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
  11
  12 #ifdef __GNUG__
  13 #pragma implementation "fs_zip.h"
  14 #endif
  15
  16 #include "wx/wxprec.h"
  17
  18 #ifdef __BORLANDC__
  19 #pragma hdrstop
  20 #endif
  21
  22 #if wxUSE_FILESYSTEM && wxUSE_FS_ZIP && wxUSE_ZIPSTREAM
  23
  24 #ifndef WXPRECOMP
  25     #include "wx/intl.h"
  26     #include "wx/log.h"
  27 #endif
  28
  29 #include "wx/hash.h"
  30 #include "wx/filesys.h"
  31 #include "wx/zipstrm.h"
  32 #include "wx/fs_zip.h"
  33
  34 /* Not the right solution (paths in makefiles) but... */
  35 #ifdef __BORLANDC__
  36 #include "../common/unzip.h"
  37 #else
  38 #include "unzip.h"
  39 #endif
  40
  41
  42 //----------------------------------------------------------------------------
  43 // wxZipFSHandler
  44 //----------------------------------------------------------------------------
  45
  46
  47
  48 wxZipFSHandler::wxZipFSHandler() : wxFileSystemHandler()
  49 {
  50     m_Archive = NULL;
  51     m_ZipFile = m_Pattern = m_BaseDir = wxEmptyString;
  52     m_AllowDirs = m_AllowFiles = TRUE;
  53     m_DirsFound = NULL;
  54 }
  55
  56
  57
  58 wxZipFSHandler::~wxZipFSHandler()
  59 {
  60     if (m_Archive)
  61         unzClose((unzFile)m_Archive);
  62     if (m_DirsFound)
  63         delete m_DirsFound;
  64 }
  65
  66
  67
  68 bool wxZipFSHandler::CanOpen(const wxString& location)
  69 {
  70     wxString p = GetProtocol(location);
  71     return (p == wxT("zip")) &&
  72            (GetProtocol(GetLeftLocation(location)) == wxT("file"));
  73 }
  74
  75
  76
  77
  78 wxFSFile* wxZipFSHandler::OpenFile(wxFileSystem& WXUNUSED(fs), const wxString& location)
  79 {
  80     wxString right = GetRightLocation(location);
  81     wxString left = GetLeftLocation(location);
  82     wxInputStream *s;
  83
  84     if (GetProtocol(left) != wxT("file"))
  85     {
  86         wxLogError(_("ZIP handler currently supports only local files!"));
  87         return NULL;
  88     }
  89
  90     if (right.Contains(wxT("./")))
  91     {
  92         if (right.GetChar(0) != wxT('/')) right = wxT('/') + right;
  93         wxFileName rightPart(right, wxPATH_UNIX);
  94         rightPart.Normalize(wxPATH_NORM_DOTS, wxT("/"), wxPATH_UNIX);
  95         right = rightPart.GetFullPath(wxPATH_UNIX);
  96     }
  97
  98     if (right.GetChar(0) == wxT('/')) right = right.Mid(1);
  99
 100     wxFileName leftFilename = wxFileSystem::URLToFileName(left);
 101
 102     s = new wxZipInputStream(leftFilename.GetFullPath(), right);
 103     if (s && s->IsOk() )
 104     {
 105         return new wxFSFile(s,
 106                             left + wxT("#zip:") + right,
 107                             GetMimeTypeFromExt(location),
 108                             GetAnchor(location)
 109 #if wxUSE_DATETIME
 110                             , wxDateTime(wxFileModificationTime(left))
 111 #endif // wxUSE_DATETIME
 112                             );
 113     }
 114
 115     delete s;
 116     return NULL;
 117 }
 118
 119
 120
 121 wxString wxZipFSHandler::FindFirst(const wxString& spec, int flags)
 122 {
 123     wxString right = GetRightLocation(spec);
 124     wxString left = GetLeftLocation(spec);
 125
 126     if (right.Last() == wxT('/')) right.RemoveLast();
 127
 128     if (m_Archive)
 129     {
 130         unzClose((unzFile)m_Archive);
 131         m_Archive = NULL;
 132     }
 133
 134     if (GetProtocol(left) != wxT("file"))
 135     {
 136         wxLogError(_("ZIP handler currently supports only local files!"));
 137         return wxEmptyString;
 138     }
 139
 140     switch (flags)
 141     {
 142         case wxFILE:
 143             m_AllowDirs = FALSE, m_AllowFiles = TRUE; break;
 144         case wxDIR:
 145             m_AllowDirs = TRUE, m_AllowFiles = FALSE; break;
 146         default:
 147             m_AllowDirs = m_AllowFiles = TRUE; break;
 148     }
 149
 150     m_ZipFile = left;
 151     wxString nativename = wxFileSystem::URLToFileName(m_ZipFile).GetFullPath();
 152     m_Archive = (void*) unzOpen(nativename.mb_str());
 153     m_Pattern = right.AfterLast(wxT('/'));
 154     m_BaseDir = right.BeforeLast(wxT('/'));
 155
 156     if (m_Archive)
 157     {
 158         if (unzGoToFirstFile((unzFile)m_Archive) != UNZ_OK)
 159         {
 160             unzClose((unzFile)m_Archive);
 161             m_Archive = NULL;
 162         }
 163         else
 164         {
 165             if (m_AllowDirs)
 166             {
 167                 delete m_DirsFound;
 168                 m_DirsFound = new wxHashTableLong();
 169             }
 170             return DoFind();
 171         }
 172     }
 173     return wxEmptyString;
 174 }
 175
 176
 177
 178 wxString wxZipFSHandler::FindNext()
 179 {
 180     if (!m_Archive) return wxEmptyString;
 181     return DoFind();
 182 }
 183
 184
 185
 186 wxString wxZipFSHandler::DoFind()
 187 {
 188     static char namebuf[1024]; // char, not wxChar!
 189     char *c;
 190     wxString namestr, dir, filename;
 191     wxString match = wxEmptyString;
 192
 193     while (match == wxEmptyString)
 194     {
 195         unzGetCurrentFileInfo((unzFile)m_Archive, NULL, namebuf, 1024, NULL, 0, NULL, 0);
 196         for (c = namebuf; *c; c++) if (*c == '\\') *c = '/';
 197         namestr = wxString::FromAscii( namebuf );    // TODO what encoding does ZIP use?
 198
 199         if (m_AllowDirs)
 200         {
 201             dir = namestr.BeforeLast(wxT('/'));
 202             while (!dir.IsEmpty())
 203             {
 204                 long key = 0;
 205                 for (size_t i = 0; i < dir.Length(); i++) key += (wxUChar)dir[i];
 206                 if (m_DirsFound->Get(key) == wxNOT_FOUND)
 207                 {
 208                     m_DirsFound->Put(key, 1);
 209                     filename = dir.AfterLast(wxT('/'));
 210                     dir = dir.BeforeLast(wxT('/'));
 211                     if (!filename.IsEmpty() && m_BaseDir == dir &&
 212                                 wxMatchWild(m_Pattern, filename, FALSE))
 213                         match = m_ZipFile + wxT("#zip:") + dir + wxT("/") + filename;
 214                 }
 215                 else
 216                     break; // already tranversed
 217             }
 218         }
 219
 220         filename = namestr.AfterLast(wxT('/'));
 221         dir = namestr.BeforeLast(wxT('/'));
 222         if (m_AllowFiles && !filename.IsEmpty() && m_BaseDir == dir &&
 223                             wxMatchWild(m_Pattern, filename, FALSE))
 224             match = m_ZipFile + wxT("#zip:") + namestr;
 225
 226         if (unzGoToNextFile((unzFile)m_Archive) != UNZ_OK)
 227         {
 228             unzClose((unzFile)m_Archive);
 229             m_Archive = NULL;
 230             break;
 231         }
 232     }
 233
 234     return match;
 235 }
 236
 237
 238
 239 #endif
 240       //wxUSE_FILESYSTEM && wxUSE_FS_ZIP && wxUSE_ZIPSTREAM