1/*
2 * Copyright (c) 1987, 1993
3 * The Regents of the University of California. All rights reserved.
4 *
5 * This code is derived from software contributed to Berkeley by
6 * Symmetric Computer Systems.
7 *
8 * Redistribution and use in source and binary forms, with or without
9 * modification, are permitted provided that the following conditions
10 * are met:
11 * 1. Redistributions of source code must retain the above copyright
12 * notice, this list of conditions and the following disclaimer.
13 * 2. Redistributions in binary form must reproduce the above copyright
14 * notice, this list of conditions and the following disclaimer in the
15 * documentation and/or other materials provided with the distribution.
16 * 3. All advertising materials mentioning features or use of this software
17 * must display the following acknowledgement:
18 * This product includes software developed by the University of
19 * California, Berkeley and its contributors.
20 * 4. Neither the name of the University nor the names of its contributors
21 * may be used to endorse or promote products derived from this software
22 * without specific prior written permission.
23 *
24 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
25 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
26 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
27 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
28 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
29 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
30 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
31 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
32 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
33 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
34 * SUCH DAMAGE.
35 */
36
37#ifndef lint
38static const char copyright[] =
39"@(#) Copyright (c) 1987, 1993\n\
40 The Regents of the University of California. All rights reserved.\n";
41#endif /* not lint */
42
43#ifndef lint
44#if 0
45static char sccsid[] = "@(#)disklabel.c 8.2 (Berkeley) 1/7/94";
46/* from static char sccsid[] = "@(#)disklabel.c 1.2 (Symmetric) 11/28/85"; */
47#endif
48static const char rcsid[] =
| 1/*
2 * Copyright (c) 1987, 1993
3 * The Regents of the University of California. All rights reserved.
4 *
5 * This code is derived from software contributed to Berkeley by
6 * Symmetric Computer Systems.
7 *
8 * Redistribution and use in source and binary forms, with or without
9 * modification, are permitted provided that the following conditions
10 * are met:
11 * 1. Redistributions of source code must retain the above copyright
12 * notice, this list of conditions and the following disclaimer.
13 * 2. Redistributions in binary form must reproduce the above copyright
14 * notice, this list of conditions and the following disclaimer in the
15 * documentation and/or other materials provided with the distribution.
16 * 3. All advertising materials mentioning features or use of this software
17 * must display the following acknowledgement:
18 * This product includes software developed by the University of
19 * California, Berkeley and its contributors.
20 * 4. Neither the name of the University nor the names of its contributors
21 * may be used to endorse or promote products derived from this software
22 * without specific prior written permission.
23 *
24 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
25 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
26 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
27 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
28 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
29 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
30 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
31 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
32 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
33 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
34 * SUCH DAMAGE.
35 */
36
37#ifndef lint
38static const char copyright[] =
39"@(#) Copyright (c) 1987, 1993\n\
40 The Regents of the University of California. All rights reserved.\n";
41#endif /* not lint */
42
43#ifndef lint
44#if 0
45static char sccsid[] = "@(#)disklabel.c 8.2 (Berkeley) 1/7/94";
46/* from static char sccsid[] = "@(#)disklabel.c 1.2 (Symmetric) 11/28/85"; */
47#endif
48static const char rcsid[] =
|
49 "$FreeBSD: head/sbin/bsdlabel/bsdlabel.c 68044 2000-10-31 07:07:39Z jkh $";
| 49 "$FreeBSD: head/sbin/bsdlabel/bsdlabel.c 73034 2001-02-25 16:47:44Z jwd $";
|
50#endif /* not lint */
51
52#include <sys/param.h>
53#include <sys/file.h>
54#include <sys/stat.h>
55#include <sys/wait.h>
56#define DKTYPENAMES
57#include <sys/disklabel.h>
58#include <ufs/ffs/fs.h>
59#include <unistd.h>
60#include <string.h>
61#include <stdio.h>
62#include <stdlib.h>
63#include <signal.h>
64#include <stdarg.h>
65#include <ctype.h>
66#include <err.h>
67#include <errno.h>
68#include "pathnames.h"
69
70/*
71 * Disklabel: read and write disklabels.
72 * The label is usually placed on one of the first sectors of the disk.
73 * Many machines also place a bootstrap in the same area,
74 * in which case the label is embedded in the bootstrap.
75 * The bootstrap source must leave space at the proper offset
76 * for the label on such machines.
77 */
78
79#ifndef BBSIZE
80#define BBSIZE 8192 /* size of boot area, with label */
81#endif
82
| 50#endif /* not lint */
51
52#include <sys/param.h>
53#include <sys/file.h>
54#include <sys/stat.h>
55#include <sys/wait.h>
56#define DKTYPENAMES
57#include <sys/disklabel.h>
58#include <ufs/ffs/fs.h>
59#include <unistd.h>
60#include <string.h>
61#include <stdio.h>
62#include <stdlib.h>
63#include <signal.h>
64#include <stdarg.h>
65#include <ctype.h>
66#include <err.h>
67#include <errno.h>
68#include "pathnames.h"
69
70/*
71 * Disklabel: read and write disklabels.
72 * The label is usually placed on one of the first sectors of the disk.
73 * Many machines also place a bootstrap in the same area,
74 * in which case the label is embedded in the bootstrap.
75 * The bootstrap source must leave space at the proper offset
76 * for the label on such machines.
77 */
78
79#ifndef BBSIZE
80#define BBSIZE 8192 /* size of boot area, with label */
81#endif
82
|
| 83/* FIX! These are too low, but are traditional */
84#define DEFAULT_NEWFS_BLOCK 8192U
85#define DEFAULT_NEWFS_FRAG 1024U
86#define DEFAULT_NEWFS_CPG 16U
87
88#define BIG_NEWFS_BLOCK 16384U
89#define BIG_NEWFS_FRAG 4096U
90#define BIG_NEWFS_CPG 64U
91
|
83#ifdef tahoe
84#define NUMBOOT 0
85#else
86#if defined(__alpha__) || defined(hp300) || defined(hp800)
87#define NUMBOOT 1
88#else
89#define NUMBOOT 2
90#endif
91#endif
92
93void makelabel __P((char *, char *, struct disklabel *));
94int writelabel __P((int, char *, struct disklabel *));
95void l_perror __P((char *));
96struct disklabel * readlabel __P((int));
97struct disklabel * makebootarea __P((char *, struct disklabel *, int));
98void display __P((FILE *, struct disklabel *));
99int edit __P((struct disklabel *, int));
100int editit __P((void));
101char * skip __P((char *));
102char * word __P((char *));
103int getasciilabel __P((FILE *, struct disklabel *));
104int checklabel __P((struct disklabel *));
105void setbootflag __P((struct disklabel *));
106void Warning (char *, ...);
107void usage __P((void));
108struct disklabel * getvirginlabel __P((void));
109
110#define DEFEDITOR _PATH_VI
111#define streq(a,b) (strcmp(a,b) == 0)
112
113char *dkname;
114char *specname;
115char tmpfil[] = PATH_TMPFILE;
116
117char namebuf[BBSIZE], *np = namebuf;
118struct disklabel lab;
119char bootarea[BBSIZE];
120
| 92#ifdef tahoe
93#define NUMBOOT 0
94#else
95#if defined(__alpha__) || defined(hp300) || defined(hp800)
96#define NUMBOOT 1
97#else
98#define NUMBOOT 2
99#endif
100#endif
101
102void makelabel __P((char *, char *, struct disklabel *));
103int writelabel __P((int, char *, struct disklabel *));
104void l_perror __P((char *));
105struct disklabel * readlabel __P((int));
106struct disklabel * makebootarea __P((char *, struct disklabel *, int));
107void display __P((FILE *, struct disklabel *));
108int edit __P((struct disklabel *, int));
109int editit __P((void));
110char * skip __P((char *));
111char * word __P((char *));
112int getasciilabel __P((FILE *, struct disklabel *));
113int checklabel __P((struct disklabel *));
114void setbootflag __P((struct disklabel *));
115void Warning (char *, ...);
116void usage __P((void));
117struct disklabel * getvirginlabel __P((void));
118
119#define DEFEDITOR _PATH_VI
120#define streq(a,b) (strcmp(a,b) == 0)
121
122char *dkname;
123char *specname;
124char tmpfil[] = PATH_TMPFILE;
125
126char namebuf[BBSIZE], *np = namebuf;
127struct disklabel lab;
128char bootarea[BBSIZE];
129
|
| 130/* partition 'c' is the full disk and is special */
131#define FULL_DISK_PART 2
132#define MAX_PART ('z')
133#define MAX_NUM_PARTS (1 + MAX_PART - 'a')
134char part_size_type[MAX_NUM_PARTS];
135char part_offset_type[MAX_NUM_PARTS];
136int part_set[MAX_NUM_PARTS];
137
|
121#if NUMBOOT > 0
122int installboot; /* non-zero if we should install a boot program */
123char *bootbuf; /* pointer to buffer with remainder of boot prog */
124int bootsize; /* size of remaining boot program */
125char *xxboot; /* primary boot */
126char *bootxx; /* secondary boot */
127char boot0[MAXPATHLEN];
128char boot1[MAXPATHLEN];
129#endif
130
131enum {
132 UNSPEC, EDIT, NOWRITE, READ, RESTORE, WRITE, WRITEABLE, WRITEBOOT
133} op = UNSPEC;
134
135int rflag;
| 138#if NUMBOOT > 0
139int installboot; /* non-zero if we should install a boot program */
140char *bootbuf; /* pointer to buffer with remainder of boot prog */
141int bootsize; /* size of remaining boot program */
142char *xxboot; /* primary boot */
143char *bootxx; /* secondary boot */
144char boot0[MAXPATHLEN];
145char boot1[MAXPATHLEN];
146#endif
147
148enum {
149 UNSPEC, EDIT, NOWRITE, READ, RESTORE, WRITE, WRITEABLE, WRITEBOOT
150} op = UNSPEC;
151
152int rflag;
|
| 153int disable_write; /* set to disable writing to disk label */
|
136
137#ifdef DEBUG
138int debug;
| 154
155#ifdef DEBUG
156int debug;
|
139#define OPTIONS "BNRWb:ders:w"
| 157#define OPTIONS "BNRWb:denrs:w"
|
140#else
| 158#else
|
141#define OPTIONS "BNRWb:ers:w"
| 159#define OPTIONS "BNRWb:enrs:w"
|
142#endif
143
144int
145main(argc, argv)
146 int argc;
147 char *argv[];
148{
149 register struct disklabel *lp;
150 FILE *t;
151 int ch, f = 0, flag, error = 0;
152 char *name = 0;
153
154 while ((ch = getopt(argc, argv, OPTIONS)) != -1)
155 switch (ch) {
156#if NUMBOOT > 0
157 case 'B':
158 ++installboot;
159 break;
160 case 'b':
161 xxboot = optarg;
162 break;
163#if NUMBOOT > 1
164 case 's':
165 bootxx = optarg;
166 break;
167#endif
168#endif
169 case 'N':
170 if (op != UNSPEC)
171 usage();
172 op = NOWRITE;
173 break;
| 160#endif
161
162int
163main(argc, argv)
164 int argc;
165 char *argv[];
166{
167 register struct disklabel *lp;
168 FILE *t;
169 int ch, f = 0, flag, error = 0;
170 char *name = 0;
171
172 while ((ch = getopt(argc, argv, OPTIONS)) != -1)
173 switch (ch) {
174#if NUMBOOT > 0
175 case 'B':
176 ++installboot;
177 break;
178 case 'b':
179 xxboot = optarg;
180 break;
181#if NUMBOOT > 1
182 case 's':
183 bootxx = optarg;
184 break;
185#endif
186#endif
187 case 'N':
188 if (op != UNSPEC)
189 usage();
190 op = NOWRITE;
191 break;
|
| 192 case 'n':
193 disable_write = 1;
194 break;
|
174 case 'R':
175 if (op != UNSPEC)
176 usage();
177 op = RESTORE;
178 break;
179 case 'W':
180 if (op != UNSPEC)
181 usage();
182 op = WRITEABLE;
183 break;
184 case 'e':
185 if (op != UNSPEC)
186 usage();
187 op = EDIT;
188 break;
189 case 'r':
190 ++rflag;
191 break;
192 case 'w':
193 if (op != UNSPEC)
194 usage();
195 op = WRITE;
196 break;
197#ifdef DEBUG
198 case 'd':
199 debug++;
200 break;
201#endif
202 case '?':
203 default:
204 usage();
205 }
206 argc -= optind;
207 argv += optind;
208#if NUMBOOT > 0
209 if (installboot) {
210 rflag++;
211 if (op == UNSPEC)
212 op = WRITEBOOT;
213 } else {
214 if (op == UNSPEC)
215 op = READ;
216 xxboot = bootxx = 0;
217 }
218#else
219 if (op == UNSPEC)
220 op = READ;
221#endif
222 if (argc < 1)
223 usage();
224
225 dkname = argv[0];
226 if (dkname[0] != '/') {
227 (void)sprintf(np, "%s%s%c", _PATH_DEV, dkname, 'a' + RAW_PART);
228 specname = np;
229 np += strlen(specname) + 1;
230 } else
231 specname = dkname;
232 f = open(specname, op == READ ? O_RDONLY : O_RDWR);
233 if (f < 0 && errno == ENOENT && dkname[0] != '/') {
234 (void)sprintf(specname, "%s%s", _PATH_DEV, dkname);
235 np = namebuf + strlen(specname) + 1;
236 f = open(specname, op == READ ? O_RDONLY : O_RDWR);
237 }
238 if (f < 0)
239 err(4, "%s", specname);
240
241 switch(op) {
242
243 case UNSPEC:
244 break;
245
246 case EDIT:
247 if (argc != 1)
248 usage();
249 lp = readlabel(f);
250 error = edit(lp, f);
251 break;
252
253 case NOWRITE:
254 flag = 0;
255 if (ioctl(f, DIOCWLABEL, (char *)&flag) < 0)
256 err(4, "ioctl DIOCWLABEL");
257 break;
258
259 case READ:
260 if (argc != 1)
261 usage();
262 lp = readlabel(f);
263 display(stdout, lp);
264 error = checklabel(lp);
265 break;
266
267 case RESTORE:
268#if NUMBOOT > 0
269 if (installboot && argc == 3) {
270 makelabel(argv[2], 0, &lab);
271 argc--;
272
273 /*
274 * We only called makelabel() for its side effect
275 * of setting the bootstrap file names. Discard
276 * all changes to `lab' so that all values in the
277 * final label come from the ASCII label.
278 */
279 bzero((char *)&lab, sizeof(lab));
280 }
281#endif
282 if (argc != 2)
283 usage();
284 if (!(t = fopen(argv[1], "r")))
285 err(4, "%s", argv[1]);
286 if (!getasciilabel(t, &lab))
287 exit(1);
288 lp = makebootarea(bootarea, &lab, f);
289 *lp = lab;
290 error = writelabel(f, bootarea, lp);
291 break;
292
293 case WRITE:
294 if (argc == 3) {
295 name = argv[2];
296 argc--;
297 }
298 if (argc != 2)
299 usage();
300 makelabel(argv[1], name, &lab);
301 lp = makebootarea(bootarea, &lab, f);
302 *lp = lab;
303 if (checklabel(lp) == 0)
304 error = writelabel(f, bootarea, lp);
305 break;
306
307 case WRITEABLE:
308 flag = 1;
309 if (ioctl(f, DIOCWLABEL, (char *)&flag) < 0)
310 err(4, "ioctl DIOCWLABEL");
311 break;
312
313#if NUMBOOT > 0
314 case WRITEBOOT:
315 {
316 struct disklabel tlab;
317
318 lp = readlabel(f);
319 tlab = *lp;
320 if (argc == 2)
321 makelabel(argv[1], 0, &lab);
322 lp = makebootarea(bootarea, &lab, f);
323 *lp = tlab;
324 if (checklabel(lp) == 0)
325 error = writelabel(f, bootarea, lp);
326 break;
327 }
328#endif
329 }
330 exit(error);
331}
332
333/*
334 * Construct a prototype disklabel from /etc/disktab. As a side
335 * effect, set the names of the primary and secondary boot files
336 * if specified.
337 */
338void
339makelabel(type, name, lp)
340 char *type, *name;
341 register struct disklabel *lp;
342{
343 register struct disklabel *dp;
344
345 if (strcmp(type, "auto") == 0)
346 dp = getvirginlabel();
347 else
348 dp = getdiskbyname(type);
349 if (dp == NULL)
350 errx(1, "%s: unknown disk type", type);
351 *lp = *dp;
352#if NUMBOOT > 0
353 /*
354 * Set bootstrap name(s).
355 * 1. If set from command line, use those,
356 * 2. otherwise, check if disktab specifies them (b0 or b1),
357 * 3. otherwise, makebootarea() will choose ones based on the name
358 * of the disk special file. E.g. /dev/ra0 -> raboot, bootra
359 */
360 if (!xxboot && lp->d_boot0) {
361 if (*lp->d_boot0 != '/')
362 (void)sprintf(boot0, "%s/%s",
363 _PATH_BOOTDIR, lp->d_boot0);
364 else
365 (void)strcpy(boot0, lp->d_boot0);
366 xxboot = boot0;
367 }
368#if NUMBOOT > 1
369 if (!bootxx && lp->d_boot1) {
370 if (*lp->d_boot1 != '/')
371 (void)sprintf(boot1, "%s/%s",
372 _PATH_BOOTDIR, lp->d_boot1);
373 else
374 (void)strcpy(boot1, lp->d_boot1);
375 bootxx = boot1;
376 }
377#endif
378#endif
379 /* d_packname is union d_boot[01], so zero */
380 bzero(lp->d_packname, sizeof(lp->d_packname));
381 if (name)
382 (void)strncpy(lp->d_packname, name, sizeof(lp->d_packname));
383}
384
385int
386writelabel(f, boot, lp)
387 int f;
388 char *boot;
389 register struct disklabel *lp;
390{
391 int flag;
392#ifdef __alpha__
393 u_long *p, sum;
394 int i;
395#endif
396#ifdef vax
397 register int i;
398#endif
399
| 195 case 'R':
196 if (op != UNSPEC)
197 usage();
198 op = RESTORE;
199 break;
200 case 'W':
201 if (op != UNSPEC)
202 usage();
203 op = WRITEABLE;
204 break;
205 case 'e':
206 if (op != UNSPEC)
207 usage();
208 op = EDIT;
209 break;
210 case 'r':
211 ++rflag;
212 break;
213 case 'w':
214 if (op != UNSPEC)
215 usage();
216 op = WRITE;
217 break;
218#ifdef DEBUG
219 case 'd':
220 debug++;
221 break;
222#endif
223 case '?':
224 default:
225 usage();
226 }
227 argc -= optind;
228 argv += optind;
229#if NUMBOOT > 0
230 if (installboot) {
231 rflag++;
232 if (op == UNSPEC)
233 op = WRITEBOOT;
234 } else {
235 if (op == UNSPEC)
236 op = READ;
237 xxboot = bootxx = 0;
238 }
239#else
240 if (op == UNSPEC)
241 op = READ;
242#endif
243 if (argc < 1)
244 usage();
245
246 dkname = argv[0];
247 if (dkname[0] != '/') {
248 (void)sprintf(np, "%s%s%c", _PATH_DEV, dkname, 'a' + RAW_PART);
249 specname = np;
250 np += strlen(specname) + 1;
251 } else
252 specname = dkname;
253 f = open(specname, op == READ ? O_RDONLY : O_RDWR);
254 if (f < 0 && errno == ENOENT && dkname[0] != '/') {
255 (void)sprintf(specname, "%s%s", _PATH_DEV, dkname);
256 np = namebuf + strlen(specname) + 1;
257 f = open(specname, op == READ ? O_RDONLY : O_RDWR);
258 }
259 if (f < 0)
260 err(4, "%s", specname);
261
262 switch(op) {
263
264 case UNSPEC:
265 break;
266
267 case EDIT:
268 if (argc != 1)
269 usage();
270 lp = readlabel(f);
271 error = edit(lp, f);
272 break;
273
274 case NOWRITE:
275 flag = 0;
276 if (ioctl(f, DIOCWLABEL, (char *)&flag) < 0)
277 err(4, "ioctl DIOCWLABEL");
278 break;
279
280 case READ:
281 if (argc != 1)
282 usage();
283 lp = readlabel(f);
284 display(stdout, lp);
285 error = checklabel(lp);
286 break;
287
288 case RESTORE:
289#if NUMBOOT > 0
290 if (installboot && argc == 3) {
291 makelabel(argv[2], 0, &lab);
292 argc--;
293
294 /*
295 * We only called makelabel() for its side effect
296 * of setting the bootstrap file names. Discard
297 * all changes to `lab' so that all values in the
298 * final label come from the ASCII label.
299 */
300 bzero((char *)&lab, sizeof(lab));
301 }
302#endif
303 if (argc != 2)
304 usage();
305 if (!(t = fopen(argv[1], "r")))
306 err(4, "%s", argv[1]);
307 if (!getasciilabel(t, &lab))
308 exit(1);
309 lp = makebootarea(bootarea, &lab, f);
310 *lp = lab;
311 error = writelabel(f, bootarea, lp);
312 break;
313
314 case WRITE:
315 if (argc == 3) {
316 name = argv[2];
317 argc--;
318 }
319 if (argc != 2)
320 usage();
321 makelabel(argv[1], name, &lab);
322 lp = makebootarea(bootarea, &lab, f);
323 *lp = lab;
324 if (checklabel(lp) == 0)
325 error = writelabel(f, bootarea, lp);
326 break;
327
328 case WRITEABLE:
329 flag = 1;
330 if (ioctl(f, DIOCWLABEL, (char *)&flag) < 0)
331 err(4, "ioctl DIOCWLABEL");
332 break;
333
334#if NUMBOOT > 0
335 case WRITEBOOT:
336 {
337 struct disklabel tlab;
338
339 lp = readlabel(f);
340 tlab = *lp;
341 if (argc == 2)
342 makelabel(argv[1], 0, &lab);
343 lp = makebootarea(bootarea, &lab, f);
344 *lp = tlab;
345 if (checklabel(lp) == 0)
346 error = writelabel(f, bootarea, lp);
347 break;
348 }
349#endif
350 }
351 exit(error);
352}
353
354/*
355 * Construct a prototype disklabel from /etc/disktab. As a side
356 * effect, set the names of the primary and secondary boot files
357 * if specified.
358 */
359void
360makelabel(type, name, lp)
361 char *type, *name;
362 register struct disklabel *lp;
363{
364 register struct disklabel *dp;
365
366 if (strcmp(type, "auto") == 0)
367 dp = getvirginlabel();
368 else
369 dp = getdiskbyname(type);
370 if (dp == NULL)
371 errx(1, "%s: unknown disk type", type);
372 *lp = *dp;
373#if NUMBOOT > 0
374 /*
375 * Set bootstrap name(s).
376 * 1. If set from command line, use those,
377 * 2. otherwise, check if disktab specifies them (b0 or b1),
378 * 3. otherwise, makebootarea() will choose ones based on the name
379 * of the disk special file. E.g. /dev/ra0 -> raboot, bootra
380 */
381 if (!xxboot && lp->d_boot0) {
382 if (*lp->d_boot0 != '/')
383 (void)sprintf(boot0, "%s/%s",
384 _PATH_BOOTDIR, lp->d_boot0);
385 else
386 (void)strcpy(boot0, lp->d_boot0);
387 xxboot = boot0;
388 }
389#if NUMBOOT > 1
390 if (!bootxx && lp->d_boot1) {
391 if (*lp->d_boot1 != '/')
392 (void)sprintf(boot1, "%s/%s",
393 _PATH_BOOTDIR, lp->d_boot1);
394 else
395 (void)strcpy(boot1, lp->d_boot1);
396 bootxx = boot1;
397 }
398#endif
399#endif
400 /* d_packname is union d_boot[01], so zero */
401 bzero(lp->d_packname, sizeof(lp->d_packname));
402 if (name)
403 (void)strncpy(lp->d_packname, name, sizeof(lp->d_packname));
404}
405
406int
407writelabel(f, boot, lp)
408 int f;
409 char *boot;
410 register struct disklabel *lp;
411{
412 int flag;
413#ifdef __alpha__
414 u_long *p, sum;
415 int i;
416#endif
417#ifdef vax
418 register int i;
419#endif
420
|
400 setbootflag(lp);
401 lp->d_magic = DISKMAGIC;
402 lp->d_magic2 = DISKMAGIC;
403 lp->d_checksum = 0;
404 lp->d_checksum = dkcksum(lp);
405 if (rflag) {
406 /*
407 * First set the kernel disk label,
408 * then write a label to the raw disk.
409 * If the SDINFO ioctl fails because it is unimplemented,
410 * keep going; otherwise, the kernel consistency checks
411 * may prevent us from changing the current (in-core)
412 * label.
413 */
414 if (ioctl(f, DIOCSDINFO, lp) < 0 &&
415 errno != ENODEV && errno != ENOTTY) {
416 l_perror("ioctl DIOCSDINFO");
417 return (1);
418 }
419 (void)lseek(f, (off_t)0, SEEK_SET);
420
| 421 if (disable_write) {
422 Warning("write to disk label supressed - label was as follows:");
423 display(stdout, lp);
424 return (0);
425 } else {
426 setbootflag(lp);
427 lp->d_magic = DISKMAGIC;
428 lp->d_magic2 = DISKMAGIC;
429 lp->d_checksum = 0;
430 lp->d_checksum = dkcksum(lp);
431 if (rflag) {
432 /*
433 * First set the kernel disk label,
434 * then write a label to the raw disk.
435 * If the SDINFO ioctl fails because it is unimplemented,
436 * keep going; otherwise, the kernel consistency checks
437 * may prevent us from changing the current (in-core)
438 * label.
439 */
440 if (ioctl(f, DIOCSDINFO, lp) < 0 &&
441 errno != ENODEV && errno != ENOTTY) {
442 l_perror("ioctl DIOCSDINFO");
443 return (1);
444 }
445 (void)lseek(f, (off_t)0, SEEK_SET);
446
|
421#ifdef __alpha__
| 447#ifdef __alpha__
|
422 /*
423 * Generate the bootblock checksum for the SRM console.
424 */
425 for (p = (u_long *)boot, i = 0, sum = 0; i < 63; i++)
426 sum += p[i];
427 p[63] = sum;
| 448 /*
449 * Generate the bootblock checksum for the SRM console.
450 */
451 for (p = (u_long *)boot, i = 0, sum = 0; i < 63; i++)
452 sum += p[i];
453 p[63] = sum;
|
428#endif
| 454#endif
|
429
430 /*
431 * write enable label sector before write (if necessary),
432 * disable after writing.
433 */
434 flag = 1;
435 if (ioctl(f, DIOCWLABEL, &flag) < 0)
436 warn("ioctl DIOCWLABEL");
437 if (write(f, boot, lp->d_bbsize) != lp->d_bbsize) {
438 warn("write");
439 return (1);
440 }
| 455
456 /*
457 * write enable label sector before write (if necessary),
458 * disable after writing.
459 */
460 flag = 1;
461 if (ioctl(f, DIOCWLABEL, &flag) < 0)
462 warn("ioctl DIOCWLABEL");
463 if (write(f, boot, lp->d_bbsize) != lp->d_bbsize) {
464 warn("write");
465 return (1);
466 }
|
441#if NUMBOOT > 0
| 467#if NUMBOOT > 0
|
442 /*
443 * Output the remainder of the disklabel
444 */
445 if (bootbuf && write(f, bootbuf, bootsize) != bootsize) {
446 warn("write");
447 return(1);
448 }
| 468 /*
469 * Output the remainder of the disklabel
470 */
471 if (bootbuf && write(f, bootbuf, bootsize) != bootsize) {
472 warn("write");
473 return(1);
474 }
|
449#endif
| 475#endif
|
450 flag = 0;
451 (void) ioctl(f, DIOCWLABEL, &flag);
452 } else if (ioctl(f, DIOCWDINFO, lp) < 0) {
453 l_perror("ioctl DIOCWDINFO");
454 return (1);
455 }
| 476 flag = 0;
477 (void) ioctl(f, DIOCWLABEL, &flag);
478 } else if (ioctl(f, DIOCWDINFO, lp) < 0) {
479 l_perror("ioctl DIOCWDINFO");
480 return (1);
481 }
|
456#ifdef vax
| 482#ifdef vax
|
457 if (lp->d_type == DTYPE_SMD && lp->d_flags & D_BADSECT) {
458 daddr_t alt;
459
460 alt = lp->d_ncylinders * lp->d_secpercyl - lp->d_nsectors;
461 for (i = 1; i < 11 && i < lp->d_nsectors; i += 2) {
462 (void)lseek(f, (off_t)((alt + i) * lp->d_secsize),
463 SEEK_SET);
464 if (write(f, boot, lp->d_secsize) < lp->d_secsize)
465 warn("alternate label %d write", i/2);
| 483 if (lp->d_type == DTYPE_SMD && lp->d_flags & D_BADSECT) {
484 daddr_t alt;
485
486 alt = lp->d_ncylinders * lp->d_secpercyl - lp->d_nsectors;
487 for (i = 1; i < 11 && i < lp->d_nsectors; i += 2) {
488 (void)lseek(f, (off_t)((alt + i) * lp->d_secsize),
489 SEEK_SET);
490 if (write(f, boot, lp->d_secsize) < lp->d_secsize)
491 warn("alternate label %d write", i/2);
492 }
|
466 }
| 493 }
|
467 }
| |
468#endif
| 494#endif
|
| 495 }
|
469 return (0);
470}
471
472void
473l_perror(s)
474 char *s;
475{
476 switch (errno) {
477
478 case ESRCH:
479 warnx("%s: no disk label on disk;", s);
480 fprintf(stderr,
481 "use \"disklabel -r\" to install initial label\n");
482 break;
483
484 case EINVAL:
485 warnx("%s: label magic number or checksum is wrong!", s);
486 fprintf(stderr, "(disklabel or kernel is out of date?)\n");
487 break;
488
489 case EBUSY:
490 warnx("%s: open partition would move or shrink", s);
491 break;
492
493 case EXDEV:
494 warnx("%s: '%c' partition must start at beginning of disk",
495 s, 'a' + RAW_PART);
496 break;
497
498 default:
499 warn((char *)NULL);
500 break;
501 }
502}
503
504/*
505 * Fetch disklabel for disk.
506 * Use ioctl to get label unless -r flag is given.
507 */
508struct disklabel *
509readlabel(f)
510 int f;
511{
512 register struct disklabel *lp;
513
514 if (rflag) {
515 if (read(f, bootarea, BBSIZE) < BBSIZE)
516 err(4, "%s", specname);
517 for (lp = (struct disklabel *)bootarea;
518 lp <= (struct disklabel *)(bootarea + BBSIZE - sizeof(*lp));
519 lp = (struct disklabel *)((char *)lp + 16))
520 if (lp->d_magic == DISKMAGIC &&
521 lp->d_magic2 == DISKMAGIC)
522 break;
523 if (lp > (struct disklabel *)(bootarea+BBSIZE-sizeof(*lp)) ||
524 lp->d_magic != DISKMAGIC || lp->d_magic2 != DISKMAGIC ||
525 dkcksum(lp) != 0)
526 errx(1,
527 "bad pack magic number (label is damaged, or pack is unlabeled)");
528 } else {
529 lp = &lab;
530 if (ioctl(f, DIOCGDINFO, lp) < 0)
531 err(4, "ioctl DIOCGDINFO");
532 }
533 return (lp);
534}
535
536/*
537 * Construct a bootarea (d_bbsize bytes) in the specified buffer ``boot''
538 * Returns a pointer to the disklabel portion of the bootarea.
539 */
540struct disklabel *
541makebootarea(boot, dp, f)
542 char *boot;
543 register struct disklabel *dp;
544 int f;
545{
546 struct disklabel *lp;
547 register char *p;
548 int b;
549#if NUMBOOT > 0
550 char *dkbasename;
551 struct stat sb;
552#endif
553#ifdef __alpha__
554 u_long *bootinfo;
555 int n;
556#endif
557#ifdef __i386__
558 char *tmpbuf;
559 int i, found;
560#endif
561
562 /* XXX */
563 if (dp->d_secsize == 0) {
564 dp->d_secsize = DEV_BSIZE;
565 dp->d_bbsize = BBSIZE;
566 }
567 lp = (struct disklabel *)
568 (boot + (LABELSECTOR * dp->d_secsize) + LABELOFFSET);
569 bzero((char *)lp, sizeof *lp);
570#if NUMBOOT > 0
571 /*
572 * If we are not installing a boot program but we are installing a
573 * label on disk then we must read the current bootarea so we don't
574 * clobber the existing boot.
575 */
576 if (!installboot) {
577 if (rflag) {
578 if (read(f, boot, BBSIZE) < BBSIZE)
579 err(4, "%s", specname);
580 bzero((char *)lp, sizeof *lp);
581 }
582 return (lp);
583 }
584 /*
585 * We are installing a boot program. Determine the name(s) and
586 * read them into the appropriate places in the boot area.
587 */
588 if (!xxboot || !bootxx) {
589 dkbasename = np;
590 if ((p = rindex(dkname, '/')) == NULL)
591 p = dkname;
592 else
593 p++;
594 while (*p && !isdigit(*p))
595 *np++ = *p++;
596 *np++ = '\0';
597
598 if (!xxboot) {
599 (void)sprintf(boot0, "%s/boot1", _PATH_BOOTDIR);
600 xxboot = boot0;
601 }
602#if NUMBOOT > 1
603 if (!bootxx) {
604 (void)sprintf(boot1, "%s/boot2", _PATH_BOOTDIR);
605 bootxx = boot1;
606 }
607#endif
608 }
609#ifdef DEBUG
610 if (debug)
611 fprintf(stderr, "bootstraps: xxboot = %s, bootxx = %s\n",
612 xxboot, bootxx ? bootxx : "NONE");
613#endif
614
615 /*
616 * Strange rules:
617 * 1. One-piece bootstrap (hp300/hp800)
618 * up to d_bbsize bytes of ``xxboot'' go in bootarea, the rest
619 * is remembered and written later following the bootarea.
620 * 2. Two-piece bootstraps (vax/i386?/mips?)
621 * up to d_secsize bytes of ``xxboot'' go in first d_secsize
622 * bytes of bootarea, remaining d_bbsize-d_secsize filled
623 * from ``bootxx''.
624 */
625 b = open(xxboot, O_RDONLY);
626 if (b < 0)
627 err(4, "%s", xxboot);
628#if NUMBOOT > 1
629#ifdef __i386__
630 /*
631 * XXX Botch alert.
632 * The i386 has the so-called fdisk table embedded into the
633 * primary bootstrap. We take care to not clobber it, but
634 * only if it does already contain some data. (Otherwise,
635 * the xxboot provides a template.)
636 */
637 if ((tmpbuf = (char *)malloc((int)dp->d_secsize)) == 0)
638 err(4, "%s", xxboot);
639 memcpy((void *)tmpbuf, (void *)boot, (int)dp->d_secsize);
640#endif /* i386 */
641 if (read(b, boot, (int)dp->d_secsize) < 0)
642 err(4, "%s", xxboot);
643 (void)close(b);
644#ifdef __i386__
645 for (i = DOSPARTOFF, found = 0;
646 !found && i < DOSPARTOFF + NDOSPART*sizeof(struct dos_partition);
647 i++)
648 found = tmpbuf[i] != 0;
649 if (found)
650 memcpy((void *)&boot[DOSPARTOFF],
651 (void *)&tmpbuf[DOSPARTOFF],
652 NDOSPART * sizeof(struct dos_partition));
653 free(tmpbuf);
654#endif /* i386 */
655 b = open(bootxx, O_RDONLY);
656 if (b < 0)
657 err(4, "%s", bootxx);
658 if (fstat(b, &sb) != 0)
659 err(4, "%s", bootxx);
660 if (dp->d_secsize + sb.st_size > dp->d_bbsize)
661 errx(4, "%s too large", bootxx);
662 if (read(b, &boot[dp->d_secsize],
663 (int)(dp->d_bbsize-dp->d_secsize)) < 0)
664 err(4, "%s", bootxx);
665#else /* !(NUMBOOT > 1) */
666#ifdef __alpha__
667 /*
668 * On the alpha, the primary bootstrap starts at the
669 * second sector of the boot area. The first sector
670 * contains the label and must be edited to contain the
671 * size and location of the primary bootstrap.
672 */
673 n = read(b, boot + dp->d_secsize, (int)dp->d_bbsize);
674 if (n < 0)
675 err(4, "%s", xxboot);
676 bootinfo = (u_long *)(boot + 480);
677 bootinfo[0] = (n + dp->d_secsize - 1) / dp->d_secsize;
678 bootinfo[1] = 1; /* start at sector 1 */
679 bootinfo[2] = 0; /* flags (must be zero) */
680#else /* !__alpha__ */
681 if (read(b, boot, (int)dp->d_bbsize) < 0)
682 err(4, "%s", xxboot);
683#endif /* __alpha__ */
684 if (fstat(b, &sb) != 0)
685 err(4, "%s", xxboot);
686 bootsize = (int)sb.st_size - dp->d_bbsize;
687 if (bootsize > 0) {
688 /* XXX assume d_secsize is a power of two */
689 bootsize = (bootsize + dp->d_secsize-1) & ~(dp->d_secsize-1);
690 bootbuf = (char *)malloc((size_t)bootsize);
691 if (bootbuf == 0)
692 err(4, "%s", xxboot);
693 if (read(b, bootbuf, bootsize) < 0) {
694 free(bootbuf);
695 err(4, "%s", xxboot);
696 }
697 }
698#endif /* NUMBOOT > 1 */
699 (void)close(b);
700#endif /* NUMBOOT > 0 */
701 /*
702 * Make sure no part of the bootstrap is written in the area
703 * reserved for the label.
704 */
705 for (p = (char *)lp; p < (char *)lp + sizeof(struct disklabel); p++)
706 if (*p)
707 errx(2, "bootstrap doesn't leave room for disk label");
708 return (lp);
709}
710
711void
712display(f, lp)
713 FILE *f;
714 register struct disklabel *lp;
715{
716 register int i, j;
717 register struct partition *pp;
718
719 fprintf(f, "# %s:\n", specname);
720 if ((unsigned) lp->d_type < DKMAXTYPES)
721 fprintf(f, "type: %s\n", dktypenames[lp->d_type]);
722 else
723 fprintf(f, "type: %u\n", lp->d_type);
724 fprintf(f, "disk: %.*s\n", (int)sizeof(lp->d_typename),
725 lp->d_typename);
726 fprintf(f, "label: %.*s\n", (int)sizeof(lp->d_packname),
727 lp->d_packname);
728 fprintf(f, "flags:");
729 if (lp->d_flags & D_REMOVABLE)
730 fprintf(f, " removeable");
731 if (lp->d_flags & D_ECC)
732 fprintf(f, " ecc");
733 if (lp->d_flags & D_BADSECT)
734 fprintf(f, " badsect");
735 fprintf(f, "\n");
736 fprintf(f, "bytes/sector: %lu\n", (u_long)lp->d_secsize);
737 fprintf(f, "sectors/track: %lu\n", (u_long)lp->d_nsectors);
738 fprintf(f, "tracks/cylinder: %lu\n", (u_long)lp->d_ntracks);
739 fprintf(f, "sectors/cylinder: %lu\n", (u_long)lp->d_secpercyl);
740 fprintf(f, "cylinders: %lu\n", (u_long)lp->d_ncylinders);
741 fprintf(f, "sectors/unit: %lu\n", (u_long)lp->d_secperunit);
742 fprintf(f, "rpm: %u\n", lp->d_rpm);
743 fprintf(f, "interleave: %u\n", lp->d_interleave);
744 fprintf(f, "trackskew: %u\n", lp->d_trackskew);
745 fprintf(f, "cylinderskew: %u\n", lp->d_cylskew);
746 fprintf(f, "headswitch: %lu\t\t# milliseconds\n",
747 (u_long)lp->d_headswitch);
748 fprintf(f, "track-to-track seek: %ld\t# milliseconds\n",
749 (u_long)lp->d_trkseek);
750 fprintf(f, "drivedata: ");
751 for (i = NDDATA - 1; i >= 0; i--)
752 if (lp->d_drivedata[i])
753 break;
754 if (i < 0)
755 i = 0;
756 for (j = 0; j <= i; j++)
757 fprintf(f, "%lu ", (u_long)lp->d_drivedata[j]);
758 fprintf(f, "\n\n%u partitions:\n", lp->d_npartitions);
759 fprintf(f,
760 "# size offset fstype [fsize bsize bps/cpg]\n");
761 pp = lp->d_partitions;
762 for (i = 0; i < lp->d_npartitions; i++, pp++) {
763 if (pp->p_size) {
764 fprintf(f, " %c: %8lu %8lu ", 'a' + i,
765 (u_long)pp->p_size, (u_long)pp->p_offset);
766 if ((unsigned) pp->p_fstype < FSMAXTYPES)
767 fprintf(f, "%8.8s", fstypenames[pp->p_fstype]);
768 else
769 fprintf(f, "%8d", pp->p_fstype);
770 switch (pp->p_fstype) {
771
772 case FS_UNUSED: /* XXX */
773 fprintf(f, " %5lu %5lu %5.5s ",
774 (u_long)pp->p_fsize,
775 (u_long)(pp->p_fsize * pp->p_frag), "");
776 break;
777
778 case FS_BSDFFS:
779 fprintf(f, " %5lu %5lu %5u ",
780 (u_long)pp->p_fsize,
781 (u_long)(pp->p_fsize * pp->p_frag),
782 pp->p_cpg);
783 break;
784
785 case FS_BSDLFS:
786 fprintf(f, " %5lu %5lu %5d",
787 (u_long)pp->p_fsize,
788 (u_long)(pp->p_fsize * pp->p_frag),
789 pp->p_cpg);
790 break;
791
792 default:
793 fprintf(f, "%20.20s", "");
794 break;
795 }
796 fprintf(f, "\t# (Cyl. %4lu",
797 (u_long)(pp->p_offset / lp->d_secpercyl));
798 if (pp->p_offset % lp->d_secpercyl)
799 putc('*', f);
800 else
801 putc(' ', f);
802 fprintf(f, "- %lu",
803 (u_long)((pp->p_offset + pp->p_size +
804 lp->d_secpercyl - 1) /
805 lp->d_secpercyl - 1));
806 if (pp->p_size % lp->d_secpercyl)
807 putc('*', f);
808 fprintf(f, ")\n");
809 }
810 }
811 fflush(f);
812}
813
814int
815edit(lp, f)
816 struct disklabel *lp;
817 int f;
818{
819 register int c, fd;
820 struct disklabel label;
821 FILE *fp;
822
823 if ((fd = mkstemp(tmpfil)) == -1 ||
824 (fp = fdopen(fd, "w")) == NULL) {
825 warnx("can't create %s", tmpfil);
826 return (1);
827 }
828 display(fp, lp);
829 fclose(fp);
830 for (;;) {
831 if (!editit())
832 break;
833 fp = fopen(tmpfil, "r");
834 if (fp == NULL) {
835 warnx("can't reopen %s for reading", tmpfil);
836 break;
837 }
838 bzero((char *)&label, sizeof(label));
839 if (getasciilabel(fp, &label)) {
840 *lp = label;
841 if (writelabel(f, bootarea, lp) == 0) {
842 fclose(fp);
843 (void) unlink(tmpfil);
844 return (0);
845 }
846 }
847 fclose(fp);
848 printf("re-edit the label? [y]: "); fflush(stdout);
849 c = getchar();
850 if (c != EOF && c != (int)'\n')
851 while (getchar() != (int)'\n')
852 ;
853 if (c == (int)'n')
854 break;
855 }
856 (void) unlink(tmpfil);
857 return (1);
858}
859
860int
861editit()
862{
863 register int pid, xpid;
864 int stat, omask;
865
866 omask = sigblock(sigmask(SIGINT)|sigmask(SIGQUIT)|sigmask(SIGHUP));
867 while ((pid = fork()) < 0) {
868 if (errno == EPROCLIM) {
869 warnx("you have too many processes");
870 return(0);
871 }
872 if (errno != EAGAIN) {
873 warn("fork");
874 return(0);
875 }
876 sleep(1);
877 }
878 if (pid == 0) {
879 register char *ed;
880
881 sigsetmask(omask);
882 setgid(getgid());
883 setuid(getuid());
884 if ((ed = getenv("EDITOR")) == (char *)0)
885 ed = DEFEDITOR;
886 execlp(ed, ed, tmpfil, 0);
887 err(1, "%s", ed);
888 }
889 while ((xpid = wait(&stat)) >= 0)
890 if (xpid == pid)
891 break;
892 sigsetmask(omask);
893 return(!stat);
894}
895
896char *
897skip(cp)
898 register char *cp;
899{
900
901 while (*cp != '\0' && isspace(*cp))
902 cp++;
903 if (*cp == '\0' || *cp == '#')
904 return ((char *)NULL);
905 return (cp);
906}
907
908char *
909word(cp)
910 register char *cp;
911{
912 register char c;
913
914 while (*cp != '\0' && !isspace(*cp) && *cp != '#')
915 cp++;
916 if ((c = *cp) != '\0') {
917 *cp++ = '\0';
918 if (c != '#')
919 return (skip(cp));
920 }
921 return ((char *)NULL);
922}
923
924/*
925 * Read an ascii label in from fd f,
926 * in the same format as that put out by display(),
927 * and fill in lp.
928 */
929int
930getasciilabel(f, lp)
931 FILE *f;
932 register struct disklabel *lp;
933{
934 register char **cpp, *cp;
935 register struct partition *pp;
| 496 return (0);
497}
498
499void
500l_perror(s)
501 char *s;
502{
503 switch (errno) {
504
505 case ESRCH:
506 warnx("%s: no disk label on disk;", s);
507 fprintf(stderr,
508 "use \"disklabel -r\" to install initial label\n");
509 break;
510
511 case EINVAL:
512 warnx("%s: label magic number or checksum is wrong!", s);
513 fprintf(stderr, "(disklabel or kernel is out of date?)\n");
514 break;
515
516 case EBUSY:
517 warnx("%s: open partition would move or shrink", s);
518 break;
519
520 case EXDEV:
521 warnx("%s: '%c' partition must start at beginning of disk",
522 s, 'a' + RAW_PART);
523 break;
524
525 default:
526 warn((char *)NULL);
527 break;
528 }
529}
530
531/*
532 * Fetch disklabel for disk.
533 * Use ioctl to get label unless -r flag is given.
534 */
535struct disklabel *
536readlabel(f)
537 int f;
538{
539 register struct disklabel *lp;
540
541 if (rflag) {
542 if (read(f, bootarea, BBSIZE) < BBSIZE)
543 err(4, "%s", specname);
544 for (lp = (struct disklabel *)bootarea;
545 lp <= (struct disklabel *)(bootarea + BBSIZE - sizeof(*lp));
546 lp = (struct disklabel *)((char *)lp + 16))
547 if (lp->d_magic == DISKMAGIC &&
548 lp->d_magic2 == DISKMAGIC)
549 break;
550 if (lp > (struct disklabel *)(bootarea+BBSIZE-sizeof(*lp)) ||
551 lp->d_magic != DISKMAGIC || lp->d_magic2 != DISKMAGIC ||
552 dkcksum(lp) != 0)
553 errx(1,
554 "bad pack magic number (label is damaged, or pack is unlabeled)");
555 } else {
556 lp = &lab;
557 if (ioctl(f, DIOCGDINFO, lp) < 0)
558 err(4, "ioctl DIOCGDINFO");
559 }
560 return (lp);
561}
562
563/*
564 * Construct a bootarea (d_bbsize bytes) in the specified buffer ``boot''
565 * Returns a pointer to the disklabel portion of the bootarea.
566 */
567struct disklabel *
568makebootarea(boot, dp, f)
569 char *boot;
570 register struct disklabel *dp;
571 int f;
572{
573 struct disklabel *lp;
574 register char *p;
575 int b;
576#if NUMBOOT > 0
577 char *dkbasename;
578 struct stat sb;
579#endif
580#ifdef __alpha__
581 u_long *bootinfo;
582 int n;
583#endif
584#ifdef __i386__
585 char *tmpbuf;
586 int i, found;
587#endif
588
589 /* XXX */
590 if (dp->d_secsize == 0) {
591 dp->d_secsize = DEV_BSIZE;
592 dp->d_bbsize = BBSIZE;
593 }
594 lp = (struct disklabel *)
595 (boot + (LABELSECTOR * dp->d_secsize) + LABELOFFSET);
596 bzero((char *)lp, sizeof *lp);
597#if NUMBOOT > 0
598 /*
599 * If we are not installing a boot program but we are installing a
600 * label on disk then we must read the current bootarea so we don't
601 * clobber the existing boot.
602 */
603 if (!installboot) {
604 if (rflag) {
605 if (read(f, boot, BBSIZE) < BBSIZE)
606 err(4, "%s", specname);
607 bzero((char *)lp, sizeof *lp);
608 }
609 return (lp);
610 }
611 /*
612 * We are installing a boot program. Determine the name(s) and
613 * read them into the appropriate places in the boot area.
614 */
615 if (!xxboot || !bootxx) {
616 dkbasename = np;
617 if ((p = rindex(dkname, '/')) == NULL)
618 p = dkname;
619 else
620 p++;
621 while (*p && !isdigit(*p))
622 *np++ = *p++;
623 *np++ = '\0';
624
625 if (!xxboot) {
626 (void)sprintf(boot0, "%s/boot1", _PATH_BOOTDIR);
627 xxboot = boot0;
628 }
629#if NUMBOOT > 1
630 if (!bootxx) {
631 (void)sprintf(boot1, "%s/boot2", _PATH_BOOTDIR);
632 bootxx = boot1;
633 }
634#endif
635 }
636#ifdef DEBUG
637 if (debug)
638 fprintf(stderr, "bootstraps: xxboot = %s, bootxx = %s\n",
639 xxboot, bootxx ? bootxx : "NONE");
640#endif
641
642 /*
643 * Strange rules:
644 * 1. One-piece bootstrap (hp300/hp800)
645 * up to d_bbsize bytes of ``xxboot'' go in bootarea, the rest
646 * is remembered and written later following the bootarea.
647 * 2. Two-piece bootstraps (vax/i386?/mips?)
648 * up to d_secsize bytes of ``xxboot'' go in first d_secsize
649 * bytes of bootarea, remaining d_bbsize-d_secsize filled
650 * from ``bootxx''.
651 */
652 b = open(xxboot, O_RDONLY);
653 if (b < 0)
654 err(4, "%s", xxboot);
655#if NUMBOOT > 1
656#ifdef __i386__
657 /*
658 * XXX Botch alert.
659 * The i386 has the so-called fdisk table embedded into the
660 * primary bootstrap. We take care to not clobber it, but
661 * only if it does already contain some data. (Otherwise,
662 * the xxboot provides a template.)
663 */
664 if ((tmpbuf = (char *)malloc((int)dp->d_secsize)) == 0)
665 err(4, "%s", xxboot);
666 memcpy((void *)tmpbuf, (void *)boot, (int)dp->d_secsize);
667#endif /* i386 */
668 if (read(b, boot, (int)dp->d_secsize) < 0)
669 err(4, "%s", xxboot);
670 (void)close(b);
671#ifdef __i386__
672 for (i = DOSPARTOFF, found = 0;
673 !found && i < DOSPARTOFF + NDOSPART*sizeof(struct dos_partition);
674 i++)
675 found = tmpbuf[i] != 0;
676 if (found)
677 memcpy((void *)&boot[DOSPARTOFF],
678 (void *)&tmpbuf[DOSPARTOFF],
679 NDOSPART * sizeof(struct dos_partition));
680 free(tmpbuf);
681#endif /* i386 */
682 b = open(bootxx, O_RDONLY);
683 if (b < 0)
684 err(4, "%s", bootxx);
685 if (fstat(b, &sb) != 0)
686 err(4, "%s", bootxx);
687 if (dp->d_secsize + sb.st_size > dp->d_bbsize)
688 errx(4, "%s too large", bootxx);
689 if (read(b, &boot[dp->d_secsize],
690 (int)(dp->d_bbsize-dp->d_secsize)) < 0)
691 err(4, "%s", bootxx);
692#else /* !(NUMBOOT > 1) */
693#ifdef __alpha__
694 /*
695 * On the alpha, the primary bootstrap starts at the
696 * second sector of the boot area. The first sector
697 * contains the label and must be edited to contain the
698 * size and location of the primary bootstrap.
699 */
700 n = read(b, boot + dp->d_secsize, (int)dp->d_bbsize);
701 if (n < 0)
702 err(4, "%s", xxboot);
703 bootinfo = (u_long *)(boot + 480);
704 bootinfo[0] = (n + dp->d_secsize - 1) / dp->d_secsize;
705 bootinfo[1] = 1; /* start at sector 1 */
706 bootinfo[2] = 0; /* flags (must be zero) */
707#else /* !__alpha__ */
708 if (read(b, boot, (int)dp->d_bbsize) < 0)
709 err(4, "%s", xxboot);
710#endif /* __alpha__ */
711 if (fstat(b, &sb) != 0)
712 err(4, "%s", xxboot);
713 bootsize = (int)sb.st_size - dp->d_bbsize;
714 if (bootsize > 0) {
715 /* XXX assume d_secsize is a power of two */
716 bootsize = (bootsize + dp->d_secsize-1) & ~(dp->d_secsize-1);
717 bootbuf = (char *)malloc((size_t)bootsize);
718 if (bootbuf == 0)
719 err(4, "%s", xxboot);
720 if (read(b, bootbuf, bootsize) < 0) {
721 free(bootbuf);
722 err(4, "%s", xxboot);
723 }
724 }
725#endif /* NUMBOOT > 1 */
726 (void)close(b);
727#endif /* NUMBOOT > 0 */
728 /*
729 * Make sure no part of the bootstrap is written in the area
730 * reserved for the label.
731 */
732 for (p = (char *)lp; p < (char *)lp + sizeof(struct disklabel); p++)
733 if (*p)
734 errx(2, "bootstrap doesn't leave room for disk label");
735 return (lp);
736}
737
738void
739display(f, lp)
740 FILE *f;
741 register struct disklabel *lp;
742{
743 register int i, j;
744 register struct partition *pp;
745
746 fprintf(f, "# %s:\n", specname);
747 if ((unsigned) lp->d_type < DKMAXTYPES)
748 fprintf(f, "type: %s\n", dktypenames[lp->d_type]);
749 else
750 fprintf(f, "type: %u\n", lp->d_type);
751 fprintf(f, "disk: %.*s\n", (int)sizeof(lp->d_typename),
752 lp->d_typename);
753 fprintf(f, "label: %.*s\n", (int)sizeof(lp->d_packname),
754 lp->d_packname);
755 fprintf(f, "flags:");
756 if (lp->d_flags & D_REMOVABLE)
757 fprintf(f, " removeable");
758 if (lp->d_flags & D_ECC)
759 fprintf(f, " ecc");
760 if (lp->d_flags & D_BADSECT)
761 fprintf(f, " badsect");
762 fprintf(f, "\n");
763 fprintf(f, "bytes/sector: %lu\n", (u_long)lp->d_secsize);
764 fprintf(f, "sectors/track: %lu\n", (u_long)lp->d_nsectors);
765 fprintf(f, "tracks/cylinder: %lu\n", (u_long)lp->d_ntracks);
766 fprintf(f, "sectors/cylinder: %lu\n", (u_long)lp->d_secpercyl);
767 fprintf(f, "cylinders: %lu\n", (u_long)lp->d_ncylinders);
768 fprintf(f, "sectors/unit: %lu\n", (u_long)lp->d_secperunit);
769 fprintf(f, "rpm: %u\n", lp->d_rpm);
770 fprintf(f, "interleave: %u\n", lp->d_interleave);
771 fprintf(f, "trackskew: %u\n", lp->d_trackskew);
772 fprintf(f, "cylinderskew: %u\n", lp->d_cylskew);
773 fprintf(f, "headswitch: %lu\t\t# milliseconds\n",
774 (u_long)lp->d_headswitch);
775 fprintf(f, "track-to-track seek: %ld\t# milliseconds\n",
776 (u_long)lp->d_trkseek);
777 fprintf(f, "drivedata: ");
778 for (i = NDDATA - 1; i >= 0; i--)
779 if (lp->d_drivedata[i])
780 break;
781 if (i < 0)
782 i = 0;
783 for (j = 0; j <= i; j++)
784 fprintf(f, "%lu ", (u_long)lp->d_drivedata[j]);
785 fprintf(f, "\n\n%u partitions:\n", lp->d_npartitions);
786 fprintf(f,
787 "# size offset fstype [fsize bsize bps/cpg]\n");
788 pp = lp->d_partitions;
789 for (i = 0; i < lp->d_npartitions; i++, pp++) {
790 if (pp->p_size) {
791 fprintf(f, " %c: %8lu %8lu ", 'a' + i,
792 (u_long)pp->p_size, (u_long)pp->p_offset);
793 if ((unsigned) pp->p_fstype < FSMAXTYPES)
794 fprintf(f, "%8.8s", fstypenames[pp->p_fstype]);
795 else
796 fprintf(f, "%8d", pp->p_fstype);
797 switch (pp->p_fstype) {
798
799 case FS_UNUSED: /* XXX */
800 fprintf(f, " %5lu %5lu %5.5s ",
801 (u_long)pp->p_fsize,
802 (u_long)(pp->p_fsize * pp->p_frag), "");
803 break;
804
805 case FS_BSDFFS:
806 fprintf(f, " %5lu %5lu %5u ",
807 (u_long)pp->p_fsize,
808 (u_long)(pp->p_fsize * pp->p_frag),
809 pp->p_cpg);
810 break;
811
812 case FS_BSDLFS:
813 fprintf(f, " %5lu %5lu %5d",
814 (u_long)pp->p_fsize,
815 (u_long)(pp->p_fsize * pp->p_frag),
816 pp->p_cpg);
817 break;
818
819 default:
820 fprintf(f, "%20.20s", "");
821 break;
822 }
823 fprintf(f, "\t# (Cyl. %4lu",
824 (u_long)(pp->p_offset / lp->d_secpercyl));
825 if (pp->p_offset % lp->d_secpercyl)
826 putc('*', f);
827 else
828 putc(' ', f);
829 fprintf(f, "- %lu",
830 (u_long)((pp->p_offset + pp->p_size +
831 lp->d_secpercyl - 1) /
832 lp->d_secpercyl - 1));
833 if (pp->p_size % lp->d_secpercyl)
834 putc('*', f);
835 fprintf(f, ")\n");
836 }
837 }
838 fflush(f);
839}
840
841int
842edit(lp, f)
843 struct disklabel *lp;
844 int f;
845{
846 register int c, fd;
847 struct disklabel label;
848 FILE *fp;
849
850 if ((fd = mkstemp(tmpfil)) == -1 ||
851 (fp = fdopen(fd, "w")) == NULL) {
852 warnx("can't create %s", tmpfil);
853 return (1);
854 }
855 display(fp, lp);
856 fclose(fp);
857 for (;;) {
858 if (!editit())
859 break;
860 fp = fopen(tmpfil, "r");
861 if (fp == NULL) {
862 warnx("can't reopen %s for reading", tmpfil);
863 break;
864 }
865 bzero((char *)&label, sizeof(label));
866 if (getasciilabel(fp, &label)) {
867 *lp = label;
868 if (writelabel(f, bootarea, lp) == 0) {
869 fclose(fp);
870 (void) unlink(tmpfil);
871 return (0);
872 }
873 }
874 fclose(fp);
875 printf("re-edit the label? [y]: "); fflush(stdout);
876 c = getchar();
877 if (c != EOF && c != (int)'\n')
878 while (getchar() != (int)'\n')
879 ;
880 if (c == (int)'n')
881 break;
882 }
883 (void) unlink(tmpfil);
884 return (1);
885}
886
887int
888editit()
889{
890 register int pid, xpid;
891 int stat, omask;
892
893 omask = sigblock(sigmask(SIGINT)|sigmask(SIGQUIT)|sigmask(SIGHUP));
894 while ((pid = fork()) < 0) {
895 if (errno == EPROCLIM) {
896 warnx("you have too many processes");
897 return(0);
898 }
899 if (errno != EAGAIN) {
900 warn("fork");
901 return(0);
902 }
903 sleep(1);
904 }
905 if (pid == 0) {
906 register char *ed;
907
908 sigsetmask(omask);
909 setgid(getgid());
910 setuid(getuid());
911 if ((ed = getenv("EDITOR")) == (char *)0)
912 ed = DEFEDITOR;
913 execlp(ed, ed, tmpfil, 0);
914 err(1, "%s", ed);
915 }
916 while ((xpid = wait(&stat)) >= 0)
917 if (xpid == pid)
918 break;
919 sigsetmask(omask);
920 return(!stat);
921}
922
923char *
924skip(cp)
925 register char *cp;
926{
927
928 while (*cp != '\0' && isspace(*cp))
929 cp++;
930 if (*cp == '\0' || *cp == '#')
931 return ((char *)NULL);
932 return (cp);
933}
934
935char *
936word(cp)
937 register char *cp;
938{
939 register char c;
940
941 while (*cp != '\0' && !isspace(*cp) && *cp != '#')
942 cp++;
943 if ((c = *cp) != '\0') {
944 *cp++ = '\0';
945 if (c != '#')
946 return (skip(cp));
947 }
948 return ((char *)NULL);
949}
950
951/*
952 * Read an ascii label in from fd f,
953 * in the same format as that put out by display(),
954 * and fill in lp.
955 */
956int
957getasciilabel(f, lp)
958 FILE *f;
959 register struct disklabel *lp;
960{
961 register char **cpp, *cp;
962 register struct partition *pp;
|
| 963 int i;
964 unsigned int part;
|
936 char *tp, *s, line[BUFSIZ];
937 int v, lineno = 0, errors = 0;
938
939 lp->d_bbsize = BBSIZE; /* XXX */
940 lp->d_sbsize = SBSIZE; /* XXX */
941 while (fgets(line, sizeof(line) - 1, f)) {
942 lineno++;
943 if ((cp = index(line,'\n')) != 0)
944 *cp = '\0';
945 cp = skip(line);
946 if (cp == NULL)
947 continue;
948 tp = index(cp, ':');
949 if (tp == NULL) {
950 fprintf(stderr, "line %d: syntax error\n", lineno);
951 errors++;
952 continue;
953 }
954 *tp++ = '\0', tp = skip(tp);
955 if (streq(cp, "type")) {
956 if (tp == NULL)
957 tp = "unknown";
958 cpp = dktypenames;
959 for (; cpp < &dktypenames[DKMAXTYPES]; cpp++)
960 if ((s = *cpp) && streq(s, tp)) {
961 lp->d_type = cpp - dktypenames;
962 goto next;
963 }
964 v = atoi(tp);
965 if ((unsigned)v >= DKMAXTYPES)
966 fprintf(stderr, "line %d:%s %d\n", lineno,
967 "Warning, unknown disk type", v);
968 lp->d_type = v;
969 continue;
970 }
971 if (streq(cp, "flags")) {
972 for (v = 0; (cp = tp) && *cp != '\0';) {
973 tp = word(cp);
974 if (streq(cp, "removeable"))
975 v |= D_REMOVABLE;
976 else if (streq(cp, "ecc"))
977 v |= D_ECC;
978 else if (streq(cp, "badsect"))
979 v |= D_BADSECT;
980 else {
981 fprintf(stderr,
982 "line %d: %s: bad flag\n",
983 lineno, cp);
984 errors++;
985 }
986 }
987 lp->d_flags = v;
988 continue;
989 }
990 if (streq(cp, "drivedata")) {
991 register int i;
992
993 for (i = 0; (cp = tp) && *cp != '\0' && i < NDDATA;) {
994 lp->d_drivedata[i++] = atoi(cp);
995 tp = word(cp);
996 }
997 continue;
998 }
999 if (sscanf(cp, "%d partitions", &v) == 1) {
1000 if (v == 0 || (unsigned)v > MAXPARTITIONS) {
1001 fprintf(stderr,
1002 "line %d: bad # of partitions\n", lineno);
1003 lp->d_npartitions = MAXPARTITIONS;
1004 errors++;
1005 } else
1006 lp->d_npartitions = v;
1007 continue;
1008 }
1009 if (tp == NULL)
1010 tp = "";
1011 if (streq(cp, "disk")) {
1012 strncpy(lp->d_typename, tp, sizeof (lp->d_typename));
1013 continue;
1014 }
1015 if (streq(cp, "label")) {
1016 strncpy(lp->d_packname, tp, sizeof (lp->d_packname));
1017 continue;
1018 }
1019 if (streq(cp, "bytes/sector")) {
1020 v = atoi(tp);
1021 if (v <= 0 || (v % DEV_BSIZE) != 0) {
1022 fprintf(stderr,
1023 "line %d: %s: bad sector size\n",
1024 lineno, tp);
1025 errors++;
1026 } else
1027 lp->d_secsize = v;
1028 continue;
1029 }
1030 if (streq(cp, "sectors/track")) {
1031 v = atoi(tp);
1032 if (v <= 0) {
1033 fprintf(stderr, "line %d: %s: bad %s\n",
1034 lineno, tp, cp);
1035 errors++;
1036 } else
1037 lp->d_nsectors = v;
1038 continue;
1039 }
1040 if (streq(cp, "sectors/cylinder")) {
1041 v = atoi(tp);
1042 if (v <= 0) {
1043 fprintf(stderr, "line %d: %s: bad %s\n",
1044 lineno, tp, cp);
1045 errors++;
1046 } else
1047 lp->d_secpercyl = v;
1048 continue;
1049 }
1050 if (streq(cp, "tracks/cylinder")) {
1051 v = atoi(tp);
1052 if (v <= 0) {
1053 fprintf(stderr, "line %d: %s: bad %s\n",
1054 lineno, tp, cp);
1055 errors++;
1056 } else
1057 lp->d_ntracks = v;
1058 continue;
1059 }
1060 if (streq(cp, "cylinders")) {
1061 v = atoi(tp);
1062 if (v <= 0) {
1063 fprintf(stderr, "line %d: %s: bad %s\n",
1064 lineno, tp, cp);
1065 errors++;
1066 } else
1067 lp->d_ncylinders = v;
1068 continue;
1069 }
1070 if (streq(cp, "sectors/unit")) {
1071 v = atoi(tp);
1072 if (v <= 0) {
1073 fprintf(stderr, "line %d: %s: bad %s\n",
1074 lineno, tp, cp);
1075 errors++;
1076 } else
1077 lp->d_secperunit = v;
1078 continue;
1079 }
1080 if (streq(cp, "rpm")) {
1081 v = atoi(tp);
1082 if (v <= 0) {
1083 fprintf(stderr, "line %d: %s: bad %s\n",
1084 lineno, tp, cp);
1085 errors++;
1086 } else
1087 lp->d_rpm = v;
1088 continue;
1089 }
1090 if (streq(cp, "interleave")) {
1091 v = atoi(tp);
1092 if (v <= 0) {
1093 fprintf(stderr, "line %d: %s: bad %s\n",
1094 lineno, tp, cp);
1095 errors++;
1096 } else
1097 lp->d_interleave = v;
1098 continue;
1099 }
1100 if (streq(cp, "trackskew")) {
1101 v = atoi(tp);
1102 if (v < 0) {
1103 fprintf(stderr, "line %d: %s: bad %s\n",
1104 lineno, tp, cp);
1105 errors++;
1106 } else
1107 lp->d_trackskew = v;
1108 continue;
1109 }
1110 if (streq(cp, "cylinderskew")) {
1111 v = atoi(tp);
1112 if (v < 0) {
1113 fprintf(stderr, "line %d: %s: bad %s\n",
1114 lineno, tp, cp);
1115 errors++;
1116 } else
1117 lp->d_cylskew = v;
1118 continue;
1119 }
1120 if (streq(cp, "headswitch")) {
1121 v = atoi(tp);
1122 if (v < 0) {
1123 fprintf(stderr, "line %d: %s: bad %s\n",
1124 lineno, tp, cp);
1125 errors++;
1126 } else
1127 lp->d_headswitch = v;
1128 continue;
1129 }
1130 if (streq(cp, "track-to-track seek")) {
1131 v = atoi(tp);
1132 if (v < 0) {
1133 fprintf(stderr, "line %d: %s: bad %s\n",
1134 lineno, tp, cp);
1135 errors++;
1136 } else
1137 lp->d_trkseek = v;
1138 continue;
1139 }
| 965 char *tp, *s, line[BUFSIZ];
966 int v, lineno = 0, errors = 0;
967
968 lp->d_bbsize = BBSIZE; /* XXX */
969 lp->d_sbsize = SBSIZE; /* XXX */
970 while (fgets(line, sizeof(line) - 1, f)) {
971 lineno++;
972 if ((cp = index(line,'\n')) != 0)
973 *cp = '\0';
974 cp = skip(line);
975 if (cp == NULL)
976 continue;
977 tp = index(cp, ':');
978 if (tp == NULL) {
979 fprintf(stderr, "line %d: syntax error\n", lineno);
980 errors++;
981 continue;
982 }
983 *tp++ = '\0', tp = skip(tp);
984 if (streq(cp, "type")) {
985 if (tp == NULL)
986 tp = "unknown";
987 cpp = dktypenames;
988 for (; cpp < &dktypenames[DKMAXTYPES]; cpp++)
989 if ((s = *cpp) && streq(s, tp)) {
990 lp->d_type = cpp - dktypenames;
991 goto next;
992 }
993 v = atoi(tp);
994 if ((unsigned)v >= DKMAXTYPES)
995 fprintf(stderr, "line %d:%s %d\n", lineno,
996 "Warning, unknown disk type", v);
997 lp->d_type = v;
998 continue;
999 }
1000 if (streq(cp, "flags")) {
1001 for (v = 0; (cp = tp) && *cp != '\0';) {
1002 tp = word(cp);
1003 if (streq(cp, "removeable"))
1004 v |= D_REMOVABLE;
1005 else if (streq(cp, "ecc"))
1006 v |= D_ECC;
1007 else if (streq(cp, "badsect"))
1008 v |= D_BADSECT;
1009 else {
1010 fprintf(stderr,
1011 "line %d: %s: bad flag\n",
1012 lineno, cp);
1013 errors++;
1014 }
1015 }
1016 lp->d_flags = v;
1017 continue;
1018 }
1019 if (streq(cp, "drivedata")) {
1020 register int i;
1021
1022 for (i = 0; (cp = tp) && *cp != '\0' && i < NDDATA;) {
1023 lp->d_drivedata[i++] = atoi(cp);
1024 tp = word(cp);
1025 }
1026 continue;
1027 }
1028 if (sscanf(cp, "%d partitions", &v) == 1) {
1029 if (v == 0 || (unsigned)v > MAXPARTITIONS) {
1030 fprintf(stderr,
1031 "line %d: bad # of partitions\n", lineno);
1032 lp->d_npartitions = MAXPARTITIONS;
1033 errors++;
1034 } else
1035 lp->d_npartitions = v;
1036 continue;
1037 }
1038 if (tp == NULL)
1039 tp = "";
1040 if (streq(cp, "disk")) {
1041 strncpy(lp->d_typename, tp, sizeof (lp->d_typename));
1042 continue;
1043 }
1044 if (streq(cp, "label")) {
1045 strncpy(lp->d_packname, tp, sizeof (lp->d_packname));
1046 continue;
1047 }
1048 if (streq(cp, "bytes/sector")) {
1049 v = atoi(tp);
1050 if (v <= 0 || (v % DEV_BSIZE) != 0) {
1051 fprintf(stderr,
1052 "line %d: %s: bad sector size\n",
1053 lineno, tp);
1054 errors++;
1055 } else
1056 lp->d_secsize = v;
1057 continue;
1058 }
1059 if (streq(cp, "sectors/track")) {
1060 v = atoi(tp);
1061 if (v <= 0) {
1062 fprintf(stderr, "line %d: %s: bad %s\n",
1063 lineno, tp, cp);
1064 errors++;
1065 } else
1066 lp->d_nsectors = v;
1067 continue;
1068 }
1069 if (streq(cp, "sectors/cylinder")) {
1070 v = atoi(tp);
1071 if (v <= 0) {
1072 fprintf(stderr, "line %d: %s: bad %s\n",
1073 lineno, tp, cp);
1074 errors++;
1075 } else
1076 lp->d_secpercyl = v;
1077 continue;
1078 }
1079 if (streq(cp, "tracks/cylinder")) {
1080 v = atoi(tp);
1081 if (v <= 0) {
1082 fprintf(stderr, "line %d: %s: bad %s\n",
1083 lineno, tp, cp);
1084 errors++;
1085 } else
1086 lp->d_ntracks = v;
1087 continue;
1088 }
1089 if (streq(cp, "cylinders")) {
1090 v = atoi(tp);
1091 if (v <= 0) {
1092 fprintf(stderr, "line %d: %s: bad %s\n",
1093 lineno, tp, cp);
1094 errors++;
1095 } else
1096 lp->d_ncylinders = v;
1097 continue;
1098 }
1099 if (streq(cp, "sectors/unit")) {
1100 v = atoi(tp);
1101 if (v <= 0) {
1102 fprintf(stderr, "line %d: %s: bad %s\n",
1103 lineno, tp, cp);
1104 errors++;
1105 } else
1106 lp->d_secperunit = v;
1107 continue;
1108 }
1109 if (streq(cp, "rpm")) {
1110 v = atoi(tp);
1111 if (v <= 0) {
1112 fprintf(stderr, "line %d: %s: bad %s\n",
1113 lineno, tp, cp);
1114 errors++;
1115 } else
1116 lp->d_rpm = v;
1117 continue;
1118 }
1119 if (streq(cp, "interleave")) {
1120 v = atoi(tp);
1121 if (v <= 0) {
1122 fprintf(stderr, "line %d: %s: bad %s\n",
1123 lineno, tp, cp);
1124 errors++;
1125 } else
1126 lp->d_interleave = v;
1127 continue;
1128 }
1129 if (streq(cp, "trackskew")) {
1130 v = atoi(tp);
1131 if (v < 0) {
1132 fprintf(stderr, "line %d: %s: bad %s\n",
1133 lineno, tp, cp);
1134 errors++;
1135 } else
1136 lp->d_trackskew = v;
1137 continue;
1138 }
1139 if (streq(cp, "cylinderskew")) {
1140 v = atoi(tp);
1141 if (v < 0) {
1142 fprintf(stderr, "line %d: %s: bad %s\n",
1143 lineno, tp, cp);
1144 errors++;
1145 } else
1146 lp->d_cylskew = v;
1147 continue;
1148 }
1149 if (streq(cp, "headswitch")) {
1150 v = atoi(tp);
1151 if (v < 0) {
1152 fprintf(stderr, "line %d: %s: bad %s\n",
1153 lineno, tp, cp);
1154 errors++;
1155 } else
1156 lp->d_headswitch = v;
1157 continue;
1158 }
1159 if (streq(cp, "track-to-track seek")) {
1160 v = atoi(tp);
1161 if (v < 0) {
1162 fprintf(stderr, "line %d: %s: bad %s\n",
1163 lineno, tp, cp);
1164 errors++;
1165 } else
1166 lp->d_trkseek = v;
1167 continue;
1168 }
|
1140 if ('a' <= *cp && *cp <= 'z' && cp[1] == '\0') {
1141 unsigned part = *cp - 'a';
1142
1143 if (part > lp->d_npartitions) {
| 1169 /* the ':' was removed above */
1170 if ('a' <= *cp && *cp <= MAX_PART && cp[1] == '\0') {
1171 part = *cp - 'a';
1172 if (part >= lp->d_npartitions) {
|
1144 fprintf(stderr,
| 1173 fprintf(stderr,
|
1145 "line %d: bad partition name\n", lineno);
| 1174 "line %d: partition name out of range a-%c: %s\n",
1175 lineno, 'a' + lp->d_npartitions - 1, cp);
|
1146 errors++;
1147 continue;
1148 }
1149 pp = &lp->d_partitions[part];
| 1176 errors++;
1177 continue;
1178 }
1179 pp = &lp->d_partitions[part];
|
| 1180 part_set[part] = 1;
|
1150#define NXTNUM(n) { \
1151 if (tp == NULL) { \
1152 fprintf(stderr, "line %d: too few numeric fields\n", lineno); \
1153 errors++; \
1154 break; \
1155 } else { \
1156 cp = tp, tp = word(cp); \
1157 if (tp == NULL) \
1158 tp = cp; \
1159 (n) = atoi(cp); \
1160 } \
1161 }
| 1181#define NXTNUM(n) { \
1182 if (tp == NULL) { \
1183 fprintf(stderr, "line %d: too few numeric fields\n", lineno); \
1184 errors++; \
1185 break; \
1186 } else { \
1187 cp = tp, tp = word(cp); \
1188 if (tp == NULL) \
1189 tp = cp; \
1190 (n) = atoi(cp); \
1191 } \
1192 }
|
1162
1163 NXTNUM(v);
1164 if (v < 0) {
| 1193/* retain 1 character following number */
1194#define NXTWORD(w,n) { \
1195 if (tp == NULL) { \
1196 fprintf(stderr, "line %d: too few numeric fields\n", lineno); \
1197 errors++; \
1198 break; \
1199 } else { \
1200 char *tmp; \
1201 cp = tp, tp = word(cp); \
1202 if (tp == NULL) \
1203 tp = cp; \
1204 (n) = strtol(cp,&tmp,10); \
1205 if (tmp) (w) = *tmp; \
1206 } \
1207 }
1208 v = 0;
1209 NXTWORD(part_size_type[part],v);
1210 if (v < 0 || (v == 0 && part_size_type[part] != '*')) {
|
1165 fprintf(stderr,
1166 "line %d: %s: bad partition size\n",
1167 lineno, cp);
1168 errors++;
| 1211 fprintf(stderr,
1212 "line %d: %s: bad partition size\n",
1213 lineno, cp);
1214 errors++;
|
1169 } else
| 1215 break;
1216 } else {
|
1170 pp->p_size = v;
| 1217 pp->p_size = v;
|
1171 NXTNUM(v);
1172 if (v < 0) {
1173 fprintf(stderr,
1174 "line %d: %s: bad partition offset\n",
1175 lineno, cp);
1176 errors++;
1177 } else
1178 pp->p_offset = v;
1179 cp = tp, tp = word(cp);
1180 cpp = fstypenames;
1181 for (; cpp < &fstypenames[FSMAXTYPES]; cpp++)
1182 if ((s = *cpp) && streq(s, cp)) {
1183 pp->p_fstype = cpp - fstypenames;
1184 goto gottype;
1185 }
1186 if (isdigit(*cp))
1187 v = atoi(cp);
1188 else
1189 v = FSMAXTYPES;
1190 if ((unsigned)v >= FSMAXTYPES) {
1191 fprintf(stderr, "line %d: %s %s\n", lineno,
1192 "Warning, unknown filesystem type", cp);
1193 v = FS_UNUSED;
1194 }
1195 pp->p_fstype = v;
1196 gottype:
| |
1197
| 1218
|
1198 switch (pp->p_fstype) {
1199
1200 case FS_UNUSED: /* XXX */
1201 NXTNUM(pp->p_fsize);
1202 if (pp->p_fsize == 0)
| 1219 v = 0;
1220 NXTWORD(part_offset_type[part],v);
1221 if (v < 0 || (v == 0 &&
1222 part_offset_type[part] != '*' &&
1223 part_offset_type[part] != '\0')) {
1224 fprintf(stderr,
1225 "line %d: %s: bad partition offset\n",
1226 lineno, cp);
1227 errors++;
|
1203 break;
| 1228 break;
|
1204 NXTNUM(v);
1205 pp->p_frag = v / pp->p_fsize;
1206 break;
| 1229 } else {
1230 pp->p_offset = v;
1231 cp = tp, tp = word(cp);
1232 cpp = fstypenames;
1233 for (; cpp < &fstypenames[FSMAXTYPES]; cpp++)
1234 if ((s = *cpp) && streq(s, cp)) {
1235 pp->p_fstype = cpp -
1236 fstypenames;
1237 goto gottype;
1238 }
1239 if (isdigit(*cp))
1240 v = atoi(cp);
1241 else
1242 v = FSMAXTYPES;
1243 if ((unsigned)v >= FSMAXTYPES) {
1244 fprintf(stderr,
1245 "line %d: Warning, unknown "
1246 "filesystem type %s\n",
1247 lineno, cp);
1248 v = FS_UNUSED;
1249 }
1250 pp->p_fstype = v;
1251 gottype:;
1252 /*
1253 * Note: NXTNUM will break us out of the
1254 * switch only!
1255 */
1256 switch (pp->p_fstype) {
1257 case FS_UNUSED:
1258 /*
1259 * allow us to accept defaults for
1260 * fsize/frag/cpg
1261 */
1262 if (tp) {
1263 NXTNUM(pp->p_fsize);
1264 if (pp->p_fsize == 0)
1265 break;
1266 NXTNUM(v);
1267 pp->p_frag = v / pp->p_fsize;
1268 }
1269 /* else default to 0's */
1270 break;
|
1207
| 1271
|
1208 case FS_BSDFFS:
1209 NXTNUM(pp->p_fsize);
1210 if (pp->p_fsize == 0)
1211 break;
1212 NXTNUM(v);
1213 pp->p_frag = v / pp->p_fsize;
1214 NXTNUM(pp->p_cpg);
1215 break;
| 1272 /* These happen to be the same */
1273 case FS_BSDFFS:
1274 case FS_BSDLFS:
1275 if (tp) {
1276 NXTNUM(pp->p_fsize);
1277 if (pp->p_fsize == 0)
1278 break;
1279 NXTNUM(v);
1280 pp->p_frag = v / pp->p_fsize;
1281 NXTNUM(pp->p_cpg);
1282 } else {
1283 /*
1284 * FIX! poor attempt at
1285 * adaptive
1286 */
1287 /* 1 GB */
1288 if (pp->p_size < 1*1024*1024*1024/lp->d_secsize) {
1289/* FIX! These are too low, but are traditional */
1290 pp->p_fsize = DEFAULT_NEWFS_BLOCK;
1291 pp->p_frag = (unsigned char) DEFAULT_NEWFS_FRAG;
1292 pp->p_cpg = DEFAULT_NEWFS_CPG;
1293 } else {
1294 pp->p_fsize = BIG_NEWFS_BLOCK;
1295 pp->p_frag = (unsigned char) BIG_NEWFS_FRAG;
1296 pp->p_cpg = BIG_NEWFS_CPG;
1297 }
1298 }
1299 break;
1300 default:
1301 break;
1302 }
|
1216
| 1303
|
1217 case FS_BSDLFS:
1218 NXTNUM(pp->p_fsize);
1219 if (pp->p_fsize == 0)
1220 break;
1221 NXTNUM(v);
1222 pp->p_frag = v / pp->p_fsize;
1223 NXTNUM(pp->p_cpg);
1224 break;
1225
1226 default:
1227 break;
| 1304 /*
1305 * note: we may not have
1306 * gotten all the entries for
1307 * the fs though if we didn't,
1308 * errors will be set.
1309 */
1310 }
|
1228 }
1229 continue;
1230 }
1231 fprintf(stderr, "line %d: %s: Unknown disklabel field\n",
1232 lineno, cp);
1233 errors++;
| 1311 }
1312 continue;
1313 }
1314 fprintf(stderr, "line %d: %s: Unknown disklabel field\n",
1315 lineno, cp);
1316 errors++;
|
1234 next:
1235 ;
| 1317 next:;
|
1236 }
1237 errors += checklabel(lp);
1238 return (errors == 0);
1239}
1240
1241/*
1242 * Check disklabel for errors and fill in
1243 * derived fields according to supplied values.
1244 */
1245int
1246checklabel(lp)
1247 register struct disklabel *lp;
1248{
1249 register struct partition *pp;
1250 int i, errors = 0;
1251 char part;
| 1318 }
1319 errors += checklabel(lp);
1320 return (errors == 0);
1321}
1322
1323/*
1324 * Check disklabel for errors and fill in
1325 * derived fields according to supplied values.
1326 */
1327int
1328checklabel(lp)
1329 register struct disklabel *lp;
1330{
1331 register struct partition *pp;
1332 int i, errors = 0;
1333 char part;
|
| 1334 unsigned long total_size,total_percent,current_offset;
1335 int seen_default_offset;
1336 int hog_part;
1337 int j;
1338 struct partition *pp2;
|
1252
1253 if (lp->d_secsize == 0) {
1254 fprintf(stderr, "sector size 0\n");
1255 return (1);
1256 }
1257 if (lp->d_nsectors == 0) {
1258 fprintf(stderr, "sectors/track 0\n");
1259 return (1);
1260 }
1261 if (lp->d_ntracks == 0) {
1262 fprintf(stderr, "tracks/cylinder 0\n");
1263 return (1);
1264 }
1265 if (lp->d_ncylinders == 0) {
1266 fprintf(stderr, "cylinders/unit 0\n");
1267 errors++;
1268 }
1269 if (lp->d_rpm == 0)
1270 Warning("revolutions/minute 0");
1271 if (lp->d_secpercyl == 0)
1272 lp->d_secpercyl = lp->d_nsectors * lp->d_ntracks;
1273 if (lp->d_secperunit == 0)
1274 lp->d_secperunit = lp->d_secpercyl * lp->d_ncylinders;
1275 if (lp->d_bbsize == 0) {
1276 fprintf(stderr, "boot block size 0\n");
1277 errors++;
1278 } else if (lp->d_bbsize % lp->d_secsize)
1279 Warning("boot block size %% sector-size != 0");
1280 if (lp->d_sbsize == 0) {
1281 fprintf(stderr, "super block size 0\n");
1282 errors++;
1283 } else if (lp->d_sbsize % lp->d_secsize)
1284 Warning("super block size %% sector-size != 0");
1285 if (lp->d_npartitions > MAXPARTITIONS)
1286 Warning("number of partitions (%lu) > MAXPARTITIONS (%d)",
1287 (u_long)lp->d_npartitions, MAXPARTITIONS);
| 1339
1340 if (lp->d_secsize == 0) {
1341 fprintf(stderr, "sector size 0\n");
1342 return (1);
1343 }
1344 if (lp->d_nsectors == 0) {
1345 fprintf(stderr, "sectors/track 0\n");
1346 return (1);
1347 }
1348 if (lp->d_ntracks == 0) {
1349 fprintf(stderr, "tracks/cylinder 0\n");
1350 return (1);
1351 }
1352 if (lp->d_ncylinders == 0) {
1353 fprintf(stderr, "cylinders/unit 0\n");
1354 errors++;
1355 }
1356 if (lp->d_rpm == 0)
1357 Warning("revolutions/minute 0");
1358 if (lp->d_secpercyl == 0)
1359 lp->d_secpercyl = lp->d_nsectors * lp->d_ntracks;
1360 if (lp->d_secperunit == 0)
1361 lp->d_secperunit = lp->d_secpercyl * lp->d_ncylinders;
1362 if (lp->d_bbsize == 0) {
1363 fprintf(stderr, "boot block size 0\n");
1364 errors++;
1365 } else if (lp->d_bbsize % lp->d_secsize)
1366 Warning("boot block size %% sector-size != 0");
1367 if (lp->d_sbsize == 0) {
1368 fprintf(stderr, "super block size 0\n");
1369 errors++;
1370 } else if (lp->d_sbsize % lp->d_secsize)
1371 Warning("super block size %% sector-size != 0");
1372 if (lp->d_npartitions > MAXPARTITIONS)
1373 Warning("number of partitions (%lu) > MAXPARTITIONS (%d)",
1374 (u_long)lp->d_npartitions, MAXPARTITIONS);
|
| 1375
1376 /* first allocate space to the partitions, then offsets */
1377 total_size = 0; /* in sectors */
1378 total_percent = 0; /* in percent */
1379 hog_part = -1;
1380 /* find all fixed partitions */
|
1288 for (i = 0; i < lp->d_npartitions; i++) {
| 1381 for (i = 0; i < lp->d_npartitions; i++) {
|
| 1382 pp = &lp->d_partitions[i];
1383 if (part_set[i]) {
1384 if (part_size_type[i] == '*') {
1385 /* partition 2 ('c') is special */
1386 if (i == FULL_DISK_PART) {
1387 pp->p_size = lp->d_secperunit;
1388 } else {
1389 if (hog_part != -1)
1390 Warning("Too many '*' partitions (%c and %c)",
1391 hog_part + 'a',i + 'a');
1392 else
1393 hog_part = i;
1394 }
1395 } else {
1396 char *type;
1397 unsigned long size;
1398
1399 size = pp->p_size;
1400 switch (part_size_type[i]) {
1401 case '%':
1402 total_percent += size;
1403 break;
1404 case 'k':
1405 case 'K':
1406 size *= 1024UL;
1407 break;
1408 case 'm':
1409 case 'M':
1410 size *= ((unsigned long) 1024*1024);
1411 break;
1412 case 'g':
1413 case 'G':
1414 size *= ((unsigned long) 1024*1024*1024);
1415 break;
1416 case '\0':
1417 break;
1418 default:
1419 Warning("unknown size specifier '%c' (K/M/G are valid)",part_size_type[i]);
1420 break;
1421 }
1422 /* don't count %'s yet */
1423 if (part_size_type[i] != '%') {
1424 /*
1425 * for all not in sectors, convert to
1426 * sectors
1427 */
1428 if (part_size_type[i] != '\0') {
1429 if (size % lp->d_secsize != 0)
1430 Warning("partition %c not an integer number of sectors",
1431 i + 'a');
1432 size /= lp->d_secsize;
1433 pp->p_size = size;
1434 }
1435 /* else already in sectors */
1436 /* partition 2 ('c') is special */
1437 if (i != FULL_DISK_PART)
1438 total_size += size;
1439 }
1440 }
1441 }
1442 }
1443 /* handle % partitions - note %'s don't need to add up to 100! */
1444 if (total_percent != 0) {
1445 long free_space = lp->d_secperunit - total_size;
1446 if (total_percent > 100) {
1447 fprintf(stderr,"total percentage %d is greater than 100\n",
1448 total_percent);
1449 errors++;
1450 }
1451
1452 if (free_space > 0) {
1453 for (i = 0; i < lp->d_npartitions; i++) {
1454 pp = &lp->d_partitions[i];
1455 if (part_set[i] && part_size_type[i] == '%') {
1456 unsigned long old_size = pp->p_size;
1457 /* careful of overflows! and integer roundoff */
1458 pp->p_size = ((double)pp->p_size/100) * free_space;
1459 total_size += pp->p_size;
1460
1461 /* FIX we can lose a sector or so due to roundoff per
1462 partition. A more complex algorithm could avoid that */
1463 }
1464 }
1465 } else {
1466 fprintf(stderr,
1467 "%ld sectors available to give to '*' and '%' partitions\n",
1468 free_space);
1469 errors++;
1470 /* fix? set all % partitions to size 0? */
1471 }
1472 }
1473 /* give anything remaining to the hog partition */
1474 if (hog_part != -1) {
1475 lp->d_partitions[hog_part].p_size = lp->d_secperunit - total_size;
1476 total_size = lp->d_secperunit;
1477 }
1478
1479 /* Now set the offsets for each partition */
1480 current_offset = 0; /* in sectors */
1481 seen_default_offset = 0;
1482 for (i = 0; i < lp->d_npartitions; i++) {
|
1289 part = 'a' + i;
1290 pp = &lp->d_partitions[i];
| 1483 part = 'a' + i;
1484 pp = &lp->d_partitions[i];
|
| 1485 if (part_set[i]) {
1486 if (part_offset_type[i] == '*') {
1487 /* partition 2 ('c') is special */
1488 if (i == FULL_DISK_PART) {
1489 pp->p_offset = 0;
1490 } else {
1491 pp->p_offset = current_offset;
1492 seen_default_offset = 1;
1493 }
1494 } else {
1495 /* allow them to be out of order for old-style tables */
1496 /* partition 2 ('c') is special */
1497 if (pp->p_offset < current_offset &&
1498 seen_default_offset && i != FULL_DISK_PART) {
1499 fprintf(stderr,
1500"Offset %ld for partition %c overlaps previous partition which ends at %ld\n",
1501 pp->p_offset,i+'a',current_offset);
1502 fprintf(stderr,
1503"Labels with any *'s for offset must be in ascending order by sector\n");
1504 errors++;
1505 } else if (pp->p_offset != current_offset &&
1506 i != FULL_DISK_PART && seen_default_offset) {
1507 /*
1508 * this may give unneeded warnings if
1509 * partitions are out-of-order
1510 */
1511 Warning(
1512"Offset %ld for partition %c doesn't match expected value %ld",
1513 pp->p_offset, i + 'a', current_offset);
1514 }
1515 }
1516 /* partition 2 ('c') is special */
1517 if (i != FULL_DISK_PART)
1518 current_offset = pp->p_offset + pp->p_size;
1519 }
1520 }
1521
1522 for (i = 0; i < lp->d_npartitions; i++) {
1523 part = 'a' + i;
1524 pp = &lp->d_partitions[i];
|
1291 if (pp->p_size == 0 && pp->p_offset != 0)
1292 Warning("partition %c: size 0, but offset %lu",
1293 part, (u_long)pp->p_offset);
1294#ifdef notdef
1295 if (pp->p_size % lp->d_secpercyl)
1296 Warning("partition %c: size %% cylinder-size != 0",
1297 part);
1298 if (pp->p_offset % lp->d_secpercyl)
1299 Warning("partition %c: offset %% cylinder-size != 0",
1300 part);
1301#endif
1302 if (pp->p_offset > lp->d_secperunit) {
1303 fprintf(stderr,
1304 "partition %c: offset past end of unit\n", part);
1305 errors++;
1306 }
1307 if (pp->p_offset + pp->p_size > lp->d_secperunit) {
1308 fprintf(stderr,
1309 "partition %c: partition extends past end of unit\n",
1310 part);
1311 errors++;
1312 }
| 1525 if (pp->p_size == 0 && pp->p_offset != 0)
1526 Warning("partition %c: size 0, but offset %lu",
1527 part, (u_long)pp->p_offset);
1528#ifdef notdef
1529 if (pp->p_size % lp->d_secpercyl)
1530 Warning("partition %c: size %% cylinder-size != 0",
1531 part);
1532 if (pp->p_offset % lp->d_secpercyl)
1533 Warning("partition %c: offset %% cylinder-size != 0",
1534 part);
1535#endif
1536 if (pp->p_offset > lp->d_secperunit) {
1537 fprintf(stderr,
1538 "partition %c: offset past end of unit\n", part);
1539 errors++;
1540 }
1541 if (pp->p_offset + pp->p_size > lp->d_secperunit) {
1542 fprintf(stderr,
1543 "partition %c: partition extends past end of unit\n",
1544 part);
1545 errors++;
1546 }
|
| 1547 if (i == FULL_DISK_PART)
1548 {
1549 if (pp->p_fstype != FS_UNUSED)
1550 Warning("partition %c is not marked as unused!",part);
1551 if (pp->p_offset != 0)
1552 Warning("partition %c doesn't start at 0!",part);
1553 if (pp->p_size != lp->d_secperunit)
1554 Warning("partition %c doesn't cover the whole unit!",part);
1555
1556 if ((pp->p_fstype != FS_UNUSED) || (pp->p_offset != 0) ||
1557 (pp->p_size != lp->d_secperunit)) {
1558 Warning("An incorrect partition %c may cause problems for "
1559 "standard system utilities",part);
1560 }
1561 }
1562
1563 /* check for overlaps */
1564 /* this will check for all possible overlaps once and only once */
1565 for (j = 0; j < i; j++) {
1566 /* partition 2 ('c') is special */
1567 if (j != FULL_DISK_PART && i != FULL_DISK_PART &&
1568 part_set[i] && part_set[j]) {
1569 pp2 = &lp->d_partitions[j];
1570 if (pp2->p_offset < pp->p_offset + pp->p_size &&
1571 (pp2->p_offset + pp2->p_size > pp->p_offset ||
1572 pp2->p_offset >= pp->p_offset)) {
1573 fprintf(stderr,"partitions %c and %c overlap!\n",
1574 j + 'a', i + 'a');
1575 errors++;
1576 }
1577 }
1578 }
|
1313 }
1314 for (; i < MAXPARTITIONS; i++) {
1315 part = 'a' + i;
1316 pp = &lp->d_partitions[i];
1317 if (pp->p_size || pp->p_offset)
1318 Warning("unused partition %c: size %d offset %lu",
1319 'a' + i, pp->p_size, (u_long)pp->p_offset);
1320 }
1321 return (errors);
1322}
1323
1324/*
1325 * When operating on a "virgin" disk, try getting an initial label
1326 * from the associated device driver. This might work for all device
1327 * drivers that are able to fetch some initial device parameters
1328 * without even having access to a (BSD) disklabel, like SCSI disks,
1329 * most IDE drives, or vn devices.
1330 *
1331 * The device name must be given in its "canonical" form.
1332 */
1333struct disklabel *
1334getvirginlabel(void)
1335{
1336 static struct disklabel lab;
1337 char namebuf[BBSIZE];
1338 int f;
1339
1340 if (dkname[0] == '/') {
1341 warnx("\"auto\" requires the usage of a canonical disk name");
1342 return (NULL);
1343 }
1344 (void)snprintf(namebuf, BBSIZE, "%s%s", _PATH_DEV, dkname);
1345 if ((f = open(namebuf, O_RDONLY)) == -1) {
1346 warn("cannot open %s", namebuf);
1347 return (NULL);
1348 }
1349
1350 /*
1351 * Try to use the new get-virgin-label ioctl. If it fails,
1352 * fallback to the old get-disdk-info ioctl.
1353 */
1354 if (ioctl(f, DIOCGDVIRGIN, &lab) < 0) {
| 1579 }
1580 for (; i < MAXPARTITIONS; i++) {
1581 part = 'a' + i;
1582 pp = &lp->d_partitions[i];
1583 if (pp->p_size || pp->p_offset)
1584 Warning("unused partition %c: size %d offset %lu",
1585 'a' + i, pp->p_size, (u_long)pp->p_offset);
1586 }
1587 return (errors);
1588}
1589
1590/*
1591 * When operating on a "virgin" disk, try getting an initial label
1592 * from the associated device driver. This might work for all device
1593 * drivers that are able to fetch some initial device parameters
1594 * without even having access to a (BSD) disklabel, like SCSI disks,
1595 * most IDE drives, or vn devices.
1596 *
1597 * The device name must be given in its "canonical" form.
1598 */
1599struct disklabel *
1600getvirginlabel(void)
1601{
1602 static struct disklabel lab;
1603 char namebuf[BBSIZE];
1604 int f;
1605
1606 if (dkname[0] == '/') {
1607 warnx("\"auto\" requires the usage of a canonical disk name");
1608 return (NULL);
1609 }
1610 (void)snprintf(namebuf, BBSIZE, "%s%s", _PATH_DEV, dkname);
1611 if ((f = open(namebuf, O_RDONLY)) == -1) {
1612 warn("cannot open %s", namebuf);
1613 return (NULL);
1614 }
1615
1616 /*
1617 * Try to use the new get-virgin-label ioctl. If it fails,
1618 * fallback to the old get-disdk-info ioctl.
1619 */
1620 if (ioctl(f, DIOCGDVIRGIN, &lab) < 0) {
|
1355 if (ioctl(f, DIOCGDINFO, &lab) < 0) {
1356 warn("ioctl DIOCGDINFO");
1357 close(f);
1358 return (NULL);
1359 }
| 1621 if (ioctl(f, DIOCGDINFO, &lab) < 0) {
1622 warn("ioctl DIOCGDINFO");
1623 close(f);
1624 return (NULL);
1625 }
|
1360 }
1361 close(f);
1362 lab.d_boot0 = NULL;
1363 lab.d_boot1 = NULL;
1364 return (&lab);
1365}
1366
1367/*
1368 * If we are installing a boot program that doesn't fit in d_bbsize
1369 * we need to mark those partitions that the boot overflows into.
1370 * This allows newfs to prevent creation of a filesystem where it might
1371 * clobber bootstrap code.
1372 */
1373void
1374setbootflag(lp)
1375 register struct disklabel *lp;
1376{
1377 register struct partition *pp;
1378 int i, errors = 0;
1379 char part;
1380 u_long boffset;
1381
1382 if (bootbuf == 0)
1383 return;
1384 boffset = bootsize / lp->d_secsize;
1385 for (i = 0; i < lp->d_npartitions; i++) {
1386 part = 'a' + i;
1387 pp = &lp->d_partitions[i];
1388 if (pp->p_size == 0)
1389 continue;
1390 if (boffset <= pp->p_offset) {
1391 if (pp->p_fstype == FS_BOOT)
1392 pp->p_fstype = FS_UNUSED;
1393 } else if (pp->p_fstype != FS_BOOT) {
1394 if (pp->p_fstype != FS_UNUSED) {
1395 fprintf(stderr,
1396 "boot overlaps used partition %c\n",
1397 part);
1398 errors++;
1399 } else {
1400 pp->p_fstype = FS_BOOT;
1401 Warning("boot overlaps partition %c, %s",
1402 part, "marked as FS_BOOT");
1403 }
1404 }
1405 }
1406 if (errors)
1407 errx(4, "cannot install boot program");
1408}
1409
1410/*VARARGS1*/
1411void
1412Warning(char *fmt, ...)
1413{
1414 va_list ap;
1415
1416 fprintf(stderr, "Warning, ");
1417 va_start(ap, fmt);
1418 vfprintf(stderr, fmt, ap);
1419 fprintf(stderr, "\n");
1420 va_end(ap);
1421}
1422
1423void
1424usage()
1425{
1426#if NUMBOOT > 0
1427 fprintf(stderr, "%s\n%s\n%s\n%s\n%s\n%s\n%s\n%s\n%s\n%s\n%s\n%s\n%s\n%s\n%s\n%s\n",
1428 "usage: disklabel [-r] disk",
1429 "\t\t(to read label)",
| 1626 }
1627 close(f);
1628 lab.d_boot0 = NULL;
1629 lab.d_boot1 = NULL;
1630 return (&lab);
1631}
1632
1633/*
1634 * If we are installing a boot program that doesn't fit in d_bbsize
1635 * we need to mark those partitions that the boot overflows into.
1636 * This allows newfs to prevent creation of a filesystem where it might
1637 * clobber bootstrap code.
1638 */
1639void
1640setbootflag(lp)
1641 register struct disklabel *lp;
1642{
1643 register struct partition *pp;
1644 int i, errors = 0;
1645 char part;
1646 u_long boffset;
1647
1648 if (bootbuf == 0)
1649 return;
1650 boffset = bootsize / lp->d_secsize;
1651 for (i = 0; i < lp->d_npartitions; i++) {
1652 part = 'a' + i;
1653 pp = &lp->d_partitions[i];
1654 if (pp->p_size == 0)
1655 continue;
1656 if (boffset <= pp->p_offset) {
1657 if (pp->p_fstype == FS_BOOT)
1658 pp->p_fstype = FS_UNUSED;
1659 } else if (pp->p_fstype != FS_BOOT) {
1660 if (pp->p_fstype != FS_UNUSED) {
1661 fprintf(stderr,
1662 "boot overlaps used partition %c\n",
1663 part);
1664 errors++;
1665 } else {
1666 pp->p_fstype = FS_BOOT;
1667 Warning("boot overlaps partition %c, %s",
1668 part, "marked as FS_BOOT");
1669 }
1670 }
1671 }
1672 if (errors)
1673 errx(4, "cannot install boot program");
1674}
1675
1676/*VARARGS1*/
1677void
1678Warning(char *fmt, ...)
1679{
1680 va_list ap;
1681
1682 fprintf(stderr, "Warning, ");
1683 va_start(ap, fmt);
1684 vfprintf(stderr, fmt, ap);
1685 fprintf(stderr, "\n");
1686 va_end(ap);
1687}
1688
1689void
1690usage()
1691{
1692#if NUMBOOT > 0
1693 fprintf(stderr, "%s\n%s\n%s\n%s\n%s\n%s\n%s\n%s\n%s\n%s\n%s\n%s\n%s\n%s\n%s\n%s\n",
1694 "usage: disklabel [-r] disk",
1695 "\t\t(to read label)",
|
1430 " disklabel -w [-r] disk type [ packid ]",
| 1696 " disklabel -w [-r] [-n] disk type [ packid ]",
|
1431 "\t\t(to write label with existing boot program)",
| 1697 "\t\t(to write label with existing boot program)",
|
1432 " disklabel -e [-r] disk",
| 1698 " disklabel -e [-r] [-n] disk",
|
1433 "\t\t(to edit label)",
| 1699 "\t\t(to edit label)",
|
1434 " disklabel -R [-r] disk protofile",
| 1700 " disklabel -R [-r] [-n] disk protofile",
|
1435 "\t\t(to restore label with existing boot program)",
1436#if NUMBOOT > 1
| 1701 "\t\t(to restore label with existing boot program)",
1702#if NUMBOOT > 1
|
1437 " disklabel -B [ -b boot1 [ -s boot2 ] ] disk [ type ]",
| 1703 " disklabel -B [-n] [ -b boot1 [ -s boot2 ] ] disk [ type ]",
|
1438 "\t\t(to install boot program with existing label)",
| 1704 "\t\t(to install boot program with existing label)",
|
1439 " disklabel -w -B [ -b boot1 [ -s boot2 ] ] disk type [ packid ]",
| 1705 " disklabel -w -B [-n] [ -b boot1 [ -s boot2 ] ] disk type [ packid ]",
|
1440 "\t\t(to write label and boot program)",
| 1706 "\t\t(to write label and boot program)",
|
1441 " disklabel -R -B [ -b boot1 [ -s boot2 ] ] disk protofile [ type ]",
| 1707 " disklabel -R -B [-n] [ -b boot1 [ -s boot2 ] ] disk protofile [ type ]",
|
1442 "\t\t(to restore label and boot program)",
1443#else
| 1708 "\t\t(to restore label and boot program)",
1709#else
|
1444 " disklabel -B [ -b bootprog ] disk [ type ]",
| 1710 " disklabel -B [-n] [ -b bootprog ] disk [ type ]",
|
1445 "\t\t(to install boot program with existing on-disk label)",
| 1711 "\t\t(to install boot program with existing on-disk label)",
|
1446 " disklabel -w -B [ -b bootprog ] disk type [ packid ]",
| 1712 " disklabel -w -B [-n] [ -b bootprog ] disk type [ packid ]",
|
1447 "\t\t(to write label and install boot program)",
| 1713 "\t\t(to write label and install boot program)",
|
1448 " disklabel -R -B [ -b bootprog ] disk protofile [ type ]",
| 1714 " disklabel -R -B [-n] [ -b bootprog ] disk protofile [ type ]",
|
1449 "\t\t(to restore label and install boot program)",
1450#endif
1451 " disklabel [-NW] disk",
1452 "\t\t(to write disable/enable label)");
1453#else
1454 fprintf(stderr, "%s\n%s\n%s\n%s\n%s\n%s\n%s\n%s\n%s\n",
1455 "usage: disklabel [-r] disk", "(to read label)",
| 1715 "\t\t(to restore label and install boot program)",
1716#endif
1717 " disklabel [-NW] disk",
1718 "\t\t(to write disable/enable label)");
1719#else
1720 fprintf(stderr, "%s\n%s\n%s\n%s\n%s\n%s\n%s\n%s\n%s\n",
1721 "usage: disklabel [-r] disk", "(to read label)",
|
1456 " disklabel -w [-r] disk type [ packid ]",
| 1722 " disklabel -w [-r] [-n] disk type [ packid ]",
|
1457 "\t\t(to write label)",
| 1723 "\t\t(to write label)",
|
1458 " disklabel -e [-r] disk",
| 1724 " disklabel -e [-r] [-n] disk",
|
1459 "\t\t(to edit label)",
| 1725 "\t\t(to edit label)",
|
1460 " disklabel -R [-r] disk protofile",
| 1726 " disklabel -R [-r] [-n] disk protofile",
|
1461 "\t\t(to restore label)",
1462 " disklabel [-NW] disk",
1463 "\t\t(to write disable/enable label)");
1464#endif
1465 exit(1);
1466}
| 1727 "\t\t(to restore label)",
1728 " disklabel [-NW] disk",
1729 "\t\t(to write disable/enable label)");
1730#endif
1731 exit(1);
1732}
|