首页 > 解决方案 > 动态分配结构的数组

问题描述

我很难在 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++;
      }

}

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

标签: cmalloc

解决方案


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

推荐阅读