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 size_t
  62 bootpart_size(const char *scheme) {
  63         size_t platlen = sizeof(platform);
  64         if (strlen(platform) == 0)
  65                 sysctlbyname(platform_sysctl, platform, &platlen, NULL, -1);
  66
  67         /* No partcode except for GPT */
  68         if (strcmp(scheme, "GPT") != 0)
  69                 return (0);
  70
  71         if (strcmp(platform, "BIOS") == 0)
  72                 return (64*1024);
  73         else
  74                 return (800*1024);
  75
  76         return (0);
  77 }
  78
  79 const char *
  80 bootpart_type(const char *scheme) {
  81         size_t platlen = sizeof(platform);
  82         if (strlen(platform) == 0)
  83                 sysctlbyname(platform_sysctl, platform, &platlen, NULL, -1);
  84
  85         if (strcmp(platform, "UEFI") == 0)
  86                 return ("efi");
  87
  88         return ("freebsd-boot");
  89 }
  90
  91 const char *
  92 bootcode_path(const char *part_type) {
  93         size_t platlen = sizeof(platform);
  94         if (strlen(platform) == 0)
  95                 sysctlbyname(platform_sysctl, platform, &platlen, NULL, -1);
  96         if (strcmp(platform, "UEFI") == 0)
  97                 return (NULL);
  98
  99         if (strcmp(part_type, "GPT") == 0)
 100                 return ("/boot/pmbr");
 101         if (strcmp(part_type, "MBR") == 0)
 102                 return ("/boot/mbr");
 103         if (strcmp(part_type, "BSD") == 0)
 104                 return ("/boot/boot");
 105
 106         return (NULL);
 107 }
 108
 109 const char *
 110 partcode_path(const char *part_type) {
 111         size_t platlen = sizeof(platform);
 112         if (strlen(platform) == 0)
 113                 sysctlbyname(platform_sysctl, platform, &platlen, NULL, -1);
 114
 115         if (strcmp(part_type, "GPT") == 0) {
 116                 if (strcmp(platform, "UEFI") == 0)
 117                         return ("/boot/boot1.efifat");
 118                 else
 119                         return ("/boot/gptboot");
 120         }
 121
 122         /* No partcode except for GPT */
 123         return (NULL);
 124 }
 125