initial import of NetBSD tree

1 files changed, 423 insertions, 0 deletions

diff --git a/sbin/edlabel/edlabel.c b/sbin/edlabel/edlabel.c
new file mode 100644
index 00000000000..4eb16a3c435
--- /dev/null
+++ b/sbin/edlabel/edlabel.c
@@ -0,0 +1,423 @@
+/*	$NetBSD: edlabel.c,v 1.1.1.1 1995/10/08 22:39:09 gwr Exp $	*/
+
+/*
+ * Copyright (c) 1995 Gordon W. Ross
+ * 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. The name of the author may not be used to endorse or promote products
+ *    derived from this software without specific prior written permission.
+ * 4. All advertising materials mentioning features or use of this software
+ *    must display the following acknowledgement:
+ *      This product includes software developed by Gordon W. Ross
+ *
+ * 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/types.h>
+#include <sys/ioccom.h>
+#include <sys/disklabel.h>
+
+#include <fcntl.h>
+#include <stdio.h>
+#include <ctype.h>
+
+/*
+ * This is a data-driven program
+ */
+struct field {
+	char *f_name;
+	int f_offset;
+	int f_type;	/* 1:char, 2:short, 4:int, >4:string */
+};
+
+/* Table describing fields in the head of a disklabel. */
+#define	dloff(f) (int)(&((struct disklabel *)0)->f)
+struct field label_head[] = {
+  { "        type_num", dloff(d_type), 2 },
+  { "        sub_type", dloff(d_subtype), 2 },
+  { "       type_name", dloff(d_typename), 16 },
+  { "       pack_name", dloff(d_packname),  16 },
+  { "    bytes/sector", dloff(d_secsize), 4 },
+  { "   sectors/track", dloff(d_nsectors), 4 },
+  { " tracks/cylinder", dloff(d_ntracks),  4 },
+  { "       cylinders", dloff(d_ncylinders), 4 },
+  { "sectors/cylinder", dloff(d_secpercyl), 4 },
+  /* Don't care about the others until later... */
+  0,
+};
+#undef dloff
+
+char tmpbuf[64];
+
+void
+edit_head_field(v, f, modify)
+	void *v;
+	struct field *f;
+	int modify;	/* also modify */
+{
+	u_int8_t  *cp;
+	u_int tmp;
+
+	cp = v;
+	cp += f->f_offset;
+
+	printf("%s: ", f->f_name);
+
+	/* Print current value... */
+	switch (f->f_type) {
+	case 1:
+		tmp = *cp;
+		printf("%d", tmp);
+		break;
+	case 2:
+		tmp = *((u_int16_t *)cp);
+		printf("%d", tmp);
+		break;
+	case 4:
+		tmp = *((u_int32_t *)cp);
+		printf("%d", tmp);
+		break;
+
+	default:
+		/* must be a string. */
+		strncpy(tmpbuf, (char*)cp, sizeof(tmpbuf));
+		printf("%s", tmpbuf);
+		break;
+	}
+
+	if (modify == 0) {
+		printf("\n");
+		return;
+	}
+	printf(" ? ");
+	fflush(stdout);
+
+	tmpbuf[0] = '\0';
+	if (fgets(tmpbuf, sizeof(tmpbuf), stdin) == NULL)
+		return;
+	if ((tmpbuf[0] == '\0') || (tmpbuf[0] == '\n')) {
+		/* no new value supplied. */
+		return;
+	}
+
+	/* store new value */
+	if (f->f_type <= 4)
+		if (sscanf(tmpbuf, "%d", &tmp) != 1)
+			return;
+
+	switch (f->f_type) {
+	case 1:
+		*cp = tmp;
+		break;
+	case 2:
+		*((u_int16_t *)cp) = tmp;
+		break;
+	case 4:
+		*((u_int32_t *)cp) = tmp;
+		break;
+	default:
+		/* Get rid of the trailing newline. */
+		tmp = strlen(tmpbuf);
+		if (tmp < 1)
+			break;
+		if (tmpbuf[tmp-1] == '\n')
+			tmpbuf[tmp-1] = '\0';
+		strncpy((char*)cp, tmpbuf, f->f_type);
+		break;
+	}
+}
+
+void
+edit_head_all(d, modify)
+	struct disklabel *d;
+	int modify;
+{
+	struct field *f;
+
+	/* Edit head stuff. */
+	for (f = label_head; f->f_name; f++)
+		edit_head_field(d, f, modify);
+}
+
+
+void
+edit_partition(d, idx, modify)
+	struct disklabel *d;
+	int idx;
+{
+	struct partition *p;
+	int start_cyl, cyls, trks, sects, tmp;
+	char letter;
+
+	if ((idx < 0) || (idx >= MAXPARTITIONS)) {
+		printf("bad partition index\n");
+		return;
+	}
+	
+	p = &d->d_partitions[idx];
+	letter = 'a' + idx;
+
+	/* Print current value... */
+	start_cyl = p->p_offset / d->d_secpercyl;
+	printf("partition %c start cyl= %6d", letter, start_cyl);
+	if (p->p_offset % d->d_secpercyl) {
+		printf(" (bad start!)");
+	}
+	printf(",  nblks= %9d ", p->p_size);
+	tmp = p->p_size;
+	sects = tmp % d->d_nsectors;
+	tmp /= d->d_nsectors;
+	trks = tmp % d->d_ntracks;
+	tmp /= d->d_ntracks;
+	cyls = tmp;
+	printf("(%d/%d/%d)\n", cyls, trks, sects);
+
+	if (modify == 0)
+		return;
+
+	/* Start cylinder: */
+	printf("partition %c start cyl: ", letter);
+	fflush(stdout);
+	if (fgets(tmpbuf, sizeof(tmpbuf), stdin) == NULL)
+		return;
+	if (sscanf(tmpbuf, "%d", &tmp) == 1)
+		p->p_offset = tmp * d->d_secpercyl;
+
+	/* number of blocks, or cyls/trks/sects */
+	printf("length as <nblks> or <cyls/trks/sects> : ");
+	fflush(stdout);
+	if (fgets(tmpbuf, sizeof(tmpbuf), stdin) == NULL)
+		return;
+	tmp = sscanf(tmpbuf, "%d/%d/%d", &cyls, &trks, &sects);
+	if (tmp == 1)
+		p->p_size = cyls;	/* really nblks! */
+	if (tmp == 3) {
+		tmp = cyls;
+		tmp *= d->d_ntracks;
+		tmp += trks;
+		tmp *= d->d_nsectors;
+		tmp += sects;
+		p->p_size = tmp;
+	}
+}
+
+/*****************************************************************/
+
+void
+check_divisors(d)
+	struct disklabel *d;
+{
+	if (d->d_nsectors == 0) {
+		d->d_nsectors = 1;
+		printf("bad sect/trk, set to 1\n");
+	}
+	if (d->d_ntracks == 0) {
+		d->d_ntracks = 1;
+		printf("bad trks/cyl, set to 1\n");
+	}
+	if (d->d_ncylinders == 0) {
+		d->d_ncylinders = 1;
+		printf("bad cyls, set to 1\n");
+	}
+	if (d->d_secpercyl == 0) {
+		d->d_secpercyl = (d->d_nsectors * d->d_ntracks);
+		printf("bad sect/cyl, set to %d\n", d->d_secpercyl);
+	}
+
+}
+
+u_short
+dkcksum(d)
+	struct disklabel *d;
+{
+	u_short *start, *end;
+	u_short sum = 0;
+
+	start = (u_short *)d;
+	end = (u_short *)&d->d_partitions[d->d_npartitions];
+	while (start < end)
+		sum ^= *start++;
+	return (sum);
+}
+
+void
+label_write(d, dn)
+	struct disklabel *d;
+	char *dn;
+{
+	int fd;
+
+	d->d_magic = DISKMAGIC;
+	d->d_magic2 = DISKMAGIC;
+	d->d_checksum = 0;
+	d->d_checksum = dkcksum(d);
+
+	fd = open(dn, O_RDWR, 0);
+	if (fd < 0) {
+		perror(dn);
+		return;
+	}
+	if (ioctl(fd, DIOCWDINFO, d) < 0) {
+		perror("ioctl DIOCWDINFO");
+	}
+	close(fd);
+}
+
+void
+label_read(dl, dn)
+	struct disklabel *dl;
+	char *dn;
+{
+	int fd;
+
+	fd = open(dn, O_RDONLY, 0);
+	if (fd < 0) {
+		perror(dn);
+		exit(1);
+	}
+	if (ioctl(fd, DIOCGDINFO, dl) < 0) {
+		perror("ioctl DIOCGDINFO");
+		exit(1);
+	}
+
+	/* Make sure divisors are non-zero. */
+	check_divisors(dl);
+
+	close(fd);
+}
+
+/*****************************************************************/
+
+void
+label_print(dl, dn)
+	struct disklabel *dl;
+	char *dn;
+{
+	struct field *f;
+	int i;
+
+	/* Print out head stuff. */
+	edit_head_all(dl, 0);
+
+	for (i = 0; i < 8; i++)
+		edit_partition(dl, i, 0);
+}
+
+char modify_cmds[] = "modify subcommands:\n\
+ @   : modify disk parameters\n\
+ a-h : modify partition\n\
+ q   : quit this subcommand\n";
+
+void
+label_modify(dl, dn)
+	struct disklabel *dl;
+	char *dn;
+{
+	int c;
+
+	printf(modify_cmds);
+	for (;;) {
+		printf("edlabel/modify> ");
+		fflush(stdout);
+		if (fgets(tmpbuf, sizeof(tmpbuf), stdin) == NULL)
+			break;
+		c = tmpbuf[0];
+		if ((c == '\0') || (c == '\n'))
+			continue;	/* blank line */
+		if (c == 'q')
+			break;
+		if (c == '@') {
+			edit_head_all(dl, 1);
+			check_divisors(dl);
+			continue;
+		}
+		if ((c < 'a') || (c > 'q')) {
+			printf("bad input.  ");
+			printf(modify_cmds);
+			continue;
+		}
+		edit_partition(dl, c - 'a', 1);
+	}
+}
+
+void
+label_quit()
+{
+	exit(0);
+}
+
+struct cmd {
+	void (*cmd_func)();
+	char *cmd_name;
+	char *cmd_descr;
+} cmds[] = {
+	{ label_print,  "print",  "display the current disk label" },
+	{ label_modify, "modify", "prompt for changes to the label" },
+	{ label_write,  "write",  "write the new label to disk" },
+	{ label_quit,   "quit",   "terminate program" },
+	0,
+};
+
+void
+menu()
+{
+	struct cmd *cmd;
+
+	printf("edlabel menu:\n");
+	for (cmd = cmds; cmd->cmd_func; cmd++)
+		printf("%s\t- %s\n", cmd->cmd_name, cmd->cmd_descr);
+}
+
+main(argc, argv)
+	int argc;
+	char **argv;
+{
+	struct disklabel dl;
+	struct cmd *cmd;
+	char *devname;
+
+	if (argc != 2) {
+		fprintf(stderr, "usage: edlabel RAWDISK\n");
+		exit(1);
+	}
+	devname = argv[1];
+
+	label_read(&dl, devname);
+
+	menu();
+
+	for (;;) {
+		printf("edlabel> ");
+		fflush(stdout);
+		if (fgets(tmpbuf, sizeof(tmpbuf), stdin) == NULL)
+			break;
+		for (cmd = cmds; cmd->cmd_func; cmd++)
+			if (cmd->cmd_name[0] == tmpbuf[0])
+				goto found;
+		printf("Invalid command.  ");
+		menu();
+		continue;
+
+	found:
+		cmd->cmd_func(&dl, devname);
+	}
+	exit(0);
+}
+