1/*	$NetBSD: interp.c,v 1.5 2012/12/27 20:21:51 martin Exp $	*/
2
3/*-
4 * Copyright (c) 1998 Michael Smith <msmith@freebsd.org>
5 * 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 *
16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
17 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
20 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
21 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
22 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
23 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
24 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
25 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
26 * SUCH DAMAGE.
27 */
28
29#include <sys/cdefs.h>
30/* __FBSDID("$FreeBSD: src/sys/boot/common/interp.c,v 1.29 2003/08/25 23:30:41 obrien Exp $"); */
31
32/*
33 * Simple commandline interpreter, toplevel and misc.
34 *
35 * XXX may be obsoleted by BootFORTH or some other, better, interpreter.
36 */
37
38#include <lib/libsa/stand.h>
39#include <lib/libsa/loadfile.h>
40#include <lib/libkern/libkern.h>
41
42#include "bootstrap.h"
43
44#ifdef BOOT_FORTH
45#include "ficl.h"
46#define	RETURN(x)	stackPushINT(bf_vm->pStack,!x); return(x)
47
48extern FICL_VM *bf_vm;
49#else
50#define	RETURN(x)	return(x)
51#endif
52
53#define	MAXARGS	20			/* maximum number of arguments allowed */
54
55static void	prompt(void);
56
57#ifndef BOOT_FORTH
58static int	perform(int argc, char *argv[]);
59
60/*
61 * Perform the command
62 */
63int
64perform(int argc, char *argv[])
65{
66    int				result;
67    struct bootblk_command	*cmdp;
68    bootblk_cmd_t		*cmd;
69
70    int i;
71
72    if (argc < 1)
73	return(CMD_OK);
74
75    /* set return defaults; a successful command will override these */
76    command_seterr("no error message");
77    cmd = NULL;
78    result = CMD_ERROR;
79
80    /* search the command set for the command */
81    for (i = 0, cmdp = commands; (cmdp->c_name != NULL) && (cmdp->c_desc != NULL ); i++, cmdp = commands + i) {
82	if ((cmdp->c_name != NULL) && !strcmp(argv[0], cmdp->c_name))
83	    cmd = cmdp->c_fn;
84    }
85    if (cmd != NULL) {
86	result = (cmd)(argc, argv);
87    } else {
88	command_seterr("unknown command");
89    }
90    RETURN(result);
91}
92#endif	/* ! BOOT_FORTH */
93
94/*
95 * Interactive mode
96 */
97void
98interact(void)
99{
100    char	input[256];			/* big enough? */
101#ifndef BOOT_FORTH
102    int		argc;
103    char	**argv;
104#endif
105
106#ifdef BOOT_FORTH
107    bf_init();
108#endif
109
110    /*
111     * Read our default configuration
112     */
113    if(include("/boot/loader.rc")!=CMD_OK)
114	include("/boot/boot.conf");
115    printf("\n");
116
117    /*
118     * XXX: Before interacting, we might want to autoboot.
119     */
120
121
122    /*
123     * Not autobooting, go manual
124     */
125    printf("\nType '?' for a list of commands, 'help' for more detailed help.\n");
126    if (getenv("prompt") == NULL)
127	setenv("prompt", "${interpret}", 1);
128    if (getenv("interpret") == NULL)
129        setenv("interpret", "OK", 1);
130
131
132    for (;;) {
133	input[0] = '\0';
134	prompt();
135	ngets(input, sizeof(input));
136#ifdef BOOT_FORTH
137	bf_vm->sourceID.i = 0;
138	bf_run(input);
139#else
140	if (!parse(&argc, &argv, input)) {
141	    if (perform(argc, argv))
142		printf("%s: %s\n", argv[0], command_geterr());
143	    free(argv);
144	} else {
145	    printf("parse error\n");
146	}
147#endif
148    }
149}
150
151/*
152 * Read commands from a file, then execute them.
153 *
154 * We store the commands in memory and close the source file so that the media
155 * holding it can safely go away while we are executing.
156 *
157 * Commands may be prefixed with '@' (so they aren't displayed) or '-' (so
158 * that the script won't stop if they fail).
159 */
160
161int
162command_include(int argc, char *argv[])
163{
164    int		i;
165    int		res;
166    char	**argvbuf;
167
168    /*
169     * Since argv is static, we need to save it here.
170     */
171    argvbuf = (char**) calloc((u_int)argc, sizeof(char*));
172    for (i = 0; i < argc; i++)
173	argvbuf[i] = strdup(argv[i]);
174
175    res=CMD_OK;
176    for (i = 1; (i < argc) && (res == CMD_OK); i++)
177	res = include(argvbuf[i]);
178
179    for (i = 0; i < argc; i++)
180	free(argvbuf[i]);
181    free(argvbuf);
182
183    return(res);
184}
185
186struct includeline
187{
188    char		*text;
189    int			flags;
190    int			line;
191#define SL_QUIET	(1<<0)
192#define SL_IGNOREERR	(1<<1)
193    struct includeline	*next;
194};
195
196int
197include(const char *filename)
198{
199    struct includeline	*script, *se, *sp;
200    char		input[256];			/* big enough? */
201#ifdef BOOT_FORTH
202    int			res;
203    char		*cp;
204    int			prevsrcid, fd, line;
205#else
206    int			argc,res;
207    char		**argv, *cp;
208    int			fd, flags, line;
209#endif
210
211    if (((fd = open(filename, O_RDONLY)) == -1)) {
212	command_seterr("can't open '%s': %s\n", filename, strerror(errno));
213	return(CMD_ERROR);
214    }
215
216    /*
217     * Read the script into memory.
218     */
219    script = se = NULL;
220    line = 0;
221
222    while (fgetstr(input, sizeof(input), fd) >= 0) {
223	line++;
224#ifdef BOOT_FORTH
225	cp = input;
226#else
227	flags = 0;
228	/* Discard comments */
229	if (strncmp(input+strspn(input, " "), "\\ ", 2) == 0)
230	    continue;
231	cp = input;
232	/* Echo? */
233	if (input[0] == '@') {
234	    cp++;
235	    flags |= SL_QUIET;
236	}
237	/* Error OK? */
238	if (input[0] == '-') {
239	    cp++;
240	    flags |= SL_IGNOREERR;
241	}
242#endif
243	/* Allocate script line structure and copy line, flags */
244	sp = alloc(sizeof(struct includeline) + strlen(cp) + 1);
245	sp->text = (char *)sp + sizeof(struct includeline);
246	strcpy(sp->text, cp);
247#ifndef BOOT_FORTH
248	sp->flags = flags;
249#endif
250	sp->line = line;
251	sp->next = NULL;
252
253	if (script == NULL) {
254	    script = sp;
255	} else {
256	    se->next = sp;
257	}
258	se = sp;
259    }
260    close(fd);
261
262    /*
263     * Execute the script
264     */
265#ifndef BOOT_FORTH
266    argv = NULL;
267#else
268    prevsrcid = bf_vm->sourceID.i;
269    bf_vm->sourceID.i = fd;
270#endif
271    res = CMD_OK;
272    for (sp = script; sp != NULL; sp = sp->next) {
273
274#ifdef BOOT_FORTH
275	res = bf_run(sp->text);
276	if (res != VM_OUTOFTEXT) {
277		command_seterr("Error while including %s, in the line:\n%s",
278		    filename, sp->text);
279		res = CMD_ERROR;
280		break;
281	} else
282		res = CMD_OK;
283#else
284	/* print if not being quiet */
285	if (!(sp->flags & SL_QUIET)) {
286	    prompt();
287	    printf("%s\n", sp->text);
288	}
289
290	/* Parse the command */
291	if (!parse(&argc, &argv, sp->text)) {
292	    if ((argc > 0) && (perform(argc, argv) != 0)) {
293		/* normal command */
294		printf("%s: %s\n", argv[0], command_geterr());
295		if (!(sp->flags & SL_IGNOREERR)) {
296		    res=CMD_ERROR;
297		    break;
298		}
299	    }
300	    free(argv);
301	    argv = NULL;
302	} else {
303	    printf("%s line %d: parse error\n", filename, sp->line);
304	    res=CMD_ERROR;
305	    break;
306	}
307#endif
308    }
309#ifndef BOOT_FORTH
310    if (argv != NULL)
311	free(argv);
312#else
313    bf_vm->sourceID.i = prevsrcid;
314#endif
315    while(script != NULL) {
316	se = script;
317	script = script->next;
318	free(se);
319    }
320    return(res);
321}
322
323/*
324 * Emit the current prompt; use the same syntax as the parser
325 * for embedding environment variables.
326 */
327static void
328prompt(void)
329{
330    char	*pr, *p, *cp, *ev;
331
332    if ((cp = getenv("prompt")) == NULL)
333	cp = ">";
334    pr = p = strdup(cp);
335
336    while (*p != 0) {
337	if ((*p == '$') && (*(p+1) == '{')) {
338	    for (cp = p + 2; (*cp != 0) && (*cp != '}'); cp++)
339		;
340	    *cp = 0;
341	    ev = getenv(p + 2);
342
343	    if (ev != NULL)
344		printf("%s", ev);
345	    p = cp + 1;
346	    continue;
347	}
348	putchar(*p++);
349    }
350    putchar(' ');
351    free(pr);
352}
353