getch.c revision 1.69
1/*	$NetBSD: getch.c,v 1.69 2018/09/27 14:07:55 roy Exp $	*/
2
3/*
4 * Copyright (c) 1981, 1993, 1994
5 *	The Regents of the University of California.  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 * 3. Neither the name of the University nor the names of its contributors
16 *    may be used to endorse or promote products derived from this software
17 *    without specific prior written permission.
18 *
19 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
20 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
21 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
23 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
24 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
25 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
26 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29 * SUCH DAMAGE.
30 */
31
32#include <sys/cdefs.h>
33#ifndef lint
34#if 0
35static char sccsid[] = "@(#)getch.c	8.2 (Berkeley) 5/4/94";
36#else
37__RCSID("$NetBSD: getch.c,v 1.69 2018/09/27 14:07:55 roy Exp $");
38#endif
39#endif					/* not lint */
40
41#include <errno.h>
42#include <string.h>
43#include <stdlib.h>
44#include <unistd.h>
45#include <stdio.h>
46#include "curses.h"
47#include "curses_private.h"
48#include "keymap.h"
49
50short	state;		/* state of the inkey function */
51
52static const struct tcdata tc[] = {
53	{TICODE_kSAV, KEY_SSAVE},
54	{TICODE_kSPD, KEY_SSUSPEND},
55	{TICODE_kUND, KEY_SUNDO},
56	{TICODE_kHLP, KEY_SHELP},
57	{TICODE_kHOM, KEY_SHOME},
58	{TICODE_kIC, KEY_SIC},
59	{TICODE_kLFT, KEY_SLEFT},
60	{TICODE_krdo, KEY_REDO},
61	{TICODE_khlp, KEY_HELP},
62	{TICODE_kmrk, KEY_MARK},
63	{TICODE_kmsg, KEY_MESSAGE},
64	{TICODE_kmov, KEY_MOVE},
65	{TICODE_knxt, KEY_NEXT},
66	{TICODE_kopn, KEY_OPEN},
67	{TICODE_kopt, KEY_OPTIONS},
68	{TICODE_kprv, KEY_PREVIOUS},
69	{TICODE_kprt, KEY_PRINT},
70	{TICODE_kMSG, KEY_SMESSAGE},
71	{TICODE_kMOV, KEY_SMOVE},
72	{TICODE_kNXT, KEY_SNEXT},
73	{TICODE_kOPT, KEY_SOPTIONS},
74	{TICODE_kPRV, KEY_SPREVIOUS},
75	{TICODE_kPRT, KEY_SPRINT},
76	{TICODE_kRDO, KEY_SREDO},
77	{TICODE_kRPL, KEY_SREPLACE},
78	{TICODE_kRIT, KEY_SRIGHT},
79	{TICODE_kRES, KEY_SRSUME},
80	{TICODE_kCAN, KEY_SCANCEL},
81	{TICODE_kref, KEY_REFERENCE},
82	{TICODE_krfr, KEY_REFRESH},
83	{TICODE_krpl, KEY_REPLACE},
84	{TICODE_krst, KEY_RESTART},
85	{TICODE_kres, KEY_RESUME},
86	{TICODE_ksav, KEY_SAVE},
87	{TICODE_kspd, KEY_SUSPEND},
88	{TICODE_kund, KEY_UNDO},
89	{TICODE_kBEG, KEY_SBEG},
90	{TICODE_kFND, KEY_SFIND},
91	{TICODE_kCMD, KEY_SCOMMAND},
92	{TICODE_kCPY, KEY_SCOPY},
93	{TICODE_kCRT, KEY_SCREATE},
94	{TICODE_kDC, KEY_SDC},
95	{TICODE_kDL, KEY_SDL},
96	{TICODE_kslt, KEY_SELECT},
97	{TICODE_kEND, KEY_SEND},
98	{TICODE_kEOL, KEY_SEOL},
99	{TICODE_kEXT, KEY_SEXIT},
100	{TICODE_kfnd, KEY_FIND},
101	{TICODE_kbeg, KEY_BEG},
102	{TICODE_kcan, KEY_CANCEL},
103	{TICODE_kclo, KEY_CLOSE},
104	{TICODE_kcmd, KEY_COMMAND},
105	{TICODE_kcpy, KEY_COPY},
106	{TICODE_kcrt, KEY_CREATE},
107	{TICODE_kend, KEY_END},
108	{TICODE_kent, KEY_ENTER},
109	{TICODE_kext, KEY_EXIT},
110	{TICODE_kf11, KEY_F(11)},
111	{TICODE_kf12, KEY_F(12)},
112	{TICODE_kf13, KEY_F(13)},
113	{TICODE_kf14, KEY_F(14)},
114	{TICODE_kf15, KEY_F(15)},
115	{TICODE_kf16, KEY_F(16)},
116	{TICODE_kf17, KEY_F(17)},
117	{TICODE_kf18, KEY_F(18)},
118	{TICODE_kf19, KEY_F(19)},
119	{TICODE_kf20, KEY_F(20)},
120	{TICODE_kf21, KEY_F(21)},
121	{TICODE_kf22, KEY_F(22)},
122	{TICODE_kf23, KEY_F(23)},
123	{TICODE_kf24, KEY_F(24)},
124	{TICODE_kf25, KEY_F(25)},
125	{TICODE_kf26, KEY_F(26)},
126	{TICODE_kf27, KEY_F(27)},
127	{TICODE_kf28, KEY_F(28)},
128	{TICODE_kf29, KEY_F(29)},
129	{TICODE_kf30, KEY_F(30)},
130	{TICODE_kf31, KEY_F(31)},
131	{TICODE_kf32, KEY_F(32)},
132	{TICODE_kf33, KEY_F(33)},
133	{TICODE_kf34, KEY_F(34)},
134	{TICODE_kf35, KEY_F(35)},
135	{TICODE_kf36, KEY_F(36)},
136	{TICODE_kf37, KEY_F(37)},
137	{TICODE_kf38, KEY_F(38)},
138	{TICODE_kf39, KEY_F(39)},
139	{TICODE_kf40, KEY_F(40)},
140	{TICODE_kf41, KEY_F(41)},
141	{TICODE_kf42, KEY_F(42)},
142	{TICODE_kf43, KEY_F(43)},
143	{TICODE_kf44, KEY_F(44)},
144	{TICODE_kf45, KEY_F(45)},
145	{TICODE_kf46, KEY_F(46)},
146	{TICODE_kf47, KEY_F(47)},
147	{TICODE_kf48, KEY_F(48)},
148	{TICODE_kf49, KEY_F(49)},
149	{TICODE_kf50, KEY_F(50)},
150	{TICODE_kf51, KEY_F(51)},
151	{TICODE_kf52, KEY_F(52)},
152	{TICODE_kf53, KEY_F(53)},
153	{TICODE_kf54, KEY_F(54)},
154	{TICODE_kf55, KEY_F(55)},
155	{TICODE_kf56, KEY_F(56)},
156	{TICODE_kf57, KEY_F(57)},
157	{TICODE_kf58, KEY_F(58)},
158	{TICODE_kf59, KEY_F(59)},
159	{TICODE_kf60, KEY_F(60)},
160	{TICODE_kf61, KEY_F(61)},
161	{TICODE_kf62, KEY_F(62)},
162	{TICODE_kf63, KEY_F(63)},
163	{TICODE_ka1, KEY_A1},
164	{TICODE_kb2, KEY_B2},
165	{TICODE_ka3, KEY_A3},
166	{TICODE_kc1, KEY_C1},
167	{TICODE_kc3, KEY_C3},
168	{TICODE_kmous, KEY_MOUSE},
169	{TICODE_kf0, KEY_F0},
170	{TICODE_kf1, KEY_F(1)},
171	{TICODE_kf2, KEY_F(2)},
172	{TICODE_kf3, KEY_F(3)},
173	{TICODE_kf4, KEY_F(4)},
174	{TICODE_kf5, KEY_F(5)},
175	{TICODE_kf6, KEY_F(6)},
176	{TICODE_kf7, KEY_F(7)},
177	{TICODE_kf8, KEY_F(8)},
178	{TICODE_kf9, KEY_F(9)},
179	{TICODE_kf10, KEY_F(10)},
180	{TICODE_kil1, KEY_IL},
181	{TICODE_ktbc, KEY_CATAB},
182	{TICODE_kcbt, KEY_BTAB},
183	{TICODE_kbs, KEY_BACKSPACE},
184	{TICODE_kclr, KEY_CLEAR},
185	{TICODE_kdch1, KEY_DC},
186	{TICODE_kcud1, KEY_DOWN},
187	{TICODE_kel, KEY_EOL},
188	{TICODE_kind, KEY_SF},
189	{TICODE_kll, KEY_LL},
190	{TICODE_khome, KEY_HOME},
191	{TICODE_kich1, KEY_IC},
192	{TICODE_kdl1, KEY_DL},
193	{TICODE_kcub1, KEY_LEFT},
194	{TICODE_krmir, KEY_EIC},
195	{TICODE_knp, KEY_NPAGE},
196	{TICODE_kpp, KEY_PPAGE},
197	{TICODE_kri, KEY_SR},
198	{TICODE_kcuf1, KEY_RIGHT},
199	{TICODE_ked, KEY_EOS},
200	{TICODE_khts, KEY_STAB},
201	{TICODE_kctab, KEY_CTAB},
202	{TICODE_kcuu1, KEY_UP}
203};
204/* Number of TC entries .... */
205static const int num_tcs = (sizeof(tc) / sizeof(struct tcdata));
206
207/* Key buffer */
208#define INBUF_SZ 16		/* size of key buffer - must be larger than
209				 * longest multi-key sequence */
210static wchar_t	inbuf[INBUF_SZ];
211static int	start, end, working; /* pointers for manipulating inbuf data */
212
213/* prototypes for private functions */
214static void add_key_sequence(SCREEN *screen, char *sequence, int key_type);
215static key_entry_t *add_new_key(keymap_t *current, char ch, int key_type,
216        int symbol);
217static void delete_key_sequence(keymap_t *current, int key_type);
218static void do_keyok(keymap_t *current, int key_type, bool set, bool flag,
219	int *retval);
220static keymap_t *new_keymap(void); /* create a new keymap */
221static key_entry_t *new_key(void); /* create a new key entry */
222static wchar_t		inkey(int to, int delay);
223
224/*
225 * Free the storage associated with the given keymap
226 */
227void
228_cursesi_free_keymap(keymap_t *map)
229{
230	int i;
231
232	  /* check for, and free, child keymaps */
233	for (i = 0; i < MAX_CHAR; i++) {
234		if (map->mapping[i] >= 0) {
235			if (map->key[map->mapping[i]]->type == KEYMAP_MULTI)
236				_cursesi_free_keymap(
237					map->key[map->mapping[i]]->value.next);
238		}
239	}
240
241	  /* now free any allocated keymap structs */
242	for (i = 0; i < map->count; i += KEYMAP_ALLOC_CHUNK) {
243		free(map->key[i]);
244	}
245
246	free(map->key);
247	free(map);
248}
249
250
251/*
252 * Add a new key entry to the keymap pointed to by current.  Entry
253 * contains the character to add to the keymap, type is the type of
254 * entry to add (either multikey or leaf) and symbol is the symbolic
255 * value for a leaf type entry.  The function returns a pointer to the
256 * new keymap entry.
257 */
258static key_entry_t *
259add_new_key(keymap_t *current, char chr, int key_type, int symbol)
260{
261	key_entry_t *the_key;
262        int i, ki;
263
264#ifdef DEBUG
265	__CTRACE(__CTRACE_MISC,
266	    "Adding character %s of type %d, symbol 0x%x\n",
267	    unctrl(chr), key_type, symbol);
268#endif
269	if (current->mapping[(unsigned char)chr] < 0) {
270		if (current->mapping[(unsigned char)chr] == MAPPING_UNUSED) {
271			  /* first time for this char */
272			current->mapping[(unsigned char)chr] =
273				current->count;	/* map new entry */
274			ki = current->count;
275
276			  /* make sure we have room in the key array first */
277			if ((current->count & (KEYMAP_ALLOC_CHUNK - 1)) == 0)
278			{
279				if ((current->key =
280				     realloc(current->key,
281					     ki * sizeof(key_entry_t *)
282					     + KEYMAP_ALLOC_CHUNK * sizeof(key_entry_t *))) == NULL) {
283					fprintf(stderr,
284					  "Could not malloc for key entry\n");
285					exit(1);
286				}
287
288				the_key = new_key();
289				for (i = 0; i < KEYMAP_ALLOC_CHUNK; i++) {
290					current->key[ki + i] = &the_key[i];
291				}
292			}
293                } else {
294			  /* the mapping was used but freed, reuse it */
295			ki = - current->mapping[(unsigned char) chr];
296			current->mapping[(unsigned char) chr] = ki;
297		}
298
299		current->count++;
300
301		  /* point at the current key array element to use */
302		the_key = current->key[ki];
303
304		the_key->type = key_type;
305
306		switch (key_type) {
307		  case KEYMAP_MULTI:
308			    /* need for next key */
309#ifdef DEBUG
310			  __CTRACE(__CTRACE_MISC, "Creating new keymap\n");
311#endif
312			  the_key->value.next = new_keymap();
313			  the_key->enable = TRUE;
314			  break;
315
316		  case KEYMAP_LEAF:
317				/* the associated symbol for the key */
318#ifdef DEBUG
319			  __CTRACE(__CTRACE_MISC, "Adding leaf key\n");
320#endif
321			  the_key->value.symbol = symbol;
322			  the_key->enable = TRUE;
323			  break;
324
325		  default:
326			  fprintf(stderr, "add_new_key: bad type passed\n");
327			  exit(1);
328		}
329	} else {
330		  /* the key is already known - just return the address. */
331#ifdef DEBUG
332		__CTRACE(__CTRACE_MISC, "Keymap already known\n");
333#endif
334		the_key = current->key[current->mapping[(unsigned char)chr]];
335	}
336
337        return the_key;
338}
339
340/*
341 * Delete the given key symbol from the key mappings for the screen.
342 *
343 */
344static void
345delete_key_sequence(keymap_t *current, int key_type)
346{
347	key_entry_t *key;
348	int i;
349
350	  /*
351	   * we need to iterate over all the keys as there may be
352	   * multiple instances of the leaf symbol.
353	   */
354	for (i = 0; i < MAX_CHAR; i++) {
355		if (current->mapping[i] < 0)
356			continue; /* no mapping for the key, next! */
357
358		key = current->key[current->mapping[i]];
359
360		if (key->type == KEYMAP_MULTI) {
361			  /* have not found the leaf, recurse down */
362			delete_key_sequence(key->value.next, key_type);
363			  /* if we deleted the last key in the map, free */
364			if (key->value.next->count == 0)
365				_cursesi_free_keymap(key->value.next);
366		} else if ((key->type == KEYMAP_LEAF)
367			   && (key->value.symbol == key_type)) {
368#ifdef DEBUG
369		__CTRACE(__CTRACE_INPUT, "delete_key_sequence: found keysym %d, deleting\n",
370		    key_type);
371#endif
372			key->enable = FALSE;
373		}
374	}
375}
376
377/*
378 * Add the sequence of characters given in sequence as the key mapping
379 * for the given key symbol.
380 */
381static void
382add_key_sequence(SCREEN *screen, char *sequence, int key_type)
383{
384	key_entry_t *tmp_key;
385	keymap_t *current;
386	int length, j, key_ent;
387
388#ifdef DEBUG
389	__CTRACE(__CTRACE_MISC, "add_key_sequence: add key sequence: %s(%s)\n",
390	    sequence, keyname(key_type));
391#endif /* DEBUG */
392	current = screen->base_keymap;	/* always start with
393					 * base keymap. */
394	length = (int)strlen(sequence);
395
396	/*
397	 * OK - we really should never get a zero length string here, either
398	 * the terminfo entry is there and it has a value or we are not called
399	 * at all.  Unfortunately, if someone assigns a terminfo string to the
400	 * ^@ value we get passed a null string which messes up our length.
401	 * So, if we get a null string then just insert a leaf value in
402	 * the 0th char position of the root keymap.  Note that we are
403	 * totally screwed if someone terminates a multichar sequence
404	 * with ^@... oh well.
405	 */
406	if (length == 0)
407		length = 1;
408
409	for (j = 0; j < length - 1; j++) {
410		  /* add the entry to the struct */
411		tmp_key = add_new_key(current, sequence[j], KEYMAP_MULTI, 0);
412
413		  /* index into the key array - it's
414		     clearer if we stash this */
415		key_ent = current->mapping[(unsigned char) sequence[j]];
416
417		current->key[key_ent] = tmp_key;
418
419		  /* next key uses this map... */
420		current = current->key[key_ent]->value.next;
421	}
422
423	/*
424	 * This is the last key in the sequence (it may have been the
425	 * only one but that does not matter) this means it is a leaf
426	 * key and should have a symbol associated with it.
427	 */
428	tmp_key = add_new_key(current, sequence[length - 1], KEYMAP_LEAF,
429			      key_type);
430	current->key[current->mapping[(int)sequence[length - 1]]] = tmp_key;
431}
432
433/*
434 * Init_getch - initialise all the pointers & structures needed to make
435 * getch work in keypad mode.
436 *
437 */
438void
439__init_getch(SCREEN *screen)
440{
441	char entry[1024], *p;
442	const char *s;
443	int     i;
444	size_t limit, l;
445#ifdef DEBUG
446	int k, length;
447#endif
448
449	/* init the inkey state variable */
450	state = INKEY_NORM;
451
452	/* init the base keymap */
453	screen->base_keymap = new_keymap();
454
455	/* key input buffer pointers */
456	start = end = working = 0;
457
458	/* now do the terminfo snarfing ... */
459
460	for (i = 0; i < num_tcs; i++) {
461		p = entry;
462		limit = 1023;
463		s = screen->term->strs[tc[i].code];
464		if (s == NULL)
465			continue;
466		l = strlen(s) + 1;
467		if (limit < l)
468			continue;
469		strlcpy(p, s, limit);
470		p += l;
471		limit -= l;
472#ifdef DEBUG
473			__CTRACE(__CTRACE_INIT,
474			    "Processing terminfo entry %d, sequence ",
475			    tc[i].code);
476			length = (int) strlen(entry);
477			for (k = 0; k <= length -1; k++)
478				__CTRACE(__CTRACE_INIT, "%s", unctrl(entry[k]));
479			__CTRACE(__CTRACE_INIT, "\n");
480#endif
481		add_key_sequence(screen, entry, tc[i].symbol);
482	}
483}
484
485
486/*
487 * new_keymap - allocates & initialises a new keymap structure.  This
488 * function returns a pointer to the new keymap.
489 *
490 */
491static keymap_t *
492new_keymap(void)
493{
494	int     i;
495	keymap_t *new_map;
496
497	if ((new_map = malloc(sizeof(keymap_t))) == NULL) {
498		perror("Inkey: Cannot allocate new keymap");
499		exit(2);
500	}
501
502	/* Initialise the new map */
503	new_map->count = 0;
504	for (i = 0; i < MAX_CHAR; i++) {
505		new_map->mapping[i] = MAPPING_UNUSED; /* no mapping for char */
506	}
507
508	/* key array will be allocated when first key is added */
509	new_map->key = NULL;
510
511	return new_map;
512}
513
514/*
515 * new_key - allocates & initialises a new key entry.  This function returns
516 * a pointer to the newly allocated key entry.
517 *
518 */
519static key_entry_t *
520new_key(void)
521{
522	key_entry_t *new_one;
523	int i;
524
525	new_one = malloc(KEYMAP_ALLOC_CHUNK * sizeof(key_entry_t));
526	if (new_one == NULL) {
527		perror("inkey: Cannot allocate new key entry chunk");
528		exit(2);
529	}
530
531	for (i = 0; i < KEYMAP_ALLOC_CHUNK; i++) {
532		new_one[i].type = 0;
533		new_one[i].value.next = NULL;
534	}
535
536	return new_one;
537}
538
539/*
540 * inkey - do the work to process keyboard input, check for multi-key
541 * sequences and return the appropriate symbol if we get a match.
542 *
543 */
544
545static wchar_t
546inkey(int to, int delay)
547{
548	wchar_t		 k;
549	int              c, mapping;
550	keymap_t	*current = _cursesi_screen->base_keymap;
551	FILE            *infd = _cursesi_screen->infd;
552
553	k = 0;		/* XXX gcc -Wuninitialized */
554
555#ifdef DEBUG
556	__CTRACE(__CTRACE_INPUT, "inkey (%d, %d)\n", to, delay);
557#endif
558	for (;;) {		/* loop until we get a complete key sequence */
559reread:
560		if (state == INKEY_NORM) {
561			if (delay && __timeout(delay) == ERR)
562				return ERR;
563			c = __fgetc_resize(infd);
564			if (c == ERR || c == KEY_RESIZE) {
565				clearerr(infd);
566				return c;
567			}
568
569			if (delay && (__notimeout() == ERR))
570				return ERR;
571
572			k = (wchar_t)c;
573#ifdef DEBUG
574			__CTRACE(__CTRACE_INPUT,
575			    "inkey (state normal) got '%s'\n", unctrl(k));
576#endif
577
578			working = start;
579			inbuf[working] = k;
580			INC_POINTER(working);
581			end = working;
582			state = INKEY_ASSEMBLING;	/* go to the assembling
583							 * state now */
584		} else if (state == INKEY_BACKOUT) {
585			k = inbuf[working];
586			INC_POINTER(working);
587			if (working == end) {	/* see if we have run
588						 * out of keys in the
589						 * backlog */
590
591				/* if we have then switch to assembling */
592				state = INKEY_ASSEMBLING;
593			}
594		} else if (state == INKEY_ASSEMBLING) {
595			/* assembling a key sequence */
596			if (delay) {
597				if (__timeout(to ? (ESCDELAY / 100) : delay)
598				    == ERR)
599					return ERR;
600			} else {
601				if (to && (__timeout(ESCDELAY / 100) == ERR))
602					return ERR;
603			}
604
605			c = __fgetc_resize(infd);
606			if (ferror(infd)) {
607				clearerr(infd);
608				return c;
609			}
610
611			if ((to || delay) && (__notimeout() == ERR))
612					return ERR;
613
614#ifdef DEBUG
615			__CTRACE(__CTRACE_INPUT,
616			    "inkey (state assembling) got '%s'\n", unctrl(k));
617#endif
618			if (feof(infd) || c == -1) {	/* inter-char timeout,
619							 * start backing out */
620				clearerr(infd);
621				if (start == end)
622					/* no chars in the buffer, restart */
623					goto reread;
624
625				k = inbuf[start];
626				state = INKEY_TIMEOUT;
627			} else {
628				k = (wchar_t) c;
629				inbuf[working] = k;
630				INC_POINTER(working);
631				end = working;
632			}
633		} else {
634			fprintf(stderr, "Inkey state screwed - exiting!!!");
635			exit(2);
636		}
637
638		  /*
639		   * Check key has no special meaning and we have not
640		   * timed out and the key has not been disabled
641		   */
642		mapping = current->mapping[k];
643		if (((state == INKEY_TIMEOUT) || (mapping < 0))
644			|| ((current->key[mapping]->type == KEYMAP_LEAF)
645			    && (current->key[mapping]->enable == FALSE))) {
646			/* return the first key we know about */
647			k = inbuf[start];
648
649			INC_POINTER(start);
650			working = start;
651
652			if (start == end) {	/* only one char processed */
653				state = INKEY_NORM;
654			} else {/* otherwise we must have more than one char
655				 * to backout */
656				state = INKEY_BACKOUT;
657			}
658			return k;
659		} else {	/* must be part of a multikey sequence */
660			/* check for completed key sequence */
661			if (current->key[current->mapping[k]]->type == KEYMAP_LEAF) {
662				start = working;	/* eat the key sequence
663							 * in inbuf */
664
665				/* check if inbuf empty now */
666				if (start == end) {
667					/* if it is go back to normal */
668					state = INKEY_NORM;
669				} else {
670					/* otherwise go to backout state */
671					state = INKEY_BACKOUT;
672				}
673
674				/* return the symbol */
675				return current->key[current->mapping[k]]->value.symbol;
676
677			} else {
678				/*
679				 * Step on to next part of the multi-key
680				 * sequence.
681				 */
682				current = current->key[current->mapping[k]]->value.next;
683			}
684		}
685	}
686}
687
688#ifndef _CURSES_USE_MACROS
689/*
690 * getch --
691 *	Read in a character from stdscr.
692 */
693int
694getch(void)
695{
696	return wgetch(stdscr);
697}
698
699/*
700 * mvgetch --
701 *      Read in a character from stdscr at the given location.
702 */
703int
704mvgetch(int y, int x)
705{
706	return mvwgetch(stdscr, y, x);
707}
708
709/*
710 * mvwgetch --
711 *      Read in a character from stdscr at the given location in the
712 *      given window.
713 */
714int
715mvwgetch(WINDOW *win, int y, int x)
716{
717	if (wmove(win, y, x) == ERR)
718		return ERR;
719
720	return wgetch(win);
721}
722
723#endif
724
725/*
726 * keyok --
727 *      Set the enable flag for a keysym, if the flag is false then
728 * getch will not return this keysym even if the matching key sequence
729 * is seen.
730 */
731int
732keyok(int key_type, bool flag)
733{
734	int result = ERR;
735
736	if (_cursesi_screen != NULL)
737		do_keyok(_cursesi_screen->base_keymap, key_type,
738		    true, flag, &result);
739	return result;
740}
741
742/*
743 * do_keyok --
744 *       Does the actual work for keyok, we need to recurse through the
745 * keymaps finding the passed key symbol.
746 */
747static void
748do_keyok(keymap_t *current, int key_type, bool set, bool flag, int *retval)
749{
750	key_entry_t *key;
751	int i;
752
753	  /*
754	   * we need to iterate over all the keys as there may be
755	   * multiple instances of the leaf symbol.
756	   */
757	for (i = 0; i < MAX_CHAR; i++) {
758		if (current->mapping[i] < 0)
759			continue; /* no mapping for the key, next! */
760
761		key = current->key[current->mapping[i]];
762
763		if (key->type == KEYMAP_MULTI)
764			do_keyok(key->value.next, key_type, set, flag, retval);
765		else if ((key->type == KEYMAP_LEAF)
766			 && (key->value.symbol == key_type)) {
767			if (set)
768				key->enable = flag;
769			*retval = OK; /* we found at least one instance, ok */
770		}
771	}
772}
773
774/*
775 * define_key --
776 *      Add a custom mapping of a key sequence to key symbol.
777 *
778 */
779int
780define_key(char *sequence, int symbol)
781{
782
783	if (symbol <= 0 || _cursesi_screen == NULL)
784		return ERR;
785
786	if (sequence == NULL) {
787#ifdef DEBUG
788		__CTRACE(__CTRACE_INPUT, "define_key: deleting keysym %d\n",
789		    symbol);
790#endif
791		delete_key_sequence(_cursesi_screen->base_keymap, symbol);
792	} else
793		add_key_sequence(_cursesi_screen, sequence, symbol);
794
795	return OK;
796}
797
798/*
799 * wgetch --
800 *	Read in a character from the window.
801 */
802int
803wgetch(WINDOW *win)
804{
805	int inp, weset;
806	int c;
807	FILE *infd = _cursesi_screen->infd;
808
809#ifdef DEBUG
810	__CTRACE(__CTRACE_INPUT, "wgetch: win(%p)\n", win);
811#endif
812	if (win == NULL)
813		return ERR;
814	if (!(win->flags & __SCROLLOK) && (win->flags & __FULLWIN)
815	    && win->curx == win->maxx - 1 && win->cury == win->maxy - 1
816	    && __echoit)
817		return ERR;
818
819	if (is_wintouched(win))
820		wrefresh(win);
821#ifdef DEBUG
822	__CTRACE(__CTRACE_INPUT, "wgetch: __echoit = %d, "
823	    "__rawmode = %d, __nl = %d, flags = %#.4x, delay = %d\n",
824	    __echoit, __rawmode, _cursesi_screen->nl, win->flags, win->delay);
825#endif
826	if (_cursesi_screen->resized) {
827		_cursesi_screen->resized = 0;
828		resizeterm(LINES, COLS);
829#ifdef DEBUG
830		__CTRACE(__CTRACE_INPUT, "wgetch returning KEY_RESIZE\n");
831#endif
832		return KEY_RESIZE;
833	}
834	if (_cursesi_screen->unget_pos) {
835#ifdef DEBUG
836		__CTRACE(__CTRACE_INPUT, "wgetch returning char at %d\n",
837		    _cursesi_screen->unget_pos);
838#endif
839		_cursesi_screen->unget_pos--;
840		c = _cursesi_screen->unget_list[_cursesi_screen->unget_pos];
841		if (__echoit)
842			waddch(win, (chtype) c);
843		return c;
844	}
845	if (__echoit && !__rawmode) {
846		cbreak();
847		weset = 1;
848	} else
849		weset = 0;
850
851	__save_termios();
852
853	if (win->flags & __KEYPAD) {
854		switch (win->delay)
855		{
856		case -1:
857			inp = inkey (win->flags & __NOTIMEOUT ? 0 : 1, 0);
858			break;
859		case 0:
860			if (__nodelay() == ERR)
861				return ERR;
862			inp = inkey(0, 0);
863			break;
864		default:
865			inp = inkey(win->flags & __NOTIMEOUT ? 0 : 1, win->delay);
866			break;
867		}
868	} else {
869		switch (win->delay)
870		{
871		case -1:
872			if (__delay() == ERR)
873				return ERR;
874			break;
875		case 0:
876			if (__nodelay() == ERR)
877				return ERR;
878			break;
879		default:
880			if (__timeout(win->delay) == ERR)
881				return ERR;
882			break;
883		}
884
885		inp = __fgetc_resize(infd);
886		if (inp == ERR || inp == KEY_RESIZE) {
887			clearerr(infd);
888			__restore_termios();
889			return inp;
890		}
891	}
892#ifdef DEBUG
893	if (inp > 255)
894		  /* we have a key symbol - treat it differently */
895		  /* XXXX perhaps __unctrl should be expanded to include
896		   * XXXX the keysyms in the table....
897		   */
898		__CTRACE(__CTRACE_INPUT, "wgetch assembled keysym 0x%x\n", inp);
899	else
900		__CTRACE(__CTRACE_INPUT, "wgetch got '%s'\n", unctrl(inp));
901#endif
902	if (win->delay > -1) {
903		if (__delay() == ERR)
904			return ERR;
905	}
906
907	__restore_termios();
908
909	if ((__echoit) && (inp < KEY_MIN))
910		waddch(win, (chtype) inp);
911
912	if (weset)
913		nocbreak();
914
915	if (_cursesi_screen->nl && inp == 13)
916		inp = 10;
917
918	return ((inp < 0) || (inp == ERR) ? ERR : inp);
919}
920
921/*
922 * ungetch --
923 *     Put the character back into the input queue.
924 */
925int
926ungetch(int c)
927{
928	return __unget((wint_t)c);
929}
930
931/*
932 * __unget --
933 *    Do the work for ungetch() and unget_wch();
934 */
935int
936__unget(wint_t c)
937{
938	wchar_t	*p;
939	int	len;
940
941#ifdef DEBUG
942	__CTRACE(__CTRACE_INPUT, "__unget(%x)\n", c);
943#endif
944	if (_cursesi_screen == NULL)
945		return ERR;
946	if (_cursesi_screen->unget_pos >= _cursesi_screen->unget_len) {
947		len = _cursesi_screen->unget_len + 32;
948		if ((p = realloc(_cursesi_screen->unget_list,
949		    sizeof(wchar_t) * len)) == NULL) {
950			/* Can't realloc(), so just lose the oldest entry */
951			memmove(_cursesi_screen->unget_list,
952			    _cursesi_screen->unget_list + sizeof(wchar_t),
953			    _cursesi_screen->unget_len - 1);
954			_cursesi_screen->unget_list[_cursesi_screen->unget_len
955			    - 1] = c;
956			_cursesi_screen->unget_pos =
957			    _cursesi_screen->unget_len;
958			return OK;
959		} else {
960			_cursesi_screen->unget_pos =
961			    _cursesi_screen->unget_len;
962			_cursesi_screen->unget_len = len;
963			_cursesi_screen->unget_list = p;
964		}
965	}
966	_cursesi_screen->unget_list[_cursesi_screen->unget_pos] = c;
967	_cursesi_screen->unget_pos++;
968	return OK;
969}
970
971int
972has_key(int key_type)
973{
974	int result = ERR;
975
976	if (_cursesi_screen != NULL)
977		do_keyok(_cursesi_screen->base_keymap, key_type,
978		    false, false, &result);
979	return result;
980}
981
982/*
983 * set_escdelay --
984 *   Sets the escape delay for the current screen.
985 */
986int
987set_escdelay(int escdelay)
988{
989
990	if (_cursesi_screen == NULL)
991		return ERR;
992	_cursesi_screen->ESCDELAY = escdelay;
993	ESCDELAY = escdelay;
994	return OK;
995}
996
997/*
998 * __fgetc_resize --
999 *    Any call to fgetc(3) should use this function instead
1000 *    and test for the return value of KEY_RESIZE as well as ERR.
1001 */
1002int
1003__fgetc_resize(FILE *infd)
1004{
1005	int c;
1006
1007	c = fgetc(infd);
1008	if (c != EOF)
1009		return c;
1010
1011	if (!ferror(infd) || errno != EINTR || !_cursesi_screen->resized)
1012		return ERR;
1013#ifdef DEBUG
1014	__CTRACE(__CTRACE_INPUT, "__fgetc_resize returning KEY_RESIZE\n");
1015#endif
1016	_cursesi_screen->resized = 0;
1017	resizeterm(LINES, COLS);
1018	return KEY_RESIZE;
1019}
1020