最近做一个VPP(Vector Packet Processing)相关的项目,有很多重复性的框架代码。突发奇想,能否用面向对象的编译期多态思想替换框架代码。编译期展开自然联想到宏定义。
原理:
当预处理器遇到宏定义时,会在代码中将定义的标识符替换为其对应的内容。这种替换发生在编译之前的预处理阶段。比如Clang 中的 用宏实现的MIN:
c
#define __NSX_PASTE__(A,B) A##B
#define MIN(A,B) __NSMIN_IMPL__(A,B,__COUNTER__)
#define __NSMIN_IMPL__(A,B,L) ({ \
__typeof__(A) __NSX_PASTE__(__a,L) = (A); \
__typeof__(B) __NSX_PASTE__(__b,L) = (B); \
(__NSX_PASTE__(__a,L) < __NSX_PASTE__(__b,L)) ? __NSX_PASTE__(__a,L) : __NSX_PASTE__(__b,L); })
经常搬砖的人,应该不陌生,MIN
是一个宏定义。不多介绍了。
创建函数宏:
1. 定义模块:
定义函数和结构体模块。 C /* 定义字符串 */ #define MODE_MOUDLE_STR2(str) #str #define MODE_MOUDLE_STR(str) MODE_MOUDLE_STR2(str) /* 定义函数 */ #define NODE_MOUDLE_FUN2(pre, node) pre##_##node #define NODE_MOUDLE_FUN(pre, node) NODE_MOUDLE_FUN2(pre, node) /* 定义结构体 */ #define NODE_MOUDLE_STRUCT2(pre, node, tail) pre##_##node##_##tail #define NODE_MOUDLE_STRUCT(pre, node, tail) NODE_MOUDLE_STRUCT2(pre, node, tail)
2. 业务流程:
重复的框架处理。不同模块,直接调用MODULE_NODE_PROC
传入模块名token即可。下面代码是各模块都有的处理逻辑: 1.查询缓存,缓存匹配,进行业务处理。 2. 缓存匹配不上,查询数据把本次缓存,业务处理
C
#define MODULE_NODE_PROC(module_name) \
static inline u16\
NODE_MOUDLE_FUN(_hpf_proc, CUR_MODULE_NODE)(const vlib_main_t *vm,\
vlib_node_runtime_t *pstNodeRun, vlib_buffer_t **ppstVbuf,\
NODE_MOUDLE_STRUCT(HPF,CUR_MODULE_NODE,CACHE_S) *pstCache) \
{\
u16 usNext = 0; u32 uiRet ; vlib_buffer_t* pstVbuf = *ppstVbuf; \
NODE_MOUDLE_STRUCT(HPF,CUR_MODULE_NODE,CACHE_KEY_S) stKey; \
NODE_MOUDLE_FUN(_hpf_fillKey, CUR_MODULE_NODE)(pstVbuf, &stKey);\
/*3 命中缓存 */ \
if(true == NODE_MOUDLE_FUN(_hpf_compareKey, CUR_MODULE_NODE)(pstCache, &stKey)){\
/* 根据查到数据做业务 */\
uiRet = NODE_MOUDLE_FUN(_hpf_work, CUR_MODULE_NODE)(vm, pstNodeRun, ppstVbuf, pstCache);\
if (uiRet == 0)\
usNext = pstCache->stData.usNext;\
return usNext;\
}\
uiRet = NODE_MOUDLE_FUN(_hpf_getDataByVbuf, CUR_MODULE_NODE)(vm, pstVbuf, pstCache);\
if (uiRet != 0) \
return usNext;\
uiRet = NODE_MOUDLE_FUN(_hpf_work, CUR_MODULE_NODE)(vm, pstNodeRun, ppstVbuf, pstCache);\
if (uiRet == 0)\
usNext = pstCache->stData.usNext;\
return usNext;\
}
上述代码使用宏定义产生的函数,不同的模块直接修改下面宏定义即可,比如udp4
模块。
C
#define CUR_MODULE_NODE udp4
3. 继承实现:
定义模块,定义具体结构体,定义所需要的函数具体实现。
C
#define CUR_MODULE_NODE udp4
#define CUR_MODULE_NODE_NAME_STR MODE_MOUDLE_STR(CUR_MODULE_NODE)
/* 报文特征,key */
typedef struct{
unsigned int uiAdjIndex;
} NODE_MOUDLE_STRUCT(HPF,CUR_MODULE_NODE,CACHE_KEY_S);
/* data */
typedef struct{
unsigned short usNext;
unsigned char ucL3Type;
}NODE_MOUDLE_STRUCT(HPF,CUR_MODULE_NODE,CACHE_DATA_S);
typedef struct{
NODE_MOUDLE_STRUCT(HPF,CUR_MODULE_NODE,CACHE_KEY_S) stKey;
NODE_MOUDLE_STRUCT(HPF,CUR_MODULE_NODE,CACHE_DATA_S) stData;
}NODE_MOUDLE_STRUCT(HPF, CUR_MODULE_NODE, CACHE_S) ;
/* rewrite */
static inline void
NODE_MOUDLE_FUN(_hpf_initCache, CUR_MODULE_NODE)(NODE_MOUDLE_STRUCT(HPF,CUR_MODULE_NODE,CACHE_S) *pstData)
{
memset(pstData, 0, sizeof(*pstData));
return ;
}
/* rewrite */
static inline unsigned int
NODE_MOUDLE_FUN(_hpf_compareKey, CUR_MODULE_NODE)(IN const NODE_MOUDLE_STRUCT(HPF,CUR_MODULE_NODE,CACHE_S)* pstData,
IN const NODE_MOUDLE_STRUCT(HPF,CUR_MODULE_NODE,CACHE_KEY_S) *pstKey)
{
return memcmp(pstKey, pstKey2, sizeof(*pstKey));
}
4. 调度流程:
上述是定义和实现,下面代码使用上述实现。通过编译阶段也能检测是否有实现错误。
C
MODULE_NODE_PROC(CUR_MODULE_NODE)
MODULE_NODE_BATCH(CUR_MODULE_NODE)
/* 要去掉宏定义 */
#undef CUR_MODULE_NODE
最佳实践:
通过宏定义,它允许通过宏来定义模块、函数和结构体,并在编译期利用这些宏定义的多态性质来实现不同模块的特定业务处理。这种方式能够简化代码、减少重复工作,并且在编译时提供错误检测的功能。 但其实并不是最佳事件。C++才是真正利器。