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#ifndef PKG_MANAGER_IMPL_H
16#define PKG_MANAGER_IMPL_H
17
18#include <functional>
19#include <map>
20#include <memory>
21#include <mutex>
22#include "pkg_lz4file.h"
23#include "pkg_manager.h"
24#include "pkg_pkgfile.h"
25#include "pkg_stream.h"
26#include "pkg_utils.h"
27
28namespace Hpackage {
29class PkgManagerImpl : public PkgManager {
30public:
31    PkgManagerImpl();
32
33    ~PkgManagerImpl() override;
34
35    void RegisterPkgFileCreator(const std::string &fileType, PkgFileConstructor constructor) override;
36
37    int32_t CreatePackage(const std::string &path, const std::string &keyName, PkgInfoPtr header,
38        std::vector<std::pair<std::string, ZipFileInfo>> &files) override;
39
40    int32_t CreatePackage(const std::string &path, const std::string &keyName, PkgInfoPtr header,
41        std::vector<std::pair<std::string, ComponentInfo>> &files) override;
42
43    int32_t CreatePackage(const std::string &path, const std::string &keyName, PkgInfoPtr header,
44        std::vector<std::pair<std::string, Lz4FileInfo>> &files) override;
45
46    int32_t VerifyPackage(const std::string &packagePath, const std::string &keyPath, const std::string &version,
47        const PkgBuffer &digest, VerifyCallback cb) override;
48
49    int32_t LoadPackage(const std::string &packagePath, const std::string &keyPath,
50        std::vector<std::string> &fileIds) override;
51
52    int32_t ExtractFile(const std::string &path, PkgManager::StreamPtr output) override;
53
54    const FileInfo *GetFileInfo(const std::string &path) override;
55
56    const PkgInfo *GetPackageInfo(const std::string &packagePath) override;
57
58    PkgEntryPtr GetPkgEntry(const std::string &path);
59
60    int32_t DecompressBuffer(FileInfoPtr info, const PkgBuffer &buffer, StreamPtr stream) const override;
61
62    int32_t CompressBuffer(FileInfoPtr info, const PkgBuffer &buffer, StreamPtr stream) const override;
63
64    int32_t LoadPackageWithoutUnPack(const std::string &packagePath, std::vector<std::string> &fileIds) override;
65
66    int32_t LoadPackageWithStream(const std::string &packagePath, const std::string &keyPath,
67        std::vector<std::string> &fileIds, uint8_t type, StreamPtr stream) override;
68
69    int32_t ParsePackage(StreamPtr stream, std::vector<std::string> &fileIds, int32_t type) override;
70
71    int32_t CreatePkgStream(StreamPtr &stream, const std::string &fileName, const PkgBuffer &buffer) override;
72
73    int32_t CreatePkgStream(PkgStreamPtr &stream, const std::string &fileName,
74        PkgStream::ExtractFileProcessor processor, const void *context) override;
75
76    int32_t CreatePkgStream(StreamPtr &stream, const std::string &fileName,
77        uint64_t fileLen, Updater::RingBuffer *buffer) override;
78    int32_t VerifyAccPackage(const std::string &packagePath, const std::string &keyPath) override;
79    int32_t VerifyOtaPackage(const std::string &packagePath) override;
80    int32_t VerifyOtaPackage(const std::string &devPath, uint64_t offset, size_t size) override;
81    int32_t VerifyBinFile(const std::string &packagePath, const std::string &keyPath,
82        const std::string &version, const PkgBuffer &digest) override;
83
84    void SetPkgDecodeProgress(PkgDecodeProgress decodeProgress) override
85    {
86        decodeProgress_ = decodeProgress;
87    }
88
89    void PostDecodeProgress(int type, size_t writeDataLen, const void *context) override;
90
91    PkgManager::StreamPtr GetPkgFileStream(const std::string &fileName) override;
92
93    int32_t CreatePkgStream(PkgStreamPtr &stream, const std::string &fileName, size_t size, int32_t type) override;
94
95    void ClosePkgStream(PkgStreamPtr &stream) override;
96
97    int32_t ParseComponents(const std::string &packagePath, std::vector<std::string> &fileName) override;
98
99    int32_t LoadPackage(const std::string &packagePath,
100        std::vector<std::string> &fileIds, PkgFile::PkgType type) override;
101private:
102    PkgFilePtr CreatePackage(PkgStreamPtr stream, PkgFile::PkgType type, PkgInfoPtr header = nullptr);
103
104    template<class T>
105    PkgFilePtr CreatePackage(const std::string &path,
106        PkgInfoPtr header, std::vector<std::pair<std::string, T>> &files, size_t &offset);
107
108    int32_t ExtraAndLoadPackage(const std::string &path, const std::string &name, PkgFile::PkgType,
109        std::vector<std::string> &fileIds);
110
111    int32_t LoadPackageWithStream(const std::string &packagePath,
112        std::vector<std::string> &fileIds, PkgFile::PkgType type, PkgStreamPtr stream);
113
114    std::string GetPkgName(const std::string &path);
115
116    PkgFile::PkgType GetPkgTypeByName(const std::string &path);
117
118    int32_t Sign(PkgStreamPtr stream, size_t offset, const PkgInfoPtr &info);
119
120    int32_t Verify(uint8_t digestMethod, const std::vector<uint8_t> &digest, const std::vector<uint8_t> &signature);
121
122    int32_t DoGenerateFileDigest(PkgStreamPtr stream, uint8_t flags, const size_t fileLen, PkgBuffer &buff,
123        std::pair<DigestAlgorithm::DigestAlgorithmPtr, DigestAlgorithm::DigestAlgorithmPtr> &algorithm);
124
125    int32_t GenerateFileDigest(PkgStreamPtr stream,
126        uint8_t digestMethod, uint8_t flags, std::vector<std::vector<uint8_t>> &digestInfos, size_t hashBufferLen = 0);
127
128    void ClearPkgFile();
129
130    int32_t SetSignVerifyKeyName(const std::string &keyName);
131
132    int32_t DoCreatePkgStream(PkgStreamPtr &stream, const std::string &fileName, int32_t type);
133
134    const std::string GetExtraPath(const std::string &path);
135
136private:
137    bool unzipToFile_ {false};
138    std::vector<PkgFilePtr> pkgFiles_ {};
139    std::map<std::string, PkgFileConstructor> pkgFileCreator_ {};
140    std::mutex mapLock_ {};
141    std::map<std::string, PkgStreamPtr> pkgStreams_ {};
142    std::string signVerifyKeyName_ {};
143    PkgDecodeProgress decodeProgress_ { nullptr };
144};
145} // namespace Hpackage
146#endif // PKG_MANAGER_IMPL_H
147