当前位置：网站首页>LLVM系列第七章：函数参数Function Arguments

LLVM系列第七章：函数参数Function Arguments

2022-08-02 14:07:00 【飞翼剑仆】

系列文章目录

本文目录

前言
一、Hello Function Arguments
二、编译
三、运行
四、总结

前言

在此记录下用LLVM为函数(Function)添加参数(Arguments)的过程，以备查阅。

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

我们知道，函数是可以接收参数的。每个参数都可以有自己的类型，比如整数、字符串、指针等等。本章，我们就来创建一个带有参数的函数。

一、Hello Function Arguments

为简单起见，假定这个函数带有两个参数，它们的名称为a和b，类型均为32位整数型。代码如下（示例）：

// HelloFunctionArguments.cpp

#include "llvm/IR/BasicBlock.h"
#include "llvm/IR/Function.h"
#include "llvm/IR/GlobalVariable.h"
#include "llvm/IR/IRBuilder.h"
#include "llvm/IR/LLVMContext.h"
#include "llvm/IR/Module.h"
#include "llvm/IR/Verifier.h"

#include <vector>

using namespace llvm;

int main(int argc, char* argv[])
{
    LLVMContext context;
    IRBuilder<> builder(context);

    // Create a module
    Module* module = new Module("HelloModule", context);

    // Add a global variable
    module->getOrInsertGlobal("helloGlobalVariable", builder.getInt32Ty());
    GlobalVariable* globalVariable = module->getNamedGlobal("helloGlobalVariable");
    globalVariable->setLinkage(GlobalValue::CommonLinkage);
    globalVariable->setAlignment(MaybeAlign(4));

    // Add a function
    std::vector<Type*> parameters(2, builder.getInt32Ty());
    FunctionType* functionType = FunctionType::get(builder.getInt32Ty(), parameters, false);
    Function* function = Function::Create(functionType, GlobalValue::ExternalLinkage, "HelloFunction", module);

    // Set arguments for the function
    function->getArg(0)->setName("a");
    function->getArg(1)->setName("b");

    // Create a block
    BasicBlock* block = BasicBlock::Create(context, "entry", function);
    builder.SetInsertPoint(block);

    // Add a return
    ConstantInt* zero = builder.getInt32(0);
    builder.CreateRet(zero);

    // Print the IR
    verifyFunction(*function);
    module->print(outs(), nullptr);

    return 0;
}

二、编译

用clang++进行编译（示例）：

# Set up C++ standard library and header path
export SDKROOT=$(xcrun --sdk macosx --show-sdk-path)

# Compile
clang++ -w -o HelloFunctionArguments `llvm-config --cxxflags --ldflags --system-libs --libs core` HelloFunctionArguments.cpp

以上命令会生成一个名为HelloFunctionArguments的可执行程序。

三、运行

运行HelloFunctionArguments（示例）：

./HelloFunctionArguments

输出结果如下（示例）：

; ModuleID = 'HelloModule'
source_filename = "HelloModule"

@helloGlobalVariable = common global i32, align 4

define i32 @HelloFunction(i32 %a, i32 %b) {
entry:
  ret i32 0
}