]> git.saurik.com Git - wxWidgets.git/blobdiff - samples/sockets/server.cpp
Changes for wxOGL since wxFatalError has been deprecated.
[wxWidgets.git] / samples / sockets / server.cpp
index 817c15657b65774b063f28287970dbb5f2f5a964..cbdcc5c5c6892954c722b2d961c4935a721a40fc 100644 (file)
@@ -29,7 +29,7 @@
 #  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
@@ -41,7 +41,7 @@
 // --------------------------------------------------------------------------
 
 // the application icon
 // --------------------------------------------------------------------------
 
 // the application icon
-#if defined(__WXGTK__) || defined(__WXMOTIF__)
+#if defined(__WXGTK__) || defined(__WXMOTIF__) || defined(__WXMAC__)
 #  include "mondrian.xpm"
 #endif
 
 #  include "mondrian.xpm"
 #endif
 
@@ -78,7 +78,6 @@ public:
 
 private:
   wxSocketServer *m_server;
 
 private:
   wxSocketServer *m_server;
-  wxPanel        *m_panel;
   wxTextCtrl     *m_text;
   wxMenu         *m_menuFile;
   wxMenuBar      *m_menuBar;
   wxTextCtrl     *m_text;
   wxMenu         *m_menuFile;
   wxMenuBar      *m_menuBar;
@@ -119,10 +118,6 @@ END_EVENT_TABLE()
 IMPLEMENT_APP(MyApp)
 
 
 IMPLEMENT_APP(MyApp)
 
 
-// To append sockets for delayed deletion
-extern wxList wxPendingDelete;
-
-
 // ==========================================================================
 // implementation
 // ==========================================================================
 // ==========================================================================
 // implementation
 // ==========================================================================
@@ -140,7 +135,7 @@ bool MyApp::OnInit()
   frame->Show(TRUE);
   SetTopWindow(frame);
 
   frame->Show(TRUE);
   SetTopWindow(frame);
 
-  // success
+  // Success
   return TRUE;
 }
 
   return TRUE;
 }
 
@@ -149,8 +144,9 @@ bool MyApp::OnInit()
 // --------------------------------------------------------------------------
 
 // frame constructor
 // --------------------------------------------------------------------------
 
 // frame constructor
+
 MyFrame::MyFrame() : wxFrame((wxFrame *)NULL, -1,
 MyFrame::MyFrame() : wxFrame((wxFrame *)NULL, -1,
-                             _T("wxSocket demo: Server"),
+                             _("wxSocket demo: Server"),
                              wxDefaultPosition, wxSize(300, 200))
 {
   // Give the frame an icon
                              wxDefaultPosition, wxSize(300, 200))
 {
   // Give the frame an icon
@@ -158,40 +154,46 @@ MyFrame::MyFrame() : wxFrame((wxFrame *)NULL, -1,
 
   // Make menus
   m_menuFile = new wxMenu();
 
   // 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->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();
 
   // 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);
 
   // Status bar
   CreateStatusBar(2);
 
   SetMenuBar(m_menuBar);
 
   // Status bar
   CreateStatusBar(2);
 
-  // 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, -1,
+                           _("Welcome to wxSocket demo: Server\n"),
+                           wxDefaultPosition, wxDefaultSize,
                            wxTE_MULTILINE | wxTE_READONLY);
 
                            wxTE_MULTILINE | wxTE_READONLY);
 
-  // Create the socket
+  // Create the address - defaults to localhost:0 initially
   wxIPV4address addr;
   addr.Service(3000);
   wxIPV4address addr;
   addr.Service(3000);
-  addr.LocalHost();
 
 
+  // Create the socket
   m_server = new wxSocketServer(addr);
   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
 
   // 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
   else
-    m_text->AppendText(_T("Could not listen at the specified port !\n\n"));
+  {
+    m_text->AppendText(_("Server listening.\n\n"));
+  }
+
+  // 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;
 
   m_busy = FALSE;
   m_numClients = 0;
@@ -200,6 +202,7 @@ MyFrame::MyFrame() : wxFrame((wxFrame *)NULL, -1,
 
 MyFrame::~MyFrame()
 {
 
 MyFrame::~MyFrame()
 {
+  // No delayed deletion here, as the frame is dying anyway
   delete m_server;
 }
 
   delete m_server;
 }
 
@@ -213,9 +216,8 @@ void MyFrame::OnQuit(wxCommandEvent& WXUNUSED(event))
 
 void MyFrame::OnAbout(wxCommandEvent& WXUNUSED(event))
 {
 
 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);
 }
 
                wxOK | wxICON_INFORMATION, this);
 }
 
@@ -224,24 +226,29 @@ void MyFrame::Test1(wxSocketBase *sock)
   unsigned char len;
   char *buf;
 
   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
 
   // 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 WAITALL flag. Also, we disabled
-  // input events so we won't have unwanted reentrance. This
-  // way we can avoid the infamous BLOCK (formerly SPEED) flag.
-  //
+  // 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->SetFlags(wxSOCKET_WAITALL);
 
-  sock->Read((char *)&len, 1);
+  // Read the size
+  sock->Read(&len, 1);
+  buf = new char[len];
 
 
-  buf = (char *)malloc(len);
+  // Read the data
   sock->Read(buf, len);
   sock->Read(buf, len);
+  m_text->AppendText(_("Got the data, sending it back\n"));
+
+  // Write it back
   sock->Write(buf, len);
   sock->Write(buf, len);
-  free(buf);
+  delete[] buf;
 
 
-  m_text->AppendText(_T("Test 1 ends\n"));
+  m_text->AppendText(_("Test 1 ends\n\n"));
 }
 
 void MyFrame::Test2(wxSocketBase *sock)
 }
 
 void MyFrame::Test2(wxSocketBase *sock)
@@ -249,43 +256,65 @@ void MyFrame::Test2(wxSocketBase *sock)
 #define MAX_MSG_SIZE 10000
 
   wxString s;
 #define MAX_MSG_SIZE 10000
 
   wxString s;
-  char *buf = (char *)malloc(MAX_MSG_SIZE);
+  wxChar *buf = new wxChar[MAX_MSG_SIZE];
   wxUint32 len;
 
   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.
 
   // 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(s);
+  m_text->AppendText(_("Sending the data back\n"));
 
 
+  // Write it back
   sock->WriteMsg(buf, len);
   sock->WriteMsg(buf, len);
-  free(buf);
+  delete[] buf;
 
 
-  m_text->AppendText(_T("Test 2 ends\n"));
+  m_text->AppendText(_("Test 2 ends\n\n"));
 
 #undef MAX_MSG_SIZE
 }
 
 void MyFrame::Test3(wxSocketBase *sock)
 {
 
 #undef MAX_MSG_SIZE
 }
 
 void MyFrame::Test3(wxSocketBase *sock)
 {
-  m_text->AppendText(_T("Test 3 begins\n"));
-  m_text->AppendText(_T("(not implemented)\n"));
-  m_text->AppendText(_T("Test 3 ends\n"));
+  unsigned char len;
+  char *buf;
+
+  m_text->AppendText(_("Test 3 begins\n"));
+
+  // 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
+  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(_("Test 3 ends\n\n"));
 }
 
 void MyFrame::OnServerEvent(wxSocketEvent& event)
 {
 }
 
 void MyFrame::OnServerEvent(wxSocketEvent& event)
 {
-  wxString s = _T("OnServerEvent: ");
+  wxString s = _("OnServerEvent: ");
   wxSocketBase *sock;
 
   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);
   }
 
   m_text->AppendText(s);
@@ -294,16 +323,16 @@ void MyFrame::OnServerEvent(wxSocketEvent& event)
   // connections queue, else exit. We use Accept(FALSE) for
   // non-blocking accept (although if we got here, there
   // should ALWAYS be a pending connection).
   // 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);
 
   if (sock)
   {
   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
   {
   }
   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;
   }
 
     return;
   }
 
@@ -317,21 +346,21 @@ void MyFrame::OnServerEvent(wxSocketEvent& event)
 
 void MyFrame::OnSocketEvent(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
   }
 
   m_text->AppendText(s);
 
   // Now we process the event
-  switch(event.SocketEvent())
+  switch(event.GetSocketEvent())
   {
     case wxSOCKET_INPUT:
     {
   {
     case wxSOCKET_INPUT:
     {
@@ -341,14 +370,15 @@ void MyFrame::OnSocketEvent(wxSocketEvent& event)
 
       // Which test are we going to run?
       unsigned char c;
 
       // 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;
 
       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.
       }
 
       // Enable input events again.
@@ -359,11 +389,17 @@ void MyFrame::OnSocketEvent(wxSocketEvent& event)
     {
       m_numClients--;
 
     {
       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: ;
       break;
     }
     default: ;
@@ -377,6 +413,6 @@ void MyFrame::OnSocketEvent(wxSocketEvent& event)
 void MyFrame::UpdateStatusBar()
 {
   wxString s;
 void MyFrame::UpdateStatusBar()
 {
   wxString s;
-  s.Printf(_T("%d clients connected"), m_numClients);
+  s.Printf(_("%d clients connected"), m_numClients);
   SetStatusText(s, 1);
 }
   SetStatusText(s, 1);
 }