NetBSD/sbin/disklabel/interact.c

749 lines
16 KiB
C

/* $NetBSD: interact.c,v 1.23 2003/12/29 21:21:25 jdc Exp $ */
/*
* Copyright (c) 1997 Christos Zoulas. All rights reserved.
*
* 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. All advertising materials mentioning features or use of this software
* must display the following acknowledgement:
* This product includes software developed by Christos Zoulas.
* 4. The name of the author may not be used to endorse or promote products
* derived from this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``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 AUTHOR 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/cdefs.h>
#ifndef lint
__RCSID("$NetBSD: interact.c,v 1.23 2003/12/29 21:21:25 jdc Exp $");
#endif /* lint */
#include <sys/param.h>
#define FSTYPENAMES
#define DKTYPENAMES
#include <sys/disklabel.h>
#include <err.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <util.h>
#include "extern.h"
static void cmd_help(struct disklabel *, char *, int);
static void cmd_chain(struct disklabel *, char *, int);
static void cmd_print(struct disklabel *, char *, int);
static void cmd_printall(struct disklabel *, char *, int);
static void cmd_info(struct disklabel *, char *, int);
static void cmd_part(struct disklabel *, char *, int);
static void cmd_label(struct disklabel *, char *, int);
static void cmd_round(struct disklabel *, char *, int);
static void cmd_name(struct disklabel *, char *, int);
static int runcmd(struct disklabel *, char *, int);
static int getinput(const char *, const char *, const char *, char *);
static int alphacmp(const void *, const void *);
static void defnum(struct disklabel *, char *, uint32_t);
static void dumpnames(const char *, const char * const *, size_t);
static int getnum(struct disklabel *, char *, int);
static int rounding = 0; /* sector rounding */
static int chaining = 0; /* make partitions contiguous */
static struct cmds {
const char *name;
void (*func)(struct disklabel *, char *, int);
const char *help;
} cmds[] = {
{ "?", cmd_help, "print this menu" },
{ "C", cmd_chain, "make partitions contiguous" },
{ "E", cmd_printall, "print disk label and current partition table"},
{ "I", cmd_info, "change label information" },
{ "N", cmd_name, "name the label" },
{ "P", cmd_print, "print current partition table" },
{ "Q", NULL, "quit" },
{ "R", cmd_round, "rounding (c)ylinders (s)ectors" },
{ "W", cmd_label, "write the current partition table" },
{ NULL, NULL, NULL }
};
static void
cmd_help(struct disklabel *lp, char *s, int fd)
{
struct cmds *cmd;
for (cmd = cmds; cmd->name != NULL; cmd++)
printf("%s\t%s\n", cmd->name, cmd->help);
printf("[a-%c]\tdefine named partition\n",
'a' + getmaxpartitions() - 1);
}
static void
cmd_chain(struct disklabel *lp, char *s, int fd)
{
int i;
char line[BUFSIZ];
i = getinput(":", "Automatically adjust partitions",
chaining ? "yes" : "no", line);
if (i <= 0)
return;
switch (line[0]) {
case 'y':
chaining = 1;
return;
case 'n':
chaining = 0;
return;
default:
printf("Invalid answer\n");
return;
}
}
static void
cmd_printall(struct disklabel *lp, char *s, int fd)
{
showinfo(stdout, lp, specname);
showpartitions(stdout, lp, Cflag);
}
static void
cmd_print(struct disklabel *lp, char *s, int fd)
{
showpartitions(stdout, lp, Cflag);
}
static void
cmd_info(struct disklabel *lp, char *s, int fd)
{
char line[BUFSIZ];
char def[BUFSIZ];
int v, i;
u_int32_t u;
printf("# Current values:\n");
showinfo(stdout, lp, specname);
/* d_type */
for (;;) {
i = lp->d_type;
if (i < 0 || i >= DKMAXTYPES)
i = 0;
snprintf(def, sizeof(def), "%s", dktypenames[i]);
i = getinput(":", "Disk type [?]", def, line);
if (i == -1)
return;
else if (i == 0)
break;
if (!strcmp(line, "?")) {
dumpnames("Supported disk types", dktypenames,
DKMAXTYPES);
continue;
}
for (i = 0; i < DKMAXTYPES; i++) {
if (!strcasecmp(dktypenames[i], line)) {
lp->d_type = i;
goto done_typename;
}
}
v = atoi(line);
if ((unsigned)v >= DKMAXTYPES) {
warnx("Unknown disk type: %s", line);
continue;
}
lp->d_type = v;
done_typename:
break;
}
/* d_typename */
snprintf(def, sizeof(def), "%.*s",
(int) sizeof(lp->d_typename), lp->d_typename);
i = getinput(":", "Disk name", def, line);
if (i == -1)
return;
else if (i == 1)
(void) strncpy(lp->d_typename, line, sizeof(lp->d_typename));
/* d_packname */
cmd_name(lp, s, fd);
/* d_npartitions */
for (;;) {
snprintf(def, sizeof(def), "%u", lp->d_npartitions);
i = getinput(":", "Number of partitions", def, line);
if (i == -1)
return;
else if (i == 0)
break;
if (sscanf(line, "%u", &u) != 1) {
printf("Invalid number of partitions `%s'\n", line);
continue;
}
lp->d_npartitions = u;
break;
}
/* d_secsize */
for (;;) {
snprintf(def, sizeof(def), "%u", lp->d_secsize);
i = getinput(":", "Sector size (bytes)", def, line);
if (i == -1)
return;
else if (i == 0)
break;
if (sscanf(line, "%u", &u) != 1) {
printf("Invalid sector size `%s'\n", line);
continue;
}
lp->d_secsize = u;
break;
}
/* d_nsectors */
for (;;) {
snprintf(def, sizeof(def), "%u", lp->d_nsectors);
i = getinput(":", "Number of sectors per track", def, line);
if (i == -1)
return;
else if (i == 0)
break;
if (sscanf(line, "%u", &u) != 1) {
printf("Invalid number of sectors `%s'\n", line);
continue;
}
lp->d_nsectors = u;
break;
}
/* d_ntracks */
for (;;) {
snprintf(def, sizeof(def), "%u", lp->d_ntracks);
i = getinput(":", "Number of tracks per cylinder", def, line);
if (i == -1)
return;
else if (i == 0)
break;
if (sscanf(line, "%u", &u) != 1) {
printf("Invalid number of tracks `%s'\n", line);
continue;
}
lp->d_ntracks = u;
break;
}
/* d_secpercyl */
for (;;) {
snprintf(def, sizeof(def), "%u", lp->d_secpercyl);
i = getinput(":", "Number of sectors/cylinder", def, line);
if (i == -1)
return;
else if (i == 0)
break;
if (sscanf(line, "%u", &u) != 1) {
printf("Invalid number of sector/cylinder `%s'\n",
line);
continue;
}
lp->d_secpercyl = u;
break;
}
/* d_ncylinders */
for (;;) {
snprintf(def, sizeof(def), "%u", lp->d_ncylinders);
i = getinput(":", "Total number of cylinders", def, line);
if (i == -1)
return;
else if (i == 0)
break;
if (sscanf(line, "%u", &u) != 1) {
printf("Invalid sector size `%s'\n", line);
continue;
}
lp->d_ncylinders = u;
break;
}
/* d_secperunit */
for (;;) {
snprintf(def, sizeof(def), "%u", lp->d_secperunit);
i = getinput(":", "Total number of sectors", def, line);
if (i == -1)
return;
else if (i == 0)
break;
if (sscanf(line, "%u", &u) != 1) {
printf("Invalid number of sectors `%s'\n", line);
continue;
}
lp->d_secperunit = u;
break;
}
/* d_rpm */
/* d_interleave */
for (;;) {
snprintf(def, sizeof(def), "%u", lp->d_interleave);
i = getinput(":", "Hardware sectors interleave", def, line);
if (i == -1)
return;
else if (i == 0)
break;
if (sscanf(line, "%u", &u) != 1) {
printf("Invalid sector interleave `%s'\n", line);
continue;
}
lp->d_interleave = u;
break;
}
/* d_trackskew */
for (;;) {
snprintf(def, sizeof(def), "%u", lp->d_trackskew);
i = getinput(":", "Sector 0 skew, per track", def, line);
if (i == -1)
return;
else if (i == 0)
break;
if (sscanf(line, "%u", &u) != 1) {
printf("Invalid track sector skew `%s'\n", line);
continue;
}
lp->d_trackskew = u;
break;
}
/* d_cylskew */
for (;;) {
snprintf(def, sizeof(def), "%u", lp->d_cylskew);
i = getinput(":", "Sector 0 skew, per cylinder", def, line);
if (i == -1)
return;
else if (i == 0)
break;
if (sscanf(line, "%u", &u) != 1) {
printf("Invalid cylinder sector `%s'\n", line);
continue;
}
lp->d_cylskew = u;
break;
}
/* d_headswitch */
for (;;) {
snprintf(def, sizeof(def), "%u", lp->d_headswitch);
i = getinput(":", "Head switch time (usec)", def, line);
if (i == -1)
return;
else if (i == 0)
break;
if (sscanf(line, "%u", &u) != 1) {
printf("Invalid head switch time `%s'\n", line);
continue;
}
lp->d_headswitch = u;
break;
}
/* d_trkseek */
for (;;) {
snprintf(def, sizeof(def), "%u", lp->d_trkseek);
i = getinput(":", "Track seek time (usec)", def, line);
if (i == -1)
return;
else if (i == 0)
break;
if (sscanf(line, "%u", &u) != 1) {
printf("Invalid track seek time `%s'\n", line);
continue;
}
lp->d_trkseek = u;
break;
}
}
static void
cmd_name(struct disklabel *lp, char *s, int fd)
{
char line[BUFSIZ];
char def[BUFSIZ];
int i;
snprintf(def, sizeof(def), "%.*s",
(int) sizeof(lp->d_packname), lp->d_packname);
i = getinput(":", "Label name", def, line);
if (i <= 0)
return;
(void) strncpy(lp->d_packname, line, sizeof(lp->d_packname));
}
static void
cmd_round(struct disklabel *lp, char *s, int fd)
{
int i;
char line[BUFSIZ];
i = getinput(":", "Rounding", rounding ? "cylinders" : "sectors", line);
if (i <= 0)
return;
switch (line[0]) {
case 'c':
rounding = 1;
return;
case 's':
rounding = 0;
return;
default:
printf("Rounding can be (c)ylinders or (s)ectors\n");
return;
}
}
static void
cmd_part(struct disklabel *lp, char *s, int fd)
{
int i;
char line[BUFSIZ];
char def[BUFSIZ];
int part;
struct partition *p, ps;
part = s[0] - 'a';
p = &lp->d_partitions[part];
if (part >= lp->d_npartitions)
lp->d_npartitions = part + 1;
(void)memcpy(&ps, p, sizeof(ps));
for (;;) {
i = p->p_fstype;
if (i < 0 || i >= FSMAXTYPES)
i = 0;
snprintf(def, sizeof(def), "%s", fstypenames[i]);
i = getinput(":", "Filesystem type [?]", def, line);
if (i == -1)
return;
else if (i == 0)
break;
if (!strcmp(line, "?")) {
dumpnames("Supported file system types",
fstypenames, FSMAXTYPES);
continue;
}
for (i = 0; i < FSMAXTYPES; i++)
if (!strcasecmp(line, fstypenames[i])) {
p->p_fstype = i;
goto done_typename;
}
printf("Invalid file system typename `%s'\n", line);
continue;
done_typename:
break;
}
for (;;) {
defnum(lp, def, p->p_offset);
i = getinput(":",
"Start offset ('x' to start after partition 'x')",
def, line);
if (i == -1)
return;
else if (i == 0)
break;
if (line[1] == '\0' &&
line[0] >= 'a' && line[0] < 'a' + getmaxpartitions()) {
struct partition *cp = lp->d_partitions;
if ((cp[line[0] - 'a'].p_offset +
cp[line[0] - 'a'].p_size) >= lp->d_secperunit) {
printf("Bad offset `%s'\n", line);
continue;
} else {
p->p_offset = cp[line[0] - 'a'].p_offset +
cp[line[0] - 'a'].p_size;
}
} else {
if ((i = getnum(lp, line, 0)) == -1) {
printf("Bad offset `%s'\n", line);
continue;
} else if (i > lp->d_secperunit) {
printf("Offset `%s' out of range\n", line);
continue;
}
p->p_offset = i;
}
break;
}
for (;;) {
defnum(lp, def, p->p_size);
i = getinput(":", "Partition size ('$' for all remaining)",
def, line);
if (i == -1)
return;
else if (i == 0)
break;
if ((i = getnum(lp, line, lp->d_secperunit - p->p_offset))
== -1) {
printf("Bad size `%s'\n", line);
continue;
} else if
((i + p->p_offset) > lp->d_secperunit) {
printf("Size `%s' out of range\n", line);
continue;
}
p->p_size = i;
break;
}
if (memcmp(&ps, p, sizeof(ps)))
showpartition(stdout, lp, part, Cflag);
if (chaining) {
int offs = -1;
struct partition *cp = lp->d_partitions;
for (i = 0; i < lp->d_npartitions; i++) {
if (cp[i].p_fstype != FS_UNUSED) {
if (offs != -1 && cp[i].p_offset != offs) {
cp[i].p_offset = offs;
showpartition(stdout, lp, i, Cflag);
}
offs = cp[i].p_offset + cp[i].p_size;
}
}
}
}
static void
cmd_label(struct disklabel *lp, char *s, int fd)
{
char line[BUFSIZ];
int i;
i = getinput("?", "Label disk", "n", line);
if (i <= 0 || (*line != 'y' && *line != 'Y') )
return;
if (checklabel(lp) != 0) {
printf("Label not written\n");
return;
}
if (writelabel(fd, bootarea, lp) != 0) {
printf("Label not written\n");
return;
}
printf("Label written\n");
}
static int
runcmd(struct disklabel *lp, char *line, int fd)
{
struct cmds *cmd;
for (cmd = cmds; cmd->name != NULL; cmd++)
if (strncmp(line, cmd->name, strlen(cmd->name)) == 0) {
if (cmd->func == NULL)
return -1;
(*cmd->func)(lp, line, fd);
return 0;
}
if (line[1] == '\0' &&
line[0] >= 'a' && line[0] < 'a' + getmaxpartitions()) {
cmd_part(lp, line, fd);
return 0;
}
printf("Unknown command %s\n", line);
return 1;
}
static int
getinput(const char *sep, const char *prompt, const char *def, char *line)
{
for (;;) {
printf("%s", prompt);
if (def)
printf(" [%s]", def);
printf("%s ", sep);
if (fgets(line, BUFSIZ, stdin) == NULL)
return -1;
if (line[0] == '\n' || line[0] == '\0') {
if (def)
return 0;
}
else {
char *p;
if ((p = strrchr(line, '\n')) != NULL)
*p = '\0';
return 1;
}
}
}
static int
alphacmp(const void *a, const void *b)
{
return (strcasecmp(*(const char **)a, *(const char **)b));
}
static void
dumpnames(const char *prompt, const char * const *olist, size_t numentries)
{
int i, j, w;
int columns, width, lines;
const char *p;
const char **list;
list = (const char **)malloc(sizeof(char *) * numentries);
width = 0;
printf("%s:\n", prompt);
for (i = 0; i < numentries; i++) {
list[i] = olist[i];
w = strlen(list[i]);
if (w > width)
width = w;
}
#if 0
for (i = 0; i < numentries; i++)
printf("%s%s", i == 0 ? "" : ", ", list[i]);
puts("");
#endif
(void)qsort(list, numentries, sizeof(char *), alphacmp);
width++; /* want two spaces between items */
width = (width + 8) &~ 7;
#define ttywidth 72
columns = ttywidth / width;
#undef ttywidth
if (columns == 0)
columns = 1;
lines = (numentries + columns - 1) / columns;
for (i = 0; i < lines; i++) {
for (j = 0; j < columns; j++) {
p = list[j * lines + i];
if (j == 0)
putc('\t', stdout);
if (p) {
fputs(p, stdout);
}
if (j * lines + i + lines >= numentries) {
putc('\n', stdout);
break;
}
w = strlen(p);
while (w < width) {
w = (w + 8) &~ 7;
putc('\t', stdout);
}
}
}
free(list);
}
static void
defnum(struct disklabel *lp, char *buf, uint32_t size)
{
(void) snprintf(buf, BUFSIZ, "%gc, %us, %gM",
size / (float) lp->d_secpercyl,
size, size * (lp->d_secsize / (float) (1024 * 1024)));
}
static int
getnum(struct disklabel *lp, char *buf, int max)
{
char *ep;
double d;
int rv;
if (max && buf[0] == '$' && buf[1] == 0)
return max;
d = strtod(buf, &ep);
if (buf == ep)
return -1;
#define ROUND(a) ((((a) / lp->d_secpercyl) + \
(((a) % lp->d_secpercyl) ? 1 : 0)) * lp->d_secpercyl)
switch (*ep) {
case '\0':
case 's':
rv = (int) d;
break;
case 'c':
rv = (int) (d * lp->d_secpercyl);
break;
case 'm':
case 'M':
rv = (int) (d * 1024 * 1024 / lp->d_secsize);
break;
default:
printf("Unit error %c\n", *ep);
return -1;
}
if (rounding)
return ROUND(rv);
else
return rv;
}
void
interact(struct disklabel *lp, int fd)
{
char line[BUFSIZ];
for (;;) {
if (getinput(">", "partition", NULL, line) == -1)
return;
if (runcmd(lp, line, fd) == -1)
return;
}
}