source: dev/trunk/ab5.0/abdev/compiler_x64/Compile_CallProc.cpp @ 514

#include "stdafx.h"

#include <Compiler.h>

#include "../BasicCompiler_Common/common.h"
#include "Opcode.h"

using namespace ActiveBasic::Compiler;

void Call_DebugSys_SaveContext(){
    //call _System_GetEip
    extern const UserProc *pSub_System_GetEip;
    compiler.codeGenerator.op_call(pSub_System_GetEip);

    //mov rdx,rax
    compiler.codeGenerator.op_mov_RR(REG_RDX,REG_RAX);

    //mov rcx,rsp
    compiler.codeGenerator.op_mov_RR(REG_RCX,REG_RSP);

    //call _DebugSys_SaveContext
    extern const UserProc *pSub_DebugSys_SaveContext;
    compiler.codeGenerator.op_call(pSub_DebugSys_SaveContext);
}

bool Opcode_CallProcPtr( const char *variable, const char *lpszParms,ProcPointer *pProcPointer)
{
    extern BOOL bDebugSupportProc;
    if( compiler.IsDebug() && bDebugSupportProc == 0 )
    {
        Call_DebugSys_SaveContext();
    }


    ////////////////////////
    // パラメータのセット
    ////////////////////////

    //パラメータオブジェクトを生成
    ParamImpl *pobj_parameter=0;
    pobj_parameter=new ParamImpl(lpszParms);

    // デフォルト引数を適用
    pobj_parameter->ApplyDefaultParameters( pProcPointer->Params() );

    //エラーチェック
    if( !pobj_parameter->ErrorCheck(variable,pProcPointer->Params() ) ){
        //パラメータにエラーがあるときは処理を終える
        return false;
    }

    //スタックフレームに存在する既存のパラメータをバックアップ
    pobj_parameter->BackupParameter( (int)pProcPointer->Params().size() );

    //一時オブジェクトを生成
    pobj_parameter->NewTempParameters( variable,pProcPointer->Params() );

    //レジスタ、スタックフレームにセット
    pobj_parameter->SetParameter(variable,pProcPointer->Params() );



    RELATIVE_VAR RelativeVar;
    GetVarOffsetReadOnly(variable,&RelativeVar,Type());
    SetVarPtrToReg(REG_RAX,&RelativeVar);

    //mov rax,qword ptr[rax]
    compiler.codeGenerator.op_mov_RM(sizeof(_int64),REG_RAX,REG_RAX,0,MOD_BASE);

    //call rax
    compiler.codeGenerator.PutOld(
        (char)0xFF,
        (char)0xD0
    );


    //レジスタのブロッキングを解除        ※パラメータセット時にロックされたレジスタ
    pobj_BlockReg->clear();

    //一時オブジェクトを破棄
    pobj_parameter->DeleteTempParameters();

    //スタックフレームに存在する既存のパラメータを復元
    pobj_parameter->RestoreParameter( (int)pProcPointer->Params().size() );

    //パラメータオブジェクトを破棄
    delete pobj_parameter;

    return true;
}

bool Opcode_CallProc(const char *Parameter,const UserProc *pUserProc,DWORD dwFlags,const char *ObjectName)
{
    if( pUserProc->IsMacro() ){
        if( lstrcmpi( pUserProc->GetName().c_str(), "Print" ) == 0 ){
            Opcode_Print(Parameter,0);
            return true;
        }
        if( lstrcmpi( pUserProc->GetName().c_str(), "Input" ) == 0 ){
            Opcode_Input(Parameter);
            return true;
        }
        if( lstrcmpi( pUserProc->GetName().c_str(), "Write" ) == 0 ){
            Opcode_Print(Parameter,1);
            return true;
        }
    }

    pUserProc->Using();

    bool isStatic = false;
    const CClass *pobj_c = NULL;
    const CMethod *pMethod = NULL;
    Type leftType;
    bool isFixedClass = false;
    if( pUserProc->GetParentClassPtr() ){
        //クラスのメンバ関数を呼び出す場合はアクセスチェックを行う
        if(ObjectName[0] && (dwFlags&PROCFLAG_NEW)==0)
        {
            if(lstrcmpi(ObjectName,"Super")==0)
            {
                //クラスメンバ関数内から基底クラスの呼び出し
                pobj_c=&compiler.pCompilingClass->GetSuperClass();

                isFixedClass = true;
            }
            else
            {
                //"->"によってオブジェクトを指定する通常のメンバ関数呼び出し
                Type varType;
                if( GetTermType( ObjectName, varType ) )
                {
                    if( varType.IsObject() )
                    {
                        pobj_c = &varType.GetClass();
                        leftType = varType;
                    }
                }

                if( !pobj_c )
                {
                    pobj_c=compiler.GetObjectModule().meta.GetClasses().Find(ObjectName);
                    if( pobj_c ){
                        isStatic = true;
                    }
                    else{
                        compiler.errorMessenger.Output(300,NULL,cp);
                    }
                }
            }
        }
        else{
            if(dwFlags&PROCFLAG_NEW){
                GetVarType( ObjectName, leftType, false );

                //New演算子によるコンストラクタ呼び出し
                pobj_c=pUserProc->GetParentClassPtr();
            }
            else{
                //クラスメンバ関数内から同一クラスのメンバ関数の呼び出し
                pobj_c=compiler.pCompilingClass;
            }
        }


        /////////////////////////////////
        // メソッド情報を取得
        /////////////////////////////////
        pMethod = NULL;
        if( ! isStatic ) pMethod = pobj_c->GetDynamicMethodOrInterfaceMethod( pUserProc );
        if( ! pMethod ){
            //動的メソッドが取得できなかったときは静的メソッドを当たる
            pMethod = pobj_c->GetStaticMethods().GetMethodPtr( pUserProc );
            if( !pMethod ){
                compiler.errorMessenger.Output(300,NULL,cp);
                return false;
            }

            //静的メンバ
            isStatic = true;
        }


        //////////////////////////////
        // アクセスエラーチェック
        //////////////////////////////

        if(ObjectName[0]){
            //外部からの呼び出し
            if(pobj_c==compiler.pCompilingClass){
                //同一クラスオブジェクトの場合はプライベートアクセスを容認する
                if( pMethod->IsNoneAccess() ){
                    compiler.errorMessenger.Output(109,pUserProc->GetName(),cp);
                    return false;
                }
            }
            else{
                if( pMethod->IsPrivate()
                    || pMethod->IsNoneAccess() ){
                    compiler.errorMessenger.Output(109,pUserProc->GetName(),cp);
                    return false;
                }
                if( !pMethod->GetUserProc().GetParentClass().IsEqualsOrSubClass( pobj_c ) && pMethod->IsProtected() ){
                    compiler.errorMessenger.Output(110,pUserProc->GetName(),cp);
                    return false;
                }
            }
        }
        else{
            //クラス内部からの呼び出し（継承によるACCESS_NONのみをエラーとする）
            if( pMethod->IsNoneAccess() ){
                compiler.errorMessenger.Output(109,pUserProc->GetName(),cp);
                return false;
            }
        }
    }


    ///////////////////////////////////////////////////////////////
    // _System_LocalThisのダミーをセット
    ///////////////////////////////////////////////////////////////

    char temporary[VN_SIZE]={0};
    if( pUserProc->GetParentClassPtr() && isStatic == false ){
        //_System_LocalThis（第一パラメータ）のダミーを作成
        lstrcpy(temporary,"0,");
    }
    if( pUserProc->ReturnType().IsStruct() ){
        // ※ByRef _System_ReturnValue パラメータのダミーをセット
        lstrcat(temporary,"0,");
    }

    if(Parameter[0]=='\0'&&temporary[0])
        temporary[lstrlen(temporary)-1]=0;
    else lstrcat(temporary,Parameter);


    //パラメータセット前のspオフセットを取得（Newの場合はここにThisポインタが格納されている）
    int this_sp_offset = pobj_sf->GetNowSp();


    ////////////////////////
    // パラメータをセット
    ////////////////////////

    //パラメータオブジェクトを生成
    ParamImpl *pobj_parameter=0;
    pobj_parameter=new ParamImpl(temporary);

    // デフォルト引数を適用
    pobj_parameter->ApplyDefaultParameters( pUserProc->RealParams() );

    // 型パラメータを適用
    pobj_parameter->SetLeftType( leftType );

    //エラーチェック
    if( !pobj_parameter->ErrorCheck(pUserProc->GetName(),pUserProc->RealParams(),pUserProc->GetSecondParmNum() ) ){
        //パラメータにエラーがあるときは処理を終える
        return false;
    }

    if(pUserProc->IsMacro()){
        //マクロ関数の場合は、パラメータ省略を考慮する
        pobj_parameter->MacroParameterSupport( pUserProc->RealParams() );
    }

    //スタックフレームに存在する既存のパラメータをバックアップ
    pobj_parameter->BackupParameter( (int)pUserProc->RealParams().size() );

    //一時オブジェクトを生成
    pobj_parameter->NewTempParameters( pUserProc->GetName(),pUserProc->RealParams(),pUserProc->GetRealSecondParmNum() );

    //レジスタ、スタックフレームにセット
    pobj_parameter->SetParameter(pUserProc->GetName(),pUserProc->RealParams(),pUserProc->GetRealSecondParmNum(), pUserProc );

    if( pUserProc->ReturnType().IsStruct() ){
        //////////////////////////////////////////////////////
        // 戻り値に構造体インスタンスを持つ場合
        // ※ByRef _System_ReturnValue パラメータをセット
        //////////////////////////////////////////////////////


        //////////////////////////////////////////////////////
        /////    レジスタ資源のバックアップ
        {   BACKUP_REGISTER_RESOURCE
        //////////////////////////////////////////////////////

            int object_size = pUserProc->ReturnType().GetClass().GetSize();

            //mov rcx,object_size
            compiler.codeGenerator.op_mov_RV(sizeof(_int64),REG_RCX,object_size);

            //call calloc
            extern const UserProc *pSub_calloc;
            compiler.codeGenerator.op_call(pSub_calloc);

            //mov r13,rax
            compiler.codeGenerator.op_mov_RR(REG_R13,REG_RAX);

        /////////////////////////////////////////////
        //////   レジスタ資源を復元
            RESTORE_REGISTER_RESOURCE
        }////////////////////////////////////////////

        if( pUserProc->GetParentClassPtr() && isStatic == false ){
            //mov rdx,r13
            compiler.codeGenerator.op_mov_RR(REG_RDX,REG_R13);
        }
        else{
            //mov rcx,r13
            compiler.codeGenerator.op_mov_RR(REG_RCX,REG_R13);
        }
    }


    if( pUserProc->GetParentClassPtr() && isStatic == false ){
        ///////////////////////////////
        // メンバ関数の場合
        // thisポインタをrcxで受け渡す
        ///////////////////////////////

        if(ObjectName[0] && (dwFlags&PROCFLAG_NEW)==0){
            if(lstrcmpi(ObjectName,"Super")==0) goto InClassMember;
            else{
                bool isLiteral, isNeedHeapFreeStructure = false;
                Type baseType( DEF_OBJECT, *pUserProc->GetParentClassPtr() ) , resultType;
                if( !TermOpe( ObjectName, baseType, resultType, isLiteral, isNeedHeapFreeStructure, NULL, false, !pMethod->IsConst() ) )
                {
                    return false;
                }
                if( !resultType.IsObject() )
                {
                    compiler.errorMessenger.OutputFatalError();
                }

                // 実態ポインタをraxにコピー
                compiler.codeGenerator.op_mov_RR( REG_RCX, REG_RAX );
            }
        }
        else{
InClassMember:
            if(dwFlags&PROCFLAG_NEW){
                //New演算子によるコンストラクタ呼び出しの場合

                //mov rcx,qword ptr[rsp+offset]     ※スタックフレームを利用
                pobj_sf->ref_offset_data(REG_RCX, this_sp_offset);
            }
            else{
                //自身のオブジェクトのThisポインタをrcxにコピー
                SetThisPtrToReg(REG_RCX);
            }
        }
    }

    if( pUserProc->IsVirtual() && !isFixedClass ){
        int vtblIndex;
        if( pobj_c->IsInterface() )
        {
            // インターフェイス メソッド呼び出し

            int offset_vtbl = compiler.GetObjectModule().meta.GetClasses().GetInterfaceInfoClassPtr()->GetMemberOffset( "__vtbl" );

            // vtblのポインタを取得
            //mov r11,qword ptr[rcx+offset_vtbl]
            compiler.codeGenerator.op_mov_RM(sizeof(_int64),REG_R11,REG_RCX,offset_vtbl,MOD_BASE_DISP8);

            int offset_this = compiler.GetObjectModule().meta.GetClasses().GetInterfaceInfoClassPtr()->GetMemberOffset( "__this" );

            // インターフェイスの場合は更に__thisを取得する
            //mov rcx,qword ptr[rcx+offset_this]
            compiler.codeGenerator.op_mov_RM(sizeof(_int64),REG_RCX,REG_RCX,offset_this,MOD_BASE_DISP8);

            int vtblMasterListIndex;
            pobj_c->GetVtblMasterListIndexAndVtblIndex( pUserProc, vtblMasterListIndex, vtblIndex );
            if( vtblMasterListIndex != 0 )
            {
                compiler.errorMessenger.OutputFatalError();
            }
        }
        else if( pobj_c->IsComInterface() )
        {
            // COMインターフェイス メソッド呼び出し

            //仮想関数（オブジェクトメソッド）呼び出し
            // pObj -> vtbl1 -> func1
            //               -> func2
            //               -> func3

            int vtblMasterListIndex;
            pobj_c->GetVtblMasterListIndexAndVtblIndex( pUserProc, vtblMasterListIndex, vtblIndex );

            // vtblのポインタを取得
            //mov r11,qword ptr[rcx]
            compiler.codeGenerator.op_mov_RM(sizeof(_int64),REG_R11,REG_RCX,0,MOD_BASE);        }
        else
        {
            //仮想関数（オブジェクトメソッド）呼び出し
            // pObj -> vtbl_master_list -> vtbl1 -> func1
            //                                   -> func2
            //                                   -> func3
            //                          -> vtbl2 -> func1
            //                                   -> func2
            //                                   -> func3

            int vtblMasterListIndex;
            pobj_c->GetVtblMasterListIndexAndVtblIndex( pUserProc, vtblMasterListIndex, vtblIndex );

            // vtblマスターリストのポインタを取得
            //mov r11,qword ptr[rcx+sizeof(com_vtbl)]
            compiler.codeGenerator.op_mov_RM(sizeof(_int64),REG_R11,REG_RCX,PTR_SIZE,MOD_BASE_DISP8);

            // vtblのポインタを取得
            //mov r11,dword ptr[r11+vtblMasterListIndex]
            compiler.codeGenerator.op_mov_RM( sizeof(_int64), REG_R11, REG_R11, vtblMasterListIndex*PTR_SIZE, MOD_BASE_DISP32 );
        }

        //call qword ptr[r11+func_index]
        if( vtblIndex * PTR_SIZE <= 0x7F ){
            compiler.codeGenerator.PutOld(
                (char)0x41,
                (char)0xFF,
                (char)0x53,
                (char)(vtblIndex*PTR_SIZE)
            );
        }
        else{
            compiler.codeGenerator.PutOld(
                (char)0x41,
                (char)0xFF,
                (char)0x93,
                (long)(vtblIndex*PTR_SIZE)
            );
        }
    }
    else{
        //通常呼び出し

        //call ProcAddr
        compiler.codeGenerator.op_call(pUserProc);
    }

    /* 64コンパイラでは不要
    if(pDllProc->bCdecl){
        //add esp,ParmSize
    }*/


    //レジスタのブロッキングを解除        ※パラメータセット時にロックされたレジスタ
    pobj_BlockReg->clear();

    //一時オブジェクトを破棄
    pobj_parameter->DeleteTempParameters();

    //スタックフレームに存在する既存のパラメータを復元
    pobj_parameter->RestoreParameter( (int)pUserProc->RealParams().size() );

    //パラメータオブジェクトを破棄
    delete pobj_parameter;

    return true;
}

bool Opcode_CallDllProc( const char *lpszParms, DllProc *pDllProc ){

    extern BOOL bDebugSupportProc;
    if( compiler.IsDebug() && bDebugSupportProc==0 && pDllProc->IsEqualSymbol( LexicalAnalyzer::FullNameToSymbol( "DebugBreak" ) ) )
    {
        Call_DebugSys_SaveContext();
    }


    ////////////////////////
    // パラメータのセット
    ////////////////////////

    //パラメータオブジェクトを生成
    ParamImpl *pobj_parameter=0;
    pobj_parameter=new ParamImpl(lpszParms);

    // デフォルト引数を適用
    pobj_parameter->ApplyDefaultParameters( pDllProc->Params() );

    //エラーチェック
    if( !pobj_parameter->ErrorCheck( pDllProc->GetName(), pDllProc->Params() ) ){
        //パラメータにエラーがあるときは処理を終える
        return false;
    }

    //スタックフレームに存在する既存のパラメータをバックアップ
    pobj_parameter->BackupParameter( (int)pDllProc->Params().size() );

    //一時オブジェクトを生成
    pobj_parameter->NewTempParameters( pDllProc->GetName(), pDllProc->Params() );

    //レジスタ、スタックフレームにセット
    pobj_parameter->SetParameter(pDllProc->GetName(), pDllProc->Params() );


    //レジスタのブロッキングを解除        ※パラメータセット時にロックされたレジスタ
    pobj_BlockReg->clear();


    //動的リンクされたプロシージャの呼び出し

    //call dword ptr[ImportTable]
    compiler.codeGenerator.op_call( pDllProc );

    /* 64コンパイラでは不要
    if(pDllProc->bCdecl){
        //add esp,ParmSize
    }*/

    //一時オブジェクトを破棄
    pobj_parameter->DeleteTempParameters();

    //スタックフレームに存在する既存のパラメータを復元
    pobj_parameter->RestoreParameter( (int)pDllProc->Params().size() );

    //パラメータオブジェクトを破棄
    delete pobj_parameter;

    return true;
}

void Opcode_CallDelegate( const Delegate &dg, const char *methodPtrValueStr, const char *objPtrValueStr, const char *params )
{
    extern BOOL bDebugSupportProc;
    if( compiler.IsDebug() && bDebugSupportProc == 0 )
    {
        Call_DebugSys_SaveContext();
    }


    ///////////////////////////////////////////////////////////////
    // _System_LocalThisのダミーをセット
    ///////////////////////////////////////////////////////////////

    char temporary[VN_SIZE]={0};
    bool isDynamicCall = false;
    if( objPtrValueStr && objPtrValueStr[0] ){
        //_System_LocalThis（第一パラメータ）のダミーを作成
        lstrcpy(temporary,"0,");

        isDynamicCall = true;
    }
    if( dg.ReturnType().IsStruct() ){
        // ※ByRef _System_ReturnValue パラメータのダミーをセット
        lstrcat(temporary,"0,");
    }

    if(params[0]=='\0'&&temporary[0])
        temporary[lstrlen(temporary)-1]=0;
    else lstrcat(temporary,params);

    const Parameters *pParams = &dg.Params();
    if( isDynamicCall )
    {
        pParams = &dg.GetDynamicParams();
    }


    ParamImpl *pobj_parameter = new ParamImpl( temporary );

    //スタックフレームに存在する既存のパラメータをバックアップ
    pobj_parameter->BackupParameter( (int)pParams->size() );

    //一時オブジェクトを生成
    pobj_parameter->NewTempParameters( dg.GetName(), *pParams );

    //レジスタ、スタックフレームにセット
    pobj_parameter->SetParameter( dg.GetName(), *pParams );


    if( objPtrValueStr && objPtrValueStr[0] )
    {
        RELATIVE_VAR RelativeVar;
        //Constアクセスが不可能なメソッドの場合
        if( !GetVarOffsetReadWrite( objPtrValueStr, &RelativeVar, Type() ) ){
            Jenga::Throw( "Opcode_CallDelegate関数内で呼ばれるGetVarOffsetReadWrite関数に失敗" );
            return;
        }

        SetVarPtrToReg(REG_RCX,&RelativeVar);

        // 参照を実体ポインタにする
        //mov rcx,qword ptr[rcx]
        compiler.codeGenerator.op_mov_RM(sizeof(_int64),REG_RCX,REG_RCX,0,MOD_BASE);
    }


    {
        ////////////////////////
        // call
        ////////////////////////
        RELATIVE_VAR RelativeVar;
        GetVarOffsetReadOnly( methodPtrValueStr, &RelativeVar, Type() );
        SetVarPtrToReg(REG_RAX,&RelativeVar);

        //mov rax,qword ptr[rax]
        compiler.codeGenerator.op_mov_RM(sizeof(_int64),REG_RAX,REG_RAX,0,MOD_BASE);

        //call rax
        compiler.codeGenerator.PutOld(
            (char)0xFF,
            (char)0xD0
        );
    }


    //レジスタのブロッキングを解除        ※パラメータセット時にロックされたレジスタ
    pobj_BlockReg->clear();

    //一時オブジェクトを破棄
    pobj_parameter->DeleteTempParameters();

    //スタックフレームに存在する既存のパラメータを復元
    pobj_parameter->RestoreParameter( (int)pParams->size() );

    //パラメータオブジェクトを破棄
    delete pobj_parameter;
}