如何将一个字符指针数组从函数返回到main？

Question

我正在尝试将具有 .txt 扩展名的文件列表存储到我当前工作目录中的数组中。 我已经把数组放在一起了，现在我只需要返回它。 我是 C 新手，不习惯指针情况。 如何返回一个字符指针数组？ 我创建了数组，并根据我在目录中找到的文本文件数量进行分配。

当我尝试编译时遇到两个错误，这让我相信我对指针的理解与我想象的不同。 我也一直读到我的数组在从函数返回时会被销毁，因为它在堆栈上。 我不知道如何解决这个问题。 欢迎任何帮助或批评。

    // prototypes
    char* getLine();
    void replaceWord();
    char * getTxtFilesInDir(char*);
    bool isTxt(char *);

    int main(int argc, char **argv) {

       // Check to see if correct number of arguments is given
       if (argc < 4) {
          printf("ERROR: Not enough parameters supplied. Please supply a find,"
                "replace, and prefix parameter and try again.\n");
          return -1;
       }

       // Initialize variables
       char *find, *replace, *prefix;
       find=argv[1];
       replace=argv[2];
       prefix=argv[3];
       char cd[1024];

       // Get the Current working directory to grab files
       if (getcwd(cd, sizeof(cd)) != NULL) {
          printf("Current working dir is: %s\n", cd);
       } else {
          perror("getcwd() error");
       }

       // Get the values of the arguments
       printf("Value of find: %s\nValue of replace: %s\nValue of prefix: %s\n",
             find, replace, prefix);

       // create an array of the files that are valid
       char* files = getTxtFilesInDir(cd);

       return 0;

    }
char*  getTxtFilesInDir(char* directory) {
   DIR *d;
   struct dirent *dir;
   d = opendir(directory);
   int n=0, i=0;

   // get the number of text files in the directory
   if (d) {
      while((dir = readdir(d)) != NULL) {
         if (isTxt(dir->d_name)) {
            n++;
         }
      }
   }

   rewinddir(d);
   // Create the array of text files depending on the size
   static char* txtFiles[n];

   // add the text files to the array
   if (d) {
      printf("Found: \n");
      while ((dir = readdir(d)) != NULL)
      {
         if (isTxt(dir->d_name)) {
            printf("%s\n", dir->d_name);
            txtFiles[i]= (char*) malloc (strlen(dir->d_name)+1);
            strncpy(txtFiles[i], dir->d_name, strlen(dir->d_name));
            i++;

         }
      }
      closedir(d);
   }
   return txtFiles;
}

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

Answer 1

如果您希望getTxtFilesInDir返回一个字符串数组，它应该返回一个char** ，而不是一个char* 。

此外，您不需要将变量声明为static 。 当您希望该变量对于函数的所有调用保持相同时，您可以将函数中的变量声明为static 。 在这种情况下，这可能不是您想要做的。

无需修改太多初始代码，您可以先为一组字符串分配内存，然后在有新条目时调整其大小。 在本例中，我立即执行此操作，因为realloc(NULL, somesize)与执行malloc(somesize) 。 这就是为什么初始化*tmp = 0和txtFiles = NULL很重要，所以这个技巧有效。 您还应该传递一个指向size_t的指针，您可以在其中存储条目数：

char **getTxtFilesInDir(const char* directory, size_t *len) {
    if(directory == NULL || len == NULL)
        return NULL;

    ...

    *len = 0;
    char **txtFiles = NULL, **tmp;
    char *str;

    if (d) {
      printf("Found: \n");
      while ((dir = readdir(d)) != NULL)
      {
         if (isTxt(dir->d_name))
         {
             tmp = realloc(txtFiles, (len+1) * sizeof *textFiles);
             if(tmp == NULL)
                 return txtFiles; // return what you've got so far


             str = malloc(strlen(dir->d_name) + 1);
             if(str == NULL)
             {
                 if(txtFiles == NULL) // first time, free everything
                 {
                     free(tmp);
                     return NULL;
                 }

                 return tmp; // return all you've got so far
             }

             strcpy(str, dir->d_name); // no need of strcnpy, you've allocated
                                       // enough memory
             txtFiles = tmp;
             txtFiles[(*len)++] = tmp;
         }
      }
      closedir(d);

      return txtFiles;
}

这里的重要部分是 1. 如何使用realloc扩展内存。 然后它使用malloc为字符串分配内存。 我在txtFiles = tmp;之前这样做了 ，这样我就不必写很多if(...==NULL) 。 如果在此过程中出现问题，该函数将返回它已经存储的所有文件名。

现在main你做：

int main(int argc, char **argv)
{
    ...

    size_t len = 0;
    char **files = getTxtFilesInDir(cd, &len);

    if(file == NULL)
    {
        fprintf(stderr, "No files found\n");
        return 1;
    }

    for(size_t i = 0; i < len; ++i)
        printf("file: %s\n", files[i]);

    // freeing memory
    for(size_t i = 0; i < len; ++i)
        free(files[i]);

    free(files);

    return 0;
}

我还想评论一下您的原始复制方式：

strncpy(txtFiles[i], dir->d_name, strlen(dir->d_name));

当您想限制要复制的字节数时， strncpy非常strncpy 。 在您的情况下，您已经分配了足够的内存，因此无需分配。 并且当您使用strncpy ，限制应绑定到目标可用的字节数，而不是源可用的字节数，否则您复制的字节数可能会多于应有的字节数。 在您的情况下，您将无法获得有效的 c 字符串，因为您只能复制到dir->d_name的长度，因此不会复制'\\0'终止。

男人

#include <string.h> char *strcpy(char *dest, const char *src); char *strncpy(char *dest, const char *src, size_t n);

[...]

strncpy()函数类似，除了最多复制src n个字节。 警告：如果src的前n个字节中没有空字节，则放置在 dest 中的字符串不会以空值结尾。

当您使用strncpy您必须通过自己设置'\\0'终止字节来确保副本是有效的字符串。 因为您已经为字符串分配了strlen(dir->d_name)+1个字节：

strncpy(txtFiles[i], dir->d_name, strlen(dir->d_name));
textFiles[i][strlen(dir->d_name)] = 0;

此外，程序的退出值是一个unsigned值，它从 0 到 255。在main您应该在出错时返回1而不是-1 。