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 } 343