c - 动态分配结构体数组

Question

我在处理 C 中的 malloc 时遇到了困难，尤其是在分配结构数组时。我有一个程序，基本上将所有文件名和文件大小存储在一个结构数组中。我的程序在不使用 malloc 的情况下运行，但我不太喜欢这种编程方式。我可以在程序中使用 malloc 获得任何帮助吗？

int getNumberOfFiles(char *path)
{
 int totalfiles = 0;
 DIR *d;
 struct dirent *dir;
 struct stat cstat;
 d = opendir(path);
 while(( dir = readdir(d)) != NULL) {
     totalfiles++;
}
return totalfiles;
}

int main()
{

      int totalfiles = getNumberOfFiles(".");
      int i =0;
      DIR *d;
      struct dirent *dir; 
      d = opendir(".");
      struct fileStruct fileobjarray[totalfiles];
      struct stat mystat;

       while(( dir = readdir(d)) != NULL) {
        fileobjarray[i].filesize=mystat.st_size;
        strcpy (fileobjarray[i].filename ,dir->d_name );
        i++;
      }

}

如您所见，我创建了一个名为 getnumberoffiles() 的函数来获取静态分配的大小。

Answer 1

struct fileStruct fileobjarray[totalfiles];

fileobjarray 是 VLA fileStruct 结构的大小为 Totalfiles 的数组。使用 malloc 进行分配我们可以写:

struct fileStruct *fileobjarray = malloc(totalfiles * sizeof(*fileobjarray));

我们将为 totalfiles 元素数组分配内存，每个元素的 sizeof(*fileobjarray) = sizeof(struct fileStruct) 大小。有时calloc在 C 中更可取，因为它可以(在某些平台上)防止溢出:

struct fileStruct *fileobjarray = calloc(totalfiles, sizeof(*fileobjarray));

并记住free()。

int getNumberOfFiles(char *path)
{
 int totalfiles = 0;
 DIR *d;
 struct dirent *dir;
 struct stat cstat;
 d = opendir(path);
 while(( dir = readdir(d)) != NULL) {
     totalfiles++;
}
return totalfiles;
}

int main()
{

      int totalfiles = getNumberOfFiles(".");
      int i =0;
      DIR *d;
      struct dirent *dir; 
      d = opendir(".");
      struct fileStruct *fileobjarray = calloc(totalfiles, sizeof(*fileobjarray));
      if (fileobjarray == NULL) {
            fprintf(stderr, "Error allocating memory!\n");
            return -1;
      }
      struct stat mystat;

      while(( dir = readdir(d)) != NULL) {
        fileobjarray[i].filesize=mystat.st_size;
        strcpy (fileobjarray[i].filename ,dir->d_name );
        i++;
      }
      free(fileobjarray);
}