X-Git-Url: https://git.saurik.com/wxWidgets.git/blobdiff_plain/856bc99a57f7a01952e9621e4b014420abcaa7ca..c437b3f4e47b74715a2f2385d4862972babd7802:/samples/sockets/server.cpp diff --git a/samples/sockets/server.cpp b/samples/sockets/server.cpp index 441d5721ba..b7063e93d0 100644 --- a/samples/sockets/server.cpp +++ b/samples/sockets/server.cpp @@ -17,11 +17,6 @@ // headers // -------------------------------------------------------------------------- -#ifdef __GNUG__ -# pragma implementation "server.cpp" -# pragma interface "server.cpp" -#endif - // For compilers that support precompilation, includes "wx/wx.h". #include "wx/wxprec.h" @@ -29,7 +24,7 @@ # 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 @@ -41,9 +36,7 @@ // -------------------------------------------------------------------------- // the application icon -#if defined(__WXGTK__) || defined(__WXMOTIF__) -# include "mondrian.xpm" -#endif +#include "mondrian.xpm" // -------------------------------------------------------------------------- // classes @@ -84,10 +77,28 @@ private: bool m_busy; int m_numClients; - // 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 begins ===", m_name); + } + + ~TestLogger() + { + wxLogMessage("=== %s ends ===", m_name); + } + +private: + const wxString m_name; +}; + // -------------------------------------------------------------------------- // constants // -------------------------------------------------------------------------- @@ -96,16 +107,16 @@ private: enum { // menu items - SERVER_QUIT = 1000, - SERVER_ABOUT, + SERVER_QUIT = wxID_EXIT, + SERVER_ABOUT = wxID_ABOUT, // id for sockets - SERVER_ID, + SERVER_ID = 100, SOCKET_ID }; // -------------------------------------------------------------------------- -// event tables and other macros for wxWindows +// event tables and other macros for wxWidgets // -------------------------------------------------------------------------- BEGIN_EVENT_TABLE(MyFrame, wxFrame) @@ -128,15 +139,18 @@ IMPLEMENT_APP(MyApp) 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; } // -------------------------------------------------------------------------- @@ -145,7 +159,7 @@ bool MyApp::OnInit() // frame constructor -MyFrame::MyFrame() : wxFrame((wxFrame *)NULL, -1, +MyFrame::MyFrame() : wxFrame((wxFrame *)NULL, wxID_ANY, _("wxSocket demo: Server"), wxDefaultPosition, wxSize(300, 200)) { @@ -163,17 +177,24 @@ MyFrame::MyFrame() : wxFrame((wxFrame *)NULL, -1, m_menuBar->Append(m_menuFile, _("&File")); 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: Server\n"), wxDefaultPosition, wxDefaultSize, wxTE_MULTILINE | wxTE_READONLY); + delete wxLog::SetActiveTarget(new wxLogTextCtrl(m_text)); // Create the address - defaults to localhost:0 initially +#if wxUSE_IPV6 + wxIPV6address addr; +#else wxIPV4address addr; +#endif addr.Service(3000); // Create the socket @@ -182,20 +203,20 @@ MyFrame::MyFrame() : wxFrame((wxFrame *)NULL, -1, // We use Ok() here to see if the server is really listening if (! m_server->Ok()) { - m_text->AppendText(_("Could not listen at the specified port !\n\n")); + wxLogMessage("Could not listen at the specified port !"); return; } else { - m_text->AppendText(_("Server listening.\n\n")); + wxLogMessage("Server listening."); } // Setup the event handler and subscribe to connection events m_server->SetEventHandler(*this, SERVER_ID); m_server->SetNotify(wxSOCKET_CONNECTION_FLAG); - m_server->Notify(TRUE); + m_server->Notify(true); - m_busy = FALSE; + m_busy = false; m_numClients = 0; UpdateStatusBar(); } @@ -210,24 +231,20 @@ MyFrame::~MyFrame() 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)) { - wxMessageBox(_("wxSocket demo: Server\n" - "(c) 1999 Guillermo Rodriguez Garcia\n"), + wxMessageBox(_("wxSocket demo: Server\n(c) 1999 Guillermo Rodriguez Garcia\n"), _("About Server"), wxOK | wxICON_INFORMATION, this); } void MyFrame::Test1(wxSocketBase *sock) { - unsigned char len; - char *buf; - - m_text->AppendText(_("Test 1 begins\n")); + TestLogger logtest("Test 1"); // Receive data from socket and send it back. We will first // get a byte with the buffer size, so we can specify the @@ -237,63 +254,63 @@ void MyFrame::Test1(wxSocketBase *sock) sock->SetFlags(wxSOCKET_WAITALL); + // Read the size + unsigned char len; sock->Read(&len, 1); + wxCharBuffer buf(len); - buf = new char[len]; - sock->Read(buf, len); - sock->Write(buf, len); - delete[] buf; + // Read the data + sock->Read(buf.data(), len); + wxLogMessage("Got the data, sending it back"); - m_text->AppendText(_("Test 1 ends\n")); + // Write it back + sock->Write(buf, len); } void MyFrame::Test2(wxSocketBase *sock) { -#define MAX_MSG_SIZE 10000 + char buf[4096]; - wxString s; - char *buf = new char[MAX_MSG_SIZE]; - wxUint32 len; - - m_text->AppendText(_("Test 2 begins\n")); + TestLogger logtest("Test 2"); // We don't need to set flags because ReadMsg and WriteMsg // are not affected by them anyway. - len = sock->ReadMsg(buf, MAX_MSG_SIZE).LastCount(); + // Read the message + wxUint32 len = sock->ReadMsg(buf, sizeof(buf)).LastCount(); + if ( !len ) + { + wxLogError("Failed to read message."); + return; + } - s.Printf(_("Client says: %s\n"), buf); - m_text->AppendText(s); - m_text->AppendText(_("Sending the data back\n")); + wxLogMessage("Got \"%s\" from client.", wxString::FromUTF8(buf, len)); + wxLogMessage("Sending the data back"); + // Write it back sock->WriteMsg(buf, len); - delete[] buf; - - m_text->AppendText(_("Test 2 ends\n")); - -#undef MAX_MSG_SIZE } void MyFrame::Test3(wxSocketBase *sock) { - unsigned char len; - char *buf; - - m_text->AppendText(_("Test 3 begins\n")); + TestLogger logtest("Test 3"); - // This test is similar to the first one, but the len is + // This test is similar to the first one, but the len is // expressed in kbytes - this tests large data transfers. sock->SetFlags(wxSOCKET_WAITALL); + // Read the size + unsigned char len; sock->Read(&len, 1); - - buf = new char[len * 1024]; - sock->Read(buf, len * 1024); - sock->Write(buf, len * 1024); - delete[] buf; + wxCharBuffer buf(len*1024); - m_text->AppendText(_("Test 3 ends\n")); + // Read the data + sock->Read(buf.data(), len * 1024); + wxLogMessage("Got the data, sending it back"); + + // Write it back + sock->Write(buf, len * 1024); } void MyFrame::OnServerEvent(wxSocketEvent& event) @@ -301,7 +318,7 @@ void MyFrame::OnServerEvent(wxSocketEvent& event) wxString s = _("OnServerEvent: "); wxSocketBase *sock; - switch(event.SocketEvent()) + switch(event.GetSocketEvent()) { case wxSOCKET_CONNECTION : s.Append(_("wxSOCKET_CONNECTION\n")); break; default : s.Append(_("Unexpected event !\n")); break; @@ -310,25 +327,25 @@ void MyFrame::OnServerEvent(wxSocketEvent& event) m_text->AppendText(s); // Accept new connection if there is one in the pending - // connections queue, else exit. We use Accept(FALSE) for + // connections queue, else exit. We use Accept(false) for // non-blocking accept (although if we got here, there // should ALWAYS be a pending connection). - sock = m_server->Accept(FALSE); + sock = m_server->Accept(false); if (sock) { - m_text->AppendText(_("New client connection accepted\n")); + wxLogMessage("New client connection accepted"); } else { - m_text->AppendText(_("Error: couldn't accept a new connection")); + wxLogMessage("Error: couldn't accept a new connection"); return; } sock->SetEventHandler(*this, SOCKET_ID); sock->SetNotify(wxSOCKET_INPUT_FLAG | wxSOCKET_LOST_FLAG); - sock->Notify(TRUE); + sock->Notify(true); m_numClients++; UpdateStatusBar(); @@ -336,21 +353,21 @@ void MyFrame::OnServerEvent(wxSocketEvent& event) void MyFrame::OnSocketEvent(wxSocketEvent& event) { - wxSocketBase *sock = event.Socket(); wxString s = _("OnSocketEvent: "); + wxSocketBase *sock = event.GetSocket(); - // We first print a msg - switch(event.SocketEvent()) + // First, print a message + switch(event.GetSocketEvent()) { - case wxSOCKET_INPUT: s.Append(_("wxSOCKET_INPUT\n")); break; - case wxSOCKET_LOST: s.Append(_("wxSOCKET_LOST\n")); break; - default: s.Append(_("unexpected event !\n")); + case wxSOCKET_INPUT : s.Append(_("wxSOCKET_INPUT\n")); break; + case wxSOCKET_LOST : s.Append(_("wxSOCKET_LOST\n")); break; + default : s.Append(_("Unexpected event !\n")); break; } m_text->AppendText(s); // Now we process the event - switch(event.SocketEvent()) + switch(event.GetSocketEvent()) { case wxSOCKET_INPUT: { @@ -360,14 +377,15 @@ void MyFrame::OnSocketEvent(wxSocketEvent& event) // Which test are we going to run? unsigned char c; - sock->Read(&c ,1); + sock->Read(&c, 1); switch (c) { case 0xBE: Test1(sock); break; case 0xCE: Test2(sock); break; case 0xDE: Test3(sock); break; - default: s.Append(_("Unknown test id received from client\n")); + default: + wxLogMessage("Unknown test id received from client"); } // Enable input events again. @@ -385,11 +403,9 @@ void MyFrame::OnSocketEvent(wxSocketEvent& event) // has been deleted. Also, we might be doing some other thing with // the socket at the same time; for example, we might be in the // middle of a test or something. Destroy() takes care of all - // this for us. The only case where delete should be used instead - // is when the event handler itself is also being destroyed; for - // example, see the frame dtor above. + // this for us. - m_text->AppendText(_("Deleting socket.\n")); + wxLogMessage("Deleting socket."); sock->Destroy(); break; } @@ -403,7 +419,9 @@ void MyFrame::OnSocketEvent(wxSocketEvent& event) void MyFrame::UpdateStatusBar() { +#if wxUSE_STATUSBAR wxString s; s.Printf(_("%d clients connected"), m_numClients); SetStatusText(s, 1); +#endif // wxUSE_STATUSBAR }