main.c revision 1.46 1 1.46 christos /* $NetBSD: main.c,v 1.46 2016/01/31 18:57:29 christos Exp $ */
2 1.13 jmmv
3 1.13 jmmv /*
4 1.13 jmmv * Copyright (c) 2006 The NetBSD Foundation, Inc.
5 1.13 jmmv * All rights reserved.
6 1.13 jmmv *
7 1.13 jmmv * This code is derived from software contributed to The NetBSD Foundation
8 1.13 jmmv * by Julio M. Merino Vidal.
9 1.13 jmmv *
10 1.13 jmmv * Redistribution and use in source and binary forms, with or without
11 1.13 jmmv * modification, are permitted provided that the following conditions
12 1.13 jmmv * are met:
13 1.13 jmmv * 1. Redistributions of source code must retain the above copyright
14 1.13 jmmv * notice, this list of conditions and the following disclaimer.
15 1.13 jmmv * 2. Redistributions in binary form must reproduce the above copyright
16 1.13 jmmv * notice, this list of conditions and the following disclaimer in the
17 1.13 jmmv * documentation and/or other materials provided with the distribution.
18 1.13 jmmv *
19 1.13 jmmv * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
20 1.13 jmmv * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
21 1.13 jmmv * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
22 1.13 jmmv * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
23 1.13 jmmv * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
24 1.13 jmmv * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
25 1.13 jmmv * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
26 1.13 jmmv * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
27 1.13 jmmv * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
28 1.13 jmmv * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
29 1.13 jmmv * POSSIBILITY OF SUCH DAMAGE.
30 1.13 jmmv */
31 1.1 dyoung
32 1.1 dyoung /*
33 1.1 dyoung * Copyright (c) 1987, 1993
34 1.1 dyoung * The Regents of the University of California. All rights reserved.
35 1.1 dyoung *
36 1.1 dyoung * This code is derived from software contributed to Berkeley by
37 1.1 dyoung * Symmetric Computer Systems.
38 1.1 dyoung *
39 1.1 dyoung * Redistribution and use in source and binary forms, with or without
40 1.1 dyoung * modification, are permitted provided that the following conditions
41 1.1 dyoung * are met:
42 1.1 dyoung * 1. Redistributions of source code must retain the above copyright
43 1.1 dyoung * notice, this list of conditions and the following disclaimer.
44 1.1 dyoung * 2. Redistributions in binary form must reproduce the above copyright
45 1.1 dyoung * notice, this list of conditions and the following disclaimer in the
46 1.1 dyoung * documentation and/or other materials provided with the distribution.
47 1.1 dyoung * 3. Neither the name of the University nor the names of its contributors
48 1.1 dyoung * may be used to endorse or promote products derived from this software
49 1.1 dyoung * without specific prior written permission.
50 1.1 dyoung *
51 1.1 dyoung * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
52 1.1 dyoung * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
53 1.1 dyoung * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
54 1.1 dyoung * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
55 1.1 dyoung * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
56 1.1 dyoung * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
57 1.1 dyoung * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
58 1.1 dyoung * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
59 1.1 dyoung * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
60 1.1 dyoung * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
61 1.1 dyoung * SUCH DAMAGE.
62 1.1 dyoung */
63 1.1 dyoung
64 1.1 dyoung #if HAVE_NBTOOL_CONFIG_H
65 1.1 dyoung #include "nbtool_config.h"
66 1.1 dyoung #endif
67 1.1 dyoung
68 1.1 dyoung #include <sys/cdefs.h>
69 1.1 dyoung #ifndef lint
70 1.19 lukem __COPYRIGHT("@(#) Copyright (c) 1987, 1993\
71 1.19 lukem The Regents of the University of California. All rights reserved.");
72 1.1 dyoung #endif /* not lint */
73 1.1 dyoung
74 1.1 dyoung #ifndef lint
75 1.1 dyoung #if 0
76 1.1 dyoung static char sccsid[] = "@(#)disklabel.c 8.4 (Berkeley) 5/4/95";
77 1.1 dyoung /* from static char sccsid[] = "@(#)disklabel.c 1.2 (Symmetric) 11/28/85"; */
78 1.1 dyoung #else
79 1.46 christos __RCSID("$NetBSD: main.c,v 1.46 2016/01/31 18:57:29 christos Exp $");
80 1.1 dyoung #endif
81 1.1 dyoung #endif /* not lint */
82 1.1 dyoung
83 1.1 dyoung #include <sys/param.h>
84 1.1 dyoung #include <sys/file.h>
85 1.1 dyoung #include <sys/stat.h>
86 1.1 dyoung #include <sys/wait.h>
87 1.1 dyoung #define DKTYPENAMES
88 1.1 dyoung #define FSTYPENAMES
89 1.1 dyoung
90 1.1 dyoung #include <ctype.h>
91 1.1 dyoung #include <err.h>
92 1.1 dyoung #include <errno.h>
93 1.1 dyoung #include <signal.h>
94 1.1 dyoung #include <string.h>
95 1.1 dyoung #include <stdio.h>
96 1.1 dyoung #include <stdlib.h>
97 1.17 dyoung #include <limits.h>
98 1.1 dyoung #include <unistd.h>
99 1.1 dyoung
100 1.1 dyoung #include <ufs/ufs/dinode.h>
101 1.1 dyoung #include <ufs/ffs/fs.h>
102 1.1 dyoung
103 1.1 dyoung #if HAVE_NBTOOL_CONFIG_H
104 1.1 dyoung #include <nbinclude/sys/disklabel.h>
105 1.6 dsl #include <nbinclude/sys/disklabel_acorn.h>
106 1.1 dyoung #include <nbinclude/sys/bootblock.h>
107 1.1 dyoung #include "../../include/disktab.h"
108 1.1 dyoung #else
109 1.1 dyoung #include <sys/ioctl.h>
110 1.1 dyoung #include <sys/disklabel.h>
111 1.6 dsl #include <sys/disklabel_acorn.h>
112 1.1 dyoung #include <sys/bootblock.h>
113 1.1 dyoung #include <util.h>
114 1.1 dyoung #include <disktab.h>
115 1.1 dyoung #endif /* HAVE_NBTOOL_CONFIG_H */
116 1.1 dyoung
117 1.1 dyoung #include "pathnames.h"
118 1.1 dyoung #include "extern.h"
119 1.1 dyoung #include "dkcksum.h"
120 1.22 tsutsui #include "bswap.h"
121 1.1 dyoung
122 1.1 dyoung /*
123 1.1 dyoung * Disklabel: read and write disklabels.
124 1.1 dyoung * The label is usually placed on one of the first sectors of the disk.
125 1.1 dyoung * Many machines also place a bootstrap in the same area,
126 1.1 dyoung * in which case the label is embedded in the bootstrap.
127 1.1 dyoung * The bootstrap source must leave space at the proper offset
128 1.1 dyoung * for the label on such machines.
129 1.1 dyoung */
130 1.1 dyoung
131 1.1 dyoung #ifndef BBSIZE
132 1.1 dyoung #define BBSIZE 8192 /* size of boot area, with label */
133 1.1 dyoung #endif
134 1.1 dyoung
135 1.6 dsl #define DISKMAGIC_REV bswap32(DISKMAGIC)
136 1.6 dsl /* To delete a label, we just invert the magic numbers */
137 1.6 dsl #define DISKMAGIC_DELETED (~DISKMAGIC)
138 1.6 dsl #define DISKMAGIC_DELETED_REV bswap32(~DISKMAGIC)
139 1.6 dsl
140 1.1 dyoung #define DEFEDITOR _PATH_VI
141 1.1 dyoung
142 1.6 dsl char specname[MAXPATHLEN];
143 1.1 dyoung
144 1.6 dsl /* Some global data, all too hard to pass about */
145 1.15 drochner char bootarea[BBSIZE]; /* Buffer matching part of disk */
146 1.6 dsl int bootarea_len; /* Number of bytes we actually read */
147 1.6 dsl static struct disklabel lab; /* The label we have updated */
148 1.6 dsl
149 1.6 dsl static int Aflag; /* Action all labels */
150 1.6 dsl static int Fflag; /* Read/write from file */
151 1.6 dsl static int rflag; /* Read/write direct from disk */
152 1.6 dsl static int tflag; /* Format output as disktab */
153 1.6 dsl int Cflag; /* CHS format output */
154 1.6 dsl static int Dflag; /* Delete old labels (use with write) */
155 1.6 dsl static int Iflag; /* Read/write direct, but default if absent */
156 1.13 jmmv static int lflag; /* List all known file system types and exit */
157 1.6 dsl static int mflag; /* Expect disk to contain an MBR */
158 1.6 dsl static int verbose;
159 1.6 dsl static int read_all; /* set if op = READ && Aflag */
160 1.6 dsl
161 1.6 dsl static int write_label(int);
162 1.6 dsl static int readlabel_direct(int);
163 1.6 dsl static void writelabel_direct(int);
164 1.6 dsl static int update_label(int, u_int, u_int);
165 1.6 dsl static struct disklabel *find_label(int, u_int);
166 1.46 christos #if !defined(NATIVELABEL_ONLY)
167 1.29 matt static void getmachineparams(const char *);
168 1.46 christos #endif
169 1.1 dyoung
170 1.1 dyoung static void makedisktab(FILE *, struct disklabel *);
171 1.6 dsl static void makelabel(const char *, const char *);
172 1.1 dyoung static void l_perror(const char *);
173 1.6 dsl static void readlabel(int);
174 1.6 dsl static int edit(int);
175 1.6 dsl static int editit(const char *);
176 1.1 dyoung static char *skip(char *);
177 1.1 dyoung static char *word(char *);
178 1.1 dyoung static int getasciilabel(FILE *, struct disklabel *);
179 1.25 joerg __dead static void usage(void);
180 1.13 jmmv static int qsort_strcmp(const void *, const void *);
181 1.1 dyoung static int getulong(const char *, char, char **,
182 1.1 dyoung unsigned long *, unsigned long);
183 1.1 dyoung #define GETNUM32(a, v) getulong(a, '\0', NULL, v, UINT32_MAX)
184 1.1 dyoung #define GETNUM16(a, v) getulong(a, '\0', NULL, v, UINT16_MAX)
185 1.1 dyoung #define GETNUM8(a, v) getulong(a, '\0', NULL, v, UINT8_MAX)
186 1.1 dyoung
187 1.6 dsl static int set_writable_fd = -1;
188 1.6 dsl
189 1.46 christos #if !defined(NATIVELABEL_ONLY)
190 1.29 matt static u_int labeloffset;
191 1.29 matt static u_int labelsector;
192 1.29 matt static int labelusesmbr;
193 1.29 matt u_int maxpartitions;
194 1.29 matt static int byteorder;
195 1.29 matt
196 1.29 matt static int biendian_p;
197 1.29 matt #ifndef HAVE_NBTOOL_CONFIG_H
198 1.29 matt static int native_p = 1;
199 1.29 matt #endif
200 1.29 matt int bswap_p;
201 1.29 matt
202 1.29 matt static const struct disklabel_params {
203 1.29 matt const char *machine;
204 1.45 christos u_char labelusesmbr : 1;
205 1.45 christos u_char labelsector : 7;
206 1.44 christos u_char maxpartitions;
207 1.44 christos u_char raw_part;
208 1.44 christos u_char oldmaxpartitions;
209 1.44 christos u_short labeloffset;
210 1.44 christos u_short byteorder;
211 1.29 matt } disklabel_params[] = {
212 1.44 christos { "mvme68k", 0, 0, 8, 2, 0, 0, BIG_ENDIAN }, /* m68k */
213 1.44 christos { "next68k", 0, 0, 8, 2, 0, 0, BIG_ENDIAN }, /* m68k */
214 1.29 matt
215 1.44 christos { "algor", 0, 0, 8, 2, 0, 64, LITTLE_ENDIAN }, /* mips */
216 1.44 christos { "alpha", 0, 0, 8, 2, 0, 64, LITTLE_ENDIAN }, /* alpha */
217 1.44 christos { "luna68k", 0, 0, 8, 2, 0, 64, BIG_ENDIAN }, /* m68k */
218 1.44 christos { "mac68k", 0, 0, 8, 2, 0, 64, BIG_ENDIAN }, /* m68k */
219 1.44 christos { "news68k", 0, 0, 8, 2, 0, 64, BIG_ENDIAN }, /* m68k */
220 1.44 christos { "newsmips", 0, 0, 8, 2, 0, 64, BIG_ENDIAN }, /* mips */
221 1.44 christos { "pmax", 0, 0, 8, 2, 0, 64, LITTLE_ENDIAN }, /* mips */
222 1.44 christos { "sun2", 0, 0, 8, 2, 0, 64, BIG_ENDIAN }, /* m68k */
223 1.44 christos { "sun68k", 0, 0, 8, 2, 0, 64, BIG_ENDIAN }, /* m68010 */
224 1.44 christos { "x68k", 0, 0, 8, 2, 0, 64, BIG_ENDIAN }, /* m68010 */
225 1.44 christos
226 1.44 christos { "vax", 0, 0, 12, 2, 8, 64, LITTLE_ENDIAN }, /* vax */
227 1.44 christos
228 1.44 christos { "amiga", 0, 0, 16, 2, 0, 64, BIG_ENDIAN }, /* m68k */
229 1.44 christos { "amigappc", 0, 0, 16, 2, 0, 64, BIG_ENDIAN }, /* powerpc */
230 1.44 christos { "evbmips", 0, 0, 16, 2, 0, 64, 0 }, /* mips */
231 1.44 christos { "evbppc", 0, 0, 16, 2, 0, 64, BIG_ENDIAN }, /* powerpc */
232 1.44 christos
233 1.44 christos { "sparc", 0, 0, 8, 2, 0, 128, BIG_ENDIAN }, /* sun */
234 1.44 christos { "sparc64", 0, 0, 8, 2, 0, 128, BIG_ENDIAN }, /* sun */
235 1.44 christos { "sun3", 0, 0, 8, 2, 0, 128, BIG_ENDIAN }, /* sun */
236 1.44 christos
237 1.44 christos { "atari", 0, 0, 16, 2, 0, 516, BIG_ENDIAN }, /* m68k */
238 1.44 christos
239 1.44 christos { "mipsco", 0, 1, 8, 2, 0, 0, BIG_ENDIAN }, /* mips */
240 1.44 christos { "mvmeppc", 0, 1, 8, 3, 0, 0, BIG_ENDIAN }, /* powerpc */
241 1.44 christos
242 1.44 christos { "bebox", 0, 1, 8, 3, 0, 0, BIG_ENDIAN }, /* powerpc */
243 1.44 christos
244 1.44 christos { "emips", 0, 1, 16, 2, 0, 0, BIG_ENDIAN }, /* mips */
245 1.44 christos { "hppa", 0, 1, 16, 2, 0, 0, BIG_ENDIAN }, /* hppa */
246 1.44 christos { "ibmnws", 0, 1, 16, 2, 0, 0, BIG_ENDIAN }, /* powerpc */
247 1.44 christos { "ofppc", 0, 1, 16, 2, 0, 0, BIG_ENDIAN }, /* powerpc */
248 1.44 christos { "rs6000", 0, 1, 16, 2, 0, 0, BIG_ENDIAN }, /* powerpc */
249 1.44 christos { "sandpoint", 0, 1, 16, 2, 0, 0, BIG_ENDIAN }, /* powerpc */
250 1.44 christos { "sgimips", 0, 1, 16, 2, 0, 0, BIG_ENDIAN }, /* mips */
251 1.44 christos
252 1.44 christos { "sbmips", 0, 1, 16, 3, 0, 0, 0 }, /* mips */
253 1.44 christos
254 1.44 christos { "cesfic", 0, 2, 8, 2, 0, 0, BIG_ENDIAN }, /* m68k */
255 1.44 christos { "hp300", 0, 2, 8, 2, 0, 0, BIG_ENDIAN }, /* m68k */
256 1.44 christos
257 1.44 christos { "ews4800mips",0, 9, 16, 15, 0, 0, BIG_ENDIAN }, /* mips */
258 1.44 christos
259 1.44 christos { "macppc", 1, 0, 16, 2, 0, 64, BIG_ENDIAN }, /* powerpc */
260 1.44 christos { "pmon", 1, 0, 16, 2, 0, 64, 0 }, /* evbmips */
261 1.44 christos
262 1.45 christos { "prep", 1, 1, 8, 2, 0, 0, BIG_ENDIAN }, /* powerpc */
263 1.44 christos
264 1.45 christos { "dreamcast", 1, 1, 16, 2, 0, 0, LITTLE_ENDIAN }, /* sh3 */
265 1.45 christos { "evbarm64", 1, 1, 16, 2, 0, 0, 0 }, /* aarch64 */
266 1.45 christos { "evbcf", 1, 1, 16, 2, 0, 0, BIG_ENDIAN }, /* coldfire */
267 1.45 christos { "evbppc-mbr", 1, 1, 16, 2, 0, 0, BIG_ENDIAN }, /* powerpc */
268 1.45 christos { "evbsh3", 1, 1, 16, 2, 0, 0, 0 }, /* sh3 */
269 1.45 christos { "hpcsh", 1, 1, 16, 2, 0, 0, LITTLE_ENDIAN }, /* sh3 */
270 1.45 christos { "mmeye", 1, 1, 16, 2, 0, 0, 0 }, /* sh3 */
271 1.45 christos { "or1k", 1, 1, 16, 2, 0, 0, BIG_ENDIAN }, /* or1k */
272 1.45 christos { "riscv", 1, 1, 16, 2, 0, 0, LITTLE_ENDIAN }, /* riscv */
273 1.45 christos
274 1.45 christos { "acorn26", 1, 1, 16, 2, 8, 0, LITTLE_ENDIAN }, /* arm */
275 1.45 christos { "acorn32", 1, 1, 16, 2, 8, 0, LITTLE_ENDIAN }, /* arm */
276 1.45 christos { "cats", 1, 1, 16, 2, 8, 0, LITTLE_ENDIAN }, /* arm */
277 1.45 christos { "evbarm", 1, 1, 16, 2, 8, 0, 0 }, /* arm */
278 1.45 christos { "iyonix", 1, 1, 16, 2, 8, 0, LITTLE_ENDIAN }, /* arm */
279 1.45 christos { "netwinder", 1, 1, 16, 2, 8, 0, LITTLE_ENDIAN }, /* arm */
280 1.45 christos { "shark", 1, 1, 16, 2, 8, 0, LITTLE_ENDIAN }, /* arm */
281 1.45 christos
282 1.45 christos { "amd64", 1, 1, 16, 3, 0, 0, LITTLE_ENDIAN }, /* x86 */
283 1.45 christos { "arc", 1, 1, 16, 3, 0, 0, LITTLE_ENDIAN }, /* mips */
284 1.45 christos { "cobalt", 1, 1, 16, 3, 0, 0, LITTLE_ENDIAN }, /* mips */
285 1.45 christos { "landisk", 1, 1, 16, 3, 0, 0, LITTLE_ENDIAN }, /* sh3 */
286 1.45 christos
287 1.45 christos { "epoc32", 1, 1, 16, 3, 8, 0, LITTLE_ENDIAN }, /* arm */
288 1.45 christos { "hpcarm", 1, 1, 16, 3, 8, 0, LITTLE_ENDIAN }, /* arm */
289 1.45 christos { "hpcmips", 1, 1, 16, 3, 8, 0, LITTLE_ENDIAN }, /* mips */
290 1.45 christos { "i386", 1, 1, 16, 3, 8, 0, LITTLE_ENDIAN }, /* x86 */
291 1.45 christos { "ia64", 1, 1, 16, 3, 8, 0, LITTLE_ENDIAN }, /* x86 */
292 1.45 christos { "zaurus", 1, 1, 16, 3, 8, 0, LITTLE_ENDIAN }, /* arm */
293 1.29 matt
294 1.44 christos { NULL, 0, 0, 0, 0, 0, 0, 0 }, /* must be last */
295 1.29 matt };
296 1.29 matt
297 1.29 matt #ifndef HAVE_NBTOOL_CONFIG_H
298 1.29 matt static struct disklabel_params native_params;
299 1.1 dyoung #endif
300 1.1 dyoung
301 1.29 matt static const struct arch_endian {
302 1.29 matt int byteorder;
303 1.29 matt const char *arch;
304 1.29 matt } arch_endians[] = {
305 1.40 matt { LITTLE_ENDIAN, "aarch64" },
306 1.29 matt { LITTLE_ENDIAN, "alpha" },
307 1.29 matt { LITTLE_ENDIAN, "arm" },
308 1.29 matt { LITTLE_ENDIAN, "earm" },
309 1.29 matt { LITTLE_ENDIAN, "earmhf" },
310 1.36 riz { LITTLE_ENDIAN, "earmv4" },
311 1.36 riz { LITTLE_ENDIAN, "earmv5" },
312 1.36 riz { LITTLE_ENDIAN, "earmv6" },
313 1.36 riz { LITTLE_ENDIAN, "earmv6hf" },
314 1.36 riz { LITTLE_ENDIAN, "earmv7" },
315 1.36 riz { LITTLE_ENDIAN, "earmv7hf" },
316 1.29 matt { LITTLE_ENDIAN, "i386" },
317 1.29 matt { LITTLE_ENDIAN, "ia64" },
318 1.29 matt { LITTLE_ENDIAN, "mipsel" },
319 1.29 matt { LITTLE_ENDIAN, "mips64el" },
320 1.42 matt { LITTLE_ENDIAN, "riscv32" },
321 1.42 matt { LITTLE_ENDIAN, "riscv64" },
322 1.29 matt { LITTLE_ENDIAN, "sh3el" },
323 1.29 matt { LITTLE_ENDIAN, "vax" },
324 1.32 skrll { LITTLE_ENDIAN, "x86_64" },
325 1.29 matt
326 1.40 matt { BIG_ENDIAN, "aarch64eb" },
327 1.29 matt { BIG_ENDIAN, "armeb" },
328 1.37 matt { BIG_ENDIAN, "coldfire" },
329 1.29 matt { BIG_ENDIAN, "earmeb" },
330 1.29 matt { BIG_ENDIAN, "earmhfeb" },
331 1.36 riz { BIG_ENDIAN, "earmv4eb" },
332 1.36 riz { BIG_ENDIAN, "earmv5eb" },
333 1.36 riz { BIG_ENDIAN, "earmv6eb" },
334 1.36 riz { BIG_ENDIAN, "earmv6hfeb" },
335 1.36 riz { BIG_ENDIAN, "earmv7eb" },
336 1.36 riz { BIG_ENDIAN, "earmv7hfeb" },
337 1.29 matt { BIG_ENDIAN, "hppa" },
338 1.37 matt { BIG_ENDIAN, "m68000" },
339 1.29 matt { BIG_ENDIAN, "m68k" },
340 1.29 matt { BIG_ENDIAN, "mipseb" },
341 1.29 matt { BIG_ENDIAN, "mips64eb" },
342 1.42 matt { BIG_ENDIAN, "or1k" },
343 1.29 matt { BIG_ENDIAN, "powerpc" },
344 1.29 matt { BIG_ENDIAN, "sh3eb" },
345 1.29 matt { BIG_ENDIAN, "sparc" },
346 1.29 matt { BIG_ENDIAN, "sparc64" },
347 1.29 matt
348 1.29 matt { 0, NULL },
349 1.29 matt };
350 1.29 matt
351 1.6 dsl /* Default location for label - only used if we don't find one to update */
352 1.17 dyoung #define LABEL_OFFSET (dklabel_getlabelsector() * DEV_BSIZE + dklabel_getlabeloffset())
353 1.46 christos #else
354 1.46 christos #define labeloffset LABELOFFSET
355 1.46 christos #define labelsector LABELSECTOR
356 1.46 christos #define labelusesmbr LABELUSESMBR
357 1.46 christos #define maxpartitions MAXPARTITIONS
358 1.46 christos #define LABEL_OFFSET LABELOFFSET
359 1.46 christos #endif /* !NATIVELABEL_ONLY */
360 1.6 dsl
361 1.6 dsl /*
362 1.6 dsl * For portability it doesn't make sense to use any other value....
363 1.6 dsl * Except, maybe, the size of a physical sector.
364 1.6 dsl * This value is used if we have to write a label to the start of an mbr ptn.
365 1.6 dsl */
366 1.6 dsl #ifndef LABELOFFSET_MBR
367 1.6 dsl #define LABELOFFSET_MBR 512
368 1.6 dsl #endif
369 1.6 dsl
370 1.6 dsl #if HAVE_NBTOOL_CONFIG_H
371 1.6 dsl static int
372 1.6 dsl opendisk(const char *path, int flags, char *buf, int buflen, int cooked)
373 1.6 dsl {
374 1.6 dsl int f;
375 1.6 dsl f = open(path, flags, 0);
376 1.6 dsl strlcpy(buf, path, buflen);
377 1.6 dsl return f;
378 1.6 dsl }
379 1.6 dsl #endif /* HAVE_NBTOOL_CONFIG_H */
380 1.6 dsl
381 1.46 christos #if !defined(NATIVELABEL_ONLY)
382 1.29 matt static void
383 1.29 matt setbyteorder(int new_byteorder)
384 1.29 matt {
385 1.29 matt static int set_p;
386 1.29 matt
387 1.29 matt if ((!biendian_p || set_p)
388 1.29 matt && byteorder != 0
389 1.29 matt && byteorder != new_byteorder) {
390 1.29 matt warn("changing %s byteorder to %s",
391 1.29 matt byteorder == LITTLE_ENDIAN ? "le" : "be",
392 1.29 matt new_byteorder == LITTLE_ENDIAN ? "le" : "be");
393 1.29 matt }
394 1.29 matt byteorder = new_byteorder;
395 1.29 matt biendian_p = 0;
396 1.29 matt set_p = 1;
397 1.29 matt }
398 1.29 matt
399 1.29 matt static void
400 1.29 matt getmachineparams(const char *mach)
401 1.29 matt {
402 1.29 matt const struct disklabel_params *dp = disklabel_params;
403 1.29 matt for (; dp->machine != NULL; dp++) {
404 1.29 matt if (!strcmp(mach, dp->machine)) {
405 1.29 matt labelusesmbr = dp->labelusesmbr;
406 1.29 matt labelsector = dp->labelsector;
407 1.29 matt labeloffset = dp->labeloffset;
408 1.29 matt maxpartitions = dp->maxpartitions;
409 1.29 matt biendian_p = (dp->byteorder == 0);
410 1.29 matt if (!biendian_p)
411 1.29 matt setbyteorder(dp->byteorder);
412 1.29 matt return;
413 1.29 matt }
414 1.29 matt }
415 1.29 matt errx(1, "%s: unknown machine type", mach);
416 1.29 matt }
417 1.29 matt
418 1.29 matt static void
419 1.29 matt getarchbyteorder(const char *arch)
420 1.29 matt {
421 1.29 matt const struct arch_endian *p = arch_endians;
422 1.29 matt for (; p->arch != NULL; p++) {
423 1.29 matt if (!strcmp(arch, p->arch)) {
424 1.29 matt setbyteorder(p->byteorder);
425 1.29 matt return;
426 1.29 matt }
427 1.29 matt }
428 1.29 matt errx(1, "%s: unknown arch", arch);
429 1.29 matt }
430 1.29 matt
431 1.17 dyoung static daddr_t
432 1.17 dyoung dklabel_getlabelsector(void)
433 1.17 dyoung {
434 1.17 dyoung unsigned long int nval;
435 1.17 dyoung char *end;
436 1.17 dyoung const char *val;
437 1.17 dyoung
438 1.17 dyoung if ((val = getenv("DISKLABELSECTOR")) == NULL)
439 1.29 matt return labelsector;
440 1.17 dyoung if ((nval = strtoul(val, &end, 10)) == ULONG_MAX && errno == ERANGE)
441 1.17 dyoung err(EXIT_FAILURE, "DISKLABELSECTOR in environment");
442 1.17 dyoung return nval;
443 1.17 dyoung }
444 1.17 dyoung
445 1.17 dyoung static off_t
446 1.17 dyoung dklabel_getlabeloffset(void)
447 1.17 dyoung {
448 1.17 dyoung unsigned long int nval;
449 1.17 dyoung char *end;
450 1.17 dyoung const char *val;
451 1.17 dyoung
452 1.17 dyoung if ((val = getenv("DISKLABELOFFSET")) == NULL)
453 1.29 matt return labeloffset;
454 1.17 dyoung if ((nval = strtoul(val, &end, 10)) == ULONG_MAX && errno == ERANGE)
455 1.17 dyoung err(EXIT_FAILURE, "DISKLABELOFFSET in environment");
456 1.17 dyoung return nval;
457 1.17 dyoung }
458 1.46 christos #endif /* !NATIVELABEL_ONLY */
459 1.17 dyoung
460 1.6 dsl static void
461 1.6 dsl clear_writable(void)
462 1.6 dsl {
463 1.6 dsl static int zero = 0;
464 1.6 dsl dk_ioctl(set_writable_fd, DIOCWLABEL, &zero);
465 1.6 dsl }
466 1.6 dsl
467 1.1 dyoung int
468 1.1 dyoung main(int argc, char *argv[])
469 1.1 dyoung {
470 1.1 dyoung FILE *t;
471 1.6 dsl int ch, f, error;
472 1.6 dsl char *dkname;
473 1.46 christos #if !defined(NATIVELABEL_ONLY)
474 1.29 matt char *cp;
475 1.46 christos #endif
476 1.6 dsl struct stat sb;
477 1.6 dsl int writable;
478 1.6 dsl enum {
479 1.6 dsl UNSPEC, EDIT, READ, RESTORE, SETWRITABLE, SETREADONLY,
480 1.46 christos WRITE,
481 1.46 christos #if !defined(NO_INTERACT)
482 1.46 christos INTERACT,
483 1.46 christos #endif
484 1.46 christos DELETE
485 1.6 dsl } op = UNSPEC, old_op;
486 1.6 dsl
487 1.29 matt #ifndef HAVE_NBTOOL_CONFIG_H
488 1.46 christos #if !defined(NATIVELABEL_ONLY)
489 1.29 matt labeloffset = native_params.labeloffset = getlabeloffset();
490 1.29 matt labelsector = native_params.labelsector = getlabelsector();
491 1.29 matt labelusesmbr = native_params.labelusesmbr = getlabelusesmbr();
492 1.29 matt maxpartitions = native_params.maxpartitions = getmaxpartitions();
493 1.31 matt byteorder = native_params.byteorder = BYTE_ORDER;
494 1.29 matt #endif
495 1.46 christos #endif
496 1.29 matt
497 1.46 christos #if !defined(NATIVELABEL_ONLY)
498 1.29 matt if ((cp = getenv("MACHINE")) != NULL) {
499 1.29 matt getmachineparams(cp);
500 1.29 matt }
501 1.29 matt
502 1.29 matt if ((cp = getenv("MACHINE_ARCH")) != NULL) {
503 1.29 matt getarchbyteorder(cp);
504 1.29 matt }
505 1.46 christos #endif
506 1.29 matt
507 1.29 matt mflag = labelusesmbr;
508 1.26 bouyer if (mflag < 0) {
509 1.29 matt #if HAVE_NBTOOL_CONFIG_H
510 1.29 matt warn("getlabelusesmbr() failed");
511 1.29 matt #else
512 1.26 bouyer warn("getlabelusesmbr() failed");
513 1.26 bouyer mflag = LABELUSESMBR;
514 1.29 matt #endif
515 1.26 bouyer }
516 1.6 dsl #if HAVE_NBTOOL_CONFIG_H
517 1.6 dsl /* We must avoid doing any ioctl requests */
518 1.6 dsl Fflag = rflag = 1;
519 1.6 dsl #endif
520 1.1 dyoung
521 1.1 dyoung error = 0;
522 1.29 matt while ((ch = getopt(argc, argv, "AB:CDFIM:NRWef:ilmrtvw")) != -1) {
523 1.6 dsl old_op = op;
524 1.1 dyoung switch (ch) {
525 1.6 dsl case 'A': /* Action all labels */
526 1.6 dsl Aflag = 1;
527 1.6 dsl rflag = 1;
528 1.6 dsl break;
529 1.6 dsl case 'C': /* Display in CHS format */
530 1.6 dsl Cflag = 1;
531 1.6 dsl break;
532 1.6 dsl case 'D': /* Delete all existing labels */
533 1.6 dsl Dflag = 1;
534 1.6 dsl rflag = 1;
535 1.6 dsl break;
536 1.6 dsl case 'F': /* Treat 'disk' as a regular file */
537 1.6 dsl Fflag = 1;
538 1.6 dsl rflag = 1; /* Force direct access */
539 1.6 dsl break;
540 1.6 dsl case 'I': /* Use default label if none found */
541 1.6 dsl Iflag = 1;
542 1.6 dsl rflag = 1; /* Implies direct access */
543 1.1 dyoung break;
544 1.6 dsl case 'R': /* Restore label from text file */
545 1.1 dyoung op = RESTORE;
546 1.1 dyoung break;
547 1.46 christos #if !defined(NATIVELABEL_ONLY)
548 1.29 matt case 'B': /* byteorder */
549 1.29 matt if (!strcmp(optarg, "be")) {
550 1.29 matt setbyteorder(BIG_ENDIAN);
551 1.29 matt } else if (!strcmp(optarg, "le")) {
552 1.29 matt setbyteorder(LITTLE_ENDIAN);
553 1.29 matt } else {
554 1.29 matt errx(1, "%s: not be or le", optarg);
555 1.29 matt }
556 1.29 matt break;
557 1.29 matt case 'M': /* machine type */
558 1.29 matt getmachineparams(optarg);
559 1.29 matt break;
560 1.46 christos #endif
561 1.6 dsl case 'N': /* Disallow writes to label sector */
562 1.6 dsl op = SETREADONLY;
563 1.1 dyoung break;
564 1.6 dsl case 'W': /* Allow writes to label sector */
565 1.1 dyoung op = SETWRITABLE;
566 1.1 dyoung break;
567 1.6 dsl case 'e': /* Edit label with $EDITOR */
568 1.1 dyoung op = EDIT;
569 1.1 dyoung break;
570 1.6 dsl case 'f': /* Name of disktab file */
571 1.1 dyoung if (setdisktab(optarg) == -1)
572 1.1 dyoung usage();
573 1.1 dyoung break;
574 1.46 christos #if !defined(NO_INTERACT)
575 1.6 dsl case 'i': /* Edit using built-in editor */
576 1.1 dyoung op = INTERACT;
577 1.1 dyoung break;
578 1.46 christos #endif /* !NO_INTERACT */
579 1.13 jmmv case 'l': /* List all known file system types and exit */
580 1.13 jmmv lflag = 1;
581 1.13 jmmv break;
582 1.6 dsl case 'm': /* Expect disk to have an MBR */
583 1.6 dsl mflag ^= 1;
584 1.6 dsl break;
585 1.6 dsl case 'r': /* Read/write label directly from disk */
586 1.6 dsl rflag = 1;
587 1.1 dyoung break;
588 1.6 dsl case 't': /* Format output as a disktab entry */
589 1.6 dsl tflag = 1;
590 1.1 dyoung break;
591 1.6 dsl case 'v': /* verbose/diag output */
592 1.6 dsl verbose++;
593 1.6 dsl break;
594 1.6 dsl case 'w': /* Write label based on disktab entry */
595 1.1 dyoung op = WRITE;
596 1.1 dyoung break;
597 1.1 dyoung case '?':
598 1.1 dyoung default:
599 1.1 dyoung usage();
600 1.6 dsl }
601 1.6 dsl if (old_op != UNSPEC && old_op != op)
602 1.6 dsl usage();
603 1.1 dyoung }
604 1.29 matt
605 1.46 christos #if !defined(NATIVELABEL_ONLY)
606 1.29 matt if (maxpartitions == 0) {
607 1.29 matt errx(1, "unknown label: use -M/-B and $MACHINE/$MACHINE_ARCH");
608 1.29 matt }
609 1.29 matt if (byteorder != BIG_ENDIAN && byteorder != LITTLE_ENDIAN) {
610 1.29 matt errx(1, "unknown byteorder");
611 1.29 matt }
612 1.29 matt bswap_p = (byteorder != BYTE_ORDER);
613 1.29 matt #ifdef DEBUG
614 1.30 matt printf("labelusesmbr=%d labelsector=%u labeloffset=%u maxpartitions=%u\n",
615 1.29 matt labelusesmbr, labelsector, labeloffset, maxpartitions);
616 1.29 matt printf("byteorder=%d bswap_p=%d\n", byteorder, bswap_p);
617 1.29 matt #endif
618 1.29 matt #ifndef HAVE_NBTOOL_CONFIG_H
619 1.29 matt /*
620 1.29 matt * If the disklabel has the same location as the native disklabel and
621 1.29 matt * fewer or equal paritions, we can use the native ioctls. Otherwise
622 1.29 matt * force file/raw access.
623 1.29 matt */
624 1.29 matt native_p = native_params.labelusesmbr == labelusesmbr
625 1.29 matt && native_params.labelsector == labelsector
626 1.29 matt && native_params.labeloffset == labeloffset
627 1.30 matt && maxpartitions <= native_params.maxpartitions
628 1.29 matt && !bswap_p;
629 1.29 matt if (!native_p)
630 1.29 matt Fflag = rflag = 1;
631 1.29 matt #endif
632 1.46 christos #endif /* !NATIVELABEL_ONLY */
633 1.29 matt
634 1.1 dyoung argc -= optind;
635 1.1 dyoung argv += optind;
636 1.1 dyoung
637 1.13 jmmv if (lflag)
638 1.13 jmmv exit(list_fs_types() ? EXIT_SUCCESS : EXIT_FAILURE);
639 1.13 jmmv
640 1.1 dyoung if (op == UNSPEC)
641 1.6 dsl op = Dflag ? DELETE : READ;
642 1.1 dyoung
643 1.1 dyoung if (argc < 1)
644 1.1 dyoung usage();
645 1.1 dyoung
646 1.46 christos if (Iflag && op != EDIT
647 1.46 christos #if !defined(NO_INTERACT)
648 1.46 christos && op != INTERACT
649 1.46 christos #endif
650 1.46 christos )
651 1.1 dyoung usage();
652 1.1 dyoung
653 1.1 dyoung dkname = argv[0];
654 1.6 dsl f = opendisk(dkname, op == READ ? O_RDONLY : O_RDWR,
655 1.6 dsl specname, sizeof specname, 0);
656 1.1 dyoung if (f < 0)
657 1.1 dyoung err(4, "%s", specname);
658 1.1 dyoung
659 1.6 dsl if (!Fflag && fstat(f, &sb) == 0 && S_ISREG(sb.st_mode))
660 1.6 dsl Fflag = rflag = 1;
661 1.1 dyoung
662 1.1 dyoung switch (op) {
663 1.1 dyoung
664 1.6 dsl case DELETE: /* Remove all existing labels */
665 1.6 dsl if (argc != 1)
666 1.6 dsl usage();
667 1.6 dsl Dflag = 2;
668 1.6 dsl writelabel_direct(f);
669 1.6 dsl break;
670 1.6 dsl
671 1.1 dyoung case EDIT:
672 1.1 dyoung if (argc != 1)
673 1.1 dyoung usage();
674 1.6 dsl readlabel(f);
675 1.6 dsl error = edit(f);
676 1.1 dyoung break;
677 1.1 dyoung
678 1.46 christos #if !defined(NO_INTERACT)
679 1.1 dyoung case INTERACT:
680 1.1 dyoung if (argc != 1)
681 1.1 dyoung usage();
682 1.6 dsl readlabel(f);
683 1.1 dyoung /*
684 1.1 dyoung * XXX: Fill some default values so checklabel does not fail
685 1.1 dyoung */
686 1.6 dsl if (lab.d_bbsize == 0)
687 1.6 dsl lab.d_bbsize = BBSIZE;
688 1.6 dsl if (lab.d_sbsize == 0)
689 1.6 dsl lab.d_sbsize = SBLOCKSIZE;
690 1.6 dsl interact(&lab, f);
691 1.1 dyoung break;
692 1.46 christos #endif /* !NO_INTERACT */
693 1.1 dyoung
694 1.1 dyoung case READ:
695 1.1 dyoung if (argc != 1)
696 1.1 dyoung usage();
697 1.6 dsl read_all = Aflag;
698 1.6 dsl readlabel(f);
699 1.6 dsl if (read_all)
700 1.6 dsl /* Label got printed in the bowels of readlabel */
701 1.6 dsl break;
702 1.1 dyoung if (tflag)
703 1.6 dsl makedisktab(stdout, &lab);
704 1.1 dyoung else {
705 1.6 dsl showinfo(stdout, &lab, specname);
706 1.6 dsl showpartitions(stdout, &lab, Cflag);
707 1.1 dyoung }
708 1.6 dsl error = checklabel(&lab);
709 1.1 dyoung if (error)
710 1.1 dyoung error += 100;
711 1.1 dyoung break;
712 1.1 dyoung
713 1.1 dyoung case RESTORE:
714 1.6 dsl if (argc != 2)
715 1.1 dyoung usage();
716 1.1 dyoung if (!(t = fopen(argv[1], "r")))
717 1.1 dyoung err(4, "%s", argv[1]);
718 1.6 dsl if (getasciilabel(t, &lab))
719 1.6 dsl error = write_label(f);
720 1.1 dyoung else
721 1.1 dyoung error = 1;
722 1.1 dyoung break;
723 1.1 dyoung
724 1.6 dsl case SETREADONLY:
725 1.6 dsl writable = 0;
726 1.6 dsl goto do_diocwlabel;
727 1.1 dyoung case SETWRITABLE:
728 1.6 dsl writable = 1;
729 1.6 dsl do_diocwlabel:
730 1.6 dsl if (argc != 1)
731 1.6 dsl usage();
732 1.6 dsl if (dk_ioctl(f, DIOCWLABEL, &writable) < 0)
733 1.1 dyoung err(4, "ioctl DIOCWLABEL");
734 1.1 dyoung break;
735 1.1 dyoung
736 1.6 dsl case WRITE: /* Create label from /etc/disktab entry & write */
737 1.1 dyoung if (argc < 2 || argc > 3)
738 1.1 dyoung usage();
739 1.6 dsl makelabel(argv[1], argv[2]);
740 1.6 dsl if (checklabel(&lab) == 0)
741 1.6 dsl error = write_label(f);
742 1.1 dyoung else
743 1.1 dyoung error = 1;
744 1.1 dyoung break;
745 1.1 dyoung
746 1.1 dyoung case UNSPEC:
747 1.1 dyoung usage();
748 1.1 dyoung
749 1.1 dyoung }
750 1.1 dyoung exit(error);
751 1.1 dyoung }
752 1.1 dyoung
753 1.1 dyoung /*
754 1.6 dsl * Construct a prototype disklabel from /etc/disktab.
755 1.1 dyoung */
756 1.1 dyoung static void
757 1.6 dsl makelabel(const char *type, const char *name)
758 1.1 dyoung {
759 1.1 dyoung struct disklabel *dp;
760 1.1 dyoung
761 1.1 dyoung dp = getdiskbyname(type);
762 1.1 dyoung if (dp == NULL)
763 1.1 dyoung errx(1, "unknown disk type: %s", type);
764 1.6 dsl lab = *dp;
765 1.1 dyoung
766 1.1 dyoung /* d_packname is union d_boot[01], so zero */
767 1.6 dsl (void)memset(lab.d_packname, 0, sizeof(lab.d_packname));
768 1.1 dyoung if (name)
769 1.6 dsl (void)strncpy(lab.d_packname, name, sizeof(lab.d_packname));
770 1.1 dyoung }
771 1.1 dyoung
772 1.6 dsl static int
773 1.6 dsl write_label(int f)
774 1.1 dyoung {
775 1.6 dsl int writable;
776 1.1 dyoung
777 1.6 dsl lab.d_magic = DISKMAGIC;
778 1.6 dsl lab.d_magic2 = DISKMAGIC;
779 1.6 dsl lab.d_checksum = 0;
780 1.6 dsl lab.d_checksum = dkcksum(&lab);
781 1.1 dyoung
782 1.6 dsl if (rflag) {
783 1.6 dsl /* Write the label directly to the disk */
784 1.1 dyoung
785 1.1 dyoung /*
786 1.1 dyoung * First set the kernel disk label,
787 1.1 dyoung * then write a label to the raw disk.
788 1.1 dyoung * If the SDINFO ioctl fails because it is unimplemented,
789 1.1 dyoung * keep going; otherwise, the kernel consistency checks
790 1.1 dyoung * may prevent us from changing the current (in-core)
791 1.1 dyoung * label.
792 1.1 dyoung */
793 1.6 dsl if (!Fflag && dk_ioctl(f, DIOCSDINFO, &lab) < 0 &&
794 1.1 dyoung errno != ENODEV && errno != ENOTTY) {
795 1.1 dyoung l_perror("ioctl DIOCSDINFO");
796 1.1 dyoung return (1);
797 1.1 dyoung }
798 1.1 dyoung /*
799 1.1 dyoung * write enable label sector before write (if necessary),
800 1.1 dyoung * disable after writing.
801 1.1 dyoung */
802 1.1 dyoung writable = 1;
803 1.6 dsl if (!Fflag) {
804 1.6 dsl if (dk_ioctl(f, DIOCWLABEL, &writable) < 0)
805 1.6 dsl perror("ioctl DIOCWLABEL");
806 1.6 dsl set_writable_fd = f;
807 1.6 dsl atexit(clear_writable);
808 1.1 dyoung }
809 1.1 dyoung
810 1.6 dsl writelabel_direct(f);
811 1.1 dyoung
812 1.1 dyoung /*
813 1.1 dyoung * Now issue a DIOCWDINFO. This will let the kernel convert the
814 1.1 dyoung * disklabel to some machdep format if needed.
815 1.1 dyoung */
816 1.6 dsl /* XXX: This is stupid! */
817 1.6 dsl if (!Fflag && dk_ioctl(f, DIOCWDINFO, &lab) < 0) {
818 1.1 dyoung l_perror("ioctl DIOCWDINFO");
819 1.1 dyoung return (1);
820 1.1 dyoung }
821 1.1 dyoung } else {
822 1.6 dsl /* Get the kernel to write the label */
823 1.6 dsl if (dk_ioctl(f, DIOCWDINFO, &lab) < 0) {
824 1.1 dyoung l_perror("ioctl DIOCWDINFO");
825 1.1 dyoung return (1);
826 1.1 dyoung }
827 1.1 dyoung }
828 1.1 dyoung
829 1.21 tsutsui #ifdef VAX_ALTLABELS
830 1.43 christos if (lab.d_type == DKTYPE_SMD && lab.d_flags & D_BADSECT &&
831 1.6 dsl lab.d_secsize == 512) {
832 1.6 dsl /* Write the label to the odd sectors of the last track! */
833 1.1 dyoung daddr_t alt;
834 1.1 dyoung int i;
835 1.6 dsl uint8_t sec0[512];
836 1.6 dsl
837 1.6 dsl if (pread(f, sec0, 512, 0) < 512) {
838 1.6 dsl warn("read master label to write alternates");
839 1.6 dsl return 0;
840 1.6 dsl }
841 1.1 dyoung
842 1.6 dsl alt = lab.d_ncylinders * lab.d_secpercyl - lab.d_nsectors;
843 1.20 mhitch for (i = 1; i < 11 && (uint32_t)i < lab.d_nsectors; i += 2) {
844 1.6 dsl if (pwrite(f, sec0, 512, (off_t)(alt + i) * 512) < 512)
845 1.1 dyoung warn("alternate label %d write", i/2);
846 1.1 dyoung }
847 1.1 dyoung }
848 1.21 tsutsui #endif /* VAX_ALTLABELS */
849 1.1 dyoung
850 1.6 dsl return 0;
851 1.6 dsl }
852 1.6 dsl
853 1.6 dsl int
854 1.6 dsl writelabel(int f, struct disklabel *lp)
855 1.6 dsl {
856 1.6 dsl if (lp != &lab)
857 1.6 dsl lab = *lp;
858 1.6 dsl return write_label(f);
859 1.1 dyoung }
860 1.1 dyoung
861 1.1 dyoung static void
862 1.1 dyoung l_perror(const char *s)
863 1.1 dyoung {
864 1.1 dyoung
865 1.1 dyoung switch (errno) {
866 1.1 dyoung
867 1.1 dyoung case ESRCH:
868 1.1 dyoung warnx("%s: No disk label on disk;\n"
869 1.1 dyoung "use \"disklabel -I\" to install initial label", s);
870 1.1 dyoung break;
871 1.1 dyoung
872 1.1 dyoung case EINVAL:
873 1.1 dyoung warnx("%s: Label magic number or checksum is wrong!\n"
874 1.1 dyoung "(disklabel or kernel is out of date?)", s);
875 1.1 dyoung break;
876 1.1 dyoung
877 1.1 dyoung case EBUSY:
878 1.1 dyoung warnx("%s: Open partition would move or shrink", s);
879 1.1 dyoung break;
880 1.1 dyoung
881 1.1 dyoung case EXDEV:
882 1.1 dyoung warnx("%s: Labeled partition or 'a' partition must start"
883 1.1 dyoung " at beginning of disk", s);
884 1.1 dyoung break;
885 1.1 dyoung
886 1.1 dyoung default:
887 1.1 dyoung warn("%s", s);
888 1.1 dyoung break;
889 1.1 dyoung }
890 1.1 dyoung }
891 1.1 dyoung
892 1.6 dsl #ifdef NO_MBR_SUPPORT
893 1.6 dsl #define process_mbr(f, action) 1
894 1.6 dsl #else
895 1.1 dyoung /*
896 1.6 dsl * Scan DOS/MBR partition table and extended partition list for NetBSD ptns.
897 1.1 dyoung */
898 1.6 dsl static int
899 1.6 dsl process_mbr(int f, int (*action)(int, u_int))
900 1.1 dyoung {
901 1.1 dyoung struct mbr_partition *dp;
902 1.1 dyoung struct mbr_sector mbr;
903 1.6 dsl int rval = 1, res;
904 1.1 dyoung int part;
905 1.6 dsl u_int ext_base, next_ext, this_ext, start;
906 1.1 dyoung
907 1.1 dyoung ext_base = 0;
908 1.1 dyoung next_ext = 0;
909 1.1 dyoung for (;;) {
910 1.1 dyoung this_ext = next_ext;
911 1.1 dyoung next_ext = 0;
912 1.6 dsl if (verbose > 1)
913 1.6 dsl warnx("reading mbr sector %u", this_ext);
914 1.1 dyoung if (pread(f, &mbr, sizeof mbr, this_ext * (off_t)DEV_BSIZE)
915 1.1 dyoung != sizeof(mbr)) {
916 1.6 dsl if (verbose)
917 1.24 apb warn("Can't read master boot record %u",
918 1.6 dsl this_ext);
919 1.6 dsl break;
920 1.1 dyoung }
921 1.1 dyoung
922 1.1 dyoung /* Check if table is valid. */
923 1.1 dyoung if (mbr.mbr_magic != htole16(MBR_MAGIC)) {
924 1.6 dsl if (verbose)
925 1.24 apb warnx("Invalid signature in mbr record %u",
926 1.6 dsl this_ext);
927 1.6 dsl break;
928 1.1 dyoung }
929 1.1 dyoung
930 1.1 dyoung dp = &mbr.mbr_parts[0];
931 1.1 dyoung
932 1.6 dsl /* Find NetBSD partition(s). */
933 1.1 dyoung for (part = 0; part < MBR_PART_COUNT; dp++, part++) {
934 1.6 dsl start = le32toh(dp->mbrp_start);
935 1.1 dyoung switch (dp->mbrp_type) {
936 1.6 dsl #ifdef COMPAT_386BSD_MBRPART
937 1.6 dsl case MBR_PTYPE_386BSD:
938 1.6 dsl if (ext_base != 0)
939 1.6 dsl break;
940 1.6 dsl /* FALLTHROUGH */
941 1.6 dsl #endif
942 1.1 dyoung case MBR_PTYPE_NETBSD:
943 1.6 dsl res = action(f, this_ext + start);
944 1.6 dsl if (res <= 0)
945 1.6 dsl /* Found or failure */
946 1.6 dsl return res;
947 1.6 dsl if (res > rval)
948 1.6 dsl /* Keep largest value */
949 1.6 dsl rval = res;
950 1.1 dyoung break;
951 1.1 dyoung case MBR_PTYPE_EXT:
952 1.1 dyoung case MBR_PTYPE_EXT_LBA:
953 1.1 dyoung case MBR_PTYPE_EXT_LNX:
954 1.6 dsl next_ext = start;
955 1.6 dsl break;
956 1.1 dyoung default:
957 1.6 dsl break;
958 1.1 dyoung }
959 1.1 dyoung }
960 1.1 dyoung if (next_ext == 0)
961 1.1 dyoung /* No more extended partitions */
962 1.1 dyoung break;
963 1.1 dyoung next_ext += ext_base;
964 1.1 dyoung if (ext_base == 0)
965 1.1 dyoung ext_base = next_ext;
966 1.1 dyoung
967 1.1 dyoung if (next_ext <= this_ext) {
968 1.6 dsl if (verbose)
969 1.6 dsl warnx("Invalid extended chain %x <= %x",
970 1.6 dsl next_ext, this_ext);
971 1.1 dyoung break;
972 1.1 dyoung }
973 1.6 dsl /* Maybe we should check against the disk size... */
974 1.1 dyoung }
975 1.1 dyoung
976 1.6 dsl return rval;
977 1.6 dsl }
978 1.6 dsl
979 1.6 dsl static int
980 1.6 dsl readlabel_mbr(int f, u_int sector)
981 1.6 dsl {
982 1.22 tsutsui struct disklabel *disk_lp;
983 1.6 dsl
984 1.22 tsutsui disk_lp = find_label(f, sector);
985 1.22 tsutsui if (disk_lp == NULL)
986 1.6 dsl return 1;
987 1.22 tsutsui targettohlabel(&lab, disk_lp);
988 1.6 dsl return 0;
989 1.6 dsl }
990 1.1 dyoung
991 1.6 dsl static int
992 1.6 dsl writelabel_mbr(int f, u_int sector)
993 1.6 dsl {
994 1.6 dsl return update_label(f, sector, mflag ? LABELOFFSET_MBR : ~0U) ? 2 : 0;
995 1.1 dyoung }
996 1.1 dyoung
997 1.6 dsl #endif /* !NO_MBR_SUPPORT */
998 1.6 dsl
999 1.6 dsl #ifndef USE_ACORN
1000 1.6 dsl #define get_filecore_partition(f) 0
1001 1.6 dsl #else
1002 1.1 dyoung /*
1003 1.1 dyoung * static int filecore_checksum(u_char *bootblock)
1004 1.1 dyoung *
1005 1.1 dyoung * Calculates the filecore boot block checksum. This is used to validate
1006 1.1 dyoung * a filecore boot block on the disk. If a boot block is validated then
1007 1.1 dyoung * it is used to locate the partition table. If the boot block is not
1008 1.1 dyoung * validated, it is assumed that the whole disk is NetBSD.
1009 1.1 dyoung *
1010 1.1 dyoung * The basic algorithm is:
1011 1.1 dyoung *
1012 1.1 dyoung * for (each byte in block, excluding checksum) {
1013 1.1 dyoung * sum += byte;
1014 1.1 dyoung * if (sum > 255)
1015 1.1 dyoung * sum -= 255;
1016 1.1 dyoung * }
1017 1.1 dyoung *
1018 1.1 dyoung * That's equivalent to summing all of the bytes in the block
1019 1.1 dyoung * (excluding the checksum byte, of course), then calculating the
1020 1.1 dyoung * checksum as "cksum = sum - ((sum - 1) / 255) * 255)". That
1021 1.1 dyoung * expression may or may not yield a faster checksum function,
1022 1.1 dyoung * but it's easier to reason about.
1023 1.1 dyoung *
1024 1.1 dyoung * Note that if you have a block filled with bytes of a single
1025 1.1 dyoung * value "X" (regardless of that value!) and calculate the cksum
1026 1.1 dyoung * of the block (excluding the checksum byte), you will _always_
1027 1.1 dyoung * end up with a checksum of X. (Do the math; that can be derived
1028 1.1 dyoung * from the checksum calculation function!) That means that
1029 1.1 dyoung * blocks which contain bytes which all have the same value will
1030 1.1 dyoung * always checksum properly. That's a _very_ unlikely occurence
1031 1.1 dyoung * (probably impossible, actually) for a valid filecore boot block,
1032 1.1 dyoung * so we treat such blocks as invalid.
1033 1.1 dyoung */
1034 1.1 dyoung static int
1035 1.1 dyoung filecore_checksum(u_char *bootblock)
1036 1.1 dyoung {
1037 1.1 dyoung u_char byte0, accum_diff;
1038 1.1 dyoung u_int sum;
1039 1.1 dyoung int i;
1040 1.1 dyoung
1041 1.1 dyoung sum = 0;
1042 1.1 dyoung accum_diff = 0;
1043 1.1 dyoung byte0 = bootblock[0];
1044 1.1 dyoung
1045 1.1 dyoung /*
1046 1.1 dyoung * Sum the contents of the block, keeping track of whether
1047 1.1 dyoung * or not all bytes are the same. If 'accum_diff' ends up
1048 1.1 dyoung * being zero, all of the bytes are, in fact, the same.
1049 1.1 dyoung */
1050 1.1 dyoung for (i = 0; i < 511; ++i) {
1051 1.1 dyoung sum += bootblock[i];
1052 1.1 dyoung accum_diff |= bootblock[i] ^ byte0;
1053 1.1 dyoung }
1054 1.1 dyoung
1055 1.1 dyoung /*
1056 1.1 dyoung * Check to see if the checksum byte is the same as the
1057 1.1 dyoung * rest of the bytes, too. (Note that if all of the bytes
1058 1.1 dyoung * are the same except the checksum, a checksum compare
1059 1.1 dyoung * won't succeed, but that's not our problem.)
1060 1.1 dyoung */
1061 1.1 dyoung accum_diff |= bootblock[i] ^ byte0;
1062 1.1 dyoung
1063 1.1 dyoung /* All bytes in block are the same; call it invalid. */
1064 1.1 dyoung if (accum_diff == 0)
1065 1.1 dyoung return (-1);
1066 1.1 dyoung
1067 1.1 dyoung return (sum - ((sum - 1) / 255) * 255);
1068 1.1 dyoung }
1069 1.1 dyoung
1070 1.1 dyoung /*
1071 1.6 dsl * Check for the presence of a RiscOS filecore boot block
1072 1.6 dsl * indicating an ADFS file system on the disc.
1073 1.6 dsl * Return the offset to the NetBSD part of the disc if
1074 1.6 dsl * this can be determined.
1075 1.6 dsl * This routine will terminate disklabel if the disc
1076 1.6 dsl * is found to be ADFS only.
1077 1.1 dyoung */
1078 1.1 dyoung static u_int
1079 1.1 dyoung get_filecore_partition(int f)
1080 1.1 dyoung {
1081 1.1 dyoung struct filecore_bootblock *fcbb;
1082 1.12 christos static u_char bb[DEV_BSIZE];
1083 1.1 dyoung u_int offset;
1084 1.6 dsl struct riscix_partition_table *riscix_part;
1085 1.6 dsl int loop;
1086 1.1 dyoung
1087 1.6 dsl if (pread(f, bb, sizeof(bb), (off_t)FILECORE_BOOT_SECTOR * DEV_BSIZE) != sizeof(bb))
1088 1.1 dyoung err(4, "can't read filecore boot block");
1089 1.1 dyoung fcbb = (struct filecore_bootblock *)bb;
1090 1.1 dyoung
1091 1.1 dyoung /* Check if table is valid. */
1092 1.1 dyoung if (filecore_checksum(bb) != fcbb->checksum)
1093 1.1 dyoung return (0);
1094 1.1 dyoung
1095 1.1 dyoung /*
1096 1.1 dyoung * Check for NetBSD/arm32 (RiscBSD) partition marker.
1097 1.1 dyoung * If found the NetBSD disklabel location is easy.
1098 1.1 dyoung */
1099 1.1 dyoung offset = (fcbb->partition_cyl_low + (fcbb->partition_cyl_high << 8))
1100 1.1 dyoung * fcbb->heads * fcbb->secspertrack;
1101 1.6 dsl
1102 1.6 dsl switch (fcbb->partition_type) {
1103 1.6 dsl
1104 1.6 dsl case PARTITION_FORMAT_RISCBSD:
1105 1.1 dyoung return (offset);
1106 1.1 dyoung
1107 1.6 dsl case PARTITION_FORMAT_RISCIX:
1108 1.1 dyoung /*
1109 1.1 dyoung * Read the RISCiX partition table and search for the
1110 1.1 dyoung * first partition named "RiscBSD", "NetBSD", or "Empty:"
1111 1.1 dyoung *
1112 1.1 dyoung * XXX is use of 'Empty:' really desirable?! -- cgd
1113 1.1 dyoung */
1114 1.1 dyoung
1115 1.6 dsl if (pread(f, bb, sizeof(bb), (off_t)offset * DEV_BSIZE) != sizeof(bb))
1116 1.1 dyoung err(4, "can't read riscix partition table");
1117 1.1 dyoung riscix_part = (struct riscix_partition_table *)bb;
1118 1.1 dyoung
1119 1.1 dyoung for (loop = 0; loop < NRISCIX_PARTITIONS; ++loop) {
1120 1.12 christos if (strcmp((char *)riscix_part->partitions[loop].rp_name,
1121 1.6 dsl "RiscBSD") == 0 ||
1122 1.12 christos strcmp((char *)riscix_part->partitions[loop].rp_name,
1123 1.6 dsl "NetBSD") == 0 ||
1124 1.12 christos strcmp((char *)riscix_part->partitions[loop].rp_name,
1125 1.6 dsl "Empty:") == 0) {
1126 1.6 dsl return riscix_part->partitions[loop].rp_start;
1127 1.1 dyoung break;
1128 1.1 dyoung }
1129 1.1 dyoung }
1130 1.6 dsl /*
1131 1.6 dsl * Valid filecore boot block, RISCiX partition table
1132 1.6 dsl * but no NetBSD partition. We should leave this
1133 1.6 dsl * disc alone.
1134 1.6 dsl */
1135 1.6 dsl errx(4, "cannot label: no NetBSD partition found"
1136 1.6 dsl " in RISCiX partition table");
1137 1.6 dsl
1138 1.6 dsl default:
1139 1.1 dyoung /*
1140 1.1 dyoung * Valid filecore boot block and no non-ADFS partition.
1141 1.1 dyoung * This means that the whole disc is allocated for ADFS
1142 1.1 dyoung * so do not trash ! If the user really wants to put a
1143 1.1 dyoung * NetBSD disklabel on the disc then they should remove
1144 1.1 dyoung * the filecore boot block first with dd.
1145 1.1 dyoung */
1146 1.1 dyoung errx(4, "cannot label: filecore-only disk"
1147 1.1 dyoung " (no non-ADFS partition)");
1148 1.1 dyoung }
1149 1.1 dyoung return (0);
1150 1.1 dyoung }
1151 1.1 dyoung #endif /* USE_ACORN */
1152 1.1 dyoung
1153 1.1 dyoung /*
1154 1.6 dsl * Fetch disklabel for disk to 'lab'.
1155 1.1 dyoung * Use ioctl to get label unless -r flag is given.
1156 1.1 dyoung */
1157 1.6 dsl static void
1158 1.1 dyoung readlabel(int f)
1159 1.1 dyoung {
1160 1.6 dsl if (rflag) {
1161 1.6 dsl /* Get label directly from disk */
1162 1.6 dsl if (readlabel_direct(f) == 0)
1163 1.6 dsl return;
1164 1.1 dyoung /*
1165 1.1 dyoung * There was no label on the disk. Get the fictious one
1166 1.1 dyoung * as a basis for initialisation.
1167 1.1 dyoung */
1168 1.6 dsl if (!Fflag && Iflag && (dk_ioctl(f, DIOCGDINFO, &lab) == 0 ||
1169 1.6 dsl dk_ioctl(f, DIOCGDEFLABEL, &lab) == 0))
1170 1.6 dsl return;
1171 1.1 dyoung } else {
1172 1.6 dsl /* Get label from kernel. */
1173 1.6 dsl if (dk_ioctl(f, DIOCGDINFO, &lab) < 0)
1174 1.1 dyoung err(4, "ioctl DIOCGDINFO");
1175 1.6 dsl return;
1176 1.1 dyoung }
1177 1.6 dsl
1178 1.6 dsl if (read_all == 2)
1179 1.6 dsl /* We actually found one, and printed it... */
1180 1.6 dsl exit(0);
1181 1.6 dsl errx(1, "could not read existing label");
1182 1.1 dyoung }
1183 1.1 dyoung
1184 1.1 dyoung /*
1185 1.6 dsl * Reading the label from the disk is largely a case of 'hunt the label'.
1186 1.6 dsl * and since different architectures default to different places there
1187 1.6 dsl * could even be more than one label that contradict each other!
1188 1.6 dsl * For now we look in the expected place, then search through likely
1189 1.6 dsl * other locations.
1190 1.1 dyoung */
1191 1.1 dyoung static struct disklabel *
1192 1.6 dsl find_label(int f, u_int sector)
1193 1.1 dyoung {
1194 1.22 tsutsui struct disklabel *disk_lp, hlp;
1195 1.24 apb int i;
1196 1.34 christos off_t offset;
1197 1.6 dsl const char *is_deleted;
1198 1.6 dsl
1199 1.6 dsl bootarea_len = pread(f, bootarea, sizeof bootarea,
1200 1.6 dsl sector * (off_t)DEV_BSIZE);
1201 1.6 dsl if (bootarea_len <= 0) {
1202 1.6 dsl if (verbose)
1203 1.6 dsl warn("failed to read bootarea from sector %u", sector);
1204 1.6 dsl return NULL;
1205 1.6 dsl }
1206 1.6 dsl
1207 1.6 dsl if (verbose > 2)
1208 1.24 apb warnx("read sector %u len %d looking for label",
1209 1.6 dsl sector, bootarea_len);
1210 1.6 dsl
1211 1.6 dsl /* Check expected offset first */
1212 1.6 dsl for (offset = LABEL_OFFSET, i = -4;; offset = i += 4) {
1213 1.6 dsl is_deleted = "";
1214 1.22 tsutsui disk_lp = (void *)(bootarea + offset);
1215 1.6 dsl if (i == LABEL_OFFSET)
1216 1.6 dsl continue;
1217 1.22 tsutsui if ((char *)(disk_lp + 1) > bootarea + bootarea_len)
1218 1.6 dsl break;
1219 1.22 tsutsui if (disk_lp->d_magic2 != disk_lp->d_magic)
1220 1.6 dsl continue;
1221 1.22 tsutsui if (read_all && (disk_lp->d_magic == DISKMAGIC_DELETED ||
1222 1.22 tsutsui disk_lp->d_magic == DISKMAGIC_DELETED_REV)) {
1223 1.22 tsutsui disk_lp->d_magic ^= ~0u;
1224 1.22 tsutsui disk_lp->d_magic2 ^= ~0u;
1225 1.6 dsl is_deleted = "deleted ";
1226 1.6 dsl }
1227 1.22 tsutsui if (target32toh(disk_lp->d_magic) != DISKMAGIC) {
1228 1.6 dsl /* XXX: Do something about byte-swapped labels ? */
1229 1.22 tsutsui if (target32toh(disk_lp->d_magic) == DISKMAGIC_REV &&
1230 1.22 tsutsui target32toh(disk_lp->d_magic2) == DISKMAGIC_REV)
1231 1.6 dsl warnx("ignoring %sbyteswapped label"
1232 1.35 christos " at offset %jd from sector %u",
1233 1.35 christos is_deleted, (intmax_t)offset, sector);
1234 1.6 dsl continue;
1235 1.6 dsl }
1236 1.29 matt if (target16toh(disk_lp->d_npartitions) > maxpartitions ||
1237 1.22 tsutsui dkcksum_target(disk_lp) != 0) {
1238 1.6 dsl if (verbose > 0)
1239 1.35 christos warnx("corrupt label found at offset %jd in "
1240 1.35 christos "sector %u", (intmax_t)offset, sector);
1241 1.6 dsl continue;
1242 1.6 dsl }
1243 1.6 dsl if (verbose > 1)
1244 1.35 christos warnx("%slabel found at offset %jd from sector %u",
1245 1.35 christos is_deleted, (intmax_t)offset, sector);
1246 1.6 dsl if (!read_all)
1247 1.22 tsutsui return disk_lp;
1248 1.6 dsl
1249 1.6 dsl /* To print all the labels we have to do it here */
1250 1.6 dsl /* XXX: maybe we should compare them? */
1251 1.22 tsutsui targettohlabel(&hlp, disk_lp);
1252 1.35 christos printf("# %ssector %u offset %jd bytes\n",
1253 1.35 christos is_deleted, sector, (intmax_t)offset);
1254 1.6 dsl if (tflag)
1255 1.22 tsutsui makedisktab(stdout, &hlp);
1256 1.6 dsl else {
1257 1.22 tsutsui showinfo(stdout, &hlp, specname);
1258 1.22 tsutsui showpartitions(stdout, &hlp, Cflag);
1259 1.6 dsl }
1260 1.22 tsutsui checklabel(&hlp);
1261 1.22 tsutsui htotargetlabel(disk_lp, &hlp);
1262 1.6 dsl /* Remember we've found a label */
1263 1.6 dsl read_all = 2;
1264 1.1 dyoung }
1265 1.6 dsl return NULL;
1266 1.6 dsl }
1267 1.6 dsl
1268 1.6 dsl static void
1269 1.6 dsl write_bootarea(int f, u_int sector)
1270 1.6 dsl {
1271 1.6 dsl int wlen;
1272 1.1 dyoung
1273 1.6 dsl if (bootarea_len <= 0)
1274 1.6 dsl errx(1, "attempting to write after failed read");
1275 1.6 dsl
1276 1.21 tsutsui #ifdef ALPHA_BOOTBLOCK_CKSUM
1277 1.1 dyoung /*
1278 1.6 dsl * The Alpha requires that the boot block be checksummed.
1279 1.21 tsutsui * <sys/bootblock.h> provides a macro to do it.
1280 1.1 dyoung */
1281 1.6 dsl if (sector == 0) {
1282 1.6 dsl struct alpha_boot_block *bb;
1283 1.6 dsl
1284 1.10 christos bb = (struct alpha_boot_block *)(void *)bootarea;
1285 1.6 dsl bb->bb_cksum = 0;
1286 1.6 dsl ALPHA_BOOT_BLOCK_CKSUM(bb, &bb->bb_cksum);
1287 1.6 dsl }
1288 1.21 tsutsui #endif /* ALPHA_BOOTBLOCK_CKSUM */
1289 1.6 dsl
1290 1.6 dsl wlen = pwrite(f, bootarea, bootarea_len, sector * (off_t)DEV_BSIZE);
1291 1.6 dsl if (wlen == bootarea_len)
1292 1.6 dsl return;
1293 1.6 dsl if (wlen == -1)
1294 1.24 apb err(1, "disklabel write (sector %u) size %d failed",
1295 1.6 dsl sector, bootarea_len);
1296 1.24 apb errx(1, "disklabel write (sector %u) size %d truncated to %d",
1297 1.6 dsl sector, bootarea_len, wlen);
1298 1.6 dsl }
1299 1.1 dyoung
1300 1.6 dsl static int
1301 1.6 dsl update_label(int f, u_int label_sector, u_int label_offset)
1302 1.6 dsl {
1303 1.6 dsl struct disklabel *disk_lp;
1304 1.6 dsl
1305 1.6 dsl disk_lp = find_label(f, label_sector);
1306 1.6 dsl
1307 1.6 dsl if (disk_lp && Dflag) {
1308 1.6 dsl /* Invalidate the existing label */
1309 1.6 dsl disk_lp->d_magic ^= ~0u;
1310 1.6 dsl disk_lp->d_magic2 ^= ~0u;
1311 1.6 dsl if (Dflag == 2)
1312 1.6 dsl write_bootarea(f, label_sector);
1313 1.6 dsl /* Force label to default location */
1314 1.6 dsl disk_lp = NULL;
1315 1.1 dyoung }
1316 1.1 dyoung
1317 1.6 dsl if (Dflag == 2)
1318 1.6 dsl /* We are just deleting the label */
1319 1.6 dsl return 0;
1320 1.6 dsl
1321 1.6 dsl if (disk_lp == NULL) {
1322 1.6 dsl if (label_offset == ~0u)
1323 1.6 dsl return 0;
1324 1.6 dsl /* Nothing on the disk - we need to add it */
1325 1.15 drochner disk_lp = (void *)(bootarea + label_offset);
1326 1.15 drochner if ((char *)(disk_lp + 1) > bootarea + bootarea_len)
1327 1.6 dsl errx(1, "no space in bootarea (sector %u) "
1328 1.6 dsl "to create label", label_sector);
1329 1.6 dsl }
1330 1.6 dsl
1331 1.22 tsutsui htotargetlabel(disk_lp, &lab);
1332 1.6 dsl write_bootarea(f, label_sector);
1333 1.6 dsl return 1;
1334 1.6 dsl }
1335 1.6 dsl
1336 1.6 dsl static void
1337 1.6 dsl writelabel_direct(int f)
1338 1.6 dsl {
1339 1.6 dsl u_int label_sector;
1340 1.6 dsl int written = 0;
1341 1.6 dsl int rval;
1342 1.6 dsl
1343 1.6 dsl label_sector = get_filecore_partition(f);
1344 1.6 dsl if (label_sector != 0)
1345 1.6 dsl /* The offset needs to be that from the acorn ports... */
1346 1.6 dsl written = update_label(f, label_sector, DEV_BSIZE);
1347 1.6 dsl
1348 1.6 dsl rval = process_mbr(f, writelabel_mbr);
1349 1.6 dsl
1350 1.6 dsl if (rval == 2 || written)
1351 1.6 dsl /* Don't add a label to sector 0, but update one if there */
1352 1.6 dsl update_label(f, 0, ~0u);
1353 1.6 dsl else
1354 1.6 dsl update_label(f, 0, LABEL_OFFSET);
1355 1.6 dsl }
1356 1.6 dsl
1357 1.6 dsl static int
1358 1.6 dsl readlabel_direct(int f)
1359 1.6 dsl {
1360 1.6 dsl struct disklabel *disk_lp;
1361 1.6 dsl u_int filecore_partition_offset;
1362 1.6 dsl
1363 1.6 dsl filecore_partition_offset = get_filecore_partition(f);
1364 1.6 dsl if (filecore_partition_offset != 0) {
1365 1.6 dsl disk_lp = find_label(f, filecore_partition_offset);
1366 1.6 dsl if (disk_lp != NULL) {
1367 1.22 tsutsui targettohlabel(&lab, disk_lp);
1368 1.6 dsl return 0;
1369 1.6 dsl }
1370 1.6 dsl }
1371 1.6 dsl
1372 1.6 dsl if (mflag && process_mbr(f, readlabel_mbr) == 0)
1373 1.6 dsl return 0;
1374 1.6 dsl
1375 1.6 dsl disk_lp = find_label(f, 0);
1376 1.6 dsl if (disk_lp != NULL) {
1377 1.22 tsutsui targettohlabel(&lab, disk_lp);
1378 1.6 dsl return 0;
1379 1.6 dsl }
1380 1.6 dsl
1381 1.6 dsl if (!mflag && process_mbr(f, readlabel_mbr) == 0)
1382 1.6 dsl return 0;
1383 1.6 dsl
1384 1.6 dsl return 1;
1385 1.1 dyoung }
1386 1.1 dyoung
1387 1.1 dyoung static void
1388 1.1 dyoung makedisktab(FILE *f, struct disklabel *lp)
1389 1.1 dyoung {
1390 1.1 dyoung int i;
1391 1.1 dyoung const char *did;
1392 1.1 dyoung struct partition *pp;
1393 1.1 dyoung
1394 1.1 dyoung did = "\\\n\t:";
1395 1.1 dyoung (void) fprintf(f, "%.*s|Automatically generated label:\\\n\t:dt=",
1396 1.1 dyoung (int) sizeof(lp->d_typename), lp->d_typename);
1397 1.1 dyoung if ((unsigned) lp->d_type < DKMAXTYPES)
1398 1.1 dyoung (void) fprintf(f, "%s:", dktypenames[lp->d_type]);
1399 1.1 dyoung else
1400 1.23 christos (void) fprintf(f, "unknown%" PRIu16 ":", lp->d_type);
1401 1.1 dyoung
1402 1.23 christos (void) fprintf(f, "se#%" PRIu32 ":", lp->d_secsize);
1403 1.23 christos (void) fprintf(f, "ns#%" PRIu32 ":", lp->d_nsectors);
1404 1.23 christos (void) fprintf(f, "nt#%" PRIu32 ":", lp->d_ntracks);
1405 1.23 christos (void) fprintf(f, "sc#%" PRIu32 ":", lp->d_secpercyl);
1406 1.23 christos (void) fprintf(f, "nc#%" PRIu32 ":", lp->d_ncylinders);
1407 1.1 dyoung
1408 1.1 dyoung if ((lp->d_secpercyl * lp->d_ncylinders) != lp->d_secperunit) {
1409 1.23 christos (void) fprintf(f, "%ssu#%" PRIu32 ":", did, lp->d_secperunit);
1410 1.1 dyoung did = "";
1411 1.1 dyoung }
1412 1.1 dyoung if (lp->d_rpm != 3600) {
1413 1.23 christos (void) fprintf(f, "%srm#%" PRIu16 ":", did, lp->d_rpm);
1414 1.1 dyoung did = "";
1415 1.1 dyoung }
1416 1.1 dyoung if (lp->d_interleave != 1) {
1417 1.23 christos (void) fprintf(f, "%sil#%" PRIu16 ":", did, lp->d_interleave);
1418 1.1 dyoung did = "";
1419 1.1 dyoung }
1420 1.1 dyoung if (lp->d_trackskew != 0) {
1421 1.23 christos (void) fprintf(f, "%ssk#%" PRIu16 ":", did, lp->d_trackskew);
1422 1.1 dyoung did = "";
1423 1.1 dyoung }
1424 1.1 dyoung if (lp->d_cylskew != 0) {
1425 1.23 christos (void) fprintf(f, "%scs#%" PRIu16 ":", did, lp->d_cylskew);
1426 1.1 dyoung did = "";
1427 1.1 dyoung }
1428 1.1 dyoung if (lp->d_headswitch != 0) {
1429 1.39 joerg (void) fprintf(f, "%shs#%" PRIu32 ":", did, lp->d_headswitch);
1430 1.1 dyoung did = "";
1431 1.1 dyoung }
1432 1.1 dyoung if (lp->d_trkseek != 0) {
1433 1.23 christos (void) fprintf(f, "%sts#%" PRIu32 ":", did, lp->d_trkseek);
1434 1.1 dyoung did = "";
1435 1.1 dyoung }
1436 1.1 dyoung #ifdef notyet
1437 1.1 dyoung (void) fprintf(f, "drivedata: ");
1438 1.1 dyoung for (i = NDDATA - 1; i >= 0; i--)
1439 1.1 dyoung if (lp->d_drivedata[i])
1440 1.1 dyoung break;
1441 1.1 dyoung if (i < 0)
1442 1.1 dyoung i = 0;
1443 1.1 dyoung for (j = 0; j <= i; j++)
1444 1.23 christos (void) fprintf(f, "%" PRIu32 " ", lp->d_drivedata[j]);
1445 1.1 dyoung #endif /* notyet */
1446 1.1 dyoung pp = lp->d_partitions;
1447 1.1 dyoung for (i = 0; i < lp->d_npartitions; i++, pp++) {
1448 1.1 dyoung if (pp->p_size) {
1449 1.1 dyoung char c = 'a' + i;
1450 1.1 dyoung (void) fprintf(f, "\\\n\t:");
1451 1.23 christos (void) fprintf(f, "p%c#%" PRIu32 ":", c, pp->p_size);
1452 1.23 christos (void) fprintf(f, "o%c#%" PRIu32 ":", c, pp->p_offset);
1453 1.1 dyoung if (pp->p_fstype != FS_UNUSED) {
1454 1.1 dyoung if ((unsigned) pp->p_fstype < FSMAXTYPES)
1455 1.1 dyoung (void) fprintf(f, "t%c=%s:", c,
1456 1.1 dyoung fstypenames[pp->p_fstype]);
1457 1.1 dyoung else
1458 1.24 apb (void) fprintf(f,
1459 1.24 apb "t%c=unknown%" PRIu8 ":",
1460 1.24 apb c, pp->p_fstype);
1461 1.1 dyoung }
1462 1.1 dyoung switch (pp->p_fstype) {
1463 1.1 dyoung
1464 1.1 dyoung case FS_UNUSED:
1465 1.1 dyoung break;
1466 1.1 dyoung
1467 1.1 dyoung case FS_BSDFFS:
1468 1.1 dyoung case FS_BSDLFS:
1469 1.1 dyoung case FS_EX2FS:
1470 1.1 dyoung case FS_ADOS:
1471 1.1 dyoung case FS_APPLEUFS:
1472 1.23 christos (void) fprintf(f, "b%c#%" PRIu64 ":", c,
1473 1.23 christos (uint64_t)pp->p_fsize * pp->p_frag);
1474 1.23 christos (void) fprintf(f, "f%c#%" PRIu32 ":", c,
1475 1.23 christos pp->p_fsize);
1476 1.1 dyoung break;
1477 1.1 dyoung default:
1478 1.1 dyoung break;
1479 1.1 dyoung }
1480 1.1 dyoung }
1481 1.1 dyoung }
1482 1.1 dyoung (void) fprintf(f, "\n");
1483 1.1 dyoung (void) fflush(f);
1484 1.1 dyoung }
1485 1.1 dyoung
1486 1.1 dyoung static int
1487 1.6 dsl edit(int f)
1488 1.1 dyoung {
1489 1.1 dyoung const char *tmpdir;
1490 1.6 dsl char tmpfil[MAXPATHLEN];
1491 1.1 dyoung int first, ch, fd;
1492 1.9 dsl int get_ok;
1493 1.1 dyoung FILE *fp;
1494 1.1 dyoung
1495 1.1 dyoung if ((tmpdir = getenv("TMPDIR")) == NULL)
1496 1.1 dyoung tmpdir = _PATH_TMP;
1497 1.1 dyoung (void)snprintf(tmpfil, sizeof(tmpfil), "%s/%s", tmpdir, TMPFILE);
1498 1.1 dyoung if ((fd = mkstemp(tmpfil)) == -1 || (fp = fdopen(fd, "w")) == NULL) {
1499 1.1 dyoung warn("%s", tmpfil);
1500 1.1 dyoung return (1);
1501 1.1 dyoung }
1502 1.1 dyoung (void)fchmod(fd, 0600);
1503 1.6 dsl showinfo(fp, &lab, specname);
1504 1.6 dsl showpartitions(fp, &lab, Cflag);
1505 1.1 dyoung (void) fclose(fp);
1506 1.1 dyoung for (;;) {
1507 1.6 dsl if (!editit(tmpfil))
1508 1.1 dyoung break;
1509 1.1 dyoung fp = fopen(tmpfil, "r");
1510 1.1 dyoung if (fp == NULL) {
1511 1.1 dyoung warn("%s", tmpfil);
1512 1.1 dyoung break;
1513 1.1 dyoung }
1514 1.6 dsl (void) memset(&lab, 0, sizeof(lab));
1515 1.9 dsl get_ok = getasciilabel(fp, &lab);
1516 1.9 dsl fclose(fp);
1517 1.9 dsl if (get_ok && write_label(f) == 0) {
1518 1.9 dsl (void) unlink(tmpfil);
1519 1.9 dsl return (0);
1520 1.1 dyoung }
1521 1.1 dyoung (void) printf("re-edit the label? [y]: ");
1522 1.1 dyoung (void) fflush(stdout);
1523 1.1 dyoung first = ch = getchar();
1524 1.1 dyoung while (ch != '\n' && ch != EOF)
1525 1.1 dyoung ch = getchar();
1526 1.1 dyoung if (first == 'n' || first == 'N')
1527 1.1 dyoung break;
1528 1.1 dyoung }
1529 1.1 dyoung (void)unlink(tmpfil);
1530 1.1 dyoung return (1);
1531 1.1 dyoung }
1532 1.1 dyoung
1533 1.1 dyoung static int
1534 1.6 dsl editit(const char *tmpfil)
1535 1.1 dyoung {
1536 1.1 dyoung int pid, xpid;
1537 1.1 dyoung int status;
1538 1.1 dyoung sigset_t nsigset, osigset;
1539 1.1 dyoung
1540 1.1 dyoung sigemptyset(&nsigset);
1541 1.1 dyoung sigaddset(&nsigset, SIGINT);
1542 1.1 dyoung sigaddset(&nsigset, SIGQUIT);
1543 1.1 dyoung sigaddset(&nsigset, SIGHUP);
1544 1.1 dyoung sigprocmask(SIG_BLOCK, &nsigset, &osigset);
1545 1.1 dyoung while ((pid = fork()) < 0) {
1546 1.1 dyoung if (errno != EAGAIN) {
1547 1.1 dyoung sigprocmask(SIG_SETMASK, &osigset, (sigset_t *)0);
1548 1.1 dyoung warn("fork");
1549 1.1 dyoung return (0);
1550 1.1 dyoung }
1551 1.1 dyoung sleep(1);
1552 1.1 dyoung }
1553 1.1 dyoung if (pid == 0) {
1554 1.1 dyoung const char *ed;
1555 1.1 dyoung char *buf;
1556 1.1 dyoung int retval;
1557 1.1 dyoung
1558 1.1 dyoung sigprocmask(SIG_SETMASK, &osigset, (sigset_t *)0);
1559 1.1 dyoung setgid(getgid());
1560 1.1 dyoung setuid(getuid());
1561 1.1 dyoung if ((ed = getenv("EDITOR")) == (char *)0)
1562 1.1 dyoung ed = DEFEDITOR;
1563 1.1 dyoung /*
1564 1.1 dyoung * Jump through a few extra hoops in case someone's editor
1565 1.1 dyoung * is "editor arg1 arg2".
1566 1.1 dyoung */
1567 1.1 dyoung asprintf(&buf, "%s %s", ed, tmpfil);
1568 1.1 dyoung if (!buf)
1569 1.1 dyoung err(1, "malloc");
1570 1.1 dyoung retval = execlp(_PATH_BSHELL, _PATH_BSHELL, "-c", buf, NULL);
1571 1.1 dyoung if (retval == -1)
1572 1.1 dyoung perror(ed);
1573 1.1 dyoung exit(retval);
1574 1.1 dyoung }
1575 1.1 dyoung while ((xpid = wait(&status)) >= 0)
1576 1.1 dyoung if (xpid == pid)
1577 1.1 dyoung break;
1578 1.1 dyoung sigprocmask(SIG_SETMASK, &osigset, (sigset_t *)0);
1579 1.1 dyoung return (!status);
1580 1.1 dyoung }
1581 1.1 dyoung
1582 1.1 dyoung static char *
1583 1.1 dyoung skip(char *cp)
1584 1.1 dyoung {
1585 1.1 dyoung
1586 1.1 dyoung cp += strspn(cp, " \t");
1587 1.1 dyoung if (*cp == '\0')
1588 1.1 dyoung return (NULL);
1589 1.1 dyoung return (cp);
1590 1.1 dyoung }
1591 1.1 dyoung
1592 1.1 dyoung static char *
1593 1.1 dyoung word(char *cp)
1594 1.1 dyoung {
1595 1.1 dyoung
1596 1.1 dyoung if (cp == NULL || *cp == '\0')
1597 1.1 dyoung return (NULL);
1598 1.1 dyoung
1599 1.1 dyoung cp += strcspn(cp, " \t");
1600 1.1 dyoung if (*cp == '\0')
1601 1.1 dyoung return (NULL);
1602 1.1 dyoung *cp++ = '\0';
1603 1.1 dyoung cp += strspn(cp, " \t");
1604 1.1 dyoung if (*cp == '\0')
1605 1.1 dyoung return (NULL);
1606 1.1 dyoung return (cp);
1607 1.1 dyoung }
1608 1.1 dyoung
1609 1.1 dyoung #define _CHECKLINE \
1610 1.1 dyoung if (tp == NULL || *tp == '\0') { \
1611 1.1 dyoung warnx("line %d: too few fields", lineno); \
1612 1.1 dyoung errors++; \
1613 1.1 dyoung break; \
1614 1.1 dyoung }
1615 1.1 dyoung
1616 1.1 dyoung #define __CHECKLINE \
1617 1.1 dyoung if (*tp == NULL || **tp == '\0') { \
1618 1.1 dyoung warnx("line %d: too few fields", lineno); \
1619 1.1 dyoung *tp = _error_; \
1620 1.1 dyoung return 0; \
1621 1.1 dyoung }
1622 1.1 dyoung
1623 1.1 dyoung static char _error_[] = "";
1624 1.1 dyoung #define NXTNUM(n) if ((n = nxtnum(&tp, lineno),0) + tp != _error_) \
1625 1.1 dyoung ; else goto error
1626 1.1 dyoung #define NXTXNUM(n) if ((n = nxtxnum(&tp, lp, lineno),0) + tp != _error_) \
1627 1.1 dyoung ; else goto error
1628 1.1 dyoung
1629 1.1 dyoung static unsigned long
1630 1.1 dyoung nxtnum(char **tp, int lineno)
1631 1.1 dyoung {
1632 1.1 dyoung char *cp;
1633 1.1 dyoung unsigned long v;
1634 1.1 dyoung
1635 1.1 dyoung __CHECKLINE
1636 1.1 dyoung if (getulong(*tp, '\0', &cp, &v, UINT32_MAX) != 0) {
1637 1.1 dyoung warnx("line %d: syntax error", lineno);
1638 1.1 dyoung *tp = _error_;
1639 1.1 dyoung return 0;
1640 1.1 dyoung }
1641 1.1 dyoung *tp = cp;
1642 1.1 dyoung return v;
1643 1.1 dyoung }
1644 1.1 dyoung
1645 1.1 dyoung static unsigned long
1646 1.1 dyoung nxtxnum(char **tp, struct disklabel *lp, int lineno)
1647 1.1 dyoung {
1648 1.1 dyoung char *cp, *ncp;
1649 1.1 dyoung unsigned long n, v;
1650 1.1 dyoung
1651 1.1 dyoung __CHECKLINE
1652 1.1 dyoung cp = *tp;
1653 1.1 dyoung if (getulong(cp, '/', &ncp, &n, UINT32_MAX) != 0)
1654 1.1 dyoung goto bad;
1655 1.1 dyoung
1656 1.1 dyoung if (*ncp == '/') {
1657 1.1 dyoung n *= lp->d_secpercyl;
1658 1.1 dyoung cp = ncp + 1;
1659 1.1 dyoung if (getulong(cp, '/', &ncp, &v, UINT32_MAX) != 0)
1660 1.1 dyoung goto bad;
1661 1.1 dyoung n += v * lp->d_nsectors;
1662 1.1 dyoung cp = ncp + 1;
1663 1.1 dyoung if (getulong(cp, '\0', &ncp, &v, UINT32_MAX) != 0)
1664 1.1 dyoung goto bad;
1665 1.1 dyoung n += v;
1666 1.1 dyoung }
1667 1.1 dyoung *tp = ncp;
1668 1.1 dyoung return n;
1669 1.1 dyoung bad:
1670 1.1 dyoung warnx("line %d: invalid format", lineno);
1671 1.1 dyoung *tp = _error_;
1672 1.1 dyoung return 0;
1673 1.1 dyoung }
1674 1.1 dyoung
1675 1.1 dyoung /*
1676 1.1 dyoung * Read an ascii label in from fd f,
1677 1.1 dyoung * in the same format as that put out by showinfo() and showpartitions(),
1678 1.1 dyoung * and fill in lp.
1679 1.1 dyoung */
1680 1.1 dyoung static int
1681 1.1 dyoung getasciilabel(FILE *f, struct disklabel *lp)
1682 1.1 dyoung {
1683 1.1 dyoung const char *const *cpp, *s;
1684 1.1 dyoung struct partition *pp;
1685 1.1 dyoung char *cp, *tp, line[BUFSIZ], tbuf[15];
1686 1.1 dyoung int lineno, errors;
1687 1.1 dyoung unsigned long v;
1688 1.1 dyoung unsigned int part;
1689 1.1 dyoung
1690 1.1 dyoung lineno = 0;
1691 1.1 dyoung errors = 0;
1692 1.1 dyoung lp->d_bbsize = BBSIZE; /* XXX */
1693 1.1 dyoung lp->d_sbsize = SBLOCKSIZE; /* XXX */
1694 1.1 dyoung while (fgets(line, sizeof(line) - 1, f)) {
1695 1.1 dyoung lineno++;
1696 1.1 dyoung if ((cp = strpbrk(line, "#\r\n")) != NULL)
1697 1.1 dyoung *cp = '\0';
1698 1.1 dyoung cp = skip(line);
1699 1.1 dyoung if (cp == NULL) /* blank line or comment line */
1700 1.1 dyoung continue;
1701 1.1 dyoung tp = strchr(cp, ':'); /* everything has a colon in it */
1702 1.1 dyoung if (tp == NULL) {
1703 1.1 dyoung warnx("line %d: syntax error", lineno);
1704 1.1 dyoung errors++;
1705 1.1 dyoung continue;
1706 1.1 dyoung }
1707 1.1 dyoung *tp++ = '\0', tp = skip(tp);
1708 1.1 dyoung if (!strcmp(cp, "type")) {
1709 1.1 dyoung if (tp == NULL) {
1710 1.1 dyoung strlcpy(tbuf, "unknown", sizeof(tbuf));
1711 1.1 dyoung tp = tbuf;
1712 1.1 dyoung }
1713 1.1 dyoung cpp = dktypenames;
1714 1.1 dyoung for (; cpp < &dktypenames[DKMAXTYPES]; cpp++)
1715 1.1 dyoung if ((s = *cpp) && !strcasecmp(s, tp)) {
1716 1.1 dyoung lp->d_type = cpp - dktypenames;
1717 1.1 dyoung goto next;
1718 1.1 dyoung }
1719 1.1 dyoung if (GETNUM16(tp, &v) != 0) {
1720 1.1 dyoung warnx("line %d: syntax error", lineno);
1721 1.1 dyoung errors++;
1722 1.1 dyoung continue;
1723 1.1 dyoung }
1724 1.1 dyoung if (v >= DKMAXTYPES)
1725 1.1 dyoung warnx("line %d: warning, unknown disk type: %s",
1726 1.1 dyoung lineno, tp);
1727 1.1 dyoung lp->d_type = v;
1728 1.1 dyoung continue;
1729 1.1 dyoung }
1730 1.1 dyoung if (!strcmp(cp, "flags")) {
1731 1.1 dyoung for (v = 0; (cp = tp) && *cp != '\0';) {
1732 1.1 dyoung tp = word(cp);
1733 1.1 dyoung if (!strcasecmp(cp, "removable"))
1734 1.1 dyoung v |= D_REMOVABLE;
1735 1.1 dyoung else if (!strcasecmp(cp, "ecc"))
1736 1.1 dyoung v |= D_ECC;
1737 1.1 dyoung else if (!strcasecmp(cp, "badsect"))
1738 1.1 dyoung v |= D_BADSECT;
1739 1.1 dyoung else {
1740 1.1 dyoung warnx("line %d: bad flag: %s",
1741 1.1 dyoung lineno, cp);
1742 1.1 dyoung errors++;
1743 1.1 dyoung }
1744 1.1 dyoung }
1745 1.1 dyoung lp->d_flags = v;
1746 1.1 dyoung continue;
1747 1.1 dyoung }
1748 1.1 dyoung if (!strcmp(cp, "drivedata")) {
1749 1.1 dyoung int i;
1750 1.1 dyoung
1751 1.1 dyoung for (i = 0; (cp = tp) && *cp != '\0' && i < NDDATA;) {
1752 1.1 dyoung if (GETNUM32(cp, &v) != 0) {
1753 1.1 dyoung warnx("line %d: bad drive data",
1754 1.1 dyoung lineno);
1755 1.1 dyoung errors++;
1756 1.1 dyoung } else
1757 1.1 dyoung lp->d_drivedata[i] = v;
1758 1.1 dyoung i++;
1759 1.1 dyoung tp = word(cp);
1760 1.1 dyoung }
1761 1.1 dyoung continue;
1762 1.1 dyoung }
1763 1.1 dyoung if (sscanf(cp, "%lu partitions", &v) == 1) {
1764 1.29 matt if (v == 0 || v > maxpartitions) {
1765 1.1 dyoung warnx("line %d: bad # of partitions", lineno);
1766 1.29 matt lp->d_npartitions = maxpartitions;
1767 1.1 dyoung errors++;
1768 1.1 dyoung } else
1769 1.1 dyoung lp->d_npartitions = v;
1770 1.1 dyoung continue;
1771 1.1 dyoung }
1772 1.1 dyoung if (tp == NULL) {
1773 1.1 dyoung tbuf[0] = '\0';
1774 1.1 dyoung tp = tbuf;
1775 1.1 dyoung }
1776 1.1 dyoung if (!strcmp(cp, "disk")) {
1777 1.1 dyoung strncpy(lp->d_typename, tp, sizeof(lp->d_typename));
1778 1.1 dyoung continue;
1779 1.1 dyoung }
1780 1.1 dyoung if (!strcmp(cp, "label")) {
1781 1.1 dyoung strncpy(lp->d_packname, tp, sizeof(lp->d_packname));
1782 1.1 dyoung continue;
1783 1.1 dyoung }
1784 1.1 dyoung if (!strcmp(cp, "bytes/sector")) {
1785 1.1 dyoung if (GETNUM32(tp, &v) != 0 || v <= 0 || (v % 512) != 0) {
1786 1.1 dyoung warnx("line %d: bad %s: %s", lineno, cp, tp);
1787 1.1 dyoung errors++;
1788 1.1 dyoung } else
1789 1.1 dyoung lp->d_secsize = v;
1790 1.1 dyoung continue;
1791 1.1 dyoung }
1792 1.1 dyoung if (!strcmp(cp, "sectors/track")) {
1793 1.1 dyoung if (GETNUM32(tp, &v) != 0) {
1794 1.1 dyoung warnx("line %d: bad %s: %s", lineno, cp, tp);
1795 1.1 dyoung errors++;
1796 1.1 dyoung } else
1797 1.1 dyoung lp->d_nsectors = v;
1798 1.1 dyoung continue;
1799 1.1 dyoung }
1800 1.1 dyoung if (!strcmp(cp, "sectors/cylinder")) {
1801 1.1 dyoung if (GETNUM32(tp, &v) != 0) {
1802 1.1 dyoung warnx("line %d: bad %s: %s", lineno, cp, tp);
1803 1.1 dyoung errors++;
1804 1.1 dyoung } else
1805 1.1 dyoung lp->d_secpercyl = v;
1806 1.1 dyoung continue;
1807 1.1 dyoung }
1808 1.1 dyoung if (!strcmp(cp, "tracks/cylinder")) {
1809 1.1 dyoung if (GETNUM32(tp, &v) != 0) {
1810 1.1 dyoung warnx("line %d: bad %s: %s", lineno, cp, tp);
1811 1.1 dyoung errors++;
1812 1.1 dyoung } else
1813 1.1 dyoung lp->d_ntracks = v;
1814 1.1 dyoung continue;
1815 1.1 dyoung }
1816 1.1 dyoung if (!strcmp(cp, "cylinders")) {
1817 1.1 dyoung if (GETNUM32(tp, &v) != 0) {
1818 1.1 dyoung warnx("line %d: bad %s: %s", lineno, cp, tp);
1819 1.1 dyoung errors++;
1820 1.1 dyoung } else
1821 1.1 dyoung lp->d_ncylinders = v;
1822 1.1 dyoung continue;
1823 1.1 dyoung }
1824 1.11 christos if (!strcmp(cp, "total sectors") ||
1825 1.11 christos !strcmp(cp, "sectors/unit")) {
1826 1.1 dyoung if (GETNUM32(tp, &v) != 0) {
1827 1.1 dyoung warnx("line %d: bad %s: %s", lineno, cp, tp);
1828 1.1 dyoung errors++;
1829 1.1 dyoung } else
1830 1.1 dyoung lp->d_secperunit = v;
1831 1.1 dyoung continue;
1832 1.1 dyoung }
1833 1.1 dyoung if (!strcmp(cp, "rpm")) {
1834 1.1 dyoung if (GETNUM16(tp, &v) != 0) {
1835 1.1 dyoung warnx("line %d: bad %s: %s", lineno, cp, tp);
1836 1.1 dyoung errors++;
1837 1.1 dyoung } else
1838 1.1 dyoung lp->d_rpm = v;
1839 1.1 dyoung continue;
1840 1.1 dyoung }
1841 1.1 dyoung if (!strcmp(cp, "interleave")) {
1842 1.1 dyoung if (GETNUM16(tp, &v) != 0) {
1843 1.1 dyoung warnx("line %d: bad %s: %s", lineno, cp, tp);
1844 1.1 dyoung errors++;
1845 1.1 dyoung } else
1846 1.1 dyoung lp->d_interleave = v;
1847 1.1 dyoung continue;
1848 1.1 dyoung }
1849 1.1 dyoung if (!strcmp(cp, "trackskew")) {
1850 1.1 dyoung if (GETNUM16(tp, &v) != 0) {
1851 1.1 dyoung warnx("line %d: bad %s: %s", lineno, cp, tp);
1852 1.1 dyoung errors++;
1853 1.1 dyoung } else
1854 1.1 dyoung lp->d_trackskew = v;
1855 1.1 dyoung continue;
1856 1.1 dyoung }
1857 1.1 dyoung if (!strcmp(cp, "cylinderskew")) {
1858 1.1 dyoung if (GETNUM16(tp, &v) != 0) {
1859 1.1 dyoung warnx("line %d: bad %s: %s", lineno, cp, tp);
1860 1.1 dyoung errors++;
1861 1.1 dyoung } else
1862 1.1 dyoung lp->d_cylskew = v;
1863 1.1 dyoung continue;
1864 1.1 dyoung }
1865 1.1 dyoung if (!strcmp(cp, "headswitch")) {
1866 1.1 dyoung if (GETNUM32(tp, &v) != 0) {
1867 1.1 dyoung warnx("line %d: bad %s: %s", lineno, cp, tp);
1868 1.1 dyoung errors++;
1869 1.1 dyoung } else
1870 1.1 dyoung lp->d_headswitch = v;
1871 1.1 dyoung continue;
1872 1.1 dyoung }
1873 1.1 dyoung if (!strcmp(cp, "track-to-track seek")) {
1874 1.1 dyoung if (GETNUM32(tp, &v) != 0) {
1875 1.1 dyoung warnx("line %d: bad %s: %s", lineno, cp, tp);
1876 1.1 dyoung errors++;
1877 1.1 dyoung } else
1878 1.1 dyoung lp->d_trkseek = v;
1879 1.1 dyoung continue;
1880 1.1 dyoung }
1881 1.1 dyoung if ('a' > *cp || *cp > 'z' || cp[1] != '\0') {
1882 1.1 dyoung warnx("line %d: unknown field: %s", lineno, cp);
1883 1.1 dyoung errors++;
1884 1.1 dyoung continue;
1885 1.1 dyoung }
1886 1.1 dyoung
1887 1.1 dyoung /* We have a partition entry */
1888 1.1 dyoung part = *cp - 'a';
1889 1.1 dyoung
1890 1.29 matt if (part >= maxpartitions) {
1891 1.1 dyoung warnx("line %d: bad partition name: %s", lineno, cp);
1892 1.1 dyoung errors++;
1893 1.1 dyoung continue;
1894 1.1 dyoung }
1895 1.33 christos if (part >= __arraycount(lp->d_partitions)) {
1896 1.33 christos warnx("line %d: partition id %s, >= %zu", lineno,
1897 1.33 christos cp, __arraycount(lp->d_partitions));
1898 1.33 christos errors++;
1899 1.33 christos continue;
1900 1.33 christos }
1901 1.1 dyoung pp = &lp->d_partitions[part];
1902 1.1 dyoung
1903 1.1 dyoung NXTXNUM(pp->p_size);
1904 1.1 dyoung NXTXNUM(pp->p_offset);
1905 1.1 dyoung /* can't use word() here because of blanks in fstypenames[] */
1906 1.1 dyoung tp += strspn(tp, " \t");
1907 1.1 dyoung _CHECKLINE
1908 1.1 dyoung cp = tp;
1909 1.1 dyoung cpp = fstypenames;
1910 1.1 dyoung for (; cpp < &fstypenames[FSMAXTYPES]; cpp++) {
1911 1.1 dyoung s = *cpp;
1912 1.1 dyoung if (s == NULL ||
1913 1.1 dyoung (cp[strlen(s)] != ' ' &&
1914 1.1 dyoung cp[strlen(s)] != '\t' &&
1915 1.1 dyoung cp[strlen(s)] != '\0'))
1916 1.1 dyoung continue;
1917 1.1 dyoung if (!memcmp(s, cp, strlen(s))) {
1918 1.1 dyoung pp->p_fstype = cpp - fstypenames;
1919 1.1 dyoung tp += strlen(s);
1920 1.1 dyoung if (*tp == '\0')
1921 1.1 dyoung tp = NULL;
1922 1.1 dyoung else {
1923 1.1 dyoung tp += strspn(tp, " \t");
1924 1.1 dyoung if (*tp == '\0')
1925 1.1 dyoung tp = NULL;
1926 1.1 dyoung }
1927 1.1 dyoung goto gottype;
1928 1.1 dyoung }
1929 1.1 dyoung }
1930 1.1 dyoung tp = word(cp);
1931 1.1 dyoung if (isdigit(*cp & 0xff)) {
1932 1.1 dyoung if (GETNUM8(cp, &v) != 0) {
1933 1.1 dyoung warnx("line %d: syntax error", lineno);
1934 1.1 dyoung errors++;
1935 1.1 dyoung }
1936 1.1 dyoung } else
1937 1.1 dyoung v = FSMAXTYPES;
1938 1.1 dyoung if ((unsigned)v >= FSMAXTYPES) {
1939 1.13 jmmv warnx("line %d: warning, unknown file system type: %s",
1940 1.1 dyoung lineno, cp);
1941 1.13 jmmv warnx("tip: use -l to see all valid file system "
1942 1.13 jmmv "types");
1943 1.1 dyoung v = FS_UNUSED;
1944 1.1 dyoung }
1945 1.1 dyoung pp->p_fstype = v;
1946 1.1 dyoung gottype:
1947 1.1 dyoung switch (pp->p_fstype) {
1948 1.1 dyoung
1949 1.1 dyoung case FS_UNUSED: /* XXX */
1950 1.1 dyoung NXTNUM(pp->p_fsize);
1951 1.1 dyoung if (pp->p_fsize == 0)
1952 1.1 dyoung break;
1953 1.1 dyoung NXTNUM(v);
1954 1.1 dyoung pp->p_frag = v / pp->p_fsize;
1955 1.1 dyoung break;
1956 1.1 dyoung
1957 1.1 dyoung case FS_BSDFFS:
1958 1.1 dyoung case FS_ADOS:
1959 1.1 dyoung case FS_APPLEUFS:
1960 1.1 dyoung NXTNUM(pp->p_fsize);
1961 1.1 dyoung if (pp->p_fsize == 0)
1962 1.1 dyoung break;
1963 1.1 dyoung NXTNUM(v);
1964 1.1 dyoung pp->p_frag = v / pp->p_fsize;
1965 1.1 dyoung NXTNUM(pp->p_cpg);
1966 1.1 dyoung break;
1967 1.1 dyoung case FS_BSDLFS:
1968 1.1 dyoung NXTNUM(pp->p_fsize);
1969 1.1 dyoung if (pp->p_fsize == 0)
1970 1.1 dyoung break;
1971 1.1 dyoung NXTNUM(v);
1972 1.1 dyoung pp->p_frag = v / pp->p_fsize;
1973 1.1 dyoung NXTNUM(pp->p_sgs);
1974 1.1 dyoung break;
1975 1.1 dyoung case FS_EX2FS:
1976 1.1 dyoung NXTNUM(pp->p_fsize);
1977 1.1 dyoung if (pp->p_fsize == 0)
1978 1.1 dyoung break;
1979 1.1 dyoung NXTNUM(v);
1980 1.1 dyoung pp->p_frag = v / pp->p_fsize;
1981 1.1 dyoung break;
1982 1.1 dyoung case FS_ISO9660:
1983 1.1 dyoung NXTNUM(pp->p_cdsession);
1984 1.1 dyoung break;
1985 1.1 dyoung default:
1986 1.1 dyoung break;
1987 1.1 dyoung }
1988 1.1 dyoung continue;
1989 1.1 dyoung error:
1990 1.1 dyoung errors++;
1991 1.1 dyoung next:
1992 1.1 dyoung ;
1993 1.1 dyoung }
1994 1.1 dyoung errors += checklabel(lp);
1995 1.1 dyoung return (errors == 0);
1996 1.1 dyoung }
1997 1.1 dyoung
1998 1.1 dyoung /*
1999 1.1 dyoung * Check disklabel for errors and fill in
2000 1.1 dyoung * derived fields according to supplied values.
2001 1.1 dyoung */
2002 1.1 dyoung int
2003 1.1 dyoung checklabel(struct disklabel *lp)
2004 1.1 dyoung {
2005 1.1 dyoung struct partition *pp, *qp;
2006 1.1 dyoung int i, j, errors;
2007 1.1 dyoung char part;
2008 1.1 dyoung
2009 1.1 dyoung errors = 0;
2010 1.1 dyoung if (lp->d_secsize == 0) {
2011 1.23 christos warnx("sector size %" PRIu32, lp->d_secsize);
2012 1.1 dyoung return (1);
2013 1.1 dyoung }
2014 1.1 dyoung if (lp->d_nsectors == 0) {
2015 1.23 christos warnx("sectors/track %" PRIu32, lp->d_nsectors);
2016 1.1 dyoung return (1);
2017 1.1 dyoung }
2018 1.1 dyoung if (lp->d_ntracks == 0) {
2019 1.23 christos warnx("tracks/cylinder %" PRIu32, lp->d_ntracks);
2020 1.1 dyoung return (1);
2021 1.1 dyoung }
2022 1.1 dyoung if (lp->d_ncylinders == 0) {
2023 1.23 christos warnx("cylinders/unit %" PRIu32, lp->d_ncylinders);
2024 1.1 dyoung errors++;
2025 1.1 dyoung }
2026 1.1 dyoung if (lp->d_rpm == 0)
2027 1.23 christos warnx("warning, revolutions/minute %" PRIu16, lp->d_rpm);
2028 1.1 dyoung if (lp->d_secpercyl == 0)
2029 1.1 dyoung lp->d_secpercyl = lp->d_nsectors * lp->d_ntracks;
2030 1.1 dyoung if (lp->d_secperunit == 0)
2031 1.1 dyoung lp->d_secperunit = lp->d_secpercyl * lp->d_ncylinders;
2032 1.1 dyoung if (lp->d_bbsize == 0) {
2033 1.23 christos warnx("boot block size %" PRIu32, lp->d_bbsize);
2034 1.1 dyoung errors++;
2035 1.1 dyoung } else if (lp->d_bbsize % lp->d_secsize)
2036 1.1 dyoung warnx("warning, boot block size %% sector-size != 0");
2037 1.1 dyoung if (lp->d_sbsize == 0) {
2038 1.23 christos warnx("super block size %" PRIu32, lp->d_sbsize);
2039 1.1 dyoung errors++;
2040 1.1 dyoung } else if (lp->d_sbsize % lp->d_secsize)
2041 1.1 dyoung warnx("warning, super block size %% sector-size != 0");
2042 1.29 matt if (lp->d_npartitions > maxpartitions)
2043 1.24 apb warnx("warning, number of partitions (%" PRIu16 ") > "
2044 1.24 apb "MAXPARTITIONS (%d)",
2045 1.29 matt lp->d_npartitions, maxpartitions);
2046 1.1 dyoung else
2047 1.29 matt for (i = maxpartitions - 1; i >= lp->d_npartitions; i--) {
2048 1.1 dyoung part = 'a' + i;
2049 1.1 dyoung pp = &lp->d_partitions[i];
2050 1.1 dyoung if (pp->p_size || pp->p_offset) {
2051 1.1 dyoung warnx("warning, partition %c increased "
2052 1.24 apb "number of partitions from %" PRIu16
2053 1.24 apb " to %d",
2054 1.1 dyoung part, lp->d_npartitions, i + 1);
2055 1.1 dyoung lp->d_npartitions = i + 1;
2056 1.1 dyoung break;
2057 1.1 dyoung }
2058 1.1 dyoung }
2059 1.1 dyoung for (i = 0; i < lp->d_npartitions; i++) {
2060 1.1 dyoung part = 'a' + i;
2061 1.1 dyoung pp = &lp->d_partitions[i];
2062 1.1 dyoung if (pp->p_size == 0 && pp->p_offset != 0)
2063 1.24 apb warnx("warning, partition %c: size 0, but "
2064 1.24 apb "offset %" PRIu32,
2065 1.1 dyoung part, pp->p_offset);
2066 1.1 dyoung #ifdef STRICT_CYLINDER_ALIGNMENT
2067 1.1 dyoung if (pp->p_offset % lp->d_secpercyl) {
2068 1.1 dyoung warnx("warning, partition %c:"
2069 1.1 dyoung " not starting on cylinder boundary",
2070 1.1 dyoung part);
2071 1.1 dyoung errors++;
2072 1.1 dyoung }
2073 1.1 dyoung #endif /* STRICT_CYLINDER_ALIGNMENT */
2074 1.1 dyoung if (pp->p_offset > lp->d_secperunit) {
2075 1.1 dyoung warnx("partition %c: offset past end of unit", part);
2076 1.1 dyoung errors++;
2077 1.1 dyoung }
2078 1.1 dyoung if (pp->p_offset + pp->p_size > lp->d_secperunit) {
2079 1.1 dyoung warnx("partition %c: partition extends"
2080 1.1 dyoung " past end of unit",
2081 1.1 dyoung part);
2082 1.1 dyoung errors++;
2083 1.1 dyoung }
2084 1.1 dyoung if (pp->p_fstype != FS_UNUSED)
2085 1.1 dyoung for (j = i + 1; j < lp->d_npartitions; j++) {
2086 1.1 dyoung qp = &lp->d_partitions[j];
2087 1.1 dyoung if (qp->p_fstype == FS_UNUSED)
2088 1.1 dyoung continue;
2089 1.1 dyoung if (pp->p_offset < qp->p_offset + qp->p_size &&
2090 1.1 dyoung qp->p_offset < pp->p_offset + pp->p_size)
2091 1.1 dyoung warnx("partitions %c and %c overlap",
2092 1.1 dyoung part, 'a' + j);
2093 1.1 dyoung }
2094 1.1 dyoung }
2095 1.1 dyoung return (errors);
2096 1.1 dyoung }
2097 1.1 dyoung
2098 1.1 dyoung static void
2099 1.1 dyoung usage(void)
2100 1.1 dyoung {
2101 1.1 dyoung static const struct {
2102 1.1 dyoung const char *name;
2103 1.1 dyoung const char *expn;
2104 1.1 dyoung } usages[] = {
2105 1.29 matt { "[-ABCFMrtv] disk", "(to read label)" },
2106 1.29 matt { "-w [-BDFMrv] [-f disktab] disk disktype [packid]", "(to write label)" },
2107 1.29 matt { "-e [-BCDFMIrv] disk", "(to edit label)" },
2108 1.46 christos #if !defined(NO_INTERACT)
2109 1.29 matt { "-i [-BDFMIrv] disk", "(to create a label interactively)" },
2110 1.46 christos #endif
2111 1.6 dsl { "-D [-v] disk", "(to delete existing label(s))" },
2112 1.29 matt { "-R [-BDFMrv] disk protofile", "(to restore label)" },
2113 1.4 dsl { "[-NW] disk", "(to write disable/enable label)" },
2114 1.13 jmmv { "-l", "(to show all known file system types)" },
2115 1.4 dsl { NULL, NULL }
2116 1.1 dyoung };
2117 1.1 dyoung int i;
2118 1.4 dsl const char *pn = getprogname();
2119 1.4 dsl const char *t = "usage:";
2120 1.1 dyoung
2121 1.4 dsl for (i = 0; usages[i].name != NULL; i++) {
2122 1.4 dsl (void)fprintf(stderr, "%s %s %s\n\t%s\n",
2123 1.4 dsl t, pn, usages[i].name, usages[i].expn);
2124 1.4 dsl t = "or";
2125 1.1 dyoung }
2126 1.1 dyoung exit(1);
2127 1.1 dyoung }
2128 1.1 dyoung
2129 1.1 dyoung static int
2130 1.1 dyoung getulong(const char *str, char sep, char **epp, unsigned long *ul,
2131 1.1 dyoung unsigned long max)
2132 1.1 dyoung {
2133 1.1 dyoung char *ep;
2134 1.1 dyoung
2135 1.1 dyoung if (epp == NULL)
2136 1.1 dyoung epp = &ep;
2137 1.1 dyoung
2138 1.1 dyoung *ul = strtoul(str, epp, 10);
2139 1.1 dyoung
2140 1.1 dyoung if ((*ul == ULONG_MAX && errno == ERANGE) || *ul > max)
2141 1.1 dyoung return ERANGE;
2142 1.1 dyoung
2143 1.1 dyoung if (*str == '\0' || (**epp != '\0' && **epp != sep &&
2144 1.1 dyoung !isspace((unsigned char)**epp)))
2145 1.1 dyoung return EFTYPE;
2146 1.1 dyoung
2147 1.1 dyoung return 0;
2148 1.1 dyoung }
2149 1.13 jmmv
2150 1.13 jmmv /*
2151 1.13 jmmv * This is a wrapper over the standard strcmp function to be used with
2152 1.13 jmmv * qsort on an array of pointers to strings.
2153 1.13 jmmv */
2154 1.13 jmmv static int
2155 1.13 jmmv qsort_strcmp(const void *v1, const void *v2)
2156 1.13 jmmv {
2157 1.13 jmmv const char *const *sp1 = (const char *const *)v1;
2158 1.13 jmmv const char *const *sp2 = (const char *const *)v2;
2159 1.13 jmmv
2160 1.13 jmmv return strcmp(*sp1, *sp2);
2161 1.13 jmmv }
2162 1.13 jmmv
2163 1.13 jmmv /*
2164 1.13 jmmv * Prints all know file system types for a partition.
2165 1.13 jmmv * Returns 1 on success, 0 on failure.
2166 1.13 jmmv */
2167 1.13 jmmv int
2168 1.13 jmmv list_fs_types(void)
2169 1.13 jmmv {
2170 1.13 jmmv int ret;
2171 1.13 jmmv size_t nelems;
2172 1.13 jmmv
2173 1.13 jmmv nelems = 0;
2174 1.13 jmmv {
2175 1.13 jmmv const char *const *namep;
2176 1.13 jmmv
2177 1.13 jmmv namep = fstypenames;
2178 1.13 jmmv while (*namep++ != NULL)
2179 1.13 jmmv nelems++;
2180 1.13 jmmv }
2181 1.13 jmmv
2182 1.13 jmmv ret = 1;
2183 1.13 jmmv if (nelems > 0) {
2184 1.13 jmmv const char **list;
2185 1.13 jmmv size_t i;
2186 1.13 jmmv
2187 1.13 jmmv list = (const char **)malloc(sizeof(char *) * nelems);
2188 1.13 jmmv if (list == NULL) {
2189 1.13 jmmv warnx("sorry, could not allocate memory for list");
2190 1.13 jmmv ret = 0;
2191 1.13 jmmv } else {
2192 1.13 jmmv for (i = 0; i < nelems; i++)
2193 1.13 jmmv list[i] = fstypenames[i];
2194 1.13 jmmv
2195 1.13 jmmv qsort(list, nelems, sizeof(char *), qsort_strcmp);
2196 1.13 jmmv
2197 1.13 jmmv for (i = 0; i < nelems; i++)
2198 1.13 jmmv (void)printf("%s\n", list[i]);
2199 1.13 jmmv
2200 1.13 jmmv free(list);
2201 1.13 jmmv }
2202 1.13 jmmv }
2203 1.13 jmmv
2204 1.13 jmmv return ret;
2205 1.13 jmmv }
2206 1.29 matt
2207 1.29 matt #ifndef HAVE_NBTOOL_CONFIG_H
2208 1.29 matt int
2209 1.29 matt dk_ioctl(int f, u_long cmd, void *arg)
2210 1.29 matt {
2211 1.46 christos #if !defined(NATIVELABEL_ONLY)
2212 1.29 matt if (!native_p) {
2213 1.29 matt errno = ENOTTY;
2214 1.29 matt return -1;
2215 1.29 matt }
2216 1.46 christos #endif
2217 1.29 matt return ioctl(f, cmd, arg);
2218 1.29 matt }
2219 1.29 matt #endif
2220