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