X-Git-Url: https://git.saurik.com/wxWidgets.git/blobdiff_plain/985acf8734d61067c66b4812a9dd9da90d725851..80cc5fc7ad102221a1ce9e160eac3c29a0586664:/include/wx/unix/gsockunx.h diff --git a/include/wx/unix/gsockunx.h b/include/wx/unix/gsockunx.h index a01f6013d2..9fa9e77fbf 100644 --- a/include/wx/unix/gsockunx.h +++ b/include/wx/unix/gsockunx.h @@ -8,51 +8,23 @@ * ------------------------------------------------------------------------- */ -#ifndef __GSOCK_UNX_H -#define __GSOCK_UNX_H +#ifndef _WX_UNIX_GSOCKUNX_H_ +#define _WX_UNIX_GSOCKUNX_H_ -#ifndef __GSOCKET_STANDALONE__ -#include "wx/setup.h" -#endif +#include class wxGSocketIOHandler; -#if wxUSE_SOCKETS || defined(__GSOCKET_STANDALONE__) - -#ifndef __GSOCKET_STANDALONE__ -#include "wx/gsocket.h" -#else -#include "gsocket.h" -#endif - -class GSocketGUIFunctionsTableConcrete : public GSocketGUIFunctionsTable -{ -public: - virtual bool OnInit(); - virtual void OnExit(); - virtual bool CanUseEventLoop(); - virtual bool Init_Socket(GSocket *socket); - virtual void Destroy_Socket(GSocket *socket); - virtual void Install_Callback(GSocket *socket, GSocketEvent event); - virtual void Uninstall_Callback(GSocket *socket, GSocketEvent event); - virtual void Enable_Events(GSocket *socket); - virtual void Disable_Events(GSocket *socket); -}; - -class GSocket +class GSocket : public GSocketBase { public: - GSocket(); + GSocket(wxSocketBase& wxsocket); virtual ~GSocket(); - bool IsOk() { return m_ok; } - void Close(); - void Shutdown(); - GSocketError SetLocal(GAddress *address); - GSocketError SetPeer(GAddress *address); - GAddress *GetLocal(); - GAddress *GetPeer(); + + virtual void Shutdown(); + virtual GSocket *WaitConnection(wxSocketBase& wxsocket); + GSocketError SetServer(); - GSocket *WaitConnection(); bool SetReusable(); bool SetBroadcast(); bool DontDoBind(); @@ -60,93 +32,181 @@ public: GSocketError SetNonOriented(); int Read(char *buffer, int size); int Write(const char *buffer, int size); - GSocketEventFlags Select(GSocketEventFlags flags); void SetNonBlocking(bool non_block); - void SetTimeout(unsigned long millisec); GSocketError WXDLLIMPEXP_NET GetError(); - void SetCallback(GSocketEventFlags flags, - GSocketCallback callback, char *cdata); - void UnsetCallback(GSocketEventFlags flags); GSocketError GetSockOpt(int level, int optname, void *optval, int *optlen); GSocketError SetSockOpt(int level, int optname, const void *optval, int optlen); - virtual void Detected_Read(); - virtual void Detected_Write(); -protected: - void Enable(GSocketEvent event); - void Disable(GSocketEvent event); + //attach or detach from main loop + void Notify(bool flag); + void Detected_Read(); + void Detected_Write(); + +private: + // enable or disable notifications for socket input/output events but only + // if m_use_events is true; do nothing otherwise + 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 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(GSocketEvent event); + void DisableEvent(GSocketEvent event); + + GSocketError Input_Timeout(); GSocketError 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); - bool m_ok; public: /* DFE: We can't protect these data member until the GUI code is updated */ /* protected: */ - int m_fd; wxGSocketIOHandler *m_handler; - GAddress *m_local; - GAddress *m_peer; - GSocketError m_error; - - bool m_non_blocking; - bool m_server; - bool m_stream; - bool m_establishing; - bool m_reusable; - bool m_broadcast; - bool m_dobind; - unsigned long m_timeout; - - /* Callbacks */ - GSocketEventFlags m_detected; - GSocketCallback m_cbacks[GSOCK_MAX_EVENT]; - char *m_data[GSOCK_MAX_EVENT]; - - char *m_gui_dependent; -}; - -#ifdef __cplusplus -extern "C" { -#endif /* __cplusplus */ -/* Definition of GAddress */ -struct _GAddress -{ - struct sockaddr *m_addr; - size_t m_len; + // true if socket should fire events + bool m_use_events; - GAddressType m_family; - int m_realfamily; + // pointer for storing extra (usually GUI-specific) data + void *m_gui_dependent; - GSocketError m_error; +private: + // notify the associated wxSocket about a change in socket state and shut + // down the socket if the event is GSOCK_LOST + void OnStateChange(GSocketEvent event); }; -#ifdef __cplusplus -} -#endif /* __cplusplus */ +// A version of GSocketManager which uses FDs for socket IO +// +// This class uses GSocket::m_gui_dependent field to store the 2 (for input and +// output) FDs associated with the socket. +class GSocketFDBasedManager : public GSocketManager +{ +public: + // no special initialization/cleanup needed when using FDs + virtual bool OnInit() { return true; } + virtual void OnExit() { } -#ifdef __cplusplus -extern "C" { -#endif /* __cplusplus */ + // allocate/free the storage we need + virtual bool Init_Socket(GSocket *socket) + { + socket->m_gui_dependent = malloc(sizeof(int)*2); + int * const fds = static_cast(socket->m_gui_dependent); + fds[0] = -1; + fds[1] = -1; -/* GAddress */ + return true; + } -GSocketError _GAddress_translate_from(GAddress *address, - struct sockaddr *addr, int len); -GSocketError _GAddress_translate_to (GAddress *address, - struct sockaddr **addr, int *len); -GSocketError _GAddress_Init_INET(GAddress *address); -GSocketError _GAddress_Init_UNIX(GAddress *address); + virtual void Close_Socket(GSocket *socket) + { + Uninstall_Callback(socket, GSOCK_INPUT); + Uninstall_Callback(socket, GSOCK_OUTPUT); + close(socket->m_fd); + } -#ifdef __cplusplus -} -#endif /* __cplusplus */ + virtual void Destroy_Socket(GSocket *socket) + { + free(socket->m_gui_dependent); + } -#endif /* wxUSE_SOCKETS || defined(__GSOCKET_STANDALONE__) */ +protected: + // identifies either input or output direction + // + // NB: the values of this enum shouldn't change + enum SocketDir + { + FD_INPUT, + FD_OUTPUT + }; + + // get the FD index corresponding to the given GSocketEvent + SocketDir GetDirForEvent(GSocket *socket, GSocketEvent event) + { + switch ( event ) + { + default: + wxFAIL_MSG( "unexpected socket event" ); + // fall through + + case GSOCK_LOST: + // fall through + + case GSOCK_INPUT: + return FD_INPUT; + + case GSOCK_OUTPUT: + return FD_OUTPUT; + + case GSOCK_CONNECTION: + // FIXME: explain this? + return socket->m_server ? FD_INPUT : FD_OUTPUT; + } + } + + // access the FDs we store + int& FD(GSocket *socket, SocketDir d) + { + return static_cast(socket->m_gui_dependent)[d]; + } +}; + +// Common base class for all ports using X11-like (and hence implemented in +// X11, Motif and GTK) AddInput() and RemoveInput() functions +class GSocketInputBasedManager : public GSocketFDBasedManager +{ +public: + virtual void Install_Callback(GSocket *socket, GSocketEvent event) + { + 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); + } + + virtual void Uninstall_Callback(GSocket *socket, GSocketEvent event) + { + const SocketDir d = GetDirForEvent(socket, event); + + int& fd = FD(socket, d); + if ( fd != -1 ) + { + RemoveInput(fd); + fd = -1; + } + } + +private: + // these functions map directly to XtAdd/RemoveInput() or + // gdk_input_add/remove() + virtual int AddInput(GSocket *socket, SocketDir d) = 0; + virtual void RemoveInput(int fd) = 0; +}; -#endif /* __GSOCK_UNX_H */ +#endif /* _WX_UNIX_GSOCKUNX_H_ */