source: dev/BasicCompiler64/MakePeHdr.cpp@ 105

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

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