xref: /onnv-gate/usr/src/cmd/boot/bootadm/bootadm.c (revision 662:524269c34bb5)
10Sstevel@tonic-gate /*
20Sstevel@tonic-gate  * CDDL HEADER START
30Sstevel@tonic-gate  *
40Sstevel@tonic-gate  * The contents of this file are subject to the terms of the
50Sstevel@tonic-gate  * Common Development and Distribution License, Version 1.0 only
60Sstevel@tonic-gate  * (the "License").  You may not use this file except in compliance
70Sstevel@tonic-gate  * with the License.
80Sstevel@tonic-gate  *
90Sstevel@tonic-gate  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
100Sstevel@tonic-gate  * or http://www.opensolaris.org/os/licensing.
110Sstevel@tonic-gate  * See the License for the specific language governing permissions
120Sstevel@tonic-gate  * and limitations under the License.
130Sstevel@tonic-gate  *
140Sstevel@tonic-gate  * When distributing Covered Code, include this CDDL HEADER in each
150Sstevel@tonic-gate  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
160Sstevel@tonic-gate  * If applicable, add the following below this CDDL HEADER, with the
170Sstevel@tonic-gate  * fields enclosed by brackets "[]" replaced with your own identifying
180Sstevel@tonic-gate  * information: Portions Copyright [yyyy] [name of copyright owner]
190Sstevel@tonic-gate  *
200Sstevel@tonic-gate  * CDDL HEADER END
210Sstevel@tonic-gate  */
220Sstevel@tonic-gate /*
230Sstevel@tonic-gate  * Copyright 2005 Sun Microsystems, Inc.  All rights reserved.
240Sstevel@tonic-gate  * Use is subject to license terms.
250Sstevel@tonic-gate  */
260Sstevel@tonic-gate 
270Sstevel@tonic-gate #pragma ident	"%Z%%M%	%I%	%E% SMI"
280Sstevel@tonic-gate 
290Sstevel@tonic-gate /*
300Sstevel@tonic-gate  * bootadm(1M) is a new utility for managing bootability of
310Sstevel@tonic-gate  * Solaris *Newboot* environments. It has two primary tasks:
320Sstevel@tonic-gate  * 	- Allow end users to manage bootability of Newboot Solaris instances
330Sstevel@tonic-gate  *	- Provide services to other subsystems in Solaris (primarily Install)
340Sstevel@tonic-gate  */
350Sstevel@tonic-gate 
360Sstevel@tonic-gate /* Headers */
370Sstevel@tonic-gate #include <stdio.h>
380Sstevel@tonic-gate #include <errno.h>
390Sstevel@tonic-gate #include <stdlib.h>
400Sstevel@tonic-gate #include <string.h>
410Sstevel@tonic-gate #include <unistd.h>
420Sstevel@tonic-gate #include <sys/types.h>
430Sstevel@tonic-gate #include <sys/stat.h>
440Sstevel@tonic-gate #include <stdarg.h>
450Sstevel@tonic-gate #include <limits.h>
460Sstevel@tonic-gate #include <signal.h>
470Sstevel@tonic-gate #include <sys/wait.h>
480Sstevel@tonic-gate #include <sys/mnttab.h>
490Sstevel@tonic-gate #include <sys/statvfs.h>
500Sstevel@tonic-gate #include <libnvpair.h>
510Sstevel@tonic-gate #include <ftw.h>
520Sstevel@tonic-gate #include <fcntl.h>
530Sstevel@tonic-gate #include <strings.h>
540Sstevel@tonic-gate #include <sys/systeminfo.h>
550Sstevel@tonic-gate #include <sys/dktp/fdisk.h>
560Sstevel@tonic-gate 
570Sstevel@tonic-gate #include <pwd.h>
580Sstevel@tonic-gate #include <grp.h>
590Sstevel@tonic-gate #include <device_info.h>
600Sstevel@tonic-gate 
610Sstevel@tonic-gate #include <libintl.h>
620Sstevel@tonic-gate #include <locale.h>
630Sstevel@tonic-gate 
640Sstevel@tonic-gate #include <assert.h>
650Sstevel@tonic-gate 
660Sstevel@tonic-gate #include "message.h"
670Sstevel@tonic-gate 
680Sstevel@tonic-gate #ifndef TEXT_DOMAIN
690Sstevel@tonic-gate #define	TEXT_DOMAIN	"SUNW_OST_OSCMD"
700Sstevel@tonic-gate #endif	/* TEXT_DOMAIN */
710Sstevel@tonic-gate 
720Sstevel@tonic-gate /* Type definitions */
730Sstevel@tonic-gate 
740Sstevel@tonic-gate /* Primary subcmds */
750Sstevel@tonic-gate typedef enum {
760Sstevel@tonic-gate 	BAM_MENU = 3,
770Sstevel@tonic-gate 	BAM_ARCHIVE
780Sstevel@tonic-gate } subcmd_t;
790Sstevel@tonic-gate 
800Sstevel@tonic-gate /* GRUB menu per-line classification */
810Sstevel@tonic-gate typedef enum {
820Sstevel@tonic-gate 	BAM_INVALID = 0,
830Sstevel@tonic-gate 	BAM_EMPTY,
840Sstevel@tonic-gate 	BAM_COMMENT,
850Sstevel@tonic-gate 	BAM_GLOBAL,
860Sstevel@tonic-gate 	BAM_ENTRY,
870Sstevel@tonic-gate 	BAM_TITLE
880Sstevel@tonic-gate } menu_flag_t;
890Sstevel@tonic-gate 
900Sstevel@tonic-gate /* struct for menu.lst contents */
910Sstevel@tonic-gate typedef struct line {
920Sstevel@tonic-gate 	int  lineNum;	/* Line number in menu.lst */
930Sstevel@tonic-gate 	int  entryNum;	/* menu boot entry #. ENTRY_INIT if not applicable */
940Sstevel@tonic-gate 	char *cmd;
950Sstevel@tonic-gate 	char *sep;
960Sstevel@tonic-gate 	char *arg;
970Sstevel@tonic-gate 	char *line;
980Sstevel@tonic-gate 	menu_flag_t flags;
990Sstevel@tonic-gate 	struct line *next;
100*662Sszhou 	struct line *prev;
1010Sstevel@tonic-gate } line_t;
1020Sstevel@tonic-gate 
103*662Sszhou typedef struct entry {
104*662Sszhou 	struct entry *next;
105*662Sszhou 	struct entry *prev;
106*662Sszhou 	line_t *start;
107*662Sszhou 	line_t *end;
108*662Sszhou } entry_t;
109*662Sszhou 
1100Sstevel@tonic-gate typedef struct {
1110Sstevel@tonic-gate 	line_t *start;
1120Sstevel@tonic-gate 	line_t *end;
113*662Sszhou 	line_t *curdefault;	/* line containing default */
114*662Sszhou 	line_t *olddefault;	/* old default line (commented) */
115*662Sszhou 	entry_t *entries;	/* os entries */
1160Sstevel@tonic-gate } menu_t;
1170Sstevel@tonic-gate 
1180Sstevel@tonic-gate typedef enum {
1190Sstevel@tonic-gate     OPT_ABSENT = 0,	/* No option */
1200Sstevel@tonic-gate     OPT_REQ,		/* option required */
1210Sstevel@tonic-gate     OPT_OPTIONAL	/* option may or may not be present */
1220Sstevel@tonic-gate } option_t;
1230Sstevel@tonic-gate 
1240Sstevel@tonic-gate typedef enum {
125316Svikram     BAM_ERROR = -1,	/* Must be negative. add_boot_entry() depends on it */
1260Sstevel@tonic-gate     BAM_SUCCESS = 0,
1270Sstevel@tonic-gate     BAM_WRITE = 2
1280Sstevel@tonic-gate } error_t;
1290Sstevel@tonic-gate 
1300Sstevel@tonic-gate typedef struct {
1310Sstevel@tonic-gate 	char	*subcmd;
1320Sstevel@tonic-gate 	option_t option;
1330Sstevel@tonic-gate 	error_t (*handler)();
1340Sstevel@tonic-gate } subcmd_defn_t;
1350Sstevel@tonic-gate 
1360Sstevel@tonic-gate 
1370Sstevel@tonic-gate #define	BAM_MAXLINE	8192
1380Sstevel@tonic-gate 
1390Sstevel@tonic-gate #define	LINE_INIT	0	/* lineNum initial value */
1400Sstevel@tonic-gate #define	ENTRY_INIT	-1	/* entryNum initial value */
1410Sstevel@tonic-gate #define	ALL_ENTRIES	-2	/* selects all boot entries */
1420Sstevel@tonic-gate 
1430Sstevel@tonic-gate #define	GRUB_DIR		"/boot/grub"
1440Sstevel@tonic-gate #define	MULTI_BOOT		"/platform/i86pc/multiboot"
1450Sstevel@tonic-gate #define	BOOT_ARCHIVE		"/platform/i86pc/boot_archive"
1460Sstevel@tonic-gate #define	GRUB_MENU		"/boot/grub/menu.lst"
1470Sstevel@tonic-gate #define	MENU_TMP		"/boot/grub/menu.lst.tmp"
1480Sstevel@tonic-gate #define	RAMDISK_SPECIAL		"/ramdisk"
149621Svikram #define	STUBBOOT		"/stubboot"
1500Sstevel@tonic-gate 
1510Sstevel@tonic-gate /* lock related */
1520Sstevel@tonic-gate #define	BAM_LOCK_FILE		"/var/run/bootadm.lock"
1530Sstevel@tonic-gate #define	LOCK_FILE_PERMS		(S_IRUSR|S_IWUSR|S_IRGRP|S_IROTH)
1540Sstevel@tonic-gate 
1550Sstevel@tonic-gate #define	CREATE_RAMDISK		"/boot/solaris/bin/create_ramdisk"
1560Sstevel@tonic-gate #define	CREATE_DISKMAP		"/boot/solaris/bin/create_diskmap"
1570Sstevel@tonic-gate #define	GRUBDISK_MAP		"/var/run/solaris_grubdisk.map"
1580Sstevel@tonic-gate 
159316Svikram #define	GRUB_slice		"/etc/lu/GRUB_slice"
160316Svikram #define	GRUB_root		"/etc/lu/GRUB_root"
161316Svikram #define	GRUB_backup_menu	"/etc/lu/GRUB_backup_menu"
162316Svikram #define	GRUB_slice_mntpt	"/tmp/GRUB_slice_mntpt"
163316Svikram #define	LU_ACTIVATE_FILE	"/etc/lu/DelayUpdate/activate.sh"
164316Svikram 
165316Svikram #define	INSTALLGRUB		"/sbin/installgrub"
166316Svikram #define	STAGE1			"/boot/grub/stage1"
167316Svikram #define	STAGE2			"/boot/grub/stage2"
168316Svikram 
1690Sstevel@tonic-gate /*
1700Sstevel@tonic-gate  * Default file attributes
1710Sstevel@tonic-gate  */
1720Sstevel@tonic-gate #define	DEFAULT_DEV_MODE	0644	/* default permissions */
1730Sstevel@tonic-gate #define	DEFAULT_DEV_UID		0	/* user root */
1740Sstevel@tonic-gate #define	DEFAULT_DEV_GID		3	/* group sys */
1750Sstevel@tonic-gate 
1760Sstevel@tonic-gate /*
1770Sstevel@tonic-gate  * Menu related
1780Sstevel@tonic-gate  * menu_cmd_t and menu_cmds must be kept in sync
1790Sstevel@tonic-gate  */
1800Sstevel@tonic-gate typedef enum {
1810Sstevel@tonic-gate 	DEFAULT_CMD = 0,
1820Sstevel@tonic-gate 	TIMEOUT_CMD,
1830Sstevel@tonic-gate 	TITLE_CMD,
1840Sstevel@tonic-gate 	ROOT_CMD,
1850Sstevel@tonic-gate 	KERNEL_CMD,
1860Sstevel@tonic-gate 	MODULE_CMD,
1870Sstevel@tonic-gate 	SEP_CMD,
1880Sstevel@tonic-gate 	COMMENT_CMD
1890Sstevel@tonic-gate } menu_cmd_t;
1900Sstevel@tonic-gate 
1910Sstevel@tonic-gate static char *menu_cmds[] = {
1920Sstevel@tonic-gate 	"default",	/* DEFAULT_CMD */
1930Sstevel@tonic-gate 	"timeout",	/* TIMEOUT_CMD */
1940Sstevel@tonic-gate 	"title",	/* TITLE_CMD */
1950Sstevel@tonic-gate 	"root",		/* ROOT_CMD */
1960Sstevel@tonic-gate 	"kernel",	/* KERNEL_CMD */
1970Sstevel@tonic-gate 	"module",	/* MODULE_CMD */
1980Sstevel@tonic-gate 	" ",		/* SEP_CMD */
1990Sstevel@tonic-gate 	"#",		/* COMMENT_CMD */
2000Sstevel@tonic-gate 	NULL
2010Sstevel@tonic-gate };
2020Sstevel@tonic-gate 
2030Sstevel@tonic-gate #define	OPT_ENTRY_NUM	"entry"
2040Sstevel@tonic-gate 
2050Sstevel@tonic-gate /*
2060Sstevel@tonic-gate  * archive related
2070Sstevel@tonic-gate  */
2080Sstevel@tonic-gate typedef struct {
2090Sstevel@tonic-gate 	line_t *head;
2100Sstevel@tonic-gate 	line_t *tail;
2110Sstevel@tonic-gate } filelist_t;
2120Sstevel@tonic-gate 
2130Sstevel@tonic-gate #define	BOOT_FILE_LIST	"boot/solaris/filelist.ramdisk"
2140Sstevel@tonic-gate #define	ETC_FILE_LIST	"etc/boot/solaris/filelist.ramdisk"
2150Sstevel@tonic-gate 
2160Sstevel@tonic-gate #define	FILE_STAT	"boot/solaris/filestat.ramdisk"
2170Sstevel@tonic-gate #define	FILE_STAT_TMP	"boot/solaris/filestat.ramdisk.tmp"
2180Sstevel@tonic-gate #define	DIR_PERMS	(S_IRWXU|S_IRGRP|S_IXGRP|S_IROTH|S_IXOTH)
2190Sstevel@tonic-gate #define	FILE_STAT_MODE	(S_IRUSR|S_IWUSR|S_IRGRP|S_IROTH)
2200Sstevel@tonic-gate 
2210Sstevel@tonic-gate #define	BAM_HDR		"---------- ADDED BY BOOTADM - DO NOT EDIT ----------"
2220Sstevel@tonic-gate #define	BAM_FTR		"---------------------END BOOTADM--------------------"
223*662Sszhou #define	BAM_OLDDEF	"BOOTADM SAVED DEFAULT: "
2240Sstevel@tonic-gate 
2250Sstevel@tonic-gate /* Globals */
2260Sstevel@tonic-gate static char *prog;
2270Sstevel@tonic-gate static subcmd_t bam_cmd;
2280Sstevel@tonic-gate static char *bam_root;
2290Sstevel@tonic-gate static int bam_rootlen;
2300Sstevel@tonic-gate static int bam_root_readonly;
231621Svikram static int bam_alt_root;
2320Sstevel@tonic-gate static char *bam_subcmd;
2330Sstevel@tonic-gate static char *bam_opt;
2340Sstevel@tonic-gate static int bam_debug;
2350Sstevel@tonic-gate static char **bam_argv;
2360Sstevel@tonic-gate static int bam_argc;
2370Sstevel@tonic-gate static int bam_force;
2380Sstevel@tonic-gate static int bam_verbose;
2390Sstevel@tonic-gate static int bam_check;
2400Sstevel@tonic-gate static int bam_smf_check;
2410Sstevel@tonic-gate static int bam_lock_fd = -1;
2420Sstevel@tonic-gate static char rootbuf[PATH_MAX] = "/";
243316Svikram static int bam_update_all;
2440Sstevel@tonic-gate 
2450Sstevel@tonic-gate /* function prototypes */
2460Sstevel@tonic-gate static void parse_args_internal(int argc, char *argv[]);
2470Sstevel@tonic-gate static void parse_args(int argc, char *argv[]);
2480Sstevel@tonic-gate static error_t bam_menu(char *subcmd, char *opt, int argc, char *argv[]);
2490Sstevel@tonic-gate static error_t bam_archive(char *subcmd, char *opt);
2500Sstevel@tonic-gate 
2510Sstevel@tonic-gate static void bam_error(char *format, ...);
2520Sstevel@tonic-gate static void bam_print(char *format, ...);
2530Sstevel@tonic-gate static void bam_exit(int excode);
2540Sstevel@tonic-gate static void bam_lock(void);
2550Sstevel@tonic-gate static void bam_unlock(void);
2560Sstevel@tonic-gate 
2570Sstevel@tonic-gate static int exec_cmd(char *cmdline, char *output, int64_t osize);
2580Sstevel@tonic-gate static error_t read_globals(menu_t *mp, char *menu_path,
2590Sstevel@tonic-gate     char *globalcmd, int quiet);
2600Sstevel@tonic-gate 
2610Sstevel@tonic-gate static menu_t *menu_read(char *menu_path);
2620Sstevel@tonic-gate static error_t menu_write(char *root, menu_t *mp);
2630Sstevel@tonic-gate static void linelist_free(line_t *start);
2640Sstevel@tonic-gate static void menu_free(menu_t *mp);
2650Sstevel@tonic-gate static void line_free(line_t *lp);
2660Sstevel@tonic-gate static void filelist_free(filelist_t *flistp);
2670Sstevel@tonic-gate static error_t list2file(char *root, char *tmp,
2680Sstevel@tonic-gate     char *final, line_t *start);
2690Sstevel@tonic-gate static error_t list_entry(menu_t *mp, char *menu_path, char *opt);
2700Sstevel@tonic-gate static error_t delete_all_entries(menu_t *mp, char *menu_path, char *opt);
2710Sstevel@tonic-gate static error_t update_entry(menu_t *mp, char *root, char *opt);
2720Sstevel@tonic-gate static error_t update_temp(menu_t *mp, char *root, char *opt);
2730Sstevel@tonic-gate 
2740Sstevel@tonic-gate static error_t update_archive(char *root, char *opt);
2750Sstevel@tonic-gate static error_t list_archive(char *root, char *opt);
2760Sstevel@tonic-gate static error_t update_all(char *root, char *opt);
2770Sstevel@tonic-gate static error_t read_list(char *root, filelist_t  *flistp);
2780Sstevel@tonic-gate static error_t set_global(menu_t *mp, char *globalcmd, int val);
2790Sstevel@tonic-gate static error_t set_option(menu_t *mp, char *globalcmd, char *opt);
2800Sstevel@tonic-gate 
2810Sstevel@tonic-gate static long s_strtol(char *str);
2820Sstevel@tonic-gate static char *s_fgets(char *buf, int n, FILE *fp);
2830Sstevel@tonic-gate static int s_fputs(char *str, FILE *fp);
2840Sstevel@tonic-gate 
2850Sstevel@tonic-gate static void *s_calloc(size_t nelem, size_t sz);
2860Sstevel@tonic-gate static char *s_strdup(char *str);
2870Sstevel@tonic-gate static int is_readonly(char *);
2880Sstevel@tonic-gate static int is_amd64(void);
2890Sstevel@tonic-gate static void append_to_flist(filelist_t *, char *);
2900Sstevel@tonic-gate 
2910Sstevel@tonic-gate #if defined(__sparc)
2920Sstevel@tonic-gate static void sparc_abort(void);
2930Sstevel@tonic-gate #endif
2940Sstevel@tonic-gate 
2950Sstevel@tonic-gate /* Menu related sub commands */
2960Sstevel@tonic-gate static subcmd_defn_t menu_subcmds[] = {
2970Sstevel@tonic-gate 	"set_option",		OPT_OPTIONAL,	set_option,	/* PUB */
2980Sstevel@tonic-gate 	"list_entry",		OPT_OPTIONAL,	list_entry,	/* PUB */
2990Sstevel@tonic-gate 	"delete_all_entries",	OPT_ABSENT,	delete_all_entries, /* PVT */
3000Sstevel@tonic-gate 	"update_entry",		OPT_REQ,	update_entry,	/* menu */
3010Sstevel@tonic-gate 	"update_temp",		OPT_OPTIONAL,	update_temp,	/* reboot */
3020Sstevel@tonic-gate 	NULL,			0,		NULL	/* must be last */
3030Sstevel@tonic-gate };
3040Sstevel@tonic-gate 
3050Sstevel@tonic-gate /* Archive related sub commands */
3060Sstevel@tonic-gate static subcmd_defn_t arch_subcmds[] = {
3070Sstevel@tonic-gate 	"update",		OPT_ABSENT,	update_archive, /* PUB */
3080Sstevel@tonic-gate 	"update_all",		OPT_ABSENT,	update_all,	/* PVT */
3090Sstevel@tonic-gate 	"list",			OPT_OPTIONAL,	list_archive,	/* PUB */
3100Sstevel@tonic-gate 	NULL,			0,		NULL	/* must be last */
3110Sstevel@tonic-gate };
3120Sstevel@tonic-gate 
3130Sstevel@tonic-gate static struct {
3140Sstevel@tonic-gate 	nvlist_t *new_nvlp;
3150Sstevel@tonic-gate 	nvlist_t *old_nvlp;
3160Sstevel@tonic-gate 	int need_update;
3170Sstevel@tonic-gate } walk_arg;
3180Sstevel@tonic-gate 
3190Sstevel@tonic-gate static void
3200Sstevel@tonic-gate usage(void)
3210Sstevel@tonic-gate {
3220Sstevel@tonic-gate 	(void) fprintf(stderr, "USAGE:\n");
3230Sstevel@tonic-gate 
3240Sstevel@tonic-gate 
3250Sstevel@tonic-gate 	/* archive usage */
3260Sstevel@tonic-gate 	(void) fprintf(stderr, "\t%s update-archive [-vn] [-R altroot]\n",
3270Sstevel@tonic-gate 	    prog);
3280Sstevel@tonic-gate 	(void) fprintf(stderr, "\t%s list-archive [-R altroot]\n", prog);
3290Sstevel@tonic-gate #ifndef __sparc
3300Sstevel@tonic-gate 	/* x86 only */
3310Sstevel@tonic-gate 	(void) fprintf(stderr, "\t%s set-menu [-R altroot] key=value\n", prog);
3320Sstevel@tonic-gate 	(void) fprintf(stderr, "\t%s list-menu [-R altroot]\n", prog);
3330Sstevel@tonic-gate #endif
3340Sstevel@tonic-gate }
3350Sstevel@tonic-gate 
3360Sstevel@tonic-gate int
3370Sstevel@tonic-gate main(int argc, char *argv[])
3380Sstevel@tonic-gate {
3390Sstevel@tonic-gate 	error_t ret;
3400Sstevel@tonic-gate 
3410Sstevel@tonic-gate 	(void) setlocale(LC_ALL, "");
3420Sstevel@tonic-gate 	(void) textdomain(TEXT_DOMAIN);
3430Sstevel@tonic-gate 
3440Sstevel@tonic-gate 	if ((prog = strrchr(argv[0], '/')) == NULL) {
3450Sstevel@tonic-gate 		prog = argv[0];
3460Sstevel@tonic-gate 	} else {
3470Sstevel@tonic-gate 		prog++;
3480Sstevel@tonic-gate 	}
3490Sstevel@tonic-gate 
3500Sstevel@tonic-gate 	if (geteuid() != 0) {
3510Sstevel@tonic-gate 		bam_error(MUST_BE_ROOT);
3520Sstevel@tonic-gate 		bam_exit(1);
3530Sstevel@tonic-gate 	}
3540Sstevel@tonic-gate 
3550Sstevel@tonic-gate 	bam_lock();
3560Sstevel@tonic-gate 
3570Sstevel@tonic-gate 	/*
3580Sstevel@tonic-gate 	 * Don't depend on caller's umask
3590Sstevel@tonic-gate 	 */
3600Sstevel@tonic-gate 	(void) umask(0022);
3610Sstevel@tonic-gate 
3620Sstevel@tonic-gate 	parse_args(argc, argv);
3630Sstevel@tonic-gate 
3640Sstevel@tonic-gate #if defined(__sparc)
3650Sstevel@tonic-gate 	/*
3660Sstevel@tonic-gate 	 * There are only two valid invocations of bootadm
3670Sstevel@tonic-gate 	 * on SPARC:
3680Sstevel@tonic-gate 	 *
3690Sstevel@tonic-gate 	 *	- SPARC diskless server creating boot_archive for i386 clients
3700Sstevel@tonic-gate 	 *	- archive creation call during reboot of a SPARC system
3710Sstevel@tonic-gate 	 *
3720Sstevel@tonic-gate 	 *	The latter should be a NOP
3730Sstevel@tonic-gate 	 */
3740Sstevel@tonic-gate 	if (bam_cmd != BAM_ARCHIVE) {
3750Sstevel@tonic-gate 		sparc_abort();
3760Sstevel@tonic-gate 	}
3770Sstevel@tonic-gate #endif
3780Sstevel@tonic-gate 
3790Sstevel@tonic-gate 	switch (bam_cmd) {
3800Sstevel@tonic-gate 		case BAM_MENU:
3810Sstevel@tonic-gate 			ret = bam_menu(bam_subcmd, bam_opt, bam_argc, bam_argv);
3820Sstevel@tonic-gate 			break;
3830Sstevel@tonic-gate 		case BAM_ARCHIVE:
3840Sstevel@tonic-gate 			ret = bam_archive(bam_subcmd, bam_opt);
3850Sstevel@tonic-gate 			break;
3860Sstevel@tonic-gate 		default:
3870Sstevel@tonic-gate 			usage();
3880Sstevel@tonic-gate 			bam_exit(1);
3890Sstevel@tonic-gate 	}
3900Sstevel@tonic-gate 
3910Sstevel@tonic-gate 	if (ret != BAM_SUCCESS)
3920Sstevel@tonic-gate 		bam_exit(1);
3930Sstevel@tonic-gate 
3940Sstevel@tonic-gate 	bam_unlock();
3950Sstevel@tonic-gate 	return (0);
3960Sstevel@tonic-gate }
3970Sstevel@tonic-gate 
3980Sstevel@tonic-gate #if defined(__sparc)
3990Sstevel@tonic-gate 
4000Sstevel@tonic-gate static void
4010Sstevel@tonic-gate sparc_abort(void)
4020Sstevel@tonic-gate {
4030Sstevel@tonic-gate 	bam_error(NOT_ON_SPARC);
4040Sstevel@tonic-gate 	bam_exit(1);
4050Sstevel@tonic-gate }
4060Sstevel@tonic-gate 
4070Sstevel@tonic-gate #endif
4080Sstevel@tonic-gate 
4090Sstevel@tonic-gate /*
4100Sstevel@tonic-gate  * Equivalence of public and internal commands:
4110Sstevel@tonic-gate  *	update-archive  -- -a update
4120Sstevel@tonic-gate  *	list-archive	-- -a list
4130Sstevel@tonic-gate  *	set-menu	-- -m set_option
4140Sstevel@tonic-gate  *	list-menu	-- -m list_entry
4150Sstevel@tonic-gate  *	update-menu	-- -m update_entry
4160Sstevel@tonic-gate  */
4170Sstevel@tonic-gate static struct cmd_map {
4180Sstevel@tonic-gate 	char *bam_cmdname;
4190Sstevel@tonic-gate 	int bam_cmd;
4200Sstevel@tonic-gate 	char *bam_subcmd;
4210Sstevel@tonic-gate } cmd_map[] = {
4220Sstevel@tonic-gate 	{ "update-archive",	BAM_ARCHIVE,	"update"},
4230Sstevel@tonic-gate 	{ "list-archive",	BAM_ARCHIVE,	"list"},
4240Sstevel@tonic-gate 	{ "set-menu",		BAM_MENU,	"set_option"},
4250Sstevel@tonic-gate 	{ "list-menu",		BAM_MENU,	"list_entry"},
4260Sstevel@tonic-gate 	{ "update-menu",	BAM_MENU,	"update_entry"},
4270Sstevel@tonic-gate 	{ NULL,			0,		NULL}
4280Sstevel@tonic-gate };
4290Sstevel@tonic-gate 
4300Sstevel@tonic-gate /*
4310Sstevel@tonic-gate  * Commands syntax published in bootadm(1M) are parsed here
4320Sstevel@tonic-gate  */
4330Sstevel@tonic-gate static void
4340Sstevel@tonic-gate parse_args(int argc, char *argv[])
4350Sstevel@tonic-gate {
4360Sstevel@tonic-gate 	struct cmd_map *cmp = cmd_map;
4370Sstevel@tonic-gate 
4380Sstevel@tonic-gate 	/* command conforming to the final spec */
4390Sstevel@tonic-gate 	if (argc > 1 && argv[1][0] != '-') {
4400Sstevel@tonic-gate 		/*
4410Sstevel@tonic-gate 		 * Map commands to internal table.
4420Sstevel@tonic-gate 		 */
4430Sstevel@tonic-gate 		while (cmp->bam_cmdname) {
4440Sstevel@tonic-gate 			if (strcmp(argv[1], cmp->bam_cmdname) == 0) {
4450Sstevel@tonic-gate 				bam_cmd = cmp->bam_cmd;
4460Sstevel@tonic-gate 				bam_subcmd = cmp->bam_subcmd;
4470Sstevel@tonic-gate 				break;
4480Sstevel@tonic-gate 			}
4490Sstevel@tonic-gate 			cmp++;
4500Sstevel@tonic-gate 		}
4510Sstevel@tonic-gate 		if (cmp->bam_cmdname == NULL) {
4520Sstevel@tonic-gate 			usage();
4530Sstevel@tonic-gate 			bam_exit(1);
4540Sstevel@tonic-gate 		}
4550Sstevel@tonic-gate 		argc--;
4560Sstevel@tonic-gate 		argv++;
4570Sstevel@tonic-gate 	}
4580Sstevel@tonic-gate 
4590Sstevel@tonic-gate 	parse_args_internal(argc, argv);
4600Sstevel@tonic-gate }
4610Sstevel@tonic-gate 
4620Sstevel@tonic-gate /*
4630Sstevel@tonic-gate  * A combination of public and private commands are parsed here.
4640Sstevel@tonic-gate  * The internal syntax and the corresponding functionality are:
4650Sstevel@tonic-gate  *	-a update	-- update-archive
4660Sstevel@tonic-gate  *	-a list		-- list-archive
4670Sstevel@tonic-gate  *	-a update-all	-- (reboot to sync all mounted OS archive)
4680Sstevel@tonic-gate  *	-m update_entry	-- update-menu
4690Sstevel@tonic-gate  *	-m list_entry	-- list-menu
4700Sstevel@tonic-gate  *	-m update_temp	-- (reboot -- [boot-args])
4710Sstevel@tonic-gate  *	-m delete_all_entries -- (called from install)
4720Sstevel@tonic-gate  */
4730Sstevel@tonic-gate static void
4740Sstevel@tonic-gate parse_args_internal(int argc, char *argv[])
4750Sstevel@tonic-gate {
4760Sstevel@tonic-gate 	int c, error;
4770Sstevel@tonic-gate 	extern char *optarg;
4780Sstevel@tonic-gate 	extern int optind, opterr;
4790Sstevel@tonic-gate 
4800Sstevel@tonic-gate 	/* Suppress error message from getopt */
4810Sstevel@tonic-gate 	opterr = 0;
4820Sstevel@tonic-gate 
4830Sstevel@tonic-gate 	error = 0;
484*662Sszhou 	while ((c = getopt(argc, argv, "a:d:fm:no:vCR:")) != -1) {
4850Sstevel@tonic-gate 		switch (c) {
4860Sstevel@tonic-gate 		case 'a':
4870Sstevel@tonic-gate 			if (bam_cmd) {
4880Sstevel@tonic-gate 				error = 1;
4890Sstevel@tonic-gate 				bam_error(MULT_CMDS, c);
4900Sstevel@tonic-gate 			}
4910Sstevel@tonic-gate 			bam_cmd = BAM_ARCHIVE;
4920Sstevel@tonic-gate 			bam_subcmd = optarg;
4930Sstevel@tonic-gate 			break;
4940Sstevel@tonic-gate 		case 'd':
4950Sstevel@tonic-gate 			if (bam_debug) {
4960Sstevel@tonic-gate 				error = 1;
4970Sstevel@tonic-gate 				bam_error(DUP_OPT, c);
4980Sstevel@tonic-gate 			}
4990Sstevel@tonic-gate 			bam_debug = s_strtol(optarg);
5000Sstevel@tonic-gate 			break;
5010Sstevel@tonic-gate 		case 'f':
5020Sstevel@tonic-gate 			if (bam_force) {
5030Sstevel@tonic-gate 				error = 1;
5040Sstevel@tonic-gate 				bam_error(DUP_OPT, c);
5050Sstevel@tonic-gate 			}
5060Sstevel@tonic-gate 			bam_force = 1;
5070Sstevel@tonic-gate 			break;
5080Sstevel@tonic-gate 		case 'm':
5090Sstevel@tonic-gate 			if (bam_cmd) {
5100Sstevel@tonic-gate 				error = 1;
5110Sstevel@tonic-gate 				bam_error(MULT_CMDS, c);
5120Sstevel@tonic-gate 			}
5130Sstevel@tonic-gate 			bam_cmd = BAM_MENU;
5140Sstevel@tonic-gate 			bam_subcmd = optarg;
5150Sstevel@tonic-gate 			break;
5160Sstevel@tonic-gate 		case 'n':
5170Sstevel@tonic-gate 			if (bam_check) {
5180Sstevel@tonic-gate 				error = 1;
5190Sstevel@tonic-gate 				bam_error(DUP_OPT, c);
5200Sstevel@tonic-gate 			}
5210Sstevel@tonic-gate 			bam_check = 1;
5220Sstevel@tonic-gate 			break;
5230Sstevel@tonic-gate 		case 'o':
5240Sstevel@tonic-gate 			if (bam_opt) {
5250Sstevel@tonic-gate 				error = 1;
5260Sstevel@tonic-gate 				bam_error(DUP_OPT, c);
5270Sstevel@tonic-gate 			}
5280Sstevel@tonic-gate 			bam_opt = optarg;
5290Sstevel@tonic-gate 			break;
5300Sstevel@tonic-gate 		case 'v':
5310Sstevel@tonic-gate 			if (bam_verbose) {
5320Sstevel@tonic-gate 				error = 1;
5330Sstevel@tonic-gate 				bam_error(DUP_OPT, c);
5340Sstevel@tonic-gate 			}
5350Sstevel@tonic-gate 			bam_verbose = 1;
5360Sstevel@tonic-gate 			break;
537*662Sszhou 		case 'C':
538*662Sszhou 			bam_smf_check = 1;
539*662Sszhou 			break;
5400Sstevel@tonic-gate 		case 'R':
5410Sstevel@tonic-gate 			if (bam_root) {
5420Sstevel@tonic-gate 				error = 1;
5430Sstevel@tonic-gate 				bam_error(DUP_OPT, c);
5440Sstevel@tonic-gate 				break;
5450Sstevel@tonic-gate 			} else if (realpath(optarg, rootbuf) == NULL) {
5460Sstevel@tonic-gate 				error = 1;
5470Sstevel@tonic-gate 				bam_error(CANT_RESOLVE, optarg,
5480Sstevel@tonic-gate 				    strerror(errno));
5490Sstevel@tonic-gate 				break;
5500Sstevel@tonic-gate 			}
551621Svikram 			bam_alt_root = 1;
5520Sstevel@tonic-gate 			bam_root = rootbuf;
5530Sstevel@tonic-gate 			bam_rootlen = strlen(rootbuf);
5540Sstevel@tonic-gate 			break;
5550Sstevel@tonic-gate 		case '?':
5560Sstevel@tonic-gate 			error = 1;
5570Sstevel@tonic-gate 			bam_error(BAD_OPT, optopt);
5580Sstevel@tonic-gate 			break;
5590Sstevel@tonic-gate 		default :
5600Sstevel@tonic-gate 			error = 1;
5610Sstevel@tonic-gate 			bam_error(BAD_OPT, c);
5620Sstevel@tonic-gate 			break;
5630Sstevel@tonic-gate 		}
5640Sstevel@tonic-gate 	}
5650Sstevel@tonic-gate 
5660Sstevel@tonic-gate 	/*
5670Sstevel@tonic-gate 	 * A command option must be specfied
5680Sstevel@tonic-gate 	 */
5690Sstevel@tonic-gate 	if (!bam_cmd) {
5700Sstevel@tonic-gate 		if (bam_opt && strcmp(bam_opt, "all") == 0) {
5710Sstevel@tonic-gate 			usage();
5720Sstevel@tonic-gate 			bam_exit(0);
5730Sstevel@tonic-gate 		}
5740Sstevel@tonic-gate 		bam_error(NEED_CMD);
5750Sstevel@tonic-gate 		error = 1;
5760Sstevel@tonic-gate 	}
5770Sstevel@tonic-gate 
5780Sstevel@tonic-gate 	if (error) {
5790Sstevel@tonic-gate 		usage();
5800Sstevel@tonic-gate 		bam_exit(1);
5810Sstevel@tonic-gate 	}
5820Sstevel@tonic-gate 
5830Sstevel@tonic-gate 	if (optind > argc) {
5840Sstevel@tonic-gate 		bam_error(INT_ERROR, "parse_args");
5850Sstevel@tonic-gate 		bam_exit(1);
5860Sstevel@tonic-gate 	} else if (optind < argc) {
5870Sstevel@tonic-gate 		bam_argv = &argv[optind];
5880Sstevel@tonic-gate 		bam_argc = argc - optind;
5890Sstevel@tonic-gate 	}
5900Sstevel@tonic-gate 
5910Sstevel@tonic-gate 	/*
5920Sstevel@tonic-gate 	 * -n implies verbose mode
5930Sstevel@tonic-gate 	 */
5940Sstevel@tonic-gate 	if (bam_check)
5950Sstevel@tonic-gate 		bam_verbose = 1;
5960Sstevel@tonic-gate }
5970Sstevel@tonic-gate 
5980Sstevel@tonic-gate static error_t
5990Sstevel@tonic-gate check_subcmd_and_options(
6000Sstevel@tonic-gate 	char *subcmd,
6010Sstevel@tonic-gate 	char *opt,
6020Sstevel@tonic-gate 	subcmd_defn_t *table,
6030Sstevel@tonic-gate 	error_t (**fp)())
6040Sstevel@tonic-gate {
6050Sstevel@tonic-gate 	int i;
6060Sstevel@tonic-gate 
6070Sstevel@tonic-gate 	if (subcmd == NULL) {
6080Sstevel@tonic-gate 		bam_error(NEED_SUBCMD);
6090Sstevel@tonic-gate 		return (BAM_ERROR);
6100Sstevel@tonic-gate 	}
6110Sstevel@tonic-gate 
6120Sstevel@tonic-gate 	if (bam_root == NULL) {
6130Sstevel@tonic-gate 		bam_root = rootbuf;
6140Sstevel@tonic-gate 		bam_rootlen = 1;
6150Sstevel@tonic-gate 	}
6160Sstevel@tonic-gate 
6170Sstevel@tonic-gate 	/* verify that subcmd is valid */
6180Sstevel@tonic-gate 	for (i = 0; table[i].subcmd != NULL; i++) {
6190Sstevel@tonic-gate 		if (strcmp(table[i].subcmd, subcmd) == 0)
6200Sstevel@tonic-gate 			break;
6210Sstevel@tonic-gate 	}
6220Sstevel@tonic-gate 
6230Sstevel@tonic-gate 	if (table[i].subcmd == NULL) {
6240Sstevel@tonic-gate 		bam_error(INVALID_SUBCMD, subcmd);
6250Sstevel@tonic-gate 		return (BAM_ERROR);
6260Sstevel@tonic-gate 	}
6270Sstevel@tonic-gate 
6280Sstevel@tonic-gate 	/* subcmd verifies that opt is appropriate */
6290Sstevel@tonic-gate 	if (table[i].option != OPT_OPTIONAL) {
6300Sstevel@tonic-gate 		if ((table[i].option == OPT_REQ) ^ (opt != NULL)) {
6310Sstevel@tonic-gate 			if (opt)
6320Sstevel@tonic-gate 				bam_error(NO_OPT_REQ, subcmd);
6330Sstevel@tonic-gate 			else
6340Sstevel@tonic-gate 				bam_error(MISS_OPT, subcmd);
6350Sstevel@tonic-gate 			return (BAM_ERROR);
6360Sstevel@tonic-gate 		}
6370Sstevel@tonic-gate 	}
6380Sstevel@tonic-gate 
6390Sstevel@tonic-gate 	*fp = table[i].handler;
6400Sstevel@tonic-gate 
6410Sstevel@tonic-gate 	return (BAM_SUCCESS);
6420Sstevel@tonic-gate }
6430Sstevel@tonic-gate 
644316Svikram 
645316Svikram static char *
646621Svikram mount_grub_slice(int *mnted, char **physlice, char **logslice, char **fs_type)
647316Svikram {
648316Svikram 	struct extmnttab mnt;
649316Svikram 	struct stat sb;
650316Svikram 	char buf[BAM_MAXLINE], dev[PATH_MAX], phys[PATH_MAX], fstype[32];
651348Svikram 	char cmd[PATH_MAX];
652316Svikram 	char *mntpt;
653316Svikram 	int p, l, f;
654316Svikram 	FILE *fp;
655316Svikram 
656316Svikram 	assert(mnted);
657316Svikram 	*mnted = 0;
658316Svikram 
659316Svikram 	/*
660621Svikram 	 * physlice, logslice, fs_type  args may be NULL
661316Svikram 	 */
662316Svikram 	if (physlice)
663316Svikram 		*physlice = NULL;
664621Svikram 	if (logslice)
665621Svikram 		*logslice = NULL;
666621Svikram 	if (fs_type)
667621Svikram 		*fs_type = NULL;
668316Svikram 
669316Svikram 	if (stat(GRUB_slice, &sb) != 0) {
670316Svikram 		bam_error(MISSING_SLICE_FILE, GRUB_slice, strerror(errno));
671316Svikram 		return (NULL);
672316Svikram 	}
673316Svikram 
674316Svikram 	fp = fopen(GRUB_slice, "r");
675316Svikram 	if (fp == NULL) {
676316Svikram 		bam_error(OPEN_FAIL, GRUB_slice, strerror(errno));
677316Svikram 		return (NULL);
678316Svikram 	}
679316Svikram 
680316Svikram 	dev[0] = fstype[0] = phys[0] = '\0';
681316Svikram 	p = sizeof ("PHYS_SLICE=") - 1;
682316Svikram 	l = sizeof ("LOG_SLICE=") - 1;
683316Svikram 	f = sizeof ("LOG_FSTYP=") - 1;
684316Svikram 	while (s_fgets(buf, sizeof (buf), fp) != NULL) {
685316Svikram 		if (strncmp(buf, "PHYS_SLICE=", p) == 0) {
686316Svikram 			(void) strlcpy(phys, buf + p, sizeof (phys));
687316Svikram 			continue;
688316Svikram 		}
689316Svikram 		if (strncmp(buf, "LOG_SLICE=", l) == 0) {
690316Svikram 			(void) strlcpy(dev, buf + l, sizeof (dev));
691316Svikram 			continue;
692316Svikram 		}
693316Svikram 		if (strncmp(buf, "LOG_FSTYP=", f) == 0) {
694316Svikram 			(void) strlcpy(fstype, buf + f, sizeof (fstype));
695316Svikram 			continue;
696316Svikram 		}
697316Svikram 	}
698316Svikram 	(void) fclose(fp);
699316Svikram 
700316Svikram 	if (dev[0] == '\0' || fstype[0] == '\0' || phys[0] == '\0') {
701316Svikram 		bam_error(BAD_SLICE_FILE, GRUB_slice);
702316Svikram 		return (NULL);
703316Svikram 	}
704316Svikram 
705316Svikram 	if (physlice) {
706316Svikram 		*physlice = s_strdup(phys);
707316Svikram 	}
708621Svikram 	if (logslice) {
709621Svikram 		*logslice = s_strdup(dev);
710621Svikram 	}
711621Svikram 	if (fs_type) {
712621Svikram 		*fs_type = s_strdup(fstype);
713621Svikram 	}
714316Svikram 
715316Svikram 	/*
716316Svikram 	 * Check if the slice is already mounted
717316Svikram 	 */
718316Svikram 	fp = fopen(MNTTAB, "r");
719316Svikram 	if (fp == NULL) {
720316Svikram 		bam_error(OPEN_FAIL, MNTTAB, strerror(errno));
721621Svikram 		goto error;
722316Svikram 	}
723316Svikram 
724316Svikram 	resetmnttab(fp);
725316Svikram 
726316Svikram 	mntpt = NULL;
727316Svikram 	while (getextmntent(fp, &mnt, sizeof (mnt)) == 0) {
728316Svikram 		if (strcmp(mnt.mnt_special, dev) == 0) {
729316Svikram 			mntpt = s_strdup(mnt.mnt_mountp);
730316Svikram 			break;
731316Svikram 		}
732316Svikram 	}
733316Svikram 
734316Svikram 	(void) fclose(fp);
735316Svikram 
736316Svikram 	if (mntpt) {
737316Svikram 		return (mntpt);
738316Svikram 	}
739316Svikram 
740316Svikram 
741316Svikram 	/*
742316Svikram 	 * GRUB slice is not mounted, we need to mount it now.
743316Svikram 	 * First create the mountpoint
744316Svikram 	 */
745316Svikram 	mntpt = s_calloc(1, PATH_MAX);
746316Svikram 	(void) snprintf(mntpt, PATH_MAX, "%s.%d", GRUB_slice_mntpt, getpid());
747316Svikram 	if (mkdir(mntpt, 0755) == -1 && errno != EEXIST) {
748316Svikram 		bam_error(MKDIR_FAILED, mntpt, strerror(errno));
749316Svikram 		free(mntpt);
750621Svikram 		goto error;
751316Svikram 	}
752316Svikram 
753348Svikram 	(void) snprintf(cmd, sizeof (cmd), "/sbin/mount -F %s %s %s",
754348Svikram 	    fstype, dev, mntpt);
755348Svikram 
756348Svikram 	if (exec_cmd(cmd, NULL, 0) != 0) {
757621Svikram 		bam_error(MOUNT_FAILED, dev, fstype);
758316Svikram 		if (rmdir(mntpt) != 0) {
759316Svikram 			bam_error(RMDIR_FAILED, mntpt, strerror(errno));
760316Svikram 		}
761316Svikram 		free(mntpt);
762621Svikram 		goto error;
763316Svikram 	}
764316Svikram 
765316Svikram 	*mnted = 1;
766316Svikram 	return (mntpt);
767621Svikram 
768621Svikram error:
769621Svikram 	if (physlice) {
770621Svikram 		free(*physlice);
771621Svikram 		*physlice = NULL;
772621Svikram 	}
773621Svikram 	if (logslice) {
774621Svikram 		free(*logslice);
775621Svikram 		*logslice = NULL;
776621Svikram 	}
777621Svikram 	if (fs_type) {
778621Svikram 		free(*fs_type);
779621Svikram 		*fs_type = NULL;
780621Svikram 	}
781621Svikram 	return (NULL);
782316Svikram }
783316Svikram 
784316Svikram static void
785621Svikram umount_grub_slice(
786621Svikram 	int mnted,
787621Svikram 	char *mntpt,
788621Svikram 	char *physlice,
789621Svikram 	char *logslice,
790621Svikram 	char *fs_type)
791316Svikram {
792348Svikram 	char cmd[PATH_MAX];
793348Svikram 
794316Svikram 	/*
795316Svikram 	 * If we have not dealt with GRUB slice
796316Svikram 	 * we have nothing to do - just return.
797316Svikram 	 */
798316Svikram 	if (mntpt == NULL)
799316Svikram 		return;
800316Svikram 
801316Svikram 
802316Svikram 	/*
803316Svikram 	 * If we mounted the filesystem earlier in mount_grub_slice()
804316Svikram 	 * unmount it now.
805316Svikram 	 */
806316Svikram 	if (mnted) {
807348Svikram 		(void) snprintf(cmd, sizeof (cmd), "/sbin/umount %s",
808348Svikram 		    mntpt);
809348Svikram 		if (exec_cmd(cmd, NULL, 0) != 0) {
810348Svikram 			bam_error(UMOUNT_FAILED, mntpt);
811316Svikram 		}
812316Svikram 		if (rmdir(mntpt) != 0) {
813316Svikram 			bam_error(RMDIR_FAILED, mntpt, strerror(errno));
814316Svikram 		}
815316Svikram 	}
816621Svikram 
817316Svikram 	if (physlice)
818316Svikram 		free(physlice);
819621Svikram 	if (logslice)
820621Svikram 		free(logslice);
821621Svikram 	if (fs_type)
822621Svikram 		free(fs_type);
823621Svikram 
824316Svikram 	free(mntpt);
825316Svikram }
826316Svikram 
827621Svikram static char *
828621Svikram use_stubboot(void)
829621Svikram {
830621Svikram 	int mnted;
831621Svikram 	struct stat sb;
832621Svikram 	struct extmnttab mnt;
833621Svikram 	FILE *fp;
834621Svikram 	char cmd[PATH_MAX];
835621Svikram 
836621Svikram 	if (stat(STUBBOOT, &sb) != 0) {
837621Svikram 		bam_error(STUBBOOT_DIR_NOT_FOUND);
838621Svikram 		return (NULL);
839621Svikram 	}
840621Svikram 
841621Svikram 	/*
842621Svikram 	 * Check if stubboot is mounted. If not, mount it
843621Svikram 	 */
844621Svikram 	fp = fopen(MNTTAB, "r");
845621Svikram 	if (fp == NULL) {
846621Svikram 		bam_error(OPEN_FAIL, MNTTAB, strerror(errno));
847621Svikram 		return (NULL);
848621Svikram 	}
849621Svikram 
850621Svikram 	resetmnttab(fp);
851621Svikram 
852621Svikram 	mnted = 0;
853621Svikram 	while (getextmntent(fp, &mnt, sizeof (mnt)) == 0) {
854621Svikram 		if (strcmp(mnt.mnt_mountp, STUBBOOT) == 0) {
855621Svikram 			mnted = 1;
856621Svikram 			break;
857621Svikram 		}
858621Svikram 	}
859621Svikram 
860621Svikram 	(void) fclose(fp);
861621Svikram 
862621Svikram 	if (mnted)
863621Svikram 		return (STUBBOOT);
864621Svikram 
865621Svikram 	/*
866621Svikram 	 * Stubboot is not mounted, mount it now.
867621Svikram 	 * It should exist in /etc/vfstab
868621Svikram 	 */
869621Svikram 	(void) snprintf(cmd, sizeof (cmd), "/sbin/mount %s",
870621Svikram 	    STUBBOOT);
871621Svikram 	if (exec_cmd(cmd, NULL, 0) != 0) {
872621Svikram 		bam_error(MOUNT_MNTPT_FAILED, STUBBOOT);
873621Svikram 		return (NULL);
874621Svikram 	}
875621Svikram 
876621Svikram 	return (STUBBOOT);
877621Svikram }
878621Svikram 
879621Svikram static void
880621Svikram disp_active_menu_locn(char *menu_path, char *logslice, char *fstype, int mnted)
881621Svikram {
882621Svikram 	/*
883621Svikram 	 * Check if we did a temp mount of an unmounted device.
884621Svikram 	 * If yes, print the block device and fstype for that device
885621Svikram 	 * else it is already mounted, so we print the path to the GRUB menu.
886621Svikram 	 */
887621Svikram 	if (mnted) {
888621Svikram 		bam_print(GRUB_MENU_DEVICE, logslice);
889621Svikram 		bam_print(GRUB_MENU_FSTYPE, fstype);
890621Svikram 	} else {
891621Svikram 		bam_print(GRUB_MENU_PATH, menu_path);
892621Svikram 	}
893621Svikram }
894621Svikram 
895621Svikram /*
896621Svikram  * NOTE: A single "/" is also considered a trailing slash and will
897621Svikram  * be deleted.
898621Svikram  */
899621Svikram static void
900621Svikram elide_trailing_slash(const char *src, char *dst, size_t dstsize)
901621Svikram {
902621Svikram 	size_t dstlen;
903621Svikram 
904621Svikram 	assert(src);
905621Svikram 	assert(dst);
906621Svikram 
907621Svikram 	(void) strlcpy(dst, src, dstsize);
908621Svikram 
909621Svikram 	dstlen = strlen(dst);
910621Svikram 	if (dst[dstlen - 1] == '/') {
911621Svikram 		dst[dstlen - 1] = '\0';
912621Svikram 	}
913621Svikram }
914621Svikram 
9150Sstevel@tonic-gate static error_t
9160Sstevel@tonic-gate bam_menu(char *subcmd, char *opt, int largc, char *largv[])
9170Sstevel@tonic-gate {
9180Sstevel@tonic-gate 	error_t ret;
9190Sstevel@tonic-gate 	char menu_path[PATH_MAX];
9200Sstevel@tonic-gate 	menu_t *menu;
921621Svikram 	char *mntpt, *menu_root, *logslice, *fstype;
922316Svikram 	struct stat sb;
923316Svikram 	int mnted;	/* set if we did a mount */
9240Sstevel@tonic-gate 	error_t (*f)(menu_t *mp, char *menu_path, char *opt);
9250Sstevel@tonic-gate 
9260Sstevel@tonic-gate 	/*
9270Sstevel@tonic-gate 	 * Check arguments
9280Sstevel@tonic-gate 	 */
9290Sstevel@tonic-gate 	ret = check_subcmd_and_options(subcmd, opt, menu_subcmds, &f);
9300Sstevel@tonic-gate 	if (ret == BAM_ERROR) {
9310Sstevel@tonic-gate 		return (BAM_ERROR);
9320Sstevel@tonic-gate 	}
9330Sstevel@tonic-gate 
934316Svikram 	mntpt = NULL;
935316Svikram 	mnted = 0;
936621Svikram 	logslice = fstype = NULL;
937621Svikram 
938621Svikram 	/*
939621Svikram 	 * If the user provides an alternate root, we
940621Svikram 	 * assume they know what they are doing and we
941621Svikram 	 * use it. Else we check if there is an
942621Svikram 	 * alternate location (other than /boot/grub)
943621Svikram 	 * for the GRUB menu
944621Svikram 	 */
945621Svikram 	if (bam_alt_root) {
946621Svikram 		menu_root = bam_root;
947621Svikram 	} else if (stat(GRUB_slice, &sb) == 0) {
948621Svikram 		mntpt = mount_grub_slice(&mnted, NULL, &logslice, &fstype);
949316Svikram 		menu_root = mntpt;
950621Svikram 	} else if (stat(STUBBOOT, &sb) == 0) {
951621Svikram 		menu_root = use_stubboot();
952316Svikram 	} else {
953316Svikram 		menu_root = bam_root;
954316Svikram 	}
955316Svikram 
956621Svikram 	if (menu_root == NULL) {
957621Svikram 		bam_error(CANNOT_LOCATE_GRUB_MENU);
958621Svikram 		return (BAM_ERROR);
959621Svikram 	}
960621Svikram 
961621Svikram 	elide_trailing_slash(menu_root, menu_path, sizeof (menu_path));
962621Svikram 	(void) strlcat(menu_path, GRUB_MENU, sizeof (menu_path));
963621Svikram 
964621Svikram 	/*
965621Svikram 	 * If listing the menu, display the active menu
966621Svikram 	 * location
967621Svikram 	 */
968621Svikram 	if (strcmp(subcmd, "list_entry") == 0) {
969621Svikram 		disp_active_menu_locn(menu_path, logslice, fstype, mnted);
970621Svikram 	}
9710Sstevel@tonic-gate 
9720Sstevel@tonic-gate 	menu = menu_read(menu_path);
9730Sstevel@tonic-gate 	assert(menu);
9740Sstevel@tonic-gate 
9750Sstevel@tonic-gate 	/*
9760Sstevel@tonic-gate 	 * Special handling for setting timeout and default
9770Sstevel@tonic-gate 	 */
9780Sstevel@tonic-gate 	if (strcmp(subcmd, "set_option") == 0) {
9790Sstevel@tonic-gate 		if (largc != 1 || largv[0] == NULL) {
9800Sstevel@tonic-gate 			usage();
981621Svikram 			menu_free(menu);
982621Svikram 			umount_grub_slice(mnted, mntpt, NULL, logslice, fstype);
9830Sstevel@tonic-gate 			return (BAM_ERROR);
9840Sstevel@tonic-gate 		}
9850Sstevel@tonic-gate 		opt = largv[0];
9860Sstevel@tonic-gate 	} else if (largc != 0) {
9870Sstevel@tonic-gate 		usage();
988621Svikram 		menu_free(menu);
989621Svikram 		umount_grub_slice(mnted, mntpt, NULL, logslice, fstype);
9900Sstevel@tonic-gate 		return (BAM_ERROR);
9910Sstevel@tonic-gate 	}
9920Sstevel@tonic-gate 
9930Sstevel@tonic-gate 	/*
9940Sstevel@tonic-gate 	 * Once the sub-cmd handler has run
9950Sstevel@tonic-gate 	 * only the line field is guaranteed to have valid values
9960Sstevel@tonic-gate 	 */
9970Sstevel@tonic-gate 	if (strcmp(subcmd, "update_entry") == 0)
9980Sstevel@tonic-gate 		ret = f(menu, bam_root, opt);
9990Sstevel@tonic-gate 	else
10000Sstevel@tonic-gate 		ret = f(menu, menu_path, opt);
10010Sstevel@tonic-gate 	if (ret == BAM_WRITE) {
1002316Svikram 		ret = menu_write(menu_root, menu);
10030Sstevel@tonic-gate 	}
10040Sstevel@tonic-gate 
10050Sstevel@tonic-gate 	menu_free(menu);
10060Sstevel@tonic-gate 
1007621Svikram 	umount_grub_slice(mnted, mntpt, NULL, logslice, fstype);
1008316Svikram 
10090Sstevel@tonic-gate 	return (ret);
10100Sstevel@tonic-gate }
10110Sstevel@tonic-gate 
10120Sstevel@tonic-gate 
10130Sstevel@tonic-gate static error_t
10140Sstevel@tonic-gate bam_archive(
10150Sstevel@tonic-gate 	char *subcmd,
10160Sstevel@tonic-gate 	char *opt)
10170Sstevel@tonic-gate {
10180Sstevel@tonic-gate 	error_t ret;
10190Sstevel@tonic-gate 	error_t (*f)(char *root, char *opt);
10200Sstevel@tonic-gate 
10210Sstevel@tonic-gate 	/*
1022*662Sszhou 	 * Add trailing / for archive subcommands
1023*662Sszhou 	 */
1024*662Sszhou 	if (rootbuf[strlen(rootbuf) - 1] != '/')
1025*662Sszhou 		(void) strcat(rootbuf, "/");
1026*662Sszhou 	bam_rootlen = strlen(rootbuf);
1027*662Sszhou 
1028*662Sszhou 	/*
10290Sstevel@tonic-gate 	 * Check arguments
10300Sstevel@tonic-gate 	 */
10310Sstevel@tonic-gate 	ret = check_subcmd_and_options(subcmd, opt, arch_subcmds, &f);
10320Sstevel@tonic-gate 	if (ret != BAM_SUCCESS) {
10330Sstevel@tonic-gate 		return (BAM_ERROR);
10340Sstevel@tonic-gate 	}
10350Sstevel@tonic-gate 
10360Sstevel@tonic-gate #if defined(__sparc)
10370Sstevel@tonic-gate 	/*
10380Sstevel@tonic-gate 	 * A NOP if called on SPARC during reboot
10390Sstevel@tonic-gate 	 */
10400Sstevel@tonic-gate 	if (strcmp(subcmd, "update_all") == 0)
10410Sstevel@tonic-gate 		return (BAM_SUCCESS);
10420Sstevel@tonic-gate 	else if (strcmp(subcmd, "update") != 0)
10430Sstevel@tonic-gate 		sparc_abort();
10440Sstevel@tonic-gate #endif
10450Sstevel@tonic-gate 
10460Sstevel@tonic-gate 	/*
10470Sstevel@tonic-gate 	 * Check archive not supported with update_all
10480Sstevel@tonic-gate 	 * since it is awkward to display out-of-sync
10490Sstevel@tonic-gate 	 * information for each BE.
10500Sstevel@tonic-gate 	 */
10510Sstevel@tonic-gate 	if (bam_check && strcmp(subcmd, "update_all") == 0) {
10520Sstevel@tonic-gate 		bam_error(CHECK_NOT_SUPPORTED, subcmd);
10530Sstevel@tonic-gate 		return (BAM_ERROR);
10540Sstevel@tonic-gate 	}
10550Sstevel@tonic-gate 
1056316Svikram 	if (strcmp(subcmd, "update_all") == 0)
1057316Svikram 		bam_update_all = 1;
1058316Svikram 
1059316Svikram 	ret = f(bam_root, opt);
1060316Svikram 
1061316Svikram 	bam_update_all = 0;
1062316Svikram 
1063316Svikram 	return (ret);
10640Sstevel@tonic-gate }
10650Sstevel@tonic-gate 
10660Sstevel@tonic-gate /*PRINTFLIKE1*/
10670Sstevel@tonic-gate static void
10680Sstevel@tonic-gate bam_error(char *format, ...)
10690Sstevel@tonic-gate {
10700Sstevel@tonic-gate 	va_list ap;
10710Sstevel@tonic-gate 
10720Sstevel@tonic-gate 	va_start(ap, format);
10730Sstevel@tonic-gate 	(void) fprintf(stderr, "%s: ", prog);
10740Sstevel@tonic-gate 	(void) vfprintf(stderr, format, ap);
10750Sstevel@tonic-gate 	va_end(ap);
10760Sstevel@tonic-gate }
10770Sstevel@tonic-gate 
10780Sstevel@tonic-gate /*PRINTFLIKE1*/
10790Sstevel@tonic-gate static void
10800Sstevel@tonic-gate bam_print(char *format, ...)
10810Sstevel@tonic-gate {
10820Sstevel@tonic-gate 	va_list ap;
10830Sstevel@tonic-gate 
10840Sstevel@tonic-gate 	va_start(ap, format);
10850Sstevel@tonic-gate 	(void) vfprintf(stdout, format, ap);
10860Sstevel@tonic-gate 	va_end(ap);
10870Sstevel@tonic-gate }
10880Sstevel@tonic-gate 
10890Sstevel@tonic-gate static void
10900Sstevel@tonic-gate bam_exit(int excode)
10910Sstevel@tonic-gate {
10920Sstevel@tonic-gate 	bam_unlock();
10930Sstevel@tonic-gate 	exit(excode);
10940Sstevel@tonic-gate }
10950Sstevel@tonic-gate 
10960Sstevel@tonic-gate static void
10970Sstevel@tonic-gate bam_lock(void)
10980Sstevel@tonic-gate {
10990Sstevel@tonic-gate 	struct flock lock;
11000Sstevel@tonic-gate 	pid_t pid;
11010Sstevel@tonic-gate 
11020Sstevel@tonic-gate 	bam_lock_fd = open(BAM_LOCK_FILE, O_CREAT|O_RDWR, LOCK_FILE_PERMS);
11030Sstevel@tonic-gate 	if (bam_lock_fd < 0) {
11040Sstevel@tonic-gate 		/*
11050Sstevel@tonic-gate 		 * We may be invoked early in boot for archive verification.
11060Sstevel@tonic-gate 		 * In this case, root is readonly and /var/run may not exist.
11070Sstevel@tonic-gate 		 * Proceed without the lock
11080Sstevel@tonic-gate 		 */
11090Sstevel@tonic-gate 		if (errno == EROFS || errno == ENOENT) {
11100Sstevel@tonic-gate 			bam_root_readonly = 1;
11110Sstevel@tonic-gate 			return;
11120Sstevel@tonic-gate 		}
11130Sstevel@tonic-gate 
11140Sstevel@tonic-gate 		bam_error(OPEN_FAIL, BAM_LOCK_FILE, strerror(errno));
11150Sstevel@tonic-gate 		bam_exit(1);
11160Sstevel@tonic-gate 	}
11170Sstevel@tonic-gate 
11180Sstevel@tonic-gate 	lock.l_type = F_WRLCK;
11190Sstevel@tonic-gate 	lock.l_whence = SEEK_SET;
11200Sstevel@tonic-gate 	lock.l_start = 0;
11210Sstevel@tonic-gate 	lock.l_len = 0;
11220Sstevel@tonic-gate 
11230Sstevel@tonic-gate 	if (fcntl(bam_lock_fd, F_SETLK, &lock) == -1) {
11240Sstevel@tonic-gate 		if (errno != EACCES && errno != EAGAIN) {
11250Sstevel@tonic-gate 			bam_error(LOCK_FAIL, BAM_LOCK_FILE, strerror(errno));
11260Sstevel@tonic-gate 			(void) close(bam_lock_fd);
11270Sstevel@tonic-gate 			bam_lock_fd = -1;
11280Sstevel@tonic-gate 			bam_exit(1);
11290Sstevel@tonic-gate 		}
11300Sstevel@tonic-gate 		pid = 0;
11310Sstevel@tonic-gate 		(void) pread(bam_lock_fd, &pid, sizeof (pid_t), 0);
11320Sstevel@tonic-gate 		bam_print(FILE_LOCKED, pid);
11330Sstevel@tonic-gate 
11340Sstevel@tonic-gate 		lock.l_type = F_WRLCK;
11350Sstevel@tonic-gate 		lock.l_whence = SEEK_SET;
11360Sstevel@tonic-gate 		lock.l_start = 0;
11370Sstevel@tonic-gate 		lock.l_len = 0;
11380Sstevel@tonic-gate 		if (fcntl(bam_lock_fd, F_SETLKW, &lock) == -1) {
11390Sstevel@tonic-gate 			bam_error(LOCK_FAIL, BAM_LOCK_FILE, strerror(errno));
11400Sstevel@tonic-gate 			(void) close(bam_lock_fd);
11410Sstevel@tonic-gate 			bam_lock_fd = -1;
11420Sstevel@tonic-gate 			bam_exit(1);
11430Sstevel@tonic-gate 		}
11440Sstevel@tonic-gate 	}
11450Sstevel@tonic-gate 
11460Sstevel@tonic-gate 	/* We own the lock now */
11470Sstevel@tonic-gate 	pid = getpid();
11480Sstevel@tonic-gate 	(void) write(bam_lock_fd, &pid, sizeof (pid));
11490Sstevel@tonic-gate }
11500Sstevel@tonic-gate 
11510Sstevel@tonic-gate static void
11520Sstevel@tonic-gate bam_unlock(void)
11530Sstevel@tonic-gate {
11540Sstevel@tonic-gate 	struct flock unlock;
11550Sstevel@tonic-gate 
11560Sstevel@tonic-gate 	/*
11570Sstevel@tonic-gate 	 * NOP if we don't hold the lock
11580Sstevel@tonic-gate 	 */
11590Sstevel@tonic-gate 	if (bam_lock_fd < 0) {
11600Sstevel@tonic-gate 		return;
11610Sstevel@tonic-gate 	}
11620Sstevel@tonic-gate 
11630Sstevel@tonic-gate 	unlock.l_type = F_UNLCK;
11640Sstevel@tonic-gate 	unlock.l_whence = SEEK_SET;
11650Sstevel@tonic-gate 	unlock.l_start = 0;
11660Sstevel@tonic-gate 	unlock.l_len = 0;
11670Sstevel@tonic-gate 
11680Sstevel@tonic-gate 	if (fcntl(bam_lock_fd, F_SETLK, &unlock) == -1) {
11690Sstevel@tonic-gate 		bam_error(UNLOCK_FAIL, BAM_LOCK_FILE, strerror(errno));
11700Sstevel@tonic-gate 	}
11710Sstevel@tonic-gate 
11720Sstevel@tonic-gate 	if (close(bam_lock_fd) == -1) {
11730Sstevel@tonic-gate 		bam_error(CLOSE_FAIL, BAM_LOCK_FILE, strerror(errno));
11740Sstevel@tonic-gate 	}
11750Sstevel@tonic-gate 	bam_lock_fd = -1;
11760Sstevel@tonic-gate }
11770Sstevel@tonic-gate 
11780Sstevel@tonic-gate static error_t
11790Sstevel@tonic-gate list_archive(char *root, char *opt)
11800Sstevel@tonic-gate {
11810Sstevel@tonic-gate 	filelist_t flist;
11820Sstevel@tonic-gate 	filelist_t *flistp = &flist;
11830Sstevel@tonic-gate 	line_t *lp;
11840Sstevel@tonic-gate 
11850Sstevel@tonic-gate 	assert(root);
11860Sstevel@tonic-gate 	assert(opt == NULL);
11870Sstevel@tonic-gate 
11880Sstevel@tonic-gate 	flistp->head = flistp->tail = NULL;
11890Sstevel@tonic-gate 	if (read_list(root, flistp) != BAM_SUCCESS) {
11900Sstevel@tonic-gate 		return (BAM_ERROR);
11910Sstevel@tonic-gate 	}
11920Sstevel@tonic-gate 	assert(flistp->head && flistp->tail);
11930Sstevel@tonic-gate 
11940Sstevel@tonic-gate 	for (lp = flistp->head; lp; lp = lp->next) {
11950Sstevel@tonic-gate 		bam_print(PRINT, lp->line);
11960Sstevel@tonic-gate 	}
11970Sstevel@tonic-gate 
11980Sstevel@tonic-gate 	filelist_free(flistp);
11990Sstevel@tonic-gate 
12000Sstevel@tonic-gate 	return (BAM_SUCCESS);
12010Sstevel@tonic-gate }
12020Sstevel@tonic-gate 
12030Sstevel@tonic-gate /*
12040Sstevel@tonic-gate  * This routine writes a list of lines to a file.
12050Sstevel@tonic-gate  * The list is *not* freed
12060Sstevel@tonic-gate  */
12070Sstevel@tonic-gate static error_t
12080Sstevel@tonic-gate list2file(char *root, char *tmp, char *final, line_t *start)
12090Sstevel@tonic-gate {
12100Sstevel@tonic-gate 	char tmpfile[PATH_MAX];
12110Sstevel@tonic-gate 	char path[PATH_MAX];
12120Sstevel@tonic-gate 	FILE *fp;
12130Sstevel@tonic-gate 	int ret;
12140Sstevel@tonic-gate 	struct stat sb;
12150Sstevel@tonic-gate 	mode_t mode;
12160Sstevel@tonic-gate 	uid_t root_uid;
12170Sstevel@tonic-gate 	gid_t sys_gid;
12180Sstevel@tonic-gate 	struct passwd *pw;
12190Sstevel@tonic-gate 	struct group *gp;
12200Sstevel@tonic-gate 
12210Sstevel@tonic-gate 
12220Sstevel@tonic-gate 	(void) snprintf(path, sizeof (path), "%s%s", root, final);
12230Sstevel@tonic-gate 
12240Sstevel@tonic-gate 	if (start == NULL) {
12250Sstevel@tonic-gate 		if (stat(path, &sb) != -1) {
12260Sstevel@tonic-gate 			bam_print(UNLINK_EMPTY, path);
12270Sstevel@tonic-gate 			if (unlink(path) != 0) {
12280Sstevel@tonic-gate 				bam_error(UNLINK_FAIL, path, strerror(errno));
12290Sstevel@tonic-gate 				return (BAM_ERROR);
12300Sstevel@tonic-gate 			} else {
12310Sstevel@tonic-gate 				return (BAM_SUCCESS);
12320Sstevel@tonic-gate 			}
12330Sstevel@tonic-gate 		}
12340Sstevel@tonic-gate 	}
12350Sstevel@tonic-gate 
12360Sstevel@tonic-gate 	/*
12370Sstevel@tonic-gate 	 * Preserve attributes of existing file if possible,
12380Sstevel@tonic-gate 	 * otherwise ask the system for uid/gid of root/sys.
12390Sstevel@tonic-gate 	 * If all fails, fall back on hard-coded defaults.
12400Sstevel@tonic-gate 	 */
12410Sstevel@tonic-gate 	if (stat(path, &sb) != -1) {
12420Sstevel@tonic-gate 		mode = sb.st_mode;
12430Sstevel@tonic-gate 		root_uid = sb.st_uid;
12440Sstevel@tonic-gate 		sys_gid = sb.st_gid;
12450Sstevel@tonic-gate 	} else {
12460Sstevel@tonic-gate 		mode = DEFAULT_DEV_MODE;
12470Sstevel@tonic-gate 		if ((pw = getpwnam(DEFAULT_DEV_USER)) != NULL) {
12480Sstevel@tonic-gate 			root_uid = pw->pw_uid;
12490Sstevel@tonic-gate 		} else {
12500Sstevel@tonic-gate 			if (bam_verbose)
12510Sstevel@tonic-gate 				bam_error(CANT_FIND_USER,
12520Sstevel@tonic-gate 				    DEFAULT_DEV_USER, DEFAULT_DEV_UID);
12530Sstevel@tonic-gate 			root_uid = (uid_t)DEFAULT_DEV_UID;
12540Sstevel@tonic-gate 		}
12550Sstevel@tonic-gate 		if ((gp = getgrnam(DEFAULT_DEV_GROUP)) != NULL) {
12560Sstevel@tonic-gate 			sys_gid = gp->gr_gid;
12570Sstevel@tonic-gate 		} else {
12580Sstevel@tonic-gate 			if (bam_verbose)
12590Sstevel@tonic-gate 				bam_error(CANT_FIND_GROUP,
12600Sstevel@tonic-gate 				    DEFAULT_DEV_GROUP, DEFAULT_DEV_GID);
12610Sstevel@tonic-gate 			sys_gid = (gid_t)DEFAULT_DEV_GID;
12620Sstevel@tonic-gate 		}
12630Sstevel@tonic-gate 	}
12640Sstevel@tonic-gate 
12650Sstevel@tonic-gate 	(void) snprintf(tmpfile, sizeof (tmpfile), "%s%s", root, tmp);
12660Sstevel@tonic-gate 
12670Sstevel@tonic-gate 	/* Truncate tmpfile first */
12680Sstevel@tonic-gate 	fp = fopen(tmpfile, "w");
12690Sstevel@tonic-gate 	if (fp == NULL) {
12700Sstevel@tonic-gate 		bam_error(OPEN_FAIL, tmpfile, strerror(errno));
12710Sstevel@tonic-gate 		return (BAM_ERROR);
12720Sstevel@tonic-gate 	}
12730Sstevel@tonic-gate 	ret = fclose(fp);
12740Sstevel@tonic-gate 	if (ret == EOF) {
12750Sstevel@tonic-gate 		bam_error(CLOSE_FAIL, tmpfile, strerror(errno));
12760Sstevel@tonic-gate 		return (BAM_ERROR);
12770Sstevel@tonic-gate 	}
12780Sstevel@tonic-gate 
12790Sstevel@tonic-gate 	/* Now open it in append mode */
12800Sstevel@tonic-gate 	fp = fopen(tmpfile, "a");
12810Sstevel@tonic-gate 	if (fp == NULL) {
12820Sstevel@tonic-gate 		bam_error(OPEN_FAIL, tmpfile, strerror(errno));
12830Sstevel@tonic-gate 		return (BAM_ERROR);
12840Sstevel@tonic-gate 	}
12850Sstevel@tonic-gate 
12860Sstevel@tonic-gate 	for (; start; start = start->next) {
12870Sstevel@tonic-gate 		ret = s_fputs(start->line, fp);
12880Sstevel@tonic-gate 		if (ret == EOF) {
12890Sstevel@tonic-gate 			bam_error(WRITE_FAIL, tmpfile, strerror(errno));
12900Sstevel@tonic-gate 			(void) fclose(fp);
12910Sstevel@tonic-gate 			return (BAM_ERROR);
12920Sstevel@tonic-gate 		}
12930Sstevel@tonic-gate 	}
12940Sstevel@tonic-gate 
12950Sstevel@tonic-gate 	ret = fclose(fp);
12960Sstevel@tonic-gate 	if (ret == EOF) {
12970Sstevel@tonic-gate 		bam_error(CLOSE_FAIL, tmpfile, strerror(errno));
12980Sstevel@tonic-gate 		return (BAM_ERROR);
12990Sstevel@tonic-gate 	}
13000Sstevel@tonic-gate 
13010Sstevel@tonic-gate 	/*
1302271Sjg 	 * Set up desired attributes.  Ignore failures on filesystems
1303271Sjg 	 * not supporting these operations - pcfs reports unsupported
1304271Sjg 	 * operations as EINVAL.
13050Sstevel@tonic-gate 	 */
13060Sstevel@tonic-gate 	ret = chmod(tmpfile, mode);
1307271Sjg 	if (ret == -1 &&
1308271Sjg 	    errno != EINVAL && errno != ENOTSUP) {
13090Sstevel@tonic-gate 		bam_error(CHMOD_FAIL, tmpfile, strerror(errno));
13100Sstevel@tonic-gate 		return (BAM_ERROR);
13110Sstevel@tonic-gate 	}
13120Sstevel@tonic-gate 
13130Sstevel@tonic-gate 	ret = chown(tmpfile, root_uid, sys_gid);
1314271Sjg 	if (ret == -1 &&
1315271Sjg 	    errno != EINVAL && errno != ENOTSUP) {
13160Sstevel@tonic-gate 		bam_error(CHOWN_FAIL, tmpfile, strerror(errno));
13170Sstevel@tonic-gate 		return (BAM_ERROR);
13180Sstevel@tonic-gate 	}
13190Sstevel@tonic-gate 
13200Sstevel@tonic-gate 
13210Sstevel@tonic-gate 	/*
13220Sstevel@tonic-gate 	 * Do an atomic rename
13230Sstevel@tonic-gate 	 */
13240Sstevel@tonic-gate 	ret = rename(tmpfile, path);
13250Sstevel@tonic-gate 	if (ret != 0) {
13260Sstevel@tonic-gate 		bam_error(RENAME_FAIL, path, strerror(errno));
13270Sstevel@tonic-gate 		return (BAM_ERROR);
13280Sstevel@tonic-gate 	}
13290Sstevel@tonic-gate 
13300Sstevel@tonic-gate 	return (BAM_SUCCESS);
13310Sstevel@tonic-gate }
13320Sstevel@tonic-gate 
13330Sstevel@tonic-gate /*
13340Sstevel@tonic-gate  * This function should always return 0 - since we want
13350Sstevel@tonic-gate  * to create stat data for *all* files in the list.
13360Sstevel@tonic-gate  */
13370Sstevel@tonic-gate /*ARGSUSED*/
13380Sstevel@tonic-gate static int
13390Sstevel@tonic-gate cmpstat(
13400Sstevel@tonic-gate 	const char *file,
13410Sstevel@tonic-gate 	const struct stat *stat,
13420Sstevel@tonic-gate 	int flags,
13430Sstevel@tonic-gate 	struct FTW *ftw)
13440Sstevel@tonic-gate {
13450Sstevel@tonic-gate 	uint_t sz;
13460Sstevel@tonic-gate 	uint64_t *value;
13470Sstevel@tonic-gate 	uint64_t filestat[2];
13480Sstevel@tonic-gate 	int error;
13490Sstevel@tonic-gate 
13500Sstevel@tonic-gate 	/*
13510Sstevel@tonic-gate 	 * We only want regular files
13520Sstevel@tonic-gate 	 */
13530Sstevel@tonic-gate 	if (!S_ISREG(stat->st_mode))
13540Sstevel@tonic-gate 		return (0);
13550Sstevel@tonic-gate 
13560Sstevel@tonic-gate 	/*
13570Sstevel@tonic-gate 	 * new_nvlp may be NULL if there were errors earlier
13580Sstevel@tonic-gate 	 * but this is not fatal to update determination.
13590Sstevel@tonic-gate 	 */
13600Sstevel@tonic-gate 	if (walk_arg.new_nvlp) {
13610Sstevel@tonic-gate 		filestat[0] = stat->st_size;
13620Sstevel@tonic-gate 		filestat[1] = stat->st_mtime;
13630Sstevel@tonic-gate 		error = nvlist_add_uint64_array(walk_arg.new_nvlp,
13640Sstevel@tonic-gate 		    file + bam_rootlen, filestat, 2);
13650Sstevel@tonic-gate 		if (error)
13660Sstevel@tonic-gate 			bam_error(NVADD_FAIL, file, strerror(error));
13670Sstevel@tonic-gate 	}
13680Sstevel@tonic-gate 
13690Sstevel@tonic-gate 	/*
13700Sstevel@tonic-gate 	 * The remaining steps are only required if we haven't made a
13710Sstevel@tonic-gate 	 * decision about update or if we are checking (-n)
13720Sstevel@tonic-gate 	 */
13730Sstevel@tonic-gate 	if (walk_arg.need_update && !bam_check)
13740Sstevel@tonic-gate 		return (0);
13750Sstevel@tonic-gate 
13760Sstevel@tonic-gate 	/*
13770Sstevel@tonic-gate 	 * If we are invoked as part of system/filesyste/boot-archive
13780Sstevel@tonic-gate 	 * SMF service, ignore amd64 modules unless we are booted amd64.
13790Sstevel@tonic-gate 	 */
1380*662Sszhou 	if (bam_smf_check && !is_amd64() && strstr(file, "/amd64/") != 0)
13810Sstevel@tonic-gate 		return (0);
13820Sstevel@tonic-gate 
13830Sstevel@tonic-gate 	/*
13840Sstevel@tonic-gate 	 * We need an update if file doesn't exist in old archive
13850Sstevel@tonic-gate 	 */
13860Sstevel@tonic-gate 	if (walk_arg.old_nvlp == NULL ||
13870Sstevel@tonic-gate 	    nvlist_lookup_uint64_array(walk_arg.old_nvlp,
13880Sstevel@tonic-gate 	    file + bam_rootlen, &value, &sz) != 0) {
13890Sstevel@tonic-gate 		if (bam_smf_check)	/* ignore new during smf check */
13900Sstevel@tonic-gate 			return (0);
13910Sstevel@tonic-gate 		walk_arg.need_update = 1;
13920Sstevel@tonic-gate 		if (bam_verbose)
13930Sstevel@tonic-gate 			bam_print(PARSEABLE_NEW_FILE, file);
13940Sstevel@tonic-gate 		return (0);
13950Sstevel@tonic-gate 	}
13960Sstevel@tonic-gate 
13970Sstevel@tonic-gate 	/*
13980Sstevel@tonic-gate 	 * File exists in old archive. Check if file has changed
13990Sstevel@tonic-gate 	 */
14000Sstevel@tonic-gate 	assert(sz == 2);
14010Sstevel@tonic-gate 	bcopy(value, filestat, sizeof (filestat));
14020Sstevel@tonic-gate 
14030Sstevel@tonic-gate 	if (filestat[0] != stat->st_size ||
14040Sstevel@tonic-gate 	    filestat[1] != stat->st_mtime) {
14050Sstevel@tonic-gate 		walk_arg.need_update = 1;
14060Sstevel@tonic-gate 		if (bam_verbose)
14070Sstevel@tonic-gate 			if (bam_smf_check)
14080Sstevel@tonic-gate 				bam_print("    %s\n", file);
14090Sstevel@tonic-gate 			else
14100Sstevel@tonic-gate 				bam_print(PARSEABLE_OUT_DATE, file);
14110Sstevel@tonic-gate 	}
14120Sstevel@tonic-gate 
14130Sstevel@tonic-gate 	return (0);
14140Sstevel@tonic-gate }
14150Sstevel@tonic-gate 
14160Sstevel@tonic-gate /*
14170Sstevel@tonic-gate  * Check flags and presence of required files.
14180Sstevel@tonic-gate  * The force flag and/or absence of files should
14190Sstevel@tonic-gate  * trigger an update.
14200Sstevel@tonic-gate  * Suppress stdout output if check (-n) option is set
14210Sstevel@tonic-gate  * (as -n should only produce parseable output.)
14220Sstevel@tonic-gate  */
14230Sstevel@tonic-gate static void
14240Sstevel@tonic-gate check_flags_and_files(char *root)
14250Sstevel@tonic-gate {
14260Sstevel@tonic-gate 	char path[PATH_MAX];
14270Sstevel@tonic-gate 	struct stat sb;
14280Sstevel@tonic-gate 
14290Sstevel@tonic-gate 	/*
14300Sstevel@tonic-gate 	 * if force, create archive unconditionally
14310Sstevel@tonic-gate 	 */
14320Sstevel@tonic-gate 	if (bam_force) {
14330Sstevel@tonic-gate 		walk_arg.need_update = 1;
14340Sstevel@tonic-gate 		if (bam_verbose && !bam_check)
14350Sstevel@tonic-gate 			bam_print(UPDATE_FORCE);
14360Sstevel@tonic-gate 		return;
14370Sstevel@tonic-gate 	}
14380Sstevel@tonic-gate 
14390Sstevel@tonic-gate 	/*
14400Sstevel@tonic-gate 	 * If archive is missing, create archive
14410Sstevel@tonic-gate 	 */
14420Sstevel@tonic-gate 	(void) snprintf(path, sizeof (path), "%s%s", root, BOOT_ARCHIVE);
14430Sstevel@tonic-gate 	if (stat(path, &sb) != 0) {
14440Sstevel@tonic-gate 		if (bam_verbose && !bam_check)
14450Sstevel@tonic-gate 			bam_print(UPDATE_ARCH_MISS, path);
14460Sstevel@tonic-gate 		walk_arg.need_update = 1;
14470Sstevel@tonic-gate 		return;
14480Sstevel@tonic-gate 	}
14490Sstevel@tonic-gate }
14500Sstevel@tonic-gate 
14510Sstevel@tonic-gate static error_t
14520Sstevel@tonic-gate read_one_list(char *root, filelist_t  *flistp, char *filelist)
14530Sstevel@tonic-gate {
14540Sstevel@tonic-gate 	char path[PATH_MAX];
14550Sstevel@tonic-gate 	FILE *fp;
14560Sstevel@tonic-gate 	char buf[BAM_MAXLINE];
14570Sstevel@tonic-gate 
14580Sstevel@tonic-gate 	(void) snprintf(path, sizeof (path), "%s%s", root, filelist);
14590Sstevel@tonic-gate 
14600Sstevel@tonic-gate 	fp = fopen(path, "r");
14610Sstevel@tonic-gate 	if (fp == NULL) {
14620Sstevel@tonic-gate 		if (bam_debug)
14630Sstevel@tonic-gate 			bam_error(FLIST_FAIL, path, strerror(errno));
14640Sstevel@tonic-gate 		return (BAM_ERROR);
14650Sstevel@tonic-gate 	}
14660Sstevel@tonic-gate 	while (s_fgets(buf, sizeof (buf), fp) != NULL) {
1467316Svikram 		/* skip blank lines */
1468316Svikram 		if (strspn(buf, " \t") == strlen(buf))
1469316Svikram 			continue;
14700Sstevel@tonic-gate 		append_to_flist(flistp, buf);
14710Sstevel@tonic-gate 	}
14720Sstevel@tonic-gate 	if (fclose(fp) != 0) {
14730Sstevel@tonic-gate 		bam_error(CLOSE_FAIL, path, strerror(errno));
14740Sstevel@tonic-gate 		return (BAM_ERROR);
14750Sstevel@tonic-gate 	}
14760Sstevel@tonic-gate 	return (BAM_SUCCESS);
14770Sstevel@tonic-gate }
14780Sstevel@tonic-gate 
14790Sstevel@tonic-gate static error_t
14800Sstevel@tonic-gate read_list(char *root, filelist_t  *flistp)
14810Sstevel@tonic-gate {
14820Sstevel@tonic-gate 	int rval;
14830Sstevel@tonic-gate 
14840Sstevel@tonic-gate 	flistp->head = flistp->tail = NULL;
14850Sstevel@tonic-gate 
14860Sstevel@tonic-gate 	/*
14870Sstevel@tonic-gate 	 * Read current lists of files - only the first is mandatory
14880Sstevel@tonic-gate 	 */
14890Sstevel@tonic-gate 	rval = read_one_list(root, flistp, BOOT_FILE_LIST);
14900Sstevel@tonic-gate 	if (rval != BAM_SUCCESS)
14910Sstevel@tonic-gate 		return (rval);
14920Sstevel@tonic-gate 	(void) read_one_list(root, flistp, ETC_FILE_LIST);
14930Sstevel@tonic-gate 
14940Sstevel@tonic-gate 	if (flistp->head == NULL) {
14950Sstevel@tonic-gate 		bam_error(NO_FLIST);
14960Sstevel@tonic-gate 		return (BAM_ERROR);
14970Sstevel@tonic-gate 	}
14980Sstevel@tonic-gate 
14990Sstevel@tonic-gate 	return (BAM_SUCCESS);
15000Sstevel@tonic-gate }
15010Sstevel@tonic-gate 
15020Sstevel@tonic-gate static void
15030Sstevel@tonic-gate getoldstat(char *root)
15040Sstevel@tonic-gate {
15050Sstevel@tonic-gate 	char path[PATH_MAX];
15060Sstevel@tonic-gate 	int fd, error;
15070Sstevel@tonic-gate 	struct stat sb;
15080Sstevel@tonic-gate 	char *ostat;
15090Sstevel@tonic-gate 
15100Sstevel@tonic-gate 	(void) snprintf(path, sizeof (path), "%s%s", root, FILE_STAT);
15110Sstevel@tonic-gate 	fd = open(path, O_RDONLY);
15120Sstevel@tonic-gate 	if (fd == -1) {
15130Sstevel@tonic-gate 		if (bam_verbose)
15140Sstevel@tonic-gate 			bam_print(OPEN_FAIL, path, strerror(errno));
15150Sstevel@tonic-gate 		walk_arg.need_update = 1;
15160Sstevel@tonic-gate 		return;
15170Sstevel@tonic-gate 	}
15180Sstevel@tonic-gate 
15190Sstevel@tonic-gate 	if (fstat(fd, &sb) != 0) {
15200Sstevel@tonic-gate 		bam_error(STAT_FAIL, path, strerror(errno));
15210Sstevel@tonic-gate 		(void) close(fd);
15220Sstevel@tonic-gate 		walk_arg.need_update = 1;
15230Sstevel@tonic-gate 		return;
15240Sstevel@tonic-gate 	}
15250Sstevel@tonic-gate 
15260Sstevel@tonic-gate 	ostat = s_calloc(1, sb.st_size);
15270Sstevel@tonic-gate 
15280Sstevel@tonic-gate 	if (read(fd, ostat, sb.st_size) != sb.st_size) {
15290Sstevel@tonic-gate 		bam_error(READ_FAIL, path, strerror(errno));
15300Sstevel@tonic-gate 		(void) close(fd);
15310Sstevel@tonic-gate 		free(ostat);
15320Sstevel@tonic-gate 		walk_arg.need_update = 1;
15330Sstevel@tonic-gate 		return;
15340Sstevel@tonic-gate 	}
15350Sstevel@tonic-gate 
15360Sstevel@tonic-gate 	(void) close(fd);
15370Sstevel@tonic-gate 
15380Sstevel@tonic-gate 	walk_arg.old_nvlp = NULL;
15390Sstevel@tonic-gate 	error = nvlist_unpack(ostat, sb.st_size, &walk_arg.old_nvlp, 0);
15400Sstevel@tonic-gate 
15410Sstevel@tonic-gate 	free(ostat);
15420Sstevel@tonic-gate 
15430Sstevel@tonic-gate 	if (error) {
15440Sstevel@tonic-gate 		bam_error(UNPACK_FAIL, path, strerror(error));
15450Sstevel@tonic-gate 		walk_arg.old_nvlp = NULL;
15460Sstevel@tonic-gate 		walk_arg.need_update = 1;
15470Sstevel@tonic-gate 		return;
15480Sstevel@tonic-gate 	}
15490Sstevel@tonic-gate }
15500Sstevel@tonic-gate 
15510Sstevel@tonic-gate static void
15520Sstevel@tonic-gate create_newstat(void)
15530Sstevel@tonic-gate {
15540Sstevel@tonic-gate 	int error;
15550Sstevel@tonic-gate 
15560Sstevel@tonic-gate 	error = nvlist_alloc(&walk_arg.new_nvlp, NV_UNIQUE_NAME, 0);
15570Sstevel@tonic-gate 	if (error) {
15580Sstevel@tonic-gate 		/*
15590Sstevel@tonic-gate 		 * Not fatal - we can still create archive
15600Sstevel@tonic-gate 		 */
15610Sstevel@tonic-gate 		walk_arg.new_nvlp = NULL;
15620Sstevel@tonic-gate 		bam_error(NVALLOC_FAIL, strerror(error));
15630Sstevel@tonic-gate 	}
15640Sstevel@tonic-gate }
15650Sstevel@tonic-gate 
15660Sstevel@tonic-gate static void
15670Sstevel@tonic-gate walk_list(char *root, filelist_t *flistp)
15680Sstevel@tonic-gate {
15690Sstevel@tonic-gate 	char path[PATH_MAX];
15700Sstevel@tonic-gate 	line_t *lp;
15710Sstevel@tonic-gate 
15720Sstevel@tonic-gate 	for (lp = flistp->head; lp; lp = lp->next) {
15730Sstevel@tonic-gate 		(void) snprintf(path, sizeof (path), "%s%s", root, lp->line);
15740Sstevel@tonic-gate 		/* XXX shouldn't we use FTW_MOUNT ? */
15750Sstevel@tonic-gate 		if (nftw(path, cmpstat, 20, 0) == -1) {
15760Sstevel@tonic-gate 			/*
15770Sstevel@tonic-gate 			 * Some files may not exist.
15780Sstevel@tonic-gate 			 * For example: etc/rtc_config on a x86 diskless system
15790Sstevel@tonic-gate 			 * Emit verbose message only
15800Sstevel@tonic-gate 			 */
15810Sstevel@tonic-gate 			if (bam_verbose)
15820Sstevel@tonic-gate 				bam_print(NFTW_FAIL, path, strerror(errno));
15830Sstevel@tonic-gate 		}
15840Sstevel@tonic-gate 	}
15850Sstevel@tonic-gate }
15860Sstevel@tonic-gate 
15870Sstevel@tonic-gate static void
15880Sstevel@tonic-gate savenew(char *root)
15890Sstevel@tonic-gate {
15900Sstevel@tonic-gate 	char path[PATH_MAX];
15910Sstevel@tonic-gate 	char path2[PATH_MAX];
15920Sstevel@tonic-gate 	size_t sz;
15930Sstevel@tonic-gate 	char *nstat;
15940Sstevel@tonic-gate 	int fd, wrote, error;
15950Sstevel@tonic-gate 
15960Sstevel@tonic-gate 	nstat = NULL;
15970Sstevel@tonic-gate 	sz = 0;
15980Sstevel@tonic-gate 	error = nvlist_pack(walk_arg.new_nvlp, &nstat, &sz,
15990Sstevel@tonic-gate 	    NV_ENCODE_XDR, 0);
16000Sstevel@tonic-gate 	if (error) {
16010Sstevel@tonic-gate 		bam_error(PACK_FAIL, strerror(error));
16020Sstevel@tonic-gate 		return;
16030Sstevel@tonic-gate 	}
16040Sstevel@tonic-gate 
16050Sstevel@tonic-gate 	(void) snprintf(path, sizeof (path), "%s%s", root, FILE_STAT_TMP);
16060Sstevel@tonic-gate 	fd = open(path, O_RDWR|O_CREAT|O_TRUNC, FILE_STAT_MODE);
16070Sstevel@tonic-gate 	if (fd == -1) {
16080Sstevel@tonic-gate 		bam_error(OPEN_FAIL, path, strerror(errno));
16090Sstevel@tonic-gate 		free(nstat);
16100Sstevel@tonic-gate 		return;
16110Sstevel@tonic-gate 	}
16120Sstevel@tonic-gate 	wrote = write(fd, nstat, sz);
16130Sstevel@tonic-gate 	if (wrote != sz) {
16140Sstevel@tonic-gate 		bam_error(WRITE_FAIL, path, strerror(errno));
16150Sstevel@tonic-gate 		(void) close(fd);
16160Sstevel@tonic-gate 		free(nstat);
16170Sstevel@tonic-gate 		return;
16180Sstevel@tonic-gate 	}
16190Sstevel@tonic-gate 	(void) close(fd);
16200Sstevel@tonic-gate 	free(nstat);
16210Sstevel@tonic-gate 
16220Sstevel@tonic-gate 	(void) snprintf(path2, sizeof (path2), "%s%s", root, FILE_STAT);
16230Sstevel@tonic-gate 	if (rename(path, path2) != 0) {
16240Sstevel@tonic-gate 		bam_error(RENAME_FAIL, path2, strerror(errno));
16250Sstevel@tonic-gate 	}
16260Sstevel@tonic-gate }
16270Sstevel@tonic-gate 
16280Sstevel@tonic-gate static void
16290Sstevel@tonic-gate clear_walk_args(void)
16300Sstevel@tonic-gate {
16310Sstevel@tonic-gate 	if (walk_arg.old_nvlp)
16320Sstevel@tonic-gate 		nvlist_free(walk_arg.old_nvlp);
16330Sstevel@tonic-gate 	if (walk_arg.new_nvlp)
16340Sstevel@tonic-gate 		nvlist_free(walk_arg.new_nvlp);
16350Sstevel@tonic-gate 	walk_arg.need_update = 0;
16360Sstevel@tonic-gate 	walk_arg.old_nvlp = NULL;
16370Sstevel@tonic-gate 	walk_arg.new_nvlp = NULL;
16380Sstevel@tonic-gate }
16390Sstevel@tonic-gate 
16400Sstevel@tonic-gate /*
16410Sstevel@tonic-gate  * Returns:
16420Sstevel@tonic-gate  *	0 - no update necessary
16430Sstevel@tonic-gate  *	1 - update required.
16440Sstevel@tonic-gate  *	BAM_ERROR (-1) - An error occurred
16450Sstevel@tonic-gate  *
16460Sstevel@tonic-gate  * Special handling for check (-n):
16470Sstevel@tonic-gate  * ================================
16480Sstevel@tonic-gate  * The check (-n) option produces parseable output.
16490Sstevel@tonic-gate  * To do this, we suppress all stdout messages unrelated
16500Sstevel@tonic-gate  * to out of sync files.
16510Sstevel@tonic-gate  * All stderr messages are still printed though.
16520Sstevel@tonic-gate  *
16530Sstevel@tonic-gate  */
16540Sstevel@tonic-gate static int
16550Sstevel@tonic-gate update_required(char *root)
16560Sstevel@tonic-gate {
16570Sstevel@tonic-gate 	struct stat sb;
16580Sstevel@tonic-gate 	char path[PATH_MAX];
16590Sstevel@tonic-gate 	filelist_t flist;
16600Sstevel@tonic-gate 	filelist_t *flistp = &flist;
16610Sstevel@tonic-gate 	int need_update;
16620Sstevel@tonic-gate 
16630Sstevel@tonic-gate 	flistp->head = flistp->tail = NULL;
16640Sstevel@tonic-gate 
16650Sstevel@tonic-gate 	walk_arg.need_update = 0;
16660Sstevel@tonic-gate 
16670Sstevel@tonic-gate 	/*
16680Sstevel@tonic-gate 	 * Without consulting stat data, check if we need update
16690Sstevel@tonic-gate 	 */
16700Sstevel@tonic-gate 	check_flags_and_files(root);
16710Sstevel@tonic-gate 
16720Sstevel@tonic-gate 	/*
16730Sstevel@tonic-gate 	 * In certain deployment scenarios, filestat may not
16740Sstevel@tonic-gate 	 * exist. Ignore it during boot-archive SMF check.
16750Sstevel@tonic-gate 	 */
16760Sstevel@tonic-gate 	if (bam_smf_check) {
16770Sstevel@tonic-gate 		(void) snprintf(path, sizeof (path), "%s%s", root, FILE_STAT);
16780Sstevel@tonic-gate 		if (stat(path, &sb) != 0)
16790Sstevel@tonic-gate 			return (0);
16800Sstevel@tonic-gate 	}
16810Sstevel@tonic-gate 
16820Sstevel@tonic-gate 	/*
16830Sstevel@tonic-gate 	 * consult stat data only if we haven't made a decision
16840Sstevel@tonic-gate 	 * about update. If checking (-n) however, we always
16850Sstevel@tonic-gate 	 * need stat data (since we want to compare old and new)
16860Sstevel@tonic-gate 	 */
16870Sstevel@tonic-gate 	if (!walk_arg.need_update || bam_check)
16880Sstevel@tonic-gate 		getoldstat(root);
16890Sstevel@tonic-gate 
16900Sstevel@tonic-gate 	/*
16910Sstevel@tonic-gate 	 * read list of files
16920Sstevel@tonic-gate 	 */
16930Sstevel@tonic-gate 	if (read_list(root, flistp) != BAM_SUCCESS) {
16940Sstevel@tonic-gate 		clear_walk_args();
16950Sstevel@tonic-gate 		return (BAM_ERROR);
16960Sstevel@tonic-gate 	}
16970Sstevel@tonic-gate 
16980Sstevel@tonic-gate 	assert(flistp->head && flistp->tail);
16990Sstevel@tonic-gate 
17000Sstevel@tonic-gate 	/*
17010Sstevel@tonic-gate 	 * At this point either the update is required
17020Sstevel@tonic-gate 	 * or the decision is pending. In either case
17030Sstevel@tonic-gate 	 * we need to create new stat nvlist
17040Sstevel@tonic-gate 	 */
17050Sstevel@tonic-gate 	create_newstat();
17060Sstevel@tonic-gate 
17070Sstevel@tonic-gate 	/*
17080Sstevel@tonic-gate 	 * This walk does 2 things:
17090Sstevel@tonic-gate 	 *  	- gets new stat data for every file
17100Sstevel@tonic-gate 	 *	- (optional) compare old and new stat data
17110Sstevel@tonic-gate 	 */
17120Sstevel@tonic-gate 	walk_list(root, &flist);
17130Sstevel@tonic-gate 
17140Sstevel@tonic-gate 	/* done with the file list */
17150Sstevel@tonic-gate 	filelist_free(flistp);
17160Sstevel@tonic-gate 
17170Sstevel@tonic-gate 	/*
17180Sstevel@tonic-gate 	 * if we didn't succeed in  creating new stat data above
17190Sstevel@tonic-gate 	 * just return result of update check so that archive is built.
17200Sstevel@tonic-gate 	 */
17210Sstevel@tonic-gate 	if (walk_arg.new_nvlp == NULL) {
17220Sstevel@tonic-gate 		bam_error(NO_NEW_STAT);
17230Sstevel@tonic-gate 		need_update = walk_arg.need_update;
17240Sstevel@tonic-gate 		clear_walk_args();
17250Sstevel@tonic-gate 		return (need_update ? 1 : 0);
17260Sstevel@tonic-gate 	}
17270Sstevel@tonic-gate 
17280Sstevel@tonic-gate 
17290Sstevel@tonic-gate 	/*
17300Sstevel@tonic-gate 	 * If no update required, discard newstat
17310Sstevel@tonic-gate 	 */
17320Sstevel@tonic-gate 	if (!walk_arg.need_update) {
17330Sstevel@tonic-gate 		clear_walk_args();
17340Sstevel@tonic-gate 		return (0);
17350Sstevel@tonic-gate 	}
17360Sstevel@tonic-gate 
17370Sstevel@tonic-gate 	/*
17380Sstevel@tonic-gate 	 * At this point we need an update - so save new stat data
17390Sstevel@tonic-gate 	 * However, if only checking (-n), don't save new stat data.
17400Sstevel@tonic-gate 	 */
17410Sstevel@tonic-gate 	if (!bam_check)
17420Sstevel@tonic-gate 		savenew(root);
17430Sstevel@tonic-gate 
17440Sstevel@tonic-gate 	clear_walk_args();
17450Sstevel@tonic-gate 
17460Sstevel@tonic-gate 	return (1);
17470Sstevel@tonic-gate }
17480Sstevel@tonic-gate 
17490Sstevel@tonic-gate static error_t
17500Sstevel@tonic-gate create_ramdisk(char *root)
17510Sstevel@tonic-gate {
17520Sstevel@tonic-gate 	char *cmdline, path[PATH_MAX];
17530Sstevel@tonic-gate 	size_t len;
17540Sstevel@tonic-gate 	struct stat sb;
17550Sstevel@tonic-gate 
17560Sstevel@tonic-gate 	/*
17570Sstevel@tonic-gate 	 * Setup command args for create_ramdisk.ksh
17580Sstevel@tonic-gate 	 */
17590Sstevel@tonic-gate 	(void) snprintf(path, sizeof (path), "%s%s", root, CREATE_RAMDISK);
17600Sstevel@tonic-gate 	if (stat(path, &sb) != 0) {
17610Sstevel@tonic-gate 		bam_error(ARCH_EXEC_MISS, path, strerror(errno));
17620Sstevel@tonic-gate 		return (BAM_ERROR);
17630Sstevel@tonic-gate 	}
17640Sstevel@tonic-gate 
17650Sstevel@tonic-gate 	len = strlen(path) + strlen(root) + 10;	/* room for space + -R */
17660Sstevel@tonic-gate 	cmdline = s_calloc(1, len);
17670Sstevel@tonic-gate 
17680Sstevel@tonic-gate 	if (strlen(root) > 1) {
17690Sstevel@tonic-gate 		(void) snprintf(cmdline, len, "%s -R %s", path, root);
17700Sstevel@tonic-gate 		/* chop off / at the end */
17710Sstevel@tonic-gate 		cmdline[strlen(cmdline) - 1] = '\0';
17720Sstevel@tonic-gate 	} else
17730Sstevel@tonic-gate 		(void) snprintf(cmdline, len, "%s", path);
17740Sstevel@tonic-gate 
17750Sstevel@tonic-gate 	if (exec_cmd(cmdline, NULL, 0) != 0) {
17760Sstevel@tonic-gate 		bam_error(ARCHIVE_FAIL, cmdline);
17770Sstevel@tonic-gate 		free(cmdline);
17780Sstevel@tonic-gate 		return (BAM_ERROR);
17790Sstevel@tonic-gate 	}
17800Sstevel@tonic-gate 	free(cmdline);
17810Sstevel@tonic-gate 
17820Sstevel@tonic-gate 	/*
17830Sstevel@tonic-gate 	 * Verify that the archive has been created
17840Sstevel@tonic-gate 	 */
17850Sstevel@tonic-gate 	(void) snprintf(path, sizeof (path), "%s%s", root, BOOT_ARCHIVE);
17860Sstevel@tonic-gate 	if (stat(path, &sb) != 0) {
17870Sstevel@tonic-gate 		bam_error(ARCHIVE_NOT_CREATED, path);
17880Sstevel@tonic-gate 		return (BAM_ERROR);
17890Sstevel@tonic-gate 	}
17900Sstevel@tonic-gate 
17910Sstevel@tonic-gate 	return (BAM_SUCCESS);
17920Sstevel@tonic-gate }
17930Sstevel@tonic-gate 
17940Sstevel@tonic-gate /*
17950Sstevel@tonic-gate  * Checks if target filesystem is on a ramdisk
17960Sstevel@tonic-gate  * 1 - is miniroot
17970Sstevel@tonic-gate  * 0 - is not
17980Sstevel@tonic-gate  * When in doubt assume it is not a ramdisk.
17990Sstevel@tonic-gate  */
18000Sstevel@tonic-gate static int
18010Sstevel@tonic-gate is_ramdisk(char *root)
18020Sstevel@tonic-gate {
18030Sstevel@tonic-gate 	struct extmnttab mnt;
18040Sstevel@tonic-gate 	FILE *fp;
18050Sstevel@tonic-gate 	int found;
1806316Svikram 	char mntpt[PATH_MAX];
1807316Svikram 	char *cp;
18080Sstevel@tonic-gate 
18090Sstevel@tonic-gate 	/*
18100Sstevel@tonic-gate 	 * There are 3 situations where creating archive is
18110Sstevel@tonic-gate 	 * of dubious value:
1812316Svikram 	 *	- create boot_archive on a lofi-mounted boot_archive
18130Sstevel@tonic-gate 	 *	- create it on a ramdisk which is the root filesystem
18140Sstevel@tonic-gate 	 *	- create it on a ramdisk mounted somewhere else
18150Sstevel@tonic-gate 	 * The first is not easy to detect and checking for it is not
18160Sstevel@tonic-gate 	 * worth it.
18170Sstevel@tonic-gate 	 * The other two conditions are handled here
18180Sstevel@tonic-gate 	 */
18190Sstevel@tonic-gate 
18200Sstevel@tonic-gate 	fp = fopen(MNTTAB, "r");
18210Sstevel@tonic-gate 	if (fp == NULL) {
18220Sstevel@tonic-gate 		bam_error(OPEN_FAIL, MNTTAB, strerror(errno));
18230Sstevel@tonic-gate 		return (0);
18240Sstevel@tonic-gate 	}
18250Sstevel@tonic-gate 
18260Sstevel@tonic-gate 	resetmnttab(fp);
18270Sstevel@tonic-gate 
1828316Svikram 	/*
1829316Svikram 	 * Remove any trailing / from the mount point
1830316Svikram 	 */
1831316Svikram 	(void) strlcpy(mntpt, root, sizeof (mntpt));
1832316Svikram 	if (strcmp(root, "/") != 0) {
1833316Svikram 		cp = mntpt + strlen(mntpt) - 1;
1834316Svikram 		if (*cp == '/')
1835316Svikram 			*cp = '\0';
1836316Svikram 	}
18370Sstevel@tonic-gate 	found = 0;
18380Sstevel@tonic-gate 	while (getextmntent(fp, &mnt, sizeof (mnt)) == 0) {
1839316Svikram 		if (strcmp(mnt.mnt_mountp, mntpt) == 0) {
18400Sstevel@tonic-gate 			found = 1;
18410Sstevel@tonic-gate 			break;
18420Sstevel@tonic-gate 		}
18430Sstevel@tonic-gate 	}
18440Sstevel@tonic-gate 
18450Sstevel@tonic-gate 	if (!found) {
18460Sstevel@tonic-gate 		if (bam_verbose)
1847316Svikram 			bam_error(NOT_IN_MNTTAB, mntpt);
18480Sstevel@tonic-gate 		(void) fclose(fp);
18490Sstevel@tonic-gate 		return (0);
18500Sstevel@tonic-gate 	}
18510Sstevel@tonic-gate 
18520Sstevel@tonic-gate 	if (strstr(mnt.mnt_special, RAMDISK_SPECIAL) != NULL) {
18530Sstevel@tonic-gate 		if (bam_verbose)
18540Sstevel@tonic-gate 			bam_error(IS_RAMDISK, bam_root);
18550Sstevel@tonic-gate 		(void) fclose(fp);
18560Sstevel@tonic-gate 		return (1);
18570Sstevel@tonic-gate 	}
18580Sstevel@tonic-gate 
18590Sstevel@tonic-gate 	(void) fclose(fp);
18600Sstevel@tonic-gate 
18610Sstevel@tonic-gate 	return (0);
18620Sstevel@tonic-gate }
18630Sstevel@tonic-gate 
18640Sstevel@tonic-gate static int
18650Sstevel@tonic-gate is_newboot(char *root)
18660Sstevel@tonic-gate {
18670Sstevel@tonic-gate 	char path[PATH_MAX];
18680Sstevel@tonic-gate 	struct stat sb;
18690Sstevel@tonic-gate 
18700Sstevel@tonic-gate 	/*
18710Sstevel@tonic-gate 	 * We can't boot without MULTI_BOOT
18720Sstevel@tonic-gate 	 */
18730Sstevel@tonic-gate 	(void) snprintf(path, sizeof (path), "%s%s", root, MULTI_BOOT);
18740Sstevel@tonic-gate 	if (stat(path, &sb) == -1) {
18750Sstevel@tonic-gate 		if (bam_verbose)
18760Sstevel@tonic-gate 			bam_print(FILE_MISS, path);
18770Sstevel@tonic-gate 		return (0);
18780Sstevel@tonic-gate 	}
18790Sstevel@tonic-gate 
18800Sstevel@tonic-gate 	/*
18810Sstevel@tonic-gate 	 * We can't generate archive without GRUB_DIR
18820Sstevel@tonic-gate 	 */
18830Sstevel@tonic-gate 	(void) snprintf(path, sizeof (path), "%s%s", root, GRUB_DIR);
18840Sstevel@tonic-gate 	if (stat(path, &sb) == -1) {
18850Sstevel@tonic-gate 		if (bam_verbose)
18860Sstevel@tonic-gate 			bam_print(DIR_MISS, path);
18870Sstevel@tonic-gate 		return (0);
18880Sstevel@tonic-gate 	}
18890Sstevel@tonic-gate 
18900Sstevel@tonic-gate 	return (1);
18910Sstevel@tonic-gate }
18920Sstevel@tonic-gate 
18930Sstevel@tonic-gate static int
18940Sstevel@tonic-gate is_readonly(char *root)
18950Sstevel@tonic-gate {
18960Sstevel@tonic-gate 	struct statvfs vfs;
18970Sstevel@tonic-gate 
18980Sstevel@tonic-gate 	/*
18990Sstevel@tonic-gate 	 * Check for RDONLY filesystem
19000Sstevel@tonic-gate 	 * When in doubt assume it is not readonly
19010Sstevel@tonic-gate 	 */
19020Sstevel@tonic-gate 	if (statvfs(root, &vfs) != 0) {
19030Sstevel@tonic-gate 		if (bam_verbose)
19040Sstevel@tonic-gate 			bam_error(STATVFS_FAIL, root, strerror(errno));
19050Sstevel@tonic-gate 		return (0);
19060Sstevel@tonic-gate 	}
19070Sstevel@tonic-gate 
19080Sstevel@tonic-gate 	if (vfs.f_flag & ST_RDONLY) {
19090Sstevel@tonic-gate 		return (1);
19100Sstevel@tonic-gate 	}
19110Sstevel@tonic-gate 
19120Sstevel@tonic-gate 	return (0);
19130Sstevel@tonic-gate }
19140Sstevel@tonic-gate 
19150Sstevel@tonic-gate static error_t
19160Sstevel@tonic-gate update_archive(char *root, char *opt)
19170Sstevel@tonic-gate {
19180Sstevel@tonic-gate 	error_t ret;
19190Sstevel@tonic-gate 
19200Sstevel@tonic-gate 	assert(root);
19210Sstevel@tonic-gate 	assert(opt == NULL);
19220Sstevel@tonic-gate 
19230Sstevel@tonic-gate 	/*
1924316Svikram 	 * root must belong to a GRUB boot OS,
19250Sstevel@tonic-gate 	 * don't care on sparc except for diskless clients
19260Sstevel@tonic-gate 	 */
19270Sstevel@tonic-gate 	if (!is_newboot(root)) {
1928316Svikram 		/*
1929316Svikram 		 * Emit message only if not in context of update_all.
1930316Svikram 		 * If in update_all, emit only if verbose flag is set.
1931316Svikram 		 */
1932316Svikram 		if (!bam_update_all || bam_verbose)
1933316Svikram 			bam_print(NOT_GRUB_BOOT, root);
19340Sstevel@tonic-gate 		return (BAM_SUCCESS);
19350Sstevel@tonic-gate 	}
19360Sstevel@tonic-gate 
19370Sstevel@tonic-gate 	/*
1938*662Sszhou 	 * If smf check is requested when / is writable (can happen
1939*662Sszhou 	 * on first reboot following an upgrade because service
1940*662Sszhou 	 * dependency is messed up), skip the check.
1941*662Sszhou 	 */
1942*662Sszhou 	if (bam_smf_check && !bam_root_readonly)
1943*662Sszhou 		return (BAM_SUCCESS);
1944*662Sszhou 
1945*662Sszhou 	/*
1946*662Sszhou 	 * root must be writable. This check applies to alternate
1947*662Sszhou 	 * root (-R option); bam_root_readonly applies to '/' only.
19480Sstevel@tonic-gate 	 * Note: statvfs() does not always report the truth
19490Sstevel@tonic-gate 	 */
1950*662Sszhou 	if (!bam_smf_check && is_readonly(root)) {
1951*662Sszhou 		if (bam_verbose)
19520Sstevel@tonic-gate 			bam_print(RDONLY_FS, root);
19530Sstevel@tonic-gate 		return (BAM_SUCCESS);
19540Sstevel@tonic-gate 	}
19550Sstevel@tonic-gate 
19560Sstevel@tonic-gate 	/*
19570Sstevel@tonic-gate 	 * Don't generate archive on ramdisk
19580Sstevel@tonic-gate 	 */
19590Sstevel@tonic-gate 	if (is_ramdisk(root)) {
19600Sstevel@tonic-gate 		if (bam_verbose)
19610Sstevel@tonic-gate 			bam_print(SKIP_RAMDISK);
19620Sstevel@tonic-gate 		return (BAM_SUCCESS);
19630Sstevel@tonic-gate 	}
19640Sstevel@tonic-gate 
19650Sstevel@tonic-gate 	/*
19660Sstevel@tonic-gate 	 * Now check if updated is really needed
19670Sstevel@tonic-gate 	 */
19680Sstevel@tonic-gate 	ret = update_required(root);
19690Sstevel@tonic-gate 
19700Sstevel@tonic-gate 	/*
19710Sstevel@tonic-gate 	 * The check command (-n) is *not* a dry run
19720Sstevel@tonic-gate 	 * It only checks if the archive is in sync.
19730Sstevel@tonic-gate 	 */
19740Sstevel@tonic-gate 	if (bam_check) {
19750Sstevel@tonic-gate 		bam_exit((ret != 0) ? 1 : 0);
19760Sstevel@tonic-gate 	}
19770Sstevel@tonic-gate 
19780Sstevel@tonic-gate 	if (ret == 1) {
19790Sstevel@tonic-gate 		/* create the ramdisk */
19800Sstevel@tonic-gate 		ret = create_ramdisk(root);
19810Sstevel@tonic-gate 	}
19820Sstevel@tonic-gate 	return (ret);
19830Sstevel@tonic-gate }
19840Sstevel@tonic-gate 
1985316Svikram static void
1986316Svikram restore_grub_slice(void)
1987316Svikram {
1988316Svikram 	struct stat sb;
1989316Svikram 	char *mntpt, *physlice;
1990316Svikram 	int mnted;	/* set if we did a mount */
1991316Svikram 	char menupath[PATH_MAX], cmd[PATH_MAX];
1992316Svikram 
1993316Svikram 	if (stat(GRUB_slice, &sb) != 0) {
1994316Svikram 		bam_error(MISSING_SLICE_FILE, GRUB_slice, strerror(errno));
1995316Svikram 		return;
1996316Svikram 	}
1997316Svikram 
1998316Svikram 	/*
1999316Svikram 	 * If we are doing an luactivate, don't attempt to restore GRUB or else
2000316Svikram 	 * we may not be able to get to DCA boot environments. Let luactivate
2001316Svikram 	 * handle GRUB/DCA installation
2002316Svikram 	 */
2003316Svikram 	if (stat(LU_ACTIVATE_FILE, &sb) == 0) {
2004316Svikram 		return;
2005316Svikram 	}
2006316Svikram 
2007316Svikram 	mnted = 0;
2008316Svikram 	physlice = NULL;
2009621Svikram 	mntpt = mount_grub_slice(&mnted, &physlice, NULL, NULL);
2010316Svikram 	if (mntpt == NULL) {
2011316Svikram 		bam_error(CANNOT_RESTORE_GRUB_SLICE);
2012316Svikram 		return;
2013316Svikram 	}
2014316Svikram 
2015316Svikram 	(void) snprintf(menupath, sizeof (menupath), "%s%s", mntpt, GRUB_MENU);
2016316Svikram 	if (stat(menupath, &sb) == 0) {
2017621Svikram 		umount_grub_slice(mnted, mntpt, physlice, NULL, NULL);
2018316Svikram 		return;
2019316Svikram 	}
2020316Svikram 
2021316Svikram 	/*
2022316Svikram 	 * The menu is missing - we need to do a restore
2023316Svikram 	 */
2024316Svikram 	bam_print(RESTORING_GRUB);
2025316Svikram 
2026316Svikram 	(void) snprintf(cmd, sizeof (cmd), "%s %s %s %s",
2027316Svikram 	    INSTALLGRUB, STAGE1, STAGE2, physlice);
2028316Svikram 
2029316Svikram 	if (exec_cmd(cmd, NULL, 0) != 0) {
2030316Svikram 		bam_error(RESTORE_GRUB_FAILED);
2031621Svikram 		umount_grub_slice(mnted, mntpt, physlice, NULL, NULL);
2032316Svikram 		return;
2033316Svikram 	}
2034316Svikram 
2035316Svikram 	if (stat(GRUB_backup_menu, &sb) != 0) {
2036316Svikram 		bam_error(MISSING_BACKUP_MENU,
2037316Svikram 		    GRUB_backup_menu, strerror(errno));
2038621Svikram 		umount_grub_slice(mnted, mntpt, physlice, NULL, NULL);
2039316Svikram 		return;
2040316Svikram 	}
2041316Svikram 
2042316Svikram 	(void) snprintf(cmd, sizeof (cmd), "/bin/cp %s %s",
2043316Svikram 	    GRUB_backup_menu, menupath);
2044316Svikram 
2045316Svikram 	if (exec_cmd(cmd, NULL, 0) != 0) {
2046316Svikram 		bam_error(RESTORE_MENU_FAILED, menupath);
2047621Svikram 		umount_grub_slice(mnted, mntpt, physlice, NULL, NULL);
2048316Svikram 		return;
2049316Svikram 	}
2050316Svikram 
2051316Svikram 	/* Success */
2052621Svikram 	umount_grub_slice(mnted, mntpt, physlice, NULL, NULL);
2053316Svikram }
2054316Svikram 
20550Sstevel@tonic-gate static error_t
20560Sstevel@tonic-gate update_all(char *root, char *opt)
20570Sstevel@tonic-gate {
20580Sstevel@tonic-gate 	struct extmnttab mnt;
20590Sstevel@tonic-gate 	struct stat sb;
20600Sstevel@tonic-gate 	FILE *fp;
20610Sstevel@tonic-gate 	char multibt[PATH_MAX];
20620Sstevel@tonic-gate 	error_t ret = BAM_SUCCESS;
20630Sstevel@tonic-gate 
2064621Svikram 	assert(root);
20650Sstevel@tonic-gate 	assert(opt == NULL);
20660Sstevel@tonic-gate 
2067621Svikram 	if (bam_rootlen != 1 || *root != '/') {
2068621Svikram 		elide_trailing_slash(root, multibt, sizeof (multibt));
2069621Svikram 		bam_error(ALT_ROOT_INVALID, multibt);
2070621Svikram 		return (BAM_ERROR);
2071621Svikram 	}
2072621Svikram 
20730Sstevel@tonic-gate 	/*
20740Sstevel@tonic-gate 	 * First update archive for current root
20750Sstevel@tonic-gate 	 */
20760Sstevel@tonic-gate 	if (update_archive(root, opt) != BAM_SUCCESS)
20770Sstevel@tonic-gate 		ret = BAM_ERROR;
20780Sstevel@tonic-gate 
20790Sstevel@tonic-gate 	/*
20800Sstevel@tonic-gate 	 * Now walk the mount table, performing archive update
20810Sstevel@tonic-gate 	 * for all mounted Newboot root filesystems
20820Sstevel@tonic-gate 	 */
20830Sstevel@tonic-gate 	fp = fopen(MNTTAB, "r");
20840Sstevel@tonic-gate 	if (fp == NULL) {
20850Sstevel@tonic-gate 		bam_error(OPEN_FAIL, MNTTAB, strerror(errno));
2086316Svikram 		ret = BAM_ERROR;
2087316Svikram 		goto out;
20880Sstevel@tonic-gate 	}
20890Sstevel@tonic-gate 
20900Sstevel@tonic-gate 	resetmnttab(fp);
20910Sstevel@tonic-gate 
20920Sstevel@tonic-gate 	while (getextmntent(fp, &mnt, sizeof (mnt)) == 0) {
20930Sstevel@tonic-gate 		if (mnt.mnt_special == NULL)
20940Sstevel@tonic-gate 			continue;
20950Sstevel@tonic-gate 		if (strncmp(mnt.mnt_special, "/dev/", strlen("/dev/")) != 0)
20960Sstevel@tonic-gate 			continue;
20970Sstevel@tonic-gate 		if (strcmp(mnt.mnt_mountp, "/") == 0)
20980Sstevel@tonic-gate 			continue;
20990Sstevel@tonic-gate 
21000Sstevel@tonic-gate 		(void) snprintf(multibt, sizeof (multibt), "%s%s",
21010Sstevel@tonic-gate 		    mnt.mnt_mountp, MULTI_BOOT);
21020Sstevel@tonic-gate 
21030Sstevel@tonic-gate 		if (stat(multibt, &sb) == -1)
21040Sstevel@tonic-gate 			continue;
21050Sstevel@tonic-gate 
21060Sstevel@tonic-gate 		/*
21070Sstevel@tonic-gate 		 * We put a trailing slash to be consistent with root = "/"
21080Sstevel@tonic-gate 		 * case, such that we don't have to print // in some cases.
21090Sstevel@tonic-gate 		 */
21100Sstevel@tonic-gate 		(void) snprintf(rootbuf, sizeof (rootbuf), "%s/",
21110Sstevel@tonic-gate 		    mnt.mnt_mountp);
21120Sstevel@tonic-gate 		bam_rootlen = strlen(rootbuf);
21130Sstevel@tonic-gate 		if (update_archive(rootbuf, opt) != BAM_SUCCESS)
21140Sstevel@tonic-gate 			ret = BAM_ERROR;
21150Sstevel@tonic-gate 	}
21160Sstevel@tonic-gate 
21170Sstevel@tonic-gate 	(void) fclose(fp);
21180Sstevel@tonic-gate 
2119316Svikram out:
2120316Svikram 	if (stat(GRUB_slice, &sb) == 0) {
2121316Svikram 		restore_grub_slice();
2122316Svikram 	}
2123316Svikram 
21240Sstevel@tonic-gate 	return (ret);
21250Sstevel@tonic-gate }
21260Sstevel@tonic-gate 
21270Sstevel@tonic-gate static void
21280Sstevel@tonic-gate append_line(menu_t *mp, line_t *lp)
21290Sstevel@tonic-gate {
21300Sstevel@tonic-gate 	if (mp->start == NULL) {
21310Sstevel@tonic-gate 		mp->start = lp;
21320Sstevel@tonic-gate 	} else {
21330Sstevel@tonic-gate 		mp->end->next = lp;
2134*662Sszhou 		lp->prev = mp->end;
21350Sstevel@tonic-gate 	}
21360Sstevel@tonic-gate 	mp->end = lp;
21370Sstevel@tonic-gate }
21380Sstevel@tonic-gate 
2139*662Sszhou static void
2140*662Sszhou unlink_line(menu_t *mp, line_t *lp)
2141*662Sszhou {
2142*662Sszhou 	/* unlink from list */
2143*662Sszhou 	if (lp->prev)
2144*662Sszhou 		lp->prev->next = lp->next;
2145*662Sszhou 	else
2146*662Sszhou 		mp->start = lp->next;
2147*662Sszhou 	if (lp->next)
2148*662Sszhou 		lp->next->prev = lp->prev;
2149*662Sszhou 	else
2150*662Sszhou 		mp->end = lp->prev;
2151*662Sszhou }
2152*662Sszhou 
2153*662Sszhou static entry_t *
2154*662Sszhou boot_entry_new(menu_t *mp, line_t *start, line_t *end)
2155*662Sszhou {
2156*662Sszhou 	entry_t *ent, *prev;
2157*662Sszhou 
2158*662Sszhou 	ent = s_calloc(1, sizeof (entry_t));
2159*662Sszhou 	ent->start = start;
2160*662Sszhou 	ent->end = end;
2161*662Sszhou 
2162*662Sszhou 	if (mp->entries == NULL) {
2163*662Sszhou 		mp->entries = ent;
2164*662Sszhou 		return (ent);
2165*662Sszhou 	}
2166*662Sszhou 
2167*662Sszhou 	prev = mp->entries;
2168*662Sszhou 	while (prev->next)
2169*662Sszhou 		prev = prev-> next;
2170*662Sszhou 	prev->next = ent;
2171*662Sszhou 	ent->prev = prev;
2172*662Sszhou 	return (ent);
2173*662Sszhou }
2174*662Sszhou 
2175*662Sszhou static void
2176*662Sszhou boot_entry_addline(entry_t *ent, line_t *lp)
2177*662Sszhou {
2178*662Sszhou 	if (ent)
2179*662Sszhou 		ent->end = lp;
2180*662Sszhou }
2181*662Sszhou 
21820Sstevel@tonic-gate /*
21830Sstevel@tonic-gate  * A line in menu.lst looks like
21840Sstevel@tonic-gate  * [ ]*<cmd>[ \t=]*<arg>*
21850Sstevel@tonic-gate  */
21860Sstevel@tonic-gate static void
21870Sstevel@tonic-gate line_parser(menu_t *mp, char *str, int *lineNum, int *entryNum)
21880Sstevel@tonic-gate {
21890Sstevel@tonic-gate 	/*
21900Sstevel@tonic-gate 	 * save state across calls. This is so that
21910Sstevel@tonic-gate 	 * header gets the right entry# after title has
21920Sstevel@tonic-gate 	 * been processed
21930Sstevel@tonic-gate 	 */
2194*662Sszhou 	static line_t *prev = NULL;
2195*662Sszhou 	static entry_t *curr_ent = NULL;
21960Sstevel@tonic-gate 
21970Sstevel@tonic-gate 	line_t	*lp;
21980Sstevel@tonic-gate 	char *cmd, *sep, *arg;
21990Sstevel@tonic-gate 	char save, *cp, *line;
22000Sstevel@tonic-gate 	menu_flag_t flag = BAM_INVALID;
22010Sstevel@tonic-gate 
22020Sstevel@tonic-gate 	if (str == NULL) {
22030Sstevel@tonic-gate 		return;
22040Sstevel@tonic-gate 	}
22050Sstevel@tonic-gate 
22060Sstevel@tonic-gate 	/*
22070Sstevel@tonic-gate 	 * First save a copy of the entire line.
22080Sstevel@tonic-gate 	 * We use this later to set the line field.
22090Sstevel@tonic-gate 	 */
22100Sstevel@tonic-gate 	line = s_strdup(str);
22110Sstevel@tonic-gate 
22120Sstevel@tonic-gate 	/* Eat up leading whitespace */
22130Sstevel@tonic-gate 	while (*str == ' ' || *str == '\t')
22140Sstevel@tonic-gate 		str++;
22150Sstevel@tonic-gate 
22160Sstevel@tonic-gate 	if (*str == '#') {		/* comment */
22170Sstevel@tonic-gate 		cmd = s_strdup("#");
22180Sstevel@tonic-gate 		sep = NULL;
22190Sstevel@tonic-gate 		arg = s_strdup(str + 1);
22200Sstevel@tonic-gate 		flag = BAM_COMMENT;
22210Sstevel@tonic-gate 	} else if (*str == '\0') {	/* blank line */
22220Sstevel@tonic-gate 		cmd = sep = arg = NULL;
22230Sstevel@tonic-gate 		flag = BAM_EMPTY;
22240Sstevel@tonic-gate 	} else {
22250Sstevel@tonic-gate 		/*
22260Sstevel@tonic-gate 		 * '=' is not a documented separator in grub syntax.
22270Sstevel@tonic-gate 		 * However various development bits use '=' as a
22280Sstevel@tonic-gate 		 * separator. In addition, external users also
22290Sstevel@tonic-gate 		 * use = as a separator. So we will allow that usage.
22300Sstevel@tonic-gate 		 */
22310Sstevel@tonic-gate 		cp = str;
22320Sstevel@tonic-gate 		while (*str != ' ' && *str != '\t' && *str != '=') {
22330Sstevel@tonic-gate 			if (*str == '\0') {
22340Sstevel@tonic-gate 				cmd = s_strdup(cp);
22350Sstevel@tonic-gate 				sep = arg = NULL;
22360Sstevel@tonic-gate 				break;
22370Sstevel@tonic-gate 			}
22380Sstevel@tonic-gate 			str++;
22390Sstevel@tonic-gate 		}
22400Sstevel@tonic-gate 
22410Sstevel@tonic-gate 		if (*str != '\0') {
22420Sstevel@tonic-gate 			save = *str;
22430Sstevel@tonic-gate 			*str = '\0';
22440Sstevel@tonic-gate 			cmd = s_strdup(cp);
22450Sstevel@tonic-gate 			*str = save;
22460Sstevel@tonic-gate 
22470Sstevel@tonic-gate 			str++;
22480Sstevel@tonic-gate 			save = *str;
22490Sstevel@tonic-gate 			*str = '\0';
22500Sstevel@tonic-gate 			sep = s_strdup(str - 1);
22510Sstevel@tonic-gate 			*str = save;
22520Sstevel@tonic-gate 
22530Sstevel@tonic-gate 			while (*str == ' ' || *str == '\t')
22540Sstevel@tonic-gate 				str++;
22550Sstevel@tonic-gate 			if (*str == '\0')
22560Sstevel@tonic-gate 				arg = NULL;
22570Sstevel@tonic-gate 			else
22580Sstevel@tonic-gate 				arg = s_strdup(str);
22590Sstevel@tonic-gate 		}
22600Sstevel@tonic-gate 	}
22610Sstevel@tonic-gate 
22620Sstevel@tonic-gate 	lp = s_calloc(1, sizeof (line_t));
22630Sstevel@tonic-gate 
22640Sstevel@tonic-gate 	lp->cmd = cmd;
22650Sstevel@tonic-gate 	lp->sep = sep;
22660Sstevel@tonic-gate 	lp->arg = arg;
22670Sstevel@tonic-gate 	lp->line = line;
22680Sstevel@tonic-gate 	lp->lineNum = ++(*lineNum);
22690Sstevel@tonic-gate 	if (cmd && strcmp(cmd, menu_cmds[TITLE_CMD]) == 0) {
22700Sstevel@tonic-gate 		lp->entryNum = ++(*entryNum);
22710Sstevel@tonic-gate 		lp->flags = BAM_TITLE;
22720Sstevel@tonic-gate 		if (prev && prev->flags == BAM_COMMENT &&
2273*662Sszhou 		    prev->arg && strcmp(prev->arg, BAM_HDR) == 0) {
22740Sstevel@tonic-gate 			prev->entryNum = lp->entryNum;
2275*662Sszhou 			curr_ent = boot_entry_new(mp, prev, lp);
2276*662Sszhou 		} else {
2277*662Sszhou 			curr_ent = boot_entry_new(mp, lp, lp);
2278*662Sszhou 		}
22790Sstevel@tonic-gate 	} else if (flag != BAM_INVALID) {
22800Sstevel@tonic-gate 		/*
22810Sstevel@tonic-gate 		 * For header comments, the entry# is "fixed up"
22820Sstevel@tonic-gate 		 * by the subsequent title
22830Sstevel@tonic-gate 		 */
22840Sstevel@tonic-gate 		lp->entryNum = *entryNum;
22850Sstevel@tonic-gate 		lp->flags = flag;
22860Sstevel@tonic-gate 	} else {
22870Sstevel@tonic-gate 		lp->entryNum = *entryNum;
22880Sstevel@tonic-gate 		lp->flags = (*entryNum == ENTRY_INIT) ? BAM_GLOBAL : BAM_ENTRY;
22890Sstevel@tonic-gate 	}
22900Sstevel@tonic-gate 
2291*662Sszhou 	/* record default, old default, and entry line ranges */
2292*662Sszhou 	if (lp->flags == BAM_GLOBAL &&
2293*662Sszhou 	    strcmp(lp->cmd, menu_cmds[DEFAULT_CMD]) == 0) {
2294*662Sszhou 		mp->curdefault = lp;
2295*662Sszhou 	} else if (lp->flags == BAM_COMMENT &&
2296*662Sszhou 	    strncmp(lp->arg, BAM_OLDDEF, strlen(BAM_OLDDEF)) == 0) {
2297*662Sszhou 		mp->olddefault = lp;
2298*662Sszhou 	} else if (lp->flags == BAM_ENTRY ||
2299*662Sszhou 	    (lp->flags == BAM_COMMENT && strcmp(lp->arg, BAM_FTR) == 0)) {
2300*662Sszhou 		boot_entry_addline(curr_ent, lp);
2301*662Sszhou 	}
23020Sstevel@tonic-gate 	append_line(mp, lp);
23030Sstevel@tonic-gate 
23040Sstevel@tonic-gate 	prev = lp;
23050Sstevel@tonic-gate }
23060Sstevel@tonic-gate 
2307621Svikram static void
2308621Svikram update_numbering(menu_t *mp)
2309621Svikram {
2310621Svikram 	int lineNum;
2311621Svikram 	int entryNum;
2312621Svikram 	int old_default_value;
2313621Svikram 	line_t *lp, *prev, *default_lp, *default_entry;
2314621Svikram 	char buf[PATH_MAX];
2315621Svikram 
2316621Svikram 	if (mp->start == NULL) {
2317621Svikram 		return;
2318621Svikram 	}
2319621Svikram 
2320621Svikram 	lineNum = LINE_INIT;
2321621Svikram 	entryNum = ENTRY_INIT;
2322621Svikram 	old_default_value = ENTRY_INIT;
2323621Svikram 	lp = default_lp = default_entry = NULL;
2324621Svikram 
2325621Svikram 	prev = NULL;
2326621Svikram 	for (lp = mp->start; lp; prev = lp, lp = lp->next) {
2327621Svikram 		lp->lineNum = ++lineNum;
2328621Svikram 
2329621Svikram 		/*
2330621Svikram 		 * Get the value of the default command
2331621Svikram 		 */
2332621Svikram 		if (lp->entryNum == ENTRY_INIT && lp->cmd &&
2333621Svikram 		    strcmp(lp->cmd, menu_cmds[DEFAULT_CMD]) == 0 &&
2334621Svikram 		    lp->arg) {
2335621Svikram 			old_default_value = atoi(lp->arg);
2336621Svikram 			default_lp = lp;
2337621Svikram 		}
2338621Svikram 
2339621Svikram 		/*
2340621Svikram 		 * If not boot entry, nothing else to fix for this
2341621Svikram 		 * entry
2342621Svikram 		 */
2343621Svikram 		if (lp->entryNum == ENTRY_INIT)
2344621Svikram 			continue;
2345621Svikram 
2346621Svikram 		/*
2347621Svikram 		 * Record the position of the default entry.
2348621Svikram 		 * The following works because global
2349621Svikram 		 * commands like default and timeout should precede
2350621Svikram 		 * actual boot entries, so old_default_value
2351621Svikram 		 * is already known (or default cmd is missing).
2352621Svikram 		 */
2353621Svikram 		if (default_entry == NULL &&
2354621Svikram 		    old_default_value != ENTRY_INIT &&
2355621Svikram 		    lp->entryNum == old_default_value) {
2356621Svikram 			default_entry = lp;
2357621Svikram 		}
2358621Svikram 
2359621Svikram 		/*
2360621Svikram 		 * Now fixup the entry number
2361621Svikram 		 */
2362621Svikram 		if (lp->cmd && strcmp(lp->cmd, menu_cmds[TITLE_CMD]) == 0) {
2363621Svikram 			lp->entryNum = ++entryNum;
2364621Svikram 			/* fixup the bootadm header */
2365621Svikram 			if (prev && prev->flags == BAM_COMMENT &&
2366621Svikram 			    prev->arg && strcmp(prev->arg, BAM_HDR) == 0) {
2367621Svikram 				prev->entryNum = lp->entryNum;
2368621Svikram 			}
2369621Svikram 		} else {
2370621Svikram 			lp->entryNum = entryNum;
2371621Svikram 		}
2372621Svikram 	}
2373621Svikram 
2374621Svikram 	/*
2375621Svikram 	 * No default command in menu, simply return
2376621Svikram 	 */
2377621Svikram 	if (default_lp == NULL) {
2378621Svikram 		return;
2379621Svikram 	}
2380621Svikram 
2381621Svikram 	free(default_lp->arg);
2382621Svikram 	free(default_lp->line);
2383621Svikram 
2384621Svikram 	if (default_entry == NULL) {
2385621Svikram 		default_lp->arg = s_strdup("0");
2386621Svikram 	} else {
2387621Svikram 		(void) snprintf(buf, sizeof (buf), "%d",
2388621Svikram 		    default_entry->entryNum);
2389621Svikram 		default_lp->arg = s_strdup(buf);
2390621Svikram 	}
2391621Svikram 
2392621Svikram 	/*
2393621Svikram 	 * The following is required since only the line field gets
2394621Svikram 	 * written back to menu.lst
2395621Svikram 	 */
2396621Svikram 	(void) snprintf(buf, sizeof (buf), "%s%s%s",
2397621Svikram 	    menu_cmds[DEFAULT_CMD], menu_cmds[SEP_CMD], default_lp->arg);
2398621Svikram 	default_lp->line = s_strdup(buf);
2399621Svikram }
2400621Svikram 
2401621Svikram 
24020Sstevel@tonic-gate static menu_t *
24030Sstevel@tonic-gate menu_read(char *menu_path)
24040Sstevel@tonic-gate {
24050Sstevel@tonic-gate 	FILE *fp;
24060Sstevel@tonic-gate 	char buf[BAM_MAXLINE], *cp;
24070Sstevel@tonic-gate 	menu_t *mp;
24080Sstevel@tonic-gate 	int line, entry, len, n;
24090Sstevel@tonic-gate 
24100Sstevel@tonic-gate 	mp = s_calloc(1, sizeof (menu_t));
24110Sstevel@tonic-gate 
24120Sstevel@tonic-gate 	fp = fopen(menu_path, "r");
24130Sstevel@tonic-gate 	if (fp == NULL) { /* Let the caller handle this error */
24140Sstevel@tonic-gate 		return (mp);
24150Sstevel@tonic-gate 	}
24160Sstevel@tonic-gate 
24170Sstevel@tonic-gate 
24180Sstevel@tonic-gate 	/* Note: GRUB boot entry number starts with 0 */
24190Sstevel@tonic-gate 	line = LINE_INIT;
24200Sstevel@tonic-gate 	entry = ENTRY_INIT;
24210Sstevel@tonic-gate 	cp = buf;
24220Sstevel@tonic-gate 	len = sizeof (buf);
24230Sstevel@tonic-gate 	while (s_fgets(cp, len, fp) != NULL) {
24240Sstevel@tonic-gate 		n = strlen(cp);
24250Sstevel@tonic-gate 		if (cp[n - 1] == '\\') {
24260Sstevel@tonic-gate 			len -= n - 1;
24270Sstevel@tonic-gate 			assert(len >= 2);
24280Sstevel@tonic-gate 			cp += n - 1;
24290Sstevel@tonic-gate 			continue;
24300Sstevel@tonic-gate 		}
24310Sstevel@tonic-gate 		line_parser(mp, buf, &line, &entry);
24320Sstevel@tonic-gate 		cp = buf;
24330Sstevel@tonic-gate 		len = sizeof (buf);
24340Sstevel@tonic-gate 	}
24350Sstevel@tonic-gate 
24360Sstevel@tonic-gate 	if (fclose(fp) == EOF) {
24370Sstevel@tonic-gate 		bam_error(CLOSE_FAIL, menu_path, strerror(errno));
24380Sstevel@tonic-gate 	}
24390Sstevel@tonic-gate 
24400Sstevel@tonic-gate 	return (mp);
24410Sstevel@tonic-gate }
24420Sstevel@tonic-gate 
24430Sstevel@tonic-gate static error_t
24440Sstevel@tonic-gate selector(menu_t *mp, char *opt, int *entry, char **title)
24450Sstevel@tonic-gate {
24460Sstevel@tonic-gate 	char *eq;
24470Sstevel@tonic-gate 	char *opt_dup;
24480Sstevel@tonic-gate 	int entryNum;
24490Sstevel@tonic-gate 
24500Sstevel@tonic-gate 	assert(mp);
24510Sstevel@tonic-gate 	assert(mp->start);
24520Sstevel@tonic-gate 	assert(opt);
24530Sstevel@tonic-gate 
24540Sstevel@tonic-gate 	opt_dup = s_strdup(opt);
24550Sstevel@tonic-gate 
24560Sstevel@tonic-gate 	if (entry)
24570Sstevel@tonic-gate 		*entry = ENTRY_INIT;
24580Sstevel@tonic-gate 	if (title)
24590Sstevel@tonic-gate 		*title = NULL;
24600Sstevel@tonic-gate 
24610Sstevel@tonic-gate 	eq = strchr(opt_dup, '=');
24620Sstevel@tonic-gate 	if (eq == NULL) {
24630Sstevel@tonic-gate 		bam_error(INVALID_OPT, opt);
24640Sstevel@tonic-gate 		free(opt_dup);
24650Sstevel@tonic-gate 		return (BAM_ERROR);
24660Sstevel@tonic-gate 	}
24670Sstevel@tonic-gate 
24680Sstevel@tonic-gate 	*eq = '\0';
24690Sstevel@tonic-gate 	if (entry && strcmp(opt_dup, OPT_ENTRY_NUM) == 0) {
24700Sstevel@tonic-gate 		assert(mp->end);
24710Sstevel@tonic-gate 		entryNum = s_strtol(eq + 1);
24720Sstevel@tonic-gate 		if (entryNum < 0 || entryNum > mp->end->entryNum) {
24730Sstevel@tonic-gate 			bam_error(INVALID_ENTRY, eq + 1);
24740Sstevel@tonic-gate 			free(opt_dup);
24750Sstevel@tonic-gate 			return (BAM_ERROR);
24760Sstevel@tonic-gate 		}
24770Sstevel@tonic-gate 		*entry = entryNum;
24780Sstevel@tonic-gate 	} else if (title && strcmp(opt_dup, menu_cmds[TITLE_CMD]) == 0) {
24790Sstevel@tonic-gate 		*title = opt + (eq - opt_dup) + 1;
24800Sstevel@tonic-gate 	} else {
24810Sstevel@tonic-gate 		bam_error(INVALID_OPT, opt);
24820Sstevel@tonic-gate 		free(opt_dup);
24830Sstevel@tonic-gate 		return (BAM_ERROR);
24840Sstevel@tonic-gate 	}
24850Sstevel@tonic-gate 
24860Sstevel@tonic-gate 	free(opt_dup);
24870Sstevel@tonic-gate 	return (BAM_SUCCESS);
24880Sstevel@tonic-gate }
24890Sstevel@tonic-gate 
24900Sstevel@tonic-gate /*
24910Sstevel@tonic-gate  * If invoked with no titles/entries (opt == NULL)
24920Sstevel@tonic-gate  * only title lines in file are printed.
24930Sstevel@tonic-gate  *
24940Sstevel@tonic-gate  * If invoked with a title or entry #, all
24950Sstevel@tonic-gate  * lines in *every* matching entry are listed
24960Sstevel@tonic-gate  */
24970Sstevel@tonic-gate static error_t
24980Sstevel@tonic-gate list_entry(menu_t *mp, char *menu_path, char *opt)
24990Sstevel@tonic-gate {
25000Sstevel@tonic-gate 	line_t *lp;
25010Sstevel@tonic-gate 	int entry = ENTRY_INIT;
25020Sstevel@tonic-gate 	int found;
25030Sstevel@tonic-gate 	char *title = NULL;
25040Sstevel@tonic-gate 
25050Sstevel@tonic-gate 	assert(mp);
25060Sstevel@tonic-gate 	assert(menu_path);
25070Sstevel@tonic-gate 
25080Sstevel@tonic-gate 	if (mp->start == NULL) {
25090Sstevel@tonic-gate 		bam_error(NO_MENU, menu_path);
25100Sstevel@tonic-gate 		return (BAM_ERROR);
25110Sstevel@tonic-gate 	}
25120Sstevel@tonic-gate 
25130Sstevel@tonic-gate 	if (opt != NULL) {
25140Sstevel@tonic-gate 		if (selector(mp, opt, &entry, &title) != BAM_SUCCESS) {
25150Sstevel@tonic-gate 			return (BAM_ERROR);
25160Sstevel@tonic-gate 		}
25170Sstevel@tonic-gate 		assert((entry != ENTRY_INIT) ^ (title != NULL));
25180Sstevel@tonic-gate 	} else {
25190Sstevel@tonic-gate 		(void) read_globals(mp, menu_path, menu_cmds[DEFAULT_CMD], 0);
25200Sstevel@tonic-gate 		(void) read_globals(mp, menu_path, menu_cmds[TIMEOUT_CMD], 0);
25210Sstevel@tonic-gate 	}
25220Sstevel@tonic-gate 
25230Sstevel@tonic-gate 	found = 0;
25240Sstevel@tonic-gate 	for (lp = mp->start; lp; lp = lp->next) {
25250Sstevel@tonic-gate 		if (lp->flags == BAM_COMMENT || lp->flags == BAM_EMPTY)
25260Sstevel@tonic-gate 			continue;
25270Sstevel@tonic-gate 		if (opt == NULL && lp->flags == BAM_TITLE) {
25280Sstevel@tonic-gate 			bam_print(PRINT_TITLE, lp->entryNum,
25290Sstevel@tonic-gate 			    lp->arg);
25300Sstevel@tonic-gate 			found = 1;
25310Sstevel@tonic-gate 			continue;
25320Sstevel@tonic-gate 		}
25330Sstevel@tonic-gate 		if (entry != ENTRY_INIT && lp->entryNum == entry) {
25340Sstevel@tonic-gate 			bam_print(PRINT, lp->line);
25350Sstevel@tonic-gate 			found = 1;
25360Sstevel@tonic-gate 			continue;
25370Sstevel@tonic-gate 		}
25380Sstevel@tonic-gate 
25390Sstevel@tonic-gate 		/*
25400Sstevel@tonic-gate 		 * We set the entry value here so that all lines
25410Sstevel@tonic-gate 		 * in entry get printed. If we subsequently match
25420Sstevel@tonic-gate 		 * title in other entries, all lines in those
25430Sstevel@tonic-gate 		 * entries get printed as well.
25440Sstevel@tonic-gate 		 */
25450Sstevel@tonic-gate 		if (title && lp->flags == BAM_TITLE && lp->arg &&
25460Sstevel@tonic-gate 		    strncmp(title, lp->arg, strlen(title)) == 0) {
25470Sstevel@tonic-gate 			bam_print(PRINT, lp->line);
25480Sstevel@tonic-gate 			entry = lp->entryNum;
25490Sstevel@tonic-gate 			found = 1;
25500Sstevel@tonic-gate 			continue;
25510Sstevel@tonic-gate 		}
25520Sstevel@tonic-gate 	}
25530Sstevel@tonic-gate 
25540Sstevel@tonic-gate 	if (!found) {
25550Sstevel@tonic-gate 		bam_error(NO_MATCH_ENTRY);
25560Sstevel@tonic-gate 		return (BAM_ERROR);
25570Sstevel@tonic-gate 	}
25580Sstevel@tonic-gate 
25590Sstevel@tonic-gate 	return (BAM_SUCCESS);
25600Sstevel@tonic-gate }
25610Sstevel@tonic-gate 
25620Sstevel@tonic-gate static int
25630Sstevel@tonic-gate add_boot_entry(menu_t *mp,
25640Sstevel@tonic-gate 	char *title,
25650Sstevel@tonic-gate 	char *root,
25660Sstevel@tonic-gate 	char *kernel,
25670Sstevel@tonic-gate 	char *module)
25680Sstevel@tonic-gate {
25690Sstevel@tonic-gate 	int lineNum, entryNum;
25700Sstevel@tonic-gate 	char linebuf[BAM_MAXLINE];
25710Sstevel@tonic-gate 
25720Sstevel@tonic-gate 	assert(mp);
25730Sstevel@tonic-gate 
25740Sstevel@tonic-gate 	if (title == NULL) {
2575656Sszhou 		title = "Solaris";	/* default to Solaris */
25760Sstevel@tonic-gate 	}
25770Sstevel@tonic-gate 	if (kernel == NULL) {
25780Sstevel@tonic-gate 		bam_error(SUBOPT_MISS, menu_cmds[KERNEL_CMD]);
25790Sstevel@tonic-gate 		return (BAM_ERROR);
25800Sstevel@tonic-gate 	}
25810Sstevel@tonic-gate 	if (module == NULL) {
25820Sstevel@tonic-gate 		bam_error(SUBOPT_MISS, menu_cmds[MODULE_CMD]);
25830Sstevel@tonic-gate 		return (BAM_ERROR);
25840Sstevel@tonic-gate 	}
25850Sstevel@tonic-gate 
25860Sstevel@tonic-gate 	if (mp->start) {
25870Sstevel@tonic-gate 		lineNum = mp->end->lineNum;
25880Sstevel@tonic-gate 		entryNum = mp->end->entryNum;
25890Sstevel@tonic-gate 	} else {
25900Sstevel@tonic-gate 		lineNum = LINE_INIT;
25910Sstevel@tonic-gate 		entryNum = ENTRY_INIT;
25920Sstevel@tonic-gate 	}
25930Sstevel@tonic-gate 
25940Sstevel@tonic-gate 	/*
25950Sstevel@tonic-gate 	 * No separator for comment (HDR/FTR) commands
25960Sstevel@tonic-gate 	 * The syntax for comments is #<comment>
25970Sstevel@tonic-gate 	 */
25980Sstevel@tonic-gate 	(void) snprintf(linebuf, sizeof (linebuf), "%s%s",
25990Sstevel@tonic-gate 	    menu_cmds[COMMENT_CMD], BAM_HDR);
2600*662Sszhou 	line_parser(mp, linebuf, &lineNum, &entryNum);
26010Sstevel@tonic-gate 
26020Sstevel@tonic-gate 	(void) snprintf(linebuf, sizeof (linebuf), "%s%s%s",
26030Sstevel@tonic-gate 	    menu_cmds[TITLE_CMD], menu_cmds[SEP_CMD], title);
2604*662Sszhou 	line_parser(mp, linebuf, &lineNum, &entryNum);
2605*662Sszhou 
2606*662Sszhou 	if (root) {
2607*662Sszhou 		(void) snprintf(linebuf, sizeof (linebuf), "%s%s%s",
2608*662Sszhou 		    menu_cmds[ROOT_CMD], menu_cmds[SEP_CMD], root);
2609*662Sszhou 		line_parser(mp, linebuf, &lineNum, &entryNum);
26100Sstevel@tonic-gate 	}
26110Sstevel@tonic-gate 
26120Sstevel@tonic-gate 	(void) snprintf(linebuf, sizeof (linebuf), "%s%s%s",
26130Sstevel@tonic-gate 	    menu_cmds[KERNEL_CMD], menu_cmds[SEP_CMD], kernel);
2614*662Sszhou 	line_parser(mp, linebuf, &lineNum, &entryNum);
26150Sstevel@tonic-gate 
26160Sstevel@tonic-gate 	(void) snprintf(linebuf, sizeof (linebuf), "%s%s%s",
26170Sstevel@tonic-gate 	    menu_cmds[MODULE_CMD], menu_cmds[SEP_CMD], module);
2618*662Sszhou 	line_parser(mp, linebuf, &lineNum, &entryNum);
26190Sstevel@tonic-gate 
26200Sstevel@tonic-gate 	(void) snprintf(linebuf, sizeof (linebuf), "%s%s",
26210Sstevel@tonic-gate 	    menu_cmds[COMMENT_CMD], BAM_FTR);
2622*662Sszhou 	line_parser(mp, linebuf, &lineNum, &entryNum);
26230Sstevel@tonic-gate 
26240Sstevel@tonic-gate 	return (entryNum);
26250Sstevel@tonic-gate }
26260Sstevel@tonic-gate 
26270Sstevel@tonic-gate static error_t
26280Sstevel@tonic-gate do_delete(menu_t *mp, int entryNum)
26290Sstevel@tonic-gate {
2630*662Sszhou 	line_t *lp, *freed;
2631*662Sszhou 	entry_t *ent, *tmp;
26320Sstevel@tonic-gate 	int deleted;
26330Sstevel@tonic-gate 
26340Sstevel@tonic-gate 	assert(entryNum != ENTRY_INIT);
26350Sstevel@tonic-gate 
2636*662Sszhou 	ent = mp->entries;
2637*662Sszhou 	while (ent) {
2638*662Sszhou 		lp = ent->start;
2639*662Sszhou 		/* check entry number and make sure it's a bootadm entry */
2640*662Sszhou 		if (lp->flags != BAM_COMMENT ||
2641*662Sszhou 		    strcmp(lp->arg, BAM_HDR) != 0 ||
2642*662Sszhou 		    (entryNum != ALL_ENTRIES && lp->entryNum != entryNum)) {
2643*662Sszhou 			ent = ent->next;
26440Sstevel@tonic-gate 			continue;
26450Sstevel@tonic-gate 		}
26460Sstevel@tonic-gate 
2647*662Sszhou 		/* free the entry content */
2648*662Sszhou 		do {
2649*662Sszhou 			freed = lp;
2650*662Sszhou 			lp = lp->next;	/* prev stays the same */
2651*662Sszhou 			unlink_line(mp, freed);
2652*662Sszhou 			line_free(freed);
2653*662Sszhou 		} while (freed != ent->end);
2654*662Sszhou 
2655*662Sszhou 		/* free the entry_t structure */
2656*662Sszhou 		tmp = ent;
2657*662Sszhou 		ent = ent->next;
2658*662Sszhou 		if (tmp->prev)
2659*662Sszhou 			tmp->prev->next = ent;
26600Sstevel@tonic-gate 		else
2661*662Sszhou 			mp->entries = ent;
2662*662Sszhou 		if (ent)
2663*662Sszhou 			ent->prev = tmp->prev;
26640Sstevel@tonic-gate 		deleted = 1;
26650Sstevel@tonic-gate 	}
26660Sstevel@tonic-gate 
26670Sstevel@tonic-gate 	if (!deleted && entryNum != ALL_ENTRIES) {
26680Sstevel@tonic-gate 		bam_error(NO_BOOTADM_MATCH);
26690Sstevel@tonic-gate 		return (BAM_ERROR);
26700Sstevel@tonic-gate 	}
26710Sstevel@tonic-gate 
2672621Svikram 	/*
2673621Svikram 	 * Now that we have deleted an entry, update
2674621Svikram 	 * the entry numbering and the default cmd.
2675621Svikram 	 */
2676621Svikram 	update_numbering(mp);
2677621Svikram 
26780Sstevel@tonic-gate 	return (BAM_SUCCESS);
26790Sstevel@tonic-gate }
26800Sstevel@tonic-gate 
26810Sstevel@tonic-gate static error_t
26820Sstevel@tonic-gate delete_all_entries(menu_t *mp, char *menu_path, char *opt)
26830Sstevel@tonic-gate {
26840Sstevel@tonic-gate 	assert(mp);
26850Sstevel@tonic-gate 	assert(opt == NULL);
26860Sstevel@tonic-gate 
26870Sstevel@tonic-gate 	if (mp->start == NULL) {
26880Sstevel@tonic-gate 		bam_print(EMPTY_FILE, menu_path);
26890Sstevel@tonic-gate 		return (BAM_SUCCESS);
26900Sstevel@tonic-gate 	}
26910Sstevel@tonic-gate 
26920Sstevel@tonic-gate 	if (do_delete(mp, ALL_ENTRIES) != BAM_SUCCESS) {
26930Sstevel@tonic-gate 		return (BAM_ERROR);
26940Sstevel@tonic-gate 	}
26950Sstevel@tonic-gate 
26960Sstevel@tonic-gate 	return (BAM_WRITE);
26970Sstevel@tonic-gate }
26980Sstevel@tonic-gate 
26990Sstevel@tonic-gate static FILE *
2700*662Sszhou open_diskmap(char *root)
27010Sstevel@tonic-gate {
27020Sstevel@tonic-gate 	FILE *fp;
27030Sstevel@tonic-gate 	char cmd[PATH_MAX];
27040Sstevel@tonic-gate 
27050Sstevel@tonic-gate 	/* make sure we have a map file */
27060Sstevel@tonic-gate 	fp = fopen(GRUBDISK_MAP, "r");
27070Sstevel@tonic-gate 	if (fp == NULL) {
27080Sstevel@tonic-gate 		(void) snprintf(cmd, sizeof (cmd),
2709*662Sszhou 		    "%s%s > /dev/null", root, CREATE_DISKMAP);
27100Sstevel@tonic-gate 		(void) system(cmd);
27110Sstevel@tonic-gate 		fp = fopen(GRUBDISK_MAP, "r");
27120Sstevel@tonic-gate 	}
27130Sstevel@tonic-gate 	return (fp);
27140Sstevel@tonic-gate }
27150Sstevel@tonic-gate 
27160Sstevel@tonic-gate #define	SECTOR_SIZE	512
27170Sstevel@tonic-gate 
27180Sstevel@tonic-gate static int
27190Sstevel@tonic-gate get_partition(char *device)
27200Sstevel@tonic-gate {
27210Sstevel@tonic-gate 	int i, fd, is_pcfs, partno = -1;
27220Sstevel@tonic-gate 	struct mboot *mboot;
27230Sstevel@tonic-gate 	char boot_sect[SECTOR_SIZE];
27240Sstevel@tonic-gate 	char *wholedisk, *slice;
27250Sstevel@tonic-gate 
27260Sstevel@tonic-gate 	/* form whole disk (p0) */
27270Sstevel@tonic-gate 	slice = device + strlen(device) - 2;
27280Sstevel@tonic-gate 	is_pcfs = (*slice != 's');
27290Sstevel@tonic-gate 	if (!is_pcfs)
27300Sstevel@tonic-gate 		*slice = '\0';
27310Sstevel@tonic-gate 	wholedisk = s_calloc(1, strlen(device) + 3);
27320Sstevel@tonic-gate 	(void) snprintf(wholedisk, strlen(device) + 3, "%sp0", device);
27330Sstevel@tonic-gate 	if (!is_pcfs)
27340Sstevel@tonic-gate 		*slice = 's';
27350Sstevel@tonic-gate 
27360Sstevel@tonic-gate 	/* read boot sector */
27370Sstevel@tonic-gate 	fd = open(wholedisk, O_RDONLY);
27380Sstevel@tonic-gate 	free(wholedisk);
27390Sstevel@tonic-gate 	if (fd == -1 || read(fd, boot_sect, SECTOR_SIZE) != SECTOR_SIZE) {
27400Sstevel@tonic-gate 		return (partno);
27410Sstevel@tonic-gate 	}
27420Sstevel@tonic-gate 	(void) close(fd);
27430Sstevel@tonic-gate 
27440Sstevel@tonic-gate 	/* parse fdisk table */
27450Sstevel@tonic-gate 	mboot = (struct mboot *)((void *)boot_sect);
27460Sstevel@tonic-gate 	for (i = 0; i < FD_NUMPART; i++) {
27470Sstevel@tonic-gate 		struct ipart *part =
27480Sstevel@tonic-gate 		    (struct ipart *)(uintptr_t)mboot->parts + i;
27490Sstevel@tonic-gate 		if (is_pcfs) {	/* looking for solaris boot part */
27500Sstevel@tonic-gate 			if (part->systid == 0xbe) {
27510Sstevel@tonic-gate 				partno = i;
27520Sstevel@tonic-gate 				break;
27530Sstevel@tonic-gate 			}
27540Sstevel@tonic-gate 		} else {	/* look for solaris partition, old and new */
27550Sstevel@tonic-gate 			if (part->systid == SUNIXOS ||
27560Sstevel@tonic-gate 			    part->systid == SUNIXOS2) {
27570Sstevel@tonic-gate 				partno = i;
27580Sstevel@tonic-gate 				break;
27590Sstevel@tonic-gate 			}
27600Sstevel@tonic-gate 		}
27610Sstevel@tonic-gate 	}
27620Sstevel@tonic-gate 	return (partno);
27630Sstevel@tonic-gate }
27640Sstevel@tonic-gate 
27650Sstevel@tonic-gate static char *
27660Sstevel@tonic-gate get_grubdisk(char *rootdev, FILE *fp, int on_bootdev)
27670Sstevel@tonic-gate {
27680Sstevel@tonic-gate 	char *grubdisk;	/* (hd#,#,#) */
27690Sstevel@tonic-gate 	char *slice;
27700Sstevel@tonic-gate 	char *grubhd;
27710Sstevel@tonic-gate 	int fdiskpart;
27720Sstevel@tonic-gate 	int found = 0;
27730Sstevel@tonic-gate 	char *devname, *ctdname = strstr(rootdev, "dsk/");
27740Sstevel@tonic-gate 	char linebuf[PATH_MAX];
27750Sstevel@tonic-gate 
27760Sstevel@tonic-gate 	if (ctdname == NULL)
27770Sstevel@tonic-gate 		return (NULL);
27780Sstevel@tonic-gate 
27790Sstevel@tonic-gate 	ctdname += strlen("dsk/");
27800Sstevel@tonic-gate 	slice = strrchr(ctdname, 's');
27810Sstevel@tonic-gate 	if (slice)
27820Sstevel@tonic-gate 		*slice = '\0';
27830Sstevel@tonic-gate 
27840Sstevel@tonic-gate 	rewind(fp);
27850Sstevel@tonic-gate 	while (s_fgets(linebuf, sizeof (linebuf), fp) != NULL) {
27860Sstevel@tonic-gate 		grubhd = strtok(linebuf, " \t\n");
27870Sstevel@tonic-gate 		if (grubhd)
27880Sstevel@tonic-gate 			devname = strtok(NULL, " \t\n");
27890Sstevel@tonic-gate 		else
27900Sstevel@tonic-gate 			devname = NULL;
27910Sstevel@tonic-gate 		if (devname && strcmp(devname, ctdname) == 0) {
27920Sstevel@tonic-gate 			found = 1;
27930Sstevel@tonic-gate 			break;
27940Sstevel@tonic-gate 		}
27950Sstevel@tonic-gate 	}
27960Sstevel@tonic-gate 
27970Sstevel@tonic-gate 	if (slice)
27980Sstevel@tonic-gate 		*slice = 's';
27990Sstevel@tonic-gate 
28000Sstevel@tonic-gate 	if (found == 0) {
28010Sstevel@tonic-gate 		if (bam_verbose)
28020Sstevel@tonic-gate 			bam_print(DISKMAP_FAIL_NONFATAL, rootdev);
28030Sstevel@tonic-gate 		grubhd = "0";	/* assume disk 0 if can't match */
28040Sstevel@tonic-gate 	}
28050Sstevel@tonic-gate 
28060Sstevel@tonic-gate 	fdiskpart = get_partition(rootdev);
28070Sstevel@tonic-gate 	if (fdiskpart == -1)
28080Sstevel@tonic-gate 		return (NULL);
28090Sstevel@tonic-gate 
28100Sstevel@tonic-gate 	grubdisk = s_calloc(1, 10);
28110Sstevel@tonic-gate 	if (slice) {
28120Sstevel@tonic-gate 		(void) snprintf(grubdisk, 10, "(hd%s,%d,%c)",
28130Sstevel@tonic-gate 		    grubhd, fdiskpart, slice[1] + 'a' - '0');
28140Sstevel@tonic-gate 	} else
28150Sstevel@tonic-gate 		(void) snprintf(grubdisk, 10, "(hd%s,%d)",
28160Sstevel@tonic-gate 		    grubhd, fdiskpart);
28170Sstevel@tonic-gate 
28180Sstevel@tonic-gate 	/* if root not on bootdev, change GRUB disk to 0 */
28190Sstevel@tonic-gate 	if (!on_bootdev)
28200Sstevel@tonic-gate 		grubdisk[3] = '0';
28210Sstevel@tonic-gate 	return (grubdisk);
28220Sstevel@tonic-gate }
28230Sstevel@tonic-gate 
2824656Sszhou static char *
2825656Sszhou get_title(char *rootdir)
28260Sstevel@tonic-gate {
28270Sstevel@tonic-gate 	static char title[80];	/* from /etc/release */
2828*662Sszhou 	char *cp = NULL, release[PATH_MAX];
28290Sstevel@tonic-gate 	FILE *fp;
28300Sstevel@tonic-gate 
28310Sstevel@tonic-gate 	/* open the /etc/release file */
28320Sstevel@tonic-gate 	(void) snprintf(release, sizeof (release), "%s/etc/release", rootdir);
28330Sstevel@tonic-gate 
28340Sstevel@tonic-gate 	fp = fopen(release, "r");
28350Sstevel@tonic-gate 	if (fp == NULL)
2836656Sszhou 		return (NULL);
28370Sstevel@tonic-gate 
28380Sstevel@tonic-gate 	while (s_fgets(title, sizeof (title), fp) != NULL) {
28390Sstevel@tonic-gate 		cp = strstr(title, "Solaris");
28400Sstevel@tonic-gate 		if (cp)
28410Sstevel@tonic-gate 			break;
28420Sstevel@tonic-gate 	}
28430Sstevel@tonic-gate 	(void) fclose(fp);
2844*662Sszhou 	return (cp == NULL ? "Solaris" : cp);
28450Sstevel@tonic-gate }
28460Sstevel@tonic-gate 
28470Sstevel@tonic-gate static char *
28480Sstevel@tonic-gate get_special(char *mountp)
28490Sstevel@tonic-gate {
28500Sstevel@tonic-gate 	FILE *mntfp;
28510Sstevel@tonic-gate 	struct mnttab mp = {0}, mpref = {0};
28520Sstevel@tonic-gate 
28530Sstevel@tonic-gate 	mntfp = fopen(MNTTAB, "r");
28540Sstevel@tonic-gate 	if (mntfp == NULL) {
28550Sstevel@tonic-gate 		return (0);
28560Sstevel@tonic-gate 	}
28570Sstevel@tonic-gate 
28580Sstevel@tonic-gate 	if (*mountp == '\0')
28590Sstevel@tonic-gate 		mpref.mnt_mountp = "/";
28600Sstevel@tonic-gate 	else
28610Sstevel@tonic-gate 		mpref.mnt_mountp = mountp;
28620Sstevel@tonic-gate 	if (getmntany(mntfp, &mp, &mpref) != 0) {
28630Sstevel@tonic-gate 		(void) fclose(mntfp);
28640Sstevel@tonic-gate 		return (NULL);
28650Sstevel@tonic-gate 	}
28660Sstevel@tonic-gate 	(void) fclose(mntfp);
28670Sstevel@tonic-gate 
28680Sstevel@tonic-gate 	return (s_strdup(mp.mnt_special));
28690Sstevel@tonic-gate }
28700Sstevel@tonic-gate 
28710Sstevel@tonic-gate static char *
28720Sstevel@tonic-gate os_to_grubdisk(char *osdisk, int on_bootdev)
28730Sstevel@tonic-gate {
28740Sstevel@tonic-gate 	FILE *fp;
28750Sstevel@tonic-gate 	char *grubdisk;
28760Sstevel@tonic-gate 
28770Sstevel@tonic-gate 	/* translate /dev/dsk name to grub disk name */
2878*662Sszhou 	fp = open_diskmap("");
28790Sstevel@tonic-gate 	if (fp == NULL) {
28800Sstevel@tonic-gate 		bam_error(DISKMAP_FAIL, osdisk);
28810Sstevel@tonic-gate 		return (NULL);
28820Sstevel@tonic-gate 	}
28830Sstevel@tonic-gate 	grubdisk = get_grubdisk(osdisk, fp, on_bootdev);
28840Sstevel@tonic-gate 	(void) fclose(fp);
28850Sstevel@tonic-gate 	return (grubdisk);
28860Sstevel@tonic-gate }
28870Sstevel@tonic-gate 
28880Sstevel@tonic-gate /*
28890Sstevel@tonic-gate  * Check if root is on the boot device
28900Sstevel@tonic-gate  * Return 0 (false) on error
28910Sstevel@tonic-gate  */
28920Sstevel@tonic-gate static int
28930Sstevel@tonic-gate menu_on_bootdev(char *menu_root, FILE *fp)
28940Sstevel@tonic-gate {
28950Sstevel@tonic-gate 	int ret;
28960Sstevel@tonic-gate 	char *grubhd, *bootp, *special;
28970Sstevel@tonic-gate 
28980Sstevel@tonic-gate 	special = get_special(menu_root);
28990Sstevel@tonic-gate 	if (special == NULL)
29000Sstevel@tonic-gate 		return (0);
29010Sstevel@tonic-gate 	bootp = strstr(special, "p0:boot");
29020Sstevel@tonic-gate 	if (bootp)
29030Sstevel@tonic-gate 		*bootp = '\0';
29040Sstevel@tonic-gate 	grubhd = get_grubdisk(special, fp, 1);
29050Sstevel@tonic-gate 	free(special);
29060Sstevel@tonic-gate 
29070Sstevel@tonic-gate 	if (grubhd == NULL)
29080Sstevel@tonic-gate 		return (0);
29090Sstevel@tonic-gate 	ret = grubhd[3] == '0';
29100Sstevel@tonic-gate 	free(grubhd);
29110Sstevel@tonic-gate 	return (ret);
29120Sstevel@tonic-gate }
29130Sstevel@tonic-gate 
2914*662Sszhou /*
2915*662Sszhou  * look for matching bootadm entry with specified parameters
2916*662Sszhou  * Here are the rules (based on existing usage):
2917*662Sszhou  * - If title is specified, match on title only
2918*662Sszhou  * - Else, match on grubdisk and module (don't care about kernel line).
2919*662Sszhou  *   note that, if root_opt is non-zero, the absence of root line is
2920*662Sszhou  *   considered a match.
2921*662Sszhou  */
2922*662Sszhou static entry_t *
2923*662Sszhou find_boot_entry(menu_t *mp, char *title, char *root, char *module,
2924*662Sszhou     int root_opt, int *entry_num)
2925*662Sszhou {
2926*662Sszhou 	int i;
2927*662Sszhou 	line_t *lp;
2928*662Sszhou 	entry_t *ent;
2929*662Sszhou 
2930*662Sszhou 	/* find matching entry */
2931*662Sszhou 	for (i = 0, ent = mp->entries; ent; i++, ent = ent->next) {
2932*662Sszhou 		lp = ent->start;
2933*662Sszhou 
2934*662Sszhou 		/* first line of entry must be bootadm comment */
2935*662Sszhou 		lp = ent->start;
2936*662Sszhou 		if (lp->flags != BAM_COMMENT || strcmp(lp->arg, BAM_HDR) != 0) {
2937*662Sszhou 			continue;
2938*662Sszhou 		}
2939*662Sszhou 
2940*662Sszhou 		/* advance to title line */
2941*662Sszhou 		lp = lp->next;
2942*662Sszhou 		if (title) {
2943*662Sszhou 			if (lp->flags == BAM_TITLE && lp->arg &&
2944*662Sszhou 			    strcmp(lp->arg, title) == 0)
2945*662Sszhou 				break;
2946*662Sszhou 			continue;	/* check title only */
2947*662Sszhou 		}
2948*662Sszhou 
2949*662Sszhou 		lp = lp->next;	/* advance to root line */
2950*662Sszhou 		if (lp == NULL || strcmp(lp->cmd, menu_cmds[ROOT_CMD]) == 0) {
2951*662Sszhou 			/* root command found, match grub disk */
2952*662Sszhou 			if (strcmp(lp->arg, root) != 0) {
2953*662Sszhou 				continue;
2954*662Sszhou 			}
2955*662Sszhou 			lp = lp->next;	/* advance to kernel line */
2956*662Sszhou 		} else {
2957*662Sszhou 			/* no root command, see if root is optional */
2958*662Sszhou 			if (root_opt == 0) {
2959*662Sszhou 				continue;
2960*662Sszhou 			}
2961*662Sszhou 		}
2962*662Sszhou 
2963*662Sszhou 		if (lp == NULL || lp->next == NULL) {
2964*662Sszhou 			continue;
2965*662Sszhou 		}
2966*662Sszhou 
2967*662Sszhou 		/* check for matching module entry (failsafe or normal) */
2968*662Sszhou 		lp = lp->next;	/* advance to module line */
2969*662Sszhou 		if (strcmp(lp->cmd, menu_cmds[MODULE_CMD]) != 0 ||
2970*662Sszhou 		    strcmp(lp->arg, module) != 0) {
2971*662Sszhou 			continue;
2972*662Sszhou 		}
2973*662Sszhou 		break;	/* match found */
2974*662Sszhou 	}
2975*662Sszhou 
2976*662Sszhou 	*entry_num = i;
2977*662Sszhou 	return (ent);
2978*662Sszhou }
2979*662Sszhou 
2980*662Sszhou static int
2981*662Sszhou update_boot_entry(menu_t *mp, char *title, char *root, char *kernel,
2982*662Sszhou     char *module, int root_opt)
2983*662Sszhou {
2984*662Sszhou 	int i;
2985*662Sszhou 	entry_t *ent;
2986*662Sszhou 	line_t *lp;
2987*662Sszhou 	char linebuf[BAM_MAXLINE];
2988*662Sszhou 
2989*662Sszhou 	/* note: don't match on title, it's updated on upgrade */
2990*662Sszhou 	ent = find_boot_entry(mp, NULL, root, module, root_opt, &i);
2991*662Sszhou 	if (ent == NULL)
2992*662Sszhou 		return (add_boot_entry(mp, title, root_opt ? NULL : root,
2993*662Sszhou 		    kernel, module));
2994*662Sszhou 
2995*662Sszhou 	/* replace title of exiting entry and delete root line */
2996*662Sszhou 	lp = ent->start;
2997*662Sszhou 	lp = lp->next;	/* title line */
2998*662Sszhou 	(void) snprintf(linebuf, sizeof (linebuf), "%s%s%s",
2999*662Sszhou 	    menu_cmds[TITLE_CMD], menu_cmds[SEP_CMD], title);
3000*662Sszhou 	free(lp->arg);
3001*662Sszhou 	free(lp->line);
3002*662Sszhou 	lp->arg = s_strdup(title);
3003*662Sszhou 	lp->line = s_strdup(linebuf);
3004*662Sszhou 
3005*662Sszhou 	lp = lp->next;	/* root line */
3006*662Sszhou 	if (strcmp(lp->cmd, menu_cmds[ROOT_CMD]) == 0) {
3007*662Sszhou 		if (root_opt) {		/* root line not needed */
3008*662Sszhou 			line_t *tmp = lp;
3009*662Sszhou 			lp = lp->next;
3010*662Sszhou 			unlink_line(mp, tmp);
3011*662Sszhou 			line_free(tmp);
3012*662Sszhou 		} else
3013*662Sszhou 			lp = lp->next;
3014*662Sszhou 	}
3015*662Sszhou 	return (i);
3016*662Sszhou }
3017*662Sszhou 
30180Sstevel@tonic-gate /*ARGSUSED*/
30190Sstevel@tonic-gate static error_t
30200Sstevel@tonic-gate update_entry(menu_t *mp, char *menu_root, char *opt)
30210Sstevel@tonic-gate {
30220Sstevel@tonic-gate 	FILE *fp;
30230Sstevel@tonic-gate 	int entry;
30240Sstevel@tonic-gate 	char *grubdisk, *title, *osdev, *osroot;
3025*662Sszhou 	struct stat sbuf;
3026*662Sszhou 	char failsafe[256];
30270Sstevel@tonic-gate 
30280Sstevel@tonic-gate 	assert(mp);
30290Sstevel@tonic-gate 	assert(opt);
30300Sstevel@tonic-gate 
30310Sstevel@tonic-gate 	osdev = strtok(opt, ",");
30320Sstevel@tonic-gate 	osroot = strtok(NULL, ",");
30330Sstevel@tonic-gate 	if (osroot == NULL)
30340Sstevel@tonic-gate 		osroot = menu_root;
30350Sstevel@tonic-gate 	title = get_title(osroot);
30360Sstevel@tonic-gate 
30370Sstevel@tonic-gate 	/* translate /dev/dsk name to grub disk name */
3038*662Sszhou 	fp = open_diskmap(osroot);
30390Sstevel@tonic-gate 	if (fp == NULL) {
30400Sstevel@tonic-gate 		bam_error(DISKMAP_FAIL, osdev);
30410Sstevel@tonic-gate 		return (BAM_ERROR);
30420Sstevel@tonic-gate 	}
30430Sstevel@tonic-gate 	grubdisk = get_grubdisk(osdev, fp, menu_on_bootdev(menu_root, fp));
30440Sstevel@tonic-gate 	(void) fclose(fp);
30450Sstevel@tonic-gate 	if (grubdisk == NULL) {
30460Sstevel@tonic-gate 		bam_error(DISKMAP_FAIL, osdev);
30470Sstevel@tonic-gate 		return (BAM_ERROR);
30480Sstevel@tonic-gate 	}
30490Sstevel@tonic-gate 
30500Sstevel@tonic-gate 	/* add the entry for normal Solaris */
3051*662Sszhou 	entry = update_boot_entry(mp, title, grubdisk,
30520Sstevel@tonic-gate 	    "/platform/i86pc/multiboot",
3053*662Sszhou 	    "/platform/i86pc/boot_archive",
3054*662Sszhou 	    osroot == menu_root);
30550Sstevel@tonic-gate 
30560Sstevel@tonic-gate 	/* add the entry for failsafe archive */
3057*662Sszhou 	(void) snprintf(failsafe, sizeof (failsafe),
3058*662Sszhou 	    "%s/boot/x86.miniroot-safe", osroot);
3059*662Sszhou 	if (stat(failsafe, &sbuf) == 0)
3060*662Sszhou 		(void) update_boot_entry(mp, "Solaris failsafe", grubdisk,
3061*662Sszhou 		    "/boot/multiboot kernel/unix -s",
3062*662Sszhou 		    "/boot/x86.miniroot-safe",
3063*662Sszhou 		    osroot == menu_root);
30640Sstevel@tonic-gate 	free(grubdisk);
30650Sstevel@tonic-gate 
30660Sstevel@tonic-gate 	if (entry == BAM_ERROR) {
30670Sstevel@tonic-gate 		return (BAM_ERROR);
30680Sstevel@tonic-gate 	}
30690Sstevel@tonic-gate 	(void) set_global(mp, menu_cmds[DEFAULT_CMD], entry);
30700Sstevel@tonic-gate 	return (BAM_WRITE);
30710Sstevel@tonic-gate }
30720Sstevel@tonic-gate 
3073316Svikram static char *
3074316Svikram read_grub_root(void)
3075316Svikram {
3076316Svikram 	FILE *fp;
3077316Svikram 	struct stat sb;
3078316Svikram 	char buf[BAM_MAXLINE];
3079316Svikram 	char *rootstr;
3080316Svikram 
3081316Svikram 	if (stat(GRUB_slice, &sb) != 0) {
3082316Svikram 		bam_error(MISSING_SLICE_FILE, GRUB_slice, strerror(errno));
3083316Svikram 		return (NULL);
3084316Svikram 	}
3085316Svikram 
3086316Svikram 	if (stat(GRUB_root, &sb) != 0) {
3087316Svikram 		bam_error(MISSING_ROOT_FILE, GRUB_root, strerror(errno));
3088316Svikram 		return (NULL);
3089316Svikram 	}
3090316Svikram 
3091316Svikram 	fp = fopen(GRUB_root, "r");
3092316Svikram 	if (fp == NULL) {
3093316Svikram 		bam_error(OPEN_FAIL, GRUB_root, strerror(errno));
3094316Svikram 		return (NULL);
3095316Svikram 	}
3096316Svikram 
3097316Svikram 	if (s_fgets(buf, sizeof (buf), fp) == NULL) {
3098316Svikram 		bam_error(EMPTY_FILE, GRUB_root, strerror(errno));
3099316Svikram 		(void) fclose(fp);
3100316Svikram 		return (NULL);
3101316Svikram 	}
3102316Svikram 
3103316Svikram 	/*
3104316Svikram 	 * Copy buf here as check below may trash the buffer
3105316Svikram 	 */
3106316Svikram 	rootstr = s_strdup(buf);
3107316Svikram 
3108316Svikram 	if (s_fgets(buf, sizeof (buf), fp) != NULL) {
3109316Svikram 		bam_error(BAD_ROOT_FILE, GRUB_root);
3110316Svikram 		free(rootstr);
3111316Svikram 		rootstr = NULL;
3112316Svikram 	}
3113316Svikram 
3114316Svikram 	(void) fclose(fp);
3115316Svikram 
3116316Svikram 	return (rootstr);
3117316Svikram }
3118316Svikram 
3119*662Sszhou static void
3120*662Sszhou save_default_entry(menu_t *mp)
3121*662Sszhou {
3122*662Sszhou 	int lineNum, entryNum;
3123*662Sszhou 	int entry = 0;	/* default is 0 */
3124*662Sszhou 	char linebuf[BAM_MAXLINE];
3125*662Sszhou 	line_t *lp = mp->curdefault;
3126*662Sszhou 
3127*662Sszhou 	if (lp)
3128*662Sszhou 		entry = s_strtol(lp->arg);
3129*662Sszhou 
3130*662Sszhou 	(void) snprintf(linebuf, sizeof (linebuf), "#%s%d", BAM_OLDDEF, entry);
3131*662Sszhou 	line_parser(mp, linebuf, &lineNum, &entryNum);
3132*662Sszhou }
3133*662Sszhou 
3134*662Sszhou static void
3135*662Sszhou restore_default_entry(menu_t *mp)
3136*662Sszhou {
3137*662Sszhou 	int entry;
3138*662Sszhou 	char *str;
3139*662Sszhou 	line_t *lp = mp->olddefault;
3140*662Sszhou 
3141*662Sszhou 	if (lp == NULL)
3142*662Sszhou 		return;		/* nothing to restore */
3143*662Sszhou 
3144*662Sszhou 	str = lp->arg + strlen(BAM_OLDDEF);
3145*662Sszhou 	entry = s_strtol(str);
3146*662Sszhou 	(void) set_global(mp, menu_cmds[DEFAULT_CMD], entry);
3147*662Sszhou 
3148*662Sszhou 	/* delete saved old default line */
3149*662Sszhou 	mp->olddefault = NULL;
3150*662Sszhou 	unlink_line(mp, lp);
3151*662Sszhou 	line_free(lp);
3152*662Sszhou }
3153*662Sszhou 
31540Sstevel@tonic-gate /*
31550Sstevel@tonic-gate  * This function is for supporting reboot with args.
31560Sstevel@tonic-gate  * The opt value can be:
31570Sstevel@tonic-gate  * NULL		delete temp entry, if present
31580Sstevel@tonic-gate  * entry=#	switches default entry to 1
31590Sstevel@tonic-gate  * else		treated as boot-args and setup a temperary menu entry
31600Sstevel@tonic-gate  *		and make it the default
31610Sstevel@tonic-gate  */
31620Sstevel@tonic-gate #define	REBOOT_TITLE	"Solaris_reboot_transient"
31630Sstevel@tonic-gate 
3164*662Sszhou /*ARGSUSED*/
31650Sstevel@tonic-gate static error_t
31660Sstevel@tonic-gate update_temp(menu_t *mp, char *menupath, char *opt)
31670Sstevel@tonic-gate {
31680Sstevel@tonic-gate 	int entry;
31690Sstevel@tonic-gate 	char *grubdisk, *rootdev;
31700Sstevel@tonic-gate 	char kernbuf[1024];
3171316Svikram 	struct stat sb;
31720Sstevel@tonic-gate 
31730Sstevel@tonic-gate 	assert(mp);
31740Sstevel@tonic-gate 
3175*662Sszhou 	/* If no option, delete exiting reboot menu entry */
3176*662Sszhou 	if (opt == NULL) {
3177*662Sszhou 		entry_t *ent = find_boot_entry(mp, REBOOT_TITLE, NULL, NULL,
3178*662Sszhou 		    0, &entry);
3179*662Sszhou 		if (ent == NULL)	/* not found is ok */
3180*662Sszhou 			return (BAM_SUCCESS);
3181*662Sszhou 		(void) do_delete(mp, entry);
3182*662Sszhou 		restore_default_entry(mp);
3183*662Sszhou 		return (BAM_WRITE);
3184*662Sszhou 	}
3185*662Sszhou 
3186*662Sszhou 	/* if entry= is specified, set the default entry */
3187*662Sszhou 	if (strncmp(opt, "entry=", strlen("entry=")) == 0 &&
31880Sstevel@tonic-gate 	    selector(mp, opt, &entry, NULL) == BAM_SUCCESS) {
31890Sstevel@tonic-gate 		/* this is entry=# option */
31900Sstevel@tonic-gate 		return (set_global(mp, menu_cmds[DEFAULT_CMD], entry));
31910Sstevel@tonic-gate 	}
31920Sstevel@tonic-gate 
31930Sstevel@tonic-gate 	/*
31940Sstevel@tonic-gate 	 * add a new menu entry base on opt and make it the default
31950Sstevel@tonic-gate 	 */
3196316Svikram 	grubdisk = NULL;
3197316Svikram 	if (stat(GRUB_slice, &sb) != 0) {
3198316Svikram 		/*
3199316Svikram 		 * 1. First get root disk name from mnttab
3200316Svikram 		 * 2. Translate disk name to grub name
3201316Svikram 		 * 3. Add the new menu entry
3202316Svikram 		 */
3203316Svikram 		rootdev = get_special("/");
3204316Svikram 		if (rootdev) {
3205316Svikram 			grubdisk = os_to_grubdisk(rootdev, 1);
3206316Svikram 			free(rootdev);
3207316Svikram 		}
3208316Svikram 	} else {
3209316Svikram 		/*
3210316Svikram 		 * This is an LU BE. The GRUB_root file
3211316Svikram 		 * contains entry for GRUB's "root" cmd.
3212316Svikram 		 */
3213316Svikram 		grubdisk = read_grub_root();
32140Sstevel@tonic-gate 	}
32150Sstevel@tonic-gate 	if (grubdisk == NULL) {
3216316Svikram 		bam_error(REBOOT_WITH_ARGS_FAILED);
32170Sstevel@tonic-gate 		return (BAM_ERROR);
32180Sstevel@tonic-gate 	}
32190Sstevel@tonic-gate 
32200Sstevel@tonic-gate 	/* add an entry for Solaris reboot */
32210Sstevel@tonic-gate 	(void) snprintf(kernbuf, sizeof (kernbuf),
32220Sstevel@tonic-gate 	    "/platform/i86pc/multiboot %s", opt);
32230Sstevel@tonic-gate 	entry = add_boot_entry(mp, REBOOT_TITLE, grubdisk, kernbuf,
32240Sstevel@tonic-gate 	    "/platform/i86pc/boot_archive");
32250Sstevel@tonic-gate 	free(grubdisk);
32260Sstevel@tonic-gate 
32270Sstevel@tonic-gate 	if (entry == BAM_ERROR) {
3228316Svikram 		bam_error(REBOOT_WITH_ARGS_FAILED);
32290Sstevel@tonic-gate 		return (BAM_ERROR);
32300Sstevel@tonic-gate 	}
3231*662Sszhou 
3232*662Sszhou 	save_default_entry(mp);
32330Sstevel@tonic-gate 	(void) set_global(mp, menu_cmds[DEFAULT_CMD], entry);
32340Sstevel@tonic-gate 	return (BAM_WRITE);
32350Sstevel@tonic-gate }
32360Sstevel@tonic-gate 
32370Sstevel@tonic-gate static error_t
32380Sstevel@tonic-gate set_global(menu_t *mp, char *globalcmd, int val)
32390Sstevel@tonic-gate {
32400Sstevel@tonic-gate 	line_t *lp, *found, *last;
32410Sstevel@tonic-gate 	char *cp, *str;
32420Sstevel@tonic-gate 	char prefix[BAM_MAXLINE];
32430Sstevel@tonic-gate 	size_t len;
32440Sstevel@tonic-gate 
32450Sstevel@tonic-gate 	assert(mp);
32460Sstevel@tonic-gate 	assert(globalcmd);
32470Sstevel@tonic-gate 
3248316Svikram 	if (strcmp(globalcmd, menu_cmds[DEFAULT_CMD]) == 0) {
3249316Svikram 		if (val < 0 || mp->end == NULL || val > mp->end->entryNum) {
3250316Svikram 			(void) snprintf(prefix, sizeof (prefix), "%d", val);
3251316Svikram 			bam_error(INVALID_ENTRY, prefix);
3252316Svikram 			return (BAM_ERROR);
3253316Svikram 		}
3254316Svikram 	}
3255316Svikram 
32560Sstevel@tonic-gate 	found = last = NULL;
32570Sstevel@tonic-gate 	for (lp = mp->start; lp; lp = lp->next) {
32580Sstevel@tonic-gate 		if (lp->flags != BAM_GLOBAL)
32590Sstevel@tonic-gate 			continue;
32600Sstevel@tonic-gate 
32610Sstevel@tonic-gate 		last = lp; /* track the last global found */
32620Sstevel@tonic-gate 
32630Sstevel@tonic-gate 		if (lp->cmd == NULL) {
32640Sstevel@tonic-gate 			bam_error(NO_CMD, lp->lineNum);
32650Sstevel@tonic-gate 			continue;
32660Sstevel@tonic-gate 		}
32670Sstevel@tonic-gate 		if (strcmp(globalcmd, lp->cmd) != 0)
32680Sstevel@tonic-gate 			continue;
32690Sstevel@tonic-gate 
32700Sstevel@tonic-gate 		if (found) {
32710Sstevel@tonic-gate 			bam_error(DUP_CMD, globalcmd, lp->lineNum, bam_root);
32720Sstevel@tonic-gate 		}
32730Sstevel@tonic-gate 		found = lp;
32740Sstevel@tonic-gate 	}
32750Sstevel@tonic-gate 
32760Sstevel@tonic-gate 	if (found == NULL) {
32770Sstevel@tonic-gate 		lp = s_calloc(1, sizeof (line_t));
32780Sstevel@tonic-gate 		if (last == NULL) {
32790Sstevel@tonic-gate 			lp->next = mp->start;
32800Sstevel@tonic-gate 			mp->start = lp;
32810Sstevel@tonic-gate 			mp->end = (mp->end) ? mp->end : lp;
32820Sstevel@tonic-gate 		} else {
32830Sstevel@tonic-gate 			lp->next = last->next;
32840Sstevel@tonic-gate 			last->next = lp;
32850Sstevel@tonic-gate 			if (lp->next == NULL)
32860Sstevel@tonic-gate 				mp->end = lp;
32870Sstevel@tonic-gate 		}
32880Sstevel@tonic-gate 		lp->flags = BAM_GLOBAL; /* other fields not needed for writes */
32890Sstevel@tonic-gate 		len = strlen(globalcmd) + strlen(menu_cmds[SEP_CMD]);
32900Sstevel@tonic-gate 		len += 10;	/* val < 10 digits */
32910Sstevel@tonic-gate 		lp->line = s_calloc(1, len);
32920Sstevel@tonic-gate 		(void) snprintf(lp->line, len, "%s%s%d",
32930Sstevel@tonic-gate 		    globalcmd, menu_cmds[SEP_CMD], val);
32940Sstevel@tonic-gate 		return (BAM_WRITE);
32950Sstevel@tonic-gate 	}
32960Sstevel@tonic-gate 
32970Sstevel@tonic-gate 	/*
32980Sstevel@tonic-gate 	 * We are changing an existing entry. Retain any prefix whitespace,
32990Sstevel@tonic-gate 	 * but overwrite everything else. This preserves tabs added for
33000Sstevel@tonic-gate 	 * readability.
33010Sstevel@tonic-gate 	 */
33020Sstevel@tonic-gate 	str = found->line;
33030Sstevel@tonic-gate 	cp = prefix;
33040Sstevel@tonic-gate 	while (*str == ' ' || *str == '\t')
33050Sstevel@tonic-gate 		*(cp++) = *(str++);
33060Sstevel@tonic-gate 	*cp = '\0'; /* Terminate prefix */
33070Sstevel@tonic-gate 	len = strlen(prefix) + strlen(globalcmd);
33080Sstevel@tonic-gate 	len += strlen(menu_cmds[SEP_CMD]) + 10;
33090Sstevel@tonic-gate 
33100Sstevel@tonic-gate 	free(found->line);
33110Sstevel@tonic-gate 	found->line = s_calloc(1, len);
33120Sstevel@tonic-gate 	(void) snprintf(found->line, len,
33130Sstevel@tonic-gate 		"%s%s%s%d", prefix, globalcmd, menu_cmds[SEP_CMD], val);
33140Sstevel@tonic-gate 
33150Sstevel@tonic-gate 	return (BAM_WRITE); /* need a write to menu */
33160Sstevel@tonic-gate }
33170Sstevel@tonic-gate 
33180Sstevel@tonic-gate /*ARGSUSED*/
33190Sstevel@tonic-gate static error_t
33200Sstevel@tonic-gate set_option(menu_t *mp, char *menu_path, char *opt)
33210Sstevel@tonic-gate {
33220Sstevel@tonic-gate 	int optnum, optval;
33230Sstevel@tonic-gate 	char *val;
33240Sstevel@tonic-gate 
33250Sstevel@tonic-gate 	assert(mp);
33260Sstevel@tonic-gate 	assert(opt);
33270Sstevel@tonic-gate 
33280Sstevel@tonic-gate 	val = strchr(opt, '=');
33290Sstevel@tonic-gate 	if (val == NULL) {
33300Sstevel@tonic-gate 		bam_error(INVALID_ENTRY, opt);
33310Sstevel@tonic-gate 		return (BAM_ERROR);
33320Sstevel@tonic-gate 	}
33330Sstevel@tonic-gate 
33340Sstevel@tonic-gate 	*val = '\0';
33350Sstevel@tonic-gate 	if (strcmp(opt, "default") == 0) {
33360Sstevel@tonic-gate 		optnum = DEFAULT_CMD;
33370Sstevel@tonic-gate 	} else if (strcmp(opt, "timeout") == 0) {
33380Sstevel@tonic-gate 		optnum = TIMEOUT_CMD;
33390Sstevel@tonic-gate 	} else {
33400Sstevel@tonic-gate 		bam_error(INVALID_ENTRY, opt);
33410Sstevel@tonic-gate 		return (BAM_ERROR);
33420Sstevel@tonic-gate 	}
33430Sstevel@tonic-gate 
33440Sstevel@tonic-gate 	optval = s_strtol(val + 1);
33450Sstevel@tonic-gate 	*val = '=';
33460Sstevel@tonic-gate 	return (set_global(mp, menu_cmds[optnum], optval));
33470Sstevel@tonic-gate }
33480Sstevel@tonic-gate 
33490Sstevel@tonic-gate /*
33500Sstevel@tonic-gate  * The quiet argument suppresses messages. This is used
33510Sstevel@tonic-gate  * when invoked in the context of other commands (e.g. list_entry)
33520Sstevel@tonic-gate  */
33530Sstevel@tonic-gate static error_t
33540Sstevel@tonic-gate read_globals(menu_t *mp, char *menu_path, char *globalcmd, int quiet)
33550Sstevel@tonic-gate {
33560Sstevel@tonic-gate 	line_t *lp;
33570Sstevel@tonic-gate 	char *arg;
33580Sstevel@tonic-gate 	int done, ret = BAM_SUCCESS;
33590Sstevel@tonic-gate 
33600Sstevel@tonic-gate 	assert(mp);
33610Sstevel@tonic-gate 	assert(menu_path);
33620Sstevel@tonic-gate 	assert(globalcmd);
33630Sstevel@tonic-gate 
33640Sstevel@tonic-gate 	if (mp->start == NULL) {
33650Sstevel@tonic-gate 		if (!quiet)
33660Sstevel@tonic-gate 			bam_error(NO_MENU, menu_path);
33670Sstevel@tonic-gate 		return (BAM_ERROR);
33680Sstevel@tonic-gate 	}
33690Sstevel@tonic-gate 
33700Sstevel@tonic-gate 	done = 0;
33710Sstevel@tonic-gate 	for (lp = mp->start; lp; lp = lp->next) {
33720Sstevel@tonic-gate 		if (lp->flags != BAM_GLOBAL)
33730Sstevel@tonic-gate 			continue;
33740Sstevel@tonic-gate 
33750Sstevel@tonic-gate 		if (lp->cmd == NULL) {
33760Sstevel@tonic-gate 			if (!quiet)
33770Sstevel@tonic-gate 				bam_error(NO_CMD, lp->lineNum);
33780Sstevel@tonic-gate 			continue;
33790Sstevel@tonic-gate 		}
33800Sstevel@tonic-gate 
33810Sstevel@tonic-gate 		if (strcmp(globalcmd, lp->cmd) != 0)
33820Sstevel@tonic-gate 			continue;
33830Sstevel@tonic-gate 
33840Sstevel@tonic-gate 		/* Found global. Check for duplicates */
33850Sstevel@tonic-gate 		if (done && !quiet) {
33860Sstevel@tonic-gate 			bam_error(DUP_CMD, globalcmd, lp->lineNum, bam_root);
33870Sstevel@tonic-gate 			ret = BAM_ERROR;
33880Sstevel@tonic-gate 		}
33890Sstevel@tonic-gate 
33900Sstevel@tonic-gate 		arg = lp->arg ? lp->arg : "";
33910Sstevel@tonic-gate 		bam_print(GLOBAL_CMD, globalcmd, arg);
33920Sstevel@tonic-gate 		done = 1;
33930Sstevel@tonic-gate 	}
33940Sstevel@tonic-gate 
33950Sstevel@tonic-gate 	if (!done && bam_verbose)
33960Sstevel@tonic-gate 		bam_print(NO_ENTRY, globalcmd);
33970Sstevel@tonic-gate 
33980Sstevel@tonic-gate 	return (ret);
33990Sstevel@tonic-gate }
34000Sstevel@tonic-gate 
34010Sstevel@tonic-gate static error_t
34020Sstevel@tonic-gate menu_write(char *root, menu_t *mp)
34030Sstevel@tonic-gate {
34040Sstevel@tonic-gate 	return (list2file(root, MENU_TMP, GRUB_MENU, mp->start));
34050Sstevel@tonic-gate }
34060Sstevel@tonic-gate 
34070Sstevel@tonic-gate static void
34080Sstevel@tonic-gate line_free(line_t *lp)
34090Sstevel@tonic-gate {
34100Sstevel@tonic-gate 	if (lp == NULL)
34110Sstevel@tonic-gate 		return;
34120Sstevel@tonic-gate 
34130Sstevel@tonic-gate 	if (lp->cmd)
34140Sstevel@tonic-gate 		free(lp->cmd);
34150Sstevel@tonic-gate 	if (lp->sep)
34160Sstevel@tonic-gate 		free(lp->sep);
34170Sstevel@tonic-gate 	if (lp->arg)
34180Sstevel@tonic-gate 		free(lp->arg);
34190Sstevel@tonic-gate 	if (lp->line)
34200Sstevel@tonic-gate 		free(lp->line);
34210Sstevel@tonic-gate 	free(lp);
34220Sstevel@tonic-gate }
34230Sstevel@tonic-gate 
34240Sstevel@tonic-gate static void
34250Sstevel@tonic-gate linelist_free(line_t *start)
34260Sstevel@tonic-gate {
34270Sstevel@tonic-gate 	line_t *lp;
34280Sstevel@tonic-gate 
34290Sstevel@tonic-gate 	while (start) {
34300Sstevel@tonic-gate 		lp = start;
34310Sstevel@tonic-gate 		start = start->next;
34320Sstevel@tonic-gate 		line_free(lp);
34330Sstevel@tonic-gate 	}
34340Sstevel@tonic-gate }
34350Sstevel@tonic-gate 
34360Sstevel@tonic-gate static void
34370Sstevel@tonic-gate filelist_free(filelist_t *flistp)
34380Sstevel@tonic-gate {
34390Sstevel@tonic-gate 	linelist_free(flistp->head);
34400Sstevel@tonic-gate 	flistp->head = NULL;
34410Sstevel@tonic-gate 	flistp->tail = NULL;
34420Sstevel@tonic-gate }
34430Sstevel@tonic-gate 
34440Sstevel@tonic-gate static void
34450Sstevel@tonic-gate menu_free(menu_t *mp)
34460Sstevel@tonic-gate {
3447*662Sszhou 	entry_t *ent, *tmp;
34480Sstevel@tonic-gate 	assert(mp);
34490Sstevel@tonic-gate 
34500Sstevel@tonic-gate 	if (mp->start)
34510Sstevel@tonic-gate 		linelist_free(mp->start);
3452*662Sszhou 	ent = mp->entries;
3453*662Sszhou 	while (ent) {
3454*662Sszhou 		tmp = ent;
3455*662Sszhou 		ent = tmp->next;
3456*662Sszhou 		free(tmp);
3457*662Sszhou 	}
3458*662Sszhou 
34590Sstevel@tonic-gate 	free(mp);
34600Sstevel@tonic-gate }
34610Sstevel@tonic-gate 
34620Sstevel@tonic-gate /*
34630Sstevel@tonic-gate  * Utility routines
34640Sstevel@tonic-gate  */
34650Sstevel@tonic-gate 
34660Sstevel@tonic-gate 
34670Sstevel@tonic-gate /*
34680Sstevel@tonic-gate  * Returns 0 on success
34690Sstevel@tonic-gate  * Any other value indicates an error
34700Sstevel@tonic-gate  */
34710Sstevel@tonic-gate static int
34720Sstevel@tonic-gate exec_cmd(char *cmdline, char *output, int64_t osize)
34730Sstevel@tonic-gate {
34740Sstevel@tonic-gate 	char buf[BUFSIZ];
34750Sstevel@tonic-gate 	int ret;
34760Sstevel@tonic-gate 	FILE *ptr;
34770Sstevel@tonic-gate 	size_t len;
34780Sstevel@tonic-gate 	sigset_t set;
34790Sstevel@tonic-gate 	void (*disp)(int);
34800Sstevel@tonic-gate 
34810Sstevel@tonic-gate 	/*
34820Sstevel@tonic-gate 	 * For security
34830Sstevel@tonic-gate 	 * - only absolute paths are allowed
34840Sstevel@tonic-gate 	 * - set IFS to space and tab
34850Sstevel@tonic-gate 	 */
34860Sstevel@tonic-gate 	if (*cmdline != '/') {
34870Sstevel@tonic-gate 		bam_error(ABS_PATH_REQ, cmdline);
34880Sstevel@tonic-gate 		return (-1);
34890Sstevel@tonic-gate 	}
34900Sstevel@tonic-gate 	(void) putenv("IFS= \t");
34910Sstevel@tonic-gate 
34920Sstevel@tonic-gate 	/*
34930Sstevel@tonic-gate 	 * We may have been exec'ed with SIGCHLD blocked
34940Sstevel@tonic-gate 	 * unblock it here
34950Sstevel@tonic-gate 	 */
34960Sstevel@tonic-gate 	(void) sigemptyset(&set);
34970Sstevel@tonic-gate 	(void) sigaddset(&set, SIGCHLD);
34980Sstevel@tonic-gate 	if (sigprocmask(SIG_UNBLOCK, &set, NULL) != 0) {
34990Sstevel@tonic-gate 		bam_error(CANT_UNBLOCK_SIGCHLD, strerror(errno));
35000Sstevel@tonic-gate 		return (-1);
35010Sstevel@tonic-gate 	}
35020Sstevel@tonic-gate 
35030Sstevel@tonic-gate 	/*
35040Sstevel@tonic-gate 	 * Set SIGCHLD disposition to SIG_DFL for popen/pclose
35050Sstevel@tonic-gate 	 */
35060Sstevel@tonic-gate 	disp = sigset(SIGCHLD, SIG_DFL);
35070Sstevel@tonic-gate 	if (disp == SIG_ERR) {
35080Sstevel@tonic-gate 		bam_error(FAILED_SIG, strerror(errno));
35090Sstevel@tonic-gate 		return (-1);
35100Sstevel@tonic-gate 	}
35110Sstevel@tonic-gate 	if (disp == SIG_HOLD) {
35120Sstevel@tonic-gate 		bam_error(BLOCKED_SIG, cmdline);
35130Sstevel@tonic-gate 		return (-1);
35140Sstevel@tonic-gate 	}
35150Sstevel@tonic-gate 
35160Sstevel@tonic-gate 	ptr = popen(cmdline, "r");
35170Sstevel@tonic-gate 	if (ptr == NULL) {
35180Sstevel@tonic-gate 		bam_error(POPEN_FAIL, cmdline, strerror(errno));
35190Sstevel@tonic-gate 		return (-1);
35200Sstevel@tonic-gate 	}
35210Sstevel@tonic-gate 
35220Sstevel@tonic-gate 	/*
35230Sstevel@tonic-gate 	 * If we simply do a pclose() following a popen(), pclose()
35240Sstevel@tonic-gate 	 * will close the reader end of the pipe immediately even
35250Sstevel@tonic-gate 	 * if the child process has not started/exited. pclose()
35260Sstevel@tonic-gate 	 * does wait for cmd to terminate before returning though.
35270Sstevel@tonic-gate 	 * When the executed command writes its output to the pipe
35280Sstevel@tonic-gate 	 * there is no reader process and the command dies with
35290Sstevel@tonic-gate 	 * SIGPIPE. To avoid this we read repeatedly until read
35300Sstevel@tonic-gate 	 * terminates with EOF. This indicates that the command
35310Sstevel@tonic-gate 	 * (writer) has closed the pipe and we can safely do a
35320Sstevel@tonic-gate 	 * pclose().
35330Sstevel@tonic-gate 	 *
35340Sstevel@tonic-gate 	 * Since pclose() does wait for the command to exit,
35350Sstevel@tonic-gate 	 * we can safely reap the exit status of the command
35360Sstevel@tonic-gate 	 * from the value returned by pclose()
35370Sstevel@tonic-gate 	 */
35380Sstevel@tonic-gate 	while (fgets(buf, sizeof (buf), ptr) != NULL) {
35390Sstevel@tonic-gate 		/* if (bam_verbose)  XXX */
35400Sstevel@tonic-gate 			bam_print(PRINT_NO_NEWLINE, buf);
35410Sstevel@tonic-gate 		if (output && osize > 0) {
35420Sstevel@tonic-gate 			(void) snprintf(output, osize, "%s", buf);
35430Sstevel@tonic-gate 			len = strlen(buf);
35440Sstevel@tonic-gate 			output += len;
35450Sstevel@tonic-gate 			osize -= len;
35460Sstevel@tonic-gate 		}
35470Sstevel@tonic-gate 	}
35480Sstevel@tonic-gate 
35490Sstevel@tonic-gate 	ret = pclose(ptr);
35500Sstevel@tonic-gate 	if (ret == -1) {
35510Sstevel@tonic-gate 		bam_error(PCLOSE_FAIL, cmdline, strerror(errno));
35520Sstevel@tonic-gate 		return (-1);
35530Sstevel@tonic-gate 	}
35540Sstevel@tonic-gate 
35550Sstevel@tonic-gate 	if (WIFEXITED(ret)) {
35560Sstevel@tonic-gate 		return (WEXITSTATUS(ret));
35570Sstevel@tonic-gate 	} else {
35580Sstevel@tonic-gate 		bam_error(EXEC_FAIL, cmdline, ret);
35590Sstevel@tonic-gate 		return (-1);
35600Sstevel@tonic-gate 	}
35610Sstevel@tonic-gate }
35620Sstevel@tonic-gate 
35630Sstevel@tonic-gate /*
35640Sstevel@tonic-gate  * Since this function returns -1 on error
35650Sstevel@tonic-gate  * it cannot be used to convert -1. However,
35660Sstevel@tonic-gate  * that is sufficient for what we need.
35670Sstevel@tonic-gate  */
35680Sstevel@tonic-gate static long
35690Sstevel@tonic-gate s_strtol(char *str)
35700Sstevel@tonic-gate {
35710Sstevel@tonic-gate 	long l;
35720Sstevel@tonic-gate 	char *res = NULL;
35730Sstevel@tonic-gate 
35740Sstevel@tonic-gate 	if (str == NULL) {
35750Sstevel@tonic-gate 		return (-1);
35760Sstevel@tonic-gate 	}
35770Sstevel@tonic-gate 
35780Sstevel@tonic-gate 	errno = 0;
35790Sstevel@tonic-gate 	l = strtol(str, &res, 10);
35800Sstevel@tonic-gate 	if (errno || *res != '\0') {
35810Sstevel@tonic-gate 		return (-1);
35820Sstevel@tonic-gate 	}
35830Sstevel@tonic-gate 
35840Sstevel@tonic-gate 	return (l);
35850Sstevel@tonic-gate }
35860Sstevel@tonic-gate 
35870Sstevel@tonic-gate /*
35880Sstevel@tonic-gate  * Wrapper around fputs, that adds a newline (since fputs doesn't)
35890Sstevel@tonic-gate  */
35900Sstevel@tonic-gate static int
35910Sstevel@tonic-gate s_fputs(char *str, FILE *fp)
35920Sstevel@tonic-gate {
35930Sstevel@tonic-gate 	char linebuf[BAM_MAXLINE];
35940Sstevel@tonic-gate 
35950Sstevel@tonic-gate 	(void) snprintf(linebuf, sizeof (linebuf), "%s\n", str);
35960Sstevel@tonic-gate 	return (fputs(linebuf, fp));
35970Sstevel@tonic-gate }
35980Sstevel@tonic-gate 
35990Sstevel@tonic-gate /*
36000Sstevel@tonic-gate  * Wrapper around fgets, that strips newlines returned by fgets
36010Sstevel@tonic-gate  */
36020Sstevel@tonic-gate static char *
36030Sstevel@tonic-gate s_fgets(char *buf, int buflen, FILE *fp)
36040Sstevel@tonic-gate {
36050Sstevel@tonic-gate 	int n;
36060Sstevel@tonic-gate 
36070Sstevel@tonic-gate 	buf = fgets(buf, buflen, fp);
36080Sstevel@tonic-gate 	if (buf) {
36090Sstevel@tonic-gate 		n = strlen(buf);
36100Sstevel@tonic-gate 		if (n == buflen - 1 && buf[n-1] != '\n')
36110Sstevel@tonic-gate 			bam_error(TOO_LONG, buflen - 1, buf);
36120Sstevel@tonic-gate 		buf[n-1] = (buf[n-1] == '\n') ? '\0' : buf[n-1];
36130Sstevel@tonic-gate 	}
36140Sstevel@tonic-gate 
36150Sstevel@tonic-gate 	return (buf);
36160Sstevel@tonic-gate }
36170Sstevel@tonic-gate 
36180Sstevel@tonic-gate static void *
36190Sstevel@tonic-gate s_calloc(size_t nelem, size_t sz)
36200Sstevel@tonic-gate {
36210Sstevel@tonic-gate 	void *ptr;
36220Sstevel@tonic-gate 
36230Sstevel@tonic-gate 	ptr = calloc(nelem, sz);
36240Sstevel@tonic-gate 	if (ptr == NULL) {
36250Sstevel@tonic-gate 		bam_error(NO_MEM, nelem*sz);
36260Sstevel@tonic-gate 		bam_exit(1);
36270Sstevel@tonic-gate 	}
36280Sstevel@tonic-gate 	return (ptr);
36290Sstevel@tonic-gate }
36300Sstevel@tonic-gate 
36310Sstevel@tonic-gate static char *
36320Sstevel@tonic-gate s_strdup(char *str)
36330Sstevel@tonic-gate {
36340Sstevel@tonic-gate 	char *ptr;
36350Sstevel@tonic-gate 
36360Sstevel@tonic-gate 	if (str == NULL)
36370Sstevel@tonic-gate 		return (NULL);
36380Sstevel@tonic-gate 
36390Sstevel@tonic-gate 	ptr = strdup(str);
36400Sstevel@tonic-gate 	if (ptr == NULL) {
36410Sstevel@tonic-gate 		bam_error(NO_MEM, strlen(str) + 1);
36420Sstevel@tonic-gate 		bam_exit(1);
36430Sstevel@tonic-gate 	}
36440Sstevel@tonic-gate 	return (ptr);
36450Sstevel@tonic-gate }
36460Sstevel@tonic-gate 
36470Sstevel@tonic-gate /*
36480Sstevel@tonic-gate  * Returns 1 if amd64 (or sparc, for syncing x86 diskless clients)
36490Sstevel@tonic-gate  * Returns 0 otherwise
36500Sstevel@tonic-gate  */
36510Sstevel@tonic-gate static int
36520Sstevel@tonic-gate is_amd64(void)
36530Sstevel@tonic-gate {
36540Sstevel@tonic-gate 	static int amd64 = -1;
36550Sstevel@tonic-gate 	char isabuf[257];	/* from sysinfo(2) manpage */
36560Sstevel@tonic-gate 
36570Sstevel@tonic-gate 	if (amd64 != -1)
36580Sstevel@tonic-gate 		return (amd64);
36590Sstevel@tonic-gate 
36600Sstevel@tonic-gate 	if (sysinfo(SI_ISALIST, isabuf, sizeof (isabuf)) > 0 &&
36610Sstevel@tonic-gate 	    strncmp(isabuf, "amd64 ", strlen("amd64 ")) == 0)
36620Sstevel@tonic-gate 		amd64 = 1;
36630Sstevel@tonic-gate 	else if (strstr(isabuf, "i386") == NULL)
36640Sstevel@tonic-gate 		amd64 = 1;		/* diskless server */
36650Sstevel@tonic-gate 	else
36660Sstevel@tonic-gate 		amd64 = 0;
36670Sstevel@tonic-gate 
36680Sstevel@tonic-gate 	return (amd64);
36690Sstevel@tonic-gate }
36700Sstevel@tonic-gate 
36710Sstevel@tonic-gate static void
36720Sstevel@tonic-gate append_to_flist(filelist_t *flistp, char *s)
36730Sstevel@tonic-gate {
36740Sstevel@tonic-gate 	line_t *lp;
36750Sstevel@tonic-gate 
36760Sstevel@tonic-gate 	lp = s_calloc(1, sizeof (line_t));
36770Sstevel@tonic-gate 	lp->line = s_strdup(s);
36780Sstevel@tonic-gate 	if (flistp->head == NULL)
36790Sstevel@tonic-gate 		flistp->head = lp;
36800Sstevel@tonic-gate 	else
36810Sstevel@tonic-gate 		flistp->tail->next = lp;
36820Sstevel@tonic-gate 	flistp->tail = lp;
36830Sstevel@tonic-gate }
3684