76 lines
1.8 KiB
C++
76 lines
1.8 KiB
C++
#include "../stdafx.h"
|
|
#include "network.h"
|
|
#include <stdio.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();
|
|
}
|
|
|
|
static const char *clientGreeting = "hello, synctracker!";
|
|
static const char *serverGreeting = "hello, demo!";
|
|
|
|
SOCKET clientConnect(SOCKET serverSocket)
|
|
{
|
|
SOCKET clientSocket = accept(serverSocket, NULL, NULL);
|
|
if (INVALID_SOCKET == clientSocket) return INVALID_SOCKET;
|
|
printf("%x\n", clientSocket);
|
|
|
|
const char *expectedGreeting = clientGreeting;
|
|
char recievedGreeting[128];
|
|
|
|
recv(clientSocket, recievedGreeting, int(strlen(expectedGreeting)), 0);
|
|
|
|
if (strncmp(expectedGreeting, recievedGreeting, strlen(expectedGreeting)) != 0)
|
|
{
|
|
closesocket(clientSocket);
|
|
return INVALID_SOCKET;
|
|
}
|
|
|
|
const char *greeting = serverGreeting;
|
|
send(clientSocket, greeting, int(strlen(greeting)), 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 *greeting = clientGreeting;
|
|
send(serverSocket, greeting, int(strlen(greeting)), 0);
|
|
|
|
const char *expectedGreeting = serverGreeting;
|
|
char recievedGreeting[128];
|
|
recv(serverSocket, recievedGreeting, int(sizeof(recievedGreeting)), 0);
|
|
if (strncmp(expectedGreeting, recievedGreeting, strlen(expectedGreeting)) != 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;
|
|
}
|
|
|