blob: 6bbfbd775c13d5367d27ca9cf24105ce258afd3a [file] [log] [blame]
Bram Moolenaaredf3f972016-08-29 22:49:24 +02001/* vi:set ts=8 sts=4 sw=4 noet:
Bram Moolenaar071d4272004-06-13 20:20:40 +00002 *
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 * misc2.c: Various functions.
12 */
13#include "vim.h"
14
Bram Moolenaar85a20022019-12-21 18:25:54 +010015static char_u *username = NULL; // cached result of mch_get_user_name()
Bram Moolenaarf461c8e2005-06-25 23:04:51 +000016
Bram Moolenaar92b8b2d2016-01-29 22:36:45 +010017static int coladvance2(pos_T *pos, int addspaces, int finetune, colnr_T wcol);
Bram Moolenaar071d4272004-06-13 20:20:40 +000018
19/*
20 * Return TRUE if in the current mode we need to use virtual.
21 */
22 int
Bram Moolenaar9b578142016-01-30 19:39:49 +010023virtual_active(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +000024{
Bram Moolenaar85a20022019-12-21 18:25:54 +010025 // While an operator is being executed we return "virtual_op", because
26 // VIsual_active has already been reset, thus we can't check for "block"
27 // being used.
Bram Moolenaar071d4272004-06-13 20:20:40 +000028 if (virtual_op != MAYBE)
29 return virtual_op;
30 return (ve_flags == VE_ALL
Bram Moolenaar071d4272004-06-13 20:20:40 +000031 || ((ve_flags & VE_BLOCK) && VIsual_active && VIsual_mode == Ctrl_V)
Bram Moolenaar071d4272004-06-13 20:20:40 +000032 || ((ve_flags & VE_INSERT) && (State & INSERT)));
33}
34
35/*
36 * Get the screen position of the cursor.
37 */
38 int
Bram Moolenaar9b578142016-01-30 19:39:49 +010039getviscol(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +000040{
41 colnr_T x;
42
43 getvvcol(curwin, &curwin->w_cursor, &x, NULL, NULL);
44 return (int)x;
45}
46
47/*
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +000048 * Go to column "wcol", and add/insert white space as necessary to get the
Bram Moolenaar071d4272004-06-13 20:20:40 +000049 * cursor in that column.
50 * The caller must have saved the cursor line for undo!
51 */
52 int
Bram Moolenaar9b578142016-01-30 19:39:49 +010053coladvance_force(colnr_T wcol)
Bram Moolenaar071d4272004-06-13 20:20:40 +000054{
55 int rc = coladvance2(&curwin->w_cursor, TRUE, FALSE, wcol);
56
57 if (wcol == MAXCOL)
58 curwin->w_valid &= ~VALID_VIRTCOL;
59 else
60 {
Bram Moolenaar85a20022019-12-21 18:25:54 +010061 // Virtcol is valid
Bram Moolenaar071d4272004-06-13 20:20:40 +000062 curwin->w_valid |= VALID_VIRTCOL;
63 curwin->w_virtcol = wcol;
64 }
65 return rc;
66}
Bram Moolenaar071d4272004-06-13 20:20:40 +000067
68/*
Bram Moolenaar977239e2019-01-11 16:16:01 +010069 * Get the screen position of character col with a coladd in the cursor line.
70 */
71 int
Bram Moolenaarfc3abf42019-01-24 15:54:21 +010072getviscol2(colnr_T col, colnr_T coladd UNUSED)
Bram Moolenaar977239e2019-01-11 16:16:01 +010073{
74 colnr_T x;
75 pos_T pos;
76
77 pos.lnum = curwin->w_cursor.lnum;
78 pos.col = col;
Bram Moolenaar977239e2019-01-11 16:16:01 +010079 pos.coladd = coladd;
Bram Moolenaar977239e2019-01-11 16:16:01 +010080 getvvcol(curwin, &pos, &x, NULL, NULL);
81 return (int)x;
82}
83
84/*
Bram Moolenaar071d4272004-06-13 20:20:40 +000085 * Try to advance the Cursor to the specified screen column.
86 * If virtual editing: fine tune the cursor position.
87 * Note that all virtual positions off the end of a line should share
88 * a curwin->w_cursor.col value (n.b. this is equal to STRLEN(line)),
89 * beginning at coladd 0.
90 *
91 * return OK if desired column is reached, FAIL if not
92 */
93 int
Bram Moolenaar9b578142016-01-30 19:39:49 +010094coladvance(colnr_T wcol)
Bram Moolenaar071d4272004-06-13 20:20:40 +000095{
96 int rc = getvpos(&curwin->w_cursor, wcol);
97
98 if (wcol == MAXCOL || rc == FAIL)
99 curwin->w_valid &= ~VALID_VIRTCOL;
Bram Moolenaardfccaf02004-12-31 20:56:11 +0000100 else if (*ml_get_cursor() != TAB)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000101 {
Bram Moolenaar85a20022019-12-21 18:25:54 +0100102 // Virtcol is valid when not on a TAB
Bram Moolenaar071d4272004-06-13 20:20:40 +0000103 curwin->w_valid |= VALID_VIRTCOL;
104 curwin->w_virtcol = wcol;
105 }
106 return rc;
107}
108
109/*
110 * Return in "pos" the position of the cursor advanced to screen column "wcol".
111 * return OK if desired column is reached, FAIL if not
112 */
113 int
Bram Moolenaar9b578142016-01-30 19:39:49 +0100114getvpos(pos_T *pos, colnr_T wcol)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000115{
Bram Moolenaar071d4272004-06-13 20:20:40 +0000116 return coladvance2(pos, FALSE, virtual_active(), wcol);
117}
118
119 static int
Bram Moolenaar9b578142016-01-30 19:39:49 +0100120coladvance2(
121 pos_T *pos,
Bram Moolenaarceba3dd2019-10-12 16:12:54 +0200122 int addspaces, // change the text to achieve our goal?
123 int finetune, // change char offset for the exact column
124 colnr_T wcol_arg) // column to move to (can be negative)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000125{
Bram Moolenaarceba3dd2019-10-12 16:12:54 +0200126 colnr_T wcol = wcol_arg;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000127 int idx;
128 char_u *ptr;
129 char_u *line;
130 colnr_T col = 0;
131 int csize = 0;
132 int one_more;
133#ifdef FEAT_LINEBREAK
134 int head = 0;
135#endif
136
Bram Moolenaarefd2bf12006-03-16 21:41:35 +0000137 one_more = (State & INSERT)
138 || restart_edit != NUL
Bram Moolenaarefd2bf12006-03-16 21:41:35 +0000139 || (VIsual_active && *p_sel != 'o')
Bram Moolenaarceba3dd2019-10-12 16:12:54 +0200140 || ((ve_flags & VE_ONEMORE) && wcol < MAXCOL);
Bram Moolenaara1381de2009-11-03 15:44:21 +0000141 line = ml_get_buf(curbuf, pos->lnum, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000142
143 if (wcol >= MAXCOL)
144 {
145 idx = (int)STRLEN(line) - 1 + one_more;
146 col = wcol;
147
Bram Moolenaar071d4272004-06-13 20:20:40 +0000148 if ((addspaces || finetune) && !VIsual_active)
149 {
150 curwin->w_curswant = linetabsize(line) + one_more;
151 if (curwin->w_curswant > 0)
152 --curwin->w_curswant;
153 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000154 }
155 else
156 {
Bram Moolenaar02631462017-09-22 15:20:32 +0200157 int width = curwin->w_width - win_col_off(curwin);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000158
Bram Moolenaarebefac62005-12-28 22:39:57 +0000159 if (finetune
Bram Moolenaar071d4272004-06-13 20:20:40 +0000160 && curwin->w_p_wrap
Bram Moolenaar071d4272004-06-13 20:20:40 +0000161 && curwin->w_width != 0
Bram Moolenaar071d4272004-06-13 20:20:40 +0000162 && wcol >= (colnr_T)width)
163 {
164 csize = linetabsize(line);
165 if (csize > 0)
166 csize--;
167
Bram Moolenaarebefac62005-12-28 22:39:57 +0000168 if (wcol / width > (colnr_T)csize / width
169 && ((State & INSERT) == 0 || (int)wcol > csize + 1))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000170 {
Bram Moolenaar85a20022019-12-21 18:25:54 +0100171 // In case of line wrapping don't move the cursor beyond the
172 // right screen edge. In Insert mode allow going just beyond
173 // the last character (like what happens when typing and
174 // reaching the right window edge).
Bram Moolenaar071d4272004-06-13 20:20:40 +0000175 wcol = (csize / width + 1) * width - 1;
176 }
177 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000178
Bram Moolenaar071d4272004-06-13 20:20:40 +0000179 ptr = line;
180 while (col <= wcol && *ptr != NUL)
181 {
Bram Moolenaar85a20022019-12-21 18:25:54 +0100182 // Count a tab for what it's worth (if list mode not on)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000183#ifdef FEAT_LINEBREAK
Bram Moolenaar597a4222014-06-25 14:39:50 +0200184 csize = win_lbr_chartabsize(curwin, line, ptr, col, &head);
Bram Moolenaar91acfff2017-03-12 19:22:36 +0100185 MB_PTR_ADV(ptr);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000186#else
Bram Moolenaar597a4222014-06-25 14:39:50 +0200187 csize = lbr_chartabsize_adv(line, &ptr, col);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000188#endif
189 col += csize;
190 }
191 idx = (int)(ptr - line);
192 /*
193 * Handle all the special cases. The virtual_active() check
194 * is needed to ensure that a virtual position off the end of
195 * a line has the correct indexing. The one_more comparison
196 * replaces an explicit add of one_more later on.
197 */
198 if (col > wcol || (!virtual_active() && one_more == 0))
199 {
200 idx -= 1;
201# ifdef FEAT_LINEBREAK
Bram Moolenaar85a20022019-12-21 18:25:54 +0100202 // Don't count the chars from 'showbreak'.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000203 csize -= head;
204# endif
205 col -= csize;
206 }
207
Bram Moolenaar071d4272004-06-13 20:20:40 +0000208 if (virtual_active()
209 && addspaces
Bram Moolenaarceba3dd2019-10-12 16:12:54 +0200210 && wcol >= 0
Bram Moolenaar071d4272004-06-13 20:20:40 +0000211 && ((col != wcol && col != wcol + 1) || csize > 1))
212 {
Bram Moolenaar85a20022019-12-21 18:25:54 +0100213 // 'virtualedit' is set: The difference between wcol and col is
214 // filled with spaces.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000215
216 if (line[idx] == NUL)
217 {
Bram Moolenaar85a20022019-12-21 18:25:54 +0100218 // Append spaces
Bram Moolenaar071d4272004-06-13 20:20:40 +0000219 int correct = wcol - col;
220 char_u *newline = alloc(idx + correct + 1);
221 int t;
222
223 if (newline == NULL)
224 return FAIL;
225
226 for (t = 0; t < idx; ++t)
227 newline[t] = line[t];
228
229 for (t = 0; t < correct; ++t)
230 newline[t + idx] = ' ';
231
232 newline[idx + correct] = NUL;
233
234 ml_replace(pos->lnum, newline, FALSE);
235 changed_bytes(pos->lnum, (colnr_T)idx);
236 idx += correct;
237 col = wcol;
238 }
239 else
240 {
Bram Moolenaar85a20022019-12-21 18:25:54 +0100241 // Break a tab
Bram Moolenaar071d4272004-06-13 20:20:40 +0000242 int linelen = (int)STRLEN(line);
Bram Moolenaar85a20022019-12-21 18:25:54 +0100243 int correct = wcol - col - csize + 1; // negative!!
Bram Moolenaareb3593b2006-04-22 22:33:57 +0000244 char_u *newline;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000245 int t, s = 0;
246 int v;
247
Bram Moolenaareb3593b2006-04-22 22:33:57 +0000248 if (-correct > csize)
249 return FAIL;
250
251 newline = alloc(linelen + csize);
252 if (newline == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000253 return FAIL;
254
255 for (t = 0; t < linelen; t++)
256 {
257 if (t != idx)
258 newline[s++] = line[t];
259 else
260 for (v = 0; v < csize; v++)
261 newline[s++] = ' ';
262 }
263
264 newline[linelen + csize - 1] = NUL;
265
266 ml_replace(pos->lnum, newline, FALSE);
267 changed_bytes(pos->lnum, idx);
268 idx += (csize - 1 + correct);
269 col += correct;
270 }
271 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000272 }
273
274 if (idx < 0)
275 pos->col = 0;
276 else
277 pos->col = idx;
278
Bram Moolenaar071d4272004-06-13 20:20:40 +0000279 pos->coladd = 0;
280
281 if (finetune)
282 {
283 if (wcol == MAXCOL)
284 {
Bram Moolenaar85a20022019-12-21 18:25:54 +0100285 // The width of the last character is used to set coladd.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000286 if (!one_more)
287 {
288 colnr_T scol, ecol;
289
290 getvcol(curwin, pos, &scol, NULL, &ecol);
291 pos->coladd = ecol - scol;
292 }
293 }
294 else
295 {
296 int b = (int)wcol - (int)col;
297
Bram Moolenaar85a20022019-12-21 18:25:54 +0100298 // The difference between wcol and col is used to set coladd.
Bram Moolenaar02631462017-09-22 15:20:32 +0200299 if (b > 0 && b < (MAXCOL - 2 * curwin->w_width))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000300 pos->coladd = b;
301
302 col += b;
303 }
304 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000305
Bram Moolenaar85a20022019-12-21 18:25:54 +0100306 // prevent from moving onto a trail byte
Bram Moolenaar071d4272004-06-13 20:20:40 +0000307 if (has_mbyte)
Bram Moolenaar03a807a2011-07-07 15:08:58 +0200308 mb_adjustpos(curbuf, pos);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000309
Bram Moolenaarceba3dd2019-10-12 16:12:54 +0200310 if (wcol < 0 || col < wcol)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000311 return FAIL;
312 return OK;
313}
314
315/*
Bram Moolenaar446cb832008-06-24 21:56:24 +0000316 * Increment the cursor position. See inc() for return values.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000317 */
318 int
Bram Moolenaar9b578142016-01-30 19:39:49 +0100319inc_cursor(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000320{
321 return inc(&curwin->w_cursor);
322}
323
Bram Moolenaar446cb832008-06-24 21:56:24 +0000324/*
325 * Increment the line pointer "lp" crossing line boundaries as necessary.
326 * Return 1 when going to the next line.
327 * Return 2 when moving forward onto a NUL at the end of the line).
328 * Return -1 when at the end of file.
329 * Return 0 otherwise.
330 */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000331 int
Bram Moolenaar9b578142016-01-30 19:39:49 +0100332inc(pos_T *lp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000333{
Bram Moolenaar8ada6aa2017-12-19 21:23:21 +0100334 char_u *p;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000335
Bram Moolenaar85a20022019-12-21 18:25:54 +0100336 // when searching position may be set to end of a line
Bram Moolenaar8ada6aa2017-12-19 21:23:21 +0100337 if (lp->col != MAXCOL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000338 {
Bram Moolenaar8ada6aa2017-12-19 21:23:21 +0100339 p = ml_get_pos(lp);
Bram Moolenaar85a20022019-12-21 18:25:54 +0100340 if (*p != NUL) // still within line, move to next char (may be NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000341 {
Bram Moolenaar8ada6aa2017-12-19 21:23:21 +0100342 if (has_mbyte)
343 {
344 int l = (*mb_ptr2len)(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000345
Bram Moolenaar8ada6aa2017-12-19 21:23:21 +0100346 lp->col += l;
347 return ((p[l] != NUL) ? 0 : 2);
348 }
Bram Moolenaar8ada6aa2017-12-19 21:23:21 +0100349 lp->col++;
Bram Moolenaar8ada6aa2017-12-19 21:23:21 +0100350 lp->coladd = 0;
Bram Moolenaar8ada6aa2017-12-19 21:23:21 +0100351 return ((p[1] != NUL) ? 0 : 2);
352 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000353 }
Bram Moolenaar85a20022019-12-21 18:25:54 +0100354 if (lp->lnum != curbuf->b_ml.ml_line_count) // there is a next line
Bram Moolenaar071d4272004-06-13 20:20:40 +0000355 {
356 lp->col = 0;
357 lp->lnum++;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000358 lp->coladd = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000359 return 1;
360 }
361 return -1;
362}
363
364/*
365 * incl(lp): same as inc(), but skip the NUL at the end of non-empty lines
366 */
367 int
Bram Moolenaar9b578142016-01-30 19:39:49 +0100368incl(pos_T *lp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000369{
370 int r;
371
372 if ((r = inc(lp)) >= 1 && lp->col)
373 r = inc(lp);
374 return r;
375}
376
377/*
378 * dec(p)
379 *
380 * Decrement the line pointer 'p' crossing line boundaries as necessary.
381 * Return 1 when crossing a line, -1 when at start of file, 0 otherwise.
382 */
383 int
Bram Moolenaar9b578142016-01-30 19:39:49 +0100384dec_cursor(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000385{
Bram Moolenaarcaa55b62017-01-10 13:51:09 +0100386 return dec(&curwin->w_cursor);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000387}
388
389 int
Bram Moolenaar9b578142016-01-30 19:39:49 +0100390dec(pos_T *lp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000391{
392 char_u *p;
393
Bram Moolenaar071d4272004-06-13 20:20:40 +0000394 lp->coladd = 0;
Bram Moolenaar1bd999f2017-12-19 22:25:40 +0100395 if (lp->col == MAXCOL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000396 {
Bram Moolenaar85a20022019-12-21 18:25:54 +0100397 // past end of line
Bram Moolenaar1bd999f2017-12-19 22:25:40 +0100398 p = ml_get(lp->lnum);
399 lp->col = (colnr_T)STRLEN(p);
Bram Moolenaar1bd999f2017-12-19 22:25:40 +0100400 if (has_mbyte)
401 lp->col -= (*mb_head_off)(p, p + lp->col);
Bram Moolenaar1bd999f2017-12-19 22:25:40 +0100402 return 0;
403 }
404
405 if (lp->col > 0)
406 {
Bram Moolenaar85a20022019-12-21 18:25:54 +0100407 // still within line
Bram Moolenaar071d4272004-06-13 20:20:40 +0000408 lp->col--;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000409 if (has_mbyte)
410 {
411 p = ml_get(lp->lnum);
412 lp->col -= (*mb_head_off)(p, p + lp->col);
413 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000414 return 0;
415 }
Bram Moolenaar1bd999f2017-12-19 22:25:40 +0100416
417 if (lp->lnum > 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000418 {
Bram Moolenaar85a20022019-12-21 18:25:54 +0100419 // there is a prior line
Bram Moolenaar071d4272004-06-13 20:20:40 +0000420 lp->lnum--;
421 p = ml_get(lp->lnum);
422 lp->col = (colnr_T)STRLEN(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000423 if (has_mbyte)
424 lp->col -= (*mb_head_off)(p, p + lp->col);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000425 return 1;
426 }
Bram Moolenaar1bd999f2017-12-19 22:25:40 +0100427
Bram Moolenaar85a20022019-12-21 18:25:54 +0100428 // at start of file
Bram Moolenaar1bd999f2017-12-19 22:25:40 +0100429 return -1;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000430}
431
432/*
433 * decl(lp): same as dec(), but skip the NUL at the end of non-empty lines
434 */
435 int
Bram Moolenaar9b578142016-01-30 19:39:49 +0100436decl(pos_T *lp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000437{
438 int r;
439
440 if ((r = dec(lp)) == 1 && lp->col)
441 r = dec(lp);
442 return r;
443}
444
445/*
Bram Moolenaar64486672010-05-16 15:46:46 +0200446 * Get the line number relative to the current cursor position, i.e. the
447 * difference between line number and cursor position. Only look for lines that
448 * can be visible, folded lines don't count.
449 */
450 linenr_T
Bram Moolenaar9b578142016-01-30 19:39:49 +0100451get_cursor_rel_lnum(
452 win_T *wp,
Bram Moolenaar85a20022019-12-21 18:25:54 +0100453 linenr_T lnum) // line number to get the result for
Bram Moolenaar64486672010-05-16 15:46:46 +0200454{
455 linenr_T cursor = wp->w_cursor.lnum;
456 linenr_T retval = 0;
457
458#ifdef FEAT_FOLDING
459 if (hasAnyFolding(wp))
460 {
461 if (lnum > cursor)
462 {
463 while (lnum > cursor)
464 {
Bram Moolenaar0bd7b3f2013-12-14 12:48:58 +0100465 (void)hasFoldingWin(wp, lnum, &lnum, NULL, TRUE, NULL);
Bram Moolenaar85a20022019-12-21 18:25:54 +0100466 // if lnum and cursor are in the same fold,
467 // now lnum <= cursor
Bram Moolenaar64486672010-05-16 15:46:46 +0200468 if (lnum > cursor)
469 retval++;
470 lnum--;
471 }
472 }
473 else if (lnum < cursor)
474 {
475 while (lnum < cursor)
476 {
Bram Moolenaar0bd7b3f2013-12-14 12:48:58 +0100477 (void)hasFoldingWin(wp, lnum, NULL, &lnum, TRUE, NULL);
Bram Moolenaar85a20022019-12-21 18:25:54 +0100478 // if lnum and cursor are in the same fold,
479 // now lnum >= cursor
Bram Moolenaar64486672010-05-16 15:46:46 +0200480 if (lnum < cursor)
481 retval--;
482 lnum++;
483 }
484 }
Bram Moolenaar85a20022019-12-21 18:25:54 +0100485 // else if (lnum == cursor)
486 // retval = 0;
Bram Moolenaar64486672010-05-16 15:46:46 +0200487 }
488 else
489#endif
490 retval = lnum - cursor;
491
492 return retval;
493}
494
495/*
Bram Moolenaard5824ce2016-09-04 20:35:01 +0200496 * Make sure "pos.lnum" and "pos.col" are valid in "buf".
497 * This allows for the col to be on the NUL byte.
498 */
499 void
500check_pos(buf_T *buf, pos_T *pos)
501{
502 char_u *line;
503 colnr_T len;
504
505 if (pos->lnum > buf->b_ml.ml_line_count)
506 pos->lnum = buf->b_ml.ml_line_count;
507
508 if (pos->col > 0)
509 {
510 line = ml_get_buf(buf, pos->lnum, FALSE);
511 len = (colnr_T)STRLEN(line);
512 if (pos->col > len)
513 pos->col = len;
514 }
515}
516
517/*
Bram Moolenaar071d4272004-06-13 20:20:40 +0000518 * Make sure curwin->w_cursor.lnum is valid.
519 */
520 void
Bram Moolenaar9b578142016-01-30 19:39:49 +0100521check_cursor_lnum(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000522{
523 if (curwin->w_cursor.lnum > curbuf->b_ml.ml_line_count)
524 {
525#ifdef FEAT_FOLDING
Bram Moolenaar85a20022019-12-21 18:25:54 +0100526 // If there is a closed fold at the end of the file, put the cursor in
527 // its first line. Otherwise in the last line.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000528 if (!hasFolding(curbuf->b_ml.ml_line_count,
529 &curwin->w_cursor.lnum, NULL))
530#endif
531 curwin->w_cursor.lnum = curbuf->b_ml.ml_line_count;
532 }
533 if (curwin->w_cursor.lnum <= 0)
534 curwin->w_cursor.lnum = 1;
535}
536
537/*
538 * Make sure curwin->w_cursor.col is valid.
539 */
540 void
Bram Moolenaar9b578142016-01-30 19:39:49 +0100541check_cursor_col(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000542{
Bram Moolenaar03a807a2011-07-07 15:08:58 +0200543 check_cursor_col_win(curwin);
544}
545
546/*
547 * Make sure win->w_cursor.col is valid.
548 */
549 void
Bram Moolenaar9b578142016-01-30 19:39:49 +0100550check_cursor_col_win(win_T *win)
Bram Moolenaar03a807a2011-07-07 15:08:58 +0200551{
Bram Moolenaar071d4272004-06-13 20:20:40 +0000552 colnr_T len;
Bram Moolenaar03a807a2011-07-07 15:08:58 +0200553 colnr_T oldcol = win->w_cursor.col;
554 colnr_T oldcoladd = win->w_cursor.col + win->w_cursor.coladd;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000555
Bram Moolenaar03a807a2011-07-07 15:08:58 +0200556 len = (colnr_T)STRLEN(ml_get_buf(win->w_buffer, win->w_cursor.lnum, FALSE));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000557 if (len == 0)
Bram Moolenaar03a807a2011-07-07 15:08:58 +0200558 win->w_cursor.col = 0;
559 else if (win->w_cursor.col >= len)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000560 {
Bram Moolenaar85a20022019-12-21 18:25:54 +0100561 // Allow cursor past end-of-line when:
562 // - in Insert mode or restarting Insert mode
563 // - in Visual mode and 'selection' isn't "old"
564 // - 'virtualedit' is set
Bram Moolenaarebefac62005-12-28 22:39:57 +0000565 if ((State & INSERT) || restart_edit
Bram Moolenaar071d4272004-06-13 20:20:40 +0000566 || (VIsual_active && *p_sel != 'o')
Bram Moolenaar33f54432008-01-04 20:25:44 +0000567 || (ve_flags & VE_ONEMORE)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000568 || virtual_active())
Bram Moolenaar03a807a2011-07-07 15:08:58 +0200569 win->w_cursor.col = len;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000570 else
Bram Moolenaar87c19962007-04-26 08:54:21 +0000571 {
Bram Moolenaar03a807a2011-07-07 15:08:58 +0200572 win->w_cursor.col = len - 1;
Bram Moolenaar85a20022019-12-21 18:25:54 +0100573 // Move the cursor to the head byte.
Bram Moolenaar87c19962007-04-26 08:54:21 +0000574 if (has_mbyte)
Bram Moolenaar03a807a2011-07-07 15:08:58 +0200575 mb_adjustpos(win->w_buffer, &win->w_cursor);
Bram Moolenaar87c19962007-04-26 08:54:21 +0000576 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000577 }
Bram Moolenaar03a807a2011-07-07 15:08:58 +0200578 else if (win->w_cursor.col < 0)
579 win->w_cursor.col = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000580
Bram Moolenaar85a20022019-12-21 18:25:54 +0100581 // If virtual editing is on, we can leave the cursor on the old position,
582 // only we must set it to virtual. But don't do it when at the end of the
583 // line.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000584 if (oldcol == MAXCOL)
Bram Moolenaar03a807a2011-07-07 15:08:58 +0200585 win->w_cursor.coladd = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000586 else if (ve_flags == VE_ALL)
Bram Moolenaar552c8a52009-03-11 16:29:20 +0000587 {
Bram Moolenaar03a807a2011-07-07 15:08:58 +0200588 if (oldcoladd > win->w_cursor.col)
Bram Moolenaar9aa15692017-08-19 15:05:32 +0200589 {
Bram Moolenaar03a807a2011-07-07 15:08:58 +0200590 win->w_cursor.coladd = oldcoladd - win->w_cursor.col;
Bram Moolenaard41babe2017-08-30 17:01:35 +0200591
Bram Moolenaar85a20022019-12-21 18:25:54 +0100592 // Make sure that coladd is not more than the char width.
593 // Not for the last character, coladd is then used when the cursor
594 // is actually after the last character.
Bram Moolenaard41babe2017-08-30 17:01:35 +0200595 if (win->w_cursor.col + 1 < len && win->w_cursor.coladd > 0)
Bram Moolenaar9aa15692017-08-19 15:05:32 +0200596 {
597 int cs, ce;
598
Bram Moolenaar9aa15692017-08-19 15:05:32 +0200599 getvcol(win, &win->w_cursor, &cs, NULL, &ce);
600 if (win->w_cursor.coladd > ce - cs)
601 win->w_cursor.coladd = ce - cs;
602 }
603 }
Bram Moolenaar552c8a52009-03-11 16:29:20 +0000604 else
Bram Moolenaar85a20022019-12-21 18:25:54 +0100605 // avoid weird number when there is a miscalculation or overflow
Bram Moolenaar03a807a2011-07-07 15:08:58 +0200606 win->w_cursor.coladd = 0;
Bram Moolenaar552c8a52009-03-11 16:29:20 +0000607 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000608}
609
610/*
611 * make sure curwin->w_cursor in on a valid character
612 */
613 void
Bram Moolenaar9b578142016-01-30 19:39:49 +0100614check_cursor(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000615{
616 check_cursor_lnum();
617 check_cursor_col();
618}
619
620#if defined(FEAT_TEXTOBJ) || defined(PROTO)
621/*
622 * Make sure curwin->w_cursor is not on the NUL at the end of the line.
623 * Allow it when in Visual mode and 'selection' is not "old".
624 */
625 void
Bram Moolenaar9b578142016-01-30 19:39:49 +0100626adjust_cursor_col(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000627{
628 if (curwin->w_cursor.col > 0
Bram Moolenaar071d4272004-06-13 20:20:40 +0000629 && (!VIsual_active || *p_sel == 'o')
Bram Moolenaar071d4272004-06-13 20:20:40 +0000630 && gchar_cursor() == NUL)
631 --curwin->w_cursor.col;
632}
633#endif
634
635/*
636 * When curwin->w_leftcol has changed, adjust the cursor position.
637 * Return TRUE if the cursor was moved.
638 */
639 int
Bram Moolenaar9b578142016-01-30 19:39:49 +0100640leftcol_changed(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000641{
642 long lastcol;
643 colnr_T s, e;
644 int retval = FALSE;
Bram Moolenaar375e3392019-01-31 18:26:10 +0100645 long siso = get_sidescrolloff_value();
Bram Moolenaar071d4272004-06-13 20:20:40 +0000646
647 changed_cline_bef_curs();
Bram Moolenaar02631462017-09-22 15:20:32 +0200648 lastcol = curwin->w_leftcol + curwin->w_width - curwin_col_off() - 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000649 validate_virtcol();
650
651 /*
652 * If the cursor is right or left of the screen, move it to last or first
653 * character.
654 */
Bram Moolenaar375e3392019-01-31 18:26:10 +0100655 if (curwin->w_virtcol > (colnr_T)(lastcol - siso))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000656 {
657 retval = TRUE;
Bram Moolenaar375e3392019-01-31 18:26:10 +0100658 coladvance((colnr_T)(lastcol - siso));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000659 }
Bram Moolenaar375e3392019-01-31 18:26:10 +0100660 else if (curwin->w_virtcol < curwin->w_leftcol + siso)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000661 {
662 retval = TRUE;
Bram Moolenaar375e3392019-01-31 18:26:10 +0100663 (void)coladvance((colnr_T)(curwin->w_leftcol + siso));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000664 }
665
666 /*
667 * If the start of the character under the cursor is not on the screen,
668 * advance the cursor one more char. If this fails (last char of the
669 * line) adjust the scrolling.
670 */
671 getvvcol(curwin, &curwin->w_cursor, &s, NULL, &e);
672 if (e > (colnr_T)lastcol)
673 {
674 retval = TRUE;
675 coladvance(s - 1);
676 }
677 else if (s < curwin->w_leftcol)
678 {
679 retval = TRUE;
Bram Moolenaar85a20022019-12-21 18:25:54 +0100680 if (coladvance(e + 1) == FAIL) // there isn't another character
Bram Moolenaar071d4272004-06-13 20:20:40 +0000681 {
Bram Moolenaar85a20022019-12-21 18:25:54 +0100682 curwin->w_leftcol = s; // adjust w_leftcol instead
Bram Moolenaar071d4272004-06-13 20:20:40 +0000683 changed_cline_bef_curs();
684 }
685 }
686
687 if (retval)
688 curwin->w_set_curswant = TRUE;
689 redraw_later(NOT_VALID);
690 return retval;
691}
692
693/**********************************************************************
694 * Various routines dealing with allocation and deallocation of memory.
695 */
696
697#if defined(MEM_PROFILE) || defined(PROTO)
698
699# define MEM_SIZES 8200
700static long_u mem_allocs[MEM_SIZES];
701static long_u mem_frees[MEM_SIZES];
702static long_u mem_allocated;
703static long_u mem_freed;
704static long_u mem_peak;
705static long_u num_alloc;
706static long_u num_freed;
707
Bram Moolenaar071d4272004-06-13 20:20:40 +0000708 static void
Bram Moolenaar9b578142016-01-30 19:39:49 +0100709mem_pre_alloc_s(size_t *sizep)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000710{
711 *sizep += sizeof(size_t);
712}
713
714 static void
Bram Moolenaar964b3742019-05-24 18:54:09 +0200715mem_pre_alloc_l(size_t *sizep)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000716{
717 *sizep += sizeof(size_t);
718}
719
720 static void
Bram Moolenaar9b578142016-01-30 19:39:49 +0100721mem_post_alloc(
722 void **pp,
723 size_t size)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000724{
725 if (*pp == NULL)
726 return;
727 size -= sizeof(size_t);
728 *(long_u *)*pp = size;
729 if (size <= MEM_SIZES-1)
730 mem_allocs[size-1]++;
731 else
732 mem_allocs[MEM_SIZES-1]++;
733 mem_allocated += size;
734 if (mem_allocated - mem_freed > mem_peak)
735 mem_peak = mem_allocated - mem_freed;
736 num_alloc++;
737 *pp = (void *)((char *)*pp + sizeof(size_t));
738}
739
740 static void
Bram Moolenaar9b578142016-01-30 19:39:49 +0100741mem_pre_free(void **pp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000742{
743 long_u size;
744
745 *pp = (void *)((char *)*pp - sizeof(size_t));
746 size = *(size_t *)*pp;
747 if (size <= MEM_SIZES-1)
748 mem_frees[size-1]++;
749 else
750 mem_frees[MEM_SIZES-1]++;
751 mem_freed += size;
752 num_freed++;
753}
754
755/*
756 * called on exit via atexit()
757 */
758 void
Bram Moolenaar9b578142016-01-30 19:39:49 +0100759vim_mem_profile_dump(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000760{
761 int i, j;
762
763 printf("\r\n");
764 j = 0;
765 for (i = 0; i < MEM_SIZES - 1; i++)
766 {
767 if (mem_allocs[i] || mem_frees[i])
768 {
769 if (mem_frees[i] > mem_allocs[i])
770 printf("\r\n%s", _("ERROR: "));
771 printf("[%4d / %4lu-%-4lu] ", i + 1, mem_allocs[i], mem_frees[i]);
772 j++;
773 if (j > 3)
774 {
775 j = 0;
776 printf("\r\n");
777 }
778 }
779 }
780
781 i = MEM_SIZES - 1;
782 if (mem_allocs[i])
783 {
784 printf("\r\n");
785 if (mem_frees[i] > mem_allocs[i])
Bram Moolenaar2f1e0502010-08-13 11:18:02 +0200786 puts(_("ERROR: "));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000787 printf("[>%d / %4lu-%-4lu]", i, mem_allocs[i], mem_frees[i]);
788 }
789
790 printf(_("\n[bytes] total alloc-freed %lu-%lu, in use %lu, peak use %lu\n"),
791 mem_allocated, mem_freed, mem_allocated - mem_freed, mem_peak);
792 printf(_("[calls] total re/malloc()'s %lu, total free()'s %lu\n\n"),
793 num_alloc, num_freed);
794}
795
Bram Moolenaar85a20022019-12-21 18:25:54 +0100796#endif // MEM_PROFILE
Bram Moolenaar071d4272004-06-13 20:20:40 +0000797
Bram Moolenaar75bdf6a2016-01-07 21:25:08 +0100798#ifdef FEAT_EVAL
Bram Moolenaarf49cc602018-11-11 15:21:05 +0100799 int
Bram Moolenaar964b3742019-05-24 18:54:09 +0200800alloc_does_fail(size_t size)
Bram Moolenaar75bdf6a2016-01-07 21:25:08 +0100801{
802 if (alloc_fail_countdown == 0)
803 {
804 if (--alloc_fail_repeat <= 0)
805 alloc_fail_id = 0;
Bram Moolenaara260b872016-01-15 20:48:22 +0100806 do_outofmem_msg(size);
Bram Moolenaar75bdf6a2016-01-07 21:25:08 +0100807 return TRUE;
808 }
809 --alloc_fail_countdown;
810 return FALSE;
811}
812#endif
813
Bram Moolenaar071d4272004-06-13 20:20:40 +0000814/*
815 * Some memory is reserved for error messages and for being able to
816 * call mf_release_all(), which needs some memory for mf_trans_add().
817 */
Bram Moolenaar48e330a2016-02-23 14:53:34 +0100818#define KEEP_ROOM (2 * 8192L)
Bram Moolenaar11b73d62012-06-29 15:51:30 +0200819#define KEEP_ROOM_KB (KEEP_ROOM / 1024L)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000820
821/*
Bram Moolenaar964b3742019-05-24 18:54:09 +0200822 * The normal way to allocate memory. This handles an out-of-memory situation
823 * as well as possible, still returns NULL when we're completely out.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000824 */
Bram Moolenaarc799fe22019-05-28 23:08:19 +0200825 void *
Bram Moolenaar964b3742019-05-24 18:54:09 +0200826alloc(size_t size)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000827{
Bram Moolenaar964b3742019-05-24 18:54:09 +0200828 return lalloc(size, TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000829}
830
831/*
Bram Moolenaar75bdf6a2016-01-07 21:25:08 +0100832 * alloc() with an ID for alloc_fail().
Bram Moolenaar75bdf6a2016-01-07 21:25:08 +0100833 */
Bram Moolenaarc799fe22019-05-28 23:08:19 +0200834 void *
Bram Moolenaar964b3742019-05-24 18:54:09 +0200835alloc_id(size_t size, alloc_id_T id UNUSED)
Bram Moolenaar75bdf6a2016-01-07 21:25:08 +0100836{
837#ifdef FEAT_EVAL
Bram Moolenaar964b3742019-05-24 18:54:09 +0200838 if (alloc_fail_id == id && alloc_does_fail(size))
Bram Moolenaar75bdf6a2016-01-07 21:25:08 +0100839 return NULL;
840#endif
Bram Moolenaar964b3742019-05-24 18:54:09 +0200841 return lalloc(size, TRUE);
Bram Moolenaar75bdf6a2016-01-07 21:25:08 +0100842}
843
844/*
Bram Moolenaar071d4272004-06-13 20:20:40 +0000845 * Allocate memory and set all bytes to zero.
846 */
Bram Moolenaarc799fe22019-05-28 23:08:19 +0200847 void *
Bram Moolenaar964b3742019-05-24 18:54:09 +0200848alloc_clear(size_t size)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000849{
Bram Moolenaarc799fe22019-05-28 23:08:19 +0200850 void *p;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000851
Bram Moolenaar964b3742019-05-24 18:54:09 +0200852 p = lalloc(size, TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000853 if (p != NULL)
Bram Moolenaar964b3742019-05-24 18:54:09 +0200854 (void)vim_memset(p, 0, size);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000855 return p;
856}
857
858/*
Bram Moolenaar162b7142018-12-21 15:17:36 +0100859 * Same as alloc_clear() but with allocation id for testing
860 */
Bram Moolenaarc799fe22019-05-28 23:08:19 +0200861 void *
Bram Moolenaar964b3742019-05-24 18:54:09 +0200862alloc_clear_id(size_t size, alloc_id_T id UNUSED)
Bram Moolenaar162b7142018-12-21 15:17:36 +0100863{
864#ifdef FEAT_EVAL
Bram Moolenaar964b3742019-05-24 18:54:09 +0200865 if (alloc_fail_id == id && alloc_does_fail(size))
Bram Moolenaar162b7142018-12-21 15:17:36 +0100866 return NULL;
867#endif
868 return alloc_clear(size);
869}
870
871/*
Bram Moolenaar071d4272004-06-13 20:20:40 +0000872 * Allocate memory like lalloc() and set all bytes to zero.
873 */
Bram Moolenaarc799fe22019-05-28 23:08:19 +0200874 void *
Bram Moolenaar964b3742019-05-24 18:54:09 +0200875lalloc_clear(size_t size, int message)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000876{
Bram Moolenaarc799fe22019-05-28 23:08:19 +0200877 void *p;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000878
Bram Moolenaarc799fe22019-05-28 23:08:19 +0200879 p = lalloc(size, message);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000880 if (p != NULL)
Bram Moolenaar964b3742019-05-24 18:54:09 +0200881 (void)vim_memset(p, 0, size);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000882 return p;
883}
884
885/*
886 * Low level memory allocation function.
887 * This is used often, KEEP IT FAST!
888 */
Bram Moolenaarc799fe22019-05-28 23:08:19 +0200889 void *
Bram Moolenaar964b3742019-05-24 18:54:09 +0200890lalloc(size_t size, int message)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000891{
Bram Moolenaar85a20022019-12-21 18:25:54 +0100892 void *p; // pointer to new storage space
893 static int releasing = FALSE; // don't do mf_release_all() recursive
Bram Moolenaar071d4272004-06-13 20:20:40 +0000894 int try_again;
Bram Moolenaar48e330a2016-02-23 14:53:34 +0100895#if defined(HAVE_AVAIL_MEM)
Bram Moolenaar85a20022019-12-21 18:25:54 +0100896 static size_t allocated = 0; // allocated since last avail check
Bram Moolenaar071d4272004-06-13 20:20:40 +0000897#endif
898
Bram Moolenaar964b3742019-05-24 18:54:09 +0200899 // Safety check for allocating zero bytes
Bram Moolenaar071d4272004-06-13 20:20:40 +0000900 if (size == 0)
901 {
Bram Moolenaar964b3742019-05-24 18:54:09 +0200902 // Don't hide this message
Bram Moolenaar071d4272004-06-13 20:20:40 +0000903 emsg_silent = 0;
Bram Moolenaar964b3742019-05-24 18:54:09 +0200904 iemsg(_("E341: Internal error: lalloc(0, )"));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000905 return NULL;
906 }
907
908#ifdef MEM_PROFILE
909 mem_pre_alloc_l(&size);
910#endif
911
Bram Moolenaar071d4272004-06-13 20:20:40 +0000912 /*
913 * Loop when out of memory: Try to release some memfile blocks and
914 * if some blocks are released call malloc again.
915 */
916 for (;;)
917 {
918 /*
919 * Handle three kind of systems:
920 * 1. No check for available memory: Just return.
921 * 2. Slow check for available memory: call mch_avail_mem() after
922 * allocating KEEP_ROOM amount of memory.
923 * 3. Strict check for available memory: call mch_avail_mem()
924 */
Bram Moolenaarc799fe22019-05-28 23:08:19 +0200925 if ((p = malloc(size)) != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000926 {
927#ifndef HAVE_AVAIL_MEM
Bram Moolenaar85a20022019-12-21 18:25:54 +0100928 // 1. No check for available memory: Just return.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000929 goto theend;
930#else
Bram Moolenaar85a20022019-12-21 18:25:54 +0100931 // 2. Slow check for available memory: call mch_avail_mem() after
932 // allocating (KEEP_ROOM / 2) amount of memory.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000933 allocated += size;
934 if (allocated < KEEP_ROOM / 2)
935 goto theend;
936 allocated = 0;
Bram Moolenaar48e330a2016-02-23 14:53:34 +0100937
Bram Moolenaar85a20022019-12-21 18:25:54 +0100938 // 3. check for available memory: call mch_avail_mem()
Bram Moolenaar11b73d62012-06-29 15:51:30 +0200939 if (mch_avail_mem(TRUE) < KEEP_ROOM_KB && !releasing)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000940 {
Bram Moolenaar85a20022019-12-21 18:25:54 +0100941 free(p); // System is low... no go!
Bram Moolenaar071d4272004-06-13 20:20:40 +0000942 p = NULL;
943 }
944 else
945 goto theend;
946#endif
947 }
948 /*
949 * Remember that mf_release_all() is being called to avoid an endless
950 * loop, because mf_release_all() may call alloc() recursively.
951 */
952 if (releasing)
953 break;
954 releasing = TRUE;
Bram Moolenaar661b1822005-07-28 22:36:45 +0000955
Bram Moolenaar85a20022019-12-21 18:25:54 +0100956 clear_sb_text(TRUE); // free any scrollback text
957 try_again = mf_release_all(); // release as many blocks as possible
Bram Moolenaar661b1822005-07-28 22:36:45 +0000958
Bram Moolenaar071d4272004-06-13 20:20:40 +0000959 releasing = FALSE;
960 if (!try_again)
961 break;
962 }
963
964 if (message && p == NULL)
965 do_outofmem_msg(size);
966
967theend:
968#ifdef MEM_PROFILE
Bram Moolenaarc799fe22019-05-28 23:08:19 +0200969 mem_post_alloc(&p, size);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000970#endif
971 return p;
972}
973
Bram Moolenaar75bdf6a2016-01-07 21:25:08 +0100974/*
975 * lalloc() with an ID for alloc_fail().
Bram Moolenaar75bdf6a2016-01-07 21:25:08 +0100976 */
Bram Moolenaar113e1072019-01-20 15:30:40 +0100977#if defined(FEAT_SIGNS) || defined(PROTO)
Bram Moolenaarc799fe22019-05-28 23:08:19 +0200978 void *
Bram Moolenaar964b3742019-05-24 18:54:09 +0200979lalloc_id(size_t size, int message, alloc_id_T id UNUSED)
Bram Moolenaar75bdf6a2016-01-07 21:25:08 +0100980{
981#ifdef FEAT_EVAL
Bram Moolenaara260b872016-01-15 20:48:22 +0100982 if (alloc_fail_id == id && alloc_does_fail(size))
Bram Moolenaar75bdf6a2016-01-07 21:25:08 +0100983 return NULL;
984#endif
Bram Moolenaar964b3742019-05-24 18:54:09 +0200985 return (lalloc(size, message));
Bram Moolenaar75bdf6a2016-01-07 21:25:08 +0100986}
Bram Moolenaar113e1072019-01-20 15:30:40 +0100987#endif
Bram Moolenaar75bdf6a2016-01-07 21:25:08 +0100988
Bram Moolenaar071d4272004-06-13 20:20:40 +0000989#if defined(MEM_PROFILE) || defined(PROTO)
990/*
991 * realloc() with memory profiling.
992 */
993 void *
Bram Moolenaar9b578142016-01-30 19:39:49 +0100994mem_realloc(void *ptr, size_t size)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000995{
996 void *p;
997
998 mem_pre_free(&ptr);
999 mem_pre_alloc_s(&size);
1000
1001 p = realloc(ptr, size);
1002
1003 mem_post_alloc(&p, size);
1004
1005 return p;
1006}
1007#endif
1008
1009/*
1010* Avoid repeating the error message many times (they take 1 second each).
1011* Did_outofmem_msg is reset when a character is read.
1012*/
1013 void
Bram Moolenaar964b3742019-05-24 18:54:09 +02001014do_outofmem_msg(size_t size)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001015{
1016 if (!did_outofmem_msg)
1017 {
Bram Moolenaar4dc8f492019-08-21 13:06:55 +02001018 // Don't hide this message
Bram Moolenaar071d4272004-06-13 20:20:40 +00001019 emsg_silent = 0;
Bram Moolenaar79739e12011-10-26 11:41:00 +02001020
Bram Moolenaar4dc8f492019-08-21 13:06:55 +02001021 // Must come first to avoid coming back here when printing the error
1022 // message fails, e.g. when setting v:errmsg.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001023 did_outofmem_msg = TRUE;
Bram Moolenaar79739e12011-10-26 11:41:00 +02001024
Bram Moolenaar964b3742019-05-24 18:54:09 +02001025 semsg(_("E342: Out of memory! (allocating %lu bytes)"), (long_u)size);
Bram Moolenaar4dc8f492019-08-21 13:06:55 +02001026
1027 if (starting == NO_SCREEN)
1028 // Not even finished with initializations and already out of
1029 // memory? Then nothing is going to work, exit.
1030 mch_exit(123);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001031 }
1032}
1033
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001034#if defined(EXITFREE) || defined(PROTO)
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001035
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001036/*
1037 * Free everything that we allocated.
1038 * Can be used to detect memory leaks, e.g., with ccmalloc.
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001039 * NOTE: This is tricky! Things are freed that functions depend on. Don't be
1040 * surprised if Vim crashes...
1041 * Some things can't be freed, esp. things local to a library function.
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001042 */
1043 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01001044free_all_mem(void)
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001045{
1046 buf_T *buf, *nextbuf;
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001047
Bram Moolenaar85a20022019-12-21 18:25:54 +01001048 // When we cause a crash here it is caught and Vim tries to exit cleanly.
1049 // Don't try freeing everything again.
Bram Moolenaarb89a25f2016-06-01 23:08:39 +02001050 if (entered_free_all_mem)
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001051 return;
Bram Moolenaarb89a25f2016-06-01 23:08:39 +02001052 entered_free_all_mem = TRUE;
Bram Moolenaar85a20022019-12-21 18:25:54 +01001053 // Don't want to trigger autocommands from here on.
Bram Moolenaar5d2bae82014-09-19 14:26:36 +02001054 block_autocmds();
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001055
Bram Moolenaar85a20022019-12-21 18:25:54 +01001056 // Close all tabs and windows. Reset 'equalalways' to avoid redraws.
Bram Moolenaar5bedfc62010-07-20 22:30:01 +02001057 p_ea = FALSE;
Bram Moolenaare5c83282019-05-03 23:15:37 +02001058 if (first_tabpage != NULL && first_tabpage->tp_next != NULL)
Bram Moolenaar7d47b6e2006-03-15 22:59:18 +00001059 do_cmdline_cmd((char_u *)"tabonly!");
Bram Moolenaar95f09602016-11-10 20:01:45 +01001060 if (!ONE_WINDOW)
Bram Moolenaar7d47b6e2006-03-15 22:59:18 +00001061 do_cmdline_cmd((char_u *)"only!");
Bram Moolenaar997fb4b2006-02-17 21:53:23 +00001062
Bram Moolenaarc4956c82006-03-12 21:58:43 +00001063# if defined(FEAT_SPELL)
Bram Moolenaar85a20022019-12-21 18:25:54 +01001064 // Free all spell info.
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001065 spell_free_all();
1066# endif
1067
Bram Moolenaare2c453d2019-08-21 14:37:09 +02001068# if defined(FEAT_BEVAL_TERM)
Bram Moolenaarb301f6b2018-02-10 15:38:35 +01001069 ui_remove_balloon();
1070# endif
Bram Moolenaar03a9f842020-05-13 13:40:16 +02001071# ifdef FEAT_PROP_POPUP
Bram Moolenaar06f08532020-05-12 23:45:16 +02001072 if (curwin != NULL)
Bram Moolenaar03a9f842020-05-13 13:40:16 +02001073 close_all_popups(TRUE);
Bram Moolenaar06f08532020-05-12 23:45:16 +02001074# endif
Bram Moolenaarb301f6b2018-02-10 15:38:35 +01001075
Bram Moolenaarac9fb182019-04-27 13:04:13 +02001076 // Clear user commands (before deleting buffers).
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001077 ex_comclear(NULL);
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001078
Bram Moolenaare5c83282019-05-03 23:15:37 +02001079 // When exiting from mainerr_arg_missing curbuf has not been initialized,
1080 // and not much else.
1081 if (curbuf != NULL)
1082 {
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001083# ifdef FEAT_MENU
Bram Moolenaare5c83282019-05-03 23:15:37 +02001084 // Clear menus.
1085 do_cmdline_cmd((char_u *)"aunmenu *");
Bram Moolenaar21160b92009-11-11 15:56:10 +00001086# ifdef FEAT_MULTI_LANG
Bram Moolenaare5c83282019-05-03 23:15:37 +02001087 do_cmdline_cmd((char_u *)"menutranslate clear");
Bram Moolenaar21160b92009-11-11 15:56:10 +00001088# endif
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001089# endif
Bram Moolenaare5c83282019-05-03 23:15:37 +02001090 // Clear mappings, abbreviations, breakpoints.
1091 do_cmdline_cmd((char_u *)"lmapclear");
1092 do_cmdline_cmd((char_u *)"xmapclear");
1093 do_cmdline_cmd((char_u *)"mapclear");
1094 do_cmdline_cmd((char_u *)"mapclear!");
1095 do_cmdline_cmd((char_u *)"abclear");
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001096# if defined(FEAT_EVAL)
Bram Moolenaare5c83282019-05-03 23:15:37 +02001097 do_cmdline_cmd((char_u *)"breakdel *");
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001098# endif
Bram Moolenaar1e498f52005-06-26 22:29:44 +00001099# if defined(FEAT_PROFILE)
Bram Moolenaare5c83282019-05-03 23:15:37 +02001100 do_cmdline_cmd((char_u *)"profdel *");
Bram Moolenaar1e498f52005-06-26 22:29:44 +00001101# endif
Bram Moolenaarde0dfed2009-02-24 03:30:14 +00001102# if defined(FEAT_KEYMAP)
Bram Moolenaare5c83282019-05-03 23:15:37 +02001103 do_cmdline_cmd((char_u *)"set keymap=");
Bram Moolenaar48ac6712019-07-04 20:26:21 +02001104# endif
Bram Moolenaare5c83282019-05-03 23:15:37 +02001105 }
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001106
1107# ifdef FEAT_TITLE
1108 free_titles();
1109# endif
1110# if defined(FEAT_SEARCHPATH)
1111 free_findfile();
1112# endif
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001113
Bram Moolenaar85a20022019-12-21 18:25:54 +01001114 // Obviously named calls.
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001115 free_all_autocmds();
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001116 clear_termcodes();
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001117 free_all_marks();
1118 alist_clear(&global_alist);
1119 free_homedir();
Bram Moolenaar24305862012-08-15 14:05:05 +02001120 free_users();
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001121 free_search_patterns();
1122 free_old_sub();
1123 free_last_insert();
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001124 free_insexpand_stuff();
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001125 free_prev_shellcmd();
1126 free_regexp_stuff();
1127 free_tag_stuff();
1128 free_cd_dir();
Bram Moolenaarde0dfed2009-02-24 03:30:14 +00001129# ifdef FEAT_SIGNS
1130 free_signs();
1131# endif
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001132# ifdef FEAT_EVAL
Bram Moolenaarb4bcea42020-10-28 13:53:50 +01001133 set_expr_line(NULL, NULL);
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001134# endif
1135# ifdef FEAT_DIFF
Bram Moolenaare5c83282019-05-03 23:15:37 +02001136 if (curtab != NULL)
1137 diff_clear(curtab);
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001138# endif
Bram Moolenaar85a20022019-12-21 18:25:54 +01001139 clear_sb_text(TRUE); // free any scrollback text
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001140
Bram Moolenaar85a20022019-12-21 18:25:54 +01001141 // Free some global vars.
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001142 vim_free(username);
Bram Moolenaaraf92ee82007-10-07 13:45:08 +00001143# ifdef FEAT_CLIPBOARD
Bram Moolenaar473de612013-06-08 18:19:48 +02001144 vim_regfree(clip_exclude_prog);
Bram Moolenaaraf92ee82007-10-07 13:45:08 +00001145# endif
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001146 vim_free(last_cmdline);
1147 vim_free(new_last_cmdline);
Bram Moolenaar238a5642006-02-21 22:12:05 +00001148 set_keep_msg(NULL, 0);
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001149
Bram Moolenaar85a20022019-12-21 18:25:54 +01001150 // Clear cmdline history.
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001151 p_hi = 0;
1152 init_history();
Bram Moolenaar05ad5ff2019-11-30 22:48:27 +01001153# ifdef FEAT_PROP_POPUP
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001154 clear_global_prop_types();
Bram Moolenaar48ac6712019-07-04 20:26:21 +02001155# endif
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001156
Bram Moolenaar48ac6712019-07-04 20:26:21 +02001157# ifdef FEAT_QUICKFIX
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001158 {
Bram Moolenaarbd1e5d22009-04-29 09:02:44 +00001159 win_T *win;
1160 tabpage_T *tab;
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001161
1162 qf_free_all(NULL);
Bram Moolenaare5c83282019-05-03 23:15:37 +02001163 // Free all location lists
Bram Moolenaarbd1e5d22009-04-29 09:02:44 +00001164 FOR_ALL_TAB_WINDOWS(tab, win)
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001165 qf_free_all(win);
1166 }
Bram Moolenaar48ac6712019-07-04 20:26:21 +02001167# endif
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001168
Bram Moolenaare5c83282019-05-03 23:15:37 +02001169 // Close all script inputs.
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001170 close_all_scripts();
1171
Bram Moolenaare5c83282019-05-03 23:15:37 +02001172 if (curwin != NULL)
1173 // Destroy all windows. Must come before freeing buffers.
1174 win_free_all();
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001175
Bram Moolenaar85a20022019-12-21 18:25:54 +01001176 // Free all option values. Must come after closing windows.
Bram Moolenaar4f198282017-10-23 21:53:30 +02001177 free_all_options();
1178
Bram Moolenaar85a20022019-12-21 18:25:54 +01001179 // Free all buffers. Reset 'autochdir' to avoid accessing things that
1180 // were freed already.
Bram Moolenaar48ac6712019-07-04 20:26:21 +02001181# ifdef FEAT_AUTOCHDIR
Bram Moolenaar2a329742008-04-01 12:53:43 +00001182 p_acd = FALSE;
Bram Moolenaar48ac6712019-07-04 20:26:21 +02001183# endif
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001184 for (buf = firstbuf; buf != NULL; )
1185 {
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001186 bufref_T bufref;
1187
1188 set_bufref(&bufref, buf);
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001189 nextbuf = buf->b_next;
Bram Moolenaara6e8f882019-12-14 16:18:15 +01001190 close_buffer(NULL, buf, DOBUF_WIPE, FALSE, FALSE);
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001191 if (bufref_valid(&bufref))
Bram Moolenaar85a20022019-12-21 18:25:54 +01001192 buf = nextbuf; // didn't work, try next one
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001193 else
1194 buf = firstbuf;
1195 }
1196
Bram Moolenaar48ac6712019-07-04 20:26:21 +02001197# ifdef FEAT_ARABIC
1198 free_arshape_buf();
1199# endif
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001200
Bram Moolenaar85a20022019-12-21 18:25:54 +01001201 // Clear registers.
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001202 clear_registers();
1203 ResetRedobuff();
1204 ResetRedobuff();
1205
Bram Moolenaar9f84ded2018-10-20 20:54:02 +02001206# if defined(FEAT_CLIENTSERVER) && defined(FEAT_X11)
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001207 vim_free(serverDelayedStartName);
Bram Moolenaar9f84ded2018-10-20 20:54:02 +02001208# endif
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001209
Bram Moolenaar85a20022019-12-21 18:25:54 +01001210 // highlight info
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001211 free_highlight();
1212
Bram Moolenaar1e498f52005-06-26 22:29:44 +00001213 reset_last_sourcing();
1214
Bram Moolenaare5c83282019-05-03 23:15:37 +02001215 if (first_tabpage != NULL)
1216 {
1217 free_tabpage(first_tabpage);
1218 first_tabpage = NULL;
1219 }
Bram Moolenaar7d47b6e2006-03-15 22:59:18 +00001220
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001221# ifdef UNIX
Bram Moolenaar85a20022019-12-21 18:25:54 +01001222 // Machine-specific free.
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001223 mch_free_mem();
1224# endif
1225
Bram Moolenaar85a20022019-12-21 18:25:54 +01001226 // message history
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001227 for (;;)
1228 if (delete_first_msg() == FAIL)
1229 break;
1230
Bram Moolenaar655da312016-05-28 22:22:34 +02001231# ifdef FEAT_JOB_CHANNEL
1232 channel_free_all();
Bram Moolenaar655da312016-05-28 22:22:34 +02001233# endif
Bram Moolenaar9f84ded2018-10-20 20:54:02 +02001234# ifdef FEAT_TIMERS
Bram Moolenaar623e2632016-07-30 22:47:56 +02001235 timer_free_all();
Bram Moolenaar9f84ded2018-10-20 20:54:02 +02001236# endif
Bram Moolenaar437bafe2016-08-01 15:40:54 +02001237# ifdef FEAT_EVAL
Bram Moolenaar85a20022019-12-21 18:25:54 +01001238 // must be after channel_free_all() with unrefs partials
Bram Moolenaar437bafe2016-08-01 15:40:54 +02001239 eval_clear();
1240# endif
1241# ifdef FEAT_JOB_CHANNEL
Bram Moolenaar85a20022019-12-21 18:25:54 +01001242 // must be after eval_clear() with unrefs jobs
Bram Moolenaar437bafe2016-08-01 15:40:54 +02001243 job_free_all();
1244# endif
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001245
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001246 free_termoptions();
1247
Bram Moolenaar85a20022019-12-21 18:25:54 +01001248 // screenlines (can't display anything now!)
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001249 free_screenlines();
1250
Bram Moolenaar82febc12019-06-10 14:48:59 +02001251# if defined(FEAT_SOUND)
1252 sound_free();
1253# endif
Bram Moolenaar9f84ded2018-10-20 20:54:02 +02001254# if defined(USE_XSMP)
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001255 xsmp_close();
Bram Moolenaar9f84ded2018-10-20 20:54:02 +02001256# endif
1257# ifdef FEAT_GUI_GTK
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001258 gui_mch_free_all();
Bram Moolenaar9f84ded2018-10-20 20:54:02 +02001259# endif
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001260 clear_hl_tables();
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001261
1262 vim_free(IObuff);
1263 vim_free(NameBuff);
Bram Moolenaar9f84ded2018-10-20 20:54:02 +02001264# ifdef FEAT_QUICKFIX
1265 check_quickfix_busy();
1266# endif
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001267}
1268#endif
1269
Bram Moolenaar071d4272004-06-13 20:20:40 +00001270/*
Bram Moolenaare980d8a2010-12-08 13:11:21 +01001271 * Copy "string" into newly allocated memory.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001272 */
1273 char_u *
Bram Moolenaar9b578142016-01-30 19:39:49 +01001274vim_strsave(char_u *string)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001275{
1276 char_u *p;
Bram Moolenaar964b3742019-05-24 18:54:09 +02001277 size_t len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001278
Bram Moolenaar964b3742019-05-24 18:54:09 +02001279 len = STRLEN(string) + 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001280 p = alloc(len);
1281 if (p != NULL)
Bram Moolenaar964b3742019-05-24 18:54:09 +02001282 mch_memmove(p, string, len);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001283 return p;
1284}
1285
Bram Moolenaare980d8a2010-12-08 13:11:21 +01001286/*
1287 * Copy up to "len" bytes of "string" into newly allocated memory and
1288 * terminate with a NUL.
1289 * The allocated memory always has size "len + 1", also when "string" is
1290 * shorter.
1291 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001292 char_u *
Bram Moolenaardf44a272020-06-07 20:49:05 +02001293vim_strnsave(char_u *string, size_t len)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001294{
1295 char_u *p;
1296
Bram Moolenaar51e14382019-05-25 20:21:28 +02001297 p = alloc(len + 1);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001298 if (p != NULL)
1299 {
1300 STRNCPY(p, string, len);
1301 p[len] = NUL;
1302 }
1303 return p;
1304}
1305
Bram Moolenaar071d4272004-06-13 20:20:40 +00001306/*
Bram Moolenaarc1a9bc12018-12-28 21:59:29 +01001307 * Copy "p[len]" into allocated memory, ignoring NUL characters.
1308 * Returns NULL when out of memory.
1309 */
1310 char_u *
Bram Moolenaar964b3742019-05-24 18:54:09 +02001311vim_memsave(char_u *p, size_t len)
Bram Moolenaarc1a9bc12018-12-28 21:59:29 +01001312{
Bram Moolenaar964b3742019-05-24 18:54:09 +02001313 char_u *ret = alloc(len);
Bram Moolenaarc1a9bc12018-12-28 21:59:29 +01001314
1315 if (ret != NULL)
Bram Moolenaar964b3742019-05-24 18:54:09 +02001316 mch_memmove(ret, p, len);
Bram Moolenaarc1a9bc12018-12-28 21:59:29 +01001317 return ret;
1318}
1319
1320/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00001321 * Same as vim_strsave(), but any characters found in esc_chars are preceded
1322 * by a backslash.
1323 */
1324 char_u *
Bram Moolenaar9b578142016-01-30 19:39:49 +01001325vim_strsave_escaped(char_u *string, char_u *esc_chars)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001326{
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001327 return vim_strsave_escaped_ext(string, esc_chars, '\\', FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001328}
1329
1330/*
1331 * Same as vim_strsave_escaped(), but when "bsl" is TRUE also escape
1332 * characters where rem_backslash() would remove the backslash.
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001333 * Escape the characters with "cc".
Bram Moolenaar071d4272004-06-13 20:20:40 +00001334 */
1335 char_u *
Bram Moolenaar9b578142016-01-30 19:39:49 +01001336vim_strsave_escaped_ext(
1337 char_u *string,
1338 char_u *esc_chars,
1339 int cc,
1340 int bsl)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001341{
1342 char_u *p;
1343 char_u *p2;
1344 char_u *escaped_string;
1345 unsigned length;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001346 int l;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001347
1348 /*
1349 * First count the number of backslashes required.
1350 * Then allocate the memory and insert them.
1351 */
Bram Moolenaar85a20022019-12-21 18:25:54 +01001352 length = 1; // count the trailing NUL
Bram Moolenaar071d4272004-06-13 20:20:40 +00001353 for (p = string; *p; p++)
1354 {
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001355 if (has_mbyte && (l = (*mb_ptr2len)(p)) > 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001356 {
Bram Moolenaar85a20022019-12-21 18:25:54 +01001357 length += l; // count a multibyte char
Bram Moolenaar071d4272004-06-13 20:20:40 +00001358 p += l - 1;
1359 continue;
1360 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001361 if (vim_strchr(esc_chars, *p) != NULL || (bsl && rem_backslash(p)))
Bram Moolenaar85a20022019-12-21 18:25:54 +01001362 ++length; // count a backslash
1363 ++length; // count an ordinary char
Bram Moolenaar071d4272004-06-13 20:20:40 +00001364 }
1365 escaped_string = alloc(length);
1366 if (escaped_string != NULL)
1367 {
1368 p2 = escaped_string;
1369 for (p = string; *p; p++)
1370 {
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001371 if (has_mbyte && (l = (*mb_ptr2len)(p)) > 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001372 {
1373 mch_memmove(p2, p, (size_t)l);
1374 p2 += l;
Bram Moolenaar85a20022019-12-21 18:25:54 +01001375 p += l - 1; // skip multibyte char
Bram Moolenaar071d4272004-06-13 20:20:40 +00001376 continue;
1377 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001378 if (vim_strchr(esc_chars, *p) != NULL || (bsl && rem_backslash(p)))
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001379 *p2++ = cc;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001380 *p2++ = *p;
1381 }
1382 *p2 = NUL;
1383 }
1384 return escaped_string;
1385}
1386
Bram Moolenaar7693ec62008-07-24 18:29:37 +00001387/*
1388 * Return TRUE when 'shell' has "csh" in the tail.
1389 */
1390 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01001391csh_like_shell(void)
Bram Moolenaar7693ec62008-07-24 18:29:37 +00001392{
1393 return (strstr((char *)gettail(p_sh), "csh") != NULL);
1394}
Bram Moolenaar7693ec62008-07-24 18:29:37 +00001395
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001396/*
1397 * Escape "string" for use as a shell argument with system().
Bram Moolenaar21160b92009-11-11 15:56:10 +00001398 * This uses single quotes, except when we know we need to use double quotes
Mike Williams12795022021-06-28 20:53:58 +02001399 * (MS-DOS and MS-Windows not using PowerShell and without 'shellslash' set).
1400 * PowerShell also uses a novel escaping for enclosed single quotes - double
1401 * them up.
Bram Moolenaarffd66c42008-07-16 20:43:37 +00001402 * Escape a newline, depending on the 'shell' option.
1403 * When "do_special" is TRUE also replace "!", "%", "#" and things starting
1404 * with "<" like "<cfile>".
Bram Moolenaar26df0922014-02-23 23:39:13 +01001405 * When "do_newline" is FALSE do not escape newline unless it is csh shell.
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001406 * Returns the result in allocated memory, NULL if we have run out.
1407 */
1408 char_u *
Bram Moolenaar9b578142016-01-30 19:39:49 +01001409vim_strsave_shellescape(char_u *string, int do_special, int do_newline)
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001410{
1411 unsigned length;
1412 char_u *p;
1413 char_u *d;
1414 char_u *escaped_string;
Bram Moolenaar05bb9532008-07-04 09:44:11 +00001415 int l;
Bram Moolenaarffd66c42008-07-16 20:43:37 +00001416 int csh_like;
Mike Williams12795022021-06-28 20:53:58 +02001417# ifdef MSWIN
1418 int powershell;
1419 int double_quotes;
1420# endif
Bram Moolenaarffd66c42008-07-16 20:43:37 +00001421
Bram Moolenaar85a20022019-12-21 18:25:54 +01001422 // Only csh and similar shells expand '!' within single quotes. For sh and
1423 // the like we must not put a backslash before it, it will be taken
1424 // literally. If do_special is set the '!' will be escaped twice.
1425 // Csh also needs to have "\n" escaped twice when do_special is set.
Bram Moolenaar7693ec62008-07-24 18:29:37 +00001426 csh_like = csh_like_shell();
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001427
Mike Williams12795022021-06-28 20:53:58 +02001428# ifdef MSWIN
1429 // PowerShell only accepts single quotes so override p_ssl.
1430 powershell = strstr((char *)gettail(p_sh), "powershell") != NULL;
1431 double_quotes = !powershell && !p_ssl;
1432# endif
1433
Bram Moolenaar85a20022019-12-21 18:25:54 +01001434 // First count the number of extra bytes required.
1435 length = (unsigned)STRLEN(string) + 3; // two quotes and a trailing NUL
Bram Moolenaar91acfff2017-03-12 19:22:36 +01001436 for (p = string; *p != NUL; MB_PTR_ADV(p))
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001437 {
Bram Moolenaar4f974752019-02-17 17:44:42 +01001438# ifdef MSWIN
Mike Williams12795022021-06-28 20:53:58 +02001439 if (double_quotes)
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001440 {
1441 if (*p == '"')
Bram Moolenaar85a20022019-12-21 18:25:54 +01001442 ++length; // " -> ""
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001443 }
1444 else
1445# endif
1446 if (*p == '\'')
Mike Williams12795022021-06-28 20:53:58 +02001447 {
1448# ifdef MSWIN
1449 if (powershell)
1450 length +=2; // ' => ''
1451 else
1452# endif
1453 length += 3; // ' => '\''
1454 }
Bram Moolenaar26df0922014-02-23 23:39:13 +01001455 if ((*p == '\n' && (csh_like || do_newline))
1456 || (*p == '!' && (csh_like || do_special)))
Bram Moolenaarffd66c42008-07-16 20:43:37 +00001457 {
Bram Moolenaar85a20022019-12-21 18:25:54 +01001458 ++length; // insert backslash
Bram Moolenaarffd66c42008-07-16 20:43:37 +00001459 if (csh_like && do_special)
Bram Moolenaar85a20022019-12-21 18:25:54 +01001460 ++length; // insert backslash
Bram Moolenaarffd66c42008-07-16 20:43:37 +00001461 }
Bram Moolenaar05bb9532008-07-04 09:44:11 +00001462 if (do_special && find_cmdline_var(p, &l) >= 0)
1463 {
Bram Moolenaar85a20022019-12-21 18:25:54 +01001464 ++length; // insert backslash
Bram Moolenaar05bb9532008-07-04 09:44:11 +00001465 p += l - 1;
1466 }
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001467 }
1468
Bram Moolenaar85a20022019-12-21 18:25:54 +01001469 // Allocate memory for the result and fill it.
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001470 escaped_string = alloc(length);
1471 if (escaped_string != NULL)
1472 {
1473 d = escaped_string;
1474
Bram Moolenaar85a20022019-12-21 18:25:54 +01001475 // add opening quote
Bram Moolenaar4f974752019-02-17 17:44:42 +01001476# ifdef MSWIN
Mike Williams12795022021-06-28 20:53:58 +02001477 if (double_quotes)
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001478 *d++ = '"';
1479 else
1480# endif
1481 *d++ = '\'';
1482
1483 for (p = string; *p != NUL; )
1484 {
Bram Moolenaar4f974752019-02-17 17:44:42 +01001485# ifdef MSWIN
Mike Williams12795022021-06-28 20:53:58 +02001486 if (double_quotes)
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001487 {
1488 if (*p == '"')
1489 {
1490 *d++ = '"';
1491 *d++ = '"';
1492 ++p;
1493 continue;
1494 }
1495 }
1496 else
1497# endif
1498 if (*p == '\'')
1499 {
Mike Williams12795022021-06-28 20:53:58 +02001500# ifdef MSWIN
1501 if (powershell)
1502 {
1503 *d++ = '\'';
1504 *d++ = '\'';
1505 }
1506 else
1507# endif
1508 {
1509 *d++ = '\'';
1510 *d++ = '\\';
1511 *d++ = '\'';
1512 *d++ = '\'';
1513 }
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001514 ++p;
1515 continue;
1516 }
Bram Moolenaar26df0922014-02-23 23:39:13 +01001517 if ((*p == '\n' && (csh_like || do_newline))
1518 || (*p == '!' && (csh_like || do_special)))
Bram Moolenaarffd66c42008-07-16 20:43:37 +00001519 {
1520 *d++ = '\\';
1521 if (csh_like && do_special)
1522 *d++ = '\\';
1523 *d++ = *p++;
1524 continue;
1525 }
Bram Moolenaar05bb9532008-07-04 09:44:11 +00001526 if (do_special && find_cmdline_var(p, &l) >= 0)
1527 {
Bram Moolenaar85a20022019-12-21 18:25:54 +01001528 *d++ = '\\'; // insert backslash
1529 while (--l >= 0) // copy the var
Bram Moolenaar05bb9532008-07-04 09:44:11 +00001530 *d++ = *p++;
Bram Moolenaar099d01d2009-11-25 16:14:45 +00001531 continue;
Bram Moolenaar05bb9532008-07-04 09:44:11 +00001532 }
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001533
1534 MB_COPY_CHAR(p, d);
1535 }
1536
Bram Moolenaar85a20022019-12-21 18:25:54 +01001537 // add terminating quote and finish with a NUL
Bram Moolenaar4f974752019-02-17 17:44:42 +01001538# ifdef MSWIN
Mike Williams12795022021-06-28 20:53:58 +02001539 if (double_quotes)
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001540 *d++ = '"';
1541 else
1542# endif
1543 *d++ = '\'';
1544 *d = NUL;
1545 }
1546
1547 return escaped_string;
1548}
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001549
Bram Moolenaar071d4272004-06-13 20:20:40 +00001550/*
1551 * Like vim_strsave(), but make all characters uppercase.
1552 * This uses ASCII lower-to-upper case translation, language independent.
1553 */
1554 char_u *
Bram Moolenaar9b578142016-01-30 19:39:49 +01001555vim_strsave_up(char_u *string)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001556{
1557 char_u *p1;
1558
1559 p1 = vim_strsave(string);
1560 vim_strup(p1);
1561 return p1;
1562}
1563
1564/*
1565 * Like vim_strnsave(), but make all characters uppercase.
1566 * This uses ASCII lower-to-upper case translation, language independent.
1567 */
1568 char_u *
Bram Moolenaardf44a272020-06-07 20:49:05 +02001569vim_strnsave_up(char_u *string, size_t len)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001570{
1571 char_u *p1;
1572
1573 p1 = vim_strnsave(string, len);
1574 vim_strup(p1);
1575 return p1;
1576}
1577
1578/*
1579 * ASCII lower-to-upper case translation, language independent.
1580 */
1581 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01001582vim_strup(
1583 char_u *p)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001584{
1585 char_u *p2;
1586 int c;
1587
1588 if (p != NULL)
1589 {
1590 p2 = p;
1591 while ((c = *p2) != NUL)
1592#ifdef EBCDIC
1593 *p2++ = isalpha(c) ? toupper(c) : c;
1594#else
1595 *p2++ = (c < 'a' || c > 'z') ? c : (c - 0x20);
1596#endif
1597 }
1598}
1599
Bram Moolenaarc4956c82006-03-12 21:58:43 +00001600#if defined(FEAT_EVAL) || defined(FEAT_SPELL) || defined(PROTO)
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001601/*
1602 * Make string "s" all upper-case and return it in allocated memory.
1603 * Handles multi-byte characters as well as possible.
1604 * Returns NULL when out of memory.
1605 */
1606 char_u *
Bram Moolenaar9b578142016-01-30 19:39:49 +01001607strup_save(char_u *orig)
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001608{
1609 char_u *p;
1610 char_u *res;
1611
1612 res = p = vim_strsave(orig);
1613
1614 if (res != NULL)
1615 while (*p != NUL)
1616 {
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001617 int l;
1618
1619 if (enc_utf8)
1620 {
1621 int c, uc;
Bram Moolenaar70b2a562012-01-10 22:26:17 +01001622 int newl;
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001623 char_u *s;
1624
1625 c = utf_ptr2char(p);
Bram Moolenaare6640ad2017-12-22 21:06:56 +01001626 l = utf_ptr2len(p);
1627 if (c == 0)
1628 {
Bram Moolenaar85a20022019-12-21 18:25:54 +01001629 // overlong sequence, use only the first byte
Bram Moolenaare6640ad2017-12-22 21:06:56 +01001630 c = *p;
1631 l = 1;
1632 }
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001633 uc = utf_toupper(c);
1634
Bram Moolenaar85a20022019-12-21 18:25:54 +01001635 // Reallocate string when byte count changes. This is rare,
1636 // thus it's OK to do another malloc()/free().
Bram Moolenaar70b2a562012-01-10 22:26:17 +01001637 newl = utf_char2len(uc);
1638 if (newl != l)
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001639 {
Bram Moolenaar964b3742019-05-24 18:54:09 +02001640 s = alloc(STRLEN(res) + 1 + newl - l);
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001641 if (s == NULL)
Bram Moolenaarcc5b22b2017-01-26 22:51:56 +01001642 {
1643 vim_free(res);
1644 return NULL;
1645 }
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001646 mch_memmove(s, res, p - res);
Bram Moolenaar70b2a562012-01-10 22:26:17 +01001647 STRCPY(s + (p - res) + newl, p + l);
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001648 p = s + (p - res);
1649 vim_free(res);
1650 res = s;
1651 }
1652
1653 utf_char2bytes(uc, p);
Bram Moolenaar70b2a562012-01-10 22:26:17 +01001654 p += newl;
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001655 }
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001656 else if (has_mbyte && (l = (*mb_ptr2len)(p)) > 1)
Bram Moolenaar85a20022019-12-21 18:25:54 +01001657 p += l; // skip multi-byte character
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001658 else
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001659 {
Bram Moolenaar85a20022019-12-21 18:25:54 +01001660 *p = TOUPPER_LOC(*p); // note that toupper() can be a macro
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001661 p++;
1662 }
1663 }
1664
1665 return res;
1666}
Bram Moolenaarcc5b22b2017-01-26 22:51:56 +01001667
1668/*
1669 * Make string "s" all lower-case and return it in allocated memory.
1670 * Handles multi-byte characters as well as possible.
1671 * Returns NULL when out of memory.
1672 */
1673 char_u *
1674strlow_save(char_u *orig)
1675{
1676 char_u *p;
1677 char_u *res;
1678
1679 res = p = vim_strsave(orig);
1680
1681 if (res != NULL)
1682 while (*p != NUL)
1683 {
Bram Moolenaarcc5b22b2017-01-26 22:51:56 +01001684 int l;
1685
1686 if (enc_utf8)
1687 {
1688 int c, lc;
1689 int newl;
1690 char_u *s;
1691
1692 c = utf_ptr2char(p);
Bram Moolenaare6640ad2017-12-22 21:06:56 +01001693 l = utf_ptr2len(p);
1694 if (c == 0)
1695 {
Bram Moolenaar85a20022019-12-21 18:25:54 +01001696 // overlong sequence, use only the first byte
Bram Moolenaare6640ad2017-12-22 21:06:56 +01001697 c = *p;
1698 l = 1;
1699 }
Bram Moolenaarcc5b22b2017-01-26 22:51:56 +01001700 lc = utf_tolower(c);
1701
Bram Moolenaar85a20022019-12-21 18:25:54 +01001702 // Reallocate string when byte count changes. This is rare,
1703 // thus it's OK to do another malloc()/free().
Bram Moolenaarcc5b22b2017-01-26 22:51:56 +01001704 newl = utf_char2len(lc);
1705 if (newl != l)
1706 {
Bram Moolenaar964b3742019-05-24 18:54:09 +02001707 s = alloc(STRLEN(res) + 1 + newl - l);
Bram Moolenaarcc5b22b2017-01-26 22:51:56 +01001708 if (s == NULL)
1709 {
1710 vim_free(res);
1711 return NULL;
1712 }
1713 mch_memmove(s, res, p - res);
1714 STRCPY(s + (p - res) + newl, p + l);
1715 p = s + (p - res);
1716 vim_free(res);
1717 res = s;
1718 }
1719
1720 utf_char2bytes(lc, p);
1721 p += newl;
1722 }
1723 else if (has_mbyte && (l = (*mb_ptr2len)(p)) > 1)
Bram Moolenaar85a20022019-12-21 18:25:54 +01001724 p += l; // skip multi-byte character
Bram Moolenaarcc5b22b2017-01-26 22:51:56 +01001725 else
Bram Moolenaarcc5b22b2017-01-26 22:51:56 +01001726 {
Bram Moolenaar85a20022019-12-21 18:25:54 +01001727 *p = TOLOWER_LOC(*p); // note that tolower() can be a macro
Bram Moolenaarcc5b22b2017-01-26 22:51:56 +01001728 p++;
1729 }
1730 }
1731
1732 return res;
1733}
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001734#endif
1735
Bram Moolenaar071d4272004-06-13 20:20:40 +00001736/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00001737 * delete spaces at the end of a string
1738 */
1739 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01001740del_trailing_spaces(char_u *ptr)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001741{
1742 char_u *q;
1743
1744 q = ptr + STRLEN(ptr);
Bram Moolenaar1c465442017-03-12 20:10:05 +01001745 while (--q > ptr && VIM_ISWHITE(q[0]) && q[-1] != '\\' && q[-1] != Ctrl_V)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001746 *q = NUL;
1747}
1748
1749/*
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001750 * Like strncpy(), but always terminate the result with one NUL.
Bram Moolenaard042c562005-06-30 22:04:15 +00001751 * "to" must be "len + 1" long!
Bram Moolenaar071d4272004-06-13 20:20:40 +00001752 */
1753 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01001754vim_strncpy(char_u *to, char_u *from, size_t len)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001755{
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001756 STRNCPY(to, from, len);
1757 to[len] = NUL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001758}
1759
1760/*
Bram Moolenaaref9d6aa2011-04-11 16:56:35 +02001761 * Like strcat(), but make sure the result fits in "tosize" bytes and is
Bram Moolenaar45600ce2017-01-27 21:54:07 +01001762 * always NUL terminated. "from" and "to" may overlap.
Bram Moolenaaref9d6aa2011-04-11 16:56:35 +02001763 */
1764 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01001765vim_strcat(char_u *to, char_u *from, size_t tosize)
Bram Moolenaaref9d6aa2011-04-11 16:56:35 +02001766{
1767 size_t tolen = STRLEN(to);
1768 size_t fromlen = STRLEN(from);
1769
1770 if (tolen + fromlen + 1 > tosize)
1771 {
1772 mch_memmove(to + tolen, from, tosize - tolen - 1);
1773 to[tosize - 1] = NUL;
1774 }
1775 else
Bram Moolenaar45600ce2017-01-27 21:54:07 +01001776 mch_memmove(to + tolen, from, fromlen + 1);
Bram Moolenaaref9d6aa2011-04-11 16:56:35 +02001777}
1778
1779/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00001780 * Isolate one part of a string option where parts are separated with
1781 * "sep_chars".
Bram Moolenaar83bab712005-08-01 21:58:57 +00001782 * The part is copied into "buf[maxlen]".
Bram Moolenaar071d4272004-06-13 20:20:40 +00001783 * "*option" is advanced to the next part.
1784 * The length is returned.
1785 */
1786 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01001787copy_option_part(
1788 char_u **option,
1789 char_u *buf,
1790 int maxlen,
1791 char *sep_chars)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001792{
1793 int len = 0;
1794 char_u *p = *option;
1795
Bram Moolenaar85a20022019-12-21 18:25:54 +01001796 // skip '.' at start of option part, for 'suffixes'
Bram Moolenaar071d4272004-06-13 20:20:40 +00001797 if (*p == '.')
1798 buf[len++] = *p++;
1799 while (*p != NUL && vim_strchr((char_u *)sep_chars, *p) == NULL)
1800 {
1801 /*
1802 * Skip backslash before a separator character and space.
1803 */
1804 if (p[0] == '\\' && vim_strchr((char_u *)sep_chars, p[1]) != NULL)
1805 ++p;
1806 if (len < maxlen - 1)
1807 buf[len++] = *p;
1808 ++p;
1809 }
1810 buf[len] = NUL;
1811
Bram Moolenaar85a20022019-12-21 18:25:54 +01001812 if (*p != NUL && *p != ',') // skip non-standard separator
Bram Moolenaar071d4272004-06-13 20:20:40 +00001813 ++p;
Bram Moolenaar85a20022019-12-21 18:25:54 +01001814 p = skip_to_option_part(p); // p points to next file name
Bram Moolenaar071d4272004-06-13 20:20:40 +00001815
1816 *option = p;
1817 return len;
1818}
1819
1820/*
Bram Moolenaar4770d092006-01-12 23:22:24 +00001821 * Replacement for free() that ignores NULL pointers.
1822 * Also skip free() when exiting for sure, this helps when we caught a deadly
1823 * signal that was caused by a crash in free().
Bram Moolenaard23a8232018-02-10 18:45:26 +01001824 * If you want to set NULL after calling this function, you should use
1825 * VIM_CLEAR() instead.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001826 */
1827 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01001828vim_free(void *x)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001829{
Bram Moolenaar4770d092006-01-12 23:22:24 +00001830 if (x != NULL && !really_exiting)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001831 {
1832#ifdef MEM_PROFILE
1833 mem_pre_free(&x);
1834#endif
1835 free(x);
1836 }
1837}
1838
1839#ifndef HAVE_MEMSET
1840 void *
Bram Moolenaar9b578142016-01-30 19:39:49 +01001841vim_memset(void *ptr, int c, size_t size)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001842{
1843 char *p = ptr;
1844
1845 while (size-- > 0)
1846 *p++ = c;
1847 return ptr;
1848}
1849#endif
1850
Bram Moolenaar071d4272004-06-13 20:20:40 +00001851#if (!defined(HAVE_STRCASECMP) && !defined(HAVE_STRICMP)) || defined(PROTO)
1852/*
1853 * Compare two strings, ignoring case, using current locale.
1854 * Doesn't work for multi-byte characters.
1855 * return 0 for match, < 0 for smaller, > 0 for bigger
1856 */
1857 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01001858vim_stricmp(char *s1, char *s2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001859{
1860 int i;
1861
1862 for (;;)
1863 {
1864 i = (int)TOLOWER_LOC(*s1) - (int)TOLOWER_LOC(*s2);
1865 if (i != 0)
Bram Moolenaar85a20022019-12-21 18:25:54 +01001866 return i; // this character different
Bram Moolenaar071d4272004-06-13 20:20:40 +00001867 if (*s1 == NUL)
Bram Moolenaar85a20022019-12-21 18:25:54 +01001868 break; // strings match until NUL
Bram Moolenaar071d4272004-06-13 20:20:40 +00001869 ++s1;
1870 ++s2;
1871 }
Bram Moolenaar85a20022019-12-21 18:25:54 +01001872 return 0; // strings match
Bram Moolenaar071d4272004-06-13 20:20:40 +00001873}
1874#endif
1875
1876#if (!defined(HAVE_STRNCASECMP) && !defined(HAVE_STRNICMP)) || defined(PROTO)
1877/*
1878 * Compare two strings, for length "len", ignoring case, using current locale.
1879 * Doesn't work for multi-byte characters.
1880 * return 0 for match, < 0 for smaller, > 0 for bigger
1881 */
1882 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01001883vim_strnicmp(char *s1, char *s2, size_t len)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001884{
1885 int i;
1886
1887 while (len > 0)
1888 {
1889 i = (int)TOLOWER_LOC(*s1) - (int)TOLOWER_LOC(*s2);
1890 if (i != 0)
Bram Moolenaar85a20022019-12-21 18:25:54 +01001891 return i; // this character different
Bram Moolenaar071d4272004-06-13 20:20:40 +00001892 if (*s1 == NUL)
Bram Moolenaar85a20022019-12-21 18:25:54 +01001893 break; // strings match until NUL
Bram Moolenaar071d4272004-06-13 20:20:40 +00001894 ++s1;
1895 ++s2;
1896 --len;
1897 }
Bram Moolenaar85a20022019-12-21 18:25:54 +01001898 return 0; // strings match
Bram Moolenaar071d4272004-06-13 20:20:40 +00001899}
1900#endif
1901
Bram Moolenaar071d4272004-06-13 20:20:40 +00001902/*
Bram Moolenaar21a83bd2021-02-23 19:19:58 +01001903 * Search for first occurrence of "c" in "string".
1904 * Version of strchr() that handles unsigned char strings with characters from
1905 * 128 to 255 correctly. It also doesn't return a pointer to the NUL at the
1906 * end of the string.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001907 */
1908 char_u *
Bram Moolenaar9b578142016-01-30 19:39:49 +01001909vim_strchr(char_u *string, int c)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001910{
1911 char_u *p;
1912 int b;
1913
1914 p = string;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001915 if (enc_utf8 && c >= 0x80)
1916 {
1917 while (*p != NUL)
1918 {
Bram Moolenaarace95982017-03-29 17:30:27 +02001919 int l = utfc_ptr2len(p);
Bram Moolenaard82a2a92015-04-21 14:02:35 +02001920
Bram Moolenaar85a20022019-12-21 18:25:54 +01001921 // Avoid matching an illegal byte here.
Bram Moolenaard82a2a92015-04-21 14:02:35 +02001922 if (utf_ptr2char(p) == c && l > 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001923 return p;
Bram Moolenaard82a2a92015-04-21 14:02:35 +02001924 p += l;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001925 }
1926 return NULL;
1927 }
1928 if (enc_dbcs != 0 && c > 255)
1929 {
1930 int n2 = c & 0xff;
1931
1932 c = ((unsigned)c >> 8) & 0xff;
1933 while ((b = *p) != NUL)
1934 {
1935 if (b == c && p[1] == n2)
1936 return p;
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001937 p += (*mb_ptr2len)(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001938 }
1939 return NULL;
1940 }
1941 if (has_mbyte)
1942 {
1943 while ((b = *p) != NUL)
1944 {
1945 if (b == c)
1946 return p;
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001947 p += (*mb_ptr2len)(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001948 }
1949 return NULL;
1950 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001951 while ((b = *p) != NUL)
1952 {
1953 if (b == c)
1954 return p;
1955 ++p;
1956 }
1957 return NULL;
1958}
1959
1960/*
Bram Moolenaar05159a02005-02-26 23:04:13 +00001961 * Version of strchr() that only works for bytes and handles unsigned char
1962 * strings with characters above 128 correctly. It also doesn't return a
1963 * pointer to the NUL at the end of the string.
1964 */
1965 char_u *
Bram Moolenaar9b578142016-01-30 19:39:49 +01001966vim_strbyte(char_u *string, int c)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001967{
1968 char_u *p = string;
1969
1970 while (*p != NUL)
1971 {
1972 if (*p == c)
1973 return p;
1974 ++p;
1975 }
1976 return NULL;
1977}
1978
1979/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00001980 * Search for last occurrence of "c" in "string".
Bram Moolenaar21a83bd2021-02-23 19:19:58 +01001981 * Version of strrchr() that handles unsigned char strings with characters from
1982 * 128 to 255 correctly. It also doesn't return a pointer to the NUL at the
1983 * end of the string.
Bram Moolenaarac6e65f2005-08-29 22:25:38 +00001984 * Return NULL if not found.
Bram Moolenaar05159a02005-02-26 23:04:13 +00001985 * Does not handle multi-byte char for "c"!
Bram Moolenaar071d4272004-06-13 20:20:40 +00001986 */
1987 char_u *
Bram Moolenaar9b578142016-01-30 19:39:49 +01001988vim_strrchr(char_u *string, int c)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001989{
1990 char_u *retval = NULL;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001991 char_u *p = string;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001992
Bram Moolenaar05159a02005-02-26 23:04:13 +00001993 while (*p)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001994 {
Bram Moolenaar05159a02005-02-26 23:04:13 +00001995 if (*p == c)
1996 retval = p;
Bram Moolenaar91acfff2017-03-12 19:22:36 +01001997 MB_PTR_ADV(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001998 }
1999 return retval;
2000}
2001
2002/*
2003 * Vim's version of strpbrk(), in case it's missing.
2004 * Don't generate a prototype for this, causes problems when it's not used.
2005 */
2006#ifndef PROTO
2007# ifndef HAVE_STRPBRK
2008# ifdef vim_strpbrk
2009# undef vim_strpbrk
2010# endif
2011 char_u *
Bram Moolenaar9b578142016-01-30 19:39:49 +01002012vim_strpbrk(char_u *s, char_u *charset)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002013{
2014 while (*s)
2015 {
2016 if (vim_strchr(charset, *s) != NULL)
2017 return s;
Bram Moolenaar91acfff2017-03-12 19:22:36 +01002018 MB_PTR_ADV(s);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002019 }
2020 return NULL;
2021}
2022# endif
2023#endif
2024
2025/*
2026 * Vim has its own isspace() function, because on some machines isspace()
2027 * can't handle characters above 128.
2028 */
2029 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01002030vim_isspace(int x)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002031{
2032 return ((x >= 9 && x <= 13) || x == ' ');
2033}
2034
2035/************************************************************************
Bram Moolenaar383f9bc2005-01-19 22:18:32 +00002036 * Functions for handling growing arrays.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002037 */
2038
2039/*
2040 * Clear an allocated growing array.
2041 */
2042 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01002043ga_clear(garray_T *gap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002044{
2045 vim_free(gap->ga_data);
2046 ga_init(gap);
2047}
2048
2049/*
2050 * Clear a growing array that contains a list of strings.
2051 */
2052 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01002053ga_clear_strings(garray_T *gap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002054{
2055 int i;
2056
Bram Moolenaar7a6eaa02021-03-21 20:53:29 +01002057 if (gap->ga_data != NULL)
2058 for (i = 0; i < gap->ga_len; ++i)
2059 vim_free(((char_u **)(gap->ga_data))[i]);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002060 ga_clear(gap);
2061}
2062
2063/*
Bram Moolenaar38ddf332020-07-31 22:05:04 +02002064 * Copy a growing array that contains a list of strings.
2065 */
2066 int
2067ga_copy_strings(garray_T *from, garray_T *to)
2068{
2069 int i;
2070
2071 ga_init2(to, sizeof(char_u *), 1);
2072 if (ga_grow(to, from->ga_len) == FAIL)
2073 return FAIL;
2074
2075 for (i = 0; i < from->ga_len; ++i)
2076 {
2077 char_u *orig = ((char_u **)from->ga_data)[i];
2078 char_u *copy;
2079
2080 if (orig == NULL)
2081 copy = NULL;
2082 else
2083 {
2084 copy = vim_strsave(orig);
2085 if (copy == NULL)
2086 {
2087 to->ga_len = i;
2088 ga_clear_strings(to);
2089 return FAIL;
2090 }
2091 }
2092 ((char_u **)to->ga_data)[i] = copy;
2093 }
2094 to->ga_len = from->ga_len;
2095 return OK;
2096}
2097
2098/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002099 * Initialize a growing array. Don't forget to set ga_itemsize and
2100 * ga_growsize! Or use ga_init2().
2101 */
2102 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01002103ga_init(garray_T *gap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002104{
2105 gap->ga_data = NULL;
Bram Moolenaar86b68352004-12-27 21:59:20 +00002106 gap->ga_maxlen = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002107 gap->ga_len = 0;
2108}
2109
2110 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01002111ga_init2(garray_T *gap, int itemsize, int growsize)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002112{
2113 ga_init(gap);
2114 gap->ga_itemsize = itemsize;
2115 gap->ga_growsize = growsize;
2116}
2117
2118/*
2119 * Make room in growing array "gap" for at least "n" items.
2120 * Return FAIL for failure, OK otherwise.
2121 */
2122 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01002123ga_grow(garray_T *gap, int n)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002124{
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02002125 if (gap->ga_maxlen - gap->ga_len < n)
2126 return ga_grow_inner(gap, n);
2127 return OK;
2128}
2129
2130 int
2131ga_grow_inner(garray_T *gap, int n)
2132{
Bram Moolenaar7282bc32012-02-22 18:12:32 +01002133 size_t old_len;
2134 size_t new_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002135 char_u *pp;
2136
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02002137 if (n < gap->ga_growsize)
2138 n = gap->ga_growsize;
Bram Moolenaarc47ed442019-06-01 14:36:26 +02002139
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02002140 // A linear growth is very inefficient when the array grows big. This
2141 // is a compromise between allocating memory that won't be used and too
2142 // many copy operations. A factor of 1.5 seems reasonable.
2143 if (n < gap->ga_len / 2)
2144 n = gap->ga_len / 2;
Bram Moolenaarc47ed442019-06-01 14:36:26 +02002145
Bram Moolenaar7e9f3512020-05-13 22:44:22 +02002146 new_len = gap->ga_itemsize * (gap->ga_len + n);
2147 pp = vim_realloc(gap->ga_data, new_len);
2148 if (pp == NULL)
2149 return FAIL;
2150 old_len = gap->ga_itemsize * gap->ga_maxlen;
2151 vim_memset(pp + old_len, 0, new_len - old_len);
2152 gap->ga_maxlen = gap->ga_len + n;
2153 gap->ga_data = pp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002154 return OK;
2155}
2156
Bram Moolenaar113e1072019-01-20 15:30:40 +01002157#if defined(FEAT_EVAL) || defined(FEAT_SEARCHPATH) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002158/*
Bram Moolenaar80a7dcf2010-08-04 17:07:20 +02002159 * For a growing array that contains a list of strings: concatenate all the
Bram Moolenaar1b1063a2014-05-07 18:35:30 +02002160 * strings with a separating "sep".
Bram Moolenaar80a7dcf2010-08-04 17:07:20 +02002161 * Returns NULL when out of memory.
2162 */
2163 char_u *
Bram Moolenaar9b578142016-01-30 19:39:49 +01002164ga_concat_strings(garray_T *gap, char *sep)
Bram Moolenaar80a7dcf2010-08-04 17:07:20 +02002165{
2166 int i;
2167 int len = 0;
Bram Moolenaar1b1063a2014-05-07 18:35:30 +02002168 int sep_len = (int)STRLEN(sep);
Bram Moolenaar80a7dcf2010-08-04 17:07:20 +02002169 char_u *s;
Bram Moolenaar1b1063a2014-05-07 18:35:30 +02002170 char_u *p;
Bram Moolenaar80a7dcf2010-08-04 17:07:20 +02002171
2172 for (i = 0; i < gap->ga_len; ++i)
Bram Moolenaar1b1063a2014-05-07 18:35:30 +02002173 len += (int)STRLEN(((char_u **)(gap->ga_data))[i]) + sep_len;
Bram Moolenaar80a7dcf2010-08-04 17:07:20 +02002174
2175 s = alloc(len + 1);
2176 if (s != NULL)
2177 {
2178 *s = NUL;
Bram Moolenaar1b1063a2014-05-07 18:35:30 +02002179 p = s;
Bram Moolenaar80a7dcf2010-08-04 17:07:20 +02002180 for (i = 0; i < gap->ga_len; ++i)
2181 {
Bram Moolenaar1b1063a2014-05-07 18:35:30 +02002182 if (p != s)
2183 {
2184 STRCPY(p, sep);
2185 p += sep_len;
2186 }
2187 STRCPY(p, ((char_u **)(gap->ga_data))[i]);
2188 p += STRLEN(p);
Bram Moolenaar80a7dcf2010-08-04 17:07:20 +02002189 }
2190 }
2191 return s;
2192}
Bram Moolenaar113e1072019-01-20 15:30:40 +01002193#endif
Bram Moolenaar80a7dcf2010-08-04 17:07:20 +02002194
Bram Moolenaar5a66dfb2017-03-01 20:40:39 +01002195#if defined(FEAT_VIMINFO) || defined(FEAT_EVAL) || defined(PROTO)
Bram Moolenaarb20e3342016-01-18 23:29:01 +01002196/*
2197 * Make a copy of string "p" and add it to "gap".
2198 * When out of memory nothing changes.
2199 */
2200 void
2201ga_add_string(garray_T *gap, char_u *p)
2202{
2203 char_u *cp = vim_strsave(p);
2204
2205 if (cp != NULL)
2206 {
2207 if (ga_grow(gap, 1) == OK)
2208 ((char_u **)(gap->ga_data))[gap->ga_len++] = cp;
2209 else
2210 vim_free(cp);
2211 }
2212}
2213#endif
2214
Bram Moolenaar80a7dcf2010-08-04 17:07:20 +02002215/*
Bram Moolenaar29ddebe2019-01-26 17:28:26 +01002216 * Concatenate a string to a growarray which contains bytes.
Bram Moolenaar43345542015-11-29 17:35:35 +01002217 * When "s" is NULL does not do anything.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002218 * Note: Does NOT copy the NUL at the end!
2219 */
2220 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01002221ga_concat(garray_T *gap, char_u *s)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002222{
Bram Moolenaar43345542015-11-29 17:35:35 +01002223 int len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002224
Bram Moolenaar478af672017-04-10 22:22:42 +02002225 if (s == NULL || *s == NUL)
Bram Moolenaar43345542015-11-29 17:35:35 +01002226 return;
2227 len = (int)STRLEN(s);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002228 if (ga_grow(gap, len) == OK)
2229 {
2230 mch_memmove((char *)gap->ga_data + gap->ga_len, s, (size_t)len);
2231 gap->ga_len += len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002232 }
2233}
2234
2235/*
2236 * Append one byte to a growarray which contains bytes.
2237 */
2238 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01002239ga_append(garray_T *gap, int c)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002240{
2241 if (ga_grow(gap, 1) == OK)
2242 {
2243 *((char *)gap->ga_data + gap->ga_len) = c;
2244 ++gap->ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002245 }
2246}
2247
Bram Moolenaar4f974752019-02-17 17:44:42 +01002248#if (defined(UNIX) && !defined(USE_SYSTEM)) || defined(MSWIN) \
Bram Moolenaar597a4222014-06-25 14:39:50 +02002249 || defined(PROTO)
Bram Moolenaar4b9669f2011-07-07 16:20:52 +02002250/*
2251 * Append the text in "gap" below the cursor line and clear "gap".
2252 */
2253 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01002254append_ga_line(garray_T *gap)
Bram Moolenaar4b9669f2011-07-07 16:20:52 +02002255{
Bram Moolenaar85a20022019-12-21 18:25:54 +01002256 // Remove trailing CR.
Bram Moolenaar4b9669f2011-07-07 16:20:52 +02002257 if (gap->ga_len > 0
2258 && !curbuf->b_p_bin
2259 && ((char_u *)gap->ga_data)[gap->ga_len - 1] == CAR)
2260 --gap->ga_len;
2261 ga_append(gap, NUL);
2262 ml_append(curwin->w_cursor.lnum++, gap->ga_data, 0, FALSE);
2263 gap->ga_len = 0;
2264}
2265#endif
2266
Bram Moolenaar071d4272004-06-13 20:20:40 +00002267/************************************************************************
2268 * functions that use lookup tables for various things, generally to do with
2269 * special key codes.
2270 */
2271
2272/*
2273 * Some useful tables.
2274 */
2275
2276static struct modmasktable
2277{
Bram Moolenaar85a20022019-12-21 18:25:54 +01002278 short mod_mask; // Bit-mask for particular key modifier
2279 short mod_flag; // Bit(s) for particular key modifier
2280 char_u name; // Single letter name of modifier
Bram Moolenaar071d4272004-06-13 20:20:40 +00002281} mod_mask_table[] =
2282{
2283 {MOD_MASK_ALT, MOD_MASK_ALT, (char_u)'M'},
Bram Moolenaar19a09a12005-03-04 23:39:37 +00002284 {MOD_MASK_META, MOD_MASK_META, (char_u)'T'},
Bram Moolenaar071d4272004-06-13 20:20:40 +00002285 {MOD_MASK_CTRL, MOD_MASK_CTRL, (char_u)'C'},
2286 {MOD_MASK_SHIFT, MOD_MASK_SHIFT, (char_u)'S'},
2287 {MOD_MASK_MULTI_CLICK, MOD_MASK_2CLICK, (char_u)'2'},
2288 {MOD_MASK_MULTI_CLICK, MOD_MASK_3CLICK, (char_u)'3'},
2289 {MOD_MASK_MULTI_CLICK, MOD_MASK_4CLICK, (char_u)'4'},
Bram Moolenaard0573012017-10-28 21:11:06 +02002290#ifdef MACOS_X
Bram Moolenaar071d4272004-06-13 20:20:40 +00002291 {MOD_MASK_CMD, MOD_MASK_CMD, (char_u)'D'},
2292#endif
Bram Moolenaar85a20022019-12-21 18:25:54 +01002293 // 'A' must be the last one
Bram Moolenaar071d4272004-06-13 20:20:40 +00002294 {MOD_MASK_ALT, MOD_MASK_ALT, (char_u)'A'},
2295 {0, 0, NUL}
Bram Moolenaar85a20022019-12-21 18:25:54 +01002296 // NOTE: when adding an entry, update MAX_KEY_NAME_LEN!
Bram Moolenaar071d4272004-06-13 20:20:40 +00002297};
2298
2299/*
2300 * Shifted key terminal codes and their unshifted equivalent.
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00002301 * Don't add mouse codes here, they are handled separately!
Bram Moolenaar071d4272004-06-13 20:20:40 +00002302 */
2303#define MOD_KEYS_ENTRY_SIZE 5
2304
2305static char_u modifier_keys_table[] =
2306{
Bram Moolenaar85a20022019-12-21 18:25:54 +01002307// mod mask with modifier without modifier
2308 MOD_MASK_SHIFT, '&', '9', '@', '1', // begin
2309 MOD_MASK_SHIFT, '&', '0', '@', '2', // cancel
2310 MOD_MASK_SHIFT, '*', '1', '@', '4', // command
2311 MOD_MASK_SHIFT, '*', '2', '@', '5', // copy
2312 MOD_MASK_SHIFT, '*', '3', '@', '6', // create
2313 MOD_MASK_SHIFT, '*', '4', 'k', 'D', // delete char
2314 MOD_MASK_SHIFT, '*', '5', 'k', 'L', // delete line
2315 MOD_MASK_SHIFT, '*', '7', '@', '7', // end
2316 MOD_MASK_CTRL, KS_EXTRA, (int)KE_C_END, '@', '7', // end
2317 MOD_MASK_SHIFT, '*', '9', '@', '9', // exit
2318 MOD_MASK_SHIFT, '*', '0', '@', '0', // find
2319 MOD_MASK_SHIFT, '#', '1', '%', '1', // help
2320 MOD_MASK_SHIFT, '#', '2', 'k', 'h', // home
2321 MOD_MASK_CTRL, KS_EXTRA, (int)KE_C_HOME, 'k', 'h', // home
2322 MOD_MASK_SHIFT, '#', '3', 'k', 'I', // insert
2323 MOD_MASK_SHIFT, '#', '4', 'k', 'l', // left arrow
2324 MOD_MASK_CTRL, KS_EXTRA, (int)KE_C_LEFT, 'k', 'l', // left arrow
2325 MOD_MASK_SHIFT, '%', 'a', '%', '3', // message
2326 MOD_MASK_SHIFT, '%', 'b', '%', '4', // move
2327 MOD_MASK_SHIFT, '%', 'c', '%', '5', // next
2328 MOD_MASK_SHIFT, '%', 'd', '%', '7', // options
2329 MOD_MASK_SHIFT, '%', 'e', '%', '8', // previous
2330 MOD_MASK_SHIFT, '%', 'f', '%', '9', // print
2331 MOD_MASK_SHIFT, '%', 'g', '%', '0', // redo
2332 MOD_MASK_SHIFT, '%', 'h', '&', '3', // replace
2333 MOD_MASK_SHIFT, '%', 'i', 'k', 'r', // right arr.
2334 MOD_MASK_CTRL, KS_EXTRA, (int)KE_C_RIGHT, 'k', 'r', // right arr.
2335 MOD_MASK_SHIFT, '%', 'j', '&', '5', // resume
2336 MOD_MASK_SHIFT, '!', '1', '&', '6', // save
2337 MOD_MASK_SHIFT, '!', '2', '&', '7', // suspend
2338 MOD_MASK_SHIFT, '!', '3', '&', '8', // undo
2339 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_UP, 'k', 'u', // up arrow
2340 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_DOWN, 'k', 'd', // down arrow
Bram Moolenaar071d4272004-06-13 20:20:40 +00002341
Bram Moolenaar85a20022019-12-21 18:25:54 +01002342 // vt100 F1
Bram Moolenaar071d4272004-06-13 20:20:40 +00002343 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_XF1, KS_EXTRA, (int)KE_XF1,
2344 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_XF2, KS_EXTRA, (int)KE_XF2,
2345 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_XF3, KS_EXTRA, (int)KE_XF3,
2346 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_XF4, KS_EXTRA, (int)KE_XF4,
2347
Bram Moolenaar85a20022019-12-21 18:25:54 +01002348 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F1, 'k', '1', // F1
Bram Moolenaar071d4272004-06-13 20:20:40 +00002349 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F2, 'k', '2',
2350 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F3, 'k', '3',
2351 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F4, 'k', '4',
2352 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F5, 'k', '5',
2353 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F6, 'k', '6',
2354 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F7, 'k', '7',
2355 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F8, 'k', '8',
2356 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F9, 'k', '9',
Bram Moolenaar85a20022019-12-21 18:25:54 +01002357 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F10, 'k', ';', // F10
Bram Moolenaar071d4272004-06-13 20:20:40 +00002358
2359 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F11, 'F', '1',
2360 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F12, 'F', '2',
2361 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F13, 'F', '3',
2362 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F14, 'F', '4',
2363 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F15, 'F', '5',
2364 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F16, 'F', '6',
2365 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F17, 'F', '7',
2366 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F18, 'F', '8',
2367 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F19, 'F', '9',
2368 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F20, 'F', 'A',
2369
2370 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F21, 'F', 'B',
2371 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F22, 'F', 'C',
2372 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F23, 'F', 'D',
2373 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F24, 'F', 'E',
2374 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F25, 'F', 'F',
2375 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F26, 'F', 'G',
2376 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F27, 'F', 'H',
2377 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F28, 'F', 'I',
2378 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F29, 'F', 'J',
2379 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F30, 'F', 'K',
2380
2381 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F31, 'F', 'L',
2382 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F32, 'F', 'M',
2383 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F33, 'F', 'N',
2384 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F34, 'F', 'O',
2385 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F35, 'F', 'P',
2386 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F36, 'F', 'Q',
2387 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F37, 'F', 'R',
2388
Bram Moolenaar85a20022019-12-21 18:25:54 +01002389 // TAB pseudo code
Bram Moolenaar071d4272004-06-13 20:20:40 +00002390 MOD_MASK_SHIFT, 'k', 'B', KS_EXTRA, (int)KE_TAB,
2391
2392 NUL
2393};
2394
2395static struct key_name_entry
2396{
Bram Moolenaar85a20022019-12-21 18:25:54 +01002397 int key; // Special key code or ascii value
2398 char_u *name; // Name of key
Bram Moolenaar071d4272004-06-13 20:20:40 +00002399} key_names_table[] =
2400{
2401 {' ', (char_u *)"Space"},
2402 {TAB, (char_u *)"Tab"},
2403 {K_TAB, (char_u *)"Tab"},
2404 {NL, (char_u *)"NL"},
Bram Moolenaar85a20022019-12-21 18:25:54 +01002405 {NL, (char_u *)"NewLine"}, // Alternative name
2406 {NL, (char_u *)"LineFeed"}, // Alternative name
2407 {NL, (char_u *)"LF"}, // Alternative name
Bram Moolenaar071d4272004-06-13 20:20:40 +00002408 {CAR, (char_u *)"CR"},
Bram Moolenaar85a20022019-12-21 18:25:54 +01002409 {CAR, (char_u *)"Return"}, // Alternative name
2410 {CAR, (char_u *)"Enter"}, // Alternative name
Bram Moolenaar071d4272004-06-13 20:20:40 +00002411 {K_BS, (char_u *)"BS"},
Bram Moolenaar85a20022019-12-21 18:25:54 +01002412 {K_BS, (char_u *)"BackSpace"}, // Alternative name
Bram Moolenaar071d4272004-06-13 20:20:40 +00002413 {ESC, (char_u *)"Esc"},
2414 {CSI, (char_u *)"CSI"},
2415 {K_CSI, (char_u *)"xCSI"},
2416 {'|', (char_u *)"Bar"},
2417 {'\\', (char_u *)"Bslash"},
2418 {K_DEL, (char_u *)"Del"},
Bram Moolenaar85a20022019-12-21 18:25:54 +01002419 {K_DEL, (char_u *)"Delete"}, // Alternative name
Bram Moolenaar071d4272004-06-13 20:20:40 +00002420 {K_KDEL, (char_u *)"kDel"},
2421 {K_UP, (char_u *)"Up"},
2422 {K_DOWN, (char_u *)"Down"},
2423 {K_LEFT, (char_u *)"Left"},
2424 {K_RIGHT, (char_u *)"Right"},
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002425 {K_XUP, (char_u *)"xUp"},
2426 {K_XDOWN, (char_u *)"xDown"},
2427 {K_XLEFT, (char_u *)"xLeft"},
2428 {K_XRIGHT, (char_u *)"xRight"},
Bram Moolenaarec2da362017-01-21 20:04:22 +01002429 {K_PS, (char_u *)"PasteStart"},
2430 {K_PE, (char_u *)"PasteEnd"},
Bram Moolenaar071d4272004-06-13 20:20:40 +00002431
2432 {K_F1, (char_u *)"F1"},
2433 {K_F2, (char_u *)"F2"},
2434 {K_F3, (char_u *)"F3"},
2435 {K_F4, (char_u *)"F4"},
2436 {K_F5, (char_u *)"F5"},
2437 {K_F6, (char_u *)"F6"},
2438 {K_F7, (char_u *)"F7"},
2439 {K_F8, (char_u *)"F8"},
2440 {K_F9, (char_u *)"F9"},
2441 {K_F10, (char_u *)"F10"},
2442
2443 {K_F11, (char_u *)"F11"},
2444 {K_F12, (char_u *)"F12"},
2445 {K_F13, (char_u *)"F13"},
2446 {K_F14, (char_u *)"F14"},
2447 {K_F15, (char_u *)"F15"},
2448 {K_F16, (char_u *)"F16"},
2449 {K_F17, (char_u *)"F17"},
2450 {K_F18, (char_u *)"F18"},
2451 {K_F19, (char_u *)"F19"},
2452 {K_F20, (char_u *)"F20"},
2453
2454 {K_F21, (char_u *)"F21"},
2455 {K_F22, (char_u *)"F22"},
2456 {K_F23, (char_u *)"F23"},
2457 {K_F24, (char_u *)"F24"},
2458 {K_F25, (char_u *)"F25"},
2459 {K_F26, (char_u *)"F26"},
2460 {K_F27, (char_u *)"F27"},
2461 {K_F28, (char_u *)"F28"},
2462 {K_F29, (char_u *)"F29"},
2463 {K_F30, (char_u *)"F30"},
2464
2465 {K_F31, (char_u *)"F31"},
2466 {K_F32, (char_u *)"F32"},
2467 {K_F33, (char_u *)"F33"},
2468 {K_F34, (char_u *)"F34"},
2469 {K_F35, (char_u *)"F35"},
2470 {K_F36, (char_u *)"F36"},
2471 {K_F37, (char_u *)"F37"},
2472
2473 {K_XF1, (char_u *)"xF1"},
2474 {K_XF2, (char_u *)"xF2"},
2475 {K_XF3, (char_u *)"xF3"},
2476 {K_XF4, (char_u *)"xF4"},
2477
2478 {K_HELP, (char_u *)"Help"},
2479 {K_UNDO, (char_u *)"Undo"},
2480 {K_INS, (char_u *)"Insert"},
Bram Moolenaar85a20022019-12-21 18:25:54 +01002481 {K_INS, (char_u *)"Ins"}, // Alternative name
Bram Moolenaar071d4272004-06-13 20:20:40 +00002482 {K_KINS, (char_u *)"kInsert"},
2483 {K_HOME, (char_u *)"Home"},
2484 {K_KHOME, (char_u *)"kHome"},
2485 {K_XHOME, (char_u *)"xHome"},
Bram Moolenaar68b76a62005-03-25 21:53:48 +00002486 {K_ZHOME, (char_u *)"zHome"},
Bram Moolenaar071d4272004-06-13 20:20:40 +00002487 {K_END, (char_u *)"End"},
2488 {K_KEND, (char_u *)"kEnd"},
2489 {K_XEND, (char_u *)"xEnd"},
Bram Moolenaar68b76a62005-03-25 21:53:48 +00002490 {K_ZEND, (char_u *)"zEnd"},
Bram Moolenaar071d4272004-06-13 20:20:40 +00002491 {K_PAGEUP, (char_u *)"PageUp"},
2492 {K_PAGEDOWN, (char_u *)"PageDown"},
2493 {K_KPAGEUP, (char_u *)"kPageUp"},
2494 {K_KPAGEDOWN, (char_u *)"kPageDown"},
2495
2496 {K_KPLUS, (char_u *)"kPlus"},
2497 {K_KMINUS, (char_u *)"kMinus"},
2498 {K_KDIVIDE, (char_u *)"kDivide"},
2499 {K_KMULTIPLY, (char_u *)"kMultiply"},
2500 {K_KENTER, (char_u *)"kEnter"},
2501 {K_KPOINT, (char_u *)"kPoint"},
2502
2503 {K_K0, (char_u *)"k0"},
2504 {K_K1, (char_u *)"k1"},
2505 {K_K2, (char_u *)"k2"},
2506 {K_K3, (char_u *)"k3"},
2507 {K_K4, (char_u *)"k4"},
2508 {K_K5, (char_u *)"k5"},
2509 {K_K6, (char_u *)"k6"},
2510 {K_K7, (char_u *)"k7"},
2511 {K_K8, (char_u *)"k8"},
2512 {K_K9, (char_u *)"k9"},
2513
2514 {'<', (char_u *)"lt"},
2515
2516 {K_MOUSE, (char_u *)"Mouse"},
Bram Moolenaar5af7d712012-01-20 17:15:51 +01002517#ifdef FEAT_MOUSE_NET
Bram Moolenaar071d4272004-06-13 20:20:40 +00002518 {K_NETTERM_MOUSE, (char_u *)"NetMouse"},
Bram Moolenaar5af7d712012-01-20 17:15:51 +01002519#endif
2520#ifdef FEAT_MOUSE_DEC
Bram Moolenaar071d4272004-06-13 20:20:40 +00002521 {K_DEC_MOUSE, (char_u *)"DecMouse"},
Bram Moolenaar5af7d712012-01-20 17:15:51 +01002522#endif
2523#ifdef FEAT_MOUSE_JSB
Bram Moolenaar071d4272004-06-13 20:20:40 +00002524 {K_JSBTERM_MOUSE, (char_u *)"JsbMouse"},
Bram Moolenaar5af7d712012-01-20 17:15:51 +01002525#endif
2526#ifdef FEAT_MOUSE_PTERM
Bram Moolenaar071d4272004-06-13 20:20:40 +00002527 {K_PTERM_MOUSE, (char_u *)"PtermMouse"},
Bram Moolenaar5af7d712012-01-20 17:15:51 +01002528#endif
2529#ifdef FEAT_MOUSE_URXVT
2530 {K_URXVT_MOUSE, (char_u *)"UrxvtMouse"},
2531#endif
Bram Moolenaar2b9578f2012-08-15 16:21:32 +02002532 {K_SGR_MOUSE, (char_u *)"SgrMouse"},
Bram Moolenaar21a83bd2021-02-23 19:19:58 +01002533 {K_SGR_MOUSERELEASE, (char_u *)"SgrMouseRelease"},
Bram Moolenaar071d4272004-06-13 20:20:40 +00002534 {K_LEFTMOUSE, (char_u *)"LeftMouse"},
2535 {K_LEFTMOUSE_NM, (char_u *)"LeftMouseNM"},
2536 {K_LEFTDRAG, (char_u *)"LeftDrag"},
2537 {K_LEFTRELEASE, (char_u *)"LeftRelease"},
2538 {K_LEFTRELEASE_NM, (char_u *)"LeftReleaseNM"},
Bram Moolenaar51b0f372017-11-18 18:52:04 +01002539 {K_MOUSEMOVE, (char_u *)"MouseMove"},
Bram Moolenaar071d4272004-06-13 20:20:40 +00002540 {K_MIDDLEMOUSE, (char_u *)"MiddleMouse"},
2541 {K_MIDDLEDRAG, (char_u *)"MiddleDrag"},
2542 {K_MIDDLERELEASE, (char_u *)"MiddleRelease"},
2543 {K_RIGHTMOUSE, (char_u *)"RightMouse"},
2544 {K_RIGHTDRAG, (char_u *)"RightDrag"},
2545 {K_RIGHTRELEASE, (char_u *)"RightRelease"},
Bram Moolenaar8d9b40e2010-07-25 15:49:07 +02002546 {K_MOUSEDOWN, (char_u *)"ScrollWheelUp"},
2547 {K_MOUSEUP, (char_u *)"ScrollWheelDown"},
2548 {K_MOUSELEFT, (char_u *)"ScrollWheelRight"},
2549 {K_MOUSERIGHT, (char_u *)"ScrollWheelLeft"},
Bram Moolenaar85a20022019-12-21 18:25:54 +01002550 {K_MOUSEDOWN, (char_u *)"MouseDown"}, // OBSOLETE: Use
2551 {K_MOUSEUP, (char_u *)"MouseUp"}, // ScrollWheelXXX instead
Bram Moolenaar071d4272004-06-13 20:20:40 +00002552 {K_X1MOUSE, (char_u *)"X1Mouse"},
2553 {K_X1DRAG, (char_u *)"X1Drag"},
2554 {K_X1RELEASE, (char_u *)"X1Release"},
2555 {K_X2MOUSE, (char_u *)"X2Mouse"},
2556 {K_X2DRAG, (char_u *)"X2Drag"},
2557 {K_X2RELEASE, (char_u *)"X2Release"},
2558 {K_DROP, (char_u *)"Drop"},
2559 {K_ZERO, (char_u *)"Nul"},
2560#ifdef FEAT_EVAL
2561 {K_SNR, (char_u *)"SNR"},
2562#endif
2563 {K_PLUG, (char_u *)"Plug"},
Bram Moolenaar1db60c42014-09-23 16:49:46 +02002564 {K_CURSORHOLD, (char_u *)"CursorHold"},
Bram Moolenaar2f106582019-05-08 21:59:25 +02002565 {K_IGNORE, (char_u *)"Ignore"},
Bram Moolenaar957cf672020-11-12 14:21:06 +01002566 {K_COMMAND, (char_u *)"Cmd"},
Bram Moolenaarccb47a22021-01-21 13:36:43 +01002567 {K_FOCUSGAINED, (char_u *)"FocusGained"},
2568 {K_FOCUSLOST, (char_u *)"FocusLost"},
Bram Moolenaar071d4272004-06-13 20:20:40 +00002569 {0, NULL}
Bram Moolenaar85a20022019-12-21 18:25:54 +01002570 // NOTE: When adding a long name update MAX_KEY_NAME_LEN.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002571};
2572
K.Takataeeec2542021-06-02 13:28:16 +02002573#define KEY_NAMES_TABLE_LEN ARRAY_LENGTH(key_names_table)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002574
Bram Moolenaar071d4272004-06-13 20:20:40 +00002575/*
2576 * Return the modifier mask bit (MOD_MASK_*) which corresponds to the given
2577 * modifier name ('S' for Shift, 'C' for Ctrl etc).
2578 */
Bram Moolenaar5843f5f2019-08-20 20:13:45 +02002579 static int
Bram Moolenaar9b578142016-01-30 19:39:49 +01002580name_to_mod_mask(int c)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002581{
2582 int i;
2583
2584 c = TOUPPER_ASC(c);
2585 for (i = 0; mod_mask_table[i].mod_mask != 0; i++)
2586 if (c == mod_mask_table[i].name)
2587 return mod_mask_table[i].mod_flag;
2588 return 0;
2589}
2590
Bram Moolenaar071d4272004-06-13 20:20:40 +00002591/*
2592 * Check if if there is a special key code for "key" that includes the
2593 * modifiers specified.
2594 */
2595 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01002596simplify_key(int key, int *modifiers)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002597{
2598 int i;
2599 int key0;
2600 int key1;
2601
2602 if (*modifiers & (MOD_MASK_SHIFT | MOD_MASK_CTRL | MOD_MASK_ALT))
2603 {
Bram Moolenaar85a20022019-12-21 18:25:54 +01002604 // TAB is a special case
Bram Moolenaar071d4272004-06-13 20:20:40 +00002605 if (key == TAB && (*modifiers & MOD_MASK_SHIFT))
2606 {
2607 *modifiers &= ~MOD_MASK_SHIFT;
2608 return K_S_TAB;
2609 }
2610 key0 = KEY2TERMCAP0(key);
2611 key1 = KEY2TERMCAP1(key);
2612 for (i = 0; modifier_keys_table[i] != NUL; i += MOD_KEYS_ENTRY_SIZE)
2613 if (key0 == modifier_keys_table[i + 3]
2614 && key1 == modifier_keys_table[i + 4]
2615 && (*modifiers & modifier_keys_table[i]))
2616 {
2617 *modifiers &= ~modifier_keys_table[i];
2618 return TERMCAP2KEY(modifier_keys_table[i + 1],
2619 modifier_keys_table[i + 2]);
2620 }
2621 }
2622 return key;
2623}
2624
2625/*
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002626 * Change <xHome> to <Home>, <xUp> to <Up>, etc.
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002627 */
2628 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01002629handle_x_keys(int key)
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002630{
2631 switch (key)
2632 {
2633 case K_XUP: return K_UP;
2634 case K_XDOWN: return K_DOWN;
2635 case K_XLEFT: return K_LEFT;
2636 case K_XRIGHT: return K_RIGHT;
2637 case K_XHOME: return K_HOME;
Bram Moolenaar68b76a62005-03-25 21:53:48 +00002638 case K_ZHOME: return K_HOME;
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002639 case K_XEND: return K_END;
Bram Moolenaar68b76a62005-03-25 21:53:48 +00002640 case K_ZEND: return K_END;
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002641 case K_XF1: return K_F1;
2642 case K_XF2: return K_F2;
2643 case K_XF3: return K_F3;
2644 case K_XF4: return K_F4;
2645 case K_S_XF1: return K_S_F1;
2646 case K_S_XF2: return K_S_F2;
2647 case K_S_XF3: return K_S_F3;
2648 case K_S_XF4: return K_S_F4;
2649 }
2650 return key;
2651}
2652
2653/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002654 * Return a string which contains the name of the given key when the given
2655 * modifiers are down.
2656 */
2657 char_u *
Bram Moolenaar9b578142016-01-30 19:39:49 +01002658get_special_key_name(int c, int modifiers)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002659{
2660 static char_u string[MAX_KEY_NAME_LEN + 1];
2661
2662 int i, idx;
2663 int table_idx;
2664 char_u *s;
2665
2666 string[0] = '<';
2667 idx = 1;
2668
Bram Moolenaar85a20022019-12-21 18:25:54 +01002669 // Key that stands for a normal character.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002670 if (IS_SPECIAL(c) && KEY2TERMCAP0(c) == KS_KEY)
2671 c = KEY2TERMCAP1(c);
2672
2673 /*
2674 * Translate shifted special keys into unshifted keys and set modifier.
2675 * Same for CTRL and ALT modifiers.
2676 */
2677 if (IS_SPECIAL(c))
2678 {
2679 for (i = 0; modifier_keys_table[i] != 0; i += MOD_KEYS_ENTRY_SIZE)
2680 if ( KEY2TERMCAP0(c) == (int)modifier_keys_table[i + 1]
2681 && (int)KEY2TERMCAP1(c) == (int)modifier_keys_table[i + 2])
2682 {
2683 modifiers |= modifier_keys_table[i];
2684 c = TERMCAP2KEY(modifier_keys_table[i + 3],
2685 modifier_keys_table[i + 4]);
2686 break;
2687 }
2688 }
2689
Bram Moolenaar85a20022019-12-21 18:25:54 +01002690 // try to find the key in the special key table
Bram Moolenaar071d4272004-06-13 20:20:40 +00002691 table_idx = find_special_key_in_table(c);
2692
2693 /*
2694 * When not a known special key, and not a printable character, try to
2695 * extract modifiers.
2696 */
Bram Moolenaarfc3abf42019-01-24 15:54:21 +01002697 if (c > 0 && (*mb_char2len)(c) == 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002698 {
2699 if (table_idx < 0
2700 && (!vim_isprintc(c) || (c & 0x7f) == ' ')
2701 && (c & 0x80))
2702 {
2703 c &= 0x7f;
2704 modifiers |= MOD_MASK_ALT;
Bram Moolenaar85a20022019-12-21 18:25:54 +01002705 // try again, to find the un-alted key in the special key table
Bram Moolenaar071d4272004-06-13 20:20:40 +00002706 table_idx = find_special_key_in_table(c);
2707 }
2708 if (table_idx < 0 && !vim_isprintc(c) && c < ' ')
2709 {
2710#ifdef EBCDIC
2711 c = CtrlChar(c);
2712#else
2713 c += '@';
2714#endif
2715 modifiers |= MOD_MASK_CTRL;
2716 }
2717 }
2718
Bram Moolenaar85a20022019-12-21 18:25:54 +01002719 // translate the modifier into a string
Bram Moolenaar071d4272004-06-13 20:20:40 +00002720 for (i = 0; mod_mask_table[i].name != 'A'; i++)
2721 if ((modifiers & mod_mask_table[i].mod_mask)
2722 == mod_mask_table[i].mod_flag)
2723 {
2724 string[idx++] = mod_mask_table[i].name;
2725 string[idx++] = (char_u)'-';
2726 }
2727
Bram Moolenaar85a20022019-12-21 18:25:54 +01002728 if (table_idx < 0) // unknown special key, may output t_xx
Bram Moolenaar071d4272004-06-13 20:20:40 +00002729 {
2730 if (IS_SPECIAL(c))
2731 {
2732 string[idx++] = 't';
2733 string[idx++] = '_';
2734 string[idx++] = KEY2TERMCAP0(c);
2735 string[idx++] = KEY2TERMCAP1(c);
2736 }
Bram Moolenaar85a20022019-12-21 18:25:54 +01002737 // Not a special key, only modifiers, output directly
Bram Moolenaar071d4272004-06-13 20:20:40 +00002738 else
2739 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00002740 if (has_mbyte && (*mb_char2len)(c) > 1)
2741 idx += (*mb_char2bytes)(c, string + idx);
Bram Moolenaarfc3abf42019-01-24 15:54:21 +01002742 else if (vim_isprintc(c))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002743 string[idx++] = c;
2744 else
2745 {
2746 s = transchar(c);
2747 while (*s)
2748 string[idx++] = *s++;
2749 }
2750 }
2751 }
Bram Moolenaar85a20022019-12-21 18:25:54 +01002752 else // use name of special key
Bram Moolenaar071d4272004-06-13 20:20:40 +00002753 {
Bram Moolenaar423977d2017-01-22 15:05:12 +01002754 size_t len = STRLEN(key_names_table[table_idx].name);
2755
2756 if (len + idx + 2 <= MAX_KEY_NAME_LEN)
2757 {
2758 STRCPY(string + idx, key_names_table[table_idx].name);
2759 idx += (int)len;
2760 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002761 }
2762 string[idx++] = '>';
2763 string[idx] = NUL;
2764 return string;
2765}
2766
2767/*
2768 * Try translating a <> name at (*srcp)[] to dst[].
2769 * Return the number of characters added to dst[], zero for no match.
2770 * If there is a match, srcp is advanced to after the <> name.
2771 * dst[] must be big enough to hold the result (up to six characters)!
2772 */
2773 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01002774trans_special(
2775 char_u **srcp,
2776 char_u *dst,
Bram Moolenaarebe9d342020-05-30 21:52:54 +02002777 int flags, // FSK_ values
2778 int *did_simplify) // FSK_SIMPLIFY and found <C-H> or <A-x>
Bram Moolenaar071d4272004-06-13 20:20:40 +00002779{
2780 int modifiers = 0;
2781 int key;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002782
Bram Moolenaarebe9d342020-05-30 21:52:54 +02002783 key = find_special_key(srcp, &modifiers, flags, did_simplify);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002784 if (key == 0)
2785 return 0;
2786
Bram Moolenaarebe9d342020-05-30 21:52:54 +02002787 return special_to_buf(key, modifiers, flags & FSK_KEYCODE, dst);
Bram Moolenaarbf0eff02019-06-01 17:13:36 +02002788}
2789
2790/*
2791 * Put the character sequence for "key" with "modifiers" into "dst" and return
2792 * the resulting length.
2793 * When "keycode" is TRUE prefer key code, e.g. K_DEL instead of DEL.
2794 * The sequence is not NUL terminated.
2795 * This is how characters in a string are encoded.
2796 */
2797 int
2798special_to_buf(int key, int modifiers, int keycode, char_u *dst)
2799{
2800 int dlen = 0;
2801
Bram Moolenaar85a20022019-12-21 18:25:54 +01002802 // Put the appropriate modifier in a string
Bram Moolenaar071d4272004-06-13 20:20:40 +00002803 if (modifiers != 0)
2804 {
2805 dst[dlen++] = K_SPECIAL;
2806 dst[dlen++] = KS_MODIFIER;
2807 dst[dlen++] = modifiers;
2808 }
2809
2810 if (IS_SPECIAL(key))
2811 {
2812 dst[dlen++] = K_SPECIAL;
2813 dst[dlen++] = KEY2TERMCAP0(key);
2814 dst[dlen++] = KEY2TERMCAP1(key);
2815 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002816 else if (has_mbyte && !keycode)
2817 dlen += (*mb_char2bytes)(key, dst + dlen);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002818 else if (keycode)
2819 dlen = (int)(add_char2buf(key, dst + dlen) - dst);
2820 else
2821 dst[dlen++] = key;
2822
2823 return dlen;
2824}
2825
2826/*
2827 * Try translating a <> name at (*srcp)[], return the key and modifiers.
2828 * srcp is advanced to after the <> name.
2829 * returns 0 if there is no match.
2830 */
2831 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01002832find_special_key(
2833 char_u **srcp,
2834 int *modp,
Bram Moolenaarebe9d342020-05-30 21:52:54 +02002835 int flags, // FSK_ values
Bram Moolenaar459fd782019-10-13 16:43:39 +02002836 int *did_simplify) // found <C-H> or <A-x>
Bram Moolenaar071d4272004-06-13 20:20:40 +00002837{
2838 char_u *last_dash;
2839 char_u *end_of_name;
2840 char_u *src;
2841 char_u *bp;
Bram Moolenaarebe9d342020-05-30 21:52:54 +02002842 int in_string = flags & FSK_IN_STRING;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002843 int modifiers;
2844 int bit;
2845 int key;
Bram Moolenaar22fcfad2016-07-01 18:17:26 +02002846 uvarnumber_T n;
Bram Moolenaarb8bf5412011-08-17 20:33:22 +02002847 int l;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002848
2849 src = *srcp;
Bram Moolenaarfccd93f2020-05-31 22:06:51 +02002850 if (src[0] != '<')
Bram Moolenaar071d4272004-06-13 20:20:40 +00002851 return 0;
Bram Moolenaarfccd93f2020-05-31 22:06:51 +02002852 if (src[1] == '*') // <*xxx>: do not simplify
2853 ++src;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002854
Bram Moolenaar85a20022019-12-21 18:25:54 +01002855 // Find end of modifier list
Bram Moolenaar071d4272004-06-13 20:20:40 +00002856 last_dash = src;
Bram Moolenaare3d1f4c2021-04-06 20:21:59 +02002857 for (bp = src + 1; *bp == '-' || vim_isNormalIDc(*bp); bp++)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002858 {
2859 if (*bp == '-')
2860 {
2861 last_dash = bp;
Bram Moolenaarb8bf5412011-08-17 20:33:22 +02002862 if (bp[1] != NUL)
2863 {
Bram Moolenaarb8bf5412011-08-17 20:33:22 +02002864 if (has_mbyte)
2865 l = mb_ptr2len(bp + 1);
2866 else
Bram Moolenaarb8bf5412011-08-17 20:33:22 +02002867 l = 1;
Bram Moolenaarc8fd33d2019-08-16 20:33:05 +02002868 // Anything accepted, like <C-?>.
2869 // <C-"> or <M-"> are not special in strings as " is
2870 // the string delimiter. With a backslash it works: <M-\">
Bram Moolenaarfccd93f2020-05-31 22:06:51 +02002871 if (!(in_string && bp[1] == '"') && bp[l + 1] == '>')
Bram Moolenaar1d90a5a2016-07-01 11:59:47 +02002872 bp += l;
Bram Moolenaar35a4cfa2016-08-14 16:07:48 +02002873 else if (in_string && bp[1] == '\\' && bp[2] == '"'
Bram Moolenaarfccd93f2020-05-31 22:06:51 +02002874 && bp[3] == '>')
Bram Moolenaar35a4cfa2016-08-14 16:07:48 +02002875 bp += 2;
Bram Moolenaarb8bf5412011-08-17 20:33:22 +02002876 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002877 }
2878 if (bp[0] == 't' && bp[1] == '_' && bp[2] && bp[3])
Bram Moolenaarfccd93f2020-05-31 22:06:51 +02002879 bp += 3; // skip t_xx, xx may be '-' or '>'
Bram Moolenaar792826c2011-08-19 22:29:02 +02002880 else if (STRNICMP(bp, "char-", 5) == 0)
2881 {
Bram Moolenaar16e9b852019-05-19 19:59:35 +02002882 vim_str2nr(bp + 5, NULL, &l, STR2NR_ALL, NULL, NULL, 0, TRUE);
2883 if (l == 0)
2884 {
2885 emsg(_(e_invarg));
2886 return 0;
2887 }
Bram Moolenaar792826c2011-08-19 22:29:02 +02002888 bp += l + 5;
2889 break;
2890 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002891 }
2892
Bram Moolenaarfccd93f2020-05-31 22:06:51 +02002893 if (*bp == '>') // found matching '>'
Bram Moolenaar071d4272004-06-13 20:20:40 +00002894 {
2895 end_of_name = bp + 1;
2896
Bram Moolenaar85a20022019-12-21 18:25:54 +01002897 // Which modifiers are given?
Bram Moolenaar071d4272004-06-13 20:20:40 +00002898 modifiers = 0x0;
2899 for (bp = src + 1; bp < last_dash; bp++)
2900 {
2901 if (*bp != '-')
2902 {
2903 bit = name_to_mod_mask(*bp);
2904 if (bit == 0x0)
Bram Moolenaar85a20022019-12-21 18:25:54 +01002905 break; // Illegal modifier name
Bram Moolenaar071d4272004-06-13 20:20:40 +00002906 modifiers |= bit;
2907 }
2908 }
2909
2910 /*
2911 * Legal modifier name.
2912 */
2913 if (bp >= last_dash)
2914 {
Bram Moolenaarb8bf5412011-08-17 20:33:22 +02002915 if (STRNICMP(last_dash + 1, "char-", 5) == 0
2916 && VIM_ISDIGIT(last_dash[6]))
2917 {
Bram Moolenaar85a20022019-12-21 18:25:54 +01002918 // <Char-123> or <Char-033> or <Char-0x33>
Bram Moolenaar459fd782019-10-13 16:43:39 +02002919 vim_str2nr(last_dash + 6, NULL, &l, STR2NR_ALL, NULL,
2920 &n, 0, TRUE);
Bram Moolenaar16e9b852019-05-19 19:59:35 +02002921 if (l == 0)
2922 {
2923 emsg(_(e_invarg));
2924 return 0;
2925 }
Bram Moolenaar792826c2011-08-19 22:29:02 +02002926 key = (int)n;
Bram Moolenaarb8bf5412011-08-17 20:33:22 +02002927 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002928 else
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002929 {
Bram Moolenaar35a4cfa2016-08-14 16:07:48 +02002930 int off = 1;
2931
Bram Moolenaar85a20022019-12-21 18:25:54 +01002932 // Modifier with single letter, or special key name.
Bram Moolenaar35a4cfa2016-08-14 16:07:48 +02002933 if (in_string && last_dash[1] == '\\' && last_dash[2] == '"')
2934 off = 2;
Bram Moolenaar792826c2011-08-19 22:29:02 +02002935 if (has_mbyte)
Bram Moolenaar35a4cfa2016-08-14 16:07:48 +02002936 l = mb_ptr2len(last_dash + off);
Bram Moolenaar792826c2011-08-19 22:29:02 +02002937 else
Bram Moolenaar792826c2011-08-19 22:29:02 +02002938 l = 1;
Bram Moolenaarfccd93f2020-05-31 22:06:51 +02002939 if (modifiers != 0 && last_dash[l + off] == '>')
Bram Moolenaar35a4cfa2016-08-14 16:07:48 +02002940 key = PTR2CHAR(last_dash + off);
Bram Moolenaar792826c2011-08-19 22:29:02 +02002941 else
2942 {
Bram Moolenaar35a4cfa2016-08-14 16:07:48 +02002943 key = get_special_key_code(last_dash + off);
Bram Moolenaarebe9d342020-05-30 21:52:54 +02002944 if (!(flags & FSK_KEEP_X_KEY))
Bram Moolenaar792826c2011-08-19 22:29:02 +02002945 key = handle_x_keys(key);
2946 }
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002947 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002948
2949 /*
2950 * get_special_key_code() may return NUL for invalid
2951 * special key name.
2952 */
2953 if (key != NUL)
2954 {
2955 /*
2956 * Only use a modifier when there is no special key code that
2957 * includes the modifier.
2958 */
2959 key = simplify_key(key, &modifiers);
2960
Bram Moolenaarebe9d342020-05-30 21:52:54 +02002961 if (!(flags & FSK_KEYCODE))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002962 {
Bram Moolenaar85a20022019-12-21 18:25:54 +01002963 // don't want keycode, use single byte code
Bram Moolenaar071d4272004-06-13 20:20:40 +00002964 if (key == K_BS)
2965 key = BS;
2966 else if (key == K_DEL || key == K_KDEL)
2967 key = DEL;
2968 }
2969
Bram Moolenaar459fd782019-10-13 16:43:39 +02002970 // Normal Key with modifier: Try to make a single byte code.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002971 if (!IS_SPECIAL(key))
Bram Moolenaar459fd782019-10-13 16:43:39 +02002972 key = extract_modifiers(key, &modifiers,
Bram Moolenaarebe9d342020-05-30 21:52:54 +02002973 flags & FSK_SIMPLIFY, did_simplify);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002974
2975 *modp = modifiers;
2976 *srcp = end_of_name;
2977 return key;
2978 }
2979 }
2980 }
2981 return 0;
2982}
2983
Bram Moolenaaref6746f2020-06-20 14:43:23 +02002984
2985/*
Bram Moolenaar4e2114e2020-10-07 16:12:37 +02002986 * Some keys are used with Ctrl without Shift and are still expected to be
2987 * mapped as if Shift was pressed:
2988 * CTRL-2 is CTRL-@
2989 * CTRL-6 is CTRL-^
2990 * CTRL-- is CTRL-_
2991 * Also, <C-H> and <C-h> mean the same thing, always use "H".
2992 * Returns the possibly adjusted key.
2993 */
2994 int
2995may_adjust_key_for_ctrl(int modifiers, int key)
2996{
2997 if (modifiers & MOD_MASK_CTRL)
2998 {
2999 if (ASCII_ISALPHA(key))
3000 return TOUPPER_ASC(key);
3001 if (key == '2')
3002 return '@';
3003 if (key == '6')
3004 return '^';
3005 if (key == '-')
3006 return '_';
3007 }
3008 return key;
3009}
3010
3011/*
Bram Moolenaaref6746f2020-06-20 14:43:23 +02003012 * Some keys already have Shift included, pass them as normal keys.
Bram Moolenaar9a033d72020-10-07 17:29:48 +02003013 * When Ctrl is also used <C-H> and <C-S-H> are different, but <C-S-{> should
3014 * be <C-{>. Same for <C-S-}> and <C-S-|>.
Bram Moolenaaref6746f2020-06-20 14:43:23 +02003015 * Also for <A-S-a> and <M-S-a>.
Bram Moolenaardaff0fb2020-09-27 13:16:46 +02003016 * This includes all printable ASCII characters except numbers and a-z.
Bram Moolenaaref6746f2020-06-20 14:43:23 +02003017 */
3018 int
3019may_remove_shift_modifier(int modifiers, int key)
3020{
3021 if ((modifiers == MOD_MASK_SHIFT
3022 || modifiers == (MOD_MASK_SHIFT | MOD_MASK_ALT)
3023 || modifiers == (MOD_MASK_SHIFT | MOD_MASK_META))
Bram Moolenaardaff0fb2020-09-27 13:16:46 +02003024 && ((key >= '!' && key <= '/')
3025 || (key >= ':' && key <= 'Z')
3026 || (key >= '[' && key <= '`')
Bram Moolenaaref6746f2020-06-20 14:43:23 +02003027 || (key >= '{' && key <= '~')))
3028 return modifiers & ~MOD_MASK_SHIFT;
Bram Moolenaar9a033d72020-10-07 17:29:48 +02003029
3030 if (modifiers == (MOD_MASK_SHIFT | MOD_MASK_CTRL)
3031 && (key == '{' || key == '}' || key == '|'))
3032 return modifiers & ~MOD_MASK_SHIFT;
3033
Bram Moolenaaref6746f2020-06-20 14:43:23 +02003034 return modifiers;
3035}
3036
Bram Moolenaar071d4272004-06-13 20:20:40 +00003037/*
3038 * Try to include modifiers in the key.
3039 * Changes "Shift-a" to 'A', "Alt-A" to 0xc0, etc.
Bram Moolenaar459fd782019-10-13 16:43:39 +02003040 * When "simplify" is FALSE don't do Ctrl and Alt.
3041 * When "simplify" is TRUE and Ctrl or Alt is removed from modifiers set
3042 * "did_simplify" when it's not NULL.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003043 */
3044 int
Bram Moolenaar459fd782019-10-13 16:43:39 +02003045extract_modifiers(int key, int *modp, int simplify, int *did_simplify)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003046{
3047 int modifiers = *modp;
3048
Bram Moolenaard0573012017-10-28 21:11:06 +02003049#ifdef MACOS_X
Bram Moolenaar459fd782019-10-13 16:43:39 +02003050 // Command-key really special, no fancynest
Bram Moolenaar071d4272004-06-13 20:20:40 +00003051 if (!(modifiers & MOD_MASK_CMD))
3052#endif
3053 if ((modifiers & MOD_MASK_SHIFT) && ASCII_ISALPHA(key))
3054 {
3055 key = TOUPPER_ASC(key);
Bram Moolenaar20298ce2020-06-19 21:46:52 +02003056 // With <C-S-a> we keep the shift modifier.
3057 // With <S-a>, <A-S-a> and <S-A> we don't keep the shift modifier.
3058 if (simplify || modifiers == MOD_MASK_SHIFT
3059 || modifiers == (MOD_MASK_SHIFT | MOD_MASK_ALT)
3060 || modifiers == (MOD_MASK_SHIFT | MOD_MASK_META))
Bram Moolenaar459fd782019-10-13 16:43:39 +02003061 modifiers &= ~MOD_MASK_SHIFT;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003062 }
Bram Moolenaar459fd782019-10-13 16:43:39 +02003063
3064 // <C-H> and <C-h> mean the same thing, always use "H"
3065 if ((modifiers & MOD_MASK_CTRL) && ASCII_ISALPHA(key))
3066 key = TOUPPER_ASC(key);
3067
3068 if (simplify && (modifiers & MOD_MASK_CTRL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003069#ifdef EBCDIC
Bram Moolenaar459fd782019-10-13 16:43:39 +02003070 // TODO: EBCDIC Better use:
3071 // && (Ctrl_chr(key) || key == '?')
3072 // ???
Bram Moolenaar071d4272004-06-13 20:20:40 +00003073 && strchr("?@ABCDEFGHIJKLMNOPQRSTUVWXYZ[\\]^_", key)
3074 != NULL
3075#else
3076 && ((key >= '?' && key <= '_') || ASCII_ISALPHA(key))
3077#endif
3078 )
3079 {
3080 key = Ctrl_chr(key);
3081 modifiers &= ~MOD_MASK_CTRL;
Bram Moolenaar85a20022019-12-21 18:25:54 +01003082 // <C-@> is <Nul>
Bram Moolenaar071d4272004-06-13 20:20:40 +00003083 if (key == 0)
3084 key = K_ZERO;
Bram Moolenaar459fd782019-10-13 16:43:39 +02003085 if (did_simplify != NULL)
3086 *did_simplify = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003087 }
Bram Moolenaar459fd782019-10-13 16:43:39 +02003088
Bram Moolenaard0573012017-10-28 21:11:06 +02003089#ifdef MACOS_X
Bram Moolenaar85a20022019-12-21 18:25:54 +01003090 // Command-key really special, no fancynest
Bram Moolenaar071d4272004-06-13 20:20:40 +00003091 if (!(modifiers & MOD_MASK_CMD))
3092#endif
Bram Moolenaar459fd782019-10-13 16:43:39 +02003093 if (simplify && (modifiers & MOD_MASK_ALT) && key < 0x80
Bram Moolenaarfc3abf42019-01-24 15:54:21 +01003094 && !enc_dbcs) // avoid creating a lead byte
Bram Moolenaar071d4272004-06-13 20:20:40 +00003095 {
3096 key |= 0x80;
Bram Moolenaar85a20022019-12-21 18:25:54 +01003097 modifiers &= ~MOD_MASK_ALT; // remove the META modifier
Bram Moolenaar459fd782019-10-13 16:43:39 +02003098 if (did_simplify != NULL)
3099 *did_simplify = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003100 }
3101
3102 *modp = modifiers;
3103 return key;
3104}
3105
3106/*
3107 * Try to find key "c" in the special key table.
3108 * Return the index when found, -1 when not found.
3109 */
3110 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01003111find_special_key_in_table(int c)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003112{
3113 int i;
3114
3115 for (i = 0; key_names_table[i].name != NULL; i++)
3116 if (c == key_names_table[i].key)
3117 break;
3118 if (key_names_table[i].name == NULL)
3119 i = -1;
3120 return i;
3121}
3122
3123/*
3124 * Find the special key with the given name (the given string does not have to
3125 * end with NUL, the name is assumed to end before the first non-idchar).
3126 * If the name starts with "t_" the next two characters are interpreted as a
3127 * termcap name.
3128 * Return the key code, or 0 if not found.
3129 */
3130 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01003131get_special_key_code(char_u *name)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003132{
3133 char_u *table_name;
3134 char_u string[3];
3135 int i, j;
3136
3137 /*
3138 * If it's <t_xx> we get the code for xx from the termcap
3139 */
3140 if (name[0] == 't' && name[1] == '_' && name[2] != NUL && name[3] != NUL)
3141 {
3142 string[0] = name[2];
3143 string[1] = name[3];
3144 string[2] = NUL;
3145 if (add_termcap_entry(string, FALSE) == OK)
3146 return TERMCAP2KEY(name[2], name[3]);
3147 }
3148 else
3149 for (i = 0; key_names_table[i].name != NULL; i++)
3150 {
3151 table_name = key_names_table[i].name;
Bram Moolenaare3d1f4c2021-04-06 20:21:59 +02003152 for (j = 0; vim_isNormalIDc(name[j]) && table_name[j] != NUL; j++)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003153 if (TOLOWER_ASC(table_name[j]) != TOLOWER_ASC(name[j]))
3154 break;
Bram Moolenaare3d1f4c2021-04-06 20:21:59 +02003155 if (!vim_isNormalIDc(name[j]) && table_name[j] == NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003156 return key_names_table[i].key;
3157 }
3158 return 0;
3159}
3160
Bram Moolenaar071d4272004-06-13 20:20:40 +00003161 char_u *
Bram Moolenaar9b578142016-01-30 19:39:49 +01003162get_key_name(int i)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003163{
Bram Moolenaare1fbddc2009-05-16 19:07:03 +00003164 if (i >= (int)KEY_NAMES_TABLE_LEN)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003165 return NULL;
3166 return key_names_table[i].name;
3167}
Bram Moolenaar071d4272004-06-13 20:20:40 +00003168
Bram Moolenaar071d4272004-06-13 20:20:40 +00003169/*
3170 * Return the current end-of-line type: EOL_DOS, EOL_UNIX or EOL_MAC.
3171 */
3172 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01003173get_fileformat(buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003174{
3175 int c = *buf->b_p_ff;
3176
3177 if (buf->b_p_bin || c == 'u')
3178 return EOL_UNIX;
3179 if (c == 'm')
3180 return EOL_MAC;
3181 return EOL_DOS;
3182}
3183
3184/*
3185 * Like get_fileformat(), but override 'fileformat' with "p" for "++opt=val"
3186 * argument.
3187 */
3188 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01003189get_fileformat_force(
3190 buf_T *buf,
Bram Moolenaar85a20022019-12-21 18:25:54 +01003191 exarg_T *eap) // can be NULL!
Bram Moolenaar071d4272004-06-13 20:20:40 +00003192{
3193 int c;
3194
3195 if (eap != NULL && eap->force_ff != 0)
Bram Moolenaar333b80a2018-04-04 22:57:29 +02003196 c = eap->force_ff;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003197 else
3198 {
3199 if ((eap != NULL && eap->force_bin != 0)
3200 ? (eap->force_bin == FORCE_BIN) : buf->b_p_bin)
3201 return EOL_UNIX;
3202 c = *buf->b_p_ff;
3203 }
3204 if (c == 'u')
3205 return EOL_UNIX;
3206 if (c == 'm')
3207 return EOL_MAC;
3208 return EOL_DOS;
3209}
3210
3211/*
3212 * Set the current end-of-line type to EOL_DOS, EOL_UNIX or EOL_MAC.
3213 * Sets both 'textmode' and 'fileformat'.
3214 * Note: Does _not_ set global value of 'textmode'!
3215 */
3216 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01003217set_fileformat(
3218 int t,
Bram Moolenaar85a20022019-12-21 18:25:54 +01003219 int opt_flags) // OPT_LOCAL and/or OPT_GLOBAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00003220{
3221 char *p = NULL;
3222
3223 switch (t)
3224 {
3225 case EOL_DOS:
3226 p = FF_DOS;
3227 curbuf->b_p_tx = TRUE;
3228 break;
3229 case EOL_UNIX:
3230 p = FF_UNIX;
3231 curbuf->b_p_tx = FALSE;
3232 break;
3233 case EOL_MAC:
3234 p = FF_MAC;
3235 curbuf->b_p_tx = FALSE;
3236 break;
3237 }
3238 if (p != NULL)
3239 set_string_option_direct((char_u *)"ff", -1, (char_u *)p,
Bram Moolenaar5e3cb7e2006-02-27 23:58:35 +00003240 OPT_FREE | opt_flags, 0);
3241
Bram Moolenaar85a20022019-12-21 18:25:54 +01003242 // This may cause the buffer to become (un)modified.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003243 check_status(curbuf);
Bram Moolenaar997fb4b2006-02-17 21:53:23 +00003244 redraw_tabline = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003245#ifdef FEAT_TITLE
Bram Moolenaar85a20022019-12-21 18:25:54 +01003246 need_maketitle = TRUE; // set window title later
Bram Moolenaar071d4272004-06-13 20:20:40 +00003247#endif
3248}
3249
3250/*
3251 * Return the default fileformat from 'fileformats'.
3252 */
3253 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01003254default_fileformat(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003255{
3256 switch (*p_ffs)
3257 {
3258 case 'm': return EOL_MAC;
3259 case 'd': return EOL_DOS;
3260 }
3261 return EOL_UNIX;
3262}
3263
3264/*
3265 * Call shell. Calls mch_call_shell, with 'shellxquote' added.
3266 */
3267 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01003268call_shell(char_u *cmd, int opt)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003269{
3270 char_u *ncmd;
3271 int retval;
Bram Moolenaar05159a02005-02-26 23:04:13 +00003272#ifdef FEAT_PROFILE
3273 proftime_T wait_time;
3274#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003275
3276 if (p_verbose > 3)
3277 {
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00003278 verbose_enter();
Bram Moolenaar06f08532020-05-12 23:45:16 +02003279 smsg(_("Calling shell to execute: \"%s\""), cmd == NULL ? p_sh : cmd);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003280 out_char('\n');
3281 cursor_on();
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00003282 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003283 }
3284
Bram Moolenaar05159a02005-02-26 23:04:13 +00003285#ifdef FEAT_PROFILE
Bram Moolenaar01265852006-03-20 21:50:15 +00003286 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003287 prof_child_enter(&wait_time);
3288#endif
3289
Bram Moolenaar071d4272004-06-13 20:20:40 +00003290 if (*p_sh == NUL)
3291 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003292 emsg(_(e_shellempty));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003293 retval = -1;
3294 }
3295 else
3296 {
3297#ifdef FEAT_GUI_MSWIN
Bram Moolenaar85a20022019-12-21 18:25:54 +01003298 // Don't hide the pointer while executing a shell command.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003299 gui_mch_mousehide(FALSE);
3300#endif
3301#ifdef FEAT_GUI
3302 ++hold_gui_events;
3303#endif
Bram Moolenaar85a20022019-12-21 18:25:54 +01003304 // The external command may update a tags file, clear cached tags.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003305 tag_freematch();
3306
Bram Moolenaar01257a72019-06-10 14:46:04 +02003307 if (cmd == NULL || *p_sxq == NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003308 retval = mch_call_shell(cmd, opt);
3309 else
3310 {
Bram Moolenaarf66b3fc2012-02-20 22:18:30 +01003311 char_u *ecmd = cmd;
3312
Bram Moolenaar1a613392019-09-28 15:51:37 +02003313 if (*p_sxe != NUL && *p_sxq == '(')
Bram Moolenaarf66b3fc2012-02-20 22:18:30 +01003314 {
3315 ecmd = vim_strsave_escaped_ext(cmd, p_sxe, '^', FALSE);
3316 if (ecmd == NULL)
3317 ecmd = cmd;
3318 }
Bram Moolenaar964b3742019-05-24 18:54:09 +02003319 ncmd = alloc(STRLEN(ecmd) + STRLEN(p_sxq) * 2 + 1);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003320 if (ncmd != NULL)
3321 {
3322 STRCPY(ncmd, p_sxq);
Bram Moolenaarf66b3fc2012-02-20 22:18:30 +01003323 STRCAT(ncmd, ecmd);
Bram Moolenaar1a613392019-09-28 15:51:37 +02003324 // When 'shellxquote' is ( append ).
3325 // When 'shellxquote' is "( append )".
3326 STRCAT(ncmd, *p_sxq == '(' ? (char_u *)")"
3327 : *p_sxq == '"' && *(p_sxq+1) == '(' ? (char_u *)")\""
3328 : p_sxq);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003329 retval = mch_call_shell(ncmd, opt);
3330 vim_free(ncmd);
3331 }
3332 else
3333 retval = -1;
Bram Moolenaarf66b3fc2012-02-20 22:18:30 +01003334 if (ecmd != cmd)
3335 vim_free(ecmd);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003336 }
3337#ifdef FEAT_GUI
3338 --hold_gui_events;
3339#endif
3340 /*
3341 * Check the window size, in case it changed while executing the
3342 * external command.
3343 */
3344 shell_resized_check();
3345 }
3346
3347#ifdef FEAT_EVAL
3348 set_vim_var_nr(VV_SHELL_ERROR, (long)retval);
Bram Moolenaar05159a02005-02-26 23:04:13 +00003349# ifdef FEAT_PROFILE
Bram Moolenaar01265852006-03-20 21:50:15 +00003350 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003351 prof_child_exit(&wait_time);
3352# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003353#endif
3354
3355 return retval;
3356}
3357
3358/*
Bram Moolenaar01265852006-03-20 21:50:15 +00003359 * VISUAL, SELECTMODE and OP_PENDING State are never set, they are equal to
3360 * NORMAL State with a condition. This function returns the real State.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003361 */
3362 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01003363get_real_state(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003364{
3365 if (State & NORMAL)
3366 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00003367 if (VIsual_active)
Bram Moolenaar01265852006-03-20 21:50:15 +00003368 {
3369 if (VIsual_select)
3370 return SELECTMODE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003371 return VISUAL;
Bram Moolenaar01265852006-03-20 21:50:15 +00003372 }
Bram Moolenaarf7ff6e82014-03-23 15:13:05 +01003373 else if (finish_op)
3374 return OP_PENDING;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003375 }
3376 return State;
3377}
3378
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003379/*
3380 * Return TRUE if "p" points to just after a path separator.
Bram Moolenaarb5ce04d2011-07-07 17:15:33 +02003381 * Takes care of multi-byte characters.
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003382 * "b" must point to the start of the file name
3383 */
3384 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01003385after_pathsep(char_u *b, char_u *p)
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003386{
Bram Moolenaarb5ce04d2011-07-07 17:15:33 +02003387 return p > b && vim_ispathsep(p[-1])
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003388 && (!has_mbyte || (*mb_head_off)(b, p - 1) == 0);
3389}
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003390
3391/*
3392 * Return TRUE if file names "f1" and "f2" are in the same directory.
3393 * "f1" may be a short name, "f2" must be a full path.
3394 */
3395 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01003396same_directory(char_u *f1, char_u *f2)
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003397{
3398 char_u ffname[MAXPATHL];
3399 char_u *t1;
3400 char_u *t2;
3401
Bram Moolenaar85a20022019-12-21 18:25:54 +01003402 // safety check
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003403 if (f1 == NULL || f2 == NULL)
3404 return FALSE;
3405
3406 (void)vim_FullName(f1, ffname, MAXPATHL, FALSE);
3407 t1 = gettail_sep(ffname);
3408 t2 = gettail_sep(f2);
3409 return (t1 - ffname == t2 - f2
3410 && pathcmp((char *)ffname, (char *)f2, (int)(t1 - ffname)) == 0);
3411}
3412
Bram Moolenaar7c365fb2018-06-29 20:28:31 +02003413#if defined(FEAT_SESSION) || defined(FEAT_AUTOCHDIR) \
Bram Moolenaar097148e2020-08-11 21:58:20 +02003414 || defined(MSWIN) || defined(FEAT_GUI_GTK) \
Bram Moolenaarbb1969b2019-01-17 15:45:25 +01003415 || defined(FEAT_NETBEANS_INTG) \
Bram Moolenaar071d4272004-06-13 20:20:40 +00003416 || defined(PROTO)
3417/*
3418 * Change to a file's directory.
3419 * Caller must call shorten_fnames()!
3420 * Return OK or FAIL.
3421 */
3422 int
Bram Moolenaar2caad3f2018-12-16 15:38:02 +01003423vim_chdirfile(char_u *fname, char *trigger_autocmd)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003424{
Bram Moolenaar2caad3f2018-12-16 15:38:02 +01003425 char_u old_dir[MAXPATHL];
3426 char_u new_dir[MAXPATHL];
Bram Moolenaarb5cb65b2018-02-03 18:01:37 +01003427 int res;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003428
Bram Moolenaar2caad3f2018-12-16 15:38:02 +01003429 if (mch_dirname(old_dir, MAXPATHL) != OK)
3430 *old_dir = NUL;
3431
3432 vim_strncpy(new_dir, fname, MAXPATHL - 1);
3433 *gettail_sep(new_dir) = NUL;
3434
Bram Moolenaar9eb76af2018-12-16 16:30:21 +01003435 if (pathcmp((char *)old_dir, (char *)new_dir, -1) == 0)
Bram Moolenaar2caad3f2018-12-16 15:38:02 +01003436 // nothing to do
3437 res = OK;
3438 else
3439 {
3440 res = mch_chdir((char *)new_dir) == 0 ? OK : FAIL;
3441
3442 if (res == OK && trigger_autocmd != NULL)
3443 apply_autocmds(EVENT_DIRCHANGED, (char_u *)trigger_autocmd,
3444 new_dir, FALSE, curbuf);
3445 }
Bram Moolenaarb5cb65b2018-02-03 18:01:37 +01003446 return res;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003447}
3448#endif
3449
3450#if defined(STAT_IGNORES_SLASH) || defined(PROTO)
3451/*
3452 * Check if "name" ends in a slash and is not a directory.
3453 * Used for systems where stat() ignores a trailing slash on a file name.
3454 * The Vim code assumes a trailing slash is only ignored for a directory.
3455 */
Bram Moolenaar91acfff2017-03-12 19:22:36 +01003456 static int
Bram Moolenaard8492792017-03-16 12:22:38 +01003457illegal_slash(const char *name)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003458{
3459 if (name[0] == NUL)
Bram Moolenaar85a20022019-12-21 18:25:54 +01003460 return FALSE; // no file name is not illegal
Bram Moolenaar071d4272004-06-13 20:20:40 +00003461 if (name[strlen(name) - 1] != '/')
Bram Moolenaar85a20022019-12-21 18:25:54 +01003462 return FALSE; // no trailing slash
Bram Moolenaar071d4272004-06-13 20:20:40 +00003463 if (mch_isdir((char_u *)name))
Bram Moolenaar85a20022019-12-21 18:25:54 +01003464 return FALSE; // trailing slash for a directory
Bram Moolenaar071d4272004-06-13 20:20:40 +00003465 return TRUE;
3466}
Bram Moolenaar91acfff2017-03-12 19:22:36 +01003467
3468/*
3469 * Special implementation of mch_stat() for Solaris.
3470 */
3471 int
3472vim_stat(const char *name, stat_T *stp)
3473{
Bram Moolenaar85a20022019-12-21 18:25:54 +01003474 // On Solaris stat() accepts "file/" as if it was "file". Return -1 if
3475 // the name ends in "/" and it's not a directory.
Bram Moolenaard8492792017-03-16 12:22:38 +01003476 return illegal_slash(name) ? -1 : stat(name, stp);
Bram Moolenaar91acfff2017-03-12 19:22:36 +01003477}
Bram Moolenaar071d4272004-06-13 20:20:40 +00003478#endif
3479
3480#if defined(CURSOR_SHAPE) || defined(PROTO)
3481
3482/*
3483 * Handling of cursor and mouse pointer shapes in various modes.
3484 */
3485
3486cursorentry_T shape_table[SHAPE_IDX_COUNT] =
3487{
Bram Moolenaar85a20022019-12-21 18:25:54 +01003488 // The values will be filled in from the 'guicursor' and 'mouseshape'
3489 // defaults when Vim starts.
3490 // Adjust the SHAPE_IDX_ defines when making changes!
Bram Moolenaar071d4272004-06-13 20:20:40 +00003491 {0, 0, 0, 700L, 400L, 250L, 0, 0, "n", SHAPE_CURSOR+SHAPE_MOUSE},
3492 {0, 0, 0, 700L, 400L, 250L, 0, 0, "v", SHAPE_CURSOR+SHAPE_MOUSE},
3493 {0, 0, 0, 700L, 400L, 250L, 0, 0, "i", SHAPE_CURSOR+SHAPE_MOUSE},
3494 {0, 0, 0, 700L, 400L, 250L, 0, 0, "r", SHAPE_CURSOR+SHAPE_MOUSE},
3495 {0, 0, 0, 700L, 400L, 250L, 0, 0, "c", SHAPE_CURSOR+SHAPE_MOUSE},
3496 {0, 0, 0, 700L, 400L, 250L, 0, 0, "ci", SHAPE_CURSOR+SHAPE_MOUSE},
3497 {0, 0, 0, 700L, 400L, 250L, 0, 0, "cr", SHAPE_CURSOR+SHAPE_MOUSE},
3498 {0, 0, 0, 700L, 400L, 250L, 0, 0, "o", SHAPE_CURSOR+SHAPE_MOUSE},
3499 {0, 0, 0, 700L, 400L, 250L, 0, 0, "ve", SHAPE_CURSOR+SHAPE_MOUSE},
3500 {0, 0, 0, 0L, 0L, 0L, 0, 0, "e", SHAPE_MOUSE},
3501 {0, 0, 0, 0L, 0L, 0L, 0, 0, "s", SHAPE_MOUSE},
3502 {0, 0, 0, 0L, 0L, 0L, 0, 0, "sd", SHAPE_MOUSE},
3503 {0, 0, 0, 0L, 0L, 0L, 0, 0, "vs", SHAPE_MOUSE},
3504 {0, 0, 0, 0L, 0L, 0L, 0, 0, "vd", SHAPE_MOUSE},
3505 {0, 0, 0, 0L, 0L, 0L, 0, 0, "m", SHAPE_MOUSE},
3506 {0, 0, 0, 0L, 0L, 0L, 0, 0, "ml", SHAPE_MOUSE},
3507 {0, 0, 0, 100L, 100L, 100L, 0, 0, "sm", SHAPE_CURSOR},
3508};
3509
3510#ifdef FEAT_MOUSESHAPE
3511/*
3512 * Table with names for mouse shapes. Keep in sync with all the tables for
3513 * mch_set_mouse_shape()!.
3514 */
3515static char * mshape_names[] =
3516{
Bram Moolenaar85a20022019-12-21 18:25:54 +01003517 "arrow", // default, must be the first one
3518 "blank", // hidden
Bram Moolenaar071d4272004-06-13 20:20:40 +00003519 "beam",
3520 "updown",
3521 "udsizing",
3522 "leftright",
3523 "lrsizing",
3524 "busy",
3525 "no",
3526 "crosshair",
3527 "hand1",
3528 "hand2",
3529 "pencil",
3530 "question",
3531 "rightup-arrow",
3532 "up-arrow",
3533 NULL
3534};
3535#endif
3536
3537/*
3538 * Parse the 'guicursor' option ("what" is SHAPE_CURSOR) or 'mouseshape'
3539 * ("what" is SHAPE_MOUSE).
3540 * Returns error message for an illegal option, NULL otherwise.
3541 */
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003542 char *
Bram Moolenaar9b578142016-01-30 19:39:49 +01003543parse_shape_opt(int what)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003544{
3545 char_u *modep;
3546 char_u *colonp;
3547 char_u *commap;
3548 char_u *slashp;
3549 char_u *p, *endp;
Bram Moolenaar85a20022019-12-21 18:25:54 +01003550 int idx = 0; // init for GCC
Bram Moolenaar071d4272004-06-13 20:20:40 +00003551 int all_idx;
3552 int len;
3553 int i;
3554 long n;
Bram Moolenaar85a20022019-12-21 18:25:54 +01003555 int found_ve = FALSE; // found "ve" flag
Bram Moolenaar071d4272004-06-13 20:20:40 +00003556 int round;
3557
3558 /*
3559 * First round: check for errors; second round: do it for real.
3560 */
3561 for (round = 1; round <= 2; ++round)
3562 {
3563 /*
3564 * Repeat for all comma separated parts.
3565 */
3566#ifdef FEAT_MOUSESHAPE
3567 if (what == SHAPE_MOUSE)
3568 modep = p_mouseshape;
3569 else
3570#endif
3571 modep = p_guicursor;
3572 while (*modep != NUL)
3573 {
3574 colonp = vim_strchr(modep, ':');
Bram Moolenaar24922ec2017-02-23 17:59:22 +01003575 commap = vim_strchr(modep, ',');
3576
3577 if (colonp == NULL || (commap != NULL && commap < colonp))
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003578 return N_("E545: Missing colon");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003579 if (colonp == modep)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003580 return N_("E546: Illegal mode");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003581
3582 /*
3583 * Repeat for all mode's before the colon.
3584 * For the 'a' mode, we loop to handle all the modes.
3585 */
3586 all_idx = -1;
3587 while (modep < colonp || all_idx >= 0)
3588 {
3589 if (all_idx < 0)
3590 {
Bram Moolenaar85a20022019-12-21 18:25:54 +01003591 // Find the mode.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003592 if (modep[1] == '-' || modep[1] == ':')
3593 len = 1;
3594 else
3595 len = 2;
3596 if (len == 1 && TOLOWER_ASC(modep[0]) == 'a')
3597 all_idx = SHAPE_IDX_COUNT - 1;
3598 else
3599 {
3600 for (idx = 0; idx < SHAPE_IDX_COUNT; ++idx)
3601 if (STRNICMP(modep, shape_table[idx].name, len)
3602 == 0)
3603 break;
3604 if (idx == SHAPE_IDX_COUNT
3605 || (shape_table[idx].used_for & what) == 0)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003606 return N_("E546: Illegal mode");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003607 if (len == 2 && modep[0] == 'v' && modep[1] == 'e')
3608 found_ve = TRUE;
3609 }
3610 modep += len + 1;
3611 }
3612
3613 if (all_idx >= 0)
3614 idx = all_idx--;
3615 else if (round == 2)
3616 {
3617#ifdef FEAT_MOUSESHAPE
3618 if (what == SHAPE_MOUSE)
3619 {
Bram Moolenaar85a20022019-12-21 18:25:54 +01003620 // Set the default, for the missing parts
Bram Moolenaar071d4272004-06-13 20:20:40 +00003621 shape_table[idx].mshape = 0;
3622 }
3623 else
3624#endif
3625 {
Bram Moolenaar85a20022019-12-21 18:25:54 +01003626 // Set the defaults, for the missing parts
Bram Moolenaar071d4272004-06-13 20:20:40 +00003627 shape_table[idx].shape = SHAPE_BLOCK;
3628 shape_table[idx].blinkwait = 700L;
3629 shape_table[idx].blinkon = 400L;
3630 shape_table[idx].blinkoff = 250L;
3631 }
3632 }
3633
Bram Moolenaar85a20022019-12-21 18:25:54 +01003634 // Parse the part after the colon
Bram Moolenaar071d4272004-06-13 20:20:40 +00003635 for (p = colonp + 1; *p && *p != ','; )
3636 {
3637#ifdef FEAT_MOUSESHAPE
3638 if (what == SHAPE_MOUSE)
3639 {
3640 for (i = 0; ; ++i)
3641 {
3642 if (mshape_names[i] == NULL)
3643 {
3644 if (!VIM_ISDIGIT(*p))
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003645 return N_("E547: Illegal mouseshape");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003646 if (round == 2)
3647 shape_table[idx].mshape =
3648 getdigits(&p) + MSHAPE_NUMBERED;
3649 else
3650 (void)getdigits(&p);
3651 break;
3652 }
3653 len = (int)STRLEN(mshape_names[i]);
3654 if (STRNICMP(p, mshape_names[i], len) == 0)
3655 {
3656 if (round == 2)
3657 shape_table[idx].mshape = i;
3658 p += len;
3659 break;
3660 }
3661 }
3662 }
Bram Moolenaar85a20022019-12-21 18:25:54 +01003663 else // if (what == SHAPE_MOUSE)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003664#endif
3665 {
3666 /*
3667 * First handle the ones with a number argument.
3668 */
3669 i = *p;
3670 len = 0;
3671 if (STRNICMP(p, "ver", 3) == 0)
3672 len = 3;
3673 else if (STRNICMP(p, "hor", 3) == 0)
3674 len = 3;
3675 else if (STRNICMP(p, "blinkwait", 9) == 0)
3676 len = 9;
3677 else if (STRNICMP(p, "blinkon", 7) == 0)
3678 len = 7;
3679 else if (STRNICMP(p, "blinkoff", 8) == 0)
3680 len = 8;
3681 if (len != 0)
3682 {
3683 p += len;
3684 if (!VIM_ISDIGIT(*p))
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003685 return N_("E548: digit expected");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003686 n = getdigits(&p);
Bram Moolenaar85a20022019-12-21 18:25:54 +01003687 if (len == 3) // "ver" or "hor"
Bram Moolenaar071d4272004-06-13 20:20:40 +00003688 {
3689 if (n == 0)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003690 return N_("E549: Illegal percentage");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003691 if (round == 2)
3692 {
3693 if (TOLOWER_ASC(i) == 'v')
3694 shape_table[idx].shape = SHAPE_VER;
3695 else
3696 shape_table[idx].shape = SHAPE_HOR;
3697 shape_table[idx].percentage = n;
3698 }
3699 }
3700 else if (round == 2)
3701 {
3702 if (len == 9)
3703 shape_table[idx].blinkwait = n;
3704 else if (len == 7)
3705 shape_table[idx].blinkon = n;
3706 else
3707 shape_table[idx].blinkoff = n;
3708 }
3709 }
3710 else if (STRNICMP(p, "block", 5) == 0)
3711 {
3712 if (round == 2)
3713 shape_table[idx].shape = SHAPE_BLOCK;
3714 p += 5;
3715 }
Bram Moolenaar85a20022019-12-21 18:25:54 +01003716 else // must be a highlight group name then
Bram Moolenaar071d4272004-06-13 20:20:40 +00003717 {
3718 endp = vim_strchr(p, '-');
Bram Moolenaar85a20022019-12-21 18:25:54 +01003719 if (commap == NULL) // last part
Bram Moolenaar071d4272004-06-13 20:20:40 +00003720 {
3721 if (endp == NULL)
Bram Moolenaar85a20022019-12-21 18:25:54 +01003722 endp = p + STRLEN(p); // find end of part
Bram Moolenaar071d4272004-06-13 20:20:40 +00003723 }
3724 else if (endp > commap || endp == NULL)
3725 endp = commap;
3726 slashp = vim_strchr(p, '/');
3727 if (slashp != NULL && slashp < endp)
3728 {
Bram Moolenaar85a20022019-12-21 18:25:54 +01003729 // "group/langmap_group"
Bram Moolenaar071d4272004-06-13 20:20:40 +00003730 i = syn_check_group(p, (int)(slashp - p));
3731 p = slashp + 1;
3732 }
3733 if (round == 2)
3734 {
3735 shape_table[idx].id = syn_check_group(p,
3736 (int)(endp - p));
3737 shape_table[idx].id_lm = shape_table[idx].id;
3738 if (slashp != NULL && slashp < endp)
3739 shape_table[idx].id = i;
3740 }
3741 p = endp;
3742 }
Bram Moolenaar85a20022019-12-21 18:25:54 +01003743 } // if (what != SHAPE_MOUSE)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003744
3745 if (*p == '-')
3746 ++p;
3747 }
3748 }
3749 modep = p;
3750 if (*modep == ',')
3751 ++modep;
3752 }
3753 }
3754
Bram Moolenaar85a20022019-12-21 18:25:54 +01003755 // If the 's' flag is not given, use the 'v' cursor for 's'
Bram Moolenaar071d4272004-06-13 20:20:40 +00003756 if (!found_ve)
3757 {
3758#ifdef FEAT_MOUSESHAPE
3759 if (what == SHAPE_MOUSE)
3760 {
3761 shape_table[SHAPE_IDX_VE].mshape = shape_table[SHAPE_IDX_V].mshape;
3762 }
3763 else
3764#endif
3765 {
3766 shape_table[SHAPE_IDX_VE].shape = shape_table[SHAPE_IDX_V].shape;
3767 shape_table[SHAPE_IDX_VE].percentage =
3768 shape_table[SHAPE_IDX_V].percentage;
3769 shape_table[SHAPE_IDX_VE].blinkwait =
3770 shape_table[SHAPE_IDX_V].blinkwait;
3771 shape_table[SHAPE_IDX_VE].blinkon =
3772 shape_table[SHAPE_IDX_V].blinkon;
3773 shape_table[SHAPE_IDX_VE].blinkoff =
3774 shape_table[SHAPE_IDX_V].blinkoff;
3775 shape_table[SHAPE_IDX_VE].id = shape_table[SHAPE_IDX_V].id;
3776 shape_table[SHAPE_IDX_VE].id_lm = shape_table[SHAPE_IDX_V].id_lm;
3777 }
3778 }
3779
3780 return NULL;
3781}
3782
Bram Moolenaarac6e65f2005-08-29 22:25:38 +00003783# if defined(MCH_CURSOR_SHAPE) || defined(FEAT_GUI) \
3784 || defined(FEAT_MOUSESHAPE) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003785/*
3786 * Return the index into shape_table[] for the current mode.
3787 * When "mouse" is TRUE, consider indexes valid for the mouse pointer.
3788 */
3789 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01003790get_shape_idx(int mouse)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003791{
3792#ifdef FEAT_MOUSESHAPE
3793 if (mouse && (State == HITRETURN || State == ASKMORE))
3794 {
3795# ifdef FEAT_GUI
Bram Moolenaar9588a0f2005-01-08 21:45:39 +00003796 int x, y;
3797 gui_mch_getmouse(&x, &y);
3798 if (Y_2_ROW(y) == Rows - 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003799 return SHAPE_IDX_MOREL;
3800# endif
3801 return SHAPE_IDX_MORE;
3802 }
3803 if (mouse && drag_status_line)
3804 return SHAPE_IDX_SDRAG;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003805 if (mouse && drag_sep_line)
3806 return SHAPE_IDX_VDRAG;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003807#endif
3808 if (!mouse && State == SHOWMATCH)
3809 return SHAPE_IDX_SM;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003810 if (State & VREPLACE_FLAG)
3811 return SHAPE_IDX_R;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003812 if (State & REPLACE_FLAG)
3813 return SHAPE_IDX_R;
3814 if (State & INSERT)
3815 return SHAPE_IDX_I;
3816 if (State & CMDLINE)
3817 {
3818 if (cmdline_at_end())
3819 return SHAPE_IDX_C;
3820 if (cmdline_overstrike())
3821 return SHAPE_IDX_CR;
3822 return SHAPE_IDX_CI;
3823 }
3824 if (finish_op)
3825 return SHAPE_IDX_O;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003826 if (VIsual_active)
3827 {
3828 if (*p_sel == 'e')
3829 return SHAPE_IDX_VE;
3830 else
3831 return SHAPE_IDX_V;
3832 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003833 return SHAPE_IDX_N;
3834}
Bram Moolenaarac6e65f2005-08-29 22:25:38 +00003835#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003836
3837# if defined(FEAT_MOUSESHAPE) || defined(PROTO)
3838static int old_mouse_shape = 0;
3839
3840/*
3841 * Set the mouse shape:
3842 * If "shape" is -1, use shape depending on the current mode,
3843 * depending on the current state.
3844 * If "shape" is -2, only update the shape when it's CLINE or STATUS (used
3845 * when the mouse moves off the status or command line).
3846 */
3847 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01003848update_mouseshape(int shape_idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003849{
3850 int new_mouse_shape;
3851
Bram Moolenaar85a20022019-12-21 18:25:54 +01003852 // Only works in GUI mode.
Bram Moolenaar6bb68362005-03-22 23:03:44 +00003853 if (!gui.in_use || gui.starting)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003854 return;
3855
Bram Moolenaar85a20022019-12-21 18:25:54 +01003856 // Postpone the updating when more is to come. Speeds up executing of
3857 // mappings.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003858 if (shape_idx == -1 && char_avail())
3859 {
3860 postponed_mouseshape = TRUE;
3861 return;
3862 }
3863
Bram Moolenaar85a20022019-12-21 18:25:54 +01003864 // When ignoring the mouse don't change shape on the statusline.
Bram Moolenaar14716812006-05-04 21:54:08 +00003865 if (*p_mouse == NUL
3866 && (shape_idx == SHAPE_IDX_CLINE
3867 || shape_idx == SHAPE_IDX_STATUS
3868 || shape_idx == SHAPE_IDX_VSEP))
3869 shape_idx = -2;
3870
Bram Moolenaar071d4272004-06-13 20:20:40 +00003871 if (shape_idx == -2
3872 && old_mouse_shape != shape_table[SHAPE_IDX_CLINE].mshape
3873 && old_mouse_shape != shape_table[SHAPE_IDX_STATUS].mshape
3874 && old_mouse_shape != shape_table[SHAPE_IDX_VSEP].mshape)
3875 return;
3876 if (shape_idx < 0)
3877 new_mouse_shape = shape_table[get_shape_idx(TRUE)].mshape;
3878 else
3879 new_mouse_shape = shape_table[shape_idx].mshape;
3880 if (new_mouse_shape != old_mouse_shape)
3881 {
3882 mch_set_mouse_shape(new_mouse_shape);
3883 old_mouse_shape = new_mouse_shape;
3884 }
3885 postponed_mouseshape = FALSE;
3886}
3887# endif
3888
Bram Moolenaar85a20022019-12-21 18:25:54 +01003889#endif // CURSOR_SHAPE
Bram Moolenaar071d4272004-06-13 20:20:40 +00003890
3891
Bram Moolenaar071d4272004-06-13 20:20:40 +00003892/*
3893 * Change directory to "new_dir". If FEAT_SEARCHPATH is defined, search
3894 * 'cdpath' for relative directory names, otherwise just mch_chdir().
3895 */
3896 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01003897vim_chdir(char_u *new_dir)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003898{
3899#ifndef FEAT_SEARCHPATH
3900 return mch_chdir((char *)new_dir);
3901#else
3902 char_u *dir_name;
3903 int r;
3904
3905 dir_name = find_directory_in_path(new_dir, (int)STRLEN(new_dir),
3906 FNAME_MESS, curbuf->b_ffname);
3907 if (dir_name == NULL)
3908 return -1;
3909 r = mch_chdir((char *)dir_name);
3910 vim_free(dir_name);
3911 return r;
3912#endif
3913}
3914
3915/*
Bram Moolenaarbbebc852005-07-18 21:47:53 +00003916 * Get user name from machine-specific function.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003917 * Returns the user name in "buf[len]".
Bram Moolenaarbbebc852005-07-18 21:47:53 +00003918 * Some systems are quite slow in obtaining the user name (Windows NT), thus
3919 * cache the result.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003920 * Returns OK or FAIL.
3921 */
3922 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01003923get_user_name(char_u *buf, int len)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003924{
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00003925 if (username == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003926 {
3927 if (mch_get_user_name(buf, len) == FAIL)
3928 return FAIL;
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00003929 username = vim_strsave(buf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003930 }
3931 else
Bram Moolenaarbbebc852005-07-18 21:47:53 +00003932 vim_strncpy(buf, username, len - 1);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003933 return OK;
3934}
3935
3936#ifndef HAVE_QSORT
3937/*
3938 * Our own qsort(), for systems that don't have it.
3939 * It's simple and slow. From the K&R C book.
3940 */
3941 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01003942qsort(
3943 void *base,
3944 size_t elm_count,
3945 size_t elm_size,
3946 int (*cmp)(const void *, const void *))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003947{
3948 char_u *buf;
3949 char_u *p1;
3950 char_u *p2;
3951 int i, j;
3952 int gap;
3953
Bram Moolenaar964b3742019-05-24 18:54:09 +02003954 buf = alloc(elm_size);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003955 if (buf == NULL)
3956 return;
3957
3958 for (gap = elm_count / 2; gap > 0; gap /= 2)
3959 for (i = gap; i < elm_count; ++i)
3960 for (j = i - gap; j >= 0; j -= gap)
3961 {
Bram Moolenaar85a20022019-12-21 18:25:54 +01003962 // Compare the elements.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003963 p1 = (char_u *)base + j * elm_size;
3964 p2 = (char_u *)base + (j + gap) * elm_size;
3965 if ((*cmp)((void *)p1, (void *)p2) <= 0)
3966 break;
Bram Moolenaar85a20022019-12-21 18:25:54 +01003967 // Exchange the elements.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003968 mch_memmove(buf, p1, elm_size);
3969 mch_memmove(p1, p2, elm_size);
3970 mch_memmove(p2, buf, elm_size);
3971 }
3972
3973 vim_free(buf);
3974}
3975#endif
3976
Bram Moolenaar071d4272004-06-13 20:20:40 +00003977/*
3978 * Sort an array of strings.
3979 */
Bram Moolenaareae1b912019-05-09 15:12:55 +02003980static int sort_compare(const void *s1, const void *s2);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003981
3982 static int
Bram Moolenaar9b578142016-01-30 19:39:49 +01003983sort_compare(const void *s1, const void *s2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003984{
3985 return STRCMP(*(char **)s1, *(char **)s2);
3986}
3987
3988 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01003989sort_strings(
3990 char_u **files,
3991 int count)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003992{
3993 qsort((void *)files, (size_t)count, sizeof(char_u *), sort_compare);
3994}
Bram Moolenaar071d4272004-06-13 20:20:40 +00003995
Bram Moolenaar071d4272004-06-13 20:20:40 +00003996/*
3997 * The putenv() implementation below comes from the "screen" program.
3998 * Included with permission from Juergen Weigert.
3999 * See pty.c for the copyright notice.
4000 */
4001
4002/*
4003 * putenv -- put value into environment
4004 *
4005 * Usage: i = putenv (string)
4006 * int i;
4007 * char *string;
4008 *
4009 * where string is of the form <name>=<value>.
4010 * Putenv returns 0 normally, -1 on error (not enough core for malloc).
4011 *
4012 * Putenv may need to add a new name into the environment, or to
4013 * associate a value longer than the current value with a particular
4014 * name. So, to make life simpler, putenv() copies your entire
4015 * environment into the heap (i.e. malloc()) from the stack
4016 * (i.e. where it resides when your process is initiated) the first
4017 * time you call it.
4018 *
4019 * (history removed, not very interesting. See the "screen" sources.)
4020 */
4021
4022#if !defined(HAVE_SETENV) && !defined(HAVE_PUTENV)
4023
Bram Moolenaar85a20022019-12-21 18:25:54 +01004024#define EXTRASIZE 5 // increment to add to env. size
Bram Moolenaar071d4272004-06-13 20:20:40 +00004025
Bram Moolenaar85a20022019-12-21 18:25:54 +01004026static int envsize = -1; // current size of environment
4027extern char **environ; // the global which is your env.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004028
Bram Moolenaar85a20022019-12-21 18:25:54 +01004029static int findenv(char *name); // look for a name in the env.
4030static int newenv(void); // copy env. from stack to heap
4031static int moreenv(void); // incr. size of env.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004032
4033 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01004034putenv(const char *string)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004035{
4036 int i;
4037 char *p;
4038
4039 if (envsize < 0)
Bram Moolenaar85a20022019-12-21 18:25:54 +01004040 { // first time putenv called
4041 if (newenv() < 0) // copy env. to heap
Bram Moolenaar071d4272004-06-13 20:20:40 +00004042 return -1;
4043 }
4044
Bram Moolenaar85a20022019-12-21 18:25:54 +01004045 i = findenv((char *)string); // look for name in environment
Bram Moolenaar071d4272004-06-13 20:20:40 +00004046
4047 if (i < 0)
Bram Moolenaar85a20022019-12-21 18:25:54 +01004048 { // name must be added
Bram Moolenaar071d4272004-06-13 20:20:40 +00004049 for (i = 0; environ[i]; i++);
4050 if (i >= (envsize - 1))
Bram Moolenaar85a20022019-12-21 18:25:54 +01004051 { // need new slot
Bram Moolenaar071d4272004-06-13 20:20:40 +00004052 if (moreenv() < 0)
4053 return -1;
4054 }
Bram Moolenaarc799fe22019-05-28 23:08:19 +02004055 p = alloc(strlen(string) + 1);
Bram Moolenaar85a20022019-12-21 18:25:54 +01004056 if (p == NULL) // not enough core
Bram Moolenaar071d4272004-06-13 20:20:40 +00004057 return -1;
Bram Moolenaar85a20022019-12-21 18:25:54 +01004058 environ[i + 1] = 0; // new end of env.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004059 }
4060 else
Bram Moolenaar85a20022019-12-21 18:25:54 +01004061 { // name already in env.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004062 p = vim_realloc(environ[i], strlen(string) + 1);
4063 if (p == NULL)
4064 return -1;
4065 }
Bram Moolenaar85a20022019-12-21 18:25:54 +01004066 sprintf(p, "%s", string); // copy into env.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004067 environ[i] = p;
4068
4069 return 0;
4070}
4071
4072 static int
Bram Moolenaar9b578142016-01-30 19:39:49 +01004073findenv(char *name)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004074{
4075 char *namechar, *envchar;
4076 int i, found;
4077
4078 found = 0;
4079 for (i = 0; environ[i] && !found; i++)
4080 {
4081 envchar = environ[i];
4082 namechar = name;
4083 while (*namechar && *namechar != '=' && (*namechar == *envchar))
4084 {
4085 namechar++;
4086 envchar++;
4087 }
4088 found = ((*namechar == '\0' || *namechar == '=') && *envchar == '=');
4089 }
4090 return found ? i - 1 : -1;
4091}
4092
4093 static int
Bram Moolenaar9b578142016-01-30 19:39:49 +01004094newenv(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004095{
4096 char **env, *elem;
4097 int i, esize;
4098
Bram Moolenaar071d4272004-06-13 20:20:40 +00004099 for (i = 0; environ[i]; i++)
4100 ;
Bram Moolenaard0573012017-10-28 21:11:06 +02004101
Bram Moolenaar071d4272004-06-13 20:20:40 +00004102 esize = i + EXTRASIZE + 1;
Bram Moolenaarc799fe22019-05-28 23:08:19 +02004103 env = ALLOC_MULT(char *, esize);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004104 if (env == NULL)
4105 return -1;
4106
Bram Moolenaar071d4272004-06-13 20:20:40 +00004107 for (i = 0; environ[i]; i++)
4108 {
Bram Moolenaarc799fe22019-05-28 23:08:19 +02004109 elem = alloc(strlen(environ[i]) + 1);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004110 if (elem == NULL)
4111 return -1;
4112 env[i] = elem;
4113 strcpy(elem, environ[i]);
4114 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004115
4116 env[i] = 0;
4117 environ = env;
4118 envsize = esize;
4119 return 0;
4120}
4121
4122 static int
Bram Moolenaar9b578142016-01-30 19:39:49 +01004123moreenv(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004124{
4125 int esize;
4126 char **env;
4127
4128 esize = envsize + EXTRASIZE;
Bram Moolenaarc799fe22019-05-28 23:08:19 +02004129 env = vim_realloc((char *)environ, esize * sizeof (*env));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004130 if (env == 0)
4131 return -1;
4132 environ = env;
4133 envsize = esize;
4134 return 0;
4135}
4136
4137# ifdef USE_VIMPTY_GETENV
Bram Moolenaar613fe7a2017-07-22 21:11:53 +02004138/*
4139 * Used for mch_getenv() for Mac.
4140 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004141 char_u *
Bram Moolenaar9b578142016-01-30 19:39:49 +01004142vimpty_getenv(const char_u *string)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004143{
4144 int i;
4145 char_u *p;
4146
4147 if (envsize < 0)
4148 return NULL;
4149
4150 i = findenv((char *)string);
4151
4152 if (i < 0)
4153 return NULL;
4154
4155 p = vim_strchr((char_u *)environ[i], '=');
4156 return (p + 1);
4157}
4158# endif
4159
Bram Moolenaar85a20022019-12-21 18:25:54 +01004160#endif // !defined(HAVE_SETENV) && !defined(HAVE_PUTENV)
Bram Moolenaarc4a06d32005-06-07 21:04:49 +00004161
Bram Moolenaarc4956c82006-03-12 21:58:43 +00004162#if defined(FEAT_EVAL) || defined(FEAT_SPELL) || defined(PROTO)
Bram Moolenaarc4a06d32005-06-07 21:04:49 +00004163/*
4164 * Return 0 for not writable, 1 for writable file, 2 for a dir which we have
4165 * rights to write into.
4166 */
4167 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01004168filewritable(char_u *fname)
Bram Moolenaarc4a06d32005-06-07 21:04:49 +00004169{
4170 int retval = 0;
4171#if defined(UNIX) || defined(VMS)
4172 int perm = 0;
4173#endif
4174
4175#if defined(UNIX) || defined(VMS)
4176 perm = mch_getperm(fname);
4177#endif
Bram Moolenaarc4a06d32005-06-07 21:04:49 +00004178 if (
Bram Moolenaar4f974752019-02-17 17:44:42 +01004179# ifdef MSWIN
Bram Moolenaarc4a06d32005-06-07 21:04:49 +00004180 mch_writable(fname) &&
4181# else
4182# if defined(UNIX) || defined(VMS)
4183 (perm & 0222) &&
4184# endif
4185# endif
4186 mch_access((char *)fname, W_OK) == 0
4187 )
Bram Moolenaarc4a06d32005-06-07 21:04:49 +00004188 {
4189 ++retval;
4190 if (mch_isdir(fname))
4191 ++retval;
4192 }
4193 return retval;
4194}
4195#endif
Bram Moolenaar6bab4d12005-06-16 21:53:56 +00004196
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004197#if defined(FEAT_SPELL) || defined(FEAT_PERSISTENT_UNDO) || defined(PROTO)
4198/*
4199 * Read 2 bytes from "fd" and turn them into an int, MSB first.
Bram Moolenaare26e0d22018-03-20 12:34:04 +01004200 * Returns -1 when encountering EOF.
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004201 */
4202 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01004203get2c(FILE *fd)
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004204{
Bram Moolenaare26e0d22018-03-20 12:34:04 +01004205 int c, n;
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004206
4207 n = getc(fd);
Bram Moolenaare26e0d22018-03-20 12:34:04 +01004208 if (n == EOF) return -1;
4209 c = getc(fd);
4210 if (c == EOF) return -1;
4211 return (n << 8) + c;
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004212}
4213
4214/*
4215 * Read 3 bytes from "fd" and turn them into an int, MSB first.
Bram Moolenaare26e0d22018-03-20 12:34:04 +01004216 * Returns -1 when encountering EOF.
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004217 */
4218 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01004219get3c(FILE *fd)
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004220{
Bram Moolenaare26e0d22018-03-20 12:34:04 +01004221 int c, n;
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004222
4223 n = getc(fd);
Bram Moolenaare26e0d22018-03-20 12:34:04 +01004224 if (n == EOF) return -1;
4225 c = getc(fd);
4226 if (c == EOF) return -1;
4227 n = (n << 8) + c;
4228 c = getc(fd);
4229 if (c == EOF) return -1;
4230 return (n << 8) + c;
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004231}
4232
4233/*
4234 * Read 4 bytes from "fd" and turn them into an int, MSB first.
Bram Moolenaare26e0d22018-03-20 12:34:04 +01004235 * Returns -1 when encountering EOF.
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004236 */
4237 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01004238get4c(FILE *fd)
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004239{
Bram Moolenaare26e0d22018-03-20 12:34:04 +01004240 int c;
Bram Moolenaar85a20022019-12-21 18:25:54 +01004241 // Use unsigned rather than int otherwise result is undefined
4242 // when left-shift sets the MSB.
Bram Moolenaar95235e62013-09-08 16:07:07 +02004243 unsigned n;
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004244
Bram Moolenaare26e0d22018-03-20 12:34:04 +01004245 c = getc(fd);
4246 if (c == EOF) return -1;
4247 n = (unsigned)c;
4248 c = getc(fd);
4249 if (c == EOF) return -1;
4250 n = (n << 8) + (unsigned)c;
4251 c = getc(fd);
4252 if (c == EOF) return -1;
4253 n = (n << 8) + (unsigned)c;
4254 c = getc(fd);
4255 if (c == EOF) return -1;
4256 n = (n << 8) + (unsigned)c;
Bram Moolenaar95235e62013-09-08 16:07:07 +02004257 return (int)n;
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004258}
4259
4260/*
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004261 * Read a string of length "cnt" from "fd" into allocated memory.
4262 * Returns NULL when out of memory or unable to read that many bytes.
4263 */
4264 char_u *
Bram Moolenaar9b578142016-01-30 19:39:49 +01004265read_string(FILE *fd, int cnt)
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004266{
4267 char_u *str;
4268 int i;
4269 int c;
4270
Bram Moolenaar85a20022019-12-21 18:25:54 +01004271 // allocate memory
Bram Moolenaar964b3742019-05-24 18:54:09 +02004272 str = alloc(cnt + 1);
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004273 if (str != NULL)
4274 {
Bram Moolenaar85a20022019-12-21 18:25:54 +01004275 // Read the string. Quit when running into the EOF.
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004276 for (i = 0; i < cnt; ++i)
4277 {
4278 c = getc(fd);
4279 if (c == EOF)
4280 {
4281 vim_free(str);
4282 return NULL;
4283 }
4284 str[i] = c;
4285 }
4286 str[i] = NUL;
4287 }
4288 return str;
4289}
4290
4291/*
4292 * Write a number to file "fd", MSB first, in "len" bytes.
4293 */
4294 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01004295put_bytes(FILE *fd, long_u nr, int len)
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004296{
4297 int i;
4298
4299 for (i = len - 1; i >= 0; --i)
4300 if (putc((int)(nr >> (i * 8)), fd) == EOF)
4301 return FAIL;
4302 return OK;
4303}
4304
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004305#endif
Bram Moolenaar10b7b392012-01-10 16:28:45 +01004306
Bram Moolenaarfc3abf42019-01-24 15:54:21 +01004307#if defined(FEAT_QUICKFIX) || defined(FEAT_SPELL) || defined(PROTO)
Bram Moolenaar10b7b392012-01-10 16:28:45 +01004308/*
4309 * Return TRUE if string "s" contains a non-ASCII character (128 or higher).
4310 * When "s" is NULL FALSE is returned.
4311 */
4312 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01004313has_non_ascii(char_u *s)
Bram Moolenaar10b7b392012-01-10 16:28:45 +01004314{
4315 char_u *p;
4316
4317 if (s != NULL)
4318 for (p = s; *p != NUL; ++p)
4319 if (*p >= 128)
4320 return TRUE;
4321 return FALSE;
4322}
4323#endif
Bram Moolenaar93c88e02015-09-15 14:12:05 +02004324
Bram Moolenaar85a20022019-12-21 18:25:54 +01004325#ifndef PROTO // proto is defined in vim.h
Bram Moolenaar51628222016-12-01 23:03:28 +01004326# ifdef ELAPSED_TIMEVAL
Bram Moolenaar833eb1d2016-11-24 17:22:50 +01004327/*
4328 * Return time in msec since "start_tv".
4329 */
4330 long
4331elapsed(struct timeval *start_tv)
4332{
4333 struct timeval now_tv;
4334
4335 gettimeofday(&now_tv, NULL);
4336 return (now_tv.tv_sec - start_tv->tv_sec) * 1000L
4337 + (now_tv.tv_usec - start_tv->tv_usec) / 1000L;
4338}
Bram Moolenaar51628222016-12-01 23:03:28 +01004339# endif
Bram Moolenaar833eb1d2016-11-24 17:22:50 +01004340
Bram Moolenaar51628222016-12-01 23:03:28 +01004341# ifdef ELAPSED_TICKCOUNT
Bram Moolenaar833eb1d2016-11-24 17:22:50 +01004342/*
4343 * Return time in msec since "start_tick".
4344 */
4345 long
4346elapsed(DWORD start_tick)
4347{
4348 DWORD now = GetTickCount();
4349
Bram Moolenaar833eb1d2016-11-24 17:22:50 +01004350 return (long)now - (long)start_tick;
4351}
Bram Moolenaar51628222016-12-01 23:03:28 +01004352# endif
Bram Moolenaar833eb1d2016-11-24 17:22:50 +01004353#endif
Bram Moolenaar20608922018-04-21 22:30:08 +02004354
4355#if defined(FEAT_JOB_CHANNEL) \
4356 || (defined(UNIX) && (!defined(USE_SYSTEM) \
4357 || (defined(FEAT_GUI) && defined(FEAT_TERMINAL)))) \
4358 || defined(PROTO)
4359/*
4360 * Parse "cmd" and put the white-separated parts in "argv".
4361 * "argv" is an allocated array with "argc" entries and room for 4 more.
4362 * Returns FAIL when out of memory.
4363 */
4364 int
4365mch_parse_cmd(char_u *cmd, int use_shcf, char ***argv, int *argc)
4366{
4367 int i;
4368 char_u *p, *d;
4369 int inquote;
4370
4371 /*
4372 * Do this loop twice:
4373 * 1: find number of arguments
4374 * 2: separate them and build argv[]
4375 */
Bram Moolenaar7851b1c2020-03-26 16:27:38 +01004376 for (i = 1; i <= 2; ++i)
Bram Moolenaar20608922018-04-21 22:30:08 +02004377 {
4378 p = skipwhite(cmd);
4379 inquote = FALSE;
4380 *argc = 0;
Bram Moolenaar7851b1c2020-03-26 16:27:38 +01004381 while (*p != NUL)
Bram Moolenaar20608922018-04-21 22:30:08 +02004382 {
Bram Moolenaar7851b1c2020-03-26 16:27:38 +01004383 if (i == 2)
Bram Moolenaar20608922018-04-21 22:30:08 +02004384 (*argv)[*argc] = (char *)p;
4385 ++*argc;
4386 d = p;
4387 while (*p != NUL && (inquote || (*p != ' ' && *p != TAB)))
4388 {
4389 if (p[0] == '"')
Bram Moolenaar1df2fa42018-10-07 21:36:11 +02004390 // quotes surrounding an argument and are dropped
Bram Moolenaar20608922018-04-21 22:30:08 +02004391 inquote = !inquote;
4392 else
4393 {
Bram Moolenaar1df2fa42018-10-07 21:36:11 +02004394 if (rem_backslash(p))
Bram Moolenaar20608922018-04-21 22:30:08 +02004395 {
Bram Moolenaar1df2fa42018-10-07 21:36:11 +02004396 // First pass: skip over "\ " and "\"".
4397 // Second pass: Remove the backslash.
Bram Moolenaar20608922018-04-21 22:30:08 +02004398 ++p;
4399 }
Bram Moolenaar7851b1c2020-03-26 16:27:38 +01004400 if (i == 2)
Bram Moolenaar20608922018-04-21 22:30:08 +02004401 *d++ = *p;
4402 }
4403 ++p;
4404 }
4405 if (*p == NUL)
4406 {
Bram Moolenaar7851b1c2020-03-26 16:27:38 +01004407 if (i == 2)
Bram Moolenaar20608922018-04-21 22:30:08 +02004408 *d++ = NUL;
4409 break;
4410 }
Bram Moolenaar7851b1c2020-03-26 16:27:38 +01004411 if (i == 2)
Bram Moolenaar20608922018-04-21 22:30:08 +02004412 *d++ = NUL;
4413 p = skipwhite(p + 1);
4414 }
4415 if (*argv == NULL)
4416 {
4417 if (use_shcf)
4418 {
Bram Moolenaar85a20022019-12-21 18:25:54 +01004419 // Account for possible multiple args in p_shcf.
Bram Moolenaar20608922018-04-21 22:30:08 +02004420 p = p_shcf;
4421 for (;;)
4422 {
4423 p = skiptowhite(p);
4424 if (*p == NUL)
4425 break;
4426 ++*argc;
4427 p = skipwhite(p);
4428 }
4429 }
4430
Bram Moolenaarc799fe22019-05-28 23:08:19 +02004431 *argv = ALLOC_MULT(char *, *argc + 4);
Bram Moolenaar85a20022019-12-21 18:25:54 +01004432 if (*argv == NULL) // out of memory
Bram Moolenaar20608922018-04-21 22:30:08 +02004433 return FAIL;
4434 }
4435 }
4436 return OK;
4437}
Bram Moolenaarebe74b72018-04-21 23:34:43 +02004438
4439# if defined(FEAT_JOB_CHANNEL) || defined(PROTO)
4440/*
4441 * Build "argv[argc]" from the string "cmd".
4442 * "argv[argc]" is set to NULL;
4443 * Return FAIL when out of memory.
4444 */
4445 int
4446build_argv_from_string(char_u *cmd, char ***argv, int *argc)
4447{
4448 char_u *cmd_copy;
4449 int i;
4450
Bram Moolenaar85a20022019-12-21 18:25:54 +01004451 // Make a copy, parsing will modify "cmd".
Bram Moolenaarebe74b72018-04-21 23:34:43 +02004452 cmd_copy = vim_strsave(cmd);
4453 if (cmd_copy == NULL
4454 || mch_parse_cmd(cmd_copy, FALSE, argv, argc) == FAIL)
4455 {
4456 vim_free(cmd_copy);
4457 return FAIL;
4458 }
4459 for (i = 0; i < *argc; i++)
4460 (*argv)[i] = (char *)vim_strsave((char_u *)(*argv)[i]);
4461 (*argv)[*argc] = NULL;
4462 vim_free(cmd_copy);
4463 return OK;
4464}
4465
4466/*
4467 * Build "argv[argc]" from the list "l".
4468 * "argv[argc]" is set to NULL;
4469 * Return FAIL when out of memory.
4470 */
4471 int
4472build_argv_from_list(list_T *l, char ***argv, int *argc)
4473{
4474 listitem_T *li;
4475 char_u *s;
4476
Bram Moolenaar85a20022019-12-21 18:25:54 +01004477 // Pass argv[] to mch_call_shell().
Bram Moolenaarc799fe22019-05-28 23:08:19 +02004478 *argv = ALLOC_MULT(char *, l->lv_len + 1);
Bram Moolenaarebe74b72018-04-21 23:34:43 +02004479 if (*argv == NULL)
4480 return FAIL;
4481 *argc = 0;
Bram Moolenaaraeea7212020-04-02 18:50:46 +02004482 FOR_ALL_LIST_ITEMS(l, li)
Bram Moolenaarebe74b72018-04-21 23:34:43 +02004483 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004484 s = tv_get_string_chk(&li->li_tv);
Bram Moolenaarebe74b72018-04-21 23:34:43 +02004485 if (s == NULL)
4486 {
4487 int i;
4488
4489 for (i = 0; i < *argc; ++i)
Bram Moolenaardf195602020-04-13 18:13:33 +02004490 VIM_CLEAR((*argv)[i]);
Bram Moolenaarebe74b72018-04-21 23:34:43 +02004491 return FAIL;
4492 }
4493 (*argv)[*argc] = (char *)vim_strsave(s);
4494 *argc += 1;
4495 }
4496 (*argv)[*argc] = NULL;
4497 return OK;
4498}
4499# endif
Bram Moolenaar20608922018-04-21 22:30:08 +02004500#endif
Bram Moolenaar57da6982019-09-13 22:30:11 +02004501
4502/*
4503 * Change the behavior of vterm.
4504 * 0: As usual.
4505 * 1: Windows 10 version 1809
4506 * The bug causes unstable handling of ambiguous width character.
Bram Moolenaar36e7a822019-11-13 21:49:24 +01004507 * 2: Windows 10 version 1903 & 1909
Bram Moolenaar57da6982019-09-13 22:30:11 +02004508 * Use the wrong result because each result is different.
4509 * 3: Windows 10 insider preview (current latest logic)
4510 */
4511 int
4512get_special_pty_type(void)
4513{
4514#ifdef MSWIN
4515 return get_conpty_type();
4516#else
4517 return 0;
4518#endif
4519}