main.c revision 1.13 1 1.13 christos /* $NetBSD: main.c,v 1.13 2006/05/13 21:42:45 christos Exp $ */
2 1.2 tls
3 1.1 jtc /*
4 1.7 wiz * startup, main loop, environments and error handling
5 1.1 jtc */
6 1.10 agc #include <sys/cdefs.h>
7 1.10 agc
8 1.10 agc #ifndef lint
9 1.13 christos __RCSID("$NetBSD: main.c,v 1.13 2006/05/13 21:42:45 christos Exp $");
10 1.10 agc #endif
11 1.10 agc
12 1.1 jtc
13 1.1 jtc #define EXTERN /* define EXTERNs in sh.h */
14 1.1 jtc
15 1.1 jtc #include "sh.h"
16 1.1 jtc #include "ksh_stat.h"
17 1.1 jtc #include "ksh_time.h"
18 1.1 jtc
19 1.1 jtc extern char **environ;
20 1.1 jtc
21 1.1 jtc /*
22 1.1 jtc * global data
23 1.1 jtc */
24 1.1 jtc
25 1.2 tls static void reclaim ARGS((void));
26 1.2 tls static void remove_temps ARGS((struct temp *tp));
27 1.2 tls static int is_restricted ARGS((char *name));
28 1.1 jtc
29 1.1 jtc /*
30 1.1 jtc * shell initialization
31 1.1 jtc */
32 1.1 jtc
33 1.6 hubertf static const char initifs[] = "IFS= \t\n";
34 1.1 jtc
35 1.6 hubertf static const char initsubs[] = "${PS2=> } ${PS3=#? } ${PS4=+ }";
36 1.1 jtc
37 1.1 jtc static const char version_param[] =
38 1.1 jtc #ifdef KSH
39 1.1 jtc "KSH_VERSION"
40 1.1 jtc #else /* KSH */
41 1.1 jtc "SH_VERSION"
42 1.1 jtc #endif /* KSH */
43 1.1 jtc ;
44 1.1 jtc
45 1.2 tls static const char *const initcoms [] = {
46 1.1 jtc "typeset", "-x", "SHELL", "PATH", "HOME", NULL,
47 1.1 jtc "typeset", "-r", version_param, NULL,
48 1.6 hubertf "typeset", "-i", "PPID", NULL,
49 1.6 hubertf "typeset", "-i", "OPTIND=1", NULL,
50 1.1 jtc #ifdef KSH
51 1.6 hubertf "eval", "typeset -i RANDOM MAILCHECK=\"${MAILCHECK-600}\" SECONDS=\"${SECONDS-0}\" TMOUT=\"${TMOUT-0}\"", NULL,
52 1.1 jtc #endif /* KSH */
53 1.1 jtc "alias",
54 1.1 jtc /* Standard ksh aliases */
55 1.1 jtc "hash=alias -t", /* not "alias -t --": hash -r needs to work */
56 1.1 jtc "type=whence -v",
57 1.1 jtc #ifdef JOBS
58 1.1 jtc "stop=kill -STOP",
59 1.1 jtc "suspend=kill -STOP $$",
60 1.1 jtc #endif
61 1.1 jtc #ifdef KSH
62 1.1 jtc "autoload=typeset -fu",
63 1.1 jtc "functions=typeset -f",
64 1.2 tls # ifdef HISTORY
65 1.1 jtc "history=fc -l",
66 1.2 tls # endif /* HISTORY */
67 1.1 jtc "integer=typeset -i",
68 1.1 jtc "nohup=nohup ",
69 1.1 jtc "local=typeset",
70 1.1 jtc "r=fc -e -",
71 1.1 jtc #endif /* KSH */
72 1.1 jtc #ifdef KSH
73 1.1 jtc /* Aliases that are builtin commands in at&t */
74 1.1 jtc "login=exec login",
75 1.8 jschauma #ifndef __NetBSD__
76 1.1 jtc "newgrp=exec newgrp",
77 1.8 jschauma #endif /* __NetBSD__ */
78 1.1 jtc #endif /* KSH */
79 1.1 jtc NULL,
80 1.1 jtc /* this is what at&t ksh seems to track, with the addition of emacs */
81 1.1 jtc "alias", "-tU",
82 1.1 jtc "cat", "cc", "chmod", "cp", "date", "ed", "emacs", "grep", "ls",
83 1.1 jtc "mail", "make", "mv", "pr", "rm", "sed", "sh", "vi", "who",
84 1.1 jtc NULL,
85 1.1 jtc #ifdef EXTRA_INITCOMS
86 1.1 jtc EXTRA_INITCOMS, NULL,
87 1.1 jtc #endif /* EXTRA_INITCOMS */
88 1.1 jtc NULL
89 1.1 jtc };
90 1.3 christos
91 1.1 jtc int
92 1.11 mycroft main(int argc, char *argv[])
93 1.1 jtc {
94 1.1 jtc register int i;
95 1.1 jtc int argi;
96 1.1 jtc Source *s;
97 1.1 jtc struct block *l;
98 1.6 hubertf int restricted, errexit;
99 1.1 jtc char **wp;
100 1.1 jtc struct env env;
101 1.6 hubertf pid_t ppid;
102 1.1 jtc
103 1.1 jtc #ifdef MEM_DEBUG
104 1.6 hubertf chmem_set_defaults("ct", 1);
105 1.6 hubertf /* chmem_push("+c", 1); */
106 1.6 hubertf #endif /* MEM_DEBUG */
107 1.1 jtc
108 1.1 jtc #ifdef OS2
109 1.1 jtc setmode (0, O_BINARY);
110 1.1 jtc setmode (1, O_TEXT);
111 1.1 jtc #endif
112 1.1 jtc
113 1.1 jtc /* make sure argv[] is sane */
114 1.1 jtc if (!*argv) {
115 1.1 jtc static const char *empty_argv[] = {
116 1.1 jtc "pdksh", (char *) 0
117 1.1 jtc };
118 1.1 jtc
119 1.12 christos argv = (char **)__UNCONST(empty_argv);
120 1.1 jtc argc = 1;
121 1.1 jtc }
122 1.1 jtc kshname = *argv;
123 1.1 jtc
124 1.1 jtc ainit(&aperm); /* initialize permanent Area */
125 1.1 jtc
126 1.7 wiz /* set up base environment */
127 1.2 tls memset(&env, 0, sizeof(env));
128 1.1 jtc env.type = E_NONE;
129 1.1 jtc ainit(&env.area);
130 1.1 jtc e = &env;
131 1.1 jtc newblock(); /* set up global l->vars and l->funs */
132 1.1 jtc
133 1.1 jtc /* Do this first so output routines (eg, errorf, shellf) can work */
134 1.1 jtc initio();
135 1.1 jtc
136 1.1 jtc initvar();
137 1.1 jtc
138 1.1 jtc initctypes();
139 1.1 jtc
140 1.1 jtc inittraps();
141 1.1 jtc
142 1.1 jtc #ifdef KSH
143 1.1 jtc coproc_init();
144 1.1 jtc #endif /* KSH */
145 1.1 jtc
146 1.1 jtc /* set up variable and command dictionaries */
147 1.1 jtc tinit(&taliases, APERM, 0);
148 1.1 jtc tinit(&aliases, APERM, 0);
149 1.1 jtc tinit(&homedirs, APERM, 0);
150 1.1 jtc
151 1.1 jtc /* define shell keywords */
152 1.1 jtc initkeywords();
153 1.1 jtc
154 1.1 jtc /* define built-in commands */
155 1.1 jtc tinit(&builtins, APERM, 64); /* must be 2^n (currently 40 builtins) */
156 1.1 jtc for (i = 0; shbuiltins[i].name != NULL; i++)
157 1.1 jtc builtin(shbuiltins[i].name, shbuiltins[i].func);
158 1.1 jtc for (i = 0; kshbuiltins[i].name != NULL; i++)
159 1.1 jtc builtin(kshbuiltins[i].name, kshbuiltins[i].func);
160 1.1 jtc
161 1.1 jtc init_histvec();
162 1.1 jtc
163 1.1 jtc def_path = DEFAULT__PATH;
164 1.1 jtc #if defined(HAVE_CONFSTR) && defined(_CS_PATH)
165 1.1 jtc {
166 1.1 jtc size_t len = confstr(_CS_PATH, (char *) 0, 0);
167 1.1 jtc char *new;
168 1.1 jtc
169 1.1 jtc if (len > 0) {
170 1.1 jtc confstr(_CS_PATH, new = alloc(len + 1, APERM), len + 1);
171 1.1 jtc def_path = new;
172 1.1 jtc }
173 1.1 jtc }
174 1.1 jtc #endif /* HAVE_CONFSTR && _CS_PATH */
175 1.6 hubertf
176 1.6 hubertf /* Set PATH to def_path (will set the path global variable).
177 1.6 hubertf * (import of environment below will probably change this setting).
178 1.6 hubertf */
179 1.6 hubertf {
180 1.6 hubertf struct tbl *vp = global("PATH");
181 1.6 hubertf /* setstr can't fail here */
182 1.6 hubertf setstr(vp, def_path, KSH_RETURN_ERROR);
183 1.6 hubertf }
184 1.1 jtc
185 1.1 jtc
186 1.11 mycroft /* Turn on nohup by default for now - will change to off
187 1.11 mycroft * by default once people are aware of its existence
188 1.1 jtc * (at&t ksh does not have a nohup option - it always sends
189 1.1 jtc * the hup).
190 1.1 jtc */
191 1.1 jtc Flag(FNOHUP) = 1;
192 1.1 jtc
193 1.1 jtc /* Turn on brace expansion by default. At&t ksh's that have
194 1.1 jtc * alternation always have it on. BUT, posix doesn't have
195 1.1 jtc * brace expansion, so set this before setting up FPOSIX
196 1.1 jtc * (change_flag() clears FBRACEEXPAND when FPOSIX is set).
197 1.1 jtc */
198 1.1 jtc #ifdef BRACE_EXPAND
199 1.1 jtc Flag(FBRACEEXPAND) = 1;
200 1.1 jtc #endif /* BRACE_EXPAND */
201 1.1 jtc
202 1.1 jtc /* set posix flag just before environment so that it will have
203 1.1 jtc * exactly the same effect as the POSIXLY_CORRECT environment
204 1.1 jtc * variable. If this needs to be done sooner to ensure correct posix
205 1.1 jtc * operation, an initial scan of the environment will also have
206 1.1 jtc * done sooner.
207 1.1 jtc */
208 1.1 jtc #ifdef POSIXLY_CORRECT
209 1.1 jtc change_flag(FPOSIX, OF_SPECIAL, 1);
210 1.1 jtc #endif /* POSIXLY_CORRECT */
211 1.9 provos
212 1.9 provos /* Set edit mode to emacs by default, may be overridden
213 1.9 provos * by the environment or the user. Also, we want tab completion
214 1.9 provos * on in vi by default. */
215 1.9 provos #if defined(EDIT) && defined(EMACS)
216 1.9 provos change_flag(FEMACS, OF_SPECIAL, 1);
217 1.9 provos #endif /* EDIT && EMACS */
218 1.9 provos #if defined(EDIT) && defined(VI)
219 1.9 provos Flag(FVITABCOMPLETE) = 1;
220 1.9 provos #endif /* EDIT && VI */
221 1.1 jtc
222 1.7 wiz /* import environment */
223 1.1 jtc if (environ != NULL)
224 1.1 jtc for (wp = environ; *wp != NULL; wp++)
225 1.1 jtc typeset(*wp, IMPORT|EXPORT, 0, 0, 0);
226 1.1 jtc
227 1.1 jtc kshpid = procpid = getpid();
228 1.1 jtc typeset(initifs, 0, 0, 0, 0); /* for security */
229 1.1 jtc
230 1.1 jtc /* assign default shell variable values */
231 1.1 jtc substitute(initsubs, 0);
232 1.1 jtc
233 1.1 jtc /* Figure out the current working directory and set $PWD */
234 1.1 jtc {
235 1.1 jtc struct stat s_pwd, s_dot;
236 1.1 jtc struct tbl *pwd_v = global("PWD");
237 1.1 jtc char *pwd = str_val(pwd_v);
238 1.1 jtc char *pwdx = pwd;
239 1.1 jtc
240 1.1 jtc /* Try to use existing $PWD if it is valid */
241 1.1 jtc if (!ISABSPATH(pwd)
242 1.1 jtc || stat(pwd, &s_pwd) < 0 || stat(".", &s_dot) < 0
243 1.1 jtc || s_pwd.st_dev != s_dot.st_dev
244 1.1 jtc || s_pwd.st_ino != s_dot.st_ino)
245 1.1 jtc pwdx = (char *) 0;
246 1.1 jtc set_current_wd(pwdx);
247 1.1 jtc if (current_wd[0])
248 1.1 jtc simplify_path(current_wd);
249 1.1 jtc /* Only set pwd if we know where we are or if it had a
250 1.1 jtc * bogus value
251 1.1 jtc */
252 1.1 jtc if (current_wd[0] || pwd != null)
253 1.6 hubertf /* setstr can't fail here */
254 1.6 hubertf setstr(pwd_v, current_wd, KSH_RETURN_ERROR);
255 1.1 jtc }
256 1.6 hubertf ppid = getppid();
257 1.6 hubertf setint(global("PPID"), (long) ppid);
258 1.1 jtc #ifdef KSH
259 1.6 hubertf setint(global("RANDOM"), (long) (time((time_t *)0) * kshpid * ppid));
260 1.1 jtc #endif /* KSH */
261 1.6 hubertf /* setstr can't fail here */
262 1.6 hubertf setstr(global(version_param), ksh_version, KSH_RETURN_ERROR);
263 1.1 jtc
264 1.1 jtc /* execute initialization statements */
265 1.12 christos for (wp = (char**)__UNCONST(initcoms); *wp != NULL; wp++) {
266 1.1 jtc shcomexec(wp);
267 1.1 jtc for (; *wp != NULL; wp++)
268 1.1 jtc ;
269 1.1 jtc }
270 1.1 jtc
271 1.6 hubertf
272 1.6 hubertf ksheuid = geteuid();
273 1.6 hubertf safe_prompt = ksheuid ? "$ " : "# ";
274 1.1 jtc {
275 1.1 jtc struct tbl *vp = global("PS1");
276 1.1 jtc
277 1.1 jtc /* Set PS1 if it isn't set, or we are root and prompt doesn't
278 1.1 jtc * contain a #.
279 1.1 jtc */
280 1.6 hubertf if (!(vp->flag & ISSET)
281 1.6 hubertf || (!ksheuid && !strchr(str_val(vp), '#')))
282 1.6 hubertf /* setstr can't fail here */
283 1.6 hubertf setstr(vp, safe_prompt, KSH_RETURN_ERROR);
284 1.1 jtc }
285 1.1 jtc
286 1.1 jtc /* Set this before parsing arguments */
287 1.6 hubertf Flag(FPRIVILEGED) = getuid() != ksheuid || getgid() != getegid();
288 1.1 jtc
289 1.1 jtc /* this to note if monitor is set on command line (see below) */
290 1.1 jtc Flag(FMONITOR) = 127;
291 1.1 jtc argi = parse_args(argv, OF_CMDLINE, (int *) 0);
292 1.4 mycroft if (argi < 0) {
293 1.1 jtc exit(1);
294 1.4 mycroft /* NOTREACHED */
295 1.4 mycroft }
296 1.1 jtc
297 1.1 jtc if (Flag(FCOMMAND)) {
298 1.1 jtc s = pushs(SSTRING, ATEMP);
299 1.1 jtc if (!(s->start = s->str = argv[argi++]))
300 1.1 jtc errorf("-c requires an argument");
301 1.1 jtc if (argv[argi])
302 1.1 jtc kshname = argv[argi++];
303 1.1 jtc } else if (argi < argc && !Flag(FSTDIN)) {
304 1.1 jtc s = pushs(SFILE, ATEMP);
305 1.1 jtc #ifdef OS2
306 1.1 jtc /* a bug in os2 extproc shell processing doesn't
307 1.1 jtc * pass full pathnames so we have to search for it.
308 1.1 jtc * This changes the behavior of 'ksh arg' to search
309 1.1 jtc * the users search path but it can't be helped.
310 1.1 jtc */
311 1.1 jtc s->file = search(argv[argi++], path, R_OK, (int *) 0);
312 1.1 jtc if (!s->file || !*s->file)
313 1.1 jtc s->file = argv[argi - 1];
314 1.1 jtc #else
315 1.1 jtc s->file = argv[argi++];
316 1.1 jtc #endif /* OS2 */
317 1.1 jtc s->u.shf = shf_open(s->file, O_RDONLY, 0, SHF_MAPHI|SHF_CLEXEC);
318 1.1 jtc if (s->u.shf == NULL) {
319 1.1 jtc exstat = 127; /* POSIX */
320 1.1 jtc errorf("%s: %s", s->file, strerror(errno));
321 1.1 jtc }
322 1.1 jtc kshname = s->file;
323 1.1 jtc } else {
324 1.1 jtc Flag(FSTDIN) = 1;
325 1.1 jtc s = pushs(SSTDIN, ATEMP);
326 1.1 jtc s->file = "<stdin>";
327 1.1 jtc s->u.shf = shf_fdopen(0, SHF_RD | can_seek(0),
328 1.1 jtc (struct shf *) 0);
329 1.1 jtc if (isatty(0) && isatty(2)) {
330 1.6 hubertf Flag(FTALKING) = Flag(FTALKING_I) = 1;
331 1.1 jtc /* The following only if isatty(0) */
332 1.1 jtc s->flags |= SF_TTY;
333 1.1 jtc s->u.shf->flags |= SHF_INTERRUPT;
334 1.1 jtc s->file = (char *) 0;
335 1.1 jtc }
336 1.1 jtc }
337 1.1 jtc
338 1.1 jtc /* This bizarreness is mandated by POSIX */
339 1.1 jtc {
340 1.1 jtc struct stat s_stdin;
341 1.1 jtc
342 1.11 mycroft if (fstat(0, &s_stdin) >= 0 && S_ISCHR(s_stdin.st_mode) &&
343 1.11 mycroft Flag(FTALKING))
344 1.1 jtc reset_nonblock(0);
345 1.1 jtc }
346 1.1 jtc
347 1.1 jtc /* initialize job control */
348 1.1 jtc i = Flag(FMONITOR) != 127;
349 1.1 jtc Flag(FMONITOR) = 0;
350 1.1 jtc j_init(i);
351 1.1 jtc #ifdef EDIT
352 1.1 jtc /* Do this after j_init(), as tty_fd is not initialized 'til then */
353 1.1 jtc if (Flag(FTALKING))
354 1.1 jtc x_init();
355 1.1 jtc #endif
356 1.1 jtc
357 1.1 jtc l = e->loc;
358 1.1 jtc l->argv = &argv[argi - 1];
359 1.1 jtc l->argc = argc - argi;
360 1.12 christos l->argv[0] = (char *)__UNCONST(kshname);
361 1.1 jtc getopts_reset(1);
362 1.1 jtc
363 1.1 jtc /* Disable during .profile/ENV reading */
364 1.1 jtc restricted = Flag(FRESTRICTED);
365 1.1 jtc Flag(FRESTRICTED) = 0;
366 1.6 hubertf errexit = Flag(FERREXIT);
367 1.6 hubertf Flag(FERREXIT) = 0;
368 1.1 jtc
369 1.1 jtc /* Do this before profile/$ENV so that if it causes problems in them,
370 1.1 jtc * user will know why things broke.
371 1.1 jtc */
372 1.1 jtc if (!current_wd[0] && Flag(FTALKING))
373 1.1 jtc warningf(FALSE, "Cannot determine current working directory");
374 1.1 jtc
375 1.1 jtc if (Flag(FLOGIN)) {
376 1.1 jtc #ifdef OS2
377 1.1 jtc char *profile;
378 1.1 jtc
379 1.1 jtc /* Try to find a profile - first see if $INIT has a value,
380 1.1 jtc * then try /etc/profile.ksh, then c:/usr/etc/profile.ksh.
381 1.1 jtc */
382 1.1 jtc if (!Flag(FPRIVILEGED)
383 1.1 jtc && strcmp(profile = substitute("$INIT/profile.ksh", 0),
384 1.1 jtc "/profile.ksh"))
385 1.1 jtc include(profile, 0, (char **) 0, 1);
386 1.1 jtc else if (include("/etc/profile.ksh", 0, (char **) 0, 1) < 0)
387 1.1 jtc include("c:/usr/etc/profile.ksh", 0, (char **) 0, 1);
388 1.1 jtc if (!Flag(FPRIVILEGED))
389 1.1 jtc include(substitute("$HOME/profile.ksh", 0), 0,
390 1.1 jtc (char **) 0, 1);
391 1.1 jtc #else /* OS2 */
392 1.2 tls include(KSH_SYSTEM_PROFILE, 0, (char **) 0, 1);
393 1.1 jtc if (!Flag(FPRIVILEGED))
394 1.1 jtc include(substitute("$HOME/.profile", 0), 0,
395 1.1 jtc (char **) 0, 1);
396 1.1 jtc #endif /* OS2 */
397 1.1 jtc }
398 1.1 jtc
399 1.1 jtc if (Flag(FPRIVILEGED))
400 1.1 jtc include("/etc/suid_profile", 0, (char **) 0, 1);
401 1.1 jtc else {
402 1.1 jtc char *env_file;
403 1.1 jtc
404 1.1 jtc #ifndef KSH
405 1.1 jtc if (!Flag(FPOSIX))
406 1.1 jtc env_file = null;
407 1.1 jtc else
408 1.1 jtc #endif /* !KSH */
409 1.1 jtc /* include $ENV */
410 1.1 jtc env_file = str_val(global("ENV"));
411 1.1 jtc
412 1.1 jtc #ifdef DEFAULT_ENV
413 1.1 jtc /* If env isn't set, include default environment */
414 1.1 jtc if (env_file == null)
415 1.12 christos env_file = __UNCONST(DEFAULT_ENV);
416 1.1 jtc #endif /* DEFAULT_ENV */
417 1.1 jtc env_file = substitute(env_file, DOTILDE);
418 1.1 jtc if (*env_file != '\0')
419 1.1 jtc include(env_file, 0, (char **) 0, 1);
420 1.1 jtc #ifdef OS2
421 1.1 jtc else if (Flag(FTALKING))
422 1.1 jtc include(substitute("$HOME/kshrc.ksh", 0), 0,
423 1.1 jtc (char **) 0, 1);
424 1.1 jtc #endif /* OS2 */
425 1.1 jtc }
426 1.1 jtc
427 1.1 jtc if (is_restricted(argv[0]) || is_restricted(str_val(global("SHELL"))))
428 1.1 jtc restricted = 1;
429 1.1 jtc if (restricted) {
430 1.1 jtc static const char *const restr_com[] = {
431 1.1 jtc "typeset", "-r", "PATH",
432 1.1 jtc "ENV", "SHELL",
433 1.1 jtc (char *) 0
434 1.1 jtc };
435 1.12 christos shcomexec((char **)__UNCONST(restr_com));
436 1.1 jtc /* After typeset command... */
437 1.1 jtc Flag(FRESTRICTED) = 1;
438 1.1 jtc }
439 1.6 hubertf if (errexit)
440 1.6 hubertf Flag(FERREXIT) = 1;
441 1.1 jtc
442 1.1 jtc if (Flag(FTALKING)) {
443 1.1 jtc hist_init(s);
444 1.1 jtc #ifdef KSH
445 1.1 jtc alarm_init();
446 1.1 jtc #endif /* KSH */
447 1.1 jtc } else
448 1.1 jtc Flag(FTRACKALL) = 1; /* set after ENV */
449 1.1 jtc
450 1.1 jtc shell(s, TRUE); /* doesn't return */
451 1.1 jtc return 0;
452 1.1 jtc }
453 1.1 jtc
454 1.1 jtc int
455 1.1 jtc include(name, argc, argv, intr_ok)
456 1.1 jtc const char *name;
457 1.1 jtc int argc;
458 1.1 jtc char **argv;
459 1.1 jtc int intr_ok;
460 1.1 jtc {
461 1.1 jtc register Source *volatile s = NULL;
462 1.1 jtc struct shf *shf;
463 1.1 jtc char **volatile old_argv;
464 1.1 jtc volatile int old_argc;
465 1.1 jtc int i;
466 1.1 jtc
467 1.1 jtc shf = shf_open(name, O_RDONLY, 0, SHF_MAPHI|SHF_CLEXEC);
468 1.1 jtc if (shf == NULL)
469 1.1 jtc return -1;
470 1.1 jtc
471 1.1 jtc if (argv) {
472 1.1 jtc old_argv = e->loc->argv;
473 1.1 jtc old_argc = e->loc->argc;
474 1.1 jtc } else {
475 1.1 jtc old_argv = (char **) 0;
476 1.1 jtc old_argc = 0;
477 1.1 jtc }
478 1.1 jtc newenv(E_INCL);
479 1.1 jtc i = ksh_sigsetjmp(e->jbuf, 0);
480 1.1 jtc if (i) {
481 1.6 hubertf if (s) /* Do this before quitenv(), which frees the memory */
482 1.1 jtc shf_close(s->u.shf);
483 1.6 hubertf quitenv();
484 1.1 jtc if (old_argv) {
485 1.1 jtc e->loc->argv = old_argv;
486 1.1 jtc e->loc->argc = old_argc;
487 1.1 jtc }
488 1.1 jtc switch (i) {
489 1.1 jtc case LRETURN:
490 1.1 jtc case LERROR:
491 1.1 jtc return exstat & 0xff; /* see below */
492 1.1 jtc case LINTR:
493 1.1 jtc /* intr_ok is set if we are including .profile or $ENV.
494 1.1 jtc * If user ^C's out, we don't want to kill the shell...
495 1.1 jtc */
496 1.1 jtc if (intr_ok && (exstat - 128) != SIGTERM)
497 1.1 jtc return 1;
498 1.1 jtc /* fall through... */
499 1.1 jtc case LEXIT:
500 1.1 jtc case LLEAVE:
501 1.1 jtc case LSHELL:
502 1.1 jtc unwind(i);
503 1.1 jtc /*NOREACHED*/
504 1.1 jtc default:
505 1.1 jtc internal_errorf(1, "include: %d", i);
506 1.1 jtc /*NOREACHED*/
507 1.1 jtc }
508 1.1 jtc }
509 1.1 jtc if (argv) {
510 1.1 jtc e->loc->argv = argv;
511 1.1 jtc e->loc->argc = argc;
512 1.1 jtc }
513 1.1 jtc s = pushs(SFILE, ATEMP);
514 1.1 jtc s->u.shf = shf;
515 1.1 jtc s->file = str_save(name, ATEMP);
516 1.1 jtc i = shell(s, FALSE);
517 1.1 jtc shf_close(s->u.shf);
518 1.6 hubertf quitenv();
519 1.1 jtc if (old_argv) {
520 1.1 jtc e->loc->argv = old_argv;
521 1.1 jtc e->loc->argc = old_argc;
522 1.1 jtc }
523 1.1 jtc return i & 0xff; /* & 0xff to ensure value not -1 */
524 1.1 jtc }
525 1.1 jtc
526 1.1 jtc int
527 1.1 jtc command(comm)
528 1.1 jtc const char *comm;
529 1.1 jtc {
530 1.1 jtc register Source *s;
531 1.13 christos int r;
532 1.1 jtc
533 1.1 jtc s = pushs(SSTRING, ATEMP);
534 1.1 jtc s->start = s->str = comm;
535 1.13 christos r = shell(s, FALSE);
536 1.13 christos afree(s, ATEMP);
537 1.13 christos return r;
538 1.1 jtc }
539 1.1 jtc
540 1.1 jtc /*
541 1.1 jtc * run the commands from the input source, returning status.
542 1.1 jtc */
543 1.1 jtc int
544 1.1 jtc shell(s, toplevel)
545 1.1 jtc Source *volatile s; /* input source */
546 1.1 jtc int volatile toplevel;
547 1.1 jtc {
548 1.1 jtc struct op *t;
549 1.1 jtc volatile int wastty = s->flags & SF_TTY;
550 1.1 jtc volatile int attempts = 13;
551 1.1 jtc volatile int interactive = Flag(FTALKING) && toplevel;
552 1.11 mycroft Source *volatile old_source = source;
553 1.1 jtc int i;
554 1.1 jtc
555 1.1 jtc newenv(E_PARSE);
556 1.1 jtc if (interactive)
557 1.1 jtc really_exit = 0;
558 1.1 jtc i = ksh_sigsetjmp(e->jbuf, 0);
559 1.1 jtc if (i) {
560 1.1 jtc switch (i) {
561 1.1 jtc case LINTR: /* we get here if SIGINT not caught or ignored */
562 1.1 jtc case LERROR:
563 1.1 jtc case LSHELL:
564 1.1 jtc if (interactive) {
565 1.1 jtc if (i == LINTR)
566 1.1 jtc shellf(newline);
567 1.1 jtc /* Reset any eof that was read as part of a
568 1.1 jtc * multiline command.
569 1.1 jtc */
570 1.1 jtc if (Flag(FIGNOREEOF) && s->type == SEOF
571 1.1 jtc && wastty)
572 1.1 jtc s->type = SSTDIN;
573 1.1 jtc /* Used by exit command to get back to
574 1.1 jtc * top level shell. Kind of strange since
575 1.1 jtc * interactive is set if we are reading from
576 1.1 jtc * a tty, but to have stopped jobs, one only
577 1.1 jtc * needs FMONITOR set (not FTALKING/SF_TTY)...
578 1.1 jtc */
579 1.11 mycroft /* toss any input we have so far */
580 1.11 mycroft s->start = s->str = null;
581 1.1 jtc break;
582 1.1 jtc }
583 1.1 jtc /* fall through... */
584 1.1 jtc case LEXIT:
585 1.1 jtc case LLEAVE:
586 1.1 jtc case LRETURN:
587 1.11 mycroft source = old_source;
588 1.1 jtc quitenv();
589 1.1 jtc unwind(i); /* keep on going */
590 1.1 jtc /*NOREACHED*/
591 1.1 jtc default:
592 1.11 mycroft source = old_source;
593 1.1 jtc quitenv();
594 1.1 jtc internal_errorf(1, "shell: %d", i);
595 1.1 jtc /*NOREACHED*/
596 1.1 jtc }
597 1.1 jtc }
598 1.1 jtc
599 1.1 jtc while (1) {
600 1.1 jtc if (trap)
601 1.1 jtc runtraps(0);
602 1.1 jtc
603 1.5 thorpej if (s->next == NULL) {
604 1.1 jtc if (Flag(FVERBOSE))
605 1.1 jtc s->flags |= SF_ECHO;
606 1.1 jtc else
607 1.1 jtc s->flags &= ~SF_ECHO;
608 1.5 thorpej }
609 1.1 jtc
610 1.1 jtc if (interactive) {
611 1.1 jtc j_notify();
612 1.1 jtc #ifdef KSH
613 1.1 jtc mcheck();
614 1.1 jtc #endif /* KSH */
615 1.1 jtc set_prompt(PS1, s);
616 1.1 jtc }
617 1.1 jtc
618 1.1 jtc t = compile(s);
619 1.1 jtc if (t != NULL && t->type == TEOF) {
620 1.1 jtc if (wastty && Flag(FIGNOREEOF) && --attempts > 0) {
621 1.1 jtc shellf("Use `exit' to leave ksh\n");
622 1.1 jtc s->type = SSTDIN;
623 1.1 jtc } else if (wastty && !really_exit
624 1.1 jtc && j_stopped_running())
625 1.1 jtc {
626 1.1 jtc really_exit = 1;
627 1.1 jtc s->type = SSTDIN;
628 1.1 jtc } else {
629 1.1 jtc /* this for POSIX, which says EXIT traps
630 1.1 jtc * shall be taken in the environment
631 1.1 jtc * immediately after the last command
632 1.1 jtc * executed.
633 1.1 jtc */
634 1.1 jtc if (toplevel)
635 1.1 jtc unwind(LEXIT);
636 1.1 jtc break;
637 1.1 jtc }
638 1.1 jtc }
639 1.1 jtc
640 1.1 jtc if (t && (!Flag(FNOEXEC) || (s->flags & SF_TTY)))
641 1.1 jtc exstat = execute(t, 0);
642 1.1 jtc
643 1.1 jtc if (t != NULL && t->type != TEOF && interactive && really_exit)
644 1.1 jtc really_exit = 0;
645 1.1 jtc
646 1.1 jtc reclaim();
647 1.1 jtc }
648 1.1 jtc quitenv();
649 1.11 mycroft source = old_source;
650 1.1 jtc return exstat;
651 1.1 jtc }
652 1.1 jtc
653 1.1 jtc /* return to closest error handler or shell(), exit if none found */
654 1.1 jtc void
655 1.1 jtc unwind(i)
656 1.1 jtc int i;
657 1.1 jtc {
658 1.1 jtc /* ordering for EXIT vs ERR is a bit odd (this is what at&t ksh does) */
659 1.1 jtc if (i == LEXIT || (Flag(FERREXIT) && (i == LERROR || i == LINTR)
660 1.1 jtc && sigtraps[SIGEXIT_].trap))
661 1.1 jtc {
662 1.1 jtc runtrap(&sigtraps[SIGEXIT_]);
663 1.1 jtc i = LLEAVE;
664 1.1 jtc } else if (Flag(FERREXIT) && (i == LERROR || i == LINTR)) {
665 1.1 jtc runtrap(&sigtraps[SIGERR_]);
666 1.1 jtc i = LLEAVE;
667 1.1 jtc }
668 1.1 jtc while (1) {
669 1.1 jtc switch (e->type) {
670 1.1 jtc case E_PARSE:
671 1.1 jtc case E_FUNC:
672 1.1 jtc case E_INCL:
673 1.1 jtc case E_LOOP:
674 1.1 jtc case E_ERRH:
675 1.1 jtc ksh_siglongjmp(e->jbuf, i);
676 1.1 jtc /*NOTREACHED*/
677 1.1 jtc
678 1.6 hubertf case E_NONE:
679 1.6 hubertf if (i == LINTR)
680 1.6 hubertf e->flags |= EF_FAKE_SIGDIE;
681 1.6 hubertf /* Fall through... */
682 1.1 jtc
683 1.1 jtc default:
684 1.1 jtc quitenv();
685 1.1 jtc }
686 1.1 jtc }
687 1.1 jtc }
688 1.1 jtc
689 1.1 jtc void
690 1.1 jtc newenv(type)
691 1.1 jtc int type;
692 1.1 jtc {
693 1.1 jtc register struct env *ep;
694 1.1 jtc
695 1.1 jtc ep = (struct env *) alloc(sizeof(*ep), ATEMP);
696 1.1 jtc ep->type = type;
697 1.1 jtc ep->flags = 0;
698 1.1 jtc ainit(&ep->area);
699 1.1 jtc ep->loc = e->loc;
700 1.1 jtc ep->savefd = NULL;
701 1.1 jtc ep->oenv = e;
702 1.1 jtc ep->temps = NULL;
703 1.1 jtc e = ep;
704 1.1 jtc }
705 1.1 jtc
706 1.1 jtc void
707 1.1 jtc quitenv()
708 1.1 jtc {
709 1.1 jtc register struct env *ep = e;
710 1.1 jtc register int fd;
711 1.1 jtc
712 1.6 hubertf if (ep->oenv && ep->oenv->loc != ep->loc)
713 1.1 jtc popblock();
714 1.1 jtc if (ep->savefd != NULL) {
715 1.1 jtc for (fd = 0; fd < NUFILE; fd++)
716 1.1 jtc /* if ep->savefd[fd] < 0, means fd was closed */
717 1.1 jtc if (ep->savefd[fd])
718 1.1 jtc restfd(fd, ep->savefd[fd]);
719 1.1 jtc if (ep->savefd[2]) /* Clear any write errors */
720 1.1 jtc shf_reopen(2, SHF_WR, shl_out);
721 1.1 jtc }
722 1.1 jtc reclaim();
723 1.6 hubertf
724 1.6 hubertf /* Bottom of the stack.
725 1.6 hubertf * Either main shell is exiting or cleanup_parents_env() was called.
726 1.6 hubertf */
727 1.6 hubertf if (ep->oenv == NULL) {
728 1.6 hubertf if (ep->type == E_NONE) { /* Main shell exiting? */
729 1.6 hubertf if (Flag(FTALKING))
730 1.6 hubertf hist_finish();
731 1.6 hubertf j_exit();
732 1.6 hubertf if (ep->flags & EF_FAKE_SIGDIE) {
733 1.6 hubertf int sig = exstat - 128;
734 1.6 hubertf
735 1.6 hubertf /* ham up our death a bit (at&t ksh
736 1.6 hubertf * only seems to do this for SIGTERM)
737 1.6 hubertf * Don't do it for SIGQUIT, since we'd
738 1.6 hubertf * dump a core..
739 1.6 hubertf */
740 1.6 hubertf if (sig == SIGINT || sig == SIGTERM) {
741 1.6 hubertf setsig(&sigtraps[sig], SIG_DFL,
742 1.6 hubertf SS_RESTORE_CURR|SS_FORCE);
743 1.6 hubertf kill(0, sig);
744 1.6 hubertf }
745 1.6 hubertf }
746 1.6 hubertf #ifdef MEM_DEBUG
747 1.6 hubertf chmem_allfree();
748 1.6 hubertf #endif /* MEM_DEBUG */
749 1.6 hubertf }
750 1.6 hubertf exit(exstat);
751 1.6 hubertf }
752 1.6 hubertf
753 1.1 jtc e = e->oenv;
754 1.1 jtc afree(ep, ATEMP);
755 1.1 jtc }
756 1.1 jtc
757 1.1 jtc /* Called after a fork to cleanup stuff left over from parents environment */
758 1.1 jtc void
759 1.1 jtc cleanup_parents_env()
760 1.1 jtc {
761 1.1 jtc struct env *ep;
762 1.1 jtc int fd;
763 1.1 jtc
764 1.1 jtc /* Don't clean up temporary files - parent will probably need them.
765 1.1 jtc * Also, can't easily reclaim memory since variables, etc. could be
766 1.11 mycroft * anywhere.
767 1.1 jtc */
768 1.1 jtc
769 1.1 jtc /* close all file descriptors hiding in savefd */
770 1.1 jtc for (ep = e; ep; ep = ep->oenv) {
771 1.6 hubertf if (ep->savefd) {
772 1.1 jtc for (fd = 0; fd < NUFILE; fd++)
773 1.1 jtc if (ep->savefd[fd] > 0)
774 1.1 jtc close(ep->savefd[fd]);
775 1.6 hubertf afree(ep->savefd, &ep->area);
776 1.6 hubertf ep->savefd = (short *) 0;
777 1.6 hubertf }
778 1.1 jtc }
779 1.1 jtc e->oenv = (struct env *) 0;
780 1.1 jtc }
781 1.1 jtc
782 1.2 tls /* Called just before an execve cleanup stuff temporary files */
783 1.2 tls void
784 1.2 tls cleanup_proc_env()
785 1.2 tls {
786 1.2 tls struct env *ep;
787 1.2 tls
788 1.2 tls for (ep = e; ep; ep = ep->oenv)
789 1.2 tls remove_temps(ep->temps);
790 1.2 tls }
791 1.2 tls
792 1.1 jtc /* remove temp files and free ATEMP Area */
793 1.1 jtc static void
794 1.1 jtc reclaim()
795 1.1 jtc {
796 1.1 jtc remove_temps(e->temps);
797 1.1 jtc e->temps = NULL;
798 1.1 jtc afreeall(&e->area);
799 1.1 jtc }
800 1.1 jtc
801 1.1 jtc static void
802 1.1 jtc remove_temps(tp)
803 1.1 jtc struct temp *tp;
804 1.1 jtc {
805 1.1 jtc #ifdef OS2
806 1.2 tls static struct temp *delayed_remove;
807 1.2 tls struct temp *t, **tprev;
808 1.1 jtc
809 1.2 tls if (delayed_remove) {
810 1.2 tls for (tprev = &delayed_remove, t = delayed_remove; t; t = *tprev)
811 1.2 tls /* No need to check t->pid here... */
812 1.2 tls if (unlink(t->name) >= 0 || errno == ENOENT) {
813 1.2 tls *tprev = t->next;
814 1.2 tls afree(t, APERM);
815 1.2 tls } else
816 1.2 tls tprev = &t->next;
817 1.2 tls }
818 1.1 jtc #endif /* OS2 */
819 1.1 jtc
820 1.1 jtc for (; tp != NULL; tp = tp->next)
821 1.2 tls if (tp->pid == procpid) {
822 1.1 jtc #ifdef OS2
823 1.2 tls /* OS/2 (and dos) do not allow files that are currently
824 1.2 tls * open to be removed, so we cache it away for future
825 1.2 tls * removal.
826 1.2 tls * XXX should only do this if errno
827 1.2 tls * is Efile-still-open-can't-remove
828 1.2 tls * (but I don't know what that is...)
829 1.2 tls */
830 1.2 tls if (unlink(tp->name) < 0 && errno != ENOENT) {
831 1.2 tls t = (struct temp *) alloc(
832 1.2 tls sizeof(struct temp) + strlen(tp->name) + 1,
833 1.2 tls APERM);
834 1.2 tls memset(t, 0, sizeof(struct temp));
835 1.6 hubertf t->name = (char *) &t[1];
836 1.11 mycroft strlcpy(t->name, tp->name, strlen(tp->name) + 1);
837 1.2 tls t->next = delayed_remove;
838 1.2 tls delayed_remove = t;
839 1.2 tls }
840 1.1 jtc #else /* OS2 */
841 1.1 jtc unlink(tp->name);
842 1.1 jtc #endif /* OS2 */
843 1.2 tls }
844 1.1 jtc }
845 1.1 jtc
846 1.1 jtc /* Returns true if name refers to a restricted shell */
847 1.1 jtc static int
848 1.1 jtc is_restricted(name)
849 1.1 jtc char *name;
850 1.1 jtc {
851 1.1 jtc char *p;
852 1.1 jtc
853 1.1 jtc if ((p = ksh_strrchr_dirsep(name)))
854 1.1 jtc name = p;
855 1.1 jtc /* accepts rsh, rksh, rpdksh, pdrksh, etc. */
856 1.1 jtc return (p = strchr(name, 'r')) && strstr(p, "sh");
857 1.1 jtc }
858 1.1 jtc
859 1.1 jtc void
860 1.1 jtc aerror(ap, msg)
861 1.1 jtc Area *ap;
862 1.1 jtc const char *msg;
863 1.1 jtc {
864 1.1 jtc internal_errorf(1, "alloc: %s", msg);
865 1.1 jtc errorf(null); /* this is never executed - keeps gcc quiet */
866 1.1 jtc /*NOTREACHED*/
867 1.1 jtc }
868