// Author: Vaclav Slavik
// Copyright: (c) 1999 Vaclav Slavik
// CVS-ID: $Id$
-// Licence: wxWindows Licence
+// Licence: wxWindows licence
/////////////////////////////////////////////////////////////////////////////
-
-
-#ifdef __GNUG__
-#pragma implementation
-#endif
-
#include "wx/wxprec.h"
-#ifdef __BORDLANDC__
+#ifdef __BORLANDC__
#pragma hdrstop
#endif
-#if wxUSE_FILESYSTEM && wxUSE_FS_ZIP && wxUSE_ZIPSTREAM
+#if wxUSE_FILESYSTEM && wxUSE_FS_ZIP && wxUSE_ZIPSTREAM && wxUSE_ZLIB
#ifndef WXPRECOMP
-#include "wx/wx.h"
+ #include "wx/intl.h"
+ #include "wx/log.h"
#endif
-#include "wx/hash.h"
#include "wx/filesys.h"
+#include "wx/wfstream.h"
#include "wx/zipstrm.h"
#include "wx/fs_zip.h"
-/* Not the right solution (paths in makefiles) but... */
-#ifdef __BORLANDC__
-#include "../common/unzip.h"
-#else
-#include "unzip.h"
-#endif
+//---------------------------------------------------------------------------
+// wxZipFSInputStream
+//---------------------------------------------------------------------------
+// Helper class for wxZipFSHandler
-//--------------------------------------------------------------------------------
-// wxZipFSHandler
-//--------------------------------------------------------------------------------
+class wxZipFSInputStream : public wxZipInputStream
+{
+ public:
+ wxZipFSInputStream(wxFSFile *file)
+ : wxZipInputStream(*file->GetStream())
+ {
+ m_file = file;
+#if 1 //WXWIN_COMPATIBILITY_2_6
+ m_allowSeeking = true;
+#endif
+ }
+ virtual ~wxZipFSInputStream() { delete m_file; }
+ private:
+ wxFSFile *m_file;
+};
+
+//----------------------------------------------------------------------------
+// wxZipFSHandler
+//----------------------------------------------------------------------------
wxZipFSHandler::wxZipFSHandler() : wxFileSystemHandler()
{
m_Archive = NULL;
m_ZipFile = m_Pattern = m_BaseDir = wxEmptyString;
- m_AllowDirs = m_AllowFiles = TRUE;
+ m_AllowDirs = m_AllowFiles = true;
m_DirsFound = NULL;
}
wxZipFSHandler::~wxZipFSHandler()
{
- if (m_Archive)
- unzClose((unzFile)m_Archive);
- if (m_DirsFound)
- delete m_DirsFound;
+ Cleanup();
}
+void wxZipFSHandler::Cleanup()
+{
+ wxDELETE(m_Archive);
+ wxDELETE(m_DirsFound);
+}
+
+
bool wxZipFSHandler::CanOpen(const wxString& location)
{
wxString p = GetProtocol(location);
- return (p == wxT("zip")) &&
- (GetProtocol(GetLeftLocation(location)) == wxT("file"));
+ return (p == wxT("zip"));
}
-
-
wxFSFile* wxZipFSHandler::OpenFile(wxFileSystem& WXUNUSED(fs), const wxString& location)
{
wxString right = GetRightLocation(location);
wxString left = GetLeftLocation(location);
- wxInputStream *s;
+ wxZipInputStream *s;
- if (GetProtocol(left) != wxT("file"))
+ if (right.Contains(wxT("./")))
{
- wxLogError(_("ZIP handler currently supports only local files!"));
- return NULL;
+ if (right.GetChar(0) != wxT('/')) right = wxT('/') + right;
+ wxFileName rightPart(right, wxPATH_UNIX);
+ rightPart.Normalize(wxPATH_NORM_DOTS, wxT("/"), wxPATH_UNIX);
+ right = rightPart.GetFullPath(wxPATH_UNIX);
}
- s = new wxZipInputStream(left, right);
- if (s && (s->LastError() == wxStream_NOERROR))
+ if (right.GetChar(0) == wxT('/')) right = right.Mid(1);
+
+ // a new wxFileSystem object is needed here to avoid infinite recursion
+ wxFSFile *leftFile = wxFileSystem().OpenFile(left);
+ if (!leftFile)
+ return NULL;
+
+ s = new wxZipFSInputStream(leftFile);
+ if (s && s->IsOk())
{
- return new wxFSFile(s,
+ bool found = false;
+ while (!found)
+ {
+ wxZipEntry *ent = s->GetNextEntry();
+ if (!ent)
+ break;
+ if (ent->GetInternalName() == right)
+ found = true;
+ delete ent;
+ }
+ if (found)
+ return new wxFSFile(s,
left + wxT("#zip:") + right,
GetMimeTypeFromExt(location),
- GetAnchor(location),
- wxDateTime(wxFileModificationTime(left)));
+ GetAnchor(location)
+#if wxUSE_DATETIME
+ , wxDateTime(wxFileModificationTime(left))
+#endif // wxUSE_DATETIME
+ );
}
delete s;
{
wxString right = GetRightLocation(spec);
wxString left = GetLeftLocation(spec);
-
- if (right.Last() == wxT('/')) right.RemoveLast();
- if (m_Archive)
- {
- unzClose((unzFile)m_Archive);
- m_Archive = NULL;
- }
+ if (!right.empty() && right.Last() == wxT('/')) right.RemoveLast();
- if (GetProtocol(left) != wxT("file"))
+ if (m_Archive)
{
- wxLogError(_("ZIP handler currently supports only local files!"));
- return wxEmptyString;
+ delete m_Archive;
+ m_Archive = NULL;
}
- switch (flags)
+ switch (flags)
{
- case wxFILE:
- m_AllowDirs = FALSE, m_AllowFiles = TRUE; break;
- case wxDIR:
- m_AllowDirs = TRUE, m_AllowFiles = FALSE; break;
- default:
- m_AllowDirs = m_AllowFiles = TRUE; break;
+ case wxFILE:
+ m_AllowDirs = false, m_AllowFiles = true; break;
+ case wxDIR:
+ m_AllowDirs = true, m_AllowFiles = false; break;
+ default:
+ m_AllowDirs = m_AllowFiles = true; break;
}
m_ZipFile = left;
- m_Archive = (void*) unzOpen(m_ZipFile.mb_str());
+
+ wxFSFile *leftFile = wxFileSystem().OpenFile(left);
+ if (leftFile)
+ m_Archive = new wxZipFSInputStream(leftFile);
+
m_Pattern = right.AfterLast(wxT('/'));
m_BaseDir = right.BeforeLast(wxT('/'));
+ if (m_BaseDir.StartsWith(wxT("/")))
+ m_BaseDir = m_BaseDir.Mid(1);
- if (m_Archive)
+ if (m_Archive)
{
- if (unzGoToFirstFile((unzFile)m_Archive) != UNZ_OK)
- {
- unzClose((unzFile)m_Archive);
- m_Archive = NULL;
- }
- else
+ if (m_AllowDirs)
{
- if (m_AllowDirs)
- {
- delete m_DirsFound;
- m_DirsFound = new wxHashTableLong();
- }
- return DoFind();
+ delete m_DirsFound;
+ m_DirsFound = new wxZipFilenameHashMap();
+ if (right.empty()) // allow "/" to match the archive root
+ return spec;
}
+ return DoFind();
}
return wxEmptyString;
}
wxString wxZipFSHandler::DoFind()
{
- static char namebuf[1024]; // char, not wxChar!
- char *c;
wxString namestr, dir, filename;
wxString match = wxEmptyString;
while (match == wxEmptyString)
{
- unzGetCurrentFileInfo((unzFile)m_Archive, NULL, namebuf, 1024, NULL, 0, NULL, 0);
- for (c = namebuf; *c; c++) if (*c == wxT('\\')) *c = wxT('/');
- namestr = namebuf;
+ wxZipEntry *entry = m_Archive->GetNextEntry();
+ if (!entry)
+ {
+ delete m_Archive;
+ m_Archive = NULL;
+ break;
+ }
+ namestr = entry->GetName(wxPATH_UNIX);
+ delete entry;
if (m_AllowDirs)
{
- dir = namestr.BeforeLast(wxT('/'));
- while (!dir.IsEmpty())
+ dir = namestr.BeforeLast(wxT('/'));
+ while (!dir.empty())
{
- long key = 0;
- for (size_t i = 0; i < dir.Length(); i++) key += (wxUChar)dir[i];
- if (m_DirsFound->Get(key) == wxNOT_FOUND)
+ if( m_DirsFound->find(dir) == m_DirsFound->end() )
{
- m_DirsFound->Put(key, 1);
+ (*m_DirsFound)[dir] = 1;
filename = dir.AfterLast(wxT('/'));
dir = dir.BeforeLast(wxT('/'));
- if (!filename.IsEmpty() && m_BaseDir == dir &&
- wxMatchWild(m_Pattern, filename, FALSE))
+ if (!filename.empty() && m_BaseDir == dir &&
+ wxMatchWild(m_Pattern, filename, false))
match = m_ZipFile + wxT("#zip:") + dir + wxT("/") + filename;
}
- else
+ else
break; // already tranversed
}
}
filename = namestr.AfterLast(wxT('/'));
dir = namestr.BeforeLast(wxT('/'));
- if (m_AllowFiles && !filename.IsEmpty() && m_BaseDir == dir &&
- wxMatchWild(m_Pattern, filename, FALSE))
+ if (m_AllowFiles && !filename.empty() && m_BaseDir == dir &&
+ wxMatchWild(m_Pattern, filename, false))
match = m_ZipFile + wxT("#zip:") + namestr;
-
- 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