xref: /illumos-gate/usr/src/cmd/format/menu_fdisk.c (revision d8a7fe16f62711cdc5c4267da8b34ff24a6b668c)
1 /*
2  * CDDL HEADER START
3  *
4  * The contents of this file are subject to the terms of the
5  * Common Development and Distribution License (the "License").
6  * You may not use this file except in compliance with the License.
7  *
8  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9  * or http://www.opensolaris.org/os/licensing.
10  * See the License for the specific language governing permissions
11  * and limitations under the License.
12  *
13  * When distributing Covered Code, include this CDDL HEADER in each
14  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15  * If applicable, add the following below this CDDL HEADER, with the
16  * fields enclosed by brackets "[]" replaced with your own identifying
17  * information: Portions Copyright [yyyy] [name of copyright owner]
18  *
19  * CDDL HEADER END
20  */
21 /*
22  * Copyright 2009 Sun Microsystems, Inc.  All rights reserved.
23  * Use is subject to license terms.
24  */
25 
26 /*
27  * This file contains functions that implement the fdisk menu commands.
28  */
29 #include "global.h"
30 #include <errno.h>
31 #include <sys/time.h>
32 #include <sys/resource.h>
33 #include <sys/wait.h>
34 #include <signal.h>
35 #include <string.h>
36 #include <fcntl.h>
37 #include <stdlib.h>
38 #include <sys/dktp/fdisk.h>
39 #include <sys/stat.h>
40 #include <sys/dklabel.h>
41 #ifdef i386
42 #include <libfdisk.h>
43 #endif
44 
45 #include "main.h"
46 #include "analyze.h"
47 #include "menu.h"
48 #include "menu_command.h"
49 #include "menu_defect.h"
50 #include "menu_partition.h"
51 #include "menu_fdisk.h"
52 #include "param.h"
53 #include "misc.h"
54 #include "label.h"
55 #include "startup.h"
56 #include "partition.h"
57 #include "prompts.h"
58 #include "checkdev.h"
59 #include "io.h"
60 #include "ctlr_scsi.h"
61 #include "auto_sense.h"
62 
63 extern	struct menu_item menu_fdisk[];
64 
65 /*
66  * Byte swapping macros for accessing struct ipart
67  *	to resolve little endian on Sparc.
68  */
69 #if defined(sparc)
70 #define	les(val)	((((val)&0xFF)<<8)|(((val)>>8)&0xFF))
71 #define	lel(val)	(((unsigned)(les((val)&0x0000FFFF))<<16) | \
72 			(les((unsigned)((val)&0xffff0000)>>16)))
73 
74 #elif	defined(i386)
75 
76 #define	les(val)	(val)
77 #define	lel(val)	(val)
78 
79 #else	/* defined(sparc) */
80 
81 #error	No Platform defined
82 
83 #endif	/* defined(sparc) */
84 
85 
86 /* Function prototypes */
87 #ifdef	__STDC__
88 
89 #if	defined(sparc)
90 
91 static int getbyte(uchar_t **);
92 static int getlong(uchar_t **);
93 
94 #endif	/* defined(sparc) */
95 
96 static int get_solaris_part(int fd, struct ipart *ipart);
97 
98 #else	/* __STDC__ */
99 
100 #if	defined(sparc)
101 
102 static int getbyte();
103 static int getlong();
104 
105 #endif	/* defined(sparc) */
106 
107 static int get_solaris_part();
108 
109 #endif	/* __STDC__ */
110 
111 #ifdef i386
112 int extpart_init(ext_part_t **epp);
113 #endif
114 /*
115  * Handling the alignment problem of struct ipart.
116  */
117 static void
118 fill_ipart(char *bootptr, struct ipart *partp)
119 {
120 #if defined(sparc)
121 	/*
122 	 * Sparc platform:
123 	 *
124 	 * Packing short/word for struct ipart to resolve
125 	 *	little endian on Sparc since it is not
126 	 *	properly aligned on Sparc.
127 	 */
128 	partp->bootid = getbyte((uchar_t **)&bootptr);
129 	partp->beghead = getbyte((uchar_t **)&bootptr);
130 	partp->begsect = getbyte((uchar_t **)&bootptr);
131 	partp->begcyl = getbyte((uchar_t **)&bootptr);
132 	partp->systid = getbyte((uchar_t **)&bootptr);
133 	partp->endhead = getbyte((uchar_t **)&bootptr);
134 	partp->endsect = getbyte((uchar_t **)&bootptr);
135 	partp->endcyl = getbyte((uchar_t **)&bootptr);
136 	partp->relsect = getlong((uchar_t **)&bootptr);
137 	partp->numsect = getlong((uchar_t **)&bootptr);
138 #elif defined(i386)
139 	/*
140 	 * i386 platform:
141 	 *
142 	 * The fdisk table does not begin on a 4-byte boundary within
143 	 * the master boot record; so, we need to recopy its contents
144 	 * to another data structure to avoid an alignment exception.
145 	 */
146 	(void) bcopy(bootptr, partp, sizeof (struct ipart));
147 #else
148 #error  No Platform defined
149 #endif /* defined(sparc) */
150 }
151 
152 /*
153  * Get a correct byte/short/word routines for Sparc platform.
154  */
155 #if defined(sparc)
156 static int
157 getbyte(uchar_t **bp)
158 {
159 	int	b;
160 
161 	b = **bp;
162 	*bp = *bp + 1;
163 	return (b);
164 }
165 
166 #ifdef DEADCODE
167 static int
168 getshort(uchar_t **bp)
169 {
170 	int	b;
171 
172 	b = ((**bp) << 8) | *(*bp + 1);
173 	*bp += 2;
174 	return (b);
175 }
176 #endif /* DEADCODE */
177 
178 static int
179 getlong(uchar_t **bp)
180 {
181 	int	b, bh, bl;
182 
183 	bh = ((**bp) << 8) | *(*bp + 1);
184 	*bp += 2;
185 	bl = ((**bp) << 8) | *(*bp + 1);
186 	*bp += 2;
187 
188 	b = (bh << 16) | bl;
189 	return (b);
190 }
191 #endif /* defined(sparc) */
192 
193 /*
194  * Convert cn[tn]dn to cn[tn]dns2 path
195  */
196 static void
197 get_sname(char *name)
198 {
199 	char		buf[MAXPATHLEN];
200 	char		*devp = "/dev/dsk";
201 	char		*rdevp = "/dev/rdsk";
202 	char		np[MAXNAMELEN];
203 	char		*npt;
204 
205 	/*
206 	 * If it is a full path /dev/[r]dsk/cn[tn]dn, use this path
207 	 */
208 	(void) strcpy(np, cur_disk->disk_name);
209 	if (strncmp(rdevp, cur_disk->disk_name, strlen(rdevp)) == 0 ||
210 	    strncmp(devp, cur_disk->disk_name, strlen(devp)) == 0) {
211 		/*
212 		 * Skip if the path is already included with sN
213 		 */
214 		if (strchr(np, 's') == strrchr(np, 's')) {
215 			npt = strrchr(np, 'p');
216 			/* If pN is found, do not include it */
217 			if (npt != NULL) {
218 				*npt = '\0';
219 			}
220 			(void) snprintf(buf, sizeof (buf), "%ss2", np);
221 		} else {
222 			(void) snprintf(buf, sizeof (buf), "%s", np);
223 		}
224 	} else {
225 		(void) snprintf(buf, sizeof (buf), "/dev/rdsk/%ss2", np);
226 	}
227 	(void) strcpy(name, buf);
228 }
229 
230 /*
231  * Convert cn[tn]dnsn to cn[tn]dnp0 path
232  */
233 static void
234 get_pname(char *name)
235 {
236 	char		buf[MAXPATHLEN];
237 	char		*devp = "/dev/dsk";
238 	char		*rdevp = "/dev/rdsk";
239 	char		np[MAXNAMELEN];
240 	char		*npt;
241 
242 	/*
243 	 * If it is a full path /dev/[r]dsk/cn[tn]dnsn, use this path
244 	 */
245 	if (cur_disk == NULL) {
246 		(void) strcpy(np, x86_devname);
247 	} else {
248 		(void) strcpy(np, cur_disk->disk_name);
249 	}
250 
251 	if (strncmp(rdevp, np, strlen(rdevp)) == 0 ||
252 	    strncmp(devp, np, strlen(devp)) == 0) {
253 		/*
254 		 * Skip if the path is already included with pN
255 		 */
256 		if (strchr(np, 'p') == NULL) {
257 			npt = strrchr(np, 's');
258 			/* If sN is found, do not include it */
259 			if (isdigit(*++npt)) {
260 				*--npt = '\0';
261 			}
262 			(void) snprintf(buf, sizeof (buf), "%sp0", np);
263 		} else {
264 			(void) snprintf(buf, sizeof (buf), "%s", np);
265 		}
266 	} else {
267 		(void) snprintf(buf, sizeof (buf), "/dev/rdsk/%sp0", np);
268 	}
269 	(void) strcpy(name, buf);
270 }
271 
272 /*
273  * Open file descriptor for current disk (cur_file)
274  *	with "p0" path or cur_disk->disk_path
275  */
276 void
277 open_cur_file(int mode)
278 {
279 	char	*dkpath;
280 	char	pbuf[MAXPATHLEN];
281 
282 	switch (mode) {
283 		case FD_USE_P0_PATH:
284 			(void) get_pname(&pbuf[0]);
285 			dkpath = pbuf;
286 			break;
287 		case FD_USE_CUR_DISK_PATH:
288 			if (cur_disk->fdisk_part.systid == SUNIXOS ||
289 			    cur_disk->fdisk_part.systid == SUNIXOS2) {
290 				(void) get_sname(&pbuf[0]);
291 				dkpath = pbuf;
292 			} else {
293 				dkpath = cur_disk->disk_path;
294 			}
295 			break;
296 		default:
297 			err_print("Error: Invalid mode option for opening "
298 			    "cur_file\n");
299 			fullabort();
300 	}
301 
302 	/* Close previous cur_file */
303 	(void) close(cur_file);
304 	/* Open cur_file with the required path dkpath */
305 	if ((cur_file = open_disk(dkpath, O_RDWR | O_NDELAY)) < 0) {
306 		err_print(
307 		    "Error: can't open selected disk '%s'.\n", dkpath);
308 		fullabort();
309 	}
310 }
311 
312 
313 /*
314  * This routine implements the 'fdisk' command.  It simply runs
315  * the fdisk command on the current disk.
316  * Use of this is restricted to interactive mode only.
317  */
318 int
319 c_fdisk()
320 {
321 
322 	char		buf[MAXPATHLEN];
323 	char		pbuf[MAXPATHLEN];
324 	struct stat	statbuf;
325 
326 	/*
327 	 * We must be in interactive mode to use the fdisk command
328 	 */
329 	if (option_f != (char *)NULL || isatty(0) != 1 || isatty(1) != 1) {
330 		err_print("Fdisk command is for interactive use only!\n");
331 		return (-1);
332 	}
333 
334 	/*
335 	 * There must be a current disk type and a current disk
336 	 */
337 	if (cur_dtype == NULL) {
338 		err_print("Current Disk Type is not set.\n");
339 		return (-1);
340 	}
341 
342 	/*
343 	 * Before running the fdisk command, get file status of
344 	 *	/dev/rdsk/cn[tn]dnp0 path to see if this disk
345 	 *	supports fixed disk partition table.
346 	 */
347 	(void) get_pname(&pbuf[0]);
348 	if (stat(pbuf, (struct stat *)&statbuf) == -1 ||
349 	    !S_ISCHR(statbuf.st_mode)) {
350 		err_print(
351 		"Disk does not support fixed disk partition table\n");
352 		return (0);
353 	}
354 
355 	/*
356 	 * Run the fdisk program.
357 	 */
358 	(void) snprintf(buf, sizeof (buf), "fdisk %s\n", pbuf);
359 	(void) system(buf);
360 
361 	/*
362 	 * Open cur_file with "p0" path for accessing the fdisk table
363 	 */
364 	(void) open_cur_file(FD_USE_P0_PATH);
365 
366 	/*
367 	 * Get solaris partition information in the fdisk partition table
368 	 */
369 	if (get_solaris_part(cur_file, &cur_disk->fdisk_part) == -1) {
370 		err_print("No fdisk solaris partition found\n");
371 		cur_disk->fdisk_part.numsect = 0;  /* No Solaris */
372 	}
373 
374 	/*
375 	 * Restore cur_file with cur_disk->disk_path
376 	 */
377 	(void) open_cur_file(FD_USE_CUR_DISK_PATH);
378 
379 	return (0);
380 }
381 
382 /*
383  * Read MBR on the disk
384  * if the Solaris partition has changed,
385  *	reread the vtoc
386  */
387 #ifdef DEADCODE
388 static void
389 update_cur_parts()
390 {
391 
392 	int i;
393 	register struct partition_info *parts;
394 
395 	for (i = 0; i < NDKMAP; i++) {
396 #if defined(_SUNOS_VTOC_16)
397 		if (cur_parts->vtoc.v_part[i].p_tag &&
398 		    cur_parts->vtoc.v_part[i].p_tag != V_ALTSCTR) {
399 			cur_parts->vtoc.v_part[i].p_start = 0;
400 			cur_parts->vtoc.v_part[i].p_size = 0;
401 
402 #endif
403 			cur_parts->pinfo_map[i].dkl_nblk = 0;
404 			cur_parts->pinfo_map[i].dkl_cylno = 0;
405 			cur_parts->vtoc.v_part[i].p_tag =
406 			    default_vtoc_map[i].p_tag;
407 			cur_parts->vtoc.v_part[i].p_flag =
408 			    default_vtoc_map[i].p_flag;
409 #if defined(_SUNOS_VTOC_16)
410 		}
411 #endif
412 	}
413 	cur_parts->pinfo_map[C_PARTITION].dkl_nblk = ncyl * spc();
414 
415 #if defined(_SUNOS_VTOC_16)
416 	/*
417 	 * Adjust for the boot partitions
418 	 */
419 	cur_parts->pinfo_map[I_PARTITION].dkl_nblk = spc();
420 	cur_parts->pinfo_map[I_PARTITION].dkl_cylno = 0;
421 	cur_parts->vtoc.v_part[C_PARTITION].p_start =
422 	    cur_parts->pinfo_map[C_PARTITION].dkl_cylno * nhead * nsect;
423 	cur_parts->vtoc.v_part[C_PARTITION].p_size =
424 	    cur_parts->pinfo_map[C_PARTITION].dkl_nblk;
425 
426 	cur_parts->vtoc.v_part[I_PARTITION].p_start =
427 	    cur_parts->pinfo_map[I_PARTITION].dkl_cylno;
428 	cur_parts->vtoc.v_part[I_PARTITION].p_size =
429 	    cur_parts->pinfo_map[I_PARTITION].dkl_nblk;
430 
431 #endif	/* defined(_SUNOS_VTOC_16) */
432 	parts = cur_dtype->dtype_plist;
433 	cur_dtype->dtype_ncyl = ncyl;
434 	cur_dtype->dtype_plist = cur_parts;
435 	parts->pinfo_name = cur_parts->pinfo_name;
436 	cur_disk->disk_parts = cur_parts;
437 	cur_ctype->ctype_dlist = cur_dtype;
438 
439 }
440 #endif /* DEADCODE */
441 
442 static int
443 get_solaris_part(int fd, struct ipart *ipart)
444 {
445 	int		i;
446 	struct ipart	ip;
447 	int		status;
448 	char		*mbr;
449 	char		*bootptr;
450 	struct dk_label	update_label;
451 	ushort_t	found = 0;
452 #ifdef i386
453 	uint32_t	relsec, numsec;
454 	int		pno, rval, ext_part_found = 0;
455 	ext_part_t	*epp;
456 #endif
457 
458 	(void) lseek(fd, 0, 0);
459 
460 	/*
461 	 * We may get mbr of different size, but the first 512 bytes
462 	 * are valid information.
463 	 */
464 	mbr = malloc(cur_blksz);
465 	if (mbr == NULL) {
466 		err_print("No memory available.\n");
467 		return (-1);
468 	}
469 	status = read(fd, mbr, cur_blksz);
470 
471 	if (status != cur_blksz) {
472 		err_print("Bad read of fdisk partition. Status = %x\n", status);
473 		err_print("Cannot read fdisk partition information.\n");
474 		free(mbr);
475 		return (-1);
476 	}
477 
478 	(void) memcpy(&boot_sec, mbr, sizeof (struct mboot));
479 	free(mbr);
480 
481 #ifdef i386
482 	(void) extpart_init(&epp);
483 #endif
484 	for (i = 0; i < FD_NUMPART; i++) {
485 		int	ipc;
486 
487 		ipc = i * sizeof (struct ipart);
488 
489 		/* Handling the alignment problem of struct ipart */
490 		bootptr = &boot_sec.parts[ipc];
491 		(void) fill_ipart(bootptr, &ip);
492 
493 #ifdef i386
494 		if (fdisk_is_dos_extended(ip.systid) && (ext_part_found == 0)) {
495 			/* We support only one extended partition per disk */
496 			ext_part_found = 1;
497 			rval = fdisk_get_solaris_part(epp, &pno, &relsec,
498 			    &numsec);
499 			if (rval == FDISK_SUCCESS) {
500 				/*
501 				 * Found a solaris partition inside the
502 				 * extended partition. Update the statistics.
503 				 */
504 				if (nhead != 0 && nsect != 0) {
505 					pcyl = numsec / (nhead * nsect);
506 					xstart = relsec / (nhead * nsect);
507 					ncyl = pcyl - acyl;
508 				}
509 				solaris_offset = relsec;
510 				found = 2;
511 				ip.bootid = 0;
512 				ip.beghead = ip.begsect = ip.begcyl = 0xff;
513 				ip.endhead = ip.endsect = ip.endcyl = 0xff;
514 				ip.systid = SUNIXOS2;
515 				ip.relsect = relsec;
516 				ip.numsect = numsec;
517 				ipart->bootid = ip.bootid;
518 				status = bcmp(&ip, ipart,
519 				    sizeof (struct ipart));
520 				bcopy(&ip, ipart, sizeof (struct ipart));
521 			}
522 			continue;
523 		}
524 #endif
525 
526 		/*
527 		 * we are interested in Solaris and EFI partition types
528 		 */
529 #ifdef i386
530 		if ((ip.systid == SUNIXOS &&
531 		    (fdisk_is_linux_swap(epp, lel(ip.relsect), NULL) != 0)) ||
532 		    ip.systid == SUNIXOS2 ||
533 		    ip.systid == EFI_PMBR) {
534 #else
535 		if (ip.systid == SUNIXOS ||
536 		    ip.systid == SUNIXOS2 ||
537 		    ip.systid == EFI_PMBR) {
538 #endif
539 			/*
540 			 * if the disk has an EFI label, nhead and nsect may
541 			 * be zero.  This test protects us from FPE's, and
542 			 * format still seems to work fine
543 			 */
544 			if (nhead != 0 && nsect != 0) {
545 				pcyl = lel(ip.numsect) / (nhead * nsect);
546 				xstart = lel(ip.relsect) / (nhead * nsect);
547 				ncyl = pcyl - acyl;
548 			}
549 #ifdef DEBUG
550 			else {
551 				err_print("Critical geometry values are zero:\n"
552 				    "\tnhead = %d; nsect = %d\n", nhead, nsect);
553 			}
554 #endif /* DEBUG */
555 
556 			solaris_offset = (uint_t)lel(ip.relsect);
557 			found = 1;
558 			break;
559 		}
560 	}
561 
562 #ifdef i386
563 	libfdisk_fini(&epp);
564 #endif
565 
566 	if (!found) {
567 		err_print("Solaris fdisk partition not found\n");
568 		return (-1);
569 	} else if (found == 1) {
570 		/*
571 		 * Found a primary solaris partition.
572 		 * compare the previous and current Solaris partition
573 		 * but don't use bootid in determination of Solaris partition
574 		 * changes
575 		 */
576 		ipart->bootid = ip.bootid;
577 		status = bcmp(&ip, ipart, sizeof (struct ipart));
578 
579 		bcopy(&ip, ipart, sizeof (struct ipart));
580 	}
581 
582 	/* if the disk partitioning has changed - get the VTOC */
583 	if (status) {
584 		struct extvtoc exvtoc;
585 		struct vtoc vtoc;
586 
587 		status = ioctl(fd, DKIOCGEXTVTOC, &exvtoc);
588 		if (status == -1) {
589 			i = errno;
590 			/* Try the old ioctl DKIOCGVTOC */
591 			status = ioctl(fd, DKIOCGVTOC, &vtoc);
592 			if (status == -1) {
593 				err_print("Bad ioctl DKIOCGEXTVTOC.\n");
594 				err_print("errno=%d %s\n", i, strerror(i));
595 				err_print("Cannot read vtoc information.\n");
596 				return (-1);
597 			}
598 		}
599 
600 		status = read_label(fd, &update_label);
601 		if (status == -1) {
602 			err_print("Cannot read label information.\n");
603 			return (-1);
604 		}
605 
606 		/* copy vtoc information */
607 		cur_parts->vtoc = update_label.dkl_vtoc;
608 
609 #if defined(_SUNOS_VTOC_16)
610 		/*
611 		 * this is to update the slice table on x86
612 		 * we don't care about VTOC8 here
613 		 */
614 		for (i = 0; i < NDKMAP; i ++) {
615 			cur_parts->pinfo_map[i].dkl_cylno =
616 			    update_label.dkl_vtoc.v_part[i].p_start /
617 			    ((int)(update_label.dkl_nhead *
618 			    update_label.dkl_nsect));
619 			cur_parts->pinfo_map[i].dkl_nblk =
620 			    update_label.dkl_vtoc.v_part[i].p_size;
621 		}
622 #endif /* defined(_SUNOS_VTOC_16) */
623 
624 		cur_dtype->dtype_ncyl = update_label.dkl_ncyl;
625 		cur_dtype->dtype_pcyl = update_label.dkl_pcyl;
626 		cur_dtype->dtype_acyl = update_label.dkl_acyl;
627 		cur_dtype->dtype_nhead = update_label.dkl_nhead;
628 		cur_dtype->dtype_nsect = update_label.dkl_nsect;
629 		ncyl = cur_dtype->dtype_ncyl;
630 		acyl = cur_dtype->dtype_acyl;
631 		pcyl = cur_dtype->dtype_pcyl;
632 		nsect = cur_dtype->dtype_nsect;
633 		nhead = cur_dtype->dtype_nhead;
634 	}
635 	return (0);
636 }
637 
638 
639 int
640 copy_solaris_part(struct ipart *ipart)
641 {
642 
643 	int		status, i, fd;
644 	struct mboot	mboot;
645 	char		*mbr;
646 	struct ipart	ip;
647 	char		buf[MAXPATHLEN];
648 	char		*bootptr;
649 	struct stat	statbuf;
650 #ifdef i386
651 	uint32_t	relsec, numsec;
652 	int		pno, rval, ext_part_found = 0;
653 	ext_part_t	*epp;
654 #endif
655 
656 	(void) get_pname(&buf[0]);
657 	if (stat(buf, &statbuf) == -1 ||
658 	    !S_ISCHR(statbuf.st_mode) ||
659 	    ((cur_label == L_TYPE_EFI) &&
660 	    (cur_disk->disk_flags & DSK_LABEL_DIRTY))) {
661 		/*
662 		 * Make sure to reset solaris_offset to zero if it is
663 		 *	previously set by a selected disk that
664 		 *	supports the fdisk table.
665 		 */
666 		solaris_offset = 0;
667 		/*
668 		 * Return if this disk does not support fdisk table or
669 		 * if it uses an EFI label but has not yet been labelled.
670 		 * If the EFI label has not been written then the open
671 		 * on the partition will fail.
672 		 */
673 		return (0);
674 	}
675 
676 	if ((fd = open(buf, O_RDONLY)) < 0) {
677 		err_print("Error: can't open disk '%s'.\n", buf);
678 		return (-1);
679 	}
680 
681 	/*
682 	 * We may get mbr of different size, but the first 512 bytes
683 	 * are valid information.
684 	 */
685 	mbr = malloc(cur_blksz);
686 	if (mbr == NULL) {
687 		err_print("No memory available.\n");
688 		return (-1);
689 	}
690 	status = read(fd, mbr, cur_blksz);
691 
692 	if (status != cur_blksz) {
693 		err_print("Bad read of fdisk partition.\n");
694 		(void) close(fd);
695 		free(mbr);
696 		return (-1);
697 	}
698 
699 	(void) memcpy(&mboot, mbr, sizeof (struct mboot));
700 
701 #ifdef i386
702 	(void) extpart_init(&epp);
703 #endif
704 	for (i = 0; i < FD_NUMPART; i++) {
705 		int	ipc;
706 
707 		ipc = i * sizeof (struct ipart);
708 
709 		/* Handling the alignment problem of struct ipart */
710 		bootptr = &mboot.parts[ipc];
711 		(void) fill_ipart(bootptr, &ip);
712 
713 #ifdef i386
714 		if (fdisk_is_dos_extended(ip.systid) && (ext_part_found == 0)) {
715 			/* We support only one extended partition per disk */
716 			ext_part_found = 1;
717 			rval = fdisk_get_solaris_part(epp, &pno, &relsec,
718 			    &numsec);
719 			if (rval == FDISK_SUCCESS) {
720 				/*
721 				 * Found a solaris partition inside the
722 				 * extended partition. Update the statistics.
723 				 */
724 				if (nhead != 0 && nsect != 0) {
725 					pcyl = numsec / (nhead * nsect);
726 					ncyl = pcyl - acyl;
727 				}
728 				solaris_offset = relsec;
729 				ip.bootid = 0;
730 				ip.beghead = ip.begsect = ip.begcyl = 0xff;
731 				ip.endhead = ip.endsect = ip.endcyl = 0xff;
732 				ip.systid = SUNIXOS2;
733 				ip.relsect = relsec;
734 				ip.numsect = numsec;
735 				bcopy(&ip, ipart, sizeof (struct ipart));
736 			}
737 			continue;
738 		}
739 #endif
740 
741 
742 #ifdef i386
743 		if ((ip.systid == SUNIXOS &&
744 		    (fdisk_is_linux_swap(epp, lel(ip.relsect), NULL) != 0)) ||
745 		    ip.systid == SUNIXOS2 ||
746 		    ip.systid == EFI_PMBR) {
747 #else
748 		if (ip.systid == SUNIXOS ||
749 		    ip.systid == SUNIXOS2 ||
750 		    ip.systid == EFI_PMBR) {
751 #endif
752 			solaris_offset = lel(ip.relsect);
753 			bcopy(&ip, ipart, sizeof (struct ipart));
754 
755 			/*
756 			 * if the disk has an EFI label, we typically won't
757 			 * have values for nhead and nsect.  format seems to
758 			 * work without them, and we need to protect ourselves
759 			 * from FPE's
760 			 */
761 			if (nhead != 0 && nsect != 0) {
762 				pcyl = lel(ip.numsect) / (nhead * nsect);
763 				ncyl = pcyl - acyl;
764 			}
765 #ifdef DEBUG
766 			else {
767 				err_print("Critical geometry values are zero:\n"
768 				    "\tnhead = %d; nsect = %d\n", nhead, nsect);
769 			}
770 #endif /* DEBUG */
771 
772 			break;
773 		}
774 	}
775 #ifdef i386
776 	libfdisk_fini(&epp);
777 #endif
778 
779 	(void) close(fd);
780 	free(mbr);
781 	return (0);
782 }
783 
784 #if defined(_FIRMWARE_NEEDS_FDISK)
785 int
786 auto_solaris_part(struct dk_label *label)
787 {
788 
789 	int		status, i, fd;
790 	struct mboot	mboot;
791 	char		*mbr;
792 	struct ipart	ip;
793 	char		*bootptr;
794 	char		pbuf[MAXPATHLEN];
795 #ifdef i386
796 	uint32_t	relsec, numsec;
797 	int		pno, rval, ext_part_found = 0;
798 	ext_part_t	*epp;
799 #endif
800 
801 	(void) get_pname(&pbuf[0]);
802 	if ((fd = open_disk(pbuf, O_RDONLY)) < 0) {
803 		err_print("Error: can't open selected disk '%s'.\n", pbuf);
804 		return (-1);
805 	}
806 
807 	/*
808 	 * We may get mbr of different size, but the first 512 bytes
809 	 * are valid information.
810 	 */
811 	mbr = malloc(cur_blksz);
812 	if (mbr == NULL) {
813 		err_print("No memory available.\n");
814 		return (-1);
815 	}
816 	status = read(fd, mbr, cur_blksz);
817 
818 	if (status != cur_blksz) {
819 		err_print("Bad read of fdisk partition.\n");
820 		free(mbr);
821 		return (-1);
822 	}
823 
824 	(void) memcpy(&mboot, mbr, sizeof (struct mboot));
825 
826 #ifdef i386
827 	(void) extpart_init(&epp);
828 #endif
829 	for (i = 0; i < FD_NUMPART; i++) {
830 		int	ipc;
831 
832 		ipc = i * sizeof (struct ipart);
833 
834 		/* Handling the alignment problem of struct ipart */
835 		bootptr = &mboot.parts[ipc];
836 		(void) fill_ipart(bootptr, &ip);
837 
838 #ifdef i386
839 		if (fdisk_is_dos_extended(ip.systid) && (ext_part_found == 0)) {
840 			/* We support only one extended partition per disk */
841 			ext_part_found = 1;
842 			rval = fdisk_get_solaris_part(epp, &pno, &relsec,
843 			    &numsec);
844 			if (rval == FDISK_SUCCESS) {
845 				/*
846 				 * Found a solaris partition inside the
847 				 * extended partition. Update the statistics.
848 				 */
849 				if ((label->dkl_nhead != 0) &&
850 				    (label->dkl_nsect != 0)) {
851 					label->dkl_pcyl =
852 					    numsec / (label->dkl_nhead *
853 					    label->dkl_nsect);
854 					label->dkl_ncyl = label->dkl_pcyl -
855 					    label->dkl_acyl;
856 				}
857 				solaris_offset = relsec;
858 			}
859 			continue;
860 		}
861 #endif
862 
863 		/*
864 		 * if the disk has an EFI label, the nhead and nsect fields
865 		 * the label may be zero.  This protects us from FPE's, and
866 		 * format still seems to work happily
867 		 */
868 
869 
870 #ifdef i386
871 		if ((ip.systid == SUNIXOS &&
872 		    (fdisk_is_linux_swap(epp, lel(ip.relsect), NULL) != 0)) ||
873 		    ip.systid == SUNIXOS2 ||
874 		    ip.systid == EFI_PMBR) {
875 #else
876 		if (ip.systid == SUNIXOS ||
877 		    ip.systid == SUNIXOS2 ||
878 		    ip.systid == EFI_PMBR) {
879 #endif
880 			if ((label->dkl_nhead != 0) &&
881 			    (label->dkl_nsect != 0)) {
882 				label->dkl_pcyl = lel(ip.numsect) /
883 				    (label->dkl_nhead * label->dkl_nsect);
884 				label->dkl_ncyl = label->dkl_pcyl -
885 				    label->dkl_acyl;
886 			}
887 #ifdef DEBUG
888 			else {
889 				err_print("Critical label fields aren't "
890 				    "non-zero:\n"
891 				    "\tlabel->dkl_nhead = %d; "
892 				    "label->dkl_nsect = "
893 				    "%d\n", label->dkl_nhead,
894 				    label->dkl_nsect);
895 			}
896 #endif /* DEBUG */
897 
898 		solaris_offset = lel(ip.relsect);
899 		break;
900 		}
901 	}
902 
903 #ifdef i386
904 	libfdisk_fini(&epp);
905 #endif
906 	(void) close(fd);
907 	free(mbr);
908 	return (0);
909 }
910 #endif	/* defined(_FIRMWARE_NEEDS_FDISK) */
911 
912 
913 int
914 good_fdisk()
915 {
916 	char		buf[MAXPATHLEN];
917 	struct stat	statbuf;
918 
919 	(void) get_pname(&buf[0]);
920 	if (stat(buf, &statbuf) == -1 ||
921 	    !S_ISCHR(statbuf.st_mode) ||
922 	    cur_label == L_TYPE_EFI) {
923 		/*
924 		 * Return if this disk does not support fdisk table or
925 		 * if the disk is labeled with EFI.
926 		 */
927 		return (1);
928 	}
929 
930 	if (lel(cur_disk->fdisk_part.numsect) > 0) {
931 		return (1);
932 	} else {
933 		err_print("WARNING - ");
934 		err_print("This disk may be in use by an application "
935 		    "that has\n\t  modified the fdisk table. Ensure "
936 		    "that this disk is\n\t  not currently in use "
937 		    "before proceeding to use fdisk.\n");
938 		return (0);
939 	}
940 }
941 
942 #ifdef i386
943 int
944 extpart_init(ext_part_t **epp)
945 {
946 	int		rval, lf_op_flag = 0;
947 	char		p0_path[MAXPATHLEN];
948 
949 	get_pname(&p0_path[0]);
950 	lf_op_flag |= FDISK_READ_DISK;
951 	if ((rval = libfdisk_init(epp, p0_path, NULL, lf_op_flag)) !=
952 	    FDISK_SUCCESS) {
953 		switch (rval) {
954 			/*
955 			 * FDISK_EBADLOGDRIVE, FDISK_ENOLOGDRIVE
956 			 * and FDISK_EBADMAGIC can be considered
957 			 * as soft errors and hence we do not exit.
958 			 */
959 			case FDISK_EBADLOGDRIVE:
960 				break;
961 			case FDISK_ENOLOGDRIVE:
962 				break;
963 			case FDISK_EBADMAGIC:
964 				break;
965 			case FDISK_ENOVGEOM:
966 				err_print("Could not get virtual geometry for"
967 				    " this device\n");
968 				libfdisk_fini(epp);
969 				fullabort();
970 				break;
971 			case FDISK_ENOPGEOM:
972 				err_print("Could not get physical geometry for"
973 				    " this device\n");
974 				libfdisk_fini(epp);
975 				fullabort();
976 				break;
977 			case FDISK_ENOLGEOM:
978 				err_print("Could not get label geometry for "
979 				    " this device\n");
980 				libfdisk_fini(epp);
981 				fullabort();
982 				break;
983 			default:
984 				err_print("Failed to initialise libfdisk.\n");
985 				libfdisk_fini(epp);
986 				fullabort();
987 				break;
988 		}
989 	}
990 	return (0);
991 }
992 #endif
993