1 /*-
2 * SPDX-License-Identifier: BSD-3-Clause
3 *
4 * Copyright (c) 1988, 1993, 1994
5 * The Regents of the University of California. All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 * 1. Redistributions of source code must retain the above copyright
11 * notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 * notice, this list of conditions and the following disclaimer in the
14 * documentation and/or other materials provided with the distribution.
15 * 3. Neither the name of the University nor the names of its contributors
16 * may be used to endorse or promote products derived from this software
17 * without specific prior written permission.
18 *
19 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
20 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
21 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
23 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
24 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
25 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
26 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29 * SUCH DAMAGE.
30 */
31
32 #ifndef lint
33 static const char copyright[] =
34 "@(#) Copyright (c) 1988, 1993, 1994\n\
35 The Regents of the University of California. All rights reserved.\n";
36 #endif /* not lint */
37
38 #if 0
39 #ifndef lint
40 static char sccsid[] = "@(#)env.c 8.3 (Berkeley) 4/2/94";
41 #endif /* not lint */
42 #endif
43
44 #include <sys/cdefs.h>
45 __FBSDID("$FreeBSD$");
46
47 #include <sys/types.h>
48
49 #include <err.h>
50 #include <errno.h>
51 #include <login_cap.h>
52 #include <pwd.h>
53 #include <stdbool.h>
54 #include <stdio.h>
55 #include <stdlib.h>
56 #include <string.h>
57 #include <unistd.h>
58
59 #include "envopts.h"
60
61 extern char **environ;
62
63 int env_verbosity;
64
65 static void usage(void);
66
67 /*
68 * Exit codes.
69 */
70 #define EXIT_CANCELED 125 /* Internal error prior to exec attempt. */
71 #define EXIT_CANNOT_INVOKE 126 /* Program located, but not usable. */
72 #define EXIT_ENOENT 127 /* Could not find program to exec. */
73
74 int
main(int argc,char ** argv)75 main(int argc, char **argv)
76 {
77 char *altpath, **ep, *p, **parg, term;
78 char *cleanenv[1];
79 char *login_class, *login_name;
80 struct passwd *pw;
81 login_cap_t *lc;
82 bool login_as_user;
83 uid_t uid;
84 int ch, want_clear;
85 int rtrn;
86
87 altpath = NULL;
88 login_class = NULL;
89 login_name = NULL;
90 pw = NULL;
91 lc = NULL;
92 login_as_user = false;
93 want_clear = 0;
94 term = '\n';
95 while ((ch = getopt(argc, argv, "-0iL:P:S:U:u:v")) != -1)
96 switch(ch) {
97 case '-':
98 case 'i':
99 want_clear = 1;
100 break;
101 case '0':
102 term = '\0';
103 break;
104 case 'U':
105 login_as_user = true;
106 /* FALLTHROUGH */
107 case 'L':
108 login_name = optarg;
109 break;
110 case 'P':
111 altpath = strdup(optarg);
112 break;
113 case 'S':
114 /*
115 * The -S option, for "split string on spaces, with
116 * support for some simple substitutions"...
117 */
118 split_spaces(optarg, &optind, &argc, &argv);
119 break;
120 case 'u':
121 if (env_verbosity)
122 fprintf(stderr, "#env unset:\t%s\n", optarg);
123 rtrn = unsetenv(optarg);
124 if (rtrn == -1)
125 err(EXIT_FAILURE, "unsetenv %s", optarg);
126 break;
127 case 'v':
128 env_verbosity++;
129 if (env_verbosity > 1)
130 fprintf(stderr, "#env verbosity now at %d\n",
131 env_verbosity);
132 break;
133 case '?':
134 default:
135 usage();
136 }
137 if (want_clear) {
138 environ = cleanenv;
139 cleanenv[0] = NULL;
140 if (env_verbosity)
141 fprintf(stderr, "#env clearing environ\n");
142 }
143 if (login_name != NULL) {
144 login_class = strchr(login_name, '/');
145 if (login_class)
146 *login_class++ = '\0';
147 if (*login_name != '\0' && strcmp(login_name, "-") != 0) {
148 pw = getpwnam(login_name);
149 if (pw == NULL) {
150 char *endp = NULL;
151 errno = 0;
152 uid = strtoul(login_name, &endp, 10);
153 if (errno == 0 && *endp == '\0')
154 pw = getpwuid(uid);
155 }
156 if (pw == NULL)
157 errx(EXIT_FAILURE, "no such user: %s", login_name);
158 }
159 /*
160 * Note that it is safe for pw to be null here; the libutil
161 * code handles that, bypassing substitution of $ and using
162 * the class "default" if no class name is given either.
163 */
164 if (login_class != NULL) {
165 lc = login_getclass(login_class);
166 if (lc == NULL)
167 errx(EXIT_FAILURE, "no such login class: %s",
168 login_class);
169 } else {
170 lc = login_getpwclass(pw);
171 if (lc == NULL)
172 errx(EXIT_FAILURE, "login_getpwclass failed");
173 }
174
175 /*
176 * This is not done with setusercontext() because that will
177 * try and use ~/.login_conf even when we don't want it to.
178 */
179 setclassenvironment(lc, pw, 1);
180 setclassenvironment(lc, pw, 0);
181 if (login_as_user) {
182 login_close(lc);
183 if ((lc = login_getuserclass(pw)) != NULL) {
184 setclassenvironment(lc, pw, 1);
185 setclassenvironment(lc, pw, 0);
186 }
187 }
188 endpwent();
189 if (lc != NULL)
190 login_close(lc);
191 }
192 for (argv += optind; *argv && (p = strchr(*argv, '=')); ++argv) {
193 if (env_verbosity)
194 fprintf(stderr, "#env setenv:\t%s\n", *argv);
195 *p = '\0';
196 rtrn = setenv(*argv, p + 1, 1);
197 *p = '=';
198 if (rtrn == -1)
199 err(EXIT_FAILURE, "setenv %s", *argv);
200 }
201 if (*argv) {
202 if (term == '\0')
203 errx(EXIT_CANCELED, "cannot specify command with -0");
204 if (altpath)
205 search_paths(altpath, argv);
206 if (env_verbosity) {
207 fprintf(stderr, "#env executing:\t%s\n", *argv);
208 for (parg = argv, argc = 0; *parg; parg++, argc++)
209 fprintf(stderr, "#env arg[%d]=\t'%s'\n",
210 argc, *parg);
211 if (env_verbosity > 1)
212 sleep(1);
213 }
214 execvp(*argv, argv);
215 err(errno == ENOENT ? EXIT_ENOENT : EXIT_CANNOT_INVOKE,
216 "%s", *argv);
217 }
218 for (ep = environ; *ep; ep++)
219 (void)printf("%s%c", *ep, term);
220 exit(0);
221 }
222
223 static void
usage(void)224 usage(void)
225 {
226 (void)fprintf(stderr,
227 "usage: env [-0iv] [-L|-U user[/class]] [-P utilpath] [-S string] [-u name]\n"
228 " [name=value ...] [utility [argument ...]]\n");
229 exit(1);
230 }
231