]> git.saurik.com Git - wxWidgets.git/blobdiff - src/common/socket.cpp
projects / wxWidgets.git / blobdiff
? search:
summary | shortlog | log | commit | commitdiff | tree
raw | inline | side by side
remove tests for bool and {const,static,reinterpret}_cast in configure and wx/defs...
[wxWidgets.git] / src / common / socket.cpp
diff --git a/src/common/socket.cpp b/src/common/socket.cpp
index cc5c83e84f3c9fbfd56c8b6a9860ee6474fa56c2..88ad1f1548cca1b7a5a3f09d701e07c8d76a636e 100644 (file)
--- a/src/common/socket.cpp
+++ b/src/common/socket.cpp
@@ -1,1171 +1,1507 @@
 /////////////////////////////////////////////////////////////////////////////
 /////////////////////////////////////////////////////////////////////////////
-// Name:       socket.cpp
+// Name:       src/common/socket.cpp
 // Purpose:    Socket handler classes
 // Authors:    Guilhem Lavaux, Guillermo Rodriguez Garcia
 // Created:    April 1997
 // Purpose:    Socket handler classes
 // Authors:    Guilhem Lavaux, Guillermo Rodriguez Garcia
 // Created:    April 1997
-// Updated:    July 1999
-// Copyright:  (C) 1999, 1998, 1997, Guilhem Lavaux
+// Copyright:  (C) 1999-1997, Guilhem Lavaux
+//             (C) 2000-1999, Guillermo Rodriguez Garcia
 // RCS_ID:     $Id$
 // RCS_ID:     $Id$
-// License:    see wxWindows license
+// License:    wxWindows licence
 /////////////////////////////////////////////////////////////////////////////
 
 /////////////////////////////////////////////////////////////////////////////
 
-#ifdef __GNUG__
-#pragma implementation "socket.h"
-#endif
+// ==========================================================================
+// Declarations
+// ==========================================================================
 
 // For compilers that support precompilation, includes "wx.h".
 #include "wx/wxprec.h"
 
 #ifdef __BORLANDC__
 
 // For compilers that support precompilation, includes "wx.h".
 #include "wx/wxprec.h"
 
 #ifdef __BORLANDC__
-#pragma hdrstop
+    #pragma hdrstop
 #endif
 
 #if wxUSE_SOCKETS
 
 #endif
 
 #if wxUSE_SOCKETS
 
-/////////////////////////////////////////////////////////////////////////////
-// wxWindows headers
-/////////////////////////////////////////////////////////////////////////////
-#include "wx/defs.h"
-#include "wx/object.h"
-#include "wx/string.h"
-#include "wx/timer.h"
-#include "wx/utils.h"
-#include "wx/module.h"
-#include "wx/log.h"
-
-#include <stdlib.h>
-#include <string.h>
-#include <ctype.h>
+#include "wx/socket.h"
 
 
-/////////////////////////////////////////////////////////////////////////////
-// wxSocket headers
-/////////////////////////////////////////////////////////////////////////////
+#ifndef WX_PRECOMP
+    #include "wx/object.h"
+    #include "wx/string.h"
+    #include "wx/intl.h"
+    #include "wx/log.h"
+    #include "wx/event.h"
+    #include "wx/app.h"
+    #include "wx/utils.h"
+    #include "wx/timer.h"
+    #include "wx/module.h"
+#endif
+
+#include "wx/apptrait.h"
 #include "wx/sckaddr.h"
 #include "wx/sckaddr.h"
-#include "wx/socket.h"
+#include "wx/stopwatch.h"
+#include "wx/thread.h"
+#include "wx/evtloop.h"
+
+// DLL options compatibility check:
+#include "wx/build.h"
+WX_CHECK_BUILD_OPTIONS("wxNet")
+
+// --------------------------------------------------------------------------
+// macros and constants
+// --------------------------------------------------------------------------
+
+// discard buffer
+#define MAX_DISCARD_SIZE (10 * 1024)
+
+#define wxTRACE_Socket _T("wxSocket")
+
+// --------------------------------------------------------------------------
+// wxWin macros
+// --------------------------------------------------------------------------
 
 
-// --------------------------------------------------------------
-// ClassInfos
-// --------------------------------------------------------------
-#if !USE_SHARED_LIBRARY
 IMPLEMENT_CLASS(wxSocketBase, wxObject)
 IMPLEMENT_CLASS(wxSocketServer, wxSocketBase)
 IMPLEMENT_CLASS(wxSocketClient, wxSocketBase)
 IMPLEMENT_CLASS(wxSocketBase, wxObject)
 IMPLEMENT_CLASS(wxSocketServer, wxSocketBase)
 IMPLEMENT_CLASS(wxSocketClient, wxSocketBase)
+IMPLEMENT_CLASS(wxDatagramSocket, wxSocketBase)
 IMPLEMENT_DYNAMIC_CLASS(wxSocketEvent, wxEvent)
 IMPLEMENT_DYNAMIC_CLASS(wxSocketEvent, wxEvent)
-#endif
+
+// --------------------------------------------------------------------------
+// private classes
+// --------------------------------------------------------------------------
 
 class wxSocketState : public wxObject
 {
 public:
 
 class wxSocketState : public wxObject
 {
 public:
-  bool notify_state;
-  GSocketEventFlags evt_notify_state;
-  wxSocketBase::wxSockFlags socket_flags;
-  wxSocketBase::wxSockCbk c_callback;
-  char *c_callback_data;
+    wxSocketFlags            m_flags;
+    wxSocketEventFlags       m_eventmask;
+    bool                     m_notify;
+    void                    *m_clientData;
+
+public:
+    wxSocketState() : wxObject() {}
+
+    DECLARE_NO_COPY_CLASS(wxSocketState)
+};
 
 
+// Conditionally make the socket non-blocking for the lifetime of this object.
+class wxSocketUnblocker
+{
 public:
 public:
-  wxSocketState() : wxObject() {}
+    wxSocketUnblocker(GSocket *socket, bool unblock = true)
+        : m_socket(socket),
+          m_unblock(unblock)
+    {
+        if ( m_unblock )
+            m_socket->SetNonBlocking(true);
+    }
+
+    ~wxSocketUnblocker()
+    {
+        if ( m_unblock )
+            m_socket->SetNonBlocking(false);
+    }
+
+private:
+    GSocket * const m_socket;
+    bool m_unblock;
+
+    DECLARE_NO_COPY_CLASS(wxSocketUnblocker)
 };
 
 };
 
-// --------------------------------------------------------------
-// wxSocketBase ctor and dtor
-// --------------------------------------------------------------
-
-wxSocketBase::wxSocketBase(wxSocketBase::wxSockFlags _flags,
-         wxSocketBase::wxSockType _type) :
-  wxEvtHandler(),
-  m_socket(NULL), m_flags(_flags), m_type(_type),
-  m_neededreq(0),
-  m_lcount(0), m_timeout(600),
-  m_unread(NULL), m_unrd_size(0), m_unrd_cur(0),
-  m_cbk(NULL), m_cdata(NULL),
-  m_connected(FALSE), m_establishing(FALSE),
-  m_notify_state(FALSE), m_id(-1),
-  m_defering(NO_DEFER), m_error(FALSE),
-  m_states()
+// ============================================================================
+// GSocketManager
+// ============================================================================
+
+GSocketManager *GSocketManager::ms_manager = NULL;
+
+/* static */
+void GSocketManager::Set(GSocketManager *manager)
 {
 {
+    wxASSERT_MSG( !ms_manager, "too late to set manager now" );
+
+    ms_manager = manager;
 }
 
 }
 
-wxSocketBase::wxSocketBase() :
-  wxEvtHandler(),
-  m_socket(NULL), m_flags(WAITALL | SPEED), m_type(SOCK_UNINIT),
-  m_neededreq(0),
-  m_lcount(0), m_timeout(600),
-  m_unread(NULL), m_unrd_size(0), m_unrd_cur(0),
-  m_cbk(NULL), m_cdata(NULL),
-  m_connected(FALSE), m_establishing(FALSE),
-  m_notify_state(FALSE), m_id(-1),
-  m_defering(NO_DEFER), m_error(FALSE),
-  m_states()
+/* static */
+void GSocketManager::Init()
 {
 {
+    wxASSERT_MSG( !ms_manager, "shouldn't be initialized twice" );
+
+    /*
+        Details: Initialize() creates a hidden window as a sink for socket
+        events, such as 'read completed'. wxMSW has only one message loop
+        for the main thread. If Initialize is called in a secondary thread,
+        the socket window will be created for the secondary thread, but
+        since there is no message loop on this thread, it will never
+        receive events and all socket operations will time out.
+        BTW, the main thread must not be stopped using sleep or block
+        on a semaphore (a bad idea in any case) or socket operations
+        will time out.
+
+        On the Mac side, Initialize() stores a pointer to the CFRunLoop for
+        the main thread. Because secondary threads do not have run loops,
+        adding event notifications to the "Current" loop would have no
+        effect at all, events would never fire.
+    */
+    wxASSERT_MSG( wxIsMainThread(),
+                    "sockets must be initialized from the main thread" );
+
+    wxAppConsole * const app = wxAppConsole::GetInstance();
+    wxCHECK_RET( app, "sockets can't be initialized without wxApp" );
+
+    ms_manager = app->GetTraits()->GetSocketManager();
 }
 
 }
 
-wxSocketBase::~wxSocketBase()
+// ==========================================================================
+// wxSocketBase
+// ==========================================================================
+
+// --------------------------------------------------------------------------
+// Initialization and shutdown
+// --------------------------------------------------------------------------
+
+// FIXME-MT: all this is MT-unsafe, of course, we should protect all accesses
+//           to m_countInit with a crit section
+size_t wxSocketBase::m_countInit = 0;
+
+bool wxSocketBase::IsInitialized()
+{
+    return m_countInit > 0;
+}
+
+bool wxSocketBase::Initialize()
 {
 {
-  if (m_unread)
-    free(m_unread);
+    if ( !m_countInit++ )
+    {
+        if ( !GSocket_Init() )
+        {
+            m_countInit--;
 
 
-  // Shutdown and close the socket
-  Close();
+            return false;
+        }
+    }
 
 
-  // Destroy the GSocket object
-  if (m_socket)
-    GSocket_destroy(m_socket);
+    return true;
 }
 
 }
 
-bool wxSocketBase::Close()
+void wxSocketBase::Shutdown()
 {
 {
-  if (m_socket)
-  {
-    // Disable callbacks
-    GSocket_UnsetCallback(m_socket, GSOCK_INPUT_FLAG | GSOCK_OUTPUT_FLAG |
-                                    GSOCK_LOST_FLAG | GSOCK_CONNECTION_FLAG);
-
-    // Shutdown the connection
-    GSocket_Shutdown(m_socket);
-    m_connected = FALSE;
-    m_establishing = FALSE;
-  }
-
-  return TRUE;
+    // we should be initialized
+    wxASSERT_MSG( m_countInit > 0, _T("extra call to Shutdown()") );
+    if ( --m_countInit == 0 )
+    {
+        GSocket_Cleanup();
+    }
 }
 
 }
 
-// --------------------------------------------------------------
-// wxSocketBase basic IO operations
-// --------------------------------------------------------------
-
-// GRG: I have made some changes to wxSocket internal event
-// system; now, all events (INPUT, OUTPUT, CONNECTION, LOST)
-// are always internally monitored; but users will only be
-// notified of these events they are interested in. So we
-// no longer have to change the event mask with SetNotify()
-// in internal functions like DeferRead, DeferWrite, and
-// the like. This solves a lot of problems.
-
-// GRG: I added m_error handling to IO operations. Now,
-// wxSocketBase::Error() correctly indicates if the last
-// operation from {Read, Write, ReadMsg, WriteMsg, Peek,
-// Unread, Discard} failed. Note that now, every function
-// that updates m_lcount, also updates m_error. While I
-// was at it, also fixed an UGLY bug in ReadMsg.
-
-class _wxSocketInternalTimer: public wxTimer
+// --------------------------------------------------------------------------
+// Ctor and dtor
+// --------------------------------------------------------------------------
+
+void wxSocketBase::Init()
 {
 {
-public:
-  int *m_state;
-  unsigned long m_new_val;
+    m_socket       = NULL;
+    m_type         = wxSOCKET_UNINIT;
+
+    // state
+    m_flags        = 0;
+    m_connected    =
+    m_establishing =
+    m_reading      =
+    m_writing      =
+    m_error        =
+    m_closed       = false;
+    m_lcount       = 0;
+    m_timeout      = 600;
+    m_beingDeleted = false;
+
+    // pushback buffer
+    m_unread       = NULL;
+    m_unrd_size    = 0;
+    m_unrd_cur     = 0;
+
+    // events
+    m_id           = wxID_ANY;
+    m_handler      = NULL;
+    m_clientData   = NULL;
+    m_notify       = false;
+    m_eventmask    = 0;
+
+    if ( !IsInitialized() )
+    {
+        // this Initialize() will be undone by wxSocketModule::OnExit(), all
+        // the other calls to it should be matched by a call to Shutdown()
+        Initialize();
+    }
+}
 
 
-  void Notify()
-  {
-    *m_state = m_new_val;  // Change the value
-  }
-};
+wxSocketBase::wxSocketBase()
+{
+    Init();
+}
 
 
-int wxSocketBase::DeferRead(char *buffer, wxUint32 nbytes)
+wxSocketBase::wxSocketBase(wxSocketFlags flags, wxSocketType type)
 {
 {
-  // Timer for timeout
-  _wxSocketInternalTimer timer;
+    Init();
 
 
-  wxASSERT(m_defering == NO_DEFER);
+    SetFlags(flags);
 
 
-  // Set the defering mode to READ.
-  m_defering = DEFER_READ;
+    m_type = type;
+}
+
+wxSocketBase::~wxSocketBase()
+{
+    // Just in case the app called Destroy() *and* then deleted the socket
+    // immediately: don't leave dangling pointers.
+    wxAppTraits *traits = wxTheApp ? wxTheApp->GetTraits() : NULL;
+    if ( traits )
+        traits->RemoveFromPendingDelete(this);
+
+    // Shutdown and close the socket
+    if (!m_beingDeleted)
+        Close();
+
+    // Destroy the GSocket object
+    if (m_socket)
+        delete m_socket;
+
+    // Free the pushback buffer
+    if (m_unread)
+        free(m_unread);
+}
 
 
-  // Set the current buffer.
-  m_defer_buffer = buffer;
-  m_defer_nbytes = nbytes;
-  m_defer_timer  = &timer;
+bool wxSocketBase::Destroy()
+{
+    // Delayed destruction: the socket will be deleted during the next idle
+    // loop iteration. This ensures that all pending events have been
+    // processed.
+    m_beingDeleted = true;
+
+    // Shutdown and close the socket
+    Close();
 
 
-  timer.m_state = (int *)&m_defer_buffer;
-  timer.m_new_val = 0;
+    // Supress events from now on
+    Notify(false);
+
+    // schedule this object for deletion
+    wxAppTraits *traits = wxTheApp ? wxTheApp->GetTraits() : NULL;
+    if ( traits )
+    {
+        // let the traits object decide what to do with us
+        traits->ScheduleForDestroy(this);
+    }
+    else // no app or no traits
+    {
+        // in wxBase we might have no app object at all, don't leak memory
+        delete this;
+    }
 
 
-  timer.Start(m_timeout * 1000, FALSE);
+    return true;
+}
 
 
-  // If the socket is readable, call DoDefer for the first time
-  if (GSocket_Select(m_socket, GSOCK_INPUT_FLAG))
-    DoDefer();
+// --------------------------------------------------------------------------
+// Basic IO calls
+// --------------------------------------------------------------------------
 
 
-  // Wait for buffer completion.
-  while (m_defer_buffer != NULL)
-    wxYield();
+// The following IO operations update m_error and m_lcount:
+// {Read, Write, ReadMsg, WriteMsg, Peek, Unread, Discard}
+//
+// TODO: Should Connect, Accept and AcceptWith update m_error?
 
 
-  timer.Stop();
+bool wxSocketBase::Close()
+{
+    // Interrupt pending waits
+    InterruptWait();
 
 
-  // Disable defering mode.
-  m_defering = NO_DEFER;
-  m_defer_timer = NULL;
+    if (m_socket)
+    {
+        // Disable callbacks
+        m_socket->UnsetCallback(
+            GSOCK_INPUT_FLAG |
+            GSOCK_OUTPUT_FLAG |
+            GSOCK_LOST_FLAG |
+            GSOCK_CONNECTION_FLAG
+        );
+
+        // Shutdown the connection
+        m_socket->Shutdown();
+    }
 
 
-  // Return the number of bytes read from the socket.
-  return nbytes-m_defer_nbytes;
+    m_connected = false;
+    m_establishing = false;
+    return true;
 }
 
 }
 
-wxSocketBase& wxSocketBase::Read(char* buffer, wxUint32 nbytes)
+wxSocketBase& wxSocketBase::Read(void* buffer, wxUint32 nbytes)
 {
 {
-  int ret = 1;
+    // Mask read events
+    m_reading = true;
 
 
-  m_error = FALSE;
+    m_lcount = DoRead(buffer, nbytes);
 
 
-  // we try this even if the connection has already been closed.
-  m_lcount = GetPushback(buffer, nbytes, FALSE);
-  nbytes -= m_lcount;
-  buffer += m_lcount;
+    // If in wxSOCKET_WAITALL mode, all bytes should have been read.
+    if (m_flags & wxSOCKET_WAITALL)
+        m_error = (m_lcount != nbytes);
+    else
+        m_error = (m_lcount == 0);
 
 
-  if (!m_connected || !m_socket)
-  {
-    // if no data retrieved AND not connected, it is an error.
-    if (!m_lcount)
-      m_error = TRUE;
+    // Allow read events from now on
+    m_reading = false;
 
     return *this;
 
     return *this;
-  }
+}
 
 
-  // If we have got the whole needed buffer, return immediately
-  if (!nbytes)
-  {
-    return *this;
-  }
+wxUint32 wxSocketBase::DoRead(void* buffer_, wxUint32 nbytes)
+{
+    // We use pointer arithmetic here which doesn't work with void pointers.
+    char *buffer = static_cast<char *>(buffer_);
+
+    // Try the push back buffer first, even before checking whether the socket
+    // is valid to allow reading previously pushed back data from an already
+    // closed socket.
+    wxUint32 total = GetPushback(buffer, nbytes, false);
+    nbytes -= total;
+    buffer += total;
+
+    // If it's indeed closed or if read everything, there is nothing more to do.
+    if ( !m_socket || !nbytes )
+        return total;
+
+    wxCHECK_MSG( buffer, 0, "NULL buffer" );
 
 
-  if (m_flags & SPEED & WAITALL)    // SPEED && WAITALL
-  {
-    while (ret > 0 && nbytes > 0)
+
+    // wxSOCKET_NOWAIT overrides all the other flags and means that we are
+    // polling the socket and don't block at all.
+    if ( m_flags & wxSOCKET_NOWAIT )
     {
     {
-      ret = GSocket_Read(m_socket, buffer, nbytes);
-      m_lcount += ret;
-      buffer += ret;
-      nbytes -= ret;
+        wxSocketUnblocker unblock(m_socket);
+        int ret = m_socket->Read(buffer, nbytes);
+        if ( ret < 0 )
+            return 0;
+
+        total += ret;
     }
     }
-    // In case the last call was an error ...
-    if (ret < 0)
-      m_lcount ++;
-  }
-  else if (m_flags & SPEED)         // SPEED && !WAITALL
-  {
-    ret = GSocket_Read(m_socket, buffer, nbytes);
-
-    if (ret > 0)
-      m_lcount += ret;
-  }
-  else                              // !SPEED
-  {
-    ret = DeferRead(buffer, nbytes);
-
-    if (ret > 0)
-      m_lcount += ret;
-  }
-
-  // If we have read some data, then it is not an error, even
-  // when in WAITALL mode, the last low-level IO call might
-  // have failed.
-  if (!m_lcount)
-    m_error = TRUE;
-
-  return *this;
+    else // blocking socket
+    {
+        for ( ;; )
+        {
+            // Wait until socket becomes ready for reading dispatching the GUI
+            // events in the meanwhile unless wxSOCKET_BLOCK was explicitly
+            // specified to disable this.
+            if ( !(m_flags & wxSOCKET_BLOCK) && !WaitForRead() )
+                break;
+
+            const int ret = m_socket->Read(buffer, nbytes);
+            if ( ret == 0 )
+            {
+                // for connection-oriented (e.g. TCP) sockets we can only read
+                // 0 bytes if the other end has been closed, and for
+                // connectionless ones (UDP) this flag doesn't make sense
+                // anyhow so we can set it to true too without doing any harm
+                m_closed = true;
+                break;
+            }
+
+            if ( ret < 0 )
+            {
+                // this will be always interpreted as error by Read()
+                return 0;
+            }
+
+            total += ret;
+
+            // If wxSOCKET_WAITALL is not set, we can leave now as we did read
+            // something and we don't need to wait for all nbytes bytes to be
+            // read.
+            if ( !(m_flags & wxSOCKET_WAITALL) )
+                break;
+
+            // Otherwise continue reading until we do read everything.
+            nbytes -= ret;
+            if ( !nbytes )
+                break;
+
+            buffer += ret;
+        }
+    }
+
+    return total;
 }
 
 }
 
-wxSocketBase& wxSocketBase::ReadMsg(char* buffer, wxUint32 nbytes)
+wxSocketBase& wxSocketBase::ReadMsg(void* buffer, wxUint32 nbytes)
 {
 {
-#define MAX_BUFSIZE (10 * 1024)
+    wxUint32 len, len2, sig, total;
+    bool error;
+    int old_flags;
+    struct
+    {
+        unsigned char sig[4];
+        unsigned char len[4];
+    } msg;
 
 
-  int old_flags;
-  unsigned long len, len2, sig;
-  struct
-  {
-    char sig[4];
-    char len[4];
-  } msg;
+    // Mask read events
+    m_reading = true;
 
 
-  // sig should be an explicit 32-bit unsigned integer; I've seen
-  // compilers in which wxUint32 was actually a 16-bit unsigned integer
+    total = 0;
+    error = true;
+    old_flags = m_flags;
+    SetFlags((m_flags & wxSOCKET_BLOCK) | wxSOCKET_WAITALL);
 
 
-  old_flags = m_flags;
-  SetFlags(WAITALL | SPEED);
+    if (DoRead(&msg, sizeof(msg)) != sizeof(msg))
+        goto exit;
 
 
-  Read((char *)&msg, sizeof(msg));
-  if (m_lcount != sizeof(msg))
-  {
-    SetFlags(old_flags);
-    m_error = TRUE;
-    return *this;
-  }
+    sig = (wxUint32)msg.sig[0];
+    sig |= (wxUint32)(msg.sig[1] << 8);
+    sig |= (wxUint32)(msg.sig[2] << 16);
+    sig |= (wxUint32)(msg.sig[3] << 24);
 
 
-  sig = msg.sig[0] & 0xff;
-  sig |= (wxUint32)(msg.sig[1] & 0xff) << 8;
-  sig |= (wxUint32)(msg.sig[2] & 0xff) << 16;
-  sig |= (wxUint32)(msg.sig[3] & 0xff) << 24;
+    if (sig != 0xfeeddead)
+    {
+        wxLogWarning(_("wxSocket: invalid signature in ReadMsg."));
+        goto exit;
+    }
 
 
-  if (sig != 0xfeeddead)
-  {
-    wxLogDebug(_T("Warning: invalid signature returned to ReadMsg\n"));
-    SetFlags(old_flags);
-    m_error = TRUE;
-    return *this;
-  }
-
-  len = msg.len[0] & 0xff;
-  len |= (wxUint32)(msg.len[1] & 0xff) << 8;
-  len |= (wxUint32)(msg.len[2] & 0xff) << 16;
-  len |= (wxUint32)(msg.len[3] & 0xff) << 24;
-
-  if (len > nbytes)
-  {
-    len2 = len - nbytes;
-    len = nbytes;
-  }
-  else
-    len2 = 0;
-
-  // The "len &&" in the following statements is necessary so
-  // that we don't attempt to read (and possibly hang the system)
-  // if the message was zero bytes long
-  if (len && Read(buffer, len).LastCount() != len)
-  {
-    SetFlags(old_flags);
-    m_error = TRUE;
-    return *this;
-  }
-  if (len2)
-  {
-    char *discard_buffer = new char[MAX_BUFSIZE];
-    long discard_len;
+    len = (wxUint32)msg.len[0];
+    len |= (wxUint32)(msg.len[1] << 8);
+    len |= (wxUint32)(msg.len[2] << 16);
+    len |= (wxUint32)(msg.len[3] << 24);
 
 
-    do
+    if (len > nbytes)
     {
     {
-      discard_len = ((len2 > MAX_BUFSIZE)? MAX_BUFSIZE : len2);
-      discard_len = Read(discard_buffer, discard_len).LastCount();
-      len2 -= discard_len;
+        len2 = len - nbytes;
+        len = nbytes;
     }
     }
-    while ((discard_len > 0) && len2);
+    else
+        len2 = 0;
 
 
-    delete [] discard_buffer;
-
-    if (len2 != 0)
+    // Don't attempt to read if the msg was zero bytes long.
+    if (len)
     {
     {
-      SetFlags(old_flags);
-      m_error = TRUE;
-      return *this;
+        total = DoRead(buffer, len);
+
+        if (total != len)
+            goto exit;
     }
     }
-  }
-  if (Read((char *)&msg, sizeof(msg)).LastCount() != sizeof(msg))
-  {
-    SetFlags(old_flags);
-    m_error = TRUE;
-    return *this;
-  }
 
 
-  sig = msg.sig[0] & 0xff;
-  sig |= (wxUint32)(msg.sig[1] & 0xff) << 8;
-  sig |= (wxUint32)(msg.sig[2] & 0xff) << 16;
-  sig |= (wxUint32)(msg.sig[3] & 0xff) << 24;
+    if (len2)
+    {
+        char *discard_buffer = new char[MAX_DISCARD_SIZE];
+        long discard_len;
+
+        // NOTE: discarded bytes don't add to m_lcount.
+        do
+        {
+            discard_len = ((len2 > MAX_DISCARD_SIZE)? MAX_DISCARD_SIZE : len2);
+            discard_len = DoRead(discard_buffer, (wxUint32)discard_len);
+            len2 -= (wxUint32)discard_len;
+        }
+        while ((discard_len > 0) && len2);
+
+        delete [] discard_buffer;
+
+        if (len2 != 0)
+            goto exit;
+    }
+    if (DoRead(&msg, sizeof(msg)) != sizeof(msg))
+        goto exit;
 
 
-  if (sig != 0xdeadfeed)
-  {
-    m_error = TRUE;
-    wxLogDebug(_T("Warning: invalid signature returned to ReadMsg\n"));
-  }
+    sig = (wxUint32)msg.sig[0];
+    sig |= (wxUint32)(msg.sig[1] << 8);
+    sig |= (wxUint32)(msg.sig[2] << 16);
+    sig |= (wxUint32)(msg.sig[3] << 24);
 
 
-  SetFlags(old_flags);
-  return *this;
+    if (sig != 0xdeadfeed)
+    {
+        wxLogWarning(_("wxSocket: invalid signature in ReadMsg."));
+        goto exit;
+    }
 
 
-#undef MAX_BUFSIZE
-}
+    // everything was OK
+    error = false;
 
 
-wxSocketBase& wxSocketBase::Peek(char* buffer, wxUint32 nbytes)
-{
-  Read(buffer, nbytes);
-  CreatePushbackAfter(buffer, nbytes);
+exit:
+    m_error = error;
+    m_lcount = total;
+    m_reading = false;
+    SetFlags(old_flags);
 
 
-  return *this;
+    return *this;
 }
 
 }
 
-int wxSocketBase::DeferWrite(const char *buffer, wxUint32 nbytes)
+wxSocketBase& wxSocketBase::Peek(void* buffer, wxUint32 nbytes)
 {
 {
-  // Timer for timeout
-  _wxSocketInternalTimer timer;
-
-  wxASSERT(m_defering == NO_DEFER);
+    // Mask read events
+    m_reading = true;
 
 
-  m_defering = DEFER_WRITE;
+    m_lcount = DoRead(buffer, nbytes);
+    Pushback(buffer, m_lcount);
 
 
-  // Set the current buffer
-  m_defer_buffer = (char *)buffer;
-  m_defer_nbytes = nbytes;
-  m_defer_timer  = &timer;
+    // If in wxSOCKET_WAITALL mode, all bytes should have been read.
+    if (m_flags & wxSOCKET_WAITALL)
+        m_error = (m_lcount != nbytes);
+    else
+        m_error = (m_lcount == 0);
 
 
-  // Start timer
-  timer.m_state   = (int *)&m_defer_buffer;
-  timer.m_new_val = 0;
+    // Allow read events again
+    m_reading = false;
 
 
-  timer.Start(m_timeout * 1000, FALSE);
+    return *this;
+}
 
 
-  // If the socket is writable, call DoDefer for the first time
-  if (GSocket_Select(m_socket, GSOCK_OUTPUT_FLAG))
-    DoDefer();
+wxSocketBase& wxSocketBase::Write(const void *buffer, wxUint32 nbytes)
+{
+    // Mask write events
+    m_writing = true;
 
 
-  // Wait for buffer completion.
-  while (m_defer_buffer != NULL)
-    wxYield();
+    m_lcount = DoWrite(buffer, nbytes);
 
 
-  // Stop timer
-  m_defer_timer = NULL;
-  timer.Stop();
+    // If in wxSOCKET_WAITALL mode, all bytes should have been written.
+    if (m_flags & wxSOCKET_WAITALL)
+        m_error = (m_lcount != nbytes);
+    else
+        m_error = (m_lcount == 0);
 
 
-  m_defering = NO_DEFER;
+    // Allow write events again
+    m_writing = false;
 
 
-  return nbytes-m_defer_nbytes;
+    return *this;
 }
 
 }
 
-wxSocketBase& wxSocketBase::Write(const char *buffer, wxUint32 nbytes)
+// This function is a mirror image of DoRead() except that it doesn't use the
+// push back buffer, please see comments there
+wxUint32 wxSocketBase::DoWrite(const void *buffer_, wxUint32 nbytes)
 {
 {
-  int ret = 1;
+    const char *buffer = static_cast<const char *>(buffer_);
 
 
-  m_lcount = 0;
-  m_error = FALSE;
+    // Return if there is nothing to read or the socket is (already?) closed.
+    if ( !m_socket || !nbytes )
+        return 0;
 
 
-  if (!m_connected || !m_socket)
-  {
-    m_error = TRUE;
-    return *this;
-  }
+    wxCHECK_MSG( buffer, 0, "NULL buffer" );
 
 
-  if (m_flags & SPEED & WAITALL)    // SPEED && WAITALL
-  {
-    while (ret > 0 && nbytes > 0)
+    wxUint32 total = 0;
+    if ( m_flags & wxSOCKET_NOWAIT )
     {
     {
-      ret = GSocket_Write(m_socket, buffer, nbytes);
-      m_lcount += ret;
-      buffer += ret;
-      nbytes -= ret;
+        wxSocketUnblocker unblock(m_socket);
+        const int ret = m_socket->Write(buffer, nbytes);
+        if ( ret > 0 )
+            total += ret;
+    }
+    else // blocking socket
+    {
+        for ( ;; )
+        {
+            if ( !(m_flags & wxSOCKET_BLOCK) && !WaitForWrite() )
+                break;
+
+            const int ret = m_socket->Write(buffer, nbytes);
+            if ( ret == 0 )
+            {
+                m_closed = true;
+                break;
+            }
+
+            if ( ret < 0 )
+                return 0;
+
+            total += ret;
+            if ( !(m_flags & wxSOCKET_WAITALL) )
+                break;
+
+            nbytes -= ret;
+            if ( !nbytes )
+                break;
+
+            buffer += ret;
+        }
     }
     }
-    // In case the last call was an error ...
-    if (ret < 0)
-      m_lcount ++;
-  }
-  else if (m_flags & SPEED)         // SPEED && !WAITALL
-  {
-    ret = GSocket_Write(m_socket, buffer, nbytes);
-
-    if (ret > 0)
-      m_lcount += ret;
-  }
-  else                              // !SPEED
-  {
-    ret = DeferWrite(buffer, nbytes);
-
-    if (ret > 0)
-      m_lcount += ret;
-  }
-
-  // If we have written some data, then it is not an error,
-  // even when in WAITALL mode, the last low-level IO call
-  // might have failed.
-  if (!m_lcount)
-    m_error = TRUE;
-
-  return *this;
-}
-
-wxSocketBase& wxSocketBase::WriteMsg(const char *buffer, wxUint32 nbytes)
-{
-  int old_flags;
-  struct {
-    char sig[4];
-    char len[4];
-  } msg;
-
-  // warning about 'cast truncates constant value'
-#ifdef __VISUALC__
-    #pragma warning(disable: 4310)
-#endif // __VISUALC__
-
-  msg.sig[0] = (char) 0xad;
-  msg.sig[1] = (char) 0xde;
-  msg.sig[2] = (char) 0xed;
-  msg.sig[3] = (char) 0xfe;
-
-  msg.len[0] = (char) nbytes & 0xff;
-  msg.len[1] = (char) (nbytes >> 8) & 0xff;
-  msg.len[2] = (char) (nbytes >> 16) & 0xff;
-  msg.len[3] = (char) (nbytes >> 24) & 0xff;
-
-  // GRG: We need WAITALL | SPEED
-  old_flags = m_flags;
-  SetFlags(WAITALL | SPEED);
-
-  if (Write((char *)&msg, sizeof(msg)).LastCount() < sizeof(msg))
-  {
-    SetFlags(old_flags);
-    m_error = TRUE;
-    return *this;
-  }
-  if (Write(buffer, nbytes).LastCount() < nbytes)
-  {
-    SetFlags(old_flags);
-    m_error = TRUE;
-    return *this;
-  }
-
-  msg.sig[0] = (char) 0xed;
-  msg.sig[1] = (char) 0xfe;
-  msg.sig[2] = (char) 0xad;
-  msg.sig[3] = (char) 0xde;
-  msg.len[0] = msg.len[1] = msg.len[2] = msg.len[3] = (char) 0;
-
-  if (Write((char *)&msg, sizeof(msg)).LastCount() < sizeof(msg))
-    m_error = TRUE;
-
-  SetFlags(old_flags);
-  return *this;
 
 
-#ifdef __VISUALC__
-    #pragma warning(default: 4310)
-#endif // __VISUALC__
+    return total;
 }
 
 }
 
-wxSocketBase& wxSocketBase::Unread(const char *buffer, wxUint32 nbytes)
+wxSocketBase& wxSocketBase::WriteMsg(const void *buffer, wxUint32 nbytes)
 {
 {
-  m_error = FALSE;
-  m_lcount = 0;
+    wxUint32 total;
+    bool error;
+    struct
+    {
+        unsigned char sig[4];
+        unsigned char len[4];
+    } msg;
 
 
-  if (nbytes != 0)
-  {
-    CreatePushbackAfter(buffer, nbytes);
-    m_lcount = nbytes;
-  }
-  return *this;
-}
+    // Mask write events
+    m_writing = true;
 
 
-bool wxSocketBase::IsData() const
-{
-  if (!m_socket)
-    return FALSE;
+    error = true;
+    total = 0;
+    SetFlags((m_flags & wxSOCKET_BLOCK) | wxSOCKET_WAITALL);
 
 
-  return (GSocket_Select(m_socket, GSOCK_INPUT_FLAG));
-}
+    msg.sig[0] = (unsigned char) 0xad;
+    msg.sig[1] = (unsigned char) 0xde;
+    msg.sig[2] = (unsigned char) 0xed;
+    msg.sig[3] = (unsigned char) 0xfe;
 
 
-// GRG: DoDefer() no longer needs to know which event occured,
-// because this was only used to catch LOST events and set
-// m_defer_buffer = NULL; this is done in OnRequest() now.
+    msg.len[0] = (unsigned char) (nbytes & 0xff);
+    msg.len[1] = (unsigned char) ((nbytes >> 8) & 0xff);
+    msg.len[2] = (unsigned char) ((nbytes >> 16) & 0xff);
+    msg.len[3] = (unsigned char) ((nbytes >> 24) & 0xff);
 
 
-void wxSocketBase::DoDefer()
-{
-  int ret;
-
-  if (!m_defer_buffer)
-    return;
-
-  switch (m_defering)
-  {
-    case DEFER_READ:
-      ret = GSocket_Read(m_socket, m_defer_buffer, m_defer_nbytes);
-      break;
-    case DEFER_WRITE:
-      ret = GSocket_Write(m_socket, m_defer_buffer, m_defer_nbytes);
-      break;
-    default:
-      ret = -1;
-      break;
-  }
-
-  if (ret >= 0)
-    m_defer_nbytes -= ret;
-
-  // If we are waiting for all bytes to be acquired, keep the defering
-  // mode enabled.
-  if ((m_flags & WAITALL) == 0 || m_defer_nbytes == 0 || ret < 0)
-  {
-    m_defer_buffer = NULL;
-  }
-  else
-  {
-    m_defer_buffer += ret;
-    m_defer_timer->Start(m_timeout * 1000, FALSE);
-  }
-}
+    if (DoWrite(&msg, sizeof(msg)) < sizeof(msg))
+        goto exit;
 
 
-void wxSocketBase::Discard()
-{
-#define MAX_BUFSIZE (10*1024)
+    total = DoWrite(buffer, nbytes);
 
 
-  char *my_data = new char[MAX_BUFSIZE];
-  wxUint32 recv_size = MAX_BUFSIZE;
-  wxUint32 total = 0;
+    if (total < nbytes)
+        goto exit;
 
 
-  SaveState();
-  SetFlags(NOWAIT);     // GRG: SPEED was not needed here!
+    msg.sig[0] = (unsigned char) 0xed;
+    msg.sig[1] = (unsigned char) 0xfe;
+    msg.sig[2] = (unsigned char) 0xad;
+    msg.sig[3] = (unsigned char) 0xde;
+    msg.len[0] =
+    msg.len[1] =
+    msg.len[2] =
+    msg.len[3] = (char) 0;
 
 
-  while (recv_size == MAX_BUFSIZE)
-  {
-    recv_size = Read(my_data, MAX_BUFSIZE).LastCount();
-    total += recv_size;
-  }
+    if ((DoWrite(&msg, sizeof(msg))) < sizeof(msg))
+        goto exit;
 
 
-  RestoreState();
-  delete [] my_data;
+    // everything was OK
+    error = false;
 
 
-  m_lcount = total;
-  m_error = FALSE;
+exit:
+    m_error = error;
+    m_lcount = total;
+    m_writing = false;
 
 
-#undef MAX_BUFSIZE
+    return *this;
 }
 
 }
 
-// --------------------------------------------------------------
-// wxSocketBase get local or peer addresses
-// --------------------------------------------------------------
-
-bool wxSocketBase::GetPeer(wxSockAddress& addr_man) const
+wxSocketBase& wxSocketBase::Unread(const void *buffer, wxUint32 nbytes)
 {
 {
-  GAddress *peer;
-
-  if (!m_socket)
-    return FALSE;
+    if (nbytes != 0)
+        Pushback(buffer, nbytes);
 
 
-  peer = GSocket_GetPeer(m_socket);
-  addr_man.SetAddress(peer);
-  GAddress_destroy(peer);
+    m_error = false;
+    m_lcount = nbytes;
 
 
-  return TRUE;
+    return *this;
 }
 
 }
 
-bool wxSocketBase::GetLocal(wxSockAddress& addr_man) const
+wxSocketBase& wxSocketBase::Discard()
 {
 {
-  GAddress *local;
+    char *buffer = new char[MAX_DISCARD_SIZE];
+    wxUint32 ret;
+    wxUint32 total = 0;
 
 
-  if (!m_socket)
-    return FALSE;
+    // Mask read events
+    m_reading = true;
 
 
-  local = GSocket_GetLocal(m_socket);
-  addr_man.SetAddress(local);
-  GAddress_destroy(local);
+    SetFlags(wxSOCKET_NOWAIT);
 
 
-  return TRUE;
-}
+    do
+    {
+        ret = DoRead(buffer, MAX_DISCARD_SIZE);
+        total += ret;
+    }
+    while (ret == MAX_DISCARD_SIZE);
 
 
-// --------------------------------------------------------------
-// wxSocketBase save and restore socket state
-// --------------------------------------------------------------
+    delete[] buffer;
+    m_lcount = total;
+    m_error  = false;
 
 
-void wxSocketBase::SaveState()
-{
-  wxSocketState *state;
+    // Allow read events again
+    m_reading = false;
 
 
-  state = new wxSocketState();
+    return *this;
+}
 
 
-  state->notify_state     = m_notify_state;
-  state->evt_notify_state = m_neededreq;
-  state->socket_flags     = m_flags;
-  state->c_callback       = m_cbk;
-  state->c_callback_data  = m_cdata;
+// --------------------------------------------------------------------------
+// Wait functions
+// --------------------------------------------------------------------------
 
 
-  m_states.Append(state);
-}
+// All Wait functions poll the socket using GSocket_Select() to
+// check for the specified combination of conditions, until one
+// of these conditions become true, an error occurs, or the
+// timeout elapses. The polling loop runs the event loop so that
+// this won't block the GUI.
 
 
-void wxSocketBase::RestoreState()
+bool
+wxSocketBase::DoWait(long seconds, long milliseconds, wxSocketEventFlags flags)
 {
 {
-  wxNode *node;
-  wxSocketState *state;
+    wxCHECK_MSG( m_socket, false, "can't wait on invalid socket" );
 
 
-  node = m_states.Last();
-  if (!node)
-    return;
+    // This can be set to true from Interrupt() to exit this function a.s.a.p.
+    m_interrupt = false;
 
 
-  state = (wxSocketState *)node->Data();
 
 
-  SetFlags(state->socket_flags);
-  m_neededreq = state->evt_notify_state;
-  m_cbk       = state->c_callback;
-  m_cdata     = state->c_callback_data;
-  Notify(state->notify_state);
-
-  delete node;
-  delete state;
-}
+    // Use either the provided timeout or the default timeout value associated
+    // with this socket.
+    //
+    // TODO: allow waiting forever, see #9443
+    const long timeout = seconds == -1 ? m_timeout * 1000
+                                       : seconds * 1000 + milliseconds;
+    const wxMilliClock_t timeEnd = wxGetLocalTimeMillis() + timeout;
 
 
+    // Get the active event loop which we'll use for the message dispatching
+    // when running in the main thread
+    wxEventLoopBase *eventLoop;
+    if ( wxIsMainThread() )
+    {
+        eventLoop = wxEventLoop::GetActive();
 
 
-// --------------------------------------------------------------
-// wxSocketBase Wait functions
-// --------------------------------------------------------------
+#ifdef __WXMSW__
+        wxASSERT_MSG( eventLoop,
+                      "Sockets won't work without a running event loop" );
+#endif // __WXMSW__
+    }
+    else // in worker thread
+    {
+        // We never dispatch messages from threads other than the main one.
+        eventLoop = NULL;
+    }
 
 
-// GRG: I have completely rewritten this family of functions
-// so that they don't depend on event notifications; instead,
-// they poll the socket, using GSocket_Select(), to check for
-// the specified combination of event flags, until an event
-// occurs or until the timeout ellapses. The polling loop
-// calls wxYield(), so this won't block the GUI.
+    // Wait in an active polling loop: notice that the loop is executed at
+    // least once, even if timeout is 0 (i.e. polling).
+    bool gotEvent = false;
+    for ( ;; )
+    {
+        // We always stop waiting when the connection is lost as it doesn't
+        // make sense to continue further, even if GSOCK_LOST_FLAG is not
+        // specified in flags to wait for.
+        const GSocketEventFlags
+            result = m_socket->Select(flags | GSOCK_LOST_FLAG);
+
+        // Incoming connection (server) or connection established (client)?
+        if ( result & GSOCK_CONNECTION_FLAG )
+        {
+            m_connected = true;
+            m_establishing = false;
+            gotEvent = true;
+            break;
+        }
+
+        // Data available or output buffer ready?
+        if ( (result & GSOCK_INPUT_FLAG) || (result & GSOCK_OUTPUT_FLAG) )
+        {
+            gotEvent = true;
+            break;
+        }
+
+        // Connection lost
+        if ( result & GSOCK_LOST_FLAG )
+        {
+            m_connected = false;
+            m_establishing = false;
+            if ( flags & GSOCK_LOST_FLAG )
+                gotEvent = true;
+            break;
+        }
+
+        if ( m_interrupt )
+            break;
+
+        // Wait more?
+        const wxMilliClock_t timeNow = wxGetLocalTimeMillis();
+        if ( timeNow >= timeEnd )
+            break;
+
+        if ( eventLoop )
+        {
+            // Dispatch the events when we run in the main thread and have an
+            // active event loop: without this sockets don't work at all under
+            // MSW as socket flags are only updated when socket messages are
+            // processed.
+            if ( eventLoop->Pending() )
+                eventLoop->Dispatch();
+        }
+#if wxUSE_THREADS
+        else // no event loop or waiting in another thread
+        {
+            // We're busy waiting but at least give up the rest of our current
+            // time slice.
+            wxThread::Yield();
+        }
+#endif // wxUSE_THREADS
+    }
 
 
-bool wxSocketBase::_Wait(long seconds, long milliseconds, wxSocketEventFlags flags)
-{
-  GSocketEventFlags result;
-  _wxSocketInternalTimer timer;
-  long timeout;
-  int state = -1;
-
-  // Check for valid socket
-  if ((!m_connected && !m_establishing) || !m_socket)
-    return FALSE;
-
-  // Check for valid timeout value
-  if (seconds != -1)
-    timeout = seconds * 1000 + milliseconds;
-  else
-    timeout = m_timeout * 1000;
-
-  // Activate timer
-  timer.m_state = &state;
-  timer.m_new_val = 0;
-  timer.Start(timeout, TRUE);
-
-  // Active polling (without using events)
-  result = GSocket_Select(m_socket, flags);
-
-  while ((result == 0) && (state == -1))
-  {
-    wxYield();
-    result = GSocket_Select(m_socket, flags);
-  }
-
-  timer.Stop();
-
-  return (result != 0);
+    return gotEvent;
 }
 
 bool wxSocketBase::Wait(long seconds, long milliseconds)
 {
 }
 
 bool wxSocketBase::Wait(long seconds, long milliseconds)
 {
-  return _Wait(seconds, milliseconds, GSOCK_INPUT_FLAG |
-                                      GSOCK_OUTPUT_FLAG |
-                                      GSOCK_CONNECTION_FLAG |
-                                      GSOCK_LOST_FLAG);
+    return DoWait(seconds, milliseconds,
+            GSOCK_INPUT_FLAG |
+            GSOCK_OUTPUT_FLAG |
+            GSOCK_CONNECTION_FLAG |
+            GSOCK_LOST_FLAG
+        );
 }
 
 bool wxSocketBase::WaitForRead(long seconds, long milliseconds)
 {
 }
 
 bool wxSocketBase::WaitForRead(long seconds, long milliseconds)
 {
-  return _Wait(seconds, milliseconds, GSOCK_INPUT_FLAG | GSOCK_LOST_FLAG);
+    // Check pushback buffer before entering DoWait
+    if ( m_unread )
+        return true;
+
+    // Note that GSOCK_INPUT_LOST has to be explicitly passed to DoWait
+    // because of the semantics of WaitForRead: a return value of true means
+    // that a GSocket_Read call will return immediately, not that there is
+    // actually data to read.
+    return DoWait(seconds, milliseconds, GSOCK_INPUT_FLAG | GSOCK_LOST_FLAG);
 }
 
 }
 
+
 bool wxSocketBase::WaitForWrite(long seconds, long milliseconds)
 {
 bool wxSocketBase::WaitForWrite(long seconds, long milliseconds)
 {
-  return _Wait(seconds, milliseconds, GSOCK_OUTPUT_FLAG | GSOCK_LOST_FLAG);
+    return DoWait(seconds, milliseconds, GSOCK_OUTPUT_FLAG | GSOCK_LOST_FLAG);
 }
 
 bool wxSocketBase::WaitForLost(long seconds, long milliseconds)
 {
 }
 
 bool wxSocketBase::WaitForLost(long seconds, long milliseconds)
 {
-  return _Wait(seconds, milliseconds, GSOCK_LOST_FLAG);
+    return DoWait(seconds, milliseconds, GSOCK_LOST_FLAG);
 }
 
 }
 
-void wxSocketBase::SetTimeout(long seconds)
+// --------------------------------------------------------------------------
+// Miscellaneous
+// --------------------------------------------------------------------------
+
+//
+// Get local or peer address
+//
+
+bool wxSocketBase::GetPeer(wxSockAddress& addr_man) const
 {
 {
-  m_timeout = seconds;
+    GAddress *peer;
 
 
-  if (m_socket)
-    GSocket_SetTimeout(m_socket, m_timeout);
-}
+    if (!m_socket)
+        return false;
 
 
-// --------------------------------------------------------------
-// wxSocketBase flags
-// --------------------------------------------------------------
+    peer = m_socket->GetPeer();
 
 
-void wxSocketBase::SetFlags(wxSockFlags _flags)
-{
-  m_flags = _flags;
+    // copying a null address would just trigger an assert anyway
+
+    if (!peer)
+        return false;
+
+    addr_man.SetAddress(peer);
+    GAddress_destroy(peer);
+
+    return true;
 }
 
 }
 
-wxSocketBase::wxSockFlags wxSocketBase::GetFlags() const
+bool wxSocketBase::GetLocal(wxSockAddress& addr_man) const
 {
 {
-  return m_flags;
+    GAddress *local;
+
+    if (!m_socket)
+        return false;
+
+    local = m_socket->GetLocal();
+    addr_man.SetAddress(local);
+    GAddress_destroy(local);
+
+    return true;
 }
 
 }
 
-// --------------------------------------------------------------
-// wxSocketBase callback management
-// --------------------------------------------------------------
+//
+// Save and restore socket state
+//
 
 
-wxSocketBase::wxSockCbk wxSocketBase::Callback(wxSockCbk cbk_)
+void wxSocketBase::SaveState()
 {
 {
-  wxSockCbk old_cbk = cbk_;
+    wxSocketState *state;
+
+    state = new wxSocketState();
 
 
-  m_cbk = cbk_;
-  return old_cbk;
+    state->m_flags      = m_flags;
+    state->m_notify     = m_notify;
+    state->m_eventmask  = m_eventmask;
+    state->m_clientData = m_clientData;
+
+    m_states.Append(state);
 }
 
 }
 
-char *wxSocketBase::CallbackData(char *data)
+void wxSocketBase::RestoreState()
 {
 {
-  char *old_data = m_cdata;
+    wxList::compatibility_iterator node;
+    wxSocketState *state;
 
 
-  m_cdata = data;
-  return old_data;
+    node = m_states.GetLast();
+    if (!node)
+        return;
+
+    state = (wxSocketState *)node->GetData();
+
+    m_flags      = state->m_flags;
+    m_notify     = state->m_notify;
+    m_eventmask  = state->m_eventmask;
+    m_clientData = state->m_clientData;
+
+    m_states.Erase(node);
+    delete state;
 }
 
 }
 
-// --------------------------------------------------------------
-// wxSocketBase automatic notifier
-// --------------------------------------------------------------
+//
+// Timeout and flags
+//
 
 
-static void LINKAGEMODE wx_socket_callback(GSocket *socket, GSocketEvent event, char *cdata)
+void wxSocketBase::SetTimeout(long seconds)
 {
 {
-  wxSocketBase *sckobj = (wxSocketBase *)cdata;
+    m_timeout = seconds;
 
 
-  sckobj->OnRequest((wxSocketNotify)event);
+    if (m_socket)
+        m_socket->SetTimeout(m_timeout * 1000);
 }
 
 }
 
-wxSocketEventFlags wxSocketBase::EventToNotify(wxSocketNotify evt)
+void wxSocketBase::SetFlags(wxSocketFlags flags)
 {
 {
-  switch (evt)
-  {
-    case GSOCK_INPUT:      return GSOCK_INPUT_FLAG;
-    case GSOCK_OUTPUT:     return GSOCK_OUTPUT_FLAG;
-    case GSOCK_CONNECTION: return GSOCK_CONNECTION_FLAG;
-    case GSOCK_LOST:       return GSOCK_LOST_FLAG;
-  }
-  return 0;
+    // Do some sanity checking on the flags used: not all values can be used
+    // together.
+    wxASSERT_MSG( !(flags & wxSOCKET_NOWAIT) ||
+                  !(flags & (wxSOCKET_WAITALL | wxSOCKET_BLOCK)),
+                  "Using wxSOCKET_WAITALL or wxSOCKET_BLOCK with "
+                  "wxSOCKET_NOWAIT doesn't make sense" );
+
+    m_flags = flags;
 }
 
 }
 
-void wxSocketBase::SetNotify(wxSocketEventFlags flags)
+
+// --------------------------------------------------------------------------
+// Event handling
+// --------------------------------------------------------------------------
+
+// A note on how events are processed, which is probably the most
+// difficult thing to get working right while keeping the same API
+// and functionality for all platforms.
+//
+// When GSocket detects an event, it calls wx_socket_callback, which in
+// turn just calls wxSocketBase::OnRequest in the corresponding wxSocket
+// object. OnRequest does some housekeeping, and if the event is to be
+// propagated to the user, it creates a new wxSocketEvent object and
+// posts it. The event is not processed immediately, but delayed with
+// AddPendingEvent instead. This is necessary in order to decouple the
+// event processing from wx_socket_callback; otherwise, subsequent IO
+// calls made from the user event handler would fail, as gtk callbacks
+// are not reentrant.
+//
+// Note that, unlike events, user callbacks (now deprecated) are _not_
+// decoupled from wx_socket_callback and thus they suffer from a variety
+// of problems. Avoid them where possible and use events instead.
+
+extern "C"
+void LINKAGEMODE wx_socket_callback(GSocket * WXUNUSED(socket),
+                                    GSocketEvent notification,
+                                    char *cdata)
 {
 {
-  m_neededreq = flags;
+    wxSocketBase *sckobj = (wxSocketBase *)cdata;
+
+    sckobj->OnRequest((wxSocketNotify) notification);
 }
 
 }
 
-void wxSocketBase::Notify(bool notify)
+void wxSocketBase::OnRequest(wxSocketNotify notification)
 {
 {
-  m_notify_state = notify;
+    switch(notification)
+    {
+        case wxSOCKET_CONNECTION:
+            m_establishing = false;
+            m_connected = true;
+            break;
+
+            // If we are in the middle of a R/W operation, do not
+            // propagate events to users. Also, filter 'late' events
+            // which are no longer valid.
+
+        case wxSOCKET_INPUT:
+            if (m_reading || !m_socket->Select(GSOCK_INPUT_FLAG))
+                return;
+            break;
+
+        case wxSOCKET_OUTPUT:
+            if (m_writing || !m_socket->Select(GSOCK_OUTPUT_FLAG))
+                return;
+            break;
+
+        case wxSOCKET_LOST:
+            m_connected = false;
+            m_establishing = false;
+            break;
+    }
+
+    // Schedule the event
+
+    wxSocketEventFlags flag = 0;
+    wxUnusedVar(flag);
+    switch (notification)
+    {
+        case GSOCK_INPUT:      flag = GSOCK_INPUT_FLAG; break;
+        case GSOCK_OUTPUT:     flag = GSOCK_OUTPUT_FLAG; break;
+        case GSOCK_CONNECTION: flag = GSOCK_CONNECTION_FLAG; break;
+        case GSOCK_LOST:       flag = GSOCK_LOST_FLAG; break;
+        default:
+                               wxLogWarning(_("wxSocket: unknown event!."));
+                               return;
+    }
+
+    if (((m_eventmask & flag) == flag) && m_notify)
+    {
+        if (m_handler)
+        {
+            wxSocketEvent event(m_id);
+            event.m_event      = notification;
+            event.m_clientData = m_clientData;
+            event.SetEventObject(this);
+
+            m_handler->AddPendingEvent(event);
+        }
+    }
 }
 
 }
 
-void wxSocketBase::OnRequest(wxSocketNotify req_evt)
+void wxSocketBase::Notify(bool notify)
 {
 {
-  wxSocketEvent event(m_id);
-  wxSocketEventFlags flag = EventToNotify(req_evt);
-
-  switch(req_evt)
-  {
-    case wxSOCKET_CONNECTION:
-      m_establishing = FALSE;
-      m_connected = TRUE;
-      break;
-    case wxSOCKET_LOST:
-      m_defer_buffer = NULL;
-      Close();
-      break;
-    case wxSOCKET_INPUT:
-    case wxSOCKET_OUTPUT:
-      if (m_defer_buffer)
-      {
-        // GRG: DoDefer() no longer needs to know which
-        // event occured, because this was only used to
-        // catch LOST events and set m_defer_buffer to
-        // NULL, and this is done in OnRequest() now.
-        DoDefer();
-        // Do not notify to user
-        return;
-      }
-      break;
-  }
-
-  if (((m_neededreq & flag) == flag) && m_notify_state)
-  {
-    event.m_socket = this;
-    event.m_skevt  = req_evt;
-    ProcessEvent(event);
-    OldOnNotify(req_evt);
-
-    if (m_cbk)
-      m_cbk(*this, req_evt, m_cdata);
-  }
+    m_notify = notify;
+    if (m_socket)
+        m_socket->Notify(notify);
 }
 
 }
 
-void wxSocketBase::OldOnNotify(wxSocketNotify evt)
+void wxSocketBase::SetNotify(wxSocketEventFlags flags)
 {
 {
+    m_eventmask = flags;
 }
 
 }
 
-// --------------------------------------------------------------
-// wxSocketBase set event handler
-// --------------------------------------------------------------
-
-void wxSocketBase::SetEventHandler(wxEvtHandler& h_evt, int id)
+void wxSocketBase::SetEventHandler(wxEvtHandler& handler, int id)
 {
 {
-  SetNextHandler(&h_evt);
-  m_id = id;
+    m_handler = &handler;
+    m_id      = id;
 }
 
 }
 
-// --------------------------------------------------------------
-// wxSocketBase pushback library
-// --------------------------------------------------------------
+// --------------------------------------------------------------------------
+// Pushback buffer
+// --------------------------------------------------------------------------
 
 
-void wxSocketBase::CreatePushbackAfter(const char *buffer, wxUint32 size)
+void wxSocketBase::Pushback(const void *buffer, wxUint32 size)
 {
 {
-  char *curr_pos;
+    if (!size) return;
+
+    if (m_unread == NULL)
+        m_unread = malloc(size);
+    else
+    {
+        void *tmp;
+
+        tmp = malloc(m_unrd_size + size);
+        memcpy((char *)tmp + size, m_unread, m_unrd_size);
+        free(m_unread);
 
 
-  if (m_unread != NULL)
-    m_unread = (char *) realloc(m_unread, m_unrd_size+size);
-  else
-    m_unread = (char *) malloc(size);
+        m_unread = tmp;
+    }
 
 
-  curr_pos = m_unread + m_unrd_size;
+    m_unrd_size += size;
 
 
-  memcpy(curr_pos, buffer, size);
-  m_unrd_size += size;
+    memcpy(m_unread, buffer, size);
 }
 
 }
 
-void wxSocketBase::CreatePushbackBefore(const char *buffer, wxUint32 size)
+wxUint32 wxSocketBase::GetPushback(void *buffer, wxUint32 size, bool peek)
 {
 {
-  if (m_unread == NULL)
-    m_unread = (char *)malloc(size);
-  else {
-    char *tmp;
-
-    tmp = (char *)malloc(m_unrd_size + size);
-    memcpy(tmp+size, m_unread, m_unrd_size);
-    free(m_unread);
+    wxCHECK_MSG( buffer, 0, "NULL buffer" );
 
 
-    m_unread = tmp;
-  }
+    if (!m_unrd_size)
+        return 0;
 
 
-  m_unrd_size += size;
+    if (size > (m_unrd_size-m_unrd_cur))
+        size = m_unrd_size-m_unrd_cur;
 
 
-  memcpy(m_unread, buffer, size);
-}
+    memcpy(buffer, (char *)m_unread + m_unrd_cur, size);
 
 
-wxUint32 wxSocketBase::GetPushback(char *buffer, wxUint32 size, bool peek)
-{
-  if (!m_unrd_size)
-    return 0;
-
-  if (size > (m_unrd_size-m_unrd_cur))
-    size = m_unrd_size-m_unrd_cur;
-  memcpy(buffer, (m_unread+m_unrd_cur), size);
-
-  if (!peek) {
-    m_unrd_cur += size;
-    if (m_unrd_size == m_unrd_cur) {
-      free(m_unread);
-      m_unread = NULL;
-      m_unrd_size = 0;
-      m_unrd_cur  = 0;
+    if (!peek)
+    {
+        m_unrd_cur += size;
+        if (m_unrd_size == m_unrd_cur)
+        {
+            free(m_unread);
+            m_unread = NULL;
+            m_unrd_size = 0;
+            m_unrd_cur  = 0;
+        }
     }
     }
-  }
 
 
-  return size;
+    return size;
 }
 
 }
 
-// --------------------------------------------------------------
+
+// ==========================================================================
 // wxSocketServer
 // wxSocketServer
-// --------------------------------------------------------------
+// ==========================================================================
 
 
-// --------------------------------------------------------------
-// wxSocketServer ctor and dtor
-// --------------------------------------------------------------
+// --------------------------------------------------------------------------
+// Ctor
+// --------------------------------------------------------------------------
 
 
-wxSocketServer::wxSocketServer(wxSockAddress& addr_man,
-             wxSockFlags flags) :
-  wxSocketBase(flags, SOCK_SERVER)
+wxSocketServer::wxSocketServer(const wxSockAddress& addr_man,
+                               wxSocketFlags flags)
+              : wxSocketBase(flags, wxSOCKET_SERVER)
 {
 {
-  // Create the socket
-  m_socket = GSocket_new();
-
-  if (!m_socket)
-    return;
-
-  // Setup the socket as server
-  GSocket_SetLocal(m_socket, addr_man.GetAddress());
-  if (GSocket_SetServer(m_socket) != GSOCK_NOERROR)
-  {
-    GSocket_destroy(m_socket);
-    m_socket = NULL;
-    return;
-  }
-
-  GSocket_SetTimeout(m_socket, m_timeout);
-  GSocket_SetCallback(m_socket, GSOCK_INPUT_FLAG | GSOCK_OUTPUT_FLAG |
-                                GSOCK_LOST_FLAG | GSOCK_CONNECTION_FLAG,
-                                wx_socket_callback, (char *)this);
+    wxLogTrace( wxTRACE_Socket, _T("Opening wxSocketServer") );
+
+    m_socket = GSocket_new();
+
+    if (!m_socket)
+    {
+        wxLogTrace( wxTRACE_Socket, _T("*** GSocket_new failed") );
+        return;
+    }
+
+    // Setup the socket as server
+    m_socket->Notify(m_notify);
+    m_socket->SetLocal(addr_man.GetAddress());
+
+    if (GetFlags() & wxSOCKET_REUSEADDR) {
+        m_socket->SetReusable();
+    }
+    if (GetFlags() & wxSOCKET_BROADCAST) {
+        m_socket->SetBroadcast();
+    }
+    if (GetFlags() & wxSOCKET_NOBIND) {
+        m_socket->DontDoBind();
+    }
+
+    if (m_socket->SetServer() != GSOCK_NOERROR)
+    {
+        delete m_socket;
+        m_socket = NULL;
 
 
+        wxLogTrace( wxTRACE_Socket, _T("*** GSocket_SetServer failed") );
+        return;
+    }
+
+    m_socket->SetTimeout(m_timeout * 1000);
+    m_socket->SetCallback(GSOCK_INPUT_FLAG | GSOCK_OUTPUT_FLAG |
+                                  GSOCK_LOST_FLAG | GSOCK_CONNECTION_FLAG,
+                                  wx_socket_callback, (char *)this);
+
+    wxLogTrace( wxTRACE_Socket, _T("wxSocketServer on fd %d"), m_socket->m_fd );
 }
 
 }
 
-// --------------------------------------------------------------
-// wxSocketServer Accept
-// --------------------------------------------------------------
+// --------------------------------------------------------------------------
+// Accept
+// --------------------------------------------------------------------------
 
 bool wxSocketServer::AcceptWith(wxSocketBase& sock, bool wait)
 {
 
 bool wxSocketServer::AcceptWith(wxSocketBase& sock, bool wait)
 {
-  GSocket *child_socket;
+    if (!m_socket)
+        return false;
 
 
-  if (!m_socket)
-    return FALSE;
+    // If wait == false, then the call should be nonblocking.
+    // When we are finished, we put the socket to blocking mode
+    // again.
+    wxSocketUnblocker unblock(m_socket, !wait);
+    GSocket * const child_socket = m_socket->WaitConnection();
 
 
-  // GRG: If wait == FALSE, then the call should be nonblocking.
-  // When we are finished, we put the socket to blocking mode
-  // again.
+    if (!child_socket)
+        return false;
 
 
-  if (!wait)
-    GSocket_SetNonBlocking(m_socket, TRUE);
+    sock.m_type = wxSOCKET_BASE;
+    sock.m_socket = child_socket;
+    sock.m_connected = true;
 
 
-  child_socket = GSocket_WaitConnection(m_socket);
+    sock.m_socket->SetTimeout(sock.m_timeout * 1000);
+    sock.m_socket->SetCallback(GSOCK_INPUT_FLAG | GSOCK_OUTPUT_FLAG |
+            GSOCK_LOST_FLAG | GSOCK_CONNECTION_FLAG,
+            wx_socket_callback, (char *)&sock);
 
 
-  if (!wait)
-    GSocket_SetNonBlocking(m_socket, FALSE);
+    return true;
+}
 
 
-  // GRG: this was not being handled!
-  if (child_socket == NULL)
-    return FALSE;
+wxSocketBase *wxSocketServer::Accept(bool wait)
+{
+    wxSocketBase* sock = new wxSocketBase();
 
 
-  sock.m_type = SOCK_INTERNAL;
-  sock.m_socket = child_socket;
-  sock.m_connected = TRUE;
+    sock->SetFlags(m_flags);
 
 
-  GSocket_SetTimeout(sock.m_socket, sock.m_timeout);
-  GSocket_SetCallback(sock.m_socket, GSOCK_INPUT_FLAG | GSOCK_OUTPUT_FLAG |
-                                     GSOCK_LOST_FLAG | GSOCK_CONNECTION_FLAG,
-                                     wx_socket_callback, (char *)&sock);
+    if (!AcceptWith(*sock, wait))
+    {
+        sock->Destroy();
+        sock = NULL;
+    }
 
 
-  return TRUE;
+    return sock;
 }
 
 }
 
-wxSocketBase *wxSocketServer::Accept(bool wait)
+bool wxSocketServer::WaitForAccept(long seconds, long milliseconds)
 {
 {
-  wxSocketBase* sock = new wxSocketBase();
+    return DoWait(seconds, milliseconds, GSOCK_CONNECTION_FLAG);
+}
 
 
-  sock->SetFlags((wxSockFlags)m_flags);
+bool wxSocketBase::GetOption(int level, int optname, void *optval, int *optlen)
+{
+    wxASSERT_MSG( m_socket, _T("Socket not initialised") );
 
 
-  if (!AcceptWith(*sock, wait))
-    return NULL;
+    if (m_socket->GetSockOpt(level, optname, optval, optlen)
+        != GSOCK_NOERROR)
+    {
+        return false;
+    }
+    return true;
+}
 
 
-  return sock;
+bool wxSocketBase::SetOption(int level, int optname, const void *optval,
+                              int optlen)
+{
+    wxASSERT_MSG( m_socket, _T("Socket not initialised") );
+
+    if (m_socket->SetSockOpt(level, optname, optval, optlen)
+        != GSOCK_NOERROR)
+    {
+        return false;
+    }
+    return true;
 }
 
 }
 
-bool wxSocketServer::WaitForAccept(long seconds, long milliseconds)
+bool wxSocketBase::SetLocal(const wxIPV4address& local)
 {
 {
-  return _Wait(seconds, milliseconds, GSOCK_CONNECTION_FLAG | GSOCK_LOST_FLAG);
+    GAddress* la = local.GetAddress();
+
+    // If the address is valid, save it for use when we call Connect
+    if (la && la->m_addr)
+    {
+        m_localAddress = local;
+
+        return true;
+    }
+
+    return false;
 }
 
 }
 
-// --------------------------------------------------------------
+// ==========================================================================
 // wxSocketClient
 // wxSocketClient
-// --------------------------------------------------------------
+// ==========================================================================
 
 
-// --------------------------------------------------------------
-// wxSocketClient ctor and dtor
-// --------------------------------------------------------------
+// --------------------------------------------------------------------------
+// Ctor and dtor
+// --------------------------------------------------------------------------
 
 
-wxSocketClient::wxSocketClient(wxSockFlags _flags) :
-  wxSocketBase(_flags, SOCK_CLIENT)
+wxSocketClient::wxSocketClient(wxSocketFlags flags)
+              : wxSocketBase(flags, wxSOCKET_CLIENT)
 {
 {
+    m_initialRecvBufferSize =
+    m_initialSendBufferSize = -1;
 }
 
 wxSocketClient::~wxSocketClient()
 {
 }
 
 }
 
 wxSocketClient::~wxSocketClient()
 {
 }
 
-// --------------------------------------------------------------
-// wxSocketClient Connect functions
-// --------------------------------------------------------------
-bool wxSocketClient::Connect(wxSockAddress& addr_man, bool wait)
+// --------------------------------------------------------------------------
+// Connect
+// --------------------------------------------------------------------------
+
+bool wxSocketClient::DoConnect(const wxSockAddress& addr_man,
+                               const wxSockAddress* local,
+                               bool wait)
 {
 {
-  GSocketError err;
+    if (m_socket)
+    {
+        // Shutdown and destroy the socket
+        Close();
+        delete m_socket;
+    }
 
 
-  if (IsConnected())
-    Close();
+    m_socket = GSocket_new();
+    m_connected = false;
+    m_establishing = false;
+
+    if (!m_socket)
+        return false;
+
+    m_socket->SetTimeout(m_timeout * 1000);
+    m_socket->SetCallback(
+        GSOCK_INPUT_FLAG |
+        GSOCK_OUTPUT_FLAG |
+        GSOCK_LOST_FLAG |
+        GSOCK_CONNECTION_FLAG,
+        wx_socket_callback,
+        (char *)this
+    );
+
+    // If wait == false, then the call should be nonblocking. When we are
+    // finished, we put the socket to blocking mode again.
+    wxSocketUnblocker unblock(m_socket, !wait);
+
+    // Reuse makes sense for clients too, if we are trying to rebind to the same port
+    if (GetFlags() & wxSOCKET_REUSEADDR)
+    {
+        m_socket->SetReusable();
+    }
+    if (GetFlags() & wxSOCKET_BROADCAST)
+    {
+        m_socket->SetBroadcast();
+    }
+    if (GetFlags() & wxSOCKET_NOBIND)
+    {
+        m_socket->DontDoBind();
+    }
 
 
-  // This should never happen.
-  if (m_socket)
-    GSocket_destroy(m_socket);
+    // If no local address was passed and one has been set, use the one that was Set
+    if (!local && m_localAddress.GetAddress())
+    {
+        local = &m_localAddress;
+    }
 
 
-  // Initialize all socket stuff ...
-  m_socket = GSocket_new();
-  m_connected = FALSE;
-  m_establishing = FALSE;
+    // Bind to the local IP address and port, when provided
+    if (local)
+    {
+        GAddress* la = local->GetAddress();
 
 
-  if (!m_socket)
-    return FALSE;
+        if (la && la->m_addr)
+            m_socket->SetLocal(la);
+    }
 
 
-  GSocket_SetTimeout(m_socket, m_timeout);
+#if defined(__WXMSW__) || defined(__WXGTK__)
+    m_socket->SetInitialSocketBuffers(m_initialRecvBufferSize, m_initialSendBufferSize);
+#endif
 
 
-  // GRG: If wait == FALSE, then the call should be nonblocking.
-  // When we are finished, we put the socket to blocking mode
-  // again.
+    m_socket->SetPeer(addr_man.GetAddress());
+    const GSocketError err = m_socket->Connect(GSOCK_STREAMED);
 
 
-  if (!wait)
-    GSocket_SetNonBlocking(m_socket, TRUE);
+    //this will register for callbacks - must be called after m_socket->m_fd was initialized
+    m_socket->Notify(m_notify);
 
 
-  GSocket_SetPeer(m_socket, addr_man.GetAddress());
-  err = GSocket_Connect(m_socket, GSOCK_STREAMED);
-  GSocket_SetCallback(m_socket, GSOCK_INPUT_FLAG | GSOCK_OUTPUT_FLAG |
-                                GSOCK_LOST_FLAG | GSOCK_CONNECTION_FLAG,
-                                wx_socket_callback, (char *)this);
+    if (err != GSOCK_NOERROR)
+    {
+        if (err == GSOCK_WOULDBLOCK)
+            m_establishing = true;
 
 
-  if (!wait)
-    GSocket_SetNonBlocking(m_socket, FALSE);
+        return false;
+    }
 
 
-  if (err != GSOCK_NOERROR)
-  {
-    if (err == GSOCK_WOULDBLOCK)
-      m_establishing = TRUE;
+    m_connected = true;
+    return true;
+}
 
 
-    return FALSE;
-  }
+bool wxSocketClient::Connect(const wxSockAddress& addr_man, bool wait)
+{
+    return DoConnect(addr_man, NULL, wait);
+}
 
 
-  m_connected = TRUE;
-  return TRUE;
+bool wxSocketClient::Connect(const wxSockAddress& addr_man,
+                             const wxSockAddress& local,
+                             bool wait)
+{
+    return DoConnect(addr_man, &local, wait);
 }
 
 bool wxSocketClient::WaitOnConnect(long seconds, long milliseconds)
 {
 }
 
 bool wxSocketClient::WaitOnConnect(long seconds, long milliseconds)
 {
-  bool ret;
-
-  if (m_connected)      // Already connected
-    return TRUE;
-
-  if (!m_establishing || !m_socket)  // No connection in progress
-    return FALSE;
-
-  ret = _Wait(seconds, milliseconds, GSOCK_CONNECTION_FLAG | GSOCK_LOST_FLAG);
-
-  // GRG: m_connected and m_establishing will be updated in
-  // OnRequest(), but we do it here anyway because sometimes
-  // the event might be a bit delayed, and if that happens,
-  // when WaitOnConnect() returns, m_connected will still be
-  // FALSE. We have to do it as well in OnRequest because
-  // maybe WaitOnConnect() is not being used...
+    if ( m_connected )
+    {
+        // this happens if the initial attempt to connect succeeded without
+        // blocking
+        return true;
+    }
 
 
-  if (ret)
-  {
-    m_connected = GSocket_Select(m_socket, GSOCK_CONNECTION_FLAG);
-    m_establishing = FALSE;
-  }
+    wxCHECK_MSG( m_establishing && m_socket, false,
+                 "No connection establishment attempt in progress" );
 
 
-  return m_connected;
+    // we must specify GSOCK_LOST_FLAG here explicitly because we must return
+    // true if the connection establishment process is finished, whether it is
+    // over because we successfully connected or because we were not able to
+    // connect
+    return DoWait(seconds, milliseconds,
+        GSOCK_CONNECTION_FLAG | GSOCK_LOST_FLAG);
 }
 
 }
 
+// ==========================================================================
+// wxDatagramSocket
+// ==========================================================================
 
 
-// --------------------------------------------------------------
-// wxSocketEvent
-// --------------------------------------------------------------
+/* NOTE: experimental stuff - might change */
 
 
-wxSocketEvent::wxSocketEvent(int id)
-  : wxEvent(id)
+wxDatagramSocket::wxDatagramSocket( const wxSockAddress& addr,
+                                    wxSocketFlags flags )
+                : wxSocketBase( flags, wxSOCKET_DATAGRAM )
 {
 {
-  wxEventType type = (wxEventType)wxEVT_SOCKET;
+    // Create the socket
+    m_socket = GSocket_new();
+
+    if (!m_socket)
+    {
+        wxFAIL_MSG( _T("datagram socket not new'd") );
+        return;
+    }
+    m_socket->Notify(m_notify);
+    // Setup the socket as non connection oriented
+    m_socket->SetLocal(addr.GetAddress());
+    if (flags & wxSOCKET_REUSEADDR)
+    {
+        m_socket->SetReusable();
+    }
+    if (GetFlags() & wxSOCKET_BROADCAST)
+    {
+        m_socket->SetBroadcast();
+    }
+    if (GetFlags() & wxSOCKET_NOBIND)
+    {
+        m_socket->DontDoBind();
+    }
+    if ( m_socket->SetNonOriented() != GSOCK_NOERROR )
+    {
+        delete m_socket;
+        m_socket = NULL;
+        return;
+    }
 
 
-  SetEventType(type);
+    // Initialize all stuff
+    m_connected = false;
+    m_establishing = false;
+    m_socket->SetTimeout( m_timeout );
+    m_socket->SetCallback( GSOCK_INPUT_FLAG | GSOCK_OUTPUT_FLAG |
+                           GSOCK_LOST_FLAG | GSOCK_CONNECTION_FLAG,
+                           wx_socket_callback, (char*)this );
 }
 
 }
 
-void wxSocketEvent::CopyObject(wxObject& obj_d) const
+wxDatagramSocket& wxDatagramSocket::RecvFrom( wxSockAddress& addr,
+                                              void* buf,
+                                              wxUint32 nBytes )
 {
 {
-  wxSocketEvent *event = (wxSocketEvent *)&obj_d;
+    Read(buf, nBytes);
+    GetPeer(addr);
+    return (*this);
+}
 
 
-  wxEvent::CopyObject(obj_d);
+wxDatagramSocket& wxDatagramSocket::SendTo( const wxSockAddress& addr,
+                                            const void* buf,
+                                            wxUint32 nBytes )
+{
+    wxASSERT_MSG( m_socket, _T("Socket not initialised") );
 
 
-  event->m_skevt = m_skevt;
-  event->m_socket = m_socket;
+    m_socket->SetPeer(addr.GetAddress());
+    Write(buf, nBytes);
+    return (*this);
 }
 
 }
 
-// --------------------------------------------------------------------------
+// ==========================================================================
 // wxSocketModule
 // wxSocketModule
-// --------------------------------------------------------------------------
-class WXDLLEXPORT wxSocketModule: public wxModule {
-  DECLARE_DYNAMIC_CLASS(wxSocketModule)
- public:
-  bool OnInit() {
-    return GSocket_Init();
-  }
-  void OnExit() {
-    GSocket_Cleanup();
-  }
+// ==========================================================================
+
+class wxSocketModule : public wxModule
+{
+public:
+    virtual bool OnInit()
+    {
+        // wxSocketBase will call GSocket_Init() itself when/if needed
+        return true;
+    }
+
+    virtual void OnExit()
+    {
+        if ( wxSocketBase::IsInitialized() )
+            wxSocketBase::Shutdown();
+    }
+
+private:
+    DECLARE_DYNAMIC_CLASS(wxSocketModule)
 };
 
 IMPLEMENT_DYNAMIC_CLASS(wxSocketModule, wxModule)
 
 };
 
 IMPLEMENT_DYNAMIC_CLASS(wxSocketModule, wxModule)
 
-#endif
-  // wxUSE_SOCKETS
+#endif // wxUSE_SOCKETS
wxWidgets (Impactor)