Small bug fix

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

/*
 * File:	client.cpp
 * Purpose:	wxSocket: client demo
 * Author:	LAVAUX Guilhem
 * Created:	June 1997
 * CVS ID:	$Id$
 * Copyright:	(c) 1997, LAVAUX Guilhem
 */

#ifdef __GNUG__
#pragma implementation
#pragma interface
#endif

// For compilers that support precompilation, includes "wx.h".
#include "wx/wxprec.h"

#ifdef __BORLANDC__
#pragma hdrstop
#endif

#ifndef WX_PRECOMP
#include "wx/wx.h"
#endif

#include "wx/wfstream.h"
#include "wx/socket.h"
#include "wx/url.h"
#include "wx/protocol/http.h"
#include "wx/thread.h"
#include "wx/progdlg.h"

#if defined(__WXMOTIF__) || defined(__WXGTK__)
#include "mondrian.xpm"
#endif

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

class MyClient;

// Define a new frame type
class MyFrame: public wxFrame
{ 
  DECLARE_CLASS(MyFrame)
public:
  MyClient *sock;
  int m_good;

  MyFrame(void);
  virtual ~MyFrame();
  void OnCloseTest(wxCommandEvent& evt);
  void OnExecTest1(wxCommandEvent& evt);
  void OnExecUrlTest(wxCommandEvent& evt);
  void OnQuitApp(wxCommandEvent& evt);
  void OnExecOpenConnection(wxCommandEvent& evt);
  void OnExecCloseConnection(wxCommandEvent& evt);
  void OnSocketEvent(wxSocketEvent& evt);
  void UpdateStatus();

  void Download(wxInputStream *input);

  DECLARE_EVENT_TABLE()
};


IMPLEMENT_CLASS(MyFrame, wxFrame)

/*
 * Define a new derived SocketClient
 */
class MyClient: public wxSocketClient
{
public:
  MyFrame *frame;

  void OnNotify(GSocketEventFlags WXUNUSED(flags)) { frame->UpdateStatus(); }
};

// ID for the menu quit command
const int SKDEMO_QUIT    = 101;
const int SKDEMO_CONNECT = 102;
const int SKDEMO_TEST1   = 103;
const int SKDEMO_TEST2   = 104;
const int SKDEMO_CLOSE   = 105;
const int SKDEMO_TEST3   = 106;
const int ID_TEST_CLOSE  = 107;
const int SKDEMO_SCK     = 108;

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();

  // Give it an icon
  frame->SetIcon(wxICON(mondrian));

  // Make a menubar
  wxMenu *file_menu = new wxMenu();

  file_menu->Append(SKDEMO_QUIT, "Exit");
  wxMenuBar *menu_bar = new wxMenuBar;
  menu_bar->Append(file_menu, "File");

  wxMenu *socket_menu = new wxMenu();
  socket_menu->Append(SKDEMO_CONNECT, "Open session");
  socket_menu->AppendSeparator();
  socket_menu->Append(SKDEMO_TEST1, "Start test 1");
  socket_menu->AppendSeparator();
  socket_menu->Append(SKDEMO_CLOSE, "Close session");
  socket_menu->AppendSeparator();
  socket_menu->Append(SKDEMO_TEST3, "Start URL test");

  menu_bar->Append(socket_menu, "Socket");

  frame->SetMenuBar(menu_bar);

  // Make a panel with a message
  (void)new wxPanel(frame, -1, wxPoint(0, 0), wxSize(300, 100));

  // Show the frame
  frame->Show(TRUE);
  
  // Return the main frame window
  return TRUE;
}

/*
 * MyFrame Constructor
 */
MyFrame::MyFrame():
  wxFrame(NULL, -1, "wxSocket client demo",
          wxDefaultPosition, wxSize(300, 200), wxDEFAULT_FRAME_STYLE)
{
  sock = new MyClient();
  sock->SetFlags((wxSocketBase::wxSockFlags) (wxSocketBase::WAITALL | wxSocketBase::SPEED));
  sock->frame = this;
  sock->SetNotify(wxSOCKET_LOST_FLAG);
  CreateStatusBar(2);
  UpdateStatus();
}

MyFrame::~MyFrame()
{
  delete sock;
}

void MyFrame::OnQuitApp(wxCommandEvent& WXUNUSED(evt))
{
  Close(TRUE);
}

void MyFrame::OnExecOpenConnection(wxCommandEvent& WXUNUSED(evt))
{
  wxIPV4address addr;
    
  if (sock->IsConnected())
    sock->Close();

  wxString hname = wxGetTextFromUser("Enter the address of the wxSocket Sample Server", 
                                  "Connect ...", "localhost");
  addr.Hostname(hname);
  addr.Service(3000);
  sock->SetNotify(0);
  sock->Connect(addr, TRUE);
  sock->SetFlags(wxSocketBase::NONE);
  if (!sock->IsConnected())
    wxMessageBox("Can't connect to the specified host", "Alert !");

  UpdateStatus();
}

void MyFrame::OnExecCloseConnection(wxCommandEvent& WXUNUSED(evt))
{
  sock->Close();
  UpdateStatus();
}

BEGIN_EVENT_TABLE(MyFrame, wxFrame)
  EVT_BUTTON(ID_TEST_CLOSE, MyFrame::OnCloseTest)
  EVT_MENU(SKDEMO_TEST1, MyFrame::OnExecTest1)
  EVT_MENU(SKDEMO_TEST3, MyFrame::OnExecUrlTest)
  EVT_MENU(SKDEMO_QUIT, MyFrame::OnQuitApp)
  EVT_MENU(SKDEMO_CONNECT, MyFrame::OnExecOpenConnection)
  EVT_MENU(SKDEMO_CLOSE, MyFrame::OnExecCloseConnection)
  EVT_SOCKET(SKDEMO_SCK, MyFrame::OnSocketEvent)
END_EVENT_TABLE()

class MyFrameSocketTimer: public wxTimer {
 public:
  void Notify() {
    *m_var = 0;
  }

  int *m_var;
};

void MyFrame::OnSocketEvent(wxSocketEvent& evt)
{
  m_good = 1;
}

void MyFrame::OnCloseTest(wxCommandEvent& evt)
{
  wxButton *button = (wxButton *)evt.GetEventObject();
  wxDialog *dlg = (wxDialog *)button->GetParent();

  dlg->EndModal(0);
}

void MyFrame::UpdateStatus()
{
  if (!sock->IsConnected()) {
    SetStatusText("Not connected", 0);
    SetStatusText("", 1);
  } else {
    wxIPV4address addr;
    wxChar s[100];

    sock->GetPeer(addr);
    wxSprintf(s, _T("Connected to %s"), WXSTRINGCAST addr.Hostname());
    SetStatusText(s, 0);
    wxSprintf(s, _T("Service: %d"), addr.Service());
    SetStatusText(s, 1);
  } 
}

void MyFrame::OnExecTest1(wxCommandEvent& WXUNUSED(evt))
{
  if (!sock->IsConnected())
    return;

  wxDialog *dlgbox = new wxDialog(this, -1, "Test 1", wxDefaultPosition, wxSize(414, 250));
  wxTextCtrl *text_win = new wxTextCtrl(dlgbox, -1, "",
                                        wxPoint(0, 0), wxSize(400, 200),
					wxTE_MULTILINE);
  (void)new wxButton(dlgbox, ID_TEST_CLOSE, "Close",
                     wxPoint(100, 210), wxSize(100, -1));
  wxChar *buf, *buf2;

  dlgbox->Layout();
  dlgbox->Show(TRUE);

  text_win->WriteText("Initializing test 1 ...\n");
  
  wxYield();
  
  /* Init */
  buf = copystring(_T("Hi ! Hi ! Hi !\n"));
  buf2 = new wxChar[wxStrlen(buf)+1];
  char c = 0xbe;
  sock->Write(&c, 1);

  /* No 1 */
  text_win->WriteText("Sending some byte to the server ...");
  wxYield();
  sock->Write((char *)buf, wxStrlen(buf)+1);
  text_win->WriteText("done\n");
  wxYield();
  text_win->WriteText("Receiving some byte from the server ...");
  wxYield();
  sock->Read((char *)buf2, wxStrlen(buf)+1);
  text_win->WriteText("done\n");
  wxYield();
  
  text_win->WriteText("Comparing the two buffers ...");
  if (memcmp(buf, buf2, wxStrlen(buf)+1) != 0) {
    text_win->WriteText("Fail\n");
    sock->Close();
    UpdateStatus();
  } else
    text_win->WriteText("done\nTest 1A passed !\n");

  /* No 2 */
  sock->SetEventHandler(*this, SKDEMO_SCK);
  sock->SetNotify(wxSOCKET_INPUT_FLAG | wxSOCKET_LOST_FLAG);
  sock->Notify(TRUE);

  text_win->WriteText("Test 1B: sending bytes to the server\n");
  wxYield();
  sock->Write((char *)buf, wxStrlen(buf)+1);
  text_win->WriteText("Waiting for incoming bytes (timeout = 2 sec) ...");
  wxYield();
  
  m_good = 2;

  MyFrameSocketTimer timer;

  timer.m_var = &m_good; 
  timer.Start(2000, TRUE);

  while (m_good == 2)
   wxYield();

  if (!m_good) {
    text_win->WriteText("Timeout ! Failed.\n");
    sock->Close();
    UpdateStatus();
  } else
    text_win->WriteText("Success.");

  sock->Read((char *)buf2, wxStrlen(buf)+1);

  dlgbox->Layout();
  dlgbox->ShowModal();

  delete [] buf;
  delete [] buf2;
  delete text_win;
  delete dlgbox;
}


void MyFrame::Download(wxInputStream *input)
{
  wxProgressDialog progress("Downloading ...", "0% downloaded");
  wxFileOutputStream f_out("test.url");
  size_t downloaded;
  int BUFSIZE, bytes_read;
  size_t file_size;
  wxString message;
  int percents;

  char *buf;

  if (input->GetSize() == (size_t)-1) {
    file_size = (size_t)-1;
    bytes_read = BUFSIZE = 10240;
  } else {
    file_size = input->GetSize();
    if (file_size > 10240)
      bytes_read = BUFSIZE = file_size / 1024;
    else
      bytes_read = BUFSIZE = 1024;
  }
  buf = new char[BUFSIZE];

  downloaded = 0;
  bytes_read = BUFSIZE;
  while (downloaded < file_size && bytes_read != 0) {
    bytes_read = input->Read(buf, BUFSIZE).LastRead();
    f_out.Write(buf, bytes_read);
    downloaded += bytes_read;

    percents = downloaded * 100 / file_size;

    message = _T("");
    message << percents << _T("% downloaded");
    progress.Update(percents, message);
  }

  delete[] buf;   
}

void MyFrame::OnExecUrlTest(wxCommandEvent& WXUNUSED(evt))
{
  wxString urlname = wxGetTextFromUser("Enter an URL to get",
                                     "URL:", "http://localhost");

  wxURL url(urlname);
  wxInputStream *datas = url.GetInputStream();

  if (!datas) {
    wxString error;
    error.Printf(_T("Error in getting data from the URL. (error = %d)"), url.GetError());
    wxMessageBox(error, "Alert !");
  } else {
    Download(datas);

    delete datas;
  }
}