1 /*
2  * Copyright (c) 2021 Huawei Device Co., Ltd.
3  * Licensed under the Apache License, Version 2.0 (the "License");
4  * you may not use this file except in compliance with the License.
5  * You may obtain a copy of the License at
6  *
7  * http://www.apache.org/licenses/LICENSE-2.0
8  *
9  * Unless required by applicable law or agreed to in writing, software
10  * distributed under the License is distributed on an "AS IS" BASIS,
11  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12  * See the License for the specific language governing permissions and
13  * limitations under the License.
14  */
15 
16 #ifndef LZ4_PKG_FILE_H
17 #define LZ4_PKG_FILE_H
18 
19 #include "pkg_pkgfile.h"
20 #include "pkg_utils.h"
21 
22 namespace Hpackage {
23 class Lz4FileEntry : public PkgEntry {
24 public:
Lz4FileEntry(PkgFilePtr pkgFile,uint32_t nodeId)25     Lz4FileEntry(PkgFilePtr pkgFile, uint32_t nodeId) : PkgEntry(pkgFile, nodeId)
26     {
27         fileInfo_.compressionLevel = 0;
28         fileInfo_.blockIndependence = 0;
29         fileInfo_.blockSizeID = 0;
30         fileInfo_.contentChecksumFlag = 0;
31     }
32 
~Lz4FileEntry()33     ~Lz4FileEntry() override {}
34 
35     int32_t Init(const PkgManager::FileInfoPtr fileInfo, PkgStreamPtr inStream) override;
36 
GetFileInfo()37     const FileInfo *GetFileInfo() const override
38     {
39         return &fileInfo_.fileInfo;
40     };
41 
42     int32_t EncodeHeader(PkgStreamPtr inStream, size_t startOffset, size_t &encodeLen) override;
43 
44     int32_t Pack(PkgStreamPtr inStream, size_t startOffset, size_t &encodeLen) override;
45 
46     int32_t Unpack(PkgStreamPtr outStream) override;
47 
48     int32_t DecodeHeader(PkgBuffer &buffer, size_t, size_t, size_t &decodeLen) override;
49 
50 private:
51     Lz4FileInfo fileInfo_ {};
52 };
53 
54 class Lz4PkgFile : public PkgFileImpl {
55 public:
56     explicit Lz4PkgFile(PkgManager::PkgManagerPtr manager, PkgStreamPtr stream, PkgInfoPtr header = nullptr)
57         : PkgFileImpl(manager, stream, PkgFile::PKG_TYPE_LZ4)
58     {
59         UNUSED(header);
60         pkgInfo_.signMethod = PKG_SIGN_METHOD_RSA;
61         pkgInfo_.digestMethod = PKG_DIGEST_TYPE_SHA256;
62     }
~Lz4PkgFile()63     ~Lz4PkgFile() override {}
64 
65     int32_t AddEntry(const PkgManager::FileInfoPtr file, const PkgStreamPtr inStream) override;
66 
67     int32_t SavePackage(size_t &offset) override;
68 
69     int32_t LoadPackage(std::vector<std::string> &fileNames, VerifyFunction verifier = nullptr) override;
70 
71 private:
72     PkgInfo pkgInfo_ {};
73     size_t currentOffset_ {0};
74 };
75 } // namespace Hpackage
76 #endif // LZ4_PKG_FILE_H