src/common/ftp.cpp

   1 /////////////////////////////////////////////////////////////////////////////
   2 // Name:        ftp.cpp
   3 // Purpose:     FTP protocol
   4 // Author:      Guilhem Lavaux
   5 // Modified by:
   6 // Created:     07/07/1997
   7 // RCS-ID:      $Id$
   8 // Copyright:   (c) 1997, 1998 Guilhem Lavaux
   9 // Licence:     wxWindows license
  10 /////////////////////////////////////////////////////////////////////////////
  11
  12 #ifdef __GNUG__
  13 #pragma implementation "ftp.h"
  14 #endif
  15
  16 // For compilers that support precompilation, includes "wx.h".
  17 #include "wx/wxprec.h"
  18
  19 #ifdef __BORLANDC__
  20 #pragma hdrstop
  21 #endif
  22
  23 #include <memory.h>
  24 #include <stdlib.h>
  25 #include "wx/string.h"
  26 #include "wx/utils.h"
  27 // #include "wx/data.h"
  28 #define WXSOCK_INTERNAL
  29 #include "wx/sckaddr.h"
  30 #undef WXSOCK_INTERNAL
  31 #include "wx/socket.h"
  32 #include "wx/url.h"
  33 #include "wx/sckstrm.h"
  34 #include "wx/protocol/protocol.h"
  35 #include "wx/protocol/ftp.h"
  36
  37 #ifdef __BORLANDC__
  38 #pragma hdrstop
  39 #endif
  40
  41 #define FTP_BSIZE 1024
  42
  43 #if !USE_SHARED_LIBRARY
  44 IMPLEMENT_DYNAMIC_CLASS(wxFTP, wxProtocol)
  45 IMPLEMENT_PROTOCOL(wxFTP, "ftp", "ftp", TRUE)
  46 #endif
  47
  48 ////////////////////////////////////////////////////////////////
  49 ////// wxFTP constructor and destructor ////////////////////////
  50 ////////////////////////////////////////////////////////////////
  51
  52 wxFTP::wxFTP()
  53   : wxProtocol()
  54 {
  55   char tmp[256];
  56
  57   m_lastError = wxPROTO_NOERR;
  58   m_streaming = FALSE;
  59
  60   m_user = "anonymous";
  61   wxGetUserName(tmp, 256);
  62   m_passwd.sprintf("%s@",tmp);
  63   wxGetHostName(tmp, 256);
  64   m_passwd += tmp;
  65
  66   SetNotify(0);
  67 }
  68
  69 wxFTP::~wxFTP()
  70 {
  71   SendCommand("QUIT", '2');
  72 }
  73
  74 ////////////////////////////////////////////////////////////////
  75 ////// wxFTP connect and login methods /////////////////////////
  76 ////////////////////////////////////////////////////////////////
  77 bool wxFTP::Connect(wxSockAddress& addr)
  78 {
  79   if (!m_handler) {
  80     m_lastError = wxPROTO_NOHNDLR;
  81     return FALSE;
  82   }
  83
  84   if (!wxProtocol::Connect(addr)) {
  85     m_lastError = wxPROTO_NETERR;
  86     return FALSE;
  87   }
  88
  89   if (!m_user || !m_passwd) {
  90     m_lastError = wxPROTO_CONNERR;
  91     return FALSE;
  92   }
  93
  94   wxString command;
  95
  96   if (!GetResult('2')) {
  97     Close();
  98     return FALSE;
  99   }
 100
 101   command.sprintf("USER %s", (const char *)m_user);
 102   if (!SendCommand(command, '3')) {
 103     Close();
 104     return FALSE;
 105   }
 106
 107   command.sprintf("PASS %s", (const char *)m_passwd);
 108   if (!SendCommand(command, '2')) {
 109     Close();
 110     return FALSE;
 111   }
 112
 113   return TRUE;
 114 }
 115
 116 bool wxFTP::Connect(const wxString& host)
 117 {
 118   wxIPV4address addr;
 119   wxString my_host = host;
 120
 121   addr.Hostname(my_host);
 122   addr.Service("ftp");
 123
 124   return Connect(addr);
 125 }
 126
 127 bool wxFTP::Close()
 128 {
 129   if (m_streaming) {
 130     m_lastError = wxPROTO_STREAMING;
 131     return FALSE;
 132   }
 133   if (m_connected)
 134     SendCommand(wxString("QUIT"), '2');
 135   return wxSocketClient::Close();
 136 }
 137
 138 ////////////////////////////////////////////////////////////////
 139 ////// wxFTP low-level methods /////////////////////////////////
 140 ////////////////////////////////////////////////////////////////
 141 bool wxFTP::SendCommand(const wxString& command, char exp_ret)
 142 {
 143   wxString tmp_str;
 144
 145   if (m_streaming) {
 146     m_lastError = wxPROTO_STREAMING;
 147     return FALSE;
 148   }
 149   tmp_str = command + "\r\n";
 150   if (Write((char *)tmp_str.GetData(), tmp_str.Length()).Error()) {
 151     m_lastError = wxPROTO_NETERR;
 152     return FALSE;
 153   }
 154   return GetResult(exp_ret);
 155 }
 156
 157 bool wxFTP::GetResult(char exp)
 158 {
 159   if ((m_lastError = GetLine(this, m_lastResult)))
 160     return FALSE;
 161   if (m_lastResult.GetChar(0) != exp) {
 162     m_lastError = wxPROTO_PROTERR;
 163     return FALSE;
 164   }
 165
 166   if (m_lastResult.GetChar(3) == '-') {
 167     wxString key = m_lastResult.Left((size_t)3);
 168
 169     key += ' ';
 170
 171     while (m_lastResult.Index(key) != 0) {
 172       if ((m_lastError = GetLine(this, m_lastResult)))
 173         return FALSE;
 174     }
 175   }
 176   return TRUE;
 177 }
 178
 179 ////////////////////////////////////////////////////////////////
 180 ////// wxFTP low-level methods /////////////////////////////////
 181 ////////////////////////////////////////////////////////////////
 182 bool wxFTP::ChDir(const wxString& dir)
 183 {
 184   wxString str = dir;
 185
 186   str.Prepend("CWD ");
 187   return SendCommand(str, '2');
 188 }
 189
 190 bool wxFTP::MkDir(const wxString& dir)
 191 {
 192   wxString str = dir;
 193   str.Prepend("MKD ");
 194   return SendCommand(str, '2');
 195 }
 196
 197 bool wxFTP::RmDir(const wxString& dir)
 198 {
 199   wxString str = dir;
 200
 201   str.Prepend("PWD ");
 202   return SendCommand(str, '2');
 203 }
 204
 205 wxString wxFTP::Pwd()
 206 {
 207   int beg, end;
 208
 209   if (!SendCommand("PWD", '2'))
 210     return wxString((char *)NULL);
 211
 212   beg = m_lastResult.Find('\"',FALSE);
 213   end = m_lastResult.Find('\"',TRUE);
 214
 215   return wxString(beg+1, end);
 216 }
 217
 218 bool wxFTP::Rename(const wxString& src, const wxString& dst)
 219 {
 220   wxString str;
 221
 222   str = "RNFR " + src;
 223   if (!SendCommand(str, '3'))
 224     return FALSE;
 225
 226   str = "RNTO " + dst;
 227   return SendCommand(str, '2');
 228 }
 229
 230 bool wxFTP::RmFile(const wxString& path)
 231 {
 232   wxString str;
 233
 234   str = "DELE ";
 235   str += path;
 236   return SendCommand(str, '2');
 237 }
 238
 239 ////////////////////////////////////////////////////////////////
 240 ////// wxFTP download*upload ///////////////////////////////////
 241 ////////////////////////////////////////////////////////////////
 242
 243 class wxInputFTPStream : public wxSocketInputStream {
 244 public:
 245   wxFTP *m_ftp;
 246
 247   wxInputFTPStream(wxFTP *ftp_clt, wxSocketBase *sock)
 248     : wxSocketInputStream(*sock), m_ftp(ftp_clt) {}
 249   virtual ~wxInputFTPStream(void)
 250   {
 251      if (LastError() != wxStream_NOERROR)
 252        m_ftp->GetResult('2');
 253      else
 254        m_ftp->Abort();
 255      delete m_i_socket;
 256   }
 257 };
 258
 259 class wxOutputFTPStream : public wxSocketOutputStream {
 260 public:
 261   wxFTP *m_ftp;
 262
 263   wxOutputFTPStream(wxFTP *ftp_clt, wxSocketBase *sock)
 264     : wxSocketOutputStream(*sock), m_ftp(ftp_clt) {}
 265   virtual ~wxOutputFTPStream(void)
 266   {
 267      if (LastError() != wxStream_NOERROR)
 268        m_ftp->GetResult('2');
 269      else
 270        m_ftp->Abort();
 271      delete m_o_socket;
 272   }
 273 };
 274
 275 wxSocketClient *wxFTP::GetPort()
 276 {
 277   wxIPV4address addr;
 278   wxSocketClient *client;
 279   struct sockaddr sin;
 280   int a[6];
 281   wxString straddr;
 282   int addr_pos;
 283
 284   if (!SendCommand("PASV", '2'))
 285     return NULL;
 286
 287   sin.sa_family = AF_INET;
 288   addr_pos = m_lastResult.Find('(');
 289   if (addr_pos == -1) {
 290     m_lastError = wxPROTO_PROTERR;
 291     return NULL;
 292   }
 293   straddr = m_lastResult(addr_pos+1, m_lastResult.Length());
 294   sscanf((const char *)straddr,"%d,%d,%d,%d,%d,%d",&a[2],&a[3],&a[4],&a[5],&a[0],&a[1]);
 295   sin.sa_data[2] = (char)a[2];
 296   sin.sa_data[3] = (char)a[3];
 297   sin.sa_data[4] = (char)a[4];
 298   sin.sa_data[5] = (char)a[5];
 299   sin.sa_data[0] = (char)a[0];
 300   sin.sa_data[1] = (char)a[1];
 301
 302   addr.Disassemble(&sin, sizeof(sin));
 303
 304   client = m_handler->CreateClient();
 305   if (!client->Connect(addr)) {
 306     delete client;
 307     return NULL;
 308   }
 309   client->Notify(FALSE);
 310
 311   return client;
 312 }
 313
 314 bool wxFTP::Abort(void)
 315 {
 316   m_streaming = FALSE;
 317   if (!SendCommand("ABOR", '4'))
 318     return FALSE;
 319   return GetResult('2');
 320 }
 321
 322 wxInputStream *wxFTP::GetInputStream(const wxString& path)
 323 {
 324   wxString tmp_str;
 325
 326   if (!SendCommand("TYPE I", '2'))
 327     return NULL;
 328
 329   wxSocketClient *sock = GetPort();
 330
 331   if (!sock) {
 332     m_lastError = wxPROTO_NETERR;
 333     return NULL;
 334   }
 335
 336   tmp_str = "RETR " + path;
 337   if (!SendCommand(tmp_str, '1'))
 338     return NULL;
 339
 340   return new wxInputFTPStream(this, sock);
 341 }
 342
 343 wxOutputStream *wxFTP::GetOutputStream(const wxString& path)
 344 {
 345   wxString tmp_str;
 346
 347   if (!SendCommand("TYPE I", '2'))
 348     return NULL;
 349
 350   wxSocketClient *sock = GetPort();
 351
 352   tmp_str = "STOR " + path;
 353   if (!SendCommand(tmp_str, '1'))
 354     return FALSE;
 355
 356   return new wxOutputFTPStream(this, sock);
 357 }
 358
 359 wxList *wxFTP::GetList(const wxString& wildcard)
 360 {
 361   wxList *file_list = new wxList;
 362   wxSocketBase *sock = GetPort();
 363   wxString tmp_str = "NLST";
 364
 365   if (!wildcard.IsNull())
 366     tmp_str += wildcard;
 367
 368   if (!SendCommand(tmp_str, '1')) {
 369     delete sock;
 370     delete file_list;
 371     return NULL;
 372   }
 373
 374   while (GetLine(sock, tmp_str) == wxPROTO_NOERR) {
 375     file_list->Append((wxObject *)(new wxString(tmp_str)));
 376   }
 377
 378   if (!GetResult('2')) {
 379     delete sock;
 380     file_list->DeleteContents(TRUE);
 381     delete file_list;
 382     return NULL;
 383   }
 384
 385   return file_list;
 386 }