source: dev/trunk/abdev/BasicCompiler_Common/src/Linker.cpp @ 370

#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 )
            );
        }
    }

    BOOST_FOREACH( const Schedule &schedule, dataTable.schedules )
    {
        if( schedule.GetType() == Schedule::DataTable )
        {
#ifdef _WIN64
            dataTable.OverwriteInt64(
                schedule.GetOffset(),
                dataTable.GetInt64( schedule.GetOffset() ) + imageBase + dataSectionBaseOffset
            );
#else
            dataTable.Overwrite(
                schedule.GetOffset(),
                dataTable.GetLong( schedule.GetOffset() ) + imageBase + dataSectionBaseOffset
            );
#endif
        }
    }
}

// Catchアドレス スケジュール
void Linker::ResolveCatchAddressSchedules( long codeSectionBaseOffset )
{
    BOOST_FOREACH( const Schedule &schedule, nativeCode.GetSchedules() )
    {
        if( schedule.GetType() == Schedule::CatchAddress )
        {
            if( nativeCode.GetLong( schedule.GetOffset() ) != 0 )
            {
                // 置き換える値が0の場合を除く
                nativeCode.Overwrite(
                    schedule.GetOffset(),
                    static_cast<long>( nativeCode.GetLong( schedule.GetOffset() ) + schedule.GetUserProc().GetBeginOpAddress() + imageBase + codeSectionBaseOffset )
                );
            }
        }
    }

    BOOST_FOREACH( const Schedule &schedule, dataTable.schedules )
    {
        if( schedule.GetType() == Schedule::CatchAddress )
        {
            if( nativeCode.GetLong( schedule.GetOffset() ) != 0 )
            {
                // 置き換える値が0の場合を除く
#ifdef _WIN64
                dataTable.OverwriteInt64(
                    schedule.GetOffset(),
                    dataTable.GetInt64( schedule.GetOffset() ) + schedule.GetUserProc().GetBeginOpAddress() + imageBase + codeSectionBaseOffset
                );
#else
                dataTable.Overwrite(
                    schedule.GetOffset(),
                    dataTable.GetLong( schedule.GetOffset() ) + schedule.GetUserProc().GetBeginOpAddress() + imageBase + codeSectionBaseOffset
                );
#endif
            }
        }
    }
}

// DLL関数スケジュール
void Linker::ResolveDllProcSchedules( long codeSectionBaseOffset, long importSectionBaseOffset, long lookupSize, long hintSize )
{
    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
            nativeCode.Overwrite(
                schedule.GetOffset(),
                static_cast<long>( imageBase + importSectionBaseOffset + lookupSize + hintSize
                    + schedule.GetDllProc().GetLookupAddress() )
            );
#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 )
{
    int allInitVarSize = compiler.GetObjectModule().meta.GetGlobalVars().initAreaBuffer.GetSize();

    BOOST_FOREACH( const Schedule &schedule, nativeCode.GetSchedules() )
    {
        if( schedule.GetType() == Schedule::GlobalVar )
        {
            if( nativeCode.GetLong( schedule.GetOffset() ) & 0x80000000 )
            {
                nativeCode.Overwrite(
                    schedule.GetOffset(),
                    static_cast<long>( allInitVarSize + (nativeCode.GetLong( schedule.GetOffset() ) & 0x7FFFFFFF) + imageBase + rwSectionBaseOffset )
                );
            }
            else
            {
                nativeCode.Overwrite(
                    schedule.GetOffset(),
                    static_cast<long>( nativeCode.GetLong( schedule.GetOffset() ) + imageBase + rwSectionBaseOffset )
                );
            }
        }
    }
}

void Linker::ResolveVtblSchedule( long dataSectionBaseOffset )
{
    BOOST_FOREACH( const Schedule &schedule, nativeCode.GetSchedules() )
    {
        if( schedule.GetType() == Schedule::ComVtbl )
        {
            LONG_PTR vtblOffset = schedule.GetClass().GetComVtblOffset();

            nativeCode.Overwrite(
                schedule.GetOffset(),
                static_cast<long>( vtblOffset + imageBase + dataSectionBaseOffset )
            );
        }

        if( schedule.GetType() == Schedule::Vtbl )
        {
            LONG_PTR vtblMasterListOffset = schedule.GetClass().GetVtblMasterListOffset();

            nativeCode.Overwrite(
                schedule.GetOffset(),
                static_cast<long>( vtblMasterListOffset + imageBase + dataSectionBaseOffset )
            );
        }
    }

    BOOST_FOREACH( const Schedule &schedule, dataTable.schedules )
    {
        if( schedule.GetType() == Schedule::ComVtbl )
        {
            LONG_PTR vtblOffset = schedule.GetClass().GetComVtblOffset();

#ifdef _WIN64
            dataTable.OverwriteInt64(
                schedule.GetOffset(),
                vtblOffset + imageBase + dataSectionBaseOffset
            );
#else
            dataTable.Overwrite(
                schedule.GetOffset(),
                vtblOffset + imageBase + dataSectionBaseOffset
            );
#endif
        }

        if( schedule.GetType() == Schedule::Vtbl )
        {
            LONG_PTR vtblMasterListOffset = schedule.GetClass().GetVtblMasterListOffset();

#ifdef _WIN64
            dataTable.OverwriteInt64(
                schedule.GetOffset(),
                vtblMasterListOffset + imageBase + dataSectionBaseOffset
            );
#else
            dataTable.Overwrite(
                schedule.GetOffset(),
                vtblMasterListOffset + imageBase + dataSectionBaseOffset
            );
#endif
        }
    }
}

void Linker::ResolveTypeInfoSchedule( long dataSectionBaseOffset )
{
    BOOST_FOREACH( const Schedule &schedule, dataTable.schedules )
    {
        if( schedule.GetType() == Schedule::TypeInfo )
        {
            LONG_PTR typeInfoDataTableOffset = schedule.GetClass().GetTypeInfoDataTableOffset();

#ifdef _WIN64
            dataTable.OverwriteInt64(
                schedule.GetOffset(),
                typeInfoDataTableOffset + imageBase + dataSectionBaseOffset
            );
#else
            dataTable.Overwrite(
                schedule.GetOffset(),
                typeInfoDataTableOffset + imageBase + dataSectionBaseOffset
            );
#endif
        }
    }
}

void Linker::Link( ObjectModule &masterObjectModule )
{
    // nativeCodeは初期状態でなければならない
    if( nativeCode.GetSize() > 0 )
    {
        SetError();
    }

    nativeCode.PutEx( masterObjectModule.globalNativeCode );

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

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

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

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

void Linker::SetDataTable( DataTable &dataTable )
{
    this->dataTable.Add( dataTable );
}