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 {
  30     u_char  sun_len;        /* sockaddr len including null */
  31     u_char  sun_family;     /* AF_UNIX */
  32     char    sun_path[108];  /* path name (gag) */
  33 };
  34 #else
  35 #include <sys/socket.h>
  36 #include <sys/un.h>
  37 #endif
  38
  39 #include <sys/time.h>
  40 #include <netinet/in.h>
  41 #include <arpa/inet.h>
  42 #include <errno.h>
  43 #include <string.h>
  44 #include <unistd.h>
  45 #include <stdio.h>
  46 #include <stdlib.h>
  47 #include <stddef.h>
  48 #include <ctype.h>
  49 #ifdef sun
  50 #  include <sys/filio.h>
  51 #endif
  52 #ifdef sgi
  53 #  include <bstring.h>
  54 #endif
  55 #include <signal.h>
  56
  57 #ifndef SOCKLEN_T
  58
  59 #ifdef VMS
  60 #  define SOCKLEN_T unsigned int
  61 #else
  62 #  ifdef __GLIBC__
  63 #    if __GLIBC__ == 2
  64 #      define SOCKLEN_T socklen_t
  65 #    endif
  66 #  else
  67 #    define SOCKLEN_T int
  68 #  endif
  69 #endif
  70
  71 #endif /* SOCKLEN_T */
  72
  73 /*
  74  * MSW defines this, Unices don't.
  75  */
  76 #ifndef INVALID_SOCKET
  77 #define INVALID_SOCKET -1
  78 #endif
  79
  80 /*
  81  * INADDR_BROADCAST is identical to INADDR_NONE which is not defined
  82  * on all systems. INADDR_BROADCAST should be fine to indicate an error.
  83  */
  84 #ifndef INADDR_NONE
  85 #define INADDR_NONE INADDR_BROADCAST
  86 #endif
  87
  88 #define MASK_SIGNAL()                       \
  89 {                                           \
  90   void (*old_handler)(int);                 \
  91                                             \
  92   old_handler = signal(SIGPIPE, SIG_IGN);
  93
  94 #define UNMASK_SIGNAL()                     \
  95   signal(SIGPIPE, old_handler);             \
  96 }
  97
  98
  99 #ifndef __GSOCKET_STANDALONE__
 100 #  include "wx/unix/gsockunx.h"
 101 #  include "wx/gsocket.h"
 102 #else
 103 #  include "gsockunx.h"
 104 #  include "gsocket.h"
 105 #endif /* __GSOCKET_STANDALONE__ */
 106
 107 /* redefine some GUI-only functions to do nothing in console mode */
 108 #if defined(wxUSE_GUI) && !wxUSE_GUI
 109 #  define _GSocket_GUI_Init(socket) (1)
 110 #  define _GSocket_GUI_Destroy(socket)
 111 #  define _GSocket_Enable_Events(socket)
 112 #  define _GSocket_Disable_Events(socket)
 113 #  define _GSocket_Install_Callback(socket, event)
 114 #  define _GSocket_Uninstall_Callback(socket, event)
 115 #endif /* wxUSE_GUI */
 116
 117 /* debugging helpers */
 118 #ifdef __GSOCKET_DEBUG__
 119 #  define GSocket_Debug(args) printf args
 120 #else
 121 #  define GSocket_Debug(args)
 122 #endif /* __GSOCKET_DEBUG__ */
 123
 124 /* Global initialisers */
 125
 126 int GSocket_Init(void)
 127 {
 128   return 1;
 129 }
 130
 131 void GSocket_Cleanup(void)
 132 {
 133 }
 134
 135 /* Constructors / Destructors for GSocket */
 136
 137 GSocket *GSocket_new(void)
 138 {
 139   int i, success;
 140   GSocket *socket;
 141
 142   socket = (GSocket *)malloc(sizeof(GSocket));
 143
 144   if (socket == NULL)
 145     return NULL;
 146
 147   socket->m_fd                  = INVALID_SOCKET;
 148   for (i=0;i<GSOCK_MAX_EVENT;i++)
 149   {
 150     socket->m_cbacks[i]         = NULL;
 151   }
 152   socket->m_detected            = 0;
 153   socket->m_local               = NULL;
 154   socket->m_peer                = NULL;
 155   socket->m_error               = GSOCK_NOERROR;
 156   socket->m_server              = FALSE;
 157   socket->m_stream              = TRUE;
 158   socket->m_gui_dependent       = NULL;
 159   socket->m_non_blocking        = FALSE;
 160   socket->m_timeout             = 10*60*1000;
 161                                 /* 10 minutes * 60 sec * 1000 millisec */
 162   socket->m_establishing        = FALSE;
 163
 164   /* Per-socket GUI-specific initialization */
 165   success = _GSocket_GUI_Init(socket);
 166   if (!success)
 167   {
 168     free(socket);
 169     return NULL;
 170   }
 171
 172   return socket;
 173 }
 174
 175 void GSocket_destroy(GSocket *socket)
 176 {
 177   assert(socket != NULL);
 178
 179   /* Check that the socket is really shutdowned */
 180   if (socket->m_fd != INVALID_SOCKET)
 181     GSocket_Shutdown(socket);
 182
 183   /* Per-socket GUI-specific cleanup */
 184   _GSocket_GUI_Destroy(socket);
 185
 186   /* Destroy private addresses */
 187   if (socket->m_local)
 188     GAddress_destroy(socket->m_local);
 189
 190   if (socket->m_peer)
 191     GAddress_destroy(socket->m_peer);
 192
 193   /* Destroy the socket itself */
 194   free(socket);
 195 }
 196
 197 /* GSocket_Shutdown:
 198  *  Disallow further read/write operations on this socket, close
 199  *  the fd and disable all callbacks.
 200  */
 201 void GSocket_Shutdown(GSocket *socket)
 202 {
 203   int evt;
 204
 205   assert(socket != NULL);
 206
 207   /* If socket has been created, shutdown it */
 208   if (socket->m_fd != INVALID_SOCKET)
 209   {
 210     shutdown(socket->m_fd, 2);
 211     close(socket->m_fd);
 212     socket->m_fd = INVALID_SOCKET;
 213   }
 214
 215   /* Disable GUI callbacks */
 216   for (evt = 0; evt < GSOCK_MAX_EVENT; evt++)
 217     socket->m_cbacks[evt] = NULL;
 218
 219   socket->m_detected = GSOCK_LOST_FLAG;
 220   _GSocket_Disable_Events(socket);
 221 }
 222
 223 /* Address handling */
 224
 225 /* GSocket_SetLocal:
 226  * GSocket_GetLocal:
 227  * GSocket_SetPeer:
 228  * GSocket_GetPeer:
 229  *  Set or get the local or peer address for this socket. The 'set'
 230  *  functions return GSOCK_NOERROR on success, an error code otherwise.
 231  *  The 'get' functions return a pointer to a GAddress object on success,
 232  *  or NULL otherwise, in which case they set the error code of the
 233  *  corresponding GSocket.
 234  *
 235  *  Error codes:
 236  *    GSOCK_INVSOCK - the socket is not valid.
 237  *    GSOCK_INVADDR - the address is not valid.
 238  */
 239 GSocketError GSocket_SetLocal(GSocket *socket, GAddress *address)
 240 {
 241   assert(socket != NULL);
 242
 243   /* the socket must be initialized, or it must be a server */
 244   if ((socket->m_fd != INVALID_SOCKET && !socket->m_server))
 245   {
 246     socket->m_error = GSOCK_INVSOCK;
 247     return GSOCK_INVSOCK;
 248   }
 249
 250   /* check address */
 251   if (address == NULL || address->m_family == GSOCK_NOFAMILY)
 252   {
 253     socket->m_error = GSOCK_INVADDR;
 254     return GSOCK_INVADDR;
 255   }
 256
 257   if (socket->m_local)
 258     GAddress_destroy(socket->m_local);
 259
 260   socket->m_local = GAddress_copy(address);
 261
 262   return GSOCK_NOERROR;
 263 }
 264
 265 GSocketError GSocket_SetPeer(GSocket *socket, GAddress *address)
 266 {
 267   assert(socket != NULL);
 268
 269   /* check address */
 270   if (address == NULL || address->m_family == GSOCK_NOFAMILY)
 271   {
 272     socket->m_error = GSOCK_INVADDR;
 273     return GSOCK_INVADDR;
 274   }
 275
 276   if (socket->m_peer)
 277     GAddress_destroy(socket->m_peer);
 278
 279   socket->m_peer = GAddress_copy(address);
 280
 281   return GSOCK_NOERROR;
 282 }
 283
 284 GAddress *GSocket_GetLocal(GSocket *socket)
 285 {
 286   GAddress *address;
 287   struct sockaddr addr;
 288   SOCKLEN_T size = sizeof(addr);
 289   GSocketError err;
 290
 291   assert(socket != NULL);
 292
 293   /* try to get it from the m_local var first */
 294   if (socket->m_local)
 295     return GAddress_copy(socket->m_local);
 296
 297   /* else, if the socket is initialized, try getsockname */
 298   if (socket->m_fd == INVALID_SOCKET)
 299   {
 300     socket->m_error = GSOCK_INVSOCK;
 301     return NULL;
 302   }
 303
 304   if (getsockname(socket->m_fd, &addr, (SOCKLEN_T *) &size) < 0)
 305   {
 306     socket->m_error = GSOCK_IOERR;
 307     return NULL;
 308   }
 309
 310   /* got a valid address from getsockname, create a GAddress object */
 311   address = GAddress_new();
 312   if (address == NULL)
 313   {
 314     socket->m_error = GSOCK_MEMERR;
 315     return NULL;
 316   }
 317
 318   err = _GAddress_translate_from(address, &addr, size);
 319   if (err != GSOCK_NOERROR)
 320   {
 321     GAddress_destroy(address);
 322     socket->m_error = err;
 323     return NULL;
 324   }
 325
 326   return address;
 327 }
 328
 329 GAddress *GSocket_GetPeer(GSocket *socket)
 330 {
 331   assert(socket != NULL);
 332
 333   /* try to get it from the m_peer var */
 334   if (socket->m_peer)
 335     return GAddress_copy(socket->m_peer);
 336
 337   return NULL;
 338 }
 339
 340 /* Server specific parts */
 341
 342 /* GSocket_SetServer:
 343  *  Sets up this socket as a server. The local address must have been
 344  *  set with GSocket_SetLocal() before GSocket_SetServer() is called.
 345  *  Returns GSOCK_NOERROR on success, one of the following otherwise:
 346  *
 347  *  Error codes:
 348  *    GSOCK_INVSOCK - the socket is in use.
 349  *    GSOCK_INVADDR - the local address has not been set.
 350  *    GSOCK_IOERR   - low-level error.
 351  */
 352 GSocketError GSocket_SetServer(GSocket *sck)
 353 {
 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, int 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 address family is either
1246  * GSOCK_NOFAMILY or GSOCK_*family*, and if it is GSOCK_NOFAMILY, it
1247  * initalizes it to be a GSOCK_*family*. In other cases, it returns
1248  * an appropiate error code.
1249  *
1250  * CHECK_ADDRESS_RETVAL does the same but returning 'retval' on error.
1251  */
1252 #define CHECK_ADDRESS(address, family)                              \
1253 {                                                                   \
1254   if (address->m_family == GSOCK_NOFAMILY)                          \
1255     if (_GAddress_Init_##family(address) != GSOCK_NOERROR)          \
1256       return address->m_error;                                      \
1257   if (address->m_family != GSOCK_##family)                          \
1258   {                                                                 \
1259     address->m_error = GSOCK_INVADDR;                               \
1260     return GSOCK_INVADDR;                                           \
1261   }                                                                 \
1262 }
1263
1264 #define CHECK_ADDRESS_RETVAL(address, family, retval)               \
1265 {                                                                   \
1266   if (address->m_family == GSOCK_NOFAMILY)                          \
1267     if (_GAddress_Init_##family(address) != GSOCK_NOERROR)          \
1268       return retval;                                                \
1269   if (address->m_family != GSOCK_##family)                          \
1270   {                                                                 \
1271     address->m_error = GSOCK_INVADDR;                               \
1272     return retval;                                                  \
1273   }                                                                 \
1274 }
1275
1276
1277 GAddress *GAddress_new(void)
1278 {
1279   GAddress *address;
1280
1281   if ((address = (GAddress *) malloc(sizeof(GAddress))) == NULL)
1282     return NULL;
1283
1284   address->m_family  = GSOCK_NOFAMILY;
1285   address->m_addr    = NULL;
1286   address->m_len     = 0;
1287
1288   return address;
1289 }
1290
1291 GAddress *GAddress_copy(GAddress *address)
1292 {
1293   GAddress *addr2;
1294
1295   assert(address != NULL);
1296
1297   if ((addr2 = (GAddress *) malloc(sizeof(GAddress))) == NULL)
1298     return NULL;
1299
1300   memcpy(addr2, address, sizeof(GAddress));
1301
1302   if (address->m_addr)
1303   {
1304     addr2->m_addr = (struct sockaddr *)malloc(addr2->m_len);
1305     if (addr2->m_addr == NULL)
1306     {
1307       free(addr2);
1308       return NULL;
1309     }
1310     memcpy(addr2->m_addr, address->m_addr, addr2->m_len);
1311   }
1312
1313   return addr2;
1314 }
1315
1316 void GAddress_destroy(GAddress *address)
1317 {
1318   assert(address != NULL);
1319
1320   if (address->m_addr)
1321     free(address->m_addr);
1322
1323   free(address);
1324 }
1325
1326 void GAddress_SetFamily(GAddress *address, GAddressType type)
1327 {
1328   assert(address != NULL);
1329
1330   address->m_family = type;
1331 }
1332
1333 GAddressType GAddress_GetFamily(GAddress *address)
1334 {
1335   assert(address != NULL);
1336
1337   return address->m_family;
1338 }
1339
1340 GSocketError _GAddress_translate_from(GAddress *address,
1341                                       struct sockaddr *addr, int len)
1342 {
1343   address->m_realfamily = addr->sa_family;
1344   switch (addr->sa_family)
1345   {
1346     case AF_INET:
1347       address->m_family = GSOCK_INET;
1348       break;
1349     case AF_UNIX:
1350       address->m_family = GSOCK_UNIX;
1351       break;
1352 #ifdef AF_INET6
1353     case AF_INET6:
1354       address->m_family = GSOCK_INET6;
1355       break;
1356 #endif
1357     default:
1358     {
1359       address->m_error = GSOCK_INVOP;
1360       return GSOCK_INVOP;
1361     }
1362   }
1363
1364   if (address->m_addr)
1365     free(address->m_addr);
1366
1367   address->m_len  = len;
1368   address->m_addr = (struct sockaddr *)malloc(len);
1369
1370   if (address->m_addr == NULL)
1371   {
1372     address->m_error = GSOCK_MEMERR;
1373     return GSOCK_MEMERR;
1374   }
1375   memcpy(address->m_addr, addr, len);
1376
1377   return GSOCK_NOERROR;
1378 }
1379
1380 GSocketError _GAddress_translate_to(GAddress *address,
1381                                     struct sockaddr **addr, int *len)
1382 {
1383   if (!address->m_addr)
1384   {
1385     address->m_error = GSOCK_INVADDR;
1386     return GSOCK_INVADDR;
1387   }
1388
1389   *len = address->m_len;
1390   *addr = (struct sockaddr *)malloc(address->m_len);
1391   if (*addr == NULL)
1392   {
1393     address->m_error = GSOCK_MEMERR;
1394     return GSOCK_MEMERR;
1395   }
1396
1397   memcpy(*addr, address->m_addr, address->m_len);
1398   return GSOCK_NOERROR;
1399 }
1400
1401 /*
1402  * -------------------------------------------------------------------------
1403  * Internet address family
1404  * -------------------------------------------------------------------------
1405  */
1406
1407 GSocketError _GAddress_Init_INET(GAddress *address)
1408 {
1409   address->m_len  = sizeof(struct sockaddr_in);
1410   address->m_addr = (struct sockaddr *) malloc(address->m_len);
1411   if (address->m_addr == NULL)
1412   {
1413     address->m_error = GSOCK_MEMERR;
1414     return GSOCK_MEMERR;
1415   }
1416
1417   address->m_family = GSOCK_INET;
1418   address->m_realfamily = PF_INET;
1419   ((struct sockaddr_in *)address->m_addr)->sin_family = AF_INET;
1420   ((struct sockaddr_in *)address->m_addr)->sin_addr.s_addr = INADDR_ANY;
1421
1422   return GSOCK_NOERROR;
1423 }
1424
1425 GSocketError GAddress_INET_SetHostName(GAddress *address, const char *hostname)
1426 {
1427   struct hostent *he;
1428   struct in_addr *addr;
1429
1430   assert(address != NULL);
1431
1432   CHECK_ADDRESS(address, INET);
1433
1434   addr = &(((struct sockaddr_in *)address->m_addr)->sin_addr);
1435
1436   /* If it is a numeric host name, convert it now */
1437 #if defined(HAVE_INET_ATON)
1438   if (inet_aton(hostname, addr) == 0)
1439   {
1440 #elif defined(HAVE_INET_ADDR)
1441   if ( (addr->s_addr = inet_addr(hostname)) == -1 )
1442   {
1443 #else
1444   /* Use gethostbyname by default */
1445   if (1)
1446   {
1447 #endif
1448     struct in_addr *array_addr;
1449
1450     /* It is a real name, we solve it */
1451     if ((he = gethostbyname(hostname)) == NULL)
1452     {
1453       /* Reset to invalid address */
1454       addr->s_addr = INADDR_NONE;
1455       address->m_error = GSOCK_NOHOST;
1456       return GSOCK_NOHOST;
1457     }
1458     array_addr = (struct in_addr *) *(he->h_addr_list);
1459     addr->s_addr = array_addr[0].s_addr;
1460   }
1461   return GSOCK_NOERROR;
1462 }
1463
1464 GSocketError GAddress_INET_SetAnyAddress(GAddress *address)
1465 {
1466   return GAddress_INET_SetHostAddress(address, INADDR_ANY);
1467 }
1468
1469 GSocketError GAddress_INET_SetHostAddress(GAddress *address,
1470                                           unsigned long hostaddr)
1471 {
1472   struct in_addr *addr;
1473
1474   assert(address != NULL);
1475
1476   CHECK_ADDRESS(address, INET);
1477
1478   addr = &(((struct sockaddr_in *)address->m_addr)->sin_addr);
1479   addr->s_addr = hostaddr;
1480
1481   return GSOCK_NOERROR;
1482 }
1483
1484 GSocketError GAddress_INET_SetPortName(GAddress *address, const char *port,
1485                                        const char *protocol)
1486 {
1487   struct servent *se;
1488   struct sockaddr_in *addr;
1489
1490   assert(address != NULL);
1491   CHECK_ADDRESS(address, INET);
1492
1493   if (!port)
1494   {
1495     address->m_error = GSOCK_INVPORT;
1496     return GSOCK_INVPORT;
1497   }
1498
1499   se = getservbyname(port, protocol);
1500   if (!se)
1501   {
1502     if (isdigit(port[0]))
1503     {
1504       int port_int;
1505
1506       port_int = atoi(port);
1507       addr = (struct sockaddr_in *)address->m_addr;
1508       addr->sin_port = htons(port_int);
1509       return GSOCK_NOERROR;
1510     }
1511
1512     address->m_error = GSOCK_INVPORT;
1513     return GSOCK_INVPORT;
1514   }
1515
1516   addr = (struct sockaddr_in *)address->m_addr;
1517   addr->sin_port = se->s_port;
1518
1519   return GSOCK_NOERROR;
1520 }
1521
1522 GSocketError GAddress_INET_SetPort(GAddress *address, unsigned short port)
1523 {
1524   struct sockaddr_in *addr;
1525
1526   assert(address != NULL);
1527   CHECK_ADDRESS(address, INET);
1528
1529   addr = (struct sockaddr_in *)address->m_addr;
1530   addr->sin_port = htons(port);
1531
1532   return GSOCK_NOERROR;
1533 }
1534
1535 GSocketError GAddress_INET_GetHostName(GAddress *address, char *hostname, size_t sbuf)
1536 {
1537   struct hostent *he;
1538   char *addr_buf;
1539   struct sockaddr_in *addr;
1540
1541   assert(address != NULL);
1542   CHECK_ADDRESS(address, INET);
1543
1544   addr = (struct sockaddr_in *)address->m_addr;
1545   addr_buf = (char *)&(addr->sin_addr);
1546
1547   he = gethostbyaddr(addr_buf, sizeof(addr->sin_addr), AF_INET);
1548   if (he == NULL)
1549   {
1550     address->m_error = GSOCK_NOHOST;
1551     return GSOCK_NOHOST;
1552   }
1553
1554   strncpy(hostname, he->h_name, sbuf);
1555
1556   return GSOCK_NOERROR;
1557 }
1558
1559 unsigned long GAddress_INET_GetHostAddress(GAddress *address)
1560 {
1561   struct sockaddr_in *addr;
1562
1563   assert(address != NULL);
1564   CHECK_ADDRESS_RETVAL(address, INET, 0);
1565
1566   addr = (struct sockaddr_in *)address->m_addr;
1567
1568   return addr->sin_addr.s_addr;
1569 }
1570
1571 unsigned short GAddress_INET_GetPort(GAddress *address)
1572 {
1573   struct sockaddr_in *addr;
1574
1575   assert(address != NULL);
1576   CHECK_ADDRESS_RETVAL(address, INET, 0);
1577
1578   addr = (struct sockaddr_in *)address->m_addr;
1579   return ntohs(addr->sin_port);
1580 }
1581
1582 /*
1583  * -------------------------------------------------------------------------
1584  * Unix address family
1585  * -------------------------------------------------------------------------
1586  */
1587
1588 GSocketError _GAddress_Init_UNIX(GAddress *address)
1589 {
1590   address->m_len  = sizeof(struct sockaddr_un);
1591   address->m_addr = (struct sockaddr *)malloc(address->m_len);
1592   if (address->m_addr == NULL)
1593   {
1594     address->m_error = GSOCK_MEMERR;
1595     return GSOCK_MEMERR;
1596   }
1597
1598   address->m_family = GSOCK_UNIX;
1599   address->m_realfamily = PF_UNIX;
1600   ((struct sockaddr_un *)address->m_addr)->sun_family = AF_UNIX;
1601   ((struct sockaddr_un *)address->m_addr)->sun_path[0] = 0;
1602
1603   return GSOCK_NOERROR;
1604 }
1605
1606 GSocketError GAddress_UNIX_SetPath(GAddress *address, const char *path)
1607 {
1608   struct sockaddr_un *addr;
1609
1610   assert(address != NULL);
1611
1612   CHECK_ADDRESS(address, UNIX);
1613
1614   addr = ((struct sockaddr_un *)address->m_addr);
1615   memcpy(addr->sun_path, path, strlen(path));
1616
1617   return GSOCK_NOERROR;
1618 }
1619
1620 GSocketError GAddress_UNIX_GetPath(GAddress *address, char *path, size_t sbuf)
1621 {
1622   struct sockaddr_un *addr;
1623
1624   assert(address != NULL);
1625   CHECK_ADDRESS(address, UNIX);
1626
1627   addr = (struct sockaddr_un *)address->m_addr;
1628
1629   strncpy(path, addr->sun_path, sbuf);
1630
1631   return GSOCK_NOERROR;
1632 }
1633
1634 #endif  /* wxUSE_SOCKETS || defined(__GSOCKET_STANDALONE__) */
1635