cpu/Cyclone/epoc/patchtable_symb2.c

   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