检查目录路径以“.”、“..”结尾

Question

我有一个编程问题，希望有人能帮我解决。我正在尝试为工作中的一项任务学习 C 编程，并且我为自己设置了一个小项目，其中包括阅读文件树，其中包括获取有关每个文件的信息的所有子目录。

我遇到的问题是我的程序不会忽略以 / 结尾的目录路径。或 /.. 当它打印所有目录时，我想在子目录前面留出空间以提高可读性。

所以这部分出现了错误：

int isDir(const char *parent, char *name) {

    struct stat st_buf; // file info
    char buf[BUF_SIZE];

    if (strcmp(name, ".") == 0 || strcmp(name, "..") == 0) {
        return 0;
    }   
    char *path = malloc(strlen(name) + strlen(parent) + 2);
    //sprintf(char *buf, const char *format, [arg1],[arg2],...)
    sprintf(path, "%s/%s", parent, name);
    stat(path, &st_buf); //

    return S_ISDIR(st_buf.st_mode); //directory
}

这是主函数和列表函数：

int list(const char *name) {
    DIR *dirp = opendir(name);
    struct dirent *dentry;
    char buf[BUF_SIZE];

    while ((dentry = readdir(dirp)) != NULL) {
    char *dir_name = dentry->d_name;        

            printf(" %s\n", dir_name);

        //if it's dir, then go into dir
        if (isDir(name, dir_name)) { //name : parent, dir_name : child
            chdir(dir_name);
            getcwd(buf, BUF_SIZE);
            list(buf);
        }
    }
    closedir(dirp);
}

int main()
{
    list(".");
    return 0;
}

结果是这样的：

hm1.c
Data
lab1.txt
result1
lab3.txt
.
..
.
..
result2
lab3.txt
.
..
result3
lab3.txt
.
..
a.c
.
..
a.out

我要打印的结果

hm1.c
Data
    lab1.txt
    result1
        lab3.txt
    result2
        lab3.txt
    result3
        lab3.txt
a.c
a.out

Answer 1

如果您有 . 或 ..，则您的 isDir 返回 true/false，在其他情况下返回 false（或零），然后在其他情况下返回 S_ISDIR 的 true/false 值

您真正需要的是返回 SKIP、isFILE 或 isDIR 3 个值之一的函数，然后在此基础上编写打印逻辑。

您还需要修复内存泄漏

还要注意chdir(dir_name); 会更改进程的实际目录，因此一旦您在循环中从list 返回，您将无法再打开您正在循环的文件或目录（因为您现在在不同的目录中）

这将解决您的问题并打印您想要的格式

enum { doSkip, isFile, isDir } testDir(char *path, char *name) 
{ 
     struct stat st_buf;        
     if (strcmp(name, ".") == 0 || strcmp(name, "..") == 0) {
         return doSkip;
     }   
     stat(path, &st_buf);
     if (S_ISDIR(st_buf.st_mode))
         return isDir;
     return isFile;
}

void list(const char *path, int indentlevel)
{
     DIR *dirp = opendir(path);
     struct dirent *dentry;
     char buf[10000]; // Lets just make the buffer sufficently big for this example
     if (!dirp) {
      printf("%*sNo access\n",indentlevel,"");
      return;
     }

     while ((dentry = readdir(dirp)) != NULL) {

        sprintf(buf,"%s/%s", path, dentry->d_name);
        switch (testDir(buf,dentry->d_name)) {
        case doSkip:
           /* do nothing */
           break;
        case isDir:
           printf("%*s%s:\n",indentlevel,"",dentry->d_name);
           list(buf,indentlevel+4);
           break;
        case isFile:
           printf("%*s%s\n",indentlevel,"",dentry->d_name);
           break;
        }
     }
     closedir(dirp);
}

int main()
{
     list(".", 0);
     return 0;
}

Answer 2

如果您愿意使用 C++，另一种方法是使用std::experimental::filesystem，也（主要）称为Boost.Filesystem。有了这个，你会做这样的事情：

#include <experimental/filesystem> // Could substitute <boost/filesystem.hpp>
#include <boost/range/iterator_range.hpp>
#include <iostream>

using namespace std::experimental;

int main(int argc, char *argv[]) 
{
    const auto path = filesystem::path{ argc > 1 ? argv[1] : "." };

    if( filesystem::is_directory(path) ) 
    {
        std::cout << path << " is a directory containing:\n";

        for( const auto& entry : boost::make_iterator_range( filesystem::recursive_directory_iterator{path}, {} ) )
        {
            std::cout << entry << "\n";
        }
    }
}

看到它运行here。请注意，目录迭代器会自动跳过. 和..。