1 /*- 2 * Copyright (c) 2002 Marcel Moolenaar 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 * 9 * 1. Redistributions of source code must retain the above copyright 10 * notice, this list of conditions and the following disclaimer. 11 * 2. Redistributions in binary form must reproduce the above copyright 12 * notice, this list of conditions and the following disclaimer in the 13 * documentation and/or other materials provided with the distribution. 14 * 15 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 16 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 17 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 18 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, 19 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 20 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 21 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 22 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 23 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 24 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 25 */ 26 27 #if HAVE_NBTOOL_CONFIG_H 28 #include "nbtool_config.h" 29 #endif 30 31 #include <sys/cdefs.h> 32 #ifdef __FBSDID 33 __FBSDID("$FreeBSD: src/sbin/gpt/add.c,v 1.14 2006/06/22 22:05:28 marcel Exp $"); 34 #endif 35 #ifdef __RCSID 36 __RCSID("$NetBSD: resizedisk.c,v 1.17 2015/12/04 21:39:18 christos Exp $"); 37 #endif 38 39 #include <sys/bootblock.h> 40 #include <sys/types.h> 41 42 #include <err.h> 43 #include <stddef.h> 44 #include <stdio.h> 45 #include <stdlib.h> 46 #include <string.h> 47 #include <unistd.h> 48 49 #include "map.h" 50 #include "gpt.h" 51 #include "gpt_private.h" 52 53 54 static int cmd_resizedisk(gpt_t, int, char *[]); 55 56 static const char *resizediskhelp[] = { 57 "[-s size]", 58 }; 59 60 struct gpt_cmd c_resizedisk = { 61 "resizedisk", 62 cmd_resizedisk, 63 resizediskhelp, __arraycount(resizediskhelp), 64 0, 65 }; 66 67 #define usage() gpt_usage(NULL, &c_resizedisk) 68 69 /* 70 * relocate the secondary GPT based on the following criteria: 71 * - size not specified 72 * - disk has not changed size, do nothing 73 * - disk has grown, relocate secondary 74 * - disk has shrunk, create new secondary 75 * - size specified 76 * - size is larger then disk or same as current location, do nothing 77 * - relocate or create new secondary 78 * - when shrinking, verify that table fits 79 */ 80 static int 81 resizedisk(gpt_t gpt, off_t sector, off_t size) 82 { 83 map_t mbrmap; 84 struct gpt_hdr *hdr; 85 struct gpt_ent *ent; 86 struct mbr *mbr; 87 off_t last, oldloc, newloc, lastdata, gpt_size; 88 int i; 89 90 last = gpt->mediasz / gpt->secsz - 1; 91 lastdata = 0; 92 newloc = 0; 93 94 if (sector > last) { 95 gpt_warnx(gpt, "specified number of sectors %jd" 96 " is larger then the disk %jd", (uintmax_t)sector, 97 (uintmax_t)last); 98 return -1; 99 } 100 101 mbrmap = map_find(gpt, MAP_TYPE_PMBR); 102 if (mbrmap == NULL || mbrmap->map_start != 0) { 103 gpt_warnx(gpt, "No valid PMBR found"); 104 return -1; 105 } 106 mbr = mbrmap->map_data; 107 108 gpt->gpt = map_find(gpt, MAP_TYPE_PRI_GPT_HDR); 109 if (gpt == NULL) { 110 gpt_warnx(gpt, "No primary GPT header; run create or recover"); 111 return -1; 112 } 113 114 gpt->tbl = map_find(gpt, MAP_TYPE_PRI_GPT_TBL); 115 if (gpt->tbl == NULL) { 116 gpt_warnx(gpt, "No primary GPT table; Run recover"); 117 return -1; 118 } 119 120 hdr = gpt->gpt->map_data; 121 oldloc = (off_t)le64toh((uint64_t)hdr->hdr_lba_alt); 122 123 gpt->tpg = map_find(gpt, MAP_TYPE_SEC_GPT_HDR); 124 gpt->lbt = map_find(gpt, MAP_TYPE_SEC_GPT_TBL); 125 if (gpt->tpg == NULL || gpt->lbt == NULL) { 126 if (gpt_gpt(gpt, oldloc, 1) == -1) { 127 gpt_warnx(gpt, 128 "Error reading backup GPT information at %#jx", 129 oldloc); 130 return -1; 131 } 132 } 133 134 gpt->tpg = map_find(gpt, MAP_TYPE_SEC_GPT_HDR); 135 if (gpt->tpg == NULL) { 136 gpt_warnx(gpt, "No secondary GPT header; Run recover"); 137 return -1; 138 } 139 gpt->lbt = map_find(gpt, MAP_TYPE_SEC_GPT_TBL); 140 if (gpt->lbt == NULL) { 141 gpt_warnx(gpt, "No secondary GPT table; Run recover"); 142 return -1; 143 } 144 145 gpt_size = gpt->tbl->map_size; 146 if (sector == oldloc) { 147 gpt_warnx(gpt, "Device is already the specified size"); 148 return 0; 149 } 150 151 if (sector == 0 && last == oldloc) { 152 gpt_warnx(gpt, "Device hasn't changed size"); 153 return 0; 154 } 155 156 for (ent = gpt->tbl->map_data; ent < 157 (struct gpt_ent *)((char *)gpt->tbl->map_data + 158 le32toh(hdr->hdr_entries) * le32toh(hdr->hdr_entsz)); ent++) { 159 if (!gpt_uuid_is_nil(ent->ent_type) && 160 ((off_t)le64toh(ent->ent_lba_end) > lastdata)) { 161 lastdata = (off_t)le64toh((uint64_t)ent->ent_lba_end); 162 } 163 } 164 165 if (sector - gpt_size <= lastdata) { 166 gpt_warnx(gpt, "Not enough space at %" PRIu64 167 " for secondary GPT table", sector); 168 return -1; 169 } 170 171 if (last - gpt_size <= lastdata) { 172 gpt_warnx(gpt, "Not enough space for new secondary GPT table"); 173 return -1; 174 } 175 176 if (sector > oldloc) 177 newloc = sector; 178 if (sector > 0 && sector < oldloc && last >= oldloc) 179 newloc = sector; 180 if (sector == 0 && last > oldloc) 181 newloc = last; 182 183 if (newloc > 0) { 184 if (gpt->tpg == NULL) { 185 gpt_warnx(gpt, "No secondary GPT header; run recover"); 186 return -1; 187 } 188 if (gpt->lbt == NULL) { 189 gpt_warnx(gpt, "Run recover"); 190 return -1; 191 } 192 gpt->tpg->map_start = newloc; 193 gpt->lbt->map_start = newloc - gpt_size; 194 } else { 195 if (sector > 0) 196 newloc = sector; 197 else 198 newloc = last; 199 200 if (gpt_add_hdr(gpt, MAP_TYPE_SEC_GPT_HDR, newloc) == -1) 201 return -1; 202 203 gpt->lbt = map_add(gpt, newloc - gpt_size, gpt_size, 204 MAP_TYPE_SEC_GPT_TBL, gpt->tbl->map_data, 0); 205 if (gpt->lbt == NULL) { 206 gpt_warn(gpt, "Error adding secondary GPT table"); 207 return -1; 208 } 209 memcpy(gpt->tpg->map_data, gpt->gpt->map_data, gpt->secsz); 210 } 211 212 hdr = gpt->gpt->map_data; 213 hdr->hdr_lba_alt = (uint64_t)gpt->tpg->map_start; 214 hdr->hdr_crc_self = 0; 215 hdr->hdr_lba_end = htole64((uint64_t)(gpt->lbt->map_start - 1)); 216 hdr->hdr_crc_self = 217 htole32(crc32(gpt->gpt->map_data, GPT_HDR_SIZE)); 218 gpt_write(gpt, gpt->gpt); 219 220 hdr = gpt->tpg->map_data; 221 hdr->hdr_lba_self = htole64((uint64_t)gpt->tpg->map_start); 222 hdr->hdr_lba_alt = htole64((uint64_t)gpt->gpt->map_start); 223 hdr->hdr_lba_end = htole64((uint64_t)(gpt->lbt->map_start - 1)); 224 hdr->hdr_lba_table = htole64((uint64_t)gpt->lbt->map_start); 225 226 if (gpt_write_backup(gpt) == -1) 227 return -1; 228 229 for (i = 0; i < 4; i++) 230 if (mbr->mbr_part[0].part_typ == MBR_PTYPE_PMBR) 231 break; 232 if (i == 4) { 233 gpt_warnx(gpt, "No valid PMBR partition found"); 234 return -1; 235 } 236 if (last > 0xffffffff) { 237 mbr->mbr_part[0].part_size_lo = htole16(0xffff); 238 mbr->mbr_part[0].part_size_hi = htole16(0xffff); 239 } else { 240 mbr->mbr_part[0].part_size_lo = htole16((uint16_t)last); 241 mbr->mbr_part[0].part_size_hi = htole16((uint16_t)(last >> 16)); 242 } 243 if (gpt_write(gpt, mbrmap) == -1) { 244 gpt_warnx(gpt, "Error writing PMBR"); 245 return -1; 246 } 247 248 return 0; 249 } 250 251 static int 252 cmd_resizedisk(gpt_t gpt, int argc, char *argv[]) 253 { 254 int ch; 255 off_t sector, size = gpt->mediasz; 256 257 while ((ch = getopt(argc, argv, "s:")) != -1) { 258 switch(ch) { 259 case 's': 260 if (gpt_add_ais(gpt, NULL, NULL, &size, ch) == -1) 261 return -1; 262 break; 263 default: 264 return usage(); 265 } 266 } 267 268 if (argc != optind) 269 return usage(); 270 271 if ((sector = gpt_check_ais(gpt, 0, (u_int)~0, size)) == -1) 272 return -1; 273 274 if (--sector == 0) { 275 gpt_warnx(gpt, "New size %ju too small", (uintmax_t)size); 276 return -1; 277 } 278 279 return resizedisk(gpt, sector, size); 280 } 281