зеркало из https://github.com/mozilla/gecko-dev.git
915 строки
25 KiB
C++
915 строки
25 KiB
C++
/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* ***** BEGIN LICENSE BLOCK *****
|
|
* Version: MPL 1.1/GPL 2.0/LGPL 2.1
|
|
*
|
|
* The contents of this file are subject to the Mozilla Public License Version
|
|
* 1.1 (the "License"); you may not use this file except in compliance with
|
|
* the License. You may obtain a copy of the License at
|
|
* http://www.mozilla.org/MPL/
|
|
*
|
|
* Software distributed under the License is distributed on an "AS IS" basis,
|
|
* WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
|
|
* for the specific language governing rights and limitations under the
|
|
* License.
|
|
*
|
|
* The Original Code is Mozilla Communicator client code, released
|
|
* March 31, 1998.
|
|
*
|
|
* The Initial Developer of the Original Code is
|
|
* Netscape Communications Corporation.
|
|
* Portions created by the Initial Developer are Copyright (C) 1998
|
|
* the Initial Developer. All Rights Reserved.
|
|
*
|
|
* Contributor(s):
|
|
* Daniel Veditz <dveditz@netscape.com>
|
|
* Samir Gehani <sgehani@netscape.com>
|
|
* Mitch Stoltz <mstoltz@netscape.com>
|
|
* Jeroen Dobbelaere <jeroen.dobbelaere@acunia.com>
|
|
* Jeff Walden <jwalden+code@mit.edu>
|
|
* Taras Glek <tglek@mozilla.com>
|
|
*
|
|
* Alternatively, the contents of this file may be used under the terms of
|
|
* either the GNU General Public License Version 2 or later (the "GPL"), or
|
|
* the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
|
|
* in which case the provisions of the GPL or the LGPL are applicable instead
|
|
* of those above. If you wish to allow use of your version of this file only
|
|
* under the terms of either the GPL or the LGPL, and not to allow others to
|
|
* use your version of this file under the terms of the MPL, indicate your
|
|
* decision by deleting the provisions above and replace them with the notice
|
|
* and other provisions required by the GPL or the LGPL. If you do not delete
|
|
* the provisions above, a recipient may use your version of this file under
|
|
* the terms of any one of the MPL, the GPL or the LGPL.
|
|
*
|
|
* ***** END LICENSE BLOCK ***** */
|
|
|
|
/*
|
|
* This module implements a simple archive extractor for the PKZIP format.
|
|
*
|
|
* The underlying nsZipArchive is NOT thread-safe. Do not pass references
|
|
* or pointers to it across thread boundaries.
|
|
*/
|
|
|
|
|
|
#define READTYPE PRInt32
|
|
#include "zlib.h"
|
|
#include "nsISupportsUtils.h"
|
|
#include "nsRecyclingAllocator.h"
|
|
#include "prio.h"
|
|
#include "plstr.h"
|
|
#include "prlog.h"
|
|
#include "stdlib.h"
|
|
#include "nsWildCard.h"
|
|
#include "zipstruct.h"
|
|
#include "nsZipArchive.h"
|
|
|
|
/**
|
|
* Global allocator used with zlib. Destroyed in module shutdown.
|
|
*/
|
|
#define NBUCKETS 6
|
|
nsRecyclingAllocator *gZlibAllocator = NULL;
|
|
|
|
// For placement new used for arena allocations of zip file list
|
|
#include NEW_H
|
|
#define ZIP_ARENABLOCKSIZE (1*1024)
|
|
|
|
#ifdef XP_UNIX
|
|
#include <sys/types.h>
|
|
#include <sys/stat.h>
|
|
#include <limits.h>
|
|
#include <unistd.h>
|
|
#elif defined(XP_WIN) || defined(XP_OS2)
|
|
#include <io.h>
|
|
#elif defined(XP_BEOS)
|
|
#include <unistd.h>
|
|
#endif
|
|
|
|
#ifdef __SYMBIAN32__
|
|
#include <sys/syslimits.h>
|
|
#endif /*__SYMBIAN32__*/
|
|
|
|
|
|
#ifndef XP_UNIX /* we need some constants defined in limits.h and unistd.h */
|
|
# ifndef S_IFMT
|
|
# define S_IFMT 0170000
|
|
# endif
|
|
# ifndef S_IFLNK
|
|
# define S_IFLNK 0120000
|
|
# endif
|
|
# ifndef PATH_MAX
|
|
# define PATH_MAX 1024
|
|
# endif
|
|
#endif /* XP_UNIX */
|
|
|
|
static PRUint16 xtoint(unsigned char *ii);
|
|
static PRUint32 xtolong(unsigned char *ll);
|
|
static PRUint16 ExtractMode(unsigned char *ll);
|
|
static PRUint32 HashName(const char* aName);
|
|
#if defined(XP_UNIX) || defined(XP_BEOS)
|
|
static PRBool IsSymlink(unsigned char *ll);
|
|
static nsresult ResolveSymlink(const char *path);
|
|
#endif
|
|
|
|
//***********************************************************
|
|
// Allocators for use with zlib
|
|
//
|
|
// Use a recycling allocator, for re-use of of the zlib buffers.
|
|
// For every inflation the following allocations are done:
|
|
// zlibAlloc(1, 9520)
|
|
// zlibAlloc(32768, 1)
|
|
//***********************************************************
|
|
|
|
static void *
|
|
zlibAlloc(void *opaque, uInt items, uInt size)
|
|
{
|
|
nsRecyclingAllocator *zallocator = (nsRecyclingAllocator *)opaque;
|
|
if (gZlibAllocator) {
|
|
return gZlibAllocator->Calloc(items, size);
|
|
}
|
|
return calloc(items, size);
|
|
}
|
|
|
|
static void
|
|
zlibFree(void *opaque, void *ptr)
|
|
{
|
|
nsRecyclingAllocator *zallocator = (nsRecyclingAllocator *)opaque;
|
|
if (zallocator)
|
|
zallocator->Free(ptr);
|
|
else
|
|
free(ptr);
|
|
}
|
|
|
|
nsresult gZlibInit(z_stream *zs)
|
|
{
|
|
memset(zs, 0, sizeof(z_stream));
|
|
//-- ensure we have our zlib allocator for better performance
|
|
if (!gZlibAllocator) {
|
|
gZlibAllocator = new nsRecyclingAllocator(NBUCKETS, NS_DEFAULT_RECYCLE_TIMEOUT, "libjar");
|
|
}
|
|
if (gZlibAllocator) {
|
|
zs->zalloc = zlibAlloc;
|
|
zs->zfree = zlibFree;
|
|
zs->opaque = gZlibAllocator;
|
|
}
|
|
int zerr = inflateInit2(zs, -MAX_WBITS);
|
|
if (zerr != Z_OK) return NS_ERROR_OUT_OF_MEMORY;
|
|
|
|
return NS_OK;
|
|
}
|
|
|
|
nsZipHandle::nsZipHandle()
|
|
: mFd(nsnull)
|
|
, mFileData(nsnull)
|
|
, mLen(0)
|
|
, mMap(nsnull)
|
|
, mRefCnt(0)
|
|
{
|
|
MOZ_COUNT_CTOR(nsZipHandle);
|
|
}
|
|
|
|
NS_IMPL_THREADSAFE_ADDREF(nsZipHandle)
|
|
NS_IMPL_THREADSAFE_RELEASE(nsZipHandle)
|
|
|
|
nsresult nsZipHandle::Init(PRFileDesc *fd, nsZipHandle **ret)
|
|
{
|
|
PRInt64 size = PR_Available64(fd);
|
|
if (size >= PR_INT32_MAX)
|
|
return NS_ERROR_FILE_TOO_BIG;
|
|
|
|
PRFileMap *map = PR_CreateFileMap(fd, size, PR_PROT_READONLY);
|
|
if (!map)
|
|
return NS_ERROR_FAILURE;
|
|
|
|
nsZipHandle *handle = new nsZipHandle();
|
|
if (!handle) {
|
|
PR_CloseFileMap(map);
|
|
return NS_ERROR_OUT_OF_MEMORY;
|
|
}
|
|
|
|
handle->mFd = fd;
|
|
handle->mMap = map;
|
|
handle->mLen = (PRUint32) size;
|
|
handle->mFileData = (PRUint8*) PR_MemMap(map, 0, handle->mLen);
|
|
handle->AddRef();
|
|
*ret = handle;
|
|
return NS_OK;
|
|
}
|
|
|
|
nsZipHandle::~nsZipHandle()
|
|
{
|
|
if (mFileData) {
|
|
PR_MemUnmap(mFileData, mLen);
|
|
PR_CloseFileMap(mMap);
|
|
mFileData = nsnull;
|
|
mMap = nsnull;
|
|
}
|
|
if (mFd) {
|
|
PR_Close(mFd);
|
|
mFd = nsnull;
|
|
}
|
|
MOZ_COUNT_DTOR(nsZipHandle);
|
|
}
|
|
|
|
//***********************************************************
|
|
// nsZipArchive -- public methods
|
|
//***********************************************************
|
|
|
|
|
|
//---------------------------------------------
|
|
// nsZipArchive::OpenArchive
|
|
//---------------------------------------------
|
|
nsresult nsZipArchive::OpenArchive(PRFileDesc * fd)
|
|
{
|
|
nsresult rv = nsZipHandle::Init(fd, getter_AddRefs(mFd));
|
|
if (NS_FAILED(rv))
|
|
return rv;
|
|
|
|
// Initialize our arena
|
|
PL_INIT_ARENA_POOL(&mArena, "ZipArena", ZIP_ARENABLOCKSIZE);
|
|
|
|
//-- get table of contents for archive
|
|
return BuildFileList();
|
|
}
|
|
|
|
//---------------------------------------------
|
|
// nsZipArchive::Test
|
|
//---------------------------------------------
|
|
nsresult nsZipArchive::Test(const char *aEntryName)
|
|
{
|
|
nsZipItem* currItem;
|
|
|
|
if (aEntryName) // only test specified item
|
|
{
|
|
currItem = GetItem(aEntryName);
|
|
if (!currItem)
|
|
return NS_ERROR_FILE_TARGET_DOES_NOT_EXIST;
|
|
//-- don't test (synthetic) directory items
|
|
if (currItem->isDirectory)
|
|
return NS_OK;
|
|
return ExtractFile(currItem, 0, 0);
|
|
}
|
|
|
|
// test all items in archive
|
|
for (int i = 0; i < ZIP_TABSIZE; i++) {
|
|
for (currItem = mFiles[i]; currItem; currItem = currItem->next) {
|
|
//-- don't test (synthetic) directory items
|
|
if (currItem->isDirectory)
|
|
continue;
|
|
nsresult rv = ExtractFile(currItem, 0, 0);
|
|
if (rv != NS_OK)
|
|
return rv;
|
|
}
|
|
}
|
|
|
|
return NS_OK;
|
|
}
|
|
|
|
//---------------------------------------------
|
|
// nsZipArchive::CloseArchive
|
|
//---------------------------------------------
|
|
nsresult nsZipArchive::CloseArchive()
|
|
{
|
|
if (mFd) {
|
|
PL_FinishArenaPool(&mArena);
|
|
mFd = NULL;
|
|
}
|
|
|
|
// CAUTION:
|
|
// We don't need to delete each of the nsZipItem as the memory for
|
|
// the zip item and the filename it holds are both allocated from the Arena.
|
|
// Hence, destroying the Arena is like destroying all the memory
|
|
// for all the nsZipItem in one shot. But if the ~nsZipItem is doing
|
|
// anything more than cleaning up memory, we should start calling it.
|
|
// Let us also cleanup the mFiles table for re-use on the next 'open' call
|
|
memset(mFiles, 0, sizeof(mFiles));
|
|
mBuiltSynthetics = false;
|
|
return NS_OK;
|
|
}
|
|
|
|
//---------------------------------------------
|
|
// nsZipArchive::GetItem
|
|
//---------------------------------------------
|
|
nsZipItem* nsZipArchive::GetItem(const char * aEntryName)
|
|
{
|
|
if (aEntryName) {
|
|
//-- If the request is for a directory, make sure that synthetic entries
|
|
//-- are created for the directories without their own entry.
|
|
if (!mBuiltSynthetics) {
|
|
PRUint32 len = strlen(aEntryName);
|
|
if ((len > 0) && (aEntryName[len-1] == '/')) {
|
|
if (BuildSynthetics() != NS_OK)
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
nsZipItem* item = mFiles[ HashName(aEntryName) ];
|
|
while (item) {
|
|
if (!strcmp(aEntryName, item->name))
|
|
return item; //-- found it
|
|
item = item->next;
|
|
}
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
//---------------------------------------------
|
|
// nsZipArchive::ExtractFile
|
|
// This extracts the item to the filehandle provided.
|
|
// If 'aFd' is null, it only tests the extraction.
|
|
// On extraction error(s) it removes the file.
|
|
// When needed, it also resolves the symlink.
|
|
//---------------------------------------------
|
|
nsresult nsZipArchive::ExtractFile(nsZipItem *item, const char *outname,
|
|
PRFileDesc* aFd)
|
|
{
|
|
if (!item)
|
|
return NS_ERROR_ILLEGAL_VALUE;
|
|
if (!mFd)
|
|
return NS_ERROR_FAILURE;
|
|
|
|
// Directory extraction is handled in nsJAR::Extract,
|
|
// so the item to be extracted should never be a directory
|
|
PR_ASSERT(!item->isDirectory);
|
|
|
|
nsresult rv;
|
|
|
|
//-- extract the file using the appropriate method
|
|
switch(item->compression)
|
|
{
|
|
case STORED:
|
|
rv = CopyItemToDisk(item, aFd);
|
|
break;
|
|
|
|
case DEFLATED:
|
|
rv = InflateItem(item, aFd);
|
|
break;
|
|
|
|
default:
|
|
//-- unsupported compression type
|
|
rv = NS_ERROR_NOT_IMPLEMENTED;
|
|
}
|
|
|
|
//-- delete the file on errors, or resolve symlink if needed
|
|
if (aFd) {
|
|
PR_Close(aFd);
|
|
if (rv != NS_OK)
|
|
PR_Delete(outname);
|
|
#if defined(XP_UNIX) || defined(XP_BEOS)
|
|
else if (item->isSymlink)
|
|
rv = ResolveSymlink(outname);
|
|
#endif
|
|
}
|
|
|
|
return rv;
|
|
}
|
|
|
|
//---------------------------------------------
|
|
// nsZipArchive::FindInit
|
|
//---------------------------------------------
|
|
PRInt32
|
|
nsZipArchive::FindInit(const char * aPattern, nsZipFind **aFind)
|
|
{
|
|
if (!aFind)
|
|
return NS_ERROR_ILLEGAL_VALUE;
|
|
|
|
// null out param in case an error happens
|
|
*aFind = NULL;
|
|
|
|
PRBool regExp = PR_FALSE;
|
|
char* pattern = 0;
|
|
|
|
// Create synthetic directory entries on demand
|
|
nsresult rv = BuildSynthetics();
|
|
if (rv != NS_OK)
|
|
return rv;
|
|
|
|
// validate the pattern
|
|
if (aPattern)
|
|
{
|
|
switch (NS_WildCardValid((char*)aPattern))
|
|
{
|
|
case INVALID_SXP:
|
|
return NS_ERROR_ILLEGAL_VALUE;
|
|
|
|
case NON_SXP:
|
|
regExp = PR_FALSE;
|
|
break;
|
|
|
|
case VALID_SXP:
|
|
regExp = PR_TRUE;
|
|
break;
|
|
|
|
default:
|
|
// undocumented return value from RegExpValid!
|
|
PR_ASSERT(PR_FALSE);
|
|
return NS_ERROR_ILLEGAL_VALUE;
|
|
}
|
|
|
|
pattern = PL_strdup(aPattern);
|
|
if (!pattern)
|
|
return NS_ERROR_OUT_OF_MEMORY;
|
|
}
|
|
|
|
*aFind = new nsZipFind(this, pattern, regExp);
|
|
if (!*aFind) {
|
|
PL_strfree(pattern);
|
|
return NS_ERROR_OUT_OF_MEMORY;
|
|
}
|
|
|
|
return NS_OK;
|
|
}
|
|
|
|
|
|
|
|
//---------------------------------------------
|
|
// nsZipFind::FindNext
|
|
//---------------------------------------------
|
|
nsresult nsZipFind::FindNext(const char ** aResult)
|
|
{
|
|
if (!mArchive || !aResult)
|
|
return NS_ERROR_ILLEGAL_VALUE;
|
|
|
|
*aResult = 0;
|
|
|
|
// we start from last match, look for next
|
|
while (mSlot < ZIP_TABSIZE)
|
|
{
|
|
// move to next in current chain, or move to new slot
|
|
mItem = mItem ? mItem->next : mArchive->mFiles[mSlot];
|
|
|
|
PRBool found = PR_FALSE;
|
|
if (!mItem)
|
|
++mSlot; // no more in this chain, move to next slot
|
|
else if (!mPattern)
|
|
found = PR_TRUE; // always match
|
|
else if (mRegExp)
|
|
found = (NS_WildCardMatch(mItem->name, mPattern, PR_FALSE) == MATCH);
|
|
else
|
|
found = (PL_strcmp(mItem->name, mPattern) == 0);
|
|
|
|
if (found) {
|
|
*aResult = mItem->name;
|
|
return NS_OK;
|
|
}
|
|
}
|
|
|
|
return NS_ERROR_FILE_TARGET_DOES_NOT_EXIST;
|
|
}
|
|
|
|
#if defined(XP_UNIX) || defined(XP_BEOS)
|
|
//---------------------------------------------
|
|
// ResolveSymlink
|
|
//---------------------------------------------
|
|
static nsresult ResolveSymlink(const char *path)
|
|
{
|
|
PRFileDesc * fIn = PR_Open(path, PR_RDONLY, 0000);
|
|
if (!fIn)
|
|
return NS_ERROR_FILE_DISK_FULL;
|
|
|
|
char buf[PATH_MAX+1];
|
|
PRInt32 length = PR_Read(fIn, (void*)buf, PATH_MAX);
|
|
PR_Close(fIn);
|
|
|
|
if ( (length <= 0)
|
|
|| ((buf[length] = 0, PR_Delete(path)) != 0)
|
|
|| (symlink(buf, path) != 0))
|
|
{
|
|
return NS_ERROR_FILE_DISK_FULL;
|
|
}
|
|
return NS_OK;
|
|
}
|
|
#endif
|
|
|
|
//***********************************************************
|
|
// nsZipArchive -- private implementation
|
|
//***********************************************************
|
|
|
|
#define BR_BUF_SIZE 1024 /* backward read buffer size */
|
|
|
|
//---------------------------------------------
|
|
// nsZipArchive::CreateZipItem
|
|
//---------------------------------------------
|
|
nsZipItem* nsZipArchive::CreateZipItem(PRUint16 namelen)
|
|
{
|
|
// sizeof(nsZipItem) includes space for name's null byte
|
|
// Arena allocate the nsZipItem
|
|
void *mem;
|
|
PL_ARENA_ALLOCATE(mem, &mArena, sizeof(nsZipItem)+namelen);
|
|
return (nsZipItem*)mem;
|
|
}
|
|
|
|
//---------------------------------------------
|
|
// nsZipArchive::BuildFileList
|
|
//---------------------------------------------
|
|
nsresult nsZipArchive::BuildFileList()
|
|
{
|
|
// Get archive size using end pos
|
|
PRUint8* buf;
|
|
PRUint8* endp = mFd->mFileData + mFd->mLen;
|
|
|
|
for (buf = endp - ZIPEND_SIZE; xtolong(buf) != ENDSIG; buf--)
|
|
{
|
|
if (buf == mFd->mFileData) {
|
|
// We're at the beginning of the file, and still no sign
|
|
// of the end signature. File must be corrupted!
|
|
return NS_ERROR_FILE_CORRUPTED;
|
|
}
|
|
}
|
|
PRUint32 central = xtolong(((ZipEnd *)buf)->offset_central_dir);
|
|
|
|
//-- Read the central directory headers
|
|
buf = mFd->mFileData + central;
|
|
PRUint32 sig = xtolong(buf);
|
|
while (sig == CENTRALSIG) {
|
|
// Make sure there is enough data available.
|
|
if (endp - buf < ZIPCENTRAL_SIZE)
|
|
return NS_ERROR_FILE_CORRUPTED;
|
|
|
|
// Read the fixed-size data.
|
|
ZipCentral* central = (ZipCentral*)buf;
|
|
|
|
PRUint16 namelen = xtoint(central->filename_len);
|
|
PRUint16 extralen = xtoint(central->extrafield_len);
|
|
PRUint16 commentlen = xtoint(central->commentfield_len);
|
|
|
|
// Sanity check variable sizes and refuse to deal with
|
|
// anything too big: it's likely a corrupt archive.
|
|
if (namelen > BR_BUF_SIZE || extralen > BR_BUF_SIZE || commentlen > 2*BR_BUF_SIZE)
|
|
return NS_ERROR_FILE_CORRUPTED;
|
|
|
|
nsZipItem* item = CreateZipItem(namelen);
|
|
if (!item)
|
|
return NS_ERROR_OUT_OF_MEMORY;
|
|
|
|
item->headerOffset = xtolong(central->localhdr_offset);
|
|
item->size = xtolong(central->size);
|
|
item->realsize = xtolong(central->orglen);
|
|
item->crc32 = xtolong(central->crc32);
|
|
item->time = xtoint(central->time);
|
|
item->date = xtoint(central->date);
|
|
item->isSynthetic = PR_FALSE;
|
|
item->compression = PR_MIN(xtoint(central->method), UNSUPPORTED);
|
|
item->mode = ExtractMode(central->external_attributes);
|
|
#if defined(XP_UNIX) || defined(XP_BEOS)
|
|
// Check if item is a symlink
|
|
item->isSymlink = IsSymlink(central->external_attributes);
|
|
#endif
|
|
|
|
buf += ZIPCENTRAL_SIZE;
|
|
|
|
// Get the item name
|
|
memcpy(item->name, buf, namelen);
|
|
item->name[namelen] = 0;
|
|
// An item whose name ends with '/' is a directory
|
|
item->isDirectory = ('/' == item->name[namelen - 1]);
|
|
|
|
// Add item to file table
|
|
PRUint32 hash = HashName(item->name);
|
|
item->next = mFiles[hash];
|
|
mFiles[hash] = item;
|
|
|
|
// Point to the next item at the top of loop
|
|
buf += namelen + extralen + commentlen;
|
|
sig = xtolong(buf);
|
|
} /* while reading central directory records */
|
|
|
|
if (sig != ENDSIG)
|
|
return NS_ERROR_FILE_CORRUPTED;
|
|
return NS_OK;
|
|
}
|
|
|
|
//---------------------------------------------
|
|
// nsZipArchive::BuildSynthetics
|
|
//---------------------------------------------
|
|
nsresult nsZipArchive::BuildSynthetics()
|
|
{
|
|
if (mBuiltSynthetics)
|
|
return NS_OK;
|
|
mBuiltSynthetics = true;
|
|
|
|
// Create synthetic entries for any missing directories.
|
|
// Do this when all ziptable has scanned to prevent double entries.
|
|
for (int i = 0; i < ZIP_TABSIZE; ++i)
|
|
{
|
|
for (nsZipItem* item = mFiles[i]; item != 0; item = item->next)
|
|
{
|
|
if (item->isSynthetic)
|
|
continue;
|
|
|
|
//-- add entries for directories in the current item's path
|
|
//-- go from end to beginning, because then we can stop trying
|
|
//-- to create diritems if we find that the diritem we want to
|
|
//-- create already exists
|
|
//-- start just before the last char so as to not add the item
|
|
//-- twice if it's a directory
|
|
PRUint16 namelen = strlen(item->name);
|
|
for (char* p = item->name + namelen - 2; p >= item->name; p--)
|
|
{
|
|
if ('/' != *p)
|
|
continue;
|
|
|
|
// See whether we need to create any more implicit directories,
|
|
// because if we don't we can avoid a lot of work.
|
|
// We can even avoid (de)allocating space for a bogus dirname with
|
|
// a little trickery -- save the char at item->name[dirnamelen],
|
|
// set it to 0, compare the strings, and restore the saved
|
|
// char when done
|
|
const PRUint32 dirnamelen = p + 1 - item->name;
|
|
const char savedChar = item->name[dirnamelen];
|
|
item->name[dirnamelen] = 0;
|
|
|
|
// Is the directory in the file table?
|
|
PRUint32 hash = HashName(item->name);
|
|
PRBool found = PR_FALSE;
|
|
for (nsZipItem* zi = mFiles[hash]; zi != NULL; zi = zi->next)
|
|
{
|
|
if (0 == strcmp(item->name, zi->name))
|
|
{
|
|
// we've already added this dir and all its parents
|
|
found = PR_TRUE;
|
|
break;
|
|
}
|
|
}
|
|
|
|
// restore the char immediately
|
|
item->name[dirnamelen] = savedChar;
|
|
|
|
// if the directory was found, break out of the directory
|
|
// creation loop now that we know all implicit directories
|
|
// are there -- otherwise, start creating the zip item
|
|
if (found)
|
|
break;
|
|
|
|
nsZipItem* diritem = CreateZipItem(dirnamelen);
|
|
if (!diritem)
|
|
return NS_ERROR_OUT_OF_MEMORY;
|
|
|
|
memcpy(diritem->name, item->name, dirnamelen);
|
|
diritem->name[dirnamelen] = 0;
|
|
|
|
diritem->isDirectory = PR_TRUE;
|
|
diritem->isSynthetic = PR_TRUE;
|
|
diritem->compression = STORED;
|
|
diritem->size = diritem->realsize = 0;
|
|
diritem->crc32 = 0;
|
|
diritem->mode = 0755;
|
|
|
|
// Set an obviously wrong last-modified date/time, because
|
|
// finding something more accurate like the most recent
|
|
// last-modified date/time of the dir's contents is a lot
|
|
// of effort. The date/time corresponds to 1980-01-01 00:00.
|
|
diritem->time = 0;
|
|
diritem->date = 1 + (1 << 5) + (0 << 9);
|
|
|
|
// add diritem to the file table
|
|
diritem->next = mFiles[hash];
|
|
mFiles[hash] = diritem;
|
|
} /* end processing of dirs in item's name */
|
|
}
|
|
}
|
|
return NS_OK;
|
|
}
|
|
|
|
nsZipHandle* nsZipArchive::GetFD()
|
|
{
|
|
if (!mFd)
|
|
return NULL;
|
|
return mFd.get();
|
|
}
|
|
|
|
//---------------------------------------------
|
|
// nsZipArchive::GetData
|
|
//---------------------------------------------
|
|
PRUint8* nsZipArchive::GetData(nsZipItem* aItem)
|
|
{
|
|
PR_ASSERT (aItem);
|
|
|
|
//-- read local header to get variable length values and calculate
|
|
//-- the real data offset
|
|
if (aItem->headerOffset + ZIPLOCAL_SIZE > mFd->mLen)
|
|
return nsnull;
|
|
|
|
// -- check signature before using the structure, in case the zip file is corrupt
|
|
ZipLocal* Local = (ZipLocal*)(mFd->mFileData + aItem->headerOffset);
|
|
if ((xtolong(Local->signature) != LOCALSIG))
|
|
return nsnull;
|
|
|
|
//-- NOTE: extralen is different in central header and local header
|
|
//-- for archives created using the Unix "zip" utility. To set
|
|
//-- the offset accurately we need the _local_ extralen.
|
|
PRUint32 dataOffset = aItem->headerOffset +
|
|
ZIPLOCAL_SIZE +
|
|
xtoint(Local->filename_len) +
|
|
xtoint(Local->extrafield_len);
|
|
|
|
// -- check if there is enough source data in the file
|
|
if (dataOffset + aItem->size > mFd->mLen)
|
|
return nsnull;
|
|
|
|
return mFd->mFileData + dataOffset;
|
|
}
|
|
|
|
//---------------------------------------------
|
|
// nsZipArchive::CopyItemToDisk
|
|
//---------------------------------------------
|
|
nsresult
|
|
nsZipArchive::CopyItemToDisk(nsZipItem *item, PRFileDesc* outFD)
|
|
{
|
|
PR_ASSERT(item);
|
|
|
|
//-- get to the start of file's data
|
|
const PRUint8* itemData = GetData(item);
|
|
if (!itemData)
|
|
return NS_ERROR_FILE_CORRUPTED;
|
|
|
|
if (outFD && PR_Write(outFD, itemData, item->size) < (READTYPE)item->size)
|
|
{
|
|
//-- Couldn't write all the data (disk full?)
|
|
return NS_ERROR_FILE_DISK_FULL;
|
|
}
|
|
|
|
//-- Calculate crc
|
|
PRUint32 crc = crc32(0L, (const unsigned char*)itemData, item->size);
|
|
//-- verify crc32
|
|
if (crc != item->crc32)
|
|
return NS_ERROR_FILE_CORRUPTED;
|
|
|
|
return NS_OK;
|
|
}
|
|
|
|
|
|
//---------------------------------------------
|
|
// nsZipArchive::InflateItem
|
|
//---------------------------------------------
|
|
nsresult nsZipArchive::InflateItem(nsZipItem * item, PRFileDesc* outFD)
|
|
/*
|
|
* This function inflates an archive item to disk, to the
|
|
* file specified by outFD. If outFD is zero, the extracted data is
|
|
* not written, only checked for CRC, so this is in effect same as 'Test'.
|
|
*/
|
|
{
|
|
PR_ASSERT(item);
|
|
//-- allocate deflation buffers
|
|
Bytef outbuf[ZIP_BUFLEN];
|
|
|
|
//-- set up the inflate
|
|
z_stream zs;
|
|
nsresult status = gZlibInit(&zs);
|
|
if (status != NS_OK)
|
|
return NS_ERROR_FAILURE;
|
|
|
|
//-- inflate loop
|
|
zs.avail_in = item->size;
|
|
zs.next_in = (Bytef*)GetData(item);
|
|
if (!zs.next_in)
|
|
return NS_ERROR_FILE_CORRUPTED;
|
|
|
|
PRUint32 crc = crc32(0L, Z_NULL, 0);
|
|
int zerr = Z_OK;
|
|
while (zerr == Z_OK)
|
|
{
|
|
zs.next_out = outbuf;
|
|
zs.avail_out = ZIP_BUFLEN;
|
|
|
|
zerr = inflate(&zs, Z_PARTIAL_FLUSH);
|
|
if (zerr != Z_OK && zerr != Z_STREAM_END)
|
|
{
|
|
status = (zerr == Z_MEM_ERROR) ? NS_ERROR_OUT_OF_MEMORY : NS_ERROR_FILE_CORRUPTED;
|
|
break;
|
|
}
|
|
PRUint32 count = zs.next_out - outbuf;
|
|
|
|
//-- incrementally update crc32
|
|
crc = crc32(crc, (const unsigned char*)outbuf, count);
|
|
|
|
if (outFD && PR_Write(outFD, outbuf, count) < (READTYPE)count)
|
|
{
|
|
status = NS_ERROR_FILE_DISK_FULL;
|
|
break;
|
|
}
|
|
} // while
|
|
|
|
//-- free zlib internal state
|
|
inflateEnd(&zs);
|
|
|
|
//-- verify crc32
|
|
if ((status == NS_OK) && (crc != item->crc32))
|
|
{
|
|
status = NS_ERROR_FILE_CORRUPTED;
|
|
}
|
|
return status;
|
|
}
|
|
|
|
//------------------------------------------
|
|
// nsZipArchive constructor and destructor
|
|
//------------------------------------------
|
|
|
|
nsZipArchive::nsZipArchive() :
|
|
mBuiltSynthetics(false)
|
|
{
|
|
MOZ_COUNT_CTOR(nsZipArchive);
|
|
|
|
// initialize the table to NULL
|
|
memset(mFiles, 0, sizeof(mFiles));
|
|
}
|
|
|
|
nsZipArchive::~nsZipArchive()
|
|
{
|
|
CloseArchive();
|
|
|
|
MOZ_COUNT_DTOR(nsZipArchive);
|
|
}
|
|
|
|
|
|
//------------------------------------------
|
|
// nsZipFind constructor and destructor
|
|
//------------------------------------------
|
|
|
|
nsZipFind::nsZipFind(nsZipArchive* aZip, char* aPattern, PRBool aRegExp) :
|
|
mArchive(aZip),
|
|
mPattern(aPattern),
|
|
mItem(0),
|
|
mSlot(0),
|
|
mRegExp(aRegExp)
|
|
{
|
|
MOZ_COUNT_CTOR(nsZipFind);
|
|
}
|
|
|
|
nsZipFind::~nsZipFind()
|
|
{
|
|
PL_strfree(mPattern);
|
|
|
|
MOZ_COUNT_DTOR(nsZipFind);
|
|
}
|
|
|
|
//------------------------------------------
|
|
// helper functions
|
|
//------------------------------------------
|
|
|
|
/*
|
|
* HashName
|
|
*
|
|
* returns a hash key for the entry name
|
|
*/
|
|
static PRUint32 HashName(const char* aName)
|
|
{
|
|
PR_ASSERT(aName != 0);
|
|
|
|
PRUint32 val = 0;
|
|
for (PRUint8* c = (PRUint8*)aName; *c != 0; c++) {
|
|
val = val*37 + *c;
|
|
}
|
|
|
|
return (val % ZIP_TABSIZE);
|
|
}
|
|
|
|
/*
|
|
* x t o i n t
|
|
*
|
|
* Converts a two byte ugly endianed integer
|
|
* to our platform's integer.
|
|
*/
|
|
static PRUint16 xtoint (unsigned char *ii)
|
|
{
|
|
return (PRUint16) ((ii [0]) | (ii [1] << 8));
|
|
}
|
|
|
|
/*
|
|
* x t o l o n g
|
|
*
|
|
* Converts a four byte ugly endianed integer
|
|
* to our platform's integer.
|
|
*/
|
|
static PRUint32 xtolong (unsigned char *ll)
|
|
{
|
|
return (PRUint32)( (ll [0] << 0) |
|
|
(ll [1] << 8) |
|
|
(ll [2] << 16) |
|
|
(ll [3] << 24) );
|
|
}
|
|
|
|
/*
|
|
* ExtractMode
|
|
*
|
|
* Extracts bits 17-24 from a 32-bit unsigned long
|
|
* representation of the external attributes field.
|
|
* Subsequently it tacks on the implicit user-read
|
|
* bit.
|
|
*/
|
|
static PRUint16 ExtractMode(unsigned char *ll)
|
|
{
|
|
return ((PRUint16)(ll[2])) | 0x0100;
|
|
}
|
|
|
|
#if defined(XP_UNIX) || defined(XP_BEOS)
|
|
/*
|
|
*
|
|
* Return true if the attributes are for a symbolic link
|
|
*
|
|
*/
|
|
|
|
static PRBool IsSymlink(unsigned char *ll)
|
|
{
|
|
return ((xtoint(ll+2) & S_IFMT) == S_IFLNK);
|
|
}
|
|
#endif
|