geom_part.c revision 221952
1/*- 2 * Copyright (c) 2007, 2008 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 * 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 AUTHORS 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 AUTHORS 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 27#include <sys/cdefs.h> 28__FBSDID("$FreeBSD: head/sbin/geom/class/part/geom_part.c 221952 2011-05-15 11:45:13Z ae $"); 29 30#include <sys/stat.h> 31#include <sys/vtoc.h> 32 33#include <assert.h> 34#include <ctype.h> 35#include <err.h> 36#include <errno.h> 37#include <fcntl.h> 38#include <libgeom.h> 39#include <libutil.h> 40#include <paths.h> 41#include <signal.h> 42#include <stdint.h> 43#include <stdio.h> 44#include <stdlib.h> 45#include <limits.h> 46#include <inttypes.h> 47#include <string.h> 48#include <strings.h> 49#include <unistd.h> 50 51#include "core/geom.h" 52#include "misc/subr.h" 53 54#ifdef STATIC_GEOM_CLASSES 55#define PUBSYM(x) gpart_##x 56#else 57#define PUBSYM(x) x 58#endif 59 60uint32_t PUBSYM(lib_version) = G_LIB_VERSION; 61uint32_t PUBSYM(version) = 0; 62 63static char sstart[32]; 64static char ssize[32]; 65volatile sig_atomic_t undo_restore; 66 67#define GPART_AUTOFILL "*" 68#define GPART_FLAGS "C" 69 70#define GPART_PARAM_BOOTCODE "bootcode" 71#define GPART_PARAM_INDEX "index" 72#define GPART_PARAM_PARTCODE "partcode" 73 74static struct gclass *find_class(struct gmesh *, const char *); 75static struct ggeom * find_geom(struct gclass *, const char *); 76static const char *find_geomcfg(struct ggeom *, const char *); 77static const char *find_provcfg(struct gprovider *, const char *); 78static struct gprovider *find_provider(struct ggeom *, off_t); 79static const char *fmtsize(int64_t); 80static int gpart_autofill(struct gctl_req *); 81static int gpart_autofill_resize(struct gctl_req *); 82static void gpart_bootcode(struct gctl_req *, unsigned int); 83static void *gpart_bootfile_read(const char *, ssize_t *); 84static void gpart_issue(struct gctl_req *, unsigned int); 85static void gpart_show(struct gctl_req *, unsigned int); 86static void gpart_show_geom(struct ggeom *, const char *, int); 87static int gpart_show_hasopt(struct gctl_req *, const char *, const char *); 88static void gpart_write_partcode(struct ggeom *, int, void *, ssize_t); 89static void gpart_write_partcode_vtoc8(struct ggeom *, int, void *); 90static void gpart_print_error(const char *); 91static void gpart_backup(struct gctl_req *, unsigned int); 92static void gpart_restore(struct gctl_req *, unsigned int); 93 94struct g_command PUBSYM(class_commands)[] = { 95 { "add", 0, gpart_issue, { 96 { 'a', "alignment", GPART_AUTOFILL, G_TYPE_STRING }, 97 { 'b', "start", GPART_AUTOFILL, G_TYPE_STRING }, 98 { 's', "size", GPART_AUTOFILL, G_TYPE_STRING }, 99 { 't', "type", NULL, G_TYPE_STRING }, 100 { 'i', GPART_PARAM_INDEX, G_VAL_OPTIONAL, G_TYPE_NUMBER }, 101 { 'l', "label", G_VAL_OPTIONAL, G_TYPE_STRING }, 102 { 'f', "flags", GPART_FLAGS, G_TYPE_STRING }, 103 G_OPT_SENTINEL }, 104 "[-a alignment] [-b start] [-s size] -t type [-i index] " 105 "[-l label] [-f flags] geom" 106 }, 107 { "backup", 0, gpart_backup, G_NULL_OPTS, 108 "geom" 109 }, 110 { "bootcode", 0, gpart_bootcode, { 111 { 'b', GPART_PARAM_BOOTCODE, G_VAL_OPTIONAL, G_TYPE_STRING }, 112 { 'p', GPART_PARAM_PARTCODE, G_VAL_OPTIONAL, G_TYPE_STRING }, 113 { 'i', GPART_PARAM_INDEX, G_VAL_OPTIONAL, G_TYPE_NUMBER }, 114 { 'f', "flags", GPART_FLAGS, G_TYPE_STRING }, 115 G_OPT_SENTINEL }, 116 "[-b bootcode] [-p partcode] [-i index] [-f flags] geom" 117 }, 118 { "commit", 0, gpart_issue, G_NULL_OPTS, 119 "geom" 120 }, 121 { "create", 0, gpart_issue, { 122 { 's', "scheme", NULL, G_TYPE_STRING }, 123 { 'n', "entries", G_VAL_OPTIONAL, G_TYPE_NUMBER }, 124 { 'f', "flags", GPART_FLAGS, G_TYPE_STRING }, 125 G_OPT_SENTINEL }, 126 "-s scheme [-n entries] [-f flags] provider" 127 }, 128 { "delete", 0, gpart_issue, { 129 { 'i', GPART_PARAM_INDEX, NULL, G_TYPE_NUMBER }, 130 { 'f', "flags", GPART_FLAGS, G_TYPE_STRING }, 131 G_OPT_SENTINEL }, 132 "-i index [-f flags] geom" 133 }, 134 { "destroy", 0, gpart_issue, { 135 { 'F', "force", NULL, G_TYPE_BOOL }, 136 { 'f', "flags", GPART_FLAGS, G_TYPE_STRING }, 137 G_OPT_SENTINEL }, 138 "[-F] [-f flags] geom" 139 }, 140 { "modify", 0, gpart_issue, { 141 { 'i', GPART_PARAM_INDEX, NULL, G_TYPE_NUMBER }, 142 { 'l', "label", G_VAL_OPTIONAL, G_TYPE_STRING }, 143 { 't', "type", G_VAL_OPTIONAL, G_TYPE_STRING }, 144 { 'f', "flags", GPART_FLAGS, G_TYPE_STRING }, 145 G_OPT_SENTINEL }, 146 "-i index [-l label] [-t type] [-f flags] geom" 147 }, 148 { "set", 0, gpart_issue, { 149 { 'a', "attrib", NULL, G_TYPE_STRING }, 150 { 'i', GPART_PARAM_INDEX, NULL, G_TYPE_NUMBER }, 151 { 'f', "flags", GPART_FLAGS, G_TYPE_STRING }, 152 G_OPT_SENTINEL }, 153 "-a attrib -i index [-f flags] geom" 154 }, 155 { "show", 0, gpart_show, { 156 { 'l', "show_label", NULL, G_TYPE_BOOL }, 157 { 'r', "show_rawtype", NULL, G_TYPE_BOOL }, 158 { 'p', "show_providers", NULL, G_TYPE_BOOL }, 159 G_OPT_SENTINEL }, 160 "[-lrp] [geom ...]" 161 }, 162 { "undo", 0, gpart_issue, G_NULL_OPTS, 163 "geom" 164 }, 165 { "unset", 0, gpart_issue, { 166 { 'a', "attrib", NULL, G_TYPE_STRING }, 167 { 'i', GPART_PARAM_INDEX, NULL, G_TYPE_NUMBER }, 168 { 'f', "flags", GPART_FLAGS, G_TYPE_STRING }, 169 G_OPT_SENTINEL }, 170 "-a attrib -i index [-f flags] geom" 171 }, 172 { "resize", 0, gpart_issue, { 173 { 'a', "alignment", GPART_AUTOFILL, G_TYPE_STRING }, 174 { 's', "size", GPART_AUTOFILL, G_TYPE_STRING }, 175 { 'i', GPART_PARAM_INDEX, NULL, G_TYPE_NUMBER }, 176 { 'f', "flags", GPART_FLAGS, G_TYPE_STRING }, 177 G_OPT_SENTINEL }, 178 "[-a alignment] [-s size] -i index [-f flags] geom" 179 }, 180 { "restore", 0, gpart_restore, { 181 { 'F', "force", NULL, G_TYPE_BOOL }, 182 { 'l', "restore_labels", NULL, G_TYPE_BOOL }, 183 { 'f', "flags", GPART_FLAGS, G_TYPE_STRING }, 184 G_OPT_SENTINEL }, 185 "[-lF] [-f flags] provider [...]" 186 }, 187 { "recover", 0, gpart_issue, { 188 { 'f', "flags", GPART_FLAGS, G_TYPE_STRING }, 189 G_OPT_SENTINEL }, 190 "[-f flags] geom" 191 }, 192 G_CMD_SENTINEL 193}; 194 195static struct gclass * 196find_class(struct gmesh *mesh, const char *name) 197{ 198 struct gclass *classp; 199 200 LIST_FOREACH(classp, &mesh->lg_class, lg_class) { 201 if (strcmp(classp->lg_name, name) == 0) 202 return (classp); 203 } 204 return (NULL); 205} 206 207static struct ggeom * 208find_geom(struct gclass *classp, const char *name) 209{ 210 struct ggeom *gp; 211 212 if (strncmp(name, _PATH_DEV, sizeof(_PATH_DEV) - 1) == 0) 213 name += sizeof(_PATH_DEV) - 1; 214 LIST_FOREACH(gp, &classp->lg_geom, lg_geom) { 215 if (strcmp(gp->lg_name, name) == 0) 216 return (gp); 217 } 218 return (NULL); 219} 220 221static const char * 222find_geomcfg(struct ggeom *gp, const char *cfg) 223{ 224 struct gconfig *gc; 225 226 LIST_FOREACH(gc, &gp->lg_config, lg_config) { 227 if (!strcmp(gc->lg_name, cfg)) 228 return (gc->lg_val); 229 } 230 return (NULL); 231} 232 233static const char * 234find_provcfg(struct gprovider *pp, const char *cfg) 235{ 236 struct gconfig *gc; 237 238 LIST_FOREACH(gc, &pp->lg_config, lg_config) { 239 if (!strcmp(gc->lg_name, cfg)) 240 return (gc->lg_val); 241 } 242 return (NULL); 243} 244 245static struct gprovider * 246find_provider(struct ggeom *gp, off_t minsector) 247{ 248 struct gprovider *pp, *bestpp; 249 const char *s; 250 off_t sector, bestsector; 251 252 bestpp = NULL; 253 bestsector = 0; 254 LIST_FOREACH(pp, &gp->lg_provider, lg_provider) { 255 s = find_provcfg(pp, "start"); 256 sector = (off_t)strtoimax(s, NULL, 0); 257 if (sector < minsector) 258 continue; 259 if (bestpp != NULL && sector >= bestsector) 260 continue; 261 262 bestpp = pp; 263 bestsector = sector; 264 } 265 return (bestpp); 266} 267 268static const char * 269fmtsize(int64_t rawsz) 270{ 271 static char buf[5]; 272 273 humanize_number(buf, sizeof(buf), rawsz, "", HN_AUTOSCALE, 274 HN_B | HN_NOSPACE | HN_DECIMAL); 275 return (buf); 276} 277 278static const char * 279fmtattrib(struct gprovider *pp) 280{ 281 static char buf[128]; 282 struct gconfig *gc; 283 u_int idx; 284 285 buf[0] = '\0'; 286 idx = 0; 287 LIST_FOREACH(gc, &pp->lg_config, lg_config) { 288 if (strcmp(gc->lg_name, "attrib") != 0) 289 continue; 290 idx += snprintf(buf + idx, sizeof(buf) - idx, "%s%s", 291 (idx == 0) ? " [" : ",", gc->lg_val); 292 } 293 if (idx > 0) 294 snprintf(buf + idx, sizeof(buf) - idx, "] "); 295 return (buf); 296} 297 298#define ALIGNDOWN(d, a) (-(a) & (d)) 299#define ALIGNUP(d, a) (-(-(a) & -(d))) 300 301static int 302gpart_autofill_resize(struct gctl_req *req) 303{ 304 struct gmesh mesh; 305 struct gclass *cp; 306 struct ggeom *gp; 307 struct gprovider *pp; 308 off_t last, size, start, new_size; 309 off_t lba, new_lba, alignment; 310 const char *s; 311 int error, idx; 312 313 idx = (int)gctl_get_intmax(req, GPART_PARAM_INDEX); 314 if (idx < 1) 315 errx(EXIT_FAILURE, "invalid partition index"); 316 317 error = geom_gettree(&mesh); 318 if (error) 319 return (error); 320 s = gctl_get_ascii(req, "class"); 321 if (s == NULL) 322 abort(); 323 cp = find_class(&mesh, s); 324 if (cp == NULL) 325 errx(EXIT_FAILURE, "Class %s not found.", s); 326 s = gctl_get_ascii(req, "arg0"); 327 if (s == NULL) 328 abort(); 329 gp = find_geom(cp, s); 330 if (gp == NULL) 331 errx(EXIT_FAILURE, "No such geom: %s.", s); 332 pp = LIST_FIRST(&gp->lg_consumer)->lg_provider; 333 if (pp == NULL) 334 errx(EXIT_FAILURE, "Provider for geom %s not found.", s); 335 336 s = gctl_get_ascii(req, "alignment"); 337 alignment = 1; 338 if (*s != '*') { 339 error = g_parse_lba(s, pp->lg_sectorsize, &alignment); 340 if (error) 341 errc(EXIT_FAILURE, error, "Invalid alignment param"); 342 if (alignment == 0) 343 errx(EXIT_FAILURE, "Invalid alignment param"); 344 } 345 error = gctl_delete_param(req, "alignment"); 346 if (error) 347 errc(EXIT_FAILURE, error, "internal error"); 348 349 s = gctl_get_ascii(req, "size"); 350 if (*s == '*') 351 new_size = 0; 352 else { 353 error = g_parse_lba(s, pp->lg_sectorsize, &new_size); 354 if (error) 355 errc(EXIT_FAILURE, error, "Invalid size param"); 356 /* no autofill necessary. */ 357 if (alignment == 1) 358 goto done; 359 if (new_size > alignment) 360 new_size = ALIGNDOWN(new_size, alignment); 361 } 362 363 last = (off_t)strtoimax(find_geomcfg(gp, "last"), NULL, 0); 364 last = ALIGNDOWN(last, alignment); 365 LIST_FOREACH(pp, &gp->lg_provider, lg_provider) { 366 s = find_provcfg(pp, "index"); 367 if (s == NULL) 368 continue; 369 if (atoi(s) == idx) 370 break; 371 } 372 if (pp == NULL) 373 errx(EXIT_FAILURE, "invalid partition index"); 374 375 s = find_provcfg(pp, "start"); 376 start = (off_t)strtoimax(s, NULL, 0); 377 s = find_provcfg(pp, "end"); 378 lba = (off_t)strtoimax(s, NULL, 0) + 1; 379 380 if (lba > last) { 381 geom_deletetree(&mesh); 382 return (ENOSPC); 383 } 384 size = lba - start; 385 pp = find_provider(gp, lba); 386 if (pp == NULL) 387 new_size = ALIGNDOWN(last - start + 1, alignment); 388 else { 389 s = find_provcfg(pp, "start"); 390 new_lba = (off_t)strtoimax(s, NULL, 0); 391 /* 392 * Is there any free space between current and 393 * next providers? 394 */ 395 new_lba = ALIGNUP(new_lba, alignment); 396 if (new_lba > lba) 397 new_size = new_lba - start; 398 else { 399 geom_deletetree(&mesh); 400 return (ENOSPC); 401 } 402 } 403done: 404 snprintf(ssize, sizeof(ssize), "%jd", (intmax_t)new_size); 405 gctl_change_param(req, "size", -1, ssize); 406 geom_deletetree(&mesh); 407 return (0); 408} 409 410static int 411gpart_autofill(struct gctl_req *req) 412{ 413 struct gmesh mesh; 414 struct gclass *cp; 415 struct ggeom *gp; 416 struct gprovider *pp; 417 off_t first, last, a_first; 418 off_t size, start, a_lba; 419 off_t lba, len, alignment; 420 uintmax_t grade; 421 const char *s; 422 int error, has_size, has_start, has_alignment; 423 424 s = gctl_get_ascii(req, "verb"); 425 if (strcmp(s, "resize") == 0) 426 return gpart_autofill_resize(req); 427 if (strcmp(s, "add") != 0) 428 return (0); 429 430 error = geom_gettree(&mesh); 431 if (error) 432 return (error); 433 s = gctl_get_ascii(req, "class"); 434 if (s == NULL) 435 abort(); 436 cp = find_class(&mesh, s); 437 if (cp == NULL) 438 errx(EXIT_FAILURE, "Class %s not found.", s); 439 s = gctl_get_ascii(req, "arg0"); 440 if (s == NULL) 441 abort(); 442 gp = find_geom(cp, s); 443 if (gp == NULL) 444 errx(EXIT_FAILURE, "No such geom: %s.", s); 445 pp = LIST_FIRST(&gp->lg_consumer)->lg_provider; 446 if (pp == NULL) 447 errx(EXIT_FAILURE, "Provider for geom %s not found.", s); 448 449 s = gctl_get_ascii(req, "alignment"); 450 has_alignment = (*s == '*') ? 0 : 1; 451 alignment = 1; 452 if (has_alignment) { 453 error = g_parse_lba(s, pp->lg_sectorsize, &alignment); 454 if (error) 455 errc(EXIT_FAILURE, error, "Invalid alignment param"); 456 if (alignment == 0) 457 errx(EXIT_FAILURE, "Invalid alignment param"); 458 } 459 error = gctl_delete_param(req, "alignment"); 460 if (error) 461 errc(EXIT_FAILURE, error, "internal error"); 462 463 s = gctl_get_ascii(req, "size"); 464 has_size = (*s == '*') ? 0 : 1; 465 size = 0; 466 if (has_size) { 467 error = g_parse_lba(s, pp->lg_sectorsize, &size); 468 if (error) 469 errc(EXIT_FAILURE, error, "Invalid size param"); 470 if (size > alignment) 471 size = ALIGNDOWN(size, alignment); 472 } 473 474 s = gctl_get_ascii(req, "start"); 475 has_start = (*s == '*') ? 0 : 1; 476 start = 0ULL; 477 if (has_start) { 478 error = g_parse_lba(s, pp->lg_sectorsize, &start); 479 if (error) 480 errc(EXIT_FAILURE, error, "Invalid start param"); 481 start = ALIGNUP(start, alignment); 482 } 483 484 /* No autofill necessary. */ 485 if (has_size && has_start && !has_alignment) 486 goto done; 487 488 first = (off_t)strtoimax(find_geomcfg(gp, "first"), NULL, 0); 489 last = (off_t)strtoimax(find_geomcfg(gp, "last"), NULL, 0); 490 grade = ~0ULL; 491 a_first = ALIGNUP(first, alignment); 492 last = ALIGNDOWN(last, alignment); 493 while ((pp = find_provider(gp, first)) != NULL) { 494 s = find_provcfg(pp, "start"); 495 lba = (off_t)strtoimax(s, NULL, 0); 496 a_lba = ALIGNDOWN(lba, alignment); 497 if (first < a_lba && a_first < a_lba) { 498 /* Free space [first, lba> */ 499 len = a_lba - a_first; 500 if (has_size) { 501 if (len >= size && 502 (uintmax_t)(len - size) < grade) { 503 start = a_first; 504 grade = len - size; 505 } 506 } else if (has_start) { 507 if (start >= a_first && start < a_lba) { 508 size = a_lba - start; 509 grade = start - a_first; 510 } 511 } else { 512 if (grade == ~0ULL || len > size) { 513 start = a_first; 514 size = len; 515 grade = 0; 516 } 517 } 518 } 519 520 s = find_provcfg(pp, "end"); 521 first = (off_t)strtoimax(s, NULL, 0) + 1; 522 a_first = ALIGNUP(first, alignment); 523 } 524 if (a_first <= last) { 525 /* Free space [first-last] */ 526 len = ALIGNDOWN(last - a_first + 1, alignment); 527 if (has_size) { 528 if (len >= size && 529 (uintmax_t)(len - size) < grade) { 530 start = a_first; 531 grade = len - size; 532 } 533 } else if (has_start) { 534 if (start >= a_first && start <= last) { 535 size = ALIGNDOWN(last - start + 1, alignment); 536 grade = start - a_first; 537 } 538 } else { 539 if (grade == ~0ULL || len > size) { 540 start = a_first; 541 size = len; 542 grade = 0; 543 } 544 } 545 } 546 547 if (grade == ~0ULL) { 548 geom_deletetree(&mesh); 549 return (ENOSPC); 550 } 551 552done: 553 snprintf(ssize, sizeof(ssize), "%jd", (intmax_t)size); 554 gctl_change_param(req, "size", -1, ssize); 555 snprintf(sstart, sizeof(sstart), "%jd", (intmax_t)start); 556 gctl_change_param(req, "start", -1, sstart); 557 geom_deletetree(&mesh); 558 return (0); 559} 560 561static void 562gpart_show_geom(struct ggeom *gp, const char *element, int show_providers) 563{ 564 struct gprovider *pp; 565 const char *s, *scheme; 566 off_t first, last, sector, end; 567 off_t length, secsz; 568 int idx, wblocks, wname, wmax; 569 570 scheme = find_geomcfg(gp, "scheme"); 571 s = find_geomcfg(gp, "first"); 572 first = (off_t)strtoimax(s, NULL, 0); 573 s = find_geomcfg(gp, "last"); 574 last = (off_t)strtoimax(s, NULL, 0); 575 wblocks = strlen(s); 576 s = find_geomcfg(gp, "state"); 577 if (s != NULL && *s != 'C') 578 s = NULL; 579 wmax = strlen(gp->lg_name); 580 if (show_providers) { 581 LIST_FOREACH(pp, &gp->lg_provider, lg_provider) { 582 wname = strlen(pp->lg_name); 583 if (wname > wmax) 584 wmax = wname; 585 } 586 } 587 wname = wmax; 588 pp = LIST_FIRST(&gp->lg_consumer)->lg_provider; 589 secsz = pp->lg_sectorsize; 590 printf("=>%*jd %*jd %*s %s (%s)%s\n", 591 wblocks, (intmax_t)first, wblocks, (intmax_t)(last - first + 1), 592 wname, gp->lg_name, 593 scheme, fmtsize(pp->lg_mediasize), 594 s ? " [CORRUPT]": ""); 595 596 while ((pp = find_provider(gp, first)) != NULL) { 597 s = find_provcfg(pp, "start"); 598 sector = (off_t)strtoimax(s, NULL, 0); 599 600 s = find_provcfg(pp, "end"); 601 end = (off_t)strtoimax(s, NULL, 0); 602 length = end - sector + 1; 603 604 s = find_provcfg(pp, "index"); 605 idx = atoi(s); 606 if (first < sector) { 607 printf(" %*jd %*jd %*s - free - (%s)\n", 608 wblocks, (intmax_t)first, wblocks, 609 (intmax_t)(sector - first), wname, "", 610 fmtsize((sector - first) * secsz)); 611 } 612 if (show_providers) { 613 printf(" %*jd %*jd %*s %s %s (%s)\n", 614 wblocks, (intmax_t)sector, wblocks, 615 (intmax_t)length, wname, pp->lg_name, 616 find_provcfg(pp, element), fmtattrib(pp), 617 fmtsize(pp->lg_mediasize)); 618 } else 619 printf(" %*jd %*jd %*d %s %s (%s)\n", 620 wblocks, (intmax_t)sector, wblocks, 621 (intmax_t)length, wname, idx, 622 find_provcfg(pp, element), fmtattrib(pp), 623 fmtsize(pp->lg_mediasize)); 624 first = end + 1; 625 } 626 if (first <= last) { 627 length = last - first + 1; 628 printf(" %*jd %*jd %*s - free - (%s)\n", 629 wblocks, (intmax_t)first, wblocks, (intmax_t)length, 630 wname, "", 631 fmtsize(length * secsz)); 632 } 633 printf("\n"); 634} 635 636static int 637gpart_show_hasopt(struct gctl_req *req, const char *opt, const char *elt) 638{ 639 640 if (!gctl_get_int(req, "%s", opt)) 641 return (0); 642 643 if (elt != NULL) 644 errx(EXIT_FAILURE, "-l and -r are mutually exclusive"); 645 646 return (1); 647} 648 649static void 650gpart_show(struct gctl_req *req, unsigned int fl __unused) 651{ 652 struct gmesh mesh; 653 struct gclass *classp; 654 struct ggeom *gp; 655 const char *element, *name; 656 int error, i, nargs, show_providers; 657 658 element = NULL; 659 if (gpart_show_hasopt(req, "show_label", element)) 660 element = "label"; 661 if (gpart_show_hasopt(req, "show_rawtype", element)) 662 element = "rawtype"; 663 if (element == NULL) 664 element = "type"; 665 666 name = gctl_get_ascii(req, "class"); 667 if (name == NULL) 668 abort(); 669 error = geom_gettree(&mesh); 670 if (error != 0) 671 errc(EXIT_FAILURE, error, "Cannot get GEOM tree"); 672 classp = find_class(&mesh, name); 673 if (classp == NULL) { 674 geom_deletetree(&mesh); 675 errx(EXIT_FAILURE, "Class %s not found.", name); 676 } 677 show_providers = gctl_get_int(req, "show_providers"); 678 nargs = gctl_get_int(req, "nargs"); 679 if (nargs > 0) { 680 for (i = 0; i < nargs; i++) { 681 name = gctl_get_ascii(req, "arg%d", i); 682 gp = find_geom(classp, name); 683 if (gp != NULL) 684 gpart_show_geom(gp, element, show_providers); 685 else 686 errx(EXIT_FAILURE, "No such geom: %s.", name); 687 } 688 } else { 689 LIST_FOREACH(gp, &classp->lg_geom, lg_geom) { 690 gpart_show_geom(gp, element, show_providers); 691 } 692 } 693 geom_deletetree(&mesh); 694} 695 696static void 697gpart_backup(struct gctl_req *req, unsigned int fl __unused) 698{ 699 struct gmesh mesh; 700 struct gclass *classp; 701 struct gprovider *pp; 702 struct ggeom *gp; 703 const char *s, *scheme; 704 off_t sector, end; 705 off_t length, secsz; 706 int error, i, windex, wblocks, wtype; 707 708 if (gctl_get_int(req, "nargs") != 1) 709 errx(EXIT_FAILURE, "Invalid number of arguments."); 710 error = geom_gettree(&mesh); 711 if (error != 0) 712 errc(EXIT_FAILURE, error, "Cannot get GEOM tree"); 713 s = gctl_get_ascii(req, "class"); 714 if (s == NULL) 715 abort(); 716 classp = find_class(&mesh, s); 717 if (classp == NULL) { 718 geom_deletetree(&mesh); 719 errx(EXIT_FAILURE, "Class %s not found.", s); 720 } 721 s = gctl_get_ascii(req, "arg0"); 722 if (s == NULL) 723 abort(); 724 gp = find_geom(classp, s); 725 if (gp == NULL) 726 errx(EXIT_FAILURE, "No such geom: %s.", s); 727 scheme = find_geomcfg(gp, "scheme"); 728 if (scheme == NULL) 729 abort(); 730 pp = LIST_FIRST(&gp->lg_consumer)->lg_provider; 731 secsz = pp->lg_sectorsize; 732 s = find_geomcfg(gp, "last"); 733 wblocks = strlen(s); 734 wtype = 0; 735 LIST_FOREACH(pp, &gp->lg_provider, lg_provider) { 736 s = find_provcfg(pp, "type"); 737 i = strlen(s); 738 if (i > wtype) 739 wtype = i; 740 } 741 s = find_geomcfg(gp, "entries"); 742 windex = strlen(s); 743 printf("%s %s\n", scheme, s); 744 LIST_FOREACH(pp, &gp->lg_provider, lg_provider) { 745 s = find_provcfg(pp, "start"); 746 sector = (off_t)strtoimax(s, NULL, 0); 747 748 s = find_provcfg(pp, "end"); 749 end = (off_t)strtoimax(s, NULL, 0); 750 length = end - sector + 1; 751 752 s = find_provcfg(pp, "label"); 753 printf("%-*s %*s %*jd %*jd %s %s\n", 754 windex, find_provcfg(pp, "index"), 755 wtype, find_provcfg(pp, "type"), 756 wblocks, (intmax_t)sector, 757 wblocks, (intmax_t)length, 758 (s != NULL) ? s: "", fmtattrib(pp)); 759 } 760 geom_deletetree(&mesh); 761} 762 763static int 764skip_line(const char *p) 765{ 766 767 while (*p != '\0') { 768 if (*p == '#') 769 return (1); 770 if (isspace(*p) == 0) 771 return (0); 772 p++; 773 } 774 return (1); 775} 776 777static void 778gpart_sighndl(int sig __unused) 779{ 780 undo_restore = 1; 781} 782 783static void 784gpart_restore(struct gctl_req *req, unsigned int fl __unused) 785{ 786 struct gmesh mesh; 787 struct gclass *classp; 788 struct gctl_req *r; 789 struct ggeom *gp; 790 struct sigaction si_sa; 791 const char *s, *flags, *errstr, *label; 792 char **ap, *argv[6], line[BUFSIZ], *pline; 793 int error, forced, i, l, nargs, created, rl; 794 intmax_t n; 795 796 nargs = gctl_get_int(req, "nargs"); 797 if (nargs < 1) 798 errx(EXIT_FAILURE, "Invalid number of arguments."); 799 800 forced = gctl_get_int(req, "force"); 801 flags = gctl_get_ascii(req, "flags"); 802 rl = gctl_get_int(req, "restore_labels"); 803 s = gctl_get_ascii(req, "class"); 804 if (s == NULL) 805 abort(); 806 error = geom_gettree(&mesh); 807 if (error != 0) 808 errc(EXIT_FAILURE, error, "Cannot get GEOM tree"); 809 classp = find_class(&mesh, s); 810 if (classp == NULL) { 811 geom_deletetree(&mesh); 812 errx(EXIT_FAILURE, "Class %s not found.", s); 813 } 814 815 sigemptyset(&si_sa.sa_mask); 816 si_sa.sa_flags = 0; 817 si_sa.sa_handler = gpart_sighndl; 818 if (sigaction(SIGINT, &si_sa, 0) == -1) 819 err(EXIT_FAILURE, "sigaction SIGINT"); 820 821 if (forced) { 822 /* destroy existent partition table before restore */ 823 for (i = 0; i < nargs; i++) { 824 s = gctl_get_ascii(req, "arg%d", i); 825 gp = find_geom(classp, s); 826 if (gp != NULL) { 827 r = gctl_get_handle(); 828 gctl_ro_param(r, "class", -1, 829 classp->lg_name); 830 gctl_ro_param(r, "verb", -1, "destroy"); 831 gctl_ro_param(r, "flags", -1, "restore"); 832 gctl_ro_param(r, "force", sizeof(forced), 833 &forced); 834 gctl_ro_param(r, "arg0", -1, s); 835 errstr = gctl_issue(r); 836 if (errstr != NULL && errstr[0] != '\0') { 837 gpart_print_error(errstr); 838 gctl_free(r); 839 goto backout; 840 } 841 gctl_free(r); 842 } 843 } 844 } 845 created = 0; 846 while (undo_restore == 0 && 847 fgets(line, sizeof(line) - 1, stdin) != NULL) { 848 /* Format of backup entries: 849 * <scheme name> <number of entries> 850 * <index> <type> <start> <size> [label] ['['attrib[,attrib]']'] 851 */ 852 pline = (char *)line; 853 pline[strlen(line) - 1] = 0; 854 if (skip_line(pline)) 855 continue; 856 for (ap = argv; 857 (*ap = strsep(&pline, " \t")) != NULL;) 858 if (**ap != '\0' && ++ap >= &argv[6]) 859 break; 860 l = ap - &argv[0]; 861 label = pline = NULL; 862 if (l == 1 || l == 2) { /* create table */ 863 if (created) 864 errx(EXIT_FAILURE, "Incorrect backup format."); 865 if (l == 2) 866 n = strtoimax(argv[1], NULL, 0); 867 for (i = 0; i < nargs; i++) { 868 s = gctl_get_ascii(req, "arg%d", i); 869 r = gctl_get_handle(); 870 gctl_ro_param(r, "class", -1, 871 classp->lg_name); 872 gctl_ro_param(r, "verb", -1, "create"); 873 gctl_ro_param(r, "scheme", -1, argv[0]); 874 if (l == 2) 875 gctl_ro_param(r, "entries", 876 sizeof(n), &n); 877 gctl_ro_param(r, "flags", -1, "restore"); 878 gctl_ro_param(r, "arg0", -1, s); 879 errstr = gctl_issue(r); 880 if (errstr != NULL && errstr[0] != '\0') { 881 gpart_print_error(errstr); 882 gctl_free(r); 883 goto backout; 884 } 885 gctl_free(r); 886 } 887 created = 1; 888 continue; 889 } else if (l < 4 || created == 0) 890 errx(EXIT_FAILURE, "Incorrect backup format."); 891 else if (l == 5) { 892 if (strchr(argv[4], '[') == NULL) 893 label = argv[4]; 894 else 895 pline = argv[4]; 896 } else if (l == 6) { 897 label = argv[4]; 898 pline = argv[5]; 899 } 900 /* Add partitions to each table */ 901 for (i = 0; i < nargs; i++) { 902 s = gctl_get_ascii(req, "arg%d", i); 903 r = gctl_get_handle(); 904 n = strtoimax(argv[0], NULL, 0); 905 gctl_ro_param(r, "class", -1, classp->lg_name); 906 gctl_ro_param(r, "verb", -1, "add"); 907 gctl_ro_param(r, "flags", -1, "restore"); 908 gctl_ro_param(r, GPART_PARAM_INDEX, sizeof(n), &n); 909 gctl_ro_param(r, "type", -1, argv[1]); 910 gctl_ro_param(r, "start", -1, argv[2]); 911 gctl_ro_param(r, "size", -1, argv[3]); 912 if (rl != 0 && label != NULL) 913 gctl_ro_param(r, "label", -1, argv[4]); 914 gctl_ro_param(r, "arg0", -1, s); 915 error = gpart_autofill(r); 916 if (error != 0) 917 errc(EXIT_FAILURE, error, "autofill"); 918 errstr = gctl_issue(r); 919 if (errstr != NULL && errstr[0] != '\0') { 920 gpart_print_error(errstr); 921 gctl_free(r); 922 goto backout; 923 } 924 gctl_free(r); 925 } 926 if (pline == NULL || *pline != '[') 927 continue; 928 /* set attributes */ 929 pline++; 930 for (ap = argv; 931 (*ap = strsep(&pline, ",]")) != NULL;) 932 if (**ap != '\0' && ++ap >= &argv[6]) 933 break; 934 for (i = 0; i < nargs; i++) { 935 l = ap - &argv[0]; 936 s = gctl_get_ascii(req, "arg%d", i); 937 while (l > 0) { 938 r = gctl_get_handle(); 939 gctl_ro_param(r, "class", -1, classp->lg_name); 940 gctl_ro_param(r, "verb", -1, "set"); 941 gctl_ro_param(r, "flags", -1, "restore"); 942 gctl_ro_param(r, GPART_PARAM_INDEX, 943 sizeof(n), &n); 944 gctl_ro_param(r, "attrib", -1, argv[--l]); 945 gctl_ro_param(r, "arg0", -1, s); 946 errstr = gctl_issue(r); 947 if (errstr != NULL && errstr[0] != '\0') { 948 gpart_print_error(errstr); 949 gctl_free(r); 950 goto backout; 951 } 952 gctl_free(r); 953 } 954 } 955 } 956 if (undo_restore) 957 goto backout; 958 /* commit changes if needed */ 959 if (strchr(flags, 'C') != NULL) { 960 for (i = 0; i < nargs; i++) { 961 s = gctl_get_ascii(req, "arg%d", i); 962 r = gctl_get_handle(); 963 gctl_ro_param(r, "class", -1, classp->lg_name); 964 gctl_ro_param(r, "verb", -1, "commit"); 965 gctl_ro_param(r, "arg0", -1, s); 966 errstr = gctl_issue(r); 967 if (errstr != NULL && errstr[0] != '\0') { 968 gpart_print_error(errstr); 969 gctl_free(r); 970 goto backout; 971 } 972 gctl_free(r); 973 } 974 } 975 gctl_free(req); 976 geom_deletetree(&mesh); 977 exit(EXIT_SUCCESS); 978 979backout: 980 for (i = 0; i < nargs; i++) { 981 s = gctl_get_ascii(req, "arg%d", i); 982 r = gctl_get_handle(); 983 gctl_ro_param(r, "class", -1, classp->lg_name); 984 gctl_ro_param(r, "verb", -1, "undo"); 985 gctl_ro_param(r, "arg0", -1, s); 986 gctl_issue(r); 987 gctl_free(r); 988 } 989 gctl_free(req); 990 geom_deletetree(&mesh); 991 exit(EXIT_FAILURE); 992} 993 994static void * 995gpart_bootfile_read(const char *bootfile, ssize_t *size) 996{ 997 struct stat sb; 998 void *code; 999 int fd; 1000 1001 if (stat(bootfile, &sb) == -1) 1002 err(EXIT_FAILURE, "%s", bootfile); 1003 if (!S_ISREG(sb.st_mode)) 1004 errx(EXIT_FAILURE, "%s: not a regular file", bootfile); 1005 if (sb.st_size == 0) 1006 errx(EXIT_FAILURE, "%s: empty file", bootfile); 1007 if (*size > 0 && sb.st_size > *size) 1008 errx(EXIT_FAILURE, "%s: file too big (%zu limit)", bootfile, 1009 *size); 1010 1011 *size = sb.st_size; 1012 1013 fd = open(bootfile, O_RDONLY); 1014 if (fd == -1) 1015 err(EXIT_FAILURE, "%s", bootfile); 1016 code = malloc(*size); 1017 if (code == NULL) 1018 err(EXIT_FAILURE, NULL); 1019 if (read(fd, code, *size) != *size) 1020 err(EXIT_FAILURE, "%s", bootfile); 1021 close(fd); 1022 1023 return (code); 1024} 1025 1026static void 1027gpart_write_partcode(struct ggeom *gp, int idx, void *code, ssize_t size) 1028{ 1029 char dsf[128]; 1030 struct gprovider *pp; 1031 const char *s; 1032 char *buf; 1033 off_t bsize; 1034 int fd; 1035 1036 LIST_FOREACH(pp, &gp->lg_provider, lg_provider) { 1037 s = find_provcfg(pp, "index"); 1038 if (s == NULL) 1039 continue; 1040 if (atoi(s) == idx) 1041 break; 1042 } 1043 1044 if (pp != NULL) { 1045 snprintf(dsf, sizeof(dsf), "/dev/%s", pp->lg_name); 1046 fd = open(dsf, O_WRONLY); 1047 if (fd == -1) 1048 err(EXIT_FAILURE, "%s", dsf); 1049 if (lseek(fd, size, SEEK_SET) != size) 1050 errx(EXIT_FAILURE, "%s: not enough space", dsf); 1051 if (lseek(fd, 0, SEEK_SET) != 0) 1052 err(EXIT_FAILURE, "%s", dsf); 1053 1054 /* 1055 * When writing to a disk device, the write must be 1056 * sector aligned and not write to any partial sectors, 1057 * so round up the buffer size to the next sector and zero it. 1058 */ 1059 bsize = (size + pp->lg_sectorsize - 1) / 1060 pp->lg_sectorsize * pp->lg_sectorsize; 1061 buf = calloc(1, bsize); 1062 if (buf == NULL) 1063 err(EXIT_FAILURE, "%s", dsf); 1064 bcopy(code, buf, size); 1065 if (write(fd, buf, bsize) != bsize) 1066 err(EXIT_FAILURE, "%s", dsf); 1067 free(buf); 1068 close(fd); 1069 } else 1070 errx(EXIT_FAILURE, "invalid partition index"); 1071} 1072 1073static void 1074gpart_write_partcode_vtoc8(struct ggeom *gp, int idx, void *code) 1075{ 1076 char dsf[128]; 1077 struct gprovider *pp; 1078 const char *s; 1079 int installed, fd; 1080 1081 installed = 0; 1082 LIST_FOREACH(pp, &gp->lg_provider, lg_provider) { 1083 s = find_provcfg(pp, "index"); 1084 if (s == NULL) 1085 continue; 1086 if (idx != 0 && atoi(s) != idx) 1087 continue; 1088 snprintf(dsf, sizeof(dsf), "/dev/%s", pp->lg_name); 1089 if (pp->lg_sectorsize != sizeof(struct vtoc8)) 1090 errx(EXIT_FAILURE, "%s: unexpected sector " 1091 "size (%d)\n", dsf, pp->lg_sectorsize); 1092 fd = open(dsf, O_WRONLY); 1093 if (fd == -1) 1094 err(EXIT_FAILURE, "%s", dsf); 1095 if (lseek(fd, VTOC_BOOTSIZE, SEEK_SET) != VTOC_BOOTSIZE) 1096 continue; 1097 /* 1098 * We ignore the first VTOC_BOOTSIZE bytes of boot code in 1099 * order to avoid overwriting the label. 1100 */ 1101 if (lseek(fd, sizeof(struct vtoc8), SEEK_SET) != 1102 sizeof(struct vtoc8)) 1103 err(EXIT_FAILURE, "%s", dsf); 1104 if (write(fd, (caddr_t)code + sizeof(struct vtoc8), 1105 VTOC_BOOTSIZE - sizeof(struct vtoc8)) != VTOC_BOOTSIZE - 1106 sizeof(struct vtoc8)) 1107 err(EXIT_FAILURE, "%s", dsf); 1108 installed++; 1109 close(fd); 1110 if (idx != 0 && atoi(s) == idx) 1111 break; 1112 } 1113 if (installed == 0) 1114 errx(EXIT_FAILURE, "%s: no partitions", gp->lg_name); 1115} 1116 1117static void 1118gpart_bootcode(struct gctl_req *req, unsigned int fl) 1119{ 1120 struct gmesh mesh; 1121 struct gclass *classp; 1122 struct ggeom *gp; 1123 const char *s; 1124 void *bootcode, *partcode; 1125 size_t bootsize, partsize; 1126 int error, idx, vtoc8; 1127 1128 if (gctl_has_param(req, GPART_PARAM_BOOTCODE)) { 1129 s = gctl_get_ascii(req, GPART_PARAM_BOOTCODE); 1130 bootsize = 800 * 1024; /* Arbitrary limit. */ 1131 bootcode = gpart_bootfile_read(s, &bootsize); 1132 error = gctl_change_param(req, GPART_PARAM_BOOTCODE, bootsize, 1133 bootcode); 1134 if (error) 1135 errc(EXIT_FAILURE, error, "internal error"); 1136 } else { 1137 bootcode = NULL; 1138 bootsize = 0; 1139 } 1140 1141 s = gctl_get_ascii(req, "class"); 1142 if (s == NULL) 1143 abort(); 1144 error = geom_gettree(&mesh); 1145 if (error != 0) 1146 errc(EXIT_FAILURE, error, "Cannot get GEOM tree"); 1147 classp = find_class(&mesh, s); 1148 if (classp == NULL) { 1149 geom_deletetree(&mesh); 1150 errx(EXIT_FAILURE, "Class %s not found.", s); 1151 } 1152 if (gctl_get_int(req, "nargs") != 1) 1153 errx(EXIT_FAILURE, "Invalid number of arguments."); 1154 s = gctl_get_ascii(req, "arg0"); 1155 if (s == NULL) 1156 abort(); 1157 gp = find_geom(classp, s); 1158 if (gp == NULL) 1159 errx(EXIT_FAILURE, "No such geom: %s.", s); 1160 s = find_geomcfg(gp, "scheme"); 1161 vtoc8 = 0; 1162 if (strcmp(s, "VTOC8") == 0) 1163 vtoc8 = 1; 1164 1165 if (gctl_has_param(req, GPART_PARAM_PARTCODE)) { 1166 s = gctl_get_ascii(req, GPART_PARAM_PARTCODE); 1167 partsize = vtoc8 != 0 ? VTOC_BOOTSIZE : bootsize * 1024; 1168 partcode = gpart_bootfile_read(s, &partsize); 1169 error = gctl_delete_param(req, GPART_PARAM_PARTCODE); 1170 if (error) 1171 errc(EXIT_FAILURE, error, "internal error"); 1172 } else { 1173 partcode = NULL; 1174 partsize = 0; 1175 } 1176 1177 if (gctl_has_param(req, GPART_PARAM_INDEX)) { 1178 if (partcode == NULL) 1179 errx(EXIT_FAILURE, "-i is only valid with -p"); 1180 idx = (int)gctl_get_intmax(req, GPART_PARAM_INDEX); 1181 if (idx < 1) 1182 errx(EXIT_FAILURE, "invalid partition index"); 1183 error = gctl_delete_param(req, GPART_PARAM_INDEX); 1184 if (error) 1185 errc(EXIT_FAILURE, error, "internal error"); 1186 } else 1187 idx = 0; 1188 1189 if (partcode != NULL) { 1190 if (vtoc8 == 0) { 1191 if (idx == 0) 1192 errx(EXIT_FAILURE, "missing -i option"); 1193 gpart_write_partcode(gp, idx, partcode, partsize); 1194 } else 1195 gpart_write_partcode_vtoc8(gp, idx, partcode); 1196 } else 1197 if (bootcode == NULL) 1198 errx(EXIT_FAILURE, "no -b nor -p"); 1199 1200 if (bootcode != NULL) 1201 gpart_issue(req, fl); 1202 1203 geom_deletetree(&mesh); 1204} 1205 1206static void 1207gpart_print_error(const char *errstr) 1208{ 1209 char *errmsg; 1210 int error; 1211 1212 error = strtol(errstr, &errmsg, 0); 1213 if (errmsg != errstr) { 1214 while (errmsg[0] == ' ') 1215 errmsg++; 1216 if (errmsg[0] != '\0') 1217 warnc(error, "%s", errmsg); 1218 else 1219 warnc(error, NULL); 1220 } else 1221 warnx("%s", errmsg); 1222} 1223 1224static void 1225gpart_issue(struct gctl_req *req, unsigned int fl __unused) 1226{ 1227 char buf[4096]; 1228 const char *errstr; 1229 int error, status; 1230 1231 if (gctl_get_int(req, "nargs") != 1) 1232 errx(EXIT_FAILURE, "Invalid number of arguments."); 1233 (void)gctl_delete_param(req, "nargs"); 1234 1235 /* autofill parameters (if applicable). */ 1236 error = gpart_autofill(req); 1237 if (error) { 1238 warnc(error, "autofill"); 1239 status = EXIT_FAILURE; 1240 goto done; 1241 } 1242 1243 bzero(buf, sizeof(buf)); 1244 gctl_rw_param(req, "output", sizeof(buf), buf); 1245 errstr = gctl_issue(req); 1246 if (errstr == NULL || errstr[0] == '\0') { 1247 if (buf[0] != '\0') 1248 printf("%s", buf); 1249 status = EXIT_SUCCESS; 1250 goto done; 1251 } 1252 1253 gpart_print_error(errstr); 1254 status = EXIT_FAILURE; 1255 1256 done: 1257 gctl_free(req); 1258 exit(status); 1259} 1260