1 /*-
2 * SPDX-License-Identifier: BSD-2-Clause
3 *
4 * Copyright (c) 2011 Nathan Whitehorn
5 * All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 * 1. Redistributions of source code must retain the above copyright
11 * notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 * notice, this list of conditions and the following disclaimer in the
14 * documentation and/or other materials provided with the distribution.
15 *
16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
17 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
20 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
21 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
22 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
23 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
24 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
25 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
26 * SUCH DAMAGE.
27 */
28
29 #include <sys/types.h>
30 #include <sys/sysctl.h>
31 #include <string.h>
32
33 #include "partedit.h"
34
35 /* EFI partition size in bytes */
36 #define EFI_BOOTPART_SIZE (260 * 1024 * 1024)
37
38 static const char *
x86_bootmethod(void)39 x86_bootmethod(void)
40 {
41 static char fw[255] = "";
42 size_t len = sizeof(fw);
43 int error;
44
45 if (strlen(fw) == 0) {
46 error = sysctlbyname("machdep.bootmethod", fw, &len, NULL, -1);
47 if (error != 0)
48 return ("BIOS");
49 }
50
51 return (fw);
52 }
53
54 const char *
default_scheme(void)55 default_scheme(void)
56 {
57 if (strcmp(x86_bootmethod(), "UEFI") == 0)
58 return ("GPT");
59 else
60 return ("MBR");
61 }
62
63 int
is_scheme_bootable(const char * part_type)64 is_scheme_bootable(const char *part_type)
65 {
66
67 if (strcmp(part_type, "GPT") == 0)
68 return (1);
69 if (strcmp(x86_bootmethod(), "BIOS") == 0) {
70 if (strcmp(part_type, "BSD") == 0)
71 return (1);
72 if (strcmp(part_type, "MBR") == 0)
73 return (1);
74 }
75
76 return (0);
77 }
78
79 int
is_fs_bootable(const char * part_type,const char * fs)80 is_fs_bootable(const char *part_type, const char *fs)
81 {
82
83 if (strcmp(fs, "freebsd-ufs") == 0)
84 return (1);
85
86 if (strcmp(fs, "freebsd-zfs") == 0 &&
87 strcmp(part_type, "GPT") == 0 &&
88 strcmp(x86_bootmethod(), "BIOS") == 0)
89 return (1);
90
91 return (0);
92 }
93
94 size_t
bootpart_size(const char * scheme)95 bootpart_size(const char *scheme)
96 {
97
98 /* No partcode except for GPT */
99 if (strcmp(scheme, "GPT") != 0)
100 return (0);
101
102 if (strcmp(x86_bootmethod(), "BIOS") == 0)
103 return (512*1024);
104 else
105 return (EFI_BOOTPART_SIZE);
106
107 return (0);
108 }
109
110 const char *
bootpart_type(const char * scheme,const char ** mountpoint)111 bootpart_type(const char *scheme, const char **mountpoint)
112 {
113
114 if (strcmp(x86_bootmethod(), "UEFI") == 0) {
115 *mountpoint = "/boot/efi";
116 return ("efi");
117 }
118
119 return ("freebsd-boot");
120 }
121
122 const char *
bootcode_path(const char * part_type)123 bootcode_path(const char *part_type)
124 {
125
126 if (strcmp(x86_bootmethod(), "UEFI") == 0)
127 return (NULL);
128
129 if (strcmp(part_type, "GPT") == 0)
130 return ("/boot/pmbr");
131 if (strcmp(part_type, "MBR") == 0)
132 return ("/boot/mbr");
133 if (strcmp(part_type, "BSD") == 0)
134 return ("/boot/boot");
135
136 return (NULL);
137 }
138
139 const char *
partcode_path(const char * part_type,const char * fs_type)140 partcode_path(const char *part_type, const char *fs_type)
141 {
142
143 if (strcmp(part_type, "GPT") == 0 && strcmp(x86_bootmethod(), "UEFI") != 0) {
144 if (strcmp(fs_type, "zfs") == 0)
145 return ("/boot/gptzfsboot");
146 else
147 return ("/boot/gptboot");
148 }
149
150 /* No partcode except for non-UEFI GPT */
151 return (NULL);
152 }
153
154