src/unix/gsocket.cpp

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