/*      $OpenBSD: disklabel.c,v 1.260 2025/10/17 17:52:32 deraadt 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] = DL_PARTNUM2NAME(i);
                partduid[strlen(partduid) - 1] = DL_PARTNUM2NAME(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 = DL_PARTNUM2NAME(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 ", DL_PARTNUM2NAME(i),
                            DL_GETPSIZE(pp), DL_GETPOFFSET(pp));
                else
                        fprintf(f, "  %c: %15.*f%c %16llu ", DL_PARTNUM2NAME(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 %d partitions are valid, named from 'a' to %s.  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",
            MAXPARTITIONS,
            MAXPARTITIONS == 16 ? "'p'" : "'z' and 'A' to 'Z'");
        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 (cp[1] == '\0') {
                        int part = DL_PARTNAME2NUM(*cp);

                        if (part == -1 || part >= lp->d_npartitions) {
                                if (part == -1 || 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 = DL_PARTNUM2NAME(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 = DL_PARTNUM2NAME(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);
}