forked from KolibriOS/kolibrios
129 lines
4.1 KiB
NASM
129 lines
4.1 KiB
NASM
; If you know macro language of FASM, there is almost nothing to comment here.
|
|
; If you don't know macro language of FASM, comments would not help you.
|
|
|
|
filename equ '%EXENAME%'
|
|
|
|
SPE_DIR_ORDER fix IMPORT EXPORT BASERELOC EXCEPTION TLS BOUND_IMPORT RESOURCE
|
|
count = 0
|
|
irps dir,SPE_DIR_ORDER
|
|
{
|
|
SPE_DIRECTORY_#dir = count
|
|
count = count + 1
|
|
}
|
|
|
|
IMAGE_DIRECTORY_ENTRY_EXPORT = 0
|
|
IMAGE_DIRECTORY_ENTRY_IMPORT = 1
|
|
IMAGE_DIRECTORY_ENTRY_RESOURCE = 2
|
|
IMAGE_DIRECTORY_ENTRY_EXCEPTION = 3
|
|
IMAGE_DIRECTORY_ENTRY_BASERELOC = 5
|
|
IMAGE_DIRECTORY_ENTRY_TLS = 9
|
|
IMAGE_DIRECTORY_ENTRY_BOUND_IMPORT = 11
|
|
|
|
virtual at 0
|
|
file filename:3Ch,4
|
|
load pehea dword from 0
|
|
end virtual
|
|
|
|
virtual at 0
|
|
file filename:pehea,0F8h
|
|
load NumberOfSections word from 6
|
|
load TimeDateStamp dword from 8
|
|
load SizeOfOptionalHeader word from 14h
|
|
if SizeOfOptionalHeader<>0E0h
|
|
error Nonstandard PE header
|
|
end if
|
|
load Characteristics word from 16h
|
|
load AddressOfEntryPoint dword from 28h
|
|
load ImageBase dword from 34h
|
|
load SectionAlignment dword from 38h
|
|
load FileAlignment dword from 3Ch
|
|
load MajorOperatingSystemVersion word from 40h
|
|
load MinorOperatingSystemVersion word from 42h
|
|
load MajorSubsystemVersion word from 48h
|
|
load MinorSubsystemVersion word from 4Ah
|
|
load SizeOfImage dword from 50h
|
|
load SizeOfHeaders dword from 54h
|
|
load Subsystem word from 5Ch
|
|
load SizeOfStackReserve dword from 60h
|
|
load SizeOfHeapReserve dword from 68h
|
|
load SrcNumberOfRvaAndSizes dword from 74h
|
|
|
|
DstNumberOfRvaAndSizes = 0
|
|
irps dir,SPE_DIR_ORDER
|
|
{
|
|
if IMAGE_DIRECTORY_ENTRY_#dir < SrcNumberOfRvaAndSizes
|
|
load DirRVA_#dir dword from 78h + 8*IMAGE_DIRECTORY_ENTRY_#dir
|
|
load DirSize_#dir dword from 7Ch + 8*IMAGE_DIRECTORY_ENTRY_#dir
|
|
else
|
|
DirRVA_#dir = 0
|
|
DirSize_#dir = 0
|
|
end if
|
|
if DirRVA_#dir > 0 & DirSize_#dir > 0
|
|
DstNumberOfRvaAndSizes = SPE_DIRECTORY_#dir + 1
|
|
end if
|
|
}
|
|
|
|
end virtual
|
|
|
|
SectionAlignmentLog = 0
|
|
while SectionAlignment <> 1 shl SectionAlignmentLog
|
|
SectionAlignmentLog = SectionAlignmentLog + 1
|
|
end while
|
|
FileAlignmentLog = 0
|
|
while FileAlignment <> 1 shl FileAlignmentLog
|
|
FileAlignmentLog = FileAlignmentLog + 1
|
|
end while
|
|
|
|
; header
|
|
dw 'PE' xor 'S' ; Signature
|
|
dw Characteristics or 0x100 ; IMAGE_FILE_32BIT_MACHINE
|
|
dd AddressOfEntryPoint
|
|
dd ImageBase
|
|
db SectionAlignmentLog
|
|
db FileAlignmentLog
|
|
db MajorSubsystemVersion
|
|
db MinorSubsystemVersion
|
|
dd SizeOfImage
|
|
dd SizeOfStackReserve
|
|
dd SizeOfHeapReserve
|
|
SizeOfHeadersField:
|
|
dd 0
|
|
db Subsystem
|
|
db DstNumberOfRvaAndSizes
|
|
dw NumberOfSections
|
|
; directories
|
|
irps dir,SPE_DIR_ORDER
|
|
{
|
|
if SPE_DIRECTORY_#dir < DstNumberOfRvaAndSizes
|
|
dd DirRVA_#dir, DirSize_#dir
|
|
end if
|
|
}
|
|
|
|
NumBytesDeleted = pehea + 0F8h - $ + NumberOfSections*0Ch
|
|
DeltaDeleted = NumBytesDeleted and not (FileAlignment - 1)
|
|
; Use store instead of declaring SizeOfHeaders - DeltaDeleted directly in dd
|
|
; to avoid the second compilation pass.
|
|
store dword SizeOfHeaders - DeltaDeleted at SizeOfHeadersField
|
|
TimeStampInExportTable = 0
|
|
; sections
|
|
repeat NumberOfSections
|
|
file filename:pehea+0F8h+(%-1)*28h,18h
|
|
load VirtualSize dword from $-10h
|
|
load VirtualAddress dword from $-0Ch
|
|
load SizeOfRawData dword from $-8
|
|
load PointerToRawData dword from $-4
|
|
PointerToRawData = PointerToRawData - DeltaDeleted
|
|
store dword PointerToRawData at $-4
|
|
if DirRVA_EXPORT <> 0 & DirRVA_EXPORT+4 >= VirtualAddress & DirRVA_EXPORT+8 <= VirtualAddress + VirtualSize & DirRVA_EXPORT+8 <= VirtualAddress + SizeOfRawData
|
|
TimeStampInExportTable = DirRVA_EXPORT+4 - VirtualAddress + PointerToRawData
|
|
end if
|
|
file filename:pehea+0F8h+(%-1)*28h+24h,4
|
|
end repeat
|
|
; padding to keep FileAlignment
|
|
times NumBytesDeleted - DeltaDeleted db 0
|
|
; data
|
|
file filename:pehea+0F8h+NumberOfSections*28h
|
|
if TimeDateStamp <> 0 & TimeStampInExportTable <> 0
|
|
store dword TimeDateStamp at TimeStampInExportTable
|
|
end if
|