blob: 00d1ff4a0f4d5ce73bf3667b4ed9138bdaa0f86c [file] [log] [blame]
Bram Moolenaar071d4272004-06-13 20:20:40 +00001/* vi:set ts=8 sts=4 sw=4:
2 *
3 * VIM - Vi IMproved by Bram Moolenaar
4 *
5 * Do ":help uganda" in Vim to read copying and usage conditions.
6 * Do ":help credits" in Vim to see a list of people who contributed.
7 * See README.txt for an overview of the Vim source code.
8 */
9
10/*
11 * 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
17static char_u *ff_expand_buffer = NULL; /* used for expanding filenames */
18
Bram Moolenaar071d4272004-06-13 20:20:40 +000019#if defined(FEAT_VIRTUALEDIT) || defined(PROTO)
20static int coladvance2 __ARGS((pos_T *pos, int addspaces, int finetune, colnr_T wcol));
21
22/*
23 * Return TRUE if in the current mode we need to use virtual.
24 */
25 int
26virtual_active()
27{
28 /* While an operator is being executed we return "virtual_op", because
29 * VIsual_active has already been reset, thus we can't check for "block"
30 * being used. */
31 if (virtual_op != MAYBE)
32 return virtual_op;
33 return (ve_flags == VE_ALL
34# ifdef FEAT_VISUAL
35 || ((ve_flags & VE_BLOCK) && VIsual_active && VIsual_mode == Ctrl_V)
36# endif
37 || ((ve_flags & VE_INSERT) && (State & INSERT)));
38}
39
40/*
41 * Get the screen position of the cursor.
42 */
43 int
44getviscol()
45{
46 colnr_T x;
47
48 getvvcol(curwin, &curwin->w_cursor, &x, NULL, NULL);
49 return (int)x;
50}
51
52/*
53 * Get the screen position of character col with a coladd in the cursor line.
54 */
55 int
56getviscol2(col, coladd)
57 colnr_T col;
58 colnr_T coladd;
59{
60 colnr_T x;
61 pos_T pos;
62
63 pos.lnum = curwin->w_cursor.lnum;
64 pos.col = col;
65 pos.coladd = coladd;
66 getvvcol(curwin, &pos, &x, NULL, NULL);
67 return (int)x;
68}
69
70/*
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +000071 * Go to column "wcol", and add/insert white space as necessary to get the
Bram Moolenaar071d4272004-06-13 20:20:40 +000072 * cursor in that column.
73 * The caller must have saved the cursor line for undo!
74 */
75 int
76coladvance_force(wcol)
77 colnr_T wcol;
78{
79 int rc = coladvance2(&curwin->w_cursor, TRUE, FALSE, wcol);
80
81 if (wcol == MAXCOL)
82 curwin->w_valid &= ~VALID_VIRTCOL;
83 else
84 {
85 /* Virtcol is valid */
86 curwin->w_valid |= VALID_VIRTCOL;
87 curwin->w_virtcol = wcol;
88 }
89 return rc;
90}
91#endif
92
93/*
94 * Try to advance the Cursor to the specified screen column.
95 * If virtual editing: fine tune the cursor position.
96 * Note that all virtual positions off the end of a line should share
97 * a curwin->w_cursor.col value (n.b. this is equal to STRLEN(line)),
98 * beginning at coladd 0.
99 *
100 * return OK if desired column is reached, FAIL if not
101 */
102 int
103coladvance(wcol)
104 colnr_T wcol;
105{
106 int rc = getvpos(&curwin->w_cursor, wcol);
107
108 if (wcol == MAXCOL || rc == FAIL)
109 curwin->w_valid &= ~VALID_VIRTCOL;
Bram Moolenaardfccaf02004-12-31 20:56:11 +0000110 else if (*ml_get_cursor() != TAB)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000111 {
Bram Moolenaardfccaf02004-12-31 20:56:11 +0000112 /* Virtcol is valid when not on a TAB */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000113 curwin->w_valid |= VALID_VIRTCOL;
114 curwin->w_virtcol = wcol;
115 }
116 return rc;
117}
118
119/*
120 * Return in "pos" the position of the cursor advanced to screen column "wcol".
121 * return OK if desired column is reached, FAIL if not
122 */
123 int
124getvpos(pos, wcol)
125 pos_T *pos;
126 colnr_T wcol;
127{
128#ifdef FEAT_VIRTUALEDIT
129 return coladvance2(pos, FALSE, virtual_active(), wcol);
130}
131
132 static int
133coladvance2(pos, addspaces, finetune, wcol)
134 pos_T *pos;
135 int addspaces; /* change the text to achieve our goal? */
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +0000136 int finetune; /* change char offset for the exact column */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000137 colnr_T wcol; /* column to move to */
138{
139#endif
140 int idx;
141 char_u *ptr;
142 char_u *line;
143 colnr_T col = 0;
144 int csize = 0;
145 int one_more;
146#ifdef FEAT_LINEBREAK
147 int head = 0;
148#endif
149
Bram Moolenaarefd2bf12006-03-16 21:41:35 +0000150 one_more = (State & INSERT)
151 || restart_edit != NUL
Bram Moolenaar071d4272004-06-13 20:20:40 +0000152#ifdef FEAT_VISUAL
Bram Moolenaarefd2bf12006-03-16 21:41:35 +0000153 || (VIsual_active && *p_sel != 'o')
Bram Moolenaar071d4272004-06-13 20:20:40 +0000154#endif
Bram Moolenaarefd2bf12006-03-16 21:41:35 +0000155#ifdef FEAT_VIRTUALEDIT
Bram Moolenaar97b2ad32006-03-18 21:40:56 +0000156 || ((ve_flags & VE_ONEMORE) && wcol < MAXCOL)
Bram Moolenaarefd2bf12006-03-16 21:41:35 +0000157#endif
158 ;
Bram Moolenaara1381de2009-11-03 15:44:21 +0000159 line = ml_get_buf(curbuf, pos->lnum, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000160
161 if (wcol >= MAXCOL)
162 {
163 idx = (int)STRLEN(line) - 1 + one_more;
164 col = wcol;
165
166#ifdef FEAT_VIRTUALEDIT
167 if ((addspaces || finetune) && !VIsual_active)
168 {
169 curwin->w_curswant = linetabsize(line) + one_more;
170 if (curwin->w_curswant > 0)
171 --curwin->w_curswant;
172 }
173#endif
174 }
175 else
176 {
177#ifdef FEAT_VIRTUALEDIT
178 int width = W_WIDTH(curwin) - win_col_off(curwin);
179
Bram Moolenaarebefac62005-12-28 22:39:57 +0000180 if (finetune
Bram Moolenaar071d4272004-06-13 20:20:40 +0000181 && curwin->w_p_wrap
182# ifdef FEAT_VERTSPLIT
183 && curwin->w_width != 0
184# endif
185 && wcol >= (colnr_T)width)
186 {
187 csize = linetabsize(line);
188 if (csize > 0)
189 csize--;
190
Bram Moolenaarebefac62005-12-28 22:39:57 +0000191 if (wcol / width > (colnr_T)csize / width
192 && ((State & INSERT) == 0 || (int)wcol > csize + 1))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000193 {
194 /* In case of line wrapping don't move the cursor beyond the
Bram Moolenaarebefac62005-12-28 22:39:57 +0000195 * right screen edge. In Insert mode allow going just beyond
196 * the last character (like what happens when typing and
197 * reaching the right window edge). */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000198 wcol = (csize / width + 1) * width - 1;
199 }
200 }
201#endif
202
Bram Moolenaar071d4272004-06-13 20:20:40 +0000203 ptr = line;
204 while (col <= wcol && *ptr != NUL)
205 {
206 /* Count a tab for what it's worth (if list mode not on) */
207#ifdef FEAT_LINEBREAK
208 csize = win_lbr_chartabsize(curwin, ptr, col, &head);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +0000209 mb_ptr_adv(ptr);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000210#else
211 csize = lbr_chartabsize_adv(&ptr, col);
212#endif
213 col += csize;
214 }
215 idx = (int)(ptr - line);
216 /*
217 * Handle all the special cases. The virtual_active() check
218 * is needed to ensure that a virtual position off the end of
219 * a line has the correct indexing. The one_more comparison
220 * replaces an explicit add of one_more later on.
221 */
222 if (col > wcol || (!virtual_active() && one_more == 0))
223 {
224 idx -= 1;
225# ifdef FEAT_LINEBREAK
226 /* Don't count the chars from 'showbreak'. */
227 csize -= head;
228# endif
229 col -= csize;
230 }
231
232#ifdef FEAT_VIRTUALEDIT
233 if (virtual_active()
234 && addspaces
235 && ((col != wcol && col != wcol + 1) || csize > 1))
236 {
237 /* 'virtualedit' is set: The difference between wcol and col is
238 * filled with spaces. */
239
240 if (line[idx] == NUL)
241 {
242 /* Append spaces */
243 int correct = wcol - col;
244 char_u *newline = alloc(idx + correct + 1);
245 int t;
246
247 if (newline == NULL)
248 return FAIL;
249
250 for (t = 0; t < idx; ++t)
251 newline[t] = line[t];
252
253 for (t = 0; t < correct; ++t)
254 newline[t + idx] = ' ';
255
256 newline[idx + correct] = NUL;
257
258 ml_replace(pos->lnum, newline, FALSE);
259 changed_bytes(pos->lnum, (colnr_T)idx);
260 idx += correct;
261 col = wcol;
262 }
263 else
264 {
265 /* Break a tab */
266 int linelen = (int)STRLEN(line);
267 int correct = wcol - col - csize + 1; /* negative!! */
Bram Moolenaareb3593b2006-04-22 22:33:57 +0000268 char_u *newline;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000269 int t, s = 0;
270 int v;
271
Bram Moolenaareb3593b2006-04-22 22:33:57 +0000272 if (-correct > csize)
273 return FAIL;
274
275 newline = alloc(linelen + csize);
276 if (newline == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000277 return FAIL;
278
279 for (t = 0; t < linelen; t++)
280 {
281 if (t != idx)
282 newline[s++] = line[t];
283 else
284 for (v = 0; v < csize; v++)
285 newline[s++] = ' ';
286 }
287
288 newline[linelen + csize - 1] = NUL;
289
290 ml_replace(pos->lnum, newline, FALSE);
291 changed_bytes(pos->lnum, idx);
292 idx += (csize - 1 + correct);
293 col += correct;
294 }
295 }
296#endif
297 }
298
299 if (idx < 0)
300 pos->col = 0;
301 else
302 pos->col = idx;
303
304#ifdef FEAT_VIRTUALEDIT
305 pos->coladd = 0;
306
307 if (finetune)
308 {
309 if (wcol == MAXCOL)
310 {
311 /* The width of the last character is used to set coladd. */
312 if (!one_more)
313 {
314 colnr_T scol, ecol;
315
316 getvcol(curwin, pos, &scol, NULL, &ecol);
317 pos->coladd = ecol - scol;
318 }
319 }
320 else
321 {
322 int b = (int)wcol - (int)col;
323
324 /* The difference between wcol and col is used to set coladd. */
325 if (b > 0 && b < (MAXCOL - 2 * W_WIDTH(curwin)))
326 pos->coladd = b;
327
328 col += b;
329 }
330 }
331#endif
332
333#ifdef FEAT_MBYTE
Bram Moolenaara1381de2009-11-03 15:44:21 +0000334 /* prevent from moving onto a trail byte */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000335 if (has_mbyte)
Bram Moolenaar03a807a2011-07-07 15:08:58 +0200336 mb_adjustpos(curbuf, pos);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000337#endif
338
339 if (col < wcol)
340 return FAIL;
341 return OK;
342}
343
344/*
Bram Moolenaar446cb832008-06-24 21:56:24 +0000345 * Increment the cursor position. See inc() for return values.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000346 */
347 int
348inc_cursor()
349{
350 return inc(&curwin->w_cursor);
351}
352
Bram Moolenaar446cb832008-06-24 21:56:24 +0000353/*
354 * Increment the line pointer "lp" crossing line boundaries as necessary.
355 * Return 1 when going to the next line.
356 * Return 2 when moving forward onto a NUL at the end of the line).
357 * Return -1 when at the end of file.
358 * Return 0 otherwise.
359 */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000360 int
361inc(lp)
362 pos_T *lp;
363{
364 char_u *p = ml_get_pos(lp);
365
366 if (*p != NUL) /* still within line, move to next char (may be NUL) */
367 {
368#ifdef FEAT_MBYTE
369 if (has_mbyte)
370 {
Bram Moolenaar0fa313a2005-08-10 21:07:57 +0000371 int l = (*mb_ptr2len)(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000372
373 lp->col += l;
374 return ((p[l] != NUL) ? 0 : 2);
375 }
376#endif
377 lp->col++;
378#ifdef FEAT_VIRTUALEDIT
379 lp->coladd = 0;
380#endif
381 return ((p[1] != NUL) ? 0 : 2);
382 }
383 if (lp->lnum != curbuf->b_ml.ml_line_count) /* there is a next line */
384 {
385 lp->col = 0;
386 lp->lnum++;
387#ifdef FEAT_VIRTUALEDIT
388 lp->coladd = 0;
389#endif
390 return 1;
391 }
392 return -1;
393}
394
395/*
396 * incl(lp): same as inc(), but skip the NUL at the end of non-empty lines
397 */
398 int
399incl(lp)
400 pos_T *lp;
401{
402 int r;
403
404 if ((r = inc(lp)) >= 1 && lp->col)
405 r = inc(lp);
406 return r;
407}
408
409/*
410 * dec(p)
411 *
412 * Decrement the line pointer 'p' crossing line boundaries as necessary.
413 * Return 1 when crossing a line, -1 when at start of file, 0 otherwise.
414 */
415 int
416dec_cursor()
417{
418 return dec(&curwin->w_cursor);
419}
420
421 int
422dec(lp)
423 pos_T *lp;
424{
425 char_u *p;
426
427#ifdef FEAT_VIRTUALEDIT
428 lp->coladd = 0;
429#endif
430 if (lp->col > 0) /* still within line */
431 {
432 lp->col--;
433#ifdef FEAT_MBYTE
434 if (has_mbyte)
435 {
436 p = ml_get(lp->lnum);
437 lp->col -= (*mb_head_off)(p, p + lp->col);
438 }
439#endif
440 return 0;
441 }
442 if (lp->lnum > 1) /* there is a prior line */
443 {
444 lp->lnum--;
445 p = ml_get(lp->lnum);
446 lp->col = (colnr_T)STRLEN(p);
447#ifdef FEAT_MBYTE
448 if (has_mbyte)
449 lp->col -= (*mb_head_off)(p, p + lp->col);
450#endif
451 return 1;
452 }
453 return -1; /* at start of file */
454}
455
456/*
457 * decl(lp): same as dec(), but skip the NUL at the end of non-empty lines
458 */
459 int
460decl(lp)
461 pos_T *lp;
462{
463 int r;
464
465 if ((r = dec(lp)) == 1 && lp->col)
466 r = dec(lp);
467 return r;
468}
469
470/*
Bram Moolenaar64486672010-05-16 15:46:46 +0200471 * Get the line number relative to the current cursor position, i.e. the
472 * difference between line number and cursor position. Only look for lines that
473 * can be visible, folded lines don't count.
474 */
475 linenr_T
476get_cursor_rel_lnum(wp, lnum)
477 win_T *wp;
478 linenr_T lnum; /* line number to get the result for */
479{
480 linenr_T cursor = wp->w_cursor.lnum;
481 linenr_T retval = 0;
482
483#ifdef FEAT_FOLDING
484 if (hasAnyFolding(wp))
485 {
486 if (lnum > cursor)
487 {
488 while (lnum > cursor)
489 {
490 (void)hasFolding(lnum, &lnum, NULL);
491 /* if lnum and cursor are in the same fold,
492 * now lnum <= cursor */
493 if (lnum > cursor)
494 retval++;
495 lnum--;
496 }
497 }
498 else if (lnum < cursor)
499 {
500 while (lnum < cursor)
501 {
502 (void)hasFolding(lnum, NULL, &lnum);
503 /* if lnum and cursor are in the same fold,
504 * now lnum >= cursor */
505 if (lnum < cursor)
506 retval--;
507 lnum++;
508 }
509 }
510 /* else if (lnum == cursor)
511 * retval = 0;
512 */
513 }
514 else
515#endif
516 retval = lnum - cursor;
517
518 return retval;
519}
520
521/*
Bram Moolenaar071d4272004-06-13 20:20:40 +0000522 * Make sure curwin->w_cursor.lnum is valid.
523 */
524 void
525check_cursor_lnum()
526{
527 if (curwin->w_cursor.lnum > curbuf->b_ml.ml_line_count)
528 {
529#ifdef FEAT_FOLDING
530 /* If there is a closed fold at the end of the file, put the cursor in
531 * its first line. Otherwise in the last line. */
532 if (!hasFolding(curbuf->b_ml.ml_line_count,
533 &curwin->w_cursor.lnum, NULL))
534#endif
535 curwin->w_cursor.lnum = curbuf->b_ml.ml_line_count;
536 }
537 if (curwin->w_cursor.lnum <= 0)
538 curwin->w_cursor.lnum = 1;
539}
540
541/*
542 * Make sure curwin->w_cursor.col is valid.
543 */
544 void
545check_cursor_col()
546{
Bram Moolenaar03a807a2011-07-07 15:08:58 +0200547 check_cursor_col_win(curwin);
548}
549
550/*
551 * Make sure win->w_cursor.col is valid.
552 */
553 void
554check_cursor_col_win(win)
555 win_T *win;
556{
Bram Moolenaar071d4272004-06-13 20:20:40 +0000557 colnr_T len;
558#ifdef FEAT_VIRTUALEDIT
Bram Moolenaar03a807a2011-07-07 15:08:58 +0200559 colnr_T oldcol = win->w_cursor.col;
560 colnr_T oldcoladd = win->w_cursor.col + win->w_cursor.coladd;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000561#endif
562
Bram Moolenaar03a807a2011-07-07 15:08:58 +0200563 len = (colnr_T)STRLEN(ml_get_buf(win->w_buffer, win->w_cursor.lnum, FALSE));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000564 if (len == 0)
Bram Moolenaar03a807a2011-07-07 15:08:58 +0200565 win->w_cursor.col = 0;
566 else if (win->w_cursor.col >= len)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000567 {
Bram Moolenaar33f54432008-01-04 20:25:44 +0000568 /* Allow cursor past end-of-line when:
569 * - in Insert mode or restarting Insert mode
570 * - in Visual mode and 'selection' isn't "old"
571 * - 'virtualedit' is set */
Bram Moolenaarebefac62005-12-28 22:39:57 +0000572 if ((State & INSERT) || restart_edit
Bram Moolenaar071d4272004-06-13 20:20:40 +0000573#ifdef FEAT_VISUAL
574 || (VIsual_active && *p_sel != 'o')
575#endif
Bram Moolenaar33f54432008-01-04 20:25:44 +0000576#ifdef FEAT_VIRTUALEDIT
577 || (ve_flags & VE_ONEMORE)
578#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000579 || virtual_active())
Bram Moolenaar03a807a2011-07-07 15:08:58 +0200580 win->w_cursor.col = len;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000581 else
Bram Moolenaar87c19962007-04-26 08:54:21 +0000582 {
Bram Moolenaar03a807a2011-07-07 15:08:58 +0200583 win->w_cursor.col = len - 1;
Bram Moolenaar87c19962007-04-26 08:54:21 +0000584#ifdef FEAT_MBYTE
Bram Moolenaar03a807a2011-07-07 15:08:58 +0200585 /* Move the cursor to the head byte. */
Bram Moolenaar87c19962007-04-26 08:54:21 +0000586 if (has_mbyte)
Bram Moolenaar03a807a2011-07-07 15:08:58 +0200587 mb_adjustpos(win->w_buffer, &win->w_cursor);
Bram Moolenaar87c19962007-04-26 08:54:21 +0000588#endif
589 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000590 }
Bram Moolenaar03a807a2011-07-07 15:08:58 +0200591 else if (win->w_cursor.col < 0)
592 win->w_cursor.col = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000593
594#ifdef FEAT_VIRTUALEDIT
595 /* If virtual editing is on, we can leave the cursor on the old position,
596 * only we must set it to virtual. But don't do it when at the end of the
597 * line. */
598 if (oldcol == MAXCOL)
Bram Moolenaar03a807a2011-07-07 15:08:58 +0200599 win->w_cursor.coladd = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000600 else if (ve_flags == VE_ALL)
Bram Moolenaar552c8a52009-03-11 16:29:20 +0000601 {
Bram Moolenaar03a807a2011-07-07 15:08:58 +0200602 if (oldcoladd > win->w_cursor.col)
603 win->w_cursor.coladd = oldcoladd - win->w_cursor.col;
Bram Moolenaar552c8a52009-03-11 16:29:20 +0000604 else
605 /* avoid weird number when there is a miscalculation or overflow */
Bram Moolenaar03a807a2011-07-07 15:08:58 +0200606 win->w_cursor.coladd = 0;
Bram Moolenaar552c8a52009-03-11 16:29:20 +0000607 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000608#endif
609}
610
611/*
612 * make sure curwin->w_cursor in on a valid character
613 */
614 void
615check_cursor()
616{
617 check_cursor_lnum();
618 check_cursor_col();
619}
620
621#if defined(FEAT_TEXTOBJ) || defined(PROTO)
622/*
623 * Make sure curwin->w_cursor is not on the NUL at the end of the line.
624 * Allow it when in Visual mode and 'selection' is not "old".
625 */
626 void
627adjust_cursor_col()
628{
629 if (curwin->w_cursor.col > 0
630# ifdef FEAT_VISUAL
631 && (!VIsual_active || *p_sel == 'o')
632# endif
633 && gchar_cursor() == NUL)
634 --curwin->w_cursor.col;
635}
636#endif
637
638/*
639 * When curwin->w_leftcol has changed, adjust the cursor position.
640 * Return TRUE if the cursor was moved.
641 */
642 int
643leftcol_changed()
644{
645 long lastcol;
646 colnr_T s, e;
647 int retval = FALSE;
648
649 changed_cline_bef_curs();
650 lastcol = curwin->w_leftcol + W_WIDTH(curwin) - curwin_col_off() - 1;
651 validate_virtcol();
652
653 /*
654 * If the cursor is right or left of the screen, move it to last or first
655 * character.
656 */
657 if (curwin->w_virtcol > (colnr_T)(lastcol - p_siso))
658 {
659 retval = TRUE;
660 coladvance((colnr_T)(lastcol - p_siso));
661 }
662 else if (curwin->w_virtcol < curwin->w_leftcol + p_siso)
663 {
664 retval = TRUE;
665 (void)coladvance((colnr_T)(curwin->w_leftcol + p_siso));
666 }
667
668 /*
669 * If the start of the character under the cursor is not on the screen,
670 * advance the cursor one more char. If this fails (last char of the
671 * line) adjust the scrolling.
672 */
673 getvvcol(curwin, &curwin->w_cursor, &s, NULL, &e);
674 if (e > (colnr_T)lastcol)
675 {
676 retval = TRUE;
677 coladvance(s - 1);
678 }
679 else if (s < curwin->w_leftcol)
680 {
681 retval = TRUE;
682 if (coladvance(e + 1) == FAIL) /* there isn't another character */
683 {
684 curwin->w_leftcol = s; /* adjust w_leftcol instead */
685 changed_cline_bef_curs();
686 }
687 }
688
689 if (retval)
690 curwin->w_set_curswant = TRUE;
691 redraw_later(NOT_VALID);
692 return retval;
693}
694
695/**********************************************************************
696 * Various routines dealing with allocation and deallocation of memory.
697 */
698
699#if defined(MEM_PROFILE) || defined(PROTO)
700
701# define MEM_SIZES 8200
702static long_u mem_allocs[MEM_SIZES];
703static long_u mem_frees[MEM_SIZES];
704static long_u mem_allocated;
705static long_u mem_freed;
706static long_u mem_peak;
707static long_u num_alloc;
708static long_u num_freed;
709
710static void mem_pre_alloc_s __ARGS((size_t *sizep));
711static void mem_pre_alloc_l __ARGS((long_u *sizep));
712static void mem_post_alloc __ARGS((void **pp, size_t size));
713static void mem_pre_free __ARGS((void **pp));
714
715 static void
716mem_pre_alloc_s(sizep)
717 size_t *sizep;
718{
719 *sizep += sizeof(size_t);
720}
721
722 static void
723mem_pre_alloc_l(sizep)
724 long_u *sizep;
725{
726 *sizep += sizeof(size_t);
727}
728
729 static void
730mem_post_alloc(pp, size)
731 void **pp;
732 size_t size;
733{
734 if (*pp == NULL)
735 return;
736 size -= sizeof(size_t);
737 *(long_u *)*pp = size;
738 if (size <= MEM_SIZES-1)
739 mem_allocs[size-1]++;
740 else
741 mem_allocs[MEM_SIZES-1]++;
742 mem_allocated += size;
743 if (mem_allocated - mem_freed > mem_peak)
744 mem_peak = mem_allocated - mem_freed;
745 num_alloc++;
746 *pp = (void *)((char *)*pp + sizeof(size_t));
747}
748
749 static void
750mem_pre_free(pp)
751 void **pp;
752{
753 long_u size;
754
755 *pp = (void *)((char *)*pp - sizeof(size_t));
756 size = *(size_t *)*pp;
757 if (size <= MEM_SIZES-1)
758 mem_frees[size-1]++;
759 else
760 mem_frees[MEM_SIZES-1]++;
761 mem_freed += size;
762 num_freed++;
763}
764
765/*
766 * called on exit via atexit()
767 */
768 void
769vim_mem_profile_dump()
770{
771 int i, j;
772
773 printf("\r\n");
774 j = 0;
775 for (i = 0; i < MEM_SIZES - 1; i++)
776 {
777 if (mem_allocs[i] || mem_frees[i])
778 {
779 if (mem_frees[i] > mem_allocs[i])
780 printf("\r\n%s", _("ERROR: "));
781 printf("[%4d / %4lu-%-4lu] ", i + 1, mem_allocs[i], mem_frees[i]);
782 j++;
783 if (j > 3)
784 {
785 j = 0;
786 printf("\r\n");
787 }
788 }
789 }
790
791 i = MEM_SIZES - 1;
792 if (mem_allocs[i])
793 {
794 printf("\r\n");
795 if (mem_frees[i] > mem_allocs[i])
Bram Moolenaar2f1e0502010-08-13 11:18:02 +0200796 puts(_("ERROR: "));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000797 printf("[>%d / %4lu-%-4lu]", i, mem_allocs[i], mem_frees[i]);
798 }
799
800 printf(_("\n[bytes] total alloc-freed %lu-%lu, in use %lu, peak use %lu\n"),
801 mem_allocated, mem_freed, mem_allocated - mem_freed, mem_peak);
802 printf(_("[calls] total re/malloc()'s %lu, total free()'s %lu\n\n"),
803 num_alloc, num_freed);
804}
805
806#endif /* MEM_PROFILE */
807
808/*
809 * Some memory is reserved for error messages and for being able to
810 * call mf_release_all(), which needs some memory for mf_trans_add().
811 */
812#if defined(MSDOS) && !defined(DJGPP)
813# define SMALL_MEM
814# define KEEP_ROOM 8192L
815#else
816# define KEEP_ROOM (2 * 8192L)
817#endif
818
819/*
Bram Moolenaar2a329742008-04-01 12:53:43 +0000820 * Note: if unsigned is 16 bits we can only allocate up to 64K with alloc().
Bram Moolenaar071d4272004-06-13 20:20:40 +0000821 * Use lalloc for larger blocks.
822 */
823 char_u *
824alloc(size)
825 unsigned size;
826{
827 return (lalloc((long_u)size, TRUE));
828}
829
830/*
831 * Allocate memory and set all bytes to zero.
832 */
833 char_u *
834alloc_clear(size)
835 unsigned size;
836{
837 char_u *p;
838
Bram Moolenaar2f1e0502010-08-13 11:18:02 +0200839 p = lalloc((long_u)size, TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000840 if (p != NULL)
841 (void)vim_memset(p, 0, (size_t)size);
842 return p;
843}
844
845/*
846 * alloc() with check for maximum line length
847 */
848 char_u *
849alloc_check(size)
850 unsigned size;
851{
852#if !defined(UNIX) && !defined(__EMX__)
853 if (sizeof(int) == 2 && size > 0x7fff)
854 {
855 /* Don't hide this message */
856 emsg_silent = 0;
857 EMSG(_("E340: Line is becoming too long"));
858 return NULL;
859 }
860#endif
861 return (lalloc((long_u)size, TRUE));
862}
863
864/*
865 * Allocate memory like lalloc() and set all bytes to zero.
866 */
867 char_u *
868lalloc_clear(size, message)
869 long_u size;
870 int message;
871{
872 char_u *p;
873
874 p = (lalloc(size, message));
875 if (p != NULL)
876 (void)vim_memset(p, 0, (size_t)size);
877 return p;
878}
879
880/*
881 * Low level memory allocation function.
882 * This is used often, KEEP IT FAST!
883 */
884 char_u *
885lalloc(size, message)
886 long_u size;
887 int message;
888{
889 char_u *p; /* pointer to new storage space */
890 static int releasing = FALSE; /* don't do mf_release_all() recursive */
891 int try_again;
892#if defined(HAVE_AVAIL_MEM) && !defined(SMALL_MEM)
893 static long_u allocated = 0; /* allocated since last avail check */
894#endif
895
896 /* Safety check for allocating zero bytes */
897 if (size == 0)
898 {
899 /* Don't hide this message */
900 emsg_silent = 0;
901 EMSGN(_("E341: Internal error: lalloc(%ld, )"), size);
902 return NULL;
903 }
904
905#ifdef MEM_PROFILE
906 mem_pre_alloc_l(&size);
907#endif
908
909#if defined(MSDOS) && !defined(DJGPP)
910 if (size >= 0xfff0) /* in MSDOS we can't deal with >64K blocks */
911 p = NULL;
912 else
913#endif
914
915 /*
916 * Loop when out of memory: Try to release some memfile blocks and
917 * if some blocks are released call malloc again.
918 */
919 for (;;)
920 {
921 /*
922 * Handle three kind of systems:
923 * 1. No check for available memory: Just return.
924 * 2. Slow check for available memory: call mch_avail_mem() after
925 * allocating KEEP_ROOM amount of memory.
926 * 3. Strict check for available memory: call mch_avail_mem()
927 */
928 if ((p = (char_u *)malloc((size_t)size)) != NULL)
929 {
930#ifndef HAVE_AVAIL_MEM
931 /* 1. No check for available memory: Just return. */
932 goto theend;
933#else
934# ifndef SMALL_MEM
935 /* 2. Slow check for available memory: call mch_avail_mem() after
936 * allocating (KEEP_ROOM / 2) amount of memory. */
937 allocated += size;
938 if (allocated < KEEP_ROOM / 2)
939 goto theend;
940 allocated = 0;
941# endif
942 /* 3. check for available memory: call mch_avail_mem() */
943 if (mch_avail_mem(TRUE) < KEEP_ROOM && !releasing)
944 {
Bram Moolenaar5a221812008-11-12 12:08:45 +0000945 free((char *)p); /* System is low... no go! */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000946 p = NULL;
947 }
948 else
949 goto theend;
950#endif
951 }
952 /*
953 * Remember that mf_release_all() is being called to avoid an endless
954 * loop, because mf_release_all() may call alloc() recursively.
955 */
956 if (releasing)
957 break;
958 releasing = TRUE;
Bram Moolenaar661b1822005-07-28 22:36:45 +0000959
960 clear_sb_text(); /* free any scrollback text */
961 try_again = mf_release_all(); /* release as many blocks as possible */
Bram Moolenaar39a58ca2005-06-27 22:42:44 +0000962#ifdef FEAT_EVAL
Bram Moolenaar661b1822005-07-28 22:36:45 +0000963 try_again |= garbage_collect(); /* cleanup recursive lists/dicts */
Bram Moolenaar39a58ca2005-06-27 22:42:44 +0000964#endif
Bram Moolenaar661b1822005-07-28 22:36:45 +0000965
Bram Moolenaar071d4272004-06-13 20:20:40 +0000966 releasing = FALSE;
967 if (!try_again)
968 break;
969 }
970
971 if (message && p == NULL)
972 do_outofmem_msg(size);
973
974theend:
975#ifdef MEM_PROFILE
976 mem_post_alloc((void **)&p, (size_t)size);
977#endif
978 return p;
979}
980
981#if defined(MEM_PROFILE) || defined(PROTO)
982/*
983 * realloc() with memory profiling.
984 */
985 void *
986mem_realloc(ptr, size)
987 void *ptr;
988 size_t size;
989{
990 void *p;
991
992 mem_pre_free(&ptr);
993 mem_pre_alloc_s(&size);
994
995 p = realloc(ptr, size);
996
997 mem_post_alloc(&p, size);
998
999 return p;
1000}
1001#endif
1002
1003/*
1004* Avoid repeating the error message many times (they take 1 second each).
1005* Did_outofmem_msg is reset when a character is read.
1006*/
1007 void
1008do_outofmem_msg(size)
1009 long_u size;
1010{
1011 if (!did_outofmem_msg)
1012 {
1013 /* Don't hide this message */
1014 emsg_silent = 0;
Bram Moolenaar79739e12011-10-26 11:41:00 +02001015
1016 /* Must come first to avoid coming back here when printing the error
1017 * message fails, e.g. when setting v:errmsg. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001018 did_outofmem_msg = TRUE;
Bram Moolenaar79739e12011-10-26 11:41:00 +02001019
1020 EMSGN(_("E342: Out of memory! (allocating %lu bytes)"), size);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001021 }
1022}
1023
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001024#if defined(EXITFREE) || defined(PROTO)
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001025
1026# if defined(FEAT_SEARCHPATH)
1027static void free_findfile __ARGS((void));
1028# endif
1029
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
1038free_all_mem()
1039{
1040 buf_T *buf, *nextbuf;
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001041 static int entered = FALSE;
1042
1043 /* When we cause a crash here it is caught and Vim tries to exit cleanly.
1044 * Don't try freeing everything again. */
1045 if (entered)
1046 return;
1047 entered = TRUE;
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001048
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001049# ifdef FEAT_AUTOCMD
Bram Moolenaar78ab3312007-09-29 12:16:41 +00001050 block_autocmds(); /* don't want to trigger autocommands here */
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001051# endif
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001052
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001053# ifdef FEAT_WINDOWS
Bram Moolenaar5bedfc62010-07-20 22:30:01 +02001054 /* Close all tabs and windows. Reset 'equalalways' to avoid redraws. */
1055 p_ea = FALSE;
Bram Moolenaar7d47b6e2006-03-15 22:59:18 +00001056 if (first_tabpage->tp_next != NULL)
1057 do_cmdline_cmd((char_u *)"tabonly!");
1058 if (firstwin != lastwin)
1059 do_cmdline_cmd((char_u *)"only!");
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001060# endif
Bram Moolenaar997fb4b2006-02-17 21:53:23 +00001061
Bram Moolenaarc4956c82006-03-12 21:58:43 +00001062# if defined(FEAT_SPELL)
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001063 /* Free all spell info. */
1064 spell_free_all();
1065# endif
1066
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001067# if defined(FEAT_USR_CMDS)
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001068 /* Clear user commands (before deleting buffers). */
1069 ex_comclear(NULL);
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001070# endif
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001071
1072# ifdef FEAT_MENU
1073 /* Clear menus. */
1074 do_cmdline_cmd((char_u *)"aunmenu *");
Bram Moolenaar21160b92009-11-11 15:56:10 +00001075# ifdef FEAT_MULTI_LANG
1076 do_cmdline_cmd((char_u *)"menutranslate clear");
1077# endif
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001078# endif
1079
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001080 /* Clear mappings, abbreviations, breakpoints. */
Bram Moolenaar21160b92009-11-11 15:56:10 +00001081 do_cmdline_cmd((char_u *)"lmapclear");
1082 do_cmdline_cmd((char_u *)"xmapclear");
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001083 do_cmdline_cmd((char_u *)"mapclear");
1084 do_cmdline_cmd((char_u *)"mapclear!");
1085 do_cmdline_cmd((char_u *)"abclear");
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001086# if defined(FEAT_EVAL)
1087 do_cmdline_cmd((char_u *)"breakdel *");
1088# endif
Bram Moolenaar1e498f52005-06-26 22:29:44 +00001089# if defined(FEAT_PROFILE)
1090 do_cmdline_cmd((char_u *)"profdel *");
1091# endif
Bram Moolenaarde0dfed2009-02-24 03:30:14 +00001092# if defined(FEAT_KEYMAP)
1093 do_cmdline_cmd((char_u *)"set keymap=");
1094#endif
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001095
1096# ifdef FEAT_TITLE
1097 free_titles();
1098# endif
1099# if defined(FEAT_SEARCHPATH)
1100 free_findfile();
1101# endif
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001102
1103 /* Obviously named calls. */
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001104# if defined(FEAT_AUTOCMD)
1105 free_all_autocmds();
1106# endif
1107 clear_termcodes();
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001108 free_all_options();
1109 free_all_marks();
1110 alist_clear(&global_alist);
1111 free_homedir();
1112 free_search_patterns();
1113 free_old_sub();
1114 free_last_insert();
1115 free_prev_shellcmd();
1116 free_regexp_stuff();
1117 free_tag_stuff();
1118 free_cd_dir();
Bram Moolenaarde0dfed2009-02-24 03:30:14 +00001119# ifdef FEAT_SIGNS
1120 free_signs();
1121# endif
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001122# ifdef FEAT_EVAL
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001123 set_expr_line(NULL);
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001124# endif
1125# ifdef FEAT_DIFF
Bram Moolenaar997fb4b2006-02-17 21:53:23 +00001126 diff_clear(curtab);
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001127# endif
Bram Moolenaara40ceaf2006-01-13 22:35:40 +00001128 clear_sb_text(); /* free any scrollback text */
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001129
1130 /* Free some global vars. */
1131 vim_free(username);
Bram Moolenaaraf92ee82007-10-07 13:45:08 +00001132# ifdef FEAT_CLIPBOARD
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001133 vim_free(clip_exclude_prog);
Bram Moolenaaraf92ee82007-10-07 13:45:08 +00001134# endif
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001135 vim_free(last_cmdline);
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001136# ifdef FEAT_CMDHIST
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001137 vim_free(new_last_cmdline);
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001138# endif
Bram Moolenaar238a5642006-02-21 22:12:05 +00001139 set_keep_msg(NULL, 0);
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001140 vim_free(ff_expand_buffer);
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001141
1142 /* Clear cmdline history. */
1143 p_hi = 0;
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001144# ifdef FEAT_CMDHIST
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001145 init_history();
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001146# endif
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001147
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001148#ifdef FEAT_QUICKFIX
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001149 {
Bram Moolenaarbd1e5d22009-04-29 09:02:44 +00001150 win_T *win;
1151 tabpage_T *tab;
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001152
1153 qf_free_all(NULL);
1154 /* Free all location lists */
Bram Moolenaarbd1e5d22009-04-29 09:02:44 +00001155 FOR_ALL_TAB_WINDOWS(tab, win)
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001156 qf_free_all(win);
1157 }
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001158#endif
1159
1160 /* Close all script inputs. */
1161 close_all_scripts();
1162
1163#if defined(FEAT_WINDOWS)
1164 /* Destroy all windows. Must come before freeing buffers. */
1165 win_free_all();
1166#endif
1167
Bram Moolenaar2a329742008-04-01 12:53:43 +00001168 /* Free all buffers. Reset 'autochdir' to avoid accessing things that
1169 * were freed already. */
1170#ifdef FEAT_AUTOCHDIR
1171 p_acd = FALSE;
1172#endif
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001173 for (buf = firstbuf; buf != NULL; )
1174 {
1175 nextbuf = buf->b_next;
1176 close_buffer(NULL, buf, DOBUF_WIPE);
1177 if (buf_valid(buf))
1178 buf = nextbuf; /* didn't work, try next one */
1179 else
1180 buf = firstbuf;
1181 }
1182
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001183#ifdef FEAT_ARABIC
1184 free_cmdline_buf();
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001185#endif
1186
1187 /* Clear registers. */
1188 clear_registers();
1189 ResetRedobuff();
1190 ResetRedobuff();
1191
Bram Moolenaar85c79d32007-02-20 01:59:20 +00001192#if defined(FEAT_CLIENTSERVER) && defined(FEAT_X11)
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001193 vim_free(serverDelayedStartName);
1194#endif
1195
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001196 /* highlight info */
1197 free_highlight();
1198
Bram Moolenaar1e498f52005-06-26 22:29:44 +00001199 reset_last_sourcing();
1200
Bram Moolenaar7d47b6e2006-03-15 22:59:18 +00001201#ifdef FEAT_WINDOWS
Bram Moolenaar06a89a52006-04-29 22:01:03 +00001202 free_tabpage(first_tabpage);
1203 first_tabpage = NULL;
Bram Moolenaar7d47b6e2006-03-15 22:59:18 +00001204#endif
1205
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001206# ifdef UNIX
1207 /* Machine-specific free. */
1208 mch_free_mem();
1209# endif
1210
1211 /* message history */
1212 for (;;)
1213 if (delete_first_msg() == FAIL)
1214 break;
1215
1216# ifdef FEAT_EVAL
1217 eval_clear();
1218# endif
1219
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001220 free_termoptions();
1221
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001222 /* screenlines (can't display anything now!) */
1223 free_screenlines();
1224
1225#if defined(USE_XSMP)
1226 xsmp_close();
1227#endif
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001228#ifdef FEAT_GUI_GTK
1229 gui_mch_free_all();
1230#endif
1231 clear_hl_tables();
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001232
1233 vim_free(IObuff);
1234 vim_free(NameBuff);
1235}
1236#endif
1237
Bram Moolenaar071d4272004-06-13 20:20:40 +00001238/*
Bram Moolenaare980d8a2010-12-08 13:11:21 +01001239 * Copy "string" into newly allocated memory.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001240 */
1241 char_u *
1242vim_strsave(string)
1243 char_u *string;
1244{
1245 char_u *p;
1246 unsigned len;
1247
1248 len = (unsigned)STRLEN(string) + 1;
1249 p = alloc(len);
1250 if (p != NULL)
1251 mch_memmove(p, string, (size_t)len);
1252 return p;
1253}
1254
Bram Moolenaare980d8a2010-12-08 13:11:21 +01001255/*
1256 * Copy up to "len" bytes of "string" into newly allocated memory and
1257 * terminate with a NUL.
1258 * The allocated memory always has size "len + 1", also when "string" is
1259 * shorter.
1260 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001261 char_u *
1262vim_strnsave(string, len)
1263 char_u *string;
1264 int len;
1265{
1266 char_u *p;
1267
1268 p = alloc((unsigned)(len + 1));
1269 if (p != NULL)
1270 {
1271 STRNCPY(p, string, len);
1272 p[len] = NUL;
1273 }
1274 return p;
1275}
1276
Bram Moolenaar071d4272004-06-13 20:20:40 +00001277/*
1278 * Same as vim_strsave(), but any characters found in esc_chars are preceded
1279 * by a backslash.
1280 */
1281 char_u *
1282vim_strsave_escaped(string, esc_chars)
1283 char_u *string;
1284 char_u *esc_chars;
1285{
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001286 return vim_strsave_escaped_ext(string, esc_chars, '\\', FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001287}
1288
1289/*
1290 * Same as vim_strsave_escaped(), but when "bsl" is TRUE also escape
1291 * characters where rem_backslash() would remove the backslash.
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001292 * Escape the characters with "cc".
Bram Moolenaar071d4272004-06-13 20:20:40 +00001293 */
1294 char_u *
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001295vim_strsave_escaped_ext(string, esc_chars, cc, bsl)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001296 char_u *string;
1297 char_u *esc_chars;
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001298 int cc;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001299 int bsl;
1300{
1301 char_u *p;
1302 char_u *p2;
1303 char_u *escaped_string;
1304 unsigned length;
1305#ifdef FEAT_MBYTE
1306 int l;
1307#endif
1308
1309 /*
1310 * First count the number of backslashes required.
1311 * Then allocate the memory and insert them.
1312 */
1313 length = 1; /* count the trailing NUL */
1314 for (p = string; *p; p++)
1315 {
1316#ifdef FEAT_MBYTE
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001317 if (has_mbyte && (l = (*mb_ptr2len)(p)) > 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001318 {
1319 length += l; /* count a multibyte char */
1320 p += l - 1;
1321 continue;
1322 }
1323#endif
1324 if (vim_strchr(esc_chars, *p) != NULL || (bsl && rem_backslash(p)))
1325 ++length; /* count a backslash */
1326 ++length; /* count an ordinary char */
1327 }
1328 escaped_string = alloc(length);
1329 if (escaped_string != NULL)
1330 {
1331 p2 = escaped_string;
1332 for (p = string; *p; p++)
1333 {
1334#ifdef FEAT_MBYTE
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001335 if (has_mbyte && (l = (*mb_ptr2len)(p)) > 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001336 {
1337 mch_memmove(p2, p, (size_t)l);
1338 p2 += l;
1339 p += l - 1; /* skip multibyte char */
1340 continue;
1341 }
1342#endif
1343 if (vim_strchr(esc_chars, *p) != NULL || (bsl && rem_backslash(p)))
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001344 *p2++ = cc;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001345 *p2++ = *p;
1346 }
1347 *p2 = NUL;
1348 }
1349 return escaped_string;
1350}
1351
Bram Moolenaar7693ec62008-07-24 18:29:37 +00001352/*
1353 * Return TRUE when 'shell' has "csh" in the tail.
1354 */
1355 int
1356csh_like_shell()
1357{
1358 return (strstr((char *)gettail(p_sh), "csh") != NULL);
1359}
Bram Moolenaar7693ec62008-07-24 18:29:37 +00001360
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001361/*
1362 * Escape "string" for use as a shell argument with system().
Bram Moolenaar21160b92009-11-11 15:56:10 +00001363 * This uses single quotes, except when we know we need to use double quotes
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001364 * (MS-DOS and MS-Windows without 'shellslash' set).
Bram Moolenaarffd66c42008-07-16 20:43:37 +00001365 * Escape a newline, depending on the 'shell' option.
1366 * When "do_special" is TRUE also replace "!", "%", "#" and things starting
1367 * with "<" like "<cfile>".
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001368 * Returns the result in allocated memory, NULL if we have run out.
1369 */
1370 char_u *
Bram Moolenaar05bb9532008-07-04 09:44:11 +00001371vim_strsave_shellescape(string, do_special)
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001372 char_u *string;
Bram Moolenaar05bb9532008-07-04 09:44:11 +00001373 int do_special;
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001374{
1375 unsigned length;
1376 char_u *p;
1377 char_u *d;
1378 char_u *escaped_string;
Bram Moolenaar05bb9532008-07-04 09:44:11 +00001379 int l;
Bram Moolenaarffd66c42008-07-16 20:43:37 +00001380 int csh_like;
1381
1382 /* Only csh and similar shells expand '!' within single quotes. For sh and
1383 * the like we must not put a backslash before it, it will be taken
1384 * literally. If do_special is set the '!' will be escaped twice.
1385 * Csh also needs to have "\n" escaped twice when do_special is set. */
Bram Moolenaar7693ec62008-07-24 18:29:37 +00001386 csh_like = csh_like_shell();
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001387
1388 /* First count the number of extra bytes required. */
Bram Moolenaar77f66d62007-02-20 02:16:18 +00001389 length = (unsigned)STRLEN(string) + 3; /* two quotes and a trailing NUL */
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001390 for (p = string; *p != NUL; mb_ptr_adv(p))
1391 {
1392# if defined(WIN32) || defined(WIN16) || defined(DOS)
1393 if (!p_ssl)
1394 {
1395 if (*p == '"')
1396 ++length; /* " -> "" */
1397 }
1398 else
1399# endif
1400 if (*p == '\'')
1401 length += 3; /* ' => '\'' */
Bram Moolenaarffd66c42008-07-16 20:43:37 +00001402 if (*p == '\n' || (*p == '!' && (csh_like || do_special)))
1403 {
1404 ++length; /* insert backslash */
1405 if (csh_like && do_special)
1406 ++length; /* insert backslash */
1407 }
Bram Moolenaar05bb9532008-07-04 09:44:11 +00001408 if (do_special && find_cmdline_var(p, &l) >= 0)
1409 {
1410 ++length; /* insert backslash */
1411 p += l - 1;
1412 }
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001413 }
1414
1415 /* Allocate memory for the result and fill it. */
1416 escaped_string = alloc(length);
1417 if (escaped_string != NULL)
1418 {
1419 d = escaped_string;
1420
1421 /* add opening quote */
1422# if defined(WIN32) || defined(WIN16) || defined(DOS)
1423 if (!p_ssl)
1424 *d++ = '"';
1425 else
1426# endif
1427 *d++ = '\'';
1428
1429 for (p = string; *p != NUL; )
1430 {
1431# if defined(WIN32) || defined(WIN16) || defined(DOS)
1432 if (!p_ssl)
1433 {
1434 if (*p == '"')
1435 {
1436 *d++ = '"';
1437 *d++ = '"';
1438 ++p;
1439 continue;
1440 }
1441 }
1442 else
1443# endif
1444 if (*p == '\'')
1445 {
Bram Moolenaar05bb9532008-07-04 09:44:11 +00001446 *d++ = '\'';
1447 *d++ = '\\';
1448 *d++ = '\'';
1449 *d++ = '\'';
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001450 ++p;
1451 continue;
1452 }
Bram Moolenaarffd66c42008-07-16 20:43:37 +00001453 if (*p == '\n' || (*p == '!' && (csh_like || do_special)))
1454 {
1455 *d++ = '\\';
1456 if (csh_like && do_special)
1457 *d++ = '\\';
1458 *d++ = *p++;
1459 continue;
1460 }
Bram Moolenaar05bb9532008-07-04 09:44:11 +00001461 if (do_special && find_cmdline_var(p, &l) >= 0)
1462 {
1463 *d++ = '\\'; /* insert backslash */
1464 while (--l >= 0) /* copy the var */
1465 *d++ = *p++;
Bram Moolenaar099d01d2009-11-25 16:14:45 +00001466 continue;
Bram Moolenaar05bb9532008-07-04 09:44:11 +00001467 }
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001468
1469 MB_COPY_CHAR(p, d);
1470 }
1471
1472 /* add terminating quote and finish with a NUL */
1473# if defined(WIN32) || defined(WIN16) || defined(DOS)
1474 if (!p_ssl)
1475 *d++ = '"';
1476 else
1477# endif
1478 *d++ = '\'';
1479 *d = NUL;
1480 }
1481
1482 return escaped_string;
1483}
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001484
Bram Moolenaar071d4272004-06-13 20:20:40 +00001485/*
1486 * Like vim_strsave(), but make all characters uppercase.
1487 * This uses ASCII lower-to-upper case translation, language independent.
1488 */
1489 char_u *
1490vim_strsave_up(string)
1491 char_u *string;
1492{
1493 char_u *p1;
1494
1495 p1 = vim_strsave(string);
1496 vim_strup(p1);
1497 return p1;
1498}
1499
1500/*
1501 * Like vim_strnsave(), but make all characters uppercase.
1502 * This uses ASCII lower-to-upper case translation, language independent.
1503 */
1504 char_u *
1505vim_strnsave_up(string, len)
1506 char_u *string;
1507 int len;
1508{
1509 char_u *p1;
1510
1511 p1 = vim_strnsave(string, len);
1512 vim_strup(p1);
1513 return p1;
1514}
1515
1516/*
1517 * ASCII lower-to-upper case translation, language independent.
1518 */
1519 void
1520vim_strup(p)
1521 char_u *p;
1522{
1523 char_u *p2;
1524 int c;
1525
1526 if (p != NULL)
1527 {
1528 p2 = p;
1529 while ((c = *p2) != NUL)
1530#ifdef EBCDIC
1531 *p2++ = isalpha(c) ? toupper(c) : c;
1532#else
1533 *p2++ = (c < 'a' || c > 'z') ? c : (c - 0x20);
1534#endif
1535 }
1536}
1537
Bram Moolenaarc4956c82006-03-12 21:58:43 +00001538#if defined(FEAT_EVAL) || defined(FEAT_SPELL) || defined(PROTO)
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001539/*
1540 * Make string "s" all upper-case and return it in allocated memory.
1541 * Handles multi-byte characters as well as possible.
1542 * Returns NULL when out of memory.
1543 */
1544 char_u *
1545strup_save(orig)
1546 char_u *orig;
1547{
1548 char_u *p;
1549 char_u *res;
1550
1551 res = p = vim_strsave(orig);
1552
1553 if (res != NULL)
1554 while (*p != NUL)
1555 {
1556# ifdef FEAT_MBYTE
1557 int l;
1558
1559 if (enc_utf8)
1560 {
1561 int c, uc;
Bram Moolenaar70b2a562012-01-10 22:26:17 +01001562 int newl;
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001563 char_u *s;
1564
1565 c = utf_ptr2char(p);
1566 uc = utf_toupper(c);
1567
1568 /* Reallocate string when byte count changes. This is rare,
1569 * thus it's OK to do another malloc()/free(). */
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001570 l = utf_ptr2len(p);
Bram Moolenaar70b2a562012-01-10 22:26:17 +01001571 newl = utf_char2len(uc);
1572 if (newl != l)
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001573 {
Bram Moolenaar70b2a562012-01-10 22:26:17 +01001574 s = alloc((unsigned)STRLEN(res) + 1 + newl - l);
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001575 if (s == NULL)
1576 break;
1577 mch_memmove(s, res, p - res);
Bram Moolenaar70b2a562012-01-10 22:26:17 +01001578 STRCPY(s + (p - res) + newl, p + l);
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001579 p = s + (p - res);
1580 vim_free(res);
1581 res = s;
1582 }
1583
1584 utf_char2bytes(uc, p);
Bram Moolenaar70b2a562012-01-10 22:26:17 +01001585 p += newl;
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001586 }
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001587 else if (has_mbyte && (l = (*mb_ptr2len)(p)) > 1)
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001588 p += l; /* skip multi-byte character */
1589 else
1590# endif
1591 {
1592 *p = TOUPPER_LOC(*p); /* note that toupper() can be a macro */
1593 p++;
1594 }
1595 }
1596
1597 return res;
1598}
1599#endif
1600
Bram Moolenaar071d4272004-06-13 20:20:40 +00001601/*
1602 * copy a space a number of times
1603 */
1604 void
1605copy_spaces(ptr, count)
1606 char_u *ptr;
1607 size_t count;
1608{
1609 size_t i = count;
1610 char_u *p = ptr;
1611
1612 while (i--)
1613 *p++ = ' ';
1614}
1615
1616#if defined(FEAT_VISUALEXTRA) || defined(PROTO)
1617/*
1618 * Copy a character a number of times.
Bram Moolenaar21160b92009-11-11 15:56:10 +00001619 * Does not work for multi-byte characters!
Bram Moolenaar071d4272004-06-13 20:20:40 +00001620 */
1621 void
1622copy_chars(ptr, count, c)
1623 char_u *ptr;
1624 size_t count;
1625 int c;
1626{
1627 size_t i = count;
1628 char_u *p = ptr;
1629
1630 while (i--)
1631 *p++ = c;
1632}
1633#endif
1634
1635/*
1636 * delete spaces at the end of a string
1637 */
1638 void
1639del_trailing_spaces(ptr)
1640 char_u *ptr;
1641{
1642 char_u *q;
1643
1644 q = ptr + STRLEN(ptr);
1645 while (--q > ptr && vim_iswhite(q[0]) && q[-1] != '\\' && q[-1] != Ctrl_V)
1646 *q = NUL;
1647}
1648
1649/*
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001650 * Like strncpy(), but always terminate the result with one NUL.
Bram Moolenaard042c562005-06-30 22:04:15 +00001651 * "to" must be "len + 1" long!
Bram Moolenaar071d4272004-06-13 20:20:40 +00001652 */
1653 void
1654vim_strncpy(to, from, len)
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001655 char_u *to;
1656 char_u *from;
Bram Moolenaarbbebc852005-07-18 21:47:53 +00001657 size_t len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001658{
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001659 STRNCPY(to, from, len);
1660 to[len] = NUL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001661}
1662
1663/*
Bram Moolenaaref9d6aa2011-04-11 16:56:35 +02001664 * Like strcat(), but make sure the result fits in "tosize" bytes and is
1665 * always NUL terminated.
1666 */
1667 void
1668vim_strcat(to, from, tosize)
1669 char_u *to;
1670 char_u *from;
1671 size_t tosize;
1672{
1673 size_t tolen = STRLEN(to);
1674 size_t fromlen = STRLEN(from);
1675
1676 if (tolen + fromlen + 1 > tosize)
1677 {
1678 mch_memmove(to + tolen, from, tosize - tolen - 1);
1679 to[tosize - 1] = NUL;
1680 }
1681 else
1682 STRCPY(to + tolen, from);
1683}
1684
1685/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00001686 * Isolate one part of a string option where parts are separated with
1687 * "sep_chars".
Bram Moolenaar83bab712005-08-01 21:58:57 +00001688 * The part is copied into "buf[maxlen]".
Bram Moolenaar071d4272004-06-13 20:20:40 +00001689 * "*option" is advanced to the next part.
1690 * The length is returned.
1691 */
1692 int
1693copy_option_part(option, buf, maxlen, sep_chars)
1694 char_u **option;
1695 char_u *buf;
1696 int maxlen;
1697 char *sep_chars;
1698{
1699 int len = 0;
1700 char_u *p = *option;
1701
1702 /* skip '.' at start of option part, for 'suffixes' */
1703 if (*p == '.')
1704 buf[len++] = *p++;
1705 while (*p != NUL && vim_strchr((char_u *)sep_chars, *p) == NULL)
1706 {
1707 /*
1708 * Skip backslash before a separator character and space.
1709 */
1710 if (p[0] == '\\' && vim_strchr((char_u *)sep_chars, p[1]) != NULL)
1711 ++p;
1712 if (len < maxlen - 1)
1713 buf[len++] = *p;
1714 ++p;
1715 }
1716 buf[len] = NUL;
1717
1718 if (*p != NUL && *p != ',') /* skip non-standard separator */
1719 ++p;
1720 p = skip_to_option_part(p); /* p points to next file name */
1721
1722 *option = p;
1723 return len;
1724}
1725
1726/*
Bram Moolenaar4770d092006-01-12 23:22:24 +00001727 * Replacement for free() that ignores NULL pointers.
1728 * Also skip free() when exiting for sure, this helps when we caught a deadly
1729 * signal that was caused by a crash in free().
Bram Moolenaar071d4272004-06-13 20:20:40 +00001730 */
1731 void
1732vim_free(x)
1733 void *x;
1734{
Bram Moolenaar4770d092006-01-12 23:22:24 +00001735 if (x != NULL && !really_exiting)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001736 {
1737#ifdef MEM_PROFILE
1738 mem_pre_free(&x);
1739#endif
1740 free(x);
1741 }
1742}
1743
1744#ifndef HAVE_MEMSET
1745 void *
1746vim_memset(ptr, c, size)
1747 void *ptr;
1748 int c;
1749 size_t size;
1750{
1751 char *p = ptr;
1752
1753 while (size-- > 0)
1754 *p++ = c;
1755 return ptr;
1756}
1757#endif
1758
1759#ifdef VIM_MEMCMP
1760/*
1761 * Return zero when "b1" and "b2" are the same for "len" bytes.
1762 * Return non-zero otherwise.
1763 */
1764 int
1765vim_memcmp(b1, b2, len)
1766 void *b1;
1767 void *b2;
1768 size_t len;
1769{
1770 char_u *p1 = (char_u *)b1, *p2 = (char_u *)b2;
1771
1772 for ( ; len > 0; --len)
1773 {
1774 if (*p1 != *p2)
1775 return 1;
1776 ++p1;
1777 ++p2;
1778 }
1779 return 0;
1780}
1781#endif
1782
1783#ifdef VIM_MEMMOVE
1784/*
1785 * Version of memmove() that handles overlapping source and destination.
1786 * For systems that don't have a function that is guaranteed to do that (SYSV).
1787 */
1788 void
1789mch_memmove(dst_arg, src_arg, len)
1790 void *src_arg, *dst_arg;
1791 size_t len;
1792{
1793 /*
1794 * A void doesn't have a size, we use char pointers.
1795 */
1796 char *dst = dst_arg, *src = src_arg;
1797
1798 /* overlap, copy backwards */
1799 if (dst > src && dst < src + len)
1800 {
1801 src += len;
1802 dst += len;
1803 while (len-- > 0)
1804 *--dst = *--src;
1805 }
1806 else /* copy forwards */
1807 while (len-- > 0)
1808 *dst++ = *src++;
1809}
1810#endif
1811
1812#if (!defined(HAVE_STRCASECMP) && !defined(HAVE_STRICMP)) || defined(PROTO)
1813/*
1814 * Compare two strings, ignoring case, using current locale.
1815 * Doesn't work for multi-byte characters.
1816 * return 0 for match, < 0 for smaller, > 0 for bigger
1817 */
1818 int
1819vim_stricmp(s1, s2)
1820 char *s1;
1821 char *s2;
1822{
1823 int i;
1824
1825 for (;;)
1826 {
1827 i = (int)TOLOWER_LOC(*s1) - (int)TOLOWER_LOC(*s2);
1828 if (i != 0)
1829 return i; /* this character different */
1830 if (*s1 == NUL)
1831 break; /* strings match until NUL */
1832 ++s1;
1833 ++s2;
1834 }
1835 return 0; /* strings match */
1836}
1837#endif
1838
1839#if (!defined(HAVE_STRNCASECMP) && !defined(HAVE_STRNICMP)) || defined(PROTO)
1840/*
1841 * Compare two strings, for length "len", ignoring case, using current locale.
1842 * Doesn't work for multi-byte characters.
1843 * return 0 for match, < 0 for smaller, > 0 for bigger
1844 */
1845 int
1846vim_strnicmp(s1, s2, len)
1847 char *s1;
1848 char *s2;
1849 size_t len;
1850{
1851 int i;
1852
1853 while (len > 0)
1854 {
1855 i = (int)TOLOWER_LOC(*s1) - (int)TOLOWER_LOC(*s2);
1856 if (i != 0)
1857 return i; /* this character different */
1858 if (*s1 == NUL)
1859 break; /* strings match until NUL */
1860 ++s1;
1861 ++s2;
1862 --len;
1863 }
1864 return 0; /* strings match */
1865}
1866#endif
1867
Bram Moolenaar071d4272004-06-13 20:20:40 +00001868/*
1869 * Version of strchr() and strrchr() that handle unsigned char strings
Bram Moolenaar05159a02005-02-26 23:04:13 +00001870 * with characters from 128 to 255 correctly. It also doesn't return a
1871 * pointer to the NUL at the end of the string.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001872 */
1873 char_u *
1874vim_strchr(string, c)
1875 char_u *string;
1876 int c;
1877{
1878 char_u *p;
1879 int b;
1880
1881 p = string;
1882#ifdef FEAT_MBYTE
1883 if (enc_utf8 && c >= 0x80)
1884 {
1885 while (*p != NUL)
1886 {
1887 if (utf_ptr2char(p) == c)
1888 return p;
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001889 p += (*mb_ptr2len)(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001890 }
1891 return NULL;
1892 }
1893 if (enc_dbcs != 0 && c > 255)
1894 {
1895 int n2 = c & 0xff;
1896
1897 c = ((unsigned)c >> 8) & 0xff;
1898 while ((b = *p) != NUL)
1899 {
1900 if (b == c && p[1] == n2)
1901 return p;
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001902 p += (*mb_ptr2len)(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001903 }
1904 return NULL;
1905 }
1906 if (has_mbyte)
1907 {
1908 while ((b = *p) != NUL)
1909 {
1910 if (b == c)
1911 return p;
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001912 p += (*mb_ptr2len)(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001913 }
1914 return NULL;
1915 }
1916#endif
1917 while ((b = *p) != NUL)
1918 {
1919 if (b == c)
1920 return p;
1921 ++p;
1922 }
1923 return NULL;
1924}
1925
1926/*
Bram Moolenaar05159a02005-02-26 23:04:13 +00001927 * Version of strchr() that only works for bytes and handles unsigned char
1928 * strings with characters above 128 correctly. It also doesn't return a
1929 * pointer to the NUL at the end of the string.
1930 */
1931 char_u *
1932vim_strbyte(string, c)
1933 char_u *string;
1934 int c;
1935{
1936 char_u *p = string;
1937
1938 while (*p != NUL)
1939 {
1940 if (*p == c)
1941 return p;
1942 ++p;
1943 }
1944 return NULL;
1945}
1946
1947/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00001948 * Search for last occurrence of "c" in "string".
Bram Moolenaarac6e65f2005-08-29 22:25:38 +00001949 * Return NULL if not found.
Bram Moolenaar05159a02005-02-26 23:04:13 +00001950 * Does not handle multi-byte char for "c"!
Bram Moolenaar071d4272004-06-13 20:20:40 +00001951 */
1952 char_u *
1953vim_strrchr(string, c)
1954 char_u *string;
1955 int c;
1956{
1957 char_u *retval = NULL;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001958 char_u *p = string;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001959
Bram Moolenaar05159a02005-02-26 23:04:13 +00001960 while (*p)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001961 {
Bram Moolenaar05159a02005-02-26 23:04:13 +00001962 if (*p == c)
1963 retval = p;
1964 mb_ptr_adv(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001965 }
1966 return retval;
1967}
1968
1969/*
1970 * Vim's version of strpbrk(), in case it's missing.
1971 * Don't generate a prototype for this, causes problems when it's not used.
1972 */
1973#ifndef PROTO
1974# ifndef HAVE_STRPBRK
1975# ifdef vim_strpbrk
1976# undef vim_strpbrk
1977# endif
1978 char_u *
1979vim_strpbrk(s, charset)
1980 char_u *s;
1981 char_u *charset;
1982{
1983 while (*s)
1984 {
1985 if (vim_strchr(charset, *s) != NULL)
1986 return s;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001987 mb_ptr_adv(s);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001988 }
1989 return NULL;
1990}
1991# endif
1992#endif
1993
1994/*
1995 * Vim has its own isspace() function, because on some machines isspace()
1996 * can't handle characters above 128.
1997 */
1998 int
1999vim_isspace(x)
2000 int x;
2001{
2002 return ((x >= 9 && x <= 13) || x == ' ');
2003}
2004
2005/************************************************************************
Bram Moolenaar383f9bc2005-01-19 22:18:32 +00002006 * Functions for handling growing arrays.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002007 */
2008
2009/*
2010 * Clear an allocated growing array.
2011 */
2012 void
2013ga_clear(gap)
2014 garray_T *gap;
2015{
2016 vim_free(gap->ga_data);
2017 ga_init(gap);
2018}
2019
2020/*
2021 * Clear a growing array that contains a list of strings.
2022 */
2023 void
2024ga_clear_strings(gap)
2025 garray_T *gap;
2026{
2027 int i;
2028
2029 for (i = 0; i < gap->ga_len; ++i)
2030 vim_free(((char_u **)(gap->ga_data))[i]);
2031 ga_clear(gap);
2032}
2033
2034/*
2035 * Initialize a growing array. Don't forget to set ga_itemsize and
2036 * ga_growsize! Or use ga_init2().
2037 */
2038 void
2039ga_init(gap)
2040 garray_T *gap;
2041{
2042 gap->ga_data = NULL;
Bram Moolenaar86b68352004-12-27 21:59:20 +00002043 gap->ga_maxlen = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002044 gap->ga_len = 0;
2045}
2046
2047 void
2048ga_init2(gap, itemsize, growsize)
2049 garray_T *gap;
2050 int itemsize;
2051 int growsize;
2052{
2053 ga_init(gap);
2054 gap->ga_itemsize = itemsize;
2055 gap->ga_growsize = growsize;
2056}
2057
2058/*
2059 * Make room in growing array "gap" for at least "n" items.
2060 * Return FAIL for failure, OK otherwise.
2061 */
2062 int
2063ga_grow(gap, n)
2064 garray_T *gap;
2065 int n;
2066{
2067 size_t len;
2068 char_u *pp;
2069
Bram Moolenaar86b68352004-12-27 21:59:20 +00002070 if (gap->ga_maxlen - gap->ga_len < n)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002071 {
2072 if (n < gap->ga_growsize)
2073 n = gap->ga_growsize;
2074 len = gap->ga_itemsize * (gap->ga_len + n);
2075 pp = alloc_clear((unsigned)len);
2076 if (pp == NULL)
2077 return FAIL;
Bram Moolenaar86b68352004-12-27 21:59:20 +00002078 gap->ga_maxlen = gap->ga_len + n;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002079 if (gap->ga_data != NULL)
2080 {
2081 mch_memmove(pp, gap->ga_data,
2082 (size_t)(gap->ga_itemsize * gap->ga_len));
2083 vim_free(gap->ga_data);
2084 }
2085 gap->ga_data = pp;
2086 }
2087 return OK;
2088}
2089
2090/*
Bram Moolenaar80a7dcf2010-08-04 17:07:20 +02002091 * For a growing array that contains a list of strings: concatenate all the
2092 * strings with a separating comma.
2093 * Returns NULL when out of memory.
2094 */
2095 char_u *
2096ga_concat_strings(gap)
2097 garray_T *gap;
2098{
2099 int i;
2100 int len = 0;
2101 char_u *s;
2102
2103 for (i = 0; i < gap->ga_len; ++i)
2104 len += (int)STRLEN(((char_u **)(gap->ga_data))[i]) + 1;
2105
2106 s = alloc(len + 1);
2107 if (s != NULL)
2108 {
2109 *s = NUL;
2110 for (i = 0; i < gap->ga_len; ++i)
2111 {
2112 if (*s != NUL)
2113 STRCAT(s, ",");
2114 STRCAT(s, ((char_u **)(gap->ga_data))[i]);
2115 }
2116 }
2117 return s;
2118}
2119
2120/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002121 * Concatenate a string to a growarray which contains characters.
2122 * Note: Does NOT copy the NUL at the end!
2123 */
2124 void
2125ga_concat(gap, s)
2126 garray_T *gap;
2127 char_u *s;
2128{
2129 int len = (int)STRLEN(s);
2130
2131 if (ga_grow(gap, len) == OK)
2132 {
2133 mch_memmove((char *)gap->ga_data + gap->ga_len, s, (size_t)len);
2134 gap->ga_len += len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002135 }
2136}
2137
2138/*
2139 * Append one byte to a growarray which contains bytes.
2140 */
2141 void
2142ga_append(gap, c)
2143 garray_T *gap;
2144 int c;
2145{
2146 if (ga_grow(gap, 1) == OK)
2147 {
2148 *((char *)gap->ga_data + gap->ga_len) = c;
2149 ++gap->ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002150 }
2151}
2152
Bram Moolenaar4b9669f2011-07-07 16:20:52 +02002153#if (defined(UNIX) && !defined(USE_SYSTEM)) || defined(WIN3264)
2154/*
2155 * Append the text in "gap" below the cursor line and clear "gap".
2156 */
2157 void
2158append_ga_line(gap)
2159 garray_T *gap;
2160{
2161 /* Remove trailing CR. */
2162 if (gap->ga_len > 0
2163 && !curbuf->b_p_bin
2164 && ((char_u *)gap->ga_data)[gap->ga_len - 1] == CAR)
2165 --gap->ga_len;
2166 ga_append(gap, NUL);
2167 ml_append(curwin->w_cursor.lnum++, gap->ga_data, 0, FALSE);
2168 gap->ga_len = 0;
2169}
2170#endif
2171
Bram Moolenaar071d4272004-06-13 20:20:40 +00002172/************************************************************************
2173 * functions that use lookup tables for various things, generally to do with
2174 * special key codes.
2175 */
2176
2177/*
2178 * Some useful tables.
2179 */
2180
2181static struct modmasktable
2182{
2183 short mod_mask; /* Bit-mask for particular key modifier */
2184 short mod_flag; /* Bit(s) for particular key modifier */
2185 char_u name; /* Single letter name of modifier */
2186} mod_mask_table[] =
2187{
2188 {MOD_MASK_ALT, MOD_MASK_ALT, (char_u)'M'},
Bram Moolenaar19a09a12005-03-04 23:39:37 +00002189 {MOD_MASK_META, MOD_MASK_META, (char_u)'T'},
Bram Moolenaar071d4272004-06-13 20:20:40 +00002190 {MOD_MASK_CTRL, MOD_MASK_CTRL, (char_u)'C'},
2191 {MOD_MASK_SHIFT, MOD_MASK_SHIFT, (char_u)'S'},
2192 {MOD_MASK_MULTI_CLICK, MOD_MASK_2CLICK, (char_u)'2'},
2193 {MOD_MASK_MULTI_CLICK, MOD_MASK_3CLICK, (char_u)'3'},
2194 {MOD_MASK_MULTI_CLICK, MOD_MASK_4CLICK, (char_u)'4'},
2195#ifdef MACOS
2196 {MOD_MASK_CMD, MOD_MASK_CMD, (char_u)'D'},
2197#endif
2198 /* 'A' must be the last one */
2199 {MOD_MASK_ALT, MOD_MASK_ALT, (char_u)'A'},
2200 {0, 0, NUL}
2201};
2202
2203/*
2204 * Shifted key terminal codes and their unshifted equivalent.
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00002205 * Don't add mouse codes here, they are handled separately!
Bram Moolenaar071d4272004-06-13 20:20:40 +00002206 */
2207#define MOD_KEYS_ENTRY_SIZE 5
2208
2209static char_u modifier_keys_table[] =
2210{
2211/* mod mask with modifier without modifier */
2212 MOD_MASK_SHIFT, '&', '9', '@', '1', /* begin */
2213 MOD_MASK_SHIFT, '&', '0', '@', '2', /* cancel */
2214 MOD_MASK_SHIFT, '*', '1', '@', '4', /* command */
2215 MOD_MASK_SHIFT, '*', '2', '@', '5', /* copy */
2216 MOD_MASK_SHIFT, '*', '3', '@', '6', /* create */
2217 MOD_MASK_SHIFT, '*', '4', 'k', 'D', /* delete char */
2218 MOD_MASK_SHIFT, '*', '5', 'k', 'L', /* delete line */
2219 MOD_MASK_SHIFT, '*', '7', '@', '7', /* end */
2220 MOD_MASK_CTRL, KS_EXTRA, (int)KE_C_END, '@', '7', /* end */
2221 MOD_MASK_SHIFT, '*', '9', '@', '9', /* exit */
2222 MOD_MASK_SHIFT, '*', '0', '@', '0', /* find */
2223 MOD_MASK_SHIFT, '#', '1', '%', '1', /* help */
2224 MOD_MASK_SHIFT, '#', '2', 'k', 'h', /* home */
2225 MOD_MASK_CTRL, KS_EXTRA, (int)KE_C_HOME, 'k', 'h', /* home */
2226 MOD_MASK_SHIFT, '#', '3', 'k', 'I', /* insert */
2227 MOD_MASK_SHIFT, '#', '4', 'k', 'l', /* left arrow */
2228 MOD_MASK_CTRL, KS_EXTRA, (int)KE_C_LEFT, 'k', 'l', /* left arrow */
2229 MOD_MASK_SHIFT, '%', 'a', '%', '3', /* message */
2230 MOD_MASK_SHIFT, '%', 'b', '%', '4', /* move */
2231 MOD_MASK_SHIFT, '%', 'c', '%', '5', /* next */
2232 MOD_MASK_SHIFT, '%', 'd', '%', '7', /* options */
2233 MOD_MASK_SHIFT, '%', 'e', '%', '8', /* previous */
2234 MOD_MASK_SHIFT, '%', 'f', '%', '9', /* print */
2235 MOD_MASK_SHIFT, '%', 'g', '%', '0', /* redo */
2236 MOD_MASK_SHIFT, '%', 'h', '&', '3', /* replace */
2237 MOD_MASK_SHIFT, '%', 'i', 'k', 'r', /* right arr. */
2238 MOD_MASK_CTRL, KS_EXTRA, (int)KE_C_RIGHT, 'k', 'r', /* right arr. */
2239 MOD_MASK_SHIFT, '%', 'j', '&', '5', /* resume */
2240 MOD_MASK_SHIFT, '!', '1', '&', '6', /* save */
2241 MOD_MASK_SHIFT, '!', '2', '&', '7', /* suspend */
2242 MOD_MASK_SHIFT, '!', '3', '&', '8', /* undo */
2243 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_UP, 'k', 'u', /* up arrow */
2244 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_DOWN, 'k', 'd', /* down arrow */
2245
2246 /* vt100 F1 */
2247 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_XF1, KS_EXTRA, (int)KE_XF1,
2248 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_XF2, KS_EXTRA, (int)KE_XF2,
2249 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_XF3, KS_EXTRA, (int)KE_XF3,
2250 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_XF4, KS_EXTRA, (int)KE_XF4,
2251
2252 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F1, 'k', '1', /* F1 */
2253 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F2, 'k', '2',
2254 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F3, 'k', '3',
2255 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F4, 'k', '4',
2256 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F5, 'k', '5',
2257 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F6, 'k', '6',
2258 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F7, 'k', '7',
2259 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F8, 'k', '8',
2260 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F9, 'k', '9',
2261 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F10, 'k', ';', /* F10 */
2262
2263 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F11, 'F', '1',
2264 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F12, 'F', '2',
2265 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F13, 'F', '3',
2266 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F14, 'F', '4',
2267 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F15, 'F', '5',
2268 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F16, 'F', '6',
2269 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F17, 'F', '7',
2270 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F18, 'F', '8',
2271 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F19, 'F', '9',
2272 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F20, 'F', 'A',
2273
2274 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F21, 'F', 'B',
2275 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F22, 'F', 'C',
2276 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F23, 'F', 'D',
2277 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F24, 'F', 'E',
2278 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F25, 'F', 'F',
2279 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F26, 'F', 'G',
2280 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F27, 'F', 'H',
2281 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F28, 'F', 'I',
2282 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F29, 'F', 'J',
2283 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F30, 'F', 'K',
2284
2285 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F31, 'F', 'L',
2286 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F32, 'F', 'M',
2287 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F33, 'F', 'N',
2288 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F34, 'F', 'O',
2289 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F35, 'F', 'P',
2290 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F36, 'F', 'Q',
2291 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F37, 'F', 'R',
2292
2293 /* TAB pseudo code*/
2294 MOD_MASK_SHIFT, 'k', 'B', KS_EXTRA, (int)KE_TAB,
2295
2296 NUL
2297};
2298
2299static struct key_name_entry
2300{
2301 int key; /* Special key code or ascii value */
2302 char_u *name; /* Name of key */
2303} key_names_table[] =
2304{
2305 {' ', (char_u *)"Space"},
2306 {TAB, (char_u *)"Tab"},
2307 {K_TAB, (char_u *)"Tab"},
2308 {NL, (char_u *)"NL"},
2309 {NL, (char_u *)"NewLine"}, /* Alternative name */
2310 {NL, (char_u *)"LineFeed"}, /* Alternative name */
2311 {NL, (char_u *)"LF"}, /* Alternative name */
2312 {CAR, (char_u *)"CR"},
2313 {CAR, (char_u *)"Return"}, /* Alternative name */
2314 {CAR, (char_u *)"Enter"}, /* Alternative name */
2315 {K_BS, (char_u *)"BS"},
2316 {K_BS, (char_u *)"BackSpace"}, /* Alternative name */
2317 {ESC, (char_u *)"Esc"},
2318 {CSI, (char_u *)"CSI"},
2319 {K_CSI, (char_u *)"xCSI"},
2320 {'|', (char_u *)"Bar"},
2321 {'\\', (char_u *)"Bslash"},
2322 {K_DEL, (char_u *)"Del"},
2323 {K_DEL, (char_u *)"Delete"}, /* Alternative name */
2324 {K_KDEL, (char_u *)"kDel"},
2325 {K_UP, (char_u *)"Up"},
2326 {K_DOWN, (char_u *)"Down"},
2327 {K_LEFT, (char_u *)"Left"},
2328 {K_RIGHT, (char_u *)"Right"},
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002329 {K_XUP, (char_u *)"xUp"},
2330 {K_XDOWN, (char_u *)"xDown"},
2331 {K_XLEFT, (char_u *)"xLeft"},
2332 {K_XRIGHT, (char_u *)"xRight"},
Bram Moolenaar071d4272004-06-13 20:20:40 +00002333
2334 {K_F1, (char_u *)"F1"},
2335 {K_F2, (char_u *)"F2"},
2336 {K_F3, (char_u *)"F3"},
2337 {K_F4, (char_u *)"F4"},
2338 {K_F5, (char_u *)"F5"},
2339 {K_F6, (char_u *)"F6"},
2340 {K_F7, (char_u *)"F7"},
2341 {K_F8, (char_u *)"F8"},
2342 {K_F9, (char_u *)"F9"},
2343 {K_F10, (char_u *)"F10"},
2344
2345 {K_F11, (char_u *)"F11"},
2346 {K_F12, (char_u *)"F12"},
2347 {K_F13, (char_u *)"F13"},
2348 {K_F14, (char_u *)"F14"},
2349 {K_F15, (char_u *)"F15"},
2350 {K_F16, (char_u *)"F16"},
2351 {K_F17, (char_u *)"F17"},
2352 {K_F18, (char_u *)"F18"},
2353 {K_F19, (char_u *)"F19"},
2354 {K_F20, (char_u *)"F20"},
2355
2356 {K_F21, (char_u *)"F21"},
2357 {K_F22, (char_u *)"F22"},
2358 {K_F23, (char_u *)"F23"},
2359 {K_F24, (char_u *)"F24"},
2360 {K_F25, (char_u *)"F25"},
2361 {K_F26, (char_u *)"F26"},
2362 {K_F27, (char_u *)"F27"},
2363 {K_F28, (char_u *)"F28"},
2364 {K_F29, (char_u *)"F29"},
2365 {K_F30, (char_u *)"F30"},
2366
2367 {K_F31, (char_u *)"F31"},
2368 {K_F32, (char_u *)"F32"},
2369 {K_F33, (char_u *)"F33"},
2370 {K_F34, (char_u *)"F34"},
2371 {K_F35, (char_u *)"F35"},
2372 {K_F36, (char_u *)"F36"},
2373 {K_F37, (char_u *)"F37"},
2374
2375 {K_XF1, (char_u *)"xF1"},
2376 {K_XF2, (char_u *)"xF2"},
2377 {K_XF3, (char_u *)"xF3"},
2378 {K_XF4, (char_u *)"xF4"},
2379
2380 {K_HELP, (char_u *)"Help"},
2381 {K_UNDO, (char_u *)"Undo"},
2382 {K_INS, (char_u *)"Insert"},
2383 {K_INS, (char_u *)"Ins"}, /* Alternative name */
2384 {K_KINS, (char_u *)"kInsert"},
2385 {K_HOME, (char_u *)"Home"},
2386 {K_KHOME, (char_u *)"kHome"},
2387 {K_XHOME, (char_u *)"xHome"},
Bram Moolenaar68b76a62005-03-25 21:53:48 +00002388 {K_ZHOME, (char_u *)"zHome"},
Bram Moolenaar071d4272004-06-13 20:20:40 +00002389 {K_END, (char_u *)"End"},
2390 {K_KEND, (char_u *)"kEnd"},
2391 {K_XEND, (char_u *)"xEnd"},
Bram Moolenaar68b76a62005-03-25 21:53:48 +00002392 {K_ZEND, (char_u *)"zEnd"},
Bram Moolenaar071d4272004-06-13 20:20:40 +00002393 {K_PAGEUP, (char_u *)"PageUp"},
2394 {K_PAGEDOWN, (char_u *)"PageDown"},
2395 {K_KPAGEUP, (char_u *)"kPageUp"},
2396 {K_KPAGEDOWN, (char_u *)"kPageDown"},
2397
2398 {K_KPLUS, (char_u *)"kPlus"},
2399 {K_KMINUS, (char_u *)"kMinus"},
2400 {K_KDIVIDE, (char_u *)"kDivide"},
2401 {K_KMULTIPLY, (char_u *)"kMultiply"},
2402 {K_KENTER, (char_u *)"kEnter"},
2403 {K_KPOINT, (char_u *)"kPoint"},
2404
2405 {K_K0, (char_u *)"k0"},
2406 {K_K1, (char_u *)"k1"},
2407 {K_K2, (char_u *)"k2"},
2408 {K_K3, (char_u *)"k3"},
2409 {K_K4, (char_u *)"k4"},
2410 {K_K5, (char_u *)"k5"},
2411 {K_K6, (char_u *)"k6"},
2412 {K_K7, (char_u *)"k7"},
2413 {K_K8, (char_u *)"k8"},
2414 {K_K9, (char_u *)"k9"},
2415
2416 {'<', (char_u *)"lt"},
2417
2418 {K_MOUSE, (char_u *)"Mouse"},
Bram Moolenaar5af7d712012-01-20 17:15:51 +01002419#ifdef FEAT_MOUSE_NET
Bram Moolenaar071d4272004-06-13 20:20:40 +00002420 {K_NETTERM_MOUSE, (char_u *)"NetMouse"},
Bram Moolenaar5af7d712012-01-20 17:15:51 +01002421#endif
2422#ifdef FEAT_MOUSE_DEC
Bram Moolenaar071d4272004-06-13 20:20:40 +00002423 {K_DEC_MOUSE, (char_u *)"DecMouse"},
Bram Moolenaar5af7d712012-01-20 17:15:51 +01002424#endif
2425#ifdef FEAT_MOUSE_JSB
Bram Moolenaar071d4272004-06-13 20:20:40 +00002426 {K_JSBTERM_MOUSE, (char_u *)"JsbMouse"},
Bram Moolenaar5af7d712012-01-20 17:15:51 +01002427#endif
2428#ifdef FEAT_MOUSE_PTERM
Bram Moolenaar071d4272004-06-13 20:20:40 +00002429 {K_PTERM_MOUSE, (char_u *)"PtermMouse"},
Bram Moolenaar5af7d712012-01-20 17:15:51 +01002430#endif
2431#ifdef FEAT_MOUSE_URXVT
2432 {K_URXVT_MOUSE, (char_u *)"UrxvtMouse"},
2433#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002434 {K_LEFTMOUSE, (char_u *)"LeftMouse"},
2435 {K_LEFTMOUSE_NM, (char_u *)"LeftMouseNM"},
2436 {K_LEFTDRAG, (char_u *)"LeftDrag"},
2437 {K_LEFTRELEASE, (char_u *)"LeftRelease"},
2438 {K_LEFTRELEASE_NM, (char_u *)"LeftReleaseNM"},
2439 {K_MIDDLEMOUSE, (char_u *)"MiddleMouse"},
2440 {K_MIDDLEDRAG, (char_u *)"MiddleDrag"},
2441 {K_MIDDLERELEASE, (char_u *)"MiddleRelease"},
2442 {K_RIGHTMOUSE, (char_u *)"RightMouse"},
2443 {K_RIGHTDRAG, (char_u *)"RightDrag"},
2444 {K_RIGHTRELEASE, (char_u *)"RightRelease"},
Bram Moolenaar8d9b40e2010-07-25 15:49:07 +02002445 {K_MOUSEDOWN, (char_u *)"ScrollWheelUp"},
2446 {K_MOUSEUP, (char_u *)"ScrollWheelDown"},
2447 {K_MOUSELEFT, (char_u *)"ScrollWheelRight"},
2448 {K_MOUSERIGHT, (char_u *)"ScrollWheelLeft"},
2449 {K_MOUSEDOWN, (char_u *)"MouseDown"}, /* OBSOLETE: Use */
2450 {K_MOUSEUP, (char_u *)"MouseUp"}, /* ScrollWheelXXX instead */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002451 {K_X1MOUSE, (char_u *)"X1Mouse"},
2452 {K_X1DRAG, (char_u *)"X1Drag"},
2453 {K_X1RELEASE, (char_u *)"X1Release"},
2454 {K_X2MOUSE, (char_u *)"X2Mouse"},
2455 {K_X2DRAG, (char_u *)"X2Drag"},
2456 {K_X2RELEASE, (char_u *)"X2Release"},
2457 {K_DROP, (char_u *)"Drop"},
2458 {K_ZERO, (char_u *)"Nul"},
2459#ifdef FEAT_EVAL
2460 {K_SNR, (char_u *)"SNR"},
2461#endif
2462 {K_PLUG, (char_u *)"Plug"},
2463 {0, NULL}
2464};
2465
2466#define KEY_NAMES_TABLE_LEN (sizeof(key_names_table) / sizeof(struct key_name_entry))
2467
2468#ifdef FEAT_MOUSE
2469static struct mousetable
2470{
2471 int pseudo_code; /* Code for pseudo mouse event */
2472 int button; /* Which mouse button is it? */
2473 int is_click; /* Is it a mouse button click event? */
2474 int is_drag; /* Is it a mouse drag event? */
2475} mouse_table[] =
2476{
2477 {(int)KE_LEFTMOUSE, MOUSE_LEFT, TRUE, FALSE},
2478#ifdef FEAT_GUI
2479 {(int)KE_LEFTMOUSE_NM, MOUSE_LEFT, TRUE, FALSE},
2480#endif
2481 {(int)KE_LEFTDRAG, MOUSE_LEFT, FALSE, TRUE},
2482 {(int)KE_LEFTRELEASE, MOUSE_LEFT, FALSE, FALSE},
2483#ifdef FEAT_GUI
2484 {(int)KE_LEFTRELEASE_NM, MOUSE_LEFT, FALSE, FALSE},
2485#endif
2486 {(int)KE_MIDDLEMOUSE, MOUSE_MIDDLE, TRUE, FALSE},
2487 {(int)KE_MIDDLEDRAG, MOUSE_MIDDLE, FALSE, TRUE},
2488 {(int)KE_MIDDLERELEASE, MOUSE_MIDDLE, FALSE, FALSE},
2489 {(int)KE_RIGHTMOUSE, MOUSE_RIGHT, TRUE, FALSE},
2490 {(int)KE_RIGHTDRAG, MOUSE_RIGHT, FALSE, TRUE},
2491 {(int)KE_RIGHTRELEASE, MOUSE_RIGHT, FALSE, FALSE},
2492 {(int)KE_X1MOUSE, MOUSE_X1, TRUE, FALSE},
2493 {(int)KE_X1DRAG, MOUSE_X1, FALSE, TRUE},
2494 {(int)KE_X1RELEASE, MOUSE_X1, FALSE, FALSE},
2495 {(int)KE_X2MOUSE, MOUSE_X2, TRUE, FALSE},
2496 {(int)KE_X2DRAG, MOUSE_X2, FALSE, TRUE},
2497 {(int)KE_X2RELEASE, MOUSE_X2, FALSE, FALSE},
2498 /* DRAG without CLICK */
2499 {(int)KE_IGNORE, MOUSE_RELEASE, FALSE, TRUE},
2500 /* RELEASE without CLICK */
2501 {(int)KE_IGNORE, MOUSE_RELEASE, FALSE, FALSE},
2502 {0, 0, 0, 0},
2503};
2504#endif /* FEAT_MOUSE */
2505
2506/*
2507 * Return the modifier mask bit (MOD_MASK_*) which corresponds to the given
2508 * modifier name ('S' for Shift, 'C' for Ctrl etc).
2509 */
2510 int
2511name_to_mod_mask(c)
2512 int c;
2513{
2514 int i;
2515
2516 c = TOUPPER_ASC(c);
2517 for (i = 0; mod_mask_table[i].mod_mask != 0; i++)
2518 if (c == mod_mask_table[i].name)
2519 return mod_mask_table[i].mod_flag;
2520 return 0;
2521}
2522
Bram Moolenaar071d4272004-06-13 20:20:40 +00002523/*
2524 * Check if if there is a special key code for "key" that includes the
2525 * modifiers specified.
2526 */
2527 int
2528simplify_key(key, modifiers)
2529 int key;
2530 int *modifiers;
2531{
2532 int i;
2533 int key0;
2534 int key1;
2535
2536 if (*modifiers & (MOD_MASK_SHIFT | MOD_MASK_CTRL | MOD_MASK_ALT))
2537 {
2538 /* TAB is a special case */
2539 if (key == TAB && (*modifiers & MOD_MASK_SHIFT))
2540 {
2541 *modifiers &= ~MOD_MASK_SHIFT;
2542 return K_S_TAB;
2543 }
2544 key0 = KEY2TERMCAP0(key);
2545 key1 = KEY2TERMCAP1(key);
2546 for (i = 0; modifier_keys_table[i] != NUL; i += MOD_KEYS_ENTRY_SIZE)
2547 if (key0 == modifier_keys_table[i + 3]
2548 && key1 == modifier_keys_table[i + 4]
2549 && (*modifiers & modifier_keys_table[i]))
2550 {
2551 *modifiers &= ~modifier_keys_table[i];
2552 return TERMCAP2KEY(modifier_keys_table[i + 1],
2553 modifier_keys_table[i + 2]);
2554 }
2555 }
2556 return key;
2557}
2558
2559/*
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002560 * Change <xHome> to <Home>, <xUp> to <Up>, etc.
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002561 */
2562 int
2563handle_x_keys(key)
2564 int key;
2565{
2566 switch (key)
2567 {
2568 case K_XUP: return K_UP;
2569 case K_XDOWN: return K_DOWN;
2570 case K_XLEFT: return K_LEFT;
2571 case K_XRIGHT: return K_RIGHT;
2572 case K_XHOME: return K_HOME;
Bram Moolenaar68b76a62005-03-25 21:53:48 +00002573 case K_ZHOME: return K_HOME;
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002574 case K_XEND: return K_END;
Bram Moolenaar68b76a62005-03-25 21:53:48 +00002575 case K_ZEND: return K_END;
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002576 case K_XF1: return K_F1;
2577 case K_XF2: return K_F2;
2578 case K_XF3: return K_F3;
2579 case K_XF4: return K_F4;
2580 case K_S_XF1: return K_S_F1;
2581 case K_S_XF2: return K_S_F2;
2582 case K_S_XF3: return K_S_F3;
2583 case K_S_XF4: return K_S_F4;
2584 }
2585 return key;
2586}
2587
2588/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002589 * Return a string which contains the name of the given key when the given
2590 * modifiers are down.
2591 */
2592 char_u *
2593get_special_key_name(c, modifiers)
2594 int c;
2595 int modifiers;
2596{
2597 static char_u string[MAX_KEY_NAME_LEN + 1];
2598
2599 int i, idx;
2600 int table_idx;
2601 char_u *s;
2602
2603 string[0] = '<';
2604 idx = 1;
2605
2606 /* Key that stands for a normal character. */
2607 if (IS_SPECIAL(c) && KEY2TERMCAP0(c) == KS_KEY)
2608 c = KEY2TERMCAP1(c);
2609
2610 /*
2611 * Translate shifted special keys into unshifted keys and set modifier.
2612 * Same for CTRL and ALT modifiers.
2613 */
2614 if (IS_SPECIAL(c))
2615 {
2616 for (i = 0; modifier_keys_table[i] != 0; i += MOD_KEYS_ENTRY_SIZE)
2617 if ( KEY2TERMCAP0(c) == (int)modifier_keys_table[i + 1]
2618 && (int)KEY2TERMCAP1(c) == (int)modifier_keys_table[i + 2])
2619 {
2620 modifiers |= modifier_keys_table[i];
2621 c = TERMCAP2KEY(modifier_keys_table[i + 3],
2622 modifier_keys_table[i + 4]);
2623 break;
2624 }
2625 }
2626
2627 /* try to find the key in the special key table */
2628 table_idx = find_special_key_in_table(c);
2629
2630 /*
2631 * When not a known special key, and not a printable character, try to
2632 * extract modifiers.
2633 */
2634 if (c > 0
2635#ifdef FEAT_MBYTE
2636 && (*mb_char2len)(c) == 1
2637#endif
2638 )
2639 {
2640 if (table_idx < 0
2641 && (!vim_isprintc(c) || (c & 0x7f) == ' ')
2642 && (c & 0x80))
2643 {
2644 c &= 0x7f;
2645 modifiers |= MOD_MASK_ALT;
2646 /* try again, to find the un-alted key in the special key table */
2647 table_idx = find_special_key_in_table(c);
2648 }
2649 if (table_idx < 0 && !vim_isprintc(c) && c < ' ')
2650 {
2651#ifdef EBCDIC
2652 c = CtrlChar(c);
2653#else
2654 c += '@';
2655#endif
2656 modifiers |= MOD_MASK_CTRL;
2657 }
2658 }
2659
2660 /* translate the modifier into a string */
2661 for (i = 0; mod_mask_table[i].name != 'A'; i++)
2662 if ((modifiers & mod_mask_table[i].mod_mask)
2663 == mod_mask_table[i].mod_flag)
2664 {
2665 string[idx++] = mod_mask_table[i].name;
2666 string[idx++] = (char_u)'-';
2667 }
2668
2669 if (table_idx < 0) /* unknown special key, may output t_xx */
2670 {
2671 if (IS_SPECIAL(c))
2672 {
2673 string[idx++] = 't';
2674 string[idx++] = '_';
2675 string[idx++] = KEY2TERMCAP0(c);
2676 string[idx++] = KEY2TERMCAP1(c);
2677 }
2678 /* Not a special key, only modifiers, output directly */
2679 else
2680 {
2681#ifdef FEAT_MBYTE
2682 if (has_mbyte && (*mb_char2len)(c) > 1)
2683 idx += (*mb_char2bytes)(c, string + idx);
2684 else
2685#endif
2686 if (vim_isprintc(c))
2687 string[idx++] = c;
2688 else
2689 {
2690 s = transchar(c);
2691 while (*s)
2692 string[idx++] = *s++;
2693 }
2694 }
2695 }
2696 else /* use name of special key */
2697 {
2698 STRCPY(string + idx, key_names_table[table_idx].name);
2699 idx = (int)STRLEN(string);
2700 }
2701 string[idx++] = '>';
2702 string[idx] = NUL;
2703 return string;
2704}
2705
2706/*
2707 * Try translating a <> name at (*srcp)[] to dst[].
2708 * Return the number of characters added to dst[], zero for no match.
2709 * If there is a match, srcp is advanced to after the <> name.
2710 * dst[] must be big enough to hold the result (up to six characters)!
2711 */
2712 int
2713trans_special(srcp, dst, keycode)
2714 char_u **srcp;
2715 char_u *dst;
2716 int keycode; /* prefer key code, e.g. K_DEL instead of DEL */
2717{
2718 int modifiers = 0;
2719 int key;
2720 int dlen = 0;
2721
Bram Moolenaar2a8ced02008-12-24 11:54:31 +00002722 key = find_special_key(srcp, &modifiers, keycode, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002723 if (key == 0)
2724 return 0;
2725
2726 /* Put the appropriate modifier in a string */
2727 if (modifiers != 0)
2728 {
2729 dst[dlen++] = K_SPECIAL;
2730 dst[dlen++] = KS_MODIFIER;
2731 dst[dlen++] = modifiers;
2732 }
2733
2734 if (IS_SPECIAL(key))
2735 {
2736 dst[dlen++] = K_SPECIAL;
2737 dst[dlen++] = KEY2TERMCAP0(key);
2738 dst[dlen++] = KEY2TERMCAP1(key);
2739 }
2740#ifdef FEAT_MBYTE
2741 else if (has_mbyte && !keycode)
2742 dlen += (*mb_char2bytes)(key, dst + dlen);
2743#endif
2744 else if (keycode)
2745 dlen = (int)(add_char2buf(key, dst + dlen) - dst);
2746 else
2747 dst[dlen++] = key;
2748
2749 return dlen;
2750}
2751
2752/*
2753 * Try translating a <> name at (*srcp)[], return the key and modifiers.
2754 * srcp is advanced to after the <> name.
2755 * returns 0 if there is no match.
2756 */
2757 int
Bram Moolenaar2a8ced02008-12-24 11:54:31 +00002758find_special_key(srcp, modp, keycode, keep_x_key)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002759 char_u **srcp;
2760 int *modp;
Bram Moolenaar2a8ced02008-12-24 11:54:31 +00002761 int keycode; /* prefer key code, e.g. K_DEL instead of DEL */
2762 int keep_x_key; /* don't translate xHome to Home key */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002763{
2764 char_u *last_dash;
2765 char_u *end_of_name;
2766 char_u *src;
2767 char_u *bp;
2768 int modifiers;
2769 int bit;
2770 int key;
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00002771 unsigned long n;
Bram Moolenaarb8bf5412011-08-17 20:33:22 +02002772 int l;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002773
2774 src = *srcp;
2775 if (src[0] != '<')
2776 return 0;
2777
2778 /* Find end of modifier list */
2779 last_dash = src;
2780 for (bp = src + 1; *bp == '-' || vim_isIDc(*bp); bp++)
2781 {
2782 if (*bp == '-')
2783 {
2784 last_dash = bp;
Bram Moolenaarb8bf5412011-08-17 20:33:22 +02002785 if (bp[1] != NUL)
2786 {
2787#ifdef FEAT_MBYTE
2788 if (has_mbyte)
2789 l = mb_ptr2len(bp + 1);
2790 else
2791#endif
2792 l = 1;
2793 if (bp[l + 1] == '>')
2794 bp += l; /* anything accepted, like <C-?> */
2795 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002796 }
2797 if (bp[0] == 't' && bp[1] == '_' && bp[2] && bp[3])
2798 bp += 3; /* skip t_xx, xx may be '-' or '>' */
Bram Moolenaar792826c2011-08-19 22:29:02 +02002799 else if (STRNICMP(bp, "char-", 5) == 0)
2800 {
2801 vim_str2nr(bp + 5, NULL, &l, TRUE, TRUE, NULL, NULL);
2802 bp += l + 5;
2803 break;
2804 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002805 }
2806
2807 if (*bp == '>') /* found matching '>' */
2808 {
2809 end_of_name = bp + 1;
2810
Bram Moolenaar071d4272004-06-13 20:20:40 +00002811 /* Which modifiers are given? */
2812 modifiers = 0x0;
2813 for (bp = src + 1; bp < last_dash; bp++)
2814 {
2815 if (*bp != '-')
2816 {
2817 bit = name_to_mod_mask(*bp);
2818 if (bit == 0x0)
2819 break; /* Illegal modifier name */
2820 modifiers |= bit;
2821 }
2822 }
2823
2824 /*
2825 * Legal modifier name.
2826 */
2827 if (bp >= last_dash)
2828 {
Bram Moolenaarb8bf5412011-08-17 20:33:22 +02002829 if (STRNICMP(last_dash + 1, "char-", 5) == 0
2830 && VIM_ISDIGIT(last_dash[6]))
2831 {
2832 /* <Char-123> or <Char-033> or <Char-0x33> */
2833 vim_str2nr(last_dash + 6, NULL, NULL, TRUE, TRUE, NULL, &n);
Bram Moolenaar792826c2011-08-19 22:29:02 +02002834 key = (int)n;
Bram Moolenaarb8bf5412011-08-17 20:33:22 +02002835 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002836 else
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002837 {
Bram Moolenaar792826c2011-08-19 22:29:02 +02002838 /*
2839 * Modifier with single letter, or special key name.
2840 */
2841#ifdef FEAT_MBYTE
2842 if (has_mbyte)
2843 l = mb_ptr2len(last_dash + 1);
2844 else
2845#endif
2846 l = 1;
2847 if (modifiers != 0 && last_dash[l + 1] == '>')
2848 key = PTR2CHAR(last_dash + 1);
2849 else
2850 {
2851 key = get_special_key_code(last_dash + 1);
2852 if (!keep_x_key)
2853 key = handle_x_keys(key);
2854 }
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002855 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002856
2857 /*
2858 * get_special_key_code() may return NUL for invalid
2859 * special key name.
2860 */
2861 if (key != NUL)
2862 {
2863 /*
2864 * Only use a modifier when there is no special key code that
2865 * includes the modifier.
2866 */
2867 key = simplify_key(key, &modifiers);
2868
2869 if (!keycode)
2870 {
2871 /* don't want keycode, use single byte code */
2872 if (key == K_BS)
2873 key = BS;
2874 else if (key == K_DEL || key == K_KDEL)
2875 key = DEL;
2876 }
2877
2878 /*
2879 * Normal Key with modifier: Try to make a single byte code.
2880 */
2881 if (!IS_SPECIAL(key))
2882 key = extract_modifiers(key, &modifiers);
2883
2884 *modp = modifiers;
2885 *srcp = end_of_name;
2886 return key;
2887 }
2888 }
2889 }
2890 return 0;
2891}
2892
2893/*
2894 * Try to include modifiers in the key.
2895 * Changes "Shift-a" to 'A', "Alt-A" to 0xc0, etc.
2896 */
2897 int
2898extract_modifiers(key, modp)
2899 int key;
2900 int *modp;
2901{
2902 int modifiers = *modp;
2903
2904#ifdef MACOS
2905 /* Command-key really special, No fancynest */
2906 if (!(modifiers & MOD_MASK_CMD))
2907#endif
2908 if ((modifiers & MOD_MASK_SHIFT) && ASCII_ISALPHA(key))
2909 {
2910 key = TOUPPER_ASC(key);
2911 modifiers &= ~MOD_MASK_SHIFT;
2912 }
2913 if ((modifiers & MOD_MASK_CTRL)
2914#ifdef EBCDIC
2915 /* * TODO: EBCDIC Better use:
2916 * && (Ctrl_chr(key) || key == '?')
2917 * ??? */
2918 && strchr("?@ABCDEFGHIJKLMNOPQRSTUVWXYZ[\\]^_", key)
2919 != NULL
2920#else
2921 && ((key >= '?' && key <= '_') || ASCII_ISALPHA(key))
2922#endif
2923 )
2924 {
2925 key = Ctrl_chr(key);
2926 modifiers &= ~MOD_MASK_CTRL;
2927 /* <C-@> is <Nul> */
2928 if (key == 0)
2929 key = K_ZERO;
2930 }
2931#ifdef MACOS
2932 /* Command-key really special, No fancynest */
2933 if (!(modifiers & MOD_MASK_CMD))
2934#endif
2935 if ((modifiers & MOD_MASK_ALT) && key < 0x80
2936#ifdef FEAT_MBYTE
2937 && !enc_dbcs /* avoid creating a lead byte */
2938#endif
2939 )
2940 {
2941 key |= 0x80;
2942 modifiers &= ~MOD_MASK_ALT; /* remove the META modifier */
2943 }
2944
2945 *modp = modifiers;
2946 return key;
2947}
2948
2949/*
2950 * Try to find key "c" in the special key table.
2951 * Return the index when found, -1 when not found.
2952 */
2953 int
2954find_special_key_in_table(c)
2955 int c;
2956{
2957 int i;
2958
2959 for (i = 0; key_names_table[i].name != NULL; i++)
2960 if (c == key_names_table[i].key)
2961 break;
2962 if (key_names_table[i].name == NULL)
2963 i = -1;
2964 return i;
2965}
2966
2967/*
2968 * Find the special key with the given name (the given string does not have to
2969 * end with NUL, the name is assumed to end before the first non-idchar).
2970 * If the name starts with "t_" the next two characters are interpreted as a
2971 * termcap name.
2972 * Return the key code, or 0 if not found.
2973 */
2974 int
2975get_special_key_code(name)
2976 char_u *name;
2977{
2978 char_u *table_name;
2979 char_u string[3];
2980 int i, j;
2981
2982 /*
2983 * If it's <t_xx> we get the code for xx from the termcap
2984 */
2985 if (name[0] == 't' && name[1] == '_' && name[2] != NUL && name[3] != NUL)
2986 {
2987 string[0] = name[2];
2988 string[1] = name[3];
2989 string[2] = NUL;
2990 if (add_termcap_entry(string, FALSE) == OK)
2991 return TERMCAP2KEY(name[2], name[3]);
2992 }
2993 else
2994 for (i = 0; key_names_table[i].name != NULL; i++)
2995 {
2996 table_name = key_names_table[i].name;
2997 for (j = 0; vim_isIDc(name[j]) && table_name[j] != NUL; j++)
2998 if (TOLOWER_ASC(table_name[j]) != TOLOWER_ASC(name[j]))
2999 break;
3000 if (!vim_isIDc(name[j]) && table_name[j] == NUL)
3001 return key_names_table[i].key;
3002 }
3003 return 0;
3004}
3005
Bram Moolenaar05bb9532008-07-04 09:44:11 +00003006#if defined(FEAT_CMDL_COMPL) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003007 char_u *
3008get_key_name(i)
3009 int i;
3010{
Bram Moolenaare1fbddc2009-05-16 19:07:03 +00003011 if (i >= (int)KEY_NAMES_TABLE_LEN)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003012 return NULL;
3013 return key_names_table[i].name;
3014}
3015#endif
3016
Bram Moolenaar05bb9532008-07-04 09:44:11 +00003017#if defined(FEAT_MOUSE) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003018/*
3019 * Look up the given mouse code to return the relevant information in the other
3020 * arguments. Return which button is down or was released.
3021 */
3022 int
3023get_mouse_button(code, is_click, is_drag)
3024 int code;
3025 int *is_click;
3026 int *is_drag;
3027{
3028 int i;
3029
3030 for (i = 0; mouse_table[i].pseudo_code; i++)
3031 if (code == mouse_table[i].pseudo_code)
3032 {
3033 *is_click = mouse_table[i].is_click;
3034 *is_drag = mouse_table[i].is_drag;
3035 return mouse_table[i].button;
3036 }
3037 return 0; /* Shouldn't get here */
3038}
3039
3040/*
3041 * Return the appropriate pseudo mouse event token (KE_LEFTMOUSE etc) based on
3042 * the given information about which mouse button is down, and whether the
3043 * mouse was clicked, dragged or released.
3044 */
3045 int
3046get_pseudo_mouse_code(button, is_click, is_drag)
3047 int button; /* eg MOUSE_LEFT */
3048 int is_click;
3049 int is_drag;
3050{
3051 int i;
3052
3053 for (i = 0; mouse_table[i].pseudo_code; i++)
3054 if (button == mouse_table[i].button
3055 && is_click == mouse_table[i].is_click
3056 && is_drag == mouse_table[i].is_drag)
3057 {
3058#ifdef FEAT_GUI
Bram Moolenaarc91506a2005-04-24 22:04:21 +00003059 /* Trick: a non mappable left click and release has mouse_col -1
3060 * or added MOUSE_COLOFF. Used for 'mousefocus' in
3061 * gui_mouse_moved() */
3062 if (mouse_col < 0 || mouse_col > MOUSE_COLOFF)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003063 {
Bram Moolenaarc91506a2005-04-24 22:04:21 +00003064 if (mouse_col < 0)
3065 mouse_col = 0;
3066 else
3067 mouse_col -= MOUSE_COLOFF;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003068 if (mouse_table[i].pseudo_code == (int)KE_LEFTMOUSE)
3069 return (int)KE_LEFTMOUSE_NM;
3070 if (mouse_table[i].pseudo_code == (int)KE_LEFTRELEASE)
3071 return (int)KE_LEFTRELEASE_NM;
3072 }
3073#endif
3074 return mouse_table[i].pseudo_code;
3075 }
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00003076 return (int)KE_IGNORE; /* not recognized, ignore it */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003077}
3078#endif /* FEAT_MOUSE */
3079
3080/*
3081 * Return the current end-of-line type: EOL_DOS, EOL_UNIX or EOL_MAC.
3082 */
3083 int
3084get_fileformat(buf)
3085 buf_T *buf;
3086{
3087 int c = *buf->b_p_ff;
3088
3089 if (buf->b_p_bin || c == 'u')
3090 return EOL_UNIX;
3091 if (c == 'm')
3092 return EOL_MAC;
3093 return EOL_DOS;
3094}
3095
3096/*
3097 * Like get_fileformat(), but override 'fileformat' with "p" for "++opt=val"
3098 * argument.
3099 */
3100 int
3101get_fileformat_force(buf, eap)
3102 buf_T *buf;
3103 exarg_T *eap; /* can be NULL! */
3104{
3105 int c;
3106
3107 if (eap != NULL && eap->force_ff != 0)
3108 c = eap->cmd[eap->force_ff];
3109 else
3110 {
3111 if ((eap != NULL && eap->force_bin != 0)
3112 ? (eap->force_bin == FORCE_BIN) : buf->b_p_bin)
3113 return EOL_UNIX;
3114 c = *buf->b_p_ff;
3115 }
3116 if (c == 'u')
3117 return EOL_UNIX;
3118 if (c == 'm')
3119 return EOL_MAC;
3120 return EOL_DOS;
3121}
3122
3123/*
3124 * Set the current end-of-line type to EOL_DOS, EOL_UNIX or EOL_MAC.
3125 * Sets both 'textmode' and 'fileformat'.
3126 * Note: Does _not_ set global value of 'textmode'!
3127 */
3128 void
3129set_fileformat(t, opt_flags)
3130 int t;
3131 int opt_flags; /* OPT_LOCAL and/or OPT_GLOBAL */
3132{
3133 char *p = NULL;
3134
3135 switch (t)
3136 {
3137 case EOL_DOS:
3138 p = FF_DOS;
3139 curbuf->b_p_tx = TRUE;
3140 break;
3141 case EOL_UNIX:
3142 p = FF_UNIX;
3143 curbuf->b_p_tx = FALSE;
3144 break;
3145 case EOL_MAC:
3146 p = FF_MAC;
3147 curbuf->b_p_tx = FALSE;
3148 break;
3149 }
3150 if (p != NULL)
3151 set_string_option_direct((char_u *)"ff", -1, (char_u *)p,
Bram Moolenaar5e3cb7e2006-02-27 23:58:35 +00003152 OPT_FREE | opt_flags, 0);
3153
Bram Moolenaar071d4272004-06-13 20:20:40 +00003154#ifdef FEAT_WINDOWS
Bram Moolenaarf740b292006-02-16 22:11:02 +00003155 /* This may cause the buffer to become (un)modified. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003156 check_status(curbuf);
Bram Moolenaar997fb4b2006-02-17 21:53:23 +00003157 redraw_tabline = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003158#endif
3159#ifdef FEAT_TITLE
3160 need_maketitle = TRUE; /* set window title later */
3161#endif
3162}
3163
3164/*
3165 * Return the default fileformat from 'fileformats'.
3166 */
3167 int
3168default_fileformat()
3169{
3170 switch (*p_ffs)
3171 {
3172 case 'm': return EOL_MAC;
3173 case 'd': return EOL_DOS;
3174 }
3175 return EOL_UNIX;
3176}
3177
3178/*
3179 * Call shell. Calls mch_call_shell, with 'shellxquote' added.
3180 */
3181 int
3182call_shell(cmd, opt)
3183 char_u *cmd;
3184 int opt;
3185{
3186 char_u *ncmd;
3187 int retval;
Bram Moolenaar05159a02005-02-26 23:04:13 +00003188#ifdef FEAT_PROFILE
3189 proftime_T wait_time;
3190#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003191
3192 if (p_verbose > 3)
3193 {
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00003194 verbose_enter();
Bram Moolenaar051b7822005-05-19 21:00:46 +00003195 smsg((char_u *)_("Calling shell to execute: \"%s\""),
Bram Moolenaar071d4272004-06-13 20:20:40 +00003196 cmd == NULL ? p_sh : cmd);
3197 out_char('\n');
3198 cursor_on();
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00003199 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003200 }
3201
Bram Moolenaar05159a02005-02-26 23:04:13 +00003202#ifdef FEAT_PROFILE
Bram Moolenaar01265852006-03-20 21:50:15 +00003203 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003204 prof_child_enter(&wait_time);
3205#endif
3206
Bram Moolenaar071d4272004-06-13 20:20:40 +00003207 if (*p_sh == NUL)
3208 {
3209 EMSG(_(e_shellempty));
3210 retval = -1;
3211 }
3212 else
3213 {
3214#ifdef FEAT_GUI_MSWIN
3215 /* Don't hide the pointer while executing a shell command. */
3216 gui_mch_mousehide(FALSE);
3217#endif
3218#ifdef FEAT_GUI
3219 ++hold_gui_events;
3220#endif
3221 /* The external command may update a tags file, clear cached tags. */
3222 tag_freematch();
3223
3224 if (cmd == NULL || *p_sxq == NUL)
3225 retval = mch_call_shell(cmd, opt);
3226 else
3227 {
3228 ncmd = alloc((unsigned)(STRLEN(cmd) + STRLEN(p_sxq) * 2 + 1));
3229 if (ncmd != NULL)
3230 {
3231 STRCPY(ncmd, p_sxq);
3232 STRCAT(ncmd, cmd);
3233 STRCAT(ncmd, p_sxq);
3234 retval = mch_call_shell(ncmd, opt);
3235 vim_free(ncmd);
3236 }
3237 else
3238 retval = -1;
3239 }
3240#ifdef FEAT_GUI
3241 --hold_gui_events;
3242#endif
3243 /*
3244 * Check the window size, in case it changed while executing the
3245 * external command.
3246 */
3247 shell_resized_check();
3248 }
3249
3250#ifdef FEAT_EVAL
3251 set_vim_var_nr(VV_SHELL_ERROR, (long)retval);
Bram Moolenaar05159a02005-02-26 23:04:13 +00003252# ifdef FEAT_PROFILE
Bram Moolenaar01265852006-03-20 21:50:15 +00003253 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003254 prof_child_exit(&wait_time);
3255# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003256#endif
3257
3258 return retval;
3259}
3260
3261/*
Bram Moolenaar01265852006-03-20 21:50:15 +00003262 * VISUAL, SELECTMODE and OP_PENDING State are never set, they are equal to
3263 * NORMAL State with a condition. This function returns the real State.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003264 */
3265 int
3266get_real_state()
3267{
3268 if (State & NORMAL)
3269 {
3270#ifdef FEAT_VISUAL
3271 if (VIsual_active)
Bram Moolenaar01265852006-03-20 21:50:15 +00003272 {
3273 if (VIsual_select)
3274 return SELECTMODE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003275 return VISUAL;
Bram Moolenaar01265852006-03-20 21:50:15 +00003276 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003277 else
3278#endif
3279 if (finish_op)
Bram Moolenaar01265852006-03-20 21:50:15 +00003280 return OP_PENDING;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003281 }
3282 return State;
3283}
3284
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003285#if defined(FEAT_MBYTE) || defined(PROTO)
3286/*
3287 * Return TRUE if "p" points to just after a path separator.
Bram Moolenaarb5ce04d2011-07-07 17:15:33 +02003288 * Takes care of multi-byte characters.
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003289 * "b" must point to the start of the file name
3290 */
3291 int
3292after_pathsep(b, p)
3293 char_u *b;
3294 char_u *p;
3295{
Bram Moolenaarb5ce04d2011-07-07 17:15:33 +02003296 return p > b && vim_ispathsep(p[-1])
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003297 && (!has_mbyte || (*mb_head_off)(b, p - 1) == 0);
3298}
3299#endif
3300
3301/*
3302 * Return TRUE if file names "f1" and "f2" are in the same directory.
3303 * "f1" may be a short name, "f2" must be a full path.
3304 */
3305 int
3306same_directory(f1, f2)
3307 char_u *f1;
3308 char_u *f2;
3309{
3310 char_u ffname[MAXPATHL];
3311 char_u *t1;
3312 char_u *t2;
3313
3314 /* safety check */
3315 if (f1 == NULL || f2 == NULL)
3316 return FALSE;
3317
3318 (void)vim_FullName(f1, ffname, MAXPATHL, FALSE);
3319 t1 = gettail_sep(ffname);
3320 t2 = gettail_sep(f2);
3321 return (t1 - ffname == t2 - f2
3322 && pathcmp((char *)ffname, (char *)f2, (int)(t1 - ffname)) == 0);
3323}
3324
Bram Moolenaar071d4272004-06-13 20:20:40 +00003325#if defined(FEAT_SESSION) || defined(MSWIN) || defined(FEAT_GUI_MAC) \
Bram Moolenaar9372a112005-12-06 19:59:18 +00003326 || ((defined(FEAT_GUI_GTK)) \
Bram Moolenaar843ee412004-06-30 16:16:41 +00003327 && ( defined(FEAT_WINDOWS) || defined(FEAT_DND)) ) \
Bram Moolenaar071d4272004-06-13 20:20:40 +00003328 || defined(FEAT_SUN_WORKSHOP) || defined(FEAT_NETBEANS_INTG) \
3329 || defined(PROTO)
3330/*
3331 * Change to a file's directory.
3332 * Caller must call shorten_fnames()!
3333 * Return OK or FAIL.
3334 */
3335 int
3336vim_chdirfile(fname)
3337 char_u *fname;
3338{
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003339 char_u dir[MAXPATHL];
Bram Moolenaar071d4272004-06-13 20:20:40 +00003340
Bram Moolenaarbbebc852005-07-18 21:47:53 +00003341 vim_strncpy(dir, fname, MAXPATHL - 1);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003342 *gettail_sep(dir) = NUL;
3343 return mch_chdir((char *)dir) == 0 ? OK : FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003344}
3345#endif
3346
3347#if defined(STAT_IGNORES_SLASH) || defined(PROTO)
3348/*
3349 * Check if "name" ends in a slash and is not a directory.
3350 * Used for systems where stat() ignores a trailing slash on a file name.
3351 * The Vim code assumes a trailing slash is only ignored for a directory.
3352 */
3353 int
3354illegal_slash(name)
3355 char *name;
3356{
3357 if (name[0] == NUL)
3358 return FALSE; /* no file name is not illegal */
3359 if (name[strlen(name) - 1] != '/')
3360 return FALSE; /* no trailing slash */
3361 if (mch_isdir((char_u *)name))
3362 return FALSE; /* trailing slash for a directory */
3363 return TRUE;
3364}
3365#endif
3366
3367#if defined(CURSOR_SHAPE) || defined(PROTO)
3368
3369/*
3370 * Handling of cursor and mouse pointer shapes in various modes.
3371 */
3372
3373cursorentry_T shape_table[SHAPE_IDX_COUNT] =
3374{
3375 /* The values will be filled in from the 'guicursor' and 'mouseshape'
3376 * defaults when Vim starts.
3377 * Adjust the SHAPE_IDX_ defines when making changes! */
3378 {0, 0, 0, 700L, 400L, 250L, 0, 0, "n", SHAPE_CURSOR+SHAPE_MOUSE},
3379 {0, 0, 0, 700L, 400L, 250L, 0, 0, "v", SHAPE_CURSOR+SHAPE_MOUSE},
3380 {0, 0, 0, 700L, 400L, 250L, 0, 0, "i", SHAPE_CURSOR+SHAPE_MOUSE},
3381 {0, 0, 0, 700L, 400L, 250L, 0, 0, "r", SHAPE_CURSOR+SHAPE_MOUSE},
3382 {0, 0, 0, 700L, 400L, 250L, 0, 0, "c", SHAPE_CURSOR+SHAPE_MOUSE},
3383 {0, 0, 0, 700L, 400L, 250L, 0, 0, "ci", SHAPE_CURSOR+SHAPE_MOUSE},
3384 {0, 0, 0, 700L, 400L, 250L, 0, 0, "cr", SHAPE_CURSOR+SHAPE_MOUSE},
3385 {0, 0, 0, 700L, 400L, 250L, 0, 0, "o", SHAPE_CURSOR+SHAPE_MOUSE},
3386 {0, 0, 0, 700L, 400L, 250L, 0, 0, "ve", SHAPE_CURSOR+SHAPE_MOUSE},
3387 {0, 0, 0, 0L, 0L, 0L, 0, 0, "e", SHAPE_MOUSE},
3388 {0, 0, 0, 0L, 0L, 0L, 0, 0, "s", SHAPE_MOUSE},
3389 {0, 0, 0, 0L, 0L, 0L, 0, 0, "sd", SHAPE_MOUSE},
3390 {0, 0, 0, 0L, 0L, 0L, 0, 0, "vs", SHAPE_MOUSE},
3391 {0, 0, 0, 0L, 0L, 0L, 0, 0, "vd", SHAPE_MOUSE},
3392 {0, 0, 0, 0L, 0L, 0L, 0, 0, "m", SHAPE_MOUSE},
3393 {0, 0, 0, 0L, 0L, 0L, 0, 0, "ml", SHAPE_MOUSE},
3394 {0, 0, 0, 100L, 100L, 100L, 0, 0, "sm", SHAPE_CURSOR},
3395};
3396
3397#ifdef FEAT_MOUSESHAPE
3398/*
3399 * Table with names for mouse shapes. Keep in sync with all the tables for
3400 * mch_set_mouse_shape()!.
3401 */
3402static char * mshape_names[] =
3403{
3404 "arrow", /* default, must be the first one */
3405 "blank", /* hidden */
3406 "beam",
3407 "updown",
3408 "udsizing",
3409 "leftright",
3410 "lrsizing",
3411 "busy",
3412 "no",
3413 "crosshair",
3414 "hand1",
3415 "hand2",
3416 "pencil",
3417 "question",
3418 "rightup-arrow",
3419 "up-arrow",
3420 NULL
3421};
3422#endif
3423
3424/*
3425 * Parse the 'guicursor' option ("what" is SHAPE_CURSOR) or 'mouseshape'
3426 * ("what" is SHAPE_MOUSE).
3427 * Returns error message for an illegal option, NULL otherwise.
3428 */
3429 char_u *
3430parse_shape_opt(what)
3431 int what;
3432{
3433 char_u *modep;
3434 char_u *colonp;
3435 char_u *commap;
3436 char_u *slashp;
3437 char_u *p, *endp;
3438 int idx = 0; /* init for GCC */
3439 int all_idx;
3440 int len;
3441 int i;
3442 long n;
3443 int found_ve = FALSE; /* found "ve" flag */
3444 int round;
3445
3446 /*
3447 * First round: check for errors; second round: do it for real.
3448 */
3449 for (round = 1; round <= 2; ++round)
3450 {
3451 /*
3452 * Repeat for all comma separated parts.
3453 */
3454#ifdef FEAT_MOUSESHAPE
3455 if (what == SHAPE_MOUSE)
3456 modep = p_mouseshape;
3457 else
3458#endif
3459 modep = p_guicursor;
3460 while (*modep != NUL)
3461 {
3462 colonp = vim_strchr(modep, ':');
3463 if (colonp == NULL)
3464 return (char_u *)N_("E545: Missing colon");
3465 if (colonp == modep)
3466 return (char_u *)N_("E546: Illegal mode");
3467 commap = vim_strchr(modep, ',');
3468
3469 /*
3470 * Repeat for all mode's before the colon.
3471 * For the 'a' mode, we loop to handle all the modes.
3472 */
3473 all_idx = -1;
3474 while (modep < colonp || all_idx >= 0)
3475 {
3476 if (all_idx < 0)
3477 {
3478 /* Find the mode. */
3479 if (modep[1] == '-' || modep[1] == ':')
3480 len = 1;
3481 else
3482 len = 2;
3483 if (len == 1 && TOLOWER_ASC(modep[0]) == 'a')
3484 all_idx = SHAPE_IDX_COUNT - 1;
3485 else
3486 {
3487 for (idx = 0; idx < SHAPE_IDX_COUNT; ++idx)
3488 if (STRNICMP(modep, shape_table[idx].name, len)
3489 == 0)
3490 break;
3491 if (idx == SHAPE_IDX_COUNT
3492 || (shape_table[idx].used_for & what) == 0)
3493 return (char_u *)N_("E546: Illegal mode");
3494 if (len == 2 && modep[0] == 'v' && modep[1] == 'e')
3495 found_ve = TRUE;
3496 }
3497 modep += len + 1;
3498 }
3499
3500 if (all_idx >= 0)
3501 idx = all_idx--;
3502 else if (round == 2)
3503 {
3504#ifdef FEAT_MOUSESHAPE
3505 if (what == SHAPE_MOUSE)
3506 {
3507 /* Set the default, for the missing parts */
3508 shape_table[idx].mshape = 0;
3509 }
3510 else
3511#endif
3512 {
3513 /* Set the defaults, for the missing parts */
3514 shape_table[idx].shape = SHAPE_BLOCK;
3515 shape_table[idx].blinkwait = 700L;
3516 shape_table[idx].blinkon = 400L;
3517 shape_table[idx].blinkoff = 250L;
3518 }
3519 }
3520
3521 /* Parse the part after the colon */
3522 for (p = colonp + 1; *p && *p != ','; )
3523 {
3524#ifdef FEAT_MOUSESHAPE
3525 if (what == SHAPE_MOUSE)
3526 {
3527 for (i = 0; ; ++i)
3528 {
3529 if (mshape_names[i] == NULL)
3530 {
3531 if (!VIM_ISDIGIT(*p))
3532 return (char_u *)N_("E547: Illegal mouseshape");
3533 if (round == 2)
3534 shape_table[idx].mshape =
3535 getdigits(&p) + MSHAPE_NUMBERED;
3536 else
3537 (void)getdigits(&p);
3538 break;
3539 }
3540 len = (int)STRLEN(mshape_names[i]);
3541 if (STRNICMP(p, mshape_names[i], len) == 0)
3542 {
3543 if (round == 2)
3544 shape_table[idx].mshape = i;
3545 p += len;
3546 break;
3547 }
3548 }
3549 }
3550 else /* if (what == SHAPE_MOUSE) */
3551#endif
3552 {
3553 /*
3554 * First handle the ones with a number argument.
3555 */
3556 i = *p;
3557 len = 0;
3558 if (STRNICMP(p, "ver", 3) == 0)
3559 len = 3;
3560 else if (STRNICMP(p, "hor", 3) == 0)
3561 len = 3;
3562 else if (STRNICMP(p, "blinkwait", 9) == 0)
3563 len = 9;
3564 else if (STRNICMP(p, "blinkon", 7) == 0)
3565 len = 7;
3566 else if (STRNICMP(p, "blinkoff", 8) == 0)
3567 len = 8;
3568 if (len != 0)
3569 {
3570 p += len;
3571 if (!VIM_ISDIGIT(*p))
3572 return (char_u *)N_("E548: digit expected");
3573 n = getdigits(&p);
3574 if (len == 3) /* "ver" or "hor" */
3575 {
3576 if (n == 0)
3577 return (char_u *)N_("E549: Illegal percentage");
3578 if (round == 2)
3579 {
3580 if (TOLOWER_ASC(i) == 'v')
3581 shape_table[idx].shape = SHAPE_VER;
3582 else
3583 shape_table[idx].shape = SHAPE_HOR;
3584 shape_table[idx].percentage = n;
3585 }
3586 }
3587 else if (round == 2)
3588 {
3589 if (len == 9)
3590 shape_table[idx].blinkwait = n;
3591 else if (len == 7)
3592 shape_table[idx].blinkon = n;
3593 else
3594 shape_table[idx].blinkoff = n;
3595 }
3596 }
3597 else if (STRNICMP(p, "block", 5) == 0)
3598 {
3599 if (round == 2)
3600 shape_table[idx].shape = SHAPE_BLOCK;
3601 p += 5;
3602 }
3603 else /* must be a highlight group name then */
3604 {
3605 endp = vim_strchr(p, '-');
3606 if (commap == NULL) /* last part */
3607 {
3608 if (endp == NULL)
3609 endp = p + STRLEN(p); /* find end of part */
3610 }
3611 else if (endp > commap || endp == NULL)
3612 endp = commap;
3613 slashp = vim_strchr(p, '/');
3614 if (slashp != NULL && slashp < endp)
3615 {
3616 /* "group/langmap_group" */
3617 i = syn_check_group(p, (int)(slashp - p));
3618 p = slashp + 1;
3619 }
3620 if (round == 2)
3621 {
3622 shape_table[idx].id = syn_check_group(p,
3623 (int)(endp - p));
3624 shape_table[idx].id_lm = shape_table[idx].id;
3625 if (slashp != NULL && slashp < endp)
3626 shape_table[idx].id = i;
3627 }
3628 p = endp;
3629 }
3630 } /* if (what != SHAPE_MOUSE) */
3631
3632 if (*p == '-')
3633 ++p;
3634 }
3635 }
3636 modep = p;
3637 if (*modep == ',')
3638 ++modep;
3639 }
3640 }
3641
3642 /* If the 's' flag is not given, use the 'v' cursor for 's' */
3643 if (!found_ve)
3644 {
3645#ifdef FEAT_MOUSESHAPE
3646 if (what == SHAPE_MOUSE)
3647 {
3648 shape_table[SHAPE_IDX_VE].mshape = shape_table[SHAPE_IDX_V].mshape;
3649 }
3650 else
3651#endif
3652 {
3653 shape_table[SHAPE_IDX_VE].shape = shape_table[SHAPE_IDX_V].shape;
3654 shape_table[SHAPE_IDX_VE].percentage =
3655 shape_table[SHAPE_IDX_V].percentage;
3656 shape_table[SHAPE_IDX_VE].blinkwait =
3657 shape_table[SHAPE_IDX_V].blinkwait;
3658 shape_table[SHAPE_IDX_VE].blinkon =
3659 shape_table[SHAPE_IDX_V].blinkon;
3660 shape_table[SHAPE_IDX_VE].blinkoff =
3661 shape_table[SHAPE_IDX_V].blinkoff;
3662 shape_table[SHAPE_IDX_VE].id = shape_table[SHAPE_IDX_V].id;
3663 shape_table[SHAPE_IDX_VE].id_lm = shape_table[SHAPE_IDX_V].id_lm;
3664 }
3665 }
3666
3667 return NULL;
3668}
3669
Bram Moolenaarac6e65f2005-08-29 22:25:38 +00003670# if defined(MCH_CURSOR_SHAPE) || defined(FEAT_GUI) \
3671 || defined(FEAT_MOUSESHAPE) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003672/*
3673 * Return the index into shape_table[] for the current mode.
3674 * When "mouse" is TRUE, consider indexes valid for the mouse pointer.
3675 */
3676 int
3677get_shape_idx(mouse)
3678 int mouse;
3679{
3680#ifdef FEAT_MOUSESHAPE
3681 if (mouse && (State == HITRETURN || State == ASKMORE))
3682 {
3683# ifdef FEAT_GUI
Bram Moolenaar9588a0f2005-01-08 21:45:39 +00003684 int x, y;
3685 gui_mch_getmouse(&x, &y);
3686 if (Y_2_ROW(y) == Rows - 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003687 return SHAPE_IDX_MOREL;
3688# endif
3689 return SHAPE_IDX_MORE;
3690 }
3691 if (mouse && drag_status_line)
3692 return SHAPE_IDX_SDRAG;
3693# ifdef FEAT_VERTSPLIT
3694 if (mouse && drag_sep_line)
3695 return SHAPE_IDX_VDRAG;
3696# endif
3697#endif
3698 if (!mouse && State == SHOWMATCH)
3699 return SHAPE_IDX_SM;
3700#ifdef FEAT_VREPLACE
3701 if (State & VREPLACE_FLAG)
3702 return SHAPE_IDX_R;
3703#endif
3704 if (State & REPLACE_FLAG)
3705 return SHAPE_IDX_R;
3706 if (State & INSERT)
3707 return SHAPE_IDX_I;
3708 if (State & CMDLINE)
3709 {
3710 if (cmdline_at_end())
3711 return SHAPE_IDX_C;
3712 if (cmdline_overstrike())
3713 return SHAPE_IDX_CR;
3714 return SHAPE_IDX_CI;
3715 }
3716 if (finish_op)
3717 return SHAPE_IDX_O;
3718#ifdef FEAT_VISUAL
3719 if (VIsual_active)
3720 {
3721 if (*p_sel == 'e')
3722 return SHAPE_IDX_VE;
3723 else
3724 return SHAPE_IDX_V;
3725 }
3726#endif
3727 return SHAPE_IDX_N;
3728}
Bram Moolenaarac6e65f2005-08-29 22:25:38 +00003729#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003730
3731# if defined(FEAT_MOUSESHAPE) || defined(PROTO)
3732static int old_mouse_shape = 0;
3733
3734/*
3735 * Set the mouse shape:
3736 * If "shape" is -1, use shape depending on the current mode,
3737 * depending on the current state.
3738 * If "shape" is -2, only update the shape when it's CLINE or STATUS (used
3739 * when the mouse moves off the status or command line).
3740 */
3741 void
3742update_mouseshape(shape_idx)
3743 int shape_idx;
3744{
3745 int new_mouse_shape;
3746
3747 /* Only works in GUI mode. */
Bram Moolenaar6bb68362005-03-22 23:03:44 +00003748 if (!gui.in_use || gui.starting)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003749 return;
3750
3751 /* Postpone the updating when more is to come. Speeds up executing of
3752 * mappings. */
3753 if (shape_idx == -1 && char_avail())
3754 {
3755 postponed_mouseshape = TRUE;
3756 return;
3757 }
3758
Bram Moolenaar14716812006-05-04 21:54:08 +00003759 /* When ignoring the mouse don't change shape on the statusline. */
3760 if (*p_mouse == NUL
3761 && (shape_idx == SHAPE_IDX_CLINE
3762 || shape_idx == SHAPE_IDX_STATUS
3763 || shape_idx == SHAPE_IDX_VSEP))
3764 shape_idx = -2;
3765
Bram Moolenaar071d4272004-06-13 20:20:40 +00003766 if (shape_idx == -2
3767 && old_mouse_shape != shape_table[SHAPE_IDX_CLINE].mshape
3768 && old_mouse_shape != shape_table[SHAPE_IDX_STATUS].mshape
3769 && old_mouse_shape != shape_table[SHAPE_IDX_VSEP].mshape)
3770 return;
3771 if (shape_idx < 0)
3772 new_mouse_shape = shape_table[get_shape_idx(TRUE)].mshape;
3773 else
3774 new_mouse_shape = shape_table[shape_idx].mshape;
3775 if (new_mouse_shape != old_mouse_shape)
3776 {
3777 mch_set_mouse_shape(new_mouse_shape);
3778 old_mouse_shape = new_mouse_shape;
3779 }
3780 postponed_mouseshape = FALSE;
3781}
3782# endif
3783
3784#endif /* CURSOR_SHAPE */
3785
3786
3787#ifdef FEAT_CRYPT
3788/*
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00003789 * Optional encryption support.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003790 * Mohsin Ahmed, mosh@sasi.com, 98-09-24
3791 * Based on zip/crypt sources.
3792 *
3793 * NOTE FOR USA: Since 2000 exporting this code from the USA is allowed to
3794 * most countries. There are a few exceptions, but that still should not be a
3795 * problem since this code was originally created in Europe and India.
Bram Moolenaar40e6a712010-05-16 22:32:54 +02003796 *
3797 * Blowfish addition originally made by Mohsin Ahmed,
3798 * http://www.cs.albany.edu/~mosh 2010-03-14
3799 * Based on blowfish by Bruce Schneier (http://www.schneier.com/blowfish.html)
3800 * and sha256 by Christophe Devine.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003801 */
3802
3803/* from zip.h */
3804
3805typedef unsigned short ush; /* unsigned 16-bit value */
3806typedef unsigned long ulg; /* unsigned 32-bit value */
3807
3808static void make_crc_tab __ARGS((void));
3809
Bram Moolenaard6f676d2005-06-01 21:51:55 +00003810static ulg crc_32_tab[256];
Bram Moolenaar071d4272004-06-13 20:20:40 +00003811
3812/*
3813 * Fill the CRC table.
3814 */
3815 static void
3816make_crc_tab()
3817{
3818 ulg s,t,v;
3819 static int done = FALSE;
3820
3821 if (done)
3822 return;
3823 for (t = 0; t < 256; t++)
3824 {
3825 v = t;
3826 for (s = 0; s < 8; s++)
3827 v = (v >> 1) ^ ((v & 1) * (ulg)0xedb88320L);
3828 crc_32_tab[t] = v;
3829 }
3830 done = TRUE;
3831}
3832
3833#define CRC32(c, b) (crc_32_tab[((int)(c) ^ (b)) & 0xff] ^ ((c) >> 8))
3834
Bram Moolenaar071d4272004-06-13 20:20:40 +00003835static ulg keys[3]; /* keys defining the pseudo-random sequence */
3836
3837/*
Bram Moolenaar04c9baf2010-06-01 23:37:39 +02003838 * Return the next byte in the pseudo-random sequence.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003839 */
Bram Moolenaar04c9baf2010-06-01 23:37:39 +02003840#define DECRYPT_BYTE_ZIP(t) { \
3841 ush temp; \
3842 \
3843 temp = (ush)keys[2] | 2; \
3844 t = (int)(((unsigned)(temp * (temp ^ 1)) >> 8) & 0xff); \
Bram Moolenaar071d4272004-06-13 20:20:40 +00003845}
3846
3847/*
Bram Moolenaar04c9baf2010-06-01 23:37:39 +02003848 * Update the encryption keys with the next byte of plain text.
3849 */
3850#define UPDATE_KEYS_ZIP(c) { \
3851 keys[0] = CRC32(keys[0], (c)); \
3852 keys[1] += keys[0] & 0xff; \
3853 keys[1] = keys[1] * 134775813L + 1; \
3854 keys[2] = CRC32(keys[2], (int)(keys[1] >> 24)); \
3855}
3856
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02003857static int crypt_busy = 0;
3858static ulg saved_keys[3];
3859static int saved_crypt_method;
3860
3861/*
Bram Moolenaar49771f42010-07-20 17:32:38 +02003862 * Return int value for crypt method string:
3863 * 0 for "zip", the old method. Also for any non-valid value.
3864 * 1 for "blowfish".
3865 */
3866 int
3867crypt_method_from_string(s)
3868 char_u *s;
3869{
3870 return *s == 'b' ? 1 : 0;
3871}
3872
3873/*
3874 * Get the crypt method for buffer "buf" as a number.
3875 */
3876 int
3877get_crypt_method(buf)
3878 buf_T *buf;
3879{
3880 return crypt_method_from_string(*buf->b_p_cm == NUL ? p_cm : buf->b_p_cm);
3881}
3882
3883/*
3884 * Set the crypt method for buffer "buf" to "method" using the int value as
3885 * returned by crypt_method_from_string().
3886 */
3887 void
3888set_crypt_method(buf, method)
3889 buf_T *buf;
3890 int method;
3891{
3892 free_string_option(buf->b_p_cm);
3893 buf->b_p_cm = vim_strsave((char_u *)(method == 0 ? "zip" : "blowfish"));
3894}
3895
3896/*
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02003897 * Prepare for initializing encryption. If already doing encryption then save
3898 * the state.
Bram Moolenaar2f1e0502010-08-13 11:18:02 +02003899 * Must always be called symmetrically with crypt_pop_state().
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02003900 */
3901 void
3902crypt_push_state()
3903{
3904 if (crypt_busy == 1)
3905 {
3906 /* save the state */
3907 if (use_crypt_method == 0)
3908 {
3909 saved_keys[0] = keys[0];
3910 saved_keys[1] = keys[1];
3911 saved_keys[2] = keys[2];
3912 }
3913 else
3914 bf_crypt_save();
3915 saved_crypt_method = use_crypt_method;
3916 }
3917 else if (crypt_busy > 1)
3918 EMSG2(_(e_intern2), "crypt_push_state()");
3919 ++crypt_busy;
3920}
3921
3922/*
3923 * End encryption. If doing encryption before crypt_push_state() then restore
3924 * the saved state.
Bram Moolenaar2f1e0502010-08-13 11:18:02 +02003925 * Must always be called symmetrically with crypt_push_state().
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02003926 */
3927 void
3928crypt_pop_state()
3929{
3930 --crypt_busy;
3931 if (crypt_busy == 1)
3932 {
3933 use_crypt_method = saved_crypt_method;
3934 if (use_crypt_method == 0)
3935 {
3936 keys[0] = saved_keys[0];
3937 keys[1] = saved_keys[1];
3938 keys[2] = saved_keys[2];
3939 }
3940 else
3941 bf_crypt_restore();
3942 }
3943}
3944
Bram Moolenaar04c9baf2010-06-01 23:37:39 +02003945/*
3946 * Encrypt "from[len]" into "to[len]".
3947 * "from" and "to" can be equal to encrypt in place.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003948 */
Bram Moolenaar40e6a712010-05-16 22:32:54 +02003949 void
Bram Moolenaar04c9baf2010-06-01 23:37:39 +02003950crypt_encode(from, len, to)
3951 char_u *from;
3952 size_t len;
3953 char_u *to;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003954{
Bram Moolenaar04c9baf2010-06-01 23:37:39 +02003955 size_t i;
3956 int ztemp, t;
3957
3958 if (use_crypt_method == 0)
3959 for (i = 0; i < len; ++i)
3960 {
3961 ztemp = from[i];
3962 DECRYPT_BYTE_ZIP(t);
3963 UPDATE_KEYS_ZIP(ztemp);
3964 to[i] = t ^ ztemp;
3965 }
Bram Moolenaar40e6a712010-05-16 22:32:54 +02003966 else
Bram Moolenaarbbd6afe2010-06-02 20:32:23 +02003967 bf_crypt_encode(from, len, to);
Bram Moolenaar04c9baf2010-06-01 23:37:39 +02003968}
3969
3970/*
3971 * Decrypt "ptr[len]" in place.
3972 */
3973 void
3974crypt_decode(ptr, len)
3975 char_u *ptr;
3976 long len;
3977{
3978 char_u *p;
3979
3980 if (use_crypt_method == 0)
3981 for (p = ptr; p < ptr + len; ++p)
3982 {
3983 ush temp;
3984
3985 temp = (ush)keys[2] | 2;
3986 temp = (int)(((unsigned)(temp * (temp ^ 1)) >> 8) & 0xff);
3987 UPDATE_KEYS_ZIP(*p ^= temp);
3988 }
3989 else
Bram Moolenaarbbd6afe2010-06-02 20:32:23 +02003990 bf_crypt_decode(ptr, len);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003991}
3992
3993/*
3994 * Initialize the encryption keys and the random header according to
3995 * the given password.
3996 * If "passwd" is NULL or empty, don't do anything.
3997 */
3998 void
3999crypt_init_keys(passwd)
4000 char_u *passwd; /* password string with which to modify keys */
4001{
4002 if (passwd != NULL && *passwd != NUL)
4003 {
Bram Moolenaar04c9baf2010-06-01 23:37:39 +02004004 if (use_crypt_method == 0)
Bram Moolenaarbbd6afe2010-06-02 20:32:23 +02004005 {
4006 char_u *p;
4007
4008 make_crc_tab();
4009 keys[0] = 305419896L;
4010 keys[1] = 591751049L;
4011 keys[2] = 878082192L;
4012 for (p = passwd; *p!= NUL; ++p)
Bram Moolenaar04c9baf2010-06-01 23:37:39 +02004013 {
Bram Moolenaarbbd6afe2010-06-02 20:32:23 +02004014 UPDATE_KEYS_ZIP((int)*p);
Bram Moolenaar04c9baf2010-06-01 23:37:39 +02004015 }
Bram Moolenaarbbd6afe2010-06-02 20:32:23 +02004016 }
Bram Moolenaar04c9baf2010-06-01 23:37:39 +02004017 else
Bram Moolenaarbbd6afe2010-06-02 20:32:23 +02004018 bf_crypt_init_keys(passwd);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004019 }
4020}
4021
4022/*
Bram Moolenaar40e6a712010-05-16 22:32:54 +02004023 * Free an allocated crypt key. Clear the text to make sure it doesn't stay
4024 * in memory anywhere.
4025 */
4026 void
4027free_crypt_key(key)
4028 char_u *key;
4029{
4030 char_u *p;
4031
4032 if (key != NULL)
4033 {
4034 for (p = key; *p != NUL; ++p)
Bram Moolenaarbe18d102010-05-22 21:37:53 +02004035 *p = 0;
Bram Moolenaar40e6a712010-05-16 22:32:54 +02004036 vim_free(key);
4037 }
4038}
4039
4040/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00004041 * Ask the user for a crypt key.
Bram Moolenaar40e6a712010-05-16 22:32:54 +02004042 * When "store" is TRUE, the new key is stored in the 'key' option, and the
Bram Moolenaar071d4272004-06-13 20:20:40 +00004043 * 'key' option value is returned: Don't free it.
4044 * When "store" is FALSE, the typed key is returned in allocated memory.
4045 * Returns NULL on failure.
4046 */
4047 char_u *
4048get_crypt_key(store, twice)
4049 int store;
4050 int twice; /* Ask for the key twice. */
4051{
4052 char_u *p1, *p2 = NULL;
4053 int round;
4054
4055 for (round = 0; ; ++round)
4056 {
4057 cmdline_star = TRUE;
4058 cmdline_row = msg_row;
4059 p1 = getcmdline_prompt(NUL, round == 0
4060 ? (char_u *)_("Enter encryption key: ")
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +00004061 : (char_u *)_("Enter same key again: "), 0, EXPAND_NOTHING,
4062 NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004063 cmdline_star = FALSE;
4064
4065 if (p1 == NULL)
4066 break;
4067
4068 if (round == twice)
4069 {
4070 if (p2 != NULL && STRCMP(p1, p2) != 0)
4071 {
4072 MSG(_("Keys don't match!"));
Bram Moolenaar40e6a712010-05-16 22:32:54 +02004073 free_crypt_key(p1);
4074 free_crypt_key(p2);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004075 p2 = NULL;
4076 round = -1; /* do it again */
4077 continue;
4078 }
Bram Moolenaar40e6a712010-05-16 22:32:54 +02004079
Bram Moolenaar071d4272004-06-13 20:20:40 +00004080 if (store)
4081 {
4082 set_option_value((char_u *)"key", 0L, p1, OPT_LOCAL);
Bram Moolenaar40e6a712010-05-16 22:32:54 +02004083 free_crypt_key(p1);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004084 p1 = curbuf->b_p_key;
4085 }
4086 break;
4087 }
4088 p2 = p1;
4089 }
4090
4091 /* since the user typed this, no need to wait for return */
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02004092 if (msg_didout)
4093 msg_putchar('\n');
Bram Moolenaare4ce65d2010-08-04 20:12:32 +02004094 need_wait_return = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004095 msg_didout = FALSE;
4096
Bram Moolenaar40e6a712010-05-16 22:32:54 +02004097 free_crypt_key(p2);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004098 return p1;
4099}
4100
4101#endif /* FEAT_CRYPT */
4102
4103/* TODO: make some #ifdef for this */
4104/*--------[ file searching ]-------------------------------------------------*/
4105/*
4106 * File searching functions for 'path', 'tags' and 'cdpath' options.
4107 * External visible functions:
4108 * vim_findfile_init() creates/initialises the search context
4109 * vim_findfile_free_visited() free list of visited files/dirs of search
4110 * context
4111 * vim_findfile() find a file in the search context
4112 * vim_findfile_cleanup() cleanup/free search context created by
4113 * vim_findfile_init()
4114 *
4115 * All static functions and variables start with 'ff_'
4116 *
4117 * In general it works like this:
4118 * First you create yourself a search context by calling vim_findfile_init().
4119 * It is possible to give a search context from a previous call to
4120 * vim_findfile_init(), so it can be reused. After this you call vim_findfile()
4121 * until you are satisfied with the result or it returns NULL. On every call it
4122 * returns the next file which matches the conditions given to
4123 * vim_findfile_init(). If it doesn't find a next file it returns NULL.
4124 *
4125 * It is possible to call vim_findfile_init() again to reinitialise your search
4126 * with some new parameters. Don't forget to pass your old search context to
4127 * it, so it can reuse it and especially reuse the list of already visited
4128 * directories. If you want to delete the list of already visited directories
4129 * simply call vim_findfile_free_visited().
4130 *
4131 * When you are done call vim_findfile_cleanup() to free the search context.
4132 *
4133 * The function vim_findfile_init() has a long comment, which describes the
4134 * needed parameters.
4135 *
4136 *
4137 *
4138 * ATTENTION:
4139 * ==========
Bram Moolenaar77f66d62007-02-20 02:16:18 +00004140 * Also we use an allocated search context here, this functions are NOT
Bram Moolenaar071d4272004-06-13 20:20:40 +00004141 * thread-safe!!!!!
4142 *
4143 * To minimize parameter passing (or because I'm to lazy), only the
4144 * external visible functions get a search context as a parameter. This is
4145 * then assigned to a static global, which is used throughout the local
4146 * functions.
4147 */
4148
4149/*
4150 * type for the directory search stack
4151 */
4152typedef struct ff_stack
4153{
4154 struct ff_stack *ffs_prev;
4155
4156 /* the fix part (no wildcards) and the part containing the wildcards
4157 * of the search path
4158 */
4159 char_u *ffs_fix_path;
4160#ifdef FEAT_PATH_EXTRA
4161 char_u *ffs_wc_path;
4162#endif
4163
4164 /* files/dirs found in the above directory, matched by the first wildcard
4165 * of wc_part
4166 */
4167 char_u **ffs_filearray;
4168 int ffs_filearray_size;
4169 char_u ffs_filearray_cur; /* needed for partly handled dirs */
4170
4171 /* to store status of partly handled directories
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004172 * 0: we work on this directory for the first time
Bram Moolenaar071d4272004-06-13 20:20:40 +00004173 * 1: this directory was partly searched in an earlier step
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004174 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004175 int ffs_stage;
4176
4177 /* How deep are we in the directory tree?
4178 * Counts backward from value of level parameter to vim_findfile_init
4179 */
4180 int ffs_level;
4181
4182 /* Did we already expand '**' to an empty string? */
4183 int ffs_star_star_empty;
4184} ff_stack_T;
4185
4186/*
4187 * type for already visited directories or files.
4188 */
4189typedef struct ff_visited
4190{
4191 struct ff_visited *ffv_next;
4192
4193#ifdef FEAT_PATH_EXTRA
4194 /* Visited directories are different if the wildcard string are
4195 * different. So we have to save it.
4196 */
4197 char_u *ffv_wc_path;
4198#endif
4199 /* for unix use inode etc for comparison (needed because of links), else
4200 * use filename.
4201 */
4202#ifdef UNIX
Bram Moolenaare1fbddc2009-05-16 19:07:03 +00004203 int ffv_dev_valid; /* ffv_dev and ffv_ino were set */
4204 dev_t ffv_dev; /* device number */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004205 ino_t ffv_ino; /* inode number */
4206#endif
4207 /* The memory for this struct is allocated according to the length of
4208 * ffv_fname.
4209 */
4210 char_u ffv_fname[1]; /* actually longer */
4211} ff_visited_T;
4212
4213/*
4214 * We might have to manage several visited lists during a search.
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00004215 * This is especially needed for the tags option. If tags is set to:
Bram Moolenaar071d4272004-06-13 20:20:40 +00004216 * "./++/tags,./++/TAGS,++/tags" (replace + with *)
4217 * So we have to do 3 searches:
4218 * 1) search from the current files directory downward for the file "tags"
4219 * 2) search from the current files directory downward for the file "TAGS"
4220 * 3) search from Vims current directory downwards for the file "tags"
4221 * As you can see, the first and the third search are for the same file, so for
4222 * the third search we can use the visited list of the first search. For the
4223 * second search we must start from a empty visited list.
4224 * The struct ff_visited_list_hdr is used to manage a linked list of already
4225 * visited lists.
4226 */
4227typedef struct ff_visited_list_hdr
4228{
4229 struct ff_visited_list_hdr *ffvl_next;
4230
4231 /* the filename the attached visited list is for */
4232 char_u *ffvl_filename;
4233
4234 ff_visited_T *ffvl_visited_list;
4235
4236} ff_visited_list_hdr_T;
4237
4238
4239/*
4240 * '**' can be expanded to several directory levels.
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00004241 * Set the default maximum depth.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004242 */
4243#define FF_MAX_STAR_STAR_EXPAND ((char_u)30)
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004244
Bram Moolenaar071d4272004-06-13 20:20:40 +00004245/*
4246 * The search context:
4247 * ffsc_stack_ptr: the stack for the dirs to search
4248 * ffsc_visited_list: the currently active visited list
4249 * ffsc_dir_visited_list: the currently active visited list for search dirs
4250 * ffsc_visited_lists_list: the list of all visited lists
4251 * ffsc_dir_visited_lists_list: the list of all visited lists for search dirs
4252 * ffsc_file_to_search: the file to search for
4253 * ffsc_start_dir: the starting directory, if search path was relative
4254 * ffsc_fix_path: the fix part of the given path (without wildcards)
4255 * Needed for upward search.
4256 * ffsc_wc_path: the part of the given path containing wildcards
4257 * ffsc_level: how many levels of dirs to search downwards
4258 * ffsc_stopdirs_v: array of stop directories for upward search
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004259 * ffsc_find_what: FINDFILE_BOTH, FINDFILE_DIR or FINDFILE_FILE
Bram Moolenaar463ee342010-08-08 18:17:52 +02004260 * ffsc_tagfile: searching for tags file, don't use 'suffixesadd'
Bram Moolenaar071d4272004-06-13 20:20:40 +00004261 */
4262typedef struct ff_search_ctx_T
4263{
4264 ff_stack_T *ffsc_stack_ptr;
4265 ff_visited_list_hdr_T *ffsc_visited_list;
4266 ff_visited_list_hdr_T *ffsc_dir_visited_list;
4267 ff_visited_list_hdr_T *ffsc_visited_lists_list;
4268 ff_visited_list_hdr_T *ffsc_dir_visited_lists_list;
4269 char_u *ffsc_file_to_search;
4270 char_u *ffsc_start_dir;
4271 char_u *ffsc_fix_path;
4272#ifdef FEAT_PATH_EXTRA
4273 char_u *ffsc_wc_path;
4274 int ffsc_level;
4275 char_u **ffsc_stopdirs_v;
4276#endif
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004277 int ffsc_find_what;
Bram Moolenaar463ee342010-08-08 18:17:52 +02004278 int ffsc_tagfile;
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00004279} ff_search_ctx_T;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004280
Bram Moolenaar071d4272004-06-13 20:20:40 +00004281/* locally needed functions */
4282#ifdef FEAT_PATH_EXTRA
4283static int ff_check_visited __ARGS((ff_visited_T **, char_u *, char_u *));
4284#else
4285static int ff_check_visited __ARGS((ff_visited_T **, char_u *));
4286#endif
4287static void vim_findfile_free_visited_list __ARGS((ff_visited_list_hdr_T **list_headp));
4288static void ff_free_visited_list __ARGS((ff_visited_T *vl));
4289static ff_visited_list_hdr_T* ff_get_visited_list __ARGS((char_u *, ff_visited_list_hdr_T **list_headp));
4290#ifdef FEAT_PATH_EXTRA
4291static int ff_wc_equal __ARGS((char_u *s1, char_u *s2));
4292#endif
4293
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004294static void ff_push __ARGS((ff_search_ctx_T *search_ctx, ff_stack_T *stack_ptr));
4295static ff_stack_T *ff_pop __ARGS((ff_search_ctx_T *search_ctx));
4296static void ff_clear __ARGS((ff_search_ctx_T *search_ctx));
4297static void ff_free_stack_element __ARGS((ff_stack_T *stack_ptr));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004298#ifdef FEAT_PATH_EXTRA
4299static ff_stack_T *ff_create_stack_element __ARGS((char_u *, char_u *, int, int));
4300#else
4301static ff_stack_T *ff_create_stack_element __ARGS((char_u *, int, int));
4302#endif
4303#ifdef FEAT_PATH_EXTRA
4304static int ff_path_in_stoplist __ARGS((char_u *, int, char_u **));
4305#endif
4306
Bram Moolenaarb9ba4032011-12-08 17:49:35 +01004307static char_u e_pathtoolong[] = N_("E854: path too long for completion");
4308
Bram Moolenaar071d4272004-06-13 20:20:40 +00004309#if 0
4310/*
4311 * if someone likes findfirst/findnext, here are the functions
4312 * NOT TESTED!!
4313 */
4314
4315static void *ff_fn_search_context = NULL;
4316
4317 char_u *
4318vim_findfirst(path, filename, level)
4319 char_u *path;
4320 char_u *filename;
4321 int level;
4322{
4323 ff_fn_search_context =
4324 vim_findfile_init(path, filename, NULL, level, TRUE, FALSE,
4325 ff_fn_search_context, rel_fname);
4326 if (NULL == ff_fn_search_context)
4327 return NULL;
4328 else
4329 return vim_findnext()
4330}
4331
4332 char_u *
4333vim_findnext()
4334{
4335 char_u *ret = vim_findfile(ff_fn_search_context);
4336
4337 if (NULL == ret)
4338 {
4339 vim_findfile_cleanup(ff_fn_search_context);
4340 ff_fn_search_context = NULL;
4341 }
4342 return ret;
4343}
4344#endif
4345
4346/*
Bram Moolenaare2b590e2010-08-08 18:29:48 +02004347 * Initialization routine for vim_findfile().
Bram Moolenaar071d4272004-06-13 20:20:40 +00004348 *
Bram Moolenaarbe18d102010-05-22 21:37:53 +02004349 * Returns the newly allocated search context or NULL if an error occurred.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004350 *
4351 * Don't forget to clean up by calling vim_findfile_cleanup() if you are done
4352 * with the search context.
4353 *
4354 * Find the file 'filename' in the directory 'path'.
4355 * The parameter 'path' may contain wildcards. If so only search 'level'
4356 * directories deep. The parameter 'level' is the absolute maximum and is
4357 * not related to restricts given to the '**' wildcard. If 'level' is 100
4358 * and you use '**200' vim_findfile() will stop after 100 levels.
4359 *
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004360 * 'filename' cannot contain wildcards! It is used as-is, no backslashes to
4361 * escape special characters.
4362 *
Bram Moolenaar071d4272004-06-13 20:20:40 +00004363 * If 'stopdirs' is not NULL and nothing is found downward, the search is
4364 * restarted on the next higher directory level. This is repeated until the
4365 * start-directory of a search is contained in 'stopdirs'. 'stopdirs' has the
4366 * format ";*<dirname>*\(;<dirname>\)*;\=$".
4367 *
4368 * If the 'path' is relative, the starting dir for the search is either VIM's
4369 * current dir or if the path starts with "./" the current files dir.
Bram Moolenaarbe18d102010-05-22 21:37:53 +02004370 * If the 'path' is absolute, the starting dir is that part of the path before
Bram Moolenaar071d4272004-06-13 20:20:40 +00004371 * the first wildcard.
4372 *
4373 * Upward search is only done on the starting dir.
4374 *
4375 * If 'free_visited' is TRUE the list of already visited files/directories is
4376 * cleared. Set this to FALSE if you just want to search from another
4377 * directory, but want to be sure that no directory from a previous search is
4378 * searched again. This is useful if you search for a file at different places.
4379 * The list of visited files/dirs can also be cleared with the function
4380 * vim_findfile_free_visited().
4381 *
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004382 * Set the parameter 'find_what' to FINDFILE_DIR if you want to search for
4383 * directories only, FINDFILE_FILE for files only, FINDFILE_BOTH for both.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004384 *
4385 * A search context returned by a previous call to vim_findfile_init() can be
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004386 * passed in the parameter "search_ctx_arg". This context is reused and
4387 * reinitialized with the new parameters. The list of already visited
Bram Moolenaar071d4272004-06-13 20:20:40 +00004388 * directories from this context is only deleted if the parameter
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004389 * "free_visited" is true. Be aware that the passed "search_ctx_arg" is freed
4390 * if the reinitialization fails.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004391 *
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004392 * If you don't have a search context from a previous call "search_ctx_arg"
4393 * must be NULL.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004394 *
4395 * This function silently ignores a few errors, vim_findfile() will have
4396 * limited functionality then.
4397 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004398 void *
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004399vim_findfile_init(path, filename, stopdirs, level, free_visited, find_what,
4400 search_ctx_arg, tagfile, rel_fname)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004401 char_u *path;
4402 char_u *filename;
Bram Moolenaare1fbddc2009-05-16 19:07:03 +00004403 char_u *stopdirs UNUSED;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004404 int level;
4405 int free_visited;
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004406 int find_what;
4407 void *search_ctx_arg;
Bram Moolenaar463ee342010-08-08 18:17:52 +02004408 int tagfile; /* expanding names of tags files */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004409 char_u *rel_fname; /* file name to use for "." */
4410{
4411#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004412 char_u *wc_part;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004413#endif
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004414 ff_stack_T *sptr;
4415 ff_search_ctx_T *search_ctx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004416
4417 /* If a search context is given by the caller, reuse it, else allocate a
4418 * new one.
4419 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004420 if (search_ctx_arg != NULL)
4421 search_ctx = search_ctx_arg;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004422 else
4423 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004424 search_ctx = (ff_search_ctx_T*)alloc((unsigned)sizeof(ff_search_ctx_T));
4425 if (search_ctx == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004426 goto error_return;
Bram Moolenaar7db5fc82010-05-24 11:59:29 +02004427 vim_memset(search_ctx, 0, sizeof(ff_search_ctx_T));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004428 }
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004429 search_ctx->ffsc_find_what = find_what;
Bram Moolenaar463ee342010-08-08 18:17:52 +02004430 search_ctx->ffsc_tagfile = tagfile;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004431
4432 /* clear the search context, but NOT the visited lists */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004433 ff_clear(search_ctx);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004434
4435 /* clear visited list if wanted */
4436 if (free_visited == TRUE)
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004437 vim_findfile_free_visited(search_ctx);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004438 else
4439 {
4440 /* Reuse old visited lists. Get the visited list for the given
4441 * filename. If no list for the current filename exists, creates a new
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004442 * one. */
4443 search_ctx->ffsc_visited_list = ff_get_visited_list(filename,
4444 &search_ctx->ffsc_visited_lists_list);
4445 if (search_ctx->ffsc_visited_list == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004446 goto error_return;
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004447 search_ctx->ffsc_dir_visited_list = ff_get_visited_list(filename,
4448 &search_ctx->ffsc_dir_visited_lists_list);
4449 if (search_ctx->ffsc_dir_visited_list == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004450 goto error_return;
4451 }
4452
4453 if (ff_expand_buffer == NULL)
4454 {
4455 ff_expand_buffer = (char_u*)alloc(MAXPATHL);
4456 if (ff_expand_buffer == NULL)
4457 goto error_return;
4458 }
4459
4460 /* Store information on starting dir now if path is relative.
Bram Moolenaar5eb86f92004-07-26 12:53:41 +00004461 * If path is absolute, we do that later. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004462 if (path[0] == '.'
4463 && (vim_ispathsep(path[1]) || path[1] == NUL)
4464 && (!tagfile || vim_strchr(p_cpo, CPO_DOTTAG) == NULL)
4465 && rel_fname != NULL)
4466 {
4467 int len = (int)(gettail(rel_fname) - rel_fname);
4468
4469 if (!vim_isAbsName(rel_fname) && len + 1 < MAXPATHL)
4470 {
4471 /* Make the start dir an absolute path name. */
Bram Moolenaarbbebc852005-07-18 21:47:53 +00004472 vim_strncpy(ff_expand_buffer, rel_fname, len);
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004473 search_ctx->ffsc_start_dir = FullName_save(ff_expand_buffer, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004474 }
4475 else
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004476 search_ctx->ffsc_start_dir = vim_strnsave(rel_fname, len);
4477 if (search_ctx->ffsc_start_dir == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004478 goto error_return;
4479 if (*++path != NUL)
4480 ++path;
4481 }
4482 else if (*path == NUL || !vim_isAbsName(path))
4483 {
4484#ifdef BACKSLASH_IN_FILENAME
4485 /* "c:dir" needs "c:" to be expanded, otherwise use current dir */
4486 if (*path != NUL && path[1] == ':')
4487 {
4488 char_u drive[3];
4489
4490 drive[0] = path[0];
4491 drive[1] = ':';
4492 drive[2] = NUL;
4493 if (vim_FullName(drive, ff_expand_buffer, MAXPATHL, TRUE) == FAIL)
4494 goto error_return;
4495 path += 2;
4496 }
4497 else
4498#endif
4499 if (mch_dirname(ff_expand_buffer, MAXPATHL) == FAIL)
4500 goto error_return;
4501
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004502 search_ctx->ffsc_start_dir = vim_strsave(ff_expand_buffer);
4503 if (search_ctx->ffsc_start_dir == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004504 goto error_return;
4505
4506#ifdef BACKSLASH_IN_FILENAME
4507 /* A path that starts with "/dir" is relative to the drive, not to the
4508 * directory (but not for "//machine/dir"). Only use the drive name. */
4509 if ((*path == '/' || *path == '\\')
4510 && path[1] != path[0]
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004511 && search_ctx->ffsc_start_dir[1] == ':')
4512 search_ctx->ffsc_start_dir[2] = NUL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004513#endif
4514 }
4515
4516#ifdef FEAT_PATH_EXTRA
4517 /*
4518 * If stopdirs are given, split them into an array of pointers.
4519 * If this fails (mem allocation), there is no upward search at all or a
4520 * stop directory is not recognized -> continue silently.
4521 * If stopdirs just contains a ";" or is empty,
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004522 * search_ctx->ffsc_stopdirs_v will only contain a NULL pointer. This
Bram Moolenaar071d4272004-06-13 20:20:40 +00004523 * is handled as unlimited upward search. See function
4524 * ff_path_in_stoplist() for details.
4525 */
4526 if (stopdirs != NULL)
4527 {
4528 char_u *walker = stopdirs;
4529 int dircount;
4530
4531 while (*walker == ';')
4532 walker++;
4533
4534 dircount = 1;
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004535 search_ctx->ffsc_stopdirs_v =
4536 (char_u **)alloc((unsigned)sizeof(char_u *));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004537
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004538 if (search_ctx->ffsc_stopdirs_v != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004539 {
4540 do
4541 {
4542 char_u *helper;
4543 void *ptr;
4544
4545 helper = walker;
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004546 ptr = vim_realloc(search_ctx->ffsc_stopdirs_v,
Bram Moolenaar071d4272004-06-13 20:20:40 +00004547 (dircount + 1) * sizeof(char_u *));
4548 if (ptr)
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004549 search_ctx->ffsc_stopdirs_v = ptr;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004550 else
4551 /* ignore, keep what we have and continue */
4552 break;
4553 walker = vim_strchr(walker, ';');
4554 if (walker)
4555 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004556 search_ctx->ffsc_stopdirs_v[dircount-1] =
4557 vim_strnsave(helper, (int)(walker - helper));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004558 walker++;
4559 }
4560 else
4561 /* this might be "", which means ascent till top
4562 * of directory tree.
4563 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004564 search_ctx->ffsc_stopdirs_v[dircount-1] =
4565 vim_strsave(helper);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004566
4567 dircount++;
4568
4569 } while (walker != NULL);
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004570 search_ctx->ffsc_stopdirs_v[dircount-1] = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004571 }
4572 }
4573#endif
4574
4575#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004576 search_ctx->ffsc_level = level;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004577
4578 /* split into:
4579 * -fix path
4580 * -wildcard_stuff (might be NULL)
4581 */
4582 wc_part = vim_strchr(path, '*');
4583 if (wc_part != NULL)
4584 {
4585 int llevel;
4586 int len;
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00004587 char *errpt;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004588
4589 /* save the fix part of the path */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004590 search_ctx->ffsc_fix_path = vim_strnsave(path, (int)(wc_part - path));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004591
4592 /*
4593 * copy wc_path and add restricts to the '**' wildcard.
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00004594 * The octet after a '**' is used as a (binary) counter.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004595 * So '**3' is transposed to '**^C' ('^C' is ASCII value 3)
4596 * or '**76' is transposed to '**N'( 'N' is ASCII value 76).
4597 * For EBCDIC you get different character values.
4598 * If no restrict is given after '**' the default is used.
Bram Moolenaar21160b92009-11-11 15:56:10 +00004599 * Due to this technique the path looks awful if you print it as a
Bram Moolenaar071d4272004-06-13 20:20:40 +00004600 * string.
4601 */
4602 len = 0;
4603 while (*wc_part != NUL)
4604 {
Bram Moolenaarb9ba4032011-12-08 17:49:35 +01004605 if (len + 5 >= MAXPATHL)
4606 {
4607 EMSG(_(e_pathtoolong));
4608 break;
4609 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004610 if (STRNCMP(wc_part, "**", 2) == 0)
4611 {
4612 ff_expand_buffer[len++] = *wc_part++;
4613 ff_expand_buffer[len++] = *wc_part++;
4614
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00004615 llevel = strtol((char *)wc_part, &errpt, 10);
4616 if ((char_u *)errpt != wc_part && llevel > 0 && llevel < 255)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004617 ff_expand_buffer[len++] = llevel;
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00004618 else if ((char_u *)errpt != wc_part && llevel == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004619 /* restrict is 0 -> remove already added '**' */
4620 len -= 2;
4621 else
4622 ff_expand_buffer[len++] = FF_MAX_STAR_STAR_EXPAND;
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00004623 wc_part = (char_u *)errpt;
Bram Moolenaar1d94f9b2005-08-04 21:29:45 +00004624 if (*wc_part != NUL && !vim_ispathsep(*wc_part))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004625 {
4626 EMSG2(_("E343: Invalid path: '**[number]' must be at the end of the path or be followed by '%s'."), PATHSEPSTR);
4627 goto error_return;
4628 }
4629 }
4630 else
4631 ff_expand_buffer[len++] = *wc_part++;
4632 }
4633 ff_expand_buffer[len] = NUL;
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004634 search_ctx->ffsc_wc_path = vim_strsave(ff_expand_buffer);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004635
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004636 if (search_ctx->ffsc_wc_path == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004637 goto error_return;
4638 }
4639 else
4640#endif
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004641 search_ctx->ffsc_fix_path = vim_strsave(path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004642
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004643 if (search_ctx->ffsc_start_dir == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004644 {
4645 /* store the fix part as startdir.
4646 * This is needed if the parameter path is fully qualified.
4647 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004648 search_ctx->ffsc_start_dir = vim_strsave(search_ctx->ffsc_fix_path);
Bram Moolenaara9d52e32010-07-31 16:44:19 +02004649 if (search_ctx->ffsc_start_dir == NULL)
4650 goto error_return;
4651 search_ctx->ffsc_fix_path[0] = NUL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004652 }
4653
4654 /* create an absolute path */
Bram Moolenaarb9ba4032011-12-08 17:49:35 +01004655 if (STRLEN(search_ctx->ffsc_start_dir)
4656 + STRLEN(search_ctx->ffsc_fix_path) + 3 >= MAXPATHL)
4657 {
4658 EMSG(_(e_pathtoolong));
4659 goto error_return;
4660 }
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004661 STRCPY(ff_expand_buffer, search_ctx->ffsc_start_dir);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004662 add_pathsep(ff_expand_buffer);
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004663 STRCAT(ff_expand_buffer, search_ctx->ffsc_fix_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004664 add_pathsep(ff_expand_buffer);
4665
4666 sptr = ff_create_stack_element(ff_expand_buffer,
4667#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004668 search_ctx->ffsc_wc_path,
Bram Moolenaar071d4272004-06-13 20:20:40 +00004669#endif
4670 level, 0);
4671
4672 if (sptr == NULL)
4673 goto error_return;
4674
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004675 ff_push(search_ctx, sptr);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004676
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004677 search_ctx->ffsc_file_to_search = vim_strsave(filename);
4678 if (search_ctx->ffsc_file_to_search == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004679 goto error_return;
4680
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004681 return search_ctx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004682
4683error_return:
4684 /*
4685 * We clear the search context now!
4686 * Even when the caller gave us a (perhaps valid) context we free it here,
4687 * as we might have already destroyed it.
4688 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004689 vim_findfile_cleanup(search_ctx);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004690 return NULL;
4691}
4692
4693#if defined(FEAT_PATH_EXTRA) || defined(PROTO)
4694/*
4695 * Get the stopdir string. Check that ';' is not escaped.
4696 */
4697 char_u *
4698vim_findfile_stopdir(buf)
4699 char_u *buf;
4700{
4701 char_u *r_ptr = buf;
4702
4703 while (*r_ptr != NUL && *r_ptr != ';')
4704 {
4705 if (r_ptr[0] == '\\' && r_ptr[1] == ';')
4706 {
Bram Moolenaar0b573a52011-07-27 17:31:47 +02004707 /* Overwrite the escape char,
4708 * use STRLEN(r_ptr) to move the trailing '\0'. */
Bram Moolenaar446cb832008-06-24 21:56:24 +00004709 STRMOVE(r_ptr, r_ptr + 1);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004710 r_ptr++;
4711 }
4712 r_ptr++;
4713 }
4714 if (*r_ptr == ';')
4715 {
4716 *r_ptr = 0;
4717 r_ptr++;
4718 }
4719 else if (*r_ptr == NUL)
4720 r_ptr = NULL;
4721 return r_ptr;
4722}
4723#endif
4724
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004725/*
4726 * Clean up the given search context. Can handle a NULL pointer.
4727 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004728 void
4729vim_findfile_cleanup(ctx)
4730 void *ctx;
4731{
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00004732 if (ctx == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004733 return;
4734
Bram Moolenaar071d4272004-06-13 20:20:40 +00004735 vim_findfile_free_visited(ctx);
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004736 ff_clear(ctx);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004737 vim_free(ctx);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004738}
4739
4740/*
4741 * Find a file in a search context.
4742 * The search context was created with vim_findfile_init() above.
4743 * Return a pointer to an allocated file name or NULL if nothing found.
4744 * To get all matching files call this function until you get NULL.
4745 *
Bram Moolenaar5eb86f92004-07-26 12:53:41 +00004746 * If the passed search_context is NULL, NULL is returned.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004747 *
4748 * The search algorithm is depth first. To change this replace the
4749 * stack with a list (don't forget to leave partly searched directories on the
4750 * top of the list).
4751 */
4752 char_u *
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004753vim_findfile(search_ctx_arg)
4754 void *search_ctx_arg;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004755{
4756 char_u *file_path;
4757#ifdef FEAT_PATH_EXTRA
4758 char_u *rest_of_wildcards;
4759 char_u *path_end = NULL;
4760#endif
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004761 ff_stack_T *stackp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004762#if defined(FEAT_SEARCHPATH) || defined(FEAT_PATH_EXTRA)
4763 int len;
4764#endif
4765 int i;
4766 char_u *p;
4767#ifdef FEAT_SEARCHPATH
4768 char_u *suf;
4769#endif
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004770 ff_search_ctx_T *search_ctx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004771
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004772 if (search_ctx_arg == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004773 return NULL;
4774
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004775 search_ctx = (ff_search_ctx_T *)search_ctx_arg;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004776
4777 /*
4778 * filepath is used as buffer for various actions and as the storage to
4779 * return a found filename.
4780 */
4781 if ((file_path = alloc((int)MAXPATHL)) == NULL)
4782 return NULL;
4783
4784#ifdef FEAT_PATH_EXTRA
4785 /* store the end of the start dir -- needed for upward search */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004786 if (search_ctx->ffsc_start_dir != NULL)
4787 path_end = &search_ctx->ffsc_start_dir[
4788 STRLEN(search_ctx->ffsc_start_dir)];
Bram Moolenaar071d4272004-06-13 20:20:40 +00004789#endif
4790
4791#ifdef FEAT_PATH_EXTRA
4792 /* upward search loop */
4793 for (;;)
4794 {
4795#endif
4796 /* downward search loop */
4797 for (;;)
4798 {
4799 /* check if user user wants to stop the search*/
4800 ui_breakcheck();
4801 if (got_int)
4802 break;
4803
4804 /* get directory to work on from stack */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004805 stackp = ff_pop(search_ctx);
4806 if (stackp == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004807 break;
4808
4809 /*
4810 * TODO: decide if we leave this test in
4811 *
4812 * GOOD: don't search a directory(-tree) twice.
4813 * BAD: - check linked list for every new directory entered.
4814 * - check for double files also done below
4815 *
4816 * Here we check if we already searched this directory.
4817 * We already searched a directory if:
4818 * 1) The directory is the same.
4819 * 2) We would use the same wildcard string.
4820 *
4821 * Good if you have links on same directory via several ways
4822 * or you have selfreferences in directories (e.g. SuSE Linux 6.3:
4823 * /etc/rc.d/init.d is linked to /etc/rc.d -> endless loop)
4824 *
4825 * This check is only needed for directories we work on for the
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004826 * first time (hence stackp->ff_filearray == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004827 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004828 if (stackp->ffs_filearray == NULL
4829 && ff_check_visited(&search_ctx->ffsc_dir_visited_list
Bram Moolenaar071d4272004-06-13 20:20:40 +00004830 ->ffvl_visited_list,
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004831 stackp->ffs_fix_path
Bram Moolenaar071d4272004-06-13 20:20:40 +00004832#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004833 , stackp->ffs_wc_path
Bram Moolenaar071d4272004-06-13 20:20:40 +00004834#endif
4835 ) == FAIL)
4836 {
4837#ifdef FF_VERBOSE
4838 if (p_verbose >= 5)
4839 {
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004840 verbose_enter_scroll();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004841 smsg((char_u *)"Already Searched: %s (%s)",
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004842 stackp->ffs_fix_path, stackp->ffs_wc_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004843 /* don't overwrite this either */
4844 msg_puts((char_u *)"\n");
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004845 verbose_leave_scroll();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004846 }
4847#endif
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004848 ff_free_stack_element(stackp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004849 continue;
4850 }
4851#ifdef FF_VERBOSE
4852 else if (p_verbose >= 5)
4853 {
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004854 verbose_enter_scroll();
Bram Moolenaar051b7822005-05-19 21:00:46 +00004855 smsg((char_u *)"Searching: %s (%s)",
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004856 stackp->ffs_fix_path, stackp->ffs_wc_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004857 /* don't overwrite this either */
4858 msg_puts((char_u *)"\n");
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004859 verbose_leave_scroll();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004860 }
4861#endif
4862
4863 /* check depth */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004864 if (stackp->ffs_level <= 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004865 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004866 ff_free_stack_element(stackp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004867 continue;
4868 }
4869
4870 file_path[0] = NUL;
4871
4872 /*
4873 * If no filearray till now expand wildcards
4874 * The function expand_wildcards() can handle an array of paths
4875 * and all possible expands are returned in one array. We use this
4876 * to handle the expansion of '**' into an empty string.
4877 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004878 if (stackp->ffs_filearray == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004879 {
4880 char_u *dirptrs[2];
4881
4882 /* we use filepath to build the path expand_wildcards() should
4883 * expand.
4884 */
4885 dirptrs[0] = file_path;
4886 dirptrs[1] = NULL;
4887
4888 /* if we have a start dir copy it in */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004889 if (!vim_isAbsName(stackp->ffs_fix_path)
4890 && search_ctx->ffsc_start_dir)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004891 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004892 STRCPY(file_path, search_ctx->ffsc_start_dir);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004893 add_pathsep(file_path);
4894 }
4895
4896 /* append the fix part of the search path */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004897 STRCAT(file_path, stackp->ffs_fix_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004898 add_pathsep(file_path);
4899
4900#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004901 rest_of_wildcards = stackp->ffs_wc_path;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004902 if (*rest_of_wildcards != NUL)
4903 {
4904 len = (int)STRLEN(file_path);
4905 if (STRNCMP(rest_of_wildcards, "**", 2) == 0)
4906 {
4907 /* pointer to the restrict byte
4908 * The restrict byte is not a character!
4909 */
4910 p = rest_of_wildcards + 2;
4911
4912 if (*p > 0)
4913 {
4914 (*p)--;
4915 file_path[len++] = '*';
4916 }
4917
4918 if (*p == 0)
4919 {
4920 /* remove '**<numb> from wildcards */
Bram Moolenaar446cb832008-06-24 21:56:24 +00004921 STRMOVE(rest_of_wildcards, rest_of_wildcards + 3);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004922 }
4923 else
4924 rest_of_wildcards += 3;
4925
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004926 if (stackp->ffs_star_star_empty == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004927 {
4928 /* if not done before, expand '**' to empty */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004929 stackp->ffs_star_star_empty = 1;
4930 dirptrs[1] = stackp->ffs_fix_path;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004931 }
4932 }
4933
4934 /*
4935 * Here we copy until the next path separator or the end of
4936 * the path. If we stop at a path separator, there is
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00004937 * still something else left. This is handled below by
Bram Moolenaar071d4272004-06-13 20:20:40 +00004938 * pushing every directory returned from expand_wildcards()
4939 * on the stack again for further search.
4940 */
4941 while (*rest_of_wildcards
4942 && !vim_ispathsep(*rest_of_wildcards))
4943 file_path[len++] = *rest_of_wildcards++;
4944
4945 file_path[len] = NUL;
4946 if (vim_ispathsep(*rest_of_wildcards))
4947 rest_of_wildcards++;
4948 }
4949#endif
4950
4951 /*
4952 * Expand wildcards like "*" and "$VAR".
4953 * If the path is a URL don't try this.
4954 */
4955 if (path_with_url(dirptrs[0]))
4956 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004957 stackp->ffs_filearray = (char_u **)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004958 alloc((unsigned)sizeof(char *));
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004959 if (stackp->ffs_filearray != NULL
4960 && (stackp->ffs_filearray[0]
Bram Moolenaar071d4272004-06-13 20:20:40 +00004961 = vim_strsave(dirptrs[0])) != NULL)
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004962 stackp->ffs_filearray_size = 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004963 else
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004964 stackp->ffs_filearray_size = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004965 }
4966 else
Bram Moolenaar0b573a52011-07-27 17:31:47 +02004967 /* Add EW_NOTWILD because the expanded path may contain
4968 * wildcard characters that are to be taken literally.
4969 * This is a bit of a hack. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004970 expand_wildcards((dirptrs[1] == NULL) ? 1 : 2, dirptrs,
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004971 &stackp->ffs_filearray_size,
4972 &stackp->ffs_filearray,
Bram Moolenaar0b573a52011-07-27 17:31:47 +02004973 EW_DIR|EW_ADDSLASH|EW_SILENT|EW_NOTWILD);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004974
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004975 stackp->ffs_filearray_cur = 0;
4976 stackp->ffs_stage = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004977 }
4978#ifdef FEAT_PATH_EXTRA
4979 else
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004980 rest_of_wildcards = &stackp->ffs_wc_path[
4981 STRLEN(stackp->ffs_wc_path)];
Bram Moolenaar071d4272004-06-13 20:20:40 +00004982#endif
4983
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004984 if (stackp->ffs_stage == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004985 {
4986 /* this is the first time we work on this directory */
4987#ifdef FEAT_PATH_EXTRA
4988 if (*rest_of_wildcards == NUL)
4989#endif
4990 {
4991 /*
4992 * we don't have further wildcards to expand, so we have to
4993 * check for the final file now
4994 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004995 for (i = stackp->ffs_filearray_cur;
4996 i < stackp->ffs_filearray_size; ++i)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004997 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004998 if (!path_with_url(stackp->ffs_filearray[i])
4999 && !mch_isdir(stackp->ffs_filearray[i]))
Bram Moolenaar071d4272004-06-13 20:20:40 +00005000 continue; /* not a directory */
5001
Bram Moolenaar21160b92009-11-11 15:56:10 +00005002 /* prepare the filename to be checked for existence
Bram Moolenaar071d4272004-06-13 20:20:40 +00005003 * below */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005004 STRCPY(file_path, stackp->ffs_filearray[i]);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005005 add_pathsep(file_path);
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005006 STRCAT(file_path, search_ctx->ffsc_file_to_search);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005007
5008 /*
5009 * Try without extra suffix and then with suffixes
5010 * from 'suffixesadd'.
5011 */
5012#ifdef FEAT_SEARCHPATH
5013 len = (int)STRLEN(file_path);
Bram Moolenaar463ee342010-08-08 18:17:52 +02005014 if (search_ctx->ffsc_tagfile)
5015 suf = (char_u *)"";
5016 else
5017 suf = curbuf->b_p_sua;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005018 for (;;)
5019#endif
5020 {
5021 /* if file exists and we didn't already find it */
5022 if ((path_with_url(file_path)
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005023 || (mch_getperm(file_path) >= 0
5024 && (search_ctx->ffsc_find_what
5025 == FINDFILE_BOTH
5026 || ((search_ctx->ffsc_find_what
5027 == FINDFILE_DIR)
5028 == mch_isdir(file_path)))))
Bram Moolenaar071d4272004-06-13 20:20:40 +00005029#ifndef FF_VERBOSE
5030 && (ff_check_visited(
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005031 &search_ctx->ffsc_visited_list->ffvl_visited_list,
Bram Moolenaar071d4272004-06-13 20:20:40 +00005032 file_path
5033#ifdef FEAT_PATH_EXTRA
5034 , (char_u *)""
5035#endif
5036 ) == OK)
5037#endif
5038 )
5039 {
5040#ifdef FF_VERBOSE
5041 if (ff_check_visited(
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005042 &search_ctx->ffsc_visited_list->ffvl_visited_list,
Bram Moolenaar071d4272004-06-13 20:20:40 +00005043 file_path
5044#ifdef FEAT_PATH_EXTRA
5045 , (char_u *)""
5046#endif
5047 ) == FAIL)
5048 {
5049 if (p_verbose >= 5)
5050 {
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00005051 verbose_enter_scroll();
Bram Moolenaar051b7822005-05-19 21:00:46 +00005052 smsg((char_u *)"Already: %s",
Bram Moolenaar071d4272004-06-13 20:20:40 +00005053 file_path);
5054 /* don't overwrite this either */
5055 msg_puts((char_u *)"\n");
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00005056 verbose_leave_scroll();
Bram Moolenaar071d4272004-06-13 20:20:40 +00005057 }
5058 continue;
5059 }
5060#endif
5061
5062 /* push dir to examine rest of subdirs later */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005063 stackp->ffs_filearray_cur = i + 1;
5064 ff_push(search_ctx, stackp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005065
Bram Moolenaar6bab9fa2009-01-22 20:32:12 +00005066 if (!path_with_url(file_path))
5067 simplify_filename(file_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005068 if (mch_dirname(ff_expand_buffer, MAXPATHL)
5069 == OK)
5070 {
5071 p = shorten_fname(file_path,
5072 ff_expand_buffer);
5073 if (p != NULL)
Bram Moolenaar446cb832008-06-24 21:56:24 +00005074 STRMOVE(file_path, p);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005075 }
5076#ifdef FF_VERBOSE
5077 if (p_verbose >= 5)
5078 {
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00005079 verbose_enter_scroll();
Bram Moolenaar051b7822005-05-19 21:00:46 +00005080 smsg((char_u *)"HIT: %s", file_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005081 /* don't overwrite this either */
5082 msg_puts((char_u *)"\n");
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00005083 verbose_leave_scroll();
Bram Moolenaar071d4272004-06-13 20:20:40 +00005084 }
5085#endif
5086 return file_path;
5087 }
5088
5089#ifdef FEAT_SEARCHPATH
5090 /* Not found or found already, try next suffix. */
5091 if (*suf == NUL)
5092 break;
5093 copy_option_part(&suf, file_path + len,
5094 MAXPATHL - len, ",");
5095#endif
5096 }
5097 }
5098 }
5099#ifdef FEAT_PATH_EXTRA
5100 else
5101 {
5102 /*
5103 * still wildcards left, push the directories for further
5104 * search
5105 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005106 for (i = stackp->ffs_filearray_cur;
5107 i < stackp->ffs_filearray_size; ++i)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005108 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005109 if (!mch_isdir(stackp->ffs_filearray[i]))
Bram Moolenaar071d4272004-06-13 20:20:40 +00005110 continue; /* not a directory */
5111
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005112 ff_push(search_ctx,
5113 ff_create_stack_element(
5114 stackp->ffs_filearray[i],
5115 rest_of_wildcards,
5116 stackp->ffs_level - 1, 0));
Bram Moolenaar071d4272004-06-13 20:20:40 +00005117 }
5118 }
5119#endif
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005120 stackp->ffs_filearray_cur = 0;
5121 stackp->ffs_stage = 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005122 }
5123
5124#ifdef FEAT_PATH_EXTRA
5125 /*
5126 * if wildcards contains '**' we have to descent till we reach the
5127 * leaves of the directory tree.
5128 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005129 if (STRNCMP(stackp->ffs_wc_path, "**", 2) == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005130 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005131 for (i = stackp->ffs_filearray_cur;
5132 i < stackp->ffs_filearray_size; ++i)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005133 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005134 if (fnamecmp(stackp->ffs_filearray[i],
5135 stackp->ffs_fix_path) == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005136 continue; /* don't repush same directory */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005137 if (!mch_isdir(stackp->ffs_filearray[i]))
Bram Moolenaar071d4272004-06-13 20:20:40 +00005138 continue; /* not a directory */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005139 ff_push(search_ctx,
5140 ff_create_stack_element(stackp->ffs_filearray[i],
5141 stackp->ffs_wc_path, stackp->ffs_level - 1, 1));
Bram Moolenaar071d4272004-06-13 20:20:40 +00005142 }
5143 }
5144#endif
5145
5146 /* we are done with the current directory */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005147 ff_free_stack_element(stackp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005148
5149 }
5150
5151#ifdef FEAT_PATH_EXTRA
5152 /* If we reached this, we didn't find anything downwards.
5153 * Let's check if we should do an upward search.
5154 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005155 if (search_ctx->ffsc_start_dir
5156 && search_ctx->ffsc_stopdirs_v != NULL && !got_int)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005157 {
5158 ff_stack_T *sptr;
5159
5160 /* is the last starting directory in the stop list? */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005161 if (ff_path_in_stoplist(search_ctx->ffsc_start_dir,
5162 (int)(path_end - search_ctx->ffsc_start_dir),
5163 search_ctx->ffsc_stopdirs_v) == TRUE)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005164 break;
5165
5166 /* cut of last dir */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005167 while (path_end > search_ctx->ffsc_start_dir
5168 && vim_ispathsep(*path_end))
Bram Moolenaar071d4272004-06-13 20:20:40 +00005169 path_end--;
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005170 while (path_end > search_ctx->ffsc_start_dir
5171 && !vim_ispathsep(path_end[-1]))
Bram Moolenaar071d4272004-06-13 20:20:40 +00005172 path_end--;
5173 *path_end = 0;
5174 path_end--;
5175
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005176 if (*search_ctx->ffsc_start_dir == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005177 break;
5178
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005179 STRCPY(file_path, search_ctx->ffsc_start_dir);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005180 add_pathsep(file_path);
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005181 STRCAT(file_path, search_ctx->ffsc_fix_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005182
5183 /* create a new stack entry */
5184 sptr = ff_create_stack_element(file_path,
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005185 search_ctx->ffsc_wc_path, search_ctx->ffsc_level, 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005186 if (sptr == NULL)
5187 break;
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005188 ff_push(search_ctx, sptr);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005189 }
5190 else
5191 break;
5192 }
5193#endif
5194
5195 vim_free(file_path);
5196 return NULL;
5197}
5198
5199/*
5200 * Free the list of lists of visited files and directories
5201 * Can handle it if the passed search_context is NULL;
5202 */
5203 void
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005204vim_findfile_free_visited(search_ctx_arg)
5205 void *search_ctx_arg;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005206{
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005207 ff_search_ctx_T *search_ctx;
5208
5209 if (search_ctx_arg == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005210 return;
5211
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005212 search_ctx = (ff_search_ctx_T *)search_ctx_arg;
5213 vim_findfile_free_visited_list(&search_ctx->ffsc_visited_lists_list);
5214 vim_findfile_free_visited_list(&search_ctx->ffsc_dir_visited_lists_list);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005215}
5216
5217 static void
5218vim_findfile_free_visited_list(list_headp)
5219 ff_visited_list_hdr_T **list_headp;
5220{
5221 ff_visited_list_hdr_T *vp;
5222
5223 while (*list_headp != NULL)
5224 {
5225 vp = (*list_headp)->ffvl_next;
5226 ff_free_visited_list((*list_headp)->ffvl_visited_list);
5227
5228 vim_free((*list_headp)->ffvl_filename);
5229 vim_free(*list_headp);
5230 *list_headp = vp;
5231 }
5232 *list_headp = NULL;
5233}
5234
5235 static void
5236ff_free_visited_list(vl)
5237 ff_visited_T *vl;
5238{
5239 ff_visited_T *vp;
5240
5241 while (vl != NULL)
5242 {
5243 vp = vl->ffv_next;
5244#ifdef FEAT_PATH_EXTRA
5245 vim_free(vl->ffv_wc_path);
5246#endif
5247 vim_free(vl);
5248 vl = vp;
5249 }
5250 vl = NULL;
5251}
5252
5253/*
5254 * Returns the already visited list for the given filename. If none is found it
5255 * allocates a new one.
5256 */
5257 static ff_visited_list_hdr_T*
5258ff_get_visited_list(filename, list_headp)
5259 char_u *filename;
5260 ff_visited_list_hdr_T **list_headp;
5261{
5262 ff_visited_list_hdr_T *retptr = NULL;
5263
5264 /* check if a visited list for the given filename exists */
5265 if (*list_headp != NULL)
5266 {
5267 retptr = *list_headp;
5268 while (retptr != NULL)
5269 {
5270 if (fnamecmp(filename, retptr->ffvl_filename) == 0)
5271 {
5272#ifdef FF_VERBOSE
5273 if (p_verbose >= 5)
5274 {
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00005275 verbose_enter_scroll();
Bram Moolenaar051b7822005-05-19 21:00:46 +00005276 smsg((char_u *)"ff_get_visited_list: FOUND list for %s",
Bram Moolenaar071d4272004-06-13 20:20:40 +00005277 filename);
5278 /* don't overwrite this either */
5279 msg_puts((char_u *)"\n");
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00005280 verbose_leave_scroll();
Bram Moolenaar071d4272004-06-13 20:20:40 +00005281 }
5282#endif
5283 return retptr;
5284 }
5285 retptr = retptr->ffvl_next;
5286 }
5287 }
5288
5289#ifdef FF_VERBOSE
5290 if (p_verbose >= 5)
5291 {
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00005292 verbose_enter_scroll();
Bram Moolenaar051b7822005-05-19 21:00:46 +00005293 smsg((char_u *)"ff_get_visited_list: new list for %s", filename);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005294 /* don't overwrite this either */
5295 msg_puts((char_u *)"\n");
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00005296 verbose_leave_scroll();
Bram Moolenaar071d4272004-06-13 20:20:40 +00005297 }
5298#endif
5299
5300 /*
5301 * if we reach this we didn't find a list and we have to allocate new list
5302 */
5303 retptr = (ff_visited_list_hdr_T*)alloc((unsigned)sizeof(*retptr));
5304 if (retptr == NULL)
5305 return NULL;
5306
5307 retptr->ffvl_visited_list = NULL;
5308 retptr->ffvl_filename = vim_strsave(filename);
5309 if (retptr->ffvl_filename == NULL)
5310 {
5311 vim_free(retptr);
5312 return NULL;
5313 }
5314 retptr->ffvl_next = *list_headp;
5315 *list_headp = retptr;
5316
5317 return retptr;
5318}
5319
5320#ifdef FEAT_PATH_EXTRA
5321/*
5322 * check if two wildcard paths are equal. Returns TRUE or FALSE.
5323 * They are equal if:
5324 * - both paths are NULL
5325 * - they have the same length
5326 * - char by char comparison is OK
5327 * - the only differences are in the counters behind a '**', so
5328 * '**\20' is equal to '**\24'
5329 */
5330 static int
5331ff_wc_equal(s1, s2)
5332 char_u *s1;
5333 char_u *s2;
5334{
5335 int i;
5336
5337 if (s1 == s2)
5338 return TRUE;
5339
5340 if (s1 == NULL || s2 == NULL)
5341 return FALSE;
5342
5343 if (STRLEN(s1) != STRLEN(s2))
5344 return FAIL;
5345
5346 for (i = 0; s1[i] != NUL && s2[i] != NUL; i++)
5347 {
5348 if (s1[i] != s2[i]
5349#ifdef CASE_INSENSITIVE_FILENAME
5350 && TOUPPER_LOC(s1[i]) != TOUPPER_LOC(s2[i])
5351#endif
5352 )
5353 {
5354 if (i >= 2)
5355 if (s1[i-1] == '*' && s1[i-2] == '*')
5356 continue;
5357 else
5358 return FAIL;
5359 else
5360 return FAIL;
5361 }
5362 }
5363 return TRUE;
5364}
5365#endif
5366
5367/*
5368 * maintains the list of already visited files and dirs
5369 * returns FAIL if the given file/dir is already in the list
5370 * returns OK if it is newly added
5371 *
5372 * TODO: What to do on memory allocation problems?
5373 * -> return TRUE - Better the file is found several times instead of
5374 * never.
5375 */
5376 static int
5377ff_check_visited(visited_list, fname
5378#ifdef FEAT_PATH_EXTRA
5379 , wc_path
5380#endif
5381 )
5382 ff_visited_T **visited_list;
5383 char_u *fname;
5384#ifdef FEAT_PATH_EXTRA
5385 char_u *wc_path;
5386#endif
5387{
5388 ff_visited_T *vp;
5389#ifdef UNIX
5390 struct stat st;
5391 int url = FALSE;
5392#endif
5393
5394 /* For an URL we only compare the name, otherwise we compare the
5395 * device/inode (unix) or the full path name (not Unix). */
5396 if (path_with_url(fname))
5397 {
Bram Moolenaarbbebc852005-07-18 21:47:53 +00005398 vim_strncpy(ff_expand_buffer, fname, MAXPATHL - 1);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005399#ifdef UNIX
5400 url = TRUE;
5401#endif
5402 }
5403 else
5404 {
5405 ff_expand_buffer[0] = NUL;
5406#ifdef UNIX
5407 if (mch_stat((char *)fname, &st) < 0)
5408#else
5409 if (vim_FullName(fname, ff_expand_buffer, MAXPATHL, TRUE) == FAIL)
5410#endif
5411 return FAIL;
5412 }
5413
5414 /* check against list of already visited files */
5415 for (vp = *visited_list; vp != NULL; vp = vp->ffv_next)
5416 {
5417 if (
5418#ifdef UNIX
Bram Moolenaare1fbddc2009-05-16 19:07:03 +00005419 !url ? (vp->ffv_dev_valid && vp->ffv_dev == st.st_dev
5420 && vp->ffv_ino == st.st_ino)
5421 :
Bram Moolenaar071d4272004-06-13 20:20:40 +00005422#endif
5423 fnamecmp(vp->ffv_fname, ff_expand_buffer) == 0
5424 )
5425 {
5426#ifdef FEAT_PATH_EXTRA
5427 /* are the wildcard parts equal */
5428 if (ff_wc_equal(vp->ffv_wc_path, wc_path) == TRUE)
5429#endif
5430 /* already visited */
5431 return FAIL;
5432 }
5433 }
5434
5435 /*
5436 * New file/dir. Add it to the list of visited files/dirs.
5437 */
5438 vp = (ff_visited_T *)alloc((unsigned)(sizeof(ff_visited_T)
5439 + STRLEN(ff_expand_buffer)));
5440
5441 if (vp != NULL)
5442 {
5443#ifdef UNIX
5444 if (!url)
5445 {
Bram Moolenaare1fbddc2009-05-16 19:07:03 +00005446 vp->ffv_dev_valid = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005447 vp->ffv_ino = st.st_ino;
5448 vp->ffv_dev = st.st_dev;
5449 vp->ffv_fname[0] = NUL;
5450 }
5451 else
5452 {
Bram Moolenaare1fbddc2009-05-16 19:07:03 +00005453 vp->ffv_dev_valid = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005454#endif
5455 STRCPY(vp->ffv_fname, ff_expand_buffer);
5456#ifdef UNIX
5457 }
5458#endif
5459#ifdef FEAT_PATH_EXTRA
5460 if (wc_path != NULL)
5461 vp->ffv_wc_path = vim_strsave(wc_path);
5462 else
5463 vp->ffv_wc_path = NULL;
5464#endif
5465
5466 vp->ffv_next = *visited_list;
5467 *visited_list = vp;
5468 }
5469
5470 return OK;
5471}
5472
5473/*
5474 * create stack element from given path pieces
5475 */
5476 static ff_stack_T *
5477ff_create_stack_element(fix_part,
5478#ifdef FEAT_PATH_EXTRA
5479 wc_part,
5480#endif
5481 level, star_star_empty)
5482 char_u *fix_part;
5483#ifdef FEAT_PATH_EXTRA
5484 char_u *wc_part;
5485#endif
5486 int level;
5487 int star_star_empty;
5488{
5489 ff_stack_T *new;
5490
5491 new = (ff_stack_T *)alloc((unsigned)sizeof(ff_stack_T));
5492 if (new == NULL)
5493 return NULL;
5494
5495 new->ffs_prev = NULL;
5496 new->ffs_filearray = NULL;
5497 new->ffs_filearray_size = 0;
5498 new->ffs_filearray_cur = 0;
5499 new->ffs_stage = 0;
5500 new->ffs_level = level;
5501 new->ffs_star_star_empty = star_star_empty;;
5502
5503 /* the following saves NULL pointer checks in vim_findfile */
5504 if (fix_part == NULL)
5505 fix_part = (char_u *)"";
5506 new->ffs_fix_path = vim_strsave(fix_part);
5507
5508#ifdef FEAT_PATH_EXTRA
5509 if (wc_part == NULL)
5510 wc_part = (char_u *)"";
5511 new->ffs_wc_path = vim_strsave(wc_part);
5512#endif
5513
5514 if (new->ffs_fix_path == NULL
5515#ifdef FEAT_PATH_EXTRA
5516 || new->ffs_wc_path == NULL
5517#endif
5518 )
5519 {
5520 ff_free_stack_element(new);
5521 new = NULL;
5522 }
5523
5524 return new;
5525}
5526
5527/*
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005528 * Push a dir on the directory stack.
Bram Moolenaar071d4272004-06-13 20:20:40 +00005529 */
5530 static void
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005531ff_push(search_ctx, stack_ptr)
5532 ff_search_ctx_T *search_ctx;
5533 ff_stack_T *stack_ptr;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005534{
5535 /* check for NULL pointer, not to return an error to the user, but
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005536 * to prevent a crash */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005537 if (stack_ptr != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005538 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005539 stack_ptr->ffs_prev = search_ctx->ffsc_stack_ptr;
5540 search_ctx->ffsc_stack_ptr = stack_ptr;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005541 }
5542}
5543
5544/*
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005545 * Pop a dir from the directory stack.
5546 * Returns NULL if stack is empty.
Bram Moolenaar071d4272004-06-13 20:20:40 +00005547 */
5548 static ff_stack_T *
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005549ff_pop(search_ctx)
5550 ff_search_ctx_T *search_ctx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005551{
5552 ff_stack_T *sptr;
5553
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005554 sptr = search_ctx->ffsc_stack_ptr;
5555 if (search_ctx->ffsc_stack_ptr != NULL)
5556 search_ctx->ffsc_stack_ptr = search_ctx->ffsc_stack_ptr->ffs_prev;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005557
5558 return sptr;
5559}
5560
5561/*
5562 * free the given stack element
5563 */
5564 static void
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005565ff_free_stack_element(stack_ptr)
5566 ff_stack_T *stack_ptr;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005567{
5568 /* vim_free handles possible NULL pointers */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005569 vim_free(stack_ptr->ffs_fix_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005570#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005571 vim_free(stack_ptr->ffs_wc_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005572#endif
5573
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005574 if (stack_ptr->ffs_filearray != NULL)
5575 FreeWild(stack_ptr->ffs_filearray_size, stack_ptr->ffs_filearray);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005576
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005577 vim_free(stack_ptr);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005578}
5579
5580/*
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005581 * Clear the search context, but NOT the visited list.
Bram Moolenaar071d4272004-06-13 20:20:40 +00005582 */
5583 static void
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005584ff_clear(search_ctx)
5585 ff_search_ctx_T *search_ctx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005586{
5587 ff_stack_T *sptr;
5588
5589 /* clear up stack */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005590 while ((sptr = ff_pop(search_ctx)) != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005591 ff_free_stack_element(sptr);
5592
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005593 vim_free(search_ctx->ffsc_file_to_search);
5594 vim_free(search_ctx->ffsc_start_dir);
5595 vim_free(search_ctx->ffsc_fix_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005596#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005597 vim_free(search_ctx->ffsc_wc_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005598#endif
5599
5600#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005601 if (search_ctx->ffsc_stopdirs_v != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005602 {
5603 int i = 0;
5604
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005605 while (search_ctx->ffsc_stopdirs_v[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005606 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005607 vim_free(search_ctx->ffsc_stopdirs_v[i]);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005608 i++;
5609 }
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005610 vim_free(search_ctx->ffsc_stopdirs_v);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005611 }
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005612 search_ctx->ffsc_stopdirs_v = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005613#endif
5614
5615 /* reset everything */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005616 search_ctx->ffsc_file_to_search = NULL;
5617 search_ctx->ffsc_start_dir = NULL;
5618 search_ctx->ffsc_fix_path = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005619#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005620 search_ctx->ffsc_wc_path = NULL;
5621 search_ctx->ffsc_level = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005622#endif
5623}
5624
5625#ifdef FEAT_PATH_EXTRA
5626/*
5627 * check if the given path is in the stopdirs
5628 * returns TRUE if yes else FALSE
5629 */
5630 static int
5631ff_path_in_stoplist(path, path_len, stopdirs_v)
5632 char_u *path;
5633 int path_len;
5634 char_u **stopdirs_v;
5635{
5636 int i = 0;
5637
5638 /* eat up trailing path separators, except the first */
Bram Moolenaar1d94f9b2005-08-04 21:29:45 +00005639 while (path_len > 1 && vim_ispathsep(path[path_len - 1]))
Bram Moolenaar071d4272004-06-13 20:20:40 +00005640 path_len--;
5641
5642 /* if no path consider it as match */
5643 if (path_len == 0)
5644 return TRUE;
5645
5646 for (i = 0; stopdirs_v[i] != NULL; i++)
5647 {
5648 if ((int)STRLEN(stopdirs_v[i]) > path_len)
5649 {
5650 /* match for parent directory. So '/home' also matches
5651 * '/home/rks'. Check for PATHSEP in stopdirs_v[i], else
5652 * '/home/r' would also match '/home/rks'
5653 */
5654 if (fnamencmp(stopdirs_v[i], path, path_len) == 0
Bram Moolenaar1d94f9b2005-08-04 21:29:45 +00005655 && vim_ispathsep(stopdirs_v[i][path_len]))
Bram Moolenaar071d4272004-06-13 20:20:40 +00005656 return TRUE;
5657 }
5658 else
5659 {
5660 if (fnamecmp(stopdirs_v[i], path) == 0)
5661 return TRUE;
5662 }
5663 }
5664 return FALSE;
5665}
5666#endif
5667
5668#if defined(FEAT_SEARCHPATH) || defined(PROTO)
5669/*
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005670 * Find the file name "ptr[len]" in the path. Also finds directory names.
Bram Moolenaar071d4272004-06-13 20:20:40 +00005671 *
5672 * On the first call set the parameter 'first' to TRUE to initialize
5673 * the search. For repeating calls to FALSE.
5674 *
5675 * Repeating calls will return other files called 'ptr[len]' from the path.
5676 *
5677 * Only on the first call 'ptr' and 'len' are used. For repeating calls they
5678 * don't need valid values.
5679 *
5680 * If nothing found on the first call the option FNAME_MESS will issue the
5681 * message:
5682 * 'Can't find file "<file>" in path'
5683 * On repeating calls:
5684 * 'No more file "<file>" found in path'
5685 *
5686 * options:
5687 * FNAME_MESS give error message when not found
5688 *
5689 * Uses NameBuff[]!
5690 *
5691 * Returns an allocated string for the file name. NULL for error.
5692 *
5693 */
5694 char_u *
5695find_file_in_path(ptr, len, options, first, rel_fname)
5696 char_u *ptr; /* file name */
5697 int len; /* length of file name */
5698 int options;
5699 int first; /* use count'th matching file name */
5700 char_u *rel_fname; /* file name searching relative to */
5701{
5702 return find_file_in_path_option(ptr, len, options, first,
5703 *curbuf->b_p_path == NUL ? p_path : curbuf->b_p_path,
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005704 FINDFILE_BOTH, rel_fname, curbuf->b_p_sua);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005705}
5706
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005707static char_u *ff_file_to_find = NULL;
5708static void *fdip_search_ctx = NULL;
5709
5710#if defined(EXITFREE)
5711 static void
5712free_findfile()
5713{
5714 vim_free(ff_file_to_find);
5715 vim_findfile_cleanup(fdip_search_ctx);
5716}
5717#endif
5718
Bram Moolenaar071d4272004-06-13 20:20:40 +00005719/*
5720 * Find the directory name "ptr[len]" in the path.
5721 *
5722 * options:
5723 * FNAME_MESS give error message when not found
5724 *
5725 * Uses NameBuff[]!
5726 *
5727 * Returns an allocated string for the file name. NULL for error.
5728 */
5729 char_u *
5730find_directory_in_path(ptr, len, options, rel_fname)
5731 char_u *ptr; /* file name */
5732 int len; /* length of file name */
5733 int options;
5734 char_u *rel_fname; /* file name searching relative to */
5735{
5736 return find_file_in_path_option(ptr, len, options, TRUE, p_cdpath,
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005737 FINDFILE_DIR, rel_fname, (char_u *)"");
Bram Moolenaar071d4272004-06-13 20:20:40 +00005738}
5739
Bram Moolenaar89cb5e02004-07-19 20:55:54 +00005740 char_u *
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005741find_file_in_path_option(ptr, len, options, first, path_option, find_what, rel_fname, suffixes)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005742 char_u *ptr; /* file name */
5743 int len; /* length of file name */
5744 int options;
5745 int first; /* use count'th matching file name */
5746 char_u *path_option; /* p_path or p_cdpath */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005747 int find_what; /* FINDFILE_FILE, _DIR or _BOTH */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005748 char_u *rel_fname; /* file name we are looking relative to. */
Bram Moolenaar433f7c82006-03-21 21:29:36 +00005749 char_u *suffixes; /* list of suffixes, 'suffixesadd' option */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005750{
Bram Moolenaar071d4272004-06-13 20:20:40 +00005751 static char_u *dir;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005752 static int did_findfile_init = FALSE;
5753 char_u save_char;
5754 char_u *file_name = NULL;
5755 char_u *buf = NULL;
5756 int rel_to_curdir;
5757#ifdef AMIGA
5758 struct Process *proc = (struct Process *)FindTask(0L);
5759 APTR save_winptr = proc->pr_WindowPtr;
5760
5761 /* Avoid a requester here for a volume that doesn't exist. */
5762 proc->pr_WindowPtr = (APTR)-1L;
5763#endif
5764
5765 if (first == TRUE)
5766 {
5767 /* copy file name into NameBuff, expanding environment variables */
5768 save_char = ptr[len];
5769 ptr[len] = NUL;
5770 expand_env(ptr, NameBuff, MAXPATHL);
5771 ptr[len] = save_char;
5772
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005773 vim_free(ff_file_to_find);
5774 ff_file_to_find = vim_strsave(NameBuff);
5775 if (ff_file_to_find == NULL) /* out of memory */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005776 {
5777 file_name = NULL;
5778 goto theend;
5779 }
5780 }
5781
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005782 rel_to_curdir = (ff_file_to_find[0] == '.'
5783 && (ff_file_to_find[1] == NUL
5784 || vim_ispathsep(ff_file_to_find[1])
5785 || (ff_file_to_find[1] == '.'
5786 && (ff_file_to_find[2] == NUL
5787 || vim_ispathsep(ff_file_to_find[2])))));
5788 if (vim_isAbsName(ff_file_to_find)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005789 /* "..", "../path", "." and "./path": don't use the path_option */
5790 || rel_to_curdir
5791#if defined(MSWIN) || defined(MSDOS) || defined(OS2)
5792 /* handle "\tmp" as absolute path */
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005793 || vim_ispathsep(ff_file_to_find[0])
Bram Moolenaar21160b92009-11-11 15:56:10 +00005794 /* handle "c:name" as absolute path */
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005795 || (ff_file_to_find[0] != NUL && ff_file_to_find[1] == ':')
Bram Moolenaar071d4272004-06-13 20:20:40 +00005796#endif
5797#ifdef AMIGA
5798 /* handle ":tmp" as absolute path */
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005799 || ff_file_to_find[0] == ':'
Bram Moolenaar071d4272004-06-13 20:20:40 +00005800#endif
5801 )
5802 {
5803 /*
5804 * Absolute path, no need to use "path_option".
5805 * If this is not a first call, return NULL. We already returned a
5806 * filename on the first call.
5807 */
5808 if (first == TRUE)
5809 {
5810 int l;
5811 int run;
5812
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005813 if (path_with_url(ff_file_to_find))
Bram Moolenaar071d4272004-06-13 20:20:40 +00005814 {
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005815 file_name = vim_strsave(ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005816 goto theend;
5817 }
5818
5819 /* When FNAME_REL flag given first use the directory of the file.
5820 * Otherwise or when this fails use the current directory. */
5821 for (run = 1; run <= 2; ++run)
5822 {
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005823 l = (int)STRLEN(ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005824 if (run == 1
5825 && rel_to_curdir
5826 && (options & FNAME_REL)
5827 && rel_fname != NULL
5828 && STRLEN(rel_fname) + l < MAXPATHL)
5829 {
5830 STRCPY(NameBuff, rel_fname);
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005831 STRCPY(gettail(NameBuff), ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005832 l = (int)STRLEN(NameBuff);
5833 }
5834 else
5835 {
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005836 STRCPY(NameBuff, ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005837 run = 2;
5838 }
5839
5840 /* When the file doesn't exist, try adding parts of
5841 * 'suffixesadd'. */
Bram Moolenaar433f7c82006-03-21 21:29:36 +00005842 buf = suffixes;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005843 for (;;)
5844 {
5845 if (
5846#ifdef DJGPP
5847 /* "C:" by itself will fail for mch_getperm(),
5848 * assume it's always valid. */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005849 (find_what != FINDFILE_FILE && NameBuff[0] != NUL
Bram Moolenaar071d4272004-06-13 20:20:40 +00005850 && NameBuff[1] == ':'
5851 && NameBuff[2] == NUL) ||
5852#endif
5853 (mch_getperm(NameBuff) >= 0
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005854 && (find_what == FINDFILE_BOTH
5855 || ((find_what == FINDFILE_DIR)
5856 == mch_isdir(NameBuff)))))
Bram Moolenaar071d4272004-06-13 20:20:40 +00005857 {
5858 file_name = vim_strsave(NameBuff);
5859 goto theend;
5860 }
5861 if (*buf == NUL)
5862 break;
5863 copy_option_part(&buf, NameBuff + l, MAXPATHL - l, ",");
5864 }
5865 }
5866 }
5867 }
5868 else
5869 {
5870 /*
5871 * Loop over all paths in the 'path' or 'cdpath' option.
5872 * When "first" is set, first setup to the start of the option.
5873 * Otherwise continue to find the next match.
5874 */
5875 if (first == TRUE)
5876 {
5877 /* vim_findfile_free_visited can handle a possible NULL pointer */
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005878 vim_findfile_free_visited(fdip_search_ctx);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005879 dir = path_option;
5880 did_findfile_init = FALSE;
5881 }
5882
5883 for (;;)
5884 {
5885 if (did_findfile_init)
5886 {
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005887 file_name = vim_findfile(fdip_search_ctx);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005888 if (file_name != NULL)
5889 break;
5890
5891 did_findfile_init = FALSE;
5892 }
5893 else
5894 {
5895 char_u *r_ptr;
5896
5897 if (dir == NULL || *dir == NUL)
5898 {
5899 /* We searched all paths of the option, now we can
5900 * free the search context. */
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005901 vim_findfile_cleanup(fdip_search_ctx);
5902 fdip_search_ctx = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005903 break;
5904 }
5905
5906 if ((buf = alloc((int)(MAXPATHL))) == NULL)
5907 break;
5908
5909 /* copy next path */
5910 buf[0] = 0;
5911 copy_option_part(&dir, buf, MAXPATHL, " ,");
5912
5913#ifdef FEAT_PATH_EXTRA
5914 /* get the stopdir string */
5915 r_ptr = vim_findfile_stopdir(buf);
5916#else
5917 r_ptr = NULL;
5918#endif
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005919 fdip_search_ctx = vim_findfile_init(buf, ff_file_to_find,
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005920 r_ptr, 100, FALSE, find_what,
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005921 fdip_search_ctx, FALSE, rel_fname);
5922 if (fdip_search_ctx != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005923 did_findfile_init = TRUE;
5924 vim_free(buf);
5925 }
5926 }
5927 }
5928 if (file_name == NULL && (options & FNAME_MESS))
5929 {
5930 if (first == TRUE)
5931 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005932 if (find_what == FINDFILE_DIR)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005933 EMSG2(_("E344: Can't find directory \"%s\" in cdpath"),
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005934 ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005935 else
5936 EMSG2(_("E345: Can't find file \"%s\" in path"),
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005937 ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005938 }
5939 else
5940 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005941 if (find_what == FINDFILE_DIR)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005942 EMSG2(_("E346: No more directory \"%s\" found in cdpath"),
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005943 ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005944 else
5945 EMSG2(_("E347: No more file \"%s\" found in path"),
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005946 ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005947 }
5948 }
5949
5950theend:
5951#ifdef AMIGA
5952 proc->pr_WindowPtr = save_winptr;
5953#endif
5954 return file_name;
5955}
5956
5957#endif /* FEAT_SEARCHPATH */
5958
5959/*
5960 * Change directory to "new_dir". If FEAT_SEARCHPATH is defined, search
5961 * 'cdpath' for relative directory names, otherwise just mch_chdir().
5962 */
5963 int
5964vim_chdir(new_dir)
5965 char_u *new_dir;
5966{
5967#ifndef FEAT_SEARCHPATH
5968 return mch_chdir((char *)new_dir);
5969#else
5970 char_u *dir_name;
5971 int r;
5972
5973 dir_name = find_directory_in_path(new_dir, (int)STRLEN(new_dir),
5974 FNAME_MESS, curbuf->b_ffname);
5975 if (dir_name == NULL)
5976 return -1;
5977 r = mch_chdir((char *)dir_name);
5978 vim_free(dir_name);
5979 return r;
5980#endif
5981}
5982
5983/*
Bram Moolenaarbbebc852005-07-18 21:47:53 +00005984 * Get user name from machine-specific function.
Bram Moolenaar071d4272004-06-13 20:20:40 +00005985 * Returns the user name in "buf[len]".
Bram Moolenaarbbebc852005-07-18 21:47:53 +00005986 * Some systems are quite slow in obtaining the user name (Windows NT), thus
5987 * cache the result.
Bram Moolenaar071d4272004-06-13 20:20:40 +00005988 * Returns OK or FAIL.
5989 */
5990 int
5991get_user_name(buf, len)
5992 char_u *buf;
5993 int len;
5994{
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005995 if (username == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005996 {
5997 if (mch_get_user_name(buf, len) == FAIL)
5998 return FAIL;
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005999 username = vim_strsave(buf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00006000 }
6001 else
Bram Moolenaarbbebc852005-07-18 21:47:53 +00006002 vim_strncpy(buf, username, len - 1);
Bram Moolenaar071d4272004-06-13 20:20:40 +00006003 return OK;
6004}
6005
6006#ifndef HAVE_QSORT
6007/*
6008 * Our own qsort(), for systems that don't have it.
6009 * It's simple and slow. From the K&R C book.
6010 */
6011 void
6012qsort(base, elm_count, elm_size, cmp)
6013 void *base;
6014 size_t elm_count;
6015 size_t elm_size;
6016 int (*cmp) __ARGS((const void *, const void *));
6017{
6018 char_u *buf;
6019 char_u *p1;
6020 char_u *p2;
6021 int i, j;
6022 int gap;
6023
6024 buf = alloc((unsigned)elm_size);
6025 if (buf == NULL)
6026 return;
6027
6028 for (gap = elm_count / 2; gap > 0; gap /= 2)
6029 for (i = gap; i < elm_count; ++i)
6030 for (j = i - gap; j >= 0; j -= gap)
6031 {
6032 /* Compare the elements. */
6033 p1 = (char_u *)base + j * elm_size;
6034 p2 = (char_u *)base + (j + gap) * elm_size;
6035 if ((*cmp)((void *)p1, (void *)p2) <= 0)
6036 break;
Bram Moolenaar21160b92009-11-11 15:56:10 +00006037 /* Exchange the elements. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00006038 mch_memmove(buf, p1, elm_size);
6039 mch_memmove(p1, p2, elm_size);
6040 mch_memmove(p2, buf, elm_size);
6041 }
6042
6043 vim_free(buf);
6044}
6045#endif
6046
Bram Moolenaar071d4272004-06-13 20:20:40 +00006047/*
6048 * Sort an array of strings.
6049 */
6050static int
6051#ifdef __BORLANDC__
6052_RTLENTRYF
6053#endif
6054sort_compare __ARGS((const void *s1, const void *s2));
6055
6056 static int
6057#ifdef __BORLANDC__
6058_RTLENTRYF
6059#endif
6060sort_compare(s1, s2)
6061 const void *s1;
6062 const void *s2;
6063{
6064 return STRCMP(*(char **)s1, *(char **)s2);
6065}
6066
6067 void
6068sort_strings(files, count)
6069 char_u **files;
6070 int count;
6071{
6072 qsort((void *)files, (size_t)count, sizeof(char_u *), sort_compare);
6073}
Bram Moolenaar071d4272004-06-13 20:20:40 +00006074
6075#if !defined(NO_EXPANDPATH) || defined(PROTO)
6076/*
6077 * Compare path "p[]" to "q[]".
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00006078 * If "maxlen" >= 0 compare "p[maxlen]" to "q[maxlen]"
Bram Moolenaar071d4272004-06-13 20:20:40 +00006079 * Return value like strcmp(p, q), but consider path separators.
6080 */
6081 int
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00006082pathcmp(p, q, maxlen)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006083 const char *p, *q;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00006084 int maxlen;
Bram Moolenaar071d4272004-06-13 20:20:40 +00006085{
6086 int i;
Bram Moolenaar86b68352004-12-27 21:59:20 +00006087 const char *s = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00006088
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00006089 for (i = 0; maxlen < 0 || i < maxlen; ++i)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006090 {
6091 /* End of "p": check if "q" also ends or just has a slash. */
6092 if (p[i] == NUL)
6093 {
6094 if (q[i] == NUL) /* full match */
6095 return 0;
6096 s = q;
6097 break;
6098 }
6099
6100 /* End of "q": check if "p" just has a slash. */
6101 if (q[i] == NUL)
6102 {
6103 s = p;
6104 break;
6105 }
6106
6107 if (
6108#ifdef CASE_INSENSITIVE_FILENAME
6109 TOUPPER_LOC(p[i]) != TOUPPER_LOC(q[i])
6110#else
6111 p[i] != q[i]
6112#endif
6113#ifdef BACKSLASH_IN_FILENAME
6114 /* consider '/' and '\\' to be equal */
6115 && !((p[i] == '/' && q[i] == '\\')
6116 || (p[i] == '\\' && q[i] == '/'))
6117#endif
6118 )
6119 {
6120 if (vim_ispathsep(p[i]))
6121 return -1;
6122 if (vim_ispathsep(q[i]))
6123 return 1;
6124 return ((char_u *)p)[i] - ((char_u *)q)[i]; /* no match */
6125 }
6126 }
Bram Moolenaar86b68352004-12-27 21:59:20 +00006127 if (s == NULL) /* "i" ran into "maxlen" */
6128 return 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00006129
6130 /* ignore a trailing slash, but not "//" or ":/" */
Bram Moolenaar86b68352004-12-27 21:59:20 +00006131 if (s[i + 1] == NUL
6132 && i > 0
6133 && !after_pathsep((char_u *)s, (char_u *)s + i)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006134#ifdef BACKSLASH_IN_FILENAME
Bram Moolenaar86b68352004-12-27 21:59:20 +00006135 && (s[i] == '/' || s[i] == '\\')
Bram Moolenaar071d4272004-06-13 20:20:40 +00006136#else
Bram Moolenaar86b68352004-12-27 21:59:20 +00006137 && s[i] == '/'
Bram Moolenaar071d4272004-06-13 20:20:40 +00006138#endif
Bram Moolenaar86b68352004-12-27 21:59:20 +00006139 )
Bram Moolenaar071d4272004-06-13 20:20:40 +00006140 return 0; /* match with trailing slash */
6141 if (s == q)
6142 return -1; /* no match */
6143 return 1;
6144}
6145#endif
6146
Bram Moolenaar071d4272004-06-13 20:20:40 +00006147/*
6148 * The putenv() implementation below comes from the "screen" program.
6149 * Included with permission from Juergen Weigert.
6150 * See pty.c for the copyright notice.
6151 */
6152
6153/*
6154 * putenv -- put value into environment
6155 *
6156 * Usage: i = putenv (string)
6157 * int i;
6158 * char *string;
6159 *
6160 * where string is of the form <name>=<value>.
6161 * Putenv returns 0 normally, -1 on error (not enough core for malloc).
6162 *
6163 * Putenv may need to add a new name into the environment, or to
6164 * associate a value longer than the current value with a particular
6165 * name. So, to make life simpler, putenv() copies your entire
6166 * environment into the heap (i.e. malloc()) from the stack
6167 * (i.e. where it resides when your process is initiated) the first
6168 * time you call it.
6169 *
6170 * (history removed, not very interesting. See the "screen" sources.)
6171 */
6172
6173#if !defined(HAVE_SETENV) && !defined(HAVE_PUTENV)
6174
6175#define EXTRASIZE 5 /* increment to add to env. size */
6176
6177static int envsize = -1; /* current size of environment */
6178#ifndef MACOS_CLASSIC
6179extern
6180#endif
6181 char **environ; /* the global which is your env. */
6182
6183static int findenv __ARGS((char *name)); /* look for a name in the env. */
6184static int newenv __ARGS((void)); /* copy env. from stack to heap */
6185static int moreenv __ARGS((void)); /* incr. size of env. */
6186
6187 int
6188putenv(string)
6189 const char *string;
6190{
6191 int i;
6192 char *p;
6193
6194 if (envsize < 0)
6195 { /* first time putenv called */
6196 if (newenv() < 0) /* copy env. to heap */
6197 return -1;
6198 }
6199
6200 i = findenv((char *)string); /* look for name in environment */
6201
6202 if (i < 0)
6203 { /* name must be added */
6204 for (i = 0; environ[i]; i++);
6205 if (i >= (envsize - 1))
6206 { /* need new slot */
6207 if (moreenv() < 0)
6208 return -1;
6209 }
6210 p = (char *)alloc((unsigned)(strlen(string) + 1));
6211 if (p == NULL) /* not enough core */
6212 return -1;
6213 environ[i + 1] = 0; /* new end of env. */
6214 }
6215 else
6216 { /* name already in env. */
6217 p = vim_realloc(environ[i], strlen(string) + 1);
6218 if (p == NULL)
6219 return -1;
6220 }
6221 sprintf(p, "%s", string); /* copy into env. */
6222 environ[i] = p;
6223
6224 return 0;
6225}
6226
6227 static int
6228findenv(name)
6229 char *name;
6230{
6231 char *namechar, *envchar;
6232 int i, found;
6233
6234 found = 0;
6235 for (i = 0; environ[i] && !found; i++)
6236 {
6237 envchar = environ[i];
6238 namechar = name;
6239 while (*namechar && *namechar != '=' && (*namechar == *envchar))
6240 {
6241 namechar++;
6242 envchar++;
6243 }
6244 found = ((*namechar == '\0' || *namechar == '=') && *envchar == '=');
6245 }
6246 return found ? i - 1 : -1;
6247}
6248
6249 static int
6250newenv()
6251{
6252 char **env, *elem;
6253 int i, esize;
6254
6255#ifdef MACOS
6256 /* for Mac a new, empty environment is created */
6257 i = 0;
6258#else
6259 for (i = 0; environ[i]; i++)
6260 ;
6261#endif
6262 esize = i + EXTRASIZE + 1;
6263 env = (char **)alloc((unsigned)(esize * sizeof (elem)));
6264 if (env == NULL)
6265 return -1;
6266
6267#ifndef MACOS
6268 for (i = 0; environ[i]; i++)
6269 {
6270 elem = (char *)alloc((unsigned)(strlen(environ[i]) + 1));
6271 if (elem == NULL)
6272 return -1;
6273 env[i] = elem;
6274 strcpy(elem, environ[i]);
6275 }
6276#endif
6277
6278 env[i] = 0;
6279 environ = env;
6280 envsize = esize;
6281 return 0;
6282}
6283
6284 static int
6285moreenv()
6286{
6287 int esize;
6288 char **env;
6289
6290 esize = envsize + EXTRASIZE;
6291 env = (char **)vim_realloc((char *)environ, esize * sizeof (*env));
6292 if (env == 0)
6293 return -1;
6294 environ = env;
6295 envsize = esize;
6296 return 0;
6297}
6298
6299# ifdef USE_VIMPTY_GETENV
6300 char_u *
6301vimpty_getenv(string)
6302 const char_u *string;
6303{
6304 int i;
6305 char_u *p;
6306
6307 if (envsize < 0)
6308 return NULL;
6309
6310 i = findenv((char *)string);
6311
6312 if (i < 0)
6313 return NULL;
6314
6315 p = vim_strchr((char_u *)environ[i], '=');
6316 return (p + 1);
6317}
6318# endif
6319
6320#endif /* !defined(HAVE_SETENV) && !defined(HAVE_PUTENV) */
Bram Moolenaarc4a06d32005-06-07 21:04:49 +00006321
Bram Moolenaarc4956c82006-03-12 21:58:43 +00006322#if defined(FEAT_EVAL) || defined(FEAT_SPELL) || defined(PROTO)
Bram Moolenaarc4a06d32005-06-07 21:04:49 +00006323/*
6324 * Return 0 for not writable, 1 for writable file, 2 for a dir which we have
6325 * rights to write into.
6326 */
6327 int
6328filewritable(fname)
6329 char_u *fname;
6330{
6331 int retval = 0;
6332#if defined(UNIX) || defined(VMS)
6333 int perm = 0;
6334#endif
6335
6336#if defined(UNIX) || defined(VMS)
6337 perm = mch_getperm(fname);
6338#endif
6339#ifndef MACOS_CLASSIC /* TODO: get either mch_writable or mch_access */
6340 if (
6341# ifdef WIN3264
6342 mch_writable(fname) &&
6343# else
6344# if defined(UNIX) || defined(VMS)
6345 (perm & 0222) &&
6346# endif
6347# endif
6348 mch_access((char *)fname, W_OK) == 0
6349 )
6350#endif
6351 {
6352 ++retval;
6353 if (mch_isdir(fname))
6354 ++retval;
6355 }
6356 return retval;
6357}
6358#endif
Bram Moolenaar6bab4d12005-06-16 21:53:56 +00006359
6360/*
6361 * Print an error message with one or two "%s" and one or two string arguments.
6362 * This is not in message.c to avoid a warning for prototypes.
6363 */
6364 int
6365emsg3(s, a1, a2)
6366 char_u *s, *a1, *a2;
6367{
Bram Moolenaareb3593b2006-04-22 22:33:57 +00006368 if (emsg_not_now())
Bram Moolenaar6bab4d12005-06-16 21:53:56 +00006369 return TRUE; /* no error messages at the moment */
Bram Moolenaar6f192452007-11-08 19:49:02 +00006370#ifdef HAVE_STDARG_H
6371 vim_snprintf((char *)IObuff, IOSIZE, (char *)s, a1, a2);
6372#else
Bram Moolenaareb3593b2006-04-22 22:33:57 +00006373 vim_snprintf((char *)IObuff, IOSIZE, (char *)s, (long_u)a1, (long_u)a2);
Bram Moolenaar6f192452007-11-08 19:49:02 +00006374#endif
Bram Moolenaar6bab4d12005-06-16 21:53:56 +00006375 return emsg(IObuff);
6376}
6377
6378/*
6379 * Print an error message with one "%ld" and one long int argument.
6380 * This is not in message.c to avoid a warning for prototypes.
6381 */
6382 int
6383emsgn(s, n)
6384 char_u *s;
6385 long n;
6386{
Bram Moolenaareb3593b2006-04-22 22:33:57 +00006387 if (emsg_not_now())
Bram Moolenaar6bab4d12005-06-16 21:53:56 +00006388 return TRUE; /* no error messages at the moment */
6389 vim_snprintf((char *)IObuff, IOSIZE, (char *)s, n);
6390 return emsg(IObuff);
6391}
Bram Moolenaarcdf04202010-05-29 15:11:47 +02006392
6393#if defined(FEAT_SPELL) || defined(FEAT_PERSISTENT_UNDO) || defined(PROTO)
6394/*
6395 * Read 2 bytes from "fd" and turn them into an int, MSB first.
6396 */
6397 int
6398get2c(fd)
6399 FILE *fd;
6400{
Bram Moolenaar9db58062010-05-29 20:33:07 +02006401 int n;
Bram Moolenaarcdf04202010-05-29 15:11:47 +02006402
6403 n = getc(fd);
6404 n = (n << 8) + getc(fd);
6405 return n;
6406}
6407
6408/*
6409 * Read 3 bytes from "fd" and turn them into an int, MSB first.
6410 */
6411 int
6412get3c(fd)
6413 FILE *fd;
6414{
Bram Moolenaar9db58062010-05-29 20:33:07 +02006415 int n;
Bram Moolenaarcdf04202010-05-29 15:11:47 +02006416
6417 n = getc(fd);
6418 n = (n << 8) + getc(fd);
6419 n = (n << 8) + getc(fd);
6420 return n;
6421}
6422
6423/*
6424 * Read 4 bytes from "fd" and turn them into an int, MSB first.
6425 */
6426 int
6427get4c(fd)
6428 FILE *fd;
6429{
Bram Moolenaar9db58062010-05-29 20:33:07 +02006430 int n;
Bram Moolenaarcdf04202010-05-29 15:11:47 +02006431
6432 n = getc(fd);
6433 n = (n << 8) + getc(fd);
6434 n = (n << 8) + getc(fd);
6435 n = (n << 8) + getc(fd);
6436 return n;
6437}
6438
6439/*
6440 * Read 8 bytes from "fd" and turn them into a time_t, MSB first.
6441 */
6442 time_t
6443get8ctime(fd)
6444 FILE *fd;
6445{
6446 time_t n = 0;
6447 int i;
6448
6449 for (i = 0; i < 8; ++i)
6450 n = (n << 8) + getc(fd);
6451 return n;
6452}
6453
6454/*
6455 * Read a string of length "cnt" from "fd" into allocated memory.
6456 * Returns NULL when out of memory or unable to read that many bytes.
6457 */
6458 char_u *
6459read_string(fd, cnt)
6460 FILE *fd;
6461 int cnt;
6462{
6463 char_u *str;
6464 int i;
6465 int c;
6466
6467 /* allocate memory */
6468 str = alloc((unsigned)cnt + 1);
6469 if (str != NULL)
6470 {
6471 /* Read the string. Quit when running into the EOF. */
6472 for (i = 0; i < cnt; ++i)
6473 {
6474 c = getc(fd);
6475 if (c == EOF)
6476 {
6477 vim_free(str);
6478 return NULL;
6479 }
6480 str[i] = c;
6481 }
6482 str[i] = NUL;
6483 }
6484 return str;
6485}
6486
6487/*
6488 * Write a number to file "fd", MSB first, in "len" bytes.
6489 */
6490 int
6491put_bytes(fd, nr, len)
6492 FILE *fd;
6493 long_u nr;
6494 int len;
6495{
6496 int i;
6497
6498 for (i = len - 1; i >= 0; --i)
6499 if (putc((int)(nr >> (i * 8)), fd) == EOF)
6500 return FAIL;
6501 return OK;
6502}
6503
6504#ifdef _MSC_VER
6505# if (_MSC_VER <= 1200)
6506/* This line is required for VC6 without the service pack. Also see the
6507 * matching #pragma below. */
6508 # pragma optimize("", off)
6509# endif
6510#endif
6511
6512/*
6513 * Write time_t to file "fd" in 8 bytes.
6514 */
6515 void
6516put_time(fd, the_time)
6517 FILE *fd;
6518 time_t the_time;
6519{
6520 int c;
6521 int i;
6522 time_t wtime = the_time;
6523
6524 /* time_t can be up to 8 bytes in size, more than long_u, thus we
6525 * can't use put_bytes() here.
6526 * Another problem is that ">>" may do an arithmetic shift that keeps the
Bram Moolenaar644fdff2010-05-30 13:26:21 +02006527 * sign. This happens for large values of wtime. A cast to long_u may
6528 * truncate if time_t is 8 bytes. So only use a cast when it is 4 bytes,
6529 * it's safe to assume that long_u is 4 bytes or more and when using 8
6530 * bytes the top bit won't be set. */
Bram Moolenaarcdf04202010-05-29 15:11:47 +02006531 for (i = 7; i >= 0; --i)
6532 {
6533 if (i + 1 > (int)sizeof(time_t))
6534 /* ">>" doesn't work well when shifting more bits than avail */
6535 putc(0, fd);
6536 else
6537 {
Bram Moolenaar644fdff2010-05-30 13:26:21 +02006538#if defined(SIZEOF_TIME_T) && SIZEOF_TIME_T > 4
Bram Moolenaarbbd6afe2010-06-02 20:32:23 +02006539 c = (int)(wtime >> (i * 8));
Bram Moolenaar644fdff2010-05-30 13:26:21 +02006540#else
Bram Moolenaarbbd6afe2010-06-02 20:32:23 +02006541 c = (int)((long_u)wtime >> (i * 8));
Bram Moolenaar644fdff2010-05-30 13:26:21 +02006542#endif
Bram Moolenaarcdf04202010-05-29 15:11:47 +02006543 putc(c, fd);
6544 }
6545 }
6546}
6547
6548#ifdef _MSC_VER
6549# if (_MSC_VER <= 1200)
6550 # pragma optimize("", on)
6551# endif
6552#endif
6553
6554#endif
Bram Moolenaar10b7b392012-01-10 16:28:45 +01006555
6556#if (defined(FEAT_MBYTE) && defined(FEAT_QUICKFIX)) \
6557 || defined(FEAT_SPELL) || defined(PROTO)
6558/*
6559 * Return TRUE if string "s" contains a non-ASCII character (128 or higher).
6560 * When "s" is NULL FALSE is returned.
6561 */
6562 int
6563has_non_ascii(s)
6564 char_u *s;
6565{
6566 char_u *p;
6567
6568 if (s != NULL)
6569 for (p = s; *p != NUL; ++p)
6570 if (*p >= 128)
6571 return TRUE;
6572 return FALSE;
6573}
6574#endif