]> git.saurik.com Git - wxWidgets.git/blobdiff - samples/sockets/server.cpp
Do not use Tooltips if they are disabled
[wxWidgets.git] / samples / sockets / server.cpp
index 441d5721ba0d3ab8dcee7886f3251f04ede436d8..b7063e93d041871a22d9ebe2336730795d1fd2e9 100644 (file)
 // 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(&c1);
 
       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
 }