C语言利用头文件宏定义设置数组buffer内容的正确实现方式

你当前的实现逻辑功能上是可行的，但存在两处问题：

1. 宏定义 #define CONTENT_MAX_SIZE = 20; 语法错误，C语言的对象宏不需要等号和结尾分号，正确写法为 #define CONTENT_MAX_SIZE 20，否则编译时会将=、;带入替换逻辑导致报错。
2. 代码冗余度高，每新增一种内容模板都需要手动新增switch分支，维护成本高。

采用查表法替代switch分支，所有模板统一存入静态模板表，通过下标直接访问，代码扩展性和运行效率更高。

步骤1：修改头文件定义 #

#include <stdint.h>
#include <string.h>

// 原有内容模板宏
#define CONTENT_VARIANT_A { 5, 3, 8, 1, 4, 23 }
#define CONTENT_VARIANT_B { 1, 10, 2 }
#define CONTENT_VARIANT_C { 4, 39, 2, 39 }
// 修正后的最大长度宏
#define CONTENT_MAX_SIZE 20

// 内容选项枚举，替代魔数，编译期可校验合法性
typedef enum {
    CHOICE_VARIANT_A = 0,
    CHOICE_VARIANT_B,
    CHOICE_VARIANT_C,
    CONTENT_VARIANT_COUNT // 自动计算模板总数量，用于越界校验
} ContentChoice;

步骤2：定义静态模板表（放在.c文件中） #

// 模板表项结构：存储模板首地址和长度
typedef struct {
    const uint8_t *data;
    uint8_t len;
} ContentTemplate;

// 全局静态模板表，编译时自动计算每个模板的长度
static const ContentTemplate g_content_templates[] = {
    [CHOICE_VARIANT_A] = {.data = (const uint8_t[])CONTENT_VARIANT_A, .len = sizeof((uint8_t[])CONTENT_VARIANT_A)},
    [CHOICE_VARIANT_B] = {.data = (const uint8_t[])CONTENT_VARIANT_B, .len = sizeof((uint8_t[])CONTENT_VARIANT_B)},
    [CHOICE_VARIANT_C] = {.data = (const uint8_t[])CONTENT_VARIANT_C, .len = sizeof((uint8_t[])CONTENT_VARIANT_C)},
};

如果使用的编译器不支持C99及以上标准的复合字面量，可以提前定义独立模板数组再填入表中：

static const uint8_t tpl_a[] = CONTENT_VARIANT_A;
static const uint8_t tpl_b[] = CONTENT_VARIANT_B;
static const uint8_t tpl_c[] = CONTENT_VARIANT_C;

static const ContentTemplate g_content_templates[] = {
    [CHOICE_VARIANT_A] = {.data = tpl_a, .len = sizeof(tpl_a)},
    [CHOICE_VARIANT_B] = {.data = tpl_b, .len = sizeof(tpl_b)},
    [CHOICE_VARIANT_C] = {.data = tpl_c, .len = sizeof(tpl_c)},
};

步骤3：简化SetBuffer实现 #

void SetBuffer(uint8_t *my_buffer, ContentChoice chosen_content) {
    // 保持原有清0逻辑
    memset(my_buffer, 0, CONTENT_MAX_SIZE);
    // 入参越界校验，避免非法访问
    if (chosen_content >= CONTENT_VARIANT_COUNT) {
        return;
    }
    // 直接查表复制内容
    const ContentTemplate *tpl = &g_content_templates[chosen_content];
    memcpy(my_buffer, tpl->data, tpl->len);
}

• 无冗余代码，新增模板仅需新增枚举值、补充模板表项即可，无需修改SetBuffer逻辑
• 入参使用枚举类型，编译期即可发现非法传值，比uint8_t类型安全性更高
• 运行效率为O(1)查表，比switch分支判断性能更好
• 完全兼容你现有的调用逻辑，不需要修改上层使用代码

内容的提问来源于stack exchange，提问作者LWChris