Angr中REP指令常用于字符串操作,例如REP MOVSB用于将内存中的一些字节连续地复制到另一个内存区域中。然而,由于REP指令涉及到循环操作,因此Angr将其视为基本块的结束,这会导致在分析时丢失一些关键信息。
为了解决这个问题,我们可以通过在程序中自定义字符串操作函数,并在Angr中将其视为一个基本块的结尾来实现。下面是一个示例代码:
#include
void my_strcpy(char* dst, const char* src, size_t len)
{
int i;
for (i = 0; i < len; i++)
{
dst[i] = src[i];
}
dst[len] = '\0';
}
int main()
{
char buf[20];
my_strcpy(buf, "Hello, world!", 13);
printf("%s\n", buf);
return 0;
}
在Angr中,我们可以将my_strcpy这个函数作为基本块的结尾,而不是REP指令。这样可以确保Angr在分析过程中不会失去关键信息。
import angr
# Load the binary
proj = angr.Project("/path/to/binary")
# Define the entry state
entry_state = proj.factory.entry_state()
# Define the custom function
my_strcpy_addr = 0x400500
# Create a new basic block with the custom function
block = proj.factory.block(my_strcpy_addr)
# Add the new basic block to the CFG
cfg = proj.analyses.CFGFast()
cfg._graph.add_node(block)
# Create a path group to explore all paths from the entry point
pg = proj.factory.path_group(entry_state)
pg.explore()
# Get the final states of each path
final_states = pg.deadended
# Print the contents of the buffer in each final state
for state in final_states:
buf = state.memory.load(buf_addr, 13)
print(buf)
在上面的代码中,我们首先定义了my_strcpy函数的地址,并创建了一个新的基本块以将其添加到CFG中。然后我们定义了程序的入口状态,并使用path_group来探索所有可能的路径。最后,我们获取所有结束状态,并在每个状态中打印缓冲区的内容。通过这种方法