#if HAVE_NBTOOL_CONFIG_H
#include "nbtool_config.h"
#endif
#include <sys/cdefs.h>
#ifndef lint
__COPYRIGHT("@(#) Copyright (c) 1987, 1993\
The Regents of the University of California. All rights reserved.");
#endif
#ifndef lint
#if 0
static char sccsid[] = "@(#)disklabel.c 8.4 (Berkeley) 5/4/95";
#else
__RCSID("$NetBSD: main.c,v 1.59 2025/01/19 04:37:15 tsutsui Exp $");
#endif
#endif
#include <sys/param.h>
#include <sys/file.h>
#include <sys/stat.h>
#include <sys/wait.h>
#define DKTYPENAMES
#define FSTYPENAMES
#include <ctype.h>
#include <err.h>
#include <errno.h>
#include <signal.h>
#include <string.h>
#include <stdio.h>
#include <stdlib.h>
#include <limits.h>
#include <unistd.h>
#include <ufs/ufs/dinode.h>
#include <ufs/ffs/fs.h>
#if HAVE_NBTOOL_CONFIG_H
#include <nbinclude/sys/disklabel.h>
#include <nbinclude/sys/disklabel_acorn.h>
#include <nbinclude/sys/bootblock.h>
#include "../../include/disktab.h"
#else
#include <sys/ioctl.h>
#include <sys/disklabel.h>
#include <sys/disklabel_acorn.h>
#include <sys/bootblock.h>
#include <util.h>
#include <disktab.h>
#endif
#include "pathnames.h"
#include "extern.h"
#include "dkcksum.h"
#include "bswap.h"
#ifndef BBSIZE
#define BBSIZE 8192
#endif
#define DISKMAGIC_REV bswap32(DISKMAGIC)
#define DISKMAGIC_DELETED (~DISKMAGIC)
#define DISKMAGIC_DELETED_REV bswap32(~DISKMAGIC)
#define DEFEDITOR _PATH_VI
char specname[MAXPATHLEN];
char bootarea[BBSIZE];
int bootarea_len;
static struct disklabel lab;
static int Aflag;
static int Fflag;
static int rflag;
static int tflag;
int Cflag;
static int Dflag;
static int Iflag;
static int lflag;
static int verbose;
static int read_all;
static int write_label(int);
static int readlabel_direct(int);
static void writelabel_direct(int);
static int update_label(int, u_int, u_int);
static struct disklabel *find_label(int, u_int);
#if !defined(NATIVELABEL_ONLY)
static void getmachineparams(const char *);
#endif
static void makedisktab(FILE *, struct disklabel *);
static void makelabel(const char *, const char *);
static void l_perror(const char *);
static void readlabel(int);
static int edit(int);
static int editit(const char *);
static char *skip(char *);
static char *word(char *);
static int getasciilabel(FILE *, struct disklabel *);
__dead static void usage(void);
static int qsort_strcmp(const void *, const void *);
static int getulong(const char *, char, char **,
unsigned long *, unsigned long);
#define GETNUM32(a, v) getulong(a, '\0', NULL, v, UINT32_MAX)
#define GETNUM16(a, v) getulong(a, '\0', NULL, v, UINT16_MAX)
#define GETNUM8(a, v) getulong(a, '\0', NULL, v, UINT8_MAX)
static int set_writable_fd = -1;
#if !defined(NATIVELABEL_ONLY)
static u_int labeloffset;
static u_int labelsector;
static int labelusesmbr;
u_int maxpartitions;
static int byteorder;
static int biendian_p;
#ifndef HAVE_NBTOOL_CONFIG_H
static int native_p = 1;
#endif
int bswap_p;
static const struct disklabel_params {
const char *machine;
u_char labelusesmbr : 1;
u_char labelsector : 7;
u_char maxpartitions;
u_char raw_part;
u_char oldmaxpartitions;
u_short labeloffset;
u_short byteorder;
} disklabel_params[] = {
{ "mvme68k", 0, 0, 8, 2, 0, 0, BIG_ENDIAN },
{ "next68k", 0, 0, 8, 2, 0, 0, BIG_ENDIAN },
{ "algor", 0, 0, 8, 2, 0, 64, LITTLE_ENDIAN },
{ "alpha", 0, 0, 8, 2, 0, 64, LITTLE_ENDIAN },
{ "luna68k", 0, 0, 8, 2, 0, 64, BIG_ENDIAN },
{ "mac68k", 0, 0, 8, 2, 0, 64, BIG_ENDIAN },
{ "news68k", 0, 0, 8, 2, 0, 64, BIG_ENDIAN },
{ "newsmips", 0, 0, 8, 2, 0, 64, BIG_ENDIAN },
{ "pmax", 0, 0, 8, 2, 0, 64, LITTLE_ENDIAN },
{ "sun2", 0, 0, 8, 2, 0, 64, BIG_ENDIAN },
{ "sun68k", 0, 0, 8, 2, 0, 64, BIG_ENDIAN },
{ "virt68k", 0, 0, 8, 2, 0, 64, BIG_ENDIAN },
{ "x68k", 0, 0, 8, 2, 0, 64, BIG_ENDIAN },
{ "vax", 0, 0, 12, 2, 8, 64, LITTLE_ENDIAN },
{ "amiga", 0, 0, 16, 2, 0, 64, BIG_ENDIAN },
{ "amigappc", 0, 0, 16, 2, 0, 64, BIG_ENDIAN },
{ "evbmips", 0, 0, 16, 2, 0, 64, 0 },
{ "evbppc", 0, 0, 16, 2, 0, 64, BIG_ENDIAN },
{ "sparc", 0, 0, 8, 2, 0, 128, BIG_ENDIAN },
{ "sparc64", 0, 0, 8, 2, 0, 128, BIG_ENDIAN },
{ "sun3", 0, 0, 8, 2, 0, 128, BIG_ENDIAN },
{ "atari", 0, 0, 16, 2, 0, 516, BIG_ENDIAN },
{ "mipsco", 0, 1, 8, 2, 0, 0, BIG_ENDIAN },
{ "mvmeppc", 0, 1, 8, 3, 0, 0, BIG_ENDIAN },
{ "bebox", 0, 1, 8, 3, 0, 0, BIG_ENDIAN },
{ "emips", 0, 1, 16, 2, 0, 0, BIG_ENDIAN },
{ "hppa", 0, 1, 16, 2, 0, 0, BIG_ENDIAN },
{ "ibmnws", 0, 1, 16, 2, 0, 0, BIG_ENDIAN },
{ "ofppc", 0, 1, 16, 2, 0, 0, BIG_ENDIAN },
{ "rs6000", 0, 1, 16, 2, 0, 0, BIG_ENDIAN },
{ "sandpoint", 0, 1, 16, 2, 0, 0, BIG_ENDIAN },
{ "sgimips", 0, 1, 16, 2, 0, 0, BIG_ENDIAN },
{ "sbmips", 0, 1, 16, 3, 0, 0, 0 },
{ "cesfic", 0, 2, 8, 2, 0, 0, BIG_ENDIAN },
{ "hp300", 0, 2, 8, 2, 0, 0, BIG_ENDIAN },
{ "ews4800mips",0, 9, 16, 15, 0, 0, BIG_ENDIAN },
{ "macppc", 1, 0, 16, 2, 0, 64, BIG_ENDIAN },
{ "pmon", 1, 0, 16, 2, 0, 64, 0 },
{ "prep", 1, 1, 8, 2, 0, 0, BIG_ENDIAN },
{ "dreamcast", 1, 1, 16, 2, 0, 0, LITTLE_ENDIAN },
{ "evbcf", 1, 1, 16, 2, 0, 0, BIG_ENDIAN },
{ "evbppc-mbr", 1, 1, 16, 2, 0, 0, BIG_ENDIAN },
{ "evbsh3", 1, 1, 16, 2, 0, 0, 0 },
{ "hpcsh", 1, 1, 16, 2, 0, 0, LITTLE_ENDIAN },
{ "mmeye", 1, 1, 16, 2, 0, 0, 0 },
{ "or1k", 1, 1, 16, 2, 0, 0, BIG_ENDIAN },
{ "riscv", 1, 1, 16, 2, 0, 0, LITTLE_ENDIAN },
{ "acorn32", 1, 1, 16, 2, 8, 0, LITTLE_ENDIAN },
{ "cats", 1, 1, 16, 2, 8, 0, LITTLE_ENDIAN },
{ "evbarm", 1, 1, 16, 2, 8, 0, 0 },
{ "iyonix", 1, 1, 16, 2, 8, 0, LITTLE_ENDIAN },
{ "netwinder", 1, 1, 16, 2, 8, 0, LITTLE_ENDIAN },
{ "shark", 1, 1, 16, 2, 8, 0, LITTLE_ENDIAN },
{ "amd64", 1, 1, 16, 3, 0, 0, LITTLE_ENDIAN },
{ "arc", 1, 1, 16, 3, 0, 0, LITTLE_ENDIAN },
{ "cobalt", 1, 1, 16, 3, 0, 0, LITTLE_ENDIAN },
{ "landisk", 1, 1, 16, 3, 0, 0, LITTLE_ENDIAN },
{ "epoc32", 1, 1, 16, 3, 8, 0, LITTLE_ENDIAN },
{ "hpcarm", 1, 1, 16, 3, 8, 0, LITTLE_ENDIAN },
{ "hpcmips", 1, 1, 16, 3, 8, 0, LITTLE_ENDIAN },
{ "i386", 1, 1, 16, 3, 8, 0, LITTLE_ENDIAN },
{ "ia64", 1, 1, 16, 3, 8, 0, LITTLE_ENDIAN },
{ "zaurus", 1, 1, 16, 3, 8, 0, LITTLE_ENDIAN },
{ NULL, 0, 0, 0, 0, 0, 0, 0 },
};
#ifndef HAVE_NBTOOL_CONFIG_H
static struct disklabel_params native_params;
#endif
static const struct arch_endian {
int byteorder;
const char *arch;
} arch_endians[] = {
{ LITTLE_ENDIAN, "aarch64" },
{ LITTLE_ENDIAN, "alpha" },
{ LITTLE_ENDIAN, "arm" },
{ LITTLE_ENDIAN, "earm" },
{ LITTLE_ENDIAN, "earmhf" },
{ LITTLE_ENDIAN, "earmv4" },
{ LITTLE_ENDIAN, "earmv5" },
{ LITTLE_ENDIAN, "earmv6" },
{ LITTLE_ENDIAN, "earmv6hf" },
{ LITTLE_ENDIAN, "earmv7" },
{ LITTLE_ENDIAN, "earmv7hf" },
{ LITTLE_ENDIAN, "i386" },
{ LITTLE_ENDIAN, "ia64" },
{ LITTLE_ENDIAN, "mipsel" },
{ LITTLE_ENDIAN, "mips64el" },
{ LITTLE_ENDIAN, "riscv32" },
{ LITTLE_ENDIAN, "riscv64" },
{ LITTLE_ENDIAN, "sh3el" },
{ LITTLE_ENDIAN, "vax" },
{ LITTLE_ENDIAN, "x86_64" },
{ BIG_ENDIAN, "aarch64eb" },
{ BIG_ENDIAN, "armeb" },
{ BIG_ENDIAN, "coldfire" },
{ BIG_ENDIAN, "earmeb" },
{ BIG_ENDIAN, "earmhfeb" },
{ BIG_ENDIAN, "earmv4eb" },
{ BIG_ENDIAN, "earmv5eb" },
{ BIG_ENDIAN, "earmv6eb" },
{ BIG_ENDIAN, "earmv6hfeb" },
{ BIG_ENDIAN, "earmv7eb" },
{ BIG_ENDIAN, "earmv7hfeb" },
{ BIG_ENDIAN, "hppa" },
{ BIG_ENDIAN, "m68000" },
{ BIG_ENDIAN, "m68k" },
{ BIG_ENDIAN, "mipseb" },
{ BIG_ENDIAN, "mips64eb" },
{ BIG_ENDIAN, "or1k" },
{ BIG_ENDIAN, "powerpc" },
{ BIG_ENDIAN, "sh3eb" },
{ BIG_ENDIAN, "sparc" },
{ BIG_ENDIAN, "sparc64" },
{ 0, NULL },
};
#define LABEL_OFFSET (dklabel_getlabelsector() * DEV_BSIZE + dklabel_getlabeloffset())
#else
#define labeloffset LABELOFFSET
#define labelsector LABELSECTOR
#define labelusesmbr LABELUSESMBR
#define maxpartitions MAXPARTITIONS
#define LABEL_OFFSET (LABELSECTOR * DEV_BSIZE + LABELOFFSET)
#endif
#ifndef LABELOFFSET_MBR
#define LABELOFFSET_MBR 512
#endif
#if HAVE_NBTOOL_CONFIG_H
static int
opendisk(const char *path, int flags, char *buf, int buflen, int cooked)
{
int f;
f = open(path, flags, 0);
strlcpy(buf, path, buflen);
return f;
}
#endif
#if !defined(NATIVELABEL_ONLY)
static void
setbyteorder(int new_byteorder)
{
static int set_p;
if ((!biendian_p || set_p)
&& byteorder != 0
&& byteorder != new_byteorder) {
warnx("changing %s byteorder to %s",
byteorder == LITTLE_ENDIAN ? "le" : "be",
new_byteorder == LITTLE_ENDIAN ? "le" : "be");
}
byteorder = new_byteorder;
biendian_p = 0;
set_p = 1;
}
static void
getmachineparams(const char *mach)
{
const struct disklabel_params *dp = disklabel_params;
for (; dp->machine != NULL; dp++) {
if (!strcmp(mach, dp->machine)) {
labelusesmbr = dp->labelusesmbr;
labelsector = dp->labelsector;
labeloffset = dp->labeloffset;
maxpartitions = dp->maxpartitions;
biendian_p = (dp->byteorder == 0);
if (!biendian_p)
setbyteorder(dp->byteorder);
return;
}
}
errx(1, "%s: unknown machine type", mach);
}
static void
getarchbyteorder(const char *arch)
{
const struct arch_endian *p = arch_endians;
for (; p->arch != NULL; p++) {
if (!strcmp(arch, p->arch)) {
setbyteorder(p->byteorder);
return;
}
}
errx(1, "%s: unknown arch", arch);
}
static daddr_t
dklabel_getlabelsector(void)
{
unsigned long int nval;
char *end;
const char *val;
if ((val = getenv("DISKLABELSECTOR")) == NULL)
return labelsector;
if ((nval = strtoul(val, &end, 10)) == ULONG_MAX && errno == ERANGE)
err(EXIT_FAILURE, "DISKLABELSECTOR in environment");
return nval;
}
static off_t
dklabel_getlabeloffset(void)
{
unsigned long int nval;
char *end;
const char *val;
if ((val = getenv("DISKLABELOFFSET")) == NULL)
return labeloffset;
if ((nval = strtoul(val, &end, 10)) == ULONG_MAX && errno == ERANGE)
err(EXIT_FAILURE, "DISKLABELOFFSET in environment");
return nval;
}
#endif
static void
clear_writable(void)
{
static int zero = 0;
dk_ioctl(set_writable_fd, DIOCWLABEL, &zero);
}
int
main(int argc, char *argv[])
{
FILE *t;
int ch, f, error;
char *dkname;
#if !defined(NATIVELABEL_ONLY)
char *cp;
#endif
struct stat sb;
int writable;
enum {
UNSPEC, EDIT, READ, RESTORE, SETWRITABLE, SETREADONLY,
WRITE,
#if !defined(NO_INTERACT)
INTERACT,
#endif
DELETE
} op = UNSPEC, old_op;
#if !defined(NATIVELABEL_ONLY)
unsigned long val;
#endif
#ifndef HAVE_NBTOOL_CONFIG_H
#if !defined(NATIVELABEL_ONLY)
labeloffset = native_params.labeloffset = getlabeloffset();
labelsector = native_params.labelsector = getlabelsector();
labelusesmbr = native_params.labelusesmbr = getlabelusesmbr();
maxpartitions = native_params.maxpartitions = getmaxpartitions();
byteorder = native_params.byteorder = BYTE_ORDER;
#endif
#endif
#if !defined(NATIVELABEL_ONLY)
if ((cp = getenv("MACHINE")) != NULL) {
getmachineparams(cp);
}
if ((cp = getenv("MACHINE_ARCH")) != NULL) {
getarchbyteorder(cp);
}
#endif
#if HAVE_NBTOOL_CONFIG_H
Fflag = rflag = 1;
#endif
error = 0;
#if !defined(NATIVELABEL_ONLY)
while ((ch = getopt(argc, argv, "AB:CDFIL:M:NO:P:RWef:ilmnrtvw")) != -1) {
#else
while ((ch = getopt(argc, argv, "ACDFINRWef:ilrtvw")) != -1) {
#endif
old_op = op;
switch (ch) {
case 'A':
Aflag = 1;
rflag = 1;
break;
case 'C':
Cflag = 1;
break;
case 'D':
Dflag = 1;
rflag = 1;
break;
case 'F':
Fflag = 1;
rflag = 1;
break;
case 'I':
Iflag = 1;
rflag = 1;
break;
case 'R':
op = RESTORE;
break;
#if !defined(NATIVELABEL_ONLY)
case 'B':
if (!strcmp(optarg, "be")) {
setbyteorder(BIG_ENDIAN);
} else if (!strcmp(optarg, "le")) {
setbyteorder(LITTLE_ENDIAN);
} else {
errx(1, "%s: not be or le", optarg);
}
break;
case 'M':
getmachineparams(optarg);
break;
#endif
case 'N':
op = SETREADONLY;
break;
#if !defined(NATIVELABEL_ONLY)
case 'L':
val = strtoul(optarg, NULL, 10);
if ((val == ULONG_MAX && errno == ERANGE) || val > UINT_MAX)
err(EXIT_FAILURE, "invalid label sector: %s", optarg);
labelsector = val;
break;
case 'O':
val = strtoul(optarg, NULL, 10);
if ((val == ULONG_MAX && errno == ERANGE) || val > UINT_MAX)
err(EXIT_FAILURE, "invalid label offset: %s", optarg);
labeloffset = val;
break;
case 'P':
val = strtoul(optarg, NULL, 10);
if ((val == ULONG_MAX && errno == ERANGE) || val < 1 || val > UINT_MAX)
err(EXIT_FAILURE, "invalid max partitions: %s", optarg);
maxpartitions = val;
break;
#endif
case 'W':
op = SETWRITABLE;
break;
case 'e':
op = EDIT;
break;
case 'f':
if (setdisktab(optarg) == -1)
usage();
break;
#if !defined(NO_INTERACT)
case 'i':
op = INTERACT;
break;
#endif
case 'l':
lflag = 1;
break;
#if !defined(NATIVELABEL_ONLY)
case 'm':
labelusesmbr = 1;
break;
case 'n':
labelusesmbr = 0;
break;
#endif
case 'r':
rflag = 1;
break;
case 't':
tflag = 1;
break;
case 'v':
verbose++;
break;
case 'w':
op = WRITE;
break;
case '?':
default:
usage();
}
if (old_op != UNSPEC && old_op != op)
usage();
}
if (maxpartitions > MAXPARTITIONS) {
errx(1, "too large maxpartitions > %u\n", MAXPARTITIONS);
}
#if !defined(NATIVELABEL_ONLY)
if (maxpartitions == 0) {
errx(1, "unknown label: use -M/-B and $MACHINE/$MACHINE_ARCH");
}
if (byteorder != BIG_ENDIAN && byteorder != LITTLE_ENDIAN) {
errx(1, "unknown byteorder");
}
bswap_p = (byteorder != BYTE_ORDER);
#ifdef DEBUG
printf("labelusesmbr=%d labelsector=%u labeloffset=%u maxpartitions=%u\n",
labelusesmbr, labelsector, labeloffset, maxpartitions);
printf("byteorder=%d bswap_p=%d\n", byteorder, bswap_p);
#endif
#ifndef HAVE_NBTOOL_CONFIG_H
native_p = native_params.labelusesmbr == labelusesmbr
&& native_params.labelsector == labelsector
&& native_params.labeloffset == labeloffset
&& maxpartitions <= native_params.maxpartitions
&& !bswap_p;
if (!native_p)
Fflag = rflag = 1;
#endif
#endif
argc -= optind;
argv += optind;
if (lflag)
exit(list_fs_types() ? EXIT_SUCCESS : EXIT_FAILURE);
if (op == UNSPEC)
op = Dflag ? DELETE : READ;
if (argc < 1)
usage();
if (Iflag && op != EDIT
#if !defined(NO_INTERACT)
&& op != INTERACT
#endif
)
usage();
dkname = argv[0];
f = opendisk(dkname, op == READ ? O_RDONLY : O_RDWR,
specname, sizeof specname, 0);
if (f < 0)
err(4, "%s", specname);
if (!Fflag && fstat(f, &sb) == 0 && S_ISREG(sb.st_mode))
Fflag = rflag = 1;
switch (op) {
case DELETE:
if (argc != 1)
usage();
Dflag = 2;
writelabel_direct(f);
break;
case EDIT:
if (argc != 1)
usage();
readlabel(f);
error = edit(f);
break;
#if !defined(NO_INTERACT)
case INTERACT:
if (argc != 1)
usage();
readlabel(f);
if (lab.d_bbsize == 0)
lab.d_bbsize = BBSIZE;
if (lab.d_sbsize == 0)
lab.d_sbsize = SBLOCKSIZE;
interact(&lab, f);
break;
#endif
case READ:
if (argc != 1)
usage();
read_all = Aflag;
readlabel(f);
if (read_all)
break;
if (tflag)
makedisktab(stdout, &lab);
else {
showinfo(stdout, &lab, specname);
showpartitions(stdout, &lab, Cflag);
}
error = checklabel(&lab);
if (error)
error += 100;
break;
case RESTORE:
if (argc != 2)
usage();
if (!(t = fopen(argv[1], "r")))
err(4, "%s", argv[1]);
if (getasciilabel(t, &lab))
error = write_label(f);
else
error = 1;
break;
case SETREADONLY:
writable = 0;
goto do_diocwlabel;
case SETWRITABLE:
writable = 1;
do_diocwlabel:
if (argc != 1)
usage();
if (dk_ioctl(f, DIOCWLABEL, &writable) < 0)
err(4, "ioctl DIOCWLABEL");
break;
case WRITE:
if (argc < 2 || argc > 3)
usage();
makelabel(argv[1], argv[2]);
if (checklabel(&lab) == 0)
error = write_label(f);
else
error = 1;
break;
case UNSPEC:
usage();
}
exit(error);
}
static void
makelabel(const char *type, const char *name)
{
struct disklabel *dp;
dp = getdiskbyname(type);
if (dp == NULL)
errx(1, "unknown disk type: %s", type);
lab = *dp;
(void)memset(lab.d_packname, 0, sizeof(lab.d_packname));
if (name)
(void)strncpy(lab.d_packname, name, sizeof(lab.d_packname));
}
static int
write_label(int f)
{
int writable;
lab.d_magic = DISKMAGIC;
lab.d_magic2 = DISKMAGIC;
lab.d_checksum = 0;
lab.d_checksum = dkcksum(&lab);
if (rflag) {
if (!Fflag && dk_ioctl(f, DIOCSDINFO, &lab) < 0 &&
errno != ENODEV && errno != ENOTTY) {
l_perror("ioctl DIOCSDINFO");
return (1);
}
writable = 1;
if (!Fflag) {
if (dk_ioctl(f, DIOCWLABEL, &writable) < 0)
perror("ioctl DIOCWLABEL");
set_writable_fd = f;
atexit(clear_writable);
}
writelabel_direct(f);
if (!Fflag && dk_ioctl(f, DIOCWDINFO, &lab) < 0) {
l_perror("ioctl DIOCWDINFO");
return (1);
}
} else {
if (dk_ioctl(f, DIOCWDINFO, &lab) < 0) {
l_perror("ioctl DIOCWDINFO");
return (1);
}
}
#ifdef VAX_ALTLABELS
if (lab.d_type == DKTYPE_SMD && lab.d_flags & D_BADSECT &&
lab.d_secsize == 512) {
daddr_t alt;
int i;
uint8_t sec0[512];
if (pread(f, sec0, 512, 0) < 512) {
warn("read master label to write alternates");
return 0;
}
alt = lab.d_ncylinders * lab.d_secpercyl - lab.d_nsectors;
for (i = 1; i < 11 && (uint32_t)i < lab.d_nsectors; i += 2) {
if (pwrite(f, sec0, 512, (off_t)(alt + i) * 512) < 512)
warn("alternate label %d write", i/2);
}
}
#endif
return 0;
}
int
writelabel(int f, struct disklabel *lp)
{
if (lp != &lab)
lab = *lp;
return write_label(f);
}
static void
l_perror(const char *s)
{
switch (errno) {
case ESRCH:
warnx("%s: No disk label on disk;\n"
"use \"disklabel -I\" to install initial label", s);
break;
case EINVAL:
warnx("%s: Label magic number or checksum is wrong!\n"
"(disklabel or kernel is out of date?)", s);
break;
case EBUSY:
warnx("%s: Open partition would move or shrink", s);
break;
case EXDEV:
warnx("%s: Labeled partition or 'a' partition must start"
" at beginning of disk", s);
break;
default:
warn("%s", s);
break;
}
}
#ifdef NO_MBR_SUPPORT
#define process_mbr(f, action) 1
#else
static int
process_mbr(int f, int (*action)(int, u_int))
{
struct mbr_partition *dp;
struct mbr_sector mbr;
int rval = 1, res;
int part;
u_int ext_base, next_ext, this_ext, start;
ext_base = 0;
next_ext = 0;
for (;;) {
this_ext = next_ext;
next_ext = 0;
if (verbose > 1)
warnx("reading mbr sector %u", this_ext);
if (pread(f, &mbr, sizeof mbr, this_ext * (off_t)DEV_BSIZE)
!= sizeof(mbr)) {
if (verbose)
warn("Can't read master boot record %u",
this_ext);
break;
}
if (mbr.mbr_magic != htole16(MBR_MAGIC)) {
if (verbose)
warnx("Invalid signature in mbr record %u",
this_ext);
break;
}
dp = &mbr.mbr_parts[0];
for (part = 0; part < MBR_PART_COUNT; dp++, part++) {
start = le32toh(dp->mbrp_start);
switch (dp->mbrp_type) {
#ifdef COMPAT_386BSD_MBRPART
case MBR_PTYPE_386BSD:
if (ext_base != 0)
break;
#endif
case MBR_PTYPE_NETBSD:
res = action(f, this_ext + start);
if (res <= 0)
return res;
if (res > rval)
rval = res;
break;
case MBR_PTYPE_EXT:
case MBR_PTYPE_EXT_LBA:
case MBR_PTYPE_EXT_LNX:
next_ext = start;
break;
default:
break;
}
}
if (next_ext == 0)
break;
next_ext += ext_base;
if (ext_base == 0)
ext_base = next_ext;
if (next_ext <= this_ext) {
if (verbose)
warnx("Invalid extended chain %x <= %x",
next_ext, this_ext);
break;
}
}
return rval;
}
static int
readlabel_mbr(int f, u_int sector)
{
struct disklabel *disk_lp;
disk_lp = find_label(f, sector);
if (disk_lp == NULL)
return 1;
targettohlabel(&lab, disk_lp);
return 0;
}
static int
writelabel_mbr(int f, u_int sector)
{
return update_label(f, sector, labelusesmbr ? LABELOFFSET_MBR : ~0U) ? 2 : 0;
}
#endif
#ifndef USE_ACORN
#define get_filecore_partition(f) 0
#else
static int
filecore_checksum(u_char *bootblock)
{
u_char byte0, accum_diff;
u_int sum;
int i;
sum = 0;
accum_diff = 0;
byte0 = bootblock[0];
for (i = 0; i < 511; ++i) {
sum += bootblock[i];
accum_diff |= bootblock[i] ^ byte0;
}
accum_diff |= bootblock[i] ^ byte0;
if (accum_diff == 0)
return (-1);
return (sum - ((sum - 1) / 255) * 255);
}
static u_int
get_filecore_partition(int f)
{
struct filecore_bootblock *fcbb;
static u_char bb[DEV_BSIZE];
u_int offset;
struct riscix_partition_table *riscix_part;
int loop;
if (pread(f, bb, sizeof(bb), (off_t)FILECORE_BOOT_SECTOR * DEV_BSIZE) != sizeof(bb))
err(4, "can't read filecore boot block");
fcbb = (struct filecore_bootblock *)bb;
if (filecore_checksum(bb) != fcbb->checksum)
return (0);
offset = (fcbb->partition_cyl_low + (fcbb->partition_cyl_high << 8))
* fcbb->heads * fcbb->secspertrack;
switch (fcbb->partition_type) {
case PARTITION_FORMAT_RISCBSD:
return (offset);
case PARTITION_FORMAT_RISCIX:
if (pread(f, bb, sizeof(bb), (off_t)offset * DEV_BSIZE) != sizeof(bb))
err(4, "can't read riscix partition table");
riscix_part = (struct riscix_partition_table *)bb;
for (loop = 0; loop < NRISCIX_PARTITIONS; ++loop) {
if (strcmp((char *)riscix_part->partitions[loop].rp_name,
"RiscBSD") == 0 ||
strcmp((char *)riscix_part->partitions[loop].rp_name,
"NetBSD") == 0 ||
strcmp((char *)riscix_part->partitions[loop].rp_name,
"Empty:") == 0) {
return riscix_part->partitions[loop].rp_start;
break;
}
}
errx(4, "cannot label: no NetBSD partition found"
" in RISCiX partition table");
default:
errx(4, "cannot label: filecore-only disk"
" (no non-ADFS partition)");
}
return (0);
}
#endif
static void
readlabel(int f)
{
if (rflag) {
if (readlabel_direct(f) == 0)
return;
if (!Fflag && Iflag && (dk_ioctl(f, DIOCGDINFO, &lab) == 0 ||
dk_ioctl(f, DIOCGDEFLABEL, &lab) == 0))
return;
} else {
if (dk_ioctl(f, DIOCGDINFO, &lab) < 0)
err(4, "ioctl DIOCGDINFO");
return;
}
if (read_all == 2)
exit(0);
errx(1, "could not read existing label");
}
static struct disklabel *
find_label(int f, u_int sector)
{
struct disklabel *disk_lp, hlp, tlp;
int i;
off_t offset;
const char *is_deleted;
bootarea_len = pread(f, bootarea, sizeof bootarea,
sector * (off_t)DEV_BSIZE);
if (bootarea_len <= 0) {
if (verbose)
warn("failed to read bootarea from sector %u", sector);
return NULL;
}
if (verbose > 2)
warnx("read sector %u len %d looking for label",
sector, bootarea_len);
for (offset = LABEL_OFFSET, i = -4;; offset = i += 4) {
is_deleted = "";
if (i == LABEL_OFFSET)
continue;
disk_lp = (void *)(bootarea + offset);
if ((char *)(disk_lp + 1) > bootarea + bootarea_len)
break;
memcpy(&tlp, disk_lp, sizeof(tlp));
if (tlp.d_magic2 != tlp.d_magic)
continue;
if (read_all && (tlp.d_magic == DISKMAGIC_DELETED ||
tlp.d_magic == DISKMAGIC_DELETED_REV)) {
tlp.d_magic ^= ~0u;
tlp.d_magic2 ^= ~0u;
is_deleted = "deleted ";
}
if (target32toh(tlp.d_magic) != DISKMAGIC) {
if (target32toh(tlp.d_magic) == DISKMAGIC_REV &&
target32toh(tlp.d_magic2) == DISKMAGIC_REV)
warnx("ignoring %sbyteswapped label"
" at offset %jd from sector %u",
is_deleted, (intmax_t)offset, sector);
continue;
}
if (target16toh(tlp.d_npartitions) > maxpartitions ||
dkcksum_target(&tlp) != 0) {
if (verbose > 0)
warnx("corrupt label found at offset %jd in "
"sector %u", (intmax_t)offset, sector);
continue;
}
if (verbose > 1)
warnx("%slabel found at offset %jd from sector %u",
is_deleted, (intmax_t)offset, sector);
if (!read_all)
return disk_lp;
targettohlabel(&hlp, &tlp);
printf("# %ssector %u offset %jd bytes\n",
is_deleted, sector, (intmax_t)offset);
if (tflag)
makedisktab(stdout, &hlp);
else {
showinfo(stdout, &hlp, specname);
showpartitions(stdout, &hlp, Cflag);
}
checklabel(&hlp);
htotargetlabel(&tlp, &hlp);
memcpy(disk_lp, &tlp, sizeof(tlp));
read_all = 2;
}
return NULL;
}
static void
write_bootarea(int f, u_int sector)
{
int wlen;
if (bootarea_len <= 0)
errx(1, "attempting to write after failed read");
#ifdef ALPHA_BOOTBLOCK_CKSUM
if (sector == 0) {
struct alpha_boot_block *bb;
bb = (struct alpha_boot_block *)(void *)bootarea;
bb->bb_cksum = 0;
ALPHA_BOOT_BLOCK_CKSUM(bb, &bb->bb_cksum);
}
#endif
wlen = pwrite(f, bootarea, bootarea_len, sector * (off_t)DEV_BSIZE);
if (wlen == bootarea_len)
return;
if (wlen == -1)
err(1, "disklabel write (sector %u) size %d failed",
sector, bootarea_len);
errx(1, "disklabel write (sector %u) size %d truncated to %d",
sector, bootarea_len, wlen);
}
static int
update_label(int f, u_int label_sector, u_int label_offset)
{
struct disklabel *disk_lp;
disk_lp = find_label(f, label_sector);
if (disk_lp && Dflag) {
disk_lp->d_magic ^= ~0u;
disk_lp->d_magic2 ^= ~0u;
if (Dflag == 2)
write_bootarea(f, label_sector);
disk_lp = NULL;
}
if (Dflag == 2)
return 0;
if (disk_lp == NULL) {
if (label_offset == ~0u)
return 0;
disk_lp = (void *)(bootarea + label_offset);
if ((char *)(disk_lp + 1) > bootarea + bootarea_len)
errx(1, "no space in bootarea (sector %u) "
"to create label", label_sector);
}
htotargetlabel(disk_lp, &lab);
write_bootarea(f, label_sector);
return 1;
}
static void
writelabel_direct(int f)
{
u_int label_sector;
int written = 0;
int rval;
label_sector = get_filecore_partition(f);
if (label_sector != 0)
written = update_label(f, label_sector, DEV_BSIZE);
rval = process_mbr(f, writelabel_mbr);
if (rval == 2 || written)
update_label(f, 0, ~0u);
else
update_label(f, 0, LABEL_OFFSET);
}
static int
readlabel_direct(int f)
{
struct disklabel *disk_lp;
u_int filecore_partition_offset;
filecore_partition_offset = get_filecore_partition(f);
if (filecore_partition_offset != 0) {
disk_lp = find_label(f, filecore_partition_offset);
if (disk_lp != NULL) {
targettohlabel(&lab, disk_lp);
return 0;
}
}
if (labelusesmbr && process_mbr(f, readlabel_mbr) == 0)
return 0;
disk_lp = find_label(f, 0);
if (disk_lp != NULL) {
targettohlabel(&lab, disk_lp);
return 0;
}
if (!labelusesmbr && process_mbr(f, readlabel_mbr) == 0)
return 0;
return 1;
}
static void
makedisktab(FILE *f, struct disklabel *lp)
{
int i;
const char *did;
struct partition *pp;
did = "\\\n\t:";
(void) fprintf(f, "%.*s|Automatically generated label:\\\n\t:dt=",
(int) sizeof(lp->d_typename), lp->d_typename);
if ((unsigned) lp->d_type < DKMAXTYPES)
(void) fprintf(f, "%s:", dktypenames[lp->d_type]);
else
(void) fprintf(f, "unknown%" PRIu16 ":", lp->d_type);
(void) fprintf(f, "se#%" PRIu32 ":", lp->d_secsize);
(void) fprintf(f, "ns#%" PRIu32 ":", lp->d_nsectors);
(void) fprintf(f, "nt#%" PRIu32 ":", lp->d_ntracks);
(void) fprintf(f, "sc#%" PRIu32 ":", lp->d_secpercyl);
(void) fprintf(f, "nc#%" PRIu32 ":", lp->d_ncylinders);
if ((lp->d_secpercyl * lp->d_ncylinders) != lp->d_secperunit) {
(void) fprintf(f, "%ssu#%" PRIu32 ":", did, lp->d_secperunit);
did = "";
}
if (lp->d_rpm != 3600) {
(void) fprintf(f, "%srm#%" PRIu16 ":", did, lp->d_rpm);
did = "";
}
if (lp->d_interleave != 1) {
(void) fprintf(f, "%sil#%" PRIu16 ":", did, lp->d_interleave);
did = "";
}
if (lp->d_trackskew != 0) {
(void) fprintf(f, "%ssk#%" PRIu16 ":", did, lp->d_trackskew);
did = "";
}
if (lp->d_cylskew != 0) {
(void) fprintf(f, "%scs#%" PRIu16 ":", did, lp->d_cylskew);
did = "";
}
if (lp->d_headswitch != 0) {
(void) fprintf(f, "%shs#%" PRIu32 ":", did, lp->d_headswitch);
did = "";
}
if (lp->d_trkseek != 0) {
(void) fprintf(f, "%sts#%" PRIu32 ":", did, lp->d_trkseek);
did = "";
}
#ifdef notyet
(void) fprintf(f, "drivedata: ");
for (i = NDDATA - 1; i >= 0; i--)
if (lp->d_drivedata[i])
break;
if (i < 0)
i = 0;
for (j = 0; j <= i; j++)
(void) fprintf(f, "%" PRIu32 " ", lp->d_drivedata[j]);
#endif
pp = lp->d_partitions;
for (i = 0; i < lp->d_npartitions; i++, pp++) {
if (pp->p_size) {
char c = 'a' + i;
(void) fprintf(f, "\\\n\t:");
(void) fprintf(f, "p%c#%" PRIu32 ":", c, pp->p_size);
(void) fprintf(f, "o%c#%" PRIu32 ":", c, pp->p_offset);
if (pp->p_fstype != FS_UNUSED) {
if ((unsigned) pp->p_fstype < FSMAXTYPES)
(void) fprintf(f, "t%c=%s:", c,
fstypenames[pp->p_fstype]);
else
(void) fprintf(f,
"t%c=unknown%" PRIu8 ":",
c, pp->p_fstype);
}
switch (pp->p_fstype) {
case FS_UNUSED:
break;
case FS_BSDFFS:
case FS_BSDLFS:
case FS_EX2FS:
case FS_ADOS:
case FS_APPLEUFS:
(void) fprintf(f, "b%c#%" PRIu64 ":", c,
(uint64_t)pp->p_fsize * pp->p_frag);
(void) fprintf(f, "f%c#%" PRIu32 ":", c,
pp->p_fsize);
break;
default:
break;
}
}
}
(void) fprintf(f, "\n");
(void) fflush(f);
}
static int
edit(int f)
{
const char *tmpdir;
char tmpfil[MAXPATHLEN];
int first, ch, fd;
int get_ok;
FILE *fp;
if ((tmpdir = getenv("TMPDIR")) == NULL)
tmpdir = _PATH_TMP;
(void)snprintf(tmpfil, sizeof(tmpfil), "%s/%s", tmpdir, TMPFILE);
if ((fd = mkstemp(tmpfil)) == -1 || (fp = fdopen(fd, "w")) == NULL) {
warn("%s", tmpfil);
return (1);
}
(void)fchmod(fd, 0600);
showinfo(fp, &lab, specname);
showpartitions(fp, &lab, Cflag);
(void) fclose(fp);
for (;;) {
if (!editit(tmpfil))
break;
fp = fopen(tmpfil, "r");
if (fp == NULL) {
warn("%s", tmpfil);
break;
}
(void) memset(&lab, 0, sizeof(lab));
get_ok = getasciilabel(fp, &lab);
fclose(fp);
if (get_ok && write_label(f) == 0) {
(void) unlink(tmpfil);
return (0);
}
(void) printf("re-edit the label? [y]: ");
(void) fflush(stdout);
first = ch = getchar();
while (ch != '\n' && ch != EOF)
ch = getchar();
if (first == 'n' || first == 'N')
break;
}
(void)unlink(tmpfil);
return (1);
}
static int
editit(const char *tmpfil)
{
int pid, xpid;
int status;
sigset_t nsigset, osigset;
sigemptyset(&nsigset);
sigaddset(&nsigset, SIGINT);
sigaddset(&nsigset, SIGQUIT);
sigaddset(&nsigset, SIGHUP);
sigprocmask(SIG_BLOCK, &nsigset, &osigset);
while ((pid = fork()) < 0) {
if (errno != EAGAIN) {
sigprocmask(SIG_SETMASK, &osigset, (sigset_t *)0);
warn("fork");
return (0);
}
sleep(1);
}
if (pid == 0) {
const char *ed;
char *buf;
int retval;
sigprocmask(SIG_SETMASK, &osigset, (sigset_t *)0);
setgid(getgid());
setuid(getuid());
if ((ed = getenv("EDITOR")) == (char *)0)
ed = DEFEDITOR;
asprintf(&buf, "%s %s", ed, tmpfil);
if (!buf)
err(1, "malloc");
retval = execlp(_PATH_BSHELL, _PATH_BSHELL, "-c", buf, NULL);
if (retval == -1)
perror(ed);
exit(retval);
}
while ((xpid = wait(&status)) >= 0)
if (xpid == pid)
break;
sigprocmask(SIG_SETMASK, &osigset, (sigset_t *)0);
return (!status);
}
static char *
skip(char *cp)
{
cp += strspn(cp, " \t");
if (*cp == '\0')
return (NULL);
return (cp);
}
static char *
word(char *cp)
{
if (cp == NULL || *cp == '\0')
return (NULL);
cp += strcspn(cp, " \t");
if (*cp == '\0')
return (NULL);
*cp++ = '\0';
cp += strspn(cp, " \t");
if (*cp == '\0')
return (NULL);
return (cp);
}
#define _CHECKLINE \
if (tp == NULL || *tp == '\0') { \
warnx("line %d: too few fields", lineno); \
errors++; \
break; \
}
#define __CHECKLINE \
if (*tp == NULL || **tp == '\0') { \
warnx("line %d: too few fields", lineno); \
*tp = _error_; \
return 0; \
}
static char _error_[] = "";
#define NXTNUM(n) if ((n = nxtnum(&tp, lineno),0) + tp != _error_) \
; else goto error
#define NXTXNUM(n) if ((n = nxtxnum(&tp, lp, lineno),0) + tp != _error_) \
; else goto error
static unsigned long
nxtnum(char **tp, int lineno)
{
char *cp;
unsigned long v;
__CHECKLINE
if (getulong(*tp, '\0', &cp, &v, UINT32_MAX) != 0) {
warnx("line %d: syntax error", lineno);
*tp = _error_;
return 0;
}
*tp = cp;
return v;
}
static unsigned long
nxtxnum(char **tp, struct disklabel *lp, int lineno)
{
char *cp, *ncp;
unsigned long n, v;
__CHECKLINE
cp = *tp;
if (getulong(cp, '/', &ncp, &n, UINT32_MAX) != 0)
goto bad;
if (*ncp == '/') {
n *= lp->d_secpercyl;
cp = ncp + 1;
if (getulong(cp, '/', &ncp, &v, UINT32_MAX) != 0)
goto bad;
n += v * lp->d_nsectors;
cp = ncp + 1;
if (getulong(cp, '\0', &ncp, &v, UINT32_MAX) != 0)
goto bad;
n += v;
}
*tp = ncp;
return n;
bad:
warnx("line %d: invalid format", lineno);
*tp = _error_;
return 0;
}
static int
getasciilabel(FILE *f, struct disklabel *lp)
{
const char *const *cpp, *s;
struct partition *pp;
char *cp, *tp, line[BUFSIZ], tbuf[15];
int lineno, errors;
unsigned long v;
unsigned int part;
lineno = 0;
errors = 0;
lp->d_bbsize = BBSIZE;
lp->d_sbsize = SBLOCKSIZE;
while (fgets(line, sizeof(line) - 1, f)) {
lineno++;
if ((cp = strpbrk(line, "#\r\n")) != NULL)
*cp = '\0';
cp = skip(line);
if (cp == NULL)
continue;
tp = strchr(cp, ':');
if (tp == NULL) {
warnx("line %d: syntax error", lineno);
errors++;
continue;
}
*tp++ = '\0', tp = skip(tp);
if (!strcmp(cp, "type")) {
if (tp == NULL) {
strlcpy(tbuf, "unknown", sizeof(tbuf));
tp = tbuf;
}
cpp = dktypenames;
for (; cpp < &dktypenames[DKMAXTYPES]; cpp++)
if ((s = *cpp) && !strcasecmp(s, tp)) {
lp->d_type = cpp - dktypenames;
goto next;
}
if (GETNUM16(tp, &v) != 0) {
warnx("line %d: syntax error", lineno);
errors++;
continue;
}
if (v >= DKMAXTYPES)
warnx("line %d: warning, unknown disk type: %s",
lineno, tp);
lp->d_type = v;
continue;
}
if (!strcmp(cp, "flags")) {
for (v = 0; (cp = tp) && *cp != '\0';) {
tp = word(cp);
if (!strcasecmp(cp, "removable"))
v |= D_REMOVABLE;
else if (!strcasecmp(cp, "ecc"))
v |= D_ECC;
else if (!strcasecmp(cp, "badsect"))
v |= D_BADSECT;
else {
warnx("line %d: bad flag: %s",
lineno, cp);
errors++;
}
}
lp->d_flags = v;
continue;
}
if (!strcmp(cp, "drivedata")) {
int i;
for (i = 0; (cp = tp) && *cp != '\0' && i < NDDATA;) {
if (GETNUM32(cp, &v) != 0) {
warnx("line %d: bad drive data",
lineno);
errors++;
} else
lp->d_drivedata[i] = v;
i++;
tp = word(cp);
}
continue;
}
if (sscanf(cp, "%lu partitions", &v) == 1) {
if (v == 0 || v > maxpartitions) {
warnx("line %d: bad # of partitions", lineno);
lp->d_npartitions = maxpartitions;
errors++;
} else
lp->d_npartitions = v;
continue;
}
if (tp == NULL) {
tbuf[0] = '\0';
tp = tbuf;
}
if (!strcmp(cp, "disk")) {
strncpy(lp->d_typename, tp, sizeof(lp->d_typename));
continue;
}
if (!strcmp(cp, "label")) {
strncpy(lp->d_packname, tp, sizeof(lp->d_packname));
continue;
}
if (!strcmp(cp, "bytes/sector")) {
if (GETNUM32(tp, &v) != 0 || v <= 0 || (v % 512) != 0) {
warnx("line %d: bad %s: %s", lineno, cp, tp);
errors++;
} else
lp->d_secsize = v;
continue;
}
if (!strcmp(cp, "sectors/track")) {
if (GETNUM32(tp, &v) != 0) {
warnx("line %d: bad %s: %s", lineno, cp, tp);
errors++;
} else
lp->d_nsectors = v;
continue;
}
if (!strcmp(cp, "sectors/cylinder")) {
if (GETNUM32(tp, &v) != 0) {
warnx("line %d: bad %s: %s", lineno, cp, tp);
errors++;
} else
lp->d_secpercyl = v;
continue;
}
if (!strcmp(cp, "tracks/cylinder")) {
if (GETNUM32(tp, &v) != 0) {
warnx("line %d: bad %s: %s", lineno, cp, tp);
errors++;
} else
lp->d_ntracks = v;
continue;
}
if (!strcmp(cp, "cylinders")) {
if (GETNUM32(tp, &v) != 0) {
warnx("line %d: bad %s: %s", lineno, cp, tp);
errors++;
} else
lp->d_ncylinders = v;
continue;
}
if (!strcmp(cp, "total sectors") ||
!strcmp(cp, "sectors/unit")) {
if (GETNUM32(tp, &v) != 0) {
warnx("line %d: bad %s: %s", lineno, cp, tp);
errors++;
} else
lp->d_secperunit = v;
continue;
}
if (!strcmp(cp, "rpm")) {
if (GETNUM16(tp, &v) != 0) {
warnx("line %d: bad %s: %s", lineno, cp, tp);
errors++;
} else
lp->d_rpm = v;
continue;
}
if (!strcmp(cp, "interleave")) {
if (GETNUM16(tp, &v) != 0) {
warnx("line %d: bad %s: %s", lineno, cp, tp);
errors++;
} else
lp->d_interleave = v;
continue;
}
if (!strcmp(cp, "trackskew")) {
if (GETNUM16(tp, &v) != 0) {
warnx("line %d: bad %s: %s", lineno, cp, tp);
errors++;
} else
lp->d_trackskew = v;
continue;
}
if (!strcmp(cp, "cylinderskew")) {
if (GETNUM16(tp, &v) != 0) {
warnx("line %d: bad %s: %s", lineno, cp, tp);
errors++;
} else
lp->d_cylskew = v;
continue;
}
if (!strcmp(cp, "headswitch")) {
if (GETNUM32(tp, &v) != 0) {
warnx("line %d: bad %s: %s", lineno, cp, tp);
errors++;
} else
lp->d_headswitch = v;
continue;
}
if (!strcmp(cp, "track-to-track seek")) {
if (GETNUM32(tp, &v) != 0) {
warnx("line %d: bad %s: %s", lineno, cp, tp);
errors++;
} else
lp->d_trkseek = v;
continue;
}
if ('a' > *cp || *cp > 'z' || cp[1] != '\0') {
warnx("line %d: unknown field: %s", lineno, cp);
errors++;
continue;
}
part = *cp - 'a';
if (part >= maxpartitions) {
warnx("line %d: bad partition name: %s", lineno, cp);
errors++;
continue;
}
if (part >= __arraycount(lp->d_partitions)) {
warnx("line %d: partition id %s, >= %zu", lineno,
cp, __arraycount(lp->d_partitions));
errors++;
continue;
}
pp = &lp->d_partitions[part];
NXTXNUM(pp->p_size);
NXTXNUM(pp->p_offset);
tp += strspn(tp, " \t");
_CHECKLINE
cp = tp;
cpp = fstypenames;
for (; cpp < &fstypenames[FSMAXTYPES]; cpp++) {
s = *cpp;
if (s == NULL ||
(cp[strlen(s)] != ' ' &&
cp[strlen(s)] != '\t' &&
cp[strlen(s)] != '\0'))
continue;
if (!memcmp(s, cp, strlen(s))) {
pp->p_fstype = cpp - fstypenames;
tp += strlen(s);
if (*tp == '\0')
tp = NULL;
else {
tp += strspn(tp, " \t");
if (*tp == '\0')
tp = NULL;
}
goto gottype;
}
}
tp = word(cp);
if (isdigit(*cp & 0xff)) {
if (GETNUM8(cp, &v) != 0) {
warnx("line %d: syntax error", lineno);
errors++;
}
} else
v = FSMAXTYPES;
if ((unsigned)v >= FSMAXTYPES) {
warnx("line %d: warning, unknown file system type: %s",
lineno, cp);
warnx("tip: use -l to see all valid file system "
"types");
v = FS_UNUSED;
}
pp->p_fstype = v;
gottype:
switch (pp->p_fstype) {
case FS_UNUSED:
NXTNUM(pp->p_fsize);
if (pp->p_fsize == 0)
break;
NXTNUM(v);
pp->p_frag = v / pp->p_fsize;
break;
case FS_BSDFFS:
case FS_ADOS:
case FS_APPLEUFS:
NXTNUM(pp->p_fsize);
if (pp->p_fsize == 0)
break;
NXTNUM(v);
pp->p_frag = v / pp->p_fsize;
NXTNUM(pp->p_cpg);
break;
case FS_BSDLFS:
NXTNUM(pp->p_fsize);
if (pp->p_fsize == 0)
break;
NXTNUM(v);
pp->p_frag = v / pp->p_fsize;
NXTNUM(pp->p_sgs);
break;
case FS_EX2FS:
NXTNUM(pp->p_fsize);
if (pp->p_fsize == 0)
break;
NXTNUM(v);
pp->p_frag = v / pp->p_fsize;
break;
case FS_ISO9660:
NXTNUM(pp->p_cdsession);
break;
default:
break;
}
continue;
error:
errors++;
next:
;
}
errors += checklabel(lp);
return (errors == 0);
}
int
checklabel(struct disklabel *lp)
{
struct partition *pp, *qp;
int i, j, errors;
char part;
errors = 0;
if (lp->d_secsize == 0) {
warnx("sector size %" PRIu32, lp->d_secsize);
return (1);
}
if (lp->d_nsectors == 0) {
warnx("sectors/track %" PRIu32, lp->d_nsectors);
return (1);
}
if (lp->d_ntracks == 0) {
warnx("tracks/cylinder %" PRIu32, lp->d_ntracks);
return (1);
}
if (lp->d_ncylinders == 0) {
warnx("cylinders/unit %" PRIu32, lp->d_ncylinders);
errors++;
}
if (lp->d_rpm == 0)
warnx("warning, revolutions/minute %" PRIu16, lp->d_rpm);
if (lp->d_secpercyl == 0)
lp->d_secpercyl = lp->d_nsectors * lp->d_ntracks;
if (lp->d_secperunit == 0)
lp->d_secperunit = lp->d_secpercyl * lp->d_ncylinders;
if (lp->d_bbsize == 0) {
warnx("boot block size %" PRIu32, lp->d_bbsize);
errors++;
} else if (lp->d_bbsize % lp->d_secsize)
warnx("warning, boot block size %% sector-size != 0");
if (lp->d_sbsize == 0) {
warnx("super block size %" PRIu32, lp->d_sbsize);
errors++;
} else if (lp->d_sbsize % lp->d_secsize)
warnx("warning, super block size %% sector-size != 0");
if (lp->d_npartitions > maxpartitions)
warnx("warning, number of partitions (%" PRIu16 ") > "
"MAXPARTITIONS (%d)",
lp->d_npartitions, maxpartitions);
else
for (i = maxpartitions - 1; i >= lp->d_npartitions; i--) {
part = 'a' + i;
pp = &lp->d_partitions[i];
if (pp->p_size || pp->p_offset) {
warnx("warning, partition %c increased "
"number of partitions from %" PRIu16
" to %d",
part, lp->d_npartitions, i + 1);
lp->d_npartitions = i + 1;
break;
}
}
for (i = 0; i < lp->d_npartitions; i++) {
part = 'a' + i;
pp = &lp->d_partitions[i];
if (pp->p_size == 0 && pp->p_offset != 0)
warnx("warning, partition %c: size 0, but "
"offset %" PRIu32,
part, pp->p_offset);
#ifdef STRICT_CYLINDER_ALIGNMENT
if (pp->p_offset % lp->d_secpercyl) {
warnx("warning, partition %c:"
" not starting on cylinder boundary",
part);
errors++;
}
#endif
if (pp->p_offset > lp->d_secperunit) {
warnx("partition %c: offset past end of unit", part);
errors++;
}
if (pp->p_offset + pp->p_size > lp->d_secperunit) {
warnx("partition %c: partition extends"
" past end of unit",
part);
errors++;
}
if (pp->p_fstype != FS_UNUSED)
for (j = i + 1; j < lp->d_npartitions; j++) {
qp = &lp->d_partitions[j];
if (qp->p_fstype == FS_UNUSED)
continue;
if (pp->p_offset < qp->p_offset + qp->p_size &&
qp->p_offset < pp->p_offset + pp->p_size)
warnx("partitions %c and %c overlap",
part, 'a' + j);
}
}
return (errors);
}
static void
usage(void)
{
static const struct {
const char *name;
const char *expn;
} usages[] = {
{ "[-ABCFMrtv] disk", "(to read label)" },
{ "-w [-BDFMrv] [-f disktab] disk disktype [packid]", "(to write label)" },
{ "-e [-BCDFMIrv] disk", "(to edit label)" },
#if !defined(NO_INTERACT)
{ "-i [-BDFMIrv] disk", "(to create a label interactively)" },
#endif
{ "-D [-v] disk", "(to delete existing label(s))" },
{ "-R [-BDFMrv] disk protofile", "(to restore label)" },
{ "[-NW] disk", "(to write disable/enable label)" },
{ "-l", "(to show all known file system types)" },
{ NULL, NULL }
};
int i;
const char *pn = getprogname();
const char *t = "usage:";
for (i = 0; usages[i].name != NULL; i++) {
(void)fprintf(stderr, "%s %s %s\n\t%s\n",
t, pn, usages[i].name, usages[i].expn);
t = "or";
}
exit(1);
}
static int
getulong(const char *str, char sep, char **epp, unsigned long *ul,
unsigned long max)
{
char *ep;
if (epp == NULL)
epp = &ep;
*ul = strtoul(str, epp, 10);
if ((*ul == ULONG_MAX && errno == ERANGE) || *ul > max)
return ERANGE;
if (*str == '\0' || (**epp != '\0' && **epp != sep &&
!isspace((unsigned char)**epp)))
return EFTYPE;
return 0;
}
static int
qsort_strcmp(const void *v1, const void *v2)
{
const char *const *sp1 = (const char *const *)v1;
const char *const *sp2 = (const char *const *)v2;
return strcmp(*sp1, *sp2);
}
int
list_fs_types(void)
{
int ret;
size_t nelems;
nelems = 0;
{
const char *const *namep;
namep = fstypenames;
while (*namep++ != NULL)
nelems++;
}
ret = 1;
if (nelems > 0) {
const char **list = NULL;
size_t i;
if (reallocarr(&list, nelems, sizeof(char *)) != 0) {
warnx("sorry, could not allocate memory for list");
ret = 0;
} else {
for (i = 0; i < nelems; i++)
list[i] = fstypenames[i];
qsort(list, nelems, sizeof(char *), qsort_strcmp);
for (i = 0; i < nelems; i++)
(void)printf("%s\n", list[i]);
free(list);
}
}
return ret;
}
#ifndef HAVE_NBTOOL_CONFIG_H
int
dk_ioctl(int f, u_long cmd, void *arg)
{
#if !defined(NATIVELABEL_ONLY)
if (!native_p) {
errno = ENOTTY;
return -1;
}
#endif
return ioctl(f, cmd, arg);
}
#endif
