src/lib/assembly/vbmeta/tests/integration_test.rs

// Copyright 2021 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

use include_bytes_from_working_dir::include_bytes_from_working_dir_env;
use include_str_from_working_dir::include_str_from_working_dir_env;

use vbmeta::Key;
use vbmeta::VBMeta;
use vbmeta::{HashDescriptor, Salt};

const PEM: &str = include_str_from_working_dir_env!("AVB_KEY");
const METADATA: &[u8] = include_bytes_from_working_dir_env!("AVB_METADATA");
const SALT: &str = env!("SALT");
const IMAGE: &[u8] = include_bytes_from_working_dir_env!("IMAGE");
const EXPECTED_VBMETA: &[u8] = include_bytes_from_working_dir_env!("EXPECTED_VBMETA");

#[test]
fn avbtool_comparison() {
    let key = Key::try_new(PEM, METADATA).unwrap();

    let salt_bytes: &[u8] = &hex::decode(SALT).unwrap();
    let salt = Salt::try_from(salt_bytes).unwrap();
    let descriptor = HashDescriptor::new("zircon", IMAGE, salt);
    let descriptors = vec![descriptor];

    let vbmeta = VBMeta::sign(descriptors, key).unwrap();
    assert_eq!(vbmeta.as_bytes(), EXPECTED_VBMETA);
}