blob: 000f043e138dc74c672afba041f7487abfde6740 [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 Moolenaarf461c8e2005-06-25 23:04:51 +000015static char_u *username = NULL; /* cached result of mch_get_user_name() */
16
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{
25 /* 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. */
28 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 {
61 /* Virtcol is valid */
62 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 Moolenaardfccaf02004-12-31 20:56:11 +0000102 /* 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,
122 int addspaces, /* change the text to achieve our goal? */
123 int finetune, /* change char offset for the exact column */
124 colnr_T wcol) /* column to move to */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000125{
Bram Moolenaar071d4272004-06-13 20:20:40 +0000126 int idx;
127 char_u *ptr;
128 char_u *line;
129 colnr_T col = 0;
130 int csize = 0;
131 int one_more;
132#ifdef FEAT_LINEBREAK
133 int head = 0;
134#endif
135
Bram Moolenaarefd2bf12006-03-16 21:41:35 +0000136 one_more = (State & INSERT)
137 || restart_edit != NUL
Bram Moolenaarefd2bf12006-03-16 21:41:35 +0000138 || (VIsual_active && *p_sel != 'o')
Bram Moolenaar29ddebe2019-01-26 17:28:26 +0100139 || ((ve_flags & VE_ONEMORE) && wcol < MAXCOL) ;
Bram Moolenaara1381de2009-11-03 15:44:21 +0000140 line = ml_get_buf(curbuf, pos->lnum, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000141
142 if (wcol >= MAXCOL)
143 {
144 idx = (int)STRLEN(line) - 1 + one_more;
145 col = wcol;
146
Bram Moolenaar071d4272004-06-13 20:20:40 +0000147 if ((addspaces || finetune) && !VIsual_active)
148 {
149 curwin->w_curswant = linetabsize(line) + one_more;
150 if (curwin->w_curswant > 0)
151 --curwin->w_curswant;
152 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000153 }
154 else
155 {
Bram Moolenaar02631462017-09-22 15:20:32 +0200156 int width = curwin->w_width - win_col_off(curwin);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000157
Bram Moolenaarebefac62005-12-28 22:39:57 +0000158 if (finetune
Bram Moolenaar071d4272004-06-13 20:20:40 +0000159 && curwin->w_p_wrap
Bram Moolenaar071d4272004-06-13 20:20:40 +0000160 && curwin->w_width != 0
Bram Moolenaar071d4272004-06-13 20:20:40 +0000161 && wcol >= (colnr_T)width)
162 {
163 csize = linetabsize(line);
164 if (csize > 0)
165 csize--;
166
Bram Moolenaarebefac62005-12-28 22:39:57 +0000167 if (wcol / width > (colnr_T)csize / width
168 && ((State & INSERT) == 0 || (int)wcol > csize + 1))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000169 {
170 /* In case of line wrapping don't move the cursor beyond the
Bram Moolenaarebefac62005-12-28 22:39:57 +0000171 * right screen edge. In Insert mode allow going just beyond
172 * the last character (like what happens when typing and
173 * reaching the right window edge). */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000174 wcol = (csize / width + 1) * width - 1;
175 }
176 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000177
Bram Moolenaar071d4272004-06-13 20:20:40 +0000178 ptr = line;
179 while (col <= wcol && *ptr != NUL)
180 {
181 /* Count a tab for what it's worth (if list mode not on) */
182#ifdef FEAT_LINEBREAK
Bram Moolenaar597a4222014-06-25 14:39:50 +0200183 csize = win_lbr_chartabsize(curwin, line, ptr, col, &head);
Bram Moolenaar91acfff2017-03-12 19:22:36 +0100184 MB_PTR_ADV(ptr);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000185#else
Bram Moolenaar597a4222014-06-25 14:39:50 +0200186 csize = lbr_chartabsize_adv(line, &ptr, col);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000187#endif
188 col += csize;
189 }
190 idx = (int)(ptr - line);
191 /*
192 * Handle all the special cases. The virtual_active() check
193 * is needed to ensure that a virtual position off the end of
194 * a line has the correct indexing. The one_more comparison
195 * replaces an explicit add of one_more later on.
196 */
197 if (col > wcol || (!virtual_active() && one_more == 0))
198 {
199 idx -= 1;
200# ifdef FEAT_LINEBREAK
201 /* Don't count the chars from 'showbreak'. */
202 csize -= head;
203# endif
204 col -= csize;
205 }
206
Bram Moolenaar071d4272004-06-13 20:20:40 +0000207 if (virtual_active()
208 && addspaces
209 && ((col != wcol && col != wcol + 1) || csize > 1))
210 {
211 /* 'virtualedit' is set: The difference between wcol and col is
212 * filled with spaces. */
213
214 if (line[idx] == NUL)
215 {
216 /* Append spaces */
217 int correct = wcol - col;
218 char_u *newline = alloc(idx + correct + 1);
219 int t;
220
221 if (newline == NULL)
222 return FAIL;
223
224 for (t = 0; t < idx; ++t)
225 newline[t] = line[t];
226
227 for (t = 0; t < correct; ++t)
228 newline[t + idx] = ' ';
229
230 newline[idx + correct] = NUL;
231
232 ml_replace(pos->lnum, newline, FALSE);
233 changed_bytes(pos->lnum, (colnr_T)idx);
234 idx += correct;
235 col = wcol;
236 }
237 else
238 {
239 /* Break a tab */
240 int linelen = (int)STRLEN(line);
241 int correct = wcol - col - csize + 1; /* negative!! */
Bram Moolenaareb3593b2006-04-22 22:33:57 +0000242 char_u *newline;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000243 int t, s = 0;
244 int v;
245
Bram Moolenaareb3593b2006-04-22 22:33:57 +0000246 if (-correct > csize)
247 return FAIL;
248
249 newline = alloc(linelen + csize);
250 if (newline == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000251 return FAIL;
252
253 for (t = 0; t < linelen; t++)
254 {
255 if (t != idx)
256 newline[s++] = line[t];
257 else
258 for (v = 0; v < csize; v++)
259 newline[s++] = ' ';
260 }
261
262 newline[linelen + csize - 1] = NUL;
263
264 ml_replace(pos->lnum, newline, FALSE);
265 changed_bytes(pos->lnum, idx);
266 idx += (csize - 1 + correct);
267 col += correct;
268 }
269 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000270 }
271
272 if (idx < 0)
273 pos->col = 0;
274 else
275 pos->col = idx;
276
Bram Moolenaar071d4272004-06-13 20:20:40 +0000277 pos->coladd = 0;
278
279 if (finetune)
280 {
281 if (wcol == MAXCOL)
282 {
283 /* The width of the last character is used to set coladd. */
284 if (!one_more)
285 {
286 colnr_T scol, ecol;
287
288 getvcol(curwin, pos, &scol, NULL, &ecol);
289 pos->coladd = ecol - scol;
290 }
291 }
292 else
293 {
294 int b = (int)wcol - (int)col;
295
296 /* The difference between wcol and col is used to set coladd. */
Bram Moolenaar02631462017-09-22 15:20:32 +0200297 if (b > 0 && b < (MAXCOL - 2 * curwin->w_width))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000298 pos->coladd = b;
299
300 col += b;
301 }
302 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000303
Bram Moolenaara1381de2009-11-03 15:44:21 +0000304 /* prevent from moving onto a trail byte */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000305 if (has_mbyte)
Bram Moolenaar03a807a2011-07-07 15:08:58 +0200306 mb_adjustpos(curbuf, pos);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000307
308 if (col < wcol)
309 return FAIL;
310 return OK;
311}
312
313/*
Bram Moolenaar446cb832008-06-24 21:56:24 +0000314 * Increment the cursor position. See inc() for return values.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000315 */
316 int
Bram Moolenaar9b578142016-01-30 19:39:49 +0100317inc_cursor(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000318{
319 return inc(&curwin->w_cursor);
320}
321
Bram Moolenaar446cb832008-06-24 21:56:24 +0000322/*
323 * Increment the line pointer "lp" crossing line boundaries as necessary.
324 * Return 1 when going to the next line.
325 * Return 2 when moving forward onto a NUL at the end of the line).
326 * Return -1 when at the end of file.
327 * Return 0 otherwise.
328 */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000329 int
Bram Moolenaar9b578142016-01-30 19:39:49 +0100330inc(pos_T *lp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000331{
Bram Moolenaar8ada6aa2017-12-19 21:23:21 +0100332 char_u *p;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000333
Bram Moolenaar8ada6aa2017-12-19 21:23:21 +0100334 /* when searching position may be set to end of a line */
335 if (lp->col != MAXCOL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000336 {
Bram Moolenaar8ada6aa2017-12-19 21:23:21 +0100337 p = ml_get_pos(lp);
338 if (*p != NUL) /* still within line, move to next char (may be NUL) */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000339 {
Bram Moolenaar8ada6aa2017-12-19 21:23:21 +0100340 if (has_mbyte)
341 {
342 int l = (*mb_ptr2len)(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000343
Bram Moolenaar8ada6aa2017-12-19 21:23:21 +0100344 lp->col += l;
345 return ((p[l] != NUL) ? 0 : 2);
346 }
Bram Moolenaar8ada6aa2017-12-19 21:23:21 +0100347 lp->col++;
Bram Moolenaar8ada6aa2017-12-19 21:23:21 +0100348 lp->coladd = 0;
Bram Moolenaar8ada6aa2017-12-19 21:23:21 +0100349 return ((p[1] != NUL) ? 0 : 2);
350 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000351 }
352 if (lp->lnum != curbuf->b_ml.ml_line_count) /* there is a next line */
353 {
354 lp->col = 0;
355 lp->lnum++;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000356 lp->coladd = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000357 return 1;
358 }
359 return -1;
360}
361
362/*
363 * incl(lp): same as inc(), but skip the NUL at the end of non-empty lines
364 */
365 int
Bram Moolenaar9b578142016-01-30 19:39:49 +0100366incl(pos_T *lp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000367{
368 int r;
369
370 if ((r = inc(lp)) >= 1 && lp->col)
371 r = inc(lp);
372 return r;
373}
374
375/*
376 * dec(p)
377 *
378 * Decrement the line pointer 'p' crossing line boundaries as necessary.
379 * Return 1 when crossing a line, -1 when at start of file, 0 otherwise.
380 */
381 int
Bram Moolenaar9b578142016-01-30 19:39:49 +0100382dec_cursor(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000383{
Bram Moolenaarcaa55b62017-01-10 13:51:09 +0100384 return dec(&curwin->w_cursor);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000385}
386
387 int
Bram Moolenaar9b578142016-01-30 19:39:49 +0100388dec(pos_T *lp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000389{
390 char_u *p;
391
Bram Moolenaar071d4272004-06-13 20:20:40 +0000392 lp->coladd = 0;
Bram Moolenaar1bd999f2017-12-19 22:25:40 +0100393 if (lp->col == MAXCOL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000394 {
Bram Moolenaar1bd999f2017-12-19 22:25:40 +0100395 /* past end of line */
396 p = ml_get(lp->lnum);
397 lp->col = (colnr_T)STRLEN(p);
Bram Moolenaar1bd999f2017-12-19 22:25:40 +0100398 if (has_mbyte)
399 lp->col -= (*mb_head_off)(p, p + lp->col);
Bram Moolenaar1bd999f2017-12-19 22:25:40 +0100400 return 0;
401 }
402
403 if (lp->col > 0)
404 {
405 /* still within line */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000406 lp->col--;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000407 if (has_mbyte)
408 {
409 p = ml_get(lp->lnum);
410 lp->col -= (*mb_head_off)(p, p + lp->col);
411 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000412 return 0;
413 }
Bram Moolenaar1bd999f2017-12-19 22:25:40 +0100414
415 if (lp->lnum > 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000416 {
Bram Moolenaar1bd999f2017-12-19 22:25:40 +0100417 /* there is a prior line */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000418 lp->lnum--;
419 p = ml_get(lp->lnum);
420 lp->col = (colnr_T)STRLEN(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000421 if (has_mbyte)
422 lp->col -= (*mb_head_off)(p, p + lp->col);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000423 return 1;
424 }
Bram Moolenaar1bd999f2017-12-19 22:25:40 +0100425
426 /* at start of file */
427 return -1;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000428}
429
430/*
431 * decl(lp): same as dec(), but skip the NUL at the end of non-empty lines
432 */
433 int
Bram Moolenaar9b578142016-01-30 19:39:49 +0100434decl(pos_T *lp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000435{
436 int r;
437
438 if ((r = dec(lp)) == 1 && lp->col)
439 r = dec(lp);
440 return r;
441}
442
443/*
Bram Moolenaar64486672010-05-16 15:46:46 +0200444 * Get the line number relative to the current cursor position, i.e. the
445 * difference between line number and cursor position. Only look for lines that
446 * can be visible, folded lines don't count.
447 */
448 linenr_T
Bram Moolenaar9b578142016-01-30 19:39:49 +0100449get_cursor_rel_lnum(
450 win_T *wp,
451 linenr_T lnum) /* line number to get the result for */
Bram Moolenaar64486672010-05-16 15:46:46 +0200452{
453 linenr_T cursor = wp->w_cursor.lnum;
454 linenr_T retval = 0;
455
456#ifdef FEAT_FOLDING
457 if (hasAnyFolding(wp))
458 {
459 if (lnum > cursor)
460 {
461 while (lnum > cursor)
462 {
Bram Moolenaar0bd7b3f2013-12-14 12:48:58 +0100463 (void)hasFoldingWin(wp, lnum, &lnum, NULL, TRUE, NULL);
Bram Moolenaar64486672010-05-16 15:46:46 +0200464 /* if lnum and cursor are in the same fold,
465 * now lnum <= cursor */
466 if (lnum > cursor)
467 retval++;
468 lnum--;
469 }
470 }
471 else if (lnum < cursor)
472 {
473 while (lnum < cursor)
474 {
Bram Moolenaar0bd7b3f2013-12-14 12:48:58 +0100475 (void)hasFoldingWin(wp, lnum, NULL, &lnum, TRUE, NULL);
Bram Moolenaar64486672010-05-16 15:46:46 +0200476 /* if lnum and cursor are in the same fold,
477 * now lnum >= cursor */
478 if (lnum < cursor)
479 retval--;
480 lnum++;
481 }
482 }
483 /* else if (lnum == cursor)
484 * retval = 0;
485 */
486 }
487 else
488#endif
489 retval = lnum - cursor;
490
491 return retval;
492}
493
494/*
Bram Moolenaard5824ce2016-09-04 20:35:01 +0200495 * Make sure "pos.lnum" and "pos.col" are valid in "buf".
496 * This allows for the col to be on the NUL byte.
497 */
498 void
499check_pos(buf_T *buf, pos_T *pos)
500{
501 char_u *line;
502 colnr_T len;
503
504 if (pos->lnum > buf->b_ml.ml_line_count)
505 pos->lnum = buf->b_ml.ml_line_count;
506
507 if (pos->col > 0)
508 {
509 line = ml_get_buf(buf, pos->lnum, FALSE);
510 len = (colnr_T)STRLEN(line);
511 if (pos->col > len)
512 pos->col = len;
513 }
514}
515
516/*
Bram Moolenaar071d4272004-06-13 20:20:40 +0000517 * Make sure curwin->w_cursor.lnum is valid.
518 */
519 void
Bram Moolenaar9b578142016-01-30 19:39:49 +0100520check_cursor_lnum(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000521{
522 if (curwin->w_cursor.lnum > curbuf->b_ml.ml_line_count)
523 {
524#ifdef FEAT_FOLDING
525 /* If there is a closed fold at the end of the file, put the cursor in
526 * its first line. Otherwise in the last line. */
527 if (!hasFolding(curbuf->b_ml.ml_line_count,
528 &curwin->w_cursor.lnum, NULL))
529#endif
530 curwin->w_cursor.lnum = curbuf->b_ml.ml_line_count;
531 }
532 if (curwin->w_cursor.lnum <= 0)
533 curwin->w_cursor.lnum = 1;
534}
535
536/*
537 * Make sure curwin->w_cursor.col is valid.
538 */
539 void
Bram Moolenaar9b578142016-01-30 19:39:49 +0100540check_cursor_col(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000541{
Bram Moolenaar03a807a2011-07-07 15:08:58 +0200542 check_cursor_col_win(curwin);
543}
544
545/*
546 * Make sure win->w_cursor.col is valid.
547 */
548 void
Bram Moolenaar9b578142016-01-30 19:39:49 +0100549check_cursor_col_win(win_T *win)
Bram Moolenaar03a807a2011-07-07 15:08:58 +0200550{
Bram Moolenaar071d4272004-06-13 20:20:40 +0000551 colnr_T len;
Bram Moolenaar03a807a2011-07-07 15:08:58 +0200552 colnr_T oldcol = win->w_cursor.col;
553 colnr_T oldcoladd = win->w_cursor.col + win->w_cursor.coladd;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000554
Bram Moolenaar03a807a2011-07-07 15:08:58 +0200555 len = (colnr_T)STRLEN(ml_get_buf(win->w_buffer, win->w_cursor.lnum, FALSE));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000556 if (len == 0)
Bram Moolenaar03a807a2011-07-07 15:08:58 +0200557 win->w_cursor.col = 0;
558 else if (win->w_cursor.col >= len)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000559 {
Bram Moolenaar33f54432008-01-04 20:25:44 +0000560 /* Allow cursor past end-of-line when:
561 * - in Insert mode or restarting Insert mode
562 * - in Visual mode and 'selection' isn't "old"
563 * - 'virtualedit' is set */
Bram Moolenaarebefac62005-12-28 22:39:57 +0000564 if ((State & INSERT) || restart_edit
Bram Moolenaar071d4272004-06-13 20:20:40 +0000565 || (VIsual_active && *p_sel != 'o')
Bram Moolenaar33f54432008-01-04 20:25:44 +0000566 || (ve_flags & VE_ONEMORE)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000567 || virtual_active())
Bram Moolenaar03a807a2011-07-07 15:08:58 +0200568 win->w_cursor.col = len;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000569 else
Bram Moolenaar87c19962007-04-26 08:54:21 +0000570 {
Bram Moolenaar03a807a2011-07-07 15:08:58 +0200571 win->w_cursor.col = len - 1;
Bram Moolenaar03a807a2011-07-07 15:08:58 +0200572 /* Move the cursor to the head byte. */
Bram Moolenaar87c19962007-04-26 08:54:21 +0000573 if (has_mbyte)
Bram Moolenaar03a807a2011-07-07 15:08:58 +0200574 mb_adjustpos(win->w_buffer, &win->w_cursor);
Bram Moolenaar87c19962007-04-26 08:54:21 +0000575 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000576 }
Bram Moolenaar03a807a2011-07-07 15:08:58 +0200577 else if (win->w_cursor.col < 0)
578 win->w_cursor.col = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000579
Bram Moolenaar071d4272004-06-13 20:20:40 +0000580 /* If virtual editing is on, we can leave the cursor on the old position,
581 * only we must set it to virtual. But don't do it when at the end of the
582 * line. */
583 if (oldcol == MAXCOL)
Bram Moolenaar03a807a2011-07-07 15:08:58 +0200584 win->w_cursor.coladd = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000585 else if (ve_flags == VE_ALL)
Bram Moolenaar552c8a52009-03-11 16:29:20 +0000586 {
Bram Moolenaar03a807a2011-07-07 15:08:58 +0200587 if (oldcoladd > win->w_cursor.col)
Bram Moolenaar9aa15692017-08-19 15:05:32 +0200588 {
Bram Moolenaar03a807a2011-07-07 15:08:58 +0200589 win->w_cursor.coladd = oldcoladd - win->w_cursor.col;
Bram Moolenaard41babe2017-08-30 17:01:35 +0200590
591 /* Make sure that coladd is not more than the char width.
592 * Not for the last character, coladd is then used when the cursor
593 * is actually after the last character. */
594 if (win->w_cursor.col + 1 < len && win->w_cursor.coladd > 0)
Bram Moolenaar9aa15692017-08-19 15:05:32 +0200595 {
596 int cs, ce;
597
Bram Moolenaar9aa15692017-08-19 15:05:32 +0200598 getvcol(win, &win->w_cursor, &cs, NULL, &ce);
599 if (win->w_cursor.coladd > ce - cs)
600 win->w_cursor.coladd = ce - cs;
601 }
602 }
Bram Moolenaar552c8a52009-03-11 16:29:20 +0000603 else
604 /* avoid weird number when there is a miscalculation or overflow */
Bram Moolenaar03a807a2011-07-07 15:08:58 +0200605 win->w_cursor.coladd = 0;
Bram Moolenaar552c8a52009-03-11 16:29:20 +0000606 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000607}
608
609/*
610 * make sure curwin->w_cursor in on a valid character
611 */
612 void
Bram Moolenaar9b578142016-01-30 19:39:49 +0100613check_cursor(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000614{
615 check_cursor_lnum();
616 check_cursor_col();
617}
618
619#if defined(FEAT_TEXTOBJ) || defined(PROTO)
620/*
621 * Make sure curwin->w_cursor is not on the NUL at the end of the line.
622 * Allow it when in Visual mode and 'selection' is not "old".
623 */
624 void
Bram Moolenaar9b578142016-01-30 19:39:49 +0100625adjust_cursor_col(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000626{
627 if (curwin->w_cursor.col > 0
Bram Moolenaar071d4272004-06-13 20:20:40 +0000628 && (!VIsual_active || *p_sel == 'o')
Bram Moolenaar071d4272004-06-13 20:20:40 +0000629 && gchar_cursor() == NUL)
630 --curwin->w_cursor.col;
631}
632#endif
633
634/*
635 * When curwin->w_leftcol has changed, adjust the cursor position.
636 * Return TRUE if the cursor was moved.
637 */
638 int
Bram Moolenaar9b578142016-01-30 19:39:49 +0100639leftcol_changed(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000640{
641 long lastcol;
642 colnr_T s, e;
643 int retval = FALSE;
Bram Moolenaar375e3392019-01-31 18:26:10 +0100644 long siso = get_sidescrolloff_value();
Bram Moolenaar071d4272004-06-13 20:20:40 +0000645
646 changed_cline_bef_curs();
Bram Moolenaar02631462017-09-22 15:20:32 +0200647 lastcol = curwin->w_leftcol + curwin->w_width - curwin_col_off() - 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000648 validate_virtcol();
649
650 /*
651 * If the cursor is right or left of the screen, move it to last or first
652 * character.
653 */
Bram Moolenaar375e3392019-01-31 18:26:10 +0100654 if (curwin->w_virtcol > (colnr_T)(lastcol - siso))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000655 {
656 retval = TRUE;
Bram Moolenaar375e3392019-01-31 18:26:10 +0100657 coladvance((colnr_T)(lastcol - siso));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000658 }
Bram Moolenaar375e3392019-01-31 18:26:10 +0100659 else if (curwin->w_virtcol < curwin->w_leftcol + siso)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000660 {
661 retval = TRUE;
Bram Moolenaar375e3392019-01-31 18:26:10 +0100662 (void)coladvance((colnr_T)(curwin->w_leftcol + siso));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000663 }
664
665 /*
666 * If the start of the character under the cursor is not on the screen,
667 * advance the cursor one more char. If this fails (last char of the
668 * line) adjust the scrolling.
669 */
670 getvvcol(curwin, &curwin->w_cursor, &s, NULL, &e);
671 if (e > (colnr_T)lastcol)
672 {
673 retval = TRUE;
674 coladvance(s - 1);
675 }
676 else if (s < curwin->w_leftcol)
677 {
678 retval = TRUE;
679 if (coladvance(e + 1) == FAIL) /* there isn't another character */
680 {
681 curwin->w_leftcol = s; /* adjust w_leftcol instead */
682 changed_cline_bef_curs();
683 }
684 }
685
686 if (retval)
687 curwin->w_set_curswant = TRUE;
688 redraw_later(NOT_VALID);
689 return retval;
690}
691
692/**********************************************************************
693 * Various routines dealing with allocation and deallocation of memory.
694 */
695
696#if defined(MEM_PROFILE) || defined(PROTO)
697
698# define MEM_SIZES 8200
699static long_u mem_allocs[MEM_SIZES];
700static long_u mem_frees[MEM_SIZES];
701static long_u mem_allocated;
702static long_u mem_freed;
703static long_u mem_peak;
704static long_u num_alloc;
705static long_u num_freed;
706
Bram Moolenaar071d4272004-06-13 20:20:40 +0000707 static void
Bram Moolenaar9b578142016-01-30 19:39:49 +0100708mem_pre_alloc_s(size_t *sizep)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000709{
710 *sizep += sizeof(size_t);
711}
712
713 static void
Bram Moolenaar964b3742019-05-24 18:54:09 +0200714mem_pre_alloc_l(size_t *sizep)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000715{
716 *sizep += sizeof(size_t);
717}
718
719 static void
Bram Moolenaar9b578142016-01-30 19:39:49 +0100720mem_post_alloc(
721 void **pp,
722 size_t size)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000723{
724 if (*pp == NULL)
725 return;
726 size -= sizeof(size_t);
727 *(long_u *)*pp = size;
728 if (size <= MEM_SIZES-1)
729 mem_allocs[size-1]++;
730 else
731 mem_allocs[MEM_SIZES-1]++;
732 mem_allocated += size;
733 if (mem_allocated - mem_freed > mem_peak)
734 mem_peak = mem_allocated - mem_freed;
735 num_alloc++;
736 *pp = (void *)((char *)*pp + sizeof(size_t));
737}
738
739 static void
Bram Moolenaar9b578142016-01-30 19:39:49 +0100740mem_pre_free(void **pp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000741{
742 long_u size;
743
744 *pp = (void *)((char *)*pp - sizeof(size_t));
745 size = *(size_t *)*pp;
746 if (size <= MEM_SIZES-1)
747 mem_frees[size-1]++;
748 else
749 mem_frees[MEM_SIZES-1]++;
750 mem_freed += size;
751 num_freed++;
752}
753
754/*
755 * called on exit via atexit()
756 */
757 void
Bram Moolenaar9b578142016-01-30 19:39:49 +0100758vim_mem_profile_dump(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000759{
760 int i, j;
761
762 printf("\r\n");
763 j = 0;
764 for (i = 0; i < MEM_SIZES - 1; i++)
765 {
766 if (mem_allocs[i] || mem_frees[i])
767 {
768 if (mem_frees[i] > mem_allocs[i])
769 printf("\r\n%s", _("ERROR: "));
770 printf("[%4d / %4lu-%-4lu] ", i + 1, mem_allocs[i], mem_frees[i]);
771 j++;
772 if (j > 3)
773 {
774 j = 0;
775 printf("\r\n");
776 }
777 }
778 }
779
780 i = MEM_SIZES - 1;
781 if (mem_allocs[i])
782 {
783 printf("\r\n");
784 if (mem_frees[i] > mem_allocs[i])
Bram Moolenaar2f1e0502010-08-13 11:18:02 +0200785 puts(_("ERROR: "));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000786 printf("[>%d / %4lu-%-4lu]", i, mem_allocs[i], mem_frees[i]);
787 }
788
789 printf(_("\n[bytes] total alloc-freed %lu-%lu, in use %lu, peak use %lu\n"),
790 mem_allocated, mem_freed, mem_allocated - mem_freed, mem_peak);
791 printf(_("[calls] total re/malloc()'s %lu, total free()'s %lu\n\n"),
792 num_alloc, num_freed);
793}
794
795#endif /* MEM_PROFILE */
796
Bram Moolenaar75bdf6a2016-01-07 21:25:08 +0100797#ifdef FEAT_EVAL
Bram Moolenaarf49cc602018-11-11 15:21:05 +0100798 int
Bram Moolenaar964b3742019-05-24 18:54:09 +0200799alloc_does_fail(size_t size)
Bram Moolenaar75bdf6a2016-01-07 21:25:08 +0100800{
801 if (alloc_fail_countdown == 0)
802 {
803 if (--alloc_fail_repeat <= 0)
804 alloc_fail_id = 0;
Bram Moolenaara260b872016-01-15 20:48:22 +0100805 do_outofmem_msg(size);
Bram Moolenaar75bdf6a2016-01-07 21:25:08 +0100806 return TRUE;
807 }
808 --alloc_fail_countdown;
809 return FALSE;
810}
811#endif
812
Bram Moolenaar071d4272004-06-13 20:20:40 +0000813/*
814 * Some memory is reserved for error messages and for being able to
815 * call mf_release_all(), which needs some memory for mf_trans_add().
816 */
Bram Moolenaar48e330a2016-02-23 14:53:34 +0100817#define KEEP_ROOM (2 * 8192L)
Bram Moolenaar11b73d62012-06-29 15:51:30 +0200818#define KEEP_ROOM_KB (KEEP_ROOM / 1024L)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000819
820/*
Bram Moolenaar964b3742019-05-24 18:54:09 +0200821 * The normal way to allocate memory. This handles an out-of-memory situation
822 * as well as possible, still returns NULL when we're completely out.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000823 */
Bram Moolenaarc799fe22019-05-28 23:08:19 +0200824 void *
Bram Moolenaar964b3742019-05-24 18:54:09 +0200825alloc(size_t size)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000826{
Bram Moolenaar964b3742019-05-24 18:54:09 +0200827 return lalloc(size, TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000828}
829
830/*
Bram Moolenaar75bdf6a2016-01-07 21:25:08 +0100831 * alloc() with an ID for alloc_fail().
Bram Moolenaar75bdf6a2016-01-07 21:25:08 +0100832 */
Bram Moolenaarc799fe22019-05-28 23:08:19 +0200833 void *
Bram Moolenaar964b3742019-05-24 18:54:09 +0200834alloc_id(size_t size, alloc_id_T id UNUSED)
Bram Moolenaar75bdf6a2016-01-07 21:25:08 +0100835{
836#ifdef FEAT_EVAL
Bram Moolenaar964b3742019-05-24 18:54:09 +0200837 if (alloc_fail_id == id && alloc_does_fail(size))
Bram Moolenaar75bdf6a2016-01-07 21:25:08 +0100838 return NULL;
839#endif
Bram Moolenaar964b3742019-05-24 18:54:09 +0200840 return lalloc(size, TRUE);
Bram Moolenaar75bdf6a2016-01-07 21:25:08 +0100841}
842
843/*
Bram Moolenaar071d4272004-06-13 20:20:40 +0000844 * Allocate memory and set all bytes to zero.
845 */
Bram Moolenaarc799fe22019-05-28 23:08:19 +0200846 void *
Bram Moolenaar964b3742019-05-24 18:54:09 +0200847alloc_clear(size_t size)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000848{
Bram Moolenaarc799fe22019-05-28 23:08:19 +0200849 void *p;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000850
Bram Moolenaar964b3742019-05-24 18:54:09 +0200851 p = lalloc(size, TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000852 if (p != NULL)
Bram Moolenaar964b3742019-05-24 18:54:09 +0200853 (void)vim_memset(p, 0, size);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000854 return p;
855}
856
857/*
Bram Moolenaar162b7142018-12-21 15:17:36 +0100858 * Same as alloc_clear() but with allocation id for testing
859 */
Bram Moolenaarc799fe22019-05-28 23:08:19 +0200860 void *
Bram Moolenaar964b3742019-05-24 18:54:09 +0200861alloc_clear_id(size_t size, alloc_id_T id UNUSED)
Bram Moolenaar162b7142018-12-21 15:17:36 +0100862{
863#ifdef FEAT_EVAL
Bram Moolenaar964b3742019-05-24 18:54:09 +0200864 if (alloc_fail_id == id && alloc_does_fail(size))
Bram Moolenaar162b7142018-12-21 15:17:36 +0100865 return NULL;
866#endif
867 return alloc_clear(size);
868}
869
870/*
Bram Moolenaar071d4272004-06-13 20:20:40 +0000871 * Allocate memory like lalloc() and set all bytes to zero.
872 */
Bram Moolenaarc799fe22019-05-28 23:08:19 +0200873 void *
Bram Moolenaar964b3742019-05-24 18:54:09 +0200874lalloc_clear(size_t size, int message)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000875{
Bram Moolenaarc799fe22019-05-28 23:08:19 +0200876 void *p;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000877
Bram Moolenaarc799fe22019-05-28 23:08:19 +0200878 p = lalloc(size, message);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000879 if (p != NULL)
Bram Moolenaar964b3742019-05-24 18:54:09 +0200880 (void)vim_memset(p, 0, size);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000881 return p;
882}
883
884/*
885 * Low level memory allocation function.
886 * This is used often, KEEP IT FAST!
887 */
Bram Moolenaarc799fe22019-05-28 23:08:19 +0200888 void *
Bram Moolenaar964b3742019-05-24 18:54:09 +0200889lalloc(size_t size, int message)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000890{
Bram Moolenaarc799fe22019-05-28 23:08:19 +0200891 void *p; /* pointer to new storage space */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000892 static int releasing = FALSE; /* don't do mf_release_all() recursive */
893 int try_again;
Bram Moolenaar48e330a2016-02-23 14:53:34 +0100894#if defined(HAVE_AVAIL_MEM)
Bram Moolenaar964b3742019-05-24 18:54:09 +0200895 static size_t allocated = 0; /* allocated since last avail check */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000896#endif
897
Bram Moolenaar964b3742019-05-24 18:54:09 +0200898 // Safety check for allocating zero bytes
Bram Moolenaar071d4272004-06-13 20:20:40 +0000899 if (size == 0)
900 {
Bram Moolenaar964b3742019-05-24 18:54:09 +0200901 // Don't hide this message
Bram Moolenaar071d4272004-06-13 20:20:40 +0000902 emsg_silent = 0;
Bram Moolenaar964b3742019-05-24 18:54:09 +0200903 iemsg(_("E341: Internal error: lalloc(0, )"));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000904 return NULL;
905 }
906
907#ifdef MEM_PROFILE
908 mem_pre_alloc_l(&size);
909#endif
910
Bram Moolenaar071d4272004-06-13 20:20:40 +0000911 /*
912 * Loop when out of memory: Try to release some memfile blocks and
913 * if some blocks are released call malloc again.
914 */
915 for (;;)
916 {
917 /*
918 * Handle three kind of systems:
919 * 1. No check for available memory: Just return.
920 * 2. Slow check for available memory: call mch_avail_mem() after
921 * allocating KEEP_ROOM amount of memory.
922 * 3. Strict check for available memory: call mch_avail_mem()
923 */
Bram Moolenaarc799fe22019-05-28 23:08:19 +0200924 if ((p = malloc(size)) != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000925 {
926#ifndef HAVE_AVAIL_MEM
927 /* 1. No check for available memory: Just return. */
928 goto theend;
929#else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000930 /* 2. Slow check for available memory: call mch_avail_mem() after
931 * allocating (KEEP_ROOM / 2) amount of memory. */
932 allocated += size;
933 if (allocated < KEEP_ROOM / 2)
934 goto theend;
935 allocated = 0;
Bram Moolenaar48e330a2016-02-23 14:53:34 +0100936
Bram Moolenaar071d4272004-06-13 20:20:40 +0000937 /* 3. check for available memory: call mch_avail_mem() */
Bram Moolenaar11b73d62012-06-29 15:51:30 +0200938 if (mch_avail_mem(TRUE) < KEEP_ROOM_KB && !releasing)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000939 {
Bram Moolenaarc799fe22019-05-28 23:08:19 +0200940 free(p); /* System is low... no go! */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000941 p = NULL;
942 }
943 else
944 goto theend;
945#endif
946 }
947 /*
948 * Remember that mf_release_all() is being called to avoid an endless
949 * loop, because mf_release_all() may call alloc() recursively.
950 */
951 if (releasing)
952 break;
953 releasing = TRUE;
Bram Moolenaar661b1822005-07-28 22:36:45 +0000954
Bram Moolenaarf2405ed2017-03-16 19:58:25 +0100955 clear_sb_text(TRUE); /* free any scrollback text */
Bram Moolenaar661b1822005-07-28 22:36:45 +0000956 try_again = mf_release_all(); /* release as many blocks as possible */
Bram Moolenaar661b1822005-07-28 22:36:45 +0000957
Bram Moolenaar071d4272004-06-13 20:20:40 +0000958 releasing = FALSE;
959 if (!try_again)
960 break;
961 }
962
963 if (message && p == NULL)
964 do_outofmem_msg(size);
965
966theend:
967#ifdef MEM_PROFILE
Bram Moolenaarc799fe22019-05-28 23:08:19 +0200968 mem_post_alloc(&p, size);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000969#endif
970 return p;
971}
972
Bram Moolenaar75bdf6a2016-01-07 21:25:08 +0100973/*
974 * lalloc() with an ID for alloc_fail().
Bram Moolenaar75bdf6a2016-01-07 21:25:08 +0100975 */
Bram Moolenaar113e1072019-01-20 15:30:40 +0100976#if defined(FEAT_SIGNS) || defined(PROTO)
Bram Moolenaarc799fe22019-05-28 23:08:19 +0200977 void *
Bram Moolenaar964b3742019-05-24 18:54:09 +0200978lalloc_id(size_t size, int message, alloc_id_T id UNUSED)
Bram Moolenaar75bdf6a2016-01-07 21:25:08 +0100979{
980#ifdef FEAT_EVAL
Bram Moolenaara260b872016-01-15 20:48:22 +0100981 if (alloc_fail_id == id && alloc_does_fail(size))
Bram Moolenaar75bdf6a2016-01-07 21:25:08 +0100982 return NULL;
983#endif
Bram Moolenaar964b3742019-05-24 18:54:09 +0200984 return (lalloc(size, message));
Bram Moolenaar75bdf6a2016-01-07 21:25:08 +0100985}
Bram Moolenaar113e1072019-01-20 15:30:40 +0100986#endif
Bram Moolenaar75bdf6a2016-01-07 21:25:08 +0100987
Bram Moolenaar071d4272004-06-13 20:20:40 +0000988#if defined(MEM_PROFILE) || defined(PROTO)
989/*
990 * realloc() with memory profiling.
991 */
992 void *
Bram Moolenaar9b578142016-01-30 19:39:49 +0100993mem_realloc(void *ptr, size_t size)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000994{
995 void *p;
996
997 mem_pre_free(&ptr);
998 mem_pre_alloc_s(&size);
999
1000 p = realloc(ptr, size);
1001
1002 mem_post_alloc(&p, size);
1003
1004 return p;
1005}
1006#endif
1007
1008/*
1009* Avoid repeating the error message many times (they take 1 second each).
1010* Did_outofmem_msg is reset when a character is read.
1011*/
1012 void
Bram Moolenaar964b3742019-05-24 18:54:09 +02001013do_outofmem_msg(size_t size)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001014{
1015 if (!did_outofmem_msg)
1016 {
1017 /* Don't hide this message */
1018 emsg_silent = 0;
Bram Moolenaar79739e12011-10-26 11:41:00 +02001019
1020 /* Must come first to avoid coming back here when printing the error
1021 * message fails, e.g. when setting v:errmsg. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001022 did_outofmem_msg = TRUE;
Bram Moolenaar79739e12011-10-26 11:41:00 +02001023
Bram Moolenaar964b3742019-05-24 18:54:09 +02001024 semsg(_("E342: Out of memory! (allocating %lu bytes)"), (long_u)size);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001025 }
1026}
1027
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001028#if defined(EXITFREE) || defined(PROTO)
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001029
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001030/*
1031 * Free everything that we allocated.
1032 * Can be used to detect memory leaks, e.g., with ccmalloc.
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001033 * NOTE: This is tricky! Things are freed that functions depend on. Don't be
1034 * surprised if Vim crashes...
1035 * Some things can't be freed, esp. things local to a library function.
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001036 */
1037 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01001038free_all_mem(void)
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001039{
1040 buf_T *buf, *nextbuf;
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001041
1042 /* When we cause a crash here it is caught and Vim tries to exit cleanly.
1043 * Don't try freeing everything again. */
Bram Moolenaarb89a25f2016-06-01 23:08:39 +02001044 if (entered_free_all_mem)
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001045 return;
Bram Moolenaarb89a25f2016-06-01 23:08:39 +02001046 entered_free_all_mem = TRUE;
Bram Moolenaara9673212016-06-01 22:21:06 +02001047
Bram Moolenaar5d2bae82014-09-19 14:26:36 +02001048 /* Don't want to trigger autocommands from here on. */
1049 block_autocmds();
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001050
Bram Moolenaar5bedfc62010-07-20 22:30:01 +02001051 /* Close all tabs and windows. Reset 'equalalways' to avoid redraws. */
1052 p_ea = FALSE;
Bram Moolenaare5c83282019-05-03 23:15:37 +02001053 if (first_tabpage != NULL && first_tabpage->tp_next != NULL)
Bram Moolenaar7d47b6e2006-03-15 22:59:18 +00001054 do_cmdline_cmd((char_u *)"tabonly!");
Bram Moolenaar95f09602016-11-10 20:01:45 +01001055 if (!ONE_WINDOW)
Bram Moolenaar7d47b6e2006-03-15 22:59:18 +00001056 do_cmdline_cmd((char_u *)"only!");
Bram Moolenaar997fb4b2006-02-17 21:53:23 +00001057
Bram Moolenaarc4956c82006-03-12 21:58:43 +00001058# if defined(FEAT_SPELL)
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001059 /* Free all spell info. */
1060 spell_free_all();
1061# endif
1062
Bram Moolenaar48ac6712019-07-04 20:26:21 +02001063# if defined(FEAT_INS_EXPAND) && defined(FEAT_BEVAL_TERM)
Bram Moolenaarb301f6b2018-02-10 15:38:35 +01001064 ui_remove_balloon();
1065# endif
1066
Bram Moolenaarac9fb182019-04-27 13:04:13 +02001067 // Clear user commands (before deleting buffers).
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001068 ex_comclear(NULL);
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001069
Bram Moolenaare5c83282019-05-03 23:15:37 +02001070 // When exiting from mainerr_arg_missing curbuf has not been initialized,
1071 // and not much else.
1072 if (curbuf != NULL)
1073 {
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001074# ifdef FEAT_MENU
Bram Moolenaare5c83282019-05-03 23:15:37 +02001075 // Clear menus.
1076 do_cmdline_cmd((char_u *)"aunmenu *");
Bram Moolenaar21160b92009-11-11 15:56:10 +00001077# ifdef FEAT_MULTI_LANG
Bram Moolenaare5c83282019-05-03 23:15:37 +02001078 do_cmdline_cmd((char_u *)"menutranslate clear");
Bram Moolenaar21160b92009-11-11 15:56:10 +00001079# endif
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001080# endif
Bram Moolenaare5c83282019-05-03 23:15:37 +02001081 // Clear mappings, abbreviations, breakpoints.
1082 do_cmdline_cmd((char_u *)"lmapclear");
1083 do_cmdline_cmd((char_u *)"xmapclear");
1084 do_cmdline_cmd((char_u *)"mapclear");
1085 do_cmdline_cmd((char_u *)"mapclear!");
1086 do_cmdline_cmd((char_u *)"abclear");
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001087# if defined(FEAT_EVAL)
Bram Moolenaare5c83282019-05-03 23:15:37 +02001088 do_cmdline_cmd((char_u *)"breakdel *");
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001089# endif
Bram Moolenaar1e498f52005-06-26 22:29:44 +00001090# if defined(FEAT_PROFILE)
Bram Moolenaare5c83282019-05-03 23:15:37 +02001091 do_cmdline_cmd((char_u *)"profdel *");
Bram Moolenaar1e498f52005-06-26 22:29:44 +00001092# endif
Bram Moolenaarde0dfed2009-02-24 03:30:14 +00001093# if defined(FEAT_KEYMAP)
Bram Moolenaare5c83282019-05-03 23:15:37 +02001094 do_cmdline_cmd((char_u *)"set keymap=");
Bram Moolenaar48ac6712019-07-04 20:26:21 +02001095# endif
Bram Moolenaare5c83282019-05-03 23:15:37 +02001096 }
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001097
1098# ifdef FEAT_TITLE
1099 free_titles();
1100# endif
1101# if defined(FEAT_SEARCHPATH)
1102 free_findfile();
1103# endif
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001104
1105 /* Obviously named calls. */
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001106 free_all_autocmds();
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001107 clear_termcodes();
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001108 free_all_marks();
1109 alist_clear(&global_alist);
1110 free_homedir();
Bram Moolenaar24305862012-08-15 14:05:05 +02001111# if defined(FEAT_CMDL_COMPL)
1112 free_users();
1113# endif
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001114 free_search_patterns();
1115 free_old_sub();
1116 free_last_insert();
Bram Moolenaarac9fb182019-04-27 13:04:13 +02001117# if defined(FEAT_INS_EXPAND)
Bram Moolenaar7591bb32019-03-30 13:53:47 +01001118 free_insexpand_stuff();
Bram Moolenaarac9fb182019-04-27 13:04:13 +02001119# endif
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001120 free_prev_shellcmd();
1121 free_regexp_stuff();
1122 free_tag_stuff();
1123 free_cd_dir();
Bram Moolenaarde0dfed2009-02-24 03:30:14 +00001124# ifdef FEAT_SIGNS
1125 free_signs();
1126# endif
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001127# ifdef FEAT_EVAL
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001128 set_expr_line(NULL);
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001129# endif
1130# ifdef FEAT_DIFF
Bram Moolenaare5c83282019-05-03 23:15:37 +02001131 if (curtab != NULL)
1132 diff_clear(curtab);
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001133# endif
Bram Moolenaarf2405ed2017-03-16 19:58:25 +01001134 clear_sb_text(TRUE); /* free any scrollback text */
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001135
1136 /* Free some global vars. */
1137 vim_free(username);
Bram Moolenaaraf92ee82007-10-07 13:45:08 +00001138# ifdef FEAT_CLIPBOARD
Bram Moolenaar473de612013-06-08 18:19:48 +02001139 vim_regfree(clip_exclude_prog);
Bram Moolenaaraf92ee82007-10-07 13:45:08 +00001140# endif
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001141 vim_free(last_cmdline);
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001142# ifdef FEAT_CMDHIST
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001143 vim_free(new_last_cmdline);
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001144# endif
Bram Moolenaar238a5642006-02-21 22:12:05 +00001145 set_keep_msg(NULL, 0);
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001146
1147 /* Clear cmdline history. */
1148 p_hi = 0;
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001149# ifdef FEAT_CMDHIST
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001150 init_history();
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001151# endif
Bram Moolenaar48ac6712019-07-04 20:26:21 +02001152# ifdef FEAT_TEXT_PROP
Bram Moolenaar98aefe72018-12-13 22:20:09 +01001153 clear_global_prop_types();
Bram Moolenaar48ac6712019-07-04 20:26:21 +02001154# endif
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001155
Bram Moolenaar48ac6712019-07-04 20:26:21 +02001156# ifdef FEAT_QUICKFIX
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001157 {
Bram Moolenaarbd1e5d22009-04-29 09:02:44 +00001158 win_T *win;
1159 tabpage_T *tab;
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001160
1161 qf_free_all(NULL);
Bram Moolenaare5c83282019-05-03 23:15:37 +02001162 // Free all location lists
Bram Moolenaarbd1e5d22009-04-29 09:02:44 +00001163 FOR_ALL_TAB_WINDOWS(tab, win)
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001164 qf_free_all(win);
1165 }
Bram Moolenaar48ac6712019-07-04 20:26:21 +02001166# endif
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001167
Bram Moolenaare5c83282019-05-03 23:15:37 +02001168 // Close all script inputs.
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001169 close_all_scripts();
1170
Bram Moolenaare5c83282019-05-03 23:15:37 +02001171 if (curwin != NULL)
1172 // Destroy all windows. Must come before freeing buffers.
1173 win_free_all();
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001174
Bram Moolenaar4f198282017-10-23 21:53:30 +02001175 /* Free all option values. Must come after closing windows. */
1176 free_all_options();
1177
Bram Moolenaar2a329742008-04-01 12:53:43 +00001178 /* Free all buffers. Reset 'autochdir' to avoid accessing things that
1179 * were freed already. */
Bram Moolenaar48ac6712019-07-04 20:26:21 +02001180# ifdef FEAT_AUTOCHDIR
Bram Moolenaar2a329742008-04-01 12:53:43 +00001181 p_acd = FALSE;
Bram Moolenaar48ac6712019-07-04 20:26:21 +02001182# endif
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001183 for (buf = firstbuf; buf != NULL; )
1184 {
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001185 bufref_T bufref;
1186
1187 set_bufref(&bufref, buf);
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001188 nextbuf = buf->b_next;
Bram Moolenaar42ec6562012-02-22 14:58:37 +01001189 close_buffer(NULL, buf, DOBUF_WIPE, FALSE);
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001190 if (bufref_valid(&bufref))
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001191 buf = nextbuf; /* didn't work, try next one */
1192 else
1193 buf = firstbuf;
1194 }
1195
Bram Moolenaar48ac6712019-07-04 20:26:21 +02001196# ifdef FEAT_ARABIC
1197 free_arshape_buf();
1198# endif
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001199
1200 /* Clear registers. */
1201 clear_registers();
1202 ResetRedobuff();
1203 ResetRedobuff();
1204
Bram Moolenaar9f84ded2018-10-20 20:54:02 +02001205# if defined(FEAT_CLIENTSERVER) && defined(FEAT_X11)
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001206 vim_free(serverDelayedStartName);
Bram Moolenaar9f84ded2018-10-20 20:54:02 +02001207# endif
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001208
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001209 /* highlight info */
1210 free_highlight();
1211
Bram Moolenaar1e498f52005-06-26 22:29:44 +00001212 reset_last_sourcing();
1213
Bram Moolenaare5c83282019-05-03 23:15:37 +02001214 if (first_tabpage != NULL)
1215 {
1216 free_tabpage(first_tabpage);
1217 first_tabpage = NULL;
1218 }
Bram Moolenaar7d47b6e2006-03-15 22:59:18 +00001219
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001220# ifdef UNIX
1221 /* Machine-specific free. */
1222 mch_free_mem();
1223# endif
1224
1225 /* message history */
1226 for (;;)
1227 if (delete_first_msg() == FAIL)
1228 break;
1229
Bram Moolenaar655da312016-05-28 22:22:34 +02001230# ifdef FEAT_JOB_CHANNEL
1231 channel_free_all();
Bram Moolenaar655da312016-05-28 22:22:34 +02001232# endif
Bram Moolenaar9f84ded2018-10-20 20:54:02 +02001233# ifdef FEAT_TIMERS
Bram Moolenaar623e2632016-07-30 22:47:56 +02001234 timer_free_all();
Bram Moolenaar9f84ded2018-10-20 20:54:02 +02001235# endif
Bram Moolenaar437bafe2016-08-01 15:40:54 +02001236# ifdef FEAT_EVAL
1237 /* must be after channel_free_all() with unrefs partials */
1238 eval_clear();
1239# endif
1240# ifdef FEAT_JOB_CHANNEL
1241 /* must be after eval_clear() with unrefs jobs */
1242 job_free_all();
1243# endif
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001244
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001245 free_termoptions();
1246
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001247 /* screenlines (can't display anything now!) */
1248 free_screenlines();
1249
Bram Moolenaar82febc12019-06-10 14:48:59 +02001250# if defined(FEAT_SOUND)
1251 sound_free();
1252# endif
Bram Moolenaar9f84ded2018-10-20 20:54:02 +02001253# if defined(USE_XSMP)
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001254 xsmp_close();
Bram Moolenaar9f84ded2018-10-20 20:54:02 +02001255# endif
1256# ifdef FEAT_GUI_GTK
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001257 gui_mch_free_all();
Bram Moolenaar9f84ded2018-10-20 20:54:02 +02001258# endif
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001259 clear_hl_tables();
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001260
1261 vim_free(IObuff);
1262 vim_free(NameBuff);
Bram Moolenaar9f84ded2018-10-20 20:54:02 +02001263# ifdef FEAT_QUICKFIX
1264 check_quickfix_busy();
1265# endif
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001266}
1267#endif
1268
Bram Moolenaar071d4272004-06-13 20:20:40 +00001269/*
Bram Moolenaare980d8a2010-12-08 13:11:21 +01001270 * Copy "string" into newly allocated memory.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001271 */
1272 char_u *
Bram Moolenaar9b578142016-01-30 19:39:49 +01001273vim_strsave(char_u *string)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001274{
1275 char_u *p;
Bram Moolenaar964b3742019-05-24 18:54:09 +02001276 size_t len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001277
Bram Moolenaar964b3742019-05-24 18:54:09 +02001278 len = STRLEN(string) + 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001279 p = alloc(len);
1280 if (p != NULL)
Bram Moolenaar964b3742019-05-24 18:54:09 +02001281 mch_memmove(p, string, len);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001282 return p;
1283}
1284
Bram Moolenaare980d8a2010-12-08 13:11:21 +01001285/*
1286 * Copy up to "len" bytes of "string" into newly allocated memory and
1287 * terminate with a NUL.
1288 * The allocated memory always has size "len + 1", also when "string" is
1289 * shorter.
1290 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001291 char_u *
Bram Moolenaar9b578142016-01-30 19:39:49 +01001292vim_strnsave(char_u *string, int len)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001293{
1294 char_u *p;
1295
Bram Moolenaar51e14382019-05-25 20:21:28 +02001296 p = alloc(len + 1);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001297 if (p != NULL)
1298 {
1299 STRNCPY(p, string, len);
1300 p[len] = NUL;
1301 }
1302 return p;
1303}
1304
Bram Moolenaar071d4272004-06-13 20:20:40 +00001305/*
Bram Moolenaarc1a9bc12018-12-28 21:59:29 +01001306 * Copy "p[len]" into allocated memory, ignoring NUL characters.
1307 * Returns NULL when out of memory.
1308 */
1309 char_u *
Bram Moolenaar964b3742019-05-24 18:54:09 +02001310vim_memsave(char_u *p, size_t len)
Bram Moolenaarc1a9bc12018-12-28 21:59:29 +01001311{
Bram Moolenaar964b3742019-05-24 18:54:09 +02001312 char_u *ret = alloc(len);
Bram Moolenaarc1a9bc12018-12-28 21:59:29 +01001313
1314 if (ret != NULL)
Bram Moolenaar964b3742019-05-24 18:54:09 +02001315 mch_memmove(ret, p, len);
Bram Moolenaarc1a9bc12018-12-28 21:59:29 +01001316 return ret;
1317}
1318
1319/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00001320 * Same as vim_strsave(), but any characters found in esc_chars are preceded
1321 * by a backslash.
1322 */
1323 char_u *
Bram Moolenaar9b578142016-01-30 19:39:49 +01001324vim_strsave_escaped(char_u *string, char_u *esc_chars)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001325{
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001326 return vim_strsave_escaped_ext(string, esc_chars, '\\', FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001327}
1328
1329/*
1330 * Same as vim_strsave_escaped(), but when "bsl" is TRUE also escape
1331 * characters where rem_backslash() would remove the backslash.
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001332 * Escape the characters with "cc".
Bram Moolenaar071d4272004-06-13 20:20:40 +00001333 */
1334 char_u *
Bram Moolenaar9b578142016-01-30 19:39:49 +01001335vim_strsave_escaped_ext(
1336 char_u *string,
1337 char_u *esc_chars,
1338 int cc,
1339 int bsl)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001340{
1341 char_u *p;
1342 char_u *p2;
1343 char_u *escaped_string;
1344 unsigned length;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001345 int l;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001346
1347 /*
1348 * First count the number of backslashes required.
1349 * Then allocate the memory and insert them.
1350 */
1351 length = 1; /* count the trailing NUL */
1352 for (p = string; *p; p++)
1353 {
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001354 if (has_mbyte && (l = (*mb_ptr2len)(p)) > 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001355 {
1356 length += l; /* count a multibyte char */
1357 p += l - 1;
1358 continue;
1359 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001360 if (vim_strchr(esc_chars, *p) != NULL || (bsl && rem_backslash(p)))
1361 ++length; /* count a backslash */
1362 ++length; /* count an ordinary char */
1363 }
1364 escaped_string = alloc(length);
1365 if (escaped_string != NULL)
1366 {
1367 p2 = escaped_string;
1368 for (p = string; *p; p++)
1369 {
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001370 if (has_mbyte && (l = (*mb_ptr2len)(p)) > 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001371 {
1372 mch_memmove(p2, p, (size_t)l);
1373 p2 += l;
1374 p += l - 1; /* skip multibyte char */
1375 continue;
1376 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001377 if (vim_strchr(esc_chars, *p) != NULL || (bsl && rem_backslash(p)))
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001378 *p2++ = cc;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001379 *p2++ = *p;
1380 }
1381 *p2 = NUL;
1382 }
1383 return escaped_string;
1384}
1385
Bram Moolenaar7693ec62008-07-24 18:29:37 +00001386/*
1387 * Return TRUE when 'shell' has "csh" in the tail.
1388 */
1389 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01001390csh_like_shell(void)
Bram Moolenaar7693ec62008-07-24 18:29:37 +00001391{
1392 return (strstr((char *)gettail(p_sh), "csh") != NULL);
1393}
Bram Moolenaar7693ec62008-07-24 18:29:37 +00001394
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001395/*
1396 * Escape "string" for use as a shell argument with system().
Bram Moolenaar21160b92009-11-11 15:56:10 +00001397 * This uses single quotes, except when we know we need to use double quotes
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001398 * (MS-DOS and MS-Windows without 'shellslash' set).
Bram Moolenaarffd66c42008-07-16 20:43:37 +00001399 * Escape a newline, depending on the 'shell' option.
1400 * When "do_special" is TRUE also replace "!", "%", "#" and things starting
1401 * with "<" like "<cfile>".
Bram Moolenaar26df0922014-02-23 23:39:13 +01001402 * When "do_newline" is FALSE do not escape newline unless it is csh shell.
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001403 * Returns the result in allocated memory, NULL if we have run out.
1404 */
1405 char_u *
Bram Moolenaar9b578142016-01-30 19:39:49 +01001406vim_strsave_shellescape(char_u *string, int do_special, int do_newline)
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001407{
1408 unsigned length;
1409 char_u *p;
1410 char_u *d;
1411 char_u *escaped_string;
Bram Moolenaar05bb9532008-07-04 09:44:11 +00001412 int l;
Bram Moolenaarffd66c42008-07-16 20:43:37 +00001413 int csh_like;
1414
1415 /* Only csh and similar shells expand '!' within single quotes. For sh and
1416 * the like we must not put a backslash before it, it will be taken
1417 * literally. If do_special is set the '!' will be escaped twice.
1418 * Csh also needs to have "\n" escaped twice when do_special is set. */
Bram Moolenaar7693ec62008-07-24 18:29:37 +00001419 csh_like = csh_like_shell();
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001420
1421 /* First count the number of extra bytes required. */
Bram Moolenaar77f66d62007-02-20 02:16:18 +00001422 length = (unsigned)STRLEN(string) + 3; /* two quotes and a trailing NUL */
Bram Moolenaar91acfff2017-03-12 19:22:36 +01001423 for (p = string; *p != NUL; MB_PTR_ADV(p))
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001424 {
Bram Moolenaar4f974752019-02-17 17:44:42 +01001425# ifdef MSWIN
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001426 if (!p_ssl)
1427 {
1428 if (*p == '"')
1429 ++length; /* " -> "" */
1430 }
1431 else
1432# endif
1433 if (*p == '\'')
1434 length += 3; /* ' => '\'' */
Bram Moolenaar26df0922014-02-23 23:39:13 +01001435 if ((*p == '\n' && (csh_like || do_newline))
1436 || (*p == '!' && (csh_like || do_special)))
Bram Moolenaarffd66c42008-07-16 20:43:37 +00001437 {
1438 ++length; /* insert backslash */
1439 if (csh_like && do_special)
1440 ++length; /* insert backslash */
1441 }
Bram Moolenaar05bb9532008-07-04 09:44:11 +00001442 if (do_special && find_cmdline_var(p, &l) >= 0)
1443 {
1444 ++length; /* insert backslash */
1445 p += l - 1;
1446 }
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001447 }
1448
1449 /* Allocate memory for the result and fill it. */
1450 escaped_string = alloc(length);
1451 if (escaped_string != NULL)
1452 {
1453 d = escaped_string;
1454
1455 /* add opening quote */
Bram Moolenaar4f974752019-02-17 17:44:42 +01001456# ifdef MSWIN
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001457 if (!p_ssl)
1458 *d++ = '"';
1459 else
1460# endif
1461 *d++ = '\'';
1462
1463 for (p = string; *p != NUL; )
1464 {
Bram Moolenaar4f974752019-02-17 17:44:42 +01001465# ifdef MSWIN
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001466 if (!p_ssl)
1467 {
1468 if (*p == '"')
1469 {
1470 *d++ = '"';
1471 *d++ = '"';
1472 ++p;
1473 continue;
1474 }
1475 }
1476 else
1477# endif
1478 if (*p == '\'')
1479 {
Bram Moolenaar05bb9532008-07-04 09:44:11 +00001480 *d++ = '\'';
1481 *d++ = '\\';
1482 *d++ = '\'';
1483 *d++ = '\'';
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001484 ++p;
1485 continue;
1486 }
Bram Moolenaar26df0922014-02-23 23:39:13 +01001487 if ((*p == '\n' && (csh_like || do_newline))
1488 || (*p == '!' && (csh_like || do_special)))
Bram Moolenaarffd66c42008-07-16 20:43:37 +00001489 {
1490 *d++ = '\\';
1491 if (csh_like && do_special)
1492 *d++ = '\\';
1493 *d++ = *p++;
1494 continue;
1495 }
Bram Moolenaar05bb9532008-07-04 09:44:11 +00001496 if (do_special && find_cmdline_var(p, &l) >= 0)
1497 {
1498 *d++ = '\\'; /* insert backslash */
1499 while (--l >= 0) /* copy the var */
1500 *d++ = *p++;
Bram Moolenaar099d01d2009-11-25 16:14:45 +00001501 continue;
Bram Moolenaar05bb9532008-07-04 09:44:11 +00001502 }
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001503
1504 MB_COPY_CHAR(p, d);
1505 }
1506
1507 /* add terminating quote and finish with a NUL */
Bram Moolenaar4f974752019-02-17 17:44:42 +01001508# ifdef MSWIN
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001509 if (!p_ssl)
1510 *d++ = '"';
1511 else
1512# endif
1513 *d++ = '\'';
1514 *d = NUL;
1515 }
1516
1517 return escaped_string;
1518}
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001519
Bram Moolenaar071d4272004-06-13 20:20:40 +00001520/*
1521 * Like vim_strsave(), but make all characters uppercase.
1522 * This uses ASCII lower-to-upper case translation, language independent.
1523 */
1524 char_u *
Bram Moolenaar9b578142016-01-30 19:39:49 +01001525vim_strsave_up(char_u *string)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001526{
1527 char_u *p1;
1528
1529 p1 = vim_strsave(string);
1530 vim_strup(p1);
1531 return p1;
1532}
1533
1534/*
1535 * Like vim_strnsave(), but make all characters uppercase.
1536 * This uses ASCII lower-to-upper case translation, language independent.
1537 */
1538 char_u *
Bram Moolenaar9b578142016-01-30 19:39:49 +01001539vim_strnsave_up(char_u *string, int len)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001540{
1541 char_u *p1;
1542
1543 p1 = vim_strnsave(string, len);
1544 vim_strup(p1);
1545 return p1;
1546}
1547
1548/*
1549 * ASCII lower-to-upper case translation, language independent.
1550 */
1551 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01001552vim_strup(
1553 char_u *p)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001554{
1555 char_u *p2;
1556 int c;
1557
1558 if (p != NULL)
1559 {
1560 p2 = p;
1561 while ((c = *p2) != NUL)
1562#ifdef EBCDIC
1563 *p2++ = isalpha(c) ? toupper(c) : c;
1564#else
1565 *p2++ = (c < 'a' || c > 'z') ? c : (c - 0x20);
1566#endif
1567 }
1568}
1569
Bram Moolenaarc4956c82006-03-12 21:58:43 +00001570#if defined(FEAT_EVAL) || defined(FEAT_SPELL) || defined(PROTO)
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001571/*
1572 * Make string "s" all upper-case and return it in allocated memory.
1573 * Handles multi-byte characters as well as possible.
1574 * Returns NULL when out of memory.
1575 */
1576 char_u *
Bram Moolenaar9b578142016-01-30 19:39:49 +01001577strup_save(char_u *orig)
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001578{
1579 char_u *p;
1580 char_u *res;
1581
1582 res = p = vim_strsave(orig);
1583
1584 if (res != NULL)
1585 while (*p != NUL)
1586 {
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001587 int l;
1588
1589 if (enc_utf8)
1590 {
1591 int c, uc;
Bram Moolenaar70b2a562012-01-10 22:26:17 +01001592 int newl;
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001593 char_u *s;
1594
1595 c = utf_ptr2char(p);
Bram Moolenaare6640ad2017-12-22 21:06:56 +01001596 l = utf_ptr2len(p);
1597 if (c == 0)
1598 {
1599 /* overlong sequence, use only the first byte */
1600 c = *p;
1601 l = 1;
1602 }
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001603 uc = utf_toupper(c);
1604
1605 /* Reallocate string when byte count changes. This is rare,
1606 * thus it's OK to do another malloc()/free(). */
Bram Moolenaar70b2a562012-01-10 22:26:17 +01001607 newl = utf_char2len(uc);
1608 if (newl != l)
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001609 {
Bram Moolenaar964b3742019-05-24 18:54:09 +02001610 s = alloc(STRLEN(res) + 1 + newl - l);
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001611 if (s == NULL)
Bram Moolenaarcc5b22b2017-01-26 22:51:56 +01001612 {
1613 vim_free(res);
1614 return NULL;
1615 }
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001616 mch_memmove(s, res, p - res);
Bram Moolenaar70b2a562012-01-10 22:26:17 +01001617 STRCPY(s + (p - res) + newl, p + l);
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001618 p = s + (p - res);
1619 vim_free(res);
1620 res = s;
1621 }
1622
1623 utf_char2bytes(uc, p);
Bram Moolenaar70b2a562012-01-10 22:26:17 +01001624 p += newl;
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001625 }
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001626 else if (has_mbyte && (l = (*mb_ptr2len)(p)) > 1)
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001627 p += l; /* skip multi-byte character */
1628 else
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001629 {
1630 *p = TOUPPER_LOC(*p); /* note that toupper() can be a macro */
1631 p++;
1632 }
1633 }
1634
1635 return res;
1636}
Bram Moolenaarcc5b22b2017-01-26 22:51:56 +01001637
1638/*
1639 * Make string "s" all lower-case and return it in allocated memory.
1640 * Handles multi-byte characters as well as possible.
1641 * Returns NULL when out of memory.
1642 */
1643 char_u *
1644strlow_save(char_u *orig)
1645{
1646 char_u *p;
1647 char_u *res;
1648
1649 res = p = vim_strsave(orig);
1650
1651 if (res != NULL)
1652 while (*p != NUL)
1653 {
Bram Moolenaarcc5b22b2017-01-26 22:51:56 +01001654 int l;
1655
1656 if (enc_utf8)
1657 {
1658 int c, lc;
1659 int newl;
1660 char_u *s;
1661
1662 c = utf_ptr2char(p);
Bram Moolenaare6640ad2017-12-22 21:06:56 +01001663 l = utf_ptr2len(p);
1664 if (c == 0)
1665 {
1666 /* overlong sequence, use only the first byte */
1667 c = *p;
1668 l = 1;
1669 }
Bram Moolenaarcc5b22b2017-01-26 22:51:56 +01001670 lc = utf_tolower(c);
1671
1672 /* Reallocate string when byte count changes. This is rare,
1673 * thus it's OK to do another malloc()/free(). */
Bram Moolenaarcc5b22b2017-01-26 22:51:56 +01001674 newl = utf_char2len(lc);
1675 if (newl != l)
1676 {
Bram Moolenaar964b3742019-05-24 18:54:09 +02001677 s = alloc(STRLEN(res) + 1 + newl - l);
Bram Moolenaarcc5b22b2017-01-26 22:51:56 +01001678 if (s == NULL)
1679 {
1680 vim_free(res);
1681 return NULL;
1682 }
1683 mch_memmove(s, res, p - res);
1684 STRCPY(s + (p - res) + newl, p + l);
1685 p = s + (p - res);
1686 vim_free(res);
1687 res = s;
1688 }
1689
1690 utf_char2bytes(lc, p);
1691 p += newl;
1692 }
1693 else if (has_mbyte && (l = (*mb_ptr2len)(p)) > 1)
1694 p += l; /* skip multi-byte character */
1695 else
Bram Moolenaarcc5b22b2017-01-26 22:51:56 +01001696 {
1697 *p = TOLOWER_LOC(*p); /* note that tolower() can be a macro */
1698 p++;
1699 }
1700 }
1701
1702 return res;
1703}
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001704#endif
1705
Bram Moolenaar071d4272004-06-13 20:20:40 +00001706/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00001707 * delete spaces at the end of a string
1708 */
1709 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01001710del_trailing_spaces(char_u *ptr)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001711{
1712 char_u *q;
1713
1714 q = ptr + STRLEN(ptr);
Bram Moolenaar1c465442017-03-12 20:10:05 +01001715 while (--q > ptr && VIM_ISWHITE(q[0]) && q[-1] != '\\' && q[-1] != Ctrl_V)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001716 *q = NUL;
1717}
1718
1719/*
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001720 * Like strncpy(), but always terminate the result with one NUL.
Bram Moolenaard042c562005-06-30 22:04:15 +00001721 * "to" must be "len + 1" long!
Bram Moolenaar071d4272004-06-13 20:20:40 +00001722 */
1723 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01001724vim_strncpy(char_u *to, char_u *from, size_t len)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001725{
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001726 STRNCPY(to, from, len);
1727 to[len] = NUL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001728}
1729
1730/*
Bram Moolenaaref9d6aa2011-04-11 16:56:35 +02001731 * Like strcat(), but make sure the result fits in "tosize" bytes and is
Bram Moolenaar45600ce2017-01-27 21:54:07 +01001732 * always NUL terminated. "from" and "to" may overlap.
Bram Moolenaaref9d6aa2011-04-11 16:56:35 +02001733 */
1734 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01001735vim_strcat(char_u *to, char_u *from, size_t tosize)
Bram Moolenaaref9d6aa2011-04-11 16:56:35 +02001736{
1737 size_t tolen = STRLEN(to);
1738 size_t fromlen = STRLEN(from);
1739
1740 if (tolen + fromlen + 1 > tosize)
1741 {
1742 mch_memmove(to + tolen, from, tosize - tolen - 1);
1743 to[tosize - 1] = NUL;
1744 }
1745 else
Bram Moolenaar45600ce2017-01-27 21:54:07 +01001746 mch_memmove(to + tolen, from, fromlen + 1);
Bram Moolenaaref9d6aa2011-04-11 16:56:35 +02001747}
1748
1749/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00001750 * Isolate one part of a string option where parts are separated with
1751 * "sep_chars".
Bram Moolenaar83bab712005-08-01 21:58:57 +00001752 * The part is copied into "buf[maxlen]".
Bram Moolenaar071d4272004-06-13 20:20:40 +00001753 * "*option" is advanced to the next part.
1754 * The length is returned.
1755 */
1756 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01001757copy_option_part(
1758 char_u **option,
1759 char_u *buf,
1760 int maxlen,
1761 char *sep_chars)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001762{
1763 int len = 0;
1764 char_u *p = *option;
1765
1766 /* skip '.' at start of option part, for 'suffixes' */
1767 if (*p == '.')
1768 buf[len++] = *p++;
1769 while (*p != NUL && vim_strchr((char_u *)sep_chars, *p) == NULL)
1770 {
1771 /*
1772 * Skip backslash before a separator character and space.
1773 */
1774 if (p[0] == '\\' && vim_strchr((char_u *)sep_chars, p[1]) != NULL)
1775 ++p;
1776 if (len < maxlen - 1)
1777 buf[len++] = *p;
1778 ++p;
1779 }
1780 buf[len] = NUL;
1781
1782 if (*p != NUL && *p != ',') /* skip non-standard separator */
1783 ++p;
1784 p = skip_to_option_part(p); /* p points to next file name */
1785
1786 *option = p;
1787 return len;
1788}
1789
1790/*
Bram Moolenaar4770d092006-01-12 23:22:24 +00001791 * Replacement for free() that ignores NULL pointers.
1792 * Also skip free() when exiting for sure, this helps when we caught a deadly
1793 * signal that was caused by a crash in free().
Bram Moolenaard23a8232018-02-10 18:45:26 +01001794 * If you want to set NULL after calling this function, you should use
1795 * VIM_CLEAR() instead.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001796 */
1797 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01001798vim_free(void *x)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001799{
Bram Moolenaar4770d092006-01-12 23:22:24 +00001800 if (x != NULL && !really_exiting)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001801 {
1802#ifdef MEM_PROFILE
1803 mem_pre_free(&x);
1804#endif
1805 free(x);
1806 }
1807}
1808
1809#ifndef HAVE_MEMSET
1810 void *
Bram Moolenaar9b578142016-01-30 19:39:49 +01001811vim_memset(void *ptr, int c, size_t size)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001812{
1813 char *p = ptr;
1814
1815 while (size-- > 0)
1816 *p++ = c;
1817 return ptr;
1818}
1819#endif
1820
Bram Moolenaar071d4272004-06-13 20:20:40 +00001821#if (!defined(HAVE_STRCASECMP) && !defined(HAVE_STRICMP)) || defined(PROTO)
1822/*
1823 * Compare two strings, ignoring case, using current locale.
1824 * Doesn't work for multi-byte characters.
1825 * return 0 for match, < 0 for smaller, > 0 for bigger
1826 */
1827 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01001828vim_stricmp(char *s1, char *s2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001829{
1830 int i;
1831
1832 for (;;)
1833 {
1834 i = (int)TOLOWER_LOC(*s1) - (int)TOLOWER_LOC(*s2);
1835 if (i != 0)
1836 return i; /* this character different */
1837 if (*s1 == NUL)
1838 break; /* strings match until NUL */
1839 ++s1;
1840 ++s2;
1841 }
1842 return 0; /* strings match */
1843}
1844#endif
1845
1846#if (!defined(HAVE_STRNCASECMP) && !defined(HAVE_STRNICMP)) || defined(PROTO)
1847/*
1848 * Compare two strings, for length "len", ignoring case, using current locale.
1849 * Doesn't work for multi-byte characters.
1850 * return 0 for match, < 0 for smaller, > 0 for bigger
1851 */
1852 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01001853vim_strnicmp(char *s1, char *s2, size_t len)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001854{
1855 int i;
1856
1857 while (len > 0)
1858 {
1859 i = (int)TOLOWER_LOC(*s1) - (int)TOLOWER_LOC(*s2);
1860 if (i != 0)
1861 return i; /* this character different */
1862 if (*s1 == NUL)
1863 break; /* strings match until NUL */
1864 ++s1;
1865 ++s2;
1866 --len;
1867 }
1868 return 0; /* strings match */
1869}
1870#endif
1871
Bram Moolenaar071d4272004-06-13 20:20:40 +00001872/*
1873 * Version of strchr() and strrchr() that handle unsigned char strings
Bram Moolenaar05159a02005-02-26 23:04:13 +00001874 * with characters from 128 to 255 correctly. It also doesn't return a
1875 * pointer to the NUL at the end of the string.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001876 */
1877 char_u *
Bram Moolenaar9b578142016-01-30 19:39:49 +01001878vim_strchr(char_u *string, int c)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001879{
1880 char_u *p;
1881 int b;
1882
1883 p = string;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001884 if (enc_utf8 && c >= 0x80)
1885 {
1886 while (*p != NUL)
1887 {
Bram Moolenaarace95982017-03-29 17:30:27 +02001888 int l = utfc_ptr2len(p);
Bram Moolenaard82a2a92015-04-21 14:02:35 +02001889
1890 /* Avoid matching an illegal byte here. */
1891 if (utf_ptr2char(p) == c && l > 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001892 return p;
Bram Moolenaard82a2a92015-04-21 14:02:35 +02001893 p += l;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001894 }
1895 return NULL;
1896 }
1897 if (enc_dbcs != 0 && c > 255)
1898 {
1899 int n2 = c & 0xff;
1900
1901 c = ((unsigned)c >> 8) & 0xff;
1902 while ((b = *p) != NUL)
1903 {
1904 if (b == c && p[1] == n2)
1905 return p;
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001906 p += (*mb_ptr2len)(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001907 }
1908 return NULL;
1909 }
1910 if (has_mbyte)
1911 {
1912 while ((b = *p) != NUL)
1913 {
1914 if (b == c)
1915 return p;
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001916 p += (*mb_ptr2len)(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001917 }
1918 return NULL;
1919 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001920 while ((b = *p) != NUL)
1921 {
1922 if (b == c)
1923 return p;
1924 ++p;
1925 }
1926 return NULL;
1927}
1928
1929/*
Bram Moolenaar05159a02005-02-26 23:04:13 +00001930 * Version of strchr() that only works for bytes and handles unsigned char
1931 * strings with characters above 128 correctly. It also doesn't return a
1932 * pointer to the NUL at the end of the string.
1933 */
1934 char_u *
Bram Moolenaar9b578142016-01-30 19:39:49 +01001935vim_strbyte(char_u *string, int c)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001936{
1937 char_u *p = string;
1938
1939 while (*p != NUL)
1940 {
1941 if (*p == c)
1942 return p;
1943 ++p;
1944 }
1945 return NULL;
1946}
1947
1948/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00001949 * Search for last occurrence of "c" in "string".
Bram Moolenaarac6e65f2005-08-29 22:25:38 +00001950 * Return NULL if not found.
Bram Moolenaar05159a02005-02-26 23:04:13 +00001951 * Does not handle multi-byte char for "c"!
Bram Moolenaar071d4272004-06-13 20:20:40 +00001952 */
1953 char_u *
Bram Moolenaar9b578142016-01-30 19:39:49 +01001954vim_strrchr(char_u *string, int c)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001955{
1956 char_u *retval = NULL;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001957 char_u *p = string;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001958
Bram Moolenaar05159a02005-02-26 23:04:13 +00001959 while (*p)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001960 {
Bram Moolenaar05159a02005-02-26 23:04:13 +00001961 if (*p == c)
1962 retval = p;
Bram Moolenaar91acfff2017-03-12 19:22:36 +01001963 MB_PTR_ADV(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001964 }
1965 return retval;
1966}
1967
1968/*
1969 * Vim's version of strpbrk(), in case it's missing.
1970 * Don't generate a prototype for this, causes problems when it's not used.
1971 */
1972#ifndef PROTO
1973# ifndef HAVE_STRPBRK
1974# ifdef vim_strpbrk
1975# undef vim_strpbrk
1976# endif
1977 char_u *
Bram Moolenaar9b578142016-01-30 19:39:49 +01001978vim_strpbrk(char_u *s, char_u *charset)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001979{
1980 while (*s)
1981 {
1982 if (vim_strchr(charset, *s) != NULL)
1983 return s;
Bram Moolenaar91acfff2017-03-12 19:22:36 +01001984 MB_PTR_ADV(s);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001985 }
1986 return NULL;
1987}
1988# endif
1989#endif
1990
1991/*
1992 * Vim has its own isspace() function, because on some machines isspace()
1993 * can't handle characters above 128.
1994 */
1995 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01001996vim_isspace(int x)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001997{
1998 return ((x >= 9 && x <= 13) || x == ' ');
1999}
2000
2001/************************************************************************
Bram Moolenaar383f9bc2005-01-19 22:18:32 +00002002 * Functions for handling growing arrays.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002003 */
2004
2005/*
2006 * Clear an allocated growing array.
2007 */
2008 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01002009ga_clear(garray_T *gap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002010{
2011 vim_free(gap->ga_data);
2012 ga_init(gap);
2013}
2014
2015/*
2016 * Clear a growing array that contains a list of strings.
2017 */
2018 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01002019ga_clear_strings(garray_T *gap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002020{
2021 int i;
2022
2023 for (i = 0; i < gap->ga_len; ++i)
2024 vim_free(((char_u **)(gap->ga_data))[i]);
2025 ga_clear(gap);
2026}
2027
2028/*
2029 * Initialize a growing array. Don't forget to set ga_itemsize and
2030 * ga_growsize! Or use ga_init2().
2031 */
2032 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01002033ga_init(garray_T *gap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002034{
2035 gap->ga_data = NULL;
Bram Moolenaar86b68352004-12-27 21:59:20 +00002036 gap->ga_maxlen = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002037 gap->ga_len = 0;
2038}
2039
2040 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01002041ga_init2(garray_T *gap, int itemsize, int growsize)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002042{
2043 ga_init(gap);
2044 gap->ga_itemsize = itemsize;
2045 gap->ga_growsize = growsize;
2046}
2047
2048/*
2049 * Make room in growing array "gap" for at least "n" items.
2050 * Return FAIL for failure, OK otherwise.
2051 */
2052 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01002053ga_grow(garray_T *gap, int n)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002054{
Bram Moolenaar7282bc32012-02-22 18:12:32 +01002055 size_t old_len;
2056 size_t new_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002057 char_u *pp;
2058
Bram Moolenaar86b68352004-12-27 21:59:20 +00002059 if (gap->ga_maxlen - gap->ga_len < n)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002060 {
2061 if (n < gap->ga_growsize)
2062 n = gap->ga_growsize;
Bram Moolenaarc47ed442019-06-01 14:36:26 +02002063
2064 // A linear growth is very inefficient when the array grows big. This
2065 // is a compromise between allocating memory that won't be used and too
2066 // many copy operations. A factor of 1.5 seems reasonable.
2067 if (n < gap->ga_len / 2)
2068 n = gap->ga_len / 2;
2069
Bram Moolenaar7282bc32012-02-22 18:12:32 +01002070 new_len = gap->ga_itemsize * (gap->ga_len + n);
Bram Moolenaarc799fe22019-05-28 23:08:19 +02002071 pp = vim_realloc(gap->ga_data, new_len);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002072 if (pp == NULL)
2073 return FAIL;
Bram Moolenaar7282bc32012-02-22 18:12:32 +01002074 old_len = gap->ga_itemsize * gap->ga_maxlen;
2075 vim_memset(pp + old_len, 0, new_len - old_len);
Bram Moolenaar86b68352004-12-27 21:59:20 +00002076 gap->ga_maxlen = gap->ga_len + n;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002077 gap->ga_data = pp;
2078 }
2079 return OK;
2080}
2081
Bram Moolenaar113e1072019-01-20 15:30:40 +01002082#if defined(FEAT_EVAL) || defined(FEAT_SEARCHPATH) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002083/*
Bram Moolenaar80a7dcf2010-08-04 17:07:20 +02002084 * For a growing array that contains a list of strings: concatenate all the
Bram Moolenaar1b1063a2014-05-07 18:35:30 +02002085 * strings with a separating "sep".
Bram Moolenaar80a7dcf2010-08-04 17:07:20 +02002086 * Returns NULL when out of memory.
2087 */
2088 char_u *
Bram Moolenaar9b578142016-01-30 19:39:49 +01002089ga_concat_strings(garray_T *gap, char *sep)
Bram Moolenaar80a7dcf2010-08-04 17:07:20 +02002090{
2091 int i;
2092 int len = 0;
Bram Moolenaar1b1063a2014-05-07 18:35:30 +02002093 int sep_len = (int)STRLEN(sep);
Bram Moolenaar80a7dcf2010-08-04 17:07:20 +02002094 char_u *s;
Bram Moolenaar1b1063a2014-05-07 18:35:30 +02002095 char_u *p;
Bram Moolenaar80a7dcf2010-08-04 17:07:20 +02002096
2097 for (i = 0; i < gap->ga_len; ++i)
Bram Moolenaar1b1063a2014-05-07 18:35:30 +02002098 len += (int)STRLEN(((char_u **)(gap->ga_data))[i]) + sep_len;
Bram Moolenaar80a7dcf2010-08-04 17:07:20 +02002099
2100 s = alloc(len + 1);
2101 if (s != NULL)
2102 {
2103 *s = NUL;
Bram Moolenaar1b1063a2014-05-07 18:35:30 +02002104 p = s;
Bram Moolenaar80a7dcf2010-08-04 17:07:20 +02002105 for (i = 0; i < gap->ga_len; ++i)
2106 {
Bram Moolenaar1b1063a2014-05-07 18:35:30 +02002107 if (p != s)
2108 {
2109 STRCPY(p, sep);
2110 p += sep_len;
2111 }
2112 STRCPY(p, ((char_u **)(gap->ga_data))[i]);
2113 p += STRLEN(p);
Bram Moolenaar80a7dcf2010-08-04 17:07:20 +02002114 }
2115 }
2116 return s;
2117}
Bram Moolenaar113e1072019-01-20 15:30:40 +01002118#endif
Bram Moolenaar80a7dcf2010-08-04 17:07:20 +02002119
Bram Moolenaar5a66dfb2017-03-01 20:40:39 +01002120#if defined(FEAT_VIMINFO) || defined(FEAT_EVAL) || defined(PROTO)
Bram Moolenaarb20e3342016-01-18 23:29:01 +01002121/*
2122 * Make a copy of string "p" and add it to "gap".
2123 * When out of memory nothing changes.
2124 */
2125 void
2126ga_add_string(garray_T *gap, char_u *p)
2127{
2128 char_u *cp = vim_strsave(p);
2129
2130 if (cp != NULL)
2131 {
2132 if (ga_grow(gap, 1) == OK)
2133 ((char_u **)(gap->ga_data))[gap->ga_len++] = cp;
2134 else
2135 vim_free(cp);
2136 }
2137}
2138#endif
2139
Bram Moolenaar80a7dcf2010-08-04 17:07:20 +02002140/*
Bram Moolenaar29ddebe2019-01-26 17:28:26 +01002141 * Concatenate a string to a growarray which contains bytes.
Bram Moolenaar43345542015-11-29 17:35:35 +01002142 * When "s" is NULL does not do anything.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002143 * Note: Does NOT copy the NUL at the end!
2144 */
2145 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01002146ga_concat(garray_T *gap, char_u *s)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002147{
Bram Moolenaar43345542015-11-29 17:35:35 +01002148 int len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002149
Bram Moolenaar478af672017-04-10 22:22:42 +02002150 if (s == NULL || *s == NUL)
Bram Moolenaar43345542015-11-29 17:35:35 +01002151 return;
2152 len = (int)STRLEN(s);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002153 if (ga_grow(gap, len) == OK)
2154 {
2155 mch_memmove((char *)gap->ga_data + gap->ga_len, s, (size_t)len);
2156 gap->ga_len += len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002157 }
2158}
2159
2160/*
2161 * Append one byte to a growarray which contains bytes.
2162 */
2163 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01002164ga_append(garray_T *gap, int c)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002165{
2166 if (ga_grow(gap, 1) == OK)
2167 {
2168 *((char *)gap->ga_data + gap->ga_len) = c;
2169 ++gap->ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002170 }
2171}
2172
Bram Moolenaar4f974752019-02-17 17:44:42 +01002173#if (defined(UNIX) && !defined(USE_SYSTEM)) || defined(MSWIN) \
Bram Moolenaar597a4222014-06-25 14:39:50 +02002174 || defined(PROTO)
Bram Moolenaar4b9669f2011-07-07 16:20:52 +02002175/*
2176 * Append the text in "gap" below the cursor line and clear "gap".
2177 */
2178 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01002179append_ga_line(garray_T *gap)
Bram Moolenaar4b9669f2011-07-07 16:20:52 +02002180{
2181 /* Remove trailing CR. */
2182 if (gap->ga_len > 0
2183 && !curbuf->b_p_bin
2184 && ((char_u *)gap->ga_data)[gap->ga_len - 1] == CAR)
2185 --gap->ga_len;
2186 ga_append(gap, NUL);
2187 ml_append(curwin->w_cursor.lnum++, gap->ga_data, 0, FALSE);
2188 gap->ga_len = 0;
2189}
2190#endif
2191
Bram Moolenaar071d4272004-06-13 20:20:40 +00002192/************************************************************************
2193 * functions that use lookup tables for various things, generally to do with
2194 * special key codes.
2195 */
2196
2197/*
2198 * Some useful tables.
2199 */
2200
2201static struct modmasktable
2202{
2203 short mod_mask; /* Bit-mask for particular key modifier */
2204 short mod_flag; /* Bit(s) for particular key modifier */
2205 char_u name; /* Single letter name of modifier */
2206} mod_mask_table[] =
2207{
2208 {MOD_MASK_ALT, MOD_MASK_ALT, (char_u)'M'},
Bram Moolenaar19a09a12005-03-04 23:39:37 +00002209 {MOD_MASK_META, MOD_MASK_META, (char_u)'T'},
Bram Moolenaar071d4272004-06-13 20:20:40 +00002210 {MOD_MASK_CTRL, MOD_MASK_CTRL, (char_u)'C'},
2211 {MOD_MASK_SHIFT, MOD_MASK_SHIFT, (char_u)'S'},
2212 {MOD_MASK_MULTI_CLICK, MOD_MASK_2CLICK, (char_u)'2'},
2213 {MOD_MASK_MULTI_CLICK, MOD_MASK_3CLICK, (char_u)'3'},
2214 {MOD_MASK_MULTI_CLICK, MOD_MASK_4CLICK, (char_u)'4'},
Bram Moolenaard0573012017-10-28 21:11:06 +02002215#ifdef MACOS_X
Bram Moolenaar071d4272004-06-13 20:20:40 +00002216 {MOD_MASK_CMD, MOD_MASK_CMD, (char_u)'D'},
2217#endif
2218 /* 'A' must be the last one */
2219 {MOD_MASK_ALT, MOD_MASK_ALT, (char_u)'A'},
2220 {0, 0, NUL}
Bram Moolenaar423977d2017-01-22 15:05:12 +01002221 /* NOTE: when adding an entry, update MAX_KEY_NAME_LEN! */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002222};
2223
2224/*
2225 * Shifted key terminal codes and their unshifted equivalent.
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00002226 * Don't add mouse codes here, they are handled separately!
Bram Moolenaar071d4272004-06-13 20:20:40 +00002227 */
2228#define MOD_KEYS_ENTRY_SIZE 5
2229
2230static char_u modifier_keys_table[] =
2231{
2232/* mod mask with modifier without modifier */
2233 MOD_MASK_SHIFT, '&', '9', '@', '1', /* begin */
2234 MOD_MASK_SHIFT, '&', '0', '@', '2', /* cancel */
2235 MOD_MASK_SHIFT, '*', '1', '@', '4', /* command */
2236 MOD_MASK_SHIFT, '*', '2', '@', '5', /* copy */
2237 MOD_MASK_SHIFT, '*', '3', '@', '6', /* create */
2238 MOD_MASK_SHIFT, '*', '4', 'k', 'D', /* delete char */
2239 MOD_MASK_SHIFT, '*', '5', 'k', 'L', /* delete line */
2240 MOD_MASK_SHIFT, '*', '7', '@', '7', /* end */
2241 MOD_MASK_CTRL, KS_EXTRA, (int)KE_C_END, '@', '7', /* end */
2242 MOD_MASK_SHIFT, '*', '9', '@', '9', /* exit */
2243 MOD_MASK_SHIFT, '*', '0', '@', '0', /* find */
2244 MOD_MASK_SHIFT, '#', '1', '%', '1', /* help */
2245 MOD_MASK_SHIFT, '#', '2', 'k', 'h', /* home */
2246 MOD_MASK_CTRL, KS_EXTRA, (int)KE_C_HOME, 'k', 'h', /* home */
2247 MOD_MASK_SHIFT, '#', '3', 'k', 'I', /* insert */
2248 MOD_MASK_SHIFT, '#', '4', 'k', 'l', /* left arrow */
2249 MOD_MASK_CTRL, KS_EXTRA, (int)KE_C_LEFT, 'k', 'l', /* left arrow */
2250 MOD_MASK_SHIFT, '%', 'a', '%', '3', /* message */
2251 MOD_MASK_SHIFT, '%', 'b', '%', '4', /* move */
2252 MOD_MASK_SHIFT, '%', 'c', '%', '5', /* next */
2253 MOD_MASK_SHIFT, '%', 'd', '%', '7', /* options */
2254 MOD_MASK_SHIFT, '%', 'e', '%', '8', /* previous */
2255 MOD_MASK_SHIFT, '%', 'f', '%', '9', /* print */
2256 MOD_MASK_SHIFT, '%', 'g', '%', '0', /* redo */
2257 MOD_MASK_SHIFT, '%', 'h', '&', '3', /* replace */
2258 MOD_MASK_SHIFT, '%', 'i', 'k', 'r', /* right arr. */
2259 MOD_MASK_CTRL, KS_EXTRA, (int)KE_C_RIGHT, 'k', 'r', /* right arr. */
2260 MOD_MASK_SHIFT, '%', 'j', '&', '5', /* resume */
2261 MOD_MASK_SHIFT, '!', '1', '&', '6', /* save */
2262 MOD_MASK_SHIFT, '!', '2', '&', '7', /* suspend */
2263 MOD_MASK_SHIFT, '!', '3', '&', '8', /* undo */
2264 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_UP, 'k', 'u', /* up arrow */
2265 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_DOWN, 'k', 'd', /* down arrow */
2266
2267 /* vt100 F1 */
2268 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_XF1, KS_EXTRA, (int)KE_XF1,
2269 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_XF2, KS_EXTRA, (int)KE_XF2,
2270 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_XF3, KS_EXTRA, (int)KE_XF3,
2271 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_XF4, KS_EXTRA, (int)KE_XF4,
2272
2273 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F1, 'k', '1', /* F1 */
2274 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F2, 'k', '2',
2275 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F3, 'k', '3',
2276 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F4, 'k', '4',
2277 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F5, 'k', '5',
2278 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F6, 'k', '6',
2279 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F7, 'k', '7',
2280 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F8, 'k', '8',
2281 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F9, 'k', '9',
2282 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F10, 'k', ';', /* F10 */
2283
2284 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F11, 'F', '1',
2285 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F12, 'F', '2',
2286 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F13, 'F', '3',
2287 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F14, 'F', '4',
2288 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F15, 'F', '5',
2289 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F16, 'F', '6',
2290 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F17, 'F', '7',
2291 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F18, 'F', '8',
2292 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F19, 'F', '9',
2293 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F20, 'F', 'A',
2294
2295 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F21, 'F', 'B',
2296 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F22, 'F', 'C',
2297 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F23, 'F', 'D',
2298 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F24, 'F', 'E',
2299 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F25, 'F', 'F',
2300 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F26, 'F', 'G',
2301 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F27, 'F', 'H',
2302 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F28, 'F', 'I',
2303 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F29, 'F', 'J',
2304 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F30, 'F', 'K',
2305
2306 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F31, 'F', 'L',
2307 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F32, 'F', 'M',
2308 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F33, 'F', 'N',
2309 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F34, 'F', 'O',
2310 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F35, 'F', 'P',
2311 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F36, 'F', 'Q',
2312 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F37, 'F', 'R',
2313
2314 /* TAB pseudo code*/
2315 MOD_MASK_SHIFT, 'k', 'B', KS_EXTRA, (int)KE_TAB,
2316
2317 NUL
2318};
2319
2320static struct key_name_entry
2321{
2322 int key; /* Special key code or ascii value */
2323 char_u *name; /* Name of key */
2324} key_names_table[] =
2325{
2326 {' ', (char_u *)"Space"},
2327 {TAB, (char_u *)"Tab"},
2328 {K_TAB, (char_u *)"Tab"},
2329 {NL, (char_u *)"NL"},
2330 {NL, (char_u *)"NewLine"}, /* Alternative name */
2331 {NL, (char_u *)"LineFeed"}, /* Alternative name */
2332 {NL, (char_u *)"LF"}, /* Alternative name */
2333 {CAR, (char_u *)"CR"},
2334 {CAR, (char_u *)"Return"}, /* Alternative name */
2335 {CAR, (char_u *)"Enter"}, /* Alternative name */
2336 {K_BS, (char_u *)"BS"},
2337 {K_BS, (char_u *)"BackSpace"}, /* Alternative name */
2338 {ESC, (char_u *)"Esc"},
2339 {CSI, (char_u *)"CSI"},
2340 {K_CSI, (char_u *)"xCSI"},
2341 {'|', (char_u *)"Bar"},
2342 {'\\', (char_u *)"Bslash"},
2343 {K_DEL, (char_u *)"Del"},
2344 {K_DEL, (char_u *)"Delete"}, /* Alternative name */
2345 {K_KDEL, (char_u *)"kDel"},
2346 {K_UP, (char_u *)"Up"},
2347 {K_DOWN, (char_u *)"Down"},
2348 {K_LEFT, (char_u *)"Left"},
2349 {K_RIGHT, (char_u *)"Right"},
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002350 {K_XUP, (char_u *)"xUp"},
2351 {K_XDOWN, (char_u *)"xDown"},
2352 {K_XLEFT, (char_u *)"xLeft"},
2353 {K_XRIGHT, (char_u *)"xRight"},
Bram Moolenaarec2da362017-01-21 20:04:22 +01002354 {K_PS, (char_u *)"PasteStart"},
2355 {K_PE, (char_u *)"PasteEnd"},
Bram Moolenaar071d4272004-06-13 20:20:40 +00002356
2357 {K_F1, (char_u *)"F1"},
2358 {K_F2, (char_u *)"F2"},
2359 {K_F3, (char_u *)"F3"},
2360 {K_F4, (char_u *)"F4"},
2361 {K_F5, (char_u *)"F5"},
2362 {K_F6, (char_u *)"F6"},
2363 {K_F7, (char_u *)"F7"},
2364 {K_F8, (char_u *)"F8"},
2365 {K_F9, (char_u *)"F9"},
2366 {K_F10, (char_u *)"F10"},
2367
2368 {K_F11, (char_u *)"F11"},
2369 {K_F12, (char_u *)"F12"},
2370 {K_F13, (char_u *)"F13"},
2371 {K_F14, (char_u *)"F14"},
2372 {K_F15, (char_u *)"F15"},
2373 {K_F16, (char_u *)"F16"},
2374 {K_F17, (char_u *)"F17"},
2375 {K_F18, (char_u *)"F18"},
2376 {K_F19, (char_u *)"F19"},
2377 {K_F20, (char_u *)"F20"},
2378
2379 {K_F21, (char_u *)"F21"},
2380 {K_F22, (char_u *)"F22"},
2381 {K_F23, (char_u *)"F23"},
2382 {K_F24, (char_u *)"F24"},
2383 {K_F25, (char_u *)"F25"},
2384 {K_F26, (char_u *)"F26"},
2385 {K_F27, (char_u *)"F27"},
2386 {K_F28, (char_u *)"F28"},
2387 {K_F29, (char_u *)"F29"},
2388 {K_F30, (char_u *)"F30"},
2389
2390 {K_F31, (char_u *)"F31"},
2391 {K_F32, (char_u *)"F32"},
2392 {K_F33, (char_u *)"F33"},
2393 {K_F34, (char_u *)"F34"},
2394 {K_F35, (char_u *)"F35"},
2395 {K_F36, (char_u *)"F36"},
2396 {K_F37, (char_u *)"F37"},
2397
2398 {K_XF1, (char_u *)"xF1"},
2399 {K_XF2, (char_u *)"xF2"},
2400 {K_XF3, (char_u *)"xF3"},
2401 {K_XF4, (char_u *)"xF4"},
2402
2403 {K_HELP, (char_u *)"Help"},
2404 {K_UNDO, (char_u *)"Undo"},
2405 {K_INS, (char_u *)"Insert"},
2406 {K_INS, (char_u *)"Ins"}, /* Alternative name */
2407 {K_KINS, (char_u *)"kInsert"},
2408 {K_HOME, (char_u *)"Home"},
2409 {K_KHOME, (char_u *)"kHome"},
2410 {K_XHOME, (char_u *)"xHome"},
Bram Moolenaar68b76a62005-03-25 21:53:48 +00002411 {K_ZHOME, (char_u *)"zHome"},
Bram Moolenaar071d4272004-06-13 20:20:40 +00002412 {K_END, (char_u *)"End"},
2413 {K_KEND, (char_u *)"kEnd"},
2414 {K_XEND, (char_u *)"xEnd"},
Bram Moolenaar68b76a62005-03-25 21:53:48 +00002415 {K_ZEND, (char_u *)"zEnd"},
Bram Moolenaar071d4272004-06-13 20:20:40 +00002416 {K_PAGEUP, (char_u *)"PageUp"},
2417 {K_PAGEDOWN, (char_u *)"PageDown"},
2418 {K_KPAGEUP, (char_u *)"kPageUp"},
2419 {K_KPAGEDOWN, (char_u *)"kPageDown"},
2420
2421 {K_KPLUS, (char_u *)"kPlus"},
2422 {K_KMINUS, (char_u *)"kMinus"},
2423 {K_KDIVIDE, (char_u *)"kDivide"},
2424 {K_KMULTIPLY, (char_u *)"kMultiply"},
2425 {K_KENTER, (char_u *)"kEnter"},
2426 {K_KPOINT, (char_u *)"kPoint"},
2427
2428 {K_K0, (char_u *)"k0"},
2429 {K_K1, (char_u *)"k1"},
2430 {K_K2, (char_u *)"k2"},
2431 {K_K3, (char_u *)"k3"},
2432 {K_K4, (char_u *)"k4"},
2433 {K_K5, (char_u *)"k5"},
2434 {K_K6, (char_u *)"k6"},
2435 {K_K7, (char_u *)"k7"},
2436 {K_K8, (char_u *)"k8"},
2437 {K_K9, (char_u *)"k9"},
2438
2439 {'<', (char_u *)"lt"},
2440
2441 {K_MOUSE, (char_u *)"Mouse"},
Bram Moolenaar5af7d712012-01-20 17:15:51 +01002442#ifdef FEAT_MOUSE_NET
Bram Moolenaar071d4272004-06-13 20:20:40 +00002443 {K_NETTERM_MOUSE, (char_u *)"NetMouse"},
Bram Moolenaar5af7d712012-01-20 17:15:51 +01002444#endif
2445#ifdef FEAT_MOUSE_DEC
Bram Moolenaar071d4272004-06-13 20:20:40 +00002446 {K_DEC_MOUSE, (char_u *)"DecMouse"},
Bram Moolenaar5af7d712012-01-20 17:15:51 +01002447#endif
2448#ifdef FEAT_MOUSE_JSB
Bram Moolenaar071d4272004-06-13 20:20:40 +00002449 {K_JSBTERM_MOUSE, (char_u *)"JsbMouse"},
Bram Moolenaar5af7d712012-01-20 17:15:51 +01002450#endif
2451#ifdef FEAT_MOUSE_PTERM
Bram Moolenaar071d4272004-06-13 20:20:40 +00002452 {K_PTERM_MOUSE, (char_u *)"PtermMouse"},
Bram Moolenaar5af7d712012-01-20 17:15:51 +01002453#endif
2454#ifdef FEAT_MOUSE_URXVT
2455 {K_URXVT_MOUSE, (char_u *)"UrxvtMouse"},
2456#endif
Bram Moolenaar2b9578f2012-08-15 16:21:32 +02002457 {K_SGR_MOUSE, (char_u *)"SgrMouse"},
Bram Moolenaara529ce02017-06-22 22:37:57 +02002458 {K_SGR_MOUSERELEASE, (char_u *)"SgrMouseRelelase"},
Bram Moolenaar071d4272004-06-13 20:20:40 +00002459 {K_LEFTMOUSE, (char_u *)"LeftMouse"},
2460 {K_LEFTMOUSE_NM, (char_u *)"LeftMouseNM"},
2461 {K_LEFTDRAG, (char_u *)"LeftDrag"},
2462 {K_LEFTRELEASE, (char_u *)"LeftRelease"},
2463 {K_LEFTRELEASE_NM, (char_u *)"LeftReleaseNM"},
Bram Moolenaar51b0f372017-11-18 18:52:04 +01002464 {K_MOUSEMOVE, (char_u *)"MouseMove"},
Bram Moolenaar071d4272004-06-13 20:20:40 +00002465 {K_MIDDLEMOUSE, (char_u *)"MiddleMouse"},
2466 {K_MIDDLEDRAG, (char_u *)"MiddleDrag"},
2467 {K_MIDDLERELEASE, (char_u *)"MiddleRelease"},
2468 {K_RIGHTMOUSE, (char_u *)"RightMouse"},
2469 {K_RIGHTDRAG, (char_u *)"RightDrag"},
2470 {K_RIGHTRELEASE, (char_u *)"RightRelease"},
Bram Moolenaar8d9b40e2010-07-25 15:49:07 +02002471 {K_MOUSEDOWN, (char_u *)"ScrollWheelUp"},
2472 {K_MOUSEUP, (char_u *)"ScrollWheelDown"},
2473 {K_MOUSELEFT, (char_u *)"ScrollWheelRight"},
2474 {K_MOUSERIGHT, (char_u *)"ScrollWheelLeft"},
2475 {K_MOUSEDOWN, (char_u *)"MouseDown"}, /* OBSOLETE: Use */
2476 {K_MOUSEUP, (char_u *)"MouseUp"}, /* ScrollWheelXXX instead */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002477 {K_X1MOUSE, (char_u *)"X1Mouse"},
2478 {K_X1DRAG, (char_u *)"X1Drag"},
2479 {K_X1RELEASE, (char_u *)"X1Release"},
2480 {K_X2MOUSE, (char_u *)"X2Mouse"},
2481 {K_X2DRAG, (char_u *)"X2Drag"},
2482 {K_X2RELEASE, (char_u *)"X2Release"},
2483 {K_DROP, (char_u *)"Drop"},
2484 {K_ZERO, (char_u *)"Nul"},
2485#ifdef FEAT_EVAL
2486 {K_SNR, (char_u *)"SNR"},
2487#endif
2488 {K_PLUG, (char_u *)"Plug"},
Bram Moolenaar1db60c42014-09-23 16:49:46 +02002489 {K_CURSORHOLD, (char_u *)"CursorHold"},
Bram Moolenaar2f106582019-05-08 21:59:25 +02002490 {K_IGNORE, (char_u *)"Ignore"},
Bram Moolenaar071d4272004-06-13 20:20:40 +00002491 {0, NULL}
Bram Moolenaar423977d2017-01-22 15:05:12 +01002492 /* NOTE: When adding a long name update MAX_KEY_NAME_LEN. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002493};
2494
2495#define KEY_NAMES_TABLE_LEN (sizeof(key_names_table) / sizeof(struct key_name_entry))
2496
2497#ifdef FEAT_MOUSE
2498static struct mousetable
2499{
2500 int pseudo_code; /* Code for pseudo mouse event */
2501 int button; /* Which mouse button is it? */
2502 int is_click; /* Is it a mouse button click event? */
2503 int is_drag; /* Is it a mouse drag event? */
2504} mouse_table[] =
2505{
2506 {(int)KE_LEFTMOUSE, MOUSE_LEFT, TRUE, FALSE},
2507#ifdef FEAT_GUI
2508 {(int)KE_LEFTMOUSE_NM, MOUSE_LEFT, TRUE, FALSE},
2509#endif
2510 {(int)KE_LEFTDRAG, MOUSE_LEFT, FALSE, TRUE},
2511 {(int)KE_LEFTRELEASE, MOUSE_LEFT, FALSE, FALSE},
2512#ifdef FEAT_GUI
2513 {(int)KE_LEFTRELEASE_NM, MOUSE_LEFT, FALSE, FALSE},
2514#endif
2515 {(int)KE_MIDDLEMOUSE, MOUSE_MIDDLE, TRUE, FALSE},
2516 {(int)KE_MIDDLEDRAG, MOUSE_MIDDLE, FALSE, TRUE},
2517 {(int)KE_MIDDLERELEASE, MOUSE_MIDDLE, FALSE, FALSE},
2518 {(int)KE_RIGHTMOUSE, MOUSE_RIGHT, TRUE, FALSE},
2519 {(int)KE_RIGHTDRAG, MOUSE_RIGHT, FALSE, TRUE},
2520 {(int)KE_RIGHTRELEASE, MOUSE_RIGHT, FALSE, FALSE},
2521 {(int)KE_X1MOUSE, MOUSE_X1, TRUE, FALSE},
2522 {(int)KE_X1DRAG, MOUSE_X1, FALSE, TRUE},
2523 {(int)KE_X1RELEASE, MOUSE_X1, FALSE, FALSE},
2524 {(int)KE_X2MOUSE, MOUSE_X2, TRUE, FALSE},
2525 {(int)KE_X2DRAG, MOUSE_X2, FALSE, TRUE},
2526 {(int)KE_X2RELEASE, MOUSE_X2, FALSE, FALSE},
2527 /* DRAG without CLICK */
Bram Moolenaar51b0f372017-11-18 18:52:04 +01002528 {(int)KE_MOUSEMOVE, MOUSE_RELEASE, FALSE, TRUE},
Bram Moolenaar071d4272004-06-13 20:20:40 +00002529 /* RELEASE without CLICK */
2530 {(int)KE_IGNORE, MOUSE_RELEASE, FALSE, FALSE},
2531 {0, 0, 0, 0},
2532};
2533#endif /* FEAT_MOUSE */
2534
2535/*
2536 * Return the modifier mask bit (MOD_MASK_*) which corresponds to the given
2537 * modifier name ('S' for Shift, 'C' for Ctrl etc).
2538 */
2539 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01002540name_to_mod_mask(int c)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002541{
2542 int i;
2543
2544 c = TOUPPER_ASC(c);
2545 for (i = 0; mod_mask_table[i].mod_mask != 0; i++)
2546 if (c == mod_mask_table[i].name)
2547 return mod_mask_table[i].mod_flag;
2548 return 0;
2549}
2550
Bram Moolenaar071d4272004-06-13 20:20:40 +00002551/*
2552 * Check if if there is a special key code for "key" that includes the
2553 * modifiers specified.
2554 */
2555 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01002556simplify_key(int key, int *modifiers)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002557{
2558 int i;
2559 int key0;
2560 int key1;
2561
2562 if (*modifiers & (MOD_MASK_SHIFT | MOD_MASK_CTRL | MOD_MASK_ALT))
2563 {
2564 /* TAB is a special case */
2565 if (key == TAB && (*modifiers & MOD_MASK_SHIFT))
2566 {
2567 *modifiers &= ~MOD_MASK_SHIFT;
2568 return K_S_TAB;
2569 }
2570 key0 = KEY2TERMCAP0(key);
2571 key1 = KEY2TERMCAP1(key);
2572 for (i = 0; modifier_keys_table[i] != NUL; i += MOD_KEYS_ENTRY_SIZE)
2573 if (key0 == modifier_keys_table[i + 3]
2574 && key1 == modifier_keys_table[i + 4]
2575 && (*modifiers & modifier_keys_table[i]))
2576 {
2577 *modifiers &= ~modifier_keys_table[i];
2578 return TERMCAP2KEY(modifier_keys_table[i + 1],
2579 modifier_keys_table[i + 2]);
2580 }
2581 }
2582 return key;
2583}
2584
2585/*
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002586 * Change <xHome> to <Home>, <xUp> to <Up>, etc.
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002587 */
2588 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01002589handle_x_keys(int key)
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002590{
2591 switch (key)
2592 {
2593 case K_XUP: return K_UP;
2594 case K_XDOWN: return K_DOWN;
2595 case K_XLEFT: return K_LEFT;
2596 case K_XRIGHT: return K_RIGHT;
2597 case K_XHOME: return K_HOME;
Bram Moolenaar68b76a62005-03-25 21:53:48 +00002598 case K_ZHOME: return K_HOME;
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002599 case K_XEND: return K_END;
Bram Moolenaar68b76a62005-03-25 21:53:48 +00002600 case K_ZEND: return K_END;
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002601 case K_XF1: return K_F1;
2602 case K_XF2: return K_F2;
2603 case K_XF3: return K_F3;
2604 case K_XF4: return K_F4;
2605 case K_S_XF1: return K_S_F1;
2606 case K_S_XF2: return K_S_F2;
2607 case K_S_XF3: return K_S_F3;
2608 case K_S_XF4: return K_S_F4;
2609 }
2610 return key;
2611}
2612
2613/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002614 * Return a string which contains the name of the given key when the given
2615 * modifiers are down.
2616 */
2617 char_u *
Bram Moolenaar9b578142016-01-30 19:39:49 +01002618get_special_key_name(int c, int modifiers)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002619{
2620 static char_u string[MAX_KEY_NAME_LEN + 1];
2621
2622 int i, idx;
2623 int table_idx;
2624 char_u *s;
2625
2626 string[0] = '<';
2627 idx = 1;
2628
2629 /* Key that stands for a normal character. */
2630 if (IS_SPECIAL(c) && KEY2TERMCAP0(c) == KS_KEY)
2631 c = KEY2TERMCAP1(c);
2632
2633 /*
2634 * Translate shifted special keys into unshifted keys and set modifier.
2635 * Same for CTRL and ALT modifiers.
2636 */
2637 if (IS_SPECIAL(c))
2638 {
2639 for (i = 0; modifier_keys_table[i] != 0; i += MOD_KEYS_ENTRY_SIZE)
2640 if ( KEY2TERMCAP0(c) == (int)modifier_keys_table[i + 1]
2641 && (int)KEY2TERMCAP1(c) == (int)modifier_keys_table[i + 2])
2642 {
2643 modifiers |= modifier_keys_table[i];
2644 c = TERMCAP2KEY(modifier_keys_table[i + 3],
2645 modifier_keys_table[i + 4]);
2646 break;
2647 }
2648 }
2649
2650 /* try to find the key in the special key table */
2651 table_idx = find_special_key_in_table(c);
2652
2653 /*
2654 * When not a known special key, and not a printable character, try to
2655 * extract modifiers.
2656 */
Bram Moolenaarfc3abf42019-01-24 15:54:21 +01002657 if (c > 0 && (*mb_char2len)(c) == 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002658 {
2659 if (table_idx < 0
2660 && (!vim_isprintc(c) || (c & 0x7f) == ' ')
2661 && (c & 0x80))
2662 {
2663 c &= 0x7f;
2664 modifiers |= MOD_MASK_ALT;
2665 /* try again, to find the un-alted key in the special key table */
2666 table_idx = find_special_key_in_table(c);
2667 }
2668 if (table_idx < 0 && !vim_isprintc(c) && c < ' ')
2669 {
2670#ifdef EBCDIC
2671 c = CtrlChar(c);
2672#else
2673 c += '@';
2674#endif
2675 modifiers |= MOD_MASK_CTRL;
2676 }
2677 }
2678
2679 /* translate the modifier into a string */
2680 for (i = 0; mod_mask_table[i].name != 'A'; i++)
2681 if ((modifiers & mod_mask_table[i].mod_mask)
2682 == mod_mask_table[i].mod_flag)
2683 {
2684 string[idx++] = mod_mask_table[i].name;
2685 string[idx++] = (char_u)'-';
2686 }
2687
2688 if (table_idx < 0) /* unknown special key, may output t_xx */
2689 {
2690 if (IS_SPECIAL(c))
2691 {
2692 string[idx++] = 't';
2693 string[idx++] = '_';
2694 string[idx++] = KEY2TERMCAP0(c);
2695 string[idx++] = KEY2TERMCAP1(c);
2696 }
2697 /* Not a special key, only modifiers, output directly */
2698 else
2699 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00002700 if (has_mbyte && (*mb_char2len)(c) > 1)
2701 idx += (*mb_char2bytes)(c, string + idx);
Bram Moolenaarfc3abf42019-01-24 15:54:21 +01002702 else if (vim_isprintc(c))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002703 string[idx++] = c;
2704 else
2705 {
2706 s = transchar(c);
2707 while (*s)
2708 string[idx++] = *s++;
2709 }
2710 }
2711 }
2712 else /* use name of special key */
2713 {
Bram Moolenaar423977d2017-01-22 15:05:12 +01002714 size_t len = STRLEN(key_names_table[table_idx].name);
2715
2716 if (len + idx + 2 <= MAX_KEY_NAME_LEN)
2717 {
2718 STRCPY(string + idx, key_names_table[table_idx].name);
2719 idx += (int)len;
2720 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002721 }
2722 string[idx++] = '>';
2723 string[idx] = NUL;
2724 return string;
2725}
2726
2727/*
2728 * Try translating a <> name at (*srcp)[] to dst[].
2729 * Return the number of characters added to dst[], zero for no match.
2730 * If there is a match, srcp is advanced to after the <> name.
2731 * dst[] must be big enough to hold the result (up to six characters)!
2732 */
2733 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01002734trans_special(
2735 char_u **srcp,
2736 char_u *dst,
Bram Moolenaarbf0eff02019-06-01 17:13:36 +02002737 int keycode, // prefer key code, e.g. K_DEL instead of DEL
2738 int in_string) // TRUE when inside a double quoted string
Bram Moolenaar071d4272004-06-13 20:20:40 +00002739{
2740 int modifiers = 0;
2741 int key;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002742
Bram Moolenaar35a4cfa2016-08-14 16:07:48 +02002743 key = find_special_key(srcp, &modifiers, keycode, FALSE, in_string);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002744 if (key == 0)
2745 return 0;
2746
Bram Moolenaarbf0eff02019-06-01 17:13:36 +02002747 return special_to_buf(key, modifiers, keycode, dst);
2748}
2749
2750/*
2751 * Put the character sequence for "key" with "modifiers" into "dst" and return
2752 * the resulting length.
2753 * When "keycode" is TRUE prefer key code, e.g. K_DEL instead of DEL.
2754 * The sequence is not NUL terminated.
2755 * This is how characters in a string are encoded.
2756 */
2757 int
2758special_to_buf(int key, int modifiers, int keycode, char_u *dst)
2759{
2760 int dlen = 0;
2761
Bram Moolenaar071d4272004-06-13 20:20:40 +00002762 /* Put the appropriate modifier in a string */
2763 if (modifiers != 0)
2764 {
2765 dst[dlen++] = K_SPECIAL;
2766 dst[dlen++] = KS_MODIFIER;
2767 dst[dlen++] = modifiers;
2768 }
2769
2770 if (IS_SPECIAL(key))
2771 {
2772 dst[dlen++] = K_SPECIAL;
2773 dst[dlen++] = KEY2TERMCAP0(key);
2774 dst[dlen++] = KEY2TERMCAP1(key);
2775 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002776 else if (has_mbyte && !keycode)
2777 dlen += (*mb_char2bytes)(key, dst + dlen);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002778 else if (keycode)
2779 dlen = (int)(add_char2buf(key, dst + dlen) - dst);
2780 else
2781 dst[dlen++] = key;
2782
2783 return dlen;
2784}
2785
2786/*
2787 * Try translating a <> name at (*srcp)[], return the key and modifiers.
2788 * srcp is advanced to after the <> name.
2789 * returns 0 if there is no match.
2790 */
2791 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01002792find_special_key(
2793 char_u **srcp,
2794 int *modp,
2795 int keycode, /* prefer key code, e.g. K_DEL instead of DEL */
Bram Moolenaar35a4cfa2016-08-14 16:07:48 +02002796 int keep_x_key, /* don't translate xHome to Home key */
2797 int in_string) /* TRUE in string, double quote is escaped */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002798{
2799 char_u *last_dash;
2800 char_u *end_of_name;
2801 char_u *src;
2802 char_u *bp;
2803 int modifiers;
2804 int bit;
2805 int key;
Bram Moolenaar22fcfad2016-07-01 18:17:26 +02002806 uvarnumber_T n;
Bram Moolenaarb8bf5412011-08-17 20:33:22 +02002807 int l;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002808
2809 src = *srcp;
2810 if (src[0] != '<')
2811 return 0;
2812
2813 /* Find end of modifier list */
2814 last_dash = src;
2815 for (bp = src + 1; *bp == '-' || vim_isIDc(*bp); bp++)
2816 {
2817 if (*bp == '-')
2818 {
2819 last_dash = bp;
Bram Moolenaarb8bf5412011-08-17 20:33:22 +02002820 if (bp[1] != NUL)
2821 {
Bram Moolenaarb8bf5412011-08-17 20:33:22 +02002822 if (has_mbyte)
2823 l = mb_ptr2len(bp + 1);
2824 else
Bram Moolenaarb8bf5412011-08-17 20:33:22 +02002825 l = 1;
Bram Moolenaar35a4cfa2016-08-14 16:07:48 +02002826 /* Anything accepted, like <C-?>.
2827 * <C-"> or <M-"> are not special in strings as " is
2828 * the string delimiter. With a backslash it works: <M-\"> */
2829 if (!(in_string && bp[1] == '"') && bp[2] == '>')
Bram Moolenaar1d90a5a2016-07-01 11:59:47 +02002830 bp += l;
Bram Moolenaar35a4cfa2016-08-14 16:07:48 +02002831 else if (in_string && bp[1] == '\\' && bp[2] == '"'
2832 && bp[3] == '>')
2833 bp += 2;
Bram Moolenaarb8bf5412011-08-17 20:33:22 +02002834 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002835 }
2836 if (bp[0] == 't' && bp[1] == '_' && bp[2] && bp[3])
2837 bp += 3; /* skip t_xx, xx may be '-' or '>' */
Bram Moolenaar792826c2011-08-19 22:29:02 +02002838 else if (STRNICMP(bp, "char-", 5) == 0)
2839 {
Bram Moolenaar16e9b852019-05-19 19:59:35 +02002840 vim_str2nr(bp + 5, NULL, &l, STR2NR_ALL, NULL, NULL, 0, TRUE);
2841 if (l == 0)
2842 {
2843 emsg(_(e_invarg));
2844 return 0;
2845 }
Bram Moolenaar792826c2011-08-19 22:29:02 +02002846 bp += l + 5;
2847 break;
2848 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002849 }
2850
2851 if (*bp == '>') /* found matching '>' */
2852 {
2853 end_of_name = bp + 1;
2854
Bram Moolenaar071d4272004-06-13 20:20:40 +00002855 /* Which modifiers are given? */
2856 modifiers = 0x0;
2857 for (bp = src + 1; bp < last_dash; bp++)
2858 {
2859 if (*bp != '-')
2860 {
2861 bit = name_to_mod_mask(*bp);
2862 if (bit == 0x0)
2863 break; /* Illegal modifier name */
2864 modifiers |= bit;
2865 }
2866 }
2867
2868 /*
2869 * Legal modifier name.
2870 */
2871 if (bp >= last_dash)
2872 {
Bram Moolenaarb8bf5412011-08-17 20:33:22 +02002873 if (STRNICMP(last_dash + 1, "char-", 5) == 0
2874 && VIM_ISDIGIT(last_dash[6]))
2875 {
2876 /* <Char-123> or <Char-033> or <Char-0x33> */
Bram Moolenaar16e9b852019-05-19 19:59:35 +02002877 vim_str2nr(last_dash + 6, NULL, &l, STR2NR_ALL, NULL, &n, 0, TRUE);
2878 if (l == 0)
2879 {
2880 emsg(_(e_invarg));
2881 return 0;
2882 }
Bram Moolenaar792826c2011-08-19 22:29:02 +02002883 key = (int)n;
Bram Moolenaarb8bf5412011-08-17 20:33:22 +02002884 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002885 else
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002886 {
Bram Moolenaar35a4cfa2016-08-14 16:07:48 +02002887 int off = 1;
2888
2889 /* Modifier with single letter, or special key name. */
2890 if (in_string && last_dash[1] == '\\' && last_dash[2] == '"')
2891 off = 2;
Bram Moolenaar792826c2011-08-19 22:29:02 +02002892 if (has_mbyte)
Bram Moolenaar35a4cfa2016-08-14 16:07:48 +02002893 l = mb_ptr2len(last_dash + off);
Bram Moolenaar792826c2011-08-19 22:29:02 +02002894 else
Bram Moolenaar792826c2011-08-19 22:29:02 +02002895 l = 1;
Bram Moolenaar35a4cfa2016-08-14 16:07:48 +02002896 if (modifiers != 0 && last_dash[l + off] == '>')
2897 key = PTR2CHAR(last_dash + off);
Bram Moolenaar792826c2011-08-19 22:29:02 +02002898 else
2899 {
Bram Moolenaar35a4cfa2016-08-14 16:07:48 +02002900 key = get_special_key_code(last_dash + off);
Bram Moolenaar792826c2011-08-19 22:29:02 +02002901 if (!keep_x_key)
2902 key = handle_x_keys(key);
2903 }
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002904 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002905
2906 /*
2907 * get_special_key_code() may return NUL for invalid
2908 * special key name.
2909 */
2910 if (key != NUL)
2911 {
2912 /*
2913 * Only use a modifier when there is no special key code that
2914 * includes the modifier.
2915 */
2916 key = simplify_key(key, &modifiers);
2917
2918 if (!keycode)
2919 {
2920 /* don't want keycode, use single byte code */
2921 if (key == K_BS)
2922 key = BS;
2923 else if (key == K_DEL || key == K_KDEL)
2924 key = DEL;
2925 }
2926
2927 /*
2928 * Normal Key with modifier: Try to make a single byte code.
2929 */
2930 if (!IS_SPECIAL(key))
2931 key = extract_modifiers(key, &modifiers);
2932
2933 *modp = modifiers;
2934 *srcp = end_of_name;
2935 return key;
2936 }
2937 }
2938 }
2939 return 0;
2940}
2941
2942/*
2943 * Try to include modifiers in the key.
2944 * Changes "Shift-a" to 'A', "Alt-A" to 0xc0, etc.
2945 */
2946 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01002947extract_modifiers(int key, int *modp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002948{
2949 int modifiers = *modp;
2950
Bram Moolenaard0573012017-10-28 21:11:06 +02002951#ifdef MACOS_X
Bram Moolenaar84a05ac2013-05-06 04:24:17 +02002952 /* Command-key really special, no fancynest */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002953 if (!(modifiers & MOD_MASK_CMD))
2954#endif
2955 if ((modifiers & MOD_MASK_SHIFT) && ASCII_ISALPHA(key))
2956 {
2957 key = TOUPPER_ASC(key);
2958 modifiers &= ~MOD_MASK_SHIFT;
2959 }
2960 if ((modifiers & MOD_MASK_CTRL)
2961#ifdef EBCDIC
2962 /* * TODO: EBCDIC Better use:
2963 * && (Ctrl_chr(key) || key == '?')
2964 * ??? */
2965 && strchr("?@ABCDEFGHIJKLMNOPQRSTUVWXYZ[\\]^_", key)
2966 != NULL
2967#else
2968 && ((key >= '?' && key <= '_') || ASCII_ISALPHA(key))
2969#endif
2970 )
2971 {
2972 key = Ctrl_chr(key);
2973 modifiers &= ~MOD_MASK_CTRL;
2974 /* <C-@> is <Nul> */
2975 if (key == 0)
2976 key = K_ZERO;
2977 }
Bram Moolenaard0573012017-10-28 21:11:06 +02002978#ifdef MACOS_X
Bram Moolenaar84a05ac2013-05-06 04:24:17 +02002979 /* Command-key really special, no fancynest */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002980 if (!(modifiers & MOD_MASK_CMD))
2981#endif
2982 if ((modifiers & MOD_MASK_ALT) && key < 0x80
Bram Moolenaarfc3abf42019-01-24 15:54:21 +01002983 && !enc_dbcs) // avoid creating a lead byte
Bram Moolenaar071d4272004-06-13 20:20:40 +00002984 {
2985 key |= 0x80;
2986 modifiers &= ~MOD_MASK_ALT; /* remove the META modifier */
2987 }
2988
2989 *modp = modifiers;
2990 return key;
2991}
2992
2993/*
2994 * Try to find key "c" in the special key table.
2995 * Return the index when found, -1 when not found.
2996 */
2997 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01002998find_special_key_in_table(int c)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002999{
3000 int i;
3001
3002 for (i = 0; key_names_table[i].name != NULL; i++)
3003 if (c == key_names_table[i].key)
3004 break;
3005 if (key_names_table[i].name == NULL)
3006 i = -1;
3007 return i;
3008}
3009
3010/*
3011 * Find the special key with the given name (the given string does not have to
3012 * end with NUL, the name is assumed to end before the first non-idchar).
3013 * If the name starts with "t_" the next two characters are interpreted as a
3014 * termcap name.
3015 * Return the key code, or 0 if not found.
3016 */
3017 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01003018get_special_key_code(char_u *name)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003019{
3020 char_u *table_name;
3021 char_u string[3];
3022 int i, j;
3023
3024 /*
3025 * If it's <t_xx> we get the code for xx from the termcap
3026 */
3027 if (name[0] == 't' && name[1] == '_' && name[2] != NUL && name[3] != NUL)
3028 {
3029 string[0] = name[2];
3030 string[1] = name[3];
3031 string[2] = NUL;
3032 if (add_termcap_entry(string, FALSE) == OK)
3033 return TERMCAP2KEY(name[2], name[3]);
3034 }
3035 else
3036 for (i = 0; key_names_table[i].name != NULL; i++)
3037 {
3038 table_name = key_names_table[i].name;
3039 for (j = 0; vim_isIDc(name[j]) && table_name[j] != NUL; j++)
3040 if (TOLOWER_ASC(table_name[j]) != TOLOWER_ASC(name[j]))
3041 break;
3042 if (!vim_isIDc(name[j]) && table_name[j] == NUL)
3043 return key_names_table[i].key;
3044 }
3045 return 0;
3046}
3047
Bram Moolenaar05bb9532008-07-04 09:44:11 +00003048#if defined(FEAT_CMDL_COMPL) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003049 char_u *
Bram Moolenaar9b578142016-01-30 19:39:49 +01003050get_key_name(int i)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003051{
Bram Moolenaare1fbddc2009-05-16 19:07:03 +00003052 if (i >= (int)KEY_NAMES_TABLE_LEN)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003053 return NULL;
3054 return key_names_table[i].name;
3055}
3056#endif
3057
Bram Moolenaar05bb9532008-07-04 09:44:11 +00003058#if defined(FEAT_MOUSE) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003059/*
3060 * Look up the given mouse code to return the relevant information in the other
3061 * arguments. Return which button is down or was released.
3062 */
3063 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01003064get_mouse_button(int code, int *is_click, int *is_drag)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003065{
3066 int i;
3067
3068 for (i = 0; mouse_table[i].pseudo_code; i++)
3069 if (code == mouse_table[i].pseudo_code)
3070 {
3071 *is_click = mouse_table[i].is_click;
3072 *is_drag = mouse_table[i].is_drag;
3073 return mouse_table[i].button;
3074 }
3075 return 0; /* Shouldn't get here */
3076}
3077
3078/*
3079 * Return the appropriate pseudo mouse event token (KE_LEFTMOUSE etc) based on
3080 * the given information about which mouse button is down, and whether the
3081 * mouse was clicked, dragged or released.
3082 */
3083 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01003084get_pseudo_mouse_code(
3085 int button, /* eg MOUSE_LEFT */
3086 int is_click,
3087 int is_drag)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003088{
3089 int i;
3090
3091 for (i = 0; mouse_table[i].pseudo_code; i++)
3092 if (button == mouse_table[i].button
3093 && is_click == mouse_table[i].is_click
3094 && is_drag == mouse_table[i].is_drag)
3095 {
3096#ifdef FEAT_GUI
Bram Moolenaarc91506a2005-04-24 22:04:21 +00003097 /* Trick: a non mappable left click and release has mouse_col -1
3098 * or added MOUSE_COLOFF. Used for 'mousefocus' in
3099 * gui_mouse_moved() */
3100 if (mouse_col < 0 || mouse_col > MOUSE_COLOFF)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003101 {
Bram Moolenaarc91506a2005-04-24 22:04:21 +00003102 if (mouse_col < 0)
3103 mouse_col = 0;
3104 else
3105 mouse_col -= MOUSE_COLOFF;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003106 if (mouse_table[i].pseudo_code == (int)KE_LEFTMOUSE)
3107 return (int)KE_LEFTMOUSE_NM;
3108 if (mouse_table[i].pseudo_code == (int)KE_LEFTRELEASE)
3109 return (int)KE_LEFTRELEASE_NM;
3110 }
3111#endif
3112 return mouse_table[i].pseudo_code;
3113 }
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00003114 return (int)KE_IGNORE; /* not recognized, ignore it */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003115}
3116#endif /* FEAT_MOUSE */
3117
3118/*
3119 * Return the current end-of-line type: EOL_DOS, EOL_UNIX or EOL_MAC.
3120 */
3121 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01003122get_fileformat(buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003123{
3124 int c = *buf->b_p_ff;
3125
3126 if (buf->b_p_bin || c == 'u')
3127 return EOL_UNIX;
3128 if (c == 'm')
3129 return EOL_MAC;
3130 return EOL_DOS;
3131}
3132
3133/*
3134 * Like get_fileformat(), but override 'fileformat' with "p" for "++opt=val"
3135 * argument.
3136 */
3137 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01003138get_fileformat_force(
3139 buf_T *buf,
3140 exarg_T *eap) /* can be NULL! */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003141{
3142 int c;
3143
3144 if (eap != NULL && eap->force_ff != 0)
Bram Moolenaar333b80a2018-04-04 22:57:29 +02003145 c = eap->force_ff;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003146 else
3147 {
3148 if ((eap != NULL && eap->force_bin != 0)
3149 ? (eap->force_bin == FORCE_BIN) : buf->b_p_bin)
3150 return EOL_UNIX;
3151 c = *buf->b_p_ff;
3152 }
3153 if (c == 'u')
3154 return EOL_UNIX;
3155 if (c == 'm')
3156 return EOL_MAC;
3157 return EOL_DOS;
3158}
3159
3160/*
3161 * Set the current end-of-line type to EOL_DOS, EOL_UNIX or EOL_MAC.
3162 * Sets both 'textmode' and 'fileformat'.
3163 * Note: Does _not_ set global value of 'textmode'!
3164 */
3165 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01003166set_fileformat(
3167 int t,
3168 int opt_flags) /* OPT_LOCAL and/or OPT_GLOBAL */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003169{
3170 char *p = NULL;
3171
3172 switch (t)
3173 {
3174 case EOL_DOS:
3175 p = FF_DOS;
3176 curbuf->b_p_tx = TRUE;
3177 break;
3178 case EOL_UNIX:
3179 p = FF_UNIX;
3180 curbuf->b_p_tx = FALSE;
3181 break;
3182 case EOL_MAC:
3183 p = FF_MAC;
3184 curbuf->b_p_tx = FALSE;
3185 break;
3186 }
3187 if (p != NULL)
3188 set_string_option_direct((char_u *)"ff", -1, (char_u *)p,
Bram Moolenaar5e3cb7e2006-02-27 23:58:35 +00003189 OPT_FREE | opt_flags, 0);
3190
Bram Moolenaarf740b292006-02-16 22:11:02 +00003191 /* This may cause the buffer to become (un)modified. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003192 check_status(curbuf);
Bram Moolenaar997fb4b2006-02-17 21:53:23 +00003193 redraw_tabline = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003194#ifdef FEAT_TITLE
3195 need_maketitle = TRUE; /* set window title later */
3196#endif
3197}
3198
3199/*
3200 * Return the default fileformat from 'fileformats'.
3201 */
3202 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01003203default_fileformat(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003204{
3205 switch (*p_ffs)
3206 {
3207 case 'm': return EOL_MAC;
3208 case 'd': return EOL_DOS;
3209 }
3210 return EOL_UNIX;
3211}
3212
3213/*
3214 * Call shell. Calls mch_call_shell, with 'shellxquote' added.
3215 */
3216 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01003217call_shell(char_u *cmd, int opt)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003218{
3219 char_u *ncmd;
3220 int retval;
Bram Moolenaar05159a02005-02-26 23:04:13 +00003221#ifdef FEAT_PROFILE
3222 proftime_T wait_time;
3223#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003224
3225 if (p_verbose > 3)
3226 {
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00003227 verbose_enter();
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003228 smsg(_("Calling shell to execute: \"%s\""),
Bram Moolenaar071d4272004-06-13 20:20:40 +00003229 cmd == NULL ? p_sh : cmd);
3230 out_char('\n');
3231 cursor_on();
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00003232 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003233 }
3234
Bram Moolenaar05159a02005-02-26 23:04:13 +00003235#ifdef FEAT_PROFILE
Bram Moolenaar01265852006-03-20 21:50:15 +00003236 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003237 prof_child_enter(&wait_time);
3238#endif
3239
Bram Moolenaar071d4272004-06-13 20:20:40 +00003240 if (*p_sh == NUL)
3241 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003242 emsg(_(e_shellempty));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003243 retval = -1;
3244 }
3245 else
3246 {
3247#ifdef FEAT_GUI_MSWIN
3248 /* Don't hide the pointer while executing a shell command. */
3249 gui_mch_mousehide(FALSE);
3250#endif
3251#ifdef FEAT_GUI
3252 ++hold_gui_events;
3253#endif
3254 /* The external command may update a tags file, clear cached tags. */
3255 tag_freematch();
3256
Bram Moolenaar01257a72019-06-10 14:46:04 +02003257 if (cmd == NULL || *p_sxq == NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003258 retval = mch_call_shell(cmd, opt);
3259 else
3260 {
Bram Moolenaarf66b3fc2012-02-20 22:18:30 +01003261 char_u *ecmd = cmd;
3262
3263 if (*p_sxe != NUL && STRCMP(p_sxq, "(") == 0)
3264 {
3265 ecmd = vim_strsave_escaped_ext(cmd, p_sxe, '^', FALSE);
3266 if (ecmd == NULL)
3267 ecmd = cmd;
3268 }
Bram Moolenaar964b3742019-05-24 18:54:09 +02003269 ncmd = alloc(STRLEN(ecmd) + STRLEN(p_sxq) * 2 + 1);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003270 if (ncmd != NULL)
3271 {
3272 STRCPY(ncmd, p_sxq);
Bram Moolenaarf66b3fc2012-02-20 22:18:30 +01003273 STRCAT(ncmd, ecmd);
Bram Moolenaar034b1152012-02-19 18:19:30 +01003274 /* When 'shellxquote' is ( append ).
3275 * When 'shellxquote' is "( append )". */
3276 STRCAT(ncmd, STRCMP(p_sxq, "(") == 0 ? (char_u *)")"
3277 : STRCMP(p_sxq, "\"(") == 0 ? (char_u *)")\""
3278 : p_sxq);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003279 retval = mch_call_shell(ncmd, opt);
3280 vim_free(ncmd);
3281 }
3282 else
3283 retval = -1;
Bram Moolenaarf66b3fc2012-02-20 22:18:30 +01003284 if (ecmd != cmd)
3285 vim_free(ecmd);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003286 }
3287#ifdef FEAT_GUI
3288 --hold_gui_events;
3289#endif
3290 /*
3291 * Check the window size, in case it changed while executing the
3292 * external command.
3293 */
3294 shell_resized_check();
3295 }
3296
3297#ifdef FEAT_EVAL
3298 set_vim_var_nr(VV_SHELL_ERROR, (long)retval);
Bram Moolenaar05159a02005-02-26 23:04:13 +00003299# ifdef FEAT_PROFILE
Bram Moolenaar01265852006-03-20 21:50:15 +00003300 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003301 prof_child_exit(&wait_time);
3302# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003303#endif
3304
3305 return retval;
3306}
3307
3308/*
Bram Moolenaar01265852006-03-20 21:50:15 +00003309 * VISUAL, SELECTMODE and OP_PENDING State are never set, they are equal to
3310 * NORMAL State with a condition. This function returns the real State.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003311 */
3312 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01003313get_real_state(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003314{
3315 if (State & NORMAL)
3316 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00003317 if (VIsual_active)
Bram Moolenaar01265852006-03-20 21:50:15 +00003318 {
3319 if (VIsual_select)
3320 return SELECTMODE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003321 return VISUAL;
Bram Moolenaar01265852006-03-20 21:50:15 +00003322 }
Bram Moolenaarf7ff6e82014-03-23 15:13:05 +01003323 else if (finish_op)
3324 return OP_PENDING;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003325 }
3326 return State;
3327}
3328
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003329/*
3330 * Return TRUE if "p" points to just after a path separator.
Bram Moolenaarb5ce04d2011-07-07 17:15:33 +02003331 * Takes care of multi-byte characters.
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003332 * "b" must point to the start of the file name
3333 */
3334 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01003335after_pathsep(char_u *b, char_u *p)
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003336{
Bram Moolenaarb5ce04d2011-07-07 17:15:33 +02003337 return p > b && vim_ispathsep(p[-1])
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003338 && (!has_mbyte || (*mb_head_off)(b, p - 1) == 0);
3339}
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003340
3341/*
3342 * Return TRUE if file names "f1" and "f2" are in the same directory.
3343 * "f1" may be a short name, "f2" must be a full path.
3344 */
3345 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01003346same_directory(char_u *f1, char_u *f2)
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003347{
3348 char_u ffname[MAXPATHL];
3349 char_u *t1;
3350 char_u *t2;
3351
3352 /* safety check */
3353 if (f1 == NULL || f2 == NULL)
3354 return FALSE;
3355
3356 (void)vim_FullName(f1, ffname, MAXPATHL, FALSE);
3357 t1 = gettail_sep(ffname);
3358 t2 = gettail_sep(f2);
3359 return (t1 - ffname == t2 - f2
3360 && pathcmp((char *)ffname, (char *)f2, (int)(t1 - ffname)) == 0);
3361}
3362
Bram Moolenaar7c365fb2018-06-29 20:28:31 +02003363#if defined(FEAT_SESSION) || defined(FEAT_AUTOCHDIR) \
3364 || defined(MSWIN) || defined(FEAT_GUI_MAC) || defined(FEAT_GUI_GTK) \
Bram Moolenaarbb1969b2019-01-17 15:45:25 +01003365 || defined(FEAT_NETBEANS_INTG) \
Bram Moolenaar071d4272004-06-13 20:20:40 +00003366 || defined(PROTO)
3367/*
3368 * Change to a file's directory.
3369 * Caller must call shorten_fnames()!
3370 * Return OK or FAIL.
3371 */
3372 int
Bram Moolenaar2caad3f2018-12-16 15:38:02 +01003373vim_chdirfile(char_u *fname, char *trigger_autocmd)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003374{
Bram Moolenaar2caad3f2018-12-16 15:38:02 +01003375 char_u old_dir[MAXPATHL];
3376 char_u new_dir[MAXPATHL];
Bram Moolenaarb5cb65b2018-02-03 18:01:37 +01003377 int res;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003378
Bram Moolenaar2caad3f2018-12-16 15:38:02 +01003379 if (mch_dirname(old_dir, MAXPATHL) != OK)
3380 *old_dir = NUL;
3381
3382 vim_strncpy(new_dir, fname, MAXPATHL - 1);
3383 *gettail_sep(new_dir) = NUL;
3384
Bram Moolenaar9eb76af2018-12-16 16:30:21 +01003385 if (pathcmp((char *)old_dir, (char *)new_dir, -1) == 0)
Bram Moolenaar2caad3f2018-12-16 15:38:02 +01003386 // nothing to do
3387 res = OK;
3388 else
3389 {
3390 res = mch_chdir((char *)new_dir) == 0 ? OK : FAIL;
3391
3392 if (res == OK && trigger_autocmd != NULL)
3393 apply_autocmds(EVENT_DIRCHANGED, (char_u *)trigger_autocmd,
3394 new_dir, FALSE, curbuf);
3395 }
Bram Moolenaarb5cb65b2018-02-03 18:01:37 +01003396 return res;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003397}
3398#endif
3399
3400#if defined(STAT_IGNORES_SLASH) || defined(PROTO)
3401/*
3402 * Check if "name" ends in a slash and is not a directory.
3403 * Used for systems where stat() ignores a trailing slash on a file name.
3404 * The Vim code assumes a trailing slash is only ignored for a directory.
3405 */
Bram Moolenaar91acfff2017-03-12 19:22:36 +01003406 static int
Bram Moolenaard8492792017-03-16 12:22:38 +01003407illegal_slash(const char *name)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003408{
3409 if (name[0] == NUL)
3410 return FALSE; /* no file name is not illegal */
3411 if (name[strlen(name) - 1] != '/')
3412 return FALSE; /* no trailing slash */
3413 if (mch_isdir((char_u *)name))
3414 return FALSE; /* trailing slash for a directory */
3415 return TRUE;
3416}
Bram Moolenaar91acfff2017-03-12 19:22:36 +01003417
3418/*
3419 * Special implementation of mch_stat() for Solaris.
3420 */
3421 int
3422vim_stat(const char *name, stat_T *stp)
3423{
3424 /* On Solaris stat() accepts "file/" as if it was "file". Return -1 if
3425 * the name ends in "/" and it's not a directory. */
Bram Moolenaard8492792017-03-16 12:22:38 +01003426 return illegal_slash(name) ? -1 : stat(name, stp);
Bram Moolenaar91acfff2017-03-12 19:22:36 +01003427}
Bram Moolenaar071d4272004-06-13 20:20:40 +00003428#endif
3429
3430#if defined(CURSOR_SHAPE) || defined(PROTO)
3431
3432/*
3433 * Handling of cursor and mouse pointer shapes in various modes.
3434 */
3435
3436cursorentry_T shape_table[SHAPE_IDX_COUNT] =
3437{
3438 /* The values will be filled in from the 'guicursor' and 'mouseshape'
3439 * defaults when Vim starts.
3440 * Adjust the SHAPE_IDX_ defines when making changes! */
3441 {0, 0, 0, 700L, 400L, 250L, 0, 0, "n", SHAPE_CURSOR+SHAPE_MOUSE},
3442 {0, 0, 0, 700L, 400L, 250L, 0, 0, "v", SHAPE_CURSOR+SHAPE_MOUSE},
3443 {0, 0, 0, 700L, 400L, 250L, 0, 0, "i", SHAPE_CURSOR+SHAPE_MOUSE},
3444 {0, 0, 0, 700L, 400L, 250L, 0, 0, "r", SHAPE_CURSOR+SHAPE_MOUSE},
3445 {0, 0, 0, 700L, 400L, 250L, 0, 0, "c", SHAPE_CURSOR+SHAPE_MOUSE},
3446 {0, 0, 0, 700L, 400L, 250L, 0, 0, "ci", SHAPE_CURSOR+SHAPE_MOUSE},
3447 {0, 0, 0, 700L, 400L, 250L, 0, 0, "cr", SHAPE_CURSOR+SHAPE_MOUSE},
3448 {0, 0, 0, 700L, 400L, 250L, 0, 0, "o", SHAPE_CURSOR+SHAPE_MOUSE},
3449 {0, 0, 0, 700L, 400L, 250L, 0, 0, "ve", SHAPE_CURSOR+SHAPE_MOUSE},
3450 {0, 0, 0, 0L, 0L, 0L, 0, 0, "e", SHAPE_MOUSE},
3451 {0, 0, 0, 0L, 0L, 0L, 0, 0, "s", SHAPE_MOUSE},
3452 {0, 0, 0, 0L, 0L, 0L, 0, 0, "sd", SHAPE_MOUSE},
3453 {0, 0, 0, 0L, 0L, 0L, 0, 0, "vs", SHAPE_MOUSE},
3454 {0, 0, 0, 0L, 0L, 0L, 0, 0, "vd", SHAPE_MOUSE},
3455 {0, 0, 0, 0L, 0L, 0L, 0, 0, "m", SHAPE_MOUSE},
3456 {0, 0, 0, 0L, 0L, 0L, 0, 0, "ml", SHAPE_MOUSE},
3457 {0, 0, 0, 100L, 100L, 100L, 0, 0, "sm", SHAPE_CURSOR},
3458};
3459
3460#ifdef FEAT_MOUSESHAPE
3461/*
3462 * Table with names for mouse shapes. Keep in sync with all the tables for
3463 * mch_set_mouse_shape()!.
3464 */
3465static char * mshape_names[] =
3466{
3467 "arrow", /* default, must be the first one */
3468 "blank", /* hidden */
3469 "beam",
3470 "updown",
3471 "udsizing",
3472 "leftright",
3473 "lrsizing",
3474 "busy",
3475 "no",
3476 "crosshair",
3477 "hand1",
3478 "hand2",
3479 "pencil",
3480 "question",
3481 "rightup-arrow",
3482 "up-arrow",
3483 NULL
3484};
3485#endif
3486
3487/*
3488 * Parse the 'guicursor' option ("what" is SHAPE_CURSOR) or 'mouseshape'
3489 * ("what" is SHAPE_MOUSE).
3490 * Returns error message for an illegal option, NULL otherwise.
3491 */
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003492 char *
Bram Moolenaar9b578142016-01-30 19:39:49 +01003493parse_shape_opt(int what)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003494{
3495 char_u *modep;
3496 char_u *colonp;
3497 char_u *commap;
3498 char_u *slashp;
3499 char_u *p, *endp;
3500 int idx = 0; /* init for GCC */
3501 int all_idx;
3502 int len;
3503 int i;
3504 long n;
3505 int found_ve = FALSE; /* found "ve" flag */
3506 int round;
3507
3508 /*
3509 * First round: check for errors; second round: do it for real.
3510 */
3511 for (round = 1; round <= 2; ++round)
3512 {
3513 /*
3514 * Repeat for all comma separated parts.
3515 */
3516#ifdef FEAT_MOUSESHAPE
3517 if (what == SHAPE_MOUSE)
3518 modep = p_mouseshape;
3519 else
3520#endif
3521 modep = p_guicursor;
3522 while (*modep != NUL)
3523 {
3524 colonp = vim_strchr(modep, ':');
Bram Moolenaar24922ec2017-02-23 17:59:22 +01003525 commap = vim_strchr(modep, ',');
3526
3527 if (colonp == NULL || (commap != NULL && commap < colonp))
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003528 return N_("E545: Missing colon");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003529 if (colonp == modep)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003530 return N_("E546: Illegal mode");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003531
3532 /*
3533 * Repeat for all mode's before the colon.
3534 * For the 'a' mode, we loop to handle all the modes.
3535 */
3536 all_idx = -1;
3537 while (modep < colonp || all_idx >= 0)
3538 {
3539 if (all_idx < 0)
3540 {
3541 /* Find the mode. */
3542 if (modep[1] == '-' || modep[1] == ':')
3543 len = 1;
3544 else
3545 len = 2;
3546 if (len == 1 && TOLOWER_ASC(modep[0]) == 'a')
3547 all_idx = SHAPE_IDX_COUNT - 1;
3548 else
3549 {
3550 for (idx = 0; idx < SHAPE_IDX_COUNT; ++idx)
3551 if (STRNICMP(modep, shape_table[idx].name, len)
3552 == 0)
3553 break;
3554 if (idx == SHAPE_IDX_COUNT
3555 || (shape_table[idx].used_for & what) == 0)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003556 return N_("E546: Illegal mode");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003557 if (len == 2 && modep[0] == 'v' && modep[1] == 'e')
3558 found_ve = TRUE;
3559 }
3560 modep += len + 1;
3561 }
3562
3563 if (all_idx >= 0)
3564 idx = all_idx--;
3565 else if (round == 2)
3566 {
3567#ifdef FEAT_MOUSESHAPE
3568 if (what == SHAPE_MOUSE)
3569 {
3570 /* Set the default, for the missing parts */
3571 shape_table[idx].mshape = 0;
3572 }
3573 else
3574#endif
3575 {
3576 /* Set the defaults, for the missing parts */
3577 shape_table[idx].shape = SHAPE_BLOCK;
3578 shape_table[idx].blinkwait = 700L;
3579 shape_table[idx].blinkon = 400L;
3580 shape_table[idx].blinkoff = 250L;
3581 }
3582 }
3583
3584 /* Parse the part after the colon */
3585 for (p = colonp + 1; *p && *p != ','; )
3586 {
3587#ifdef FEAT_MOUSESHAPE
3588 if (what == SHAPE_MOUSE)
3589 {
3590 for (i = 0; ; ++i)
3591 {
3592 if (mshape_names[i] == NULL)
3593 {
3594 if (!VIM_ISDIGIT(*p))
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003595 return N_("E547: Illegal mouseshape");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003596 if (round == 2)
3597 shape_table[idx].mshape =
3598 getdigits(&p) + MSHAPE_NUMBERED;
3599 else
3600 (void)getdigits(&p);
3601 break;
3602 }
3603 len = (int)STRLEN(mshape_names[i]);
3604 if (STRNICMP(p, mshape_names[i], len) == 0)
3605 {
3606 if (round == 2)
3607 shape_table[idx].mshape = i;
3608 p += len;
3609 break;
3610 }
3611 }
3612 }
3613 else /* if (what == SHAPE_MOUSE) */
3614#endif
3615 {
3616 /*
3617 * First handle the ones with a number argument.
3618 */
3619 i = *p;
3620 len = 0;
3621 if (STRNICMP(p, "ver", 3) == 0)
3622 len = 3;
3623 else if (STRNICMP(p, "hor", 3) == 0)
3624 len = 3;
3625 else if (STRNICMP(p, "blinkwait", 9) == 0)
3626 len = 9;
3627 else if (STRNICMP(p, "blinkon", 7) == 0)
3628 len = 7;
3629 else if (STRNICMP(p, "blinkoff", 8) == 0)
3630 len = 8;
3631 if (len != 0)
3632 {
3633 p += len;
3634 if (!VIM_ISDIGIT(*p))
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003635 return N_("E548: digit expected");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003636 n = getdigits(&p);
3637 if (len == 3) /* "ver" or "hor" */
3638 {
3639 if (n == 0)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003640 return N_("E549: Illegal percentage");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003641 if (round == 2)
3642 {
3643 if (TOLOWER_ASC(i) == 'v')
3644 shape_table[idx].shape = SHAPE_VER;
3645 else
3646 shape_table[idx].shape = SHAPE_HOR;
3647 shape_table[idx].percentage = n;
3648 }
3649 }
3650 else if (round == 2)
3651 {
3652 if (len == 9)
3653 shape_table[idx].blinkwait = n;
3654 else if (len == 7)
3655 shape_table[idx].blinkon = n;
3656 else
3657 shape_table[idx].blinkoff = n;
3658 }
3659 }
3660 else if (STRNICMP(p, "block", 5) == 0)
3661 {
3662 if (round == 2)
3663 shape_table[idx].shape = SHAPE_BLOCK;
3664 p += 5;
3665 }
3666 else /* must be a highlight group name then */
3667 {
3668 endp = vim_strchr(p, '-');
3669 if (commap == NULL) /* last part */
3670 {
3671 if (endp == NULL)
3672 endp = p + STRLEN(p); /* find end of part */
3673 }
3674 else if (endp > commap || endp == NULL)
3675 endp = commap;
3676 slashp = vim_strchr(p, '/');
3677 if (slashp != NULL && slashp < endp)
3678 {
3679 /* "group/langmap_group" */
3680 i = syn_check_group(p, (int)(slashp - p));
3681 p = slashp + 1;
3682 }
3683 if (round == 2)
3684 {
3685 shape_table[idx].id = syn_check_group(p,
3686 (int)(endp - p));
3687 shape_table[idx].id_lm = shape_table[idx].id;
3688 if (slashp != NULL && slashp < endp)
3689 shape_table[idx].id = i;
3690 }
3691 p = endp;
3692 }
3693 } /* if (what != SHAPE_MOUSE) */
3694
3695 if (*p == '-')
3696 ++p;
3697 }
3698 }
3699 modep = p;
3700 if (*modep == ',')
3701 ++modep;
3702 }
3703 }
3704
3705 /* If the 's' flag is not given, use the 'v' cursor for 's' */
3706 if (!found_ve)
3707 {
3708#ifdef FEAT_MOUSESHAPE
3709 if (what == SHAPE_MOUSE)
3710 {
3711 shape_table[SHAPE_IDX_VE].mshape = shape_table[SHAPE_IDX_V].mshape;
3712 }
3713 else
3714#endif
3715 {
3716 shape_table[SHAPE_IDX_VE].shape = shape_table[SHAPE_IDX_V].shape;
3717 shape_table[SHAPE_IDX_VE].percentage =
3718 shape_table[SHAPE_IDX_V].percentage;
3719 shape_table[SHAPE_IDX_VE].blinkwait =
3720 shape_table[SHAPE_IDX_V].blinkwait;
3721 shape_table[SHAPE_IDX_VE].blinkon =
3722 shape_table[SHAPE_IDX_V].blinkon;
3723 shape_table[SHAPE_IDX_VE].blinkoff =
3724 shape_table[SHAPE_IDX_V].blinkoff;
3725 shape_table[SHAPE_IDX_VE].id = shape_table[SHAPE_IDX_V].id;
3726 shape_table[SHAPE_IDX_VE].id_lm = shape_table[SHAPE_IDX_V].id_lm;
3727 }
3728 }
3729
3730 return NULL;
3731}
3732
Bram Moolenaarac6e65f2005-08-29 22:25:38 +00003733# if defined(MCH_CURSOR_SHAPE) || defined(FEAT_GUI) \
3734 || defined(FEAT_MOUSESHAPE) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003735/*
3736 * Return the index into shape_table[] for the current mode.
3737 * When "mouse" is TRUE, consider indexes valid for the mouse pointer.
3738 */
3739 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01003740get_shape_idx(int mouse)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003741{
3742#ifdef FEAT_MOUSESHAPE
3743 if (mouse && (State == HITRETURN || State == ASKMORE))
3744 {
3745# ifdef FEAT_GUI
Bram Moolenaar9588a0f2005-01-08 21:45:39 +00003746 int x, y;
3747 gui_mch_getmouse(&x, &y);
3748 if (Y_2_ROW(y) == Rows - 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003749 return SHAPE_IDX_MOREL;
3750# endif
3751 return SHAPE_IDX_MORE;
3752 }
3753 if (mouse && drag_status_line)
3754 return SHAPE_IDX_SDRAG;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003755 if (mouse && drag_sep_line)
3756 return SHAPE_IDX_VDRAG;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003757#endif
3758 if (!mouse && State == SHOWMATCH)
3759 return SHAPE_IDX_SM;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003760 if (State & VREPLACE_FLAG)
3761 return SHAPE_IDX_R;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003762 if (State & REPLACE_FLAG)
3763 return SHAPE_IDX_R;
3764 if (State & INSERT)
3765 return SHAPE_IDX_I;
3766 if (State & CMDLINE)
3767 {
3768 if (cmdline_at_end())
3769 return SHAPE_IDX_C;
3770 if (cmdline_overstrike())
3771 return SHAPE_IDX_CR;
3772 return SHAPE_IDX_CI;
3773 }
3774 if (finish_op)
3775 return SHAPE_IDX_O;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003776 if (VIsual_active)
3777 {
3778 if (*p_sel == 'e')
3779 return SHAPE_IDX_VE;
3780 else
3781 return SHAPE_IDX_V;
3782 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003783 return SHAPE_IDX_N;
3784}
Bram Moolenaarac6e65f2005-08-29 22:25:38 +00003785#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003786
3787# if defined(FEAT_MOUSESHAPE) || defined(PROTO)
3788static int old_mouse_shape = 0;
3789
3790/*
3791 * Set the mouse shape:
3792 * If "shape" is -1, use shape depending on the current mode,
3793 * depending on the current state.
3794 * If "shape" is -2, only update the shape when it's CLINE or STATUS (used
3795 * when the mouse moves off the status or command line).
3796 */
3797 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01003798update_mouseshape(int shape_idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003799{
3800 int new_mouse_shape;
3801
3802 /* Only works in GUI mode. */
Bram Moolenaar6bb68362005-03-22 23:03:44 +00003803 if (!gui.in_use || gui.starting)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003804 return;
3805
3806 /* Postpone the updating when more is to come. Speeds up executing of
3807 * mappings. */
3808 if (shape_idx == -1 && char_avail())
3809 {
3810 postponed_mouseshape = TRUE;
3811 return;
3812 }
3813
Bram Moolenaar14716812006-05-04 21:54:08 +00003814 /* When ignoring the mouse don't change shape on the statusline. */
3815 if (*p_mouse == NUL
3816 && (shape_idx == SHAPE_IDX_CLINE
3817 || shape_idx == SHAPE_IDX_STATUS
3818 || shape_idx == SHAPE_IDX_VSEP))
3819 shape_idx = -2;
3820
Bram Moolenaar071d4272004-06-13 20:20:40 +00003821 if (shape_idx == -2
3822 && old_mouse_shape != shape_table[SHAPE_IDX_CLINE].mshape
3823 && old_mouse_shape != shape_table[SHAPE_IDX_STATUS].mshape
3824 && old_mouse_shape != shape_table[SHAPE_IDX_VSEP].mshape)
3825 return;
3826 if (shape_idx < 0)
3827 new_mouse_shape = shape_table[get_shape_idx(TRUE)].mshape;
3828 else
3829 new_mouse_shape = shape_table[shape_idx].mshape;
3830 if (new_mouse_shape != old_mouse_shape)
3831 {
3832 mch_set_mouse_shape(new_mouse_shape);
3833 old_mouse_shape = new_mouse_shape;
3834 }
3835 postponed_mouseshape = FALSE;
3836}
3837# endif
3838
3839#endif /* CURSOR_SHAPE */
3840
3841
Bram Moolenaar071d4272004-06-13 20:20:40 +00003842/*
3843 * Change directory to "new_dir". If FEAT_SEARCHPATH is defined, search
3844 * 'cdpath' for relative directory names, otherwise just mch_chdir().
3845 */
3846 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01003847vim_chdir(char_u *new_dir)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003848{
3849#ifndef FEAT_SEARCHPATH
3850 return mch_chdir((char *)new_dir);
3851#else
3852 char_u *dir_name;
3853 int r;
3854
3855 dir_name = find_directory_in_path(new_dir, (int)STRLEN(new_dir),
3856 FNAME_MESS, curbuf->b_ffname);
3857 if (dir_name == NULL)
3858 return -1;
3859 r = mch_chdir((char *)dir_name);
3860 vim_free(dir_name);
3861 return r;
3862#endif
3863}
3864
3865/*
Bram Moolenaarbbebc852005-07-18 21:47:53 +00003866 * Get user name from machine-specific function.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003867 * Returns the user name in "buf[len]".
Bram Moolenaarbbebc852005-07-18 21:47:53 +00003868 * Some systems are quite slow in obtaining the user name (Windows NT), thus
3869 * cache the result.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003870 * Returns OK or FAIL.
3871 */
3872 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01003873get_user_name(char_u *buf, int len)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003874{
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00003875 if (username == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003876 {
3877 if (mch_get_user_name(buf, len) == FAIL)
3878 return FAIL;
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00003879 username = vim_strsave(buf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003880 }
3881 else
Bram Moolenaarbbebc852005-07-18 21:47:53 +00003882 vim_strncpy(buf, username, len - 1);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003883 return OK;
3884}
3885
3886#ifndef HAVE_QSORT
3887/*
3888 * Our own qsort(), for systems that don't have it.
3889 * It's simple and slow. From the K&R C book.
3890 */
3891 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01003892qsort(
3893 void *base,
3894 size_t elm_count,
3895 size_t elm_size,
3896 int (*cmp)(const void *, const void *))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003897{
3898 char_u *buf;
3899 char_u *p1;
3900 char_u *p2;
3901 int i, j;
3902 int gap;
3903
Bram Moolenaar964b3742019-05-24 18:54:09 +02003904 buf = alloc(elm_size);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003905 if (buf == NULL)
3906 return;
3907
3908 for (gap = elm_count / 2; gap > 0; gap /= 2)
3909 for (i = gap; i < elm_count; ++i)
3910 for (j = i - gap; j >= 0; j -= gap)
3911 {
3912 /* Compare the elements. */
3913 p1 = (char_u *)base + j * elm_size;
3914 p2 = (char_u *)base + (j + gap) * elm_size;
3915 if ((*cmp)((void *)p1, (void *)p2) <= 0)
3916 break;
Bram Moolenaar21160b92009-11-11 15:56:10 +00003917 /* Exchange the elements. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003918 mch_memmove(buf, p1, elm_size);
3919 mch_memmove(p1, p2, elm_size);
3920 mch_memmove(p2, buf, elm_size);
3921 }
3922
3923 vim_free(buf);
3924}
3925#endif
3926
Bram Moolenaar071d4272004-06-13 20:20:40 +00003927/*
3928 * Sort an array of strings.
3929 */
Bram Moolenaareae1b912019-05-09 15:12:55 +02003930static int sort_compare(const void *s1, const void *s2);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003931
3932 static int
Bram Moolenaar9b578142016-01-30 19:39:49 +01003933sort_compare(const void *s1, const void *s2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003934{
3935 return STRCMP(*(char **)s1, *(char **)s2);
3936}
3937
3938 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01003939sort_strings(
3940 char_u **files,
3941 int count)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003942{
3943 qsort((void *)files, (size_t)count, sizeof(char_u *), sort_compare);
3944}
Bram Moolenaar071d4272004-06-13 20:20:40 +00003945
3946#if !defined(NO_EXPANDPATH) || defined(PROTO)
3947/*
3948 * Compare path "p[]" to "q[]".
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003949 * If "maxlen" >= 0 compare "p[maxlen]" to "q[maxlen]"
Bram Moolenaar071d4272004-06-13 20:20:40 +00003950 * Return value like strcmp(p, q), but consider path separators.
3951 */
3952 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01003953pathcmp(const char *p, const char *q, int maxlen)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003954{
Bram Moolenaarf6470c22015-08-25 16:31:40 +02003955 int i, j;
Bram Moolenaar38ec50b2013-04-12 14:42:39 +02003956 int c1, c2;
Bram Moolenaar86b68352004-12-27 21:59:20 +00003957 const char *s = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003958
Bram Moolenaarf6470c22015-08-25 16:31:40 +02003959 for (i = 0, j = 0; maxlen < 0 || (i < maxlen && j < maxlen);)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003960 {
Bram Moolenaar38ec50b2013-04-12 14:42:39 +02003961 c1 = PTR2CHAR((char_u *)p + i);
Bram Moolenaarf6470c22015-08-25 16:31:40 +02003962 c2 = PTR2CHAR((char_u *)q + j);
Bram Moolenaar38ec50b2013-04-12 14:42:39 +02003963
Bram Moolenaar071d4272004-06-13 20:20:40 +00003964 /* End of "p": check if "q" also ends or just has a slash. */
Bram Moolenaar38ec50b2013-04-12 14:42:39 +02003965 if (c1 == NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003966 {
Bram Moolenaar38ec50b2013-04-12 14:42:39 +02003967 if (c2 == NUL) /* full match */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003968 return 0;
3969 s = q;
Bram Moolenaar15675582018-02-09 12:29:56 +01003970 i = j;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003971 break;
3972 }
3973
3974 /* End of "q": check if "p" just has a slash. */
Bram Moolenaar38ec50b2013-04-12 14:42:39 +02003975 if (c2 == NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003976 {
3977 s = p;
3978 break;
3979 }
3980
Bram Moolenaar38ec50b2013-04-12 14:42:39 +02003981 if ((p_fic ? MB_TOUPPER(c1) != MB_TOUPPER(c2) : c1 != c2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003982#ifdef BACKSLASH_IN_FILENAME
3983 /* consider '/' and '\\' to be equal */
Bram Moolenaar38ec50b2013-04-12 14:42:39 +02003984 && !((c1 == '/' && c2 == '\\')
3985 || (c1 == '\\' && c2 == '/'))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003986#endif
3987 )
3988 {
Bram Moolenaar38ec50b2013-04-12 14:42:39 +02003989 if (vim_ispathsep(c1))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003990 return -1;
Bram Moolenaar38ec50b2013-04-12 14:42:39 +02003991 if (vim_ispathsep(c2))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003992 return 1;
Bram Moolenaar38ec50b2013-04-12 14:42:39 +02003993 return p_fic ? MB_TOUPPER(c1) - MB_TOUPPER(c2)
3994 : c1 - c2; /* no match */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003995 }
Bram Moolenaarf6470c22015-08-25 16:31:40 +02003996
3997 i += MB_PTR2LEN((char_u *)p + i);
3998 j += MB_PTR2LEN((char_u *)q + j);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003999 }
Bram Moolenaarf6470c22015-08-25 16:31:40 +02004000 if (s == NULL) /* "i" or "j" ran into "maxlen" */
Bram Moolenaar86b68352004-12-27 21:59:20 +00004001 return 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004002
Bram Moolenaar38ec50b2013-04-12 14:42:39 +02004003 c1 = PTR2CHAR((char_u *)s + i);
4004 c2 = PTR2CHAR((char_u *)s + i + MB_PTR2LEN((char_u *)s + i));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004005 /* ignore a trailing slash, but not "//" or ":/" */
Bram Moolenaar38ec50b2013-04-12 14:42:39 +02004006 if (c2 == NUL
Bram Moolenaar86b68352004-12-27 21:59:20 +00004007 && i > 0
4008 && !after_pathsep((char_u *)s, (char_u *)s + i)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004009#ifdef BACKSLASH_IN_FILENAME
Bram Moolenaar38ec50b2013-04-12 14:42:39 +02004010 && (c1 == '/' || c1 == '\\')
Bram Moolenaar071d4272004-06-13 20:20:40 +00004011#else
Bram Moolenaar38ec50b2013-04-12 14:42:39 +02004012 && c1 == '/'
Bram Moolenaar071d4272004-06-13 20:20:40 +00004013#endif
Bram Moolenaar86b68352004-12-27 21:59:20 +00004014 )
Bram Moolenaar071d4272004-06-13 20:20:40 +00004015 return 0; /* match with trailing slash */
4016 if (s == q)
4017 return -1; /* no match */
4018 return 1;
4019}
4020#endif
4021
Bram Moolenaar071d4272004-06-13 20:20:40 +00004022/*
4023 * The putenv() implementation below comes from the "screen" program.
4024 * Included with permission from Juergen Weigert.
4025 * See pty.c for the copyright notice.
4026 */
4027
4028/*
4029 * putenv -- put value into environment
4030 *
4031 * Usage: i = putenv (string)
4032 * int i;
4033 * char *string;
4034 *
4035 * where string is of the form <name>=<value>.
4036 * Putenv returns 0 normally, -1 on error (not enough core for malloc).
4037 *
4038 * Putenv may need to add a new name into the environment, or to
4039 * associate a value longer than the current value with a particular
4040 * name. So, to make life simpler, putenv() copies your entire
4041 * environment into the heap (i.e. malloc()) from the stack
4042 * (i.e. where it resides when your process is initiated) the first
4043 * time you call it.
4044 *
4045 * (history removed, not very interesting. See the "screen" sources.)
4046 */
4047
4048#if !defined(HAVE_SETENV) && !defined(HAVE_PUTENV)
4049
4050#define EXTRASIZE 5 /* increment to add to env. size */
4051
4052static int envsize = -1; /* current size of environment */
Bram Moolenaard0573012017-10-28 21:11:06 +02004053extern char **environ; /* the global which is your env. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004054
Bram Moolenaar92b8b2d2016-01-29 22:36:45 +01004055static int findenv(char *name); /* look for a name in the env. */
4056static int newenv(void); /* copy env. from stack to heap */
4057static int moreenv(void); /* incr. size of env. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004058
4059 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01004060putenv(const char *string)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004061{
4062 int i;
4063 char *p;
4064
4065 if (envsize < 0)
4066 { /* first time putenv called */
4067 if (newenv() < 0) /* copy env. to heap */
4068 return -1;
4069 }
4070
4071 i = findenv((char *)string); /* look for name in environment */
4072
4073 if (i < 0)
4074 { /* name must be added */
4075 for (i = 0; environ[i]; i++);
4076 if (i >= (envsize - 1))
4077 { /* need new slot */
4078 if (moreenv() < 0)
4079 return -1;
4080 }
Bram Moolenaarc799fe22019-05-28 23:08:19 +02004081 p = alloc(strlen(string) + 1);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004082 if (p == NULL) /* not enough core */
4083 return -1;
4084 environ[i + 1] = 0; /* new end of env. */
4085 }
4086 else
4087 { /* name already in env. */
4088 p = vim_realloc(environ[i], strlen(string) + 1);
4089 if (p == NULL)
4090 return -1;
4091 }
4092 sprintf(p, "%s", string); /* copy into env. */
4093 environ[i] = p;
4094
4095 return 0;
4096}
4097
4098 static int
Bram Moolenaar9b578142016-01-30 19:39:49 +01004099findenv(char *name)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004100{
4101 char *namechar, *envchar;
4102 int i, found;
4103
4104 found = 0;
4105 for (i = 0; environ[i] && !found; i++)
4106 {
4107 envchar = environ[i];
4108 namechar = name;
4109 while (*namechar && *namechar != '=' && (*namechar == *envchar))
4110 {
4111 namechar++;
4112 envchar++;
4113 }
4114 found = ((*namechar == '\0' || *namechar == '=') && *envchar == '=');
4115 }
4116 return found ? i - 1 : -1;
4117}
4118
4119 static int
Bram Moolenaar9b578142016-01-30 19:39:49 +01004120newenv(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004121{
4122 char **env, *elem;
4123 int i, esize;
4124
Bram Moolenaar071d4272004-06-13 20:20:40 +00004125 for (i = 0; environ[i]; i++)
4126 ;
Bram Moolenaard0573012017-10-28 21:11:06 +02004127
Bram Moolenaar071d4272004-06-13 20:20:40 +00004128 esize = i + EXTRASIZE + 1;
Bram Moolenaarc799fe22019-05-28 23:08:19 +02004129 env = ALLOC_MULT(char *, esize);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004130 if (env == NULL)
4131 return -1;
4132
Bram Moolenaar071d4272004-06-13 20:20:40 +00004133 for (i = 0; environ[i]; i++)
4134 {
Bram Moolenaarc799fe22019-05-28 23:08:19 +02004135 elem = alloc(strlen(environ[i]) + 1);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004136 if (elem == NULL)
4137 return -1;
4138 env[i] = elem;
4139 strcpy(elem, environ[i]);
4140 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004141
4142 env[i] = 0;
4143 environ = env;
4144 envsize = esize;
4145 return 0;
4146}
4147
4148 static int
Bram Moolenaar9b578142016-01-30 19:39:49 +01004149moreenv(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004150{
4151 int esize;
4152 char **env;
4153
4154 esize = envsize + EXTRASIZE;
Bram Moolenaarc799fe22019-05-28 23:08:19 +02004155 env = vim_realloc((char *)environ, esize * sizeof (*env));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004156 if (env == 0)
4157 return -1;
4158 environ = env;
4159 envsize = esize;
4160 return 0;
4161}
4162
4163# ifdef USE_VIMPTY_GETENV
Bram Moolenaar613fe7a2017-07-22 21:11:53 +02004164/*
4165 * Used for mch_getenv() for Mac.
4166 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004167 char_u *
Bram Moolenaar9b578142016-01-30 19:39:49 +01004168vimpty_getenv(const char_u *string)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004169{
4170 int i;
4171 char_u *p;
4172
4173 if (envsize < 0)
4174 return NULL;
4175
4176 i = findenv((char *)string);
4177
4178 if (i < 0)
4179 return NULL;
4180
4181 p = vim_strchr((char_u *)environ[i], '=');
4182 return (p + 1);
4183}
4184# endif
4185
4186#endif /* !defined(HAVE_SETENV) && !defined(HAVE_PUTENV) */
Bram Moolenaarc4a06d32005-06-07 21:04:49 +00004187
Bram Moolenaarc4956c82006-03-12 21:58:43 +00004188#if defined(FEAT_EVAL) || defined(FEAT_SPELL) || defined(PROTO)
Bram Moolenaarc4a06d32005-06-07 21:04:49 +00004189/*
4190 * Return 0 for not writable, 1 for writable file, 2 for a dir which we have
4191 * rights to write into.
4192 */
4193 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01004194filewritable(char_u *fname)
Bram Moolenaarc4a06d32005-06-07 21:04:49 +00004195{
4196 int retval = 0;
4197#if defined(UNIX) || defined(VMS)
4198 int perm = 0;
4199#endif
4200
4201#if defined(UNIX) || defined(VMS)
4202 perm = mch_getperm(fname);
4203#endif
Bram Moolenaarc4a06d32005-06-07 21:04:49 +00004204 if (
Bram Moolenaar4f974752019-02-17 17:44:42 +01004205# ifdef MSWIN
Bram Moolenaarc4a06d32005-06-07 21:04:49 +00004206 mch_writable(fname) &&
4207# else
4208# if defined(UNIX) || defined(VMS)
4209 (perm & 0222) &&
4210# endif
4211# endif
4212 mch_access((char *)fname, W_OK) == 0
4213 )
Bram Moolenaarc4a06d32005-06-07 21:04:49 +00004214 {
4215 ++retval;
4216 if (mch_isdir(fname))
4217 ++retval;
4218 }
4219 return retval;
4220}
4221#endif
Bram Moolenaar6bab4d12005-06-16 21:53:56 +00004222
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004223#if defined(FEAT_SPELL) || defined(FEAT_PERSISTENT_UNDO) || defined(PROTO)
4224/*
4225 * Read 2 bytes from "fd" and turn them into an int, MSB first.
Bram Moolenaare26e0d22018-03-20 12:34:04 +01004226 * Returns -1 when encountering EOF.
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004227 */
4228 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01004229get2c(FILE *fd)
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004230{
Bram Moolenaare26e0d22018-03-20 12:34:04 +01004231 int c, n;
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004232
4233 n = getc(fd);
Bram Moolenaare26e0d22018-03-20 12:34:04 +01004234 if (n == EOF) return -1;
4235 c = getc(fd);
4236 if (c == EOF) return -1;
4237 return (n << 8) + c;
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004238}
4239
4240/*
4241 * Read 3 bytes from "fd" and turn them into an int, MSB first.
Bram Moolenaare26e0d22018-03-20 12:34:04 +01004242 * Returns -1 when encountering EOF.
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004243 */
4244 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01004245get3c(FILE *fd)
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004246{
Bram Moolenaare26e0d22018-03-20 12:34:04 +01004247 int c, n;
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004248
4249 n = getc(fd);
Bram Moolenaare26e0d22018-03-20 12:34:04 +01004250 if (n == EOF) return -1;
4251 c = getc(fd);
4252 if (c == EOF) return -1;
4253 n = (n << 8) + c;
4254 c = getc(fd);
4255 if (c == EOF) return -1;
4256 return (n << 8) + c;
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004257}
4258
4259/*
4260 * Read 4 bytes from "fd" and turn them into an int, MSB first.
Bram Moolenaare26e0d22018-03-20 12:34:04 +01004261 * Returns -1 when encountering EOF.
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004262 */
4263 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01004264get4c(FILE *fd)
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004265{
Bram Moolenaare26e0d22018-03-20 12:34:04 +01004266 int c;
Bram Moolenaar95235e62013-09-08 16:07:07 +02004267 /* Use unsigned rather than int otherwise result is undefined
4268 * when left-shift sets the MSB. */
4269 unsigned n;
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004270
Bram Moolenaare26e0d22018-03-20 12:34:04 +01004271 c = getc(fd);
4272 if (c == EOF) return -1;
4273 n = (unsigned)c;
4274 c = getc(fd);
4275 if (c == EOF) return -1;
4276 n = (n << 8) + (unsigned)c;
4277 c = getc(fd);
4278 if (c == EOF) return -1;
4279 n = (n << 8) + (unsigned)c;
4280 c = getc(fd);
4281 if (c == EOF) return -1;
4282 n = (n << 8) + (unsigned)c;
Bram Moolenaar95235e62013-09-08 16:07:07 +02004283 return (int)n;
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004284}
4285
4286/*
Bram Moolenaarf4fba6d2016-06-26 16:44:24 +02004287 * Read 8 bytes from "fd" and turn them into a time_T, MSB first.
Bram Moolenaare26e0d22018-03-20 12:34:04 +01004288 * Returns -1 when encountering EOF.
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004289 */
Bram Moolenaarf4fba6d2016-06-26 16:44:24 +02004290 time_T
Bram Moolenaar9b578142016-01-30 19:39:49 +01004291get8ctime(FILE *fd)
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004292{
Bram Moolenaare26e0d22018-03-20 12:34:04 +01004293 int c;
Bram Moolenaarf4fba6d2016-06-26 16:44:24 +02004294 time_T n = 0;
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004295 int i;
4296
4297 for (i = 0; i < 8; ++i)
Bram Moolenaare26e0d22018-03-20 12:34:04 +01004298 {
4299 c = getc(fd);
4300 if (c == EOF) return -1;
4301 n = (n << 8) + c;
4302 }
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004303 return n;
4304}
4305
4306/*
4307 * Read a string of length "cnt" from "fd" into allocated memory.
4308 * Returns NULL when out of memory or unable to read that many bytes.
4309 */
4310 char_u *
Bram Moolenaar9b578142016-01-30 19:39:49 +01004311read_string(FILE *fd, int cnt)
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004312{
4313 char_u *str;
4314 int i;
4315 int c;
4316
4317 /* allocate memory */
Bram Moolenaar964b3742019-05-24 18:54:09 +02004318 str = alloc(cnt + 1);
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004319 if (str != NULL)
4320 {
4321 /* Read the string. Quit when running into the EOF. */
4322 for (i = 0; i < cnt; ++i)
4323 {
4324 c = getc(fd);
4325 if (c == EOF)
4326 {
4327 vim_free(str);
4328 return NULL;
4329 }
4330 str[i] = c;
4331 }
4332 str[i] = NUL;
4333 }
4334 return str;
4335}
4336
4337/*
4338 * Write a number to file "fd", MSB first, in "len" bytes.
4339 */
4340 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01004341put_bytes(FILE *fd, long_u nr, int len)
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004342{
4343 int i;
4344
4345 for (i = len - 1; i >= 0; --i)
4346 if (putc((int)(nr >> (i * 8)), fd) == EOF)
4347 return FAIL;
4348 return OK;
4349}
4350
4351#ifdef _MSC_VER
4352# if (_MSC_VER <= 1200)
4353/* This line is required for VC6 without the service pack. Also see the
4354 * matching #pragma below. */
4355 # pragma optimize("", off)
4356# endif
4357#endif
4358
4359/*
Bram Moolenaarf4fba6d2016-06-26 16:44:24 +02004360 * Write time_T to file "fd" in 8 bytes.
Bram Moolenaar285bf842016-01-07 22:34:01 +01004361 * Returns FAIL when the write failed.
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004362 */
Bram Moolenaar285bf842016-01-07 22:34:01 +01004363 int
Bram Moolenaarf4fba6d2016-06-26 16:44:24 +02004364put_time(FILE *fd, time_T the_time)
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004365{
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02004366 char_u buf[8];
4367
4368 time_to_bytes(the_time, buf);
Bram Moolenaar285bf842016-01-07 22:34:01 +01004369 return fwrite(buf, (size_t)8, (size_t)1, fd) == 1 ? OK : FAIL;
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02004370}
4371
4372/*
Bram Moolenaarf4fba6d2016-06-26 16:44:24 +02004373 * Write time_T to "buf[8]".
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02004374 */
4375 void
Bram Moolenaarf4fba6d2016-06-26 16:44:24 +02004376time_to_bytes(time_T the_time, char_u *buf)
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02004377{
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004378 int c;
4379 int i;
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02004380 int bi = 0;
Bram Moolenaarf4fba6d2016-06-26 16:44:24 +02004381 time_T wtime = the_time;
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004382
Bram Moolenaarf4fba6d2016-06-26 16:44:24 +02004383 /* time_T can be up to 8 bytes in size, more than long_u, thus we
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004384 * can't use put_bytes() here.
4385 * Another problem is that ">>" may do an arithmetic shift that keeps the
Bram Moolenaar644fdff2010-05-30 13:26:21 +02004386 * sign. This happens for large values of wtime. A cast to long_u may
Bram Moolenaarf4fba6d2016-06-26 16:44:24 +02004387 * truncate if time_T is 8 bytes. So only use a cast when it is 4 bytes,
Bram Moolenaar644fdff2010-05-30 13:26:21 +02004388 * it's safe to assume that long_u is 4 bytes or more and when using 8
4389 * bytes the top bit won't be set. */
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004390 for (i = 7; i >= 0; --i)
4391 {
Bram Moolenaarf4fba6d2016-06-26 16:44:24 +02004392 if (i + 1 > (int)sizeof(time_T))
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004393 /* ">>" doesn't work well when shifting more bits than avail */
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02004394 buf[bi++] = 0;
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004395 else
4396 {
Bram Moolenaar644fdff2010-05-30 13:26:21 +02004397#if defined(SIZEOF_TIME_T) && SIZEOF_TIME_T > 4
Bram Moolenaarbbd6afe2010-06-02 20:32:23 +02004398 c = (int)(wtime >> (i * 8));
Bram Moolenaar644fdff2010-05-30 13:26:21 +02004399#else
Bram Moolenaarbbd6afe2010-06-02 20:32:23 +02004400 c = (int)((long_u)wtime >> (i * 8));
Bram Moolenaar644fdff2010-05-30 13:26:21 +02004401#endif
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02004402 buf[bi++] = c;
Bram Moolenaarcdf04202010-05-29 15:11:47 +02004403 }
4404 }
4405}
4406
4407#ifdef _MSC_VER
4408# if (_MSC_VER <= 1200)
4409 # pragma optimize("", on)
4410# endif
4411#endif
4412
4413#endif
Bram Moolenaar10b7b392012-01-10 16:28:45 +01004414
Bram Moolenaarfc3abf42019-01-24 15:54:21 +01004415#if defined(FEAT_QUICKFIX) || defined(FEAT_SPELL) || defined(PROTO)
Bram Moolenaar10b7b392012-01-10 16:28:45 +01004416/*
4417 * Return TRUE if string "s" contains a non-ASCII character (128 or higher).
4418 * When "s" is NULL FALSE is returned.
4419 */
4420 int
Bram Moolenaar9b578142016-01-30 19:39:49 +01004421has_non_ascii(char_u *s)
Bram Moolenaar10b7b392012-01-10 16:28:45 +01004422{
4423 char_u *p;
4424
4425 if (s != NULL)
4426 for (p = s; *p != NUL; ++p)
4427 if (*p >= 128)
4428 return TRUE;
4429 return FALSE;
4430}
4431#endif
Bram Moolenaar93c88e02015-09-15 14:12:05 +02004432
4433#if defined(MESSAGE_QUEUE) || defined(PROTO)
Bram Moolenaarcd1a62d2018-12-14 21:32:02 +01004434# define MAX_REPEAT_PARSE 8
4435
Bram Moolenaar93c88e02015-09-15 14:12:05 +02004436/*
4437 * Process messages that have been queued for netbeans or clientserver.
Bram Moolenaar3a117e12016-10-30 21:57:52 +01004438 * Also check if any jobs have ended.
Bram Moolenaar93c88e02015-09-15 14:12:05 +02004439 * These functions can call arbitrary vimscript and should only be called when
4440 * it is safe to do so.
4441 */
4442 void
Bram Moolenaar9b578142016-01-30 19:39:49 +01004443parse_queued_messages(void)
Bram Moolenaar93c88e02015-09-15 14:12:05 +02004444{
Bram Moolenaarcd1a62d2018-12-14 21:32:02 +01004445 win_T *old_curwin = curwin;
4446 int i;
Bram Moolenaar6cc7e212019-07-07 14:05:24 +02004447 int save_may_garbage_collect = may_garbage_collect;
Bram Moolenaara3f7e582017-11-09 13:21:58 +01004448
Bram Moolenaar94f01952018-09-01 15:30:03 +02004449 // Do not handle messages while redrawing, because it may cause buffers to
4450 // change or be wiped while they are being redrawn.
4451 if (updating_screen)
4452 return;
4453
Bram Moolenaar6cc7e212019-07-07 14:05:24 +02004454 // may_garbage_collect is set in main_loop() to do garbage collection when
4455 // blocking to wait on a character. We don't want that while parsing
4456 // messages, a callback may invoke vgetc() while lists and dicts are in use
4457 // in the call stack.
4458 may_garbage_collect = FALSE;
4459
Bram Moolenaarcd1a62d2018-12-14 21:32:02 +01004460 // Loop when a job ended, but don't keep looping forever.
4461 for (i = 0; i < MAX_REPEAT_PARSE; ++i)
4462 {
4463 // For Win32 mch_breakcheck() does not check for input, do it here.
Bram Moolenaar4f974752019-02-17 17:44:42 +01004464# if defined(MSWIN) && defined(FEAT_JOB_CHANNEL)
Bram Moolenaarcd1a62d2018-12-14 21:32:02 +01004465 channel_handle_events(FALSE);
Bram Moolenaarb7522a22016-02-21 17:20:55 +01004466# endif
4467
Bram Moolenaar93c88e02015-09-15 14:12:05 +02004468# ifdef FEAT_NETBEANS_INTG
Bram Moolenaarcd1a62d2018-12-14 21:32:02 +01004469 // Process the queued netbeans messages.
4470 netbeans_parse_messages();
Bram Moolenaar93c88e02015-09-15 14:12:05 +02004471# endif
Bram Moolenaar509ce2a2016-03-11 22:52:15 +01004472# ifdef FEAT_JOB_CHANNEL
Bram Moolenaarcd1a62d2018-12-14 21:32:02 +01004473 // Write any buffer lines still to be written.
4474 channel_write_any_lines();
Bram Moolenaar8b877ac2016-03-28 19:16:20 +02004475
Bram Moolenaarcd1a62d2018-12-14 21:32:02 +01004476 // Process the messages queued on channels.
4477 channel_parse_messages();
Bram Moolenaar20fb9f32016-01-30 23:20:33 +01004478# endif
Bram Moolenaar95346802015-09-15 15:57:29 +02004479# if defined(FEAT_CLIENTSERVER) && defined(FEAT_X11)
Bram Moolenaarcd1a62d2018-12-14 21:32:02 +01004480 // Process the queued clientserver messages.
4481 server_parse_messages();
Bram Moolenaar93c88e02015-09-15 14:12:05 +02004482# endif
Bram Moolenaar509ce2a2016-03-11 22:52:15 +01004483# ifdef FEAT_JOB_CHANNEL
Bram Moolenaarcd1a62d2018-12-14 21:32:02 +01004484 // Check if any jobs have ended. If so, repeat the above to handle
4485 // changes, e.g. stdin may have been closed.
4486 if (job_check_ended())
4487 continue;
Bram Moolenaaree1cffc2016-02-21 19:14:41 +01004488# endif
Bram Moolenaar2a4857a2019-01-29 22:29:07 +01004489# ifdef FEAT_TERMINAL
4490 free_unused_terminals();
4491# endif
Bram Moolenaarcd1a62d2018-12-14 21:32:02 +01004492 break;
4493 }
Bram Moolenaara3f7e582017-11-09 13:21:58 +01004494
Bram Moolenaar6cc7e212019-07-07 14:05:24 +02004495 may_garbage_collect = save_may_garbage_collect;
4496
Bram Moolenaar94f01952018-09-01 15:30:03 +02004497 // If the current window changed we need to bail out of the waiting loop.
4498 // E.g. when a job exit callback closes the terminal window.
Bram Moolenaara3f7e582017-11-09 13:21:58 +01004499 if (curwin != old_curwin)
4500 ins_char_typebuf(K_IGNORE);
Bram Moolenaar93c88e02015-09-15 14:12:05 +02004501}
4502#endif
Bram Moolenaar833eb1d2016-11-24 17:22:50 +01004503
Bram Moolenaar51628222016-12-01 23:03:28 +01004504#ifndef PROTO /* proto is defined in vim.h */
4505# ifdef ELAPSED_TIMEVAL
Bram Moolenaar833eb1d2016-11-24 17:22:50 +01004506/*
4507 * Return time in msec since "start_tv".
4508 */
4509 long
4510elapsed(struct timeval *start_tv)
4511{
4512 struct timeval now_tv;
4513
4514 gettimeofday(&now_tv, NULL);
4515 return (now_tv.tv_sec - start_tv->tv_sec) * 1000L
4516 + (now_tv.tv_usec - start_tv->tv_usec) / 1000L;
4517}
Bram Moolenaar51628222016-12-01 23:03:28 +01004518# endif
Bram Moolenaar833eb1d2016-11-24 17:22:50 +01004519
Bram Moolenaar51628222016-12-01 23:03:28 +01004520# ifdef ELAPSED_TICKCOUNT
Bram Moolenaar833eb1d2016-11-24 17:22:50 +01004521/*
4522 * Return time in msec since "start_tick".
4523 */
4524 long
4525elapsed(DWORD start_tick)
4526{
4527 DWORD now = GetTickCount();
4528
Bram Moolenaar833eb1d2016-11-24 17:22:50 +01004529 return (long)now - (long)start_tick;
4530}
Bram Moolenaar51628222016-12-01 23:03:28 +01004531# endif
Bram Moolenaar833eb1d2016-11-24 17:22:50 +01004532#endif
Bram Moolenaar20608922018-04-21 22:30:08 +02004533
4534#if defined(FEAT_JOB_CHANNEL) \
4535 || (defined(UNIX) && (!defined(USE_SYSTEM) \
4536 || (defined(FEAT_GUI) && defined(FEAT_TERMINAL)))) \
4537 || defined(PROTO)
4538/*
4539 * Parse "cmd" and put the white-separated parts in "argv".
4540 * "argv" is an allocated array with "argc" entries and room for 4 more.
4541 * Returns FAIL when out of memory.
4542 */
4543 int
4544mch_parse_cmd(char_u *cmd, int use_shcf, char ***argv, int *argc)
4545{
4546 int i;
4547 char_u *p, *d;
4548 int inquote;
4549
4550 /*
4551 * Do this loop twice:
4552 * 1: find number of arguments
4553 * 2: separate them and build argv[]
4554 */
4555 for (i = 0; i < 2; ++i)
4556 {
4557 p = skipwhite(cmd);
4558 inquote = FALSE;
4559 *argc = 0;
4560 for (;;)
4561 {
4562 if (i == 1)
4563 (*argv)[*argc] = (char *)p;
4564 ++*argc;
4565 d = p;
4566 while (*p != NUL && (inquote || (*p != ' ' && *p != TAB)))
4567 {
4568 if (p[0] == '"')
Bram Moolenaar1df2fa42018-10-07 21:36:11 +02004569 // quotes surrounding an argument and are dropped
Bram Moolenaar20608922018-04-21 22:30:08 +02004570 inquote = !inquote;
4571 else
4572 {
Bram Moolenaar1df2fa42018-10-07 21:36:11 +02004573 if (rem_backslash(p))
Bram Moolenaar20608922018-04-21 22:30:08 +02004574 {
Bram Moolenaar1df2fa42018-10-07 21:36:11 +02004575 // First pass: skip over "\ " and "\"".
4576 // Second pass: Remove the backslash.
Bram Moolenaar20608922018-04-21 22:30:08 +02004577 ++p;
4578 }
4579 if (i == 1)
4580 *d++ = *p;
4581 }
4582 ++p;
4583 }
4584 if (*p == NUL)
4585 {
4586 if (i == 1)
4587 *d++ = NUL;
4588 break;
4589 }
4590 if (i == 1)
4591 *d++ = NUL;
4592 p = skipwhite(p + 1);
4593 }
4594 if (*argv == NULL)
4595 {
4596 if (use_shcf)
4597 {
4598 /* Account for possible multiple args in p_shcf. */
4599 p = p_shcf;
4600 for (;;)
4601 {
4602 p = skiptowhite(p);
4603 if (*p == NUL)
4604 break;
4605 ++*argc;
4606 p = skipwhite(p);
4607 }
4608 }
4609
Bram Moolenaarc799fe22019-05-28 23:08:19 +02004610 *argv = ALLOC_MULT(char *, *argc + 4);
Bram Moolenaar20608922018-04-21 22:30:08 +02004611 if (*argv == NULL) /* out of memory */
4612 return FAIL;
4613 }
4614 }
4615 return OK;
4616}
Bram Moolenaarebe74b72018-04-21 23:34:43 +02004617
4618# if defined(FEAT_JOB_CHANNEL) || defined(PROTO)
4619/*
4620 * Build "argv[argc]" from the string "cmd".
4621 * "argv[argc]" is set to NULL;
4622 * Return FAIL when out of memory.
4623 */
4624 int
4625build_argv_from_string(char_u *cmd, char ***argv, int *argc)
4626{
4627 char_u *cmd_copy;
4628 int i;
4629
4630 /* Make a copy, parsing will modify "cmd". */
4631 cmd_copy = vim_strsave(cmd);
4632 if (cmd_copy == NULL
4633 || mch_parse_cmd(cmd_copy, FALSE, argv, argc) == FAIL)
4634 {
4635 vim_free(cmd_copy);
4636 return FAIL;
4637 }
4638 for (i = 0; i < *argc; i++)
4639 (*argv)[i] = (char *)vim_strsave((char_u *)(*argv)[i]);
4640 (*argv)[*argc] = NULL;
4641 vim_free(cmd_copy);
4642 return OK;
4643}
4644
4645/*
4646 * Build "argv[argc]" from the list "l".
4647 * "argv[argc]" is set to NULL;
4648 * Return FAIL when out of memory.
4649 */
4650 int
4651build_argv_from_list(list_T *l, char ***argv, int *argc)
4652{
4653 listitem_T *li;
4654 char_u *s;
4655
4656 /* Pass argv[] to mch_call_shell(). */
Bram Moolenaarc799fe22019-05-28 23:08:19 +02004657 *argv = ALLOC_MULT(char *, l->lv_len + 1);
Bram Moolenaarebe74b72018-04-21 23:34:43 +02004658 if (*argv == NULL)
4659 return FAIL;
4660 *argc = 0;
4661 for (li = l->lv_first; li != NULL; li = li->li_next)
4662 {
Bram Moolenaard155d7a2018-12-21 16:04:21 +01004663 s = tv_get_string_chk(&li->li_tv);
Bram Moolenaarebe74b72018-04-21 23:34:43 +02004664 if (s == NULL)
4665 {
4666 int i;
4667
4668 for (i = 0; i < *argc; ++i)
4669 vim_free((*argv)[i]);
4670 return FAIL;
4671 }
4672 (*argv)[*argc] = (char *)vim_strsave(s);
4673 *argc += 1;
4674 }
4675 (*argv)[*argc] = NULL;
4676 return OK;
4677}
4678# endif
Bram Moolenaar20608922018-04-21 22:30:08 +02004679#endif
Bram Moolenaarafde13b2019-04-28 19:46:49 +02004680
4681#if defined(FEAT_SESSION) || defined(PROTO)
4682/*
4683 * Generate a script that can be used to restore the current editing session.
4684 * Save the value of v:this_session before running :mksession in order to make
4685 * automagic session save fully transparent. Return TRUE on success.
4686 */
4687 int
4688write_session_file(char_u *filename)
4689{
4690 char_u *escaped_filename;
4691 char *mksession_cmdline;
4692 unsigned int save_ssop_flags;
4693 int failed;
4694
4695 /*
4696 * Build an ex command line to create a script that restores the current
4697 * session if executed. Escape the filename to avoid nasty surprises.
4698 */
4699 escaped_filename = vim_strsave_escaped(filename, escape_chars);
4700 if (escaped_filename == NULL)
4701 return FALSE;
Bram Moolenaarc799fe22019-05-28 23:08:19 +02004702 mksession_cmdline = alloc(10 + (int)STRLEN(escaped_filename) + 1);
Bram Moolenaarafde13b2019-04-28 19:46:49 +02004703 if (mksession_cmdline == NULL)
4704 {
4705 vim_free(escaped_filename);
4706 return FALSE;
4707 }
4708 strcpy(mksession_cmdline, "mksession ");
4709 STRCAT(mksession_cmdline, escaped_filename);
4710 vim_free(escaped_filename);
4711
4712 /*
4713 * Use a reasonable hardcoded set of 'sessionoptions' flags to avoid
4714 * unpredictable effects when the session is saved automatically. Also,
4715 * we definitely need SSOP_GLOBALS to be able to restore v:this_session.
4716 * Don't use SSOP_BUFFERS to prevent the buffer list from becoming
4717 * enormously large if the GNOME session feature is used regularly.
4718 */
4719 save_ssop_flags = ssop_flags;
4720 ssop_flags = (SSOP_BLANK|SSOP_CURDIR|SSOP_FOLDS|SSOP_GLOBALS
4721 |SSOP_HELP|SSOP_OPTIONS|SSOP_WINSIZE|SSOP_TABPAGES);
4722
4723 do_cmdline_cmd((char_u *)"let Save_VV_this_session = v:this_session");
4724 failed = (do_cmdline_cmd((char_u *)mksession_cmdline) == FAIL);
4725 do_cmdline_cmd((char_u *)"let v:this_session = Save_VV_this_session");
4726 do_unlet((char_u *)"Save_VV_this_session", TRUE);
4727
4728 ssop_flags = save_ssop_flags;
4729 vim_free(mksession_cmdline);
4730
4731 /*
4732 * Reopen the file and append a command to restore v:this_session,
4733 * as if this save never happened. This is to avoid conflicts with
4734 * the user's own sessions. FIXME: It's probably less hackish to add
4735 * a "stealth" flag to 'sessionoptions' -- gotta ask Bram.
4736 */
4737 if (!failed)
4738 {
4739 FILE *fd;
4740
4741 fd = open_exfile(filename, TRUE, APPENDBIN);
4742
4743 failed = (fd == NULL
4744 || put_line(fd, "let v:this_session = Save_VV_this_session") == FAIL
4745 || put_line(fd, "unlet Save_VV_this_session") == FAIL);
4746
4747 if (fd != NULL && fclose(fd) != 0)
4748 failed = TRUE;
4749
4750 if (failed)
4751 mch_remove(filename);
4752 }
4753
4754 return !failed;
4755}
4756#endif