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

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

Linkerの骨格を作成した

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