X-Git-Url: https://git.saurik.com/wxWidgets.git/blobdiff_plain/5526e819eca4465ed5520d49bccfebc6a28045e0..29149a64916d6fdc53e445adca9ef83bc58fb6c3:/src/common/fs_inet.cpp

diff --git a/src/common/fs_inet.cpp b/src/common/fs_inet.cpp
index acb0c25831..31e26d60e9 100644
--- a/src/common/fs_inet.cpp
+++ b/src/common/fs_inet.cpp
@@ -8,7 +8,7 @@
 
 /*
 
-REMARKS : 
+REMARKS :
 
 This FS creates local cache (in /tmp directory). The cache is freed
 on program exit.
@@ -20,23 +20,31 @@ limitation)
 */
 
 #ifdef __GNUG__
-#pragma implementation
+#pragma implementation "fs_inet.h"
 #endif
 
-#include <wx/wxprec.h>
+#include "wx/wxprec.h"
 
 #ifdef __BORDLANDC__
 #pragma hdrstop
 #endif
 
+#if !wxUSE_SOCKETS
+    #undef wxUSE_FS_INET
+    #define wxUSE_FS_INET 0
+#endif
+
+#if wxUSE_FILESYSTEM && wxUSE_FS_INET
+
 #ifndef WXPRECOMP
-#include <wx/wx.h>
+#include "wx/wx.h"
 #endif
 
 #include "wx/wfstream.h"
 #include "wx/url.h"
 #include "wx/filesys.h"
 #include "wx/fs_inet.h"
+#include "wx/module.h"
 
 class wxInetCacheNode : public wxObject
 {
@@ -59,18 +67,41 @@ class wxInetCacheNode : public wxObject
 //--------------------------------------------------------------------------------
 
 
-bool wxInternetFSHandler::CanOpen(const wxString& location)
+static wxString StripProtocolAnchor(const wxString& location)
 {
-    wxString p = GetProtocol(location);
-    return (p == "http") || (p == "ftp");
+    wxString myloc(location.BeforeLast(wxT('#')));
+    if (myloc.IsEmpty()) myloc = location.AfterFirst(wxT(':'));
+    else myloc = myloc.AfterFirst(wxT(':'));
+
+    // fix malformed url:
+    if (myloc.Left(2) != wxT("//")) 
+    {
+        if (myloc.GetChar(0) != wxT('/')) myloc = wxT("//") + myloc;
+        else myloc = wxT("/") + myloc;
+    }
+    if (myloc.Mid(2).Find(wxT('/')) == wxNOT_FOUND) myloc << wxT('/');
+
+    return myloc;
 }
 
 
 
+bool wxInternetFSHandler::CanOpen(const wxString& location)
+{
+    wxString p = GetProtocol(location);
+    if ((p == wxT("http")) || (p == wxT("ftp"))) 
+    {
+        wxURL url(p + wxT(":") + StripProtocolAnchor(location));
+        return (url.GetError() == wxURL_NOERR);
+    }
+
+    return FALSE;
+}
+
 
 wxFSFile* wxInternetFSHandler::OpenFile(wxFileSystem& WXUNUSED(fs), const wxString& location)
 {
-    wxString right = GetProtocol(location) + ":" + GetRightLocation(location);
+    wxString right = GetProtocol(location) + wxT(":") + StripProtocolAnchor(location);
     wxInputStream *s;
     wxString content;
     wxInetCacheNode *info;
@@ -78,36 +109,45 @@ wxFSFile* wxInternetFSHandler::OpenFile(wxFileSystem& WXUNUSED(fs), const wxStri
     info = (wxInetCacheNode*) m_Cache.Get(right);
 
     // Add item into cache:
-    if (info == NULL) {
+    if (info == NULL)
+    {
         wxURL url(right);
-        s = url.GetInputStream();
-        content = url.GetProtocol().GetContentType();
-        if (content == wxEmptyString) content = GetMimeTypeFromExt(location);
-        if (s) {
-            char buf[256];
-
-            wxGetTempFileName("wxhtml", buf);
-            info = new wxInetCacheNode(buf, content);
-            m_Cache.Put(right, info);
-
-            {   // ok, now copy it:
-                wxFileOutputStream sout(buf);
-                s -> Read(sout); // copy the stream
+        if (url.GetError() == wxURL_NOERR) 
+        {
+            s = url.GetInputStream();
+            content = url.GetProtocol().GetContentType();
+            if (content == wxEmptyString) content = GetMimeTypeFromExt(location);
+            if (s)
+            {
+                wxChar buf[256];
+
+                wxGetTempFileName( wxT("wxhtml"), buf);
+                info = new wxInetCacheNode(buf, content);
+                m_Cache.Put(right, info);
+
+                {   // ok, now copy it:
+                    wxFileOutputStream sout((wxString)buf);
+                    s -> Read(sout); // copy the stream
+                }
+                delete s;
             }
-            delete s;
+            else
+                return (wxFSFile*) NULL; // we can't open the URL
         }
-        else return NULL; //we can't open the URL
+        else
+            return (wxFSFile*) NULL; // incorrect URL
     }
 
     // Load item from cache:
-    s = new wxFileInputStream(info -> GetTemp());
-    if (s) {
-        return new wxFSFile(s,
-                            right,
-                            info -> GetMime(),
-                            GetAnchor(location));
-    }
-    else return NULL;
+    s = new wxFileInputStream(info->GetTemp());
+    if (!s)
+        return (wxFSFile*) NULL;
+
+    return new wxFSFile(s,
+                        right,
+                        info->GetMime(),
+                        GetAnchor(location),
+                        wxDateTime::Now());
 }
 
 
@@ -118,13 +158,27 @@ wxInternetFSHandler::~wxInternetFSHandler()
     wxInetCacheNode *n2;
 
     m_Cache.BeginFind();
-    while ((n = m_Cache.Next()) != NULL) {
-        n2 = (wxInetCacheNode*) n -> GetData();
-        wxRemoveFile(n2 -> GetTemp());
+    while ((n = m_Cache.Next()) != NULL)
+    {
+        n2 = (wxInetCacheNode*) n->GetData();
+        wxRemoveFile(n2->GetTemp());
         delete n2;
     }
 }
 
+class wxFileSystemInternetModule : public wxModule
+{
+    DECLARE_DYNAMIC_CLASS(wxFileSystemInternetModule)
 
+    public:
+        virtual bool OnInit()
+        {
+            wxFileSystem::AddHandler(new wxInternetFSHandler);
+            return TRUE;
+        }
+        virtual void OnExit() {}
+};
 
+IMPLEMENT_DYNAMIC_CLASS(wxFileSystemInternetModule, wxModule)
 
+#endif // wxUSE_FILESYSTEM && wxUSE_FS_INET