From: shenmeng996 <[email protected]>

fstat return block device's size of zero.
use ioctl to get block device's size.

Signed-off-by: shenmeng996 <[email protected]>
---
 lib/io.c | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/lib/io.c b/lib/io.c
index 93328d3..ae3ac37 100644
--- a/lib/io.c
+++ b/lib/io.c
@@ -9,6 +9,8 @@
 #define _LARGEFILE64_SOURCE
 #define _GNU_SOURCE
 #include <sys/stat.h>
+#include <sys/ioctl.h>
+#include <linux/fs.h>
 #include "erofs/io.h"
 #ifdef HAVE_LINUX_FALLOC_H
 #include <linux/falloc.h>
@@ -49,7 +51,12 @@ int dev_open(const char *dev)
 
        switch (st.st_mode & S_IFMT) {
        case S_IFBLK:
-               erofs_devsz = st.st_size;
+               ret = ioctl(fd, BLKGETSIZE64, &erofs_devsz);
+               if (ret) {
+                       erofs_err("failed to ioctl(%s).", dev);
+                       close(fd);
+                       return -errno;  
+               }
                break;
        case S_IFREG:
                ret = ftruncate(fd, 0);
-- 
1.8.3.1

Reply via email to