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 
28 namespace Hpackage {
29 class PkgManagerImpl : public PkgManager {
30 public:
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 
SetPkgDecodeProgress(PkgDecodeProgress decodeProgress)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;
101 private:
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 
136 private:
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