1 /* 2 * Copyright (c) 2024-2024 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 SIGNATRUETOOLS_MERKLE_TREE_EXTENSION_H 16 #define SIGNATRUETOOLS_MERKLE_TREE_EXTENSION_H 17 18 #include <vector> 19 #include <string> 20 21 #include "extension.h" 22 #include "signature_tools_log.h" 23 24 namespace OHOS { 25 namespace SignatureTools { 26 27 class MerkleTreeExtension : public Extension { 28 public: 29 static constexpr int32_t MERKLE_TREE_INLINED = 0x1; 30 static constexpr int32_t MERKLE_TREE_EXTENSION_DATA_SIZE = 80; 31 MerkleTreeExtension(); 32 MerkleTreeExtension(int64_t merkleTreeSize, int64_t merkleTreeOffset, const std::vector<int8_t> rootHash); 33 virtual ~MerkleTreeExtension(); 34 static MerkleTreeExtension* FromByteArray(std::vector<int8_t>& bytes); 35 virtual int32_t GetSize(); 36 virtual void ToByteArray(std::vector<int8_t>& ret); 37 int64_t GetMerkleTreeSize(); 38 int64_t GetMerkleTreeOffset(); 39 void SetMerkleTreeOffset(int64_t offset); 40 41 private: 42 static constexpr int32_t ROOT_HASH_SIZE = 64; 43 static constexpr int32_t PAGE_SIZE_4K = 4096; 44 int64_t merkleTreeSize; 45 int64_t merkleTreeOffset; 46 std::vector<int8_t> rootHash; 47 }; 48 } // namespace SignatureTools 49 } // namespace OHOS 50 #endif // SIGNATRUETOOLS_MERKLE_TREE_EXTENSION_H 51