source: dev/trunk/abdev/BasicCompiler32/Compile_CallProc.cpp@ 406

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

    //push eax
    compiler.codeGenerator.op_push(REG_EAX);

    //push ebp
    compiler.codeGenerator.op_push(REG_EBP);

    //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->NewTempParameters( variable,pProcPointer->Params() );

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



    ////////////////////////
    // call
    ////////////////////////
    RELATIVE_VAR RelativeVar;
    GetVarOffsetReadOnly(variable,&RelativeVar,Type());
    SetVarPtrToEax(&RelativeVar);

    //mov eax,dword ptr[eax]
    compiler.codeGenerator.op_mov_RM( sizeof(long), REG_EAX, REG_EAX, 0, MOD_BASE );

    //call eax
    compiler.codeGenerator.op_call_R( REG_EAX );



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

    //パラメータオブジェクトを破棄
    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;
                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){
                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 ){
                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->GetUserProc().GetParentClass().IsEqualsOrSubClass( pobj_c ) && 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( pUserProc->ReturnType().IsStruct() ){
        // ※ByRef _System_ReturnValue パラメータのダミーをセット
        lstrcat(temporary,"0,");
    }

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


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

    //パラメータオブジェクトを生成
    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() );
    }

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

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

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

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

        //push object_size
        compiler.codeGenerator.op_push_V(object_size);

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

        //push eax
        compiler.codeGenerator.op_push(REG_EAX);

        ParmSize += PTR_SIZE;
    }


    if( pUserProc->GetParentClassPtr() && isStatic == false ){
        //////////////////////////////////////////////////////
        // メンバ関数の場合
        // ※_System_LocalThis パラメータをecxにセット
        //////////////////////////////////////////////////////

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

                SetVarPtrToEax(&RelativeVar);

                // 参照を実体ポインタにする
                compiler.codeGenerator.op_mov_RM( sizeof(long), REG_ECX, REG_EAX, 0, MOD_BASE );
            }
        }
        else{
InClassMember:
            if(dwFlags&PROCFLAG_NEW){
                //New演算子によるコンストラクタ呼び出しの場合

                //mov ecx,dword ptr[esp+ParmSize]
                compiler.codeGenerator.op_mov_RM( sizeof(long), REG_ECX, REG_ESP, ParmSize + tempSize, MOD_BASE_DISP32 );
            }
            else{
                //Thisポインタをecxにコピー
                SetThisPtrToReg(REG_ECX);
            }
        }
    }

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

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


            // vtblのポインタを取得
            //mov edx,dword ptr[ecx+offset_vtbl]
            compiler.codeGenerator.op_mov_RM( sizeof(long), REG_EDX, REG_ECX, offset_vtbl, MOD_BASE_DISP8 );

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



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

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

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

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

            // vtblのポインタを取得
            //mov edx,dword ptr[ecx]
            compiler.codeGenerator.op_mov_RM( sizeof(long), REG_EDX, REG_ECX, 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 edx,dword ptr[ecx+sizeof(com_vtbl)]
            compiler.codeGenerator.op_mov_RM( sizeof(long), REG_EDX, REG_ECX, PTR_SIZE, MOD_BASE_DISP8 );
            
            // vtblのポインタを取得
            //mov edx,dword ptr[edx+vtblMasterListIndex]
            compiler.codeGenerator.op_mov_RM( sizeof(long), REG_EDX, REG_EDX, vtblMasterListIndex*PTR_SIZE, MOD_BASE_DISP32 );
        }

        //push ecx
        compiler.codeGenerator.op_push(REG_ECX);    

        //call dword ptr[edx+func_index]
        if( vtblIndex * PTR_SIZE <= 0x7F )
        {
            compiler.codeGenerator.PutOld(
                (char)0xFF,
                (char)0x52,
                (char)(vtblIndex*PTR_SIZE)
            );
        }
        else{
            compiler.codeGenerator.PutOld(
                (char)0xFF,
                (char)0x92
            );
            compiler.codeGenerator.PutOld( (long)(vtblIndex*PTR_SIZE), Schedule::None );
        }
    }
    else{
        //通常呼び出し

        if( pUserProc->GetParentClassPtr() && isStatic == false )
        {
            //push ecx
            compiler.codeGenerator.op_push(REG_ECX);
        }

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

    if(pUserProc->IsCdecl()){
        //add esp,ParmSize
        compiler.codeGenerator.op_add_esp(ParmSize);
    }

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

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

    return true;
}

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

    extern BOOL bDebugCompile;
    extern BOOL bDebugSupportProc;
    if(bDebugCompile&&bDebugSupportProc==0&& pDllProc->IsEqualSymbol( "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->NewTempParameters( pDllProc->GetName(), pDllProc->Params() );

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


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

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

    if(pDllProc->IsCdecl()){
        //add esp,ParmSize
        compiler.codeGenerator.op_add_esp(ParmSize);
    }

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

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

    return true;
}

void Opcode_CallDelegate( const Delegate &dg, const char *methodPtrValueStr, const char *objPtrValueStr, const char *params )
{
    extern BOOL bDebugCompile;
    extern BOOL bDebugSupportProc;
    if(bDebugCompile&&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->NewTempParameters( dg.GetName(), *pParams );

    //レジスタ、スタックフレームにセット
    int ParmSize = 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;
        }

        SetVarPtrToEax(&RelativeVar);

        // 参照を実体ポインタにする
        compiler.codeGenerator.op_mov_RM( sizeof(long), REG_ECX, REG_EAX, 0, MOD_BASE );

        //push ecx
        compiler.codeGenerator.op_push(REG_ECX);
    }


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

        //mov eax,dword ptr[eax]
        compiler.codeGenerator.op_mov_RM( sizeof(long), REG_EAX, REG_EAX, 0, MOD_BASE );

        //call eax
        compiler.codeGenerator.op_call_R( REG_EAX );
    }


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

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