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.2 2008/08/21 23:10:04 thomas 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 = (off_t)1024 * 1024 * 1024 / 512; /* 1GB */ 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 210