1*0Sstevel@tonic-gate /*
2*0Sstevel@tonic-gate  * CDDL HEADER START
3*0Sstevel@tonic-gate  *
4*0Sstevel@tonic-gate  * The contents of this file are subject to the terms of the
5*0Sstevel@tonic-gate  * Common Development and Distribution License, Version 1.0 only
6*0Sstevel@tonic-gate  * (the "License").  You may not use this file except in compliance
7*0Sstevel@tonic-gate  * with the License.
8*0Sstevel@tonic-gate  *
9*0Sstevel@tonic-gate  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
10*0Sstevel@tonic-gate  * or http://www.opensolaris.org/os/licensing.
11*0Sstevel@tonic-gate  * See the License for the specific language governing permissions
12*0Sstevel@tonic-gate  * and limitations under the License.
13*0Sstevel@tonic-gate  *
14*0Sstevel@tonic-gate  * When distributing Covered Code, include this CDDL HEADER in each
15*0Sstevel@tonic-gate  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
16*0Sstevel@tonic-gate  * If applicable, add the following below this CDDL HEADER, with the
17*0Sstevel@tonic-gate  * fields enclosed by brackets "[]" replaced with your own identifying
18*0Sstevel@tonic-gate  * information: Portions Copyright [yyyy] [name of copyright owner]
19*0Sstevel@tonic-gate  *
20*0Sstevel@tonic-gate  * CDDL HEADER END
21*0Sstevel@tonic-gate  */
22*0Sstevel@tonic-gate /*
23*0Sstevel@tonic-gate  * Copyright 2005 Sun Microsystems, Inc.  All rights reserved.
24*0Sstevel@tonic-gate  * Use is subject to license terms.
25*0Sstevel@tonic-gate  */
26*0Sstevel@tonic-gate 
27*0Sstevel@tonic-gate #pragma ident	"%Z%%M%	%I%	%E% SMI"
28*0Sstevel@tonic-gate 
29*0Sstevel@tonic-gate #pragma weak uadmin = _uadmin
30*0Sstevel@tonic-gate 
31*0Sstevel@tonic-gate /*
32*0Sstevel@tonic-gate  * Wrapper function to implement reboot w/ arguments on x86
33*0Sstevel@tonic-gate  * platforms. Extract reboot arguments and place them in
34*0Sstevel@tonic-gate  * in a transient entry in /[stub]boot/grub/menu.lst
35*0Sstevel@tonic-gate  * All other commands are passed through.
36*0Sstevel@tonic-gate  */
37*0Sstevel@tonic-gate 
38*0Sstevel@tonic-gate #include "synonyms.h"
39*0Sstevel@tonic-gate #include <fcntl.h>
40*0Sstevel@tonic-gate #include <ctype.h>
41*0Sstevel@tonic-gate #include <stdio.h>
42*0Sstevel@tonic-gate #include <stdlib.h>
43*0Sstevel@tonic-gate #include <sys/types.h>
44*0Sstevel@tonic-gate #include <sys/stat.h>
45*0Sstevel@tonic-gate #include <sys/uadmin.h>
46*0Sstevel@tonic-gate #include <unistd.h>
47*0Sstevel@tonic-gate #include <string.h>
48*0Sstevel@tonic-gate 
49*0Sstevel@tonic-gate #define	MAX_BOOTARG	256
50*0Sstevel@tonic-gate 
51*0Sstevel@tonic-gate static int
52*0Sstevel@tonic-gate legal_arg(char *bargs)
53*0Sstevel@tonic-gate {
54*0Sstevel@tonic-gate 	int i;
55*0Sstevel@tonic-gate 
56*0Sstevel@tonic-gate 	for (i = 0; i < MAX_BOOTARG; i++, bargs++) {
57*0Sstevel@tonic-gate 		if (*bargs == 0 && i > 0)
58*0Sstevel@tonic-gate 			return (i);
59*0Sstevel@tonic-gate 		if (!isprint(*bargs))
60*0Sstevel@tonic-gate 			break;
61*0Sstevel@tonic-gate 	}
62*0Sstevel@tonic-gate 	return (-1);
63*0Sstevel@tonic-gate }
64*0Sstevel@tonic-gate 
65*0Sstevel@tonic-gate static char quote[] = "\'";
66*0Sstevel@tonic-gate 
67*0Sstevel@tonic-gate int
68*0Sstevel@tonic-gate uadmin(int cmd, int fcn, uintptr_t mdep)
69*0Sstevel@tonic-gate {
70*0Sstevel@tonic-gate 	extern int __uadmin(int cmd, int fcn, uintptr_t mdep);
71*0Sstevel@tonic-gate 	char *bargs, cmdbuf[256];
72*0Sstevel@tonic-gate 	struct stat sbuf;
73*0Sstevel@tonic-gate 	char *altroot;
74*0Sstevel@tonic-gate 
75*0Sstevel@tonic-gate 	bargs = (char *)mdep;
76*0Sstevel@tonic-gate 	if (geteuid() == 0 && (cmd == A_SHUTDOWN || cmd == A_REBOOT)) {
77*0Sstevel@tonic-gate 		switch (fcn) {
78*0Sstevel@tonic-gate 		case AD_IBOOT:
79*0Sstevel@tonic-gate 		case AD_SBOOT:
80*0Sstevel@tonic-gate 		case AD_SIBOOT:
81*0Sstevel@tonic-gate 			/*
82*0Sstevel@tonic-gate 			 * These functions fabricate appropriate bootargs.
83*0Sstevel@tonic-gate 			 * If bootargs are passed in, map these functions
84*0Sstevel@tonic-gate 			 * to AD_BOOT.
85*0Sstevel@tonic-gate 			 */
86*0Sstevel@tonic-gate 			if (bargs == 0) {
87*0Sstevel@tonic-gate 				switch (fcn) {
88*0Sstevel@tonic-gate 				case AD_IBOOT:
89*0Sstevel@tonic-gate 					bargs = "-a";
90*0Sstevel@tonic-gate 					break;
91*0Sstevel@tonic-gate 				case AD_SBOOT:
92*0Sstevel@tonic-gate 					bargs = "-s";
93*0Sstevel@tonic-gate 					break;
94*0Sstevel@tonic-gate 				case AD_SIBOOT:
95*0Sstevel@tonic-gate 					bargs = "-sa";
96*0Sstevel@tonic-gate 					break;
97*0Sstevel@tonic-gate 				}
98*0Sstevel@tonic-gate 			}
99*0Sstevel@tonic-gate 			/*FALLTHROUGH*/
100*0Sstevel@tonic-gate 		case AD_BOOT:
101*0Sstevel@tonic-gate 			if (bargs == 0)
102*0Sstevel@tonic-gate 				break;	/* no args */
103*0Sstevel@tonic-gate 			if (legal_arg(bargs) < 0)
104*0Sstevel@tonic-gate 				break;	/* bad args */
105*0Sstevel@tonic-gate 
106*0Sstevel@tonic-gate 			/* check for /stubboot */
107*0Sstevel@tonic-gate 			if (stat("/stubboot/boot/grub/menu.lst", &sbuf) == 0) {
108*0Sstevel@tonic-gate 				altroot = "-R /stubboot ";
109*0Sstevel@tonic-gate 			} else {
110*0Sstevel@tonic-gate 				altroot = "";
111*0Sstevel@tonic-gate 			}
112*0Sstevel@tonic-gate 
113*0Sstevel@tonic-gate 			/* are we rebooting to a GRUB menu entry? */
114*0Sstevel@tonic-gate 			if (isdigit(bargs[0])) {
115*0Sstevel@tonic-gate 				int entry = strtol(bargs, NULL, 10);
116*0Sstevel@tonic-gate 				(void) snprintf(cmdbuf, sizeof (cmdbuf),
117*0Sstevel@tonic-gate 				    "/sbin/bootadm set-menu %sdefault=%d",
118*0Sstevel@tonic-gate 				    altroot, entry);
119*0Sstevel@tonic-gate 			} else {
120*0Sstevel@tonic-gate 				(void) snprintf(cmdbuf, sizeof (cmdbuf),
121*0Sstevel@tonic-gate 				    "/sbin/bootadm -m update_temp %s"
122*0Sstevel@tonic-gate 				    "-o %s%s%s", altroot, quote, bargs, quote);
123*0Sstevel@tonic-gate 			}
124*0Sstevel@tonic-gate 			(void) system(cmdbuf);
125*0Sstevel@tonic-gate 		}
126*0Sstevel@tonic-gate 	}
127*0Sstevel@tonic-gate 	return (__uadmin(cmd, fcn, mdep));
128*0Sstevel@tonic-gate }
129