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

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