blob: 04f038b9a265f4d5168394e8090c21fb614e37f8 [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 Moolenaar071d4272004-06-13 20:20:40 +0000159 line = ml_get_curline();
160
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
203 idx = -1;
204 ptr = line;
205 while (col <= wcol && *ptr != NUL)
206 {
207 /* Count a tab for what it's worth (if list mode not on) */
208#ifdef FEAT_LINEBREAK
209 csize = win_lbr_chartabsize(curwin, ptr, col, &head);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +0000210 mb_ptr_adv(ptr);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000211#else
212 csize = lbr_chartabsize_adv(&ptr, col);
213#endif
214 col += csize;
215 }
216 idx = (int)(ptr - line);
217 /*
218 * Handle all the special cases. The virtual_active() check
219 * is needed to ensure that a virtual position off the end of
220 * a line has the correct indexing. The one_more comparison
221 * replaces an explicit add of one_more later on.
222 */
223 if (col > wcol || (!virtual_active() && one_more == 0))
224 {
225 idx -= 1;
226# ifdef FEAT_LINEBREAK
227 /* Don't count the chars from 'showbreak'. */
228 csize -= head;
229# endif
230 col -= csize;
231 }
232
233#ifdef FEAT_VIRTUALEDIT
234 if (virtual_active()
235 && addspaces
236 && ((col != wcol && col != wcol + 1) || csize > 1))
237 {
238 /* 'virtualedit' is set: The difference between wcol and col is
239 * filled with spaces. */
240
241 if (line[idx] == NUL)
242 {
243 /* Append spaces */
244 int correct = wcol - col;
245 char_u *newline = alloc(idx + correct + 1);
246 int t;
247
248 if (newline == NULL)
249 return FAIL;
250
251 for (t = 0; t < idx; ++t)
252 newline[t] = line[t];
253
254 for (t = 0; t < correct; ++t)
255 newline[t + idx] = ' ';
256
257 newline[idx + correct] = NUL;
258
259 ml_replace(pos->lnum, newline, FALSE);
260 changed_bytes(pos->lnum, (colnr_T)idx);
261 idx += correct;
262 col = wcol;
263 }
264 else
265 {
266 /* Break a tab */
267 int linelen = (int)STRLEN(line);
268 int correct = wcol - col - csize + 1; /* negative!! */
Bram Moolenaareb3593b2006-04-22 22:33:57 +0000269 char_u *newline;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000270 int t, s = 0;
271 int v;
272
Bram Moolenaareb3593b2006-04-22 22:33:57 +0000273 if (-correct > csize)
274 return FAIL;
275
276 newline = alloc(linelen + csize);
277 if (newline == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000278 return FAIL;
279
280 for (t = 0; t < linelen; t++)
281 {
282 if (t != idx)
283 newline[s++] = line[t];
284 else
285 for (v = 0; v < csize; v++)
286 newline[s++] = ' ';
287 }
288
289 newline[linelen + csize - 1] = NUL;
290
291 ml_replace(pos->lnum, newline, FALSE);
292 changed_bytes(pos->lnum, idx);
293 idx += (csize - 1 + correct);
294 col += correct;
295 }
296 }
297#endif
298 }
299
300 if (idx < 0)
301 pos->col = 0;
302 else
303 pos->col = idx;
304
305#ifdef FEAT_VIRTUALEDIT
306 pos->coladd = 0;
307
308 if (finetune)
309 {
310 if (wcol == MAXCOL)
311 {
312 /* The width of the last character is used to set coladd. */
313 if (!one_more)
314 {
315 colnr_T scol, ecol;
316
317 getvcol(curwin, pos, &scol, NULL, &ecol);
318 pos->coladd = ecol - scol;
319 }
320 }
321 else
322 {
323 int b = (int)wcol - (int)col;
324
325 /* The difference between wcol and col is used to set coladd. */
326 if (b > 0 && b < (MAXCOL - 2 * W_WIDTH(curwin)))
327 pos->coladd = b;
328
329 col += b;
330 }
331 }
332#endif
333
334#ifdef FEAT_MBYTE
335 /* prevent cursor from moving on the trail byte */
336 if (has_mbyte)
337 mb_adjust_cursor();
338#endif
339
340 if (col < wcol)
341 return FAIL;
342 return OK;
343}
344
345/*
Bram Moolenaar446cb832008-06-24 21:56:24 +0000346 * Increment the cursor position. See inc() for return values.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000347 */
348 int
349inc_cursor()
350{
351 return inc(&curwin->w_cursor);
352}
353
Bram Moolenaar446cb832008-06-24 21:56:24 +0000354/*
355 * Increment the line pointer "lp" crossing line boundaries as necessary.
356 * Return 1 when going to the next line.
357 * Return 2 when moving forward onto a NUL at the end of the line).
358 * Return -1 when at the end of file.
359 * Return 0 otherwise.
360 */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000361 int
362inc(lp)
363 pos_T *lp;
364{
365 char_u *p = ml_get_pos(lp);
366
367 if (*p != NUL) /* still within line, move to next char (may be NUL) */
368 {
369#ifdef FEAT_MBYTE
370 if (has_mbyte)
371 {
Bram Moolenaar0fa313a2005-08-10 21:07:57 +0000372 int l = (*mb_ptr2len)(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000373
374 lp->col += l;
375 return ((p[l] != NUL) ? 0 : 2);
376 }
377#endif
378 lp->col++;
379#ifdef FEAT_VIRTUALEDIT
380 lp->coladd = 0;
381#endif
382 return ((p[1] != NUL) ? 0 : 2);
383 }
384 if (lp->lnum != curbuf->b_ml.ml_line_count) /* there is a next line */
385 {
386 lp->col = 0;
387 lp->lnum++;
388#ifdef FEAT_VIRTUALEDIT
389 lp->coladd = 0;
390#endif
391 return 1;
392 }
393 return -1;
394}
395
396/*
397 * incl(lp): same as inc(), but skip the NUL at the end of non-empty lines
398 */
399 int
400incl(lp)
401 pos_T *lp;
402{
403 int r;
404
405 if ((r = inc(lp)) >= 1 && lp->col)
406 r = inc(lp);
407 return r;
408}
409
410/*
411 * dec(p)
412 *
413 * Decrement the line pointer 'p' crossing line boundaries as necessary.
414 * Return 1 when crossing a line, -1 when at start of file, 0 otherwise.
415 */
416 int
417dec_cursor()
418{
419 return dec(&curwin->w_cursor);
420}
421
422 int
423dec(lp)
424 pos_T *lp;
425{
426 char_u *p;
427
428#ifdef FEAT_VIRTUALEDIT
429 lp->coladd = 0;
430#endif
431 if (lp->col > 0) /* still within line */
432 {
433 lp->col--;
434#ifdef FEAT_MBYTE
435 if (has_mbyte)
436 {
437 p = ml_get(lp->lnum);
438 lp->col -= (*mb_head_off)(p, p + lp->col);
439 }
440#endif
441 return 0;
442 }
443 if (lp->lnum > 1) /* there is a prior line */
444 {
445 lp->lnum--;
446 p = ml_get(lp->lnum);
447 lp->col = (colnr_T)STRLEN(p);
448#ifdef FEAT_MBYTE
449 if (has_mbyte)
450 lp->col -= (*mb_head_off)(p, p + lp->col);
451#endif
452 return 1;
453 }
454 return -1; /* at start of file */
455}
456
457/*
458 * decl(lp): same as dec(), but skip the NUL at the end of non-empty lines
459 */
460 int
461decl(lp)
462 pos_T *lp;
463{
464 int r;
465
466 if ((r = dec(lp)) == 1 && lp->col)
467 r = dec(lp);
468 return r;
469}
470
471/*
472 * Make sure curwin->w_cursor.lnum is valid.
473 */
474 void
475check_cursor_lnum()
476{
477 if (curwin->w_cursor.lnum > curbuf->b_ml.ml_line_count)
478 {
479#ifdef FEAT_FOLDING
480 /* If there is a closed fold at the end of the file, put the cursor in
481 * its first line. Otherwise in the last line. */
482 if (!hasFolding(curbuf->b_ml.ml_line_count,
483 &curwin->w_cursor.lnum, NULL))
484#endif
485 curwin->w_cursor.lnum = curbuf->b_ml.ml_line_count;
486 }
487 if (curwin->w_cursor.lnum <= 0)
488 curwin->w_cursor.lnum = 1;
489}
490
491/*
492 * Make sure curwin->w_cursor.col is valid.
493 */
494 void
495check_cursor_col()
496{
497 colnr_T len;
498#ifdef FEAT_VIRTUALEDIT
499 colnr_T oldcol = curwin->w_cursor.col + curwin->w_cursor.coladd;
500#endif
501
502 len = (colnr_T)STRLEN(ml_get_curline());
503 if (len == 0)
504 curwin->w_cursor.col = 0;
505 else if (curwin->w_cursor.col >= len)
506 {
Bram Moolenaar33f54432008-01-04 20:25:44 +0000507 /* Allow cursor past end-of-line when:
508 * - in Insert mode or restarting Insert mode
509 * - in Visual mode and 'selection' isn't "old"
510 * - 'virtualedit' is set */
Bram Moolenaarebefac62005-12-28 22:39:57 +0000511 if ((State & INSERT) || restart_edit
Bram Moolenaar071d4272004-06-13 20:20:40 +0000512#ifdef FEAT_VISUAL
513 || (VIsual_active && *p_sel != 'o')
514#endif
Bram Moolenaar33f54432008-01-04 20:25:44 +0000515#ifdef FEAT_VIRTUALEDIT
516 || (ve_flags & VE_ONEMORE)
517#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000518 || virtual_active())
519 curwin->w_cursor.col = len;
520 else
Bram Moolenaar87c19962007-04-26 08:54:21 +0000521 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000522 curwin->w_cursor.col = len - 1;
Bram Moolenaar87c19962007-04-26 08:54:21 +0000523#ifdef FEAT_MBYTE
524 /* prevent cursor from moving on the trail byte */
525 if (has_mbyte)
526 mb_adjust_cursor();
527#endif
528 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000529 }
530
531#ifdef FEAT_VIRTUALEDIT
532 /* If virtual editing is on, we can leave the cursor on the old position,
533 * only we must set it to virtual. But don't do it when at the end of the
534 * line. */
535 if (oldcol == MAXCOL)
536 curwin->w_cursor.coladd = 0;
537 else if (ve_flags == VE_ALL)
538 curwin->w_cursor.coladd = oldcol - curwin->w_cursor.col;
539#endif
540}
541
542/*
543 * make sure curwin->w_cursor in on a valid character
544 */
545 void
546check_cursor()
547{
548 check_cursor_lnum();
549 check_cursor_col();
550}
551
552#if defined(FEAT_TEXTOBJ) || defined(PROTO)
553/*
554 * Make sure curwin->w_cursor is not on the NUL at the end of the line.
555 * Allow it when in Visual mode and 'selection' is not "old".
556 */
557 void
558adjust_cursor_col()
559{
560 if (curwin->w_cursor.col > 0
561# ifdef FEAT_VISUAL
562 && (!VIsual_active || *p_sel == 'o')
563# endif
564 && gchar_cursor() == NUL)
565 --curwin->w_cursor.col;
566}
567#endif
568
569/*
570 * When curwin->w_leftcol has changed, adjust the cursor position.
571 * Return TRUE if the cursor was moved.
572 */
573 int
574leftcol_changed()
575{
576 long lastcol;
577 colnr_T s, e;
578 int retval = FALSE;
579
580 changed_cline_bef_curs();
581 lastcol = curwin->w_leftcol + W_WIDTH(curwin) - curwin_col_off() - 1;
582 validate_virtcol();
583
584 /*
585 * If the cursor is right or left of the screen, move it to last or first
586 * character.
587 */
588 if (curwin->w_virtcol > (colnr_T)(lastcol - p_siso))
589 {
590 retval = TRUE;
591 coladvance((colnr_T)(lastcol - p_siso));
592 }
593 else if (curwin->w_virtcol < curwin->w_leftcol + p_siso)
594 {
595 retval = TRUE;
596 (void)coladvance((colnr_T)(curwin->w_leftcol + p_siso));
597 }
598
599 /*
600 * If the start of the character under the cursor is not on the screen,
601 * advance the cursor one more char. If this fails (last char of the
602 * line) adjust the scrolling.
603 */
604 getvvcol(curwin, &curwin->w_cursor, &s, NULL, &e);
605 if (e > (colnr_T)lastcol)
606 {
607 retval = TRUE;
608 coladvance(s - 1);
609 }
610 else if (s < curwin->w_leftcol)
611 {
612 retval = TRUE;
613 if (coladvance(e + 1) == FAIL) /* there isn't another character */
614 {
615 curwin->w_leftcol = s; /* adjust w_leftcol instead */
616 changed_cline_bef_curs();
617 }
618 }
619
620 if (retval)
621 curwin->w_set_curswant = TRUE;
622 redraw_later(NOT_VALID);
623 return retval;
624}
625
626/**********************************************************************
627 * Various routines dealing with allocation and deallocation of memory.
628 */
629
630#if defined(MEM_PROFILE) || defined(PROTO)
631
632# define MEM_SIZES 8200
633static long_u mem_allocs[MEM_SIZES];
634static long_u mem_frees[MEM_SIZES];
635static long_u mem_allocated;
636static long_u mem_freed;
637static long_u mem_peak;
638static long_u num_alloc;
639static long_u num_freed;
640
641static void mem_pre_alloc_s __ARGS((size_t *sizep));
642static void mem_pre_alloc_l __ARGS((long_u *sizep));
643static void mem_post_alloc __ARGS((void **pp, size_t size));
644static void mem_pre_free __ARGS((void **pp));
645
646 static void
647mem_pre_alloc_s(sizep)
648 size_t *sizep;
649{
650 *sizep += sizeof(size_t);
651}
652
653 static void
654mem_pre_alloc_l(sizep)
655 long_u *sizep;
656{
657 *sizep += sizeof(size_t);
658}
659
660 static void
661mem_post_alloc(pp, size)
662 void **pp;
663 size_t size;
664{
665 if (*pp == NULL)
666 return;
667 size -= sizeof(size_t);
668 *(long_u *)*pp = size;
669 if (size <= MEM_SIZES-1)
670 mem_allocs[size-1]++;
671 else
672 mem_allocs[MEM_SIZES-1]++;
673 mem_allocated += size;
674 if (mem_allocated - mem_freed > mem_peak)
675 mem_peak = mem_allocated - mem_freed;
676 num_alloc++;
677 *pp = (void *)((char *)*pp + sizeof(size_t));
678}
679
680 static void
681mem_pre_free(pp)
682 void **pp;
683{
684 long_u size;
685
686 *pp = (void *)((char *)*pp - sizeof(size_t));
687 size = *(size_t *)*pp;
688 if (size <= MEM_SIZES-1)
689 mem_frees[size-1]++;
690 else
691 mem_frees[MEM_SIZES-1]++;
692 mem_freed += size;
693 num_freed++;
694}
695
696/*
697 * called on exit via atexit()
698 */
699 void
700vim_mem_profile_dump()
701{
702 int i, j;
703
704 printf("\r\n");
705 j = 0;
706 for (i = 0; i < MEM_SIZES - 1; i++)
707 {
708 if (mem_allocs[i] || mem_frees[i])
709 {
710 if (mem_frees[i] > mem_allocs[i])
711 printf("\r\n%s", _("ERROR: "));
712 printf("[%4d / %4lu-%-4lu] ", i + 1, mem_allocs[i], mem_frees[i]);
713 j++;
714 if (j > 3)
715 {
716 j = 0;
717 printf("\r\n");
718 }
719 }
720 }
721
722 i = MEM_SIZES - 1;
723 if (mem_allocs[i])
724 {
725 printf("\r\n");
726 if (mem_frees[i] > mem_allocs[i])
727 printf(_("ERROR: "));
728 printf("[>%d / %4lu-%-4lu]", i, mem_allocs[i], mem_frees[i]);
729 }
730
731 printf(_("\n[bytes] total alloc-freed %lu-%lu, in use %lu, peak use %lu\n"),
732 mem_allocated, mem_freed, mem_allocated - mem_freed, mem_peak);
733 printf(_("[calls] total re/malloc()'s %lu, total free()'s %lu\n\n"),
734 num_alloc, num_freed);
735}
736
737#endif /* MEM_PROFILE */
738
739/*
740 * Some memory is reserved for error messages and for being able to
741 * call mf_release_all(), which needs some memory for mf_trans_add().
742 */
743#if defined(MSDOS) && !defined(DJGPP)
744# define SMALL_MEM
745# define KEEP_ROOM 8192L
746#else
747# define KEEP_ROOM (2 * 8192L)
748#endif
749
750/*
Bram Moolenaar2a329742008-04-01 12:53:43 +0000751 * Note: if unsigned is 16 bits we can only allocate up to 64K with alloc().
Bram Moolenaar071d4272004-06-13 20:20:40 +0000752 * Use lalloc for larger blocks.
753 */
754 char_u *
755alloc(size)
756 unsigned size;
757{
758 return (lalloc((long_u)size, TRUE));
759}
760
761/*
762 * Allocate memory and set all bytes to zero.
763 */
764 char_u *
765alloc_clear(size)
766 unsigned size;
767{
768 char_u *p;
769
770 p = (lalloc((long_u)size, TRUE));
771 if (p != NULL)
772 (void)vim_memset(p, 0, (size_t)size);
773 return p;
774}
775
776/*
777 * alloc() with check for maximum line length
778 */
779 char_u *
780alloc_check(size)
781 unsigned size;
782{
783#if !defined(UNIX) && !defined(__EMX__)
784 if (sizeof(int) == 2 && size > 0x7fff)
785 {
786 /* Don't hide this message */
787 emsg_silent = 0;
788 EMSG(_("E340: Line is becoming too long"));
789 return NULL;
790 }
791#endif
792 return (lalloc((long_u)size, TRUE));
793}
794
795/*
796 * Allocate memory like lalloc() and set all bytes to zero.
797 */
798 char_u *
799lalloc_clear(size, message)
800 long_u size;
801 int message;
802{
803 char_u *p;
804
805 p = (lalloc(size, message));
806 if (p != NULL)
807 (void)vim_memset(p, 0, (size_t)size);
808 return p;
809}
810
811/*
812 * Low level memory allocation function.
813 * This is used often, KEEP IT FAST!
814 */
815 char_u *
816lalloc(size, message)
817 long_u size;
818 int message;
819{
820 char_u *p; /* pointer to new storage space */
821 static int releasing = FALSE; /* don't do mf_release_all() recursive */
822 int try_again;
823#if defined(HAVE_AVAIL_MEM) && !defined(SMALL_MEM)
824 static long_u allocated = 0; /* allocated since last avail check */
825#endif
826
827 /* Safety check for allocating zero bytes */
828 if (size == 0)
829 {
830 /* Don't hide this message */
831 emsg_silent = 0;
832 EMSGN(_("E341: Internal error: lalloc(%ld, )"), size);
833 return NULL;
834 }
835
836#ifdef MEM_PROFILE
837 mem_pre_alloc_l(&size);
838#endif
839
840#if defined(MSDOS) && !defined(DJGPP)
841 if (size >= 0xfff0) /* in MSDOS we can't deal with >64K blocks */
842 p = NULL;
843 else
844#endif
845
846 /*
847 * Loop when out of memory: Try to release some memfile blocks and
848 * if some blocks are released call malloc again.
849 */
850 for (;;)
851 {
852 /*
853 * Handle three kind of systems:
854 * 1. No check for available memory: Just return.
855 * 2. Slow check for available memory: call mch_avail_mem() after
856 * allocating KEEP_ROOM amount of memory.
857 * 3. Strict check for available memory: call mch_avail_mem()
858 */
859 if ((p = (char_u *)malloc((size_t)size)) != NULL)
860 {
861#ifndef HAVE_AVAIL_MEM
862 /* 1. No check for available memory: Just return. */
863 goto theend;
864#else
865# ifndef SMALL_MEM
866 /* 2. Slow check for available memory: call mch_avail_mem() after
867 * allocating (KEEP_ROOM / 2) amount of memory. */
868 allocated += size;
869 if (allocated < KEEP_ROOM / 2)
870 goto theend;
871 allocated = 0;
872# endif
873 /* 3. check for available memory: call mch_avail_mem() */
874 if (mch_avail_mem(TRUE) < KEEP_ROOM && !releasing)
875 {
876 vim_free((char *)p); /* System is low... no go! */
877 p = NULL;
878 }
879 else
880 goto theend;
881#endif
882 }
883 /*
884 * Remember that mf_release_all() is being called to avoid an endless
885 * loop, because mf_release_all() may call alloc() recursively.
886 */
887 if (releasing)
888 break;
889 releasing = TRUE;
Bram Moolenaar661b1822005-07-28 22:36:45 +0000890
891 clear_sb_text(); /* free any scrollback text */
892 try_again = mf_release_all(); /* release as many blocks as possible */
Bram Moolenaar39a58ca2005-06-27 22:42:44 +0000893#ifdef FEAT_EVAL
Bram Moolenaar661b1822005-07-28 22:36:45 +0000894 try_again |= garbage_collect(); /* cleanup recursive lists/dicts */
Bram Moolenaar39a58ca2005-06-27 22:42:44 +0000895#endif
Bram Moolenaar661b1822005-07-28 22:36:45 +0000896
Bram Moolenaar071d4272004-06-13 20:20:40 +0000897 releasing = FALSE;
898 if (!try_again)
899 break;
900 }
901
902 if (message && p == NULL)
903 do_outofmem_msg(size);
904
905theend:
906#ifdef MEM_PROFILE
907 mem_post_alloc((void **)&p, (size_t)size);
908#endif
909 return p;
910}
911
912#if defined(MEM_PROFILE) || defined(PROTO)
913/*
914 * realloc() with memory profiling.
915 */
916 void *
917mem_realloc(ptr, size)
918 void *ptr;
919 size_t size;
920{
921 void *p;
922
923 mem_pre_free(&ptr);
924 mem_pre_alloc_s(&size);
925
926 p = realloc(ptr, size);
927
928 mem_post_alloc(&p, size);
929
930 return p;
931}
932#endif
933
934/*
935* Avoid repeating the error message many times (they take 1 second each).
936* Did_outofmem_msg is reset when a character is read.
937*/
938 void
939do_outofmem_msg(size)
940 long_u size;
941{
942 if (!did_outofmem_msg)
943 {
944 /* Don't hide this message */
945 emsg_silent = 0;
946 EMSGN(_("E342: Out of memory! (allocating %lu bytes)"), size);
947 did_outofmem_msg = TRUE;
948 }
949}
950
Bram Moolenaar1ec484f2005-06-24 23:07:47 +0000951#if defined(EXITFREE) || defined(PROTO)
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000952
953# if defined(FEAT_SEARCHPATH)
954static void free_findfile __ARGS((void));
955# endif
956
Bram Moolenaar1ec484f2005-06-24 23:07:47 +0000957/*
958 * Free everything that we allocated.
959 * Can be used to detect memory leaks, e.g., with ccmalloc.
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000960 * NOTE: This is tricky! Things are freed that functions depend on. Don't be
961 * surprised if Vim crashes...
962 * Some things can't be freed, esp. things local to a library function.
Bram Moolenaar1ec484f2005-06-24 23:07:47 +0000963 */
964 void
965free_all_mem()
966{
967 buf_T *buf, *nextbuf;
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000968 static int entered = FALSE;
969
970 /* When we cause a crash here it is caught and Vim tries to exit cleanly.
971 * Don't try freeing everything again. */
972 if (entered)
973 return;
974 entered = TRUE;
Bram Moolenaar1ec484f2005-06-24 23:07:47 +0000975
Bram Moolenaare0ca7b22007-11-24 20:28:24 +0000976# ifdef FEAT_AUTOCMD
Bram Moolenaar78ab3312007-09-29 12:16:41 +0000977 block_autocmds(); /* don't want to trigger autocommands here */
Bram Moolenaare0ca7b22007-11-24 20:28:24 +0000978# endif
Bram Moolenaar1ec484f2005-06-24 23:07:47 +0000979
Bram Moolenaare0ca7b22007-11-24 20:28:24 +0000980# ifdef FEAT_WINDOWS
Bram Moolenaar997fb4b2006-02-17 21:53:23 +0000981 /* close all tabs and windows */
Bram Moolenaar7d47b6e2006-03-15 22:59:18 +0000982 if (first_tabpage->tp_next != NULL)
983 do_cmdline_cmd((char_u *)"tabonly!");
984 if (firstwin != lastwin)
985 do_cmdline_cmd((char_u *)"only!");
Bram Moolenaare0ca7b22007-11-24 20:28:24 +0000986# endif
Bram Moolenaar997fb4b2006-02-17 21:53:23 +0000987
Bram Moolenaarc4956c82006-03-12 21:58:43 +0000988# if defined(FEAT_SPELL)
Bram Moolenaar1ec484f2005-06-24 23:07:47 +0000989 /* Free all spell info. */
990 spell_free_all();
991# endif
992
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000993# if defined(FEAT_USR_CMDS)
Bram Moolenaar1ec484f2005-06-24 23:07:47 +0000994 /* Clear user commands (before deleting buffers). */
995 ex_comclear(NULL);
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000996# endif
Bram Moolenaar1ec484f2005-06-24 23:07:47 +0000997
998# ifdef FEAT_MENU
999 /* Clear menus. */
1000 do_cmdline_cmd((char_u *)"aunmenu *");
1001# endif
1002
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001003 /* Clear mappings, abbreviations, breakpoints. */
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001004 do_cmdline_cmd((char_u *)"mapclear");
1005 do_cmdline_cmd((char_u *)"mapclear!");
1006 do_cmdline_cmd((char_u *)"abclear");
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001007# if defined(FEAT_EVAL)
1008 do_cmdline_cmd((char_u *)"breakdel *");
1009# endif
Bram Moolenaar1e498f52005-06-26 22:29:44 +00001010# if defined(FEAT_PROFILE)
1011 do_cmdline_cmd((char_u *)"profdel *");
1012# endif
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001013
1014# ifdef FEAT_TITLE
1015 free_titles();
1016# endif
1017# if defined(FEAT_SEARCHPATH)
1018 free_findfile();
1019# endif
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001020
1021 /* Obviously named calls. */
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001022# if defined(FEAT_AUTOCMD)
1023 free_all_autocmds();
1024# endif
1025 clear_termcodes();
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001026 free_all_options();
1027 free_all_marks();
1028 alist_clear(&global_alist);
1029 free_homedir();
1030 free_search_patterns();
1031 free_old_sub();
1032 free_last_insert();
1033 free_prev_shellcmd();
1034 free_regexp_stuff();
1035 free_tag_stuff();
1036 free_cd_dir();
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001037# ifdef FEAT_EVAL
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001038 set_expr_line(NULL);
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001039# endif
1040# ifdef FEAT_DIFF
Bram Moolenaar997fb4b2006-02-17 21:53:23 +00001041 diff_clear(curtab);
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001042# endif
Bram Moolenaara40ceaf2006-01-13 22:35:40 +00001043 clear_sb_text(); /* free any scrollback text */
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001044
1045 /* Free some global vars. */
1046 vim_free(username);
Bram Moolenaaraf92ee82007-10-07 13:45:08 +00001047# ifdef FEAT_CLIPBOARD
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001048 vim_free(clip_exclude_prog);
Bram Moolenaaraf92ee82007-10-07 13:45:08 +00001049# endif
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001050 vim_free(last_cmdline);
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001051# ifdef FEAT_CMDHIST
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001052 vim_free(new_last_cmdline);
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001053# endif
Bram Moolenaar238a5642006-02-21 22:12:05 +00001054 set_keep_msg(NULL, 0);
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001055 vim_free(ff_expand_buffer);
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001056
1057 /* Clear cmdline history. */
1058 p_hi = 0;
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001059# ifdef FEAT_CMDHIST
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001060 init_history();
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001061# endif
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001062
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001063#ifdef FEAT_QUICKFIX
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001064 {
1065 win_T *win;
1066
1067 qf_free_all(NULL);
1068 /* Free all location lists */
1069 FOR_ALL_WINDOWS(win)
1070 qf_free_all(win);
1071 }
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001072#endif
1073
1074 /* Close all script inputs. */
1075 close_all_scripts();
1076
1077#if defined(FEAT_WINDOWS)
1078 /* Destroy all windows. Must come before freeing buffers. */
1079 win_free_all();
1080#endif
1081
Bram Moolenaar2a329742008-04-01 12:53:43 +00001082 /* Free all buffers. Reset 'autochdir' to avoid accessing things that
1083 * were freed already. */
1084#ifdef FEAT_AUTOCHDIR
1085 p_acd = FALSE;
1086#endif
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001087 for (buf = firstbuf; buf != NULL; )
1088 {
1089 nextbuf = buf->b_next;
1090 close_buffer(NULL, buf, DOBUF_WIPE);
1091 if (buf_valid(buf))
1092 buf = nextbuf; /* didn't work, try next one */
1093 else
1094 buf = firstbuf;
1095 }
1096
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001097#ifdef FEAT_ARABIC
1098 free_cmdline_buf();
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001099#endif
1100
1101 /* Clear registers. */
1102 clear_registers();
1103 ResetRedobuff();
1104 ResetRedobuff();
1105
Bram Moolenaar85c79d32007-02-20 01:59:20 +00001106#if defined(FEAT_CLIENTSERVER) && defined(FEAT_X11)
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001107 vim_free(serverDelayedStartName);
1108#endif
1109
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001110 /* highlight info */
1111 free_highlight();
1112
Bram Moolenaar1e498f52005-06-26 22:29:44 +00001113 reset_last_sourcing();
1114
Bram Moolenaar7d47b6e2006-03-15 22:59:18 +00001115#ifdef FEAT_WINDOWS
Bram Moolenaar06a89a52006-04-29 22:01:03 +00001116 free_tabpage(first_tabpage);
1117 first_tabpage = NULL;
Bram Moolenaar7d47b6e2006-03-15 22:59:18 +00001118#endif
1119
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001120# ifdef UNIX
1121 /* Machine-specific free. */
1122 mch_free_mem();
1123# endif
1124
1125 /* message history */
1126 for (;;)
1127 if (delete_first_msg() == FAIL)
1128 break;
1129
1130# ifdef FEAT_EVAL
1131 eval_clear();
1132# endif
1133
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001134 free_termoptions();
1135
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001136 /* screenlines (can't display anything now!) */
1137 free_screenlines();
1138
1139#if defined(USE_XSMP)
1140 xsmp_close();
1141#endif
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001142#ifdef FEAT_GUI_GTK
1143 gui_mch_free_all();
1144#endif
1145 clear_hl_tables();
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001146
1147 vim_free(IObuff);
1148 vim_free(NameBuff);
1149}
1150#endif
1151
Bram Moolenaar071d4272004-06-13 20:20:40 +00001152/*
1153 * copy a string into newly allocated memory
1154 */
1155 char_u *
1156vim_strsave(string)
1157 char_u *string;
1158{
1159 char_u *p;
1160 unsigned len;
1161
1162 len = (unsigned)STRLEN(string) + 1;
1163 p = alloc(len);
1164 if (p != NULL)
1165 mch_memmove(p, string, (size_t)len);
1166 return p;
1167}
1168
1169 char_u *
1170vim_strnsave(string, len)
1171 char_u *string;
1172 int len;
1173{
1174 char_u *p;
1175
1176 p = alloc((unsigned)(len + 1));
1177 if (p != NULL)
1178 {
1179 STRNCPY(p, string, len);
1180 p[len] = NUL;
1181 }
1182 return p;
1183}
1184
Bram Moolenaar071d4272004-06-13 20:20:40 +00001185/*
1186 * Same as vim_strsave(), but any characters found in esc_chars are preceded
1187 * by a backslash.
1188 */
1189 char_u *
1190vim_strsave_escaped(string, esc_chars)
1191 char_u *string;
1192 char_u *esc_chars;
1193{
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001194 return vim_strsave_escaped_ext(string, esc_chars, '\\', FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001195}
1196
1197/*
1198 * Same as vim_strsave_escaped(), but when "bsl" is TRUE also escape
1199 * characters where rem_backslash() would remove the backslash.
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001200 * Escape the characters with "cc".
Bram Moolenaar071d4272004-06-13 20:20:40 +00001201 */
1202 char_u *
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001203vim_strsave_escaped_ext(string, esc_chars, cc, bsl)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001204 char_u *string;
1205 char_u *esc_chars;
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001206 int cc;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001207 int bsl;
1208{
1209 char_u *p;
1210 char_u *p2;
1211 char_u *escaped_string;
1212 unsigned length;
1213#ifdef FEAT_MBYTE
1214 int l;
1215#endif
1216
1217 /*
1218 * First count the number of backslashes required.
1219 * Then allocate the memory and insert them.
1220 */
1221 length = 1; /* count the trailing NUL */
1222 for (p = string; *p; p++)
1223 {
1224#ifdef FEAT_MBYTE
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001225 if (has_mbyte && (l = (*mb_ptr2len)(p)) > 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001226 {
1227 length += l; /* count a multibyte char */
1228 p += l - 1;
1229 continue;
1230 }
1231#endif
1232 if (vim_strchr(esc_chars, *p) != NULL || (bsl && rem_backslash(p)))
1233 ++length; /* count a backslash */
1234 ++length; /* count an ordinary char */
1235 }
1236 escaped_string = alloc(length);
1237 if (escaped_string != NULL)
1238 {
1239 p2 = escaped_string;
1240 for (p = string; *p; p++)
1241 {
1242#ifdef FEAT_MBYTE
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001243 if (has_mbyte && (l = (*mb_ptr2len)(p)) > 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001244 {
1245 mch_memmove(p2, p, (size_t)l);
1246 p2 += l;
1247 p += l - 1; /* skip multibyte char */
1248 continue;
1249 }
1250#endif
1251 if (vim_strchr(esc_chars, *p) != NULL || (bsl && rem_backslash(p)))
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001252 *p2++ = cc;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001253 *p2++ = *p;
1254 }
1255 *p2 = NUL;
1256 }
1257 return escaped_string;
1258}
1259
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001260#if defined(FEAT_EVAL) || defined(PROTO)
1261/*
1262 * Escape "string" for use as a shell argument with system().
1263 * This uses single quotes, except when we know we need to use double qoutes
1264 * (MS-DOS and MS-Windows without 'shellslash' set).
Bram Moolenaar05bb9532008-07-04 09:44:11 +00001265 * Also replace "%", "#" and things like "<cfile>" when "do_special" is TRUE.
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001266 * Returns the result in allocated memory, NULL if we have run out.
1267 */
1268 char_u *
Bram Moolenaar05bb9532008-07-04 09:44:11 +00001269vim_strsave_shellescape(string, do_special)
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001270 char_u *string;
Bram Moolenaar05bb9532008-07-04 09:44:11 +00001271 int do_special;
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001272{
1273 unsigned length;
1274 char_u *p;
1275 char_u *d;
1276 char_u *escaped_string;
Bram Moolenaar05bb9532008-07-04 09:44:11 +00001277 int l;
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001278
1279 /* First count the number of extra bytes required. */
Bram Moolenaar77f66d62007-02-20 02:16:18 +00001280 length = (unsigned)STRLEN(string) + 3; /* two quotes and a trailing NUL */
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001281 for (p = string; *p != NUL; mb_ptr_adv(p))
1282 {
1283# if defined(WIN32) || defined(WIN16) || defined(DOS)
1284 if (!p_ssl)
1285 {
1286 if (*p == '"')
1287 ++length; /* " -> "" */
1288 }
1289 else
1290# endif
1291 if (*p == '\'')
1292 length += 3; /* ' => '\'' */
Bram Moolenaar05bb9532008-07-04 09:44:11 +00001293 if (do_special && find_cmdline_var(p, &l) >= 0)
1294 {
1295 ++length; /* insert backslash */
1296 p += l - 1;
1297 }
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001298 }
1299
1300 /* Allocate memory for the result and fill it. */
1301 escaped_string = alloc(length);
1302 if (escaped_string != NULL)
1303 {
1304 d = escaped_string;
1305
1306 /* add opening quote */
1307# if defined(WIN32) || defined(WIN16) || defined(DOS)
1308 if (!p_ssl)
1309 *d++ = '"';
1310 else
1311# endif
1312 *d++ = '\'';
1313
1314 for (p = string; *p != NUL; )
1315 {
1316# if defined(WIN32) || defined(WIN16) || defined(DOS)
1317 if (!p_ssl)
1318 {
1319 if (*p == '"')
1320 {
1321 *d++ = '"';
1322 *d++ = '"';
1323 ++p;
1324 continue;
1325 }
1326 }
1327 else
1328# endif
1329 if (*p == '\'')
1330 {
Bram Moolenaar05bb9532008-07-04 09:44:11 +00001331 *d++ = '\'';
1332 *d++ = '\\';
1333 *d++ = '\'';
1334 *d++ = '\'';
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001335 ++p;
1336 continue;
1337 }
Bram Moolenaar05bb9532008-07-04 09:44:11 +00001338 if (do_special && find_cmdline_var(p, &l) >= 0)
1339 {
1340 *d++ = '\\'; /* insert backslash */
1341 while (--l >= 0) /* copy the var */
1342 *d++ = *p++;
1343 }
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001344
1345 MB_COPY_CHAR(p, d);
1346 }
1347
1348 /* add terminating quote and finish with a NUL */
1349# if defined(WIN32) || defined(WIN16) || defined(DOS)
1350 if (!p_ssl)
1351 *d++ = '"';
1352 else
1353# endif
1354 *d++ = '\'';
1355 *d = NUL;
1356 }
1357
1358 return escaped_string;
1359}
1360#endif
1361
Bram Moolenaar071d4272004-06-13 20:20:40 +00001362/*
1363 * Like vim_strsave(), but make all characters uppercase.
1364 * This uses ASCII lower-to-upper case translation, language independent.
1365 */
1366 char_u *
1367vim_strsave_up(string)
1368 char_u *string;
1369{
1370 char_u *p1;
1371
1372 p1 = vim_strsave(string);
1373 vim_strup(p1);
1374 return p1;
1375}
1376
1377/*
1378 * Like vim_strnsave(), but make all characters uppercase.
1379 * This uses ASCII lower-to-upper case translation, language independent.
1380 */
1381 char_u *
1382vim_strnsave_up(string, len)
1383 char_u *string;
1384 int len;
1385{
1386 char_u *p1;
1387
1388 p1 = vim_strnsave(string, len);
1389 vim_strup(p1);
1390 return p1;
1391}
1392
1393/*
1394 * ASCII lower-to-upper case translation, language independent.
1395 */
1396 void
1397vim_strup(p)
1398 char_u *p;
1399{
1400 char_u *p2;
1401 int c;
1402
1403 if (p != NULL)
1404 {
1405 p2 = p;
1406 while ((c = *p2) != NUL)
1407#ifdef EBCDIC
1408 *p2++ = isalpha(c) ? toupper(c) : c;
1409#else
1410 *p2++ = (c < 'a' || c > 'z') ? c : (c - 0x20);
1411#endif
1412 }
1413}
1414
Bram Moolenaarc4956c82006-03-12 21:58:43 +00001415#if defined(FEAT_EVAL) || defined(FEAT_SPELL) || defined(PROTO)
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001416/*
1417 * Make string "s" all upper-case and return it in allocated memory.
1418 * Handles multi-byte characters as well as possible.
1419 * Returns NULL when out of memory.
1420 */
1421 char_u *
1422strup_save(orig)
1423 char_u *orig;
1424{
1425 char_u *p;
1426 char_u *res;
1427
1428 res = p = vim_strsave(orig);
1429
1430 if (res != NULL)
1431 while (*p != NUL)
1432 {
1433# ifdef FEAT_MBYTE
1434 int l;
1435
1436 if (enc_utf8)
1437 {
1438 int c, uc;
1439 int nl;
1440 char_u *s;
1441
1442 c = utf_ptr2char(p);
1443 uc = utf_toupper(c);
1444
1445 /* Reallocate string when byte count changes. This is rare,
1446 * thus it's OK to do another malloc()/free(). */
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001447 l = utf_ptr2len(p);
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001448 nl = utf_char2len(uc);
1449 if (nl != l)
1450 {
1451 s = alloc((unsigned)STRLEN(res) + 1 + nl - l);
1452 if (s == NULL)
1453 break;
1454 mch_memmove(s, res, p - res);
1455 STRCPY(s + (p - res) + nl, p + l);
1456 p = s + (p - res);
1457 vim_free(res);
1458 res = s;
1459 }
1460
1461 utf_char2bytes(uc, p);
1462 p += nl;
1463 }
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001464 else if (has_mbyte && (l = (*mb_ptr2len)(p)) > 1)
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001465 p += l; /* skip multi-byte character */
1466 else
1467# endif
1468 {
1469 *p = TOUPPER_LOC(*p); /* note that toupper() can be a macro */
1470 p++;
1471 }
1472 }
1473
1474 return res;
1475}
1476#endif
1477
Bram Moolenaar071d4272004-06-13 20:20:40 +00001478/*
1479 * copy a space a number of times
1480 */
1481 void
1482copy_spaces(ptr, count)
1483 char_u *ptr;
1484 size_t count;
1485{
1486 size_t i = count;
1487 char_u *p = ptr;
1488
1489 while (i--)
1490 *p++ = ' ';
1491}
1492
1493#if defined(FEAT_VISUALEXTRA) || defined(PROTO)
1494/*
1495 * Copy a character a number of times.
1496 * Does not work for multi-byte charactes!
1497 */
1498 void
1499copy_chars(ptr, count, c)
1500 char_u *ptr;
1501 size_t count;
1502 int c;
1503{
1504 size_t i = count;
1505 char_u *p = ptr;
1506
1507 while (i--)
1508 *p++ = c;
1509}
1510#endif
1511
1512/*
1513 * delete spaces at the end of a string
1514 */
1515 void
1516del_trailing_spaces(ptr)
1517 char_u *ptr;
1518{
1519 char_u *q;
1520
1521 q = ptr + STRLEN(ptr);
1522 while (--q > ptr && vim_iswhite(q[0]) && q[-1] != '\\' && q[-1] != Ctrl_V)
1523 *q = NUL;
1524}
1525
1526/*
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001527 * Like strncpy(), but always terminate the result with one NUL.
Bram Moolenaard042c562005-06-30 22:04:15 +00001528 * "to" must be "len + 1" long!
Bram Moolenaar071d4272004-06-13 20:20:40 +00001529 */
1530 void
1531vim_strncpy(to, from, len)
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001532 char_u *to;
1533 char_u *from;
Bram Moolenaarbbebc852005-07-18 21:47:53 +00001534 size_t len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001535{
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001536 STRNCPY(to, from, len);
1537 to[len] = NUL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001538}
1539
1540/*
1541 * Isolate one part of a string option where parts are separated with
1542 * "sep_chars".
Bram Moolenaar83bab712005-08-01 21:58:57 +00001543 * The part is copied into "buf[maxlen]".
Bram Moolenaar071d4272004-06-13 20:20:40 +00001544 * "*option" is advanced to the next part.
1545 * The length is returned.
1546 */
1547 int
1548copy_option_part(option, buf, maxlen, sep_chars)
1549 char_u **option;
1550 char_u *buf;
1551 int maxlen;
1552 char *sep_chars;
1553{
1554 int len = 0;
1555 char_u *p = *option;
1556
1557 /* skip '.' at start of option part, for 'suffixes' */
1558 if (*p == '.')
1559 buf[len++] = *p++;
1560 while (*p != NUL && vim_strchr((char_u *)sep_chars, *p) == NULL)
1561 {
1562 /*
1563 * Skip backslash before a separator character and space.
1564 */
1565 if (p[0] == '\\' && vim_strchr((char_u *)sep_chars, p[1]) != NULL)
1566 ++p;
1567 if (len < maxlen - 1)
1568 buf[len++] = *p;
1569 ++p;
1570 }
1571 buf[len] = NUL;
1572
1573 if (*p != NUL && *p != ',') /* skip non-standard separator */
1574 ++p;
1575 p = skip_to_option_part(p); /* p points to next file name */
1576
1577 *option = p;
1578 return len;
1579}
1580
1581/*
Bram Moolenaar4770d092006-01-12 23:22:24 +00001582 * Replacement for free() that ignores NULL pointers.
1583 * Also skip free() when exiting for sure, this helps when we caught a deadly
1584 * signal that was caused by a crash in free().
Bram Moolenaar071d4272004-06-13 20:20:40 +00001585 */
1586 void
1587vim_free(x)
1588 void *x;
1589{
Bram Moolenaar4770d092006-01-12 23:22:24 +00001590 if (x != NULL && !really_exiting)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001591 {
1592#ifdef MEM_PROFILE
1593 mem_pre_free(&x);
1594#endif
1595 free(x);
1596 }
1597}
1598
1599#ifndef HAVE_MEMSET
1600 void *
1601vim_memset(ptr, c, size)
1602 void *ptr;
1603 int c;
1604 size_t size;
1605{
1606 char *p = ptr;
1607
1608 while (size-- > 0)
1609 *p++ = c;
1610 return ptr;
1611}
1612#endif
1613
1614#ifdef VIM_MEMCMP
1615/*
1616 * Return zero when "b1" and "b2" are the same for "len" bytes.
1617 * Return non-zero otherwise.
1618 */
1619 int
1620vim_memcmp(b1, b2, len)
1621 void *b1;
1622 void *b2;
1623 size_t len;
1624{
1625 char_u *p1 = (char_u *)b1, *p2 = (char_u *)b2;
1626
1627 for ( ; len > 0; --len)
1628 {
1629 if (*p1 != *p2)
1630 return 1;
1631 ++p1;
1632 ++p2;
1633 }
1634 return 0;
1635}
1636#endif
1637
1638#ifdef VIM_MEMMOVE
1639/*
1640 * Version of memmove() that handles overlapping source and destination.
1641 * For systems that don't have a function that is guaranteed to do that (SYSV).
1642 */
1643 void
1644mch_memmove(dst_arg, src_arg, len)
1645 void *src_arg, *dst_arg;
1646 size_t len;
1647{
1648 /*
1649 * A void doesn't have a size, we use char pointers.
1650 */
1651 char *dst = dst_arg, *src = src_arg;
1652
1653 /* overlap, copy backwards */
1654 if (dst > src && dst < src + len)
1655 {
1656 src += len;
1657 dst += len;
1658 while (len-- > 0)
1659 *--dst = *--src;
1660 }
1661 else /* copy forwards */
1662 while (len-- > 0)
1663 *dst++ = *src++;
1664}
1665#endif
1666
1667#if (!defined(HAVE_STRCASECMP) && !defined(HAVE_STRICMP)) || defined(PROTO)
1668/*
1669 * Compare two strings, ignoring case, using current locale.
1670 * Doesn't work for multi-byte characters.
1671 * return 0 for match, < 0 for smaller, > 0 for bigger
1672 */
1673 int
1674vim_stricmp(s1, s2)
1675 char *s1;
1676 char *s2;
1677{
1678 int i;
1679
1680 for (;;)
1681 {
1682 i = (int)TOLOWER_LOC(*s1) - (int)TOLOWER_LOC(*s2);
1683 if (i != 0)
1684 return i; /* this character different */
1685 if (*s1 == NUL)
1686 break; /* strings match until NUL */
1687 ++s1;
1688 ++s2;
1689 }
1690 return 0; /* strings match */
1691}
1692#endif
1693
1694#if (!defined(HAVE_STRNCASECMP) && !defined(HAVE_STRNICMP)) || defined(PROTO)
1695/*
1696 * Compare two strings, for length "len", ignoring case, using current locale.
1697 * Doesn't work for multi-byte characters.
1698 * return 0 for match, < 0 for smaller, > 0 for bigger
1699 */
1700 int
1701vim_strnicmp(s1, s2, len)
1702 char *s1;
1703 char *s2;
1704 size_t len;
1705{
1706 int i;
1707
1708 while (len > 0)
1709 {
1710 i = (int)TOLOWER_LOC(*s1) - (int)TOLOWER_LOC(*s2);
1711 if (i != 0)
1712 return i; /* this character different */
1713 if (*s1 == NUL)
1714 break; /* strings match until NUL */
1715 ++s1;
1716 ++s2;
1717 --len;
1718 }
1719 return 0; /* strings match */
1720}
1721#endif
1722
1723#if 0 /* currently not used */
1724/*
1725 * Check if string "s2" appears somewhere in "s1" while ignoring case.
1726 * Return NULL if not, a pointer to the first occurrence if it does.
1727 */
1728 char_u *
1729vim_stristr(s1, s2)
1730 char_u *s1;
1731 char_u *s2;
1732{
1733 char_u *p;
1734 int len = STRLEN(s2);
1735 char_u *end = s1 + STRLEN(s1) - len;
1736
1737 for (p = s1; p <= end; ++p)
1738 if (STRNICMP(p, s2, len) == 0)
1739 return p;
1740 return NULL;
1741}
1742#endif
1743
1744/*
1745 * Version of strchr() and strrchr() that handle unsigned char strings
Bram Moolenaar05159a02005-02-26 23:04:13 +00001746 * with characters from 128 to 255 correctly. It also doesn't return a
1747 * pointer to the NUL at the end of the string.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001748 */
1749 char_u *
1750vim_strchr(string, c)
1751 char_u *string;
1752 int c;
1753{
1754 char_u *p;
1755 int b;
1756
1757 p = string;
1758#ifdef FEAT_MBYTE
1759 if (enc_utf8 && c >= 0x80)
1760 {
1761 while (*p != NUL)
1762 {
1763 if (utf_ptr2char(p) == c)
1764 return p;
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001765 p += (*mb_ptr2len)(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001766 }
1767 return NULL;
1768 }
1769 if (enc_dbcs != 0 && c > 255)
1770 {
1771 int n2 = c & 0xff;
1772
1773 c = ((unsigned)c >> 8) & 0xff;
1774 while ((b = *p) != NUL)
1775 {
1776 if (b == c && p[1] == n2)
1777 return p;
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001778 p += (*mb_ptr2len)(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001779 }
1780 return NULL;
1781 }
1782 if (has_mbyte)
1783 {
1784 while ((b = *p) != NUL)
1785 {
1786 if (b == c)
1787 return p;
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001788 p += (*mb_ptr2len)(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001789 }
1790 return NULL;
1791 }
1792#endif
1793 while ((b = *p) != NUL)
1794 {
1795 if (b == c)
1796 return p;
1797 ++p;
1798 }
1799 return NULL;
1800}
1801
1802/*
Bram Moolenaar05159a02005-02-26 23:04:13 +00001803 * Version of strchr() that only works for bytes and handles unsigned char
1804 * strings with characters above 128 correctly. It also doesn't return a
1805 * pointer to the NUL at the end of the string.
1806 */
1807 char_u *
1808vim_strbyte(string, c)
1809 char_u *string;
1810 int c;
1811{
1812 char_u *p = string;
1813
1814 while (*p != NUL)
1815 {
1816 if (*p == c)
1817 return p;
1818 ++p;
1819 }
1820 return NULL;
1821}
1822
1823/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00001824 * Search for last occurrence of "c" in "string".
Bram Moolenaarac6e65f2005-08-29 22:25:38 +00001825 * Return NULL if not found.
Bram Moolenaar05159a02005-02-26 23:04:13 +00001826 * Does not handle multi-byte char for "c"!
Bram Moolenaar071d4272004-06-13 20:20:40 +00001827 */
1828 char_u *
1829vim_strrchr(string, c)
1830 char_u *string;
1831 int c;
1832{
1833 char_u *retval = NULL;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001834 char_u *p = string;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001835
Bram Moolenaar05159a02005-02-26 23:04:13 +00001836 while (*p)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001837 {
Bram Moolenaar05159a02005-02-26 23:04:13 +00001838 if (*p == c)
1839 retval = p;
1840 mb_ptr_adv(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001841 }
1842 return retval;
1843}
1844
1845/*
1846 * Vim's version of strpbrk(), in case it's missing.
1847 * Don't generate a prototype for this, causes problems when it's not used.
1848 */
1849#ifndef PROTO
1850# ifndef HAVE_STRPBRK
1851# ifdef vim_strpbrk
1852# undef vim_strpbrk
1853# endif
1854 char_u *
1855vim_strpbrk(s, charset)
1856 char_u *s;
1857 char_u *charset;
1858{
1859 while (*s)
1860 {
1861 if (vim_strchr(charset, *s) != NULL)
1862 return s;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001863 mb_ptr_adv(s);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001864 }
1865 return NULL;
1866}
1867# endif
1868#endif
1869
1870/*
1871 * Vim has its own isspace() function, because on some machines isspace()
1872 * can't handle characters above 128.
1873 */
1874 int
1875vim_isspace(x)
1876 int x;
1877{
1878 return ((x >= 9 && x <= 13) || x == ' ');
1879}
1880
1881/************************************************************************
Bram Moolenaar383f9bc2005-01-19 22:18:32 +00001882 * Functions for handling growing arrays.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001883 */
1884
1885/*
1886 * Clear an allocated growing array.
1887 */
1888 void
1889ga_clear(gap)
1890 garray_T *gap;
1891{
1892 vim_free(gap->ga_data);
1893 ga_init(gap);
1894}
1895
1896/*
1897 * Clear a growing array that contains a list of strings.
1898 */
1899 void
1900ga_clear_strings(gap)
1901 garray_T *gap;
1902{
1903 int i;
1904
1905 for (i = 0; i < gap->ga_len; ++i)
1906 vim_free(((char_u **)(gap->ga_data))[i]);
1907 ga_clear(gap);
1908}
1909
1910/*
1911 * Initialize a growing array. Don't forget to set ga_itemsize and
1912 * ga_growsize! Or use ga_init2().
1913 */
1914 void
1915ga_init(gap)
1916 garray_T *gap;
1917{
1918 gap->ga_data = NULL;
Bram Moolenaar86b68352004-12-27 21:59:20 +00001919 gap->ga_maxlen = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001920 gap->ga_len = 0;
1921}
1922
1923 void
1924ga_init2(gap, itemsize, growsize)
1925 garray_T *gap;
1926 int itemsize;
1927 int growsize;
1928{
1929 ga_init(gap);
1930 gap->ga_itemsize = itemsize;
1931 gap->ga_growsize = growsize;
1932}
1933
1934/*
1935 * Make room in growing array "gap" for at least "n" items.
1936 * Return FAIL for failure, OK otherwise.
1937 */
1938 int
1939ga_grow(gap, n)
1940 garray_T *gap;
1941 int n;
1942{
1943 size_t len;
1944 char_u *pp;
1945
Bram Moolenaar86b68352004-12-27 21:59:20 +00001946 if (gap->ga_maxlen - gap->ga_len < n)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001947 {
1948 if (n < gap->ga_growsize)
1949 n = gap->ga_growsize;
1950 len = gap->ga_itemsize * (gap->ga_len + n);
1951 pp = alloc_clear((unsigned)len);
1952 if (pp == NULL)
1953 return FAIL;
Bram Moolenaar86b68352004-12-27 21:59:20 +00001954 gap->ga_maxlen = gap->ga_len + n;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001955 if (gap->ga_data != NULL)
1956 {
1957 mch_memmove(pp, gap->ga_data,
1958 (size_t)(gap->ga_itemsize * gap->ga_len));
1959 vim_free(gap->ga_data);
1960 }
1961 gap->ga_data = pp;
1962 }
1963 return OK;
1964}
1965
1966/*
1967 * Concatenate a string to a growarray which contains characters.
1968 * Note: Does NOT copy the NUL at the end!
1969 */
1970 void
1971ga_concat(gap, s)
1972 garray_T *gap;
1973 char_u *s;
1974{
1975 int len = (int)STRLEN(s);
1976
1977 if (ga_grow(gap, len) == OK)
1978 {
1979 mch_memmove((char *)gap->ga_data + gap->ga_len, s, (size_t)len);
1980 gap->ga_len += len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001981 }
1982}
1983
1984/*
1985 * Append one byte to a growarray which contains bytes.
1986 */
1987 void
1988ga_append(gap, c)
1989 garray_T *gap;
1990 int c;
1991{
1992 if (ga_grow(gap, 1) == OK)
1993 {
1994 *((char *)gap->ga_data + gap->ga_len) = c;
1995 ++gap->ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001996 }
1997}
1998
1999/************************************************************************
2000 * functions that use lookup tables for various things, generally to do with
2001 * special key codes.
2002 */
2003
2004/*
2005 * Some useful tables.
2006 */
2007
2008static struct modmasktable
2009{
2010 short mod_mask; /* Bit-mask for particular key modifier */
2011 short mod_flag; /* Bit(s) for particular key modifier */
2012 char_u name; /* Single letter name of modifier */
2013} mod_mask_table[] =
2014{
2015 {MOD_MASK_ALT, MOD_MASK_ALT, (char_u)'M'},
Bram Moolenaar19a09a12005-03-04 23:39:37 +00002016 {MOD_MASK_META, MOD_MASK_META, (char_u)'T'},
Bram Moolenaar071d4272004-06-13 20:20:40 +00002017 {MOD_MASK_CTRL, MOD_MASK_CTRL, (char_u)'C'},
2018 {MOD_MASK_SHIFT, MOD_MASK_SHIFT, (char_u)'S'},
2019 {MOD_MASK_MULTI_CLICK, MOD_MASK_2CLICK, (char_u)'2'},
2020 {MOD_MASK_MULTI_CLICK, MOD_MASK_3CLICK, (char_u)'3'},
2021 {MOD_MASK_MULTI_CLICK, MOD_MASK_4CLICK, (char_u)'4'},
2022#ifdef MACOS
2023 {MOD_MASK_CMD, MOD_MASK_CMD, (char_u)'D'},
2024#endif
2025 /* 'A' must be the last one */
2026 {MOD_MASK_ALT, MOD_MASK_ALT, (char_u)'A'},
2027 {0, 0, NUL}
2028};
2029
2030/*
2031 * Shifted key terminal codes and their unshifted equivalent.
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00002032 * Don't add mouse codes here, they are handled separately!
Bram Moolenaar071d4272004-06-13 20:20:40 +00002033 */
2034#define MOD_KEYS_ENTRY_SIZE 5
2035
2036static char_u modifier_keys_table[] =
2037{
2038/* mod mask with modifier without modifier */
2039 MOD_MASK_SHIFT, '&', '9', '@', '1', /* begin */
2040 MOD_MASK_SHIFT, '&', '0', '@', '2', /* cancel */
2041 MOD_MASK_SHIFT, '*', '1', '@', '4', /* command */
2042 MOD_MASK_SHIFT, '*', '2', '@', '5', /* copy */
2043 MOD_MASK_SHIFT, '*', '3', '@', '6', /* create */
2044 MOD_MASK_SHIFT, '*', '4', 'k', 'D', /* delete char */
2045 MOD_MASK_SHIFT, '*', '5', 'k', 'L', /* delete line */
2046 MOD_MASK_SHIFT, '*', '7', '@', '7', /* end */
2047 MOD_MASK_CTRL, KS_EXTRA, (int)KE_C_END, '@', '7', /* end */
2048 MOD_MASK_SHIFT, '*', '9', '@', '9', /* exit */
2049 MOD_MASK_SHIFT, '*', '0', '@', '0', /* find */
2050 MOD_MASK_SHIFT, '#', '1', '%', '1', /* help */
2051 MOD_MASK_SHIFT, '#', '2', 'k', 'h', /* home */
2052 MOD_MASK_CTRL, KS_EXTRA, (int)KE_C_HOME, 'k', 'h', /* home */
2053 MOD_MASK_SHIFT, '#', '3', 'k', 'I', /* insert */
2054 MOD_MASK_SHIFT, '#', '4', 'k', 'l', /* left arrow */
2055 MOD_MASK_CTRL, KS_EXTRA, (int)KE_C_LEFT, 'k', 'l', /* left arrow */
2056 MOD_MASK_SHIFT, '%', 'a', '%', '3', /* message */
2057 MOD_MASK_SHIFT, '%', 'b', '%', '4', /* move */
2058 MOD_MASK_SHIFT, '%', 'c', '%', '5', /* next */
2059 MOD_MASK_SHIFT, '%', 'd', '%', '7', /* options */
2060 MOD_MASK_SHIFT, '%', 'e', '%', '8', /* previous */
2061 MOD_MASK_SHIFT, '%', 'f', '%', '9', /* print */
2062 MOD_MASK_SHIFT, '%', 'g', '%', '0', /* redo */
2063 MOD_MASK_SHIFT, '%', 'h', '&', '3', /* replace */
2064 MOD_MASK_SHIFT, '%', 'i', 'k', 'r', /* right arr. */
2065 MOD_MASK_CTRL, KS_EXTRA, (int)KE_C_RIGHT, 'k', 'r', /* right arr. */
2066 MOD_MASK_SHIFT, '%', 'j', '&', '5', /* resume */
2067 MOD_MASK_SHIFT, '!', '1', '&', '6', /* save */
2068 MOD_MASK_SHIFT, '!', '2', '&', '7', /* suspend */
2069 MOD_MASK_SHIFT, '!', '3', '&', '8', /* undo */
2070 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_UP, 'k', 'u', /* up arrow */
2071 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_DOWN, 'k', 'd', /* down arrow */
2072
2073 /* vt100 F1 */
2074 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_XF1, KS_EXTRA, (int)KE_XF1,
2075 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_XF2, KS_EXTRA, (int)KE_XF2,
2076 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_XF3, KS_EXTRA, (int)KE_XF3,
2077 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_XF4, KS_EXTRA, (int)KE_XF4,
2078
2079 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F1, 'k', '1', /* F1 */
2080 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F2, 'k', '2',
2081 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F3, 'k', '3',
2082 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F4, 'k', '4',
2083 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F5, 'k', '5',
2084 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F6, 'k', '6',
2085 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F7, 'k', '7',
2086 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F8, 'k', '8',
2087 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F9, 'k', '9',
2088 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F10, 'k', ';', /* F10 */
2089
2090 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F11, 'F', '1',
2091 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F12, 'F', '2',
2092 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F13, 'F', '3',
2093 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F14, 'F', '4',
2094 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F15, 'F', '5',
2095 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F16, 'F', '6',
2096 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F17, 'F', '7',
2097 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F18, 'F', '8',
2098 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F19, 'F', '9',
2099 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F20, 'F', 'A',
2100
2101 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F21, 'F', 'B',
2102 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F22, 'F', 'C',
2103 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F23, 'F', 'D',
2104 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F24, 'F', 'E',
2105 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F25, 'F', 'F',
2106 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F26, 'F', 'G',
2107 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F27, 'F', 'H',
2108 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F28, 'F', 'I',
2109 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F29, 'F', 'J',
2110 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F30, 'F', 'K',
2111
2112 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F31, 'F', 'L',
2113 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F32, 'F', 'M',
2114 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F33, 'F', 'N',
2115 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F34, 'F', 'O',
2116 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F35, 'F', 'P',
2117 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F36, 'F', 'Q',
2118 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F37, 'F', 'R',
2119
2120 /* TAB pseudo code*/
2121 MOD_MASK_SHIFT, 'k', 'B', KS_EXTRA, (int)KE_TAB,
2122
2123 NUL
2124};
2125
2126static struct key_name_entry
2127{
2128 int key; /* Special key code or ascii value */
2129 char_u *name; /* Name of key */
2130} key_names_table[] =
2131{
2132 {' ', (char_u *)"Space"},
2133 {TAB, (char_u *)"Tab"},
2134 {K_TAB, (char_u *)"Tab"},
2135 {NL, (char_u *)"NL"},
2136 {NL, (char_u *)"NewLine"}, /* Alternative name */
2137 {NL, (char_u *)"LineFeed"}, /* Alternative name */
2138 {NL, (char_u *)"LF"}, /* Alternative name */
2139 {CAR, (char_u *)"CR"},
2140 {CAR, (char_u *)"Return"}, /* Alternative name */
2141 {CAR, (char_u *)"Enter"}, /* Alternative name */
2142 {K_BS, (char_u *)"BS"},
2143 {K_BS, (char_u *)"BackSpace"}, /* Alternative name */
2144 {ESC, (char_u *)"Esc"},
2145 {CSI, (char_u *)"CSI"},
2146 {K_CSI, (char_u *)"xCSI"},
2147 {'|', (char_u *)"Bar"},
2148 {'\\', (char_u *)"Bslash"},
2149 {K_DEL, (char_u *)"Del"},
2150 {K_DEL, (char_u *)"Delete"}, /* Alternative name */
2151 {K_KDEL, (char_u *)"kDel"},
2152 {K_UP, (char_u *)"Up"},
2153 {K_DOWN, (char_u *)"Down"},
2154 {K_LEFT, (char_u *)"Left"},
2155 {K_RIGHT, (char_u *)"Right"},
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002156 {K_XUP, (char_u *)"xUp"},
2157 {K_XDOWN, (char_u *)"xDown"},
2158 {K_XLEFT, (char_u *)"xLeft"},
2159 {K_XRIGHT, (char_u *)"xRight"},
Bram Moolenaar071d4272004-06-13 20:20:40 +00002160
2161 {K_F1, (char_u *)"F1"},
2162 {K_F2, (char_u *)"F2"},
2163 {K_F3, (char_u *)"F3"},
2164 {K_F4, (char_u *)"F4"},
2165 {K_F5, (char_u *)"F5"},
2166 {K_F6, (char_u *)"F6"},
2167 {K_F7, (char_u *)"F7"},
2168 {K_F8, (char_u *)"F8"},
2169 {K_F9, (char_u *)"F9"},
2170 {K_F10, (char_u *)"F10"},
2171
2172 {K_F11, (char_u *)"F11"},
2173 {K_F12, (char_u *)"F12"},
2174 {K_F13, (char_u *)"F13"},
2175 {K_F14, (char_u *)"F14"},
2176 {K_F15, (char_u *)"F15"},
2177 {K_F16, (char_u *)"F16"},
2178 {K_F17, (char_u *)"F17"},
2179 {K_F18, (char_u *)"F18"},
2180 {K_F19, (char_u *)"F19"},
2181 {K_F20, (char_u *)"F20"},
2182
2183 {K_F21, (char_u *)"F21"},
2184 {K_F22, (char_u *)"F22"},
2185 {K_F23, (char_u *)"F23"},
2186 {K_F24, (char_u *)"F24"},
2187 {K_F25, (char_u *)"F25"},
2188 {K_F26, (char_u *)"F26"},
2189 {K_F27, (char_u *)"F27"},
2190 {K_F28, (char_u *)"F28"},
2191 {K_F29, (char_u *)"F29"},
2192 {K_F30, (char_u *)"F30"},
2193
2194 {K_F31, (char_u *)"F31"},
2195 {K_F32, (char_u *)"F32"},
2196 {K_F33, (char_u *)"F33"},
2197 {K_F34, (char_u *)"F34"},
2198 {K_F35, (char_u *)"F35"},
2199 {K_F36, (char_u *)"F36"},
2200 {K_F37, (char_u *)"F37"},
2201
2202 {K_XF1, (char_u *)"xF1"},
2203 {K_XF2, (char_u *)"xF2"},
2204 {K_XF3, (char_u *)"xF3"},
2205 {K_XF4, (char_u *)"xF4"},
2206
2207 {K_HELP, (char_u *)"Help"},
2208 {K_UNDO, (char_u *)"Undo"},
2209 {K_INS, (char_u *)"Insert"},
2210 {K_INS, (char_u *)"Ins"}, /* Alternative name */
2211 {K_KINS, (char_u *)"kInsert"},
2212 {K_HOME, (char_u *)"Home"},
2213 {K_KHOME, (char_u *)"kHome"},
2214 {K_XHOME, (char_u *)"xHome"},
Bram Moolenaar68b76a62005-03-25 21:53:48 +00002215 {K_ZHOME, (char_u *)"zHome"},
Bram Moolenaar071d4272004-06-13 20:20:40 +00002216 {K_END, (char_u *)"End"},
2217 {K_KEND, (char_u *)"kEnd"},
2218 {K_XEND, (char_u *)"xEnd"},
Bram Moolenaar68b76a62005-03-25 21:53:48 +00002219 {K_ZEND, (char_u *)"zEnd"},
Bram Moolenaar071d4272004-06-13 20:20:40 +00002220 {K_PAGEUP, (char_u *)"PageUp"},
2221 {K_PAGEDOWN, (char_u *)"PageDown"},
2222 {K_KPAGEUP, (char_u *)"kPageUp"},
2223 {K_KPAGEDOWN, (char_u *)"kPageDown"},
2224
2225 {K_KPLUS, (char_u *)"kPlus"},
2226 {K_KMINUS, (char_u *)"kMinus"},
2227 {K_KDIVIDE, (char_u *)"kDivide"},
2228 {K_KMULTIPLY, (char_u *)"kMultiply"},
2229 {K_KENTER, (char_u *)"kEnter"},
2230 {K_KPOINT, (char_u *)"kPoint"},
2231
2232 {K_K0, (char_u *)"k0"},
2233 {K_K1, (char_u *)"k1"},
2234 {K_K2, (char_u *)"k2"},
2235 {K_K3, (char_u *)"k3"},
2236 {K_K4, (char_u *)"k4"},
2237 {K_K5, (char_u *)"k5"},
2238 {K_K6, (char_u *)"k6"},
2239 {K_K7, (char_u *)"k7"},
2240 {K_K8, (char_u *)"k8"},
2241 {K_K9, (char_u *)"k9"},
2242
2243 {'<', (char_u *)"lt"},
2244
2245 {K_MOUSE, (char_u *)"Mouse"},
2246 {K_NETTERM_MOUSE, (char_u *)"NetMouse"},
2247 {K_DEC_MOUSE, (char_u *)"DecMouse"},
2248 {K_JSBTERM_MOUSE, (char_u *)"JsbMouse"},
2249 {K_PTERM_MOUSE, (char_u *)"PtermMouse"},
2250 {K_LEFTMOUSE, (char_u *)"LeftMouse"},
2251 {K_LEFTMOUSE_NM, (char_u *)"LeftMouseNM"},
2252 {K_LEFTDRAG, (char_u *)"LeftDrag"},
2253 {K_LEFTRELEASE, (char_u *)"LeftRelease"},
2254 {K_LEFTRELEASE_NM, (char_u *)"LeftReleaseNM"},
2255 {K_MIDDLEMOUSE, (char_u *)"MiddleMouse"},
2256 {K_MIDDLEDRAG, (char_u *)"MiddleDrag"},
2257 {K_MIDDLERELEASE, (char_u *)"MiddleRelease"},
2258 {K_RIGHTMOUSE, (char_u *)"RightMouse"},
2259 {K_RIGHTDRAG, (char_u *)"RightDrag"},
2260 {K_RIGHTRELEASE, (char_u *)"RightRelease"},
2261 {K_MOUSEDOWN, (char_u *)"MouseDown"},
2262 {K_MOUSEUP, (char_u *)"MouseUp"},
2263 {K_X1MOUSE, (char_u *)"X1Mouse"},
2264 {K_X1DRAG, (char_u *)"X1Drag"},
2265 {K_X1RELEASE, (char_u *)"X1Release"},
2266 {K_X2MOUSE, (char_u *)"X2Mouse"},
2267 {K_X2DRAG, (char_u *)"X2Drag"},
2268 {K_X2RELEASE, (char_u *)"X2Release"},
2269 {K_DROP, (char_u *)"Drop"},
2270 {K_ZERO, (char_u *)"Nul"},
2271#ifdef FEAT_EVAL
2272 {K_SNR, (char_u *)"SNR"},
2273#endif
2274 {K_PLUG, (char_u *)"Plug"},
2275 {0, NULL}
2276};
2277
2278#define KEY_NAMES_TABLE_LEN (sizeof(key_names_table) / sizeof(struct key_name_entry))
2279
2280#ifdef FEAT_MOUSE
2281static struct mousetable
2282{
2283 int pseudo_code; /* Code for pseudo mouse event */
2284 int button; /* Which mouse button is it? */
2285 int is_click; /* Is it a mouse button click event? */
2286 int is_drag; /* Is it a mouse drag event? */
2287} mouse_table[] =
2288{
2289 {(int)KE_LEFTMOUSE, MOUSE_LEFT, TRUE, FALSE},
2290#ifdef FEAT_GUI
2291 {(int)KE_LEFTMOUSE_NM, MOUSE_LEFT, TRUE, FALSE},
2292#endif
2293 {(int)KE_LEFTDRAG, MOUSE_LEFT, FALSE, TRUE},
2294 {(int)KE_LEFTRELEASE, MOUSE_LEFT, FALSE, FALSE},
2295#ifdef FEAT_GUI
2296 {(int)KE_LEFTRELEASE_NM, MOUSE_LEFT, FALSE, FALSE},
2297#endif
2298 {(int)KE_MIDDLEMOUSE, MOUSE_MIDDLE, TRUE, FALSE},
2299 {(int)KE_MIDDLEDRAG, MOUSE_MIDDLE, FALSE, TRUE},
2300 {(int)KE_MIDDLERELEASE, MOUSE_MIDDLE, FALSE, FALSE},
2301 {(int)KE_RIGHTMOUSE, MOUSE_RIGHT, TRUE, FALSE},
2302 {(int)KE_RIGHTDRAG, MOUSE_RIGHT, FALSE, TRUE},
2303 {(int)KE_RIGHTRELEASE, MOUSE_RIGHT, FALSE, FALSE},
2304 {(int)KE_X1MOUSE, MOUSE_X1, TRUE, FALSE},
2305 {(int)KE_X1DRAG, MOUSE_X1, FALSE, TRUE},
2306 {(int)KE_X1RELEASE, MOUSE_X1, FALSE, FALSE},
2307 {(int)KE_X2MOUSE, MOUSE_X2, TRUE, FALSE},
2308 {(int)KE_X2DRAG, MOUSE_X2, FALSE, TRUE},
2309 {(int)KE_X2RELEASE, MOUSE_X2, FALSE, FALSE},
2310 /* DRAG without CLICK */
2311 {(int)KE_IGNORE, MOUSE_RELEASE, FALSE, TRUE},
2312 /* RELEASE without CLICK */
2313 {(int)KE_IGNORE, MOUSE_RELEASE, FALSE, FALSE},
2314 {0, 0, 0, 0},
2315};
2316#endif /* FEAT_MOUSE */
2317
2318/*
2319 * Return the modifier mask bit (MOD_MASK_*) which corresponds to the given
2320 * modifier name ('S' for Shift, 'C' for Ctrl etc).
2321 */
2322 int
2323name_to_mod_mask(c)
2324 int c;
2325{
2326 int i;
2327
2328 c = TOUPPER_ASC(c);
2329 for (i = 0; mod_mask_table[i].mod_mask != 0; i++)
2330 if (c == mod_mask_table[i].name)
2331 return mod_mask_table[i].mod_flag;
2332 return 0;
2333}
2334
Bram Moolenaar071d4272004-06-13 20:20:40 +00002335/*
2336 * Check if if there is a special key code for "key" that includes the
2337 * modifiers specified.
2338 */
2339 int
2340simplify_key(key, modifiers)
2341 int key;
2342 int *modifiers;
2343{
2344 int i;
2345 int key0;
2346 int key1;
2347
2348 if (*modifiers & (MOD_MASK_SHIFT | MOD_MASK_CTRL | MOD_MASK_ALT))
2349 {
2350 /* TAB is a special case */
2351 if (key == TAB && (*modifiers & MOD_MASK_SHIFT))
2352 {
2353 *modifiers &= ~MOD_MASK_SHIFT;
2354 return K_S_TAB;
2355 }
2356 key0 = KEY2TERMCAP0(key);
2357 key1 = KEY2TERMCAP1(key);
2358 for (i = 0; modifier_keys_table[i] != NUL; i += MOD_KEYS_ENTRY_SIZE)
2359 if (key0 == modifier_keys_table[i + 3]
2360 && key1 == modifier_keys_table[i + 4]
2361 && (*modifiers & modifier_keys_table[i]))
2362 {
2363 *modifiers &= ~modifier_keys_table[i];
2364 return TERMCAP2KEY(modifier_keys_table[i + 1],
2365 modifier_keys_table[i + 2]);
2366 }
2367 }
2368 return key;
2369}
2370
2371/*
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002372 * Change <xHome> to <Home>, <xUp> to <Up>, etc.
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002373 */
2374 int
2375handle_x_keys(key)
2376 int key;
2377{
2378 switch (key)
2379 {
2380 case K_XUP: return K_UP;
2381 case K_XDOWN: return K_DOWN;
2382 case K_XLEFT: return K_LEFT;
2383 case K_XRIGHT: return K_RIGHT;
2384 case K_XHOME: return K_HOME;
Bram Moolenaar68b76a62005-03-25 21:53:48 +00002385 case K_ZHOME: return K_HOME;
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002386 case K_XEND: return K_END;
Bram Moolenaar68b76a62005-03-25 21:53:48 +00002387 case K_ZEND: return K_END;
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002388 case K_XF1: return K_F1;
2389 case K_XF2: return K_F2;
2390 case K_XF3: return K_F3;
2391 case K_XF4: return K_F4;
2392 case K_S_XF1: return K_S_F1;
2393 case K_S_XF2: return K_S_F2;
2394 case K_S_XF3: return K_S_F3;
2395 case K_S_XF4: return K_S_F4;
2396 }
2397 return key;
2398}
2399
2400/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002401 * Return a string which contains the name of the given key when the given
2402 * modifiers are down.
2403 */
2404 char_u *
2405get_special_key_name(c, modifiers)
2406 int c;
2407 int modifiers;
2408{
2409 static char_u string[MAX_KEY_NAME_LEN + 1];
2410
2411 int i, idx;
2412 int table_idx;
2413 char_u *s;
2414
2415 string[0] = '<';
2416 idx = 1;
2417
2418 /* Key that stands for a normal character. */
2419 if (IS_SPECIAL(c) && KEY2TERMCAP0(c) == KS_KEY)
2420 c = KEY2TERMCAP1(c);
2421
2422 /*
2423 * Translate shifted special keys into unshifted keys and set modifier.
2424 * Same for CTRL and ALT modifiers.
2425 */
2426 if (IS_SPECIAL(c))
2427 {
2428 for (i = 0; modifier_keys_table[i] != 0; i += MOD_KEYS_ENTRY_SIZE)
2429 if ( KEY2TERMCAP0(c) == (int)modifier_keys_table[i + 1]
2430 && (int)KEY2TERMCAP1(c) == (int)modifier_keys_table[i + 2])
2431 {
2432 modifiers |= modifier_keys_table[i];
2433 c = TERMCAP2KEY(modifier_keys_table[i + 3],
2434 modifier_keys_table[i + 4]);
2435 break;
2436 }
2437 }
2438
2439 /* try to find the key in the special key table */
2440 table_idx = find_special_key_in_table(c);
2441
2442 /*
2443 * When not a known special key, and not a printable character, try to
2444 * extract modifiers.
2445 */
2446 if (c > 0
2447#ifdef FEAT_MBYTE
2448 && (*mb_char2len)(c) == 1
2449#endif
2450 )
2451 {
2452 if (table_idx < 0
2453 && (!vim_isprintc(c) || (c & 0x7f) == ' ')
2454 && (c & 0x80))
2455 {
2456 c &= 0x7f;
2457 modifiers |= MOD_MASK_ALT;
2458 /* try again, to find the un-alted key in the special key table */
2459 table_idx = find_special_key_in_table(c);
2460 }
2461 if (table_idx < 0 && !vim_isprintc(c) && c < ' ')
2462 {
2463#ifdef EBCDIC
2464 c = CtrlChar(c);
2465#else
2466 c += '@';
2467#endif
2468 modifiers |= MOD_MASK_CTRL;
2469 }
2470 }
2471
2472 /* translate the modifier into a string */
2473 for (i = 0; mod_mask_table[i].name != 'A'; i++)
2474 if ((modifiers & mod_mask_table[i].mod_mask)
2475 == mod_mask_table[i].mod_flag)
2476 {
2477 string[idx++] = mod_mask_table[i].name;
2478 string[idx++] = (char_u)'-';
2479 }
2480
2481 if (table_idx < 0) /* unknown special key, may output t_xx */
2482 {
2483 if (IS_SPECIAL(c))
2484 {
2485 string[idx++] = 't';
2486 string[idx++] = '_';
2487 string[idx++] = KEY2TERMCAP0(c);
2488 string[idx++] = KEY2TERMCAP1(c);
2489 }
2490 /* Not a special key, only modifiers, output directly */
2491 else
2492 {
2493#ifdef FEAT_MBYTE
2494 if (has_mbyte && (*mb_char2len)(c) > 1)
2495 idx += (*mb_char2bytes)(c, string + idx);
2496 else
2497#endif
2498 if (vim_isprintc(c))
2499 string[idx++] = c;
2500 else
2501 {
2502 s = transchar(c);
2503 while (*s)
2504 string[idx++] = *s++;
2505 }
2506 }
2507 }
2508 else /* use name of special key */
2509 {
2510 STRCPY(string + idx, key_names_table[table_idx].name);
2511 idx = (int)STRLEN(string);
2512 }
2513 string[idx++] = '>';
2514 string[idx] = NUL;
2515 return string;
2516}
2517
2518/*
2519 * Try translating a <> name at (*srcp)[] to dst[].
2520 * Return the number of characters added to dst[], zero for no match.
2521 * If there is a match, srcp is advanced to after the <> name.
2522 * dst[] must be big enough to hold the result (up to six characters)!
2523 */
2524 int
2525trans_special(srcp, dst, keycode)
2526 char_u **srcp;
2527 char_u *dst;
2528 int keycode; /* prefer key code, e.g. K_DEL instead of DEL */
2529{
2530 int modifiers = 0;
2531 int key;
2532 int dlen = 0;
2533
2534 key = find_special_key(srcp, &modifiers, keycode);
2535 if (key == 0)
2536 return 0;
2537
2538 /* Put the appropriate modifier in a string */
2539 if (modifiers != 0)
2540 {
2541 dst[dlen++] = K_SPECIAL;
2542 dst[dlen++] = KS_MODIFIER;
2543 dst[dlen++] = modifiers;
2544 }
2545
2546 if (IS_SPECIAL(key))
2547 {
2548 dst[dlen++] = K_SPECIAL;
2549 dst[dlen++] = KEY2TERMCAP0(key);
2550 dst[dlen++] = KEY2TERMCAP1(key);
2551 }
2552#ifdef FEAT_MBYTE
2553 else if (has_mbyte && !keycode)
2554 dlen += (*mb_char2bytes)(key, dst + dlen);
2555#endif
2556 else if (keycode)
2557 dlen = (int)(add_char2buf(key, dst + dlen) - dst);
2558 else
2559 dst[dlen++] = key;
2560
2561 return dlen;
2562}
2563
2564/*
2565 * Try translating a <> name at (*srcp)[], return the key and modifiers.
2566 * srcp is advanced to after the <> name.
2567 * returns 0 if there is no match.
2568 */
2569 int
2570find_special_key(srcp, modp, keycode)
2571 char_u **srcp;
2572 int *modp;
2573 int keycode; /* prefer key code, e.g. K_DEL instead of DEL */
2574{
2575 char_u *last_dash;
2576 char_u *end_of_name;
2577 char_u *src;
2578 char_u *bp;
2579 int modifiers;
2580 int bit;
2581 int key;
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00002582 unsigned long n;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002583
2584 src = *srcp;
2585 if (src[0] != '<')
2586 return 0;
2587
2588 /* Find end of modifier list */
2589 last_dash = src;
2590 for (bp = src + 1; *bp == '-' || vim_isIDc(*bp); bp++)
2591 {
2592 if (*bp == '-')
2593 {
2594 last_dash = bp;
2595 if (bp[1] != NUL && bp[2] == '>')
2596 ++bp; /* anything accepted, like <C-?> */
2597 }
2598 if (bp[0] == 't' && bp[1] == '_' && bp[2] && bp[3])
2599 bp += 3; /* skip t_xx, xx may be '-' or '>' */
2600 }
2601
2602 if (*bp == '>') /* found matching '>' */
2603 {
2604 end_of_name = bp + 1;
2605
2606 if (STRNICMP(src + 1, "char-", 5) == 0 && VIM_ISDIGIT(src[6]))
2607 {
2608 /* <Char-123> or <Char-033> or <Char-0x33> */
2609 vim_str2nr(src + 6, NULL, NULL, TRUE, TRUE, NULL, &n);
2610 *modp = 0;
2611 *srcp = end_of_name;
2612 return (int)n;
2613 }
2614
2615 /* Which modifiers are given? */
2616 modifiers = 0x0;
2617 for (bp = src + 1; bp < last_dash; bp++)
2618 {
2619 if (*bp != '-')
2620 {
2621 bit = name_to_mod_mask(*bp);
2622 if (bit == 0x0)
2623 break; /* Illegal modifier name */
2624 modifiers |= bit;
2625 }
2626 }
2627
2628 /*
2629 * Legal modifier name.
2630 */
2631 if (bp >= last_dash)
2632 {
2633 /*
2634 * Modifier with single letter, or special key name.
2635 */
2636 if (modifiers != 0 && last_dash[2] == '>')
2637 key = last_dash[1];
2638 else
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002639 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00002640 key = get_special_key_code(last_dash + 1);
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002641 key = handle_x_keys(key);
2642 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002643
2644 /*
2645 * get_special_key_code() may return NUL for invalid
2646 * special key name.
2647 */
2648 if (key != NUL)
2649 {
2650 /*
2651 * Only use a modifier when there is no special key code that
2652 * includes the modifier.
2653 */
2654 key = simplify_key(key, &modifiers);
2655
2656 if (!keycode)
2657 {
2658 /* don't want keycode, use single byte code */
2659 if (key == K_BS)
2660 key = BS;
2661 else if (key == K_DEL || key == K_KDEL)
2662 key = DEL;
2663 }
2664
2665 /*
2666 * Normal Key with modifier: Try to make a single byte code.
2667 */
2668 if (!IS_SPECIAL(key))
2669 key = extract_modifiers(key, &modifiers);
2670
2671 *modp = modifiers;
2672 *srcp = end_of_name;
2673 return key;
2674 }
2675 }
2676 }
2677 return 0;
2678}
2679
2680/*
2681 * Try to include modifiers in the key.
2682 * Changes "Shift-a" to 'A', "Alt-A" to 0xc0, etc.
2683 */
2684 int
2685extract_modifiers(key, modp)
2686 int key;
2687 int *modp;
2688{
2689 int modifiers = *modp;
2690
2691#ifdef MACOS
2692 /* Command-key really special, No fancynest */
2693 if (!(modifiers & MOD_MASK_CMD))
2694#endif
2695 if ((modifiers & MOD_MASK_SHIFT) && ASCII_ISALPHA(key))
2696 {
2697 key = TOUPPER_ASC(key);
2698 modifiers &= ~MOD_MASK_SHIFT;
2699 }
2700 if ((modifiers & MOD_MASK_CTRL)
2701#ifdef EBCDIC
2702 /* * TODO: EBCDIC Better use:
2703 * && (Ctrl_chr(key) || key == '?')
2704 * ??? */
2705 && strchr("?@ABCDEFGHIJKLMNOPQRSTUVWXYZ[\\]^_", key)
2706 != NULL
2707#else
2708 && ((key >= '?' && key <= '_') || ASCII_ISALPHA(key))
2709#endif
2710 )
2711 {
2712 key = Ctrl_chr(key);
2713 modifiers &= ~MOD_MASK_CTRL;
2714 /* <C-@> is <Nul> */
2715 if (key == 0)
2716 key = K_ZERO;
2717 }
2718#ifdef MACOS
2719 /* Command-key really special, No fancynest */
2720 if (!(modifiers & MOD_MASK_CMD))
2721#endif
2722 if ((modifiers & MOD_MASK_ALT) && key < 0x80
2723#ifdef FEAT_MBYTE
2724 && !enc_dbcs /* avoid creating a lead byte */
2725#endif
2726 )
2727 {
2728 key |= 0x80;
2729 modifiers &= ~MOD_MASK_ALT; /* remove the META modifier */
2730 }
2731
2732 *modp = modifiers;
2733 return key;
2734}
2735
2736/*
2737 * Try to find key "c" in the special key table.
2738 * Return the index when found, -1 when not found.
2739 */
2740 int
2741find_special_key_in_table(c)
2742 int c;
2743{
2744 int i;
2745
2746 for (i = 0; key_names_table[i].name != NULL; i++)
2747 if (c == key_names_table[i].key)
2748 break;
2749 if (key_names_table[i].name == NULL)
2750 i = -1;
2751 return i;
2752}
2753
2754/*
2755 * Find the special key with the given name (the given string does not have to
2756 * end with NUL, the name is assumed to end before the first non-idchar).
2757 * If the name starts with "t_" the next two characters are interpreted as a
2758 * termcap name.
2759 * Return the key code, or 0 if not found.
2760 */
2761 int
2762get_special_key_code(name)
2763 char_u *name;
2764{
2765 char_u *table_name;
2766 char_u string[3];
2767 int i, j;
2768
2769 /*
2770 * If it's <t_xx> we get the code for xx from the termcap
2771 */
2772 if (name[0] == 't' && name[1] == '_' && name[2] != NUL && name[3] != NUL)
2773 {
2774 string[0] = name[2];
2775 string[1] = name[3];
2776 string[2] = NUL;
2777 if (add_termcap_entry(string, FALSE) == OK)
2778 return TERMCAP2KEY(name[2], name[3]);
2779 }
2780 else
2781 for (i = 0; key_names_table[i].name != NULL; i++)
2782 {
2783 table_name = key_names_table[i].name;
2784 for (j = 0; vim_isIDc(name[j]) && table_name[j] != NUL; j++)
2785 if (TOLOWER_ASC(table_name[j]) != TOLOWER_ASC(name[j]))
2786 break;
2787 if (!vim_isIDc(name[j]) && table_name[j] == NUL)
2788 return key_names_table[i].key;
2789 }
2790 return 0;
2791}
2792
Bram Moolenaar05bb9532008-07-04 09:44:11 +00002793#if defined(FEAT_CMDL_COMPL) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002794 char_u *
2795get_key_name(i)
2796 int i;
2797{
2798 if (i >= KEY_NAMES_TABLE_LEN)
2799 return NULL;
2800 return key_names_table[i].name;
2801}
2802#endif
2803
Bram Moolenaar05bb9532008-07-04 09:44:11 +00002804#if defined(FEAT_MOUSE) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002805/*
2806 * Look up the given mouse code to return the relevant information in the other
2807 * arguments. Return which button is down or was released.
2808 */
2809 int
2810get_mouse_button(code, is_click, is_drag)
2811 int code;
2812 int *is_click;
2813 int *is_drag;
2814{
2815 int i;
2816
2817 for (i = 0; mouse_table[i].pseudo_code; i++)
2818 if (code == mouse_table[i].pseudo_code)
2819 {
2820 *is_click = mouse_table[i].is_click;
2821 *is_drag = mouse_table[i].is_drag;
2822 return mouse_table[i].button;
2823 }
2824 return 0; /* Shouldn't get here */
2825}
2826
2827/*
2828 * Return the appropriate pseudo mouse event token (KE_LEFTMOUSE etc) based on
2829 * the given information about which mouse button is down, and whether the
2830 * mouse was clicked, dragged or released.
2831 */
2832 int
2833get_pseudo_mouse_code(button, is_click, is_drag)
2834 int button; /* eg MOUSE_LEFT */
2835 int is_click;
2836 int is_drag;
2837{
2838 int i;
2839
2840 for (i = 0; mouse_table[i].pseudo_code; i++)
2841 if (button == mouse_table[i].button
2842 && is_click == mouse_table[i].is_click
2843 && is_drag == mouse_table[i].is_drag)
2844 {
2845#ifdef FEAT_GUI
Bram Moolenaarc91506a2005-04-24 22:04:21 +00002846 /* Trick: a non mappable left click and release has mouse_col -1
2847 * or added MOUSE_COLOFF. Used for 'mousefocus' in
2848 * gui_mouse_moved() */
2849 if (mouse_col < 0 || mouse_col > MOUSE_COLOFF)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002850 {
Bram Moolenaarc91506a2005-04-24 22:04:21 +00002851 if (mouse_col < 0)
2852 mouse_col = 0;
2853 else
2854 mouse_col -= MOUSE_COLOFF;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002855 if (mouse_table[i].pseudo_code == (int)KE_LEFTMOUSE)
2856 return (int)KE_LEFTMOUSE_NM;
2857 if (mouse_table[i].pseudo_code == (int)KE_LEFTRELEASE)
2858 return (int)KE_LEFTRELEASE_NM;
2859 }
2860#endif
2861 return mouse_table[i].pseudo_code;
2862 }
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00002863 return (int)KE_IGNORE; /* not recognized, ignore it */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002864}
2865#endif /* FEAT_MOUSE */
2866
2867/*
2868 * Return the current end-of-line type: EOL_DOS, EOL_UNIX or EOL_MAC.
2869 */
2870 int
2871get_fileformat(buf)
2872 buf_T *buf;
2873{
2874 int c = *buf->b_p_ff;
2875
2876 if (buf->b_p_bin || c == 'u')
2877 return EOL_UNIX;
2878 if (c == 'm')
2879 return EOL_MAC;
2880 return EOL_DOS;
2881}
2882
2883/*
2884 * Like get_fileformat(), but override 'fileformat' with "p" for "++opt=val"
2885 * argument.
2886 */
2887 int
2888get_fileformat_force(buf, eap)
2889 buf_T *buf;
2890 exarg_T *eap; /* can be NULL! */
2891{
2892 int c;
2893
2894 if (eap != NULL && eap->force_ff != 0)
2895 c = eap->cmd[eap->force_ff];
2896 else
2897 {
2898 if ((eap != NULL && eap->force_bin != 0)
2899 ? (eap->force_bin == FORCE_BIN) : buf->b_p_bin)
2900 return EOL_UNIX;
2901 c = *buf->b_p_ff;
2902 }
2903 if (c == 'u')
2904 return EOL_UNIX;
2905 if (c == 'm')
2906 return EOL_MAC;
2907 return EOL_DOS;
2908}
2909
2910/*
2911 * Set the current end-of-line type to EOL_DOS, EOL_UNIX or EOL_MAC.
2912 * Sets both 'textmode' and 'fileformat'.
2913 * Note: Does _not_ set global value of 'textmode'!
2914 */
2915 void
2916set_fileformat(t, opt_flags)
2917 int t;
2918 int opt_flags; /* OPT_LOCAL and/or OPT_GLOBAL */
2919{
2920 char *p = NULL;
2921
2922 switch (t)
2923 {
2924 case EOL_DOS:
2925 p = FF_DOS;
2926 curbuf->b_p_tx = TRUE;
2927 break;
2928 case EOL_UNIX:
2929 p = FF_UNIX;
2930 curbuf->b_p_tx = FALSE;
2931 break;
2932 case EOL_MAC:
2933 p = FF_MAC;
2934 curbuf->b_p_tx = FALSE;
2935 break;
2936 }
2937 if (p != NULL)
2938 set_string_option_direct((char_u *)"ff", -1, (char_u *)p,
Bram Moolenaar5e3cb7e2006-02-27 23:58:35 +00002939 OPT_FREE | opt_flags, 0);
2940
Bram Moolenaar071d4272004-06-13 20:20:40 +00002941#ifdef FEAT_WINDOWS
Bram Moolenaarf740b292006-02-16 22:11:02 +00002942 /* This may cause the buffer to become (un)modified. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002943 check_status(curbuf);
Bram Moolenaar997fb4b2006-02-17 21:53:23 +00002944 redraw_tabline = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002945#endif
2946#ifdef FEAT_TITLE
2947 need_maketitle = TRUE; /* set window title later */
2948#endif
2949}
2950
2951/*
2952 * Return the default fileformat from 'fileformats'.
2953 */
2954 int
2955default_fileformat()
2956{
2957 switch (*p_ffs)
2958 {
2959 case 'm': return EOL_MAC;
2960 case 'd': return EOL_DOS;
2961 }
2962 return EOL_UNIX;
2963}
2964
2965/*
2966 * Call shell. Calls mch_call_shell, with 'shellxquote' added.
2967 */
2968 int
2969call_shell(cmd, opt)
2970 char_u *cmd;
2971 int opt;
2972{
2973 char_u *ncmd;
2974 int retval;
Bram Moolenaar05159a02005-02-26 23:04:13 +00002975#ifdef FEAT_PROFILE
2976 proftime_T wait_time;
2977#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002978
2979 if (p_verbose > 3)
2980 {
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00002981 verbose_enter();
Bram Moolenaar051b7822005-05-19 21:00:46 +00002982 smsg((char_u *)_("Calling shell to execute: \"%s\""),
Bram Moolenaar071d4272004-06-13 20:20:40 +00002983 cmd == NULL ? p_sh : cmd);
2984 out_char('\n');
2985 cursor_on();
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00002986 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00002987 }
2988
Bram Moolenaar05159a02005-02-26 23:04:13 +00002989#ifdef FEAT_PROFILE
Bram Moolenaar01265852006-03-20 21:50:15 +00002990 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00002991 prof_child_enter(&wait_time);
2992#endif
2993
Bram Moolenaar071d4272004-06-13 20:20:40 +00002994 if (*p_sh == NUL)
2995 {
2996 EMSG(_(e_shellempty));
2997 retval = -1;
2998 }
2999 else
3000 {
3001#ifdef FEAT_GUI_MSWIN
3002 /* Don't hide the pointer while executing a shell command. */
3003 gui_mch_mousehide(FALSE);
3004#endif
3005#ifdef FEAT_GUI
3006 ++hold_gui_events;
3007#endif
3008 /* The external command may update a tags file, clear cached tags. */
3009 tag_freematch();
3010
3011 if (cmd == NULL || *p_sxq == NUL)
3012 retval = mch_call_shell(cmd, opt);
3013 else
3014 {
3015 ncmd = alloc((unsigned)(STRLEN(cmd) + STRLEN(p_sxq) * 2 + 1));
3016 if (ncmd != NULL)
3017 {
3018 STRCPY(ncmd, p_sxq);
3019 STRCAT(ncmd, cmd);
3020 STRCAT(ncmd, p_sxq);
3021 retval = mch_call_shell(ncmd, opt);
3022 vim_free(ncmd);
3023 }
3024 else
3025 retval = -1;
3026 }
3027#ifdef FEAT_GUI
3028 --hold_gui_events;
3029#endif
3030 /*
3031 * Check the window size, in case it changed while executing the
3032 * external command.
3033 */
3034 shell_resized_check();
3035 }
3036
3037#ifdef FEAT_EVAL
3038 set_vim_var_nr(VV_SHELL_ERROR, (long)retval);
Bram Moolenaar05159a02005-02-26 23:04:13 +00003039# ifdef FEAT_PROFILE
Bram Moolenaar01265852006-03-20 21:50:15 +00003040 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003041 prof_child_exit(&wait_time);
3042# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003043#endif
3044
3045 return retval;
3046}
3047
3048/*
Bram Moolenaar01265852006-03-20 21:50:15 +00003049 * VISUAL, SELECTMODE and OP_PENDING State are never set, they are equal to
3050 * NORMAL State with a condition. This function returns the real State.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003051 */
3052 int
3053get_real_state()
3054{
3055 if (State & NORMAL)
3056 {
3057#ifdef FEAT_VISUAL
3058 if (VIsual_active)
Bram Moolenaar01265852006-03-20 21:50:15 +00003059 {
3060 if (VIsual_select)
3061 return SELECTMODE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003062 return VISUAL;
Bram Moolenaar01265852006-03-20 21:50:15 +00003063 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003064 else
3065#endif
3066 if (finish_op)
Bram Moolenaar01265852006-03-20 21:50:15 +00003067 return OP_PENDING;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003068 }
3069 return State;
3070}
3071
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003072#if defined(FEAT_MBYTE) || defined(PROTO)
3073/*
3074 * Return TRUE if "p" points to just after a path separator.
3075 * Take care of multi-byte characters.
3076 * "b" must point to the start of the file name
3077 */
3078 int
3079after_pathsep(b, p)
3080 char_u *b;
3081 char_u *p;
3082{
3083 return vim_ispathsep(p[-1])
3084 && (!has_mbyte || (*mb_head_off)(b, p - 1) == 0);
3085}
3086#endif
3087
3088/*
3089 * Return TRUE if file names "f1" and "f2" are in the same directory.
3090 * "f1" may be a short name, "f2" must be a full path.
3091 */
3092 int
3093same_directory(f1, f2)
3094 char_u *f1;
3095 char_u *f2;
3096{
3097 char_u ffname[MAXPATHL];
3098 char_u *t1;
3099 char_u *t2;
3100
3101 /* safety check */
3102 if (f1 == NULL || f2 == NULL)
3103 return FALSE;
3104
3105 (void)vim_FullName(f1, ffname, MAXPATHL, FALSE);
3106 t1 = gettail_sep(ffname);
3107 t2 = gettail_sep(f2);
3108 return (t1 - ffname == t2 - f2
3109 && pathcmp((char *)ffname, (char *)f2, (int)(t1 - ffname)) == 0);
3110}
3111
Bram Moolenaar071d4272004-06-13 20:20:40 +00003112#if defined(FEAT_SESSION) || defined(MSWIN) || defined(FEAT_GUI_MAC) \
Bram Moolenaar9372a112005-12-06 19:59:18 +00003113 || ((defined(FEAT_GUI_GTK)) \
Bram Moolenaar843ee412004-06-30 16:16:41 +00003114 && ( defined(FEAT_WINDOWS) || defined(FEAT_DND)) ) \
Bram Moolenaar071d4272004-06-13 20:20:40 +00003115 || defined(FEAT_SUN_WORKSHOP) || defined(FEAT_NETBEANS_INTG) \
3116 || defined(PROTO)
3117/*
3118 * Change to a file's directory.
3119 * Caller must call shorten_fnames()!
3120 * Return OK or FAIL.
3121 */
3122 int
3123vim_chdirfile(fname)
3124 char_u *fname;
3125{
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003126 char_u dir[MAXPATHL];
Bram Moolenaar071d4272004-06-13 20:20:40 +00003127
Bram Moolenaarbbebc852005-07-18 21:47:53 +00003128 vim_strncpy(dir, fname, MAXPATHL - 1);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003129 *gettail_sep(dir) = NUL;
3130 return mch_chdir((char *)dir) == 0 ? OK : FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003131}
3132#endif
3133
3134#if defined(STAT_IGNORES_SLASH) || defined(PROTO)
3135/*
3136 * Check if "name" ends in a slash and is not a directory.
3137 * Used for systems where stat() ignores a trailing slash on a file name.
3138 * The Vim code assumes a trailing slash is only ignored for a directory.
3139 */
3140 int
3141illegal_slash(name)
3142 char *name;
3143{
3144 if (name[0] == NUL)
3145 return FALSE; /* no file name is not illegal */
3146 if (name[strlen(name) - 1] != '/')
3147 return FALSE; /* no trailing slash */
3148 if (mch_isdir((char_u *)name))
3149 return FALSE; /* trailing slash for a directory */
3150 return TRUE;
3151}
3152#endif
3153
3154#if defined(CURSOR_SHAPE) || defined(PROTO)
3155
3156/*
3157 * Handling of cursor and mouse pointer shapes in various modes.
3158 */
3159
3160cursorentry_T shape_table[SHAPE_IDX_COUNT] =
3161{
3162 /* The values will be filled in from the 'guicursor' and 'mouseshape'
3163 * defaults when Vim starts.
3164 * Adjust the SHAPE_IDX_ defines when making changes! */
3165 {0, 0, 0, 700L, 400L, 250L, 0, 0, "n", SHAPE_CURSOR+SHAPE_MOUSE},
3166 {0, 0, 0, 700L, 400L, 250L, 0, 0, "v", SHAPE_CURSOR+SHAPE_MOUSE},
3167 {0, 0, 0, 700L, 400L, 250L, 0, 0, "i", SHAPE_CURSOR+SHAPE_MOUSE},
3168 {0, 0, 0, 700L, 400L, 250L, 0, 0, "r", SHAPE_CURSOR+SHAPE_MOUSE},
3169 {0, 0, 0, 700L, 400L, 250L, 0, 0, "c", SHAPE_CURSOR+SHAPE_MOUSE},
3170 {0, 0, 0, 700L, 400L, 250L, 0, 0, "ci", SHAPE_CURSOR+SHAPE_MOUSE},
3171 {0, 0, 0, 700L, 400L, 250L, 0, 0, "cr", SHAPE_CURSOR+SHAPE_MOUSE},
3172 {0, 0, 0, 700L, 400L, 250L, 0, 0, "o", SHAPE_CURSOR+SHAPE_MOUSE},
3173 {0, 0, 0, 700L, 400L, 250L, 0, 0, "ve", SHAPE_CURSOR+SHAPE_MOUSE},
3174 {0, 0, 0, 0L, 0L, 0L, 0, 0, "e", SHAPE_MOUSE},
3175 {0, 0, 0, 0L, 0L, 0L, 0, 0, "s", SHAPE_MOUSE},
3176 {0, 0, 0, 0L, 0L, 0L, 0, 0, "sd", SHAPE_MOUSE},
3177 {0, 0, 0, 0L, 0L, 0L, 0, 0, "vs", SHAPE_MOUSE},
3178 {0, 0, 0, 0L, 0L, 0L, 0, 0, "vd", SHAPE_MOUSE},
3179 {0, 0, 0, 0L, 0L, 0L, 0, 0, "m", SHAPE_MOUSE},
3180 {0, 0, 0, 0L, 0L, 0L, 0, 0, "ml", SHAPE_MOUSE},
3181 {0, 0, 0, 100L, 100L, 100L, 0, 0, "sm", SHAPE_CURSOR},
3182};
3183
3184#ifdef FEAT_MOUSESHAPE
3185/*
3186 * Table with names for mouse shapes. Keep in sync with all the tables for
3187 * mch_set_mouse_shape()!.
3188 */
3189static char * mshape_names[] =
3190{
3191 "arrow", /* default, must be the first one */
3192 "blank", /* hidden */
3193 "beam",
3194 "updown",
3195 "udsizing",
3196 "leftright",
3197 "lrsizing",
3198 "busy",
3199 "no",
3200 "crosshair",
3201 "hand1",
3202 "hand2",
3203 "pencil",
3204 "question",
3205 "rightup-arrow",
3206 "up-arrow",
3207 NULL
3208};
3209#endif
3210
3211/*
3212 * Parse the 'guicursor' option ("what" is SHAPE_CURSOR) or 'mouseshape'
3213 * ("what" is SHAPE_MOUSE).
3214 * Returns error message for an illegal option, NULL otherwise.
3215 */
3216 char_u *
3217parse_shape_opt(what)
3218 int what;
3219{
3220 char_u *modep;
3221 char_u *colonp;
3222 char_u *commap;
3223 char_u *slashp;
3224 char_u *p, *endp;
3225 int idx = 0; /* init for GCC */
3226 int all_idx;
3227 int len;
3228 int i;
3229 long n;
3230 int found_ve = FALSE; /* found "ve" flag */
3231 int round;
3232
3233 /*
3234 * First round: check for errors; second round: do it for real.
3235 */
3236 for (round = 1; round <= 2; ++round)
3237 {
3238 /*
3239 * Repeat for all comma separated parts.
3240 */
3241#ifdef FEAT_MOUSESHAPE
3242 if (what == SHAPE_MOUSE)
3243 modep = p_mouseshape;
3244 else
3245#endif
3246 modep = p_guicursor;
3247 while (*modep != NUL)
3248 {
3249 colonp = vim_strchr(modep, ':');
3250 if (colonp == NULL)
3251 return (char_u *)N_("E545: Missing colon");
3252 if (colonp == modep)
3253 return (char_u *)N_("E546: Illegal mode");
3254 commap = vim_strchr(modep, ',');
3255
3256 /*
3257 * Repeat for all mode's before the colon.
3258 * For the 'a' mode, we loop to handle all the modes.
3259 */
3260 all_idx = -1;
3261 while (modep < colonp || all_idx >= 0)
3262 {
3263 if (all_idx < 0)
3264 {
3265 /* Find the mode. */
3266 if (modep[1] == '-' || modep[1] == ':')
3267 len = 1;
3268 else
3269 len = 2;
3270 if (len == 1 && TOLOWER_ASC(modep[0]) == 'a')
3271 all_idx = SHAPE_IDX_COUNT - 1;
3272 else
3273 {
3274 for (idx = 0; idx < SHAPE_IDX_COUNT; ++idx)
3275 if (STRNICMP(modep, shape_table[idx].name, len)
3276 == 0)
3277 break;
3278 if (idx == SHAPE_IDX_COUNT
3279 || (shape_table[idx].used_for & what) == 0)
3280 return (char_u *)N_("E546: Illegal mode");
3281 if (len == 2 && modep[0] == 'v' && modep[1] == 'e')
3282 found_ve = TRUE;
3283 }
3284 modep += len + 1;
3285 }
3286
3287 if (all_idx >= 0)
3288 idx = all_idx--;
3289 else if (round == 2)
3290 {
3291#ifdef FEAT_MOUSESHAPE
3292 if (what == SHAPE_MOUSE)
3293 {
3294 /* Set the default, for the missing parts */
3295 shape_table[idx].mshape = 0;
3296 }
3297 else
3298#endif
3299 {
3300 /* Set the defaults, for the missing parts */
3301 shape_table[idx].shape = SHAPE_BLOCK;
3302 shape_table[idx].blinkwait = 700L;
3303 shape_table[idx].blinkon = 400L;
3304 shape_table[idx].blinkoff = 250L;
3305 }
3306 }
3307
3308 /* Parse the part after the colon */
3309 for (p = colonp + 1; *p && *p != ','; )
3310 {
3311#ifdef FEAT_MOUSESHAPE
3312 if (what == SHAPE_MOUSE)
3313 {
3314 for (i = 0; ; ++i)
3315 {
3316 if (mshape_names[i] == NULL)
3317 {
3318 if (!VIM_ISDIGIT(*p))
3319 return (char_u *)N_("E547: Illegal mouseshape");
3320 if (round == 2)
3321 shape_table[idx].mshape =
3322 getdigits(&p) + MSHAPE_NUMBERED;
3323 else
3324 (void)getdigits(&p);
3325 break;
3326 }
3327 len = (int)STRLEN(mshape_names[i]);
3328 if (STRNICMP(p, mshape_names[i], len) == 0)
3329 {
3330 if (round == 2)
3331 shape_table[idx].mshape = i;
3332 p += len;
3333 break;
3334 }
3335 }
3336 }
3337 else /* if (what == SHAPE_MOUSE) */
3338#endif
3339 {
3340 /*
3341 * First handle the ones with a number argument.
3342 */
3343 i = *p;
3344 len = 0;
3345 if (STRNICMP(p, "ver", 3) == 0)
3346 len = 3;
3347 else if (STRNICMP(p, "hor", 3) == 0)
3348 len = 3;
3349 else if (STRNICMP(p, "blinkwait", 9) == 0)
3350 len = 9;
3351 else if (STRNICMP(p, "blinkon", 7) == 0)
3352 len = 7;
3353 else if (STRNICMP(p, "blinkoff", 8) == 0)
3354 len = 8;
3355 if (len != 0)
3356 {
3357 p += len;
3358 if (!VIM_ISDIGIT(*p))
3359 return (char_u *)N_("E548: digit expected");
3360 n = getdigits(&p);
3361 if (len == 3) /* "ver" or "hor" */
3362 {
3363 if (n == 0)
3364 return (char_u *)N_("E549: Illegal percentage");
3365 if (round == 2)
3366 {
3367 if (TOLOWER_ASC(i) == 'v')
3368 shape_table[idx].shape = SHAPE_VER;
3369 else
3370 shape_table[idx].shape = SHAPE_HOR;
3371 shape_table[idx].percentage = n;
3372 }
3373 }
3374 else if (round == 2)
3375 {
3376 if (len == 9)
3377 shape_table[idx].blinkwait = n;
3378 else if (len == 7)
3379 shape_table[idx].blinkon = n;
3380 else
3381 shape_table[idx].blinkoff = n;
3382 }
3383 }
3384 else if (STRNICMP(p, "block", 5) == 0)
3385 {
3386 if (round == 2)
3387 shape_table[idx].shape = SHAPE_BLOCK;
3388 p += 5;
3389 }
3390 else /* must be a highlight group name then */
3391 {
3392 endp = vim_strchr(p, '-');
3393 if (commap == NULL) /* last part */
3394 {
3395 if (endp == NULL)
3396 endp = p + STRLEN(p); /* find end of part */
3397 }
3398 else if (endp > commap || endp == NULL)
3399 endp = commap;
3400 slashp = vim_strchr(p, '/');
3401 if (slashp != NULL && slashp < endp)
3402 {
3403 /* "group/langmap_group" */
3404 i = syn_check_group(p, (int)(slashp - p));
3405 p = slashp + 1;
3406 }
3407 if (round == 2)
3408 {
3409 shape_table[idx].id = syn_check_group(p,
3410 (int)(endp - p));
3411 shape_table[idx].id_lm = shape_table[idx].id;
3412 if (slashp != NULL && slashp < endp)
3413 shape_table[idx].id = i;
3414 }
3415 p = endp;
3416 }
3417 } /* if (what != SHAPE_MOUSE) */
3418
3419 if (*p == '-')
3420 ++p;
3421 }
3422 }
3423 modep = p;
3424 if (*modep == ',')
3425 ++modep;
3426 }
3427 }
3428
3429 /* If the 's' flag is not given, use the 'v' cursor for 's' */
3430 if (!found_ve)
3431 {
3432#ifdef FEAT_MOUSESHAPE
3433 if (what == SHAPE_MOUSE)
3434 {
3435 shape_table[SHAPE_IDX_VE].mshape = shape_table[SHAPE_IDX_V].mshape;
3436 }
3437 else
3438#endif
3439 {
3440 shape_table[SHAPE_IDX_VE].shape = shape_table[SHAPE_IDX_V].shape;
3441 shape_table[SHAPE_IDX_VE].percentage =
3442 shape_table[SHAPE_IDX_V].percentage;
3443 shape_table[SHAPE_IDX_VE].blinkwait =
3444 shape_table[SHAPE_IDX_V].blinkwait;
3445 shape_table[SHAPE_IDX_VE].blinkon =
3446 shape_table[SHAPE_IDX_V].blinkon;
3447 shape_table[SHAPE_IDX_VE].blinkoff =
3448 shape_table[SHAPE_IDX_V].blinkoff;
3449 shape_table[SHAPE_IDX_VE].id = shape_table[SHAPE_IDX_V].id;
3450 shape_table[SHAPE_IDX_VE].id_lm = shape_table[SHAPE_IDX_V].id_lm;
3451 }
3452 }
3453
3454 return NULL;
3455}
3456
Bram Moolenaarac6e65f2005-08-29 22:25:38 +00003457# if defined(MCH_CURSOR_SHAPE) || defined(FEAT_GUI) \
3458 || defined(FEAT_MOUSESHAPE) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003459/*
3460 * Return the index into shape_table[] for the current mode.
3461 * When "mouse" is TRUE, consider indexes valid for the mouse pointer.
3462 */
3463 int
3464get_shape_idx(mouse)
3465 int mouse;
3466{
3467#ifdef FEAT_MOUSESHAPE
3468 if (mouse && (State == HITRETURN || State == ASKMORE))
3469 {
3470# ifdef FEAT_GUI
Bram Moolenaar9588a0f2005-01-08 21:45:39 +00003471 int x, y;
3472 gui_mch_getmouse(&x, &y);
3473 if (Y_2_ROW(y) == Rows - 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003474 return SHAPE_IDX_MOREL;
3475# endif
3476 return SHAPE_IDX_MORE;
3477 }
3478 if (mouse && drag_status_line)
3479 return SHAPE_IDX_SDRAG;
3480# ifdef FEAT_VERTSPLIT
3481 if (mouse && drag_sep_line)
3482 return SHAPE_IDX_VDRAG;
3483# endif
3484#endif
3485 if (!mouse && State == SHOWMATCH)
3486 return SHAPE_IDX_SM;
3487#ifdef FEAT_VREPLACE
3488 if (State & VREPLACE_FLAG)
3489 return SHAPE_IDX_R;
3490#endif
3491 if (State & REPLACE_FLAG)
3492 return SHAPE_IDX_R;
3493 if (State & INSERT)
3494 return SHAPE_IDX_I;
3495 if (State & CMDLINE)
3496 {
3497 if (cmdline_at_end())
3498 return SHAPE_IDX_C;
3499 if (cmdline_overstrike())
3500 return SHAPE_IDX_CR;
3501 return SHAPE_IDX_CI;
3502 }
3503 if (finish_op)
3504 return SHAPE_IDX_O;
3505#ifdef FEAT_VISUAL
3506 if (VIsual_active)
3507 {
3508 if (*p_sel == 'e')
3509 return SHAPE_IDX_VE;
3510 else
3511 return SHAPE_IDX_V;
3512 }
3513#endif
3514 return SHAPE_IDX_N;
3515}
Bram Moolenaarac6e65f2005-08-29 22:25:38 +00003516#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003517
3518# if defined(FEAT_MOUSESHAPE) || defined(PROTO)
3519static int old_mouse_shape = 0;
3520
3521/*
3522 * Set the mouse shape:
3523 * If "shape" is -1, use shape depending on the current mode,
3524 * depending on the current state.
3525 * If "shape" is -2, only update the shape when it's CLINE or STATUS (used
3526 * when the mouse moves off the status or command line).
3527 */
3528 void
3529update_mouseshape(shape_idx)
3530 int shape_idx;
3531{
3532 int new_mouse_shape;
3533
3534 /* Only works in GUI mode. */
Bram Moolenaar6bb68362005-03-22 23:03:44 +00003535 if (!gui.in_use || gui.starting)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003536 return;
3537
3538 /* Postpone the updating when more is to come. Speeds up executing of
3539 * mappings. */
3540 if (shape_idx == -1 && char_avail())
3541 {
3542 postponed_mouseshape = TRUE;
3543 return;
3544 }
3545
Bram Moolenaar14716812006-05-04 21:54:08 +00003546 /* When ignoring the mouse don't change shape on the statusline. */
3547 if (*p_mouse == NUL
3548 && (shape_idx == SHAPE_IDX_CLINE
3549 || shape_idx == SHAPE_IDX_STATUS
3550 || shape_idx == SHAPE_IDX_VSEP))
3551 shape_idx = -2;
3552
Bram Moolenaar071d4272004-06-13 20:20:40 +00003553 if (shape_idx == -2
3554 && old_mouse_shape != shape_table[SHAPE_IDX_CLINE].mshape
3555 && old_mouse_shape != shape_table[SHAPE_IDX_STATUS].mshape
3556 && old_mouse_shape != shape_table[SHAPE_IDX_VSEP].mshape)
3557 return;
3558 if (shape_idx < 0)
3559 new_mouse_shape = shape_table[get_shape_idx(TRUE)].mshape;
3560 else
3561 new_mouse_shape = shape_table[shape_idx].mshape;
3562 if (new_mouse_shape != old_mouse_shape)
3563 {
3564 mch_set_mouse_shape(new_mouse_shape);
3565 old_mouse_shape = new_mouse_shape;
3566 }
3567 postponed_mouseshape = FALSE;
3568}
3569# endif
3570
3571#endif /* CURSOR_SHAPE */
3572
3573
3574#ifdef FEAT_CRYPT
3575/*
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00003576 * Optional encryption support.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003577 * Mohsin Ahmed, mosh@sasi.com, 98-09-24
3578 * Based on zip/crypt sources.
3579 *
3580 * NOTE FOR USA: Since 2000 exporting this code from the USA is allowed to
3581 * most countries. There are a few exceptions, but that still should not be a
3582 * problem since this code was originally created in Europe and India.
3583 */
3584
3585/* from zip.h */
3586
3587typedef unsigned short ush; /* unsigned 16-bit value */
3588typedef unsigned long ulg; /* unsigned 32-bit value */
3589
3590static void make_crc_tab __ARGS((void));
3591
Bram Moolenaard6f676d2005-06-01 21:51:55 +00003592static ulg crc_32_tab[256];
Bram Moolenaar071d4272004-06-13 20:20:40 +00003593
3594/*
3595 * Fill the CRC table.
3596 */
3597 static void
3598make_crc_tab()
3599{
3600 ulg s,t,v;
3601 static int done = FALSE;
3602
3603 if (done)
3604 return;
3605 for (t = 0; t < 256; t++)
3606 {
3607 v = t;
3608 for (s = 0; s < 8; s++)
3609 v = (v >> 1) ^ ((v & 1) * (ulg)0xedb88320L);
3610 crc_32_tab[t] = v;
3611 }
3612 done = TRUE;
3613}
3614
3615#define CRC32(c, b) (crc_32_tab[((int)(c) ^ (b)) & 0xff] ^ ((c) >> 8))
3616
3617
3618static ulg keys[3]; /* keys defining the pseudo-random sequence */
3619
3620/*
3621 * Return the next byte in the pseudo-random sequence
3622 */
3623 int
3624decrypt_byte()
3625{
3626 ush temp;
3627
3628 temp = (ush)keys[2] | 2;
3629 return (int)(((unsigned)(temp * (temp ^ 1)) >> 8) & 0xff);
3630}
3631
3632/*
3633 * Update the encryption keys with the next byte of plain text
3634 */
3635 int
3636update_keys(c)
3637 int c; /* byte of plain text */
3638{
3639 keys[0] = CRC32(keys[0], c);
3640 keys[1] += keys[0] & 0xff;
3641 keys[1] = keys[1] * 134775813L + 1;
3642 keys[2] = CRC32(keys[2], (int)(keys[1] >> 24));
3643 return c;
3644}
3645
3646/*
3647 * Initialize the encryption keys and the random header according to
3648 * the given password.
3649 * If "passwd" is NULL or empty, don't do anything.
3650 */
3651 void
3652crypt_init_keys(passwd)
3653 char_u *passwd; /* password string with which to modify keys */
3654{
3655 if (passwd != NULL && *passwd != NUL)
3656 {
3657 make_crc_tab();
3658 keys[0] = 305419896L;
3659 keys[1] = 591751049L;
3660 keys[2] = 878082192L;
3661 while (*passwd != '\0')
3662 update_keys((int)*passwd++);
3663 }
3664}
3665
3666/*
3667 * Ask the user for a crypt key.
3668 * When "store" is TRUE, the new key in stored in the 'key' option, and the
3669 * 'key' option value is returned: Don't free it.
3670 * When "store" is FALSE, the typed key is returned in allocated memory.
3671 * Returns NULL on failure.
3672 */
3673 char_u *
3674get_crypt_key(store, twice)
3675 int store;
3676 int twice; /* Ask for the key twice. */
3677{
3678 char_u *p1, *p2 = NULL;
3679 int round;
3680
3681 for (round = 0; ; ++round)
3682 {
3683 cmdline_star = TRUE;
3684 cmdline_row = msg_row;
3685 p1 = getcmdline_prompt(NUL, round == 0
3686 ? (char_u *)_("Enter encryption key: ")
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +00003687 : (char_u *)_("Enter same key again: "), 0, EXPAND_NOTHING,
3688 NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003689 cmdline_star = FALSE;
3690
3691 if (p1 == NULL)
3692 break;
3693
3694 if (round == twice)
3695 {
3696 if (p2 != NULL && STRCMP(p1, p2) != 0)
3697 {
3698 MSG(_("Keys don't match!"));
3699 vim_free(p1);
3700 vim_free(p2);
3701 p2 = NULL;
3702 round = -1; /* do it again */
3703 continue;
3704 }
3705 if (store)
3706 {
3707 set_option_value((char_u *)"key", 0L, p1, OPT_LOCAL);
3708 vim_free(p1);
3709 p1 = curbuf->b_p_key;
3710 }
3711 break;
3712 }
3713 p2 = p1;
3714 }
3715
3716 /* since the user typed this, no need to wait for return */
3717 need_wait_return = FALSE;
3718 msg_didout = FALSE;
3719
3720 vim_free(p2);
3721 return p1;
3722}
3723
3724#endif /* FEAT_CRYPT */
3725
3726/* TODO: make some #ifdef for this */
3727/*--------[ file searching ]-------------------------------------------------*/
3728/*
3729 * File searching functions for 'path', 'tags' and 'cdpath' options.
3730 * External visible functions:
3731 * vim_findfile_init() creates/initialises the search context
3732 * vim_findfile_free_visited() free list of visited files/dirs of search
3733 * context
3734 * vim_findfile() find a file in the search context
3735 * vim_findfile_cleanup() cleanup/free search context created by
3736 * vim_findfile_init()
3737 *
3738 * All static functions and variables start with 'ff_'
3739 *
3740 * In general it works like this:
3741 * First you create yourself a search context by calling vim_findfile_init().
3742 * It is possible to give a search context from a previous call to
3743 * vim_findfile_init(), so it can be reused. After this you call vim_findfile()
3744 * until you are satisfied with the result or it returns NULL. On every call it
3745 * returns the next file which matches the conditions given to
3746 * vim_findfile_init(). If it doesn't find a next file it returns NULL.
3747 *
3748 * It is possible to call vim_findfile_init() again to reinitialise your search
3749 * with some new parameters. Don't forget to pass your old search context to
3750 * it, so it can reuse it and especially reuse the list of already visited
3751 * directories. If you want to delete the list of already visited directories
3752 * simply call vim_findfile_free_visited().
3753 *
3754 * When you are done call vim_findfile_cleanup() to free the search context.
3755 *
3756 * The function vim_findfile_init() has a long comment, which describes the
3757 * needed parameters.
3758 *
3759 *
3760 *
3761 * ATTENTION:
3762 * ==========
Bram Moolenaar77f66d62007-02-20 02:16:18 +00003763 * Also we use an allocated search context here, this functions are NOT
Bram Moolenaar071d4272004-06-13 20:20:40 +00003764 * thread-safe!!!!!
3765 *
3766 * To minimize parameter passing (or because I'm to lazy), only the
3767 * external visible functions get a search context as a parameter. This is
3768 * then assigned to a static global, which is used throughout the local
3769 * functions.
3770 */
3771
3772/*
3773 * type for the directory search stack
3774 */
3775typedef struct ff_stack
3776{
3777 struct ff_stack *ffs_prev;
3778
3779 /* the fix part (no wildcards) and the part containing the wildcards
3780 * of the search path
3781 */
3782 char_u *ffs_fix_path;
3783#ifdef FEAT_PATH_EXTRA
3784 char_u *ffs_wc_path;
3785#endif
3786
3787 /* files/dirs found in the above directory, matched by the first wildcard
3788 * of wc_part
3789 */
3790 char_u **ffs_filearray;
3791 int ffs_filearray_size;
3792 char_u ffs_filearray_cur; /* needed for partly handled dirs */
3793
3794 /* to store status of partly handled directories
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00003795 * 0: we work on this directory for the first time
Bram Moolenaar071d4272004-06-13 20:20:40 +00003796 * 1: this directory was partly searched in an earlier step
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00003797 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003798 int ffs_stage;
3799
3800 /* How deep are we in the directory tree?
3801 * Counts backward from value of level parameter to vim_findfile_init
3802 */
3803 int ffs_level;
3804
3805 /* Did we already expand '**' to an empty string? */
3806 int ffs_star_star_empty;
3807} ff_stack_T;
3808
3809/*
3810 * type for already visited directories or files.
3811 */
3812typedef struct ff_visited
3813{
3814 struct ff_visited *ffv_next;
3815
3816#ifdef FEAT_PATH_EXTRA
3817 /* Visited directories are different if the wildcard string are
3818 * different. So we have to save it.
3819 */
3820 char_u *ffv_wc_path;
3821#endif
3822 /* for unix use inode etc for comparison (needed because of links), else
3823 * use filename.
3824 */
3825#ifdef UNIX
3826 int ffv_dev; /* device number (-1 if not set) */
3827 ino_t ffv_ino; /* inode number */
3828#endif
3829 /* The memory for this struct is allocated according to the length of
3830 * ffv_fname.
3831 */
3832 char_u ffv_fname[1]; /* actually longer */
3833} ff_visited_T;
3834
3835/*
3836 * We might have to manage several visited lists during a search.
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00003837 * This is especially needed for the tags option. If tags is set to:
Bram Moolenaar071d4272004-06-13 20:20:40 +00003838 * "./++/tags,./++/TAGS,++/tags" (replace + with *)
3839 * So we have to do 3 searches:
3840 * 1) search from the current files directory downward for the file "tags"
3841 * 2) search from the current files directory downward for the file "TAGS"
3842 * 3) search from Vims current directory downwards for the file "tags"
3843 * As you can see, the first and the third search are for the same file, so for
3844 * the third search we can use the visited list of the first search. For the
3845 * second search we must start from a empty visited list.
3846 * The struct ff_visited_list_hdr is used to manage a linked list of already
3847 * visited lists.
3848 */
3849typedef struct ff_visited_list_hdr
3850{
3851 struct ff_visited_list_hdr *ffvl_next;
3852
3853 /* the filename the attached visited list is for */
3854 char_u *ffvl_filename;
3855
3856 ff_visited_T *ffvl_visited_list;
3857
3858} ff_visited_list_hdr_T;
3859
3860
3861/*
3862 * '**' can be expanded to several directory levels.
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00003863 * Set the default maximum depth.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003864 */
3865#define FF_MAX_STAR_STAR_EXPAND ((char_u)30)
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00003866
Bram Moolenaar071d4272004-06-13 20:20:40 +00003867/*
3868 * The search context:
3869 * ffsc_stack_ptr: the stack for the dirs to search
3870 * ffsc_visited_list: the currently active visited list
3871 * ffsc_dir_visited_list: the currently active visited list for search dirs
3872 * ffsc_visited_lists_list: the list of all visited lists
3873 * ffsc_dir_visited_lists_list: the list of all visited lists for search dirs
3874 * ffsc_file_to_search: the file to search for
3875 * ffsc_start_dir: the starting directory, if search path was relative
3876 * ffsc_fix_path: the fix part of the given path (without wildcards)
3877 * Needed for upward search.
3878 * ffsc_wc_path: the part of the given path containing wildcards
3879 * ffsc_level: how many levels of dirs to search downwards
3880 * ffsc_stopdirs_v: array of stop directories for upward search
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00003881 * ffsc_find_what: FINDFILE_BOTH, FINDFILE_DIR or FINDFILE_FILE
Bram Moolenaar071d4272004-06-13 20:20:40 +00003882 */
3883typedef struct ff_search_ctx_T
3884{
3885 ff_stack_T *ffsc_stack_ptr;
3886 ff_visited_list_hdr_T *ffsc_visited_list;
3887 ff_visited_list_hdr_T *ffsc_dir_visited_list;
3888 ff_visited_list_hdr_T *ffsc_visited_lists_list;
3889 ff_visited_list_hdr_T *ffsc_dir_visited_lists_list;
3890 char_u *ffsc_file_to_search;
3891 char_u *ffsc_start_dir;
3892 char_u *ffsc_fix_path;
3893#ifdef FEAT_PATH_EXTRA
3894 char_u *ffsc_wc_path;
3895 int ffsc_level;
3896 char_u **ffsc_stopdirs_v;
3897#endif
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00003898 int ffsc_find_what;
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00003899} ff_search_ctx_T;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003900
Bram Moolenaar071d4272004-06-13 20:20:40 +00003901/* locally needed functions */
3902#ifdef FEAT_PATH_EXTRA
3903static int ff_check_visited __ARGS((ff_visited_T **, char_u *, char_u *));
3904#else
3905static int ff_check_visited __ARGS((ff_visited_T **, char_u *));
3906#endif
3907static void vim_findfile_free_visited_list __ARGS((ff_visited_list_hdr_T **list_headp));
3908static void ff_free_visited_list __ARGS((ff_visited_T *vl));
3909static ff_visited_list_hdr_T* ff_get_visited_list __ARGS((char_u *, ff_visited_list_hdr_T **list_headp));
3910#ifdef FEAT_PATH_EXTRA
3911static int ff_wc_equal __ARGS((char_u *s1, char_u *s2));
3912#endif
3913
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00003914static void ff_push __ARGS((ff_search_ctx_T *search_ctx, ff_stack_T *stack_ptr));
3915static ff_stack_T *ff_pop __ARGS((ff_search_ctx_T *search_ctx));
3916static void ff_clear __ARGS((ff_search_ctx_T *search_ctx));
3917static void ff_free_stack_element __ARGS((ff_stack_T *stack_ptr));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003918#ifdef FEAT_PATH_EXTRA
3919static ff_stack_T *ff_create_stack_element __ARGS((char_u *, char_u *, int, int));
3920#else
3921static ff_stack_T *ff_create_stack_element __ARGS((char_u *, int, int));
3922#endif
3923#ifdef FEAT_PATH_EXTRA
3924static int ff_path_in_stoplist __ARGS((char_u *, int, char_u **));
3925#endif
3926
Bram Moolenaar071d4272004-06-13 20:20:40 +00003927#if 0
3928/*
3929 * if someone likes findfirst/findnext, here are the functions
3930 * NOT TESTED!!
3931 */
3932
3933static void *ff_fn_search_context = NULL;
3934
3935 char_u *
3936vim_findfirst(path, filename, level)
3937 char_u *path;
3938 char_u *filename;
3939 int level;
3940{
3941 ff_fn_search_context =
3942 vim_findfile_init(path, filename, NULL, level, TRUE, FALSE,
3943 ff_fn_search_context, rel_fname);
3944 if (NULL == ff_fn_search_context)
3945 return NULL;
3946 else
3947 return vim_findnext()
3948}
3949
3950 char_u *
3951vim_findnext()
3952{
3953 char_u *ret = vim_findfile(ff_fn_search_context);
3954
3955 if (NULL == ret)
3956 {
3957 vim_findfile_cleanup(ff_fn_search_context);
3958 ff_fn_search_context = NULL;
3959 }
3960 return ret;
3961}
3962#endif
3963
3964/*
3965 * Initialization routine for vim_findfile.
3966 *
3967 * Returns the newly allocated search context or NULL if an error occured.
3968 *
3969 * Don't forget to clean up by calling vim_findfile_cleanup() if you are done
3970 * with the search context.
3971 *
3972 * Find the file 'filename' in the directory 'path'.
3973 * The parameter 'path' may contain wildcards. If so only search 'level'
3974 * directories deep. The parameter 'level' is the absolute maximum and is
3975 * not related to restricts given to the '**' wildcard. If 'level' is 100
3976 * and you use '**200' vim_findfile() will stop after 100 levels.
3977 *
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00003978 * 'filename' cannot contain wildcards! It is used as-is, no backslashes to
3979 * escape special characters.
3980 *
Bram Moolenaar071d4272004-06-13 20:20:40 +00003981 * If 'stopdirs' is not NULL and nothing is found downward, the search is
3982 * restarted on the next higher directory level. This is repeated until the
3983 * start-directory of a search is contained in 'stopdirs'. 'stopdirs' has the
3984 * format ";*<dirname>*\(;<dirname>\)*;\=$".
3985 *
3986 * If the 'path' is relative, the starting dir for the search is either VIM's
3987 * current dir or if the path starts with "./" the current files dir.
3988 * If the 'path' is absolut, the starting dir is that part of the path before
3989 * the first wildcard.
3990 *
3991 * Upward search is only done on the starting dir.
3992 *
3993 * If 'free_visited' is TRUE the list of already visited files/directories is
3994 * cleared. Set this to FALSE if you just want to search from another
3995 * directory, but want to be sure that no directory from a previous search is
3996 * searched again. This is useful if you search for a file at different places.
3997 * The list of visited files/dirs can also be cleared with the function
3998 * vim_findfile_free_visited().
3999 *
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004000 * Set the parameter 'find_what' to FINDFILE_DIR if you want to search for
4001 * directories only, FINDFILE_FILE for files only, FINDFILE_BOTH for both.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004002 *
4003 * A search context returned by a previous call to vim_findfile_init() can be
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004004 * passed in the parameter "search_ctx_arg". This context is reused and
4005 * reinitialized with the new parameters. The list of already visited
Bram Moolenaar071d4272004-06-13 20:20:40 +00004006 * directories from this context is only deleted if the parameter
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004007 * "free_visited" is true. Be aware that the passed "search_ctx_arg" is freed
4008 * if the reinitialization fails.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004009 *
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004010 * If you don't have a search context from a previous call "search_ctx_arg"
4011 * must be NULL.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004012 *
4013 * This function silently ignores a few errors, vim_findfile() will have
4014 * limited functionality then.
4015 */
4016/*ARGSUSED*/
4017 void *
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004018vim_findfile_init(path, filename, stopdirs, level, free_visited, find_what,
4019 search_ctx_arg, tagfile, rel_fname)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004020 char_u *path;
4021 char_u *filename;
4022 char_u *stopdirs;
4023 int level;
4024 int free_visited;
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004025 int find_what;
4026 void *search_ctx_arg;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004027 int tagfile;
4028 char_u *rel_fname; /* file name to use for "." */
4029{
4030#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004031 char_u *wc_part;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004032#endif
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004033 ff_stack_T *sptr;
4034 ff_search_ctx_T *search_ctx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004035
4036 /* If a search context is given by the caller, reuse it, else allocate a
4037 * new one.
4038 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004039 if (search_ctx_arg != NULL)
4040 search_ctx = search_ctx_arg;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004041 else
4042 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004043 search_ctx = (ff_search_ctx_T*)alloc((unsigned)sizeof(ff_search_ctx_T));
4044 if (search_ctx == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004045 goto error_return;
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004046 memset(search_ctx, 0, sizeof(ff_search_ctx_T));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004047 }
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004048 search_ctx->ffsc_find_what = find_what;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004049
4050 /* clear the search context, but NOT the visited lists */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004051 ff_clear(search_ctx);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004052
4053 /* clear visited list if wanted */
4054 if (free_visited == TRUE)
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004055 vim_findfile_free_visited(search_ctx);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004056 else
4057 {
4058 /* Reuse old visited lists. Get the visited list for the given
4059 * filename. If no list for the current filename exists, creates a new
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004060 * one. */
4061 search_ctx->ffsc_visited_list = ff_get_visited_list(filename,
4062 &search_ctx->ffsc_visited_lists_list);
4063 if (search_ctx->ffsc_visited_list == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004064 goto error_return;
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004065 search_ctx->ffsc_dir_visited_list = ff_get_visited_list(filename,
4066 &search_ctx->ffsc_dir_visited_lists_list);
4067 if (search_ctx->ffsc_dir_visited_list == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004068 goto error_return;
4069 }
4070
4071 if (ff_expand_buffer == NULL)
4072 {
4073 ff_expand_buffer = (char_u*)alloc(MAXPATHL);
4074 if (ff_expand_buffer == NULL)
4075 goto error_return;
4076 }
4077
4078 /* Store information on starting dir now if path is relative.
Bram Moolenaar5eb86f92004-07-26 12:53:41 +00004079 * If path is absolute, we do that later. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004080 if (path[0] == '.'
4081 && (vim_ispathsep(path[1]) || path[1] == NUL)
4082 && (!tagfile || vim_strchr(p_cpo, CPO_DOTTAG) == NULL)
4083 && rel_fname != NULL)
4084 {
4085 int len = (int)(gettail(rel_fname) - rel_fname);
4086
4087 if (!vim_isAbsName(rel_fname) && len + 1 < MAXPATHL)
4088 {
4089 /* Make the start dir an absolute path name. */
Bram Moolenaarbbebc852005-07-18 21:47:53 +00004090 vim_strncpy(ff_expand_buffer, rel_fname, len);
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004091 search_ctx->ffsc_start_dir = FullName_save(ff_expand_buffer, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004092 }
4093 else
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004094 search_ctx->ffsc_start_dir = vim_strnsave(rel_fname, len);
4095 if (search_ctx->ffsc_start_dir == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004096 goto error_return;
4097 if (*++path != NUL)
4098 ++path;
4099 }
4100 else if (*path == NUL || !vim_isAbsName(path))
4101 {
4102#ifdef BACKSLASH_IN_FILENAME
4103 /* "c:dir" needs "c:" to be expanded, otherwise use current dir */
4104 if (*path != NUL && path[1] == ':')
4105 {
4106 char_u drive[3];
4107
4108 drive[0] = path[0];
4109 drive[1] = ':';
4110 drive[2] = NUL;
4111 if (vim_FullName(drive, ff_expand_buffer, MAXPATHL, TRUE) == FAIL)
4112 goto error_return;
4113 path += 2;
4114 }
4115 else
4116#endif
4117 if (mch_dirname(ff_expand_buffer, MAXPATHL) == FAIL)
4118 goto error_return;
4119
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004120 search_ctx->ffsc_start_dir = vim_strsave(ff_expand_buffer);
4121 if (search_ctx->ffsc_start_dir == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004122 goto error_return;
4123
4124#ifdef BACKSLASH_IN_FILENAME
4125 /* A path that starts with "/dir" is relative to the drive, not to the
4126 * directory (but not for "//machine/dir"). Only use the drive name. */
4127 if ((*path == '/' || *path == '\\')
4128 && path[1] != path[0]
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004129 && search_ctx->ffsc_start_dir[1] == ':')
4130 search_ctx->ffsc_start_dir[2] = NUL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004131#endif
4132 }
4133
4134#ifdef FEAT_PATH_EXTRA
4135 /*
4136 * If stopdirs are given, split them into an array of pointers.
4137 * If this fails (mem allocation), there is no upward search at all or a
4138 * stop directory is not recognized -> continue silently.
4139 * If stopdirs just contains a ";" or is empty,
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004140 * search_ctx->ffsc_stopdirs_v will only contain a NULL pointer. This
Bram Moolenaar071d4272004-06-13 20:20:40 +00004141 * is handled as unlimited upward search. See function
4142 * ff_path_in_stoplist() for details.
4143 */
4144 if (stopdirs != NULL)
4145 {
4146 char_u *walker = stopdirs;
4147 int dircount;
4148
4149 while (*walker == ';')
4150 walker++;
4151
4152 dircount = 1;
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004153 search_ctx->ffsc_stopdirs_v =
4154 (char_u **)alloc((unsigned)sizeof(char_u *));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004155
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004156 if (search_ctx->ffsc_stopdirs_v != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004157 {
4158 do
4159 {
4160 char_u *helper;
4161 void *ptr;
4162
4163 helper = walker;
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004164 ptr = vim_realloc(search_ctx->ffsc_stopdirs_v,
Bram Moolenaar071d4272004-06-13 20:20:40 +00004165 (dircount + 1) * sizeof(char_u *));
4166 if (ptr)
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004167 search_ctx->ffsc_stopdirs_v = ptr;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004168 else
4169 /* ignore, keep what we have and continue */
4170 break;
4171 walker = vim_strchr(walker, ';');
4172 if (walker)
4173 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004174 search_ctx->ffsc_stopdirs_v[dircount-1] =
4175 vim_strnsave(helper, (int)(walker - helper));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004176 walker++;
4177 }
4178 else
4179 /* this might be "", which means ascent till top
4180 * of directory tree.
4181 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004182 search_ctx->ffsc_stopdirs_v[dircount-1] =
4183 vim_strsave(helper);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004184
4185 dircount++;
4186
4187 } while (walker != NULL);
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004188 search_ctx->ffsc_stopdirs_v[dircount-1] = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004189 }
4190 }
4191#endif
4192
4193#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004194 search_ctx->ffsc_level = level;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004195
4196 /* split into:
4197 * -fix path
4198 * -wildcard_stuff (might be NULL)
4199 */
4200 wc_part = vim_strchr(path, '*');
4201 if (wc_part != NULL)
4202 {
4203 int llevel;
4204 int len;
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00004205 char *errpt;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004206
4207 /* save the fix part of the path */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004208 search_ctx->ffsc_fix_path = vim_strnsave(path, (int)(wc_part - path));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004209
4210 /*
4211 * copy wc_path and add restricts to the '**' wildcard.
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00004212 * The octet after a '**' is used as a (binary) counter.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004213 * So '**3' is transposed to '**^C' ('^C' is ASCII value 3)
4214 * or '**76' is transposed to '**N'( 'N' is ASCII value 76).
4215 * For EBCDIC you get different character values.
4216 * If no restrict is given after '**' the default is used.
4217 * Due to this technic the path looks awful if you print it as a
4218 * string.
4219 */
4220 len = 0;
4221 while (*wc_part != NUL)
4222 {
4223 if (STRNCMP(wc_part, "**", 2) == 0)
4224 {
4225 ff_expand_buffer[len++] = *wc_part++;
4226 ff_expand_buffer[len++] = *wc_part++;
4227
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00004228 llevel = strtol((char *)wc_part, &errpt, 10);
4229 if ((char_u *)errpt != wc_part && llevel > 0 && llevel < 255)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004230 ff_expand_buffer[len++] = llevel;
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00004231 else if ((char_u *)errpt != wc_part && llevel == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004232 /* restrict is 0 -> remove already added '**' */
4233 len -= 2;
4234 else
4235 ff_expand_buffer[len++] = FF_MAX_STAR_STAR_EXPAND;
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00004236 wc_part = (char_u *)errpt;
Bram Moolenaar1d94f9b2005-08-04 21:29:45 +00004237 if (*wc_part != NUL && !vim_ispathsep(*wc_part))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004238 {
4239 EMSG2(_("E343: Invalid path: '**[number]' must be at the end of the path or be followed by '%s'."), PATHSEPSTR);
4240 goto error_return;
4241 }
4242 }
4243 else
4244 ff_expand_buffer[len++] = *wc_part++;
4245 }
4246 ff_expand_buffer[len] = NUL;
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004247 search_ctx->ffsc_wc_path = vim_strsave(ff_expand_buffer);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004248
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004249 if (search_ctx->ffsc_wc_path == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004250 goto error_return;
4251 }
4252 else
4253#endif
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004254 search_ctx->ffsc_fix_path = vim_strsave(path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004255
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004256 if (search_ctx->ffsc_start_dir == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004257 {
4258 /* store the fix part as startdir.
4259 * This is needed if the parameter path is fully qualified.
4260 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004261 search_ctx->ffsc_start_dir = vim_strsave(search_ctx->ffsc_fix_path);
4262 if (search_ctx->ffsc_start_dir)
4263 search_ctx->ffsc_fix_path[0] = NUL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004264 }
4265
4266 /* create an absolute path */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004267 STRCPY(ff_expand_buffer, search_ctx->ffsc_start_dir);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004268 add_pathsep(ff_expand_buffer);
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004269 STRCAT(ff_expand_buffer, search_ctx->ffsc_fix_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004270 add_pathsep(ff_expand_buffer);
4271
4272 sptr = ff_create_stack_element(ff_expand_buffer,
4273#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004274 search_ctx->ffsc_wc_path,
Bram Moolenaar071d4272004-06-13 20:20:40 +00004275#endif
4276 level, 0);
4277
4278 if (sptr == NULL)
4279 goto error_return;
4280
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004281 ff_push(search_ctx, sptr);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004282
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004283 search_ctx->ffsc_file_to_search = vim_strsave(filename);
4284 if (search_ctx->ffsc_file_to_search == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004285 goto error_return;
4286
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004287 return search_ctx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004288
4289error_return:
4290 /*
4291 * We clear the search context now!
4292 * Even when the caller gave us a (perhaps valid) context we free it here,
4293 * as we might have already destroyed it.
4294 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004295 vim_findfile_cleanup(search_ctx);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004296 return NULL;
4297}
4298
4299#if defined(FEAT_PATH_EXTRA) || defined(PROTO)
4300/*
4301 * Get the stopdir string. Check that ';' is not escaped.
4302 */
4303 char_u *
4304vim_findfile_stopdir(buf)
4305 char_u *buf;
4306{
4307 char_u *r_ptr = buf;
4308
4309 while (*r_ptr != NUL && *r_ptr != ';')
4310 {
4311 if (r_ptr[0] == '\\' && r_ptr[1] == ';')
4312 {
4313 /* overwrite the escape char,
4314 * use STRLEN(r_ptr) to move the trailing '\0'
4315 */
Bram Moolenaar446cb832008-06-24 21:56:24 +00004316 STRMOVE(r_ptr, r_ptr + 1);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004317 r_ptr++;
4318 }
4319 r_ptr++;
4320 }
4321 if (*r_ptr == ';')
4322 {
4323 *r_ptr = 0;
4324 r_ptr++;
4325 }
4326 else if (*r_ptr == NUL)
4327 r_ptr = NULL;
4328 return r_ptr;
4329}
4330#endif
4331
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004332/*
4333 * Clean up the given search context. Can handle a NULL pointer.
4334 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004335 void
4336vim_findfile_cleanup(ctx)
4337 void *ctx;
4338{
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00004339 if (ctx == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004340 return;
4341
Bram Moolenaar071d4272004-06-13 20:20:40 +00004342 vim_findfile_free_visited(ctx);
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004343 ff_clear(ctx);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004344 vim_free(ctx);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004345}
4346
4347/*
4348 * Find a file in a search context.
4349 * The search context was created with vim_findfile_init() above.
4350 * Return a pointer to an allocated file name or NULL if nothing found.
4351 * To get all matching files call this function until you get NULL.
4352 *
Bram Moolenaar5eb86f92004-07-26 12:53:41 +00004353 * If the passed search_context is NULL, NULL is returned.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004354 *
4355 * The search algorithm is depth first. To change this replace the
4356 * stack with a list (don't forget to leave partly searched directories on the
4357 * top of the list).
4358 */
4359 char_u *
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004360vim_findfile(search_ctx_arg)
4361 void *search_ctx_arg;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004362{
4363 char_u *file_path;
4364#ifdef FEAT_PATH_EXTRA
4365 char_u *rest_of_wildcards;
4366 char_u *path_end = NULL;
4367#endif
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004368 ff_stack_T *stackp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004369#if defined(FEAT_SEARCHPATH) || defined(FEAT_PATH_EXTRA)
4370 int len;
4371#endif
4372 int i;
4373 char_u *p;
4374#ifdef FEAT_SEARCHPATH
4375 char_u *suf;
4376#endif
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004377 ff_search_ctx_T *search_ctx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004378
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004379 if (search_ctx_arg == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004380 return NULL;
4381
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004382 search_ctx = (ff_search_ctx_T *)search_ctx_arg;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004383
4384 /*
4385 * filepath is used as buffer for various actions and as the storage to
4386 * return a found filename.
4387 */
4388 if ((file_path = alloc((int)MAXPATHL)) == NULL)
4389 return NULL;
4390
4391#ifdef FEAT_PATH_EXTRA
4392 /* store the end of the start dir -- needed for upward search */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004393 if (search_ctx->ffsc_start_dir != NULL)
4394 path_end = &search_ctx->ffsc_start_dir[
4395 STRLEN(search_ctx->ffsc_start_dir)];
Bram Moolenaar071d4272004-06-13 20:20:40 +00004396#endif
4397
4398#ifdef FEAT_PATH_EXTRA
4399 /* upward search loop */
4400 for (;;)
4401 {
4402#endif
4403 /* downward search loop */
4404 for (;;)
4405 {
4406 /* check if user user wants to stop the search*/
4407 ui_breakcheck();
4408 if (got_int)
4409 break;
4410
4411 /* get directory to work on from stack */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004412 stackp = ff_pop(search_ctx);
4413 if (stackp == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004414 break;
4415
4416 /*
4417 * TODO: decide if we leave this test in
4418 *
4419 * GOOD: don't search a directory(-tree) twice.
4420 * BAD: - check linked list for every new directory entered.
4421 * - check for double files also done below
4422 *
4423 * Here we check if we already searched this directory.
4424 * We already searched a directory if:
4425 * 1) The directory is the same.
4426 * 2) We would use the same wildcard string.
4427 *
4428 * Good if you have links on same directory via several ways
4429 * or you have selfreferences in directories (e.g. SuSE Linux 6.3:
4430 * /etc/rc.d/init.d is linked to /etc/rc.d -> endless loop)
4431 *
4432 * This check is only needed for directories we work on for the
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004433 * first time (hence stackp->ff_filearray == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004434 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004435 if (stackp->ffs_filearray == NULL
4436 && ff_check_visited(&search_ctx->ffsc_dir_visited_list
Bram Moolenaar071d4272004-06-13 20:20:40 +00004437 ->ffvl_visited_list,
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004438 stackp->ffs_fix_path
Bram Moolenaar071d4272004-06-13 20:20:40 +00004439#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004440 , stackp->ffs_wc_path
Bram Moolenaar071d4272004-06-13 20:20:40 +00004441#endif
4442 ) == FAIL)
4443 {
4444#ifdef FF_VERBOSE
4445 if (p_verbose >= 5)
4446 {
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004447 verbose_enter_scroll();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004448 smsg((char_u *)"Already Searched: %s (%s)",
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004449 stackp->ffs_fix_path, stackp->ffs_wc_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004450 /* don't overwrite this either */
4451 msg_puts((char_u *)"\n");
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004452 verbose_leave_scroll();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004453 }
4454#endif
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004455 ff_free_stack_element(stackp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004456 continue;
4457 }
4458#ifdef FF_VERBOSE
4459 else if (p_verbose >= 5)
4460 {
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004461 verbose_enter_scroll();
Bram Moolenaar051b7822005-05-19 21:00:46 +00004462 smsg((char_u *)"Searching: %s (%s)",
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004463 stackp->ffs_fix_path, stackp->ffs_wc_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004464 /* don't overwrite this either */
4465 msg_puts((char_u *)"\n");
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004466 verbose_leave_scroll();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004467 }
4468#endif
4469
4470 /* check depth */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004471 if (stackp->ffs_level <= 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004472 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004473 ff_free_stack_element(stackp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004474 continue;
4475 }
4476
4477 file_path[0] = NUL;
4478
4479 /*
4480 * If no filearray till now expand wildcards
4481 * The function expand_wildcards() can handle an array of paths
4482 * and all possible expands are returned in one array. We use this
4483 * to handle the expansion of '**' into an empty string.
4484 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004485 if (stackp->ffs_filearray == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004486 {
4487 char_u *dirptrs[2];
4488
4489 /* we use filepath to build the path expand_wildcards() should
4490 * expand.
4491 */
4492 dirptrs[0] = file_path;
4493 dirptrs[1] = NULL;
4494
4495 /* if we have a start dir copy it in */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004496 if (!vim_isAbsName(stackp->ffs_fix_path)
4497 && search_ctx->ffsc_start_dir)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004498 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004499 STRCPY(file_path, search_ctx->ffsc_start_dir);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004500 add_pathsep(file_path);
4501 }
4502
4503 /* append the fix part of the search path */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004504 STRCAT(file_path, stackp->ffs_fix_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004505 add_pathsep(file_path);
4506
4507#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004508 rest_of_wildcards = stackp->ffs_wc_path;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004509 if (*rest_of_wildcards != NUL)
4510 {
4511 len = (int)STRLEN(file_path);
4512 if (STRNCMP(rest_of_wildcards, "**", 2) == 0)
4513 {
4514 /* pointer to the restrict byte
4515 * The restrict byte is not a character!
4516 */
4517 p = rest_of_wildcards + 2;
4518
4519 if (*p > 0)
4520 {
4521 (*p)--;
4522 file_path[len++] = '*';
4523 }
4524
4525 if (*p == 0)
4526 {
4527 /* remove '**<numb> from wildcards */
Bram Moolenaar446cb832008-06-24 21:56:24 +00004528 STRMOVE(rest_of_wildcards, rest_of_wildcards + 3);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004529 }
4530 else
4531 rest_of_wildcards += 3;
4532
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004533 if (stackp->ffs_star_star_empty == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004534 {
4535 /* if not done before, expand '**' to empty */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004536 stackp->ffs_star_star_empty = 1;
4537 dirptrs[1] = stackp->ffs_fix_path;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004538 }
4539 }
4540
4541 /*
4542 * Here we copy until the next path separator or the end of
4543 * the path. If we stop at a path separator, there is
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00004544 * still something else left. This is handled below by
Bram Moolenaar071d4272004-06-13 20:20:40 +00004545 * pushing every directory returned from expand_wildcards()
4546 * on the stack again for further search.
4547 */
4548 while (*rest_of_wildcards
4549 && !vim_ispathsep(*rest_of_wildcards))
4550 file_path[len++] = *rest_of_wildcards++;
4551
4552 file_path[len] = NUL;
4553 if (vim_ispathsep(*rest_of_wildcards))
4554 rest_of_wildcards++;
4555 }
4556#endif
4557
4558 /*
4559 * Expand wildcards like "*" and "$VAR".
4560 * If the path is a URL don't try this.
4561 */
4562 if (path_with_url(dirptrs[0]))
4563 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004564 stackp->ffs_filearray = (char_u **)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004565 alloc((unsigned)sizeof(char *));
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004566 if (stackp->ffs_filearray != NULL
4567 && (stackp->ffs_filearray[0]
Bram Moolenaar071d4272004-06-13 20:20:40 +00004568 = vim_strsave(dirptrs[0])) != NULL)
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004569 stackp->ffs_filearray_size = 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004570 else
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004571 stackp->ffs_filearray_size = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004572 }
4573 else
4574 expand_wildcards((dirptrs[1] == NULL) ? 1 : 2, dirptrs,
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004575 &stackp->ffs_filearray_size,
4576 &stackp->ffs_filearray,
Bram Moolenaar071d4272004-06-13 20:20:40 +00004577 EW_DIR|EW_ADDSLASH|EW_SILENT);
4578
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004579 stackp->ffs_filearray_cur = 0;
4580 stackp->ffs_stage = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004581 }
4582#ifdef FEAT_PATH_EXTRA
4583 else
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004584 rest_of_wildcards = &stackp->ffs_wc_path[
4585 STRLEN(stackp->ffs_wc_path)];
Bram Moolenaar071d4272004-06-13 20:20:40 +00004586#endif
4587
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004588 if (stackp->ffs_stage == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004589 {
4590 /* this is the first time we work on this directory */
4591#ifdef FEAT_PATH_EXTRA
4592 if (*rest_of_wildcards == NUL)
4593#endif
4594 {
4595 /*
4596 * we don't have further wildcards to expand, so we have to
4597 * check for the final file now
4598 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004599 for (i = stackp->ffs_filearray_cur;
4600 i < stackp->ffs_filearray_size; ++i)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004601 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004602 if (!path_with_url(stackp->ffs_filearray[i])
4603 && !mch_isdir(stackp->ffs_filearray[i]))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004604 continue; /* not a directory */
4605
4606 /* prepare the filename to be checked for existance
4607 * below */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004608 STRCPY(file_path, stackp->ffs_filearray[i]);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004609 add_pathsep(file_path);
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004610 STRCAT(file_path, search_ctx->ffsc_file_to_search);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004611
4612 /*
4613 * Try without extra suffix and then with suffixes
4614 * from 'suffixesadd'.
4615 */
4616#ifdef FEAT_SEARCHPATH
4617 len = (int)STRLEN(file_path);
4618 suf = curbuf->b_p_sua;
4619 for (;;)
4620#endif
4621 {
4622 /* if file exists and we didn't already find it */
4623 if ((path_with_url(file_path)
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004624 || (mch_getperm(file_path) >= 0
4625 && (search_ctx->ffsc_find_what
4626 == FINDFILE_BOTH
4627 || ((search_ctx->ffsc_find_what
4628 == FINDFILE_DIR)
4629 == mch_isdir(file_path)))))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004630#ifndef FF_VERBOSE
4631 && (ff_check_visited(
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004632 &search_ctx->ffsc_visited_list->ffvl_visited_list,
Bram Moolenaar071d4272004-06-13 20:20:40 +00004633 file_path
4634#ifdef FEAT_PATH_EXTRA
4635 , (char_u *)""
4636#endif
4637 ) == OK)
4638#endif
4639 )
4640 {
4641#ifdef FF_VERBOSE
4642 if (ff_check_visited(
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004643 &search_ctx->ffsc_visited_list->ffvl_visited_list,
Bram Moolenaar071d4272004-06-13 20:20:40 +00004644 file_path
4645#ifdef FEAT_PATH_EXTRA
4646 , (char_u *)""
4647#endif
4648 ) == FAIL)
4649 {
4650 if (p_verbose >= 5)
4651 {
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004652 verbose_enter_scroll();
Bram Moolenaar051b7822005-05-19 21:00:46 +00004653 smsg((char_u *)"Already: %s",
Bram Moolenaar071d4272004-06-13 20:20:40 +00004654 file_path);
4655 /* don't overwrite this either */
4656 msg_puts((char_u *)"\n");
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004657 verbose_leave_scroll();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004658 }
4659 continue;
4660 }
4661#endif
4662
4663 /* push dir to examine rest of subdirs later */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004664 stackp->ffs_filearray_cur = i + 1;
4665 ff_push(search_ctx, stackp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004666
4667 simplify_filename(file_path);
4668 if (mch_dirname(ff_expand_buffer, MAXPATHL)
4669 == OK)
4670 {
4671 p = shorten_fname(file_path,
4672 ff_expand_buffer);
4673 if (p != NULL)
Bram Moolenaar446cb832008-06-24 21:56:24 +00004674 STRMOVE(file_path, p);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004675 }
4676#ifdef FF_VERBOSE
4677 if (p_verbose >= 5)
4678 {
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004679 verbose_enter_scroll();
Bram Moolenaar051b7822005-05-19 21:00:46 +00004680 smsg((char_u *)"HIT: %s", file_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004681 /* don't overwrite this either */
4682 msg_puts((char_u *)"\n");
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004683 verbose_leave_scroll();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004684 }
4685#endif
4686 return file_path;
4687 }
4688
4689#ifdef FEAT_SEARCHPATH
4690 /* Not found or found already, try next suffix. */
4691 if (*suf == NUL)
4692 break;
4693 copy_option_part(&suf, file_path + len,
4694 MAXPATHL - len, ",");
4695#endif
4696 }
4697 }
4698 }
4699#ifdef FEAT_PATH_EXTRA
4700 else
4701 {
4702 /*
4703 * still wildcards left, push the directories for further
4704 * search
4705 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004706 for (i = stackp->ffs_filearray_cur;
4707 i < stackp->ffs_filearray_size; ++i)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004708 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004709 if (!mch_isdir(stackp->ffs_filearray[i]))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004710 continue; /* not a directory */
4711
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004712 ff_push(search_ctx,
4713 ff_create_stack_element(
4714 stackp->ffs_filearray[i],
4715 rest_of_wildcards,
4716 stackp->ffs_level - 1, 0));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004717 }
4718 }
4719#endif
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004720 stackp->ffs_filearray_cur = 0;
4721 stackp->ffs_stage = 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004722 }
4723
4724#ifdef FEAT_PATH_EXTRA
4725 /*
4726 * if wildcards contains '**' we have to descent till we reach the
4727 * leaves of the directory tree.
4728 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004729 if (STRNCMP(stackp->ffs_wc_path, "**", 2) == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004730 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004731 for (i = stackp->ffs_filearray_cur;
4732 i < stackp->ffs_filearray_size; ++i)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004733 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004734 if (fnamecmp(stackp->ffs_filearray[i],
4735 stackp->ffs_fix_path) == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004736 continue; /* don't repush same directory */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004737 if (!mch_isdir(stackp->ffs_filearray[i]))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004738 continue; /* not a directory */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004739 ff_push(search_ctx,
4740 ff_create_stack_element(stackp->ffs_filearray[i],
4741 stackp->ffs_wc_path, stackp->ffs_level - 1, 1));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004742 }
4743 }
4744#endif
4745
4746 /* we are done with the current directory */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004747 ff_free_stack_element(stackp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004748
4749 }
4750
4751#ifdef FEAT_PATH_EXTRA
4752 /* If we reached this, we didn't find anything downwards.
4753 * Let's check if we should do an upward search.
4754 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004755 if (search_ctx->ffsc_start_dir
4756 && search_ctx->ffsc_stopdirs_v != NULL && !got_int)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004757 {
4758 ff_stack_T *sptr;
4759
4760 /* is the last starting directory in the stop list? */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004761 if (ff_path_in_stoplist(search_ctx->ffsc_start_dir,
4762 (int)(path_end - search_ctx->ffsc_start_dir),
4763 search_ctx->ffsc_stopdirs_v) == TRUE)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004764 break;
4765
4766 /* cut of last dir */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004767 while (path_end > search_ctx->ffsc_start_dir
4768 && vim_ispathsep(*path_end))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004769 path_end--;
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004770 while (path_end > search_ctx->ffsc_start_dir
4771 && !vim_ispathsep(path_end[-1]))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004772 path_end--;
4773 *path_end = 0;
4774 path_end--;
4775
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004776 if (*search_ctx->ffsc_start_dir == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004777 break;
4778
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004779 STRCPY(file_path, search_ctx->ffsc_start_dir);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004780 add_pathsep(file_path);
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004781 STRCAT(file_path, search_ctx->ffsc_fix_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004782
4783 /* create a new stack entry */
4784 sptr = ff_create_stack_element(file_path,
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004785 search_ctx->ffsc_wc_path, search_ctx->ffsc_level, 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004786 if (sptr == NULL)
4787 break;
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004788 ff_push(search_ctx, sptr);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004789 }
4790 else
4791 break;
4792 }
4793#endif
4794
4795 vim_free(file_path);
4796 return NULL;
4797}
4798
4799/*
4800 * Free the list of lists of visited files and directories
4801 * Can handle it if the passed search_context is NULL;
4802 */
4803 void
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004804vim_findfile_free_visited(search_ctx_arg)
4805 void *search_ctx_arg;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004806{
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004807 ff_search_ctx_T *search_ctx;
4808
4809 if (search_ctx_arg == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004810 return;
4811
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004812 search_ctx = (ff_search_ctx_T *)search_ctx_arg;
4813 vim_findfile_free_visited_list(&search_ctx->ffsc_visited_lists_list);
4814 vim_findfile_free_visited_list(&search_ctx->ffsc_dir_visited_lists_list);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004815}
4816
4817 static void
4818vim_findfile_free_visited_list(list_headp)
4819 ff_visited_list_hdr_T **list_headp;
4820{
4821 ff_visited_list_hdr_T *vp;
4822
4823 while (*list_headp != NULL)
4824 {
4825 vp = (*list_headp)->ffvl_next;
4826 ff_free_visited_list((*list_headp)->ffvl_visited_list);
4827
4828 vim_free((*list_headp)->ffvl_filename);
4829 vim_free(*list_headp);
4830 *list_headp = vp;
4831 }
4832 *list_headp = NULL;
4833}
4834
4835 static void
4836ff_free_visited_list(vl)
4837 ff_visited_T *vl;
4838{
4839 ff_visited_T *vp;
4840
4841 while (vl != NULL)
4842 {
4843 vp = vl->ffv_next;
4844#ifdef FEAT_PATH_EXTRA
4845 vim_free(vl->ffv_wc_path);
4846#endif
4847 vim_free(vl);
4848 vl = vp;
4849 }
4850 vl = NULL;
4851}
4852
4853/*
4854 * Returns the already visited list for the given filename. If none is found it
4855 * allocates a new one.
4856 */
4857 static ff_visited_list_hdr_T*
4858ff_get_visited_list(filename, list_headp)
4859 char_u *filename;
4860 ff_visited_list_hdr_T **list_headp;
4861{
4862 ff_visited_list_hdr_T *retptr = NULL;
4863
4864 /* check if a visited list for the given filename exists */
4865 if (*list_headp != NULL)
4866 {
4867 retptr = *list_headp;
4868 while (retptr != NULL)
4869 {
4870 if (fnamecmp(filename, retptr->ffvl_filename) == 0)
4871 {
4872#ifdef FF_VERBOSE
4873 if (p_verbose >= 5)
4874 {
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004875 verbose_enter_scroll();
Bram Moolenaar051b7822005-05-19 21:00:46 +00004876 smsg((char_u *)"ff_get_visited_list: FOUND list for %s",
Bram Moolenaar071d4272004-06-13 20:20:40 +00004877 filename);
4878 /* don't overwrite this either */
4879 msg_puts((char_u *)"\n");
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004880 verbose_leave_scroll();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004881 }
4882#endif
4883 return retptr;
4884 }
4885 retptr = retptr->ffvl_next;
4886 }
4887 }
4888
4889#ifdef FF_VERBOSE
4890 if (p_verbose >= 5)
4891 {
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004892 verbose_enter_scroll();
Bram Moolenaar051b7822005-05-19 21:00:46 +00004893 smsg((char_u *)"ff_get_visited_list: new list for %s", filename);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004894 /* don't overwrite this either */
4895 msg_puts((char_u *)"\n");
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004896 verbose_leave_scroll();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004897 }
4898#endif
4899
4900 /*
4901 * if we reach this we didn't find a list and we have to allocate new list
4902 */
4903 retptr = (ff_visited_list_hdr_T*)alloc((unsigned)sizeof(*retptr));
4904 if (retptr == NULL)
4905 return NULL;
4906
4907 retptr->ffvl_visited_list = NULL;
4908 retptr->ffvl_filename = vim_strsave(filename);
4909 if (retptr->ffvl_filename == NULL)
4910 {
4911 vim_free(retptr);
4912 return NULL;
4913 }
4914 retptr->ffvl_next = *list_headp;
4915 *list_headp = retptr;
4916
4917 return retptr;
4918}
4919
4920#ifdef FEAT_PATH_EXTRA
4921/*
4922 * check if two wildcard paths are equal. Returns TRUE or FALSE.
4923 * They are equal if:
4924 * - both paths are NULL
4925 * - they have the same length
4926 * - char by char comparison is OK
4927 * - the only differences are in the counters behind a '**', so
4928 * '**\20' is equal to '**\24'
4929 */
4930 static int
4931ff_wc_equal(s1, s2)
4932 char_u *s1;
4933 char_u *s2;
4934{
4935 int i;
4936
4937 if (s1 == s2)
4938 return TRUE;
4939
4940 if (s1 == NULL || s2 == NULL)
4941 return FALSE;
4942
4943 if (STRLEN(s1) != STRLEN(s2))
4944 return FAIL;
4945
4946 for (i = 0; s1[i] != NUL && s2[i] != NUL; i++)
4947 {
4948 if (s1[i] != s2[i]
4949#ifdef CASE_INSENSITIVE_FILENAME
4950 && TOUPPER_LOC(s1[i]) != TOUPPER_LOC(s2[i])
4951#endif
4952 )
4953 {
4954 if (i >= 2)
4955 if (s1[i-1] == '*' && s1[i-2] == '*')
4956 continue;
4957 else
4958 return FAIL;
4959 else
4960 return FAIL;
4961 }
4962 }
4963 return TRUE;
4964}
4965#endif
4966
4967/*
4968 * maintains the list of already visited files and dirs
4969 * returns FAIL if the given file/dir is already in the list
4970 * returns OK if it is newly added
4971 *
4972 * TODO: What to do on memory allocation problems?
4973 * -> return TRUE - Better the file is found several times instead of
4974 * never.
4975 */
4976 static int
4977ff_check_visited(visited_list, fname
4978#ifdef FEAT_PATH_EXTRA
4979 , wc_path
4980#endif
4981 )
4982 ff_visited_T **visited_list;
4983 char_u *fname;
4984#ifdef FEAT_PATH_EXTRA
4985 char_u *wc_path;
4986#endif
4987{
4988 ff_visited_T *vp;
4989#ifdef UNIX
4990 struct stat st;
4991 int url = FALSE;
4992#endif
4993
4994 /* For an URL we only compare the name, otherwise we compare the
4995 * device/inode (unix) or the full path name (not Unix). */
4996 if (path_with_url(fname))
4997 {
Bram Moolenaarbbebc852005-07-18 21:47:53 +00004998 vim_strncpy(ff_expand_buffer, fname, MAXPATHL - 1);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004999#ifdef UNIX
5000 url = TRUE;
5001#endif
5002 }
5003 else
5004 {
5005 ff_expand_buffer[0] = NUL;
5006#ifdef UNIX
5007 if (mch_stat((char *)fname, &st) < 0)
5008#else
5009 if (vim_FullName(fname, ff_expand_buffer, MAXPATHL, TRUE) == FAIL)
5010#endif
5011 return FAIL;
5012 }
5013
5014 /* check against list of already visited files */
5015 for (vp = *visited_list; vp != NULL; vp = vp->ffv_next)
5016 {
5017 if (
5018#ifdef UNIX
5019 !url
5020 ? (vp->ffv_dev == st.st_dev
5021 && vp->ffv_ino == st.st_ino)
5022 :
5023#endif
5024 fnamecmp(vp->ffv_fname, ff_expand_buffer) == 0
5025 )
5026 {
5027#ifdef FEAT_PATH_EXTRA
5028 /* are the wildcard parts equal */
5029 if (ff_wc_equal(vp->ffv_wc_path, wc_path) == TRUE)
5030#endif
5031 /* already visited */
5032 return FAIL;
5033 }
5034 }
5035
5036 /*
5037 * New file/dir. Add it to the list of visited files/dirs.
5038 */
5039 vp = (ff_visited_T *)alloc((unsigned)(sizeof(ff_visited_T)
5040 + STRLEN(ff_expand_buffer)));
5041
5042 if (vp != NULL)
5043 {
5044#ifdef UNIX
5045 if (!url)
5046 {
5047 vp->ffv_ino = st.st_ino;
5048 vp->ffv_dev = st.st_dev;
5049 vp->ffv_fname[0] = NUL;
5050 }
5051 else
5052 {
5053 vp->ffv_ino = 0;
5054 vp->ffv_dev = -1;
5055#endif
5056 STRCPY(vp->ffv_fname, ff_expand_buffer);
5057#ifdef UNIX
5058 }
5059#endif
5060#ifdef FEAT_PATH_EXTRA
5061 if (wc_path != NULL)
5062 vp->ffv_wc_path = vim_strsave(wc_path);
5063 else
5064 vp->ffv_wc_path = NULL;
5065#endif
5066
5067 vp->ffv_next = *visited_list;
5068 *visited_list = vp;
5069 }
5070
5071 return OK;
5072}
5073
5074/*
5075 * create stack element from given path pieces
5076 */
5077 static ff_stack_T *
5078ff_create_stack_element(fix_part,
5079#ifdef FEAT_PATH_EXTRA
5080 wc_part,
5081#endif
5082 level, star_star_empty)
5083 char_u *fix_part;
5084#ifdef FEAT_PATH_EXTRA
5085 char_u *wc_part;
5086#endif
5087 int level;
5088 int star_star_empty;
5089{
5090 ff_stack_T *new;
5091
5092 new = (ff_stack_T *)alloc((unsigned)sizeof(ff_stack_T));
5093 if (new == NULL)
5094 return NULL;
5095
5096 new->ffs_prev = NULL;
5097 new->ffs_filearray = NULL;
5098 new->ffs_filearray_size = 0;
5099 new->ffs_filearray_cur = 0;
5100 new->ffs_stage = 0;
5101 new->ffs_level = level;
5102 new->ffs_star_star_empty = star_star_empty;;
5103
5104 /* the following saves NULL pointer checks in vim_findfile */
5105 if (fix_part == NULL)
5106 fix_part = (char_u *)"";
5107 new->ffs_fix_path = vim_strsave(fix_part);
5108
5109#ifdef FEAT_PATH_EXTRA
5110 if (wc_part == NULL)
5111 wc_part = (char_u *)"";
5112 new->ffs_wc_path = vim_strsave(wc_part);
5113#endif
5114
5115 if (new->ffs_fix_path == NULL
5116#ifdef FEAT_PATH_EXTRA
5117 || new->ffs_wc_path == NULL
5118#endif
5119 )
5120 {
5121 ff_free_stack_element(new);
5122 new = NULL;
5123 }
5124
5125 return new;
5126}
5127
5128/*
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005129 * Push a dir on the directory stack.
Bram Moolenaar071d4272004-06-13 20:20:40 +00005130 */
5131 static void
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005132ff_push(search_ctx, stack_ptr)
5133 ff_search_ctx_T *search_ctx;
5134 ff_stack_T *stack_ptr;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005135{
5136 /* check for NULL pointer, not to return an error to the user, but
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005137 * to prevent a crash */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005138 if (stack_ptr != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005139 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005140 stack_ptr->ffs_prev = search_ctx->ffsc_stack_ptr;
5141 search_ctx->ffsc_stack_ptr = stack_ptr;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005142 }
5143}
5144
5145/*
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005146 * Pop a dir from the directory stack.
5147 * Returns NULL if stack is empty.
Bram Moolenaar071d4272004-06-13 20:20:40 +00005148 */
5149 static ff_stack_T *
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005150ff_pop(search_ctx)
5151 ff_search_ctx_T *search_ctx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005152{
5153 ff_stack_T *sptr;
5154
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005155 sptr = search_ctx->ffsc_stack_ptr;
5156 if (search_ctx->ffsc_stack_ptr != NULL)
5157 search_ctx->ffsc_stack_ptr = search_ctx->ffsc_stack_ptr->ffs_prev;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005158
5159 return sptr;
5160}
5161
5162/*
5163 * free the given stack element
5164 */
5165 static void
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005166ff_free_stack_element(stack_ptr)
5167 ff_stack_T *stack_ptr;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005168{
5169 /* vim_free handles possible NULL pointers */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005170 vim_free(stack_ptr->ffs_fix_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005171#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005172 vim_free(stack_ptr->ffs_wc_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005173#endif
5174
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005175 if (stack_ptr->ffs_filearray != NULL)
5176 FreeWild(stack_ptr->ffs_filearray_size, stack_ptr->ffs_filearray);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005177
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005178 vim_free(stack_ptr);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005179}
5180
5181/*
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005182 * Clear the search context, but NOT the visited list.
Bram Moolenaar071d4272004-06-13 20:20:40 +00005183 */
5184 static void
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005185ff_clear(search_ctx)
5186 ff_search_ctx_T *search_ctx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005187{
5188 ff_stack_T *sptr;
5189
5190 /* clear up stack */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005191 while ((sptr = ff_pop(search_ctx)) != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005192 ff_free_stack_element(sptr);
5193
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005194 vim_free(search_ctx->ffsc_file_to_search);
5195 vim_free(search_ctx->ffsc_start_dir);
5196 vim_free(search_ctx->ffsc_fix_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005197#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005198 vim_free(search_ctx->ffsc_wc_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005199#endif
5200
5201#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005202 if (search_ctx->ffsc_stopdirs_v != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005203 {
5204 int i = 0;
5205
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005206 while (search_ctx->ffsc_stopdirs_v[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005207 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005208 vim_free(search_ctx->ffsc_stopdirs_v[i]);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005209 i++;
5210 }
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005211 vim_free(search_ctx->ffsc_stopdirs_v);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005212 }
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005213 search_ctx->ffsc_stopdirs_v = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005214#endif
5215
5216 /* reset everything */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005217 search_ctx->ffsc_file_to_search = NULL;
5218 search_ctx->ffsc_start_dir = NULL;
5219 search_ctx->ffsc_fix_path = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005220#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005221 search_ctx->ffsc_wc_path = NULL;
5222 search_ctx->ffsc_level = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005223#endif
5224}
5225
5226#ifdef FEAT_PATH_EXTRA
5227/*
5228 * check if the given path is in the stopdirs
5229 * returns TRUE if yes else FALSE
5230 */
5231 static int
5232ff_path_in_stoplist(path, path_len, stopdirs_v)
5233 char_u *path;
5234 int path_len;
5235 char_u **stopdirs_v;
5236{
5237 int i = 0;
5238
5239 /* eat up trailing path separators, except the first */
Bram Moolenaar1d94f9b2005-08-04 21:29:45 +00005240 while (path_len > 1 && vim_ispathsep(path[path_len - 1]))
Bram Moolenaar071d4272004-06-13 20:20:40 +00005241 path_len--;
5242
5243 /* if no path consider it as match */
5244 if (path_len == 0)
5245 return TRUE;
5246
5247 for (i = 0; stopdirs_v[i] != NULL; i++)
5248 {
5249 if ((int)STRLEN(stopdirs_v[i]) > path_len)
5250 {
5251 /* match for parent directory. So '/home' also matches
5252 * '/home/rks'. Check for PATHSEP in stopdirs_v[i], else
5253 * '/home/r' would also match '/home/rks'
5254 */
5255 if (fnamencmp(stopdirs_v[i], path, path_len) == 0
Bram Moolenaar1d94f9b2005-08-04 21:29:45 +00005256 && vim_ispathsep(stopdirs_v[i][path_len]))
Bram Moolenaar071d4272004-06-13 20:20:40 +00005257 return TRUE;
5258 }
5259 else
5260 {
5261 if (fnamecmp(stopdirs_v[i], path) == 0)
5262 return TRUE;
5263 }
5264 }
5265 return FALSE;
5266}
5267#endif
5268
5269#if defined(FEAT_SEARCHPATH) || defined(PROTO)
5270/*
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005271 * Find the file name "ptr[len]" in the path. Also finds directory names.
Bram Moolenaar071d4272004-06-13 20:20:40 +00005272 *
5273 * On the first call set the parameter 'first' to TRUE to initialize
5274 * the search. For repeating calls to FALSE.
5275 *
5276 * Repeating calls will return other files called 'ptr[len]' from the path.
5277 *
5278 * Only on the first call 'ptr' and 'len' are used. For repeating calls they
5279 * don't need valid values.
5280 *
5281 * If nothing found on the first call the option FNAME_MESS will issue the
5282 * message:
5283 * 'Can't find file "<file>" in path'
5284 * On repeating calls:
5285 * 'No more file "<file>" found in path'
5286 *
5287 * options:
5288 * FNAME_MESS give error message when not found
5289 *
5290 * Uses NameBuff[]!
5291 *
5292 * Returns an allocated string for the file name. NULL for error.
5293 *
5294 */
5295 char_u *
5296find_file_in_path(ptr, len, options, first, rel_fname)
5297 char_u *ptr; /* file name */
5298 int len; /* length of file name */
5299 int options;
5300 int first; /* use count'th matching file name */
5301 char_u *rel_fname; /* file name searching relative to */
5302{
5303 return find_file_in_path_option(ptr, len, options, first,
5304 *curbuf->b_p_path == NUL ? p_path : curbuf->b_p_path,
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005305 FINDFILE_BOTH, rel_fname, curbuf->b_p_sua);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005306}
5307
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005308static char_u *ff_file_to_find = NULL;
5309static void *fdip_search_ctx = NULL;
5310
5311#if defined(EXITFREE)
5312 static void
5313free_findfile()
5314{
5315 vim_free(ff_file_to_find);
5316 vim_findfile_cleanup(fdip_search_ctx);
5317}
5318#endif
5319
Bram Moolenaar071d4272004-06-13 20:20:40 +00005320/*
5321 * Find the directory name "ptr[len]" in the path.
5322 *
5323 * options:
5324 * FNAME_MESS give error message when not found
5325 *
5326 * Uses NameBuff[]!
5327 *
5328 * Returns an allocated string for the file name. NULL for error.
5329 */
5330 char_u *
5331find_directory_in_path(ptr, len, options, rel_fname)
5332 char_u *ptr; /* file name */
5333 int len; /* length of file name */
5334 int options;
5335 char_u *rel_fname; /* file name searching relative to */
5336{
5337 return find_file_in_path_option(ptr, len, options, TRUE, p_cdpath,
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005338 FINDFILE_DIR, rel_fname, (char_u *)"");
Bram Moolenaar071d4272004-06-13 20:20:40 +00005339}
5340
Bram Moolenaar89cb5e02004-07-19 20:55:54 +00005341 char_u *
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005342find_file_in_path_option(ptr, len, options, first, path_option, find_what, rel_fname, suffixes)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005343 char_u *ptr; /* file name */
5344 int len; /* length of file name */
5345 int options;
5346 int first; /* use count'th matching file name */
5347 char_u *path_option; /* p_path or p_cdpath */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005348 int find_what; /* FINDFILE_FILE, _DIR or _BOTH */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005349 char_u *rel_fname; /* file name we are looking relative to. */
Bram Moolenaar433f7c82006-03-21 21:29:36 +00005350 char_u *suffixes; /* list of suffixes, 'suffixesadd' option */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005351{
Bram Moolenaar071d4272004-06-13 20:20:40 +00005352 static char_u *dir;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005353 static int did_findfile_init = FALSE;
5354 char_u save_char;
5355 char_u *file_name = NULL;
5356 char_u *buf = NULL;
5357 int rel_to_curdir;
5358#ifdef AMIGA
5359 struct Process *proc = (struct Process *)FindTask(0L);
5360 APTR save_winptr = proc->pr_WindowPtr;
5361
5362 /* Avoid a requester here for a volume that doesn't exist. */
5363 proc->pr_WindowPtr = (APTR)-1L;
5364#endif
5365
5366 if (first == TRUE)
5367 {
5368 /* copy file name into NameBuff, expanding environment variables */
5369 save_char = ptr[len];
5370 ptr[len] = NUL;
5371 expand_env(ptr, NameBuff, MAXPATHL);
5372 ptr[len] = save_char;
5373
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005374 vim_free(ff_file_to_find);
5375 ff_file_to_find = vim_strsave(NameBuff);
5376 if (ff_file_to_find == NULL) /* out of memory */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005377 {
5378 file_name = NULL;
5379 goto theend;
5380 }
5381 }
5382
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005383 rel_to_curdir = (ff_file_to_find[0] == '.'
5384 && (ff_file_to_find[1] == NUL
5385 || vim_ispathsep(ff_file_to_find[1])
5386 || (ff_file_to_find[1] == '.'
5387 && (ff_file_to_find[2] == NUL
5388 || vim_ispathsep(ff_file_to_find[2])))));
5389 if (vim_isAbsName(ff_file_to_find)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005390 /* "..", "../path", "." and "./path": don't use the path_option */
5391 || rel_to_curdir
5392#if defined(MSWIN) || defined(MSDOS) || defined(OS2)
5393 /* handle "\tmp" as absolute path */
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005394 || vim_ispathsep(ff_file_to_find[0])
Bram Moolenaar071d4272004-06-13 20:20:40 +00005395 /* handle "c:name" as absulute path */
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005396 || (ff_file_to_find[0] != NUL && ff_file_to_find[1] == ':')
Bram Moolenaar071d4272004-06-13 20:20:40 +00005397#endif
5398#ifdef AMIGA
5399 /* handle ":tmp" as absolute path */
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005400 || ff_file_to_find[0] == ':'
Bram Moolenaar071d4272004-06-13 20:20:40 +00005401#endif
5402 )
5403 {
5404 /*
5405 * Absolute path, no need to use "path_option".
5406 * If this is not a first call, return NULL. We already returned a
5407 * filename on the first call.
5408 */
5409 if (first == TRUE)
5410 {
5411 int l;
5412 int run;
5413
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005414 if (path_with_url(ff_file_to_find))
Bram Moolenaar071d4272004-06-13 20:20:40 +00005415 {
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005416 file_name = vim_strsave(ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005417 goto theend;
5418 }
5419
5420 /* When FNAME_REL flag given first use the directory of the file.
5421 * Otherwise or when this fails use the current directory. */
5422 for (run = 1; run <= 2; ++run)
5423 {
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005424 l = (int)STRLEN(ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005425 if (run == 1
5426 && rel_to_curdir
5427 && (options & FNAME_REL)
5428 && rel_fname != NULL
5429 && STRLEN(rel_fname) + l < MAXPATHL)
5430 {
5431 STRCPY(NameBuff, rel_fname);
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005432 STRCPY(gettail(NameBuff), ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005433 l = (int)STRLEN(NameBuff);
5434 }
5435 else
5436 {
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005437 STRCPY(NameBuff, ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005438 run = 2;
5439 }
5440
5441 /* When the file doesn't exist, try adding parts of
5442 * 'suffixesadd'. */
Bram Moolenaar433f7c82006-03-21 21:29:36 +00005443 buf = suffixes;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005444 for (;;)
5445 {
5446 if (
5447#ifdef DJGPP
5448 /* "C:" by itself will fail for mch_getperm(),
5449 * assume it's always valid. */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005450 (find_what != FINDFILE_FILE && NameBuff[0] != NUL
Bram Moolenaar071d4272004-06-13 20:20:40 +00005451 && NameBuff[1] == ':'
5452 && NameBuff[2] == NUL) ||
5453#endif
5454 (mch_getperm(NameBuff) >= 0
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005455 && (find_what == FINDFILE_BOTH
5456 || ((find_what == FINDFILE_DIR)
5457 == mch_isdir(NameBuff)))))
Bram Moolenaar071d4272004-06-13 20:20:40 +00005458 {
5459 file_name = vim_strsave(NameBuff);
5460 goto theend;
5461 }
5462 if (*buf == NUL)
5463 break;
5464 copy_option_part(&buf, NameBuff + l, MAXPATHL - l, ",");
5465 }
5466 }
5467 }
5468 }
5469 else
5470 {
5471 /*
5472 * Loop over all paths in the 'path' or 'cdpath' option.
5473 * When "first" is set, first setup to the start of the option.
5474 * Otherwise continue to find the next match.
5475 */
5476 if (first == TRUE)
5477 {
5478 /* vim_findfile_free_visited can handle a possible NULL pointer */
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005479 vim_findfile_free_visited(fdip_search_ctx);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005480 dir = path_option;
5481 did_findfile_init = FALSE;
5482 }
5483
5484 for (;;)
5485 {
5486 if (did_findfile_init)
5487 {
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005488 file_name = vim_findfile(fdip_search_ctx);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005489 if (file_name != NULL)
5490 break;
5491
5492 did_findfile_init = FALSE;
5493 }
5494 else
5495 {
5496 char_u *r_ptr;
5497
5498 if (dir == NULL || *dir == NUL)
5499 {
5500 /* We searched all paths of the option, now we can
5501 * free the search context. */
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005502 vim_findfile_cleanup(fdip_search_ctx);
5503 fdip_search_ctx = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005504 break;
5505 }
5506
5507 if ((buf = alloc((int)(MAXPATHL))) == NULL)
5508 break;
5509
5510 /* copy next path */
5511 buf[0] = 0;
5512 copy_option_part(&dir, buf, MAXPATHL, " ,");
5513
5514#ifdef FEAT_PATH_EXTRA
5515 /* get the stopdir string */
5516 r_ptr = vim_findfile_stopdir(buf);
5517#else
5518 r_ptr = NULL;
5519#endif
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005520 fdip_search_ctx = vim_findfile_init(buf, ff_file_to_find,
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005521 r_ptr, 100, FALSE, find_what,
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005522 fdip_search_ctx, FALSE, rel_fname);
5523 if (fdip_search_ctx != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005524 did_findfile_init = TRUE;
5525 vim_free(buf);
5526 }
5527 }
5528 }
5529 if (file_name == NULL && (options & FNAME_MESS))
5530 {
5531 if (first == TRUE)
5532 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005533 if (find_what == FINDFILE_DIR)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005534 EMSG2(_("E344: Can't find directory \"%s\" in cdpath"),
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005535 ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005536 else
5537 EMSG2(_("E345: Can't find file \"%s\" in path"),
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005538 ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005539 }
5540 else
5541 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005542 if (find_what == FINDFILE_DIR)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005543 EMSG2(_("E346: No more directory \"%s\" found in cdpath"),
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005544 ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005545 else
5546 EMSG2(_("E347: No more file \"%s\" found in path"),
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005547 ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005548 }
5549 }
5550
5551theend:
5552#ifdef AMIGA
5553 proc->pr_WindowPtr = save_winptr;
5554#endif
5555 return file_name;
5556}
5557
5558#endif /* FEAT_SEARCHPATH */
5559
5560/*
5561 * Change directory to "new_dir". If FEAT_SEARCHPATH is defined, search
5562 * 'cdpath' for relative directory names, otherwise just mch_chdir().
5563 */
5564 int
5565vim_chdir(new_dir)
5566 char_u *new_dir;
5567{
5568#ifndef FEAT_SEARCHPATH
5569 return mch_chdir((char *)new_dir);
5570#else
5571 char_u *dir_name;
5572 int r;
5573
5574 dir_name = find_directory_in_path(new_dir, (int)STRLEN(new_dir),
5575 FNAME_MESS, curbuf->b_ffname);
5576 if (dir_name == NULL)
5577 return -1;
5578 r = mch_chdir((char *)dir_name);
5579 vim_free(dir_name);
5580 return r;
5581#endif
5582}
5583
5584/*
Bram Moolenaarbbebc852005-07-18 21:47:53 +00005585 * Get user name from machine-specific function.
Bram Moolenaar071d4272004-06-13 20:20:40 +00005586 * Returns the user name in "buf[len]".
Bram Moolenaarbbebc852005-07-18 21:47:53 +00005587 * Some systems are quite slow in obtaining the user name (Windows NT), thus
5588 * cache the result.
Bram Moolenaar071d4272004-06-13 20:20:40 +00005589 * Returns OK or FAIL.
5590 */
5591 int
5592get_user_name(buf, len)
5593 char_u *buf;
5594 int len;
5595{
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005596 if (username == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005597 {
5598 if (mch_get_user_name(buf, len) == FAIL)
5599 return FAIL;
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005600 username = vim_strsave(buf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005601 }
5602 else
Bram Moolenaarbbebc852005-07-18 21:47:53 +00005603 vim_strncpy(buf, username, len - 1);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005604 return OK;
5605}
5606
5607#ifndef HAVE_QSORT
5608/*
5609 * Our own qsort(), for systems that don't have it.
5610 * It's simple and slow. From the K&R C book.
5611 */
5612 void
5613qsort(base, elm_count, elm_size, cmp)
5614 void *base;
5615 size_t elm_count;
5616 size_t elm_size;
5617 int (*cmp) __ARGS((const void *, const void *));
5618{
5619 char_u *buf;
5620 char_u *p1;
5621 char_u *p2;
5622 int i, j;
5623 int gap;
5624
5625 buf = alloc((unsigned)elm_size);
5626 if (buf == NULL)
5627 return;
5628
5629 for (gap = elm_count / 2; gap > 0; gap /= 2)
5630 for (i = gap; i < elm_count; ++i)
5631 for (j = i - gap; j >= 0; j -= gap)
5632 {
5633 /* Compare the elements. */
5634 p1 = (char_u *)base + j * elm_size;
5635 p2 = (char_u *)base + (j + gap) * elm_size;
5636 if ((*cmp)((void *)p1, (void *)p2) <= 0)
5637 break;
5638 /* Exchange the elemets. */
5639 mch_memmove(buf, p1, elm_size);
5640 mch_memmove(p1, p2, elm_size);
5641 mch_memmove(p2, buf, elm_size);
5642 }
5643
5644 vim_free(buf);
5645}
5646#endif
5647
Bram Moolenaar071d4272004-06-13 20:20:40 +00005648/*
5649 * Sort an array of strings.
5650 */
5651static int
5652#ifdef __BORLANDC__
5653_RTLENTRYF
5654#endif
5655sort_compare __ARGS((const void *s1, const void *s2));
5656
5657 static int
5658#ifdef __BORLANDC__
5659_RTLENTRYF
5660#endif
5661sort_compare(s1, s2)
5662 const void *s1;
5663 const void *s2;
5664{
5665 return STRCMP(*(char **)s1, *(char **)s2);
5666}
5667
5668 void
5669sort_strings(files, count)
5670 char_u **files;
5671 int count;
5672{
5673 qsort((void *)files, (size_t)count, sizeof(char_u *), sort_compare);
5674}
Bram Moolenaar071d4272004-06-13 20:20:40 +00005675
5676#if !defined(NO_EXPANDPATH) || defined(PROTO)
5677/*
5678 * Compare path "p[]" to "q[]".
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00005679 * If "maxlen" >= 0 compare "p[maxlen]" to "q[maxlen]"
Bram Moolenaar071d4272004-06-13 20:20:40 +00005680 * Return value like strcmp(p, q), but consider path separators.
5681 */
5682 int
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00005683pathcmp(p, q, maxlen)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005684 const char *p, *q;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00005685 int maxlen;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005686{
5687 int i;
Bram Moolenaar86b68352004-12-27 21:59:20 +00005688 const char *s = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005689
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00005690 for (i = 0; maxlen < 0 || i < maxlen; ++i)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005691 {
5692 /* End of "p": check if "q" also ends or just has a slash. */
5693 if (p[i] == NUL)
5694 {
5695 if (q[i] == NUL) /* full match */
5696 return 0;
5697 s = q;
5698 break;
5699 }
5700
5701 /* End of "q": check if "p" just has a slash. */
5702 if (q[i] == NUL)
5703 {
5704 s = p;
5705 break;
5706 }
5707
5708 if (
5709#ifdef CASE_INSENSITIVE_FILENAME
5710 TOUPPER_LOC(p[i]) != TOUPPER_LOC(q[i])
5711#else
5712 p[i] != q[i]
5713#endif
5714#ifdef BACKSLASH_IN_FILENAME
5715 /* consider '/' and '\\' to be equal */
5716 && !((p[i] == '/' && q[i] == '\\')
5717 || (p[i] == '\\' && q[i] == '/'))
5718#endif
5719 )
5720 {
5721 if (vim_ispathsep(p[i]))
5722 return -1;
5723 if (vim_ispathsep(q[i]))
5724 return 1;
5725 return ((char_u *)p)[i] - ((char_u *)q)[i]; /* no match */
5726 }
5727 }
Bram Moolenaar86b68352004-12-27 21:59:20 +00005728 if (s == NULL) /* "i" ran into "maxlen" */
5729 return 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005730
5731 /* ignore a trailing slash, but not "//" or ":/" */
Bram Moolenaar86b68352004-12-27 21:59:20 +00005732 if (s[i + 1] == NUL
5733 && i > 0
5734 && !after_pathsep((char_u *)s, (char_u *)s + i)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005735#ifdef BACKSLASH_IN_FILENAME
Bram Moolenaar86b68352004-12-27 21:59:20 +00005736 && (s[i] == '/' || s[i] == '\\')
Bram Moolenaar071d4272004-06-13 20:20:40 +00005737#else
Bram Moolenaar86b68352004-12-27 21:59:20 +00005738 && s[i] == '/'
Bram Moolenaar071d4272004-06-13 20:20:40 +00005739#endif
Bram Moolenaar86b68352004-12-27 21:59:20 +00005740 )
Bram Moolenaar071d4272004-06-13 20:20:40 +00005741 return 0; /* match with trailing slash */
5742 if (s == q)
5743 return -1; /* no match */
5744 return 1;
5745}
5746#endif
5747
Bram Moolenaar071d4272004-06-13 20:20:40 +00005748/*
5749 * The putenv() implementation below comes from the "screen" program.
5750 * Included with permission from Juergen Weigert.
5751 * See pty.c for the copyright notice.
5752 */
5753
5754/*
5755 * putenv -- put value into environment
5756 *
5757 * Usage: i = putenv (string)
5758 * int i;
5759 * char *string;
5760 *
5761 * where string is of the form <name>=<value>.
5762 * Putenv returns 0 normally, -1 on error (not enough core for malloc).
5763 *
5764 * Putenv may need to add a new name into the environment, or to
5765 * associate a value longer than the current value with a particular
5766 * name. So, to make life simpler, putenv() copies your entire
5767 * environment into the heap (i.e. malloc()) from the stack
5768 * (i.e. where it resides when your process is initiated) the first
5769 * time you call it.
5770 *
5771 * (history removed, not very interesting. See the "screen" sources.)
5772 */
5773
5774#if !defined(HAVE_SETENV) && !defined(HAVE_PUTENV)
5775
5776#define EXTRASIZE 5 /* increment to add to env. size */
5777
5778static int envsize = -1; /* current size of environment */
5779#ifndef MACOS_CLASSIC
5780extern
5781#endif
5782 char **environ; /* the global which is your env. */
5783
5784static int findenv __ARGS((char *name)); /* look for a name in the env. */
5785static int newenv __ARGS((void)); /* copy env. from stack to heap */
5786static int moreenv __ARGS((void)); /* incr. size of env. */
5787
5788 int
5789putenv(string)
5790 const char *string;
5791{
5792 int i;
5793 char *p;
5794
5795 if (envsize < 0)
5796 { /* first time putenv called */
5797 if (newenv() < 0) /* copy env. to heap */
5798 return -1;
5799 }
5800
5801 i = findenv((char *)string); /* look for name in environment */
5802
5803 if (i < 0)
5804 { /* name must be added */
5805 for (i = 0; environ[i]; i++);
5806 if (i >= (envsize - 1))
5807 { /* need new slot */
5808 if (moreenv() < 0)
5809 return -1;
5810 }
5811 p = (char *)alloc((unsigned)(strlen(string) + 1));
5812 if (p == NULL) /* not enough core */
5813 return -1;
5814 environ[i + 1] = 0; /* new end of env. */
5815 }
5816 else
5817 { /* name already in env. */
5818 p = vim_realloc(environ[i], strlen(string) + 1);
5819 if (p == NULL)
5820 return -1;
5821 }
5822 sprintf(p, "%s", string); /* copy into env. */
5823 environ[i] = p;
5824
5825 return 0;
5826}
5827
5828 static int
5829findenv(name)
5830 char *name;
5831{
5832 char *namechar, *envchar;
5833 int i, found;
5834
5835 found = 0;
5836 for (i = 0; environ[i] && !found; i++)
5837 {
5838 envchar = environ[i];
5839 namechar = name;
5840 while (*namechar && *namechar != '=' && (*namechar == *envchar))
5841 {
5842 namechar++;
5843 envchar++;
5844 }
5845 found = ((*namechar == '\0' || *namechar == '=') && *envchar == '=');
5846 }
5847 return found ? i - 1 : -1;
5848}
5849
5850 static int
5851newenv()
5852{
5853 char **env, *elem;
5854 int i, esize;
5855
5856#ifdef MACOS
5857 /* for Mac a new, empty environment is created */
5858 i = 0;
5859#else
5860 for (i = 0; environ[i]; i++)
5861 ;
5862#endif
5863 esize = i + EXTRASIZE + 1;
5864 env = (char **)alloc((unsigned)(esize * sizeof (elem)));
5865 if (env == NULL)
5866 return -1;
5867
5868#ifndef MACOS
5869 for (i = 0; environ[i]; i++)
5870 {
5871 elem = (char *)alloc((unsigned)(strlen(environ[i]) + 1));
5872 if (elem == NULL)
5873 return -1;
5874 env[i] = elem;
5875 strcpy(elem, environ[i]);
5876 }
5877#endif
5878
5879 env[i] = 0;
5880 environ = env;
5881 envsize = esize;
5882 return 0;
5883}
5884
5885 static int
5886moreenv()
5887{
5888 int esize;
5889 char **env;
5890
5891 esize = envsize + EXTRASIZE;
5892 env = (char **)vim_realloc((char *)environ, esize * sizeof (*env));
5893 if (env == 0)
5894 return -1;
5895 environ = env;
5896 envsize = esize;
5897 return 0;
5898}
5899
5900# ifdef USE_VIMPTY_GETENV
5901 char_u *
5902vimpty_getenv(string)
5903 const char_u *string;
5904{
5905 int i;
5906 char_u *p;
5907
5908 if (envsize < 0)
5909 return NULL;
5910
5911 i = findenv((char *)string);
5912
5913 if (i < 0)
5914 return NULL;
5915
5916 p = vim_strchr((char_u *)environ[i], '=');
5917 return (p + 1);
5918}
5919# endif
5920
5921#endif /* !defined(HAVE_SETENV) && !defined(HAVE_PUTENV) */
Bram Moolenaarc4a06d32005-06-07 21:04:49 +00005922
Bram Moolenaarc4956c82006-03-12 21:58:43 +00005923#if defined(FEAT_EVAL) || defined(FEAT_SPELL) || defined(PROTO)
Bram Moolenaarc4a06d32005-06-07 21:04:49 +00005924/*
5925 * Return 0 for not writable, 1 for writable file, 2 for a dir which we have
5926 * rights to write into.
5927 */
5928 int
5929filewritable(fname)
5930 char_u *fname;
5931{
5932 int retval = 0;
5933#if defined(UNIX) || defined(VMS)
5934 int perm = 0;
5935#endif
5936
5937#if defined(UNIX) || defined(VMS)
5938 perm = mch_getperm(fname);
5939#endif
5940#ifndef MACOS_CLASSIC /* TODO: get either mch_writable or mch_access */
5941 if (
5942# ifdef WIN3264
5943 mch_writable(fname) &&
5944# else
5945# if defined(UNIX) || defined(VMS)
5946 (perm & 0222) &&
5947# endif
5948# endif
5949 mch_access((char *)fname, W_OK) == 0
5950 )
5951#endif
5952 {
5953 ++retval;
5954 if (mch_isdir(fname))
5955 ++retval;
5956 }
5957 return retval;
5958}
5959#endif
Bram Moolenaar6bab4d12005-06-16 21:53:56 +00005960
5961/*
5962 * Print an error message with one or two "%s" and one or two string arguments.
5963 * This is not in message.c to avoid a warning for prototypes.
5964 */
5965 int
5966emsg3(s, a1, a2)
5967 char_u *s, *a1, *a2;
5968{
Bram Moolenaareb3593b2006-04-22 22:33:57 +00005969 if (emsg_not_now())
Bram Moolenaar6bab4d12005-06-16 21:53:56 +00005970 return TRUE; /* no error messages at the moment */
Bram Moolenaar6f192452007-11-08 19:49:02 +00005971#ifdef HAVE_STDARG_H
5972 vim_snprintf((char *)IObuff, IOSIZE, (char *)s, a1, a2);
5973#else
Bram Moolenaareb3593b2006-04-22 22:33:57 +00005974 vim_snprintf((char *)IObuff, IOSIZE, (char *)s, (long_u)a1, (long_u)a2);
Bram Moolenaar6f192452007-11-08 19:49:02 +00005975#endif
Bram Moolenaar6bab4d12005-06-16 21:53:56 +00005976 return emsg(IObuff);
5977}
5978
5979/*
5980 * Print an error message with one "%ld" and one long int argument.
5981 * This is not in message.c to avoid a warning for prototypes.
5982 */
5983 int
5984emsgn(s, n)
5985 char_u *s;
5986 long n;
5987{
Bram Moolenaareb3593b2006-04-22 22:33:57 +00005988 if (emsg_not_now())
Bram Moolenaar6bab4d12005-06-16 21:53:56 +00005989 return TRUE; /* no error messages at the moment */
5990 vim_snprintf((char *)IObuff, IOSIZE, (char *)s, n);
5991 return emsg(IObuff);
5992}