前段时间看到同学写了一段基于宏和指针的继承,一直也想写一个,这种写法更加像c++中的函数覆盖,特地拿来晒晒!
//基于c语言的库函数函数覆盖
#include
#include
//这里是重点的开始
//首先将我们要扩展的函数进行宏定义,注意这里的malloc 已经不是库函数的malloc
//而xmalloc中的内容就是我们要重写的内容
#define malloc(size_t) xmalloc(size_t)
//这句话的作用是将库函数malloc的函数入口地址赋给pmalloc,这样做的目的是为了防止“宏递归”
//如果没了这句话,就没有办法扩展malloc 函数,如果我想申请空间只能malloc,而malloc又宏定义为xmalloc
//这样就会一直在xmalloc 中不停的递归
void *(*pmalloc)(size_t) = malloc;
void* xmalloc(size_t size)
{
void * point;//定义空指针
if(size <= 0)//判断申请空间大小是否合法,不合法就返回NULL申请失败
{
puts(“The size if invalid”);
return NULL ;
}
point = (void *)pmalloc(size);//申请空间,作用与库函数malloc 一样
if(point == NULL)
{
puts(“Alloc Failed!/n”);
return NULL;
}
puts(“Alloc Sucessfully!”);//告诉用户申请成功,只是简单的扩展测试,证明“覆盖”成功
printf(“The size is : %d/n”,size);
puts(“—————————/n”);
return point;//返回申请的空间的头地址
}
int main()
{
int *a;
int i;
a = (int *)malloc(8 * sizeof(int));//调用这个函数后,就会发生一个件申请的事情
for(i = 0; i < 8;i ++)
{
printf(“%d “,i);
}
puts(“”);
return 0;
}
这是运行结果: