Question

我目前正在尝试制作一个c程序，该程序计算给定目录的总文件大小（以字节为单位），包括目录本身，其中的所有文件以及所有子目录中的所有文件和目录。本质上，系统要求我将du -b的替代程序编写为命令。

我以为我有一个可行的解决方案，但是在第一个目录之后，该程序认为所有更深层的条目都是目录，即使它们只是常规文件也是如此。这包括当我直接给它提供一个深度更深的目录时，例如，给它提供输入./Directory1而不是仅仅给.。

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

int du_function(char direc[]) {
    int total = 0;
    char str[100];
    strcpy(str, direc);
    struct stat sfile;
    struct dirent *de;
    DIR *dr = opendir(direc);
    if (dr == NULL) {
        printf("Could not open directory\n");
        return 0;
    }
    while ((de = readdir(dr)) != NULL) {
        printf("%s\n", de->d_name);
        stat(de->d_name, &sfile);
        if (S_ISDIR(sfile.st_mode) && strcmp(de->d_name, ".") != 0 &&
            strcmp(de->d_name, "..") != 0) {
            strcat(str, "/");
            strcat(str, de->d_name);
            printf("This is a directory called %s\n", str);
            total = total + du_function(str);
            strcpy(str, direc);
        }
        if (S_ISREG(sfile.st_mode) || strcmp(de->d_name, ".") == 0) {
            printf("Size in bytes = %ld\n", sfile.st_size);
            total = total + sfile.st_size;
            printf("The total is %d bytes so far.\n", total);
        }
    }
    printf("The total is %d bytes.\n", total);
    closedir(dr);
    return total;
}

int main(int argc, char *argv[]) {
    if (argc < 2) {
        char cwd[1] = ".";
        du_function(cwd);
    } else
        du_function(argv[1]);
    return 0;
}

我在这里已经尽力了，并尝试了各种解决方案，但是出于某种原因，S_ISDIR必须将事物标识为不是（或者很可能从我那里接收到错误输入）的目录。）

Answer 1

char cwd[1] = ".";对于 string 而言太小。 空字符没有空间。

使用[]，让编译器确定 string 所需的大小。

char cwd[] = ".";

stat()调用基于本地名称而不是完整路径。

#if 0
    stat(de->d_name, &sfile);
#else
    char buf[500];
    snprintf(buf, sizeof buf, "%s/%s", direc, de->d_name);
    stat(buf, &sfile);
#endif

（已删除）@PSkocik

中的相似发现

也许还有其他问题。

Answer 2

就像您构造递归调用的路径一样，您必须使用stat系统调用的前导目录构造文件名。您的程序仅适用于当前目录中的文件。

修复程序的有效方法是将足够大的缓冲区传递给递归函数，并逐步构建路径和文件名：

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

long int du_function(char path[], size_t size) {
    size_t len = strlen(path);
    long int total = 0;
    struct stat sfile;
    struct dirent *de;
    DIR *dr = opendir(path);
    if (dr == NULL) {
        printf("Could not open directory %s\n", path);
        return 0;
    }
    while ((de = readdir(dr)) != NULL) {
        if (strcmp(de->d_name, "..") == 0)
            continue;
        //printf("%s\n", de->d_name);
        if (snprintf(path + len, size - len, "/%s", de->d_name) > (int)(size - len)) {
            path[len] = '\0';
            printf("Path too long: %s/%s\n", path, de->d_name);
            continue;
        }
        stat(path, &sfile);
        if (S_ISDIR(sfile.st_mode) && strcmp(de->d_name, ".") != 0) {
            //printf("This is a directory called %s\n", path);
            total = total + du_function(path, size);
        } else
        if (S_ISREG(sfile.st_mode) || strcmp(de->d_name, ".") == 0) {
            //printf("Size in bytes = %ld\n", (long)sfile.st_size);
            total = total + sfile.st_size;
            //printf("The total is %ld bytes so far.\n", total);
        }
    }
    path[len] = '\0';
    printf("%ld\t%s\n", total, path);
    closedir(dr);
    return total;
}

int main(int argc, char *argv[]) {
    char path[1024] = ".";
    if (argc > 1) {
        strcpy(path, argv[1]);
    }
    du_function(path, sizeof path);
    return 0;
}

尝试在目录和所有子目录中以字节为单位打印总空间

2 个答案: