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
39static 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
74static void error(int);
75#endif
76static int gettype(char *, const char *const *);
77
78static const char *db_array[2] = { _PATH_DISKTAB, 0 };
79
80int
81setdisktab(const char *name)
82{
83 if (!name || !*name)
84 return -1;
85
86 db_array[0] = name;
87 return 0;
88}
89
90
91struct disklabel *
92getdiskbyname(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
208static int
209gettype(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
225static void
226error(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