如何在linux中以编程方式获取dir的大小？

Question

我想通过 C 程序获取 linux 中特定目录的确切大小。 我尝试使用 statfs(path,struct statfs &) 但它没有给出确切的大小。 我也尝试过 stat() 但它返回任何目录的大小为 4096 ！

请建议我如何获取 dir 的确切大小，就像我们在“du -sh dirPath”命令之后得到的那样。

我也不想通过 system() 使用 du。

提前致谢。

Answer 1

基于 Jim Plank 的 示例 帮助您入门：

#include <dirent.h>
#include <stdio.h>
#include <stdlib.h>
#include <sys/stat.h>

int main( int argc, char **argv ) {
  DIR *d = opendir( "." );

  if( d == NULL ) {
    fprintf( stderr, "Cannot open current working directory\n" );
    return 1;
  }

  struct dirent *de;
  struct stat buf;
  int total_size = 0;

  for( de = readdir( d ); de != NULL; de = readdir( d ) ) {
    int exists = stat( de->d_name, &buf );

    if( exists < 0 ) {
      fprintf( stderr, "Cannot read file statistics for %s\n", de->d_name );
    } else {
      total_size += buf.st_size;
    }
  }

  closedir( d );
  printf( "%d\n", total_size );

  return 0;
}

注释、注意事项和读者问题：

• 此示例不完整。 请参阅 Plank 的注释更多细节。
• 如果有锁定的文件会怎样？
• 符号链接是否需要特殊处理（以避免无限循环）？
• 如何输出错误文件的完整路径名？

这个答案是一个起点，而不是一个完整且强大的计算目录大小的程序。 如果您需要更多帮助，请阅读 du 程序的源代码。

Answer 2

您需要 stat() 当前目录和子目录中的所有文件并将它们相加。

考虑为此使用递归算法。

Answer 3

我想这个解决方案对于那些仍然可能遇到问题的人来说可能会有用。

这是为模仿 Linux du 程序而编写的函数。 它递归地遍历所有目录并累加文件大小。

注意，这个函数仍然不完整，因为它在硬链接上的行为不正确。 人们应该添加一个容器来存储指向同一 inode 实体的文件描述符，并使用它来消除同一文件的多个计数。 lstat() 用于处理符号链接（又名软链接），硬链接是这里的一个问题。

size_t countDiskUsage(const char* pathname)
{
  if (pathname == NULL) {
    printf("Erorr: pathname is NULL\n");
  }

  struct stat stats;

  if (lstat(pathname, &stats) == 0) {
    if (S_ISREG(stats.st_mode)){
      return stats.st_size;
    }
  } else {
    perror("lstat\n");
  }

  DIR* dir = opendir(pathname);

  if (dir == NULL) {
    perror("Error");
    return 0;
  }

  struct dirent *dirEntry;
  size_t totalSize = 4096;

  for (dirEntry = readdir(dir); dirEntry != NULL; dirEntry =   readdir(dir)) {
    long pathLength = sizeof(char) * (strlen(pathname) + strlen(dirEntry->d_name) + 2);
    char* name = (char*)malloc(pathLength);
    strcpy(name, pathname);
    strcpy(name + strlen(pathname), "/");
    strcpy(name + strlen(pathname) + 1, dirEntry->d_name);

    if (dirEntry->d_type == DT_DIR) {
      if (strcmp(dirEntry->d_name, ".") != 0 && strcmp(dirEntry->d_name, "..") != 0) {
        totalSize += countDiskUsage(name);
      }
    } else {
      int status = lstat(name, &stats);
      if (status == 0) {
        totalSize += stats.st_size;
      } else {
        perror("lstat\n");
      }
    }
    free(name);
  }

  closedir(dir);

  return totalSize;
}

Answer 4

如果您不想使用 'system'，但可以使用 'pipe'、'fork'、'execlp ' 和 'du'，您可以构建一个管道，分叉一个新进程，重定向管道中子进程的 'STDOUT'，执行 'du' 在子级中，并在父级中读取结果。 示例代码如下：

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>

int main(void) {
  int pfd[2], n;
  char str[1000];

  if (pipe(pfd) < 0) {
    printf("Oups, pipe failed.  Exiting\n");
    exit(-1);
  }

  n = fork();

  if (n < 0) {
    printf("Oups, fork failed.  Exiting\n");
    exit(-2);
  } else if (n == 0) {
    close(pfd[0]);

    dup2(pfd[1], 1);
    close(pfd[1]);

    execlp("du", "du", "-sh", "/tmp", (char *) 0);
    printf("Oups, execlp failed.  Exiting\n"); /* This will be read by the  parent. */
    exit(-1); /* To avoid problem if execlp fails, especially if in a loop. */
  } else {
    close(pfd[1]);

    n = read(pfd[0], str, 1000); /* Should be done in a loop until read return 0, but I am lazy. */
    str[n] = '\0';

    close(pfd[0]);
    wait(&n); /* To avoid the zombie process. */

    if (n == 0) {
       printf("%s", str);
    } else {
       printf("Oups, du or execlp failed.\n");
    }
  }
}