我建议让用户指定应考虑哪些挂载点的总数,或者使用一种启发式方法来省略系统和临时挂载点。
请考虑以下示例: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
的长期使用。
在守护进程中,可以使用HUP
、USR1
或USR2
信号来指示用户何时要重新加载配置——挂载点列表,在此处。我不认为将其集成到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)
时,这两个路径才在同一个挂载点下。(一个设备可能会被多次挂载到不同的点上,例如使用
绑定挂载,但通常上述检查就足够了。)
如果您觉得以上代码太过繁琐,您可以使用
df
工具。为确保输出结果是以C/POSIX本地环境格式(而非其他语言如法语或芬兰语)显示,请使用:
handle = popen("LANG=C LC_ALL=C df -Pl", "r")
或类似语句,使用
len = getline(&line, &size, handle)
读取输出。
df
命令中添加-P
选项。 - Cyrus