Merge branch 'vendor/GCC44'
[dragonfly.git] / bin / sh / main.c
1 /*-
2  * Copyright (c) 1991, 1993
3  *      The Regents of the University of California.  All rights reserved.
4  *
5  * This code is derived from software contributed to Berkeley by
6  * Kenneth Almquist.
7  *
8  * Redistribution and use in source and binary forms, with or without
9  * modification, are permitted provided that the following conditions
10  * are met:
11  * 1. Redistributions of source code must retain the above copyright
12  *    notice, this list of conditions and the following disclaimer.
13  * 2. Redistributions in binary form must reproduce the above copyright
14  *    notice, this list of conditions and the following disclaimer in the
15  *    documentation and/or other materials provided with the distribution.
16  * 3. All advertising materials mentioning features or use of this software
17  *    must display the following acknowledgement:
18  *      This product includes software developed by the University of
19  *      California, Berkeley and its contributors.
20  * 4. Neither the name of the University nor the names of its contributors
21  *    may be used to endorse or promote products derived from this software
22  *    without specific prior written permission.
23  *
24  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
25  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
26  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
27  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
28  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
29  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
30  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
31  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
32  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
33  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
34  * SUCH DAMAGE.
35  *
36  * @(#) Copyright (c) 1991, 1993 The Regents of the University of California.  All rights reserved.
37  * @(#)main.c   8.6 (Berkeley) 5/28/95
38  * $FreeBSD: src/bin/sh/main.c,v 1.53 2012/01/01 22:17:12 jilles Exp $
39  */
40
41 #include <stdio.h>
42 #include <signal.h>
43 #include <sys/stat.h>
44 #include <unistd.h>
45 #include <fcntl.h>
46 #include <locale.h>
47 #include <errno.h>
48
49 #include "shell.h"
50 #include "main.h"
51 #include "mail.h"
52 #include "options.h"
53 #include "output.h"
54 #include "parser.h"
55 #include "nodes.h"
56 #include "expand.h"
57 #include "eval.h"
58 #include "jobs.h"
59 #include "input.h"
60 #include "trap.h"
61 #include "var.h"
62 #include "show.h"
63 #include "memalloc.h"
64 #include "error.h"
65 #include "init.h"
66 #include "mystring.h"
67 #include "exec.h"
68 #include "cd.h"
69 #include "builtins.h"
70
71 int rootpid;
72 int rootshell;
73 struct jmploc main_handler;
74 int localeisutf8, initial_localeisutf8;
75
76 static void cmdloop(int);
77 static void read_profile(const char *);
78 static const char *find_dot_file(const char *);
79
80 /*
81  * Main routine.  We initialize things, parse the arguments, execute
82  * profiles if we're a login shell, and then call cmdloop to execute
83  * commands.  The setjmp call sets up the location to jump to when an
84  * exception occurs.  When an exception occurs the variable "state"
85  * is used to figure out how far we had gotten.
86  */
87
88 int
89 main(int argc, char *argv[])
90 {
91         struct stackmark smark, smark2;
92         volatile int state;
93         char *shinit;
94
95         setlocale(LC_ALL, "");
96         initcharset();
97         state = 0;
98         if (setjmp(main_handler.loc)) {
99                 switch (exception) {
100                 case EXEXEC:
101                         exitstatus = exerrno;
102                         break;
103
104                 case EXERROR:
105                         exitstatus = 2;
106                         break;
107
108                 default:
109                         break;
110                 }
111
112                 if (state == 0 || iflag == 0 || ! rootshell ||
113                     exception == EXEXIT)
114                         exitshell(exitstatus);
115                 reset();
116                 if (exception == EXINT)
117                         out2fmt_flush("\n");
118                 popstackmark(&smark);
119                 FORCEINTON;                             /* enable interrupts */
120                 if (state == 1)
121                         goto state1;
122                 else if (state == 2)
123                         goto state2;
124                 else if (state == 3)
125                         goto state3;
126                 else
127                         goto state4;
128         }
129         handler = &main_handler;
130 #ifdef DEBUG
131         opentrace();
132         trputs("Shell args:  ");  trargs(argv);
133 #endif
134         rootpid = getpid();
135         rootshell = 1;
136         init();
137         setstackmark(&smark);
138         setstackmark(&smark2);
139         procargs(argc, argv);
140         pwd_init(iflag);
141         if (iflag)
142                 chkmail(1);
143         if (argv[0] && argv[0][0] == '-') {
144                 state = 1;
145                 read_profile("/etc/profile");
146 state1:
147                 state = 2;
148                 if (privileged == 0)
149                         read_profile("${HOME-}/.profile");
150                 else
151                         read_profile("/etc/suid_profile");
152         }
153 state2:
154         state = 3;
155         if (!privileged && iflag) {
156                 if ((shinit = lookupvar("ENV")) != NULL && *shinit != '\0') {
157                         state = 3;
158                         read_profile(shinit);
159                 }
160         }
161 state3:
162         state = 4;
163         popstackmark(&smark2);
164         if (minusc) {
165                 evalstring(minusc, sflag ? 0 : EV_EXIT);
166         }
167         if (sflag || minusc == NULL) {
168 state4: /* XXX ??? - why isn't this before the "if" statement */
169                 cmdloop(1);
170         }
171         exitshell(exitstatus);
172         /*NOTREACHED*/
173         return 0;
174 }
175
176
177 /*
178  * Read and execute commands.  "Top" is nonzero for the top level command
179  * loop; it turns on prompting if the shell is interactive.
180  */
181
182 static void
183 cmdloop(int top)
184 {
185         union node *n;
186         struct stackmark smark;
187         int inter;
188         int numeof = 0;
189
190         TRACE(("cmdloop(%d) called\n", top));
191         setstackmark(&smark);
192         for (;;) {
193                 if (pendingsigs)
194                         dotrap();
195                 inter = 0;
196                 if (iflag && top) {
197                         inter++;
198                         showjobs(1, SHOWJOBS_DEFAULT);
199                         chkmail(0);
200                         flushout(&output);
201                 }
202                 n = parsecmd(inter);
203                 /* showtree(n); DEBUG */
204                 if (n == NEOF) {
205                         if (!top || numeof >= 50)
206                                 break;
207                         if (!stoppedjobs()) {
208                                 if (!Iflag)
209                                         break;
210                                 out2fmt_flush("\nUse \"exit\" to leave shell.\n");
211                         }
212                         numeof++;
213                 } else if (n != NULL && nflag == 0) {
214                         job_warning = (job_warning == 2) ? 1 : 0;
215                         numeof = 0;
216                         evaltree(n, 0);
217                 }
218                 popstackmark(&smark);
219                 setstackmark(&smark);
220                 if (evalskip != 0) {
221                         if (evalskip == SKIPFILE)
222                                 evalskip = 0;
223                         break;
224                 }
225         }
226         popstackmark(&smark);
227 }
228
229
230
231 /*
232  * Read /etc/profile or .profile.  Return on error.
233  */
234
235 static void
236 read_profile(const char *name)
237 {
238         int fd;
239         const char *expandedname;
240
241         expandedname = expandstr(__DECONST(char *, name));
242         if (expandedname == NULL)
243                 return;
244         INTOFF;
245         if ((fd = open(expandedname, O_RDONLY)) >= 0)
246                 setinputfd(fd, 1);
247         INTON;
248         if (fd < 0)
249                 return;
250         cmdloop(0);
251         popfile();
252 }
253
254
255
256 /*
257  * Read a file containing shell functions.
258  */
259
260 void
261 readcmdfile(const char *name)
262 {
263         int fd;
264
265         INTOFF;
266         if ((fd = open(name, O_RDONLY)) >= 0)
267                 setinputfd(fd, 1);
268         else
269                 error("cannot open %s: %s", name, strerror(errno));
270         INTON;
271         cmdloop(0);
272         popfile();
273 }
274
275
276
277 /*
278  * Take commands from a file.  To be compatible we should do a path
279  * search for the file, which is necessary to find sub-commands.
280  */
281
282
283 static const char *
284 find_dot_file(const char *basename)
285 {
286         char *fullname;
287         const char *path = pathval();
288         struct stat statb;
289
290         /* don't try this for absolute or relative paths */
291         if( strchr(basename, '/'))
292                 return basename;
293
294         while ((fullname = padvance(&path, basename)) != NULL) {
295                 if ((stat(fullname, &statb) == 0) && S_ISREG(statb.st_mode)) {
296                         /*
297                          * Don't bother freeing here, since it will
298                          * be freed by the caller.
299                          */
300                         return fullname;
301                 }
302                 stunalloc(fullname);
303         }
304         return basename;
305 }
306
307 int
308 dotcmd(int argc, char **argv)
309 {
310         const char *fullname;
311         char *filename;
312
313         if (argc < 2)
314                 error("missing filename");
315
316         exitstatus = 0;
317
318         /*
319          * Because we have historically not supported any options,
320          * only treat "--" specially.
321          */
322         filename = argc > 2 && strcmp(argv[1], "--") == 0 ? argv[2] : argv[1];
323
324         fullname = find_dot_file(filename);
325         setinputfile(fullname, 1);
326         commandname = fullname;
327         cmdloop(0);
328         popfile();
329         return exitstatus;
330 }
331
332
333 int
334 exitcmd(int argc, char **argv)
335 {
336         if (stoppedjobs())
337                 return 0;
338         if (argc > 1)
339                 exitshell(number(argv[1]));
340         else
341                 exitshell_savedstatus();
342 }