/* 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"

#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(); }

    RefPtr<nsZipHandle>  mFd;         // 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__ */