source: trunk/src/kernel32/pefile.cpp@ 1036

Last change on this file since 1036 was 956, checked in by sandervl, 26 years ago

Rewrite for new win32 image classes
File size: 7.4 KB
Line 
1/* $Id: pefile.cpp,v 1.5 1999-09-15 23:38:01 sandervl Exp $ */
2
3/*
4 * PE2LX PE utility functions
5 *
6 * Project Odin Software License can be found in LICENSE.TXT
7 *
8 */
9#define INCL_BASE
10#include <os2wrap.h> //Odin32 OS/2 api wrappers
11#include <stdio.h>
12#include <string.h>
13#include <stdlib.h>
14#ifdef __WATCOMC__
15#include <mem.h>
16#endif
17#include <win32type.h>
18#include <pefile.h>
19#include <misc.h>
20#include <winres.h>
21
22//******************************************************************************
23//******************************************************************************
24char *UnicodeToFixedAsciiString(int length, WCHAR *NameString)
25{
26static char asciistring[256];
27int i;
28
29 if(length >= 255) length = 255;
30 for(i=0;i<length;i++) {
31 asciistring[i] = NameString[i] & 0xFF;
32 }
33 asciistring[length] = 0;
34 return(asciistring);
35}
36//******************************************************************************
37//******************************************************************************
38BOOL GetPEFileHeader (LPVOID lpFile, PIMAGE_FILE_HEADER pHeader)
39{
40 if(*(USHORT *)lpFile == IMAGE_DOS_SIGNATURE &&
41 *(DWORD *)PE_HEADER (lpFile) == IMAGE_NT_SIGNATURE)
42 {
43 memcpy ((LPVOID)pHeader, PEHEADEROFF (lpFile), sizeof (IMAGE_FILE_HEADER));
44 return TRUE;
45 }
46 else return FALSE;
47}
48//******************************************************************************
49//******************************************************************************
50BOOL GetPEOptionalHeader (LPVOID lpFile, PIMAGE_OPTIONAL_HEADER pHeader)
51{
52 if(*(USHORT *)lpFile == IMAGE_DOS_SIGNATURE &&
53 *(DWORD *)PE_HEADER (lpFile) == IMAGE_NT_SIGNATURE)
54 {
55 memcpy ((LPVOID)pHeader, OPTHEADEROFF (lpFile), sizeof (IMAGE_OPTIONAL_HEADER));
56 return TRUE;
57 }
58 else return FALSE;
59}
60//******************************************************************************
61//******************************************************************************
62LPVOID ImageDirectoryOffset (LPVOID lpFile, DWORD dwIMAGE_DIRECTORY)
63{
64 PIMAGE_OPTIONAL_HEADER poh = (PIMAGE_OPTIONAL_HEADER)OPTHEADEROFF (lpFile);
65 IMAGE_SECTION_HEADER sh;
66
67 if (dwIMAGE_DIRECTORY >= poh->NumberOfRvaAndSizes)
68 return NULL;
69
70 if(GetSectionHdrByRVA(lpFile, &sh, poh->DataDirectory[dwIMAGE_DIRECTORY].VirtualAddress) == FALSE)
71 {
72 return NULL;
73 }
74
75 return (LPVOID)(((ULONG)lpFile + (ULONG)(poh->DataDirectory[dwIMAGE_DIRECTORY].VirtualAddress -
76 sh.VirtualAddress) + (ULONG)sh.PointerToRawData));
77}
78//******************************************************************************
79//******************************************************************************
80BOOL GetSectionHdrByImageDir(LPVOID lpFile, DWORD dwIMAGE_DIRECTORY, PIMAGE_SECTION_HEADER pSect)
81{
82 PIMAGE_OPTIONAL_HEADER poh = (PIMAGE_OPTIONAL_HEADER)OPTHEADEROFF (lpFile);
83
84 if (dwIMAGE_DIRECTORY >= poh->NumberOfRvaAndSizes)
85 return FALSE;
86
87 return GetSectionHdrByRVA(lpFile, pSect, poh->DataDirectory[dwIMAGE_DIRECTORY].VirtualAddress);
88}
89//******************************************************************************
90//******************************************************************************
91BOOL IsImportSection(LPVOID lpFile, PIMAGE_SECTION_HEADER psh)
92{
93 PIMAGE_OPTIONAL_HEADER poh = (PIMAGE_OPTIONAL_HEADER)OPTHEADEROFF (lpFile);
94 int i = 0;
95 DWORD ImageDirVA;
96
97 ImageDirVA = poh->DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT].VirtualAddress;
98
99 if(psh->VirtualAddress <= ImageDirVA &&
100 psh->VirtualAddress + max(psh->Misc.VirtualSize,psh->SizeOfRawData) > ImageDirVA &&
101 strcmp(psh->Name, ".idata") == 0)
102 {
103 return TRUE;
104 }
105 return FALSE;
106}
107//******************************************************************************
108//******************************************************************************
109BOOL GetSectionHdrByName (LPVOID lpFile, IMAGE_SECTION_HEADER *sh, char *szSection)
110{
111 PIMAGE_SECTION_HEADER psh;
112 int nSections = NR_SECTIONS (lpFile);
113 int i;
114
115 if((psh = (PIMAGE_SECTION_HEADER)SECTIONHDROFF (lpFile)) != NULL)
116 {
117 for(i=0; i<nSections; i++)
118 {
119 if(strcmp (psh->Name, szSection) == 0)
120 {
121 memcpy ((LPVOID)sh, (LPVOID)psh, sizeof (IMAGE_SECTION_HEADER));
122 return TRUE;
123 }
124 else psh++;
125 }
126 }
127 return FALSE;
128}
129//******************************************************************************
130//******************************************************************************
131BOOL GetSectionHdrByType (LPVOID lpFile, IMAGE_SECTION_HEADER *sh, int type)
132{
133 PIMAGE_SECTION_HEADER psh;
134 int nSections = NR_SECTIONS (lpFile);
135 int i;
136
137 if((psh = (PIMAGE_SECTION_HEADER)SECTIONHDROFF (lpFile)) != NULL)
138 {
139 for(i=0; i<nSections; i++)
140 {
141 if(psh->Characteristics & type)
142 {
143 memcpy ((LPVOID)sh, (LPVOID)psh, sizeof (IMAGE_SECTION_HEADER));
144 return TRUE;
145 }
146 else psh++;
147 }
148 }
149 return FALSE;
150}
151//******************************************************************************
152//******************************************************************************
153int GetNumberOfResources(LPVOID lpFile)
154{
155 PIMAGE_RESOURCE_DIRECTORY prdRoot, prdType;
156 PIMAGE_RESOURCE_DIRECTORY_ENTRY prde;
157 int nCnt=0, i, j, id;
158 char *resname;
159
160 if ((prdRoot = (PIMAGE_RESOURCE_DIRECTORY)ImageDirectoryOffset
161 (lpFile, IMAGE_DIRECTORY_ENTRY_RESOURCE)) == NULL)
162 return 0;
163
164 prde = (PIMAGE_RESOURCE_DIRECTORY_ENTRY)((DWORD)prdRoot + sizeof (IMAGE_RESOURCE_DIRECTORY));
165
166 for (i=0; i<prdRoot->NumberOfNamedEntries+prdRoot->NumberOfIdEntries; i++)
167 {
168 prdType = (PIMAGE_RESOURCE_DIRECTORY)((ULONG)prdRoot + (ULONG)prde->u2.OffsetToData);
169
170 if(i<prdRoot->NumberOfNamedEntries) {
171 //SvL: 30-10-'97, high bit is set, so clear to get real offset
172 prde->u1.Name &= ~0x80000000;
173 for(j=0;j<MAX_RES;j++) {
174 resname = UnicodeToFixedAsciiString(*(WCHAR *)((ULONG)prdRoot + (ULONG)prde->u1.Name), (WCHAR *)((ULONG)prdRoot + (ULONG)prde->u1.Name + sizeof(WCHAR))); // first word = string length
175 if(strcmp(resname, ResTypes[j]) == 0)
176 break;
177 }
178 if(j == MAX_RES) {
179 //SvL: 30-10-'97, not found = custom resource type (correct?)
180 id = NTRT_RCDATA;
181 }
182 else id = j;
183 }
184 else id = prde->u1.Id;
185
186 prdType = (PIMAGE_RESOURCE_DIRECTORY)((DWORD)prdType ^ 0x80000000);
187
188 if(id == NTRT_STRING) {
189 //String tables can contain up to 16 individual resources!
190 nCnt += prdType->NumberOfNamedEntries*16 + prdType->NumberOfIdEntries*16;
191 }
192 else {
193 //Only icon groups are stored as resources in the LX file
194 //Icon groups contain one or more icons
195 if(id != NTRT_ICON)
196 nCnt += prdType->NumberOfNamedEntries + prdType->NumberOfIdEntries;
197 }
198 prde++;
199 }
200
201 return nCnt;
202}
203/** Get Section Header for the given RVA - returns boolean according to the result
204 *
205 * knut [Jul 22 1998 2:53am]
206 */
207BOOL GetSectionHdrByRVA (LPVOID lpFile, IMAGE_SECTION_HEADER *sh, ULONG rva)
208{
209 PIMAGE_SECTION_HEADER psh;
210 int nSections = NR_SECTIONS (lpFile);
211 int i;
212
213 if ((psh = (PIMAGE_SECTION_HEADER)SECTIONHDROFF (lpFile)) != NULL)
214 {
215 for (i=0; i<nSections; i++)
216 {
217 if (rva >= psh->VirtualAddress && rva < psh->VirtualAddress + max(psh->Misc.VirtualSize,psh->SizeOfRawData))
218 {
219 memcpy (sh, psh, sizeof(IMAGE_SECTION_HEADER));
220 return TRUE;
221 }
222 else psh++;
223 }
224 }
225 return FALSE;
226}
227//******************************************************************************
228//******************************************************************************
Note: See TracBrowser for help on using the repository browser.