start new makefile, migrate to libpicofe

[picodrive.git] / cpu / Cyclone / epoc / patchtable_symb2.c

#include <stdio.h>\r
#include <stdlib.h>\r
#include <string.h>\r
\r
#include <windows.h>\r
\r
#define symbols 2\r
\r
int main(int argc, char *argv[])\r
{\r
	FILE *f = 0;\r
	unsigned char pattern[8] = { 0x12, 0x34, 0x56, 0x78, 0x90, 0x12, 0x34, 0x56 };\r
	unsigned char *buff, *p;\r
	unsigned long patt_offset; // pattern offset in .text section\r
	unsigned long size = 0, i, insert_pos, *handler;//, symbols;\r
	unsigned short reloc_entry;\r
	IMAGE_BASE_RELOCATION *reloc_page;\r
	IMAGE_DOS_HEADER  *dos_header;\r
	IMAGE_FILE_HEADER *file_header;\r
	IMAGE_SECTION_HEADER *sect_header, *relocsect_header = 0, *codesect_header = 0;\r
\r
	if(argc != 2) {\r
		printf("usage: %s <pe_exe_or_app_before_petran>\n\n", argv[0]);\r
		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
		return 1;\r
	}\r
	\r
	f = fopen(argv[1], "rb+");\r
	if(!f) {\r
		printf("%s: couldn't open %s\n", argv[0], argv[1]);\r
		return 2;\r
	}\r
\r
	//symbols = atoi(argv[2]);\r
\r
	// read the file\r
	fseek(f,0,SEEK_END); size=ftell(f); fseek(f,0,SEEK_SET);\r
	buff = (unsigned char *) malloc(size);\r
	fread(buff,1,size,f);\r
\r
	dos_header = (IMAGE_DOS_HEADER *)      buff;\r
	file_header= (IMAGE_FILE_HEADER *)    (buff+dos_header->e_lfanew+4);\r
	sect_header= (IMAGE_SECTION_HEADER *) (buff+dos_header->e_lfanew+4+sizeof(IMAGE_FILE_HEADER)+sizeof(IMAGE_OPTIONAL_HEADER32));\r
\r
	if(size < 0x500 || dos_header->e_magic != IMAGE_DOS_SIGNATURE ||\r
	   *(DWORD *)(buff+dos_header->e_lfanew) != IMAGE_NT_SIGNATURE || file_header->Machine != 0x0A00) {\r
		printf("%s: not a PE executable image for ARM target.\n", argv[0]);\r
		fclose(f);\r
		free(buff);\r
		return 2;\r
	}\r
\r
	// scan all sections for data and reloc sections\r
	for(i = 0; i < file_header->NumberOfSections; i++, sect_header++) {\r
		     if(strncmp(sect_header->Name, ".text",  5) == 0) codesect_header  = sect_header;\r
		else if(strncmp(sect_header->Name, ".reloc", 6) == 0) relocsect_header = sect_header;\r
	}\r
	\r
	if(!codesect_header || !relocsect_header) {\r
		printf("%s: failed to find reloc and/or data section.\n", argv[0]);\r
		fclose(f);\r
		free(buff);\r
		return 3;\r
	}\r
\r
	if(relocsect_header != sect_header-1) {\r
		printf("%s: bug: reloc section is not last, this is unexpected and not supported.\n", argv[0]);\r
		fclose(f);\r
		free(buff);\r
		return 4;\r
	}\r
\r
	// find the pattern\r
	for(i = codesect_header->PointerToRawData; i < size-8; i+=2)\r
		if(memcmp(buff+i, pattern, 8) == 0) break;\r
	if(i == size-8 || i < 4) {\r
		printf("%s: failed to find the pattern.\n", argv[0]);\r
		fclose(f);\r
		free(buff);\r
		return 5;\r
	}\r
\r
	// calculate pattern offset in RVA (relative virtual address)\r
	patt_offset = i - codesect_header->PointerToRawData + codesect_header->VirtualAddress;\r
\r
	// replace the placeholders themselves\r
	handler = (unsigned long *) (buff + i - 4);\r
	for(i = 1; i <= symbols; i++)\r
		*(handler+i) = *handler;\r
\r
	// find suitable reloc section\r
	for(i = 0, p = buff+relocsect_header->PointerToRawData; i < relocsect_header->SizeOfRawData; ) {\r
		reloc_page = (IMAGE_BASE_RELOCATION *) p;\r
		if(patt_offset - reloc_page->VirtualAddress >= 0 && patt_offset - reloc_page->VirtualAddress < 0x1000 - symbols*2) break;\r
		i += reloc_page->SizeOfBlock;\r
		p += reloc_page->SizeOfBlock;\r
	}\r
\r
	if(i >= relocsect_header->SizeOfRawData) {\r
		printf("%s: suitable reloc section not found.\n", argv[0]);\r
		fclose(f);\r
		free(buff);\r
		return 6;\r
	}\r
\r
	// now find the insert pos and update everything\r
	insert_pos = p + reloc_page->SizeOfBlock - buff;\r
	reloc_page->SizeOfBlock += symbols*2;\r
	relocsect_header->SizeOfRawData += symbols*2;\r
\r
	// check for possible padding\r
	if(!*(buff+insert_pos-1)) insert_pos -= 2;\r
\r
	// write all this joy\r
	fseek(f,0,SEEK_SET);\r
	fwrite(buff, 1, insert_pos, f);\r
\r
	// write new reloc entries\r
	for(i = 0; i < symbols; i++) {\r
		handler++;\r
		reloc_entry = (unsigned short)(((unsigned char *) handler - buff) - reloc_page->VirtualAddress - codesect_header->PointerToRawData + codesect_header->VirtualAddress) | 0x3000; // quite nasty\r
		fwrite(&reloc_entry, 1, 2, f);\r
	}\r
\r
	// write the remaining data\r
	fwrite(buff+insert_pos, 1, size-insert_pos, f);\r
\r
	// done at last!\r
	fclose(f);\r
	free(buff);\r
\r
	return 0;\r
}\r