69 lines
1.6 KiB
C++
69 lines
1.6 KiB
C++
#include "network.h"
|
|
|
|
bool initNetwork()
|
|
{
|
|
WSADATA wsaData;
|
|
if (0 != WSAStartup(MAKEWORD( 2, 0 ), &wsaData)) return false;
|
|
if (LOBYTE( wsaData.wVersion ) != 2 || HIBYTE( wsaData.wVersion ) != 0) return false;
|
|
return true;
|
|
}
|
|
|
|
void closeNetwork()
|
|
{
|
|
WSACleanup();
|
|
}
|
|
|
|
SOCKET clientConnect(SOCKET serverSocket)
|
|
{
|
|
SOCKET clientSocket = accept(serverSocket, NULL, NULL);
|
|
if (INVALID_SOCKET == clientSocket) return INVALID_SOCKET;
|
|
|
|
const char *expectedHandshake = "hello, synctracker!";
|
|
char recievedHandshake[19];
|
|
|
|
recv(clientSocket, recievedHandshake, int(strlen(expectedHandshake)), 0);
|
|
if (strncmp(expectedHandshake, recievedHandshake, strlen(expectedHandshake)) != 0)
|
|
{
|
|
closesocket(clientSocket);
|
|
return INVALID_SOCKET;
|
|
}
|
|
|
|
const char *test = "hello, demo!";
|
|
send(clientSocket, test, int(strlen(test)), 0);
|
|
|
|
return clientSocket;
|
|
}
|
|
|
|
SOCKET serverConnect(struct sockaddr_in *addr)
|
|
{
|
|
SOCKET serverSocket = socket( AF_INET, SOCK_STREAM, 0 );
|
|
connect( serverSocket,(struct sockaddr *)addr, sizeof(struct sockaddr_in));
|
|
|
|
const char * request_text = "hello, synctracker!";
|
|
send(serverSocket, request_text, int(strlen(request_text)), 0);
|
|
|
|
const char *expectedHandshake = "hello, demo!";
|
|
char recievedHandshake[12];
|
|
recv(serverSocket, recievedHandshake, int(sizeof(recievedHandshake)), 0);
|
|
if (strncmp(expectedHandshake, recievedHandshake, strlen(expectedHandshake)) != 0)
|
|
{
|
|
closesocket(serverSocket);
|
|
return INVALID_SOCKET;
|
|
}
|
|
|
|
return serverSocket;
|
|
}
|
|
|
|
bool pollRead(SOCKET socket)
|
|
{
|
|
struct timeval timeout = { 0, 0 };
|
|
fd_set fds;
|
|
|
|
FD_ZERO(&fds);
|
|
FD_SET(socket, &fds);
|
|
|
|
// look for new commands
|
|
return select(0, &fds, NULL, NULL, &timeout) > 0;
|
|
}
|
|
|