转载

IDAPython：让你的生活更美好（四）

前三部分已经验证了用IDAPython能够让工作变的更简单，这一部分让我们看看逆向工程师如何使用IDAPython的颜色和强大的脚本特性。

分析者经常需要面对越来越复杂的代码，而且有时候无法轻易看出动态执行的时候执行的代码。而通过IDAPython的强大功能，我们不但能静态的标识指令，并且能够统计出对应的指令被使用了多少次。

背景

在这一部分中，我用C语言写了一个简单的程序。下面的代码是为了这次的练习而编写和编译的：

#include "stdafx.h" #include <stdlib.h> #include <time.h>  int _tmain(int argc, _TCHAR* argv[]) {   char* start = "Running the program.";   printf("[+] %s/n", start);   char* loop_string = "Looping...";     srand (time(NULL));   int bool_value = rand() % 10 + 1;   printf("[+] %d selected./n", bool_value);   if(bool_value > 5)   {     char* over_five = "Number over 5 selected. Looping 2 times.";     printf("[+] %s/n", over_five);     for(int x = 0; x < 2; x++)       printf("[+] %s/n", loop_string);   }   else   {     char* under_five = "Number under 5 selected. Looping 5 times.";     printf("[+] %s/n", under_five);     for(int x = 0; x < 5; x++)       printf("[+] %s/n", loop_string);   }   return 0; }

当我们将这个二进制加载到IDA中时，我们可以看到预期的循环与代码重定向语句。如果我们在不知道源码的情况下来看这个例子，通过静态分析能够大概判断代码实现的功能。

IDAPython：让你的生活更美好（四）

然而，如果我们想知道运行的时候执行了哪个区块的代码呢？这个问题可以用IDAPython来解决哦！

编写IDAPYTHON脚本

我们第一个需要处理的难题是如何逐句遍历每一条指令，以下代码将可以帮助我们来解决：（调试信息会输出已经被执行的指令）

RunTo(BeginEA()) event = GetDebuggerEvent(WFNE_SUSP, -1)  EnableTracing(TRACE_STEP, 1) event = GetDebuggerEvent(WFNE_ANY|WFNE_CONT, -1)  while True:   event = GetDebuggerEvent(WFNE_ANY, -1)   addr = GetEventEa()   print "Debug: current address", hex(addr), "| debug event", hex(event)   if event <= 1: break

在上面的代码中我们首先启动了调试器并且执行到函数的入口处，通过调用‘RunTo(BeginEA())’函数。接下来调用的GetDebuggerEvent()函数会等待直到断点到达。

接着我们调用EnableTracing()函数来打开IDA的跟踪功能，然后GetDebuggerEvent()函数调用会继续执行调试器，配置跟踪步骤。最后，我们会进入一个循环遍历每一个地址直到遇到结束条件。这个脚本在IDA中的输出如下所示：

IDAPython：让你的生活更美好（四）

下一个步骤是检索出执行的每一行并标识颜色。我们可以使用GetColor()和SetColor()函数来分别解决这两个问题。下面的代码会获取给定行数代码的当前颜色，并决定将代码设置成什么颜色，并进行设定。

在这个例子，我使用四种不同深浅的蓝色。深蓝色表示这一行被重复执行。（读者可以根据跟人喜好修改这部分代码）

def get_new_color(current_color):   colors = [0xffe699, 0xffcc33, 0xe6ac00, 0xb38600]   if current_color == 0xFFFFFF:     return colors[0]   if current_color in colors:     pos = colors.index(current_color)     if pos == len(colors)-1:       return colors[pos]     else:       return colors[pos+1]   return 0xFFFFFF       current_color = GetColor(addr, CIC_ITEM) new_color = get_new_color(current_color) SetColor(addr, CIC_ITEM, new_color)

运行上面的代码能够将无颜色的行数修改为高亮的蓝色。另外，如果同一行代码运行多次会变成深蓝色。

IDAPython：让你的生活更美好（四）

可以使用下面的代码来删除IDA文件中之前设置的所有颜色。将颜色设置成0XFFFFFF将会变成白色，或者高效的将之前设置的所有颜色删除。

heads = Heads(SegStart(ScreenEA()), SegEnd(ScreenEA())) for i in heads:   SetColor(i, CIC_ITEM, 0xFFFFFF)

将所有的代码合到一起，我们会得到如下结果：

heads = Heads(SegStart(ScreenEA()), SegEnd(ScreenEA())) for i in heads:   SetColor(i, CIC_ITEM, 0xFFFFFF)    def get_new_color(current_color):   colors = [0xffe699, 0xffcc33, 0xe6ac00, 0xb38600]   if current_color == 0xFFFFFF:     return colors[0]   if current_color in colors:     pos = colors.index(current_color)     if pos == len(colors)-1:       return colors[pos]     else:       return colors[pos+1]   return 0xFFFFFF    RunTo(BeginEA()) event = GetDebuggerEvent(WFNE_SUSP, -1)  EnableTracing(TRACE_STEP, 1) event = GetDebuggerEvent(WFNE_ANY|WFNE_CONT, -1) while True:   event = GetDebuggerEvent(WFNE_ANY, -1)   addr = GetEventEa()   current_color = GetColor(addr, CIC_ITEM)   new_color = get_new_color(current_color)   SetColor(addr, CIC_ITEM, new_color)   if event <= 1: break

当我在程序用运行这段代码时，我们看到执行过的反汇编指令被高亮了。如下图所示，多次执行的指令被设置成了深蓝色，让我们能够更容易理解代码执行流程。

IDAPython：让你的生活更美好（四）