NetBSD/sbin/disklabel/interact.c

754 lines
16 KiB
C

/* $NetBSD: interact.c,v 1.14 2000/08/14 22:37:08 lukem 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.14 2000/08/14 22:37:08 lukem 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 __P((struct disklabel *, char *, int));
static void cmd_chain __P((struct disklabel *, char *, int));
static void cmd_print __P((struct disklabel *, char *, int));
static void cmd_printall __P((struct disklabel *, char *, int));
static void cmd_info __P((struct disklabel *, char *, int));
static void cmd_part __P((struct disklabel *, char *, int));
static void cmd_label __P((struct disklabel *, char *, int));
static void cmd_round __P((struct disklabel *, char *, int));
static void cmd_name __P((struct disklabel *, char *, int));
static int runcmd __P((char *, struct disklabel *, int));
static int getinput __P((const char *, const char *, const char *, char *));
static int alphacmp __P((const void *, const void *));
static void defnum __P((char *, struct disklabel *, int));
static void dumpnames __P((const char *, const char * const *, size_t));
static int getnum __P((char *, int, struct disklabel *));
static int rounding = 0; /* sector rounding */
static int chaining = 0; /* make partitions contiguous */
static struct cmds {
const char *name;
void (*func) __P((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(lp, s, fd)
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(lp, s, fd)
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(lp, s, fd)
struct disklabel *lp;
char *s;
int fd;
{
showinfo(stdout, lp);
showpartitions(stdout, lp);
}
static void
cmd_print(lp, s, fd)
struct disklabel *lp;
char *s;
int fd;
{
showpartitions(stdout, lp);
}
static void
cmd_info(lp, s, fd)
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);
/* 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(lp, s, fd)
struct disklabel *lp;
char *s;
int fd;
{
char line[BUFSIZ];
int i;
snprintf(line, sizeof(line), "%.*s",
(int) sizeof(lp->d_packname), lp->d_packname);
i = getinput(":", "Label name", lp->d_packname, line);
if (i <= 0)
return;
(void) strncpy(lp->d_packname, line, sizeof(lp->d_packname));
}
static void
cmd_round(lp, s, fd)
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(lp, s, fd)
struct disklabel *lp;
char *s;
int fd;
{
int i;
char line[BUFSIZ];
char def[BUFSIZ];
int part = *s - 'a';
struct partition *p = &lp->d_partitions[part];
if (part >= lp->d_npartitions)
lp->d_npartitions = part + 1;
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(def, lp, p->p_offset);
i = getinput(":", "Start offset", def, line);
if (i == -1)
return;
else if (i == 0)
break;
if ((i = getnum(line, 0, lp)) == -1) {
printf("Bad offset `%s'\n", line);
continue;
}
p->p_offset = i;
break;
}
for (;;) {
defnum(def, lp, p->p_size);
i = getinput(":", "Partition size ('$' for all remaining)",
def, line);
if (i == -1)
return;
else if (i == 0)
break;
if ((i = getnum(line, lp->d_secperunit - p->p_offset, lp))
== -1) {
printf("Bad size `%s'\n", line);
continue;
}
p->p_size = i;
break;
}
if (chaining) {
int offs = p[0].p_offset + p[0].p_size;
p = lp->d_partitions;
part = getrawpartition();
for (i = 1; i < lp->d_npartitions; i++) {
if (i != part && p[i].p_fstype) {
p[i].p_offset = offs;
offs = p[i].p_offset + p[i].p_size;
}
}
}
}
static void
cmd_label(lp, s, fd)
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(line, lp, fd)
char *line;
struct disklabel *lp;
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(sep, prompt, def, line)
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(a, b)
const void *a, *b;
{
return (strcasecmp(*(const char **)a, *(const char **)b));
}
static void
dumpnames(prompt, olist, numentries)
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((void *)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(buf, lp, size)
char *buf;
struct disklabel *lp;
int size;
{
(void) snprintf(buf, BUFSIZ, "%gc, %ds, %gM",
size / (float) lp->d_secpercyl,
size, size * (lp->d_secsize / (float) (1024 * 1024)));
}
static int
getnum(buf, max, lp)
char *buf;
int max;
struct disklabel *lp;
{
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(lp, fd)
struct disklabel *lp;
int fd;
{
char line[BUFSIZ];
for (;;) {
if (getinput(">", "partition", NULL, line) == -1)
return;
if (runcmd(line, lp, fd) == -1)
return;
}
}