source: dev/trunk/ab5.0/abdev/BasicCompiler_Common/src/Exception.cpp @ 591

#include "stdafx.h"

#ifdef _AMD64_
#include "../../compiler_x64/opcode.h"
#else
#include "../../compiler_x86/opcode.h"
#endif

namespace Exception{

class CatchScope
{
    Type paramType;
    long codePos;
public:
    CatchScope( const Type &paramType, long codePos )
        : paramType( paramType )
        , codePos( codePos )
    {
    }
    ~CatchScope()
    {
    }

    const Type &GetParamType() const
    {
        return paramType;
    }
    long GetCodePos() const
    {
        return codePos;
    }
};
typedef std::vector<CatchScope> CatchScopes;

class TryScope
{
    int sourceCodePos;
    bool isDefinedFinally;

    CatchScopes catchScopes;

    std::vector<const PertialSchedule *> finallySchedules;
    void JmpFinally()
    {
        finallySchedules.push_back(
            compiler.codeGenerator.op_jmp( 0, sizeof(long), true )
        );
    }
    void ResolveJmpFinally()
    {
        BOOST_FOREACH( const PertialSchedule *pPertialSchedule, finallySchedules )
        {
            compiler.codeGenerator.opfix_JmpPertialSchedule( pPertialSchedule );
        }
    }

    const PertialSchedule *pPertialScheduleForCatchAddress;
    const PertialSchedule *pPertialScheduleForFinallyAddress;

public:
    TryScope( int sourceCodePos )
        : sourceCodePos( sourceCodePos )
        , isDefinedFinally( false )
    {
    }
    ~TryScope()
    {
    }

    int GetSourceCodePos() const
    {
        return sourceCodePos;
    }
    const CatchScopes &GetCatchScopes() const
    {
        return catchScopes;
    }
    bool IsCatched() const
    {
        return ( catchScopes.size() != 0 );
    }
    bool IsDefinedFinally() const
    {
        return isDefinedFinally;
    }

    void RegistPertialScheduleForCatchAddress( const PertialSchedule *pPertialScheduleForCatchAddress )
    {
        this->pPertialScheduleForCatchAddress = pPertialScheduleForCatchAddress;
    }
    const PertialSchedule *GetPertialScheduleForCatchAddress() const
    {
        return pPertialScheduleForCatchAddress;
    }
    void RegistPertialScheduleForFinallyAddress( const PertialSchedule *pPertialScheduleForFinallyAddress )
    {
        this->pPertialScheduleForFinallyAddress = pPertialScheduleForFinallyAddress;
    }
    const PertialSchedule *GetPertialScheduleForFinallyAddress() const
    {
        return pPertialScheduleForFinallyAddress;
    }
    
    

    void Try()
    {
        // レキシカルスコープをレベルアップ
        compiler.codeGenerator.lexicalScopes.Start(
            compiler.codeGenerator.GetNativeCodeSize(),
            LexicalScope::SCOPE_TRY
        );
    }

    void Catch( Type &paramType )
    {
        if( isDefinedFinally )
        {
            compiler.errorMessenger.Output(71,NULL,cp);
            return;
        }

        if( catchScopes.size() )
        {
            // 既に1回以上のCatchが存在するとき

            // レキシカルスコープをレベルダウン
            compiler.codeGenerator.lexicalScopes.End();
        }

        JmpFinally();

        catchScopes.push_back( CatchScope( paramType, compiler.codeGenerator.GetNativeCodeSize() ) );

        // レキシカルスコープをレベルアップ
        compiler.codeGenerator.lexicalScopes.Start(
            compiler.codeGenerator.GetNativeCodeSize(),
            LexicalScope::SCOPE_CATCH
        );
    }
    void Finally()
    {
        if( isDefinedFinally )
        {
            compiler.errorMessenger.Output(70,NULL,cp);
            return;
        }

        if( catchScopes.size() )
        {
            // 既に1回以上のCatchが存在するとき

            // レキシカルスコープをレベルダウン
            compiler.codeGenerator.lexicalScopes.End();
        }

        isDefinedFinally = true;

        ResolveJmpFinally();

        // レキシカルスコープをレベルアップ
        compiler.codeGenerator.lexicalScopes.Start(
            compiler.codeGenerator.GetNativeCodeSize(),
            LexicalScope::SCOPE_FINALLY
        );
    }

    void EndTry()
    {
        if( !isDefinedFinally )
        {
            compiler.errorMessenger.OutputFatalError();
        }

        if( catchScopes.size() || isDefinedFinally )
        {
            // 既に1回以上のCatch、またはFinallyが存在するとき

            // レキシカルスコープをレベルダウン
            compiler.codeGenerator.lexicalScopes.End();
        }

        // レキシカルスコープをレベルダウン
        compiler.codeGenerator.lexicalScopes.End();
    }

    int GenerateCatchTable()
    {
        int size = static_cast<int>(( (catchScopes.size()+1)*2 ) * sizeof(LONG_PTR));
        BYTE *buffer = (BYTE *)calloc( size, 1 );

        int pos = 0;
        BOOST_FOREACH( const CatchScope &catchScope, catchScopes )
        {
            // パラメータのクラス名
            char paramName[VN_SIZE] = "";
            int paramNameDataTableOffset = 0;
            if( catchScope.GetParamType().IsObject() )
            {
                lstrcpy( paramName, catchScope.GetParamType().GetClass().GetFullName().c_str() );
            }
            if( compiler.IsUnicode() )
            {
                paramNameDataTableOffset = compiler.GetObjectModule().dataTable.AddWString( Jenga::Common::ToWString( paramName ) );
            }
            else
            {
                paramNameDataTableOffset = compiler.GetObjectModule().dataTable.AddString( paramName );
            }
            *((LONG_PTR *)(buffer+pos)) = paramNameDataTableOffset;
            pos += sizeof(LONG_PTR);

            // Catchアドレス
            *((LONG_PTR *)(buffer+pos)) = catchScope.GetCodePos();
            pos += sizeof(LONG_PTR);
        }

        int dataTableOffset = compiler.GetObjectModule().dataTable.AddBinary( buffer, size );

        free( buffer );

        pos = 0;
        BOOST_FOREACH( const CatchScope &catchScope, catchScopes )
        {
            // パラメータのクラス名
            compiler.GetObjectModule().dataTable.schedules.push_back( Schedule( Schedule::DataTable, dataTableOffset + pos ) );
            pos += sizeof(LONG_PTR);

            // Catchアドレス
            const UserProc *pUserProc = &compiler.GetCompilingUserProc();
            if( compiler.IsGlobalAreaCompiling() )
            {
                pUserProc = UserProc::pGlobalProc;
            }
            compiler.GetObjectModule().dataTable.schedules.push_back(
                Schedule( Schedule::CatchAddress, pUserProc, dataTableOffset + pos )
            );
            pos += sizeof(LONG_PTR);
        }

        return dataTableOffset;
    }
};
typedef std::vector<TryScope> TryScopes;

TryScopes tryScopes;

void TryCommand()
{
    tryScopes.push_back( TryScope( cp ) );
    tryScopes.back().Try();

    int backCp = cp;

    char temporary[1024];
    lstrcpy( temporary, "ExceptionService.BeginTryScope( _System_GetNowScopeCatchAddresses() As VoidPtr, _System_GetNowScopeFinallyAddresses() As VoidPtr, _System_GetBp() As LONG_PTR, _System_GetSp() As LONG_PTR )" );
    MakeMiddleCode( temporary );
    ChangeOpcode( temporary );

    cp = backCp;
}
void CatchCommand( const char *parameter )
{
    if( tryScopes.size() == 0 )
    {
        compiler.errorMessenger.Output(1,NULL,cp);
        return;
    }

    char varName[VN_SIZE];
    Type paramType;
    if( parameter[0] )
    {
        char typeName[VN_SIZE];
        SplitSyntacticForAs( parameter, varName, typeName );
        if( !typeName[0] )
        {
            compiler.errorMessenger.Output(72,NULL,cp);
        }
        else
        {
            if( !compiler.StringToType( typeName, paramType ) )
            {
                compiler.errorMessenger.Output(74,typeName,cp);
            }
            else
            {
                if( !paramType.IsObject() )
                {
                    compiler.errorMessenger.Output(73,typeName,cp);
                }
            }
        }
    }

    tryScopes.back().Catch( paramType );

    int backCp = cp;

    char temporary[1024];
    if( paramType.IsObject() )
    {
        sprintf( temporary, "Dim %s = System.Threading.Thread.CurrentThread().__GetThrowintParamObject() As %s", varName, paramType.GetClass().GetFullName().c_str() );
        MakeMiddleCode( temporary );
        ChangeOpcode( temporary );
    }
    lstrcpy( temporary, "System.Threading.Thread.CurrentThread().__Catched()" );
    MakeMiddleCode( temporary );
    ChangeOpcode( temporary );

    cp = backCp;
}
void FinallyCommand()
{
    tryScopes.back().Finally();

    int backCp = cp;

    char temporary[1024];
    lstrcpy( temporary, "_System_pobj_AllThreads->GetCurrentException()->FinishFinally()" );
    MakeMiddleCode( temporary );
    ChangeOpcode( temporary );

    cp = backCp;

    //Tryスコープに入るときに引き渡されるパラメータ値を解決
    compiler.codeGenerator.opfix( tryScopes.back().GetPertialScheduleForFinallyAddress(), compiler.codeGenerator.GetNativeCodeSize() );
}
void EndTryCommand( bool isNoWarning )
{
    if( tryScopes.size() == 0 )
    {
        compiler.errorMessenger.Output(12,"End Try",cp);
        return;
    }

    if( !isNoWarning && !tryScopes.back().IsDefinedFinally() && !tryScopes.back().IsCatched() )
    {
        // CatchもFinallyも存在しないとき
        compiler.errorMessenger.Output(-108,NULL,tryScopes.back().GetSourceCodePos() );
    }

    int dataTableOffset = tryScopes.back().GenerateCatchTable();

    // _System_GetNowScopeCatchAddressesを解決
    compiler.codeGenerator.opfix( tryScopes.back().GetPertialScheduleForCatchAddress(), dataTableOffset );

    if( !tryScopes.back().IsDefinedFinally() )
    {
        // Finallyが定義されていないときは空のFinallyを定義しておく
        FinallyCommand();
    }

    int backCp = cp;

    char temporary[1024];
    lstrcpy( temporary, "_System_pobj_AllThreads->GetCurrentException()->EndTryScope()" );
    MakeMiddleCode( temporary );
    ChangeOpcode( temporary );

    cp = backCp;

    tryScopes.back().EndTry();
    tryScopes.pop_back();
}

void InspectTryScope()
{
    while( tryScopes.size() > 0 )
    {
        compiler.errorMessenger.Output(22, "Try", tryScopes.back().GetSourceCodePos() );

        EndTryCommand( true );
    }
}

void ThrowCommand( const char *Parameter )
{
    int backCp = cp;

    char temporary[1024];
    if( Parameter[0] )
    {
        sprintf( temporary, "_System_pobj_AllThreads->GetCurrentException()->_ThrowWithParam(%s)", Parameter );
    }
    else
    {
        lstrcpy( temporary, "_System_pobj_AllThreads->GetCurrentException()->_ThrowNoneParam()" );
    }
    MakeMiddleCode( temporary );
    ChangeOpcode( temporary );

    cp = backCp;
}

void Opcode_Func_System_GetNowScopeCatchAddress()
{
    if( tryScopes.size() == 0 )
    {
        compiler.errorMessenger.Output(1,NULL,cp);
        return;
    }

#ifdef _WIN64
    //mov rax,catchAddress
    const PertialSchedule *pPertialSchedule = compiler.codeGenerator.op_mov_RV( sizeof(long), REG_RAX, 0, Schedule::DataTable, true );
#else
    //mov eax,catchAddress
    const PertialSchedule *pPertialSchedule = compiler.codeGenerator.op_mov_RV( REG_EAX, 0, Schedule::DataTable, true );
#endif

    tryScopes.back().RegistPertialScheduleForCatchAddress( pPertialSchedule );

    /*
    int dataTableOffset = compiler.GetObjectModule().dataTable.Add( static_cast<LONG_PTR>(0) );

#ifdef _WIN64
    //mov rax,dataTableOffset
    compiler.codeGenerator.op_mov_RV( sizeof(_int64), REG_RAX, dataTableOffset, Schedule::DataTable);
#else
    //mov eax,dataTableOffset
    compiler.codeGenerator.op_mov_RV( REG_EAX, dataTableOffset, Schedule::DataTable);
#endif
    */
}

void Opcode_Func_System_GetNowScopeFinallyAddress()
{
    if( tryScopes.size() == 0 )
    {
        compiler.errorMessenger.Output(1,NULL,cp);
        return;
    }

#ifdef _WIN64
    //mov rax,finallyAddress
    const PertialSchedule *pPertialSchedule = compiler.codeGenerator.op_mov_RV( sizeof(long), REG_RAX, 0, Schedule::CatchAddress, true );
#else
    //mov eax,finallyAddress
    const PertialSchedule *pPertialSchedule = compiler.codeGenerator.op_mov_RV( REG_EAX, 0, Schedule::CatchAddress, true );
#endif

    tryScopes.back().RegistPertialScheduleForFinallyAddress( pPertialSchedule );

    /*
    int dataTableOffset = compiler.GetObjectModule().dataTable.Add( static_cast<LONG_PTR>(0) );

#ifdef _WIN64
    //mov rax,dataTableOffset
    compiler.codeGenerator.op_mov_RV( sizeof(_int64), REG_RAX, dataTableOffset, Schedule::DataTable);
#else
    //mov eax,dataTableOffset
    compiler.codeGenerator.op_mov_RV( REG_EAX, dataTableOffset, Schedule::DataTable);
#endif
    */
}


}   // Exception