samples/wxsocket/client.cpp

   1 /*
   2  * File:        client.cpp
   3  * Purpose:     wxSocket: client demo
   4  * Author:      LAVAUX Guilhem
   5  * Created:     June 1997
   6  * CVS ID:      $Id$
   7  * Copyright:   (c) 1997, LAVAUX Guilhem
   8  */
   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
  26 #include "wx/wfstream.h"
  27 #include "wx/socket.h"
  28 #include "wx/url.h"
  29 #include "wx/protocol/http.h"
  30 #include "wx/thread.h"
  31 #include "wx/progdlg.h"
  32
  33 #if defined(__WXMOTIF__) || defined(__WXGTK__)
  34 #include "mondrian.xpm"
  35 #endif
  36
  37 // Define a new application type
  38 class MyApp: public wxApp
  39 { public:
  40     virtual bool OnInit(void);
  41 };
  42
  43 class MyClient;
  44
  45 // Define a new frame type
  46 class MyFrame: public wxFrame
  47 {
  48   DECLARE_CLASS(MyFrame)
  49 public:
  50   MyClient *sock;
  51   int m_good;
  52
  53   MyFrame(void);
  54   virtual ~MyFrame();
  55   void OnCloseTest(wxCommandEvent& evt);
  56   void OnExecTest1(wxCommandEvent& evt);
  57   void OnExecUrlTest(wxCommandEvent& evt);
  58   void OnQuitApp(wxCommandEvent& evt);
  59   void OnExecOpenConnection(wxCommandEvent& evt);
  60   void OnExecCloseConnection(wxCommandEvent& evt);
  61   void OnSocketEvent(wxSocketEvent& evt);
  62   void UpdateStatus();
  63
  64   void Download(wxInputStream *input);
  65
  66   DECLARE_EVENT_TABLE()
  67 };
  68
  69
  70 IMPLEMENT_CLASS(MyFrame, wxFrame)
  71
  72 /*
  73  * Define a new derived SocketClient
  74  */
  75 class MyClient: public wxSocketClient
  76 {
  77 public:
  78   MyFrame *frame;
  79
  80   void OnNotify(GSocketEventFlags WXUNUSED(flags)) { frame->UpdateStatus(); }
  81 };
  82
  83 // ID for the menu quit command
  84 const int SKDEMO_QUIT    = 101;
  85 const int SKDEMO_CONNECT = 102;
  86 const int SKDEMO_TEST1   = 103;
  87 const int SKDEMO_TEST2   = 104;
  88 const int SKDEMO_CLOSE   = 105;
  89 const int SKDEMO_TEST3   = 106;
  90 const int ID_TEST_CLOSE  = 107;
  91 const int SKDEMO_SCK     = 108;
  92
  93 IMPLEMENT_APP(MyApp)
  94
  95 /*
  96  * `Main program' equivalent, creating windows and returning main app frame
  97  */
  98 bool MyApp::OnInit(void)
  99 {
 100   // Create the main frame window
 101   MyFrame *frame = new MyFrame();
 102
 103   // Give it an icon
 104   frame->SetIcon(wxICON(mondrian));
 105
 106   // Make a menubar
 107   wxMenu *file_menu = new wxMenu();
 108
 109   file_menu->Append(SKDEMO_QUIT, "Exit");
 110   wxMenuBar *menu_bar = new wxMenuBar;
 111   menu_bar->Append(file_menu, "File");
 112
 113   wxMenu *socket_menu = new wxMenu();
 114   socket_menu->Append(SKDEMO_CONNECT, "Open session");
 115   socket_menu->AppendSeparator();
 116   socket_menu->Append(SKDEMO_TEST1, "Start test 1");
 117   socket_menu->AppendSeparator();
 118   socket_menu->Append(SKDEMO_CLOSE, "Close session");
 119   socket_menu->AppendSeparator();
 120   socket_menu->Append(SKDEMO_TEST3, "Start URL test");
 121
 122   menu_bar->Append(socket_menu, "Socket");
 123
 124   frame->SetMenuBar(menu_bar);
 125
 126   // Make a panel with a message
 127   (void)new wxPanel(frame, -1, wxPoint(0, 0), wxSize(300, 100));
 128
 129   // Show the frame
 130   frame->Show(TRUE);
 131
 132   // Return the main frame window
 133   return TRUE;
 134 }
 135
 136 /*
 137  * MyFrame Constructor
 138  */
 139 MyFrame::MyFrame():
 140   wxFrame(NULL, -1, "wxSocket client demo",
 141           wxDefaultPosition, wxSize(300, 200), wxDEFAULT_FRAME_STYLE)
 142 {
 143   sock = new MyClient();
 144   sock->SetFlags((wxSocketBase::wxSockFlags) (wxSocketBase::WAITALL | wxSocketBase::SPEED));
 145   sock->frame = this;
 146   sock->SetNotify(wxSOCKET_LOST_FLAG);
 147   CreateStatusBar(2);
 148   UpdateStatus();
 149 }
 150
 151 MyFrame::~MyFrame()
 152 {
 153   delete sock;
 154 }
 155
 156 void MyFrame::OnQuitApp(wxCommandEvent& WXUNUSED(evt))
 157 {
 158   Close(TRUE);
 159 }
 160
 161 void MyFrame::OnExecOpenConnection(wxCommandEvent& WXUNUSED(evt))
 162 {
 163   wxIPV4address addr;
 164
 165   if (sock->IsConnected())
 166     sock->Close();
 167
 168   wxString hname = wxGetTextFromUser("Enter the address of the wxSocket Sample Server",
 169                                   "Connect ...", "localhost");
 170   addr.Hostname(hname);
 171   addr.Service(3000);
 172   sock->SetNotify(0);
 173   sock->Connect(addr, TRUE);
 174   sock->SetFlags(wxSocketBase::NONE);
 175   if (!sock->IsConnected())
 176     wxMessageBox("Can't connect to the specified host", "Alert !");
 177
 178   UpdateStatus();
 179 }
 180
 181 void MyFrame::OnExecCloseConnection(wxCommandEvent& WXUNUSED(evt))
 182 {
 183   sock->Close();
 184   UpdateStatus();
 185 }
 186
 187 BEGIN_EVENT_TABLE(MyFrame, wxFrame)
 188   EVT_BUTTON(ID_TEST_CLOSE, MyFrame::OnCloseTest)
 189   EVT_MENU(SKDEMO_TEST1, MyFrame::OnExecTest1)
 190   EVT_MENU(SKDEMO_TEST3, MyFrame::OnExecUrlTest)
 191   EVT_MENU(SKDEMO_QUIT, MyFrame::OnQuitApp)
 192   EVT_MENU(SKDEMO_CONNECT, MyFrame::OnExecOpenConnection)
 193   EVT_MENU(SKDEMO_CLOSE, MyFrame::OnExecCloseConnection)
 194   EVT_SOCKET(SKDEMO_SCK, MyFrame::OnSocketEvent)
 195 END_EVENT_TABLE()
 196
 197 class MyFrameSocketTimer: public wxTimer {
 198  public:
 199   void Notify() {
 200     *m_var = 0;
 201   }
 202
 203   int *m_var;
 204 };
 205
 206 void MyFrame::OnSocketEvent(wxSocketEvent& evt)
 207 {
 208   m_good = 1;
 209 }
 210
 211 void MyFrame::OnCloseTest(wxCommandEvent& evt)
 212 {
 213   wxButton *button = (wxButton *)evt.GetEventObject();
 214   wxDialog *dlg = (wxDialog *)button->GetParent();
 215
 216   dlg->EndModal(0);
 217 }
 218
 219 void MyFrame::UpdateStatus()
 220 {
 221   if (!sock->IsConnected()) {
 222     SetStatusText("Not connected", 0);
 223     SetStatusText("", 1);
 224   } else {
 225     wxIPV4address addr;
 226     wxChar s[100];
 227
 228     sock->GetPeer(addr);
 229     wxSprintf(s, _T("Connected to %s"), WXSTRINGCAST addr.Hostname());
 230     SetStatusText(s, 0);
 231     wxSprintf(s, _T("Service: %d"), addr.Service());
 232     SetStatusText(s, 1);
 233   }
 234 }
 235
 236 void MyFrame::OnExecTest1(wxCommandEvent& WXUNUSED(evt))
 237 {
 238   if (!sock->IsConnected())
 239     return;
 240
 241   wxDialog *dlgbox = new wxDialog(this, -1, "Test 1", wxDefaultPosition, wxSize(414, 250));
 242   wxTextCtrl *text_win = new wxTextCtrl(dlgbox, -1, "",
 243                                         wxPoint(0, 0), wxSize(400, 200),
 244                                         wxTE_MULTILINE);
 245   (void)new wxButton(dlgbox, ID_TEST_CLOSE, "Close",
 246                      wxPoint(100, 210), wxSize(100, -1));
 247   wxChar *buf, *buf2;
 248
 249   dlgbox->Layout();
 250   dlgbox->Show(TRUE);
 251
 252   text_win->WriteText("Initializing test 1 ...\n");
 253
 254   wxYield();
 255
 256   /* Init */
 257   buf = copystring(_T("Hi ! Hi ! Hi !\n"));
 258   buf2 = new wxChar[wxStrlen(buf)+1];
 259   char c = 0xbe;
 260   sock->Write(&c, 1);
 261
 262   /* No 1 */
 263   text_win->WriteText("Sending some byte to the server ...");
 264   wxYield();
 265   sock->Write((char *)buf, wxStrlen(buf)+1);
 266   text_win->WriteText("done\n");
 267   wxYield();
 268   text_win->WriteText("Receiving some byte from the server ...");
 269   wxYield();
 270   sock->Read((char *)buf2, wxStrlen(buf)+1);
 271   text_win->WriteText("done\n");
 272   wxYield();
 273
 274   text_win->WriteText("Comparing the two buffers ...");
 275   if (memcmp(buf, buf2, wxStrlen(buf)+1) != 0) {
 276     text_win->WriteText("Fail\n");
 277     sock->Close();
 278     UpdateStatus();
 279   } else
 280     text_win->WriteText("done\nTest 1A passed !\n");
 281
 282   /* No 2 */
 283   sock->SetEventHandler(*this, SKDEMO_SCK);
 284   sock->SetNotify(wxSOCKET_INPUT_FLAG | wxSOCKET_LOST_FLAG);
 285   sock->Notify(TRUE);
 286
 287   text_win->WriteText("Test 1B: sending bytes to the server\n");
 288   wxYield();
 289   sock->Write((char *)buf, wxStrlen(buf)+1);
 290   text_win->WriteText("Waiting for incoming bytes (timeout = 2 sec) ...");
 291   wxYield();
 292
 293   m_good = 2;
 294
 295   MyFrameSocketTimer timer;
 296
 297   timer.m_var = &m_good;
 298   timer.Start(2000, TRUE);
 299
 300   while (m_good == 2)
 301    wxYield();
 302
 303   if (!m_good) {
 304     text_win->WriteText("Timeout ! Failed.\n");
 305     sock->Close();
 306     UpdateStatus();
 307   } else
 308     text_win->WriteText("Success.");
 309
 310   sock->Read((char *)buf2, wxStrlen(buf)+1);
 311
 312   dlgbox->Layout();
 313   dlgbox->ShowModal();
 314
 315   delete [] buf;
 316   delete [] buf2;
 317   delete text_win;
 318   delete dlgbox;
 319 }
 320
 321
 322 void MyFrame::Download(wxInputStream *input)
 323 {
 324   wxProgressDialog progress("Downloading ...", "0% downloaded");
 325   wxFileOutputStream f_out("test.url");
 326   size_t downloaded;
 327   int BUFSIZE, bytes_read;
 328   size_t file_size;
 329   wxString message;
 330   int percents;
 331
 332   char *buf;
 333
 334   if (input->GetSize() == (size_t)-1) {
 335     file_size = (size_t)-1;
 336     bytes_read = BUFSIZE = 10240;
 337   } else {
 338     file_size = input->GetSize();
 339     if (file_size > 10240)
 340       bytes_read = BUFSIZE = file_size / 1024;
 341     else
 342       bytes_read = BUFSIZE = 1024;
 343   }
 344   buf = new char[BUFSIZE];
 345
 346   downloaded = 0;
 347   bytes_read = BUFSIZE;
 348   while (downloaded < file_size && bytes_read != 0) {
 349     bytes_read = input->Read(buf, BUFSIZE).LastRead();
 350     f_out.Write(buf, bytes_read);
 351     downloaded += bytes_read;
 352
 353     percents = downloaded * 100 / file_size;
 354
 355     message = _T("");
 356     message << percents << _T("% downloaded");
 357     progress.Update(percents, message);
 358   }
 359
 360   delete[] buf;
 361 }
 362
 363 void MyFrame::OnExecUrlTest(wxCommandEvent& WXUNUSED(evt))
 364 {
 365   wxString urlname = wxGetTextFromUser("Enter an URL to get",
 366                                      "URL:", "http://localhost");
 367
 368   wxURL url(urlname);
 369   wxInputStream *datas = url.GetInputStream();
 370
 371   if (!datas) {
 372     wxString error;
 373     error.Printf(_T("Error in getting data from the URL. (error = %d)"), url.GetError());
 374     wxMessageBox(error, "Alert !");
 375   } else {
 376     Download(datas);
 377
 378     delete datas;
 379   }
 380 }