LoadInst和StoreInst值和地址LLVM

时间:2018-01-17 10:24:32

标签: c++ c llvm

我有一个文件print.c,它有两个功能:

void printLoad(...) {
  // print address and value of memory location from which value
  printf("address=... value=...", ...); 
}

void printStore(...) {
  // print address and value of memory location from which value 
}

我有一个LLVM传递,它遍历指令并在当前的一个(load / store inst)之后添加了callInst指令printLoad或printStore(取决于指令类型)。

为了调用这个printStore或printLoad,我需要为CallInst :: Create函数添加适当的参数,它们是内存位置的地址和值。

这是我想要实现的一个例子:

define void @mains() #0 {
  %1 = alloca i32, align 4
  %2 = alloca i32, align 4
  store i32 0, i32* %1, align 4
  store i32 5, i32* %1, align 4
  store i32 2, i32* %2, align 4
  store i32 4, i32* %2, align 4
  %3 = load i32, i32* %2, align 4
  %4 = add nsw i32 %3, 5
  store i32 %4, i32* %1, align 4
  ret void
}

The output should be:
  store instruction: 
    address=...   // address of %1
    value=0
  ...
  ...
  ...
  load instruction:
    address=...  // address of %2
    value=4
  store instruction:
    address=...  // address of %1
    value=9

到目前为止的进展:

我可以在LoadInst / StoreInst上使用getPointerOperand()来获取操作数的地址。

我还可以通过将操作数转换为ConstantInt来获取前4个存储指令中StoreInst的值,但我不知道如何在最后一个StoreInst中提取值。它甚至可能吗?

编辑:

使用

void printLoad(int32_t p) 

Constant *hookLoadFunc = M.getOrInsertFunction("printLoad", Type::getVoidTy(M.getContext()), Type::getInt32Ty(M.getContext()));

  %1 = alloca i32, align 4
  %2 = alloca i32, align 4
  %3 = alloca i32, align 4
  store i32 0, i32* %1, align 4
  call void @printStore(i32 0)
  store i32 0, i32* %2, align 4
  call void @printStore(i32 0)
  store i32 5, i32* %2, align 4
  call void @printStore(i32 5)
  store i32 2, i32* %3, align 4
  call void @printStore(i32 2)
  store i32 4, i32* %3, align 4
  call void @printStore(i32 4)
  %4 = load i32, i32* %3, align 4
  %5 = add nsw i32 %4, 5
  store i32 %5, i32* %2, align 4
  call void @printStore(i32 %5)
  ret i32 0
  %2 = alloca i32, align 4
  store i32 %0, i32* %2, align 4
  call void @printStore(i32 %0)
  %3 = load i32, i32* %2, align 4
  %4 = call i32 (i8*, ...) @printf(i8* getelementptr inbounds ([22 x i8], [22 x i8]* @.str, i32 0, i32 0), i32 %3)
  ret void
  %2 = alloca i32, align 4
  store i32 %0, i32* %2, align 4
  call void @printStore(i32 %0)
  %3 = load i32, i32* %2, align 4
  %4 = call i32 (i8*, ...) @printf(i8* getelementptr inbounds ([22 x i8], [22 x i8]* @.str.1, i32 0, i32 0), i32 %3)
  ret void

这会导致分段错误:运行时为11。

解决:

想出我有无限循环(由于递归)。 printStore 实际上使用了加载/存储指令,从而创建了对 printStore 的另一个调用,依此类推。

1 个答案:

答案 0 :(得分:2)

假设您有llvm::Function代表printLoad()printStore()

llvm::Function * print_load = ....
llvm::Function * print_store = ...

您可以为每个CallInstLoadInst发出StoreInst

LoadInst

LoadInst * some_load = ...
Value * address_of_load = some_load->getOperand(0);
Value * print_load_arguments[] = { address_of_load, some_load };

// Insert a CallInst just after the load.
CallInst::Create(print_load, print_load_arguments )->insertAfter( some_load );

请记住,在llvm中,LoadInst加载的值与LoadInst本身相同。

StoreInst

StoreInst * some_store = ...
Value * value_to_store = some_store->getOperand(0);
Value * address_of_store = some_store->getOperand(1);
Value * print_store_arguments[] = { address_of_store, value_to_store };

// Insert a CallInst just after the store.
CallInst::Create(print_store, print_store_arguments)->insertAfter(some_store);

如果所有类型都匹配,这将有效。否则,您必须在致电BitCastprintStore()之前插入printLoad()说明。