source: trunk/src/kernel32/winimagebase.h@ 3313

Last change on this file since 3313 was 3059, checked in by sandervl, 25 years ago

Dll dependency changes
File size: 5.8 KB
Line 
1/* $Id: winimagebase.h,v 1.1 2000-03-09 19:03:22 sandervl Exp $ */
2
3/*
4 * Win32 PE 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 __WINIMAGEBASE_H__
13#define __WINIMAGEBASE_H__
14
15#include <peexe.h>
16#include "queue.h"
17
18#define MAGIC_WINIMAGE 0x11223344
19
20#ifndef CCHMAXPATH
21#define CCHMAXPATH 260
22#endif
23
24#define LANG_GETFIRST 0x80000000
25#define ID_GETFIRST LANG_GETFIRST
26#define IDLANG_GETFIRST LANG_GETFIRST
27
28#ifndef ENUMRESNAMEPROC
29 typedef BOOL (* CALLBACK ENUMRESTYPEPROCA)(HMODULE,LPSTR,LONG);
30 typedef BOOL (* CALLBACK ENUMRESTYPEPROCW)(HMODULE,LPWSTR,LONG);
31 typedef BOOL (* CALLBACK ENUMRESNAMEPROCA)(HMODULE,LPCSTR,LPSTR,LONG);
32 typedef BOOL (* CALLBACK ENUMRESNAMEPROCW)(HMODULE,LPCWSTR,LPWSTR,LONG);
33 typedef BOOL (* CALLBACK ENUMRESLANGPROCA)(HMODULE,LPCSTR,LPCSTR,WORD,LONG);
34 typedef BOOL (* CALLBACK ENUMRESLANGPROCW)(HMODULE,LPCWSTR,LPCWSTR,WORD,LONG);
35#endif
36
37class Win32Resource;
38class Win32DllBase;
39
40class Win32ImageBase
41{
42protected:
43 DWORD magic;
44public:
45 void checkObject()
46 {
47 if(magic != MAGIC_WINIMAGE) {
48 eprintf(("Corrupt this pointer %X %X!!", this, magic));
49 DebugInt3();
50 }
51 };
52
53public:
54 // Constructors and destructors
55 Win32ImageBase(HINSTANCE hInstance);
56virtual ~Win32ImageBase();
57
58 ULONG getError() { return errorState; };
59 HINSTANCE getInstanceHandle() { return hinstance; };
60
61//Returns required OS version for this image
62virtual ULONG getVersion();
63
64virtual void setFullPath(char *name);
65 char *getFullPath() { return fullpath; };
66
67 char *getModuleName() { return szModule; };
68
69virtual HRSRC findResourceA(LPCSTR lpszName, LPSTR lpszType, ULONG lang = LANG_GETFIRST);
70 HRSRC findResourceW(LPWSTR lpszName, LPWSTR lpszType, ULONG lang = LANG_GETFIRST);
71virtual ULONG getResourceSizeA(LPCSTR lpszName, LPSTR lpszType, ULONG lang = LANG_GETFIRST);
72 ULONG getResourceSizeW(LPCWSTR lpszName, LPWSTR lpszType, ULONG lang = LANG_GETFIRST);
73virtual BOOL enumResourceNamesA(HMODULE hmod, LPCTSTR lpszType, ENUMRESNAMEPROCA lpEnumFunc, LONG lParam);
74virtual BOOL enumResourceNamesW(HMODULE hmod, LPCWSTR lpszType, ENUMRESNAMEPROCW lpEnumFunc, LONG lParam);
75virtual BOOL enumResourceTypesA(HMODULE hmod, ENUMRESTYPEPROCA lpEnumFunc,
76 LONG lParam);
77virtual BOOL enumResourceTypesW(HMODULE hmod, ENUMRESTYPEPROCW lpEnumFunc,
78 LONG lParam);
79
80virtual ULONG getVersionSize();
81virtual BOOL getVersionStruct(char *verstruct, ULONG bufLength);
82
83static BOOL isPEImage(char *szFileName);
84
85 void setEntryPoint(ULONG startAddress) { entryPoint = startAddress; };
86
87 void setTLSAddress(LPVOID dwTlsAddress) { tlsAddress = dwTlsAddress; };
88 void setTLSIndexAddr(LPDWORD dwTlsIndexAddr) { tlsIndexAddr = dwTlsIndexAddr; };
89 void setTLSInitSize(ULONG dwTlsSize) { tlsInitSize = dwTlsSize; };
90 void setTLSTotalSize(ULONG dwTlsSize) { tlsTotalSize = dwTlsSize; };
91 void setTLSCallBackAddr(PIMAGE_TLS_CALLBACK *dwTlsCallBackAddr)
92 {
93 tlsCallBackAddr = dwTlsCallBackAddr;
94 };
95
96 void tlsAttachThread(); //setup TLS structures for new thread
97 void tlsDetachThread(); //destroy TLS structures
98
99virtual ULONG getApi(char *name) = 0;
100virtual ULONG getApi(int ordinal) = 0;
101
102virtual BOOL isDll() = 0;
103
104static Win32ImageBase * findModule(HMODULE hModule);
105
106 //Add image to dependency list of this image
107 void addDependency(Win32DllBase *dll);
108 BOOL dependsOn(Win32DllBase *dll);
109
110protected:
111 void tlsAlloc(); //Allocate TLS index for this module
112 void tlsDelete(); //Destroy TLS index for this module
113
114 Win32Resource *winres;
115
116 ULONG errorState, entryPoint;
117
118 char *fullpath;
119 char szModule[CCHMAXPATH];
120 char szFileName[CCHMAXPATH];
121
122 HINSTANCE hinstance;
123
124 LPVOID tlsAddress; //address of TLS data
125 LPDWORD tlsIndexAddr; //address of DWORD that receives the TLS index
126 ULONG tlsInitSize; //size of initialized TLS memory block
127 ULONG tlsTotalSize; //size of TLS memory block
128 PIMAGE_TLS_CALLBACK *tlsCallBackAddr; //ptr to TLS callback array
129 ULONG tlsIndex; //module TLS index
130
131 ULONG getPEResourceSize(ULONG id, ULONG type, ULONG lang = LANG_GETFIRST);
132
133 PIMAGE_RESOURCE_DATA_ENTRY getPEResourceEntry(ULONG id, ULONG type, ULONG lang = LANG_GETFIRST);
134 PIMAGE_RESOURCE_DATA_ENTRY ProcessResSubDir(PIMAGE_RESOURCE_DIRECTORY prdType,
135 ULONG *nodeData, int level);
136 PIMAGE_RESOURCE_DIRECTORY getResSubDirW(PIMAGE_RESOURCE_DIRECTORY pResDir, LPCWSTR lpszName);
137 PIMAGE_RESOURCE_DIRECTORY getResSubDirA(PIMAGE_RESOURCE_DIRECTORY pResDir, LPCTSTR lpszName);
138
139 PIMAGE_RESOURCE_DIRECTORY pResDir;
140
141 //substracted from RVA data offsets
142 ULONG ulRVAResourceSection;
143
144 //linked list of dlls loaded on behalf of this executable image (dll or exe)
145 Queue loadedDlls;
146private:
147
148 friend class Win32Resource;
149 friend ULONG SYSTEM GetVersionSize(char *modname);
150};
151
152//SvL: This structure is placed at the end of the first page of the image (header
153// page), so we can determine the Win32Image pointer from a HINSTANCE variable
154// (which is actually the address of the win32 module)
155typedef struct
156{
157 ULONG magic1;
158 Win32ImageBase *image;
159 ULONG magic2;
160} WINIMAGE_LOOKUP;
161
162#define WINIMAGE_LOOKUPADDR(a) (WINIMAGE_LOOKUP *)((ULONG)a + PAGE_SIZE - sizeof(WINIMAGE_LOOKUP))
163
164#endif //__WINIMAGEBASE_H__
Note: See TracBrowser for help on using the repository browser.