.Net8顶级技术:边界检查之IR解析(慎入)

博客 动态
0 252
羽尘
羽尘 2023-05-22 08:55:50
悬赏:0 积分 收藏

.Net8顶级技术:边界检查之IR解析(慎入)

前言

C#这种语言之所以号称安全的,面向对象的语言。这个安全两个字可不是瞎叫的哦。因为JIT会检查任何可能超出分配范围的数值,以便使其保持在安全边界内。这里有两个概念,其一边界检查,其二IR解析。后者的生成是前者的功能的保证。啥叫IR,你以为的IL是中间语言,其实并不是,还有一层IR中间表象。.Net8的顶级技术之一(非专属),晓者寥寥。本篇来看看这两项技术。

概括

1.边界检查的缺陷
也叫循环提升,这里边界检查以数组的边界检查为例,看下C#代码
C# Code


using System.Runtime.CompilerServices;
class Program
{
    static void Main()
    {
        int[] array = new int[10_000_000];
        for (int i = 0; i < 1_000_000; i++)
        {
            Test(array);
        }
    }
    [MethodImpl(MethodImplOptions.NoInlining)]
    private static bool Test(int[] array)
    {
        for (int i = 0; i < 0x12345; i++)
        {
            if (array[i] == 42)
            {
                return true;
            }
        }
        return false;
    }
}  

JIT并不知道数组array[i]里面的i索引是否超过了array数组的长度。所以每次循环都会检查索引的大小,如果超过则报异常,不超过继续循环,这种功能就叫做边界检查。是.Net6 JIT自动加上去的,但是它有缺陷。

缺陷就在于,每次循环都检查,极大消耗了代码的运行效率。为了避免这种缺陷,是否可以在循环之前判断array数组的长度小于或者循环的最大值。通过这种一次性的判断,取代每次循环的判断,最大化提升代码运行效率。
在.Net8里面这种情况是可行的。
.Net8 JIT Machine Code


G_M000_IG01:                ;; offset=0000H
       4883EC28             sub      rsp, 40
G_M000_IG02:                ;; offset=0004H
       33C0                 xor      eax, eax
       4885C9               test     rcx, rcx
       7429                 je       SHORT G_M000_IG05
       81790845230100       cmp      dword ptr [rcx+08H], 0x12345
       7C20                 jl       SHORT G_M000_IG05
       0F1F40000F1F840000000000 align    [12 bytes for IG03]
G_M000_IG03:                ;; offset=0020H
       8BD0                 mov      edx, eax
       837C91102A           cmp      dword ptr [rcx+4*rdx+10H], 42
       7429                 je       SHORT G_M000_IG08
       FFC0                 inc      eax
       3D45230100           cmp      eax, 0x12345
       7CEE                 jl       SHORT G_M000_IG03
G_M000_IG04:                ;; offset=0032H
      EB17                 jmp      SHORT G_M000_IG06
G_M000_IG05:                ;; offset=0034H
       3B4108               cmp      eax, dword ptr [rcx+08H]
       7323                 jae      SHORT G_M000_IG10
       8BD0                 mov      edx, eax
       837C91102A           cmp      dword ptr [rcx+4*rdx+10H], 42
       7410                 je       SHORT G_M000_IG08
       FFC0                 inc      eax
       3D45230100           cmp      eax, 0x12345
       7CE9                 jl       SHORT G_M000_IG05
G_M000_IG06:                ;; offset=004BH
       33C0                 xor      eax, eax
G_M000_IG07:                ;; offset=004DH
       4883C428             add      rsp, 40
       C3                   ret
G_M00_IG08:                ;; offset=0052H
       B801000000           mov      eax, 1
G_M000_IG09:                ;; offset=0057H
       4883C428             add      rsp, 40
       C3                   ret
G_M000_IG10:                ;; offset=005CH
       E89F82C25F           call     CORINFO_HELP_RNGCHKFAIL
       CC                   int3
; Total bytes of code 98

诚如上面所言,边界检查的判断放在了for循环的外面。if和else分成快速和慢速路径,前者进行了优化。逆向成C#代码如下

if(array!=null && array.length >=0x12345)//数组不能为空,且数组的长度不能小于循环的长度。否则可能边界溢出
{
   for(int i=0;i<0x12345;i++)
   {
     if(array[i]==42)//这里不再检查边界
     {
       return true;
     }
   }
   return false;
}
else
{
  for(int i=0;i<0x2345;i++)
  {
     if(array[i]==42)//边界检查
     return true;
  }
  return flase;
}

边界检查不是本节的重点,重点是这个边界检查是如何通过IR生成的,以及优化。因为IL代码里面并没有。

2.IR的生成
部分代码。常规的认为,C#的运行过程是:
C# Code->
IL ->
Machine Code
一般的认为,IL是中间语言,或者字节码。但是实际上还有一层在JIT里面。如下:
C# Code ->
IL ->
IR ->
Machine Code
这个IR是对IL进行各种骚操作。最重要的一点就是各种优化和变形。这里来看看IR是如何对IL进行边界检查优化的。

看下边界检查的核心IR代码:

***** BB02
STMT00002 ( 0x004[E-] ... 0x009 )
   [000013] ---XG+-----                         *  JTRUE     void  
   [000012] N--XG+-N-U-                         \--*  EQ        int   
   [000034] ---XG+-----                            +--*  COMMA     int   
   [000026] ---X-+-----                            |  +--*  BOUNDS_CHECK_Rng void  
   [000008] -----+-----                            |  |  +--*  LCL_VAR   int    V01 loc0         
   [000025] ---X-+-----                            |  |  \--*  ARR_LENGTH int   
   [000007] -----+-----                            |  |     \--*  LCL_VAR   ref    V00 arg0         
   [000035] n---G+-----                            |  \--*  IND       int   
   [000033] -----+-----                            |     \--*  ARR_ADDR  byref int[]
   [000032] -----+-----                            |        \--*  ADD       byref 
   [000023] -----+-----                            |           +--*  LCL_VAR   ref    V00 arg0         
   [000031] -----+-----                            |           \--*  ADD       long  
   [000029] -----+-----                            |              +--*  LSH       long  
   [000027] -----+---U-                            |              |  +--*  CAST      long <- uint
   [000024] -----+-----                            |              |  |  \--*  LCL_VAR   int    V01 loc0         
   [000028] -----+-N---                            |              |  \--*  CNS_INT   long   2
   [000030] -----+-----                            |              \--*  CNS_INT   long   16
   [000011] -----+-----                            \--*  CNS_INT   int    42

------------ BB03 [00D..019) -> BB02 (cond), preds={BB02} succs={BB04,BB02}

这种看着牛逼轰轰的代码,正是IR。从最里面看起,意思在注释里。

[000031] -----+-----                            |           \--*  ADD       long //把LSH计算的结果加上16,这个16就是下面的CNS_INT long 16的16.
     [000029] -----+-----                            |              +--*  LSH       long  //LSH表示把数组索引左移2位。这个2就是下面的CNS_INT long 2里面的2
     [000027] -----+---U-                            |              |  +--*  CAST      long <- uint//把数组索引的类型从uint转换转换成long类型
     [000024] -----+-----                            |              |  |  \--*  LCL_VAR   int    V01 loc0 //读取本地变量V01,实际上就是数组arrar的索引。
     [000028] -----+-N---                            |              |  \--*  CNS_INT   long   2 //这个2是左移的位数
     [000030] -----+-----                            |              \--*  CNS_INT   long   16//被ADD相加的数值16

继续看

 |  \--*  IND       int   
   [000033] -----+-----                            |     \--*  ARR_ADDR  byref int[]
   [000032] -----+-----                            |        \--*  ADD       byref //把前面计算的结果与array数组的地址相加。实际上就是 array + i*4+-x10。一个索引占4个字节,methodtable和array.length各占8字节,这个表达式的结果就是索引位i的array的值,也就是array[i]这个数值。
   [000023] -----+-----                            |           +--*  LCL_VAR   ref    V00 arg0 //获取本地变量V00的地址,这个地址实际上就是数组array的地址。
   [000031] -----+-----                            |           \--*  ADD       long  
   [000029] -----+-----                            |              +--*  LSH       long  
   [000027] -----+---U-                            |              |  +--*  CAST      long <- uint
   [000024] -----+-----                            |              |  |  \--*  LCL_VAR   int    V01 loc0         
   [000028] -----+-N---                            |              |  \--*  CNS_INT   long   2
   [000030] -----+-----                            |              \--*  CNS_INT   long   16

继续看

  [000013] ---XG+-----                         *  JTRUE     void //是或者否都进行相应的跳转
   [000012] N--XG+-N-U-                         \--*  EQ        int //判断获取的array[i]是否等于42,这个42是CNS_INT int 42里的42
   [000034] ---XG+-----                            +--*  COMMA     int //计算它的两个值,获取第二个值也就是array[i]
   [000026] ---X-+-----                            |  +--*  BOUNDS_CHECK_Rng void  
   [000008] -----+-----                            |  |  +--*  LCL_VAR   int    V01 loc0 //数组的索引i值
   [000025] ---X-+-----                            |  |  \--*  ARR_LENGTH int //获取数组长度
   [000007] -----+-----                            |  |     \--*  LCL_VAR   ref    V00 arg0  //数组的长度
   [000035] n---G+-----                            |  \--*  IND       int   //获取array[i]的值
   [000033] -----+-----                            |     \--*  ARR_ADDR  byref int[] //获取刚刚array数组地址
    //中间省略,上面已经写过了。
 [000011] -----+-----                            \--*  CNS_INT   int    42

那么翻译成C# Code如下:

if(array[i]==42)
{
  return true;
}
return false

这里还没有循环,因为循环在其它的Basic Block块,这里是BB02块。那么下面就是对着BB02进行优化变形,最终形成了如上边界检查去除所示的结果。关于这点,下篇再看。

结尾

作者:江湖评谈
原文:在此处
image

posted @ 2023-05-22 08:41  江湖评谈  阅读(0)  评论(0编辑  收藏  举报
回帖
    羽尘

    羽尘 (王者 段位)

    2335 积分 (2)粉丝 (11)源码

     

    温馨提示

    亦奇源码

    最新会员