[[675]], [[676]] のバリエーションの最後、DLLの遅延ロード情報を表示するPython。 遅延ロード情報はDataDirectory[IMAGE_DIRECTORY_ENTRY_DELAY_IMPORT]の示すRVAに、ImgDelayDescr構造体の配列として記録されている。 ImgDelayDescr構造体はSDKの提供する"delayimp.h"で定義されている(遅延ロードはOS自身の機能ではなく、SDKのリンカとスタブコードにより実現されている点に注意)。 #more|| #outline|| ---- * ソース全体 show_delay_loads.py: #code|python|> import sys, logging, struct, ctypes # {{{ data structure definitions BYTE = ctypes.c_ubyte WORD = ctypes.c_ushort LONG = ctypes.c_long DWORD = ctypes.c_ulong RVA = ctypes.c_ulong class IMAGE_DOS_HEADER(ctypes.Structure): _fields_ = [ ('e_magic', WORD), # Magic number ('e_cblp', WORD), # Bytes on last page of file ('e_cp', WORD), # Pages in file ('e_crlc', WORD), # Relocations ('e_cparhdr', WORD), # Size of header in paragraphs ('e_minalloc', WORD), # Minimum extra paragraphs needed ('e_maxalloc', WORD), # Maximum extra paragraphs needed ('e_ss', WORD), # Initial (relative) SS value ('e_sp', WORD), # Initial SP value ('e_csum', WORD), # Checksum ('e_ip', WORD), # Initial IP value ('e_cs', WORD), # Initial (relative) CS value ('e_lfarlc', WORD), # File address of relocation table ('e_ovno', WORD), # Overlay number ('e_res', WORD * 4), # Reserved words ('e_oemid', WORD), # OEM identifier (for e_oeminfo) ('e_oeminfo', WORD), # OEM information; e_oemid specific ('e_res2', WORD * 10), # Reserved words ('e_lfanew', LONG), # File address of new exe header ] class IMAGE_FILE_HEADER(ctypes.Structure): _fields_ = [ ('Machine', WORD), ('NumberOfSections', WORD), ('TimeDateStamp', DWORD), ('PointerToSymbolTable', DWORD), ('NumberOfSymbols', DWORD), ('SizeOfOptionalHeader', WORD), ('Characteristics', WORD), ] class IMAGE_DATA_DIRECTORY(ctypes.Structure): _fields_ = [ ('VirtualAddress', DWORD), ('Size', DWORD), ] IMAGE_NUMBEROF_DIRECTORY_ENTRIES = 16 IMAGE_DIRECTORY_ENTRY_EXPORT = 0 # Export Directory IMAGE_DIRECTORY_ENTRY_IMPORT = 1 # Import Directory IMAGE_DIRECTORY_ENTRY_RESOURCE = 2 # Resource Directory IMAGE_DIRECTORY_ENTRY_EXCEPTION = 3 # Exception Directory IMAGE_DIRECTORY_ENTRY_SECURITY = 4 # Security Directory IMAGE_DIRECTORY_ENTRY_BASERELOC = 5 # Base Relocation Table IMAGE_DIRECTORY_ENTRY_DEBUG = 6 # Debug Directory IMAGE_DIRECTORY_ENTRY_ARCHITECTURE = 7 # Architecture Specific Data IMAGE_DIRECTORY_ENTRY_GLOBALPTR = 8 # RVA of GP IMAGE_DIRECTORY_ENTRY_TLS = 9 # TLS Directory IMAGE_DIRECTORY_ENTRY_LOAD_CONFIG = 10 # Load Configuration Directory IMAGE_DIRECTORY_ENTRY_BOUND_IMPORT = 11 # Bound Import Directory in headers IMAGE_DIRECTORY_ENTRY_IAT = 12 # Import Address Table IMAGE_DIRECTORY_ENTRY_DELAY_IMPORT = 13 # Delay Load Import Descriptors IMAGE_DIRECTORY_ENTRY_COM_DESCRIPTOR = 14 # COM Runtime descriptor class IMAGE_OPTIONAL_HEADER(ctypes.Structure): _fields_ = [ ('Magic', WORD), ('MajorLinkerVersion', BYTE), ('MinorLinkerVersion', BYTE), ('SizeOfCode', DWORD), ('SizeOfInitializedData', DWORD), ('SizeOfUninitializedData', DWORD), ('AddressOfEntryPoint', DWORD), ('BaseOfCode', DWORD), ('BaseOfData', DWORD), ('ImageBase', DWORD), ('SectionAlignment', DWORD), ('FileAlignment', DWORD), ('MajorOperatingSystemVersion', WORD), ('MinorOperatingSystemVersion', WORD), ('MajorImageVersion', WORD), ('MinorImageVersion', WORD), ('MajorSubsystemVersion', WORD), ('MinorSubsystemVersion', WORD), ('Win32VersionValue', DWORD), ('SizeOfImage', DWORD), ('SizeOfHeaders', DWORD), ('CheckSum', DWORD), ('Subsystem', WORD), ('DllCharacteristics', WORD), ('SizeOfStackReserve', DWORD), ('SizeOfStackCommit', DWORD), ('SizeOfHeapReserve', DWORD), ('SizeOfHeapCommit', DWORD), ('LoaderFlags', DWORD), ('NumberOfRvaAndSizes', DWORD), ('DataDirectory', IMAGE_DATA_DIRECTORY * IMAGE_NUMBEROF_DIRECTORY_ENTRIES ), ] class IMAGE_NT_HEADERS(ctypes.Structure): _fields_ = [ ('Signature', DWORD), ('FileHeader', IMAGE_FILE_HEADER), ('OptionalHeader', IMAGE_OPTIONAL_HEADER), ] IMAGE_SIZEOF_SHORT_NAME = 8 class IMAGE_SECTION_HEADER_MISC(ctypes.Union): _fields_ = [ ('PhysicalAddress', DWORD), ('VirtualSize', DWORD), ] class IMAGE_SECTION_HEADER(ctypes.Structure): _fields_ = [ ('Name', BYTE * IMAGE_SIZEOF_SHORT_NAME), ('Misc', IMAGE_SECTION_HEADER_MISC), ('VirtualAddress', DWORD), ('SizeOfRawData', DWORD), ('PointerToRawData', DWORD), ('PointerToRelocations', DWORD), ('PointerToLinenumbers', DWORD), ('NumberOfRelocations', WORD), ('NumberOfLinenumbers', WORD), ('Characteristics', DWORD), ] class IMAGE_IMPORT_DESCRIPTOR_MISC(ctypes.Union): _fields_ = [ # 0 for terminating null import descriptor ('Characteristics', DWORD), # RVA to original unbound IAT (PIMAGE_THUNK_DATA) ('OriginalFirstThunk', DWORD), ] class IMAGE_IMPORT_DESCRIPTOR(ctypes.Structure): _fields_ = [ ('Misc', IMAGE_IMPORT_DESCRIPTOR_MISC), # 0 if not bound, # -1 if bound, and real date/time stamp # in IMAGE_DIRECTORY_ENTRY_BOUND_IMPORT (new BIND) # O.W. date/time stamp of DLL bound to (Old BIND) ('TimeDateStamp', DWORD), ('ForwarderChain', DWORD), # -1 if no forwarders ('Name', DWORD), # RVA to IAT (if bound this IAT has actual addresses) ('FirstThunk', DWORD), ] class IMAGE_IMPORT_BY_NAME(ctypes.Structure): _fields_ = [ ('Hint', WORD), ('Name', BYTE), ] class IMAGE_THUNK_DATA(ctypes.Union): _fields_ = [ ('ForwarderString', DWORD), # PBYTE ('Function', DWORD), # PDWORD ('Ordinal', DWORD), ('AddressOfData', DWORD), # PIMAGE_IMPORT_BY_NAME ] IMAGE_ORDINAL_FLAG64 = 0x8000000000000000 IMAGE_ORDINAL_FLAG32 = 0x80000000 def IMAGE_ORDINAL64(Ordinal): return (Ordinal & 0xffff) def IMAGE_ORDINAL32(Ordinal): return (Ordinal & 0xffff) def IMAGE_SNAP_BY_ORDINAL64(Ordinal): return ((Ordinal & IMAGE_ORDINAL_FLAG64) != 0) def IMAGE_SNAP_BY_ORDINAL32(Ordinal): return ((Ordinal & IMAGE_ORDINAL_FLAG32) != 0) class IMAGE_EXPORT_DIRECTORY(ctypes.Structure): _fields_ = [ ('Characteristics', DWORD), ('TimeDateStamp', DWORD), ('MajorVersion', WORD), ('MinorVersion', WORD), ('Name', DWORD), ('Base', DWORD), ('NumberOfFunctions', DWORD), ('NumberOfNames', DWORD), ('AddressOfFunctions', DWORD), ('AddressOfNames', DWORD), ('AddressOfNameOrdinals', DWORD), ] class IMAGE_BOUND_IMPORT_DESCRIPTOR(ctypes.Structure): _fields_ = [ ('TimeDateStamp', DWORD), ('OffsetModuleName', WORD), ('NumberOfModuleForwarderRefs', WORD), # Array of zero or more IMAGE_BOUND_FORWARDER_REF follows ] class IMAGE_BOUND_FORWARDER_REF(ctypes.Structure): _fields_ = [ ('TimeDateStamp', DWORD), ('OffsetModuleName', WORD), ('Reserved', WORD), ] class ImgDelayDescr(ctypes.Structure): _fields_ = [ ('grAttrs', DWORD), ('rvaDLLName', RVA), ('rvaHmod', RVA), ('rvaIAT', RVA), ('rvaINT', RVA), ('rvaBoundIAT', RVA), ('rvaUnloadIAT', RVA), ('dwTimeStamp', DWORD), ] # }}} if 2 != len(sys.argv): print 'usage: python %s filename' % sys.argv[0] quit() file_name = sys.argv[1] logging.basicConfig( level=logging.DEBUG, format='%(asctime)s %(name)s %(levelname)s %(message)s', ) log = logging.getLogger('main') try: f = open(file_name, 'rb') except IOError, e: log.error("open(%s) faileed." % file_name) log.error(e) raise e # read IMAGE_DOS_HEADER f.seek(0, 0) sz = ctypes.sizeof(IMAGE_DOS_HEADER) data = f.read(sz) dos_header = IMAGE_DOS_HEADER() fit = min(len(data), sz) ctypes.memmove(ctypes.addressof(dos_header), data, fit) # read IMAGE_NT_HEADERS f.seek(dos_header.e_lfanew, 0) sz = ctypes.sizeof(IMAGE_NT_HEADERS) data = f.read(sz) pe_header = IMAGE_NT_HEADERS() fit = min(len(data), sz) ctypes.memmove(ctypes.addressof(pe_header), data, fit) file_header = pe_header.FileHeader opt_header = pe_header.OptionalHeader # enumerate IMAGE_SECTION_HEADER array section_headers = [] section_header_sz = ctypes.sizeof(IMAGE_SECTION_HEADER) for i in range(file_header.NumberOfSections): data = f.read(section_header_sz) section_header = IMAGE_SECTION_HEADER() fit = min(len(data), sz) ctypes.memmove(ctypes.addressof(section_header), data, fit) section_headers.append(section_header) # load to memory f.seek(0, 0) file_bytes = f.read() # read all file contents image_buf = ctypes.create_string_buffer(opt_header.SizeOfImage) ctypes.memset(image_buf, 0, opt_header.SizeOfImage) image_buf_ptr = ctypes.addressof(image_buf) # copy headers ctypes.memmove( image_buf_ptr, file_bytes, opt_header.SizeOfHeaders ) # copy section datas for h in section_headers: section_virtual_addr = h.VirtualAddress pointer_to_rawdata = h.PointerToRawData size_of_rawdata = h.SizeOfRawData if pointer_to_rawdata : ctypes.memmove( image_buf_ptr + section_virtual_addr, file_bytes[pointer_to_rawdata:pointer_to_rawdata + size_of_rawdata], size_of_rawdata) # Delay Load Import Descriptor dlid_entry = opt_header.DataDirectory[IMAGE_DIRECTORY_ENTRY_DELAY_IMPORT] dlid_rva = dlid_entry.VirtualAddress dlid_head = image_buf_ptr + dlid_rva dlid_sz = ctypes.sizeof(ImgDelayDescr) offset = dlid_head dlids = [] while 1: dlid = ImgDelayDescr() ctypes.memmove( ctypes.addressof(dlid), offset, dlid_sz) if 0 == dlid.rvaHmod: break offset = offset + dlid_sz dlids.append(dlid) for dlid in dlids: print "-----------------" dllname = ctypes.c_char_p(image_buf_ptr + dlid.rvaDLLName) print "rvaDLLName = %s (0x%08X)" % (dllname.value, dlid.rvaDLLName) print "grAttrs = 0x%08X" % dlid.grAttrs print "rvaHmod = 0x%08X" % dlid.rvaHmod print "rvaIAT = 0x%08X" % dlid.rvaIAT print "rvaINT = 0x%08X" % dlid.rvaINT print "rvaBoundIAT = 0x%08X" % dlid.rvaBoundIAT print "rvaUnloadIAT = 0x%08X" % dlid.rvaUnloadIAT print "dwTimeStamp = 0x%08X" % dlid.dwTimeStamp # enumerate rvaIAT(=FirstThunk) records ft_thunk_head = image_buf_ptr + dlid.rvaIAT ft_thunk_sz = ctypes.sizeof(IMAGE_THUNK_DATA) ft_thunks = [] ft_thunk_offset = 0 while 1: ft_thunk = IMAGE_THUNK_DATA() ctypes.memmove( ctypes.addressof(ft_thunk), ft_thunk_head + ft_thunk_offset, ft_thunk_sz) ft_thunk_offset = ft_thunk_offset + ft_thunk_sz if 0 == ft_thunk.Ordinal : break ft_thunks.append(ft_thunk) for ft_thunk in ft_thunks: print "\tFirstThunk: 0x%08X" % ft_thunk.Ordinal # enumerate rvaINT(=OriginalFirstThunk) records oft_thunk_head = image_buf_ptr + dlid.rvaINT oft_thunk_sz = ctypes.sizeof(IMAGE_THUNK_DATA) oft_thunks = [] oft_thunk_offset = 0 while 1: oft_thunk = IMAGE_THUNK_DATA() ctypes.memmove( ctypes.addressof(oft_thunk), oft_thunk_head + oft_thunk_offset, oft_thunk_sz) oft_thunk_offset = oft_thunk_offset + oft_thunk_sz if 0 == oft_thunk.Ordinal : break oft_thunks.append(oft_thunk) for oft_thunk in oft_thunks: print "\tOriginalFirstThunk: %08X" % oft_thunk.Ordinal if IMAGE_SNAP_BY_ORDINAL32(oft_thunk.Ordinal): print "\t\tOrdinal: %d" % IMAGE_ORDINAL32(oft_thunk.Ordinal) else: hint_data = IMAGE_IMPORT_BY_NAME() hint_data_sz = ctypes.sizeof(IMAGE_IMPORT_BY_NAME) ctypes.memmove( ctypes.addressof(hint_data), image_buf_ptr + oft_thunk.AddressOfData, hint_data_sz) print "\t\tHint: %04X" % hint_data.Hint hint_data_name_ptr = image_buf_ptr + \ oft_thunk.AddressOfData + \ ctypes.sizeof(WORD) hint_data_name = ctypes.c_char_p(hint_data_name_ptr) print "\t\tName: %s" % hint_data_name.value ||< * 解説 セクションのコピーまでは [[676]] を初めとするこれまでのシリーズと同様。 ただし ImgDelayDescr 構造体の定義を追加している。 ** IMAGE_OPTIONAL_HEADER の DataDirectory[IMAGE_DIRECTORY_ENTRY_DELAY_IMPORT] から ImgDelayDescr配列の先頭アドレスを取得 DataDirectory[IMAGE_DIRECTORY_ENTRY_DELAY_IMPORT]は、 ImgDelayDescr配列の先頭アドレス(RVA)を指している。 ImgDelayDescr構造体の配列を取り出すのが次のソースコードになる: #code|python|> # Delay Load Import Descriptor dlid_entry = opt_header.DataDirectory[IMAGE_DIRECTORY_ENTRY_DELAY_IMPORT] dlid_rva = dlid_entry.VirtualAddress dlid_head = image_buf_ptr + dlid_rva dlid_sz = ctypes.sizeof(ImgDelayDescr) offset = dlid_head dlids = [] while 1: dlid = ImgDelayDescr() ctypes.memmove( ctypes.addressof(dlid), offset, dlid_sz) if 0 == dlid.rvaHmod: break offset = offset + dlid_sz dlids.append(dlid) ||< ** 遅延ロード用のIAT, INTの取得 遅延ロードが使うIAT, INTはImgDelayDesc構造体のrvaIAT, rvaINTから通常のインポート情報と同様に取り出せる。 ImgDelayDescr構造体の各メンバおよびIAT, INTを表示するのが次のソースコードになる: #code|python|> for dlid in dlids: print "-----------------" dllname = ctypes.c_char_p(image_buf_ptr + dlid.rvaDLLName) print "rvaDLLName = %s (0x%08X)" % (dllname.value, dlid.rvaDLLName) print "grAttrs = 0x%08X" % dlid.grAttrs print "rvaHmod = 0x%08X" % dlid.rvaHmod print "rvaIAT = 0x%08X" % dlid.rvaIAT print "rvaINT = 0x%08X" % dlid.rvaINT print "rvaBoundIAT = 0x%08X" % dlid.rvaBoundIAT print "rvaUnloadIAT = 0x%08X" % dlid.rvaUnloadIAT print "dwTimeStamp = 0x%08X" % dlid.dwTimeStamp # enumerate rvaIAT(=FirstThunk) records ft_thunk_head = image_buf_ptr + dlid.rvaIAT ft_thunk_sz = ctypes.sizeof(IMAGE_THUNK_DATA) ft_thunks = [] ft_thunk_offset = 0 # (省略) # enumerate rvaINT(=OriginalFirstThunk) records oft_thunk_head = image_buf_ptr + dlid.rvaINT oft_thunk_sz = ctypes.sizeof(IMAGE_THUNK_DATA) oft_thunks = [] oft_thunk_offset = 0 # (省略) ||< "(省略)"部分はIATとINTの内容を表示する処理で、 [[675]] と同じ。 本来は"grAttrs"のフラグをチェックし、"rva"で始まる各メンバがRVAなのかVirtualAddressなのか判別する必要があるが、簡便の為全てRVAとして処理している。