C语言:在main函数之前和之后执行函数

前段时间,需要写个简单的用C语言实现的轻量级httpd服务,想实现在添加api的时候,只要在Makefile中添加对应的api C文件,就能做到将此api加入到api列表以提供响应的功能,类似于linux的驱动程序编写,在编写驱动的时候,只要编译我们写的C程序,即可把驱动加入到内核中。主要是要实现简单的类似于module_init和module_exit宏,在api C文件的最后加入这两个宏,实现自动加入和卸载指定的api。

C++中,可以通过全局变量的构造函数实现类似功能,那C语言有吗?答案是有的,可以利用attribute关键字来达到目的,前提是你使用的是GCC。

我的实现是:

#define my_init(initfn)  \

void initfn() __attribute__((constructor));

#define my_exit(exitfn)  \

void exitfn() __attribute__((destructor));

例子:

#include <stdio.h>


int main()
{
    printf("Hello World!\n");
    return 0;
}


void before()
{
    printf("before function\n");
}


void after()
{
    printf("after function\n");
}


#define my_init(initfn)  \
void initfn() __attribute__((constructor));


#define my_exit(exitfn)  \
void exitfn() __attribute__((destructor));


my_init(before)
my_exit(after)

执行的结果为:

before function
Hello World!
after function

posted on 2018-06-30 19:19  Chris_Tsai  阅读(343)  评论(0编辑  收藏  举报

导航