2 * Copyright (c) 1994, 1995 Gordon W. Ross
3 * Copyright (c) 1994 Theo de Raadt
5 * Copyright (c) 1987, 1993
6 * The Regents of the University of California. All rights reserved.
8 * This code is derived from software contributed to Berkeley by
9 * Symmetric Computer Systems.
11 * Redistribution and use in source and binary forms, with or without
12 * modification, are permitted provided that the following conditions
14 * 1. Redistributions of source code must retain the above copyright
15 * notice, this list of conditions and the following disclaimer.
16 * 2. Redistributions in binary form must reproduce the above copyright
17 * notice, this list of conditions and the following disclaimer in the
18 * documentation and/or other materials provided with the distribution.
19 * 3. All advertising materials mentioning features or use of this software
20 * must display the following acknowledgement:
21 * This product includes software developed by the University of
22 * California, Berkeley and its contributors.
23 * This product includes software developed by Theo de Raadt.
24 * 4. Neither the name of the University nor the names of its contributors
25 * may be used to endorse or promote products derived from this software
26 * without specific prior written permission.
28 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
29 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
30 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
31 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
32 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
33 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
34 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
35 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
36 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
37 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
40 * from: $NetBSD: disksubr.c,v 1.13 2000/12/17 22:39:18 pk $
45 static const char copyright[] =
46 "@(#) Copyright (c) 1987, 1993\n\
47 The Regents of the University of California. All rights reserved.\n";
51 static char sccsid[] = "@(#)disklabel.c 8.2 (Berkeley) 1/7/94";
52 /* from static char sccsid[] = "@(#)disklabel.c 1.2 (Symmetric) 11/28/85"; */
55 #include <sys/cdefs.h>
56 __FBSDID("$FreeBSD$");
58 #include <sys/param.h>
66 #define MAXPARTITIONS 20
67 #include <sys/disklabel.h>
80 #include "pathnames.h"
82 static void makelabel(const char *, struct disklabel *);
83 static int geom_class_available(const char *);
84 static int writelabel(void);
85 static int readlabel(int flag);
86 static void display(FILE *, const struct disklabel *);
87 static int edit(void);
88 static int editit(void);
89 static void fixlabel(struct disklabel *);
90 static char *skip(char *);
91 static char *word(char *);
92 static int getasciilabel(FILE *, struct disklabel *);
93 static int getasciipartspec(char *, struct disklabel *, int, int);
94 static int checklabel(struct disklabel *);
95 static void usage(void);
96 static struct disklabel *getvirginlabel(void);
98 #define DEFEDITOR _PATH_VI
99 #define DEFPARTITIONS 8
101 static char *specname;
103 static char tmpfil[] = PATH_TMPFILE;
105 static struct disklabel lab;
106 static u_char bootarea[BBSIZE];
107 static off_t mediasize;
108 static ssize_t secsize;
109 static char blank[] = "";
110 static char unknown[] = "unknown";
112 #define MAX_PART ('z')
113 #define MAX_NUM_PARTS (1 + MAX_PART - 'a')
114 static char part_size_type[MAX_NUM_PARTS];
115 static char part_offset_type[MAX_NUM_PARTS];
116 static int part_set[MAX_NUM_PARTS];
118 static int installboot; /* non-zero if we should install a boot program */
119 static int allfields; /* present all fields in edit */
120 static char const *xxboot; /* primary boot */
122 static uint32_t lba_offset;
124 #define LABELSECTOR -1
127 #define LABELOFFSET -1
129 static int labelsoffset = LABELSECTOR;
130 static int labeloffset = LABELOFFSET;
131 static int bbsize = BBSIZE;
134 UNSPEC, EDIT, READ, RESTORE, WRITE, WRITEBOOT
138 static int disable_write; /* set to disable writing to disk label */
139 static int is_file; /* work on a file (abs. pathname), "-f" opt. */
142 main(int argc, char *argv[])
151 while ((ch = getopt(argc, argv, "ABb:efm:nRrw")) != -1)
166 if (!strcmp(optarg, "i386") ||
167 !strcmp(optarg, "amd64") ||
168 !strcmp(optarg, "ia64") ||
169 !strcmp(optarg, "pc98")) {
174 errx(1, "Unsupported architecture");
192 * We accept and ignode -r for compatibility with
193 * historically disklabel usage.
210 if (labelsoffset < 0 || labeloffset < 0)
211 errx(1, "a -m <architecture> option must be specified");
213 /* Figure out the names of the thing we're working on */
217 specname = g_device_path(argv[0]);
218 if (specname == NULL) {
219 warn("unable to get correct path for %s", argv[0]);
222 fd = open(specname, O_RDONLY);
224 warn("error opening %s", specname);
227 pname = g_providername(fd);
229 warn("error getting providername for %s", specname);
236 if (installboot && op == UNSPEC)
238 else if (op == UNSPEC)
258 display(stdout, NULL);
259 error = checklabel(NULL);
265 if (!(t = fopen(argv[1], "r")))
266 err(4, "fopen %s", argv[1]);
268 if (!getasciilabel(t, &lab))
270 error = writelabel();
281 makelabel(name, &lab);
283 if (checklabel(NULL) == 0)
284 error = writelabel();
292 makelabel(argv[1], &lab);
293 if (checklabel(NULL) == 0)
294 error = writelabel();
301 fixlabel(struct disklabel *lp)
303 struct partition *dp;
306 for (i = 0; i < lp->d_npartitions; i++) {
309 if (lp->d_partitions[i].p_size)
313 dp = &lp->d_partitions[0];
314 dp->p_offset = BBSIZE / secsize;
315 dp->p_size = lp->d_secperunit - dp->p_offset;
319 * Construct a prototype disklabel from /etc/disktab.
322 makelabel(const char *type, struct disklabel *lp)
324 struct disklabel *dp;
326 if (strcmp(type, "auto") == 0)
327 dp = getvirginlabel();
329 dp = getdiskbyname(type);
331 errx(1, "%s: unknown disk type", type);
333 bzero(lp->d_packname, sizeof(lp->d_packname));
343 xxboot = "/boot/boot";
344 fd = open(xxboot, O_RDONLY);
346 err(1, "cannot open %s", xxboot);
348 if (st.st_size <= BBSIZE) {
349 if (read(fd, bootarea, st.st_size) != st.st_size)
350 err(1, "read error %s", xxboot);
354 errx(1, "boot code %s is wrong size", xxboot);
358 geom_class_available(const char *name)
360 struct gclass *class;
364 error = geom_gettree(&mesh);
366 errc(1, error, "Cannot get GEOM tree");
368 LIST_FOREACH(class, &mesh.lg_class, lg_class) {
369 if (strcmp(class->lg_name, name) == 0) {
370 geom_deletetree(&mesh);
375 geom_deletetree(&mesh);
384 struct gctl_req *grq;
386 struct disklabel *lp = &lab;
389 warnx("write to disk label supressed - label was as follows:");
390 display(stdout, NULL);
394 lp->d_magic = DISKMAGIC;
395 lp->d_magic2 = DISKMAGIC;
397 lp->d_checksum = dkcksum(lp);
400 for (i = 0; i < lab.d_npartitions; i++)
401 if (lab.d_partitions[i].p_size)
402 lab.d_partitions[i].p_offset += lba_offset;
403 bsd_disklabel_le_enc(bootarea + labeloffset + labelsoffset * secsize,
406 fd = open(specname, O_RDWR);
409 warn("cannot open file %s for writing label", specname);
414 if (geom_class_available("PART") != 0) {
416 * Since we weren't able open provider for
417 * writing, then recommend user to use gpart(8).
420 "cannot open provider %s for writing label",
422 warnx("Try to use gpart(8).");
426 /* Give up if GEOM_BSD is not available. */
427 if (geom_class_available("BSD") == 0) {
428 warnc(serrno, "%s", specname);
432 grq = gctl_get_handle();
433 gctl_ro_param(grq, "verb", -1, "write label");
434 gctl_ro_param(grq, "class", -1, "BSD");
435 gctl_ro_param(grq, "geom", -1, pname);
436 gctl_ro_param(grq, "label", 148+16*8,
437 bootarea + labeloffset + labelsoffset * secsize);
438 errstr = gctl_issue(grq);
439 if (errstr != NULL) {
446 grq = gctl_get_handle();
447 gctl_ro_param(grq, "verb", -1, "write bootcode");
448 gctl_ro_param(grq, "class", -1, "BSD");
449 gctl_ro_param(grq, "geom", -1, pname);
450 gctl_ro_param(grq, "bootcode", BBSIZE, bootarea);
451 errstr = gctl_issue(grq);
452 if (errstr != NULL) {
460 if (write(fd, bootarea, bbsize) != bbsize) {
461 warn("write %s", specname);
471 get_file_parms(int f)
476 if (fstat(f, &sb) != 0)
477 err(4, "fstat failed");
478 i = sb.st_mode & S_IFMT;
479 if (i != S_IFREG && i != S_IFLNK)
480 errx(4, "%s is not a valid file or link", specname);
482 mediasize = sb.st_size;
486 * Fetch disklabel for disk.
496 f = open(specname, O_RDONLY);
498 err(1, "%s", specname);
502 mediasize = g_mediasize(f);
503 secsize = g_sectorsize(f);
504 if (secsize < 0 || mediasize < 0)
505 err(4, "cannot get disk geometry");
507 if (mediasize > (off_t)0xffffffff * secsize)
509 "disks with more than 2^32-1 sectors are not supported");
510 (void)lseek(f, (off_t)0, SEEK_SET);
511 nbytes = read(f, bootarea, BBSIZE);
513 err(4, "%s read", specname);
514 if (nbytes != BBSIZE)
515 errx(4, "couldn't read %d bytes from %s", BBSIZE, specname);
517 error = bsd_disklabel_le_dec(
518 bootarea + (labeloffset + labelsoffset * secsize),
519 &lab, MAXPARTITIONS);
521 errx(1, "%s: no valid label found", specname);
527 * Compensate for absolute block addressing by finding the
528 * smallest partition offset and if the offset of the 'c'
529 * partition is equal to that, subtract it from all offsets.
532 for (i = 0; i < lab.d_npartitions; i++) {
533 if (lab.d_partitions[i].p_size)
534 lba = MIN(lba, lab.d_partitions[i].p_offset);
536 if (lba != 0 && lab.d_partitions[RAW_PART].p_offset == lba) {
537 for (i = 0; i < lab.d_npartitions; i++) {
538 if (lab.d_partitions[i].p_size)
539 lab.d_partitions[i].p_offset -= lba;
542 * Save the offset so that we can write the label
543 * back with absolute block addresses.
552 display(FILE *f, const struct disklabel *lp)
555 const struct partition *pp;
560 fprintf(f, "# %s:\n", specname);
562 if (lp->d_type < DKMAXTYPES)
563 fprintf(f, "type: %s\n", dktypenames[lp->d_type]);
565 fprintf(f, "type: %u\n", lp->d_type);
566 fprintf(f, "disk: %.*s\n", (int)sizeof(lp->d_typename),
568 fprintf(f, "label: %.*s\n", (int)sizeof(lp->d_packname),
570 fprintf(f, "flags:");
571 if (lp->d_flags & D_REMOVABLE)
572 fprintf(f, " removeable");
573 if (lp->d_flags & D_ECC)
575 if (lp->d_flags & D_BADSECT)
576 fprintf(f, " badsect");
578 fprintf(f, "bytes/sector: %lu\n", (u_long)lp->d_secsize);
579 fprintf(f, "sectors/track: %lu\n", (u_long)lp->d_nsectors);
580 fprintf(f, "tracks/cylinder: %lu\n", (u_long)lp->d_ntracks);
581 fprintf(f, "sectors/cylinder: %lu\n", (u_long)lp->d_secpercyl);
582 fprintf(f, "cylinders: %lu\n", (u_long)lp->d_ncylinders);
583 fprintf(f, "sectors/unit: %lu\n", (u_long)lp->d_secperunit);
584 fprintf(f, "rpm: %u\n", lp->d_rpm);
585 fprintf(f, "interleave: %u\n", lp->d_interleave);
586 fprintf(f, "trackskew: %u\n", lp->d_trackskew);
587 fprintf(f, "cylinderskew: %u\n", lp->d_cylskew);
588 fprintf(f, "headswitch: %lu\t\t# milliseconds\n",
589 (u_long)lp->d_headswitch);
590 fprintf(f, "track-to-track seek: %ld\t# milliseconds\n",
591 (u_long)lp->d_trkseek);
592 fprintf(f, "drivedata: ");
593 for (i = NDDATA - 1; i >= 0; i--)
594 if (lp->d_drivedata[i])
598 for (j = 0; j <= i; j++)
599 fprintf(f, "%lu ", (u_long)lp->d_drivedata[j]);
602 fprintf(f, "%u partitions:\n", lp->d_npartitions);
604 "# size offset fstype [fsize bsize bps/cpg]\n");
605 pp = lp->d_partitions;
606 for (i = 0; i < lp->d_npartitions; i++, pp++) {
608 fprintf(f, " %c: %10lu %10lu ", 'a' + i,
609 (u_long)pp->p_size, (u_long)pp->p_offset);
610 if (pp->p_fstype < FSMAXTYPES)
611 fprintf(f, "%8.8s", fstypenames[pp->p_fstype]);
613 fprintf(f, "%8d", pp->p_fstype);
614 switch (pp->p_fstype) {
616 case FS_UNUSED: /* XXX */
617 fprintf(f, " %5lu %5lu %2s",
619 (u_long)(pp->p_fsize * pp->p_frag), "");
623 fprintf(f, " %5lu %5lu %5u",
625 (u_long)(pp->p_fsize * pp->p_frag),
630 fprintf(f, " %5lu %5lu %5d",
632 (u_long)(pp->p_fsize * pp->p_frag),
637 fprintf(f, "%20.20s", "");
641 fprintf(f, " # \"raw\" part, don't edit");
653 struct disklabel label;
656 if ((fd = mkstemp(tmpfil)) == -1 ||
657 (fp = fdopen(fd, "w")) == NULL) {
658 warnx("can't create %s", tmpfil);
666 fp = fopen(tmpfil, "r");
668 warnx("can't reopen %s for reading", tmpfil);
671 bzero((char *)&label, sizeof(label));
672 c = getasciilabel(fp, &label);
676 if (writelabel() == 0) {
677 (void) unlink(tmpfil);
681 printf("re-edit the label? [y]: ");
684 if (c != EOF && c != (int)'\n')
685 while (getchar() != (int)'\n')
690 (void) unlink(tmpfil);
703 omask = sigblock(sigmask(SIGINT)|sigmask(SIGQUIT)|sigmask(SIGHUP));
704 while ((pid = fork()) < 0) {
705 if (errno == EPROCLIM) {
706 warnx("you have too many processes");
709 if (errno != EAGAIN) {
718 if (setresgid(gid, gid, gid) == -1)
721 if (setresuid(uid, uid, uid) == -1)
723 if ((ed = getenv("EDITOR")) == (char *)0)
725 execlp(ed, ed, tmpfil, (char *)0);
728 while ((xpid = wait(&locstat)) >= 0)
739 while (*cp != '\0' && isspace(*cp))
741 if (*cp == '\0' || *cp == '#')
751 while (*cp != '\0' && !isspace(*cp) && *cp != '#')
753 if ((c = *cp) != '\0') {
762 * Read an ascii label in from fd f,
763 * in the same format as that put out by display(),
767 getasciilabel(FILE *f, struct disklabel *lp)
772 char *tp, line[BUFSIZ];
774 int lineno = 0, errors = 0;
777 makelabel("auto", lp);
778 bzero(&part_set, sizeof(part_set));
779 bzero(&part_size_type, sizeof(part_size_type));
780 bzero(&part_offset_type, sizeof(part_offset_type));
781 lp->d_bbsize = BBSIZE; /* XXX */
782 lp->d_sbsize = 0; /* XXX */
783 while (fgets(line, sizeof(line) - 1, f)) {
785 if ((cp = index(line,'\n')) != 0)
792 fprintf(stderr, "line %d: syntax error\n", lineno);
796 *tp++ = '\0', tp = skip(tp);
797 if (!strcmp(cp, "type")) {
801 for (; cpp < &dktypenames[DKMAXTYPES]; cpp++)
802 if (*cpp && !strcmp(*cpp, tp)) {
803 lp->d_type = cpp - dktypenames;
806 if (cpp < &dktypenames[DKMAXTYPES])
809 v = strtoul(tp, &endp, 10);
810 if (errno != 0 || *endp != '\0')
813 fprintf(stderr, "line %d:%s %lu\n", lineno,
814 "Warning, unknown disk type", v);
819 if (!strcmp(cp, "flags")) {
820 for (v = 0; (cp = tp) && *cp != '\0';) {
822 if (!strcmp(cp, "removeable"))
824 else if (!strcmp(cp, "ecc"))
826 else if (!strcmp(cp, "badsect"))
830 "line %d: %s: bad flag\n",
838 if (!strcmp(cp, "drivedata")) {
839 for (i = 0; (cp = tp) && *cp != '\0' && i < NDDATA;) {
840 lp->d_drivedata[i++] = strtoul(cp, NULL, 10);
845 if (sscanf(cp, "%lu partitions", &v) == 1) {
846 if (v > MAXPARTITIONS) {
848 "line %d: bad # of partitions\n", lineno);
849 lp->d_npartitions = MAXPARTITIONS;
851 } else if (v < DEFPARTITIONS) {
853 "line %d: bad # of partitions\n", lineno);
854 lp->d_npartitions = DEFPARTITIONS;
857 lp->d_npartitions = v;
862 if (!strcmp(cp, "disk")) {
863 strncpy(lp->d_typename, tp, sizeof (lp->d_typename));
866 if (!strcmp(cp, "label")) {
867 strncpy(lp->d_packname, tp, sizeof (lp->d_packname));
870 if (!strcmp(cp, "bytes/sector")) {
871 v = strtoul(tp, NULL, 10);
872 if (v == 0 || (v % DEV_BSIZE) != 0) {
874 "line %d: %s: bad sector size\n",
881 if (!strcmp(cp, "sectors/track")) {
882 v = strtoul(tp, NULL, 10);
883 #if (ULONG_MAX != 0xffffffffUL)
884 if (v == 0 || v > 0xffffffff)
889 fprintf(stderr, "line %d: %s: bad %s\n",
896 if (!strcmp(cp, "sectors/cylinder")) {
897 v = strtoul(tp, NULL, 10);
899 fprintf(stderr, "line %d: %s: bad %s\n",
906 if (!strcmp(cp, "tracks/cylinder")) {
907 v = strtoul(tp, NULL, 10);
909 fprintf(stderr, "line %d: %s: bad %s\n",
916 if (!strcmp(cp, "cylinders")) {
917 v = strtoul(tp, NULL, 10);
919 fprintf(stderr, "line %d: %s: bad %s\n",
923 lp->d_ncylinders = v;
926 if (!strcmp(cp, "sectors/unit")) {
927 v = strtoul(tp, NULL, 10);
929 fprintf(stderr, "line %d: %s: bad %s\n",
933 lp->d_secperunit = v;
936 if (!strcmp(cp, "rpm")) {
937 v = strtoul(tp, NULL, 10);
938 if (v == 0 || v > USHRT_MAX) {
939 fprintf(stderr, "line %d: %s: bad %s\n",
946 if (!strcmp(cp, "interleave")) {
947 v = strtoul(tp, NULL, 10);
948 if (v == 0 || v > USHRT_MAX) {
949 fprintf(stderr, "line %d: %s: bad %s\n",
953 lp->d_interleave = v;
956 if (!strcmp(cp, "trackskew")) {
957 v = strtoul(tp, NULL, 10);
959 fprintf(stderr, "line %d: %s: bad %s\n",
966 if (!strcmp(cp, "cylinderskew")) {
967 v = strtoul(tp, NULL, 10);
969 fprintf(stderr, "line %d: %s: bad %s\n",
976 if (!strcmp(cp, "headswitch")) {
977 v = strtoul(tp, NULL, 10);
978 lp->d_headswitch = v;
981 if (!strcmp(cp, "track-to-track seek")) {
982 v = strtoul(tp, NULL, 10);
986 /* the ':' was removed above */
987 if (*cp < 'a' || *cp > MAX_PART || cp[1] != '\0') {
989 "line %d: %s: Unknown disklabel field\n", lineno,
995 /* Process a partition specification line. */
997 if (part >= lp->d_npartitions) {
999 "line %d: partition name out of range a-%c: %s\n",
1000 lineno, 'a' + lp->d_npartitions - 1, cp);
1006 if (getasciipartspec(tp, lp, part, lineno) != 0) {
1011 errors += checklabel(lp);
1012 return (errors == 0);
1015 #define NXTNUM(n) do { \
1017 fprintf(stderr, "line %d: too few numeric fields\n", lineno); \
1020 cp = tp, tp = word(cp); \
1021 (n) = strtoul(cp, NULL, 10); \
1025 /* retain 1 character following number */
1026 #define NXTWORD(w,n) do { \
1028 fprintf(stderr, "line %d: too few numeric fields\n", lineno); \
1032 cp = tp, tp = word(cp); \
1033 (n) = strtoul(cp, &tmp, 10); \
1034 if (tmp) (w) = *tmp; \
1039 * Read a partition line into partition `part' in the specified disklabel.
1040 * Return 0 on success, 1 on failure.
1043 getasciipartspec(char *tp, struct disklabel *lp, int part, int lineno)
1045 struct partition *pp;
1050 pp = &lp->d_partitions[part];
1054 NXTWORD(part_size_type[part],v);
1055 if (v == 0 && part_size_type[part] != '*') {
1057 "line %d: %s: bad partition size\n", lineno, cp);
1063 NXTWORD(part_offset_type[part],v);
1064 if (v == 0 && part_offset_type[part] != '*' &&
1065 part_offset_type[part] != '\0') {
1067 "line %d: %s: bad partition offset\n", lineno, cp);
1072 fprintf(stderr, "line %d: missing file system type\n", lineno);
1075 cp = tp, tp = word(cp);
1076 for (cpp = fstypenames; cpp < &fstypenames[FSMAXTYPES]; cpp++)
1077 if (*cpp && !strcmp(*cpp, cp))
1080 pp->p_fstype = cpp - fstypenames;
1084 v = strtoul(cp, &endp, 10);
1085 if (errno != 0 || *endp != '\0')
1089 if (v >= FSMAXTYPES) {
1091 "line %d: Warning, unknown file system type %s\n",
1098 switch (pp->p_fstype) {
1102 /* accept defaults for fsize/frag/cpg */
1104 NXTNUM(pp->p_fsize);
1105 if (pp->p_fsize == 0)
1108 pp->p_frag = v / pp->p_fsize;
1112 /* else default to 0's */
1121 * Check disklabel for errors and fill in
1122 * derived fields according to supplied values.
1125 checklabel(struct disklabel *lp)
1127 struct partition *pp;
1130 u_long base_offset, needed, total_size, total_percent, current_offset;
1132 int seen_default_offset;
1135 struct partition *pp2;
1142 if (lp->d_secsize == 0) {
1143 fprintf(stderr, "sector size 0\n");
1146 if (lp->d_nsectors == 0) {
1147 fprintf(stderr, "sectors/track 0\n");
1150 if (lp->d_ntracks == 0) {
1151 fprintf(stderr, "tracks/cylinder 0\n");
1154 if (lp->d_ncylinders == 0) {
1155 fprintf(stderr, "cylinders/unit 0\n");
1159 warnx("revolutions/minute 0");
1160 if (lp->d_secpercyl == 0)
1161 lp->d_secpercyl = lp->d_nsectors * lp->d_ntracks;
1162 if (lp->d_secperunit == 0)
1163 lp->d_secperunit = lp->d_secpercyl * lp->d_ncylinders;
1164 if (lp->d_bbsize == 0) {
1165 fprintf(stderr, "boot block size 0\n");
1167 } else if (lp->d_bbsize % lp->d_secsize)
1168 warnx("boot block size %% sector-size != 0");
1169 if (lp->d_npartitions > MAXPARTITIONS) {
1170 warnx("number of partitions (%lu) > MAXPARTITIONS (%d)",
1171 (u_long)lp->d_npartitions, MAXPARTITIONS);
1174 if (lp->d_npartitions < DEFPARTITIONS) {
1175 warnx("number of partitions (%lu) < DEFPARTITIONS (%d)",
1176 (u_long)lp->d_npartitions, DEFPARTITIONS);
1180 struct disklabel *vl;
1182 vl = getvirginlabel();
1183 if (lp->d_secsize == 0)
1184 lp->d_secsize = vl->d_secsize;
1185 if (lp->d_nsectors == 0)
1186 lp->d_nsectors = vl->d_nsectors;
1187 if (lp->d_ntracks == 0)
1188 lp->d_ntracks = vl->d_ntracks;
1189 if (lp->d_ncylinders == 0)
1190 lp->d_ncylinders = vl->d_ncylinders;
1192 lp->d_rpm = vl->d_rpm;
1193 if (lp->d_interleave == 0)
1194 lp->d_interleave = vl->d_interleave;
1195 if (lp->d_secpercyl == 0)
1196 lp->d_secpercyl = vl->d_secpercyl;
1197 if (lp->d_secperunit == 0 ||
1198 lp->d_secperunit > vl->d_secperunit)
1199 lp->d_secperunit = vl->d_secperunit;
1200 if (lp->d_bbsize == 0)
1201 lp->d_bbsize = vl->d_bbsize;
1202 if (lp->d_npartitions < DEFPARTITIONS ||
1203 lp->d_npartitions > MAXPARTITIONS)
1204 lp->d_npartitions = vl->d_npartitions;
1208 /* first allocate space to the partitions, then offsets */
1209 total_size = 0; /* in sectors */
1210 total_percent = 0; /* in percent */
1212 /* find all fixed partitions */
1213 for (i = 0; i < lp->d_npartitions; i++) {
1214 pp = &lp->d_partitions[i];
1216 if (part_size_type[i] == '*') {
1217 if (i == RAW_PART) {
1218 pp->p_size = lp->d_secperunit;
1221 warnx("Too many '*' partitions (%c and %c)",
1222 hog_part + 'a',i + 'a');
1230 switch (part_size_type[i]) {
1232 total_percent += size;
1253 warnx("unknown multiplier suffix '%c' for partition %c (should be K, M, G or T)",
1254 part_size_type[i], i + 'a');
1257 /* don't count %'s yet */
1258 if (part_size_type[i] != '%') {
1260 * for all not in sectors, convert to
1263 if (part_size_type[i] != '\0') {
1264 if (size % lp->d_secsize != 0)
1265 warnx("partition %c not an integer number of sectors",
1267 size /= lp->d_secsize;
1270 /* else already in sectors */
1278 /* Find out the total free space, excluding the boot block area. */
1279 base_offset = BBSIZE / secsize;
1281 for (i = 0; i < lp->d_npartitions; i++) {
1282 pp = &lp->d_partitions[i];
1283 if (!part_set[i] || i == RAW_PART ||
1284 part_size_type[i] == '%' || part_size_type[i] == '*')
1286 if (pp->p_offset > base_offset)
1287 free_space += pp->p_offset - base_offset;
1288 if (pp->p_offset + pp->p_size > base_offset)
1289 base_offset = pp->p_offset + pp->p_size;
1291 if (base_offset < lp->d_secperunit)
1292 free_space += lp->d_secperunit - base_offset;
1294 /* handle % partitions - note %'s don't need to add up to 100! */
1295 if (total_percent != 0) {
1296 if (total_percent > 100) {
1297 fprintf(stderr,"total percentage %lu is greater than 100\n",
1302 if (free_space > 0) {
1303 for (i = 0; i < lp->d_npartitions; i++) {
1304 pp = &lp->d_partitions[i];
1305 if (part_set[i] && part_size_type[i] == '%') {
1306 /* careful of overflows! and integer roundoff */
1307 pp->p_size = ((double)pp->p_size/100) * free_space;
1308 total_size += pp->p_size;
1310 /* FIX we can lose a sector or so due to roundoff per
1311 partition. A more complex algorithm could avoid that */
1316 "%ld sectors available to give to '*' and '%%' partitions\n",
1319 /* fix? set all % partitions to size 0? */
1322 /* give anything remaining to the hog partition */
1323 if (hog_part != -1) {
1325 * Find the range of offsets usable by '*' partitions around
1326 * the hog partition and how much space they need.
1329 base_offset = BBSIZE / secsize;
1330 for (i = hog_part - 1; i >= 0; i--) {
1331 pp = &lp->d_partitions[i];
1332 if (!part_set[i] || i == RAW_PART)
1334 if (part_offset_type[i] == '*') {
1335 needed += pp->p_size;
1338 base_offset = pp->p_offset + pp->p_size;
1341 current_offset = lp->d_secperunit;
1342 for (i = lp->d_npartitions - 1; i > hog_part; i--) {
1343 pp = &lp->d_partitions[i];
1344 if (!part_set[i] || i == RAW_PART)
1346 if (part_offset_type[i] == '*') {
1347 needed += pp->p_size;
1350 current_offset = pp->p_offset;
1353 if (current_offset - base_offset <= needed) {
1354 fprintf(stderr, "Cannot find space for partition %c\n",
1357 "Need more than %lu sectors between %lu and %lu\n",
1358 needed, base_offset, current_offset);
1360 lp->d_partitions[hog_part].p_size = 0;
1362 lp->d_partitions[hog_part].p_size = current_offset -
1363 base_offset - needed;
1364 total_size += lp->d_partitions[hog_part].p_size;
1368 /* Now set the offsets for each partition */
1369 current_offset = BBSIZE / secsize; /* in sectors */
1370 seen_default_offset = 0;
1371 for (i = 0; i < lp->d_npartitions; i++) {
1373 pp = &lp->d_partitions[i];
1375 if (part_offset_type[i] == '*') {
1376 if (i == RAW_PART) {
1379 pp->p_offset = current_offset;
1380 seen_default_offset = 1;
1383 /* allow them to be out of order for old-style tables */
1384 if (pp->p_offset < current_offset &&
1385 seen_default_offset && i != RAW_PART &&
1386 pp->p_fstype != FS_VINUM) {
1388 "Offset %ld for partition %c overlaps previous partition which ends at %lu\n",
1389 (long)pp->p_offset,i+'a',current_offset);
1391 "Labels with any *'s for offset must be in ascending order by sector\n");
1393 } else if (pp->p_offset != current_offset &&
1394 i != RAW_PART && seen_default_offset) {
1396 * this may give unneeded warnings if
1397 * partitions are out-of-order
1400 "Offset %ld for partition %c doesn't match expected value %ld",
1401 (long)pp->p_offset, i + 'a', current_offset);
1405 current_offset = pp->p_offset + pp->p_size;
1409 for (i = 0; i < lp->d_npartitions; i++) {
1411 pp = &lp->d_partitions[i];
1412 if (pp->p_size == 0 && pp->p_offset != 0)
1413 warnx("partition %c: size 0, but offset %lu",
1414 part, (u_long)pp->p_offset);
1416 if (pp->p_size % lp->d_secpercyl)
1417 warnx("partition %c: size %% cylinder-size != 0",
1419 if (pp->p_offset % lp->d_secpercyl)
1420 warnx("partition %c: offset %% cylinder-size != 0",
1423 if (pp->p_offset > lp->d_secperunit) {
1425 "partition %c: offset past end of unit\n", part);
1428 if (pp->p_offset + pp->p_size > lp->d_secperunit) {
1430 "partition %c: partition extends past end of unit\n",
1434 if (i == RAW_PART) {
1435 if (pp->p_fstype != FS_UNUSED)
1436 warnx("partition %c is not marked as unused!",part);
1437 if (pp->p_offset != 0)
1438 warnx("partition %c doesn't start at 0!",part);
1439 if (pp->p_size != lp->d_secperunit)
1440 warnx("partition %c doesn't cover the whole unit!",part);
1442 if ((pp->p_fstype != FS_UNUSED) || (pp->p_offset != 0) ||
1443 (pp->p_size != lp->d_secperunit)) {
1444 warnx("An incorrect partition %c may cause problems for "
1445 "standard system utilities",part);
1449 /* check for overlaps */
1450 /* this will check for all possible overlaps once and only once */
1451 for (j = 0; j < i; j++) {
1452 pp2 = &lp->d_partitions[j];
1453 if (j != RAW_PART && i != RAW_PART &&
1454 pp->p_fstype != FS_VINUM &&
1455 pp2->p_fstype != FS_VINUM &&
1456 part_set[i] && part_set[j]) {
1457 if (pp2->p_offset < pp->p_offset + pp->p_size &&
1458 (pp2->p_offset + pp2->p_size > pp->p_offset ||
1459 pp2->p_offset >= pp->p_offset)) {
1460 fprintf(stderr,"partitions %c and %c overlap!\n",
1467 for (; i < lp->d_npartitions; i++) {
1469 pp = &lp->d_partitions[i];
1470 if (pp->p_size || pp->p_offset)
1471 warnx("unused partition %c: size %d offset %lu",
1472 'a' + i, pp->p_size, (u_long)pp->p_offset);
1478 * When operating on a "virgin" disk, try getting an initial label
1479 * from the associated device driver. This might work for all device
1480 * drivers that are able to fetch some initial device parameters
1481 * without even having access to a (BSD) disklabel, like SCSI disks,
1482 * most IDE drives, or vn devices.
1484 * The device name must be given in its "canonical" form.
1486 static struct disklabel *
1487 getvirginlabel(void)
1489 static struct disklabel loclab;
1490 struct partition *dp;
1494 if ((f = open(specname, O_RDONLY)) == -1) {
1495 warn("cannot open %s", specname);
1502 mediasize = g_mediasize(f);
1503 secsize = g_sectorsize(f);
1504 if (secsize < 0 || mediasize < 0) {
1509 memset(&loclab, 0, sizeof loclab);
1510 loclab.d_magic = DISKMAGIC;
1511 loclab.d_magic2 = DISKMAGIC;
1512 loclab.d_secsize = secsize;
1513 loclab.d_secperunit = mediasize / secsize;
1516 * Nobody in these enligthened days uses the CHS geometry for
1517 * anything, but nontheless try to get it right. If we fail
1518 * to get any good ideas from the device, construct something
1519 * which is IBM-PC friendly.
1521 if (ioctl(f, DIOCGFWSECTORS, &u) == 0)
1522 loclab.d_nsectors = u;
1524 loclab.d_nsectors = 63;
1525 if (ioctl(f, DIOCGFWHEADS, &u) == 0)
1526 loclab.d_ntracks = u;
1527 else if (loclab.d_secperunit <= 63*1*1024)
1528 loclab.d_ntracks = 1;
1529 else if (loclab.d_secperunit <= 63*16*1024)
1530 loclab.d_ntracks = 16;
1532 loclab.d_ntracks = 255;
1533 loclab.d_secpercyl = loclab.d_ntracks * loclab.d_nsectors;
1534 loclab.d_ncylinders = loclab.d_secperunit / loclab.d_secpercyl;
1535 loclab.d_npartitions = DEFPARTITIONS;
1537 /* Various (unneeded) compat stuff */
1538 loclab.d_rpm = 3600;
1539 loclab.d_bbsize = BBSIZE;
1540 loclab.d_interleave = 1;
1541 strncpy(loclab.d_typename, "amnesiac",
1542 sizeof(loclab.d_typename));
1544 dp = &loclab.d_partitions[RAW_PART];
1545 dp->p_size = loclab.d_secperunit;
1546 loclab.d_checksum = dkcksum(&loclab);
1556 "%s\n%s\n%s\n%s\n%s\n%s\n%s\n%s\n%s\n%s\n%s\n%s\n%s\n%s\n",
1557 "usage: bsdlabel disk",
1558 "\t\t(to read label)",
1559 " bsdlabel -w [-n] [-m machine] disk [type]",
1560 "\t\t(to write label with existing boot program)",
1561 " bsdlabel -e [-n] [-m machine] disk",
1562 "\t\t(to edit label)",
1563 " bsdlabel -R [-n] [-m machine] disk protofile",
1564 "\t\t(to restore label with existing boot program)",
1565 " bsdlabel -B [-b boot] [-m machine] disk",
1566 "\t\t(to install boot program with existing on-disk label)",
1567 " bsdlabel -w -B [-n] [-b boot] [-m machine] disk [type]",
1568 "\t\t(to write label and install boot program)",
1569 " bsdlabel -R -B [-n] [-b boot] [-m machine] disk protofile",
1570 "\t\t(to restore label and install boot program)"