使用C

时间:2018-04-10 06:34:57

标签: c directory segmentation-fault dirent.h

我试图删除没有系统调用且没有使用大量库的非空目录。到目前为止我的代码是......

int rmrf(char *path) {
    char* path_copy = (char *) malloc(1024 * sizeof(char));
    strcpy(path_copy, path);
    DIR *directory = opendir(path_copy);
    struct dirent *entry = readdir(directory);
    while (entry != NULL) {
        if (!strcmp(entry->d_name, ".") || !strcmp(entry->d_name, "..")) { //skip /. and /..
        } else if (entry->d_type == DT_DIR) { //directory recurse
            strcat(path_copy, "/");
            strcat(path_copy, entry->d_name);
            rmrf(path_copy);
            remove(path);
        } else { //file delete
            strcat(path_copy, "/");
            strcat(path_copy, entry->d_name);
            remove(path_copy);
        }
        entry = readdir(directory);
    }
    closedir(directory);
    return 0;
}

我当前的文件结构看起来像这样...

Who
|---Region 1
    |---County 1
        |---SubCounty 1
    |---County 2
|---Region 2
    |---County 1
|---Region 3

目前我遇到了seg故障但随着时间的推移在不同的地方。今天早些时候我会得到两个级别的递归深度然后seg故障,但到目前为止,我甚至无法完成整个级别。我无法弄清楚出了什么问题,当我使用gdb查看我得到的问题时......

malloc.c: No such file or directory.

任何帮助将不胜感激!

更新:

我已经从paxdiablo那里得到了建议,并得出了最终的函数......

int rmrf(char *path) {
    char* path_copy = malloc(1024);
    DIR *directory = opendir(path);
    struct dirent *entry = readdir(directory);
    while (entry != NULL) {
        if (!strcmp(entry->d_name, ".") || !strcmp(entry->d_name, "..")) { //skip /. and /..
        } else if (entry->d_type == DT_DIR) { //directory recurse
            strcpy(path_copy, path);
            strcat(path_copy, "/");
            strcat(path_copy, entry->d_name);
            rmrf(path_copy);
            remove(path);
        } else { //file delete
            strcpy(path_copy, path);
            strcat(path_copy, "/");
            strcat(path_copy, entry->d_name);
            remove(path_copy);
        }
        entry = readdir(directory);
    }
    closedir(directory);
    free(path_copy);
    return 0;
}
然而,虽然它在递归中越走越远,但我仍然遇到了一个seg错误。 seg故障的gdb输出如下......

Program received signal SIGSEGV, Segmentation fault.
_int_malloc (av=av@entry=0x7ffff7dd1b20 <main_arena>, bytes=bytes@entry=32816) at malloc.c:3802
3802    malloc.c: No such file or directory.
(gdb) where
#0  _int_malloc (av=av@entry=0x7ffff7dd1b20 <main_arena>, bytes=bytes@entry=32816) at malloc.c:3802
#1  0x00007ffff7a91184 in __GI___libc_malloc (bytes=32816) at malloc.c:2913
#2  0x00007ffff7ad51ba in __alloc_dir (statp=0x7fffffffe190, flags=0, close_fd=true, fd=6) at ../sysdeps/posix/opendir.c:247
#3  opendir_tail (fd=6) at ../sysdeps/posix/opendir.c:145
#4  __opendir (name=<optimized out>) at ../sysdeps/posix/opendir.c:200
#5  0x0000000000401bca in rmrf ()
#6  0x0000000000401c8d in rmrf ()
#7  0x0000000000401c8d in rmrf ()
#8  0x0000000000402380 in main ()

思想?

1 个答案:

答案 0 :(得分:4)

对于初始代码,在输入函数时执行一次

strcpy(path_copy, path);

然后对当前目录中的每个文件或目录执行此操作:

strcat(path_copy, "/");
strcat(path_copy, entry->d_name);

这意味着,如果您当前目录a中包含bc/xx文件,path_copy变量将循环显示:< / p>

/xx/a   /xx/a/b   /xx/a/b/c

而不是正确的:

/xx/a   /xx/b     /xx/c

如果文件数量足够大,您很容易就会为路径分配1024字节。

如果你想解决这个问题,那么每次都应该从头开始变量:

if ((strcmp(entry->d_name, ".") != 0) && (strcmp(entry->d_name, "..") != 0)) {
    if (entry->d_type == DT_DIR) {
        strcpy(path_copy, path);
        strcat(path_copy, "/");
        strcat(path_copy, entry->d_name);
        rmrf(path_copy);
        remove(path);
    } else {
        sprintf(path_copy, "%s/%s", path, entry->d_name);
        remove(path_copy);
    }
}

您注意到我已经稍微修改了您的初始条件,以便更有意义(如果文件既不是.也不是..,则只执行内部位)

我还在else子句中显示了使用sprintf构建要删除的字符串的简短方法,而不是一组strcpy/strcat次调用。如果您愿意,可以随意在if子句中执行此操作,我已经使用旧方法将其保留,因此您可以看到您需要做的就是添加初始路径。

只需几点,适用于您的第一个和/或第二个代码段:

  • 您还应确保免费在每个级别分配的内存,在从函数返回之前,在closedir()return之间。

  • 从不需要转换malloc的返回值,因为void *可以隐式转换为任何其他类型的指针。事实上,这样做很危险,因为它可以隐藏某些微妙的错误。

  • 同样,你从不需要乘以sizeof(char) - 也就是说,根据定义,总是一个。

  • 您可以在文件/目录检查之前将path_copy的创建移动到,因为这两个部分都很常见。

  • 最后,如果您正在处理的目录实际上并不存在,那么您将遇到麻烦,因为opendir将返回NULL并且您将立即尝试将其传递给readdir

考虑到所有这些,我开始使用以下程序,该程序实际上遍历树并打印出它找到的所有文件。一旦您对此感到满意,您可以添加删除的位:

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <dirent.h>

int rmrf(char *path) {
    char *path_copy = malloc(1024);
    DIR *directory = opendir(path);
    if (directory != NULL) {
        struct dirent *entry = readdir(directory);
        while (entry != NULL) {
            if ((strcmp(entry->d_name, ".") != 0) && (strcmp(entry->d_name, "..") != 0)) {
                sprintf(path_copy, "%s/%s", path, entry->d_name);
                if (entry->d_type == DT_DIR) {
                    rmrf(path_copy);
                    puts(path);
                } else {
                    puts(path_copy);
                }
            }
            entry = readdir(directory);
        }
        closedir(directory);
    }
    free(path_copy);
    return 0;
}

主要代码只是确保正确设置思考的驱动程序。只需确保在运行之前,您没有(在当前目录中)要保留的paxtestpaxtest2文件或目录。

int main(void) {
    system("rm -rf paxjunk");
    system("mkdir paxjunk");
    system("touch paxjunk/0.txt");
    system("mkdir paxjunk/1");
    system("touch paxjunk/1/1.txt");
    system("mkdir paxjunk/2");
    system("touch paxjunk/2/2.txt");

    rmrf("paxjunk");
    puts("===");

    system("rm -rf paxjunk2");

    rmrf("paxjunk2");
    puts("===");

    system("rm -rf paxjunk");

    return 0;
}

当你运行它时,你应该看到它正常工作:

paxjunk/0.txt
paxjunk/1/1.txt
paxjunk
paxjunk/2/2.txt
paxjunk
===
===