【LLVM实践】死循环插桩检测

已于 2022-03-04 23:27:49 修改
阅读量1k 收藏 2
点赞数 1
分类专栏： c++ 文章标签： LLVM
于 2022-03-04 23:26:38 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/yy405145590/article/details/123286466
版权

死循环检测


本篇文章的目的是通过在循环体里插桩代码，用以检测循环是否发生了不可预期的死循环。 

#include "stdio.h"
int main()
{
    int i = 0;
    for(; i < 21; i++)
    {
        printf("%d \n", i);
    }
    return 0;
}

输出

0 
1 
2 
3 
4 
5 
6 
7 
8 
9 
10 
11 
12 
13 
14 
15 
16 
17 
18 
19 
20 
many loop count 21

自动判断是否超出了预期的循环次数。

LLVM介绍

LLVM是新一代的编译工具，强大的模块化支持，让程序员很方便的介入到编译过程中的各个方面，本篇文章主要利用函数处理阶段来插桩部分代码。

FunctionPass

FunctionPass是LLVM在处理函数代码生成时，会调用的Pass模块。本篇文章中直接在LLVM自带的HelloWorldPass里加入我们的代码。

PreservedAnalyses HelloWorldPass::run(Function &F,
                                      FunctionAnalysisManager &AM) {
  LoopInfo *LI = &AM.getResult<LoopAnalysis>(F);
  errs() << F.getName() << " loopcount:" <<  LI->getTopLevelLoopsVector().size() << "\n";
  Type *intType = Type::getInt32Ty(F.getContext());
  std::vector<Type *> printfArgsTypes({Type::getInt8PtrTy(F.getContext())});
  FunctionType *printfType = FunctionType::get(intType, printfArgsTypes, true);
  auto printfFunc = F.getEntryBlock().getModule()->getOrInsertFunction("printf", printfType);
  for(auto &l : LI->getTopLevelLoopsVector())
  {
    auto pp = l->getLoopPredecessor();
    if(pp)
    {
      IRBuilder<> builderInit(pp->getFirstNonPHI());      
      auto pLoopCount = builderInit.CreateAlloca(builderInit.getInt32Ty(), nullptr, "loopcount");
      builderInit.CreateStore(ConstantInt::get(builderInit.getInt32Ty(), 0), pLoopCount);

      auto pB = l->getHeader();
      builderInit.SetInsertPoint(pB->getFirstNonPHI());
      auto *addL = builderInit.CreateLoad(builderInit.getInt32Ty(), pLoopCount, "loadLoopCount");
      auto *one = ConstantInt::get(builderInit.getInt32Ty(), 1);
      auto *pRest = builderInit.CreateAdd(addL, one, "inc loopCount");
      builderInit.CreateStore(pRest, pLoopCount);
      auto pBExit = l->getExitBlock();
      if(pBExit != nullptr)
      {
        builderInit.SetInsertPoint(pBExit->getFirstNonPHI());
        auto *addL = builderInit.CreateLoad(builderInit.getInt32Ty(), pLoopCount, "loadLoopCount");
        auto *n100W = ConstantInt::get(builderInit.getInt32Ty(), 20);
        auto *pCmp = cast<ICmpInst>(builderInit.CreateCmp(CmpInst::Predicate::ICMP_UGE, addL, n100W, "Cmp"));// builderInit.CreateCmp();
        //auto pAssert = BasicBlock::Create(pBExit->getContext(), "assert");
        Instruction *Brnch =
          SplitBlockAndInsertIfThen(pCmp, pCmp->getNextNode(), false);

      builderInit.SetInsertPoint(Brnch);
            

      Value *puts_params[] = {
        builderInit.CreateGlobalStringPtr("many loop count %d \n", "loopttip"),
        addL
      };
      builderInit.CreateCall(printfFunc, puts_params, "printlog");
       // auto pIfThen = SplitBlockAndInsertIfThen(pCmp, pBExit->getFirstNonPHI(), false, nullptr, (DominatorTree*)nullptr, nullptr, pAssert);
       // builderInit.Insert(pIfThen, "ifthen");
        //pBExit->getModule()->getOrInsertFunction()
        //builderInit.CreateSelect(pCmp, )
      }     
      
    }
  }
 
  return PreservedAnalyses::none();
}

生成的IR对比

; ModuleID = 'test.c'
source_filename = "test.c"
target datalayout = "e-m:e-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128"
target triple = "x86_64-unknown-linux-gnu"

@.str = private unnamed_addr constant [5 x i8] c"%d \0A\00", align 1
@loopttip = private unnamed_addr constant [21 x i8] c"many loop count %d \0A\00", align 1

; Function Attrs: nofree nounwind uwtable
define dso_local i32 @main() local_unnamed_addr #0 {
  %1 = alloca i32, align 4
  store i32 0, i32* %1, align 4
  %2 = load i32, i32* %1, align 4
  br label %3

3:                                                ; preds = %0, %3
  %4 = phi i32 [ %2, %0 ], [ %6, %3 ]
  %5 = phi i32 [ 0, %0 ], [ %8, %3 ]
  %6 = add i32 %4, 1
  %7 = call i32 (i8*, ...) @printf(i8* noundef nonnull dereferenceable(1) getelementptr inbounds ([5 x i8], [5 x i8]* @.str, i64 0, i64 0), i32 noundef %5)
  %8 = add nuw nsw i32 %5, 1
  %9 = icmp eq i32 %8, 21
  br i1 %9, label %10, label %3, !llvm.loop !3

10:                                               ; preds = %3
  store i32 %6, i32* %1, align 4
  %11 = load i32, i32* %1, align 4
  %12 = icmp ugt i32 %11, 19
  br i1 %12, label %13, label %15

13:                                               ; preds = %10
  %14 = call i32 (i8*, ...) @printf(i8* nonnull dereferenceable(1) getelementptr inbounds ([21 x i8], [21 x i8]* @loopttip, i64 0, i64 0), i32 %11)
  br label %15

15:                                               ; preds = %10, %13
  ret i32 0
}

; Function Attrs: nofree nounwind
declare dso_local noundef i32 @printf(i8* nocapture noundef readonly, ...) local_unnamed_addr #1

attributes #0 = { nofree nounwind uwtable "frame-pointer"="none" "min-legal-vector-width"="0" "no-trapping-math"="true" "stack-protector-buffer-size"="8" "target-cpu"="x86-64" "target-features"="+cx8,+fxsr,+mmx,+sse,+sse2,+x87" "tune-cpu"="generic" }
attributes #1 = { nofree nounwind "frame-pointer"="none" "no-trapping-math"="true" "stack-protector-buffer-size"="8" "target-cpu"="x86-64" "target-features"="+cx8,+fxsr,+mmx,+sse,+sse2,+x87" "tune-cpu"="generic" }

!llvm.module.flags = !{!0, !1}
!llvm.ident = !{!2}

!0 = !{i32 1, !"wchar_size", i32 4}
!1 = !{i32 7, !"uwtable", i32 2}
!2 = !{!"clang version 15.0.0 (https://github.com/llvm/llvm-project.git a4d2943bf6b350ea62745ab25c7dcc2f1d17a470)"}
!3 = distinct !{!3, !4, !5}
!4 = !{!"llvm.loop.mustprogress"}
!5 = !{!"llvm.loop.unroll.disable"}


; ModuleID = '../llvm-test/test.ll.out'
source_filename = "test.c"
target datalayout = "e-m:e-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128"
target triple = "x86_64-unknown-linux-gnu"

@.str = private unnamed_addr constant [5 x i8] c"%d \0A\00", align 1
@loopttip = private unnamed_addr constant [21 x i8] c"many loop count %d \0A\00", align 1
@loopttip.1 = private unnamed_addr constant [21 x i8] c"many loop count %d \0A\00", align 1

; Function Attrs: nofree nounwind uwtable
define dso_local i32 @main() local_unnamed_addr #0 {
  %loopcount = alloca i32, align 4
  store i32 0, i32* %loopcount, align 4
  %1 = alloca i32, align 4
  store i32 0, i32* %1, align 4
  %2 = load i32, i32* %1, align 4
  br label %3

3:                                                ; preds = %3, %0
  %4 = phi i32 [ %2, %0 ], [ %6, %3 ]
  %5 = phi i32 [ 0, %0 ], [ %8, %3 ]
  %loadLoopCount = load i32, i32* %loopcount, align 4;这里自增循环次数
  %"inc loopCount" = add i32 %loadLoopCount, 1
  store i32 %"inc loopCount", i32* %loopcount, align 4
  %6 = add i32 %4, 1
  %7 = call i32 (i8*, ...) @printf(i8* noundef nonnull dereferenceable(1) getelementptr inbounds ([5 x i8], [5 x i8]* @.str, i64 0, i64 0), i32 noundef %5)
  %8 = add nuw nsw i32 %5, 1
  %9 = icmp eq i32 %8, 21
  br i1 %9, label %10, label %3, !llvm.loop !3

10:                                               ; preds = %3
  %loadLoopCount1 = load i32, i32* %loopcount, align 4
  %Cmp = icmp uge i32 %loadLoopCount1, 20				;这里判断循环次数
  br i1 %Cmp, label %11, label %12

11:                                               ; preds = %10
  %printlog = call i32 (i8*, ...) @printf(i8* getelementptr inbounds ([21 x i8], [21 x i8]* @loopttip.1, i32 0, i32 0), i32 %loadLoopCount1)
  br label %12

12:                                               ; preds = %10, %11
  store i32 %6, i32* %1, align 4
  %13 = load i32, i32* %1, align 4
  %14 = icmp ugt i32 %13, 19
  br i1 %14, label %15, label %17

15:                                               ; preds = %12
  %16 = call i32 (i8*, ...) @printf(i8* nonnull dereferenceable(1) getelementptr inbounds ([21 x i8], [21 x i8]* @loopttip, i64 0, i64 0), i32 %13)
  br label %17

17:                                               ; preds = %15, %12
  ret i32 0
}

; Function Attrs: nofree nounwind
declare dso_local noundef i32 @printf(i8* nocapture noundef readonly, ...) local_unnamed_addr #1

attributes #0 = { nofree nounwind uwtable "frame-pointer"="none" "min-legal-vector-width"="0" "no-trapping-math"="true" "stack-protector-buffer-size"="8" "target-cpu"="x86-64" "target-features"="+cx8,+fxsr,+mmx,+sse,+sse2,+x87" "tune-cpu"="generic" }
attributes #1 = { nofree nounwind "frame-pointer"="none" "no-trapping-math"="true" "stack-protector-buffer-size"="8" "target-cpu"="x86-64" "target-features"="+cx8,+fxsr,+mmx,+sse,+sse2,+x87" "tune-cpu"="generic" }

!llvm.module.flags = !{!0, !1}
!llvm.ident = !{!2}

!0 = !{i32 1, !"wchar_size", i32 4}
!1 = !{i32 7, !"uwtable", i32 2}
!2 = !{!"clang version 15.0.0 (https://github.com/llvm/llvm-project.git a4d2943bf6b350ea62745ab25c7dcc2f1d17a470)"}
!3 = distinct !{!3, !4, !5}
!4 = !{!"llvm.loop.mustprogress"}
!5 = !{!"llvm.loop.unroll.disable"}

集成到Clang工具中

//clang/lib/CodeGen/BackendUtil.cpp
void EmitAssemblyHelper::RunOptimizationPipeline(
    BackendAction Action, std::unique_ptr<raw_pwrite_stream> &OS,
    std::unique_ptr<llvm::ToolOutputFile> &ThinLinkOS) {
    ....
     PB.registerScalarOptimizerLateEPCallback(
     [](FunctionPassManager &FPM, OptimizationLevel Level) {
       FPM.addPass(HelloWorldPass());
     });
    ....
    }

后记

本篇文章只是初试LLVM做的一个练习课题，仅供参考。

yy405145590
关注
  • 1
    点赞
  • 踩
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
c语言自动插桩,01 - LLVM Pass 实现 C函数 插桩
weixin_35641153的博客
05-24 1333
网上我就搜到这一篇文章介绍使用 LLVM Pass 的方式，来实现函数插桩。但是按照他文章的一步步坐下来，发现各种报错，要不就是 pass 无法被 xcode 加载，反正遇到了不少的问题。而且该作者很多核心的部分都没有写出来，不知道是因为太简单，还是不想让别人知道 ~经历了两周多左右的研究，以 LLVM 小白的身份，终于跑起来了 ~ 不容易 ~注意：如下内容，最好需要你已经有 LLVM 环境！！！...
llvm 编译器高级用法：第三方库插桩
SOHU_TECH的博客
12-10 8042
本文字数：3141字预计阅读时间：25分钟一、背景最近看到一篇有意思的技术文章：《抖音研发实践：基于二进制文件重排的解决方案 APP启动速度提升超15%》。原文结尾提到该方案无法覆盖10...
LLVM入门教学——SanitizerCoverage插桩（自定义桩代码+Win）
最新发布
计算机硕士的博客
05-18 927
LLVM入门教学——SanitizerCoverage插桩（自定义桩代码+Win），计算代码覆盖率。
LLVM PASS PWN（一）
include_voidmain的博客
11-01 573
LLVM PASS PWN
LLVM PASS 分析【1】：Rotate Loops
ZhiqianXia
03-05 2004
1. Introduction 2. 源码分析 2.1 初始化代码 在这里插入代码片
llvm问题积累
春江花月夜
07-25 441
1、load一个地址空间为空的memory %.sroa.3.0.copyload = load volatile i16, i16* bitcast (i8* getelementptr inbounds (i8, i8* bitcast (%struct.str* @global to i8*), i64 14) to i16*), align 2 store volatile i16 ...
QBDI:基于LLVM的动态二进制检测框架
05-02
介绍 QuarkslaB动态二进制工具（QBDI）是一个模块化，跨平台和跨体系结构的DBI框架。 它旨在支持在x86，x86-64，ARM和AArch64架构上运行Linux，macOS，Android，iOS和Windows操作系统。 除了C / C ++ API，Python和...
llvm编译器cookbook 中英 SSAbook
01-14
6. **实践应用**：书中可能会有实例演示如何用LLVM构建编译器或分析工具，这有助于读者将理论知识应用于实际项目。 7. **跨平台能力**：LLVM支持多种架构和操作系统，学习如何利用这一特性创建跨平台的编译解决方案...
LLVM-14.0.5-win64
06-24
它可以应用各种各样的优化策略，如死代码消除、常量折叠等，以提高生成代码的效率。 4. **LLVM Linker**：这个链接器可以将多个编译后的模块合并成一个可执行文件或库，同时处理符号解析和重定位。 5. **LLVM ...
Hikari_LLVM15.0.0.xctoolchain
10-03
标题中的“Hikari-LLVM15.0.0.xctoolchain”指的是一个针对Xcode工具链的定制版本，结合了Hikari（可能是一个特定的优化或工具集）与LLVM 15.0.0。LLVM是一个开源的编译器基础设施项目，广泛用于构建编译器、工具和...
llvm cookbook
04-12
llvm cookbook
llvm学习笔记(1)
softee的专栏
12-22 2018
下载并安装了llvm 3.4，打算认真学习一下这个著名的编译器。 llvm目录下的README.txt，doc/GettingStarted.rst说明了怎么使用llvm，doc/README.txt则是llvm的一个综述，docs/Packaging.rst则指出了如何为llvm写一个包。 GettingStarted.rst： llvm包含三个部分。 1）LLVM包。 包含l
（LLVM）中间语言（IR）基本语法简介
热门推荐
蓝色角落……
07-01 1万+
（LLVM）中间语言（IR）基本语法简介转自：http://blog.sina.com.cn/s/blog_49b6b6d001011gik.html根据编译原理知识，编译器不是直接将源语言翻译为目标语言，而是翻译为一种“中间语言”，我们编译器从业人员称之为“IR”--指令集，之后再由中间语言，利用后端程序和设备翻译为目标平台的汇编语言； 无疑，不同编译器的中间语言IR是不一样的，而IR可以说是集中
LLVM系列第二十二章：写一个简单的编译时函数调用统计器(Pass)
Zhanglin_Wu的专栏, C++, Scrum, LLVM, 编译器
07-08 1666
用LLVM提供的C++ API，写一个简单的Pass，用来统计每个函数在编译时的调用次数，并进行测试
LLVM - 生成for循环
Network/Storage/Linux Kernel
06-22 999
1、生成函数 2、使用Phi表达式实现for循环 3、本例中，KaleidoscopeJIT源码位于./llvm-8.0.1.src/examples/Kaleidoscope/include/KaleidoscopeJIT.h Phi的概念可参考https://stackoverflow.com/questions/11485531/what-exactly-phi-instruction-does-and-how-to-use-it-in-llvm 本例中，生成的llvm ir如下： define do
LLVM 介绍（一） 项目与Chris Lattner
11-24 791
目录 LLVM 项目 编译器高材生Chris Lattner（克里斯·拉特纳） CLANG 项目 LLVM基本特性 LLVM优势特性 LLVM适用人群 LLVM 项目 LLVM项目是模块化、可重用的编译器以及工具链技术的集合。 可重用的编译器以及工具链技术的集合。 美国计算机协会 (ACM) 将其2012 年软件系统奖项颁给了LLVM，之前曾经获得此奖项的软件和技术包括:Jav...
LLVM核心库入门指南：实践解析
"Getting Started with LLVM Core Libraries" 是一本实践指南，旨在帮助读者理解LLVM核心库，通过源代码参考和片段、编译器阶段之间的连接分析、重要数据结构的描述以及如何使用LLVM提供的工具来深入学习LLVM框架。...

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值