灵活使用 C 语言中的宏定义
admin
2024-05-13 08:20:54
0

最近在看内核中 qspinlock相关的代码,发现了一些有意思的实现,尤其是在 kernel/locking/qspinlock.c文件的最后几行,通过 #include "qspinlock.c"语句又把这个 C文件的内容包含了进来,但前提是在 #include之前把某些关键的宏定义改了,这就使得在处理新包含进来的文件内容时,能够得到与上面第一次处理时不一样的结果,这在很大程度上复用了代码。

下面通过一个简化版的例子来展示上述实现的细节:
首先定义 file.c文件,它的实现类似于上述 kernel/locking/qspinlock.c文件:

#ifndef _GEN_PV_LOCK_SLOWPATH#include 
#include 
#include "file.h"static void __pv_init_node(void)
{printf("dummy\n");
}#define pv_enabled()        false#define pv_init_node        __pv_init_node#ifdef CONFIG_PARAVIRT_SPINLOCKS
#define queued_spin_lock_slowpath   native_queued_spin_lock_slowpath
#endif#endif /* _GEN_PV_LOCK_SLOWPATH */EXPORT_SYMBOL(queued_spin_lock_slowpath);
void queued_spin_lock_slowpath(void)
{if (pv_enabled()) {printf("pv enabled\n");} else {printf("pv not enabled\n");}pv_init_node();printf("symbol of this function: %s\n", SYMBOL_NAME(queued_spin_lock_slowpath));
}#if !defined(_GEN_PV_LOCK_SLOWPATH) && defined(CONFIG_PARAVIRT_SPINLOCKS)
#define _GEN_PV_LOCK_SLOWPATH#undef  pv_enabled
#define pv_enabled()    true#undef pv_init_node
void pv_init_node(void)
{printf("not dummy\n");
}#undef queued_spin_lock_slowpath
#define queued_spin_lock_slowpath   __pv_queued_spin_lock_slowpath#include "file.c"#endif

再定义与之配套的 file.h文件

#ifndef _FILE_H_
#define _FILE_H_#define SYMBOL_NAME(name) __SYMBOL_NAME(name)
#define __SYMBOL_NAME(name) __sym_##name#define DECLARE_SYMBOL(name) char *SYMBOL_NAME(name)#define EXTERN_SYMBOL(name) extern DECLARE_SYMBOL(name)#define __EXPORT_SYMBOL(name) \DECLARE_SYMBOL(name) = #name#define EXPORT_SYMBOL(name) __EXPORT_SYMBOL(name)#endif

最后定义 main.c文件

#include 
#include "file.h"#ifdef CONFIG_PARAVIRT_SPINLOCKS
EXTERN_SYMBOL(native_queued_spin_lock_slowpath);
EXTERN_SYMBOL(__pv_queued_spin_lock_slowpath);
#else
EXTERN_SYMBOL(queued_spin_lock_slowpath);
#endifvoid test(void)
{
#ifdef CONFIG_PARAVIRT_SPINLOCKSnative_queued_spin_lock_slowpath();printf("=========================================================\n");__pv_queued_spin_lock_slowpath();
#elsequeued_spin_lock_slowpath();
#endif
}int main(void)
{test();return 0;
}

下面通过 CONFIG_PARAVIRT_SPINLOCKS宏来展示两种不同的用法。

如果不定义 CONFIG_PARAVIRT_SPINLOCKS宏,相当于 queued_spin_lock_slowpath()只有一种实现: 即native_queued_spin_lock_slowpath()
而如果定义了 CONFIG_PARAVIRT_SPINLOCKS宏,相当于 queued_spin_lock_slowpath()有两种实现:即native_queued_spin_lock_slowpath()__pv_queued_spin_lock_slowpath()

用法一:不定义 CONFIG_PARAVIRT_SPINLOCKS
运行结果如下所示:

$ gcc -o main main.c file.c -I./ 
$ ./main 
pv not enabled
dummy
symbol of this function: queued_spin_lock_slowpath

用法二:定义CONFIG_PARAVIRT_SPINLOCKS
运行结果如下所示:

$ gcc -o main main.c file.c -I./ -DCONFIG_PARAVIRT_SPINLOCKS
$ ./main 
pv not enabled
dummy
symbol of this function: native_queued_spin_lock_slowpath
=========================================================
pv enabled
not dummy
symbol of this function: __pv_queued_spin_lock_slowpath

相关内容

热门资讯

【看表情包学Linux】进程地...   🤣 爆笑教程 👉 《看表情包学Linux》👈 猛...
育碧GDC2018程序化大世界... 1.传统手动绘制森林的问题 采用手动绘制的方法的话,每次迭代地形都要手动再绘制森林。这...
编译原理陈火旺版第三章课后题答... 下面答案仅供参考! 1.编写一个对于 Pascal 源程序的预处理程序。该程序的作用是...
MacBookPro M2芯片... MacBookPro M2芯片下如何搭建React-Native环境目录软件下载环境配置 目录 写在...
Android studio ... 解决 Android studio 出现“The emulator process for AVD ...
pyflink学习笔记(六):... 在pyflink学习笔记(一)中简单介绍了table-sql的窗口函数,下面简单介绍下...
创建deployment 创建deployment服务编排-DeploymentDeployment工作负载均衡器介绍Depl...
gma 1.1.4 (2023... 新增   1、地图工具    a. 增加【GetWorldDEMDataSet】。提供了一套 GEO...
AI专业教您保姆级在暗影精灵8... 目录 一、Stable Diffusion介绍    二、Stable Diffusion环境搭建 ...
vue笔记 第一个Vue应用 Document{{content}}{{...
Unity自带类 --- Ti... 1.在Unity中,自己写的类(脚本)的名字不能与Unit...
托福口语21天——day5 发... 目录 一、连读纠音 二、语料输入+造句输出 三、真题 一、连读纠音 英语中的连读方式有好几种...
五、排序与分页 一、排序 1、语法 ORDER BY 字段 ASC | DESC ASC(ascen...
Linux系统中如何安装软件 文章目录一、rpm包安装方式步骤:二、deb包安装方式步骤:三、tar....
开荒手册4——Related ... 0 写在前面 最早读文献的时候,每每看到related work部分都会选择性的忽略&...
实验01:吃鸡蛋问题 1.实验目的: 通过实验理解算法的概念、算法的表示、算法的时间复杂度和空间复杂度分析&...
8个免费图片/照片压缩工具帮您... 继续查看一些最好的图像压缩工具,以提升用户体验和存储空间以及网站使用支持。 无数图像压...
Spring Cloud Al... 前言 本文小新为大家带来 Sentinel控制台规则配置 相关知识,具体内容包括流控...
多项目同时进行,如何做好进度管... 多项目同时进行,如何做好进度管理? 大多数时候,面对项目进...
ATTCK红队评估实战靶场(二... 前言 第二个靶机来喽,地址:vulunstack 环境配置 大喊一声我...