Question

它是我上一个问题的延伸：How can I get only txt files from directory in c?。现在我想保存那些文件名（我真的不知道它们中有多少是在dir中）到char **数组。我想出了一个解决方案（种类）然后我意识到我不需要char*而是char **（我知道，愚蠢的我：））

无论如何，我用这段代码得到了分段错误[core dumped]：

#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数组有点混淆，argh：/）

#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;
}

Answer 1

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）。

Answer 2

除了可疑的索引基里连科指出，你的第一个（和核心）错误是：

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

这似乎是无辜的，但allocMemory定义为：

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

txtFiles和moreTxtFiles都是指向指针的指针;没有指向chars的指针。您将返回一个指向单个已分配char的指针，然后将其存储在一个指针中，该指针希望它是指向指针的指针。

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

如何从c中的目录中只获取* .txt文件名并将其写入char **数组？

2 个答案: