递归获取目录的大小 - C

Question

我正在尝试以递归方式获取目录的大小，但我只会遇到段错误。我真的看不出我错在哪里，有人可以帮助我吗？ 附言我不需要验证文件是否存在，这只是我必须编写的另一个函数的尝试。

代码如下：

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

int main(int argc, char * argv[])
{
  printf("%d\n", size(argv[1]));
  return 0;
}

int is_folder(char * path)
{
    struct stat path_stat;
    stat(path, &path_stat);
    return !(S_ISREG(path_stat.st_mode));
}

int size(char * name)
{
  int dir_size = 0;
  struct dirent * pDirent;
  DIR * pDir = opendir(name);
  while ((pDirent = readdir(pDir)) != NULL)
  {
    char buf[PATH_MAX + 1];
    realpath(pDirent->d_name, buf);
    if (is_folder(buf))
    {
      size(buf);
    }
    else
    {
      struct stat st;
      stat(buf, &st);
      int sz = st.st_size;
      dir_size += sz;
    }
  }
   return dir_size;
}

Answer 1

基于您的代码，下面是我能够重现 du -sk 返回的最接近的内容：

#include <stdio.h>
#include <sys/stat.h>
#include <dirent.h>
#include <limits.h>
#include <string.h>

off_t directorySize(char *directory_name)
{
    off_t directory_size = 0;

    DIR *pDir;

    if ((pDir = opendir(directory_name)) != NULL)
    {
        struct dirent *pDirent;

        while ((pDirent = readdir(pDir)) != NULL)
        {
            char buffer[PATH_MAX + 1];

            strcat(strcat(strcpy(buffer, directory_name), "/"), pDirent->d_name);

            struct stat file_stat;

            if (stat(buffer, &file_stat) == 0)
            {
                directory_size += file_stat.st_blocks * S_BLKSIZE;
            }

            if (pDirent->d_type == DT_DIR)
            {
                if (strcmp(pDirent->d_name, ".") != 0 && strcmp(pDirent->d_name, "..") != 0)
                {
                    directory_size += directorySize(buffer);
                }
            }
        }

        (void) closedir(pDir);
    }

    return directory_size;
}

int main(int argc, char *argv[])
{
    printf("%lldKiB\n", directorySize(argv[1]) / 1024);

    return 0;
}

我猜您看到的差异是由于目录占用了少量空间，您需要将这些空间包含在您的总数中，并且文件空间使用是分块的，因此您需要计算块数，而不是字节数。

Answer 2

花了一晚尝试如何找到与 du -sh 报告的大小相同的大小：我写了这个，它似乎也适用于大型目录（例如 750MB 或更大的 linux 源)。

#include <stdio.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include <string.h>
#include <dirent.h>
#include <stdlib.h>
#include <limits.h>

long folder_size(char *);

int main(int argc, char * argv[])
{
  if (argc < 2)
  {
    exit(1);
  }
  printf("%ld\n", mysize(argv[1]));
  return 0;
}

int not_folder(char * path)
{
    struct stat path_stat;
    stat(path, &path_stat);
    return (S_ISREG(path_stat.st_mode));
}

long folder_size(char * name)
{
  long dir_size = 0L;
  struct dirent * pDirent;
  DIR * pDir = opendir(name);
  while ((pDirent = readdir(pDir)))
  {
    if (strcmp (pDirent->d_name, "..") != 0 && strcmp (pDirent->d_name, ".") != 0)
    {
      char buf[PATH_MAX];
      strcpy(buf, name);
      strcat(buf, "/");
      strcat(buf, pDirent->d_name);
      if (not_folder(buf))
      {
        struct stat st;
        stat(buf, &st);
        dir_size += st.st_blocks * S_BLKSIZE;
        printf("%s %ld\n", buf, (long)st.st_size);
      }
      else
      {
        dir_size += folder_size(buf);
      }
    }
  }
  (void) closedir(pDir);
  return dir_size;
}