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

#define _nsZipHeader_h_

#include "nsString.h"

#include "nsIOutputStream.h"

#include "nsIInputStream.h"

#include "nsIZipReader.h"

#include "mozilla/Attributes.h"

#include "mozilla/UniquePtr.h"

// High word is S_IFREG, low word is DOS file attribute

#define ZIP_ATTRS_FILE 0x80000000

// High word is S_IFDIR, low word is DOS dir attribute

#define ZIP_ATTRS_DIRECTORY 0x40000010

#define PERMISSIONS_FILE 0644

#define PERMISSIONS_DIR 0755

// Combine file type attributes with unix style permissions

#define ZIP_ATTRS(p, a) ((p & 0xfff) << 16) | a

class nsZipHeader final : public nsIZipEntry

{

    ~nsZipHeader()

    {

        mExtraField = nullptr;

        mLocalExtraField = nullptr;

    }

public:

    NS_DECL_ISUPPORTS

    NS_DECL_NSIZIPENTRY

    nsZipHeader() :

        mCRC(0),

        mCSize(0),

        mUSize(0),

        mEAttr(0),

        mOffset(0),

        mFieldLength(0),

        mLocalFieldLength(0),

        mVersionMade(0x0300 + 23), // Generated on Unix by v2.3 (matches infozip)

        mVersionNeeded(20), // Requires v2.0 to extract

        mFlags(0),

        mMethod(0),

        mTime(0),

        mDate(0),

        mDisk(0),

        mIAttr(0),

        mInited(false),

        mWriteOnClose(false),

        mExtraField(nullptr),

        mLocalExtraField(nullptr)

    {

    }

    uint32_t mCRC;

    uint32_t mCSize;

    uint32_t mUSize;

    uint32_t mEAttr;

    uint32_t mOffset;

    uint32_t mFieldLength;

    uint32_t mLocalFieldLength;

    uint16_t mVersionMade;

    uint16_t mVersionNeeded;

    uint16_t mFlags;

    uint16_t mMethod;

    uint16_t mTime;

    uint16_t mDate;

    uint16_t mDisk;

    uint16_t mIAttr;

    bool mInited;

    bool mWriteOnClose;

    nsCString mName;

    nsCString mComment;

    mozilla::UniquePtr<uint8_t[]> mExtraField;

    mozilla::UniquePtr<uint8_t[]> mLocalExtraField;

    void Init(const nsACString & aPath, PRTime aDate, uint32_t aAttr,

              uint32_t aOffset);

    uint32_t GetFileHeaderLength();

    nsresult WriteFileHeader(nsIOutputStream *aStream);

    uint32_t GetCDSHeaderLength();

    nsresult WriteCDSHeader(nsIOutputStream *aStream);

    nsresult ReadCDSHeader(nsIInputStream *aStream);

    const uint8_t * GetExtraField(uint16_t aTag, bool aLocal, uint16_t *aBlockSize);

    nsresult PadExtraField(uint32_t aOffset, uint16_t aAlignSize);

};

#endif