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 GZIP_PKG_FILE_H 16 #define GZIP_PKG_FILE_H 17 18 #include "pkg_pkgfile.h" 19 #include "pkg_utils.h" 20 #include "pkg_zipfile.h" 21 #include "zlib.h" 22 23 namespace hpackage { 24 struct __attribute__((packed)) GZipHeader { 25 uint16_t magic = 0; 26 uint8_t method = 0; 27 uint8_t flags = 0; 28 uint32_t mtime = 0; 29 uint8_t xfl = 0; 30 uint8_t osFile = 0; 31 }; 32 33 struct __attribute__((packed)) GZipExtra { 34 uint8_t si1 = 0; 35 uint8_t si2 = 0; 36 uint16_t xlen = 0; 37 }; 38 39 class GZipFileEntry : public ZipFileEntry { 40 public: GZipFileEntry(PkgFilePtr pkgFile,uint32_t nodeId)41 GZipFileEntry(PkgFilePtr pkgFile, uint32_t nodeId) : ZipFileEntry(pkgFile, nodeId) {} 42 ~GZipFileEntry()43 ~GZipFileEntry() override {} 44 45 int32_t EncodeHeader(PkgStreamPtr inStream, size_t startOffset, size_t &encodeLen) override; 46 47 int32_t Pack(PkgStreamPtr inStream, size_t startOffset, size_t &encodeLen) override; 48 49 int32_t Unpack(PkgStreamPtr outStream) override; 50 51 int32_t DecodeHeader(const PkgBuffer &buffer, size_t, size_t, size_t &decodeLen) override; 52 private: 53 void DecodeHeaderCalOffset(uint8_t flags, const PkgBuffer &buffer, size_t &offset, 54 std::string &fileName) const; 55 }; 56 57 class GZipPkgFile : public PkgFile { 58 public: GZipPkgFile(PkgManager::PkgManagerPtr manager,PkgStreamPtr stream)59 explicit GZipPkgFile(PkgManager::PkgManagerPtr manager, PkgStreamPtr stream) 60 : PkgFile(manager, stream, PkgFile::PKG_TYPE_GZIP) 61 { 62 pkgInfo_.signMethod = PKG_SIGN_METHOD_RSA; 63 pkgInfo_.digestMethod = PKG_DIGEST_TYPE_SHA256; 64 } 65 ~GZipPkgFile()66 ~GZipPkgFile() override {} 67 68 int32_t AddEntry(const PkgManager::FileInfoPtr file, const PkgStreamPtr inStream) override; 69 70 int32_t SavePackage(size_t &offset) override; 71 72 int32_t LoadPackage(std::vector<std::string> &fileNames, VerifyFunction verifier = nullptr) override; 73 private: 74 PkgInfo pkgInfo_ {}; 75 size_t currentOffset_ = 0; 76 }; 77 } // namespace hpackage 78 #endif // GZIP_PKG_FILE_H 79