#include "NewsServer.h"
#include <iostream>
#include <Klient.h>
#include <Globals.h>

// Inhalt der NewsServer Klasse aus NewsServer.h
// Konstruktor 
NewsServer::NewsServer(InitDatei* zIni)
	: Thread()
{
	Network::Start(100);
	std::cout << "IS: Verbindung mit Datenbank wird hergestellt...\n";
	db = new NSDatenbank(zIni);
	klients = new RCArray< NSKlient >();
	empfangen = 0;
	gesendet = 0;
	fehler = new Text();
	ini = dynamic_cast<InitDatei*>(zIni->getThis());
	id = (int)*zIni->zWert("ServerId");
	server = new Server();
	aServer = new SSLServer();
	aServer->setPrivateKeyPassword(zIni->zWert("SSLPasswort")->getText());
	aServer->setCertificateFile(zIni->zWert("SSLCert")->getText());
	aServer->setPrivateKeyFile(zIni->zWert("SSLKey")->getText());
	std::cout << "IS: Starten des Admin Servers...\n";
	if (!aServer->verbinde((unsigned short)TextZuInt(ini->zWert("AdminServerPort")->getText(), 10), 10))
	{
		std::cout << "IS: Der Admin Server konnte nicht gestartet werden. Das Programm wird beendet.\n";
		exit(1);
	}
	db->setServerStatus(id, 2);
	end = 0;
	nichtPausiert = 0;
	InitializeCriticalSection(&cs);
	if (zIni->zWert("Aktiv")->istGleich("TRUE"))
	{
		serverStarten();
		serverFortsetzen();
	}
}

// Destruktor 
NewsServer::~NewsServer()
{
	fehler->release();
	server->trenne();
	server->release();
	aServer->trenne();
	aServer->release();
	if (klients)
		klients->release();
	ini->release();
	db->release();
	DeleteCriticalSection(&cs);
}

// nicht constant 
void NewsServer::runn()
{
	while (!end && aServer->isConnected())
	{
		SSLSKlient* klient;
		klient = aServer->getKlient();
		if (end && klient)
		{
			klient->trenne();
			klient = (SSLSKlient*)klient->release();
			Sleep(1000);
			return;
		}
		if (!klient)
			continue;
		NSAKlient* clHandle = new NSAKlient(klient, dynamic_cast<NewsServer*>(getThis()));
		clHandle->start();
	}
}

void NewsServer::thread()
{
	while (server->isConnected())
	{
		SKlient* klient;
		klient = server->getKlient();
		if (!klient)
			continue;
		Framework::getThreadRegister()->cleanUpClosedThreads();
		NSKlient* clHandle = new NSKlient(klient, dynamic_cast<NewsServer*>(getThis()));
		EnterCriticalSection(&cs);
		klients->add(clHandle);
		LeaveCriticalSection(&cs);
		clHandle->start();
	}
}

void NewsServer::close()
{
	db->setServerStatus(id, 1);
	server->trenne();
#ifdef WIN32
	warteAufThread(1000);
#endif
	EnterCriticalSection(&cs);
	for (int i = 0; i < klients->getEintragAnzahl(); i++)
		klients->z(i)->absturz();
	klients = (RCArray<NSKlient>*)klients->release();
	LeaveCriticalSection(&cs);
	ende();
	run = 0;
	end = 1;
	Klient* klient = new Klient();
	klient->verbinde(aServer->getPort(), "127.0.0.1");
	Sleep(500);
	aServer->trenne();
	klient->release();
}

bool NewsServer::serverStarten()
{
	if (nichtPausiert)
	{
		fehler->setText("Der Server konnte nicht gestartet werden: Der Server läuft bereits.");
		return 0;
	}
	if (server)
		server->release();
	server = new Server();
	if (server->verbinde((unsigned short)TextZuInt(ini->zWert("ServerPort")->getText(), 10), 10))
	{
		nichtPausiert = 1;
		start();
		return 1;
	}
	else
	{
		serverBeenden();
		fehler->setText("Der Server konnte nicht gestartet werden: Eventuell ist der Port in benutzung.");
		return 0;
	}
}

bool NewsServer::serverPause()
{
	if (!nichtPausiert)
	{
		fehler->setText("Der Server konnte nicht pausiert werden: Der Server läuft nicht.");
		return 0;
	}
	if (!db->setServerStatus(id, 2))
	{
		fehler->setText("Der Server konnte nicht pausiert werden: ");
		fehler->append(db->getLetzterFehler());
		return 0;
	}
	return 1;
}

bool NewsServer::serverFortsetzen()
{
	if (!nichtPausiert)
	{
		fehler->setText("Der Server konnte nicht fortgesetzt werden: Der Server läuft nicht.");
		return 0;
	}
	if (!db->setServerStatus(id, 3))
	{
		fehler->setText("Der Server konnte nicht fortgesetzt werden: ");
		fehler->append(db->getLetzterFehler());
		return 0;
	}
	return 1;
}

bool NewsServer::serverBeenden()
{
	if (!nichtPausiert)
	{
		fehler->setText("Der Server konnte nicht beendet werden: Der Server läuft nicht.");
		return 0;
	}
	if (db->serverIstNichtPausiert(id))
	{
		fehler->setText("Der Server konnte nicht beendet werden: Der Server muss erst pausiert werden.");
		return 0;
	}
	nichtPausiert = 0;
	ende();
	if (server)
		server->trenne();
	return 1;
}

bool NewsServer::setMaxKlients(int mc)
{
	if (!db->setMaxClients(id, mc))
	{
		fehler->setText("Die maximale Anzahl der Clients konnte nicht gesetzt werden:\n");
		fehler->append(db->getLetzterFehler());
		return 0;
	}
	ini->setWert("MaxClients", Text() += mc);
	return 1;
}

bool NewsServer::absturzKlient(int klientId)
{
	bool gefunden = 0;
	EnterCriticalSection(&cs);
	for (int i = 0; i < klients->getEintragAnzahl(); i++)
	{
		if (klients->z(i)->getKlientNummer() == klientId)
		{
			klients->z(i)->absturz();
			klients->remove(i);
			gefunden = 1;
			break;
		}
	}
	LeaveCriticalSection(&cs);
	return gefunden;
}

bool NewsServer::removeKlient(NSKlient* zKlient)
{
	bool gefunden = 0;
	EnterCriticalSection(&cs);
	for (int i = 0; i < klients->getEintragAnzahl(); i++)
	{
		if (klients->z(i) == zKlient)
		{
			klients->remove(i);
			gefunden = 1;
			break;
		}
	}
	LeaveCriticalSection(&cs);
	return gefunden;
}

void NewsServer::addGesendet(int bytes)
{
	gesendet += bytes;
}

void NewsServer::addEmpfangen(int bytes)
{
	empfangen += bytes;
}

// constant 
bool NewsServer::istAn() const
{
	return db->serverIstNichtPausiert(id);
}

Server* NewsServer::zServer() const
{
	return server;
}

NSDatenbank* NewsServer::zDB() const
{
	return db;
}

bool NewsServer::hatClients() const
{
	return klients->hat(0);
}

int NewsServer::getId() const
{
	return id;
}

const char* NewsServer::getLetzterFehler() const
{
	return fehler->getText();
}


// Inhalt der NSAKlient Klasse aus NewsServer.h
// Konstruktor 
NSAKlient::NSAKlient(SSLSKlient* klient, NewsServer* ns)
	: Thread()
{
	this->klient = klient;
	name = new Text("");
	passwort = new Text("");
	adminId = 0;
	this->ns = ns;
}

// Destruktor 
NSAKlient::~NSAKlient()
{
	klient->trenne();
	klient->release();
	ns->release();
	name->release();
	passwort->release();
}

// nicht constant 
void NSAKlient::thread()
{
	while (1)
	{
		char c = 0;
		if (!klient->getNachricht(&c, 1))
			break;
		else
		{
			bool br = 0;
			switch (c)
			{
			case 1: // Login
				if (1)
				{
					klient->sende("\1", 1);
					unsigned char nLen = 0;
					klient->getNachricht((char*)&nLen, 1);
					char* n = new char[nLen + 1];
					n[(int)nLen] = 0;
					if (nLen)
						klient->getNachricht(n, nLen);
					unsigned char pLen = 0;
					klient->getNachricht((char*)&pLen, 1);
					char* p = new char[pLen + 1];
					p[(int)pLen] = 0;
					if (pLen)
						klient->getNachricht(p, pLen);
					int adminId = ns->zDB()->istAdministrator(n, p);
					if (adminId)
					{
						klient->sende("\1", 1);
						name->setText(n);
						passwort->setText(p);
						this->adminId = adminId;
					}
					else
						errorZuKlient("Falsche Kombination aus Name und Passwort.");
					delete[] n;
					delete[] p;
				}
				break;
			case 2: // Logout
				adminId = 0;
				name->setText("");
				passwort->setText("");
				klient->sende("\1", 1);
				break;
			case 3: // Trennen
				br = 1;
				klient->sende("\1", 1);
				break;
			case 4: // Server starten
				if (!adminId)
					errorZuKlient("Du musst dich einloggen.");
				else
				{
					if (ns->zDB()->adminHatRecht(adminId, Admin_Recht::NSStarten))
					{
						if (!ns->serverStarten())
						{
							Text* err = new Text();
							err->append(ns->getLetzterFehler());
							errorZuKlient(err->getText());
							err->release();
						}
						else
							klient->sende("\1", 1);
					}
					else
						errorZuKlient("Du bist nicht berechtigt den Server zu starten.");
				}
				break;
			case 5: // Server beenden
				if (!adminId)
					errorZuKlient("Du musst dich einloggen.");
				else
				{
					if (ns->zDB()->adminHatRecht(adminId, Admin_Recht::NSBeenden))
					{
						if (ns->serverBeenden())
							klient->sende("\1", 1);
						else
						{
							Text* err = new Text();
							err->append(ns->getLetzterFehler());
							errorZuKlient(err->getText());
							err->release();
						}
					}
					else
						errorZuKlient("Du bist nicht berechtigt den Server zu beenden.");
				}
				break;
			case 6: // Programm Schließen
				if (!adminId)
					errorZuKlient("Du musst dich einloggen.");
				else
				{
					bool ok = 0;
					if (ns->isRunning())
					{
						if (ns->zDB()->adminHatRecht(adminId, Admin_Recht::NSBeenden))
						{
							if (ns->serverBeenden())
								ok = 1;
							else
							{
								Text* err = new Text();
								err->append(ns->getLetzterFehler());
								errorZuKlient(err->getText());
								err->release();
							}
						}
						else
							errorZuKlient("Du bist nicht berechtigt den Server zu beenden.");
					}
					else
						ok = 1;
					if (ok && ns->hatClients())
					{
						errorZuKlient("Es sind noch Klients Online. Bitte versuche es später erneut.");
						break;
					}
					if (ok)
					{
						klient->sende("\1", 1);
						std::cout << "NS: Der Server wird von Benutzer " << adminId << " heruntergefahren.\n";
						ns->close();
						br = 1;
					}
				}
				break;
			case 7: // Programm abstürzen
				if (!adminId)
					errorZuKlient("Du musst dich einloggen.");
				else
				{
					bool ok = 0;
					if (ns->isRunning())
					{
						if (ns->zDB()->adminHatRecht(adminId, Admin_Recht::NSBeenden))
						{
							ns->serverBeenden();
							ok = 1;
						}
						else
							errorZuKlient("Du bist nicht berechtigt den Server zu beenden.");
					}
					else
						ok = 1;
					if (ok)
					{
						klient->sende("\1", 1);
						std::cout << "NS: Der Server wurde von Benutzer " << adminId << " terminiert.\n";
						ns->close();
						br = 1;
					}
				}
				break;
			case 8: // Status Frage
				if (1)
				{
					char status = 0;
					if (ns->isRunning())
					{
						status = 1;
						if (ns->istAn())
							status = 2;
					}
					klient->sende("\1", 1);
					klient->sende(&status, 1);
				}
				break;
			case 9: // Server pausieren
				if (!adminId)
					errorZuKlient("Du musst dich einloggen.");
				else
				{
					klient->sende("\1", 1);
					char pause = 0;
					klient->getNachricht(&pause, 1);
					if (ns->zDB()->adminHatRecht(adminId, Admin_Recht::NSPausieren))
					{
						bool ok = 0;
						if (pause)
							ok = ns->serverPause();
						else
							ok = ns->serverFortsetzen();
						if (ok)
							klient->sende("\1", 1);
						else
						{
							Text* err = new Text();
							err->append(ns->getLetzterFehler());
							errorZuKlient(err->getText());
							err->release();
						}
					}
					else
					{
						if (pause)
							errorZuKlient("Du bist nicht berechtigt den Server zu pausieren.");
						else
							errorZuKlient("Du bist nicht berechtigt den Server fortzusetzen.");
					}
				}
				break;
			case 0xA: // maximale Anzahl der Clients setzen
				if (!adminId)
					errorZuKlient("Du musst dich einloggen.");
				else
				{
					klient->sende("\1", 1);
					int maxC = 0;
					klient->getNachricht((char*)&maxC, 4);
					if (ns->zDB()->adminHatRecht(adminId, Admin_Recht::NSMCChange))
					{
						if (ns->setMaxKlients(maxC))
							klient->sende("\1", 1);
						else
						{
							Text* err = new Text();
							err->append(ns->getLetzterFehler());
							errorZuKlient(err->getText());
							err->release();
						}
					}
					else
						errorZuKlient("Du bist nicht berechtigt die maximale Anzahl der Clients zu verändern.");
				}
				break;
			case 0xC: // klient absturtz
				if (1)
				{
					klient->sende("\1", 1);
					int klientId = 0;
					klient->getNachricht((char*)&klientId, 4);
					if (klientId && ns->absturzKlient(klientId))
						klient->sende("\1", 1);
					else
						klient->sende("\0", 1);
				}
				break;
			default:
				errorZuKlient("Unbekannte Nachricht!");
				break;
			}
			if (br)
				break;
			ns->addEmpfangen(klient->getDownloadBytes(1));
			ns->addGesendet(klient->getUploadBytes(1));
		}
	}
	ns->addEmpfangen(klient->getDownloadBytes(1));
	ns->addGesendet(klient->getUploadBytes(1));
	delete this;
}

void NSAKlient::errorZuKlient(const char* nachricht) const // sendet eine Fehlernachricht zum Klient
{
	klient->sende("\3", 1);
	char len = (char)textLength(nachricht);
	klient->sende(&len, 1);
	klient->sende(nachricht, len);
}


// Inhalt der NSKlient aus NewsServer.h
// Konstruktor 
NSKlient::NSKlient(SKlient* klient, NewsServer* ns)
	: Thread()
{
	this->klient = klient;
	unsigned char key[20] = { 4, 150, 243, 145, 204, 71, 253, 88, 4, 51, 189, 207, 189, 248, 231, 40, 106, 7, 182, 5 };
	klient->setSendeKey((char*)key, 20);
	klient->setEmpfangKey((char*)key, 20);
	klientNummer = 0;
	this->ns = ns;
}

// Destruktor 
NSKlient::~NSKlient()
{
	klient->release();
	ns->release();
}

// nicht constant
void NSKlient::absturz()
{
	ende();
	klient->trenne();
	ns->zDB()->unregisterKlient(klientNummer, ns->getId());
}

void NSKlient::thread()
{
	while (1)
	{
		char c = 0;
		if (!klient->getNachrichtEncrypted(&c, 1))
			break;
		else
		{
			bool br = 0;
			switch (c)
			{
			case 1: // Klient identifikation
				klient->getNachrichtEncrypted((char*)&klientNummer, 4);
				if (!ns->zDB()->proveKlient(klientNummer, ns->getId()))
				{
					klientNummer = 0;
					errorZuKlient("Du bist nicht für diesen Server eingetragen");
				}
				else
				{
					Text* key = ns->zDB()->getKlientKey(klientNummer);
					if (!key)
						errorZuKlient("Es konnte kein Key ermittelt werden.");
					else
					{
						klient->sendeEncrypted("\1", 1);
						klient->setEmpfangKey(*key, key->getLength());
						klient->setSendeKey(*key, key->getLength());
						key->release();
					}
				}
				break;
			case 2: // Main / Erhaltung Server message
				if (1)
				{
					char befehl = 0;
					klient->getNachrichtEncrypted(&befehl, 1);
					switch (befehl)
					{
					case 2: // klient absturtz
						if (1)
						{
							int klientId = 0;
							klient->getNachrichtEncrypted((char*)&klientId, 4);
							if (klientId && ns->absturzKlient(klientId))
								klient->sendeEncrypted("\1", 1);
							else
								klient->sendeEncrypted("\0", 1);
						}
						break;
					default:
						errorZuKlient("Befehl nicht bekannt!");
						break;
					}
				}
				break;
			case 3: // Verbindungsende
				br = 1;
				klient->sendeEncrypted("\1", 1);
				break;
			case 4: // unregister Klient
				if (!klientNummer)
				{
					errorZuKlient("Du bist nicht Identifiziert.");
					break;
				}
				ns->zDB()->unregisterKlient(klientNummer, ns->getId());
				klient->sendeEncrypted("\1", 1);
				break;
			case 5: // frage nach Seite
				if (klientNummer)
				{
					klient->sendeEncrypted("\1", 1);
					unsigned char len = 0;
					klient->getNachrichtEncrypted((char*)&len, 1);
					char* n = new char[len + 1];
					n[(int)len] = 0;
					klient->getNachrichtEncrypted(n, len);
					Text* pfad = new Text("../news/");
					pfad->append(n);
					delete[] n;
					if (!DateiExistiert(pfad->getText()))
					{
						errorZuKlient("Die Seite ist nicht vorhanden.");
						pfad->release();
						break;
					}
					klient->sendeEncrypted("\1", 1);
					Datei* dat = new Datei();
					dat->setDatei(pfad);
					RCArray< Text >* list = dat->getDateiListe();
					int dAnz = list->getEintragAnzahl();
					for (int i = 0; i < dAnz; i++)
					{
						if (DateiIstVerzeichnis(list->get(i)) || list->z(i)->istGleich(".") || list->z(i)->istGleich(".."))
						{
							list->remove(i);
							dAnz--;
							i--;
						}
					}
					klient->sendeEncrypted((char*)&dAnz, 4);
					for (int i = 0; i < dAnz; i++)
					{
						Text* pf = new Text(dat->zPfad()->getText());
						pf->append("/");
						pf->append(list->z(i)->getText());
						Datei* d = new Datei();
						d->setDatei(pf);
						d->open(Datei::Style::lesen);
						char nl = (char)list->z(i)->getLength();
						klient->sendeEncrypted(&nl, 1);
						klient->sendeEncrypted(list->z(i)->getText(), nl);
						__int64 gr = d->getSize();
						klient->sendeEncrypted((char*)&gr, 8);
						char* bytes = new char[2048];
						while (gr > 0)
						{
							int len = gr > 2048 ? 2048 : (int)gr;
							d->lese(bytes, len);
							klient->sende(bytes, len);
							gr -= len;
						}
						delete[] bytes;
						d->close();
						d->release();
					}
					list->release();
					dat->release();
				}
				else
					errorZuKlient("Du bist nicht Identifiziert.");
				break;
			case 0x6: // ping
				if (1)
				{
					if (!klientNummer)
					{
						errorZuKlient("Du bist nicht Identifiziert.");
						break;
					}
					klient->sendeEncrypted("\1", 1);
				}
				break;
			default:
				errorZuKlient("Unbekannte Nachricht!");
				break;
			}
			if (br)
				break;
			ns->addEmpfangen(klient->getDownloadBytes(1));
			ns->addGesendet(klient->getUploadBytes(1));
		}
	}
	ns->addEmpfangen(klient->getDownloadBytes(1));
	ns->addGesendet(klient->getUploadBytes(1));
	ns->removeKlient(this); // delete this
}

// constant
void NSKlient::errorZuKlient(const char* nachricht) const // sendet eine Fehlernachricht zum Klient
{
	klient->sendeEncrypted("\3", 1);
	char len = (char)textLength(nachricht);
	klient->sendeEncrypted(&len, 1);
	klient->sendeEncrypted(nachricht, len);
}

int NSKlient::getKlientNummer() const // gibt die KlientId zurück
{
	return klientNummer;
}