C++覆盖系统函数的方法

在编写程序时,我们有时会有覆盖系统函数的需求。比如针对glibc中fork线程不安全的bug,我们可以在程序中实现一个特定的fork,伪代码如下:

pid_t fork(void)
{
    lock();
    fork();//system call
    unlock();
}
将这个程序编译为myfork.o。
我们的目标是,对于所有使用fork的库,当我们同时链接myfork.o后,可执行件执行fork时调用的都是上述实现的函数,这样就可以保证系统函数fork()串行地执行,从而避免了线程不安全的情况。当然,这里只是一个大概的思路,下面介绍一个具体实现的例子:

简单起见,我们以系统函数atoi为例,atoi的功能是将字符串转换为整型数,如atoi("10")的结果为10,atoi("12")的结果为12。这里我们要实现的atoi的结果在其系统版本结果的基础上+1,即atoi("12")的结果为13。

关键:g++编译时使用链接参数-Wl,-wrap;extern "C"。

  • 单文件实现
    single.cc:
    #include <stdlib.h>
    #include <stdio.h>
    extern "C" int __real_atoi(const char *nptr);
    extern "C" int __wrap_atoi(const char *str)
    {
      return __real_atoi(str)
  • 2
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值