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 time_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 53 private: 54 void DecodeHeaderCalOffset(uint8_t flags, const PkgBuffer &buffer, size_t &offset, 55 std::string &fileName) const; 56 }; 57 58 class GZipPkgFile : public PkgFile { 59 public: GZipPkgFile(PkgManager::PkgManagerPtr manager,PkgStreamPtr stream)60 explicit GZipPkgFile(PkgManager::PkgManagerPtr manager, PkgStreamPtr stream) 61 : PkgFile(manager, stream, PkgFile::PKG_TYPE_GZIP) 62 { 63 pkgInfo_.signMethod = PKG_SIGN_METHOD_RSA; 64 pkgInfo_.digestMethod = PKG_DIGEST_TYPE_SHA256; 65 } 66 ~GZipPkgFile()67 ~GZipPkgFile() override {} 68 69 int32_t AddEntry(const PkgManager::FileInfoPtr file, const PkgStreamPtr inStream) override; 70 71 int32_t SavePackage(size_t &offset) override; 72 73 int32_t LoadPackage(std::vector<std::string> &fileNames, VerifyFunction verifier = nullptr) override; 74 75 private: 76 PkgInfo pkgInfo_ {}; 77 size_t currentOffset_ = 0; 78 }; 79 } // namespace Hpackage 80 #endif // GZIP_PKG_FILE_H 81