Small changes

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

diff --git a/samples/wxsocket/server.cpp b/samples/wxsocket/server.cpp
index 2c521909368b49ad34cc72de46721fc61285792b..817c15657b65774b063f28287970dbb5f2f5a964 100644 (file)
--- a/samples/wxsocket/server.cpp
+++ b/samples/wxsocket/server.cpp
@@ -1,189 +1,382 @@
-/*
- * File:       server.cpp
- * Purpose:    wxSocket: server demo
- * Author:     LAVAUX Guilhem (from minimal.cc)
- * Created:    June 1997
- * Updated:    
- * Copyright:  (c) 1993, AIAI, University of Edinburgh
- *             (C) 1997, LAVAUX Guilhem
- */
+/////////////////////////////////////////////////////////////////////////////
+// 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
+// Licence:     wxWindows licence
+/////////////////////////////////////////////////////////////////////////////
+
+// ==========================================================================
+// declarations
+// ==========================================================================
+
+// --------------------------------------------------------------------------
+// headers
+// --------------------------------------------------------------------------
+

 #ifdef __GNUG__
 #ifdef __GNUG__

-#pragma implementation
-#pragma interface
+#  pragma implementation "server.cpp"
+#  pragma interface "server.cpp"

 #endif
 
 #endif
 

-// For compilers that support precompilation, includes "wx.h".
+// For compilers that support precompilation, includes "wx/wx.h".

 #include "wx/wxprec.h"
 
 #ifdef __BORLANDC__
 #include "wx/wxprec.h"
 
 #ifdef __BORLANDC__

-#pragma hdrstop
+#  pragma hdrstop

 #endif
 
 #endif
 

+// for all others, include the necessary headers 

 #ifndef WX_PRECOMP
 #ifndef WX_PRECOMP

-#include "wx/wx.h"
+#  include "wx/wx.h"

 #endif
 #endif

+

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

-// Define a new application type
-class MyApp: public wxApp
-{ public:
-    bool OnInit(void);
-};
+// --------------------------------------------------------------------------
+// resources
+// --------------------------------------------------------------------------

 
 

-class MyServer;
+// the application icon
+#if defined(__WXGTK__) || defined(__WXMOTIF__)
+#  include "mondrian.xpm"
+#endif

 
 

-// Define a new frame type
-class MyFrame: public wxFrame
-{ 
-  DECLARE_EVENT_TABLE()
+// --------------------------------------------------------------------------
+// classes
+// --------------------------------------------------------------------------
+
+// Define a new application type
+class MyApp : public wxApp
+{

 public:
 public:

-  MyServer *sock;
-  int nb_clients;
-
-  MyFrame(wxFrame *frame);
-  virtual ~MyFrame();
-  void Menu_Exit(wxCommandEvent& evt);
-  void ExecTest1(wxSocketBase *sock_o);
-  void UpdateStatus(int incr);
+  virtual bool OnInit();

 };
 
 };
 

-#define SKDEMO_QUIT 101
+// Define a new frame type: this is going to be our main frame
+class MyFrame : public wxFrame
+{
+public:
+  MyFrame();
+  ~MyFrame();
+
+  // event handlers (these functions should _not_ be virtual)
+  void OnQuit(wxCommandEvent& event);
+  void OnAbout(wxCommandEvent& event);
+  void OnServerEvent(wxSocketEvent& event);
+  void OnSocketEvent(wxSocketEvent& event);
+
+  void Test1(wxSocketBase *sock);
+  void Test2(wxSocketBase *sock);
+  void Test3(wxSocketBase *sock);
+
+  // convenience functions
+  void UpdateStatusBar();
+
+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
+  DECLARE_EVENT_TABLE()
+};

 
 

-BEGIN_EVENT_TABLE(MyFrame, wxFrame)
-  EVT_MENU(SKDEMO_QUIT, MyFrame::Menu_Exit)
-END_EVENT_TABLE()
+// --------------------------------------------------------------------------
+// constants
+// --------------------------------------------------------------------------

 
 

-class MySock: public wxSocketBase {
-public:
-  MyFrame *frame;
+// IDs for the controls and the menu commands
+enum
+{
+  // menu items
+  SERVER_QUIT = 1000,
+  SERVER_ABOUT,

 
 

-  void OldOnNotify(wxRequestEvent flags);
+  // id for sockets
+  SERVER_ID,
+  SOCKET_ID

 };
 
 };
 

-class MyServer: public wxSocketServer {
-public:
-  MyFrame *frame;
+// --------------------------------------------------------------------------
+// event tables and other macros for wxWindows
+// --------------------------------------------------------------------------

 
 

-  MyServer(wxSockAddress& addr) : wxSocketServer(addr) { }
-  void OldOnNotify(wxRequestEvent flags);
-};
+BEGIN_EVENT_TABLE(MyFrame, wxFrame)
+  EVT_MENU(SERVER_QUIT,  MyFrame::OnQuit)
+  EVT_MENU(SERVER_ABOUT, MyFrame::OnAbout)
+  EVT_SOCKET(SERVER_ID,  MyFrame::OnServerEvent)
+  EVT_SOCKET(SOCKET_ID,  MyFrame::OnSocketEvent)
+END_EVENT_TABLE()

 
 IMPLEMENT_APP(MyApp)
 
 
 IMPLEMENT_APP(MyApp)
 

-// `Main program' equivalent, creating windows and returning main app frame
-bool MyApp::OnInit(void)
-{
-  // Create the main frame window
-  MyFrame *frame = new MyFrame(NULL);

 
 

-  // Give it an icon
-#ifdef wx_msw
-  frame->SetIcon(new wxIcon("mondrian"));
-#endif
-#ifdef wx_x
-  frame->SetIcon(new wxIcon("aiai.xbm"));
-#endif
+// To append sockets for delayed deletion
+extern wxList wxPendingDelete;

 
 

-  // Make a menubar
-  wxMenu *file_menu = new wxMenu;

 
 

-  file_menu->Append(SKDEMO_QUIT, "E&xit");
-  wxMenuBar *menu_bar = new wxMenuBar;
-  menu_bar->Append(file_menu, "File");
-  frame->SetMenuBar(menu_bar);
+// ==========================================================================
+// implementation
+// ==========================================================================

 
 

-  // Make a panel with a message
-  (void)new wxPanel(frame, 0, 0, 300, 100);
+// --------------------------------------------------------------------------
+// the application class
+// --------------------------------------------------------------------------
+
+bool MyApp::OnInit()
+{
+  // Create the main application window
+  MyFrame *frame = new MyFrame();

 
 

-  // Show the frame
+  // Show it and tell the application that it's our main window

   frame->Show(TRUE);
   frame->Show(TRUE);

-  
-  // Return the main frame window
+  SetTopWindow(frame);
+
+  // success

   return TRUE;
 }
 
   return TRUE;
 }
 

-void MySock::OldOnNotify(wxRequestEvent flags)
+// --------------------------------------------------------------------------
+// main frame
+// --------------------------------------------------------------------------
+
+// frame constructor
+MyFrame::MyFrame() : wxFrame((wxFrame *)NULL, -1,
+                             _T("wxSocket demo: Server"),
+                             wxDefaultPosition, wxSize(300, 200))

 {
 {

-  extern wxList wxPendingDelete;
+  // Give the frame an icon
+  SetIcon(wxICON(mondrian));
+
+  // Make menus
+  m_menuFile = new wxMenu();
+  m_menuFile->Append(SERVER_ABOUT, _T("&About...\tCtrl-A"), _T("Show about dialog"));
+  m_menuFile->AppendSeparator();
+  m_menuFile->Append(SERVER_QUIT, _T("E&xit\tAlt-X"), _T("Quit server"));
+
+  // Append menus to the menubar
+  m_menuBar = new wxMenuBar();
+  m_menuBar->Append(m_menuFile, _T("&File"));
+  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(),
+                           wxTE_MULTILINE | wxTE_READONLY);
+
+  // Create the socket
+  wxIPV4address addr;
+  addr.Service(3000);
+  addr.LocalHost();
+
+  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"));
+  else
+    m_text->AppendText(_T("Could not listen at the specified port !\n\n"));
+
+  m_busy = FALSE;
+  m_numClients = 0;
+  UpdateStatusBar();
+}

 
 

-  switch (flags) {
-  case EVT_READ:
-    unsigned char c;
+MyFrame::~MyFrame()
+{
+  delete m_server;
+}

 
 

-    ReadMsg((char *)&c, 1);
-    if (c == 0xbe)
-      frame->ExecTest1(this);
+// event handlers

 
 

-    break;
-  case EVT_LOST:
-    frame->UpdateStatus(-1);
-    wxPendingDelete.Append(this);
-    break;
-  }
+void MyFrame::OnQuit(wxCommandEvent& WXUNUSED(event))
+{
+  // TRUE is to force the frame to close
+  Close(TRUE);

 }
 
 }
 

-void MyServer::OldOnNotify(wxRequestEvent WXUNUSED(flags))
+void MyFrame::OnAbout(wxCommandEvent& WXUNUSED(event))

 {
 {

-  MySock *sock2 = new MySock();
+  wxMessageBox(_T("wxSocket demo: Server\n")
+               _T("(c) 1999 Guillermo Rodriguez Garcia\n"),
+               _T("About Server"),
+               wxOK | wxICON_INFORMATION, this);
+}

 
 

-  if (!AcceptWith(*sock2))
-    return;
+void MyFrame::Test1(wxSocketBase *sock)
+{
+  unsigned char len;
+  char *buf;
+
+  m_text->AppendText(_T("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 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.
+  //
+  sock->SetFlags(wxSOCKET_WAITALL);
+
+  sock->Read((char *)&len, 1);

 
 

-  m_handler->Register(sock2);
+  buf = (char *)malloc(len);
+  sock->Read(buf, len);
+  sock->Write(buf, len);
+  free(buf);

 
 

-  sock2->SetFlags(NONE);
-  sock2->frame = frame;
-  sock2->SetNotify(REQ_READ | REQ_LOST);
-  sock2->Notify(TRUE);
-  frame->UpdateStatus(1);
+  m_text->AppendText(_T("Test 1 ends\n"));

 }
 
 }
 

-// My frame Constructor
-MyFrame::MyFrame(wxFrame *frame):
-  wxFrame(frame, -1, "wxSocket sample (server)", wxDefaultPosition,
-          wxSize(300, 200))
+void MyFrame::Test2(wxSocketBase *sock)

 {
 {

-  wxIPV4address addr;
-  addr.Service(3000);
+#define MAX_MSG_SIZE 10000

 
 

-  // Init all
-  wxSocketHandler::Master();
+  wxString s;
+  char *buf = (char *)malloc(MAX_MSG_SIZE);
+  wxUint32 len;

 
 

-  sock = new MyServer(addr);
-  wxSocketHandler::Master().Register(sock);
-  sock->frame = this;
-  sock->SetNotify(wxSocketBase::REQ_ACCEPT);
-  sock->Notify(TRUE);
-  nb_clients = 0;
+  m_text->AppendText(_T("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();

 
 

-  CreateStatusBar(1);
-  UpdateStatus(0);
+  s.Printf(_T("Client says: %s\n"), buf);
+  m_text->AppendText(s);
+
+  sock->WriteMsg(buf, len);
+  free(buf);
+
+  m_text->AppendText(_T("Test 2 ends\n"));
+
+#undef MAX_MSG_SIZE

 }
 
 }
 

-MyFrame::~MyFrame()
+void MyFrame::Test3(wxSocketBase *sock)

 {
 {

-  delete sock;
+  m_text->AppendText(_T("Test 3 begins\n"));
+  m_text->AppendText(_T("(not implemented)\n"));
+  m_text->AppendText(_T("Test 3 ends\n"));

 }
 
 }
 

-// Intercept menu commands
-void MyFrame::Menu_Exit(wxCommandEvent& event)
+void MyFrame::OnServerEvent(wxSocketEvent& event)

 {
 {

-  Close(TRUE);
+  wxString s = _T("OnServerEvent: ");
+  wxSocketBase *sock;
+
+  switch(event.SocketEvent())
+  {
+    case wxSOCKET_CONNECTION : s.Append(_T("wxSOCKET_CONNECTION\n")); break;
+    default                  : s.Append(_T("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
+  // non-blocking accept (although if we got here, there
+  // should ALWAYS be a pending connection).
+  //
+  sock = m_server->Accept(FALSE);
+
+  if (sock)
+  {
+    m_text->AppendText(_T("New client connection accepted\n"));
+  }
+  else
+  {
+    m_text->AppendText(_T("Error: couldn't accept a new connection"));
+    return;
+  }
+
+  sock->SetEventHandler(*this, SOCKET_ID);
+  sock->SetNotify(wxSOCKET_INPUT_FLAG | wxSOCKET_LOST_FLAG);
+  sock->Notify(TRUE);
+
+  m_numClients++;
+  UpdateStatusBar();

 }
 
 }
 

-void MyFrame::ExecTest1(wxSocketBase *sock_o)
+void MyFrame::OnSocketEvent(wxSocketEvent& event)

 {
 {

-  char *buf = new char[50];
-  size_t l;
+  wxSocketBase *sock = event.Socket();
+  wxString s = _T("OnSocketEvent: ");
+
+  // We first print a msg
+  switch(event.SocketEvent())
+  {
+    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"));
+  }
+
+  m_text->AppendText(s);
+
+  // Now we process the event
+  switch(event.SocketEvent())
+  {
+    case wxSOCKET_INPUT:
+    {
+      // We disable input events, so that the test doesn't trigger
+      // wxSocketEvent again.
+      sock->SetNotify(wxSOCKET_LOST_FLAG);
+
+      // Which test are we going to run?
+      unsigned char c;
+      sock->Read((char *)&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"));
+      }
+
+      // Enable input events again.
+      sock->SetNotify(wxSOCKET_LOST_FLAG | wxSOCKET_INPUT_FLAG);
+      break;
+    }
+    case wxSOCKET_LOST:
+    {
+      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);
+      break;
+    }
+    default: ;
+  }

 
 

-  l = sock_o->Read(buf, 50).LastCount();
-  sock_o->Write(buf, l);
+  UpdateStatusBar();

 }
 
 }
 

-void MyFrame::UpdateStatus(int incr)
+// convenience functions
+
+void MyFrame::UpdateStatusBar()

 {
 {

-  char s[30];
-  nb_clients += incr;
-  sprintf(s, "%d clients connected", nb_clients);
-  SetStatusText(s);
+  wxString s;
+  s.Printf(_T("%d clients connected"), m_numClients);
+  SetStatusText(s, 1);

 }
 }