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