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 LoadPackageWithStreamForApp(AppPkgInfo &info, 70 std::vector<std::string> &fileIds, StreamPtr stream) override; 71 72 int32_t ParsePackage(StreamPtr stream, std::vector<std::string> &fileIds, int32_t type) override; 73 74 int32_t CreatePkgStream(StreamPtr &stream, const std::string &fileName, const PkgBuffer &buffer) override; 75 76 int32_t CreatePkgStream(PkgStreamPtr &stream, const std::string &fileName, 77 PkgStream::ExtractFileProcessor processor, const void *context) override; 78 79 int32_t CreatePkgStream(StreamPtr &stream, const std::string &fileName, 80 uint64_t fileLen, Updater::RingBuffer *buffer) override; 81 int32_t VerifyAccPackage(const std::string &packagePath, const std::string &keyPath) override; 82 int32_t VerifyOtaPackage(const std::string &packagePath) override; 83 int32_t VerifyOtaPackage(const std::string &packagePath, bool isSupportOldSig) override; 84 int32_t VerifyOtaPackage(const std::string &devPath, uint64_t offset, size_t size) override; 85 int32_t VerifyBinFile(const std::string &packagePath, const std::string &keyPath, 86 const std::string &version, const PkgBuffer &digest) override; 87 SetPkgDecodeProgress(PkgDecodeProgress decodeProgress)88 void SetPkgDecodeProgress(PkgDecodeProgress decodeProgress) override 89 { 90 decodeProgress_ = decodeProgress; 91 } 92 93 void PostDecodeProgress(int type, size_t writeDataLen, const void *context) override; 94 95 PkgManager::StreamPtr GetPkgFileStream(const std::string &fileName) override; 96 97 int32_t CreatePkgStream(PkgStreamPtr &stream, const std::string &fileName, size_t size, int32_t type) override; 98 99 void ClosePkgStream(PkgStreamPtr &stream) override; 100 101 int32_t ParseComponents(const std::string &packagePath, std::vector<std::string> &fileName) override; 102 103 int32_t LoadPackage(const std::string &packagePath, 104 std::vector<std::string> &fileIds, PkgFile::PkgType type) override; 105 106 private: 107 PkgFilePtr CreatePackage(PkgStreamPtr stream, PkgFile::PkgType type, PkgInfoPtr header = nullptr); 108 109 template<class T> 110 PkgFilePtr CreatePackage(const std::string &path, 111 PkgInfoPtr header, std::vector<std::pair<std::string, T>> &files, size_t &offset); 112 113 int32_t ExtraAndLoadPackage(const std::string &path, const std::string &name, PkgFile::PkgType, 114 std::vector<std::string> &fileIds); 115 116 int32_t LoadPackageWithStream(const std::string &packagePath, 117 std::vector<std::string> &fileIds, PkgFile::PkgType type, PkgStreamPtr stream); 118 119 std::string GetPkgName(const std::string &path); 120 121 PkgFile::PkgType GetPkgTypeByName(const std::string &path); 122 123 int32_t Sign(PkgStreamPtr stream, size_t offset, const PkgInfoPtr &info); 124 125 int32_t Verify(uint8_t digestMethod, const std::vector<uint8_t> &digest, const std::vector<uint8_t> &signature); 126 127 int32_t DoGenerateFileDigest(PkgStreamPtr stream, uint8_t flags, const size_t fileLen, PkgBuffer &buff, 128 std::pair<DigestAlgorithm::DigestAlgorithmPtr, DigestAlgorithm::DigestAlgorithmPtr> &algorithm); 129 130 int32_t GenerateFileDigest(PkgStreamPtr stream, 131 uint8_t digestMethod, uint8_t flags, std::vector<std::vector<uint8_t>> &digestInfos, size_t hashBufferLen = 0); 132 133 void ClearPkgFile(); 134 135 int32_t SetSignVerifyKeyName(const std::string &keyName); 136 137 int32_t DoCreatePkgStream(PkgStreamPtr &stream, const std::string &fileName, int32_t type); 138 139 const std::string GetExtraPath(const std::string &path); 140 141 private: 142 bool unzipToFile_ {false}; 143 std::vector<PkgFilePtr> pkgFiles_ {}; 144 std::map<std::string, PkgFileConstructor> pkgFileCreator_ {}; 145 std::mutex mapLock_ {}; 146 std::map<std::string, PkgStreamPtr> pkgStreams_ {}; 147 std::string signVerifyKeyName_ {}; 148 PkgDecodeProgress decodeProgress_ { nullptr }; 149 }; 150 } // namespace Hpackage 151 #endif // PKG_MANAGER_IMPL_H 152