xref: /aosp_15_r20/external/vboot_reference/cgpt/cgpt_boot.c (revision 8617a60d3594060b7ecbd21bc622a7c14f3cf2bc)
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