samples/wxsocket/client.cpp

   1 /*
   2  * File:        client.cpp
   3  * Purpose:     wxSocket: client demo
   4  * Author:      LAVAUX Guilhem (from minimal.cc)
   5  * Created:     June 1997
   6  * Updated:
   7  * Copyright:   (c) 1993, AIAI, University of Edinburgh
   8  *              (C) 1997, LAVAUX Guilhem
   9  */
  10 #ifdef __GNUG__
  11 #pragma implementation
  12 #pragma interface
  13 #endif
  14
  15 // For compilers that support precompilation, includes "wx.h".
  16 #include "wx/wxprec.h"
  17
  18 #ifdef __BORLANDC__
  19 #pragma hdrstop
  20 #endif
  21
  22 #ifndef WX_PRECOMP
  23 #include "wx/wx.h"
  24 #endif
  25 #include "wx/socket.h"
  26 #include "wx/url.h"
  27 #include "wx/protocol/http.h"
  28
  29 // Define a new application type
  30 class MyApp: public wxApp
  31 { public:
  32     virtual bool OnInit(void);
  33 };
  34
  35 class MyClient;
  36
  37 // Define a new frame type
  38 class MyFrame: public wxFrame
  39 {
  40   DECLARE_CLASS(MyFrame)
  41 public:
  42   MyClient *sock;
  43
  44   MyFrame(void);
  45   virtual ~MyFrame();
  46   void OnCloseTest(wxCommandEvent& evt);
  47   void OnExecTest1(wxCommandEvent& evt);
  48   void OnExecUrlTest(wxCommandEvent& evt);
  49   void OnQuitApp(wxCommandEvent& evt);
  50   void OnExecOpenConnection(wxCommandEvent& evt);
  51   void OnExecCloseConnection(wxCommandEvent& evt);
  52   void UpdateStatus();
  53
  54   DECLARE_EVENT_TABLE()
  55 };
  56
  57
  58 IMPLEMENT_CLASS(MyFrame, wxFrame)
  59
  60 /*
  61  * Define a new derived SocketClient
  62  */
  63 class MyClient: public wxSocketClient
  64 {
  65 public:
  66   MyFrame *frame;
  67
  68   void OnNotify(wxRequestNotify WXUNUSED(flags)) { frame->UpdateStatus(); }
  69 };
  70
  71 // ID for the menu quit command
  72 const SKDEMO_QUIT    = 101;
  73 const SKDEMO_CONNECT = 102;
  74 const SKDEMO_TEST1   = 103;
  75 const SKDEMO_TEST2   = 104;
  76 const SKDEMO_CLOSE   = 105;
  77 const SKDEMO_TEST3   = 106;
  78 const ID_TEST_CLOSE  = 107;
  79
  80 IMPLEMENT_APP(MyApp)
  81
  82 /*
  83  * `Main program' equivalent, creating windows and returning main app frame
  84  */
  85 bool MyApp::OnInit(void)
  86 {
  87   // Create the main frame window
  88   MyFrame *frame = new MyFrame();
  89
  90   // Give it an icon
  91 #ifdef wx_msw
  92   frame->SetIcon(new wxIcon("mondrian"));
  93 #endif
  94 #ifdef wx_x
  95   frame->SetIcon(new wxIcon("mondrian.xbm"));
  96 #endif
  97
  98   // Make a menubar
  99   wxMenu *file_menu = new wxMenu();
 100
 101   file_menu->Append(SKDEMO_QUIT, "Exit");
 102   wxMenuBar *menu_bar = new wxMenuBar;
 103   menu_bar->Append(file_menu, "File");
 104
 105   wxMenu *socket_menu = new wxMenu();
 106   socket_menu->Append(SKDEMO_CONNECT, "Open session");
 107   socket_menu->AppendSeparator();
 108   socket_menu->Append(SKDEMO_TEST1, "Start test 1");
 109   socket_menu->AppendSeparator();
 110   socket_menu->Append(SKDEMO_CLOSE, "Close session");
 111   socket_menu->AppendSeparator();
 112   socket_menu->Append(SKDEMO_TEST3, "Start URL test");
 113
 114   menu_bar->Append(socket_menu, "Socket");
 115
 116   frame->SetMenuBar(menu_bar);
 117
 118   // Make a panel with a message
 119   (void)new wxPanel(frame, 0, 0, 300, 100);
 120
 121   // Show the frame
 122   frame->Show(TRUE);
 123
 124   // Return the main frame window
 125   return TRUE;
 126 }
 127
 128 /*
 129  * MyFrame Constructor
 130  */
 131 MyFrame::MyFrame():
 132   wxFrame(NULL, -1, "wxSocket client demo",
 133           wxDefaultPosition, wxSize(300, 200), wxDEFAULT_FRAME_STYLE)
 134 {
 135   // Init all
 136   wxSocketHandler::Master();
 137
 138   sock = new MyClient();
 139   sock->SetFlags(wxSocketBase::WAITALL);
 140   wxSocketHandler::Master().Register(sock);
 141   sock->frame = this;
 142   sock->SetNotify(wxSocketBase::REQ_LOST);
 143   CreateStatusBar(2);
 144   UpdateStatus();
 145 }
 146
 147 MyFrame::~MyFrame()
 148 {
 149   delete sock;
 150 }
 151
 152 void MyFrame::OnQuitApp(wxCommandEvent& WXUNUSED(evt))
 153 {
 154   Close(TRUE);
 155 }
 156
 157 void MyFrame::OnExecOpenConnection(wxCommandEvent& WXUNUSED(evt))
 158 {
 159   wxIPV4address addr;
 160
 161   if (sock->IsConnected())
 162     sock->Close();
 163
 164   wxString hname = wxGetTextFromUser("Enter the address of the wxSocket Sample Server",
 165                                   "Connect ...", "localhost");
 166   addr.Hostname(hname);
 167   addr.Service(3000);
 168   sock->SetNotify(0);
 169   sock->Connect(addr, TRUE);
 170   if (!sock->IsConnected())
 171     wxMessageBox("Can't connect to the specified host", "Alert !");
 172
 173   UpdateStatus();
 174 }
 175
 176 void MyFrame::OnExecCloseConnection(wxCommandEvent& WXUNUSED(evt))
 177 {
 178   if (sock)
 179     sock->Close();
 180   UpdateStatus();
 181 }
 182
 183 BEGIN_EVENT_TABLE(MyFrame, wxFrame)
 184   EVT_BUTTON(ID_TEST_CLOSE, MyFrame::OnCloseTest)
 185   EVT_MENU(SKDEMO_TEST1, MyFrame::OnExecTest1)
 186   EVT_MENU(SKDEMO_TEST3, MyFrame::OnExecUrlTest)
 187   EVT_MENU(SKDEMO_QUIT, MyFrame::OnQuitApp)
 188   EVT_MENU(SKDEMO_CONNECT, MyFrame::OnExecOpenConnection)
 189   EVT_MENU(SKDEMO_CLOSE, MyFrame::OnExecCloseConnection)
 190 END_EVENT_TABLE()
 191
 192 void MyFrame::OnCloseTest(wxCommandEvent& evt)
 193 {
 194   wxButton *button = (wxButton *)evt.GetEventObject();
 195   wxDialog *dlg = (wxDialog *)button->GetParent();
 196
 197   dlg->EndModal(0);
 198 }
 199
 200 void MyFrame::UpdateStatus()
 201 {
 202   if (!sock->IsConnected()) {
 203     SetStatusText("Not connected", 0);
 204     SetStatusText("", 1);
 205   } else {
 206     wxIPV4address addr;
 207     char s[100];
 208
 209     sock->GetPeer(addr);
 210     sprintf(s, "Connected to %s", (const char *)addr.Hostname());
 211     SetStatusText(s, 0);
 212     sprintf(s, "Service: %d", addr.Service());
 213     SetStatusText(s, 1);
 214   }
 215 }
 216
 217 void MyFrame::OnExecTest1(wxCommandEvent& WXUNUSED(evt))
 218 {
 219   if (!sock->IsConnected())
 220     return;
 221
 222   wxDialog *dlgbox = new wxDialog(this, -1, "Test 1", wxDefaultPosition, wxSize(410, 270));
 223   wxTextCtrl *text_win = new wxTextCtrl(dlgbox, -1, "",
 224                                         wxPoint(0, 0), wxSize(400, 200),
 225                                         wxTE_MULTILINE);
 226   (void)new wxButton(dlgbox, ID_TEST_CLOSE, "Close",
 227                      wxPoint(100, 210), wxSize(100, 40));
 228   char *buf, *buf2;
 229
 230   dlgbox->Layout();
 231   dlgbox->Show(TRUE);
 232
 233   text_win->WriteText("Initializing test 1 ...\n");
 234
 235   /* Init */
 236   buf = copystring("Salut ! Salut ! Salut ! Salut Toto\n");
 237   buf2 = new char[strlen(buf)+1];
 238   char c = 0xbe;
 239   sock->WriteMsg(&c, 1);
 240
 241   /* No 1 */
 242   text_win->WriteText("Sending some byte to the server ...");
 243   sock->Write(buf, strlen(buf)+1);
 244   text_win->WriteText("done\n");
 245   text_win->WriteText("Receiving some byte from the server ...");
 246   sock->Read(buf2, strlen(buf)+1);
 247   text_win->WriteText("done\n");
 248
 249   text_win->WriteText("Comparing the two buffers ...");
 250   if (memcmp(buf, buf2, strlen(buf)+1) != 0) {
 251     text_win->WriteText("Fail\n");
 252     sock->Close();
 253     UpdateStatus();
 254   } else
 255     text_win->WriteText("done\nTest 1 passed !\n");
 256
 257   dlgbox->Layout();
 258   dlgbox->ShowModal();
 259
 260   delete [] buf;
 261   delete [] buf2;
 262   delete text_win;
 263   delete dlgbox;
 264 }
 265
 266 void MyFrame::OnExecUrlTest(wxCommandEvent& WXUNUSED(evt))
 267 {
 268   wxString urlname = wxGetTextFromUser("Enter an URL to get",
 269                                      "URL:", "http://localhost");
 270
 271   wxURL url(urlname);
 272   wxInputStream *datas = url.GetInputStream();
 273
 274   if (!datas)
 275     wxMessageBox("Error in getting data from the URL.", "Alert !");
 276   else {
 277     wxMessageBox("Success !! Click on OK to see the text.", "OK");
 278     delete datas;
 279   }
 280 }