Home | History | Annotate | Line # | Download | only in efiboot
prompt.c revision 1.2.2.4 
      1 /*	$NetBSD: prompt.c,v 1.2.2.4 2018/11/26 01:52:52 pgoyette Exp $	*/
      2 
      3 /*
      4  * Copyright (c) 1996, 1997
      5  * 	Matthias Drochner.  All rights reserved.
      6  * Copyright (c) 1996, 1997
      7  * 	Perry E. Metzger.  All rights reserved.
      8  * Copyright (c) 1997
      9  *	Jason R. Thorpe.  All rights reserved
     10  *
     11  * Redistribution and use in source and binary forms, with or without
     12  * modification, are permitted provided that the following conditions
     13  * are met:
     14  * 1. Redistributions of source code must retain the above copyright
     15  *    notice, this list of conditions and the following disclaimer.
     16  * 2. Redistributions in binary form must reproduce the above copyright
     17  *    notice, this list of conditions and the following disclaimer in the
     18  *    documentation and/or other materials provided with the distribution.
     19  * 3. All advertising materials mentioning features or use of this software
     20  *    must display the following acknowledgements:
     21  *	This product includes software developed for the NetBSD Project
     22  *	by Matthias Drochner.
     23  *	This product includes software developed for the NetBSD Project
     24  *	by Perry E. Metzger.
     25  * 4. The names of the authors may not be used to endorse or promote products
     26  *    derived from this software without specific prior written permission.
     27  *
     28  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
     29  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
     30  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
     31  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
     32  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
     33  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
     34  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
     35  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
     36  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
     37  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
     38  */
     39 
     40 #include "efiboot.h"
     41 
     42 #include <lib/libsa/net.h>
     43 
     44 #define	POLL_FREQ	10
     45 
     46 char *
     47 gettrailer(char *arg)
     48 {
     49 	char *options;
     50 
     51 	for (options = arg; *options; options++) {
     52 		switch (*options) {
     53 		case ' ':
     54 		case '\t':
     55 			*options++ = '\0';
     56 			break;
     57 		default:
     58 			continue;
     59 		}
     60 		break;
     61 	}
     62 	if (*options == '\0')
     63 		return options;
     64 
     65 	/* trim leading blanks/tabs */
     66 	while (*options == ' ' || *options == '\t')
     67 		options++;
     68 
     69 	return options;
     70 }
     71 
     72 char
     73 awaitkey(int timeout, int tell)
     74 {
     75 	int i = timeout * POLL_FREQ;
     76 	int last_secs = -1, secs;
     77 	char c = 0;
     78 
     79 	for (;;) {
     80 		if (tell) {
     81 			char buf[32];
     82 			int len;
     83 
     84 			secs = (i + POLL_FREQ - 1) / POLL_FREQ;
     85 			if (secs != last_secs) {
     86 				len = snprintf(buf, sizeof(buf), "%d seconds. ", (i + POLL_FREQ - 1) / POLL_FREQ);
     87 				if (len > 0 && len < sizeof(buf)) {
     88 					char *p = buf;
     89 					printf("%s", buf);
     90 					while (*p)
     91 						*p++ = '\b';
     92 					printf("%s", buf);
     93 				}
     94 				last_secs = secs;
     95 			}
     96 		}
     97 		if (ischar()) {
     98 			c = getchar();
     99 			if (c == 0)
    100 				c = -1;
    101 			goto out;
    102 		}
    103 		if (--i > 0) {
    104 			efi_delay(1000000 / POLL_FREQ);
    105 		} else {
    106 			break;
    107 		}
    108 	}
    109 
    110 out:
    111 	if (tell)
    112 		printf("0 seconds.     \n");
    113 
    114 	return c;
    115 }
    116 
    117 void
    118 docommand(char *arg)
    119 {
    120 	char *options;
    121 	int i;
    122 
    123 	options = gettrailer(arg);
    124 
    125 	for (i = 0; commands[i].c_name != NULL; i++) {
    126 		if (strcmp(arg, commands[i].c_name) == 0) {
    127 			(*commands[i].c_fn)(options);
    128 			return;
    129 		}
    130 	}
    131 
    132 	printf("unknown command\n");
    133 	command_help(NULL);
    134 }
    135 
    136 __dead void
    137 bootprompt(void)
    138 {
    139 	char input[80];
    140 
    141 	for (;;) {
    142 		char *c = input;
    143 
    144 		input[0] = '\0';
    145 		printf("> ");
    146 		kgets(input, sizeof(input));
    147 
    148 		/*
    149 		 * Skip leading whitespace.
    150 		 */
    151 		while (*c == ' ')
    152 			c++;
    153 		if (*c)
    154 			docommand(c);
    155 	}
    156 }
    157