除法运算的汇编优化(用乘法和右移来代替)

今天在反编译一段程序时发现了一些奇怪的代码,耗费了一天的时间终于弄懂了这段算法

| push    ebp                  |
| mov     ebp,esp              |
| mov     ecx,ss:[ebp+0x8]     |
| mov     eax,0x88888889       |
| mul     ecx                  |
| shr     edx,0x5              |
| mov     eax,edx              |
| shl     eax,0x4              |
| sub     eax,edx              |
| add     eax,eax              |
| add     eax,eax              |
| sub     ecx,eax              |
| push    esi                  |
| mov     esi,edx              |
| mov     eax,0x88888889       |
| mul     esi                  |
| push    edi                  |
| mov     edi,ss:[ebp+0xC]     |
| mov     ds:[edi],ecx         |
| mov     ecx,edx              |
| shr     ecx,0x5              |
| mov     edx,ecx              |
| shl     edx,0x4              |
| sub     edx,ecx              |
| add     edx,edx              |
| add     edx,edx              |
| sub     esi,edx              |
| mov     eax,0xAAAAAAAB       |
| mul     ecx                  |
| mov     ds:[edi+0x4],esi     |
| mov     esi,edx              |
| shr     esi,0x4              |
| lea     eax,ds:[esi+esi*2]   |
| add     eax,eax              |
| add     eax,eax              |
| add     eax,eax              |
| sub     ecx,eax              |
| mov     eax,0x8421085        |
| mul     esi                  |
| mov     ds:[edi+0x8],ecx     |
| mov     ecx,esi              |
| sub     ecx,edx              |
| shr     ecx,0x1              |
| add     ecx,edx              |
| shr     ecx,0x4              |
| mov     edx,ecx              |
| shl     edx,0x5              |
| sub     edx,ecx              |
| sub     esi,edx              |
| mov     eax,0xAAAAAAAB       |
| mul     ecx                  |
| shr     edx,0x3              |
| lea     eax,ds:[edx+edx*2]   |
| add     eax,eax              |
| add     eax,eax              |
| sub     ecx,eax              |
| inc     ecx                  |
| add     edx,0x7D0            |
| inc     esi                  |
| push    edi                  |
| mov     ds:[edi+0xC],esi     |
| mov     ds:[edi+0x10],ecx    |
| mov     ds:[edi+0x14],edx    |
| call    <sub_45630C0>        |
| add     esp,0x4              |
| cmp     ds:[edi+0x4],0x0     |
| mov     eax,edi              |
| jge     0x4563293            |
| mov     ds:[edi+0x4],0x0     |
| pop     edi                  |
| pop     esi                  |
| pop     ebp                  |
| ret                          |

 

将除法转换为乘法的MagicNumber:

 除以60  0x88888889

MagicNumber=2^33/被除数+1
所以被除数等于2^33/(MagicNumber-1)

不过除以有2的倍数因子的数时候,后边会跟shr edx,xx一类的指令或者mov xxx,edx

shr xxx,xx
比如除以72
会先除以9,再把商右移3次

 

除数的算法,例如开头的一段汇编代码:

mov     eax,0x88888889
mul     ecx
shr     edx,0x5

除数=(2^(32+5))/(0x88888889-1)=60.000000013969838622484784252589

所以: 除数=60

 

乘法也会优化

31乘以某个数能不能写成这个数乘以2的次幂 再减去这个数。

用数学语言表达一下就是:

设这个数为x

31*x=x*2^n-x

 这个等式是否存在,如果存在,求n的值

那我们计算一下,

31=2^n -1

得2^n=32

 得n=5

也就是说存在那么一个n使得,31乘以某个数的结果等于这个数乘以2的n次幂再减去一个数。

所以,一个乘法运算,最后就转化成了一个速度较快的移位运算了。 

 

 

 

 

以下资料参考网页:

 https://www.jianshu.com/p/c2ecadbb7b19

0x01 除法优化浅析

最近花了点时间去逆向一些小程序,遇到“(R0 * 0xAAAAAAAB) >> 32”这样的运算时,一时看不出何意。后来经过搜索,才知道这是编译器对除法做的优化(因为除法指令比较耗时)。在这里做个小笔记。

对于除法操作,如果除数是2的整数次方,那直接右移就可以了。比如:R0/4可以用R0>>2代替。如果除数不是2的整数次方,那如何优化呢?简单写一下原理:

 结合示例来看:

void test(unsigned int a) 
{ LOG("unsigned int a / 3 = %d", a / 3); }

test函数很简单,看一下反汇编代码(主要关心其中的a/3):

LDR        R2, =0xAAAAAAAB
UMULL.W    R2, R3, R0, R2
LSRS       R1, R3, #1

R0即test函数的参数a,最后a/3的计算结果保存在R1中。

首先,R0和R2做无符号数乘法(UMULL),结果的高32位保存到R3,低32位保存到R2。R2的值后续并没有用到,相当于舍弃了,即只保留R0*R2的高32位,也就是相当于整个乘法运算的结果右移了32位。所以前2行代码即:(R0 * R2) >> 32。

第3行代码,又把R3逻辑右移了1位,所以这3行代码合起来就是:(R0 * R2) >> 33。而R2的值是0xAAAAAAAB,所以最终结果就是:(R0 * 0xAAAAAAAB) >> 33。也就是编译器将a/3优化成了(a * 0xAAAAAAAB) >> 33。那么这个结果,与上面提到的除法优化原理(a/b = (a*c) >> n,其中c=(2^n)/b)吻合吗?

从“(a * 0xAAAAAAAB) >> 33”可知,编译器选择的n值为33,那么c=(2 ^ 33)/b。这里除数b为3,所以c=(2^33)/3=2863311530.67,向上取整为2863311531,换成16进制,即:0xAAAAAAAB。所以,这里编译器所做的优化与上面提到的优化原理正好吻合。

刚才有一个c从2863311530.67向上取整为2863311531的操作,那么c的值就有一个0.33的误差。那为什么这个误差不会影响到最后的计算结果呢?这个是可以进行推理证明的,可以参考:https://www.cnblogs.com/shines77/p/4189074.html


0x02 由汇编反推除法

再来看一个例子,巩固一下。假设有以下3行反汇编代码,现在来反推回高级代码。

LDR        R2, =0xCCCCCCCD
UMULL.W    R2, R3, R0, R2
LSRS       R1, R3, #2

3行代码合起来即:(R0 * 0xCCCCCCCD) >> 34。

除法优化原理:a/b = (a*c) >> n,其中c=(2^n)/b。

由(R0 * 0xCCCCCCCD) >> 34,可知n=34,c=0xCCCCCCCD。根据c=(2^ n)/b,可知b=(2^ n)/c=(2^34)/0xCCCCCCCD=4.99999999971,即b=5(因为c值有一个很小的,不影响除法运算结果的误差,所以这里得到的值近似5)。所以,上述3行汇编代码对应的高级代码即:R0/5。与实际的源码正好对应的上:

void test(int a) {
  LOG("int a / 5 = %d", a / 5);
}

再回头看一下刚开始提到的“R0 * 0xAAAAAAAB >> 32”,这个对应的高级代码应该是什么?

除法优化原理:a/b = (a*c) >> n,其中c=(2^n)/b。

由“(R0 * 0xAAAAAAAB) >> 32”,可知n=32,c=0xAAAAAAAB。根据c=(2^ n)/b,可知b=(2^ n)/c=(2^32)/0xAAAAAAAB=1.49999999983,即b=1.5。所以“(R0 * 0xAAAAAAAB) >> 32”即R0/1.5。不过,这里提到的除法优化是针对整数常量来说的,所以实际就是R0/(3/2),即R0*2/3。

0x03 有符号数的除法优化

现在把test函数简单修改一下:

void test(int a) {
  LOG("int a / 3 = %d", a / 3);
}

原先参数类型是unsigned int,现在参数类型是int。看一下a/3对应的反汇编代码:

LDR        R2, =0x55555556  
MOV        R1, R0           
SMULL.W    R2, R3, R0, R2   
SUB.W      R1, R3, R1,ASR#31

这4行代码合起来就是:(R0 * 0x55555556) >> 32 – (R0 >> 31),其中R0 >> 31是算数右移。先忽略后面的减法,只关心“(R0*0x55555556)>>32”。

除法优化原理:a/b = (a*c) >> n,其中c=(2^n)/b。

由“(R0 * 0x55555556) >> 32”,可知n=32,c=0x55555556。根据c=(2^ n)/b,可知b=(2^ n)/c=(2^32)/0x55555556=2.9999999986,即b=3。所以“(R0 * 0x55555556) >> 32”即R0/3。这么一看,貌似后面的“– (R0 >> 31)”是多余的。其实不然,简单分析一下。

参数类型是int,“R0 >> 31”就是取符号位(算数右移)。那么有两种情况:

1)R0是正数,那么R0 >> 31结果为0,减法相当于什么也没做。
除法优化原理还是:a/b = (a*c) >> n,其中c=(2^n)/b。

2)R0是负数,那么R0 >> 31结果为0xFFFFFFFF,即-1,减-1相当于加1。
除法优化原理变成:a/b =( (a*c) >> n) + 1,其中c=(2^n)/b。

为什么被除数为负数时,后面要加1呢?因为“(a*c) >> n”是向下取整的结果。加1是为了向0取整,而c/c++语言对于整数除法的规定正是向0取整。

关于除法优化,还有很多更复杂的情况,以及一系列的理论推导。限于时间,我就先了解到这。对于简单的情况,能根据反汇编代码,反推回优化之前的除法操作了。



作者:十八垧
链接:https://www.jianshu.com/p/c2ecadbb7b19
来源:简书





 

posted @ 2021-06-09 17:09  czcbzc  阅读(1986)  评论(0编辑  收藏  举报