Limit max_blk to 2G by default

1 files changed, 2 insertions, 2 deletions

diff --git a/DskVIO.HC b/DskVIO.HC
index 375c3c8..f414858 100644
--- a/DskVIO.HC
+++ b/DskVIO.HC
@@ -104,7 +104,7 @@ U8 MountVirtioBlk()
   bd->type = BDT_VIRTIO_BLK;
   if (VIRTIO_BLK_MAX_BLK)
   {
-    bd->max_blk = VIRTIO_BLK_MAX_BLK;
+    bd->max_blk = Min(VIRTIO_BLK_MAX_BLK, virtio_blk.blks);
   }
   else
   {
@@ -117,4 +117,4 @@ U8 MountVirtioBlk()
   dv->data_area = bs->bitmap_sects;
   Free(bs);
   return dv->drv_let;
-}
\ No newline at end of file
+}