| /* |
| * Copyright (C) 2016 The Android Open Source Project |
| * |
| * Permission is hereby granted, free of charge, to any person |
| * obtaining a copy of this software and associated documentation |
| * files (the "Software"), to deal in the Software without |
| * restriction, including without limitation the rights to use, copy, |
| * modify, merge, publish, distribute, sublicense, and/or sell copies |
| * of the Software, and to permit persons to whom the Software is |
| * furnished to do so, subject to the following conditions: |
| * |
| * The above copyright notice and this permission notice shall be |
| * included in all copies or substantial portions of the Software. |
| * |
| * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, |
| * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF |
| * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND |
| * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS |
| * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN |
| * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN |
| * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE |
| * SOFTWARE. |
| */ |
| |
| #include "avb_unittest_util.h" |
| |
| #include <android-base/file.h> |
| |
| std::string mem_to_hexstring(const uint8_t* data, size_t len) { |
| std::string ret; |
| char digits[17] = "0123456789abcdef"; |
| for (size_t n = 0; n < len; n++) { |
| ret.push_back(digits[data[n] >> 4]); |
| ret.push_back(digits[data[n] & 0x0f]); |
| } |
| return ret; |
| } |
| |
| std::string string_trim(const std::string& str) { |
| size_t first = str.find_first_not_of(" \t\n"); |
| if (first == std::string::npos) { |
| return str; |
| } |
| size_t last = str.find_last_not_of(" \t\n"); |
| return str.substr(first, (last - first + 1)); |
| } |
| |
| namespace avb { |
| |
| void BaseAvbToolTest::SetUp() { |
| /* Change current directory to test executable directory so that relative path |
| * references to test dependencies don't rely on being manually run from |
| * correct directory */ |
| base::SetCurrentDirectory( |
| base::FilePath(android::base::GetExecutableDirectory())); |
| |
| /* Create temporary directory to stash images in. */ |
| base::FilePath ret; |
| char* buf = strdup("/tmp/libavb-tests.XXXXXX"); |
| ASSERT_TRUE(mkdtemp(buf) != nullptr); |
| testdir_ = base::FilePath(buf); |
| free(buf); |
| |
| /* Reset memory leak tracing */ |
| avb::testing_memory_reset(); |
| } |
| |
| void BaseAvbToolTest::TearDown() { |
| /* Nuke temporary directory. */ |
| ASSERT_EQ(0U, testdir_.value().find("/tmp/libavb-tests")); |
| ASSERT_TRUE(base::DeleteFile(testdir_, true /* recursive */)); |
| /* Ensure all memory has been freed. */ |
| EXPECT_TRUE(avb::testing_memory_all_freed()); |
| } |
| |
| std::string BaseAvbToolTest::CalcVBMetaDigest(const std::string& vbmeta_image, |
| const std::string& digest_alg) { |
| base::FilePath vbmeta_path = testdir_.Append(vbmeta_image); |
| base::FilePath vbmeta_digest_path = testdir_.Append("vbmeta_digest"); |
| EXPECT_COMMAND( |
| 0, |
| "./avbtool calculate_vbmeta_digest --image %s --hash_algorithm %s" |
| " --output %s", |
| vbmeta_path.value().c_str(), |
| digest_alg.c_str(), |
| vbmeta_digest_path.value().c_str()); |
| std::string vbmeta_digest_data; |
| EXPECT_TRUE(base::ReadFileToString(vbmeta_digest_path, &vbmeta_digest_data)); |
| return string_trim(vbmeta_digest_data); |
| } |
| |
| void BaseAvbToolTest::GenerateVBMetaImage( |
| const std::string& image_name, |
| const std::string& algorithm, |
| uint64_t rollback_index, |
| const base::FilePath& key_path, |
| const std::string& additional_options) { |
| std::string signing_options; |
| if (algorithm == "") { |
| signing_options = " --algorithm NONE "; |
| } else { |
| signing_options = std::string(" --algorithm ") + algorithm + " --key " + |
| key_path.value() + " "; |
| } |
| vbmeta_image_path_ = testdir_.Append(image_name); |
| EXPECT_COMMAND(0, |
| "./avbtool make_vbmeta_image" |
| " --rollback_index %" PRIu64 |
| " %s %s " |
| " --output %s", |
| rollback_index, |
| additional_options.c_str(), |
| signing_options.c_str(), |
| vbmeta_image_path_.value().c_str()); |
| int64_t file_size; |
| ASSERT_TRUE(base::GetFileSize(vbmeta_image_path_, &file_size)); |
| vbmeta_image_.resize(file_size); |
| ASSERT_TRUE(base::ReadFile(vbmeta_image_path_, |
| reinterpret_cast<char*>(vbmeta_image_.data()), |
| vbmeta_image_.size())); |
| } |
| |
| /* Generate a file with name |file_name| of size |image_size| with |
| * known content (0x00 0x01 0x02 .. 0xff 0x00 0x01 ..). |
| */ |
| base::FilePath BaseAvbToolTest::GenerateImage(const std::string file_name, |
| size_t image_size, |
| uint8_t start_byte) { |
| std::vector<uint8_t> image; |
| image.resize(image_size); |
| for (size_t n = 0; n < image_size; n++) { |
| image[n] = uint8_t(n + start_byte); |
| } |
| base::FilePath image_path = testdir_.Append(file_name); |
| EXPECT_EQ(image_size, |
| static_cast<const size_t>( |
| base::WriteFile(image_path, |
| reinterpret_cast<const char*>(image.data()), |
| image.size()))); |
| return image_path; |
| } |
| |
| std::string BaseAvbToolTest::InfoImage(const base::FilePath& image_path) { |
| base::FilePath tmp_path = testdir_.Append("info_output.txt"); |
| EXPECT_COMMAND(0, |
| "./avbtool info_image --image %s --output %s", |
| image_path.value().c_str(), |
| tmp_path.value().c_str()); |
| std::string info_data; |
| EXPECT_TRUE(base::ReadFileToString(tmp_path, &info_data)); |
| return info_data; |
| } |
| |
| std::string BaseAvbToolTest::PublicKeyAVB(const base::FilePath& key_path) { |
| base::FilePath tmp_path = testdir_.Append("public_key.bin"); |
| EXPECT_COMMAND(0, |
| "./avbtool extract_public_key --key %s" |
| " --output %s", |
| key_path.value().c_str(), |
| tmp_path.value().c_str()); |
| std::string key_data; |
| EXPECT_TRUE(base::ReadFileToString(tmp_path, &key_data)); |
| return key_data; |
| } |
| |
| } // namespace avb |