/*

 * Copyright (c) 2021 Huawei Device Co., Ltd.

 * Licensed under the Apache License, Version 2.0 (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.apache.org/licenses/LICENSE-2.0

 *

 * Unless required by applicable law or agreed to in writing, software

 * distributed under the License is distributed on an "AS IS" BASIS,

 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

 * See the License for the specific language governing permissions and

 * limitations under the License.

 */



#ifndef LZ4_ADAPTER_H

#define LZ4_ADAPTER_H

#include <iostream>

#include <vector>

#include "deflate_adapter.h"

#include "diffpatch.h"

#include "lz4.h"

#include "lz4frame.h"

#include "lz4hc.h"

#include "pkg_manager.h"

#include "securec.h"



namespace UpdatePatch {

#define LZ4_BLOCK_SIZE(blockId) (1 << (8 + (2 * (blockId))))



class Lz4Adapter : public DeflateAdapter {

public:

    static constexpr uint32_t LZ4B_BLOCK_SIZE = 1 << 22; // (4M)

    static constexpr uint32_t LZ4B_MAGIC_NUMBER = 0x184C2102;

    static constexpr uint32_t LZ4B_REVERSED_LEN = 4;

    Lz4Adapter(UpdatePatchWriterPtr outStream, size_t offset, const Hpackage::PkgManager::FileInfoPtr fileInfo);

    ~Lz4Adapter() override {}

protected:

    virtual int32_t CompressData(const BlockBuffer &srcData) = 0;

    std::vector<uint8_t> buffer_ {};

    UpdatePatchWriterPtr outStream_ { nullptr };



    size_t offset_;

    int32_t compressionLevel_ { 0 };

    int32_t blockIndependence_ { 0 };

    int32_t contentChecksumFlag_ { 0 };

    uint32_t blockSizeID_ { 0 };

    int32_t autoFlush_ {1};

};



class Lz4FrameAdapter : public Lz4Adapter {

public:

    Lz4FrameAdapter(UpdatePatchWriterPtr outStream, size_t offset,

        const Hpackage::PkgManager::FileInfoPtr fileInfo) : Lz4Adapter(outStream, offset, fileInfo) {}

    ~Lz4FrameAdapter() override

    {

        Close();

    }



    int32_t Open() override;

    int32_t Close() override;

    int32_t WriteData(const BlockBuffer &srcData) override;

    int32_t FlushData(size_t &offset) override;

protected:

    int32_t CompressData(const BlockBuffer &srcData) override;

    size_t currDataSize_ = 0;

    std::vector<uint8_t> inData_ {};

private:

    LZ4F_compressionContext_t compressionContext_ {};

};



class Lz4BlockAdapter : public Lz4FrameAdapter {

public:

    Lz4BlockAdapter(UpdatePatchWriterPtr outStream, size_t offset,

        const Hpackage::PkgManager::FileInfoPtr fileInfo) : Lz4FrameAdapter(outStream, offset, fileInfo) {}

    ~Lz4BlockAdapter() override

    {

        Close();

    }



    int32_t Open() override;

    int32_t Close() override;

    int32_t FlushData(size_t &offset) override;

private:

    int32_t CompressData(const BlockBuffer &srcData) override;

};

} // namespace UpdatePatch

#endif // LZ4_ADAPTER_H