#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.GetCompilingClass().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.FindClassSupportedTypeDef(
						LexicalAnalyzer::FullNameToSymbol( 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.GetCompilingClass();
			}
		}


		/////////////////////////////////
		// メソッド情報を取得
		/////////////////////////////////
		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( compiler.IsCompilingClass() && pobj_c == &compiler.GetCompilingClass() )
			{
				//同一クラスオブジェクトの場合はプライベートアクセスを容認する
				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;
}