#include "stdafx.h"
#include "strEx.h"
#include "NSClientSocket.h"

/**
 * Default c-tor
 */
NSClientSocket::NSClientSocket(): hStopEvent(NULL) {
}

NSClientSocket::~NSClientSocket() {
}


/**
 * Thread procedure for the socket listener
 * @param lpParameter Potential argument to the thread proc.
 * @return thread exit status
 * @todo This needs to be reworked, possibly completely redone ?
 */
DWORD NSClientSocket::threadProc(LPVOID lpParameter)
{
	hStopEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
	if (!hStopEvent) {
		NSC_LOG_ERROR_STD("Create StopEvent failed: " + strEx::itos(GetLastError()));
		return 0;
	}

	WSADATA wsaData;
	sockaddr_in local;
	int wsaret=WSAStartup(0x101,&wsaData);
	if(wsaret!=0) {
		NSC_LOG_ERROR_STD("WSA Startup failed: " + strEx::itos(wsaret));
		return 0;
	}

	local.sin_family=AF_INET;
	local.sin_addr.s_addr=INADDR_ANY;
	local.sin_port=htons(static_cast<u_short>(NSCModuleHelper::getSettingsInt("NSClient", "port", DEFAULT_TCP_PORT)));
	server=socket(AF_INET,SOCK_STREAM,0);
	if(server==INVALID_SOCKET) {
		WSACleanup();
		NSC_LOG_ERROR_STD("Could not create listening socket: " + strEx::itos(GetLastError()));
		return 0;
	}

	if(bind(server,(sockaddr*)&local,sizeof(local))!=0) {
		closesocket(server);
		WSACleanup();
		NSC_LOG_ERROR_STD("Could not bind socket: " + strEx::itos(GetLastError()));
		return 0;
	}

	if(listen(server,10)!=0) {
		closesocket(server);
		WSACleanup();
		NSC_LOG_ERROR_STD("Could not open socket: " + strEx::itos(GetLastError()));
		return 0;
	}

	SOCKET client;
	sockaddr_in from;
	int fromlen=sizeof(from);
	while (!(WaitForSingleObject(hStopEvent, 100) == WAIT_OBJECT_0)) {
		client=accept(server, (struct sockaddr*)&from,&fromlen);
		if (client != INVALID_SOCKET) 
			onAccept(client);
	}
	closesocket(server);
	WSACleanup();
	NSC_DEBUG_MSG("Socket closed!");
	return 0;
}

#define RECV_BUFFER_LEN 1024

std::string NSClientSocket::parseRequest(char *buffer)  {
	charEx::token pwd = charEx::getToken(buffer, '&');
	if ( (pwd.first.empty()) || (pwd.first != NSCModuleHelper::getSettingsString("generic", "password", "")) )
		return "ERRRO: Invalid password.";
	if (pwd.second == NULL)
		return "ERRRO: No command specified.";
	charEx::token cmd = charEx::getToken(pwd.second, '&');
	NSC_DEBUG_MSG("Command: " + cmd.first);
	std::string message, perf;
	NSCAPI::nagiosReturn ret = NSCModuleHelper::InjectSplitAndCommand(cmd.first.c_str(), cmd.second, '&', message, perf);
	int c = atoi(cmd.first.c_str());
	switch (c) {
		case REQ_CLIENTVERSION:
		case REQ_UPTIME:
		case REQ_CPULOAD:
		case REQ_USEDDISKSPACE:
			return message;
		case REQ_SERVICESTATE:
		case REQ_PROCSTATE:
			return NSCHelper::translateReturn(ret) + "&" + message;
		default:
			return NSCHelper::translateReturn(ret) + "&" + message + "|" + perf;
	}
}

void NSClientSocket::onAccept(SOCKET client) {
	char *buff = new char[RECV_BUFFER_LEN+1];
	int n=recv(client,buff,RECV_BUFFER_LEN,0);
	if ((n!=SOCKET_ERROR )&&(n > 0)&&(n < RECV_BUFFER_LEN)) {
		buff[n] = '\0';
		NSC_DEBUG_MSG("Incoming data: ");
		NSC_DEBUG_MSG(buff);
		std::string response = parseRequest(buff);	
		NSC_DEBUG_MSG("Outgoing data: " + response);
		send(client, response.c_str(), static_cast<int>(response.length()), 0);
	} else {
		std::string str = "ERROR: Unknown socket error";
		send(client,str.c_str(),static_cast<int>(str.length()),0);
	}
	delete [] buff;
	closesocket(client);
}


/**
 * Exit thread callback proc. 
 * This is called by the thread manager when the thread should initiate a shutdown procedure.
 * The thread manager is responsible for waiting for the actual termination of the thread.
 */
void NSClientSocket::exitThread(void) {
	NSC_DEBUG_MSG("Requesting shutdown!");
	if (!SetEvent(hStopEvent)) {
		NSC_LOG_ERROR_STD("SetStopEvent failed");
	}
}