// headers
// --------------------------------------------------------------------------
-#ifdef __GNUG__
-# pragma implementation "client.cpp"
-# pragma interface "client.cpp"
-#endif
-
// For compilers that support precompilation, includes "wx/wx.h".
#include "wx/wxprec.h"
# pragma hdrstop
#endif
-// for all others, include the necessary headers
+// for all others, include the necessary headers
#ifndef WX_PRECOMP
# include "wx/wx.h"
#endif
#include "wx/socket.h"
#include "wx/url.h"
-#include "wx/wfstream.h"
+#include "wx/sstream.h"
+#include <memory>
// --------------------------------------------------------------------------
// resources
// --------------------------------------------------------------------------
// the application icon
-#if defined(__WXGTK__) || defined(__WXX11__) || defined(__WXMOTIF__) || defined(__WXMAC__)
-# include "mondrian.xpm"
-#endif
+#include "mondrian.xpm"
// --------------------------------------------------------------------------
// classes
void OnTest3(wxCommandEvent& event);
void OnCloseConnection(wxCommandEvent& event);
+#if wxUSE_URL
// event handlers for Protocols menu
void OnTestURL(wxCommandEvent& event);
+#endif
+#if wxUSE_IPV6
+ void OnOpenConnectionIPv6(wxCommandEvent& event);
+#endif
+
+ void OpenConnection(wxSockAddress::Family family);
// event handlers for DatagramSocket menu (stub)
void OnDatagram(wxCommandEvent& event);
wxMenuBar *m_menuBar;
bool m_busy;
- // any class wishing to process wxWindows events must use this macro
+ // any class wishing to process wxWidgets events must use this macro
DECLARE_EVENT_TABLE()
};
+// simple helper class to log start and end of each test
+class TestLogger
+{
+public:
+ TestLogger(const wxString& name) : m_name(name)
+ {
+ wxLogMessage("=== %s test begins ===", m_name);
+ }
+
+ ~TestLogger()
+ {
+ wxLogMessage("=== %s test ends ===", m_name);
+ }
+
+private:
+ const wxString m_name;
+};
+
// --------------------------------------------------------------------------
// constants
// --------------------------------------------------------------------------
enum
{
// menu items
- CLIENT_QUIT = 1000,
- CLIENT_ABOUT,
- CLIENT_OPEN,
+ CLIENT_QUIT = wxID_EXIT,
+ CLIENT_ABOUT = wxID_ABOUT,
+ CLIENT_OPEN = 100,
+#if wxUSE_IPV6
+ CLIENT_OPENIPV6,
+#endif
CLIENT_TEST1,
CLIENT_TEST2,
CLIENT_TEST3,
CLIENT_CLOSE,
+#if wxUSE_URL
CLIENT_TESTURL,
+#endif
CLIENT_DGRAM,
// id for socket
};
// --------------------------------------------------------------------------
-// event tables and other macros for wxWindows
+// event tables and other macros for wxWidgets
// --------------------------------------------------------------------------
BEGIN_EVENT_TABLE(MyFrame, wxFrame)
EVT_MENU(CLIENT_QUIT, MyFrame::OnQuit)
EVT_MENU(CLIENT_ABOUT, MyFrame::OnAbout)
EVT_MENU(CLIENT_OPEN, MyFrame::OnOpenConnection)
+#if wxUSE_IPV6
+ EVT_MENU(CLIENT_OPENIPV6, MyFrame::OnOpenConnectionIPv6)
+#endif
EVT_MENU(CLIENT_TEST1, MyFrame::OnTest1)
EVT_MENU(CLIENT_TEST2, MyFrame::OnTest2)
EVT_MENU(CLIENT_TEST3, MyFrame::OnTest3)
EVT_MENU(CLIENT_CLOSE, MyFrame::OnCloseConnection)
EVT_MENU(CLIENT_DGRAM, MyFrame::OnDatagram)
+#if wxUSE_URL
EVT_MENU(CLIENT_TESTURL, MyFrame::OnTestURL)
+#endif
EVT_SOCKET(SOCKET_ID, MyFrame::OnSocketEvent)
END_EVENT_TABLE()
bool MyApp::OnInit()
{
+ if ( !wxApp::OnInit() )
+ return false;
+
// Create the main application window
MyFrame *frame = new MyFrame();
// Show it and tell the application that it's our main window
- frame->Show(TRUE);
+ frame->Show(true);
SetTopWindow(frame);
// success
- return TRUE;
+ return true;
}
// --------------------------------------------------------------------------
// --------------------------------------------------------------------------
// frame constructor
-MyFrame::MyFrame() : wxFrame((wxFrame *)NULL, -1,
+MyFrame::MyFrame() : wxFrame((wxFrame *)NULL, wxID_ANY,
_("wxSocket demo: Client"),
wxDefaultPosition, wxSize(300, 200))
{
m_menuFile->Append(CLIENT_QUIT, _("E&xit\tAlt-X"), _("Quit client"));
m_menuSocket = new wxMenu();
- m_menuSocket->Append(CLIENT_OPEN, _("&Open session"), _("Connect to server"));
+ m_menuSocket->Append(CLIENT_OPEN, _("&Open session\tCtrl-O"), _("Connect to server"));
+#if wxUSE_IPV6
+ m_menuSocket->Append(CLIENT_OPENIPV6, _("&Open session(IPv6)\tShift-Ctrl-O"), _("Connect to server(IPv6)"));
+#endif
m_menuSocket->AppendSeparator();
- m_menuSocket->Append(CLIENT_TEST1, _("Test &1"), _("Test basic functionality"));
- m_menuSocket->Append(CLIENT_TEST2, _("Test &2"), _("Test ReadMsg and WriteMsg"));
- m_menuSocket->Append(CLIENT_TEST3, _("Test &3"), _("Test large data transfer"));
+ m_menuSocket->Append(CLIENT_TEST1, _("Test &1\tCtrl-F1"), _("Test basic functionality"));
+ m_menuSocket->Append(CLIENT_TEST2, _("Test &2\tCtrl-F2"), _("Test ReadMsg and WriteMsg"));
+ m_menuSocket->Append(CLIENT_TEST3, _("Test &3\tCtrl-F3"), _("Test large data transfer"));
m_menuSocket->AppendSeparator();
- m_menuSocket->Append(CLIENT_CLOSE, _("&Close session"), _("Close connection"));
+ m_menuSocket->Append(CLIENT_CLOSE, _("&Close session\tCtrl-Q"), _("Close connection"));
m_menuDatagramSocket = new wxMenu();
m_menuDatagramSocket->Append(CLIENT_DGRAM, _("Send Datagram"), _("Test UDP sockets"));
+#if wxUSE_URL
m_menuProtocols = new wxMenu();
- m_menuProtocols->Append(CLIENT_TESTURL, _("Test URL"), _("Get data from the specified URL"));
+ m_menuProtocols->Append(CLIENT_TESTURL, _("Test URL\tCtrl-U"),
+ _("Get data from the specified URL"));
+#endif
// Append menus to the menubar
m_menuBar = new wxMenuBar();
m_menuBar->Append(m_menuFile, _("&File"));
m_menuBar->Append(m_menuSocket, _("&SocketClient"));
m_menuBar->Append(m_menuDatagramSocket, _("&DatagramSocket"));
+#if wxUSE_URL
m_menuBar->Append(m_menuProtocols, _("&Protocols"));
+#endif
SetMenuBar(m_menuBar);
+#if wxUSE_STATUSBAR
// Status bar
CreateStatusBar(2);
+#endif // wxUSE_STATUSBAR
// Make a textctrl for logging
- m_text = new wxTextCtrl(this, -1,
+ m_text = new wxTextCtrl(this, wxID_ANY,
_("Welcome to wxSocket demo: Client\nClient ready\n"),
wxDefaultPosition, wxDefaultSize,
wxTE_MULTILINE | wxTE_READONLY);
+ delete wxLog::SetActiveTarget(new wxLogTextCtrl(m_text));
// Create the socket
m_sock = new wxSocketClient();
m_sock->SetNotify(wxSOCKET_CONNECTION_FLAG |
wxSOCKET_INPUT_FLAG |
wxSOCKET_LOST_FLAG);
- m_sock->Notify(TRUE);
+ m_sock->Notify(true);
- m_busy = FALSE;
+ m_busy = false;
UpdateStatusBar();
}
void MyFrame::OnQuit(wxCommandEvent& WXUNUSED(event))
{
- // TRUE is to force the frame to close
- Close(TRUE);
+ // true is to force the frame to close
+ Close(true);
}
void MyFrame::OnAbout(wxCommandEvent& WXUNUSED(event))
void MyFrame::OnOpenConnection(wxCommandEvent& WXUNUSED(event))
{
- wxIPV4address addr;
+ OpenConnection(wxSockAddress::IPV4);
+}
+#if wxUSE_IPV6
+void MyFrame::OnOpenConnectionIPv6(wxCommandEvent& WXUNUSED(event))
+{
+ OpenConnection(wxSockAddress::IPV6);
+}
+#endif // wxUSE_IPV6
- m_menuSocket->Enable(CLIENT_OPEN, FALSE);
- m_menuSocket->Enable(CLIENT_CLOSE, FALSE);
+void MyFrame::OpenConnection(wxSockAddress::Family family)
+{
+ wxUnusedVar(family); // unused in !wxUSE_IPV6 case
+
+ wxIPaddress * addr;
+ wxIPV4address addr4;
+#if wxUSE_IPV6
+ wxIPV6address addr6;
+ if ( family == wxSockAddress::IPV6 )
+ addr = &addr6;
+ else
+#endif
+ addr = &addr4;
+
+ m_menuSocket->Enable(CLIENT_OPEN, false);
+#if wxUSE_IPV6
+ m_menuSocket->Enable(CLIENT_OPENIPV6, false);
+#endif
+ m_menuSocket->Enable(CLIENT_CLOSE, false);
// Ask user for server address
wxString hostname = wxGetTextFromUser(
_("Connect ..."),
_("localhost"));
- addr.Hostname(hostname);
- addr.Service(3000);
+ addr->Hostname(hostname);
+ addr->Service(3000);
// Mini-tutorial for Connect() :-)
// ---------------------------
// There are two ways to use Connect(): blocking and non-blocking,
// depending on the value passed as the 'wait' (2nd) parameter.
//
- // Connect(addr, TRUE) will wait until the connection completes,
- // returning TRUE on success and FALSE on failure. This call blocks
+ // Connect(addr, true) will wait until the connection completes,
+ // returning true on success and false on failure. This call blocks
// the GUI (this might be changed in future releases to honour the
// wxSOCKET_BLOCK flag).
//
- // Connect(addr, FALSE) will issue a nonblocking connection request
- // and return immediately. If the return value is TRUE, then the
- // connection has been already succesfully established. If it is
- // FALSE, you must wait for the request to complete, either with
+ // Connect(addr, false) will issue a nonblocking connection request
+ // and return immediately. If the return value is true, then the
+ // connection has been already successfully established. If it is
+ // false, you must wait for the request to complete, either with
// WaitOnConnect() or by watching wxSOCKET_CONNECTION / LOST
// events (please read the documentation).
//
// WaitOnConnect() itself never blocks the GUI (this might change
// in the future to honour the wxSOCKET_BLOCK flag). This call will
- // return FALSE on timeout, or TRUE if the connection request
+ // return false on timeout, or true if the connection request
// completes, which in turn might mean:
//
// a) That the connection was successfully established
//
// For blocking Connect:
//
- // bool success = client->Connect(addr, TRUE);
+ // bool success = client->Connect(addr, true);
//
// For nonblocking Connect:
//
- // client->Connect(addr, FALSE);
+ // client->Connect(addr, false);
//
- // bool waitmore = TRUE;
+ // bool waitmore = true;
// while (! client->WaitOnConnect(seconds, millis) && waitmore )
// {
// // possibly give some feedback to the user,
// // update waitmore if needed.
// }
// bool success = client->IsConnected();
- //
+ //
// And that's all :-)
m_text->AppendText(_("\nTrying to connect (timeout = 10 sec) ...\n"));
- m_sock->Connect(addr, FALSE);
+ m_sock->Connect(*addr, false);
m_sock->WaitOnConnect(10);
if (m_sock->IsConnected())
m_text->AppendText(_("Failed ! Unable to connect\n"));
wxMessageBox(_("Can't connect to the specified host"), _("Alert !"));
}
-
+
UpdateStatusBar();
}
void MyFrame::OnTest1(wxCommandEvent& WXUNUSED(event))
{
- const wxChar *buf1;
- wxChar *buf2;
- unsigned char len;
-
// Disable socket menu entries (exception: Close Session)
- m_busy = TRUE;
+ m_busy = true;
UpdateStatusBar();
m_text->AppendText(_("\n=== Test 1 begins ===\n"));
m_sock->SetFlags(wxSOCKET_WAITALL);
- buf1 = _("Test string (less than 256 chars!)");
- len = (wxStrlen(buf1) + 1) * sizeof(wxChar);
- buf2 = new wxChar[wxStrlen(buf1) + 1];
+ const char *buf1 = "Test string (less than 256 chars!)";
+ unsigned char len = (unsigned char)(wxStrlen(buf1) + 1);
+ wxCharBuffer buf2(wxStrlen(buf1));
m_text->AppendText(_("Sending a test buffer to the server ..."));
m_sock->Write(&len, 1);
m_text->AppendText(m_sock->Error() ? _("failed !\n") : _("done\n"));
m_text->AppendText(_("Receiving the buffer back from server ..."));
- m_sock->Read(buf2, len);
+ m_sock->Read(buf2.data(), len);
m_text->AppendText(m_sock->Error() ? _("failed !\n") : _("done\n"));
m_text->AppendText(_("Comparing the two buffers ..."));
}
m_text->AppendText(_("=== Test 1 ends ===\n"));
- delete[] buf2;
- m_busy = FALSE;
+ m_busy = false;
UpdateStatusBar();
}
void MyFrame::OnTest2(wxCommandEvent& WXUNUSED(event))
{
- const wxChar *msg1;
- wxChar *msg2;
- size_t len;
-
// Disable socket menu entries (exception: Close Session)
- m_busy = TRUE;
+ m_busy = true;
UpdateStatusBar();
m_text->AppendText(_("\n=== Test 2 begins ===\n"));
wxString s = wxGetTextFromUser(
_("Enter an arbitrary string to send to the server:"),
_("Test 2 ..."),
- _("Yes I like wxWindows!"));
+ _("Yes I like wxWidgets!"));
- msg1 = s.c_str();
- len = (wxStrlen(msg1) + 1) * sizeof(wxChar);
- msg2 = new wxChar[wxStrlen(msg1) + 1];
+ const wxUTF8Buf msg1(s.utf8_str());
+ size_t len = wxStrlen(msg1) + 1;
+ wxCharBuffer msg2(wxStrlen(msg1));
m_text->AppendText(_("Sending the string with WriteMsg ..."));
m_sock->WriteMsg(msg1, len);
if (m_sock->IsData())
{
m_text->AppendText(_("Reading the string back with ReadMsg ..."));
- m_sock->ReadMsg(msg2, len);
+ m_sock->ReadMsg(msg2.data(), len);
m_text->AppendText(m_sock->Error() ? _("failed !\n") : _("done\n"));
m_text->AppendText(_("Comparing the two buffers ..."));
if (memcmp(msg1, msg2, len) != 0)
{
- m_text->AppendText(_("failed!\n"));
+ m_text->AppendText(_("failed!\n"));
m_text->AppendText(_("Test 2 failed !\n"));
}
else
m_text->AppendText(_("=== Test 2 ends ===\n"));
- delete[] msg2;
- m_busy = FALSE;
+ m_busy = false;
UpdateStatusBar();
}
void MyFrame::OnTest3(wxCommandEvent& WXUNUSED(event))
{
- char *buf1;
- char *buf2;
- unsigned char len;
-
// Disable socket menu entries (exception: Close Session)
- m_busy = TRUE;
+ m_busy = true;
UpdateStatusBar();
m_text->AppendText(_("\n=== Test 3 begins ===\n"));
m_sock->SetFlags(wxSOCKET_WAITALL);
// Note that len is in kbytes here!
- len = 32;
- buf1 = new char[len * 1024];
- buf2 = new char[len * 1024];
+ const unsigned char len = 32;
+ wxCharBuffer buf1(len * 1024),
+ buf2(len * 1024);
- for (int i = 0; i < len * 1024; i ++)
- buf1[i] = (char)(i % 256);
+ for (size_t i = 0; i < len * 1024; i ++)
+ buf1.data()[i] = (char)(i % 256);
m_text->AppendText(_("Sending a large buffer (32K) to the server ..."));
m_sock->Write(&len, 1);
m_text->AppendText(m_sock->Error() ? _("failed !\n") : _("done\n"));
m_text->AppendText(_("Receiving the buffer back from server ..."));
- m_sock->Read(buf2, len * 1024);
+ m_sock->Read(buf2.data(), len * 1024);
m_text->AppendText(m_sock->Error() ? _("failed !\n") : _("done\n"));
m_text->AppendText(_("Comparing the two buffers ..."));
}
m_text->AppendText(_("=== Test 3 ends ===\n"));
- delete[] buf2;
- m_busy = FALSE;
+ m_busy = false;
UpdateStatusBar();
}
m_text->AppendText(_("=== Datagram test ends ===\n"));
}
+#if wxUSE_URL
+
void MyFrame::OnTestURL(wxCommandEvent& WXUNUSED(event))
{
- // Note that we are creating a new socket here, so this
- // won't mess with the client/server demo.
-
- // Ask for the URL
- m_text->AppendText(_("\n=== URL test begins ===\n"));
- wxString urlname = wxGetTextFromUser(_("Enter an URL to get"),
- _("URL:"),
- _T("http://localhost"));
-
- // Parse the URL
- wxURL url(urlname);
- if (url.GetError() != wxURL_NOERR)
- {
- m_text->AppendText(_("Error: couldn't parse URL\n"));
- m_text->AppendText(_("=== URL test ends ===\n"));
- return;
- }
+ // Ask for the URL
+ static wxString s_urlname("http://www.google.com/");
+ wxString urlname = wxGetTextFromUser
+ (
+ _("Enter an URL to get"),
+ _("URL:"),
+ s_urlname
+ );
+ if ( urlname.empty() )
+ return; // cancelled by user
- // Try to get the input stream (connects to the given URL)
- m_text->AppendText(_("Trying to establish connection...\n"));
- wxYield();
- wxInputStream *data = url.GetInputStream();
- if (!data)
- {
- m_text->AppendText(_("Error: couldn't read from URL\n"));
- m_text->AppendText(_("=== URL test ends ===\n"));
- return;
- }
+ s_urlname = urlname;
- // Print the contents type and file size
- wxString s;
- s.Printf(_("Contents type: %s\nFile size: %i\nStarting to download...\n"),
- url.GetProtocol().GetContentType().c_str(),
- data->GetSize());
- m_text->AppendText(s);
- wxYield();
- // Get the data
- wxFileOutputStream sout(wxString("test.url"));
- if (!sout.Ok())
- {
- m_text->AppendText(_("Error: couldn't open file for output\n"));
- m_text->AppendText(_("=== URL test ends ===\n"));
- return;
- }
+ TestLogger logtest("URL");
+
+ // Parse the URL
+ wxURL url(urlname);
+ if ( url.GetError() != wxURL_NOERR )
+ {
+ wxLogError("Failed to parse URL \"%s\"", urlname);
+ return;
+ }
- data->Read(sout);
- m_text->AppendText(_("Results written to file: test.url\n"));
- m_text->AppendText(_("Done.\n"));
- m_text->AppendText(_("=== URL test ends ===\n"));
+ // Try to get the input stream (connects to the given URL)
+ wxLogMessage("Establishing connection to \"%s\"...", urlname);
+ const std::auto_ptr<wxInputStream> data(url.GetInputStream());
+ if ( !data.get() )
+ {
+ wxLogError("Failed to retrieve URL \"%s\"", urlname);
+ return;
+ }
+
+ // Print the contents type and file size
+ wxLogMessage("Contents type: %s\nFile size: %i\nStarting to download...",
+ url.GetProtocol().GetContentType(),
+ data->GetSize());
+
+ // Get the data
+ wxStringOutputStream sout;
+ if ( data->Read(sout).GetLastError() != wxSTREAM_EOF )
+ wxLogError("Error reading the input stream.");
- delete data;
+ wxLogMessage("Text retrieved from URL \"%s\" follows:\n%s",
+ urlname, sout.GetString());
}
+#endif // wxUSE_URL
+
void MyFrame::OnSocketEvent(wxSocketEvent& event)
{
wxString s = _("OnSocketEvent: ");
}
else
{
+#if wxUSE_IPV6
+ wxIPV6address addr;
+#else
wxIPV4address addr;
+#endif
m_sock->GetPeer(addr);
s.Printf(_("%s : %d"), (addr.Hostname()).c_str(), addr.Service());
}
+#if wxUSE_STATUSBAR
SetStatusText(s, 1);
+#endif // wxUSE_STATUSBAR
m_menuSocket->Enable(CLIENT_OPEN, !m_sock->IsConnected() && !m_busy);
+#if wxUSE_IPV6
+ m_menuSocket->Enable(CLIENT_OPENIPV6, !m_sock->IsConnected() && !m_busy);
+#endif
m_menuSocket->Enable(CLIENT_TEST1, m_sock->IsConnected() && !m_busy);
m_menuSocket->Enable(CLIENT_TEST2, m_sock->IsConnected() && !m_busy);
m_menuSocket->Enable(CLIENT_TEST3, m_sock->IsConnected() && !m_busy);