инструмент вызова функции в LLVM IR

Я хочу инструментировать вызов каждой функции программы на уровне битового кода. Предположим, есть функция void f(int a), мне нужно настроить следующий код в начале основной функции.

int a;
klee_make_symbolic(&a, sizeof(a), "a");
f(a);

Я написал проход, чтобы добиться этого.

 for (Module::iterator f = M.begin(), fe = M.end(); f != fe; ++f) {
  std::vector<llvm::Value*> args;
  for(Function::arg_iterator ai = f->arg_begin(), ae = f->arg_end(); ai != ae; ++ai){
   Type* tp = ai->getType();   
    AllocaInst* arg = new AllocaInst(tp, "name", firstInst);
    args.push_back(arg);
    LLVM_TYPE_Q llvm::Type *i8Ty = Type::getInt8Ty(getGlobalContext());
    Constant *fc = M.getOrInsertFunction("klee_make_symbolic",
                                               PointerType::getUnqual(i8Ty),
                                               Type::getInt64Ty(getGlobalContext()),
                                               PointerType::getUnqual(i8Ty),
                                               NULL);
    Function* kleeMakeSymbolic = cast<Function>(fc);

    std::vector<Value* > klee_args;
    klee_args.push_back(arg);
    klee_args.push_back(ConstantInt::get(Type::getInt64Ty(getGlobalContext()),
                       dl->getTypeAllocSizeInBits(tp)));// dl is DataLayout
    klee_args.push_back(arg);//I dont't know how to pass a argument of "const char *"

    // Inject a call to klee_make_symbolic
    CallInst::Create(kleeMakeSymbolic, klee_args, "", firstInst);

  }

  // Inject a call to the function
  CallInst::Create(f, args, "", firstInst);

}

}

Но я получил ошибку утверждения:

 llvm::CallInst::init(llvm::Value*, llvm::ArrayRef<llvm::Value*>, const llvm::Twine&): Assertion `(Args.size() == FTy->getNumParams() || (FTy->isVarArg() && Args.size() > FTy->getNumParams())) && "Calling a function with bad signature!"' failed.

Я новичок в LLVm, может ли кто-нибудь сказать мне, что не так в моей реализации?


llvm
person Dingbao Xie    schedule 27.12.2014    source источник

Ответы (1)


arrow_upward
4
arrow_downward

Вы передаете указатель a в функцию f. Это проблема с вашей реализацией.

В вашем коде:

for (Module::iterator f = M.begin(), fe = M.end(); f != fe; ++f) {
  std::vector<llvm::Value*> args;
  for(Function::arg_iterator ai = f->arg_begin(), ae = f->arg_end(); ai != ae; ++ai){
   Type* tp = ai->getType();   
    AllocaInst* arg = new AllocaInst(tp, "name", firstInst);
    args.push_back(arg);
    ...

  }

  // Inject a call to the function
  CallInst::Create(f, args, "", firstInst);
}

вы нажимаете arg в своем векторе args. arg — это значение AllocaInst, поэтому это указатель. Вам нужно, чтобы значение соответствовало сигнатуре вашей функции. Вам нужно создать LoadInst, который загружается из AllocaInst, и вставить LoadInst в ваш вектор.

Для вашей проблемы:

 klee_args.push_back(arg);//I dont't know how to pass a argument of "const char *"

Взгляните на функцию CreateGlobalStringPtr из IRBuilder. Документация здесь. IRBuilder — хороший вспомогательный класс, который немного упрощает жизнь с LLVM IR.

person Michael Haidl    schedule 27.12.2014