source: dev/trunk/abdev/BasicCompiler64/Compile_CallProc.cpp@ 308

#include "stdafx.h"

#include <jenga/include/smoothie/Smoothie.h>

#include <Compiler.h>

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

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 bDebugCompile;
    extern BOOL bDebugSupportProc;
    if(bDebugCompile&&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){
    int i2;

    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;
                GetVarType( ObjectName, varType, false );
                if( NATURAL_TYPE( varType.GetBasicType() ) == DEF_OBJECT )
                {
                    pobj_c = &varType.GetClass();
                    leftType = varType;
                }
                else
                {
                    pobj_c=compiler.GetObjectModule().meta.GetClasses().Find(ObjectName);
                    if( pobj_c ){
                        isStatic = true;
                    }
                    else{
                        SetError(300,NULL,cp);
                    }
                }
            }
        }
        else{
            if(dwFlags&PROCFLAG_NEW){
                //New演算子によるコンストラクタ呼び出し
                pobj_c=pUserProc->GetParentClassPtr();
            }
            else{
                //クラスメンバ関数内から同一クラスのメンバ関数の呼び出し
                pobj_c=compiler.pCompilingClass;
            }
        }


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

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


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

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


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

    char temporary[VN_SIZE]={0};
    if( pUserProc->GetParentClassPtr() && isStatic == false ){
        //_System_LocalThis（第一パラメータ）のダミーを作成
        lstrcpy(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() );

    //エラーチェック
    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() );

    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{
                RELATIVE_VAR RelativeVar;
                if( pMethod->IsConst() ){
                    //Constアクセスが可能なメソッドの場合
                    if( !GetVarOffsetReadOnly( ObjectName, &RelativeVar, Type() ) ){
                        return false;
                    }
                }
                else{
                    //Constアクセスが不可能なメソッドの場合
                    if( !GetVarOffsetReadWrite( ObjectName, &RelativeVar, Type() ) ){
                        return false;
                    }
                }

                SetVarPtrToReg(REG_RCX,&RelativeVar);

                // 参照を実体ポインタにする
                //mov rcx,qword ptr[rcx]
                compiler.codeGenerator.op_mov_RM(sizeof(_int64),REG_RCX,REG_RCX,0,MOD_BASE);
            }
        }
        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 ){
        //仮想関数（オブジェクトメソッド）呼び出し
        //pObj->func_table->func1
        //                ->func2
        //                ->func3

        //mov r11,qword ptr[rcx]
        compiler.codeGenerator.op_mov_RM(sizeof(_int64),REG_R11,REG_RCX,0,MOD_BASE);

        i2 = pobj_c->GetFuncNumInVtbl( pUserProc );

        //call qword ptr[r11+func_index]
        if(i2*PTR_SIZE<=0x7F){
            compiler.codeGenerator.PutOld(
                (char)0x41,
                (char)0xFF,
                (char)0x53,
                (char)(i2*PTR_SIZE)
            );
        }
        else{
            compiler.codeGenerator.PutOld(
                (char)0x41,
                (char)0xFF,
                (char)0x93,
                (long)(i2*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 bDebugCompile;
    extern BOOL bDebugSupportProc;
    if(bDebugCompile&&bDebugSupportProc==0&& pDllProc->GetName() != "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;
}