// Copyright 2013 Dolphin Emulator Project
// Licensed under GPLv2
// Refer to the license.txt file included.

#include "Common.h"
#include "CommonPaths.h"
#include "Timer.h"

#include "EXI_DeviceIPL.h"
#include "../Core.h"
#include "../ConfigManager.h"
#include "MemoryUtil.h"
#include "FileUtil.h"
#include "../Movie.h"
#include "../CoreTiming.h"
#include "SystemTimers.h"

// We should provide an option to choose from the above, or figure out the checksum (the algo in yagcd seems wrong)
// so that people can change default language.

static const char iplverPAL[0x100] = "(C) 1999-2001 Nintendo.  All rights reserved."
									 "(C) 1999 ArtX Inc.  All rights reserved."
									 "PAL  Revision 1.0  ";

static const char iplverNTSC[0x100]= "(C) 1999-2001 Nintendo.  All rights reserved."
									 "(C) 1999 ArtX Inc.  All rights reserved.";

// bootrom descrambler reversed by segher
// Copyright 2008 Segher Boessenkool <segher@kernel.crashing.org>
void CEXIIPL::Descrambler(u8* data, u32 size)
{
	u8 acc = 0;
	u8 nacc = 0;

	u16 t = 0x2953;
	u16 u = 0xd9c2;
	u16 v = 0x3ff1;

	u8 x = 1;

	for (u32 it = 0; it < size;)
	{
		int t0 = t & 1;
		int t1 = (t >> 1) & 1;
		int u0 = u & 1;
		int u1 = (u >> 1) & 1;
		int v0 = v & 1;

		x ^= t1 ^ v0;
		x ^= (u0 | u1);
		x ^= (t0 ^ u1 ^ v0) & (t0 ^ u0);

		if (t0 == u0)
		{
			v >>= 1;
			if (v0)
				v ^= 0xb3d0;
		}

		if (t0 == 0)
		{
			u >>= 1;
			if (u0)
				u ^= 0xfb10;
		}

		t >>= 1;
		if (t0)
			t ^= 0xa740;

		nacc++;
		acc = 2*acc + x;
		if (nacc == 8)
		{
			data[it++] ^= acc;
			nacc = 0;
		}
	}
}

CEXIIPL::CEXIIPL() :
	m_uPosition(0),
	m_uAddress(0),
	m_uRWOffset(0),
	m_count(0),
	m_FontsLoaded(false)
{
	memset(m_szBuffer,0,sizeof(m_szBuffer));

	// Determine region
	m_bNTSC = SConfig::GetInstance().m_LocalCoreStartupParameter.bNTSC;

	// Create the IPL
	m_pIPL = (u8*)AllocateMemoryPages(ROM_SIZE);
	
	if (SConfig::GetInstance().m_LocalCoreStartupParameter.bHLE_BS2)
	{
		// Copy header
		memcpy(m_pIPL, m_bNTSC ? iplverNTSC : iplverPAL, m_bNTSC ? sizeof(iplverNTSC) : sizeof(iplverPAL));

		// Load fonts
		LoadFileToIPL((File::GetSysDirectory() + GC_SYS_DIR + DIR_SEP + FONT_SJIS), 0x1aff00);
		LoadFileToIPL((File::GetSysDirectory() + GC_SYS_DIR + DIR_SEP + FONT_ANSI), 0x1fcf00);
	}
	else
	{
		// Load whole ROM dump
		LoadFileToIPL(SConfig::GetInstance().m_LocalCoreStartupParameter.m_strBootROM, 0);
		// Descramble the encrypted section (contains BS1 and BS2)
		Descrambler(m_pIPL + 0x100, 0x1aff00);
		INFO_LOG(BOOT, "Loaded bootrom: %s", m_pIPL); // yay for null-terminated strings ;p
	}

	// Clear RTC
	memset(m_RTC, 0, sizeof(m_RTC));


	// We Overwrite language selection here since it's possible on the GC to change the language as you please
	g_SRAM.lang = SConfig::GetInstance().m_LocalCoreStartupParameter.SelectedLanguage;

	WriteProtectMemory(m_pIPL, ROM_SIZE);
	m_uAddress = 0;		
}

CEXIIPL::~CEXIIPL()
{
	if (m_count > 0)
	{
		m_szBuffer[m_count] = 0x00;
	}

	if (m_pIPL != NULL)
	{
		FreeMemoryPages(m_pIPL, ROM_SIZE);
		m_pIPL = NULL;
	}

	// SRAM
	File::IOFile file(SConfig::GetInstance().m_LocalCoreStartupParameter.m_strSRAM, "wb");
	file.WriteArray(&g_SRAM, 1);
}
void CEXIIPL::DoState(PointerWrap &p)
{
	p.Do(m_RTC);
	p.Do(m_uPosition);
	p.Do(m_uAddress);
	p.Do(m_uRWOffset);
	p.Do(m_szBuffer);
	p.Do(m_count);
	p.Do(m_FontsLoaded);
}

void CEXIIPL::LoadFileToIPL(std::string filename, u32 offset)
{
	File::IOFile pStream(filename, "rb");
	if (pStream)
	{
		u64 filesize = pStream.GetSize();

		pStream.ReadBytes(m_pIPL + offset, filesize);

		m_FontsLoaded = true;
	}
}

void CEXIIPL::SetCS(int _iCS)
{
	if (_iCS)
	{
		// cs transition to high
		m_uPosition = 0;
	}
}

bool CEXIIPL::IsPresent()
{
	return true;
}

void CEXIIPL::TransferByte(u8& _uByte)
{
	// Seconds between 1.1.2000 and 4.1.2008 16:00:38
	const u32 cWiiBias = 0x0F1114A6;

	// The first 4 bytes must be the address
	// If we haven't read it, do it now
	if (m_uPosition <= 3)
	{
		m_uAddress <<= 8;
		m_uAddress |= _uByte;
		m_uRWOffset = 0;
		_uByte = 0xFF;

		// Check if the command is complete
		if (m_uPosition == 3)
		{
			// Get the time ...
			u32 &rtc = *((u32 *)&m_RTC);
			if (SConfig::GetInstance().m_LocalCoreStartupParameter.bWii)
			{
				// Subtract Wii bias
				rtc = Common::swap32(CEXIIPL::GetGCTime() - cWiiBias);
			}
			else
			{
				rtc = Common::swap32(CEXIIPL::GetGCTime());
			}

			// Log the command
			std::string device_name;

			switch (CommandRegion())
			{
			case REGION_RTC:
				device_name = "RTC";
				break;
			case REGION_SRAM:
				device_name = "SRAM";
				break;
			case REGION_UART:
				device_name = "UART";
				break;
			case REGION_EUART:
			case REGION_EUART_UNK:
				device_name = "EUART";
				break;
			case REGION_UART_UNK:
				device_name = "UART Other?";
				break;
			case REGION_BARNACLE:
				device_name = "UART Barnacle";
				break;
			case REGION_WRTC0:
			case REGION_WRTC1:
			case REGION_WRTC2:
				device_name = "Wii RTC flags - not implemented";
				break;
			default:
				if ((m_uAddress >> 6) < ROM_SIZE)
				{
					device_name = "ROM";
				}
				else
				{
					device_name = "illegal address";
					_dbg_assert_msg_(EXPANSIONINTERFACE, 0,
						"EXI IPL-DEV: %s %08x", device_name.c_str(), m_uAddress);
				}
				break;
			}

			DEBUG_LOG(EXPANSIONINTERFACE, "%s %s %08x", device_name.c_str(),
				IsWriteCommand() ? "write" : "read", m_uAddress);
		}
	} 
	else
	{
		// Actually read or write a byte
		switch (CommandRegion())
		{
		case REGION_RTC:
			if (IsWriteCommand())
				m_RTC[(m_uAddress & 0x03) + m_uRWOffset] = _uByte;
			else
				_uByte = m_RTC[(m_uAddress & 0x03) + m_uRWOffset];
			break;

		case REGION_SRAM:
			if (IsWriteCommand())
				g_SRAM.p_SRAM[(m_uAddress & 0x3F) + m_uRWOffset] = _uByte;
			else
				_uByte = g_SRAM.p_SRAM[(m_uAddress & 0x3F) + m_uRWOffset];
			break;

		case REGION_UART:
		case REGION_EUART:
			if (IsWriteCommand())
			{
				if (_uByte != '\0')
					m_szBuffer[m_count++] = _uByte;
				if ((m_count >= 256) || (_uByte == 0xD))
				{					
					m_szBuffer[m_count] = 0x00;
					NOTICE_LOG(OSREPORT, "%s", m_szBuffer);
					memset(m_szBuffer, 0, sizeof(m_szBuffer));
					m_count = 0;
				}
			}
			else
			{
				// "Queue Length"... return 0 cause we're instant
				_uByte = 0;
			}
			break;

		case REGION_EUART_UNK:
			// Writes 0xf2 then 0xf3 on EUART init. Just need to return non-zero
			// so we can leave the byte untouched.
			break;

		case REGION_UART_UNK:
			DEBUG_LOG(OSREPORT, "UART? %x", _uByte);
			_uByte = 0xff;
			break;

		case REGION_BARNACLE:
			DEBUG_LOG(OSREPORT, "UART Barnacle %x", _uByte);
			break;

		case REGION_WRTC0:
		case REGION_WRTC1:
		case REGION_WRTC2:
			// WII only RTC flags... afaik just the wii menu initialize it
		default:
			if ((m_uAddress >> 6) < ROM_SIZE)
			{
				if (!IsWriteCommand())
				{
					u32 position = ((m_uAddress >> 6) & ROM_MASK) + m_uRWOffset;

					// Technically we should descramble here iff descrambling logic is enabled.
					// At the moment, we pre-decrypt the whole thing and
					// ignore the "enabled" bit - see CEXIIPL::CEXIIPL
					_uByte = m_pIPL[position];

					if ((position >= 0x001AFF00) && (position <= 0x001FF474) && !m_FontsLoaded)
					{
						PanicAlertT(
							"Error: Trying to access %s fonts but they are not loaded. "
							"Games may not show fonts correctly, or crash.",
							(position >= 0x001FCF00) ? "ANSI" : "SJIS");
						m_FontsLoaded = true; // Don't be a nag :p
					}
				}
			}
			else
			{
				NOTICE_LOG(OSREPORT, "EXI IPL-DEV: %s %x at %08x",
					IsWriteCommand() ? "write" : "read", _uByte, m_uAddress);
			}
			break;
		}

		m_uRWOffset++;
	}

	m_uPosition++;
}

u32 CEXIIPL::GetGCTime()
{
	u64 ltime = 0;
	static const u32 cJanuary2000 = 0x386D4380;  // Seconds between 1.1.1970 and 1.1.2000

	if (Movie::IsRecordingInput() || Movie::IsPlayingInput())
	{
		ltime = Movie::GetRecordingStartTime();

		// let's keep time moving forward, regardless of what it starts at
		ltime += CoreTiming::GetTicks() / SystemTimers::GetTicksPerSecond();
	}
	else
	{
		// hack in some netplay stuff
		ltime = NetPlay_GetGCTime();

		if (0 == ltime)
			ltime = Common::Timer::GetLocalTimeSinceJan1970();
	}

	return ((u32)ltime - cJanuary2000);

#if 0
	// (mb2): I think we can get rid of the IPL bias.
	// I know, it's another hack so I let the previous code for a while.

	// Get SRAM bias
	u32 Bias;

	for (int i=0; i<4; i++)
	{
		((u8*)&Bias)[i] = sram_dump[0xc + (i^3)];
	}

	// Get the time ...
	u64 ltime = Common::Timer::GetTimeSinceJan1970();
	return ((u32)ltime - cJanuary2000 - Bias);
#endif
}
