3.9 바이너리 계측

#include <stdio.h> #include <map> #include <string> #include <asm-generic/unistd.h> #include "pin.H" //1. 프로파일링을 위한 pintool을 구현하려면 반드시 사용해야 함. 대문자 H는 CPP 해더를 의미함. 처음알았네; KNOB<bool> ProfileCalls(KNOB_MODE_WRITEONCE, "pintool", "c", "0", "Profile function calls"); // 2 KNOB<bool> ProfileSyscalls(KNOB_MODE_WRITEONCE, "pintool", "s", "0", "Profile syscalls"); std::map<ADDRINT, std::map<ADDRINT, unsigned long> > cflows; //3 프로파일링에 사용되는 map 구조체들 std::map<ADDRINT, std::map<ADDRINT, unsigned long> > calls; std::map<ADDRINT, unsigned long> syscalls; std::map<ADDRINT, std::string> funcnames; unsigned long insn_count = 0; unsigned long cflow_count = 0; unsigned long call_count = 0; unsigned long syscall_count = 0; int main(int argc, char *argv[]) { PIN_InitSymbols(); // symbol이 존재한다면 심볼 정보를 기반으로 함수가 얼마나 호출 되었는지 통계를 제공한다. if(PIN_Init(argc,argv)) { // Init 함수. Pin과 관련된 모든것들을 초기화한다. print_usage(); return 1; } IMG_AddInstrumentFunction(parse_funcsyms, NULL); //계측시 이미지 단위로 처리 INS_AddInstrumentFunction(instrument_insn, NULL); // 인스트럭션 단위로 처리 TRACE_AddInstrumentFunction(instrument_trace, NULL); // trace 단위로 처리 if(ProfileSyscalls.Value()) { PIN_AddSyscallEntryFunction(log_syscall, NULL); // 시스콜 로깅 함수. ProfileSyscalls가 참일 경우만 실행 } PIN_AddFiniFunction(print_results, NULL); // Pin 종료함수 /* Never returns */ PIN_StartProgram(); // 분석 대상 어플리케이션 수행. return 0; }

static void parse_funcsyms(IMG img, void *v) { if(!IMG_Valid(img)) return; //1. 이미지 유효 검사 for(SEC sec = IMG_SecHead(img); SEC_Valid(sec); sec = SEC_Next(sec)) { //2 섹션 헤더가 정상일 경우 다음 섹션으로 넘겨준다. for(RTN rtn = SEC_RtnHead(sec); RTN_Valid(rtn); rtn = RTN_Next(rtn)) { //3 섹션 내에서 루틴을 통해 함수 이름을 순회한다. funcnames[RTN_Address(rtn)] = RTN_Name(rtn); //4. 함수 이름을 펑션네임 배열에 저장한다. } } }