]> git.saurik.com Git - wxWidgets.git/blobdiff - src/common/url.cpp
fixing overrelease and out-of-bounds write, fixes #13725
[wxWidgets.git] / src / common / url.cpp
index ef280923ab5553bd5b0ae68e3756127295731b90..15b9311839e476a852c5975c9cf8670264912b7f 100644 (file)
 /////////////////////////////////////////////////////////////////////////////
-// Name:        url.cpp
+// Name:        src/common/url.cpp
 // Purpose:     URL parser
 // Author:      Guilhem Lavaux
 // Modified by:
 // Created:     20/07/1997
 // RCS-ID:      $Id$
 // Copyright:   (c) 1997, 1998 Guilhem Lavaux
-// Licence:     wxWindows license
+// Licence:     wxWindows licence
 /////////////////////////////////////////////////////////////////////////////
 
-#ifdef __GNUG__
-#pragma implementation "url.h"
-#endif
-
 // For compilers that support precompilation, includes "wx.h".
 #include "wx/wxprec.h"
 
 #ifdef __BORLANDC__
-#pragma hdrstop
+    #pragma hdrstop
 #endif
 
-#if wxUSE_SOCKETS
+#if wxUSE_URL
+
+#include "wx/url.h"
 
 #ifndef WX_PRECOMP
+    #include "wx/list.h"
+    #include "wx/string.h"
+    #include "wx/utils.h"
+    #include "wx/module.h"
 #endif
 
 #include <string.h>
 #include <ctype.h>
 
-// wxWindows headers
-#include <wx/string.h>
-#include <wx/list.h>
-#include <wx/utils.h>
-
-// wxSocket header
-#include "wx/url.h"
-
-#if !USE_SHARED_LIBRARY
-IMPLEMENT_CLASS(wxProtoInfo, wxObject)
-IMPLEMENT_CLASS(wxURL, wxObject)
-#endif
+IMPLEMENT_CLASS(wxURL, wxURI)
 
 // Protocols list
-wxProtoInfo *wxURL::g_protocols = NULL;
-wxHTTP *wxURL::g_proxy;
+wxProtoInfo *wxURL::ms_protocols = NULL;
 
-/////////////////////////////////////////////////////////////////
-// wxURL ////////////////////////////////////////////////////////
-/////////////////////////////////////////////////////////////////
+// Enforce linking of protocol classes:
+USE_PROTOCOL(wxFileProto)
 
-/*
- * --------------------------------------------------------------
- * --------- wxURL CONSTRUCTOR DESTRUCTOR -----------------------
- * --------------------------------------------------------------
- */
+#if wxUSE_PROTOCOL_HTTP
+USE_PROTOCOL(wxHTTP)
 
-wxURL::wxURL(const wxString& url)
-{
-  m_protocol = NULL;
-  if (g_proxy->IsConnected()) {
-    m_protocol = g_proxy;
-    m_protoname = "proxy";
-    m_path = url;
-    return;
-  }
-  m_url = url;
-  m_error = wxURL_NOERR;
-  ParseURL();
-}
+    wxHTTP *wxURL::ms_proxyDefault = NULL;
+    bool wxURL::ms_useDefaultProxy = false;
+#endif
 
-bool wxURL::ParseURL()
-{
-  wxString last_url = m_url;
-
-  // Clean up
-  CleanData();
-
-  // Extract protocol name
-  if (!PrepProto(last_url)) {
-    m_error = wxURL_SNTXERR;
-    return FALSE;
-  }
-
-  // Find and create the protocol object
-  if (!FetchProtocol()) {
-    m_error = wxURL_NOPROTO;
-    return FALSE;
-  }
-
-  // Do we need a host name ?
-  if (m_protoinfo->m_needhost) {
-    // Extract it
-    if (!PrepHost(last_url)) {
-      m_error = wxURL_SNTXERR;
-      return FALSE;
-    }
-  }
+#if wxUSE_PROTOCOL_FTP
+USE_PROTOCOL(wxFTP)
+#endif
+
+// --------------------------------------------------------------
+//
+//                          wxURL
+//
+// --------------------------------------------------------------
 
-  // Extract full path
-  if (!PrepPath(last_url)) {
-    m_error = wxURL_NOPATH;
-    return FALSE;
-  }
+// --------------------------------------------------------------
+// Construction
+// --------------------------------------------------------------
 
-  m_error = wxURL_NOERR;
-  return TRUE;
+wxURL::wxURL(const wxString& url) : wxURI(url)
+{
+    Init(url);
+    ParseURL();
 }
 
-void wxURL::CleanData()
+wxURL::wxURL(const wxURI& uri) : wxURI(uri)
 {
-  if (m_protoname != _T("proxy"))
-    delete m_protocol;
+    Init(uri.BuildURI());
+    ParseURL();
 }
 
-wxURL::~wxURL()
+wxURL::wxURL(const wxURL& url) : wxURI(url)
 {
-  CleanData();
+    Init(url.GetURL());
+    ParseURL();
 }
 
-/*
- * --------------------------------------------------------------
- * --------- wxURL urls decoders --------------------------------
- * --------------------------------------------------------------
- */
-bool wxURL::PrepProto(wxString& url)
+void wxURL::Init(const wxString& url)
 {
-  int pos;
+    m_protocol = NULL;
+    m_error = wxURL_NOERR;
+    m_url = url;
+#if wxUSE_URL_NATIVE
+    m_nativeImp = CreateNativeImpObject();
+#endif
 
-  // Find end
-  pos = url.Find(':');
-  if (pos == -1)
-    return FALSE;
+#if wxUSE_PROTOCOL_HTTP
+    if ( ms_useDefaultProxy && !ms_proxyDefault )
+    {
+        SetDefaultProxy( wxGetenv(wxT("HTTP_PROXY")) );
 
-  m_protoname = url(0, pos);
+        if ( !ms_proxyDefault )
+        {
+            // don't try again
+            ms_useDefaultProxy = false;
+        }
+    }
 
-  url = url(pos+1, url.Length());
+    m_useProxy = ms_proxyDefault != NULL;
+    m_proxy = ms_proxyDefault;
+#endif // wxUSE_PROTOCOL_HTTP
 
-  return TRUE;
 }
 
-bool wxURL::PrepHost(wxString& url)
-{
-  wxString temp_url;
-  int pos, pos2;
+// --------------------------------------------------------------
+// Assignment
+// --------------------------------------------------------------
 
-  if ((url.GetChar(0) != '/') || (url.GetChar(1) != '/'))
-    return FALSE;
+wxURL& wxURL::operator = (const wxString& url)
+{
+    wxURI::operator = (url);
+    Free();
+    Init(url);
+    ParseURL();
 
-  url = url(2, url.Length());
+    return *this;
+}
 
-  pos = url.Find('/');
-  if (pos == -1)
-    pos = url.Length();
+wxURL& wxURL::operator = (const wxURI& uri)
+{
+    if (&uri != this)
+    {
+        wxURI::operator = (uri);
+        Free();
+        Init(uri.BuildURI());
+        ParseURL();
+    }
 
-  if (pos == 0)
-    return FALSE;
+    return *this;
+}
 
-  temp_url = url(0, pos);
-  url = url(url.Find('/'), url.Length());
+wxURL& wxURL::operator = (const wxURL& url)
+{
+    if (&url != this)
+    {
+        wxURI::operator = (url);
+        Free();
+        Init(url.GetURL());
+        ParseURL();
+    }
 
-  // Retrieve service number
-  pos2 = temp_url.Find(':', TRUE);
-  if (pos2 != -1 && pos2 < pos) {
-    m_servname = temp_url(pos2+1, pos);
-    if (!m_servname.IsNumber())
-      return FALSE;
-    temp_url = temp_url(0, pos2);
-  }
+    return *this;
+}
 
-  // Retrieve user and password.
-  pos2 = temp_url.Find('@');
-  // Even if pos2 equals -1, this code is right.
-  m_hostname = temp_url(pos2+1, temp_url.Length());
+// --------------------------------------------------------------
+// ParseURL
+//
+// Builds the URL and takes care of the old protocol stuff
+// --------------------------------------------------------------
 
-  m_user = "";
-  m_password = "";
+bool wxURL::ParseURL()
+{
+    // If the URL was already parsed (m_protocol != NULL), pass this section.
+    if (!m_protocol)
+    {
+        // Clean up
+        CleanData();
+
+        // Make sure we have a protocol/scheme
+        if (!HasScheme())
+        {
+            m_error = wxURL_SNTXERR;
+            return false;
+        }
+
+        // Find and create the protocol object
+        if (!FetchProtocol())
+        {
+            m_error = wxURL_NOPROTO;
+            return false;
+        }
+
+        // Do we need a host name ?
+        if (m_protoinfo->m_needhost)
+        {
+            //  Make sure we have one, then
+            if (!HasServer())
+            {
+                m_error = wxURL_SNTXERR;
+                return false;
+            }
+        }
+    }
 
-  if (pos2 == -1)
-    return TRUE;
+#if wxUSE_PROTOCOL_HTTP
+    if (m_useProxy)
+    {
+        // Third, we rebuild the URL.
+        m_url = m_scheme + wxT(":");
+        if (m_protoinfo->m_needhost)
+            m_url = m_url + wxT("//") + m_server;
+
+        // We initialize specific variables.
+        if (m_protocol)
+            m_protocol->Destroy();
+        m_protocol = m_proxy; // FIXME: we should clone the protocol
+    }
+#endif // wxUSE_PROTOCOL_HTTP
 
-  temp_url = temp_url(0, pos2);
-  pos2 = temp_url.Find(':');
+    m_error = wxURL_NOERR;
+    return true;
+}
 
-  if (pos2 == -1)
-    return FALSE;
+// --------------------------------------------------------------
+// Destruction/Cleanup
+// --------------------------------------------------------------
 
-  m_user = temp_url(0, pos2);
-  m_password = temp_url(pos2+1, url.Length());
+void wxURL::CleanData()
+{
+#if wxUSE_PROTOCOL_HTTP
+    if (!m_useProxy)
+#endif // wxUSE_PROTOCOL_HTTP
+    {
+        if (m_protocol)
+        {
+            // Need to safely delete the socket (pending events)
+            m_protocol->Destroy();
+            m_protocol = NULL;
+        }
+    }
+}
 
-  return TRUE;
+void wxURL::Free()
+{
+    CleanData();
+#if wxUSE_PROTOCOL_HTTP
+    if (m_proxy && m_proxy != ms_proxyDefault)
+        delete m_proxy;
+#endif // wxUSE_PROTOCOL_HTTP
+#if wxUSE_URL_NATIVE
+    delete m_nativeImp;
+#endif
 }
 
-bool wxURL::PrepPath(wxString& url)
+wxURL::~wxURL()
 {
-  if (url.Length() != 0)
-    m_path = url;
-  else
-    m_path = "/";
-  return TRUE;
+    Free();
 }
 
+// --------------------------------------------------------------
+// FetchProtocol
+// --------------------------------------------------------------
+
 bool wxURL::FetchProtocol()
 {
-  wxProtoInfo *info = g_protocols;
-
-  while (info) {
-    if (m_protoname == info->m_protoname) {
-      if (m_servname.IsNull())
-        m_servname = info->m_servname;
+    wxProtoInfo *info = ms_protocols;
 
-      m_protoinfo = info;
-      m_protocol = (wxProtocol *)m_protoinfo->m_cinfo->CreateObject();
-      wxSocketHandler::Master().Register(m_protocol);
-      return TRUE;
+    while (info)
+    {
+        if (m_scheme == info->m_protoname)
+        {
+            if ( m_port.empty() )
+                m_port = info->m_servname;
+            m_protoinfo = info;
+            m_protocol = (wxProtocol *)m_protoinfo->m_cinfo->CreateObject();
+            return true;
+        }
+        info = info->next;
     }
-    info = info->next;
-  }
-  return FALSE;
+    return false;
 }
 
-/*
- * --------------------------------------------------------------
- * --------- wxURL get ------------------------------------------
- * --------------------------------------------------------------
- */
-wxInputStream *wxURL::GetInputStream(void)
+// --------------------------------------------------------------
+// GetInputStream
+// --------------------------------------------------------------
+
+wxInputStream *wxURL::GetInputStream()
 {
-  wxIPV4address addr;
-  wxInputStream *the_i_stream = NULL;
-
-  if (!m_protocol)
-    if (!ParseURL())
-      return NULL;
-
-  if (!m_protocol) {
-    m_error = wxURL_NOPROTO;
-    return NULL;
-  }
-
-  m_error = wxURL_NOERR;
-  if (m_user != _T("")) {
-    m_protocol->SetUser(m_user);
-    m_protocol->SetPassword(m_password);
-  }
-
-  if (m_protoinfo->m_needhost) {
-    if (!addr.Hostname(m_hostname)) {
-      m_error = wxURL_NOHOST;
-      return NULL;
+    if (!m_protocol)
+    {
+        m_error = wxURL_NOPROTO;
+        return NULL;
     }
 
-    addr.Service(m_servname);
+    m_error = wxURL_NOERR;
+    if (HasUserInfo())
+    {
+        size_t dwPasswordPos = m_userinfo.find(':');
+
+        if (dwPasswordPos == wxString::npos)
+            m_protocol->SetUser(Unescape(m_userinfo));
+        else
+        {
+            m_protocol->SetUser(Unescape(m_userinfo(0, dwPasswordPos)));
+            m_protocol->SetPassword(Unescape(m_userinfo(dwPasswordPos+1, m_userinfo.length() + 1)));
+        }
+    }
+
+#if wxUSE_URL_NATIVE
+    // give the native implementation to return a better stream
+    // such as the native WinINet functionality under MS-Windows
+    if (m_nativeImp)
+    {
+        wxInputStream *rc;
+        rc = m_nativeImp->GetInputStream(this);
+        if (rc != 0)
+            return rc;
+    }
+    // else use the standard behaviour
+#endif // wxUSE_URL_NATIVE
 
-    if (!m_protocol->Connect(addr, TRUE)) // Watcom needs the 2nd arg for some reason
+#if wxUSE_SOCKETS
+    wxIPV4address addr;
+
+    // m_protoinfo is NULL when we use a proxy
+    if (
+#if wxUSE_PROTOCOL_HTTP
+         !m_useProxy &&
+#endif // wxUSE_PROTOCOL_HTTP
+         m_protoinfo->m_needhost )
     {
-      m_error = wxURL_CONNERR;
-      return NULL;
+        if (!addr.Hostname(m_server))
+        {
+            m_error = wxURL_NOHOST;
+            return NULL;
+        }
+
+        addr.Service(m_port);
+
+        if (!m_protocol->Connect(addr, true)) // Watcom needs the 2nd arg for some reason
+        {
+            m_error = wxURL_CONNERR;
+            return NULL;
+        }
     }
-  }
+#endif // wxUSE_SOCKETS
 
-  the_i_stream = m_protocol->GetInputStream(m_path);
-  if (!the_i_stream) {
-    m_error = wxURL_PROTOERR;
-    return NULL;
-  }
+    wxString fullPath;
 
-  return the_i_stream;
+#if wxUSE_PROTOCOL_HTTP
+    // When we use a proxy, we have to pass the whole URL to it.
+    if (m_useProxy)
+        fullPath += m_url;
+#endif // wxUSE_PROTOCOL_HTTP
+
+    if(m_path.empty())
+        fullPath += wxT("/");
+    else
+        fullPath += m_path;
+
+    if (HasQuery())
+        fullPath += wxT("?") + m_query;
+
+    if (HasFragment())
+        fullPath += wxT("#") + m_fragment;
+
+    wxInputStream *the_i_stream = m_protocol->GetInputStream(fullPath);
+
+    if (!the_i_stream)
+    {
+        m_error = wxURL_PROTOERR;
+        return NULL;
+    }
+
+    return the_i_stream;
 }
 
+#if wxUSE_PROTOCOL_HTTP
 void wxURL::SetDefaultProxy(const wxString& url_proxy)
 {
-  g_proxy->Close();
-
-  if (url_proxy.IsNull())
-    return;
+    if ( !url_proxy )
+    {
+        if ( ms_proxyDefault )
+        {
+            ms_proxyDefault->Close();
+            wxDELETE(ms_proxyDefault);
+        }
+    }
+    else
+    {
+        wxString tmp_str = url_proxy;
+        int pos = tmp_str.Find(wxT(':'));
+        if (pos == wxNOT_FOUND)
+            return;
+
+        wxString hostname = tmp_str(0, pos),
+        port = tmp_str(pos+1, tmp_str.length()-pos);
+        wxIPV4address addr;
+
+        if (!addr.Hostname(hostname))
+            return;
+        if (!addr.Service(port))
+            return;
+
+        if (ms_proxyDefault)
+            // Finally, when all is right, we connect the new proxy.
+            ms_proxyDefault->Close();
+        else
+            ms_proxyDefault = new wxHTTP();
+        ms_proxyDefault->Connect(addr, true); // Watcom needs the 2nd arg for some reason
+    }
+}
 
-  wxString tmp_str = url_proxy;
-  int pos = tmp_str.Find(':');
-  wxString hostname = tmp_str(0, pos),
-           port = tmp_str(pos, tmp_str.Length()-pos);
-  wxIPV4address addr;
+void wxURL::SetProxy(const wxString& url_proxy)
+{
+    if ( !url_proxy )
+    {
+        if ( m_proxy && m_proxy != ms_proxyDefault )
+        {
+            m_proxy->Close();
+            delete m_proxy;
+        }
 
-  addr.Hostname(hostname);
-  addr.Service(port);
-  
-  g_proxy->Connect(addr, TRUE); // Watcom needs the 2nd arg for some reason
+        m_useProxy = false;
+    }
+    else
+    {
+        wxString tmp_str;
+        wxString hostname, port;
+        int pos;
+        wxIPV4address addr;
+
+        tmp_str = url_proxy;
+        pos = tmp_str.Find(wxT(':'));
+        // This is an invalid proxy name.
+        if (pos == wxNOT_FOUND)
+            return;
+
+        hostname = tmp_str(0, pos);
+        port = tmp_str(pos+1, tmp_str.length()-pos);
+
+        addr.Hostname(hostname);
+        addr.Service(port);
+
+        // Finally, create the whole stuff.
+        if (m_proxy && m_proxy != ms_proxyDefault)
+            delete m_proxy;
+        m_proxy = new wxHTTP();
+        m_proxy->Connect(addr, true); // Watcom needs the 2nd arg for some reason
+
+        CleanData();
+        // Reparse url.
+        m_useProxy = true;
+        ParseURL();
+    }
 }
+#endif // wxUSE_PROTOCOL_HTTP
 
-void wxURL::SetProxy(const wxString& url_proxy)
+// ----------------------------------------------------------------------
+// wxURLModule
+//
+// A module which deletes the default proxy if we created it
+// ----------------------------------------------------------------------
+
+#if wxUSE_SOCKETS
+
+class wxURLModule : public wxModule
 {
-  if (url_proxy.IsNull()) {
-    m_proxy.Close();
-    return;
-  }
+public:
+    wxURLModule();
 
-  CleanData();
+    virtual bool OnInit();
+    virtual void OnExit();
 
-  wxString tmp_str;
-  wxString hostname, port;
-  int pos;
-  wxIPV4address addr;
+private:
+    DECLARE_DYNAMIC_CLASS(wxURLModule)
+};
 
-  tmp_str = url_proxy;
-  pos = tmp_str.Find(':');
-  hostname = tmp_str(0, pos);
-  port = tmp_str(pos, tmp_str.Length()-pos);
+IMPLEMENT_DYNAMIC_CLASS(wxURLModule, wxModule)
 
-  addr.Hostname(hostname);
-  addr.Service(port);
+wxURLModule::wxURLModule()
+{
+    // we must be cleaned up before wxSocketModule as otherwise deleting
+    // ms_proxyDefault from our OnExit() won't work (and can actually crash)
+    AddDependency(wxClassInfo::FindClass(wxT("wxSocketModule")));
+}
 
-  m_proxy.Connect(addr, TRUE); // Watcom needs the 2nd arg for some reason
+bool wxURLModule::OnInit()
+{
+#if wxUSE_PROTOCOL_HTTP
+    // env var HTTP_PROXY contains the address of the default proxy to use if
+    // set, but don't try to create this proxy right now because it will slow
+    // down the program startup (especially if there is no DNS server
+    // available, in which case it may take up to 1 minute)
 
-  m_protocol = &m_proxy;
-  m_protoname = "proxy";
-  m_path = url_proxy;
+    if ( wxGetenv(wxT("HTTP_PROXY")) )
+    {
+        wxURL::ms_useDefaultProxy = true;
+    }
+#endif // wxUSE_PROTOCOL_HTTP
+    return true;
 }
 
-#endif
-  // wxUSE_SOCKETS
+void wxURLModule::OnExit()
+{
+#if wxUSE_PROTOCOL_HTTP
+    wxDELETE(wxURL::ms_proxyDefault);
+#endif // wxUSE_PROTOCOL_HTTP
+}
+
+#endif // wxUSE_SOCKETS
+
+
+#endif // wxUSE_URL