SecBSD
/
src
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
src/sbin/disklabel/disklabel.c

1130 lines
26 KiB
C
Raw Permalink Blame History

/* $OpenBSD: disklabel.c,v 1.254 2023/07/03 15:27:07 krw Exp $ */
/*
* Copyright (c) 1987, 1993
* The Regents of the University of California. All rights reserved.
*
* This code is derived from software contributed to Berkeley by
* Symmetric Computer Systems.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
* 3. Neither the name of the University nor the names of its contributors
* may be used to endorse or promote products derived from this software
* without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
* SUCH DAMAGE.
*/
#include <sys/param.h> /* DEV_BSIZE */
#include <sys/sysctl.h>
#include <sys/ioctl.h>
#include <sys/dkio.h>
#include <sys/stat.h>
#include <sys/wait.h>
#define DKTYPENAMES
#include <sys/disklabel.h>
#include <ufs/ffs/fs.h>
#include <ctype.h>
#include <err.h>
#include <errno.h>
#include <fcntl.h>
#include <limits.h>
#include <signal.h>
#include <string.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <util.h>
#include <fstab.h>
#include "pathnames.h"
#include "extern.h"
/*
* Disklabel: read and write disklabels.
* The label is usually placed on one of the first sectors of the disk.
* Many machines also place a bootstrap in the same area,
* in which case the label is embedded in the bootstrap.
* The bootstrap source must leave space at the proper offset
* for the label on such machines.
*/
#ifndef BBSIZE
#define BBSIZE 8192 /* size of boot area, with label */
#endif
char *dkname, *specname, *fstabfile;
char tmpfil[] = _PATH_TMPFILE;
char *mountpoints[MAXPARTITIONS];
struct disklabel lab;
enum {
UNSPEC, EDIT, EDITOR, READ, RESTORE, WRITE
} op = UNSPEC;
int aflag;
int cflag;
int dflag;
int tflag;
int uidflag;
int verbose;
int quiet;
int donothing;
void makedisktab(FILE *, struct disklabel *);
int checklabel(struct disklabel *);
void readlabel(int);
void parsefstab(void);
void parsedisktab(char *, struct disklabel *);
int edit(struct disklabel *, int);
int editit(const char *);
char *skip(char *);
char *word(char *);
int getasciilabel(FILE *, struct disklabel *);
int cmplabel(struct disklabel *, struct disklabel *);
void usage(void);
u_int64_t getnum(char *, u_int64_t, u_int64_t, const char **);
int64_t physmem;
void
getphysmem(void)
{
size_t sz = sizeof(physmem);
int mib[] = { CTL_HW, HW_PHYSMEM64 };
if (sysctl(mib, 2, &physmem, &sz, NULL, (size_t)0) == -1)
errx(4, "can't get mem size");
}
int
main(int argc, char *argv[])
{
FILE *t;
char *autotable = NULL;
int ch, f, error = 0;
char print_unit = '\0';
getphysmem();
while ((ch = getopt(argc, argv, "AEf:F:hRcdenp:tT:vw")) != -1)
switch (ch) {
case 'A':
aflag = 1;
break;
case 'R':
if (op != UNSPEC)
usage();
op = RESTORE;
break;
case 'c':
cflag = 1;
break;
case 'd':
dflag = 1;
break;
case 'e':
if (op != UNSPEC)
usage();
op = EDIT;
break;
case 'E':
if (op != UNSPEC)
usage();
op = EDITOR;
break;
case 'f':
fstabfile = optarg;
uidflag = 0;
break;
case 'F':
fstabfile = optarg;
uidflag = 1;
break;
case 'h':
print_unit = '*';
break;
case 't':
tflag = 1;
break;
case 'T':
autotable = optarg;
break;
case 'w':
if (op != UNSPEC)
usage();
op = WRITE;
break;
case 'p':
if (strchr("bckmgtBCKMGT", optarg[0]) == NULL ||
optarg[1] != '\0') {
fprintf(stderr, "Valid units are bckmgt\n");
return 1;
}
print_unit = tolower((unsigned char)optarg[0]);
break;
case 'n':
donothing = 1;
break;
case 'v':
verbose = 1;
break;
default:
usage();
}
argc -= optind;
argv += optind;
if (op == UNSPEC)
op = READ;
if (argc < 1 || (fstabfile && !(op == EDITOR || op == RESTORE ||
aflag)))
usage();
if (argv[0] == NULL)
usage();
dkname = argv[0];
f = opendev(dkname, (op == READ ? O_RDONLY : O_RDWR), OPENDEV_PART,
&specname);
if (f == -1)
err(4, "%s", specname);
if (op != WRITE || aflag || dflag) {
readlabel(f);
if (op == EDIT || op == EDITOR || aflag) {
if (pledge("stdio rpath wpath cpath disklabel proc "
"exec", NULL) == -1)
err(1, "pledge");
} else if (fstabfile) {
if (pledge("stdio rpath wpath cpath disklabel", NULL)
== -1)
err(1, "pledge");
} else {
if (pledge("stdio rpath wpath disklabel", NULL) == -1)
err(1, "pledge");
}
if (autotable != NULL)
parse_autotable(autotable);
parsefstab();
error = aflag ? editor_allocspace(&lab) : 0;
if (op == WRITE && error)
errx(1, "autoalloc failed");
} else if (argc == 2 || argc == 3) {
/* Ensure f is a disk device before pledging. */
if (ioctl(f, DIOCGDINFO, &lab) == -1)
err(4, "DIOCGDINFO");
if (pledge("stdio rpath wpath disklabel", NULL) == -1)
err(1, "pledge");
parsedisktab(argv[1], &lab);
if (argc == 3)
strncpy(lab.d_packname, argv[2], sizeof(lab.d_packname));
} else
usage();
switch (op) {
case EDIT:
if (argc != 1)
usage();
error = edit(&lab, f);
break;
case EDITOR:
if (argc != 1)
usage();
error = editor(f);
break;
case READ:
if (argc != 1)
usage();
if (pledge("stdio", NULL) == -1)
err(1, "pledge");
if (tflag)
makedisktab(stdout, &lab);
else
display(stdout, &lab, print_unit, 1);
error = checklabel(&lab);
break;
case RESTORE:
if (argc < 2 || argc > 3)
usage();
if (!(t = fopen(argv[1], "r")))
err(4, "%s", argv[1]);
error = getasciilabel(t, &lab);
if (error == 0) {
memset(&lab.d_uid, 0, sizeof(lab.d_uid));
error = writelabel(f, &lab);
}
fclose(t);
break;
case WRITE:
error = checklabel(&lab);
if (error == 0)
error = writelabel(f, &lab);
break;
default:
break;
}
return error;
}
/*
* Construct a prototype disklabel from /etc/disktab.
*/
void
parsedisktab(char *type, struct disklabel *lp)
{
struct disklabel *dp;
dp = getdiskbyname(type);
if (dp == NULL)
errx(1, "unknown disk type: %s", type);
*lp = *dp;
}
int
writelabel(int f, struct disklabel *lp)
{
lp->d_magic = DISKMAGIC;
lp->d_magic2 = DISKMAGIC;
lp->d_checksum = 0;
lp->d_checksum = dkcksum(lp);
if (!donothing) {
/* Write new label to disk. */
if (ioctl(f, DIOCWDINFO, lp) == -1) {
warn("DIOCWDINFO");
return 1;
}
/* Refresh our copy of the on-disk current label to get UID. */
if (ioctl(f, DIOCGDINFO, &lab) == -1)
err(4, "DIOCGDINFO");
/* Finally, write out any mount point information. */
mpsave(lp);
}
return 0;
}
/*
* Fetch requested disklabel into 'lab' using ioctl.
*/
void
readlabel(int f)
{
struct disklabel dl;
if (cflag && ioctl(f, DIOCRLDINFO) == -1)
err(4, "DIOCRLDINFO");
if ((op == RESTORE) || dflag || aflag) {
if (ioctl(f, DIOCGPDINFO, &lab) == -1)
err(4, "DIOCGPDINFO");
} else {
if (ioctl(f, DIOCGDINFO, &lab) == -1)
err(4, "DIOCGDINFO");
if (ioctl(f, DIOCGPDINFO, &dl) == -1)
err(4, "DIOCGPDINFO");
lab.d_secsize = dl.d_secsize;
lab.d_nsectors = dl.d_nsectors;
lab.d_ntracks = dl.d_ntracks;
lab.d_secpercyl = dl.d_secpercyl;
lab.d_ncylinders = dl.d_ncylinders;
lab.d_type = dl.d_type;
}
}
void
parsefstab(void)
{
char *partname, *partduid;
struct fstab *fsent;
int i;
i = asprintf(&partname, "/dev/%s%c", dkname, 'a');
if (i == -1)
err(1, NULL);
i = asprintf(&partduid,
"%02hhx%02hhx%02hhx%02hhx%02hhx%02hhx%02hhx%02hhx.a",
lab.d_uid[0], lab.d_uid[1], lab.d_uid[2], lab.d_uid[3],
lab.d_uid[4], lab.d_uid[5], lab.d_uid[6], lab.d_uid[7]);
if (i == -1)
err(1, NULL);
setfsent();
for (i = 0; i < MAXPARTITIONS; i++) {
partname[strlen(dkname) + 5] = 'a' + i;
partduid[strlen(partduid) - 1] = 'a' + i;
fsent = getfsspec(partname);
if (fsent == NULL)
fsent = getfsspec(partduid);
if (fsent)
mountpoints[i] = strdup(fsent->fs_file);
}
endfsent();
free(partduid);
free(partname);
}
void
makedisktab(FILE *f, struct disklabel *lp)
{
int i;
struct partition *pp;
if (lp->d_packname[0])
(void)fprintf(f, "%.*s|", (int)sizeof(lp->d_packname),
lp->d_packname);
if (lp->d_typename[0])
(void)fprintf(f, "%.*s|", (int)sizeof(lp->d_typename),
lp->d_typename);
(void)fputs("Automatically generated label:\\\n\t:dt=", f);
if (lp->d_type < DKMAXTYPES)
(void)fprintf(f, "%s:", dktypenames[lp->d_type]);
else
(void)fprintf(f, "unknown%d:", lp->d_type);
(void)fprintf(f, "se#%u:", lp->d_secsize);
(void)fprintf(f, "ns#%u:", lp->d_nsectors);
(void)fprintf(f, "nt#%u:", lp->d_ntracks);
(void)fprintf(f, "nc#%u:", lp->d_ncylinders);
(void)fprintf(f, "sc#%u:", lp->d_secpercyl);
(void)fprintf(f, "su#%llu:", DL_GETDSIZE(lp));
/*
* XXX We do not print have disktab information yet for
* XXX DL_GETBSTART DL_GETBEND
*/
pp = lp->d_partitions;
for (i = 0; i < lp->d_npartitions; i++, pp++) {
if (DL_GETPSIZE(pp)) {
char c = 'a' + i;
(void)fprintf(f, "\\\n\t:");
(void)fprintf(f, "p%c#%llu:", c, DL_GETPSIZE(pp));
(void)fprintf(f, "o%c#%llu:", c, DL_GETPOFFSET(pp));
if (pp->p_fstype != FS_UNUSED) {
if (pp->p_fstype < FSMAXTYPES)
(void)fprintf(f, "t%c=%s:", c,
fstypenames[pp->p_fstype]);
else
(void)fprintf(f, "t%c=unknown%d:",
c, pp->p_fstype);
}
switch (pp->p_fstype) {
case FS_UNUSED:
break;
case FS_BSDFFS:
(void)fprintf(f, "b%c#%u:", c,
DISKLABELV1_FFS_BSIZE(pp->p_fragblock));
(void)fprintf(f, "f%c#%u:", c,
DISKLABELV1_FFS_FSIZE(pp->p_fragblock));
break;
default:
break;
}
}
}
(void)fputc('\n', f);
(void)fflush(f);
}
double
scale(u_int64_t sz, char unit, const struct disklabel *lp)
{
double fsz;
fsz = (double)sz * lp->d_secsize;
switch (unit) {
case 'B':
return fsz;
case 'C':
return fsz / lp->d_secsize / lp->d_secpercyl;
case 'K':
return fsz / 1024;
case 'M':
return fsz / (1024 * 1024);
case 'G':
return fsz / (1024 * 1024 * 1024);
case 'T':
return fsz / (1024ULL * 1024 * 1024 * 1024);
default:
return -1.0;
}
}
/*
* Display a particular partition.
*/
void
display_partition(FILE *f, const struct disklabel *lp, int i, char unit)
{
const struct partition *pp = &lp->d_partitions[i];
double p_size;
p_size = scale(DL_GETPSIZE(pp), unit, lp);
if (DL_GETPSIZE(pp)) {
u_int32_t frag = DISKLABELV1_FFS_FRAG(pp->p_fragblock);
u_int32_t fsize = DISKLABELV1_FFS_FSIZE(pp->p_fragblock);
if (p_size < 0)
fprintf(f, " %c: %16llu %16llu ", 'a' + i,
DL_GETPSIZE(pp), DL_GETPOFFSET(pp));
else
fprintf(f, " %c: %15.*f%c %16llu ", 'a' + i,
unit == 'B' ? 0 : 1, p_size, unit,
DL_GETPOFFSET(pp));
if (pp->p_fstype < FSMAXTYPES)
fprintf(f, "%7.7s", fstypenames[pp->p_fstype]);
else
fprintf(f, "%7d", pp->p_fstype);
switch (pp->p_fstype) {
case FS_BSDFFS:
fprintf(f, " %5u %5u %5hu ",
fsize, fsize * frag,
pp->p_cpg);
break;
default:
fprintf(f, "%20.20s", "");
break;
}
if (mountpoints[i] != NULL)
fprintf(f, "# %s", mountpoints[i]);
putc('\n', f);
}
}
char
canonical_unit(const struct disklabel *lp, char unit)
{
const struct partition *pp;
u_int64_t small;
int i;
if (unit == '*') {
small = DL_GETDSIZE(lp);
pp = &lp->d_partitions[0];
for (i = 0; i < lp->d_npartitions; i++, pp++)
if (DL_GETPSIZE(pp) > 0 && DL_GETPSIZE(pp) < small)
small = DL_GETPSIZE(pp);
if (small < DL_BLKTOSEC(lp, MEG(1)))
unit = 'K';
else if (small < DL_BLKTOSEC(lp, MEG(1024)))
unit = 'M';
else if (small < DL_BLKTOSEC(lp, GIG(1024)))
unit = 'G';
else
unit = 'T';
}
unit = toupper((unsigned char)unit);
return unit;
}
void
display(FILE *f, const struct disklabel *lp, char unit, int all)
{
int i;
double d;
unit = canonical_unit(lp, unit);
fprintf(f, "# %s:\n", specname);
if (lp->d_type < DKMAXTYPES)
fprintf(f, "type: %s\n", dktypenames[lp->d_type]);
else
fprintf(f, "type: %d\n", lp->d_type);
fprintf(f, "disk: %.*s\n", (int)sizeof(lp->d_typename),
lp->d_typename);
fprintf(f, "label: %.*s\n", (int)sizeof(lp->d_packname),
lp->d_packname);
fprintf(f, "duid: %02hhx%02hhx%02hhx%02hhx%02hhx%02hhx%02hhx%02hhx\n",
lp->d_uid[0], lp->d_uid[1], lp->d_uid[2], lp->d_uid[3],
lp->d_uid[4], lp->d_uid[5], lp->d_uid[6], lp->d_uid[7]);
fprintf(f, "flags:");
if (lp->d_flags & D_VENDOR)
fprintf(f, " vendor");
putc('\n', f);
fprintf(f, "bytes/sector: %u\n", lp->d_secsize);
fprintf(f, "sectors/track: %u\n", lp->d_nsectors);
fprintf(f, "tracks/cylinder: %u\n", lp->d_ntracks);
fprintf(f, "sectors/cylinder: %u\n", lp->d_secpercyl);
fprintf(f, "cylinders: %u\n", lp->d_ncylinders);
fprintf(f, "total sectors: %llu", DL_GETDSIZE(lp));
d = scale(DL_GETDSIZE(lp), unit, lp);
if (d > 0)
fprintf(f, " # total bytes: %.*f%c", unit == 'B' ? 0 : 1,
d, unit);
fprintf(f, "\n");
fprintf(f, "boundstart: %llu\n", DL_GETBSTART(lp));
fprintf(f, "boundend: %llu\n", DL_GETBEND(lp));
if (all) {
fprintf(f, "\n%hu partitions:\n", lp->d_npartitions);
fprintf(f, "# %16.16s %16.16s fstype [fsize bsize cpg]\n",
"size", "offset");
for (i = 0; i < lp->d_npartitions; i++)
display_partition(f, lp, i, unit);
}
fflush(f);
}
int
edit(struct disklabel *lp, int f)
{
int first, ch, fd, error = 0;
struct disklabel label;
FILE *fp;
if ((fd = mkstemp(tmpfil)) == -1 || (fp = fdopen(fd, "w")) == NULL) {
warn("%s", tmpfil);
if (fd != -1)
close(fd);
return 1;
}
display(fp, lp, 0, 1);
fprintf(fp, "\n# Notes:\n");
fprintf(fp,
"# Up to 16 partitions are valid, named from 'a' to 'p'. Partition 'a' is\n"
"# your root filesystem, 'b' is your swap, and 'c' should cover your whole\n"
"# disk. Any other partition is free for any use. 'size' and 'offset' are\n"
"# in 512-byte blocks. fstype should be '4.2BSD', 'swap', or 'none' or some\n"
"# other values. fsize/bsize/cpg should typically be '2048 16384 16' for a\n"
"# 4.2BSD filesystem (or '512 4096 16' except on alpha, sun4, ...)\n");
fclose(fp);
for (;;) {
if (editit(tmpfil) == -1)
break;
fp = fopen(tmpfil, "r");
if (fp == NULL) {
warn("%s", tmpfil);
break;
}
/* Start with the kernel's idea of the default label. */
if (ioctl(f, DIOCGPDINFO, &label) == -1)
err(4, "DIOCGPDINFO");
error = getasciilabel(fp, &label);
if (error == 0) {
if (cmplabel(lp, &label) == 0) {
puts("No changes.");
fclose(fp);
(void) unlink(tmpfil);
return 0;
}
*lp = label;
if (writelabel(f, lp) == 0) {
fclose(fp);
(void) unlink(tmpfil);
return 0;
}
}
fclose(fp);
printf("re-edit the label? [y]: ");
fflush(stdout);
first = ch = getchar();
while (ch != '\n' && ch != EOF)
ch = getchar();
if (first == 'n' || first == 'N')
break;
}
(void)unlink(tmpfil);
return 1;
}
/*
* Execute an editor on the specified pathname, which is interpreted
* from the shell. This means flags may be included.
*
* Returns -1 on error, or the exit value on success.
*/
int
editit(const char *pathname)
{
char *argp[] = {"sh", "-c", NULL, NULL}, *ed, *p;
sig_t sighup, sigint, sigquit, sigchld;
pid_t pid;
int saved_errno, st, ret = -1;
ed = getenv("VISUAL");
if (ed == NULL || ed[0] == '\0')
ed = getenv("EDITOR");
if (ed == NULL || ed[0] == '\0')
ed = _PATH_VI;
if (asprintf(&p, "%s %s", ed, pathname) == -1)
return -1;
argp[2] = p;
sighup = signal(SIGHUP, SIG_IGN);
sigint = signal(SIGINT, SIG_IGN);
sigquit = signal(SIGQUIT, SIG_IGN);
sigchld = signal(SIGCHLD, SIG_DFL);
if ((pid = fork()) == -1)
goto fail;
if (pid == 0) {
execv(_PATH_BSHELL, argp);
_exit(127);
}
while (waitpid(pid, &st, 0) == -1)
if (errno != EINTR)
goto fail;
if (!WIFEXITED(st))
errno = EINTR;
else
ret = WEXITSTATUS(st);
fail:
saved_errno = errno;
(void)signal(SIGHUP, sighup);
(void)signal(SIGINT, sigint);
(void)signal(SIGQUIT, sigquit);
(void)signal(SIGCHLD, sigchld);
free(p);
errno = saved_errno;
return ret;
}
char *
skip(char *cp)
{
cp += strspn(cp, " \t");
if (*cp == '\0')
return NULL;
return cp;
}
char *
word(char *cp)
{
cp += strcspn(cp, " \t");
if (*cp == '\0')
return NULL;
*cp++ = '\0';
cp += strspn(cp, " \t");
if (*cp == '\0')
return NULL;
return cp;
}
/* Base the max value on the sizeof of the value we are reading */
#define GETNUM(field, nptr, min, errstr) \
getnum((nptr), (min), \
sizeof(field) == 8 ? LLONG_MAX : \
(sizeof(field) == 4 ? UINT_MAX : \
(sizeof(field) == 2 ? USHRT_MAX : UCHAR_MAX)), (errstr))
u_int64_t
getnum(char *nptr, u_int64_t min, u_int64_t max, const char **errstr)
{
char *p, c;
u_int64_t ret;
for (p = nptr; *p != '\0' && !isspace((unsigned char)*p); p++)
;
c = *p;
*p = '\0';
ret = strtonum(nptr, min, max, errstr);
*p = c;
return ret;
}
int
duid_parse(struct disklabel *lp, char *s)
{
u_char duid[8];
char c;
int i;
if (strlen(s) != 16)
return -1;
memset(duid, 0, sizeof(duid));
for (i = 0; i < 16; i++) {
c = s[i];
if (c >= '0' && c <= '9')
c -= '0';
else if (c >= 'a' && c <= 'f')
c -= ('a' - 10);
else if (c >= 'A' && c <= 'F')
c -= ('A' - 10);
else
return -1;
duid[i / 2] <<= 4;
duid[i / 2] |= c & 0xf;
}
memcpy(lp->d_uid, &duid, sizeof(lp->d_uid));
return 0;
}
/*
* Read an ascii label in from FILE f,
* in the same format as that put out by display(),
* and fill in lp.
*/
int
getasciilabel(FILE *f, struct disklabel *lp)
{
const char * const *cpp;
const char *s;
char *cp;
const char *errstr;
struct partition *pp;
char *mp, *tp, line[BUFSIZ];
char **omountpoints = NULL;
int lineno = 0, errors = 0;
u_int32_t v, fsize;
u_int64_t lv;
unsigned int part;
lp->d_version = 1;
if (!(omountpoints = calloc(MAXPARTITIONS, sizeof(char *))))
err(1, NULL);
mpcopy(omountpoints, mountpoints);
for (part = 0; part < MAXPARTITIONS; part++) {
free(mountpoints[part]);
mountpoints[part] = NULL;
}
while (fgets(line, sizeof(line), f)) {
lineno++;
mp = NULL;
if ((cp = strpbrk(line, "\r\n")))
*cp = '\0';
if ((cp = strpbrk(line, "#"))) {
*cp = '\0';
mp = skip(cp+1);
if (mp && *mp != '/')
mp = NULL;
}
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, "flags")) {
for (v = 0; (cp = tp) && *cp != '\0';) {
tp = word(cp);
if (!strcmp(cp, "badsect"))
; /* Ignore obsolete flag. */
else if (!strcmp(cp, "vendor"))
v |= D_VENDOR;
else {
warnx("line %d: bad flag: %s",
lineno, cp);
errors++;
}
}
lp->d_flags = v;
continue;
}
if (sscanf(cp, "%d 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)
tp = "";
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, "duid")) {
if (duid_parse(lp, tp) != 0) {
warnx("line %d: bad %s: %s", lineno, cp, tp);
errors++;
}
continue;
}
/* Ignore fields that are no longer used. */
if (!strcmp(cp, "rpm") ||
!strcmp(cp, "interleave") ||
!strcmp(cp, "trackskew") ||
!strcmp(cp, "cylinderskew") ||
!strcmp(cp, "headswitch") ||
!strcmp(cp, "track-to-track seek") ||
!strcmp(cp, "drivedata"))
continue;
/* Ignore fields that are forcibly set when label is read. */
if (!strcmp(cp, "total sectors") ||
!strcmp(cp, "boundstart") ||
!strcmp(cp, "boundend") ||
!strcmp(cp, "bytes/sector") ||
!strcmp(cp, "sectors/track") ||
!strcmp(cp, "sectors/cylinder") ||
!strcmp(cp, "tracks/cylinder") ||
!strcmp(cp, "cylinders") ||
!strcmp(cp, "type"))
continue;
if ('a' <= *cp && *cp <= 'z' && cp[1] == '\0') {
unsigned int part = *cp - 'a';
if (part >= lp->d_npartitions) {
if (part >= MAXPARTITIONS) {
warnx("line %d: bad partition name: %s",
lineno, cp);
errors++;
continue;
} else {
lp->d_npartitions = part + 1;
}
}
pp = &lp->d_partitions[part];
#define NXTNUM(n, field, errstr) { \
if (tp == NULL || *tp == '\0') { \
warnx("line %d: too few fields", lineno); \
errors++; \
break; \
} else \
cp = tp, tp = word(cp), (n) = GETNUM(field, cp, 0, errstr); \
}
NXTNUM(lv, lv, &errstr);
if (errstr) {
warnx("line %d: bad partition size: %s",
lineno, cp);
errors++;
} else {
DL_SETPSIZE(pp, lv);
}
NXTNUM(lv, lv, &errstr);
if (errstr) {
warnx("line %d: bad partition offset: %s",
lineno, cp);
errors++;
} else {
DL_SETPOFFSET(pp, lv);
}
if (tp == NULL) {
pp->p_fstype = FS_UNUSED;
goto gottype;
}
cp = tp, tp = word(cp);
cpp = fstypenames;
for (; cpp < &fstypenames[FSMAXTYPES]; cpp++)
if ((s = *cpp) && !strcasecmp(s, cp)) {
pp->p_fstype = cpp - fstypenames;
goto gottype;
}
if (isdigit((unsigned char)*cp))
v = GETNUM(pp->p_fstype, cp, 0, &errstr);
else
v = FSMAXTYPES;
if (errstr || v >= FSMAXTYPES) {
warnx("line %d: warning, unknown filesystem type: %s",
lineno, cp);
v = FS_UNUSED;
}
pp->p_fstype = v;
gottype:
switch (pp->p_fstype) {
case FS_UNUSED: /* XXX */
if (tp == NULL) /* ok to skip fsize/bsize */
break;
NXTNUM(fsize, fsize, &errstr);
if (fsize == 0)
break;
NXTNUM(v, v, &errstr);
pp->p_fragblock =
DISKLABELV1_FFS_FRAGBLOCK(fsize, v / fsize);
break;
case FS_BSDFFS:
NXTNUM(fsize, fsize, &errstr);
if (fsize == 0)
break;
NXTNUM(v, v, &errstr);
pp->p_fragblock =
DISKLABELV1_FFS_FRAGBLOCK(fsize, v / fsize);
NXTNUM(pp->p_cpg, pp->p_cpg, &errstr);
break;
default:
break;
}
if (mp)
mountpoints[part] = strdup(mp);
continue;
}
warnx("line %d: unknown field: %s", lineno, cp);
errors++;
}
errors += checklabel(lp);
if (errors > 0)
mpcopy(mountpoints, omountpoints);
mpfree(omountpoints, DISCARD);
return errors > 0;
}
/*
* Check disklabel for errors and fill in
* derived fields according to supplied values.
*/
int
checklabel(struct disklabel *lp)
{
struct partition *pp;
int i, errors = 0;
char part;
if (lp->d_secsize == 0) {
warnx("sector size 0");
return 1;
}
if (lp->d_nsectors == 0) {
warnx("sectors/track 0");
return 1;
}
if (lp->d_ntracks == 0) {
warnx("tracks/cylinder 0");
return 1;
}
if (lp->d_ncylinders == 0) {
warnx("cylinders/unit 0");
errors++;
}
if (lp->d_secpercyl == 0)
lp->d_secpercyl = lp->d_nsectors * lp->d_ntracks;
if (DL_GETDSIZE(lp) == 0)
DL_SETDSIZE(lp, (u_int64_t)lp->d_secpercyl * lp->d_ncylinders);
if (lp->d_npartitions > MAXPARTITIONS)
warnx("warning, number of partitions (%d) > MAXPARTITIONS (%d)",
lp->d_npartitions, MAXPARTITIONS);
for (i = 0; i < lp->d_npartitions; i++) {
part = 'a' + i;
pp = &lp->d_partitions[i];
if (DL_GETPSIZE(pp) == 0 && DL_GETPOFFSET(pp) != 0)
warnx("warning, partition %c: size 0, but offset %llu",
part, DL_GETPOFFSET(pp));
#ifdef SUN_CYLCHECK
if (lp->d_flags & D_VENDOR) {
if (i != RAW_PART && DL_GETPSIZE(pp) % lp->d_secpercyl)
warnx("warning, partition %c: size %% "
"cylinder-size != 0", part);
if (i != RAW_PART && DL_GETPOFFSET(pp) % lp->d_secpercyl)
warnx("warning, partition %c: offset %% "
"cylinder-size != 0", part);
}
#endif
#ifdef SUN_AAT0
if ((lp->d_flags & D_VENDOR) &&
i == 0 && DL_GETPSIZE(pp) != 0 && DL_GETPOFFSET(pp) != 0) {
warnx("this architecture requires partition 'a' to "
"start at sector 0");
errors++;
}
#endif
if (DL_GETPOFFSET(pp) > DL_GETDSIZE(lp)) {
warnx("partition %c: offset past end of unit", part);
errors++;
} else if (DL_GETPOFFSET(pp) + DL_GETPSIZE(pp) >
DL_GETDSIZE(lp)) {
warnx("partition %c: extends past end of unit",
part);
errors++;
}
#if 0
if (pp->p_frag == 0 && pp->p_fsize != 0) {
warnx("partition %c: block size < fragment size", part);
errors++;
}
#endif
}
for (; i < MAXPARTITIONS; i++) {
part = 'a' + i;
pp = &lp->d_partitions[i];
if (DL_GETPSIZE(pp) || DL_GETPOFFSET(pp))
warnx("warning, unused partition %c: size %llu "
"offset %llu", part, DL_GETPSIZE(pp),
DL_GETPOFFSET(pp));
}
return errors > 0;
}
int
cmplabel(struct disklabel *lp1, struct disklabel *lp2)
{
struct disklabel lab1 = *lp1;
struct disklabel lab2 = *lp2;
/* We don't compare these fields */
lab1.d_magic = lab2.d_magic;
lab1.d_magic2 = lab2.d_magic2;
lab1.d_checksum = lab2.d_checksum;
lab1.d_bstart = lab2.d_bstart;
lab1.d_bstarth = lab2.d_bstarth;
lab1.d_bend = lab2.d_bend;
lab1.d_bendh = lab2.d_bendh;
return memcmp(&lab1, &lab2, sizeof(struct disklabel));
}
void
usage(void)
{
fprintf(stderr,
"usage: disklabel [-Acdtv] [-h | -p unit] [-T file] disk\n");
fprintf(stderr,
" disklabel -w [-Acdnv] [-T file] disk disktype [packid]\n");
fprintf(stderr,
" disklabel -e [-Acdnv] [-T file] disk\n");
fprintf(stderr,
" disklabel -E [-Acdnv] [-F|-f file] [-T file] disk\n");
fprintf(stderr,
" disklabel -R [-nv] [-F|-f file] disk protofile\n");
exit(1);
}
Reference in New Issue View Git Blame Copy Permalink