使用pycparser编辑C函数声明

Question

我需要编写一个Python程序来解析C源代码文件，并在每个函数声明的末尾添加一个隐藏参数。

更准确地说，我需要修改如下内容：

void f(int a, int b, int c) { ... }

变成类似这样的东西：

void f(int a, int b, int c, int hiddenArg) { ... }

显然，我还需要编辑其他函数对此函数的所有调用。

我需要使用pycparser来做到这一点，但是一旦我读完它，我就找不出正确的方法来编辑它。有没有一种合适的方法来解决我没有做到的事情(或者任何方式)？

Answer 1

作为修改AST的示例，下面将更改函数声明，使其具有新的参数：

class ParamAdder(c_ast.NodeVisitor):
    def visit_FuncDecl(self, node):
        ty = c_ast.TypeDecl(declname='_hidden',
                            quals=[],
                            type=c_ast.IdentifierType(['int']))
        newdecl = c_ast.Decl(
                    name='_hidden',
                    quals=[],
                    storage=[],
                    funcspec=[],
                    type=ty,
                    init=None,
                    bitsize=None,
                    coord=node.coord)
        if node.args:
            node.args.params.append(newdecl)
        else:
            node.args = c_ast.ParamList(params=[newdecl])

这会更改所有函数。显然，您可以在访问器中添加一个条件，以便只更改您感兴趣的某些函数。

您可以使用visit_FuncCall以类似的方式修改对此函数的调用。