1 /* 2 * CDDL HEADER START 3 * 4 * The contents of this file are subject to the terms of the 5 * Common Development and Distribution License (the "License"). 6 * You may not use this file except in compliance with the License. 7 * 8 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE 9 * or http://www.opensolaris.org/os/licensing. 10 * See the License for the specific language governing permissions 11 * and limitations under the License. 12 * 13 * When distributing Covered Code, include this CDDL HEADER in each 14 * file and include the License file at usr/src/OPENSOLARIS.LICENSE. 15 * If applicable, add the following below this CDDL HEADER, with the 16 * fields enclosed by brackets "[]" replaced with your own identifying 17 * information: Portions Copyright [yyyy] [name of copyright owner] 18 * 19 * CDDL HEADER END 20 */ 21 22 /* 23 * Copyright 2009 Sun Microsystems, Inc. All rights reserved. 24 * Use is subject to license terms. 25 */ 26 27 28 /* 29 * Wrapper function to implement reboot w/ arguments on x86 30 * platforms. Extract reboot arguments and place them in 31 * in a transient entry in /[stub]boot/grub/menu.lst 32 * All other commands are passed through. 33 */ 34 35 #include "lint.h" 36 #include <fcntl.h> 37 #include <ctype.h> 38 #include <stdio.h> 39 #include <stdlib.h> 40 #include <sys/types.h> 41 #include <sys/stat.h> 42 #include <sys/uadmin.h> 43 #include <unistd.h> 44 #include <strings.h> 45 #include <pthread.h> 46 #include <zone.h> 47 48 static int 49 legal_arg(char *bargs) 50 { 51 int i; 52 53 for (i = 0; i < BOOTARGS_MAX; i++, bargs++) { 54 if (*bargs == 0 && i > 0) 55 return (i); 56 if (!isprint(*bargs)) 57 break; 58 } 59 return (-1); 60 } 61 62 static char quote[] = "\'"; 63 64 int 65 uadmin(int cmd, int fcn, uintptr_t mdep) 66 { 67 extern int __uadmin(int cmd, int fcn, uintptr_t mdep); 68 char *bargs, cmdbuf[256]; 69 struct stat sbuf; 70 char *altroot; 71 72 bargs = (char *)mdep; 73 74 if (geteuid() == 0 && getzoneid() == GLOBAL_ZONEID && 75 (cmd == A_SHUTDOWN || cmd == A_REBOOT)) { 76 int off = 0; 77 78 switch (fcn) { 79 case AD_IBOOT: 80 case AD_SBOOT: 81 case AD_SIBOOT: 82 /* 83 * These functions fabricate appropriate bootargs. 84 * If bootargs are passed in, map these functions 85 * to AD_BOOT. 86 */ 87 if (bargs == 0) { 88 switch (fcn) { 89 case AD_IBOOT: 90 bargs = "-a"; 91 break; 92 case AD_SBOOT: 93 bargs = "-s"; 94 break; 95 case AD_SIBOOT: 96 bargs = "-sa"; 97 break; 98 } 99 } 100 /*FALLTHROUGH*/ 101 case AD_BOOT: 102 case AD_FASTREBOOT: 103 if (bargs == 0) 104 break; /* no args */ 105 if (legal_arg(bargs) < 0) 106 break; /* bad args */ 107 108 /* avoid cancellation in system() */ 109 (void) pthread_setcancelstate(PTHREAD_CANCEL_DISABLE, 110 NULL); 111 112 /* check for /stubboot */ 113 if (stat("/stubboot/boot/grub/menu.lst", &sbuf) == 0) { 114 altroot = "-R /stubboot "; 115 } else { 116 altroot = ""; 117 } 118 119 if (fcn == AD_FASTREBOOT) { 120 char *newarg, *head; 121 char bargs_scratch[BOOTARGS_MAX]; 122 123 bzero(bargs_scratch, BOOTARGS_MAX); 124 125 bcopy(bargs, bargs_scratch, strlen(bargs)); 126 head = bargs_scratch; 127 newarg = strtok(bargs_scratch, " "); 128 129 if (newarg == NULL || newarg[0] == '-') 130 break; 131 132 /* First argument is rootdir */ 133 if (strncmp(&newarg[strlen(newarg)-4], 134 "unix", 4) != 0) { 135 newarg = strtok(NULL, " "); 136 off = newarg - head; 137 } 138 139 /* 140 * If we are using alternate root via 141 * mountpoint or a different BE, don't 142 * bother to update the temp menu entry. 143 */ 144 if (off > 0) 145 break; 146 } 147 148 /* are we rebooting to a GRUB menu entry? */ 149 if (isdigit(bargs[0])) { 150 int entry = strtol(bargs, NULL, 10); 151 (void) snprintf(cmdbuf, sizeof (cmdbuf), 152 "/sbin/bootadm set-menu %sdefault=%d", 153 altroot, entry); 154 } else { 155 (void) snprintf(cmdbuf, sizeof (cmdbuf), 156 "/sbin/bootadm -m update_temp %s" 157 "-o %s%s%s", altroot, quote, 158 &bargs[off], quote); 159 } 160 (void) system(cmdbuf); 161 } 162 } 163 164 return (__uadmin(cmd, fcn, mdep)); 165 } 166