X-Git-Url: https://git.saurik.com/wxWidgets.git/blobdiff_plain/6097c3a28a968e4acbdbee90072cc3f4ee65ab74..2275c7326890f3322d1f140b5687a75810bc81af:/samples/sockets/server.cpp?ds=sidebyside diff --git a/samples/sockets/server.cpp b/samples/sockets/server.cpp index f24529f16c..cd8fe59d5b 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 @@ -78,14 +71,13 @@ public: private: wxSocketServer *m_server; - wxPanel *m_panel; wxTextCtrl *m_text; wxMenu *m_menuFile; wxMenuBar *m_menuBar; 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() }; @@ -97,16 +89,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) @@ -119,10 +111,6 @@ END_EVENT_TABLE() IMPLEMENT_APP(MyApp) -// To append sockets for delayed deletion [XXX: this should be removed] -extern WXDLLEXPORT wxList wxPendingDelete; - - // ========================================================================== // implementation // ========================================================================== @@ -133,15 +121,18 @@ extern WXDLLEXPORT wxList wxPendingDelete; 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; + // Success + return true; } // -------------------------------------------------------------------------- @@ -149,8 +140,9 @@ bool MyApp::OnInit() // -------------------------------------------------------------------------- // frame constructor -MyFrame::MyFrame() : wxFrame((wxFrame *)NULL, -1, - _T("wxSocket demo: Server"), + +MyFrame::MyFrame() : wxFrame((wxFrame *)NULL, wxID_ANY, + _("wxSocket demo: Server"), wxDefaultPosition, wxSize(300, 200)) { // Give the frame an icon @@ -158,47 +150,61 @@ MyFrame::MyFrame() : wxFrame((wxFrame *)NULL, -1, // Make menus m_menuFile = new wxMenu(); - m_menuFile->Append(SERVER_ABOUT, _T("&About...\tCtrl-A"), _T("Show about dialog")); + m_menuFile->Append(SERVER_ABOUT, _("&About...\tCtrl-A"), _("Show about dialog")); m_menuFile->AppendSeparator(); - m_menuFile->Append(SERVER_QUIT, _T("E&xit\tAlt-X"), _T("Quit server")); + m_menuFile->Append(SERVER_QUIT, _("E&xit\tAlt-X"), _("Quit server")); // Append menus to the menubar m_menuBar = new wxMenuBar(); - m_menuBar->Append(m_menuFile, _T("&File")); + m_menuBar->Append(m_menuFile, _("&File")); SetMenuBar(m_menuBar); +#if wxUSE_STATUSBAR // Status bar CreateStatusBar(2); +#endif // wxUSE_STATUSBAR - // Make a panel with a textctrl in it - m_panel = new wxPanel(this, -1, wxPoint(0, 0), GetClientSize()); - m_text = new wxTextCtrl(m_panel, -1, - _T("Welcome to wxSocket demo: Server\n"), - wxPoint(0, 0), m_panel->GetClientSize(), + // Make a textctrl for logging + m_text = new wxTextCtrl(this, wxID_ANY, + _("Welcome to wxSocket demo: Server\n"), + wxDefaultPosition, wxDefaultSize, wxTE_MULTILINE | wxTE_READONLY); - // Create the socket - defaults to localhost:0 + // Create the address - defaults to localhost:0 initially +#if wxUSE_IPV6 + wxIPV6address addr; +#else wxIPV4address addr; +#endif addr.Service(3000); + // Create the socket m_server = new wxSocketServer(addr); - m_server->SetEventHandler(*this, SERVER_ID); - m_server->SetNotify(wxSOCKET_CONNECTION_FLAG); - m_server->Notify(TRUE); // We use Ok() here to see if the server is really listening - if (m_server->Ok()) - m_text->AppendText(_T("Server listening.\n\n")); + if (! m_server->Ok()) + { + m_text->AppendText(_("Could not listen at the specified port !\n\n")); + return; + } else - m_text->AppendText(_T("Could not listen at the specified port !\n\n")); + { + m_text->AppendText(_("Server listening.\n\n")); + } - m_busy = FALSE; + // 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_busy = false; m_numClients = 0; UpdateStatusBar(); } MyFrame::~MyFrame() { + // No delayed deletion here, as the frame is dying anyway delete m_server; } @@ -206,15 +212,14 @@ 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(_T("wxSocket demo: Server\n") - _T("(c) 1999 Guillermo Rodriguez Garcia\n"), - _T("About Server"), + wxMessageBox(_("wxSocket demo: Server\n(c) 1999 Guillermo Rodriguez Garcia\n"), + _("About Server"), wxOK | wxICON_INFORMATION, this); } @@ -223,24 +228,29 @@ void MyFrame::Test1(wxSocketBase *sock) unsigned char len; char *buf; - m_text->AppendText(_T("Test 1 begins\n")); + m_text->AppendText(_("Test 1 begins\n")); // Receive data from socket and send it back. We will first // get a byte with the buffer size, so we can specify the // exact size and use the wxSOCKET_WAITALL flag. Also, we // disabled input events so we won't have unwanted reentrance. // This way we can avoid the infamous wxSOCKET_BLOCK flag. - // - sock->SetFlags(wxSOCKET_WAITALL); - sock->Read((char *)&len, 1); + sock->SetFlags(wxSOCKET_WAITALL); + // Read the size + sock->Read(&len, 1); buf = new char[len]; + + // Read the data sock->Read(buf, len); + m_text->AppendText(_("Got the data, sending it back\n")); + + // Write it back sock->Write(buf, len); delete[] buf; - m_text->AppendText(_T("Test 1 ends\n")); + m_text->AppendText(_("Test 1 ends\n\n")); } void MyFrame::Test2(wxSocketBase *sock) @@ -248,23 +258,25 @@ void MyFrame::Test2(wxSocketBase *sock) #define MAX_MSG_SIZE 10000 wxString s; - char *buf = new char[MAX_MSG_SIZE]; + wxChar *buf = new wxChar[MAX_MSG_SIZE]; wxUint32 len; - m_text->AppendText(_T("Test 2 begins\n")); + m_text->AppendText(_("Test 2 begins\n")); // 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(); - s.Printf(_T("Client says: %s\n"), buf); + // Read the message + len = sock->ReadMsg(buf, MAX_MSG_SIZE * sizeof(wxChar)).LastCount(); + s.Printf(_("Client says: %s\n"), buf); m_text->AppendText(s); + m_text->AppendText(_("Sending the data back\n")); + // Write it back sock->WriteMsg(buf, len); delete[] buf; - m_text->AppendText(_T("Test 2 ends\n")); + m_text->AppendText(_("Test 2 ends\n\n")); #undef MAX_MSG_SIZE } @@ -274,55 +286,61 @@ void MyFrame::Test3(wxSocketBase *sock) unsigned char len; char *buf; - m_text->AppendText(_T("Test 3 begins\n")); + m_text->AppendText(_("Test 3 begins\n")); - // 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); - sock->Read((char *)&len, 1); + // Read the size + sock->Read(&len, 1); buf = new char[len * 1024]; + + // Read the data sock->Read(buf, len * 1024); + m_text->AppendText(_("Got the data, sending it back\n")); + + // Write it back sock->Write(buf, len * 1024); delete[] buf; - m_text->AppendText(_T("Test 3 ends\n")); + m_text->AppendText(_("Test 3 ends\n\n")); } void MyFrame::OnServerEvent(wxSocketEvent& event) { - wxString s = _T("OnServerEvent: "); + wxString s = _("OnServerEvent: "); wxSocketBase *sock; - switch(event.SocketEvent()) + switch(event.GetSocketEvent()) { - case wxSOCKET_CONNECTION : s.Append(_T("wxSOCKET_CONNECTION\n")); break; - default : s.Append(_T("Unexpected event !\n")); break; + case wxSOCKET_CONNECTION : s.Append(_("wxSOCKET_CONNECTION\n")); break; + default : s.Append(_("Unexpected event !\n")); break; } 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(_T("New client connection accepted\n")); + m_text->AppendText(_("New client connection accepted\n\n")); } else { - m_text->AppendText(_T("Error: couldn't accept a new connection")); + m_text->AppendText(_("Error: couldn't accept a new connection\n\n")); return; } sock->SetEventHandler(*this, SOCKET_ID); sock->SetNotify(wxSOCKET_INPUT_FLAG | wxSOCKET_LOST_FLAG); - sock->Notify(TRUE); + sock->Notify(true); m_numClients++; UpdateStatusBar(); @@ -330,21 +348,21 @@ void MyFrame::OnServerEvent(wxSocketEvent& event) void MyFrame::OnSocketEvent(wxSocketEvent& event) { - wxSocketBase *sock = event.Socket(); - wxString s = _T("OnSocketEvent: "); + 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(_T("wxSOCKET_INPUT\n")); break; - case wxSOCKET_LOST: s.Append(_T("wxSOCKET_LOST\n")); break; - default: s.Append(_T("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: { @@ -354,14 +372,15 @@ void MyFrame::OnSocketEvent(wxSocketEvent& event) // Which test are we going to run? unsigned char c; - sock->Read((char *)&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(_T("Unknown test id received from client\n")); + default: + m_text->AppendText(_("Unknown test id received from client\n\n")); } // Enable input events again. @@ -372,11 +391,17 @@ void MyFrame::OnSocketEvent(wxSocketEvent& event) { m_numClients--; - // We cannot delete the socket right now because we can - // be in the middle of a test or something. So we append - // it to the list of objects to be deleted. - m_text->AppendText(_T("Deleting socket.\n")); - wxPendingDelete.Append(sock); + // Destroy() should be used instead of delete wherever possible, + // due to the fact that wxSocket uses 'delayed events' (see the + // documentation for wxPostEvent) and we don't want an event to + // arrive to the event handler (the frame, here) after the socket + // 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. + + m_text->AppendText(_("Deleting socket.\n\n")); + sock->Destroy(); break; } default: ; @@ -389,7 +414,9 @@ void MyFrame::OnSocketEvent(wxSocketEvent& event) void MyFrame::UpdateStatusBar() { +#if wxUSE_STATUSBAR wxString s; - s.Printf(_T("%d clients connected"), m_numClients); + s.Printf(_("%d clients connected"), m_numClients); SetStatusText(s, 1); +#endif // wxUSE_STATUSBAR }