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 35 #include <sys/types.h> 36 37 #include <err.h> 38 #include <stddef.h> 39 #include <stdio.h> 40 #include <stdlib.h> 41 #include <string.h> 42 #include <unistd.h> 43 #include <fcntl.h> 44 45 #include "map.h" 46 #include "gpt.h" 47 48 static void 49 usage_boot(void) 50 { 51 fprintf(stderr, "usage: %s device\n", getprogname()); 52 exit(1); 53 } 54 55 static void 56 bootset(int fd) 57 { 58 uuid_t uuid; 59 off_t block; 60 off_t size; 61 unsigned int entry; 62 map_t *gpt, *tpg; 63 map_t *tbl, *lbt; 64 map_t *map; 65 u_int32_t status; 66 struct gpt_hdr *hdr; 67 struct gpt_ent *ent; 68 struct mbr *mbr; 69 int bfd; 70 71 /* 72 * Paramters for boot partition 73 */ 74 uuid_name_lookup(&uuid, "DragonFly Label32", &status); 75 if (status != uuid_s_ok) 76 err(1, "unable to find uuid for 'DragonFly Label32'"); 77 entry = 0; 78 block = 0; 79 size = (off_t)1024 * 1024 * 1024 / 512; /* 1GB */ 80 81 gpt = map_find(MAP_TYPE_PRI_GPT_HDR); 82 if (gpt == NULL) 83 errx(1, "%s: error: no primary GPT header", device_name); 84 tpg = map_find(MAP_TYPE_SEC_GPT_HDR); 85 if (tpg == NULL) 86 errx(1, "%s: error: no secondary GPT header", device_name); 87 tbl = map_find(MAP_TYPE_PRI_GPT_TBL); 88 lbt = map_find(MAP_TYPE_SEC_GPT_TBL); 89 if (tbl == NULL || lbt == NULL) { 90 errx(1, "%s: error: no primary or secondary gpt table", 91 device_name); 92 } 93 94 hdr = gpt->map_data; 95 if (entry > le32toh(hdr->hdr_entries)) { 96 errx(1, "%s: error: index %u out of range (%u max)", 97 device_name, entry, le32toh(hdr->hdr_entries)); 98 } 99 100 ent = (void *)((char *)tbl->map_data + entry * 101 le32toh(hdr->hdr_entsz)); 102 if (!uuid_is_nil(&ent->ent_type, NULL)) { 103 errx(1, "%s: error: entry at index %d is not free", 104 device_name, entry); 105 } 106 map = map_alloc(block, size); 107 if (map == NULL) 108 errx(1, "%s: error: no space available on device", device_name); 109 block = map->map_start; 110 size = map->map_size; 111 112 le_uuid_enc(&ent->ent_type, &uuid); 113 ent->ent_lba_start = htole64(map->map_start); 114 ent->ent_lba_end = htole64(map->map_start + map->map_size - 1LL); 115 116 hdr->hdr_crc_table = htole32(crc32(tbl->map_data, 117 le32toh(hdr->hdr_entries) * 118 le32toh(hdr->hdr_entsz))); 119 hdr->hdr_crc_self = 0; 120 hdr->hdr_crc_self = htole32(crc32(hdr, le32toh(hdr->hdr_size))); 121 122 gpt_write(fd, gpt); 123 gpt_write(fd, tbl); 124 125 hdr = tpg->map_data; 126 ent = (void*)((char*)lbt->map_data + entry * le32toh(hdr->hdr_entsz)); 127 le_uuid_enc(&ent->ent_type, &uuid); 128 ent->ent_lba_start = htole64(map->map_start); 129 ent->ent_lba_end = htole64(map->map_start + map->map_size - 1LL); 130 131 hdr->hdr_crc_table = htole32(crc32(lbt->map_data, 132 le32toh(hdr->hdr_entries) * 133 le32toh(hdr->hdr_entsz))); 134 hdr->hdr_crc_self = 0; 135 hdr->hdr_crc_self = htole32(crc32(hdr, le32toh(hdr->hdr_size))); 136 137 gpt_write(fd, lbt); 138 gpt_write(fd, tpg); 139 140 /* 141 * Create a dummy partition 142 */ 143 map = map_find(MAP_TYPE_PMBR); 144 if (map == NULL) 145 errx(1, "I can't find the PMBR!"); 146 mbr = map->map_data; 147 if (mbr == NULL) 148 errx(1, "I can't find the PMBR's data!"); 149 150 /* 151 * Copy in real boot code 152 */ 153 bfd = open("/boot/boot0", O_RDONLY); 154 if (bfd < 0 || 155 read(bfd, mbr->mbr_code, sizeof(mbr->mbr_code)) != 156 sizeof(mbr->mbr_code)) { 157 errx(1, "Cannot read /boot/boot0"); 158 } 159 close(bfd); 160 161 /* 162 * Generate partition #1 163 */ 164 mbr->mbr_part[1].part_shd = 0xff; 165 mbr->mbr_part[1].part_ssect = 0xff; 166 mbr->mbr_part[1].part_scyl = 0xff; 167 mbr->mbr_part[1].part_ehd = 0xff; 168 mbr->mbr_part[1].part_esect = 0xff; 169 mbr->mbr_part[1].part_ecyl = 0xff; 170 mbr->mbr_part[1].part_start_lo = htole16(block); 171 mbr->mbr_part[1].part_start_hi = htole16((block) >> 16); 172 mbr->mbr_part[1].part_size_lo = htole16(size); 173 mbr->mbr_part[1].part_size_hi = htole16(size >> 16); 174 175 mbr->mbr_part[1].part_typ = 108; 176 mbr->mbr_part[1].part_flag = 0x80; 177 178 gpt_write(fd, map); 179 } 180 181 int 182 cmd_boot(int argc, char *argv[]) 183 { 184 int ch, fd; 185 186 while ((ch = getopt(argc, argv, "")) != -1) { 187 switch(ch) { 188 default: 189 usage_boot(); 190 } 191 } 192 193 if (argc == optind) 194 usage_boot(); 195 196 while (optind < argc) { 197 fd = gpt_open(argv[optind++]); 198 if (fd == -1) { 199 warn("unable to open device '%s'", device_name); 200 continue; 201 } 202 bootset(fd); 203 gpt_close(fd); 204 } 205 return (0); 206 } 207 208