Home | History | Annotate | Line # | Download | only in svhlabel
svhlabel.c revision 1.4.22.1
      1  1.4.22.1     jym /*	$NetBSD: svhlabel.c,v 1.4.22.1 2009/05/13 19:19:06 jym Exp $	*/
      2       1.1  rumble 
      3       1.1  rumble /*
      4       1.1  rumble  * Copyright (C) 2007 Stephen M. Rumble.
      5       1.1  rumble  * Copyright (C) 1998 Wolfgang Solfrank.
      6       1.1  rumble  * Copyright (C) 1998 TooLs GmbH.
      7       1.1  rumble  * All rights reserved.
      8       1.1  rumble  *
      9       1.1  rumble  * Redistribution and use in source and binary forms, with or without
     10       1.1  rumble  * modification, are permitted provided that the following conditions
     11       1.1  rumble  * are met:
     12       1.1  rumble  * 1. Redistributions of source code must retain the above copyright
     13       1.1  rumble  *    notice, this list of conditions and the following disclaimer.
     14       1.1  rumble  * 2. Redistributions in binary form must reproduce the above copyright
     15       1.1  rumble  *    notice, this list of conditions and the following disclaimer in the
     16       1.1  rumble  *    documentation and/or other materials provided with the distribution.
     17       1.1  rumble  * 3. All advertising materials mentioning features or use of this software
     18       1.1  rumble  *    must display the following acknowledgement:
     19       1.1  rumble  *	This product includes software developed by TooLs GmbH.
     20       1.1  rumble  * 4. The name of TooLs GmbH may not be used to endorse or promote products
     21       1.1  rumble  *    derived from this software without specific prior written permission.
     22       1.1  rumble  *
     23       1.1  rumble  * THIS SOFTWARE IS PROVIDED BY TOOLS GMBH ``AS IS'' AND ANY EXPRESS OR
     24       1.1  rumble  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
     25       1.1  rumble  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
     26       1.1  rumble  * IN NO EVENT SHALL TOOLS GMBH BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
     27       1.1  rumble  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
     28       1.1  rumble  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
     29       1.1  rumble  * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
     30       1.1  rumble  * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
     31       1.1  rumble  * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
     32       1.1  rumble  * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
     33       1.1  rumble  */
     34       1.1  rumble 
     35       1.1  rumble #include <sys/cdefs.h>
     36       1.1  rumble #ifndef lint
     37  1.4.22.1     jym __RCSID("$NetBSD: svhlabel.c,v 1.4.22.1 2009/05/13 19:19:06 jym Exp $");
     38       1.1  rumble #endif /* not lint */
     39       1.1  rumble 
     40       1.1  rumble #include <stdio.h>
     41       1.1  rumble #include <err.h>
     42       1.1  rumble #include <errno.h>
     43       1.1  rumble #include <fcntl.h>
     44       1.1  rumble #include <limits.h>
     45       1.1  rumble #include <stdlib.h>
     46       1.1  rumble #include <string.h>
     47       1.1  rumble #include <unistd.h>
     48       1.1  rumble #include <util.h>
     49       1.1  rumble 
     50       1.1  rumble #include <sys/param.h>
     51       1.1  rumble #define FSTYPENAMES
     52       1.1  rumble #include <sys/disklabel.h>
     53       1.1  rumble #include <sys/bootblock.h>
     54       1.1  rumble #include <sys/ioctl.h>
     55       1.1  rumble 
     56       1.4  rumble #include <fs/efs/efs.h>
     57       1.3  rumble #include <fs/efs/efs_sb.h>
     58       1.3  rumble 
     59       1.1  rumble #include "dkcksum.h"
     60       1.1  rumble #include "extern.h"
     61       1.1  rumble 
     62       1.1  rumble int		main(int, char **);
     63       1.1  rumble static void	usage(void);
     64       1.1  rumble static void	getlabel(int);
     65       1.1  rumble static void	setlabel(int, int);
     66       1.1  rumble static int	getparts(int, int);
     67       1.1  rumble static int	is_efs(int, uint32_t);
     68       1.1  rumble static struct sgi_boot_block *convert_sgi_boot_block(unsigned char *);
     69       1.1  rumble 
     70       1.1  rumble struct disklabel label;
     71       1.1  rumble 
     72       1.1  rumble static void
     73       1.1  rumble getlabel(int sd)
     74       1.1  rumble {
     75       1.1  rumble 
     76       1.1  rumble 	if (ioctl(sd, DIOCGDINFO, &label) < 0) {
     77       1.1  rumble 		perror("get label");
     78       1.1  rumble 		exit(1);
     79       1.1  rumble 	}
     80       1.1  rumble 	/*
     81       1.1  rumble 	 * Some ports seem to not set the number of partitions
     82       1.1  rumble 	 * correctly, albeit they seem to set the raw partition ok!
     83       1.1  rumble 	 */
     84       1.1  rumble 	if (label.d_npartitions <= getrawpartition())
     85       1.1  rumble 		label.d_npartitions = getrawpartition() + 1;
     86       1.1  rumble }
     87       1.1  rumble 
     88       1.1  rumble static void
     89       1.1  rumble setlabel(int sd, int doraw)
     90       1.1  rumble {
     91       1.1  rumble 	int one = 1;
     92       1.1  rumble 
     93       1.1  rumble 	label.d_checksum = 0;
     94       1.1  rumble 	label.d_checksum = dkcksum(&label);
     95       1.1  rumble 	if (ioctl(sd, doraw ? DIOCWDINFO : DIOCSDINFO, &label) < 0) {
     96       1.1  rumble 		perror("set label");
     97       1.1  rumble 		exit(1);
     98       1.1  rumble 	}
     99       1.1  rumble 	if (!doraw)
    100       1.1  rumble 		/* If we haven't written to the disk, don't discard on close */
    101       1.1  rumble 		ioctl(sd, DIOCKLABEL, &one);
    102       1.1  rumble 
    103       1.1  rumble }
    104       1.1  rumble 
    105       1.1  rumble static int
    106       1.1  rumble getparts(int sd, int verbose)
    107       1.1  rumble {
    108       1.1  rumble 	unsigned char		buf[DEV_BSIZE];
    109       1.1  rumble 	struct sgi_boot_block  *vh;
    110       1.1  rumble 	struct partition	npe;
    111       1.1  rumble 	int			i, j, changed;
    112       1.1  rumble 
    113       1.1  rumble 	changed = 0;
    114       1.1  rumble 
    115       1.1  rumble 	if (lseek(sd, 0, SEEK_SET) == -1) {
    116       1.1  rumble 		perror("seek vh");
    117       1.1  rumble 		exit(1);
    118       1.1  rumble 	}
    119       1.1  rumble 	if ((i = read(sd, buf, sizeof(buf))) != DEV_BSIZE) {
    120       1.1  rumble 		perror("read vh");
    121       1.1  rumble 		exit(1);
    122       1.1  rumble 	}
    123       1.1  rumble 	vh = convert_sgi_boot_block(buf);
    124       1.1  rumble 
    125       1.1  rumble 	if (vh->magic != SGI_BOOT_BLOCK_MAGIC)
    126       1.1  rumble 		return (changed);
    127       1.1  rumble 
    128       1.1  rumble 	if (label.d_secsize != SGI_BOOT_BLOCK_BLOCKSIZE)
    129       1.1  rumble 		changed++;
    130       1.1  rumble 	label.d_secsize = SGI_BOOT_BLOCK_BLOCKSIZE;
    131       1.1  rumble 
    132       1.1  rumble 	for (i = j = 0; i < SGI_BOOT_BLOCK_MAXPARTITIONS; i++) {
    133       1.1  rumble 		if (vh->partitions[i].blocks == 0)
    134       1.1  rumble 			continue;
    135       1.1  rumble 
    136       1.1  rumble 		if (j == MAXPARTITIONS)
    137       1.1  rumble 			break;
    138       1.1  rumble 
    139       1.1  rumble 		switch (vh->partitions[i].type) {
    140       1.1  rumble 		case SGI_PTYPE_EFS:
    141       1.1  rumble 		/*
    142       1.1  rumble 		 * For some reason, my IRIX CDs list EFS partitions as SYSV!?
    143       1.1  rumble 		 */
    144       1.1  rumble 		case SGI_PTYPE_SYSV:
    145       1.1  rumble 			if (is_efs(sd, vh->partitions[i].first)) {
    146       1.1  rumble 				npe.p_fstype = FS_EFS;
    147       1.1  rumble 				npe.p_size = vh->partitions[i].blocks;
    148       1.1  rumble 				npe.p_offset = vh->partitions[i].first;
    149       1.1  rumble 				npe.p_fsize = 0;
    150       1.1  rumble 				npe.p_frag = 0;
    151       1.1  rumble 				npe.p_cpg = 0;
    152       1.1  rumble 			}
    153       1.1  rumble 			break;
    154       1.1  rumble 
    155       1.1  rumble 		case SGI_PTYPE_VOLUME:
    156  1.4.22.1     jym 			if (label.d_secperunit != (uint32_t)vh->partitions[i].blocks)
    157       1.1  rumble 				changed++;
    158       1.1  rumble 			label.d_secperunit = vh->partitions[i].blocks;
    159       1.1  rumble 			continue;
    160       1.1  rumble 
    161       1.1  rumble 		default:
    162       1.1  rumble 			continue;
    163       1.1  rumble 		}
    164       1.1  rumble 
    165       1.1  rumble 		if (j >= label.d_npartitions)
    166       1.1  rumble 			break;
    167       1.1  rumble 
    168       1.1  rumble 		if (j == getrawpartition()) {
    169       1.1  rumble 			if (++j >= label.d_npartitions)
    170       1.1  rumble 				break;
    171       1.1  rumble 		}
    172       1.1  rumble 
    173       1.1  rumble 		if (memcmp(&label.d_partitions[j], &npe, sizeof(npe)) != 0) {
    174       1.1  rumble 			label.d_partitions[j] = npe;
    175       1.1  rumble 			changed++;
    176       1.1  rumble 		}
    177       1.1  rumble 
    178       1.1  rumble 		j++;
    179       1.1  rumble 	}
    180       1.1  rumble 
    181       1.1  rumble 	/* XXX - fudge */
    182       1.1  rumble 	if (label.d_nsectors != 1 || label.d_ntracks != 1 ||
    183       1.1  rumble 	    label.d_secpercyl != 1 || label.d_ncylinders != label.d_secperunit)
    184       1.1  rumble 		changed++;
    185       1.1  rumble 	label.d_nsectors = 1;
    186       1.1  rumble 	label.d_ntracks = 1;
    187       1.1  rumble 	label.d_secpercyl = 1;
    188       1.1  rumble 	label.d_ncylinders = label.d_secperunit;
    189       1.1  rumble 
    190       1.1  rumble 	i = getrawpartition();
    191       1.1  rumble 	if (label.d_partitions[i].p_fstype != FS_UNUSED ||
    192       1.1  rumble 	    label.d_partitions[i].p_offset != 0 ||
    193       1.1  rumble 	    label.d_partitions[i].p_size != label.d_secperunit) {
    194       1.1  rumble 		label.d_partitions[i].p_fstype = FS_UNUSED;
    195       1.1  rumble 		label.d_partitions[i].p_offset = 0;
    196       1.1  rumble 		label.d_partitions[i].p_size = label.d_secperunit;
    197       1.1  rumble 		changed++;
    198       1.1  rumble 	}
    199       1.1  rumble 
    200       1.1  rumble 	return (changed);
    201       1.1  rumble }
    202       1.1  rumble 
    203       1.1  rumble static int
    204       1.1  rumble is_efs(int sd, uint32_t blkoff)
    205       1.1  rumble {
    206       1.4  rumble 	struct efs_sb sb;
    207       1.1  rumble 	off_t oldoff;
    208       1.1  rumble 
    209       1.1  rumble 	if ((oldoff = lseek(sd, 0, SEEK_CUR)) == -1) {
    210       1.1  rumble 		perror("is_efs lseek 0");
    211       1.1  rumble 		exit(1);
    212       1.1  rumble 	}
    213       1.1  rumble 
    214       1.4  rumble 	blkoff *= SGI_BOOT_BLOCK_BLOCKSIZE;
    215       1.4  rumble 	if (lseek(sd, blkoff + (EFS_BB_SB * EFS_BB_SIZE), SEEK_SET) == -1) {
    216       1.1  rumble 		perror("is_efs lseek 1");
    217       1.1  rumble 		exit(1);
    218       1.1  rumble 	}
    219       1.1  rumble 
    220       1.4  rumble 	if (read(sd, &sb, sizeof(sb)) != sizeof(sb)) {
    221       1.1  rumble 		perror("is_efs read");
    222       1.1  rumble 		exit(1);
    223       1.1  rumble 	}
    224       1.1  rumble 
    225       1.1  rumble 	if (lseek(sd, oldoff, SEEK_SET) == -1) {
    226       1.1  rumble 		perror("is_efs lseek 2");
    227       1.1  rumble 		exit(1);
    228       1.1  rumble 	}
    229       1.1  rumble 
    230       1.4  rumble 	BE32TOH(sb.sb_magic);
    231       1.1  rumble 
    232       1.4  rumble 	return (sb.sb_magic == EFS_SB_MAGIC || sb.sb_magic == EFS_SB_NEWMAGIC);
    233       1.1  rumble }
    234       1.1  rumble 
    235       1.1  rumble static struct sgi_boot_block *
    236       1.1  rumble convert_sgi_boot_block(unsigned char *buf)
    237       1.1  rumble {
    238       1.1  rumble 	struct sgi_boot_block *vh;
    239       1.1  rumble 	int i;
    240       1.1  rumble 
    241       1.1  rumble 	vh = (struct sgi_boot_block *)buf;
    242       1.1  rumble 
    243       1.1  rumble 	BE32TOH(vh->magic);
    244       1.1  rumble 	BE16TOH(vh->root);
    245       1.1  rumble 	BE16TOH(vh->swap);
    246       1.1  rumble 
    247       1.1  rumble 	BE16TOH(vh->dp.dp_cyls);
    248       1.1  rumble 	BE16TOH(vh->dp.dp_shd0);
    249       1.1  rumble 	BE16TOH(vh->dp.dp_trks0);
    250       1.1  rumble 	BE16TOH(vh->dp.dp_secs);
    251       1.1  rumble 	BE16TOH(vh->dp.dp_secbytes);
    252       1.1  rumble 	BE16TOH(vh->dp.dp_interleave);
    253       1.1  rumble 	BE32TOH(vh->dp.dp_flags);
    254       1.1  rumble 	BE32TOH(vh->dp.dp_datarate);
    255       1.1  rumble 	BE32TOH(vh->dp.dp_nretries);
    256       1.1  rumble 	BE32TOH(vh->dp.dp_mspw);
    257       1.1  rumble 	BE16TOH(vh->dp.dp_xgap1);
    258       1.1  rumble 	BE16TOH(vh->dp.dp_xsync);
    259       1.1  rumble 	BE16TOH(vh->dp.dp_xrdly);
    260       1.1  rumble 	BE16TOH(vh->dp.dp_xgap2);
    261       1.1  rumble 	BE16TOH(vh->dp.dp_xrgate);
    262       1.1  rumble 	BE16TOH(vh->dp.dp_xwcont);
    263       1.1  rumble 
    264       1.1  rumble 	for (i = 0; i < SGI_BOOT_BLOCK_MAXVOLDIRS; i++) {
    265       1.1  rumble 		BE32TOH(vh->voldir[i].block);
    266       1.1  rumble 		BE32TOH(vh->voldir[i].bytes);
    267       1.1  rumble 	}
    268       1.1  rumble 
    269       1.1  rumble 	for (i = 0; i < SGI_BOOT_BLOCK_MAXPARTITIONS; i++) {
    270       1.1  rumble 		BE32TOH(vh->partitions[i].blocks);
    271       1.1  rumble 		BE32TOH(vh->partitions[i].first);
    272       1.1  rumble 		BE32TOH(vh->partitions[i].type);
    273       1.1  rumble 	}
    274       1.1  rumble 
    275       1.1  rumble 	BE32TOH(vh->checksum);
    276       1.1  rumble 
    277       1.1  rumble 	return (vh);
    278       1.1  rumble }
    279       1.1  rumble 
    280       1.1  rumble static void
    281       1.1  rumble usage(void)
    282       1.1  rumble {
    283       1.1  rumble 	fprintf(stderr, "usage: %s [-fqrw] rawdisk\n",
    284       1.1  rumble 	    getprogname());
    285       1.1  rumble 	exit(1);
    286       1.1  rumble }
    287       1.1  rumble 
    288       1.1  rumble 
    289       1.1  rumble int
    290       1.1  rumble main(int argc, char **argv)
    291       1.1  rumble {
    292       1.1  rumble 	int	sd, ch, changed;
    293       1.1  rumble 	char	name[MAXPATHLEN];
    294       1.1  rumble 	int	force;			/* force label update */
    295       1.1  rumble 	int	raw;			/* update on-disk label as well */
    296       1.1  rumble 	int	verbose;		/* verbose output */
    297       1.1  rumble 	int	write_it;		/* update in-core label if changed */
    298       1.1  rumble 
    299       1.1  rumble 	force = 0;
    300       1.1  rumble 	raw = 0;
    301       1.1  rumble 	verbose = 1;
    302       1.1  rumble 	write_it = 0;
    303       1.1  rumble 	while ((ch = getopt(argc, argv, "fqrw")) != -1) {
    304       1.1  rumble 		switch (ch) {
    305       1.1  rumble 		case 'f':
    306       1.1  rumble 			force = 1;
    307       1.1  rumble 			break;
    308       1.1  rumble 		case 'q':
    309       1.1  rumble 			verbose = 0;
    310       1.1  rumble 			break;
    311       1.1  rumble 		case 'r':
    312       1.1  rumble 			raw = 1;
    313       1.1  rumble 			break;
    314       1.1  rumble 		case 'w':
    315       1.1  rumble 			write_it = 1;
    316       1.1  rumble 			break;
    317       1.1  rumble 		default:
    318       1.1  rumble 			usage();
    319       1.1  rumble 		}
    320       1.1  rumble 	}
    321       1.1  rumble 	argc -= optind;
    322       1.1  rumble 	argv += optind;
    323       1.1  rumble 	if (argc != 1)
    324       1.1  rumble 		usage();
    325       1.1  rumble 
    326       1.1  rumble 	if ((sd = opendisk(argv[0], write_it ? O_RDWR : O_RDONLY, name,
    327       1.1  rumble 	    (size_t)MAXPATHLEN, 1)) < 0) {
    328       1.1  rumble 		perror(argv[0]);
    329       1.1  rumble 		exit(1);
    330       1.1  rumble 	}
    331       1.1  rumble 	getlabel(sd);
    332       1.1  rumble 	changed = getparts(sd, verbose);
    333       1.1  rumble 
    334       1.1  rumble 	if (verbose) {
    335       1.1  rumble 		putchar('\n');
    336       1.1  rumble 		showpartitions(stdout, &label, 0);
    337       1.1  rumble 		putchar('\n');
    338       1.1  rumble 	}
    339       1.1  rumble 	if (write_it) {
    340       1.1  rumble 		if (! changed && ! force)
    341       1.1  rumble 			printf("No change; not updating disk label.\n");
    342       1.1  rumble 		else {
    343       1.1  rumble 			if (verbose)
    344       1.1  rumble 				printf("Updating in-core %sdisk label.\n",
    345       1.1  rumble 				    raw ? "and on-disk " : "");
    346       1.1  rumble raw = 0; /* XXX */
    347       1.1  rumble 			setlabel(sd, raw);
    348       1.1  rumble 		}
    349       1.1  rumble 	} else {
    350       1.1  rumble 		printf("Not updating disk label.\n");
    351       1.1  rumble 	}
    352       1.1  rumble 	close(sd);
    353       1.1  rumble 	return (0);
    354       1.1  rumble }
    355