fdisk.c revision 1.11 1 1.11 ghudson /* $NetBSD: fdisk.c,v 1.11 1995/10/04 23:11:19 ghudson Exp $ */
2 1.10 cgd
3 1.4 mycroft /*
4 1.1 cgd * Mach Operating System
5 1.1 cgd * Copyright (c) 1992 Carnegie Mellon University
6 1.1 cgd * All Rights Reserved.
7 1.4 mycroft *
8 1.1 cgd * Permission to use, copy, modify and distribute this software and its
9 1.1 cgd * documentation is hereby granted, provided that both the copyright
10 1.1 cgd * notice and this permission notice appear in all copies of the
11 1.1 cgd * software, derivative works or modified versions, and any portions
12 1.1 cgd * thereof, and that both notices appear in supporting documentation.
13 1.4 mycroft *
14 1.1 cgd * CARNEGIE MELLON ALLOWS FREE USE OF THIS SOFTWARE IN ITS "AS IS"
15 1.1 cgd * CONDITION. CARNEGIE MELLON DISCLAIMS ANY LIABILITY OF ANY KIND FOR
16 1.1 cgd * ANY DAMAGES WHATSOEVER RESULTING FROM THE USE OF THIS SOFTWARE.
17 1.4 mycroft *
18 1.1 cgd * Carnegie Mellon requests users of this software to return to
19 1.4 mycroft *
20 1.1 cgd * Software Distribution Coordinator or Software.Distribution (at) CS.CMU.EDU
21 1.1 cgd * School of Computer Science
22 1.1 cgd * Carnegie Mellon University
23 1.1 cgd * Pittsburgh PA 15213-3890
24 1.4 mycroft *
25 1.1 cgd * any improvements or extensions that they make and grant Carnegie Mellon
26 1.1 cgd * the rights to redistribute these changes.
27 1.1 cgd */
28 1.1 cgd
29 1.2 mycroft #ifndef lint
30 1.11 ghudson static char rcsid[] = "$NetBSD: fdisk.c,v 1.11 1995/10/04 23:11:19 ghudson Exp $";
31 1.2 mycroft #endif /* not lint */
32 1.2 mycroft
33 1.1 cgd #include <sys/types.h>
34 1.1 cgd #include <sys/disklabel.h>
35 1.4 mycroft #include <sys/ioctl.h>
36 1.1 cgd #include <sys/stat.h>
37 1.4 mycroft
38 1.8 cgd #include <ctype.h>
39 1.4 mycroft #include <err.h>
40 1.1 cgd #include <fcntl.h>
41 1.4 mycroft #include <stdio.h>
42 1.4 mycroft #include <stdlib.h>
43 1.8 cgd #include <string.h>
44 1.8 cgd #include <unistd.h>
45 1.1 cgd
46 1.1 cgd #define LBUF 100
47 1.1 cgd static char lbuf[LBUF];
48 1.1 cgd
49 1.1 cgd /*
50 1.1 cgd * 14-Dec-89 Robert Baron (rvb) at Carnegie-Mellon University
51 1.1 cgd * Copyright (c) 1989 Robert. V. Baron
52 1.1 cgd * Created.
53 1.1 cgd */
54 1.1 cgd
55 1.1 cgd char *disk = "/dev/rwd0d";
56 1.1 cgd
57 1.1 cgd struct disklabel disklabel; /* disk parameters */
58 1.1 cgd
59 1.5 mycroft int cylinders, sectors, heads, cylindersectors, disksectors;
60 1.1 cgd
61 1.4 mycroft struct mboot {
62 1.1 cgd unsigned char padding[2]; /* force the longs to be long alligned */
63 1.1 cgd unsigned char bootinst[DOSPARTOFF];
64 1.1 cgd struct dos_partition parts[4];
65 1.1 cgd unsigned short int signature;
66 1.1 cgd };
67 1.1 cgd struct mboot mboot;
68 1.1 cgd
69 1.1 cgd #define ACTIVE 0x80
70 1.1 cgd #define BOOT_MAGIC 0xAA55
71 1.1 cgd
72 1.5 mycroft int dos_cylinders;
73 1.1 cgd int dos_heads;
74 1.1 cgd int dos_sectors;
75 1.5 mycroft int dos_cylindersectors;
76 1.1 cgd
77 1.4 mycroft #define DOSSECT(s,c) (((s) & 0x3f) | (((c) >> 2) & 0xc0))
78 1.4 mycroft #define DOSCYL(c) ((c) & 0xff)
79 1.4 mycroft int partition = -1;
80 1.4 mycroft
81 1.4 mycroft int a_flag; /* set active partition */
82 1.4 mycroft int i_flag; /* replace partition data */
83 1.4 mycroft int u_flag; /* update partition data */
84 1.4 mycroft
85 1.4 mycroft unsigned char bootcode[] = {
86 1.4 mycroft 0x33, 0xc0, 0xfa, 0x8e, 0xd0, 0xbc, 0x00, 0x7c, 0x8e, 0xc0, 0x8e, 0xd8, 0xfb, 0x8b, 0xf4, 0xbf,
87 1.4 mycroft 0x00, 0x06, 0xb9, 0x00, 0x02, 0xfc, 0xf3, 0xa4, 0xea, 0x1d, 0x06, 0x00, 0x00, 0xb0, 0x04, 0xbe,
88 1.4 mycroft 0xbe, 0x07, 0x80, 0x3c, 0x80, 0x74, 0x0c, 0x83, 0xc6, 0x10, 0xfe, 0xc8, 0x75, 0xf4, 0xbe, 0xbd,
89 1.4 mycroft 0x06, 0xeb, 0x43, 0x8b, 0xfe, 0x8b, 0x14, 0x8b, 0x4c, 0x02, 0x83, 0xc6, 0x10, 0xfe, 0xc8, 0x74,
90 1.4 mycroft 0x0a, 0x80, 0x3c, 0x80, 0x75, 0xf4, 0xbe, 0xbd, 0x06, 0xeb, 0x2b, 0xbd, 0x05, 0x00, 0xbb, 0x00,
91 1.4 mycroft 0x7c, 0xb8, 0x01, 0x02, 0xcd, 0x13, 0x73, 0x0c, 0x33, 0xc0, 0xcd, 0x13, 0x4d, 0x75, 0xef, 0xbe,
92 1.4 mycroft 0x9e, 0x06, 0xeb, 0x12, 0x81, 0x3e, 0xfe, 0x7d, 0x55, 0xaa, 0x75, 0x07, 0x8b, 0xf7, 0xea, 0x00,
93 1.4 mycroft 0x7c, 0x00, 0x00, 0xbe, 0x85, 0x06, 0x2e, 0xac, 0x0a, 0xc0, 0x74, 0x06, 0xb4, 0x0e, 0xcd, 0x10,
94 1.1 cgd 0xeb, 0xf4, 0xfb, 0xeb, 0xfe,
95 1.1 cgd 'M', 'i', 's', 's', 'i', 'n', 'g', ' ',
96 1.1 cgd 'o', 'p', 'e', 'r', 'a', 't', 'i', 'n', 'g', ' ', 's', 'y', 's', 't', 'e', 'm', 0,
97 1.4 mycroft 'E', 'r', 'r', 'o', 'r', ' ', 'l', 'o', 'a', 'd', 'i', 'n', 'g', ' ',
98 1.1 cgd 'o', 'p', 'e', 'r', 'a', 't', 'i', 'n', 'g', ' ', 's', 'y', 's', 't', 'e', 'm', 0,
99 1.1 cgd 'I', 'n', 'v', 'a', 'l', 'i', 'd', ' ',
100 1.1 cgd 'p', 'a', 'r', 't', 'i', 't', 'i', 'o', 'n', ' ', 't', 'a', 'b', 'l', 'e', 0,
101 1.1 cgd 'A', 'u', 't', 'h', 'o', 'r', ' ', '-', ' ',
102 1.4 mycroft 'S', 'i', 'e', 'g', 'm', 'a', 'r', ' ', 'S', 'c', 'h', 'm', 'i', 'd', 't', 0,0,0,
103 1.1 cgd
104 1.4 mycroft 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
105 1.4 mycroft 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
106 1.4 mycroft 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
107 1.4 mycroft 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
108 1.4 mycroft 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
109 1.4 mycroft 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
110 1.4 mycroft 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
111 1.4 mycroft 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
112 1.4 mycroft 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
113 1.4 mycroft 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
114 1.4 mycroft 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
115 1.4 mycroft 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
116 1.4 mycroft 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0
117 1.1 cgd };
118 1.1 cgd
119 1.4 mycroft struct part_type {
121 1.4 mycroft int type;
122 1.4 mycroft char *name;
123 1.4 mycroft } part_types[] = {
124 1.4 mycroft {0x00, "unused"},
125 1.4 mycroft {0x01, "Primary DOS with 12 bit FAT"},
126 1.4 mycroft {0x02, "XENIX / filesystem"},
127 1.4 mycroft {0x03, "XENIX /usr filesystem"},
128 1.4 mycroft {0x04, "Primary DOS with 16 bit FAT"},
129 1.4 mycroft {0x05, "Extended DOS"},
130 1.4 mycroft {0x06, "Primary 'big' DOS (> 32MB)"},
131 1.4 mycroft {0x07, "OS/2 HPFS, QNX or Advanced UNIX"},
132 1.4 mycroft {0x08, "AIX filesystem"},
133 1.4 mycroft {0x09, "AIX boot partition or Coherent"},
134 1.4 mycroft {0x0A, "OS/2 Boot Manager or OPUS"},
135 1.4 mycroft {0x10, "OPUS"},
136 1.4 mycroft {0x40, "VENIX 286"},
137 1.4 mycroft {0x50, "DM"},
138 1.4 mycroft {0x51, "DM"},
139 1.4 mycroft {0x52, "CP/M or Microport SysV/AT"},
140 1.4 mycroft {0x56, "GB"},
141 1.4 mycroft {0x61, "Speed"},
142 1.4 mycroft {0x63, "ISC UNIX, other System V/386, GNU HURD or Mach"},
143 1.4 mycroft {0x64, "Novell Netware 2.xx"},
144 1.4 mycroft {0x65, "Novell Netware 3.xx"},
145 1.4 mycroft {0x75, "PCIX"},
146 1.4 mycroft {0x80, "Minix 1.1 ... 1.4a"},
147 1.9 mycroft {0x81, "Minix 1.4b ... 1.5.10"},
148 1.9 mycroft {0x82, "Linux swap"},
149 1.4 mycroft {0x83, "Linux filesystem"},
150 1.4 mycroft {0x93, "Amoeba filesystem"},
151 1.9 mycroft {0x94, "Amoeba bad block table"},
152 1.4 mycroft {0xA5, "NetBSD or 386BSD"},
153 1.4 mycroft {0xB7, "BSDI BSD/386 filesystem"},
154 1.4 mycroft {0xB8, "BSDI BSD/386 swap"},
155 1.4 mycroft {0xDB, "Concurrent CPM or C.DOS or CTOS"},
156 1.4 mycroft {0xE1, "Speed"},
157 1.4 mycroft {0xE3, "Speed"},
158 1.4 mycroft {0xE4, "Speed"},
159 1.4 mycroft {0xF1, "Speed"},
160 1.4 mycroft {0xF2, "DOS 3.3+ Secondary"},
161 1.4 mycroft {0xF4, "Speed"},
162 1.1 cgd {0xFF, "BBT (Bad Blocks Table)"},
163 1.1 cgd };
164 1.4 mycroft
165 1.4 mycroft void usage __P((void));
166 1.4 mycroft void print_s0 __P((int));
167 1.4 mycroft void print_part __P((int));
168 1.11 ghudson void init_sector0 __P((int));
169 1.11 ghudson void intuit_translated_geometry __P((void));
170 1.11 ghudson int try_heads __P((quad_t, quad_t, quad_t, quad_t, quad_t, quad_t, quad_t,
171 1.11 ghudson quad_t));
172 1.4 mycroft int try_sectors __P((quad_t, quad_t, quad_t, quad_t, quad_t));
173 1.4 mycroft void change_part __P((int));
174 1.4 mycroft void print_params __P((void));
175 1.4 mycroft void change_active __P((int));
176 1.4 mycroft void get_params_to_use __P((void));
177 1.4 mycroft void dos __P((int, unsigned char *, unsigned char *, unsigned char *));
178 1.4 mycroft int open_disk __P((int));
179 1.4 mycroft int read_disk __P((int, void *));
180 1.4 mycroft int write_disk __P((int, void *));
181 1.4 mycroft int get_params __P((void));
182 1.4 mycroft int read_s0 __P((void));
183 1.4 mycroft int write_s0 __P((void));
184 1.5 mycroft int yesno __P((char *));
185 1.4 mycroft void decimal __P((char *, int *));
186 1.4 mycroft int type_match __P((const void *, const void *));
187 1.1 cgd char *get_type __P((int));
188 1.4 mycroft
189 1.1 cgd int
190 1.4 mycroft main(argc, argv)
191 1.4 mycroft int argc;
192 1.1 cgd char *argv[];
193 1.4 mycroft {
194 1.4 mycroft int ch;
195 1.1 cgd int part;
196 1.4 mycroft
197 1.4 mycroft a_flag = i_flag = u_flag = 0;
198 1.4 mycroft while ((ch = getopt(argc, argv, "0123aiu")) != -1)
199 1.4 mycroft switch (ch) {
200 1.4 mycroft case '0':
201 1.4 mycroft partition = 0;
202 1.4 mycroft break;
203 1.4 mycroft case '1':
204 1.4 mycroft partition = 1;
205 1.4 mycroft break;
206 1.4 mycroft case '2':
207 1.4 mycroft partition = 2;
208 1.4 mycroft break;
209 1.4 mycroft case '3':
210 1.4 mycroft partition = 3;
211 1.4 mycroft break;
212 1.4 mycroft case 'a':
213 1.4 mycroft a_flag = 1;
214 1.4 mycroft break;
215 1.4 mycroft case 'i':
216 1.4 mycroft i_flag = 1;
217 1.4 mycroft case 'u':
218 1.1 cgd u_flag = 1;
219 1.4 mycroft break;
220 1.4 mycroft default:
221 1.1 cgd usage();
222 1.4 mycroft }
223 1.4 mycroft argc -= optind;
224 1.1 cgd argv += optind;
225 1.1 cgd
226 1.1 cgd if (argc > 0)
227 1.4 mycroft disk = argv[0];
228 1.4 mycroft
229 1.1 cgd if (open_disk(a_flag || i_flag || u_flag) < 0)
230 1.1 cgd exit(1);
231 1.11 ghudson
232 1.11 ghudson if (read_s0())
233 1.11 ghudson init_sector0(1);
234 1.11 ghudson
235 1.11 ghudson intuit_translated_geometry();
236 1.4 mycroft
237 1.4 mycroft printf("******* Working on device %s *******\n", disk);
238 1.1 cgd if (u_flag)
239 1.1 cgd get_params_to_use();
240 1.1 cgd else
241 1.1 cgd print_params();
242 1.1 cgd
243 1.1 cgd printf("Warning: BIOS sector numbering starts with sector 1\n");
244 1.4 mycroft printf("Information from DOS bootblock is:\n");
245 1.4 mycroft if (partition == -1) {
246 1.4 mycroft for (part = 0; part < NDOSPART; part++)
247 1.4 mycroft change_part(part);
248 1.1 cgd } else
249 1.1 cgd change_part(partition);
250 1.1 cgd
251 1.1 cgd if (u_flag || a_flag)
252 1.1 cgd change_active(partition);
253 1.1 cgd
254 1.1 cgd if (u_flag || a_flag) {
255 1.1 cgd printf("\nWe haven't changed the partition table yet. ");
256 1.1 cgd printf("This is your last chance.\n");
257 1.4 mycroft print_s0(-1);
258 1.1 cgd if (yesno("Should we write new partition table?"))
259 1.1 cgd write_s0();
260 1.1 cgd }
261 1.1 cgd
262 1.4 mycroft exit(0);
263 1.1 cgd }
264 1.4 mycroft
265 1.4 mycroft void
266 1.4 mycroft usage()
267 1.4 mycroft {
268 1.4 mycroft
269 1.4 mycroft (void)fprintf(stderr, "usage: fdisk [-aiu] [-0|-1|-2|-3] [device]\n");
270 1.1 cgd exit(1);
271 1.1 cgd }
272 1.4 mycroft
273 1.1 cgd void
274 1.4 mycroft print_s0(which)
275 1.1 cgd int which;
276 1.4 mycroft {
277 1.1 cgd int part;
278 1.1 cgd
279 1.1 cgd print_params();
280 1.4 mycroft printf("Information from DOS bootblock is:\n");
281 1.4 mycroft if (which == -1) {
282 1.4 mycroft for (part = 0; part < NDOSPART; part++)
283 1.4 mycroft printf("%d: ", part), print_part(part);
284 1.1 cgd } else
285 1.1 cgd print_part(which);
286 1.1 cgd }
287 1.1 cgd
288 1.1 cgd static struct dos_partition mtpart = { 0 };
289 1.4 mycroft
290 1.4 mycroft void
291 1.4 mycroft print_part(part)
292 1.1 cgd int part;
293 1.4 mycroft {
294 1.1 cgd struct dos_partition *partp;
295 1.4 mycroft
296 1.6 mycroft partp = &mboot.parts[part];
297 1.1 cgd if (!memcmp(partp, &mtpart, sizeof(struct dos_partition))) {
298 1.1 cgd printf("<UNUSED>\n");
299 1.1 cgd return;
300 1.5 mycroft }
301 1.4 mycroft printf("sysid %d (%s)\n", partp->dp_typ, get_type(partp->dp_typ));
302 1.4 mycroft printf(" start %d, size %d (%d MB), flag %x\n",
303 1.4 mycroft partp->dp_start, partp->dp_size,
304 1.5 mycroft partp->dp_size * 512 / (1024 * 1024), partp->dp_flag);
305 1.5 mycroft printf("\tbeg: cylinder %4d, head %3d, sector %2d\n",
306 1.5 mycroft DPCYL(partp->dp_scyl, partp->dp_ssect),
307 1.5 mycroft partp->dp_shd, DPSECT(partp->dp_ssect));
308 1.5 mycroft printf("\tend: cylinder %4d, head %3d, sector %2d\n",
309 1.5 mycroft DPCYL(partp->dp_ecyl, partp->dp_esect),
310 1.1 cgd partp->dp_ehd, DPSECT(partp->dp_esect));
311 1.1 cgd }
312 1.4 mycroft
313 1.1 cgd void
314 1.4 mycroft init_sector0(start)
315 1.1 cgd int start;
316 1.4 mycroft {
317 1.1 cgd struct dos_partition *partp;
318 1.4 mycroft
319 1.1 cgd memcpy(mboot.bootinst, bootcode, sizeof(bootcode));
320 1.1 cgd mboot.signature = BOOT_MAGIC;
321 1.4 mycroft
322 1.1 cgd partp = &mboot.parts[3];
323 1.1 cgd partp->dp_typ = DOSPTYP_386BSD;
324 1.1 cgd partp->dp_flag = ACTIVE;
325 1.5 mycroft partp->dp_start = start;
326 1.1 cgd partp->dp_size = disksectors - start;
327 1.4 mycroft
328 1.4 mycroft dos(partp->dp_start,
329 1.4 mycroft &partp->dp_scyl, &partp->dp_shd, &partp->dp_ssect);
330 1.4 mycroft dos(partp->dp_start + partp->dp_size - 1,
331 1.11 ghudson &partp->dp_ecyl, &partp->dp_ehd, &partp->dp_esect);
332 1.11 ghudson }
333 1.11 ghudson
334 1.11 ghudson /* Prerequisite: the disklabel parameters and master boot record must
335 1.11 ghudson * have been read (i.e. dos_* and mboot are meaningful).
336 1.11 ghudson * Specification: modifies dos_cylinders, dos_heads, dos_sectors, and
337 1.11 ghudson * dos_cylindersectors to be consistent with what the
338 1.11 ghudson * partition table is using, if we can find a geometry
339 1.11 ghudson * which is consistent with all partition table entries.
340 1.11 ghudson * We may get the number of cylinders slightly wrong (in
341 1.11 ghudson * the conservative direction). The idea is to be able
342 1.11 ghudson * to create a NetBSD partition on a disk we don't know
343 1.11 ghudson * the translated geometry of.
344 1.11 ghudson * This whole routine should be replaced with a kernel interface to get
345 1.11 ghudson * the BIOS geometry (which in turn requires modifications to the i386
346 1.11 ghudson * boot loader to pass in the BIOS geometry for each disk). */
347 1.11 ghudson void
348 1.11 ghudson intuit_translated_geometry()
349 1.11 ghudson {
350 1.11 ghudson int cylinders = -1, heads = -1, sectors = -1, i, j;
351 1.11 ghudson int c1, h1, s1, c2, h2, s2;
352 1.11 ghudson long a1, a2;
353 1.11 ghudson quad_t num, denom;
354 1.11 ghudson
355 1.11 ghudson /* Try to deduce the number of heads from two different mappings. */
356 1.11 ghudson for (i = 0; i < NDOSPART * 2; i++) {
357 1.11 ghudson if (get_mapping(i, &c1, &h1, &s1, &a1) < 0)
358 1.11 ghudson continue;
359 1.11 ghudson for (j = 0; j < 8; j++) {
360 1.11 ghudson if (get_mapping(j, &c2, &h2, &s2, &a2) < 0)
361 1.11 ghudson continue;
362 1.11 ghudson num = (quad_t)h1*(a2-s2) - h2*(a1-s1);
363 1.11 ghudson denom = (quad_t)c2*(a1-s1) - c1*(a2-s2);
364 1.11 ghudson if (denom != 0 && num % denom == 0) {
365 1.11 ghudson heads = num / denom;
366 1.11 ghudson break;
367 1.11 ghudson }
368 1.11 ghudson }
369 1.11 ghudson if (heads != -1)
370 1.11 ghudson break;
371 1.11 ghudson }
372 1.11 ghudson
373 1.11 ghudson if (heads == -1)
374 1.11 ghudson return;
375 1.11 ghudson
376 1.11 ghudson /* Now figure out the number of sectors from a single mapping. */
377 1.11 ghudson for (i = 0; i < NDOSPART * 2; i++) {
378 1.11 ghudson if (get_mapping(i, &c1, &h1, &s1, &a1) < 0)
379 1.11 ghudson continue;
380 1.11 ghudson num = a1 - s1;
381 1.11 ghudson denom = c1 * heads + h1;
382 1.11 ghudson if (denom != 0 && num % denom == 0) {
383 1.11 ghudson sectors = num / denom;
384 1.11 ghudson break;
385 1.11 ghudson }
386 1.11 ghudson }
387 1.11 ghudson
388 1.11 ghudson if (sectors == -1)
389 1.11 ghudson return;
390 1.11 ghudson
391 1.11 ghudson /* Estimate the number of cylinders. */
392 1.11 ghudson cylinders = dos_cylinders * dos_cylindersectors / heads / sectors;
393 1.11 ghudson
394 1.11 ghudson /* Now verify consistency with each of the partition table entries.
395 1.11 ghudson * Be willing to shove cylinders up a little bit to make things work,
396 1.11 ghudson * but translation mismatches are fatal. */
397 1.11 ghudson for (i = 0; i < NDOSPART * 2; i++) {
398 1.11 ghudson if (get_mapping(i, &c1, &h1, &s1, &a1) < 0)
399 1.11 ghudson continue;
400 1.11 ghudson if (sectors * (c1 * heads + h1) + s1 != a1)
401 1.11 ghudson return;
402 1.11 ghudson if (c1 >= cylinders)
403 1.11 ghudson cylinders = c1 + 1;
404 1.11 ghudson }
405 1.11 ghudson
406 1.11 ghudson /* Everything checks out. Reset the geometry to use for further
407 1.11 ghudson * calculations. */
408 1.11 ghudson dos_cylinders = cylinders;
409 1.11 ghudson dos_heads = heads;
410 1.11 ghudson dos_sectors = sectors;
411 1.11 ghudson dos_cylindersectors = heads * sectors;
412 1.11 ghudson }
413 1.11 ghudson
414 1.11 ghudson /* For the purposes of intuit_translated_geometry(), treat the partition
415 1.11 ghudson * table as a list of eight mapping between (cylinder, head, sector)
416 1.11 ghudson * triplets and absolute sectors. Get the relevant geometry triplet and
417 1.11 ghudson * absolute sectors for a given entry, or return -1 if it isn't present.
418 1.11 ghudson * Note: for simplicity, the returned sector is 0-based. */
419 1.11 ghudson int
420 1.11 ghudson get_mapping(i, cylinder, head, sector, absolute)
421 1.11 ghudson int i, *cylinder, *head, *sector;
422 1.11 ghudson long *absolute;
423 1.11 ghudson {
424 1.11 ghudson struct dos_partition *part = &mboot.parts[i / 2];
425 1.11 ghudson
426 1.11 ghudson if (part->dp_typ == 0)
427 1.11 ghudson return -1;
428 1.11 ghudson if (i % 2 == 0) {
429 1.11 ghudson *cylinder = DPCYL(part->dp_scyl, part->dp_ssect);
430 1.11 ghudson *head = part->dp_shd;
431 1.11 ghudson *sector = DPSECT(part->dp_ssect) - 1;
432 1.11 ghudson *absolute = part->dp_start;
433 1.11 ghudson } else {
434 1.11 ghudson *cylinder = DPCYL(part->dp_ecyl, part->dp_esect);
435 1.11 ghudson *head = part->dp_ehd;
436 1.11 ghudson *sector = DPSECT(part->dp_esect) - 1;
437 1.11 ghudson *absolute = part->dp_start + part->dp_size - 1;
438 1.11 ghudson }
439 1.1 cgd return 0;
440 1.1 cgd }
441 1.4 mycroft
442 1.4 mycroft void
443 1.4 mycroft change_part(part)
444 1.1 cgd int part;
445 1.4 mycroft {
446 1.1 cgd struct dos_partition *partp;
447 1.4 mycroft
448 1.1 cgd partp = &mboot.parts[part];
449 1.4 mycroft
450 1.4 mycroft printf("The data for partition %d is:\n", part);
451 1.4 mycroft print_part(part);
452 1.4 mycroft
453 1.4 mycroft if (!u_flag || !yesno("Do you want to change it?"))
454 1.1 cgd return;
455 1.1 cgd
456 1.7 mycroft if (i_flag) {
457 1.4 mycroft memset(partp, 0, sizeof(*partp));
458 1.1 cgd if (part == 3) {
459 1.1 cgd init_sector0(1);
460 1.4 mycroft printf("\nThe static data for the DOS partition 3 has been reinitialized to:\n");
461 1.1 cgd print_part(part);
462 1.1 cgd }
463 1.1 cgd }
464 1.1 cgd
465 1.5 mycroft do {
466 1.5 mycroft {
467 1.5 mycroft int sysid, start, size;
468 1.5 mycroft
469 1.5 mycroft sysid = partp->dp_typ,
470 1.5 mycroft start = partp->dp_start,
471 1.5 mycroft size = partp->dp_size;
472 1.5 mycroft decimal("sysid", &sysid);
473 1.5 mycroft decimal("start", &start);
474 1.5 mycroft decimal("size", &size);
475 1.5 mycroft partp->dp_typ = sysid;
476 1.5 mycroft partp->dp_start = start;
477 1.5 mycroft partp->dp_size = size;
478 1.1 cgd }
479 1.7 mycroft
480 1.5 mycroft if (yesno("Explicitly specify beg/end address?")) {
481 1.4 mycroft int tsector, tcylinder, thead;
482 1.5 mycroft
483 1.5 mycroft tcylinder = DPCYL(partp->dp_scyl, partp->dp_ssect);
484 1.5 mycroft thead = partp->dp_shd;
485 1.5 mycroft tsector = DPSECT(partp->dp_ssect);
486 1.5 mycroft decimal("beginning cylinder", &tcylinder);
487 1.5 mycroft decimal("beginning head", &thead);
488 1.5 mycroft decimal("beginning sector", &tsector);
489 1.5 mycroft partp->dp_scyl = DOSCYL(tcylinder);
490 1.5 mycroft partp->dp_shd = thead;
491 1.5 mycroft partp->dp_ssect = DOSSECT(tsector, tcylinder);
492 1.5 mycroft
493 1.5 mycroft tcylinder = DPCYL(partp->dp_ecyl, partp->dp_esect);
494 1.5 mycroft thead = partp->dp_ehd;
495 1.5 mycroft tsector = DPSECT(partp->dp_esect);
496 1.5 mycroft decimal("ending cylinder", &tcylinder);
497 1.5 mycroft decimal("ending head", &thead);
498 1.5 mycroft decimal("ending sector", &tsector);
499 1.5 mycroft partp->dp_ecyl = DOSCYL(tcylinder);
500 1.5 mycroft partp->dp_ehd = thead;
501 1.1 cgd partp->dp_esect = DOSSECT(tsector, tcylinder);
502 1.1 cgd } else {
503 1.4 mycroft dos(partp->dp_start,
504 1.4 mycroft &partp->dp_scyl, &partp->dp_shd, &partp->dp_ssect);
505 1.4 mycroft dos(partp->dp_start + partp->dp_size - 1,
506 1.1 cgd &partp->dp_ecyl, &partp->dp_ehd, &partp->dp_esect);
507 1.4 mycroft }
508 1.4 mycroft
509 1.7 mycroft print_part(part);
510 1.1 cgd } while (!yesno("Is this entry okay?"));
511 1.1 cgd }
512 1.4 mycroft
513 1.1 cgd void
514 1.1 cgd print_params()
515 1.4 mycroft {
516 1.1 cgd
517 1.5 mycroft printf("parameters extracted from in-core disklabel are:\n");
518 1.5 mycroft printf("cylinders=%d heads=%d sectors/track=%d (%d sectors/cylinder)\n\n",
519 1.5 mycroft cylinders, heads, sectors, cylindersectors);
520 1.5 mycroft if (dos_sectors > 63 || dos_cylinders > 1023 || dos_heads > 255)
521 1.1 cgd printf("Figures below won't work with BIOS for partitions not in cylinder 1\n");
522 1.5 mycroft printf("parameters to be used for BIOS calculations are:\n");
523 1.5 mycroft printf("cylinders=%d heads=%d sectors/track=%d (%d sectors/cylinder)\n\n",
524 1.1 cgd dos_cylinders, dos_heads, dos_sectors, dos_cylindersectors);
525 1.1 cgd }
526 1.4 mycroft
527 1.1 cgd void
528 1.4 mycroft change_active(which)
529 1.1 cgd int which;
530 1.4 mycroft {
531 1.4 mycroft struct dos_partition *partp;
532 1.5 mycroft int part;
533 1.4 mycroft int active = 3;
534 1.4 mycroft
535 1.1 cgd partp = &mboot.parts[0];
536 1.1 cgd
537 1.1 cgd if (a_flag && which != -1)
538 1.4 mycroft active = which;
539 1.4 mycroft else {
540 1.4 mycroft for (part = 0; part < NDOSPART; part++)
541 1.4 mycroft if (partp[part].dp_flag & ACTIVE)
542 1.4 mycroft active = part;
543 1.4 mycroft }
544 1.4 mycroft if (yesno("Do you want to change the active partition?")) {
545 1.5 mycroft do {
546 1.4 mycroft decimal("active partition", &active);
547 1.1 cgd } while (!yesno("Are you happy with this choice?"));
548 1.4 mycroft }
549 1.4 mycroft for (part = 0; part < NDOSPART; part++)
550 1.4 mycroft partp[part].dp_flag &= ~ACTIVE;
551 1.1 cgd partp[active].dp_flag |= ACTIVE;
552 1.1 cgd }
553 1.4 mycroft
554 1.1 cgd void
555 1.1 cgd get_params_to_use()
556 1.4 mycroft {
557 1.1 cgd
558 1.4 mycroft print_params();
559 1.4 mycroft if (yesno("Do you want to change our idea of what BIOS thinks?")) {
560 1.5 mycroft do {
561 1.5 mycroft decimal("BIOS's idea of #cylinders", &dos_cylinders);
562 1.5 mycroft decimal("BIOS's idea of #heads", &dos_heads);
563 1.5 mycroft decimal("BIOS's idea of #sectors", &dos_sectors);
564 1.1 cgd dos_cylindersectors = dos_heads * dos_sectors;
565 1.4 mycroft print_params();
566 1.1 cgd } while (!yesno("Are you happy with this choice?"));
567 1.1 cgd }
568 1.1 cgd }
569 1.1 cgd
570 1.1 cgd /***********************************************\
571 1.1 cgd * Change real numbers into strange dos numbers *
572 1.4 mycroft \***********************************************/
573 1.5 mycroft void
574 1.5 mycroft dos(sector, cylinderp, headp, sectorp)
575 1.5 mycroft int sector;
576 1.4 mycroft unsigned char *cylinderp, *headp, *sectorp;
577 1.5 mycroft {
578 1.4 mycroft int cylinder, head;
579 1.5 mycroft
580 1.5 mycroft cylinder = sector / dos_cylindersectors;
581 1.4 mycroft sector -= cylinder * dos_cylindersectors;
582 1.5 mycroft
583 1.5 mycroft head = sector / dos_sectors;
584 1.4 mycroft sector -= head * dos_sectors;
585 1.5 mycroft
586 1.5 mycroft *cylinderp = DOSCYL(cylinder);
587 1.5 mycroft *headp = head;
588 1.1 cgd *sectorp = DOSSECT(sector + 1, cylinder);
589 1.1 cgd }
590 1.1 cgd
591 1.1 cgd int fd;
592 1.4 mycroft
593 1.1 cgd int
594 1.4 mycroft open_disk(u_flag)
595 1.1 cgd int u_flag;
596 1.4 mycroft {
597 1.1 cgd struct stat st;
598 1.4 mycroft
599 1.4 mycroft if ((fd = open(disk, u_flag ? O_RDWR : O_RDONLY)) == -1) {
600 1.4 mycroft warn("%s", disk);
601 1.4 mycroft return (-1);
602 1.4 mycroft }
603 1.4 mycroft if (fstat(fd, &st) == -1) {
604 1.4 mycroft close(fd);
605 1.4 mycroft warn("%s", disk);
606 1.1 cgd return (-1);
607 1.7 mycroft }
608 1.4 mycroft if (!S_ISCHR(st.st_mode) && !S_ISREG(st.st_mode)) {
609 1.7 mycroft close(fd);
610 1.4 mycroft warnx("%s is not a character device or regular file", disk);
611 1.1 cgd return (-1);
612 1.4 mycroft }
613 1.4 mycroft if (get_params() == -1) {
614 1.4 mycroft close(fd);
615 1.1 cgd return (-1);
616 1.4 mycroft }
617 1.1 cgd return (0);
618 1.1 cgd }
619 1.4 mycroft
620 1.1 cgd int
621 1.4 mycroft read_disk(sector, buf)
622 1.4 mycroft int sector;
623 1.1 cgd void *buf;
624 1.4 mycroft {
625 1.4 mycroft
626 1.4 mycroft if (lseek(fd, (off_t)(sector * 512), 0) == -1)
627 1.4 mycroft return (-1);
628 1.1 cgd return (read(fd, buf, 512));
629 1.1 cgd }
630 1.4 mycroft
631 1.1 cgd int
632 1.4 mycroft write_disk(sector, buf)
633 1.4 mycroft int sector;
634 1.1 cgd void *buf;
635 1.4 mycroft {
636 1.4 mycroft
637 1.4 mycroft if (lseek(fd, (off_t)(sector * 512), 0) == -1)
638 1.4 mycroft return (-1);
639 1.1 cgd return (write(fd, buf, 512));
640 1.1 cgd }
641 1.4 mycroft
642 1.4 mycroft int
643 1.1 cgd get_params()
644 1.1 cgd {
645 1.4 mycroft
646 1.4 mycroft if (ioctl(fd, DIOCGDINFO, &disklabel) == -1) {
647 1.4 mycroft warn("DIOCGDINFO");
648 1.4 mycroft return (-1);
649 1.1 cgd }
650 1.5 mycroft
651 1.4 mycroft dos_cylinders = cylinders = disklabel.d_ncylinders;
652 1.4 mycroft dos_heads = heads = disklabel.d_ntracks;
653 1.5 mycroft dos_sectors = sectors = disklabel.d_nsectors;
654 1.5 mycroft dos_cylindersectors = cylindersectors = heads * sectors;
655 1.1 cgd disksectors = cylinders * heads * sectors;
656 1.4 mycroft
657 1.1 cgd return (0);
658 1.1 cgd }
659 1.4 mycroft
660 1.1 cgd int
661 1.1 cgd read_s0()
662 1.4 mycroft {
663 1.4 mycroft
664 1.4 mycroft if (read_disk(0, mboot.bootinst) == -1) {
665 1.4 mycroft warn("can't read fdisk partition table");
666 1.1 cgd return (-1);
667 1.1 cgd }
668 1.4 mycroft if (mboot.signature != BOOT_MAGIC) {
669 1.4 mycroft warn("invalid fdisk partition table found");
670 1.4 mycroft /* So should we initialize things? */
671 1.1 cgd return (-1);
672 1.4 mycroft }
673 1.1 cgd return (0);
674 1.1 cgd }
675 1.4 mycroft
676 1.1 cgd int
677 1.1 cgd write_s0()
678 1.4 mycroft {
679 1.4 mycroft int flag;
680 1.1 cgd
681 1.1 cgd /*
682 1.1 cgd * write enable label sector before write (if necessary),
683 1.1 cgd * disable after writing.
684 1.1 cgd * needed if the disklabel protected area also protects
685 1.1 cgd * sector 0. (e.g. empty disk)
686 1.1 cgd */
687 1.1 cgd flag = 1;
688 1.4 mycroft if (ioctl(fd, DIOCWLABEL, &flag) < 0)
689 1.4 mycroft warn("DIOCWLABEL");
690 1.4 mycroft if (write_disk(0, mboot.bootinst) == -1) {
691 1.1 cgd warn("can't write fdisk partition table");
692 1.4 mycroft return -1;
693 1.1 cgd }
694 1.4 mycroft flag = 0;
695 1.4 mycroft if (ioctl(fd, DIOCWLABEL, &flag) < 0)
696 1.1 cgd warn("DIOCWLABEL");
697 1.1 cgd }
698 1.4 mycroft
699 1.4 mycroft int
700 1.4 mycroft yesno(str)
701 1.4 mycroft char *str;
702 1.4 mycroft {
703 1.1 cgd int ch, first;
704 1.1 cgd
705 1.1 cgd printf("%s [n] ", str);
706 1.4 mycroft
707 1.4 mycroft first = ch = getchar();
708 1.4 mycroft while (ch != '\n' && ch != EOF)
709 1.4 mycroft ch = getchar();
710 1.1 cgd return (first == 'y' || first == 'Y');
711 1.1 cgd }
712 1.5 mycroft
713 1.5 mycroft void
714 1.4 mycroft decimal(str, num)
715 1.5 mycroft char *str;
716 1.1 cgd int *num;
717 1.8 cgd {
718 1.4 mycroft int acc = 0;
719 1.1 cgd char *cp;
720 1.5 mycroft
721 1.5 mycroft for (;; printf("%s is not a valid decimal number.\n", lbuf)) {
722 1.1 cgd printf("Supply a decimal value for \"%s\" [%d] ", str, *num);
723 1.1 cgd
724 1.7 mycroft fgets(lbuf, LBUF, stdin);
725 1.7 mycroft lbuf[strlen(lbuf)-1] = '\0';
726 1.7 mycroft cp = lbuf;
727 1.7 mycroft
728 1.7 mycroft cp += strspn(cp, " \t");
729 1.5 mycroft if (*cp == '\0')
730 1.1 cgd return;
731 1.7 mycroft
732 1.7 mycroft if (!isdigit(*cp))
733 1.5 mycroft continue;
734 1.7 mycroft acc = strtol(lbuf, &cp, 10);
735 1.5 mycroft
736 1.5 mycroft cp += strspn(cp, " \t");
737 1.5 mycroft if (*cp != '\0')
738 1.1 cgd continue;
739 1.5 mycroft
740 1.5 mycroft *num = acc;
741 1.1 cgd return;
742 1.1 cgd }
743 1.1 cgd
744 1.1 cgd }
745 1.4 mycroft
746 1.4 mycroft int
747 1.4 mycroft type_match(key, item)
748 1.4 mycroft const void *key, *item;
749 1.4 mycroft {
750 1.4 mycroft const int *typep = key;
751 1.4 mycroft const struct part_type *ptr = item;
752 1.4 mycroft
753 1.4 mycroft if (*typep < ptr->type)
754 1.4 mycroft return (-1);
755 1.4 mycroft if (*typep > ptr->type)
756 1.4 mycroft return (1);
757 1.4 mycroft return (0);
758 1.4 mycroft }
759 1.4 mycroft
760 1.4 mycroft char *
761 1.4 mycroft get_type(type)
762 1.4 mycroft int type;
763 1.4 mycroft {
764 1.4 mycroft struct part_type *ptr;
765 1.4 mycroft
766 1.4 mycroft ptr = bsearch(&type, part_types,
767 1.4 mycroft sizeof(part_types) / sizeof(struct part_type),
768 1.4 mycroft sizeof(struct part_type), type_match);
769 1.4 mycroft if (ptr == 0)
770 1.4 mycroft return ("unknown");
771 1.4 mycroft else
772 1.1 cgd return (ptr->name);
773 }
774