1 /////////////////////////////////////////////////////////////////////////////
3 // Purpose: Client for wxSocket demo
4 // Author: Guillermo Rodriguez Garcia <guille@iies.es>
8 // Copyright: (c) 1999 Guillermo Rodriguez Garcia
9 // Licence: wxWindows licence
10 /////////////////////////////////////////////////////////////////////////////
12 // ==========================================================================
14 // ==========================================================================
16 // --------------------------------------------------------------------------
18 // --------------------------------------------------------------------------
20 // For compilers that support precompilation, includes "wx/wx.h".
21 #include "wx/wxprec.h"
27 // for all others, include the necessary headers
32 #include "wx/socket.h"
34 #include "wx/sstream.h"
37 // --------------------------------------------------------------------------
39 // --------------------------------------------------------------------------
41 // the application icon
42 #include "mondrian.xpm"
44 // --------------------------------------------------------------------------
46 // --------------------------------------------------------------------------
48 // Define a new application type
49 class MyApp
: public wxApp
52 virtual bool OnInit();
55 // Define a new frame type: this is going to be our main frame
56 class MyFrame
: public wxFrame
62 // event handlers for File menu
63 void OnQuit(wxCommandEvent
& event
);
64 void OnAbout(wxCommandEvent
& event
);
66 // event handlers for Socket menu
67 void OnOpenConnection(wxCommandEvent
& event
);
68 void OnTest1(wxCommandEvent
& event
);
69 void OnTest2(wxCommandEvent
& event
);
70 void OnTest3(wxCommandEvent
& event
);
71 void OnCloseConnection(wxCommandEvent
& event
);
74 // event handlers for Protocols menu
75 void OnTestURL(wxCommandEvent
& event
);
78 void OnOpenConnectionIPv6(wxCommandEvent
& event
);
81 void OpenConnection(wxSockAddress::Family family
);
83 // event handlers for DatagramSocket menu (stub)
84 void OnDatagram(wxCommandEvent
& event
);
86 // socket event handler
87 void OnSocketEvent(wxSocketEvent
& event
);
89 // convenience functions
90 void UpdateStatusBar();
93 wxSocketClient
*m_sock
;
97 wxMenu
*m_menuDatagramSocket
;
98 wxMenu
*m_menuProtocols
;
102 // any class wishing to process wxWidgets events must use this macro
103 DECLARE_EVENT_TABLE()
106 // simple helper class to log start and end of each test
110 TestLogger(const wxString
& name
) : m_name(name
)
112 wxLogMessage("=== %s test begins ===", m_name
);
117 wxLogMessage("=== %s test ends ===", m_name
);
121 const wxString m_name
;
124 // --------------------------------------------------------------------------
126 // --------------------------------------------------------------------------
128 // IDs for the controls and the menu commands
132 CLIENT_QUIT
= wxID_EXIT
,
133 CLIENT_ABOUT
= wxID_ABOUT
,
151 // --------------------------------------------------------------------------
152 // event tables and other macros for wxWidgets
153 // --------------------------------------------------------------------------
155 BEGIN_EVENT_TABLE(MyFrame
, wxFrame
)
156 EVT_MENU(CLIENT_QUIT
, MyFrame::OnQuit
)
157 EVT_MENU(CLIENT_ABOUT
, MyFrame::OnAbout
)
158 EVT_MENU(CLIENT_OPEN
, MyFrame::OnOpenConnection
)
160 EVT_MENU(CLIENT_OPENIPV6
, MyFrame::OnOpenConnectionIPv6
)
162 EVT_MENU(CLIENT_TEST1
, MyFrame::OnTest1
)
163 EVT_MENU(CLIENT_TEST2
, MyFrame::OnTest2
)
164 EVT_MENU(CLIENT_TEST3
, MyFrame::OnTest3
)
165 EVT_MENU(CLIENT_CLOSE
, MyFrame::OnCloseConnection
)
166 EVT_MENU(CLIENT_DGRAM
, MyFrame::OnDatagram
)
168 EVT_MENU(CLIENT_TESTURL
, MyFrame::OnTestURL
)
170 EVT_SOCKET(SOCKET_ID
, MyFrame::OnSocketEvent
)
175 // ==========================================================================
177 // ==========================================================================
179 // --------------------------------------------------------------------------
180 // the application class
181 // --------------------------------------------------------------------------
185 if ( !wxApp::OnInit() )
188 // Create the main application window
189 MyFrame
*frame
= new MyFrame();
191 // Show it and tell the application that it's our main window
199 // --------------------------------------------------------------------------
201 // --------------------------------------------------------------------------
204 MyFrame::MyFrame() : wxFrame((wxFrame
*)NULL
, wxID_ANY
,
205 _("wxSocket demo: Client"),
206 wxDefaultPosition
, wxSize(300, 200))
208 // Give the frame an icon
209 SetIcon(wxICON(mondrian
));
212 m_menuFile
= new wxMenu();
213 m_menuFile
->Append(CLIENT_ABOUT
, _("&About...\tCtrl-A"), _("Show about dialog"));
214 m_menuFile
->AppendSeparator();
215 m_menuFile
->Append(CLIENT_QUIT
, _("E&xit\tAlt-X"), _("Quit client"));
217 m_menuSocket
= new wxMenu();
218 m_menuSocket
->Append(CLIENT_OPEN
, _("&Open session"), _("Connect to server"));
220 m_menuSocket
->Append(CLIENT_OPENIPV6
, _("&Open session(IPv6)"), _("Connect to server(IPv6)"));
222 m_menuSocket
->AppendSeparator();
223 m_menuSocket
->Append(CLIENT_TEST1
, _("Test &1"), _("Test basic functionality"));
224 m_menuSocket
->Append(CLIENT_TEST2
, _("Test &2"), _("Test ReadMsg and WriteMsg"));
225 m_menuSocket
->Append(CLIENT_TEST3
, _("Test &3"), _("Test large data transfer"));
226 m_menuSocket
->AppendSeparator();
227 m_menuSocket
->Append(CLIENT_CLOSE
, _("&Close session"), _("Close connection"));
229 m_menuDatagramSocket
= new wxMenu();
230 m_menuDatagramSocket
->Append(CLIENT_DGRAM
, _("Send Datagram"), _("Test UDP sockets"));
233 m_menuProtocols
= new wxMenu();
234 m_menuProtocols
->Append(CLIENT_TESTURL
, _("Test URL\tCtrl-U"),
235 _("Get data from the specified URL"));
238 // Append menus to the menubar
239 m_menuBar
= new wxMenuBar();
240 m_menuBar
->Append(m_menuFile
, _("&File"));
241 m_menuBar
->Append(m_menuSocket
, _("&SocketClient"));
242 m_menuBar
->Append(m_menuDatagramSocket
, _("&DatagramSocket"));
244 m_menuBar
->Append(m_menuProtocols
, _("&Protocols"));
246 SetMenuBar(m_menuBar
);
251 #endif // wxUSE_STATUSBAR
253 // Make a textctrl for logging
254 m_text
= new wxTextCtrl(this, wxID_ANY
,
255 _("Welcome to wxSocket demo: Client\nClient ready\n"),
256 wxDefaultPosition
, wxDefaultSize
,
257 wxTE_MULTILINE
| wxTE_READONLY
);
258 delete wxLog::SetActiveTarget(new wxLogTextCtrl(m_text
));
261 m_sock
= new wxSocketClient();
263 // Setup the event handler and subscribe to most events
264 m_sock
->SetEventHandler(*this, SOCKET_ID
);
265 m_sock
->SetNotify(wxSOCKET_CONNECTION_FLAG
|
266 wxSOCKET_INPUT_FLAG
|
268 m_sock
->Notify(true);
276 // No delayed deletion here, as the frame is dying anyway
282 void MyFrame::OnQuit(wxCommandEvent
& WXUNUSED(event
))
284 // true is to force the frame to close
288 void MyFrame::OnAbout(wxCommandEvent
& WXUNUSED(event
))
290 wxMessageBox(_("wxSocket demo: Client\n(c) 1999 Guillermo Rodriguez Garcia\n"),
292 wxOK
| wxICON_INFORMATION
, this);
295 void MyFrame::OnOpenConnection(wxCommandEvent
& WXUNUSED(event
))
297 OpenConnection(wxSockAddress::IPV4
);
300 void MyFrame::OnOpenConnectionIPv6(wxCommandEvent
& WXUNUSED(event
))
302 OpenConnection(wxSockAddress::IPV6
);
306 void MyFrame::OpenConnection(wxSockAddress::Family family
)
308 wxUnusedVar(family
); // unused in !wxUSE_IPV6 case
314 if ( family
== wxSockAddress::IPV6
)
320 m_menuSocket
->Enable(CLIENT_OPEN
, false);
322 m_menuSocket
->Enable(CLIENT_OPENIPV6
, false);
324 m_menuSocket
->Enable(CLIENT_CLOSE
, false);
326 // Ask user for server address
327 wxString hostname
= wxGetTextFromUser(
328 _("Enter the address of the wxSocket demo server:"),
332 addr
->Hostname(hostname
);
335 // Mini-tutorial for Connect() :-)
336 // ---------------------------
338 // There are two ways to use Connect(): blocking and non-blocking,
339 // depending on the value passed as the 'wait' (2nd) parameter.
341 // Connect(addr, true) will wait until the connection completes,
342 // returning true on success and false on failure. This call blocks
343 // the GUI (this might be changed in future releases to honour the
344 // wxSOCKET_BLOCK flag).
346 // Connect(addr, false) will issue a nonblocking connection request
347 // and return immediately. If the return value is true, then the
348 // connection has been already successfully established. If it is
349 // false, you must wait for the request to complete, either with
350 // WaitOnConnect() or by watching wxSOCKET_CONNECTION / LOST
351 // events (please read the documentation).
353 // WaitOnConnect() itself never blocks the GUI (this might change
354 // in the future to honour the wxSOCKET_BLOCK flag). This call will
355 // return false on timeout, or true if the connection request
356 // completes, which in turn might mean:
358 // a) That the connection was successfully established
359 // b) That the connection request failed (for example, because
360 // it was refused by the peer.
362 // Use IsConnected() to distinguish between these two.
364 // So, in a brief, you should do one of the following things:
366 // For blocking Connect:
368 // bool success = client->Connect(addr, true);
370 // For nonblocking Connect:
372 // client->Connect(addr, false);
374 // bool waitmore = true;
375 // while (! client->WaitOnConnect(seconds, millis) && waitmore )
377 // // possibly give some feedback to the user,
378 // // update waitmore if needed.
380 // bool success = client->IsConnected();
382 // And that's all :-)
384 m_text
->AppendText(_("\nTrying to connect (timeout = 10 sec) ...\n"));
385 m_sock
->Connect(*addr
, false);
386 m_sock
->WaitOnConnect(10);
388 if (m_sock
->IsConnected())
389 m_text
->AppendText(_("Succeeded ! Connection established\n"));
393 m_text
->AppendText(_("Failed ! Unable to connect\n"));
394 wxMessageBox(_("Can't connect to the specified host"), _("Alert !"));
400 void MyFrame::OnTest1(wxCommandEvent
& WXUNUSED(event
))
402 // Disable socket menu entries (exception: Close Session)
406 m_text
->AppendText(_("\n=== Test 1 begins ===\n"));
408 // Tell the server which test we are running
409 unsigned char c
= 0xBE;
410 m_sock
->Write(&c
, 1);
412 // Send some data and read it back. We know the size of the
413 // buffer, so we can specify the exact number of bytes to be
414 // sent or received and use the wxSOCKET_WAITALL flag. Also,
415 // we have disabled menu entries which could interfere with
416 // the test, so we can safely avoid the wxSOCKET_BLOCK flag.
418 // First we send a byte with the length of the string, then
419 // we send the string itself (do NOT try to send any integral
420 // value larger than a byte "as is" across the network, or
421 // you might be in trouble! Ever heard about big and little
422 // endian computers?)
424 m_sock
->SetFlags(wxSOCKET_WAITALL
);
426 const wxChar
*buf1
= _T("Test string (less than 256 chars!)");
427 unsigned char len
= (unsigned char)((wxStrlen(buf1
) + 1)*sizeof(wxChar
));
428 wxChar
*buf2
= new wxChar
[wxStrlen(buf1
) + 1];
430 m_text
->AppendText(_("Sending a test buffer to the server ..."));
431 m_sock
->Write(&len
, 1);
432 m_sock
->Write(buf1
, len
);
433 m_text
->AppendText(m_sock
->Error() ? _("failed !\n") : _("done\n"));
435 m_text
->AppendText(_("Receiving the buffer back from server ..."));
436 m_sock
->Read(buf2
, len
);
437 m_text
->AppendText(m_sock
->Error() ? _("failed !\n") : _("done\n"));
439 m_text
->AppendText(_("Comparing the two buffers ..."));
440 if (memcmp(buf1
, buf2
, len
) != 0)
442 m_text
->AppendText(_("failed!\n"));
443 m_text
->AppendText(_("Test 1 failed !\n"));
447 m_text
->AppendText(_("done\n"));
448 m_text
->AppendText(_("Test 1 passed !\n"));
450 m_text
->AppendText(_("=== Test 1 ends ===\n"));
457 void MyFrame::OnTest2(wxCommandEvent
& WXUNUSED(event
))
463 // Disable socket menu entries (exception: Close Session)
467 m_text
->AppendText(_("\n=== Test 2 begins ===\n"));
469 // Tell the server which test we are running
470 unsigned char c
= 0xCE;
471 m_sock
->Write(&c
, 1);
473 // Here we use ReadMsg and WriteMsg to send messages with
474 // a header with size information. Also, the reception is
475 // event triggered, so we test input events as well.
477 // We need to set no flags here (ReadMsg and WriteMsg are
478 // not affected by flags)
480 m_sock
->SetFlags(wxSOCKET_WAITALL
);
482 wxString s
= wxGetTextFromUser(
483 _("Enter an arbitrary string to send to the server:"),
485 _("Yes I like wxWidgets!"));
488 len
= (wxStrlen(msg1
) + 1) * sizeof(wxChar
);
489 msg2
= new wxChar
[wxStrlen(msg1
) + 1];
491 m_text
->AppendText(_("Sending the string with WriteMsg ..."));
492 m_sock
->WriteMsg(msg1
, len
);
493 m_text
->AppendText(m_sock
->Error() ? _("failed !\n") : _("done\n"));
494 m_text
->AppendText(_("Waiting for an event (timeout = 2 sec)\n"));
496 // Wait until data available (will also return if the connection is lost)
497 m_sock
->WaitForRead(2);
499 if (m_sock
->IsData())
501 m_text
->AppendText(_("Reading the string back with ReadMsg ..."));
502 m_sock
->ReadMsg(msg2
, len
);
503 m_text
->AppendText(m_sock
->Error() ? _("failed !\n") : _("done\n"));
504 m_text
->AppendText(_("Comparing the two buffers ..."));
505 if (memcmp(msg1
, msg2
, len
) != 0)
507 m_text
->AppendText(_("failed!\n"));
508 m_text
->AppendText(_("Test 2 failed !\n"));
512 m_text
->AppendText(_("done\n"));
513 m_text
->AppendText(_("Test 2 passed !\n"));
517 m_text
->AppendText(_("Timeout ! Test 2 failed.\n"));
519 m_text
->AppendText(_("=== Test 2 ends ===\n"));
526 void MyFrame::OnTest3(wxCommandEvent
& WXUNUSED(event
))
532 // Disable socket menu entries (exception: Close Session)
536 m_text
->AppendText(_("\n=== Test 3 begins ===\n"));
538 // Tell the server which test we are running
539 unsigned char c
= 0xDE;
540 m_sock
->Write(&c
, 1);
542 // This test also is similar to the first one but it sends a
543 // large buffer so that wxSocket is actually forced to split
544 // it into pieces and take care of sending everything before
547 m_sock
->SetFlags(wxSOCKET_WAITALL
);
549 // Note that len is in kbytes here!
551 buf1
= new char[len
* 1024];
552 buf2
= new char[len
* 1024];
554 for (int i
= 0; i
< len
* 1024; i
++)
555 buf1
[i
] = (char)(i
% 256);
557 m_text
->AppendText(_("Sending a large buffer (32K) to the server ..."));
558 m_sock
->Write(&len
, 1);
559 m_sock
->Write(buf1
, len
* 1024);
560 m_text
->AppendText(m_sock
->Error() ? _("failed !\n") : _("done\n"));
562 m_text
->AppendText(_("Receiving the buffer back from server ..."));
563 m_sock
->Read(buf2
, len
* 1024);
564 m_text
->AppendText(m_sock
->Error() ? _("failed !\n") : _("done\n"));
566 m_text
->AppendText(_("Comparing the two buffers ..."));
567 if (memcmp(buf1
, buf2
, len
) != 0)
569 m_text
->AppendText(_("failed!\n"));
570 m_text
->AppendText(_("Test 3 failed !\n"));
574 m_text
->AppendText(_("done\n"));
575 m_text
->AppendText(_("Test 3 passed !\n"));
577 m_text
->AppendText(_("=== Test 3 ends ===\n"));
584 void MyFrame::OnCloseConnection(wxCommandEvent
& WXUNUSED(event
))
590 void MyFrame::OnDatagram(wxCommandEvent
& WXUNUSED(event
))
592 m_text
->AppendText(_("\n=== Datagram test begins ===\n"));
593 m_text
->AppendText(_("Sorry, not implemented\n"));
594 m_text
->AppendText(_("=== Datagram test ends ===\n"));
599 void MyFrame::OnTestURL(wxCommandEvent
& WXUNUSED(event
))
602 static wxString
s_urlname("http://www.google.com/");
603 wxString urlname
= wxGetTextFromUser
605 _("Enter an URL to get"),
609 if ( urlname
.empty() )
610 return; // cancelled by user
615 TestLogger
logtest("URL");
619 if ( url
.GetError() != wxURL_NOERR
)
621 wxLogError("Failed to parse URL \"%s\"", urlname
);
625 // Try to get the input stream (connects to the given URL)
626 wxLogMessage("Establishing connection to \"%s\"...", urlname
);
627 const std::auto_ptr
<wxInputStream
> data(url
.GetInputStream());
630 wxLogError("Failed to retrieve URL \"%s\"", urlname
);
634 // Print the contents type and file size
635 wxLogMessage("Contents type: %s\nFile size: %i\nStarting to download...",
636 url
.GetProtocol().GetContentType(),
640 wxStringOutputStream sout
;
641 if ( data
->Read(sout
).GetLastError() != wxSTREAM_EOF
)
642 wxLogError("Error reading the input stream.");
644 wxLogMessage("Text retrieved from URL \"%s\" follows:\n%s",
645 urlname
, sout
.GetString());
650 void MyFrame::OnSocketEvent(wxSocketEvent
& event
)
652 wxString s
= _("OnSocketEvent: ");
654 switch(event
.GetSocketEvent())
656 case wxSOCKET_INPUT
: s
.Append(_("wxSOCKET_INPUT\n")); break;
657 case wxSOCKET_LOST
: s
.Append(_("wxSOCKET_LOST\n")); break;
658 case wxSOCKET_CONNECTION
: s
.Append(_("wxSOCKET_CONNECTION\n")); break;
659 default : s
.Append(_("Unexpected event !\n")); break;
662 m_text
->AppendText(s
);
666 // convenience functions
668 void MyFrame::UpdateStatusBar()
672 if (!m_sock
->IsConnected())
674 s
.Printf(_("Not connected"));
684 m_sock
->GetPeer(addr
);
685 s
.Printf(_("%s : %d"), (addr
.Hostname()).c_str(), addr
.Service());
690 #endif // wxUSE_STATUSBAR
692 m_menuSocket
->Enable(CLIENT_OPEN
, !m_sock
->IsConnected() && !m_busy
);
694 m_menuSocket
->Enable(CLIENT_OPENIPV6
, !m_sock
->IsConnected() && !m_busy
);
696 m_menuSocket
->Enable(CLIENT_TEST1
, m_sock
->IsConnected() && !m_busy
);
697 m_menuSocket
->Enable(CLIENT_TEST2
, m_sock
->IsConnected() && !m_busy
);
698 m_menuSocket
->Enable(CLIENT_TEST3
, m_sock
->IsConnected() && !m_busy
);
699 m_menuSocket
->Enable(CLIENT_CLOSE
, m_sock
->IsConnected());