summaryrefslogtreecommitdiff
path: root/sbin/disklabel/interact.c
diff options
context:
space:
mode:
authorchristos <christos@NetBSD.org>1997-03-08 23:46:08 +0000
committerchristos <christos@NetBSD.org>1997-03-08 23:46:08 +0000
commit72fbacb380ea4e2dfba5a16e6bdba2004aa2af95 (patch)
treeecf83b015392c0e27cb0031506358b4a9ea7734f /sbin/disklabel/interact.c
parentd15724228ecb1298eb5d83c24be215ef943f596e (diff)
Add -i (interactive mode to create/edit labels) inspired by the SunOS/Solaris
format/partition command.
Diffstat (limited to 'sbin/disklabel/interact.c')
-rw-r--r--sbin/disklabel/interact.c370
1 files changed, 370 insertions, 0 deletions
diff --git a/sbin/disklabel/interact.c b/sbin/disklabel/interact.c
new file mode 100644
index 00000000000..e7cfdea17f1
--- /dev/null
+++ b/sbin/disklabel/interact.c
@@ -0,0 +1,370 @@
+/* $NetBSD: interact.c,v 1.1 1997/03/08 23:46:12 christos 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.
+ */
+
+#ifndef lint
+static char rcsid[] = "$NetBSD: interact.c,v 1.1 1997/03/08 23:46:12 christos Exp $";
+#endif /* lint */
+
+#include <stdio.h>
+#include <string.h>
+#include <stdlib.h>
+#include <sys/types.h>
+#include <sys/param.h>
+#define DKTYPENAMES
+#include <sys/disklabel.h>
+
+#include "extern.h"
+
+static void cmd_help __P((struct disklabel *, char *, int));
+static void cmd_print __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 void defnum __P((char *, struct disklabel *, int));
+static int getnum __P((char *, struct disklabel *));
+static void deffstypename __P((char *, int));
+static int getfstypename __P((const char *));
+
+static int rounding = 0; /* sector rounding */
+
+static struct cmds {
+ const char *name;
+ void (*func) __P((struct disklabel *, char *, int));
+ const char *help;
+} cmds[] = {
+ { "a", cmd_part, "define partition a" },
+ { "b", cmd_part, "define partition b" },
+ { "c", cmd_part, "define partition c" },
+ { "d", cmd_part, "define partition d" },
+ { "e", cmd_part, "define partition e" },
+ { "f", cmd_part, "define partition f" },
+ { "g", cmd_part, "define partition g" },
+ { "h", cmd_part, "define partition h" },
+ { "l", cmd_label, "write the current partition table" },
+ { "n", cmd_name, "name the label" },
+ { "p", cmd_print, "print current partition table" },
+ { "q", NULL, "quit" },
+ { "r", cmd_round, "rounding (c)ylinders (s)ectors" },
+ { "?", cmd_help, "print this menu" },
+ { 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);
+}
+
+
+static void
+cmd_print(lp, s, fd)
+ struct disklabel *lp;
+ char *s;
+ int fd;
+{
+ display(stdout, lp);
+}
+
+
+static void
+cmd_name(lp, s, fd)
+ struct disklabel *lp;
+ char *s;
+ int fd;
+{
+ char line[BUFSIZ];
+ int 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 (;;) {
+ deffstypename(def, p->p_fstype);
+ i = getinput(":", "Filesystem type", def, line);
+ if (i <= 0)
+ break;
+ if ((i = getfstypename(line)) == -1) {
+ printf("Invalid file system typename `%s'\n", line);
+ continue;
+ }
+ p->p_fstype = i;
+ break;
+ }
+ for (;;) {
+ defnum(def, lp, p->p_offset);
+ i = getinput(":", "Start offset", def, line);
+ if (i <= 0)
+ break;
+ if ((i = getnum(line, 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", def, line);
+ if (i <= 0)
+ break;
+ if ((i = getnum(line, lp)) == -1) {
+ printf("Bad size `%s'\n", line);
+ continue;
+ }
+ p->p_size = i;
+ break;
+ }
+}
+
+
+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')
+ return;
+
+ if (checklabel(lp) != 0) {
+ printf("Label not written\n");
+ return;
+ }
+
+ if ((i = writelabel(fd, bootarea, lp)) != 0) {
+ printf("Label not written %d\n", strerror(i));
+ 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;
+ }
+ printf("Unknown command %s\n", line);
+}
+
+
+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 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, lp)
+ char *buf;
+ struct disklabel *lp;
+{
+ char *ep;
+ double d = strtod(buf, &ep);
+ int rv;
+
+ 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':
+ 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;
+}
+
+
+static void
+deffstypename(buf, i)
+ char *buf;
+ int i;
+{
+ if (i < 0 || i >= DKMAXTYPES)
+ i = 0;
+ (void) strcpy(buf, fstypenames[i]);
+}
+
+
+static int
+getfstypename(buf)
+ const char *buf;
+{
+ int i;
+
+ for (i = 0; i < DKMAXTYPES; i++)
+ if (strcmp(buf, fstypenames[i]) == 0)
+ return i;
+ return -1;
+}
+
+
+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;
+ }
+}