xref: /titanic_50/usr/src/cmd/boot/bootadm/bootadm.c (revision 2449e17f82f6097fd2c665b64723e31ceecbeca6)
17c478bd9Sstevel@tonic-gate /*
27c478bd9Sstevel@tonic-gate  * CDDL HEADER START
37c478bd9Sstevel@tonic-gate  *
47c478bd9Sstevel@tonic-gate  * The contents of this file are subject to the terms of the
5fbac2b2bSvikram  * Common Development and Distribution License (the "License").
6fbac2b2bSvikram  * You may not use this file except in compliance with the License.
77c478bd9Sstevel@tonic-gate  *
87c478bd9Sstevel@tonic-gate  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
97c478bd9Sstevel@tonic-gate  * or http://www.opensolaris.org/os/licensing.
107c478bd9Sstevel@tonic-gate  * See the License for the specific language governing permissions
117c478bd9Sstevel@tonic-gate  * and limitations under the License.
127c478bd9Sstevel@tonic-gate  *
137c478bd9Sstevel@tonic-gate  * When distributing Covered Code, include this CDDL HEADER in each
147c478bd9Sstevel@tonic-gate  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
157c478bd9Sstevel@tonic-gate  * If applicable, add the following below this CDDL HEADER, with the
167c478bd9Sstevel@tonic-gate  * fields enclosed by brackets "[]" replaced with your own identifying
177c478bd9Sstevel@tonic-gate  * information: Portions Copyright [yyyy] [name of copyright owner]
187c478bd9Sstevel@tonic-gate  *
197c478bd9Sstevel@tonic-gate  * CDDL HEADER END
207c478bd9Sstevel@tonic-gate  */
217c478bd9Sstevel@tonic-gate /*
22bff269d0Svikram  * Copyright 2007 Sun Microsystems, Inc.  All rights reserved.
237c478bd9Sstevel@tonic-gate  * Use is subject to license terms.
247c478bd9Sstevel@tonic-gate  */
257c478bd9Sstevel@tonic-gate 
267c478bd9Sstevel@tonic-gate #pragma ident	"%Z%%M%	%I%	%E% SMI"
277c478bd9Sstevel@tonic-gate 
287c478bd9Sstevel@tonic-gate /*
297c478bd9Sstevel@tonic-gate  * bootadm(1M) is a new utility for managing bootability of
307c478bd9Sstevel@tonic-gate  * Solaris *Newboot* environments. It has two primary tasks:
317c478bd9Sstevel@tonic-gate  * 	- Allow end users to manage bootability of Newboot Solaris instances
327c478bd9Sstevel@tonic-gate  *	- Provide services to other subsystems in Solaris (primarily Install)
337c478bd9Sstevel@tonic-gate  */
347c478bd9Sstevel@tonic-gate 
357c478bd9Sstevel@tonic-gate /* Headers */
367c478bd9Sstevel@tonic-gate #include <stdio.h>
377c478bd9Sstevel@tonic-gate #include <errno.h>
387c478bd9Sstevel@tonic-gate #include <stdlib.h>
397c478bd9Sstevel@tonic-gate #include <string.h>
407c478bd9Sstevel@tonic-gate #include <unistd.h>
417c478bd9Sstevel@tonic-gate #include <sys/types.h>
427c478bd9Sstevel@tonic-gate #include <sys/stat.h>
437c478bd9Sstevel@tonic-gate #include <stdarg.h>
447c478bd9Sstevel@tonic-gate #include <limits.h>
457c478bd9Sstevel@tonic-gate #include <signal.h>
467c478bd9Sstevel@tonic-gate #include <sys/wait.h>
477c478bd9Sstevel@tonic-gate #include <sys/mnttab.h>
487c478bd9Sstevel@tonic-gate #include <sys/statvfs.h>
497c478bd9Sstevel@tonic-gate #include <libnvpair.h>
507c478bd9Sstevel@tonic-gate #include <ftw.h>
517c478bd9Sstevel@tonic-gate #include <fcntl.h>
527c478bd9Sstevel@tonic-gate #include <strings.h>
53*2449e17fSsherrym #include <utime.h>
547c478bd9Sstevel@tonic-gate #include <sys/systeminfo.h>
557c478bd9Sstevel@tonic-gate #include <sys/dktp/fdisk.h>
5658091fd8Ssetje #include <sys/param.h>
57*2449e17fSsherrym #if defined(__i386)
58*2449e17fSsherrym #include <sys/ucode.h>
59*2449e17fSsherrym #endif
607c478bd9Sstevel@tonic-gate 
617c478bd9Sstevel@tonic-gate #include <pwd.h>
627c478bd9Sstevel@tonic-gate #include <grp.h>
637c478bd9Sstevel@tonic-gate #include <device_info.h>
647c478bd9Sstevel@tonic-gate 
657c478bd9Sstevel@tonic-gate #include <locale.h>
667c478bd9Sstevel@tonic-gate 
677c478bd9Sstevel@tonic-gate #include <assert.h>
687c478bd9Sstevel@tonic-gate 
697c478bd9Sstevel@tonic-gate #include "message.h"
70ae115bc7Smrj #include "bootadm.h"
717c478bd9Sstevel@tonic-gate 
727c478bd9Sstevel@tonic-gate #ifndef TEXT_DOMAIN
737c478bd9Sstevel@tonic-gate #define	TEXT_DOMAIN	"SUNW_OST_OSCMD"
747c478bd9Sstevel@tonic-gate #endif	/* TEXT_DOMAIN */
757c478bd9Sstevel@tonic-gate 
767c478bd9Sstevel@tonic-gate /* Type definitions */
777c478bd9Sstevel@tonic-gate 
787c478bd9Sstevel@tonic-gate /* Primary subcmds */
797c478bd9Sstevel@tonic-gate typedef enum {
807c478bd9Sstevel@tonic-gate 	BAM_MENU = 3,
817c478bd9Sstevel@tonic-gate 	BAM_ARCHIVE
827c478bd9Sstevel@tonic-gate } subcmd_t;
837c478bd9Sstevel@tonic-gate 
847c478bd9Sstevel@tonic-gate typedef enum {
857c478bd9Sstevel@tonic-gate     OPT_ABSENT = 0,	/* No option */
867c478bd9Sstevel@tonic-gate     OPT_REQ,		/* option required */
877c478bd9Sstevel@tonic-gate     OPT_OPTIONAL	/* option may or may not be present */
887c478bd9Sstevel@tonic-gate } option_t;
897c478bd9Sstevel@tonic-gate 
907c478bd9Sstevel@tonic-gate typedef struct {
917c478bd9Sstevel@tonic-gate 	char	*subcmd;
927c478bd9Sstevel@tonic-gate 	option_t option;
937c478bd9Sstevel@tonic-gate 	error_t (*handler)();
941a97e40eSvikram 	int	unpriv;			/* is this an unprivileged command */
957c478bd9Sstevel@tonic-gate } subcmd_defn_t;
967c478bd9Sstevel@tonic-gate 
977c478bd9Sstevel@tonic-gate #define	LINE_INIT	0	/* lineNum initial value */
987c478bd9Sstevel@tonic-gate #define	ENTRY_INIT	-1	/* entryNum initial value */
997c478bd9Sstevel@tonic-gate #define	ALL_ENTRIES	-2	/* selects all boot entries */
1007c478bd9Sstevel@tonic-gate 
1017c478bd9Sstevel@tonic-gate #define	GRUB_DIR		"/boot/grub"
1027c478bd9Sstevel@tonic-gate #define	GRUB_MENU		"/boot/grub/menu.lst"
1037c478bd9Sstevel@tonic-gate #define	MENU_TMP		"/boot/grub/menu.lst.tmp"
1047c478bd9Sstevel@tonic-gate #define	RAMDISK_SPECIAL		"/ramdisk"
10540541d5dSvikram #define	STUBBOOT		"/stubboot"
1067c478bd9Sstevel@tonic-gate 
1077c478bd9Sstevel@tonic-gate /* lock related */
1087c478bd9Sstevel@tonic-gate #define	BAM_LOCK_FILE		"/var/run/bootadm.lock"
1097c478bd9Sstevel@tonic-gate #define	LOCK_FILE_PERMS		(S_IRUSR|S_IWUSR|S_IRGRP|S_IROTH)
1107c478bd9Sstevel@tonic-gate 
1117c478bd9Sstevel@tonic-gate #define	CREATE_RAMDISK		"/boot/solaris/bin/create_ramdisk"
1127c478bd9Sstevel@tonic-gate #define	CREATE_DISKMAP		"/boot/solaris/bin/create_diskmap"
1137c478bd9Sstevel@tonic-gate #define	GRUBDISK_MAP		"/var/run/solaris_grubdisk.map"
1147c478bd9Sstevel@tonic-gate 
115b610f78eSvikram #define	GRUB_slice		"/etc/lu/GRUB_slice"
116b610f78eSvikram #define	GRUB_root		"/etc/lu/GRUB_root"
117b610f78eSvikram #define	GRUB_backup_menu	"/etc/lu/GRUB_backup_menu"
118b610f78eSvikram #define	GRUB_slice_mntpt	"/tmp/GRUB_slice_mntpt"
119b610f78eSvikram #define	LU_ACTIVATE_FILE	"/etc/lu/DelayUpdate/activate.sh"
120fbac2b2bSvikram #define	GRUB_fdisk		"/etc/lu/GRUB_fdisk"
121fbac2b2bSvikram #define	GRUB_fdisk_target	"/etc/lu/GRUB_fdisk_target"
122b610f78eSvikram 
123b610f78eSvikram #define	INSTALLGRUB		"/sbin/installgrub"
124b610f78eSvikram #define	STAGE1			"/boot/grub/stage1"
125b610f78eSvikram #define	STAGE2			"/boot/grub/stage2"
126b610f78eSvikram 
1277c478bd9Sstevel@tonic-gate /*
12898892a30Snadkarni  * The following two defines are used to detect and create the correct
12998892a30Snadkarni  * boot archive  when safemode patching is underway.  LOFS_PATCH_FILE is a
13098892a30Snadkarni  * contracted private interface between bootadm and the install
13198892a30Snadkarni  * consolidation.  It is set by pdo.c when a patch with SUNW_PATCH_SAFEMODE
13298892a30Snadkarni  * is applied.
13398892a30Snadkarni  */
13498892a30Snadkarni 
13598892a30Snadkarni #define	LOFS_PATCH_FILE		"/var/run/.patch_loopback_mode"
13698892a30Snadkarni #define	LOFS_PATCH_MNT		"/var/run/.patch_root_loopbackmnt"
13798892a30Snadkarni 
13898892a30Snadkarni /*
1397c478bd9Sstevel@tonic-gate  * Default file attributes
1407c478bd9Sstevel@tonic-gate  */
1417c478bd9Sstevel@tonic-gate #define	DEFAULT_DEV_MODE	0644	/* default permissions */
1427c478bd9Sstevel@tonic-gate #define	DEFAULT_DEV_UID		0	/* user root */
1437c478bd9Sstevel@tonic-gate #define	DEFAULT_DEV_GID		3	/* group sys */
1447c478bd9Sstevel@tonic-gate 
1457c478bd9Sstevel@tonic-gate /*
1467c478bd9Sstevel@tonic-gate  * Menu related
1477c478bd9Sstevel@tonic-gate  * menu_cmd_t and menu_cmds must be kept in sync
1487c478bd9Sstevel@tonic-gate  */
149ae115bc7Smrj char *menu_cmds[] = {
1507c478bd9Sstevel@tonic-gate 	"default",	/* DEFAULT_CMD */
1517c478bd9Sstevel@tonic-gate 	"timeout",	/* TIMEOUT_CMD */
1527c478bd9Sstevel@tonic-gate 	"title",	/* TITLE_CMD */
1537c478bd9Sstevel@tonic-gate 	"root",		/* ROOT_CMD */
1547c478bd9Sstevel@tonic-gate 	"kernel",	/* KERNEL_CMD */
155ae115bc7Smrj 	"kernel$",	/* KERNEL_DOLLAR_CMD */
1567c478bd9Sstevel@tonic-gate 	"module",	/* MODULE_CMD */
157ae115bc7Smrj 	"module$",	/* MODULE_DOLLAR_CMD */
1587c478bd9Sstevel@tonic-gate 	" ",		/* SEP_CMD */
1597c478bd9Sstevel@tonic-gate 	"#",		/* COMMENT_CMD */
160ae115bc7Smrj 	"chainloader",	/* CHAINLOADER_CMD */
161ae115bc7Smrj 	"args",		/* ARGS_CMD */
1627c478bd9Sstevel@tonic-gate 	NULL
1637c478bd9Sstevel@tonic-gate };
1647c478bd9Sstevel@tonic-gate 
1657c478bd9Sstevel@tonic-gate #define	OPT_ENTRY_NUM	"entry"
1667c478bd9Sstevel@tonic-gate 
1677c478bd9Sstevel@tonic-gate /*
1687c478bd9Sstevel@tonic-gate  * archive related
1697c478bd9Sstevel@tonic-gate  */
1707c478bd9Sstevel@tonic-gate typedef struct {
1717c478bd9Sstevel@tonic-gate 	line_t *head;
1727c478bd9Sstevel@tonic-gate 	line_t *tail;
1737c478bd9Sstevel@tonic-gate } filelist_t;
1747c478bd9Sstevel@tonic-gate 
1757c478bd9Sstevel@tonic-gate #define	BOOT_FILE_LIST	"boot/solaris/filelist.ramdisk"
1767c478bd9Sstevel@tonic-gate #define	ETC_FILE_LIST	"etc/boot/solaris/filelist.ramdisk"
1777c478bd9Sstevel@tonic-gate 
1787c478bd9Sstevel@tonic-gate #define	FILE_STAT	"boot/solaris/filestat.ramdisk"
1797c478bd9Sstevel@tonic-gate #define	FILE_STAT_TMP	"boot/solaris/filestat.ramdisk.tmp"
1807c478bd9Sstevel@tonic-gate #define	DIR_PERMS	(S_IRWXU|S_IRGRP|S_IXGRP|S_IROTH|S_IXOTH)
1817c478bd9Sstevel@tonic-gate #define	FILE_STAT_MODE	(S_IRUSR|S_IWUSR|S_IRGRP|S_IROTH)
1827c478bd9Sstevel@tonic-gate 
1837c478bd9Sstevel@tonic-gate /* Globals */
184ae115bc7Smrj int bam_verbose;
185ae115bc7Smrj int bam_force;
1867c478bd9Sstevel@tonic-gate static char *prog;
1877c478bd9Sstevel@tonic-gate static subcmd_t bam_cmd;
1887c478bd9Sstevel@tonic-gate static char *bam_root;
1897c478bd9Sstevel@tonic-gate static int bam_rootlen;
1907c478bd9Sstevel@tonic-gate static int bam_root_readonly;
19140541d5dSvikram static int bam_alt_root;
1927c478bd9Sstevel@tonic-gate static char *bam_subcmd;
1937c478bd9Sstevel@tonic-gate static char *bam_opt;
1947c478bd9Sstevel@tonic-gate static int bam_debug;
1957c478bd9Sstevel@tonic-gate static char **bam_argv;
1967c478bd9Sstevel@tonic-gate static int bam_argc;
1977c478bd9Sstevel@tonic-gate static int bam_check;
1987c478bd9Sstevel@tonic-gate static int bam_smf_check;
1997c478bd9Sstevel@tonic-gate static int bam_lock_fd = -1;
2007c478bd9Sstevel@tonic-gate static char rootbuf[PATH_MAX] = "/";
201b610f78eSvikram static int bam_update_all;
2027c478bd9Sstevel@tonic-gate 
2037c478bd9Sstevel@tonic-gate /* function prototypes */
2047c478bd9Sstevel@tonic-gate static void parse_args_internal(int argc, char *argv[]);
2057c478bd9Sstevel@tonic-gate static void parse_args(int argc, char *argv[]);
2067c478bd9Sstevel@tonic-gate static error_t bam_menu(char *subcmd, char *opt, int argc, char *argv[]);
2077c478bd9Sstevel@tonic-gate static error_t bam_archive(char *subcmd, char *opt);
2087c478bd9Sstevel@tonic-gate 
2097c478bd9Sstevel@tonic-gate static void bam_print(char *format, ...);
2107c478bd9Sstevel@tonic-gate static void bam_exit(int excode);
2117c478bd9Sstevel@tonic-gate static void bam_lock(void);
2127c478bd9Sstevel@tonic-gate static void bam_unlock(void);
2137c478bd9Sstevel@tonic-gate 
2147c478bd9Sstevel@tonic-gate static int exec_cmd(char *cmdline, char *output, int64_t osize);
2157c478bd9Sstevel@tonic-gate static error_t read_globals(menu_t *mp, char *menu_path,
2167c478bd9Sstevel@tonic-gate     char *globalcmd, int quiet);
2177c478bd9Sstevel@tonic-gate 
2187c478bd9Sstevel@tonic-gate static menu_t *menu_read(char *menu_path);
2197c478bd9Sstevel@tonic-gate static error_t menu_write(char *root, menu_t *mp);
2207c478bd9Sstevel@tonic-gate static void linelist_free(line_t *start);
2217c478bd9Sstevel@tonic-gate static void menu_free(menu_t *mp);
2227c478bd9Sstevel@tonic-gate static void line_free(line_t *lp);
2237c478bd9Sstevel@tonic-gate static void filelist_free(filelist_t *flistp);
2247c478bd9Sstevel@tonic-gate static error_t list2file(char *root, char *tmp,
2257c478bd9Sstevel@tonic-gate     char *final, line_t *start);
2267c478bd9Sstevel@tonic-gate static error_t list_entry(menu_t *mp, char *menu_path, char *opt);
2277c478bd9Sstevel@tonic-gate static error_t delete_all_entries(menu_t *mp, char *menu_path, char *opt);
2287c478bd9Sstevel@tonic-gate static error_t update_entry(menu_t *mp, char *root, char *opt);
2297c478bd9Sstevel@tonic-gate static error_t update_temp(menu_t *mp, char *root, char *opt);
2307c478bd9Sstevel@tonic-gate 
2317c478bd9Sstevel@tonic-gate static error_t update_archive(char *root, char *opt);
2327c478bd9Sstevel@tonic-gate static error_t list_archive(char *root, char *opt);
2337c478bd9Sstevel@tonic-gate static error_t update_all(char *root, char *opt);
2347c478bd9Sstevel@tonic-gate static error_t read_list(char *root, filelist_t  *flistp);
2357c478bd9Sstevel@tonic-gate static error_t set_global(menu_t *mp, char *globalcmd, int val);
2367c478bd9Sstevel@tonic-gate static error_t set_option(menu_t *mp, char *globalcmd, char *opt);
237ae115bc7Smrj static error_t set_kernel(menu_t *mp, menu_cmd_t optnum, char *path,
238ae115bc7Smrj     char *buf, size_t bufsize);
239ae115bc7Smrj static char *expand_path(const char *partial_path);
2407c478bd9Sstevel@tonic-gate 
2417c478bd9Sstevel@tonic-gate static long s_strtol(char *str);
2427c478bd9Sstevel@tonic-gate static int s_fputs(char *str, FILE *fp);
2437c478bd9Sstevel@tonic-gate 
2447c478bd9Sstevel@tonic-gate static char *s_strdup(char *str);
2457c478bd9Sstevel@tonic-gate static int is_readonly(char *);
2467c478bd9Sstevel@tonic-gate static int is_amd64(void);
2477c478bd9Sstevel@tonic-gate static void append_to_flist(filelist_t *, char *);
2487c478bd9Sstevel@tonic-gate 
2497c478bd9Sstevel@tonic-gate #if defined(__sparc)
2507c478bd9Sstevel@tonic-gate static void sparc_abort(void);
2517c478bd9Sstevel@tonic-gate #endif
2527c478bd9Sstevel@tonic-gate 
253*2449e17fSsherrym #if defined(__i386)
254*2449e17fSsherrym static void ucode_install();
255*2449e17fSsherrym #endif
256*2449e17fSsherrym 
2577c478bd9Sstevel@tonic-gate /* Menu related sub commands */
2587c478bd9Sstevel@tonic-gate static subcmd_defn_t menu_subcmds[] = {
2591a97e40eSvikram 	"set_option",		OPT_OPTIONAL,	set_option, 0,	/* PUB */
2601a97e40eSvikram 	"list_entry",		OPT_OPTIONAL,	list_entry, 1,	/* PUB */
2611a97e40eSvikram 	"delete_all_entries",	OPT_ABSENT,	delete_all_entries, 0, /* PVT */
2621a97e40eSvikram 	"update_entry",		OPT_REQ,	update_entry, 0, /* menu */
2631a97e40eSvikram 	"update_temp",		OPT_OPTIONAL,	update_temp, 0,	/* reboot */
264ae115bc7Smrj 	"upgrade",		OPT_ABSENT,	upgrade_menu, 0, /* menu */
2651a97e40eSvikram 	NULL,			0,		NULL, 0	/* must be last */
2667c478bd9Sstevel@tonic-gate };
2677c478bd9Sstevel@tonic-gate 
2687c478bd9Sstevel@tonic-gate /* Archive related sub commands */
2697c478bd9Sstevel@tonic-gate static subcmd_defn_t arch_subcmds[] = {
2701a97e40eSvikram 	"update",		OPT_ABSENT,	update_archive, 0, /* PUB */
2711a97e40eSvikram 	"update_all",		OPT_ABSENT,	update_all, 0,	/* PVT */
2721a97e40eSvikram 	"list",			OPT_OPTIONAL,	list_archive, 1, /* PUB */
2731a97e40eSvikram 	NULL,			0,		NULL, 0	/* must be last */
2747c478bd9Sstevel@tonic-gate };
2757c478bd9Sstevel@tonic-gate 
2767c478bd9Sstevel@tonic-gate static struct {
2777c478bd9Sstevel@tonic-gate 	nvlist_t *new_nvlp;
2787c478bd9Sstevel@tonic-gate 	nvlist_t *old_nvlp;
2797c478bd9Sstevel@tonic-gate 	int need_update;
2807c478bd9Sstevel@tonic-gate } walk_arg;
2817c478bd9Sstevel@tonic-gate 
28258091fd8Ssetje 
2839632cfadSsetje struct safefile {
28458091fd8Ssetje 	char *name;
28558091fd8Ssetje 	struct safefile *next;
28658091fd8Ssetje };
28758091fd8Ssetje 
288ae115bc7Smrj static struct safefile *safefiles = NULL;
28958091fd8Ssetje #define	NEED_UPDATE_FILE "/etc/svc/volatile/boot_archive_needs_update"
29058091fd8Ssetje 
2917c478bd9Sstevel@tonic-gate static void
2927c478bd9Sstevel@tonic-gate usage(void)
2937c478bd9Sstevel@tonic-gate {
2947c478bd9Sstevel@tonic-gate 	(void) fprintf(stderr, "USAGE:\n");
2957c478bd9Sstevel@tonic-gate 
2967c478bd9Sstevel@tonic-gate 
2977c478bd9Sstevel@tonic-gate 	/* archive usage */
2987c478bd9Sstevel@tonic-gate 	(void) fprintf(stderr, "\t%s update-archive [-vn] [-R altroot]\n",
2997c478bd9Sstevel@tonic-gate 	    prog);
3007c478bd9Sstevel@tonic-gate 	(void) fprintf(stderr, "\t%s list-archive [-R altroot]\n", prog);
3017c478bd9Sstevel@tonic-gate #ifndef __sparc
3027c478bd9Sstevel@tonic-gate 	/* x86 only */
3037c478bd9Sstevel@tonic-gate 	(void) fprintf(stderr, "\t%s set-menu [-R altroot] key=value\n", prog);
3047c478bd9Sstevel@tonic-gate 	(void) fprintf(stderr, "\t%s list-menu [-R altroot]\n", prog);
3057c478bd9Sstevel@tonic-gate #endif
3067c478bd9Sstevel@tonic-gate }
3077c478bd9Sstevel@tonic-gate 
3087c478bd9Sstevel@tonic-gate int
3097c478bd9Sstevel@tonic-gate main(int argc, char *argv[])
3107c478bd9Sstevel@tonic-gate {
3117c478bd9Sstevel@tonic-gate 	error_t ret;
3127c478bd9Sstevel@tonic-gate 
3137c478bd9Sstevel@tonic-gate 	(void) setlocale(LC_ALL, "");
3147c478bd9Sstevel@tonic-gate 	(void) textdomain(TEXT_DOMAIN);
3157c478bd9Sstevel@tonic-gate 
3167c478bd9Sstevel@tonic-gate 	if ((prog = strrchr(argv[0], '/')) == NULL) {
3177c478bd9Sstevel@tonic-gate 		prog = argv[0];
3187c478bd9Sstevel@tonic-gate 	} else {
3197c478bd9Sstevel@tonic-gate 		prog++;
3207c478bd9Sstevel@tonic-gate 	}
3217c478bd9Sstevel@tonic-gate 
3227c478bd9Sstevel@tonic-gate 
3237c478bd9Sstevel@tonic-gate 	/*
3247c478bd9Sstevel@tonic-gate 	 * Don't depend on caller's umask
3257c478bd9Sstevel@tonic-gate 	 */
3267c478bd9Sstevel@tonic-gate 	(void) umask(0022);
3277c478bd9Sstevel@tonic-gate 
3287c478bd9Sstevel@tonic-gate 	parse_args(argc, argv);
3297c478bd9Sstevel@tonic-gate 
3307c478bd9Sstevel@tonic-gate #if defined(__sparc)
3317c478bd9Sstevel@tonic-gate 	/*
3327c478bd9Sstevel@tonic-gate 	 * There are only two valid invocations of bootadm
3337c478bd9Sstevel@tonic-gate 	 * on SPARC:
3347c478bd9Sstevel@tonic-gate 	 *
3357c478bd9Sstevel@tonic-gate 	 *	- SPARC diskless server creating boot_archive for i386 clients
3367c478bd9Sstevel@tonic-gate 	 *	- archive creation call during reboot of a SPARC system
3377c478bd9Sstevel@tonic-gate 	 *
3387c478bd9Sstevel@tonic-gate 	 *	The latter should be a NOP
3397c478bd9Sstevel@tonic-gate 	 */
3407c478bd9Sstevel@tonic-gate 	if (bam_cmd != BAM_ARCHIVE) {
3417c478bd9Sstevel@tonic-gate 		sparc_abort();
3427c478bd9Sstevel@tonic-gate 	}
3437c478bd9Sstevel@tonic-gate #endif
3447c478bd9Sstevel@tonic-gate 
3457c478bd9Sstevel@tonic-gate 	switch (bam_cmd) {
3467c478bd9Sstevel@tonic-gate 		case BAM_MENU:
3477c478bd9Sstevel@tonic-gate 			ret = bam_menu(bam_subcmd, bam_opt, bam_argc, bam_argv);
3487c478bd9Sstevel@tonic-gate 			break;
3497c478bd9Sstevel@tonic-gate 		case BAM_ARCHIVE:
3507c478bd9Sstevel@tonic-gate 			ret = bam_archive(bam_subcmd, bam_opt);
3517c478bd9Sstevel@tonic-gate 			break;
3527c478bd9Sstevel@tonic-gate 		default:
3537c478bd9Sstevel@tonic-gate 			usage();
3547c478bd9Sstevel@tonic-gate 			bam_exit(1);
3557c478bd9Sstevel@tonic-gate 	}
3567c478bd9Sstevel@tonic-gate 
3577c478bd9Sstevel@tonic-gate 	if (ret != BAM_SUCCESS)
3587c478bd9Sstevel@tonic-gate 		bam_exit(1);
3597c478bd9Sstevel@tonic-gate 
3607c478bd9Sstevel@tonic-gate 	bam_unlock();
3617c478bd9Sstevel@tonic-gate 	return (0);
3627c478bd9Sstevel@tonic-gate }
3637c478bd9Sstevel@tonic-gate 
3647c478bd9Sstevel@tonic-gate #if defined(__sparc)
3657c478bd9Sstevel@tonic-gate 
3667c478bd9Sstevel@tonic-gate static void
3677c478bd9Sstevel@tonic-gate sparc_abort(void)
3687c478bd9Sstevel@tonic-gate {
3697c478bd9Sstevel@tonic-gate 	bam_error(NOT_ON_SPARC);
3707c478bd9Sstevel@tonic-gate 	bam_exit(1);
3717c478bd9Sstevel@tonic-gate }
3727c478bd9Sstevel@tonic-gate 
3737c478bd9Sstevel@tonic-gate #endif
3747c478bd9Sstevel@tonic-gate 
3757c478bd9Sstevel@tonic-gate /*
3767c478bd9Sstevel@tonic-gate  * Equivalence of public and internal commands:
3777c478bd9Sstevel@tonic-gate  *	update-archive  -- -a update
3787c478bd9Sstevel@tonic-gate  *	list-archive	-- -a list
3797c478bd9Sstevel@tonic-gate  *	set-menu	-- -m set_option
3807c478bd9Sstevel@tonic-gate  *	list-menu	-- -m list_entry
3817c478bd9Sstevel@tonic-gate  *	update-menu	-- -m update_entry
3827c478bd9Sstevel@tonic-gate  */
3837c478bd9Sstevel@tonic-gate static struct cmd_map {
3847c478bd9Sstevel@tonic-gate 	char *bam_cmdname;
3857c478bd9Sstevel@tonic-gate 	int bam_cmd;
3867c478bd9Sstevel@tonic-gate 	char *bam_subcmd;
3877c478bd9Sstevel@tonic-gate } cmd_map[] = {
3887c478bd9Sstevel@tonic-gate 	{ "update-archive",	BAM_ARCHIVE,	"update"},
3897c478bd9Sstevel@tonic-gate 	{ "list-archive",	BAM_ARCHIVE,	"list"},
3907c478bd9Sstevel@tonic-gate 	{ "set-menu",		BAM_MENU,	"set_option"},
3917c478bd9Sstevel@tonic-gate 	{ "list-menu",		BAM_MENU,	"list_entry"},
3927c478bd9Sstevel@tonic-gate 	{ "update-menu",	BAM_MENU,	"update_entry"},
3937c478bd9Sstevel@tonic-gate 	{ NULL,			0,		NULL}
3947c478bd9Sstevel@tonic-gate };
3957c478bd9Sstevel@tonic-gate 
3967c478bd9Sstevel@tonic-gate /*
3977c478bd9Sstevel@tonic-gate  * Commands syntax published in bootadm(1M) are parsed here
3987c478bd9Sstevel@tonic-gate  */
3997c478bd9Sstevel@tonic-gate static void
4007c478bd9Sstevel@tonic-gate parse_args(int argc, char *argv[])
4017c478bd9Sstevel@tonic-gate {
4027c478bd9Sstevel@tonic-gate 	struct cmd_map *cmp = cmd_map;
4037c478bd9Sstevel@tonic-gate 
4047c478bd9Sstevel@tonic-gate 	/* command conforming to the final spec */
4057c478bd9Sstevel@tonic-gate 	if (argc > 1 && argv[1][0] != '-') {
4067c478bd9Sstevel@tonic-gate 		/*
4077c478bd9Sstevel@tonic-gate 		 * Map commands to internal table.
4087c478bd9Sstevel@tonic-gate 		 */
4097c478bd9Sstevel@tonic-gate 		while (cmp->bam_cmdname) {
4107c478bd9Sstevel@tonic-gate 			if (strcmp(argv[1], cmp->bam_cmdname) == 0) {
4117c478bd9Sstevel@tonic-gate 				bam_cmd = cmp->bam_cmd;
4127c478bd9Sstevel@tonic-gate 				bam_subcmd = cmp->bam_subcmd;
4137c478bd9Sstevel@tonic-gate 				break;
4147c478bd9Sstevel@tonic-gate 			}
4157c478bd9Sstevel@tonic-gate 			cmp++;
4167c478bd9Sstevel@tonic-gate 		}
4177c478bd9Sstevel@tonic-gate 		if (cmp->bam_cmdname == NULL) {
4187c478bd9Sstevel@tonic-gate 			usage();
4197c478bd9Sstevel@tonic-gate 			bam_exit(1);
4207c478bd9Sstevel@tonic-gate 		}
4217c478bd9Sstevel@tonic-gate 		argc--;
4227c478bd9Sstevel@tonic-gate 		argv++;
4237c478bd9Sstevel@tonic-gate 	}
4247c478bd9Sstevel@tonic-gate 
4257c478bd9Sstevel@tonic-gate 	parse_args_internal(argc, argv);
4267c478bd9Sstevel@tonic-gate }
4277c478bd9Sstevel@tonic-gate 
4287c478bd9Sstevel@tonic-gate /*
4297c478bd9Sstevel@tonic-gate  * A combination of public and private commands are parsed here.
4307c478bd9Sstevel@tonic-gate  * The internal syntax and the corresponding functionality are:
4317c478bd9Sstevel@tonic-gate  *	-a update	-- update-archive
4327c478bd9Sstevel@tonic-gate  *	-a list		-- list-archive
4337c478bd9Sstevel@tonic-gate  *	-a update-all	-- (reboot to sync all mounted OS archive)
4347c478bd9Sstevel@tonic-gate  *	-m update_entry	-- update-menu
4357c478bd9Sstevel@tonic-gate  *	-m list_entry	-- list-menu
4367c478bd9Sstevel@tonic-gate  *	-m update_temp	-- (reboot -- [boot-args])
4377c478bd9Sstevel@tonic-gate  *	-m delete_all_entries -- (called from install)
4387c478bd9Sstevel@tonic-gate  */
4397c478bd9Sstevel@tonic-gate static void
4407c478bd9Sstevel@tonic-gate parse_args_internal(int argc, char *argv[])
4417c478bd9Sstevel@tonic-gate {
4427c478bd9Sstevel@tonic-gate 	int c, error;
4437c478bd9Sstevel@tonic-gate 	extern char *optarg;
4447c478bd9Sstevel@tonic-gate 	extern int optind, opterr;
4457c478bd9Sstevel@tonic-gate 
4467c478bd9Sstevel@tonic-gate 	/* Suppress error message from getopt */
4477c478bd9Sstevel@tonic-gate 	opterr = 0;
4487c478bd9Sstevel@tonic-gate 
4497c478bd9Sstevel@tonic-gate 	error = 0;
4500d69385cSrscott 	while ((c = getopt(argc, argv, "a:d:fm:no:vCR:")) != -1) {
4517c478bd9Sstevel@tonic-gate 		switch (c) {
4527c478bd9Sstevel@tonic-gate 		case 'a':
4537c478bd9Sstevel@tonic-gate 			if (bam_cmd) {
4547c478bd9Sstevel@tonic-gate 				error = 1;
4557c478bd9Sstevel@tonic-gate 				bam_error(MULT_CMDS, c);
4567c478bd9Sstevel@tonic-gate 			}
4577c478bd9Sstevel@tonic-gate 			bam_cmd = BAM_ARCHIVE;
4587c478bd9Sstevel@tonic-gate 			bam_subcmd = optarg;
4597c478bd9Sstevel@tonic-gate 			break;
4607c478bd9Sstevel@tonic-gate 		case 'd':
4617c478bd9Sstevel@tonic-gate 			if (bam_debug) {
4627c478bd9Sstevel@tonic-gate 				error = 1;
4637c478bd9Sstevel@tonic-gate 				bam_error(DUP_OPT, c);
4647c478bd9Sstevel@tonic-gate 			}
4657c478bd9Sstevel@tonic-gate 			bam_debug = s_strtol(optarg);
4667c478bd9Sstevel@tonic-gate 			break;
4677c478bd9Sstevel@tonic-gate 		case 'f':
4687c478bd9Sstevel@tonic-gate 			if (bam_force) {
4697c478bd9Sstevel@tonic-gate 				error = 1;
4707c478bd9Sstevel@tonic-gate 				bam_error(DUP_OPT, c);
4717c478bd9Sstevel@tonic-gate 			}
4727c478bd9Sstevel@tonic-gate 			bam_force = 1;
4737c478bd9Sstevel@tonic-gate 			break;
4747c478bd9Sstevel@tonic-gate 		case 'm':
4757c478bd9Sstevel@tonic-gate 			if (bam_cmd) {
4767c478bd9Sstevel@tonic-gate 				error = 1;
4777c478bd9Sstevel@tonic-gate 				bam_error(MULT_CMDS, c);
4787c478bd9Sstevel@tonic-gate 			}
4797c478bd9Sstevel@tonic-gate 			bam_cmd = BAM_MENU;
4807c478bd9Sstevel@tonic-gate 			bam_subcmd = optarg;
4817c478bd9Sstevel@tonic-gate 			break;
4827c478bd9Sstevel@tonic-gate 		case 'n':
4837c478bd9Sstevel@tonic-gate 			if (bam_check) {
4847c478bd9Sstevel@tonic-gate 				error = 1;
4857c478bd9Sstevel@tonic-gate 				bam_error(DUP_OPT, c);
4867c478bd9Sstevel@tonic-gate 			}
4877c478bd9Sstevel@tonic-gate 			bam_check = 1;
4887c478bd9Sstevel@tonic-gate 			break;
4897c478bd9Sstevel@tonic-gate 		case 'o':
4907c478bd9Sstevel@tonic-gate 			if (bam_opt) {
4917c478bd9Sstevel@tonic-gate 				error = 1;
4927c478bd9Sstevel@tonic-gate 				bam_error(DUP_OPT, c);
4937c478bd9Sstevel@tonic-gate 			}
4947c478bd9Sstevel@tonic-gate 			bam_opt = optarg;
4957c478bd9Sstevel@tonic-gate 			break;
4967c478bd9Sstevel@tonic-gate 		case 'v':
4977c478bd9Sstevel@tonic-gate 			if (bam_verbose) {
4987c478bd9Sstevel@tonic-gate 				error = 1;
4997c478bd9Sstevel@tonic-gate 				bam_error(DUP_OPT, c);
5007c478bd9Sstevel@tonic-gate 			}
5017c478bd9Sstevel@tonic-gate 			bam_verbose = 1;
5027c478bd9Sstevel@tonic-gate 			break;
5038c1b6884Sszhou 		case 'C':
5048c1b6884Sszhou 			bam_smf_check = 1;
5058c1b6884Sszhou 			break;
5067c478bd9Sstevel@tonic-gate 		case 'R':
5077c478bd9Sstevel@tonic-gate 			if (bam_root) {
5087c478bd9Sstevel@tonic-gate 				error = 1;
5097c478bd9Sstevel@tonic-gate 				bam_error(DUP_OPT, c);
5107c478bd9Sstevel@tonic-gate 				break;
5117c478bd9Sstevel@tonic-gate 			} else if (realpath(optarg, rootbuf) == NULL) {
5127c478bd9Sstevel@tonic-gate 				error = 1;
5137c478bd9Sstevel@tonic-gate 				bam_error(CANT_RESOLVE, optarg,
5147c478bd9Sstevel@tonic-gate 				    strerror(errno));
5157c478bd9Sstevel@tonic-gate 				break;
5167c478bd9Sstevel@tonic-gate 			}
51740541d5dSvikram 			bam_alt_root = 1;
5187c478bd9Sstevel@tonic-gate 			bam_root = rootbuf;
5197c478bd9Sstevel@tonic-gate 			bam_rootlen = strlen(rootbuf);
5207c478bd9Sstevel@tonic-gate 			break;
5217c478bd9Sstevel@tonic-gate 		case '?':
5227c478bd9Sstevel@tonic-gate 			error = 1;
5237c478bd9Sstevel@tonic-gate 			bam_error(BAD_OPT, optopt);
5247c478bd9Sstevel@tonic-gate 			break;
5257c478bd9Sstevel@tonic-gate 		default :
5267c478bd9Sstevel@tonic-gate 			error = 1;
5277c478bd9Sstevel@tonic-gate 			bam_error(BAD_OPT, c);
5287c478bd9Sstevel@tonic-gate 			break;
5297c478bd9Sstevel@tonic-gate 		}
5307c478bd9Sstevel@tonic-gate 	}
5317c478bd9Sstevel@tonic-gate 
5327c478bd9Sstevel@tonic-gate 	/*
5337c478bd9Sstevel@tonic-gate 	 * A command option must be specfied
5347c478bd9Sstevel@tonic-gate 	 */
5357c478bd9Sstevel@tonic-gate 	if (!bam_cmd) {
5367c478bd9Sstevel@tonic-gate 		if (bam_opt && strcmp(bam_opt, "all") == 0) {
5377c478bd9Sstevel@tonic-gate 			usage();
5387c478bd9Sstevel@tonic-gate 			bam_exit(0);
5397c478bd9Sstevel@tonic-gate 		}
5407c478bd9Sstevel@tonic-gate 		bam_error(NEED_CMD);
5417c478bd9Sstevel@tonic-gate 		error = 1;
5427c478bd9Sstevel@tonic-gate 	}
5437c478bd9Sstevel@tonic-gate 
5447c478bd9Sstevel@tonic-gate 	if (error) {
5457c478bd9Sstevel@tonic-gate 		usage();
5467c478bd9Sstevel@tonic-gate 		bam_exit(1);
5477c478bd9Sstevel@tonic-gate 	}
5487c478bd9Sstevel@tonic-gate 
5497c478bd9Sstevel@tonic-gate 	if (optind > argc) {
5507c478bd9Sstevel@tonic-gate 		bam_error(INT_ERROR, "parse_args");
5517c478bd9Sstevel@tonic-gate 		bam_exit(1);
5527c478bd9Sstevel@tonic-gate 	} else if (optind < argc) {
5537c478bd9Sstevel@tonic-gate 		bam_argv = &argv[optind];
5547c478bd9Sstevel@tonic-gate 		bam_argc = argc - optind;
5557c478bd9Sstevel@tonic-gate 	}
5567c478bd9Sstevel@tonic-gate 
5577c478bd9Sstevel@tonic-gate 	/*
5587c478bd9Sstevel@tonic-gate 	 * -n implies verbose mode
5597c478bd9Sstevel@tonic-gate 	 */
5607c478bd9Sstevel@tonic-gate 	if (bam_check)
5617c478bd9Sstevel@tonic-gate 		bam_verbose = 1;
5627c478bd9Sstevel@tonic-gate }
5637c478bd9Sstevel@tonic-gate 
5647c478bd9Sstevel@tonic-gate static error_t
5657c478bd9Sstevel@tonic-gate check_subcmd_and_options(
5667c478bd9Sstevel@tonic-gate 	char *subcmd,
5677c478bd9Sstevel@tonic-gate 	char *opt,
5687c478bd9Sstevel@tonic-gate 	subcmd_defn_t *table,
5697c478bd9Sstevel@tonic-gate 	error_t (**fp)())
5707c478bd9Sstevel@tonic-gate {
5717c478bd9Sstevel@tonic-gate 	int i;
5727c478bd9Sstevel@tonic-gate 
5737c478bd9Sstevel@tonic-gate 	if (subcmd == NULL) {
5747c478bd9Sstevel@tonic-gate 		bam_error(NEED_SUBCMD);
5757c478bd9Sstevel@tonic-gate 		return (BAM_ERROR);
5767c478bd9Sstevel@tonic-gate 	}
5777c478bd9Sstevel@tonic-gate 
5781a97e40eSvikram 	if (bam_argc != 0 || bam_argv) {
5791a97e40eSvikram 		if (strcmp(subcmd, "set_option") != 0 || bam_argc != 1) {
5801a97e40eSvikram 			bam_error(TRAILING_ARGS);
5811a97e40eSvikram 			usage();
5821a97e40eSvikram 			return (BAM_ERROR);
5831a97e40eSvikram 		}
5841a97e40eSvikram 	}
5851a97e40eSvikram 
5867c478bd9Sstevel@tonic-gate 	if (bam_root == NULL) {
5877c478bd9Sstevel@tonic-gate 		bam_root = rootbuf;
5887c478bd9Sstevel@tonic-gate 		bam_rootlen = 1;
5897c478bd9Sstevel@tonic-gate 	}
5907c478bd9Sstevel@tonic-gate 
5917c478bd9Sstevel@tonic-gate 	/* verify that subcmd is valid */
5927c478bd9Sstevel@tonic-gate 	for (i = 0; table[i].subcmd != NULL; i++) {
5937c478bd9Sstevel@tonic-gate 		if (strcmp(table[i].subcmd, subcmd) == 0)
5947c478bd9Sstevel@tonic-gate 			break;
5957c478bd9Sstevel@tonic-gate 	}
5967c478bd9Sstevel@tonic-gate 
5977c478bd9Sstevel@tonic-gate 	if (table[i].subcmd == NULL) {
5987c478bd9Sstevel@tonic-gate 		bam_error(INVALID_SUBCMD, subcmd);
5997c478bd9Sstevel@tonic-gate 		return (BAM_ERROR);
6007c478bd9Sstevel@tonic-gate 	}
6017c478bd9Sstevel@tonic-gate 
6021a97e40eSvikram 	if (table[i].unpriv == 0 && geteuid() != 0) {
6031a97e40eSvikram 		bam_error(MUST_BE_ROOT);
6041a97e40eSvikram 		return (BAM_ERROR);
6051a97e40eSvikram 	}
6061a97e40eSvikram 
6071a97e40eSvikram 	/*
6081a97e40eSvikram 	 * Currently only privileged commands need a lock
6091a97e40eSvikram 	 */
6101a97e40eSvikram 	if (table[i].unpriv == 0)
6111a97e40eSvikram 		bam_lock();
6121a97e40eSvikram 
6137c478bd9Sstevel@tonic-gate 	/* subcmd verifies that opt is appropriate */
6147c478bd9Sstevel@tonic-gate 	if (table[i].option != OPT_OPTIONAL) {
6157c478bd9Sstevel@tonic-gate 		if ((table[i].option == OPT_REQ) ^ (opt != NULL)) {
6167c478bd9Sstevel@tonic-gate 			if (opt)
6177c478bd9Sstevel@tonic-gate 				bam_error(NO_OPT_REQ, subcmd);
6187c478bd9Sstevel@tonic-gate 			else
6197c478bd9Sstevel@tonic-gate 				bam_error(MISS_OPT, subcmd);
6207c478bd9Sstevel@tonic-gate 			return (BAM_ERROR);
6217c478bd9Sstevel@tonic-gate 		}
6227c478bd9Sstevel@tonic-gate 	}
6237c478bd9Sstevel@tonic-gate 
6247c478bd9Sstevel@tonic-gate 	*fp = table[i].handler;
6257c478bd9Sstevel@tonic-gate 
6267c478bd9Sstevel@tonic-gate 	return (BAM_SUCCESS);
6277c478bd9Sstevel@tonic-gate }
6287c478bd9Sstevel@tonic-gate 
629b610f78eSvikram 
630b610f78eSvikram static char *
63140541d5dSvikram mount_grub_slice(int *mnted, char **physlice, char **logslice, char **fs_type)
632b610f78eSvikram {
633b610f78eSvikram 	struct extmnttab mnt;
634b610f78eSvikram 	struct stat sb;
635b610f78eSvikram 	char buf[BAM_MAXLINE], dev[PATH_MAX], phys[PATH_MAX], fstype[32];
636f0f2c544Svikram 	char cmd[PATH_MAX];
637b610f78eSvikram 	char *mntpt;
638b610f78eSvikram 	int p, l, f;
639b610f78eSvikram 	FILE *fp;
640b610f78eSvikram 
641b610f78eSvikram 	assert(mnted);
642b610f78eSvikram 	*mnted = 0;
643b610f78eSvikram 
644b610f78eSvikram 	/*
64540541d5dSvikram 	 * physlice, logslice, fs_type  args may be NULL
646b610f78eSvikram 	 */
647b610f78eSvikram 	if (physlice)
648b610f78eSvikram 		*physlice = NULL;
64940541d5dSvikram 	if (logslice)
65040541d5dSvikram 		*logslice = NULL;
65140541d5dSvikram 	if (fs_type)
65240541d5dSvikram 		*fs_type = NULL;
653b610f78eSvikram 
654b610f78eSvikram 	if (stat(GRUB_slice, &sb) != 0) {
655b610f78eSvikram 		bam_error(MISSING_SLICE_FILE, GRUB_slice, strerror(errno));
656b610f78eSvikram 		return (NULL);
657b610f78eSvikram 	}
658b610f78eSvikram 
659b610f78eSvikram 	fp = fopen(GRUB_slice, "r");
660b610f78eSvikram 	if (fp == NULL) {
661b610f78eSvikram 		bam_error(OPEN_FAIL, GRUB_slice, strerror(errno));
662b610f78eSvikram 		return (NULL);
663b610f78eSvikram 	}
664b610f78eSvikram 
665b610f78eSvikram 	dev[0] = fstype[0] = phys[0] = '\0';
666b610f78eSvikram 	p = sizeof ("PHYS_SLICE=") - 1;
667b610f78eSvikram 	l = sizeof ("LOG_SLICE=") - 1;
668b610f78eSvikram 	f = sizeof ("LOG_FSTYP=") - 1;
669b610f78eSvikram 	while (s_fgets(buf, sizeof (buf), fp) != NULL) {
670b610f78eSvikram 		if (strncmp(buf, "PHYS_SLICE=", p) == 0) {
671b610f78eSvikram 			(void) strlcpy(phys, buf + p, sizeof (phys));
672b610f78eSvikram 			continue;
673b610f78eSvikram 		}
674b610f78eSvikram 		if (strncmp(buf, "LOG_SLICE=", l) == 0) {
675b610f78eSvikram 			(void) strlcpy(dev, buf + l, sizeof (dev));
676b610f78eSvikram 			continue;
677b610f78eSvikram 		}
678b610f78eSvikram 		if (strncmp(buf, "LOG_FSTYP=", f) == 0) {
679b610f78eSvikram 			(void) strlcpy(fstype, buf + f, sizeof (fstype));
680b610f78eSvikram 			continue;
681b610f78eSvikram 		}
682b610f78eSvikram 	}
683b610f78eSvikram 	(void) fclose(fp);
684b610f78eSvikram 
685b610f78eSvikram 	if (dev[0] == '\0' || fstype[0] == '\0' || phys[0] == '\0') {
686b610f78eSvikram 		bam_error(BAD_SLICE_FILE, GRUB_slice);
687b610f78eSvikram 		return (NULL);
688b610f78eSvikram 	}
689b610f78eSvikram 
690b610f78eSvikram 	if (physlice) {
691b610f78eSvikram 		*physlice = s_strdup(phys);
692b610f78eSvikram 	}
69340541d5dSvikram 	if (logslice) {
69440541d5dSvikram 		*logslice = s_strdup(dev);
69540541d5dSvikram 	}
69640541d5dSvikram 	if (fs_type) {
69740541d5dSvikram 		*fs_type = s_strdup(fstype);
69840541d5dSvikram 	}
699b610f78eSvikram 
700b610f78eSvikram 	/*
701b610f78eSvikram 	 * Check if the slice is already mounted
702b610f78eSvikram 	 */
703b610f78eSvikram 	fp = fopen(MNTTAB, "r");
704b610f78eSvikram 	if (fp == NULL) {
705b610f78eSvikram 		bam_error(OPEN_FAIL, MNTTAB, strerror(errno));
70640541d5dSvikram 		goto error;
707b610f78eSvikram 	}
708b610f78eSvikram 
709b610f78eSvikram 	resetmnttab(fp);
710b610f78eSvikram 
711b610f78eSvikram 	mntpt = NULL;
712b610f78eSvikram 	while (getextmntent(fp, &mnt, sizeof (mnt)) == 0) {
713b610f78eSvikram 		if (strcmp(mnt.mnt_special, dev) == 0) {
714b610f78eSvikram 			mntpt = s_strdup(mnt.mnt_mountp);
715b610f78eSvikram 			break;
716b610f78eSvikram 		}
717b610f78eSvikram 	}
718b610f78eSvikram 
719b610f78eSvikram 	(void) fclose(fp);
720b610f78eSvikram 
721b610f78eSvikram 	if (mntpt) {
722b610f78eSvikram 		return (mntpt);
723b610f78eSvikram 	}
724b610f78eSvikram 
725b610f78eSvikram 
726b610f78eSvikram 	/*
727b610f78eSvikram 	 * GRUB slice is not mounted, we need to mount it now.
728b610f78eSvikram 	 * First create the mountpoint
729b610f78eSvikram 	 */
730b610f78eSvikram 	mntpt = s_calloc(1, PATH_MAX);
731b610f78eSvikram 	(void) snprintf(mntpt, PATH_MAX, "%s.%d", GRUB_slice_mntpt, getpid());
732b610f78eSvikram 	if (mkdir(mntpt, 0755) == -1 && errno != EEXIST) {
733b610f78eSvikram 		bam_error(MKDIR_FAILED, mntpt, strerror(errno));
734b610f78eSvikram 		free(mntpt);
73540541d5dSvikram 		goto error;
736b610f78eSvikram 	}
737b610f78eSvikram 
738f0f2c544Svikram 	(void) snprintf(cmd, sizeof (cmd), "/sbin/mount -F %s %s %s",
739f0f2c544Svikram 	    fstype, dev, mntpt);
740f0f2c544Svikram 
741f0f2c544Svikram 	if (exec_cmd(cmd, NULL, 0) != 0) {
74240541d5dSvikram 		bam_error(MOUNT_FAILED, dev, fstype);
743b610f78eSvikram 		if (rmdir(mntpt) != 0) {
744b610f78eSvikram 			bam_error(RMDIR_FAILED, mntpt, strerror(errno));
745b610f78eSvikram 		}
746b610f78eSvikram 		free(mntpt);
74740541d5dSvikram 		goto error;
748b610f78eSvikram 	}
749b610f78eSvikram 
750b610f78eSvikram 	*mnted = 1;
751b610f78eSvikram 	return (mntpt);
75240541d5dSvikram 
75340541d5dSvikram error:
75440541d5dSvikram 	if (physlice) {
75540541d5dSvikram 		free(*physlice);
75640541d5dSvikram 		*physlice = NULL;
75740541d5dSvikram 	}
75840541d5dSvikram 	if (logslice) {
75940541d5dSvikram 		free(*logslice);
76040541d5dSvikram 		*logslice = NULL;
76140541d5dSvikram 	}
76240541d5dSvikram 	if (fs_type) {
76340541d5dSvikram 		free(*fs_type);
76440541d5dSvikram 		*fs_type = NULL;
76540541d5dSvikram 	}
76640541d5dSvikram 	return (NULL);
767b610f78eSvikram }
768b610f78eSvikram 
769b610f78eSvikram static void
77040541d5dSvikram umount_grub_slice(
77140541d5dSvikram 	int mnted,
77240541d5dSvikram 	char *mntpt,
77340541d5dSvikram 	char *physlice,
77440541d5dSvikram 	char *logslice,
77540541d5dSvikram 	char *fs_type)
776b610f78eSvikram {
777f0f2c544Svikram 	char cmd[PATH_MAX];
778f0f2c544Svikram 
779b610f78eSvikram 	/*
780b610f78eSvikram 	 * If we have not dealt with GRUB slice
781b610f78eSvikram 	 * we have nothing to do - just return.
782b610f78eSvikram 	 */
783b610f78eSvikram 	if (mntpt == NULL)
784b610f78eSvikram 		return;
785b610f78eSvikram 
786b610f78eSvikram 
787b610f78eSvikram 	/*
788b610f78eSvikram 	 * If we mounted the filesystem earlier in mount_grub_slice()
789b610f78eSvikram 	 * unmount it now.
790b610f78eSvikram 	 */
791b610f78eSvikram 	if (mnted) {
792f0f2c544Svikram 		(void) snprintf(cmd, sizeof (cmd), "/sbin/umount %s",
793f0f2c544Svikram 		    mntpt);
794f0f2c544Svikram 		if (exec_cmd(cmd, NULL, 0) != 0) {
795f0f2c544Svikram 			bam_error(UMOUNT_FAILED, mntpt);
796b610f78eSvikram 		}
797b610f78eSvikram 		if (rmdir(mntpt) != 0) {
798b610f78eSvikram 			bam_error(RMDIR_FAILED, mntpt, strerror(errno));
799b610f78eSvikram 		}
800b610f78eSvikram 	}
80140541d5dSvikram 
802b610f78eSvikram 	if (physlice)
803b610f78eSvikram 		free(physlice);
80440541d5dSvikram 	if (logslice)
80540541d5dSvikram 		free(logslice);
80640541d5dSvikram 	if (fs_type)
80740541d5dSvikram 		free(fs_type);
80840541d5dSvikram 
809b610f78eSvikram 	free(mntpt);
810b610f78eSvikram }
811b610f78eSvikram 
81240541d5dSvikram static char *
81340541d5dSvikram use_stubboot(void)
81440541d5dSvikram {
81540541d5dSvikram 	int mnted;
81640541d5dSvikram 	struct stat sb;
81740541d5dSvikram 	struct extmnttab mnt;
81840541d5dSvikram 	FILE *fp;
81940541d5dSvikram 	char cmd[PATH_MAX];
82040541d5dSvikram 
82140541d5dSvikram 	if (stat(STUBBOOT, &sb) != 0) {
82240541d5dSvikram 		bam_error(STUBBOOT_DIR_NOT_FOUND);
82340541d5dSvikram 		return (NULL);
82440541d5dSvikram 	}
82540541d5dSvikram 
82640541d5dSvikram 	/*
82740541d5dSvikram 	 * Check if stubboot is mounted. If not, mount it
82840541d5dSvikram 	 */
82940541d5dSvikram 	fp = fopen(MNTTAB, "r");
83040541d5dSvikram 	if (fp == NULL) {
83140541d5dSvikram 		bam_error(OPEN_FAIL, MNTTAB, strerror(errno));
83240541d5dSvikram 		return (NULL);
83340541d5dSvikram 	}
83440541d5dSvikram 
83540541d5dSvikram 	resetmnttab(fp);
83640541d5dSvikram 
83740541d5dSvikram 	mnted = 0;
83840541d5dSvikram 	while (getextmntent(fp, &mnt, sizeof (mnt)) == 0) {
83940541d5dSvikram 		if (strcmp(mnt.mnt_mountp, STUBBOOT) == 0) {
84040541d5dSvikram 			mnted = 1;
84140541d5dSvikram 			break;
84240541d5dSvikram 		}
84340541d5dSvikram 	}
84440541d5dSvikram 
84540541d5dSvikram 	(void) fclose(fp);
84640541d5dSvikram 
84740541d5dSvikram 	if (mnted)
84840541d5dSvikram 		return (STUBBOOT);
84940541d5dSvikram 
85040541d5dSvikram 	/*
85140541d5dSvikram 	 * Stubboot is not mounted, mount it now.
85240541d5dSvikram 	 * It should exist in /etc/vfstab
85340541d5dSvikram 	 */
85440541d5dSvikram 	(void) snprintf(cmd, sizeof (cmd), "/sbin/mount %s",
85540541d5dSvikram 	    STUBBOOT);
85640541d5dSvikram 	if (exec_cmd(cmd, NULL, 0) != 0) {
85740541d5dSvikram 		bam_error(MOUNT_MNTPT_FAILED, STUBBOOT);
85840541d5dSvikram 		return (NULL);
85940541d5dSvikram 	}
86040541d5dSvikram 
86140541d5dSvikram 	return (STUBBOOT);
86240541d5dSvikram }
86340541d5dSvikram 
86440541d5dSvikram static void
86540541d5dSvikram disp_active_menu_locn(char *menu_path, char *logslice, char *fstype, int mnted)
86640541d5dSvikram {
86740541d5dSvikram 	/*
86840541d5dSvikram 	 * Check if we did a temp mount of an unmounted device.
86940541d5dSvikram 	 * If yes, print the block device and fstype for that device
87040541d5dSvikram 	 * else it is already mounted, so we print the path to the GRUB menu.
87140541d5dSvikram 	 */
87240541d5dSvikram 	if (mnted) {
87340541d5dSvikram 		bam_print(GRUB_MENU_DEVICE, logslice);
87440541d5dSvikram 		bam_print(GRUB_MENU_FSTYPE, fstype);
87540541d5dSvikram 	} else {
87640541d5dSvikram 		bam_print(GRUB_MENU_PATH, menu_path);
87740541d5dSvikram 	}
87840541d5dSvikram }
87940541d5dSvikram 
88040541d5dSvikram /*
88140541d5dSvikram  * NOTE: A single "/" is also considered a trailing slash and will
88240541d5dSvikram  * be deleted.
88340541d5dSvikram  */
88440541d5dSvikram static void
88540541d5dSvikram elide_trailing_slash(const char *src, char *dst, size_t dstsize)
88640541d5dSvikram {
88740541d5dSvikram 	size_t dstlen;
88840541d5dSvikram 
88940541d5dSvikram 	assert(src);
89040541d5dSvikram 	assert(dst);
89140541d5dSvikram 
89240541d5dSvikram 	(void) strlcpy(dst, src, dstsize);
89340541d5dSvikram 
89440541d5dSvikram 	dstlen = strlen(dst);
89540541d5dSvikram 	if (dst[dstlen - 1] == '/') {
89640541d5dSvikram 		dst[dstlen - 1] = '\0';
89740541d5dSvikram 	}
89840541d5dSvikram }
89940541d5dSvikram 
9007c478bd9Sstevel@tonic-gate static error_t
9017c478bd9Sstevel@tonic-gate bam_menu(char *subcmd, char *opt, int largc, char *largv[])
9027c478bd9Sstevel@tonic-gate {
9037c478bd9Sstevel@tonic-gate 	error_t ret;
9047c478bd9Sstevel@tonic-gate 	char menu_path[PATH_MAX];
905ae115bc7Smrj 	char path[PATH_MAX];
9067c478bd9Sstevel@tonic-gate 	menu_t *menu;
90740541d5dSvikram 	char *mntpt, *menu_root, *logslice, *fstype;
908b610f78eSvikram 	struct stat sb;
909b610f78eSvikram 	int mnted;	/* set if we did a mount */
9107c478bd9Sstevel@tonic-gate 	error_t (*f)(menu_t *mp, char *menu_path, char *opt);
9117c478bd9Sstevel@tonic-gate 
9127c478bd9Sstevel@tonic-gate 	/*
9137c478bd9Sstevel@tonic-gate 	 * Check arguments
9147c478bd9Sstevel@tonic-gate 	 */
9157c478bd9Sstevel@tonic-gate 	ret = check_subcmd_and_options(subcmd, opt, menu_subcmds, &f);
9167c478bd9Sstevel@tonic-gate 	if (ret == BAM_ERROR) {
9177c478bd9Sstevel@tonic-gate 		return (BAM_ERROR);
9187c478bd9Sstevel@tonic-gate 	}
9197c478bd9Sstevel@tonic-gate 
920b610f78eSvikram 	mntpt = NULL;
921b610f78eSvikram 	mnted = 0;
92240541d5dSvikram 	logslice = fstype = NULL;
92340541d5dSvikram 
92440541d5dSvikram 	/*
925ae115bc7Smrj 	 * Check for the menu.list file:
926ae115bc7Smrj 	 *
927ae115bc7Smrj 	 * 1. Check for a GRUB_slice file, be it on / or
928ae115bc7Smrj 	 *    on the user-provided alternate root.
929ae115bc7Smrj 	 * 2. Use the alternate root, if given.
930ae115bc7Smrj 	 * 3. Check /stubboot
931ae115bc7Smrj 	 * 4. Use /
93240541d5dSvikram 	 */
93340541d5dSvikram 	if (bam_alt_root) {
934ae115bc7Smrj 		(void) snprintf(path, sizeof (path), "%s%s", bam_root,
935ae115bc7Smrj 		    GRUB_slice);
936ae115bc7Smrj 	} else {
937ae115bc7Smrj 		(void) snprintf(path, sizeof (path), "%s", GRUB_slice);
938ae115bc7Smrj 	}
939ae115bc7Smrj 
940ae115bc7Smrj 	if (stat(path, &sb) == 0) {
94140541d5dSvikram 		mntpt = mount_grub_slice(&mnted, NULL, &logslice, &fstype);
942b610f78eSvikram 		menu_root = mntpt;
943ae115bc7Smrj 	} else if (bam_alt_root) {
944ae115bc7Smrj 		menu_root = bam_root;
94540541d5dSvikram 	} else if (stat(STUBBOOT, &sb) == 0) {
94640541d5dSvikram 		menu_root = use_stubboot();
947b610f78eSvikram 	} else {
948b610f78eSvikram 		menu_root = bam_root;
949b610f78eSvikram 	}
950b610f78eSvikram 
95140541d5dSvikram 	if (menu_root == NULL) {
95240541d5dSvikram 		bam_error(CANNOT_LOCATE_GRUB_MENU);
95340541d5dSvikram 		return (BAM_ERROR);
95440541d5dSvikram 	}
95540541d5dSvikram 
95640541d5dSvikram 	elide_trailing_slash(menu_root, menu_path, sizeof (menu_path));
95740541d5dSvikram 	(void) strlcat(menu_path, GRUB_MENU, sizeof (menu_path));
95840541d5dSvikram 
95940541d5dSvikram 	/*
96040541d5dSvikram 	 * If listing the menu, display the active menu
96140541d5dSvikram 	 * location
96240541d5dSvikram 	 */
96340541d5dSvikram 	if (strcmp(subcmd, "list_entry") == 0) {
96440541d5dSvikram 		disp_active_menu_locn(menu_path, logslice, fstype, mnted);
96540541d5dSvikram 	}
9667c478bd9Sstevel@tonic-gate 
9677c478bd9Sstevel@tonic-gate 	menu = menu_read(menu_path);
9687c478bd9Sstevel@tonic-gate 	assert(menu);
9697c478bd9Sstevel@tonic-gate 
9707c478bd9Sstevel@tonic-gate 	/*
9717c478bd9Sstevel@tonic-gate 	 * Special handling for setting timeout and default
9727c478bd9Sstevel@tonic-gate 	 */
9737c478bd9Sstevel@tonic-gate 	if (strcmp(subcmd, "set_option") == 0) {
9747c478bd9Sstevel@tonic-gate 		if (largc != 1 || largv[0] == NULL) {
9757c478bd9Sstevel@tonic-gate 			usage();
97640541d5dSvikram 			menu_free(menu);
97740541d5dSvikram 			umount_grub_slice(mnted, mntpt, NULL, logslice, fstype);
9787c478bd9Sstevel@tonic-gate 			return (BAM_ERROR);
9797c478bd9Sstevel@tonic-gate 		}
9807c478bd9Sstevel@tonic-gate 		opt = largv[0];
9817c478bd9Sstevel@tonic-gate 	} else if (largc != 0) {
9827c478bd9Sstevel@tonic-gate 		usage();
98340541d5dSvikram 		menu_free(menu);
98440541d5dSvikram 		umount_grub_slice(mnted, mntpt, NULL, logslice, fstype);
9857c478bd9Sstevel@tonic-gate 		return (BAM_ERROR);
9867c478bd9Sstevel@tonic-gate 	}
9877c478bd9Sstevel@tonic-gate 
988ae115bc7Smrj 	ret = dboot_or_multiboot(bam_root);
989ae115bc7Smrj 	if (ret != BAM_SUCCESS)
990ae115bc7Smrj 		return (ret);
991ae115bc7Smrj 
9927c478bd9Sstevel@tonic-gate 	/*
9937c478bd9Sstevel@tonic-gate 	 * Once the sub-cmd handler has run
9947c478bd9Sstevel@tonic-gate 	 * only the line field is guaranteed to have valid values
9957c478bd9Sstevel@tonic-gate 	 */
996ae115bc7Smrj 	if ((strcmp(subcmd, "update_entry") == 0) ||
997ae115bc7Smrj 	    (strcmp(subcmd, "upgrade") == 0))
9987c478bd9Sstevel@tonic-gate 		ret = f(menu, bam_root, opt);
9997c478bd9Sstevel@tonic-gate 	else
10007c478bd9Sstevel@tonic-gate 		ret = f(menu, menu_path, opt);
10017c478bd9Sstevel@tonic-gate 	if (ret == BAM_WRITE) {
1002b610f78eSvikram 		ret = menu_write(menu_root, menu);
10037c478bd9Sstevel@tonic-gate 	}
10047c478bd9Sstevel@tonic-gate 
10057c478bd9Sstevel@tonic-gate 	menu_free(menu);
10067c478bd9Sstevel@tonic-gate 
100740541d5dSvikram 	umount_grub_slice(mnted, mntpt, NULL, logslice, fstype);
1008b610f78eSvikram 
10097c478bd9Sstevel@tonic-gate 	return (ret);
10107c478bd9Sstevel@tonic-gate }
10117c478bd9Sstevel@tonic-gate 
10127c478bd9Sstevel@tonic-gate 
10137c478bd9Sstevel@tonic-gate static error_t
10147c478bd9Sstevel@tonic-gate bam_archive(
10157c478bd9Sstevel@tonic-gate 	char *subcmd,
10167c478bd9Sstevel@tonic-gate 	char *opt)
10177c478bd9Sstevel@tonic-gate {
10187c478bd9Sstevel@tonic-gate 	error_t ret;
10197c478bd9Sstevel@tonic-gate 	error_t (*f)(char *root, char *opt);
10207c478bd9Sstevel@tonic-gate 
10217c478bd9Sstevel@tonic-gate 	/*
10228c1b6884Sszhou 	 * Add trailing / for archive subcommands
10238c1b6884Sszhou 	 */
10248c1b6884Sszhou 	if (rootbuf[strlen(rootbuf) - 1] != '/')
10258c1b6884Sszhou 		(void) strcat(rootbuf, "/");
10268c1b6884Sszhou 	bam_rootlen = strlen(rootbuf);
10278c1b6884Sszhou 
10288c1b6884Sszhou 	/*
10297c478bd9Sstevel@tonic-gate 	 * Check arguments
10307c478bd9Sstevel@tonic-gate 	 */
10317c478bd9Sstevel@tonic-gate 	ret = check_subcmd_and_options(subcmd, opt, arch_subcmds, &f);
10327c478bd9Sstevel@tonic-gate 	if (ret != BAM_SUCCESS) {
10337c478bd9Sstevel@tonic-gate 		return (BAM_ERROR);
10347c478bd9Sstevel@tonic-gate 	}
10357c478bd9Sstevel@tonic-gate 
10367c478bd9Sstevel@tonic-gate #if defined(__sparc)
10377c478bd9Sstevel@tonic-gate 	/*
10387c478bd9Sstevel@tonic-gate 	 * A NOP if called on SPARC during reboot
10397c478bd9Sstevel@tonic-gate 	 */
10407c478bd9Sstevel@tonic-gate 	if (strcmp(subcmd, "update_all") == 0)
10417c478bd9Sstevel@tonic-gate 		return (BAM_SUCCESS);
10427c478bd9Sstevel@tonic-gate 	else if (strcmp(subcmd, "update") != 0)
10437c478bd9Sstevel@tonic-gate 		sparc_abort();
10447c478bd9Sstevel@tonic-gate #endif
10457c478bd9Sstevel@tonic-gate 
1046ae115bc7Smrj 	ret = dboot_or_multiboot(rootbuf);
1047ae115bc7Smrj 	if (ret != BAM_SUCCESS)
1048ae115bc7Smrj 		return (ret);
1049ae115bc7Smrj 
10507c478bd9Sstevel@tonic-gate 	/*
10517c478bd9Sstevel@tonic-gate 	 * Check archive not supported with update_all
10527c478bd9Sstevel@tonic-gate 	 * since it is awkward to display out-of-sync
10537c478bd9Sstevel@tonic-gate 	 * information for each BE.
10547c478bd9Sstevel@tonic-gate 	 */
10557c478bd9Sstevel@tonic-gate 	if (bam_check && strcmp(subcmd, "update_all") == 0) {
10567c478bd9Sstevel@tonic-gate 		bam_error(CHECK_NOT_SUPPORTED, subcmd);
10577c478bd9Sstevel@tonic-gate 		return (BAM_ERROR);
10587c478bd9Sstevel@tonic-gate 	}
10597c478bd9Sstevel@tonic-gate 
1060b610f78eSvikram 	if (strcmp(subcmd, "update_all") == 0)
1061b610f78eSvikram 		bam_update_all = 1;
1062b610f78eSvikram 
1063*2449e17fSsherrym #if defined(__i386)
1064*2449e17fSsherrym 	ucode_install(bam_root);
1065*2449e17fSsherrym #endif
1066*2449e17fSsherrym 
1067b610f78eSvikram 	ret = f(bam_root, opt);
1068b610f78eSvikram 
1069b610f78eSvikram 	bam_update_all = 0;
1070b610f78eSvikram 
1071b610f78eSvikram 	return (ret);
10727c478bd9Sstevel@tonic-gate }
10737c478bd9Sstevel@tonic-gate 
10747c478bd9Sstevel@tonic-gate /*PRINTFLIKE1*/
1075ae115bc7Smrj void
10767c478bd9Sstevel@tonic-gate bam_error(char *format, ...)
10777c478bd9Sstevel@tonic-gate {
10787c478bd9Sstevel@tonic-gate 	va_list ap;
10797c478bd9Sstevel@tonic-gate 
10807c478bd9Sstevel@tonic-gate 	va_start(ap, format);
10817c478bd9Sstevel@tonic-gate 	(void) fprintf(stderr, "%s: ", prog);
10827c478bd9Sstevel@tonic-gate 	(void) vfprintf(stderr, format, ap);
10837c478bd9Sstevel@tonic-gate 	va_end(ap);
10847c478bd9Sstevel@tonic-gate }
10857c478bd9Sstevel@tonic-gate 
10867c478bd9Sstevel@tonic-gate /*PRINTFLIKE1*/
10877c478bd9Sstevel@tonic-gate static void
10887c478bd9Sstevel@tonic-gate bam_print(char *format, ...)
10897c478bd9Sstevel@tonic-gate {
10907c478bd9Sstevel@tonic-gate 	va_list ap;
10917c478bd9Sstevel@tonic-gate 
10927c478bd9Sstevel@tonic-gate 	va_start(ap, format);
10937c478bd9Sstevel@tonic-gate 	(void) vfprintf(stdout, format, ap);
10947c478bd9Sstevel@tonic-gate 	va_end(ap);
10957c478bd9Sstevel@tonic-gate }
10967c478bd9Sstevel@tonic-gate 
1097ae115bc7Smrj /*PRINTFLIKE1*/
1098ae115bc7Smrj void
1099ae115bc7Smrj bam_print_stderr(char *format, ...)
1100ae115bc7Smrj {
1101ae115bc7Smrj 	va_list ap;
1102ae115bc7Smrj 
1103ae115bc7Smrj 	va_start(ap, format);
1104ae115bc7Smrj 	(void) vfprintf(stderr, format, ap);
1105ae115bc7Smrj 	va_end(ap);
1106ae115bc7Smrj }
1107ae115bc7Smrj 
11087c478bd9Sstevel@tonic-gate static void
11097c478bd9Sstevel@tonic-gate bam_exit(int excode)
11107c478bd9Sstevel@tonic-gate {
11117c478bd9Sstevel@tonic-gate 	bam_unlock();
11127c478bd9Sstevel@tonic-gate 	exit(excode);
11137c478bd9Sstevel@tonic-gate }
11147c478bd9Sstevel@tonic-gate 
11157c478bd9Sstevel@tonic-gate static void
11167c478bd9Sstevel@tonic-gate bam_lock(void)
11177c478bd9Sstevel@tonic-gate {
11187c478bd9Sstevel@tonic-gate 	struct flock lock;
11197c478bd9Sstevel@tonic-gate 	pid_t pid;
11207c478bd9Sstevel@tonic-gate 
11217c478bd9Sstevel@tonic-gate 	bam_lock_fd = open(BAM_LOCK_FILE, O_CREAT|O_RDWR, LOCK_FILE_PERMS);
11227c478bd9Sstevel@tonic-gate 	if (bam_lock_fd < 0) {
11237c478bd9Sstevel@tonic-gate 		/*
11247c478bd9Sstevel@tonic-gate 		 * We may be invoked early in boot for archive verification.
11257c478bd9Sstevel@tonic-gate 		 * In this case, root is readonly and /var/run may not exist.
11267c478bd9Sstevel@tonic-gate 		 * Proceed without the lock
11277c478bd9Sstevel@tonic-gate 		 */
11287c478bd9Sstevel@tonic-gate 		if (errno == EROFS || errno == ENOENT) {
11297c478bd9Sstevel@tonic-gate 			bam_root_readonly = 1;
11307c478bd9Sstevel@tonic-gate 			return;
11317c478bd9Sstevel@tonic-gate 		}
11327c478bd9Sstevel@tonic-gate 
11337c478bd9Sstevel@tonic-gate 		bam_error(OPEN_FAIL, BAM_LOCK_FILE, strerror(errno));
11347c478bd9Sstevel@tonic-gate 		bam_exit(1);
11357c478bd9Sstevel@tonic-gate 	}
11367c478bd9Sstevel@tonic-gate 
11377c478bd9Sstevel@tonic-gate 	lock.l_type = F_WRLCK;
11387c478bd9Sstevel@tonic-gate 	lock.l_whence = SEEK_SET;
11397c478bd9Sstevel@tonic-gate 	lock.l_start = 0;
11407c478bd9Sstevel@tonic-gate 	lock.l_len = 0;
11417c478bd9Sstevel@tonic-gate 
11427c478bd9Sstevel@tonic-gate 	if (fcntl(bam_lock_fd, F_SETLK, &lock) == -1) {
11437c478bd9Sstevel@tonic-gate 		if (errno != EACCES && errno != EAGAIN) {
11447c478bd9Sstevel@tonic-gate 			bam_error(LOCK_FAIL, BAM_LOCK_FILE, strerror(errno));
11457c478bd9Sstevel@tonic-gate 			(void) close(bam_lock_fd);
11467c478bd9Sstevel@tonic-gate 			bam_lock_fd = -1;
11477c478bd9Sstevel@tonic-gate 			bam_exit(1);
11487c478bd9Sstevel@tonic-gate 		}
11497c478bd9Sstevel@tonic-gate 		pid = 0;
11507c478bd9Sstevel@tonic-gate 		(void) pread(bam_lock_fd, &pid, sizeof (pid_t), 0);
11517c478bd9Sstevel@tonic-gate 		bam_print(FILE_LOCKED, pid);
11527c478bd9Sstevel@tonic-gate 
11537c478bd9Sstevel@tonic-gate 		lock.l_type = F_WRLCK;
11547c478bd9Sstevel@tonic-gate 		lock.l_whence = SEEK_SET;
11557c478bd9Sstevel@tonic-gate 		lock.l_start = 0;
11567c478bd9Sstevel@tonic-gate 		lock.l_len = 0;
11577c478bd9Sstevel@tonic-gate 		if (fcntl(bam_lock_fd, F_SETLKW, &lock) == -1) {
11587c478bd9Sstevel@tonic-gate 			bam_error(LOCK_FAIL, BAM_LOCK_FILE, strerror(errno));
11597c478bd9Sstevel@tonic-gate 			(void) close(bam_lock_fd);
11607c478bd9Sstevel@tonic-gate 			bam_lock_fd = -1;
11617c478bd9Sstevel@tonic-gate 			bam_exit(1);
11627c478bd9Sstevel@tonic-gate 		}
11637c478bd9Sstevel@tonic-gate 	}
11647c478bd9Sstevel@tonic-gate 
11657c478bd9Sstevel@tonic-gate 	/* We own the lock now */
11667c478bd9Sstevel@tonic-gate 	pid = getpid();
11677c478bd9Sstevel@tonic-gate 	(void) write(bam_lock_fd, &pid, sizeof (pid));
11687c478bd9Sstevel@tonic-gate }
11697c478bd9Sstevel@tonic-gate 
11707c478bd9Sstevel@tonic-gate static void
11717c478bd9Sstevel@tonic-gate bam_unlock(void)
11727c478bd9Sstevel@tonic-gate {
11737c478bd9Sstevel@tonic-gate 	struct flock unlock;
11747c478bd9Sstevel@tonic-gate 
11757c478bd9Sstevel@tonic-gate 	/*
11767c478bd9Sstevel@tonic-gate 	 * NOP if we don't hold the lock
11777c478bd9Sstevel@tonic-gate 	 */
11787c478bd9Sstevel@tonic-gate 	if (bam_lock_fd < 0) {
11797c478bd9Sstevel@tonic-gate 		return;
11807c478bd9Sstevel@tonic-gate 	}
11817c478bd9Sstevel@tonic-gate 
11827c478bd9Sstevel@tonic-gate 	unlock.l_type = F_UNLCK;
11837c478bd9Sstevel@tonic-gate 	unlock.l_whence = SEEK_SET;
11847c478bd9Sstevel@tonic-gate 	unlock.l_start = 0;
11857c478bd9Sstevel@tonic-gate 	unlock.l_len = 0;
11867c478bd9Sstevel@tonic-gate 
11877c478bd9Sstevel@tonic-gate 	if (fcntl(bam_lock_fd, F_SETLK, &unlock) == -1) {
11887c478bd9Sstevel@tonic-gate 		bam_error(UNLOCK_FAIL, BAM_LOCK_FILE, strerror(errno));
11897c478bd9Sstevel@tonic-gate 	}
11907c478bd9Sstevel@tonic-gate 
11917c478bd9Sstevel@tonic-gate 	if (close(bam_lock_fd) == -1) {
11927c478bd9Sstevel@tonic-gate 		bam_error(CLOSE_FAIL, BAM_LOCK_FILE, strerror(errno));
11937c478bd9Sstevel@tonic-gate 	}
11947c478bd9Sstevel@tonic-gate 	bam_lock_fd = -1;
11957c478bd9Sstevel@tonic-gate }
11967c478bd9Sstevel@tonic-gate 
11977c478bd9Sstevel@tonic-gate static error_t
11987c478bd9Sstevel@tonic-gate list_archive(char *root, char *opt)
11997c478bd9Sstevel@tonic-gate {
12007c478bd9Sstevel@tonic-gate 	filelist_t flist;
12017c478bd9Sstevel@tonic-gate 	filelist_t *flistp = &flist;
12027c478bd9Sstevel@tonic-gate 	line_t *lp;
12037c478bd9Sstevel@tonic-gate 
12047c478bd9Sstevel@tonic-gate 	assert(root);
12057c478bd9Sstevel@tonic-gate 	assert(opt == NULL);
12067c478bd9Sstevel@tonic-gate 
12077c478bd9Sstevel@tonic-gate 	flistp->head = flistp->tail = NULL;
12087c478bd9Sstevel@tonic-gate 	if (read_list(root, flistp) != BAM_SUCCESS) {
12097c478bd9Sstevel@tonic-gate 		return (BAM_ERROR);
12107c478bd9Sstevel@tonic-gate 	}
12117c478bd9Sstevel@tonic-gate 	assert(flistp->head && flistp->tail);
12127c478bd9Sstevel@tonic-gate 
12137c478bd9Sstevel@tonic-gate 	for (lp = flistp->head; lp; lp = lp->next) {
12147c478bd9Sstevel@tonic-gate 		bam_print(PRINT, lp->line);
12157c478bd9Sstevel@tonic-gate 	}
12167c478bd9Sstevel@tonic-gate 
12177c478bd9Sstevel@tonic-gate 	filelist_free(flistp);
12187c478bd9Sstevel@tonic-gate 
12197c478bd9Sstevel@tonic-gate 	return (BAM_SUCCESS);
12207c478bd9Sstevel@tonic-gate }
12217c478bd9Sstevel@tonic-gate 
12227c478bd9Sstevel@tonic-gate /*
12237c478bd9Sstevel@tonic-gate  * This routine writes a list of lines to a file.
12247c478bd9Sstevel@tonic-gate  * The list is *not* freed
12257c478bd9Sstevel@tonic-gate  */
12267c478bd9Sstevel@tonic-gate static error_t
12277c478bd9Sstevel@tonic-gate list2file(char *root, char *tmp, char *final, line_t *start)
12287c478bd9Sstevel@tonic-gate {
12297c478bd9Sstevel@tonic-gate 	char tmpfile[PATH_MAX];
12307c478bd9Sstevel@tonic-gate 	char path[PATH_MAX];
12317c478bd9Sstevel@tonic-gate 	FILE *fp;
12327c478bd9Sstevel@tonic-gate 	int ret;
12337c478bd9Sstevel@tonic-gate 	struct stat sb;
12347c478bd9Sstevel@tonic-gate 	mode_t mode;
12357c478bd9Sstevel@tonic-gate 	uid_t root_uid;
12367c478bd9Sstevel@tonic-gate 	gid_t sys_gid;
12377c478bd9Sstevel@tonic-gate 	struct passwd *pw;
12387c478bd9Sstevel@tonic-gate 	struct group *gp;
12397c478bd9Sstevel@tonic-gate 
12407c478bd9Sstevel@tonic-gate 
12417c478bd9Sstevel@tonic-gate 	(void) snprintf(path, sizeof (path), "%s%s", root, final);
12427c478bd9Sstevel@tonic-gate 
12437c478bd9Sstevel@tonic-gate 	if (start == NULL) {
12447c478bd9Sstevel@tonic-gate 		if (stat(path, &sb) != -1) {
12457c478bd9Sstevel@tonic-gate 			bam_print(UNLINK_EMPTY, path);
12467c478bd9Sstevel@tonic-gate 			if (unlink(path) != 0) {
12477c478bd9Sstevel@tonic-gate 				bam_error(UNLINK_FAIL, path, strerror(errno));
12487c478bd9Sstevel@tonic-gate 				return (BAM_ERROR);
12497c478bd9Sstevel@tonic-gate 			} else {
12507c478bd9Sstevel@tonic-gate 				return (BAM_SUCCESS);
12517c478bd9Sstevel@tonic-gate 			}
12527c478bd9Sstevel@tonic-gate 		}
12537c478bd9Sstevel@tonic-gate 	}
12547c478bd9Sstevel@tonic-gate 
12557c478bd9Sstevel@tonic-gate 	/*
12567c478bd9Sstevel@tonic-gate 	 * Preserve attributes of existing file if possible,
12577c478bd9Sstevel@tonic-gate 	 * otherwise ask the system for uid/gid of root/sys.
12587c478bd9Sstevel@tonic-gate 	 * If all fails, fall back on hard-coded defaults.
12597c478bd9Sstevel@tonic-gate 	 */
12607c478bd9Sstevel@tonic-gate 	if (stat(path, &sb) != -1) {
12617c478bd9Sstevel@tonic-gate 		mode = sb.st_mode;
12627c478bd9Sstevel@tonic-gate 		root_uid = sb.st_uid;
12637c478bd9Sstevel@tonic-gate 		sys_gid = sb.st_gid;
12647c478bd9Sstevel@tonic-gate 	} else {
12657c478bd9Sstevel@tonic-gate 		mode = DEFAULT_DEV_MODE;
12667c478bd9Sstevel@tonic-gate 		if ((pw = getpwnam(DEFAULT_DEV_USER)) != NULL) {
12677c478bd9Sstevel@tonic-gate 			root_uid = pw->pw_uid;
12687c478bd9Sstevel@tonic-gate 		} else {
12697c478bd9Sstevel@tonic-gate 			if (bam_verbose)
12707c478bd9Sstevel@tonic-gate 				bam_error(CANT_FIND_USER,
12717c478bd9Sstevel@tonic-gate 				    DEFAULT_DEV_USER, DEFAULT_DEV_UID);
12727c478bd9Sstevel@tonic-gate 			root_uid = (uid_t)DEFAULT_DEV_UID;
12737c478bd9Sstevel@tonic-gate 		}
12747c478bd9Sstevel@tonic-gate 		if ((gp = getgrnam(DEFAULT_DEV_GROUP)) != NULL) {
12757c478bd9Sstevel@tonic-gate 			sys_gid = gp->gr_gid;
12767c478bd9Sstevel@tonic-gate 		} else {
12777c478bd9Sstevel@tonic-gate 			if (bam_verbose)
12787c478bd9Sstevel@tonic-gate 				bam_error(CANT_FIND_GROUP,
12797c478bd9Sstevel@tonic-gate 				    DEFAULT_DEV_GROUP, DEFAULT_DEV_GID);
12807c478bd9Sstevel@tonic-gate 			sys_gid = (gid_t)DEFAULT_DEV_GID;
12817c478bd9Sstevel@tonic-gate 		}
12827c478bd9Sstevel@tonic-gate 	}
12837c478bd9Sstevel@tonic-gate 
12847c478bd9Sstevel@tonic-gate 	(void) snprintf(tmpfile, sizeof (tmpfile), "%s%s", root, tmp);
12857c478bd9Sstevel@tonic-gate 
12867c478bd9Sstevel@tonic-gate 	/* Truncate tmpfile first */
12877c478bd9Sstevel@tonic-gate 	fp = fopen(tmpfile, "w");
12887c478bd9Sstevel@tonic-gate 	if (fp == NULL) {
12897c478bd9Sstevel@tonic-gate 		bam_error(OPEN_FAIL, tmpfile, strerror(errno));
12907c478bd9Sstevel@tonic-gate 		return (BAM_ERROR);
12917c478bd9Sstevel@tonic-gate 	}
12927c478bd9Sstevel@tonic-gate 	ret = fclose(fp);
12937c478bd9Sstevel@tonic-gate 	if (ret == EOF) {
12947c478bd9Sstevel@tonic-gate 		bam_error(CLOSE_FAIL, tmpfile, strerror(errno));
12957c478bd9Sstevel@tonic-gate 		return (BAM_ERROR);
12967c478bd9Sstevel@tonic-gate 	}
12977c478bd9Sstevel@tonic-gate 
12987c478bd9Sstevel@tonic-gate 	/* Now open it in append mode */
12997c478bd9Sstevel@tonic-gate 	fp = fopen(tmpfile, "a");
13007c478bd9Sstevel@tonic-gate 	if (fp == NULL) {
13017c478bd9Sstevel@tonic-gate 		bam_error(OPEN_FAIL, tmpfile, strerror(errno));
13027c478bd9Sstevel@tonic-gate 		return (BAM_ERROR);
13037c478bd9Sstevel@tonic-gate 	}
13047c478bd9Sstevel@tonic-gate 
13057c478bd9Sstevel@tonic-gate 	for (; start; start = start->next) {
13067c478bd9Sstevel@tonic-gate 		ret = s_fputs(start->line, fp);
13077c478bd9Sstevel@tonic-gate 		if (ret == EOF) {
13087c478bd9Sstevel@tonic-gate 			bam_error(WRITE_FAIL, tmpfile, strerror(errno));
13097c478bd9Sstevel@tonic-gate 			(void) fclose(fp);
13107c478bd9Sstevel@tonic-gate 			return (BAM_ERROR);
13117c478bd9Sstevel@tonic-gate 		}
13127c478bd9Sstevel@tonic-gate 	}
13137c478bd9Sstevel@tonic-gate 
13147c478bd9Sstevel@tonic-gate 	ret = fclose(fp);
13157c478bd9Sstevel@tonic-gate 	if (ret == EOF) {
13167c478bd9Sstevel@tonic-gate 		bam_error(CLOSE_FAIL, tmpfile, strerror(errno));
13177c478bd9Sstevel@tonic-gate 		return (BAM_ERROR);
13187c478bd9Sstevel@tonic-gate 	}
13197c478bd9Sstevel@tonic-gate 
13207c478bd9Sstevel@tonic-gate 	/*
1321de531be4Sjg 	 * Set up desired attributes.  Ignore failures on filesystems
1322de531be4Sjg 	 * not supporting these operations - pcfs reports unsupported
1323de531be4Sjg 	 * operations as EINVAL.
13247c478bd9Sstevel@tonic-gate 	 */
13257c478bd9Sstevel@tonic-gate 	ret = chmod(tmpfile, mode);
1326de531be4Sjg 	if (ret == -1 &&
1327de531be4Sjg 	    errno != EINVAL && errno != ENOTSUP) {
13287c478bd9Sstevel@tonic-gate 		bam_error(CHMOD_FAIL, tmpfile, strerror(errno));
13297c478bd9Sstevel@tonic-gate 		return (BAM_ERROR);
13307c478bd9Sstevel@tonic-gate 	}
13317c478bd9Sstevel@tonic-gate 
13327c478bd9Sstevel@tonic-gate 	ret = chown(tmpfile, root_uid, sys_gid);
1333de531be4Sjg 	if (ret == -1 &&
1334de531be4Sjg 	    errno != EINVAL && errno != ENOTSUP) {
13357c478bd9Sstevel@tonic-gate 		bam_error(CHOWN_FAIL, tmpfile, strerror(errno));
13367c478bd9Sstevel@tonic-gate 		return (BAM_ERROR);
13377c478bd9Sstevel@tonic-gate 	}
13387c478bd9Sstevel@tonic-gate 
13397c478bd9Sstevel@tonic-gate 
13407c478bd9Sstevel@tonic-gate 	/*
13417c478bd9Sstevel@tonic-gate 	 * Do an atomic rename
13427c478bd9Sstevel@tonic-gate 	 */
13437c478bd9Sstevel@tonic-gate 	ret = rename(tmpfile, path);
13447c478bd9Sstevel@tonic-gate 	if (ret != 0) {
13457c478bd9Sstevel@tonic-gate 		bam_error(RENAME_FAIL, path, strerror(errno));
13467c478bd9Sstevel@tonic-gate 		return (BAM_ERROR);
13477c478bd9Sstevel@tonic-gate 	}
13487c478bd9Sstevel@tonic-gate 
13497c478bd9Sstevel@tonic-gate 	return (BAM_SUCCESS);
13507c478bd9Sstevel@tonic-gate }
13517c478bd9Sstevel@tonic-gate 
13527c478bd9Sstevel@tonic-gate /*
13537c478bd9Sstevel@tonic-gate  * This function should always return 0 - since we want
13547c478bd9Sstevel@tonic-gate  * to create stat data for *all* files in the list.
13557c478bd9Sstevel@tonic-gate  */
13567c478bd9Sstevel@tonic-gate /*ARGSUSED*/
13577c478bd9Sstevel@tonic-gate static int
13587c478bd9Sstevel@tonic-gate cmpstat(
13597c478bd9Sstevel@tonic-gate 	const char *file,
13607c478bd9Sstevel@tonic-gate 	const struct stat *stat,
13617c478bd9Sstevel@tonic-gate 	int flags,
13627c478bd9Sstevel@tonic-gate 	struct FTW *ftw)
13637c478bd9Sstevel@tonic-gate {
13647c478bd9Sstevel@tonic-gate 	uint_t sz;
13657c478bd9Sstevel@tonic-gate 	uint64_t *value;
13667c478bd9Sstevel@tonic-gate 	uint64_t filestat[2];
13677c478bd9Sstevel@tonic-gate 	int error;
13687c478bd9Sstevel@tonic-gate 
136958091fd8Ssetje 	struct safefile *safefilep;
137058091fd8Ssetje 	FILE *fp;
137158091fd8Ssetje 
13727c478bd9Sstevel@tonic-gate 	/*
13737c478bd9Sstevel@tonic-gate 	 * We only want regular files
13747c478bd9Sstevel@tonic-gate 	 */
13757c478bd9Sstevel@tonic-gate 	if (!S_ISREG(stat->st_mode))
13767c478bd9Sstevel@tonic-gate 		return (0);
13777c478bd9Sstevel@tonic-gate 
13787c478bd9Sstevel@tonic-gate 	/*
13797c478bd9Sstevel@tonic-gate 	 * new_nvlp may be NULL if there were errors earlier
13807c478bd9Sstevel@tonic-gate 	 * but this is not fatal to update determination.
13817c478bd9Sstevel@tonic-gate 	 */
13827c478bd9Sstevel@tonic-gate 	if (walk_arg.new_nvlp) {
13837c478bd9Sstevel@tonic-gate 		filestat[0] = stat->st_size;
13847c478bd9Sstevel@tonic-gate 		filestat[1] = stat->st_mtime;
13857c478bd9Sstevel@tonic-gate 		error = nvlist_add_uint64_array(walk_arg.new_nvlp,
13867c478bd9Sstevel@tonic-gate 		    file + bam_rootlen, filestat, 2);
13877c478bd9Sstevel@tonic-gate 		if (error)
13887c478bd9Sstevel@tonic-gate 			bam_error(NVADD_FAIL, file, strerror(error));
13897c478bd9Sstevel@tonic-gate 	}
13907c478bd9Sstevel@tonic-gate 
13917c478bd9Sstevel@tonic-gate 	/*
13927c478bd9Sstevel@tonic-gate 	 * The remaining steps are only required if we haven't made a
13937c478bd9Sstevel@tonic-gate 	 * decision about update or if we are checking (-n)
13947c478bd9Sstevel@tonic-gate 	 */
13957c478bd9Sstevel@tonic-gate 	if (walk_arg.need_update && !bam_check)
13967c478bd9Sstevel@tonic-gate 		return (0);
13977c478bd9Sstevel@tonic-gate 
13987c478bd9Sstevel@tonic-gate 	/*
139958091fd8Ssetje 	 * If we are invoked as part of system/filesyste/boot-archive, then
140058091fd8Ssetje 	 * there are a number of things we should not worry about
14017c478bd9Sstevel@tonic-gate 	 */
140258091fd8Ssetje 	if (bam_smf_check) {
140358091fd8Ssetje 		/* ignore amd64 modules unless we are booted amd64. */
140458091fd8Ssetje 		if (!is_amd64() && strstr(file, "/amd64/") != 0)
14057c478bd9Sstevel@tonic-gate 			return (0);
14067c478bd9Sstevel@tonic-gate 
140758091fd8Ssetje 		/* read in list of safe files */
140858091fd8Ssetje 		if (safefiles == NULL)
140958091fd8Ssetje 			if (fp = fopen("/boot/solaris/filelist.safe", "r")) {
141058091fd8Ssetje 				safefiles = s_calloc(1,
141158091fd8Ssetje 				    sizeof (struct safefile));
141258091fd8Ssetje 				safefilep = safefiles;
141358091fd8Ssetje 				safefilep->name = s_calloc(1, MAXPATHLEN +
141458091fd8Ssetje 				    MAXNAMELEN);
141558091fd8Ssetje 				safefilep->next = NULL;
141658091fd8Ssetje 				while (s_fgets(safefilep->name, MAXPATHLEN +
141758091fd8Ssetje 				    MAXNAMELEN, fp) != NULL) {
141858091fd8Ssetje 					safefilep->next = s_calloc(1,
141958091fd8Ssetje 					    sizeof (struct safefile));
142058091fd8Ssetje 					safefilep = safefilep->next;
142158091fd8Ssetje 					safefilep->name = s_calloc(1,
142258091fd8Ssetje 					    MAXPATHLEN + MAXNAMELEN);
142358091fd8Ssetje 					safefilep->next = NULL;
142458091fd8Ssetje 				}
142558091fd8Ssetje 				(void) fclose(fp);
142658091fd8Ssetje 			}
142758091fd8Ssetje 	}
142858091fd8Ssetje 
14297c478bd9Sstevel@tonic-gate 	/*
14307c478bd9Sstevel@tonic-gate 	 * We need an update if file doesn't exist in old archive
14317c478bd9Sstevel@tonic-gate 	 */
14327c478bd9Sstevel@tonic-gate 	if (walk_arg.old_nvlp == NULL ||
14337c478bd9Sstevel@tonic-gate 	    nvlist_lookup_uint64_array(walk_arg.old_nvlp,
14347c478bd9Sstevel@tonic-gate 	    file + bam_rootlen, &value, &sz) != 0) {
14357c478bd9Sstevel@tonic-gate 		if (bam_smf_check)	/* ignore new during smf check */
14367c478bd9Sstevel@tonic-gate 			return (0);
14377c478bd9Sstevel@tonic-gate 		walk_arg.need_update = 1;
14387c478bd9Sstevel@tonic-gate 		if (bam_verbose)
14397c478bd9Sstevel@tonic-gate 			bam_print(PARSEABLE_NEW_FILE, file);
14407c478bd9Sstevel@tonic-gate 		return (0);
14417c478bd9Sstevel@tonic-gate 	}
14427c478bd9Sstevel@tonic-gate 
14437c478bd9Sstevel@tonic-gate 	/*
14447c478bd9Sstevel@tonic-gate 	 * File exists in old archive. Check if file has changed
14457c478bd9Sstevel@tonic-gate 	 */
14467c478bd9Sstevel@tonic-gate 	assert(sz == 2);
14477c478bd9Sstevel@tonic-gate 	bcopy(value, filestat, sizeof (filestat));
14487c478bd9Sstevel@tonic-gate 
14497c478bd9Sstevel@tonic-gate 	if (filestat[0] != stat->st_size ||
14507c478bd9Sstevel@tonic-gate 	    filestat[1] != stat->st_mtime) {
14517c609327Ssetje 		if (bam_smf_check) {
14527c609327Ssetje 			safefilep = safefiles;
14537c609327Ssetje 			while (safefilep != NULL) {
14547c609327Ssetje 				if (strcmp(file + bam_rootlen,
14557c609327Ssetje 				    safefilep->name) == 0) {
14567c609327Ssetje 					(void) creat(NEED_UPDATE_FILE, 0644);
14577c609327Ssetje 					return (0);
14587c609327Ssetje 				}
14597c609327Ssetje 				safefilep = safefilep->next;
14607c609327Ssetje 			}
14617c609327Ssetje 		}
14627c478bd9Sstevel@tonic-gate 		walk_arg.need_update = 1;
14637c478bd9Sstevel@tonic-gate 		if (bam_verbose)
14647c478bd9Sstevel@tonic-gate 			if (bam_smf_check)
14657c478bd9Sstevel@tonic-gate 				bam_print("    %s\n", file);
14667c478bd9Sstevel@tonic-gate 			else
14677c478bd9Sstevel@tonic-gate 				bam_print(PARSEABLE_OUT_DATE, file);
14687c478bd9Sstevel@tonic-gate 	}
14697c478bd9Sstevel@tonic-gate 
14707c478bd9Sstevel@tonic-gate 	return (0);
14717c478bd9Sstevel@tonic-gate }
14727c478bd9Sstevel@tonic-gate 
14737c478bd9Sstevel@tonic-gate /*
14747c478bd9Sstevel@tonic-gate  * Check flags and presence of required files.
14757c478bd9Sstevel@tonic-gate  * The force flag and/or absence of files should
14767c478bd9Sstevel@tonic-gate  * trigger an update.
14777c478bd9Sstevel@tonic-gate  * Suppress stdout output if check (-n) option is set
14787c478bd9Sstevel@tonic-gate  * (as -n should only produce parseable output.)
14797c478bd9Sstevel@tonic-gate  */
14807c478bd9Sstevel@tonic-gate static void
14817c478bd9Sstevel@tonic-gate check_flags_and_files(char *root)
14827c478bd9Sstevel@tonic-gate {
14837c478bd9Sstevel@tonic-gate 	char path[PATH_MAX];
14847c478bd9Sstevel@tonic-gate 	struct stat sb;
14857c478bd9Sstevel@tonic-gate 
14867c478bd9Sstevel@tonic-gate 	/*
14877c478bd9Sstevel@tonic-gate 	 * if force, create archive unconditionally
14887c478bd9Sstevel@tonic-gate 	 */
14897c478bd9Sstevel@tonic-gate 	if (bam_force) {
14907c478bd9Sstevel@tonic-gate 		walk_arg.need_update = 1;
14917c478bd9Sstevel@tonic-gate 		if (bam_verbose && !bam_check)
14927c478bd9Sstevel@tonic-gate 			bam_print(UPDATE_FORCE);
14937c478bd9Sstevel@tonic-gate 		return;
14947c478bd9Sstevel@tonic-gate 	}
14957c478bd9Sstevel@tonic-gate 
14967c478bd9Sstevel@tonic-gate 	/*
14977c478bd9Sstevel@tonic-gate 	 * If archive is missing, create archive
14987c478bd9Sstevel@tonic-gate 	 */
1499ae115bc7Smrj 	(void) snprintf(path, sizeof (path), "%s%s", root,
1500ae115bc7Smrj 	    DIRECT_BOOT_ARCHIVE_32);
15017c478bd9Sstevel@tonic-gate 	if (stat(path, &sb) != 0) {
15027c478bd9Sstevel@tonic-gate 		if (bam_verbose && !bam_check)
15037c478bd9Sstevel@tonic-gate 			bam_print(UPDATE_ARCH_MISS, path);
15047c478bd9Sstevel@tonic-gate 		walk_arg.need_update = 1;
15057c478bd9Sstevel@tonic-gate 		return;
15067c478bd9Sstevel@tonic-gate 	}
1507ae115bc7Smrj 	if (bam_direct == BAM_DIRECT_DBOOT) {
1508ae115bc7Smrj 		(void) snprintf(path, sizeof (path), "%s%s", root,
1509ae115bc7Smrj 		    DIRECT_BOOT_ARCHIVE_64);
1510ae115bc7Smrj 		if (stat(path, &sb) != 0) {
1511ae115bc7Smrj 			if (bam_verbose && !bam_check)
1512ae115bc7Smrj 				bam_print(UPDATE_ARCH_MISS, path);
1513ae115bc7Smrj 			walk_arg.need_update = 1;
1514ae115bc7Smrj 			return;
1515ae115bc7Smrj 		}
1516ae115bc7Smrj 	}
15177c478bd9Sstevel@tonic-gate }
15187c478bd9Sstevel@tonic-gate 
15197c478bd9Sstevel@tonic-gate static error_t
15207c478bd9Sstevel@tonic-gate read_one_list(char *root, filelist_t  *flistp, char *filelist)
15217c478bd9Sstevel@tonic-gate {
15227c478bd9Sstevel@tonic-gate 	char path[PATH_MAX];
15237c478bd9Sstevel@tonic-gate 	FILE *fp;
15247c478bd9Sstevel@tonic-gate 	char buf[BAM_MAXLINE];
15257c478bd9Sstevel@tonic-gate 
15267c478bd9Sstevel@tonic-gate 	(void) snprintf(path, sizeof (path), "%s%s", root, filelist);
15277c478bd9Sstevel@tonic-gate 
15287c478bd9Sstevel@tonic-gate 	fp = fopen(path, "r");
15297c478bd9Sstevel@tonic-gate 	if (fp == NULL) {
15307c478bd9Sstevel@tonic-gate 		if (bam_debug)
15317c478bd9Sstevel@tonic-gate 			bam_error(FLIST_FAIL, path, strerror(errno));
15327c478bd9Sstevel@tonic-gate 		return (BAM_ERROR);
15337c478bd9Sstevel@tonic-gate 	}
15347c478bd9Sstevel@tonic-gate 	while (s_fgets(buf, sizeof (buf), fp) != NULL) {
1535b610f78eSvikram 		/* skip blank lines */
1536b610f78eSvikram 		if (strspn(buf, " \t") == strlen(buf))
1537b610f78eSvikram 			continue;
15387c478bd9Sstevel@tonic-gate 		append_to_flist(flistp, buf);
15397c478bd9Sstevel@tonic-gate 	}
15407c478bd9Sstevel@tonic-gate 	if (fclose(fp) != 0) {
15417c478bd9Sstevel@tonic-gate 		bam_error(CLOSE_FAIL, path, strerror(errno));
15427c478bd9Sstevel@tonic-gate 		return (BAM_ERROR);
15437c478bd9Sstevel@tonic-gate 	}
15447c478bd9Sstevel@tonic-gate 	return (BAM_SUCCESS);
15457c478bd9Sstevel@tonic-gate }
15467c478bd9Sstevel@tonic-gate 
15477c478bd9Sstevel@tonic-gate static error_t
15487c478bd9Sstevel@tonic-gate read_list(char *root, filelist_t  *flistp)
15497c478bd9Sstevel@tonic-gate {
15507c478bd9Sstevel@tonic-gate 	int rval;
15517c478bd9Sstevel@tonic-gate 
15527c478bd9Sstevel@tonic-gate 	flistp->head = flistp->tail = NULL;
15537c478bd9Sstevel@tonic-gate 
15547c478bd9Sstevel@tonic-gate 	/*
15557c478bd9Sstevel@tonic-gate 	 * Read current lists of files - only the first is mandatory
15567c478bd9Sstevel@tonic-gate 	 */
15577c478bd9Sstevel@tonic-gate 	rval = read_one_list(root, flistp, BOOT_FILE_LIST);
15587c478bd9Sstevel@tonic-gate 	if (rval != BAM_SUCCESS)
15597c478bd9Sstevel@tonic-gate 		return (rval);
15607c478bd9Sstevel@tonic-gate 	(void) read_one_list(root, flistp, ETC_FILE_LIST);
15617c478bd9Sstevel@tonic-gate 
15627c478bd9Sstevel@tonic-gate 	if (flistp->head == NULL) {
15637c478bd9Sstevel@tonic-gate 		bam_error(NO_FLIST);
15647c478bd9Sstevel@tonic-gate 		return (BAM_ERROR);
15657c478bd9Sstevel@tonic-gate 	}
15667c478bd9Sstevel@tonic-gate 
15677c478bd9Sstevel@tonic-gate 	return (BAM_SUCCESS);
15687c478bd9Sstevel@tonic-gate }
15697c478bd9Sstevel@tonic-gate 
15707c478bd9Sstevel@tonic-gate static void
15717c478bd9Sstevel@tonic-gate getoldstat(char *root)
15727c478bd9Sstevel@tonic-gate {
15737c478bd9Sstevel@tonic-gate 	char path[PATH_MAX];
15747c478bd9Sstevel@tonic-gate 	int fd, error;
15757c478bd9Sstevel@tonic-gate 	struct stat sb;
15767c478bd9Sstevel@tonic-gate 	char *ostat;
15777c478bd9Sstevel@tonic-gate 
15787c478bd9Sstevel@tonic-gate 	(void) snprintf(path, sizeof (path), "%s%s", root, FILE_STAT);
15797c478bd9Sstevel@tonic-gate 	fd = open(path, O_RDONLY);
15807c478bd9Sstevel@tonic-gate 	if (fd == -1) {
15817c478bd9Sstevel@tonic-gate 		if (bam_verbose)
15827c478bd9Sstevel@tonic-gate 			bam_print(OPEN_FAIL, path, strerror(errno));
15837c478bd9Sstevel@tonic-gate 		walk_arg.need_update = 1;
15847c478bd9Sstevel@tonic-gate 		return;
15857c478bd9Sstevel@tonic-gate 	}
15867c478bd9Sstevel@tonic-gate 
15877c478bd9Sstevel@tonic-gate 	if (fstat(fd, &sb) != 0) {
15887c478bd9Sstevel@tonic-gate 		bam_error(STAT_FAIL, path, strerror(errno));
15897c478bd9Sstevel@tonic-gate 		(void) close(fd);
15907c478bd9Sstevel@tonic-gate 		walk_arg.need_update = 1;
15917c478bd9Sstevel@tonic-gate 		return;
15927c478bd9Sstevel@tonic-gate 	}
15937c478bd9Sstevel@tonic-gate 
15947c478bd9Sstevel@tonic-gate 	ostat = s_calloc(1, sb.st_size);
15957c478bd9Sstevel@tonic-gate 
15967c478bd9Sstevel@tonic-gate 	if (read(fd, ostat, sb.st_size) != sb.st_size) {
15977c478bd9Sstevel@tonic-gate 		bam_error(READ_FAIL, path, strerror(errno));
15987c478bd9Sstevel@tonic-gate 		(void) close(fd);
15997c478bd9Sstevel@tonic-gate 		free(ostat);
16007c478bd9Sstevel@tonic-gate 		walk_arg.need_update = 1;
16017c478bd9Sstevel@tonic-gate 		return;
16027c478bd9Sstevel@tonic-gate 	}
16037c478bd9Sstevel@tonic-gate 
16047c478bd9Sstevel@tonic-gate 	(void) close(fd);
16057c478bd9Sstevel@tonic-gate 
16067c478bd9Sstevel@tonic-gate 	walk_arg.old_nvlp = NULL;
16077c478bd9Sstevel@tonic-gate 	error = nvlist_unpack(ostat, sb.st_size, &walk_arg.old_nvlp, 0);
16087c478bd9Sstevel@tonic-gate 
16097c478bd9Sstevel@tonic-gate 	free(ostat);
16107c478bd9Sstevel@tonic-gate 
16117c478bd9Sstevel@tonic-gate 	if (error) {
16127c478bd9Sstevel@tonic-gate 		bam_error(UNPACK_FAIL, path, strerror(error));
16137c478bd9Sstevel@tonic-gate 		walk_arg.old_nvlp = NULL;
16147c478bd9Sstevel@tonic-gate 		walk_arg.need_update = 1;
16157c478bd9Sstevel@tonic-gate 		return;
16167c478bd9Sstevel@tonic-gate 	}
16177c478bd9Sstevel@tonic-gate }
16187c478bd9Sstevel@tonic-gate 
1619a28d77b8Svikram /*
1620a28d77b8Svikram  * Checks if a file in the current (old) archive has
1621a28d77b8Svikram  * been deleted from the root filesystem. This is needed for
1622a28d77b8Svikram  * software like Trusted Extensions (TX) that switch early
1623a28d77b8Svikram  * in boot based on presence/absence of a kernel module.
1624a28d77b8Svikram  */
1625a28d77b8Svikram static void
1626a28d77b8Svikram check4stale(char *root)
1627a28d77b8Svikram {
1628a28d77b8Svikram 	nvpair_t	*nvp;
1629a28d77b8Svikram 	nvlist_t	*nvlp;
1630a28d77b8Svikram 	char 		*file;
1631a28d77b8Svikram 	char		path[PATH_MAX];
1632a28d77b8Svikram 	struct stat	sb;
1633a28d77b8Svikram 
1634a28d77b8Svikram 	/*
1635a28d77b8Svikram 	 * Skip stale file check during smf check
1636a28d77b8Svikram 	 */
1637a28d77b8Svikram 	if (bam_smf_check)
1638a28d77b8Svikram 		return;
1639a28d77b8Svikram 
1640a28d77b8Svikram 	/* Nothing to do if no old stats */
1641a28d77b8Svikram 	if ((nvlp = walk_arg.old_nvlp) == NULL)
1642a28d77b8Svikram 		return;
1643a28d77b8Svikram 
1644a28d77b8Svikram 	for (nvp = nvlist_next_nvpair(nvlp, NULL); nvp;
1645a28d77b8Svikram 	    nvp = nvlist_next_nvpair(nvlp, nvp)) {
1646a28d77b8Svikram 		file = nvpair_name(nvp);
1647a28d77b8Svikram 		if (file == NULL)
1648a28d77b8Svikram 			continue;
1649a28d77b8Svikram 		(void) snprintf(path, sizeof (path), "%s/%s",
1650a28d77b8Svikram 		    root, file);
1651a28d77b8Svikram 		if (stat(path, &sb) == -1) {
1652a28d77b8Svikram 			walk_arg.need_update = 1;
1653a28d77b8Svikram 			if (bam_verbose)
1654a28d77b8Svikram 				bam_print(PARSEABLE_STALE_FILE, path);
1655a28d77b8Svikram 		}
1656a28d77b8Svikram 	}
1657a28d77b8Svikram }
1658a28d77b8Svikram 
16597c478bd9Sstevel@tonic-gate static void
16607c478bd9Sstevel@tonic-gate create_newstat(void)
16617c478bd9Sstevel@tonic-gate {
16627c478bd9Sstevel@tonic-gate 	int error;
16637c478bd9Sstevel@tonic-gate 
16647c478bd9Sstevel@tonic-gate 	error = nvlist_alloc(&walk_arg.new_nvlp, NV_UNIQUE_NAME, 0);
16657c478bd9Sstevel@tonic-gate 	if (error) {
16667c478bd9Sstevel@tonic-gate 		/*
16677c478bd9Sstevel@tonic-gate 		 * Not fatal - we can still create archive
16687c478bd9Sstevel@tonic-gate 		 */
16697c478bd9Sstevel@tonic-gate 		walk_arg.new_nvlp = NULL;
16707c478bd9Sstevel@tonic-gate 		bam_error(NVALLOC_FAIL, strerror(error));
16717c478bd9Sstevel@tonic-gate 	}
16727c478bd9Sstevel@tonic-gate }
16737c478bd9Sstevel@tonic-gate 
16747c478bd9Sstevel@tonic-gate static void
16757c478bd9Sstevel@tonic-gate walk_list(char *root, filelist_t *flistp)
16767c478bd9Sstevel@tonic-gate {
16777c478bd9Sstevel@tonic-gate 	char path[PATH_MAX];
16787c478bd9Sstevel@tonic-gate 	line_t *lp;
16797c478bd9Sstevel@tonic-gate 
16807c478bd9Sstevel@tonic-gate 	for (lp = flistp->head; lp; lp = lp->next) {
16817c478bd9Sstevel@tonic-gate 		(void) snprintf(path, sizeof (path), "%s%s", root, lp->line);
16827c478bd9Sstevel@tonic-gate 		/* XXX shouldn't we use FTW_MOUNT ? */
16837c478bd9Sstevel@tonic-gate 		if (nftw(path, cmpstat, 20, 0) == -1) {
16847c478bd9Sstevel@tonic-gate 			/*
16857c478bd9Sstevel@tonic-gate 			 * Some files may not exist.
16867c478bd9Sstevel@tonic-gate 			 * For example: etc/rtc_config on a x86 diskless system
16877c478bd9Sstevel@tonic-gate 			 * Emit verbose message only
16887c478bd9Sstevel@tonic-gate 			 */
16897c478bd9Sstevel@tonic-gate 			if (bam_verbose)
16907c478bd9Sstevel@tonic-gate 				bam_print(NFTW_FAIL, path, strerror(errno));
16917c478bd9Sstevel@tonic-gate 		}
16927c478bd9Sstevel@tonic-gate 	}
16937c478bd9Sstevel@tonic-gate }
16947c478bd9Sstevel@tonic-gate 
16957c478bd9Sstevel@tonic-gate static void
16967c478bd9Sstevel@tonic-gate savenew(char *root)
16977c478bd9Sstevel@tonic-gate {
16987c478bd9Sstevel@tonic-gate 	char path[PATH_MAX];
16997c478bd9Sstevel@tonic-gate 	char path2[PATH_MAX];
17007c478bd9Sstevel@tonic-gate 	size_t sz;
17017c478bd9Sstevel@tonic-gate 	char *nstat;
17027c478bd9Sstevel@tonic-gate 	int fd, wrote, error;
17037c478bd9Sstevel@tonic-gate 
17047c478bd9Sstevel@tonic-gate 	nstat = NULL;
17057c478bd9Sstevel@tonic-gate 	sz = 0;
17067c478bd9Sstevel@tonic-gate 	error = nvlist_pack(walk_arg.new_nvlp, &nstat, &sz,
17077c478bd9Sstevel@tonic-gate 	    NV_ENCODE_XDR, 0);
17087c478bd9Sstevel@tonic-gate 	if (error) {
17097c478bd9Sstevel@tonic-gate 		bam_error(PACK_FAIL, strerror(error));
17107c478bd9Sstevel@tonic-gate 		return;
17117c478bd9Sstevel@tonic-gate 	}
17127c478bd9Sstevel@tonic-gate 
17137c478bd9Sstevel@tonic-gate 	(void) snprintf(path, sizeof (path), "%s%s", root, FILE_STAT_TMP);
17147c478bd9Sstevel@tonic-gate 	fd = open(path, O_RDWR|O_CREAT|O_TRUNC, FILE_STAT_MODE);
17157c478bd9Sstevel@tonic-gate 	if (fd == -1) {
17167c478bd9Sstevel@tonic-gate 		bam_error(OPEN_FAIL, path, strerror(errno));
17177c478bd9Sstevel@tonic-gate 		free(nstat);
17187c478bd9Sstevel@tonic-gate 		return;
17197c478bd9Sstevel@tonic-gate 	}
17207c478bd9Sstevel@tonic-gate 	wrote = write(fd, nstat, sz);
17217c478bd9Sstevel@tonic-gate 	if (wrote != sz) {
17227c478bd9Sstevel@tonic-gate 		bam_error(WRITE_FAIL, path, strerror(errno));
17237c478bd9Sstevel@tonic-gate 		(void) close(fd);
17247c478bd9Sstevel@tonic-gate 		free(nstat);
17257c478bd9Sstevel@tonic-gate 		return;
17267c478bd9Sstevel@tonic-gate 	}
17277c478bd9Sstevel@tonic-gate 	(void) close(fd);
17287c478bd9Sstevel@tonic-gate 	free(nstat);
17297c478bd9Sstevel@tonic-gate 
17307c478bd9Sstevel@tonic-gate 	(void) snprintf(path2, sizeof (path2), "%s%s", root, FILE_STAT);
17317c478bd9Sstevel@tonic-gate 	if (rename(path, path2) != 0) {
17327c478bd9Sstevel@tonic-gate 		bam_error(RENAME_FAIL, path2, strerror(errno));
17337c478bd9Sstevel@tonic-gate 	}
17347c478bd9Sstevel@tonic-gate }
17357c478bd9Sstevel@tonic-gate 
17367c478bd9Sstevel@tonic-gate static void
17377c478bd9Sstevel@tonic-gate clear_walk_args(void)
17387c478bd9Sstevel@tonic-gate {
17397c478bd9Sstevel@tonic-gate 	if (walk_arg.old_nvlp)
17407c478bd9Sstevel@tonic-gate 		nvlist_free(walk_arg.old_nvlp);
17417c478bd9Sstevel@tonic-gate 	if (walk_arg.new_nvlp)
17427c478bd9Sstevel@tonic-gate 		nvlist_free(walk_arg.new_nvlp);
17437c478bd9Sstevel@tonic-gate 	walk_arg.need_update = 0;
17447c478bd9Sstevel@tonic-gate 	walk_arg.old_nvlp = NULL;
17457c478bd9Sstevel@tonic-gate 	walk_arg.new_nvlp = NULL;
17467c478bd9Sstevel@tonic-gate }
17477c478bd9Sstevel@tonic-gate 
17487c478bd9Sstevel@tonic-gate /*
17497c478bd9Sstevel@tonic-gate  * Returns:
17507c478bd9Sstevel@tonic-gate  *	0 - no update necessary
17517c478bd9Sstevel@tonic-gate  *	1 - update required.
17527c478bd9Sstevel@tonic-gate  *	BAM_ERROR (-1) - An error occurred
17537c478bd9Sstevel@tonic-gate  *
17547c478bd9Sstevel@tonic-gate  * Special handling for check (-n):
17557c478bd9Sstevel@tonic-gate  * ================================
17567c478bd9Sstevel@tonic-gate  * The check (-n) option produces parseable output.
17577c478bd9Sstevel@tonic-gate  * To do this, we suppress all stdout messages unrelated
17587c478bd9Sstevel@tonic-gate  * to out of sync files.
17597c478bd9Sstevel@tonic-gate  * All stderr messages are still printed though.
17607c478bd9Sstevel@tonic-gate  *
17617c478bd9Sstevel@tonic-gate  */
17627c478bd9Sstevel@tonic-gate static int
17637c478bd9Sstevel@tonic-gate update_required(char *root)
17647c478bd9Sstevel@tonic-gate {
17657c478bd9Sstevel@tonic-gate 	struct stat sb;
17667c478bd9Sstevel@tonic-gate 	char path[PATH_MAX];
17677c478bd9Sstevel@tonic-gate 	filelist_t flist;
17687c478bd9Sstevel@tonic-gate 	filelist_t *flistp = &flist;
17697c478bd9Sstevel@tonic-gate 	int need_update;
17707c478bd9Sstevel@tonic-gate 
17717c478bd9Sstevel@tonic-gate 	flistp->head = flistp->tail = NULL;
17727c478bd9Sstevel@tonic-gate 
17737c478bd9Sstevel@tonic-gate 	walk_arg.need_update = 0;
17747c478bd9Sstevel@tonic-gate 
17757c478bd9Sstevel@tonic-gate 	/*
17767c478bd9Sstevel@tonic-gate 	 * Without consulting stat data, check if we need update
17777c478bd9Sstevel@tonic-gate 	 */
17787c478bd9Sstevel@tonic-gate 	check_flags_and_files(root);
17797c478bd9Sstevel@tonic-gate 
17807c478bd9Sstevel@tonic-gate 	/*
17817c478bd9Sstevel@tonic-gate 	 * In certain deployment scenarios, filestat may not
17827c478bd9Sstevel@tonic-gate 	 * exist. Ignore it during boot-archive SMF check.
17837c478bd9Sstevel@tonic-gate 	 */
17847c478bd9Sstevel@tonic-gate 	if (bam_smf_check) {
17857c478bd9Sstevel@tonic-gate 		(void) snprintf(path, sizeof (path), "%s%s", root, FILE_STAT);
17867c478bd9Sstevel@tonic-gate 		if (stat(path, &sb) != 0)
17877c478bd9Sstevel@tonic-gate 			return (0);
17887c478bd9Sstevel@tonic-gate 	}
17897c478bd9Sstevel@tonic-gate 
17907c478bd9Sstevel@tonic-gate 	/*
17917c478bd9Sstevel@tonic-gate 	 * consult stat data only if we haven't made a decision
17927c478bd9Sstevel@tonic-gate 	 * about update. If checking (-n) however, we always
17937c478bd9Sstevel@tonic-gate 	 * need stat data (since we want to compare old and new)
17947c478bd9Sstevel@tonic-gate 	 */
17957c478bd9Sstevel@tonic-gate 	if (!walk_arg.need_update || bam_check)
17967c478bd9Sstevel@tonic-gate 		getoldstat(root);
17977c478bd9Sstevel@tonic-gate 
17987c478bd9Sstevel@tonic-gate 	/*
1799a28d77b8Svikram 	 * Check if the archive contains files that are no longer
1800a28d77b8Svikram 	 * present on the root filesystem.
1801a28d77b8Svikram 	 */
1802a28d77b8Svikram 	if (!walk_arg.need_update || bam_check)
1803a28d77b8Svikram 		check4stale(root);
1804a28d77b8Svikram 
1805a28d77b8Svikram 	/*
18067c478bd9Sstevel@tonic-gate 	 * read list of files
18077c478bd9Sstevel@tonic-gate 	 */
18087c478bd9Sstevel@tonic-gate 	if (read_list(root, flistp) != BAM_SUCCESS) {
18097c478bd9Sstevel@tonic-gate 		clear_walk_args();
18107c478bd9Sstevel@tonic-gate 		return (BAM_ERROR);
18117c478bd9Sstevel@tonic-gate 	}
18127c478bd9Sstevel@tonic-gate 
18137c478bd9Sstevel@tonic-gate 	assert(flistp->head && flistp->tail);
18147c478bd9Sstevel@tonic-gate 
18157c478bd9Sstevel@tonic-gate 	/*
18167c478bd9Sstevel@tonic-gate 	 * At this point either the update is required
18177c478bd9Sstevel@tonic-gate 	 * or the decision is pending. In either case
18187c478bd9Sstevel@tonic-gate 	 * we need to create new stat nvlist
18197c478bd9Sstevel@tonic-gate 	 */
18207c478bd9Sstevel@tonic-gate 	create_newstat();
18217c478bd9Sstevel@tonic-gate 
18227c478bd9Sstevel@tonic-gate 	/*
18237c478bd9Sstevel@tonic-gate 	 * This walk does 2 things:
18247c478bd9Sstevel@tonic-gate 	 *  	- gets new stat data for every file
18257c478bd9Sstevel@tonic-gate 	 *	- (optional) compare old and new stat data
18267c478bd9Sstevel@tonic-gate 	 */
18277c478bd9Sstevel@tonic-gate 	walk_list(root, &flist);
18287c478bd9Sstevel@tonic-gate 
18297c478bd9Sstevel@tonic-gate 	/* done with the file list */
18307c478bd9Sstevel@tonic-gate 	filelist_free(flistp);
18317c478bd9Sstevel@tonic-gate 
18327c478bd9Sstevel@tonic-gate 	/*
18337c478bd9Sstevel@tonic-gate 	 * if we didn't succeed in  creating new stat data above
18347c478bd9Sstevel@tonic-gate 	 * just return result of update check so that archive is built.
18357c478bd9Sstevel@tonic-gate 	 */
18367c478bd9Sstevel@tonic-gate 	if (walk_arg.new_nvlp == NULL) {
18377c478bd9Sstevel@tonic-gate 		bam_error(NO_NEW_STAT);
18387c478bd9Sstevel@tonic-gate 		need_update = walk_arg.need_update;
18397c478bd9Sstevel@tonic-gate 		clear_walk_args();
18407c478bd9Sstevel@tonic-gate 		return (need_update ? 1 : 0);
18417c478bd9Sstevel@tonic-gate 	}
18427c478bd9Sstevel@tonic-gate 
18437c478bd9Sstevel@tonic-gate 
18447c478bd9Sstevel@tonic-gate 	/*
18457c478bd9Sstevel@tonic-gate 	 * If no update required, discard newstat
18467c478bd9Sstevel@tonic-gate 	 */
18477c478bd9Sstevel@tonic-gate 	if (!walk_arg.need_update) {
18487c478bd9Sstevel@tonic-gate 		clear_walk_args();
18497c478bd9Sstevel@tonic-gate 		return (0);
18507c478bd9Sstevel@tonic-gate 	}
18517c478bd9Sstevel@tonic-gate 
18527c478bd9Sstevel@tonic-gate 	/*
18537c478bd9Sstevel@tonic-gate 	 * At this point we need an update - so save new stat data
18547c478bd9Sstevel@tonic-gate 	 * However, if only checking (-n), don't save new stat data.
18557c478bd9Sstevel@tonic-gate 	 */
18567c478bd9Sstevel@tonic-gate 	if (!bam_check)
18577c478bd9Sstevel@tonic-gate 		savenew(root);
18587c478bd9Sstevel@tonic-gate 
18597c478bd9Sstevel@tonic-gate 	clear_walk_args();
18607c478bd9Sstevel@tonic-gate 
18617c478bd9Sstevel@tonic-gate 	return (1);
18627c478bd9Sstevel@tonic-gate }
18637c478bd9Sstevel@tonic-gate 
18647c478bd9Sstevel@tonic-gate static error_t
18657c478bd9Sstevel@tonic-gate create_ramdisk(char *root)
18667c478bd9Sstevel@tonic-gate {
18677c478bd9Sstevel@tonic-gate 	char *cmdline, path[PATH_MAX];
18687c478bd9Sstevel@tonic-gate 	size_t len;
18697c478bd9Sstevel@tonic-gate 	struct stat sb;
18707c478bd9Sstevel@tonic-gate 
18717c478bd9Sstevel@tonic-gate 	/*
18727c478bd9Sstevel@tonic-gate 	 * Setup command args for create_ramdisk.ksh
18737c478bd9Sstevel@tonic-gate 	 */
18747c478bd9Sstevel@tonic-gate 	(void) snprintf(path, sizeof (path), "%s%s", root, CREATE_RAMDISK);
18757c478bd9Sstevel@tonic-gate 	if (stat(path, &sb) != 0) {
18767c478bd9Sstevel@tonic-gate 		bam_error(ARCH_EXEC_MISS, path, strerror(errno));
18777c478bd9Sstevel@tonic-gate 		return (BAM_ERROR);
18787c478bd9Sstevel@tonic-gate 	}
18797c478bd9Sstevel@tonic-gate 
18807c478bd9Sstevel@tonic-gate 	len = strlen(path) + strlen(root) + 10;	/* room for space + -R */
18817c478bd9Sstevel@tonic-gate 	cmdline = s_calloc(1, len);
18827c478bd9Sstevel@tonic-gate 
18837c478bd9Sstevel@tonic-gate 	if (strlen(root) > 1) {
18847c478bd9Sstevel@tonic-gate 		(void) snprintf(cmdline, len, "%s -R %s", path, root);
18857c478bd9Sstevel@tonic-gate 		/* chop off / at the end */
18867c478bd9Sstevel@tonic-gate 		cmdline[strlen(cmdline) - 1] = '\0';
18877c478bd9Sstevel@tonic-gate 	} else
18887c478bd9Sstevel@tonic-gate 		(void) snprintf(cmdline, len, "%s", path);
18897c478bd9Sstevel@tonic-gate 
18907c478bd9Sstevel@tonic-gate 	if (exec_cmd(cmdline, NULL, 0) != 0) {
18917c478bd9Sstevel@tonic-gate 		bam_error(ARCHIVE_FAIL, cmdline);
18927c478bd9Sstevel@tonic-gate 		free(cmdline);
18937c478bd9Sstevel@tonic-gate 		return (BAM_ERROR);
18947c478bd9Sstevel@tonic-gate 	}
18957c478bd9Sstevel@tonic-gate 	free(cmdline);
18967c478bd9Sstevel@tonic-gate 
18977c478bd9Sstevel@tonic-gate 	/*
18987c478bd9Sstevel@tonic-gate 	 * Verify that the archive has been created
18997c478bd9Sstevel@tonic-gate 	 */
1900ae115bc7Smrj 	(void) snprintf(path, sizeof (path), "%s%s", root,
1901ae115bc7Smrj 	    DIRECT_BOOT_ARCHIVE_32);
19027c478bd9Sstevel@tonic-gate 	if (stat(path, &sb) != 0) {
19037c478bd9Sstevel@tonic-gate 		bam_error(ARCHIVE_NOT_CREATED, path);
19047c478bd9Sstevel@tonic-gate 		return (BAM_ERROR);
19057c478bd9Sstevel@tonic-gate 	}
1906ae115bc7Smrj 	if (bam_direct == BAM_DIRECT_DBOOT) {
1907ae115bc7Smrj 		(void) snprintf(path, sizeof (path), "%s%s", root,
1908ae115bc7Smrj 		    DIRECT_BOOT_ARCHIVE_64);
1909ae115bc7Smrj 		if (stat(path, &sb) != 0) {
1910ae115bc7Smrj 			bam_error(ARCHIVE_NOT_CREATED, path);
1911ae115bc7Smrj 			return (BAM_ERROR);
1912ae115bc7Smrj 		}
1913ae115bc7Smrj 	}
19147c478bd9Sstevel@tonic-gate 
19157c478bd9Sstevel@tonic-gate 	return (BAM_SUCCESS);
19167c478bd9Sstevel@tonic-gate }
19177c478bd9Sstevel@tonic-gate 
19187c478bd9Sstevel@tonic-gate /*
19197c478bd9Sstevel@tonic-gate  * Checks if target filesystem is on a ramdisk
19207c478bd9Sstevel@tonic-gate  * 1 - is miniroot
19217c478bd9Sstevel@tonic-gate  * 0 - is not
19227c478bd9Sstevel@tonic-gate  * When in doubt assume it is not a ramdisk.
19237c478bd9Sstevel@tonic-gate  */
19247c478bd9Sstevel@tonic-gate static int
19257c478bd9Sstevel@tonic-gate is_ramdisk(char *root)
19267c478bd9Sstevel@tonic-gate {
19277c478bd9Sstevel@tonic-gate 	struct extmnttab mnt;
19287c478bd9Sstevel@tonic-gate 	FILE *fp;
19297c478bd9Sstevel@tonic-gate 	int found;
1930b610f78eSvikram 	char mntpt[PATH_MAX];
1931b610f78eSvikram 	char *cp;
19327c478bd9Sstevel@tonic-gate 
19337c478bd9Sstevel@tonic-gate 	/*
19347c478bd9Sstevel@tonic-gate 	 * There are 3 situations where creating archive is
19357c478bd9Sstevel@tonic-gate 	 * of dubious value:
1936b610f78eSvikram 	 *	- create boot_archive on a lofi-mounted boot_archive
19377c478bd9Sstevel@tonic-gate 	 *	- create it on a ramdisk which is the root filesystem
19387c478bd9Sstevel@tonic-gate 	 *	- create it on a ramdisk mounted somewhere else
19397c478bd9Sstevel@tonic-gate 	 * The first is not easy to detect and checking for it is not
19407c478bd9Sstevel@tonic-gate 	 * worth it.
19417c478bd9Sstevel@tonic-gate 	 * The other two conditions are handled here
19427c478bd9Sstevel@tonic-gate 	 */
19437c478bd9Sstevel@tonic-gate 
19447c478bd9Sstevel@tonic-gate 	fp = fopen(MNTTAB, "r");
19457c478bd9Sstevel@tonic-gate 	if (fp == NULL) {
19467c478bd9Sstevel@tonic-gate 		bam_error(OPEN_FAIL, MNTTAB, strerror(errno));
19477c478bd9Sstevel@tonic-gate 		return (0);
19487c478bd9Sstevel@tonic-gate 	}
19497c478bd9Sstevel@tonic-gate 
19507c478bd9Sstevel@tonic-gate 	resetmnttab(fp);
19517c478bd9Sstevel@tonic-gate 
1952b610f78eSvikram 	/*
1953b610f78eSvikram 	 * Remove any trailing / from the mount point
1954b610f78eSvikram 	 */
1955b610f78eSvikram 	(void) strlcpy(mntpt, root, sizeof (mntpt));
1956b610f78eSvikram 	if (strcmp(root, "/") != 0) {
1957b610f78eSvikram 		cp = mntpt + strlen(mntpt) - 1;
1958b610f78eSvikram 		if (*cp == '/')
1959b610f78eSvikram 			*cp = '\0';
1960b610f78eSvikram 	}
19617c478bd9Sstevel@tonic-gate 	found = 0;
19627c478bd9Sstevel@tonic-gate 	while (getextmntent(fp, &mnt, sizeof (mnt)) == 0) {
1963b610f78eSvikram 		if (strcmp(mnt.mnt_mountp, mntpt) == 0) {
19647c478bd9Sstevel@tonic-gate 			found = 1;
19657c478bd9Sstevel@tonic-gate 			break;
19667c478bd9Sstevel@tonic-gate 		}
19677c478bd9Sstevel@tonic-gate 	}
19687c478bd9Sstevel@tonic-gate 
19697c478bd9Sstevel@tonic-gate 	if (!found) {
19707c478bd9Sstevel@tonic-gate 		if (bam_verbose)
1971b610f78eSvikram 			bam_error(NOT_IN_MNTTAB, mntpt);
19727c478bd9Sstevel@tonic-gate 		(void) fclose(fp);
19737c478bd9Sstevel@tonic-gate 		return (0);
19747c478bd9Sstevel@tonic-gate 	}
19757c478bd9Sstevel@tonic-gate 
19767c478bd9Sstevel@tonic-gate 	if (strstr(mnt.mnt_special, RAMDISK_SPECIAL) != NULL) {
19777c478bd9Sstevel@tonic-gate 		if (bam_verbose)
19787c478bd9Sstevel@tonic-gate 			bam_error(IS_RAMDISK, bam_root);
19797c478bd9Sstevel@tonic-gate 		(void) fclose(fp);
19807c478bd9Sstevel@tonic-gate 		return (1);
19817c478bd9Sstevel@tonic-gate 	}
19827c478bd9Sstevel@tonic-gate 
19837c478bd9Sstevel@tonic-gate 	(void) fclose(fp);
19847c478bd9Sstevel@tonic-gate 
19857c478bd9Sstevel@tonic-gate 	return (0);
19867c478bd9Sstevel@tonic-gate }
19877c478bd9Sstevel@tonic-gate 
19887c478bd9Sstevel@tonic-gate static int
19897c478bd9Sstevel@tonic-gate is_newboot(char *root)
19907c478bd9Sstevel@tonic-gate {
19917c478bd9Sstevel@tonic-gate 	char path[PATH_MAX];
19927c478bd9Sstevel@tonic-gate 	struct stat sb;
19937c478bd9Sstevel@tonic-gate 
19947c478bd9Sstevel@tonic-gate 	/*
19957c478bd9Sstevel@tonic-gate 	 * We can't boot without MULTI_BOOT
19967c478bd9Sstevel@tonic-gate 	 */
19977c478bd9Sstevel@tonic-gate 	(void) snprintf(path, sizeof (path), "%s%s", root, MULTI_BOOT);
19987c478bd9Sstevel@tonic-gate 	if (stat(path, &sb) == -1) {
19997c478bd9Sstevel@tonic-gate 		if (bam_verbose)
20007c478bd9Sstevel@tonic-gate 			bam_print(FILE_MISS, path);
20017c478bd9Sstevel@tonic-gate 		return (0);
20027c478bd9Sstevel@tonic-gate 	}
20037c478bd9Sstevel@tonic-gate 
20047c478bd9Sstevel@tonic-gate 	/*
20057c478bd9Sstevel@tonic-gate 	 * We can't generate archive without GRUB_DIR
20067c478bd9Sstevel@tonic-gate 	 */
20077c478bd9Sstevel@tonic-gate 	(void) snprintf(path, sizeof (path), "%s%s", root, GRUB_DIR);
20087c478bd9Sstevel@tonic-gate 	if (stat(path, &sb) == -1) {
20097c478bd9Sstevel@tonic-gate 		if (bam_verbose)
20107c478bd9Sstevel@tonic-gate 			bam_print(DIR_MISS, path);
20117c478bd9Sstevel@tonic-gate 		return (0);
20127c478bd9Sstevel@tonic-gate 	}
20137c478bd9Sstevel@tonic-gate 
20147c478bd9Sstevel@tonic-gate 	return (1);
20157c478bd9Sstevel@tonic-gate }
20167c478bd9Sstevel@tonic-gate 
20177c478bd9Sstevel@tonic-gate static int
20187c478bd9Sstevel@tonic-gate is_readonly(char *root)
20197c478bd9Sstevel@tonic-gate {
20207c478bd9Sstevel@tonic-gate 	struct statvfs vfs;
20217c478bd9Sstevel@tonic-gate 
20227c478bd9Sstevel@tonic-gate 	/*
20237c478bd9Sstevel@tonic-gate 	 * Check for RDONLY filesystem
20247c478bd9Sstevel@tonic-gate 	 * When in doubt assume it is not readonly
20257c478bd9Sstevel@tonic-gate 	 */
20267c478bd9Sstevel@tonic-gate 	if (statvfs(root, &vfs) != 0) {
20277c478bd9Sstevel@tonic-gate 		if (bam_verbose)
20287c478bd9Sstevel@tonic-gate 			bam_error(STATVFS_FAIL, root, strerror(errno));
20297c478bd9Sstevel@tonic-gate 		return (0);
20307c478bd9Sstevel@tonic-gate 	}
20317c478bd9Sstevel@tonic-gate 
20327c478bd9Sstevel@tonic-gate 	if (vfs.f_flag & ST_RDONLY) {
20337c478bd9Sstevel@tonic-gate 		return (1);
20347c478bd9Sstevel@tonic-gate 	}
20357c478bd9Sstevel@tonic-gate 
20367c478bd9Sstevel@tonic-gate 	return (0);
20377c478bd9Sstevel@tonic-gate }
20387c478bd9Sstevel@tonic-gate 
20397c478bd9Sstevel@tonic-gate static error_t
20407c478bd9Sstevel@tonic-gate update_archive(char *root, char *opt)
20417c478bd9Sstevel@tonic-gate {
20427c478bd9Sstevel@tonic-gate 	error_t ret;
20437c478bd9Sstevel@tonic-gate 
20447c478bd9Sstevel@tonic-gate 	assert(root);
20457c478bd9Sstevel@tonic-gate 	assert(opt == NULL);
20467c478bd9Sstevel@tonic-gate 
20477c478bd9Sstevel@tonic-gate 	/*
2048b610f78eSvikram 	 * root must belong to a GRUB boot OS,
20497c478bd9Sstevel@tonic-gate 	 * don't care on sparc except for diskless clients
20507c478bd9Sstevel@tonic-gate 	 */
20517c478bd9Sstevel@tonic-gate 	if (!is_newboot(root)) {
2052b610f78eSvikram 		/*
2053b610f78eSvikram 		 * Emit message only if not in context of update_all.
2054b610f78eSvikram 		 * If in update_all, emit only if verbose flag is set.
2055b610f78eSvikram 		 */
2056b610f78eSvikram 		if (!bam_update_all || bam_verbose)
2057b610f78eSvikram 			bam_print(NOT_GRUB_BOOT, root);
20587c478bd9Sstevel@tonic-gate 		return (BAM_SUCCESS);
20597c478bd9Sstevel@tonic-gate 	}
20607c478bd9Sstevel@tonic-gate 
20617c478bd9Sstevel@tonic-gate 	/*
20628c1b6884Sszhou 	 * If smf check is requested when / is writable (can happen
20638c1b6884Sszhou 	 * on first reboot following an upgrade because service
20648c1b6884Sszhou 	 * dependency is messed up), skip the check.
20658c1b6884Sszhou 	 */
20668c1b6884Sszhou 	if (bam_smf_check && !bam_root_readonly)
20678c1b6884Sszhou 		return (BAM_SUCCESS);
20688c1b6884Sszhou 
20698c1b6884Sszhou 	/*
20708c1b6884Sszhou 	 * root must be writable. This check applies to alternate
20718c1b6884Sszhou 	 * root (-R option); bam_root_readonly applies to '/' only.
20727c478bd9Sstevel@tonic-gate 	 * Note: statvfs() does not always report the truth
20737c478bd9Sstevel@tonic-gate 	 */
207461cb17bdSsetje 	if (!bam_smf_check && !bam_check && is_readonly(root)) {
20758c1b6884Sszhou 		if (bam_verbose)
20767c478bd9Sstevel@tonic-gate 			bam_print(RDONLY_FS, root);
20777c478bd9Sstevel@tonic-gate 		return (BAM_SUCCESS);
20787c478bd9Sstevel@tonic-gate 	}
20797c478bd9Sstevel@tonic-gate 
20807c478bd9Sstevel@tonic-gate 	/*
20817c478bd9Sstevel@tonic-gate 	 * Don't generate archive on ramdisk
20827c478bd9Sstevel@tonic-gate 	 */
20837c478bd9Sstevel@tonic-gate 	if (is_ramdisk(root)) {
20847c478bd9Sstevel@tonic-gate 		if (bam_verbose)
20857c478bd9Sstevel@tonic-gate 			bam_print(SKIP_RAMDISK);
20867c478bd9Sstevel@tonic-gate 		return (BAM_SUCCESS);
20877c478bd9Sstevel@tonic-gate 	}
20887c478bd9Sstevel@tonic-gate 
20897c478bd9Sstevel@tonic-gate 	/*
20907c478bd9Sstevel@tonic-gate 	 * Now check if updated is really needed
20917c478bd9Sstevel@tonic-gate 	 */
20927c478bd9Sstevel@tonic-gate 	ret = update_required(root);
20937c478bd9Sstevel@tonic-gate 
20947c478bd9Sstevel@tonic-gate 	/*
20957c478bd9Sstevel@tonic-gate 	 * The check command (-n) is *not* a dry run
20967c478bd9Sstevel@tonic-gate 	 * It only checks if the archive is in sync.
20977c478bd9Sstevel@tonic-gate 	 */
20987c478bd9Sstevel@tonic-gate 	if (bam_check) {
20997c478bd9Sstevel@tonic-gate 		bam_exit((ret != 0) ? 1 : 0);
21007c478bd9Sstevel@tonic-gate 	}
21017c478bd9Sstevel@tonic-gate 
21027c478bd9Sstevel@tonic-gate 	if (ret == 1) {
21037c478bd9Sstevel@tonic-gate 		/* create the ramdisk */
21047c478bd9Sstevel@tonic-gate 		ret = create_ramdisk(root);
21057c478bd9Sstevel@tonic-gate 	}
21067c478bd9Sstevel@tonic-gate 	return (ret);
21077c478bd9Sstevel@tonic-gate }
21087c478bd9Sstevel@tonic-gate 
2109b610f78eSvikram static void
2110fbac2b2bSvikram update_fdisk(void)
2111fbac2b2bSvikram {
2112fbac2b2bSvikram 	struct stat sb;
2113fbac2b2bSvikram 	char cmd[PATH_MAX];
2114fbac2b2bSvikram 	int ret1, ret2;
2115fbac2b2bSvikram 
2116fbac2b2bSvikram 	assert(stat(GRUB_fdisk, &sb) == 0);
2117fbac2b2bSvikram 	assert(stat(GRUB_fdisk_target, &sb) == 0);
2118fbac2b2bSvikram 
2119fbac2b2bSvikram 	(void) snprintf(cmd, sizeof (cmd), "/sbin/fdisk -F %s `/bin/cat %s`",
2120fbac2b2bSvikram 	    GRUB_fdisk, GRUB_fdisk_target);
2121fbac2b2bSvikram 
2122fbac2b2bSvikram 	bam_print(UPDATING_FDISK);
2123fbac2b2bSvikram 	if (exec_cmd(cmd, NULL, 0) != 0) {
2124fbac2b2bSvikram 		bam_error(FDISK_UPDATE_FAILED);
2125fbac2b2bSvikram 	}
2126fbac2b2bSvikram 
2127fbac2b2bSvikram 	/*
2128fbac2b2bSvikram 	 * We are done, remove the files.
2129fbac2b2bSvikram 	 */
2130fbac2b2bSvikram 	ret1 = unlink(GRUB_fdisk);
2131fbac2b2bSvikram 	ret2 = unlink(GRUB_fdisk_target);
2132fbac2b2bSvikram 	if (ret1 != 0 || ret2 != 0) {
2133fbac2b2bSvikram 		bam_error(FILE_REMOVE_FAILED, GRUB_fdisk, GRUB_fdisk_target);
2134fbac2b2bSvikram 	}
2135fbac2b2bSvikram }
2136fbac2b2bSvikram 
2137fbac2b2bSvikram static void
2138b610f78eSvikram restore_grub_slice(void)
2139b610f78eSvikram {
2140b610f78eSvikram 	struct stat sb;
2141b610f78eSvikram 	char *mntpt, *physlice;
2142b610f78eSvikram 	int mnted;	/* set if we did a mount */
2143b610f78eSvikram 	char menupath[PATH_MAX], cmd[PATH_MAX];
2144b610f78eSvikram 
2145b610f78eSvikram 	if (stat(GRUB_slice, &sb) != 0) {
2146b610f78eSvikram 		bam_error(MISSING_SLICE_FILE, GRUB_slice, strerror(errno));
2147b610f78eSvikram 		return;
2148b610f78eSvikram 	}
2149b610f78eSvikram 
2150b610f78eSvikram 	/*
2151b610f78eSvikram 	 * If we are doing an luactivate, don't attempt to restore GRUB or else
2152b610f78eSvikram 	 * we may not be able to get to DCA boot environments. Let luactivate
2153b610f78eSvikram 	 * handle GRUB/DCA installation
2154b610f78eSvikram 	 */
2155b610f78eSvikram 	if (stat(LU_ACTIVATE_FILE, &sb) == 0) {
2156b610f78eSvikram 		return;
2157b610f78eSvikram 	}
2158b610f78eSvikram 
2159b610f78eSvikram 	mnted = 0;
2160b610f78eSvikram 	physlice = NULL;
216140541d5dSvikram 	mntpt = mount_grub_slice(&mnted, &physlice, NULL, NULL);
2162b610f78eSvikram 	if (mntpt == NULL) {
2163b610f78eSvikram 		bam_error(CANNOT_RESTORE_GRUB_SLICE);
2164b610f78eSvikram 		return;
2165b610f78eSvikram 	}
2166b610f78eSvikram 
2167b610f78eSvikram 	(void) snprintf(menupath, sizeof (menupath), "%s%s", mntpt, GRUB_MENU);
2168b610f78eSvikram 	if (stat(menupath, &sb) == 0) {
216940541d5dSvikram 		umount_grub_slice(mnted, mntpt, physlice, NULL, NULL);
2170b610f78eSvikram 		return;
2171b610f78eSvikram 	}
2172b610f78eSvikram 
2173b610f78eSvikram 	/*
2174b610f78eSvikram 	 * The menu is missing - we need to do a restore
2175b610f78eSvikram 	 */
2176b610f78eSvikram 	bam_print(RESTORING_GRUB);
2177b610f78eSvikram 
2178b610f78eSvikram 	(void) snprintf(cmd, sizeof (cmd), "%s %s %s %s",
2179b610f78eSvikram 	    INSTALLGRUB, STAGE1, STAGE2, physlice);
2180b610f78eSvikram 
2181b610f78eSvikram 	if (exec_cmd(cmd, NULL, 0) != 0) {
2182b610f78eSvikram 		bam_error(RESTORE_GRUB_FAILED);
218340541d5dSvikram 		umount_grub_slice(mnted, mntpt, physlice, NULL, NULL);
2184b610f78eSvikram 		return;
2185b610f78eSvikram 	}
2186b610f78eSvikram 
2187b610f78eSvikram 	if (stat(GRUB_backup_menu, &sb) != 0) {
2188b610f78eSvikram 		bam_error(MISSING_BACKUP_MENU,
2189b610f78eSvikram 		    GRUB_backup_menu, strerror(errno));
219040541d5dSvikram 		umount_grub_slice(mnted, mntpt, physlice, NULL, NULL);
2191b610f78eSvikram 		return;
2192b610f78eSvikram 	}
2193b610f78eSvikram 
2194b610f78eSvikram 	(void) snprintf(cmd, sizeof (cmd), "/bin/cp %s %s",
2195b610f78eSvikram 	    GRUB_backup_menu, menupath);
2196b610f78eSvikram 
2197b610f78eSvikram 	if (exec_cmd(cmd, NULL, 0) != 0) {
2198b610f78eSvikram 		bam_error(RESTORE_MENU_FAILED, menupath);
219940541d5dSvikram 		umount_grub_slice(mnted, mntpt, physlice, NULL, NULL);
2200b610f78eSvikram 		return;
2201b610f78eSvikram 	}
2202b610f78eSvikram 
2203b610f78eSvikram 	/* Success */
220440541d5dSvikram 	umount_grub_slice(mnted, mntpt, physlice, NULL, NULL);
2205b610f78eSvikram }
2206b610f78eSvikram 
22077c478bd9Sstevel@tonic-gate static error_t
22087c478bd9Sstevel@tonic-gate update_all(char *root, char *opt)
22097c478bd9Sstevel@tonic-gate {
22107c478bd9Sstevel@tonic-gate 	struct extmnttab mnt;
22117c478bd9Sstevel@tonic-gate 	struct stat sb;
22127c478bd9Sstevel@tonic-gate 	FILE *fp;
22137c478bd9Sstevel@tonic-gate 	char multibt[PATH_MAX];
22147c478bd9Sstevel@tonic-gate 	error_t ret = BAM_SUCCESS;
2215fbac2b2bSvikram 	int ret1, ret2;
22167c478bd9Sstevel@tonic-gate 
221740541d5dSvikram 	assert(root);
22187c478bd9Sstevel@tonic-gate 	assert(opt == NULL);
22197c478bd9Sstevel@tonic-gate 
222040541d5dSvikram 	if (bam_rootlen != 1 || *root != '/') {
222140541d5dSvikram 		elide_trailing_slash(root, multibt, sizeof (multibt));
222240541d5dSvikram 		bam_error(ALT_ROOT_INVALID, multibt);
222340541d5dSvikram 		return (BAM_ERROR);
222440541d5dSvikram 	}
222540541d5dSvikram 
22267c478bd9Sstevel@tonic-gate 	/*
222798892a30Snadkarni 	 * Check to see if we are in the midst of safemode patching
222898892a30Snadkarni 	 * If so skip building the archive for /. Instead build it
222998892a30Snadkarni 	 * against the latest bits obtained by creating a fresh lofs
223098892a30Snadkarni 	 * mount of root.
223198892a30Snadkarni 	 */
223298892a30Snadkarni 	if (stat(LOFS_PATCH_FILE, &sb) == 0)  {
223398892a30Snadkarni 		if (mkdir(LOFS_PATCH_MNT, 0755) == -1 &&
223498892a30Snadkarni 		    errno != EEXIST) {
223598892a30Snadkarni 			bam_error(MKDIR_FAILED, "%s", LOFS_PATCH_MNT,
223698892a30Snadkarni 			    strerror(errno));
223798892a30Snadkarni 			ret = BAM_ERROR;
223898892a30Snadkarni 			goto out;
223998892a30Snadkarni 		}
224098892a30Snadkarni 		(void) snprintf(multibt, sizeof (multibt),
224198892a30Snadkarni 		    "/sbin/mount -F lofs -o nosub /  %s", LOFS_PATCH_MNT);
224298892a30Snadkarni 		if (exec_cmd(multibt, NULL, 0) != 0) {
224398892a30Snadkarni 			bam_error(MOUNT_FAILED, LOFS_PATCH_MNT, "lofs");
224498892a30Snadkarni 			ret = BAM_ERROR;
224598892a30Snadkarni 		}
224698892a30Snadkarni 		if (ret != BAM_ERROR) {
224798892a30Snadkarni 			(void) snprintf(rootbuf, sizeof (rootbuf), "%s/",
224898892a30Snadkarni 			    LOFS_PATCH_MNT);
224998892a30Snadkarni 			bam_rootlen = strlen(rootbuf);
225098892a30Snadkarni 			if (update_archive(rootbuf, opt) != BAM_SUCCESS)
225198892a30Snadkarni 				ret = BAM_ERROR;
225295b1e0e9Snadkarni 			/*
225395b1e0e9Snadkarni 			 * unmount the lofs mount since there could be
225495b1e0e9Snadkarni 			 * multiple invocations of bootadm -a update_all
225595b1e0e9Snadkarni 			 */
225695b1e0e9Snadkarni 			(void) snprintf(multibt, sizeof (multibt),
225795b1e0e9Snadkarni 			    "/sbin/umount %s", LOFS_PATCH_MNT);
225895b1e0e9Snadkarni 			if (exec_cmd(multibt, NULL, 0) != 0) {
225995b1e0e9Snadkarni 				bam_error(UMOUNT_FAILED, LOFS_PATCH_MNT);
226095b1e0e9Snadkarni 				ret = BAM_ERROR;
226195b1e0e9Snadkarni 			}
226298892a30Snadkarni 		}
226398892a30Snadkarni 	} else {
226498892a30Snadkarni 		/*
22657c478bd9Sstevel@tonic-gate 		 * First update archive for current root
22667c478bd9Sstevel@tonic-gate 		 */
22677c478bd9Sstevel@tonic-gate 		if (update_archive(root, opt) != BAM_SUCCESS)
22687c478bd9Sstevel@tonic-gate 			ret = BAM_ERROR;
226998892a30Snadkarni 	}
227098892a30Snadkarni 
227198892a30Snadkarni 	if (ret == BAM_ERROR)
227298892a30Snadkarni 		goto out;
22737c478bd9Sstevel@tonic-gate 
22747c478bd9Sstevel@tonic-gate 	/*
22757c478bd9Sstevel@tonic-gate 	 * Now walk the mount table, performing archive update
22767c478bd9Sstevel@tonic-gate 	 * for all mounted Newboot root filesystems
22777c478bd9Sstevel@tonic-gate 	 */
22787c478bd9Sstevel@tonic-gate 	fp = fopen(MNTTAB, "r");
22797c478bd9Sstevel@tonic-gate 	if (fp == NULL) {
22807c478bd9Sstevel@tonic-gate 		bam_error(OPEN_FAIL, MNTTAB, strerror(errno));
2281b610f78eSvikram 		ret = BAM_ERROR;
2282b610f78eSvikram 		goto out;
22837c478bd9Sstevel@tonic-gate 	}
22847c478bd9Sstevel@tonic-gate 
22857c478bd9Sstevel@tonic-gate 	resetmnttab(fp);
22867c478bd9Sstevel@tonic-gate 
22877c478bd9Sstevel@tonic-gate 	while (getextmntent(fp, &mnt, sizeof (mnt)) == 0) {
22887c478bd9Sstevel@tonic-gate 		if (mnt.mnt_special == NULL)
22897c478bd9Sstevel@tonic-gate 			continue;
22907c478bd9Sstevel@tonic-gate 		if (strncmp(mnt.mnt_special, "/dev/", strlen("/dev/")) != 0)
22917c478bd9Sstevel@tonic-gate 			continue;
22927c478bd9Sstevel@tonic-gate 		if (strcmp(mnt.mnt_mountp, "/") == 0)
22937c478bd9Sstevel@tonic-gate 			continue;
22947c478bd9Sstevel@tonic-gate 
22957c478bd9Sstevel@tonic-gate 		(void) snprintf(multibt, sizeof (multibt), "%s%s",
22967c478bd9Sstevel@tonic-gate 		    mnt.mnt_mountp, MULTI_BOOT);
22977c478bd9Sstevel@tonic-gate 
22987c478bd9Sstevel@tonic-gate 		if (stat(multibt, &sb) == -1)
22997c478bd9Sstevel@tonic-gate 			continue;
23007c478bd9Sstevel@tonic-gate 
23017c478bd9Sstevel@tonic-gate 		/*
23027c478bd9Sstevel@tonic-gate 		 * We put a trailing slash to be consistent with root = "/"
23037c478bd9Sstevel@tonic-gate 		 * case, such that we don't have to print // in some cases.
23047c478bd9Sstevel@tonic-gate 		 */
23057c478bd9Sstevel@tonic-gate 		(void) snprintf(rootbuf, sizeof (rootbuf), "%s/",
23067c478bd9Sstevel@tonic-gate 		    mnt.mnt_mountp);
23077c478bd9Sstevel@tonic-gate 		bam_rootlen = strlen(rootbuf);
2308ae115bc7Smrj 
2309ae115bc7Smrj 		/*
2310ae115bc7Smrj 		 * It's possible that other mounts may be an alternate boot
2311ae115bc7Smrj 		 * architecture, so check it again.
2312ae115bc7Smrj 		 */
2313ae115bc7Smrj 		if ((dboot_or_multiboot(rootbuf) != BAM_SUCCESS) ||
2314ae115bc7Smrj 		    (update_archive(rootbuf, opt) != BAM_SUCCESS))
23157c478bd9Sstevel@tonic-gate 			ret = BAM_ERROR;
23167c478bd9Sstevel@tonic-gate 	}
23177c478bd9Sstevel@tonic-gate 
23187c478bd9Sstevel@tonic-gate 	(void) fclose(fp);
23197c478bd9Sstevel@tonic-gate 
2320b610f78eSvikram out:
2321b610f78eSvikram 	if (stat(GRUB_slice, &sb) == 0) {
2322b610f78eSvikram 		restore_grub_slice();
2323b610f78eSvikram 	}
2324b610f78eSvikram 
2325fbac2b2bSvikram 	/*
2326fbac2b2bSvikram 	 * Update fdisk table as we go down. Updating it when
2327fbac2b2bSvikram 	 * the system is running will confuse biosdev.
2328fbac2b2bSvikram 	 */
2329fbac2b2bSvikram 	ret1 = stat(GRUB_fdisk, &sb);
2330fbac2b2bSvikram 	ret2 = stat(GRUB_fdisk_target, &sb);
2331fbac2b2bSvikram 	if ((ret1 == 0) && (ret2 == 0)) {
2332fbac2b2bSvikram 		update_fdisk();
2333fbac2b2bSvikram 	} else if ((ret1 == 0) ^ (ret2 == 0)) {
2334fbac2b2bSvikram 		/*
2335fbac2b2bSvikram 		 * It is an error for one file to be
2336fbac2b2bSvikram 		 * present and the other absent.
2337fbac2b2bSvikram 		 * It is normal for both files to be
2338fbac2b2bSvikram 		 * absent - it indicates that no fdisk
2339fbac2b2bSvikram 		 * update is required.
2340fbac2b2bSvikram 		 */
2341fbac2b2bSvikram 		bam_error(MISSING_FDISK_FILE,
2342fbac2b2bSvikram 		    ret1 ? GRUB_fdisk : GRUB_fdisk_target);
2343fbac2b2bSvikram 		ret = BAM_ERROR;
2344fbac2b2bSvikram 	}
2345fbac2b2bSvikram 
23467c478bd9Sstevel@tonic-gate 	return (ret);
23477c478bd9Sstevel@tonic-gate }
23487c478bd9Sstevel@tonic-gate 
23497c478bd9Sstevel@tonic-gate static void
23507c478bd9Sstevel@tonic-gate append_line(menu_t *mp, line_t *lp)
23517c478bd9Sstevel@tonic-gate {
23527c478bd9Sstevel@tonic-gate 	if (mp->start == NULL) {
23537c478bd9Sstevel@tonic-gate 		mp->start = lp;
23547c478bd9Sstevel@tonic-gate 	} else {
23557c478bd9Sstevel@tonic-gate 		mp->end->next = lp;
23568c1b6884Sszhou 		lp->prev = mp->end;
23577c478bd9Sstevel@tonic-gate 	}
23587c478bd9Sstevel@tonic-gate 	mp->end = lp;
23597c478bd9Sstevel@tonic-gate }
23607c478bd9Sstevel@tonic-gate 
23618c1b6884Sszhou static void
23628c1b6884Sszhou unlink_line(menu_t *mp, line_t *lp)
23638c1b6884Sszhou {
23648c1b6884Sszhou 	/* unlink from list */
23658c1b6884Sszhou 	if (lp->prev)
23668c1b6884Sszhou 		lp->prev->next = lp->next;
23678c1b6884Sszhou 	else
23688c1b6884Sszhou 		mp->start = lp->next;
23698c1b6884Sszhou 	if (lp->next)
23708c1b6884Sszhou 		lp->next->prev = lp->prev;
23718c1b6884Sszhou 	else
23728c1b6884Sszhou 		mp->end = lp->prev;
23738c1b6884Sszhou }
23748c1b6884Sszhou 
23758c1b6884Sszhou static entry_t *
23768c1b6884Sszhou boot_entry_new(menu_t *mp, line_t *start, line_t *end)
23778c1b6884Sszhou {
23788c1b6884Sszhou 	entry_t *ent, *prev;
23798c1b6884Sszhou 
23808c1b6884Sszhou 	ent = s_calloc(1, sizeof (entry_t));
23818c1b6884Sszhou 	ent->start = start;
23828c1b6884Sszhou 	ent->end = end;
23838c1b6884Sszhou 
23848c1b6884Sszhou 	if (mp->entries == NULL) {
23858c1b6884Sszhou 		mp->entries = ent;
23868c1b6884Sszhou 		return (ent);
23878c1b6884Sszhou 	}
23888c1b6884Sszhou 
23898c1b6884Sszhou 	prev = mp->entries;
23908c1b6884Sszhou 	while (prev->next)
23918c1b6884Sszhou 		prev = prev-> next;
23928c1b6884Sszhou 	prev->next = ent;
23938c1b6884Sszhou 	ent->prev = prev;
23948c1b6884Sszhou 	return (ent);
23958c1b6884Sszhou }
23968c1b6884Sszhou 
23978c1b6884Sszhou static void
23988c1b6884Sszhou boot_entry_addline(entry_t *ent, line_t *lp)
23998c1b6884Sszhou {
24008c1b6884Sszhou 	if (ent)
24018c1b6884Sszhou 		ent->end = lp;
24028c1b6884Sszhou }
24038c1b6884Sszhou 
24047c478bd9Sstevel@tonic-gate /*
24057c478bd9Sstevel@tonic-gate  * A line in menu.lst looks like
24067c478bd9Sstevel@tonic-gate  * [ ]*<cmd>[ \t=]*<arg>*
24077c478bd9Sstevel@tonic-gate  */
24087c478bd9Sstevel@tonic-gate static void
24097c478bd9Sstevel@tonic-gate line_parser(menu_t *mp, char *str, int *lineNum, int *entryNum)
24107c478bd9Sstevel@tonic-gate {
24117c478bd9Sstevel@tonic-gate 	/*
24127c478bd9Sstevel@tonic-gate 	 * save state across calls. This is so that
24137c478bd9Sstevel@tonic-gate 	 * header gets the right entry# after title has
24147c478bd9Sstevel@tonic-gate 	 * been processed
24157c478bd9Sstevel@tonic-gate 	 */
24168c1b6884Sszhou 	static line_t *prev = NULL;
24178c1b6884Sszhou 	static entry_t *curr_ent = NULL;
2418ae115bc7Smrj 	static int in_liveupgrade = 0;
24197c478bd9Sstevel@tonic-gate 
24207c478bd9Sstevel@tonic-gate 	line_t	*lp;
24217c478bd9Sstevel@tonic-gate 	char *cmd, *sep, *arg;
24227c478bd9Sstevel@tonic-gate 	char save, *cp, *line;
24237c478bd9Sstevel@tonic-gate 	menu_flag_t flag = BAM_INVALID;
24247c478bd9Sstevel@tonic-gate 
24257c478bd9Sstevel@tonic-gate 	if (str == NULL) {
24267c478bd9Sstevel@tonic-gate 		return;
24277c478bd9Sstevel@tonic-gate 	}
24287c478bd9Sstevel@tonic-gate 
24297c478bd9Sstevel@tonic-gate 	/*
24307c478bd9Sstevel@tonic-gate 	 * First save a copy of the entire line.
24317c478bd9Sstevel@tonic-gate 	 * We use this later to set the line field.
24327c478bd9Sstevel@tonic-gate 	 */
24337c478bd9Sstevel@tonic-gate 	line = s_strdup(str);
24347c478bd9Sstevel@tonic-gate 
24357c478bd9Sstevel@tonic-gate 	/* Eat up leading whitespace */
24367c478bd9Sstevel@tonic-gate 	while (*str == ' ' || *str == '\t')
24377c478bd9Sstevel@tonic-gate 		str++;
24387c478bd9Sstevel@tonic-gate 
24397c478bd9Sstevel@tonic-gate 	if (*str == '#') {		/* comment */
24407c478bd9Sstevel@tonic-gate 		cmd = s_strdup("#");
24417c478bd9Sstevel@tonic-gate 		sep = NULL;
24427c478bd9Sstevel@tonic-gate 		arg = s_strdup(str + 1);
24437c478bd9Sstevel@tonic-gate 		flag = BAM_COMMENT;
2444ae115bc7Smrj 		if (strstr(arg, BAM_LU_HDR) != NULL) {
2445ae115bc7Smrj 			in_liveupgrade = 1;
2446ae115bc7Smrj 		} else if (strstr(arg, BAM_LU_FTR) != NULL) {
2447ae115bc7Smrj 			in_liveupgrade = 0;
2448ae115bc7Smrj 		}
24497c478bd9Sstevel@tonic-gate 	} else if (*str == '\0') {	/* blank line */
24507c478bd9Sstevel@tonic-gate 		cmd = sep = arg = NULL;
24517c478bd9Sstevel@tonic-gate 		flag = BAM_EMPTY;
24527c478bd9Sstevel@tonic-gate 	} else {
24537c478bd9Sstevel@tonic-gate 		/*
24547c478bd9Sstevel@tonic-gate 		 * '=' is not a documented separator in grub syntax.
24557c478bd9Sstevel@tonic-gate 		 * However various development bits use '=' as a
24567c478bd9Sstevel@tonic-gate 		 * separator. In addition, external users also
24577c478bd9Sstevel@tonic-gate 		 * use = as a separator. So we will allow that usage.
24587c478bd9Sstevel@tonic-gate 		 */
24597c478bd9Sstevel@tonic-gate 		cp = str;
24607c478bd9Sstevel@tonic-gate 		while (*str != ' ' && *str != '\t' && *str != '=') {
24617c478bd9Sstevel@tonic-gate 			if (*str == '\0') {
24627c478bd9Sstevel@tonic-gate 				cmd = s_strdup(cp);
24637c478bd9Sstevel@tonic-gate 				sep = arg = NULL;
24647c478bd9Sstevel@tonic-gate 				break;
24657c478bd9Sstevel@tonic-gate 			}
24667c478bd9Sstevel@tonic-gate 			str++;
24677c478bd9Sstevel@tonic-gate 		}
24687c478bd9Sstevel@tonic-gate 
24697c478bd9Sstevel@tonic-gate 		if (*str != '\0') {
24707c478bd9Sstevel@tonic-gate 			save = *str;
24717c478bd9Sstevel@tonic-gate 			*str = '\0';
24727c478bd9Sstevel@tonic-gate 			cmd = s_strdup(cp);
24737c478bd9Sstevel@tonic-gate 			*str = save;
24747c478bd9Sstevel@tonic-gate 
24757c478bd9Sstevel@tonic-gate 			str++;
24767c478bd9Sstevel@tonic-gate 			save = *str;
24777c478bd9Sstevel@tonic-gate 			*str = '\0';
24787c478bd9Sstevel@tonic-gate 			sep = s_strdup(str - 1);
24797c478bd9Sstevel@tonic-gate 			*str = save;
24807c478bd9Sstevel@tonic-gate 
24817c478bd9Sstevel@tonic-gate 			while (*str == ' ' || *str == '\t')
24827c478bd9Sstevel@tonic-gate 				str++;
24837c478bd9Sstevel@tonic-gate 			if (*str == '\0')
24847c478bd9Sstevel@tonic-gate 				arg = NULL;
24857c478bd9Sstevel@tonic-gate 			else
24867c478bd9Sstevel@tonic-gate 				arg = s_strdup(str);
24877c478bd9Sstevel@tonic-gate 		}
24887c478bd9Sstevel@tonic-gate 	}
24897c478bd9Sstevel@tonic-gate 
24907c478bd9Sstevel@tonic-gate 	lp = s_calloc(1, sizeof (line_t));
24917c478bd9Sstevel@tonic-gate 
24927c478bd9Sstevel@tonic-gate 	lp->cmd = cmd;
24937c478bd9Sstevel@tonic-gate 	lp->sep = sep;
24947c478bd9Sstevel@tonic-gate 	lp->arg = arg;
24957c478bd9Sstevel@tonic-gate 	lp->line = line;
24967c478bd9Sstevel@tonic-gate 	lp->lineNum = ++(*lineNum);
24977c478bd9Sstevel@tonic-gate 	if (cmd && strcmp(cmd, menu_cmds[TITLE_CMD]) == 0) {
24987c478bd9Sstevel@tonic-gate 		lp->entryNum = ++(*entryNum);
24997c478bd9Sstevel@tonic-gate 		lp->flags = BAM_TITLE;
25007c478bd9Sstevel@tonic-gate 		if (prev && prev->flags == BAM_COMMENT &&
2501ae115bc7Smrj 		    prev->arg && strcmp(prev->arg, BAM_BOOTADM_HDR) == 0) {
25027c478bd9Sstevel@tonic-gate 			prev->entryNum = lp->entryNum;
25038c1b6884Sszhou 			curr_ent = boot_entry_new(mp, prev, lp);
2504ae115bc7Smrj 			curr_ent->flags = BAM_ENTRY_BOOTADM;
25058c1b6884Sszhou 		} else {
25068c1b6884Sszhou 			curr_ent = boot_entry_new(mp, lp, lp);
2507ae115bc7Smrj 			if (in_liveupgrade) {
2508ae115bc7Smrj 				curr_ent->flags = BAM_ENTRY_LU;
25098c1b6884Sszhou 			}
2510ae115bc7Smrj 		}
2511ae115bc7Smrj 		curr_ent->entryNum = *entryNum;
25127c478bd9Sstevel@tonic-gate 	} else if (flag != BAM_INVALID) {
25137c478bd9Sstevel@tonic-gate 		/*
25147c478bd9Sstevel@tonic-gate 		 * For header comments, the entry# is "fixed up"
25157c478bd9Sstevel@tonic-gate 		 * by the subsequent title
25167c478bd9Sstevel@tonic-gate 		 */
25177c478bd9Sstevel@tonic-gate 		lp->entryNum = *entryNum;
25187c478bd9Sstevel@tonic-gate 		lp->flags = flag;
25197c478bd9Sstevel@tonic-gate 	} else {
25207c478bd9Sstevel@tonic-gate 		lp->entryNum = *entryNum;
2521ae115bc7Smrj 
2522ae115bc7Smrj 		if (*entryNum == ENTRY_INIT) {
2523ae115bc7Smrj 			lp->flags = BAM_GLOBAL;
2524ae115bc7Smrj 		} else {
2525ae115bc7Smrj 			lp->flags = BAM_ENTRY;
2526ae115bc7Smrj 
2527ae115bc7Smrj 			if (cmd && arg) {
2528ae115bc7Smrj 				/*
2529ae115bc7Smrj 				 * We only compare for the length of "module"
2530ae115bc7Smrj 				 * so that "module$" will also match.
2531ae115bc7Smrj 				 */
2532ae115bc7Smrj 				if ((strncmp(cmd, menu_cmds[MODULE_CMD],
2533ae115bc7Smrj 				    strlen(menu_cmds[MODULE_CMD])) == 0) &&
2534ae115bc7Smrj 				    (strcmp(arg, MINIROOT) == 0))
2535ae115bc7Smrj 					curr_ent->flags |= BAM_ENTRY_MINIROOT;
2536ae115bc7Smrj 				else if (strcmp(cmd, menu_cmds[ROOT_CMD]) == 0)
2537ae115bc7Smrj 					curr_ent->flags |= BAM_ENTRY_ROOT;
2538ae115bc7Smrj 				else if (strcmp(cmd,
2539ae115bc7Smrj 				    menu_cmds[CHAINLOADER_CMD]) == 0)
2540ae115bc7Smrj 					curr_ent->flags |=
2541ae115bc7Smrj 					    BAM_ENTRY_CHAINLOADER;
2542ae115bc7Smrj 			}
2543ae115bc7Smrj 		}
25447c478bd9Sstevel@tonic-gate 	}
25457c478bd9Sstevel@tonic-gate 
25468c1b6884Sszhou 	/* record default, old default, and entry line ranges */
25478c1b6884Sszhou 	if (lp->flags == BAM_GLOBAL &&
25488c1b6884Sszhou 	    strcmp(lp->cmd, menu_cmds[DEFAULT_CMD]) == 0) {
25498c1b6884Sszhou 		mp->curdefault = lp;
25508c1b6884Sszhou 	} else if (lp->flags == BAM_COMMENT &&
25518c1b6884Sszhou 	    strncmp(lp->arg, BAM_OLDDEF, strlen(BAM_OLDDEF)) == 0) {
25528c1b6884Sszhou 		mp->olddefault = lp;
2553ae115bc7Smrj 	} else if (lp->flags == BAM_COMMENT &&
2554ae115bc7Smrj 	    strncmp(lp->arg, BAM_OLD_RC_DEF, strlen(BAM_OLD_RC_DEF)) == 0) {
2555ae115bc7Smrj 		mp->old_rc_default = lp;
25568c1b6884Sszhou 	} else if (lp->flags == BAM_ENTRY ||
2557ae115bc7Smrj 	    (lp->flags == BAM_COMMENT &&
2558ae115bc7Smrj 	    strcmp(lp->arg, BAM_BOOTADM_FTR) == 0)) {
25598c1b6884Sszhou 		boot_entry_addline(curr_ent, lp);
25608c1b6884Sszhou 	}
25617c478bd9Sstevel@tonic-gate 	append_line(mp, lp);
25627c478bd9Sstevel@tonic-gate 
25637c478bd9Sstevel@tonic-gate 	prev = lp;
25647c478bd9Sstevel@tonic-gate }
25657c478bd9Sstevel@tonic-gate 
256640541d5dSvikram static void
256740541d5dSvikram update_numbering(menu_t *mp)
256840541d5dSvikram {
256940541d5dSvikram 	int lineNum;
257040541d5dSvikram 	int entryNum;
257140541d5dSvikram 	int old_default_value;
257240541d5dSvikram 	line_t *lp, *prev, *default_lp, *default_entry;
257340541d5dSvikram 	char buf[PATH_MAX];
257440541d5dSvikram 
257540541d5dSvikram 	if (mp->start == NULL) {
257640541d5dSvikram 		return;
257740541d5dSvikram 	}
257840541d5dSvikram 
257940541d5dSvikram 	lineNum = LINE_INIT;
258040541d5dSvikram 	entryNum = ENTRY_INIT;
258140541d5dSvikram 	old_default_value = ENTRY_INIT;
258240541d5dSvikram 	lp = default_lp = default_entry = NULL;
258340541d5dSvikram 
258440541d5dSvikram 	prev = NULL;
258540541d5dSvikram 	for (lp = mp->start; lp; prev = lp, lp = lp->next) {
258640541d5dSvikram 		lp->lineNum = ++lineNum;
258740541d5dSvikram 
258840541d5dSvikram 		/*
258940541d5dSvikram 		 * Get the value of the default command
259040541d5dSvikram 		 */
259140541d5dSvikram 		if (lp->entryNum == ENTRY_INIT && lp->cmd &&
259240541d5dSvikram 		    strcmp(lp->cmd, menu_cmds[DEFAULT_CMD]) == 0 &&
259340541d5dSvikram 		    lp->arg) {
259440541d5dSvikram 			old_default_value = atoi(lp->arg);
259540541d5dSvikram 			default_lp = lp;
259640541d5dSvikram 		}
259740541d5dSvikram 
259840541d5dSvikram 		/*
259940541d5dSvikram 		 * If not boot entry, nothing else to fix for this
260040541d5dSvikram 		 * entry
260140541d5dSvikram 		 */
260240541d5dSvikram 		if (lp->entryNum == ENTRY_INIT)
260340541d5dSvikram 			continue;
260440541d5dSvikram 
260540541d5dSvikram 		/*
260640541d5dSvikram 		 * Record the position of the default entry.
260740541d5dSvikram 		 * The following works because global
260840541d5dSvikram 		 * commands like default and timeout should precede
260940541d5dSvikram 		 * actual boot entries, so old_default_value
261040541d5dSvikram 		 * is already known (or default cmd is missing).
261140541d5dSvikram 		 */
261240541d5dSvikram 		if (default_entry == NULL &&
261340541d5dSvikram 		    old_default_value != ENTRY_INIT &&
261440541d5dSvikram 		    lp->entryNum == old_default_value) {
261540541d5dSvikram 			default_entry = lp;
261640541d5dSvikram 		}
261740541d5dSvikram 
261840541d5dSvikram 		/*
261940541d5dSvikram 		 * Now fixup the entry number
262040541d5dSvikram 		 */
262140541d5dSvikram 		if (lp->cmd && strcmp(lp->cmd, menu_cmds[TITLE_CMD]) == 0) {
262240541d5dSvikram 			lp->entryNum = ++entryNum;
262340541d5dSvikram 			/* fixup the bootadm header */
262440541d5dSvikram 			if (prev && prev->flags == BAM_COMMENT &&
2625ae115bc7Smrj 			    prev->arg &&
2626ae115bc7Smrj 			    strcmp(prev->arg, BAM_BOOTADM_HDR) == 0) {
262740541d5dSvikram 				prev->entryNum = lp->entryNum;
262840541d5dSvikram 			}
262940541d5dSvikram 		} else {
263040541d5dSvikram 			lp->entryNum = entryNum;
263140541d5dSvikram 		}
263240541d5dSvikram 	}
263340541d5dSvikram 
263440541d5dSvikram 	/*
263540541d5dSvikram 	 * No default command in menu, simply return
263640541d5dSvikram 	 */
263740541d5dSvikram 	if (default_lp == NULL) {
263840541d5dSvikram 		return;
263940541d5dSvikram 	}
264040541d5dSvikram 
264140541d5dSvikram 	free(default_lp->arg);
264240541d5dSvikram 	free(default_lp->line);
264340541d5dSvikram 
264440541d5dSvikram 	if (default_entry == NULL) {
264540541d5dSvikram 		default_lp->arg = s_strdup("0");
264640541d5dSvikram 	} else {
264740541d5dSvikram 		(void) snprintf(buf, sizeof (buf), "%d",
264840541d5dSvikram 		    default_entry->entryNum);
264940541d5dSvikram 		default_lp->arg = s_strdup(buf);
265040541d5dSvikram 	}
265140541d5dSvikram 
265240541d5dSvikram 	/*
265340541d5dSvikram 	 * The following is required since only the line field gets
265440541d5dSvikram 	 * written back to menu.lst
265540541d5dSvikram 	 */
265640541d5dSvikram 	(void) snprintf(buf, sizeof (buf), "%s%s%s",
265740541d5dSvikram 	    menu_cmds[DEFAULT_CMD], menu_cmds[SEP_CMD], default_lp->arg);
265840541d5dSvikram 	default_lp->line = s_strdup(buf);
265940541d5dSvikram }
266040541d5dSvikram 
266140541d5dSvikram 
26627c478bd9Sstevel@tonic-gate static menu_t *
26637c478bd9Sstevel@tonic-gate menu_read(char *menu_path)
26647c478bd9Sstevel@tonic-gate {
26657c478bd9Sstevel@tonic-gate 	FILE *fp;
26667c478bd9Sstevel@tonic-gate 	char buf[BAM_MAXLINE], *cp;
26677c478bd9Sstevel@tonic-gate 	menu_t *mp;
26687c478bd9Sstevel@tonic-gate 	int line, entry, len, n;
26697c478bd9Sstevel@tonic-gate 
26707c478bd9Sstevel@tonic-gate 	mp = s_calloc(1, sizeof (menu_t));
26717c478bd9Sstevel@tonic-gate 
26727c478bd9Sstevel@tonic-gate 	fp = fopen(menu_path, "r");
26737c478bd9Sstevel@tonic-gate 	if (fp == NULL) { /* Let the caller handle this error */
26747c478bd9Sstevel@tonic-gate 		return (mp);
26757c478bd9Sstevel@tonic-gate 	}
26767c478bd9Sstevel@tonic-gate 
26777c478bd9Sstevel@tonic-gate 
26787c478bd9Sstevel@tonic-gate 	/* Note: GRUB boot entry number starts with 0 */
26797c478bd9Sstevel@tonic-gate 	line = LINE_INIT;
26807c478bd9Sstevel@tonic-gate 	entry = ENTRY_INIT;
26817c478bd9Sstevel@tonic-gate 	cp = buf;
26827c478bd9Sstevel@tonic-gate 	len = sizeof (buf);
26837c478bd9Sstevel@tonic-gate 	while (s_fgets(cp, len, fp) != NULL) {
26847c478bd9Sstevel@tonic-gate 		n = strlen(cp);
26857c478bd9Sstevel@tonic-gate 		if (cp[n - 1] == '\\') {
26867c478bd9Sstevel@tonic-gate 			len -= n - 1;
26877c478bd9Sstevel@tonic-gate 			assert(len >= 2);
26887c478bd9Sstevel@tonic-gate 			cp += n - 1;
26897c478bd9Sstevel@tonic-gate 			continue;
26907c478bd9Sstevel@tonic-gate 		}
26917c478bd9Sstevel@tonic-gate 		line_parser(mp, buf, &line, &entry);
26927c478bd9Sstevel@tonic-gate 		cp = buf;
26937c478bd9Sstevel@tonic-gate 		len = sizeof (buf);
26947c478bd9Sstevel@tonic-gate 	}
26957c478bd9Sstevel@tonic-gate 
26967c478bd9Sstevel@tonic-gate 	if (fclose(fp) == EOF) {
26977c478bd9Sstevel@tonic-gate 		bam_error(CLOSE_FAIL, menu_path, strerror(errno));
26987c478bd9Sstevel@tonic-gate 	}
26997c478bd9Sstevel@tonic-gate 
27007c478bd9Sstevel@tonic-gate 	return (mp);
27017c478bd9Sstevel@tonic-gate }
27027c478bd9Sstevel@tonic-gate 
27037c478bd9Sstevel@tonic-gate static error_t
27047c478bd9Sstevel@tonic-gate selector(menu_t *mp, char *opt, int *entry, char **title)
27057c478bd9Sstevel@tonic-gate {
27067c478bd9Sstevel@tonic-gate 	char *eq;
27077c478bd9Sstevel@tonic-gate 	char *opt_dup;
27087c478bd9Sstevel@tonic-gate 	int entryNum;
27097c478bd9Sstevel@tonic-gate 
27107c478bd9Sstevel@tonic-gate 	assert(mp);
27117c478bd9Sstevel@tonic-gate 	assert(mp->start);
27127c478bd9Sstevel@tonic-gate 	assert(opt);
27137c478bd9Sstevel@tonic-gate 
27147c478bd9Sstevel@tonic-gate 	opt_dup = s_strdup(opt);
27157c478bd9Sstevel@tonic-gate 
27167c478bd9Sstevel@tonic-gate 	if (entry)
27177c478bd9Sstevel@tonic-gate 		*entry = ENTRY_INIT;
27187c478bd9Sstevel@tonic-gate 	if (title)
27197c478bd9Sstevel@tonic-gate 		*title = NULL;
27207c478bd9Sstevel@tonic-gate 
27217c478bd9Sstevel@tonic-gate 	eq = strchr(opt_dup, '=');
27227c478bd9Sstevel@tonic-gate 	if (eq == NULL) {
27237c478bd9Sstevel@tonic-gate 		bam_error(INVALID_OPT, opt);
27247c478bd9Sstevel@tonic-gate 		free(opt_dup);
27257c478bd9Sstevel@tonic-gate 		return (BAM_ERROR);
27267c478bd9Sstevel@tonic-gate 	}
27277c478bd9Sstevel@tonic-gate 
27287c478bd9Sstevel@tonic-gate 	*eq = '\0';
27297c478bd9Sstevel@tonic-gate 	if (entry && strcmp(opt_dup, OPT_ENTRY_NUM) == 0) {
27307c478bd9Sstevel@tonic-gate 		assert(mp->end);
27317c478bd9Sstevel@tonic-gate 		entryNum = s_strtol(eq + 1);
27327c478bd9Sstevel@tonic-gate 		if (entryNum < 0 || entryNum > mp->end->entryNum) {
27337c478bd9Sstevel@tonic-gate 			bam_error(INVALID_ENTRY, eq + 1);
27347c478bd9Sstevel@tonic-gate 			free(opt_dup);
27357c478bd9Sstevel@tonic-gate 			return (BAM_ERROR);
27367c478bd9Sstevel@tonic-gate 		}
27377c478bd9Sstevel@tonic-gate 		*entry = entryNum;
27387c478bd9Sstevel@tonic-gate 	} else if (title && strcmp(opt_dup, menu_cmds[TITLE_CMD]) == 0) {
27397c478bd9Sstevel@tonic-gate 		*title = opt + (eq - opt_dup) + 1;
27407c478bd9Sstevel@tonic-gate 	} else {
27417c478bd9Sstevel@tonic-gate 		bam_error(INVALID_OPT, opt);
27427c478bd9Sstevel@tonic-gate 		free(opt_dup);
27437c478bd9Sstevel@tonic-gate 		return (BAM_ERROR);
27447c478bd9Sstevel@tonic-gate 	}
27457c478bd9Sstevel@tonic-gate 
27467c478bd9Sstevel@tonic-gate 	free(opt_dup);
27477c478bd9Sstevel@tonic-gate 	return (BAM_SUCCESS);
27487c478bd9Sstevel@tonic-gate }
27497c478bd9Sstevel@tonic-gate 
27507c478bd9Sstevel@tonic-gate /*
27517c478bd9Sstevel@tonic-gate  * If invoked with no titles/entries (opt == NULL)
27527c478bd9Sstevel@tonic-gate  * only title lines in file are printed.
27537c478bd9Sstevel@tonic-gate  *
27547c478bd9Sstevel@tonic-gate  * If invoked with a title or entry #, all
27557c478bd9Sstevel@tonic-gate  * lines in *every* matching entry are listed
27567c478bd9Sstevel@tonic-gate  */
27577c478bd9Sstevel@tonic-gate static error_t
27587c478bd9Sstevel@tonic-gate list_entry(menu_t *mp, char *menu_path, char *opt)
27597c478bd9Sstevel@tonic-gate {
27607c478bd9Sstevel@tonic-gate 	line_t *lp;
27617c478bd9Sstevel@tonic-gate 	int entry = ENTRY_INIT;
27627c478bd9Sstevel@tonic-gate 	int found;
27637c478bd9Sstevel@tonic-gate 	char *title = NULL;
27647c478bd9Sstevel@tonic-gate 
27657c478bd9Sstevel@tonic-gate 	assert(mp);
27667c478bd9Sstevel@tonic-gate 	assert(menu_path);
27677c478bd9Sstevel@tonic-gate 
27687c478bd9Sstevel@tonic-gate 	if (mp->start == NULL) {
27697c478bd9Sstevel@tonic-gate 		bam_error(NO_MENU, menu_path);
27707c478bd9Sstevel@tonic-gate 		return (BAM_ERROR);
27717c478bd9Sstevel@tonic-gate 	}
27727c478bd9Sstevel@tonic-gate 
27737c478bd9Sstevel@tonic-gate 	if (opt != NULL) {
27747c478bd9Sstevel@tonic-gate 		if (selector(mp, opt, &entry, &title) != BAM_SUCCESS) {
27757c478bd9Sstevel@tonic-gate 			return (BAM_ERROR);
27767c478bd9Sstevel@tonic-gate 		}
27777c478bd9Sstevel@tonic-gate 		assert((entry != ENTRY_INIT) ^ (title != NULL));
27787c478bd9Sstevel@tonic-gate 	} else {
27797c478bd9Sstevel@tonic-gate 		(void) read_globals(mp, menu_path, menu_cmds[DEFAULT_CMD], 0);
27807c478bd9Sstevel@tonic-gate 		(void) read_globals(mp, menu_path, menu_cmds[TIMEOUT_CMD], 0);
27817c478bd9Sstevel@tonic-gate 	}
27827c478bd9Sstevel@tonic-gate 
27837c478bd9Sstevel@tonic-gate 	found = 0;
27847c478bd9Sstevel@tonic-gate 	for (lp = mp->start; lp; lp = lp->next) {
27857c478bd9Sstevel@tonic-gate 		if (lp->flags == BAM_COMMENT || lp->flags == BAM_EMPTY)
27867c478bd9Sstevel@tonic-gate 			continue;
27877c478bd9Sstevel@tonic-gate 		if (opt == NULL && lp->flags == BAM_TITLE) {
27887c478bd9Sstevel@tonic-gate 			bam_print(PRINT_TITLE, lp->entryNum,
27897c478bd9Sstevel@tonic-gate 			    lp->arg);
27907c478bd9Sstevel@tonic-gate 			found = 1;
27917c478bd9Sstevel@tonic-gate 			continue;
27927c478bd9Sstevel@tonic-gate 		}
27937c478bd9Sstevel@tonic-gate 		if (entry != ENTRY_INIT && lp->entryNum == entry) {
27947c478bd9Sstevel@tonic-gate 			bam_print(PRINT, lp->line);
27957c478bd9Sstevel@tonic-gate 			found = 1;
27967c478bd9Sstevel@tonic-gate 			continue;
27977c478bd9Sstevel@tonic-gate 		}
27987c478bd9Sstevel@tonic-gate 
27997c478bd9Sstevel@tonic-gate 		/*
28007c478bd9Sstevel@tonic-gate 		 * We set the entry value here so that all lines
28017c478bd9Sstevel@tonic-gate 		 * in entry get printed. If we subsequently match
28027c478bd9Sstevel@tonic-gate 		 * title in other entries, all lines in those
28037c478bd9Sstevel@tonic-gate 		 * entries get printed as well.
28047c478bd9Sstevel@tonic-gate 		 */
28057c478bd9Sstevel@tonic-gate 		if (title && lp->flags == BAM_TITLE && lp->arg &&
28067c478bd9Sstevel@tonic-gate 		    strncmp(title, lp->arg, strlen(title)) == 0) {
28077c478bd9Sstevel@tonic-gate 			bam_print(PRINT, lp->line);
28087c478bd9Sstevel@tonic-gate 			entry = lp->entryNum;
28097c478bd9Sstevel@tonic-gate 			found = 1;
28107c478bd9Sstevel@tonic-gate 			continue;
28117c478bd9Sstevel@tonic-gate 		}
28127c478bd9Sstevel@tonic-gate 	}
28137c478bd9Sstevel@tonic-gate 
28147c478bd9Sstevel@tonic-gate 	if (!found) {
28157c478bd9Sstevel@tonic-gate 		bam_error(NO_MATCH_ENTRY);
28167c478bd9Sstevel@tonic-gate 		return (BAM_ERROR);
28177c478bd9Sstevel@tonic-gate 	}
28187c478bd9Sstevel@tonic-gate 
28197c478bd9Sstevel@tonic-gate 	return (BAM_SUCCESS);
28207c478bd9Sstevel@tonic-gate }
28217c478bd9Sstevel@tonic-gate 
28227c478bd9Sstevel@tonic-gate static int
28237c478bd9Sstevel@tonic-gate add_boot_entry(menu_t *mp,
28247c478bd9Sstevel@tonic-gate 	char *title,
28257c478bd9Sstevel@tonic-gate 	char *root,
28267c478bd9Sstevel@tonic-gate 	char *kernel,
28277c478bd9Sstevel@tonic-gate 	char *module)
28287c478bd9Sstevel@tonic-gate {
28297c478bd9Sstevel@tonic-gate 	int lineNum, entryNum;
28307c478bd9Sstevel@tonic-gate 	char linebuf[BAM_MAXLINE];
2831ae115bc7Smrj 	menu_cmd_t k_cmd, m_cmd;
28327c478bd9Sstevel@tonic-gate 
28337c478bd9Sstevel@tonic-gate 	assert(mp);
28347c478bd9Sstevel@tonic-gate 
28357c478bd9Sstevel@tonic-gate 	if (title == NULL) {
2836ee3b8144Sszhou 		title = "Solaris";	/* default to Solaris */
28377c478bd9Sstevel@tonic-gate 	}
28387c478bd9Sstevel@tonic-gate 	if (kernel == NULL) {
28397c478bd9Sstevel@tonic-gate 		bam_error(SUBOPT_MISS, menu_cmds[KERNEL_CMD]);
28407c478bd9Sstevel@tonic-gate 		return (BAM_ERROR);
28417c478bd9Sstevel@tonic-gate 	}
28427c478bd9Sstevel@tonic-gate 	if (module == NULL) {
2843ae115bc7Smrj 		if (bam_direct != BAM_DIRECT_DBOOT) {
28447c478bd9Sstevel@tonic-gate 			bam_error(SUBOPT_MISS, menu_cmds[MODULE_CMD]);
28457c478bd9Sstevel@tonic-gate 			return (BAM_ERROR);
28467c478bd9Sstevel@tonic-gate 		}
28477c478bd9Sstevel@tonic-gate 
2848ae115bc7Smrj 		/* Figure the commands out from the kernel line */
2849ae115bc7Smrj 		if (strstr(kernel, "$ISADIR") != NULL) {
2850ae115bc7Smrj 			module = DIRECT_BOOT_ARCHIVE;
2851ae115bc7Smrj 			k_cmd = KERNEL_DOLLAR_CMD;
2852ae115bc7Smrj 			m_cmd = MODULE_DOLLAR_CMD;
2853ae115bc7Smrj 		} else if (strstr(kernel, "amd64") != NULL) {
2854ae115bc7Smrj 			module = DIRECT_BOOT_ARCHIVE_64;
2855ae115bc7Smrj 			k_cmd = KERNEL_CMD;
2856ae115bc7Smrj 			m_cmd = MODULE_CMD;
2857ae115bc7Smrj 		} else {
2858ae115bc7Smrj 			module = DIRECT_BOOT_ARCHIVE_32;
2859ae115bc7Smrj 			k_cmd = KERNEL_CMD;
2860ae115bc7Smrj 			m_cmd = MODULE_CMD;
2861ae115bc7Smrj 		}
2862ae115bc7Smrj 	} else if ((bam_direct == BAM_DIRECT_DBOOT) &&
2863ae115bc7Smrj 	    (strstr(kernel, "$ISADIR") != NULL)) {
2864ae115bc7Smrj 		/*
2865ae115bc7Smrj 		 * If it's a non-failsafe dboot kernel, use the "kernel$"
2866ae115bc7Smrj 		 * command.  Otherwise, use "kernel".
2867ae115bc7Smrj 		 */
2868ae115bc7Smrj 		k_cmd = KERNEL_DOLLAR_CMD;
2869ae115bc7Smrj 		m_cmd = MODULE_DOLLAR_CMD;
2870ae115bc7Smrj 	} else {
2871ae115bc7Smrj 		k_cmd = KERNEL_CMD;
2872ae115bc7Smrj 		m_cmd = MODULE_CMD;
2873ae115bc7Smrj 	}
2874ae115bc7Smrj 
28757c478bd9Sstevel@tonic-gate 	if (mp->start) {
28767c478bd9Sstevel@tonic-gate 		lineNum = mp->end->lineNum;
28777c478bd9Sstevel@tonic-gate 		entryNum = mp->end->entryNum;
28787c478bd9Sstevel@tonic-gate 	} else {
28797c478bd9Sstevel@tonic-gate 		lineNum = LINE_INIT;
28807c478bd9Sstevel@tonic-gate 		entryNum = ENTRY_INIT;
28817c478bd9Sstevel@tonic-gate 	}
28827c478bd9Sstevel@tonic-gate 
28837c478bd9Sstevel@tonic-gate 	/*
28847c478bd9Sstevel@tonic-gate 	 * No separator for comment (HDR/FTR) commands
28857c478bd9Sstevel@tonic-gate 	 * The syntax for comments is #<comment>
28867c478bd9Sstevel@tonic-gate 	 */
28877c478bd9Sstevel@tonic-gate 	(void) snprintf(linebuf, sizeof (linebuf), "%s%s",
2888ae115bc7Smrj 	    menu_cmds[COMMENT_CMD], BAM_BOOTADM_HDR);
28898c1b6884Sszhou 	line_parser(mp, linebuf, &lineNum, &entryNum);
28907c478bd9Sstevel@tonic-gate 
28917c478bd9Sstevel@tonic-gate 	(void) snprintf(linebuf, sizeof (linebuf), "%s%s%s",
28927c478bd9Sstevel@tonic-gate 	    menu_cmds[TITLE_CMD], menu_cmds[SEP_CMD], title);
28938c1b6884Sszhou 	line_parser(mp, linebuf, &lineNum, &entryNum);
28947c478bd9Sstevel@tonic-gate 
28958c1b6884Sszhou 	if (root) {
28967c478bd9Sstevel@tonic-gate 		(void) snprintf(linebuf, sizeof (linebuf), "%s%s%s",
28977c478bd9Sstevel@tonic-gate 		    menu_cmds[ROOT_CMD], menu_cmds[SEP_CMD], root);
28988c1b6884Sszhou 		line_parser(mp, linebuf, &lineNum, &entryNum);
28997c478bd9Sstevel@tonic-gate 	}
29007c478bd9Sstevel@tonic-gate 
29017c478bd9Sstevel@tonic-gate 	(void) snprintf(linebuf, sizeof (linebuf), "%s%s%s",
2902ae115bc7Smrj 	    menu_cmds[k_cmd], menu_cmds[SEP_CMD], kernel);
29038c1b6884Sszhou 	line_parser(mp, linebuf, &lineNum, &entryNum);
29047c478bd9Sstevel@tonic-gate 
29057c478bd9Sstevel@tonic-gate 	(void) snprintf(linebuf, sizeof (linebuf), "%s%s%s",
2906ae115bc7Smrj 	    menu_cmds[m_cmd], menu_cmds[SEP_CMD], module);
29078c1b6884Sszhou 	line_parser(mp, linebuf, &lineNum, &entryNum);
29087c478bd9Sstevel@tonic-gate 
29097c478bd9Sstevel@tonic-gate 	(void) snprintf(linebuf, sizeof (linebuf), "%s%s",
2910ae115bc7Smrj 	    menu_cmds[COMMENT_CMD], BAM_BOOTADM_FTR);
29118c1b6884Sszhou 	line_parser(mp, linebuf, &lineNum, &entryNum);
29127c478bd9Sstevel@tonic-gate 
29137c478bd9Sstevel@tonic-gate 	return (entryNum);
29147c478bd9Sstevel@tonic-gate }
29157c478bd9Sstevel@tonic-gate 
29167c478bd9Sstevel@tonic-gate static error_t
29177c478bd9Sstevel@tonic-gate do_delete(menu_t *mp, int entryNum)
29187c478bd9Sstevel@tonic-gate {
29198c1b6884Sszhou 	line_t *lp, *freed;
29208c1b6884Sszhou 	entry_t *ent, *tmp;
29217c478bd9Sstevel@tonic-gate 	int deleted;
29227c478bd9Sstevel@tonic-gate 
29237c478bd9Sstevel@tonic-gate 	assert(entryNum != ENTRY_INIT);
29247c478bd9Sstevel@tonic-gate 
29258c1b6884Sszhou 	ent = mp->entries;
29268c1b6884Sszhou 	while (ent) {
29278c1b6884Sszhou 		lp = ent->start;
29288c1b6884Sszhou 		/* check entry number and make sure it's a bootadm entry */
29298c1b6884Sszhou 		if (lp->flags != BAM_COMMENT ||
2930ae115bc7Smrj 		    strcmp(lp->arg, BAM_BOOTADM_HDR) != 0 ||
29318c1b6884Sszhou 		    (entryNum != ALL_ENTRIES && lp->entryNum != entryNum)) {
29328c1b6884Sszhou 			ent = ent->next;
29337c478bd9Sstevel@tonic-gate 			continue;
29347c478bd9Sstevel@tonic-gate 		}
29357c478bd9Sstevel@tonic-gate 
29368c1b6884Sszhou 		/* free the entry content */
29378c1b6884Sszhou 		do {
29388c1b6884Sszhou 			freed = lp;
29398c1b6884Sszhou 			lp = lp->next;	/* prev stays the same */
29408c1b6884Sszhou 			unlink_line(mp, freed);
29418c1b6884Sszhou 			line_free(freed);
29428c1b6884Sszhou 		} while (freed != ent->end);
29437c478bd9Sstevel@tonic-gate 
29448c1b6884Sszhou 		/* free the entry_t structure */
29458c1b6884Sszhou 		tmp = ent;
29468c1b6884Sszhou 		ent = ent->next;
29478c1b6884Sszhou 		if (tmp->prev)
29488c1b6884Sszhou 			tmp->prev->next = ent;
29497c478bd9Sstevel@tonic-gate 		else
29508c1b6884Sszhou 			mp->entries = ent;
29518c1b6884Sszhou 		if (ent)
29528c1b6884Sszhou 			ent->prev = tmp->prev;
29537c478bd9Sstevel@tonic-gate 		deleted = 1;
29547c478bd9Sstevel@tonic-gate 	}
29557c478bd9Sstevel@tonic-gate 
29567c478bd9Sstevel@tonic-gate 	if (!deleted && entryNum != ALL_ENTRIES) {
29577c478bd9Sstevel@tonic-gate 		bam_error(NO_BOOTADM_MATCH);
29587c478bd9Sstevel@tonic-gate 		return (BAM_ERROR);
29597c478bd9Sstevel@tonic-gate 	}
29607c478bd9Sstevel@tonic-gate 
296140541d5dSvikram 	/*
296240541d5dSvikram 	 * Now that we have deleted an entry, update
296340541d5dSvikram 	 * the entry numbering and the default cmd.
296440541d5dSvikram 	 */
296540541d5dSvikram 	update_numbering(mp);
296640541d5dSvikram 
29677c478bd9Sstevel@tonic-gate 	return (BAM_SUCCESS);
29687c478bd9Sstevel@tonic-gate }
29697c478bd9Sstevel@tonic-gate 
29707c478bd9Sstevel@tonic-gate static error_t
29717c478bd9Sstevel@tonic-gate delete_all_entries(menu_t *mp, char *menu_path, char *opt)
29727c478bd9Sstevel@tonic-gate {
29737c478bd9Sstevel@tonic-gate 	assert(mp);
29747c478bd9Sstevel@tonic-gate 	assert(opt == NULL);
29757c478bd9Sstevel@tonic-gate 
29767c478bd9Sstevel@tonic-gate 	if (mp->start == NULL) {
29777c478bd9Sstevel@tonic-gate 		bam_print(EMPTY_FILE, menu_path);
29787c478bd9Sstevel@tonic-gate 		return (BAM_SUCCESS);
29797c478bd9Sstevel@tonic-gate 	}
29807c478bd9Sstevel@tonic-gate 
29817c478bd9Sstevel@tonic-gate 	if (do_delete(mp, ALL_ENTRIES) != BAM_SUCCESS) {
29827c478bd9Sstevel@tonic-gate 		return (BAM_ERROR);
29837c478bd9Sstevel@tonic-gate 	}
29847c478bd9Sstevel@tonic-gate 
29857c478bd9Sstevel@tonic-gate 	return (BAM_WRITE);
29867c478bd9Sstevel@tonic-gate }
29877c478bd9Sstevel@tonic-gate 
29887c478bd9Sstevel@tonic-gate static FILE *
29898c1b6884Sszhou open_diskmap(char *root)
29907c478bd9Sstevel@tonic-gate {
29917c478bd9Sstevel@tonic-gate 	FILE *fp;
29927c478bd9Sstevel@tonic-gate 	char cmd[PATH_MAX];
29937c478bd9Sstevel@tonic-gate 
29947c478bd9Sstevel@tonic-gate 	/* make sure we have a map file */
29957c478bd9Sstevel@tonic-gate 	fp = fopen(GRUBDISK_MAP, "r");
29967c478bd9Sstevel@tonic-gate 	if (fp == NULL) {
29977c478bd9Sstevel@tonic-gate 		(void) snprintf(cmd, sizeof (cmd),
29988c1b6884Sszhou 		    "%s%s > /dev/null", root, CREATE_DISKMAP);
29997c478bd9Sstevel@tonic-gate 		(void) system(cmd);
30007c478bd9Sstevel@tonic-gate 		fp = fopen(GRUBDISK_MAP, "r");
30017c478bd9Sstevel@tonic-gate 	}
30027c478bd9Sstevel@tonic-gate 	return (fp);
30037c478bd9Sstevel@tonic-gate }
30047c478bd9Sstevel@tonic-gate 
30057c478bd9Sstevel@tonic-gate #define	SECTOR_SIZE	512
30067c478bd9Sstevel@tonic-gate 
30077c478bd9Sstevel@tonic-gate static int
30087c478bd9Sstevel@tonic-gate get_partition(char *device)
30097c478bd9Sstevel@tonic-gate {
30107c478bd9Sstevel@tonic-gate 	int i, fd, is_pcfs, partno = -1;
30117c478bd9Sstevel@tonic-gate 	struct mboot *mboot;
30127c478bd9Sstevel@tonic-gate 	char boot_sect[SECTOR_SIZE];
30137c478bd9Sstevel@tonic-gate 	char *wholedisk, *slice;
30147c478bd9Sstevel@tonic-gate 
30157c478bd9Sstevel@tonic-gate 	/* form whole disk (p0) */
30167c478bd9Sstevel@tonic-gate 	slice = device + strlen(device) - 2;
30177c478bd9Sstevel@tonic-gate 	is_pcfs = (*slice != 's');
30187c478bd9Sstevel@tonic-gate 	if (!is_pcfs)
30197c478bd9Sstevel@tonic-gate 		*slice = '\0';
30207c478bd9Sstevel@tonic-gate 	wholedisk = s_calloc(1, strlen(device) + 3);
30217c478bd9Sstevel@tonic-gate 	(void) snprintf(wholedisk, strlen(device) + 3, "%sp0", device);
30227c478bd9Sstevel@tonic-gate 	if (!is_pcfs)
30237c478bd9Sstevel@tonic-gate 		*slice = 's';
30247c478bd9Sstevel@tonic-gate 
30257c478bd9Sstevel@tonic-gate 	/* read boot sector */
30267c478bd9Sstevel@tonic-gate 	fd = open(wholedisk, O_RDONLY);
30277c478bd9Sstevel@tonic-gate 	free(wholedisk);
30287c478bd9Sstevel@tonic-gate 	if (fd == -1 || read(fd, boot_sect, SECTOR_SIZE) != SECTOR_SIZE) {
30297c478bd9Sstevel@tonic-gate 		return (partno);
30307c478bd9Sstevel@tonic-gate 	}
30317c478bd9Sstevel@tonic-gate 	(void) close(fd);
30327c478bd9Sstevel@tonic-gate 
30337c478bd9Sstevel@tonic-gate 	/* parse fdisk table */
30347c478bd9Sstevel@tonic-gate 	mboot = (struct mboot *)((void *)boot_sect);
30357c478bd9Sstevel@tonic-gate 	for (i = 0; i < FD_NUMPART; i++) {
30367c478bd9Sstevel@tonic-gate 		struct ipart *part =
30377c478bd9Sstevel@tonic-gate 		    (struct ipart *)(uintptr_t)mboot->parts + i;
30387c478bd9Sstevel@tonic-gate 		if (is_pcfs) {	/* looking for solaris boot part */
30397c478bd9Sstevel@tonic-gate 			if (part->systid == 0xbe) {
30407c478bd9Sstevel@tonic-gate 				partno = i;
30417c478bd9Sstevel@tonic-gate 				break;
30427c478bd9Sstevel@tonic-gate 			}
30437c478bd9Sstevel@tonic-gate 		} else {	/* look for solaris partition, old and new */
30447c478bd9Sstevel@tonic-gate 			if (part->systid == SUNIXOS ||
30457c478bd9Sstevel@tonic-gate 			    part->systid == SUNIXOS2) {
30467c478bd9Sstevel@tonic-gate 				partno = i;
30477c478bd9Sstevel@tonic-gate 				break;
30487c478bd9Sstevel@tonic-gate 			}
30497c478bd9Sstevel@tonic-gate 		}
30507c478bd9Sstevel@tonic-gate 	}
30517c478bd9Sstevel@tonic-gate 	return (partno);
30527c478bd9Sstevel@tonic-gate }
30537c478bd9Sstevel@tonic-gate 
30547c478bd9Sstevel@tonic-gate static char *
30557c478bd9Sstevel@tonic-gate get_grubdisk(char *rootdev, FILE *fp, int on_bootdev)
30567c478bd9Sstevel@tonic-gate {
30577c478bd9Sstevel@tonic-gate 	char *grubdisk;	/* (hd#,#,#) */
30587c478bd9Sstevel@tonic-gate 	char *slice;
30597c478bd9Sstevel@tonic-gate 	char *grubhd;
30607c478bd9Sstevel@tonic-gate 	int fdiskpart;
30617c478bd9Sstevel@tonic-gate 	int found = 0;
30627c478bd9Sstevel@tonic-gate 	char *devname, *ctdname = strstr(rootdev, "dsk/");
30637c478bd9Sstevel@tonic-gate 	char linebuf[PATH_MAX];
30647c478bd9Sstevel@tonic-gate 
30657c478bd9Sstevel@tonic-gate 	if (ctdname == NULL)
30667c478bd9Sstevel@tonic-gate 		return (NULL);
30677c478bd9Sstevel@tonic-gate 
30687c478bd9Sstevel@tonic-gate 	ctdname += strlen("dsk/");
30697c478bd9Sstevel@tonic-gate 	slice = strrchr(ctdname, 's');
30707c478bd9Sstevel@tonic-gate 	if (slice)
30717c478bd9Sstevel@tonic-gate 		*slice = '\0';
30727c478bd9Sstevel@tonic-gate 
30737c478bd9Sstevel@tonic-gate 	rewind(fp);
30747c478bd9Sstevel@tonic-gate 	while (s_fgets(linebuf, sizeof (linebuf), fp) != NULL) {
30757c478bd9Sstevel@tonic-gate 		grubhd = strtok(linebuf, " \t\n");
30767c478bd9Sstevel@tonic-gate 		if (grubhd)
30777c478bd9Sstevel@tonic-gate 			devname = strtok(NULL, " \t\n");
30787c478bd9Sstevel@tonic-gate 		else
30797c478bd9Sstevel@tonic-gate 			devname = NULL;
30807c478bd9Sstevel@tonic-gate 		if (devname && strcmp(devname, ctdname) == 0) {
30817c478bd9Sstevel@tonic-gate 			found = 1;
30827c478bd9Sstevel@tonic-gate 			break;
30837c478bd9Sstevel@tonic-gate 		}
30847c478bd9Sstevel@tonic-gate 	}
30857c478bd9Sstevel@tonic-gate 
30867c478bd9Sstevel@tonic-gate 	if (slice)
30877c478bd9Sstevel@tonic-gate 		*slice = 's';
30887c478bd9Sstevel@tonic-gate 
30897c478bd9Sstevel@tonic-gate 	if (found == 0) {
30907c478bd9Sstevel@tonic-gate 		if (bam_verbose)
30917c478bd9Sstevel@tonic-gate 			bam_print(DISKMAP_FAIL_NONFATAL, rootdev);
30927c478bd9Sstevel@tonic-gate 		grubhd = "0";	/* assume disk 0 if can't match */
30937c478bd9Sstevel@tonic-gate 	}
30947c478bd9Sstevel@tonic-gate 
30957c478bd9Sstevel@tonic-gate 	fdiskpart = get_partition(rootdev);
30967c478bd9Sstevel@tonic-gate 	if (fdiskpart == -1)
30977c478bd9Sstevel@tonic-gate 		return (NULL);
30987c478bd9Sstevel@tonic-gate 
30997c478bd9Sstevel@tonic-gate 	grubdisk = s_calloc(1, 10);
31007c478bd9Sstevel@tonic-gate 	if (slice) {
31017c478bd9Sstevel@tonic-gate 		(void) snprintf(grubdisk, 10, "(hd%s,%d,%c)",
31027c478bd9Sstevel@tonic-gate 		    grubhd, fdiskpart, slice[1] + 'a' - '0');
31037c478bd9Sstevel@tonic-gate 	} else
31047c478bd9Sstevel@tonic-gate 		(void) snprintf(grubdisk, 10, "(hd%s,%d)",
31057c478bd9Sstevel@tonic-gate 		    grubhd, fdiskpart);
31067c478bd9Sstevel@tonic-gate 
31077c478bd9Sstevel@tonic-gate 	/* if root not on bootdev, change GRUB disk to 0 */
31087c478bd9Sstevel@tonic-gate 	if (!on_bootdev)
31097c478bd9Sstevel@tonic-gate 		grubdisk[3] = '0';
31107c478bd9Sstevel@tonic-gate 	return (grubdisk);
31117c478bd9Sstevel@tonic-gate }
31127c478bd9Sstevel@tonic-gate 
3113ee3b8144Sszhou static char *
3114ee3b8144Sszhou get_title(char *rootdir)
31157c478bd9Sstevel@tonic-gate {
31167c478bd9Sstevel@tonic-gate 	static char title[80];	/* from /etc/release */
31178c1b6884Sszhou 	char *cp = NULL, release[PATH_MAX];
31187c478bd9Sstevel@tonic-gate 	FILE *fp;
31197c478bd9Sstevel@tonic-gate 
31207c478bd9Sstevel@tonic-gate 	/* open the /etc/release file */
31217c478bd9Sstevel@tonic-gate 	(void) snprintf(release, sizeof (release), "%s/etc/release", rootdir);
31227c478bd9Sstevel@tonic-gate 
31237c478bd9Sstevel@tonic-gate 	fp = fopen(release, "r");
31247c478bd9Sstevel@tonic-gate 	if (fp == NULL)
3125ee3b8144Sszhou 		return (NULL);
31267c478bd9Sstevel@tonic-gate 
31277c478bd9Sstevel@tonic-gate 	while (s_fgets(title, sizeof (title), fp) != NULL) {
31287c478bd9Sstevel@tonic-gate 		cp = strstr(title, "Solaris");
31297c478bd9Sstevel@tonic-gate 		if (cp)
31307c478bd9Sstevel@tonic-gate 			break;
31317c478bd9Sstevel@tonic-gate 	}
31327c478bd9Sstevel@tonic-gate 	(void) fclose(fp);
31338c1b6884Sszhou 	return (cp == NULL ? "Solaris" : cp);
31347c478bd9Sstevel@tonic-gate }
31357c478bd9Sstevel@tonic-gate 
3136ae115bc7Smrj char *
31377c478bd9Sstevel@tonic-gate get_special(char *mountp)
31387c478bd9Sstevel@tonic-gate {
31397c478bd9Sstevel@tonic-gate 	FILE *mntfp;
31407c478bd9Sstevel@tonic-gate 	struct mnttab mp = {0}, mpref = {0};
31417c478bd9Sstevel@tonic-gate 
31427c478bd9Sstevel@tonic-gate 	mntfp = fopen(MNTTAB, "r");
31437c478bd9Sstevel@tonic-gate 	if (mntfp == NULL) {
31447c478bd9Sstevel@tonic-gate 		return (0);
31457c478bd9Sstevel@tonic-gate 	}
31467c478bd9Sstevel@tonic-gate 
31477c478bd9Sstevel@tonic-gate 	if (*mountp == '\0')
31487c478bd9Sstevel@tonic-gate 		mpref.mnt_mountp = "/";
31497c478bd9Sstevel@tonic-gate 	else
31507c478bd9Sstevel@tonic-gate 		mpref.mnt_mountp = mountp;
31517c478bd9Sstevel@tonic-gate 	if (getmntany(mntfp, &mp, &mpref) != 0) {
31527c478bd9Sstevel@tonic-gate 		(void) fclose(mntfp);
31537c478bd9Sstevel@tonic-gate 		return (NULL);
31547c478bd9Sstevel@tonic-gate 	}
31557c478bd9Sstevel@tonic-gate 	(void) fclose(mntfp);
31567c478bd9Sstevel@tonic-gate 
31577c478bd9Sstevel@tonic-gate 	return (s_strdup(mp.mnt_special));
31587c478bd9Sstevel@tonic-gate }
31597c478bd9Sstevel@tonic-gate 
3160ae115bc7Smrj char *
31617c478bd9Sstevel@tonic-gate os_to_grubdisk(char *osdisk, int on_bootdev)
31627c478bd9Sstevel@tonic-gate {
31637c478bd9Sstevel@tonic-gate 	FILE *fp;
31647c478bd9Sstevel@tonic-gate 	char *grubdisk;
31657c478bd9Sstevel@tonic-gate 
31667c478bd9Sstevel@tonic-gate 	/* translate /dev/dsk name to grub disk name */
31678c1b6884Sszhou 	fp = open_diskmap("");
31687c478bd9Sstevel@tonic-gate 	if (fp == NULL) {
31697c478bd9Sstevel@tonic-gate 		bam_error(DISKMAP_FAIL, osdisk);
31707c478bd9Sstevel@tonic-gate 		return (NULL);
31717c478bd9Sstevel@tonic-gate 	}
31727c478bd9Sstevel@tonic-gate 	grubdisk = get_grubdisk(osdisk, fp, on_bootdev);
31737c478bd9Sstevel@tonic-gate 	(void) fclose(fp);
31747c478bd9Sstevel@tonic-gate 	return (grubdisk);
31757c478bd9Sstevel@tonic-gate }
31767c478bd9Sstevel@tonic-gate 
31777c478bd9Sstevel@tonic-gate /*
31787c478bd9Sstevel@tonic-gate  * Check if root is on the boot device
31797c478bd9Sstevel@tonic-gate  * Return 0 (false) on error
31807c478bd9Sstevel@tonic-gate  */
31817c478bd9Sstevel@tonic-gate static int
31827c478bd9Sstevel@tonic-gate menu_on_bootdev(char *menu_root, FILE *fp)
31837c478bd9Sstevel@tonic-gate {
31847c478bd9Sstevel@tonic-gate 	int ret;
31857c478bd9Sstevel@tonic-gate 	char *grubhd, *bootp, *special;
31867c478bd9Sstevel@tonic-gate 
31877c478bd9Sstevel@tonic-gate 	special = get_special(menu_root);
31887c478bd9Sstevel@tonic-gate 	if (special == NULL)
31897c478bd9Sstevel@tonic-gate 		return (0);
31907c478bd9Sstevel@tonic-gate 	bootp = strstr(special, "p0:boot");
31917c478bd9Sstevel@tonic-gate 	if (bootp)
31927c478bd9Sstevel@tonic-gate 		*bootp = '\0';
31937c478bd9Sstevel@tonic-gate 	grubhd = get_grubdisk(special, fp, 1);
31947c478bd9Sstevel@tonic-gate 	free(special);
31957c478bd9Sstevel@tonic-gate 
31967c478bd9Sstevel@tonic-gate 	if (grubhd == NULL)
31977c478bd9Sstevel@tonic-gate 		return (0);
31987c478bd9Sstevel@tonic-gate 	ret = grubhd[3] == '0';
31997c478bd9Sstevel@tonic-gate 	free(grubhd);
32007c478bd9Sstevel@tonic-gate 	return (ret);
32017c478bd9Sstevel@tonic-gate }
32027c478bd9Sstevel@tonic-gate 
32038c1b6884Sszhou /*
32048c1b6884Sszhou  * look for matching bootadm entry with specified parameters
32058c1b6884Sszhou  * Here are the rules (based on existing usage):
32068c1b6884Sszhou  * - If title is specified, match on title only
32078c1b6884Sszhou  * - Else, match on grubdisk and module (don't care about kernel line).
32088c1b6884Sszhou  *   note that, if root_opt is non-zero, the absence of root line is
32098c1b6884Sszhou  *   considered a match.
32108c1b6884Sszhou  */
32118c1b6884Sszhou static entry_t *
32128c1b6884Sszhou find_boot_entry(menu_t *mp, char *title, char *root, char *module,
32138c1b6884Sszhou     int root_opt, int *entry_num)
32148c1b6884Sszhou {
32158c1b6884Sszhou 	int i;
32168c1b6884Sszhou 	line_t *lp;
32178c1b6884Sszhou 	entry_t *ent;
32188c1b6884Sszhou 
32198c1b6884Sszhou 	/* find matching entry */
32208c1b6884Sszhou 	for (i = 0, ent = mp->entries; ent; i++, ent = ent->next) {
32218c1b6884Sszhou 		lp = ent->start;
32228c1b6884Sszhou 
32238c1b6884Sszhou 		/* first line of entry must be bootadm comment */
32248c1b6884Sszhou 		lp = ent->start;
3225ae115bc7Smrj 		if (lp->flags != BAM_COMMENT ||
3226ae115bc7Smrj 		    strcmp(lp->arg, BAM_BOOTADM_HDR) != 0) {
32278c1b6884Sszhou 			continue;
32288c1b6884Sszhou 		}
32298c1b6884Sszhou 
32308c1b6884Sszhou 		/* advance to title line */
32318c1b6884Sszhou 		lp = lp->next;
32328c1b6884Sszhou 		if (title) {
32338c1b6884Sszhou 			if (lp->flags == BAM_TITLE && lp->arg &&
32348c1b6884Sszhou 			    strcmp(lp->arg, title) == 0)
32358c1b6884Sszhou 				break;
32368c1b6884Sszhou 			continue;	/* check title only */
32378c1b6884Sszhou 		}
32388c1b6884Sszhou 
32398c1b6884Sszhou 		lp = lp->next;	/* advance to root line */
32408c1b6884Sszhou 		if (lp == NULL || strcmp(lp->cmd, menu_cmds[ROOT_CMD]) == 0) {
32418c1b6884Sszhou 			/* root command found, match grub disk */
32428c1b6884Sszhou 			if (strcmp(lp->arg, root) != 0) {
32438c1b6884Sszhou 				continue;
32448c1b6884Sszhou 			}
32458c1b6884Sszhou 			lp = lp->next;	/* advance to kernel line */
32468c1b6884Sszhou 		} else {
32478c1b6884Sszhou 			/* no root command, see if root is optional */
32488c1b6884Sszhou 			if (root_opt == 0) {
32498c1b6884Sszhou 				continue;
32508c1b6884Sszhou 			}
32518c1b6884Sszhou 		}
32528c1b6884Sszhou 
32538c1b6884Sszhou 		if (lp == NULL || lp->next == NULL) {
32548c1b6884Sszhou 			continue;
32558c1b6884Sszhou 		}
32568c1b6884Sszhou 
32570d69385cSrscott 		/*
32580d69385cSrscott 		 * Check for matching module entry (failsafe or normal).  We
32590d69385cSrscott 		 * use a strncmp to match "module" or "module$", since we
32600d69385cSrscott 		 * don't know which one it should be.  If it fails to match,
32610d69385cSrscott 		 * we go around the loop again.
32620d69385cSrscott 		 */
32638c1b6884Sszhou 		lp = lp->next;	/* advance to module line */
3264ae115bc7Smrj 		if ((strncmp(lp->cmd, menu_cmds[MODULE_CMD],
3265ae115bc7Smrj 		    strlen(menu_cmds[MODULE_CMD])) != 0) ||
3266ae115bc7Smrj 		    (strcmp(lp->arg, module) != 0)) {
32678c1b6884Sszhou 			continue;
32688c1b6884Sszhou 		}
32698c1b6884Sszhou 		break;	/* match found */
32708c1b6884Sszhou 	}
32718c1b6884Sszhou 
32728c1b6884Sszhou 	*entry_num = i;
32738c1b6884Sszhou 	return (ent);
32748c1b6884Sszhou }
32758c1b6884Sszhou 
32768c1b6884Sszhou static int
32778c1b6884Sszhou update_boot_entry(menu_t *mp, char *title, char *root, char *kernel,
32788c1b6884Sszhou     char *module, int root_opt)
32798c1b6884Sszhou {
3280ae115bc7Smrj 	int i, change_kernel = 0;
32818c1b6884Sszhou 	entry_t *ent;
32828c1b6884Sszhou 	line_t *lp;
32838c1b6884Sszhou 	char linebuf[BAM_MAXLINE];
32848c1b6884Sszhou 
32858c1b6884Sszhou 	/* note: don't match on title, it's updated on upgrade */
32868c1b6884Sszhou 	ent = find_boot_entry(mp, NULL, root, module, root_opt, &i);
3287ae115bc7Smrj 	if ((ent == NULL) && (bam_direct == BAM_DIRECT_DBOOT)) {
3288ae115bc7Smrj 		/*
3289ae115bc7Smrj 		 * We may be upgrading a kernel from multiboot to
3290ae115bc7Smrj 		 * directboot.  Look for a multiboot entry.
3291ae115bc7Smrj 		 */
3292ae115bc7Smrj 		ent = find_boot_entry(mp, NULL, root, MULTI_BOOT_ARCHIVE,
3293ae115bc7Smrj 		    root_opt, &i);
3294ae115bc7Smrj 		if (ent != NULL) {
3295ae115bc7Smrj 			change_kernel = 1;
3296ae115bc7Smrj 		}
3297ae115bc7Smrj 	}
32988c1b6884Sszhou 	if (ent == NULL)
32998c1b6884Sszhou 		return (add_boot_entry(mp, title, root_opt ? NULL : root,
33008c1b6884Sszhou 		    kernel, module));
33018c1b6884Sszhou 
33028c1b6884Sszhou 	/* replace title of exiting entry and delete root line */
33038c1b6884Sszhou 	lp = ent->start;
33048c1b6884Sszhou 	lp = lp->next;	/* title line */
33058c1b6884Sszhou 	(void) snprintf(linebuf, sizeof (linebuf), "%s%s%s",
33068c1b6884Sszhou 	    menu_cmds[TITLE_CMD], menu_cmds[SEP_CMD], title);
33078c1b6884Sszhou 	free(lp->arg);
33088c1b6884Sszhou 	free(lp->line);
33098c1b6884Sszhou 	lp->arg = s_strdup(title);
33108c1b6884Sszhou 	lp->line = s_strdup(linebuf);
33118c1b6884Sszhou 
33128c1b6884Sszhou 	lp = lp->next;	/* root line */
33138c1b6884Sszhou 	if (strcmp(lp->cmd, menu_cmds[ROOT_CMD]) == 0) {
33148c1b6884Sszhou 		if (root_opt) {		/* root line not needed */
33158c1b6884Sszhou 			line_t *tmp = lp;
33168c1b6884Sszhou 			lp = lp->next;
33178c1b6884Sszhou 			unlink_line(mp, tmp);
33188c1b6884Sszhou 			line_free(tmp);
33198c1b6884Sszhou 		} else
33208c1b6884Sszhou 			lp = lp->next;
33218c1b6884Sszhou 	}
3322ae115bc7Smrj 
3323ae115bc7Smrj 	if (change_kernel) {
3324ae115bc7Smrj 		/*
3325ae115bc7Smrj 		 * We're upgrading from multiboot to directboot.
3326ae115bc7Smrj 		 */
3327ae115bc7Smrj 		if (strcmp(lp->cmd, menu_cmds[KERNEL_CMD]) == 0) {
3328ae115bc7Smrj 			(void) snprintf(linebuf, sizeof (linebuf), "%s%s%s",
3329ae115bc7Smrj 			    menu_cmds[KERNEL_DOLLAR_CMD], menu_cmds[SEP_CMD],
3330ae115bc7Smrj 			    kernel);
3331ae115bc7Smrj 			free(lp->arg);
3332ae115bc7Smrj 			free(lp->line);
3333ae115bc7Smrj 			lp->arg = s_strdup(kernel);
3334ae115bc7Smrj 			lp->line = s_strdup(linebuf);
3335ae115bc7Smrj 			lp = lp->next;
3336ae115bc7Smrj 		}
3337ae115bc7Smrj 		if (strcmp(lp->cmd, menu_cmds[MODULE_CMD]) == 0) {
3338ae115bc7Smrj 			(void) snprintf(linebuf, sizeof (linebuf), "%s%s%s",
3339ae115bc7Smrj 			    menu_cmds[MODULE_DOLLAR_CMD], menu_cmds[SEP_CMD],
3340ae115bc7Smrj 			    module);
3341ae115bc7Smrj 			free(lp->arg);
3342ae115bc7Smrj 			free(lp->line);
3343ae115bc7Smrj 			lp->arg = s_strdup(module);
3344ae115bc7Smrj 			lp->line = s_strdup(linebuf);
3345ae115bc7Smrj 			lp = lp->next;
3346ae115bc7Smrj 		}
3347ae115bc7Smrj 	}
33488c1b6884Sszhou 	return (i);
33498c1b6884Sszhou }
33508c1b6884Sszhou 
33517c478bd9Sstevel@tonic-gate /*ARGSUSED*/
33527c478bd9Sstevel@tonic-gate static error_t
33537c478bd9Sstevel@tonic-gate update_entry(menu_t *mp, char *menu_root, char *opt)
33547c478bd9Sstevel@tonic-gate {
33557c478bd9Sstevel@tonic-gate 	FILE *fp;
33567c478bd9Sstevel@tonic-gate 	int entry;
335760d0a590Srscott 	char *grubdisk, *title, *osdev, *osroot, *failsafe_kernel = NULL;
33588c1b6884Sszhou 	struct stat sbuf;
33598c1b6884Sszhou 	char failsafe[256];
33607c478bd9Sstevel@tonic-gate 
33617c478bd9Sstevel@tonic-gate 	assert(mp);
33627c478bd9Sstevel@tonic-gate 	assert(opt);
33637c478bd9Sstevel@tonic-gate 
33647c478bd9Sstevel@tonic-gate 	osdev = strtok(opt, ",");
33657c478bd9Sstevel@tonic-gate 	osroot = strtok(NULL, ",");
33667c478bd9Sstevel@tonic-gate 	if (osroot == NULL)
33677c478bd9Sstevel@tonic-gate 		osroot = menu_root;
33687c478bd9Sstevel@tonic-gate 	title = get_title(osroot);
33697c478bd9Sstevel@tonic-gate 
33707c478bd9Sstevel@tonic-gate 	/* translate /dev/dsk name to grub disk name */
33718c1b6884Sszhou 	fp = open_diskmap(osroot);
33727c478bd9Sstevel@tonic-gate 	if (fp == NULL) {
33737c478bd9Sstevel@tonic-gate 		bam_error(DISKMAP_FAIL, osdev);
33747c478bd9Sstevel@tonic-gate 		return (BAM_ERROR);
33757c478bd9Sstevel@tonic-gate 	}
33767c478bd9Sstevel@tonic-gate 	grubdisk = get_grubdisk(osdev, fp, menu_on_bootdev(menu_root, fp));
33777c478bd9Sstevel@tonic-gate 	(void) fclose(fp);
33787c478bd9Sstevel@tonic-gate 	if (grubdisk == NULL) {
33797c478bd9Sstevel@tonic-gate 		bam_error(DISKMAP_FAIL, osdev);
33807c478bd9Sstevel@tonic-gate 		return (BAM_ERROR);
33817c478bd9Sstevel@tonic-gate 	}
33827c478bd9Sstevel@tonic-gate 
33837c478bd9Sstevel@tonic-gate 	/* add the entry for normal Solaris */
3384ae115bc7Smrj 	if (bam_direct == BAM_DIRECT_DBOOT) {
33858c1b6884Sszhou 		entry = update_boot_entry(mp, title, grubdisk,
3386ae115bc7Smrj 		    DIRECT_BOOT_KERNEL, DIRECT_BOOT_ARCHIVE,
33878c1b6884Sszhou 		    osroot == menu_root);
3388ae115bc7Smrj 	} else {
3389ae115bc7Smrj 		entry = update_boot_entry(mp, title, grubdisk,
3390ae115bc7Smrj 		    MULTI_BOOT, MULTI_BOOT_ARCHIVE,
3391ae115bc7Smrj 		    osroot == menu_root);
3392ae115bc7Smrj 	}
33937c478bd9Sstevel@tonic-gate 
33947c478bd9Sstevel@tonic-gate 	/* add the entry for failsafe archive */
3395ae115bc7Smrj 	(void) snprintf(failsafe, sizeof (failsafe), "%s%s", osroot, MINIROOT);
3396ae115bc7Smrj 	if (stat(failsafe, &sbuf) == 0) {
339760d0a590Srscott 
339860d0a590Srscott 		/* Figure out where the kernel line should point */
339960d0a590Srscott 		(void) snprintf(failsafe, sizeof (failsafe), "%s%s", osroot,
340060d0a590Srscott 		    DIRECT_BOOT_FAILSAFE_KERNEL);
340160d0a590Srscott 		if (stat(failsafe, &sbuf) == 0) {
340260d0a590Srscott 			failsafe_kernel = DIRECT_BOOT_FAILSAFE_LINE;
340360d0a590Srscott 		} else {
340460d0a590Srscott 			(void) snprintf(failsafe, sizeof (failsafe), "%s%s",
340560d0a590Srscott 			    osroot, MULTI_BOOT_FAILSAFE);
340660d0a590Srscott 			if (stat(failsafe, &sbuf) == 0) {
340760d0a590Srscott 				failsafe_kernel = MULTI_BOOT_FAILSAFE_LINE;
340860d0a590Srscott 			}
340960d0a590Srscott 		}
341060d0a590Srscott 		if (failsafe_kernel != NULL) {
3411ae115bc7Smrj 			(void) update_boot_entry(mp, FAILSAFE_TITLE, grubdisk,
341260d0a590Srscott 			    failsafe_kernel, MINIROOT, osroot == menu_root);
341360d0a590Srscott 		} else {
341460d0a590Srscott 			bam_error(NO_FAILSAFE_KERNEL);
341560d0a590Srscott 		}
3416ae115bc7Smrj 	}
34177c478bd9Sstevel@tonic-gate 	free(grubdisk);
34187c478bd9Sstevel@tonic-gate 
34197c478bd9Sstevel@tonic-gate 	if (entry == BAM_ERROR) {
34207c478bd9Sstevel@tonic-gate 		return (BAM_ERROR);
34217c478bd9Sstevel@tonic-gate 	}
34227c478bd9Sstevel@tonic-gate 	(void) set_global(mp, menu_cmds[DEFAULT_CMD], entry);
34237c478bd9Sstevel@tonic-gate 	return (BAM_WRITE);
34247c478bd9Sstevel@tonic-gate }
34257c478bd9Sstevel@tonic-gate 
3426b610f78eSvikram static char *
3427b610f78eSvikram read_grub_root(void)
3428b610f78eSvikram {
3429b610f78eSvikram 	FILE *fp;
3430b610f78eSvikram 	struct stat sb;
3431b610f78eSvikram 	char buf[BAM_MAXLINE];
3432b610f78eSvikram 	char *rootstr;
3433b610f78eSvikram 
3434b610f78eSvikram 	if (stat(GRUB_slice, &sb) != 0) {
3435b610f78eSvikram 		bam_error(MISSING_SLICE_FILE, GRUB_slice, strerror(errno));
3436b610f78eSvikram 		return (NULL);
3437b610f78eSvikram 	}
3438b610f78eSvikram 
3439b610f78eSvikram 	if (stat(GRUB_root, &sb) != 0) {
3440b610f78eSvikram 		bam_error(MISSING_ROOT_FILE, GRUB_root, strerror(errno));
3441b610f78eSvikram 		return (NULL);
3442b610f78eSvikram 	}
3443b610f78eSvikram 
3444b610f78eSvikram 	fp = fopen(GRUB_root, "r");
3445b610f78eSvikram 	if (fp == NULL) {
3446b610f78eSvikram 		bam_error(OPEN_FAIL, GRUB_root, strerror(errno));
3447b610f78eSvikram 		return (NULL);
3448b610f78eSvikram 	}
3449b610f78eSvikram 
3450b610f78eSvikram 	if (s_fgets(buf, sizeof (buf), fp) == NULL) {
3451b610f78eSvikram 		bam_error(EMPTY_FILE, GRUB_root, strerror(errno));
3452b610f78eSvikram 		(void) fclose(fp);
3453b610f78eSvikram 		return (NULL);
3454b610f78eSvikram 	}
3455b610f78eSvikram 
3456b610f78eSvikram 	/*
3457b610f78eSvikram 	 * Copy buf here as check below may trash the buffer
3458b610f78eSvikram 	 */
3459b610f78eSvikram 	rootstr = s_strdup(buf);
3460b610f78eSvikram 
3461b610f78eSvikram 	if (s_fgets(buf, sizeof (buf), fp) != NULL) {
3462b610f78eSvikram 		bam_error(BAD_ROOT_FILE, GRUB_root);
3463b610f78eSvikram 		free(rootstr);
3464b610f78eSvikram 		rootstr = NULL;
3465b610f78eSvikram 	}
3466b610f78eSvikram 
3467b610f78eSvikram 	(void) fclose(fp);
3468b610f78eSvikram 
3469b610f78eSvikram 	return (rootstr);
3470b610f78eSvikram }
3471b610f78eSvikram 
34728c1b6884Sszhou static void
3473ae115bc7Smrj save_default_entry(menu_t *mp, const char *which)
34748c1b6884Sszhou {
34758c1b6884Sszhou 	int lineNum, entryNum;
34768c1b6884Sszhou 	int entry = 0;	/* default is 0 */
34778c1b6884Sszhou 	char linebuf[BAM_MAXLINE];
34788c1b6884Sszhou 	line_t *lp = mp->curdefault;
34798c1b6884Sszhou 
3480bff269d0Svikram 	if (mp->start) {
3481bff269d0Svikram 		lineNum = mp->end->lineNum;
3482bff269d0Svikram 		entryNum = mp->end->entryNum;
3483bff269d0Svikram 	} else {
3484bff269d0Svikram 		lineNum = LINE_INIT;
3485bff269d0Svikram 		entryNum = ENTRY_INIT;
3486bff269d0Svikram 	}
3487bff269d0Svikram 
34888c1b6884Sszhou 	if (lp)
34898c1b6884Sszhou 		entry = s_strtol(lp->arg);
34908c1b6884Sszhou 
3491ae115bc7Smrj 	(void) snprintf(linebuf, sizeof (linebuf), "#%s%d", which, entry);
34928c1b6884Sszhou 	line_parser(mp, linebuf, &lineNum, &entryNum);
34938c1b6884Sszhou }
34948c1b6884Sszhou 
34958c1b6884Sszhou static void
3496ae115bc7Smrj restore_default_entry(menu_t *mp, const char *which, line_t *lp)
34978c1b6884Sszhou {
34988c1b6884Sszhou 	int entry;
34998c1b6884Sszhou 	char *str;
35008c1b6884Sszhou 
35018c1b6884Sszhou 	if (lp == NULL)
35028c1b6884Sszhou 		return;		/* nothing to restore */
35038c1b6884Sszhou 
3504ae115bc7Smrj 	str = lp->arg + strlen(which);
35058c1b6884Sszhou 	entry = s_strtol(str);
35068c1b6884Sszhou 	(void) set_global(mp, menu_cmds[DEFAULT_CMD], entry);
35078c1b6884Sszhou 
35088c1b6884Sszhou 	/* delete saved old default line */
35098c1b6884Sszhou 	unlink_line(mp, lp);
35108c1b6884Sszhou 	line_free(lp);
35118c1b6884Sszhou }
35128c1b6884Sszhou 
35137c478bd9Sstevel@tonic-gate /*
35147c478bd9Sstevel@tonic-gate  * This function is for supporting reboot with args.
35157c478bd9Sstevel@tonic-gate  * The opt value can be:
35167c478bd9Sstevel@tonic-gate  * NULL		delete temp entry, if present
35177c478bd9Sstevel@tonic-gate  * entry=#	switches default entry to 1
35187c478bd9Sstevel@tonic-gate  * else		treated as boot-args and setup a temperary menu entry
35197c478bd9Sstevel@tonic-gate  *		and make it the default
35207c478bd9Sstevel@tonic-gate  */
35217c478bd9Sstevel@tonic-gate #define	REBOOT_TITLE	"Solaris_reboot_transient"
35227c478bd9Sstevel@tonic-gate 
35238c1b6884Sszhou /*ARGSUSED*/
35247c478bd9Sstevel@tonic-gate static error_t
35257c478bd9Sstevel@tonic-gate update_temp(menu_t *mp, char *menupath, char *opt)
35267c478bd9Sstevel@tonic-gate {
35277c478bd9Sstevel@tonic-gate 	int entry;
3528455710d3Srscott 	char *grubdisk, *rootdev, *path, *opt_ptr;
3529ae115bc7Smrj 	char kernbuf[BUFSIZ];
3530ae115bc7Smrj 	char args_buf[BUFSIZ];
3531b610f78eSvikram 	struct stat sb;
35327c478bd9Sstevel@tonic-gate 
35337c478bd9Sstevel@tonic-gate 	assert(mp);
35347c478bd9Sstevel@tonic-gate 
35358c1b6884Sszhou 	/* If no option, delete exiting reboot menu entry */
35368c1b6884Sszhou 	if (opt == NULL) {
35378c1b6884Sszhou 		entry_t *ent = find_boot_entry(mp, REBOOT_TITLE, NULL, NULL,
35388c1b6884Sszhou 		    0, &entry);
35398c1b6884Sszhou 		if (ent == NULL)	/* not found is ok */
35408c1b6884Sszhou 			return (BAM_SUCCESS);
35418c1b6884Sszhou 		(void) do_delete(mp, entry);
3542ae115bc7Smrj 		restore_default_entry(mp, BAM_OLDDEF, mp->olddefault);
3543ae115bc7Smrj 		mp->olddefault = NULL;
35448c1b6884Sszhou 		return (BAM_WRITE);
35458c1b6884Sszhou 	}
35468c1b6884Sszhou 
35478c1b6884Sszhou 	/* if entry= is specified, set the default entry */
35488c1b6884Sszhou 	if (strncmp(opt, "entry=", strlen("entry=")) == 0 &&
35497c478bd9Sstevel@tonic-gate 	    selector(mp, opt, &entry, NULL) == BAM_SUCCESS) {
35507c478bd9Sstevel@tonic-gate 		/* this is entry=# option */
35517c478bd9Sstevel@tonic-gate 		return (set_global(mp, menu_cmds[DEFAULT_CMD], entry));
35527c478bd9Sstevel@tonic-gate 	}
35537c478bd9Sstevel@tonic-gate 
35547c478bd9Sstevel@tonic-gate 	/*
35557c478bd9Sstevel@tonic-gate 	 * add a new menu entry base on opt and make it the default
3556b610f78eSvikram 	 */
3557b610f78eSvikram 	grubdisk = NULL;
3558b610f78eSvikram 	if (stat(GRUB_slice, &sb) != 0) {
3559b610f78eSvikram 		/*
35607c478bd9Sstevel@tonic-gate 		 * 1. First get root disk name from mnttab
35617c478bd9Sstevel@tonic-gate 		 * 2. Translate disk name to grub name
35627c478bd9Sstevel@tonic-gate 		 * 3. Add the new menu entry
35637c478bd9Sstevel@tonic-gate 		 */
35647c478bd9Sstevel@tonic-gate 		rootdev = get_special("/");
35657c478bd9Sstevel@tonic-gate 		if (rootdev) {
35667c478bd9Sstevel@tonic-gate 			grubdisk = os_to_grubdisk(rootdev, 1);
35677c478bd9Sstevel@tonic-gate 			free(rootdev);
35687c478bd9Sstevel@tonic-gate 		}
3569b610f78eSvikram 	} else {
3570b610f78eSvikram 		/*
3571b610f78eSvikram 		 * This is an LU BE. The GRUB_root file
3572b610f78eSvikram 		 * contains entry for GRUB's "root" cmd.
3573b610f78eSvikram 		 */
3574b610f78eSvikram 		grubdisk = read_grub_root();
3575b610f78eSvikram 	}
35767c478bd9Sstevel@tonic-gate 	if (grubdisk == NULL) {
3577b610f78eSvikram 		bam_error(REBOOT_WITH_ARGS_FAILED);
35787c478bd9Sstevel@tonic-gate 		return (BAM_ERROR);
35797c478bd9Sstevel@tonic-gate 	}
35807c478bd9Sstevel@tonic-gate 
35817c478bd9Sstevel@tonic-gate 	/* add an entry for Solaris reboot */
3582ae115bc7Smrj 	if (bam_direct == BAM_DIRECT_DBOOT) {
3583ae115bc7Smrj 		if (opt[0] == '-') {
3584ae115bc7Smrj 			/* It's an option - first see if boot-file is set */
3585ae115bc7Smrj 			if (set_kernel(mp, KERNEL_CMD, NULL, kernbuf, BUFSIZ)
3586ae115bc7Smrj 			    != BAM_SUCCESS)
3587ae115bc7Smrj 				return (BAM_ERROR);
3588ae115bc7Smrj 			if (kernbuf[0] == '\0')
3589ae115bc7Smrj 				(void) strncpy(kernbuf, DIRECT_BOOT_KERNEL,
3590ae115bc7Smrj 				    BUFSIZ);
3591ae115bc7Smrj 			(void) strlcat(kernbuf, " ", BUFSIZ);
3592ae115bc7Smrj 			(void) strlcat(kernbuf, opt, BUFSIZ);
3593ae115bc7Smrj 		} else if (opt[0] == '/') {
3594455710d3Srscott 			/* It's a full path, so write it out. */
3595ae115bc7Smrj 			(void) strlcpy(kernbuf, opt, BUFSIZ);
3596455710d3Srscott 
3597455710d3Srscott 			/*
3598455710d3Srscott 			 * If someone runs:
3599455710d3Srscott 			 *
3600455710d3Srscott 			 *	# eeprom boot-args='-kd'
3601455710d3Srscott 			 *	# reboot /platform/i86pc/kernel/unix
3602455710d3Srscott 			 *
3603455710d3Srscott 			 * we want to use the boot-args as part of the boot
3604455710d3Srscott 			 * line.  On the other hand, if someone runs:
3605455710d3Srscott 			 *
3606455710d3Srscott 			 *	# reboot "/platform/i86pc/kernel/unix -kd"
3607455710d3Srscott 			 *
3608455710d3Srscott 			 * we don't need to mess with boot-args.  If there's
3609455710d3Srscott 			 * no space in the options string, assume we're in the
3610455710d3Srscott 			 * first case.
3611455710d3Srscott 			 */
3612455710d3Srscott 			if (strchr(opt, ' ') == NULL) {
3613455710d3Srscott 				if (set_kernel(mp, ARGS_CMD, NULL, args_buf,
3614455710d3Srscott 				    BUFSIZ) != BAM_SUCCESS)
3615455710d3Srscott 					return (BAM_ERROR);
3616455710d3Srscott 
3617455710d3Srscott 				if (args_buf[0] != '\0') {
3618455710d3Srscott 					(void) strlcat(kernbuf, " ", BUFSIZ);
3619455710d3Srscott 					(void) strlcat(kernbuf, args_buf,
3620455710d3Srscott 					    BUFSIZ);
3621455710d3Srscott 				}
3622455710d3Srscott 			}
3623ae115bc7Smrj 		} else {
3624455710d3Srscott 			/*
3625455710d3Srscott 			 * It may be a partial path, or it may be a partial
3626455710d3Srscott 			 * path followed by options.  Assume that only options
3627455710d3Srscott 			 * follow a space.  If someone sends us a kernel path
3628455710d3Srscott 			 * that includes a space, they deserve to be broken.
3629455710d3Srscott 			 */
3630455710d3Srscott 			opt_ptr = strchr(opt, ' ');
3631455710d3Srscott 			if (opt_ptr != NULL) {
3632455710d3Srscott 				*opt_ptr = '\0';
3633455710d3Srscott 			}
3634455710d3Srscott 
3635ae115bc7Smrj 			path = expand_path(opt);
3636ae115bc7Smrj 			if (path != NULL) {
3637ae115bc7Smrj 				(void) strlcpy(kernbuf, path, BUFSIZ);
3638ae115bc7Smrj 				free(path);
3639455710d3Srscott 
3640455710d3Srscott 				/*
3641455710d3Srscott 				 * If there were options given, use those.
3642455710d3Srscott 				 * Otherwise, copy over the default options.
3643455710d3Srscott 				 */
3644455710d3Srscott 				if (opt_ptr != NULL) {
3645455710d3Srscott 					/* Restore the space in opt string */
3646455710d3Srscott 					*opt_ptr = ' ';
3647455710d3Srscott 					(void) strlcat(kernbuf, opt_ptr,
3648455710d3Srscott 					    BUFSIZ);
3649455710d3Srscott 				} else {
3650ae115bc7Smrj 					if (set_kernel(mp, ARGS_CMD, NULL,
3651ae115bc7Smrj 					    args_buf, BUFSIZ) != BAM_SUCCESS)
3652ae115bc7Smrj 						return (BAM_ERROR);
3653ae115bc7Smrj 
3654ae115bc7Smrj 					if (args_buf[0] != '\0') {
3655ae115bc7Smrj 						(void) strlcat(kernbuf, " ",
3656ae115bc7Smrj 						    BUFSIZ);
3657ae115bc7Smrj 						(void) strlcat(kernbuf,
3658ae115bc7Smrj 						    args_buf, BUFSIZ);
3659ae115bc7Smrj 					}
3660ae115bc7Smrj 				}
3661455710d3Srscott 			} else {
3662455710d3Srscott 				bam_error(UNKNOWN_KERNEL, opt);
3663455710d3Srscott 				bam_print_stderr(UNKNOWN_KERNEL_REBOOT);
3664455710d3Srscott 				return (BAM_ERROR);
3665ae115bc7Smrj 			}
3666ae115bc7Smrj 		}
36677c478bd9Sstevel@tonic-gate 		entry = add_boot_entry(mp, REBOOT_TITLE, grubdisk, kernbuf,
3668ae115bc7Smrj 		    NULL);
3669ae115bc7Smrj 	} else {
3670ae115bc7Smrj 		(void) snprintf(kernbuf, sizeof (kernbuf), "%s %s",
3671ae115bc7Smrj 		    MULTI_BOOT, opt);
3672ae115bc7Smrj 		entry = add_boot_entry(mp, REBOOT_TITLE, grubdisk, kernbuf,
3673ae115bc7Smrj 		    MULTI_BOOT_ARCHIVE);
3674ae115bc7Smrj 	}
36757c478bd9Sstevel@tonic-gate 	free(grubdisk);
36767c478bd9Sstevel@tonic-gate 
36777c478bd9Sstevel@tonic-gate 	if (entry == BAM_ERROR) {
3678b610f78eSvikram 		bam_error(REBOOT_WITH_ARGS_FAILED);
36797c478bd9Sstevel@tonic-gate 		return (BAM_ERROR);
36807c478bd9Sstevel@tonic-gate 	}
36818c1b6884Sszhou 
3682ae115bc7Smrj 	save_default_entry(mp, BAM_OLDDEF);
36837c478bd9Sstevel@tonic-gate 	(void) set_global(mp, menu_cmds[DEFAULT_CMD], entry);
36847c478bd9Sstevel@tonic-gate 	return (BAM_WRITE);
36857c478bd9Sstevel@tonic-gate }
36867c478bd9Sstevel@tonic-gate 
36877c478bd9Sstevel@tonic-gate static error_t
36887c478bd9Sstevel@tonic-gate set_global(menu_t *mp, char *globalcmd, int val)
36897c478bd9Sstevel@tonic-gate {
36907c478bd9Sstevel@tonic-gate 	line_t *lp, *found, *last;
36917c478bd9Sstevel@tonic-gate 	char *cp, *str;
36927c478bd9Sstevel@tonic-gate 	char prefix[BAM_MAXLINE];
36937c478bd9Sstevel@tonic-gate 	size_t len;
36947c478bd9Sstevel@tonic-gate 
36957c478bd9Sstevel@tonic-gate 	assert(mp);
36967c478bd9Sstevel@tonic-gate 	assert(globalcmd);
36977c478bd9Sstevel@tonic-gate 
3698b610f78eSvikram 	if (strcmp(globalcmd, menu_cmds[DEFAULT_CMD]) == 0) {
3699b610f78eSvikram 		if (val < 0 || mp->end == NULL || val > mp->end->entryNum) {
3700b610f78eSvikram 			(void) snprintf(prefix, sizeof (prefix), "%d", val);
3701b610f78eSvikram 			bam_error(INVALID_ENTRY, prefix);
3702b610f78eSvikram 			return (BAM_ERROR);
3703b610f78eSvikram 		}
3704b610f78eSvikram 	}
3705b610f78eSvikram 
37067c478bd9Sstevel@tonic-gate 	found = last = NULL;
37077c478bd9Sstevel@tonic-gate 	for (lp = mp->start; lp; lp = lp->next) {
37087c478bd9Sstevel@tonic-gate 		if (lp->flags != BAM_GLOBAL)
37097c478bd9Sstevel@tonic-gate 			continue;
37107c478bd9Sstevel@tonic-gate 
37117c478bd9Sstevel@tonic-gate 		last = lp; /* track the last global found */
37127c478bd9Sstevel@tonic-gate 
37137c478bd9Sstevel@tonic-gate 		if (lp->cmd == NULL) {
37147c478bd9Sstevel@tonic-gate 			bam_error(NO_CMD, lp->lineNum);
37157c478bd9Sstevel@tonic-gate 			continue;
37167c478bd9Sstevel@tonic-gate 		}
37177c478bd9Sstevel@tonic-gate 		if (strcmp(globalcmd, lp->cmd) != 0)
37187c478bd9Sstevel@tonic-gate 			continue;
37197c478bd9Sstevel@tonic-gate 
37207c478bd9Sstevel@tonic-gate 		if (found) {
37217c478bd9Sstevel@tonic-gate 			bam_error(DUP_CMD, globalcmd, lp->lineNum, bam_root);
37227c478bd9Sstevel@tonic-gate 		}
37237c478bd9Sstevel@tonic-gate 		found = lp;
37247c478bd9Sstevel@tonic-gate 	}
37257c478bd9Sstevel@tonic-gate 
37267c478bd9Sstevel@tonic-gate 	if (found == NULL) {
37277c478bd9Sstevel@tonic-gate 		lp = s_calloc(1, sizeof (line_t));
37287c478bd9Sstevel@tonic-gate 		if (last == NULL) {
37297c478bd9Sstevel@tonic-gate 			lp->next = mp->start;
37307c478bd9Sstevel@tonic-gate 			mp->start = lp;
37317c478bd9Sstevel@tonic-gate 			mp->end = (mp->end) ? mp->end : lp;
37327c478bd9Sstevel@tonic-gate 		} else {
37337c478bd9Sstevel@tonic-gate 			lp->next = last->next;
37347c478bd9Sstevel@tonic-gate 			last->next = lp;
37357c478bd9Sstevel@tonic-gate 			if (lp->next == NULL)
37367c478bd9Sstevel@tonic-gate 				mp->end = lp;
37377c478bd9Sstevel@tonic-gate 		}
37387c478bd9Sstevel@tonic-gate 		lp->flags = BAM_GLOBAL; /* other fields not needed for writes */
37397c478bd9Sstevel@tonic-gate 		len = strlen(globalcmd) + strlen(menu_cmds[SEP_CMD]);
37407c478bd9Sstevel@tonic-gate 		len += 10;	/* val < 10 digits */
37417c478bd9Sstevel@tonic-gate 		lp->line = s_calloc(1, len);
37427c478bd9Sstevel@tonic-gate 		(void) snprintf(lp->line, len, "%s%s%d",
37437c478bd9Sstevel@tonic-gate 		    globalcmd, menu_cmds[SEP_CMD], val);
37447c478bd9Sstevel@tonic-gate 		return (BAM_WRITE);
37457c478bd9Sstevel@tonic-gate 	}
37467c478bd9Sstevel@tonic-gate 
37477c478bd9Sstevel@tonic-gate 	/*
37487c478bd9Sstevel@tonic-gate 	 * We are changing an existing entry. Retain any prefix whitespace,
37497c478bd9Sstevel@tonic-gate 	 * but overwrite everything else. This preserves tabs added for
37507c478bd9Sstevel@tonic-gate 	 * readability.
37517c478bd9Sstevel@tonic-gate 	 */
37527c478bd9Sstevel@tonic-gate 	str = found->line;
37537c478bd9Sstevel@tonic-gate 	cp = prefix;
37547c478bd9Sstevel@tonic-gate 	while (*str == ' ' || *str == '\t')
37557c478bd9Sstevel@tonic-gate 		*(cp++) = *(str++);
37567c478bd9Sstevel@tonic-gate 	*cp = '\0'; /* Terminate prefix */
37577c478bd9Sstevel@tonic-gate 	len = strlen(prefix) + strlen(globalcmd);
37587c478bd9Sstevel@tonic-gate 	len += strlen(menu_cmds[SEP_CMD]) + 10;
37597c478bd9Sstevel@tonic-gate 
37607c478bd9Sstevel@tonic-gate 	free(found->line);
37617c478bd9Sstevel@tonic-gate 	found->line = s_calloc(1, len);
37627c478bd9Sstevel@tonic-gate 	(void) snprintf(found->line, len,
37637c478bd9Sstevel@tonic-gate 	    "%s%s%s%d", prefix, globalcmd, menu_cmds[SEP_CMD], val);
37647c478bd9Sstevel@tonic-gate 
37657c478bd9Sstevel@tonic-gate 	return (BAM_WRITE); /* need a write to menu */
37667c478bd9Sstevel@tonic-gate }
37677c478bd9Sstevel@tonic-gate 
3768ae115bc7Smrj /*
3769ae115bc7Smrj  * partial_path may be anything like "kernel/unix" or "kmdb".  Try to
3770455710d3Srscott  * expand it to a full unix path.  The calling function is expected to
3771455710d3Srscott  * output a message if an error occurs and NULL is returned.
3772ae115bc7Smrj  */
3773ae115bc7Smrj static char *
3774ae115bc7Smrj expand_path(const char *partial_path)
3775ae115bc7Smrj {
3776ae115bc7Smrj 	int new_path_len;
3777ae115bc7Smrj 	char *new_path, new_path2[PATH_MAX];
3778ae115bc7Smrj 	struct stat sb;
3779ae115bc7Smrj 
3780ae115bc7Smrj 	new_path_len = strlen(partial_path) + 64;
3781ae115bc7Smrj 	new_path = s_calloc(1, new_path_len);
3782ae115bc7Smrj 
3783ae115bc7Smrj 	/* First, try the simplest case - something like "kernel/unix" */
3784ae115bc7Smrj 	(void) snprintf(new_path, new_path_len, "/platform/i86pc/%s",
3785ae115bc7Smrj 	    partial_path);
3786ae115bc7Smrj 	if (stat(new_path, &sb) == 0) {
3787ae115bc7Smrj 		return (new_path);
3788ae115bc7Smrj 	}
3789ae115bc7Smrj 
3790ae115bc7Smrj 	if (strcmp(partial_path, "kmdb") == 0) {
3791ae115bc7Smrj 		(void) snprintf(new_path, new_path_len, "%s -k",
3792ae115bc7Smrj 		    DIRECT_BOOT_KERNEL);
3793ae115bc7Smrj 		return (new_path);
3794ae115bc7Smrj 	}
3795ae115bc7Smrj 
3796ae115bc7Smrj 	/*
3797ae115bc7Smrj 	 * We've quickly reached unsupported usage.  Try once more to
3798ae115bc7Smrj 	 * see if we were just given a glom name.
3799ae115bc7Smrj 	 */
3800ae115bc7Smrj 	(void) snprintf(new_path, new_path_len, "/platform/i86pc/%s/unix",
3801ae115bc7Smrj 	    partial_path);
3802ae115bc7Smrj 	(void) snprintf(new_path2, PATH_MAX, "/platform/i86pc/%s/amd64/unix",
3803ae115bc7Smrj 	    partial_path);
3804ae115bc7Smrj 	if (stat(new_path, &sb) == 0) {
3805ae115bc7Smrj 		if (stat(new_path2, &sb) == 0) {
3806ae115bc7Smrj 			/*
3807ae115bc7Smrj 			 * We matched both, so we actually
3808ae115bc7Smrj 			 * want to write the $ISADIR version.
3809ae115bc7Smrj 			 */
3810ae115bc7Smrj 			(void) snprintf(new_path, new_path_len,
3811ae115bc7Smrj 			    "/platform/i86pc/kernel/%s/$ISADIR/unix",
3812ae115bc7Smrj 			    partial_path);
3813ae115bc7Smrj 		}
3814ae115bc7Smrj 		return (new_path);
3815ae115bc7Smrj 	}
3816ae115bc7Smrj 
3817ae115bc7Smrj 	free(new_path);
3818ae115bc7Smrj 	return (NULL);
3819ae115bc7Smrj }
3820ae115bc7Smrj 
3821ae115bc7Smrj /*
3822ae115bc7Smrj  * The kernel cmd and arg have been changed, so
3823ae115bc7Smrj  * check whether the archive line needs to change.
3824ae115bc7Smrj  */
3825ae115bc7Smrj static void
3826ae115bc7Smrj set_archive_line(entry_t *entryp, line_t *kernelp)
3827ae115bc7Smrj {
3828ae115bc7Smrj 	line_t *lp = entryp->start;
3829ae115bc7Smrj 	char *new_archive;
3830ae115bc7Smrj 	menu_cmd_t m_cmd;
3831ae115bc7Smrj 
3832ae115bc7Smrj 	for (; lp != NULL; lp = lp->next) {
3833ae115bc7Smrj 		if (strncmp(lp->cmd, menu_cmds[MODULE_CMD],
3834ae115bc7Smrj 		    sizeof (menu_cmds[MODULE_CMD]) - 1) == 0) {
3835ae115bc7Smrj 			break;
3836ae115bc7Smrj 		}
3837ae115bc7Smrj 		if (lp == entryp->end)
3838ae115bc7Smrj 			return;
3839ae115bc7Smrj 	}
3840ae115bc7Smrj 	if (lp == NULL)
3841ae115bc7Smrj 		return;
3842ae115bc7Smrj 
3843ae115bc7Smrj 	if (strstr(kernelp->arg, "$ISADIR") != NULL) {
3844ae115bc7Smrj 		new_archive = DIRECT_BOOT_ARCHIVE;
3845ae115bc7Smrj 		m_cmd = MODULE_DOLLAR_CMD;
3846ae115bc7Smrj 	} else if (strstr(kernelp->arg, "amd64") != NULL) {
3847ae115bc7Smrj 		new_archive = DIRECT_BOOT_ARCHIVE_64;
3848ae115bc7Smrj 		m_cmd = MODULE_CMD;
3849ae115bc7Smrj 	} else {
3850ae115bc7Smrj 		new_archive = DIRECT_BOOT_ARCHIVE_32;
3851ae115bc7Smrj 		m_cmd = MODULE_CMD;
3852ae115bc7Smrj 	}
3853ae115bc7Smrj 
3854ae115bc7Smrj 	if (strcmp(lp->arg, new_archive) == 0)
3855ae115bc7Smrj 		return;
3856ae115bc7Smrj 
3857ae115bc7Smrj 	if (strcmp(lp->cmd, menu_cmds[m_cmd]) != 0) {
3858ae115bc7Smrj 		free(lp->cmd);
3859ae115bc7Smrj 		lp->cmd = s_strdup(menu_cmds[m_cmd]);
3860ae115bc7Smrj 	}
3861ae115bc7Smrj 
3862ae115bc7Smrj 	free(lp->arg);
3863ae115bc7Smrj 	lp->arg = s_strdup(new_archive);
3864ae115bc7Smrj 	update_line(lp);
3865ae115bc7Smrj }
3866ae115bc7Smrj 
3867ae115bc7Smrj /*
3868ae115bc7Smrj  * Title for an entry to set properties that once went in bootenv.rc.
3869ae115bc7Smrj  */
3870ae115bc7Smrj #define	BOOTENV_RC_TITLE	"Solaris bootenv rc"
3871ae115bc7Smrj 
3872ae115bc7Smrj /*
3873ae115bc7Smrj  * If path is NULL, return the kernel (optnum == KERNEL_CMD) or arguments
3874ae115bc7Smrj  * (optnum == ARGS_CMD) in the argument buf.  If path is a zero-length
3875ae115bc7Smrj  * string, reset the value to the default.  If path is a non-zero-length
3876ae115bc7Smrj  * string, set the kernel or arguments.
3877ae115bc7Smrj  */
3878ae115bc7Smrj static error_t
3879ae115bc7Smrj set_kernel(menu_t *mp, menu_cmd_t optnum, char *path, char *buf, size_t bufsize)
3880ae115bc7Smrj {
3881ae115bc7Smrj 	int entryNum, rv = BAM_SUCCESS, free_new_path = 0;
3882ae115bc7Smrj 	entry_t *entryp;
3883ae115bc7Smrj 	line_t *ptr, *kernelp;
3884ae115bc7Smrj 	char *new_arg, *old_args, *space;
3885ae115bc7Smrj 	char *grubdisk, *rootdev, *new_path;
3886ae115bc7Smrj 	char old_space;
3887ae115bc7Smrj 	size_t old_kernel_len, new_str_len;
3888ae115bc7Smrj 	struct stat sb;
3889ae115bc7Smrj 
3890ae115bc7Smrj 	assert(bufsize > 0);
3891ae115bc7Smrj 
3892ae115bc7Smrj 	ptr = kernelp = NULL;
3893ae115bc7Smrj 	new_arg = old_args = space = NULL;
3894ae115bc7Smrj 	grubdisk = rootdev = new_path = NULL;
3895ae115bc7Smrj 	buf[0] = '\0';
3896ae115bc7Smrj 
3897ae115bc7Smrj 	if (bam_direct != BAM_DIRECT_DBOOT) {
3898ae115bc7Smrj 		bam_error(NOT_DBOOT, optnum == KERNEL_CMD ? "kernel" : "args");
3899ae115bc7Smrj 		return (BAM_ERROR);
3900ae115bc7Smrj 	}
3901ae115bc7Smrj 
3902ae115bc7Smrj 	/*
3903ae115bc7Smrj 	 * If a user changed the default entry to a non-bootadm controlled
3904ae115bc7Smrj 	 * one, we don't want to mess with it.  Just print an error and
3905ae115bc7Smrj 	 * return.
3906ae115bc7Smrj 	 */
3907ae115bc7Smrj 	if (mp->curdefault) {
3908ae115bc7Smrj 		entryNum = s_strtol(mp->curdefault->arg);
3909ae115bc7Smrj 		for (entryp = mp->entries; entryp; entryp = entryp->next) {
3910ae115bc7Smrj 			if (entryp->entryNum == entryNum)
3911ae115bc7Smrj 				break;
3912ae115bc7Smrj 		}
3913ae115bc7Smrj 		if ((entryp != NULL) &&
3914ae115bc7Smrj 		    ((entryp->flags & (BAM_ENTRY_BOOTADM|BAM_ENTRY_LU)) == 0)) {
3915ae115bc7Smrj 			bam_error(DEFAULT_NOT_BAM);
3916ae115bc7Smrj 			return (BAM_ERROR);
3917ae115bc7Smrj 		}
3918ae115bc7Smrj 	}
3919ae115bc7Smrj 
3920ae115bc7Smrj 	entryNum = -1;
3921ae115bc7Smrj 	entryp = find_boot_entry(mp, BOOTENV_RC_TITLE, NULL, NULL, 0,
3922ae115bc7Smrj 	    &entryNum);
3923ae115bc7Smrj 
3924ae115bc7Smrj 	if (entryp != NULL) {
3925ae115bc7Smrj 		for (ptr = entryp->start; ptr && ptr != entryp->end;
3926ae115bc7Smrj 		    ptr = ptr->next) {
3927ae115bc7Smrj 			if (strncmp(ptr->cmd, menu_cmds[KERNEL_CMD],
3928ae115bc7Smrj 			    sizeof (menu_cmds[KERNEL_CMD]) - 1) == 0) {
3929ae115bc7Smrj 				kernelp = ptr;
3930ae115bc7Smrj 				break;
3931ae115bc7Smrj 			}
3932ae115bc7Smrj 		}
3933ae115bc7Smrj 		if (kernelp == NULL) {
3934ae115bc7Smrj 			bam_error(NO_KERNEL, entryNum);
3935ae115bc7Smrj 			return (BAM_ERROR);
3936ae115bc7Smrj 		}
3937ae115bc7Smrj 
3938ae115bc7Smrj 		old_kernel_len = strcspn(kernelp->arg, " \t");
3939ae115bc7Smrj 		space = old_args = kernelp->arg + old_kernel_len;
3940ae115bc7Smrj 		while ((*old_args == ' ') || (*old_args == '\t'))
3941ae115bc7Smrj 			old_args++;
3942ae115bc7Smrj 	}
3943ae115bc7Smrj 
3944ae115bc7Smrj 	if (path == NULL) {
3945ae115bc7Smrj 		/* Simply report what was found */
3946ae115bc7Smrj 		if (kernelp == NULL)
3947ae115bc7Smrj 			return (BAM_SUCCESS);
3948ae115bc7Smrj 
3949ae115bc7Smrj 		if (optnum == ARGS_CMD) {
3950ae115bc7Smrj 			if (old_args[0] != '\0')
3951ae115bc7Smrj 				(void) strlcpy(buf, old_args, bufsize);
3952ae115bc7Smrj 		} else {
3953ae115bc7Smrj 			/*
3954ae115bc7Smrj 			 * We need to print the kernel, so we just turn the
3955ae115bc7Smrj 			 * first space into a '\0' and print the beginning.
3956ae115bc7Smrj 			 * We don't print anything if it's the default kernel.
3957ae115bc7Smrj 			 */
3958ae115bc7Smrj 			old_space = *space;
3959ae115bc7Smrj 			*space = '\0';
3960ae115bc7Smrj 			if (strcmp(kernelp->arg, DIRECT_BOOT_KERNEL) != 0)
3961ae115bc7Smrj 				(void) strlcpy(buf, kernelp->arg, bufsize);
3962ae115bc7Smrj 			*space = old_space;
3963ae115bc7Smrj 		}
3964ae115bc7Smrj 		return (BAM_SUCCESS);
3965ae115bc7Smrj 	}
3966ae115bc7Smrj 
3967ae115bc7Smrj 	/*
3968ae115bc7Smrj 	 * First, check if we're resetting an entry to the default.
3969ae115bc7Smrj 	 */
3970ae115bc7Smrj 	if ((path[0] == '\0') ||
3971ae115bc7Smrj 	    ((optnum == KERNEL_CMD) &&
3972ae115bc7Smrj 	    (strcmp(path, DIRECT_BOOT_KERNEL) == 0))) {
3973ae115bc7Smrj 		if ((entryp == NULL) || (kernelp == NULL)) {
3974ae115bc7Smrj 			/* No previous entry, it's already the default */
3975ae115bc7Smrj 			return (BAM_SUCCESS);
3976ae115bc7Smrj 		}
3977ae115bc7Smrj 
3978ae115bc7Smrj 		/*
3979ae115bc7Smrj 		 * Check if we can delete the entry.  If we're resetting the
3980ae115bc7Smrj 		 * kernel command, and the args is already empty, or if we're
3981ae115bc7Smrj 		 * resetting the args command, and the kernel is already the
3982ae115bc7Smrj 		 * default, we can restore the old default and delete the entry.
3983ae115bc7Smrj 		 */
3984ae115bc7Smrj 		if (((optnum == KERNEL_CMD) &&
3985ae115bc7Smrj 		    ((old_args == NULL) || (old_args[0] == '\0'))) ||
3986ae115bc7Smrj 		    ((optnum == ARGS_CMD) &&
3987ae115bc7Smrj 		    (strncmp(kernelp->arg, DIRECT_BOOT_KERNEL,
3988ae115bc7Smrj 		    sizeof (DIRECT_BOOT_KERNEL) - 1) == 0))) {
3989ae115bc7Smrj 			kernelp = NULL;
3990ae115bc7Smrj 			(void) do_delete(mp, entryNum);
3991ae115bc7Smrj 			restore_default_entry(mp, BAM_OLD_RC_DEF,
3992ae115bc7Smrj 			    mp->old_rc_default);
3993ae115bc7Smrj 			mp->old_rc_default = NULL;
3994ae115bc7Smrj 			rv = BAM_WRITE;
3995ae115bc7Smrj 			goto done;
3996ae115bc7Smrj 		}
3997ae115bc7Smrj 
3998ae115bc7Smrj 		if (optnum == KERNEL_CMD) {
3999ae115bc7Smrj 			/*
4000ae115bc7Smrj 			 * At this point, we've already checked that old_args
4001ae115bc7Smrj 			 * and entryp are valid pointers.  The "+ 2" is for
4002ae115bc7Smrj 			 * a space a the string termination character.
4003ae115bc7Smrj 			 */
4004ae115bc7Smrj 			new_str_len = (sizeof (DIRECT_BOOT_KERNEL) - 1) +
4005ae115bc7Smrj 			    strlen(old_args) + 2;
4006ae115bc7Smrj 			new_arg = s_calloc(1, new_str_len);
4007ae115bc7Smrj 			(void) snprintf(new_arg, new_str_len, "%s %s",
4008ae115bc7Smrj 			    DIRECT_BOOT_KERNEL, old_args);
4009ae115bc7Smrj 			free(kernelp->arg);
4010ae115bc7Smrj 			kernelp->arg = new_arg;
4011ae115bc7Smrj 
4012ae115bc7Smrj 			/*
4013ae115bc7Smrj 			 * We have changed the kernel line, so we may need
4014ae115bc7Smrj 			 * to update the archive line as well.
4015ae115bc7Smrj 			 */
4016ae115bc7Smrj 			set_archive_line(entryp, kernelp);
4017ae115bc7Smrj 		} else {
4018ae115bc7Smrj 			/*
4019ae115bc7Smrj 			 * We're resetting the boot args to nothing, so
4020ae115bc7Smrj 			 * we only need to copy the kernel.  We've already
4021ae115bc7Smrj 			 * checked that the kernel is not the default.
4022ae115bc7Smrj 			 */
4023ae115bc7Smrj 			new_arg = s_calloc(1, old_kernel_len + 1);
4024ae115bc7Smrj 			(void) snprintf(new_arg, old_kernel_len + 1, "%s",
4025ae115bc7Smrj 			    kernelp->arg);
4026ae115bc7Smrj 			free(kernelp->arg);
4027ae115bc7Smrj 			kernelp->arg = new_arg;
4028ae115bc7Smrj 		}
4029ae115bc7Smrj 		rv = BAM_WRITE;
4030ae115bc7Smrj 		goto done;
4031ae115bc7Smrj 	}
4032ae115bc7Smrj 
4033ae115bc7Smrj 	/*
4034ae115bc7Smrj 	 * Expand the kernel file to a full path, if necessary
4035ae115bc7Smrj 	 */
4036ae115bc7Smrj 	if ((optnum == KERNEL_CMD) && (path[0] != '/')) {
4037ae115bc7Smrj 		new_path = expand_path(path);
4038ae115bc7Smrj 		if (new_path == NULL) {
4039455710d3Srscott 			bam_error(UNKNOWN_KERNEL, path);
4040ae115bc7Smrj 			return (BAM_ERROR);
4041ae115bc7Smrj 		}
4042ae115bc7Smrj 		free_new_path = 1;
4043ae115bc7Smrj 	} else {
4044ae115bc7Smrj 		new_path = path;
4045ae115bc7Smrj 		free_new_path = 0;
4046ae115bc7Smrj 	}
4047ae115bc7Smrj 
4048ae115bc7Smrj 	/*
4049ae115bc7Smrj 	 * At this point, we know we're setting a new value.  First, take care
4050ae115bc7Smrj 	 * of the case where there was no previous entry.
4051ae115bc7Smrj 	 */
4052ae115bc7Smrj 	if (entryp == NULL) {
4053ae115bc7Smrj 		/* Similar to code in update_temp */
4054ae115bc7Smrj 		if (stat(GRUB_slice, &sb) != 0) {
4055ae115bc7Smrj 			/*
4056ae115bc7Smrj 			 * 1. First get root disk name from mnttab
4057ae115bc7Smrj 			 * 2. Translate disk name to grub name
4058ae115bc7Smrj 			 * 3. Add the new menu entry
4059ae115bc7Smrj 			 */
4060ae115bc7Smrj 			rootdev = get_special("/");
4061ae115bc7Smrj 			if (rootdev) {
4062ae115bc7Smrj 				grubdisk = os_to_grubdisk(rootdev, 1);
4063ae115bc7Smrj 				free(rootdev);
4064ae115bc7Smrj 			}
4065ae115bc7Smrj 		} else {
4066ae115bc7Smrj 			/*
4067ae115bc7Smrj 			 * This is an LU BE. The GRUB_root file
4068ae115bc7Smrj 			 * contains entry for GRUB's "root" cmd.
4069ae115bc7Smrj 			 */
4070ae115bc7Smrj 			grubdisk = read_grub_root();
4071ae115bc7Smrj 		}
4072ae115bc7Smrj 		if (grubdisk == NULL) {
4073ae115bc7Smrj 			bam_error(REBOOT_WITH_ARGS_FAILED);
4074ae115bc7Smrj 			rv = BAM_ERROR;
4075ae115bc7Smrj 			goto done;
4076ae115bc7Smrj 		}
4077ae115bc7Smrj 		if (optnum == KERNEL_CMD) {
4078ae115bc7Smrj 			entryNum = add_boot_entry(mp, BOOTENV_RC_TITLE,
4079ae115bc7Smrj 			    grubdisk, new_path, NULL);
4080ae115bc7Smrj 		} else {
4081ae115bc7Smrj 			new_str_len = strlen(DIRECT_BOOT_KERNEL) +
4082ae115bc7Smrj 			    strlen(path) + 8;
4083ae115bc7Smrj 			new_arg = s_calloc(1, new_str_len);
4084ae115bc7Smrj 
4085ae115bc7Smrj 			(void) snprintf(new_arg, new_str_len, "%s %s",
4086ae115bc7Smrj 			    DIRECT_BOOT_KERNEL, path);
4087ae115bc7Smrj 			entryNum = add_boot_entry(mp, BOOTENV_RC_TITLE,
4088ae115bc7Smrj 			    grubdisk, new_arg, DIRECT_BOOT_ARCHIVE);
4089ae115bc7Smrj 		}
4090ae115bc7Smrj 		save_default_entry(mp, BAM_OLD_RC_DEF);
4091ae115bc7Smrj 		(void) set_global(mp, menu_cmds[DEFAULT_CMD], entryNum);
4092ae115bc7Smrj 		rv = BAM_WRITE;
4093ae115bc7Smrj 		goto done;
4094ae115bc7Smrj 	}
4095ae115bc7Smrj 
4096ae115bc7Smrj 	/*
4097ae115bc7Smrj 	 * There was already an bootenv entry which we need to edit.
4098ae115bc7Smrj 	 */
4099ae115bc7Smrj 	if (optnum == KERNEL_CMD) {
4100ae115bc7Smrj 		new_str_len = strlen(new_path) + strlen(old_args) + 2;
4101ae115bc7Smrj 		new_arg = s_calloc(1, new_str_len);
4102ae115bc7Smrj 		(void) snprintf(new_arg, new_str_len, "%s %s", new_path,
4103ae115bc7Smrj 		    old_args);
4104ae115bc7Smrj 		free(kernelp->arg);
4105ae115bc7Smrj 		kernelp->arg = new_arg;
4106ae115bc7Smrj 
4107ae115bc7Smrj 		/*
4108ae115bc7Smrj 		 * If we have changed the kernel line, we may need to update
4109ae115bc7Smrj 		 * the archive line as well.
4110ae115bc7Smrj 		 */
4111ae115bc7Smrj 		set_archive_line(entryp, kernelp);
4112ae115bc7Smrj 	} else {
4113ae115bc7Smrj 		new_str_len = old_kernel_len + strlen(path) + 8;
4114ae115bc7Smrj 		new_arg = s_calloc(1, new_str_len);
4115ae115bc7Smrj 		(void) strncpy(new_arg, kernelp->arg, old_kernel_len);
4116ae115bc7Smrj 		(void) strlcat(new_arg, " ", new_str_len);
4117ae115bc7Smrj 		(void) strlcat(new_arg, path, new_str_len);
4118ae115bc7Smrj 		free(kernelp->arg);
4119ae115bc7Smrj 		kernelp->arg = new_arg;
4120ae115bc7Smrj 	}
4121ae115bc7Smrj 	rv = BAM_WRITE;
4122ae115bc7Smrj 
4123ae115bc7Smrj done:
4124ae115bc7Smrj 	if ((rv == BAM_WRITE) && kernelp)
4125ae115bc7Smrj 		update_line(kernelp);
4126ae115bc7Smrj 	if (free_new_path)
4127ae115bc7Smrj 		free(new_path);
4128ae115bc7Smrj 	return (rv);
4129ae115bc7Smrj }
4130ae115bc7Smrj 
41317c478bd9Sstevel@tonic-gate /*ARGSUSED*/
41327c478bd9Sstevel@tonic-gate static error_t
41337c478bd9Sstevel@tonic-gate set_option(menu_t *mp, char *menu_path, char *opt)
41347c478bd9Sstevel@tonic-gate {
41357c478bd9Sstevel@tonic-gate 	int optnum, optval;
41367c478bd9Sstevel@tonic-gate 	char *val;
4137ae115bc7Smrj 	char buf[BUFSIZ] = "";
4138ae115bc7Smrj 	error_t rv;
41397c478bd9Sstevel@tonic-gate 
41407c478bd9Sstevel@tonic-gate 	assert(mp);
41417c478bd9Sstevel@tonic-gate 	assert(opt);
41427c478bd9Sstevel@tonic-gate 
41437c478bd9Sstevel@tonic-gate 	val = strchr(opt, '=');
4144ae115bc7Smrj 	if (val != NULL) {
4145ae115bc7Smrj 		*val = '\0';
41467c478bd9Sstevel@tonic-gate 	}
41477c478bd9Sstevel@tonic-gate 
41487c478bd9Sstevel@tonic-gate 	if (strcmp(opt, "default") == 0) {
41497c478bd9Sstevel@tonic-gate 		optnum = DEFAULT_CMD;
41507c478bd9Sstevel@tonic-gate 	} else if (strcmp(opt, "timeout") == 0) {
41517c478bd9Sstevel@tonic-gate 		optnum = TIMEOUT_CMD;
4152ae115bc7Smrj 	} else if (strcmp(opt, menu_cmds[KERNEL_CMD]) == 0) {
4153ae115bc7Smrj 		optnum = KERNEL_CMD;
4154ae115bc7Smrj 	} else if (strcmp(opt, menu_cmds[ARGS_CMD]) == 0) {
4155ae115bc7Smrj 		optnum = ARGS_CMD;
41567c478bd9Sstevel@tonic-gate 	} else {
41577c478bd9Sstevel@tonic-gate 		bam_error(INVALID_ENTRY, opt);
41587c478bd9Sstevel@tonic-gate 		return (BAM_ERROR);
41597c478bd9Sstevel@tonic-gate 	}
41607c478bd9Sstevel@tonic-gate 
4161ae115bc7Smrj 	/*
4162ae115bc7Smrj 	 * kernel and args are allowed without "=new_value" strings.  All
4163ae115bc7Smrj 	 * others cause errors
4164ae115bc7Smrj 	 */
4165ae115bc7Smrj 	if ((val == NULL) && (optnum != KERNEL_CMD) && (optnum != ARGS_CMD)) {
4166ae115bc7Smrj 		bam_error(INVALID_ENTRY, opt);
4167ae115bc7Smrj 		return (BAM_ERROR);
4168ae115bc7Smrj 	} else if (val != NULL) {
41697c478bd9Sstevel@tonic-gate 		*val = '=';
4170ae115bc7Smrj 	}
4171ae115bc7Smrj 
4172ae115bc7Smrj 	if ((optnum == KERNEL_CMD) || (optnum == ARGS_CMD)) {
4173ae115bc7Smrj 		rv = set_kernel(mp, optnum, val ? val + 1 : NULL, buf, BUFSIZ);
4174ae115bc7Smrj 		if ((rv == BAM_SUCCESS) && (buf[0] != '\0'))
4175ae115bc7Smrj 			(void) printf("%s\n", buf);
4176ae115bc7Smrj 		return (rv);
4177ae115bc7Smrj 	} else {
4178ae115bc7Smrj 		optval = s_strtol(val + 1);
41797c478bd9Sstevel@tonic-gate 		return (set_global(mp, menu_cmds[optnum], optval));
41807c478bd9Sstevel@tonic-gate 	}
4181ae115bc7Smrj }
41827c478bd9Sstevel@tonic-gate 
41837c478bd9Sstevel@tonic-gate /*
41847c478bd9Sstevel@tonic-gate  * The quiet argument suppresses messages. This is used
41857c478bd9Sstevel@tonic-gate  * when invoked in the context of other commands (e.g. list_entry)
41867c478bd9Sstevel@tonic-gate  */
41877c478bd9Sstevel@tonic-gate static error_t
41887c478bd9Sstevel@tonic-gate read_globals(menu_t *mp, char *menu_path, char *globalcmd, int quiet)
41897c478bd9Sstevel@tonic-gate {
41907c478bd9Sstevel@tonic-gate 	line_t *lp;
41917c478bd9Sstevel@tonic-gate 	char *arg;
41927c478bd9Sstevel@tonic-gate 	int done, ret = BAM_SUCCESS;
41937c478bd9Sstevel@tonic-gate 
41947c478bd9Sstevel@tonic-gate 	assert(mp);
41957c478bd9Sstevel@tonic-gate 	assert(menu_path);
41967c478bd9Sstevel@tonic-gate 	assert(globalcmd);
41977c478bd9Sstevel@tonic-gate 
41987c478bd9Sstevel@tonic-gate 	if (mp->start == NULL) {
41997c478bd9Sstevel@tonic-gate 		if (!quiet)
42007c478bd9Sstevel@tonic-gate 			bam_error(NO_MENU, menu_path);
42017c478bd9Sstevel@tonic-gate 		return (BAM_ERROR);
42027c478bd9Sstevel@tonic-gate 	}
42037c478bd9Sstevel@tonic-gate 
42047c478bd9Sstevel@tonic-gate 	done = 0;
42057c478bd9Sstevel@tonic-gate 	for (lp = mp->start; lp; lp = lp->next) {
42067c478bd9Sstevel@tonic-gate 		if (lp->flags != BAM_GLOBAL)
42077c478bd9Sstevel@tonic-gate 			continue;
42087c478bd9Sstevel@tonic-gate 
42097c478bd9Sstevel@tonic-gate 		if (lp->cmd == NULL) {
42107c478bd9Sstevel@tonic-gate 			if (!quiet)
42117c478bd9Sstevel@tonic-gate 				bam_error(NO_CMD, lp->lineNum);
42127c478bd9Sstevel@tonic-gate 			continue;
42137c478bd9Sstevel@tonic-gate 		}
42147c478bd9Sstevel@tonic-gate 
42157c478bd9Sstevel@tonic-gate 		if (strcmp(globalcmd, lp->cmd) != 0)
42167c478bd9Sstevel@tonic-gate 			continue;
42177c478bd9Sstevel@tonic-gate 
42187c478bd9Sstevel@tonic-gate 		/* Found global. Check for duplicates */
42197c478bd9Sstevel@tonic-gate 		if (done && !quiet) {
42207c478bd9Sstevel@tonic-gate 			bam_error(DUP_CMD, globalcmd, lp->lineNum, bam_root);
42217c478bd9Sstevel@tonic-gate 			ret = BAM_ERROR;
42227c478bd9Sstevel@tonic-gate 		}
42237c478bd9Sstevel@tonic-gate 
42247c478bd9Sstevel@tonic-gate 		arg = lp->arg ? lp->arg : "";
42257c478bd9Sstevel@tonic-gate 		bam_print(GLOBAL_CMD, globalcmd, arg);
42267c478bd9Sstevel@tonic-gate 		done = 1;
42277c478bd9Sstevel@tonic-gate 	}
42287c478bd9Sstevel@tonic-gate 
42297c478bd9Sstevel@tonic-gate 	if (!done && bam_verbose)
42307c478bd9Sstevel@tonic-gate 		bam_print(NO_ENTRY, globalcmd);
42317c478bd9Sstevel@tonic-gate 
42327c478bd9Sstevel@tonic-gate 	return (ret);
42337c478bd9Sstevel@tonic-gate }
42347c478bd9Sstevel@tonic-gate 
42357c478bd9Sstevel@tonic-gate static error_t
42367c478bd9Sstevel@tonic-gate menu_write(char *root, menu_t *mp)
42377c478bd9Sstevel@tonic-gate {
42387c478bd9Sstevel@tonic-gate 	return (list2file(root, MENU_TMP, GRUB_MENU, mp->start));
42397c478bd9Sstevel@tonic-gate }
42407c478bd9Sstevel@tonic-gate 
42417c478bd9Sstevel@tonic-gate static void
42427c478bd9Sstevel@tonic-gate line_free(line_t *lp)
42437c478bd9Sstevel@tonic-gate {
42447c478bd9Sstevel@tonic-gate 	if (lp == NULL)
42457c478bd9Sstevel@tonic-gate 		return;
42467c478bd9Sstevel@tonic-gate 
42477c478bd9Sstevel@tonic-gate 	if (lp->cmd)
42487c478bd9Sstevel@tonic-gate 		free(lp->cmd);
42497c478bd9Sstevel@tonic-gate 	if (lp->sep)
42507c478bd9Sstevel@tonic-gate 		free(lp->sep);
42517c478bd9Sstevel@tonic-gate 	if (lp->arg)
42527c478bd9Sstevel@tonic-gate 		free(lp->arg);
42537c478bd9Sstevel@tonic-gate 	if (lp->line)
42547c478bd9Sstevel@tonic-gate 		free(lp->line);
42557c478bd9Sstevel@tonic-gate 	free(lp);
42567c478bd9Sstevel@tonic-gate }
42577c478bd9Sstevel@tonic-gate 
42587c478bd9Sstevel@tonic-gate static void
42597c478bd9Sstevel@tonic-gate linelist_free(line_t *start)
42607c478bd9Sstevel@tonic-gate {
42617c478bd9Sstevel@tonic-gate 	line_t *lp;
42627c478bd9Sstevel@tonic-gate 
42637c478bd9Sstevel@tonic-gate 	while (start) {
42647c478bd9Sstevel@tonic-gate 		lp = start;
42657c478bd9Sstevel@tonic-gate 		start = start->next;
42667c478bd9Sstevel@tonic-gate 		line_free(lp);
42677c478bd9Sstevel@tonic-gate 	}
42687c478bd9Sstevel@tonic-gate }
42697c478bd9Sstevel@tonic-gate 
42707c478bd9Sstevel@tonic-gate static void
42717c478bd9Sstevel@tonic-gate filelist_free(filelist_t *flistp)
42727c478bd9Sstevel@tonic-gate {
42737c478bd9Sstevel@tonic-gate 	linelist_free(flistp->head);
42747c478bd9Sstevel@tonic-gate 	flistp->head = NULL;
42757c478bd9Sstevel@tonic-gate 	flistp->tail = NULL;
42767c478bd9Sstevel@tonic-gate }
42777c478bd9Sstevel@tonic-gate 
42787c478bd9Sstevel@tonic-gate static void
42797c478bd9Sstevel@tonic-gate menu_free(menu_t *mp)
42807c478bd9Sstevel@tonic-gate {
42818c1b6884Sszhou 	entry_t *ent, *tmp;
42827c478bd9Sstevel@tonic-gate 	assert(mp);
42837c478bd9Sstevel@tonic-gate 
42847c478bd9Sstevel@tonic-gate 	if (mp->start)
42857c478bd9Sstevel@tonic-gate 		linelist_free(mp->start);
42868c1b6884Sszhou 	ent = mp->entries;
42878c1b6884Sszhou 	while (ent) {
42888c1b6884Sszhou 		tmp = ent;
42898c1b6884Sszhou 		ent = tmp->next;
42908c1b6884Sszhou 		free(tmp);
42918c1b6884Sszhou 	}
42927c478bd9Sstevel@tonic-gate 
42938c1b6884Sszhou 	free(mp);
42947c478bd9Sstevel@tonic-gate }
42957c478bd9Sstevel@tonic-gate 
42967c478bd9Sstevel@tonic-gate /*
42977c478bd9Sstevel@tonic-gate  * Utility routines
42987c478bd9Sstevel@tonic-gate  */
42997c478bd9Sstevel@tonic-gate 
43007c478bd9Sstevel@tonic-gate 
43017c478bd9Sstevel@tonic-gate /*
43027c478bd9Sstevel@tonic-gate  * Returns 0 on success
43037c478bd9Sstevel@tonic-gate  * Any other value indicates an error
43047c478bd9Sstevel@tonic-gate  */
43057c478bd9Sstevel@tonic-gate static int
43067c478bd9Sstevel@tonic-gate exec_cmd(char *cmdline, char *output, int64_t osize)
43077c478bd9Sstevel@tonic-gate {
43087c478bd9Sstevel@tonic-gate 	char buf[BUFSIZ];
43097c478bd9Sstevel@tonic-gate 	int ret;
43107c478bd9Sstevel@tonic-gate 	FILE *ptr;
43117c478bd9Sstevel@tonic-gate 	size_t len;
43127c478bd9Sstevel@tonic-gate 	sigset_t set;
43137c478bd9Sstevel@tonic-gate 	void (*disp)(int);
43147c478bd9Sstevel@tonic-gate 
43157c478bd9Sstevel@tonic-gate 	/*
43167c478bd9Sstevel@tonic-gate 	 * For security
43177c478bd9Sstevel@tonic-gate 	 * - only absolute paths are allowed
43187c478bd9Sstevel@tonic-gate 	 * - set IFS to space and tab
43197c478bd9Sstevel@tonic-gate 	 */
43207c478bd9Sstevel@tonic-gate 	if (*cmdline != '/') {
43217c478bd9Sstevel@tonic-gate 		bam_error(ABS_PATH_REQ, cmdline);
43227c478bd9Sstevel@tonic-gate 		return (-1);
43237c478bd9Sstevel@tonic-gate 	}
43247c478bd9Sstevel@tonic-gate 	(void) putenv("IFS= \t");
43257c478bd9Sstevel@tonic-gate 
43267c478bd9Sstevel@tonic-gate 	/*
43277c478bd9Sstevel@tonic-gate 	 * We may have been exec'ed with SIGCHLD blocked
43287c478bd9Sstevel@tonic-gate 	 * unblock it here
43297c478bd9Sstevel@tonic-gate 	 */
43307c478bd9Sstevel@tonic-gate 	(void) sigemptyset(&set);
43317c478bd9Sstevel@tonic-gate 	(void) sigaddset(&set, SIGCHLD);
43327c478bd9Sstevel@tonic-gate 	if (sigprocmask(SIG_UNBLOCK, &set, NULL) != 0) {
43337c478bd9Sstevel@tonic-gate 		bam_error(CANT_UNBLOCK_SIGCHLD, strerror(errno));
43347c478bd9Sstevel@tonic-gate 		return (-1);
43357c478bd9Sstevel@tonic-gate 	}
43367c478bd9Sstevel@tonic-gate 
43377c478bd9Sstevel@tonic-gate 	/*
43387c478bd9Sstevel@tonic-gate 	 * Set SIGCHLD disposition to SIG_DFL for popen/pclose
43397c478bd9Sstevel@tonic-gate 	 */
43407c478bd9Sstevel@tonic-gate 	disp = sigset(SIGCHLD, SIG_DFL);
43417c478bd9Sstevel@tonic-gate 	if (disp == SIG_ERR) {
43427c478bd9Sstevel@tonic-gate 		bam_error(FAILED_SIG, strerror(errno));
43437c478bd9Sstevel@tonic-gate 		return (-1);
43447c478bd9Sstevel@tonic-gate 	}
43457c478bd9Sstevel@tonic-gate 	if (disp == SIG_HOLD) {
43467c478bd9Sstevel@tonic-gate 		bam_error(BLOCKED_SIG, cmdline);
43477c478bd9Sstevel@tonic-gate 		return (-1);
43487c478bd9Sstevel@tonic-gate 	}
43497c478bd9Sstevel@tonic-gate 
43507c478bd9Sstevel@tonic-gate 	ptr = popen(cmdline, "r");
43517c478bd9Sstevel@tonic-gate 	if (ptr == NULL) {
43527c478bd9Sstevel@tonic-gate 		bam_error(POPEN_FAIL, cmdline, strerror(errno));
43537c478bd9Sstevel@tonic-gate 		return (-1);
43547c478bd9Sstevel@tonic-gate 	}
43557c478bd9Sstevel@tonic-gate 
43567c478bd9Sstevel@tonic-gate 	/*
43577c478bd9Sstevel@tonic-gate 	 * If we simply do a pclose() following a popen(), pclose()
43587c478bd9Sstevel@tonic-gate 	 * will close the reader end of the pipe immediately even
43597c478bd9Sstevel@tonic-gate 	 * if the child process has not started/exited. pclose()
43607c478bd9Sstevel@tonic-gate 	 * does wait for cmd to terminate before returning though.
43617c478bd9Sstevel@tonic-gate 	 * When the executed command writes its output to the pipe
43627c478bd9Sstevel@tonic-gate 	 * there is no reader process and the command dies with
43637c478bd9Sstevel@tonic-gate 	 * SIGPIPE. To avoid this we read repeatedly until read
43647c478bd9Sstevel@tonic-gate 	 * terminates with EOF. This indicates that the command
43657c478bd9Sstevel@tonic-gate 	 * (writer) has closed the pipe and we can safely do a
43667c478bd9Sstevel@tonic-gate 	 * pclose().
43677c478bd9Sstevel@tonic-gate 	 *
43687c478bd9Sstevel@tonic-gate 	 * Since pclose() does wait for the command to exit,
43697c478bd9Sstevel@tonic-gate 	 * we can safely reap the exit status of the command
43707c478bd9Sstevel@tonic-gate 	 * from the value returned by pclose()
43717c478bd9Sstevel@tonic-gate 	 */
43727c478bd9Sstevel@tonic-gate 	while (fgets(buf, sizeof (buf), ptr) != NULL) {
43737c478bd9Sstevel@tonic-gate 		/* if (bam_verbose)  XXX */
43747c478bd9Sstevel@tonic-gate 			bam_print(PRINT_NO_NEWLINE, buf);
43757c478bd9Sstevel@tonic-gate 		if (output && osize > 0) {
43767c478bd9Sstevel@tonic-gate 			(void) snprintf(output, osize, "%s", buf);
43777c478bd9Sstevel@tonic-gate 			len = strlen(buf);
43787c478bd9Sstevel@tonic-gate 			output += len;
43797c478bd9Sstevel@tonic-gate 			osize -= len;
43807c478bd9Sstevel@tonic-gate 		}
43817c478bd9Sstevel@tonic-gate 	}
43827c478bd9Sstevel@tonic-gate 
43837c478bd9Sstevel@tonic-gate 	ret = pclose(ptr);
43847c478bd9Sstevel@tonic-gate 	if (ret == -1) {
43857c478bd9Sstevel@tonic-gate 		bam_error(PCLOSE_FAIL, cmdline, strerror(errno));
43867c478bd9Sstevel@tonic-gate 		return (-1);
43877c478bd9Sstevel@tonic-gate 	}
43887c478bd9Sstevel@tonic-gate 
43897c478bd9Sstevel@tonic-gate 	if (WIFEXITED(ret)) {
43907c478bd9Sstevel@tonic-gate 		return (WEXITSTATUS(ret));
43917c478bd9Sstevel@tonic-gate 	} else {
43927c478bd9Sstevel@tonic-gate 		bam_error(EXEC_FAIL, cmdline, ret);
43937c478bd9Sstevel@tonic-gate 		return (-1);
43947c478bd9Sstevel@tonic-gate 	}
43957c478bd9Sstevel@tonic-gate }
43967c478bd9Sstevel@tonic-gate 
43977c478bd9Sstevel@tonic-gate /*
43987c478bd9Sstevel@tonic-gate  * Since this function returns -1 on error
43997c478bd9Sstevel@tonic-gate  * it cannot be used to convert -1. However,
44007c478bd9Sstevel@tonic-gate  * that is sufficient for what we need.
44017c478bd9Sstevel@tonic-gate  */
44027c478bd9Sstevel@tonic-gate static long
44037c478bd9Sstevel@tonic-gate s_strtol(char *str)
44047c478bd9Sstevel@tonic-gate {
44057c478bd9Sstevel@tonic-gate 	long l;
44067c478bd9Sstevel@tonic-gate 	char *res = NULL;
44077c478bd9Sstevel@tonic-gate 
44087c478bd9Sstevel@tonic-gate 	if (str == NULL) {
44097c478bd9Sstevel@tonic-gate 		return (-1);
44107c478bd9Sstevel@tonic-gate 	}
44117c478bd9Sstevel@tonic-gate 
44127c478bd9Sstevel@tonic-gate 	errno = 0;
44137c478bd9Sstevel@tonic-gate 	l = strtol(str, &res, 10);
44147c478bd9Sstevel@tonic-gate 	if (errno || *res != '\0') {
44157c478bd9Sstevel@tonic-gate 		return (-1);
44167c478bd9Sstevel@tonic-gate 	}
44177c478bd9Sstevel@tonic-gate 
44187c478bd9Sstevel@tonic-gate 	return (l);
44197c478bd9Sstevel@tonic-gate }
44207c478bd9Sstevel@tonic-gate 
44217c478bd9Sstevel@tonic-gate /*
44227c478bd9Sstevel@tonic-gate  * Wrapper around fputs, that adds a newline (since fputs doesn't)
44237c478bd9Sstevel@tonic-gate  */
44247c478bd9Sstevel@tonic-gate static int
44257c478bd9Sstevel@tonic-gate s_fputs(char *str, FILE *fp)
44267c478bd9Sstevel@tonic-gate {
44277c478bd9Sstevel@tonic-gate 	char linebuf[BAM_MAXLINE];
44287c478bd9Sstevel@tonic-gate 
44297c478bd9Sstevel@tonic-gate 	(void) snprintf(linebuf, sizeof (linebuf), "%s\n", str);
44307c478bd9Sstevel@tonic-gate 	return (fputs(linebuf, fp));
44317c478bd9Sstevel@tonic-gate }
44327c478bd9Sstevel@tonic-gate 
44337c478bd9Sstevel@tonic-gate /*
44347c478bd9Sstevel@tonic-gate  * Wrapper around fgets, that strips newlines returned by fgets
44357c478bd9Sstevel@tonic-gate  */
4436ae115bc7Smrj char *
44377c478bd9Sstevel@tonic-gate s_fgets(char *buf, int buflen, FILE *fp)
44387c478bd9Sstevel@tonic-gate {
44397c478bd9Sstevel@tonic-gate 	int n;
44407c478bd9Sstevel@tonic-gate 
44417c478bd9Sstevel@tonic-gate 	buf = fgets(buf, buflen, fp);
44427c478bd9Sstevel@tonic-gate 	if (buf) {
44437c478bd9Sstevel@tonic-gate 		n = strlen(buf);
44447c478bd9Sstevel@tonic-gate 		if (n == buflen - 1 && buf[n-1] != '\n')
44457c478bd9Sstevel@tonic-gate 			bam_error(TOO_LONG, buflen - 1, buf);
44467c478bd9Sstevel@tonic-gate 		buf[n-1] = (buf[n-1] == '\n') ? '\0' : buf[n-1];
44477c478bd9Sstevel@tonic-gate 	}
44487c478bd9Sstevel@tonic-gate 
44497c478bd9Sstevel@tonic-gate 	return (buf);
44507c478bd9Sstevel@tonic-gate }
44517c478bd9Sstevel@tonic-gate 
4452ae115bc7Smrj void *
44537c478bd9Sstevel@tonic-gate s_calloc(size_t nelem, size_t sz)
44547c478bd9Sstevel@tonic-gate {
44557c478bd9Sstevel@tonic-gate 	void *ptr;
44567c478bd9Sstevel@tonic-gate 
44577c478bd9Sstevel@tonic-gate 	ptr = calloc(nelem, sz);
44587c478bd9Sstevel@tonic-gate 	if (ptr == NULL) {
44597c478bd9Sstevel@tonic-gate 		bam_error(NO_MEM, nelem*sz);
44607c478bd9Sstevel@tonic-gate 		bam_exit(1);
44617c478bd9Sstevel@tonic-gate 	}
44627c478bd9Sstevel@tonic-gate 	return (ptr);
44637c478bd9Sstevel@tonic-gate }
44647c478bd9Sstevel@tonic-gate 
4465ae115bc7Smrj void *
4466ae115bc7Smrj s_realloc(void *ptr, size_t sz)
4467ae115bc7Smrj {
4468ae115bc7Smrj 	ptr = realloc(ptr, sz);
4469ae115bc7Smrj 	if (ptr == NULL) {
4470ae115bc7Smrj 		bam_error(NO_MEM, sz);
4471ae115bc7Smrj 		bam_exit(1);
4472ae115bc7Smrj 	}
4473ae115bc7Smrj 	return (ptr);
4474ae115bc7Smrj }
4475ae115bc7Smrj 
44767c478bd9Sstevel@tonic-gate static char *
44777c478bd9Sstevel@tonic-gate s_strdup(char *str)
44787c478bd9Sstevel@tonic-gate {
44797c478bd9Sstevel@tonic-gate 	char *ptr;
44807c478bd9Sstevel@tonic-gate 
44817c478bd9Sstevel@tonic-gate 	if (str == NULL)
44827c478bd9Sstevel@tonic-gate 		return (NULL);
44837c478bd9Sstevel@tonic-gate 
44847c478bd9Sstevel@tonic-gate 	ptr = strdup(str);
44857c478bd9Sstevel@tonic-gate 	if (ptr == NULL) {
44867c478bd9Sstevel@tonic-gate 		bam_error(NO_MEM, strlen(str) + 1);
44877c478bd9Sstevel@tonic-gate 		bam_exit(1);
44887c478bd9Sstevel@tonic-gate 	}
44897c478bd9Sstevel@tonic-gate 	return (ptr);
44907c478bd9Sstevel@tonic-gate }
44917c478bd9Sstevel@tonic-gate 
44927c478bd9Sstevel@tonic-gate /*
44937c478bd9Sstevel@tonic-gate  * Returns 1 if amd64 (or sparc, for syncing x86 diskless clients)
44947c478bd9Sstevel@tonic-gate  * Returns 0 otherwise
44957c478bd9Sstevel@tonic-gate  */
44967c478bd9Sstevel@tonic-gate static int
44977c478bd9Sstevel@tonic-gate is_amd64(void)
44987c478bd9Sstevel@tonic-gate {
44997c478bd9Sstevel@tonic-gate 	static int amd64 = -1;
45007c478bd9Sstevel@tonic-gate 	char isabuf[257];	/* from sysinfo(2) manpage */
45017c478bd9Sstevel@tonic-gate 
45027c478bd9Sstevel@tonic-gate 	if (amd64 != -1)
45037c478bd9Sstevel@tonic-gate 		return (amd64);
45047c478bd9Sstevel@tonic-gate 
45057c478bd9Sstevel@tonic-gate 	if (sysinfo(SI_ISALIST, isabuf, sizeof (isabuf)) > 0 &&
45067c478bd9Sstevel@tonic-gate 	    strncmp(isabuf, "amd64 ", strlen("amd64 ")) == 0)
45077c478bd9Sstevel@tonic-gate 		amd64 = 1;
45087c478bd9Sstevel@tonic-gate 	else if (strstr(isabuf, "i386") == NULL)
45097c478bd9Sstevel@tonic-gate 		amd64 = 1;		/* diskless server */
45107c478bd9Sstevel@tonic-gate 	else
45117c478bd9Sstevel@tonic-gate 		amd64 = 0;
45127c478bd9Sstevel@tonic-gate 
45137c478bd9Sstevel@tonic-gate 	return (amd64);
45147c478bd9Sstevel@tonic-gate }
45157c478bd9Sstevel@tonic-gate 
45167c478bd9Sstevel@tonic-gate static void
45177c478bd9Sstevel@tonic-gate append_to_flist(filelist_t *flistp, char *s)
45187c478bd9Sstevel@tonic-gate {
45197c478bd9Sstevel@tonic-gate 	line_t *lp;
45207c478bd9Sstevel@tonic-gate 
45217c478bd9Sstevel@tonic-gate 	lp = s_calloc(1, sizeof (line_t));
45227c478bd9Sstevel@tonic-gate 	lp->line = s_strdup(s);
45237c478bd9Sstevel@tonic-gate 	if (flistp->head == NULL)
45247c478bd9Sstevel@tonic-gate 		flistp->head = lp;
45257c478bd9Sstevel@tonic-gate 	else
45267c478bd9Sstevel@tonic-gate 		flistp->tail->next = lp;
45277c478bd9Sstevel@tonic-gate 	flistp->tail = lp;
45287c478bd9Sstevel@tonic-gate }
4529*2449e17fSsherrym 
4530*2449e17fSsherrym #if defined(__i386)
4531*2449e17fSsherrym 
4532*2449e17fSsherrym UCODE_VENDORS;
4533*2449e17fSsherrym 
4534*2449e17fSsherrym /*ARGSUSED*/
4535*2449e17fSsherrym static void
4536*2449e17fSsherrym ucode_install(char *root)
4537*2449e17fSsherrym {
4538*2449e17fSsherrym 	int i;
4539*2449e17fSsherrym 
4540*2449e17fSsherrym 	for (i = 0; ucode_vendors[i].filestr != NULL; i++) {
4541*2449e17fSsherrym 		int cmd_len = PATH_MAX + 256;
4542*2449e17fSsherrym 		char cmd[PATH_MAX + 256];
4543*2449e17fSsherrym 		char file[PATH_MAX];
4544*2449e17fSsherrym 		char timestamp[PATH_MAX];
4545*2449e17fSsherrym 		struct stat fstatus, tstatus;
4546*2449e17fSsherrym 		struct utimbuf u_times;
4547*2449e17fSsherrym 
4548*2449e17fSsherrym 		(void) snprintf(file, PATH_MAX, "%s/%s/%s-ucode.txt",
4549*2449e17fSsherrym 		    bam_root, UCODE_INSTALL_PATH, ucode_vendors[i].filestr);
4550*2449e17fSsherrym 
4551*2449e17fSsherrym 		if (stat(file, &fstatus) != 0 || !(S_ISREG(fstatus.st_mode)))
4552*2449e17fSsherrym 			continue;
4553*2449e17fSsherrym 
4554*2449e17fSsherrym 		(void) snprintf(timestamp, PATH_MAX, "%s.ts", file);
4555*2449e17fSsherrym 
4556*2449e17fSsherrym 		if (stat(timestamp, &tstatus) == 0 &&
4557*2449e17fSsherrym 		    fstatus.st_mtime <= tstatus.st_mtime)
4558*2449e17fSsherrym 			continue;
4559*2449e17fSsherrym 
4560*2449e17fSsherrym 		(void) snprintf(cmd, cmd_len, "/usr/sbin/ucodeadm -i -R "
4561*2449e17fSsherrym 		    "%s/%s/%s %s > /dev/null 2>&1", bam_root,
4562*2449e17fSsherrym 		    UCODE_INSTALL_PATH, ucode_vendors[i].vendorstr, file);
4563*2449e17fSsherrym 		if (system(cmd) != 0)
4564*2449e17fSsherrym 			return;
4565*2449e17fSsherrym 
4566*2449e17fSsherrym 		if (creat(timestamp, S_IRUSR | S_IWUSR) == -1)
4567*2449e17fSsherrym 			return;
4568*2449e17fSsherrym 
4569*2449e17fSsherrym 		u_times.actime = fstatus.st_atime;
4570*2449e17fSsherrym 		u_times.modtime = fstatus.st_mtime;
4571*2449e17fSsherrym 		(void) utime(timestamp, &u_times);
4572*2449e17fSsherrym 	}
4573*2449e17fSsherrym }
4574*2449e17fSsherrym #endif
4575