9 int main(int argc, char *argv[])
\r
12 unsigned char pattern[8] = { 0x12, 0x34, 0x56, 0x78, 0x90, 0x12, 0x34, 0x56 };
\r
13 unsigned char *buff, *p;
\r
14 unsigned long patt_offset; // pattern offset in .text section
\r
15 unsigned long size = 0, i, insert_pos, *handler;//, symbols;
\r
16 unsigned short reloc_entry;
\r
17 IMAGE_BASE_RELOCATION *reloc_page;
\r
18 IMAGE_DOS_HEADER *dos_header;
\r
19 IMAGE_FILE_HEADER *file_header;
\r
20 IMAGE_SECTION_HEADER *sect_header, *relocsect_header = 0, *codesect_header = 0;
\r
23 printf("usage: %s <pe_exe_or_app_before_petran>\n\n", argv[0]);
\r
24 printf("note: this was written to fix a problem related to Cyclone and as v.2.9-psion-98r2 and shouldn't be used for anything else. See Readme.\n", argv[0]);
\r
28 f = fopen(argv[1], "rb+");
\r
30 printf("%s: couldn't open %s\n", argv[0], argv[1]);
\r
34 //symbols = atoi(argv[2]);
\r
37 fseek(f,0,SEEK_END); size=ftell(f); fseek(f,0,SEEK_SET);
\r
38 buff = (unsigned char *) malloc(size);
\r
39 fread(buff,1,size,f);
\r
41 dos_header = (IMAGE_DOS_HEADER *) buff;
\r
42 file_header= (IMAGE_FILE_HEADER *) (buff+dos_header->e_lfanew+4);
\r
43 sect_header= (IMAGE_SECTION_HEADER *) (buff+dos_header->e_lfanew+4+sizeof(IMAGE_FILE_HEADER)+sizeof(IMAGE_OPTIONAL_HEADER32));
\r
45 if(size < 0x500 || dos_header->e_magic != IMAGE_DOS_SIGNATURE ||
\r
46 *(DWORD *)(buff+dos_header->e_lfanew) != IMAGE_NT_SIGNATURE || file_header->Machine != 0x0A00) {
\r
47 printf("%s: not a PE executable image for ARM target.\n", argv[0]);
\r
53 // scan all sections for data and reloc sections
\r
54 for(i = 0; i < file_header->NumberOfSections; i++, sect_header++) {
\r
55 if(strncmp(sect_header->Name, ".text", 5) == 0) codesect_header = sect_header;
\r
56 else if(strncmp(sect_header->Name, ".reloc", 6) == 0) relocsect_header = sect_header;
\r
59 if(!codesect_header || !relocsect_header) {
\r
60 printf("%s: failed to find reloc and/or data section.\n", argv[0]);
\r
66 if(relocsect_header != sect_header-1) {
\r
67 printf("%s: bug: reloc section is not last, this is unexpected and not supported.\n", argv[0]);
\r
74 for(i = codesect_header->PointerToRawData; i < size-8; i+=2)
\r
75 if(memcmp(buff+i, pattern, 8) == 0) break;
\r
76 if(i == size-8 || i < 4) {
\r
77 printf("%s: failed to find the pattern.\n", argv[0]);
\r
83 // calculate pattern offset in RVA (relative virtual address)
\r
84 patt_offset = i - codesect_header->PointerToRawData + codesect_header->VirtualAddress;
\r
86 // replace the placeholders themselves
\r
87 handler = (unsigned long *) (buff + i - 4);
\r
88 for(i = 1; i <= symbols; i++)
\r
89 *(handler+i) = *handler;
\r
91 // find suitable reloc section
\r
92 for(i = 0, p = buff+relocsect_header->PointerToRawData; i < relocsect_header->SizeOfRawData; ) {
\r
93 reloc_page = (IMAGE_BASE_RELOCATION *) p;
\r
94 if(patt_offset - reloc_page->VirtualAddress >= 0 && patt_offset - reloc_page->VirtualAddress < 0x1000 - symbols*2) break;
\r
95 i += reloc_page->SizeOfBlock;
\r
96 p += reloc_page->SizeOfBlock;
\r
99 if(i >= relocsect_header->SizeOfRawData) {
\r
100 printf("%s: suitable reloc section not found.\n", argv[0]);
\r
106 // now find the insert pos and update everything
\r
107 insert_pos = p + reloc_page->SizeOfBlock - buff;
\r
108 reloc_page->SizeOfBlock += symbols*2;
\r
109 relocsect_header->SizeOfRawData += symbols*2;
\r
111 // check for possible padding
\r
112 if(!*(buff+insert_pos-1)) insert_pos -= 2;
\r
114 // write all this joy
\r
115 fseek(f,0,SEEK_SET);
\r
116 fwrite(buff, 1, insert_pos, f);
\r
118 // write new reloc entries
\r
119 for(i = 0; i < symbols; i++) {
\r
121 reloc_entry = (unsigned short)(((unsigned char *) handler - buff) - reloc_page->VirtualAddress - codesect_header->PointerToRawData + codesect_header->VirtualAddress) | 0x3000; // quite nasty
\r
122 fwrite(&reloc_entry, 1, 2, f);
\r
125 // write the remaining data
\r
126 fwrite(buff+insert_pos, 1, size-insert_pos, f);
\r