1*8617a60dSAndroid Build Coastguard Worker /* Copyright 2012 The ChromiumOS Authors
2*8617a60dSAndroid Build Coastguard Worker * Use of this source code is governed by a BSD-style license that can be
3*8617a60dSAndroid Build Coastguard Worker * found in the LICENSE file.
4*8617a60dSAndroid Build Coastguard Worker */
5*8617a60dSAndroid Build Coastguard Worker
6*8617a60dSAndroid Build Coastguard Worker #include <errno.h>
7*8617a60dSAndroid Build Coastguard Worker #include <fcntl.h>
8*8617a60dSAndroid Build Coastguard Worker #include <string.h>
9*8617a60dSAndroid Build Coastguard Worker #include <unistd.h>
10*8617a60dSAndroid Build Coastguard Worker
11*8617a60dSAndroid Build Coastguard Worker #include "cgpt.h"
12*8617a60dSAndroid Build Coastguard Worker #include "cgptlib_internal.h"
13*8617a60dSAndroid Build Coastguard Worker #include "cgpt_params.h"
14*8617a60dSAndroid Build Coastguard Worker #include "vboot_host.h"
15*8617a60dSAndroid Build Coastguard Worker
CgptGetBootPartitionNumber(CgptBootParams * params)16*8617a60dSAndroid Build Coastguard Worker int CgptGetBootPartitionNumber(CgptBootParams *params) {
17*8617a60dSAndroid Build Coastguard Worker struct drive drive;
18*8617a60dSAndroid Build Coastguard Worker int gpt_retval= 0;
19*8617a60dSAndroid Build Coastguard Worker int retval;
20*8617a60dSAndroid Build Coastguard Worker
21*8617a60dSAndroid Build Coastguard Worker if (params == NULL)
22*8617a60dSAndroid Build Coastguard Worker return CGPT_FAILED;
23*8617a60dSAndroid Build Coastguard Worker
24*8617a60dSAndroid Build Coastguard Worker if (CGPT_OK != DriveOpen(params->drive_name, &drive, O_RDONLY,
25*8617a60dSAndroid Build Coastguard Worker params->drive_size))
26*8617a60dSAndroid Build Coastguard Worker return CGPT_FAILED;
27*8617a60dSAndroid Build Coastguard Worker
28*8617a60dSAndroid Build Coastguard Worker if (GPT_SUCCESS != (gpt_retval = GptValidityCheck(&drive.gpt))) {
29*8617a60dSAndroid Build Coastguard Worker Error("GptValidityCheck() returned %d: %s\n",
30*8617a60dSAndroid Build Coastguard Worker gpt_retval, GptError(gpt_retval));
31*8617a60dSAndroid Build Coastguard Worker retval = CGPT_FAILED;
32*8617a60dSAndroid Build Coastguard Worker goto done;
33*8617a60dSAndroid Build Coastguard Worker }
34*8617a60dSAndroid Build Coastguard Worker
35*8617a60dSAndroid Build Coastguard Worker if (CGPT_OK != ReadPMBR(&drive)) {
36*8617a60dSAndroid Build Coastguard Worker Error("Unable to read PMBR\n");
37*8617a60dSAndroid Build Coastguard Worker retval = CGPT_FAILED;
38*8617a60dSAndroid Build Coastguard Worker goto done;
39*8617a60dSAndroid Build Coastguard Worker }
40*8617a60dSAndroid Build Coastguard Worker
41*8617a60dSAndroid Build Coastguard Worker char buf[GUID_STRLEN];
42*8617a60dSAndroid Build Coastguard Worker GuidToStr(&drive.pmbr.boot_guid, buf, sizeof(buf));
43*8617a60dSAndroid Build Coastguard Worker
44*8617a60dSAndroid Build Coastguard Worker int numEntries = GetNumberOfEntries(&drive);
45*8617a60dSAndroid Build Coastguard Worker int i;
46*8617a60dSAndroid Build Coastguard Worker for (i = 0; i < numEntries; i++) {
47*8617a60dSAndroid Build Coastguard Worker GptEntry *entry = GetEntry(&drive.gpt, ANY_VALID, i);
48*8617a60dSAndroid Build Coastguard Worker
49*8617a60dSAndroid Build Coastguard Worker if (GuidEqual(&entry->unique, &drive.pmbr.boot_guid)) {
50*8617a60dSAndroid Build Coastguard Worker params->partition = i + 1;
51*8617a60dSAndroid Build Coastguard Worker retval = CGPT_OK;
52*8617a60dSAndroid Build Coastguard Worker goto done;
53*8617a60dSAndroid Build Coastguard Worker }
54*8617a60dSAndroid Build Coastguard Worker }
55*8617a60dSAndroid Build Coastguard Worker
56*8617a60dSAndroid Build Coastguard Worker Error("Didn't find any boot partition\n");
57*8617a60dSAndroid Build Coastguard Worker params->partition = 0;
58*8617a60dSAndroid Build Coastguard Worker retval = CGPT_FAILED;
59*8617a60dSAndroid Build Coastguard Worker
60*8617a60dSAndroid Build Coastguard Worker done:
61*8617a60dSAndroid Build Coastguard Worker (void) DriveClose(&drive, 1);
62*8617a60dSAndroid Build Coastguard Worker return retval;
63*8617a60dSAndroid Build Coastguard Worker }
64*8617a60dSAndroid Build Coastguard Worker
65*8617a60dSAndroid Build Coastguard Worker
CgptBoot(CgptBootParams * params)66*8617a60dSAndroid Build Coastguard Worker int CgptBoot(CgptBootParams *params) {
67*8617a60dSAndroid Build Coastguard Worker struct drive drive;
68*8617a60dSAndroid Build Coastguard Worker int retval = 1;
69*8617a60dSAndroid Build Coastguard Worker int gpt_retval= 0;
70*8617a60dSAndroid Build Coastguard Worker int mode = O_RDONLY;
71*8617a60dSAndroid Build Coastguard Worker
72*8617a60dSAndroid Build Coastguard Worker if (params == NULL)
73*8617a60dSAndroid Build Coastguard Worker return CGPT_FAILED;
74*8617a60dSAndroid Build Coastguard Worker
75*8617a60dSAndroid Build Coastguard Worker if (params->create_pmbr || params->partition || params->bootfile)
76*8617a60dSAndroid Build Coastguard Worker mode = O_RDWR;
77*8617a60dSAndroid Build Coastguard Worker
78*8617a60dSAndroid Build Coastguard Worker if (CGPT_OK != DriveOpen(params->drive_name, &drive, mode,
79*8617a60dSAndroid Build Coastguard Worker params->drive_size)) {
80*8617a60dSAndroid Build Coastguard Worker return CGPT_FAILED;
81*8617a60dSAndroid Build Coastguard Worker }
82*8617a60dSAndroid Build Coastguard Worker
83*8617a60dSAndroid Build Coastguard Worker if (CGPT_OK != ReadPMBR(&drive)) {
84*8617a60dSAndroid Build Coastguard Worker Error("Unable to read PMBR\n");
85*8617a60dSAndroid Build Coastguard Worker goto done;
86*8617a60dSAndroid Build Coastguard Worker }
87*8617a60dSAndroid Build Coastguard Worker
88*8617a60dSAndroid Build Coastguard Worker if (params->create_pmbr) {
89*8617a60dSAndroid Build Coastguard Worker drive.pmbr.magic[0] = 0x1d;
90*8617a60dSAndroid Build Coastguard Worker drive.pmbr.magic[1] = 0x9a;
91*8617a60dSAndroid Build Coastguard Worker drive.pmbr.sig[0] = 0x55;
92*8617a60dSAndroid Build Coastguard Worker drive.pmbr.sig[1] = 0xaa;
93*8617a60dSAndroid Build Coastguard Worker memset(&drive.pmbr.part, 0, sizeof(drive.pmbr.part));
94*8617a60dSAndroid Build Coastguard Worker drive.pmbr.part[0].f_head = 0x00;
95*8617a60dSAndroid Build Coastguard Worker drive.pmbr.part[0].f_sect = 0x02;
96*8617a60dSAndroid Build Coastguard Worker drive.pmbr.part[0].f_cyl = 0x00;
97*8617a60dSAndroid Build Coastguard Worker drive.pmbr.part[0].type = 0xee;
98*8617a60dSAndroid Build Coastguard Worker drive.pmbr.part[0].l_head = 0xff;
99*8617a60dSAndroid Build Coastguard Worker drive.pmbr.part[0].l_sect = 0xff;
100*8617a60dSAndroid Build Coastguard Worker drive.pmbr.part[0].l_cyl = 0xff;
101*8617a60dSAndroid Build Coastguard Worker drive.pmbr.part[0].f_lba = htole32(1);
102*8617a60dSAndroid Build Coastguard Worker uint32_t max = 0xffffffff;
103*8617a60dSAndroid Build Coastguard Worker if (drive.gpt.streaming_drive_sectors < 0xffffffff)
104*8617a60dSAndroid Build Coastguard Worker max = drive.gpt.streaming_drive_sectors - 1;
105*8617a60dSAndroid Build Coastguard Worker drive.pmbr.part[0].num_sect = htole32(max);
106*8617a60dSAndroid Build Coastguard Worker }
107*8617a60dSAndroid Build Coastguard Worker
108*8617a60dSAndroid Build Coastguard Worker if (params->partition) {
109*8617a60dSAndroid Build Coastguard Worker if (GPT_SUCCESS != (gpt_retval = GptValidityCheck(&drive.gpt))) {
110*8617a60dSAndroid Build Coastguard Worker Error("GptValidityCheck() returned %d: %s\n",
111*8617a60dSAndroid Build Coastguard Worker gpt_retval, GptError(gpt_retval));
112*8617a60dSAndroid Build Coastguard Worker goto done;
113*8617a60dSAndroid Build Coastguard Worker }
114*8617a60dSAndroid Build Coastguard Worker
115*8617a60dSAndroid Build Coastguard Worker if (params->partition > GetNumberOfEntries(&drive)) {
116*8617a60dSAndroid Build Coastguard Worker Error("invalid partition number: %d\n", params->partition);
117*8617a60dSAndroid Build Coastguard Worker goto done;
118*8617a60dSAndroid Build Coastguard Worker }
119*8617a60dSAndroid Build Coastguard Worker
120*8617a60dSAndroid Build Coastguard Worker uint32_t index = params->partition - 1;
121*8617a60dSAndroid Build Coastguard Worker GptEntry *entry = GetEntry(&drive.gpt, ANY_VALID, index);
122*8617a60dSAndroid Build Coastguard Worker memcpy(&drive.pmbr.boot_guid, &entry->unique, sizeof(Guid));
123*8617a60dSAndroid Build Coastguard Worker }
124*8617a60dSAndroid Build Coastguard Worker
125*8617a60dSAndroid Build Coastguard Worker if (params->bootfile) {
126*8617a60dSAndroid Build Coastguard Worker int fd = open(params->bootfile, O_RDONLY);
127*8617a60dSAndroid Build Coastguard Worker if (fd < 0) {
128*8617a60dSAndroid Build Coastguard Worker Error("Can't read %s: %s\n", params->bootfile, strerror(errno));
129*8617a60dSAndroid Build Coastguard Worker goto done;
130*8617a60dSAndroid Build Coastguard Worker }
131*8617a60dSAndroid Build Coastguard Worker
132*8617a60dSAndroid Build Coastguard Worker int n = read(fd, drive.pmbr.bootcode, sizeof(drive.pmbr.bootcode));
133*8617a60dSAndroid Build Coastguard Worker if (n < 1) {
134*8617a60dSAndroid Build Coastguard Worker Error("problem reading %s: %s\n", params->bootfile, strerror(errno));
135*8617a60dSAndroid Build Coastguard Worker close(fd);
136*8617a60dSAndroid Build Coastguard Worker goto done;
137*8617a60dSAndroid Build Coastguard Worker }
138*8617a60dSAndroid Build Coastguard Worker
139*8617a60dSAndroid Build Coastguard Worker close(fd);
140*8617a60dSAndroid Build Coastguard Worker }
141*8617a60dSAndroid Build Coastguard Worker
142*8617a60dSAndroid Build Coastguard Worker char buf[GUID_STRLEN];
143*8617a60dSAndroid Build Coastguard Worker GuidToStr(&drive.pmbr.boot_guid, buf, sizeof(buf));
144*8617a60dSAndroid Build Coastguard Worker printf("%s\n", buf);
145*8617a60dSAndroid Build Coastguard Worker
146*8617a60dSAndroid Build Coastguard Worker // Write it all out, if needed.
147*8617a60dSAndroid Build Coastguard Worker if (mode == O_RDONLY || CGPT_OK == WritePMBR(&drive))
148*8617a60dSAndroid Build Coastguard Worker retval = 0;
149*8617a60dSAndroid Build Coastguard Worker
150*8617a60dSAndroid Build Coastguard Worker done:
151*8617a60dSAndroid Build Coastguard Worker (void) DriveClose(&drive, 1);
152*8617a60dSAndroid Build Coastguard Worker return retval;
153*8617a60dSAndroid Build Coastguard Worker }
154