src/unix/gsocket.c

   1 /* -------------------------------------------------------------------------
   2  * Project: GSocket (Generic Socket) for WX
   3  * Name:    gsocket.c
   4  * Authors: Guilhem Lavaux,
   5  *          Guillermo Rodriguez Garcia <guille@iies.es> (maintainer)
   6  * Purpose: GSocket main Unix file
   7  * CVSID:   $Id$
   8  * -------------------------------------------------------------------------
   9  */
  10
  11 /*
  12  * PLEASE don't put C++ comments here - this is a C source file.
  13  */
  14
  15 #ifndef __GSOCKET_STANDALONE__
  16 #include "wx/setup.h"
  17 #endif
  18
  19 #if wxUSE_SOCKETS || defined(__GSOCKET_STANDALONE__)
  20
  21 #include <assert.h>
  22 #include <sys/types.h>
  23 #include <netdb.h>
  24 #include <sys/ioctl.h>
  25
  26 #ifdef __VMS__
  27 #include <socket.h>
  28 struct sockaddr_un {
  29     u_char  sun_len;        /* sockaddr len including null */
  30     u_char  sun_family;     /* AF_UNIX */
  31     char    sun_path[108];  /* path name (gag) */
  32 };
  33 #else
  34 #include <sys/socket.h>
  35 #include <sys/un.h>
  36 #endif
  37
  38 #include <sys/time.h>
  39 #include <netinet/in.h>
  40 #include <arpa/inet.h>
  41 #include <errno.h>
  42 #include <string.h>
  43 #include <unistd.h>
  44 #include <stdio.h>
  45 #include <stdlib.h>
  46 #include <stddef.h>
  47 #include <ctype.h>
  48 #ifdef sun
  49 #  include <sys/filio.h>
  50 #endif
  51 #ifdef sgi
  52 #  include <bstring.h>
  53 #endif
  54 #include <signal.h>
  55
  56 #ifndef SOCKLEN_T
  57
  58 #ifdef VMS
  59 #  define SOCKLEN_T unsigned int
  60 #else
  61 #  ifdef __GLIBC__
  62 #    if __GLIBC__ == 2
  63 #      define SOCKLEN_T socklen_t
  64 #    endif
  65 #  else
  66 #    define SOCKLEN_T int
  67 #  endif
  68 #endif
  69
  70 #endif /* SOCKLEN_T */
  71
  72 /*
  73  * MSW defines this, Unices don't.
  74  */
  75 #ifndef INVALID_SOCKET
  76 #define INVALID_SOCKET -1
  77 #endif
  78
  79 /*
  80  * INADDR_BROADCAST is identical to INADDR_NONE which is not defined
  81  * on all systems. INADDR_BROADCAST should be fine to indicate an error.
  82  */
  83 #ifndef INADDR_NONE
  84 #define INADDR_NONE INADDR_BROADCAST
  85 #endif
  86
  87 #define MASK_SIGNAL()                       \
  88 {                                           \
  89   void (*old_handler)(int);                 \
  90                                             \
  91   old_handler = signal(SIGPIPE, SIG_IGN);
  92
  93 #define UNMASK_SIGNAL()                     \
  94   signal(SIGPIPE, old_handler);             \
  95 }
  96
  97
  98 #ifndef __GSOCKET_STANDALONE__
  99 #  include "wx/unix/gsockunx.h"
 100 #  include "wx/gsocket.h"
 101 #else
 102 #  include "gsockunx.h"
 103 #  include "gsocket.h"
 104 #endif /* __GSOCKET_STANDALONE__ */
 105
 106 /* redefine some GUI-only functions to do nothing in console mode */
 107 #if defined(wxUSE_GUI) && !wxUSE_GUI
 108 #  define _GSocket_GUI_Init(socket) (1)
 109 #  define _GSocket_GUI_Destroy(socket)
 110 #  define _GSocket_Enable_Events(socket)
 111 #  define _GSocket_Disable_Events(socket)
 112 #  define _GSocket_Install_Callback(socket, event)
 113 #  define _GSocket_Uninstall_Callback(socket, event)
 114 #endif /* wxUSE_GUI */
 115
 116 /* debugging helpers */
 117 #ifdef __GSOCKET_DEBUG__
 118     #define GSocket_Debug(args)       printf args
 119 #else
 120     #define GSocket_Debug(args)
 121 #endif // __GSOCKET_DEBUG__
 122
 123 /* Global initialisers */
 124
 125 bool GSocket_Init(void)
 126 {
 127   return TRUE;
 128 }
 129
 130 void GSocket_Cleanup(void)
 131 {
 132 }
 133
 134 /* Constructors / Destructors for GSocket */
 135
 136 GSocket *GSocket_new(void)
 137 {
 138   int i, success;
 139   GSocket *socket;
 140
 141   socket = (GSocket *)malloc(sizeof(GSocket));
 142
 143   if (socket == NULL)
 144     return NULL;
 145
 146   socket->m_fd                  = INVALID_SOCKET;
 147   for (i=0;i<GSOCK_MAX_EVENT;i++)
 148   {
 149     socket->m_cbacks[i]         = NULL;
 150   }
 151   socket->m_detected            = 0;
 152   socket->m_local               = NULL;
 153   socket->m_peer                = NULL;
 154   socket->m_error               = GSOCK_NOERROR;
 155   socket->m_server              = FALSE;
 156   socket->m_stream              = TRUE;
 157   socket->m_gui_dependent       = NULL;
 158   socket->m_non_blocking        = FALSE;
 159   socket->m_timeout             = 10*60*1000;
 160                                 /* 10 minutes * 60 sec * 1000 millisec */
 161   socket->m_establishing        = FALSE;
 162
 163   /* Per-socket GUI-specific initialization */
 164   success = _GSocket_GUI_Init(socket);
 165   if (!success)
 166   {
 167     free(socket);
 168     return NULL;
 169   }
 170
 171   return socket;
 172 }
 173
 174 void GSocket_destroy(GSocket *socket)
 175 {
 176   assert(socket != NULL);
 177
 178   /* Check that the socket is really shutdowned */
 179   if (socket->m_fd != INVALID_SOCKET)
 180     GSocket_Shutdown(socket);
 181
 182   /* Per-socket GUI-specific cleanup */
 183   _GSocket_GUI_Destroy(socket);
 184
 185   /* Destroy private addresses */
 186   if (socket->m_local)
 187     GAddress_destroy(socket->m_local);
 188
 189   if (socket->m_peer)
 190     GAddress_destroy(socket->m_peer);
 191
 192   /* Destroy the socket itself */
 193   free(socket);
 194 }
 195
 196 /* GSocket_Shutdown:
 197  *  Disallow further read/write operations on this socket, close
 198  *  the fd and disable all callbacks.
 199  */
 200 void GSocket_Shutdown(GSocket *socket)
 201 {
 202   int evt;
 203
 204   assert(socket != NULL);
 205
 206   /* If socket has been created, shutdown it */
 207   if (socket->m_fd != INVALID_SOCKET)
 208   {
 209     shutdown(socket->m_fd, 2);
 210     close(socket->m_fd);
 211     socket->m_fd = INVALID_SOCKET;
 212   }
 213
 214   /* Disable GUI callbacks */
 215   for (evt = 0; evt < GSOCK_MAX_EVENT; evt++)
 216     socket->m_cbacks[evt] = NULL;
 217
 218   socket->m_detected = GSOCK_LOST_FLAG;
 219   _GSocket_Disable_Events(socket);
 220 }
 221
 222 /* Address handling */
 223
 224 /* GSocket_SetLocal:
 225  * GSocket_GetLocal:
 226  * GSocket_SetPeer:
 227  * GSocket_GetPeer:
 228  *  Set or get the local or peer address for this socket. The 'set'
 229  *  functions return GSOCK_NOERROR on success, an error code otherwise.
 230  *  The 'get' functions return a pointer to a GAddress object on success,
 231  *  or NULL otherwise, in which case they set the error code of the
 232  *  corresponding GSocket.
 233  *
 234  *  Error codes:
 235  *    GSOCK_INVSOCK - the socket is not valid.
 236  *    GSOCK_INVADDR - the address is not valid.
 237  */
 238 GSocketError GSocket_SetLocal(GSocket *socket, GAddress *address)
 239 {
 240   assert(socket != NULL);
 241
 242   /* the socket must be initialized, or it must be a server */
 243   if ((socket->m_fd != INVALID_SOCKET && !socket->m_server))
 244   {
 245     socket->m_error = GSOCK_INVSOCK;
 246     return GSOCK_INVSOCK;
 247   }
 248
 249   /* check address */
 250   if (address == NULL || address->m_family == GSOCK_NOFAMILY)
 251   {
 252     socket->m_error = GSOCK_INVADDR;
 253     return GSOCK_INVADDR;
 254   }
 255
 256   if (socket->m_local)
 257     GAddress_destroy(socket->m_local);
 258
 259   socket->m_local = GAddress_copy(address);
 260
 261   return GSOCK_NOERROR;
 262 }
 263
 264 GSocketError GSocket_SetPeer(GSocket *socket, GAddress *address)
 265 {
 266   assert(socket != NULL);
 267
 268   /* check address */
 269   if (address == NULL || address->m_family == GSOCK_NOFAMILY)
 270   {
 271     socket->m_error = GSOCK_INVADDR;
 272     return GSOCK_INVADDR;
 273   }
 274
 275   if (socket->m_peer)
 276     GAddress_destroy(socket->m_peer);
 277
 278   socket->m_peer = GAddress_copy(address);
 279
 280   return GSOCK_NOERROR;
 281 }
 282
 283 GAddress *GSocket_GetLocal(GSocket *socket)
 284 {
 285   GAddress *address;
 286   struct sockaddr addr;
 287   SOCKLEN_T size = sizeof(addr);
 288   GSocketError err;
 289
 290   assert(socket != NULL);
 291
 292   /* try to get it from the m_local var first */
 293   if (socket->m_local)
 294     return GAddress_copy(socket->m_local);
 295
 296   /* else, if the socket is initialized, try getsockname */
 297   if (socket->m_fd == INVALID_SOCKET)
 298   {
 299     socket->m_error = GSOCK_INVSOCK;
 300     return NULL;
 301   }
 302
 303   if (getsockname(socket->m_fd, &addr, (SOCKLEN_T *) &size) < 0)
 304   {
 305     socket->m_error = GSOCK_IOERR;
 306     return NULL;
 307   }
 308
 309   /* got a valid address from getsockname, create a GAddress object */
 310   address = GAddress_new();
 311   if (address == NULL)
 312   {
 313     socket->m_error = GSOCK_MEMERR;
 314     return NULL;
 315   }
 316
 317   err = _GAddress_translate_from(address, &addr, size);
 318   if (err != GSOCK_NOERROR)
 319   {
 320     GAddress_destroy(address);
 321     socket->m_error = err;
 322     return NULL;
 323   }
 324
 325   return address;
 326 }
 327
 328 GAddress *GSocket_GetPeer(GSocket *socket)
 329 {
 330   assert(socket != NULL);
 331
 332   /* try to get it from the m_peer var */
 333   if (socket->m_peer)
 334     return GAddress_copy(socket->m_peer);
 335
 336   return NULL;
 337 }
 338
 339 /* Server specific parts */
 340
 341 /* GSocket_SetServer:
 342  *  Sets up this socket as a server. The local address must have been
 343  *  set with GSocket_SetLocal() before GSocket_SetServer() is called.
 344  *  Returns GSOCK_NOERROR on success, one of the following otherwise:
 345  *
 346  *  Error codes:
 347  *    GSOCK_INVSOCK - the socket is in use.
 348  *    GSOCK_INVADDR - the local address has not been set.
 349  *    GSOCK_IOERR   - low-level error.
 350  */
 351 GSocketError GSocket_SetServer(GSocket *sck)
 352 {
 353   int type;
 354   int arg = 1;
 355
 356   assert(sck != NULL);
 357
 358   /* must not be in use */
 359   if (sck->m_fd != INVALID_SOCKET)
 360   {
 361     sck->m_error = GSOCK_INVSOCK;
 362     return GSOCK_INVSOCK;
 363   }
 364
 365   /* the local addr must have been set */
 366   if (!sck->m_local)
 367   {
 368     sck->m_error = GSOCK_INVADDR;
 369     return GSOCK_INVADDR;
 370   }
 371
 372   /* Initialize all fields */
 373   sck->m_stream   = TRUE;
 374   sck->m_server   = TRUE;
 375   sck->m_oriented = TRUE;
 376
 377   /* Create the socket */
 378   sck->m_fd = socket(sck->m_local->m_realfamily, SOCK_STREAM, 0);
 379
 380   if (sck->m_fd == INVALID_SOCKET)
 381   {
 382     sck->m_error = GSOCK_IOERR;
 383     return GSOCK_IOERR;
 384   }
 385
 386   ioctl(sck->m_fd, FIONBIO, &arg);
 387   _GSocket_Enable_Events(sck);
 388
 389   /* Bind to the local address,
 390    * retrieve the actual address bound,
 391    * and listen up to 5 connections.
 392    */
 393   if ((bind(sck->m_fd, sck->m_local->m_addr, sck->m_local->m_len) != 0) ||
 394       (getsockname(sck->m_fd,
 395                    sck->m_local->m_addr,
 396                    (SOCKLEN_T *) &sck->m_local->m_len) != 0) ||
 397       (listen(sck->m_fd, 5) != 0))
 398   {
 399     close(sck->m_fd);
 400     sck->m_fd = INVALID_SOCKET;
 401     sck->m_error = GSOCK_IOERR;
 402     return GSOCK_IOERR;
 403   }
 404
 405   return GSOCK_NOERROR;
 406 }
 407
 408 /* GSocket_WaitConnection:
 409  *  Waits for an incoming client connection. Returns a pointer to
 410  *  a GSocket object, or NULL if there was an error, in which case
 411  *  the last error field will be updated for the calling GSocket.
 412  *
 413  *  Error codes (set in the calling GSocket)
 414  *    GSOCK_INVSOCK    - the socket is not valid or not a server.
 415  *    GSOCK_TIMEDOUT   - timeout, no incoming connections.
 416  *    GSOCK_WOULDBLOCK - the call would block and the socket is nonblocking.
 417  *    GSOCK_MEMERR     - couldn't allocate memory.
 418  *    GSOCK_IOERR      - low-level error.
 419  */
 420 GSocket *GSocket_WaitConnection(GSocket *socket)
 421 {
 422   struct sockaddr from;
 423   SOCKLEN_T fromlen = sizeof(from);
 424   GSocket *connection;
 425   GSocketError err;
 426   int arg = 1;
 427
 428   assert(socket != NULL);
 429
 430   /* Reenable CONNECTION events */
 431   _GSocket_Enable(socket, GSOCK_CONNECTION);
 432
 433   /* If the socket has already been created, we exit immediately */
 434   if (socket->m_fd == INVALID_SOCKET || !socket->m_server)
 435   {
 436     socket->m_error = GSOCK_INVSOCK;
 437     return NULL;
 438   }
 439
 440   /* Create a GSocket object for the new connection */
 441   connection = GSocket_new();
 442
 443   if (!connection)
 444   {
 445     socket->m_error = GSOCK_MEMERR;
 446     return NULL;
 447   }
 448
 449   /* Wait for a connection (with timeout) */
 450   if (_GSocket_Input_Timeout(socket) == GSOCK_TIMEDOUT)
 451   {
 452     GSocket_destroy(connection);
 453     /* socket->m_error set by _GSocket_Input_Timeout */
 454     return NULL;
 455   }
 456
 457   connection->m_fd = accept(socket->m_fd, &from, (SOCKLEN_T *) &fromlen);
 458
 459   if (connection->m_fd == INVALID_SOCKET)
 460   {
 461     if (errno == EWOULDBLOCK)
 462       socket->m_error = GSOCK_WOULDBLOCK;
 463     else
 464       socket->m_error = GSOCK_IOERR;
 465
 466     GSocket_destroy(connection);
 467     return NULL;
 468   }
 469
 470   /* Initialize all fields */
 471   connection->m_server   = FALSE;
 472   connection->m_stream   = TRUE;
 473   connection->m_oriented = TRUE;
 474
 475   /* Setup the peer address field */
 476   connection->m_peer = GAddress_new();
 477   if (!connection->m_peer)
 478   {
 479     GSocket_destroy(connection);
 480     socket->m_error = GSOCK_MEMERR;
 481     return NULL;
 482   }
 483   err = _GAddress_translate_from(connection->m_peer, &from, fromlen);
 484   if (err != GSOCK_NOERROR)
 485   {
 486     GAddress_destroy(connection->m_peer);
 487     GSocket_destroy(connection);
 488     socket->m_error = err;
 489     return NULL;
 490   }
 491
 492   ioctl(connection->m_fd, FIONBIO, &arg);
 493   _GSocket_Enable_Events(connection);
 494
 495   return connection;
 496 }
 497
 498 /* Client specific parts */
 499
 500 /* GSocket_Connect:
 501  *  For stream (connection oriented) sockets, GSocket_Connect() tries
 502  *  to establish a client connection to a server using the peer address
 503  *  as established with GSocket_SetPeer(). Returns GSOCK_NOERROR if the
 504  *  connection has been succesfully established, or one of the error
 505  *  codes listed below. Note that for nonblocking sockets, a return
 506  *  value of GSOCK_WOULDBLOCK doesn't mean a failure. The connection
 507  *  request can be completed later; you should use GSocket_Select()
 508  *  to poll for GSOCK_CONNECTION | GSOCK_LOST, or wait for the
 509  *  corresponding asynchronous events.
 510  *
 511  *  For datagram (non connection oriented) sockets, GSocket_Connect()
 512  *  just sets the peer address established with GSocket_SetPeer() as
 513  *  default destination.
 514  *
 515  *  Error codes:
 516  *    GSOCK_INVSOCK    - the socket is in use or not valid.
 517  *    GSOCK_INVADDR    - the peer address has not been established.
 518  *    GSOCK_TIMEDOUT   - timeout, the connection failed.
 519  *    GSOCK_WOULDBLOCK - connection in progress (nonblocking sockets only)
 520  *    GSOCK_MEMERR     - couldn't allocate memory.
 521  *    GSOCK_IOERR      - low-level error.
 522  */
 523 GSocketError GSocket_Connect(GSocket *sck, GSocketStream stream)
 524 {
 525   int err, ret;
 526   int arg = 1;
 527
 528   assert(sck != NULL);
 529
 530   /* Enable CONNECTION events (needed for nonblocking connections) */
 531   _GSocket_Enable(sck, GSOCK_CONNECTION);
 532
 533   if (sck->m_fd != INVALID_SOCKET)
 534   {
 535     sck->m_error = GSOCK_INVSOCK;
 536     return GSOCK_INVSOCK;
 537   }
 538
 539   if (!sck->m_peer)
 540   {
 541     sck->m_error = GSOCK_INVADDR;
 542     return GSOCK_INVADDR;
 543   }
 544
 545   /* Streamed or dgram socket? */
 546   sck->m_stream   = (stream == GSOCK_STREAMED);
 547   sck->m_oriented = TRUE;
 548   sck->m_server   = FALSE;
 549   sck->m_establishing = FALSE;
 550
 551   /* Create the socket */
 552   sck->m_fd = socket(sck->m_peer->m_realfamily,
 553                      sck->m_stream? SOCK_STREAM : SOCK_DGRAM, 0);
 554
 555   if (sck->m_fd == INVALID_SOCKET)
 556   {
 557     sck->m_error = GSOCK_IOERR;
 558     return GSOCK_IOERR;
 559   }
 560
 561   ioctl(sck->m_fd, FIONBIO, &arg);
 562   _GSocket_Enable_Events(sck);
 563
 564   /* Connect it to the peer address, with a timeout (see below) */
 565   ret = connect(sck->m_fd, sck->m_peer->m_addr, sck->m_peer->m_len);
 566
 567   if (ret == -1)
 568   {
 569     err = errno;
 570
 571     /* If connect failed with EINPROGRESS and the GSocket object
 572      * is in blocking mode, we select() for the specified timeout
 573      * checking for writability to see if the connection request
 574      * completes.
 575      */
 576     if ((err == EINPROGRESS) && (!sck->m_non_blocking))
 577     {
 578       if (_GSocket_Output_Timeout(sck) == GSOCK_TIMEDOUT)
 579       {
 580         close(sck->m_fd);
 581         sck->m_fd = INVALID_SOCKET;
 582         /* sck->m_error is set in _GSocket_Output_Timeout */
 583         return GSOCK_TIMEDOUT;
 584       }
 585       else
 586       {
 587         int error;
 588         SOCKLEN_T len = sizeof(error);
 589
 590         getsockopt(sck->m_fd, SOL_SOCKET, SO_ERROR, (void*) &error, &len);
 591
 592         if (!error)
 593           return GSOCK_NOERROR;
 594       }
 595     }
 596
 597     /* If connect failed with EINPROGRESS and the GSocket object
 598      * is set to nonblocking, we set m_error to GSOCK_WOULDBLOCK
 599      * (and return GSOCK_WOULDBLOCK) but we don't close the socket;
 600      * this way if the connection completes, a GSOCK_CONNECTION
 601      * event will be generated, if enabled.
 602      */
 603     if ((err == EINPROGRESS) && (sck->m_non_blocking))
 604     {
 605       sck->m_establishing = TRUE;
 606       sck->m_error = GSOCK_WOULDBLOCK;
 607       return GSOCK_WOULDBLOCK;
 608     }
 609
 610     /* If connect failed with an error other than EINPROGRESS,
 611      * then the call to GSocket_Connect has failed.
 612      */
 613     close(sck->m_fd);
 614     sck->m_fd = INVALID_SOCKET;
 615     sck->m_error = GSOCK_IOERR;
 616     return GSOCK_IOERR;
 617   }
 618
 619   return GSOCK_NOERROR;
 620 }
 621
 622 /* Datagram sockets */
 623
 624 /* GSocket_SetNonOriented:
 625  *  Sets up this socket as a non-connection oriented (datagram) socket.
 626  *  Before using this function, the local address must have been set
 627  *  with GSocket_SetLocal(), or the call will fail. Returns GSOCK_NOERROR
 628  *  on success, or one of the following otherwise.
 629  *
 630  *  Error codes:
 631  *    GSOCK_INVSOCK - the socket is in use.
 632  *    GSOCK_INVADDR - the local address has not been set.
 633  *    GSOCK_IOERR   - low-level error.
 634  */
 635 GSocketError GSocket_SetNonOriented(GSocket *sck)
 636 {
 637   int arg = 1;
 638
 639   assert(sck != NULL);
 640
 641   if (sck->m_fd != INVALID_SOCKET)
 642   {
 643     sck->m_error = GSOCK_INVSOCK;
 644     return GSOCK_INVSOCK;
 645   }
 646
 647   if (!sck->m_local)
 648   {
 649     sck->m_error = GSOCK_INVADDR;
 650     return GSOCK_INVADDR;
 651   }
 652
 653   /* Initialize all fields */
 654   sck->m_stream   = FALSE;
 655   sck->m_server   = FALSE;
 656   sck->m_oriented = FALSE;
 657
 658   /* Create the socket */
 659   sck->m_fd = socket(sck->m_local->m_realfamily, SOCK_DGRAM, 0);
 660
 661   if (sck->m_fd == INVALID_SOCKET)
 662   {
 663     sck->m_error = GSOCK_IOERR;
 664     return GSOCK_IOERR;
 665   }
 666
 667   ioctl(sck->m_fd, FIONBIO, &arg);
 668   _GSocket_Enable_Events(sck);
 669
 670   /* Bind to the local address,
 671    * and retrieve the actual address bound.
 672    */
 673   if ((bind(sck->m_fd, sck->m_local->m_addr, sck->m_local->m_len) != 0) ||
 674       (getsockname(sck->m_fd,
 675                    sck->m_local->m_addr,
 676                    (SOCKLEN_T *) &sck->m_local->m_len) != 0))
 677   {
 678     close(sck->m_fd);
 679     sck->m_fd    = INVALID_SOCKET;
 680     sck->m_error = GSOCK_IOERR;
 681     return GSOCK_IOERR;
 682   }
 683
 684   return GSOCK_NOERROR;
 685 }
 686
 687 /* Generic IO */
 688
 689 /* Like recv(), send(), ... */
 690 int GSocket_Read(GSocket *socket, char *buffer, int size)
 691 {
 692   int ret;
 693
 694   assert(socket != NULL);
 695
 696   /* Reenable INPUT events */
 697   _GSocket_Enable(socket, GSOCK_INPUT);
 698
 699   if (socket->m_fd == INVALID_SOCKET || socket->m_server)
 700   {
 701     socket->m_error = GSOCK_INVSOCK;
 702     return -1;
 703   }
 704
 705   /* If the socket is blocking, wait for data (with a timeout) */
 706   if (_GSocket_Input_Timeout(socket) == GSOCK_TIMEDOUT)
 707     return -1;
 708
 709   /* Read the data */
 710   if (socket->m_stream)
 711     ret = _GSocket_Recv_Stream(socket, buffer, size);
 712   else
 713     ret = _GSocket_Recv_Dgram(socket, buffer, size);
 714
 715   if (ret == -1)
 716   {
 717     if (errno == EWOULDBLOCK)
 718       socket->m_error = GSOCK_WOULDBLOCK;
 719     else
 720       socket->m_error = GSOCK_IOERR;
 721   }
 722
 723   return ret;
 724 }
 725
 726 int GSocket_Write(GSocket *socket, const char *buffer, int size)
 727 {
 728   int ret;
 729
 730   assert(socket != NULL);
 731
 732   GSocket_Debug(( "GSocket_Write #1, size %d\n", size ));
 733
 734   if (socket->m_fd == INVALID_SOCKET || socket->m_server)
 735   {
 736     socket->m_error = GSOCK_INVSOCK;
 737     return -1;
 738   }
 739
 740   GSocket_Debug(( "GSocket_Write #2, size %d\n", size ));
 741
 742   /* If the socket is blocking, wait for writability (with a timeout) */
 743   if (_GSocket_Output_Timeout(socket) == GSOCK_TIMEDOUT)
 744     return -1;
 745
 746   GSocket_Debug(( "GSocket_Write #3, size %d\n", size ));
 747
 748   /* Write the data */
 749   if (socket->m_stream)
 750     ret = _GSocket_Send_Stream(socket, buffer, size);
 751   else
 752     ret = _GSocket_Send_Dgram(socket, buffer, size);
 753
 754   GSocket_Debug(( "GSocket_Write #4, size %d\n", size ));
 755
 756   if (ret == -1)
 757   {
 758     if (errno == EWOULDBLOCK)
 759     {
 760       socket->m_error = GSOCK_WOULDBLOCK;
 761       GSocket_Debug(( "GSocket_Write error WOULDBLOCK\n" ));
 762     }
 763     else
 764     {
 765       socket->m_error = GSOCK_IOERR;
 766       GSocket_Debug(( "GSocket_Write error IOERR\n" ));
 767     }
 768
 769     /* Only reenable OUTPUT events after an error (just like WSAAsyncSelect
 770      * in MSW). Once the first OUTPUT event is received, users can assume
 771      * that the socket is writable until a read operation fails. Only then
 772      * will further OUTPUT events be posted.
 773      */
 774     _GSocket_Enable(socket, GSOCK_OUTPUT);
 775     return -1;
 776   }
 777
 778   GSocket_Debug(( "GSocket_Write #5, size %d ret %d\n", size, ret ));
 779
 780   return ret;
 781 }
 782
 783 /* GSocket_Select:
 784  *  Polls the socket to determine its status. This function will
 785  *  check for the events specified in the 'flags' parameter, and
 786  *  it will return a mask indicating which operations can be
 787  *  performed. This function won't block, regardless of the
 788  *  mode (blocking | nonblocking) of the socket.
 789  */
 790 GSocketEventFlags GSocket_Select(GSocket *socket, GSocketEventFlags flags)
 791 {
 792 #if defined(wxUSE_GUI) && !wxUSE_GUI
 793
 794   GSocketEventFlags result = 0;
 795   fd_set readfds;
 796   fd_set writefds;
 797   fd_set exceptfds;
 798   struct timeval tv;
 799
 800   /* Do not use a static struct, Linux can garble it */
 801   tv.tv_sec = 0;
 802   tv.tv_usec = 0;
 803
 804   assert(socket != NULL);
 805
 806   FD_ZERO(&readfds);
 807   FD_ZERO(&writefds);
 808   FD_ZERO(&exceptfds);
 809   FD_SET(socket->m_fd, &readfds);
 810   FD_SET(socket->m_fd, &writefds);
 811   FD_SET(socket->m_fd, &exceptfds);
 812
 813   /* Check known state first */
 814   result |= (GSOCK_CONNECTION_FLAG & socket->m_detected & flags);
 815   result |= (GSOCK_LOST_FLAG       & socket->m_detected & flags);
 816
 817   /* Try select now */
 818   if (select(socket->m_fd + 1, &readfds, &writefds, &exceptfds, &tv) <= 0)
 819     return result;
 820
 821   /* Check for readability */
 822   if (FD_ISSET(socket->m_fd, &readfds))
 823   {
 824     char c;
 825
 826     if (recv(socket->m_fd, &c, 1, MSG_PEEK) > 0)
 827     {
 828       result |= (GSOCK_INPUT_FLAG & flags);
 829     }
 830     else
 831     {
 832       if (socket->m_server && socket->m_stream)
 833       {
 834         result |= (GSOCK_CONNECTION_FLAG & flags);
 835         socket->m_detected |= GSOCK_CONNECTION_FLAG;
 836       }
 837       else
 838       {
 839         result |= (GSOCK_LOST_FLAG & flags);
 840         socket->m_detected = GSOCK_LOST_FLAG;
 841       }
 842     }
 843   }
 844
 845   /* Check for writability */
 846   if (FD_ISSET(socket->m_fd, &writefds))
 847   {
 848     if (socket->m_establishing && !socket->m_server)
 849     {
 850       int error;
 851       SOCKLEN_T len = sizeof(error);
 852
 853       socket->m_establishing = FALSE;
 854
 855       getsockopt(socket->m_fd, SOL_SOCKET, SO_ERROR, (void*)&error, &len);
 856
 857       if (error)
 858       {
 859         result |= (GSOCK_LOST_FLAG & flags);
 860         socket->m_detected = GSOCK_LOST_FLAG;
 861       }
 862       else
 863       {
 864         result |= (GSOCK_CONNECTION_FLAG & flags);
 865         socket->m_detected |= GSOCK_CONNECTION_FLAG;
 866       }
 867     }
 868     else
 869     {
 870       result |= (GSOCK_OUTPUT_FLAG & flags);
 871     }
 872   }
 873
 874   /* Check for exceptions and errors (is this useful in Unices?) */
 875   if (FD_ISSET(socket->m_fd, &exceptfds))
 876   {
 877     result |= (GSOCK_LOST_FLAG & flags);
 878     socket->m_establishing = FALSE;
 879     socket->m_detected = GSOCK_LOST_FLAG;
 880   }
 881
 882   return result;
 883
 884 #else
 885
 886   assert(socket != NULL);
 887   return flags & socket->m_detected;
 888
 889 #endif /* !wxUSE_GUI */
 890 }
 891
 892 /* Flags */
 893
 894 /* GSocket_SetNonBlocking:
 895  *  Sets the socket to non-blocking mode. All IO calls will return
 896  *  immediately.
 897  */
 898 void GSocket_SetNonBlocking(GSocket *socket, bool non_block)
 899 {
 900   assert(socket != NULL);
 901
 902   GSocket_Debug( ("GSocket_SetNonBlocking: %d\n", (int)non_block) );
 903
 904   socket->m_non_blocking = non_block;
 905 }
 906
 907 /* GSocket_SetTimeout:
 908  *  Sets the timeout for blocking calls. Time is expressed in
 909  *  milliseconds.
 910  */
 911 void GSocket_SetTimeout(GSocket *socket, unsigned long millisec)
 912 {
 913   assert(socket != NULL);
 914
 915   socket->m_timeout = millisec;
 916 }
 917
 918 /* GSocket_GetError:
 919  *  Returns the last error occured for this socket. Note that successful
 920  *  operations do not clear this back to GSOCK_NOERROR, so use it only
 921  *  after an error.
 922  */
 923 GSocketError GSocket_GetError(GSocket *socket)
 924 {
 925   assert(socket != NULL);
 926
 927   return socket->m_error;
 928 }
 929
 930 /* Callbacks */
 931
 932 /* GSOCK_INPUT:
 933  *   There is data to be read in the input buffer. If, after a read
 934  *   operation, there is still data available, the callback function will
 935  *   be called again.
 936  * GSOCK_OUTPUT:
 937  *   The socket is available for writing. That is, the next write call
 938  *   won't block. This event is generated only once, when the connection is
 939  *   first established, and then only if a call failed with GSOCK_WOULDBLOCK,
 940  *   when the output buffer empties again. This means that the app should
 941  *   assume that it can write since the first OUTPUT event, and no more
 942  *   OUTPUT events will be generated unless an error occurs.
 943  * GSOCK_CONNECTION:
 944  *   Connection succesfully established, for client sockets, or incoming
 945  *   client connection, for server sockets. Wait for this event (also watch
 946  *   out for GSOCK_LOST) after you issue a nonblocking GSocket_Connect() call.
 947  * GSOCK_LOST:
 948  *   The connection is lost (or a connection request failed); this could
 949  *   be due to a failure, or due to the peer closing it gracefully.
 950  */
 951
 952 /* GSocket_SetCallback:
 953  *  Enables the callbacks specified by 'flags'. Note that 'flags'
 954  *  may be a combination of flags OR'ed toghether, so the same
 955  *  callback function can be made to accept different events.
 956  *  The callback function must have the following prototype:
 957  *
 958  *  void function(GSocket *socket, GSocketEvent event, char *cdata)
 959  */
 960 void GSocket_SetCallback(GSocket *socket, GSocketEventFlags flags,
 961                          GSocketCallback callback, char *cdata)
 962 {
 963   int count;
 964
 965   assert(socket != NULL);
 966
 967   for (count = 0; count < GSOCK_MAX_EVENT; count++)
 968   {
 969     if ((flags & (1 << count)) != 0)
 970     {
 971       socket->m_cbacks[count] = callback;
 972       socket->m_data[count] = cdata;
 973     }
 974   }
 975 }
 976
 977 /* GSocket_UnsetCallback:
 978  *  Disables all callbacks specified by 'flags', which may be a
 979  *  combination of flags OR'ed toghether.
 980  */
 981 void GSocket_UnsetCallback(GSocket *socket, GSocketEventFlags flags)
 982 {
 983   int count;
 984
 985   assert(socket != NULL);
 986
 987   for (count = 0; count < GSOCK_MAX_EVENT; count++)
 988   {
 989     if ((flags & (1 << count)) != 0)
 990     {
 991       socket->m_cbacks[count] = NULL;
 992       socket->m_data[count] = NULL;
 993     }
 994   }
 995 }
 996
 997
 998 #define CALL_CALLBACK(socket, event) {                                  \
 999   _GSocket_Disable(socket, event);                                      \
1000   if (socket->m_cbacks[event])                                          \
1001     socket->m_cbacks[event](socket, event, socket->m_data[event]);      \
1002 }
1003
1004
1005 void _GSocket_Enable(GSocket *socket, GSocketEvent event)
1006 {
1007   socket->m_detected &= ~(1 << event);
1008   _GSocket_Install_Callback(socket, event);
1009 }
1010
1011 void _GSocket_Disable(GSocket *socket, GSocketEvent event)
1012 {
1013   socket->m_detected |= (1 << event);
1014   _GSocket_Uninstall_Callback(socket, event);
1015 }
1016
1017 /* _GSocket_Input_Timeout:
1018  *  For blocking sockets, wait until data is available or
1019  *  until timeout ellapses.
1020  */
1021 GSocketError _GSocket_Input_Timeout(GSocket *socket)
1022 {
1023   struct timeval tv;
1024   fd_set readfds;
1025   int ret;
1026
1027   /* Linux select() will overwrite the struct on return */
1028   tv.tv_sec  = (socket->m_timeout / 1000);
1029   tv.tv_usec = (socket->m_timeout % 1000) * 1000;
1030
1031   if (!socket->m_non_blocking)
1032   {
1033     FD_ZERO(&readfds);
1034     FD_SET(socket->m_fd, &readfds);
1035     ret = select(socket->m_fd + 1, &readfds, NULL, NULL, &tv);
1036     if (ret == 0)
1037     {
1038       GSocket_Debug(( "GSocket_Input_Timeout, select returned 0\n" ));
1039       socket->m_error = GSOCK_TIMEDOUT;
1040       return GSOCK_TIMEDOUT;
1041     }
1042     if (ret == -1)
1043     {
1044       GSocket_Debug(( "GSocket_Input_Timeout, select returned -1\n" ));
1045       if (errno == EBADF) GSocket_Debug(( "Invalid file descriptor\n" ));
1046       if (errno == EINTR) GSocket_Debug(( "A non blocked signal was caught\n" ));
1047       if (errno == EINVAL) GSocket_Debug(( "The highest number descriptor is negative\n" ));
1048       if (errno == ENOMEM) GSocket_Debug(( "Not enough memory\n" ));
1049       socket->m_error = GSOCK_TIMEDOUT;
1050       return GSOCK_TIMEDOUT;
1051     }
1052   }
1053   return GSOCK_NOERROR;
1054 }
1055
1056 /* _GSocket_Output_Timeout:
1057  *  For blocking sockets, wait until data can be sent without
1058  *  blocking or until timeout ellapses.
1059  */
1060 GSocketError _GSocket_Output_Timeout(GSocket *socket)
1061 {
1062   struct timeval tv;
1063   fd_set writefds;
1064   int ret;
1065
1066   /* Linux select() will overwrite the struct on return */
1067   tv.tv_sec  = (socket->m_timeout / 1000);
1068   tv.tv_usec = (socket->m_timeout % 1000) * 1000;
1069
1070   GSocket_Debug( ("m_non_blocking has: %d\n", (int)socket->m_non_blocking) );
1071
1072   if (!socket->m_non_blocking)
1073   {
1074     FD_ZERO(&writefds);
1075     FD_SET(socket->m_fd, &writefds);
1076     ret = select(socket->m_fd + 1, NULL, &writefds, NULL, &tv);
1077     if (ret == 0)
1078     {
1079       GSocket_Debug(( "GSocket_Output_Timeout, select returned 0\n" ));
1080       socket->m_error = GSOCK_TIMEDOUT;
1081       return GSOCK_TIMEDOUT;
1082     }
1083     if (ret == -1)
1084     {
1085       GSocket_Debug(( "GSocket_Output_Timeout, select returned -1\n" ));
1086       if (errno == EBADF) GSocket_Debug(( "Invalid file descriptor\n" ));
1087       if (errno == EINTR) GSocket_Debug(( "A non blocked signal was caught\n" ));
1088       if (errno == EINVAL) GSocket_Debug(( "The highest number descriptor is negative\n" ));
1089       if (errno == ENOMEM) GSocket_Debug(( "Not enough memory\n" ));
1090       socket->m_error = GSOCK_TIMEDOUT;
1091       return GSOCK_TIMEDOUT;
1092     }
1093     if ( ! FD_ISSET(socket->m_fd, &writefds) )
1094       GSocket_Debug(( "GSocket_Output_Timeout is buggy!\n" ));
1095     else
1096       GSocket_Debug(( "GSocket_Output_Timeout seems correct\n" ));
1097   }
1098   else
1099   {
1100     GSocket_Debug(( "GSocket_Output_Timeout, didn't try select!\n" ));
1101   }
1102
1103   return GSOCK_NOERROR;
1104 }
1105
1106 int _GSocket_Recv_Stream(GSocket *socket, char *buffer, int size)
1107 {
1108   return recv(socket->m_fd, buffer, size, 0);
1109 }
1110
1111 int _GSocket_Recv_Dgram(GSocket *socket, char *buffer, int size)
1112 {
1113   struct sockaddr from;
1114   SOCKLEN_T fromlen = sizeof(from);
1115   int ret;
1116   GSocketError err;
1117
1118   fromlen = sizeof(from);
1119
1120   ret = recvfrom(socket->m_fd, buffer, size, 0, &from, (SOCKLEN_T *) &fromlen);
1121
1122   if (ret == -1)
1123     return -1;
1124
1125   /* Translate a system address into a GSocket address */
1126   if (!socket->m_peer)
1127   {
1128     socket->m_peer = GAddress_new();
1129     if (!socket->m_peer)
1130     {
1131       socket->m_error = GSOCK_MEMERR;
1132       return -1;
1133     }
1134   }
1135   err = _GAddress_translate_from(socket->m_peer, &from, fromlen);
1136   if (err != GSOCK_NOERROR)
1137   {
1138     GAddress_destroy(socket->m_peer);
1139     socket->m_peer  = NULL;
1140     socket->m_error = err;
1141     return -1;
1142   }
1143
1144   return ret;
1145 }
1146
1147 int _GSocket_Send_Stream(GSocket *socket, const char *buffer, int size)
1148 {
1149   int ret;
1150
1151   MASK_SIGNAL();
1152   ret = send(socket->m_fd, buffer, size, 0);
1153   UNMASK_SIGNAL();
1154
1155   return ret;
1156 }
1157
1158 int _GSocket_Send_Dgram(GSocket *socket, const char *buffer, int size)
1159 {
1160   struct sockaddr *addr;
1161   int len, ret;
1162   GSocketError err;
1163
1164   if (!socket->m_peer)
1165   {
1166     socket->m_error = GSOCK_INVADDR;
1167     return -1;
1168   }
1169
1170   err = _GAddress_translate_to(socket->m_peer, &addr, &len);
1171   if (err != GSOCK_NOERROR)
1172   {
1173     socket->m_error = err;
1174     return -1;
1175   }
1176
1177   MASK_SIGNAL();
1178   ret = sendto(socket->m_fd, buffer, size, 0, addr, len);
1179   UNMASK_SIGNAL();
1180
1181   /* Frees memory allocated from _GAddress_translate_to */
1182   free(addr);
1183
1184   return ret;
1185 }
1186
1187 void _GSocket_Detected_Read(GSocket *socket)
1188 {
1189   char c;
1190
1191   if (recv(socket->m_fd, &c, 1, MSG_PEEK) > 0)
1192   {
1193     CALL_CALLBACK(socket, GSOCK_INPUT);
1194   }
1195   else
1196   {
1197     if (socket->m_server && socket->m_stream)
1198     {
1199       CALL_CALLBACK(socket, GSOCK_CONNECTION);
1200     }
1201     else
1202     {
1203       CALL_CALLBACK(socket, GSOCK_LOST);
1204     }
1205   }
1206 }
1207
1208 void _GSocket_Detected_Write(GSocket *socket)
1209 {
1210   if (socket->m_establishing && !socket->m_server)
1211   {
1212     int error;
1213     SOCKLEN_T len = sizeof(error);
1214
1215     socket->m_establishing = FALSE;
1216
1217     getsockopt(socket->m_fd, SOL_SOCKET, SO_ERROR, (void*)&error, &len);
1218
1219     if (error)
1220     {
1221       CALL_CALLBACK(socket, GSOCK_LOST);
1222     }
1223     else
1224     {
1225       CALL_CALLBACK(socket, GSOCK_CONNECTION);
1226       /* We have to fire this event by hand because CONNECTION (for clients)
1227        * and OUTPUT are internally the same and we just disabled CONNECTION
1228        * events with the above macro.
1229        */
1230       CALL_CALLBACK(socket, GSOCK_OUTPUT);
1231     }
1232   }
1233   else
1234   {
1235     CALL_CALLBACK(socket, GSOCK_OUTPUT);
1236   }
1237 }
1238
1239 /*
1240  * -------------------------------------------------------------------------
1241  * GAddress
1242  * -------------------------------------------------------------------------
1243  */
1244
1245 /* CHECK_ADDRESS verifies that the current family is either GSOCK_NOFAMILY
1246  * or GSOCK_*family*, and if it is GSOCK_NOFAMILY, it initalizes address
1247  * to be a GSOCK_*family*. In other cases, it returns GSOCK_INVADDR.
1248  */
1249 #define CHECK_ADDRESS(address, family, retval)                      \
1250 {                                                                   \
1251   if (address->m_family == GSOCK_NOFAMILY)                          \
1252     if (_GAddress_Init_##family(address) != GSOCK_NOERROR)          \
1253       return address->m_error;                                      \
1254   if (address->m_family != GSOCK_##family)                          \
1255   {                                                                 \
1256     address->m_error = GSOCK_INVADDR;                               \
1257     return retval;                                                  \
1258   }                                                                 \
1259 }
1260
1261 GAddress *GAddress_new(void)
1262 {
1263   GAddress *address;
1264
1265   if ((address = (GAddress *) malloc(sizeof(GAddress))) == NULL)
1266     return NULL;
1267
1268   address->m_family  = GSOCK_NOFAMILY;
1269   address->m_addr    = NULL;
1270   address->m_len     = 0;
1271
1272   return address;
1273 }
1274
1275 GAddress *GAddress_copy(GAddress *address)
1276 {
1277   GAddress *addr2;
1278
1279   assert(address != NULL);
1280
1281   if ((addr2 = (GAddress *) malloc(sizeof(GAddress))) == NULL)
1282     return NULL;
1283
1284   memcpy(addr2, address, sizeof(GAddress));
1285
1286   if (address->m_addr)
1287   {
1288     addr2->m_addr = (struct sockaddr *)malloc(addr2->m_len);
1289     if (addr2->m_addr == NULL)
1290     {
1291       free(addr2);
1292       return NULL;
1293     }
1294     memcpy(addr2->m_addr, address->m_addr, addr2->m_len);
1295   }
1296
1297   return addr2;
1298 }
1299
1300 void GAddress_destroy(GAddress *address)
1301 {
1302   assert(address != NULL);
1303
1304   if (address->m_addr)
1305     free(address->m_addr);
1306
1307   free(address);
1308 }
1309
1310 void GAddress_SetFamily(GAddress *address, GAddressType type)
1311 {
1312   assert(address != NULL);
1313
1314   address->m_family = type;
1315 }
1316
1317 GAddressType GAddress_GetFamily(GAddress *address)
1318 {
1319   assert(address != NULL);
1320
1321   return address->m_family;
1322 }
1323
1324 GSocketError _GAddress_translate_from(GAddress *address,
1325                                       struct sockaddr *addr, int len)
1326 {
1327   address->m_realfamily = addr->sa_family;
1328   switch (addr->sa_family)
1329   {
1330     case AF_INET:
1331       address->m_family = GSOCK_INET;
1332       break;
1333     case AF_UNIX:
1334       address->m_family = GSOCK_UNIX;
1335       break;
1336 #ifdef AF_INET6
1337     case AF_INET6:
1338       address->m_family = GSOCK_INET6;
1339       break;
1340 #endif
1341     default:
1342     {
1343       address->m_error = GSOCK_INVOP;
1344       return GSOCK_INVOP;
1345     }
1346   }
1347
1348   if (address->m_addr)
1349     free(address->m_addr);
1350
1351   address->m_len  = len;
1352   address->m_addr = (struct sockaddr *)malloc(len);
1353
1354   if (address->m_addr == NULL)
1355   {
1356     address->m_error = GSOCK_MEMERR;
1357     return GSOCK_MEMERR;
1358   }
1359   memcpy(address->m_addr, addr, len);
1360
1361   return GSOCK_NOERROR;
1362 }
1363
1364 GSocketError _GAddress_translate_to(GAddress *address,
1365                                     struct sockaddr **addr, int *len)
1366 {
1367   if (!address->m_addr)
1368   {
1369     address->m_error = GSOCK_INVADDR;
1370     return GSOCK_INVADDR;
1371   }
1372
1373   *len = address->m_len;
1374   *addr = (struct sockaddr *)malloc(address->m_len);
1375   if (*addr == NULL) {
1376     address->m_error = GSOCK_MEMERR;
1377     return GSOCK_MEMERR;
1378   }
1379
1380   memcpy(*addr, address->m_addr, address->m_len);
1381   return GSOCK_NOERROR;
1382 }
1383
1384 /*
1385  * -------------------------------------------------------------------------
1386  * Internet address family
1387  * -------------------------------------------------------------------------
1388  */
1389
1390 GSocketError _GAddress_Init_INET(GAddress *address)
1391 {
1392   address->m_len  = sizeof(struct sockaddr_in);
1393   address->m_addr = (struct sockaddr *) malloc(address->m_len);
1394   if (address->m_addr == NULL)
1395   {
1396     address->m_error = GSOCK_MEMERR;
1397     return GSOCK_MEMERR;
1398   }
1399
1400   address->m_family = GSOCK_INET;
1401   address->m_realfamily = PF_INET;
1402   ((struct sockaddr_in *)address->m_addr)->sin_family = AF_INET;
1403   ((struct sockaddr_in *)address->m_addr)->sin_addr.s_addr = INADDR_ANY;
1404
1405   return GSOCK_NOERROR;
1406 }
1407
1408 GSocketError GAddress_INET_SetHostName(GAddress *address, const char *hostname)
1409 {
1410   struct hostent *he;
1411   struct in_addr *addr;
1412
1413   assert(address != NULL);
1414
1415   CHECK_ADDRESS(address, INET, GSOCK_INVADDR);
1416
1417   addr = &(((struct sockaddr_in *)address->m_addr)->sin_addr);
1418
1419   /* If it is a numeric host name, convert it now */
1420 #if defined(HAVE_INET_ATON)
1421   if (inet_aton(hostname, addr) == 0)
1422   {
1423 #elif defined(HAVE_INET_ADDR)
1424   if ( (addr->s_addr = inet_addr(hostname)) == -1 )
1425   {
1426 #else
1427   /* Use gethostbyname by default */
1428   if (1)
1429   {
1430 #endif
1431     struct in_addr *array_addr;
1432
1433     /* It is a real name, we solve it */
1434     if ((he = gethostbyname(hostname)) == NULL)
1435     {
1436       /* Reset to invalid address */
1437       addr->s_addr = INADDR_NONE;
1438       address->m_error = GSOCK_NOHOST;
1439       return GSOCK_NOHOST;
1440     }
1441     array_addr = (struct in_addr *) *(he->h_addr_list);
1442     addr->s_addr = array_addr[0].s_addr;
1443   }
1444   return GSOCK_NOERROR;
1445 }
1446
1447 GSocketError GAddress_INET_SetAnyAddress(GAddress *address)
1448 {
1449   return GAddress_INET_SetHostAddress(address, INADDR_ANY);
1450 }
1451
1452 GSocketError GAddress_INET_SetHostAddress(GAddress *address,
1453                                           unsigned long hostaddr)
1454 {
1455   struct in_addr *addr;
1456
1457   assert(address != NULL);
1458
1459   CHECK_ADDRESS(address, INET, GSOCK_INVADDR);
1460
1461   addr = &(((struct sockaddr_in *)address->m_addr)->sin_addr);
1462   addr->s_addr = hostaddr;
1463
1464   return GSOCK_NOERROR;
1465 }
1466
1467 GSocketError GAddress_INET_SetPortName(GAddress *address, const char *port,
1468                                        const char *protocol)
1469 {
1470   struct servent *se;
1471   struct sockaddr_in *addr;
1472
1473   assert(address != NULL);
1474   CHECK_ADDRESS(address, INET, GSOCK_INVADDR);
1475
1476   if (!port)
1477   {
1478     address->m_error = GSOCK_INVPORT;
1479     return GSOCK_INVPORT;
1480   }
1481
1482   se = getservbyname(port, protocol);
1483   if (!se)
1484   {
1485     if (isdigit(port[0]))
1486     {
1487       int port_int;
1488
1489       port_int = atoi(port);
1490       addr = (struct sockaddr_in *)address->m_addr;
1491       addr->sin_port = htons(port_int);
1492       return GSOCK_NOERROR;
1493     }
1494
1495     address->m_error = GSOCK_INVPORT;
1496     return GSOCK_INVPORT;
1497   }
1498
1499   addr = (struct sockaddr_in *)address->m_addr;
1500   addr->sin_port = se->s_port;
1501
1502   return GSOCK_NOERROR;
1503 }
1504
1505 GSocketError GAddress_INET_SetPort(GAddress *address, unsigned short port)
1506 {
1507   struct sockaddr_in *addr;
1508
1509   assert(address != NULL);
1510   CHECK_ADDRESS(address, INET, GSOCK_INVADDR);
1511
1512   addr = (struct sockaddr_in *)address->m_addr;
1513   addr->sin_port = htons(port);
1514
1515   return GSOCK_NOERROR;
1516 }
1517
1518 GSocketError GAddress_INET_GetHostName(GAddress *address, char *hostname, size_t sbuf)
1519 {
1520   struct hostent *he;
1521   char *addr_buf;
1522   struct sockaddr_in *addr;
1523
1524   assert(address != NULL);
1525   CHECK_ADDRESS(address, INET, GSOCK_INVADDR);
1526
1527   addr = (struct sockaddr_in *)address->m_addr;
1528   addr_buf = (char *)&(addr->sin_addr);
1529
1530   he = gethostbyaddr(addr_buf, sizeof(addr->sin_addr), AF_INET);
1531   if (he == NULL)
1532   {
1533     address->m_error = GSOCK_NOHOST;
1534     return GSOCK_NOHOST;
1535   }
1536
1537   strncpy(hostname, he->h_name, sbuf);
1538
1539   return GSOCK_NOERROR;
1540 }
1541
1542 unsigned long GAddress_INET_GetHostAddress(GAddress *address)
1543 {
1544   struct sockaddr_in *addr;
1545
1546   assert(address != NULL);
1547   CHECK_ADDRESS(address, INET, 0);
1548
1549   addr = (struct sockaddr_in *)address->m_addr;
1550
1551   return addr->sin_addr.s_addr;
1552 }
1553
1554 unsigned short GAddress_INET_GetPort(GAddress *address)
1555 {
1556   struct sockaddr_in *addr;
1557
1558   assert(address != NULL);
1559   CHECK_ADDRESS(address, INET, 0);
1560
1561   addr = (struct sockaddr_in *)address->m_addr;
1562   return ntohs(addr->sin_port);
1563 }
1564
1565 /*
1566  * -------------------------------------------------------------------------
1567  * Unix address family
1568  * -------------------------------------------------------------------------
1569  */
1570
1571 GSocketError _GAddress_Init_UNIX(GAddress *address)
1572 {
1573   address->m_len  = sizeof(struct sockaddr_un);
1574   address->m_addr = (struct sockaddr *)malloc(address->m_len);
1575   if (address->m_addr == NULL)
1576   {
1577     address->m_error = GSOCK_MEMERR;
1578     return GSOCK_MEMERR;
1579   }
1580
1581   address->m_family = GSOCK_UNIX;
1582   address->m_realfamily = PF_UNIX;
1583   ((struct sockaddr_un *)address->m_addr)->sun_family = AF_UNIX;
1584   ((struct sockaddr_un *)address->m_addr)->sun_path[0] = 0;
1585
1586   return TRUE;
1587 }
1588
1589 GSocketError GAddress_UNIX_SetPath(GAddress *address, const char *path)
1590 {
1591   struct sockaddr_un *addr;
1592
1593   assert(address != NULL);
1594
1595   CHECK_ADDRESS(address, UNIX, GSOCK_INVADDR);
1596
1597   addr = ((struct sockaddr_un *)address->m_addr);
1598   memcpy(addr->sun_path, path, strlen(path));
1599
1600   return GSOCK_NOERROR;
1601 }
1602
1603 GSocketError GAddress_UNIX_GetPath(GAddress *address, char *path, size_t sbuf)
1604 {
1605   struct sockaddr_un *addr;
1606
1607   assert(address != NULL);
1608   CHECK_ADDRESS(address, UNIX, GSOCK_INVADDR);
1609
1610   addr = (struct sockaddr_un *)address->m_addr;
1611
1612   strncpy(path, addr->sun_path, sbuf);
1613
1614   return GSOCK_NOERROR;
1615 }
1616
1617 #endif  /* wxUSE_SOCKETS || defined(__GSOCKET_STANDALONE__) */
1618