1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
|
//
// Copyright (c) Microsoft. All rights reserved.
// Licensed under the MIT license. See LICENSE file in the project root for full license information.
//
// ===========================================================================
// File: CeeFileGenWriter.h
//
// ===========================================================================
#ifndef _CEEFILEGENWRITER_H_
#define _CEEFILEGENWRITER_H_
//
// CeeFileGenWriter contains all the code necessary to actually write an exe
// while CCeeGen contains everything else. This lets CeeGen.exe and the VM
// share more code without forcing the VM to carry the extra code to write an
// exe.
#include <windef.h>
#include "ceegen.h"
#include "iceefilegen.h"
class PEWriter;
class CeeFileGenWriter;
// default setting for PE file
const UINT32 CEE_IMAGE_BASE_32 = 0x00400000;
const UINT64 CEE_IMAGE_BASE_64 = UI64(0x0000000140000000);
const int CEE_IMAGE_SUBSYSTEM_MAJOR_VERSION = 4;
const int CEE_IMAGE_SUBSYSTEM_MINOR_VERSION = 0;
class CeeFileGenWriter : public CCeeGen
{
mdToken m_entryPoint; // token for entry point
DWORD m_comImageFlags;
LPWSTR m_outputFileName;
LPWSTR m_resourceFileName;
LPWSTR m_libraryName;
GUID m_libraryGuid;
bool m_dllSwitch;
ULONG m_iatOffset;
DWORD m_dwMacroDefinitionRVA;
DWORD m_dwMacroDefinitionSize;
DWORD m_dwManifestRVA;
DWORD m_dwManifestSize;
DWORD m_dwStrongNameRVA;
DWORD m_dwStrongNameSize;
DWORD m_dwVTableRVA;
DWORD m_dwVTableSize;
bool m_linked;
bool m_fixed;
HRESULT checkForErrors();
struct IDataDllInfo {
const char *m_name;
int m_numMethods;
const char **m_methodName;
int m_iltOffset;
int m_ibnOffset;
int m_iatOffset;
int m_nameOffset;
} *m_iDataDlls;
int m_dllCount;
CeeSection *m_iDataSectionIAT;
int m_iDataOffsetIAT;
char *m_iDataIAT;
#ifdef EMIT_FIXUPS
CeeSection * m_sectionFixups;
IMAGE_DEBUG_DIRECTORY * m_pDebugDir;
bool m_fFixupsUpdated;
bool m_fEmitFixups;
#endif
HRESULT allocateIAT();
public:
// Create with one of these two methods, not operator new
static HRESULT CreateNewInstance(CCeeGen *pCeeFileGenFrom, CeeFileGenWriter* & pGenWriter,
DWORD createFlags = ICEE_CREATE_FILE_PURE_IL);
// See ICeeFileGen.h for the definition of the bits used in createFlags
static HRESULT CreateNewInstanceEx(CCeeGen *pCeeFileGenFrom, CeeFileGenWriter* & pGenWriter,
DWORD createFlags, LPCWSTR seedFileName = NULL);
virtual HRESULT Cleanup();
PEWriter &getPEWriter();
HRESULT link(); // Layout the sections and assign their starting addresses
HRESULT fixup(); // Apply relocations to any pointer data. Also generate PE base relocs
HRESULT generateImage(void **ppImage);
HRESULT setImageBase(size_t imageBase);
HRESULT setImageBase64(ULONGLONG imageBase);
HRESULT setFileAlignment(ULONG fileAlignment);
HRESULT setSubsystem(DWORD subsystem, DWORD major, DWORD minor);
HRESULT getMethodRVA(ULONG codeOffset, ULONG *codeRVA);
HRESULT setEntryPoint(mdMethodDef method);
mdMethodDef getEntryPoint();
HRESULT setComImageFlags(DWORD mask);
HRESULT clearComImageFlags(DWORD mask);
DWORD getComImageFlags();
HRESULT setOutputFileName(__in LPWSTR outputFileName);
LPWSTR getOutputFileName();
HRESULT setResourceFileName(__in LPWSTR resourceFileName);
LPWSTR getResourceFileName();
HRESULT setLibraryName(__in LPWSTR libraryName);
LPWSTR getLibraryName();
HRESULT setDirectoryEntry(CeeSection §ion, ULONG entry, ULONG size, ULONG offset=0);
HRESULT computeSectionOffset(CeeSection §ion, __in char *ptr,
unsigned *offset);
HRESULT computeOffset(__in char *ptr, CeeSection **pSection,
unsigned *offset);
HRESULT getCorHeader(IMAGE_COR20_HEADER **ppHeader);
HRESULT getFileTimeStamp(DWORD *pTimeStamp);
//<TODO>@FUTURE: this entry point is only here so that down level clients of this interface
// can import the method by name in the exports table using the original name.
// These things really ought to be exported through a v-table so there is no
// name mangling issues. It would make the export table much smaller as well.</TODO>
HRESULT emitLibraryName(IMetaDataEmit *emitter);
HRESULT setLibraryGuid(__in LPWSTR libraryGuid);
HRESULT setDllSwitch(bool dllSwitch);
bool getDllSwitch();
HRESULT setObjSwitch(bool objSwitch);
bool getObjSwitch();
HRESULT EmitMacroDefinitions(void *pData, DWORD cData);
HRESULT setManifestEntry(ULONG size, ULONG offset);
HRESULT setStrongNameEntry(ULONG size, ULONG offset);
HRESULT setEnCRvaBase(ULONG dataBase, ULONG rdataBase);
HRESULT setVTableEntry(ULONG size, ULONG offset);
HRESULT setVTableEntry64(ULONG size, void* ptr);
protected:
CeeFileGenWriter(); // ctor is protected
HRESULT emitResourceSection();
HRESULT emitExeMain();
HRESULT setAddrReloc(UCHAR *instrAddr, DWORD value);
HRESULT addAddrReloc(CeeSection &thisSection, UCHAR *instrAddr, DWORD offset, CeeSection *targetSection);
HRESULT MapTokens(CeeGenTokenMapper *pMapper, IMetaDataImport *pImport);
HRESULT MapTokensForMethod(CeeGenTokenMapper *pMapper,BYTE *pCode, LPCWSTR szMethodName);
#ifdef EMIT_FIXUPS
HRESULT InitFixupSection();
HRESULT UpdateFixups();
HRESULT setEmitFixups();
#ifdef TEST_EMIT_FIXUPS
HRESULT TestEmitFixups();
#endif
public:
HRESULT addFixup(CeeSection& secFixups, unsigned offset, CeeSectionRelocType reloc, CeeSection * relativeTo = NULL, CeeSectionRelocExtra *extra = 0);
#endif
};
inline PEWriter &CeeFileGenWriter::getPEWriter()
{
return (PEWriter &) *m_peSectionMan;
}
inline LPWSTR CeeFileGenWriter::getOutputFileName() {
return m_outputFileName;
}
inline LPWSTR CeeFileGenWriter::getResourceFileName() {
return m_resourceFileName;
}
inline HRESULT CeeFileGenWriter::setDllSwitch(bool dllSwitch) {
if(m_dllSwitch = dllSwitch) m_objSwitch = FALSE; return S_OK;
}
inline bool CeeFileGenWriter::getDllSwitch() {
return m_dllSwitch;
}
inline HRESULT CeeFileGenWriter::setObjSwitch(bool objSwitch) {
if(m_objSwitch = objSwitch) m_dllSwitch = FALSE; return S_OK;
}
inline bool CeeFileGenWriter::getObjSwitch() {
return m_objSwitch;
}
inline LPWSTR CeeFileGenWriter::getLibraryName() {
return m_libraryName;
}
inline mdMethodDef CeeFileGenWriter::getEntryPoint() {
return m_entryPoint;
}
inline HRESULT CeeFileGenWriter::setEntryPoint(mdMethodDef method) {
m_entryPoint = method;
return S_OK;
}
inline HRESULT CeeFileGenWriter::setComImageFlags(DWORD mask) {
m_comImageFlags |= mask; return S_OK;
}
inline HRESULT CeeFileGenWriter::clearComImageFlags(DWORD mask) {
m_comImageFlags &= ~mask; return S_OK;
}
inline DWORD CeeFileGenWriter::getComImageFlags() {
return m_comImageFlags;
}
//
#if defined(_IMAGE_FILE_4K_SECTION_ALIGNMENT_)
#define IMAGE_NT_OPTIONAL_HDR_SECTION_ALIGNMENT 0x1000
#else
#define IMAGE_NT_OPTIONAL_HDR_SECTION_ALIGNMENT 0x2000
#endif
// The stub is always x86 so we always mark the image as x86
#define IMAGE_FILE_MACHINE IMAGE_FILE_MACHINE_I386
#endif // _CEEFILEGENWRITER_H_
|
