source: dev/BasicCompiler64/MakePeHdr.cpp@ 88

Last change on this file since 88 was 88, checked in by dai_9181, 17 years ago

House→Smoothie
Sourceクラスを用意した。
File size: 54.3 KB
Line 
1#include "../BasicCompiler_Common/common.h"
2#include "Opcode.h"
3
4
5////////////////////////////
6// 特殊関数の構造体ポインタ
7////////////////////////////
8UserProc
9 *pSub_System_StartupProgram,
10 *pSub_DebugSys_StartProc,
11 *pSub_DebugSys_EndProc,
12 *pSub_DebugSys_SaveContext,
13 *pSub_System_GetEip,
14 *pSub_System_InitDllGlobalVariables,
15 *pSub_System_InitStaticLocalVariables,
16 *pSub_System_Call_Destructor_of_GlobalObject,
17 *pSub_System_GetSp,
18 *pSub_pow,
19 *pSub_calloc,
20 *pSub_realloc,
21 *pSub_free,
22 *pSub_System_GC_malloc_ForObject,
23 *pSub_System_GC_malloc_ForObjectPtr,
24 *pSub_System_GC_free_for_SweepingDelete,
25 *pSubStaticMethod_System_TypeBase_InitializeUserTypes;
26
27
28//////////////////////////////
29// 各セクションの位置とサイズ
30int FileSize_CodeSection,
31 FileSize_ExportSection,
32 FileSize_ImportSection,
33 FileSize_DataSection,
34 FileSize_RWSection,
35 FileSize_RSrcSection,
36 FileSize_RelocSection,
37 FileSize_DebugSection;
38int FilePos_CodeSection,
39 FilePos_ExportSection,
40 FilePos_ImportSection,
41 FilePos_DataSection,
42 FilePos_RWSection,
43 FilePos_RSrcSection,
44 FilePos_RelocSection,
45 FilePos_DebugSection;
46int MemSize_CodeSection,
47 MemSize_ExportSection,
48 MemSize_ImportSection,
49 MemSize_DataSection,
50 MemSize_RWSection,
51 MemSize_RSrcSection,
52 MemSize_RelocSection,
53 MemSize_DebugSection;
54int MemPos_CodeSection,
55 MemPos_ExportSection,
56 MemPos_ImportSection,
57 MemPos_DataSection,
58 MemPos_RWSection,
59 MemPos_RSrcSection,
60 MemPos_RelocSection,
61 MemPos_DebugSection;
62int bUse_CodeSection,
63 bUse_ExportSection,
64 bUse_ImportSection,
65 bUse_DataSection,
66 bUse_RWSection,
67 bUse_RSrcSection,
68 bUse_RelocSection,
69 bUse_DebugSection;
70
71
72void DebugVariable(void){
73 char temporary[255];
74 sprintf(temporary,"_DebugSys_dwThreadID[255]%c%cDWord",1,ESC_AS);
75 OpcodeDim(temporary,DIMFLAG_INITDEBUGVAR);
76 sprintf(temporary,"_DebugSys_ProcNum[255]%c%cDWord",1,ESC_AS);
77 OpcodeDim(temporary,DIMFLAG_INITDEBUGVAR);
78 sprintf(temporary,"_DebugSys_lplpObp[255]%c%c*ULONG_PTR",1,ESC_AS);
79 OpcodeDim(temporary,DIMFLAG_INITDEBUGVAR);
80 sprintf(temporary,"_DebugSys_lplpSpBase[255]%c%c*ULONG_PTR",1,ESC_AS);
81 OpcodeDim(temporary,DIMFLAG_INITDEBUGVAR);
82}
83
84void Compile(void){
85 extern HWND hMainDlg;
86 extern HWND hOwnerEditor;
87 extern HANDLE hHeap;
88 extern BOOL bDebugCompile;
89 extern BOOL bDll;
90 extern DWORD ImageBase;
91 extern int obp_AllocSize;
92 extern char *basbuf;
93 extern char OutputFileName[MAX_PATH];
94 int i,i2,i3,i4,i5;
95 char temporary[MAX_PATH],*temp2;
96 HANDLE hFile;
97
98
99 //コードセクションのメモリ上の位置
100 MemPos_CodeSection= 0x1000;
101
102 //データセクションのメモリ上の位置（仮定）
103 MemPos_DataSection= 0x10000000;
104
105
106 //エクスポート セクションを利用するかどうか
107 if(bDll) bUse_ExportSection=1;
108 else bUse_ExportSection=0;
109
110
111 //////////////////
112 // データテーブル
113 dataTable.Init();
114 if(bDebugCompile){
115 dataTable.Add( (long)0x00000002 );
116 }
117
118
119 //////////////////////
120 // コード生成前の準備
121 //////////////////////
122
123 //重複エラー情報管理のメモリを確保（グローバル領域コンパイル用）
124 extern char **SynonymErrorWords;
125 extern int SynonymErrorNum;
126 SynonymErrorNum=0;
127 SynonymErrorWords=(char **)HeapAlloc(hHeap,0,1);
128
129 //列挙体に関する情報の初期化
130 CEnumParent::InitEnum();
131
132 //列挙体からクラスコードを生成
133 char *temp;
134 temp=CEnumParent::GenerateCodes();
135 AddSourceCode(temp);
136 HeapDefaultFree(temp);
137
138 //関数ポインタ情報を初期化
139 Smoothie::Meta::procPointers.clear();
140
141 //クラス名を取得（詳細情報はGetAllClassInfoで取得）
142 // GetSubInfo関数の中で参照されるオブジェクト名を事前に取得する。
143 // ※オブジェクトの内容までは取得しない
144 pobj_DBClass=new CDBClass();
145 pobj_DBClass->InitNames();
146
147 //TypeDef情報を初期化
148 Smoothie::Meta::typeDefs.Init();
149
150 //定数情報を取得
151 GetConstInfo();
152
153 //サブルーチン（ユーザー定義、DLL関数）の識別子、アドレスを取得
154 pobj_CompilingClass=0;
155 GetSubInfo();
156
157 //クラス情報を取得（※注 - GetSubInfoの後に呼び出す）
158 pobj_DBClass->GetAllClassInfo();
159
160 // 実行時型情報の確定コード（InitializeUserTypesメソッドの実装コード）を生成
161 pobj_DBClass->GenerateRuntimeTypeInfo();
162
163 //コードと行番号の関係
164 extern LINEINFO *pLineInfo;
165 extern int MaxLineInfoNum;
166 pLineInfo=(LINEINFO *)HeapAlloc(hHeap,0,1);
167 MaxLineInfoNum=0;
168
169
170 ///////////////////////////
171 // 最低限必要な関数を取得
172 ///////////////////////////
173 cp=-1;
174
175 if(pSub_System_StartupProgram=GetSubHash("_System_StartupProgram",1))
176 pSub_System_StartupProgram->Using();
177
178 if(pSub_DebugSys_StartProc=GetSubHash("_DebugSys_StartProc",1))
179 pSub_DebugSys_StartProc->Using();
180
181 if(pSub_DebugSys_EndProc=GetSubHash("_DebugSys_EndProc",1))
182 pSub_DebugSys_EndProc->Using();
183
184 if(pSub_DebugSys_SaveContext=GetSubHash("_DebugSys_SaveContext",1))
185 pSub_DebugSys_SaveContext->Using();
186
187 if(pSub_System_GetEip=GetSubHash("_System_GetEip",1)){
188 pSub_System_GetEip->Using();
189 pSub_System_GetEip->ThisIsSystemProc();
190 }
191
192 if(pSub_System_InitDllGlobalVariables=GetSubHash("_System_InitDllGlobalVariables",1)){
193 pSub_System_InitDllGlobalVariables->Using();
194 pSub_System_InitDllGlobalVariables->ThisIsSystemProc();
195 }
196
197 if(pSub_System_InitStaticLocalVariables=GetSubHash("_System_InitStaticLocalVariables",1)){
198 pSub_System_InitStaticLocalVariables->Using();
199 pSub_System_InitStaticLocalVariables->ThisIsSystemProc();
200 }
201
202 if(pSub_System_Call_Destructor_of_GlobalObject=GetSubHash("_System_Call_Destructor_of_GlobalObject",1)){
203 pSub_System_Call_Destructor_of_GlobalObject->Using();
204 pSub_System_Call_Destructor_of_GlobalObject->ThisIsSystemProc();
205 }
206
207 if(pSub_System_GetSp=GetSubHash("_System_GetSp",1)){
208 pSub_System_GetSp->Using();
209 pSub_System_GetSp->ThisIsSystemProc();
210 }
211
212 if(pSub_pow=GetSubHash("pow",1))
213 pSub_pow->Using();
214
215 if(pSub_calloc=GetSubHash("calloc",1))
216 pSub_calloc->Using();
217
218 if(pSub_realloc=GetSubHash("realloc",1))
219 pSub_realloc->Using();
220
221 if(pSub_free=GetSubHash("free",1))
222 pSub_free->Using();
223
224 if( pSub_System_GC_malloc_ForObject = GetSubHash( "_System_GC_malloc_ForObject",1 ) )
225 pSub_System_GC_malloc_ForObject->Using();
226
227 if( pSub_System_GC_malloc_ForObjectPtr = GetSubHash( "_System_GC_malloc_ForObjectPtr",1 ) )
228 pSub_System_GC_malloc_ForObjectPtr->Using();
229
230 if( pSub_System_GC_free_for_SweepingDelete = GetSubHash( "_System_GC_free_for_SweepingDelete",1 ) )
231 pSub_System_GC_free_for_SweepingDelete->Using();
232
233 if( pSubStaticMethod_System_TypeBase_InitializeUserTypes = GetSubHash( "_System_TypeBase.InitializeUserTypes",1 ) ){
234 pSubStaticMethod_System_TypeBase_InitializeUserTypes->Using();
235 pSubStaticMethod_System_TypeBase_InitializeUserTypes->ThisIsSystemProc();
236 }
237
238
239
240 //データテーブルスケジュール
241 pobj_DataTableSchedule=new CSchedule();
242 pobj_DataTableSchedule->SetFlag(SCHEDULE_FLAG_RELOC);
243
244 //インポートテーブルスケジュール
245 //※x64はRPI（実行中アドレス）からのオフセットになるため、再配置が不要
246 pobj_ImportAddrSchedule=new CImportAddrSchedule();
247
248 //プロシージャポインタスケジュール
249 pobj_SubAddrSchedule=new CSubAddrSchedule();
250
251 //グローバル変数アドレススケジュール
252 pobj_GlobalVarSchedule=new CSchedule();
253 pobj_GlobalVarSchedule->SetFlag(SCHEDULE_FLAG_RELOC);
254
255 //グローバル変数の初期バッファ
256 extern BYTE *initGlobalBuf;
257 initGlobalBuf=(BYTE *)HeapAlloc(hHeap,HEAP_ZERO_MEMORY,1);
258
259 //リロケーション情報
260 pobj_Reloc=new CReloc();
261
262 //一時スケジューリングポインタ
263 pobj_TempSchedule=new CTempSchedule();
264
265 obp_AllocSize=8192*2;
266 OpBuffer=(char *)HeapAlloc(hHeap,HEAP_ZERO_MEMORY,obp_AllocSize);
267 obp=0;
268
269 //レジスタのブロッキングを管理するためのオブジェクトを生成
270 pobj_BlockReg=new CBlockReg;
271
272 //レキシカルスコープ情報を初期化
273 obj_LexScopes.Init(obp);
274
275
276 /////////////////////////////////////////////////////////////////
277 // デバッグコンパイル用のログを生成する
278 /////////////////////////////////////////////////////////////////
279#ifdef _DEBUG
280 {
281 ofstream ofs("middle_code.txt");
282 ofs << basbuf << endl;
283 ofs.close();
284 }
285#endif
286
287
288 //////////////////////
289 // グローバル実行領域
290 //////////////////////
291
292 cp=-1;
293 UserProc::CompileStartForGlobalArea();
294
295 if(!bDll){
296 //ラベル用のメモリを確保
297 extern LABEL *pLabelNames;
298 extern int MaxLabelNum;
299 pLabelNames=(LABEL *)HeapAlloc(hHeap,0,1);
300 MaxLabelNum=0;
301
302 //Gotoラベルスケジュール
303 extern GOTOLABELSCHEDULE *pGotoLabelSchedule;
304 extern int GotoLabelScheduleNum;
305 pGotoLabelSchedule=(GOTOLABELSCHEDULE *)HeapAlloc(hHeap,0,1);
306 GotoLabelScheduleNum=0;
307
308 //With情報のメモリを確保
309 extern WITHINFO WithInfo;
310 WithInfo.ppName=(char **)HeapAlloc(hHeap,0,1);
311 WithInfo.pWithCp=(int *)HeapAlloc(hHeap,0,1);
312 WithInfo.num=0;
313
314 //Continueアドレスを初期化
315 extern DWORD dwContinueAddress;
316 dwContinueAddress=-1;
317
318 //スタックフレーム管理用クラスを初期化
319 extern CStackFrame *pobj_sf;
320 pobj_sf=new CStackFrame();
321
322
323 //未完成
324 //OpBuffer[obp++]=(char)0xCC;
325
326 //sub rsp,スタックフレームサイズ
327 int StackFrameSchedule;
328 op_sub_rsp(0xFFFFFFFF);
329 StackFrameSchedule=obp-sizeof(long);
330
331 if(bDebugCompile){
332 //デバッグ用の変数を定義
333 DebugVariable();
334 }
335
336 //GC用の変数を定義
337 InitGCVariables();
338
339 //_System_StartupProgramの呼び出し
340 op_call(pSub_System_StartupProgram);
341
342 //クラスに属する静的メンバを定義
343 CMember::InitStaticMember();
344
345 //グローバル実行領域をコンパイル開始
346 CompileBuffer(0,0);
347
348 //ラベル用のメモリを解放
349 for(i=0;i<MaxLabelNum;i++){
350 if(pLabelNames[i].pName) HeapDefaultFree(pLabelNames[i].pName);
351 }
352 HeapDefaultFree(pLabelNames);
353
354 //Goto未知ラベルスケジュールを解放
355 for(i=0;i<GotoLabelScheduleNum;i++){
356 if(pGotoLabelSchedule[i].pName){
357 SetError(6,pGotoLabelSchedule[i].pName,pGotoLabelSchedule[i].now_cp);
358 HeapDefaultFree(pGotoLabelSchedule[i].pName);
359 }
360 else{
361 sprintf(temporary,"%d",pGotoLabelSchedule[i].line);
362 SetError(6,temporary,pGotoLabelSchedule[i].now_cp);
363 }
364 }
365 HeapDefaultFree(pGotoLabelSchedule);
366
367
368 ///////////////////////////////////////
369 // グローバル文字列変数の解放処理
370 ///////////////////////////////////////
371
372 //call _System_Call_Destructor_of_GlobalObject
373 extern UserProc *pSub_System_Call_Destructor_of_GlobalObject;
374 op_call(pSub_System_Call_Destructor_of_GlobalObject);
375
376
377 //add rsp,スタックフレームサイズ
378 op_add_RV(REG_RSP,pobj_sf->GetFrameSize());
379
380 //xor rax,rax（raxを0に初期化する）
381 op_zero_reg(REG_RAX);
382
383 //ret
384 OpBuffer[obp++]=(char)0xC3;
385
386 //グローバル実行領域のコードサイズ
387 extern int GlobalOpBufferSize;
388 GlobalOpBufferSize=obp;
389
390 //スタックフレームスケジュール（subコマンドに渡す値）
391 *((long *)(OpBuffer+StackFrameSchedule))=pobj_sf->GetFrameSize();
392
393 //スタックフレームスケジュールを実行
394 pobj_sf->RunningSchedule();
395 delete pobj_sf;
396 pobj_sf=0;
397
398 //With情報のメモリを解放
399 for(i=0;i<WithInfo.num;i++){
400 SetError(22,"With",WithInfo.pWithCp[i]);
401 HeapDefaultFree(WithInfo.ppName[i]);
402 }
403 HeapDefaultFree(WithInfo.ppName);
404 HeapDefaultFree(WithInfo.pWithCp);
405 }
406 else{
407 ////////////////
408 // DLLの場合
409 ////////////////
410
411
412 }
413
414 //重複エラー情報管理のメモリを解放（グローバル領域用）
415 for(i=0;i<SynonymErrorNum;i++) HeapDefaultFree(SynonymErrorWords[i]);
416 HeapDefaultFree(SynonymErrorWords);
417
418
419 StepCompileProgress();
420
421
422 /////////////////////
423 // ローカル実行領域
424 /////////////////////
425
426 //プロシージャをコンパイル開始
427 cp=0;
428 CompileLocal();
429
430 // 終了
431 ///////////////////////
432
433 StepCompileProgress();
434
435
436 //レジスタのブロッキングを管理するためのオブジェクトを破棄
437 delete pobj_BlockReg;
438
439 //ネイティブコード生成はここまで
440 //////////////////////////////////////////////////////////
441
442
443
444 ////////////////////////////////
445 // 使用するDLL関数のチェック
446 ////////////////////////////////
447 for(i=0;i<MAX_HASH;i++){
448 extern DllProc **ppDeclareHash;
449 const DllProc *pDllProc = ppDeclareHash[i];
450 while(pDllProc){
451 if( !pDllProc->IsUsing() ){
452 pDllProc=pDllProc->pNextData;
453 continue;
454 }
455
456 //エラーチェック
457 HINSTANCE hLib;
458 hLib=LoadLibrary( pDllProc->GetDllFileName().c_str() );
459 if(!hLib){
460 extern char OutputFileName[MAX_PATH];
461 char temp2[MAX_PATH],temp3[MAX_PATH];
462 _splitpath(OutputFileName,temp2,temp3,NULL,NULL);
463 lstrcat(temp2,temp3);
464 lstrcpy(temp3,pDllProc->GetDllFileName().c_str());
465 GetFullPath(temp3,temp2);
466 hLib=LoadLibrary(temp3);
467
468 if(!hLib){
469 SetError(-106,pDllProc->GetDllFileName().c_str(),pDllProc->GetCodePos());
470 }
471 }
472
473 if(hLib){
474 if(!GetProcAddress(hLib,pDllProc->GetAlias().c_str())){
475 FreeLibrary(hLib);
476 SetError(-107,pDllProc->GetAlias(),pDllProc->GetCodePos());
477 }
478 FreeLibrary(hLib);
479 }
480
481 pDllProc=pDllProc->pNextData;
482 }
483 }
484
485
486
487 /////////////////////////////
488 // リソースデータを読み込む
489 /////////////////////////////
490 extern char ResourceFileName[MAX_PATH];
491 GetResourceData(ResourceFileName);
492
493
494 //////////////////////////////
495 // エクスポート情報（DLLのみ）
496 //////////////////////////////
497 IMAGE_EXPORT_DIRECTORY ImageExportDirectory;
498 DWORD *lpdwExportAddressTable;
499 DWORD *lpdwExportNamePointerTable;
500 WORD *lpwExportOrdinalTable;
501 char lpExportNames[8192];
502 int ExportNamesLength;
503 int ExportNum;
504 int DllMain_EntryPoint;
505
506 DllMain_EntryPoint=-1;
507 ExportNum=0;
508 ExportNamesLength=0;
509 lpdwExportAddressTable=(DWORD *)HeapAlloc(hHeap,0,1);
510 lpdwExportNamePointerTable=(DWORD *)HeapAlloc(hHeap,0,1);
511 lpwExportOrdinalTable=(WORD *)HeapAlloc(hHeap,0,1);
512
513 if(bUse_ExportSection){
514 _splitpath(OutputFileName,NULL,NULL,lpExportNames,temporary);
515 lstrcat(lpExportNames,temporary);
516 ExportNamesLength=lstrlen(lpExportNames)+1;
517
518 extern UserProc **ppSubHash;
519 UserProc *pUserProc,*psi2;
520 while(1){
521 //辞書順にサーチ
522 temporary[0]=0;
523 for(i=0,psi2=0;i<MAX_HASH;i++){
524 pUserProc=ppSubHash[i];
525 while(pUserProc){
526 if(pUserProc->IsExport()){
527 if(temporary[0]=='\0'){
528 lstrcpy(temporary,pUserProc->GetName().c_str());
529 psi2=pUserProc;
530 }
531 else{
532 i3=lstrlen(temporary);
533 i4=(int)pUserProc->GetName().size();
534 if(i3<i4) i3=i4;
535 if(memcmp(temporary,pUserProc->GetName().c_str(),i3)>0){
536 lstrcpy(temporary,pUserProc->GetName().c_str());
537 psi2=pUserProc;
538 }
539 }
540 }
541 pUserProc=pUserProc->pNextData;
542 }
543 }
544 if(psi2==0) break;
545 pUserProc=psi2;
546
547 pUserProc->ExportOff();
548
549 if( pUserProc->GetName() == "DllMain" ){
550 DllMain_EntryPoint=pUserProc->beginOpAddress;
551 }
552
553 lpdwExportAddressTable=(DWORD *)HeapReAlloc(hHeap,0,lpdwExportAddressTable,(ExportNum+1)*sizeof(DWORD));
554 lpdwExportAddressTable[ExportNum]=pUserProc->beginOpAddress;
555
556 lpdwExportNamePointerTable=(DWORD *)HeapReAlloc(hHeap,0,lpdwExportNamePointerTable,(ExportNum+1)*sizeof(DWORD));
557 lpdwExportNamePointerTable[ExportNum]=ExportNamesLength;
558
559 lpwExportOrdinalTable=(WORD *)HeapReAlloc(hHeap,0,lpwExportOrdinalTable,(ExportNum+1)*sizeof(WORD));
560 lpwExportOrdinalTable[ExportNum]=ExportNum;
561
562 lstrcpy(lpExportNames+ExportNamesLength,pUserProc->GetName().c_str());
563 ExportNamesLength+=lstrlen(lpExportNames+ExportNamesLength)+1;
564
565 ExportNum++;
566 }
567 for(i=0;i<ExportNum;i++){
568 lpdwExportNamePointerTable[i]+=
569 sizeof(IMAGE_EXPORT_DIRECTORY)+ //エクスポートディレクトリテーブルを飛び越す
570 ExportNum*sizeof(DWORD)+ //エクスポート アドレス テーブルを飛び越す
571 ExportNum*sizeof(DWORD)+ //エクスポート名ポインタ テーブルを飛び越す
572 ExportNum*sizeof(WORD); //エクスポート序数テーブルを飛び越す
573 }
574
575 ImageExportDirectory.Characteristics=0;
576 ImageExportDirectory.TimeDateStamp=(DWORD)time(NULL);
577 ImageExportDirectory.MajorVersion=0;
578 ImageExportDirectory.MinorVersion=0;
579 ImageExportDirectory.Name=sizeof(IMAGE_EXPORT_DIRECTORY)+
580 ExportNum*sizeof(DWORD)+ //エクスポート アドレス テーブルを飛び越す
581 ExportNum*sizeof(DWORD)+ //エクスポート名ポインタ テーブルを飛び越す
582 ExportNum*sizeof(WORD); //エクスポート序数テーブルを飛び越す
583 ImageExportDirectory.Base=1;
584 ImageExportDirectory.NumberOfFunctions=ExportNum;
585 ImageExportDirectory.NumberOfNames=ExportNum;
586 ImageExportDirectory.AddressOfFunctions=sizeof(IMAGE_EXPORT_DIRECTORY);
587 ImageExportDirectory.AddressOfNames=ImageExportDirectory.AddressOfFunctions+ExportNum*sizeof(DWORD);
588 ImageExportDirectory.AddressOfNameOrdinals=ImageExportDirectory.AddressOfNames+ExportNum*sizeof(DWORD);
589 }
590
591
592 /////////////////////
593 //インポートDLL情報
594 /////////////////////
595 extern DllProc **ppDeclareHash;
596 DllProc *pDllProc;
597 char **ppDllNames;
598 int ImportDllNum;
599
600 /*
601 インポート アドレス テーブル（ルックアップと同じ内容だが、プログラム実行時に実行アドレスに書き換えられる）
602 ルックアップ テーブル（ヒントを示すRVA）
603 IMAGE_IMPORT_DESCRIPTOR1[size=0x14] インポートヘッダ
604 IMAGE_IMPORT_DESCRIPTOR2[size=0x14]
605 IMAGE_IMPORT_DESCRIPTOR3[size=0x14]
606 ...
607 Dll名1[size=0x10]
608 Dll名2[size=0x10]
609 Dll名3[size=0x10]
610 ...
611 ヒントテーブル（関数名を羅列したもの）
612 */
613
614 /*
615 IMAGE_IMPORT_DESCRIPTOR1[size=0x14] インポートヘッダ
616 IMAGE_IMPORT_DESCRIPTOR2[size=0x14]
617 IMAGE_IMPORT_DESCRIPTOR3[size=0x14]
618 ...
619 Dll名1[size=0x10]
620 Dll名2[size=0x10]
621 Dll名3[size=0x10]
622 ...
623 ルックアップ テーブル（ヒントを示すRVA）
624 ヒントテーブル
625 インポート アドレス テーブル（ルックアップと同じ内容だが、プログラム実行時に実行アドレスに書き換えられる）
626 */
627
628 ppDllNames=(char **)HeapAlloc(hHeap,0,1);
629
630 ImportDllNum=0;
631 for(i=0;i<MAX_HASH;i++){
632 const DllProc *pDllProc=ppDeclareHash[i];
633 while(pDllProc){
634 if( !pDllProc->IsUsing() ){
635 pDllProc=pDllProc->pNextData;
636 continue;
637 }
638
639 if( pDllProc->GetDllFileName().size() > 16 ){
640 SetError(7,NULL,cp);
641 break;
642 }
643 for(i2=0;i2<ImportDllNum;i2++){
644 if( pDllProc->GetDllFileName() == ppDllNames[i2] ){
645 break;
646 }
647 }
648 if(i2==ImportDllNum){
649 ppDllNames=(char **)HeapReAlloc(hHeap,0,ppDllNames,(ImportDllNum+1)*sizeof(char **));
650 ppDllNames[ImportDllNum]=(char *)HeapAlloc(hHeap,HEAP_ZERO_MEMORY,16);
651 lstrcpy(ppDllNames[ImportDllNum],pDllProc->GetDllFileName().c_str());
652 ImportDllNum++;
653 }
654
655 pDllProc=pDllProc->pNextData;
656 }
657 }
658
659 //IMAGE_IMPORT_DESCRIPTOR、及びルックアップ テーブル サイズの計算
660 IMAGE_IMPORT_DESCRIPTOR *pImportDescriptor;
661 int LookupSize;
662 LookupSize=0;
663 pImportDescriptor=(IMAGE_IMPORT_DESCRIPTOR *)HeapAlloc(hHeap,0,(ImportDllNum+1)*sizeof(IMAGE_IMPORT_DESCRIPTOR));
664 for(i=0;i<ImportDllNum;i++){
665 //IMAGE_IMPORT_DESCRIPTOR
666 pImportDescriptor[i].OriginalFirstThunk=LookupSize; //※すぐ下で再計算
667 pImportDescriptor[i].TimeDateStamp=0;
668 pImportDescriptor[i].ForwarderChain=0;
669 pImportDescriptor[i].Name=i*0x10; //※すぐ下で再計算
670 for(i2=0;i2<MAX_HASH;i2++){
671 pDllProc=ppDeclareHash[i2];
672 while(pDllProc){
673 if( !pDllProc->IsUsing() ){
674 pDllProc=pDllProc->pNextData;
675 continue;
676 }
677
678 if( pDllProc->GetDllFileName() == ppDllNames[i] ){
679 //ルックアップデータのサイズを追加
680 LookupSize+=sizeof(_int64);
681 }
682
683 pDllProc=pDllProc->pNextData;
684 }
685 }
686 LookupSize+=sizeof(_int64); //NULL用
687 }
688 memset(&pImportDescriptor[i],0,sizeof(IMAGE_IMPORT_DESCRIPTOR));
689 for(i=0;i<ImportDllNum;i++){
690 //インポートアドレステーブル分の差分を追加
691 pImportDescriptor[i].OriginalFirstThunk+=LookupSize;
692
693 //DLL名への差分を追加
694 pImportDescriptor[i].Name+=
695 LookupSize+ //インポート アドレス テーブル
696 LookupSize+ //ルックアップテーブル
697 (ImportDllNum+1)*sizeof(IMAGE_IMPORT_DESCRIPTOR); //IMAGE_IMPORT_DESCRIPTOR
698 }
699
700 //ルックアップ テーブル、ヒント テーブル
701 unsigned _int64 *pLookupTable;
702 pLookupTable=(unsigned _int64 *)HeapAlloc(hHeap,0,LookupSize*sizeof(_int64)+1);
703 char *pHintTable;
704 int HintSize,HintAllocSize;
705 HintSize=0;
706 HintAllocSize=128*2;
707 pHintTable=(char *)HeapAlloc(hHeap,0,HintAllocSize);
708 for(i=0,i5=0;i<ImportDllNum;i++){
709 for(i2=0;i2<MAX_HASH;i2++){
710 pDllProc=ppDeclareHash[i2];
711 while(pDllProc){
712 if( !pDllProc->IsUsing() ){
713 pDllProc=pDllProc->pNextData;
714 continue;
715 }
716
717 if( pDllProc->GetDllFileName() == ppDllNames[i] ){
718 //ルックアップの位置をセット（後にインポート アドレス テーブルにセットしなおされる）
719 pDllProc->SetLookupAddress( i5*sizeof(_int64) );
720
721 //ルックアップ テーブルの値をセット
722 pLookupTable[i5++]=
723 LookupSize+ //インポート アドレス テーブル
724 LookupSize+ //ルックアップテーブル
725 (ImportDllNum+1)*sizeof(IMAGE_IMPORT_DESCRIPTOR)+ //IMAGE_IMPORT_DESCRIPTOR
726 ImportDllNum*0x10+ //DLL名の羅列
727 HintSize; //ヒント名へのオフセット
728
729 //ヒント テーブル
730 pHintTable[HintSize++]=0;
731 pHintTable[HintSize++]=0;
732 lstrcpy(pHintTable+HintSize,pDllProc->GetAlias().c_str());
733 i4=(int)pDllProc->GetAlias().size();
734 HintSize+=i4+1;
735 if(i4%2==0) pHintTable[HintSize++]=0;
736
737 if(HintAllocSize<HintSize+128){
738 HintAllocSize+=128;
739 pHintTable=(char *)HeapReAlloc(hHeap,0,pHintTable,HintAllocSize);
740 }
741 }
742
743 pDllProc=pDllProc->pNextData;
744 }
745 }
746 pLookupTable[i5++]=0;
747 }
748
749
750 if(bDll){
751 //DLLの場合はリロケーション情報を仮生成
752 //※正式な生成は各セクションのメモリ上のサイズが決定してから再度行う。
753 pobj_Reloc->ResetRelocBuffer();
754 }
755
756
757
758 //グローバル変数情報を扱う構造体も初期バッファの有無による配置を行う
759 //（デバッグ情報で利用される）
760 extern int AllInitGlobalVarSize;
761 foreach( Variable *pVar, globalVars ){
762 if(pVar->offset&0x80000000){
763 pVar->offset=(pVar->offset&0x7FFFFFFF)+AllInitGlobalVarSize;
764 }
765 }
766
767
768
769 ////////////////////////////////////
770 // デバッグセクションを生成
771 ////////////////////////////////////
772
773 //機械語コードとBasicコードの関係を整理
774 extern LINEINFO *pLineInfo;
775 extern int MaxLineInfoNum;
776 int MinObp,MaxObp;
777 LINEINFO *pTempLineInfo;
778 pTempLineInfo=(LINEINFO *)HeapAlloc(hHeap,0,MaxLineInfoNum*sizeof(LINEINFO));
779 i2=0;
780 MaxObp=0;
781 while(1){
782 for(i3=0,MinObp=0x0FFFFFFF;i3<MaxLineInfoNum;i3++){
783 if(pLineInfo[i3].TopObp<MinObp&&MaxObp<pLineInfo[i3].TopObp){
784 MinObp=pLineInfo[i3].TopObp;
785 i4=i3;
786 }
787 }
788 if(MinObp==0x0FFFFFFF) break;
789 pTempLineInfo[i2]=pLineInfo[i4];
790 MaxObp=pTempLineInfo[i2].TopObp;
791 i2++;
792 }
793 HeapDefaultFree(pLineInfo);
794 pLineInfo=(LINEINFO *)HeapAlloc(hHeap,0,i2*sizeof(LINEINFO));
795 memcpy(pLineInfo,pTempLineInfo,i2*sizeof(LINEINFO));
796 MaxLineInfoNum=i2;
797 HeapDefaultFree(pTempLineInfo);
798
799 //デバッグセクションを生成
800 CDebugSection *pobj_DebugSection;
801 pobj_DebugSection=new CDebugSection();
802 extern BOOL bDebugCompile;
803 extern BOOL bError;
804 if(bDebugCompile&&bError==0){
805 pobj_DebugSection->make();
806 }
807
808
809
810 /////////////////////////////////////
811 // 各セクションの位置とサイズを計算
812 /////////////////////////////////////
813
814 //コードセッションのファイル上のサイズ
815 if(obp%FILE_ALIGNMENT) FileSize_CodeSection=obp+(FILE_ALIGNMENT-obp%FILE_ALIGNMENT);
816 else FileSize_CodeSection=obp;
817 if(FileSize_CodeSection) bUse_CodeSection=1;
818 else bUse_CodeSection=0;
819
820 //エクスポートセクションのファイル上のサイズ
821 i= sizeof(IMAGE_EXPORT_DIRECTORY)+ //エクスポートディレクトリテーブルを飛び越す
822 ExportNum*sizeof(DWORD)+ //エクスポート アドレス テーブルを飛び越す
823 ExportNum*sizeof(DWORD)+ //エクスポート名ポインタ テーブルを飛び越す
824 ExportNum*sizeof(WORD)+ //エクスポート序数テーブルを飛び越す
825 ExportNamesLength; //シンボル名バッファ
826 if(bUse_ExportSection){
827 if(i%FILE_ALIGNMENT) FileSize_ExportSection=i+(FILE_ALIGNMENT-i%FILE_ALIGNMENT);
828 else FileSize_ExportSection=i;
829 }
830 else FileSize_ExportSection=0;
831
832 //インポートセクションのファイル上のサイズ
833 i=
834 LookupSize+ //インポート アドレス テーブル
835 LookupSize+ //ルックアップテーブル
836 (ImportDllNum+1)*sizeof(IMAGE_IMPORT_DESCRIPTOR)+ //IMAGE_IMPORT_DESCRIPTOR
837 ImportDllNum*0x10+ //DLL名の羅列
838 HintSize; //ヒント名
839 if(i%FILE_ALIGNMENT) FileSize_ImportSection=i+(FILE_ALIGNMENT-i%FILE_ALIGNMENT);
840 else FileSize_ImportSection=i;
841 bUse_ImportSection=1; //インポートセクションは必ず存在する
842
843 //データセクションのファイル上のサイズ
844 if(dataTable.GetSize()%FILE_ALIGNMENT) FileSize_DataSection=dataTable.GetSize()+(FILE_ALIGNMENT-dataTable.GetSize()%FILE_ALIGNMENT);
845 else FileSize_DataSection=dataTable.GetSize();
846 if(FileSize_DataSection) bUse_DataSection=1;
847 else bUse_DataSection=0;
848
849 //リライタブルセクションのファイル上のサイズ（グローバル変数の初期情報のみを格納）
850 extern int AllInitGlobalVarSize;
851 if(AllInitGlobalVarSize%FILE_ALIGNMENT) FileSize_RWSection=AllInitGlobalVarSize+(FILE_ALIGNMENT-AllInitGlobalVarSize%FILE_ALIGNMENT);
852 else{
853 if(AllInitGlobalVarSize) FileSize_RWSection=AllInitGlobalVarSize;
854 else FileSize_RWSection=FILE_ALIGNMENT;
855 }
856 bUse_RWSection=1;
857
858 //リソースセクションのファイル上のサイズ
859 char *RSrcSectionBuffer;
860 int RSrcSectionSize;
861 RSrcSectionBuffer=GetRSrcSectionBuffer(&RSrcSectionSize);
862 if(RSrcSectionSize%FILE_ALIGNMENT) FileSize_RSrcSection=RSrcSectionSize+(FILE_ALIGNMENT-RSrcSectionSize%FILE_ALIGNMENT);
863 else FileSize_RSrcSection=RSrcSectionSize;
864 if(FileSize_RSrcSection) bUse_RSrcSection=1;
865 else bUse_RSrcSection=0;
866
867 //リロケーションセクションのファイル上のサイズ
868 if(pobj_Reloc->length%FILE_ALIGNMENT) FileSize_RelocSection=pobj_Reloc->length+(FILE_ALIGNMENT-pobj_Reloc->length%FILE_ALIGNMENT);
869 else FileSize_RelocSection=pobj_Reloc->length;
870 if(FileSize_RelocSection) bUse_RelocSection=1;
871 else bUse_RelocSection=0;
872
873 //デバッグセクションのファイル上のサイズ
874 if(pobj_DebugSection->length%FILE_ALIGNMENT) FileSize_DebugSection=pobj_DebugSection->length+(FILE_ALIGNMENT-pobj_DebugSection->length%FILE_ALIGNMENT);
875 else FileSize_DebugSection=pobj_DebugSection->length;
876 if(FileSize_DebugSection) bUse_DebugSection=1;
877 else bUse_DebugSection=0;
878
879
880 //各セッションのファイル上の位置
881 FilePos_CodeSection= EXE_HEADER_SIZE;
882 FilePos_ExportSection= FilePos_CodeSection+
883 FileSize_CodeSection;
884 FilePos_ImportSection= FilePos_ExportSection+
885 FileSize_ExportSection;
886 FilePos_DataSection= FilePos_ImportSection+
887 FileSize_ImportSection;
888 FilePos_RWSection= FilePos_DataSection+
889 FileSize_DataSection;
890 FilePos_RSrcSection= FilePos_RWSection+
891 FileSize_RWSection;
892 FilePos_RelocSection= FilePos_RSrcSection+
893 FileSize_RSrcSection;
894 FilePos_DebugSection= FilePos_RelocSection+
895 FileSize_RelocSection;
896
897 //コードセッションのメモリ上のサイズ
898 if(FileSize_CodeSection%MEM_ALIGNMENT) MemSize_CodeSection=FileSize_CodeSection+(MEM_ALIGNMENT-FileSize_CodeSection%MEM_ALIGNMENT);
899 else MemSize_CodeSection=FileSize_CodeSection;
900
901 //エクスポートセクションのメモリ上のサイズ
902 if(FileSize_ExportSection%MEM_ALIGNMENT) MemSize_ExportSection=FileSize_ExportSection+(MEM_ALIGNMENT-FileSize_ExportSection%MEM_ALIGNMENT);
903 else MemSize_ExportSection=FileSize_ExportSection;
904
905 //インポートセクションのメモリ上のサイズ
906 if(FileSize_ImportSection%MEM_ALIGNMENT) MemSize_ImportSection=FileSize_ImportSection+(MEM_ALIGNMENT-FileSize_ImportSection%MEM_ALIGNMENT);
907 else MemSize_ImportSection=FileSize_ImportSection;
908
909 //データセクションのメモリ上のサイズ
910 if(FileSize_DataSection%MEM_ALIGNMENT) MemSize_DataSection=FileSize_DataSection+(MEM_ALIGNMENT-FileSize_DataSection%MEM_ALIGNMENT);
911 else MemSize_DataSection=FileSize_DataSection;
912
913 //リライタブルセクションのメモリ上のサイズ
914 extern int AllGlobalVarSize;
915 i=AllInitGlobalVarSize+AllGlobalVarSize;
916 if(i%MEM_ALIGNMENT) MemSize_RWSection=i+(MEM_ALIGNMENT-i%MEM_ALIGNMENT);
917 else MemSize_RWSection=i;
918
919 //リソースセクションのメモリ上のサイズ
920 if(FileSize_RSrcSection%MEM_ALIGNMENT) MemSize_RSrcSection=FileSize_RSrcSection+(MEM_ALIGNMENT-FileSize_RSrcSection%MEM_ALIGNMENT);
921 else MemSize_RSrcSection=FileSize_RSrcSection;
922
923 //リロケーションセクションのメモリ上のサイズ
924 if(FileSize_RelocSection%MEM_ALIGNMENT) MemSize_RelocSection=FileSize_RelocSection+(MEM_ALIGNMENT-FileSize_RelocSection%MEM_ALIGNMENT);
925 else MemSize_RelocSection=FileSize_RelocSection;
926
927 //デバッグセクションのメモリ上のサイズ
928 if(FileSize_DebugSection%MEM_ALIGNMENT) MemSize_DebugSection=FileSize_DebugSection+(MEM_ALIGNMENT-FileSize_DebugSection%MEM_ALIGNMENT);
929 else MemSize_DebugSection=FileSize_DebugSection;
930
931
932 //各セッションのメモリ上の位置
933 if(bUse_ExportSection) MemPos_ExportSection= 0x1000+
934 MemSize_CodeSection;
935 else MemPos_ExportSection=0;
936 if(bUse_ImportSection) MemPos_ImportSection= 0x1000+
937 MemSize_CodeSection+
938 MemSize_ExportSection;
939 else MemPos_ImportSection=0;
940 if(bUse_DataSection) MemPos_DataSection= 0x1000+
941 MemSize_CodeSection+
942 MemSize_ExportSection+
943 MemSize_ImportSection;
944 else MemPos_DataSection=0;
945 if(bUse_RWSection) MemPos_RWSection= 0x1000+
946 MemSize_CodeSection+
947 MemSize_ExportSection+
948 MemSize_ImportSection+
949 MemSize_DataSection;
950 else MemPos_RWSection=0;
951 if(bUse_RSrcSection) MemPos_RSrcSection= 0x1000+
952 MemSize_CodeSection+
953 MemSize_ExportSection+
954 MemSize_ImportSection+
955 MemSize_DataSection+
956 MemSize_RWSection;
957 else MemPos_RSrcSection=0;
958 if(bUse_RelocSection) MemPos_RelocSection= 0x1000+
959 MemSize_CodeSection+
960 MemSize_ExportSection+
961 MemSize_ImportSection+
962 MemSize_DataSection+
963 MemSize_RWSection+
964 MemSize_RSrcSection;
965 else MemPos_RelocSection=0;
966 if(bUse_DebugSection) MemPos_DebugSection= 0x1000+
967 MemSize_CodeSection+
968 MemSize_ExportSection+
969 MemSize_ImportSection+
970 MemSize_DataSection+
971 MemSize_RWSection+
972 MemSize_RSrcSection+
973 MemSize_RelocSection;
974 else MemPos_DebugSection=0;
975
976
977
978 /////////////////////////////
979 // GC用の変数スケジュール
980 /////////////////////////////
981 PerformedGcVarSchedule();
982
983
984 ////////////////////////////
985 // エクスポート情報の再配置
986 ////////////////////////////
987 if(bUse_ExportSection){
988 for(i=0;i<ExportNum;i++){
989 lpdwExportAddressTable[i]+=MemPos_CodeSection;
990 lpdwExportNamePointerTable[i]+=MemPos_ExportSection;
991 }
992
993 ImageExportDirectory.Name+= MemPos_ExportSection;
994 ImageExportDirectory.AddressOfFunctions+= MemPos_ExportSection;
995 ImageExportDirectory.AddressOfNames+= MemPos_ExportSection;
996 ImageExportDirectory.AddressOfNameOrdinals+= MemPos_ExportSection;
997 }
998
999
1000 ////////////////////////////
1001 // インポート情報の再配置
1002 ////////////////////////////
1003 for(i=0,i5=0;i<ImportDllNum;i++){
1004 //IMAGE_IMPORT_DESCRIPTOR
1005 pImportDescriptor[i].OriginalFirstThunk+=MemPos_ImportSection;
1006 pImportDescriptor[i].Name+=MemPos_ImportSection;
1007
1008 //インポート アドレス テーブル（ルックアップ テーブルを差し引く）
1009 pImportDescriptor[i].FirstThunk=pImportDescriptor[i].OriginalFirstThunk-
1010 LookupSize; //ルックアップテーブル
1011
1012 for(i2=0;i2<MAX_HASH;i2++){
1013 const DllProc *pDllProc=ppDeclareHash[i2];
1014 while(pDllProc){
1015 if( !pDllProc->IsUsing() ){
1016 pDllProc=pDllProc->pNextData;
1017 continue;
1018 }
1019
1020 if( pDllProc->GetDllFileName() == ppDllNames[i] ){
1021 //ルックアップ テーブル
1022 pLookupTable[i5++]+=MemPos_ImportSection;
1023 }
1024
1025 pDllProc=pDllProc->pNextData;
1026 }
1027 }
1028 i5++;
1029 }
1030
1031
1032 ////////////////////////////////////////
1033 //仮想関数データテーブルスケジュール
1034 pobj_DBClass->ActionVtblSchedule(ImageBase,MemPos_CodeSection);
1035
1036
1037 //////////////////////////////
1038 // データテーブルスケジュール
1039 for(i=0;i<pobj_DataTableSchedule->num;i++){
1040 *((long *)(OpBuffer+pobj_DataTableSchedule->pObpValues[i]))+=ImageBase+MemPos_DataSection;
1041 }
1042 delete pobj_DataTableSchedule;
1043
1044
1045 //////////////////////////////////
1046 // インポートテーブルスケジュール（インポート アドレス テーブルを指し示させる）
1047 // ※x86はRVAからのオフセット。x64はRPI（実行中アドレス）からのオフセット
1048 for(i=0;i<pobj_ImportAddrSchedule->num;i++){
1049 DllProc *pDllProc;
1050 pDllProc=pobj_ImportAddrSchedule->ppdi[i];
1051 *((long *)(OpBuffer+pobj_ImportAddrSchedule->pObpValues[i]))=
1052 MemPos_ImportSection+
1053 pDllProc->GetLookupAddress()
1054 -(MemPos_CodeSection+pobj_ImportAddrSchedule->pObpValues[i]+sizeof(long));
1055 }
1056 delete pobj_ImportAddrSchedule;
1057
1058
1059 ////////////////////////////////////
1060 // プロシージャポインタスケジュール
1061 for(i=0;i<pobj_SubAddrSchedule->num;i++){
1062 if(pobj_SubAddrSchedule->ppsi[i]->beginOpAddress==0
1063 &&pobj_SubAddrSchedule->ppsi[i]->endOpAddress==0){
1064 SetError(300,NULL,-1);
1065 }
1066
1067 if(pobj_SubAddrSchedule->pbCall[i]){
1068 *((long *)(OpBuffer+pobj_SubAddrSchedule->pObpValues[i]))=
1069 pobj_SubAddrSchedule->ppsi[i]->beginOpAddress-(pobj_SubAddrSchedule->pObpValues[i]+sizeof(long));
1070 }
1071 else{
1072 *((long *)(OpBuffer+pobj_SubAddrSchedule->pObpValues[i]))=
1073 pobj_SubAddrSchedule->ppsi[i]->beginOpAddress+ImageBase+MemPos_CodeSection;
1074 }
1075 }
1076 delete pobj_SubAddrSchedule;
1077
1078
1079 if(bDll){
1080 //DLLの場合はリロケーション情報を生成
1081 pobj_Reloc->ResetRelocBuffer();
1082 }
1083
1084 delete pobj_TempSchedule;
1085
1086
1087
1088 //////////////////////////////////////
1089 // グローバル変数アドレススケジュール
1090 //////////////////////////////////////
1091
1092 /*
1093 GlobalArea1 - 初期バッファあり
1094 GlobalArea2 - 初期バッファなし
1095 */
1096 for(i=0;i<pobj_GlobalVarSchedule->num;i++){
1097 if(*((long *)(OpBuffer+pobj_GlobalVarSchedule->pObpValues[i])) & 0x80000000){
1098 //初期バッファなし
1099 *((long *)(OpBuffer+pobj_GlobalVarSchedule->pObpValues[i]))=
1100 AllInitGlobalVarSize+
1101 (*((long *)(OpBuffer+pobj_GlobalVarSchedule->pObpValues[i])) & 0x7FFFFFFF)+
1102 ImageBase+MemPos_RWSection;
1103 }
1104 else{
1105 //初期バッファあり
1106 *((long *)(OpBuffer+pobj_GlobalVarSchedule->pObpValues[i]))+=
1107 ImageBase+MemPos_RWSection;
1108 }
1109 }
1110 delete pobj_GlobalVarSchedule;
1111
1112
1113 ////////////////////////////////
1114 // リソースアドレススケジュール
1115 extern DWORD *lpdwRSrcAddrSchedule;
1116 extern int RSrcAddrScheduleNum;
1117 for(i=0;i<RSrcAddrScheduleNum;i++){
1118 *(DWORD *)(RSrcSectionBuffer+lpdwRSrcAddrSchedule[i])+=MemPos_RSrcSection;
1119 }
1120 HeapDefaultFree(lpdwRSrcAddrSchedule);
1121
1122
1123 //Dosスタブ
1124 char *DosStubBuffer;
1125 int DosStubSize;
1126 extern char BasicSystemDir[MAX_PATH];
1127 sprintf(temporary,"%sSubOperation\\dosstub.pgm",BasicSystemDir);
1128 hFile=CreateFile(temporary,GENERIC_READ,0,NULL,OPEN_EXISTING,FILE_ATTRIBUTE_NORMAL,NULL);
1129 if(hFile==INVALID_HANDLE_VALUE){
1130 MessageBox(hOwnerEditor,"dosstub.pgmの読み込みに失敗","error",MB_OK);
1131 goto EndWriteOpcode;
1132 }
1133 DosStubSize=GetFileSize(hFile,NULL);
1134 DosStubBuffer=(char *)HeapAlloc(hHeap,0,DosStubSize);
1135 ReadFile(hFile,DosStubBuffer,DosStubSize,(DWORD *)&i2,NULL);
1136 CloseHandle(hFile);
1137
1138
1139 extern BOOL bError;
1140 if(bError) goto EndWriteOpcode;
1141
1142
1143 ////////////////////////////
1144 // EXEファイルのヘッダ情報
1145 ////////////////////////////
1146
1147 IMAGE_DOS_HEADER ImageDosHeader;
1148 ImageDosHeader.e_magic= 0x5A4D;
1149 ImageDosHeader.e_cblp= 0x0090;
1150 ImageDosHeader.e_cp= 0x0003;
1151 ImageDosHeader.e_crlc= 0;
1152 ImageDosHeader.e_cparhdr=4;
1153 ImageDosHeader.e_minalloc=0x0000;
1154 ImageDosHeader.e_maxalloc=0xFFFF;
1155 ImageDosHeader.e_ss= 0x0000;
1156 ImageDosHeader.e_sp= 0x00B8;
1157 ImageDosHeader.e_csum= 0x0000;
1158 ImageDosHeader.e_ip= 0x0000;
1159 ImageDosHeader.e_cs= 0x0000;
1160 ImageDosHeader.e_lfarlc=0x0040;
1161 ImageDosHeader.e_ovno= 0x0000;
1162 ImageDosHeader.e_res[0]=0;
1163 ImageDosHeader.e_res[1]=0;
1164 ImageDosHeader.e_res[2]=0;
1165 ImageDosHeader.e_res[3]=0;
1166 ImageDosHeader.e_oemid= 0x0000;
1167 ImageDosHeader.e_oeminfo=0x0000;
1168 ImageDosHeader.e_res2[0]=0;
1169 ImageDosHeader.e_res2[1]=0;
1170 ImageDosHeader.e_res2[2]=0;
1171 ImageDosHeader.e_res2[3]=0;
1172 ImageDosHeader.e_res2[4]=0;
1173 ImageDosHeader.e_res2[5]=0;
1174 ImageDosHeader.e_res2[6]=0;
1175 ImageDosHeader.e_res2[7]=0;
1176 ImageDosHeader.e_res2[8]=0;
1177 ImageDosHeader.e_res2[9]=0;
1178 ImageDosHeader.e_lfanew=0x0100; //PEヘッダの位置
1179
1180
1181 /////////////////////////////////////////////
1182 // PEヘッダ
1183 /////////////////////////////////////////////
1184
1185 IMAGE_NT_HEADERS64 ImagePeHdr;
1186 ImagePeHdr.Signature=IMAGE_NT_SIGNATURE;
1187
1188 //マシンタイプ
1189 ImagePeHdr.FileHeader.Machine= IMAGE_FILE_MACHINE_AMD64;
1190
1191 ImagePeHdr.FileHeader.NumberOfSections= bUse_CodeSection+
1192 bUse_ExportSection+
1193 bUse_ImportSection+
1194 bUse_DataSection+
1195 bUse_RWSection+
1196 bUse_RSrcSection+
1197 bUse_RelocSection+
1198 bUse_DebugSection; //セクション数
1199 ImagePeHdr.FileHeader.TimeDateStamp= (DWORD)time(NULL);
1200 ImagePeHdr.FileHeader.PointerToSymbolTable= 0x00000000;
1201 ImagePeHdr.FileHeader.NumberOfSymbols= 0x00000000;
1202 ImagePeHdr.FileHeader.SizeOfOptionalHeader= IMAGE_SIZEOF_NT_OPTIONAL64_HEADER;
1203 if(bDll){
1204 ImagePeHdr.FileHeader.Characteristics= IMAGE_FILE_EXECUTABLE_IMAGE|
1205 IMAGE_FILE_DLL|
1206 IMAGE_FILE_LARGE_ADDRESS_AWARE;
1207 }
1208 else{
1209 ImagePeHdr.FileHeader.Characteristics= IMAGE_FILE_RELOCS_STRIPPED|
1210 IMAGE_FILE_EXECUTABLE_IMAGE|
1211 IMAGE_FILE_LARGE_ADDRESS_AWARE;
1212 }
1213
1214 ImagePeHdr.OptionalHeader.Magic= IMAGE_NT_OPTIONAL_HDR64_MAGIC;
1215 ImagePeHdr.OptionalHeader.MajorLinkerVersion= 5;
1216 ImagePeHdr.OptionalHeader.MinorLinkerVersion= 0;
1217 ImagePeHdr.OptionalHeader.SizeOfCode= FileSize_CodeSection; //コードサイズ（.textのセッションサイズ）
1218 ImagePeHdr.OptionalHeader.SizeOfInitializedData=FileSize_DataSection; //データサイズ（.dataのセッションサイズ）
1219 ImagePeHdr.OptionalHeader.SizeOfUninitializedData=0; //未初期化データのサイズ（なし）
1220 if(bDll){
1221 if(DllMain_EntryPoint==-1)
1222 ImagePeHdr.OptionalHeader.AddressOfEntryPoint=0;
1223 else
1224 ImagePeHdr.OptionalHeader.AddressOfEntryPoint=MemPos_CodeSection+DllMain_EntryPoint;
1225 }
1226 else ImagePeHdr.OptionalHeader.AddressOfEntryPoint= MemPos_CodeSection;
1227 ImagePeHdr.OptionalHeader.BaseOfCode= MemPos_CodeSection; //.textのアドレス
1228
1229 ImagePeHdr.OptionalHeader.ImageBase= ImageBase; //イメージベース
1230 ImagePeHdr.OptionalHeader.SectionAlignment= MEM_ALIGNMENT; //セクションアラインメント
1231 ImagePeHdr.OptionalHeader.FileAlignment= FILE_ALIGNMENT;
1232 ImagePeHdr.OptionalHeader.MajorOperatingSystemVersion=4;
1233 ImagePeHdr.OptionalHeader.MinorOperatingSystemVersion=0;
1234 ImagePeHdr.OptionalHeader.MajorImageVersion= 0;
1235 ImagePeHdr.OptionalHeader.MinorImageVersion= 0;
1236 ImagePeHdr.OptionalHeader.MajorSubsystemVersion=4;
1237 ImagePeHdr.OptionalHeader.MinorSubsystemVersion=0;
1238 ImagePeHdr.OptionalHeader.Win32VersionValue= 0;
1239 ImagePeHdr.OptionalHeader.SizeOfImage= EXE_HEADER_SIZE+
1240 MemSize_CodeSection+
1241 MemSize_ExportSection+
1242 MemSize_ImportSection+
1243 MemSize_DataSection+
1244 MemSize_RWSection+
1245 MemSize_RSrcSection+
1246 MemSize_RelocSection+
1247 MemSize_DebugSection;//すべてのイメージサイズ
1248 ImagePeHdr.OptionalHeader.SizeOfHeaders= EXE_HEADER_SIZE;//ヘッダサイズ
1249 ImagePeHdr.OptionalHeader.CheckSum= 0;
1250 extern unsigned short TypeOfSubSystem;
1251 ImagePeHdr.OptionalHeader.Subsystem= TypeOfSubSystem;
1252 ImagePeHdr.OptionalHeader.DllCharacteristics= 0;
1253 ImagePeHdr.OptionalHeader.SizeOfStackReserve= 0x00100000;
1254 ImagePeHdr.OptionalHeader.SizeOfStackCommit= 0x00001000;
1255 ImagePeHdr.OptionalHeader.SizeOfHeapReserve= 0x00100000;
1256 ImagePeHdr.OptionalHeader.SizeOfHeapCommit= 0x00001000;
1257 ImagePeHdr.OptionalHeader.LoaderFlags= 0;
1258 ImagePeHdr.OptionalHeader.NumberOfRvaAndSizes= IMAGE_NUMBEROF_DIRECTORY_ENTRIES;
1259
1260 //データ ディクショナリ
1261 ImagePeHdr.OptionalHeader.DataDirectory[0].VirtualAddress=MemPos_ExportSection;
1262 ImagePeHdr.OptionalHeader.DataDirectory[0].Size=FileSize_ExportSection;
1263 ImagePeHdr.OptionalHeader.DataDirectory[1].VirtualAddress= //IMAGE_IMPORT_DESCRIPTORを指す
1264 MemPos_ImportSection+
1265 LookupSize+ //インポート アドレス テーブル
1266 LookupSize; //ルックアップ テーブル
1267 ImagePeHdr.OptionalHeader.DataDirectory[1].Size=(ImportDllNum+1)*sizeof(IMAGE_IMPORT_DESCRIPTOR);
1268 ImagePeHdr.OptionalHeader.DataDirectory[2].VirtualAddress=MemPos_RSrcSection;
1269 ImagePeHdr.OptionalHeader.DataDirectory[2].Size=RSrcSectionSize;
1270 ImagePeHdr.OptionalHeader.DataDirectory[3].VirtualAddress=0;
1271 ImagePeHdr.OptionalHeader.DataDirectory[3].Size=0;
1272 ImagePeHdr.OptionalHeader.DataDirectory[4].VirtualAddress=0;
1273 ImagePeHdr.OptionalHeader.DataDirectory[4].Size=0;
1274 ImagePeHdr.OptionalHeader.DataDirectory[5].VirtualAddress=MemPos_RelocSection;
1275 ImagePeHdr.OptionalHeader.DataDirectory[5].Size=pobj_Reloc->length;
1276 ImagePeHdr.OptionalHeader.DataDirectory[6].VirtualAddress=0;
1277 ImagePeHdr.OptionalHeader.DataDirectory[6].Size=0;
1278 ImagePeHdr.OptionalHeader.DataDirectory[7].VirtualAddress=0;
1279 ImagePeHdr.OptionalHeader.DataDirectory[7].Size=0;
1280 ImagePeHdr.OptionalHeader.DataDirectory[8].VirtualAddress=0;
1281 ImagePeHdr.OptionalHeader.DataDirectory[8].Size=0;
1282 ImagePeHdr.OptionalHeader.DataDirectory[9].VirtualAddress=0;
1283 ImagePeHdr.OptionalHeader.DataDirectory[9].Size=0;
1284 ImagePeHdr.OptionalHeader.DataDirectory[10].VirtualAddress=0;
1285 ImagePeHdr.OptionalHeader.DataDirectory[10].Size=0;
1286 ImagePeHdr.OptionalHeader.DataDirectory[11].VirtualAddress=0;
1287 ImagePeHdr.OptionalHeader.DataDirectory[11].Size=0;
1288 ImagePeHdr.OptionalHeader.DataDirectory[12].VirtualAddress=MemPos_ImportSection;//インポート アドレス テーブル
1289 ImagePeHdr.OptionalHeader.DataDirectory[12].Size=LookupSize;
1290 ImagePeHdr.OptionalHeader.DataDirectory[13].VirtualAddress=0;
1291 ImagePeHdr.OptionalHeader.DataDirectory[13].Size=0;
1292 ImagePeHdr.OptionalHeader.DataDirectory[14].VirtualAddress=0;
1293 ImagePeHdr.OptionalHeader.DataDirectory[14].Size=0;
1294 ImagePeHdr.OptionalHeader.DataDirectory[15].VirtualAddress=0;
1295 ImagePeHdr.OptionalHeader.DataDirectory[15].Size=0;
1296
1297
1298 //コードセクションヘッダ
1299 IMAGE_SECTION_HEADER CodeSectionHeader;
1300 memset((char *)CodeSectionHeader.Name,0,IMAGE_SIZEOF_SHORT_NAME);
1301 lstrcpy((char *)CodeSectionHeader.Name,".text");
1302 CodeSectionHeader.Misc.VirtualSize= obp;
1303 CodeSectionHeader.VirtualAddress= MemPos_CodeSection; //開始アドレス
1304 CodeSectionHeader.SizeOfRawData= FileSize_CodeSection;
1305 CodeSectionHeader.PointerToRawData= FilePos_CodeSection; //ファイル上の開始アドレス
1306 CodeSectionHeader.PointerToRelocations= 0;
1307 CodeSectionHeader.PointerToLinenumbers= 0;
1308 CodeSectionHeader.NumberOfRelocations= 0;
1309 CodeSectionHeader.NumberOfLinenumbers= 0;
1310 CodeSectionHeader.Characteristics= IMAGE_SCN_MEM_EXECUTE|
1311 IMAGE_SCN_MEM_READ|
1312 IMAGE_SCN_CNT_CODE;
1313
1314 //エクスポートセクションヘッダ
1315 IMAGE_SECTION_HEADER ExportSectionHeader;
1316 memset((char *)ExportSectionHeader.Name,0,IMAGE_SIZEOF_SHORT_NAME);
1317 lstrcpy((char *)ExportSectionHeader.Name,".edata");
1318 ExportSectionHeader.Misc.VirtualSize= MemSize_ExportSection;
1319 ExportSectionHeader.VirtualAddress= MemPos_ExportSection; //開始アドレス
1320 ExportSectionHeader.SizeOfRawData= FileSize_ExportSection; //サイズ
1321 ExportSectionHeader.PointerToRawData= FilePos_ExportSection; //ファイル上の開始アドレス
1322 ExportSectionHeader.PointerToRelocations= 0;
1323 ExportSectionHeader.PointerToLinenumbers= 0;
1324 ExportSectionHeader.NumberOfRelocations= 0;
1325 ExportSectionHeader.NumberOfLinenumbers= 0;
1326 ExportSectionHeader.Characteristics= IMAGE_SCN_CNT_INITIALIZED_DATA|
1327 IMAGE_SCN_MEM_READ;
1328
1329 //インポートセクションヘッダ
1330 IMAGE_SECTION_HEADER ImportSectionHeader;
1331 memset((char *)ImportSectionHeader.Name,0,IMAGE_SIZEOF_SHORT_NAME);
1332 lstrcpy((char *)ImportSectionHeader.Name,".idata");
1333 ImportSectionHeader.Misc.VirtualSize=
1334 LookupSize+ //インポートアドレステーブル
1335 LookupSize+ //ルックアップテーブル
1336 (ImportDllNum+1)*sizeof(IMAGE_IMPORT_DESCRIPTOR)+
1337 16*ImportDllNum+ //DLL名
1338 HintSize; //ヒント名（関数名）テーブル
1339 ImportSectionHeader.VirtualAddress= MemPos_ImportSection; //開始アドレス
1340 ImportSectionHeader.SizeOfRawData= FileSize_ImportSection; //サイズ
1341 ImportSectionHeader.PointerToRawData= FilePos_ImportSection; //ファイル上の開始アドレス
1342 ImportSectionHeader.PointerToRelocations= 0;
1343 ImportSectionHeader.PointerToLinenumbers= 0;
1344 ImportSectionHeader.NumberOfRelocations= 0;
1345 ImportSectionHeader.NumberOfLinenumbers= 0;
1346 ImportSectionHeader.Characteristics= IMAGE_SCN_CNT_INITIALIZED_DATA|
1347 IMAGE_SCN_MEM_READ;
1348
1349 //データセクションヘッダ
1350 IMAGE_SECTION_HEADER DataSectionHeader;
1351 memset((char *)DataSectionHeader.Name,0,IMAGE_SIZEOF_SHORT_NAME);
1352 lstrcpy((char *)DataSectionHeader.Name,".sdata");
1353 DataSectionHeader.Misc.VirtualSize= MemSize_DataSection;
1354 DataSectionHeader.VirtualAddress= MemPos_DataSection;
1355 DataSectionHeader.SizeOfRawData= FileSize_DataSection;
1356 DataSectionHeader.PointerToRawData= FilePos_DataSection;
1357 DataSectionHeader.PointerToRelocations= 0;
1358 DataSectionHeader.PointerToLinenumbers= 0;
1359 DataSectionHeader.NumberOfRelocations= 0;
1360 DataSectionHeader.NumberOfLinenumbers= 0;
1361 DataSectionHeader.Characteristics= IMAGE_SCN_CNT_INITIALIZED_DATA|
1362 IMAGE_SCN_MEM_READ|
1363 IMAGE_SCN_MEM_WRITE;
1364
1365 //リライタブルセクションヘッダ
1366 IMAGE_SECTION_HEADER RWSectionHeader;
1367 memset((char *)RWSectionHeader.Name,0,IMAGE_SIZEOF_SHORT_NAME);
1368 lstrcpy((char *)RWSectionHeader.Name,".data");
1369 RWSectionHeader.Misc.VirtualSize= AllInitGlobalVarSize+AllGlobalVarSize;
1370 RWSectionHeader.VirtualAddress= MemPos_RWSection;
1371 RWSectionHeader.SizeOfRawData= FileSize_RWSection;
1372 RWSectionHeader.PointerToRawData= FilePos_RWSection;
1373 RWSectionHeader.PointerToRelocations= 0;
1374 RWSectionHeader.PointerToLinenumbers= 0;
1375 RWSectionHeader.NumberOfRelocations= 0;
1376 RWSectionHeader.NumberOfLinenumbers= 0;
1377 RWSectionHeader.Characteristics= IMAGE_SCN_CNT_INITIALIZED_DATA|
1378 IMAGE_SCN_MEM_READ|
1379 IMAGE_SCN_MEM_WRITE;
1380
1381 //リソースセクションヘッダ
1382 IMAGE_SECTION_HEADER RSrcSectionHeader;
1383 memset((char *)RSrcSectionHeader.Name,0,IMAGE_SIZEOF_SHORT_NAME);
1384 lstrcpy((char *)RSrcSectionHeader.Name,".rsrc");
1385 RSrcSectionHeader.Misc.VirtualSize= RSrcSectionSize;
1386 RSrcSectionHeader.VirtualAddress= MemPos_RSrcSection;
1387 RSrcSectionHeader.SizeOfRawData= FileSize_RSrcSection;
1388 RSrcSectionHeader.PointerToRawData= FilePos_RSrcSection;
1389 RSrcSectionHeader.PointerToRelocations= 0;
1390 RSrcSectionHeader.PointerToLinenumbers= 0;
1391 RSrcSectionHeader.NumberOfRelocations= 0;
1392 RSrcSectionHeader.NumberOfLinenumbers= 0;
1393 RSrcSectionHeader.Characteristics= IMAGE_SCN_CNT_INITIALIZED_DATA|
1394 IMAGE_SCN_MEM_READ;
1395
1396 //リロケーションセクションヘッダ
1397 IMAGE_SECTION_HEADER RelocSectionHeader;
1398 memset((char *)RelocSectionHeader.Name,0,IMAGE_SIZEOF_SHORT_NAME);
1399 lstrcpy((char *)RelocSectionHeader.Name,".reloc");
1400 RelocSectionHeader.Misc.VirtualSize= pobj_Reloc->length;
1401 RelocSectionHeader.VirtualAddress= MemPos_RelocSection; //開始アドレス
1402 RelocSectionHeader.SizeOfRawData= FileSize_RelocSection; //サイズ
1403 RelocSectionHeader.PointerToRawData= FilePos_RelocSection; //ファイル上の開始アドレス
1404 RelocSectionHeader.PointerToRelocations= 0;
1405 RelocSectionHeader.PointerToLinenumbers= 0;
1406 RelocSectionHeader.NumberOfRelocations= 0;
1407 RelocSectionHeader.NumberOfLinenumbers= 0;
1408 RelocSectionHeader.Characteristics= IMAGE_SCN_CNT_INITIALIZED_DATA|
1409 IMAGE_SCN_MEM_DISCARDABLE|
1410 IMAGE_SCN_MEM_READ;
1411
1412 //デバッグセクションヘッダ
1413 IMAGE_SECTION_HEADER DebugSectionHeader;
1414 memset((char *)DebugSectionHeader.Name,0,IMAGE_SIZEOF_SHORT_NAME);
1415 lstrcpy((char *)DebugSectionHeader.Name,".debug");
1416 DebugSectionHeader.Misc.VirtualSize= pobj_DebugSection->length;
1417 DebugSectionHeader.VirtualAddress= MemPos_DebugSection; //開始アドレス
1418 DebugSectionHeader.SizeOfRawData= FileSize_DebugSection; //サイズ
1419 DebugSectionHeader.PointerToRawData= FilePos_DebugSection; //ファイル上の開始アドレス
1420 DebugSectionHeader.PointerToRelocations= 0;
1421 DebugSectionHeader.PointerToLinenumbers= 0;
1422 DebugSectionHeader.NumberOfRelocations= 0;
1423 DebugSectionHeader.NumberOfLinenumbers= 0;
1424 DebugSectionHeader.Characteristics= IMAGE_SCN_MEM_DISCARDABLE|
1425 IMAGE_SCN_MEM_READ;
1426
1427
1428 hFile=CreateFile(OutputFileName,GENERIC_WRITE,0,NULL,CREATE_ALWAYS,FILE_ATTRIBUTE_NORMAL,NULL);
1429 if(hFile==INVALID_HANDLE_VALUE){
1430 SetError(53,OutputFileName,-1);
1431 goto EndWriteOpcode;
1432 }
1433
1434 //ヘッダ
1435 WriteFile(hFile,(void *)&ImageDosHeader,sizeof(IMAGE_DOS_HEADER),(DWORD *)&i2,NULL);
1436 i=i2;
1437
1438 //Dosスタブ
1439 WriteFile(hFile,DosStubBuffer,DosStubSize,(DWORD *)&i2,NULL);
1440 i+=i2;
1441
1442 //0x0100までNULLを並べる
1443 temp2=(char *)HeapAlloc(hHeap,HEAP_ZERO_MEMORY,0x0100-i);
1444 WriteFile(hFile,temp2,0x0100-i,(DWORD *)&i2,NULL);
1445 HeapDefaultFree(temp2);
1446 i+=i2;
1447
1448 //PEヘッダ
1449 WriteFile(hFile,&ImagePeHdr,sizeof(IMAGE_NT_HEADERS64),(DWORD *)&i2,NULL);
1450 i+=i2;
1451
1452 //コード セクション ヘッダ
1453 WriteFile(hFile,&CodeSectionHeader,sizeof(IMAGE_SECTION_HEADER),(DWORD *)&i2,NULL);
1454 i+=i2;
1455
1456 if(bUse_ExportSection){
1457 //エクスポート セクション ヘッダ
1458 WriteFile(hFile,&ExportSectionHeader,sizeof(IMAGE_SECTION_HEADER),(DWORD *)&i2,NULL);
1459 i+=i2;
1460 }
1461 if(bUse_ImportSection){
1462 //インポート セクション ヘッダ
1463 WriteFile(hFile,&ImportSectionHeader,sizeof(IMAGE_SECTION_HEADER),(DWORD *)&i2,NULL);
1464 i+=i2;
1465 }
1466 if(bUse_DataSection){
1467 //データ セクション ヘッダ
1468 WriteFile(hFile,&DataSectionHeader,sizeof(IMAGE_SECTION_HEADER),(DWORD *)&i2,NULL);
1469 i+=i2;
1470 }
1471 if(bUse_RWSection){
1472 //リライタブルセクションヘッダ
1473 WriteFile(hFile,&RWSectionHeader,sizeof(IMAGE_SECTION_HEADER),(DWORD *)&i2,NULL);
1474 i+=i2;
1475 }
1476 if(bUse_RSrcSection){
1477 //リソースセクションヘッダ
1478 WriteFile(hFile,&RSrcSectionHeader,sizeof(IMAGE_SECTION_HEADER),(DWORD *)&i2,NULL);
1479 i+=i2;
1480 }
1481 if(bUse_RelocSection){
1482 //リロケーションセクションヘッダ
1483 WriteFile(hFile,&RelocSectionHeader,sizeof(IMAGE_SECTION_HEADER),(DWORD *)&i2,NULL);
1484 i+=i2;
1485 }
1486 if(bUse_DebugSection){
1487 //デバッグセクションヘッダ
1488 WriteFile(hFile,&DebugSectionHeader,sizeof(IMAGE_SECTION_HEADER),(DWORD *)&i2,NULL);
1489 i+=i2;
1490 }
1491
1492 //EXE_HEADER_SIZEまでNULLを並べる
1493 temp2=(char *)HeapAlloc(hHeap,HEAP_ZERO_MEMORY,EXE_HEADER_SIZE-i);
1494 WriteFile(hFile,temp2,EXE_HEADER_SIZE-i,(DWORD *)&i2,NULL);
1495 HeapDefaultFree(temp2);
1496 i+=i2;
1497
1498 //コード
1499 WriteFile(hFile,OpBuffer,obp,(DWORD *)&i2,NULL);
1500 i+=i2;
1501
1502 //FilePos_ExportSectionまでNULLを並べる
1503 temp2=(char *)HeapAlloc(hHeap,HEAP_ZERO_MEMORY,FilePos_ExportSection-i);
1504 WriteFile(hFile,temp2,FilePos_ExportSection-i,(DWORD *)&i2,NULL);
1505 HeapDefaultFree(temp2);
1506 i+=i2;
1507
1508 if(bUse_ExportSection){
1509 //エクスポート ディレクトリ テーブル
1510 WriteFile(hFile,&ImageExportDirectory,sizeof(IMAGE_EXPORT_DIRECTORY),(DWORD *)&i2,NULL);
1511 i+=i2;
1512
1513 //エクスポート アドレス テーブル
1514 WriteFile(hFile,lpdwExportAddressTable,ExportNum*sizeof(DWORD),(DWORD *)&i2,NULL);
1515 i+=i2;
1516
1517 //エクスポート名ポインタ テーブル
1518 WriteFile(hFile,lpdwExportNamePointerTable,ExportNum*sizeof(DWORD),(DWORD *)&i2,NULL);
1519 i+=i2;
1520
1521 //エクスポート序数テーブル
1522 WriteFile(hFile,lpwExportOrdinalTable,ExportNum*sizeof(WORD),(DWORD *)&i2,NULL);
1523 i+=i2;
1524
1525 //シンボル名
1526 WriteFile(hFile,lpExportNames,ExportNamesLength,(DWORD *)&i2,NULL);
1527 i+=i2;
1528 }
1529
1530 //FilePos_ImportSectionまでNULLを並べる
1531 temp2=(char *)HeapAlloc(hHeap,HEAP_ZERO_MEMORY,FilePos_ImportSection-i);
1532 WriteFile(hFile,temp2,FilePos_ImportSection-i,(DWORD *)&i2,NULL);
1533 HeapDefaultFree(temp2);
1534 i+=i2;
1535
1536 if(bUse_ImportSection){
1537 //インポート アドレス テーブル
1538 WriteFile(hFile,pLookupTable,LookupSize,(DWORD *)&i2,NULL);
1539 i+=i2;
1540
1541 //ルックアップ テーブル
1542 WriteFile(hFile,pLookupTable,LookupSize,(DWORD *)&i2,NULL);
1543 i+=i2;
1544
1545 //インポート ディレクトリ テーブル（Nullディレクトリ テーブルを含む）
1546 for(i3=0;i3<(ImportDllNum+1);i3++){
1547 WriteFile(hFile,&pImportDescriptor[i3],sizeof(IMAGE_IMPORT_DESCRIPTOR),(DWORD *)&i2,NULL);
1548 i+=i2;
1549 }
1550
1551 //DLL名
1552 for(i3=0;i3<ImportDllNum;i3++){
1553 WriteFile(hFile,ppDllNames[i3],16,(DWORD *)&i2,NULL);
1554 i+=i2;
1555 }
1556
1557 //ヒント テーブル
1558 WriteFile(hFile,pHintTable,HintSize,(DWORD *)&i2,NULL);
1559 i+=i2;
1560 }
1561
1562 //FilePos_DataSectionまでNULLを並べる
1563 temp2=(char *)HeapAlloc(hHeap,HEAP_ZERO_MEMORY,FilePos_DataSection-i);
1564 WriteFile(hFile,temp2,FilePos_DataSection-i,(DWORD *)&i2,NULL);
1565 HeapDefaultFree(temp2);
1566 i+=i2;
1567
1568 if(bUse_DataSection){
1569 //データ テーブル
1570 WriteFile(hFile,dataTable.GetPtr(),dataTable.GetSize(),(DWORD *)&i2,NULL);
1571 i+=i2;
1572 }
1573
1574 //FilePos_RWSectionまでNULLを並べる
1575 temp2=(char *)HeapAlloc(hHeap,HEAP_ZERO_MEMORY,FilePos_RWSection-i);
1576 WriteFile(hFile,temp2,FilePos_RWSection-i,(DWORD *)&i2,NULL);
1577 HeapDefaultFree(temp2);
1578 i+=i2;
1579
1580 if(bUse_RWSection){
1581 //リライタブル データ テーブル（グローバル変数の初期バッファ）
1582 initGlobalBuf=(BYTE *)HeapReAlloc(hHeap,
1583 HEAP_ZERO_MEMORY,
1584 initGlobalBuf,
1585 FileSize_RWSection);
1586 WriteFile(hFile,initGlobalBuf,FileSize_RWSection,(DWORD *)&i2,NULL);
1587 i+=i2;
1588 }
1589
1590 //FilePos_RSrcSectionまでNULLを並べる
1591 temp2=(char *)HeapAlloc(hHeap,HEAP_ZERO_MEMORY,FilePos_RSrcSection-i);
1592 WriteFile(hFile,temp2,FilePos_RSrcSection-i,(DWORD *)&i2,NULL);
1593 HeapDefaultFree(temp2);
1594 i+=i2;
1595
1596 if(bUse_RSrcSection){
1597 //リソースバッファ
1598 WriteFile(hFile,RSrcSectionBuffer,RSrcSectionSize,(DWORD *)&i2,NULL);
1599 i+=i2;
1600 }
1601
1602 //FilePos_RelocSectionまでNULLを並べる
1603 temp2=(char *)HeapAlloc(hHeap,HEAP_ZERO_MEMORY,FilePos_RelocSection-i);
1604 WriteFile(hFile,temp2,FilePos_RelocSection-i,(DWORD *)&i2,NULL);
1605 HeapDefaultFree(temp2);
1606 i+=i2;
1607
1608 if(bUse_RelocSection){
1609 //リロケーション情報
1610 WriteFile(hFile,pobj_Reloc->buffer,pobj_Reloc->length,(DWORD *)&i2,NULL);
1611 i+=i2;
1612 }
1613
1614 //ファイルアラインメントを考慮
1615 if(i%FILE_ALIGNMENT){
1616 temp2=(char *)HeapAlloc(hHeap,HEAP_ZERO_MEMORY,FILE_ALIGNMENT-i%FILE_ALIGNMENT);
1617 WriteFile(hFile,temp2,FILE_ALIGNMENT-i%FILE_ALIGNMENT,(DWORD *)&i2,NULL);
1618 HeapDefaultFree(temp2);
1619 i+=i2;
1620 }
1621
1622 if(bUse_DebugSection){
1623 //デバッグセクション
1624 WriteFile(hFile,pobj_DebugSection->buffer,pobj_DebugSection->length,(DWORD *)&i2,NULL);
1625 i+=i2;
1626 }
1627
1628 //ファイルアラインメントを考慮
1629 if(i%FILE_ALIGNMENT){
1630 temp2=(char *)HeapAlloc(hHeap,HEAP_ZERO_MEMORY,FILE_ALIGNMENT-i%FILE_ALIGNMENT);
1631 WriteFile(hFile,temp2,FILE_ALIGNMENT-i%FILE_ALIGNMENT,(DWORD *)&i2,NULL);
1632 HeapDefaultFree(temp2);
1633 i+=i2;
1634 }
1635
1636 //書き込み終了
1637 CloseHandle(hFile);
1638
1639
1640EndWriteOpcode:
1641
1642 //Dosスタブ用のメモリを解放
1643 HeapDefaultFree(DosStubBuffer);
1644
1645 //エクスポート テーブル情報を解放
1646 HeapDefaultFree(lpdwExportAddressTable);
1647 HeapDefaultFree(lpdwExportNamePointerTable);
1648 HeapDefaultFree(lpwExportOrdinalTable);
1649
1650 //インポートDLL情報を解放
1651 HeapDefaultFree(pImportDescriptor);
1652 for(i=0;i<ImportDllNum;i++)
1653 HeapDefaultFree(ppDllNames[i]);
1654 HeapDefaultFree(ppDllNames);
1655
1656 //ルックアップテーブルに関する情報を解放
1657 HeapDefaultFree(pLookupTable);
1658
1659 //ヒントテーブルに関する情報を解放
1660 HeapDefaultFree(pHintTable);
1661
1662 //グローバル変数の初期バッファを解放
1663 HeapDefaultFree(initGlobalBuf);
1664
1665 //コードバッファを解放
1666 HeapDefaultFree(OpBuffer);
1667 OpBuffer=0;
1668
1669 //リソースセクションバッファを解放
1670 HeapDefaultFree(RSrcSectionBuffer);
1671
1672 //デバッグセクションを開放
1673 delete pobj_DebugSection;
1674
1675 //リロケーション情報を解放
1676 delete pobj_Reloc;
1677
1678 //列挙体に関する情報の破棄
1679 CEnumParent::DestroyEnum();
1680
1681 //クラスに関するメモリを解放
1682 delete pobj_DBClass;
1683 pobj_DBClass=0;
1684}
Note: See TracBrowser for help on using the repository browser.