SQCSimulator2023/SQCSim-srv/server.cpp

#include "server.h"

Server::Server(LogDest log) : m_log(log) {
	if (log == LogDest::LOGFILE) {
		m_logfile = std::ofstream("server.log", std::ofstream::out);
		if (!m_logfile.is_open()) {
			m_log = LogDest::CONSOLE; // Fallback console.
			Log("Ouverture fichier log: repli vers console.", true, false);
		}
	}
}

Server::~Server() {
	if (m_logfile.is_open())
		m_logfile.close();
	if (m_sock_udp)
		closesocket(m_sock_udp);
	if (m_sock_tcp)
		closesocket(m_sock_tcp);
#ifdef _WIN32
	WSACleanup();
#endif
}

int Server::Init() {
	Log("Initialisation du serveur...", false, false);

#ifdef _WIN32
	if (WSAStartup(MAKEWORD(2, 2), &m_wsaData) != 0) { /* Initialisation de l'environnement reseau (Windows only) */
		Log("Initialisation WinSock.", true, true);
		return 1;
	}
#endif

	m_sock_udp = socket(AF_INET, SOCK_DGRAM, 0);
	if (m_sock_udp == INVALID_SOCKET) { /* Creation du socket UDP */
		Log("Creation Socket UDP.", true, true);
		return 2;
	}

	m_sock_tcp = socket(AF_INET, SOCK_STREAM, 0);
	if (m_sock_tcp == INVALID_SOCKET) { /* Creation du socket TCP */
		Log("Creation Socket TCP.", true, true);
		return 3;
	}

	/* Creation structure donnes descripteur du socket serveur */
	sockaddr_in addr;
	addr.sin_family = AF_INET;
	addr.sin_port = htons(SRV_PORT);
	addr.sin_addr.s_addr = htonl(INADDR_ANY);

	if (bind(m_sock_udp, (sockaddr*)&addr, sizeof(addr)) != 0) { /* Associer le socket UDP au port */
		Log("Association Socket UDP.", true, true);
		return 4;
	}

	if (bind(m_sock_tcp, (sockaddr*)&addr, sizeof(addr)) != 0) { /* Associer le socket TCP au port */
		Log("Association Socket TCP.", true, true);
		return 5;
	}

	return 0;
}

int Server::Ready() {
	Log("Prêt à démarrer...", false, false);
	return 0;
}

void Server::Run() {
	Log("Partie en cours...", false, false);
}

inline std::string Server::Timestamp() {
	time_t rawtime;
	struct tm timeinfo;
	char buffer[80];

	time(&rawtime);

#ifdef _WIN32
	localtime_s(&timeinfo, &rawtime);
#else
	localtime_r(&rawtime, &timeinfo);
#endif

	strftime(buffer, sizeof(buffer), "%d-%m-%Y %H:%M:%S", &timeinfo);
	std::string str(buffer);

	return "[" + str + "] ";
}

void Server::Log(std::string str, bool is_error = false, bool is_fatal = false) {
	switch (m_log) {
	case LogDest::LOGFILE:
		m_logfile << Timestamp() << (is_fatal ? "FATAL " : "") << (is_error ? "ERROR " : "") << str << std::endl;
		break;
	case LogDest::CONSOLE:
	default:
		std::cout << Timestamp() << (is_fatal? "FATAL ": "") << (is_error ? "ERROR ": "") << str << std::endl;
		break;
	}

	if (is_fatal) {
		if (m_sock_udp)
			closesocket(m_sock_udp);
		if (m_sock_tcp)
			closesocket(m_sock_tcp);
#ifdef _WIN32
		WSACleanup();
#endif
	}
}

///* Recevoir paquet */
//while (true) {
//	char buffer[2048];
//	sockaddr_in client;
//
//#ifdef _WIN32 // Mais pourquoi?
//	int clen = sizeof(client);
//#else
//	unsigned int clen = sizeof(client);
//#endif
//
//	int count = recvfrom(socket_udp, buffer, sizeof(buffer) - 1, 0, (sockaddr*)&client, &clen);
//
//	if (count < 0) {
//		Log("Erreur de reception de paquet.", socket_udp);
//		return 4;
//	}
//
//	buffer[count] = '\0';
//
//	/* Gérer le paquet reçu */
//	std::string commande(buffer);
//
//	if (commande.find("echo ") == 0) {													/*		ECHO		*/
//		std::string::size_type pos = commande.find(' ');
//		std::string parametres = commande.substr(pos + 1);
//
//		sendto(socket_udp, parametres.c_str(), parametres.length(), 0, (const sockaddr*)&client, sizeof(client));
//	}
//	else if (commande.find("date ") == 0) {												/*		DATE		*/
//		time_t rawtime;
//		struct tm* timeinfo = new tm();
//		char tbuffer[80];
//
//		time(&rawtime);
//
//#ifdef _WIN32
//		localtime_s(timeinfo, &rawtime);
//#else
//		localtime_r(&rawtime, timeinfo);
//#endif
//
//		strftime(tbuffer, 80, "%a %b %e %T %G", timeinfo);
//
//		sendto(socket_udp, tbuffer, sizeof(tbuffer), 0, (const sockaddr*)&client, sizeof(client));
//		delete timeinfo;
//	}
//	else if (commande.find("ping ") == 0) {												/*		PING		*/
//		sendto(socket_udp, "pong", sizeof("pong"), 0, (const sockaddr*)&client, sizeof(client));
//	}
//	else if (commande.find("usager ") == 0) {											/*		USAGER		*/
//		std::string user;
//
//#ifdef _WIN32
//		wchar_t userbuf[30];
//		DWORD usersize = 30;
//
//		GetUserNameW(userbuf, &usersize);
//
//		std::wstring wuser = userbuf;
//		user = std::string(wuser.begin(), wuser.end());
//#else
//		char ptr[30];
//		getlogin_r(ptr, sizeof(ptr) - 1);
//		user = std::string(ptr);
//#endif
//
//		sendto(socket_udp, user.c_str(), user.length(), 0, (const sockaddr*)&client, sizeof(client));
//	}
//	else if (commande.find("exec ") == 0) {												/*		EXEC		*/
//		std::string::size_type pos = commande.find(' ');
//		std::string parametres = commande.substr(pos + 1);
//
//		FILE* pipe = nullptr;
//		char buffer[301]; // 300 caractères + '\0'
//		std::string reponse;
//
//		pipe = popen(parametres.c_str(), "r");
//
//		if (!pipe)
//			reponse = "Erreur de commande!";
//		else while (!feof(pipe)) {
//			if (fgets(buffer, sizeof(buffer) - 1, pipe))
//				reponse += buffer;
//		}
//
//		if (pipe)
//			pclose(pipe);
//
//		if (reponse.length() > 300)
//			reponse = reponse.substr(0, 300);
//		else if (reponse.length() < 1)
//			reponse = "OK!";
//
//		sendto(socket_udp, reponse.c_str(), reponse.length(), 0, (const sockaddr*)&client, sizeof(client));
//	}
//	else if (commande.find("bye ") == 0) {											/*		BYE			*/
//		Log("", 0); // Message d'erreur pas de message d'erreur!
//		return 0;
//	}
//	else sendto(socket_udp, "huh?", sizeof("huh?"), 0, (const sockaddr*)&client, sizeof(client));		/*			DEFAULT			 */
//}
//
///* Ce bout de code ne devrait theoriquement jamais etre atteint, mais au cas. */
//Log("", 0);
//return 0;
//return false;
//}
//