- update from initscripts-8.54

[projects/rc-scripts.git] / src / getkey.c

/*\r
 * Copyright (c) 1999-2003, 2006 Red Hat, Inc. All rights reserved.\r
 *\r
 * This software may be freely redistributed under the terms of the GNU\r
 * public license.\r
 *\r
 * You should have received a copy of the GNU General Public License\r
 * along with this program; if not, write to the Free Software\r
 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.\r
 *\r
 * getkey\r
 *\r
 * A very simple keygrabber.\r
 *\r
 */\r
#include <ctype.h>\r
#include <errno.h>\r
#include <signal.h>\r
#include <stdlib.h>\r
#include <stdio.h>\r
#include <string.h>\r
#include <termios.h>\r
#include <unistd.h>\r
#include <sys/poll.h>\r
#include "popt.h"\r
\r
static struct termios orig_tp;\r
\r
static void reset_term(int x) {\r
	tcsetattr(0,TCSANOW,&orig_tp);\r
	_exit(x);\r
}\r
\r
int main(int argc, char **argv) {\r
        static const char default_list[] = "";\r
\r
	const char *list;\r
	char *waitmessage = NULL;\r
	char *waitprint, *waitsprint;\r
	int waitseconds=0;\r
	int alarmlen=0;\r
	int ignore_control=0;\r
	struct termios tp;\r
	int r;\r
	struct pollfd ufds; /* only one, no need for an array... */\r
        poptContext context;\r
	struct poptOption options[] = {\r
            { "wait", 'c', POPT_ARG_INT, &waitseconds, 0, "Number of seconds to wait for keypress", NULL },\r
            { "message", 'm', POPT_ARG_STRING, &waitmessage, 0, "Message to print out while waiting for string\nNOTE: The message may have a \"%d\" in it, to hold the number of seconds left to wait.", NULL },\r
            { "ignore-control-chars", 'i', POPT_ARG_NONE, &ignore_control, 0, "Ignore Control-C and Control-D", NULL },\r
            POPT_AUTOHELP\r
            POPT_TABLEEND\r
        };\r
\r
	context = poptGetContext("getkey", argc, (const char **)argv, options,\r
                                POPT_CONTEXT_POSIXMEHARDER);\r
        poptSetOtherOptionHelp(context, "[keys]");\r
\r
        r = poptGetNextOpt(context);\r
        if (r < -1) {\r
            fprintf(stderr, "%s: %s\n", \r
                   poptBadOption(context, POPT_BADOPTION_NOALIAS),\r
                   poptStrerror(r));\r
\r
            return -1;\r
        }\r
        list = poptGetArg(context);\r
	if (list != NULL) {\r
	    char *p;\r
\r
	    p = strdup(list);\r
	    list = p;\r
	    while (*p != 0) {\r
		*p = toupper(*p);\r
		p++;\r
	    }\r
	} else\r
	    list = default_list;\r
	if (waitseconds) {\r
	    if (waitseconds < 0) {\r
		fprintf(stderr, "--wait: Invalid time %d seconds\n",\r
			waitseconds);\r
		return -1;\r
	    }\r
	    alarmlen = waitseconds;\r
	}\r
\r
	tcgetattr(0,&tp);\r
	orig_tp = tp;\r
	signal(SIGTERM,reset_term);\r
	if (alarmlen != 0) {\r
	    signal(SIGALRM,reset_term);\r
	    alarm(alarmlen);\r
	}\r
\r
	tp.c_iflag=0;\r
	tp.c_oflag &= ~OPOST;\r
	tp.c_lflag &= ~(ISIG | ICANON);\r
	tcsetattr(0,TCSANOW,&tp);\r
\r
	ufds.events = POLLIN;\r
	ufds.fd = 0;\r
\r
	if (waitmessage) {\r
	    waitprint = alloca (strlen(waitmessage)+15); /* long enough */\r
	    waitprint[0] = '\r';\r
	    waitsprint = waitprint + 1;\r
	}\r
\r
	while (1) {\r
	    if (waitmessage) {\r
		sprintf (waitsprint, waitmessage, waitseconds);\r
		write (1, waitprint, strlen(waitprint));\r
	    }\r
	    r = poll(&ufds, 1, alarmlen ? 1000 : -1);\r
	    if (r == 0) {\r
		/* we have waited a whole second with no keystroke... */\r
		waitseconds--;\r
	    }\r
	    if (r > 0) {\r
		char ch;\r
\r
		read(0, &ch, sizeof(ch));\r
		ch = toupper(ch);\r
		/* Die if we get a control-c or control-d */\r
                if (ignore_control == 0 && (ch == 3 || ch == 4))\r
		    reset_term(1);\r
		/* Don't let a null character be interpreted as a match\r
		   by strchr */\r
		if (ch != 0\r
		    && (strcmp(list, "") == 0 || strchr(list, ch) != NULL))\r
		    reset_term(0);\r
	    }\r
	}\r
}\r