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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
|
/* Copyright 2014 The ChromiumOS Authors
* Use of this source code is governed by a BSD-style license that can be
* found in the LICENSE file.
*
* Routines for verifying a kernel or disk image
*/
#include "2sysincludes.h"
#include "2api.h"
#include "2common.h"
#include "2misc.h"
#include "2nvstorage.h"
#include "2secdata.h"
#include "host_common.h"
#include "util_misc.h"
#include "vboot_api.h"
static uint8_t workbuf[VB2_KERNEL_WORKBUF_RECOMMENDED_SIZE]
__attribute__((aligned(VB2_WORKBUF_ALIGN)));
static struct vb2_context *ctx;
static uint8_t *diskbuf;
static struct vb2_kernel_params params;
static struct vb2_disk_info disk_info;
vb2_error_t VbExDiskRead(vb2ex_disk_handle_t handle, uint64_t lba_start,
uint64_t lba_count, void *buffer)
{
if (handle != (vb2ex_disk_handle_t)1)
return VB2_ERROR_UNKNOWN;
if (lba_start >= disk_info.streaming_lba_count)
return VB2_ERROR_UNKNOWN;
if (lba_start + lba_count > disk_info.streaming_lba_count)
return VB2_ERROR_UNKNOWN;
memcpy(buffer, diskbuf + lba_start * 512, lba_count * 512);
return VB2_SUCCESS;
}
vb2_error_t VbExDiskWrite(vb2ex_disk_handle_t handle, uint64_t lba_start,
uint64_t lba_count, const void *buffer)
{
if (handle != (vb2ex_disk_handle_t)1)
return VB2_ERROR_UNKNOWN;
if (lba_start >= disk_info.streaming_lba_count)
return VB2_ERROR_UNKNOWN;
if (lba_start + lba_count > disk_info.streaming_lba_count)
return VB2_ERROR_UNKNOWN;
memcpy(diskbuf + lba_start * 512, buffer, lba_count * 512);
return VB2_SUCCESS;
}
static void print_help(const char *progname)
{
printf("\nUsage: %s <disk_image> <kernel.vbpubk>\n\n",
progname);
}
int main(int argc, char *argv[])
{
uint8_t *kernkey = NULL;
uint64_t kernkey_size = 0;
uint64_t disk_bytes = 0;
vb2_error_t rv;
if (argc < 3) {
print_help(argv[0]);
return 1;
}
/* Load disk file */
/* TODO: is it better to mmap() in the long run? */
diskbuf = ReadFile(argv[1], &disk_bytes);
if (!diskbuf) {
fprintf(stderr, "Can't read disk file %s\n", argv[1]);
return 1;
}
/* Read public key */
kernkey = ReadFile(argv[2], &kernkey_size);
if (!kernkey) {
fprintf(stderr, "Can't read key file %s\n", argv[2]);
return 1;
}
/* Set up params */
disk_info.handle = (vb2ex_disk_handle_t)1;
disk_info.bytes_per_lba = 512;
disk_info.streaming_lba_count = disk_bytes / 512;
disk_info.lba_count = disk_info.streaming_lba_count;
params.kernel_buffer_size = 16 * 1024 * 1024;
params.kernel_buffer = malloc(params.kernel_buffer_size);
if (!params.kernel_buffer) {
fprintf(stderr, "Can't allocate kernel buffer\n");
return 1;
}
/* TODO(chromium:441893): support dev-mode flag and external gpt flag */
disk_info.flags = 0;
if (vb2api_init(&workbuf, sizeof(workbuf), &ctx)) {
fprintf(stderr, "Can't initialize workbuf\n");
return 1;
}
rv = vb2api_inject_kernel_subkey(ctx, kernkey, kernkey_size);
if (rv != VB2_SUCCESS) {
fprintf(stderr, "vb2api_inject_kernel_subkey failed: %x\n",
rv);
return 1;
}
/* Try loading kernel */
rv = vb2api_load_kernel(ctx, ¶ms, &disk_info);
if (rv != VB2_SUCCESS) {
fprintf(stderr, "vb2api_load_kernel() failed with code %x\n",
rv);
return 1;
}
printf("Found a good kernel.\n");
printf("Partition number: %u\n", params.partition_number);
printf("Bootloader address: 0x%" PRIx64 "\n",
params.bootloader_address);
/* TODO: print other things (partition GUID, shared_data) */
printf("Yaay!\n");
return 0;
}
|
