sbin/gpt/boot.c

   1 /*
   2  * Copyright (c) 2008 The DragonFly Project.  All rights reserved.
   3  *
   4  * This code is derived from software contributed to The DragonFly Project
   5  * by Matthew Dillon <dillon@backplane.com>
   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  *
  11  * 1. Redistributions of source code must retain the above copyright
  12  *    notice, this list of conditions and the following disclaimer.
  13  * 2. Redistributions in binary form must reproduce the above copyright
  14  *    notice, this list of conditions and the following disclaimer in
  15  *    the documentation and/or other materials provided with the
  16  *    distribution.
  17  * 3. Neither the name of The DragonFly Project nor the names of its
  18  *    contributors may be used to endorse or promote products derived
  19  *    from this software without specific, prior written permission.
  20  *
  21  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  22  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  23  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
  24  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE
  25  * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
  26  * INCIDENTAL, SPECIAL, EXEMPLARY OR CONSEQUENTIAL DAMAGES (INCLUDING,
  27  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  28  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
  29  * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
  30  * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
  31  * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  32  * SUCH DAMAGE.
  33  *
  34  * $DragonFly: src/sbin/gpt/boot.c,v 1.1 2008/07/30 00:45:26 dillon Exp $
  35  */
  36
  37 #include <sys/types.h>
  38
  39 #include <err.h>
  40 #include <stddef.h>
  41 #include <stdio.h>
  42 #include <stdlib.h>
  43 #include <string.h>
  44 #include <unistd.h>
  45 #include <fcntl.h>
  46
  47 #include "map.h"
  48 #include "gpt.h"
  49
  50 static void
  51 usage_boot(void)
  52 {
  53         fprintf(stderr, "usage: %s device\n", getprogname());
  54         exit(1);
  55 }
  56
  57 static void
  58 bootset(int fd)
  59 {
  60         uuid_t uuid;
  61         off_t  block;
  62         off_t  size;
  63         unsigned int entry;
  64         map_t *gpt, *tpg;
  65         map_t *tbl, *lbt;
  66         map_t *map;
  67         u_int32_t status;
  68         struct gpt_hdr *hdr;
  69         struct gpt_ent *ent;
  70         struct mbr *mbr;
  71         int bfd;
  72
  73         /*
  74          * Paramters for boot partition
  75          */
  76         uuid_name_lookup(&uuid, "DragonFly Label32", &status);
  77         if (status != uuid_s_ok)
  78                 err(1, "unable to find uuid for 'DragonFly Label32'");
  79         entry = 0;
  80         block = 0;
  81         size = 128 * 1024 * 1024 / 512;
  82
  83         gpt = map_find(MAP_TYPE_PRI_GPT_HDR);
  84         if (gpt == NULL)
  85                 errx(1, "%s: error: no primary GPT header", device_name);
  86         tpg = map_find(MAP_TYPE_SEC_GPT_HDR);
  87         if (tpg == NULL)
  88                 errx(1, "%s: error: no secondary GPT header", device_name);
  89         tbl = map_find(MAP_TYPE_PRI_GPT_TBL);
  90         lbt = map_find(MAP_TYPE_SEC_GPT_TBL);
  91         if (tbl == NULL || lbt == NULL) {
  92                 errx(1, "%s: error: no primary or secondary gpt table",
  93                      device_name);
  94         }
  95
  96         hdr = gpt->map_data;
  97         if (entry > le32toh(hdr->hdr_entries)) {
  98                 errx(1, "%s: error: index %u out of range (%u max)",
  99                      device_name, entry, le32toh(hdr->hdr_entries));
 100         }
 101
 102         ent = (void *)((char *)tbl->map_data + entry *
 103                        le32toh(hdr->hdr_entsz));
 104         if (!uuid_is_nil(&ent->ent_type, NULL)) {
 105                 errx(1, "%s: error: entry at index %d is not free",
 106                      device_name, entry);
 107         }
 108         map = map_alloc(block, size);
 109         if (map == NULL)
 110                 errx(1, "%s: error: no space available on device", device_name);
 111         block = map->map_start;
 112         size  = map->map_size;
 113
 114         le_uuid_enc(&ent->ent_type, &uuid);
 115         ent->ent_lba_start = htole64(map->map_start);
 116         ent->ent_lba_end = htole64(map->map_start + map->map_size - 1LL);
 117
 118         hdr->hdr_crc_table = htole32(crc32(tbl->map_data,
 119                                      le32toh(hdr->hdr_entries) *
 120                                      le32toh(hdr->hdr_entsz)));
 121         hdr->hdr_crc_self = 0;
 122         hdr->hdr_crc_self = htole32(crc32(hdr, le32toh(hdr->hdr_size)));
 123
 124         gpt_write(fd, gpt);
 125         gpt_write(fd, tbl);
 126
 127         hdr = tpg->map_data;
 128         ent = (void*)((char*)lbt->map_data + entry * le32toh(hdr->hdr_entsz));
 129         le_uuid_enc(&ent->ent_type, &uuid);
 130         ent->ent_lba_start = htole64(map->map_start);
 131         ent->ent_lba_end = htole64(map->map_start + map->map_size - 1LL);
 132
 133         hdr->hdr_crc_table = htole32(crc32(lbt->map_data,
 134                                      le32toh(hdr->hdr_entries) *
 135                                      le32toh(hdr->hdr_entsz)));
 136         hdr->hdr_crc_self = 0;
 137         hdr->hdr_crc_self = htole32(crc32(hdr, le32toh(hdr->hdr_size)));
 138
 139         gpt_write(fd, lbt);
 140         gpt_write(fd, tpg);
 141
 142         /*
 143          * Create a dummy partition
 144          */
 145         map = map_find(MAP_TYPE_PMBR);
 146         if (map == NULL)
 147                 errx(1, "I can't find the PMBR!");
 148         mbr = map->map_data;
 149         if (mbr == NULL)
 150                 errx(1, "I can't find the PMBR's data!");
 151
 152         /*
 153          * Copy in real boot code
 154          */
 155         bfd = open("/boot/boot0", O_RDONLY);
 156         if (bfd < 0 ||
 157             read(bfd, mbr->mbr_code, sizeof(mbr->mbr_code)) !=
 158             sizeof(mbr->mbr_code)) {
 159                 errx(1, "Cannot read /boot/boot0");
 160         }
 161         close(bfd);
 162
 163         /*
 164          * Generate partition #1
 165          */
 166         mbr->mbr_part[1].part_shd = 0xff;
 167         mbr->mbr_part[1].part_ssect = 0xff;
 168         mbr->mbr_part[1].part_scyl = 0xff;
 169         mbr->mbr_part[1].part_ehd = 0xff;
 170         mbr->mbr_part[1].part_esect = 0xff;
 171         mbr->mbr_part[1].part_ecyl = 0xff;
 172         mbr->mbr_part[1].part_start_lo = htole16(block);
 173         mbr->mbr_part[1].part_start_hi = htole16((block) >> 16);
 174         mbr->mbr_part[1].part_size_lo = htole16(size);
 175         mbr->mbr_part[1].part_size_hi = htole16(size >> 16);
 176
 177         mbr->mbr_part[1].part_typ = 165;
 178         mbr->mbr_part[1].part_flag = 0x80;
 179
 180         gpt_write(fd, map);
 181 }
 182
 183 int
 184 cmd_boot(int argc, char *argv[])
 185 {
 186         int ch, fd;
 187
 188         while ((ch = getopt(argc, argv, "")) != -1) {
 189                 switch(ch) {
 190                 default:
 191                         usage_boot();
 192                 }
 193         }
 194
 195         if (argc == optind)
 196                 usage_boot();
 197
 198         while (optind < argc) {
 199                 fd = gpt_open(argv[optind++]);
 200                 if (fd == -1) {
 201                         warn("unable to open device '%s'", device_name);
 202                         continue;
 203                 }
 204                 bootset(fd);
 205                 gpt_close(fd);
 206         }
 207         return (0);
 208 }
 209