我正在开发一个程序,该程序应该在注册表中搜索特定值,并将它们及其路径存储在数组中。所以我不知道程序会找到多少个键,因此我需要使用一个动态增长的数组。我现在使用这个代码,但我不确定它是否正确。
struct data
{
char * Path;
char * Key;
};
struct data **RegArray = NULL;
int ArrayCount = 0;
// ....
// ....
// search the registry here....
// value has been found, so i should add it to the array here
RegArray = ( struct data **)realloc( RegArray, ( ArrayCount + 1 ) * sizeof( struct data *) );
RegArray[ ArrayCount ] = ( struct data *)malloc( sizeof( struct data ) );
RegArray[ ArrayCount ]->Path = _strdup( CurrentPath );
RegArray[ ArrayCount ]->Key = _strdup( CurrentKey );
ArrayCount++;
有人能告诉我这是否可以吗。如果没有,我应该如何正确地做?
谢谢!
您已经了解了它的要点。但是,您应该做一些改进:
-
不要强制转换
malloc
、realloc
、calloc
等的返回值:RegArray[ ArrayCount ] = ( struct data *)malloc( sizeof( struct data ) );
变成。。。
RegArray[ ArrayCount ] = malloc( sizeof( struct data ) );
-
为了防止内存泄漏,在检查是否成功后,在分配到预期位置之前,始终将
realloc
分配给临时变量:RegArray = ( struct data **)realloc( RegArray, ( ArrayCount + 1 ) * sizeof( struct data *) );
变成。。。
struct data **tmp = realloc( RegArray, ( ArrayCount + 1 ) * sizeof( struct data *) ); if (tmp == NULL) { /* handle error case */ } RegArray = tmp;
-
始终检查
malloc
、realloc
、calloc
等的返回值:RegArray[ ArrayCount ] = ( struct data *)malloc( sizeof( struct data ) );
变成。。。
RegArray[ ArrayCount ] = malloc( sizeof( struct data ) ); if (RegArray[ ArrayCount ] == NULL) { /* handle error case */ }
-
使用
sizeof
时,请使用变量而不是类型。为了提高可读性,我通常还会在sizeof
中的表达式周围去掉无用的括号:RegArray[ ArrayCount ] = malloc( sizeof( struct data ) );
变成。。。
RegArray[ ArrayCount ] = malloc( sizeof **RegArray );
列表方式:
struct Data
{
char * Path;
char * Key;
Data * next;
};
void deallocate(Data *ptr){
free(ptr->Path);
free(ptr->Key);
free(ptr);
}
Data *removeElement(Data *list, char *Key){
Data *ptr = list;
Data *prev = NULL;
while(ptr != NULL){
if(strcmp(Key,ptr->Key) == 0){
if(prev != NULL){
prev->next = ptr->next;
deallocate(ptr);
}
else{
prev = ptr;
list = ptr->next;
deallocate(prev);
}
}
else{
ptr = ptr->next;
}
}
return list;
}
Data * addElement(Data *list, char *path, char *key){
if(list == NULL) {
list = (Data *)malloc(sizeof(Data));
return list;
}
Data *cursor = list;
while(cursor != NULL){
cursor = cursor->next;
}
cursor = (Data *)malloc(sizeof(Data));
cursor->next = NULL;
cursor->path = path;
cursor->key = key;
return list;
}
int main(){
Data *list = NULL;
// value has been found
list = addElement(list,path,key);
return 0;
}