LLVM系列第二十章：写一个简单的Function Pass

本文链接：https://blog.csdn.net/Zhanglin_Wu/article/details/125438188

系列文章目录

flex&bison系列

本文目录

前言
一、项目结构
二、项目细节
- 1. 程序模块
- 3. Simple Function Pass
三、编译
四、总结

前言

在此记录下用LLVM创建一个简单的Function Pass的过程，以备查阅。

开发环境的配置请参考第一章《LLVM系列第一章：编译LLVM源码》。

Function Pass，顾名思义，是在程序中的每个函数(function)上执行的。Function Pass不需要以特定顺序执行，并且不会修改外部函数(external function)。具体地说，Function Pass不能添加或删除当前模块的函数或全局变量，也不能分析或修改当前正在被处理的函数之外的其它函数。

本章我们就来写一个最简单的Function Pass。

一、项目结构

我们把这个简单的项目命名为SimpleFunctionPass。可以参考LLVM的源码中其它Pass流程的组织结构，来组织我们自己的代码（示例）：

llvm-project/llvm
├── ...
├── lib
│   └── Transforms
│       ├── CMakeLists.txt
│       └── SimpleFunctionPass
│           ├── CMakeLists.txt
│           └── SimpleFunctionPass.cpp
└── ...

二、项目细节

1. 程序模块

这个简单的项目只包含了一个模块：

SimpleFunctionPass，一个简单的Function Pass模块

SimpleFunctionPass将会对每一个函数进行处理，即把函数中的所有Basic Block等信息打印出来。

注意，我们需要把SimpleFunctionPass项目加入到LLVM Transforms父项目中，即指示CMake在编译LLVM源码的同时，也要编译SimpleFunctionPass项目。

以下是跟项目组织结构相关的部分CMake脚本。

(1) lib/Transforms/SimpleFunctionPass/CMakeLists.txt文件（示例）：

# CMakeLists.txt

add_llvm_library(SimpleFunctionPass MODULE BUILDTREE_ONLY
    SimpleFunctionPass.cpp

    PLUGIN_TOOL
    opt
)

(2) lib/Transforms/CMakeLists.txt文件（示例）：

...
add_subdirectory(SimpleFunctionPass)
...

3. Simple Function Pass

SimpleFunctionPass的实现在文件lib/Transforms/SimpleFunctionPass/SimpleFunctionPass.cpp中：

// SimpleFunctionPass.cpp

#include "llvm/IR/PassManager.h"
#include "llvm/Passes/PassBuilder.h"
#include "llvm/Passes/PassPlugin.h"
#include "llvm/Support/raw_ostream.h"

#include  <iostream>

using namespace llvm;
using std::cout;
using std::endl;

namespace
{
    class SimpleFunctionPass : public PassInfoMixin<SimpleFunctionPass>
    {
    public:

        PreservedAnalyses run(Function& function, FunctionAnalysisManager& analysisManager)
        {
            cout << "Function: " << function.getName().str() << endl;
            cout << "    Instruction Count: " << function.getInstructionCount() << endl;

            // Print out all the basic blocks in this function
            cout << endl << "    Basic Block Count: " << function.getBasicBlockList().size() << endl;
            for (const auto& basicBlock : function)
            {
                cout << "        BasicBlock: " << basicBlock.getName().str() << endl;
            }

            // Print out all the arguments of this function
            cout << endl << "    Argument Count: " << function.arg_size() << endl;
            for (const auto& arg : function.args())
            {
                cout << "        Argument: " << arg.getName().str() << endl;
            }

            cout << endl << "    Function Type: " << endl;
            function.getFunctionType()->print(outs());
            cout << endl << endl;

            // Assuming we did not change anything of the IR code
            return PreservedAnalyses::all();
        }
    };
}

// This part is the new way of registering your pass
extern "C" PassPluginLibraryInfo LLVM_ATTRIBUTE_WEAK llvmGetPassPluginInfo()
{
    return
    {
        LLVM_PLUGIN_API_VERSION,
        "SimpleFunctionPass",
        "v0.1",
        [](PassBuilder& passBuilder) {
            passBuilder.registerPipelineParsingCallback(
                [](StringRef name, FunctionPassManager& passManager, ArrayRef<PassBuilder::PipelineElement>) {
                    if(name == "simple-function-pass")
                    {
                        passManager.addPass(SimpleFunctionPass());
                        return true;
                    }

                    return false;
                }
            );
        }
    };
}

三、编译

1. 生成项目文件

用CMake工具生成项目文件（示例）：

cd /path/to/llvm-project
mkdir build
cd build

cmake -G Ninja -DLLVM_ENABLE_PROJECTS=clang ../llvm

输出log如下（示例）：

-- clang project is enabled
-- clang-tools-extra project is disabled
-- ...
-- Ninja version: 1.10.2
-- Found ld64 - /Applications/Xcode.app/Contents/Developer/Toolchains/XcodeDefault.xctoolchain/usr/bin/ld
-- ...
-- LLVM host triple: x86_64-apple-darwin20.6.0
-- LLVM default target triple: x86_64-apple-darwin20.6.0
-- ...
-- Configuring done
-- Generating done
-- Build files have been written to: .../llvm-project/build

2. 编译

用ninja进行编译（示例）：

ninja

如果我们是在第一章的编译LLVM完成之后，再编译此项目，则仅仅需要编译SimpleFunctionPass项目即可。当然，这是ninja自动就能识别出来的，即所谓的增量编译技术。输出log如下（示例）：

[4/4] Linking CXX shared module lib/SimpleFunctionPass.dylib

3. 运行

为了简单起见，假设我们要对以下Test.c文件中C代码进行处理（示例）：

// Test.c

int Foo(int a)
{
    int b;

    if (a > 33)
    {
        b = 66;
    }
    else
    {
        b = 77;
    }

    return b;
}

int Bar(int a, int b)
{
    return a + b;
}

int Bead(int a, int b)
{
    return a * b;
}

可以用clang生成IR代码，命令如下（示例）：

mv ../llvm/lib/Transforms/SimpleFunctionPass/Test.c.txt ../llvm/lib/Transforms/SimpleFunctionPass/Test.c

clang -S -emit-llvm ../llvm/lib/Transforms/SimpleFunctionPass/Test.c -o ../llvm/lib/Transforms/SimpleFunctionPass/Test.ll

生成IR代码大体如下，当然，并不是跟clang生成的IR代码完全一样，做了一点简化（示例）：

; ModuleID = 'Test.c'
source_filename = "Test.c"

define i32 @Foo(i32 %a) {
entry:
  %a.addr = alloca i32, align 4
  %b = alloca i32, align 4
  store i32 %a, i32* %a.addr, align 4
  %0 = load i32, i32* %a.addr, align 4
  %cmp = icmp sgt i32 %0, 33
  br i1 %cmp, label %if.then, label %if.else

if.then:                                          ; preds = %entry
  store i32 66, i32* %b, align 4
  br label %if.end

if.else:                                          ; preds = %entry
  store i32 77, i32* %b, align 4
  br label %if.end

if.end:                                           ; preds = %if.else, %if.then
  %1 = load i32, i32* %b, align 4
  ret i32 %1
}

define i32 @Bar(i32 %a, i32 %b) {
entry:
  %a.addr = alloca i32, align 4
  %b.addr = alloca i32, align 4
  store i32 %a, i32* %a.addr, align 4
  store i32 %b, i32* %b.addr, align 4
  %0 = load i32, i32* %a.addr, align 4
  %1 = load i32, i32* %b.addr, align 4
  %add = add nsw i32 %0, %1
  ret i32 %add
}

define i32 @Bead(i32 %a, i32 %b) {
entry:
  %a.addr = alloca i32, align 4
  %b.addr = alloca i32, align 4
  store i32 %a, i32* %a.addr, align 4
  store i32 %b, i32* %b.addr, align 4
  %0 = load i32, i32* %a.addr, align 4
  %1 = load i32, i32* %b.addr, align 4
  %mul = mul nsw i32 %0, %1
  ret i32 %mul
}

运行SimpleFunctionPass（示例）：

./bin/opt -load-pass-plugin=lib/SimpleFunctionPass.dylib -passes="simple-function-pass" -disable-output Test.ll

输出结果如下（示例）：

Function: Foo
    Instruction Count: 12

    Basic Block Count: 4
        BasicBlock: entry
        BasicBlock: if.then
        BasicBlock: if.else
        BasicBlock: if.end

    Argument Count: 1
        Argument: a

    Function Type: 
i32 (i32)

Function: Bar
    Instruction Count: 8

    Basic Block Count: 1
        BasicBlock: entry

    Argument Count: 2
        Argument: a
        Argument: b

    Function Type: 
i32 (i32, i32)

Function: Bead
    Instruction Count: 8

    Basic Block Count: 1
        BasicBlock: entry

    Argument Count: 2
        Argument: a
        Argument: b

    Function Type: 
i32 (i32, i32)