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
  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 /*
  75  * INADDR_BROADCAST is identical to INADDR_NONE which is not defined
  76  * on all unices. INADDR_BROADCAST should be fine to indicate an error.
  77  */
  78 #ifndef INADDR_NONE
  79 #define INADDR_NONE INADDR_BROADCAST
  80 #endif
  81
  82 #define MASK_SIGNAL()                       \
  83 {                                           \
  84   void (*old_handler)(int);                 \
  85                                             \
  86   old_handler = signal(SIGPIPE, SIG_IGN);
  87
  88 #define UNMASK_SIGNAL()                     \
  89   signal(SIGPIPE, old_handler);             \
  90 }
  91
  92
  93 #ifndef __GSOCKET_STANDALONE__
  94
  95 #include "wx/gsocket.h"
  96 #include "wx/unix/gsockunx.h"
  97
  98 #else
  99
 100 #include "gsocket.h"
 101 #include "gsockunx.h"
 102
 103 #endif /* __GSOCKET_STANDALONE__ */
 104
 105
 106 /* Global initialisers */
 107
 108 bool GSocket_Init()
 109 {
 110   return TRUE;
 111 }
 112
 113 void GSocket_Cleanup()
 114 {
 115 }
 116
 117 /* Constructors / Destructors for GSocket */
 118
 119 GSocket *GSocket_new()
 120 {
 121   int i;
 122   GSocket *socket;
 123
 124   socket = (GSocket *)malloc(sizeof(GSocket));
 125
 126   if (socket == NULL)
 127     return NULL;
 128
 129   socket->m_fd                  = -1;
 130   for (i=0;i<GSOCK_MAX_EVENT;i++)
 131   {
 132     socket->m_cbacks[i]         = NULL;
 133   }
 134   socket->m_detected            = 0;
 135   socket->m_local               = NULL;
 136   socket->m_peer                = NULL;
 137   socket->m_error               = GSOCK_NOERROR;
 138   socket->m_server              = FALSE;
 139   socket->m_stream              = TRUE;
 140   socket->m_gui_dependent       = NULL;
 141   socket->m_non_blocking        = FALSE;
 142   socket->m_timeout             = 10*60*1000;
 143                                 /* 10 minutes * 60 sec * 1000 millisec */
 144   socket->m_establishing        = FALSE;
 145
 146   /* We initialize the GUI specific entries here */
 147   _GSocket_GUI_Init(socket);
 148
 149   return socket;
 150 }
 151
 152 void GSocket_destroy(GSocket *socket)
 153 {
 154   assert(socket != NULL);
 155
 156   /* Check that the socket is really shutdowned */
 157   if (socket->m_fd != -1)
 158     GSocket_Shutdown(socket);
 159
 160   /* We destroy GUI specific variables */
 161   _GSocket_GUI_Destroy(socket);
 162
 163   /* Destroy private addresses */
 164   if (socket->m_local)
 165     GAddress_destroy(socket->m_local);
 166
 167   if (socket->m_peer)
 168     GAddress_destroy(socket->m_peer);
 169
 170   /* Destroy the socket itself */
 171   free(socket);
 172 }
 173
 174 /* GSocket_Shutdown:
 175  *  Disallow further read/write operations on this socket, close
 176  *  the fd and disable all callbacks.
 177  */
 178 void GSocket_Shutdown(GSocket *socket)
 179 {
 180   int evt;
 181
 182   assert(socket != NULL);
 183
 184   /* If socket has been created, shutdown it */
 185   if (socket->m_fd != -1)
 186   {
 187     shutdown(socket->m_fd, 2);
 188     close(socket->m_fd);
 189     socket->m_fd = -1;
 190   }
 191
 192   /* Disable GUI callbacks */
 193   for (evt = 0; evt < GSOCK_MAX_EVENT; evt++)
 194     socket->m_cbacks[evt] = NULL;
 195
 196   socket->m_detected = 0;
 197   _GSocket_Disable_Events(socket);
 198 }
 199
 200 /* Address handling */
 201
 202 /* GSocket_SetLocal:
 203  * GSocket_GetLocal:
 204  * GSocket_SetPeer:
 205  * GSocket_GetPeer:
 206  *  Set or get the local or peer address for this socket. The 'set'
 207  *  functions return GSOCK_NOERROR on success, an error code otherwise.
 208  *  The 'get' functions return a pointer to a GAddress object on success,
 209  *  or NULL otherwise, in which case they set the error code of the
 210  *  corresponding GSocket.
 211  *
 212  *  Error codes:
 213  *    GSOCK_INVSOCK - the socket is not valid.
 214  *    GSOCK_INVADDR - the address is not valid.
 215  */
 216 GSocketError GSocket_SetLocal(GSocket *socket, GAddress *address)
 217 {
 218   assert(socket != NULL);
 219
 220   /* the socket must be initialized, or it must be a server */
 221   if ((socket->m_fd != -1 && !socket->m_server))
 222   {
 223     socket->m_error = GSOCK_INVSOCK;
 224     return GSOCK_INVSOCK;
 225   }
 226
 227   /* check address */
 228   if (address == NULL || address->m_family == GSOCK_NOFAMILY)
 229   {
 230     socket->m_error = GSOCK_INVADDR;
 231     return GSOCK_INVADDR;
 232   }
 233
 234   if (socket->m_local)
 235     GAddress_destroy(socket->m_local);
 236
 237   socket->m_local = GAddress_copy(address);
 238
 239   return GSOCK_NOERROR;
 240 }
 241
 242 GSocketError GSocket_SetPeer(GSocket *socket, GAddress *address)
 243 {
 244   assert(socket != NULL);
 245
 246   /* check address */
 247   if (address == NULL || address->m_family == GSOCK_NOFAMILY)
 248   {
 249     socket->m_error = GSOCK_INVADDR;
 250     return GSOCK_INVADDR;
 251   }
 252
 253   if (socket->m_peer)
 254     GAddress_destroy(socket->m_peer);
 255
 256   socket->m_peer = GAddress_copy(address);
 257
 258   return GSOCK_NOERROR;
 259 }
 260
 261 GAddress *GSocket_GetLocal(GSocket *socket)
 262 {
 263   GAddress *address;
 264   struct sockaddr addr;
 265   SOCKLEN_T size = sizeof(addr);
 266   GSocketError err;
 267
 268   assert(socket != NULL);
 269
 270   /* try to get it from the m_local var first */
 271   if (socket->m_local)
 272     return GAddress_copy(socket->m_local);
 273
 274   /* else, if the socket is initialized, try getsockname */
 275   if (socket->m_fd == -1)
 276   {
 277     socket->m_error = GSOCK_INVSOCK;
 278     return NULL;
 279   }
 280
 281   if (getsockname(socket->m_fd, &addr, (SOCKLEN_T *) &size) < 0)
 282   {
 283     socket->m_error = GSOCK_IOERR;
 284     return NULL;
 285   }
 286
 287   /* got a valid address from getsockname, create a GAddress object */
 288   address = GAddress_new();
 289   if (address == NULL)
 290   {
 291     socket->m_error = GSOCK_MEMERR;
 292     return NULL;
 293   }
 294
 295   err = _GAddress_translate_from(address, &addr, size);
 296   if (err != GSOCK_NOERROR)
 297   {
 298     GAddress_destroy(address);
 299     socket->m_error = err;
 300     return NULL;
 301   }
 302
 303   return address;
 304 }
 305
 306 GAddress *GSocket_GetPeer(GSocket *socket)
 307 {
 308   assert(socket != NULL);
 309
 310   /* try to get it from the m_peer var */
 311   if (socket->m_peer)
 312     return GAddress_copy(socket->m_peer);
 313
 314   return NULL;
 315 }
 316
 317 /* Server specific parts */
 318
 319 /* GSocket_SetServer:
 320  *  Sets up this socket as a server. The local address must have been
 321  *  set with GSocket_SetLocal() before GSocket_SetServer() is called.
 322  *  Returns GSOCK_NOERROR on success, one of the following otherwise:
 323  *
 324  *  Error codes:
 325  *    GSOCK_INVSOCK - the socket is in use.
 326  *    GSOCK_INVADDR - the local address has not been set.
 327  *    GSOCK_IOERR   - low-level error.
 328  */
 329 GSocketError GSocket_SetServer(GSocket *sck)
 330 {
 331   int type;
 332   int arg = 1;
 333
 334   assert(sck != NULL);
 335
 336   /* must not be in use */
 337   if (sck->m_fd != -1)
 338   {
 339     sck->m_error = GSOCK_INVSOCK;
 340     return GSOCK_INVSOCK;
 341   }
 342
 343   /* the local addr must have been set */
 344   if (!sck->m_local)
 345   {
 346     sck->m_error = GSOCK_INVADDR;
 347     return GSOCK_INVADDR;
 348   }
 349
 350   /* Initialize all fields */
 351   sck->m_stream   = TRUE;
 352   sck->m_server   = TRUE;
 353   sck->m_oriented = TRUE;
 354
 355   /* Create the socket */
 356   sck->m_fd = socket(sck->m_local->m_realfamily, SOCK_STREAM, 0);
 357
 358   if (sck->m_fd == -1)
 359   {
 360     sck->m_error = GSOCK_IOERR;
 361     return GSOCK_IOERR;
 362   }
 363
 364   ioctl(sck->m_fd, FIONBIO, &arg);
 365   _GSocket_Enable_Events(sck);
 366
 367   /* Bind to the local address,
 368    * retrieve the actual address bound,
 369    * and listen up to 5 connections.
 370    */
 371   if ((bind(sck->m_fd, sck->m_local->m_addr, sck->m_local->m_len) != 0) ||
 372       (getsockname(sck->m_fd,
 373                    sck->m_local->m_addr,
 374                    (SOCKLEN_T *) &sck->m_local->m_len) != 0) ||
 375       (listen(sck->m_fd, 5) != 0))
 376   {
 377     close(sck->m_fd);
 378     sck->m_fd = -1;
 379     sck->m_error = GSOCK_IOERR;
 380     return GSOCK_IOERR;
 381   }
 382
 383   return GSOCK_NOERROR;
 384 }
 385
 386 /* GSocket_WaitConnection:
 387  *  Waits for an incoming client connection. Returns a pointer to
 388  *  a GSocket object, or NULL if there was an error, in which case
 389  *  the last error field will be updated for the calling GSocket.
 390  *
 391  *  Error codes (set in the calling GSocket)
 392  *    GSOCK_INVSOCK    - the socket is not valid or not a server.
 393  *    GSOCK_TIMEDOUT   - timeout, no incoming connections.
 394  *    GSOCK_WOULDBLOCK - the call would block and the socket is nonblocking.
 395  *    GSOCK_MEMERR     - couldn't allocate memory.
 396  *    GSOCK_IOERR      - low-level error.
 397  */
 398 GSocket *GSocket_WaitConnection(GSocket *socket)
 399 {
 400   struct sockaddr from;
 401   SOCKLEN_T fromlen = sizeof(from);
 402   GSocket *connection;
 403   GSocketError err;
 404   int arg = 1;
 405
 406   assert(socket != NULL);
 407
 408   /* Reenable CONNECTION events */
 409   _GSocket_Enable(socket, GSOCK_CONNECTION);
 410
 411   /* If the socket has already been created, we exit immediately */
 412   if (socket->m_fd == -1 || !socket->m_server)
 413   {
 414     socket->m_error = GSOCK_INVSOCK;
 415     return NULL;
 416   }
 417
 418   /* Create a GSocket object for the new connection */
 419   connection = GSocket_new();
 420
 421   if (!connection)
 422   {
 423     socket->m_error = GSOCK_MEMERR;
 424     return NULL;
 425   }
 426
 427   /* Wait for a connection (with timeout) */
 428   if (_GSocket_Input_Timeout(socket) == GSOCK_TIMEDOUT)
 429   {
 430     GSocket_destroy(connection);
 431     /* socket->m_error set by _GSocket_Input_Timeout */
 432     return NULL;
 433   }
 434
 435   connection->m_fd = accept(socket->m_fd, &from, (SOCKLEN_T *) &fromlen);
 436
 437   if (connection->m_fd == -1)
 438   {
 439     if (errno == EWOULDBLOCK)
 440       socket->m_error = GSOCK_WOULDBLOCK;
 441     else
 442       socket->m_error = GSOCK_IOERR;
 443
 444     GSocket_destroy(connection);
 445     return NULL;
 446   }
 447
 448   /* Initialize all fields */
 449   connection->m_server   = FALSE;
 450   connection->m_stream   = TRUE;
 451   connection->m_oriented = TRUE;
 452
 453   /* Setup the peer address field */
 454   connection->m_peer = GAddress_new();
 455   if (!connection->m_peer)
 456   {
 457     GSocket_destroy(connection);
 458     socket->m_error = GSOCK_MEMERR;
 459     return NULL;
 460   }
 461   err = _GAddress_translate_from(connection->m_peer, &from, fromlen);
 462   if (err != GSOCK_NOERROR)
 463   {
 464     GAddress_destroy(connection->m_peer);
 465     GSocket_destroy(connection);
 466     socket->m_error = err;
 467     return NULL;
 468   }
 469
 470   ioctl(connection->m_fd, FIONBIO, &arg);
 471   _GSocket_Enable_Events(connection);
 472
 473   return connection;
 474 }
 475
 476 /* Datagram sockets */
 477
 478 /* GSocket_SetNonOriented:
 479  *  Sets up this socket as a non-connection oriented (datagram) socket.
 480  *  Before using this function, the local address must have been set
 481  *  with GSocket_SetLocal(), or the call will fail. Returns GSOCK_NOERROR
 482  *  on success, or one of the following otherwise.
 483  *
 484  *  Error codes:
 485  *    GSOCK_INVSOCK - the socket is in use.
 486  *    GSOCK_INVADDR - the local address has not been set.
 487  *    GSOCK_IOERR   - low-level error.
 488  */
 489 GSocketError GSocket_SetNonOriented(GSocket *sck)
 490 {
 491   int arg = 1;
 492
 493   assert(sck != NULL);
 494
 495   if (sck->m_fd != -1)
 496   {
 497     sck->m_error = GSOCK_INVSOCK;
 498     return GSOCK_INVSOCK;
 499   }
 500
 501   if (!sck->m_local)
 502   {
 503     sck->m_error = GSOCK_INVADDR;
 504     return GSOCK_INVADDR;
 505   }
 506
 507   /* Initialize all fields */
 508   sck->m_stream   = FALSE;
 509   sck->m_server   = FALSE;
 510   sck->m_oriented = FALSE;
 511
 512   /* Create the socket */
 513   sck->m_fd = socket(sck->m_local->m_realfamily, SOCK_DGRAM, 0);
 514
 515   if (sck->m_fd < 0)
 516   {
 517     sck->m_error = GSOCK_IOERR;
 518     return GSOCK_IOERR;
 519   }
 520
 521   ioctl(sck->m_fd, FIONBIO, &arg);
 522   _GSocket_Enable_Events(sck);
 523
 524   /* Bind to the local address,
 525    * and retrieve the actual address bound.
 526    */
 527   if ((bind(sck->m_fd, sck->m_local->m_addr, sck->m_local->m_len) != 0) ||
 528       (getsockname(sck->m_fd,
 529                    sck->m_local->m_addr,
 530                    (SOCKLEN_T *) &sck->m_local->m_len) != 0))
 531   {
 532     close(sck->m_fd);
 533     sck->m_fd    = -1;
 534     sck->m_error = GSOCK_IOERR;
 535     return GSOCK_IOERR;
 536   }
 537
 538   return GSOCK_NOERROR;
 539 }
 540
 541 /* Client specific parts */
 542
 543 /* GSocket_Connect:
 544  *  For stream (connection oriented) sockets, GSocket_Connect() tries
 545  *  to establish a client connection to a server using the peer address
 546  *  as established with GSocket_SetPeer(). Returns GSOCK_NOERROR if the
 547  *  connection has been succesfully established, or one of the error
 548  *  codes listed below. Note that for nonblocking sockets, a return
 549  *  value of GSOCK_WOULDBLOCK doesn't mean a failure. The connection
 550  *  request can be completed later; you should use GSocket_Select()
 551  *  to poll for GSOCK_CONNECTION | GSOCK_LOST, or wait for the
 552  *  corresponding asynchronous events.
 553  *
 554  *  For datagram (non connection oriented) sockets, GSocket_Connect()
 555  *  just sets the peer address established with GSocket_SetPeer() as
 556  *  default destination.
 557  *
 558  *  Error codes:
 559  *    GSOCK_INVSOCK    - the socket is in use or not valid.
 560  *    GSOCK_INVADDR    - the peer address has not been established.
 561  *    GSOCK_TIMEDOUT   - timeout, the connection failed.
 562  *    GSOCK_WOULDBLOCK - connection in progress (nonblocking sockets only)
 563  *    GSOCK_MEMERR     - couldn't allocate memory.
 564  *    GSOCK_IOERR      - low-level error.
 565  */
 566 GSocketError GSocket_Connect(GSocket *sck, GSocketStream stream)
 567 {
 568   int err, ret;
 569   int arg = 1;
 570
 571   assert(sck != NULL);
 572
 573   /* Enable CONNECTION events (needed for nonblocking connections) */
 574   _GSocket_Enable(sck, GSOCK_CONNECTION);
 575
 576   if (sck->m_fd != -1)
 577   {
 578     sck->m_error = GSOCK_INVSOCK;
 579     return GSOCK_INVSOCK;
 580   }
 581
 582   if (!sck->m_peer)
 583   {
 584     sck->m_error = GSOCK_INVADDR;
 585     return GSOCK_INVADDR;
 586   }
 587
 588   /* Streamed or dgram socket? */
 589   sck->m_stream   = (stream == GSOCK_STREAMED);
 590   sck->m_oriented = TRUE;
 591   sck->m_server   = FALSE;
 592   sck->m_establishing = FALSE;
 593
 594   /* Create the socket */
 595   sck->m_fd = socket(sck->m_peer->m_realfamily,
 596                      sck->m_stream? SOCK_STREAM : SOCK_DGRAM, 0);
 597
 598   if (sck->m_fd == -1)
 599   {
 600     sck->m_error = GSOCK_IOERR;
 601     return GSOCK_IOERR;
 602   }
 603
 604   ioctl(sck->m_fd, FIONBIO, &arg);
 605   _GSocket_Enable_Events(sck);
 606
 607   /* Connect it to the peer address, with a timeout (see below) */
 608   ret = connect(sck->m_fd, sck->m_peer->m_addr, sck->m_peer->m_len);
 609
 610   if (ret == -1)
 611   {
 612     err = errno;
 613
 614     /* If connect failed with EINPROGRESS and the GSocket object
 615      * is in blocking mode, we select() for the specified timeout
 616      * checking for writability to see if the connection request
 617      * completes.
 618      */
 619     if ((err == EINPROGRESS) && (!sck->m_non_blocking))
 620     {
 621       if (_GSocket_Output_Timeout(sck) == GSOCK_TIMEDOUT)
 622       {
 623         close(sck->m_fd);
 624         sck->m_fd = -1;
 625         /* sck->m_error is set in _GSocket_Output_Timeout */
 626         return GSOCK_TIMEDOUT;
 627       }
 628       else
 629       {
 630         int error;
 631         SOCKLEN_T len = sizeof(error);
 632
 633         getsockopt(sck->m_fd, SOL_SOCKET, SO_ERROR, (void*) &error, &len);
 634
 635         if (!error)
 636           return GSOCK_NOERROR;
 637       }
 638     }
 639
 640     /* If connect failed with EINPROGRESS and the GSocket object
 641      * is set to nonblocking, we set m_error to GSOCK_WOULDBLOCK
 642      * (and return GSOCK_WOULDBLOCK) but we don't close the socket;
 643      * this way if the connection completes, a GSOCK_CONNECTION
 644      * event will be generated, if enabled.
 645      */
 646     if ((err == EINPROGRESS) && (sck->m_non_blocking))
 647     {
 648       sck->m_establishing = TRUE;
 649       sck->m_error = GSOCK_WOULDBLOCK;
 650       return GSOCK_WOULDBLOCK;
 651     }
 652
 653     /* If connect failed with an error other than EINPROGRESS,
 654      * then the call to GSocket_Connect has failed.
 655      */
 656     close(sck->m_fd);
 657     sck->m_fd = -1;
 658     sck->m_error = GSOCK_IOERR;
 659     return GSOCK_IOERR;
 660   }
 661
 662   return GSOCK_NOERROR;
 663 }
 664
 665 /* Generic IO */
 666
 667 /* Like recv(), send(), ... */
 668 int GSocket_Read(GSocket *socket, char *buffer, int size)
 669 {
 670   int ret;
 671
 672   assert(socket != NULL);
 673
 674   /* Reenable INPUT events */
 675   _GSocket_Enable(socket, GSOCK_INPUT);
 676
 677   if (socket->m_fd == -1 || socket->m_server)
 678   {
 679     socket->m_error = GSOCK_INVSOCK;
 680     return -1;
 681   }
 682
 683   /* If the socket is blocking, wait for data (with a timeout) */
 684   if (_GSocket_Input_Timeout(socket) == GSOCK_TIMEDOUT)
 685     return -1;
 686
 687   /* Read the data */
 688   if (socket->m_stream)
 689     ret = _GSocket_Recv_Stream(socket, buffer, size);
 690   else
 691     ret = _GSocket_Recv_Dgram(socket, buffer, size);
 692
 693   if (ret == -1)
 694   {
 695     if (errno == EWOULDBLOCK)
 696       socket->m_error = GSOCK_WOULDBLOCK;
 697     else
 698       socket->m_error = GSOCK_IOERR;
 699   }
 700
 701   return ret;
 702 }
 703
 704 int GSocket_Write(GSocket *socket, const char *buffer, int size)
 705 {
 706   int ret;
 707
 708   assert(socket != NULL);
 709
 710   if (socket->m_fd == -1 || socket->m_server)
 711   {
 712     socket->m_error = GSOCK_INVSOCK;
 713     return -1;
 714   }
 715
 716   /* If the socket is blocking, wait for writability (with a timeout) */
 717   if (_GSocket_Output_Timeout(socket) == GSOCK_TIMEDOUT)
 718     return -1;
 719
 720   /* Write the data */
 721   if (socket->m_stream)
 722     ret = _GSocket_Send_Stream(socket, buffer, size);
 723   else
 724     ret = _GSocket_Send_Dgram(socket, buffer, size);
 725
 726   if (ret == -1)
 727   {
 728     if (errno == EWOULDBLOCK)
 729       socket->m_error = GSOCK_WOULDBLOCK;
 730     else
 731       socket->m_error = GSOCK_IOERR;
 732
 733     /* Only reenable OUTPUT events after an error (just like WSAAsyncSelect
 734      * in MSW). Once the first OUTPUT event is received, users can assume
 735      * that the socket is writable until a read operation fails. Only then
 736      * will further OUTPUT events be posted.
 737      */
 738     _GSocket_Enable(socket, GSOCK_OUTPUT);
 739     return -1;
 740   }
 741
 742   return ret;
 743 }
 744
 745 /* GSocket_Select:
 746  *  Polls the socket to determine its status. This function will
 747  *  check for the events specified in the 'flags' parameter, and
 748  *  it will return a mask indicating which operations can be
 749  *  performed. This function won't block, regardless of the
 750  *  mode (blocking | nonblocking) of the socket.
 751  */
 752 GSocketEventFlags GSocket_Select(GSocket *socket, GSocketEventFlags flags)
 753 {
 754   assert(socket != NULL);
 755
 756   return (flags & socket->m_detected);
 757 }
 758
 759 /* Flags */
 760
 761 /* GSocket_SetNonBlocking:
 762  *  Sets the socket to non-blocking mode. All IO calls will return
 763  *  immediately.
 764  */
 765 void GSocket_SetNonBlocking(GSocket *socket, bool non_block)
 766 {
 767   assert(socket != NULL);
 768
 769   socket->m_non_blocking = non_block;
 770 }
 771
 772 /* GSocket_SetTimeout:
 773  *  Sets the timeout for blocking calls. Time is expressed in
 774  *  milliseconds.
 775  */
 776 void GSocket_SetTimeout(GSocket *socket, unsigned long millisec)
 777 {
 778   assert(socket != NULL);
 779
 780   socket->m_timeout = millisec;
 781 }
 782
 783 /* GSocket_GetError:
 784  *  Returns the last error occured for this socket. Note that successful
 785  *  operations do not clear this back to GSOCK_NOERROR, so use it only
 786  *  after an error.
 787  */
 788 GSocketError GSocket_GetError(GSocket *socket)
 789 {
 790   assert(socket != NULL);
 791
 792   return socket->m_error;
 793 }
 794
 795 /* Callbacks */
 796
 797 /* GSOCK_INPUT:
 798  *   There is data to be read in the input buffer. If, after a read
 799  *   operation, there is still data available, the callback function will
 800  *   be called again.
 801  * GSOCK_OUTPUT:
 802  *   The socket is available for writing. That is, the next write call
 803  *   won't block. This event is generated only once, when the connection is
 804  *   first established, and then only if a call failed with GSOCK_WOULDBLOCK,
 805  *   when the output buffer empties again. This means that the app should
 806  *   assume that it can write since the first OUTPUT event, and no more
 807  *   OUTPUT events will be generated unless an error occurs.
 808  * GSOCK_CONNECTION:
 809  *   Connection succesfully established, for client sockets, or incoming
 810  *   client connection, for server sockets. Wait for this event (also watch
 811  *   out for GSOCK_LOST) after you issue a nonblocking GSocket_Connect() call.
 812  * GSOCK_LOST:
 813  *   The connection is lost (or a connection request failed); this could
 814  *   be due to a failure, or due to the peer closing it gracefully.
 815  */
 816
 817 /* GSocket_SetCallback:
 818  *  Enables the callbacks specified by 'flags'. Note that 'flags'
 819  *  may be a combination of flags OR'ed toghether, so the same
 820  *  callback function can be made to accept different events.
 821  *  The callback function must have the following prototype:
 822  *
 823  *  void function(GSocket *socket, GSocketEvent event, char *cdata)
 824  */
 825 void GSocket_SetCallback(GSocket *socket, GSocketEventFlags flags,
 826                          GSocketCallback callback, char *cdata)
 827 {
 828   int count;
 829
 830   assert(socket != NULL);
 831
 832   for (count = 0; count < GSOCK_MAX_EVENT; count++)
 833   {
 834     if ((flags & (1 << count)) != 0)
 835     {
 836       socket->m_cbacks[count] = callback;
 837       socket->m_data[count] = cdata;
 838     }
 839   }
 840 }
 841
 842 /* GSocket_UnsetCallback:
 843  *  Disables all callbacks specified by 'flags', which may be a
 844  *  combination of flags OR'ed toghether.
 845  */
 846 void GSocket_UnsetCallback(GSocket *socket, GSocketEventFlags flags)
 847 {
 848   int count;
 849
 850   assert(socket != NULL);
 851
 852   for (count = 0; count < GSOCK_MAX_EVENT; count++)
 853   {
 854     if ((flags & (1 << count)) != 0)
 855     {
 856       socket->m_cbacks[count] = NULL;
 857       socket->m_data[count] = NULL;
 858     }
 859   }
 860 }
 861
 862
 863 #define CALL_CALLBACK(socket, event) {                                  \
 864   _GSocket_Disable(socket, event);                                      \
 865   if (socket->m_cbacks[event])                                          \
 866     socket->m_cbacks[event](socket, event, socket->m_data[event]);      \
 867 }
 868
 869
 870 void _GSocket_Enable(GSocket *socket, GSocketEvent event)
 871 {
 872   socket->m_detected &= ~(1 << event);
 873   _GSocket_Install_Callback(socket, event);
 874 }
 875
 876 void _GSocket_Disable(GSocket *socket, GSocketEvent event)
 877 {
 878   socket->m_detected |= (1 << event);
 879   _GSocket_Uninstall_Callback(socket, event);
 880 }
 881
 882 /* _GSocket_Input_Timeout:
 883  *  For blocking sockets, wait until data is available or
 884  *  until timeout ellapses.
 885  */
 886 GSocketError _GSocket_Input_Timeout(GSocket *socket)
 887 {
 888   struct timeval tv;
 889   fd_set readfds;
 890
 891   /* Linux select() will overwrite the struct on return */
 892   tv.tv_sec  = (socket->m_timeout / 1000);
 893   tv.tv_usec = (socket->m_timeout % 1000) * 1000;
 894
 895   if (!socket->m_non_blocking)
 896   {
 897     FD_ZERO(&readfds);
 898     FD_SET(socket->m_fd, &readfds);
 899     if (select(socket->m_fd + 1, &readfds, NULL, NULL, &tv) == 0)
 900     {
 901       socket->m_error = GSOCK_TIMEDOUT;
 902       return GSOCK_TIMEDOUT;
 903     }
 904   }
 905   return GSOCK_NOERROR;
 906 }
 907
 908 /* _GSocket_Output_Timeout:
 909  *  For blocking sockets, wait until data can be sent without
 910  *  blocking or until timeout ellapses.
 911  */
 912 GSocketError _GSocket_Output_Timeout(GSocket *socket)
 913 {
 914   struct timeval tv;
 915   fd_set writefds;
 916
 917   /* Linux select() will overwrite the struct on return */
 918   tv.tv_sec  = (socket->m_timeout / 1000);
 919   tv.tv_usec = (socket->m_timeout % 1000) * 1000;
 920
 921   if (!socket->m_non_blocking)
 922   {
 923     FD_ZERO(&writefds);
 924     FD_SET(socket->m_fd, &writefds);
 925     if (select(socket->m_fd + 1, NULL, &writefds, NULL, &tv) == 0)
 926     {
 927       socket->m_error = GSOCK_TIMEDOUT;
 928       return GSOCK_TIMEDOUT;
 929     }
 930   }
 931   return GSOCK_NOERROR;
 932 }
 933
 934 int _GSocket_Recv_Stream(GSocket *socket, char *buffer, int size)
 935 {
 936   int ret;
 937
 938   MASK_SIGNAL();
 939   ret = recv(socket->m_fd, buffer, size, 0);
 940   UNMASK_SIGNAL();
 941
 942   return ret;
 943 }
 944
 945 int _GSocket_Recv_Dgram(GSocket *socket, char *buffer, int size)
 946 {
 947   struct sockaddr from;
 948   SOCKLEN_T fromlen = sizeof(from);
 949   int ret;
 950   GSocketError err;
 951
 952   fromlen = sizeof(from);
 953
 954   MASK_SIGNAL();
 955   ret = recvfrom(socket->m_fd, buffer, size, 0, &from, (SOCKLEN_T *) &fromlen);
 956   UNMASK_SIGNAL();
 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