Add non-template wxWeakRef<T> constructor from T*.

[wxWidgets.git] / include / wx / unix / private / sockunix.h

diff --git a/include/wx/unix/private/sockunix.h b/include/wx/unix/private/sockunix.h
index 472d524a565e78b5b26b41c011d6a55f7d052bf4..63fc0d885dd372db74a8798124fe8a42e78603f5 100644 (file)
--- a/include/wx/unix/private/sockunix.h
+++ b/include/wx/unix/private/sockunix.h
@@ -14,8 +14,10 @@
 
 #include <unistd.h>
 #include <sys/ioctl.h>
 
 #include <unistd.h>
 #include <sys/ioctl.h>

+#include "wx/private/fdiomanager.h"

 
 

-class wxSocketImplUnix : public wxSocketImpl
+class wxSocketImplUnix : public wxSocketImpl,
+                         public wxFDIOHandler

 {
 public:
     wxSocketImplUnix(wxSocketBase& wxsocket)
 {
 public:
     wxSocketImplUnix(wxSocketBase& wxsocket)


@@ -23,30 +25,37 @@ public:
     {
         m_fds[0] =
         m_fds[1] = -1;
     {
         m_fds[0] =
         m_fds[1] = -1;

-
-        m_use_events = false;

     }
 
     }
 

-    virtual void Shutdown();
-    virtual wxSocketImpl *WaitConnection(wxSocketBase& wxsocket);
+    virtual wxSocketError GetLastError() const;

 
 

-    int Read(char *buffer, int size);
-    int Write(const char *buffer, int size);
-    //attach or detach from main loop
-    void Notify(bool flag);
-    void Detected_Read();
-    void Detected_Write();
+    virtual void ReenableEvents(wxSocketEventFlags flags)
+    {
+        // enable the notifications about input/output being available again in
+        // case they were disabled by OnRead/WriteWaiting()
+        //
+        // notice that we'd like to enable the events here only if there is
+        // nothing more left on the socket right now as otherwise we're going
+        // to get a "ready for whatever" notification immediately (well, during
+        // the next event loop iteration) and disable the event back again
+        // which is rather inefficient but unfortunately doing it like this
+        // doesn't work because the existing code (e.g. src/common/sckipc.cpp)
+        // expects to keep getting notifications about the data available from
+        // the socket even if it didn't read all the data the last time, so we
+        // absolutely have to continue generating them
+        EnableEvents(flags);
+    }
+
+    // wxFDIOHandler methods
+    virtual void OnReadWaiting();
+    virtual void OnWriteWaiting();
+    virtual void OnExceptionWaiting();
+    virtual bool IsOk() const { return m_fd != INVALID_SOCKET; }

 
 private:
 
 private:

-    virtual wxSocketError DoHandleConnect(int ret);

     virtual void DoClose()
     {
     virtual void DoClose()
     {

-        wxSocketManager * const manager = wxSocketManager::Get();
-        if ( manager )
-        {
-            manager->Uninstall_Callback(this, wxSOCKET_INPUT);
-            manager->Uninstall_Callback(this, wxSOCKET_OUTPUT);
-        }
+        DisableEvents();

 
         close(m_fd);
     }
 
         close(m_fd);
     }


@@ -59,45 +68,16 @@ private:
         EnableEvents();
     }
 
         EnableEvents();
     }
 

-    // enable or disable notifications for socket input/output events but only
-    // if m_use_events is true; do nothing otherwise
-    virtual void EnableEvents()
-    {
-        if ( m_use_events )
-            DoEnableEvents(true);
-    }
-
-    void DisableEvents()
-    {
-        if ( m_use_events )
-            DoEnableEvents(false);
-    }
-
-    // really enable or disable socket input/output events, regardless of
-    // m_use_events value
-    void DoEnableEvents(bool enable);
+    // enable or disable notifications for socket input/output events
+    void EnableEvents(int flags = wxSOCKET_INPUT_FLAG | wxSOCKET_OUTPUT_FLAG)
+        { DoEnableEvents(flags, true); }
+    void DisableEvents(int flags = wxSOCKET_INPUT_FLAG | wxSOCKET_OUTPUT_FLAG)
+        { DoEnableEvents(flags, false); }

 
 

-
-    // enable or disable events for the given event if m_use_events; do nothing
-    // otherwise
-    //
-    // notice that these functions also update m_detected: EnableEvent() clears
-    // the corresponding bit in it and DisableEvent() sets it
-    void EnableEvent(wxSocketNotify event);
-    void DisableEvent(wxSocketNotify event);
-
-
-    wxSocketError Input_Timeout();
-    wxSocketError Output_Timeout();
-    int Recv_Stream(char *buffer, int size);
-    int Recv_Dgram(char *buffer, int size);
-    int Send_Stream(const char *buffer, int size);
-    int Send_Dgram(const char *buffer, int size);
+    // really enable or disable socket input/output events
+    void DoEnableEvents(int flags, bool enable);

 
 protected:
 
 protected:

-    // true if socket should fire events
-    bool m_use_events;
-

     // descriptors for input and output event notification channels associated
     // with the socket
     int m_fds[2];
     // descriptors for input and output event notification channels associated
     // with the socket
     int m_fds[2];


@@ -107,101 +87,51 @@ private:
     // down the socket if the event is wxSOCKET_LOST
     void OnStateChange(wxSocketNotify event);
 
     // down the socket if the event is wxSOCKET_LOST
     void OnStateChange(wxSocketNotify event);
 

+    // check if there is any input available, return 1 if yes, 0 if no or -1 on
+    // error
+    int CheckForInput();
+
+

     // give it access to our m_fds
     friend class wxSocketFDBasedManager;
 };
 
     // give it access to our m_fds
     friend class wxSocketFDBasedManager;
 };
 

-// A version of wxSocketManager which uses FDs for socket IO
+// A version of wxSocketManager which uses FDs for socket IO: it is used by
+// Unix console applications and some X11-like ports (wxGTK and wxMotif but not
+// wxX11 currently) which implement their own port-specific wxFDIOManagers

 class wxSocketFDBasedManager : public wxSocketManager
 {
 public:
 class wxSocketFDBasedManager : public wxSocketManager
 {
 public:

-    // no special initialization/cleanup needed when using FDs
-    virtual bool OnInit() { return true; }
+    wxSocketFDBasedManager()
+    {
+        m_fdioManager = NULL;
+    }
+
+    virtual bool OnInit();

     virtual void OnExit() { }
 
     virtual void OnExit() { }
 

-    // allocate/free the storage we need

     virtual wxSocketImpl *CreateSocket(wxSocketBase& wxsocket)
     {
         return new wxSocketImplUnix(wxsocket);
     }
 
     virtual wxSocketImpl *CreateSocket(wxSocketBase& wxsocket)
     {
         return new wxSocketImplUnix(wxsocket);
     }
 

-protected:
-    // identifies either input or output direction
-    //
-    // NB: the values of this enum shouldn't change
-    enum SocketDir
-    {
-        FD_INPUT,
-        FD_OUTPUT
-    };
+    virtual void Install_Callback(wxSocketImpl *socket_, wxSocketNotify event);
+    virtual void Uninstall_Callback(wxSocketImpl *socket_, wxSocketNotify event);

 
 

+protected:

     // get the FD index corresponding to the given wxSocketNotify
     // get the FD index corresponding to the given wxSocketNotify

-    SocketDir GetDirForEvent(wxSocketImpl *socket, wxSocketNotify event)
-    {
-        switch ( event )
-        {
-            default:
-                wxFAIL_MSG( "unexpected socket event" );
-                // fall through
-
-            case wxSOCKET_LOST:
-                // fall through
-
-            case wxSOCKET_INPUT:
-                return FD_INPUT;
-
-            case wxSOCKET_OUTPUT:
-                return FD_OUTPUT;
-
-            case wxSOCKET_CONNECTION:
-                // FIXME: explain this?
-                return socket->m_server ? FD_INPUT : FD_OUTPUT;
-        }
-    }
+    wxFDIOManager::Direction
+    GetDirForEvent(wxSocketImpl *socket, wxSocketNotify event);

 
     // access the FDs we store
 
     // access the FDs we store

-    int& FD(wxSocketImpl *socket, SocketDir d)
-    {
-        return static_cast<wxSocketImplUnix *>(socket)->m_fds[d];
-    }
-};
-
-// Common base class for all ports using X11-like (and hence implemented in
-// X11, Motif and GTK) AddInput() and RemoveInput() functions
-class wxSocketInputBasedManager : public wxSocketFDBasedManager
-{
-public:
-    virtual void Install_Callback(wxSocketImpl *socket, wxSocketNotify event)
+    int& FD(wxSocketImplUnix *socket, wxFDIOManager::Direction d)

     {
     {

-        wxCHECK_RET( socket->m_fd != -1,
-                        "shouldn't be called on invalid socket" );
-
-        const SocketDir d = GetDirForEvent(socket, event);
-
-        int& fd = FD(socket, d);
-        if ( fd != -1 )
-            RemoveInput(fd);
-
-        fd = AddInput(socket, d);
+        return socket->m_fds[d];

     }
 
     }
 

-    virtual void Uninstall_Callback(wxSocketImpl *socket, wxSocketNotify event)
-    {
-        const SocketDir d = GetDirForEvent(socket, event);
-
-        int& fd = FD(socket, d);
-        if ( fd != -1 )
-        {
-            RemoveInput(fd);
-            fd = -1;
-        }
-    }
+    wxFDIOManager *m_fdioManager;

 
 

-private:
-    // these functions map directly to XtAdd/RemoveInput() or
-    // gdk_input_add/remove()
-    virtual int AddInput(wxSocketImpl *socket, SocketDir d) = 0;
-    virtual void RemoveInput(int fd) = 0;
+    wxDECLARE_NO_COPY_CLASS(wxSocketFDBasedManager);

 };
 
 #endif  /* _WX_UNIX_GSOCKUNX_H_ */
 };
 
 #endif  /* _WX_UNIX_GSOCKUNX_H_ */