src/common/fs_zip.cpp

   1 /////////////////////////////////////////////////////////////////////////////
   2 // Name:        fs_zip.cpp
   3 // Purpose:     ZIP file system
   4 // Author:      Vaclav Slavik
   5 // Copyright:   (c) 1999 Vaclav Slavik
   6 // CVS-ID:      $Id$
   7 // Licence:     wxWindows licence
   8 /////////////////////////////////////////////////////////////////////////////
   9
  10 #include "wx/wxprec.h"
  11
  12 #ifdef __BORLANDC__
  13 #pragma hdrstop
  14 #endif
  15
  16 #if wxUSE_FILESYSTEM && wxUSE_FS_ZIP && wxUSE_ZIPSTREAM && wxUSE_ZLIB
  17
  18 #ifndef WXPRECOMP
  19     #include "wx/intl.h"
  20     #include "wx/log.h"
  21 #endif
  22
  23 #include "wx/filesys.h"
  24 #include "wx/wfstream.h"
  25 #include "wx/zipstrm.h"
  26 #include "wx/fs_zip.h"
  27
  28
  29 //---------------------------------------------------------------------------
  30 // wxZipFSInputStream
  31 //---------------------------------------------------------------------------
  32 // Helper class for wxZipFSHandler
  33
  34 class wxZipFSInputStream : public wxZipInputStream
  35 {
  36     public:
  37        wxZipFSInputStream(wxFSFile *file)
  38                : wxZipInputStream(*file->GetStream())
  39        {
  40             m_file = file;
  41 #if 1 //WXWIN_COMPATIBILITY_2_6
  42             m_allowSeeking = true;
  43 #endif
  44        }
  45
  46        virtual ~wxZipFSInputStream() { delete m_file; }
  47
  48     private:
  49        wxFSFile *m_file;
  50 };
  51
  52 //----------------------------------------------------------------------------
  53 // wxZipFSHandler
  54 //----------------------------------------------------------------------------
  55
  56 wxZipFSHandler::wxZipFSHandler() : wxFileSystemHandler()
  57 {
  58     m_Archive = NULL;
  59     m_ZipFile = m_Pattern = m_BaseDir = wxEmptyString;
  60     m_AllowDirs = m_AllowFiles = true;
  61     m_DirsFound = NULL;
  62 }
  63
  64
  65
  66 wxZipFSHandler::~wxZipFSHandler()
  67 {
  68     Cleanup();
  69 }
  70
  71
  72 void wxZipFSHandler::Cleanup()
  73 {
  74     wxDELETE(m_Archive);
  75     wxDELETE(m_DirsFound);
  76 }
  77
  78
  79
  80 bool wxZipFSHandler::CanOpen(const wxString& location)
  81 {
  82     wxString p = GetProtocol(location);
  83     return (p == wxT("zip"));
  84 }
  85
  86
  87 wxFSFile* wxZipFSHandler::OpenFile(wxFileSystem& WXUNUSED(fs), const wxString& location)
  88 {
  89     wxString right = GetRightLocation(location);
  90     wxString left = GetLeftLocation(location);
  91     wxZipInputStream *s;
  92
  93     if (right.Contains(wxT("./")))
  94     {
  95         if (right.GetChar(0) != wxT('/')) right = wxT('/') + right;
  96         wxFileName rightPart(right, wxPATH_UNIX);
  97         rightPart.Normalize(wxPATH_NORM_DOTS, wxT("/"), wxPATH_UNIX);
  98         right = rightPart.GetFullPath(wxPATH_UNIX);
  99     }
 100
 101     if (right.GetChar(0) == wxT('/')) right = right.Mid(1);
 102
 103     // a new wxFileSystem object is needed here to avoid infinite recursion
 104     wxFSFile *leftFile = wxFileSystem().OpenFile(left);
 105     if (!leftFile)
 106        return NULL;
 107
 108     s = new wxZipFSInputStream(leftFile);
 109     if (s && s->IsOk())
 110     {
 111        bool found = false;
 112        while (!found)
 113        {
 114            wxZipEntry *ent = s->GetNextEntry();
 115            if (!ent)
 116                break;
 117            if (ent->GetInternalName() == right)
 118                found = true;
 119            delete ent;
 120        }
 121        if (found)
 122            return new wxFSFile(s,
 123                             left + wxT("#zip:") + right,
 124                             GetMimeTypeFromExt(location),
 125                             GetAnchor(location)
 126 #if wxUSE_DATETIME
 127                             , wxDateTime(wxFileModificationTime(left))
 128 #endif // wxUSE_DATETIME
 129                             );
 130     }
 131
 132     delete s;
 133     return NULL;
 134 }
 135
 136
 137
 138 wxString wxZipFSHandler::FindFirst(const wxString& spec, int flags)
 139 {
 140     wxString right = GetRightLocation(spec);
 141     wxString left = GetLeftLocation(spec);
 142
 143     if (!right.empty() && right.Last() == wxT('/')) right.RemoveLast();
 144
 145     if (m_Archive)
 146     {
 147         delete m_Archive;
 148         m_Archive = NULL;
 149     }
 150
 151     switch (flags)
 152     {
 153         case wxFILE:
 154             m_AllowDirs = false, m_AllowFiles = true; break;
 155         case wxDIR:
 156             m_AllowDirs = true, m_AllowFiles = false; break;
 157         default:
 158             m_AllowDirs = m_AllowFiles = true; break;
 159     }
 160
 161     m_ZipFile = left;
 162
 163     wxFSFile *leftFile = wxFileSystem().OpenFile(left);
 164     if (leftFile)
 165         m_Archive = new wxZipFSInputStream(leftFile);
 166
 167     m_Pattern = right.AfterLast(wxT('/'));
 168     m_BaseDir = right.BeforeLast(wxT('/'));
 169     if (m_BaseDir.StartsWith(wxT("/")))
 170         m_BaseDir = m_BaseDir.Mid(1);
 171
 172     if (m_Archive)
 173     {
 174         if (m_AllowDirs)
 175         {
 176             delete m_DirsFound;
 177             m_DirsFound = new wxZipFilenameHashMap();
 178             if (right.empty())  // allow "/" to match the archive root
 179                 return spec;
 180         }
 181         return DoFind();
 182     }
 183     return wxEmptyString;
 184 }
 185
 186
 187
 188 wxString wxZipFSHandler::FindNext()
 189 {
 190     if (!m_Archive) return wxEmptyString;
 191     return DoFind();
 192 }
 193
 194
 195
 196 wxString wxZipFSHandler::DoFind()
 197 {
 198     wxString namestr, dir, filename;
 199     wxString match = wxEmptyString;
 200
 201     while (match == wxEmptyString)
 202     {
 203         wxZipEntry *entry = m_Archive->GetNextEntry();
 204         if (!entry)
 205         {
 206             delete m_Archive;
 207             m_Archive = NULL;
 208             break;
 209         }
 210         namestr = entry->GetName(wxPATH_UNIX);
 211         delete entry;
 212
 213         if (m_AllowDirs)
 214         {
 215             dir = namestr.BeforeLast(wxT('/'));
 216             while (!dir.empty())
 217             {
 218                 if( m_DirsFound->find(dir) == m_DirsFound->end() )
 219                 {
 220                     (*m_DirsFound)[dir] = 1;
 221                     filename = dir.AfterLast(wxT('/'));
 222                     dir = dir.BeforeLast(wxT('/'));
 223                     if (!filename.empty() && m_BaseDir == dir &&
 224                                 wxMatchWild(m_Pattern, filename, false))
 225                         match = m_ZipFile + wxT("#zip:") + dir + wxT("/") + filename;
 226                 }
 227                 else
 228                     break; // already tranversed
 229             }
 230         }
 231
 232         filename = namestr.AfterLast(wxT('/'));
 233         dir = namestr.BeforeLast(wxT('/'));
 234         if (m_AllowFiles && !filename.empty() && m_BaseDir == dir &&
 235                             wxMatchWild(m_Pattern, filename, false))
 236             match = m_ZipFile + wxT("#zip:") + namestr;
 237     }
 238
 239     return match;
 240 }
 241
 242
 243
 244 #endif
 245       //wxUSE_FILESYSTEM && wxUSE_FS_ZIP && wxUSE_ZIPSTREAM