source: dev/trunk/abdev/BasicCompiler32/MakePeHdr.cpp@ 179

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