#include <iostream>
#include <fstream>
#ifdef OPENWRT
#include <utility>
#else
#ifdef PLATFORM_freebsd
#include <sys/types.h>
#endif
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#endif

#include <socketcc.h>
#include "ConfigFile.h"
#include "Kaid.h"
#include "StrUtils.h"
#include "ifaddrlist.h"

#ifndef OPENWRT
void doEnumeration()
{
	struct ifaddrlist *al;
	char err[132];
	int n, i;
	in_addr addr;

	n = ifaddrlist(&al, err);
	if(n < 0)
	{
    } else {
		for (i = n; i > 0; --i, ++al)
		{
			addr.s_addr = al->addr;
			printf("Device: %s -> %s\n", al->device, inet_ntoa(addr));
		}
	}
}
#endif

using namespace std;

CConfigFile::CConfigFile (string configFile)
{
	m_sFilename = configFile;
	// Set defaults
	
	OrbPort = 34525;
	UseIPv6 = false;
#ifdef PLATFORM_freebsd
	User = "";
#endif
	
	UIBind = "";
	UIIP = "";
	UIDev = "";
	UIPort = 34522;
	
	EngineBind = "";
	EngineIP = "";
	EngineDev = "";
	EnginePort = 0;
	EnginePAT = 0;
	Verbosity=0;
	
	#ifdef PLATFORM_macosx_jaguar
	SniffDevice = "en0";
	#else
	SniffDevice = "eth0";
	#endif	
	LearningMode = lmManual;
	IgnoreMacs = "";
	ConsoleMacs = "";
	XBoxHomebrew = false;
	ConfigURL = "";
	OrbList = "";
	Username = "";
	Password = "";
	AutoLogin = 0;
}

void
CConfigFile::Load ()
{
	ifstream file (m_sFilename.c_str ());
	
	string line;
	string name;
	string value;
	int posEqual;
	
	while (std::getline (file, line))
	{
		if (!line.length ())
			continue;	
		if (line[0] == '#')
			continue;
		if (line[0] == ';')
			continue;
	
		posEqual = line.find ('=');
		name = trim (line.substr (0, posEqual));
		value = trim (line.substr (posEqual + 1));
		m_vsFields.push_back (name);
		m_vsValues.push_back (value);
	}
	
	for (unsigned int i = 0; i < m_vsFields.size (); i++)
	{
		string fld = case_lower (m_vsFields[i]);
		string val = case_lower (m_vsValues[i]);
		if (fld == "uibind")
		{
			UIBind = m_vsValues[i];
		}
		else if (fld == "orbport")
		{
			OrbPort = atoi (m_vsValues[i].c_str ());
		}
		else if (fld == "protocol")
		{
			UseIPv6 = (m_vsValues[i] == "6");
		}
		else if (fld == "enginebind")
		{
			EngineBind = m_vsValues[i];
		}
		else if (fld == "sniffdevice")
		{
			SniffDevice = m_vsValues[i];
		}
		else if (fld == "maclearningmode")
		{
			if (val == "auto" || val == "1" || val == "bitch" || val == "allmine")
				LearningMode = lmAuto;
			else if (val == "firstrun" || val == "2")
				LearningMode = lmFirstRun;
			else if (val == "firstfound" || val == "3" || val == "first")
				LearningMode = lmFirstFound;
			else if (val == "manual" || val == "4" || val == "legacy" || val == "oldstyle"
				|| val == "turbo" || val == "turboboost" || val == "boost" || val == "knightrider")
				LearningMode = lmManual;
		}
		else if (fld == "consolemacs")
		{
			ConsoleMacs = m_vsValues[i];
			if(!ConsoleMacs.empty() && ConsoleMacs[ConsoleMacs.size() - 1] != ';')
				ConsoleMacs += ";";
		}
		else if (fld == "ignoremacs")
		{
			IgnoreMacs = m_vsValues[i];
			if(!IgnoreMacs.empty() && IgnoreMacs[IgnoreMacs.size() - 1] != ';')
				IgnoreMacs += ";";
		}		
		else if (fld == "configurl")
		{
			ConfigURL = m_vsValues[i];
		}
		else if (fld == "username")
		{
			Username = m_vsValues[i];
		}
		else if (fld == "password")
		{
			Password = m_vsValues[i];
		}
		else if (fld == "autologin")
		{
			AutoLogin = atoi (m_vsValues[i].c_str ());
		} 
		else if (fld == "verbosity")
		{
			Verbosity = atoi (m_vsValues[i].c_str ());
		} 
		else if (fld == "enginepat")
		{
			EnginePAT = atoi (m_vsValues[i].c_str ());
		} 
		else if (fld == "xboxhomebrew")
		{
			XBoxHomebrew = ((val == "1") || (val == "on") || (val == "true") ||
			(val == "yes") || (val == "yay") || (val == "foshizzle"));
		}
		else if (fld == "user")
		{
			User = m_vsValues[i];
		}
		else
		{
		}
	}
	if (LearningMode != lmManual) ConsoleMacs = "";
	ParseBindings ();
	GetOrbList ();
}

void
CConfigFile::ParseBindings ()
{
	if (EngineBind.find (":") < 0)
    {
      	if (EngineBind.find (".") < 0)
			EngineDev = EngineBind;
      	else
			EngineIP = EngineBind;
    }
  	else
    {
    	string sAux = EngineBind.substr (0, EngineBind.find (":"));
    	if( sAux.find(".") < 0)
    		EngineDev = sAux;
    	else
    		EngineIP = sAux;
      	EnginePort = atoi (EngineBind.substr (EngineBind.find (":") + 1).c_str ());
    }

	if (UIBind.find (":") < 0)
    {
      	if (UIBind.find (".") < 0)
			UIDev = UIBind;
      	else
			UIIP = UIBind;
    }
  	else
    {
    	string sAux = UIBind.substr (0, UIBind.find (":"));
    	if( sAux.find(".") < 0)
    		UIDev = sAux;
    	else
    		UIIP = sAux;
      	UIPort = atoi (UIBind.substr (UIBind.find (":") + 1).c_str ());
    }
}

void
CConfigFile::GetOrbList ()
{
  string sOrbList, sAux, sAux2;
  if (ConfigURL.empty ())
    return;
  try
  {
    debuglog("KAID", "Kai Engine for "PLATFORM" is initialising...");
    string sURL = ConfigURL;
    string sHostSep = "/";
    int iProtoSeps = sURL.find ("://");
    if (iProtoSeps > 0)
      sURL = sURL.substr (iProtoSeps + 3, sURL.size () - iProtoSeps + 3);
    int iHostSepPos = sURL.find (sHostSep);
    string sHost (sURL.substr (0, iHostSepPos));
    string sHTTPReq (sURL.substr (iHostSepPos));
    IPAddress ipOrbListServer;
    ipOrbListServer = sHost.c_str ();

    TCPClientSocket pcHTTPClient (ipOrbListServer, 80);
    string sRequest ("GET " + sHTTPReq + " HTTP/1.1\r\nHost: " +
		     sHost + "\r\nAccept: text/xml\r\n\r\n");
    pcHTTPClient.SendASCII (sRequest.c_str ());

    while (1)
    {
		char pcRecvBuffer[8192];
		int iBytesTransferred;
		iBytesTransferred = pcHTTPClient.RecvData (pcRecvBuffer,
						   sizeof (pcRecvBuffer) - 1);
		if (iBytesTransferred < 1)
	  		throw "Error getting Orb list!";
		pcRecvBuffer[iBytesTransferred] = 0;
		sAux += pcRecvBuffer;
	}
  }
  catch (SocketException & excep)
  {
	if (excep == SocketException::errNotConnected)
	{
		int iFromChar = sAux.find ("KaiOrbitalServer");
		// iFromChar = 0;
		if (iFromChar < 0)
		{
	    	debuglog("KAID", "No orbitals servers are available...");
	    	throw "No orbitals found!";
	  	}
		int iToChar = sAux.rfind ("KaiOrbitalServer=");
		sAux2 = sAux.substr (iToChar);
		iToChar += sAux2.find (";") + 1;
		sAux = sAux.substr (iFromChar, (iToChar - iFromChar));
		vector < string > vsServers;
		Tokenize (sAux, vsServers, ";");
		unsigned int i;
		for (i = 0; i < vsServers.size (); i++)
	  		sOrbList.append (vsServers[i].substr (vsServers[i].find ("=") + 1)
			   + ";");
		OrbList = sOrbList;
	}
    else
    {
		string err = (const char *) excep;
		debuglog("KAID","Failed to download orbital list...");
		throw "No orbitals found!";
	}
  }
}