Home | History | Annotate | Line # | Download | only in gen
      1 /*	$NetBSD: disklabel.c,v 1.37 2012/06/25 22:32:43 abs Exp $	*/
      2 
      3 /*
      4  * Copyright (c) 1983, 1987, 1993
      5  *	The Regents of the University of California.  All rights reserved.
      6  *
      7  * Redistribution and use in source and binary forms, with or without
      8  * modification, are permitted provided that the following conditions
      9  * are met:
     10  * 1. Redistributions of source code must retain the above copyright
     11  *    notice, this list of conditions and the following disclaimer.
     12  * 2. Redistributions in binary form must reproduce the above copyright
     13  *    notice, this list of conditions and the following disclaimer in the
     14  *    documentation and/or other materials provided with the distribution.
     15  * 3. Neither the name of the University nor the names of its contributors
     16  *    may be used to endorse or promote products derived from this software
     17  *    without specific prior written permission.
     18  *
     19  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
     20  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     21  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     22  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
     23  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     24  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
     25  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     26  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     27  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     28  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     29  * SUCH DAMAGE.
     30  */
     31 
     32 #if HAVE_NBTOOL_CONFIG_H
     33 #include "nbtool_config.h"
     34 #endif
     35 
     36 #include <sys/cdefs.h>
     37 #if defined(LIBC_SCCS) && !defined(lint)
     38 #if 0
     39 static char sccsid[] = "@(#)disklabel.c	8.2 (Berkeley) 5/3/95";
     40 #else
     41 __RCSID("$NetBSD: disklabel.c,v 1.37 2012/06/25 22:32:43 abs Exp $");
     42 #endif
     43 #endif /* LIBC_SCCS and not lint */
     44 
     45 #include "namespace.h"
     46 #include <sys/param.h>
     47 #define DKTYPENAMES
     48 #define FSTYPENAMES
     49 #include <ufs/ufs/dinode.h>
     50 #include <ufs/ffs/fs.h>
     51 
     52 #if HAVE_NBTOOL_CONFIG_H
     53 #include <nbinclude/sys/disklabel.h>
     54 #include <nbinclude/disktab.h>
     55 #else
     56 #include <sys/disklabel.h>
     57 #include <disktab.h>
     58 #endif /* HAVE_NBTOOL_CONFIG_H */
     59 
     60 #include <assert.h>
     61 #include <ctype.h>
     62 #include <errno.h>
     63 #include <fcntl.h>
     64 #include <stdio.h>
     65 #include <stdlib.h>
     66 #include <string.h>
     67 #include <unistd.h>
     68 
     69 #ifdef __weak_alias
     70 __weak_alias(getdiskbyname,_getdiskbyname)
     71 #endif
     72 
     73 #if 0
     74 static void	error(int);
     75 #endif
     76 static int	gettype(char *, const char *const *);
     77 
     78 static const char *db_array[2] = { _PATH_DISKTAB, 0 };
     79 
     80 int
     81 setdisktab(const char *name)
     82 {
     83 	if (!name || !*name)
     84 		return -1;
     85 
     86 	db_array[0] = name;
     87 	return 0;
     88 }
     89 
     90 
     91 struct disklabel *
     92 getdiskbyname(const char *name)
     93 {
     94 	static struct	disklabel disk;
     95 	struct	disklabel *dp = &disk;
     96 	struct partition *pp;
     97 	char	*buf;
     98 	char	*cp, *cq;	/* can't be */
     99 	char	p, max, psize[3], pbsize[3],
    100 		pfsize[3], poffset[3], ptype[3];
    101 	u_int32_t *dx;
    102 	long f;
    103 
    104 	_DIAGASSERT(name != NULL);
    105 
    106 	if (cgetent(&buf, db_array, name) < 0)
    107 		return NULL;
    108 
    109 	memset(&disk, 0, sizeof(disk));
    110 	/*
    111 	 * typename
    112 	 */
    113 	cq = dp->d_typename;
    114 	cp = buf;
    115 	while (cq < dp->d_typename + sizeof(dp->d_typename) - 1 &&
    116 	    (*cq = *cp) && *cq != '|' && *cq != ':')
    117 		cq++, cp++;
    118 	*cq = '\0';
    119 	/*
    120 	 * boot name (optional)  xxboot, bootxx
    121 	 */
    122 	cgetstr(buf, "b0", &dp->d_boot0);
    123 	cgetstr(buf, "b1", &dp->d_boot1);
    124 
    125 	if (cgetstr(buf, "ty", &cq) >= 0) {
    126 		if (strcmp(cq, "removable") == 0)
    127 			dp->d_flags |= D_REMOVABLE;
    128 		else if (strcmp(cq, "simulated") == 0)
    129 			dp->d_flags |= D_RAMDISK;
    130 		free(cq);
    131 	}
    132 	if (cgetcap(buf, "sf", ':') != NULL)
    133 		dp->d_flags |= D_BADSECT;
    134 
    135 #define getnumdflt(field, dname, dflt) \
    136     (field) = ((cgetnum(buf, dname, &f) == -1) ? (dflt) : (u_int32_t) f)
    137 #define	getnum(field, dname) \
    138 	if (cgetnum(buf, dname, &f) != -1) field = (u_int32_t)f
    139 
    140 	getnumdflt(dp->d_secsize, "se", DEV_BSIZE);
    141 	getnum(dp->d_ntracks, "nt");
    142 	getnum(dp->d_nsectors, "ns");
    143 	getnum(dp->d_ncylinders, "nc");
    144 
    145 	if (cgetstr(buf, "dt", &cq) >= 0) {
    146 		dp->d_type = gettype(cq, dktypenames);
    147 		free(cq);
    148 	} else
    149 		getnumdflt(dp->d_type, "dt", 0);
    150 	getnumdflt(dp->d_secpercyl, "sc", dp->d_nsectors * dp->d_ntracks);
    151 	getnumdflt(dp->d_secperunit, "su", dp->d_secpercyl * dp->d_ncylinders);
    152 	getnumdflt(dp->d_rpm, "rm", 3600);
    153 	getnumdflt(dp->d_interleave, "il", 1);
    154 	getnumdflt(dp->d_trackskew, "sk", 0);
    155 	getnumdflt(dp->d_cylskew, "cs", 0);
    156 	getnumdflt(dp->d_headswitch, "hs", 0);
    157 	getnumdflt(dp->d_trkseek, "ts", 0);
    158 	getnumdflt(dp->d_bbsize, "bs", BBSIZE);
    159 	getnumdflt(dp->d_sbsize, "sb", SBLOCKSIZE);
    160 	strcpy(psize, "px");	/* XXX: strcpy is safe */
    161 	strcpy(pbsize, "bx");	/* XXX: strcpy is safe */
    162 	strcpy(pfsize, "fx");	/* XXX: strcpy is safe */
    163 	strcpy(poffset, "ox");	/* XXX: strcpy is safe */
    164 	strcpy(ptype, "tx");	/* XXX: strcpy is safe */
    165 	max = 'a' - 1;
    166 	pp = &dp->d_partitions[0];
    167 	for (p = 'a'; p < 'a' + MAXPARTITIONS; p++, pp++) {
    168 		long ff;
    169 
    170 		psize[1] = pbsize[1] = pfsize[1] = poffset[1] = ptype[1] = p;
    171 		if (cgetnum(buf, psize, &ff) == -1)
    172 			pp->p_size = 0;
    173 		else {
    174 			pp->p_size = (u_int32_t)ff;
    175 			getnum(pp->p_offset, poffset);
    176 			getnumdflt(pp->p_fsize, pfsize, 0);
    177 			if (pp->p_fsize) {
    178 				long bsize;
    179 
    180 				if (cgetnum(buf, pbsize, &bsize) == -1)
    181 					pp->p_frag = 8;
    182 				else
    183 					pp->p_frag =
    184 					    (u_int8_t)(bsize / pp->p_fsize);
    185 			}
    186 			getnumdflt(pp->p_fstype, ptype, 0);
    187 			if (pp->p_fstype == 0)
    188 				if (cgetstr(buf, ptype, &cq) >= 0) {
    189 					pp->p_fstype = gettype(cq, fstypenames);
    190 					free(cq);
    191 				}
    192 			max = p;
    193 		}
    194 	}
    195 	dp->d_npartitions = max + 1 - 'a';
    196 	strcpy(psize, "dx");	/* XXX: strcpy is safe */
    197 	dx = dp->d_drivedata;
    198 	for (p = '0'; p < '0' + NDDATA; p++, dx++) {
    199 		psize[1] = p;
    200 		getnumdflt(*dx, psize, 0);
    201 	}
    202 	dp->d_magic = DISKMAGIC;
    203 	dp->d_magic2 = DISKMAGIC;
    204 	free(buf);
    205 	return (dp);
    206 }
    207 
    208 static int
    209 gettype(char *t, const char *const *names)
    210 {
    211 	const char *const *nm;
    212 
    213 	_DIAGASSERT(t != NULL);
    214 	_DIAGASSERT(names != NULL);
    215 
    216 	for (nm = names; *nm; nm++)
    217 		if (strcasecmp(t, *nm) == 0)
    218 			return (int)(nm - names);
    219 	if (isdigit((unsigned char) *t))
    220 		return (atoi(t));
    221 	return (0);
    222 }
    223 
    224 #if 0
    225 static void
    226 error(err)
    227 	int err;
    228 {
    229 	char *p;
    230 
    231 	(void)write(STDERR_FILENO, "disktab: ", 9);
    232 	(void)write(STDERR_FILENO, _PATH_DISKTAB, sizeof(_PATH_DISKTAB) - 1);
    233 	(void)write(STDERR_FILENO, ": ", 2);
    234 	p = strerror(err);
    235 	(void)write(STDERR_FILENO, p, strlen(p));
    236 	(void)write(STDERR_FILENO, "\n", 1);
    237 }
    238 #endif
    239