Angr安装与使用之使用篇（十）

针对angr提供的练习题，现在进行求解09_angr_hooks，它也是关于hook的问题，需要使用angr求解出正确密码。但是需要提供排除地址以减少路径求解时间。
具体代码如下所示

import angr
import claripy
import sys

def main(argv):
  path_to_binary = argv[1]
  project = angr.Project(path_to_binary)

  initial_state = project.factory.entry_state()

  # 此处地址为check_equals_XYMKBKUHNIQYNQXE函数的入口地址
  check_equals_called_address = 0x80485a5

  #check_equals_XYMKBKUHNIQYNQXE函数所占的长度
  instruction_to_skip_length = 0x4d
  @project.hook(check_equals_called_address, length=instruction_to_skip_length)
  def skip_check_equals_(state):
  
    user_input_buffer_address = 0x804a054   # 输入的buffer
    user_input_buffer_length = 0x10   #

    user_input_string = state.memory.load(
      user_input_buffer_address, 
      user_input_buffer_length
    )
    
   
    check_against_string = 'XYMKBKUHNIQYNQXE' # 目标字符串

   #使用eax存储返回值，如果user_input_string与check_against_string相等则返回1，否则为0
    state.regs.eax = claripy.If(
      user_input_string == check_against_string, 
      claripy.BVV(1, 32), 
      claripy.BVV(0, 32)
    )

  simulation = project.factory.simgr(initial_state)

  def is_successful(state):
    stdout_output = state.posix.dumps(sys.stdout.fileno())
    return b'Good Job' in stdout_output

  def should_abort(state):
    stdout_output = state.posix.dumps(sys.stdout.fileno())
    return b'Try again' in stdout_output

  simulation.explore(find=is_successful, avoid=should_abort)

  if simulation.found:
    solution_state = simulation.found[0]

    # Since we are allowing Angr to handle the input, retrieve it by printing
    # the contents of stdin. Use one of the early levels as a reference.
    solution = solution_state.posix.dumps(sys.stdin.fileno())
    print(solution)
  else:
    raise Exception('Could not find the solution')

if __name__ == '__main__':
  main(sys.argv)

check_equals_XYMKBKUHNIQYNQXE函数的入口地址为0x80485a5，其占用长度为0x80485F2-0x80485a5=0x4d，将它们作为参数输入hook，以跳过check_equals_XYMKBKUHNIQYNQXE函数。

跳过的函数需要输入两个内容，即0x804a054和0x10。

下面验证实验结果
执行刚刚写好的程序，保存为scaffold09.py，并将其与09_angr_hooks放于同一文件夹中，具体如下图所示。

再执行09_angr_hooks，然后需要我们输入angr刚刚求解出的密码，结果为Good Job。

至此，求解09_angr_hooks已全部完成。