diff --git a/Code/Network/NetworkDependencies/Connection.cpp b/Code/Network/NetworkDependencies/Connection.cpp index fdd89ef4..522caa24 100644 --- a/Code/Network/NetworkDependencies/Connection.cpp +++ b/Code/Network/NetworkDependencies/Connection.cpp @@ -2,6 +2,7 @@ #include #include +#include using namespace Oyster::Network; @@ -10,13 +11,12 @@ Connection::~Connection() closesocket( this->socket ); } -bool Connection::Connect(unsigned short port , const char serverName[]) +int Connection::Connect(unsigned short port , const char serverName[]) { struct hostent *hostEnt; if((hostEnt = gethostbyname(serverName)) == NULL) { - //couldn't find host - return false; + return WSAGetLastError(); } struct sockaddr_in server; @@ -26,20 +26,20 @@ bool Connection::Connect(unsigned short port , const char serverName[]) if(connect(this->socket, (sockaddr*)&server, sizeof(server)) == SOCKET_ERROR) { - //Error connecting to server - return false; + return WSAGetLastError(); } //connection succesfull! - return true; + return 0; } -bool Connection::InitiateServer(unsigned short port) +int Connection::InitiateServer(unsigned short port) { - if(!initiateSocket()) + int errorCode = 0; + + if((errorCode = initiateSocket()) != 0) { - //Error opening socket! - return false; + return errorCode; } struct sockaddr_in server; @@ -49,36 +49,35 @@ bool Connection::InitiateServer(unsigned short port) if(bind(this->socket, (sockaddr*)&server, sizeof(server)) == SOCKET_ERROR) { - //Bind failed!; + errorCode = WSAGetLastError(); closesocket(this->socket); - return false; + return errorCode; } //not our Listen function! its trying to keep our socket open for connections if(listen(this->socket, 5) == SOCKET_ERROR) { - //"Listen failed! + errorCode = WSAGetLastError(); closesocket(this->socket); - return false; + return errorCode; } //Server started! - return true; + return 0; } -bool Connection::InitiateClient() +int Connection::InitiateClient() { - if(!initiateSocket()) - { - return false; - } + int errorCode; + return initiateSocket(); - return true; } -void Connection::Disconnect() +int Connection::Disconnect() { closesocket(this->socket); + + return WSAGetLastError(); } bool Connection::Send(OysterByte& bytes) @@ -88,13 +87,10 @@ bool Connection::Send(OysterByte& bytes) nBytes = send(this->socket, bytes, bytes.GetSize(), 0); if(nBytes == SOCKET_ERROR) { - //Send failed! - return false; + return WSAGetLastError(); } - std::cout << "Size of the sent data: " << nBytes << " bytes" << std::endl; - - return true; + return 0; } int Connection::Recieve(OysterByte& bytes) @@ -105,8 +101,7 @@ int Connection::Recieve(OysterByte& bytes) nBytes = recv(this->socket, bytes, 500, 0); if(nBytes == SOCKET_ERROR) { - //Recv failed - return -1; + return WSAGetLastError(); } else { @@ -117,7 +112,7 @@ int Connection::Recieve(OysterByte& bytes) //bytes.byteArray[nBytes] = '\0'; - return 1; + return 0; } int Connection::Listen() @@ -125,8 +120,7 @@ int Connection::Listen() int clientSocket; if((clientSocket = accept(this->socket, NULL, NULL)) == INVALID_SOCKET) { - //failed - return -1; + return WSAGetLastError(); } return clientSocket; @@ -135,14 +129,13 @@ int Connection::Listen() /////////////////////////////////////// //Private functions /////////////////////////////////////// -bool Connection::initiateSocket() +int Connection::initiateSocket() { this->socket = ::socket(AF_INET, SOCK_STREAM, 0); if(this->socket == SOCKET_ERROR) { - //error opening socket - return false; + return WSAGetLastError(); } - return true; + return 0; } diff --git a/Code/Network/NetworkDependencies/Connection.h b/Code/Network/NetworkDependencies/Connection.h index 05b35070..ddb3f8b7 100644 --- a/Code/Network/NetworkDependencies/Connection.h +++ b/Code/Network/NetworkDependencies/Connection.h @@ -20,18 +20,18 @@ namespace Oyster Connection(int socket) { this->socket = socket; }; virtual ~Connection(); - virtual bool InitiateServer( unsigned short port ); - virtual bool InitiateClient(); + virtual int InitiateServer( unsigned short port ); + virtual int InitiateClient(); virtual bool Send( OysterByte& bytes ); virtual int Recieve( OysterByte& bytes ); - virtual void Disconnect(); - virtual bool Connect( unsigned short port , const char serverName[] ); + virtual int Disconnect(); + virtual int Connect( unsigned short port , const char serverName[] ); virtual int Listen(); private: - bool initiateSocket(); + int initiateSocket(); int socket; diff --git a/Code/Network/NetworkDependencies/IConnection.h b/Code/Network/NetworkDependencies/IConnection.h index 85b63b77..e836a9c0 100644 --- a/Code/Network/NetworkDependencies/IConnection.h +++ b/Code/Network/NetworkDependencies/IConnection.h @@ -14,13 +14,13 @@ namespace Oyster { public: - virtual void Disconnect() = 0; + virtual int Disconnect() = 0; virtual bool Send( OysterByte& bytes ) = 0; virtual int Recieve( OysterByte& bytes) = 0; - virtual bool InitiateServer( unsigned short port ) { return false; }; - virtual bool InitiateClient() { return false; }; + virtual int InitiateServer( unsigned short port ) { return false; }; + virtual int InitiateClient() { return false; }; virtual int Listen() { return -1; }; - virtual bool Connect( unsigned short port, const char serverName[] ) { return false; }; + virtual int Connect( unsigned short port, const char serverName[] ) { return false; }; }; } } diff --git a/Code/Network/NetworkDependencies/WinsockFunctions.cpp b/Code/Network/NetworkDependencies/WinsockFunctions.cpp index 92382ee9..3419965b 100644 --- a/Code/Network/NetworkDependencies/WinsockFunctions.cpp +++ b/Code/Network/NetworkDependencies/WinsockFunctions.cpp @@ -1,13 +1,37 @@ #include "WinsockFunctions.h" #include -bool InitSockets() +bool InitWinSock() { WSADATA wsaData; return WSAStartup(MAKEWORD(2, 2), &wsaData) == NO_ERROR; } -void ShutdownSockets() +void ShutdownWinSock() { WSACleanup(); +} + +std::wstring GetErrorMessage(int errorCode) +{ + LPWSTR lpMessage; + std::wstring retVal(L"Succesful"); + + DWORD bufLen = FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS , + NULL, + errorCode , + MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT) , + (LPWSTR)&lpMessage, + 0 , + NULL ); + + if(bufLen) + { + retVal = lpMessage; + + LocalFree(lpMessage); + + return retVal; + } + } \ No newline at end of file diff --git a/Code/Network/NetworkDependencies/WinsockFunctions.h b/Code/Network/NetworkDependencies/WinsockFunctions.h index a2a3eee0..92c8957a 100644 --- a/Code/Network/NetworkDependencies/WinsockFunctions.h +++ b/Code/Network/NetworkDependencies/WinsockFunctions.h @@ -4,8 +4,12 @@ ///////////////////////////////////// // Created by Pontus Fransson 2013 // ///////////////////////////////////// +#include -void ShutdownSockets(); -bool InitSockets(); + +void ShutdownWinSock(); +bool InitWinSock(); + +std::wstring GetErrorMessage(int errorCode); #endif \ No newline at end of file diff --git a/Code/Network/OysterNetworkClient/Client.cpp b/Code/Network/OysterNetworkClient/Client.cpp index 1c6357b9..daffe9b6 100644 --- a/Code/Network/OysterNetworkClient/Client.cpp +++ b/Code/Network/OysterNetworkClient/Client.cpp @@ -13,12 +13,21 @@ Client::~Client() connection = 0; } -bool Client::Connect(unsigned int port, char filename[]) +int Client::Connect(unsigned int port, char filename[]) { - connection->InitiateClient(); - connection->Connect(port, filename); + int errorCode; - return true; + if((errorCode = connection->InitiateClient()) != 0) + { + return errorCode; + } + + if((errorCode = connection->Connect(port, filename)) != 0) + { + return errorCode; + } + + return 0; } void Client::Send(Oyster::Network::OysterByte& bytes) diff --git a/Code/Network/OysterNetworkClient/Client.h b/Code/Network/OysterNetworkClient/Client.h index 0ecb2a32..6e69e657 100644 --- a/Code/Network/OysterNetworkClient/Client.h +++ b/Code/Network/OysterNetworkClient/Client.h @@ -20,7 +20,7 @@ namespace Oyster Client(); ~Client(); - bool Connect(unsigned int port, char filename[]); + int Connect(unsigned int port, char filename[]); void Send(OysterByte& bytes); void Recv(OysterByte& bytes); diff --git a/Code/Network/OysterNetworkClient/ClientMain.cpp b/Code/Network/OysterNetworkClient/ClientMain.cpp index 472ae10a..244ce1d4 100644 --- a/Code/Network/OysterNetworkClient/ClientMain.cpp +++ b/Code/Network/OysterNetworkClient/ClientMain.cpp @@ -17,9 +17,11 @@ void chat(Client &client); int main() { + int errorCode; + char msgRecv[255] = "\0"; - InitSockets(); + InitWinSock(); cout << "Client" << endl; @@ -27,11 +29,17 @@ int main() Client client; //Connect to server - client.Connect(9876, "localhost"); + errorCode = client.Connect(9876, "localhost"); + + if(errorCode != 0) + { + wstring errorTest = GetErrorMessage(errorCode); + wcout << "errorMessage: " << errorTest << endl; + } chat(client); - ShutdownSockets(); + ShutdownWinSock(); system("pause"); return 0; diff --git a/Code/Network/OysterNetworkServer/ServerMain.cpp b/Code/Network/OysterNetworkServer/ServerMain.cpp index 94e854f5..f7b358ed 100644 --- a/Code/Network/OysterNetworkServer/ServerMain.cpp +++ b/Code/Network/OysterNetworkServer/ServerMain.cpp @@ -20,10 +20,11 @@ int main() cout << "Server" << endl; Translator t; + int errorCode; - if(!InitSockets()) + if(!InitWinSock()) { - cout << "Sockets failed to initialize" << endl; + cout << "errorMessage: unable to start winsock" << endl; } //Create socket @@ -88,7 +89,8 @@ int main() client1.Send(recvBuffer); } - ShutdownSockets(); + + ShutdownWinSock(); delete set; system("pause");