欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

TIL 20:使用malloc的注意事项

程序员文章站 2022-03-22 22:21:52
...

有时候在函数中使用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;