add support for a timeout argument

1 files changed, 13 insertions, 2 deletions

diff --git a/src/getkey.c b/src/getkey.c
index b17558a3..8f1cc1f6 100644
--- a/src/getkey.c
+++ b/src/getkey.c
@@ -1,6 +1,7 @@
 
 #include <ctype.h>
 #include <signal.h>
+#include <stdlib.h>
 #include <string.h>
 #include <termios.h>
 #include <unistd.h>
@@ -16,14 +17,24 @@ void reset_term(int x) {
 
 int main(int argc, char **argv) {
 	char foo[2];
+	char *list;
+	int alarmlen=0;
 	int tp_if,tp_of,tp_lf;
 	int x;
 	
 	if (argc>1) {
-		for (x=0;argv[1][x];x++) argv[1][x]=toupper(argv[1][x]);
+		if (argv[1][0]=='-' && isdigit(argv[1][1])) {
+			alarmlen = atoi(argv[1]+1);
+			list = argv[2];
+		} else {
+			list = argv[1];
+		}
+		for (x=0;list[x];x++) list[x]=toupper(list[x]);
 	}
 	foo[0]=foo[1]='\0';
 	signal(SIGTERM,reset_term);
+	alarm(alarmlen);
+	signal(SIGALRM,reset_term);
 	tcgetattr(0,&tp);
 	tp_if=tp.c_iflag;
 	tp_of=tp.c_oflag;
@@ -40,7 +51,7 @@ int main(int argc, char **argv) {
 		foo[0]=toupper(foo[0]);
 		/* Die if we get a control-c or control-d */
 		if (foo[0]==3 || foo[0]==4) reset_term(1);
-		if ((!argv[1]) || strstr(argv[1],foo)) {
+		if ((!list) || strstr(list,foo)) {
 			reset_term(0);
 		}
 	}