xref: /netbsd-src/lib/libc/gen/disklabel.c (revision b1c86f5f087524e68db12794ee9c3e3da1ab17a0)
1 /*	$NetBSD: disklabel.c,v 1.34 2006/03/19 02:17:16 christos 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.34 2006/03/19 02:17:16 christos 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 __P((int));
75 #endif
76 static int	gettype __P((char *, const char *const *));
77 
78 static const char *db_array[2] = { _PATH_DISKTAB, 0 };
79 
80 int
81 setdisktab(name)
82 	const char *name;
83 {
84 	if (!name || !*name)
85 		return -1;
86 
87 	db_array[0] = name;
88 	return 0;
89 }
90 
91 
92 struct disklabel *
93 getdiskbyname(name)
94 	const char *name;
95 {
96 	static struct	disklabel disk;
97 	struct	disklabel *dp = &disk;
98 	struct partition *pp;
99 	char	*buf;
100 	char	*cp, *cq;	/* can't be */
101 	char	p, max, psize[3], pbsize[3],
102 		pfsize[3], poffset[3], ptype[3];
103 	u_int32_t *dx;
104 	long f;
105 
106 	_DIAGASSERT(name != NULL);
107 
108 	if (cgetent(&buf, db_array, name) < 0)
109 		return NULL;
110 
111 	memset(&disk, 0, sizeof(disk));
112 	/*
113 	 * typename
114 	 */
115 	cq = dp->d_typename;
116 	cp = buf;
117 	while (cq < dp->d_typename + sizeof(dp->d_typename) - 1 &&
118 	    (*cq = *cp) && *cq != '|' && *cq != ':')
119 		cq++, cp++;
120 	*cq = '\0';
121 	/*
122 	 * boot name (optional)  xxboot, bootxx
123 	 */
124 	cgetstr(buf, "b0", &dp->d_boot0);
125 	cgetstr(buf, "b1", &dp->d_boot1);
126 
127 	if (cgetstr(buf, "ty", &cq) >= 0) {
128 		if (strcmp(cq, "removable") == 0)
129 			dp->d_flags |= D_REMOVABLE;
130 		else if (strcmp(cq, "simulated") == 0)
131 			dp->d_flags |= D_RAMDISK;
132 		free(cq);
133 	}
134 	if (cgetcap(buf, "sf", ':') != NULL)
135 		dp->d_flags |= D_BADSECT;
136 
137 #define getnumdflt(field, dname, dflt) \
138     (field) = ((cgetnum(buf, dname, &f) == -1) ? (dflt) : (u_int32_t) f)
139 #define	getnum(field, dname) \
140 	if (cgetnum(buf, dname, &f) != -1) field = (u_int32_t)f
141 
142 	getnumdflt(dp->d_secsize, "se", DEV_BSIZE);
143 	getnum(dp->d_ntracks, "nt");
144 	getnum(dp->d_nsectors, "ns");
145 	getnum(dp->d_ncylinders, "nc");
146 
147 	if (cgetstr(buf, "dt", &cq) >= 0) {
148 		dp->d_type = gettype(cq, dktypenames);
149 		free(cq);
150 	} else
151 		getnumdflt(dp->d_type, "dt", 0);
152 	getnumdflt(dp->d_secpercyl, "sc", dp->d_nsectors * dp->d_ntracks);
153 	getnumdflt(dp->d_secperunit, "su", dp->d_secpercyl * dp->d_ncylinders);
154 	getnumdflt(dp->d_rpm, "rm", 3600);
155 	getnumdflt(dp->d_interleave, "il", 1);
156 	getnumdflt(dp->d_trackskew, "sk", 0);
157 	getnumdflt(dp->d_cylskew, "cs", 0);
158 	getnumdflt(dp->d_headswitch, "hs", 0);
159 	getnumdflt(dp->d_trkseek, "ts", 0);
160 	getnumdflt(dp->d_bbsize, "bs", BBSIZE);
161 	getnumdflt(dp->d_sbsize, "sb", SBLOCKSIZE);
162 	strcpy(psize, "px");	/* XXX: strcpy is safe */
163 	strcpy(pbsize, "bx");	/* XXX: strcpy is safe */
164 	strcpy(pfsize, "fx");	/* XXX: strcpy is safe */
165 	strcpy(poffset, "ox");	/* XXX: strcpy is safe */
166 	strcpy(ptype, "tx");	/* XXX: strcpy is safe */
167 	max = 'a' - 1;
168 	pp = &dp->d_partitions[0];
169 	for (p = 'a'; p < 'a' + MAXPARTITIONS; p++, pp++) {
170 		long ff;
171 
172 		psize[1] = pbsize[1] = pfsize[1] = poffset[1] = ptype[1] = p;
173 		if (cgetnum(buf, psize, &ff) == -1)
174 			pp->p_size = 0;
175 		else {
176 			pp->p_size = (u_int32_t)ff;
177 			getnum(pp->p_offset, poffset);
178 			getnumdflt(pp->p_fsize, pfsize, 0);
179 			if (pp->p_fsize) {
180 				long bsize;
181 
182 				if (cgetnum(buf, pbsize, &bsize) == -1)
183 					pp->p_frag = 8;
184 				else
185 					pp->p_frag =
186 					    (u_int8_t)(bsize / pp->p_fsize);
187 			}
188 			getnumdflt(pp->p_fstype, ptype, 0);
189 			if (pp->p_fstype == 0)
190 				if (cgetstr(buf, ptype, &cq) >= 0) {
191 					pp->p_fstype = gettype(cq, fstypenames);
192 					free(cq);
193 				}
194 			max = p;
195 		}
196 	}
197 	dp->d_npartitions = max + 1 - 'a';
198 	strcpy(psize, "dx");	/* XXX: strcpy is safe */
199 	dx = dp->d_drivedata;
200 	for (p = '0'; p < '0' + NDDATA; p++, dx++) {
201 		psize[1] = p;
202 		getnumdflt(*dx, psize, 0);
203 	}
204 	dp->d_magic = DISKMAGIC;
205 	dp->d_magic2 = DISKMAGIC;
206 	free(buf);
207 	return (dp);
208 }
209 
210 static int
211 gettype(t, names)
212 	char *t;
213 	const char *const *names;
214 {
215 	const char *const *nm;
216 
217 	_DIAGASSERT(t != NULL);
218 	_DIAGASSERT(names != NULL);
219 
220 	for (nm = names; *nm; nm++)
221 		if (strcasecmp(t, *nm) == 0)
222 			return (nm - names);
223 	if (isdigit((unsigned char) *t))
224 		return (atoi(t));
225 	return (0);
226 }
227 
228 #if 0
229 static void
230 error(err)
231 	int err;
232 {
233 	char *p;
234 
235 	(void)write(STDERR_FILENO, "disktab: ", 9);
236 	(void)write(STDERR_FILENO, _PATH_DISKTAB, sizeof(_PATH_DISKTAB) - 1);
237 	(void)write(STDERR_FILENO, ": ", 2);
238 	p = strerror(err);
239 	(void)write(STDERR_FILENO, p, strlen(p));
240 	(void)write(STDERR_FILENO, "\n", 1);
241 }
242 #endif
243