gecko-dev/modules/libjar/nsJARInputStream.cpp

141 строка
3.4 KiB
C++

/* nsJARInputStream.cpp
*
* 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 Netscape Communicator source code.
* The Initial Developer of the Original Code is Netscape Communications
* Corporation. Portions created by Netscape are
* Copyright (C) 1999 Netscape Communications Corporation. All
* Rights Reserved.
*
* Contributor(s):
* Mitch Stoltz <mstoltz@netscape.com>
*/
#include "nsJARInputStream.h"
#include "zipfile.h" // defines ZIP error codes
#include "nsZipArchive.h"
/*---------------------------------------------
* nsISupports implementation
*--------------------------------------------*/
NS_IMPL_THREADSAFE_ISUPPORTS1(nsJARInputStream, nsIInputStream);
/*----------------------------------------------------------
* nsJARInputStream implementation
*--------------------------------------------------------*/
NS_IMETHODIMP
nsJARInputStream::Available(PRUint32 *_retval)
{
if (Zip() == nsnull)
*_retval = 0;
else
*_retval = Zip()->Available(&mReadInfo);
return NS_OK;
}
NS_IMETHODIMP
nsJARInputStream::Read(char* buf, PRUint32 count, PRUint32 *bytesRead)
{
if (Zip() == nsnull)
{
*bytesRead = 0;
return NS_OK;
}
PRInt32 err = Zip()->Read(&mReadInfo, buf, count, bytesRead);
return err == ZIP_OK ? NS_OK : NS_ERROR_FAILURE;
}
NS_IMETHODIMP
nsJARInputStream::ReadSegments(nsWriteSegmentFun writer, void * closure, PRUint32 count, PRUint32 *_retval)
{
char *readBuf = (char *)nsMemory::Alloc(count);
PRUint32 nBytes;
if (!readBuf)
return NS_ERROR_OUT_OF_MEMORY;
nsresult rv = Read(readBuf, count, &nBytes);
*_retval = 0;
if (NS_SUCCEEDED(rv)) {
rv = writer(this, closure, readBuf, 0, nBytes, _retval);
// XXX _retval may be less than nBytes!! This is the wrong
// way to synthesize ReadSegments.
NS_ASSERTION(*_retval == nBytes, "data loss");
}
nsMemory::Free(readBuf);
return rv;
}
NS_IMETHODIMP
nsJARInputStream::IsNonBlocking(PRBool *aNonBlocking)
{
*aNonBlocking = PR_FALSE;
return NS_OK;
}
NS_IMETHODIMP
nsJARInputStream::Close()
{
NS_IF_RELEASE(mJAR);
return NS_OK;
}
nsresult
nsJARInputStream::Init(nsJAR* aJAR, const char* aFilename)
{
if (!aFilename)
return NS_ERROR_NULL_POINTER;
mJAR = aJAR;
NS_ADDREF(mJAR);
PRInt32 result;
result = Zip()->ReadInit(aFilename, &mReadInfo);
if (result != ZIP_OK)
return NS_ERROR_FAILURE;
return NS_OK;
}
NS_METHOD
nsJARInputStream::Create(nsISupports* ignored, const nsIID& aIID, void* *aResult)
{
nsJARInputStream* is = new nsJARInputStream();
if (is == nsnull)
return NS_ERROR_OUT_OF_MEMORY;
NS_ADDREF(is);
nsresult rv = is->QueryInterface(aIID, aResult);
NS_RELEASE(is);
return rv;
}
//----------------------------------------------
// nsJARInputStream constructor and destructor
//----------------------------------------------
nsJARInputStream::nsJARInputStream()
: mJAR(nsnull)
{
NS_INIT_REFCNT();
}
nsJARInputStream::~nsJARInputStream()
{
Close();
}