深入解析McSema:x86二进制转LLVM位码的技术突破

本文详细介绍了McSema框架如何将x86二进制代码转换为LLVM位码,重点分析了浮点运算指令的翻译挑战,包括FPU寄存器管理、控制流恢复和指令语义转换等关键技术实现细节。

McSema预览 - Trail of Bits博客

Artem Dinaburg
2014年6月23日
编译器, 会议, mcsema

6月28日,Artem Dinaburg和Andrew Ruef将在REcon 2014会议上介绍名为McSema的项目。McSema是一个将x86二进制文件转换为LLVM位码的框架。这种转换与编译器内部发生的过程相反:编译器将LLVM位码转换为x86机器码,而McSema则将x86机器码转换为LLVM位码。

为什么我们要做这么疯狂的事情?

因为我们想要分析现有的二进制应用程序,而对LLVM位码进行推理比x86指令要容易得多。不仅更容易推理LLVM位码,而且更容易操作和将位码重新定位到不同的架构。现在有许多为LLVM位码编写的程序分析工具(例如KLEE、PAGAI、LLBMC)可以用于现有应用程序。此外,在保持原始应用程序功能的同时,以复杂方式转换应用程序变得更加简单。

McSema将LLVM程序分析和操作工具的世界带到了二进制可执行文件中。虽然还有其他x86到LLVM位码的转换器,但McSema具有几个优势:

  • McSema将控制流恢复与翻译分离,允许使用自定义控制流恢复前端
  • McSema支持FPU指令
  • McSema是开源的,采用宽松许可证
  • McSema有文档记录,可以工作,并在我们的REcon演讲后不久将可用

这篇博客文章将是McSema的预览,并将研究将一个使用浮点运算的简单函数从x86指令转换为LLVM位码的挑战。我们将翻译的函数称为timespi。它接受一个参数k,并返回k * PI的值。timespi的源代码如下。

1
2
3
4
long double timespi(long double k) {
    long double pi = 3.14159265358979323846;
    return k*pi;
}

当使用Microsoft Visual Studio 2010编译时,汇编代码看起来像下面的IDA Pro截图。

使用McSema转换为LLVM位码,然后将位码重新发射为x86二进制文件后,汇编代码看起来有很大不同。

新的代码明显更大。下面,我们解释原因。

你可能会自言自语:“哇,这么小的函数有这么多代码膨胀?这些人在做什么?”

我们特别想使用这个例子,因为它展示了浮点支持——这是McSema独有的功能,并且因为它展示了x86到LLVM位码转换固有的困难。

翻译背景

McSema将x86指令建模为对寄存器上下文的操作。也就是说,有一个寄存器上下文结构包含所有寄存器和标志,指令语义表示为结构成员的修改。这个概念通过简化的伪代码示例最容易理解。像ADD EAX, EBX这样的操作将被翻译为context[EAX] += context[EBX]

翻译困难

现在让我们研究为什么像timespi这样的小函数会带来严重的翻译挑战。

PI值从数据部分读取

控制流恢复必须检测到第一个FLD指令引用数据,并正确识别数据大小。McSema将控制流恢复与翻译分离,因此可以通过IDAPython脚本利用IDA的优秀CFG恢复功能。

翻译需要支持x86 FPU寄存器、FPU标志和控制位

FPU寄存器不像整数寄存器。整数寄存器(EAX、ECX、EBX等)是命名且独立的。引用EAX的指令将始终引用寄存器上下文中的相同位置。

FPU寄存器是8个数据寄存器(ST(0)到ST(7))的堆栈,由TOP标志索引。引用ST(i)的指令实际上引用寄存器上下文中的st_registers[(TOP + i) % 8]

这是Intel IA-32软件开发手册中的图8-2。它很好地描述了FPU数据寄存器以及如何通过TOP标志隐式引用它们。

整数寄存器仅由寄存器内容定义。FPU寄存器部分由寄存器内容定义,部分由FPU标记字定义。FPU标记字是一个位图,定义浮点寄存器的内容是否为:

  • 有效(即正常浮点值)
  • 零值
  • 特殊值,如NaN或Infinity
  • 空(寄存器未使用)

要确定FPU寄存器的值,必须同时查阅FPU标记字和寄存器内容。

翻译需要至少支持FLD、FSTP和FMUL指令

实际指令操作(如加载、存储和乘法)相对容易支持。困难的部分是实现FPU执行语义。

例如,FPU存储有关FPU指令的状态,如:

  • 最后指令指针:最后执行的FPU指令的位置
  • 最后数据指针:FPU指令的最新内存操作数的地址
  • 操作码:最后执行的FPU指令的操作码

其中一些概念比其他概念更容易翻译到LLVM位码。存储最后内存操作数的地址翻译得很好:如果翻译的指令引用内存,则将内存地址存储在寄存器上下文的最后数据指针字段中。其他概念根本不翻译。例如,当单个FPU指令被翻译成多个LLVM操作时,“最后指令指针"意味着什么?

自引用状态并不是翻译困难的终点。像精度控制和舍入控制标志这样的FPU标志会影响指令操作。精度控制标志影响算术操作,而不是存储寄存器的精度。因此,可以通过FLDST(0)ST(1)中加载双扩展精度值,但FMUL可能在ST(0)中存储单精度结果。

翻译步骤

既然我们已经探讨了翻译的困难,让我们看看仅翻译timespi核心——FMUL指令所需的步骤。IA-32软件开发手册将此FMUL实例定义为"将ST(0)乘以m64fp并将结果存储在ST(0)中”。以下是将FMUL翻译为LLVM位码所需的一些步骤。

  1. 检查ST(0)的FPU标记字,确保其不为空
  2. 读取TOP标志
  3. 从st_registers[TOP]读取值。除非FPU标记字说值为零,否则只读取零
  4. 加载m64fp指向的值
  5. 进行乘法运算
  6. 检查精度控制标志。根据需要调整结果的精度
  7. 将调整后的结果写入st_registers[TOP]
  8. 更新ST(0)的FPU标记字以匹配结果。也许我们乘以了零?
  9. 更新寄存器上下文中的FPU状态标志。对于FMUL,这只是C1标志
  10. 更新最后FPU操作码字段
  11. 我们的指令引用数据了吗?确实引用了!将最后FPU数据字段更新为m64fp
  12. 跳过更新最后FPU指令字段,因为它目前并不真正映射到LLVM位码…

对于单个指令来说,这是很多工作,而且列表甚至还不完整。除了翻译原始指令的工作外,还必须在函数入口和出口点、外部调用以及地址被获取的函数上采取额外步骤。这些额外细节将在REcon演讲中涵盖。

结论

翻译浮点操作是一项棘手、困难的工作。看似简单的浮点指令隐藏了许多操作,并翻译成大量的LLVM位码。翻译后的代码很大,因为McSema暴露了浮点操作的隐藏复杂性。考虑到目前还没有尝试优化指令翻译,我们认为当前的输出相当不错。

要更详细地了解McSema,请参加Artem和Andrew在REcon的演讲,并继续关注Trail of Bits博客以获取更多公告。

编辑:McSema现已开源。请参阅我们的公告以获取更多信息。

comments powered by Disqus
使用 Hugo 构建
主题 StackJimmy 设计