如何将malloc放在函数里,如何写这个函数,以及如何调用是经常遇到的问题。
看题目:
1.看下面代码有什么问题
void GetMemory( char *p )
{
p = (char *) malloc( 100 );
}
void Test( void )
{
char *str = NULL;
GetMemory( str );
strcpy( str, “hello world” );
printf( str );
}
(1)这里最大的问题是p是函数的局部变量,在函数返回之后就消失了,传入形参并不能真正改变实参的值,因此不会成功。
(2):在函数GetMemory中和Test中没有malloc对应的free,造成内存泄露 。
2使用指向指针的指针
void GetMemory( char **p, int num )
{
p = (char ) malloc( num );
}
void Test( void )
{
char *str = NULL;
GetMemory( &str, 100 );
strcpy( str, “hello” );
printf( str );
}
这的问题是p = (char ) malloc( num );之后应该加上判断函数,确定是否分配成功,虽然一般都会成功,但是加上判断是很好的习惯。也即:
if ( *p == NULL )
{
…//进行申请内存失败处理
}
同时应考虑num>0;
- 未释放堆内存 动态分配的内存在程序结束之前没有释放,应该调用free, 把malloc生成的内存释放掉。
- printf(str) 改为 printf(“%s”,str)。
即使使用这样的函数,也要加上是否分配成功的判断:
void Test( void )
{
char str = (char ) malloc( 100 );
strcpy( str, “hello” );
free( str );
… //省略的其它语句
}