c代码获取二级存储大小

标签 c linux bash ipc

我需要一种简单的方法来在 linux 的(守护程序)c 代码中获取二级存储详细信息(如总大小、已用空间和可用空间);

这是我尝试过的东西

  • statvfs - 不知道如何获取磁盘详细信息而不是文件
  • 在 C 代码中使用 system("df -h --total | grep total > disk.stat") 然后读取文件。

但上面涉及文件写入和读取,效率不高,因为此 C 代码是一个守护进程,它将不断轮询系统详细信息作为图形生成的输入。

如果没有其他方法,请告诉我一个简单快速的 ipc 机制,并举例说明此 bash 和 c 代码之间的通信。

/*
* @breif returns total percentage of secondary storage used
*
*   - uses bash command to get storage data and store in a file
*   - and use c code retrive the percent of usage from file and return it
*/
int calculate_storage_size( )
{
  if ( system("df -h --total | grep total > disk.stat") >= 0 )
  {
    char *temp_char_ptr = (char *)NULL;
    int storage_size_percent = -1;  
    FILE *fp ;
    fp = fopen ("disk.stat" , "r");
    if (fp != (FILE *)NULL)
    {
      temp_char_ptr = (char*) calloc ( 6 , 1 );
      fscanf( fp,"%s %s %s %s %d", temp_char_ptr, temp_char_ptr, temp_char_ptr, temp_char_ptr, &storage_size_percent);
    }
    free (temp_char_ptr);
    fclose(fp);
    return storage_size_percent;
  }
  return -1;
}

提前致谢和问候

最佳答案

我建议最好让用户指定应该在总数中考虑哪些挂载,或者使用启发式方法来省略系统和临时挂载。

考虑以下示例,info.c:

#define  _POSIX_C_SOURCE 200809L
#define  _GNU_SOURCE
#include <stdlib.h>
#include <stdint.h>
#include <inttypes.h>
#include <sys/statvfs.h>
#include <mntent.h>
#include <string.h>
#include <errno.h>
#include <stdio.h>

static void free_array(char **array)
{
    if (array) {
        size_t i;    
        for (i = 0; array[i] != NULL; i++) {
            free(array[i]);
            array[i] = NULL;
        }
        free(array);
    }
}

static char **normal_mounts(void)
{
    char         **list = NULL, **temp;
    size_t         size = 0;
    size_t         used = 0;
    char           buffer[4096];
    struct mntent  entry;
    FILE          *mounts;

    mounts = fopen("/proc/mounts", "r");
    if (!mounts)
        return NULL;

    while (getmntent_r(mounts, &entry, buffer, sizeof buffer) == &entry)
        if (strcmp(entry.mnt_fsname, "tmpfs") &&
            strcmp(entry.mnt_fsname, "swap")  &&
            strcmp(entry.mnt_dir, "/proc")    && strncmp(entry.mnt_dir, "/proc/", 6) &&
            strcmp(entry.mnt_dir, "/boot")    && strncmp(entry.mnt_dir, "/boot/", 6) &&
            strcmp(entry.mnt_dir, "/sys")     && strncmp(entry.mnt_dir, "/sys/", 5) &&
            strcmp(entry.mnt_dir, "/run")     && strncmp(entry.mnt_dir, "/run/", 5) &&
            strcmp(entry.mnt_dir, "/dev")     && strncmp(entry.mnt_dir, "/dev/", 5) &&
            strcmp(entry.mnt_dir, "/mnt")     && strncmp(entry.mnt_dir, "/mnt/", 5) &&
            strcmp(entry.mnt_dir, "/media")   && strncmp(entry.mnt_dir, "/media/", 7) &&
            strcmp(entry.mnt_dir, "/var/run") && strncmp(entry.mnt_dir, "/var/run/", 9)) {

            if (used >= size) {
                size = (used | 15) + 17;
                temp = realloc(list, size * sizeof list[0]);
                if (!temp) {
                    endmntent(mounts);
                    free_array(list);
                    errno = ENOMEM;
                    return NULL;
                }
                list = temp;
            }

            if (!(list[used++] = strdup(entry.mnt_dir))) {
                endmntent(mounts);
                free_array(list);
                errno = ENOMEM;
                return NULL;
            }
        }

    if (ferror(mounts) || !feof(mounts)) {
        endmntent(mounts);
        free_array(list);
        errno = EIO;
        return NULL;
    } else
        endmntent(mounts);

    if (!used) {
        free_array(list);
        errno = 0;
        return NULL;
    }

    if (size != used + 1) {
        size = used + 1;
        temp = realloc(list, size * sizeof list[0]);
        if (!temp) {
            free_array(list);
            errno = ENOMEM;
            return NULL;
        }
        list = temp;
    }

    list[used] = NULL;
    errno = 0;
    return list;
}

static int statistics(const char **mountpoint, uint64_t *bytes_total, uint64_t *bytes_free)
{
    struct statvfs  info;
    uint64_t        btotal = 0;
    uint64_t        bfree = 0;
    size_t          i;

    if (!mountpoint)
        return errno = EINVAL;

    for (i = 0; mountpoint[i] != NULL; i++)
        if (statvfs(mountpoint[i], &info) != -1) {
            btotal += (uint64_t)info.f_frsize * (uint64_t)info.f_blocks;
            bfree  += (uint64_t)info.f_bsize * (uint64_t)info.f_bavail;
        } else
            return errno;

    if (bytes_total)
        *bytes_total = btotal;
    if (bytes_free)
        *bytes_free = bfree;

    return 0;
}

int main(int argc, char *argv[])
{
    uint64_t total = 0;
    uint64_t nfree = 0;

    if (argc > 1) {
        if (statistics((const char **)argv + 1, &total, &nfree)) {
            fprintf(stderr, "%s.\n", strerror(errno));
            return EXIT_FAILURE;
        }
    } else {
        char **mounts = normal_mounts();
        size_t i;
        if (!mounts) {
            if (errno)
                fprintf(stderr, "Error determining file systems: %s.\n", strerror(errno));
            else
                fprintf(stderr, "No normal file systems found.\n");
            return EXIT_FAILURE;
        }
        fprintf(stderr, "Considering mount points");
        for (i = 0; mounts[i] != NULL; i++)
            fprintf(stderr, " %s", mounts[i]);
        fprintf(stderr, "\n");

        if (statistics((const char **)mounts, &total, &nfree)) {
            fprintf(stderr, "%s.\n", strerror(errno));
            return EXIT_FAILURE;
        }
        free_array(mounts);
    }

    printf("%20" PRIu64 " bytes total\n", total);
    printf("%20" PRIu64 " bytes free\n",  nfree); 
    return EXIT_SUCCESS;
}

statistics() 函数接受一个以 NULL 结尾的挂载点数组,以及两个指向无符号 64 位整数的指针。如果成功,该函数返回 0,否则返回非零 errno 代码。如果成功,该函数会将文件系统中的总字节数设置为第一个整数,并将空闲字节数设置为第二个整数。

如果您提供一个或多个安装点作为命令行参数,则只会考虑这些安装点。 (POSIX 表示 argv[argc] == NULL,所以这种用法是安全的。)

否则,normal_mounts() 函数用于解析/proc/mounts 以获得“正常”挂载点的列表。该函数使用 getmntent()从内核提供的伪文件中读取每个条目(行)。所有 tmpfs (ramdisks) 和 swap 文件系统都被排除在外,安装在 /proc/boot, /sys, /run, /dev, /mnt, /media > 和 /var/run。这只是一种粗略的启发式方法,不是已知的好方法。

在守护进程中,甚至在图形应用程序中,您只调用(等同于)statistics() 函数,并使用相同的挂载点数组。您甚至可以考虑单独跟踪每个挂载点,让用户过滤并组合他们感兴趣的信息。事实上,我建议:我个人可能有兴趣查看我的临时文件使用情况的波动(在/tmp/var/tmp 是 tmpfs 挂载),以及跟踪我对 /home 的长期使用情况。

在守护进程中,您可以使用 HUPUSR1USR2 信号来指示用户何时希望您重新加载配置 --挂载点列表,在这里。我不认为将它集成到 DBUS 以检测可移动媒体安装/卸载会很有趣,但如果您认为它有用,当然可以。

如果您使用例如编译上述程序

gcc -Wall -O2 info.c -o info

然后运行

./info

它会输出类似的东西

Considering mount points / /home
        119989497856 bytes total
         26786156544 bytes free

其中第一行输出到标准错误,字节行输出到标准输出。您还可以具体命名挂载点——确保它们不同,因为代码不会检查重复的挂载——:

./info /home /tmp

如果您想知道如何确定两个目录是否在同一个挂载上:对一个目录调用 stat(path1, &info1),然后调用 stat(path2, &info2) 另一方面。当且仅当 (info1.st_dev == info2.st_dev) 时,这两条路径在同一个挂载上。 (一个设备可以在不同的点多次安装,例如使用绑定(bind)安装,但通常上述检查就足够了。)

如果您觉得以上所有代码都很烦人,您可以随时依赖 df 实用程序。为确保输出在 C/POSIX 语言环境中(而不是法语或芬兰语),请使用

handle = popen("LANG=C LC_ALL=C df -Pl", "r");

或类似的,并使用 len = getline(&line, &size, handle) 读取输出。

关于c代码获取二级存储大小,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/39825929/

相关文章:

bash - 如何检查是否以 root 身份在 bash 脚本中运行

c - 无符号字符左移查询

python - 如何用Python将不同类型的数据一起打包成二进制数据

c - 未初始化时检查指针是否为空

linux - 获取创建日期大于某个日期 linux 的文件列表

c - "__gmon_start__"符号是什么?

linux - 有没有办法以非root用户身份在linux上启动/重启/停止apache服务器?

c - C 中字符数据的二维数组的“压缩”输入

linux - 检查文件在 FTP 服务器上是否可用

bash - 从文本文件中删除除第一行和每第 n 行之外的行