01 - LLVM Pass 實(shí)現(xiàn) C函數(shù) 插樁

01. llvm 在 iOS 插樁

網(wǎng)上我就搜到這一篇文章介紹使用 LLVM Pass 的方式,來實(shí)現(xiàn)函數(shù)插樁。

但是按照他文章的一步步坐下來,發(fā)現(xiàn)各種報(bào)錯(cuò),要不就是 pass 無法被 xcode 加載,反正遇到了不少的問題。

而且該作者很多核心的部分都沒有寫出來,不知道是因?yàn)樘?jiǎn)單,還是不想讓別人知道 ~

經(jīng)歷了兩周多左右的研究,以 LLVM 小白的身份,終于跑起來了 ~ 不容易 ~

注意:如下內(nèi)容,最好需要你已經(jīng)有 LLVM 環(huán)境?。。?/p>

02. C 函數(shù) 手動(dòng)插樁

long _ly_fun_b()
{
  struct timeval star;
  gettimeofday(&star, NULL);
  long b = star.tv_sec * 1000000 + star.tv_usec;
  return b;
}

void _ly_fun_e(char *name, long b)
{
  struct timeval end;
  gettimeofday(&end, NULL);
  long e = end.tv_sec * 1000000 + end.tv_usec;
  long t = e - b;
  printf("%s %ld us\n",name, t);
}

int main()
{
  long b = _ly_fun_b();
  printf("hello world!");
  _ly_fun_e("main", b);
  return 0;
}

手動(dòng)插樁實(shí)際上就是將預(yù)先寫好的開始和結(jié)束兩個(gè)函數(shù)的調(diào)用,手動(dòng)寫入到 main() 執(zhí)行前 和 執(zhí)行后。

那么自動(dòng)插樁了?
就是不需要手動(dòng)在 main() 中進(jìn)行修改,而是在編譯的過程中,自動(dòng)將上面2個(gè)函數(shù),插入到 main() 執(zhí)行前 和 執(zhí)行后。

那么如何在編譯的時(shí)候,在 main() 插入其他的 函數(shù)調(diào)用 了?

03. 通過 LLVM Pass 自動(dòng)插樁的過程

001.jpg

這里涉及一些 LLVM IR 的東西,在這里就不列舉了,可以自己去學(xué)習(xí)下。
總之,我們可以在如上 橘黃色 Optimizer (優(yōu)化器)這部分,來完成 自動(dòng)插樁 的目的。

而 LLVM 留給開發(fā)者在 Optimizer (優(yōu)化器)階段 讀寫 IR 的方式也有很多,比如:Pass、ClangPlugin ..
至于 Pass 和 ClangPlugin 對(duì)比,可以自己去看下,本文通過 Pass 來實(shí)現(xiàn)。

04. 實(shí)現(xiàn) 函數(shù)插樁 Pass

04-01. $LLVM_SOURCE/lib/Transforms/MyPlacement/MyPlacementPass.cpp

#include <string>
#include <system_error>
#include <iostream>

#include "llvm/Support/raw_ostream.h"
#include "llvm/Pass.h"
#include "llvm/IR/Module.h"
#include "llvm/IR/Function.h"
#include "llvm/IR/Instructions.h"
#include "llvm/IR/LegacyPassManager.h"
#include "llvm/IR/IRBuilder.h"
//#include "llvm/IR/BasicBlock.h"
//#include "llvm/IR/Constants.h"
//#include "llvm/IR/LLVMContext.h"
//#include "llvm/IR/Type.h"
#include "llvm/Transforms/IPO/PassManagerBuilder.h"

using namespace llvm;

struct MyPlacementPass : public FunctionPass
{
  static char ID;
  MyPlacementPass() : FunctionPass(ID){}
    
  bool runOnFunction(Function &F) override
  {
    printf("------------- runOnFunction --------------\n");
    if (F.getName().startswith("_ly_fun"))
    {
      return false;
    }

    LLVMContext &context = F.getParent()->getContext();
    BasicBlock &bb = F.getEntryBlock();
    
    Instruction *beginInst = dyn_cast<Instruction>(bb.begin());
    FunctionType *type = FunctionType::get(Type::getInt64Ty(context), {}, false);
    Constant *beginFun = F.getParent()->getOrInsertFunction("_ly_fun_b", type);
    Value *beginTime = nullptr;

    if (Function *fun = dyn_cast<Function>(beginFun))
    {
      CallInst *inst = CallInst::Create(fun);
      inst->insertBefore(beginInst);
      beginTime = inst;
    }
    
    for (Function::iterator I = F.begin(), E = F.end(); I != E; ++I)
    {
      BasicBlock &BB = *I;
      for (BasicBlock::iterator I = BB.begin(), E = BB.end(); I != E; ++I)
      {
        ReturnInst *IST = dyn_cast<ReturnInst>(I);
        if (IST)
        {
          FunctionType *type = FunctionType::get(Type::getVoidTy(context), {Type::getInt8PtrTy(context),Type::getInt64Ty(context)}, false);
          Constant *s = BB.getModule()->getOrInsertFunction("_ly_fun_e", type);
          if (Function *fun = dyn_cast<Function>(s))
          {
            IRBuilder<> builder(&BB);
            CallInst *inst = CallInst::Create(fun, {builder.CreateGlobalStringPtr(BB.getParent()->getName()), beginTime});
            inst->insertBefore(IST);
          }
        }
      }
    }
    return false;
  }
};

char MyPlacementPass::ID = 0;

// Automatically enable the pass.
// http://adriansampson.net/blog/clangpass.html
static void registerSkeletonPass(const PassManagerBuilder &, legacy::PassManagerBase &PM) 
{
  PM.add(new MyPlacementPass());
}

static RegisterStandardPasses RegisterMyPass(PassManagerBuilder::EP_EarlyAsPossible, registerSkeletonPass);

如上代碼基本與前面的博客文章中的差不太多,但是底部的幾句代碼,那篇博客中并沒有貼出來,這也是我經(jīng)歷了一兩周摸索 ~

不加上這幾句,編譯出來的 dylib ,總是無法被 clang 加載執(zhí)行。

并且注意如上代碼是用的 LLVM 8.0.1 版本的代碼,對(duì)應(yīng) LLVM github 源碼倉(cāng)庫中的是 release_80 分支的代碼。

我當(dāng)時(shí)使用的是 LLVM github 源碼倉(cāng)庫中的是 release_90 分支代碼,如上代碼一直編譯報(bào)錯(cuò),發(fā)現(xiàn)很多的 API 都已經(jīng)變了,搜搜了很多文檔才找到新的 API 使用。

這是 Pass 的毛病,因?yàn)橐蕾嚨氖?libclang 庫,而這個(gè)庫 API 是不穩(wěn)定的,經(jīng)常會(huì)發(fā)生變化。

04-02. $LLVM_SOURCE/lib/Transforms/MyPlacement/CMakeLists.txt

add_llvm_library( MyPlacementPass MODULE
  MyPlacementPass.cpp
)

04-03. 構(gòu)建生成 Pass dylib

image.png
╭─xiongzenghui at xiongzenghui的MacBook Pro in ~/llvm-8.0.1.src/build/Debug/lib using ?ruby-2.4.1?
╰─○ ll | grep -i 'MyPlacementPass'
-rwxr-xr-x  1 xiongzenghui  staff   149K 10 10 22:52 MyPlacementPass.dylib

MyPlacementPass.dylib 這個(gè)動(dòng)態(tài)庫就是要使用到的 LLVM Pass。

05. 新建一個(gè) Mac App 項(xiàng)目, 使用 Pass

05-01. Build Settings 中新增 CC、CXX 兩項(xiàng)用戶定義設(shè)置, 指向自己編譯的 clang

image.png

05-02. Build Settings 修改 OTHER_CFLAGS 加載 pass dylib

image.png

05-03. 如果編譯報(bào)錯(cuò): unknown argument index ...

image.png

Build Settings 欄目中搜索 index , 然后將 Enable Index-Wihle-Building Functionality 默認(rèn)值 Default 改為 NO

image.png

05-04. 運(yùn)行 mac app

image.png

?? ?? ?? 成功自動(dòng)插樁

如上 Mac App 項(xiàng)目的代碼: https://gitee.com/garywade/LLVMDemo.git

參考的資料

https://blog.csdn.net/qq_23599965/article/details/86620219
https://www.leadroyal.cn/?p=647
http://www.itdecent.cn/p/4d392b16d831

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡(jiǎn)書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容