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_ELF_SIGN_BLOCK_H 16 #define SIGNATRUETOOLS_ELF_SIGN_BLOCK_H 17 18 #include <vector> 19 #include <string> 20 21 #include "fs_verity_descriptor_with_sign.h" 22 23 namespace OHOS { 24 namespace SignatureTools { 25 26 class ElfSignBlock { 27 public: 28 static constexpr int32_t PAGE_SIZE_4K = 4096; 29 static constexpr int32_t MERKLE_TREE_INLINED = 0x2; 30 31 public: 32 ElfSignBlock(); 33 ElfSignBlock(int32_t paddingSize, const std::vector<int8_t> &merkleTreeData, 34 const FsVerityDescriptorWithSign &descriptorWithSign); 35 int32_t Size(); 36 std::vector<int8_t>& GetMerkleTreeWithPadding(); 37 int64_t GetDataSize(); 38 int64_t GetTreeOffset(); 39 std::vector<int8_t>& GetSignature(); 40 void ToByteArray(std::vector<int8_t>& ret); 41 static bool FromByteArray(std::vector<int8_t>& bytes, ElfSignBlock& elfSignBlock); 42 static int32_t ComputeMerkleTreePaddingLength(int64_t signBlockOffset); 43 44 private: 45 int32_t type = MERKLE_TREE_INLINED; 46 int32_t treeLength; 47 std::vector<int8_t> merkleTreeWithPadding; 48 FsVerityDescriptorWithSign descriptorWithSign; 49 }; 50 } // namespace SignatureTools 51 } // namespace OHOS 52 #endif // SIGNATRUETOOLS_ELF_SIGN_BLOCK_H