]> git.saurik.com Git - wxWidgets.git/blame - include/wx/socket.h
wxCriticalSection implemented using mutexes for !MSW (sample compiles again)
[wxWidgets.git] / include / wx / socket.h
CommitLineData
f4ada568
GL
1/////////////////////////////////////////////////////////////////////////////
2// Name: socket.h
3// Purpose: Socket handling classes
4// Author: Guilhem Lavaux
5// Modified by:
6// Created: April 1997
7// RCS-ID: $Id$
8// Copyright: (c) Guilhem Lavaux
9// Licence: wxWindows licence
10/////////////////////////////////////////////////////////////////////////////
0c32066b 11
f4ada568
GL
12#ifndef _WX_NETWORK_SOCKET_H
13#define _WX_NETWORK_SOCKET_H
14
15#ifdef __GNUG__
0c32066b 16#pragma interface "socket.h"
f4ada568
GL
17#endif
18
19// ---------------------------------------------------------------------------
20// Windows(tm) specific
21// ---------------------------------------------------------------------------
22#if defined(__WINDOWS__) && defined(WXSOCK_INTERNAL)
23#include <winsock.h>
24#include <wx/msw/private.h>
25
26struct wxSockInternal {
27 UINT my_msg;
28};
29
30struct wxSockHandlerInternal {
31 HWND sockWin;
32 UINT firstAvailableMsg;
33};
34#endif // defined(__WINDOWS__) && defined(WXSOCK_INTERNAL)
35
36// ---------------------------------------------------------------------------
37// Unix specific
38// ---------------------------------------------------------------------------
39#if defined(__UNIX__) && defined(WXSOCK_INTERNAL)
40#include <sys/types.h>
41#include <sys/socket.h>
42#include <netinet/in.h>
43
44// ---------------------------------------------------------------------------
45// Athena specific
46// ---------------------------------------------------------------------------
47#if defined(__WXXT__) || defined(__WXMOTIF__)
48#include <X11/Intrinsic.h>
49
50struct wxSockInternal {
51 XtInputId sock_inputid, sock_outputid, sock_exceptid;
52};
53#endif
54
55// ---------------------------------------------------------------------------
56// GTK specific
57// ---------------------------------------------------------------------------
58#if defined(__WXGTK__)
59#include <gdk/gdk.h>
60
61struct wxSockInternal {
62 gint sock_inputid, sock_outputid, sock_exceptid;
63};
64#endif
65
66#endif // defined(__UNIX__) && defined(WXSOCK_INTERNAL)
67
68// ---------------------------------------------------------------------------
69// wxSocket headers (generic)
70// ---------------------------------------------------------------------------
71#ifdef WXPREC
72#include <wx/wxprec.h>
73#else
74#include <wx/wx.h>
75#endif
76#include "wx/sckaddr.h"
77
78class WXDLLEXPORT wxSocketEvent;
79class WXDLLEXPORT wxSocketHandler;
80class WXDLLEXPORT wxSocketBase : public wxEvtHandler
81{
82 DECLARE_CLASS(wxSocketBase)
17dff81c
SC
83#ifdef __WXMAC__
84friend void wxMacSocketOnRequestProc(void *refcon , short event) ;
85#endif
f4ada568
GL
86public:
87
88 enum wxSockFlags { NONE=0, NOWAIT=1, WAITALL=2, SPEED=4 };
89 // Type of request
90 enum { REQ_READ=0x1, REQ_PEEK=0x2, REQ_WRITE=0x4, REQ_LOST=0x8,
91 REQ_ACCEPT=0x10, REQ_CONNECT=0x20};
92 enum { EVT_READ=0, EVT_PEEK=1, EVT_WRITE=2, EVT_LOST=3, EVT_ACCEPT=4,
93 EVT_CONNECT=5 };
94
95 typedef int wxRequestNotify;
96 typedef int wxRequestEvent;
97 typedef void (*wxSockCbk)(wxSocketBase& sock,wxRequestEvent evt,char *cdata);
98
99protected:
100 wxList req_list[EVT_WRITE+1];
101
102 // Internal use for SaveState() and RestoreState()
103 class wxSockState : public wxObject {
104 public:
105 bool cbk_on;
106 wxSockCbk cbk;
107 char *cdata;
108 bool notif;
109 wxRequestNotify cbk_set;
110 wxSockFlags flags;
111 };
112 typedef struct {
113 char sig[4];
114 char len[4];
115 } SockMsg;
116 enum wxSockType { SOCK_CLIENT, SOCK_SERVER, SOCK_INTERNAL, SOCK_UNINIT };
117
118 wxSockFlags m_flags;
119 wxSockType m_type; // wxSocket type
120 bool m_connected, m_connecting; // State of the socket
121 int m_fd; // Socket file descriptors
122 int m_waitflags; // Wait flags
123 wxList m_states; // States list
124 wxSockCbk m_cbk; // C callback
125 char *m_cdata; // C callback data
126 int m_id; // Socket id (for event handler)
127 wxSocketHandler *m_handler; // the current socket handler
128 wxRequestNotify m_neededreq; // Specify which requet signals we need
129 bool m_cbkon;
130 char *m_unread; // The unread buf
131 size_t m_unrd_size; // The size of the unread buf
132 bool m_processing; // To prevent some endless loop
133 unsigned long m_timeout;
134 int m_wantbuf;
135 size_t m_lcount; // Last IO request size
136 int m_error; // Last IO error
137 bool m_notifyme;
138
139 struct wxSockInternal *m_internal; // System specific variables
140
141public:
142 wxSocketBase();
143 virtual ~wxSocketBase();
144 virtual bool Close();
145
146 // Base IO
147 wxSocketBase& Peek(char* buffer, size_t nbytes);
148 wxSocketBase& Read(char* buffer, size_t nbytes);
149 wxSocketBase& Write(const char *buffer, size_t nbytes);
150 wxSocketBase& WriteMsg(const char *buffer, size_t nbytes);
151 wxSocketBase& ReadMsg(char* buffer, size_t nbytes);
152 wxSocketBase& Unread(const char *buffer, size_t nbytes);
153 void Discard();
154
155 // Try not to use this two methods (they sould be protected)
156 void CreatePushbackAfter(const char *buffer, size_t size);
157 void CreatePushbackBefore(const char *buffer, size_t size);
158
159 // Status
160 inline bool Ok() const { return (m_fd < 0 ? 0 : 1); };
161 inline bool Error() const { return (m_error != 0); };
162 inline bool IsConnected() const { return m_connected; };
163 inline bool IsDisconnected() const { return !IsConnected(); };
164 inline bool IsNoWait() const { return m_flags & NOWAIT; };
165 bool IsData() const;
166 inline size_t LastCount() const { return m_lcount; }
167 inline int LastError() const { return m_error; }
168
169 inline void SetFlags(wxSockFlags _flags);
170 inline void SetTimeout(unsigned long sec) { m_timeout = sec; }
171
172 // seconds = -1 means infinite wait
173 // seconds = 0 means no wait
174 // seconds > 0 means specified wait
175 bool Wait(long seconds = -1, long microseconds = 0);
176 bool WaitForRead(long seconds = -1, long microseconds = 0);
177 bool WaitForWrite(long seconds = -1, long microseconds = 0);
178 bool WaitForLost(long seconds = -1, long microseconds = 0);
179
180 // Save the current state of Socket
181 void SaveState();
182 void RestoreState();
183
184 // Setup external callback
185 wxSockCbk Callback(wxSockCbk cbk_);
186 char *CallbackData(char *data);
187
188 // Setup event handler
189 void SetEventHandler(wxEvtHandler& evt_hdlr, int id = -1);
190
191 // Method called when it happens something on the socket
192 void SetNotify(wxRequestNotify flags);
193 virtual void OnRequest(wxRequestEvent req_evt);
194
195 // Public internal callback
196 virtual void OldOnNotify(wxRequestEvent WXUNUSED(evt));
197
198 // Some info on the socket...
199 virtual bool GetPeer(wxSockAddress& addr_man) const;
200 virtual bool GetLocal(wxSockAddress& addr_man) const;
201
202 // Install or uninstall callbacks
203 void Notify(bool notify);
204
205 // So you can know what the socket driver is looking for ...
206 inline wxRequestNotify NeededReq() const { return m_neededreq; }
207
208 static wxRequestNotify EventToNotify(wxRequestEvent evt);
209
210protected:
211 friend class wxSocketServer;
212 friend class wxSocketHandler;
213
214 wxSocketBase(wxSockFlags flags, wxSockType type);
215
216 bool _Wait(long seconds, long microseconds, int type);
217
218 // Set "my" handler
219 inline virtual void SetHandler(wxSocketHandler *handler)
220 { m_handler = handler; }
221
222 // Activate or disactivate callback
223 void SetupCallbacks();
224 void DestroyCallbacks();
225
226 // Pushback library
227 size_t GetPushback(char *buffer, size_t size, bool peek);
228
229 // To prevent many read or write on the same socket at the same time
230 // ==> cause strange things :-)
231 void WantSpeedBuffer(char *buffer, size_t size, wxRequestEvent req);
232 void WantBuffer(char *buffer, size_t size, wxRequestEvent req);
233
234 virtual bool DoRequests(wxRequestEvent req);
235};
236
237////////////////////////////////////////////////////////////////////////
238
239class WXDLLEXPORT wxSocketServer : public wxSocketBase
240{
241 DECLARE_CLASS(wxSocketServer)
242public:
243
244 // 'service' can be a name or a port-number
245
246 wxSocketServer(wxSockAddress& addr_man, wxSockFlags flags = wxSocketBase::NONE);
247
248 wxSocketBase* Accept();
249 bool AcceptWith(wxSocketBase& sock);
250 virtual void OnRequest(wxRequestEvent flags);
251};
252
253////////////////////////////////////////////////////////////////////////
254
255class WXDLLEXPORT wxSocketClient : public wxSocketBase
256{
257 DECLARE_CLASS(wxSocketClient)
258public:
259
260 wxSocketClient(wxSockFlags flags = wxSocketBase::NONE);
261 virtual ~wxSocketClient();
262
263 virtual bool Connect(wxSockAddress& addr_man, bool wait = TRUE);
264
75ed1d15 265 bool WaitOnConnect(long seconds = -1, long microseconds = 0);
f4ada568
GL
266
267 virtual void OnRequest(wxRequestEvent flags);
268};
269
270////////////////////////////////////////////////////////////////////////
271
272class WXDLLEXPORT wxSocketHandler : public wxObject
273{
274 DECLARE_CLASS(wxSocketHandler)
275protected:
f4ada568
GL
276#if defined(__WINDOWS__)
277 wxList *smsg_list;
278 struct wxSockHandlerInternal *internal;
279#endif
280 wxList *socks;
281
282public:
283 enum SockStatus { SOCK_NONE, SOCK_DATA, SOCK_CONNECT, SOCK_DISCONNECT,
284 SOCK_ERROR };
3b4183d8 285 static wxSocketHandler *master;
f4ada568
GL
286
287 wxSocketHandler();
288 virtual ~wxSocketHandler();
289
290 void Register(wxSocketBase* sock);
291 void UnRegister(wxSocketBase* sock);
292
293 unsigned long Count() const;
294
295 // seconds = -1 means indefinite wait
296 // seconds = 0 means no wait
297 // seconds > 0 means specified wait
298
299 int Wait(long seconds = -1, long microseconds = 0);
300 void YieldSock();
301
302 wxSocketServer *CreateServer
303 (wxSockAddress& addr,
304 wxSocketBase::wxSockFlags flags = wxSocketBase::NONE);
305 wxSocketClient *CreateClient
306 (wxSocketBase::wxSockFlags flags = wxSocketBase::NONE);
307
308 // Create or reuse a socket handler
3b4183d8 309 static wxSocketHandler& Master() { return *master; }
f4ada568
GL
310
311#if defined(WXSOCK_INTERNAL) && defined(__WINDOWS__)
312
313 friend LRESULT APIENTRY _EXPORT wxSocketHandlerWndProc(HWND hWnd,
314 UINT message, WPARAM wParam, LPARAM lParam);
315
316 UINT NewMessage(wxSocketBase *sock);
317 void DestroyMessage(UINT msg);
318
319 HWND GetHWND() const;
320#endif
321};
322
323class WXDLLEXPORT wxSocketEvent : public wxEvent {
324 DECLARE_DYNAMIC_CLASS(wxSocketEvent)
325public:
326 wxSocketEvent(int id = 0);
327
328 wxSocketBase::wxRequestEvent SocketEvent() const { return m_skevt; }
329public:
330 wxSocketBase::wxRequestEvent m_skevt;
331};
332
333typedef void (wxEvtHandler::*wxSocketEventFunction)(wxSocketEvent&);
334
17dff81c
SC
335#define wxEVT_SOCKET wxEVT_FIRST+301
336
db131261
RR
337#define EVT_SOCKET(id, func) { wxEVT_SOCKET, id, 0, \
338 (wxObjectEventFunction) (wxEventFunction) (wxSocketEventFunction) & func, \
339 (wxObject *) NULL },
f4ada568
GL
340
341#endif