1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
|
// Copyright 2023, The Android Open Source Project
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
mod cert_tests;
mod test_data;
mod test_ops;
mod verify_tests;
use avb::{slot_verify, HashtreeErrorMode, SlotVerifyData, SlotVerifyFlags, SlotVerifyResult};
use std::{ffi::CString, fs};
use test_data::*;
use test_ops::{FakeVbmetaKey, TestOps};
/// Initializes a `TestOps` object such that verification will succeed on `TEST_PARTITION_NAME`.
///
/// This usually forms the basis of the `TestOps` objects used, with tests modifying the returned
/// object as needed for the individual test case.
fn build_test_ops_one_image_one_vbmeta<'a>() -> TestOps<'a> {
let mut ops = TestOps::default();
ops.add_partition(TEST_PARTITION_NAME, fs::read(TEST_IMAGE_PATH).unwrap());
ops.add_partition("vbmeta", fs::read(TEST_VBMETA_PATH).unwrap());
ops.default_vbmeta_key = Some(FakeVbmetaKey::Avb {
public_key: fs::read(TEST_PUBLIC_KEY_PATH).unwrap(),
public_key_metadata: None,
});
ops.rollbacks.insert(TEST_VBMETA_ROLLBACK_LOCATION, 0);
ops.unlock_state = Ok(false);
ops
}
/// Calls `slot_verify()` using standard args for `build_test_ops_one_image_one_vbmeta()` setup.
fn verify_one_image_one_vbmeta<'a>(
ops: &mut TestOps<'a>,
) -> SlotVerifyResult<'a, SlotVerifyData<'a>> {
slot_verify(
ops,
&[&CString::new(TEST_PARTITION_NAME).unwrap()],
None,
SlotVerifyFlags::AVB_SLOT_VERIFY_FLAGS_NONE,
HashtreeErrorMode::AVB_HASHTREE_ERROR_MODE_EIO,
)
}
|