xref: /vim-8.2.3635/src/getchar.c (revision 0d5a12ea)
1 /* vi:set ts=8 sts=4 sw=4 noet:
2  *
3  * VIM - Vi IMproved	by Bram Moolenaar
4  *
5  * Do ":help uganda"  in Vim to read copying and usage conditions.
6  * Do ":help credits" in Vim to see a list of people who contributed.
7  * See README.txt for an overview of the Vim source code.
8  */
9 
10 /*
11  * getchar.c
12  *
13  * functions related with getting a character from the user/mapping/redo/...
14  *
15  * manipulations with redo buffer and stuff buffer
16  * mappings and abbreviations
17  */
18 
19 #include "vim.h"
20 
21 /*
22  * These buffers are used for storing:
23  * - stuffed characters: A command that is translated into another command.
24  * - redo characters: will redo the last change.
25  * - recorded characters: for the "q" command.
26  *
27  * The bytes are stored like in the typeahead buffer:
28  * - K_SPECIAL introduces a special key (two more bytes follow).  A literal
29  *   K_SPECIAL is stored as K_SPECIAL KS_SPECIAL KE_FILLER.
30  * - CSI introduces a GUI termcap code (also when gui.in_use is FALSE,
31  *   otherwise switching the GUI on would make mappings invalid).
32  *   A literal CSI is stored as CSI KS_EXTRA KE_CSI.
33  * These translations are also done on multi-byte characters!
34  *
35  * Escaping CSI bytes is done by the system-specific input functions, called
36  * by ui_inchar().
37  * Escaping K_SPECIAL is done by inchar().
38  * Un-escaping is done by vgetc().
39  */
40 
41 #define MINIMAL_SIZE 20			// minimal size for b_str
42 
43 static buffheader_T redobuff = {{NULL, {NUL}}, NULL, 0, 0};
44 static buffheader_T old_redobuff = {{NULL, {NUL}}, NULL, 0, 0};
45 static buffheader_T recordbuff = {{NULL, {NUL}}, NULL, 0, 0};
46 
47 static int typeahead_char = 0;		// typeahead char that's not flushed
48 
49 /*
50  * when block_redo is TRUE redo buffer will not be changed
51  * used by edit() to repeat insertions and 'V' command for redoing
52  */
53 static int	block_redo = FALSE;
54 
55 static int	KeyNoremap = 0;	    // remapping flags
56 
57 /*
58  * Variables used by vgetorpeek() and flush_buffers().
59  *
60  * typebuf.tb_buf[] contains all characters that are not consumed yet.
61  * typebuf.tb_buf[typebuf.tb_off] is the first valid character.
62  * typebuf.tb_buf[typebuf.tb_off + typebuf.tb_len - 1] is the last valid char.
63  * typebuf.tb_buf[typebuf.tb_off + typebuf.tb_len] must be NUL.
64  * The head of the buffer may contain the result of mappings, abbreviations
65  * and @a commands.  The length of this part is typebuf.tb_maplen.
66  * typebuf.tb_silent is the part where <silent> applies.
67  * After the head are characters that come from the terminal.
68  * typebuf.tb_no_abbr_cnt is the number of characters in typebuf.tb_buf that
69  * should not be considered for abbreviations.
70  * Some parts of typebuf.tb_buf may not be mapped. These parts are remembered
71  * in typebuf.tb_noremap[], which is the same length as typebuf.tb_buf and
72  * contains RM_NONE for the characters that are not to be remapped.
73  * typebuf.tb_noremap[typebuf.tb_off] is the first valid flag.
74  * (typebuf has been put in globals.h, because check_termcode() needs it).
75  */
76 #define RM_YES		0	// tb_noremap: remap
77 #define RM_NONE		1	// tb_noremap: don't remap
78 #define RM_SCRIPT	2	// tb_noremap: remap local script mappings
79 #define RM_ABBR		4	// tb_noremap: don't remap, do abbrev.
80 
81 // typebuf.tb_buf has three parts: room in front (for result of mappings), the
82 // middle for typeahead and room for new characters (which needs to be 3 *
83 // MAXMAPLEN) for the Amiga).
84 #define TYPELEN_INIT	(5 * (MAXMAPLEN + 3))
85 static char_u	typebuf_init[TYPELEN_INIT];	// initial typebuf.tb_buf
86 static char_u	noremapbuf_init[TYPELEN_INIT];	// initial typebuf.tb_noremap
87 
88 static int	last_recorded_len = 0;	// number of last recorded chars
89 
90 static int	read_readbuf(buffheader_T *buf, int advance);
91 static void	init_typebuf(void);
92 static void	may_sync_undo(void);
93 static void	free_typebuf(void);
94 static void	closescript(void);
95 static void	updatescript(int c);
96 static int	vgetorpeek(int);
97 static int	inchar(char_u *buf, int maxlen, long wait_time);
98 
99 /*
100  * Free and clear a buffer.
101  */
102     static void
free_buff(buffheader_T * buf)103 free_buff(buffheader_T *buf)
104 {
105     buffblock_T	*p, *np;
106 
107     for (p = buf->bh_first.b_next; p != NULL; p = np)
108     {
109 	np = p->b_next;
110 	vim_free(p);
111     }
112     buf->bh_first.b_next = NULL;
113 }
114 
115 /*
116  * Return the contents of a buffer as a single string.
117  * K_SPECIAL and CSI in the returned string are escaped.
118  */
119     static char_u *
get_buffcont(buffheader_T * buffer,int dozero)120 get_buffcont(
121     buffheader_T	*buffer,
122     int			dozero)	    // count == zero is not an error
123 {
124     long_u	    count = 0;
125     char_u	    *p = NULL;
126     char_u	    *p2;
127     char_u	    *str;
128     buffblock_T *bp;
129 
130     // compute the total length of the string
131     for (bp = buffer->bh_first.b_next; bp != NULL; bp = bp->b_next)
132 	count += (long_u)STRLEN(bp->b_str);
133 
134     if ((count || dozero) && (p = alloc(count + 1)) != NULL)
135     {
136 	p2 = p;
137 	for (bp = buffer->bh_first.b_next; bp != NULL; bp = bp->b_next)
138 	    for (str = bp->b_str; *str; )
139 		*p2++ = *str++;
140 	*p2 = NUL;
141     }
142     return (p);
143 }
144 
145 /*
146  * Return the contents of the record buffer as a single string
147  * and clear the record buffer.
148  * K_SPECIAL and CSI in the returned string are escaped.
149  */
150     char_u *
get_recorded(void)151 get_recorded(void)
152 {
153     char_u	*p;
154     size_t	len;
155 
156     p = get_buffcont(&recordbuff, TRUE);
157     free_buff(&recordbuff);
158 
159     /*
160      * Remove the characters that were added the last time, these must be the
161      * (possibly mapped) characters that stopped the recording.
162      */
163     len = STRLEN(p);
164     if ((int)len >= last_recorded_len)
165     {
166 	len -= last_recorded_len;
167 	p[len] = NUL;
168     }
169 
170     /*
171      * When stopping recording from Insert mode with CTRL-O q, also remove the
172      * CTRL-O.
173      */
174     if (len > 0 && restart_edit != 0 && p[len - 1] == Ctrl_O)
175 	p[len - 1] = NUL;
176 
177     return (p);
178 }
179 
180 /*
181  * Return the contents of the redo buffer as a single string.
182  * K_SPECIAL and CSI in the returned string are escaped.
183  */
184     char_u *
get_inserted(void)185 get_inserted(void)
186 {
187     return get_buffcont(&redobuff, FALSE);
188 }
189 
190 /*
191  * Add string "s" after the current block of buffer "buf".
192  * K_SPECIAL and CSI should have been escaped already.
193  */
194     static void
add_buff(buffheader_T * buf,char_u * s,long slen)195 add_buff(
196     buffheader_T	*buf,
197     char_u		*s,
198     long		slen)	// length of "s" or -1
199 {
200     buffblock_T *p;
201     long_u	    len;
202 
203     if (slen < 0)
204 	slen = (long)STRLEN(s);
205     if (slen == 0)				// don't add empty strings
206 	return;
207 
208     if (buf->bh_first.b_next == NULL)	// first add to list
209     {
210 	buf->bh_space = 0;
211 	buf->bh_curr = &(buf->bh_first);
212     }
213     else if (buf->bh_curr == NULL)	// buffer has already been read
214     {
215 	iemsg(_("E222: Add to read buffer"));
216 	return;
217     }
218     else if (buf->bh_index != 0)
219 	mch_memmove(buf->bh_first.b_next->b_str,
220 		    buf->bh_first.b_next->b_str + buf->bh_index,
221 		    STRLEN(buf->bh_first.b_next->b_str + buf->bh_index) + 1);
222     buf->bh_index = 0;
223 
224     if (buf->bh_space >= (int)slen)
225     {
226 	len = (long_u)STRLEN(buf->bh_curr->b_str);
227 	vim_strncpy(buf->bh_curr->b_str + len, s, (size_t)slen);
228 	buf->bh_space -= slen;
229     }
230     else
231     {
232 	if (slen < MINIMAL_SIZE)
233 	    len = MINIMAL_SIZE;
234 	else
235 	    len = slen;
236 	p = alloc(offsetof(buffblock_T, b_str) + len + 1);
237 	if (p == NULL)
238 	    return; // no space, just forget it
239 	buf->bh_space = (int)(len - slen);
240 	vim_strncpy(p->b_str, s, (size_t)slen);
241 
242 	p->b_next = buf->bh_curr->b_next;
243 	buf->bh_curr->b_next = p;
244 	buf->bh_curr = p;
245     }
246 }
247 
248 /*
249  * Add number "n" to buffer "buf".
250  */
251     static void
add_num_buff(buffheader_T * buf,long n)252 add_num_buff(buffheader_T *buf, long n)
253 {
254     char_u	number[32];
255 
256     sprintf((char *)number, "%ld", n);
257     add_buff(buf, number, -1L);
258 }
259 
260 /*
261  * Add character 'c' to buffer "buf".
262  * Translates special keys, NUL, CSI, K_SPECIAL and multibyte characters.
263  */
264     static void
add_char_buff(buffheader_T * buf,int c)265 add_char_buff(buffheader_T *buf, int c)
266 {
267     char_u	bytes[MB_MAXBYTES + 1];
268     int		len;
269     int		i;
270     char_u	temp[4];
271 
272     if (IS_SPECIAL(c))
273 	len = 1;
274     else
275 	len = (*mb_char2bytes)(c, bytes);
276     for (i = 0; i < len; ++i)
277     {
278 	if (!IS_SPECIAL(c))
279 	    c = bytes[i];
280 
281 	if (IS_SPECIAL(c) || c == K_SPECIAL || c == NUL)
282 	{
283 	    // translate special key code into three byte sequence
284 	    temp[0] = K_SPECIAL;
285 	    temp[1] = K_SECOND(c);
286 	    temp[2] = K_THIRD(c);
287 	    temp[3] = NUL;
288 	}
289 #ifdef FEAT_GUI
290 	else if (c == CSI)
291 	{
292 	    // Translate a CSI to a CSI - KS_EXTRA - KE_CSI sequence
293 	    temp[0] = CSI;
294 	    temp[1] = KS_EXTRA;
295 	    temp[2] = (int)KE_CSI;
296 	    temp[3] = NUL;
297 	}
298 #endif
299 	else
300 	{
301 	    temp[0] = c;
302 	    temp[1] = NUL;
303 	}
304 	add_buff(buf, temp, -1L);
305     }
306 }
307 
308 // First read ahead buffer. Used for translated commands.
309 static buffheader_T readbuf1 = {{NULL, {NUL}}, NULL, 0, 0};
310 
311 // Second read ahead buffer. Used for redo.
312 static buffheader_T readbuf2 = {{NULL, {NUL}}, NULL, 0, 0};
313 
314 /*
315  * Get one byte from the read buffers.  Use readbuf1 one first, use readbuf2
316  * if that one is empty.
317  * If advance == TRUE go to the next char.
318  * No translation is done K_SPECIAL and CSI are escaped.
319  */
320     static int
read_readbuffers(int advance)321 read_readbuffers(int advance)
322 {
323     int c;
324 
325     c = read_readbuf(&readbuf1, advance);
326     if (c == NUL)
327 	c = read_readbuf(&readbuf2, advance);
328     return c;
329 }
330 
331     static int
read_readbuf(buffheader_T * buf,int advance)332 read_readbuf(buffheader_T *buf, int advance)
333 {
334     char_u	c;
335     buffblock_T	*curr;
336 
337     if (buf->bh_first.b_next == NULL)  // buffer is empty
338 	return NUL;
339 
340     curr = buf->bh_first.b_next;
341     c = curr->b_str[buf->bh_index];
342 
343     if (advance)
344     {
345 	if (curr->b_str[++buf->bh_index] == NUL)
346 	{
347 	    buf->bh_first.b_next = curr->b_next;
348 	    vim_free(curr);
349 	    buf->bh_index = 0;
350 	}
351     }
352     return c;
353 }
354 
355 /*
356  * Prepare the read buffers for reading (if they contain something).
357  */
358     static void
start_stuff(void)359 start_stuff(void)
360 {
361     if (readbuf1.bh_first.b_next != NULL)
362     {
363 	readbuf1.bh_curr = &(readbuf1.bh_first);
364 	readbuf1.bh_space = 0;
365     }
366     if (readbuf2.bh_first.b_next != NULL)
367     {
368 	readbuf2.bh_curr = &(readbuf2.bh_first);
369 	readbuf2.bh_space = 0;
370     }
371 }
372 
373 /*
374  * Return TRUE if the stuff buffer is empty.
375  */
376     int
stuff_empty(void)377 stuff_empty(void)
378 {
379     return (readbuf1.bh_first.b_next == NULL
380 	 && readbuf2.bh_first.b_next == NULL);
381 }
382 
383 #if defined(FEAT_EVAL) || defined(PROTO)
384 /*
385  * Return TRUE if readbuf1 is empty.  There may still be redo characters in
386  * redbuf2.
387  */
388     int
readbuf1_empty(void)389 readbuf1_empty(void)
390 {
391     return (readbuf1.bh_first.b_next == NULL);
392 }
393 #endif
394 
395 /*
396  * Set a typeahead character that won't be flushed.
397  */
398     void
typeahead_noflush(int c)399 typeahead_noflush(int c)
400 {
401     typeahead_char = c;
402 }
403 
404 /*
405  * Remove the contents of the stuff buffer and the mapped characters in the
406  * typeahead buffer (used in case of an error).  If "flush_typeahead" is true,
407  * flush all typeahead characters (used when interrupted by a CTRL-C).
408  */
409     void
flush_buffers(flush_buffers_T flush_typeahead)410 flush_buffers(flush_buffers_T flush_typeahead)
411 {
412     init_typebuf();
413 
414     start_stuff();
415     while (read_readbuffers(TRUE) != NUL)
416 	;
417 
418     if (flush_typeahead == FLUSH_MINIMAL)
419     {
420 	// remove mapped characters at the start only
421 	typebuf.tb_off += typebuf.tb_maplen;
422 	typebuf.tb_len -= typebuf.tb_maplen;
423 #if defined(FEAT_CLIENTSERVER) || defined(FEAT_EVAL)
424 	if (typebuf.tb_len == 0)
425 	    typebuf_was_filled = FALSE;
426 #endif
427     }
428     else
429     {
430 	// remove typeahead
431 	if (flush_typeahead == FLUSH_INPUT)
432 	    // We have to get all characters, because we may delete the first
433 	    // part of an escape sequence.  In an xterm we get one char at a
434 	    // time and we have to get them all.
435 	    while (inchar(typebuf.tb_buf, typebuf.tb_buflen - 1, 10L) != 0)
436 		;
437 	typebuf.tb_off = MAXMAPLEN;
438 	typebuf.tb_len = 0;
439 #if defined(FEAT_CLIENTSERVER) || defined(FEAT_EVAL)
440 	// Reset the flag that text received from a client or from feedkeys()
441 	// was inserted in the typeahead buffer.
442 	typebuf_was_filled = FALSE;
443 #endif
444     }
445     typebuf.tb_maplen = 0;
446     typebuf.tb_silent = 0;
447     cmd_silent = FALSE;
448     typebuf.tb_no_abbr_cnt = 0;
449     if (++typebuf.tb_change_cnt == 0)
450 	typebuf.tb_change_cnt = 1;
451 }
452 
453 /*
454  * The previous contents of the redo buffer is kept in old_redobuffer.
455  * This is used for the CTRL-O <.> command in insert mode.
456  */
457     void
ResetRedobuff(void)458 ResetRedobuff(void)
459 {
460     if (!block_redo)
461     {
462 	free_buff(&old_redobuff);
463 	old_redobuff = redobuff;
464 	redobuff.bh_first.b_next = NULL;
465     }
466 }
467 
468 /*
469  * Discard the contents of the redo buffer and restore the previous redo
470  * buffer.
471  */
472     void
CancelRedo(void)473 CancelRedo(void)
474 {
475     if (!block_redo)
476     {
477 	free_buff(&redobuff);
478 	redobuff = old_redobuff;
479 	old_redobuff.bh_first.b_next = NULL;
480 	start_stuff();
481 	while (read_readbuffers(TRUE) != NUL)
482 	    ;
483     }
484 }
485 
486 /*
487  * Save redobuff and old_redobuff to save_redobuff and save_old_redobuff.
488  * Used before executing autocommands and user functions.
489  */
490     void
saveRedobuff(save_redo_T * save_redo)491 saveRedobuff(save_redo_T *save_redo)
492 {
493     char_u	*s;
494 
495     save_redo->sr_redobuff = redobuff;
496     redobuff.bh_first.b_next = NULL;
497     save_redo->sr_old_redobuff = old_redobuff;
498     old_redobuff.bh_first.b_next = NULL;
499 
500     // Make a copy, so that ":normal ." in a function works.
501     s = get_buffcont(&save_redo->sr_redobuff, FALSE);
502     if (s != NULL)
503     {
504 	add_buff(&redobuff, s, -1L);
505 	vim_free(s);
506     }
507 }
508 
509 /*
510  * Restore redobuff and old_redobuff from save_redobuff and save_old_redobuff.
511  * Used after executing autocommands and user functions.
512  */
513     void
restoreRedobuff(save_redo_T * save_redo)514 restoreRedobuff(save_redo_T *save_redo)
515 {
516     free_buff(&redobuff);
517     redobuff = save_redo->sr_redobuff;
518     free_buff(&old_redobuff);
519     old_redobuff = save_redo->sr_old_redobuff;
520 }
521 
522 /*
523  * Append "s" to the redo buffer.
524  * K_SPECIAL and CSI should already have been escaped.
525  */
526     void
AppendToRedobuff(char_u * s)527 AppendToRedobuff(char_u *s)
528 {
529     if (!block_redo)
530 	add_buff(&redobuff, s, -1L);
531 }
532 
533 /*
534  * Append to Redo buffer literally, escaping special characters with CTRL-V.
535  * K_SPECIAL and CSI are escaped as well.
536  */
537     void
AppendToRedobuffLit(char_u * str,int len)538 AppendToRedobuffLit(
539     char_u	*str,
540     int		len)	    // length of "str" or -1 for up to the NUL
541 {
542     char_u	*s = str;
543     int		c;
544     char_u	*start;
545 
546     if (block_redo)
547 	return;
548 
549     while (len < 0 ? *s != NUL : s - str < len)
550     {
551 	// Put a string of normal characters in the redo buffer (that's
552 	// faster).
553 	start = s;
554 	while (*s >= ' '
555 #ifndef EBCDIC
556 		&& *s < DEL	// EBCDIC: all chars above space are normal
557 #endif
558 		&& (len < 0 || s - str < len))
559 	    ++s;
560 
561 	// Don't put '0' or '^' as last character, just in case a CTRL-D is
562 	// typed next.
563 	if (*s == NUL && (s[-1] == '0' || s[-1] == '^'))
564 	    --s;
565 	if (s > start)
566 	    add_buff(&redobuff, start, (long)(s - start));
567 
568 	if (*s == NUL || (len >= 0 && s - str >= len))
569 	    break;
570 
571 	// Handle a special or multibyte character.
572 	if (has_mbyte)
573 	    // Handle composing chars separately.
574 	    c = mb_cptr2char_adv(&s);
575 	else
576 	    c = *s++;
577 	if (c < ' ' || c == DEL || (*s == NUL && (c == '0' || c == '^')))
578 	    add_char_buff(&redobuff, Ctrl_V);
579 
580 	// CTRL-V '0' must be inserted as CTRL-V 048 (EBCDIC: xf0)
581 	if (*s == NUL && c == '0')
582 #ifdef EBCDIC
583 	    add_buff(&redobuff, (char_u *)"xf0", 3L);
584 #else
585 	    add_buff(&redobuff, (char_u *)"048", 3L);
586 #endif
587 	else
588 	    add_char_buff(&redobuff, c);
589     }
590 }
591 
592 /*
593  * Append a character to the redo buffer.
594  * Translates special keys, NUL, CSI, K_SPECIAL and multibyte characters.
595  */
596     void
AppendCharToRedobuff(int c)597 AppendCharToRedobuff(int c)
598 {
599     if (!block_redo)
600 	add_char_buff(&redobuff, c);
601 }
602 
603 /*
604  * Append a number to the redo buffer.
605  */
606     void
AppendNumberToRedobuff(long n)607 AppendNumberToRedobuff(long n)
608 {
609     if (!block_redo)
610 	add_num_buff(&redobuff, n);
611 }
612 
613 /*
614  * Append string "s" to the stuff buffer.
615  * CSI and K_SPECIAL must already have been escaped.
616  */
617     void
stuffReadbuff(char_u * s)618 stuffReadbuff(char_u *s)
619 {
620     add_buff(&readbuf1, s, -1L);
621 }
622 
623 /*
624  * Append string "s" to the redo stuff buffer.
625  * CSI and K_SPECIAL must already have been escaped.
626  */
627     void
stuffRedoReadbuff(char_u * s)628 stuffRedoReadbuff(char_u *s)
629 {
630     add_buff(&readbuf2, s, -1L);
631 }
632 
633     static void
stuffReadbuffLen(char_u * s,long len)634 stuffReadbuffLen(char_u *s, long len)
635 {
636     add_buff(&readbuf1, s, len);
637 }
638 
639 #if defined(FEAT_EVAL) || defined(PROTO)
640 /*
641  * Stuff "s" into the stuff buffer, leaving special key codes unmodified and
642  * escaping other K_SPECIAL and CSI bytes.
643  * Change CR, LF and ESC into a space.
644  */
645     void
stuffReadbuffSpec(char_u * s)646 stuffReadbuffSpec(char_u *s)
647 {
648     int c;
649 
650     while (*s != NUL)
651     {
652 	if (*s == K_SPECIAL && s[1] != NUL && s[2] != NUL)
653 	{
654 	    // Insert special key literally.
655 	    stuffReadbuffLen(s, 3L);
656 	    s += 3;
657 	}
658 	else
659 	{
660 	    c = mb_ptr2char_adv(&s);
661 	    if (c == CAR || c == NL || c == ESC)
662 		c = ' ';
663 	    stuffcharReadbuff(c);
664 	}
665     }
666 }
667 #endif
668 
669 /*
670  * Append a character to the stuff buffer.
671  * Translates special keys, NUL, CSI, K_SPECIAL and multibyte characters.
672  */
673     void
stuffcharReadbuff(int c)674 stuffcharReadbuff(int c)
675 {
676     add_char_buff(&readbuf1, c);
677 }
678 
679 /*
680  * Append a number to the stuff buffer.
681  */
682     void
stuffnumReadbuff(long n)683 stuffnumReadbuff(long n)
684 {
685     add_num_buff(&readbuf1, n);
686 }
687 
688 /*
689  * Stuff a string into the typeahead buffer, such that edit() will insert it
690  * literally ("literally" TRUE) or interpret is as typed characters.
691  */
692     void
stuffescaped(char_u * arg,int literally)693 stuffescaped(char_u *arg, int literally)
694 {
695     int		c;
696     char_u	*start;
697 
698     while (*arg != NUL)
699     {
700 	// Stuff a sequence of normal ASCII characters, that's fast.  Also
701 	// stuff K_SPECIAL to get the effect of a special key when "literally"
702 	// is TRUE.
703 	start = arg;
704 	while ((*arg >= ' '
705 #ifndef EBCDIC
706 		    && *arg < DEL // EBCDIC: chars above space are normal
707 #endif
708 		    )
709 		|| (*arg == K_SPECIAL && !literally))
710 	    ++arg;
711 	if (arg > start)
712 	    stuffReadbuffLen(start, (long)(arg - start));
713 
714 	// stuff a single special character
715 	if (*arg != NUL)
716 	{
717 	    if (has_mbyte)
718 		c = mb_cptr2char_adv(&arg);
719 	    else
720 		c = *arg++;
721 	    if (literally && ((c < ' ' && c != TAB) || c == DEL))
722 		stuffcharReadbuff(Ctrl_V);
723 	    stuffcharReadbuff(c);
724 	}
725     }
726 }
727 
728 /*
729  * Read a character from the redo buffer.  Translates K_SPECIAL, CSI and
730  * multibyte characters.
731  * The redo buffer is left as it is.
732  * If init is TRUE, prepare for redo, return FAIL if nothing to redo, OK
733  * otherwise.
734  * If old is TRUE, use old_redobuff instead of redobuff.
735  */
736     static int
read_redo(int init,int old_redo)737 read_redo(int init, int old_redo)
738 {
739     static buffblock_T	*bp;
740     static char_u	*p;
741     int			c;
742     int			n;
743     char_u		buf[MB_MAXBYTES + 1];
744     int			i;
745 
746     if (init)
747     {
748 	if (old_redo)
749 	    bp = old_redobuff.bh_first.b_next;
750 	else
751 	    bp = redobuff.bh_first.b_next;
752 	if (bp == NULL)
753 	    return FAIL;
754 	p = bp->b_str;
755 	return OK;
756     }
757     if ((c = *p) != NUL)
758     {
759 	// Reverse the conversion done by add_char_buff()
760 	// For a multi-byte character get all the bytes and return the
761 	// converted character.
762 	if (has_mbyte && (c != K_SPECIAL || p[1] == KS_SPECIAL))
763 	    n = MB_BYTE2LEN_CHECK(c);
764 	else
765 	    n = 1;
766 	for (i = 0; ; ++i)
767 	{
768 	    if (c == K_SPECIAL) // special key or escaped K_SPECIAL
769 	    {
770 		c = TO_SPECIAL(p[1], p[2]);
771 		p += 2;
772 	    }
773 #ifdef FEAT_GUI
774 	    if (c == CSI)	// escaped CSI
775 		p += 2;
776 #endif
777 	    if (*++p == NUL && bp->b_next != NULL)
778 	    {
779 		bp = bp->b_next;
780 		p = bp->b_str;
781 	    }
782 	    buf[i] = c;
783 	    if (i == n - 1)	// last byte of a character
784 	    {
785 		if (n != 1)
786 		    c = (*mb_ptr2char)(buf);
787 		break;
788 	    }
789 	    c = *p;
790 	    if (c == NUL)	// cannot happen?
791 		break;
792 	}
793     }
794 
795     return c;
796 }
797 
798 /*
799  * Copy the rest of the redo buffer into the stuff buffer (in a slow way).
800  * If old_redo is TRUE, use old_redobuff instead of redobuff.
801  * The escaped K_SPECIAL and CSI are copied without translation.
802  */
803     static void
copy_redo(int old_redo)804 copy_redo(int old_redo)
805 {
806     int	    c;
807 
808     while ((c = read_redo(FALSE, old_redo)) != NUL)
809 	add_char_buff(&readbuf2, c);
810 }
811 
812 /*
813  * Stuff the redo buffer into readbuf2.
814  * Insert the redo count into the command.
815  * If "old_redo" is TRUE, the last but one command is repeated
816  * instead of the last command (inserting text). This is used for
817  * CTRL-O <.> in insert mode
818  *
819  * return FAIL for failure, OK otherwise
820  */
821     int
start_redo(long count,int old_redo)822 start_redo(long count, int old_redo)
823 {
824     int	    c;
825 
826     // init the pointers; return if nothing to redo
827     if (read_redo(TRUE, old_redo) == FAIL)
828 	return FAIL;
829 
830     c = read_redo(FALSE, old_redo);
831 
832     // copy the buffer name, if present
833     if (c == '"')
834     {
835 	add_buff(&readbuf2, (char_u *)"\"", 1L);
836 	c = read_redo(FALSE, old_redo);
837 
838 	// if a numbered buffer is used, increment the number
839 	if (c >= '1' && c < '9')
840 	    ++c;
841 	add_char_buff(&readbuf2, c);
842 
843 	// the expression register should be re-evaluated
844 	if (c == '=')
845 	{
846 	    add_char_buff(&readbuf2, CAR);
847 	    cmd_silent = TRUE;
848 	}
849 
850 	c = read_redo(FALSE, old_redo);
851     }
852 
853     if (c == 'v')   // redo Visual
854     {
855 	VIsual = curwin->w_cursor;
856 	VIsual_active = TRUE;
857 	VIsual_select = FALSE;
858 	VIsual_reselect = TRUE;
859 	redo_VIsual_busy = TRUE;
860 	c = read_redo(FALSE, old_redo);
861     }
862 
863     // try to enter the count (in place of a previous count)
864     if (count)
865     {
866 	while (VIM_ISDIGIT(c))	// skip "old" count
867 	    c = read_redo(FALSE, old_redo);
868 	add_num_buff(&readbuf2, count);
869     }
870 
871     // copy from the redo buffer into the stuff buffer
872     add_char_buff(&readbuf2, c);
873     copy_redo(old_redo);
874     return OK;
875 }
876 
877 /*
878  * Repeat the last insert (R, o, O, a, A, i or I command) by stuffing
879  * the redo buffer into readbuf2.
880  * return FAIL for failure, OK otherwise
881  */
882     int
start_redo_ins(void)883 start_redo_ins(void)
884 {
885     int	    c;
886 
887     if (read_redo(TRUE, FALSE) == FAIL)
888 	return FAIL;
889     start_stuff();
890 
891     // skip the count and the command character
892     while ((c = read_redo(FALSE, FALSE)) != NUL)
893     {
894 	if (vim_strchr((char_u *)"AaIiRrOo", c) != NULL)
895 	{
896 	    if (c == 'O' || c == 'o')
897 		add_buff(&readbuf2, NL_STR, -1L);
898 	    break;
899 	}
900     }
901 
902     // copy the typed text from the redo buffer into the stuff buffer
903     copy_redo(FALSE);
904     block_redo = TRUE;
905     return OK;
906 }
907 
908     void
stop_redo_ins(void)909 stop_redo_ins(void)
910 {
911     block_redo = FALSE;
912 }
913 
914 /*
915  * Initialize typebuf.tb_buf to point to typebuf_init.
916  * alloc() cannot be used here: In out-of-memory situations it would
917  * be impossible to type anything.
918  */
919     static void
init_typebuf(void)920 init_typebuf(void)
921 {
922     if (typebuf.tb_buf == NULL)
923     {
924 	typebuf.tb_buf = typebuf_init;
925 	typebuf.tb_noremap = noremapbuf_init;
926 	typebuf.tb_buflen = TYPELEN_INIT;
927 	typebuf.tb_len = 0;
928 	typebuf.tb_off = MAXMAPLEN + 4;
929 	typebuf.tb_change_cnt = 1;
930     }
931 }
932 
933 /*
934  * Returns TRUE when keys cannot be remapped.
935  */
936     int
noremap_keys(void)937 noremap_keys(void)
938 {
939     return KeyNoremap & (RM_NONE|RM_SCRIPT);
940 }
941 
942 /*
943  * Insert a string in position 'offset' in the typeahead buffer (for "@r"
944  * and ":normal" command, vgetorpeek() and check_termcode()).
945  *
946  * If noremap is REMAP_YES, new string can be mapped again.
947  * If noremap is REMAP_NONE, new string cannot be mapped again.
948  * If noremap is REMAP_SKIP, fist char of new string cannot be mapped again,
949  * but abbreviations are allowed.
950  * If noremap is REMAP_SCRIPT, new string cannot be mapped again, except for
951  *			script-local mappings.
952  * If noremap is > 0, that many characters of the new string cannot be mapped.
953  *
954  * If nottyped is TRUE, the string does not return KeyTyped (don't use when
955  * offset is non-zero!).
956  *
957  * If silent is TRUE, cmd_silent is set when the characters are obtained.
958  *
959  * return FAIL for failure, OK otherwise
960  */
961     int
ins_typebuf(char_u * str,int noremap,int offset,int nottyped,int silent)962 ins_typebuf(
963     char_u	*str,
964     int		noremap,
965     int		offset,
966     int		nottyped,
967     int		silent)
968 {
969     char_u	*s1, *s2;
970     int		newlen;
971     int		addlen;
972     int		i;
973     int		newoff;
974     int		val;
975     int		nrm;
976 
977     init_typebuf();
978     if (++typebuf.tb_change_cnt == 0)
979 	typebuf.tb_change_cnt = 1;
980     state_no_longer_safe("ins_typebuf()");
981 
982     addlen = (int)STRLEN(str);
983 
984     if (offset == 0 && addlen <= typebuf.tb_off)
985     {
986 	/*
987 	 * Easy case: there is room in front of typebuf.tb_buf[typebuf.tb_off]
988 	 */
989 	typebuf.tb_off -= addlen;
990 	mch_memmove(typebuf.tb_buf + typebuf.tb_off, str, (size_t)addlen);
991     }
992     else if (typebuf.tb_len == 0 && typebuf.tb_buflen
993 					       >= addlen + 3 * (MAXMAPLEN + 4))
994     {
995 	/*
996 	 * Buffer is empty and string fits in the existing buffer.
997 	 * Leave some space before and after, if possible.
998 	 */
999 	typebuf.tb_off = (typebuf.tb_buflen - addlen - 3 * (MAXMAPLEN + 4)) / 2;
1000 	mch_memmove(typebuf.tb_buf + typebuf.tb_off, str, (size_t)addlen);
1001     }
1002     else
1003     {
1004 	int extra;
1005 
1006 	/*
1007 	 * Need to allocate a new buffer.
1008 	 * In typebuf.tb_buf there must always be room for 3 * (MAXMAPLEN + 4)
1009 	 * characters.  We add some extra room to avoid having to allocate too
1010 	 * often.
1011 	 */
1012 	newoff = MAXMAPLEN + 4;
1013 	extra = addlen + newoff + 4 * (MAXMAPLEN + 4);
1014 	if (typebuf.tb_len > 2147483647 - extra)
1015 	{
1016 	    // string is getting too long for a 32 bit int
1017 	    emsg(_(e_toocompl));    // also calls flush_buffers
1018 	    setcursor();
1019 	    return FAIL;
1020 	}
1021 	newlen = typebuf.tb_len + extra;
1022 	s1 = alloc(newlen);
1023 	if (s1 == NULL)		    // out of memory
1024 	    return FAIL;
1025 	s2 = alloc(newlen);
1026 	if (s2 == NULL)		    // out of memory
1027 	{
1028 	    vim_free(s1);
1029 	    return FAIL;
1030 	}
1031 	typebuf.tb_buflen = newlen;
1032 
1033 	// copy the old chars, before the insertion point
1034 	mch_memmove(s1 + newoff, typebuf.tb_buf + typebuf.tb_off,
1035 							      (size_t)offset);
1036 	// copy the new chars
1037 	mch_memmove(s1 + newoff + offset, str, (size_t)addlen);
1038 	// copy the old chars, after the insertion point, including the	NUL at
1039 	// the end
1040 	mch_memmove(s1 + newoff + offset + addlen,
1041 				     typebuf.tb_buf + typebuf.tb_off + offset,
1042 				       (size_t)(typebuf.tb_len - offset + 1));
1043 	if (typebuf.tb_buf != typebuf_init)
1044 	    vim_free(typebuf.tb_buf);
1045 	typebuf.tb_buf = s1;
1046 
1047 	mch_memmove(s2 + newoff, typebuf.tb_noremap + typebuf.tb_off,
1048 							      (size_t)offset);
1049 	mch_memmove(s2 + newoff + offset + addlen,
1050 		   typebuf.tb_noremap + typebuf.tb_off + offset,
1051 					   (size_t)(typebuf.tb_len - offset));
1052 	if (typebuf.tb_noremap != noremapbuf_init)
1053 	    vim_free(typebuf.tb_noremap);
1054 	typebuf.tb_noremap = s2;
1055 
1056 	typebuf.tb_off = newoff;
1057     }
1058     typebuf.tb_len += addlen;
1059 
1060     // If noremap == REMAP_SCRIPT: do remap script-local mappings.
1061     if (noremap == REMAP_SCRIPT)
1062 	val = RM_SCRIPT;
1063     else if (noremap == REMAP_SKIP)
1064 	val = RM_ABBR;
1065     else
1066 	val = RM_NONE;
1067 
1068     /*
1069      * Adjust typebuf.tb_noremap[] for the new characters:
1070      * If noremap == REMAP_NONE or REMAP_SCRIPT: new characters are
1071      *			(sometimes) not remappable
1072      * If noremap == REMAP_YES: all the new characters are mappable
1073      * If noremap  > 0: "noremap" characters are not remappable, the rest
1074      *			mappable
1075      */
1076     if (noremap == REMAP_SKIP)
1077 	nrm = 1;
1078     else if (noremap < 0)
1079 	nrm = addlen;
1080     else
1081 	nrm = noremap;
1082     for (i = 0; i < addlen; ++i)
1083 	typebuf.tb_noremap[typebuf.tb_off + i + offset] =
1084 						  (--nrm >= 0) ? val : RM_YES;
1085 
1086     // tb_maplen and tb_silent only remember the length of mapped and/or
1087     // silent mappings at the start of the buffer, assuming that a mapped
1088     // sequence doesn't result in typed characters.
1089     if (nottyped || typebuf.tb_maplen > offset)
1090 	typebuf.tb_maplen += addlen;
1091     if (silent || typebuf.tb_silent > offset)
1092     {
1093 	typebuf.tb_silent += addlen;
1094 	cmd_silent = TRUE;
1095     }
1096     if (typebuf.tb_no_abbr_cnt && offset == 0)	// and not used for abbrev.s
1097 	typebuf.tb_no_abbr_cnt += addlen;
1098 
1099     return OK;
1100 }
1101 
1102 /*
1103  * Put character "c" back into the typeahead buffer.
1104  * Can be used for a character obtained by vgetc() that needs to be put back.
1105  * Uses cmd_silent, KeyTyped and KeyNoremap to restore the flags belonging to
1106  * the char.
1107  */
1108     void
ins_char_typebuf(int c,int modifier)1109 ins_char_typebuf(int c, int modifier)
1110 {
1111     char_u	buf[MB_MAXBYTES + 4];
1112     int		idx = 0;
1113 
1114     if (modifier != 0)
1115     {
1116 	buf[0] = K_SPECIAL;
1117 	buf[1] = KS_MODIFIER;
1118 	buf[2] = modifier;
1119 	buf[3] = NUL;
1120 	idx = 3;
1121     }
1122     if (IS_SPECIAL(c))
1123     {
1124 	buf[idx] = K_SPECIAL;
1125 	buf[idx + 1] = K_SECOND(c);
1126 	buf[idx + 2] = K_THIRD(c);
1127 	buf[idx + 3] = NUL;
1128 	idx += 3;
1129     }
1130     else
1131 	buf[(*mb_char2bytes)(c, buf + idx) + idx] = NUL;
1132     (void)ins_typebuf(buf, KeyNoremap, 0, !KeyTyped, cmd_silent);
1133 }
1134 
1135 /*
1136  * Return TRUE if the typeahead buffer was changed (while waiting for a
1137  * character to arrive).  Happens when a message was received from a client or
1138  * from feedkeys().
1139  * But check in a more generic way to avoid trouble: When "typebuf.tb_buf"
1140  * changed it was reallocated and the old pointer can no longer be used.
1141  * Or "typebuf.tb_off" may have been changed and we would overwrite characters
1142  * that was just added.
1143  */
1144     int
typebuf_changed(int tb_change_cnt)1145 typebuf_changed(
1146     int		tb_change_cnt)	// old value of typebuf.tb_change_cnt
1147 {
1148     return (tb_change_cnt != 0 && (typebuf.tb_change_cnt != tb_change_cnt
1149 #if defined(FEAT_CLIENTSERVER) || defined(FEAT_EVAL)
1150 	    || typebuf_was_filled
1151 #endif
1152 	   ));
1153 }
1154 
1155 /*
1156  * Return TRUE if there are no characters in the typeahead buffer that have
1157  * not been typed (result from a mapping or come from ":normal").
1158  */
1159     int
typebuf_typed(void)1160 typebuf_typed(void)
1161 {
1162     return typebuf.tb_maplen == 0;
1163 }
1164 
1165 /*
1166  * Return the number of characters that are mapped (or not typed).
1167  */
1168     int
typebuf_maplen(void)1169 typebuf_maplen(void)
1170 {
1171     return typebuf.tb_maplen;
1172 }
1173 
1174 /*
1175  * remove "len" characters from typebuf.tb_buf[typebuf.tb_off + offset]
1176  */
1177     void
del_typebuf(int len,int offset)1178 del_typebuf(int len, int offset)
1179 {
1180     int	    i;
1181 
1182     if (len == 0)
1183 	return;		// nothing to do
1184 
1185     typebuf.tb_len -= len;
1186 
1187     /*
1188      * Easy case: Just increase typebuf.tb_off.
1189      */
1190     if (offset == 0 && typebuf.tb_buflen - (typebuf.tb_off + len)
1191 							 >= 3 * MAXMAPLEN + 3)
1192 	typebuf.tb_off += len;
1193     /*
1194      * Have to move the characters in typebuf.tb_buf[] and typebuf.tb_noremap[]
1195      */
1196     else
1197     {
1198 	i = typebuf.tb_off + offset;
1199 	/*
1200 	 * Leave some extra room at the end to avoid reallocation.
1201 	 */
1202 	if (typebuf.tb_off > MAXMAPLEN)
1203 	{
1204 	    mch_memmove(typebuf.tb_buf + MAXMAPLEN,
1205 			     typebuf.tb_buf + typebuf.tb_off, (size_t)offset);
1206 	    mch_memmove(typebuf.tb_noremap + MAXMAPLEN,
1207 			 typebuf.tb_noremap + typebuf.tb_off, (size_t)offset);
1208 	    typebuf.tb_off = MAXMAPLEN;
1209 	}
1210 	// adjust typebuf.tb_buf (include the NUL at the end)
1211 	mch_memmove(typebuf.tb_buf + typebuf.tb_off + offset,
1212 						     typebuf.tb_buf + i + len,
1213 				       (size_t)(typebuf.tb_len - offset + 1));
1214 	// adjust typebuf.tb_noremap[]
1215 	mch_memmove(typebuf.tb_noremap + typebuf.tb_off + offset,
1216 						 typebuf.tb_noremap + i + len,
1217 					   (size_t)(typebuf.tb_len - offset));
1218     }
1219 
1220     if (typebuf.tb_maplen > offset)		// adjust tb_maplen
1221     {
1222 	if (typebuf.tb_maplen < offset + len)
1223 	    typebuf.tb_maplen = offset;
1224 	else
1225 	    typebuf.tb_maplen -= len;
1226     }
1227     if (typebuf.tb_silent > offset)		// adjust tb_silent
1228     {
1229 	if (typebuf.tb_silent < offset + len)
1230 	    typebuf.tb_silent = offset;
1231 	else
1232 	    typebuf.tb_silent -= len;
1233     }
1234     if (typebuf.tb_no_abbr_cnt > offset)	// adjust tb_no_abbr_cnt
1235     {
1236 	if (typebuf.tb_no_abbr_cnt < offset + len)
1237 	    typebuf.tb_no_abbr_cnt = offset;
1238 	else
1239 	    typebuf.tb_no_abbr_cnt -= len;
1240     }
1241 
1242 #if defined(FEAT_CLIENTSERVER) || defined(FEAT_EVAL)
1243     // Reset the flag that text received from a client or from feedkeys()
1244     // was inserted in the typeahead buffer.
1245     typebuf_was_filled = FALSE;
1246 #endif
1247     if (++typebuf.tb_change_cnt == 0)
1248 	typebuf.tb_change_cnt = 1;
1249 }
1250 
1251 /*
1252  * Write typed characters to script file.
1253  * If recording is on put the character in the recordbuffer.
1254  */
1255     static void
gotchars(char_u * chars,int len)1256 gotchars(char_u *chars, int len)
1257 {
1258     char_u		*s = chars;
1259     int			i;
1260     static char_u	buf[4];
1261     static int		buflen = 0;
1262     int			todo = len;
1263 
1264     while (todo--)
1265     {
1266 	buf[buflen++] = *s++;
1267 
1268 	// When receiving a special key sequence, store it until we have all
1269 	// the bytes and we can decide what to do with it.
1270 	if (buflen == 1 && buf[0] == K_SPECIAL)
1271 	    continue;
1272 	if (buflen == 2)
1273 	    continue;
1274 	if (buflen == 3 && buf[1] == KS_EXTRA
1275 		       && (buf[2] == KE_FOCUSGAINED || buf[2] == KE_FOCUSLOST))
1276 	{
1277 	    // Drop K_FOCUSGAINED and K_FOCUSLOST, they are not useful in a
1278 	    // recording.
1279 	    buflen = 0;
1280 	    continue;
1281 	}
1282 
1283 	// Handle one byte at a time; no translation to be done.
1284 	for (i = 0; i < buflen; ++i)
1285 	    updatescript(buf[i]);
1286 
1287 	if (reg_recording != 0)
1288 	{
1289 	    buf[buflen] = NUL;
1290 	    add_buff(&recordbuff, buf, (long)buflen);
1291 	    // remember how many chars were last recorded
1292 	    last_recorded_len += buflen;
1293 	}
1294 	buflen = 0;
1295     }
1296     may_sync_undo();
1297 
1298 #ifdef FEAT_EVAL
1299     // output "debug mode" message next time in debug mode
1300     debug_did_msg = FALSE;
1301 #endif
1302 
1303     // Since characters have been typed, consider the following to be in
1304     // another mapping.  Search string will be kept in history.
1305     ++maptick;
1306 }
1307 
1308 /*
1309  * Sync undo.  Called when typed characters are obtained from the typeahead
1310  * buffer, or when a menu is used.
1311  * Do not sync:
1312  * - In Insert mode, unless cursor key has been used.
1313  * - While reading a script file.
1314  * - When no_u_sync is non-zero.
1315  */
1316     static void
may_sync_undo(void)1317 may_sync_undo(void)
1318 {
1319     if ((!(State & (INSERT + CMDLINE)) || arrow_used)
1320 					       && scriptin[curscript] == NULL)
1321 	u_sync(FALSE);
1322 }
1323 
1324 /*
1325  * Make "typebuf" empty and allocate new buffers.
1326  * Returns FAIL when out of memory.
1327  */
1328     static int
alloc_typebuf(void)1329 alloc_typebuf(void)
1330 {
1331     typebuf.tb_buf = alloc(TYPELEN_INIT);
1332     typebuf.tb_noremap = alloc(TYPELEN_INIT);
1333     if (typebuf.tb_buf == NULL || typebuf.tb_noremap == NULL)
1334     {
1335 	free_typebuf();
1336 	return FAIL;
1337     }
1338     typebuf.tb_buflen = TYPELEN_INIT;
1339     typebuf.tb_off = MAXMAPLEN + 4;  // can insert without realloc
1340     typebuf.tb_len = 0;
1341     typebuf.tb_maplen = 0;
1342     typebuf.tb_silent = 0;
1343     typebuf.tb_no_abbr_cnt = 0;
1344     if (++typebuf.tb_change_cnt == 0)
1345 	typebuf.tb_change_cnt = 1;
1346 #if defined(FEAT_CLIENTSERVER) || defined(FEAT_EVAL)
1347     typebuf_was_filled = FALSE;
1348 #endif
1349     return OK;
1350 }
1351 
1352 /*
1353  * Free the buffers of "typebuf".
1354  */
1355     static void
free_typebuf(void)1356 free_typebuf(void)
1357 {
1358     if (typebuf.tb_buf == typebuf_init)
1359 	internal_error("Free typebuf 1");
1360     else
1361 	VIM_CLEAR(typebuf.tb_buf);
1362     if (typebuf.tb_noremap == noremapbuf_init)
1363 	internal_error("Free typebuf 2");
1364     else
1365 	VIM_CLEAR(typebuf.tb_noremap);
1366 }
1367 
1368 /*
1369  * When doing ":so! file", the current typeahead needs to be saved, and
1370  * restored when "file" has been read completely.
1371  */
1372 static typebuf_T saved_typebuf[NSCRIPT];
1373 
1374     int
save_typebuf(void)1375 save_typebuf(void)
1376 {
1377     init_typebuf();
1378     saved_typebuf[curscript] = typebuf;
1379     // If out of memory: restore typebuf and close file.
1380     if (alloc_typebuf() == FAIL)
1381     {
1382 	closescript();
1383 	return FAIL;
1384     }
1385     return OK;
1386 }
1387 
1388 static int old_char = -1;	// character put back by vungetc()
1389 static int old_mod_mask;	// mod_mask for ungotten character
1390 static int old_mouse_row;	// mouse_row related to old_char
1391 static int old_mouse_col;	// mouse_col related to old_char
1392 
1393 /*
1394  * Save all three kinds of typeahead, so that the user must type at a prompt.
1395  */
1396     void
save_typeahead(tasave_T * tp)1397 save_typeahead(tasave_T *tp)
1398 {
1399     tp->save_typebuf = typebuf;
1400     tp->typebuf_valid = (alloc_typebuf() == OK);
1401     if (!tp->typebuf_valid)
1402 	typebuf = tp->save_typebuf;
1403 
1404     tp->old_char = old_char;
1405     tp->old_mod_mask = old_mod_mask;
1406     old_char = -1;
1407 
1408     tp->save_readbuf1 = readbuf1;
1409     readbuf1.bh_first.b_next = NULL;
1410     tp->save_readbuf2 = readbuf2;
1411     readbuf2.bh_first.b_next = NULL;
1412 # ifdef USE_INPUT_BUF
1413     tp->save_inputbuf = get_input_buf();
1414 # endif
1415 }
1416 
1417 /*
1418  * Restore the typeahead to what it was before calling save_typeahead().
1419  * The allocated memory is freed, can only be called once!
1420  * When "overwrite" is FALSE input typed later is kept.
1421  */
1422     void
restore_typeahead(tasave_T * tp,int overwrite UNUSED)1423 restore_typeahead(tasave_T *tp, int overwrite UNUSED)
1424 {
1425     if (tp->typebuf_valid)
1426     {
1427 	free_typebuf();
1428 	typebuf = tp->save_typebuf;
1429     }
1430 
1431     old_char = tp->old_char;
1432     old_mod_mask = tp->old_mod_mask;
1433 
1434     free_buff(&readbuf1);
1435     readbuf1 = tp->save_readbuf1;
1436     free_buff(&readbuf2);
1437     readbuf2 = tp->save_readbuf2;
1438 # ifdef USE_INPUT_BUF
1439     set_input_buf(tp->save_inputbuf, overwrite);
1440 # endif
1441 }
1442 
1443 /*
1444  * Open a new script file for the ":source!" command.
1445  */
1446     void
openscript(char_u * name,int directly)1447 openscript(
1448     char_u	*name,
1449     int		directly)	// when TRUE execute directly
1450 {
1451     if (curscript + 1 == NSCRIPT)
1452     {
1453 	emsg(_(e_scripts_nested_too_deep));
1454 	return;
1455     }
1456 
1457     // Disallow sourcing a file in the sandbox, the commands would be executed
1458     // later, possibly outside of the sandbox.
1459     if (check_secure())
1460 	return;
1461 
1462 #ifdef FEAT_EVAL
1463     if (ignore_script)
1464 	// Not reading from script, also don't open one.  Warning message?
1465 	return;
1466 #endif
1467 
1468     if (scriptin[curscript] != NULL)	// already reading script
1469 	++curscript;
1470 				// use NameBuff for expanded name
1471     expand_env(name, NameBuff, MAXPATHL);
1472     if ((scriptin[curscript] = mch_fopen((char *)NameBuff, READBIN)) == NULL)
1473     {
1474 	semsg(_(e_notopen), name);
1475 	if (curscript)
1476 	    --curscript;
1477 	return;
1478     }
1479     if (save_typebuf() == FAIL)
1480 	return;
1481 
1482     /*
1483      * Execute the commands from the file right now when using ":source!"
1484      * after ":global" or ":argdo" or in a loop.  Also when another command
1485      * follows.  This means the display won't be updated.  Don't do this
1486      * always, "make test" would fail.
1487      */
1488     if (directly)
1489     {
1490 	oparg_T	oa;
1491 	int	oldcurscript;
1492 	int	save_State = State;
1493 	int	save_restart_edit = restart_edit;
1494 	int	save_insertmode = p_im;
1495 	int	save_finish_op = finish_op;
1496 	int	save_msg_scroll = msg_scroll;
1497 
1498 	State = NORMAL;
1499 	msg_scroll = FALSE;	// no msg scrolling in Normal mode
1500 	restart_edit = 0;	// don't go to Insert mode
1501 	p_im = FALSE;		// don't use 'insertmode'
1502 	clear_oparg(&oa);
1503 	finish_op = FALSE;
1504 
1505 	oldcurscript = curscript;
1506 	do
1507 	{
1508 	    update_topline_cursor();	// update cursor position and topline
1509 	    normal_cmd(&oa, FALSE);	// execute one command
1510 	    (void)vpeekc();		// check for end of file
1511 	}
1512 	while (scriptin[oldcurscript] != NULL);
1513 
1514 	State = save_State;
1515 	msg_scroll = save_msg_scroll;
1516 	restart_edit = save_restart_edit;
1517 	p_im = save_insertmode;
1518 	finish_op = save_finish_op;
1519     }
1520 }
1521 
1522 /*
1523  * Close the currently active input script.
1524  */
1525     static void
closescript(void)1526 closescript(void)
1527 {
1528     free_typebuf();
1529     typebuf = saved_typebuf[curscript];
1530 
1531     fclose(scriptin[curscript]);
1532     scriptin[curscript] = NULL;
1533     if (curscript > 0)
1534 	--curscript;
1535 }
1536 
1537 #if defined(EXITFREE) || defined(PROTO)
1538     void
close_all_scripts(void)1539 close_all_scripts(void)
1540 {
1541     while (scriptin[0] != NULL)
1542 	closescript();
1543 }
1544 #endif
1545 
1546 /*
1547  * Return TRUE when reading keys from a script file.
1548  */
1549     int
using_script(void)1550 using_script(void)
1551 {
1552     return scriptin[curscript] != NULL;
1553 }
1554 
1555 /*
1556  * This function is called just before doing a blocking wait.  Thus after
1557  * waiting 'updatetime' for a character to arrive.
1558  */
1559     void
before_blocking(void)1560 before_blocking(void)
1561 {
1562     updatescript(0);
1563 #ifdef FEAT_EVAL
1564     if (may_garbage_collect)
1565 	garbage_collect(FALSE);
1566 #endif
1567 }
1568 
1569 /*
1570  * updatescript() is called when a character can be written into the script file
1571  * or when we have waited some time for a character (c == 0)
1572  *
1573  * All the changed memfiles are synced if c == 0 or when the number of typed
1574  * characters reaches 'updatecount' and 'updatecount' is non-zero.
1575  */
1576     static void
updatescript(int c)1577 updatescript(int c)
1578 {
1579     static int	    count = 0;
1580 
1581     if (c && scriptout)
1582 	putc(c, scriptout);
1583     if (c == 0 || (p_uc > 0 && ++count >= p_uc))
1584     {
1585 	ml_sync_all(c == 0, TRUE);
1586 	count = 0;
1587     }
1588 }
1589 
1590 /*
1591  * Convert "c" plus "modifiers" to merge the effect of modifyOtherKeys into the
1592  * character.
1593  */
1594     int
merge_modifyOtherKeys(int c_arg,int * modifiers)1595 merge_modifyOtherKeys(int c_arg, int *modifiers)
1596 {
1597     int c = c_arg;
1598 
1599     if (*modifiers & MOD_MASK_CTRL)
1600     {
1601 	if ((c >= '`' && c <= 0x7f) || (c >= '@' && c <= '_'))
1602 	    c &= 0x1f;
1603 	else if (c == '6')
1604 	    // CTRL-6 is equivalent to CTRL-^
1605 	    c = 0x1e;
1606 #ifdef FEAT_GUI_GTK
1607 	// These mappings look arbitrary at the first glance, but in fact
1608 	// resemble quite exactly the behaviour of the GTK+ 1.2 GUI on my
1609 	// machine.  The only difference is BS vs. DEL for CTRL-8 (makes
1610 	// more sense and is consistent with usual terminal behaviour).
1611 	else if (c == '2')
1612 	    c = NUL;
1613 	else if (c >= '3' && c <= '7')
1614 	    c = c ^ 0x28;
1615 	else if (c == '8')
1616 	    c = BS;
1617 	else if (c == '?')
1618 	    c = DEL;
1619 #endif
1620 	if (c != c_arg)
1621 	    *modifiers &= ~MOD_MASK_CTRL;
1622     }
1623     if ((*modifiers & (MOD_MASK_META | MOD_MASK_ALT))
1624 	    && c >= 0 && c <= 127)
1625     {
1626 	c += 0x80;
1627 	*modifiers &= ~(MOD_MASK_META|MOD_MASK_ALT);
1628     }
1629     return c;
1630 }
1631 
1632 /*
1633  * Get the next input character.
1634  * Can return a special key or a multi-byte character.
1635  * Can return NUL when called recursively, use safe_vgetc() if that's not
1636  * wanted.
1637  * This translates escaped K_SPECIAL and CSI bytes to a K_SPECIAL or CSI byte.
1638  * Collects the bytes of a multibyte character into the whole character.
1639  * Returns the modifiers in the global "mod_mask".
1640  */
1641     int
vgetc(void)1642 vgetc(void)
1643 {
1644     int		c, c2;
1645     int		n;
1646     char_u	buf[MB_MAXBYTES + 1];
1647     int		i;
1648 
1649 #ifdef FEAT_EVAL
1650     // Do garbage collection when garbagecollect() was called previously and
1651     // we are now at the toplevel.
1652     if (may_garbage_collect && want_garbage_collect)
1653 	garbage_collect(FALSE);
1654 #endif
1655 
1656     /*
1657      * If a character was put back with vungetc, it was already processed.
1658      * Return it directly.
1659      */
1660     if (old_char != -1)
1661     {
1662 	c = old_char;
1663 	old_char = -1;
1664 	mod_mask = old_mod_mask;
1665 	mouse_row = old_mouse_row;
1666 	mouse_col = old_mouse_col;
1667     }
1668     else
1669     {
1670 	mod_mask = 0;
1671 	vgetc_mod_mask = 0;
1672 	vgetc_char = 0;
1673 	last_recorded_len = 0;
1674 
1675 	for (;;)		// this is done twice if there are modifiers
1676 	{
1677 	    int did_inc = FALSE;
1678 
1679 	    if (mod_mask
1680 #if defined(FEAT_XIM) && defined(FEAT_GUI_GTK)
1681 		    || im_is_preediting()
1682 #endif
1683 #if defined(FEAT_PROP_POPUP)
1684 		    || popup_no_mapping()
1685 #endif
1686 		    )
1687 	    {
1688 		// no mapping after modifier has been read
1689 		++no_mapping;
1690 		++allow_keys;
1691 		did_inc = TRUE;	// mod_mask may change value
1692 	    }
1693 	    c = vgetorpeek(TRUE);
1694 	    if (did_inc)
1695 	    {
1696 		--no_mapping;
1697 		--allow_keys;
1698 	    }
1699 
1700 	    // Get two extra bytes for special keys
1701 	    if (c == K_SPECIAL
1702 #ifdef FEAT_GUI
1703 		    || (c == CSI)
1704 #endif
1705 	       )
1706 	    {
1707 		int	    save_allow_keys = allow_keys;
1708 
1709 		++no_mapping;
1710 		allow_keys = 0;		// make sure BS is not found
1711 		c2 = vgetorpeek(TRUE);	// no mapping for these chars
1712 		c = vgetorpeek(TRUE);
1713 		--no_mapping;
1714 		allow_keys = save_allow_keys;
1715 		if (c2 == KS_MODIFIER)
1716 		{
1717 		    mod_mask = c;
1718 		    continue;
1719 		}
1720 		c = TO_SPECIAL(c2, c);
1721 
1722 #if defined(FEAT_GUI_MSWIN) && defined(FEAT_MENU) && defined(FEAT_TEAROFF)
1723 		// Handle K_TEAROFF here, the caller of vgetc() doesn't need to
1724 		// know that a menu was torn off
1725 		if (
1726 # ifdef VIMDLL
1727 		    gui.in_use &&
1728 # endif
1729 		    c == K_TEAROFF)
1730 		{
1731 		    char_u	name[200];
1732 		    int		i;
1733 
1734 		    // get menu path, it ends with a <CR>
1735 		    for (i = 0; (c = vgetorpeek(TRUE)) != '\r'; )
1736 		    {
1737 			name[i] = c;
1738 			if (i < 199)
1739 			    ++i;
1740 		    }
1741 		    name[i] = NUL;
1742 		    gui_make_tearoff(name);
1743 		    continue;
1744 		}
1745 #endif
1746 #if defined(FEAT_GUI) && defined(FEAT_GUI_GTK) && defined(FEAT_MENU)
1747 		// GTK: <F10> normally selects the menu, but it's passed until
1748 		// here to allow mapping it.  Intercept and invoke the GTK
1749 		// behavior if it's not mapped.
1750 		if (c == K_F10 && gui.menubar != NULL)
1751 		{
1752 		    gtk_menu_shell_select_first(
1753 					   GTK_MENU_SHELL(gui.menubar), FALSE);
1754 		    continue;
1755 		}
1756 #endif
1757 #ifdef FEAT_GUI
1758 		// Handle focus event here, so that the caller doesn't need to
1759 		// know about it.  Return K_IGNORE so that we loop once (needed
1760 		// if 'lazyredraw' is set).
1761 		if (c == K_FOCUSGAINED || c == K_FOCUSLOST)
1762 		{
1763 		    ui_focus_change(c == K_FOCUSGAINED);
1764 		    c = K_IGNORE;
1765 		}
1766 
1767 		// Translate K_CSI to CSI.  The special key is only used to
1768 		// avoid it being recognized as the start of a special key.
1769 		if (c == K_CSI)
1770 		    c = CSI;
1771 #endif
1772 	    }
1773 	    // a keypad or special function key was not mapped, use it like
1774 	    // its ASCII equivalent
1775 	    switch (c)
1776 	    {
1777 		case K_KPLUS:	c = '+'; break;
1778 		case K_KMINUS:	c = '-'; break;
1779 		case K_KDIVIDE:	c = '/'; break;
1780 		case K_KMULTIPLY: c = '*'; break;
1781 		case K_KENTER:	c = CAR; break;
1782 		case K_KPOINT:
1783 #ifdef MSWIN
1784 				// Can be either '.' or a ',',
1785 				// depending on the type of keypad.
1786 				c = MapVirtualKey(VK_DECIMAL, 2); break;
1787 #else
1788 				c = '.'; break;
1789 #endif
1790 		case K_K0:	c = '0'; break;
1791 		case K_K1:	c = '1'; break;
1792 		case K_K2:	c = '2'; break;
1793 		case K_K3:	c = '3'; break;
1794 		case K_K4:	c = '4'; break;
1795 		case K_K5:	c = '5'; break;
1796 		case K_K6:	c = '6'; break;
1797 		case K_K7:	c = '7'; break;
1798 		case K_K8:	c = '8'; break;
1799 		case K_K9:	c = '9'; break;
1800 
1801 		case K_XHOME:
1802 		case K_ZHOME:	if (mod_mask == MOD_MASK_SHIFT)
1803 				{
1804 				    c = K_S_HOME;
1805 				    mod_mask = 0;
1806 				}
1807 				else if (mod_mask == MOD_MASK_CTRL)
1808 				{
1809 				    c = K_C_HOME;
1810 				    mod_mask = 0;
1811 				}
1812 				else
1813 				    c = K_HOME;
1814 				break;
1815 		case K_XEND:
1816 		case K_ZEND:	if (mod_mask == MOD_MASK_SHIFT)
1817 				{
1818 				    c = K_S_END;
1819 				    mod_mask = 0;
1820 				}
1821 				else if (mod_mask == MOD_MASK_CTRL)
1822 				{
1823 				    c = K_C_END;
1824 				    mod_mask = 0;
1825 				}
1826 				else
1827 				    c = K_END;
1828 				break;
1829 
1830 		case K_XUP:	c = K_UP; break;
1831 		case K_XDOWN:	c = K_DOWN; break;
1832 		case K_XLEFT:	c = K_LEFT; break;
1833 		case K_XRIGHT:	c = K_RIGHT; break;
1834 	    }
1835 
1836 	    // For a multi-byte character get all the bytes and return the
1837 	    // converted character.
1838 	    // Note: This will loop until enough bytes are received!
1839 	    if (has_mbyte && (n = MB_BYTE2LEN_CHECK(c)) > 1)
1840 	    {
1841 		++no_mapping;
1842 		buf[0] = c;
1843 		for (i = 1; i < n; ++i)
1844 		{
1845 		    buf[i] = vgetorpeek(TRUE);
1846 		    if (buf[i] == K_SPECIAL
1847 #ifdef FEAT_GUI
1848 			    || (buf[i] == CSI)
1849 #endif
1850 			    )
1851 		    {
1852 			// Must be a K_SPECIAL - KS_SPECIAL - KE_FILLER
1853 			// sequence, which represents a K_SPECIAL (0x80),
1854 			// or a CSI - KS_EXTRA - KE_CSI sequence, which
1855 			// represents a CSI (0x9B),
1856 			// or a K_SPECIAL - KS_EXTRA - KE_CSI, which is CSI
1857 			// too.
1858 			c = vgetorpeek(TRUE);
1859 			if (vgetorpeek(TRUE) == KE_CSI && c == KS_EXTRA)
1860 			    buf[i] = CSI;
1861 		    }
1862 		}
1863 		--no_mapping;
1864 		c = (*mb_ptr2char)(buf);
1865 	    }
1866 
1867 	    if (vgetc_char == 0)
1868 	    {
1869 		vgetc_mod_mask = mod_mask;
1870 		vgetc_char = c;
1871 	    }
1872 
1873 	    break;
1874 	}
1875     }
1876 
1877 #ifdef FEAT_EVAL
1878     /*
1879      * In the main loop "may_garbage_collect" can be set to do garbage
1880      * collection in the first next vgetc().  It's disabled after that to
1881      * avoid internally used Lists and Dicts to be freed.
1882      */
1883     may_garbage_collect = FALSE;
1884 #endif
1885 
1886 #ifdef FEAT_BEVAL_TERM
1887     if (c != K_MOUSEMOVE && c != K_IGNORE && c != K_CURSORHOLD)
1888     {
1889 	// Don't trigger 'balloonexpr' unless only the mouse was moved.
1890 	bevalexpr_due_set = FALSE;
1891 	ui_remove_balloon();
1892     }
1893 #endif
1894 #ifdef FEAT_PROP_POPUP
1895     // Only filter keys that do not come from ":normal".  Keys from feedkeys()
1896     // are filtered.
1897     if ((!ex_normal_busy || in_feedkeys) && popup_do_filter(c))
1898     {
1899 	if (c == Ctrl_C)
1900 	    got_int = FALSE;  // avoid looping
1901 	c = K_IGNORE;
1902     }
1903 #endif
1904 
1905     // Need to process the character before we know it's safe to do something
1906     // else.
1907     if (c != K_IGNORE)
1908 	state_no_longer_safe("key typed");
1909 
1910     return c;
1911 }
1912 
1913 /*
1914  * Like vgetc(), but never return a NUL when called recursively, get a key
1915  * directly from the user (ignoring typeahead).
1916  */
1917     int
safe_vgetc(void)1918 safe_vgetc(void)
1919 {
1920     int	c;
1921 
1922     c = vgetc();
1923     if (c == NUL)
1924 	c = get_keystroke();
1925     return c;
1926 }
1927 
1928 /*
1929  * Like safe_vgetc(), but loop to handle K_IGNORE.
1930  * Also ignore scrollbar events.
1931  */
1932     int
plain_vgetc(void)1933 plain_vgetc(void)
1934 {
1935     int c;
1936 
1937     do
1938 	c = safe_vgetc();
1939     while (c == K_IGNORE
1940 	    || c == K_VER_SCROLLBAR || c == K_HOR_SCROLLBAR
1941 	    || c == K_MOUSEMOVE);
1942 
1943     if (c == K_PS)
1944 	// Only handle the first pasted character.  Drop the rest, since we
1945 	// don't know what to do with it.
1946 	c = bracketed_paste(PASTE_ONE_CHAR, FALSE, NULL);
1947 
1948     return c;
1949 }
1950 
1951 /*
1952  * Check if a character is available, such that vgetc() will not block.
1953  * If the next character is a special character or multi-byte, the returned
1954  * character is not valid!.
1955  * Returns NUL if no character is available.
1956  */
1957     int
vpeekc(void)1958 vpeekc(void)
1959 {
1960     if (old_char != -1)
1961 	return old_char;
1962     return vgetorpeek(FALSE);
1963 }
1964 
1965 #if defined(FEAT_TERMRESPONSE) || defined(FEAT_TERMINAL) || defined(PROTO)
1966 /*
1967  * Like vpeekc(), but don't allow mapping.  Do allow checking for terminal
1968  * codes.
1969  */
1970     int
vpeekc_nomap(void)1971 vpeekc_nomap(void)
1972 {
1973     int		c;
1974 
1975     ++no_mapping;
1976     ++allow_keys;
1977     c = vpeekc();
1978     --no_mapping;
1979     --allow_keys;
1980     return c;
1981 }
1982 #endif
1983 
1984 /*
1985  * Check if any character is available, also half an escape sequence.
1986  * Trick: when no typeahead found, but there is something in the typeahead
1987  * buffer, it must be an ESC that is recognized as the start of a key code.
1988  */
1989     int
vpeekc_any(void)1990 vpeekc_any(void)
1991 {
1992     int		c;
1993 
1994     c = vpeekc();
1995     if (c == NUL && typebuf.tb_len > 0)
1996 	c = ESC;
1997     return c;
1998 }
1999 
2000 /*
2001  * Call vpeekc() without causing anything to be mapped.
2002  * Return TRUE if a character is available, FALSE otherwise.
2003  */
2004     int
char_avail(void)2005 char_avail(void)
2006 {
2007     int	    retval;
2008 
2009 #ifdef FEAT_EVAL
2010     // When test_override("char_avail", 1) was called pretend there is no
2011     // typeahead.
2012     if (disable_char_avail_for_testing)
2013 	return FALSE;
2014 #endif
2015     ++no_mapping;
2016     retval = vpeekc();
2017     --no_mapping;
2018     return (retval != NUL);
2019 }
2020 
2021 #if defined(FEAT_EVAL) || defined(PROTO)
2022 /*
2023  * "getchar()" and "getcharstr()" functions
2024  */
2025     static void
getchar_common(typval_T * argvars,typval_T * rettv)2026 getchar_common(typval_T *argvars, typval_T *rettv)
2027 {
2028     varnumber_T		n;
2029     int			error = FALSE;
2030 
2031     if (in_vim9script() && check_for_opt_bool_arg(argvars, 0) == FAIL)
2032 	return;
2033 
2034 #ifdef MESSAGE_QUEUE
2035     // vpeekc() used to check for messages, but that caused problems, invoking
2036     // a callback where it was not expected.  Some plugins use getchar(1) in a
2037     // loop to await a message, therefore make sure we check for messages here.
2038     parse_queued_messages();
2039 #endif
2040 
2041     // Position the cursor.  Needed after a message that ends in a space.
2042     windgoto(msg_row, msg_col);
2043 
2044     ++no_mapping;
2045     ++allow_keys;
2046     for (;;)
2047     {
2048 	if (argvars[0].v_type == VAR_UNKNOWN)
2049 	    // getchar(): blocking wait.
2050 	    n = plain_vgetc();
2051 	else if (tv_get_bool_chk(&argvars[0], &error))
2052 	    // getchar(1): only check if char avail
2053 	    n = vpeekc_any();
2054 	else if (error || vpeekc_any() == NUL)
2055 	    // illegal argument or getchar(0) and no char avail: return zero
2056 	    n = 0;
2057 	else
2058 	    // getchar(0) and char avail() != NUL: get a character.
2059 	    // Note that vpeekc_any() returns K_SPECIAL for K_IGNORE.
2060 	    n = safe_vgetc();
2061 
2062 	if (n == K_IGNORE || n == K_MOUSEMOVE
2063 		|| n == K_VER_SCROLLBAR || n == K_HOR_SCROLLBAR)
2064 	    continue;
2065 	break;
2066     }
2067     --no_mapping;
2068     --allow_keys;
2069 
2070     set_vim_var_nr(VV_MOUSE_WIN, 0);
2071     set_vim_var_nr(VV_MOUSE_WINID, 0);
2072     set_vim_var_nr(VV_MOUSE_LNUM, 0);
2073     set_vim_var_nr(VV_MOUSE_COL, 0);
2074 
2075     rettv->vval.v_number = n;
2076     if (IS_SPECIAL(n) || mod_mask != 0)
2077     {
2078 	char_u		temp[10];   // modifier: 3, mbyte-char: 6, NUL: 1
2079 	int		i = 0;
2080 
2081 	// Turn a special key into three bytes, plus modifier.
2082 	if (mod_mask != 0)
2083 	{
2084 	    temp[i++] = K_SPECIAL;
2085 	    temp[i++] = KS_MODIFIER;
2086 	    temp[i++] = mod_mask;
2087 	}
2088 	if (IS_SPECIAL(n))
2089 	{
2090 	    temp[i++] = K_SPECIAL;
2091 	    temp[i++] = K_SECOND(n);
2092 	    temp[i++] = K_THIRD(n);
2093 	}
2094 	else if (has_mbyte)
2095 	    i += (*mb_char2bytes)(n, temp + i);
2096 	else
2097 	    temp[i++] = n;
2098 	temp[i++] = NUL;
2099 	rettv->v_type = VAR_STRING;
2100 	rettv->vval.v_string = vim_strsave(temp);
2101 
2102 	if (is_mouse_key(n))
2103 	{
2104 	    int		row = mouse_row;
2105 	    int		col = mouse_col;
2106 	    win_T	*win;
2107 	    linenr_T	lnum;
2108 	    win_T	*wp;
2109 	    int		winnr = 1;
2110 
2111 	    if (row >= 0 && col >= 0)
2112 	    {
2113 		// Find the window at the mouse coordinates and compute the
2114 		// text position.
2115 		win = mouse_find_win(&row, &col, FIND_POPUP);
2116 		if (win == NULL)
2117 		    return;
2118 		(void)mouse_comp_pos(win, &row, &col, &lnum, NULL);
2119 #ifdef FEAT_PROP_POPUP
2120 		if (WIN_IS_POPUP(win))
2121 		    winnr = 0;
2122 		else
2123 #endif
2124 		    for (wp = firstwin; wp != win && wp != NULL;
2125 							       wp = wp->w_next)
2126 			++winnr;
2127 		set_vim_var_nr(VV_MOUSE_WIN, winnr);
2128 		set_vim_var_nr(VV_MOUSE_WINID, win->w_id);
2129 		set_vim_var_nr(VV_MOUSE_LNUM, lnum);
2130 		set_vim_var_nr(VV_MOUSE_COL, col + 1);
2131 	    }
2132 	}
2133     }
2134 }
2135 
2136 /*
2137  * "getchar()" function
2138  */
2139     void
f_getchar(typval_T * argvars,typval_T * rettv)2140 f_getchar(typval_T *argvars, typval_T *rettv)
2141 {
2142     getchar_common(argvars, rettv);
2143 }
2144 
2145 /*
2146  * "getcharstr()" function
2147  */
2148     void
f_getcharstr(typval_T * argvars,typval_T * rettv)2149 f_getcharstr(typval_T *argvars, typval_T *rettv)
2150 {
2151     getchar_common(argvars, rettv);
2152 
2153     if (rettv->v_type == VAR_NUMBER)
2154     {
2155 	char_u		temp[7];   // mbyte-char: 6, NUL: 1
2156 	varnumber_T	n = rettv->vval.v_number;
2157 	int		i = 0;
2158 
2159 	if (n != 0)
2160 	{
2161 	    if (has_mbyte)
2162 		i += (*mb_char2bytes)(n, temp + i);
2163 	    else
2164 		temp[i++] = n;
2165 	}
2166 	temp[i++] = NUL;
2167 	rettv->v_type = VAR_STRING;
2168 	rettv->vval.v_string = vim_strsave(temp);
2169     }
2170 }
2171 
2172 /*
2173  * "getcharmod()" function
2174  */
2175     void
f_getcharmod(typval_T * argvars UNUSED,typval_T * rettv)2176 f_getcharmod(typval_T *argvars UNUSED, typval_T *rettv)
2177 {
2178     rettv->vval.v_number = mod_mask;
2179 }
2180 #endif // FEAT_EVAL
2181 
2182 #if defined(MESSAGE_QUEUE) || defined(PROTO)
2183 # define MAX_REPEAT_PARSE 8
2184 
2185 /*
2186  * Process messages that have been queued for netbeans or clientserver.
2187  * Also check if any jobs have ended.
2188  * These functions can call arbitrary Vim script and should only be called when
2189  * it is safe to do so.
2190  */
2191     void
parse_queued_messages(void)2192 parse_queued_messages(void)
2193 {
2194     int	    old_curwin_id;
2195     int	    old_curbuf_fnum;
2196     int	    i;
2197     int	    save_may_garbage_collect = may_garbage_collect;
2198     static int entered = 0;
2199     int	    was_safe = get_was_safe_state();
2200 
2201     // Do not handle messages while redrawing, because it may cause buffers to
2202     // change or be wiped while they are being redrawn.
2203     // Also bail out when parsing messages was explicitly disabled.
2204     if (updating_screen || dont_parse_messages)
2205 	return;
2206 
2207     // If memory allocation fails during startup we'll exit but curbuf or
2208     // curwin could be NULL.
2209     if (curbuf == NULL || curwin == NULL)
2210        return;
2211 
2212     old_curbuf_fnum = curbuf->b_fnum;
2213     old_curwin_id = curwin->w_id;
2214 
2215     ++entered;
2216 
2217     // may_garbage_collect is set in main_loop() to do garbage collection when
2218     // blocking to wait on a character.  We don't want that while parsing
2219     // messages, a callback may invoke vgetc() while lists and dicts are in use
2220     // in the call stack.
2221     may_garbage_collect = FALSE;
2222 
2223     // Loop when a job ended, but don't keep looping forever.
2224     for (i = 0; i < MAX_REPEAT_PARSE; ++i)
2225     {
2226 	// For Win32 mch_breakcheck() does not check for input, do it here.
2227 # if (defined(MSWIN) || defined(__HAIKU__)) && defined(FEAT_JOB_CHANNEL)
2228 	channel_handle_events(FALSE);
2229 # endif
2230 
2231 # ifdef FEAT_NETBEANS_INTG
2232 	// Process the queued netbeans messages.
2233 	netbeans_parse_messages();
2234 # endif
2235 # ifdef FEAT_JOB_CHANNEL
2236 	// Write any buffer lines still to be written.
2237 	channel_write_any_lines();
2238 
2239 	// Process the messages queued on channels.
2240 	channel_parse_messages();
2241 # endif
2242 # if defined(FEAT_CLIENTSERVER) && defined(FEAT_X11)
2243 	// Process the queued clientserver messages.
2244 	server_parse_messages();
2245 # endif
2246 # ifdef FEAT_JOB_CHANNEL
2247 	// Check if any jobs have ended.  If so, repeat the above to handle
2248 	// changes, e.g. stdin may have been closed.
2249 	if (job_check_ended())
2250 	    continue;
2251 # endif
2252 # ifdef FEAT_TERMINAL
2253 	free_unused_terminals();
2254 # endif
2255 # ifdef FEAT_SOUND_CANBERRA
2256 	if (has_sound_callback_in_queue())
2257 	    invoke_sound_callback();
2258 # endif
2259 #ifdef SIGUSR1
2260 	if (got_sigusr1)
2261 	{
2262 	    apply_autocmds(EVENT_SIGUSR1, NULL, NULL, FALSE, curbuf);
2263 	    got_sigusr1 = FALSE;
2264 	}
2265 #endif
2266 	break;
2267     }
2268 
2269     // When not nested we'll go back to waiting for a typed character.  If it
2270     // was safe before then this triggers a SafeStateAgain autocommand event.
2271     if (entered == 1 && was_safe)
2272 	may_trigger_safestateagain();
2273 
2274     may_garbage_collect = save_may_garbage_collect;
2275 
2276     // If the current window or buffer changed we need to bail out of the
2277     // waiting loop.  E.g. when a job exit callback closes the terminal window.
2278     if (curwin->w_id != old_curwin_id || curbuf->b_fnum != old_curbuf_fnum)
2279 	ins_char_typebuf(K_IGNORE, 0);
2280 
2281     --entered;
2282 }
2283 #endif
2284 
2285 
2286 typedef enum {
2287     map_result_fail,    // failed, break loop
2288     map_result_get,     // get a character from typeahead
2289     map_result_retry,   // try to map again
2290     map_result_nomatch  // no matching mapping, get char
2291 } map_result_T;
2292 
2293 /*
2294  * Check if the bytes at the start of the typeahead buffer are a character used
2295  * in CTRL-X mode.  This includes the form with a CTRL modifier.
2296  */
2297     static int
at_ctrl_x_key(void)2298 at_ctrl_x_key(void)
2299 {
2300     char_u  *p = typebuf.tb_buf + typebuf.tb_off;
2301     int	    c = *p;
2302 
2303     if (typebuf.tb_len > 3
2304 	    && c == K_SPECIAL
2305 	    && p[1] == KS_MODIFIER
2306 	    && (p[2] & MOD_MASK_CTRL))
2307 	c = p[3] & 0x1f;
2308     return vim_is_ctrl_x_key(c);
2309 }
2310 
2311 /*
2312  * Check if typebuf.tb_buf[] contains a modifier plus key that can be changed
2313  * into just a key, apply that.
2314  * Check from typebuf.tb_buf[typebuf.tb_off] to typebuf.tb_buf[typebuf.tb_off
2315  * + "max_offset"].
2316  * Return the length of the replaced bytes, zero if nothing changed.
2317  */
2318     static int
check_simplify_modifier(int max_offset)2319 check_simplify_modifier(int max_offset)
2320 {
2321     int		offset;
2322     char_u	*tp;
2323 
2324     for (offset = 0; offset < max_offset; ++offset)
2325     {
2326 	if (offset + 3 >= typebuf.tb_len)
2327 	    break;
2328 	tp = typebuf.tb_buf + typebuf.tb_off + offset;
2329 	if (tp[0] == K_SPECIAL && tp[1] == KS_MODIFIER)
2330 	{
2331 	    // A modifier was not used for a mapping, apply it to ASCII keys.
2332 	    // Shift would already have been applied.
2333 	    int modifier = tp[2];
2334 	    int	c = tp[3];
2335 	    int new_c = merge_modifyOtherKeys(c, &modifier);
2336 
2337 	    if (new_c != c)
2338 	    {
2339 		char_u	new_string[MB_MAXBYTES];
2340 		int	len;
2341 
2342 		if (offset == 0)
2343 		{
2344 		    // At the start: remember the character and mod_mask before
2345 		    // merging, in some cases, e.g. at the hit-return prompt,
2346 		    // they are put back in the typeahead buffer.
2347 		    vgetc_char = c;
2348 		    vgetc_mod_mask = tp[2];
2349 		}
2350 		len = mb_char2bytes(new_c, new_string);
2351 		if (modifier == 0)
2352 		{
2353 		    if (put_string_in_typebuf(offset, 4, new_string, len,
2354 							   NULL, 0, 0) == FAIL)
2355 		    return -1;
2356 		}
2357 		else
2358 		{
2359 		    tp[2] = modifier;
2360 		    if (put_string_in_typebuf(offset + 3, 1, new_string, len,
2361 							   NULL, 0, 0) == FAIL)
2362 		    return -1;
2363 		}
2364 		return len;
2365 	    }
2366 	}
2367     }
2368     return 0;
2369 }
2370 
2371 /*
2372  * Handle mappings in the typeahead buffer.
2373  * - When something was mapped, return map_result_retry for recursive mappings.
2374  * - When nothing mapped and typeahead has a character: return map_result_get.
2375  * - When there is no match yet, return map_result_nomatch, need to get more
2376  *   typeahead.
2377  */
2378     static int
handle_mapping(int * keylenp,int * timedout,int * mapdepth)2379 handle_mapping(
2380 	    int *keylenp,
2381 	    int *timedout,
2382 	    int *mapdepth)
2383 {
2384     mapblock_T	*mp = NULL;
2385     mapblock_T	*mp2;
2386     mapblock_T	*mp_match;
2387     int		mp_match_len = 0;
2388     int		max_mlen = 0;
2389     int		tb_c1;
2390     int		mlen;
2391 #ifdef FEAT_LANGMAP
2392     int		nolmaplen;
2393 #endif
2394     int		keylen = *keylenp;
2395     int		i;
2396     int		local_State = get_real_state();
2397 
2398     /*
2399      * Check for a mappable key sequence.
2400      * Walk through one maphash[] list until we find an entry that matches.
2401      *
2402      * Don't look for mappings if:
2403      * - no_mapping set: mapping disabled (e.g. for CTRL-V)
2404      * - maphash_valid not set: no mappings present.
2405      * - typebuf.tb_buf[typebuf.tb_off] should not be remapped
2406      * - in insert or cmdline mode and 'paste' option set
2407      * - waiting for "hit return to continue" and CR or SPACE typed
2408      * - waiting for a char with --more--
2409      * - in Ctrl-X mode, and we get a valid char for that mode
2410      */
2411     tb_c1 = typebuf.tb_buf[typebuf.tb_off];
2412     if (no_mapping == 0 && is_maphash_valid()
2413 	    && (no_zero_mapping == 0 || tb_c1 != '0')
2414 	    && (typebuf.tb_maplen == 0
2415 		|| (p_remap
2416 		    && (typebuf.tb_noremap[typebuf.tb_off]
2417 				    & (RM_NONE|RM_ABBR)) == 0))
2418 	    && !(p_paste && (State & (INSERT + CMDLINE)))
2419 	    && !(State == HITRETURN && (tb_c1 == CAR || tb_c1 == ' '))
2420 	    && State != ASKMORE
2421 	    && State != CONFIRM
2422 	    && !((ctrl_x_mode_not_default() && at_ctrl_x_key())
2423 		    || ((compl_cont_status & CONT_LOCAL)
2424 			&& (tb_c1 == Ctrl_N || tb_c1 == Ctrl_P))))
2425     {
2426 #ifdef FEAT_GUI
2427 	if (gui.in_use && tb_c1 == CSI && typebuf.tb_len >= 2
2428 		&& typebuf.tb_buf[typebuf.tb_off + 1] == KS_MODIFIER)
2429 	{
2430 	    // The GUI code sends CSI KS_MODIFIER {flags}, but mappings expect
2431 	    // K_SPECIAL KS_MODIFIER {flags}.
2432 	    tb_c1 = K_SPECIAL;
2433 	}
2434 #endif
2435 #ifdef FEAT_LANGMAP
2436 	if (tb_c1 == K_SPECIAL)
2437 	    nolmaplen = 2;
2438 	else
2439 	{
2440 	    LANGMAP_ADJUST(tb_c1, (State & (CMDLINE | INSERT)) == 0
2441 					    && get_real_state() != SELECTMODE);
2442 	    nolmaplen = 0;
2443 	}
2444 #endif
2445 	// First try buffer-local mappings.
2446 	mp = get_buf_maphash_list(local_State, tb_c1);
2447 	mp2 = get_maphash_list(local_State, tb_c1);
2448 	if (mp == NULL)
2449 	{
2450 	    // There are no buffer-local mappings.
2451 	    mp = mp2;
2452 	    mp2 = NULL;
2453 	}
2454 
2455 	/*
2456 	 * Loop until a partly matching mapping is found or all (local)
2457 	 * mappings have been checked.
2458 	 * The longest full match is remembered in "mp_match".
2459 	 * A full match is only accepted if there is no partly match, so "aa"
2460 	 * and "aaa" can both be mapped.
2461 	 */
2462 	mp_match = NULL;
2463 	mp_match_len = 0;
2464 	for ( ; mp != NULL;
2465 	       mp->m_next == NULL ? (mp = mp2, mp2 = NULL) : (mp = mp->m_next))
2466 	{
2467 	    // Only consider an entry if the first character matches and it is
2468 	    // for the current state.
2469 	    // Skip ":lmap" mappings if keys were mapped.
2470 	    if (mp->m_keys[0] == tb_c1
2471 		    && (mp->m_mode & local_State)
2472 		    && !(mp->m_simplified && seenModifyOtherKeys
2473 						     && typebuf.tb_maplen == 0)
2474 		    && ((mp->m_mode & LANGMAP) == 0 || typebuf.tb_maplen == 0))
2475 	    {
2476 #ifdef FEAT_LANGMAP
2477 		int	nomap = nolmaplen;
2478 		int	c2;
2479 #endif
2480 		// find the match length of this mapping
2481 		for (mlen = 1; mlen < typebuf.tb_len; ++mlen)
2482 		{
2483 #ifdef FEAT_LANGMAP
2484 		    c2 = typebuf.tb_buf[typebuf.tb_off + mlen];
2485 		    if (nomap > 0)
2486 			--nomap;
2487 		    else if (c2 == K_SPECIAL)
2488 			nomap = 2;
2489 		    else
2490 			LANGMAP_ADJUST(c2, TRUE);
2491 		    if (mp->m_keys[mlen] != c2)
2492 #else
2493 		    if (mp->m_keys[mlen] !=
2494 					 typebuf.tb_buf[typebuf.tb_off + mlen])
2495 #endif
2496 			break;
2497 		}
2498 
2499 		// Don't allow mapping the first byte(s) of a multi-byte char.
2500 		// Happens when mapping <M-a> and then changing 'encoding'.
2501 		// Beware that 0x80 is escaped.
2502 		{
2503 		    char_u *p1 = mp->m_keys;
2504 		    char_u *p2 = mb_unescape(&p1);
2505 
2506 		    if (has_mbyte && p2 != NULL
2507 					&& MB_BYTE2LEN(tb_c1) > mb_ptr2len(p2))
2508 			mlen = 0;
2509 		}
2510 
2511 		// Check an entry whether it matches.
2512 		// - Full match: mlen == keylen
2513 		// - Partly match: mlen == typebuf.tb_len
2514 		keylen = mp->m_keylen;
2515 		if (mlen == keylen || (mlen == typebuf.tb_len
2516 						   && typebuf.tb_len < keylen))
2517 		{
2518 		    char_u  *s;
2519 		    int	    n;
2520 
2521 		    // If only script-local mappings are allowed, check if the
2522 		    // mapping starts with K_SNR.
2523 		    s = typebuf.tb_noremap + typebuf.tb_off;
2524 		    if (*s == RM_SCRIPT
2525 			    && (mp->m_keys[0] != K_SPECIAL
2526 				|| mp->m_keys[1] != KS_EXTRA
2527 				|| mp->m_keys[2] != KE_SNR))
2528 			continue;
2529 
2530 		    // If one of the typed keys cannot be remapped, skip the
2531 		    // entry.
2532 		    for (n = mlen; --n >= 0; )
2533 			if (*s++ & (RM_NONE|RM_ABBR))
2534 			    break;
2535 		    if (n >= 0)
2536 			continue;
2537 
2538 		    if (keylen > typebuf.tb_len)
2539 		    {
2540 			if (!*timedout && !(mp_match != NULL
2541 							&& mp_match->m_nowait))
2542 			{
2543 			    // break at a partly match
2544 			    keylen = KEYLEN_PART_MAP;
2545 			    break;
2546 			}
2547 		    }
2548 		    else if (keylen > mp_match_len)
2549 		    {
2550 			// found a longer match
2551 			mp_match = mp;
2552 			mp_match_len = keylen;
2553 		    }
2554 		}
2555 		else
2556 		    // No match; may have to check for termcode at next
2557 		    // character.
2558 		    if (max_mlen < mlen)
2559 			max_mlen = mlen;
2560 	    }
2561 	}
2562 
2563 	// If no partly match found, use the longest full match.
2564 	if (keylen != KEYLEN_PART_MAP)
2565 	{
2566 	    mp = mp_match;
2567 	    keylen = mp_match_len;
2568 	}
2569     }
2570 
2571     /*
2572      * Check for match with 'pastetoggle'
2573      */
2574     if (*p_pt != NUL && mp == NULL && (State & (INSERT|NORMAL)))
2575     {
2576 	for (mlen = 0; mlen < typebuf.tb_len && p_pt[mlen]; ++mlen)
2577 	    if (p_pt[mlen] != typebuf.tb_buf[typebuf.tb_off + mlen])
2578 		    break;
2579 	if (p_pt[mlen] == NUL)	// match
2580 	{
2581 	    // write chars to script file(s)
2582 	    if (mlen > typebuf.tb_maplen)
2583 		gotchars(typebuf.tb_buf + typebuf.tb_off + typebuf.tb_maplen,
2584 						     mlen - typebuf.tb_maplen);
2585 
2586 	    del_typebuf(mlen, 0); // remove the chars
2587 	    set_option_value((char_u *)"paste", (long)!p_paste, NULL, 0);
2588 	    if (!(State & INSERT))
2589 	    {
2590 		msg_col = 0;
2591 		msg_row = Rows - 1;
2592 		msg_clr_eos();		// clear ruler
2593 	    }
2594 	    status_redraw_all();
2595 	    redraw_statuslines();
2596 	    showmode();
2597 	    setcursor();
2598 	    *keylenp = keylen;
2599 	    return map_result_retry;
2600 	}
2601 	// Need more chars for partly match.
2602 	if (mlen == typebuf.tb_len)
2603 	    keylen = KEYLEN_PART_KEY;
2604 	else if (max_mlen < mlen)
2605 	    // no match, may have to check for termcode at next character
2606 	    max_mlen = mlen + 1;
2607     }
2608 
2609     if ((mp == NULL || max_mlen >= mp_match_len) && keylen != KEYLEN_PART_MAP)
2610     {
2611 	int	save_keylen = keylen;
2612 
2613 	/*
2614 	 * When no matching mapping found or found a non-matching mapping that
2615 	 * matches at least what the matching mapping matched:
2616 	 * Check if we have a terminal code, when:
2617 	 * - mapping is allowed,
2618 	 * - keys have not been mapped,
2619 	 * - and not an ESC sequence, not in insert mode or p_ek is on,
2620 	 * - and when not timed out,
2621 	 */
2622 	if ((no_mapping == 0 || allow_keys != 0)
2623 		&& (typebuf.tb_maplen == 0
2624 		    || (p_remap && typebuf.tb_noremap[
2625 						    typebuf.tb_off] == RM_YES))
2626 		&& !*timedout)
2627 	{
2628 	    keylen = check_termcode(max_mlen + 1, NULL, 0, NULL);
2629 
2630 	    // If no termcode matched but 'pastetoggle' matched partially
2631 	    // it's like an incomplete key sequence.
2632 	    if (keylen == 0 && save_keylen == KEYLEN_PART_KEY)
2633 		keylen = KEYLEN_PART_KEY;
2634 
2635 	    // If no termcode matched, try to include the modifier into the
2636 	    // key.  This for when modifyOtherKeys is working.
2637 	    if (keylen == 0 && !no_reduce_keys)
2638 		keylen = check_simplify_modifier(max_mlen + 1);
2639 
2640 	    // When getting a partial match, but the last characters were not
2641 	    // typed, don't wait for a typed character to complete the
2642 	    // termcode.  This helps a lot when a ":normal" command ends in an
2643 	    // ESC.
2644 	    if (keylen < 0 && typebuf.tb_len == typebuf.tb_maplen)
2645 		keylen = 0;
2646 	}
2647 	else
2648 	    keylen = 0;
2649 	if (keylen == 0)	// no matching terminal code
2650 	{
2651 #ifdef AMIGA
2652 	    // check for window bounds report
2653 	    if (typebuf.tb_maplen == 0 && (typebuf.tb_buf[
2654 						typebuf.tb_off] & 0xff) == CSI)
2655 	    {
2656 		char_u *s;
2657 
2658 		for (s = typebuf.tb_buf + typebuf.tb_off + 1;
2659 			   s < typebuf.tb_buf + typebuf.tb_off + typebuf.tb_len
2660 		   && (VIM_ISDIGIT(*s) || *s == ';' || *s == ' ');
2661 			++s)
2662 		    ;
2663 		if (*s == 'r' || *s == '|') // found one
2664 		{
2665 		    del_typebuf(
2666 			  (int)(s + 1 - (typebuf.tb_buf + typebuf.tb_off)), 0);
2667 		    // get size and redraw screen
2668 		    shell_resized();
2669 		    *keylenp = keylen;
2670 		    return map_result_retry;
2671 		}
2672 		if (*s == NUL)	    // need more characters
2673 		    keylen = KEYLEN_PART_KEY;
2674 	    }
2675 	    if (keylen >= 0)
2676 #endif
2677 		// When there was a matching mapping and no termcode could be
2678 		// replaced after another one, use that mapping (loop around).
2679 		// If there was no mapping at all use the character from the
2680 		// typeahead buffer right here.
2681 		if (mp == NULL)
2682 		{
2683 		    *keylenp = keylen;
2684 		    return map_result_get;    // got character, break for loop
2685 		}
2686 	}
2687 
2688 	if (keylen > 0)	    // full matching terminal code
2689 	{
2690 #if defined(FEAT_GUI) && defined(FEAT_MENU)
2691 	    if (typebuf.tb_len >= 2
2692 		    && typebuf.tb_buf[typebuf.tb_off] == K_SPECIAL
2693 			      && typebuf.tb_buf[typebuf.tb_off + 1] == KS_MENU)
2694 	    {
2695 		int	idx;
2696 
2697 		// Using a menu may cause a break in undo!  It's like using
2698 		// gotchars(), but without recording or writing to a script
2699 		// file.
2700 		may_sync_undo();
2701 		del_typebuf(3, 0);
2702 		idx = get_menu_index(current_menu, local_State);
2703 		if (idx != MENU_INDEX_INVALID)
2704 		{
2705 		    // In Select mode and a Visual mode menu is used:  Switch
2706 		    // to Visual mode temporarily.  Append K_SELECT to switch
2707 		    // back to Select mode.
2708 		    if (VIsual_active && VIsual_select
2709 					     && (current_menu->modes & VISUAL))
2710 		    {
2711 			VIsual_select = FALSE;
2712 			(void)ins_typebuf(K_SELECT_STRING,
2713 						   REMAP_NONE, 0, TRUE, FALSE);
2714 		    }
2715 		    ins_typebuf(current_menu->strings[idx],
2716 				current_menu->noremap[idx],
2717 				0, TRUE, current_menu->silent[idx]);
2718 		}
2719 	    }
2720 #endif // FEAT_GUI && FEAT_MENU
2721 	    *keylenp = keylen;
2722 	    return map_result_retry;	// try mapping again
2723 	}
2724 
2725 	// Partial match: get some more characters.  When a matching mapping
2726 	// was found use that one.
2727 	if (mp == NULL || keylen < 0)
2728 	    keylen = KEYLEN_PART_KEY;
2729 	else
2730 	    keylen = mp_match_len;
2731     }
2732 
2733     /*
2734      * complete match
2735      */
2736     if (keylen >= 0 && keylen <= typebuf.tb_len)
2737     {
2738 	char_u *map_str;
2739 
2740 #ifdef FEAT_EVAL
2741 	int	save_m_expr;
2742 	int	save_m_noremap;
2743 	int	save_m_silent;
2744 	char_u	*save_m_keys;
2745 	char_u	*save_m_str;
2746 #else
2747 # define save_m_noremap mp->m_noremap
2748 # define save_m_silent mp->m_silent
2749 #endif
2750 
2751 	// write chars to script file(s)
2752 	if (keylen > typebuf.tb_maplen)
2753 	    gotchars(typebuf.tb_buf + typebuf.tb_off + typebuf.tb_maplen,
2754 						   keylen - typebuf.tb_maplen);
2755 
2756 	cmd_silent = (typebuf.tb_silent > 0);
2757 	del_typebuf(keylen, 0);	// remove the mapped keys
2758 
2759 	/*
2760 	 * Put the replacement string in front of mapstr.
2761 	 * The depth check catches ":map x y" and ":map y x".
2762 	 */
2763 	if (++*mapdepth >= p_mmd)
2764 	{
2765 	    emsg(_("E223: recursive mapping"));
2766 	    if (State & CMDLINE)
2767 		redrawcmdline();
2768 	    else
2769 		setcursor();
2770 	    flush_buffers(FLUSH_MINIMAL);
2771 	    *mapdepth = 0;	// for next one
2772 	    *keylenp = keylen;
2773 	    return map_result_fail;
2774 	}
2775 
2776 	/*
2777 	 * In Select mode and a Visual mode mapping is used: Switch to Visual
2778 	 * mode temporarily.  Append K_SELECT to switch back to Select mode.
2779 	 */
2780 	if (VIsual_active && VIsual_select && (mp->m_mode & VISUAL))
2781 	{
2782 	    VIsual_select = FALSE;
2783 	    (void)ins_typebuf(K_SELECT_STRING, REMAP_NONE, 0, TRUE, FALSE);
2784 	}
2785 
2786 #ifdef FEAT_EVAL
2787 	// Copy the values from *mp that are used, because evaluating the
2788 	// expression may invoke a function that redefines the mapping, thereby
2789 	// making *mp invalid.
2790 	save_m_expr = mp->m_expr;
2791 	save_m_noremap = mp->m_noremap;
2792 	save_m_silent = mp->m_silent;
2793 	save_m_keys = NULL;  // only saved when needed
2794 	save_m_str = NULL;  // only saved when needed
2795 
2796 	/*
2797 	 * Handle ":map <expr>": evaluate the {rhs} as an expression.  Also
2798 	 * save and restore the command line for "normal :".
2799 	 */
2800 	if (mp->m_expr)
2801 	{
2802 	    int save_vgetc_busy = vgetc_busy;
2803 	    int save_may_garbage_collect = may_garbage_collect;
2804 	    int was_screen_col = screen_cur_col;
2805 	    int was_screen_row = screen_cur_row;
2806 
2807 	    vgetc_busy = 0;
2808 	    may_garbage_collect = FALSE;
2809 
2810 	    save_m_keys = vim_strsave(mp->m_keys);
2811 	    save_m_str = vim_strsave(mp->m_str);
2812 	    map_str = eval_map_expr(save_m_str, NUL);
2813 
2814 	    // The mapping may do anything, but we expect it to take care of
2815 	    // redrawing.  Do put the cursor back where it was.
2816 	    windgoto(was_screen_row, was_screen_col);
2817 	    out_flush();
2818 
2819 	    vgetc_busy = save_vgetc_busy;
2820 	    may_garbage_collect = save_may_garbage_collect;
2821 	}
2822 	else
2823 #endif
2824 	    map_str = mp->m_str;
2825 
2826 	/*
2827 	 * Insert the 'to' part in the typebuf.tb_buf.
2828 	 * If 'from' field is the same as the start of the 'to' field, don't
2829 	 * remap the first character (but do allow abbreviations).
2830 	 * If m_noremap is set, don't remap the whole 'to' part.
2831 	 */
2832 	if (map_str == NULL)
2833 	    i = FAIL;
2834 	else
2835 	{
2836 	    int noremap;
2837 
2838 	    if (save_m_noremap != REMAP_YES)
2839 		noremap = save_m_noremap;
2840 	    else if (
2841 #ifdef FEAT_EVAL
2842 		STRNCMP(map_str, save_m_keys != NULL ? save_m_keys : mp->m_keys,
2843 								(size_t)keylen)
2844 #else
2845 		STRNCMP(map_str, mp->m_keys, (size_t)keylen)
2846 #endif
2847 		   != 0)
2848 		noremap = REMAP_YES;
2849 	    else
2850 		noremap = REMAP_SKIP;
2851 	    i = ins_typebuf(map_str, noremap,
2852 					 0, TRUE, cmd_silent || save_m_silent);
2853 #ifdef FEAT_EVAL
2854 	    if (save_m_expr)
2855 		vim_free(map_str);
2856 #endif
2857 	}
2858 #ifdef FEAT_EVAL
2859 	vim_free(save_m_keys);
2860 	vim_free(save_m_str);
2861 #endif
2862 	*keylenp = keylen;
2863 	if (i == FAIL)
2864 	    return map_result_fail;
2865 	return map_result_retry;
2866     }
2867 
2868     *keylenp = keylen;
2869     return map_result_nomatch;
2870 }
2871 
2872 /*
2873  * unget one character (can only be done once!)
2874  */
2875     void
vungetc(int c)2876 vungetc(int c)
2877 {
2878     old_char = c;
2879     old_mod_mask = mod_mask;
2880     old_mouse_row = mouse_row;
2881     old_mouse_col = mouse_col;
2882 }
2883 
2884 /*
2885  * Get a byte:
2886  * 1. from the stuffbuffer
2887  *	This is used for abbreviated commands like "D" -> "d$".
2888  *	Also used to redo a command for ".".
2889  * 2. from the typeahead buffer
2890  *	Stores text obtained previously but not used yet.
2891  *	Also stores the result of mappings.
2892  *	Also used for the ":normal" command.
2893  * 3. from the user
2894  *	This may do a blocking wait if "advance" is TRUE.
2895  *
2896  * if "advance" is TRUE (vgetc()):
2897  *	Really get the character.
2898  *	KeyTyped is set to TRUE in the case the user typed the key.
2899  *	KeyStuffed is TRUE if the character comes from the stuff buffer.
2900  * if "advance" is FALSE (vpeekc()):
2901  *	Just look whether there is a character available.
2902  *	Return NUL if not.
2903  *
2904  * When "no_mapping" is zero, checks for mappings in the current mode.
2905  * Only returns one byte (of a multi-byte character).
2906  * K_SPECIAL and CSI may be escaped, need to get two more bytes then.
2907  */
2908     static int
vgetorpeek(int advance)2909 vgetorpeek(int advance)
2910 {
2911     int		c, c1;
2912     int		timedout = FALSE;	// waited for more than 1 second
2913 					// for mapping to complete
2914     int		mapdepth = 0;		// check for recursive mapping
2915     int		mode_deleted = FALSE;   // set when mode has been deleted
2916 #ifdef FEAT_CMDL_INFO
2917     int		new_wcol, new_wrow;
2918 #endif
2919 #ifdef FEAT_GUI
2920     int		shape_changed = FALSE;  // adjusted cursor shape
2921 #endif
2922     int		n;
2923     int		old_wcol, old_wrow;
2924     int		wait_tb_len;
2925 
2926     /*
2927      * This function doesn't work very well when called recursively.  This may
2928      * happen though, because of:
2929      * 1. The call to add_to_showcmd().	char_avail() is then used to check if
2930      * there is a character available, which calls this function.  In that
2931      * case we must return NUL, to indicate no character is available.
2932      * 2. A GUI callback function writes to the screen, causing a
2933      * wait_return().
2934      * Using ":normal" can also do this, but it saves the typeahead buffer,
2935      * thus it should be OK.  But don't get a key from the user then.
2936      */
2937     if (vgetc_busy > 0 && ex_normal_busy == 0)
2938 	return NUL;
2939 
2940     ++vgetc_busy;
2941 
2942     if (advance)
2943 	KeyStuffed = FALSE;
2944 
2945     init_typebuf();
2946     start_stuff();
2947     if (advance && typebuf.tb_maplen == 0)
2948 	reg_executing = 0;
2949     do
2950     {
2951 /*
2952  * get a character: 1. from the stuffbuffer
2953  */
2954 	if (typeahead_char != 0)
2955 	{
2956 	    c = typeahead_char;
2957 	    if (advance)
2958 		typeahead_char = 0;
2959 	}
2960 	else
2961 	    c = read_readbuffers(advance);
2962 	if (c != NUL && !got_int)
2963 	{
2964 	    if (advance)
2965 	    {
2966 		// KeyTyped = FALSE;  When the command that stuffed something
2967 		// was typed, behave like the stuffed command was typed.
2968 		// needed for CTRL-W CTRL-] to open a fold, for example.
2969 		KeyStuffed = TRUE;
2970 	    }
2971 	    if (typebuf.tb_no_abbr_cnt == 0)
2972 		typebuf.tb_no_abbr_cnt = 1;	// no abbreviations now
2973 	}
2974 	else
2975 	{
2976 	    /*
2977 	     * Loop until we either find a matching mapped key, or we
2978 	     * are sure that it is not a mapped key.
2979 	     * If a mapped key sequence is found we go back to the start to
2980 	     * try re-mapping.
2981 	     */
2982 	    for (;;)
2983 	    {
2984 		long	wait_time;
2985 		int	keylen = 0;
2986 #ifdef FEAT_CMDL_INFO
2987 		int	showcmd_idx;
2988 #endif
2989 		/*
2990 		 * ui_breakcheck() is slow, don't use it too often when
2991 		 * inside a mapping.  But call it each time for typed
2992 		 * characters.
2993 		 */
2994 		if (typebuf.tb_maplen)
2995 		    line_breakcheck();
2996 		else
2997 		    ui_breakcheck();		// check for CTRL-C
2998 		if (got_int)
2999 		{
3000 		    // flush all input
3001 		    c = inchar(typebuf.tb_buf, typebuf.tb_buflen - 1, 0L);
3002 
3003 		    /*
3004 		     * If inchar() returns TRUE (script file was active) or we
3005 		     * are inside a mapping, get out of Insert mode.
3006 		     * Otherwise we behave like having gotten a CTRL-C.
3007 		     * As a result typing CTRL-C in insert mode will
3008 		     * really insert a CTRL-C.
3009 		     */
3010 		    if ((c || typebuf.tb_maplen)
3011 					      && (State & (INSERT + CMDLINE)))
3012 			c = ESC;
3013 		    else
3014 			c = Ctrl_C;
3015 		    flush_buffers(FLUSH_INPUT);	// flush all typeahead
3016 
3017 		    if (advance)
3018 		    {
3019 			// Also record this character, it might be needed to
3020 			// get out of Insert mode.
3021 			*typebuf.tb_buf = c;
3022 			gotchars(typebuf.tb_buf, 1);
3023 		    }
3024 		    cmd_silent = FALSE;
3025 
3026 		    break;
3027 		}
3028 		else if (typebuf.tb_len > 0)
3029 		{
3030 		    /*
3031 		     * Check for a mapping in "typebuf".
3032 		     */
3033 		    map_result_T result = handle_mapping(
3034 						&keylen, &timedout, &mapdepth);
3035 
3036 		    if (result == map_result_retry)
3037 			// try mapping again
3038 			continue;
3039 		    if (result == map_result_fail)
3040 		    {
3041 			// failed, use the outer loop
3042 			c = -1;
3043 			break;
3044 		    }
3045 		    if (result == map_result_get)
3046 		    {
3047 /*
3048  * get a character: 2. from the typeahead buffer
3049  */
3050 			c = typebuf.tb_buf[typebuf.tb_off];
3051 			if (advance)	// remove chars from tb_buf
3052 			{
3053 			    cmd_silent = (typebuf.tb_silent > 0);
3054 			    if (typebuf.tb_maplen > 0)
3055 				KeyTyped = FALSE;
3056 			    else
3057 			    {
3058 				KeyTyped = TRUE;
3059 				// write char to script file(s)
3060 				gotchars(typebuf.tb_buf
3061 						 + typebuf.tb_off, 1);
3062 			    }
3063 			    KeyNoremap = typebuf.tb_noremap[
3064 						      typebuf.tb_off];
3065 			    del_typebuf(1, 0);
3066 			}
3067 			break;
3068 		    }
3069 
3070 		    // not enough characters, get more
3071 		}
3072 
3073 /*
3074  * get a character: 3. from the user - handle <Esc> in Insert mode
3075  */
3076 		/*
3077 		 * Special case: if we get an <ESC> in insert mode and there
3078 		 * are no more characters at once, we pretend to go out of
3079 		 * insert mode.  This prevents the one second delay after
3080 		 * typing an <ESC>.  If we get something after all, we may
3081 		 * have to redisplay the mode. That the cursor is in the wrong
3082 		 * place does not matter.
3083 		 */
3084 		c = 0;
3085 #ifdef FEAT_CMDL_INFO
3086 		new_wcol = curwin->w_wcol;
3087 		new_wrow = curwin->w_wrow;
3088 #endif
3089 		if (	   advance
3090 			&& typebuf.tb_len == 1
3091 			&& typebuf.tb_buf[typebuf.tb_off] == ESC
3092 			&& !no_mapping
3093 			&& ex_normal_busy == 0
3094 			&& typebuf.tb_maplen == 0
3095 			&& (State & INSERT)
3096 			&& (p_timeout
3097 			    || (keylen == KEYLEN_PART_KEY && p_ttimeout))
3098 			&& (c = inchar(typebuf.tb_buf + typebuf.tb_off
3099 					       + typebuf.tb_len, 3, 25L)) == 0)
3100 		{
3101 		    colnr_T	col = 0, vcol;
3102 		    char_u	*ptr;
3103 
3104 		    if (mode_displayed)
3105 		    {
3106 			unshowmode(TRUE);
3107 			mode_deleted = TRUE;
3108 		    }
3109 #ifdef FEAT_GUI
3110 		    // may show a different cursor shape
3111 		    if (gui.in_use && State != NORMAL && !cmd_silent)
3112 		    {
3113 			int	    save_State;
3114 
3115 			save_State = State;
3116 			State = NORMAL;
3117 			gui_update_cursor(TRUE, FALSE);
3118 			State = save_State;
3119 			shape_changed = TRUE;
3120 		    }
3121 #endif
3122 		    validate_cursor();
3123 		    old_wcol = curwin->w_wcol;
3124 		    old_wrow = curwin->w_wrow;
3125 
3126 		    // move cursor left, if possible
3127 		    if (curwin->w_cursor.col != 0)
3128 		    {
3129 			if (curwin->w_wcol > 0)
3130 			{
3131 			    if (did_ai)
3132 			    {
3133 				/*
3134 				 * We are expecting to truncate the trailing
3135 				 * white-space, so find the last non-white
3136 				 * character -- webb
3137 				 */
3138 				col = vcol = curwin->w_wcol = 0;
3139 				ptr = ml_get_curline();
3140 				while (col < curwin->w_cursor.col)
3141 				{
3142 				    if (!VIM_ISWHITE(ptr[col]))
3143 					curwin->w_wcol = vcol;
3144 				    vcol += lbr_chartabsize(ptr, ptr + col,
3145 							       vcol);
3146 				    if (has_mbyte)
3147 					col += (*mb_ptr2len)(ptr + col);
3148 				    else
3149 					++col;
3150 				}
3151 				curwin->w_wrow = curwin->w_cline_row
3152 					   + curwin->w_wcol / curwin->w_width;
3153 				curwin->w_wcol %= curwin->w_width;
3154 				curwin->w_wcol += curwin_col_off();
3155 				col = 0;	// no correction needed
3156 			    }
3157 			    else
3158 			    {
3159 				--curwin->w_wcol;
3160 				col = curwin->w_cursor.col - 1;
3161 			    }
3162 			}
3163 			else if (curwin->w_p_wrap && curwin->w_wrow)
3164 			{
3165 			    --curwin->w_wrow;
3166 			    curwin->w_wcol = curwin->w_width - 1;
3167 			    col = curwin->w_cursor.col - 1;
3168 			}
3169 			if (has_mbyte && col > 0 && curwin->w_wcol > 0)
3170 			{
3171 			    // Correct when the cursor is on the right halve
3172 			    // of a double-wide character.
3173 			    ptr = ml_get_curline();
3174 			    col -= (*mb_head_off)(ptr, ptr + col);
3175 			    if ((*mb_ptr2cells)(ptr + col) > 1)
3176 				--curwin->w_wcol;
3177 			}
3178 		    }
3179 		    setcursor();
3180 		    out_flush();
3181 #ifdef FEAT_CMDL_INFO
3182 		    new_wcol = curwin->w_wcol;
3183 		    new_wrow = curwin->w_wrow;
3184 #endif
3185 		    curwin->w_wcol = old_wcol;
3186 		    curwin->w_wrow = old_wrow;
3187 		}
3188 		if (c < 0)
3189 		    continue;	// end of input script reached
3190 
3191 		// Allow mapping for just typed characters. When we get here c
3192 		// is the number of extra bytes and typebuf.tb_len is 1.
3193 		for (n = 1; n <= c; ++n)
3194 		    typebuf.tb_noremap[typebuf.tb_off + n] = RM_YES;
3195 		typebuf.tb_len += c;
3196 
3197 		// buffer full, don't map
3198 		if (typebuf.tb_len >= typebuf.tb_maplen + MAXMAPLEN)
3199 		{
3200 		    timedout = TRUE;
3201 		    continue;
3202 		}
3203 
3204 		if (ex_normal_busy > 0)
3205 		{
3206 #ifdef FEAT_CMDWIN
3207 		    static int tc = 0;
3208 #endif
3209 
3210 		    // No typeahead left and inside ":normal".  Must return
3211 		    // something to avoid getting stuck.  When an incomplete
3212 		    // mapping is present, behave like it timed out.
3213 		    if (typebuf.tb_len > 0)
3214 		    {
3215 			timedout = TRUE;
3216 			continue;
3217 		    }
3218 
3219 		    // When 'insertmode' is set, ESC just beeps in Insert
3220 		    // mode.  Use CTRL-L to make edit() return.
3221 		    // For the command line only CTRL-C always breaks it.
3222 		    // For the cmdline window: Alternate between ESC and
3223 		    // CTRL-C: ESC for most situations and CTRL-C to close the
3224 		    // cmdline window.
3225 		    if (p_im && (State & INSERT))
3226 			c = Ctrl_L;
3227 #ifdef FEAT_TERMINAL
3228 		    else if (terminal_is_active())
3229 			c = K_CANCEL;
3230 #endif
3231 		    else if ((State & CMDLINE)
3232 #ifdef FEAT_CMDWIN
3233 			    || (cmdwin_type > 0 && tc == ESC)
3234 #endif
3235 			    )
3236 			c = Ctrl_C;
3237 		    else
3238 			c = ESC;
3239 #ifdef FEAT_CMDWIN
3240 		    tc = c;
3241 #endif
3242 		    // return from main_loop()
3243 		    if (pending_exmode_active)
3244 			exmode_active = EXMODE_NORMAL;
3245 
3246 		    break;
3247 		}
3248 
3249 /*
3250  * get a character: 3. from the user - update display
3251  */
3252 		// In insert mode a screen update is skipped when characters
3253 		// are still available.  But when those available characters
3254 		// are part of a mapping, and we are going to do a blocking
3255 		// wait here.  Need to update the screen to display the
3256 		// changed text so far. Also for when 'lazyredraw' is set and
3257 		// redrawing was postponed because there was something in the
3258 		// input buffer (e.g., termresponse).
3259 		if (((State & INSERT) != 0 || p_lz) && (State & CMDLINE) == 0
3260 			  && advance && must_redraw != 0 && !need_wait_return)
3261 		{
3262 		    update_screen(0);
3263 		    setcursor(); // put cursor back where it belongs
3264 		}
3265 
3266 		/*
3267 		 * If we have a partial match (and are going to wait for more
3268 		 * input from the user), show the partially matched characters
3269 		 * to the user with showcmd.
3270 		 */
3271 #ifdef FEAT_CMDL_INFO
3272 		showcmd_idx = 0;
3273 #endif
3274 		c1 = 0;
3275 		if (typebuf.tb_len > 0 && advance && !exmode_active)
3276 		{
3277 		    if (((State & (NORMAL | INSERT)) || State == LANGMAP)
3278 			    && State != HITRETURN)
3279 		    {
3280 			// this looks nice when typing a dead character map
3281 			if (State & INSERT
3282 			    && ptr2cells(typebuf.tb_buf + typebuf.tb_off
3283 						   + typebuf.tb_len - 1) == 1)
3284 			{
3285 			    edit_putchar(typebuf.tb_buf[typebuf.tb_off
3286 						+ typebuf.tb_len - 1], FALSE);
3287 			    setcursor(); // put cursor back where it belongs
3288 			    c1 = 1;
3289 			}
3290 #ifdef FEAT_CMDL_INFO
3291 			// need to use the col and row from above here
3292 			old_wcol = curwin->w_wcol;
3293 			old_wrow = curwin->w_wrow;
3294 			curwin->w_wcol = new_wcol;
3295 			curwin->w_wrow = new_wrow;
3296 			push_showcmd();
3297 			if (typebuf.tb_len > SHOWCMD_COLS)
3298 			    showcmd_idx = typebuf.tb_len - SHOWCMD_COLS;
3299 			while (showcmd_idx < typebuf.tb_len)
3300 			    (void)add_to_showcmd(
3301 			       typebuf.tb_buf[typebuf.tb_off + showcmd_idx++]);
3302 			curwin->w_wcol = old_wcol;
3303 			curwin->w_wrow = old_wrow;
3304 #endif
3305 		    }
3306 
3307 		    // this looks nice when typing a dead character map
3308 		    if ((State & CMDLINE)
3309 #if defined(FEAT_CRYPT) || defined(FEAT_EVAL)
3310 			    && cmdline_star == 0
3311 #endif
3312 			    && ptr2cells(typebuf.tb_buf + typebuf.tb_off
3313 						   + typebuf.tb_len - 1) == 1)
3314 		    {
3315 			putcmdline(typebuf.tb_buf[typebuf.tb_off
3316 						+ typebuf.tb_len - 1], FALSE);
3317 			c1 = 1;
3318 		    }
3319 		}
3320 
3321 /*
3322  * get a character: 3. from the user - get it
3323  */
3324 		if (typebuf.tb_len == 0)
3325 		    // timedout may have been set while waiting for a mapping
3326 		    // that has a <Nop> RHS.
3327 		    timedout = FALSE;
3328 
3329 		if (advance)
3330 		{
3331 		    if (typebuf.tb_len == 0
3332 			    || !(p_timeout
3333 				 || (p_ttimeout && keylen == KEYLEN_PART_KEY)))
3334 			// blocking wait
3335 			wait_time = -1L;
3336 		    else if (keylen == KEYLEN_PART_KEY && p_ttm >= 0)
3337 			wait_time = p_ttm;
3338 		    else
3339 			wait_time = p_tm;
3340 		}
3341 		else
3342 		    wait_time = 0;
3343 
3344 		wait_tb_len = typebuf.tb_len;
3345 		c = inchar(typebuf.tb_buf + typebuf.tb_off + typebuf.tb_len,
3346 			typebuf.tb_buflen - typebuf.tb_off - typebuf.tb_len - 1,
3347 			wait_time);
3348 
3349 #ifdef FEAT_CMDL_INFO
3350 		if (showcmd_idx != 0)
3351 		    pop_showcmd();
3352 #endif
3353 		if (c1 == 1)
3354 		{
3355 		    if (State & INSERT)
3356 			edit_unputchar();
3357 		    if (State & CMDLINE)
3358 			unputcmdline();
3359 		    else
3360 			setcursor();	// put cursor back where it belongs
3361 		}
3362 
3363 		if (c < 0)
3364 		    continue;		// end of input script reached
3365 		if (c == NUL)		// no character available
3366 		{
3367 		    if (!advance)
3368 			break;
3369 		    if (wait_tb_len > 0)	// timed out
3370 		    {
3371 			timedout = TRUE;
3372 			continue;
3373 		    }
3374 		}
3375 		else
3376 		{	    // allow mapping for just typed characters
3377 		    while (typebuf.tb_buf[typebuf.tb_off
3378 						     + typebuf.tb_len] != NUL)
3379 			typebuf.tb_noremap[typebuf.tb_off
3380 						 + typebuf.tb_len++] = RM_YES;
3381 #ifdef HAVE_INPUT_METHOD
3382 		    // Get IM status right after getting keys, not after the
3383 		    // timeout for a mapping (focus may be lost by then).
3384 		    vgetc_im_active = im_get_status();
3385 #endif
3386 		}
3387 	    }	    // for (;;)
3388 	}	// if (!character from stuffbuf)
3389 
3390 	// if advance is FALSE don't loop on NULs
3391     } while ((c < 0 && c != K_CANCEL) || (advance && c == NUL));
3392 
3393     /*
3394      * The "INSERT" message is taken care of here:
3395      *	 if we return an ESC to exit insert mode, the message is deleted
3396      *	 if we don't return an ESC but deleted the message before, redisplay it
3397      */
3398     if (advance && p_smd && msg_silent == 0 && (State & INSERT))
3399     {
3400 	if (c == ESC && !mode_deleted && !no_mapping && mode_displayed)
3401 	{
3402 	    if (typebuf.tb_len && !KeyTyped)
3403 		redraw_cmdline = TRUE;	    // delete mode later
3404 	    else
3405 		unshowmode(FALSE);
3406 	}
3407 	else if (c != ESC && mode_deleted)
3408 	{
3409 	    if (typebuf.tb_len && !KeyTyped)
3410 		redraw_cmdline = TRUE;	    // show mode later
3411 	    else
3412 		showmode();
3413 	}
3414     }
3415 #ifdef FEAT_GUI
3416     // may unshow different cursor shape
3417     if (gui.in_use && shape_changed)
3418 	gui_update_cursor(TRUE, FALSE);
3419 #endif
3420     if (timedout && c == ESC)
3421     {
3422 	char_u nop_buf[3];
3423 
3424 	// When recording there will be no timeout.  Add a <Nop> after the ESC
3425 	// to avoid that it forms a key code with following characters.
3426 	nop_buf[0] = K_SPECIAL;
3427 	nop_buf[1] = KS_EXTRA;
3428 	nop_buf[2] = KE_NOP;
3429 	gotchars(nop_buf, 3);
3430     }
3431 
3432     --vgetc_busy;
3433 
3434     return c;
3435 }
3436 
3437 /*
3438  * inchar() - get one character from
3439  *	1. a scriptfile
3440  *	2. the keyboard
3441  *
3442  *  As many characters as we can get (up to 'maxlen') are put in "buf" and
3443  *  NUL terminated (buffer length must be 'maxlen' + 1).
3444  *  Minimum for "maxlen" is 3!!!!
3445  *
3446  *  "tb_change_cnt" is the value of typebuf.tb_change_cnt if "buf" points into
3447  *  it.  When typebuf.tb_change_cnt changes (e.g., when a message is received
3448  *  from a remote client) "buf" can no longer be used.  "tb_change_cnt" is 0
3449  *  otherwise.
3450  *
3451  *  If we got an interrupt all input is read until none is available.
3452  *
3453  *  If wait_time == 0  there is no waiting for the char.
3454  *  If wait_time == n  we wait for n msec for a character to arrive.
3455  *  If wait_time == -1 we wait forever for a character to arrive.
3456  *
3457  *  Return the number of obtained characters.
3458  *  Return -1 when end of input script reached.
3459  */
3460     static int
inchar(char_u * buf,int maxlen,long wait_time)3461 inchar(
3462     char_u	*buf,
3463     int		maxlen,
3464     long	wait_time)	    // milli seconds
3465 {
3466     int		len = 0;	    // init for GCC
3467     int		retesc = FALSE;	    // return ESC with gotint
3468     int		script_char;
3469     int		tb_change_cnt = typebuf.tb_change_cnt;
3470 
3471     if (wait_time == -1L || wait_time > 100L)  // flush output before waiting
3472     {
3473 	cursor_on();
3474 	out_flush_cursor(FALSE, FALSE);
3475 #if defined(FEAT_GUI) && defined(FEAT_MOUSESHAPE)
3476 	if (gui.in_use && postponed_mouseshape)
3477 	    update_mouseshape(-1);
3478 #endif
3479     }
3480 
3481     /*
3482      * Don't reset these when at the hit-return prompt, otherwise a endless
3483      * recursive loop may result (write error in swapfile, hit-return, timeout
3484      * on char wait, flush swapfile, write error....).
3485      */
3486     if (State != HITRETURN)
3487     {
3488 	did_outofmem_msg = FALSE;   // display out of memory message (again)
3489 	did_swapwrite_msg = FALSE;  // display swap file write error again
3490     }
3491     undo_off = FALSE;		    // restart undo now
3492 
3493     /*
3494      * Get a character from a script file if there is one.
3495      * If interrupted: Stop reading script files, close them all.
3496      */
3497     script_char = -1;
3498     while (scriptin[curscript] != NULL && script_char < 0
3499 #ifdef FEAT_EVAL
3500 	    && !ignore_script
3501 #endif
3502 	    )
3503     {
3504 #ifdef MESSAGE_QUEUE
3505 	parse_queued_messages();
3506 #endif
3507 
3508 	if (got_int || (script_char = getc(scriptin[curscript])) < 0)
3509 	{
3510 	    // Reached EOF.
3511 	    // Careful: closescript() frees typebuf.tb_buf[] and buf[] may
3512 	    // point inside typebuf.tb_buf[].  Don't use buf[] after this!
3513 	    closescript();
3514 	    /*
3515 	     * When reading script file is interrupted, return an ESC to get
3516 	     * back to normal mode.
3517 	     * Otherwise return -1, because typebuf.tb_buf[] has changed.
3518 	     */
3519 	    if (got_int)
3520 		retesc = TRUE;
3521 	    else
3522 		return -1;
3523 	}
3524 	else
3525 	{
3526 	    buf[0] = script_char;
3527 	    len = 1;
3528 	}
3529     }
3530 
3531     if (script_char < 0)	// did not get a character from script
3532     {
3533 	/*
3534 	 * If we got an interrupt, skip all previously typed characters and
3535 	 * return TRUE if quit reading script file.
3536 	 * Stop reading typeahead when a single CTRL-C was read,
3537 	 * fill_input_buf() returns this when not able to read from stdin.
3538 	 * Don't use buf[] here, closescript() may have freed typebuf.tb_buf[]
3539 	 * and buf may be pointing inside typebuf.tb_buf[].
3540 	 */
3541 	if (got_int)
3542 	{
3543 #define DUM_LEN MAXMAPLEN * 3 + 3
3544 	    char_u	dum[DUM_LEN + 1];
3545 
3546 	    for (;;)
3547 	    {
3548 		len = ui_inchar(dum, DUM_LEN, 0L, 0);
3549 		if (len == 0 || (len == 1 && dum[0] == 3))
3550 		    break;
3551 	    }
3552 	    return retesc;
3553 	}
3554 
3555 	/*
3556 	 * Always flush the output characters when getting input characters
3557 	 * from the user and not just peeking.
3558 	 */
3559 	if (wait_time == -1L || wait_time > 10L)
3560 	    out_flush();
3561 
3562 	/*
3563 	 * Fill up to a third of the buffer, because each character may be
3564 	 * tripled below.
3565 	 */
3566 	len = ui_inchar(buf, maxlen / 3, wait_time, tb_change_cnt);
3567     }
3568 
3569     // If the typebuf was changed further down, it is like nothing was added by
3570     // this call.
3571     if (typebuf_changed(tb_change_cnt))
3572 	return 0;
3573 
3574     // Note the change in the typeahead buffer, this matters for when
3575     // vgetorpeek() is called recursively, e.g. using getchar(1) in a timer
3576     // function.
3577     if (len > 0 && ++typebuf.tb_change_cnt == 0)
3578 	typebuf.tb_change_cnt = 1;
3579 
3580     return fix_input_buffer(buf, len);
3581 }
3582 
3583 /*
3584  * Fix typed characters for use by vgetc() and check_termcode().
3585  * "buf[]" must have room to triple the number of bytes!
3586  * Returns the new length.
3587  */
3588     int
fix_input_buffer(char_u * buf,int len)3589 fix_input_buffer(char_u *buf, int len)
3590 {
3591     int		i;
3592     char_u	*p = buf;
3593 
3594     /*
3595      * Two characters are special: NUL and K_SPECIAL.
3596      * When compiled With the GUI CSI is also special.
3597      * Replace	     NUL by K_SPECIAL KS_ZERO	 KE_FILLER
3598      * Replace K_SPECIAL by K_SPECIAL KS_SPECIAL KE_FILLER
3599      * Replace       CSI by K_SPECIAL KS_EXTRA   KE_CSI
3600      */
3601     for (i = len; --i >= 0; ++p)
3602     {
3603 #ifdef FEAT_GUI
3604 	// When the GUI is used any character can come after a CSI, don't
3605 	// escape it.
3606 	if (gui.in_use && p[0] == CSI && i >= 2)
3607 	{
3608 	    p += 2;
3609 	    i -= 2;
3610 	}
3611 # ifndef MSWIN
3612 	// When the GUI is not used CSI needs to be escaped.
3613 	else if (!gui.in_use && p[0] == CSI)
3614 	{
3615 	    mch_memmove(p + 3, p + 1, (size_t)i);
3616 	    *p++ = K_SPECIAL;
3617 	    *p++ = KS_EXTRA;
3618 	    *p = (int)KE_CSI;
3619 	    len += 2;
3620 	}
3621 # endif
3622 	else
3623 #endif
3624 	if (p[0] == NUL || (p[0] == K_SPECIAL
3625 		    // timeout may generate K_CURSORHOLD
3626 		    && (i < 2 || p[1] != KS_EXTRA || p[2] != (int)KE_CURSORHOLD)
3627 #if defined(MSWIN) && (!defined(FEAT_GUI) || defined(VIMDLL))
3628 		    // Win32 console passes modifiers
3629 		    && (
3630 # ifdef VIMDLL
3631 			gui.in_use ||
3632 # endif
3633 			(i < 2 || p[1] != KS_MODIFIER))
3634 #endif
3635 		    ))
3636 	{
3637 	    mch_memmove(p + 3, p + 1, (size_t)i);
3638 	    p[2] = K_THIRD(p[0]);
3639 	    p[1] = K_SECOND(p[0]);
3640 	    p[0] = K_SPECIAL;
3641 	    p += 2;
3642 	    len += 2;
3643 	}
3644     }
3645     *p = NUL;		// add trailing NUL
3646     return len;
3647 }
3648 
3649 #if defined(USE_INPUT_BUF) || defined(PROTO)
3650 /*
3651  * Return TRUE when bytes are in the input buffer or in the typeahead buffer.
3652  * Normally the input buffer would be sufficient, but the server_to_input_buf()
3653  * or feedkeys() may insert characters in the typeahead buffer while we are
3654  * waiting for input to arrive.
3655  */
3656     int
input_available(void)3657 input_available(void)
3658 {
3659     return (!vim_is_input_buf_empty()
3660 # if defined(FEAT_CLIENTSERVER) || defined(FEAT_EVAL)
3661 	    || typebuf_was_filled
3662 # endif
3663 	    );
3664 }
3665 #endif
3666 
3667 /*
3668  * Function passed to do_cmdline() to get the command after a <Cmd> key from
3669  * typeahead.
3670  */
3671     char_u *
getcmdkeycmd(int promptc UNUSED,void * cookie UNUSED,int indent UNUSED,getline_opt_T do_concat UNUSED)3672 getcmdkeycmd(
3673 	int		promptc UNUSED,
3674 	void		*cookie UNUSED,
3675 	int		indent UNUSED,
3676 	getline_opt_T	do_concat UNUSED)
3677 {
3678     garray_T	line_ga;
3679     int		c1 = -1;
3680     int		c2;
3681     int		cmod = 0;
3682     int		aborted = FALSE;
3683 
3684     ga_init2(&line_ga, 1, 32);
3685 
3686     // no mapping for these characters
3687     no_mapping++;
3688 
3689     got_int = FALSE;
3690     while (c1 != NUL && !aborted)
3691     {
3692 	if (ga_grow(&line_ga, 32) != OK)
3693 	{
3694 	    aborted = TRUE;
3695 	    break;
3696 	}
3697 
3698 	if (vgetorpeek(FALSE) == NUL)
3699 	{
3700 	    // incomplete <Cmd> is an error, because there is not much the user
3701 	    // could do in this state.
3702 	    emsg(_(e_cmd_mapping_must_end_with_cr));
3703 	    aborted = TRUE;
3704 	    break;
3705 	}
3706 
3707 	// Get one character at a time.
3708 	c1 = vgetorpeek(TRUE);
3709 
3710 	// Get two extra bytes for special keys
3711 	if (c1 == K_SPECIAL)
3712 	{
3713 	    c1 = vgetorpeek(TRUE);
3714 	    c2 = vgetorpeek(TRUE);
3715 	    if (c1 == KS_MODIFIER)
3716 	    {
3717 		cmod = c2;
3718 		continue;
3719 	    }
3720 	    c1 = TO_SPECIAL(c1, c2);
3721 	}
3722 	if (c1 == Ctrl_V)
3723 	{
3724 	    // CTRL-V is followed by octal, hex or other characters, reverses
3725 	    // what AppendToRedobuffLit() does.
3726 	    no_reduce_keys = TRUE;  //  don't merge modifyOtherKeys
3727 	    c1 = get_literal(TRUE);
3728 	    no_reduce_keys = FALSE;
3729 	}
3730 
3731 	if (got_int)
3732 	    aborted = TRUE;
3733 	else if (c1 == '\r' || c1 == '\n')
3734 	    c1 = NUL;  // end the line
3735 	else if (c1 == ESC)
3736 	    aborted = TRUE;
3737 	else if (c1 == K_COMMAND)
3738 	{
3739 	    // give a nicer error message for this special case
3740 	    emsg(_(e_cmd_mapping_must_end_with_cr_before_second_cmd));
3741 	    aborted = TRUE;
3742 	}
3743 	else if (IS_SPECIAL(c1))
3744 	{
3745 	    if (c1 == K_SNR)
3746 		ga_concat(&line_ga, (char_u *)"<SNR>");
3747 	    else
3748 	    {
3749 		semsg(e_cmd_maping_must_not_include_str_key,
3750 					       get_special_key_name(c1, cmod));
3751 		aborted = TRUE;
3752 	    }
3753 	}
3754 	else
3755 	    ga_append(&line_ga, (char)c1);
3756 
3757 	cmod = 0;
3758     }
3759 
3760     no_mapping--;
3761 
3762     if (aborted)
3763 	ga_clear(&line_ga);
3764 
3765     return (char_u *)line_ga.ga_data;
3766 }
3767