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