通过函数给指针申请内存说到底就是一个参数传递的问题,很简单但是也很容易犯错误。
一、例子(一) 使用一级指针作为参数
#include "stdio.h"
#include "malloc.h"
void mem_malloc(int *p, int num)
{
p = (int *)malloc(num * sizeof(int));
return;
}
void main()
{
int *pi;
mem_malloc(pi, 10);
}
细心的读者朋友应该发现,上面代码存在着问题。mem_malloc函数里的这个参数只是pi的一个副本,在mem_malloc函数调用结束后就没了,malloc分配的内存首地址也没带出来,pi还是原来的pi没变。函数调用结束后,由于没有带回首地址,申请的堆内存无法释放,还发生了内存泄漏。
如果要用一级指针实现参数传递,可以按照如下方式修改。
#include "stdio.h"
#include "malloc.h"
int* mem_malloc(int *p, int num)
{
p = (int *)malloc(num * sizeof(int));
return p;
}
void main()
{
int *pi;
pi = mem_malloc(pi, 10);
free(pi);
pi = NULL;
}
这样看起来是返回了一个局部变量的地址,但这个地址是个堆地址,函数结束后任然有效,使用完之后一定要记得释放,不然会引起内存泄漏。
二、例子(二) 使用二级指针作为参数
如下例子在调用memory_alloc函数后,申请的内存首地址也带回了,使用完之后释放内存。
#include "stdio.h"
#include "malloc.h"
#include "string.h"
void memory_alloc(void **pp, int num)
{
*pp= (void *)malloc(num * sizeof(int));
return;
}
void main()
{
char *p;
memory_alloc((void **)&p, 20);
memset(p,0x00,10);
sprintf(p,"%s","Hello World!\n");
printf("%s\n",p);
free(p);
p= NULL;
return;
}