| 1 | /* $Id: winimagepeldr.h,v 1.8 1999-11-30 19:39:14 sandervl Exp $ */
|
|---|
| 2 |
|
|---|
| 3 | /*
|
|---|
| 4 | * Win32 PE loader Image base class
|
|---|
| 5 | *
|
|---|
| 6 | * Copyright 1998-1999 Sander van Leeuwen (sandervl@xs4all.nl)
|
|---|
| 7 | *
|
|---|
| 8 | *
|
|---|
| 9 | * Project Odin Software License can be found in LICENSE.TXT
|
|---|
| 10 | *
|
|---|
| 11 | */
|
|---|
| 12 | #ifndef __WINIMAGEPELDR_H__
|
|---|
| 13 | #define __WINIMAGEPELDR_H__
|
|---|
| 14 |
|
|---|
| 15 | #include <winimagebase.h>
|
|---|
| 16 |
|
|---|
| 17 | #define SINGLE_PAGE 0 //commit single page
|
|---|
| 18 | #define COMPLETE_SECTION 1 //commit entire section
|
|---|
| 19 | #define SECTION_PAGES 2 //commit default nr of pages
|
|---|
| 20 |
|
|---|
| 21 | #define DEFAULT_NR_PAGES 16 //default nr of pages to commit during exception
|
|---|
| 22 |
|
|---|
| 23 | //SvL: To load a dll/exe for i.e. getting a single resource (GetVersionSize/Resource)
|
|---|
| 24 | #define REAL_LOAD 0
|
|---|
| 25 | #define RSRC_LOAD 1
|
|---|
| 26 |
|
|---|
| 27 | //SvL: Amount of memory the peldr dll reserves for win32 exes without fixups
|
|---|
| 28 | //(most of them need to be loaded at 4 MB; except MS Office apps of course)
|
|---|
| 29 | #define PELDR_RESERVEDMEMSIZE 16*1024*1024
|
|---|
| 30 |
|
|---|
| 31 | #define ERROR_INTERNAL 1
|
|---|
| 32 |
|
|---|
| 33 | #define SECTION_CODE 1
|
|---|
| 34 | #define SECTION_INITDATA 2
|
|---|
| 35 | #define SECTION_UNINITDATA 4
|
|---|
| 36 | #define SECTION_READONLYDATA 8
|
|---|
| 37 | #define SECTION_IMPORT 16
|
|---|
| 38 | #define SECTION_RESOURCE 32
|
|---|
| 39 | #define SECTION_RELOC 64
|
|---|
| 40 | #define SECTION_EXPORT 128
|
|---|
| 41 | #define SECTION_DEBUG 256
|
|---|
| 42 | #define SECTION_TLS 512
|
|---|
| 43 |
|
|---|
| 44 | #define PAGE_SIZE 4096
|
|---|
| 45 |
|
|---|
| 46 | #define MAX_SECTION 64 /*PLF Mon 98-02-09 23:47:16*/
|
|---|
| 47 |
|
|---|
| 48 | typedef struct {
|
|---|
| 49 | ULONG rawoffset;
|
|---|
| 50 | ULONG rawsize;
|
|---|
| 51 | ULONG virtaddr;
|
|---|
| 52 | ULONG realvirtaddr; //as allocated in OS/2
|
|---|
| 53 | ULONG virtualsize;
|
|---|
| 54 | ULONG type;
|
|---|
| 55 | ULONG pageflags;
|
|---|
| 56 | } Section;
|
|---|
| 57 |
|
|---|
| 58 | typedef struct {
|
|---|
| 59 | ULONG virtaddr;
|
|---|
| 60 | ULONG ordinal;
|
|---|
| 61 | ULONG nlength;
|
|---|
| 62 | char name[4];
|
|---|
| 63 | } NameExport;
|
|---|
| 64 |
|
|---|
| 65 | typedef struct {
|
|---|
| 66 | ULONG virtaddr;
|
|---|
| 67 | ULONG ordinal;
|
|---|
| 68 | } OrdExport;
|
|---|
| 69 |
|
|---|
| 70 | class Win32DllBase;
|
|---|
| 71 | class Win32MemMap;
|
|---|
| 72 |
|
|---|
| 73 | class Win32PeLdrImage : public virtual Win32ImageBase
|
|---|
| 74 | {
|
|---|
| 75 | public:
|
|---|
| 76 | Win32PeLdrImage(char *szFileName, BOOL isExe, int loadtype = REAL_LOAD);
|
|---|
| 77 | virtual ~Win32PeLdrImage();
|
|---|
| 78 |
|
|---|
| 79 | //reservedMem is address of memory reserved in peldr.dll (allocated before
|
|---|
| 80 | //any dlls are loaded, so that exes without fixups can be loaded at a low
|
|---|
| 81 | //address)
|
|---|
| 82 | virtual BOOL init(ULONG reservedMem);
|
|---|
| 83 |
|
|---|
| 84 | virtual ULONG getApi(char *name);
|
|---|
| 85 | virtual ULONG getApi(int ordinal);
|
|---|
| 86 |
|
|---|
| 87 | //Returns required OS version for this image
|
|---|
| 88 | virtual ULONG getVersion();
|
|---|
| 89 |
|
|---|
| 90 | //commits image page(s) when an access violation exception is dispatched
|
|---|
| 91 | BOOL commitPage(ULONG virtAddress, BOOL fWriteAccess, int fPageCmd = SECTION_PAGES);
|
|---|
| 92 |
|
|---|
| 93 | protected:
|
|---|
| 94 | void StoreImportByOrd(Win32DllBase *WinDll, ULONG ordinal, ULONG impaddr);
|
|---|
| 95 | void StoreImportByName(Win32DllBase *WinDll, char *impname, ULONG impaddr);
|
|---|
| 96 |
|
|---|
| 97 | void addSection(ULONG type, ULONG rawoffset, ULONG rawsize, ULONG virtaddress, ULONG virtsize);
|
|---|
| 98 | BOOL allocSections(ULONG reservedMem);
|
|---|
| 99 | BOOL allocFixedMem(ULONG reservedMem);
|
|---|
| 100 | Section *findSection(ULONG type);
|
|---|
| 101 | Section *findSectionByAddr(ULONG addr);
|
|---|
| 102 | Section *findSectionByOS2Addr(ULONG addr);
|
|---|
| 103 | Section *findPreviousSectionByOS2Addr(ULONG addr);
|
|---|
| 104 |
|
|---|
| 105 | BOOL setMemFlags();
|
|---|
| 106 | BOOL setFixups(PIMAGE_BASE_RELOCATION prel);
|
|---|
| 107 | BOOL setFixups(ULONG virtAddress, ULONG size);
|
|---|
| 108 | void AddOff32Fixup(ULONG fixupaddr);
|
|---|
| 109 | void AddOff16Fixup(ULONG fixupaddr, BOOL fHighFixup);
|
|---|
| 110 |
|
|---|
| 111 | BOOL processImports(char *win32file);
|
|---|
| 112 |
|
|---|
| 113 | BOOL processExports(char *win32file);
|
|---|
| 114 | void AddNameExport(ULONG virtaddr, char *apiname, ULONG ordinal);
|
|---|
| 115 | void AddOrdExport(ULONG virtaddr, ULONG ordinal);
|
|---|
| 116 |
|
|---|
| 117 | IMAGE_OPTIONAL_HEADER oh;
|
|---|
| 118 | IMAGE_FILE_HEADER fh;
|
|---|
| 119 |
|
|---|
| 120 | ULONG nrNameExports, nameExportSize;
|
|---|
| 121 | ULONG nrOrdExports;
|
|---|
| 122 | NameExport *nameexports, *curnameexport;
|
|---|
| 123 | OrdExport *ordexports, *curordexport;
|
|---|
| 124 |
|
|---|
| 125 | ULONG nrsections, imageSize, imageVirtBase, imageVirtEnd;
|
|---|
| 126 | //OS/2 virtual base address
|
|---|
| 127 | ULONG realBaseAddress;
|
|---|
| 128 | Section section[MAX_SECTION];
|
|---|
| 129 |
|
|---|
| 130 | ULONG loadType;
|
|---|
| 131 |
|
|---|
| 132 | HFILE hFile;
|
|---|
| 133 |
|
|---|
| 134 | PIMAGE_BASE_RELOCATION pFixups;
|
|---|
| 135 |
|
|---|
| 136 | Win32MemMap *memmap;
|
|---|
| 137 | private:
|
|---|
| 138 | };
|
|---|
| 139 |
|
|---|
| 140 | #include <iostream.h>
|
|---|
| 141 | #include <fstream.h>
|
|---|
| 142 | extern ofstream fout;
|
|---|
| 143 |
|
|---|
| 144 | #endif //__WINIMAGEPELDR_H__
|
|---|
| 145 |
|
|---|
