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\tCtrl-O"), _("Connect to server"));
220 m_menuSocket
->Append(CLIENT_OPENIPV6
, _("&Open session(IPv6)\tShift-Ctrl-O"), _("Connect to server(IPv6)"));
222 m_menuSocket
->AppendSeparator();
223 m_menuSocket
->Append(CLIENT_TEST1
, _("Test &1\tCtrl-F1"), _("Test basic functionality"));
224 m_menuSocket
->Append(CLIENT_TEST2
, _("Test &2\tCtrl-F2"), _("Test ReadMsg and WriteMsg"));
225 m_menuSocket
->Append(CLIENT_TEST3
, _("Test &3\tCtrl-F3"), _("Test large data transfer"));
226 m_menuSocket
->AppendSeparator();
227 m_menuSocket
->Append(CLIENT_CLOSE
, _("&Close session\tCtrl-Q"), _("Close connection"));
229 m_menuDatagramSocket
= new wxMenu();
230 m_menuDatagramSocket
->Append(CLIENT_DGRAM
, _("&Datagram test\tCtrl-D"), _("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
, _("&TCP"));
242 m_menuBar
->Append(m_menuDatagramSocket
, _("&UDP"));
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:"),
331 if ( hostname
.empty() )
334 addr
->Hostname(hostname
);
337 // we connect asynchronously and will get a wxSOCKET_CONNECTION event when
338 // the connection is really established
340 // if you want to make sure that connection is established right here you
341 // could call WaitOnConnect(timeout) instead
342 wxLogMessage("Trying to connect to %s:%d", hostname
, addr
->Service());
344 m_sock
->Connect(*addr
, false);
347 void MyFrame::OnTest1(wxCommandEvent
& WXUNUSED(event
))
349 // Disable socket menu entries (exception: Close Session)
353 m_text
->AppendText(_("\n=== Test 1 begins ===\n"));
355 // Tell the server which test we are running
356 unsigned char c
= 0xBE;
357 m_sock
->Write(&c
, 1);
359 // Send some data and read it back. We know the size of the
360 // buffer, so we can specify the exact number of bytes to be
361 // sent or received and use the wxSOCKET_WAITALL flag. Also,
362 // we have disabled menu entries which could interfere with
363 // the test, so we can safely avoid the wxSOCKET_BLOCK flag.
365 // First we send a byte with the length of the string, then
366 // we send the string itself (do NOT try to send any integral
367 // value larger than a byte "as is" across the network, or
368 // you might be in trouble! Ever heard about big and little
369 // endian computers?)
371 m_sock
->SetFlags(wxSOCKET_WAITALL
);
373 const char *buf1
= "Test string (less than 256 chars!)";
374 unsigned char len
= (unsigned char)(wxStrlen(buf1
) + 1);
375 wxCharBuffer
buf2(wxStrlen(buf1
));
377 m_text
->AppendText(_("Sending a test buffer to the server ..."));
378 m_sock
->Write(&len
, 1);
379 m_sock
->Write(buf1
, len
);
380 m_text
->AppendText(m_sock
->Error() ? _("failed !\n") : _("done\n"));
382 m_text
->AppendText(_("Receiving the buffer back from server ..."));
383 m_sock
->Read(buf2
.data(), len
);
384 m_text
->AppendText(m_sock
->Error() ? _("failed !\n") : _("done\n"));
386 m_text
->AppendText(_("Comparing the two buffers ..."));
387 if (memcmp(buf1
, buf2
, len
) != 0)
389 m_text
->AppendText(_("failed!\n"));
390 m_text
->AppendText(_("Test 1 failed !\n"));
394 m_text
->AppendText(_("done\n"));
395 m_text
->AppendText(_("Test 1 passed !\n"));
397 m_text
->AppendText(_("=== Test 1 ends ===\n"));
403 void MyFrame::OnTest2(wxCommandEvent
& WXUNUSED(event
))
405 // Disable socket menu entries (exception: Close Session)
409 m_text
->AppendText(_("\n=== Test 2 begins ===\n"));
411 // Tell the server which test we are running
412 unsigned char c
= 0xCE;
413 m_sock
->Write(&c
, 1);
415 // Here we use ReadMsg and WriteMsg to send messages with
416 // a header with size information. Also, the reception is
417 // event triggered, so we test input events as well.
419 // We need to set no flags here (ReadMsg and WriteMsg are
420 // not affected by flags)
422 m_sock
->SetFlags(wxSOCKET_WAITALL
);
424 wxString s
= wxGetTextFromUser(
425 _("Enter an arbitrary string to send to the server:"),
427 _("Yes I like wxWidgets!"));
429 const wxScopedCharBuffer
msg1(s
.utf8_str());
430 size_t len
= wxStrlen(msg1
) + 1;
431 wxCharBuffer
msg2(wxStrlen(msg1
));
433 m_text
->AppendText(_("Sending the string with WriteMsg ..."));
434 m_sock
->WriteMsg(msg1
, len
);
435 m_text
->AppendText(m_sock
->Error() ? _("failed !\n") : _("done\n"));
436 m_text
->AppendText(_("Waiting for an event (timeout = 2 sec)\n"));
438 // Wait until data available (will also return if the connection is lost)
439 m_sock
->WaitForRead(2);
441 if (m_sock
->IsData())
443 m_text
->AppendText(_("Reading the string back with ReadMsg ..."));
444 m_sock
->ReadMsg(msg2
.data(), len
);
445 m_text
->AppendText(m_sock
->Error() ? _("failed !\n") : _("done\n"));
446 m_text
->AppendText(_("Comparing the two buffers ..."));
447 if (memcmp(msg1
, msg2
, len
) != 0)
449 m_text
->AppendText(_("failed!\n"));
450 m_text
->AppendText(_("Test 2 failed !\n"));
454 m_text
->AppendText(_("done\n"));
455 m_text
->AppendText(_("Test 2 passed !\n"));
459 m_text
->AppendText(_("Timeout ! Test 2 failed.\n"));
461 m_text
->AppendText(_("=== Test 2 ends ===\n"));
467 void MyFrame::OnTest3(wxCommandEvent
& WXUNUSED(event
))
469 // Disable socket menu entries (exception: Close Session)
473 m_text
->AppendText(_("\n=== Test 3 begins ===\n"));
475 // Tell the server which test we are running
476 unsigned char c
= 0xDE;
477 m_sock
->Write(&c
, 1);
479 // This test also is similar to the first one but it sends a
480 // large buffer so that wxSocket is actually forced to split
481 // it into pieces and take care of sending everything before
484 m_sock
->SetFlags(wxSOCKET_WAITALL
);
486 // Note that len is in kbytes here!
487 const unsigned char len
= 32;
488 wxCharBuffer
buf1(len
* 1024),
491 for (size_t i
= 0; i
< len
* 1024; i
++)
492 buf1
.data()[i
] = (char)(i
% 256);
494 m_text
->AppendText(_("Sending a large buffer (32K) to the server ..."));
495 m_sock
->Write(&len
, 1);
496 m_sock
->Write(buf1
, len
* 1024);
497 m_text
->AppendText(m_sock
->Error() ? _("failed !\n") : _("done\n"));
499 m_text
->AppendText(_("Receiving the buffer back from server ..."));
500 m_sock
->Read(buf2
.data(), len
* 1024);
501 m_text
->AppendText(m_sock
->Error() ? _("failed !\n") : _("done\n"));
503 m_text
->AppendText(_("Comparing the two buffers ..."));
504 if (memcmp(buf1
, buf2
, len
) != 0)
506 m_text
->AppendText(_("failed!\n"));
507 m_text
->AppendText(_("Test 3 failed !\n"));
511 m_text
->AppendText(_("done\n"));
512 m_text
->AppendText(_("Test 3 passed !\n"));
514 m_text
->AppendText(_("=== Test 3 ends ===\n"));
520 void MyFrame::OnCloseConnection(wxCommandEvent
& WXUNUSED(event
))
526 void MyFrame::OnDatagram(wxCommandEvent
& WXUNUSED(event
))
528 wxString hostname
= wxGetTextFromUser
530 "Enter the address of the wxSocket demo server:",
534 if ( hostname
.empty() )
537 TestLogger
logtest("UDP");
539 wxIPV4address addrLocal
;
540 addrLocal
.Hostname();
541 wxDatagramSocket
sock(addrLocal
);
544 wxLogMessage("ERROR: failed to create UDP socket");
548 wxLogMessage("Created UDP socket at %s:%u",
549 addrLocal
.IPAddress(), addrLocal
.Service());
551 wxIPV4address addrPeer
;
552 addrPeer
.Hostname(hostname
);
553 addrPeer
.Service(3000);
555 wxLogMessage("Testing UDP with peer at %s:%u",
556 addrPeer
.IPAddress(), addrPeer
.Service());
558 char buf
[] = "Uryyb sebz pyvrag!";
559 if ( sock
.SendTo(addrPeer
, buf
, sizeof(buf
)).LastCount() != sizeof(buf
) )
561 wxLogMessage("ERROR: failed to send data");
565 if ( sock
.RecvFrom(addrPeer
, buf
, sizeof(buf
)).LastCount() != sizeof(buf
) )
567 wxLogMessage("ERROR: failed to receive data");
571 wxLogMessage("Received \"%s\" from %s:%u.",
572 wxString::From8BitData(buf
, sock
.LastCount()),
573 addrPeer
.IPAddress(), addrPeer
.Service());
578 void MyFrame::OnTestURL(wxCommandEvent
& WXUNUSED(event
))
581 static wxString
s_urlname("http://www.google.com/");
582 wxString urlname
= wxGetTextFromUser
584 _("Enter an URL to get"),
588 if ( urlname
.empty() )
589 return; // cancelled by user
594 TestLogger
logtest("URL");
598 if ( url
.GetError() != wxURL_NOERR
)
600 wxLogError("Failed to parse URL \"%s\"", urlname
);
604 // Try to get the input stream (connects to the given URL)
605 wxLogMessage("Establishing connection to \"%s\"...", urlname
);
606 const std::auto_ptr
<wxInputStream
> data(url
.GetInputStream());
609 wxLogError("Failed to retrieve URL \"%s\"", urlname
);
613 // Print the contents type and file size
614 wxLogMessage("Contents type: %s\nFile size: %i\nStarting to download...",
615 url
.GetProtocol().GetContentType(),
619 wxStringOutputStream sout
;
620 if ( data
->Read(sout
).GetLastError() != wxSTREAM_EOF
)
622 wxLogError("Error reading the input stream.");
625 wxLogMessage("Text retrieved from URL \"%s\" follows:\n%s",
626 urlname
, sout
.GetString());
631 void MyFrame::OnSocketEvent(wxSocketEvent
& event
)
633 switch ( event
.GetSocketEvent() )
636 wxLogMessage("Input available on the socket");
640 wxLogMessage("Socket connection was unexpectedly lost.");
644 case wxSOCKET_CONNECTION
:
645 wxLogMessage("... socket is now connected.");
650 wxLogMessage("Unknown socket event!!!");
655 // convenience functions
657 void MyFrame::UpdateStatusBar()
662 if (!m_sock
->IsConnected())
674 m_sock
->GetPeer(addr
);
675 s
.Printf("%s : %d", addr
.Hostname(), addr
.Service());
679 #endif // wxUSE_STATUSBAR
681 m_menuSocket
->Enable(CLIENT_OPEN
, !m_sock
->IsConnected() && !m_busy
);
683 m_menuSocket
->Enable(CLIENT_OPENIPV6
, !m_sock
->IsConnected() && !m_busy
);
685 m_menuSocket
->Enable(CLIENT_TEST1
, m_sock
->IsConnected() && !m_busy
);
686 m_menuSocket
->Enable(CLIENT_TEST2
, m_sock
->IsConnected() && !m_busy
);
687 m_menuSocket
->Enable(CLIENT_TEST3
, m_sock
->IsConnected() && !m_busy
);
688 m_menuSocket
->Enable(CLIENT_CLOSE
, m_sock
->IsConnected());