将clang 打印的AST转成svg

将clang 打印的AST转成svg

start.sh

将clang打印的ast结果放入input.txt

并用toDot.py将input.txt转成output.dot文件

最后使用graphviz通过output.dot生成output.svg

如果C文件出现头文件，可能造成ast过多导致生成的svg无法打开，因此可以手动删除一些input.txt里的ast结果，再运行start.sh

#!/bin/bash
# $1 可以为C文件名，如果没有参数默认使用input.txt文件作为生成的ast结果
# 示例1:	./start.sh test.c
# 示例2:  ./start.sh

if [ $# -eq 1 ];
then
	clang $1 -Xclang -ast-dump -fsyntax-only  -std=c99 | sed "s,\x1B\[[0-9;]*[a-zA-Z],,g" > input.txt
fi 
# 运行toDot.py程序
python ./toDot.py input.txt output.dot 
# 生成svg 
dot -Nfontsize=10 -Tsvg output.dot -o output.svg

toDot.py

#!/usr/bin/python3
import re
import shlex
import sys

#
# argv[1]:输入文件名
# argv[2]:输出文件名

# 中间print的内容都被输出到output
addrMap = {}
gIndex = 0

class AST_Node:
    def __init__(self, index, spelling, address, loc, other):
        self.index = index
        self.spelling = spelling
        self.address = address
        self.other = other
        self.loc = loc
        self.children = []

        self.loc = self.loc.replace("<", "\\<")
        self.loc = self.loc.replace(">", "\\>")

        # for i,x in enumerate(self.other):
        #     if re.search("<(.*?)>",x):
        #         self.other[i]=x.replace(">","\\>")
        #         self.other[i]=x.replace("<","\\<")

        if len(self.other) > 0 and ("line:" in self.other[0] or "col:" in self.other[0]):
            self.loc = self.loc + " " + self.other[0]
            del self.other[0]


def process(line):
    global gIndex
    cnt = 0
    for i in range(len(line)):
        if line[i].isalpha():
            cnt = i
            x = line[i]
            break
    line = line[cnt:]
    split = shlex.split(line)
    spelling = split[0]
    address = split[1]
    loc = ""
    loc_tmp = ""
    if len(split) > 2:
        loc_tmp = re.findall("<.*?>", split[2])
    if len(split) > 2 and len(loc_tmp) != 0:
        loc = loc_tmp[0]
        other = split[3:]
    else:
        other = split[2:]

    other = [x.strip() for x in other if x.strip() != '']
    ans = AST_Node(gIndex, spelling, address, loc, other)
    gIndex = gIndex + 1
    return cnt / 2, ans


index = 0


def take():
    global index
    oldVal, oldNode = allList[index]
    while (index + 1) < len(allList):
        x = allList[index + 1]
        newVal, newNode = x
        if newVal == oldVal + 1:
            index = index + 1
            oldNode.children.append(take())
        else:
            break
    return oldNode


def search(node: AST_Node):
    def processAddr(list):
        if list is None:
            return "None"
        return ",".join([str(x.index) for x in list])

    print(" " + str(node.index) + " [label = \"{" + str(
        node.index) + ":" + node.spelling + " | <addr>addr:" + node.address, end=" ")
    if node.loc is not None and len(node.loc) != 0:
        print(" | <0>loc:" + node.loc, end="")

    if len(node.other) > 0:
        s_tmp = " ".join([x for x in node.other])
        # res = re.search("<(.*?)>",s_tmp)
        # if res:
        #     s_tmp = re.sub("<(.*?)>","\\<"+res.group(1)+"\\>",s_tmp)
        s_tmp = s_tmp.replace("<", "\\<")
        s_tmp = s_tmp.replace(">", "\\>")
        print(" | <other>" + s_tmp, end="")
    # for i, other in enumerate(node.other):
    #     print(" | <" + str(i + 1) + ">" + other, end="")
    print("}\"];")
    for child in node.children:
        print(" " + processAddr(addrMap[node.address]) + " -> " + processAddr(addrMap[child.address]), end=";\n")
    for i, other in enumerate(node.other):
        if re.search("0x[0-9a-zA-Z]*", other) and addrMap.get(other):
            print(" " + processAddr(addrMap[node.address]) + "-> " + processAddr(addrMap[other]), end=";\n")

    for child in node.children:
        search(child)


if __name__ == '__main__':
    savedStdout = sys.stdout  # 保存标准输出流
    inputFile = sys.argv[1]
    outputFile = sys.argv[2]

    print_log = open(outputFile, "w")
    sys.stdout = print_log

    with open(inputFile, "r") as f:
        allList = []
        # list=re.findall("0x[0-9a-zA-Z]*",f.read())
        # s=set(list)

        for line in f.readlines():
            if "<<<NULL>>>" in line:
                continue
            line = line.replace("<<invalid sloc>> <invalid sloc>", "")
            p = re.compile("<((.*?)(:[0-9]*)+)>")
            res = p.search(line)
            # print(line)
            if res:
                line = p.sub("\"<" + res.group(1) + ">\"", line)
            # print(line)
            # line = line.replace("<", "\"<")
            # line = line.replace(">", ">\"")

            # print(line)
            val, node = process(line)
            allList.append((val, node))

        for alist in allList:
            i, node = alist
            if addrMap.get(node.address) is None:
                addrMap[node.address] = []
            addrMap[node.address].append(node)

        # 调试
        # for t in allList:
        #     i, x = t
        #     print("loc:"+x.loc + " other:" + str(x.other))

        # 生成的树
        root = take()
        print("digraph G{\n node [shape = record]")
        search(root)
        print("}")
        print_log.close()

示例

示例C代码:test.c

#include <stdio.h>
int main(int argc,char*argv[])
{
	printf("hello world");
	return 0;
}

对如下代码运行./start.sh test.c
删除input.txt中头文件的大多部分，只保留printf
再次运行./start.sh

示例生成的svg结果