2

它是我之前问题的扩展：如何从 c 中的目录中仅获取 txt 文件？. 现在我想将这些文件名（我真的不知道目录中有多少）保存到char **数组中。我想出了一个解决方案（有点），但后来我意识到我不需要char*但是char **（我知道，愚蠢的我 :]）

无论如何，我用这段代码得到了分段错误[核心转储]：

#include <stdio.h>
#include <fcntl.h>
#include <unistd.h>
#include <stdlib.h>
#include <dirent.h>
#include <stdbool.h>

char* allocMemory(int n)
{
    char *tab = (char *) malloc(n*sizeof(char));
    return tab;
}

void freeMemory(char **tab, int n, int m)
{
    int i=0;
    for(i=0; i<m; i++)
        free(tab[i]);
    free(tab);
    tab = NULL;
}

bool hasTxtExtension(char const *filename)
{
    size_t len = strlen(filename);
    return len > 4 && strcmp(filename + len - 4, ".txt") == 0;
}

char** getTxtFilenames(const char *dirname)
{
    DIR *directory = NULL;
    struct dirent *ent = NULL;
    int fileCounter = 0;

    char **txtFiles = allocMemory(1);
    char **moreTxtFiles = allocMemory(1);

    directory = opendir (dirname);
    if(directory == NULL)
        return NULL;
    int i = 0;

     while ((ent = readdir (directory)) != NULL)
     {
         if(hasTxtExtension(ent->d_name))
         {
             fileCounter ++;
             moreTxtFiles = (char**) realloc (txtFiles, fileCounter * sizeof(char*));

             if(moreTxtFiles[i] != NULL)
             {
                 txtFiles = moreTxtFiles;
                 txtFiles[i] = allocMemory(strlen(ent->d_name));
                 txtFiles[i][fileCounter - 1] = ent->d_name;
             }
             else
             {
                 freeMemory(txtFiles, 1, fileCounter);
                 return NULL;
             }
         }
         i ++;
     }

     if(closedir(directory) < 0)
        return NULL;

    return txtFiles;
}

int main(int argc, char **argv)
{
    char **txtFilenames = getTxtFilenames("dir");
    if(txtFilenames == NULL)
        return -1;
    printf("%s\n", txtFilenames[0][1]);
    return 0;
}

编辑：

我也试过这个：（请注意，我对 C 中那些不可爱的 char 数组有点困惑，啊 :/）

#include <stdio.h>
#include <fcntl.h>
#include <unistd.h>
#include <stdlib.h>
#include <dirent.h>
#include <stdbool.h>

char* allocMemory(int n)
{
    char *tab = (char *) malloc(n*sizeof(char));
    return tab;
}

void freeMemory(char *tab)
{
    free(tab);
    tab = NULL;
}

bool hasTxtExtension(char const *filename)
{
    size_t len = strlen(filename);
    return len > 4 && strcmp(filename + len - 4, ".txt") == 0;
}

char* getTxtFilenames(const char *dirname)
{
    DIR *directory = NULL;
    struct dirent *ent = NULL;
    int fileCounter = 0;

    char *txtFiles = NULL;
    char *moreTxtFiles = NULL;

    directory = opendir (dirname);
    if(directory == NULL)
        return NULL;

     while ((ent = readdir (directory)) != NULL)
     {
         if(hasTxtExtension(ent->d_name))
         {
             fileCounter ++;
             moreTxtFiles = (char*) realloc (txtFiles, fileCounter * sizeof(char));

             if(moreTxtFiles != NULL)
             {
                 txtFiles = moreTxtFiles;
                 txtFiles[fileCounter - 1] = ent->d_name;
             }
             else
             {
                 freeMemory(txtFiles);
                 return NULL;
             }
         }
     }

     if(closedir(directory) < 0)
        return NULL;

    return txtFiles;
}

int main(int argc, char **argv)
{
    char **txtFilenames = getTxtFilenames("dir");
    if(txtFilenames == NULL)
        return -1;
    printf("%s\n", txtFilenames[0]);
    return 0;
}
4

2 回答 2

3
  • txtFilenames[0][1]是一个字符，而不是一个字符串。
  • txtFiles[i][fileCounter - 1]是一个字符，但是ent->d_name是一个字符串。

我不明白你为什么使用两个索引（i和fileCounter）。只需使用一个字符串数组。您还应该包括<string.h>.

解决方案更简单：

#include <direct.h>
#include <stdlib.h>
#include <string.h>

char **f(const char *s)
{
    char **p = NULL;
    DIR *dir;
    struct dirent *ent;
    size_t i = 0;

    dir = opendir(s);

    while ((ent = readdir(dir)) != NULL) {
        if (hasTxtExtension(ent->d_name)) {
            p = realloc(p, (i + 1) * sizeof(char *));
            p[i] = malloc(strlen(ent->d_name) + 1);
            strcpy(p[i], ent->d_name);
            ++i;
        }
    }

    closedir(dir);
    return p;
}

以下是改进此代码的非详尽列表：

  • 处理重新分配错误：如果realloc失败，则存在内存泄漏。
  • 处理目录错误：特别是处理错误s目录路径的情况。
  • 优化重新分配（不要realloc在每次迭代中使用）。
于 2012-10-19T17:18:09.203 回答
2

除了 Kirilenko 指出的可疑索引之外，您的第一个（也是核心）错误是：

char **txtFiles = allocMemory(1);
char **moreTxtFiles = allocMemory(1);

这似乎很无辜，但 allocMemory 被定义为：

char* allocMemory(int n)
{
    char *tab = (char *) malloc(n*sizeof(char));
    return tab;
}

txtFiles 和 moreTxtFiles 都是指向指针的指针；不是指向字符的指针。您正在返回一个指向单个已分配字符的指针，然后将其存储在一个指针中，该指针期望它是一个指向指针的指针。

对于您的情况，不要使用 allocMemory 来分配列表条目。只需使用它来分配您的字符串。

于 2012-10-19T17:39:11.960 回答