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 * 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 * 55 default_scheme(void) 56 { 57 if (strcmp(x86_bootmethod(), "UEFI") == 0) 58 return ("GPT"); 59 else 60 return ("MBR"); 61 } 62 63 int 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 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 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 * 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 * 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 * 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