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

//ローカル変数アドレススケジュール
DWORD *pLocalVarAddrSchedule;
int LocalVarAddrScheduleNum;

void Call_DebugSys_SaveContext(){
	//call _System_GetEip
	extern SUBINFO *pSub_System_GetEip;
	op_call(pSub_System_GetEip);

	//push eax
	op_push(REG_EAX);

	//push ebp
	op_push(REG_EBP);

	//call _DebugSys_SaveContext
	extern SUBINFO *pSub_DebugSys_SaveContext;
	op_call(pSub_DebugSys_SaveContext);
}

void AddLocalVarAddrSchedule(){
	extern HANDLE hHeap;

	//ローカル変数アドレススケジュールに追加する
	pLocalVarAddrSchedule=(DWORD *)HeapReAlloc(hHeap,0,pLocalVarAddrSchedule,(LocalVarAddrScheduleNum+1)*sizeof(DWORD));
	pLocalVarAddrSchedule[LocalVarAddrScheduleNum]=obp;
	LocalVarAddrScheduleNum++;
}


int Opcode_CallProcPtr(char *variable,char *Parameter,PROCPTRINFO *pi,LONG_PTR *plpIndex){
	extern HANDLE hHeap;
	int i;


	extern BOOL bDebugCompile;
	extern BOOL bDebugSupportProc;
	if(bDebugCompile&&bDebugSupportProc==0)
		Call_DebugSys_SaveContext();


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

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

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

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

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



	////////////////////////
	// call
	////////////////////////
	RELATIVE_VAR RelativeVar;
	LONG_PTR lp;
	GetVarOffsetReadOnly(variable,&i,&RelativeVar,&lp);
	SetVarPtrToEax(&RelativeVar);

	//mov eax,dword ptr[eax]
	OpBuffer[obp++]=(char)0x8B;
	OpBuffer[obp++]=(char)0x00;

	//call eax
	OpBuffer[obp++]=(char)0xFF;
	OpBuffer[obp++]=(char)0xD0;



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

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

	if(plpIndex) *plpIndex=pi->u.ReturnIndex;

	return pi->ReturnType;
}

void Opcode_CallProc(const char *Parameter,SUBINFO *psi,DWORD dwFlags,char *ObjectName,int RefType){
	int i2;

	if(psi->dwType==SUBTYPE_MACRO){
		if(lstrcmpi(psi->name,"Print")==0){
			Opcode_Print(Parameter,0);
			return;
		}
		if(lstrcmpi(psi->name,"Input")==0){
			Opcode_Input(Parameter);
			return;
		}
		if(lstrcmpi(psi->name,"Write")==0){
			Opcode_Print(Parameter,1);
			return;
		}
	}

	psi->bUse=1;

	bool isStatic = false;
	CClass *pobj_c = NULL;
	CMethod *pMethod = NULL;
	if(psi->pobj_ParentClass){
		//クラスのメンバ関数を呼び出す場合はアクセスチェックを行う
		if(ObjectName[0]){
			if(lstrcmpi(ObjectName,"Super")==0){
				//クラスメンバ関数内から基底クラスの呼び出し
				pobj_c=pobj_CompilingClass;
			}
			else{
				//"->"によってオブジェクトを指定する通常のメンバ関数呼び出し
				int type = GetVarType(ObjectName,(LONG_PTR *)&pobj_c,0);
				if(!(NATURAL_TYPE(type)==DEF_OBJECT)){
					pobj_c=pobj_DBClass->check(ObjectName);
					if( pobj_c ){
						isStatic = true;
					}
					else{
						SetError(300,NULL,cp);
					}
				}
			}
		}
		else{
			if(dwFlags&PROCFLAG_NEW){
				//New演算子によるコンストラクタ呼び出し
				pobj_c=psi->pobj_ParentClass;
			}
			else{
				//クラスメンバ関数内から同一クラスのメンバ関数の呼び出し
				pobj_c=pobj_CompilingClass;
			}
		}


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

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


		//////////////////////////////
		// アクセスエラーチェック
		//////////////////////////////
		DWORD dwAccess = pMethod->dwAccess;

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



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

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

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



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

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

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

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

	//一時オブジェクトを生成
	int tempSize = pobj_parameter->NewTempParameters( psi->name,psi->pRealParmInfo,psi->RealParmNum,psi->RealSecondParmNum );

	//レジスタ、スタックフレームにセット
	int ParmSize;
	ParmSize=pobj_parameter->SetParameter(psi->name,psi->pRealParmInfo,psi->RealParmNum,psi->RealSecondParmNum);

	if(psi->ReturnType==DEF_STRUCT && psi->isReturnRef == false){
		//////////////////////////////////////////////////////
		// 戻り値に構造体インスタンスを持つ場合
		// ※ByRef _System_ReturnValue パラメータをセット
		//////////////////////////////////////////////////////

		int object_size = psi->u.Return_pobj_c->GetSize();

		//push object_size
		op_push_value(object_size);

		//call calloc
		extern SUBINFO *pSub_calloc;
		op_call(pSub_calloc);

		//push eax
		op_push(REG_EAX);
	}


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

		if(ObjectName[0]){
			if(lstrcmpi(ObjectName,"Super")==0) goto InClassMember;
			else{
				RELATIVE_VAR RelativeVar;
				if( pMethod->isConst ){
					//Constアクセスが可能なメソッドの場合
					if( !GetVarOffsetReadOnly( ObjectName, &i2, &RelativeVar, 0 ) ) return;
				}
				else{
					//Constアクセスが不可能なメソッドの場合
					if( !GetVarOffsetReadWrite( ObjectName, &i2, &RelativeVar, 0 ) ) return;
				}

				SetVarPtrToEax(&RelativeVar);

				// 参照を実体ポインタにする
				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]
				op_mov_RM( sizeof(long), REG_ECX, REG_ESP, ParmSize + tempSize, MOD_BASE_DISP32 );
			}
			else{
				//Thisポインタをecxにコピー
				SetThisPtrToReg(REG_ECX);
			}
		}

		//push ecx
		op_push(REG_ECX);
	}

	if(psi->bVirtual){
		//仮想関数（オブジェクトメソッド）呼び出し
		//pObj->func_table->func1
		//                ->func2
		//                ->func3

		//mov edx,dword ptr[ecx]
		OpBuffer[obp++]=(char)0x8B;
		OpBuffer[obp++]=(char)0x11;

		i2 = pobj_c->GetFuncNumInVtbl( psi );

		//call dword ptr[edx+func_index]
		if(i2*PTR_SIZE<=0x7F){
			OpBuffer[obp++]=(char)0xFF;
			OpBuffer[obp++]=(char)0x52;
			OpBuffer[obp++]=(char)(i2*PTR_SIZE);
		}
		else{
			OpBuffer[obp++]=(char)0xFF;
			OpBuffer[obp++]=(char)0x92;
			*((long *)(OpBuffer+obp))=i2*PTR_SIZE;
			obp+=sizeof(long);
		}
	}
	else{
		//通常呼び出し

		//call ProcAddr
		op_call(psi);
	}

	if(psi->bCdecl){
		//add esp,ParmSize
		op_add_esp(ParmSize);
	}

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

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

int Opcode_CallDllProc(char *Parameter,DECLAREINFO *pdi,LONG_PTR *plpIndex){
	char *temporary;

	temporary=(char *)HeapAlloc(hHeap,0,lstrlen(Parameter)+1);

	extern BOOL bDebugCompile;
	extern BOOL bDebugSupportProc;
	if(bDebugCompile&&bDebugSupportProc==0&&lstrcmp(pdi->name,"DebugBreak")!=0)
		Call_DebugSys_SaveContext();

	pdi->bUse=1;


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

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

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

	//一時オブジェクトを生成
	pobj_parameter->NewTempParameters( pdi->name,pdi->pParmInfo,pdi->ParmNum,pdi->ParmNum );

	//レジスタ、スタックフレームにセット
	int ParmSize;
	ParmSize=pobj_parameter->SetParameter(pdi->name,pdi->pParmInfo,pdi->ParmNum,pdi->ParmNum);


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

	//call dword ptr[LookupTable]
	OpBuffer[obp++]=(char)0xFF;
	OpBuffer[obp++]=(char)0x15;
	pobj_ImportAddrSchedule->add(pdi);
	obp+=sizeof(long);

	if(pdi->bCdecl){
		//add esp,ParmSize
		op_add_esp(ParmSize);
	}

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

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

	if(plpIndex) *plpIndex=pdi->u.ReturnIndex;

	HeapDefaultFree(temporary);

	return pdi->ReturnType;
}