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