#include "stdafx.h"

#include <Compiler.h>


// データテーブルスケジュール
void Linker::ResolveDataTableSchedules( long dataSectionBaseOffset )
{
	BOOST_FOREACH( const Schedule &schedule, nativeCode.GetSchedules() )
	{
		if( schedule.GetType() == Schedule::DataTable )
		{
			nativeCode.Overwrite(
				schedule.GetOffset(),
				static_cast<long>( nativeCode.GetLong( schedule.GetOffset() ) + imageBase + dataSectionBaseOffset )
			);
		}
	}
}

// DLL関数スケジュール
void Linker::ResolveDllProcSchedules( long codeSectionBaseOffset, long importSectionBaseOffset )
{
	BOOST_FOREACH( const Schedule &schedule, nativeCode.GetSchedules() )
	{
		if( schedule.GetType() == Schedule::DllProc )
		{
#ifdef _AMD64
			nativeCode.Overwrite(
				schedule.GetOffset(),
				static_cast<long>( importSectionBaseOffset + schedule.GetDllProc().GetLookupAddress()
					- ( codeSectionBaseOffset + schedule.GetOffset() + sizeof(long) ) )
			);
#else
			// TODO: 未完成
			SetError();
#endif
		}
	}
}

// ユーザ定義関数スケジュール
void Linker::ResolveUserProcSchedules( long codeSectionBaseOffset )
{
	BOOST_FOREACH( const Schedule &schedule, nativeCode.GetSchedules() )
	{
		if( schedule.GetType() == Schedule::UserProc
			|| schedule.GetType() == Schedule::AddressOf )
		{
			if( schedule.GetUserProc().GetBeginOpAddress() == 0
				&& schedule.GetUserProc().GetEndOpAddress() == 0 )
			{
				SetError();
			}

			if( schedule.GetType() == Schedule::UserProc )
			{
				nativeCode.Overwrite(
					schedule.GetOffset(),
					static_cast<long>( schedule.GetUserProc().GetBeginOpAddress() - ( schedule.GetOffset() + sizeof(long) ) )
				);
			}
			else if( schedule.GetType() == Schedule::AddressOf )
			{
				nativeCode.Overwrite(
					schedule.GetOffset(),
					static_cast<long>( schedule.GetUserProc().GetBeginOpAddress() + imageBase + codeSectionBaseOffset )
				);
			}
		}
	}
}

// グローバル変数スケジュール
void Linker::ResolveGlobalVarSchedules( long rwSectionBaseOffset )
{
	BOOST_FOREACH( const Schedule &schedule, nativeCode.GetSchedules() )
	{
		if( schedule.GetType() == Schedule::GlobalVar )
		{
			if( nativeCode.GetLong( schedule.GetOffset() ) & 0x80000000 )
			{
				extern int AllInitGlobalVarSize;
				nativeCode.Overwrite(
					schedule.GetOffset(),
					static_cast<long>( AllInitGlobalVarSize + (nativeCode.GetLong( schedule.GetOffset() ) & 0x7FFFFFFF) + imageBase + rwSectionBaseOffset )
				);
			}
			else
			{
				nativeCode.Overwrite(
					schedule.GetOffset(),
					static_cast<long>( nativeCode.GetLong( schedule.GetOffset() ) + imageBase + rwSectionBaseOffset )
				);
			}
		}
	}
}

void Linker::Link( vector<ObjectModule *> &objectModules )
{
	/*
	BOOST_FOREACH( ObjectModule *pObjectModule, objectModules )
	{
	}*/
	ObjectModule &masterObjectModule = *objectModules[0];

	nativeCode.Put( masterObjectModule.globalNativeCode );

	masterObjectModule.meta.GetUserProcs().Iterator_Reset();
	while( masterObjectModule.meta.GetUserProcs().Iterator_HasNext() )
	{
		const UserProc *pUserProc = masterObjectModule.meta.GetUserProcs().Iterator_GetNext();

		if( pUserProc->IsUsing() )
		{
			pUserProc->SetBeginOpAddress( nativeCode.GetSize() );

			nativeCode.Put( pUserProc->GetNativeCode() );

			pUserProc->SetEndOpAddress( nativeCode.GetSize() );
		}
	}
}