samples/wxsocket/client.cpp

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