使用ioctl获取磁盘大小

落日映苍穹つ 2022-06-05 07:22 255阅读 0赞

获取磁盘空间的方法，大部分是根据读取/proc/mounts文件来获取得到，磁盘必须被挂载了之后才能看到，有的还是看不到的，比如：

![Center][]

但是此时在mounts中却查询不到SWAP分区的信息这样计算出来的磁盘大小就是不准确的。mounts文件内容如下：

[root@bogon ~]# cat /proc/mounts 
    rootfs / rootfs rw 0 0
    sysfs /sys sysfs rw,nosuid,nodev,noexec,relatime 0 0
    proc /proc proc rw,nosuid,nodev,noexec,relatime 0 0
    devtmpfs /dev devtmpfs rw,nosuid,size=485404k,nr_inodes=121351,mode=755 0 0
    securityfs /sys/kernel/security securityfs rw,nosuid,nodev,noexec,relatime 0 0
    tmpfs /dev/shm tmpfs rw,nosuid,nodev 0 0
    devpts /dev/pts devpts rw,nosuid,noexec,relatime,gid=5,mode=620,ptmxmode=000 0 0
    tmpfs /run tmpfs rw,nosuid,nodev,mode=755 0 0
    tmpfs /sys/fs/cgroup tmpfs ro,nosuid,nodev,noexec,mode=755 0 0
    cgroup /sys/fs/cgroup/systemd cgroup rw,nosuid,nodev,noexec,relatime,xattr,release_agent=/usr/lib/systemd/systemd-cgroups-agent,name=systemd 0 0
    pstore /sys/fs/pstore pstore rw,nosuid,nodev,noexec,relatime 0 0
    cgroup /sys/fs/cgroup/net_cls,net_prio cgroup rw,nosuid,nodev,noexec,relatime,net_prio,net_cls 0 0
    cgroup /sys/fs/cgroup/blkio cgroup rw,nosuid,nodev,noexec,relatime,blkio 0 0
    cgroup /sys/fs/cgroup/cpu,cpuacct cgroup rw,nosuid,nodev,noexec,relatime,cpuacct,cpu 0 0
    cgroup /sys/fs/cgroup/pids cgroup rw,nosuid,nodev,noexec,relatime,pids 0 0
    cgroup /sys/fs/cgroup/hugetlb cgroup rw,nosuid,nodev,noexec,relatime,hugetlb 0 0
    cgroup /sys/fs/cgroup/perf_event cgroup rw,nosuid,nodev,noexec,relatime,perf_event 0 0
    cgroup /sys/fs/cgroup/devices cgroup rw,nosuid,nodev,noexec,relatime,devices 0 0
    cgroup /sys/fs/cgroup/freezer cgroup rw,nosuid,nodev,noexec,relatime,freezer 0 0
    cgroup /sys/fs/cgroup/memory cgroup rw,nosuid,nodev,noexec,relatime,memory 0 0
    cgroup /sys/fs/cgroup/cpuset cgroup rw,nosuid,nodev,noexec,relatime,cpuset 0 0
    configfs /sys/kernel/config configfs rw,relatime 0 0
    /dev/sda3 / xfs rw,relatime,attr2,inode64,noquota 0 0
    systemd-1 /proc/sys/fs/binfmt_misc autofs rw,relatime,fd=26,pgrp=1,timeout=300,minproto=5,maxproto=5,direct 0 0
    mqueue /dev/mqueue mqueue rw,relatime 0 0
    debugfs /sys/kernel/debug debugfs rw,relatime 0 0
    hugetlbfs /dev/hugepages hugetlbfs rw,relatime 0 0
    nfsd /proc/fs/nfsd nfsd rw,relatime 0 0
    /dev/sda1 /boot xfs rw,relatime,attr2,inode64,noquota 0 0
    sunrpc /var/lib/nfs/rpc_pipefs rpc_pipefs rw,relatime 0 0
    tmpfs /run/user/0 tmpfs rw,nosuid,nodev,relatime,size=99992k,mode=700 0 0
    tmpfs /run/user/1000 tmpfs rw,nosuid,nodev,relatime,size=99992k,mode=700,uid=1000,gid=1000 0 0

这时候使用ioctl函数就比较准确了，代码如下：

#include <stdio.h>  
    #include <fcntl.h>  
    #include <linux/fs.h>  
      
    int main(void)  
    {  
        int fd;  
        //off_t size  
        unsigned long long size;  
        int len;  
        int r;  
      
        if ((fd = open("/dev/sda", O_RDONLY)) < 0)  
        {  
            printf("open error %d\n");  
            return -1;  
        }  
      
        if ((r = ioctl(fd, BLKGETSIZE64, &size)) < 0)  
        {  
            printf("ioctl error \n");  
            return -1;   
        }  
      
        len = (size>>30);  
        printf("size of sda = %d G\n", len);  
        return 0;  
    }

运行结果如下：

[root@bogon disk]# gcc disk.c 
    [root@bogon disk]# ./a.out 
    size of sda = 40 G

这里还可以指定分区获取，比如获取sda1、sda2、sda3获取到的内容和lsblk查询出来的是一样的。

除此吃之外还可以使用fdisk这个源码就比较复杂了，也可以使用lsblk，这个源码还没看过。不过看过fdisk的源码相对于ioctrl函数直接获取代码量就大了去了。

[Center]: /images/20220605/d49fb8c701b149878b1d91afc8b057dc.png