source: dev/BasicCompiler64/MakePeHdr.cpp@ 91

Last change on this file since 91 was 91, checked in by dai_9181, 18 years ago

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