correcting text alignment flag

[wxWidgets.git] / samples / sockets / server.cpp

diff --git a/samples/sockets/server.cpp b/samples/sockets/server.cpp
index 47acc20ab4fc1d0c711ded571a78032d2c501c03..04b4ce33f28d6bfe534c3205678a5fd5e92aae84 100644 (file)
--- a/samples/sockets/server.cpp
+++ b/samples/sockets/server.cpp
@@ -2,10 +2,10 @@
 // Name:        server.cpp
 // Purpose:     Server for wxSocket demo
 // Author:      Guillermo Rodriguez Garcia <guille@iies.es>
 // Name:        server.cpp
 // Purpose:     Server for wxSocket demo
 // Author:      Guillermo Rodriguez Garcia <guille@iies.es>

-// Modified by:

 // Created:     1999/09/19
 // RCS-ID:      $Id$
 // Copyright:   (c) 1999 Guillermo Rodriguez Garcia
 // Created:     1999/09/19
 // RCS-ID:      $Id$
 // Copyright:   (c) 1999 Guillermo Rodriguez Garcia

+//              (c) 2009 Vadim Zeitlin

 // Licence:     wxWindows licence
 /////////////////////////////////////////////////////////////////////////////
 
 // Licence:     wxWindows licence
 /////////////////////////////////////////////////////////////////////////////
 


@@ -17,11 +17,6 @@
 // headers
 // --------------------------------------------------------------------------
 
 // 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"
 
 // For compilers that support precompilation, includes "wx/wx.h".
 #include "wx/wxprec.h"
 


@@ -29,20 +24,29 @@
 #  pragma hdrstop
 #endif
 
 #  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
 
 #ifndef WX_PRECOMP
 #  include "wx/wx.h"
 #endif
 

+#include "wx/busyinfo.h"

 #include "wx/socket.h"
 
 #include "wx/socket.h"
 

+// this example is currently written to use only IP or only IPv6 sockets, it
+// should be extended to allow using either in the future
+#if wxUSE_IPV6
+    typedef wxIPV6address IPaddress;
+#else
+    typedef wxIPV4address IPaddress;
+#endif
+

 // --------------------------------------------------------------------------
 // resources
 // --------------------------------------------------------------------------
 
 // the application icon
 // --------------------------------------------------------------------------
 // resources
 // --------------------------------------------------------------------------
 
 // the application icon

-#if defined(__WXGTK__) || defined(__WXMOTIF__)
-#  include "mondrian.xpm"
+#ifndef wxHAS_IMAGES_IN_RESOURCES
+    #include "../sample.xpm"

 #endif
 
 // --------------------------------------------------------------------------
 #endif
 
 // --------------------------------------------------------------------------


@@ -64,6 +68,8 @@ public:
   ~MyFrame();
 
   // event handlers (these functions should _not_ be virtual)
   ~MyFrame();
 
   // event handlers (these functions should _not_ be virtual)

+  void OnUDPTest(wxCommandEvent& event);
+  void OnWaitForAccept(wxCommandEvent& event);

   void OnQuit(wxCommandEvent& event);
   void OnAbout(wxCommandEvent& event);
   void OnServerEvent(wxSocketEvent& event);
   void OnQuit(wxCommandEvent& event);
   void OnAbout(wxCommandEvent& event);
   void OnServerEvent(wxSocketEvent& event);


@@ -84,10 +90,28 @@ private:
   bool            m_busy;
   int             m_numClients;
 
   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()
 };
 
   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
 // --------------------------------------------------------------------------
 // --------------------------------------------------------------------------
 // constants
 // --------------------------------------------------------------------------


@@ -96,21 +120,25 @@ private:
 enum
 {
   // menu items
 enum
 {
   // menu items

-  SERVER_QUIT = 1000,
-  SERVER_ABOUT,
+  SERVER_UDPTEST = 10,
+  SERVER_WAITFORACCEPT,
+  SERVER_QUIT = wxID_EXIT,
+  SERVER_ABOUT = wxID_ABOUT,

 
   // id for sockets
 
   // id for sockets

-  SERVER_ID,
+  SERVER_ID = 100,

   SOCKET_ID
 };
 
 // --------------------------------------------------------------------------
   SOCKET_ID
 };
 
 // --------------------------------------------------------------------------

-// event tables and other macros for wxWindows
+// event tables and other macros for wxWidgets

 // --------------------------------------------------------------------------
 
 BEGIN_EVENT_TABLE(MyFrame, wxFrame)
   EVT_MENU(SERVER_QUIT,  MyFrame::OnQuit)
   EVT_MENU(SERVER_ABOUT, MyFrame::OnAbout)
 // --------------------------------------------------------------------------
 
 BEGIN_EVENT_TABLE(MyFrame, wxFrame)
   EVT_MENU(SERVER_QUIT,  MyFrame::OnQuit)
   EVT_MENU(SERVER_ABOUT, MyFrame::OnAbout)

+  EVT_MENU(SERVER_UDPTEST, MyFrame::OnUDPTest)
+  EVT_MENU(SERVER_WAITFORACCEPT, MyFrame::OnWaitForAccept)

   EVT_SOCKET(SERVER_ID,  MyFrame::OnServerEvent)
   EVT_SOCKET(SOCKET_ID,  MyFrame::OnSocketEvent)
 END_EVENT_TABLE()
   EVT_SOCKET(SERVER_ID,  MyFrame::OnServerEvent)
   EVT_SOCKET(SOCKET_ID,  MyFrame::OnSocketEvent)
 END_EVENT_TABLE()


@@ -128,15 +156,17 @@ IMPLEMENT_APP(MyApp)
 
 bool MyApp::OnInit()
 {
 
 bool MyApp::OnInit()
 {

+  if ( !wxApp::OnInit() )
+      return false;
+

   // Create the main application window
   MyFrame *frame = new MyFrame();
 
   // Create the main application window
   MyFrame *frame = new MyFrame();
 

-  // Show it and tell the application that it's our main window
-  frame->Show(TRUE);
-  SetTopWindow(frame);
+  // Show it
+  frame->Show(true);

 
   // Success
 
   // Success

-  return TRUE;
+  return true;

 }
 
 // --------------------------------------------------------------------------
 }
 
 // --------------------------------------------------------------------------


@@ -145,16 +175,19 @@ bool MyApp::OnInit()
 
 // frame constructor
 
 
 // frame constructor
 

-MyFrame::MyFrame() : wxFrame((wxFrame *)NULL, -1,
+MyFrame::MyFrame() : wxFrame((wxFrame *)NULL, wxID_ANY,

                              _("wxSocket demo: Server"),
                              wxDefaultPosition, wxSize(300, 200))
 {
   // Give the frame an icon
                              _("wxSocket demo: Server"),
                              wxDefaultPosition, wxSize(300, 200))
 {
   // Give the frame an icon

-  SetIcon(wxICON(mondrian));
+  SetIcon(wxICON(sample));

 
   // Make menus
   m_menuFile = new wxMenu();
 
   // Make menus
   m_menuFile = new wxMenu();

-  m_menuFile->Append(SERVER_ABOUT, _("&About...\tCtrl-A"), _("Show about dialog"));
+  m_menuFile->Append(SERVER_WAITFORACCEPT, "&Wait for connection\tCtrl-W");
+  m_menuFile->Append(SERVER_UDPTEST, "&UDP test\tCtrl-U");
+  m_menuFile->AppendSeparator();
+  m_menuFile->Append(SERVER_ABOUT, _("&About\tCtrl-A"), _("Show about dialog"));

   m_menuFile->AppendSeparator();
   m_menuFile->Append(SERVER_QUIT, _("E&xit\tAlt-X"), _("Quit server"));
 
   m_menuFile->AppendSeparator();
   m_menuFile->Append(SERVER_QUIT, _("E&xit\tAlt-X"), _("Quit server"));
 


@@ -163,39 +196,51 @@ MyFrame::MyFrame() : wxFrame((wxFrame *)NULL, -1,
   m_menuBar->Append(m_menuFile, _("&File"));
   SetMenuBar(m_menuBar);
 
   m_menuBar->Append(m_menuFile, _("&File"));
   SetMenuBar(m_menuBar);
 

+#if wxUSE_STATUSBAR

   // Status bar
   CreateStatusBar(2);
   // Status bar
   CreateStatusBar(2);

+#endif // wxUSE_STATUSBAR

 
   // Make a textctrl for logging
 
   // 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);
                            _("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
 
   // Create the address - defaults to localhost:0 initially

-  wxIPV4address addr;
+  IPaddress addr;

   addr.Service(3000);
 
   addr.Service(3000);
 

+  wxLogMessage("Creating server at %s:%u", addr.IPAddress(), addr.Service());
+

   // Create the socket
   m_server = new wxSocketServer(addr);
 
   // Create the socket
   m_server = new wxSocketServer(addr);
 

-  // We use Ok() here to see if the server is really listening
-  if (! m_server->Ok())
+  // We use IsOk() here to see if the server is really listening
+  if (! m_server->IsOk())

   {
   {

-    m_text->AppendText(_("Could not listen at the specified port !\n\n"));
+    wxLogMessage("Could not listen at the specified port !");

     return;
   }
     return;
   }

+
+  IPaddress addrReal;
+  if ( !m_server->GetLocal(addrReal) )
+  {
+    wxLogMessage("ERROR: couldn't get the address we bound to");
+  }

   else
   {
   else
   {

-    m_text->AppendText(_("Server listening.\n\n"));
+    wxLogMessage("Server listening at %s:%u",
+                 addrReal.IPAddress(), addrReal.Service());

   }
 
   // Setup the event handler and subscribe to connection events
   m_server->SetEventHandler(*this, SERVER_ID);
   m_server->SetNotify(wxSOCKET_CONNECTION_FLAG);
   }
 
   // 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();
 }
   m_numClients = 0;
   UpdateStatusBar();
 }


@@ -210,24 +255,67 @@ MyFrame::~MyFrame()
 
 void MyFrame::OnQuit(wxCommandEvent& WXUNUSED(event))
 {
 
 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::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);
 }
 
                _("About Server"),
                wxOK | wxICON_INFORMATION, this);
 }
 

-void MyFrame::Test1(wxSocketBase *sock)
+void MyFrame::OnUDPTest(wxCommandEvent& WXUNUSED(event))

 {
 {

-  unsigned char len;
-  char *buf;
+    TestLogger logtest("UDP test");

 
 

-  m_text->AppendText(_("Test 1 begins\n"));
+    IPaddress addr;
+    addr.Service(3000);
+    wxDatagramSocket sock(addr);
+
+    char buf[1024];
+    size_t n = sock.RecvFrom(addr, buf, sizeof(buf)).LastCount();
+    if ( !n )
+    {
+        wxLogMessage("ERROR: failed to receive data");
+        return;
+    }
+
+    wxLogMessage("Received \"%s\" from %s:%u.",
+                 wxString::From8BitData(buf, n),
+                 addr.IPAddress(), addr.Service());
+
+    for ( size_t i = 0; i < n; i++ )
+    {
+        char& c = buf[i];
+        if ( (c >= 'A' && c <= 'M') || (c >= 'a' && c <= 'm') )
+            c += 13;
+        else if ( (c >= 'N' && c <= 'Z') || (c >= 'n' && c <= 'z') )
+            c -= 13;
+    }
+
+    if ( sock.SendTo(addr, buf, n).LastCount() != n )
+    {
+        wxLogMessage("ERROR: failed to send data");
+        return;
+    }
+}
+
+void MyFrame::OnWaitForAccept(wxCommandEvent& WXUNUSED(event))
+{
+    TestLogger logtest("WaitForAccept() test");
+
+    wxBusyInfo("Waiting for connection for 10 seconds...", this);
+    if ( m_server->WaitForAccept(10) )
+        wxLogMessage("Accepted client connection.");
+    else
+        wxLogMessage("Connection error or timeout expired.");
+}
+
+void MyFrame::Test1(wxSocketBase *sock)
+{
+  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
 
   // Receive data from socket and send it back. We will first
   // get a byte with the buffer size, so we can specify the


@@ -238,73 +326,62 @@ void MyFrame::Test1(wxSocketBase *sock)
   sock->SetFlags(wxSOCKET_WAITALL);
 
   // Read the size
   sock->SetFlags(wxSOCKET_WAITALL);
 
   // Read the size

+  unsigned char len;

   sock->Read(&len, 1);
   sock->Read(&len, 1);

-  buf = new char[len];
+  wxCharBuffer buf(len);

 
   // Read the data
 
   // Read the data

-  sock->Read(buf, len);
-  m_text->AppendText(_("Got the data, sending it back\n"));
+  sock->Read(buf.data(), len);
+  wxLogMessage("Got the data, sending it back");

 
   // Write it back
   sock->Write(buf, len);
 
   // Write it back
   sock->Write(buf, len);

-  delete[] buf;
-
-  m_text->AppendText(_("Test 1 ends\n\n"));

 }
 
 void MyFrame::Test2(wxSocketBase *sock)
 {
 }
 
 void MyFrame::Test2(wxSocketBase *sock)
 {

-#define MAX_MSG_SIZE 10000
-
-  wxString s;
-  char *buf = new char[MAX_MSG_SIZE];
-  wxUint32 len;
+  char buf[4096];

 
 

-  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.
 
   // Read the message
 
   // We don't need to set flags because ReadMsg and WriteMsg
   // are not affected by them anyway.
 
   // Read the message

-  len = sock->ReadMsg(buf, MAX_MSG_SIZE).LastCount();
-  s.Printf(_("Client says: %s\n"), buf);
-  m_text->AppendText(s);
-  m_text->AppendText(_("Sending the data back\n"));
+  wxUint32 len = sock->ReadMsg(buf, sizeof(buf)).LastCount();
+  if ( !len )
+  {
+      wxLogError("Failed to read message.");
+      return;
+  }
+
+  wxLogMessage("Got \"%s\" from client.", wxString::FromUTF8(buf, len));
+  wxLogMessage("Sending the data back");

 
   // Write it back
   sock->WriteMsg(buf, len);
 
   // Write it back
   sock->WriteMsg(buf, len);

-  delete[] buf;
-
-  m_text->AppendText(_("Test 2 ends\n\n"));
-
-#undef MAX_MSG_SIZE

 }
 
 void MyFrame::Test3(wxSocketBase *sock)
 {
 }
 
 void MyFrame::Test3(wxSocketBase *sock)
 {

-  unsigned char len;
-  char *buf;
+  TestLogger logtest("Test 3");

 
 

-  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);
 
   // Read the size
   // expressed in kbytes - this tests large data transfers.
 
   sock->SetFlags(wxSOCKET_WAITALL);
 
   // Read the size

+  unsigned char len;

   sock->Read(&len, 1);
   sock->Read(&len, 1);

-  buf = new char[len * 1024];
+  wxCharBuffer buf(len*1024);

 
   // Read the data
 
   // Read the data

-  sock->Read(buf, len * 1024);
-  m_text->AppendText(_("Got the data, sending it back\n"));
+  sock->Read(buf.data(), len * 1024);
+  wxLogMessage("Got the data, sending it back");

 
   // Write it back
   sock->Write(buf, len * 1024);
 
   // Write it back
   sock->Write(buf, len * 1024);

-  delete[] buf;
-
-  m_text->AppendText(_("Test 3 ends\n\n"));

 }
 
 void MyFrame::OnServerEvent(wxSocketEvent& event)
 }
 
 void MyFrame::OnServerEvent(wxSocketEvent& event)


@@ -321,26 +398,34 @@ void MyFrame::OnServerEvent(wxSocketEvent& event)
   m_text->AppendText(s);
 
   // Accept new connection if there is one in the pending
   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).
 
   // 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)
   {
 
   if (sock)
   {

-    m_text->AppendText(_("New client connection accepted\n\n"));
+    IPaddress addr;
+    if ( !sock->GetPeer(addr) )
+    {
+      wxLogMessage("New connection from unknown client accepted.");
+    }
+    else
+    {
+      wxLogMessage("New client connection from %s:%u accepted",
+                   addr.IPAddress(), addr.Service());
+    }

   }
   else
   {
   }
   else
   {

-    m_text->AppendText(_("Error: couldn't accept a new connection\n\n"));
-    sock->Destroy();
+    wxLogMessage("Error: couldn't accept a new connection");

     return;
   }
 
   sock->SetEventHandler(*this, SOCKET_ID);
   sock->SetNotify(wxSOCKET_INPUT_FLAG | wxSOCKET_LOST_FLAG);
     return;
   }
 
   sock->SetEventHandler(*this, SOCKET_ID);
   sock->SetNotify(wxSOCKET_INPUT_FLAG | wxSOCKET_LOST_FLAG);

-  sock->Notify(TRUE);
+  sock->Notify(true);

 
   m_numClients++;
   UpdateStatusBar();
 
   m_numClients++;
   UpdateStatusBar();


@@ -380,7 +465,7 @@ void MyFrame::OnSocketEvent(wxSocketEvent& event)
         case 0xCE: Test2(sock); break;
         case 0xDE: Test3(sock); break;
         default:
         case 0xCE: Test2(sock); break;
         case 0xDE: Test3(sock); break;
         default:

-          m_text->AppendText(_("Unknown test id received from client\n\n"));
+          wxLogMessage("Unknown test id received from client");

       }
 
       // Enable input events again.
       }
 
       // Enable input events again.


@@ -400,7 +485,7 @@ void MyFrame::OnSocketEvent(wxSocketEvent& event)
       // middle of a test or something. Destroy() takes care of all
       // this for us.
 
       // middle of a test or something. Destroy() takes care of all
       // this for us.
 

-      m_text->AppendText(_("Deleting socket.\n\n"));
+      wxLogMessage("Deleting socket.");

       sock->Destroy();
       break;
     }
       sock->Destroy();
       break;
     }


@@ -414,7 +499,9 @@ void MyFrame::OnSocketEvent(wxSocketEvent& event)
 
 void MyFrame::UpdateStatusBar()
 {
 
 void MyFrame::UpdateStatusBar()
 {

+#if wxUSE_STATUSBAR

   wxString s;
   s.Printf(_("%d clients connected"), m_numClients);
   SetStatusText(s, 1);
   wxString s;
   s.Printf(_("%d clients connected"), m_numClients);
   SetStatusText(s, 1);

+#endif // wxUSE_STATUSBAR

 }
 }