Another place we need an autorelease pool.

[wxWidgets.git] / src / common / ftp.cpp

diff --git a/src/common/ftp.cpp b/src/common/ftp.cpp
index 5edc4b2e8bc775ec19be6c3b8d5407bc60baa29f..9b1cd2513b62964d71b2a02b112eeadf84ccaf04 100644 (file)
--- a/src/common/ftp.cpp
+++ b/src/common/ftp.cpp
@@ -1,5 +1,5 @@
 /////////////////////////////////////////////////////////////////////////////
 /////////////////////////////////////////////////////////////////////////////

-// Name:        ftp.cpp
+// Name:        common/ftp.cpp

 // Purpose:     FTP protocol
 // Author:      Guilhem Lavaux
 // Modified by: Mark Johnson, wxWindows@mj10777.de
 // Purpose:     FTP protocol
 // Author:      Guilhem Lavaux
 // Modified by: Mark Johnson, wxWindows@mj10777.de


@@ -7,9 +7,11 @@
 //              Vadim Zeitlin (numerous fixes and rewrites to all part of the
 //              code, support ASCII/Binary modes, better error reporting, more
 //              robust Abort(), support for arbitrary FTP commands, ...)
 //              Vadim Zeitlin (numerous fixes and rewrites to all part of the
 //              code, support ASCII/Binary modes, better error reporting, more
 //              robust Abort(), support for arbitrary FTP commands, ...)

+//              Randall Fox (support for active mode)

 // Created:     07/07/1997
 // RCS-ID:      $Id$
 // Copyright:   (c) 1997, 1998 Guilhem Lavaux
 // Created:     07/07/1997
 // RCS-ID:      $Id$
 // Copyright:   (c) 1997, 1998 Guilhem Lavaux

+//              (c) 1998-2004 wxWidgets team

 // Licence:     wxWindows licence
 /////////////////////////////////////////////////////////////////////////////
 
 // Licence:     wxWindows licence
 /////////////////////////////////////////////////////////////////////////////
 


@@ -17,10 +19,6 @@
 // declarations
 // ============================================================================
 
 // declarations
 // ============================================================================
 

-#if defined(__GNUG__) && !defined(NO_GCC_PRAGMA)
-  #pragma implementation "ftp.h"
-#endif
-

 // ----------------------------------------------------------------------------
 // headers
 // ----------------------------------------------------------------------------
 // ----------------------------------------------------------------------------
 // headers
 // ----------------------------------------------------------------------------


@@ -40,6 +38,7 @@
     #include "wx/utils.h"
     #include "wx/log.h"
     #include "wx/intl.h"
     #include "wx/utils.h"
     #include "wx/log.h"
     #include "wx/intl.h"

+    #include "wx/wxcrtvararg.h"

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


@@ -49,10 +48,6 @@
 #include "wx/protocol/protocol.h"
 #include "wx/protocol/ftp.h"
 
 #include "wx/protocol/protocol.h"
 #include "wx/protocol/ftp.h"
 

-#if defined(__WXMAC__)
-    #include "wx/mac/macsock.h"
-#endif
-

 #ifndef __MWERKS__
     #include <memory.h>
 #endif
 #ifndef __MWERKS__
     #include <memory.h>
 #endif


@@ -69,7 +64,7 @@ static const size_t LEN_CODE = 3;
 // ----------------------------------------------------------------------------
 
 IMPLEMENT_DYNAMIC_CLASS(wxFTP, wxProtocol)
 // ----------------------------------------------------------------------------
 
 IMPLEMENT_DYNAMIC_CLASS(wxFTP, wxProtocol)

-IMPLEMENT_PROTOCOL(wxFTP, wxT("ftp"), wxT("ftp"), TRUE)
+IMPLEMENT_PROTOCOL(wxFTP, wxT("ftp"), wxT("ftp"), true)

 
 // ============================================================================
 // implementation
 
 // ============================================================================
 // implementation


@@ -81,24 +76,28 @@ IMPLEMENT_PROTOCOL(wxFTP, wxT("ftp"), wxT("ftp"), TRUE)
 
 wxFTP::wxFTP()
 {
 
 wxFTP::wxFTP()
 {

-    m_lastError = wxPROTO_NOERR;
-    m_streaming = FALSE;
+    m_streaming = false;

     m_currentTransfermode = NONE;
 
     m_currentTransfermode = NONE;
 

-    m_user = wxT("anonymous");
-    m_passwd << wxGetUserId() << wxT('@') << wxGetFullHostName();
+    m_username = wxT("anonymous");
+    m_password << wxGetUserId() << wxT('@') << wxGetFullHostName();

 
     SetNotify(0);
 
     SetNotify(0);

-    SetFlags(wxSOCKET_NONE);
+    SetFlags(wxSOCKET_NOWAIT);
+    m_bPassive = true;
+    m_bEncounteredError = false;

 }
 
 wxFTP::~wxFTP()
 {
     if ( m_streaming )
     {
 }
 
 wxFTP::~wxFTP()
 {
     if ( m_streaming )
     {

+        // if we are streaming, this will issue
+        // an FTP ABORT command, to tell the server we are aborting

         (void)Abort();
     }
 
         (void)Abort();
     }
 

+    // now this issues a "QUIT" command to tell the server we are

     Close();
 }
 
     Close();
 }
 


@@ -106,50 +105,54 @@ wxFTP::~wxFTP()
 // wxFTP connect and login methods
 // ----------------------------------------------------------------------------
 
 // wxFTP connect and login methods
 // ----------------------------------------------------------------------------
 

-bool wxFTP::Connect(wxSockAddress& addr, bool WXUNUSED(wait))
+bool wxFTP::Connect(const wxSockAddress& addr, bool WXUNUSED(wait))

 {
     if ( !wxProtocol::Connect(addr) )
     {
         m_lastError = wxPROTO_NETERR;
 {
     if ( !wxProtocol::Connect(addr) )
     {
         m_lastError = wxPROTO_NETERR;

-        return FALSE;
+        return false;

     }
 
     }
 

-    if ( !m_user )
+    if ( !m_username )

     {
         m_lastError = wxPROTO_CONNERR;
     {
         m_lastError = wxPROTO_CONNERR;

-        return FALSE;
+        return false;

     }
 
     // we should have 220 welcome message
     if ( !CheckResult('2') )
     {
         Close();
     }
 
     // we should have 220 welcome message
     if ( !CheckResult('2') )
     {
         Close();

-        return FALSE;
+        return false;

     }
 
     wxString command;
     }
 
     wxString command;

-    command.Printf(wxT("USER %s"), m_user.c_str());
+    command.Printf(wxT("USER %s"), m_username.c_str());

     char rc = SendCommand(command);
     if ( rc == '2' )
     {
         // 230 return: user accepted without password
     char rc = SendCommand(command);
     if ( rc == '2' )
     {
         // 230 return: user accepted without password

-        return TRUE;
+        m_lastError = wxPROTO_NOERR;
+        return true;

     }
 
     if ( rc != '3' )
     {
     }
 
     if ( rc != '3' )
     {

+        m_lastError = wxPROTO_CONNERR;

         Close();
         Close();

-        return FALSE;
+        return false;

     }
 
     }
 

-    command.Printf(wxT("PASS %s"), m_passwd.c_str());
+    command.Printf(wxT("PASS %s"), m_password.c_str());

     if ( !CheckCommand(command, '2') )
     {
     if ( !CheckCommand(command, '2') )
     {

+        m_lastError = wxPROTO_CONNERR;

         Close();
         Close();

-        return FALSE;
+        return false;

     }
 
     }
 

-    return TRUE;
+    m_lastError = wxPROTO_NOERR;
+    return true;

 }
 
 bool wxFTP::Connect(const wxString& host)
 }
 
 bool wxFTP::Connect(const wxString& host)


@@ -166,13 +169,14 @@ bool wxFTP::Close()
     if ( m_streaming )
     {
         m_lastError = wxPROTO_STREAMING;
     if ( m_streaming )
     {
         m_lastError = wxPROTO_STREAMING;

-        return FALSE;
+        return false;

     }
 
     if ( IsConnected() )
     {
         if ( !CheckCommand(wxT("QUIT"), '2') )
         {
     }
 
     if ( IsConnected() )
     {
         if ( !CheckCommand(wxT("QUIT"), '2') )
         {

+            m_lastError = wxPROTO_CONNERR;

             wxLogDebug(_T("Failed to close connection gracefully."));
         }
     }
             wxLogDebug(_T("Failed to close connection gracefully."));
         }
     }


@@ -184,6 +188,43 @@ bool wxFTP::Close()
 // low level methods
 // ============================================================================
 
 // low level methods
 // ============================================================================
 

+wxSocketBase *wxFTP::AcceptIfActive(wxSocketBase *sock)
+{
+    if ( m_bPassive )
+        return sock;
+
+    // now wait for a connection from server
+    wxSocketServer *sockSrv = (wxSocketServer *)sock;
+    if ( !sockSrv->WaitForAccept() )
+    {
+        m_lastError = wxPROTO_CONNERR;
+        wxLogError(_("Timeout while waiting for FTP server to connect, try passive mode."));
+        delete sock;
+        sock = NULL;
+    }
+    else
+    {
+        m_lastError = wxPROTO_NOERR;
+        sock = sockSrv->Accept(true);
+        delete sockSrv;
+    }
+
+    return sock;
+}
+
+bool wxFTP::Abort()
+{
+    if ( !m_streaming )
+        return true;
+
+    m_streaming = false;
+    if ( !CheckCommand(wxT("ABOR"), '4') )
+        return false;
+
+    return CheckResult('2');
+}
+
+

 // ----------------------------------------------------------------------------
 // Send command to FTP server
 // ----------------------------------------------------------------------------
 // ----------------------------------------------------------------------------
 // Send command to FTP server
 // ----------------------------------------------------------------------------


@@ -216,18 +257,24 @@ char wxFTP::SendCommand(const wxString& command)
         cmd = command;
     }
 
         cmd = command;
     }
 

-    wxLogTrace(FTP_TRACE_MASK, _T("==> %s"), cmd.c_str());
+    LogRequest(cmd);

 #endif // __WXDEBUG__
 
 #endif // __WXDEBUG__
 

+    m_lastError = wxPROTO_NOERR;

     return GetResult();
 }
 
 // ----------------------------------------------------------------------------
     return GetResult();
 }
 
 // ----------------------------------------------------------------------------

-// Recieve servers reply
+// Receive servers reply

 // ----------------------------------------------------------------------------
 
 char wxFTP::GetResult()
 {
 // ----------------------------------------------------------------------------
 
 char wxFTP::GetResult()
 {

+    // if we've already had a read or write timeout error, the connection is
+    // probably toast, so don't bother, it just wastes the users time
+    if ( m_bEncounteredError )
+        return 0;
+

     wxString code;
 
     // m_lastResult will contain the entire server response, possibly on
     wxString code;
 
     // m_lastResult will contain the entire server response, possibly on


@@ -241,15 +288,20 @@ char wxFTP::GetResult()
     //      ...
     //      xyz ...
     // and the intermeidate lines may start with xyz or not
     //      ...
     //      xyz ...
     // and the intermeidate lines may start with xyz or not

-    bool badReply = FALSE;
-    bool firstLine = TRUE;
-    bool endOfReply = FALSE;
+    bool badReply = false;
+    bool firstLine = true;
+    bool endOfReply = false;

     while ( !endOfReply && !badReply )
     {
         wxString line;
     while ( !endOfReply && !badReply )
     {
         wxString line;

-        m_lastError = ReadLine(line);
+        m_lastError = ReadLine(this,line);

         if ( m_lastError )
         if ( m_lastError )

+        {
+            m_bEncounteredError = true;

             return 0;
             return 0;

+        }
+
+        LogResponse(line);

 
         if ( !m_lastResult.empty() )
         {
 
         if ( !m_lastResult.empty() )
         {


@@ -265,12 +317,7 @@ char wxFTP::GetResult()
         {
             if ( firstLine )
             {
         {
             if ( firstLine )
             {

-                badReply = TRUE;
-            }
-            else
-            {
-                wxLogTrace(FTP_TRACE_MASK, _T("<== %s %s"),
-                           code.c_str(), line.c_str());
+                badReply = true;

             }
         }
         else // line has at least 4 chars
             }
         }
         else // line has at least 4 chars


@@ -281,41 +328,30 @@ char wxFTP::GetResult()
             if ( firstLine )
             {
                 code = wxString(line, LEN_CODE);
             if ( firstLine )
             {
                 code = wxString(line, LEN_CODE);

-                wxLogTrace(FTP_TRACE_MASK, _T("<== %s %s"),
-                           code.c_str(), line.c_str() + LEN_CODE + 1);

 
                 switch ( chMarker )
                 {
                     case _T(' '):
 
                 switch ( chMarker )
                 {
                     case _T(' '):

-                        endOfReply = TRUE;
+                        endOfReply = true;

                         break;
 
                     case _T('-'):
                         break;
 
                     case _T('-'):

-                        firstLine = FALSE;
+                        firstLine = false;

                         break;
 
                     default:
                         // unexpected
                         break;
 
                     default:
                         // unexpected

-                        badReply = TRUE;
+                        badReply = true;

                 }
             }
             else // subsequent line of multiline reply
             {
                 }
             }
             else // subsequent line of multiline reply
             {

-                if ( wxStrncmp(line, code, LEN_CODE) == 0 )
+                if ( line.compare(0, LEN_CODE, code) == 0 )

                 {
                     if ( chMarker == _T(' ') )
                     {
                 {
                     if ( chMarker == _T(' ') )
                     {

-                        endOfReply = TRUE;
+                        endOfReply = true;

                     }
                     }

-
-                    wxLogTrace(FTP_TRACE_MASK, _T("<== %s %s"),
-                               code.c_str(), line.c_str() + LEN_CODE + 1);
-                }
-                else
-                {
-                    // just part of reply
-                    wxLogTrace(FTP_TRACE_MASK, _T("<== %s %s"),
-                               code.c_str(), line.c_str());

                 }
             }
         }
                 }
             }
         }


@@ -330,9 +366,11 @@ char wxFTP::GetResult()
 
         return 0;
     }
 
         return 0;
     }

+    else
+        m_lastError = wxPROTO_NOERR;

 
     // if we got here we must have a non empty code string
 
     // if we got here we must have a non empty code string

-    return code[0u];
+    return (char)code[0u];

 }
 
 // ----------------------------------------------------------------------------
 }
 
 // ----------------------------------------------------------------------------


@@ -344,7 +382,7 @@ bool wxFTP::SetTransferMode(TransferMode transferMode)
     if ( transferMode == m_currentTransfermode )
     {
         // nothing to do
     if ( transferMode == m_currentTransfermode )
     {
         // nothing to do

-        return TRUE;
+        return true;

     }
 
     wxString mode;
     }
 
     wxString mode;


@@ -365,17 +403,17 @@ bool wxFTP::SetTransferMode(TransferMode transferMode)
 
     if ( !DoSimpleCommand(_T("TYPE"), mode) )
     {
 
     if ( !DoSimpleCommand(_T("TYPE"), mode) )
     {

-        wxLogError(_("Failed to set FTP transfer mode to %s."), (const wxChar*)
+        wxLogError(_("Failed to set FTP transfer mode to %s."),

                    (transferMode == ASCII ? _("ASCII") : _("binary")));
 
                    (transferMode == ASCII ? _("ASCII") : _("binary")));
 

-        return FALSE;
+        return false;

     }
 
     }
 

-    // If we get here the operation has been succesfully completed
+    // If we get here the operation has been successfully completed

     // Set the status-member
     m_currentTransfermode = transferMode;
 
     // Set the status-member
     m_currentTransfermode = transferMode;
 

-    return TRUE;
+    return true;

 }
 
 bool wxFTP::DoSimpleCommand(const wxChar *command, const wxString& arg)
 }
 
 bool wxFTP::DoSimpleCommand(const wxChar *command, const wxString& arg)


@@ -389,11 +427,13 @@ bool wxFTP::DoSimpleCommand(const wxChar *command, const wxString& arg)
     if ( !CheckCommand(fullcmd, '2') )
     {
         wxLogDebug(_T("FTP command '%s' failed."), fullcmd.c_str());
     if ( !CheckCommand(fullcmd, '2') )
     {
         wxLogDebug(_T("FTP command '%s' failed."), fullcmd.c_str());

+        m_lastError = wxPROTO_NETERR;

 
 

-        return FALSE;
+        return false;

     }
 
     }
 

-    return TRUE;
+    m_lastError = wxPROTO_NOERR;
+    return true;

 }
 
 bool wxFTP::ChDir(const wxString& dir)
 }
 
 bool wxFTP::ChDir(const wxString& dir)


@@ -422,19 +462,20 @@ wxString wxFTP::Pwd()
     if ( CheckCommand(wxT("PWD"), '2') )
     {
         // the result is at least that long if CheckCommand() succeeded
     if ( CheckCommand(wxT("PWD"), '2') )
     {
         // the result is at least that long if CheckCommand() succeeded

-        const wxChar *p = m_lastResult.c_str() + LEN_CODE + 1;
+        wxString::const_iterator p = m_lastResult.begin() + LEN_CODE + 1;

         if ( *p != _T('"') )
         {
         if ( *p != _T('"') )
         {

-            wxLogDebug(_T("Missing starting quote in reply for PWD: %s"), p);
+            wxLogDebug(_T("Missing starting quote in reply for PWD: %s"),
+                       wxString(p, m_lastResult.end()));

         }
         else
         {
         }
         else
         {

-            for ( p++; *p; p++ )
+            for ( ++p; (bool)*p; ++p ) // FIXME-DMARS

             {
                 if ( *p == _T('"') )
                 {
                     // check if the quote is doubled
             {
                 if ( *p == _T('"') )
                 {
                     // check if the quote is doubled

-                    p++;
+                    ++p;

                     if ( !*p || *p != _T('"') )
                     {
                         // no, this is the end
                     if ( !*p || *p != _T('"') )
                     {
                         // no, this is the end


@@ -456,6 +497,7 @@ wxString wxFTP::Pwd()
     }
     else
     {
     }
     else
     {

+        m_lastError = wxPROTO_PROTERR;

         wxLogDebug(_T("FTP PWD command failed."));
     }
 
         wxLogDebug(_T("FTP PWD command failed."));
     }
 


@@ -468,7 +510,7 @@ bool wxFTP::Rename(const wxString& src, const wxString& dst)
 
     str = wxT("RNFR ") + src;
     if ( !CheckCommand(str, '3') )
 
     str = wxT("RNFR ") + src;
     if ( !CheckCommand(str, '3') )

-        return FALSE;
+        return false;

 
     str = wxT("RNTO ") + dst;
 
 
     str = wxT("RNTO ") + dst;
 


@@ -484,115 +526,117 @@ bool wxFTP::RmFile(const wxString& path)
 }
 
 // ----------------------------------------------------------------------------
 }
 
 // ----------------------------------------------------------------------------

-// wxFTP download and upload
+// wxFTP port methods

 // ----------------------------------------------------------------------------
 
 // ----------------------------------------------------------------------------
 

-class wxInputFTPStream : public wxSocketInputStream
+wxSocketBase *wxFTP::GetPort()

 {
 {

-public:
-    wxInputFTPStream(wxFTP *ftp, wxSocketBase *sock)
-        : wxSocketInputStream(*sock)
-    {
-        m_ftp = ftp;
-
-        // FIXME make the timeout configurable
-
-        // set a shorter than default timeout
-        m_i_socket->SetTimeout(60); // 1 minute
+    /*
+    PASSIVE:    Client sends a "PASV" to the server.  The server responds with
+                an address and port number which it will be listening on. Then
+                the client connects to the server at the specified address and
+                port.
+
+    ACTIVE:     Client sends the server a PORT command which includes an
+                address and port number which the client will be listening on.
+                The server then connects to the client at that address and
+                port.
+    */
+
+    wxSocketBase *socket = m_bPassive ? GetPassivePort() : GetActivePort();
+    if ( !socket )
+    {
+        m_bEncounteredError = true;
+        return NULL;

     }
 
     }
 

-    size_t GetSize() const { return m_ftpsize; }
+    // Now set the time for the new socket to the default or user selected
+    // timeout period
+    socket->SetTimeout(m_uiDefaultTimeout);

 
 

-    virtual ~wxInputFTPStream()
-    {
-        delete m_i_socket;
+    return socket;
+}

 
 

-        if ( IsOk() )
-        {
-            // wait for "226 transfer completed"
-            m_ftp->CheckResult('2');
+wxString wxFTP::GetPortCmdArgument(const wxIPV4address& addrLocal,
+                                   const wxIPV4address& addrNew)
+{
+    // Just fills in the return value with the local IP
+    // address of the current socket.  Also it fill in the
+    // PORT which the client will be listening on

 
 

-            m_ftp->m_streaming = FALSE;
-        }
-        else
-        {
-            m_ftp->Abort();
-        }
+    wxString addrIP = addrLocal.IPAddress();
+    int portNew = addrNew.Service();

 
 

-        // delete m_i_socket; // moved to top of destructor to accomodate wu-FTPd >= 2.6.0
-    }
+    // We need to break the PORT number in bytes
+    addrIP.Replace(_T("."), _T(","));
+    addrIP << _T(',')
+           << wxString::Format(_T("%d"), portNew >> 8) << _T(',')
+           << wxString::Format(_T("%d"), portNew & 0xff);

 
 

-    wxFTP *m_ftp;
-    size_t m_ftpsize;
-
-    DECLARE_NO_COPY_CLASS(wxInputFTPStream)
-};
+    // Now we have a value like "10,0,0,1,5,23"
+    return addrIP;
+}

 
 

-class wxOutputFTPStream : public wxSocketOutputStream
+wxSocketBase *wxFTP::GetActivePort()

 {
 {

-public:
-    wxOutputFTPStream(wxFTP *ftp_clt, wxSocketBase *sock)
-        : wxSocketOutputStream(*sock), m_ftp(ftp_clt)
-    {
-    }
+    // we need an address to listen on
+    wxIPV4address addrNew, addrLocal;
+    GetLocal(addrLocal);
+    addrNew.AnyAddress();
+    addrNew.Service(0); // pick an open port number.

 
 

-    virtual ~wxOutputFTPStream(void)
+    wxSocketServer *sockSrv = new wxSocketServer(addrNew);
+    if (!sockSrv->Ok())

     {
     {

-        if ( IsOk() )
-        {
-            // close data connection first, this will generate "transfer
-            // completed" reply
-            delete m_o_socket;
+        // We use Ok() here to see if everything is ok
+        m_lastError = wxPROTO_PROTERR;
+        delete sockSrv;
+        return NULL;
+    }

 
 

-            // read this reply
-            m_ftp->CheckResult('2');
+    //gets the new address, actually it is just the port number
+    sockSrv->GetLocal(addrNew);

 
 

-            m_ftp->m_streaming = FALSE;
-        }
-        else
-        {
-            // abort data connection first
-            m_ftp->Abort();
-
-            // and close it after
-            delete m_o_socket;
-        }
+    // Now we create the argument of the PORT command, we send in both
+    // addresses because the addrNew has an IP of "0.0.0.0", so we need the
+    // value in addrLocal
+    wxString port = GetPortCmdArgument(addrLocal, addrNew);
+    if ( !DoSimpleCommand(_T("PORT"), port) )
+    {
+        m_lastError = wxPROTO_PROTERR;
+        delete sockSrv;
+        wxLogError(_("The FTP server doesn't support the PORT command."));
+        return NULL;

     }
 
     }
 

-    wxFTP *m_ftp;
-
-    DECLARE_NO_COPY_CLASS(wxOutputFTPStream)
-};
+    m_lastError = wxPROTO_NOERR;
+    sockSrv->Notify(false); // Don't send any events
+    return sockSrv;
+}

 
 

-wxSocketClient *wxFTP::GetPort()
+wxSocketBase *wxFTP::GetPassivePort()

 {
 {

-    int a[6];
-

     if ( !DoSimpleCommand(_T("PASV")) )
     {
     if ( !DoSimpleCommand(_T("PASV")) )
     {

+        m_lastError = wxPROTO_PROTERR;

         wxLogError(_("The FTP server doesn't support passive mode."));
         wxLogError(_("The FTP server doesn't support passive mode."));

-

         return NULL;
     }
 
         return NULL;
     }
 

-    const wxChar *addrStart = wxStrchr(m_lastResult, _T('('));
-    if ( !addrStart )
-    {
-        m_lastError = wxPROTO_PROTERR;
+    size_t addrStart = m_lastResult.find(_T('('));
+    size_t addrEnd = (addrStart == wxString::npos)
+                     ? wxString::npos
+                     : m_lastResult.find(_T(')'), addrStart);

 
 

-        return NULL;
-    }
-
-    const wxChar *addrEnd = wxStrchr(addrStart, _T(')'));
-    if ( !addrEnd )
+    if ( addrEnd == wxString::npos )

     {
         m_lastError = wxPROTO_PROTERR;
     {
         m_lastError = wxPROTO_PROTERR;

-

         return NULL;
     }
 
         return NULL;
     }
 

-    wxString straddr(addrStart + 1, addrEnd);
-
+    // get the port number and address
+    int a[6];
+    wxString straddr(m_lastResult, addrStart + 1, addrEnd - (addrStart + 1));

     wxSscanf(straddr, wxT("%d,%d,%d,%d,%d,%d"),
              &a[2],&a[3],&a[4],&a[5],&a[0],&a[1]);
 
     wxSscanf(straddr, wxT("%d,%d,%d,%d,%d,%d"),
              &a[2],&a[3],&a[4],&a[5],&a[0],&a[1]);
 


@@ -600,7 +644,7 @@ wxSocketClient *wxFTP::GetPort()
                         (wxUint16)a[3] << 16 |
                         (wxUint16)a[4] << 8 |
                         a[5];
                         (wxUint16)a[3] << 16 |
                         (wxUint16)a[4] << 8 |
                         a[5];

-    wxUint16 port = (wxUint16)a[0] << 8 | a[1];
+    wxUint16 port = (wxUint16)(a[0] << 8 | a[1]);

 
     wxIPV4address addr;
     addr.Hostname(hostaddr);
 
     wxIPV4address addr;
     addr.Hostname(hostaddr);


@@ -609,36 +653,115 @@ wxSocketClient *wxFTP::GetPort()
     wxSocketClient *client = new wxSocketClient();
     if ( !client->Connect(addr) )
     {
     wxSocketClient *client = new wxSocketClient();
     if ( !client->Connect(addr) )
     {

+        m_lastError = wxPROTO_CONNERR;

         delete client;
         return NULL;
     }
 
         delete client;
         return NULL;
     }
 

-    client->Notify(FALSE);
+    client->Notify(false);

 
 

+    m_lastError = wxPROTO_NOERR;

     return client;
 }
 
     return client;
 }
 

-bool wxFTP::Abort()
+
+// ----------------------------------------------------------------------------
+// wxFTP download and upload
+// ----------------------------------------------------------------------------
+
+class wxInputFTPStream : public wxSocketInputStream

 {
 {

-    if ( !m_streaming )
-        return TRUE;
+public:
+    wxInputFTPStream(wxFTP *ftp, wxSocketBase *sock)
+        : wxSocketInputStream(*sock)
+    {
+        m_ftp = ftp;
+        // socket timeout automatically set in GetPort function
+    }

 
 

-    m_streaming = FALSE;
-    if ( !CheckCommand(wxT("ABOR"), '4') )
-        return FALSE;
+    virtual ~wxInputFTPStream()
+    {
+        delete m_i_socket;   // keep at top

 
 

-    return CheckResult('2');
-}
+        // when checking the result, the stream will
+        // almost always show an error, even if the file was
+        // properly transfered, thus, lets just grab the result

 
 

-wxInputStream *wxFTP::GetInputStream(const wxString& path)
+        // we are looking for "226 transfer completed"
+        char code = m_ftp->GetResult();
+        if ('2' == code)
+        {
+            // it was a good transfer.
+            // we're done!
+             m_ftp->m_streaming = false;
+            return;
+        }
+        // did we timeout?
+        if (0 == code)
+        {
+            // the connection is probably toast. issue an abort, and
+            // then a close. there won't be any more waiting
+            // for this connection
+            m_ftp->Abort();
+            m_ftp->Close();
+            return;
+        }
+        // There was a problem with the transfer and the server
+        // has acknowledged it.  If we issue an "ABORT" now, the user
+        // would get the "226" for the abort and think the xfer was
+        // complete, thus, don't do anything here, just return
+    }
+
+    wxFTP *m_ftp;
+
+    wxDECLARE_NO_COPY_CLASS(wxInputFTPStream);
+};
+
+class wxOutputFTPStream : public wxSocketOutputStream

 {
 {

-    int pos_size;
-    wxInputFTPStream *in_stream;
+public:
+    wxOutputFTPStream(wxFTP *ftp_clt, wxSocketBase *sock)
+        : wxSocketOutputStream(*sock), m_ftp(ftp_clt)
+    {
+    }
+
+    virtual ~wxOutputFTPStream(void)
+    {
+        if ( IsOk() )
+        {
+            // close data connection first, this will generate "transfer
+            // completed" reply
+            delete m_o_socket;
+
+            // read this reply
+            m_ftp->GetResult(); // save result so user can get to it
+
+            m_ftp->m_streaming = false;
+        }
+        else
+        {
+            // abort data connection first
+            m_ftp->Abort();

 
 

+            // and close it after
+            delete m_o_socket;
+        }
+    }
+
+    wxFTP *m_ftp;
+
+    wxDECLARE_NO_COPY_CLASS(wxOutputFTPStream);
+};
+
+wxInputStream *wxFTP::GetInputStream(const wxString& path)
+{

     if ( ( m_currentTransfermode == NONE ) && !SetTransferMode(BINARY) )
     if ( ( m_currentTransfermode == NONE ) && !SetTransferMode(BINARY) )

+    {
+        m_lastError = wxPROTO_CONNERR;

         return NULL;
         return NULL;

+    }

 
 

-    wxSocketClient *sock = GetPort();
+    wxSocketBase *sock = GetPort();

 
     if ( !sock )
     {
 
     if ( !sock )
     {


@@ -646,40 +769,46 @@ wxInputStream *wxFTP::GetInputStream(const wxString& path)
         return NULL;
     }
 
         return NULL;
     }
 

-    wxString tmp_str = wxT("RETR ") + wxURL::ConvertFromURI(path);
+    wxString tmp_str = wxT("RETR ") + wxURI::Unescape(path);

     if ( !CheckCommand(tmp_str, '1') )
         return NULL;
 
     if ( !CheckCommand(tmp_str, '1') )
         return NULL;
 

-    m_streaming = TRUE;
-
-    in_stream = new wxInputFTPStream(this, sock);
-
-    pos_size = m_lastResult.Index(wxT('('));
-    if ( pos_size != wxNOT_FOUND )
+    sock = AcceptIfActive(sock);
+    if ( !sock )

     {
     {

-        wxString str_size = m_lastResult(pos_size+1, m_lastResult.Index(wxT(')'))-1);
-
-        in_stream->m_ftpsize = wxAtoi(WXSTRINGCAST str_size);
+        m_lastError = wxPROTO_CONNERR;
+        return NULL;

     }
 
     sock->SetFlags(wxSOCKET_WAITALL);
 
     }
 
     sock->SetFlags(wxSOCKET_WAITALL);
 

+    m_streaming = true;
+
+    wxInputFTPStream *in_stream = new wxInputFTPStream(this, sock);
+
+    m_lastError = wxPROTO_NOERR;

     return in_stream;
 }
 
 wxOutputStream *wxFTP::GetOutputStream(const wxString& path)
 {
     if ( ( m_currentTransfermode == NONE ) && !SetTransferMode(BINARY) )
     return in_stream;
 }
 
 wxOutputStream *wxFTP::GetOutputStream(const wxString& path)
 {
     if ( ( m_currentTransfermode == NONE ) && !SetTransferMode(BINARY) )

+    {
+        m_lastError = wxPROTO_CONNERR;

         return NULL;
         return NULL;

+    }

 
 

-    wxSocketClient *sock = GetPort();
+    wxSocketBase *sock = GetPort();

 
     wxString tmp_str = wxT("STOR ") + path;
     if ( !CheckCommand(tmp_str, '1') )
         return NULL;
 
 
     wxString tmp_str = wxT("STOR ") + path;
     if ( !CheckCommand(tmp_str, '1') )
         return NULL;
 

-    m_streaming = TRUE;
+    sock = AcceptIfActive(sock);
+
+    m_streaming = true;

 
 

+    m_lastError = wxPROTO_NOERR;

     return new wxOutputFTPStream(this, sock);
 }
 
     return new wxOutputFTPStream(this, sock);
 }
 


@@ -692,8 +821,10 @@ bool wxFTP::GetList(wxArrayString& files,
                     bool details)
 {
     wxSocketBase *sock = GetPort();
                     bool details)
 {
     wxSocketBase *sock = GetPort();

-    if (!sock)
-        return FALSE;
+    if (!sock) {
+        m_lastError = wxPROTO_NETERR;
+        return false;
+    }

 
     // NLST : List of Filenames (including Directory's !)
     // LIST : depending on BS of FTP-Server
 
     // NLST : List of Filenames (including Directory's !)
     // LIST : depending on BS of FTP-Server


@@ -701,27 +832,36 @@ bool wxFTP::GetList(wxArrayString& files,
     //        - Windows : like "dir" command
     //        - others  : ?
     wxString line(details ? _T("LIST") : _T("NLST"));
     //        - Windows : like "dir" command
     //        - others  : ?
     wxString line(details ? _T("LIST") : _T("NLST"));

-    if ( !!wildcard )
+    if ( !wildcard.empty() )

     {
         line << _T(' ') << wildcard;
     }
 
     {
         line << _T(' ') << wildcard;
     }
 

-    if (!CheckCommand(line, '1'))
+    if ( !CheckCommand(line, '1') )

     {
     {

-        return FALSE;
+        m_lastError = wxPROTO_PROTERR;
+        wxLogDebug(_T("FTP 'LIST' command returned unexpected result from server"));
+        delete sock;
+        return false;
+    }
+
+    sock = AcceptIfActive(sock);
+    if ( !sock ) {
+        m_lastError = wxPROTO_CONNERR;
+        return false;

     }
     }

+

     files.Empty();
     files.Empty();

-    while ( ReadLine(sock, line) == wxPROTO_NOERR )
+    while (ReadLine(sock, line) == wxPROTO_NOERR )

     {
         files.Add(line);
     }
     {
         files.Add(line);
     }

+

     delete sock;
 
     // the file list should be terminated by "226 Transfer complete""
     delete sock;
 
     // the file list should be terminated by "226 Transfer complete""

-    if ( !CheckResult('2') )
-        return FALSE;
-
-    return TRUE;
+    m_lastError = wxPROTO_NOERR;
+    return CheckResult('2');

 }
 
 bool wxFTP::FileExists(const wxString& fileName)
 }
 
 bool wxFTP::FileExists(const wxString& fileName)


@@ -730,13 +870,13 @@ bool wxFTP::FileExists(const wxString& fileName)
     // current dir. It does so by simply doing an NLST (via GetList).
     // If this succeeds (and the list is not empty) the file exists.
 
     // current dir. It does so by simply doing an NLST (via GetList).
     // If this succeeds (and the list is not empty) the file exists.
 

-    bool retval = FALSE;
+    bool retval = false;

     wxArrayString fileList;
 
     wxArrayString fileList;
 

-    if ( GetList(fileList, fileName, FALSE) )
+    if ( GetList(fileList, fileName, false) )

     {
         // Some ftp-servers (Ipswitch WS_FTP Server 1.0.5 does this)
     {
         // Some ftp-servers (Ipswitch WS_FTP Server 1.0.5 does this)

-        // displays this behaviour when queried on a non-existing file:
+        // displays this behaviour when queried on a nonexistent file:

         // NLST this_file_does_not_exist
         // 150 Opening ASCII data connection for directory listing
         // (no data transferred)
         // NLST this_file_does_not_exist
         // 150 Opening ASCII data connection for directory listing
         // (no data transferred)


@@ -787,13 +927,13 @@ int wxFTP::GetFileSize(const wxString& fileName)
                           &statuscode, &filesize) == 2 )
             {
                 // We've gotten a good reply.
                           &statuscode, &filesize) == 2 )
             {
                 // We've gotten a good reply.

-                ok = TRUE;
+                ok = true;

             }
             else
             {
                 // Something bad happened.. A "2yz" reply with no size
                 // Fallback
             }
             else
             {
                 // Something bad happened.. A "2yz" reply with no size
                 // Fallback

-                ok = FALSE;
+                ok = false;

             }
         }
 
             }
         }
 


@@ -813,7 +953,7 @@ int wxFTP::GetFileSize(const wxString& fileName)
             // We now try to get details for the file with a "LIST"-command
             // and then parse the output from there..
             wxArrayString fileList;
             // We now try to get details for the file with a "LIST"-command
             // and then parse the output from there..
             wxArrayString fileList;

-            if ( GetList(fileList, fileName, TRUE) )
+            if ( GetList(fileList, fileName, true) )

             {
                 if ( !fileList.IsEmpty() )
                 {
             {
                 if ( !fileList.IsEmpty() )
                 {


@@ -822,10 +962,10 @@ int wxFTP::GetFileSize(const wxString& fileName)
                     // substring containing the name we are looking for. We
                     // stop the iteration at the first occurrence of the
                     // filename. The search is not case-sensitive.
                     // substring containing the name we are looking for. We
                     // stop the iteration at the first occurrence of the
                     // filename. The search is not case-sensitive.

-                    bool foundIt = FALSE;
+                    bool foundIt = false;

 
                     size_t i;
 
                     size_t i;

-                    for ( i = 0; !foundIt && i < fileList.Count(); i++ )
+                    for ( i = 0; !foundIt && i < fileList.GetCount(); i++ )

                     {
                         foundIt = fileList[i].Upper().Contains(fileName.Upper());
                     }
                     {
                         foundIt = fileList[i].Upper().Contains(fileName.Upper());
                     }


@@ -854,8 +994,7 @@ int wxFTP::GetFileSize(const wxString& fileName)
                                           &filesize) != 9 )
                             {
                                 // Hmm... Invalid response
                                           &filesize) != 9 )
                             {
                                 // Hmm... Invalid response

-                                wxLogTrace(FTP_TRACE_MASK,
-                                           _T("Invalid LIST response"));
+                                wxLogDebug(wxT("Invalid LIST response"));

                             }
                         }
                         else // Windows-style response (?)
                             }
                         }
                         else // Windows-style response (?)


@@ -865,8 +1004,7 @@ int wxFTP::GetFileSize(const wxString& fileName)
                                           &filesize) != 4 )
                             {
                                 // something bad happened..?
                                           &filesize) != 4 )
                             {
                                 // something bad happened..?

-                                wxLogTrace(FTP_TRACE_MASK,
-                                           _T("Invalid or unknown LIST response"));
+                                wxLogDebug(wxT("Invalid or unknown LIST response"));

                             }
                         }
                     }
                             }
                         }
                     }