[IDAPython] 조건부 브레이크 포인트(SetBptCnd)를 이용한 예제

# 조건부 브레이크 포인트를 이용해 0x33으로 XOR 인코딩된 데이터를 디코딩 후 패치해주는 코드

#-*- coding: utf-8 -*-
 
RunPlugin("python", 3)
 
# IDAPython 라이브러리 로드 
from idaapi import *
from idautils import *
from idc import *
 
# 아래는 구글링 하다가 발견한 소스코드이며 구현 방법은 다양하다는 것을 느낌
'''
def xor(size, key, buff):
    for index in range(0, size):
        cur_addr = buff + index
        tmp = Byte(cur_addr) ^ key
        PatchByte(cur_addr, temp)
        
xor(5, 0x33, addr)
'''
 
def DecodeString():
    addr = GetRegValue("EAX") # _printf 함수에 도달했을 때 EAX 레지스터 값을 불러옴
 
    while True: # 0일 때까지 1만큼 주소 값 증가
        if Byte(addr) != 0: 
            DecodeByte = Byte(addr) ^ 0x33 # 현재 주소의 Byte 값을 불러 0x33과 xor 인코딩
            PatchByte(addr, DecodeByte)
        else:
            break
        addr += 1
 
base_addr = get_imagebase() # 이미지 베이스 주소를 가져옴
print_addr = 0x164F # 브레이크 포인트를 하려는 주소의 RVA 주소를 가져옴
addr = base_addr + print_addr # VA(ImageBase + RVA) 주소를 구함(환경마다 이미지 베이스가 달라질 수 있기 때문)
        
AddBpt(addr) # 브레이크 포인트
SetBptCnd(addr, "DecodeString()") # 해당 주소에 브레이크 포인트가 도달한 순간 DecodeString 함수를 실행

AddBpt() - 특정 주소에 브레이크 포인트 설정

SetBptCnd() - 브레이크 포인트에 사용자 정의 핸들러 연결

GetRegValue() - 레지스터 값 가져오기

SetRegValue() - 레지스터 값 설정

GetDebuggerEvent() - 특정 시간 동안 디버거 이벤트를 대기

RunTo() - 기술한 위치에 도달하거나 브레이크포인트를 만날 때까지 프로세스 실행

StepInto() / StepOver - 호출 함수로 진입 또는 건너뛰기