Version 4.0

src/Socket.cpp

@@ -0,0 +1,240 @@
+/***************************************************/
+/*! \class Socket
+    \brief STK TCP socket client/server class.
+
+    This class provides a uniform cross-platform
+    TCP socket client or socket server interface.
+    Methods are provided for reading or writing
+    data buffers to/from connections.  This class
+    also provides a number of static functions for
+    use with external socket descriptors.
+
+    The user is responsible for checking the values
+    returned by the read/write methods.  Values
+    less than or equal to zero indicate a closed
+    or lost connection or the occurence of an error.
+
+    by Perry R. Cook and Gary P. Scavone, 1995 - 2002.
+*/
+/***************************************************/
+
+#include "Socket.h"
+#include <stdio.h>
+#include <string.h>
+
+#if (defined(__OS_IRIX__) || defined(__OS_LINUX__))
+
+  #include <sys/socket.h>
+  #include <sys/types.h>
+  #include <arpa/inet.h>
+  #include <netdb.h>
+  #include <unistd.h>
+  #include <fcntl.h>
+
+#elif defined(__OS_WINDOWS__)
+
+  #include <winsock.h>
+
+#endif
+
+Socket :: Socket( int port )
+{
+  soket = -1;
+  server = true;
+  poort = port;
+
+  // Create a socket server.
+#if defined(__OS_WINDOWS__)  // windoze-only stuff
+  WSADATA wsaData;
+  WORD wVersionRequested = MAKEWORD(1,1);
+
+  WSAStartup(wVersionRequested, &wsaData);
+  if (wsaData.wVersion != wVersionRequested) {
+    sprintf( msg, "Socket: Incompatible Windows socket library version!" );
+    handleError( msg, StkError::PROCESS_SOCKET );
+  }
+#endif
+
+  // Create the server-side socket
+  soket = ::socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
+  if (soket < 0) {
+    sprintf(msg, "Socket: Couldn't create socket server!");
+    handleError( msg, StkError::PROCESS_SOCKET );
+  }
+
+  struct sockaddr_in mysocket;
+  mysocket.sin_family=AF_INET;
+  mysocket.sin_addr.s_addr=INADDR_ANY;
+  mysocket.sin_port=htons( port );
+
+  // Bind socket to the appropriate port and interface (INADDR_ANY)
+  if (bind(soket, (struct sockaddr *)&mysocket, sizeof(mysocket)) < 0) {
+    sprintf(msg, "Socket: Couldn't bind socket!");
+    handleError( msg, StkError::PROCESS_SOCKET );
+  }
+
+  // Listen for incoming connection(s)
+  if ( listen(soket, 1) < 0 ) {
+    sprintf(msg, "Socket: Couldn't start server listening!");
+    handleError( msg, StkError::PROCESS_SOCKET );
+  }
+}
+
+Socket :: Socket(int port, const char *hostname )
+{
+  soket = -1;
+  server = false;
+  poort = port;
+
+#if defined(__OS_WINDOWS__)  // windoze-only stuff
+  WSADATA wsaData;
+  WORD wVersionRequested = MAKEWORD(1,1);
+
+  WSAStartup(wVersionRequested, &wsaData);
+  if (wsaData.wVersion != wVersionRequested) {
+    sprintf( msg, "Socket: Incompatible Windows socket library version!" );
+    handleError( msg, StkError::PROCESS_SOCKET );
+  }
+#endif
+
+  // Create a socket client connection.
+  connect( port, hostname );
+}
+
+Socket :: ~Socket()
+{
+#if (defined(__OS_IRIX__) || defined(__OS_LINUX__))
+
+  ::close( soket );
+
+#elif defined(__OS_WINDOWS__)
+
+  ::closesocket( soket );
+  WSACleanup();
+
+#endif
+}
+
+int Socket :: connect( int port, const char *hostname )
+{
+  // This method is for client connections only!
+  if ( server == true ) return -1;
+
+  // Close an existing connection if it exists.
+  if ( isValid( soket ) ) this->close();
+
+  // Create the client-side socket
+  soket = ::socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
+  if (soket < 0) {
+    sprintf(msg, "Socket: Couldn't create socket client!");
+    handleError( msg, StkError::PROCESS_SOCKET );
+  }
+
+  struct hostent *hostp;
+  if ( (hostp = gethostbyname(hostname)) == 0 ) {
+    sprintf(msg, "Socket: unknown host (%s)!", hostname);
+    handleError( msg, StkError::PROCESS_SOCKET_IPADDR );
+  }
+
+  // Fill in the address structure
+  struct sockaddr_in server_address;
+  server_address.sin_family = AF_INET;
+  memcpy((void *)&server_address.sin_addr, hostp->h_addr, hostp->h_length);
+  server_address.sin_port = htons(port);
+
+  // Connect to the server
+  if ( ::connect(soket, (struct sockaddr *)&server_address,
+                 sizeof(server_address) ) < 0) {
+    sprintf(msg, "Socket: Couldn't connect to socket server!");
+    handleError( msg, StkError::PROCESS_SOCKET );
+  }
+
+  return soket;
+}
+
+int Socket :: socket( void ) const
+{
+  return soket;
+}
+
+int Socket :: port( void ) const
+{
+  return poort;
+}
+
+int Socket :: accept( void )
+{
+  if ( server )
+    return ::accept( soket, NULL, NULL );
+  else
+    return -1;
+}
+
+bool Socket :: isValid( int socket )
+{
+  return socket != -1;
+}
+
+void Socket :: setBlocking( int socket, bool enable )
+{
+  if ( !isValid( socket ) ) return;
+
+#if (defined(__OS_IRIX__) || defined(__OS_LINUX__))
+
+  int tmp = ::fcntl(socket, F_GETFL, 0);
+  if ( tmp >= 0 )
+    tmp = ::fcntl( socket, F_SETFL, enable ? (tmp &~ O_NONBLOCK) : (tmp | O_NONBLOCK) );
+
+#elif defined(__OS_WINDOWS__)
+
+  unsigned long non_block = !enable;
+  ioctlsocket( socket, FIONBIO, &non_block );
+
+#endif
+}
+
+void Socket :: close( void )
+{
+  if ( !isValid( soket ) ) return;
+  this->close( soket );
+  soket = -1;
+}
+
+void Socket :: close( int socket )
+{
+  if ( !isValid( socket ) ) return;
+
+#if (defined(__OS_IRIX__) || defined(__OS_LINUX__))
+
+  ::close( socket );
+
+#elif defined(__OS_WINDOWS__)
+
+  ::closesocket( socket );
+
+#endif
+}
+
+int Socket :: writeBuffer(const void *buffer, long bufferSize, int flags )
+{
+  if ( !isValid( soket ) ) return -1;
+  return send( soket, (const char *)buffer, bufferSize, flags );
+}
+
+int Socket :: writeBuffer(int socket, const void *buffer, long bufferSize, int flags )
+{
+  if ( !isValid( socket ) ) return -1;
+  return send( socket, (const char *)buffer, bufferSize, flags );
+}
+
+int Socket :: readBuffer(void *buffer, long bufferSize, int flags )
+{
+  if ( !isValid( soket ) ) return -1;
+  return recv( soket, (char *)buffer, bufferSize, flags );
+}
+
+int Socket :: readBuffer(int socket, void *buffer, long bufferSize, int flags )
+{
+  if ( !isValid( socket ) ) return -1;
+  return recv( socket, (char *)buffer, bufferSize, flags );
+}