forked from cafeberry/cafeberry
92 lines
2.1 KiB
C++
92 lines
2.1 KiB
C++
#include "stdafx.h"
|
|
|
|
#include "NetworkSocketManager.h"
|
|
|
|
#include <WinSock2.h>
|
|
#include <WS2tcpip.h>
|
|
|
|
// PIEBOT: Adding this because Sony systems uses BSD apis
|
|
// which uses socketclose instead of closesocket.
|
|
int NetworkSocketManager::CloseSocket(SOCKET sock)
|
|
{
|
|
return closesocket(sock);
|
|
}
|
|
|
|
int NetworkSocketManager::getSocketError(SOCKET socket)
|
|
{
|
|
return WSAGetLastError();
|
|
}
|
|
|
|
bool NetworkSocketManager::IsSocketValid(SOCKET sock)
|
|
{
|
|
return sock != INVALID_SOCKET;
|
|
}
|
|
|
|
bool NetworkSocketManager::IsSentValid(int sent)
|
|
{
|
|
return sent != SOCKET_ERROR;
|
|
}
|
|
|
|
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 addrinfo hints = {};
|
|
struct addrinfo *result = NULL;
|
|
|
|
hints.ai_family = AF_INET;
|
|
hints.ai_socktype = SOCK_STREAM;
|
|
hints.ai_protocol = IPPROTO_TCP;
|
|
hints.ai_flags = AI_PASSIVE;
|
|
|
|
char portStr[16];
|
|
sprintf_s(portStr, "%d", port);
|
|
|
|
int iResult = getaddrinfo(NULL, portStr, &hints, &result);
|
|
if (iResult != 0)
|
|
{
|
|
app.DebugPrintf("getaddrinfo failed: %d\n", iResult);
|
|
return INVALID_SOCKET;
|
|
}
|
|
|
|
s_listenSocket = socket(result->ai_family, result->ai_socktype, result->ai_protocol);
|
|
if (s_listenSocket == INVALID_SOCKET)
|
|
{
|
|
app.DebugPrintf("socket() failed: %d\n", WSAGetLastError());
|
|
freeaddrinfo(result);
|
|
return INVALID_SOCKET;
|
|
}
|
|
|
|
// PIEBOT: time for part 2, binding and listening.
|
|
int opt = 1;
|
|
setsockopt(s_listenSocket, SOL_SOCKET, SO_REUSEADDR, (const char *)&opt, sizeof(opt));
|
|
|
|
iResult = ::bind(s_listenSocket, result->ai_addr, (int)result->ai_addrlen);
|
|
freeaddrinfo(result);
|
|
if (iResult == SOCKET_ERROR)
|
|
{
|
|
app.DebugPrintf("bind() failed: %d\n", WSAGetLastError());
|
|
closesocket(s_listenSocket);
|
|
return INVALID_SOCKET;
|
|
}
|
|
|
|
iResult = listen(s_listenSocket, SOMAXCONN);
|
|
if (iResult == SOCKET_ERROR)
|
|
{
|
|
app.DebugPrintf("listen() failed: %d\n", WSAGetLastError());
|
|
closesocket(s_listenSocket);
|
|
return INVALID_SOCKET;
|
|
}
|
|
|
|
return s_listenSocket;
|
|
} |