Initial commit

CrossSocket.cpp

@@ -0,0 +1,252 @@
+// CrossSockets
+// Copyright Jeroen Saey
+// Created 27-01-2013
+// CrossSocket.cpp
+
+#include "CrossSocket.h"
+
+#ifndef __WIN32__
+  #include <netdb.h>
+  #include <arpa/inet.h>
+  #include <fcntl.h>
+  #include <sys/select.h>
+  #include <sys/time.h>
+#else
+  #define F_SETFL FIONBIO
+  #define O_NONBLOCK 1
+  
+  extern int close(int fd);
+  extern int fcntl(int fd, int cmd, long arg);
+#endif
+
+using namespace std;
+
+CrossSocket::CrossSocket(blockTypeEnum block)
+{
+    _blockingMode = block;    
+}
+
+CrossSocket::~CrossSocket()
+{
+
+}
+
+void CrossSocket::getSocket()
+{
+    if (_socketHandle < 0)
+    {
+        _socketHandle = socket(PF_INET, SOCK_STREAM, 0);
+
+        if (_blockingMode == nonblocking)
+        {
+            fcntl(_socketHandle, F_SETFL, O_NONBLOCK);
+        }
+
+        //reset state	
+        reset();
+    }
+}
+
+BaseCrossSocket* CrossSocket::create(int socketdescriptor, CrossSocketErrors *error)
+{
+    CrossSocket* remoteClass;
+
+    remoteClass = new CrossSocket(_blockingMode);
+    remoteClass->_socketHandle = socketdescriptor;
+
+    noError(error);
+    return remoteClass;
+}
+
+bool CrossSocket::bind(int port, CrossSocketErrors *error)
+{
+    return bind(port, "", error);
+}
+
+bool CrossSocket::bind(int port, string host, CrossSocketErrors *error)
+{
+    hostent *hostEntry;
+    in_addr networkAddress; 	
+
+    if (host.size() > 0 )
+    {
+        // Bind to a specific address
+
+        if ((hostEntry = gethostbyname(host.data())) == NULL)
+        {
+            setError(error, fatal, "CrossSocket::bind() - Can't get host by name");
+            return false;
+        }
+
+        networkAddress = *((in_addr *)hostEntry->h_addr);
+     }
+     else
+     {
+        // Bind to any address
+        networkAddress.s_addr = INADDR_ANY;
+     }
+
+    getSocket();
+
+    sockaddr_in serverAddress;
+
+    memset(&serverAddress, 0, sizeof(serverAddress));
+    serverAddress.sin_family = AF_INET;
+    serverAddress.sin_port = htons(port);
+    serverAddress.sin_addr.s_addr = networkAddress.s_addr;
+
+    if (::bind(_socketHandle, (sockaddr *)&serverAddress, sizeof(serverAddress)) == -1)
+    {
+        handleError(error, "CrossSocket::bind() error: ");
+        return false;
+    }
+
+    noError(error);
+    return true;
+}
+
+bool CrossSocket::connect(int port, string hostname, CrossSocketErrors *error)
+{
+    getSocket();
+
+    hostent *host;
+
+    if ((host = gethostbyname(hostname.data())) == NULL )
+    {
+        setError(error, fatal, "CrossSocket::connect() - Can't get host by name");
+        return false;
+    }
+
+    sockaddr_in clientAddress;
+
+    memset(&clientAddress, 0, sizeof(clientAddress));
+    clientAddress.sin_family = AF_INET;
+    clientAddress.sin_port = htons(port);
+    clientAddress.sin_addr = *((in_addr *)host->h_addr);
+
+    if (::connect(_socketHandle, (sockaddr *)&clientAddress, sizeof(clientAddress)) == -1)
+    {
+        handleError(error, "CrossSocket::connect() error: ");
+        return false;
+    }
+    
+    noError(error);
+    return true;
+}
+
+
+string CrossSocket::getClientAddress(CrossSocketErrors *error)
+{
+    sockaddr_in clientAddress;
+
+    if (!getClientHost((sockaddr *)&clientAddress, error))
+    {
+        return "";
+    }
+
+    char *addressPointer;
+
+    if ((addressPointer = inet_ntoa(clientAddress.sin_addr)) == NULL )
+    {
+        setError(error, fatal, "CrossSocket::getClientHostName() - Can't get client address");
+        return "";
+    }
+    
+    noError(error);
+    return string (addressPointer);
+}
+
+int CrossSocket::getClientPort(CrossSocketErrors *error)
+{
+    sockaddr_in clientAddress;
+
+    if (!getClientHost((sockaddr *)&clientAddress, error))
+    {
+        return -1;
+    }
+
+    noError(error);
+
+    return ntohs(clientAddress.sin_port);
+}
+
+string CrossSocket::getClientHostName(CrossSocketErrors *error)
+{
+    string name = getClientAddress(error);
+    
+    if (name.size() < 1)
+    {
+        return "";
+    }
+
+    hostent *host; 	
+
+    if ((host = gethostbyname(name.data())) == NULL )
+    {
+        setError(error, fatal, "CrossSocket::getClientHostName() - Can't get client by address");
+        return "";
+    }
+    
+    noError(error);
+    return string(host->h_name);;
+}
+
+string CrossSocket::getServerAddress(CrossSocketErrors *error)
+{
+    //We need to get the real address, so we must
+    //first get this computers host name and then
+    //translate that into an address!
+
+    string name = getServerHostName(error);
+    if (name.size() < 1)
+    {
+        return "";
+    }
+
+    hostent *host;
+
+    if ((host = gethostbyname(name.data())) == NULL )
+    {
+        setError(error, fatal, "CrossSocket::getServerAddress() - Can't get host by name");
+        return "";
+    }
+
+    char *addressPointer;
+
+    if ((addressPointer = inet_ntoa(*((in_addr *)host->h_addr))) == NULL)
+    {
+        setError(error, fatal, "CrossSocket::getServerAddress() - Can't get host address");
+        return "";
+    }
+    return string(addressPointer);
+}
+
+int CrossSocket::getServerPort(CrossSocketErrors *error)
+{
+    sockaddr_in address;
+
+    if (!getServerHost((sockaddr *)&address, error))
+    {
+        return -1;
+    }
+
+    noError(error);
+
+    return ntohs(address.sin_port);
+}
+
+string CrossSocket::getServerHostName(CrossSocketErrors *error)
+{
+    char buffer[256];
+
+    if (gethostname(buffer, 256) != 0)
+    {
+        handleError(error, "CrossSocket::gethostname() error: ");
+        return "";
+    }
+
+    string message(buffer);
+
+    noError(error);
+    return message;
+}