]>
Commit | Line | Data |
---|---|---|
f85d901f GRG |
1 | ///////////////////////////////////////////////////////////////////////////// |
2 | // Name: server.cpp | |
3 | // Purpose: Server for wxSocket demo | |
4 | // Author: Guillermo Rodriguez Garcia <guille@iies.es> | |
5 | // Modified by: | |
6 | // Created: 1999/09/19 | |
7 | // RCS-ID: $Id$ | |
8 | // Copyright: (c) 1999 Guillermo Rodriguez Garcia | |
9 | // Licence: wxWindows licence | |
10 | ///////////////////////////////////////////////////////////////////////////// | |
11 | ||
12 | // ========================================================================== | |
13 | // declarations | |
14 | // ========================================================================== | |
15 | ||
16 | // -------------------------------------------------------------------------- | |
17 | // headers | |
18 | // -------------------------------------------------------------------------- | |
e2a6f233 | 19 | |
788233da | 20 | #if defined(__GNUG__) && !defined(__APPLE__) |
f85d901f GRG |
21 | # pragma implementation "server.cpp" |
22 | # pragma interface "server.cpp" | |
f4ada568 GL |
23 | #endif |
24 | ||
f85d901f | 25 | // For compilers that support precompilation, includes "wx/wx.h". |
f4ada568 GL |
26 | #include "wx/wxprec.h" |
27 | ||
28 | #ifdef __BORLANDC__ | |
f85d901f | 29 | # pragma hdrstop |
f4ada568 GL |
30 | #endif |
31 | ||
677a9f0c | 32 | // for all others, include the necessary headers |
f4ada568 | 33 | #ifndef WX_PRECOMP |
f85d901f | 34 | # include "wx/wx.h" |
f4ada568 | 35 | #endif |
e2a6f233 | 36 | |
09fb22cf JS |
37 | #include "wx/socket.h" |
38 | ||
f85d901f GRG |
39 | // -------------------------------------------------------------------------- |
40 | // resources | |
41 | // -------------------------------------------------------------------------- | |
f4ada568 | 42 | |
f85d901f | 43 | // the application icon |
618f2efa | 44 | #if defined(__WXGTK__) || defined(__WXX11__) || defined(__WXMOTIF__) || defined(__WXMAC__) |
f85d901f | 45 | # include "mondrian.xpm" |
e2a6f233 JS |
46 | #endif |
47 | ||
f85d901f GRG |
48 | // -------------------------------------------------------------------------- |
49 | // classes | |
50 | // -------------------------------------------------------------------------- | |
51 | ||
f4ada568 | 52 | // Define a new application type |
f85d901f GRG |
53 | class MyApp : public wxApp |
54 | { | |
55 | public: | |
56 | virtual bool OnInit(); | |
f4ada568 GL |
57 | }; |
58 | ||
f85d901f GRG |
59 | // Define a new frame type: this is going to be our main frame |
60 | class MyFrame : public wxFrame | |
61 | { | |
f4ada568 | 62 | public: |
f85d901f GRG |
63 | MyFrame(); |
64 | ~MyFrame(); | |
65 | ||
66 | // event handlers (these functions should _not_ be virtual) | |
67 | void OnQuit(wxCommandEvent& event); | |
68 | void OnAbout(wxCommandEvent& event); | |
69 | void OnServerEvent(wxSocketEvent& event); | |
70 | void OnSocketEvent(wxSocketEvent& event); | |
71 | ||
72 | void Test1(wxSocketBase *sock); | |
73 | void Test2(wxSocketBase *sock); | |
74 | void Test3(wxSocketBase *sock); | |
75 | ||
76 | // convenience functions | |
77 | void UpdateStatusBar(); | |
78 | ||
79 | private: | |
80 | wxSocketServer *m_server; | |
f85d901f GRG |
81 | wxTextCtrl *m_text; |
82 | wxMenu *m_menuFile; | |
83 | wxMenuBar *m_menuBar; | |
84 | bool m_busy; | |
85 | int m_numClients; | |
86 | ||
be5a51fb | 87 | // any class wishing to process wxWidgets events must use this macro |
f85d901f GRG |
88 | DECLARE_EVENT_TABLE() |
89 | }; | |
90 | ||
91 | // -------------------------------------------------------------------------- | |
92 | // constants | |
93 | // -------------------------------------------------------------------------- | |
94 | ||
95 | // IDs for the controls and the menu commands | |
96 | enum | |
97 | { | |
98 | // menu items | |
99 | SERVER_QUIT = 1000, | |
100 | SERVER_ABOUT, | |
101 | ||
102 | // id for sockets | |
103 | SERVER_ID, | |
104 | SOCKET_ID | |
f4ada568 GL |
105 | }; |
106 | ||
f85d901f | 107 | // -------------------------------------------------------------------------- |
be5a51fb | 108 | // event tables and other macros for wxWidgets |
f85d901f | 109 | // -------------------------------------------------------------------------- |
f4ada568 GL |
110 | |
111 | BEGIN_EVENT_TABLE(MyFrame, wxFrame) | |
f85d901f GRG |
112 | EVT_MENU(SERVER_QUIT, MyFrame::OnQuit) |
113 | EVT_MENU(SERVER_ABOUT, MyFrame::OnAbout) | |
114 | EVT_SOCKET(SERVER_ID, MyFrame::OnServerEvent) | |
115 | EVT_SOCKET(SOCKET_ID, MyFrame::OnSocketEvent) | |
f4ada568 GL |
116 | END_EVENT_TABLE() |
117 | ||
f4ada568 GL |
118 | IMPLEMENT_APP(MyApp) |
119 | ||
f4ada568 | 120 | |
f85d901f GRG |
121 | // ========================================================================== |
122 | // implementation | |
123 | // ========================================================================== | |
f4ada568 | 124 | |
f85d901f GRG |
125 | // -------------------------------------------------------------------------- |
126 | // the application class | |
127 | // -------------------------------------------------------------------------- | |
f4ada568 | 128 | |
f85d901f GRG |
129 | bool MyApp::OnInit() |
130 | { | |
131 | // Create the main application window | |
132 | MyFrame *frame = new MyFrame(); | |
f4ada568 | 133 | |
f85d901f | 134 | // Show it and tell the application that it's our main window |
b62ca03d | 135 | frame->Show(true); |
f85d901f | 136 | SetTopWindow(frame); |
2b98cb1f | 137 | |
bd4d918f | 138 | // Success |
b62ca03d | 139 | return true; |
f4ada568 GL |
140 | } |
141 | ||
f85d901f GRG |
142 | // -------------------------------------------------------------------------- |
143 | // main frame | |
144 | // -------------------------------------------------------------------------- | |
a737331d | 145 | |
f85d901f | 146 | // frame constructor |
bd4d918f | 147 | |
b62ca03d | 148 | MyFrame::MyFrame() : wxFrame((wxFrame *)NULL, wxID_ANY, |
bd4d918f | 149 | _("wxSocket demo: Server"), |
f85d901f | 150 | wxDefaultPosition, wxSize(300, 200)) |
f4ada568 | 151 | { |
f85d901f GRG |
152 | // Give the frame an icon |
153 | SetIcon(wxICON(mondrian)); | |
154 | ||
155 | // Make menus | |
156 | m_menuFile = new wxMenu(); | |
bd4d918f | 157 | m_menuFile->Append(SERVER_ABOUT, _("&About...\tCtrl-A"), _("Show about dialog")); |
f85d901f | 158 | m_menuFile->AppendSeparator(); |
bd4d918f | 159 | m_menuFile->Append(SERVER_QUIT, _("E&xit\tAlt-X"), _("Quit server")); |
f85d901f GRG |
160 | |
161 | // Append menus to the menubar | |
162 | m_menuBar = new wxMenuBar(); | |
bd4d918f | 163 | m_menuBar->Append(m_menuFile, _("&File")); |
f85d901f GRG |
164 | SetMenuBar(m_menuBar); |
165 | ||
166 | // Status bar | |
167 | CreateStatusBar(2); | |
168 | ||
bd4d918f | 169 | // Make a textctrl for logging |
b62ca03d | 170 | m_text = new wxTextCtrl(this, wxID_ANY, |
bd4d918f GRG |
171 | _("Welcome to wxSocket demo: Server\n"), |
172 | wxDefaultPosition, wxDefaultSize, | |
f85d901f GRG |
173 | wxTE_MULTILINE | wxTE_READONLY); |
174 | ||
bd4d918f | 175 | // Create the address - defaults to localhost:0 initially |
f85d901f GRG |
176 | wxIPV4address addr; |
177 | addr.Service(3000); | |
178 | ||
bd4d918f | 179 | // Create the socket |
f85d901f | 180 | m_server = new wxSocketServer(addr); |
f85d901f GRG |
181 | |
182 | // We use Ok() here to see if the server is really listening | |
bd4d918f GRG |
183 | if (! m_server->Ok()) |
184 | { | |
185 | m_text->AppendText(_("Could not listen at the specified port !\n\n")); | |
186 | return; | |
187 | } | |
f85d901f | 188 | else |
bd4d918f GRG |
189 | { |
190 | m_text->AppendText(_("Server listening.\n\n")); | |
191 | } | |
192 | ||
193 | // Setup the event handler and subscribe to connection events | |
194 | m_server->SetEventHandler(*this, SERVER_ID); | |
195 | m_server->SetNotify(wxSOCKET_CONNECTION_FLAG); | |
b62ca03d | 196 | m_server->Notify(true); |
f85d901f | 197 | |
b62ca03d | 198 | m_busy = false; |
f85d901f GRG |
199 | m_numClients = 0; |
200 | UpdateStatusBar(); | |
f4ada568 GL |
201 | } |
202 | ||
f85d901f | 203 | MyFrame::~MyFrame() |
f4ada568 | 204 | { |
bd4d918f | 205 | // No delayed deletion here, as the frame is dying anyway |
f85d901f GRG |
206 | delete m_server; |
207 | } | |
f4ada568 | 208 | |
f85d901f | 209 | // event handlers |
f4ada568 | 210 | |
f85d901f GRG |
211 | void MyFrame::OnQuit(wxCommandEvent& WXUNUSED(event)) |
212 | { | |
b62ca03d WS |
213 | // true is to force the frame to close |
214 | Close(true); | |
f85d901f | 215 | } |
f4ada568 | 216 | |
f85d901f GRG |
217 | void MyFrame::OnAbout(wxCommandEvent& WXUNUSED(event)) |
218 | { | |
4693b20c | 219 | wxMessageBox(_("wxSocket demo: Server\n(c) 1999 Guillermo Rodriguez Garcia\n"), |
bd4d918f | 220 | _("About Server"), |
f85d901f | 221 | wxOK | wxICON_INFORMATION, this); |
f4ada568 GL |
222 | } |
223 | ||
f85d901f | 224 | void MyFrame::Test1(wxSocketBase *sock) |
f4ada568 | 225 | { |
f85d901f GRG |
226 | unsigned char len; |
227 | char *buf; | |
f4ada568 | 228 | |
bd4d918f | 229 | m_text->AppendText(_("Test 1 begins\n")); |
f4ada568 | 230 | |
f85d901f GRG |
231 | // Receive data from socket and send it back. We will first |
232 | // get a byte with the buffer size, so we can specify the | |
6097c3a2 GRG |
233 | // exact size and use the wxSOCKET_WAITALL flag. Also, we |
234 | // disabled input events so we won't have unwanted reentrance. | |
235 | // This way we can avoid the infamous wxSOCKET_BLOCK flag. | |
e51b0130 | 236 | |
f85d901f GRG |
237 | sock->SetFlags(wxSOCKET_WAITALL); |
238 | ||
bc7e88ae | 239 | // Read the size |
f187448d | 240 | sock->Read(&len, 1); |
6097c3a2 | 241 | buf = new char[len]; |
bc7e88ae GRG |
242 | |
243 | // Read the data | |
f85d901f | 244 | sock->Read(buf, len); |
bc7e88ae GRG |
245 | m_text->AppendText(_("Got the data, sending it back\n")); |
246 | ||
247 | // Write it back | |
f85d901f | 248 | sock->Write(buf, len); |
6097c3a2 | 249 | delete[] buf; |
f85d901f | 250 | |
bc7e88ae | 251 | m_text->AppendText(_("Test 1 ends\n\n")); |
f4ada568 GL |
252 | } |
253 | ||
f85d901f | 254 | void MyFrame::Test2(wxSocketBase *sock) |
f4ada568 | 255 | { |
f85d901f GRG |
256 | #define MAX_MSG_SIZE 10000 |
257 | ||
258 | wxString s; | |
4693b20c | 259 | wxChar *buf = new wxChar[MAX_MSG_SIZE]; |
f85d901f GRG |
260 | wxUint32 len; |
261 | ||
bd4d918f | 262 | m_text->AppendText(_("Test 2 begins\n")); |
f85d901f GRG |
263 | |
264 | // We don't need to set flags because ReadMsg and WriteMsg | |
265 | // are not affected by them anyway. | |
e51b0130 | 266 | |
bc7e88ae | 267 | // Read the message |
4693b20c | 268 | len = sock->ReadMsg(buf, MAX_MSG_SIZE * sizeof(wxChar)).LastCount(); |
bd4d918f | 269 | s.Printf(_("Client says: %s\n"), buf); |
f85d901f | 270 | m_text->AppendText(s); |
36bec0ac | 271 | m_text->AppendText(_("Sending the data back\n")); |
f85d901f | 272 | |
bc7e88ae | 273 | // Write it back |
f85d901f | 274 | sock->WriteMsg(buf, len); |
6097c3a2 | 275 | delete[] buf; |
f85d901f | 276 | |
bc7e88ae | 277 | m_text->AppendText(_("Test 2 ends\n\n")); |
f85d901f GRG |
278 | |
279 | #undef MAX_MSG_SIZE | |
f4ada568 GL |
280 | } |
281 | ||
f85d901f | 282 | void MyFrame::Test3(wxSocketBase *sock) |
f4ada568 | 283 | { |
6097c3a2 GRG |
284 | unsigned char len; |
285 | char *buf; | |
286 | ||
bd4d918f | 287 | m_text->AppendText(_("Test 3 begins\n")); |
6097c3a2 | 288 | |
677a9f0c | 289 | // This test is similar to the first one, but the len is |
6097c3a2 | 290 | // expressed in kbytes - this tests large data transfers. |
e51b0130 | 291 | |
6097c3a2 GRG |
292 | sock->SetFlags(wxSOCKET_WAITALL); |
293 | ||
bc7e88ae | 294 | // Read the size |
f187448d | 295 | sock->Read(&len, 1); |
6097c3a2 | 296 | buf = new char[len * 1024]; |
bc7e88ae GRG |
297 | |
298 | // Read the data | |
6097c3a2 | 299 | sock->Read(buf, len * 1024); |
bc7e88ae GRG |
300 | m_text->AppendText(_("Got the data, sending it back\n")); |
301 | ||
302 | // Write it back | |
6097c3a2 GRG |
303 | sock->Write(buf, len * 1024); |
304 | delete[] buf; | |
305 | ||
bc7e88ae | 306 | m_text->AppendText(_("Test 3 ends\n\n")); |
f4ada568 GL |
307 | } |
308 | ||
f85d901f | 309 | void MyFrame::OnServerEvent(wxSocketEvent& event) |
f4ada568 | 310 | { |
bd4d918f | 311 | wxString s = _("OnServerEvent: "); |
f85d901f GRG |
312 | wxSocketBase *sock; |
313 | ||
bc7e88ae | 314 | switch(event.GetSocketEvent()) |
f85d901f | 315 | { |
bd4d918f GRG |
316 | case wxSOCKET_CONNECTION : s.Append(_("wxSOCKET_CONNECTION\n")); break; |
317 | default : s.Append(_("Unexpected event !\n")); break; | |
f85d901f | 318 | } |
f4ada568 | 319 | |
f85d901f | 320 | m_text->AppendText(s); |
765e386b | 321 | |
f85d901f | 322 | // Accept new connection if there is one in the pending |
b62ca03d | 323 | // connections queue, else exit. We use Accept(false) for |
f85d901f GRG |
324 | // non-blocking accept (although if we got here, there |
325 | // should ALWAYS be a pending connection). | |
e51b0130 | 326 | |
b62ca03d | 327 | sock = m_server->Accept(false); |
f85d901f GRG |
328 | |
329 | if (sock) | |
330 | { | |
bc7e88ae | 331 | m_text->AppendText(_("New client connection accepted\n\n")); |
f85d901f GRG |
332 | } |
333 | else | |
334 | { | |
bc7e88ae | 335 | m_text->AppendText(_("Error: couldn't accept a new connection\n\n")); |
f85d901f GRG |
336 | return; |
337 | } | |
338 | ||
339 | sock->SetEventHandler(*this, SOCKET_ID); | |
340 | sock->SetNotify(wxSOCKET_INPUT_FLAG | wxSOCKET_LOST_FLAG); | |
b62ca03d | 341 | sock->Notify(true); |
f85d901f GRG |
342 | |
343 | m_numClients++; | |
344 | UpdateStatusBar(); | |
f4ada568 GL |
345 | } |
346 | ||
f85d901f GRG |
347 | void MyFrame::OnSocketEvent(wxSocketEvent& event) |
348 | { | |
bd4d918f | 349 | wxString s = _("OnSocketEvent: "); |
f4d5e009 | 350 | wxSocketBase *sock = event.GetSocket(); |
f85d901f | 351 | |
f4d5e009 | 352 | // First, print a message |
bc7e88ae | 353 | switch(event.GetSocketEvent()) |
f85d901f | 354 | { |
f4d5e009 GRG |
355 | case wxSOCKET_INPUT : s.Append(_("wxSOCKET_INPUT\n")); break; |
356 | case wxSOCKET_LOST : s.Append(_("wxSOCKET_LOST\n")); break; | |
357 | default : s.Append(_("Unexpected event !\n")); break; | |
f85d901f GRG |
358 | } |
359 | ||
360 | m_text->AppendText(s); | |
361 | ||
362 | // Now we process the event | |
bc7e88ae | 363 | switch(event.GetSocketEvent()) |
f85d901f GRG |
364 | { |
365 | case wxSOCKET_INPUT: | |
366 | { | |
367 | // We disable input events, so that the test doesn't trigger | |
368 | // wxSocketEvent again. | |
369 | sock->SetNotify(wxSOCKET_LOST_FLAG); | |
370 | ||
371 | // Which test are we going to run? | |
372 | unsigned char c; | |
f4d5e009 | 373 | sock->Read(&c, 1); |
f85d901f GRG |
374 | |
375 | switch (c) | |
376 | { | |
377 | case 0xBE: Test1(sock); break; | |
378 | case 0xCE: Test2(sock); break; | |
379 | case 0xDE: Test3(sock); break; | |
f4d5e009 GRG |
380 | default: |
381 | m_text->AppendText(_("Unknown test id received from client\n\n")); | |
f85d901f GRG |
382 | } |
383 | ||
384 | // Enable input events again. | |
385 | sock->SetNotify(wxSOCKET_LOST_FLAG | wxSOCKET_INPUT_FLAG); | |
386 | break; | |
387 | } | |
388 | case wxSOCKET_LOST: | |
389 | { | |
390 | m_numClients--; | |
391 | ||
bd4d918f GRG |
392 | // Destroy() should be used instead of delete wherever possible, |
393 | // due to the fact that wxSocket uses 'delayed events' (see the | |
394 | // documentation for wxPostEvent) and we don't want an event to | |
395 | // arrive to the event handler (the frame, here) after the socket | |
396 | // has been deleted. Also, we might be doing some other thing with | |
397 | // the socket at the same time; for example, we might be in the | |
398 | // middle of a test or something. Destroy() takes care of all | |
bc7e88ae | 399 | // this for us. |
bd4d918f | 400 | |
bc7e88ae | 401 | m_text->AppendText(_("Deleting socket.\n\n")); |
bd4d918f | 402 | sock->Destroy(); |
f85d901f GRG |
403 | break; |
404 | } | |
405 | default: ; | |
406 | } | |
407 | ||
408 | UpdateStatusBar(); | |
409 | } | |
410 | ||
411 | // convenience functions | |
412 | ||
413 | void MyFrame::UpdateStatusBar() | |
f4ada568 | 414 | { |
f85d901f | 415 | wxString s; |
bd4d918f | 416 | s.Printf(_("%d clients connected"), m_numClients); |
f85d901f | 417 | SetStatusText(s, 1); |
f4ada568 | 418 | } |