usr.sbin/bsdinstall/partedit/partedit_x86.c

   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$
  27  */
  28
  29 #include <sys/types.h>
  30 #include <sys/sysctl.h>
  31 #include <string.h>
  32
  33 #include "partedit.h"
  34
  35 static char platform[255] = "";
  36 static const char *platform_sysctl = "machdep.bootmethod";
  37
  38 const char *
  39 default_scheme(void) {
  40         return ("GPT");
  41 }
  42
  43 int
  44 is_scheme_bootable(const char *part_type) {
  45         size_t platlen = sizeof(platform);
  46         if (strlen(platform) == 0)
  47                 sysctlbyname(platform_sysctl, platform, &platlen, NULL, -1);
  48
  49         if (strcmp(part_type, "GPT") == 0)
  50                 return (1);
  51         if (strcmp(platform, "BIOS") == 0) {
  52                 if (strcmp(part_type, "BSD") == 0)
  53                         return (1);
  54                 if (strcmp(part_type, "MBR") == 0)
  55                         return (1);
  56         }
  57
  58         return (0);
  59 }
  60
  61 int
  62 is_fs_bootable(const char *part_type, const char *fs) {
  63         size_t platlen = sizeof(platform);
  64         if (strlen(platform) == 0)
  65                 sysctlbyname(platform_sysctl, platform, &platlen, NULL, -1);
  66
  67         if (strcmp(fs, "freebsd-ufs") == 0)
  68                 return (1);
  69
  70         if (strcmp(fs, "freebsd-zfs") == 0 && strcmp(platform, "BIOS") == 0)
  71                 return (1);
  72
  73         return (0);
  74 }
  75
  76 size_t
  77 bootpart_size(const char *scheme) {
  78         size_t platlen = sizeof(platform);
  79         if (strlen(platform) == 0)
  80                 sysctlbyname(platform_sysctl, platform, &platlen, NULL, -1);
  81
  82         /* No partcode except for GPT */
  83         if (strcmp(scheme, "GPT") != 0)
  84                 return (0);
  85
  86         if (strcmp(platform, "BIOS") == 0)
  87                 return (512*1024);
  88         else
  89                 return (800*1024);
  90
  91         return (0);
  92 }
  93
  94 const char *
  95 bootpart_type(const char *scheme) {
  96         size_t platlen = sizeof(platform);
  97         if (strlen(platform) == 0)
  98                 sysctlbyname(platform_sysctl, platform, &platlen, NULL, -1);
  99
 100         if (strcmp(platform, "UEFI") == 0)
 101                 return ("efi");
 102
 103         return ("freebsd-boot");
 104 }
 105
 106 const char *
 107 bootcode_path(const char *part_type) {
 108         size_t platlen = sizeof(platform);
 109         if (strlen(platform) == 0)
 110                 sysctlbyname(platform_sysctl, platform, &platlen, NULL, -1);
 111         if (strcmp(platform, "UEFI") == 0)
 112                 return (NULL);
 113
 114         if (strcmp(part_type, "GPT") == 0)
 115                 return ("/boot/pmbr");
 116         if (strcmp(part_type, "MBR") == 0)
 117                 return ("/boot/mbr");
 118         if (strcmp(part_type, "BSD") == 0)
 119                 return ("/boot/boot");
 120
 121         return (NULL);
 122 }
 123
 124 const char *
 125 partcode_path(const char *part_type, const char *fs_type) {
 126         size_t platlen = sizeof(platform);
 127         if (strlen(platform) == 0)
 128                 sysctlbyname(platform_sysctl, platform, &platlen, NULL, -1);
 129
 130         if (strcmp(part_type, "GPT") == 0) {
 131                 if (strcmp(platform, "UEFI") == 0)
 132                         return ("/boot/boot1.efifat");
 133                 else if (strcmp(fs_type, "zfs") == 0)
 134                         return ("/boot/gptzfsboot");
 135                 else
 136                         return ("/boot/gptboot");
 137         }
 138
 139         /* No partcode except for GPT */
 140         return (NULL);
 141 }
 142