blob: 9479a53c20d8f85dd663da953a02a615edd89fae [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;
1015 EMSGN(_("E342: Out of memory! (allocating %lu bytes)"), size);
1016 did_outofmem_msg = TRUE;
1017 }
1018}
1019
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001020#if defined(EXITFREE) || defined(PROTO)
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001021
1022# if defined(FEAT_SEARCHPATH)
1023static void free_findfile __ARGS((void));
1024# endif
1025
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001026/*
1027 * Free everything that we allocated.
1028 * Can be used to detect memory leaks, e.g., with ccmalloc.
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001029 * NOTE: This is tricky! Things are freed that functions depend on. Don't be
1030 * surprised if Vim crashes...
1031 * Some things can't be freed, esp. things local to a library function.
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001032 */
1033 void
1034free_all_mem()
1035{
1036 buf_T *buf, *nextbuf;
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001037 static int entered = FALSE;
1038
1039 /* When we cause a crash here it is caught and Vim tries to exit cleanly.
1040 * Don't try freeing everything again. */
1041 if (entered)
1042 return;
1043 entered = TRUE;
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001044
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001045# ifdef FEAT_AUTOCMD
Bram Moolenaar78ab3312007-09-29 12:16:41 +00001046 block_autocmds(); /* don't want to trigger autocommands here */
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001047# endif
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001048
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001049# ifdef FEAT_WINDOWS
Bram Moolenaar5bedfc62010-07-20 22:30:01 +02001050 /* Close all tabs and windows. Reset 'equalalways' to avoid redraws. */
1051 p_ea = FALSE;
Bram Moolenaar7d47b6e2006-03-15 22:59:18 +00001052 if (first_tabpage->tp_next != NULL)
1053 do_cmdline_cmd((char_u *)"tabonly!");
1054 if (firstwin != lastwin)
1055 do_cmdline_cmd((char_u *)"only!");
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001056# endif
Bram Moolenaar997fb4b2006-02-17 21:53:23 +00001057
Bram Moolenaarc4956c82006-03-12 21:58:43 +00001058# if defined(FEAT_SPELL)
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001059 /* Free all spell info. */
1060 spell_free_all();
1061# endif
1062
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001063# if defined(FEAT_USR_CMDS)
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001064 /* Clear user commands (before deleting buffers). */
1065 ex_comclear(NULL);
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001066# endif
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001067
1068# ifdef FEAT_MENU
1069 /* Clear menus. */
1070 do_cmdline_cmd((char_u *)"aunmenu *");
Bram Moolenaar21160b92009-11-11 15:56:10 +00001071# ifdef FEAT_MULTI_LANG
1072 do_cmdline_cmd((char_u *)"menutranslate clear");
1073# endif
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001074# endif
1075
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001076 /* Clear mappings, abbreviations, breakpoints. */
Bram Moolenaar21160b92009-11-11 15:56:10 +00001077 do_cmdline_cmd((char_u *)"lmapclear");
1078 do_cmdline_cmd((char_u *)"xmapclear");
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001079 do_cmdline_cmd((char_u *)"mapclear");
1080 do_cmdline_cmd((char_u *)"mapclear!");
1081 do_cmdline_cmd((char_u *)"abclear");
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001082# if defined(FEAT_EVAL)
1083 do_cmdline_cmd((char_u *)"breakdel *");
1084# endif
Bram Moolenaar1e498f52005-06-26 22:29:44 +00001085# if defined(FEAT_PROFILE)
1086 do_cmdline_cmd((char_u *)"profdel *");
1087# endif
Bram Moolenaarde0dfed2009-02-24 03:30:14 +00001088# if defined(FEAT_KEYMAP)
1089 do_cmdline_cmd((char_u *)"set keymap=");
1090#endif
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001091
1092# ifdef FEAT_TITLE
1093 free_titles();
1094# endif
1095# if defined(FEAT_SEARCHPATH)
1096 free_findfile();
1097# endif
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001098
1099 /* Obviously named calls. */
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001100# if defined(FEAT_AUTOCMD)
1101 free_all_autocmds();
1102# endif
1103 clear_termcodes();
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001104 free_all_options();
1105 free_all_marks();
1106 alist_clear(&global_alist);
1107 free_homedir();
1108 free_search_patterns();
1109 free_old_sub();
1110 free_last_insert();
1111 free_prev_shellcmd();
1112 free_regexp_stuff();
1113 free_tag_stuff();
1114 free_cd_dir();
Bram Moolenaarde0dfed2009-02-24 03:30:14 +00001115# ifdef FEAT_SIGNS
1116 free_signs();
1117# endif
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001118# ifdef FEAT_EVAL
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001119 set_expr_line(NULL);
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001120# endif
1121# ifdef FEAT_DIFF
Bram Moolenaar997fb4b2006-02-17 21:53:23 +00001122 diff_clear(curtab);
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001123# endif
Bram Moolenaara40ceaf2006-01-13 22:35:40 +00001124 clear_sb_text(); /* free any scrollback text */
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001125
1126 /* Free some global vars. */
1127 vim_free(username);
Bram Moolenaaraf92ee82007-10-07 13:45:08 +00001128# ifdef FEAT_CLIPBOARD
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001129 vim_free(clip_exclude_prog);
Bram Moolenaaraf92ee82007-10-07 13:45:08 +00001130# endif
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001131 vim_free(last_cmdline);
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001132# ifdef FEAT_CMDHIST
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001133 vim_free(new_last_cmdline);
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001134# endif
Bram Moolenaar238a5642006-02-21 22:12:05 +00001135 set_keep_msg(NULL, 0);
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001136 vim_free(ff_expand_buffer);
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001137
1138 /* Clear cmdline history. */
1139 p_hi = 0;
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001140# ifdef FEAT_CMDHIST
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001141 init_history();
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001142# endif
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001143
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001144#ifdef FEAT_QUICKFIX
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001145 {
Bram Moolenaarbd1e5d22009-04-29 09:02:44 +00001146 win_T *win;
1147 tabpage_T *tab;
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001148
1149 qf_free_all(NULL);
1150 /* Free all location lists */
Bram Moolenaarbd1e5d22009-04-29 09:02:44 +00001151 FOR_ALL_TAB_WINDOWS(tab, win)
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001152 qf_free_all(win);
1153 }
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001154#endif
1155
1156 /* Close all script inputs. */
1157 close_all_scripts();
1158
1159#if defined(FEAT_WINDOWS)
1160 /* Destroy all windows. Must come before freeing buffers. */
1161 win_free_all();
1162#endif
1163
Bram Moolenaar2a329742008-04-01 12:53:43 +00001164 /* Free all buffers. Reset 'autochdir' to avoid accessing things that
1165 * were freed already. */
1166#ifdef FEAT_AUTOCHDIR
1167 p_acd = FALSE;
1168#endif
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001169 for (buf = firstbuf; buf != NULL; )
1170 {
1171 nextbuf = buf->b_next;
1172 close_buffer(NULL, buf, DOBUF_WIPE);
1173 if (buf_valid(buf))
1174 buf = nextbuf; /* didn't work, try next one */
1175 else
1176 buf = firstbuf;
1177 }
1178
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001179#ifdef FEAT_ARABIC
1180 free_cmdline_buf();
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001181#endif
1182
1183 /* Clear registers. */
1184 clear_registers();
1185 ResetRedobuff();
1186 ResetRedobuff();
1187
Bram Moolenaar85c79d32007-02-20 01:59:20 +00001188#if defined(FEAT_CLIENTSERVER) && defined(FEAT_X11)
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001189 vim_free(serverDelayedStartName);
1190#endif
1191
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001192 /* highlight info */
1193 free_highlight();
1194
Bram Moolenaar1e498f52005-06-26 22:29:44 +00001195 reset_last_sourcing();
1196
Bram Moolenaar7d47b6e2006-03-15 22:59:18 +00001197#ifdef FEAT_WINDOWS
Bram Moolenaar06a89a52006-04-29 22:01:03 +00001198 free_tabpage(first_tabpage);
1199 first_tabpage = NULL;
Bram Moolenaar7d47b6e2006-03-15 22:59:18 +00001200#endif
1201
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001202# ifdef UNIX
1203 /* Machine-specific free. */
1204 mch_free_mem();
1205# endif
1206
1207 /* message history */
1208 for (;;)
1209 if (delete_first_msg() == FAIL)
1210 break;
1211
1212# ifdef FEAT_EVAL
1213 eval_clear();
1214# endif
1215
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001216 free_termoptions();
1217
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001218 /* screenlines (can't display anything now!) */
1219 free_screenlines();
1220
1221#if defined(USE_XSMP)
1222 xsmp_close();
1223#endif
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001224#ifdef FEAT_GUI_GTK
1225 gui_mch_free_all();
1226#endif
1227 clear_hl_tables();
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001228
1229 vim_free(IObuff);
1230 vim_free(NameBuff);
1231}
1232#endif
1233
Bram Moolenaar071d4272004-06-13 20:20:40 +00001234/*
Bram Moolenaare980d8a2010-12-08 13:11:21 +01001235 * Copy "string" into newly allocated memory.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001236 */
1237 char_u *
1238vim_strsave(string)
1239 char_u *string;
1240{
1241 char_u *p;
1242 unsigned len;
1243
1244 len = (unsigned)STRLEN(string) + 1;
1245 p = alloc(len);
1246 if (p != NULL)
1247 mch_memmove(p, string, (size_t)len);
1248 return p;
1249}
1250
Bram Moolenaare980d8a2010-12-08 13:11:21 +01001251/*
1252 * Copy up to "len" bytes of "string" into newly allocated memory and
1253 * terminate with a NUL.
1254 * The allocated memory always has size "len + 1", also when "string" is
1255 * shorter.
1256 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001257 char_u *
1258vim_strnsave(string, len)
1259 char_u *string;
1260 int len;
1261{
1262 char_u *p;
1263
1264 p = alloc((unsigned)(len + 1));
1265 if (p != NULL)
1266 {
1267 STRNCPY(p, string, len);
1268 p[len] = NUL;
1269 }
1270 return p;
1271}
1272
Bram Moolenaar071d4272004-06-13 20:20:40 +00001273/*
1274 * Same as vim_strsave(), but any characters found in esc_chars are preceded
1275 * by a backslash.
1276 */
1277 char_u *
1278vim_strsave_escaped(string, esc_chars)
1279 char_u *string;
1280 char_u *esc_chars;
1281{
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001282 return vim_strsave_escaped_ext(string, esc_chars, '\\', FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001283}
1284
1285/*
1286 * Same as vim_strsave_escaped(), but when "bsl" is TRUE also escape
1287 * characters where rem_backslash() would remove the backslash.
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001288 * Escape the characters with "cc".
Bram Moolenaar071d4272004-06-13 20:20:40 +00001289 */
1290 char_u *
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001291vim_strsave_escaped_ext(string, esc_chars, cc, bsl)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001292 char_u *string;
1293 char_u *esc_chars;
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001294 int cc;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001295 int bsl;
1296{
1297 char_u *p;
1298 char_u *p2;
1299 char_u *escaped_string;
1300 unsigned length;
1301#ifdef FEAT_MBYTE
1302 int l;
1303#endif
1304
1305 /*
1306 * First count the number of backslashes required.
1307 * Then allocate the memory and insert them.
1308 */
1309 length = 1; /* count the trailing NUL */
1310 for (p = string; *p; p++)
1311 {
1312#ifdef FEAT_MBYTE
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001313 if (has_mbyte && (l = (*mb_ptr2len)(p)) > 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001314 {
1315 length += l; /* count a multibyte char */
1316 p += l - 1;
1317 continue;
1318 }
1319#endif
1320 if (vim_strchr(esc_chars, *p) != NULL || (bsl && rem_backslash(p)))
1321 ++length; /* count a backslash */
1322 ++length; /* count an ordinary char */
1323 }
1324 escaped_string = alloc(length);
1325 if (escaped_string != NULL)
1326 {
1327 p2 = escaped_string;
1328 for (p = string; *p; p++)
1329 {
1330#ifdef FEAT_MBYTE
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001331 if (has_mbyte && (l = (*mb_ptr2len)(p)) > 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001332 {
1333 mch_memmove(p2, p, (size_t)l);
1334 p2 += l;
1335 p += l - 1; /* skip multibyte char */
1336 continue;
1337 }
1338#endif
1339 if (vim_strchr(esc_chars, *p) != NULL || (bsl && rem_backslash(p)))
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001340 *p2++ = cc;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001341 *p2++ = *p;
1342 }
1343 *p2 = NUL;
1344 }
1345 return escaped_string;
1346}
1347
Bram Moolenaar7693ec62008-07-24 18:29:37 +00001348/*
1349 * Return TRUE when 'shell' has "csh" in the tail.
1350 */
1351 int
1352csh_like_shell()
1353{
1354 return (strstr((char *)gettail(p_sh), "csh") != NULL);
1355}
Bram Moolenaar7693ec62008-07-24 18:29:37 +00001356
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001357/*
1358 * Escape "string" for use as a shell argument with system().
Bram Moolenaar21160b92009-11-11 15:56:10 +00001359 * This uses single quotes, except when we know we need to use double quotes
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001360 * (MS-DOS and MS-Windows without 'shellslash' set).
Bram Moolenaarffd66c42008-07-16 20:43:37 +00001361 * Escape a newline, depending on the 'shell' option.
1362 * When "do_special" is TRUE also replace "!", "%", "#" and things starting
1363 * with "<" like "<cfile>".
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001364 * Returns the result in allocated memory, NULL if we have run out.
1365 */
1366 char_u *
Bram Moolenaar05bb9532008-07-04 09:44:11 +00001367vim_strsave_shellescape(string, do_special)
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001368 char_u *string;
Bram Moolenaar05bb9532008-07-04 09:44:11 +00001369 int do_special;
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001370{
1371 unsigned length;
1372 char_u *p;
1373 char_u *d;
1374 char_u *escaped_string;
Bram Moolenaar05bb9532008-07-04 09:44:11 +00001375 int l;
Bram Moolenaarffd66c42008-07-16 20:43:37 +00001376 int csh_like;
1377
1378 /* Only csh and similar shells expand '!' within single quotes. For sh and
1379 * the like we must not put a backslash before it, it will be taken
1380 * literally. If do_special is set the '!' will be escaped twice.
1381 * Csh also needs to have "\n" escaped twice when do_special is set. */
Bram Moolenaar7693ec62008-07-24 18:29:37 +00001382 csh_like = csh_like_shell();
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001383
1384 /* First count the number of extra bytes required. */
Bram Moolenaar77f66d62007-02-20 02:16:18 +00001385 length = (unsigned)STRLEN(string) + 3; /* two quotes and a trailing NUL */
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001386 for (p = string; *p != NUL; mb_ptr_adv(p))
1387 {
1388# if defined(WIN32) || defined(WIN16) || defined(DOS)
1389 if (!p_ssl)
1390 {
1391 if (*p == '"')
1392 ++length; /* " -> "" */
1393 }
1394 else
1395# endif
1396 if (*p == '\'')
1397 length += 3; /* ' => '\'' */
Bram Moolenaarffd66c42008-07-16 20:43:37 +00001398 if (*p == '\n' || (*p == '!' && (csh_like || do_special)))
1399 {
1400 ++length; /* insert backslash */
1401 if (csh_like && do_special)
1402 ++length; /* insert backslash */
1403 }
Bram Moolenaar05bb9532008-07-04 09:44:11 +00001404 if (do_special && find_cmdline_var(p, &l) >= 0)
1405 {
1406 ++length; /* insert backslash */
1407 p += l - 1;
1408 }
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001409 }
1410
1411 /* Allocate memory for the result and fill it. */
1412 escaped_string = alloc(length);
1413 if (escaped_string != NULL)
1414 {
1415 d = escaped_string;
1416
1417 /* add opening quote */
1418# if defined(WIN32) || defined(WIN16) || defined(DOS)
1419 if (!p_ssl)
1420 *d++ = '"';
1421 else
1422# endif
1423 *d++ = '\'';
1424
1425 for (p = string; *p != NUL; )
1426 {
1427# if defined(WIN32) || defined(WIN16) || defined(DOS)
1428 if (!p_ssl)
1429 {
1430 if (*p == '"')
1431 {
1432 *d++ = '"';
1433 *d++ = '"';
1434 ++p;
1435 continue;
1436 }
1437 }
1438 else
1439# endif
1440 if (*p == '\'')
1441 {
Bram Moolenaar05bb9532008-07-04 09:44:11 +00001442 *d++ = '\'';
1443 *d++ = '\\';
1444 *d++ = '\'';
1445 *d++ = '\'';
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001446 ++p;
1447 continue;
1448 }
Bram Moolenaarffd66c42008-07-16 20:43:37 +00001449 if (*p == '\n' || (*p == '!' && (csh_like || do_special)))
1450 {
1451 *d++ = '\\';
1452 if (csh_like && do_special)
1453 *d++ = '\\';
1454 *d++ = *p++;
1455 continue;
1456 }
Bram Moolenaar05bb9532008-07-04 09:44:11 +00001457 if (do_special && find_cmdline_var(p, &l) >= 0)
1458 {
1459 *d++ = '\\'; /* insert backslash */
1460 while (--l >= 0) /* copy the var */
1461 *d++ = *p++;
Bram Moolenaar099d01d2009-11-25 16:14:45 +00001462 continue;
Bram Moolenaar05bb9532008-07-04 09:44:11 +00001463 }
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001464
1465 MB_COPY_CHAR(p, d);
1466 }
1467
1468 /* add terminating quote and finish with a NUL */
1469# if defined(WIN32) || defined(WIN16) || defined(DOS)
1470 if (!p_ssl)
1471 *d++ = '"';
1472 else
1473# endif
1474 *d++ = '\'';
1475 *d = NUL;
1476 }
1477
1478 return escaped_string;
1479}
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001480
Bram Moolenaar071d4272004-06-13 20:20:40 +00001481/*
1482 * Like vim_strsave(), but make all characters uppercase.
1483 * This uses ASCII lower-to-upper case translation, language independent.
1484 */
1485 char_u *
1486vim_strsave_up(string)
1487 char_u *string;
1488{
1489 char_u *p1;
1490
1491 p1 = vim_strsave(string);
1492 vim_strup(p1);
1493 return p1;
1494}
1495
1496/*
1497 * Like vim_strnsave(), but make all characters uppercase.
1498 * This uses ASCII lower-to-upper case translation, language independent.
1499 */
1500 char_u *
1501vim_strnsave_up(string, len)
1502 char_u *string;
1503 int len;
1504{
1505 char_u *p1;
1506
1507 p1 = vim_strnsave(string, len);
1508 vim_strup(p1);
1509 return p1;
1510}
1511
1512/*
1513 * ASCII lower-to-upper case translation, language independent.
1514 */
1515 void
1516vim_strup(p)
1517 char_u *p;
1518{
1519 char_u *p2;
1520 int c;
1521
1522 if (p != NULL)
1523 {
1524 p2 = p;
1525 while ((c = *p2) != NUL)
1526#ifdef EBCDIC
1527 *p2++ = isalpha(c) ? toupper(c) : c;
1528#else
1529 *p2++ = (c < 'a' || c > 'z') ? c : (c - 0x20);
1530#endif
1531 }
1532}
1533
Bram Moolenaarc4956c82006-03-12 21:58:43 +00001534#if defined(FEAT_EVAL) || defined(FEAT_SPELL) || defined(PROTO)
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001535/*
1536 * Make string "s" all upper-case and return it in allocated memory.
1537 * Handles multi-byte characters as well as possible.
1538 * Returns NULL when out of memory.
1539 */
1540 char_u *
1541strup_save(orig)
1542 char_u *orig;
1543{
1544 char_u *p;
1545 char_u *res;
1546
1547 res = p = vim_strsave(orig);
1548
1549 if (res != NULL)
1550 while (*p != NUL)
1551 {
1552# ifdef FEAT_MBYTE
1553 int l;
1554
1555 if (enc_utf8)
1556 {
1557 int c, uc;
1558 int nl;
1559 char_u *s;
1560
1561 c = utf_ptr2char(p);
1562 uc = utf_toupper(c);
1563
1564 /* Reallocate string when byte count changes. This is rare,
1565 * thus it's OK to do another malloc()/free(). */
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001566 l = utf_ptr2len(p);
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001567 nl = utf_char2len(uc);
1568 if (nl != l)
1569 {
1570 s = alloc((unsigned)STRLEN(res) + 1 + nl - l);
1571 if (s == NULL)
1572 break;
1573 mch_memmove(s, res, p - res);
1574 STRCPY(s + (p - res) + nl, p + l);
1575 p = s + (p - res);
1576 vim_free(res);
1577 res = s;
1578 }
1579
1580 utf_char2bytes(uc, p);
1581 p += nl;
1582 }
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001583 else if (has_mbyte && (l = (*mb_ptr2len)(p)) > 1)
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001584 p += l; /* skip multi-byte character */
1585 else
1586# endif
1587 {
1588 *p = TOUPPER_LOC(*p); /* note that toupper() can be a macro */
1589 p++;
1590 }
1591 }
1592
1593 return res;
1594}
1595#endif
1596
Bram Moolenaar071d4272004-06-13 20:20:40 +00001597/*
1598 * copy a space a number of times
1599 */
1600 void
1601copy_spaces(ptr, count)
1602 char_u *ptr;
1603 size_t count;
1604{
1605 size_t i = count;
1606 char_u *p = ptr;
1607
1608 while (i--)
1609 *p++ = ' ';
1610}
1611
1612#if defined(FEAT_VISUALEXTRA) || defined(PROTO)
1613/*
1614 * Copy a character a number of times.
Bram Moolenaar21160b92009-11-11 15:56:10 +00001615 * Does not work for multi-byte characters!
Bram Moolenaar071d4272004-06-13 20:20:40 +00001616 */
1617 void
1618copy_chars(ptr, count, c)
1619 char_u *ptr;
1620 size_t count;
1621 int c;
1622{
1623 size_t i = count;
1624 char_u *p = ptr;
1625
1626 while (i--)
1627 *p++ = c;
1628}
1629#endif
1630
1631/*
1632 * delete spaces at the end of a string
1633 */
1634 void
1635del_trailing_spaces(ptr)
1636 char_u *ptr;
1637{
1638 char_u *q;
1639
1640 q = ptr + STRLEN(ptr);
1641 while (--q > ptr && vim_iswhite(q[0]) && q[-1] != '\\' && q[-1] != Ctrl_V)
1642 *q = NUL;
1643}
1644
1645/*
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001646 * Like strncpy(), but always terminate the result with one NUL.
Bram Moolenaard042c562005-06-30 22:04:15 +00001647 * "to" must be "len + 1" long!
Bram Moolenaar071d4272004-06-13 20:20:40 +00001648 */
1649 void
1650vim_strncpy(to, from, len)
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001651 char_u *to;
1652 char_u *from;
Bram Moolenaarbbebc852005-07-18 21:47:53 +00001653 size_t len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001654{
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001655 STRNCPY(to, from, len);
1656 to[len] = NUL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001657}
1658
1659/*
Bram Moolenaaref9d6aa2011-04-11 16:56:35 +02001660 * Like strcat(), but make sure the result fits in "tosize" bytes and is
1661 * always NUL terminated.
1662 */
1663 void
1664vim_strcat(to, from, tosize)
1665 char_u *to;
1666 char_u *from;
1667 size_t tosize;
1668{
1669 size_t tolen = STRLEN(to);
1670 size_t fromlen = STRLEN(from);
1671
1672 if (tolen + fromlen + 1 > tosize)
1673 {
1674 mch_memmove(to + tolen, from, tosize - tolen - 1);
1675 to[tosize - 1] = NUL;
1676 }
1677 else
1678 STRCPY(to + tolen, from);
1679}
1680
1681/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00001682 * Isolate one part of a string option where parts are separated with
1683 * "sep_chars".
Bram Moolenaar83bab712005-08-01 21:58:57 +00001684 * The part is copied into "buf[maxlen]".
Bram Moolenaar071d4272004-06-13 20:20:40 +00001685 * "*option" is advanced to the next part.
1686 * The length is returned.
1687 */
1688 int
1689copy_option_part(option, buf, maxlen, sep_chars)
1690 char_u **option;
1691 char_u *buf;
1692 int maxlen;
1693 char *sep_chars;
1694{
1695 int len = 0;
1696 char_u *p = *option;
1697
1698 /* skip '.' at start of option part, for 'suffixes' */
1699 if (*p == '.')
1700 buf[len++] = *p++;
1701 while (*p != NUL && vim_strchr((char_u *)sep_chars, *p) == NULL)
1702 {
1703 /*
1704 * Skip backslash before a separator character and space.
1705 */
1706 if (p[0] == '\\' && vim_strchr((char_u *)sep_chars, p[1]) != NULL)
1707 ++p;
1708 if (len < maxlen - 1)
1709 buf[len++] = *p;
1710 ++p;
1711 }
1712 buf[len] = NUL;
1713
1714 if (*p != NUL && *p != ',') /* skip non-standard separator */
1715 ++p;
1716 p = skip_to_option_part(p); /* p points to next file name */
1717
1718 *option = p;
1719 return len;
1720}
1721
1722/*
Bram Moolenaar4770d092006-01-12 23:22:24 +00001723 * Replacement for free() that ignores NULL pointers.
1724 * Also skip free() when exiting for sure, this helps when we caught a deadly
1725 * signal that was caused by a crash in free().
Bram Moolenaar071d4272004-06-13 20:20:40 +00001726 */
1727 void
1728vim_free(x)
1729 void *x;
1730{
Bram Moolenaar4770d092006-01-12 23:22:24 +00001731 if (x != NULL && !really_exiting)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001732 {
1733#ifdef MEM_PROFILE
1734 mem_pre_free(&x);
1735#endif
1736 free(x);
1737 }
1738}
1739
1740#ifndef HAVE_MEMSET
1741 void *
1742vim_memset(ptr, c, size)
1743 void *ptr;
1744 int c;
1745 size_t size;
1746{
1747 char *p = ptr;
1748
1749 while (size-- > 0)
1750 *p++ = c;
1751 return ptr;
1752}
1753#endif
1754
1755#ifdef VIM_MEMCMP
1756/*
1757 * Return zero when "b1" and "b2" are the same for "len" bytes.
1758 * Return non-zero otherwise.
1759 */
1760 int
1761vim_memcmp(b1, b2, len)
1762 void *b1;
1763 void *b2;
1764 size_t len;
1765{
1766 char_u *p1 = (char_u *)b1, *p2 = (char_u *)b2;
1767
1768 for ( ; len > 0; --len)
1769 {
1770 if (*p1 != *p2)
1771 return 1;
1772 ++p1;
1773 ++p2;
1774 }
1775 return 0;
1776}
1777#endif
1778
1779#ifdef VIM_MEMMOVE
1780/*
1781 * Version of memmove() that handles overlapping source and destination.
1782 * For systems that don't have a function that is guaranteed to do that (SYSV).
1783 */
1784 void
1785mch_memmove(dst_arg, src_arg, len)
1786 void *src_arg, *dst_arg;
1787 size_t len;
1788{
1789 /*
1790 * A void doesn't have a size, we use char pointers.
1791 */
1792 char *dst = dst_arg, *src = src_arg;
1793
1794 /* overlap, copy backwards */
1795 if (dst > src && dst < src + len)
1796 {
1797 src += len;
1798 dst += len;
1799 while (len-- > 0)
1800 *--dst = *--src;
1801 }
1802 else /* copy forwards */
1803 while (len-- > 0)
1804 *dst++ = *src++;
1805}
1806#endif
1807
1808#if (!defined(HAVE_STRCASECMP) && !defined(HAVE_STRICMP)) || defined(PROTO)
1809/*
1810 * Compare two strings, ignoring case, using current locale.
1811 * Doesn't work for multi-byte characters.
1812 * return 0 for match, < 0 for smaller, > 0 for bigger
1813 */
1814 int
1815vim_stricmp(s1, s2)
1816 char *s1;
1817 char *s2;
1818{
1819 int i;
1820
1821 for (;;)
1822 {
1823 i = (int)TOLOWER_LOC(*s1) - (int)TOLOWER_LOC(*s2);
1824 if (i != 0)
1825 return i; /* this character different */
1826 if (*s1 == NUL)
1827 break; /* strings match until NUL */
1828 ++s1;
1829 ++s2;
1830 }
1831 return 0; /* strings match */
1832}
1833#endif
1834
1835#if (!defined(HAVE_STRNCASECMP) && !defined(HAVE_STRNICMP)) || defined(PROTO)
1836/*
1837 * Compare two strings, for length "len", ignoring case, using current locale.
1838 * Doesn't work for multi-byte characters.
1839 * return 0 for match, < 0 for smaller, > 0 for bigger
1840 */
1841 int
1842vim_strnicmp(s1, s2, len)
1843 char *s1;
1844 char *s2;
1845 size_t len;
1846{
1847 int i;
1848
1849 while (len > 0)
1850 {
1851 i = (int)TOLOWER_LOC(*s1) - (int)TOLOWER_LOC(*s2);
1852 if (i != 0)
1853 return i; /* this character different */
1854 if (*s1 == NUL)
1855 break; /* strings match until NUL */
1856 ++s1;
1857 ++s2;
1858 --len;
1859 }
1860 return 0; /* strings match */
1861}
1862#endif
1863
Bram Moolenaar071d4272004-06-13 20:20:40 +00001864/*
1865 * Version of strchr() and strrchr() that handle unsigned char strings
Bram Moolenaar05159a02005-02-26 23:04:13 +00001866 * with characters from 128 to 255 correctly. It also doesn't return a
1867 * pointer to the NUL at the end of the string.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001868 */
1869 char_u *
1870vim_strchr(string, c)
1871 char_u *string;
1872 int c;
1873{
1874 char_u *p;
1875 int b;
1876
1877 p = string;
1878#ifdef FEAT_MBYTE
1879 if (enc_utf8 && c >= 0x80)
1880 {
1881 while (*p != NUL)
1882 {
1883 if (utf_ptr2char(p) == c)
1884 return p;
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001885 p += (*mb_ptr2len)(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001886 }
1887 return NULL;
1888 }
1889 if (enc_dbcs != 0 && c > 255)
1890 {
1891 int n2 = c & 0xff;
1892
1893 c = ((unsigned)c >> 8) & 0xff;
1894 while ((b = *p) != NUL)
1895 {
1896 if (b == c && p[1] == n2)
1897 return p;
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001898 p += (*mb_ptr2len)(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001899 }
1900 return NULL;
1901 }
1902 if (has_mbyte)
1903 {
1904 while ((b = *p) != NUL)
1905 {
1906 if (b == c)
1907 return p;
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001908 p += (*mb_ptr2len)(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001909 }
1910 return NULL;
1911 }
1912#endif
1913 while ((b = *p) != NUL)
1914 {
1915 if (b == c)
1916 return p;
1917 ++p;
1918 }
1919 return NULL;
1920}
1921
1922/*
Bram Moolenaar05159a02005-02-26 23:04:13 +00001923 * Version of strchr() that only works for bytes and handles unsigned char
1924 * strings with characters above 128 correctly. It also doesn't return a
1925 * pointer to the NUL at the end of the string.
1926 */
1927 char_u *
1928vim_strbyte(string, c)
1929 char_u *string;
1930 int c;
1931{
1932 char_u *p = string;
1933
1934 while (*p != NUL)
1935 {
1936 if (*p == c)
1937 return p;
1938 ++p;
1939 }
1940 return NULL;
1941}
1942
1943/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00001944 * Search for last occurrence of "c" in "string".
Bram Moolenaarac6e65f2005-08-29 22:25:38 +00001945 * Return NULL if not found.
Bram Moolenaar05159a02005-02-26 23:04:13 +00001946 * Does not handle multi-byte char for "c"!
Bram Moolenaar071d4272004-06-13 20:20:40 +00001947 */
1948 char_u *
1949vim_strrchr(string, c)
1950 char_u *string;
1951 int c;
1952{
1953 char_u *retval = NULL;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001954 char_u *p = string;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001955
Bram Moolenaar05159a02005-02-26 23:04:13 +00001956 while (*p)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001957 {
Bram Moolenaar05159a02005-02-26 23:04:13 +00001958 if (*p == c)
1959 retval = p;
1960 mb_ptr_adv(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001961 }
1962 return retval;
1963}
1964
1965/*
1966 * Vim's version of strpbrk(), in case it's missing.
1967 * Don't generate a prototype for this, causes problems when it's not used.
1968 */
1969#ifndef PROTO
1970# ifndef HAVE_STRPBRK
1971# ifdef vim_strpbrk
1972# undef vim_strpbrk
1973# endif
1974 char_u *
1975vim_strpbrk(s, charset)
1976 char_u *s;
1977 char_u *charset;
1978{
1979 while (*s)
1980 {
1981 if (vim_strchr(charset, *s) != NULL)
1982 return s;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001983 mb_ptr_adv(s);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001984 }
1985 return NULL;
1986}
1987# endif
1988#endif
1989
1990/*
1991 * Vim has its own isspace() function, because on some machines isspace()
1992 * can't handle characters above 128.
1993 */
1994 int
1995vim_isspace(x)
1996 int x;
1997{
1998 return ((x >= 9 && x <= 13) || x == ' ');
1999}
2000
2001/************************************************************************
Bram Moolenaar383f9bc2005-01-19 22:18:32 +00002002 * Functions for handling growing arrays.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002003 */
2004
2005/*
2006 * Clear an allocated growing array.
2007 */
2008 void
2009ga_clear(gap)
2010 garray_T *gap;
2011{
2012 vim_free(gap->ga_data);
2013 ga_init(gap);
2014}
2015
2016/*
2017 * Clear a growing array that contains a list of strings.
2018 */
2019 void
2020ga_clear_strings(gap)
2021 garray_T *gap;
2022{
2023 int i;
2024
2025 for (i = 0; i < gap->ga_len; ++i)
2026 vim_free(((char_u **)(gap->ga_data))[i]);
2027 ga_clear(gap);
2028}
2029
2030/*
2031 * Initialize a growing array. Don't forget to set ga_itemsize and
2032 * ga_growsize! Or use ga_init2().
2033 */
2034 void
2035ga_init(gap)
2036 garray_T *gap;
2037{
2038 gap->ga_data = NULL;
Bram Moolenaar86b68352004-12-27 21:59:20 +00002039 gap->ga_maxlen = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002040 gap->ga_len = 0;
2041}
2042
2043 void
2044ga_init2(gap, itemsize, growsize)
2045 garray_T *gap;
2046 int itemsize;
2047 int growsize;
2048{
2049 ga_init(gap);
2050 gap->ga_itemsize = itemsize;
2051 gap->ga_growsize = growsize;
2052}
2053
2054/*
2055 * Make room in growing array "gap" for at least "n" items.
2056 * Return FAIL for failure, OK otherwise.
2057 */
2058 int
2059ga_grow(gap, n)
2060 garray_T *gap;
2061 int n;
2062{
2063 size_t len;
2064 char_u *pp;
2065
Bram Moolenaar86b68352004-12-27 21:59:20 +00002066 if (gap->ga_maxlen - gap->ga_len < n)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002067 {
2068 if (n < gap->ga_growsize)
2069 n = gap->ga_growsize;
2070 len = gap->ga_itemsize * (gap->ga_len + n);
2071 pp = alloc_clear((unsigned)len);
2072 if (pp == NULL)
2073 return FAIL;
Bram Moolenaar86b68352004-12-27 21:59:20 +00002074 gap->ga_maxlen = gap->ga_len + n;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002075 if (gap->ga_data != NULL)
2076 {
2077 mch_memmove(pp, gap->ga_data,
2078 (size_t)(gap->ga_itemsize * gap->ga_len));
2079 vim_free(gap->ga_data);
2080 }
2081 gap->ga_data = pp;
2082 }
2083 return OK;
2084}
2085
2086/*
Bram Moolenaar80a7dcf2010-08-04 17:07:20 +02002087 * For a growing array that contains a list of strings: concatenate all the
2088 * strings with a separating comma.
2089 * Returns NULL when out of memory.
2090 */
2091 char_u *
2092ga_concat_strings(gap)
2093 garray_T *gap;
2094{
2095 int i;
2096 int len = 0;
2097 char_u *s;
2098
2099 for (i = 0; i < gap->ga_len; ++i)
2100 len += (int)STRLEN(((char_u **)(gap->ga_data))[i]) + 1;
2101
2102 s = alloc(len + 1);
2103 if (s != NULL)
2104 {
2105 *s = NUL;
2106 for (i = 0; i < gap->ga_len; ++i)
2107 {
2108 if (*s != NUL)
2109 STRCAT(s, ",");
2110 STRCAT(s, ((char_u **)(gap->ga_data))[i]);
2111 }
2112 }
2113 return s;
2114}
2115
2116/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002117 * Concatenate a string to a growarray which contains characters.
2118 * Note: Does NOT copy the NUL at the end!
2119 */
2120 void
2121ga_concat(gap, s)
2122 garray_T *gap;
2123 char_u *s;
2124{
2125 int len = (int)STRLEN(s);
2126
2127 if (ga_grow(gap, len) == OK)
2128 {
2129 mch_memmove((char *)gap->ga_data + gap->ga_len, s, (size_t)len);
2130 gap->ga_len += len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002131 }
2132}
2133
2134/*
2135 * Append one byte to a growarray which contains bytes.
2136 */
2137 void
2138ga_append(gap, c)
2139 garray_T *gap;
2140 int c;
2141{
2142 if (ga_grow(gap, 1) == OK)
2143 {
2144 *((char *)gap->ga_data + gap->ga_len) = c;
2145 ++gap->ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002146 }
2147}
2148
2149/************************************************************************
2150 * functions that use lookup tables for various things, generally to do with
2151 * special key codes.
2152 */
2153
2154/*
2155 * Some useful tables.
2156 */
2157
2158static struct modmasktable
2159{
2160 short mod_mask; /* Bit-mask for particular key modifier */
2161 short mod_flag; /* Bit(s) for particular key modifier */
2162 char_u name; /* Single letter name of modifier */
2163} mod_mask_table[] =
2164{
2165 {MOD_MASK_ALT, MOD_MASK_ALT, (char_u)'M'},
Bram Moolenaar19a09a12005-03-04 23:39:37 +00002166 {MOD_MASK_META, MOD_MASK_META, (char_u)'T'},
Bram Moolenaar071d4272004-06-13 20:20:40 +00002167 {MOD_MASK_CTRL, MOD_MASK_CTRL, (char_u)'C'},
2168 {MOD_MASK_SHIFT, MOD_MASK_SHIFT, (char_u)'S'},
2169 {MOD_MASK_MULTI_CLICK, MOD_MASK_2CLICK, (char_u)'2'},
2170 {MOD_MASK_MULTI_CLICK, MOD_MASK_3CLICK, (char_u)'3'},
2171 {MOD_MASK_MULTI_CLICK, MOD_MASK_4CLICK, (char_u)'4'},
2172#ifdef MACOS
2173 {MOD_MASK_CMD, MOD_MASK_CMD, (char_u)'D'},
2174#endif
2175 /* 'A' must be the last one */
2176 {MOD_MASK_ALT, MOD_MASK_ALT, (char_u)'A'},
2177 {0, 0, NUL}
2178};
2179
2180/*
2181 * Shifted key terminal codes and their unshifted equivalent.
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00002182 * Don't add mouse codes here, they are handled separately!
Bram Moolenaar071d4272004-06-13 20:20:40 +00002183 */
2184#define MOD_KEYS_ENTRY_SIZE 5
2185
2186static char_u modifier_keys_table[] =
2187{
2188/* mod mask with modifier without modifier */
2189 MOD_MASK_SHIFT, '&', '9', '@', '1', /* begin */
2190 MOD_MASK_SHIFT, '&', '0', '@', '2', /* cancel */
2191 MOD_MASK_SHIFT, '*', '1', '@', '4', /* command */
2192 MOD_MASK_SHIFT, '*', '2', '@', '5', /* copy */
2193 MOD_MASK_SHIFT, '*', '3', '@', '6', /* create */
2194 MOD_MASK_SHIFT, '*', '4', 'k', 'D', /* delete char */
2195 MOD_MASK_SHIFT, '*', '5', 'k', 'L', /* delete line */
2196 MOD_MASK_SHIFT, '*', '7', '@', '7', /* end */
2197 MOD_MASK_CTRL, KS_EXTRA, (int)KE_C_END, '@', '7', /* end */
2198 MOD_MASK_SHIFT, '*', '9', '@', '9', /* exit */
2199 MOD_MASK_SHIFT, '*', '0', '@', '0', /* find */
2200 MOD_MASK_SHIFT, '#', '1', '%', '1', /* help */
2201 MOD_MASK_SHIFT, '#', '2', 'k', 'h', /* home */
2202 MOD_MASK_CTRL, KS_EXTRA, (int)KE_C_HOME, 'k', 'h', /* home */
2203 MOD_MASK_SHIFT, '#', '3', 'k', 'I', /* insert */
2204 MOD_MASK_SHIFT, '#', '4', 'k', 'l', /* left arrow */
2205 MOD_MASK_CTRL, KS_EXTRA, (int)KE_C_LEFT, 'k', 'l', /* left arrow */
2206 MOD_MASK_SHIFT, '%', 'a', '%', '3', /* message */
2207 MOD_MASK_SHIFT, '%', 'b', '%', '4', /* move */
2208 MOD_MASK_SHIFT, '%', 'c', '%', '5', /* next */
2209 MOD_MASK_SHIFT, '%', 'd', '%', '7', /* options */
2210 MOD_MASK_SHIFT, '%', 'e', '%', '8', /* previous */
2211 MOD_MASK_SHIFT, '%', 'f', '%', '9', /* print */
2212 MOD_MASK_SHIFT, '%', 'g', '%', '0', /* redo */
2213 MOD_MASK_SHIFT, '%', 'h', '&', '3', /* replace */
2214 MOD_MASK_SHIFT, '%', 'i', 'k', 'r', /* right arr. */
2215 MOD_MASK_CTRL, KS_EXTRA, (int)KE_C_RIGHT, 'k', 'r', /* right arr. */
2216 MOD_MASK_SHIFT, '%', 'j', '&', '5', /* resume */
2217 MOD_MASK_SHIFT, '!', '1', '&', '6', /* save */
2218 MOD_MASK_SHIFT, '!', '2', '&', '7', /* suspend */
2219 MOD_MASK_SHIFT, '!', '3', '&', '8', /* undo */
2220 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_UP, 'k', 'u', /* up arrow */
2221 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_DOWN, 'k', 'd', /* down arrow */
2222
2223 /* vt100 F1 */
2224 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_XF1, KS_EXTRA, (int)KE_XF1,
2225 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_XF2, KS_EXTRA, (int)KE_XF2,
2226 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_XF3, KS_EXTRA, (int)KE_XF3,
2227 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_XF4, KS_EXTRA, (int)KE_XF4,
2228
2229 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F1, 'k', '1', /* F1 */
2230 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F2, 'k', '2',
2231 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F3, 'k', '3',
2232 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F4, 'k', '4',
2233 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F5, 'k', '5',
2234 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F6, 'k', '6',
2235 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F7, 'k', '7',
2236 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F8, 'k', '8',
2237 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F9, 'k', '9',
2238 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F10, 'k', ';', /* F10 */
2239
2240 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F11, 'F', '1',
2241 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F12, 'F', '2',
2242 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F13, 'F', '3',
2243 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F14, 'F', '4',
2244 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F15, 'F', '5',
2245 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F16, 'F', '6',
2246 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F17, 'F', '7',
2247 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F18, 'F', '8',
2248 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F19, 'F', '9',
2249 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F20, 'F', 'A',
2250
2251 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F21, 'F', 'B',
2252 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F22, 'F', 'C',
2253 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F23, 'F', 'D',
2254 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F24, 'F', 'E',
2255 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F25, 'F', 'F',
2256 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F26, 'F', 'G',
2257 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F27, 'F', 'H',
2258 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F28, 'F', 'I',
2259 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F29, 'F', 'J',
2260 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F30, 'F', 'K',
2261
2262 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F31, 'F', 'L',
2263 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F32, 'F', 'M',
2264 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F33, 'F', 'N',
2265 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F34, 'F', 'O',
2266 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F35, 'F', 'P',
2267 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F36, 'F', 'Q',
2268 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F37, 'F', 'R',
2269
2270 /* TAB pseudo code*/
2271 MOD_MASK_SHIFT, 'k', 'B', KS_EXTRA, (int)KE_TAB,
2272
2273 NUL
2274};
2275
2276static struct key_name_entry
2277{
2278 int key; /* Special key code or ascii value */
2279 char_u *name; /* Name of key */
2280} key_names_table[] =
2281{
2282 {' ', (char_u *)"Space"},
2283 {TAB, (char_u *)"Tab"},
2284 {K_TAB, (char_u *)"Tab"},
2285 {NL, (char_u *)"NL"},
2286 {NL, (char_u *)"NewLine"}, /* Alternative name */
2287 {NL, (char_u *)"LineFeed"}, /* Alternative name */
2288 {NL, (char_u *)"LF"}, /* Alternative name */
2289 {CAR, (char_u *)"CR"},
2290 {CAR, (char_u *)"Return"}, /* Alternative name */
2291 {CAR, (char_u *)"Enter"}, /* Alternative name */
2292 {K_BS, (char_u *)"BS"},
2293 {K_BS, (char_u *)"BackSpace"}, /* Alternative name */
2294 {ESC, (char_u *)"Esc"},
2295 {CSI, (char_u *)"CSI"},
2296 {K_CSI, (char_u *)"xCSI"},
2297 {'|', (char_u *)"Bar"},
2298 {'\\', (char_u *)"Bslash"},
2299 {K_DEL, (char_u *)"Del"},
2300 {K_DEL, (char_u *)"Delete"}, /* Alternative name */
2301 {K_KDEL, (char_u *)"kDel"},
2302 {K_UP, (char_u *)"Up"},
2303 {K_DOWN, (char_u *)"Down"},
2304 {K_LEFT, (char_u *)"Left"},
2305 {K_RIGHT, (char_u *)"Right"},
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002306 {K_XUP, (char_u *)"xUp"},
2307 {K_XDOWN, (char_u *)"xDown"},
2308 {K_XLEFT, (char_u *)"xLeft"},
2309 {K_XRIGHT, (char_u *)"xRight"},
Bram Moolenaar071d4272004-06-13 20:20:40 +00002310
2311 {K_F1, (char_u *)"F1"},
2312 {K_F2, (char_u *)"F2"},
2313 {K_F3, (char_u *)"F3"},
2314 {K_F4, (char_u *)"F4"},
2315 {K_F5, (char_u *)"F5"},
2316 {K_F6, (char_u *)"F6"},
2317 {K_F7, (char_u *)"F7"},
2318 {K_F8, (char_u *)"F8"},
2319 {K_F9, (char_u *)"F9"},
2320 {K_F10, (char_u *)"F10"},
2321
2322 {K_F11, (char_u *)"F11"},
2323 {K_F12, (char_u *)"F12"},
2324 {K_F13, (char_u *)"F13"},
2325 {K_F14, (char_u *)"F14"},
2326 {K_F15, (char_u *)"F15"},
2327 {K_F16, (char_u *)"F16"},
2328 {K_F17, (char_u *)"F17"},
2329 {K_F18, (char_u *)"F18"},
2330 {K_F19, (char_u *)"F19"},
2331 {K_F20, (char_u *)"F20"},
2332
2333 {K_F21, (char_u *)"F21"},
2334 {K_F22, (char_u *)"F22"},
2335 {K_F23, (char_u *)"F23"},
2336 {K_F24, (char_u *)"F24"},
2337 {K_F25, (char_u *)"F25"},
2338 {K_F26, (char_u *)"F26"},
2339 {K_F27, (char_u *)"F27"},
2340 {K_F28, (char_u *)"F28"},
2341 {K_F29, (char_u *)"F29"},
2342 {K_F30, (char_u *)"F30"},
2343
2344 {K_F31, (char_u *)"F31"},
2345 {K_F32, (char_u *)"F32"},
2346 {K_F33, (char_u *)"F33"},
2347 {K_F34, (char_u *)"F34"},
2348 {K_F35, (char_u *)"F35"},
2349 {K_F36, (char_u *)"F36"},
2350 {K_F37, (char_u *)"F37"},
2351
2352 {K_XF1, (char_u *)"xF1"},
2353 {K_XF2, (char_u *)"xF2"},
2354 {K_XF3, (char_u *)"xF3"},
2355 {K_XF4, (char_u *)"xF4"},
2356
2357 {K_HELP, (char_u *)"Help"},
2358 {K_UNDO, (char_u *)"Undo"},
2359 {K_INS, (char_u *)"Insert"},
2360 {K_INS, (char_u *)"Ins"}, /* Alternative name */
2361 {K_KINS, (char_u *)"kInsert"},
2362 {K_HOME, (char_u *)"Home"},
2363 {K_KHOME, (char_u *)"kHome"},
2364 {K_XHOME, (char_u *)"xHome"},
Bram Moolenaar68b76a62005-03-25 21:53:48 +00002365 {K_ZHOME, (char_u *)"zHome"},
Bram Moolenaar071d4272004-06-13 20:20:40 +00002366 {K_END, (char_u *)"End"},
2367 {K_KEND, (char_u *)"kEnd"},
2368 {K_XEND, (char_u *)"xEnd"},
Bram Moolenaar68b76a62005-03-25 21:53:48 +00002369 {K_ZEND, (char_u *)"zEnd"},
Bram Moolenaar071d4272004-06-13 20:20:40 +00002370 {K_PAGEUP, (char_u *)"PageUp"},
2371 {K_PAGEDOWN, (char_u *)"PageDown"},
2372 {K_KPAGEUP, (char_u *)"kPageUp"},
2373 {K_KPAGEDOWN, (char_u *)"kPageDown"},
2374
2375 {K_KPLUS, (char_u *)"kPlus"},
2376 {K_KMINUS, (char_u *)"kMinus"},
2377 {K_KDIVIDE, (char_u *)"kDivide"},
2378 {K_KMULTIPLY, (char_u *)"kMultiply"},
2379 {K_KENTER, (char_u *)"kEnter"},
2380 {K_KPOINT, (char_u *)"kPoint"},
2381
2382 {K_K0, (char_u *)"k0"},
2383 {K_K1, (char_u *)"k1"},
2384 {K_K2, (char_u *)"k2"},
2385 {K_K3, (char_u *)"k3"},
2386 {K_K4, (char_u *)"k4"},
2387 {K_K5, (char_u *)"k5"},
2388 {K_K6, (char_u *)"k6"},
2389 {K_K7, (char_u *)"k7"},
2390 {K_K8, (char_u *)"k8"},
2391 {K_K9, (char_u *)"k9"},
2392
2393 {'<', (char_u *)"lt"},
2394
2395 {K_MOUSE, (char_u *)"Mouse"},
2396 {K_NETTERM_MOUSE, (char_u *)"NetMouse"},
2397 {K_DEC_MOUSE, (char_u *)"DecMouse"},
2398 {K_JSBTERM_MOUSE, (char_u *)"JsbMouse"},
2399 {K_PTERM_MOUSE, (char_u *)"PtermMouse"},
2400 {K_LEFTMOUSE, (char_u *)"LeftMouse"},
2401 {K_LEFTMOUSE_NM, (char_u *)"LeftMouseNM"},
2402 {K_LEFTDRAG, (char_u *)"LeftDrag"},
2403 {K_LEFTRELEASE, (char_u *)"LeftRelease"},
2404 {K_LEFTRELEASE_NM, (char_u *)"LeftReleaseNM"},
2405 {K_MIDDLEMOUSE, (char_u *)"MiddleMouse"},
2406 {K_MIDDLEDRAG, (char_u *)"MiddleDrag"},
2407 {K_MIDDLERELEASE, (char_u *)"MiddleRelease"},
2408 {K_RIGHTMOUSE, (char_u *)"RightMouse"},
2409 {K_RIGHTDRAG, (char_u *)"RightDrag"},
2410 {K_RIGHTRELEASE, (char_u *)"RightRelease"},
Bram Moolenaar8d9b40e2010-07-25 15:49:07 +02002411 {K_MOUSEDOWN, (char_u *)"ScrollWheelUp"},
2412 {K_MOUSEUP, (char_u *)"ScrollWheelDown"},
2413 {K_MOUSELEFT, (char_u *)"ScrollWheelRight"},
2414 {K_MOUSERIGHT, (char_u *)"ScrollWheelLeft"},
2415 {K_MOUSEDOWN, (char_u *)"MouseDown"}, /* OBSOLETE: Use */
2416 {K_MOUSEUP, (char_u *)"MouseUp"}, /* ScrollWheelXXX instead */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002417 {K_X1MOUSE, (char_u *)"X1Mouse"},
2418 {K_X1DRAG, (char_u *)"X1Drag"},
2419 {K_X1RELEASE, (char_u *)"X1Release"},
2420 {K_X2MOUSE, (char_u *)"X2Mouse"},
2421 {K_X2DRAG, (char_u *)"X2Drag"},
2422 {K_X2RELEASE, (char_u *)"X2Release"},
2423 {K_DROP, (char_u *)"Drop"},
2424 {K_ZERO, (char_u *)"Nul"},
2425#ifdef FEAT_EVAL
2426 {K_SNR, (char_u *)"SNR"},
2427#endif
2428 {K_PLUG, (char_u *)"Plug"},
2429 {0, NULL}
2430};
2431
2432#define KEY_NAMES_TABLE_LEN (sizeof(key_names_table) / sizeof(struct key_name_entry))
2433
2434#ifdef FEAT_MOUSE
2435static struct mousetable
2436{
2437 int pseudo_code; /* Code for pseudo mouse event */
2438 int button; /* Which mouse button is it? */
2439 int is_click; /* Is it a mouse button click event? */
2440 int is_drag; /* Is it a mouse drag event? */
2441} mouse_table[] =
2442{
2443 {(int)KE_LEFTMOUSE, MOUSE_LEFT, TRUE, FALSE},
2444#ifdef FEAT_GUI
2445 {(int)KE_LEFTMOUSE_NM, MOUSE_LEFT, TRUE, FALSE},
2446#endif
2447 {(int)KE_LEFTDRAG, MOUSE_LEFT, FALSE, TRUE},
2448 {(int)KE_LEFTRELEASE, MOUSE_LEFT, FALSE, FALSE},
2449#ifdef FEAT_GUI
2450 {(int)KE_LEFTRELEASE_NM, MOUSE_LEFT, FALSE, FALSE},
2451#endif
2452 {(int)KE_MIDDLEMOUSE, MOUSE_MIDDLE, TRUE, FALSE},
2453 {(int)KE_MIDDLEDRAG, MOUSE_MIDDLE, FALSE, TRUE},
2454 {(int)KE_MIDDLERELEASE, MOUSE_MIDDLE, FALSE, FALSE},
2455 {(int)KE_RIGHTMOUSE, MOUSE_RIGHT, TRUE, FALSE},
2456 {(int)KE_RIGHTDRAG, MOUSE_RIGHT, FALSE, TRUE},
2457 {(int)KE_RIGHTRELEASE, MOUSE_RIGHT, FALSE, FALSE},
2458 {(int)KE_X1MOUSE, MOUSE_X1, TRUE, FALSE},
2459 {(int)KE_X1DRAG, MOUSE_X1, FALSE, TRUE},
2460 {(int)KE_X1RELEASE, MOUSE_X1, FALSE, FALSE},
2461 {(int)KE_X2MOUSE, MOUSE_X2, TRUE, FALSE},
2462 {(int)KE_X2DRAG, MOUSE_X2, FALSE, TRUE},
2463 {(int)KE_X2RELEASE, MOUSE_X2, FALSE, FALSE},
2464 /* DRAG without CLICK */
2465 {(int)KE_IGNORE, MOUSE_RELEASE, FALSE, TRUE},
2466 /* RELEASE without CLICK */
2467 {(int)KE_IGNORE, MOUSE_RELEASE, FALSE, FALSE},
2468 {0, 0, 0, 0},
2469};
2470#endif /* FEAT_MOUSE */
2471
2472/*
2473 * Return the modifier mask bit (MOD_MASK_*) which corresponds to the given
2474 * modifier name ('S' for Shift, 'C' for Ctrl etc).
2475 */
2476 int
2477name_to_mod_mask(c)
2478 int c;
2479{
2480 int i;
2481
2482 c = TOUPPER_ASC(c);
2483 for (i = 0; mod_mask_table[i].mod_mask != 0; i++)
2484 if (c == mod_mask_table[i].name)
2485 return mod_mask_table[i].mod_flag;
2486 return 0;
2487}
2488
Bram Moolenaar071d4272004-06-13 20:20:40 +00002489/*
2490 * Check if if there is a special key code for "key" that includes the
2491 * modifiers specified.
2492 */
2493 int
2494simplify_key(key, modifiers)
2495 int key;
2496 int *modifiers;
2497{
2498 int i;
2499 int key0;
2500 int key1;
2501
2502 if (*modifiers & (MOD_MASK_SHIFT | MOD_MASK_CTRL | MOD_MASK_ALT))
2503 {
2504 /* TAB is a special case */
2505 if (key == TAB && (*modifiers & MOD_MASK_SHIFT))
2506 {
2507 *modifiers &= ~MOD_MASK_SHIFT;
2508 return K_S_TAB;
2509 }
2510 key0 = KEY2TERMCAP0(key);
2511 key1 = KEY2TERMCAP1(key);
2512 for (i = 0; modifier_keys_table[i] != NUL; i += MOD_KEYS_ENTRY_SIZE)
2513 if (key0 == modifier_keys_table[i + 3]
2514 && key1 == modifier_keys_table[i + 4]
2515 && (*modifiers & modifier_keys_table[i]))
2516 {
2517 *modifiers &= ~modifier_keys_table[i];
2518 return TERMCAP2KEY(modifier_keys_table[i + 1],
2519 modifier_keys_table[i + 2]);
2520 }
2521 }
2522 return key;
2523}
2524
2525/*
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002526 * Change <xHome> to <Home>, <xUp> to <Up>, etc.
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002527 */
2528 int
2529handle_x_keys(key)
2530 int key;
2531{
2532 switch (key)
2533 {
2534 case K_XUP: return K_UP;
2535 case K_XDOWN: return K_DOWN;
2536 case K_XLEFT: return K_LEFT;
2537 case K_XRIGHT: return K_RIGHT;
2538 case K_XHOME: return K_HOME;
Bram Moolenaar68b76a62005-03-25 21:53:48 +00002539 case K_ZHOME: return K_HOME;
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002540 case K_XEND: return K_END;
Bram Moolenaar68b76a62005-03-25 21:53:48 +00002541 case K_ZEND: return K_END;
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002542 case K_XF1: return K_F1;
2543 case K_XF2: return K_F2;
2544 case K_XF3: return K_F3;
2545 case K_XF4: return K_F4;
2546 case K_S_XF1: return K_S_F1;
2547 case K_S_XF2: return K_S_F2;
2548 case K_S_XF3: return K_S_F3;
2549 case K_S_XF4: return K_S_F4;
2550 }
2551 return key;
2552}
2553
2554/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002555 * Return a string which contains the name of the given key when the given
2556 * modifiers are down.
2557 */
2558 char_u *
2559get_special_key_name(c, modifiers)
2560 int c;
2561 int modifiers;
2562{
2563 static char_u string[MAX_KEY_NAME_LEN + 1];
2564
2565 int i, idx;
2566 int table_idx;
2567 char_u *s;
2568
2569 string[0] = '<';
2570 idx = 1;
2571
2572 /* Key that stands for a normal character. */
2573 if (IS_SPECIAL(c) && KEY2TERMCAP0(c) == KS_KEY)
2574 c = KEY2TERMCAP1(c);
2575
2576 /*
2577 * Translate shifted special keys into unshifted keys and set modifier.
2578 * Same for CTRL and ALT modifiers.
2579 */
2580 if (IS_SPECIAL(c))
2581 {
2582 for (i = 0; modifier_keys_table[i] != 0; i += MOD_KEYS_ENTRY_SIZE)
2583 if ( KEY2TERMCAP0(c) == (int)modifier_keys_table[i + 1]
2584 && (int)KEY2TERMCAP1(c) == (int)modifier_keys_table[i + 2])
2585 {
2586 modifiers |= modifier_keys_table[i];
2587 c = TERMCAP2KEY(modifier_keys_table[i + 3],
2588 modifier_keys_table[i + 4]);
2589 break;
2590 }
2591 }
2592
2593 /* try to find the key in the special key table */
2594 table_idx = find_special_key_in_table(c);
2595
2596 /*
2597 * When not a known special key, and not a printable character, try to
2598 * extract modifiers.
2599 */
2600 if (c > 0
2601#ifdef FEAT_MBYTE
2602 && (*mb_char2len)(c) == 1
2603#endif
2604 )
2605 {
2606 if (table_idx < 0
2607 && (!vim_isprintc(c) || (c & 0x7f) == ' ')
2608 && (c & 0x80))
2609 {
2610 c &= 0x7f;
2611 modifiers |= MOD_MASK_ALT;
2612 /* try again, to find the un-alted key in the special key table */
2613 table_idx = find_special_key_in_table(c);
2614 }
2615 if (table_idx < 0 && !vim_isprintc(c) && c < ' ')
2616 {
2617#ifdef EBCDIC
2618 c = CtrlChar(c);
2619#else
2620 c += '@';
2621#endif
2622 modifiers |= MOD_MASK_CTRL;
2623 }
2624 }
2625
2626 /* translate the modifier into a string */
2627 for (i = 0; mod_mask_table[i].name != 'A'; i++)
2628 if ((modifiers & mod_mask_table[i].mod_mask)
2629 == mod_mask_table[i].mod_flag)
2630 {
2631 string[idx++] = mod_mask_table[i].name;
2632 string[idx++] = (char_u)'-';
2633 }
2634
2635 if (table_idx < 0) /* unknown special key, may output t_xx */
2636 {
2637 if (IS_SPECIAL(c))
2638 {
2639 string[idx++] = 't';
2640 string[idx++] = '_';
2641 string[idx++] = KEY2TERMCAP0(c);
2642 string[idx++] = KEY2TERMCAP1(c);
2643 }
2644 /* Not a special key, only modifiers, output directly */
2645 else
2646 {
2647#ifdef FEAT_MBYTE
2648 if (has_mbyte && (*mb_char2len)(c) > 1)
2649 idx += (*mb_char2bytes)(c, string + idx);
2650 else
2651#endif
2652 if (vim_isprintc(c))
2653 string[idx++] = c;
2654 else
2655 {
2656 s = transchar(c);
2657 while (*s)
2658 string[idx++] = *s++;
2659 }
2660 }
2661 }
2662 else /* use name of special key */
2663 {
2664 STRCPY(string + idx, key_names_table[table_idx].name);
2665 idx = (int)STRLEN(string);
2666 }
2667 string[idx++] = '>';
2668 string[idx] = NUL;
2669 return string;
2670}
2671
2672/*
2673 * Try translating a <> name at (*srcp)[] to dst[].
2674 * Return the number of characters added to dst[], zero for no match.
2675 * If there is a match, srcp is advanced to after the <> name.
2676 * dst[] must be big enough to hold the result (up to six characters)!
2677 */
2678 int
2679trans_special(srcp, dst, keycode)
2680 char_u **srcp;
2681 char_u *dst;
2682 int keycode; /* prefer key code, e.g. K_DEL instead of DEL */
2683{
2684 int modifiers = 0;
2685 int key;
2686 int dlen = 0;
2687
Bram Moolenaar2a8ced02008-12-24 11:54:31 +00002688 key = find_special_key(srcp, &modifiers, keycode, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002689 if (key == 0)
2690 return 0;
2691
2692 /* Put the appropriate modifier in a string */
2693 if (modifiers != 0)
2694 {
2695 dst[dlen++] = K_SPECIAL;
2696 dst[dlen++] = KS_MODIFIER;
2697 dst[dlen++] = modifiers;
2698 }
2699
2700 if (IS_SPECIAL(key))
2701 {
2702 dst[dlen++] = K_SPECIAL;
2703 dst[dlen++] = KEY2TERMCAP0(key);
2704 dst[dlen++] = KEY2TERMCAP1(key);
2705 }
2706#ifdef FEAT_MBYTE
2707 else if (has_mbyte && !keycode)
2708 dlen += (*mb_char2bytes)(key, dst + dlen);
2709#endif
2710 else if (keycode)
2711 dlen = (int)(add_char2buf(key, dst + dlen) - dst);
2712 else
2713 dst[dlen++] = key;
2714
2715 return dlen;
2716}
2717
2718/*
2719 * Try translating a <> name at (*srcp)[], return the key and modifiers.
2720 * srcp is advanced to after the <> name.
2721 * returns 0 if there is no match.
2722 */
2723 int
Bram Moolenaar2a8ced02008-12-24 11:54:31 +00002724find_special_key(srcp, modp, keycode, keep_x_key)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002725 char_u **srcp;
2726 int *modp;
Bram Moolenaar2a8ced02008-12-24 11:54:31 +00002727 int keycode; /* prefer key code, e.g. K_DEL instead of DEL */
2728 int keep_x_key; /* don't translate xHome to Home key */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002729{
2730 char_u *last_dash;
2731 char_u *end_of_name;
2732 char_u *src;
2733 char_u *bp;
2734 int modifiers;
2735 int bit;
2736 int key;
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00002737 unsigned long n;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002738
2739 src = *srcp;
2740 if (src[0] != '<')
2741 return 0;
2742
2743 /* Find end of modifier list */
2744 last_dash = src;
2745 for (bp = src + 1; *bp == '-' || vim_isIDc(*bp); bp++)
2746 {
2747 if (*bp == '-')
2748 {
2749 last_dash = bp;
2750 if (bp[1] != NUL && bp[2] == '>')
2751 ++bp; /* anything accepted, like <C-?> */
2752 }
2753 if (bp[0] == 't' && bp[1] == '_' && bp[2] && bp[3])
2754 bp += 3; /* skip t_xx, xx may be '-' or '>' */
2755 }
2756
2757 if (*bp == '>') /* found matching '>' */
2758 {
2759 end_of_name = bp + 1;
2760
2761 if (STRNICMP(src + 1, "char-", 5) == 0 && VIM_ISDIGIT(src[6]))
2762 {
2763 /* <Char-123> or <Char-033> or <Char-0x33> */
2764 vim_str2nr(src + 6, NULL, NULL, TRUE, TRUE, NULL, &n);
2765 *modp = 0;
2766 *srcp = end_of_name;
2767 return (int)n;
2768 }
2769
2770 /* Which modifiers are given? */
2771 modifiers = 0x0;
2772 for (bp = src + 1; bp < last_dash; bp++)
2773 {
2774 if (*bp != '-')
2775 {
2776 bit = name_to_mod_mask(*bp);
2777 if (bit == 0x0)
2778 break; /* Illegal modifier name */
2779 modifiers |= bit;
2780 }
2781 }
2782
2783 /*
2784 * Legal modifier name.
2785 */
2786 if (bp >= last_dash)
2787 {
2788 /*
2789 * Modifier with single letter, or special key name.
2790 */
2791 if (modifiers != 0 && last_dash[2] == '>')
2792 key = last_dash[1];
2793 else
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002794 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00002795 key = get_special_key_code(last_dash + 1);
Bram Moolenaar2a8ced02008-12-24 11:54:31 +00002796 if (!keep_x_key)
2797 key = handle_x_keys(key);
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002798 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002799
2800 /*
2801 * get_special_key_code() may return NUL for invalid
2802 * special key name.
2803 */
2804 if (key != NUL)
2805 {
2806 /*
2807 * Only use a modifier when there is no special key code that
2808 * includes the modifier.
2809 */
2810 key = simplify_key(key, &modifiers);
2811
2812 if (!keycode)
2813 {
2814 /* don't want keycode, use single byte code */
2815 if (key == K_BS)
2816 key = BS;
2817 else if (key == K_DEL || key == K_KDEL)
2818 key = DEL;
2819 }
2820
2821 /*
2822 * Normal Key with modifier: Try to make a single byte code.
2823 */
2824 if (!IS_SPECIAL(key))
2825 key = extract_modifiers(key, &modifiers);
2826
2827 *modp = modifiers;
2828 *srcp = end_of_name;
2829 return key;
2830 }
2831 }
2832 }
2833 return 0;
2834}
2835
2836/*
2837 * Try to include modifiers in the key.
2838 * Changes "Shift-a" to 'A', "Alt-A" to 0xc0, etc.
2839 */
2840 int
2841extract_modifiers(key, modp)
2842 int key;
2843 int *modp;
2844{
2845 int modifiers = *modp;
2846
2847#ifdef MACOS
2848 /* Command-key really special, No fancynest */
2849 if (!(modifiers & MOD_MASK_CMD))
2850#endif
2851 if ((modifiers & MOD_MASK_SHIFT) && ASCII_ISALPHA(key))
2852 {
2853 key = TOUPPER_ASC(key);
2854 modifiers &= ~MOD_MASK_SHIFT;
2855 }
2856 if ((modifiers & MOD_MASK_CTRL)
2857#ifdef EBCDIC
2858 /* * TODO: EBCDIC Better use:
2859 * && (Ctrl_chr(key) || key == '?')
2860 * ??? */
2861 && strchr("?@ABCDEFGHIJKLMNOPQRSTUVWXYZ[\\]^_", key)
2862 != NULL
2863#else
2864 && ((key >= '?' && key <= '_') || ASCII_ISALPHA(key))
2865#endif
2866 )
2867 {
2868 key = Ctrl_chr(key);
2869 modifiers &= ~MOD_MASK_CTRL;
2870 /* <C-@> is <Nul> */
2871 if (key == 0)
2872 key = K_ZERO;
2873 }
2874#ifdef MACOS
2875 /* Command-key really special, No fancynest */
2876 if (!(modifiers & MOD_MASK_CMD))
2877#endif
2878 if ((modifiers & MOD_MASK_ALT) && key < 0x80
2879#ifdef FEAT_MBYTE
2880 && !enc_dbcs /* avoid creating a lead byte */
2881#endif
2882 )
2883 {
2884 key |= 0x80;
2885 modifiers &= ~MOD_MASK_ALT; /* remove the META modifier */
2886 }
2887
2888 *modp = modifiers;
2889 return key;
2890}
2891
2892/*
2893 * Try to find key "c" in the special key table.
2894 * Return the index when found, -1 when not found.
2895 */
2896 int
2897find_special_key_in_table(c)
2898 int c;
2899{
2900 int i;
2901
2902 for (i = 0; key_names_table[i].name != NULL; i++)
2903 if (c == key_names_table[i].key)
2904 break;
2905 if (key_names_table[i].name == NULL)
2906 i = -1;
2907 return i;
2908}
2909
2910/*
2911 * Find the special key with the given name (the given string does not have to
2912 * end with NUL, the name is assumed to end before the first non-idchar).
2913 * If the name starts with "t_" the next two characters are interpreted as a
2914 * termcap name.
2915 * Return the key code, or 0 if not found.
2916 */
2917 int
2918get_special_key_code(name)
2919 char_u *name;
2920{
2921 char_u *table_name;
2922 char_u string[3];
2923 int i, j;
2924
2925 /*
2926 * If it's <t_xx> we get the code for xx from the termcap
2927 */
2928 if (name[0] == 't' && name[1] == '_' && name[2] != NUL && name[3] != NUL)
2929 {
2930 string[0] = name[2];
2931 string[1] = name[3];
2932 string[2] = NUL;
2933 if (add_termcap_entry(string, FALSE) == OK)
2934 return TERMCAP2KEY(name[2], name[3]);
2935 }
2936 else
2937 for (i = 0; key_names_table[i].name != NULL; i++)
2938 {
2939 table_name = key_names_table[i].name;
2940 for (j = 0; vim_isIDc(name[j]) && table_name[j] != NUL; j++)
2941 if (TOLOWER_ASC(table_name[j]) != TOLOWER_ASC(name[j]))
2942 break;
2943 if (!vim_isIDc(name[j]) && table_name[j] == NUL)
2944 return key_names_table[i].key;
2945 }
2946 return 0;
2947}
2948
Bram Moolenaar05bb9532008-07-04 09:44:11 +00002949#if defined(FEAT_CMDL_COMPL) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002950 char_u *
2951get_key_name(i)
2952 int i;
2953{
Bram Moolenaare1fbddc2009-05-16 19:07:03 +00002954 if (i >= (int)KEY_NAMES_TABLE_LEN)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002955 return NULL;
2956 return key_names_table[i].name;
2957}
2958#endif
2959
Bram Moolenaar05bb9532008-07-04 09:44:11 +00002960#if defined(FEAT_MOUSE) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002961/*
2962 * Look up the given mouse code to return the relevant information in the other
2963 * arguments. Return which button is down or was released.
2964 */
2965 int
2966get_mouse_button(code, is_click, is_drag)
2967 int code;
2968 int *is_click;
2969 int *is_drag;
2970{
2971 int i;
2972
2973 for (i = 0; mouse_table[i].pseudo_code; i++)
2974 if (code == mouse_table[i].pseudo_code)
2975 {
2976 *is_click = mouse_table[i].is_click;
2977 *is_drag = mouse_table[i].is_drag;
2978 return mouse_table[i].button;
2979 }
2980 return 0; /* Shouldn't get here */
2981}
2982
2983/*
2984 * Return the appropriate pseudo mouse event token (KE_LEFTMOUSE etc) based on
2985 * the given information about which mouse button is down, and whether the
2986 * mouse was clicked, dragged or released.
2987 */
2988 int
2989get_pseudo_mouse_code(button, is_click, is_drag)
2990 int button; /* eg MOUSE_LEFT */
2991 int is_click;
2992 int is_drag;
2993{
2994 int i;
2995
2996 for (i = 0; mouse_table[i].pseudo_code; i++)
2997 if (button == mouse_table[i].button
2998 && is_click == mouse_table[i].is_click
2999 && is_drag == mouse_table[i].is_drag)
3000 {
3001#ifdef FEAT_GUI
Bram Moolenaarc91506a2005-04-24 22:04:21 +00003002 /* Trick: a non mappable left click and release has mouse_col -1
3003 * or added MOUSE_COLOFF. Used for 'mousefocus' in
3004 * gui_mouse_moved() */
3005 if (mouse_col < 0 || mouse_col > MOUSE_COLOFF)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003006 {
Bram Moolenaarc91506a2005-04-24 22:04:21 +00003007 if (mouse_col < 0)
3008 mouse_col = 0;
3009 else
3010 mouse_col -= MOUSE_COLOFF;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003011 if (mouse_table[i].pseudo_code == (int)KE_LEFTMOUSE)
3012 return (int)KE_LEFTMOUSE_NM;
3013 if (mouse_table[i].pseudo_code == (int)KE_LEFTRELEASE)
3014 return (int)KE_LEFTRELEASE_NM;
3015 }
3016#endif
3017 return mouse_table[i].pseudo_code;
3018 }
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00003019 return (int)KE_IGNORE; /* not recognized, ignore it */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003020}
3021#endif /* FEAT_MOUSE */
3022
3023/*
3024 * Return the current end-of-line type: EOL_DOS, EOL_UNIX or EOL_MAC.
3025 */
3026 int
3027get_fileformat(buf)
3028 buf_T *buf;
3029{
3030 int c = *buf->b_p_ff;
3031
3032 if (buf->b_p_bin || c == 'u')
3033 return EOL_UNIX;
3034 if (c == 'm')
3035 return EOL_MAC;
3036 return EOL_DOS;
3037}
3038
3039/*
3040 * Like get_fileformat(), but override 'fileformat' with "p" for "++opt=val"
3041 * argument.
3042 */
3043 int
3044get_fileformat_force(buf, eap)
3045 buf_T *buf;
3046 exarg_T *eap; /* can be NULL! */
3047{
3048 int c;
3049
3050 if (eap != NULL && eap->force_ff != 0)
3051 c = eap->cmd[eap->force_ff];
3052 else
3053 {
3054 if ((eap != NULL && eap->force_bin != 0)
3055 ? (eap->force_bin == FORCE_BIN) : buf->b_p_bin)
3056 return EOL_UNIX;
3057 c = *buf->b_p_ff;
3058 }
3059 if (c == 'u')
3060 return EOL_UNIX;
3061 if (c == 'm')
3062 return EOL_MAC;
3063 return EOL_DOS;
3064}
3065
3066/*
3067 * Set the current end-of-line type to EOL_DOS, EOL_UNIX or EOL_MAC.
3068 * Sets both 'textmode' and 'fileformat'.
3069 * Note: Does _not_ set global value of 'textmode'!
3070 */
3071 void
3072set_fileformat(t, opt_flags)
3073 int t;
3074 int opt_flags; /* OPT_LOCAL and/or OPT_GLOBAL */
3075{
3076 char *p = NULL;
3077
3078 switch (t)
3079 {
3080 case EOL_DOS:
3081 p = FF_DOS;
3082 curbuf->b_p_tx = TRUE;
3083 break;
3084 case EOL_UNIX:
3085 p = FF_UNIX;
3086 curbuf->b_p_tx = FALSE;
3087 break;
3088 case EOL_MAC:
3089 p = FF_MAC;
3090 curbuf->b_p_tx = FALSE;
3091 break;
3092 }
3093 if (p != NULL)
3094 set_string_option_direct((char_u *)"ff", -1, (char_u *)p,
Bram Moolenaar5e3cb7e2006-02-27 23:58:35 +00003095 OPT_FREE | opt_flags, 0);
3096
Bram Moolenaar071d4272004-06-13 20:20:40 +00003097#ifdef FEAT_WINDOWS
Bram Moolenaarf740b292006-02-16 22:11:02 +00003098 /* This may cause the buffer to become (un)modified. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003099 check_status(curbuf);
Bram Moolenaar997fb4b2006-02-17 21:53:23 +00003100 redraw_tabline = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003101#endif
3102#ifdef FEAT_TITLE
3103 need_maketitle = TRUE; /* set window title later */
3104#endif
3105}
3106
3107/*
3108 * Return the default fileformat from 'fileformats'.
3109 */
3110 int
3111default_fileformat()
3112{
3113 switch (*p_ffs)
3114 {
3115 case 'm': return EOL_MAC;
3116 case 'd': return EOL_DOS;
3117 }
3118 return EOL_UNIX;
3119}
3120
3121/*
3122 * Call shell. Calls mch_call_shell, with 'shellxquote' added.
3123 */
3124 int
3125call_shell(cmd, opt)
3126 char_u *cmd;
3127 int opt;
3128{
3129 char_u *ncmd;
3130 int retval;
Bram Moolenaar05159a02005-02-26 23:04:13 +00003131#ifdef FEAT_PROFILE
3132 proftime_T wait_time;
3133#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003134
3135 if (p_verbose > 3)
3136 {
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00003137 verbose_enter();
Bram Moolenaar051b7822005-05-19 21:00:46 +00003138 smsg((char_u *)_("Calling shell to execute: \"%s\""),
Bram Moolenaar071d4272004-06-13 20:20:40 +00003139 cmd == NULL ? p_sh : cmd);
3140 out_char('\n');
3141 cursor_on();
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00003142 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003143 }
3144
Bram Moolenaar05159a02005-02-26 23:04:13 +00003145#ifdef FEAT_PROFILE
Bram Moolenaar01265852006-03-20 21:50:15 +00003146 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003147 prof_child_enter(&wait_time);
3148#endif
3149
Bram Moolenaar071d4272004-06-13 20:20:40 +00003150 if (*p_sh == NUL)
3151 {
3152 EMSG(_(e_shellempty));
3153 retval = -1;
3154 }
3155 else
3156 {
3157#ifdef FEAT_GUI_MSWIN
3158 /* Don't hide the pointer while executing a shell command. */
3159 gui_mch_mousehide(FALSE);
3160#endif
3161#ifdef FEAT_GUI
3162 ++hold_gui_events;
3163#endif
3164 /* The external command may update a tags file, clear cached tags. */
3165 tag_freematch();
3166
3167 if (cmd == NULL || *p_sxq == NUL)
3168 retval = mch_call_shell(cmd, opt);
3169 else
3170 {
3171 ncmd = alloc((unsigned)(STRLEN(cmd) + STRLEN(p_sxq) * 2 + 1));
3172 if (ncmd != NULL)
3173 {
3174 STRCPY(ncmd, p_sxq);
3175 STRCAT(ncmd, cmd);
3176 STRCAT(ncmd, p_sxq);
3177 retval = mch_call_shell(ncmd, opt);
3178 vim_free(ncmd);
3179 }
3180 else
3181 retval = -1;
3182 }
3183#ifdef FEAT_GUI
3184 --hold_gui_events;
3185#endif
3186 /*
3187 * Check the window size, in case it changed while executing the
3188 * external command.
3189 */
3190 shell_resized_check();
3191 }
3192
3193#ifdef FEAT_EVAL
3194 set_vim_var_nr(VV_SHELL_ERROR, (long)retval);
Bram Moolenaar05159a02005-02-26 23:04:13 +00003195# ifdef FEAT_PROFILE
Bram Moolenaar01265852006-03-20 21:50:15 +00003196 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003197 prof_child_exit(&wait_time);
3198# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003199#endif
3200
3201 return retval;
3202}
3203
3204/*
Bram Moolenaar01265852006-03-20 21:50:15 +00003205 * VISUAL, SELECTMODE and OP_PENDING State are never set, they are equal to
3206 * NORMAL State with a condition. This function returns the real State.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003207 */
3208 int
3209get_real_state()
3210{
3211 if (State & NORMAL)
3212 {
3213#ifdef FEAT_VISUAL
3214 if (VIsual_active)
Bram Moolenaar01265852006-03-20 21:50:15 +00003215 {
3216 if (VIsual_select)
3217 return SELECTMODE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003218 return VISUAL;
Bram Moolenaar01265852006-03-20 21:50:15 +00003219 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003220 else
3221#endif
3222 if (finish_op)
Bram Moolenaar01265852006-03-20 21:50:15 +00003223 return OP_PENDING;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003224 }
3225 return State;
3226}
3227
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003228#if defined(FEAT_MBYTE) || defined(PROTO)
3229/*
3230 * Return TRUE if "p" points to just after a path separator.
3231 * Take care of multi-byte characters.
3232 * "b" must point to the start of the file name
3233 */
3234 int
3235after_pathsep(b, p)
3236 char_u *b;
3237 char_u *p;
3238{
3239 return vim_ispathsep(p[-1])
3240 && (!has_mbyte || (*mb_head_off)(b, p - 1) == 0);
3241}
3242#endif
3243
3244/*
3245 * Return TRUE if file names "f1" and "f2" are in the same directory.
3246 * "f1" may be a short name, "f2" must be a full path.
3247 */
3248 int
3249same_directory(f1, f2)
3250 char_u *f1;
3251 char_u *f2;
3252{
3253 char_u ffname[MAXPATHL];
3254 char_u *t1;
3255 char_u *t2;
3256
3257 /* safety check */
3258 if (f1 == NULL || f2 == NULL)
3259 return FALSE;
3260
3261 (void)vim_FullName(f1, ffname, MAXPATHL, FALSE);
3262 t1 = gettail_sep(ffname);
3263 t2 = gettail_sep(f2);
3264 return (t1 - ffname == t2 - f2
3265 && pathcmp((char *)ffname, (char *)f2, (int)(t1 - ffname)) == 0);
3266}
3267
Bram Moolenaar071d4272004-06-13 20:20:40 +00003268#if defined(FEAT_SESSION) || defined(MSWIN) || defined(FEAT_GUI_MAC) \
Bram Moolenaar9372a112005-12-06 19:59:18 +00003269 || ((defined(FEAT_GUI_GTK)) \
Bram Moolenaar843ee412004-06-30 16:16:41 +00003270 && ( defined(FEAT_WINDOWS) || defined(FEAT_DND)) ) \
Bram Moolenaar071d4272004-06-13 20:20:40 +00003271 || defined(FEAT_SUN_WORKSHOP) || defined(FEAT_NETBEANS_INTG) \
3272 || defined(PROTO)
3273/*
3274 * Change to a file's directory.
3275 * Caller must call shorten_fnames()!
3276 * Return OK or FAIL.
3277 */
3278 int
3279vim_chdirfile(fname)
3280 char_u *fname;
3281{
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003282 char_u dir[MAXPATHL];
Bram Moolenaar071d4272004-06-13 20:20:40 +00003283
Bram Moolenaarbbebc852005-07-18 21:47:53 +00003284 vim_strncpy(dir, fname, MAXPATHL - 1);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003285 *gettail_sep(dir) = NUL;
3286 return mch_chdir((char *)dir) == 0 ? OK : FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003287}
3288#endif
3289
3290#if defined(STAT_IGNORES_SLASH) || defined(PROTO)
3291/*
3292 * Check if "name" ends in a slash and is not a directory.
3293 * Used for systems where stat() ignores a trailing slash on a file name.
3294 * The Vim code assumes a trailing slash is only ignored for a directory.
3295 */
3296 int
3297illegal_slash(name)
3298 char *name;
3299{
3300 if (name[0] == NUL)
3301 return FALSE; /* no file name is not illegal */
3302 if (name[strlen(name) - 1] != '/')
3303 return FALSE; /* no trailing slash */
3304 if (mch_isdir((char_u *)name))
3305 return FALSE; /* trailing slash for a directory */
3306 return TRUE;
3307}
3308#endif
3309
3310#if defined(CURSOR_SHAPE) || defined(PROTO)
3311
3312/*
3313 * Handling of cursor and mouse pointer shapes in various modes.
3314 */
3315
3316cursorentry_T shape_table[SHAPE_IDX_COUNT] =
3317{
3318 /* The values will be filled in from the 'guicursor' and 'mouseshape'
3319 * defaults when Vim starts.
3320 * Adjust the SHAPE_IDX_ defines when making changes! */
3321 {0, 0, 0, 700L, 400L, 250L, 0, 0, "n", SHAPE_CURSOR+SHAPE_MOUSE},
3322 {0, 0, 0, 700L, 400L, 250L, 0, 0, "v", SHAPE_CURSOR+SHAPE_MOUSE},
3323 {0, 0, 0, 700L, 400L, 250L, 0, 0, "i", SHAPE_CURSOR+SHAPE_MOUSE},
3324 {0, 0, 0, 700L, 400L, 250L, 0, 0, "r", SHAPE_CURSOR+SHAPE_MOUSE},
3325 {0, 0, 0, 700L, 400L, 250L, 0, 0, "c", SHAPE_CURSOR+SHAPE_MOUSE},
3326 {0, 0, 0, 700L, 400L, 250L, 0, 0, "ci", SHAPE_CURSOR+SHAPE_MOUSE},
3327 {0, 0, 0, 700L, 400L, 250L, 0, 0, "cr", SHAPE_CURSOR+SHAPE_MOUSE},
3328 {0, 0, 0, 700L, 400L, 250L, 0, 0, "o", SHAPE_CURSOR+SHAPE_MOUSE},
3329 {0, 0, 0, 700L, 400L, 250L, 0, 0, "ve", SHAPE_CURSOR+SHAPE_MOUSE},
3330 {0, 0, 0, 0L, 0L, 0L, 0, 0, "e", SHAPE_MOUSE},
3331 {0, 0, 0, 0L, 0L, 0L, 0, 0, "s", SHAPE_MOUSE},
3332 {0, 0, 0, 0L, 0L, 0L, 0, 0, "sd", SHAPE_MOUSE},
3333 {0, 0, 0, 0L, 0L, 0L, 0, 0, "vs", SHAPE_MOUSE},
3334 {0, 0, 0, 0L, 0L, 0L, 0, 0, "vd", SHAPE_MOUSE},
3335 {0, 0, 0, 0L, 0L, 0L, 0, 0, "m", SHAPE_MOUSE},
3336 {0, 0, 0, 0L, 0L, 0L, 0, 0, "ml", SHAPE_MOUSE},
3337 {0, 0, 0, 100L, 100L, 100L, 0, 0, "sm", SHAPE_CURSOR},
3338};
3339
3340#ifdef FEAT_MOUSESHAPE
3341/*
3342 * Table with names for mouse shapes. Keep in sync with all the tables for
3343 * mch_set_mouse_shape()!.
3344 */
3345static char * mshape_names[] =
3346{
3347 "arrow", /* default, must be the first one */
3348 "blank", /* hidden */
3349 "beam",
3350 "updown",
3351 "udsizing",
3352 "leftright",
3353 "lrsizing",
3354 "busy",
3355 "no",
3356 "crosshair",
3357 "hand1",
3358 "hand2",
3359 "pencil",
3360 "question",
3361 "rightup-arrow",
3362 "up-arrow",
3363 NULL
3364};
3365#endif
3366
3367/*
3368 * Parse the 'guicursor' option ("what" is SHAPE_CURSOR) or 'mouseshape'
3369 * ("what" is SHAPE_MOUSE).
3370 * Returns error message for an illegal option, NULL otherwise.
3371 */
3372 char_u *
3373parse_shape_opt(what)
3374 int what;
3375{
3376 char_u *modep;
3377 char_u *colonp;
3378 char_u *commap;
3379 char_u *slashp;
3380 char_u *p, *endp;
3381 int idx = 0; /* init for GCC */
3382 int all_idx;
3383 int len;
3384 int i;
3385 long n;
3386 int found_ve = FALSE; /* found "ve" flag */
3387 int round;
3388
3389 /*
3390 * First round: check for errors; second round: do it for real.
3391 */
3392 for (round = 1; round <= 2; ++round)
3393 {
3394 /*
3395 * Repeat for all comma separated parts.
3396 */
3397#ifdef FEAT_MOUSESHAPE
3398 if (what == SHAPE_MOUSE)
3399 modep = p_mouseshape;
3400 else
3401#endif
3402 modep = p_guicursor;
3403 while (*modep != NUL)
3404 {
3405 colonp = vim_strchr(modep, ':');
3406 if (colonp == NULL)
3407 return (char_u *)N_("E545: Missing colon");
3408 if (colonp == modep)
3409 return (char_u *)N_("E546: Illegal mode");
3410 commap = vim_strchr(modep, ',');
3411
3412 /*
3413 * Repeat for all mode's before the colon.
3414 * For the 'a' mode, we loop to handle all the modes.
3415 */
3416 all_idx = -1;
3417 while (modep < colonp || all_idx >= 0)
3418 {
3419 if (all_idx < 0)
3420 {
3421 /* Find the mode. */
3422 if (modep[1] == '-' || modep[1] == ':')
3423 len = 1;
3424 else
3425 len = 2;
3426 if (len == 1 && TOLOWER_ASC(modep[0]) == 'a')
3427 all_idx = SHAPE_IDX_COUNT - 1;
3428 else
3429 {
3430 for (idx = 0; idx < SHAPE_IDX_COUNT; ++idx)
3431 if (STRNICMP(modep, shape_table[idx].name, len)
3432 == 0)
3433 break;
3434 if (idx == SHAPE_IDX_COUNT
3435 || (shape_table[idx].used_for & what) == 0)
3436 return (char_u *)N_("E546: Illegal mode");
3437 if (len == 2 && modep[0] == 'v' && modep[1] == 'e')
3438 found_ve = TRUE;
3439 }
3440 modep += len + 1;
3441 }
3442
3443 if (all_idx >= 0)
3444 idx = all_idx--;
3445 else if (round == 2)
3446 {
3447#ifdef FEAT_MOUSESHAPE
3448 if (what == SHAPE_MOUSE)
3449 {
3450 /* Set the default, for the missing parts */
3451 shape_table[idx].mshape = 0;
3452 }
3453 else
3454#endif
3455 {
3456 /* Set the defaults, for the missing parts */
3457 shape_table[idx].shape = SHAPE_BLOCK;
3458 shape_table[idx].blinkwait = 700L;
3459 shape_table[idx].blinkon = 400L;
3460 shape_table[idx].blinkoff = 250L;
3461 }
3462 }
3463
3464 /* Parse the part after the colon */
3465 for (p = colonp + 1; *p && *p != ','; )
3466 {
3467#ifdef FEAT_MOUSESHAPE
3468 if (what == SHAPE_MOUSE)
3469 {
3470 for (i = 0; ; ++i)
3471 {
3472 if (mshape_names[i] == NULL)
3473 {
3474 if (!VIM_ISDIGIT(*p))
3475 return (char_u *)N_("E547: Illegal mouseshape");
3476 if (round == 2)
3477 shape_table[idx].mshape =
3478 getdigits(&p) + MSHAPE_NUMBERED;
3479 else
3480 (void)getdigits(&p);
3481 break;
3482 }
3483 len = (int)STRLEN(mshape_names[i]);
3484 if (STRNICMP(p, mshape_names[i], len) == 0)
3485 {
3486 if (round == 2)
3487 shape_table[idx].mshape = i;
3488 p += len;
3489 break;
3490 }
3491 }
3492 }
3493 else /* if (what == SHAPE_MOUSE) */
3494#endif
3495 {
3496 /*
3497 * First handle the ones with a number argument.
3498 */
3499 i = *p;
3500 len = 0;
3501 if (STRNICMP(p, "ver", 3) == 0)
3502 len = 3;
3503 else if (STRNICMP(p, "hor", 3) == 0)
3504 len = 3;
3505 else if (STRNICMP(p, "blinkwait", 9) == 0)
3506 len = 9;
3507 else if (STRNICMP(p, "blinkon", 7) == 0)
3508 len = 7;
3509 else if (STRNICMP(p, "blinkoff", 8) == 0)
3510 len = 8;
3511 if (len != 0)
3512 {
3513 p += len;
3514 if (!VIM_ISDIGIT(*p))
3515 return (char_u *)N_("E548: digit expected");
3516 n = getdigits(&p);
3517 if (len == 3) /* "ver" or "hor" */
3518 {
3519 if (n == 0)
3520 return (char_u *)N_("E549: Illegal percentage");
3521 if (round == 2)
3522 {
3523 if (TOLOWER_ASC(i) == 'v')
3524 shape_table[idx].shape = SHAPE_VER;
3525 else
3526 shape_table[idx].shape = SHAPE_HOR;
3527 shape_table[idx].percentage = n;
3528 }
3529 }
3530 else if (round == 2)
3531 {
3532 if (len == 9)
3533 shape_table[idx].blinkwait = n;
3534 else if (len == 7)
3535 shape_table[idx].blinkon = n;
3536 else
3537 shape_table[idx].blinkoff = n;
3538 }
3539 }
3540 else if (STRNICMP(p, "block", 5) == 0)
3541 {
3542 if (round == 2)
3543 shape_table[idx].shape = SHAPE_BLOCK;
3544 p += 5;
3545 }
3546 else /* must be a highlight group name then */
3547 {
3548 endp = vim_strchr(p, '-');
3549 if (commap == NULL) /* last part */
3550 {
3551 if (endp == NULL)
3552 endp = p + STRLEN(p); /* find end of part */
3553 }
3554 else if (endp > commap || endp == NULL)
3555 endp = commap;
3556 slashp = vim_strchr(p, '/');
3557 if (slashp != NULL && slashp < endp)
3558 {
3559 /* "group/langmap_group" */
3560 i = syn_check_group(p, (int)(slashp - p));
3561 p = slashp + 1;
3562 }
3563 if (round == 2)
3564 {
3565 shape_table[idx].id = syn_check_group(p,
3566 (int)(endp - p));
3567 shape_table[idx].id_lm = shape_table[idx].id;
3568 if (slashp != NULL && slashp < endp)
3569 shape_table[idx].id = i;
3570 }
3571 p = endp;
3572 }
3573 } /* if (what != SHAPE_MOUSE) */
3574
3575 if (*p == '-')
3576 ++p;
3577 }
3578 }
3579 modep = p;
3580 if (*modep == ',')
3581 ++modep;
3582 }
3583 }
3584
3585 /* If the 's' flag is not given, use the 'v' cursor for 's' */
3586 if (!found_ve)
3587 {
3588#ifdef FEAT_MOUSESHAPE
3589 if (what == SHAPE_MOUSE)
3590 {
3591 shape_table[SHAPE_IDX_VE].mshape = shape_table[SHAPE_IDX_V].mshape;
3592 }
3593 else
3594#endif
3595 {
3596 shape_table[SHAPE_IDX_VE].shape = shape_table[SHAPE_IDX_V].shape;
3597 shape_table[SHAPE_IDX_VE].percentage =
3598 shape_table[SHAPE_IDX_V].percentage;
3599 shape_table[SHAPE_IDX_VE].blinkwait =
3600 shape_table[SHAPE_IDX_V].blinkwait;
3601 shape_table[SHAPE_IDX_VE].blinkon =
3602 shape_table[SHAPE_IDX_V].blinkon;
3603 shape_table[SHAPE_IDX_VE].blinkoff =
3604 shape_table[SHAPE_IDX_V].blinkoff;
3605 shape_table[SHAPE_IDX_VE].id = shape_table[SHAPE_IDX_V].id;
3606 shape_table[SHAPE_IDX_VE].id_lm = shape_table[SHAPE_IDX_V].id_lm;
3607 }
3608 }
3609
3610 return NULL;
3611}
3612
Bram Moolenaarac6e65f2005-08-29 22:25:38 +00003613# if defined(MCH_CURSOR_SHAPE) || defined(FEAT_GUI) \
3614 || defined(FEAT_MOUSESHAPE) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003615/*
3616 * Return the index into shape_table[] for the current mode.
3617 * When "mouse" is TRUE, consider indexes valid for the mouse pointer.
3618 */
3619 int
3620get_shape_idx(mouse)
3621 int mouse;
3622{
3623#ifdef FEAT_MOUSESHAPE
3624 if (mouse && (State == HITRETURN || State == ASKMORE))
3625 {
3626# ifdef FEAT_GUI
Bram Moolenaar9588a0f2005-01-08 21:45:39 +00003627 int x, y;
3628 gui_mch_getmouse(&x, &y);
3629 if (Y_2_ROW(y) == Rows - 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003630 return SHAPE_IDX_MOREL;
3631# endif
3632 return SHAPE_IDX_MORE;
3633 }
3634 if (mouse && drag_status_line)
3635 return SHAPE_IDX_SDRAG;
3636# ifdef FEAT_VERTSPLIT
3637 if (mouse && drag_sep_line)
3638 return SHAPE_IDX_VDRAG;
3639# endif
3640#endif
3641 if (!mouse && State == SHOWMATCH)
3642 return SHAPE_IDX_SM;
3643#ifdef FEAT_VREPLACE
3644 if (State & VREPLACE_FLAG)
3645 return SHAPE_IDX_R;
3646#endif
3647 if (State & REPLACE_FLAG)
3648 return SHAPE_IDX_R;
3649 if (State & INSERT)
3650 return SHAPE_IDX_I;
3651 if (State & CMDLINE)
3652 {
3653 if (cmdline_at_end())
3654 return SHAPE_IDX_C;
3655 if (cmdline_overstrike())
3656 return SHAPE_IDX_CR;
3657 return SHAPE_IDX_CI;
3658 }
3659 if (finish_op)
3660 return SHAPE_IDX_O;
3661#ifdef FEAT_VISUAL
3662 if (VIsual_active)
3663 {
3664 if (*p_sel == 'e')
3665 return SHAPE_IDX_VE;
3666 else
3667 return SHAPE_IDX_V;
3668 }
3669#endif
3670 return SHAPE_IDX_N;
3671}
Bram Moolenaarac6e65f2005-08-29 22:25:38 +00003672#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003673
3674# if defined(FEAT_MOUSESHAPE) || defined(PROTO)
3675static int old_mouse_shape = 0;
3676
3677/*
3678 * Set the mouse shape:
3679 * If "shape" is -1, use shape depending on the current mode,
3680 * depending on the current state.
3681 * If "shape" is -2, only update the shape when it's CLINE or STATUS (used
3682 * when the mouse moves off the status or command line).
3683 */
3684 void
3685update_mouseshape(shape_idx)
3686 int shape_idx;
3687{
3688 int new_mouse_shape;
3689
3690 /* Only works in GUI mode. */
Bram Moolenaar6bb68362005-03-22 23:03:44 +00003691 if (!gui.in_use || gui.starting)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003692 return;
3693
3694 /* Postpone the updating when more is to come. Speeds up executing of
3695 * mappings. */
3696 if (shape_idx == -1 && char_avail())
3697 {
3698 postponed_mouseshape = TRUE;
3699 return;
3700 }
3701
Bram Moolenaar14716812006-05-04 21:54:08 +00003702 /* When ignoring the mouse don't change shape on the statusline. */
3703 if (*p_mouse == NUL
3704 && (shape_idx == SHAPE_IDX_CLINE
3705 || shape_idx == SHAPE_IDX_STATUS
3706 || shape_idx == SHAPE_IDX_VSEP))
3707 shape_idx = -2;
3708
Bram Moolenaar071d4272004-06-13 20:20:40 +00003709 if (shape_idx == -2
3710 && old_mouse_shape != shape_table[SHAPE_IDX_CLINE].mshape
3711 && old_mouse_shape != shape_table[SHAPE_IDX_STATUS].mshape
3712 && old_mouse_shape != shape_table[SHAPE_IDX_VSEP].mshape)
3713 return;
3714 if (shape_idx < 0)
3715 new_mouse_shape = shape_table[get_shape_idx(TRUE)].mshape;
3716 else
3717 new_mouse_shape = shape_table[shape_idx].mshape;
3718 if (new_mouse_shape != old_mouse_shape)
3719 {
3720 mch_set_mouse_shape(new_mouse_shape);
3721 old_mouse_shape = new_mouse_shape;
3722 }
3723 postponed_mouseshape = FALSE;
3724}
3725# endif
3726
3727#endif /* CURSOR_SHAPE */
3728
3729
3730#ifdef FEAT_CRYPT
3731/*
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00003732 * Optional encryption support.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003733 * Mohsin Ahmed, mosh@sasi.com, 98-09-24
3734 * Based on zip/crypt sources.
3735 *
3736 * NOTE FOR USA: Since 2000 exporting this code from the USA is allowed to
3737 * most countries. There are a few exceptions, but that still should not be a
3738 * problem since this code was originally created in Europe and India.
Bram Moolenaar40e6a712010-05-16 22:32:54 +02003739 *
3740 * Blowfish addition originally made by Mohsin Ahmed,
3741 * http://www.cs.albany.edu/~mosh 2010-03-14
3742 * Based on blowfish by Bruce Schneier (http://www.schneier.com/blowfish.html)
3743 * and sha256 by Christophe Devine.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003744 */
3745
3746/* from zip.h */
3747
3748typedef unsigned short ush; /* unsigned 16-bit value */
3749typedef unsigned long ulg; /* unsigned 32-bit value */
3750
3751static void make_crc_tab __ARGS((void));
3752
Bram Moolenaard6f676d2005-06-01 21:51:55 +00003753static ulg crc_32_tab[256];
Bram Moolenaar071d4272004-06-13 20:20:40 +00003754
3755/*
3756 * Fill the CRC table.
3757 */
3758 static void
3759make_crc_tab()
3760{
3761 ulg s,t,v;
3762 static int done = FALSE;
3763
3764 if (done)
3765 return;
3766 for (t = 0; t < 256; t++)
3767 {
3768 v = t;
3769 for (s = 0; s < 8; s++)
3770 v = (v >> 1) ^ ((v & 1) * (ulg)0xedb88320L);
3771 crc_32_tab[t] = v;
3772 }
3773 done = TRUE;
3774}
3775
3776#define CRC32(c, b) (crc_32_tab[((int)(c) ^ (b)) & 0xff] ^ ((c) >> 8))
3777
Bram Moolenaar071d4272004-06-13 20:20:40 +00003778static ulg keys[3]; /* keys defining the pseudo-random sequence */
3779
3780/*
Bram Moolenaar04c9baf2010-06-01 23:37:39 +02003781 * Return the next byte in the pseudo-random sequence.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003782 */
Bram Moolenaar04c9baf2010-06-01 23:37:39 +02003783#define DECRYPT_BYTE_ZIP(t) { \
3784 ush temp; \
3785 \
3786 temp = (ush)keys[2] | 2; \
3787 t = (int)(((unsigned)(temp * (temp ^ 1)) >> 8) & 0xff); \
Bram Moolenaar071d4272004-06-13 20:20:40 +00003788}
3789
3790/*
Bram Moolenaar04c9baf2010-06-01 23:37:39 +02003791 * Update the encryption keys with the next byte of plain text.
3792 */
3793#define UPDATE_KEYS_ZIP(c) { \
3794 keys[0] = CRC32(keys[0], (c)); \
3795 keys[1] += keys[0] & 0xff; \
3796 keys[1] = keys[1] * 134775813L + 1; \
3797 keys[2] = CRC32(keys[2], (int)(keys[1] >> 24)); \
3798}
3799
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02003800static int crypt_busy = 0;
3801static ulg saved_keys[3];
3802static int saved_crypt_method;
3803
3804/*
Bram Moolenaar49771f42010-07-20 17:32:38 +02003805 * Return int value for crypt method string:
3806 * 0 for "zip", the old method. Also for any non-valid value.
3807 * 1 for "blowfish".
3808 */
3809 int
3810crypt_method_from_string(s)
3811 char_u *s;
3812{
3813 return *s == 'b' ? 1 : 0;
3814}
3815
3816/*
3817 * Get the crypt method for buffer "buf" as a number.
3818 */
3819 int
3820get_crypt_method(buf)
3821 buf_T *buf;
3822{
3823 return crypt_method_from_string(*buf->b_p_cm == NUL ? p_cm : buf->b_p_cm);
3824}
3825
3826/*
3827 * Set the crypt method for buffer "buf" to "method" using the int value as
3828 * returned by crypt_method_from_string().
3829 */
3830 void
3831set_crypt_method(buf, method)
3832 buf_T *buf;
3833 int method;
3834{
3835 free_string_option(buf->b_p_cm);
3836 buf->b_p_cm = vim_strsave((char_u *)(method == 0 ? "zip" : "blowfish"));
3837}
3838
3839/*
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02003840 * Prepare for initializing encryption. If already doing encryption then save
3841 * the state.
Bram Moolenaar2f1e0502010-08-13 11:18:02 +02003842 * Must always be called symmetrically with crypt_pop_state().
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02003843 */
3844 void
3845crypt_push_state()
3846{
3847 if (crypt_busy == 1)
3848 {
3849 /* save the state */
3850 if (use_crypt_method == 0)
3851 {
3852 saved_keys[0] = keys[0];
3853 saved_keys[1] = keys[1];
3854 saved_keys[2] = keys[2];
3855 }
3856 else
3857 bf_crypt_save();
3858 saved_crypt_method = use_crypt_method;
3859 }
3860 else if (crypt_busy > 1)
3861 EMSG2(_(e_intern2), "crypt_push_state()");
3862 ++crypt_busy;
3863}
3864
3865/*
3866 * End encryption. If doing encryption before crypt_push_state() then restore
3867 * the saved state.
Bram Moolenaar2f1e0502010-08-13 11:18:02 +02003868 * Must always be called symmetrically with crypt_push_state().
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02003869 */
3870 void
3871crypt_pop_state()
3872{
3873 --crypt_busy;
3874 if (crypt_busy == 1)
3875 {
3876 use_crypt_method = saved_crypt_method;
3877 if (use_crypt_method == 0)
3878 {
3879 keys[0] = saved_keys[0];
3880 keys[1] = saved_keys[1];
3881 keys[2] = saved_keys[2];
3882 }
3883 else
3884 bf_crypt_restore();
3885 }
3886}
3887
Bram Moolenaar04c9baf2010-06-01 23:37:39 +02003888/*
3889 * Encrypt "from[len]" into "to[len]".
3890 * "from" and "to" can be equal to encrypt in place.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003891 */
Bram Moolenaar40e6a712010-05-16 22:32:54 +02003892 void
Bram Moolenaar04c9baf2010-06-01 23:37:39 +02003893crypt_encode(from, len, to)
3894 char_u *from;
3895 size_t len;
3896 char_u *to;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003897{
Bram Moolenaar04c9baf2010-06-01 23:37:39 +02003898 size_t i;
3899 int ztemp, t;
3900
3901 if (use_crypt_method == 0)
3902 for (i = 0; i < len; ++i)
3903 {
3904 ztemp = from[i];
3905 DECRYPT_BYTE_ZIP(t);
3906 UPDATE_KEYS_ZIP(ztemp);
3907 to[i] = t ^ ztemp;
3908 }
Bram Moolenaar40e6a712010-05-16 22:32:54 +02003909 else
Bram Moolenaarbbd6afe2010-06-02 20:32:23 +02003910 bf_crypt_encode(from, len, to);
Bram Moolenaar04c9baf2010-06-01 23:37:39 +02003911}
3912
3913/*
3914 * Decrypt "ptr[len]" in place.
3915 */
3916 void
3917crypt_decode(ptr, len)
3918 char_u *ptr;
3919 long len;
3920{
3921 char_u *p;
3922
3923 if (use_crypt_method == 0)
3924 for (p = ptr; p < ptr + len; ++p)
3925 {
3926 ush temp;
3927
3928 temp = (ush)keys[2] | 2;
3929 temp = (int)(((unsigned)(temp * (temp ^ 1)) >> 8) & 0xff);
3930 UPDATE_KEYS_ZIP(*p ^= temp);
3931 }
3932 else
Bram Moolenaarbbd6afe2010-06-02 20:32:23 +02003933 bf_crypt_decode(ptr, len);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003934}
3935
3936/*
3937 * Initialize the encryption keys and the random header according to
3938 * the given password.
3939 * If "passwd" is NULL or empty, don't do anything.
3940 */
3941 void
3942crypt_init_keys(passwd)
3943 char_u *passwd; /* password string with which to modify keys */
3944{
3945 if (passwd != NULL && *passwd != NUL)
3946 {
Bram Moolenaar04c9baf2010-06-01 23:37:39 +02003947 if (use_crypt_method == 0)
Bram Moolenaarbbd6afe2010-06-02 20:32:23 +02003948 {
3949 char_u *p;
3950
3951 make_crc_tab();
3952 keys[0] = 305419896L;
3953 keys[1] = 591751049L;
3954 keys[2] = 878082192L;
3955 for (p = passwd; *p!= NUL; ++p)
Bram Moolenaar04c9baf2010-06-01 23:37:39 +02003956 {
Bram Moolenaarbbd6afe2010-06-02 20:32:23 +02003957 UPDATE_KEYS_ZIP((int)*p);
Bram Moolenaar04c9baf2010-06-01 23:37:39 +02003958 }
Bram Moolenaarbbd6afe2010-06-02 20:32:23 +02003959 }
Bram Moolenaar04c9baf2010-06-01 23:37:39 +02003960 else
Bram Moolenaarbbd6afe2010-06-02 20:32:23 +02003961 bf_crypt_init_keys(passwd);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003962 }
3963}
3964
3965/*
Bram Moolenaar40e6a712010-05-16 22:32:54 +02003966 * Free an allocated crypt key. Clear the text to make sure it doesn't stay
3967 * in memory anywhere.
3968 */
3969 void
3970free_crypt_key(key)
3971 char_u *key;
3972{
3973 char_u *p;
3974
3975 if (key != NULL)
3976 {
3977 for (p = key; *p != NUL; ++p)
Bram Moolenaarbe18d102010-05-22 21:37:53 +02003978 *p = 0;
Bram Moolenaar40e6a712010-05-16 22:32:54 +02003979 vim_free(key);
3980 }
3981}
3982
3983/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00003984 * Ask the user for a crypt key.
Bram Moolenaar40e6a712010-05-16 22:32:54 +02003985 * When "store" is TRUE, the new key is stored in the 'key' option, and the
Bram Moolenaar071d4272004-06-13 20:20:40 +00003986 * 'key' option value is returned: Don't free it.
3987 * When "store" is FALSE, the typed key is returned in allocated memory.
3988 * Returns NULL on failure.
3989 */
3990 char_u *
3991get_crypt_key(store, twice)
3992 int store;
3993 int twice; /* Ask for the key twice. */
3994{
3995 char_u *p1, *p2 = NULL;
3996 int round;
3997
3998 for (round = 0; ; ++round)
3999 {
4000 cmdline_star = TRUE;
4001 cmdline_row = msg_row;
4002 p1 = getcmdline_prompt(NUL, round == 0
4003 ? (char_u *)_("Enter encryption key: ")
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +00004004 : (char_u *)_("Enter same key again: "), 0, EXPAND_NOTHING,
4005 NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004006 cmdline_star = FALSE;
4007
4008 if (p1 == NULL)
4009 break;
4010
4011 if (round == twice)
4012 {
4013 if (p2 != NULL && STRCMP(p1, p2) != 0)
4014 {
4015 MSG(_("Keys don't match!"));
Bram Moolenaar40e6a712010-05-16 22:32:54 +02004016 free_crypt_key(p1);
4017 free_crypt_key(p2);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004018 p2 = NULL;
4019 round = -1; /* do it again */
4020 continue;
4021 }
Bram Moolenaar40e6a712010-05-16 22:32:54 +02004022
Bram Moolenaar071d4272004-06-13 20:20:40 +00004023 if (store)
4024 {
4025 set_option_value((char_u *)"key", 0L, p1, OPT_LOCAL);
Bram Moolenaar40e6a712010-05-16 22:32:54 +02004026 free_crypt_key(p1);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004027 p1 = curbuf->b_p_key;
4028 }
4029 break;
4030 }
4031 p2 = p1;
4032 }
4033
4034 /* since the user typed this, no need to wait for return */
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02004035 if (msg_didout)
4036 msg_putchar('\n');
Bram Moolenaare4ce65d2010-08-04 20:12:32 +02004037 need_wait_return = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004038 msg_didout = FALSE;
4039
Bram Moolenaar40e6a712010-05-16 22:32:54 +02004040 free_crypt_key(p2);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004041 return p1;
4042}
4043
4044#endif /* FEAT_CRYPT */
4045
4046/* TODO: make some #ifdef for this */
4047/*--------[ file searching ]-------------------------------------------------*/
4048/*
4049 * File searching functions for 'path', 'tags' and 'cdpath' options.
4050 * External visible functions:
4051 * vim_findfile_init() creates/initialises the search context
4052 * vim_findfile_free_visited() free list of visited files/dirs of search
4053 * context
4054 * vim_findfile() find a file in the search context
4055 * vim_findfile_cleanup() cleanup/free search context created by
4056 * vim_findfile_init()
4057 *
4058 * All static functions and variables start with 'ff_'
4059 *
4060 * In general it works like this:
4061 * First you create yourself a search context by calling vim_findfile_init().
4062 * It is possible to give a search context from a previous call to
4063 * vim_findfile_init(), so it can be reused. After this you call vim_findfile()
4064 * until you are satisfied with the result or it returns NULL. On every call it
4065 * returns the next file which matches the conditions given to
4066 * vim_findfile_init(). If it doesn't find a next file it returns NULL.
4067 *
4068 * It is possible to call vim_findfile_init() again to reinitialise your search
4069 * with some new parameters. Don't forget to pass your old search context to
4070 * it, so it can reuse it and especially reuse the list of already visited
4071 * directories. If you want to delete the list of already visited directories
4072 * simply call vim_findfile_free_visited().
4073 *
4074 * When you are done call vim_findfile_cleanup() to free the search context.
4075 *
4076 * The function vim_findfile_init() has a long comment, which describes the
4077 * needed parameters.
4078 *
4079 *
4080 *
4081 * ATTENTION:
4082 * ==========
Bram Moolenaar77f66d62007-02-20 02:16:18 +00004083 * Also we use an allocated search context here, this functions are NOT
Bram Moolenaar071d4272004-06-13 20:20:40 +00004084 * thread-safe!!!!!
4085 *
4086 * To minimize parameter passing (or because I'm to lazy), only the
4087 * external visible functions get a search context as a parameter. This is
4088 * then assigned to a static global, which is used throughout the local
4089 * functions.
4090 */
4091
4092/*
4093 * type for the directory search stack
4094 */
4095typedef struct ff_stack
4096{
4097 struct ff_stack *ffs_prev;
4098
4099 /* the fix part (no wildcards) and the part containing the wildcards
4100 * of the search path
4101 */
4102 char_u *ffs_fix_path;
4103#ifdef FEAT_PATH_EXTRA
4104 char_u *ffs_wc_path;
4105#endif
4106
4107 /* files/dirs found in the above directory, matched by the first wildcard
4108 * of wc_part
4109 */
4110 char_u **ffs_filearray;
4111 int ffs_filearray_size;
4112 char_u ffs_filearray_cur; /* needed for partly handled dirs */
4113
4114 /* to store status of partly handled directories
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004115 * 0: we work on this directory for the first time
Bram Moolenaar071d4272004-06-13 20:20:40 +00004116 * 1: this directory was partly searched in an earlier step
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004117 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004118 int ffs_stage;
4119
4120 /* How deep are we in the directory tree?
4121 * Counts backward from value of level parameter to vim_findfile_init
4122 */
4123 int ffs_level;
4124
4125 /* Did we already expand '**' to an empty string? */
4126 int ffs_star_star_empty;
4127} ff_stack_T;
4128
4129/*
4130 * type for already visited directories or files.
4131 */
4132typedef struct ff_visited
4133{
4134 struct ff_visited *ffv_next;
4135
4136#ifdef FEAT_PATH_EXTRA
4137 /* Visited directories are different if the wildcard string are
4138 * different. So we have to save it.
4139 */
4140 char_u *ffv_wc_path;
4141#endif
4142 /* for unix use inode etc for comparison (needed because of links), else
4143 * use filename.
4144 */
4145#ifdef UNIX
Bram Moolenaare1fbddc2009-05-16 19:07:03 +00004146 int ffv_dev_valid; /* ffv_dev and ffv_ino were set */
4147 dev_t ffv_dev; /* device number */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004148 ino_t ffv_ino; /* inode number */
4149#endif
4150 /* The memory for this struct is allocated according to the length of
4151 * ffv_fname.
4152 */
4153 char_u ffv_fname[1]; /* actually longer */
4154} ff_visited_T;
4155
4156/*
4157 * We might have to manage several visited lists during a search.
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00004158 * This is especially needed for the tags option. If tags is set to:
Bram Moolenaar071d4272004-06-13 20:20:40 +00004159 * "./++/tags,./++/TAGS,++/tags" (replace + with *)
4160 * So we have to do 3 searches:
4161 * 1) search from the current files directory downward for the file "tags"
4162 * 2) search from the current files directory downward for the file "TAGS"
4163 * 3) search from Vims current directory downwards for the file "tags"
4164 * As you can see, the first and the third search are for the same file, so for
4165 * the third search we can use the visited list of the first search. For the
4166 * second search we must start from a empty visited list.
4167 * The struct ff_visited_list_hdr is used to manage a linked list of already
4168 * visited lists.
4169 */
4170typedef struct ff_visited_list_hdr
4171{
4172 struct ff_visited_list_hdr *ffvl_next;
4173
4174 /* the filename the attached visited list is for */
4175 char_u *ffvl_filename;
4176
4177 ff_visited_T *ffvl_visited_list;
4178
4179} ff_visited_list_hdr_T;
4180
4181
4182/*
4183 * '**' can be expanded to several directory levels.
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00004184 * Set the default maximum depth.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004185 */
4186#define FF_MAX_STAR_STAR_EXPAND ((char_u)30)
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004187
Bram Moolenaar071d4272004-06-13 20:20:40 +00004188/*
4189 * The search context:
4190 * ffsc_stack_ptr: the stack for the dirs to search
4191 * ffsc_visited_list: the currently active visited list
4192 * ffsc_dir_visited_list: the currently active visited list for search dirs
4193 * ffsc_visited_lists_list: the list of all visited lists
4194 * ffsc_dir_visited_lists_list: the list of all visited lists for search dirs
4195 * ffsc_file_to_search: the file to search for
4196 * ffsc_start_dir: the starting directory, if search path was relative
4197 * ffsc_fix_path: the fix part of the given path (without wildcards)
4198 * Needed for upward search.
4199 * ffsc_wc_path: the part of the given path containing wildcards
4200 * ffsc_level: how many levels of dirs to search downwards
4201 * ffsc_stopdirs_v: array of stop directories for upward search
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004202 * ffsc_find_what: FINDFILE_BOTH, FINDFILE_DIR or FINDFILE_FILE
Bram Moolenaar463ee342010-08-08 18:17:52 +02004203 * ffsc_tagfile: searching for tags file, don't use 'suffixesadd'
Bram Moolenaar071d4272004-06-13 20:20:40 +00004204 */
4205typedef struct ff_search_ctx_T
4206{
4207 ff_stack_T *ffsc_stack_ptr;
4208 ff_visited_list_hdr_T *ffsc_visited_list;
4209 ff_visited_list_hdr_T *ffsc_dir_visited_list;
4210 ff_visited_list_hdr_T *ffsc_visited_lists_list;
4211 ff_visited_list_hdr_T *ffsc_dir_visited_lists_list;
4212 char_u *ffsc_file_to_search;
4213 char_u *ffsc_start_dir;
4214 char_u *ffsc_fix_path;
4215#ifdef FEAT_PATH_EXTRA
4216 char_u *ffsc_wc_path;
4217 int ffsc_level;
4218 char_u **ffsc_stopdirs_v;
4219#endif
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004220 int ffsc_find_what;
Bram Moolenaar463ee342010-08-08 18:17:52 +02004221 int ffsc_tagfile;
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00004222} ff_search_ctx_T;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004223
Bram Moolenaar071d4272004-06-13 20:20:40 +00004224/* locally needed functions */
4225#ifdef FEAT_PATH_EXTRA
4226static int ff_check_visited __ARGS((ff_visited_T **, char_u *, char_u *));
4227#else
4228static int ff_check_visited __ARGS((ff_visited_T **, char_u *));
4229#endif
4230static void vim_findfile_free_visited_list __ARGS((ff_visited_list_hdr_T **list_headp));
4231static void ff_free_visited_list __ARGS((ff_visited_T *vl));
4232static ff_visited_list_hdr_T* ff_get_visited_list __ARGS((char_u *, ff_visited_list_hdr_T **list_headp));
4233#ifdef FEAT_PATH_EXTRA
4234static int ff_wc_equal __ARGS((char_u *s1, char_u *s2));
4235#endif
4236
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004237static void ff_push __ARGS((ff_search_ctx_T *search_ctx, ff_stack_T *stack_ptr));
4238static ff_stack_T *ff_pop __ARGS((ff_search_ctx_T *search_ctx));
4239static void ff_clear __ARGS((ff_search_ctx_T *search_ctx));
4240static void ff_free_stack_element __ARGS((ff_stack_T *stack_ptr));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004241#ifdef FEAT_PATH_EXTRA
4242static ff_stack_T *ff_create_stack_element __ARGS((char_u *, char_u *, int, int));
4243#else
4244static ff_stack_T *ff_create_stack_element __ARGS((char_u *, int, int));
4245#endif
4246#ifdef FEAT_PATH_EXTRA
4247static int ff_path_in_stoplist __ARGS((char_u *, int, char_u **));
4248#endif
4249
Bram Moolenaar071d4272004-06-13 20:20:40 +00004250#if 0
4251/*
4252 * if someone likes findfirst/findnext, here are the functions
4253 * NOT TESTED!!
4254 */
4255
4256static void *ff_fn_search_context = NULL;
4257
4258 char_u *
4259vim_findfirst(path, filename, level)
4260 char_u *path;
4261 char_u *filename;
4262 int level;
4263{
4264 ff_fn_search_context =
4265 vim_findfile_init(path, filename, NULL, level, TRUE, FALSE,
4266 ff_fn_search_context, rel_fname);
4267 if (NULL == ff_fn_search_context)
4268 return NULL;
4269 else
4270 return vim_findnext()
4271}
4272
4273 char_u *
4274vim_findnext()
4275{
4276 char_u *ret = vim_findfile(ff_fn_search_context);
4277
4278 if (NULL == ret)
4279 {
4280 vim_findfile_cleanup(ff_fn_search_context);
4281 ff_fn_search_context = NULL;
4282 }
4283 return ret;
4284}
4285#endif
4286
4287/*
Bram Moolenaare2b590e2010-08-08 18:29:48 +02004288 * Initialization routine for vim_findfile().
Bram Moolenaar071d4272004-06-13 20:20:40 +00004289 *
Bram Moolenaarbe18d102010-05-22 21:37:53 +02004290 * Returns the newly allocated search context or NULL if an error occurred.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004291 *
4292 * Don't forget to clean up by calling vim_findfile_cleanup() if you are done
4293 * with the search context.
4294 *
4295 * Find the file 'filename' in the directory 'path'.
4296 * The parameter 'path' may contain wildcards. If so only search 'level'
4297 * directories deep. The parameter 'level' is the absolute maximum and is
4298 * not related to restricts given to the '**' wildcard. If 'level' is 100
4299 * and you use '**200' vim_findfile() will stop after 100 levels.
4300 *
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004301 * 'filename' cannot contain wildcards! It is used as-is, no backslashes to
4302 * escape special characters.
4303 *
Bram Moolenaar071d4272004-06-13 20:20:40 +00004304 * If 'stopdirs' is not NULL and nothing is found downward, the search is
4305 * restarted on the next higher directory level. This is repeated until the
4306 * start-directory of a search is contained in 'stopdirs'. 'stopdirs' has the
4307 * format ";*<dirname>*\(;<dirname>\)*;\=$".
4308 *
4309 * If the 'path' is relative, the starting dir for the search is either VIM's
4310 * current dir or if the path starts with "./" the current files dir.
Bram Moolenaarbe18d102010-05-22 21:37:53 +02004311 * If the 'path' is absolute, the starting dir is that part of the path before
Bram Moolenaar071d4272004-06-13 20:20:40 +00004312 * the first wildcard.
4313 *
4314 * Upward search is only done on the starting dir.
4315 *
4316 * If 'free_visited' is TRUE the list of already visited files/directories is
4317 * cleared. Set this to FALSE if you just want to search from another
4318 * directory, but want to be sure that no directory from a previous search is
4319 * searched again. This is useful if you search for a file at different places.
4320 * The list of visited files/dirs can also be cleared with the function
4321 * vim_findfile_free_visited().
4322 *
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004323 * Set the parameter 'find_what' to FINDFILE_DIR if you want to search for
4324 * directories only, FINDFILE_FILE for files only, FINDFILE_BOTH for both.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004325 *
4326 * A search context returned by a previous call to vim_findfile_init() can be
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004327 * passed in the parameter "search_ctx_arg". This context is reused and
4328 * reinitialized with the new parameters. The list of already visited
Bram Moolenaar071d4272004-06-13 20:20:40 +00004329 * directories from this context is only deleted if the parameter
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004330 * "free_visited" is true. Be aware that the passed "search_ctx_arg" is freed
4331 * if the reinitialization fails.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004332 *
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004333 * If you don't have a search context from a previous call "search_ctx_arg"
4334 * must be NULL.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004335 *
4336 * This function silently ignores a few errors, vim_findfile() will have
4337 * limited functionality then.
4338 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004339 void *
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004340vim_findfile_init(path, filename, stopdirs, level, free_visited, find_what,
4341 search_ctx_arg, tagfile, rel_fname)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004342 char_u *path;
4343 char_u *filename;
Bram Moolenaare1fbddc2009-05-16 19:07:03 +00004344 char_u *stopdirs UNUSED;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004345 int level;
4346 int free_visited;
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004347 int find_what;
4348 void *search_ctx_arg;
Bram Moolenaar463ee342010-08-08 18:17:52 +02004349 int tagfile; /* expanding names of tags files */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004350 char_u *rel_fname; /* file name to use for "." */
4351{
4352#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004353 char_u *wc_part;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004354#endif
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004355 ff_stack_T *sptr;
4356 ff_search_ctx_T *search_ctx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004357
4358 /* If a search context is given by the caller, reuse it, else allocate a
4359 * new one.
4360 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004361 if (search_ctx_arg != NULL)
4362 search_ctx = search_ctx_arg;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004363 else
4364 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004365 search_ctx = (ff_search_ctx_T*)alloc((unsigned)sizeof(ff_search_ctx_T));
4366 if (search_ctx == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004367 goto error_return;
Bram Moolenaar7db5fc82010-05-24 11:59:29 +02004368 vim_memset(search_ctx, 0, sizeof(ff_search_ctx_T));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004369 }
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004370 search_ctx->ffsc_find_what = find_what;
Bram Moolenaar463ee342010-08-08 18:17:52 +02004371 search_ctx->ffsc_tagfile = tagfile;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004372
4373 /* clear the search context, but NOT the visited lists */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004374 ff_clear(search_ctx);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004375
4376 /* clear visited list if wanted */
4377 if (free_visited == TRUE)
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004378 vim_findfile_free_visited(search_ctx);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004379 else
4380 {
4381 /* Reuse old visited lists. Get the visited list for the given
4382 * filename. If no list for the current filename exists, creates a new
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004383 * one. */
4384 search_ctx->ffsc_visited_list = ff_get_visited_list(filename,
4385 &search_ctx->ffsc_visited_lists_list);
4386 if (search_ctx->ffsc_visited_list == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004387 goto error_return;
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004388 search_ctx->ffsc_dir_visited_list = ff_get_visited_list(filename,
4389 &search_ctx->ffsc_dir_visited_lists_list);
4390 if (search_ctx->ffsc_dir_visited_list == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004391 goto error_return;
4392 }
4393
4394 if (ff_expand_buffer == NULL)
4395 {
4396 ff_expand_buffer = (char_u*)alloc(MAXPATHL);
4397 if (ff_expand_buffer == NULL)
4398 goto error_return;
4399 }
4400
4401 /* Store information on starting dir now if path is relative.
Bram Moolenaar5eb86f92004-07-26 12:53:41 +00004402 * If path is absolute, we do that later. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004403 if (path[0] == '.'
4404 && (vim_ispathsep(path[1]) || path[1] == NUL)
4405 && (!tagfile || vim_strchr(p_cpo, CPO_DOTTAG) == NULL)
4406 && rel_fname != NULL)
4407 {
4408 int len = (int)(gettail(rel_fname) - rel_fname);
4409
4410 if (!vim_isAbsName(rel_fname) && len + 1 < MAXPATHL)
4411 {
4412 /* Make the start dir an absolute path name. */
Bram Moolenaarbbebc852005-07-18 21:47:53 +00004413 vim_strncpy(ff_expand_buffer, rel_fname, len);
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004414 search_ctx->ffsc_start_dir = FullName_save(ff_expand_buffer, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004415 }
4416 else
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004417 search_ctx->ffsc_start_dir = vim_strnsave(rel_fname, len);
4418 if (search_ctx->ffsc_start_dir == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004419 goto error_return;
4420 if (*++path != NUL)
4421 ++path;
4422 }
4423 else if (*path == NUL || !vim_isAbsName(path))
4424 {
4425#ifdef BACKSLASH_IN_FILENAME
4426 /* "c:dir" needs "c:" to be expanded, otherwise use current dir */
4427 if (*path != NUL && path[1] == ':')
4428 {
4429 char_u drive[3];
4430
4431 drive[0] = path[0];
4432 drive[1] = ':';
4433 drive[2] = NUL;
4434 if (vim_FullName(drive, ff_expand_buffer, MAXPATHL, TRUE) == FAIL)
4435 goto error_return;
4436 path += 2;
4437 }
4438 else
4439#endif
4440 if (mch_dirname(ff_expand_buffer, MAXPATHL) == FAIL)
4441 goto error_return;
4442
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004443 search_ctx->ffsc_start_dir = vim_strsave(ff_expand_buffer);
4444 if (search_ctx->ffsc_start_dir == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004445 goto error_return;
4446
4447#ifdef BACKSLASH_IN_FILENAME
4448 /* A path that starts with "/dir" is relative to the drive, not to the
4449 * directory (but not for "//machine/dir"). Only use the drive name. */
4450 if ((*path == '/' || *path == '\\')
4451 && path[1] != path[0]
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004452 && search_ctx->ffsc_start_dir[1] == ':')
4453 search_ctx->ffsc_start_dir[2] = NUL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004454#endif
4455 }
4456
4457#ifdef FEAT_PATH_EXTRA
4458 /*
4459 * If stopdirs are given, split them into an array of pointers.
4460 * If this fails (mem allocation), there is no upward search at all or a
4461 * stop directory is not recognized -> continue silently.
4462 * If stopdirs just contains a ";" or is empty,
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004463 * search_ctx->ffsc_stopdirs_v will only contain a NULL pointer. This
Bram Moolenaar071d4272004-06-13 20:20:40 +00004464 * is handled as unlimited upward search. See function
4465 * ff_path_in_stoplist() for details.
4466 */
4467 if (stopdirs != NULL)
4468 {
4469 char_u *walker = stopdirs;
4470 int dircount;
4471
4472 while (*walker == ';')
4473 walker++;
4474
4475 dircount = 1;
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004476 search_ctx->ffsc_stopdirs_v =
4477 (char_u **)alloc((unsigned)sizeof(char_u *));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004478
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004479 if (search_ctx->ffsc_stopdirs_v != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004480 {
4481 do
4482 {
4483 char_u *helper;
4484 void *ptr;
4485
4486 helper = walker;
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004487 ptr = vim_realloc(search_ctx->ffsc_stopdirs_v,
Bram Moolenaar071d4272004-06-13 20:20:40 +00004488 (dircount + 1) * sizeof(char_u *));
4489 if (ptr)
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004490 search_ctx->ffsc_stopdirs_v = ptr;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004491 else
4492 /* ignore, keep what we have and continue */
4493 break;
4494 walker = vim_strchr(walker, ';');
4495 if (walker)
4496 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004497 search_ctx->ffsc_stopdirs_v[dircount-1] =
4498 vim_strnsave(helper, (int)(walker - helper));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004499 walker++;
4500 }
4501 else
4502 /* this might be "", which means ascent till top
4503 * of directory tree.
4504 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004505 search_ctx->ffsc_stopdirs_v[dircount-1] =
4506 vim_strsave(helper);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004507
4508 dircount++;
4509
4510 } while (walker != NULL);
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004511 search_ctx->ffsc_stopdirs_v[dircount-1] = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004512 }
4513 }
4514#endif
4515
4516#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004517 search_ctx->ffsc_level = level;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004518
4519 /* split into:
4520 * -fix path
4521 * -wildcard_stuff (might be NULL)
4522 */
4523 wc_part = vim_strchr(path, '*');
4524 if (wc_part != NULL)
4525 {
4526 int llevel;
4527 int len;
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00004528 char *errpt;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004529
4530 /* save the fix part of the path */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004531 search_ctx->ffsc_fix_path = vim_strnsave(path, (int)(wc_part - path));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004532
4533 /*
4534 * copy wc_path and add restricts to the '**' wildcard.
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00004535 * The octet after a '**' is used as a (binary) counter.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004536 * So '**3' is transposed to '**^C' ('^C' is ASCII value 3)
4537 * or '**76' is transposed to '**N'( 'N' is ASCII value 76).
4538 * For EBCDIC you get different character values.
4539 * If no restrict is given after '**' the default is used.
Bram Moolenaar21160b92009-11-11 15:56:10 +00004540 * Due to this technique the path looks awful if you print it as a
Bram Moolenaar071d4272004-06-13 20:20:40 +00004541 * string.
4542 */
4543 len = 0;
4544 while (*wc_part != NUL)
4545 {
4546 if (STRNCMP(wc_part, "**", 2) == 0)
4547 {
4548 ff_expand_buffer[len++] = *wc_part++;
4549 ff_expand_buffer[len++] = *wc_part++;
4550
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00004551 llevel = strtol((char *)wc_part, &errpt, 10);
4552 if ((char_u *)errpt != wc_part && llevel > 0 && llevel < 255)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004553 ff_expand_buffer[len++] = llevel;
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00004554 else if ((char_u *)errpt != wc_part && llevel == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004555 /* restrict is 0 -> remove already added '**' */
4556 len -= 2;
4557 else
4558 ff_expand_buffer[len++] = FF_MAX_STAR_STAR_EXPAND;
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00004559 wc_part = (char_u *)errpt;
Bram Moolenaar1d94f9b2005-08-04 21:29:45 +00004560 if (*wc_part != NUL && !vim_ispathsep(*wc_part))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004561 {
4562 EMSG2(_("E343: Invalid path: '**[number]' must be at the end of the path or be followed by '%s'."), PATHSEPSTR);
4563 goto error_return;
4564 }
4565 }
4566 else
4567 ff_expand_buffer[len++] = *wc_part++;
4568 }
4569 ff_expand_buffer[len] = NUL;
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004570 search_ctx->ffsc_wc_path = vim_strsave(ff_expand_buffer);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004571
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004572 if (search_ctx->ffsc_wc_path == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004573 goto error_return;
4574 }
4575 else
4576#endif
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004577 search_ctx->ffsc_fix_path = vim_strsave(path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004578
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004579 if (search_ctx->ffsc_start_dir == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004580 {
4581 /* store the fix part as startdir.
4582 * This is needed if the parameter path is fully qualified.
4583 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004584 search_ctx->ffsc_start_dir = vim_strsave(search_ctx->ffsc_fix_path);
Bram Moolenaara9d52e32010-07-31 16:44:19 +02004585 if (search_ctx->ffsc_start_dir == NULL)
4586 goto error_return;
4587 search_ctx->ffsc_fix_path[0] = NUL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004588 }
4589
4590 /* create an absolute path */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004591 STRCPY(ff_expand_buffer, search_ctx->ffsc_start_dir);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004592 add_pathsep(ff_expand_buffer);
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004593 STRCAT(ff_expand_buffer, search_ctx->ffsc_fix_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004594 add_pathsep(ff_expand_buffer);
4595
4596 sptr = ff_create_stack_element(ff_expand_buffer,
4597#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004598 search_ctx->ffsc_wc_path,
Bram Moolenaar071d4272004-06-13 20:20:40 +00004599#endif
4600 level, 0);
4601
4602 if (sptr == NULL)
4603 goto error_return;
4604
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004605 ff_push(search_ctx, sptr);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004606
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004607 search_ctx->ffsc_file_to_search = vim_strsave(filename);
4608 if (search_ctx->ffsc_file_to_search == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004609 goto error_return;
4610
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004611 return search_ctx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004612
4613error_return:
4614 /*
4615 * We clear the search context now!
4616 * Even when the caller gave us a (perhaps valid) context we free it here,
4617 * as we might have already destroyed it.
4618 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004619 vim_findfile_cleanup(search_ctx);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004620 return NULL;
4621}
4622
4623#if defined(FEAT_PATH_EXTRA) || defined(PROTO)
4624/*
4625 * Get the stopdir string. Check that ';' is not escaped.
4626 */
4627 char_u *
4628vim_findfile_stopdir(buf)
4629 char_u *buf;
4630{
4631 char_u *r_ptr = buf;
4632
4633 while (*r_ptr != NUL && *r_ptr != ';')
4634 {
4635 if (r_ptr[0] == '\\' && r_ptr[1] == ';')
4636 {
4637 /* overwrite the escape char,
4638 * use STRLEN(r_ptr) to move the trailing '\0'
4639 */
Bram Moolenaar446cb832008-06-24 21:56:24 +00004640 STRMOVE(r_ptr, r_ptr + 1);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004641 r_ptr++;
4642 }
4643 r_ptr++;
4644 }
4645 if (*r_ptr == ';')
4646 {
4647 *r_ptr = 0;
4648 r_ptr++;
4649 }
4650 else if (*r_ptr == NUL)
4651 r_ptr = NULL;
4652 return r_ptr;
4653}
4654#endif
4655
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004656/*
4657 * Clean up the given search context. Can handle a NULL pointer.
4658 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004659 void
4660vim_findfile_cleanup(ctx)
4661 void *ctx;
4662{
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00004663 if (ctx == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004664 return;
4665
Bram Moolenaar071d4272004-06-13 20:20:40 +00004666 vim_findfile_free_visited(ctx);
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004667 ff_clear(ctx);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004668 vim_free(ctx);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004669}
4670
4671/*
4672 * Find a file in a search context.
4673 * The search context was created with vim_findfile_init() above.
4674 * Return a pointer to an allocated file name or NULL if nothing found.
4675 * To get all matching files call this function until you get NULL.
4676 *
Bram Moolenaar5eb86f92004-07-26 12:53:41 +00004677 * If the passed search_context is NULL, NULL is returned.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004678 *
4679 * The search algorithm is depth first. To change this replace the
4680 * stack with a list (don't forget to leave partly searched directories on the
4681 * top of the list).
4682 */
4683 char_u *
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004684vim_findfile(search_ctx_arg)
4685 void *search_ctx_arg;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004686{
4687 char_u *file_path;
4688#ifdef FEAT_PATH_EXTRA
4689 char_u *rest_of_wildcards;
4690 char_u *path_end = NULL;
4691#endif
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004692 ff_stack_T *stackp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004693#if defined(FEAT_SEARCHPATH) || defined(FEAT_PATH_EXTRA)
4694 int len;
4695#endif
4696 int i;
4697 char_u *p;
4698#ifdef FEAT_SEARCHPATH
4699 char_u *suf;
4700#endif
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004701 ff_search_ctx_T *search_ctx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004702
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004703 if (search_ctx_arg == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004704 return NULL;
4705
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004706 search_ctx = (ff_search_ctx_T *)search_ctx_arg;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004707
4708 /*
4709 * filepath is used as buffer for various actions and as the storage to
4710 * return a found filename.
4711 */
4712 if ((file_path = alloc((int)MAXPATHL)) == NULL)
4713 return NULL;
4714
4715#ifdef FEAT_PATH_EXTRA
4716 /* store the end of the start dir -- needed for upward search */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004717 if (search_ctx->ffsc_start_dir != NULL)
4718 path_end = &search_ctx->ffsc_start_dir[
4719 STRLEN(search_ctx->ffsc_start_dir)];
Bram Moolenaar071d4272004-06-13 20:20:40 +00004720#endif
4721
4722#ifdef FEAT_PATH_EXTRA
4723 /* upward search loop */
4724 for (;;)
4725 {
4726#endif
4727 /* downward search loop */
4728 for (;;)
4729 {
4730 /* check if user user wants to stop the search*/
4731 ui_breakcheck();
4732 if (got_int)
4733 break;
4734
4735 /* get directory to work on from stack */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004736 stackp = ff_pop(search_ctx);
4737 if (stackp == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004738 break;
4739
4740 /*
4741 * TODO: decide if we leave this test in
4742 *
4743 * GOOD: don't search a directory(-tree) twice.
4744 * BAD: - check linked list for every new directory entered.
4745 * - check for double files also done below
4746 *
4747 * Here we check if we already searched this directory.
4748 * We already searched a directory if:
4749 * 1) The directory is the same.
4750 * 2) We would use the same wildcard string.
4751 *
4752 * Good if you have links on same directory via several ways
4753 * or you have selfreferences in directories (e.g. SuSE Linux 6.3:
4754 * /etc/rc.d/init.d is linked to /etc/rc.d -> endless loop)
4755 *
4756 * This check is only needed for directories we work on for the
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004757 * first time (hence stackp->ff_filearray == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004758 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004759 if (stackp->ffs_filearray == NULL
4760 && ff_check_visited(&search_ctx->ffsc_dir_visited_list
Bram Moolenaar071d4272004-06-13 20:20:40 +00004761 ->ffvl_visited_list,
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004762 stackp->ffs_fix_path
Bram Moolenaar071d4272004-06-13 20:20:40 +00004763#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004764 , stackp->ffs_wc_path
Bram Moolenaar071d4272004-06-13 20:20:40 +00004765#endif
4766 ) == FAIL)
4767 {
4768#ifdef FF_VERBOSE
4769 if (p_verbose >= 5)
4770 {
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004771 verbose_enter_scroll();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004772 smsg((char_u *)"Already Searched: %s (%s)",
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004773 stackp->ffs_fix_path, stackp->ffs_wc_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004774 /* don't overwrite this either */
4775 msg_puts((char_u *)"\n");
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004776 verbose_leave_scroll();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004777 }
4778#endif
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004779 ff_free_stack_element(stackp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004780 continue;
4781 }
4782#ifdef FF_VERBOSE
4783 else if (p_verbose >= 5)
4784 {
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004785 verbose_enter_scroll();
Bram Moolenaar051b7822005-05-19 21:00:46 +00004786 smsg((char_u *)"Searching: %s (%s)",
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004787 stackp->ffs_fix_path, stackp->ffs_wc_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004788 /* don't overwrite this either */
4789 msg_puts((char_u *)"\n");
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004790 verbose_leave_scroll();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004791 }
4792#endif
4793
4794 /* check depth */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004795 if (stackp->ffs_level <= 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004796 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004797 ff_free_stack_element(stackp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004798 continue;
4799 }
4800
4801 file_path[0] = NUL;
4802
4803 /*
4804 * If no filearray till now expand wildcards
4805 * The function expand_wildcards() can handle an array of paths
4806 * and all possible expands are returned in one array. We use this
4807 * to handle the expansion of '**' into an empty string.
4808 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004809 if (stackp->ffs_filearray == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004810 {
4811 char_u *dirptrs[2];
4812
4813 /* we use filepath to build the path expand_wildcards() should
4814 * expand.
4815 */
4816 dirptrs[0] = file_path;
4817 dirptrs[1] = NULL;
4818
4819 /* if we have a start dir copy it in */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004820 if (!vim_isAbsName(stackp->ffs_fix_path)
4821 && search_ctx->ffsc_start_dir)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004822 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004823 STRCPY(file_path, search_ctx->ffsc_start_dir);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004824 add_pathsep(file_path);
4825 }
4826
4827 /* append the fix part of the search path */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004828 STRCAT(file_path, stackp->ffs_fix_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004829 add_pathsep(file_path);
4830
4831#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004832 rest_of_wildcards = stackp->ffs_wc_path;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004833 if (*rest_of_wildcards != NUL)
4834 {
4835 len = (int)STRLEN(file_path);
4836 if (STRNCMP(rest_of_wildcards, "**", 2) == 0)
4837 {
4838 /* pointer to the restrict byte
4839 * The restrict byte is not a character!
4840 */
4841 p = rest_of_wildcards + 2;
4842
4843 if (*p > 0)
4844 {
4845 (*p)--;
4846 file_path[len++] = '*';
4847 }
4848
4849 if (*p == 0)
4850 {
4851 /* remove '**<numb> from wildcards */
Bram Moolenaar446cb832008-06-24 21:56:24 +00004852 STRMOVE(rest_of_wildcards, rest_of_wildcards + 3);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004853 }
4854 else
4855 rest_of_wildcards += 3;
4856
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004857 if (stackp->ffs_star_star_empty == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004858 {
4859 /* if not done before, expand '**' to empty */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004860 stackp->ffs_star_star_empty = 1;
4861 dirptrs[1] = stackp->ffs_fix_path;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004862 }
4863 }
4864
4865 /*
4866 * Here we copy until the next path separator or the end of
4867 * the path. If we stop at a path separator, there is
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00004868 * still something else left. This is handled below by
Bram Moolenaar071d4272004-06-13 20:20:40 +00004869 * pushing every directory returned from expand_wildcards()
4870 * on the stack again for further search.
4871 */
4872 while (*rest_of_wildcards
4873 && !vim_ispathsep(*rest_of_wildcards))
4874 file_path[len++] = *rest_of_wildcards++;
4875
4876 file_path[len] = NUL;
4877 if (vim_ispathsep(*rest_of_wildcards))
4878 rest_of_wildcards++;
4879 }
4880#endif
4881
4882 /*
4883 * Expand wildcards like "*" and "$VAR".
4884 * If the path is a URL don't try this.
4885 */
4886 if (path_with_url(dirptrs[0]))
4887 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004888 stackp->ffs_filearray = (char_u **)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004889 alloc((unsigned)sizeof(char *));
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004890 if (stackp->ffs_filearray != NULL
4891 && (stackp->ffs_filearray[0]
Bram Moolenaar071d4272004-06-13 20:20:40 +00004892 = vim_strsave(dirptrs[0])) != NULL)
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004893 stackp->ffs_filearray_size = 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004894 else
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004895 stackp->ffs_filearray_size = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004896 }
4897 else
4898 expand_wildcards((dirptrs[1] == NULL) ? 1 : 2, dirptrs,
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004899 &stackp->ffs_filearray_size,
4900 &stackp->ffs_filearray,
Bram Moolenaar071d4272004-06-13 20:20:40 +00004901 EW_DIR|EW_ADDSLASH|EW_SILENT);
4902
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004903 stackp->ffs_filearray_cur = 0;
4904 stackp->ffs_stage = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004905 }
4906#ifdef FEAT_PATH_EXTRA
4907 else
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004908 rest_of_wildcards = &stackp->ffs_wc_path[
4909 STRLEN(stackp->ffs_wc_path)];
Bram Moolenaar071d4272004-06-13 20:20:40 +00004910#endif
4911
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004912 if (stackp->ffs_stage == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004913 {
4914 /* this is the first time we work on this directory */
4915#ifdef FEAT_PATH_EXTRA
4916 if (*rest_of_wildcards == NUL)
4917#endif
4918 {
4919 /*
4920 * we don't have further wildcards to expand, so we have to
4921 * check for the final file now
4922 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004923 for (i = stackp->ffs_filearray_cur;
4924 i < stackp->ffs_filearray_size; ++i)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004925 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004926 if (!path_with_url(stackp->ffs_filearray[i])
4927 && !mch_isdir(stackp->ffs_filearray[i]))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004928 continue; /* not a directory */
4929
Bram Moolenaar21160b92009-11-11 15:56:10 +00004930 /* prepare the filename to be checked for existence
Bram Moolenaar071d4272004-06-13 20:20:40 +00004931 * below */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004932 STRCPY(file_path, stackp->ffs_filearray[i]);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004933 add_pathsep(file_path);
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004934 STRCAT(file_path, search_ctx->ffsc_file_to_search);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004935
4936 /*
4937 * Try without extra suffix and then with suffixes
4938 * from 'suffixesadd'.
4939 */
4940#ifdef FEAT_SEARCHPATH
4941 len = (int)STRLEN(file_path);
Bram Moolenaar463ee342010-08-08 18:17:52 +02004942 if (search_ctx->ffsc_tagfile)
4943 suf = (char_u *)"";
4944 else
4945 suf = curbuf->b_p_sua;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004946 for (;;)
4947#endif
4948 {
4949 /* if file exists and we didn't already find it */
4950 if ((path_with_url(file_path)
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004951 || (mch_getperm(file_path) >= 0
4952 && (search_ctx->ffsc_find_what
4953 == FINDFILE_BOTH
4954 || ((search_ctx->ffsc_find_what
4955 == FINDFILE_DIR)
4956 == mch_isdir(file_path)))))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004957#ifndef FF_VERBOSE
4958 && (ff_check_visited(
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004959 &search_ctx->ffsc_visited_list->ffvl_visited_list,
Bram Moolenaar071d4272004-06-13 20:20:40 +00004960 file_path
4961#ifdef FEAT_PATH_EXTRA
4962 , (char_u *)""
4963#endif
4964 ) == OK)
4965#endif
4966 )
4967 {
4968#ifdef FF_VERBOSE
4969 if (ff_check_visited(
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004970 &search_ctx->ffsc_visited_list->ffvl_visited_list,
Bram Moolenaar071d4272004-06-13 20:20:40 +00004971 file_path
4972#ifdef FEAT_PATH_EXTRA
4973 , (char_u *)""
4974#endif
4975 ) == FAIL)
4976 {
4977 if (p_verbose >= 5)
4978 {
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004979 verbose_enter_scroll();
Bram Moolenaar051b7822005-05-19 21:00:46 +00004980 smsg((char_u *)"Already: %s",
Bram Moolenaar071d4272004-06-13 20:20:40 +00004981 file_path);
4982 /* don't overwrite this either */
4983 msg_puts((char_u *)"\n");
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004984 verbose_leave_scroll();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004985 }
4986 continue;
4987 }
4988#endif
4989
4990 /* push dir to examine rest of subdirs later */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004991 stackp->ffs_filearray_cur = i + 1;
4992 ff_push(search_ctx, stackp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004993
Bram Moolenaar6bab9fa2009-01-22 20:32:12 +00004994 if (!path_with_url(file_path))
4995 simplify_filename(file_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004996 if (mch_dirname(ff_expand_buffer, MAXPATHL)
4997 == OK)
4998 {
4999 p = shorten_fname(file_path,
5000 ff_expand_buffer);
5001 if (p != NULL)
Bram Moolenaar446cb832008-06-24 21:56:24 +00005002 STRMOVE(file_path, p);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005003 }
5004#ifdef FF_VERBOSE
5005 if (p_verbose >= 5)
5006 {
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00005007 verbose_enter_scroll();
Bram Moolenaar051b7822005-05-19 21:00:46 +00005008 smsg((char_u *)"HIT: %s", file_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005009 /* don't overwrite this either */
5010 msg_puts((char_u *)"\n");
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00005011 verbose_leave_scroll();
Bram Moolenaar071d4272004-06-13 20:20:40 +00005012 }
5013#endif
5014 return file_path;
5015 }
5016
5017#ifdef FEAT_SEARCHPATH
5018 /* Not found or found already, try next suffix. */
5019 if (*suf == NUL)
5020 break;
5021 copy_option_part(&suf, file_path + len,
5022 MAXPATHL - len, ",");
5023#endif
5024 }
5025 }
5026 }
5027#ifdef FEAT_PATH_EXTRA
5028 else
5029 {
5030 /*
5031 * still wildcards left, push the directories for further
5032 * search
5033 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005034 for (i = stackp->ffs_filearray_cur;
5035 i < stackp->ffs_filearray_size; ++i)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005036 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005037 if (!mch_isdir(stackp->ffs_filearray[i]))
Bram Moolenaar071d4272004-06-13 20:20:40 +00005038 continue; /* not a directory */
5039
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005040 ff_push(search_ctx,
5041 ff_create_stack_element(
5042 stackp->ffs_filearray[i],
5043 rest_of_wildcards,
5044 stackp->ffs_level - 1, 0));
Bram Moolenaar071d4272004-06-13 20:20:40 +00005045 }
5046 }
5047#endif
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005048 stackp->ffs_filearray_cur = 0;
5049 stackp->ffs_stage = 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005050 }
5051
5052#ifdef FEAT_PATH_EXTRA
5053 /*
5054 * if wildcards contains '**' we have to descent till we reach the
5055 * leaves of the directory tree.
5056 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005057 if (STRNCMP(stackp->ffs_wc_path, "**", 2) == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005058 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005059 for (i = stackp->ffs_filearray_cur;
5060 i < stackp->ffs_filearray_size; ++i)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005061 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005062 if (fnamecmp(stackp->ffs_filearray[i],
5063 stackp->ffs_fix_path) == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005064 continue; /* don't repush same directory */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005065 if (!mch_isdir(stackp->ffs_filearray[i]))
Bram Moolenaar071d4272004-06-13 20:20:40 +00005066 continue; /* not a directory */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005067 ff_push(search_ctx,
5068 ff_create_stack_element(stackp->ffs_filearray[i],
5069 stackp->ffs_wc_path, stackp->ffs_level - 1, 1));
Bram Moolenaar071d4272004-06-13 20:20:40 +00005070 }
5071 }
5072#endif
5073
5074 /* we are done with the current directory */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005075 ff_free_stack_element(stackp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005076
5077 }
5078
5079#ifdef FEAT_PATH_EXTRA
5080 /* If we reached this, we didn't find anything downwards.
5081 * Let's check if we should do an upward search.
5082 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005083 if (search_ctx->ffsc_start_dir
5084 && search_ctx->ffsc_stopdirs_v != NULL && !got_int)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005085 {
5086 ff_stack_T *sptr;
5087
5088 /* is the last starting directory in the stop list? */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005089 if (ff_path_in_stoplist(search_ctx->ffsc_start_dir,
5090 (int)(path_end - search_ctx->ffsc_start_dir),
5091 search_ctx->ffsc_stopdirs_v) == TRUE)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005092 break;
5093
5094 /* cut of last dir */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005095 while (path_end > search_ctx->ffsc_start_dir
5096 && vim_ispathsep(*path_end))
Bram Moolenaar071d4272004-06-13 20:20:40 +00005097 path_end--;
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005098 while (path_end > search_ctx->ffsc_start_dir
5099 && !vim_ispathsep(path_end[-1]))
Bram Moolenaar071d4272004-06-13 20:20:40 +00005100 path_end--;
5101 *path_end = 0;
5102 path_end--;
5103
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005104 if (*search_ctx->ffsc_start_dir == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005105 break;
5106
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005107 STRCPY(file_path, search_ctx->ffsc_start_dir);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005108 add_pathsep(file_path);
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005109 STRCAT(file_path, search_ctx->ffsc_fix_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005110
5111 /* create a new stack entry */
5112 sptr = ff_create_stack_element(file_path,
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005113 search_ctx->ffsc_wc_path, search_ctx->ffsc_level, 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005114 if (sptr == NULL)
5115 break;
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005116 ff_push(search_ctx, sptr);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005117 }
5118 else
5119 break;
5120 }
5121#endif
5122
5123 vim_free(file_path);
5124 return NULL;
5125}
5126
5127/*
5128 * Free the list of lists of visited files and directories
5129 * Can handle it if the passed search_context is NULL;
5130 */
5131 void
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005132vim_findfile_free_visited(search_ctx_arg)
5133 void *search_ctx_arg;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005134{
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005135 ff_search_ctx_T *search_ctx;
5136
5137 if (search_ctx_arg == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005138 return;
5139
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005140 search_ctx = (ff_search_ctx_T *)search_ctx_arg;
5141 vim_findfile_free_visited_list(&search_ctx->ffsc_visited_lists_list);
5142 vim_findfile_free_visited_list(&search_ctx->ffsc_dir_visited_lists_list);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005143}
5144
5145 static void
5146vim_findfile_free_visited_list(list_headp)
5147 ff_visited_list_hdr_T **list_headp;
5148{
5149 ff_visited_list_hdr_T *vp;
5150
5151 while (*list_headp != NULL)
5152 {
5153 vp = (*list_headp)->ffvl_next;
5154 ff_free_visited_list((*list_headp)->ffvl_visited_list);
5155
5156 vim_free((*list_headp)->ffvl_filename);
5157 vim_free(*list_headp);
5158 *list_headp = vp;
5159 }
5160 *list_headp = NULL;
5161}
5162
5163 static void
5164ff_free_visited_list(vl)
5165 ff_visited_T *vl;
5166{
5167 ff_visited_T *vp;
5168
5169 while (vl != NULL)
5170 {
5171 vp = vl->ffv_next;
5172#ifdef FEAT_PATH_EXTRA
5173 vim_free(vl->ffv_wc_path);
5174#endif
5175 vim_free(vl);
5176 vl = vp;
5177 }
5178 vl = NULL;
5179}
5180
5181/*
5182 * Returns the already visited list for the given filename. If none is found it
5183 * allocates a new one.
5184 */
5185 static ff_visited_list_hdr_T*
5186ff_get_visited_list(filename, list_headp)
5187 char_u *filename;
5188 ff_visited_list_hdr_T **list_headp;
5189{
5190 ff_visited_list_hdr_T *retptr = NULL;
5191
5192 /* check if a visited list for the given filename exists */
5193 if (*list_headp != NULL)
5194 {
5195 retptr = *list_headp;
5196 while (retptr != NULL)
5197 {
5198 if (fnamecmp(filename, retptr->ffvl_filename) == 0)
5199 {
5200#ifdef FF_VERBOSE
5201 if (p_verbose >= 5)
5202 {
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00005203 verbose_enter_scroll();
Bram Moolenaar051b7822005-05-19 21:00:46 +00005204 smsg((char_u *)"ff_get_visited_list: FOUND list for %s",
Bram Moolenaar071d4272004-06-13 20:20:40 +00005205 filename);
5206 /* don't overwrite this either */
5207 msg_puts((char_u *)"\n");
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00005208 verbose_leave_scroll();
Bram Moolenaar071d4272004-06-13 20:20:40 +00005209 }
5210#endif
5211 return retptr;
5212 }
5213 retptr = retptr->ffvl_next;
5214 }
5215 }
5216
5217#ifdef FF_VERBOSE
5218 if (p_verbose >= 5)
5219 {
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00005220 verbose_enter_scroll();
Bram Moolenaar051b7822005-05-19 21:00:46 +00005221 smsg((char_u *)"ff_get_visited_list: new list for %s", filename);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005222 /* don't overwrite this either */
5223 msg_puts((char_u *)"\n");
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00005224 verbose_leave_scroll();
Bram Moolenaar071d4272004-06-13 20:20:40 +00005225 }
5226#endif
5227
5228 /*
5229 * if we reach this we didn't find a list and we have to allocate new list
5230 */
5231 retptr = (ff_visited_list_hdr_T*)alloc((unsigned)sizeof(*retptr));
5232 if (retptr == NULL)
5233 return NULL;
5234
5235 retptr->ffvl_visited_list = NULL;
5236 retptr->ffvl_filename = vim_strsave(filename);
5237 if (retptr->ffvl_filename == NULL)
5238 {
5239 vim_free(retptr);
5240 return NULL;
5241 }
5242 retptr->ffvl_next = *list_headp;
5243 *list_headp = retptr;
5244
5245 return retptr;
5246}
5247
5248#ifdef FEAT_PATH_EXTRA
5249/*
5250 * check if two wildcard paths are equal. Returns TRUE or FALSE.
5251 * They are equal if:
5252 * - both paths are NULL
5253 * - they have the same length
5254 * - char by char comparison is OK
5255 * - the only differences are in the counters behind a '**', so
5256 * '**\20' is equal to '**\24'
5257 */
5258 static int
5259ff_wc_equal(s1, s2)
5260 char_u *s1;
5261 char_u *s2;
5262{
5263 int i;
5264
5265 if (s1 == s2)
5266 return TRUE;
5267
5268 if (s1 == NULL || s2 == NULL)
5269 return FALSE;
5270
5271 if (STRLEN(s1) != STRLEN(s2))
5272 return FAIL;
5273
5274 for (i = 0; s1[i] != NUL && s2[i] != NUL; i++)
5275 {
5276 if (s1[i] != s2[i]
5277#ifdef CASE_INSENSITIVE_FILENAME
5278 && TOUPPER_LOC(s1[i]) != TOUPPER_LOC(s2[i])
5279#endif
5280 )
5281 {
5282 if (i >= 2)
5283 if (s1[i-1] == '*' && s1[i-2] == '*')
5284 continue;
5285 else
5286 return FAIL;
5287 else
5288 return FAIL;
5289 }
5290 }
5291 return TRUE;
5292}
5293#endif
5294
5295/*
5296 * maintains the list of already visited files and dirs
5297 * returns FAIL if the given file/dir is already in the list
5298 * returns OK if it is newly added
5299 *
5300 * TODO: What to do on memory allocation problems?
5301 * -> return TRUE - Better the file is found several times instead of
5302 * never.
5303 */
5304 static int
5305ff_check_visited(visited_list, fname
5306#ifdef FEAT_PATH_EXTRA
5307 , wc_path
5308#endif
5309 )
5310 ff_visited_T **visited_list;
5311 char_u *fname;
5312#ifdef FEAT_PATH_EXTRA
5313 char_u *wc_path;
5314#endif
5315{
5316 ff_visited_T *vp;
5317#ifdef UNIX
5318 struct stat st;
5319 int url = FALSE;
5320#endif
5321
5322 /* For an URL we only compare the name, otherwise we compare the
5323 * device/inode (unix) or the full path name (not Unix). */
5324 if (path_with_url(fname))
5325 {
Bram Moolenaarbbebc852005-07-18 21:47:53 +00005326 vim_strncpy(ff_expand_buffer, fname, MAXPATHL - 1);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005327#ifdef UNIX
5328 url = TRUE;
5329#endif
5330 }
5331 else
5332 {
5333 ff_expand_buffer[0] = NUL;
5334#ifdef UNIX
5335 if (mch_stat((char *)fname, &st) < 0)
5336#else
5337 if (vim_FullName(fname, ff_expand_buffer, MAXPATHL, TRUE) == FAIL)
5338#endif
5339 return FAIL;
5340 }
5341
5342 /* check against list of already visited files */
5343 for (vp = *visited_list; vp != NULL; vp = vp->ffv_next)
5344 {
5345 if (
5346#ifdef UNIX
Bram Moolenaare1fbddc2009-05-16 19:07:03 +00005347 !url ? (vp->ffv_dev_valid && vp->ffv_dev == st.st_dev
5348 && vp->ffv_ino == st.st_ino)
5349 :
Bram Moolenaar071d4272004-06-13 20:20:40 +00005350#endif
5351 fnamecmp(vp->ffv_fname, ff_expand_buffer) == 0
5352 )
5353 {
5354#ifdef FEAT_PATH_EXTRA
5355 /* are the wildcard parts equal */
5356 if (ff_wc_equal(vp->ffv_wc_path, wc_path) == TRUE)
5357#endif
5358 /* already visited */
5359 return FAIL;
5360 }
5361 }
5362
5363 /*
5364 * New file/dir. Add it to the list of visited files/dirs.
5365 */
5366 vp = (ff_visited_T *)alloc((unsigned)(sizeof(ff_visited_T)
5367 + STRLEN(ff_expand_buffer)));
5368
5369 if (vp != NULL)
5370 {
5371#ifdef UNIX
5372 if (!url)
5373 {
Bram Moolenaare1fbddc2009-05-16 19:07:03 +00005374 vp->ffv_dev_valid = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005375 vp->ffv_ino = st.st_ino;
5376 vp->ffv_dev = st.st_dev;
5377 vp->ffv_fname[0] = NUL;
5378 }
5379 else
5380 {
Bram Moolenaare1fbddc2009-05-16 19:07:03 +00005381 vp->ffv_dev_valid = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005382#endif
5383 STRCPY(vp->ffv_fname, ff_expand_buffer);
5384#ifdef UNIX
5385 }
5386#endif
5387#ifdef FEAT_PATH_EXTRA
5388 if (wc_path != NULL)
5389 vp->ffv_wc_path = vim_strsave(wc_path);
5390 else
5391 vp->ffv_wc_path = NULL;
5392#endif
5393
5394 vp->ffv_next = *visited_list;
5395 *visited_list = vp;
5396 }
5397
5398 return OK;
5399}
5400
5401/*
5402 * create stack element from given path pieces
5403 */
5404 static ff_stack_T *
5405ff_create_stack_element(fix_part,
5406#ifdef FEAT_PATH_EXTRA
5407 wc_part,
5408#endif
5409 level, star_star_empty)
5410 char_u *fix_part;
5411#ifdef FEAT_PATH_EXTRA
5412 char_u *wc_part;
5413#endif
5414 int level;
5415 int star_star_empty;
5416{
5417 ff_stack_T *new;
5418
5419 new = (ff_stack_T *)alloc((unsigned)sizeof(ff_stack_T));
5420 if (new == NULL)
5421 return NULL;
5422
5423 new->ffs_prev = NULL;
5424 new->ffs_filearray = NULL;
5425 new->ffs_filearray_size = 0;
5426 new->ffs_filearray_cur = 0;
5427 new->ffs_stage = 0;
5428 new->ffs_level = level;
5429 new->ffs_star_star_empty = star_star_empty;;
5430
5431 /* the following saves NULL pointer checks in vim_findfile */
5432 if (fix_part == NULL)
5433 fix_part = (char_u *)"";
5434 new->ffs_fix_path = vim_strsave(fix_part);
5435
5436#ifdef FEAT_PATH_EXTRA
5437 if (wc_part == NULL)
5438 wc_part = (char_u *)"";
5439 new->ffs_wc_path = vim_strsave(wc_part);
5440#endif
5441
5442 if (new->ffs_fix_path == NULL
5443#ifdef FEAT_PATH_EXTRA
5444 || new->ffs_wc_path == NULL
5445#endif
5446 )
5447 {
5448 ff_free_stack_element(new);
5449 new = NULL;
5450 }
5451
5452 return new;
5453}
5454
5455/*
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005456 * Push a dir on the directory stack.
Bram Moolenaar071d4272004-06-13 20:20:40 +00005457 */
5458 static void
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005459ff_push(search_ctx, stack_ptr)
5460 ff_search_ctx_T *search_ctx;
5461 ff_stack_T *stack_ptr;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005462{
5463 /* check for NULL pointer, not to return an error to the user, but
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005464 * to prevent a crash */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005465 if (stack_ptr != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005466 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005467 stack_ptr->ffs_prev = search_ctx->ffsc_stack_ptr;
5468 search_ctx->ffsc_stack_ptr = stack_ptr;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005469 }
5470}
5471
5472/*
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005473 * Pop a dir from the directory stack.
5474 * Returns NULL if stack is empty.
Bram Moolenaar071d4272004-06-13 20:20:40 +00005475 */
5476 static ff_stack_T *
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005477ff_pop(search_ctx)
5478 ff_search_ctx_T *search_ctx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005479{
5480 ff_stack_T *sptr;
5481
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005482 sptr = search_ctx->ffsc_stack_ptr;
5483 if (search_ctx->ffsc_stack_ptr != NULL)
5484 search_ctx->ffsc_stack_ptr = search_ctx->ffsc_stack_ptr->ffs_prev;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005485
5486 return sptr;
5487}
5488
5489/*
5490 * free the given stack element
5491 */
5492 static void
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005493ff_free_stack_element(stack_ptr)
5494 ff_stack_T *stack_ptr;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005495{
5496 /* vim_free handles possible NULL pointers */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005497 vim_free(stack_ptr->ffs_fix_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005498#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005499 vim_free(stack_ptr->ffs_wc_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005500#endif
5501
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005502 if (stack_ptr->ffs_filearray != NULL)
5503 FreeWild(stack_ptr->ffs_filearray_size, stack_ptr->ffs_filearray);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005504
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005505 vim_free(stack_ptr);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005506}
5507
5508/*
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005509 * Clear the search context, but NOT the visited list.
Bram Moolenaar071d4272004-06-13 20:20:40 +00005510 */
5511 static void
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005512ff_clear(search_ctx)
5513 ff_search_ctx_T *search_ctx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005514{
5515 ff_stack_T *sptr;
5516
5517 /* clear up stack */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005518 while ((sptr = ff_pop(search_ctx)) != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005519 ff_free_stack_element(sptr);
5520
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005521 vim_free(search_ctx->ffsc_file_to_search);
5522 vim_free(search_ctx->ffsc_start_dir);
5523 vim_free(search_ctx->ffsc_fix_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005524#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005525 vim_free(search_ctx->ffsc_wc_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005526#endif
5527
5528#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005529 if (search_ctx->ffsc_stopdirs_v != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005530 {
5531 int i = 0;
5532
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005533 while (search_ctx->ffsc_stopdirs_v[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005534 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005535 vim_free(search_ctx->ffsc_stopdirs_v[i]);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005536 i++;
5537 }
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005538 vim_free(search_ctx->ffsc_stopdirs_v);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005539 }
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005540 search_ctx->ffsc_stopdirs_v = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005541#endif
5542
5543 /* reset everything */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005544 search_ctx->ffsc_file_to_search = NULL;
5545 search_ctx->ffsc_start_dir = NULL;
5546 search_ctx->ffsc_fix_path = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005547#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005548 search_ctx->ffsc_wc_path = NULL;
5549 search_ctx->ffsc_level = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005550#endif
5551}
5552
5553#ifdef FEAT_PATH_EXTRA
5554/*
5555 * check if the given path is in the stopdirs
5556 * returns TRUE if yes else FALSE
5557 */
5558 static int
5559ff_path_in_stoplist(path, path_len, stopdirs_v)
5560 char_u *path;
5561 int path_len;
5562 char_u **stopdirs_v;
5563{
5564 int i = 0;
5565
5566 /* eat up trailing path separators, except the first */
Bram Moolenaar1d94f9b2005-08-04 21:29:45 +00005567 while (path_len > 1 && vim_ispathsep(path[path_len - 1]))
Bram Moolenaar071d4272004-06-13 20:20:40 +00005568 path_len--;
5569
5570 /* if no path consider it as match */
5571 if (path_len == 0)
5572 return TRUE;
5573
5574 for (i = 0; stopdirs_v[i] != NULL; i++)
5575 {
5576 if ((int)STRLEN(stopdirs_v[i]) > path_len)
5577 {
5578 /* match for parent directory. So '/home' also matches
5579 * '/home/rks'. Check for PATHSEP in stopdirs_v[i], else
5580 * '/home/r' would also match '/home/rks'
5581 */
5582 if (fnamencmp(stopdirs_v[i], path, path_len) == 0
Bram Moolenaar1d94f9b2005-08-04 21:29:45 +00005583 && vim_ispathsep(stopdirs_v[i][path_len]))
Bram Moolenaar071d4272004-06-13 20:20:40 +00005584 return TRUE;
5585 }
5586 else
5587 {
5588 if (fnamecmp(stopdirs_v[i], path) == 0)
5589 return TRUE;
5590 }
5591 }
5592 return FALSE;
5593}
5594#endif
5595
5596#if defined(FEAT_SEARCHPATH) || defined(PROTO)
5597/*
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005598 * Find the file name "ptr[len]" in the path. Also finds directory names.
Bram Moolenaar071d4272004-06-13 20:20:40 +00005599 *
5600 * On the first call set the parameter 'first' to TRUE to initialize
5601 * the search. For repeating calls to FALSE.
5602 *
5603 * Repeating calls will return other files called 'ptr[len]' from the path.
5604 *
5605 * Only on the first call 'ptr' and 'len' are used. For repeating calls they
5606 * don't need valid values.
5607 *
5608 * If nothing found on the first call the option FNAME_MESS will issue the
5609 * message:
5610 * 'Can't find file "<file>" in path'
5611 * On repeating calls:
5612 * 'No more file "<file>" found in path'
5613 *
5614 * options:
5615 * FNAME_MESS give error message when not found
5616 *
5617 * Uses NameBuff[]!
5618 *
5619 * Returns an allocated string for the file name. NULL for error.
5620 *
5621 */
5622 char_u *
5623find_file_in_path(ptr, len, options, first, rel_fname)
5624 char_u *ptr; /* file name */
5625 int len; /* length of file name */
5626 int options;
5627 int first; /* use count'th matching file name */
5628 char_u *rel_fname; /* file name searching relative to */
5629{
5630 return find_file_in_path_option(ptr, len, options, first,
5631 *curbuf->b_p_path == NUL ? p_path : curbuf->b_p_path,
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005632 FINDFILE_BOTH, rel_fname, curbuf->b_p_sua);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005633}
5634
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005635static char_u *ff_file_to_find = NULL;
5636static void *fdip_search_ctx = NULL;
5637
5638#if defined(EXITFREE)
5639 static void
5640free_findfile()
5641{
5642 vim_free(ff_file_to_find);
5643 vim_findfile_cleanup(fdip_search_ctx);
5644}
5645#endif
5646
Bram Moolenaar071d4272004-06-13 20:20:40 +00005647/*
5648 * Find the directory name "ptr[len]" in the path.
5649 *
5650 * options:
5651 * FNAME_MESS give error message when not found
5652 *
5653 * Uses NameBuff[]!
5654 *
5655 * Returns an allocated string for the file name. NULL for error.
5656 */
5657 char_u *
5658find_directory_in_path(ptr, len, options, rel_fname)
5659 char_u *ptr; /* file name */
5660 int len; /* length of file name */
5661 int options;
5662 char_u *rel_fname; /* file name searching relative to */
5663{
5664 return find_file_in_path_option(ptr, len, options, TRUE, p_cdpath,
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005665 FINDFILE_DIR, rel_fname, (char_u *)"");
Bram Moolenaar071d4272004-06-13 20:20:40 +00005666}
5667
Bram Moolenaar89cb5e02004-07-19 20:55:54 +00005668 char_u *
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005669find_file_in_path_option(ptr, len, options, first, path_option, find_what, rel_fname, suffixes)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005670 char_u *ptr; /* file name */
5671 int len; /* length of file name */
5672 int options;
5673 int first; /* use count'th matching file name */
5674 char_u *path_option; /* p_path or p_cdpath */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005675 int find_what; /* FINDFILE_FILE, _DIR or _BOTH */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005676 char_u *rel_fname; /* file name we are looking relative to. */
Bram Moolenaar433f7c82006-03-21 21:29:36 +00005677 char_u *suffixes; /* list of suffixes, 'suffixesadd' option */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005678{
Bram Moolenaar071d4272004-06-13 20:20:40 +00005679 static char_u *dir;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005680 static int did_findfile_init = FALSE;
5681 char_u save_char;
5682 char_u *file_name = NULL;
5683 char_u *buf = NULL;
5684 int rel_to_curdir;
5685#ifdef AMIGA
5686 struct Process *proc = (struct Process *)FindTask(0L);
5687 APTR save_winptr = proc->pr_WindowPtr;
5688
5689 /* Avoid a requester here for a volume that doesn't exist. */
5690 proc->pr_WindowPtr = (APTR)-1L;
5691#endif
5692
5693 if (first == TRUE)
5694 {
5695 /* copy file name into NameBuff, expanding environment variables */
5696 save_char = ptr[len];
5697 ptr[len] = NUL;
5698 expand_env(ptr, NameBuff, MAXPATHL);
5699 ptr[len] = save_char;
5700
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005701 vim_free(ff_file_to_find);
5702 ff_file_to_find = vim_strsave(NameBuff);
5703 if (ff_file_to_find == NULL) /* out of memory */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005704 {
5705 file_name = NULL;
5706 goto theend;
5707 }
5708 }
5709
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005710 rel_to_curdir = (ff_file_to_find[0] == '.'
5711 && (ff_file_to_find[1] == NUL
5712 || vim_ispathsep(ff_file_to_find[1])
5713 || (ff_file_to_find[1] == '.'
5714 && (ff_file_to_find[2] == NUL
5715 || vim_ispathsep(ff_file_to_find[2])))));
5716 if (vim_isAbsName(ff_file_to_find)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005717 /* "..", "../path", "." and "./path": don't use the path_option */
5718 || rel_to_curdir
5719#if defined(MSWIN) || defined(MSDOS) || defined(OS2)
5720 /* handle "\tmp" as absolute path */
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005721 || vim_ispathsep(ff_file_to_find[0])
Bram Moolenaar21160b92009-11-11 15:56:10 +00005722 /* handle "c:name" as absolute path */
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005723 || (ff_file_to_find[0] != NUL && ff_file_to_find[1] == ':')
Bram Moolenaar071d4272004-06-13 20:20:40 +00005724#endif
5725#ifdef AMIGA
5726 /* handle ":tmp" as absolute path */
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005727 || ff_file_to_find[0] == ':'
Bram Moolenaar071d4272004-06-13 20:20:40 +00005728#endif
5729 )
5730 {
5731 /*
5732 * Absolute path, no need to use "path_option".
5733 * If this is not a first call, return NULL. We already returned a
5734 * filename on the first call.
5735 */
5736 if (first == TRUE)
5737 {
5738 int l;
5739 int run;
5740
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005741 if (path_with_url(ff_file_to_find))
Bram Moolenaar071d4272004-06-13 20:20:40 +00005742 {
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005743 file_name = vim_strsave(ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005744 goto theend;
5745 }
5746
5747 /* When FNAME_REL flag given first use the directory of the file.
5748 * Otherwise or when this fails use the current directory. */
5749 for (run = 1; run <= 2; ++run)
5750 {
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005751 l = (int)STRLEN(ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005752 if (run == 1
5753 && rel_to_curdir
5754 && (options & FNAME_REL)
5755 && rel_fname != NULL
5756 && STRLEN(rel_fname) + l < MAXPATHL)
5757 {
5758 STRCPY(NameBuff, rel_fname);
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005759 STRCPY(gettail(NameBuff), ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005760 l = (int)STRLEN(NameBuff);
5761 }
5762 else
5763 {
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005764 STRCPY(NameBuff, ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005765 run = 2;
5766 }
5767
5768 /* When the file doesn't exist, try adding parts of
5769 * 'suffixesadd'. */
Bram Moolenaar433f7c82006-03-21 21:29:36 +00005770 buf = suffixes;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005771 for (;;)
5772 {
5773 if (
5774#ifdef DJGPP
5775 /* "C:" by itself will fail for mch_getperm(),
5776 * assume it's always valid. */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005777 (find_what != FINDFILE_FILE && NameBuff[0] != NUL
Bram Moolenaar071d4272004-06-13 20:20:40 +00005778 && NameBuff[1] == ':'
5779 && NameBuff[2] == NUL) ||
5780#endif
5781 (mch_getperm(NameBuff) >= 0
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005782 && (find_what == FINDFILE_BOTH
5783 || ((find_what == FINDFILE_DIR)
5784 == mch_isdir(NameBuff)))))
Bram Moolenaar071d4272004-06-13 20:20:40 +00005785 {
5786 file_name = vim_strsave(NameBuff);
5787 goto theend;
5788 }
5789 if (*buf == NUL)
5790 break;
5791 copy_option_part(&buf, NameBuff + l, MAXPATHL - l, ",");
5792 }
5793 }
5794 }
5795 }
5796 else
5797 {
5798 /*
5799 * Loop over all paths in the 'path' or 'cdpath' option.
5800 * When "first" is set, first setup to the start of the option.
5801 * Otherwise continue to find the next match.
5802 */
5803 if (first == TRUE)
5804 {
5805 /* vim_findfile_free_visited can handle a possible NULL pointer */
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005806 vim_findfile_free_visited(fdip_search_ctx);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005807 dir = path_option;
5808 did_findfile_init = FALSE;
5809 }
5810
5811 for (;;)
5812 {
5813 if (did_findfile_init)
5814 {
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005815 file_name = vim_findfile(fdip_search_ctx);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005816 if (file_name != NULL)
5817 break;
5818
5819 did_findfile_init = FALSE;
5820 }
5821 else
5822 {
5823 char_u *r_ptr;
5824
5825 if (dir == NULL || *dir == NUL)
5826 {
5827 /* We searched all paths of the option, now we can
5828 * free the search context. */
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005829 vim_findfile_cleanup(fdip_search_ctx);
5830 fdip_search_ctx = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005831 break;
5832 }
5833
5834 if ((buf = alloc((int)(MAXPATHL))) == NULL)
5835 break;
5836
5837 /* copy next path */
5838 buf[0] = 0;
5839 copy_option_part(&dir, buf, MAXPATHL, " ,");
5840
5841#ifdef FEAT_PATH_EXTRA
5842 /* get the stopdir string */
5843 r_ptr = vim_findfile_stopdir(buf);
5844#else
5845 r_ptr = NULL;
5846#endif
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005847 fdip_search_ctx = vim_findfile_init(buf, ff_file_to_find,
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005848 r_ptr, 100, FALSE, find_what,
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005849 fdip_search_ctx, FALSE, rel_fname);
5850 if (fdip_search_ctx != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005851 did_findfile_init = TRUE;
5852 vim_free(buf);
5853 }
5854 }
5855 }
5856 if (file_name == NULL && (options & FNAME_MESS))
5857 {
5858 if (first == TRUE)
5859 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005860 if (find_what == FINDFILE_DIR)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005861 EMSG2(_("E344: Can't find directory \"%s\" in cdpath"),
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005862 ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005863 else
5864 EMSG2(_("E345: Can't find file \"%s\" in path"),
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005865 ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005866 }
5867 else
5868 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005869 if (find_what == FINDFILE_DIR)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005870 EMSG2(_("E346: No more directory \"%s\" found in cdpath"),
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005871 ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005872 else
5873 EMSG2(_("E347: No more file \"%s\" found in path"),
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005874 ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005875 }
5876 }
5877
5878theend:
5879#ifdef AMIGA
5880 proc->pr_WindowPtr = save_winptr;
5881#endif
5882 return file_name;
5883}
5884
5885#endif /* FEAT_SEARCHPATH */
5886
5887/*
5888 * Change directory to "new_dir". If FEAT_SEARCHPATH is defined, search
5889 * 'cdpath' for relative directory names, otherwise just mch_chdir().
5890 */
5891 int
5892vim_chdir(new_dir)
5893 char_u *new_dir;
5894{
5895#ifndef FEAT_SEARCHPATH
5896 return mch_chdir((char *)new_dir);
5897#else
5898 char_u *dir_name;
5899 int r;
5900
5901 dir_name = find_directory_in_path(new_dir, (int)STRLEN(new_dir),
5902 FNAME_MESS, curbuf->b_ffname);
5903 if (dir_name == NULL)
5904 return -1;
5905 r = mch_chdir((char *)dir_name);
5906 vim_free(dir_name);
5907 return r;
5908#endif
5909}
5910
5911/*
Bram Moolenaarbbebc852005-07-18 21:47:53 +00005912 * Get user name from machine-specific function.
Bram Moolenaar071d4272004-06-13 20:20:40 +00005913 * Returns the user name in "buf[len]".
Bram Moolenaarbbebc852005-07-18 21:47:53 +00005914 * Some systems are quite slow in obtaining the user name (Windows NT), thus
5915 * cache the result.
Bram Moolenaar071d4272004-06-13 20:20:40 +00005916 * Returns OK or FAIL.
5917 */
5918 int
5919get_user_name(buf, len)
5920 char_u *buf;
5921 int len;
5922{
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005923 if (username == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005924 {
5925 if (mch_get_user_name(buf, len) == FAIL)
5926 return FAIL;
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005927 username = vim_strsave(buf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005928 }
5929 else
Bram Moolenaarbbebc852005-07-18 21:47:53 +00005930 vim_strncpy(buf, username, len - 1);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005931 return OK;
5932}
5933
5934#ifndef HAVE_QSORT
5935/*
5936 * Our own qsort(), for systems that don't have it.
5937 * It's simple and slow. From the K&R C book.
5938 */
5939 void
5940qsort(base, elm_count, elm_size, cmp)
5941 void *base;
5942 size_t elm_count;
5943 size_t elm_size;
5944 int (*cmp) __ARGS((const void *, const void *));
5945{
5946 char_u *buf;
5947 char_u *p1;
5948 char_u *p2;
5949 int i, j;
5950 int gap;
5951
5952 buf = alloc((unsigned)elm_size);
5953 if (buf == NULL)
5954 return;
5955
5956 for (gap = elm_count / 2; gap > 0; gap /= 2)
5957 for (i = gap; i < elm_count; ++i)
5958 for (j = i - gap; j >= 0; j -= gap)
5959 {
5960 /* Compare the elements. */
5961 p1 = (char_u *)base + j * elm_size;
5962 p2 = (char_u *)base + (j + gap) * elm_size;
5963 if ((*cmp)((void *)p1, (void *)p2) <= 0)
5964 break;
Bram Moolenaar21160b92009-11-11 15:56:10 +00005965 /* Exchange the elements. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005966 mch_memmove(buf, p1, elm_size);
5967 mch_memmove(p1, p2, elm_size);
5968 mch_memmove(p2, buf, elm_size);
5969 }
5970
5971 vim_free(buf);
5972}
5973#endif
5974
Bram Moolenaar071d4272004-06-13 20:20:40 +00005975/*
5976 * Sort an array of strings.
5977 */
5978static int
5979#ifdef __BORLANDC__
5980_RTLENTRYF
5981#endif
5982sort_compare __ARGS((const void *s1, const void *s2));
5983
5984 static int
5985#ifdef __BORLANDC__
5986_RTLENTRYF
5987#endif
5988sort_compare(s1, s2)
5989 const void *s1;
5990 const void *s2;
5991{
5992 return STRCMP(*(char **)s1, *(char **)s2);
5993}
5994
5995 void
5996sort_strings(files, count)
5997 char_u **files;
5998 int count;
5999{
6000 qsort((void *)files, (size_t)count, sizeof(char_u *), sort_compare);
6001}
Bram Moolenaar071d4272004-06-13 20:20:40 +00006002
6003#if !defined(NO_EXPANDPATH) || defined(PROTO)
6004/*
6005 * Compare path "p[]" to "q[]".
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00006006 * If "maxlen" >= 0 compare "p[maxlen]" to "q[maxlen]"
Bram Moolenaar071d4272004-06-13 20:20:40 +00006007 * Return value like strcmp(p, q), but consider path separators.
6008 */
6009 int
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00006010pathcmp(p, q, maxlen)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006011 const char *p, *q;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00006012 int maxlen;
Bram Moolenaar071d4272004-06-13 20:20:40 +00006013{
6014 int i;
Bram Moolenaar86b68352004-12-27 21:59:20 +00006015 const char *s = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00006016
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00006017 for (i = 0; maxlen < 0 || i < maxlen; ++i)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006018 {
6019 /* End of "p": check if "q" also ends or just has a slash. */
6020 if (p[i] == NUL)
6021 {
6022 if (q[i] == NUL) /* full match */
6023 return 0;
6024 s = q;
6025 break;
6026 }
6027
6028 /* End of "q": check if "p" just has a slash. */
6029 if (q[i] == NUL)
6030 {
6031 s = p;
6032 break;
6033 }
6034
6035 if (
6036#ifdef CASE_INSENSITIVE_FILENAME
6037 TOUPPER_LOC(p[i]) != TOUPPER_LOC(q[i])
6038#else
6039 p[i] != q[i]
6040#endif
6041#ifdef BACKSLASH_IN_FILENAME
6042 /* consider '/' and '\\' to be equal */
6043 && !((p[i] == '/' && q[i] == '\\')
6044 || (p[i] == '\\' && q[i] == '/'))
6045#endif
6046 )
6047 {
6048 if (vim_ispathsep(p[i]))
6049 return -1;
6050 if (vim_ispathsep(q[i]))
6051 return 1;
6052 return ((char_u *)p)[i] - ((char_u *)q)[i]; /* no match */
6053 }
6054 }
Bram Moolenaar86b68352004-12-27 21:59:20 +00006055 if (s == NULL) /* "i" ran into "maxlen" */
6056 return 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00006057
6058 /* ignore a trailing slash, but not "//" or ":/" */
Bram Moolenaar86b68352004-12-27 21:59:20 +00006059 if (s[i + 1] == NUL
6060 && i > 0
6061 && !after_pathsep((char_u *)s, (char_u *)s + i)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006062#ifdef BACKSLASH_IN_FILENAME
Bram Moolenaar86b68352004-12-27 21:59:20 +00006063 && (s[i] == '/' || s[i] == '\\')
Bram Moolenaar071d4272004-06-13 20:20:40 +00006064#else
Bram Moolenaar86b68352004-12-27 21:59:20 +00006065 && s[i] == '/'
Bram Moolenaar071d4272004-06-13 20:20:40 +00006066#endif
Bram Moolenaar86b68352004-12-27 21:59:20 +00006067 )
Bram Moolenaar071d4272004-06-13 20:20:40 +00006068 return 0; /* match with trailing slash */
6069 if (s == q)
6070 return -1; /* no match */
6071 return 1;
6072}
6073#endif
6074
Bram Moolenaar071d4272004-06-13 20:20:40 +00006075/*
6076 * The putenv() implementation below comes from the "screen" program.
6077 * Included with permission from Juergen Weigert.
6078 * See pty.c for the copyright notice.
6079 */
6080
6081/*
6082 * putenv -- put value into environment
6083 *
6084 * Usage: i = putenv (string)
6085 * int i;
6086 * char *string;
6087 *
6088 * where string is of the form <name>=<value>.
6089 * Putenv returns 0 normally, -1 on error (not enough core for malloc).
6090 *
6091 * Putenv may need to add a new name into the environment, or to
6092 * associate a value longer than the current value with a particular
6093 * name. So, to make life simpler, putenv() copies your entire
6094 * environment into the heap (i.e. malloc()) from the stack
6095 * (i.e. where it resides when your process is initiated) the first
6096 * time you call it.
6097 *
6098 * (history removed, not very interesting. See the "screen" sources.)
6099 */
6100
6101#if !defined(HAVE_SETENV) && !defined(HAVE_PUTENV)
6102
6103#define EXTRASIZE 5 /* increment to add to env. size */
6104
6105static int envsize = -1; /* current size of environment */
6106#ifndef MACOS_CLASSIC
6107extern
6108#endif
6109 char **environ; /* the global which is your env. */
6110
6111static int findenv __ARGS((char *name)); /* look for a name in the env. */
6112static int newenv __ARGS((void)); /* copy env. from stack to heap */
6113static int moreenv __ARGS((void)); /* incr. size of env. */
6114
6115 int
6116putenv(string)
6117 const char *string;
6118{
6119 int i;
6120 char *p;
6121
6122 if (envsize < 0)
6123 { /* first time putenv called */
6124 if (newenv() < 0) /* copy env. to heap */
6125 return -1;
6126 }
6127
6128 i = findenv((char *)string); /* look for name in environment */
6129
6130 if (i < 0)
6131 { /* name must be added */
6132 for (i = 0; environ[i]; i++);
6133 if (i >= (envsize - 1))
6134 { /* need new slot */
6135 if (moreenv() < 0)
6136 return -1;
6137 }
6138 p = (char *)alloc((unsigned)(strlen(string) + 1));
6139 if (p == NULL) /* not enough core */
6140 return -1;
6141 environ[i + 1] = 0; /* new end of env. */
6142 }
6143 else
6144 { /* name already in env. */
6145 p = vim_realloc(environ[i], strlen(string) + 1);
6146 if (p == NULL)
6147 return -1;
6148 }
6149 sprintf(p, "%s", string); /* copy into env. */
6150 environ[i] = p;
6151
6152 return 0;
6153}
6154
6155 static int
6156findenv(name)
6157 char *name;
6158{
6159 char *namechar, *envchar;
6160 int i, found;
6161
6162 found = 0;
6163 for (i = 0; environ[i] && !found; i++)
6164 {
6165 envchar = environ[i];
6166 namechar = name;
6167 while (*namechar && *namechar != '=' && (*namechar == *envchar))
6168 {
6169 namechar++;
6170 envchar++;
6171 }
6172 found = ((*namechar == '\0' || *namechar == '=') && *envchar == '=');
6173 }
6174 return found ? i - 1 : -1;
6175}
6176
6177 static int
6178newenv()
6179{
6180 char **env, *elem;
6181 int i, esize;
6182
6183#ifdef MACOS
6184 /* for Mac a new, empty environment is created */
6185 i = 0;
6186#else
6187 for (i = 0; environ[i]; i++)
6188 ;
6189#endif
6190 esize = i + EXTRASIZE + 1;
6191 env = (char **)alloc((unsigned)(esize * sizeof (elem)));
6192 if (env == NULL)
6193 return -1;
6194
6195#ifndef MACOS
6196 for (i = 0; environ[i]; i++)
6197 {
6198 elem = (char *)alloc((unsigned)(strlen(environ[i]) + 1));
6199 if (elem == NULL)
6200 return -1;
6201 env[i] = elem;
6202 strcpy(elem, environ[i]);
6203 }
6204#endif
6205
6206 env[i] = 0;
6207 environ = env;
6208 envsize = esize;
6209 return 0;
6210}
6211
6212 static int
6213moreenv()
6214{
6215 int esize;
6216 char **env;
6217
6218 esize = envsize + EXTRASIZE;
6219 env = (char **)vim_realloc((char *)environ, esize * sizeof (*env));
6220 if (env == 0)
6221 return -1;
6222 environ = env;
6223 envsize = esize;
6224 return 0;
6225}
6226
6227# ifdef USE_VIMPTY_GETENV
6228 char_u *
6229vimpty_getenv(string)
6230 const char_u *string;
6231{
6232 int i;
6233 char_u *p;
6234
6235 if (envsize < 0)
6236 return NULL;
6237
6238 i = findenv((char *)string);
6239
6240 if (i < 0)
6241 return NULL;
6242
6243 p = vim_strchr((char_u *)environ[i], '=');
6244 return (p + 1);
6245}
6246# endif
6247
6248#endif /* !defined(HAVE_SETENV) && !defined(HAVE_PUTENV) */
Bram Moolenaarc4a06d32005-06-07 21:04:49 +00006249
Bram Moolenaarc4956c82006-03-12 21:58:43 +00006250#if defined(FEAT_EVAL) || defined(FEAT_SPELL) || defined(PROTO)
Bram Moolenaarc4a06d32005-06-07 21:04:49 +00006251/*
6252 * Return 0 for not writable, 1 for writable file, 2 for a dir which we have
6253 * rights to write into.
6254 */
6255 int
6256filewritable(fname)
6257 char_u *fname;
6258{
6259 int retval = 0;
6260#if defined(UNIX) || defined(VMS)
6261 int perm = 0;
6262#endif
6263
6264#if defined(UNIX) || defined(VMS)
6265 perm = mch_getperm(fname);
6266#endif
6267#ifndef MACOS_CLASSIC /* TODO: get either mch_writable or mch_access */
6268 if (
6269# ifdef WIN3264
6270 mch_writable(fname) &&
6271# else
6272# if defined(UNIX) || defined(VMS)
6273 (perm & 0222) &&
6274# endif
6275# endif
6276 mch_access((char *)fname, W_OK) == 0
6277 )
6278#endif
6279 {
6280 ++retval;
6281 if (mch_isdir(fname))
6282 ++retval;
6283 }
6284 return retval;
6285}
6286#endif
Bram Moolenaar6bab4d12005-06-16 21:53:56 +00006287
6288/*
6289 * Print an error message with one or two "%s" and one or two string arguments.
6290 * This is not in message.c to avoid a warning for prototypes.
6291 */
6292 int
6293emsg3(s, a1, a2)
6294 char_u *s, *a1, *a2;
6295{
Bram Moolenaareb3593b2006-04-22 22:33:57 +00006296 if (emsg_not_now())
Bram Moolenaar6bab4d12005-06-16 21:53:56 +00006297 return TRUE; /* no error messages at the moment */
Bram Moolenaar6f192452007-11-08 19:49:02 +00006298#ifdef HAVE_STDARG_H
6299 vim_snprintf((char *)IObuff, IOSIZE, (char *)s, a1, a2);
6300#else
Bram Moolenaareb3593b2006-04-22 22:33:57 +00006301 vim_snprintf((char *)IObuff, IOSIZE, (char *)s, (long_u)a1, (long_u)a2);
Bram Moolenaar6f192452007-11-08 19:49:02 +00006302#endif
Bram Moolenaar6bab4d12005-06-16 21:53:56 +00006303 return emsg(IObuff);
6304}
6305
6306/*
6307 * Print an error message with one "%ld" and one long int argument.
6308 * This is not in message.c to avoid a warning for prototypes.
6309 */
6310 int
6311emsgn(s, n)
6312 char_u *s;
6313 long n;
6314{
Bram Moolenaareb3593b2006-04-22 22:33:57 +00006315 if (emsg_not_now())
Bram Moolenaar6bab4d12005-06-16 21:53:56 +00006316 return TRUE; /* no error messages at the moment */
6317 vim_snprintf((char *)IObuff, IOSIZE, (char *)s, n);
6318 return emsg(IObuff);
6319}
Bram Moolenaarcdf04202010-05-29 15:11:47 +02006320
6321#if defined(FEAT_SPELL) || defined(FEAT_PERSISTENT_UNDO) || defined(PROTO)
6322/*
6323 * Read 2 bytes from "fd" and turn them into an int, MSB first.
6324 */
6325 int
6326get2c(fd)
6327 FILE *fd;
6328{
Bram Moolenaar9db58062010-05-29 20:33:07 +02006329 int n;
Bram Moolenaarcdf04202010-05-29 15:11:47 +02006330
6331 n = getc(fd);
6332 n = (n << 8) + getc(fd);
6333 return n;
6334}
6335
6336/*
6337 * Read 3 bytes from "fd" and turn them into an int, MSB first.
6338 */
6339 int
6340get3c(fd)
6341 FILE *fd;
6342{
Bram Moolenaar9db58062010-05-29 20:33:07 +02006343 int n;
Bram Moolenaarcdf04202010-05-29 15:11:47 +02006344
6345 n = getc(fd);
6346 n = (n << 8) + getc(fd);
6347 n = (n << 8) + getc(fd);
6348 return n;
6349}
6350
6351/*
6352 * Read 4 bytes from "fd" and turn them into an int, MSB first.
6353 */
6354 int
6355get4c(fd)
6356 FILE *fd;
6357{
Bram Moolenaar9db58062010-05-29 20:33:07 +02006358 int n;
Bram Moolenaarcdf04202010-05-29 15:11:47 +02006359
6360 n = getc(fd);
6361 n = (n << 8) + getc(fd);
6362 n = (n << 8) + getc(fd);
6363 n = (n << 8) + getc(fd);
6364 return n;
6365}
6366
6367/*
6368 * Read 8 bytes from "fd" and turn them into a time_t, MSB first.
6369 */
6370 time_t
6371get8ctime(fd)
6372 FILE *fd;
6373{
6374 time_t n = 0;
6375 int i;
6376
6377 for (i = 0; i < 8; ++i)
6378 n = (n << 8) + getc(fd);
6379 return n;
6380}
6381
6382/*
6383 * Read a string of length "cnt" from "fd" into allocated memory.
6384 * Returns NULL when out of memory or unable to read that many bytes.
6385 */
6386 char_u *
6387read_string(fd, cnt)
6388 FILE *fd;
6389 int cnt;
6390{
6391 char_u *str;
6392 int i;
6393 int c;
6394
6395 /* allocate memory */
6396 str = alloc((unsigned)cnt + 1);
6397 if (str != NULL)
6398 {
6399 /* Read the string. Quit when running into the EOF. */
6400 for (i = 0; i < cnt; ++i)
6401 {
6402 c = getc(fd);
6403 if (c == EOF)
6404 {
6405 vim_free(str);
6406 return NULL;
6407 }
6408 str[i] = c;
6409 }
6410 str[i] = NUL;
6411 }
6412 return str;
6413}
6414
6415/*
6416 * Write a number to file "fd", MSB first, in "len" bytes.
6417 */
6418 int
6419put_bytes(fd, nr, len)
6420 FILE *fd;
6421 long_u nr;
6422 int len;
6423{
6424 int i;
6425
6426 for (i = len - 1; i >= 0; --i)
6427 if (putc((int)(nr >> (i * 8)), fd) == EOF)
6428 return FAIL;
6429 return OK;
6430}
6431
6432#ifdef _MSC_VER
6433# if (_MSC_VER <= 1200)
6434/* This line is required for VC6 without the service pack. Also see the
6435 * matching #pragma below. */
6436 # pragma optimize("", off)
6437# endif
6438#endif
6439
6440/*
6441 * Write time_t to file "fd" in 8 bytes.
6442 */
6443 void
6444put_time(fd, the_time)
6445 FILE *fd;
6446 time_t the_time;
6447{
6448 int c;
6449 int i;
6450 time_t wtime = the_time;
6451
6452 /* time_t can be up to 8 bytes in size, more than long_u, thus we
6453 * can't use put_bytes() here.
6454 * Another problem is that ">>" may do an arithmetic shift that keeps the
Bram Moolenaar644fdff2010-05-30 13:26:21 +02006455 * sign. This happens for large values of wtime. A cast to long_u may
6456 * truncate if time_t is 8 bytes. So only use a cast when it is 4 bytes,
6457 * it's safe to assume that long_u is 4 bytes or more and when using 8
6458 * bytes the top bit won't be set. */
Bram Moolenaarcdf04202010-05-29 15:11:47 +02006459 for (i = 7; i >= 0; --i)
6460 {
6461 if (i + 1 > (int)sizeof(time_t))
6462 /* ">>" doesn't work well when shifting more bits than avail */
6463 putc(0, fd);
6464 else
6465 {
Bram Moolenaar644fdff2010-05-30 13:26:21 +02006466#if defined(SIZEOF_TIME_T) && SIZEOF_TIME_T > 4
Bram Moolenaarbbd6afe2010-06-02 20:32:23 +02006467 c = (int)(wtime >> (i * 8));
Bram Moolenaar644fdff2010-05-30 13:26:21 +02006468#else
Bram Moolenaarbbd6afe2010-06-02 20:32:23 +02006469 c = (int)((long_u)wtime >> (i * 8));
Bram Moolenaar644fdff2010-05-30 13:26:21 +02006470#endif
Bram Moolenaarcdf04202010-05-29 15:11:47 +02006471 putc(c, fd);
6472 }
6473 }
6474}
6475
6476#ifdef _MSC_VER
6477# if (_MSC_VER <= 1200)
6478 # pragma optimize("", on)
6479# endif
6480#endif
6481
6482#endif