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

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

BasicSourceのシリアライズがうまくいっていない

File size: 58.4 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 // データテーブル
147 compiler.GetObjectModule().dataTable.Clear();
148 if(bDebugCompile){
149 compiler.GetObjectModule().dataTable.Add( (long)0x00000002 );
150 }
151
152
153 //////////////////////
154 // コード生成前の準備
155 //////////////////////
156
157 //重複エラー情報管理のメモリを確保(グローバル領域コンパイル用)
158 extern char **SynonymErrorWords;
159 extern int SynonymErrorNum;
160 SynonymErrorNum=0;
161 SynonymErrorWords=(char **)HeapAlloc(hHeap,0,1);
162
163 //列挙体に関する情報の初期化
164 CEnumParent::InitEnum();
165
166 //列挙体からクラスコードを生成
167 char *temp;
168 temp=CEnumParent::GenerateCodes();
169 AddSourceCode(temp);
170 HeapDefaultFree(temp);
171
172 //関数ポインタ情報を初期化
173 compiler.GetObjectModule().meta.GetProcPointers().clear();
174
175 // 名前空間情報を取得
176 NamespaceSupporter::CollectNamespaces(
177 compiler.GetObjectModule().source.GetBuffer(),
178 compiler.GetObjectModule().meta.GetNamespaces()
179 );
180
181 //クラス名を取得(詳細情報はGetAllClassInfoで取得)
182 // CollectProcedures関数の中で参照されるオブジェクト名を事前に取得する。
183 // ※オブジェクトの内容までは取得しない
184 compiler.GetObjectModule().meta.GetClasses().CollectClassesForNameOnly( compiler.GetObjectModule().source );
185
186 //TypeDef情報を初期化
187 compiler.GetObjectModule().meta.GetTypeDefs().Init();
188
189 //定数情報を取得
190 GetConstInfo();
191
192 // サブルーチン(ユーザー定義、DLL関数)の識別子、アドレスを取得
193 compiler.pCompilingClass = NULL;
194 CollectProcedures(
195 compiler.GetObjectModule().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=obp;
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 vector<ObjectModule *> masterObjectModules;
543 masterObjectModules.push_back( &compiler.GetObjectModule() );
544 compiler.linker.Link( masterObjectModules );
545
546 oldSourceLines = compiler.linker.GetNativeCode().GetSourceLines();
547
548
549/*
550 int t,t2;
551 t=GetTickCount();
552 if( !compiler.GetObjectModule().WriteText( Jenga::Common::Environment::GetAppDir() + "\\objectModule_test.xml" ) )
553 {
554 MessageBox(0,"XML書き込みに失敗","test",0);
555 }
556 if( !compiler.GetObjectModule().ReadText( Jenga::Common::Environment::GetAppDir() + "\\objectModule_test.xml" ) )
557 {
558 MessageBox(0,"XML読み込みに失敗","test",0);
559 }
560 t2=GetTickCount();
561 t2-=t;
562 char s[100];
563 sprintf(s,"%d",t2);
564 MessageBox(0,s,"test",0);
565
566 if( !compiler.GetObjectModule().meta.WriteBinary( Jenga::Common::Environment::GetAppDir() + "\\objectModule_test.dat" ) )
567 {
568 MessageBox(0,"バイナリ書き込みに失敗","test",0);
569 }
570 Meta *pTempMeta = new Meta();
571 if( !pTempMeta->ReadBinary( Jenga::Common::Environment::GetAppDir() + "\\objectModule_test.dat" ) )
572 {
573 MessageBox(0,"バイナリ読み込みに失敗","test",0);
574 }
575 if( !compiler.GetObjectModule().meta.WriteText( Jenga::Common::Environment::GetAppDir() + "\\objectModule_test.dat" ) )
576 {
577 MessageBox(0,"バイナリ書き込みに失敗","test",0);
578 }
579 Meta *pTempMeta = new Meta();
580 if( !pTempMeta->ReadText( Jenga::Common::Environment::GetAppDir() + "\\objectModule_test.dat" ) )
581 {
582 MessageBox(0,"バイナリ読み込みに失敗","test",0);
583 }*/
584
585
586
587 ////////////////////////////////
588 // ここで一旦ログを取る
589 ////////////////////////////////
590 Diagnose();
591
592
593
594 ////////////////////////////////
595 // 使用するDLL関数のチェック
596 ////////////////////////////////
597 compiler.GetObjectModule().meta.GetDllProcs().Iterator_Reset();
598 while( compiler.GetObjectModule().meta.GetDllProcs().Iterator_HasNext() )
599 {
600 const DllProc *pDllProc = compiler.GetObjectModule().meta.GetDllProcs().Iterator_GetNext();
601
602 if( !pDllProc->IsUsing() ){
603 pDllProc=pDllProc->pNextData;
604 continue;
605 }
606
607 //エラーチェック
608 HINSTANCE hLib;
609 hLib=LoadLibrary( pDllProc->GetDllFileName().c_str() );
610 if(!hLib){
611 extern char OutputFileName[MAX_PATH];
612 char temp2[MAX_PATH],temp3[MAX_PATH];
613 _splitpath(OutputFileName,temp2,temp3,NULL,NULL);
614 lstrcat(temp2,temp3);
615 lstrcpy(temp3,pDllProc->GetDllFileName().c_str());
616 GetFullPath(temp3,temp2);
617 hLib=LoadLibrary(temp3);
618
619 if(!hLib){
620 SetError(-106,pDllProc->GetDllFileName().c_str(),pDllProc->GetCodePos());
621 }
622 }
623
624 if(hLib){
625 if(!GetProcAddress(hLib,pDllProc->GetAlias().c_str())){
626 FreeLibrary(hLib);
627 SetError(-107,pDllProc->GetAlias(),pDllProc->GetCodePos());
628 }
629 FreeLibrary(hLib);
630 }
631 }
632
633
634
635 /////////////////////////////
636 // リソースデータを読み込む
637 /////////////////////////////
638 extern char ResourceFileName[MAX_PATH];
639 GetResourceData(ResourceFileName);
640
641
642 //////////////////////////////
643 // エクスポート情報(DLLのみ)
644 //////////////////////////////
645 IMAGE_EXPORT_DIRECTORY ImageExportDirectory;
646 DWORD *lpdwExportAddressTable;
647 DWORD *lpdwExportNamePointerTable;
648 WORD *lpwExportOrdinalTable;
649 char lpExportNames[8192];
650 int ExportNamesLength;
651 int ExportNum;
652 int DllMain_EntryPoint;
653
654 DllMain_EntryPoint=-1;
655 ExportNum=0;
656 ExportNamesLength=0;
657 lpdwExportAddressTable=(DWORD *)HeapAlloc(hHeap,0,1);
658 lpdwExportNamePointerTable=(DWORD *)HeapAlloc(hHeap,0,1);
659 lpwExportOrdinalTable=(WORD *)HeapAlloc(hHeap,0,1);
660
661 if(bUse_ExportSection){
662 _splitpath(OutputFileName,NULL,NULL,lpExportNames,temporary);
663 lstrcat(lpExportNames,temporary);
664 ExportNamesLength=lstrlen(lpExportNames)+1;
665
666 UserProc *pUserProc,*psi2;
667 while(1){
668 //辞書順にサーチ
669 temporary[0]=0;
670 compiler.GetObjectModule().meta.GetUserProcs().Iterator_Reset();
671 while( compiler.GetObjectModule().meta.GetUserProcs().Iterator_HasNext() )
672 {
673 pUserProc = compiler.GetObjectModule().meta.GetUserProcs().Iterator_GetNext();
674 if(pUserProc->IsExport()){
675 if(temporary[0]=='\0'){
676 lstrcpy(temporary,pUserProc->GetName().c_str());
677 psi2=pUserProc;
678 }
679 else{
680 i3=lstrlen(temporary);
681 i4=(int)pUserProc->GetName().size();
682 if(i3<i4) i3=i4;
683 if(memcmp(temporary,pUserProc->GetName().c_str(),i3)>0){
684 lstrcpy(temporary,pUserProc->GetName().c_str());
685 psi2=pUserProc;
686 }
687 }
688 }
689 }
690 if(psi2==0) break;
691 pUserProc=psi2;
692
693 pUserProc->ExportOff();
694
695 if( pUserProc->GetName() == "DllMain" ){
696 // TODO: 未完成
697 DllMain_EntryPoint=pUserProc->_beginOpAddressOld;
698 }
699
700 lpdwExportAddressTable=(DWORD *)HeapReAlloc(hHeap,0,lpdwExportAddressTable,(ExportNum+1)*sizeof(DWORD));
701 lpdwExportAddressTable[ExportNum]=pUserProc->_beginOpAddressOld;
702
703 lpdwExportNamePointerTable=(DWORD *)HeapReAlloc(hHeap,0,lpdwExportNamePointerTable,(ExportNum+1)*sizeof(DWORD));
704 lpdwExportNamePointerTable[ExportNum]=ExportNamesLength;
705
706 lpwExportOrdinalTable=(WORD *)HeapReAlloc(hHeap,0,lpwExportOrdinalTable,(ExportNum+1)*sizeof(WORD));
707 lpwExportOrdinalTable[ExportNum]=ExportNum;
708
709 lstrcpy(lpExportNames+ExportNamesLength,pUserProc->GetName().c_str());
710 ExportNamesLength+=lstrlen(lpExportNames+ExportNamesLength)+1;
711
712 ExportNum++;
713 }
714 for(i=0;i<ExportNum;i++){
715 lpdwExportNamePointerTable[i]+=
716 sizeof(IMAGE_EXPORT_DIRECTORY)+ //エクスポートディレクトリテーブルを飛び越す
717 ExportNum*sizeof(DWORD)+ //エクスポート アドレス テーブルを飛び越す
718 ExportNum*sizeof(DWORD)+ //エクスポート名ポインタ テーブルを飛び越す
719 ExportNum*sizeof(WORD); //エクスポート序数テーブルを飛び越す
720 }
721
722 ImageExportDirectory.Characteristics=0;
723 ImageExportDirectory.TimeDateStamp=(DWORD)time(NULL);
724 ImageExportDirectory.MajorVersion=0;
725 ImageExportDirectory.MinorVersion=0;
726 ImageExportDirectory.Name=sizeof(IMAGE_EXPORT_DIRECTORY)+
727 ExportNum*sizeof(DWORD)+ //エクスポート アドレス テーブルを飛び越す
728 ExportNum*sizeof(DWORD)+ //エクスポート名ポインタ テーブルを飛び越す
729 ExportNum*sizeof(WORD); //エクスポート序数テーブルを飛び越す
730 ImageExportDirectory.Base=1;
731 ImageExportDirectory.NumberOfFunctions=ExportNum;
732 ImageExportDirectory.NumberOfNames=ExportNum;
733 ImageExportDirectory.AddressOfFunctions=sizeof(IMAGE_EXPORT_DIRECTORY);
734 ImageExportDirectory.AddressOfNames=ImageExportDirectory.AddressOfFunctions+ExportNum*sizeof(DWORD);
735 ImageExportDirectory.AddressOfNameOrdinals=ImageExportDirectory.AddressOfNames+ExportNum*sizeof(DWORD);
736 }
737
738
739 /////////////////////
740 //インポートDLL情報
741 /////////////////////
742
743 /*
744 IMAGE_IMPORT_DESCRIPTOR1[size=0x14] インポートヘッダ
745 IMAGE_IMPORT_DESCRIPTOR2[size=0x14]
746 IMAGE_IMPORT_DESCRIPTOR3[size=0x14]
747 ...
748 Dll名1[size=0x10]
749 Dll名2[size=0x10]
750 Dll名3[size=0x10]
751 ...
752 ルックアップ テーブル(ヒントを示すRVA)
753 ヒントテーブル
754 インポート アドレス テーブル(ルックアップと同じ内容だが、プログラム実行時に実行アドレスに書き換えられる)
755 */
756
757 char **ppDllNames=(char **)HeapAlloc(hHeap,0,1);
758
759 int ImportDllNum=0;
760
761 compiler.GetObjectModule().meta.GetDllProcs().Iterator_Reset();
762 while( compiler.GetObjectModule().meta.GetDllProcs().Iterator_HasNext() )
763 {
764 const DllProc *pDllProc = compiler.GetObjectModule().meta.GetDllProcs().Iterator_GetNext();
765
766 if( !pDllProc->IsUsing() ){
767 pDllProc=pDllProc->pNextData;
768 continue;
769 }
770
771 if( pDllProc->GetDllFileName().size() > 16 ){
772 SetError(7,NULL,cp);
773 break;
774 }
775 for(i2=0;i2<ImportDllNum;i2++){
776 if( pDllProc->GetDllFileName() == ppDllNames[i2] ){
777 break;
778 }
779 }
780 if(i2==ImportDllNum){
781 ppDllNames=(char **)HeapReAlloc(hHeap,0,ppDllNames,(ImportDllNum+1)*sizeof(char **));
782 ppDllNames[ImportDllNum]=(char *)HeapAlloc(hHeap,HEAP_ZERO_MEMORY,16);
783 lstrcpy(ppDllNames[ImportDllNum],pDllProc->GetDllFileName().c_str());
784 ImportDllNum++;
785 }
786 }
787
788 //インポート テーブル、及びルックアップ テーブル サイズの計算
789 IMAGE_IMPORT_DESCRIPTOR *pImportTable;
790 int LookupSize;
791 LookupSize=0;
792 pImportTable=(IMAGE_IMPORT_DESCRIPTOR *)HeapAlloc(hHeap,0,(ImportDllNum+1)*sizeof(IMAGE_IMPORT_DESCRIPTOR));
793 i3=(ImportDllNum+1)*sizeof(IMAGE_IMPORT_DESCRIPTOR);
794 for(i=0;i<ImportDllNum;i++){
795 //インポート テーブル(IMAGE_IMPORT_DESCRIPTOR)
796 pImportTable[i].OriginalFirstThunk=i3+(ImportDllNum*0x10)+LookupSize;
797 pImportTable[i].TimeDateStamp=0;
798 pImportTable[i].ForwarderChain=0;
799 pImportTable[i].Name=i3+i*0x10;
800
801 compiler.GetObjectModule().meta.GetDllProcs().Iterator_Reset();
802 while( compiler.GetObjectModule().meta.GetDllProcs().Iterator_HasNext() )
803 {
804 const DllProc *pDllProc = compiler.GetObjectModule().meta.GetDllProcs().Iterator_GetNext();
805
806 if( !pDllProc->IsUsing() ){
807 pDllProc=pDllProc->pNextData;
808 continue;
809 }
810
811 if( pDllProc->GetDllFileName() == ppDllNames[i] ){
812 //ルックアップデータのサイズを追加
813 LookupSize+=sizeof(DWORD);
814 }
815 }
816 LookupSize+=sizeof(DWORD); //NULL用
817 }
818 memset(&pImportTable[i],0,sizeof(IMAGE_IMPORT_DESCRIPTOR));
819
820 //ルックアップ テーブル、ヒント テーブル
821 DWORD *pLookupTable;
822 pLookupTable=(DWORD *)HeapAlloc(hHeap,0,LookupSize*sizeof(DWORD)+1);
823 char *pHintTable;
824 int HintSize,HintAllocSize;
825 HintSize=0;
826 HintAllocSize=128*2;
827 pHintTable=(char *)HeapAlloc(hHeap,0,HintAllocSize);
828 i3+=ImportDllNum*0x10;
829 for(i=0,i5=0;i<ImportDllNum;i++){
830 compiler.GetObjectModule().meta.GetDllProcs().Iterator_Reset();
831 while( compiler.GetObjectModule().meta.GetDllProcs().Iterator_HasNext() )
832 {
833 DllProc *pDllProc = compiler.GetObjectModule().meta.GetDllProcs().Iterator_GetNext();
834
835 if( !pDllProc->IsUsing() ){
836 pDllProc=pDllProc->pNextData;
837 continue;
838 }
839
840 if( pDllProc->GetDllFileName() == ppDllNames[i] ){
841 //ルックアップの位置をセット(後にインポート アドレス テーブルにセットしなおされる)
842 pDllProc->SetLookupAddress( i3+(i5*sizeof(DWORD)) );
843
844 //ルックアップ テーブル
845 pLookupTable[i5++]=i3+LookupSize+HintSize;
846
847 //ヒント テーブル
848 pHintTable[HintSize++]=0;
849 pHintTable[HintSize++]=0;
850 lstrcpy(pHintTable+HintSize,pDllProc->GetAlias().c_str());
851 i4=(int)pDllProc->GetAlias().size();
852 HintSize+=i4+1;
853 if(i4%2==0) pHintTable[HintSize++]=0;
854
855 if(HintAllocSize<HintSize+128){
856 HintAllocSize+=128;
857 pHintTable=(char *)HeapReAlloc(hHeap,0,pHintTable,HintAllocSize);
858 }
859 }
860 }
861 pLookupTable[i5++]=0;
862 }
863
864
865 if( compiler.IsDll() ){
866 //DLLの場合はリロケーション情報を仮生成
867 //※正式な生成は各セクションのメモリ上のサイズが決定してから再度行う。
868 pobj_Reloc->ResetRelocBuffer();
869 }
870
871
872
873 //グローバル変数情報を扱う構造体も初期バッファの有無による配置を行う
874 //(デバッグ情報で利用される)
875 extern int AllInitGlobalVarSize;
876 BOOST_FOREACH( Variable *pVar, compiler.GetObjectModule().meta.GetGlobalVars() ){
877 if(pVar->GetOffsetAddress()&0x80000000){
878 pVar->SetOffsetAddress( (pVar->GetOffsetAddress()&0x7FFFFFFF)+AllInitGlobalVarSize );
879 }
880 }
881
882
883
884 ////////////////////////////////////
885 // デバッグセクションを生成
886 ////////////////////////////////////
887
888 //デバッグセクションを生成
889 CDebugSection *pobj_DebugSection;
890 pobj_DebugSection=new CDebugSection();
891 extern BOOL bDebugCompile;
892 extern BOOL bError;
893 if(bDebugCompile&&bError==0){
894 pobj_DebugSection->make();
895 }
896
897
898
899 /////////////////////////////////////
900 // 各セクションの位置とサイズを計算
901 /////////////////////////////////////
902
903 //コードセッションのファイル上のサイズ
904 if(obp%FILE_ALIGNMENT) FileSize_CodeSection=obp+(FILE_ALIGNMENT-obp%FILE_ALIGNMENT);
905 else FileSize_CodeSection=obp;
906 if(FileSize_CodeSection) bUse_CodeSection=1;
907 else bUse_CodeSection=0;
908
909 //エクスポートセクションのファイル上のサイズ
910 i= sizeof(IMAGE_EXPORT_DIRECTORY)+ //エクスポートディレクトリテーブルを飛び越す
911 ExportNum*sizeof(DWORD)+ //エクスポート アドレス テーブルを飛び越す
912 ExportNum*sizeof(DWORD)+ //エクスポート名ポインタ テーブルを飛び越す
913 ExportNum*sizeof(WORD)+ //エクスポート序数テーブルを飛び越す
914 ExportNamesLength; //シンボル名バッファ
915 if(bUse_ExportSection){
916 if(i%FILE_ALIGNMENT) FileSize_ExportSection=i+(FILE_ALIGNMENT-i%FILE_ALIGNMENT);
917 else FileSize_ExportSection=i;
918 }
919 else FileSize_ExportSection=0;
920
921 //インポートセクションのファイル上のサイズ
922 i=(ImportDllNum+1)*sizeof(IMAGE_IMPORT_DESCRIPTOR)+
923 16*ImportDllNum+ //DLL名
924 LookupSize+ //ルックアップテーブル
925 HintSize+ //ヒント名(関数名)テーブル
926 LookupSize; //インポート アドレス テーブル
927 if(i%FILE_ALIGNMENT) FileSize_ImportSection=i+(FILE_ALIGNMENT-i%FILE_ALIGNMENT);
928 else FileSize_ImportSection=i;
929 bUse_ImportSection=1; //インポートセクションは必ず存在する
930
931 //データセクションのファイル上のサイズ
932 if(compiler.GetObjectModule().dataTable.GetSize()%FILE_ALIGNMENT) FileSize_DataSection=compiler.GetObjectModule().dataTable.GetSize()+(FILE_ALIGNMENT-compiler.GetObjectModule().dataTable.GetSize()%FILE_ALIGNMENT);
933 else FileSize_DataSection=compiler.GetObjectModule().dataTable.GetSize();
934 if(FileSize_DataSection) bUse_DataSection=1;
935 else bUse_DataSection=0;
936
937 //リライタブルセクションのファイル上のサイズ(グローバル変数の初期情報のみを格納)
938 if(AllInitGlobalVarSize%FILE_ALIGNMENT) FileSize_RWSection=AllInitGlobalVarSize+(FILE_ALIGNMENT-AllInitGlobalVarSize%FILE_ALIGNMENT);
939 else{
940 if(AllInitGlobalVarSize) FileSize_RWSection=AllInitGlobalVarSize;
941 else FileSize_RWSection=FILE_ALIGNMENT;
942 }
943 bUse_RWSection=1;
944
945 //リソースセクションのファイル上のサイズ
946 char *RSrcSectionBuffer;
947 int RSrcSectionSize;
948 RSrcSectionBuffer=GetRSrcSectionBuffer(&RSrcSectionSize);
949 if(RSrcSectionSize%FILE_ALIGNMENT) FileSize_RSrcSection=RSrcSectionSize+(FILE_ALIGNMENT-RSrcSectionSize%FILE_ALIGNMENT);
950 else FileSize_RSrcSection=RSrcSectionSize;
951 if(FileSize_RSrcSection) bUse_RSrcSection=1;
952 else bUse_RSrcSection=0;
953
954 //リロケーションセクションのファイル上のサイズ
955 if(pobj_Reloc->length%FILE_ALIGNMENT) FileSize_RelocSection=pobj_Reloc->length+(FILE_ALIGNMENT-pobj_Reloc->length%FILE_ALIGNMENT);
956 else FileSize_RelocSection=pobj_Reloc->length;
957 if(FileSize_RelocSection) bUse_RelocSection=1;
958 else bUse_RelocSection=0;
959
960 //デバッグセクションのファイル上のサイズ
961 if(pobj_DebugSection->length%FILE_ALIGNMENT) FileSize_DebugSection=pobj_DebugSection->length+(FILE_ALIGNMENT-pobj_DebugSection->length%FILE_ALIGNMENT);
962 else FileSize_DebugSection=pobj_DebugSection->length;
963 if(FileSize_DebugSection) bUse_DebugSection=1;
964 else bUse_DebugSection=0;
965
966
967 //各セッションのファイル上の位置
968 FilePos_CodeSection= EXE_HEADER_SIZE;
969 FilePos_ExportSection= FilePos_CodeSection+
970 FileSize_CodeSection;
971 FilePos_ImportSection= FilePos_ExportSection+
972 FileSize_ExportSection;
973 FilePos_DataSection= FilePos_ImportSection+
974 FileSize_ImportSection;
975 FilePos_RWSection= FilePos_DataSection+
976 FileSize_DataSection;
977 FilePos_RSrcSection= FilePos_RWSection+
978 FileSize_RWSection;
979 FilePos_RelocSection= FilePos_RSrcSection+
980 FileSize_RSrcSection;
981 FilePos_DebugSection= FilePos_RelocSection+
982 FileSize_RelocSection;
983
984
985 //コードセッションのメモリ上のサイズ
986 if(FileSize_CodeSection%MEM_ALIGNMENT) MemSize_CodeSection=FileSize_CodeSection+(MEM_ALIGNMENT-FileSize_CodeSection%MEM_ALIGNMENT);
987 else MemSize_CodeSection=FileSize_CodeSection;
988
989 //エクスポートセクションのメモリ上のサイズ
990 if(FileSize_ExportSection%MEM_ALIGNMENT) MemSize_ExportSection=FileSize_ExportSection+(MEM_ALIGNMENT-FileSize_ExportSection%MEM_ALIGNMENT);
991 else MemSize_ExportSection=FileSize_ExportSection;
992
993 //インポートセクションのメモリ上のサイズ
994 if(FileSize_ImportSection%MEM_ALIGNMENT) MemSize_ImportSection=FileSize_ImportSection+(MEM_ALIGNMENT-FileSize_ImportSection%MEM_ALIGNMENT);
995 else MemSize_ImportSection=FileSize_ImportSection;
996
997 //データセクションのメモリ上のサイズ
998 if(FileSize_DataSection%MEM_ALIGNMENT) MemSize_DataSection=FileSize_DataSection+(MEM_ALIGNMENT-FileSize_DataSection%MEM_ALIGNMENT);
999 else MemSize_DataSection=FileSize_DataSection;
1000
1001 //リライタブルセクションのメモリ上のサイズ
1002 extern int AllGlobalVarSize;
1003 i=AllInitGlobalVarSize+AllGlobalVarSize;
1004 if(i%MEM_ALIGNMENT) MemSize_RWSection=i+(MEM_ALIGNMENT-i%MEM_ALIGNMENT);
1005 else MemSize_RWSection=i;
1006
1007 //リソースセクションのメモリ上のサイズ
1008 if(FileSize_RSrcSection%MEM_ALIGNMENT) MemSize_RSrcSection=FileSize_RSrcSection+(MEM_ALIGNMENT-FileSize_RSrcSection%MEM_ALIGNMENT);
1009 else MemSize_RSrcSection=FileSize_RSrcSection;
1010
1011 //リロケーションセクションのメモリ上のサイズ
1012 if(FileSize_RelocSection%MEM_ALIGNMENT) MemSize_RelocSection=FileSize_RelocSection+(MEM_ALIGNMENT-FileSize_RelocSection%MEM_ALIGNMENT);
1013 else MemSize_RelocSection=FileSize_RelocSection;
1014
1015 //デバッグセクションのメモリ上のサイズ
1016 if(FileSize_DebugSection%MEM_ALIGNMENT) MemSize_DebugSection=FileSize_DebugSection+(MEM_ALIGNMENT-FileSize_DebugSection%MEM_ALIGNMENT);
1017 else MemSize_DebugSection=FileSize_DebugSection;
1018
1019
1020 //各セッションのメモリ上の位置
1021 if(bUse_ExportSection) MemPos_ExportSection= 0x1000+
1022 MemSize_CodeSection;
1023 else MemPos_ExportSection=0;
1024 if(bUse_ImportSection) MemPos_ImportSection= 0x1000+
1025 MemSize_CodeSection+
1026 MemSize_ExportSection;
1027 else MemPos_ImportSection=0;
1028 if(bUse_DataSection) MemPos_DataSection= 0x1000+
1029 MemSize_CodeSection+
1030 MemSize_ExportSection+
1031 MemSize_ImportSection;
1032 else MemPos_DataSection=0;
1033 if(bUse_RWSection) MemPos_RWSection= 0x1000+
1034 MemSize_CodeSection+
1035 MemSize_ExportSection+
1036 MemSize_ImportSection+
1037 MemSize_DataSection;
1038 else MemPos_RWSection=0;
1039 if(bUse_RSrcSection) MemPos_RSrcSection= 0x1000+
1040 MemSize_CodeSection+
1041 MemSize_ExportSection+
1042 MemSize_ImportSection+
1043 MemSize_DataSection+
1044 MemSize_RWSection;
1045 else MemPos_RSrcSection=0;
1046 if(bUse_RelocSection) MemPos_RelocSection= 0x1000+
1047 MemSize_CodeSection+
1048 MemSize_ExportSection+
1049 MemSize_ImportSection+
1050 MemSize_DataSection+
1051 MemSize_RWSection+
1052 MemSize_RSrcSection;
1053 else MemPos_RelocSection=0;
1054 if(bUse_DebugSection) MemPos_DebugSection= 0x1000+
1055 MemSize_CodeSection+
1056 MemSize_ExportSection+
1057 MemSize_ImportSection+
1058 MemSize_DataSection+
1059 MemSize_RWSection+
1060 MemSize_RSrcSection+
1061 MemSize_RelocSection;
1062 else MemPos_DebugSection=0;
1063
1064
1065
1066 ////////////////////////////
1067 // エクスポート情報の再配置
1068 ////////////////////////////
1069 if(bUse_ExportSection){
1070 for(i=0;i<ExportNum;i++){
1071 lpdwExportAddressTable[i]+=MemPos_CodeSection;
1072 lpdwExportNamePointerTable[i]+=MemPos_ExportSection;
1073 }
1074
1075 ImageExportDirectory.Name+= MemPos_ExportSection;
1076 ImageExportDirectory.AddressOfFunctions+= MemPos_ExportSection;
1077 ImageExportDirectory.AddressOfNames+= MemPos_ExportSection;
1078 ImageExportDirectory.AddressOfNameOrdinals+= MemPos_ExportSection;
1079 }
1080
1081
1082 ////////////////////////////
1083 // インポート情報の再配置
1084 ////////////////////////////
1085 for(i=0,i5=0;i<ImportDllNum;i++){
1086 //インポート テーブル(IMAGE_IMPORT_DESCRIPTOR)
1087 pImportTable[i].OriginalFirstThunk+=MemPos_ImportSection;
1088 pImportTable[i].Name+=MemPos_ImportSection;
1089
1090 //インポート アドレス テーブル(ルックアップとヒントを飛び越す)
1091 pImportTable[i].FirstThunk=pImportTable[i].OriginalFirstThunk+
1092 LookupSize+ //ルックアップテーブル
1093 HintSize; //ヒント名(関数名)テーブル
1094
1095 compiler.GetObjectModule().meta.GetDllProcs().Iterator_Reset();
1096 while( compiler.GetObjectModule().meta.GetDllProcs().Iterator_HasNext() )
1097 {
1098 const DllProc *pDllProc = compiler.GetObjectModule().meta.GetDllProcs().Iterator_GetNext();
1099
1100 if( !pDllProc->IsUsing() ){
1101 pDllProc=pDllProc->pNextData;
1102 continue;
1103 }
1104
1105 if( pDllProc->GetDllFileName() == ppDllNames[i] ){
1106 //ルックアップ テーブル
1107 pLookupTable[i5++]+=MemPos_ImportSection;
1108 }
1109 }
1110 i5++;
1111 }
1112
1113
1114 ////////////////////////////////////////
1115 //仮想関数データテーブルスケジュール
1116 compiler.GetObjectModule().meta.GetClasses().ActionVtblSchedule(ImageBase,MemPos_CodeSection);
1117
1118
1119 //////////////////////////////
1120 // データテーブルスケジュール
1121 for(i=0;i<pobj_DataTableSchedule->num;i++){
1122 *((long *)(OpBuffer+pobj_DataTableSchedule->pObpValues[i]))+=ImageBase+MemPos_DataSection;
1123 }
1124 delete pobj_DataTableSchedule;
1125
1126
1127 //////////////////////////////////
1128 // インポートテーブルスケジュール(インポート アドレス テーブルを指し示させる)
1129 for(i=0;i<pobj_ImportAddrSchedule->num;i++){
1130 const DllProc *pDllProc=pobj_ImportAddrSchedule->ppdi[i];
1131 *((long *)(OpBuffer+pobj_ImportAddrSchedule->pObpValues[i]))=
1132 ImageBase+
1133 MemPos_ImportSection+
1134 LookupSize+ //ルックアップテーブル
1135 HintSize+ //ヒント名(関数名)テーブル
1136 pDllProc->GetLookupAddress();
1137 }
1138 delete pobj_ImportAddrSchedule;
1139
1140
1141 ////////////////////////////////////
1142 // プロシージャポインタスケジュール
1143 for(i=0;i<pobj_SubAddrSchedule->num;i++){
1144 if(pobj_SubAddrSchedule->ppsi[i]->_beginOpAddressOld==0
1145 &&pobj_SubAddrSchedule->ppsi[i]->_endOpAddressOld==0){
1146 SetError(300,NULL,-1);
1147 }
1148
1149 if(pobj_SubAddrSchedule->pbCall[i]){
1150 *((long *)(OpBuffer+pobj_SubAddrSchedule->pObpValues[i]))=
1151 pobj_SubAddrSchedule->ppsi[i]->_beginOpAddressOld-(pobj_SubAddrSchedule->pObpValues[i]+sizeof(long));
1152 }
1153 else{
1154 *((long *)(OpBuffer+pobj_SubAddrSchedule->pObpValues[i]))=
1155 pobj_SubAddrSchedule->ppsi[i]->_beginOpAddressOld+ImageBase+MemPos_CodeSection;
1156 }
1157 }
1158 delete pobj_SubAddrSchedule;
1159
1160
1161 if( compiler.IsDll() ){
1162 //DLLの場合はリロケーション情報を生成
1163 pobj_Reloc->ResetRelocBuffer();
1164 }
1165
1166
1167
1168 //////////////////////////////////////
1169 // グローバル変数アドレススケジュール
1170 //////////////////////////////////////
1171
1172 /*
1173 GlobalArea1 - 初期バッファあり
1174 GlobalArea2 - 初期バッファなし
1175 */
1176 for(i=0;i<pobj_GlobalVarSchedule->num;i++){
1177 if(*((long *)(OpBuffer+pobj_GlobalVarSchedule->pObpValues[i])) & 0x80000000){
1178 //初期バッファなし
1179 *((long *)(OpBuffer+pobj_GlobalVarSchedule->pObpValues[i]))=
1180 AllInitGlobalVarSize+
1181 (*((long *)(OpBuffer+pobj_GlobalVarSchedule->pObpValues[i])) & 0x7FFFFFFF)+
1182 ImageBase+MemPos_RWSection;
1183 }
1184 else{
1185 //初期バッファあり
1186 *((long *)(OpBuffer+pobj_GlobalVarSchedule->pObpValues[i]))+=
1187 ImageBase+MemPos_RWSection;
1188 }
1189 }
1190 delete pobj_GlobalVarSchedule;
1191
1192
1193 compiler.linker.SetImageBase( ImageBase );
1194 compiler.linker.ResolveDataTableSchedules( MemPos_DataSection );
1195 compiler.linker.ResolveDllProcSchedules( MemPos_CodeSection, MemPos_ImportSection, LookupSize, HintSize );
1196 compiler.linker.ResolveUserProcSchedules( MemPos_CodeSection );
1197 compiler.linker.ResolveGlobalVarSchedules( MemPos_RWSection );
1198
1199 if( obp == compiler.linker.GetNativeCode().GetSize() )
1200 {
1201 memcpy( OpBuffer, compiler.linker.GetNativeCode().GetCodeBuffer(), compiler.linker.GetNativeCode().GetSize() );
1202 }
1203 else
1204 {
1205 SetError();
1206 }
1207
1208
1209
1210 ////////////////////////////////
1211 // リソースアドレススケジュール
1212 extern DWORD *lpdwRSrcAddrSchedule;
1213 extern int RSrcAddrScheduleNum;
1214 for(i=0;i<RSrcAddrScheduleNum;i++){
1215 *(DWORD *)(RSrcSectionBuffer+lpdwRSrcAddrSchedule[i])+=MemPos_RSrcSection;
1216 }
1217 HeapDefaultFree(lpdwRSrcAddrSchedule);
1218
1219
1220 //Dosスタブ
1221 char *DosStubBuffer;
1222 int DosStubSize;
1223 hFile=CreateFile(
1224 ( Jenga::Common::Environment::GetAppDir() + "\\SubOperation\\dosstub.pgm" ).c_str(),
1225 GENERIC_READ,0,NULL,OPEN_EXISTING,FILE_ATTRIBUTE_NORMAL,NULL);
1226 if(hFile==INVALID_HANDLE_VALUE){
1227 MessageBox(hOwnerEditor,"dosstub.pgmの読み込みに失敗","error",MB_OK);
1228 goto EndWriteOpcode;
1229 }
1230 DosStubSize=GetFileSize(hFile,NULL);
1231 DosStubBuffer=(char *)HeapAlloc(hHeap,0,DosStubSize);
1232 ReadFile(hFile,DosStubBuffer,DosStubSize,(DWORD *)&i2,NULL);
1233 CloseHandle(hFile);
1234
1235
1236 extern BOOL bError;
1237 if(bError) goto EndWriteOpcode;
1238
1239
1240 ////////////////////////////
1241 // EXEファイルのヘッダ情報
1242 ////////////////////////////
1243
1244 IMAGE_DOS_HEADER ImageDosHeader;
1245 ImageDosHeader.e_magic= 0x5A4D;
1246 ImageDosHeader.e_cblp= 0x0090;
1247 ImageDosHeader.e_cp= 0x0003;
1248 ImageDosHeader.e_crlc= 0;
1249 ImageDosHeader.e_cparhdr=4;
1250 ImageDosHeader.e_minalloc=0x0000;
1251 ImageDosHeader.e_maxalloc=0xFFFF;
1252 ImageDosHeader.e_ss= 0x0000;
1253 ImageDosHeader.e_sp= 0x00B8;
1254 ImageDosHeader.e_csum= 0x0000;
1255 ImageDosHeader.e_ip= 0x0000;
1256 ImageDosHeader.e_cs= 0x0000;
1257 ImageDosHeader.e_lfarlc=0x0040;
1258 ImageDosHeader.e_ovno= 0x0000;
1259 ImageDosHeader.e_res[0]=0;
1260 ImageDosHeader.e_res[1]=0;
1261 ImageDosHeader.e_res[2]=0;
1262 ImageDosHeader.e_res[3]=0;
1263 ImageDosHeader.e_oemid= 0x0000;
1264 ImageDosHeader.e_oeminfo=0x0000;
1265 ImageDosHeader.e_res2[0]=0;
1266 ImageDosHeader.e_res2[1]=0;
1267 ImageDosHeader.e_res2[2]=0;
1268 ImageDosHeader.e_res2[3]=0;
1269 ImageDosHeader.e_res2[4]=0;
1270 ImageDosHeader.e_res2[5]=0;
1271 ImageDosHeader.e_res2[6]=0;
1272 ImageDosHeader.e_res2[7]=0;
1273 ImageDosHeader.e_res2[8]=0;
1274 ImageDosHeader.e_res2[9]=0;
1275 ImageDosHeader.e_lfanew=0x0100; //PEヘッダの位置
1276
1277
1278 /////////////////////////////////////////////
1279 // PEヘッダ
1280 /////////////////////////////////////////////
1281
1282 IMAGE_NT_HEADERS ImagePeHdr;
1283 ImagePeHdr.Signature=IMAGE_NT_SIGNATURE;
1284
1285 //マシンタイプ
1286 ImagePeHdr.FileHeader.Machine= IMAGE_FILE_MACHINE_I386;
1287
1288 ImagePeHdr.FileHeader.NumberOfSections= bUse_CodeSection+
1289 bUse_ExportSection+
1290 bUse_ImportSection+
1291 bUse_DataSection+
1292 bUse_RWSection+
1293 bUse_RSrcSection+
1294 bUse_RelocSection+
1295 bUse_DebugSection; //セクション数
1296 ImagePeHdr.FileHeader.TimeDateStamp= (DWORD)time(NULL);
1297 ImagePeHdr.FileHeader.PointerToSymbolTable= 0x00000000;
1298 ImagePeHdr.FileHeader.NumberOfSymbols= 0x00000000;
1299 ImagePeHdr.FileHeader.SizeOfOptionalHeader= IMAGE_SIZEOF_NT_OPTIONAL32_HEADER;
1300 if( compiler.IsDll() ){
1301 ImagePeHdr.FileHeader.Characteristics= IMAGE_FILE_EXECUTABLE_IMAGE|
1302 IMAGE_FILE_32BIT_MACHINE|
1303 IMAGE_FILE_LINE_NUMS_STRIPPED|
1304 IMAGE_FILE_LOCAL_SYMS_STRIPPED|
1305 IMAGE_FILE_DLL;
1306 }
1307 else{
1308 ImagePeHdr.FileHeader.Characteristics= IMAGE_FILE_EXECUTABLE_IMAGE|
1309 IMAGE_FILE_32BIT_MACHINE|
1310 IMAGE_FILE_LINE_NUMS_STRIPPED|
1311 IMAGE_FILE_LOCAL_SYMS_STRIPPED;
1312 }
1313
1314 ImagePeHdr.OptionalHeader.Magic= 0x010B;
1315 ImagePeHdr.OptionalHeader.MajorLinkerVersion= 4;
1316 ImagePeHdr.OptionalHeader.MinorLinkerVersion= 0;
1317 ImagePeHdr.OptionalHeader.SizeOfCode= FileSize_CodeSection; //コードサイズ(.textのセッションサイズ)
1318 ImagePeHdr.OptionalHeader.SizeOfInitializedData=FileSize_DataSection; //データサイズ(.dataのセッションサイズ)
1319 ImagePeHdr.OptionalHeader.SizeOfUninitializedData=0; //未初期化データのサイズ(なし)
1320 if( compiler.IsDll() ){
1321 if(DllMain_EntryPoint==-1)
1322 ImagePeHdr.OptionalHeader.AddressOfEntryPoint=0;
1323 else
1324 ImagePeHdr.OptionalHeader.AddressOfEntryPoint=MemPos_CodeSection+DllMain_EntryPoint;
1325 }
1326 else ImagePeHdr.OptionalHeader.AddressOfEntryPoint= MemPos_CodeSection;
1327 ImagePeHdr.OptionalHeader.BaseOfCode= MemPos_CodeSection; //.textのアドレス
1328 ImagePeHdr.OptionalHeader.BaseOfData= MemPos_DataSection; //.dataのアドレス
1329
1330 ImagePeHdr.OptionalHeader.ImageBase= ImageBase; //イメージベース
1331 ImagePeHdr.OptionalHeader.SectionAlignment= MEM_ALIGNMENT; //セクションアラインメント
1332 ImagePeHdr.OptionalHeader.FileAlignment= FILE_ALIGNMENT;
1333 ImagePeHdr.OptionalHeader.MajorOperatingSystemVersion=4;
1334 ImagePeHdr.OptionalHeader.MinorOperatingSystemVersion=0;
1335 ImagePeHdr.OptionalHeader.MajorImageVersion= 0;
1336 ImagePeHdr.OptionalHeader.MinorImageVersion= 0;
1337 ImagePeHdr.OptionalHeader.MajorSubsystemVersion=4;
1338 ImagePeHdr.OptionalHeader.MinorSubsystemVersion=0;
1339 ImagePeHdr.OptionalHeader.Win32VersionValue= 0;
1340 ImagePeHdr.OptionalHeader.SizeOfImage= EXE_HEADER_SIZE+
1341 MemSize_CodeSection+
1342 MemSize_ExportSection+
1343 MemSize_ImportSection+
1344 MemSize_DataSection+
1345 MemSize_RWSection+
1346 MemSize_RSrcSection+
1347 MemSize_RelocSection+
1348 MemSize_DebugSection;//すべてのイメージサイズ
1349 ImagePeHdr.OptionalHeader.SizeOfHeaders= EXE_HEADER_SIZE;//ヘッダサイズ
1350 ImagePeHdr.OptionalHeader.CheckSum= 0;
1351 extern unsigned short TypeOfSubSystem;
1352 ImagePeHdr.OptionalHeader.Subsystem= TypeOfSubSystem;
1353 ImagePeHdr.OptionalHeader.DllCharacteristics= 0;
1354 ImagePeHdr.OptionalHeader.SizeOfStackReserve= 0x00100000;
1355 ImagePeHdr.OptionalHeader.SizeOfStackCommit= 0x00001000;
1356 ImagePeHdr.OptionalHeader.SizeOfHeapReserve= 0x00100000;
1357 ImagePeHdr.OptionalHeader.SizeOfHeapCommit= 0x00001000;
1358 ImagePeHdr.OptionalHeader.LoaderFlags= 0;
1359 ImagePeHdr.OptionalHeader.NumberOfRvaAndSizes= IMAGE_NUMBEROF_DIRECTORY_ENTRIES;
1360
1361 //データ ディクショナリ
1362 ImagePeHdr.OptionalHeader.DataDirectory[0].VirtualAddress=MemPos_ExportSection;
1363 ImagePeHdr.OptionalHeader.DataDirectory[0].Size=FileSize_ExportSection;
1364 ImagePeHdr.OptionalHeader.DataDirectory[1].VirtualAddress=MemPos_ImportSection;//インポートテーブル
1365 ImagePeHdr.OptionalHeader.DataDirectory[1].Size=FileSize_ImportSection;
1366 ImagePeHdr.OptionalHeader.DataDirectory[2].VirtualAddress=MemPos_RSrcSection;
1367 ImagePeHdr.OptionalHeader.DataDirectory[2].Size=RSrcSectionSize;
1368 ImagePeHdr.OptionalHeader.DataDirectory[3].VirtualAddress=0;
1369 ImagePeHdr.OptionalHeader.DataDirectory[3].Size=0;
1370 ImagePeHdr.OptionalHeader.DataDirectory[4].VirtualAddress=0;
1371 ImagePeHdr.OptionalHeader.DataDirectory[4].Size=0;
1372 ImagePeHdr.OptionalHeader.DataDirectory[5].VirtualAddress=MemPos_RelocSection;
1373 ImagePeHdr.OptionalHeader.DataDirectory[5].Size=pobj_Reloc->length;
1374 ImagePeHdr.OptionalHeader.DataDirectory[6].VirtualAddress=0;
1375 ImagePeHdr.OptionalHeader.DataDirectory[6].Size=0;
1376 ImagePeHdr.OptionalHeader.DataDirectory[7].VirtualAddress=0;
1377 ImagePeHdr.OptionalHeader.DataDirectory[7].Size=0;
1378 ImagePeHdr.OptionalHeader.DataDirectory[8].VirtualAddress=0;
1379 ImagePeHdr.OptionalHeader.DataDirectory[8].Size=0;
1380 ImagePeHdr.OptionalHeader.DataDirectory[9].VirtualAddress=0;
1381 ImagePeHdr.OptionalHeader.DataDirectory[9].Size=0;
1382 ImagePeHdr.OptionalHeader.DataDirectory[10].VirtualAddress=0;
1383 ImagePeHdr.OptionalHeader.DataDirectory[10].Size=0;
1384 ImagePeHdr.OptionalHeader.DataDirectory[11].VirtualAddress=0;
1385 ImagePeHdr.OptionalHeader.DataDirectory[11].Size=0;
1386 ImagePeHdr.OptionalHeader.DataDirectory[12].VirtualAddress=MemPos_ImportSection+
1387 (ImportDllNum+1)*sizeof(IMAGE_IMPORT_DESCRIPTOR)+
1388 16*ImportDllNum+ //DLL名
1389 LookupSize+ //ルックアップテーブル
1390 HintSize; //ヒント名(関数名)テーブル
1391 ImagePeHdr.OptionalHeader.DataDirectory[12].Size=LookupSize;
1392 ImagePeHdr.OptionalHeader.DataDirectory[13].VirtualAddress=0;
1393 ImagePeHdr.OptionalHeader.DataDirectory[13].Size=0;
1394 ImagePeHdr.OptionalHeader.DataDirectory[14].VirtualAddress=0;
1395 ImagePeHdr.OptionalHeader.DataDirectory[14].Size=0;
1396 ImagePeHdr.OptionalHeader.DataDirectory[15].VirtualAddress=0;
1397 ImagePeHdr.OptionalHeader.DataDirectory[15].Size=0;
1398
1399
1400 //コードセクションヘッダ
1401 IMAGE_SECTION_HEADER CodeSectionHeader;
1402 memset((char *)CodeSectionHeader.Name,0,IMAGE_SIZEOF_SHORT_NAME);
1403 lstrcpy((char *)CodeSectionHeader.Name,".text");
1404 CodeSectionHeader.Misc.VirtualSize= MemSize_CodeSection;
1405 CodeSectionHeader.VirtualAddress= MemPos_CodeSection; //開始アドレス
1406 CodeSectionHeader.SizeOfRawData= FileSize_CodeSection;
1407 CodeSectionHeader.PointerToRawData= FilePos_CodeSection; //ファイル上の開始アドレス
1408 CodeSectionHeader.PointerToRelocations= 0;
1409 CodeSectionHeader.PointerToLinenumbers= 0;
1410 CodeSectionHeader.NumberOfRelocations= 0;
1411 CodeSectionHeader.NumberOfLinenumbers= 0;
1412 CodeSectionHeader.Characteristics= IMAGE_SCN_MEM_EXECUTE|
1413 IMAGE_SCN_MEM_READ|
1414 IMAGE_SCN_CNT_CODE;
1415
1416 //エクスポートセクションヘッダ
1417 IMAGE_SECTION_HEADER ExportSectionHeader;
1418 memset((char *)ExportSectionHeader.Name,0,IMAGE_SIZEOF_SHORT_NAME);
1419 lstrcpy((char *)ExportSectionHeader.Name,".edata");
1420 ExportSectionHeader.Misc.VirtualSize= MemSize_ExportSection;
1421 ExportSectionHeader.VirtualAddress= MemPos_ExportSection; //開始アドレス
1422 ExportSectionHeader.SizeOfRawData= FileSize_ExportSection; //サイズ
1423 ExportSectionHeader.PointerToRawData= FilePos_ExportSection; //ファイル上の開始アドレス
1424 ExportSectionHeader.PointerToRelocations= 0;
1425 ExportSectionHeader.PointerToLinenumbers= 0;
1426 ExportSectionHeader.NumberOfRelocations= 0;
1427 ExportSectionHeader.NumberOfLinenumbers= 0;
1428 ExportSectionHeader.Characteristics= IMAGE_SCN_CNT_INITIALIZED_DATA|
1429 IMAGE_SCN_MEM_READ;
1430
1431 //インポートセクションヘッダ
1432 IMAGE_SECTION_HEADER ImportSectionHeader;
1433 memset((char *)ImportSectionHeader.Name,0,IMAGE_SIZEOF_SHORT_NAME);
1434 lstrcpy((char *)ImportSectionHeader.Name,".idata");
1435 ImportSectionHeader.Misc.VirtualSize= MemSize_ImportSection;
1436 ImportSectionHeader.VirtualAddress= MemPos_ImportSection; //開始アドレス
1437 ImportSectionHeader.SizeOfRawData= FileSize_ImportSection; //サイズ
1438 ImportSectionHeader.PointerToRawData= FilePos_ImportSection; //ファイル上の開始アドレス
1439 ImportSectionHeader.PointerToRelocations= 0;
1440 ImportSectionHeader.PointerToLinenumbers= 0;
1441 ImportSectionHeader.NumberOfRelocations= 0;
1442 ImportSectionHeader.NumberOfLinenumbers= 0;
1443 ImportSectionHeader.Characteristics= IMAGE_SCN_CNT_INITIALIZED_DATA|
1444 IMAGE_SCN_MEM_READ;
1445
1446 //データセクションヘッダ
1447 IMAGE_SECTION_HEADER DataSectionHeader;
1448 memset((char *)DataSectionHeader.Name,0,IMAGE_SIZEOF_SHORT_NAME);
1449 lstrcpy((char *)DataSectionHeader.Name,".sdata");
1450 DataSectionHeader.Misc.VirtualSize= MemSize_DataSection;
1451 DataSectionHeader.VirtualAddress= MemPos_DataSection;
1452 DataSectionHeader.SizeOfRawData= FileSize_DataSection;
1453 DataSectionHeader.PointerToRawData= FilePos_DataSection;
1454 DataSectionHeader.PointerToRelocations= 0;
1455 DataSectionHeader.PointerToLinenumbers= 0;
1456 DataSectionHeader.NumberOfRelocations= 0;
1457 DataSectionHeader.NumberOfLinenumbers= 0;
1458 DataSectionHeader.Characteristics= IMAGE_SCN_CNT_INITIALIZED_DATA|
1459 IMAGE_SCN_MEM_READ|
1460 IMAGE_SCN_MEM_WRITE;
1461
1462 //リライタブルセクションヘッダ
1463 IMAGE_SECTION_HEADER RWSectionHeader;
1464 memset((char *)RWSectionHeader.Name,0,IMAGE_SIZEOF_SHORT_NAME);
1465 lstrcpy((char *)RWSectionHeader.Name,".data");
1466 RWSectionHeader.Misc.VirtualSize= AllInitGlobalVarSize+AllGlobalVarSize;
1467 RWSectionHeader.VirtualAddress= MemPos_RWSection;
1468 RWSectionHeader.SizeOfRawData= FileSize_RWSection;
1469 RWSectionHeader.PointerToRawData= FilePos_RWSection;
1470 RWSectionHeader.PointerToRelocations= 0;
1471 RWSectionHeader.PointerToLinenumbers= 0;
1472 RWSectionHeader.NumberOfRelocations= 0;
1473 RWSectionHeader.NumberOfLinenumbers= 0;
1474 RWSectionHeader.Characteristics= IMAGE_SCN_CNT_INITIALIZED_DATA|
1475 IMAGE_SCN_MEM_READ|
1476 IMAGE_SCN_MEM_WRITE;
1477
1478 //リソースセクションヘッダ
1479 IMAGE_SECTION_HEADER RSrcSectionHeader;
1480 memset((char *)RSrcSectionHeader.Name,0,IMAGE_SIZEOF_SHORT_NAME);
1481 lstrcpy((char *)RSrcSectionHeader.Name,".rsrc");
1482 RSrcSectionHeader.Misc.VirtualSize= RSrcSectionSize;
1483 RSrcSectionHeader.VirtualAddress= MemPos_RSrcSection;
1484 RSrcSectionHeader.SizeOfRawData= FileSize_RSrcSection;
1485 RSrcSectionHeader.PointerToRawData= FilePos_RSrcSection;
1486 RSrcSectionHeader.PointerToRelocations= 0;
1487 RSrcSectionHeader.PointerToLinenumbers= 0;
1488 RSrcSectionHeader.NumberOfRelocations= 0;
1489 RSrcSectionHeader.NumberOfLinenumbers= 0;
1490 RSrcSectionHeader.Characteristics= IMAGE_SCN_CNT_INITIALIZED_DATA|
1491 IMAGE_SCN_MEM_READ;
1492
1493 //リロケーションセクションヘッダ
1494 IMAGE_SECTION_HEADER RelocSectionHeader;
1495 memset((char *)RelocSectionHeader.Name,0,IMAGE_SIZEOF_SHORT_NAME);
1496 lstrcpy((char *)RelocSectionHeader.Name,".reloc");
1497 RelocSectionHeader.Misc.VirtualSize= pobj_Reloc->length;
1498 RelocSectionHeader.VirtualAddress= MemPos_RelocSection; //開始アドレス
1499 RelocSectionHeader.SizeOfRawData= FileSize_RelocSection; //サイズ
1500 RelocSectionHeader.PointerToRawData= FilePos_RelocSection; //ファイル上の開始アドレス
1501 RelocSectionHeader.PointerToRelocations= 0;
1502 RelocSectionHeader.PointerToLinenumbers= 0;
1503 RelocSectionHeader.NumberOfRelocations= 0;
1504 RelocSectionHeader.NumberOfLinenumbers= 0;
1505 RelocSectionHeader.Characteristics= IMAGE_SCN_CNT_INITIALIZED_DATA|
1506 IMAGE_SCN_MEM_DISCARDABLE|
1507 IMAGE_SCN_MEM_READ;
1508
1509 //デバッグセクションヘッダ
1510 IMAGE_SECTION_HEADER DebugSectionHeader;
1511 memset((char *)DebugSectionHeader.Name,0,IMAGE_SIZEOF_SHORT_NAME);
1512 lstrcpy((char *)DebugSectionHeader.Name,".debug");
1513 DebugSectionHeader.Misc.VirtualSize= pobj_DebugSection->length;
1514 DebugSectionHeader.VirtualAddress= MemPos_DebugSection; //開始アドレス
1515 DebugSectionHeader.SizeOfRawData= FileSize_DebugSection; //サイズ
1516 DebugSectionHeader.PointerToRawData= FilePos_DebugSection; //ファイル上の開始アドレス
1517 DebugSectionHeader.PointerToRelocations= 0;
1518 DebugSectionHeader.PointerToLinenumbers= 0;
1519 DebugSectionHeader.NumberOfRelocations= 0;
1520 DebugSectionHeader.NumberOfLinenumbers= 0;
1521 DebugSectionHeader.Characteristics= IMAGE_SCN_MEM_DISCARDABLE|
1522 IMAGE_SCN_MEM_READ;
1523
1524
1525 hFile=CreateFile(OutputFileName,GENERIC_WRITE,0,NULL,CREATE_ALWAYS,FILE_ATTRIBUTE_NORMAL,NULL);
1526 if(hFile==INVALID_HANDLE_VALUE){
1527 SetError(53,OutputFileName,-1);
1528 goto EndWriteOpcode;
1529 }
1530
1531 //ヘッダ
1532 WriteFile(hFile,(void *)&ImageDosHeader,sizeof(IMAGE_DOS_HEADER),(DWORD *)&i2,NULL);
1533 i=i2;
1534
1535 //Dosスタブ
1536 WriteFile(hFile,DosStubBuffer,DosStubSize,(DWORD *)&i2,NULL);
1537 i+=i2;
1538
1539 //0x0100までNULLを並べる
1540 temp2=(char *)HeapAlloc(hHeap,HEAP_ZERO_MEMORY,0x0100-i);
1541 WriteFile(hFile,temp2,0x0100-i,(DWORD *)&i2,NULL);
1542 HeapDefaultFree(temp2);
1543 i+=i2;
1544
1545 //PEヘッダ
1546 WriteFile(hFile,&ImagePeHdr,sizeof(IMAGE_NT_HEADERS),(DWORD *)&i2,NULL);
1547 i+=i2;
1548
1549 //コード セクション ヘッダ
1550 WriteFile(hFile,&CodeSectionHeader,sizeof(IMAGE_SECTION_HEADER),(DWORD *)&i2,NULL);
1551 i+=i2;
1552
1553 if(bUse_ExportSection){
1554 //エクスポート セクション ヘッダ
1555 WriteFile(hFile,&ExportSectionHeader,sizeof(IMAGE_SECTION_HEADER),(DWORD *)&i2,NULL);
1556 i+=i2;
1557 }
1558 if(bUse_ImportSection){
1559 //インポート セクション ヘッダ
1560 WriteFile(hFile,&ImportSectionHeader,sizeof(IMAGE_SECTION_HEADER),(DWORD *)&i2,NULL);
1561 i+=i2;
1562 }
1563 if(bUse_DataSection){
1564 //データ セクション ヘッダ
1565 WriteFile(hFile,&DataSectionHeader,sizeof(IMAGE_SECTION_HEADER),(DWORD *)&i2,NULL);
1566 i+=i2;
1567 }
1568 if(bUse_RWSection){
1569 //リライタブルセクションヘッダ
1570 WriteFile(hFile,&RWSectionHeader,sizeof(IMAGE_SECTION_HEADER),(DWORD *)&i2,NULL);
1571 i+=i2;
1572 }
1573 if(bUse_RSrcSection){
1574 //リソースセクションヘッダ
1575 WriteFile(hFile,&RSrcSectionHeader,sizeof(IMAGE_SECTION_HEADER),(DWORD *)&i2,NULL);
1576 i+=i2;
1577 }
1578 if(bUse_RelocSection){
1579 //リロケーションセクションヘッダ
1580 WriteFile(hFile,&RelocSectionHeader,sizeof(IMAGE_SECTION_HEADER),(DWORD *)&i2,NULL);
1581 i+=i2;
1582 }
1583 if(bUse_DebugSection){
1584 //デバッグセクションヘッダ
1585 WriteFile(hFile,&DebugSectionHeader,sizeof(IMAGE_SECTION_HEADER),(DWORD *)&i2,NULL);
1586 i+=i2;
1587 }
1588
1589 //EXE_HEADER_SIZEまでNULLを並べる
1590 temp2=(char *)HeapAlloc(hHeap,HEAP_ZERO_MEMORY,EXE_HEADER_SIZE-i);
1591 WriteFile(hFile,temp2,EXE_HEADER_SIZE-i,(DWORD *)&i2,NULL);
1592 HeapDefaultFree(temp2);
1593 i+=i2;
1594
1595 //コード
1596 WriteFile(hFile,OpBuffer,obp,(DWORD *)&i2,NULL);
1597 i+=i2;
1598
1599 //FilePos_ExportSectionまでNULLを並べる
1600 temp2=(char *)HeapAlloc(hHeap,HEAP_ZERO_MEMORY,FilePos_ExportSection-i);
1601 WriteFile(hFile,temp2,FilePos_ExportSection-i,(DWORD *)&i2,NULL);
1602 HeapDefaultFree(temp2);
1603 i+=i2;
1604
1605 if(bUse_ExportSection){
1606 //エクスポート ディレクトリ テーブル
1607 WriteFile(hFile,&ImageExportDirectory,sizeof(IMAGE_EXPORT_DIRECTORY),(DWORD *)&i2,NULL);
1608 i+=i2;
1609
1610 //エクスポート アドレス テーブル
1611 WriteFile(hFile,lpdwExportAddressTable,ExportNum*sizeof(DWORD),(DWORD *)&i2,NULL);
1612 i+=i2;
1613
1614 //エクスポート名ポインタ テーブル
1615 WriteFile(hFile,lpdwExportNamePointerTable,ExportNum*sizeof(DWORD),(DWORD *)&i2,NULL);
1616 i+=i2;
1617
1618 //エクスポート序数テーブル
1619 WriteFile(hFile,lpwExportOrdinalTable,ExportNum*sizeof(WORD),(DWORD *)&i2,NULL);
1620 i+=i2;
1621
1622 //シンボル名
1623 WriteFile(hFile,lpExportNames,ExportNamesLength,(DWORD *)&i2,NULL);
1624 i+=i2;
1625 }
1626
1627 //FilePos_ImportSectionまでNULLを並べる
1628 temp2=(char *)HeapAlloc(hHeap,HEAP_ZERO_MEMORY,FilePos_ImportSection-i);
1629 WriteFile(hFile,temp2,FilePos_ImportSection-i,(DWORD *)&i2,NULL);
1630 HeapDefaultFree(temp2);
1631 i+=i2;
1632
1633 if(bUse_ImportSection){
1634 //インポート ディレクトリ テーブル(Nullディレクトリ テーブルを含む)
1635 for(i3=0;i3<(ImportDllNum+1);i3++){
1636 WriteFile(hFile,&pImportTable[i3],sizeof(IMAGE_IMPORT_DESCRIPTOR),(DWORD *)&i2,NULL);
1637 i+=i2;
1638 }
1639
1640 //DLL名
1641 for(i3=0;i3<ImportDllNum;i3++){
1642 WriteFile(hFile,ppDllNames[i3],16,(DWORD *)&i2,NULL);
1643 i+=i2;
1644 }
1645
1646 //ルックアップ テーブル
1647 WriteFile(hFile,pLookupTable,LookupSize,(DWORD *)&i2,NULL);
1648 i+=i2;
1649
1650 //ヒント テーブル
1651 WriteFile(hFile,pHintTable,HintSize,(DWORD *)&i2,NULL);
1652 i+=i2;
1653
1654 //インポート アドレス テーブル
1655 WriteFile(hFile,pLookupTable,LookupSize,(DWORD *)&i2,NULL);
1656 i+=i2;
1657 }
1658
1659 //FilePos_DataSectionまでNULLを並べる
1660 temp2=(char *)HeapAlloc(hHeap,HEAP_ZERO_MEMORY,FilePos_DataSection-i);
1661 WriteFile(hFile,temp2,FilePos_DataSection-i,(DWORD *)&i2,NULL);
1662 HeapDefaultFree(temp2);
1663 i+=i2;
1664
1665 if(bUse_DataSection){
1666 //データ テーブル
1667 WriteFile(hFile,compiler.GetObjectModule().dataTable.GetPtr(),compiler.GetObjectModule().dataTable.GetSize(),(DWORD *)&i2,NULL);
1668 i+=i2;
1669 }
1670
1671 //FilePos_RWSectionまでNULLを並べる
1672 temp2=(char *)HeapAlloc(hHeap,HEAP_ZERO_MEMORY,FilePos_RWSection-i);
1673 WriteFile(hFile,temp2,FilePos_RWSection-i,(DWORD *)&i2,NULL);
1674 HeapDefaultFree(temp2);
1675 i+=i2;
1676
1677 if(bUse_RWSection){
1678 //リライタブル データ テーブル(グローバル変数の初期バッファ)
1679 initGlobalBuf=(BYTE *)HeapReAlloc(hHeap,
1680 HEAP_ZERO_MEMORY,
1681 initGlobalBuf,
1682 FileSize_RWSection);
1683 WriteFile(hFile,initGlobalBuf,FileSize_RWSection,(DWORD *)&i2,NULL);
1684 i+=i2;
1685 }
1686
1687 //FilePos_RSrcSectionまでNULLを並べる
1688 temp2=(char *)HeapAlloc(hHeap,HEAP_ZERO_MEMORY,FilePos_RSrcSection-i);
1689 WriteFile(hFile,temp2,FilePos_RSrcSection-i,(DWORD *)&i2,NULL);
1690 HeapDefaultFree(temp2);
1691 i+=i2;
1692
1693 if(bUse_RSrcSection){
1694 //リソースバッファ
1695 WriteFile(hFile,RSrcSectionBuffer,RSrcSectionSize,(DWORD *)&i2,NULL);
1696 i+=i2;
1697 }
1698
1699 //FilePos_RelocSectionまでNULLを並べる
1700 temp2=(char *)HeapAlloc(hHeap,HEAP_ZERO_MEMORY,FilePos_RelocSection-i);
1701 WriteFile(hFile,temp2,FilePos_RelocSection-i,(DWORD *)&i2,NULL);
1702 HeapDefaultFree(temp2);
1703 i+=i2;
1704
1705 if(bUse_RelocSection){
1706 //リロケーション情報
1707 WriteFile(hFile,pobj_Reloc->buffer,pobj_Reloc->length,(DWORD *)&i2,NULL);
1708 i+=i2;
1709 }
1710
1711 //ファイルアラインメントを考慮
1712 if(i%FILE_ALIGNMENT){
1713 temp2=(char *)HeapAlloc(hHeap,HEAP_ZERO_MEMORY,FILE_ALIGNMENT-i%FILE_ALIGNMENT);
1714 WriteFile(hFile,temp2,FILE_ALIGNMENT-i%FILE_ALIGNMENT,(DWORD *)&i2,NULL);
1715 HeapDefaultFree(temp2);
1716 i+=i2;
1717 }
1718
1719 if(bUse_DebugSection){
1720 //デバッグセクション
1721 WriteFile(hFile,pobj_DebugSection->buffer,pobj_DebugSection->length,(DWORD *)&i2,NULL);
1722 i+=i2;
1723 }
1724
1725 //ファイルアラインメントを考慮
1726 if(i%FILE_ALIGNMENT){
1727 temp2=(char *)HeapAlloc(hHeap,HEAP_ZERO_MEMORY,FILE_ALIGNMENT-i%FILE_ALIGNMENT);
1728 WriteFile(hFile,temp2,FILE_ALIGNMENT-i%FILE_ALIGNMENT,(DWORD *)&i2,NULL);
1729 HeapDefaultFree(temp2);
1730 i+=i2;
1731 }
1732
1733 //書き込み終了
1734 CloseHandle(hFile);
1735
1736
1737EndWriteOpcode:
1738
1739 //Dosスタブ用のメモリを解放
1740 HeapDefaultFree(DosStubBuffer);
1741
1742 //エクスポート テーブル情報を解放
1743 HeapDefaultFree(lpdwExportAddressTable);
1744 HeapDefaultFree(lpdwExportNamePointerTable);
1745 HeapDefaultFree(lpwExportOrdinalTable);
1746
1747 //インポートDLL情報を解放
1748 HeapDefaultFree(pImportTable);
1749 for(i=0;i<ImportDllNum;i++)
1750 HeapDefaultFree(ppDllNames[i]);
1751 HeapDefaultFree(ppDllNames);
1752
1753 //ルックアップテーブルに関する情報を解放
1754 HeapDefaultFree(pLookupTable);
1755
1756 //ヒントテーブルに関する情報を解放
1757 HeapDefaultFree(pHintTable);
1758
1759 //グローバル変数の初期バッファを解放
1760 HeapDefaultFree(initGlobalBuf);
1761
1762 //コードバッファを解放
1763 free(OpBuffer);
1764 OpBuffer=0;
1765
1766 //リソースセクションバッファを解放
1767 HeapDefaultFree(RSrcSectionBuffer);
1768
1769 //デバッグセクションを開放
1770 delete pobj_DebugSection;
1771
1772 //リロケーション情報を解放
1773 delete pobj_Reloc;
1774
1775 //列挙体に関する情報の破棄
1776 CEnumParent::DestroyEnum();
1777
1778 //クラスに関するメモリを解放
1779 compiler.GetObjectModule().meta.GetClasses().Clear();
1780}
Note: See TracBrowser for help on using the repository browser.