gecko-dev/modules/libjar/nsJARInputStream.h

91 строка
2.8 KiB
C++

/* nsJARInputStream.h
*
* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
#ifndef nsJARINPUTSTREAM_h__
#define nsJARINPUTSTREAM_h__
#include "nsIInputStream.h"
#include "nsJAR.h"
#include "nsTArray.h"
#include "mozilla/Attributes.h"
#include "mozilla/Omnijar.h"
#ifdef MOZ_JAR_BROTLI
struct BrotliDecoderStateStruct;
#endif
/*-------------------------------------------------------------------------
* Class nsJARInputStream declaration. This class defines the type of the
* object returned by calls to nsJAR::GetInputStream(filename) for the
* purpose of reading a file item out of a JAR file.
*------------------------------------------------------------------------*/
class nsJARInputStream final : public nsIInputStream {
public:
nsJARInputStream()
: mOutSize(0),
mInCrc(0),
mOutCrc(0)
#ifdef MOZ_JAR_BROTLI
,
mBrotliState(nullptr)
#endif
,
mNameLen(0),
mCurPos(0),
mArrPos(0),
mMode(MODE_NOTINITED) {
memset(&mZs, 0, sizeof(z_stream));
}
NS_DECL_THREADSAFE_ISUPPORTS
NS_DECL_NSIINPUTSTREAM
// takes ownership of |fd|, even on failure
nsresult InitFile(nsJAR* aJar, nsZipItem* item);
nsresult InitDirectory(nsJAR* aJar, const nsACString& aJarDirSpec,
const char* aDir);
private:
~nsJARInputStream() { Close(); }
mozilla::CacheAwareZipHandle mItemHandle; // handle for reading
uint32_t mOutSize; // inflated size
uint32_t mInCrc; // CRC as provided by the zipentry
uint32_t mOutCrc; // CRC as calculated by me
z_stream mZs; // zip data structure
#ifdef MOZ_JAR_BROTLI
BrotliDecoderStateStruct* mBrotliState; // Brotli decoder state
#endif
/* For directory reading */
RefPtr<nsJAR> mJar; // string reference to zipreader
uint32_t mNameLen; // length of dirname
nsCString mBuffer; // storage for generated text of stream
uint32_t mCurPos; // Current position in buffer
uint32_t mArrPos; // current position within mArray
nsTArray<nsCString> mArray; // array of names in (zip) directory
typedef enum {
MODE_NOTINITED,
MODE_CLOSED,
MODE_DIRECTORY,
MODE_INFLATE,
#ifdef MOZ_JAR_BROTLI
MODE_BROTLI,
#endif
MODE_COPY
} JISMode;
JISMode mMode; // Modus of the stream
nsresult ContinueInflate(char* aBuf, uint32_t aCount, uint32_t* aBytesRead);
nsresult ReadDirectory(char* aBuf, uint32_t aCount, uint32_t* aBytesRead);
uint32_t CopyDataToBuffer(char*& aBuffer, uint32_t& aCount);
};
#endif /* nsJARINPUTSTREAM_h__ */