/////////////////////////////////////////////////////////////////////////////
// Name: socket.h
// Purpose: Socket handling classes
-// Author: Guilhem Lavaux
+// Authors: Guilhem Lavaux, Guillermo Rodriguez Garcia
// Modified by:
// Created: April 1997
// RCS-ID: $Id$
#if wxUSE_SOCKETS
-// ---------------------------------------------------------------------------
-// Windows(tm) specific
-// ---------------------------------------------------------------------------
-#if defined(__WINDOWS__) && defined(WXSOCK_INTERNAL)
-#include <winsock.h>
-#include <wx/msw/private.h>
-#endif // defined(__WINDOWS__) && defined(WXSOCK_INTERNAL)
-
-// ---------------------------------------------------------------------------
-// Unix specific
-// ---------------------------------------------------------------------------
-#if defined(__UNIX__) && defined(WXSOCK_INTERNAL)
-#include <sys/types.h>
-#include <sys/socket.h>
-#include <netinet/in.h>
-#endif // defined(__UNIX__) && defined(WXSOCK_INTERNAL)
-
// ---------------------------------------------------------------------------
// wxSocket headers (generic)
// ---------------------------------------------------------------------------
#ifdef WXPREC
-#include <wx/wxprec.h>
+# include "wx/wxprec.h"
#else
-#include <wx/wx.h>
+# include "wx/event.h"
+# include "wx/string.h"
#endif
+
#include "wx/sckaddr.h"
+#include "wx/gsocket.h"
+
+// ------------------------------------------------------------------------
+// GSocket type alias
+// ------------------------------------------------------------------------
+
+typedef enum {
+ wxSOCKET_INPUT = GSOCK_INPUT,
+ wxSOCKET_OUTPUT = GSOCK_OUTPUT,
+ wxSOCKET_CONNECTION = GSOCK_CONNECTION,
+ wxSOCKET_LOST = GSOCK_LOST
+} wxSocketNotify;
+
+enum {
+ wxSOCKET_INPUT_FLAG = GSOCK_INPUT_FLAG,
+ wxSOCKET_OUTPUT_FLAG = GSOCK_OUTPUT_FLAG,
+ wxSOCKET_CONNECTION_FLAG = GSOCK_CONNECTION_FLAG,
+ wxSOCKET_LOST_FLAG = GSOCK_LOST_FLAG
+};
+typedef GSocketEventFlags wxSocketEventFlags;
+
+typedef enum {
+ wxSOCKET_NOERROR = GSOCK_NOERROR,
+ wxSOCKET_INPOP = GSOCK_INVOP,
+ wxSOCKET_IOERR = GSOCK_IOERR,
+ wxSOCKET_INVADDR = GSOCK_INVADDR,
+ wxSOCKET_INVSOCK = GSOCK_INVSOCK,
+ wxSOCKET_NOHOST = GSOCK_NOHOST,
+ wxSOCKET_INVPORT = GSOCK_INVPORT,
+ wxSOCKET_WOULDBLOCK = GSOCK_WOULDBLOCK,
+ wxSOCKET_TIMEDOUT = GSOCK_TIMEDOUT,
+ wxSOCKET_MEMERR = GSOCK_MEMERR
+} wxSocketError;
+
+// ------------------------------------------------------------------------
+// wxSocket base
+// ------------------------------------------------------------------------
+
+class WXDLLEXPORT wxTimer;
class WXDLLEXPORT wxSocketEvent;
-class WXDLLEXPORT wxSocketHandler;
-class wxSocketInternal;
class WXDLLEXPORT wxSocketBase : public wxEvtHandler
{
DECLARE_CLASS(wxSocketBase)
-#ifdef __WXMAC__
- friend void wxMacSocketOnRequestProc(void *refcon , short event) ;
-#endif
public:
- enum wxSockFlags { NONE=0, NOWAIT=1, WAITALL=2, SPEED=4 };
+ enum { NONE=0, NOWAIT=1, WAITALL=2, SPEED=4 };
+ typedef int wxSockFlags;
// Type of request
- enum { REQ_READ=0x1, REQ_PEEK=0x2, REQ_WRITE=0x4, REQ_LOST=0x8,
- REQ_ACCEPT=0x10, REQ_CONNECT=0x20, REQ_WAIT=0x40};
- enum { EVT_READ=0, EVT_PEEK=1, EVT_WRITE=2, EVT_LOST=3, EVT_ACCEPT=4,
- EVT_CONNECT=5 };
- typedef int wxRequestNotify;
- typedef int wxRequestEvent;
- enum wxSockType { SOCK_CLIENT, SOCK_SERVER, SOCK_INTERNAL, SOCK_UNINIT };
- typedef void (*wxSockCbk)(wxSocketBase& sock,wxRequestEvent evt,char *cdata);
+ enum wxSockType { SOCK_CLIENT, SOCK_SERVER, SOCK_INTERNAL, SOCK_UNINIT };
+ typedef void (*wxSockCbk)(wxSocketBase& sock, wxSocketNotify evt, char *cdata);
protected:
- wxSockFlags m_flags;
+ GSocket *m_socket; // wxSocket socket
+ wxSockFlags m_flags; // wxSocket flags
wxSockType m_type; // wxSocket type
- bool m_connected, m_connecting; // State of the socket
- int m_fd; // Socket file descriptors
- wxList m_states; // States list
- int m_id; // Socket id (for event handler)
- wxSocketHandler *m_handler; // the current socket handler
- wxRequestNotify m_neededreq; // Specify which requet signals we need
- unsigned long m_timeout;
- size_t m_lcount; // Last IO request size
- int m_error; // Last IO error
- wxSocketInternal *m_internal;
- char *m_unread; // Pushback buffer
- size_t m_unrd_size; // Pushback buffer size
- wxSockCbk m_cbk;
- char *m_cdata;
- bool m_notify_state;
-
+ wxSocketEventFlags m_neededreq; // Which events we are interested in
+ wxUint32 m_lcount; // Last IO request size
+ unsigned long m_timeout; // IO timeout value
+
+ char *m_unread; // Pushback buffer
+ wxUint32 m_unrd_size; // Pushback buffer size
+ wxUint32 m_unrd_cur; // Pushback pointer
+
+ wxSockCbk m_cbk; // C callback
+ char *m_cdata; // C callback data
+
+ bool m_connected; // Connected ?
+ bool m_establishing; // Pending connections ?
+ bool m_notify_state; // Notify state
+ int m_id; // Socket id (for event handler)
+
+ // Defering variables
+ enum {
+ DEFER_READ, DEFER_WRITE, NO_DEFER
+ } m_defering; // Defering state
+ char *m_defer_buffer; // Defering target buffer
+ wxUint32 m_defer_nbytes; // Defering buffer size
+ wxTimer *m_defer_timer; // Timer for defering mode
+
+ bool m_error; // Did an error occur in last IO call ?
+ wxList m_states; // Stack of states
+
public:
wxSocketBase();
virtual ~wxSocketBase();
virtual bool Close();
// Base IO
- wxSocketBase& Peek(char* buffer, size_t nbytes);
- wxSocketBase& Read(char* buffer, size_t nbytes);
- wxSocketBase& Write(const char *buffer, size_t nbytes);
- wxSocketBase& Unread(const char *buffer, size_t nbytes);
- wxSocketBase& ReadMsg(char *buffer, size_t nbytes);
- wxSocketBase& WriteMsg(const char *buffer, size_t nbytes);
+ wxSocketBase& Peek(char* buffer, wxUint32 nbytes);
+ wxSocketBase& Read(char* buffer, wxUint32 nbytes);
+ wxSocketBase& Write(const char *buffer, wxUint32 nbytes);
+ wxSocketBase& Unread(const char *buffer, wxUint32 nbytes);
+ wxSocketBase& ReadMsg(char *buffer, wxUint32 nbytes);
+ wxSocketBase& WriteMsg(const char *buffer, wxUint32 nbytes);
void Discard();
- // Try not to use this two methods (they sould be protected)
- void CreatePushbackAfter(const char *buffer, size_t size);
- void CreatePushbackBefore(const char *buffer, size_t size);
-
// Status
- inline bool Ok() const { return (m_fd < 0 ? 0 : 1); };
- inline bool Error() const { return (m_error != 0); };
+ inline bool Ok() const { return (m_socket != NULL); };
+ inline bool Error() const { return m_error; };
inline bool IsConnected() const { return m_connected; };
inline bool IsDisconnected() const { return !IsConnected(); };
- inline bool IsNoWait() const { return m_flags & NOWAIT; };
+ inline bool IsNoWait() const { return ((m_flags & NOWAIT) != 0); };
bool IsData() const;
- inline size_t LastCount() const { return m_lcount; }
- inline int LastError() const { return m_error; }
+ inline wxUint32 LastCount() const { return m_lcount; }
+ inline wxSocketError LastError() const { return (wxSocketError)GSocket_GetError(m_socket); }
inline wxSockType GetType() const { return m_type; }
-
+
+ // Some info on the socket...
+ virtual bool GetPeer(wxSockAddress& addr_man) const;
+ virtual bool GetLocal(wxSockAddress& addr_man) const;
+
+ // Set attributes and flags
void SetFlags(wxSockFlags _flags);
wxSockFlags GetFlags() const;
- inline void SetTimeout(unsigned long sec) { m_timeout = sec; }
-
- // seconds = -1 means infinite wait
- // seconds = 0 means no wait
- // seconds > 0 means specified wait
- bool Wait(long seconds = -1, long microseconds = 0);
- bool WaitForRead(long seconds = -1, long microseconds = 0);
- bool WaitForWrite(long seconds = -1, long microseconds = 0);
- bool WaitForLost(long seconds = -1, long microseconds = 0);
-
+ void SetTimeout(long seconds);
+
+ // Wait functions
+ // seconds = -1 means default timeout (change with SetTimeout)
+ // seconds, milliseconds = 0 means no wait
+ // seconds, milliseconds > 0 means specified wait
+ bool Wait(long seconds = -1, long milliseconds = 0);
+ bool WaitForRead(long seconds = -1, long milliseconds = 0);
+ bool WaitForWrite(long seconds = -1, long milliseconds = 0);
+ bool WaitForLost(long seconds = -1, long milliseconds = 0);
+
// Save the current state of Socket
void SaveState();
void RestoreState();
- // Setup external callback
- wxSockCbk Callback(wxSockCbk cbk_);
- char *CallbackData(char *data);
-
// Setup event handler
void SetEventHandler(wxEvtHandler& evt_hdlr, int id = -1);
-
- // Method called when it happens something on the socket
- void SetNotify(wxRequestNotify flags);
- virtual void OnRequest(wxRequestEvent req_evt);
-
- // Public internal callback
- virtual void OldOnNotify(wxRequestEvent WXUNUSED(evt));
- // Some info on the socket...
- virtual bool GetPeer(wxSockAddress& addr_man) const;
- virtual bool GetLocal(wxSockAddress& addr_man) const;
-
- // Install or uninstall callbacks
+ // Tell wxSocket which events to notify
+ void SetNotify(wxSocketEventFlags flags);
void Notify(bool notify);
+ static wxSocketEventFlags EventToNotify(wxSocketNotify evt);
+ inline wxSocketEventFlags NeededReq() const { return m_neededreq; }
- // So you can know what the socket driver is looking for ...
- inline wxRequestNotify NeededReq() const { return m_neededreq; }
+ // External callback
+ wxSockCbk Callback(wxSockCbk cbk_);
+ char *CallbackData(char *data);
- static wxRequestNotify EventToNotify(wxRequestEvent evt);
+ // Public internal callback
+ virtual void OldOnNotify(wxSocketNotify WXUNUSED(evt));
+
+ // Do NOT use these functions; they should be protected!
+ void CreatePushbackAfter(const char *buffer, wxUint32 size);
+ void CreatePushbackBefore(const char *buffer, wxUint32 size);
+ void OnRequest(wxSocketNotify req_evt);
protected:
friend class wxSocketServer;
+ friend class wxSocketClient;
friend class wxSocketHandler;
- friend class wxSocketInternal;
#ifdef __SALFORDC__
public:
#ifdef __SALFORDC__
protected:
#endif
-
- bool _Wait(long seconds, long microseconds, int type);
-
- // Set "my" handler
- inline virtual void SetHandler(wxSocketHandler *handler)
- { m_handler = handler; }
- // Pushback library
- size_t GetPushback(char *buffer, size_t size, bool peek);
+ bool _Wait(long seconds, long milliseconds, wxSocketEventFlags flags);
- // To prevent many read or write on the same socket at the same time
- // ==> cause strange things :-)
- void WantSpeedBuffer(char *buffer, size_t size, wxRequestEvent req);
- void WantBuffer(char *buffer, size_t size, wxRequestEvent req);
+ int DeferRead(char *buffer, wxUint32 nbytes);
+ int DeferWrite(const char *buffer, wxUint32 nbytes);
+ void DoDefer();
+
+ // Pushback library
+ wxUint32 GetPushback(char *buffer, wxUint32 size, bool peek);
};
////////////////////////////////////////////////////////////////////////
{
DECLARE_CLASS(wxSocketServer)
public:
-
// 'service' can be a name or a port-number
wxSocketServer(wxSockAddress& addr_man, wxSockFlags flags = wxSocketBase::NONE);
- wxSocketBase* Accept();
- bool AcceptWith(wxSocketBase& sock);
+ wxSocketBase* Accept(bool wait = TRUE);
+ bool AcceptWith(wxSocketBase& sock, bool wait = TRUE);
+
+ bool WaitForAccept(long seconds = -1, long milliseconds = 0);
};
////////////////////////////////////////////////////////////////////////
virtual bool Connect(wxSockAddress& addr_man, bool wait = TRUE);
- bool WaitOnConnect(long seconds = -1, long microseconds = 0);
-
- virtual void OnRequest(wxRequestEvent flags);
-};
-
-////////////////////////////////////////////////////////////////////////
-
-class WXDLLEXPORT wxSocketHandler : public wxObject
-{
- DECLARE_CLASS(wxSocketHandler)
-protected:
- wxList *socks;
-
-public:
- enum SockStatus { SOCK_NONE, SOCK_DATA, SOCK_CONNECT, SOCK_DISCONNECT,
- SOCK_ERROR };
- static wxSocketHandler *master;
-
- wxSocketHandler();
- virtual ~wxSocketHandler();
-
- void Register(wxSocketBase* sock);
- void UnRegister(wxSocketBase* sock);
-
- unsigned long Count() const;
-
- // seconds = -1 means indefinite wait
- // seconds = 0 means no wait
- // seconds > 0 means specified wait
-
- int Wait(long seconds = -1, long microseconds = 0);
- void YieldSock();
-
- wxSocketServer *CreateServer
- (wxSockAddress& addr,
- wxSocketBase::wxSockFlags flags = wxSocketBase::NONE);
- wxSocketClient *CreateClient
- (wxSocketBase::wxSockFlags flags = wxSocketBase::NONE);
-
- // Create or reuse a socket handler
- static wxSocketHandler& Master() { return *master; }
+ bool WaitOnConnect(long seconds = -1, long milliseconds = 0);
};
class WXDLLEXPORT wxSocketEvent : public wxEvent {
public:
wxSocketEvent(int id = 0);
- wxSocketBase::wxRequestEvent SocketEvent() const { return m_skevt; }
+ wxSocketNotify SocketEvent() const { return m_skevt; }
wxSocketBase *Socket() const { return m_socket; }
void CopyObject(wxObject& obj_d) const;
public:
- wxSocketBase::wxRequestEvent m_skevt;
+ wxSocketNotify m_skevt;
wxSocketBase *m_socket;
};