X-Git-Url: https://git.saurik.com/wxWidgets.git/blobdiff_plain/de0702d065066f8ef63a9581985def172777ebb6..9ac71bef070f553f4ecef2c549dba33b10422f06:/src/common/fs_zip.cpp

diff --git a/src/common/fs_zip.cpp b/src/common/fs_zip.cpp
index 3a4f8deab6..97cfa0bb1b 100644
--- a/src/common/fs_zip.cpp
+++ b/src/common/fs_zip.cpp
@@ -10,7 +10,7 @@
 
 
 #ifdef __GNUG__
-#pragma implementation
+#pragma implementation "fs_zip.h"
 #endif
 
 #include "wx/wxprec.h"
@@ -22,7 +22,8 @@
 #if wxUSE_FILESYSTEM && wxUSE_FS_ZIP && wxUSE_ZIPSTREAM
 
 #ifndef WXPRECOMP
-#include "wx/wx.h"
+    #include "wx/intl.h"
+    #include "wx/log.h"
 #endif
 
 #include "wx/hash.h"
@@ -56,7 +57,7 @@ wxZipFSHandler::wxZipFSHandler() : wxFileSystemHandler()
 
 wxZipFSHandler::~wxZipFSHandler()
 {
-    if (m_Archive)         
+    if (m_Archive)
         unzClose((unzFile)m_Archive);
     if (m_DirsFound)
         delete m_DirsFound;
@@ -67,7 +68,8 @@ wxZipFSHandler::~wxZipFSHandler()
 bool wxZipFSHandler::CanOpen(const wxString& location)
 {
     wxString p = GetProtocol(location);
-    return (p == wxT("zip"));
+    return (p == wxT("zip")) &&
+           (GetProtocol(GetLeftLocation(location)) == wxT("file"));
 }
 
 
@@ -79,14 +81,16 @@ wxFSFile* wxZipFSHandler::OpenFile(wxFileSystem& WXUNUSED(fs), const wxString& l
     wxString left = GetLeftLocation(location);
     wxInputStream *s;
 
-    if (GetProtocol(left) != wxT("file")) 
+    if (GetProtocol(left) != wxT("file"))
     {
         wxLogError(_("ZIP handler currently supports only local files!"));
         return NULL;
     }
 
+    if (right.GetChar(0) == wxT('/')) right = right.Mid(1);
+
     s = new wxZipInputStream(left, right);
-    if (s && (s->LastError() == wxStream_NOERROR)) 
+    if (s && (s->LastError() == wxStream_NOERROR))
     {
         return new wxFSFile(s,
                             left + wxT("#zip:") + right,
@@ -105,10 +109,10 @@ wxString wxZipFSHandler::FindFirst(const wxString& spec, int flags)
 {
     wxString right = GetRightLocation(spec);
     wxString left = GetLeftLocation(spec);
-    
+
     if (right.Last() == wxT('/')) right.RemoveLast();
 
-    if (m_Archive) 
+    if (m_Archive)
     {
         unzClose((unzFile)m_Archive);
         m_Archive = NULL;
@@ -120,13 +124,13 @@ wxString wxZipFSHandler::FindFirst(const wxString& spec, int flags)
         return wxEmptyString;
     }
 
-    switch (flags) 
+    switch (flags)
     {
-        case wxFILE: 
+        case wxFILE:
             m_AllowDirs = FALSE, m_AllowFiles = TRUE; break;
-        case wxDIR: 
+        case wxDIR:
             m_AllowDirs = TRUE, m_AllowFiles = FALSE; break;
-        default: 
+        default:
             m_AllowDirs = m_AllowFiles = TRUE; break;
     }
 
@@ -135,14 +139,14 @@ wxString wxZipFSHandler::FindFirst(const wxString& spec, int flags)
     m_Pattern = right.AfterLast(wxT('/'));
     m_BaseDir = right.BeforeLast(wxT('/'));
 
-    if (m_Archive) 
+    if (m_Archive)
     {
-        if (unzGoToFirstFile((unzFile)m_Archive) != UNZ_OK) 
+        if (unzGoToFirstFile((unzFile)m_Archive) != UNZ_OK)
         {
             unzClose((unzFile)m_Archive);
-            m_Archive = NULL;        
+            m_Archive = NULL;
         }
-        else 
+        else
         {
             if (m_AllowDirs)
             {
@@ -180,7 +184,7 @@ wxString wxZipFSHandler::DoFind()
 
         if (m_AllowDirs)
         {
-            dir = namestr.BeforeLast(wxT('/'));       
+            dir = namestr.BeforeLast(wxT('/'));
             while (!dir.IsEmpty())
             {
                 long key = 0;
@@ -194,7 +198,7 @@ wxString wxZipFSHandler::DoFind()
                                 wxMatchWild(m_Pattern, filename, FALSE))
                         match = m_ZipFile + wxT("#zip:") + dir + wxT("/") + filename;
                 }
-                else 
+                else
                     break; // already tranversed
             }
         }
@@ -204,19 +208,19 @@ wxString wxZipFSHandler::DoFind()
         if (m_AllowFiles && !filename.IsEmpty() && m_BaseDir == dir &&
                             wxMatchWild(m_Pattern, filename, FALSE))
             match = m_ZipFile + wxT("#zip:") + namestr;
-        
-        if (unzGoToNextFile((unzFile)m_Archive) != UNZ_OK) 
+
+        if (unzGoToNextFile((unzFile)m_Archive) != UNZ_OK)
         {
             unzClose((unzFile)m_Archive);
             m_Archive = NULL;
             break;
         }
     }
-    
+
     return match;
 }
 
 
 
-#endif 
+#endif
       //wxUSE_FILESYSTEM && wxUSE_FS_ZIP && wxUSE_ZIPSTREAM