зеркало из https://github.com/mozilla/pjs.git
1316 строки
37 KiB
C++
1316 строки
37 KiB
C++
/* -*- Mode: C++; tab-width: 2; 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@netsape.com>
|
|
* Pierre Phaneuf <pp@ludusdesign.com>
|
|
* Jeff Walden <jwalden+code@mit.edu>
|
|
*
|
|
* 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 ***** */
|
|
#include <string.h>
|
|
#include "nsJARInputStream.h"
|
|
#include "nsJAR.h"
|
|
#include "nsILocalFile.h"
|
|
#include "nsIConsoleService.h"
|
|
#include "nsICryptoHash.h"
|
|
#include "prprf.h"
|
|
|
|
#ifdef XP_UNIX
|
|
#include <sys/stat.h>
|
|
#elif defined (XP_WIN) || defined(XP_OS2)
|
|
#include <io.h>
|
|
#endif
|
|
|
|
//----------------------------------------------
|
|
// nsJARManifestItem declaration
|
|
//----------------------------------------------
|
|
/*
|
|
* nsJARManifestItem contains meta-information pertaining
|
|
* to an individual JAR entry, taken from the
|
|
* META-INF/MANIFEST.MF and META-INF/ *.SF files.
|
|
* This is security-critical information, defined here so it is not
|
|
* accessible from anywhere else.
|
|
*/
|
|
typedef enum
|
|
{
|
|
JAR_INVALID = 1,
|
|
JAR_INTERNAL = 2,
|
|
JAR_EXTERNAL = 3
|
|
} JARManifestItemType;
|
|
|
|
class nsJARManifestItem
|
|
{
|
|
public:
|
|
JARManifestItemType mType;
|
|
|
|
// True if the second step of verification (VerifyEntry)
|
|
// has taken place:
|
|
PRBool entryVerified;
|
|
|
|
// Not signed, valid, or failure code
|
|
PRInt16 status;
|
|
|
|
// Internal storage of digests
|
|
char* calculatedSectionDigest;
|
|
char* storedEntryDigest;
|
|
|
|
nsJARManifestItem();
|
|
virtual ~nsJARManifestItem();
|
|
};
|
|
|
|
//-------------------------------------------------
|
|
// nsJARManifestItem constructors and destructor
|
|
//-------------------------------------------------
|
|
nsJARManifestItem::nsJARManifestItem(): mType(JAR_INTERNAL),
|
|
entryVerified(PR_FALSE),
|
|
status(JAR_NOT_SIGNED),
|
|
calculatedSectionDigest(nsnull),
|
|
storedEntryDigest(nsnull)
|
|
{
|
|
}
|
|
|
|
nsJARManifestItem::~nsJARManifestItem()
|
|
{
|
|
// Delete digests if necessary
|
|
PR_FREEIF(calculatedSectionDigest);
|
|
PR_FREEIF(storedEntryDigest);
|
|
}
|
|
|
|
//----------------------------------------------
|
|
// nsJAR constructor/destructor
|
|
//----------------------------------------------
|
|
PR_STATIC_CALLBACK(PRBool)
|
|
DeleteManifestEntry(nsHashKey* aKey, void* aData, void* closure)
|
|
{
|
|
//-- deletes an entry in mManifestData.
|
|
delete (nsJARManifestItem*)aData;
|
|
return PR_TRUE;
|
|
}
|
|
|
|
// The following initialization makes a guess of 10 entries per jarfile.
|
|
nsJAR::nsJAR(): mManifestData(nsnull, nsnull, DeleteManifestEntry, nsnull, 10),
|
|
mParsedManifest(PR_FALSE),
|
|
mGlobalStatus(JAR_MANIFEST_NOT_PARSED),
|
|
mReleaseTime(PR_INTERVAL_NO_TIMEOUT),
|
|
mCache(nsnull),
|
|
mLock(nsnull),
|
|
mTotalItemsInManifest(0)
|
|
{
|
|
}
|
|
|
|
nsJAR::~nsJAR()
|
|
{
|
|
Close();
|
|
}
|
|
|
|
NS_IMPL_THREADSAFE_QUERY_INTERFACE2(nsJAR, nsIZipReader, nsIJAR)
|
|
NS_IMPL_THREADSAFE_ADDREF(nsJAR)
|
|
|
|
// Custom Release method works with nsZipReaderCache...
|
|
nsrefcnt nsJAR::Release(void)
|
|
{
|
|
nsrefcnt count;
|
|
NS_PRECONDITION(0 != mRefCnt, "dup release");
|
|
count = PR_AtomicDecrement((PRInt32 *)&mRefCnt);
|
|
NS_LOG_RELEASE(this, count, "nsJAR");
|
|
if (0 == count) {
|
|
mRefCnt = 1; /* stabilize */
|
|
/* enable this to find non-threadsafe destructors: */
|
|
/* NS_ASSERT_OWNINGTHREAD(nsJAR); */
|
|
NS_DELETEXPCOM(this);
|
|
return 0;
|
|
}
|
|
else if (1 == count && mCache) {
|
|
nsresult rv = mCache->ReleaseZip(this);
|
|
NS_ASSERTION(NS_SUCCEEDED(rv), "failed to release zip file");
|
|
}
|
|
return count;
|
|
}
|
|
|
|
//----------------------------------------------
|
|
// nsIZipReader implementation
|
|
//----------------------------------------------
|
|
|
|
NS_IMETHODIMP
|
|
nsJAR::Open(nsIFile* zipFile)
|
|
{
|
|
if (mLock) return NS_ERROR_FAILURE; // Already open!
|
|
|
|
mZipFile = zipFile;
|
|
mLock = PR_NewLock();
|
|
NS_ENSURE_TRUE(mLock, NS_ERROR_OUT_OF_MEMORY);
|
|
|
|
PRFileDesc *fd = OpenFile();
|
|
NS_ENSURE_TRUE(fd, NS_ERROR_FAILURE);
|
|
|
|
nsresult rv = mZip.OpenArchive(fd);
|
|
if (NS_FAILED(rv)) Close();
|
|
|
|
return rv;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
nsJAR::GetFile(nsIFile* *result)
|
|
{
|
|
*result = mZipFile;
|
|
NS_IF_ADDREF(*result);
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
nsJAR::Close()
|
|
{
|
|
if (mLock) {
|
|
PR_DestroyLock(mLock);
|
|
mLock = nsnull;
|
|
}
|
|
|
|
mParsedManifest = PR_FALSE;
|
|
mGlobalStatus = JAR_MANIFEST_NOT_PARSED;
|
|
mTotalItemsInManifest = 0;
|
|
|
|
return mZip.CloseArchive();
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
nsJAR::Test(const char *aEntryName)
|
|
{
|
|
return mZip.Test(aEntryName);
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
nsJAR::Extract(const char *zipEntry, nsIFile* outFile)
|
|
{
|
|
// nsZipArchive and zlib are not thread safe
|
|
// we need to use a lock to prevent bug #51267
|
|
nsAutoLock lock(mLock);
|
|
|
|
nsresult rv;
|
|
nsCOMPtr<nsILocalFile> localFile = do_QueryInterface(outFile, &rv);
|
|
if (NS_FAILED(rv)) return rv;
|
|
|
|
nsZipItem *item = mZip.GetItem(zipEntry);
|
|
NS_ENSURE_TRUE(item, NS_ERROR_FILE_TARGET_DOES_NOT_EXIST);
|
|
|
|
// Remove existing file or directory so we set permissions correctly.
|
|
// If it's a directory that already exists and contains files, throw
|
|
// an exception and return.
|
|
|
|
//XXX Bug 332139:
|
|
//XXX If we guarantee that rv in the case of a non-empty directory
|
|
//XXX is always FILE_DIR_NOT_EMPTY, we can remove
|
|
//XXX |rv == NS_ERROR_FAILURE| - bug 322183 needs to be completely
|
|
//XXX fixed before that can happen
|
|
rv = localFile->Remove(PR_FALSE);
|
|
if (rv == NS_ERROR_FILE_DIR_NOT_EMPTY ||
|
|
rv == NS_ERROR_FAILURE)
|
|
return rv;
|
|
|
|
if (item->isDirectory)
|
|
{
|
|
rv = localFile->Create(nsIFile::DIRECTORY_TYPE, item->mode);
|
|
//XXX Do this in nsZipArchive? It would be nice to keep extraction
|
|
//XXX code completely there, but that would require a way to get a
|
|
//XXX PRDir from localFile.
|
|
}
|
|
else
|
|
{
|
|
PRFileDesc* fd;
|
|
rv = localFile->OpenNSPRFileDesc(PR_WRONLY | PR_CREATE_FILE, item->mode, &fd);
|
|
if (NS_FAILED(rv)) return rv;
|
|
|
|
// ExtractFile also closes the fd handle and resolves the symlink if needed
|
|
nsCAutoString path;
|
|
rv = outFile->GetNativePath(path);
|
|
if (NS_FAILED(rv)) return rv;
|
|
|
|
rv = mZip.ExtractFile(item, path.get(), fd);
|
|
}
|
|
if (NS_FAILED(rv)) return rv;
|
|
|
|
PRTime prtime = GetModTime(item->date, item->time);
|
|
// nsIFile needs milliseconds, while prtime is in microseconds.
|
|
PRTime conversion = LL_ZERO;
|
|
PRTime newTime = LL_ZERO;
|
|
LL_I2L(conversion, PR_USEC_PER_MSEC);
|
|
LL_DIV(newTime, prtime, conversion);
|
|
// non-fatal if this fails, ignore errors
|
|
outFile->SetLastModifiedTime(newTime);
|
|
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
nsJAR::GetEntry(const char *aEntryName, nsIZipEntry* *result)
|
|
{
|
|
nsZipItem* zipItem = mZip.GetItem(aEntryName);
|
|
NS_ENSURE_TRUE(zipItem, NS_ERROR_FILE_TARGET_DOES_NOT_EXIST);
|
|
|
|
nsJARItem* jarItem = new nsJARItem(zipItem);
|
|
NS_ENSURE_TRUE(jarItem, NS_ERROR_OUT_OF_MEMORY);
|
|
|
|
NS_ADDREF(*result = jarItem);
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
nsJAR::HasEntry(const nsACString &aEntryName, PRBool *result)
|
|
{
|
|
*result = mZip.GetItem(PromiseFlatCString(aEntryName).get()) != nsnull;
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
nsJAR::FindEntries(const char *aPattern, nsIUTF8StringEnumerator **result)
|
|
{
|
|
NS_ENSURE_ARG_POINTER(result);
|
|
|
|
nsZipFind *find;
|
|
nsresult rv = mZip.FindInit(aPattern, &find);
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
nsIUTF8StringEnumerator *zipEnum = new nsJAREnumerator(find);
|
|
if (!zipEnum) {
|
|
delete find;
|
|
return NS_ERROR_OUT_OF_MEMORY;
|
|
}
|
|
|
|
NS_ADDREF(*result = zipEnum);
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
nsJAR::GetInputStream(const char* aFilename, nsIInputStream** result)
|
|
{
|
|
return GetInputStreamWithSpec(EmptyCString(), aFilename, result);
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
nsJAR::GetInputStreamWithSpec(const nsACString& aJarDirSpec,
|
|
const char* aEntryName, nsIInputStream** result)
|
|
{
|
|
NS_ENSURE_ARG_POINTER(aEntryName);
|
|
NS_ENSURE_ARG_POINTER(result);
|
|
|
|
// Watch out for the jar:foo.zip!/ (aDir is empty) top-level special case!
|
|
nsZipItem *item = nsnull;
|
|
if (*aEntryName) {
|
|
// First check if item exists in jar
|
|
item = mZip.GetItem(aEntryName);
|
|
if (!item) return NS_ERROR_FILE_TARGET_DOES_NOT_EXIST;
|
|
}
|
|
nsJARInputStream* jis = new nsJARInputStream();
|
|
// addref now so we can call InitFile/InitDirectory()
|
|
NS_ENSURE_TRUE(jis, NS_ERROR_OUT_OF_MEMORY);
|
|
NS_ADDREF(*result = jis);
|
|
|
|
nsresult rv = NS_OK;
|
|
if (!item || item->isDirectory) {
|
|
rv = jis->InitDirectory(&mZip, aJarDirSpec, aEntryName);
|
|
} else {
|
|
// Open jarfile, to get its own filedescriptor for the stream
|
|
// XXX The file may have been overwritten, so |item| might not be
|
|
// valid. We really want to work from inode rather than file name.
|
|
PRFileDesc *fd = nsnull;
|
|
fd = OpenFile();
|
|
if (fd) {
|
|
rv = jis->InitFile(&mZip, item, fd);
|
|
// |jis| now owns |fd|
|
|
} else {
|
|
rv = NS_ERROR_FAILURE;
|
|
}
|
|
}
|
|
if (NS_FAILED(rv)) {
|
|
NS_RELEASE(*result);
|
|
}
|
|
return rv;
|
|
}
|
|
|
|
//----------------------------------------------
|
|
// nsIJAR implementation
|
|
//----------------------------------------------
|
|
|
|
NS_IMETHODIMP
|
|
nsJAR::GetCertificatePrincipal(const char* aFilename, nsIPrincipal** aPrincipal)
|
|
{
|
|
//-- Parameter check
|
|
if (!aPrincipal)
|
|
return NS_ERROR_NULL_POINTER;
|
|
*aPrincipal = nsnull;
|
|
|
|
//-- Get the signature verifier service
|
|
nsresult rv;
|
|
nsCOMPtr<nsISignatureVerifier> verifier =
|
|
do_GetService(SIGNATURE_VERIFIER_CONTRACTID, &rv);
|
|
if (NS_FAILED(rv)) // No signature verifier available
|
|
return NS_OK;
|
|
|
|
//-- Parse the manifest
|
|
rv = ParseManifest(verifier);
|
|
if (NS_FAILED(rv)) return rv;
|
|
if (mGlobalStatus == JAR_NO_MANIFEST)
|
|
return NS_OK;
|
|
|
|
PRInt16 requestedStatus;
|
|
if (aFilename)
|
|
{
|
|
//-- Find the item
|
|
nsCStringKey key(aFilename);
|
|
nsJARManifestItem* manItem = static_cast<nsJARManifestItem*>(mManifestData.Get(&key));
|
|
if (!manItem)
|
|
return NS_OK;
|
|
//-- Verify the item against the manifest
|
|
if (!manItem->entryVerified)
|
|
{
|
|
nsXPIDLCString entryData;
|
|
PRUint32 entryDataLen;
|
|
rv = LoadEntry(aFilename, getter_Copies(entryData), &entryDataLen);
|
|
if (NS_FAILED(rv)) return rv;
|
|
rv = VerifyEntry(manItem, entryData, entryDataLen);
|
|
if (NS_FAILED(rv)) return rv;
|
|
}
|
|
requestedStatus = manItem->status;
|
|
}
|
|
else // User wants identity of signer w/o verifying any entries
|
|
requestedStatus = mGlobalStatus;
|
|
|
|
if (requestedStatus != JAR_VALID_MANIFEST)
|
|
ReportError(aFilename, requestedStatus);
|
|
else // Valid signature
|
|
{
|
|
*aPrincipal = mPrincipal;
|
|
NS_IF_ADDREF(*aPrincipal);
|
|
}
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
nsJAR::GetManifestEntriesCount(PRUint32* count)
|
|
{
|
|
*count = mTotalItemsInManifest;
|
|
return NS_OK;
|
|
}
|
|
|
|
nsresult
|
|
nsJAR::GetJarPath(nsACString& aResult)
|
|
{
|
|
NS_ENSURE_ARG_POINTER(mZipFile);
|
|
|
|
return mZipFile->GetNativePath(aResult);
|
|
}
|
|
|
|
PRFileDesc*
|
|
nsJAR::OpenFile()
|
|
{
|
|
nsresult rv;
|
|
nsCOMPtr<nsILocalFile> localFile = do_QueryInterface(mZipFile, &rv);
|
|
if (NS_FAILED(rv)) return nsnull;
|
|
|
|
PRFileDesc* fd;
|
|
rv = localFile->OpenNSPRFileDesc(PR_RDONLY, 0000, &fd);
|
|
if (NS_FAILED(rv)) return nsnull;
|
|
|
|
return fd;
|
|
}
|
|
|
|
//----------------------------------------------
|
|
// nsJAR private implementation
|
|
//----------------------------------------------
|
|
nsresult
|
|
nsJAR::LoadEntry(const char* aFilename, char** aBuf, PRUint32* aBufLen)
|
|
{
|
|
//-- Get a stream for reading the file
|
|
nsresult rv;
|
|
nsCOMPtr<nsIInputStream> manifestStream;
|
|
rv = GetInputStream(aFilename, getter_AddRefs(manifestStream));
|
|
if (NS_FAILED(rv)) return NS_ERROR_FILE_TARGET_DOES_NOT_EXIST;
|
|
|
|
//-- Read the manifest file into memory
|
|
char* buf;
|
|
PRUint32 len;
|
|
rv = manifestStream->Available(&len);
|
|
if (NS_FAILED(rv)) return rv;
|
|
if (len == PRUint32(-1))
|
|
return NS_ERROR_FILE_CORRUPTED; // bug 164695
|
|
buf = (char*)PR_MALLOC(len+1);
|
|
if (!buf) return NS_ERROR_OUT_OF_MEMORY;
|
|
PRUint32 bytesRead;
|
|
rv = manifestStream->Read(buf, len, &bytesRead);
|
|
if (bytesRead != len)
|
|
rv = NS_ERROR_FILE_CORRUPTED;
|
|
if (NS_FAILED(rv)) {
|
|
PR_FREEIF(buf);
|
|
return rv;
|
|
}
|
|
buf[len] = '\0'; //Null-terminate the buffer
|
|
*aBuf = buf;
|
|
if (aBufLen)
|
|
*aBufLen = len;
|
|
return NS_OK;
|
|
}
|
|
|
|
|
|
PRInt32
|
|
nsJAR::ReadLine(const char** src)
|
|
{
|
|
//--Moves pointer to beginning of next line and returns line length
|
|
// not including CR/LF.
|
|
PRInt32 length;
|
|
char* eol = PL_strpbrk(*src, "\r\n");
|
|
|
|
if (eol == nsnull) // Probably reached end of file before newline
|
|
{
|
|
length = PL_strlen(*src);
|
|
if (length == 0) // immediate end-of-file
|
|
*src = nsnull;
|
|
else // some data left on this line
|
|
*src += length;
|
|
}
|
|
else
|
|
{
|
|
length = eol - *src;
|
|
if (eol[0] == '\r' && eol[1] == '\n') // CR LF, so skip 2
|
|
*src = eol+2;
|
|
else // Either CR or LF, so skip 1
|
|
*src = eol+1;
|
|
}
|
|
return length;
|
|
}
|
|
|
|
//-- The following #defines are used by ParseManifest()
|
|
// and ParseOneFile(). The header strings are defined in the JAR specification.
|
|
#define JAR_MF 1
|
|
#define JAR_SF 2
|
|
#define JAR_MF_SEARCH_STRING "(M|/M)ETA-INF/(M|m)(ANIFEST|anifest).(MF|mf)$"
|
|
#define JAR_SF_SEARCH_STRING "(M|/M)ETA-INF/*.(SF|sf)$"
|
|
#define JAR_MF_HEADER (const char*)"Manifest-Version: 1.0"
|
|
#define JAR_SF_HEADER (const char*)"Signature-Version: 1.0"
|
|
|
|
nsresult
|
|
nsJAR::ParseManifest(nsISignatureVerifier* verifier)
|
|
{
|
|
//-- Verification Step 1
|
|
if (mParsedManifest)
|
|
return NS_OK;
|
|
//-- (1)Manifest (MF) file
|
|
nsCOMPtr<nsIUTF8StringEnumerator> files;
|
|
nsresult rv = FindEntries(JAR_MF_SEARCH_STRING, getter_AddRefs(files));
|
|
if (!files) rv = NS_ERROR_FAILURE;
|
|
if (NS_FAILED(rv)) return rv;
|
|
|
|
//-- Load the file into memory
|
|
PRBool more;
|
|
rv = files->HasMore(&more);
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
if (!more)
|
|
{
|
|
mGlobalStatus = JAR_NO_MANIFEST;
|
|
mParsedManifest = PR_TRUE;
|
|
return NS_OK;
|
|
}
|
|
|
|
nsCAutoString manifestFilename;
|
|
rv = files->GetNext(manifestFilename);
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
// Check if there is more than one manifest, if so then error!
|
|
rv = files->HasMore(&more);
|
|
if (NS_FAILED(rv)) return rv;
|
|
if (more)
|
|
{
|
|
mParsedManifest = PR_TRUE;
|
|
return NS_ERROR_FILE_CORRUPTED; // More than one MF file
|
|
}
|
|
|
|
nsXPIDLCString manifestBuffer;
|
|
PRUint32 manifestLen;
|
|
rv = LoadEntry(manifestFilename.get(), getter_Copies(manifestBuffer), &manifestLen);
|
|
if (NS_FAILED(rv)) return rv;
|
|
|
|
//-- Parse it
|
|
rv = ParseOneFile(manifestBuffer, JAR_MF);
|
|
if (NS_FAILED(rv)) return rv;
|
|
|
|
//-- (2)Signature (SF) file
|
|
// If there are multiple signatures, we select one.
|
|
rv = FindEntries(JAR_SF_SEARCH_STRING, getter_AddRefs(files));
|
|
if (!files) rv = NS_ERROR_FAILURE;
|
|
if (NS_FAILED(rv)) return rv;
|
|
//-- Get an SF file
|
|
rv = files->HasMore(&more);
|
|
if (NS_FAILED(rv)) return rv;
|
|
if (!more)
|
|
{
|
|
mGlobalStatus = JAR_NO_MANIFEST;
|
|
mParsedManifest = PR_TRUE;
|
|
return NS_OK;
|
|
}
|
|
rv = files->GetNext(manifestFilename);
|
|
if (NS_FAILED(rv)) return rv;
|
|
|
|
rv = LoadEntry(manifestFilename.get(), getter_Copies(manifestBuffer), &manifestLen);
|
|
if (NS_FAILED(rv)) return rv;
|
|
|
|
//-- Get its corresponding signature file
|
|
nsCAutoString sigFilename(manifestFilename);
|
|
PRInt32 extension = sigFilename.RFindChar('.') + 1;
|
|
NS_ASSERTION(extension != 0, "Manifest Parser: Missing file extension.");
|
|
(void)sigFilename.Cut(extension, 2);
|
|
nsXPIDLCString sigBuffer;
|
|
PRUint32 sigLen;
|
|
{
|
|
nsCAutoString tempFilename(sigFilename); tempFilename.Append("rsa", 3);
|
|
rv = LoadEntry(tempFilename.get(), getter_Copies(sigBuffer), &sigLen);
|
|
}
|
|
if (NS_FAILED(rv))
|
|
{
|
|
nsCAutoString tempFilename(sigFilename); tempFilename.Append("RSA", 3);
|
|
rv = LoadEntry(tempFilename.get(), getter_Copies(sigBuffer), &sigLen);
|
|
}
|
|
if (NS_FAILED(rv))
|
|
{
|
|
mGlobalStatus = JAR_NO_MANIFEST;
|
|
mParsedManifest = PR_TRUE;
|
|
return NS_OK;
|
|
}
|
|
|
|
//-- Verify that the signature file is a valid signature of the SF file
|
|
PRInt32 verifyError;
|
|
rv = verifier->VerifySignature(sigBuffer, sigLen, manifestBuffer, manifestLen,
|
|
&verifyError, getter_AddRefs(mPrincipal));
|
|
if (NS_FAILED(rv)) return rv;
|
|
if (mPrincipal && verifyError == 0)
|
|
mGlobalStatus = JAR_VALID_MANIFEST;
|
|
else if (verifyError == nsISignatureVerifier::VERIFY_ERROR_UNKNOWN_CA)
|
|
mGlobalStatus = JAR_INVALID_UNKNOWN_CA;
|
|
else
|
|
mGlobalStatus = JAR_INVALID_SIG;
|
|
|
|
//-- Parse the SF file. If the verification above failed, principal
|
|
// is null, and ParseOneFile will mark the relevant entries as invalid.
|
|
// if ParseOneFile fails, then it has no effect, and we can safely
|
|
// continue to the next SF file, or return.
|
|
ParseOneFile(manifestBuffer, JAR_SF);
|
|
mParsedManifest = PR_TRUE;
|
|
|
|
return NS_OK;
|
|
}
|
|
|
|
nsresult
|
|
nsJAR::ParseOneFile(const char* filebuf, PRInt16 aFileType)
|
|
{
|
|
//-- Check file header
|
|
const char* nextLineStart = filebuf;
|
|
nsCAutoString curLine;
|
|
PRInt32 linelen;
|
|
linelen = ReadLine(&nextLineStart);
|
|
curLine.Assign(filebuf, linelen);
|
|
|
|
if ( ((aFileType == JAR_MF) && !curLine.Equals(JAR_MF_HEADER) ) ||
|
|
((aFileType == JAR_SF) && !curLine.Equals(JAR_SF_HEADER) ) )
|
|
return NS_ERROR_FILE_CORRUPTED;
|
|
|
|
//-- Skip header section
|
|
do {
|
|
linelen = ReadLine(&nextLineStart);
|
|
} while (linelen > 0);
|
|
|
|
//-- Set up parsing variables
|
|
const char* curPos;
|
|
const char* sectionStart = nextLineStart;
|
|
|
|
nsJARManifestItem* curItemMF = nsnull;
|
|
PRBool foundName = PR_FALSE;
|
|
if (aFileType == JAR_MF)
|
|
if (!(curItemMF = new nsJARManifestItem()))
|
|
return NS_ERROR_OUT_OF_MEMORY;
|
|
|
|
nsCAutoString curItemName;
|
|
nsCAutoString storedSectionDigest;
|
|
|
|
for(;;)
|
|
{
|
|
curPos = nextLineStart;
|
|
linelen = ReadLine(&nextLineStart);
|
|
curLine.Assign(curPos, linelen);
|
|
if (linelen == 0)
|
|
// end of section (blank line or end-of-file)
|
|
{
|
|
if (aFileType == JAR_MF)
|
|
{
|
|
mTotalItemsInManifest++;
|
|
if (curItemMF->mType != JAR_INVALID)
|
|
{
|
|
//-- Did this section have a name: line?
|
|
if(!foundName)
|
|
curItemMF->mType = JAR_INVALID;
|
|
else
|
|
{
|
|
//-- If it's an internal item, it must correspond
|
|
// to a valid jar entry
|
|
if (curItemMF->mType == JAR_INTERNAL)
|
|
{
|
|
PRBool exists;
|
|
PRInt32 result = HasEntry(curItemName, &exists);
|
|
if (result != ZIP_OK || !exists)
|
|
curItemMF->mType = JAR_INVALID;
|
|
}
|
|
//-- Check for duplicates
|
|
nsCStringKey key(curItemName);
|
|
if (mManifestData.Exists(&key))
|
|
curItemMF->mType = JAR_INVALID;
|
|
}
|
|
}
|
|
|
|
if (curItemMF->mType == JAR_INVALID)
|
|
delete curItemMF;
|
|
else //-- calculate section digest
|
|
{
|
|
PRUint32 sectionLength = curPos - sectionStart;
|
|
CalculateDigest(sectionStart, sectionLength,
|
|
&(curItemMF->calculatedSectionDigest));
|
|
//-- Save item in the hashtable
|
|
nsCStringKey itemKey(curItemName);
|
|
mManifestData.Put(&itemKey, (void*)curItemMF);
|
|
}
|
|
if (nextLineStart == nsnull) // end-of-file
|
|
break;
|
|
|
|
sectionStart = nextLineStart;
|
|
if (!(curItemMF = new nsJARManifestItem()))
|
|
return NS_ERROR_OUT_OF_MEMORY;
|
|
} // (aFileType == JAR_MF)
|
|
else
|
|
//-- file type is SF, compare digest with calculated
|
|
// section digests from MF file.
|
|
{
|
|
if (foundName)
|
|
{
|
|
nsJARManifestItem* curItemSF;
|
|
nsCStringKey key(curItemName);
|
|
curItemSF = (nsJARManifestItem*)mManifestData.Get(&key);
|
|
if(curItemSF)
|
|
{
|
|
NS_ASSERTION(curItemSF->status == JAR_NOT_SIGNED,
|
|
"SECURITY ERROR: nsJARManifestItem not correctly initialized");
|
|
curItemSF->status = mGlobalStatus;
|
|
if (curItemSF->status == JAR_VALID_MANIFEST)
|
|
{ // Compare digests
|
|
if (storedSectionDigest.IsEmpty())
|
|
curItemSF->status = JAR_NOT_SIGNED;
|
|
else
|
|
{
|
|
if (!storedSectionDigest.Equals((const char*)curItemSF->calculatedSectionDigest))
|
|
curItemSF->status = JAR_INVALID_MANIFEST;
|
|
PR_FREEIF(curItemSF->calculatedSectionDigest)
|
|
storedSectionDigest = "";
|
|
}
|
|
} // (aPrincipal != nsnull)
|
|
} // if(curItemSF)
|
|
} // if(foundName)
|
|
|
|
if(nextLineStart == nsnull) // end-of-file
|
|
break;
|
|
} // aFileType == JAR_SF
|
|
foundName = PR_FALSE;
|
|
continue;
|
|
} // if(linelen == 0)
|
|
|
|
//-- Look for continuations (beginning with a space) on subsequent lines
|
|
// and append them to the current line.
|
|
while(*nextLineStart == ' ')
|
|
{
|
|
curPos = nextLineStart;
|
|
PRInt32 continuationLen = ReadLine(&nextLineStart) - 1;
|
|
nsCAutoString continuation(curPos+1, continuationLen);
|
|
curLine += continuation;
|
|
linelen += continuationLen;
|
|
}
|
|
|
|
//-- Find colon in current line, this separates name from value
|
|
PRInt32 colonPos = curLine.FindChar(':');
|
|
if (colonPos == -1) // No colon on line, ignore line
|
|
continue;
|
|
//-- Break down the line
|
|
nsCAutoString lineName;
|
|
curLine.Left(lineName, colonPos);
|
|
nsCAutoString lineData;
|
|
curLine.Mid(lineData, colonPos+2, linelen - (colonPos+2));
|
|
|
|
//-- Lines to look for:
|
|
// (1) Digest:
|
|
if (lineName.Equals(NS_LITERAL_CSTRING("SHA1-Digest"),
|
|
nsCaseInsensitiveCStringComparator()))
|
|
//-- This is a digest line, save the data in the appropriate place
|
|
{
|
|
if(aFileType == JAR_MF)
|
|
{
|
|
curItemMF->storedEntryDigest = (char*)PR_MALLOC(lineData.Length()+1);
|
|
if (!(curItemMF->storedEntryDigest))
|
|
return NS_ERROR_OUT_OF_MEMORY;
|
|
PL_strcpy(curItemMF->storedEntryDigest, lineData.get());
|
|
}
|
|
else
|
|
storedSectionDigest = lineData;
|
|
continue;
|
|
}
|
|
|
|
// (2) Name: associates this manifest section with a file in the jar.
|
|
if (!foundName && lineName.Equals(NS_LITERAL_CSTRING("Name"),
|
|
nsCaseInsensitiveCStringComparator()))
|
|
{
|
|
curItemName = lineData;
|
|
foundName = PR_TRUE;
|
|
continue;
|
|
}
|
|
|
|
// (3) Magic: this may be an inline Javascript.
|
|
// We can't do any other kind of magic.
|
|
if ( aFileType == JAR_MF &&
|
|
lineName.Equals(NS_LITERAL_CSTRING("Magic"),
|
|
nsCaseInsensitiveCStringComparator()))
|
|
{
|
|
if(lineData.Equals(NS_LITERAL_CSTRING("javascript"),
|
|
nsCaseInsensitiveCStringComparator()))
|
|
curItemMF->mType = JAR_EXTERNAL;
|
|
else
|
|
curItemMF->mType = JAR_INVALID;
|
|
continue;
|
|
}
|
|
|
|
} // for (;;)
|
|
return NS_OK;
|
|
} //ParseOneFile()
|
|
|
|
nsresult
|
|
nsJAR::VerifyEntry(nsJARManifestItem* aManItem, const char* aEntryData,
|
|
PRUint32 aLen)
|
|
{
|
|
if (aManItem->status == JAR_VALID_MANIFEST)
|
|
{
|
|
if(!aManItem->storedEntryDigest)
|
|
// No entry digests in manifest file. Entry is unsigned.
|
|
aManItem->status = JAR_NOT_SIGNED;
|
|
else
|
|
{ //-- Calculate and compare digests
|
|
char* calculatedEntryDigest;
|
|
nsresult rv = CalculateDigest(aEntryData, aLen, &calculatedEntryDigest);
|
|
if (NS_FAILED(rv)) return NS_ERROR_FAILURE;
|
|
if (PL_strcmp(aManItem->storedEntryDigest, calculatedEntryDigest) != 0)
|
|
aManItem->status = JAR_INVALID_ENTRY;
|
|
PR_FREEIF(calculatedEntryDigest)
|
|
PR_FREEIF(aManItem->storedEntryDigest)
|
|
}
|
|
}
|
|
aManItem->entryVerified = PR_TRUE;
|
|
return NS_OK;
|
|
}
|
|
|
|
void nsJAR::ReportError(const char* aFilename, PRInt16 errorCode)
|
|
{
|
|
//-- Generate error message
|
|
nsAutoString message;
|
|
message.AssignLiteral("Signature Verification Error: the signature on ");
|
|
if (aFilename)
|
|
message.AppendWithConversion(aFilename);
|
|
else
|
|
message.AppendLiteral("this .jar archive");
|
|
message.AppendLiteral(" is invalid because ");
|
|
switch(errorCode)
|
|
{
|
|
case JAR_NOT_SIGNED:
|
|
message.AppendLiteral("the archive did not contain a valid PKCS7 signature.");
|
|
break;
|
|
case JAR_INVALID_SIG:
|
|
message.AppendLiteral("the digital signature (*.RSA) file is not a valid signature of the signature instruction file (*.SF).");
|
|
break;
|
|
case JAR_INVALID_UNKNOWN_CA:
|
|
message.AppendLiteral("the certificate used to sign this file has an unrecognized issuer.");
|
|
break;
|
|
case JAR_INVALID_MANIFEST:
|
|
message.AppendLiteral("the signature instruction file (*.SF) does not contain a valid hash of the MANIFEST.MF file.");
|
|
break;
|
|
case JAR_INVALID_ENTRY:
|
|
message.AppendLiteral("the MANIFEST.MF file does not contain a valid hash of the file being verified.");
|
|
break;
|
|
case JAR_NO_MANIFEST:
|
|
message.AppendLiteral("the archive did not contain a manifest.");
|
|
break;
|
|
default:
|
|
message.AppendLiteral("of an unknown problem.");
|
|
}
|
|
|
|
// Report error in JS console
|
|
nsCOMPtr<nsIConsoleService> console(do_GetService("@mozilla.org/consoleservice;1"));
|
|
if (console)
|
|
{
|
|
console->LogStringMessage(message.get());
|
|
}
|
|
#ifdef DEBUG
|
|
char* messageCstr = ToNewCString(message);
|
|
if (!messageCstr) return;
|
|
fprintf(stderr, "%s\n", messageCstr);
|
|
nsMemory::Free(messageCstr);
|
|
#endif
|
|
}
|
|
|
|
|
|
nsresult nsJAR::CalculateDigest(const char* aInBuf, PRUint32 aLen,
|
|
char** digest)
|
|
{
|
|
*digest = nsnull;
|
|
nsresult rv;
|
|
|
|
|
|
nsCOMPtr<nsICryptoHash> hasher = do_CreateInstance("@mozilla.org/security/hash;1", &rv);
|
|
if (NS_FAILED(rv)) return rv;
|
|
|
|
rv = hasher->Init(nsICryptoHash::SHA1);
|
|
if (NS_FAILED(rv)) return rv;
|
|
|
|
rv = hasher->Update((const PRUint8*) aInBuf, aLen);
|
|
if (NS_FAILED(rv)) return rv;
|
|
|
|
nsCAutoString hashString;
|
|
rv = hasher->Finish(PR_TRUE, hashString);
|
|
if (NS_FAILED(rv)) return rv;
|
|
|
|
*digest = ToNewCString(hashString);
|
|
|
|
return *digest ? NS_OK : NS_ERROR_OUT_OF_MEMORY;
|
|
}
|
|
|
|
NS_IMPL_THREADSAFE_ISUPPORTS1(nsJAREnumerator, nsIUTF8StringEnumerator)
|
|
|
|
//----------------------------------------------
|
|
// nsJAREnumerator::HasMore
|
|
//----------------------------------------------
|
|
NS_IMETHODIMP
|
|
nsJAREnumerator::HasMore(PRBool* aResult)
|
|
{
|
|
// try to get the next element
|
|
if (!mCurr) {
|
|
NS_ASSERTION(mFind, "nsJAREnumerator: Missing zipFind.");
|
|
nsresult rv = mFind->FindNext( &mCurr );
|
|
if (rv == NS_ERROR_FILE_TARGET_DOES_NOT_EXIST) {
|
|
*aResult = PR_FALSE; // No more matches available
|
|
return NS_OK;
|
|
}
|
|
NS_ENSURE_SUCCESS(rv, NS_ERROR_FAILURE); // no error translation
|
|
}
|
|
|
|
*aResult = PR_TRUE;
|
|
return NS_OK;
|
|
}
|
|
|
|
//----------------------------------------------
|
|
// nsJAREnumerator::GetNext
|
|
//----------------------------------------------
|
|
NS_IMETHODIMP
|
|
nsJAREnumerator::GetNext(nsACString& aResult)
|
|
{
|
|
// check if the current item is "stale"
|
|
if (!mCurr) {
|
|
PRBool bMore;
|
|
nsresult rv = HasMore(&bMore);
|
|
if (NS_FAILED(rv) || !bMore)
|
|
return NS_ERROR_FAILURE; // no error translation
|
|
}
|
|
aResult = mCurr;
|
|
mCurr = 0; // we just gave this one away
|
|
return NS_OK;
|
|
}
|
|
|
|
|
|
NS_IMPL_THREADSAFE_ISUPPORTS1(nsJARItem, nsIZipEntry)
|
|
|
|
nsJARItem::nsJARItem(nsZipItem* aZipItem)
|
|
: mSize(aZipItem->size),
|
|
mRealsize(aZipItem->realsize),
|
|
mCrc32(aZipItem->crc32),
|
|
mDate(aZipItem->date),
|
|
mTime(aZipItem->time),
|
|
mCompression(aZipItem->compression),
|
|
mIsDirectory(aZipItem->isDirectory),
|
|
mIsSynthetic(aZipItem->isSynthetic)
|
|
{
|
|
}
|
|
|
|
//------------------------------------------
|
|
// nsJARItem::GetCompression
|
|
//------------------------------------------
|
|
NS_IMETHODIMP
|
|
nsJARItem::GetCompression(PRUint16 *aCompression)
|
|
{
|
|
NS_ENSURE_ARG_POINTER(aCompression);
|
|
|
|
*aCompression = mCompression;
|
|
return NS_OK;
|
|
}
|
|
|
|
//------------------------------------------
|
|
// nsJARItem::GetSize
|
|
//------------------------------------------
|
|
NS_IMETHODIMP
|
|
nsJARItem::GetSize(PRUint32 *aSize)
|
|
{
|
|
NS_ENSURE_ARG_POINTER(aSize);
|
|
|
|
*aSize = mSize;
|
|
return NS_OK;
|
|
}
|
|
|
|
//------------------------------------------
|
|
// nsJARItem::GetRealSize
|
|
//------------------------------------------
|
|
NS_IMETHODIMP
|
|
nsJARItem::GetRealSize(PRUint32 *aRealsize)
|
|
{
|
|
NS_ENSURE_ARG_POINTER(aRealsize);
|
|
|
|
*aRealsize = mRealsize;
|
|
return NS_OK;
|
|
}
|
|
|
|
//------------------------------------------
|
|
// nsJARItem::GetCrc32
|
|
//------------------------------------------
|
|
NS_IMETHODIMP
|
|
nsJARItem::GetCRC32(PRUint32 *aCrc32)
|
|
{
|
|
NS_ENSURE_ARG_POINTER(aCrc32);
|
|
|
|
*aCrc32 = mCrc32;
|
|
return NS_OK;
|
|
}
|
|
|
|
//------------------------------------------
|
|
// nsJARItem::GetIsDirectory
|
|
//------------------------------------------
|
|
NS_IMETHODIMP
|
|
nsJARItem::GetIsDirectory(PRBool *aIsDirectory)
|
|
{
|
|
NS_ENSURE_ARG_POINTER(aIsDirectory);
|
|
|
|
*aIsDirectory = mIsDirectory;
|
|
return NS_OK;
|
|
}
|
|
|
|
//------------------------------------------
|
|
// nsJARItem::GetIsSynthetic
|
|
//------------------------------------------
|
|
NS_IMETHODIMP
|
|
nsJARItem::GetIsSynthetic(PRBool *aIsSynthetic)
|
|
{
|
|
NS_ENSURE_ARG_POINTER(aIsSynthetic);
|
|
|
|
*aIsSynthetic = mIsSynthetic;
|
|
return NS_OK;
|
|
}
|
|
|
|
//------------------------------------------
|
|
// nsJARItem::GetLastModifiedTime
|
|
//------------------------------------------
|
|
NS_IMETHODIMP
|
|
nsJARItem::GetLastModifiedTime(PRTime* aLastModTime)
|
|
{
|
|
NS_ENSURE_ARG_POINTER(aLastModTime);
|
|
|
|
*aLastModTime = GetModTime(mDate, mTime);
|
|
return NS_OK;
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
// nsIZipReaderCache
|
|
|
|
NS_IMPL_THREADSAFE_ISUPPORTS3(nsZipReaderCache, nsIZipReaderCache, nsIObserver, nsISupportsWeakReference)
|
|
|
|
nsZipReaderCache::nsZipReaderCache()
|
|
: mLock(nsnull),
|
|
mZips(16)
|
|
#ifdef ZIP_CACHE_HIT_RATE
|
|
,
|
|
mZipCacheLookups(0),
|
|
mZipCacheHits(0),
|
|
mZipCacheFlushes(0),
|
|
mZipSyncMisses(0)
|
|
#endif
|
|
{
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
nsZipReaderCache::Init(PRUint32 cacheSize)
|
|
{
|
|
mCacheSize = cacheSize;
|
|
|
|
// Register as a memory pressure observer
|
|
nsCOMPtr<nsIObserverService> os =
|
|
do_GetService("@mozilla.org/observer-service;1");
|
|
if (os)
|
|
{
|
|
os->AddObserver(this, "memory-pressure", PR_TRUE);
|
|
os->AddObserver(this, "chrome-flush-caches", PR_TRUE);
|
|
}
|
|
// ignore failure of the observer registration.
|
|
|
|
mLock = PR_NewLock();
|
|
return mLock ? NS_OK : NS_ERROR_OUT_OF_MEMORY;
|
|
}
|
|
|
|
static PRBool PR_CALLBACK
|
|
DropZipReaderCache(nsHashKey *aKey, void *aData, void* closure)
|
|
{
|
|
nsJAR* zip = (nsJAR*)aData;
|
|
zip->SetZipReaderCache(nsnull);
|
|
return PR_TRUE;
|
|
}
|
|
|
|
nsZipReaderCache::~nsZipReaderCache()
|
|
{
|
|
if (mLock)
|
|
PR_DestroyLock(mLock);
|
|
mZips.Enumerate(DropZipReaderCache, nsnull);
|
|
|
|
#ifdef ZIP_CACHE_HIT_RATE
|
|
printf("nsZipReaderCache size=%d hits=%d lookups=%d rate=%f%% flushes=%d missed %d\n",
|
|
mCacheSize, mZipCacheHits, mZipCacheLookups,
|
|
(float)mZipCacheHits / mZipCacheLookups,
|
|
mZipCacheFlushes, mZipSyncMisses);
|
|
#endif
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
nsZipReaderCache::GetZip(nsIFile* zipFile, nsIZipReader* *result)
|
|
{
|
|
nsresult rv;
|
|
nsAutoLock lock(mLock);
|
|
|
|
#ifdef ZIP_CACHE_HIT_RATE
|
|
mZipCacheLookups++;
|
|
#endif
|
|
|
|
nsCAutoString path;
|
|
rv = zipFile->GetNativePath(path);
|
|
if (NS_FAILED(rv)) return rv;
|
|
|
|
nsCStringKey key(path);
|
|
nsJAR* zip = static_cast<nsJAR*>(static_cast<nsIZipReader*>(mZips.Get(&key))); // AddRefs
|
|
if (zip) {
|
|
#ifdef ZIP_CACHE_HIT_RATE
|
|
mZipCacheHits++;
|
|
#endif
|
|
zip->ClearReleaseTime();
|
|
}
|
|
else {
|
|
zip = new nsJAR();
|
|
if (zip == nsnull)
|
|
return NS_ERROR_OUT_OF_MEMORY;
|
|
NS_ADDREF(zip);
|
|
zip->SetZipReaderCache(this);
|
|
|
|
rv = zip->Open(zipFile);
|
|
if (NS_FAILED(rv)) {
|
|
NS_RELEASE(zip);
|
|
return rv;
|
|
}
|
|
|
|
PRBool collision = mZips.Put(&key, static_cast<nsIZipReader*>(zip)); // AddRefs to 2
|
|
NS_ASSERTION(!collision, "horked");
|
|
}
|
|
*result = zip;
|
|
return rv;
|
|
}
|
|
|
|
static PRBool PR_CALLBACK
|
|
FindOldestZip(nsHashKey *aKey, void *aData, void* closure)
|
|
{
|
|
nsJAR** oldestPtr = (nsJAR**)closure;
|
|
nsJAR* oldest = *oldestPtr;
|
|
nsJAR* current = (nsJAR*)aData;
|
|
PRIntervalTime currentReleaseTime = current->GetReleaseTime();
|
|
if (currentReleaseTime != PR_INTERVAL_NO_TIMEOUT) {
|
|
if (oldest == nsnull ||
|
|
currentReleaseTime < oldest->GetReleaseTime()) {
|
|
*oldestPtr = current;
|
|
}
|
|
}
|
|
return PR_TRUE;
|
|
}
|
|
|
|
struct ZipFindData {nsJAR* zip; PRBool found;};
|
|
|
|
static PRBool PR_CALLBACK
|
|
FindZip(nsHashKey *aKey, void *aData, void* closure)
|
|
{
|
|
ZipFindData* find_data = (ZipFindData*)closure;
|
|
|
|
if (find_data->zip == (nsJAR*)aData) {
|
|
find_data->found = PR_TRUE;
|
|
return PR_FALSE;
|
|
}
|
|
return PR_TRUE;
|
|
}
|
|
|
|
nsresult
|
|
nsZipReaderCache::ReleaseZip(nsJAR* zip)
|
|
{
|
|
nsresult rv;
|
|
nsAutoLock lock(mLock);
|
|
|
|
// It is possible that two thread compete for this zip. The dangerous
|
|
// case is where one thread Releases the zip and discovers that the ref
|
|
// count has gone to one. Before it can call this ReleaseZip method
|
|
// another thread calls our GetZip method. The ref count goes to two. That
|
|
// second thread then Releases the zip and the ref count goes to one. It
|
|
// then tries to enter this ReleaseZip method and blocks while the first
|
|
// thread is still here. The first thread continues and remove the zip from
|
|
// the cache and calls its Release method sending the ref count to 0 and
|
|
// deleting the zip. However, the second thread is still blocked at the
|
|
// start of ReleaseZip, but the 'zip' param now hold a reference to a
|
|
// deleted zip!
|
|
//
|
|
// So, we are going to try safeguarding here by searching our hashtable while
|
|
// locked here for the zip. We return fast if it is not found.
|
|
|
|
ZipFindData find_data = {zip, PR_FALSE};
|
|
mZips.Enumerate(FindZip, &find_data);
|
|
if (!find_data.found) {
|
|
#ifdef ZIP_CACHE_HIT_RATE
|
|
mZipSyncMisses++;
|
|
#endif
|
|
return NS_OK;
|
|
}
|
|
|
|
zip->SetReleaseTime();
|
|
|
|
if (mZips.Count() <= mCacheSize)
|
|
return NS_OK;
|
|
|
|
nsJAR* oldest = nsnull;
|
|
mZips.Enumerate(FindOldestZip, &oldest);
|
|
|
|
// Because of the craziness above it is possible that there is no zip that
|
|
// needs removing.
|
|
if (!oldest)
|
|
return NS_OK;
|
|
|
|
#ifdef ZIP_CACHE_HIT_RATE
|
|
mZipCacheFlushes++;
|
|
#endif
|
|
|
|
// Clear the cache pointer in case we gave out this oldest guy while
|
|
// his Release call was being made. Otherwise we could nest on ReleaseZip
|
|
// when the second owner calls Release and we are still here in this lock.
|
|
oldest->SetZipReaderCache(nsnull);
|
|
|
|
// remove from hashtable
|
|
nsCAutoString path;
|
|
rv = oldest->GetJarPath(path);
|
|
if (NS_FAILED(rv)) return rv;
|
|
|
|
nsCStringKey key(path);
|
|
PRBool removed = mZips.Remove(&key); // Releases
|
|
NS_ASSERTION(removed, "botched");
|
|
|
|
return NS_OK;
|
|
}
|
|
|
|
static PRBool PR_CALLBACK
|
|
FindFlushableZip(nsHashKey *aKey, void *aData, void* closure)
|
|
{
|
|
nsHashKey** flushableKeyPtr = (nsHashKey**)closure;
|
|
nsJAR* current = (nsJAR*)aData;
|
|
|
|
if (current->GetReleaseTime() != PR_INTERVAL_NO_TIMEOUT) {
|
|
*flushableKeyPtr = aKey;
|
|
current->SetZipReaderCache(nsnull);
|
|
return PR_FALSE;
|
|
}
|
|
return PR_TRUE;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
nsZipReaderCache::Observe(nsISupports *aSubject,
|
|
const char *aTopic,
|
|
const PRUnichar *aSomeData)
|
|
{
|
|
if (strcmp(aTopic, "memory-pressure") == 0) {
|
|
nsAutoLock lock(mLock);
|
|
while (PR_TRUE) {
|
|
nsHashKey* flushable = nsnull;
|
|
mZips.Enumerate(FindFlushableZip, &flushable);
|
|
if ( ! flushable )
|
|
break;
|
|
PRBool removed = mZips.Remove(flushable); // Releases
|
|
NS_ASSERTION(removed, "botched");
|
|
|
|
#ifdef xDEBUG_jband
|
|
printf("flushed something from the jar cache\n");
|
|
#endif
|
|
}
|
|
}
|
|
else if (strcmp(aTopic, "chrome-flush-caches") == 0) {
|
|
mZips.Enumerate(DropZipReaderCache, nsnull);
|
|
mZips.Reset();
|
|
}
|
|
return NS_OK;
|
|
}
|
|
|
|
PRTime GetModTime(PRUint16 aDate, PRUint16 aTime)
|
|
{
|
|
PRExplodedTime time;
|
|
|
|
time.tm_usec = 0;
|
|
|
|
time.tm_hour = (aTime >> 11) & 0x1F;
|
|
time.tm_min = (aTime >> 5) & 0x3F;
|
|
time.tm_sec = (aTime & 0x1F) * 2;
|
|
|
|
time.tm_year = (aDate >> 9) + 1980;
|
|
time.tm_month = ((aDate >> 5) & 0x0F)-1;
|
|
time.tm_mday = aDate & 0x1F;
|
|
|
|
time.tm_params.tp_gmt_offset = 0;
|
|
time.tm_params.tp_dst_offset = 0;
|
|
|
|
PR_NormalizeTime(&time, PR_GMTParameters);
|
|
time.tm_params = PR_LocalTimeParameters(&time);
|
|
|
|
return PR_ImplodeTime(&time);
|
|
}
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|