天天看点

使用malloc的注意事项

有时候在函数中使用malloc时,如果分配内存失败,可能会打算让函数直接返回,例如:

char *T;
	T = malloc(4);
	if (T == NULL)
		return -1;
	else
		//...
           

但是如果使用了多次malloc,就要注意在判断到某次分配失败而导致返回时,将已经分配了的空间释放掉。虽然通常在函数的结尾处不会忘记free,但是却很容易在这种中途返回时疏忽了。

例如先分配了T1的空间,那么在给T2分配空间失败时,在返回之前就要free(T1):

char *T1;
	T1 = malloc(4);
	if (T == NULL)
		return -1;
	char *T2;
	T2 = malloc(4);
	if (T2 == NULL) {
		free(T1);
		return -1;
	}
		return -1;