#include "stdafx.h"

#include "NetworkSocketManager.h"

#include <sys/socket.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/tcp.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <sys/time.h>
#include <sys/select.h>

// PIEBOT: Adding this because Sony systems uses BSD apis
//		   which uses socketclose instead of closesocket.
int NetworkSocketManager::CloseSocket(SOCKET sock)
{
	return socketclose(sock);
}

int NetworkSocketManager::getSocketError(SOCKET socket)
{
	return socket;
}

bool NetworkSocketManager::IsSocketValid(SOCKET sock)
{
	return sock >= 0;
}

bool NetworkSocketManager::IsSentValid(int sent)
{
	return sent >= 0;
}

bool NetworkSocketManager::InitNetwork()
{
	WSADATA wsaData;
	int result = WSAStartup(MAKEWORD(2, 2), &wsaData);
	if (result != 0)
	{
		app.DebugPrintf("WSAStartup failed: %d\n", result);
		return false;
	}
}

SOCKET NetworkSocketManager::MakeHostSocket(int port)
{
	SOCKET s_listenSocket = NULL;

	struct sockaddr_in addr = {};

	addr.sin_family = AF_INET;
	addr.sin_addr.s_addr = htonl(INADDR_ANY);
	addr.sin_port = htons((uint16_t)port);

	s_listenSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
	if (s_listenSocket < 0)
	{
		app.DebugPrintf("socket() failed: %d\n", s_listenSocket);
		return NULL;
	}

	// PIEBOT: time for part 2, binding and listening.
	int opt = 1;
	setsockopt(s_listenSocket, SOL_SOCKET, SO_REUSEADDR, (const char *)&opt, sizeof(opt));

	int iResult = ::bind(s_listenSocket, (struct sockaddr*)&addr, sizeof(addr));

	if (iResult < 0)
	{
		app.DebugPrintf("bind() failed: %d\n", iResult);
		closesocket(s_listenSocket);
		return INVALID_SOCKET;
	}

	iResult = listen(s_listenSocket, 5);
	if (iResult < 0)
	{
		app.DebugPrintf("listen() failed: %d\n", iResult);
		closesocket(s_listenSocket);
		return INVALID_SOCKET;
	}

	return s_listenSocket;
}