/*
|
* 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
|
