| 1 | #include <stdio.h>\r |
| 2 | #include <stdlib.h>\r |
| 3 | #include <string.h>\r |
| 4 | \r |
| 5 | #include <windows.h>\r |
| 6 | \r |
| 7 | #define symbols 2\r |
| 8 | \r |
| 9 | int main(int argc, char *argv[])\r |
| 10 | {\r |
| 11 | FILE *f = 0;\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 |
| 21 | \r |
| 22 | if(argc != 2) {\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 |
| 25 | return 1;\r |
| 26 | }\r |
| 27 | \r |
| 28 | f = fopen(argv[1], "rb+");\r |
| 29 | if(!f) {\r |
| 30 | printf("%s: couldn't open %s\n", argv[0], argv[1]);\r |
| 31 | return 2;\r |
| 32 | }\r |
| 33 | \r |
| 34 | //symbols = atoi(argv[2]);\r |
| 35 | \r |
| 36 | // read the file\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 |
| 40 | \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 |
| 44 | \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 |
| 48 | fclose(f);\r |
| 49 | free(buff);\r |
| 50 | return 2;\r |
| 51 | }\r |
| 52 | \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 |
| 57 | }\r |
| 58 | \r |
| 59 | if(!codesect_header || !relocsect_header) {\r |
| 60 | printf("%s: failed to find reloc and/or data section.\n", argv[0]);\r |
| 61 | fclose(f);\r |
| 62 | free(buff);\r |
| 63 | return 3;\r |
| 64 | }\r |
| 65 | \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 |
| 68 | fclose(f);\r |
| 69 | free(buff);\r |
| 70 | return 4;\r |
| 71 | }\r |
| 72 | \r |
| 73 | // find the pattern\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 |
| 78 | fclose(f);\r |
| 79 | free(buff);\r |
| 80 | return 5;\r |
| 81 | }\r |
| 82 | \r |
| 83 | // calculate pattern offset in RVA (relative virtual address)\r |
| 84 | patt_offset = i - codesect_header->PointerToRawData + codesect_header->VirtualAddress;\r |
| 85 | \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 |
| 90 | \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 |
| 97 | }\r |
| 98 | \r |
| 99 | if(i >= relocsect_header->SizeOfRawData) {\r |
| 100 | printf("%s: suitable reloc section not found.\n", argv[0]);\r |
| 101 | fclose(f);\r |
| 102 | free(buff);\r |
| 103 | return 6;\r |
| 104 | }\r |
| 105 | \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 |
| 110 | \r |
| 111 | // check for possible padding\r |
| 112 | if(!*(buff+insert_pos-1)) insert_pos -= 2;\r |
| 113 | \r |
| 114 | // write all this joy\r |
| 115 | fseek(f,0,SEEK_SET);\r |
| 116 | fwrite(buff, 1, insert_pos, f);\r |
| 117 | \r |
| 118 | // write new reloc entries\r |
| 119 | for(i = 0; i < symbols; i++) {\r |
| 120 | handler++;\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 |
| 123 | }\r |
| 124 | \r |
| 125 | // write the remaining data\r |
| 126 | fwrite(buff+insert_pos, 1, size-insert_pos, f);\r |
| 127 | \r |
| 128 | // done at last!\r |
| 129 | fclose(f);\r |
| 130 | free(buff);\r |
| 131 | \r |
| 132 | return 0;\r |
| 133 | }\r |