如何将C程序中的算术运算替换为带条件选择的三元运算？

一、放弃手写解析器，用现成AST工具 #

自己从零开发C解析器需要处理语法歧义、运算符优先级、类型匹配等复杂问题，效率极低，直接用成熟的抽象语法树（AST）处理工具是最优选择：

• Clang LibTooling：C/C++官方级别的代码分析与重构工具，能精准遍历、修改AST节点，适合生产级工具开发
• Pycparser：Python实现的轻量C解析器，上手快，适合快速验证原型

二、核心处理流程 #

无论选用哪种工具，核心步骤统一为：

1. 生成AST：解析目标C代码，生成抽象语法树
2. 映射逆运算：建立二元算术运算符与逆运算的对应关系：
   • + ↔ -
   • - ↔ +（仅处理二元减，排除一元负号）
   • * ↔ /
   • / ↔ *
3. 替换运算节点：遍历所有二元算术运算节点，将其替换为[布尔变量] ? [原运算] : [逆运算]，同时为每个运算生成唯一布尔变量名（如b1、b2）
4. 新增函数参数：收集所有生成的布尔变量，作为参数追加到对应函数的参数列表末尾
5. 生成目标代码：将修改后的AST重新转换为C代码

三、Pycparser实现示例（快速验证） #

以下是简化版代码，可处理二元加减乘除运算：

from pycparser import c_parser, c_ast
from pycparser.plyparser import Coord

# 运算符逆运算映射表
OP_INVERSE = {
    '+': '-',
    '-': '+',
    '*': '/',
    '/': '*'
}

# 全局计数器，生成唯一布尔变量名
bool_var_idx = 1

class ArithRewriter(c_ast.NodeVisitor):
    def __init__(self, func_node):
        self.func = func_node
        self.bool_vars = []

    def visit_BinaryOp(self, node):
        # 递归处理左右子节点，确保嵌套运算被优先替换
        self.visit(node.left)
        self.visit(node.right)

        if node.op in OP_INVERSE:
            global bool_var_idx
            var_name = f'b{bool_var_idx}'
            bool_var_idx += 1
            self.bool_vars.append(var_name)

            # 构造逆运算节点
            inverse_op = c_ast.BinaryOp(
                op=OP_INVERSE[node.op],
                left=node.left,
                right=node.right,
                coord=node.coord
            )
            # 构造三元表达式替换原运算节点
            ternary_node = c_ast.TernaryOp(
                cond=c_ast.ID(name=var_name),
                iftrue=node,
                iffalse=inverse_op,
                coord=node.coord
            )

            # 替换父节点中的当前运算节点
            for field, child in node.parent.children():
                if child is node:
                    setattr(node.parent, field, ternary_node)
                    break

    def add_bool_params(self):
        # 为函数添加布尔参数（示例用int简化，实际可改用_Bool）
        for var in self.bool_vars:
            param_decl = c_ast.Decl(
                name=var,
                type=c_ast.TypeDecl(
                    name=var,
                    type=c_ast.IdentifierType(names=['int'])
                ),
                coord=Coord('', 0)
            )
            self.func.decl.type.args.params.append(param_decl)

# 测试代码
test_code = """
int function(int a, int b, int c) {
  int d = a + b * c;
  return d;
}
"""

# 解析代码生成AST
parser = c_parser.CParser()
ast = parser.parse(test_code)
func_def = ast.ext[0]

# 重写算术运算节点
rewriter = ArithRewriter(func_def)
rewriter.visit(func_def.body)
rewriter.add_bool_params()

# 打印修改后的代码
from pycparser.prettyprinter import ASTPrettyPrinter
print(ASTPrettyPrinter(indent='  ').visit(ast))

四、关键注意事项 #

1. 运算符优先级：AST会自动保留原运算的优先级，替换后的表达式无需额外手动添加括号
2. 一元运算符区分：需过滤一元负号（如-a），避免错误替换
3. 整数运算特殊性：C中整数除法是截断向零，逆运算乘法可能无法还原原值，需根据业务需求处理此类场景
4. 布尔变量类型：建议使用C标准的_Bool类型（需包含<stdbool.h>），示例中用int是简化处理
5. 多函数处理：若目标程序包含多个函数，需遍历所有函数定义，分别处理每个函数内的运算并添加对应布尔参数

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