#include "lib.h"

// Private Prototypes
int SetRsaKeySet(u8 **PrivDest, u8 *PrivSource, u8 **PubDest, u8 *PubSource);
int SetUnFixedKey(keys_struct *keys, u8 *UnFixedKey);

int SetTIK_RsaKey(keys_struct *keys, u8 *PrivateExp, u8 *PublicMod);
int SetTMD_RsaKey(keys_struct *keys, u8 *PrivateExp, u8 *PublicMod);
int SetCFA_RsaKey(keys_struct *keys, u8 *PrivateExp, u8 *PublicMod);
int SetCCI_RsaKey(keys_struct *keys, u8 *PrivateExp, u8 *PublicMod);
int SetAccessDesc_RsaKey(keys_struct *keys, u8 *PrivateExp, u8 *PublicMod);
int SetCXI_RsaKey(keys_struct *keys, u8 *PrivateExp, u8 *PublicMod);

int SetCaCert(keys_struct *keys, u8 *Cert);
int SetTikCert(keys_struct *keys, u8 *Cert);
int SetTmdCert(keys_struct *keys, u8 *Cert);


// Code
void InitKeys(keys_struct *keys)
{
	memset(keys,0,sizeof(keys_struct));
	
#ifdef RETAIL_FSIGN
	/* AES Keys */
	// CIA
	SetCommonKey(keys,(u8*)zeros_fixed_aesKey,1);
	SetCurrentCommonKey(keys,1);
	
	// NCCH
	keys->aes.NormalKey = (u8*)zeros_fixed_aesKey;
	SetSystemFixedKey(keys,(u8*)zeros_fixed_aesKey);

	/* RSA Keys */
	// CIA
	SetTIK_RsaKey(keys,(u8*)Dummy_rsa_privExp,(u8*)Dummy_rsa_pubMod);
	SetTMD_RsaKey(keys,(u8*)Dummy_rsa_privExp,(u8*)Dummy_rsa_pubMod);
	// CFA
	SetCFA_RsaKey(keys,(u8*)Dummy_rsa_privExp,(u8*)Dummy_rsa_pubMod);
	// CCI
	SetCCI_RsaKey(keys,(u8*)Dummy_rsa_privExp,(u8*)Dummy_rsa_pubMod);
	// CXI
	SetAccessDesc_RsaKey(keys,(u8*)Dummy_rsa_privExp,(u8*)Dummy_rsa_pubMod);
	
	/* Certs */
	SetCaCert(keys,(u8*)ca3_dpki_cert);
	SetTikCert(keys,(u8*)xsC_dpki_cert);
	SetTmdCert(keys,(u8*)cpB_dpki_cert);
#else // DEBUG KEYS
	/* AES Keys */
	// CIA
	SetCommonKey(keys,(u8*)ctr_aes_common_key_dev0,0);
	SetCommonKey(keys,(u8*)ctr_aes_common_key_dev1,1);
	SetCurrentCommonKey(keys,0);
	
	// NCCH
	keys->aes.NormalKey = (u8*)zeros_fixed_aesKey;
	SetSystemFixedKey(keys,(u8*)system_fixed_aesKey);

	/* RSA Keys */
	// CIA
	SetTIK_RsaKey(keys,(u8*)xs9_dpki_rsa_privExp,(u8*)xs9_dpki_rsa_pubMod);
	SetTMD_RsaKey(keys,(u8*)cpA_dpki_rsa_privExp,(u8*)cpA_dpki_rsa_pubMod);
	// CFA
	SetCFA_RsaKey(keys,(u8*)DevNcsdCfa_privExp,(u8*)DevNcsdCfa_pubMod);
	// CCI
	SetCCI_RsaKey(keys,(u8*)DevNcsdCfa_privExp,(u8*)DevNcsdCfa_pubMod);
	// CXI
	SetAccessDesc_RsaKey(keys,(u8*)AccessDesc_privExp,(u8*)AccessDesc_pubMod);
	
	/* Certs */
	SetCaCert(keys,(u8*)ca4_dpki_cert);
	SetTikCert(keys,(u8*)xs9_dpki_cert);
	SetTmdCert(keys,(u8*)cpA_dpki_cert);
#endif
}

void FreeKeys(keys_struct *keys)
{
	// AES
	if(keys->aes.CommonKey){
		for(int i = 0; i < 256; i++){
			free(keys->aes.CommonKey[i]);
		}
	}
	free(keys->aes.CommonKey);
	free(keys->aes.SystemFixedKey);
	free(keys->aes.UnFixedKey);
	
	// RSA
	free(keys->rsa.TIK_Priv);
	free(keys->rsa.TIK_Pub);
	free(keys->rsa.TMD_Priv);
	free(keys->rsa.TMD_Pub);

	free(keys->rsa.CFA_Priv);
	free(keys->rsa.CFA_Pub);
	
	free(keys->rsa.CCI_Priv);
	free(keys->rsa.CCI_Pub);
	
	free(keys->rsa.AccessDesc_Priv);
	free(keys->rsa.AccessDesc_Pub);
	
	// Certs
	free(keys->certs.ca_cert);
	free(keys->certs.tik_cert);
	free(keys->certs.tmd_cert);
	memset(keys,0,sizeof(keys_struct));
}

int SetRsaKeySet(u8 **PrivDest, u8 *PrivSource, u8 **PubDest, u8 *PubSource)
{
	int result = 0;
	if(PrivSource){
		result = CopyData(PrivDest,PrivSource,0x100);
		if(result) return result;
	}
	if(PubSource){
		result = CopyData(PubDest,PubSource,0x100);
		if(result) return result;
	}
	return 0;
}

int SetCommonKey(keys_struct *keys, u8 *CommonKey, u8 Index)
{
	if(!keys) return -1;
	if(!keys->aes.CommonKey){
		keys->aes.CommonKey = malloc(sizeof(u8*)*256);
		memset(keys->aes.CommonKey,0,sizeof(u8*)*256);
	}
	return CopyData(&keys->aes.CommonKey[Index],CommonKey,16);
}

int SetCurrentCommonKey(keys_struct *keys, u8 Index)
{
	if(!keys) return -1;
	keys->aes.CurrentCommonKey = Index;
	return 0;
}

int SetSystemFixedKey(keys_struct *keys, u8 *SystemFixedKey)
{
	if(!keys) return -1;
	return CopyData(&keys->aes.SystemFixedKey,SystemFixedKey,16);
}

int SetUnFixedKey(keys_struct *keys, u8 *UnFixedKey)
{
	if(!keys) return -1;
	return CopyData(&keys->aes.UnFixedKey,UnFixedKey,16);
}

int SetTIK_RsaKey(keys_struct *keys, u8 *PrivateExp, u8 *PublicMod)
{
	if(!keys) return -1;
	return SetRsaKeySet(&keys->rsa.TIK_Priv,PrivateExp,&keys->rsa.TIK_Pub,PublicMod);
}

int SetTMD_RsaKey(keys_struct *keys, u8 *PrivateExp, u8 *PublicMod)
{
	if(!keys) return -1;
	return SetRsaKeySet(&keys->rsa.TMD_Priv,PrivateExp,&keys->rsa.TMD_Pub,PublicMod);
}

int SetCFA_RsaKey(keys_struct *keys, u8 *PrivateExp, u8 *PublicMod)
{
	if(!keys) return -1;
	return SetRsaKeySet(&keys->rsa.CFA_Priv,PrivateExp,&keys->rsa.CFA_Pub,PublicMod);
}

int SetCCI_RsaKey(keys_struct *keys, u8 *PrivateExp, u8 *PublicMod)
{
	if(!keys) return -1;
	return SetRsaKeySet(&keys->rsa.CCI_Priv,PrivateExp,&keys->rsa.CCI_Pub,PublicMod);
}

int SetAccessDesc_RsaKey(keys_struct *keys, u8 *PrivateExp, u8 *PublicMod)
{
	if(!keys) return -1;
	return SetRsaKeySet(&keys->rsa.AccessDesc_Priv,PrivateExp,&keys->rsa.AccessDesc_Pub,PublicMod);
}

int SetCaCert(keys_struct *keys, u8 *Cert)
{
	if(!keys) return -1;
	return CopyData(&keys->certs.ca_cert,Cert,0x400);
}
int SetTikCert(keys_struct *keys, u8 *Cert)
{
	if(!keys) return -1;
	return CopyData(&keys->certs.tik_cert,Cert,0x300);
}

int SetTmdCert(keys_struct *keys, u8 *Cert)
{
	if(!keys) return -1;
	return CopyData(&keys->certs.tmd_cert,Cert,0x400);
}