blob: fafe931e67300968091e80927047b315f0b58003 [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 {
Bram Moolenaar5a221812008-11-12 12:08:45 +0000876 free((char *)p); /* System is low... no go! */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000877 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 Moolenaarde0dfed2009-02-24 03:30:14 +00001013# if defined(FEAT_KEYMAP)
1014 do_cmdline_cmd((char_u *)"set keymap=");
1015#endif
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001016
1017# ifdef FEAT_TITLE
1018 free_titles();
1019# endif
1020# if defined(FEAT_SEARCHPATH)
1021 free_findfile();
1022# endif
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001023
1024 /* Obviously named calls. */
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001025# if defined(FEAT_AUTOCMD)
1026 free_all_autocmds();
1027# endif
1028 clear_termcodes();
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001029 free_all_options();
1030 free_all_marks();
1031 alist_clear(&global_alist);
1032 free_homedir();
1033 free_search_patterns();
1034 free_old_sub();
1035 free_last_insert();
1036 free_prev_shellcmd();
1037 free_regexp_stuff();
1038 free_tag_stuff();
1039 free_cd_dir();
Bram Moolenaarde0dfed2009-02-24 03:30:14 +00001040# ifdef FEAT_SIGNS
1041 free_signs();
1042# endif
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001043# ifdef FEAT_EVAL
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001044 set_expr_line(NULL);
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001045# endif
1046# ifdef FEAT_DIFF
Bram Moolenaar997fb4b2006-02-17 21:53:23 +00001047 diff_clear(curtab);
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001048# endif
Bram Moolenaara40ceaf2006-01-13 22:35:40 +00001049 clear_sb_text(); /* free any scrollback text */
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001050
1051 /* Free some global vars. */
1052 vim_free(username);
Bram Moolenaaraf92ee82007-10-07 13:45:08 +00001053# ifdef FEAT_CLIPBOARD
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001054 vim_free(clip_exclude_prog);
Bram Moolenaaraf92ee82007-10-07 13:45:08 +00001055# endif
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001056 vim_free(last_cmdline);
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001057# ifdef FEAT_CMDHIST
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001058 vim_free(new_last_cmdline);
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001059# endif
Bram Moolenaar238a5642006-02-21 22:12:05 +00001060 set_keep_msg(NULL, 0);
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001061 vim_free(ff_expand_buffer);
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001062
1063 /* Clear cmdline history. */
1064 p_hi = 0;
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001065# ifdef FEAT_CMDHIST
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001066 init_history();
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001067# endif
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001068
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001069#ifdef FEAT_QUICKFIX
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001070 {
1071 win_T *win;
1072
1073 qf_free_all(NULL);
1074 /* Free all location lists */
1075 FOR_ALL_WINDOWS(win)
1076 qf_free_all(win);
1077 }
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001078#endif
1079
1080 /* Close all script inputs. */
1081 close_all_scripts();
1082
1083#if defined(FEAT_WINDOWS)
1084 /* Destroy all windows. Must come before freeing buffers. */
1085 win_free_all();
1086#endif
1087
Bram Moolenaar2a329742008-04-01 12:53:43 +00001088 /* Free all buffers. Reset 'autochdir' to avoid accessing things that
1089 * were freed already. */
1090#ifdef FEAT_AUTOCHDIR
1091 p_acd = FALSE;
1092#endif
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001093 for (buf = firstbuf; buf != NULL; )
1094 {
1095 nextbuf = buf->b_next;
1096 close_buffer(NULL, buf, DOBUF_WIPE);
1097 if (buf_valid(buf))
1098 buf = nextbuf; /* didn't work, try next one */
1099 else
1100 buf = firstbuf;
1101 }
1102
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001103#ifdef FEAT_ARABIC
1104 free_cmdline_buf();
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001105#endif
1106
1107 /* Clear registers. */
1108 clear_registers();
1109 ResetRedobuff();
1110 ResetRedobuff();
1111
Bram Moolenaar85c79d32007-02-20 01:59:20 +00001112#if defined(FEAT_CLIENTSERVER) && defined(FEAT_X11)
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001113 vim_free(serverDelayedStartName);
1114#endif
1115
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001116 /* highlight info */
1117 free_highlight();
1118
Bram Moolenaar1e498f52005-06-26 22:29:44 +00001119 reset_last_sourcing();
1120
Bram Moolenaar7d47b6e2006-03-15 22:59:18 +00001121#ifdef FEAT_WINDOWS
Bram Moolenaar06a89a52006-04-29 22:01:03 +00001122 free_tabpage(first_tabpage);
1123 first_tabpage = NULL;
Bram Moolenaar7d47b6e2006-03-15 22:59:18 +00001124#endif
1125
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001126# ifdef UNIX
1127 /* Machine-specific free. */
1128 mch_free_mem();
1129# endif
1130
1131 /* message history */
1132 for (;;)
1133 if (delete_first_msg() == FAIL)
1134 break;
1135
1136# ifdef FEAT_EVAL
1137 eval_clear();
1138# endif
1139
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001140 free_termoptions();
1141
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001142 /* screenlines (can't display anything now!) */
1143 free_screenlines();
1144
1145#if defined(USE_XSMP)
1146 xsmp_close();
1147#endif
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001148#ifdef FEAT_GUI_GTK
1149 gui_mch_free_all();
1150#endif
1151 clear_hl_tables();
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001152
1153 vim_free(IObuff);
1154 vim_free(NameBuff);
1155}
1156#endif
1157
Bram Moolenaar071d4272004-06-13 20:20:40 +00001158/*
1159 * copy a string into newly allocated memory
1160 */
1161 char_u *
1162vim_strsave(string)
1163 char_u *string;
1164{
1165 char_u *p;
1166 unsigned len;
1167
1168 len = (unsigned)STRLEN(string) + 1;
1169 p = alloc(len);
1170 if (p != NULL)
1171 mch_memmove(p, string, (size_t)len);
1172 return p;
1173}
1174
1175 char_u *
1176vim_strnsave(string, len)
1177 char_u *string;
1178 int len;
1179{
1180 char_u *p;
1181
1182 p = alloc((unsigned)(len + 1));
1183 if (p != NULL)
1184 {
1185 STRNCPY(p, string, len);
1186 p[len] = NUL;
1187 }
1188 return p;
1189}
1190
Bram Moolenaar071d4272004-06-13 20:20:40 +00001191/*
1192 * Same as vim_strsave(), but any characters found in esc_chars are preceded
1193 * by a backslash.
1194 */
1195 char_u *
1196vim_strsave_escaped(string, esc_chars)
1197 char_u *string;
1198 char_u *esc_chars;
1199{
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001200 return vim_strsave_escaped_ext(string, esc_chars, '\\', FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001201}
1202
1203/*
1204 * Same as vim_strsave_escaped(), but when "bsl" is TRUE also escape
1205 * characters where rem_backslash() would remove the backslash.
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001206 * Escape the characters with "cc".
Bram Moolenaar071d4272004-06-13 20:20:40 +00001207 */
1208 char_u *
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001209vim_strsave_escaped_ext(string, esc_chars, cc, bsl)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001210 char_u *string;
1211 char_u *esc_chars;
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001212 int cc;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001213 int bsl;
1214{
1215 char_u *p;
1216 char_u *p2;
1217 char_u *escaped_string;
1218 unsigned length;
1219#ifdef FEAT_MBYTE
1220 int l;
1221#endif
1222
1223 /*
1224 * First count the number of backslashes required.
1225 * Then allocate the memory and insert them.
1226 */
1227 length = 1; /* count the trailing NUL */
1228 for (p = string; *p; p++)
1229 {
1230#ifdef FEAT_MBYTE
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001231 if (has_mbyte && (l = (*mb_ptr2len)(p)) > 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001232 {
1233 length += l; /* count a multibyte char */
1234 p += l - 1;
1235 continue;
1236 }
1237#endif
1238 if (vim_strchr(esc_chars, *p) != NULL || (bsl && rem_backslash(p)))
1239 ++length; /* count a backslash */
1240 ++length; /* count an ordinary char */
1241 }
1242 escaped_string = alloc(length);
1243 if (escaped_string != NULL)
1244 {
1245 p2 = escaped_string;
1246 for (p = string; *p; p++)
1247 {
1248#ifdef FEAT_MBYTE
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001249 if (has_mbyte && (l = (*mb_ptr2len)(p)) > 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001250 {
1251 mch_memmove(p2, p, (size_t)l);
1252 p2 += l;
1253 p += l - 1; /* skip multibyte char */
1254 continue;
1255 }
1256#endif
1257 if (vim_strchr(esc_chars, *p) != NULL || (bsl && rem_backslash(p)))
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001258 *p2++ = cc;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001259 *p2++ = *p;
1260 }
1261 *p2 = NUL;
1262 }
1263 return escaped_string;
1264}
1265
Bram Moolenaar7693ec62008-07-24 18:29:37 +00001266/*
1267 * Return TRUE when 'shell' has "csh" in the tail.
1268 */
1269 int
1270csh_like_shell()
1271{
1272 return (strstr((char *)gettail(p_sh), "csh") != NULL);
1273}
Bram Moolenaar7693ec62008-07-24 18:29:37 +00001274
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001275/*
1276 * Escape "string" for use as a shell argument with system().
1277 * This uses single quotes, except when we know we need to use double qoutes
1278 * (MS-DOS and MS-Windows without 'shellslash' set).
Bram Moolenaarffd66c42008-07-16 20:43:37 +00001279 * Escape a newline, depending on the 'shell' option.
1280 * When "do_special" is TRUE also replace "!", "%", "#" and things starting
1281 * with "<" like "<cfile>".
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001282 * Returns the result in allocated memory, NULL if we have run out.
1283 */
1284 char_u *
Bram Moolenaar05bb9532008-07-04 09:44:11 +00001285vim_strsave_shellescape(string, do_special)
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001286 char_u *string;
Bram Moolenaar05bb9532008-07-04 09:44:11 +00001287 int do_special;
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001288{
1289 unsigned length;
1290 char_u *p;
1291 char_u *d;
1292 char_u *escaped_string;
Bram Moolenaar05bb9532008-07-04 09:44:11 +00001293 int l;
Bram Moolenaarffd66c42008-07-16 20:43:37 +00001294 int csh_like;
1295
1296 /* Only csh and similar shells expand '!' within single quotes. For sh and
1297 * the like we must not put a backslash before it, it will be taken
1298 * literally. If do_special is set the '!' will be escaped twice.
1299 * Csh also needs to have "\n" escaped twice when do_special is set. */
Bram Moolenaar7693ec62008-07-24 18:29:37 +00001300 csh_like = csh_like_shell();
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001301
1302 /* First count the number of extra bytes required. */
Bram Moolenaar77f66d62007-02-20 02:16:18 +00001303 length = (unsigned)STRLEN(string) + 3; /* two quotes and a trailing NUL */
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001304 for (p = string; *p != NUL; mb_ptr_adv(p))
1305 {
1306# if defined(WIN32) || defined(WIN16) || defined(DOS)
1307 if (!p_ssl)
1308 {
1309 if (*p == '"')
1310 ++length; /* " -> "" */
1311 }
1312 else
1313# endif
1314 if (*p == '\'')
1315 length += 3; /* ' => '\'' */
Bram Moolenaarffd66c42008-07-16 20:43:37 +00001316 if (*p == '\n' || (*p == '!' && (csh_like || do_special)))
1317 {
1318 ++length; /* insert backslash */
1319 if (csh_like && do_special)
1320 ++length; /* insert backslash */
1321 }
Bram Moolenaar05bb9532008-07-04 09:44:11 +00001322 if (do_special && find_cmdline_var(p, &l) >= 0)
1323 {
1324 ++length; /* insert backslash */
1325 p += l - 1;
1326 }
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001327 }
1328
1329 /* Allocate memory for the result and fill it. */
1330 escaped_string = alloc(length);
1331 if (escaped_string != NULL)
1332 {
1333 d = escaped_string;
1334
1335 /* add opening quote */
1336# if defined(WIN32) || defined(WIN16) || defined(DOS)
1337 if (!p_ssl)
1338 *d++ = '"';
1339 else
1340# endif
1341 *d++ = '\'';
1342
1343 for (p = string; *p != NUL; )
1344 {
1345# if defined(WIN32) || defined(WIN16) || defined(DOS)
1346 if (!p_ssl)
1347 {
1348 if (*p == '"')
1349 {
1350 *d++ = '"';
1351 *d++ = '"';
1352 ++p;
1353 continue;
1354 }
1355 }
1356 else
1357# endif
1358 if (*p == '\'')
1359 {
Bram Moolenaar05bb9532008-07-04 09:44:11 +00001360 *d++ = '\'';
1361 *d++ = '\\';
1362 *d++ = '\'';
1363 *d++ = '\'';
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001364 ++p;
1365 continue;
1366 }
Bram Moolenaarffd66c42008-07-16 20:43:37 +00001367 if (*p == '\n' || (*p == '!' && (csh_like || do_special)))
1368 {
1369 *d++ = '\\';
1370 if (csh_like && do_special)
1371 *d++ = '\\';
1372 *d++ = *p++;
1373 continue;
1374 }
Bram Moolenaar05bb9532008-07-04 09:44:11 +00001375 if (do_special && find_cmdline_var(p, &l) >= 0)
1376 {
1377 *d++ = '\\'; /* insert backslash */
1378 while (--l >= 0) /* copy the var */
1379 *d++ = *p++;
1380 }
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001381
1382 MB_COPY_CHAR(p, d);
1383 }
1384
1385 /* add terminating quote and finish with a NUL */
1386# if defined(WIN32) || defined(WIN16) || defined(DOS)
1387 if (!p_ssl)
1388 *d++ = '"';
1389 else
1390# endif
1391 *d++ = '\'';
1392 *d = NUL;
1393 }
1394
1395 return escaped_string;
1396}
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001397
Bram Moolenaar071d4272004-06-13 20:20:40 +00001398/*
1399 * Like vim_strsave(), but make all characters uppercase.
1400 * This uses ASCII lower-to-upper case translation, language independent.
1401 */
1402 char_u *
1403vim_strsave_up(string)
1404 char_u *string;
1405{
1406 char_u *p1;
1407
1408 p1 = vim_strsave(string);
1409 vim_strup(p1);
1410 return p1;
1411}
1412
1413/*
1414 * Like vim_strnsave(), but make all characters uppercase.
1415 * This uses ASCII lower-to-upper case translation, language independent.
1416 */
1417 char_u *
1418vim_strnsave_up(string, len)
1419 char_u *string;
1420 int len;
1421{
1422 char_u *p1;
1423
1424 p1 = vim_strnsave(string, len);
1425 vim_strup(p1);
1426 return p1;
1427}
1428
1429/*
1430 * ASCII lower-to-upper case translation, language independent.
1431 */
1432 void
1433vim_strup(p)
1434 char_u *p;
1435{
1436 char_u *p2;
1437 int c;
1438
1439 if (p != NULL)
1440 {
1441 p2 = p;
1442 while ((c = *p2) != NUL)
1443#ifdef EBCDIC
1444 *p2++ = isalpha(c) ? toupper(c) : c;
1445#else
1446 *p2++ = (c < 'a' || c > 'z') ? c : (c - 0x20);
1447#endif
1448 }
1449}
1450
Bram Moolenaarc4956c82006-03-12 21:58:43 +00001451#if defined(FEAT_EVAL) || defined(FEAT_SPELL) || defined(PROTO)
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001452/*
1453 * Make string "s" all upper-case and return it in allocated memory.
1454 * Handles multi-byte characters as well as possible.
1455 * Returns NULL when out of memory.
1456 */
1457 char_u *
1458strup_save(orig)
1459 char_u *orig;
1460{
1461 char_u *p;
1462 char_u *res;
1463
1464 res = p = vim_strsave(orig);
1465
1466 if (res != NULL)
1467 while (*p != NUL)
1468 {
1469# ifdef FEAT_MBYTE
1470 int l;
1471
1472 if (enc_utf8)
1473 {
1474 int c, uc;
1475 int nl;
1476 char_u *s;
1477
1478 c = utf_ptr2char(p);
1479 uc = utf_toupper(c);
1480
1481 /* Reallocate string when byte count changes. This is rare,
1482 * thus it's OK to do another malloc()/free(). */
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001483 l = utf_ptr2len(p);
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001484 nl = utf_char2len(uc);
1485 if (nl != l)
1486 {
1487 s = alloc((unsigned)STRLEN(res) + 1 + nl - l);
1488 if (s == NULL)
1489 break;
1490 mch_memmove(s, res, p - res);
1491 STRCPY(s + (p - res) + nl, p + l);
1492 p = s + (p - res);
1493 vim_free(res);
1494 res = s;
1495 }
1496
1497 utf_char2bytes(uc, p);
1498 p += nl;
1499 }
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001500 else if (has_mbyte && (l = (*mb_ptr2len)(p)) > 1)
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001501 p += l; /* skip multi-byte character */
1502 else
1503# endif
1504 {
1505 *p = TOUPPER_LOC(*p); /* note that toupper() can be a macro */
1506 p++;
1507 }
1508 }
1509
1510 return res;
1511}
1512#endif
1513
Bram Moolenaar071d4272004-06-13 20:20:40 +00001514/*
1515 * copy a space a number of times
1516 */
1517 void
1518copy_spaces(ptr, count)
1519 char_u *ptr;
1520 size_t count;
1521{
1522 size_t i = count;
1523 char_u *p = ptr;
1524
1525 while (i--)
1526 *p++ = ' ';
1527}
1528
1529#if defined(FEAT_VISUALEXTRA) || defined(PROTO)
1530/*
1531 * Copy a character a number of times.
1532 * Does not work for multi-byte charactes!
1533 */
1534 void
1535copy_chars(ptr, count, c)
1536 char_u *ptr;
1537 size_t count;
1538 int c;
1539{
1540 size_t i = count;
1541 char_u *p = ptr;
1542
1543 while (i--)
1544 *p++ = c;
1545}
1546#endif
1547
1548/*
1549 * delete spaces at the end of a string
1550 */
1551 void
1552del_trailing_spaces(ptr)
1553 char_u *ptr;
1554{
1555 char_u *q;
1556
1557 q = ptr + STRLEN(ptr);
1558 while (--q > ptr && vim_iswhite(q[0]) && q[-1] != '\\' && q[-1] != Ctrl_V)
1559 *q = NUL;
1560}
1561
1562/*
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001563 * Like strncpy(), but always terminate the result with one NUL.
Bram Moolenaard042c562005-06-30 22:04:15 +00001564 * "to" must be "len + 1" long!
Bram Moolenaar071d4272004-06-13 20:20:40 +00001565 */
1566 void
1567vim_strncpy(to, from, len)
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001568 char_u *to;
1569 char_u *from;
Bram Moolenaarbbebc852005-07-18 21:47:53 +00001570 size_t len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001571{
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001572 STRNCPY(to, from, len);
1573 to[len] = NUL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001574}
1575
1576/*
1577 * Isolate one part of a string option where parts are separated with
1578 * "sep_chars".
Bram Moolenaar83bab712005-08-01 21:58:57 +00001579 * The part is copied into "buf[maxlen]".
Bram Moolenaar071d4272004-06-13 20:20:40 +00001580 * "*option" is advanced to the next part.
1581 * The length is returned.
1582 */
1583 int
1584copy_option_part(option, buf, maxlen, sep_chars)
1585 char_u **option;
1586 char_u *buf;
1587 int maxlen;
1588 char *sep_chars;
1589{
1590 int len = 0;
1591 char_u *p = *option;
1592
1593 /* skip '.' at start of option part, for 'suffixes' */
1594 if (*p == '.')
1595 buf[len++] = *p++;
1596 while (*p != NUL && vim_strchr((char_u *)sep_chars, *p) == NULL)
1597 {
1598 /*
1599 * Skip backslash before a separator character and space.
1600 */
1601 if (p[0] == '\\' && vim_strchr((char_u *)sep_chars, p[1]) != NULL)
1602 ++p;
1603 if (len < maxlen - 1)
1604 buf[len++] = *p;
1605 ++p;
1606 }
1607 buf[len] = NUL;
1608
1609 if (*p != NUL && *p != ',') /* skip non-standard separator */
1610 ++p;
1611 p = skip_to_option_part(p); /* p points to next file name */
1612
1613 *option = p;
1614 return len;
1615}
1616
1617/*
Bram Moolenaar4770d092006-01-12 23:22:24 +00001618 * Replacement for free() that ignores NULL pointers.
1619 * Also skip free() when exiting for sure, this helps when we caught a deadly
1620 * signal that was caused by a crash in free().
Bram Moolenaar071d4272004-06-13 20:20:40 +00001621 */
1622 void
1623vim_free(x)
1624 void *x;
1625{
Bram Moolenaar4770d092006-01-12 23:22:24 +00001626 if (x != NULL && !really_exiting)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001627 {
1628#ifdef MEM_PROFILE
1629 mem_pre_free(&x);
1630#endif
1631 free(x);
1632 }
1633}
1634
1635#ifndef HAVE_MEMSET
1636 void *
1637vim_memset(ptr, c, size)
1638 void *ptr;
1639 int c;
1640 size_t size;
1641{
1642 char *p = ptr;
1643
1644 while (size-- > 0)
1645 *p++ = c;
1646 return ptr;
1647}
1648#endif
1649
1650#ifdef VIM_MEMCMP
1651/*
1652 * Return zero when "b1" and "b2" are the same for "len" bytes.
1653 * Return non-zero otherwise.
1654 */
1655 int
1656vim_memcmp(b1, b2, len)
1657 void *b1;
1658 void *b2;
1659 size_t len;
1660{
1661 char_u *p1 = (char_u *)b1, *p2 = (char_u *)b2;
1662
1663 for ( ; len > 0; --len)
1664 {
1665 if (*p1 != *p2)
1666 return 1;
1667 ++p1;
1668 ++p2;
1669 }
1670 return 0;
1671}
1672#endif
1673
1674#ifdef VIM_MEMMOVE
1675/*
1676 * Version of memmove() that handles overlapping source and destination.
1677 * For systems that don't have a function that is guaranteed to do that (SYSV).
1678 */
1679 void
1680mch_memmove(dst_arg, src_arg, len)
1681 void *src_arg, *dst_arg;
1682 size_t len;
1683{
1684 /*
1685 * A void doesn't have a size, we use char pointers.
1686 */
1687 char *dst = dst_arg, *src = src_arg;
1688
1689 /* overlap, copy backwards */
1690 if (dst > src && dst < src + len)
1691 {
1692 src += len;
1693 dst += len;
1694 while (len-- > 0)
1695 *--dst = *--src;
1696 }
1697 else /* copy forwards */
1698 while (len-- > 0)
1699 *dst++ = *src++;
1700}
1701#endif
1702
1703#if (!defined(HAVE_STRCASECMP) && !defined(HAVE_STRICMP)) || defined(PROTO)
1704/*
1705 * Compare two strings, ignoring case, using current locale.
1706 * Doesn't work for multi-byte characters.
1707 * return 0 for match, < 0 for smaller, > 0 for bigger
1708 */
1709 int
1710vim_stricmp(s1, s2)
1711 char *s1;
1712 char *s2;
1713{
1714 int i;
1715
1716 for (;;)
1717 {
1718 i = (int)TOLOWER_LOC(*s1) - (int)TOLOWER_LOC(*s2);
1719 if (i != 0)
1720 return i; /* this character different */
1721 if (*s1 == NUL)
1722 break; /* strings match until NUL */
1723 ++s1;
1724 ++s2;
1725 }
1726 return 0; /* strings match */
1727}
1728#endif
1729
1730#if (!defined(HAVE_STRNCASECMP) && !defined(HAVE_STRNICMP)) || defined(PROTO)
1731/*
1732 * Compare two strings, for length "len", ignoring case, using current locale.
1733 * Doesn't work for multi-byte characters.
1734 * return 0 for match, < 0 for smaller, > 0 for bigger
1735 */
1736 int
1737vim_strnicmp(s1, s2, len)
1738 char *s1;
1739 char *s2;
1740 size_t len;
1741{
1742 int i;
1743
1744 while (len > 0)
1745 {
1746 i = (int)TOLOWER_LOC(*s1) - (int)TOLOWER_LOC(*s2);
1747 if (i != 0)
1748 return i; /* this character different */
1749 if (*s1 == NUL)
1750 break; /* strings match until NUL */
1751 ++s1;
1752 ++s2;
1753 --len;
1754 }
1755 return 0; /* strings match */
1756}
1757#endif
1758
1759#if 0 /* currently not used */
1760/*
1761 * Check if string "s2" appears somewhere in "s1" while ignoring case.
1762 * Return NULL if not, a pointer to the first occurrence if it does.
1763 */
1764 char_u *
1765vim_stristr(s1, s2)
1766 char_u *s1;
1767 char_u *s2;
1768{
1769 char_u *p;
1770 int len = STRLEN(s2);
1771 char_u *end = s1 + STRLEN(s1) - len;
1772
1773 for (p = s1; p <= end; ++p)
1774 if (STRNICMP(p, s2, len) == 0)
1775 return p;
1776 return NULL;
1777}
1778#endif
1779
1780/*
1781 * Version of strchr() and strrchr() that handle unsigned char strings
Bram Moolenaar05159a02005-02-26 23:04:13 +00001782 * with characters from 128 to 255 correctly. It also doesn't return a
1783 * pointer to the NUL at the end of the string.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001784 */
1785 char_u *
1786vim_strchr(string, c)
1787 char_u *string;
1788 int c;
1789{
1790 char_u *p;
1791 int b;
1792
1793 p = string;
1794#ifdef FEAT_MBYTE
1795 if (enc_utf8 && c >= 0x80)
1796 {
1797 while (*p != NUL)
1798 {
1799 if (utf_ptr2char(p) == c)
1800 return p;
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001801 p += (*mb_ptr2len)(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001802 }
1803 return NULL;
1804 }
1805 if (enc_dbcs != 0 && c > 255)
1806 {
1807 int n2 = c & 0xff;
1808
1809 c = ((unsigned)c >> 8) & 0xff;
1810 while ((b = *p) != NUL)
1811 {
1812 if (b == c && p[1] == n2)
1813 return p;
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001814 p += (*mb_ptr2len)(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001815 }
1816 return NULL;
1817 }
1818 if (has_mbyte)
1819 {
1820 while ((b = *p) != NUL)
1821 {
1822 if (b == c)
1823 return p;
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001824 p += (*mb_ptr2len)(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001825 }
1826 return NULL;
1827 }
1828#endif
1829 while ((b = *p) != NUL)
1830 {
1831 if (b == c)
1832 return p;
1833 ++p;
1834 }
1835 return NULL;
1836}
1837
1838/*
Bram Moolenaar05159a02005-02-26 23:04:13 +00001839 * Version of strchr() that only works for bytes and handles unsigned char
1840 * strings with characters above 128 correctly. It also doesn't return a
1841 * pointer to the NUL at the end of the string.
1842 */
1843 char_u *
1844vim_strbyte(string, c)
1845 char_u *string;
1846 int c;
1847{
1848 char_u *p = string;
1849
1850 while (*p != NUL)
1851 {
1852 if (*p == c)
1853 return p;
1854 ++p;
1855 }
1856 return NULL;
1857}
1858
1859/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00001860 * Search for last occurrence of "c" in "string".
Bram Moolenaarac6e65f2005-08-29 22:25:38 +00001861 * Return NULL if not found.
Bram Moolenaar05159a02005-02-26 23:04:13 +00001862 * Does not handle multi-byte char for "c"!
Bram Moolenaar071d4272004-06-13 20:20:40 +00001863 */
1864 char_u *
1865vim_strrchr(string, c)
1866 char_u *string;
1867 int c;
1868{
1869 char_u *retval = NULL;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001870 char_u *p = string;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001871
Bram Moolenaar05159a02005-02-26 23:04:13 +00001872 while (*p)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001873 {
Bram Moolenaar05159a02005-02-26 23:04:13 +00001874 if (*p == c)
1875 retval = p;
1876 mb_ptr_adv(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001877 }
1878 return retval;
1879}
1880
1881/*
1882 * Vim's version of strpbrk(), in case it's missing.
1883 * Don't generate a prototype for this, causes problems when it's not used.
1884 */
1885#ifndef PROTO
1886# ifndef HAVE_STRPBRK
1887# ifdef vim_strpbrk
1888# undef vim_strpbrk
1889# endif
1890 char_u *
1891vim_strpbrk(s, charset)
1892 char_u *s;
1893 char_u *charset;
1894{
1895 while (*s)
1896 {
1897 if (vim_strchr(charset, *s) != NULL)
1898 return s;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001899 mb_ptr_adv(s);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001900 }
1901 return NULL;
1902}
1903# endif
1904#endif
1905
1906/*
1907 * Vim has its own isspace() function, because on some machines isspace()
1908 * can't handle characters above 128.
1909 */
1910 int
1911vim_isspace(x)
1912 int x;
1913{
1914 return ((x >= 9 && x <= 13) || x == ' ');
1915}
1916
1917/************************************************************************
Bram Moolenaar383f9bc2005-01-19 22:18:32 +00001918 * Functions for handling growing arrays.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001919 */
1920
1921/*
1922 * Clear an allocated growing array.
1923 */
1924 void
1925ga_clear(gap)
1926 garray_T *gap;
1927{
1928 vim_free(gap->ga_data);
1929 ga_init(gap);
1930}
1931
1932/*
1933 * Clear a growing array that contains a list of strings.
1934 */
1935 void
1936ga_clear_strings(gap)
1937 garray_T *gap;
1938{
1939 int i;
1940
1941 for (i = 0; i < gap->ga_len; ++i)
1942 vim_free(((char_u **)(gap->ga_data))[i]);
1943 ga_clear(gap);
1944}
1945
1946/*
1947 * Initialize a growing array. Don't forget to set ga_itemsize and
1948 * ga_growsize! Or use ga_init2().
1949 */
1950 void
1951ga_init(gap)
1952 garray_T *gap;
1953{
1954 gap->ga_data = NULL;
Bram Moolenaar86b68352004-12-27 21:59:20 +00001955 gap->ga_maxlen = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001956 gap->ga_len = 0;
1957}
1958
1959 void
1960ga_init2(gap, itemsize, growsize)
1961 garray_T *gap;
1962 int itemsize;
1963 int growsize;
1964{
1965 ga_init(gap);
1966 gap->ga_itemsize = itemsize;
1967 gap->ga_growsize = growsize;
1968}
1969
1970/*
1971 * Make room in growing array "gap" for at least "n" items.
1972 * Return FAIL for failure, OK otherwise.
1973 */
1974 int
1975ga_grow(gap, n)
1976 garray_T *gap;
1977 int n;
1978{
1979 size_t len;
1980 char_u *pp;
1981
Bram Moolenaar86b68352004-12-27 21:59:20 +00001982 if (gap->ga_maxlen - gap->ga_len < n)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001983 {
1984 if (n < gap->ga_growsize)
1985 n = gap->ga_growsize;
1986 len = gap->ga_itemsize * (gap->ga_len + n);
1987 pp = alloc_clear((unsigned)len);
1988 if (pp == NULL)
1989 return FAIL;
Bram Moolenaar86b68352004-12-27 21:59:20 +00001990 gap->ga_maxlen = gap->ga_len + n;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001991 if (gap->ga_data != NULL)
1992 {
1993 mch_memmove(pp, gap->ga_data,
1994 (size_t)(gap->ga_itemsize * gap->ga_len));
1995 vim_free(gap->ga_data);
1996 }
1997 gap->ga_data = pp;
1998 }
1999 return OK;
2000}
2001
2002/*
2003 * Concatenate a string to a growarray which contains characters.
2004 * Note: Does NOT copy the NUL at the end!
2005 */
2006 void
2007ga_concat(gap, s)
2008 garray_T *gap;
2009 char_u *s;
2010{
2011 int len = (int)STRLEN(s);
2012
2013 if (ga_grow(gap, len) == OK)
2014 {
2015 mch_memmove((char *)gap->ga_data + gap->ga_len, s, (size_t)len);
2016 gap->ga_len += len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002017 }
2018}
2019
2020/*
2021 * Append one byte to a growarray which contains bytes.
2022 */
2023 void
2024ga_append(gap, c)
2025 garray_T *gap;
2026 int c;
2027{
2028 if (ga_grow(gap, 1) == OK)
2029 {
2030 *((char *)gap->ga_data + gap->ga_len) = c;
2031 ++gap->ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002032 }
2033}
2034
2035/************************************************************************
2036 * functions that use lookup tables for various things, generally to do with
2037 * special key codes.
2038 */
2039
2040/*
2041 * Some useful tables.
2042 */
2043
2044static struct modmasktable
2045{
2046 short mod_mask; /* Bit-mask for particular key modifier */
2047 short mod_flag; /* Bit(s) for particular key modifier */
2048 char_u name; /* Single letter name of modifier */
2049} mod_mask_table[] =
2050{
2051 {MOD_MASK_ALT, MOD_MASK_ALT, (char_u)'M'},
Bram Moolenaar19a09a12005-03-04 23:39:37 +00002052 {MOD_MASK_META, MOD_MASK_META, (char_u)'T'},
Bram Moolenaar071d4272004-06-13 20:20:40 +00002053 {MOD_MASK_CTRL, MOD_MASK_CTRL, (char_u)'C'},
2054 {MOD_MASK_SHIFT, MOD_MASK_SHIFT, (char_u)'S'},
2055 {MOD_MASK_MULTI_CLICK, MOD_MASK_2CLICK, (char_u)'2'},
2056 {MOD_MASK_MULTI_CLICK, MOD_MASK_3CLICK, (char_u)'3'},
2057 {MOD_MASK_MULTI_CLICK, MOD_MASK_4CLICK, (char_u)'4'},
2058#ifdef MACOS
2059 {MOD_MASK_CMD, MOD_MASK_CMD, (char_u)'D'},
2060#endif
2061 /* 'A' must be the last one */
2062 {MOD_MASK_ALT, MOD_MASK_ALT, (char_u)'A'},
2063 {0, 0, NUL}
2064};
2065
2066/*
2067 * Shifted key terminal codes and their unshifted equivalent.
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00002068 * Don't add mouse codes here, they are handled separately!
Bram Moolenaar071d4272004-06-13 20:20:40 +00002069 */
2070#define MOD_KEYS_ENTRY_SIZE 5
2071
2072static char_u modifier_keys_table[] =
2073{
2074/* mod mask with modifier without modifier */
2075 MOD_MASK_SHIFT, '&', '9', '@', '1', /* begin */
2076 MOD_MASK_SHIFT, '&', '0', '@', '2', /* cancel */
2077 MOD_MASK_SHIFT, '*', '1', '@', '4', /* command */
2078 MOD_MASK_SHIFT, '*', '2', '@', '5', /* copy */
2079 MOD_MASK_SHIFT, '*', '3', '@', '6', /* create */
2080 MOD_MASK_SHIFT, '*', '4', 'k', 'D', /* delete char */
2081 MOD_MASK_SHIFT, '*', '5', 'k', 'L', /* delete line */
2082 MOD_MASK_SHIFT, '*', '7', '@', '7', /* end */
2083 MOD_MASK_CTRL, KS_EXTRA, (int)KE_C_END, '@', '7', /* end */
2084 MOD_MASK_SHIFT, '*', '9', '@', '9', /* exit */
2085 MOD_MASK_SHIFT, '*', '0', '@', '0', /* find */
2086 MOD_MASK_SHIFT, '#', '1', '%', '1', /* help */
2087 MOD_MASK_SHIFT, '#', '2', 'k', 'h', /* home */
2088 MOD_MASK_CTRL, KS_EXTRA, (int)KE_C_HOME, 'k', 'h', /* home */
2089 MOD_MASK_SHIFT, '#', '3', 'k', 'I', /* insert */
2090 MOD_MASK_SHIFT, '#', '4', 'k', 'l', /* left arrow */
2091 MOD_MASK_CTRL, KS_EXTRA, (int)KE_C_LEFT, 'k', 'l', /* left arrow */
2092 MOD_MASK_SHIFT, '%', 'a', '%', '3', /* message */
2093 MOD_MASK_SHIFT, '%', 'b', '%', '4', /* move */
2094 MOD_MASK_SHIFT, '%', 'c', '%', '5', /* next */
2095 MOD_MASK_SHIFT, '%', 'd', '%', '7', /* options */
2096 MOD_MASK_SHIFT, '%', 'e', '%', '8', /* previous */
2097 MOD_MASK_SHIFT, '%', 'f', '%', '9', /* print */
2098 MOD_MASK_SHIFT, '%', 'g', '%', '0', /* redo */
2099 MOD_MASK_SHIFT, '%', 'h', '&', '3', /* replace */
2100 MOD_MASK_SHIFT, '%', 'i', 'k', 'r', /* right arr. */
2101 MOD_MASK_CTRL, KS_EXTRA, (int)KE_C_RIGHT, 'k', 'r', /* right arr. */
2102 MOD_MASK_SHIFT, '%', 'j', '&', '5', /* resume */
2103 MOD_MASK_SHIFT, '!', '1', '&', '6', /* save */
2104 MOD_MASK_SHIFT, '!', '2', '&', '7', /* suspend */
2105 MOD_MASK_SHIFT, '!', '3', '&', '8', /* undo */
2106 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_UP, 'k', 'u', /* up arrow */
2107 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_DOWN, 'k', 'd', /* down arrow */
2108
2109 /* vt100 F1 */
2110 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_XF1, KS_EXTRA, (int)KE_XF1,
2111 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_XF2, KS_EXTRA, (int)KE_XF2,
2112 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_XF3, KS_EXTRA, (int)KE_XF3,
2113 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_XF4, KS_EXTRA, (int)KE_XF4,
2114
2115 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F1, 'k', '1', /* F1 */
2116 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F2, 'k', '2',
2117 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F3, 'k', '3',
2118 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F4, 'k', '4',
2119 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F5, 'k', '5',
2120 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F6, 'k', '6',
2121 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F7, 'k', '7',
2122 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F8, 'k', '8',
2123 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F9, 'k', '9',
2124 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F10, 'k', ';', /* F10 */
2125
2126 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F11, 'F', '1',
2127 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F12, 'F', '2',
2128 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F13, 'F', '3',
2129 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F14, 'F', '4',
2130 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F15, 'F', '5',
2131 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F16, 'F', '6',
2132 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F17, 'F', '7',
2133 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F18, 'F', '8',
2134 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F19, 'F', '9',
2135 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F20, 'F', 'A',
2136
2137 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F21, 'F', 'B',
2138 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F22, 'F', 'C',
2139 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F23, 'F', 'D',
2140 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F24, 'F', 'E',
2141 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F25, 'F', 'F',
2142 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F26, 'F', 'G',
2143 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F27, 'F', 'H',
2144 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F28, 'F', 'I',
2145 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F29, 'F', 'J',
2146 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F30, 'F', 'K',
2147
2148 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F31, 'F', 'L',
2149 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F32, 'F', 'M',
2150 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F33, 'F', 'N',
2151 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F34, 'F', 'O',
2152 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F35, 'F', 'P',
2153 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F36, 'F', 'Q',
2154 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F37, 'F', 'R',
2155
2156 /* TAB pseudo code*/
2157 MOD_MASK_SHIFT, 'k', 'B', KS_EXTRA, (int)KE_TAB,
2158
2159 NUL
2160};
2161
2162static struct key_name_entry
2163{
2164 int key; /* Special key code or ascii value */
2165 char_u *name; /* Name of key */
2166} key_names_table[] =
2167{
2168 {' ', (char_u *)"Space"},
2169 {TAB, (char_u *)"Tab"},
2170 {K_TAB, (char_u *)"Tab"},
2171 {NL, (char_u *)"NL"},
2172 {NL, (char_u *)"NewLine"}, /* Alternative name */
2173 {NL, (char_u *)"LineFeed"}, /* Alternative name */
2174 {NL, (char_u *)"LF"}, /* Alternative name */
2175 {CAR, (char_u *)"CR"},
2176 {CAR, (char_u *)"Return"}, /* Alternative name */
2177 {CAR, (char_u *)"Enter"}, /* Alternative name */
2178 {K_BS, (char_u *)"BS"},
2179 {K_BS, (char_u *)"BackSpace"}, /* Alternative name */
2180 {ESC, (char_u *)"Esc"},
2181 {CSI, (char_u *)"CSI"},
2182 {K_CSI, (char_u *)"xCSI"},
2183 {'|', (char_u *)"Bar"},
2184 {'\\', (char_u *)"Bslash"},
2185 {K_DEL, (char_u *)"Del"},
2186 {K_DEL, (char_u *)"Delete"}, /* Alternative name */
2187 {K_KDEL, (char_u *)"kDel"},
2188 {K_UP, (char_u *)"Up"},
2189 {K_DOWN, (char_u *)"Down"},
2190 {K_LEFT, (char_u *)"Left"},
2191 {K_RIGHT, (char_u *)"Right"},
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002192 {K_XUP, (char_u *)"xUp"},
2193 {K_XDOWN, (char_u *)"xDown"},
2194 {K_XLEFT, (char_u *)"xLeft"},
2195 {K_XRIGHT, (char_u *)"xRight"},
Bram Moolenaar071d4272004-06-13 20:20:40 +00002196
2197 {K_F1, (char_u *)"F1"},
2198 {K_F2, (char_u *)"F2"},
2199 {K_F3, (char_u *)"F3"},
2200 {K_F4, (char_u *)"F4"},
2201 {K_F5, (char_u *)"F5"},
2202 {K_F6, (char_u *)"F6"},
2203 {K_F7, (char_u *)"F7"},
2204 {K_F8, (char_u *)"F8"},
2205 {K_F9, (char_u *)"F9"},
2206 {K_F10, (char_u *)"F10"},
2207
2208 {K_F11, (char_u *)"F11"},
2209 {K_F12, (char_u *)"F12"},
2210 {K_F13, (char_u *)"F13"},
2211 {K_F14, (char_u *)"F14"},
2212 {K_F15, (char_u *)"F15"},
2213 {K_F16, (char_u *)"F16"},
2214 {K_F17, (char_u *)"F17"},
2215 {K_F18, (char_u *)"F18"},
2216 {K_F19, (char_u *)"F19"},
2217 {K_F20, (char_u *)"F20"},
2218
2219 {K_F21, (char_u *)"F21"},
2220 {K_F22, (char_u *)"F22"},
2221 {K_F23, (char_u *)"F23"},
2222 {K_F24, (char_u *)"F24"},
2223 {K_F25, (char_u *)"F25"},
2224 {K_F26, (char_u *)"F26"},
2225 {K_F27, (char_u *)"F27"},
2226 {K_F28, (char_u *)"F28"},
2227 {K_F29, (char_u *)"F29"},
2228 {K_F30, (char_u *)"F30"},
2229
2230 {K_F31, (char_u *)"F31"},
2231 {K_F32, (char_u *)"F32"},
2232 {K_F33, (char_u *)"F33"},
2233 {K_F34, (char_u *)"F34"},
2234 {K_F35, (char_u *)"F35"},
2235 {K_F36, (char_u *)"F36"},
2236 {K_F37, (char_u *)"F37"},
2237
2238 {K_XF1, (char_u *)"xF1"},
2239 {K_XF2, (char_u *)"xF2"},
2240 {K_XF3, (char_u *)"xF3"},
2241 {K_XF4, (char_u *)"xF4"},
2242
2243 {K_HELP, (char_u *)"Help"},
2244 {K_UNDO, (char_u *)"Undo"},
2245 {K_INS, (char_u *)"Insert"},
2246 {K_INS, (char_u *)"Ins"}, /* Alternative name */
2247 {K_KINS, (char_u *)"kInsert"},
2248 {K_HOME, (char_u *)"Home"},
2249 {K_KHOME, (char_u *)"kHome"},
2250 {K_XHOME, (char_u *)"xHome"},
Bram Moolenaar68b76a62005-03-25 21:53:48 +00002251 {K_ZHOME, (char_u *)"zHome"},
Bram Moolenaar071d4272004-06-13 20:20:40 +00002252 {K_END, (char_u *)"End"},
2253 {K_KEND, (char_u *)"kEnd"},
2254 {K_XEND, (char_u *)"xEnd"},
Bram Moolenaar68b76a62005-03-25 21:53:48 +00002255 {K_ZEND, (char_u *)"zEnd"},
Bram Moolenaar071d4272004-06-13 20:20:40 +00002256 {K_PAGEUP, (char_u *)"PageUp"},
2257 {K_PAGEDOWN, (char_u *)"PageDown"},
2258 {K_KPAGEUP, (char_u *)"kPageUp"},
2259 {K_KPAGEDOWN, (char_u *)"kPageDown"},
2260
2261 {K_KPLUS, (char_u *)"kPlus"},
2262 {K_KMINUS, (char_u *)"kMinus"},
2263 {K_KDIVIDE, (char_u *)"kDivide"},
2264 {K_KMULTIPLY, (char_u *)"kMultiply"},
2265 {K_KENTER, (char_u *)"kEnter"},
2266 {K_KPOINT, (char_u *)"kPoint"},
2267
2268 {K_K0, (char_u *)"k0"},
2269 {K_K1, (char_u *)"k1"},
2270 {K_K2, (char_u *)"k2"},
2271 {K_K3, (char_u *)"k3"},
2272 {K_K4, (char_u *)"k4"},
2273 {K_K5, (char_u *)"k5"},
2274 {K_K6, (char_u *)"k6"},
2275 {K_K7, (char_u *)"k7"},
2276 {K_K8, (char_u *)"k8"},
2277 {K_K9, (char_u *)"k9"},
2278
2279 {'<', (char_u *)"lt"},
2280
2281 {K_MOUSE, (char_u *)"Mouse"},
2282 {K_NETTERM_MOUSE, (char_u *)"NetMouse"},
2283 {K_DEC_MOUSE, (char_u *)"DecMouse"},
2284 {K_JSBTERM_MOUSE, (char_u *)"JsbMouse"},
2285 {K_PTERM_MOUSE, (char_u *)"PtermMouse"},
2286 {K_LEFTMOUSE, (char_u *)"LeftMouse"},
2287 {K_LEFTMOUSE_NM, (char_u *)"LeftMouseNM"},
2288 {K_LEFTDRAG, (char_u *)"LeftDrag"},
2289 {K_LEFTRELEASE, (char_u *)"LeftRelease"},
2290 {K_LEFTRELEASE_NM, (char_u *)"LeftReleaseNM"},
2291 {K_MIDDLEMOUSE, (char_u *)"MiddleMouse"},
2292 {K_MIDDLEDRAG, (char_u *)"MiddleDrag"},
2293 {K_MIDDLERELEASE, (char_u *)"MiddleRelease"},
2294 {K_RIGHTMOUSE, (char_u *)"RightMouse"},
2295 {K_RIGHTDRAG, (char_u *)"RightDrag"},
2296 {K_RIGHTRELEASE, (char_u *)"RightRelease"},
2297 {K_MOUSEDOWN, (char_u *)"MouseDown"},
2298 {K_MOUSEUP, (char_u *)"MouseUp"},
2299 {K_X1MOUSE, (char_u *)"X1Mouse"},
2300 {K_X1DRAG, (char_u *)"X1Drag"},
2301 {K_X1RELEASE, (char_u *)"X1Release"},
2302 {K_X2MOUSE, (char_u *)"X2Mouse"},
2303 {K_X2DRAG, (char_u *)"X2Drag"},
2304 {K_X2RELEASE, (char_u *)"X2Release"},
2305 {K_DROP, (char_u *)"Drop"},
2306 {K_ZERO, (char_u *)"Nul"},
2307#ifdef FEAT_EVAL
2308 {K_SNR, (char_u *)"SNR"},
2309#endif
2310 {K_PLUG, (char_u *)"Plug"},
2311 {0, NULL}
2312};
2313
2314#define KEY_NAMES_TABLE_LEN (sizeof(key_names_table) / sizeof(struct key_name_entry))
2315
2316#ifdef FEAT_MOUSE
2317static struct mousetable
2318{
2319 int pseudo_code; /* Code for pseudo mouse event */
2320 int button; /* Which mouse button is it? */
2321 int is_click; /* Is it a mouse button click event? */
2322 int is_drag; /* Is it a mouse drag event? */
2323} mouse_table[] =
2324{
2325 {(int)KE_LEFTMOUSE, MOUSE_LEFT, TRUE, FALSE},
2326#ifdef FEAT_GUI
2327 {(int)KE_LEFTMOUSE_NM, MOUSE_LEFT, TRUE, FALSE},
2328#endif
2329 {(int)KE_LEFTDRAG, MOUSE_LEFT, FALSE, TRUE},
2330 {(int)KE_LEFTRELEASE, MOUSE_LEFT, FALSE, FALSE},
2331#ifdef FEAT_GUI
2332 {(int)KE_LEFTRELEASE_NM, MOUSE_LEFT, FALSE, FALSE},
2333#endif
2334 {(int)KE_MIDDLEMOUSE, MOUSE_MIDDLE, TRUE, FALSE},
2335 {(int)KE_MIDDLEDRAG, MOUSE_MIDDLE, FALSE, TRUE},
2336 {(int)KE_MIDDLERELEASE, MOUSE_MIDDLE, FALSE, FALSE},
2337 {(int)KE_RIGHTMOUSE, MOUSE_RIGHT, TRUE, FALSE},
2338 {(int)KE_RIGHTDRAG, MOUSE_RIGHT, FALSE, TRUE},
2339 {(int)KE_RIGHTRELEASE, MOUSE_RIGHT, FALSE, FALSE},
2340 {(int)KE_X1MOUSE, MOUSE_X1, TRUE, FALSE},
2341 {(int)KE_X1DRAG, MOUSE_X1, FALSE, TRUE},
2342 {(int)KE_X1RELEASE, MOUSE_X1, FALSE, FALSE},
2343 {(int)KE_X2MOUSE, MOUSE_X2, TRUE, FALSE},
2344 {(int)KE_X2DRAG, MOUSE_X2, FALSE, TRUE},
2345 {(int)KE_X2RELEASE, MOUSE_X2, FALSE, FALSE},
2346 /* DRAG without CLICK */
2347 {(int)KE_IGNORE, MOUSE_RELEASE, FALSE, TRUE},
2348 /* RELEASE without CLICK */
2349 {(int)KE_IGNORE, MOUSE_RELEASE, FALSE, FALSE},
2350 {0, 0, 0, 0},
2351};
2352#endif /* FEAT_MOUSE */
2353
2354/*
2355 * Return the modifier mask bit (MOD_MASK_*) which corresponds to the given
2356 * modifier name ('S' for Shift, 'C' for Ctrl etc).
2357 */
2358 int
2359name_to_mod_mask(c)
2360 int c;
2361{
2362 int i;
2363
2364 c = TOUPPER_ASC(c);
2365 for (i = 0; mod_mask_table[i].mod_mask != 0; i++)
2366 if (c == mod_mask_table[i].name)
2367 return mod_mask_table[i].mod_flag;
2368 return 0;
2369}
2370
Bram Moolenaar071d4272004-06-13 20:20:40 +00002371/*
2372 * Check if if there is a special key code for "key" that includes the
2373 * modifiers specified.
2374 */
2375 int
2376simplify_key(key, modifiers)
2377 int key;
2378 int *modifiers;
2379{
2380 int i;
2381 int key0;
2382 int key1;
2383
2384 if (*modifiers & (MOD_MASK_SHIFT | MOD_MASK_CTRL | MOD_MASK_ALT))
2385 {
2386 /* TAB is a special case */
2387 if (key == TAB && (*modifiers & MOD_MASK_SHIFT))
2388 {
2389 *modifiers &= ~MOD_MASK_SHIFT;
2390 return K_S_TAB;
2391 }
2392 key0 = KEY2TERMCAP0(key);
2393 key1 = KEY2TERMCAP1(key);
2394 for (i = 0; modifier_keys_table[i] != NUL; i += MOD_KEYS_ENTRY_SIZE)
2395 if (key0 == modifier_keys_table[i + 3]
2396 && key1 == modifier_keys_table[i + 4]
2397 && (*modifiers & modifier_keys_table[i]))
2398 {
2399 *modifiers &= ~modifier_keys_table[i];
2400 return TERMCAP2KEY(modifier_keys_table[i + 1],
2401 modifier_keys_table[i + 2]);
2402 }
2403 }
2404 return key;
2405}
2406
2407/*
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002408 * Change <xHome> to <Home>, <xUp> to <Up>, etc.
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002409 */
2410 int
2411handle_x_keys(key)
2412 int key;
2413{
2414 switch (key)
2415 {
2416 case K_XUP: return K_UP;
2417 case K_XDOWN: return K_DOWN;
2418 case K_XLEFT: return K_LEFT;
2419 case K_XRIGHT: return K_RIGHT;
2420 case K_XHOME: return K_HOME;
Bram Moolenaar68b76a62005-03-25 21:53:48 +00002421 case K_ZHOME: return K_HOME;
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002422 case K_XEND: return K_END;
Bram Moolenaar68b76a62005-03-25 21:53:48 +00002423 case K_ZEND: return K_END;
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002424 case K_XF1: return K_F1;
2425 case K_XF2: return K_F2;
2426 case K_XF3: return K_F3;
2427 case K_XF4: return K_F4;
2428 case K_S_XF1: return K_S_F1;
2429 case K_S_XF2: return K_S_F2;
2430 case K_S_XF3: return K_S_F3;
2431 case K_S_XF4: return K_S_F4;
2432 }
2433 return key;
2434}
2435
2436/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002437 * Return a string which contains the name of the given key when the given
2438 * modifiers are down.
2439 */
2440 char_u *
2441get_special_key_name(c, modifiers)
2442 int c;
2443 int modifiers;
2444{
2445 static char_u string[MAX_KEY_NAME_LEN + 1];
2446
2447 int i, idx;
2448 int table_idx;
2449 char_u *s;
2450
2451 string[0] = '<';
2452 idx = 1;
2453
2454 /* Key that stands for a normal character. */
2455 if (IS_SPECIAL(c) && KEY2TERMCAP0(c) == KS_KEY)
2456 c = KEY2TERMCAP1(c);
2457
2458 /*
2459 * Translate shifted special keys into unshifted keys and set modifier.
2460 * Same for CTRL and ALT modifiers.
2461 */
2462 if (IS_SPECIAL(c))
2463 {
2464 for (i = 0; modifier_keys_table[i] != 0; i += MOD_KEYS_ENTRY_SIZE)
2465 if ( KEY2TERMCAP0(c) == (int)modifier_keys_table[i + 1]
2466 && (int)KEY2TERMCAP1(c) == (int)modifier_keys_table[i + 2])
2467 {
2468 modifiers |= modifier_keys_table[i];
2469 c = TERMCAP2KEY(modifier_keys_table[i + 3],
2470 modifier_keys_table[i + 4]);
2471 break;
2472 }
2473 }
2474
2475 /* try to find the key in the special key table */
2476 table_idx = find_special_key_in_table(c);
2477
2478 /*
2479 * When not a known special key, and not a printable character, try to
2480 * extract modifiers.
2481 */
2482 if (c > 0
2483#ifdef FEAT_MBYTE
2484 && (*mb_char2len)(c) == 1
2485#endif
2486 )
2487 {
2488 if (table_idx < 0
2489 && (!vim_isprintc(c) || (c & 0x7f) == ' ')
2490 && (c & 0x80))
2491 {
2492 c &= 0x7f;
2493 modifiers |= MOD_MASK_ALT;
2494 /* try again, to find the un-alted key in the special key table */
2495 table_idx = find_special_key_in_table(c);
2496 }
2497 if (table_idx < 0 && !vim_isprintc(c) && c < ' ')
2498 {
2499#ifdef EBCDIC
2500 c = CtrlChar(c);
2501#else
2502 c += '@';
2503#endif
2504 modifiers |= MOD_MASK_CTRL;
2505 }
2506 }
2507
2508 /* translate the modifier into a string */
2509 for (i = 0; mod_mask_table[i].name != 'A'; i++)
2510 if ((modifiers & mod_mask_table[i].mod_mask)
2511 == mod_mask_table[i].mod_flag)
2512 {
2513 string[idx++] = mod_mask_table[i].name;
2514 string[idx++] = (char_u)'-';
2515 }
2516
2517 if (table_idx < 0) /* unknown special key, may output t_xx */
2518 {
2519 if (IS_SPECIAL(c))
2520 {
2521 string[idx++] = 't';
2522 string[idx++] = '_';
2523 string[idx++] = KEY2TERMCAP0(c);
2524 string[idx++] = KEY2TERMCAP1(c);
2525 }
2526 /* Not a special key, only modifiers, output directly */
2527 else
2528 {
2529#ifdef FEAT_MBYTE
2530 if (has_mbyte && (*mb_char2len)(c) > 1)
2531 idx += (*mb_char2bytes)(c, string + idx);
2532 else
2533#endif
2534 if (vim_isprintc(c))
2535 string[idx++] = c;
2536 else
2537 {
2538 s = transchar(c);
2539 while (*s)
2540 string[idx++] = *s++;
2541 }
2542 }
2543 }
2544 else /* use name of special key */
2545 {
2546 STRCPY(string + idx, key_names_table[table_idx].name);
2547 idx = (int)STRLEN(string);
2548 }
2549 string[idx++] = '>';
2550 string[idx] = NUL;
2551 return string;
2552}
2553
2554/*
2555 * Try translating a <> name at (*srcp)[] to dst[].
2556 * Return the number of characters added to dst[], zero for no match.
2557 * If there is a match, srcp is advanced to after the <> name.
2558 * dst[] must be big enough to hold the result (up to six characters)!
2559 */
2560 int
2561trans_special(srcp, dst, keycode)
2562 char_u **srcp;
2563 char_u *dst;
2564 int keycode; /* prefer key code, e.g. K_DEL instead of DEL */
2565{
2566 int modifiers = 0;
2567 int key;
2568 int dlen = 0;
2569
Bram Moolenaar2a8ced02008-12-24 11:54:31 +00002570 key = find_special_key(srcp, &modifiers, keycode, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002571 if (key == 0)
2572 return 0;
2573
2574 /* Put the appropriate modifier in a string */
2575 if (modifiers != 0)
2576 {
2577 dst[dlen++] = K_SPECIAL;
2578 dst[dlen++] = KS_MODIFIER;
2579 dst[dlen++] = modifiers;
2580 }
2581
2582 if (IS_SPECIAL(key))
2583 {
2584 dst[dlen++] = K_SPECIAL;
2585 dst[dlen++] = KEY2TERMCAP0(key);
2586 dst[dlen++] = KEY2TERMCAP1(key);
2587 }
2588#ifdef FEAT_MBYTE
2589 else if (has_mbyte && !keycode)
2590 dlen += (*mb_char2bytes)(key, dst + dlen);
2591#endif
2592 else if (keycode)
2593 dlen = (int)(add_char2buf(key, dst + dlen) - dst);
2594 else
2595 dst[dlen++] = key;
2596
2597 return dlen;
2598}
2599
2600/*
2601 * Try translating a <> name at (*srcp)[], return the key and modifiers.
2602 * srcp is advanced to after the <> name.
2603 * returns 0 if there is no match.
2604 */
2605 int
Bram Moolenaar2a8ced02008-12-24 11:54:31 +00002606find_special_key(srcp, modp, keycode, keep_x_key)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002607 char_u **srcp;
2608 int *modp;
Bram Moolenaar2a8ced02008-12-24 11:54:31 +00002609 int keycode; /* prefer key code, e.g. K_DEL instead of DEL */
2610 int keep_x_key; /* don't translate xHome to Home key */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002611{
2612 char_u *last_dash;
2613 char_u *end_of_name;
2614 char_u *src;
2615 char_u *bp;
2616 int modifiers;
2617 int bit;
2618 int key;
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00002619 unsigned long n;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002620
2621 src = *srcp;
2622 if (src[0] != '<')
2623 return 0;
2624
2625 /* Find end of modifier list */
2626 last_dash = src;
2627 for (bp = src + 1; *bp == '-' || vim_isIDc(*bp); bp++)
2628 {
2629 if (*bp == '-')
2630 {
2631 last_dash = bp;
2632 if (bp[1] != NUL && bp[2] == '>')
2633 ++bp; /* anything accepted, like <C-?> */
2634 }
2635 if (bp[0] == 't' && bp[1] == '_' && bp[2] && bp[3])
2636 bp += 3; /* skip t_xx, xx may be '-' or '>' */
2637 }
2638
2639 if (*bp == '>') /* found matching '>' */
2640 {
2641 end_of_name = bp + 1;
2642
2643 if (STRNICMP(src + 1, "char-", 5) == 0 && VIM_ISDIGIT(src[6]))
2644 {
2645 /* <Char-123> or <Char-033> or <Char-0x33> */
2646 vim_str2nr(src + 6, NULL, NULL, TRUE, TRUE, NULL, &n);
2647 *modp = 0;
2648 *srcp = end_of_name;
2649 return (int)n;
2650 }
2651
2652 /* Which modifiers are given? */
2653 modifiers = 0x0;
2654 for (bp = src + 1; bp < last_dash; bp++)
2655 {
2656 if (*bp != '-')
2657 {
2658 bit = name_to_mod_mask(*bp);
2659 if (bit == 0x0)
2660 break; /* Illegal modifier name */
2661 modifiers |= bit;
2662 }
2663 }
2664
2665 /*
2666 * Legal modifier name.
2667 */
2668 if (bp >= last_dash)
2669 {
2670 /*
2671 * Modifier with single letter, or special key name.
2672 */
2673 if (modifiers != 0 && last_dash[2] == '>')
2674 key = last_dash[1];
2675 else
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002676 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00002677 key = get_special_key_code(last_dash + 1);
Bram Moolenaar2a8ced02008-12-24 11:54:31 +00002678 if (!keep_x_key)
2679 key = handle_x_keys(key);
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002680 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002681
2682 /*
2683 * get_special_key_code() may return NUL for invalid
2684 * special key name.
2685 */
2686 if (key != NUL)
2687 {
2688 /*
2689 * Only use a modifier when there is no special key code that
2690 * includes the modifier.
2691 */
2692 key = simplify_key(key, &modifiers);
2693
2694 if (!keycode)
2695 {
2696 /* don't want keycode, use single byte code */
2697 if (key == K_BS)
2698 key = BS;
2699 else if (key == K_DEL || key == K_KDEL)
2700 key = DEL;
2701 }
2702
2703 /*
2704 * Normal Key with modifier: Try to make a single byte code.
2705 */
2706 if (!IS_SPECIAL(key))
2707 key = extract_modifiers(key, &modifiers);
2708
2709 *modp = modifiers;
2710 *srcp = end_of_name;
2711 return key;
2712 }
2713 }
2714 }
2715 return 0;
2716}
2717
2718/*
2719 * Try to include modifiers in the key.
2720 * Changes "Shift-a" to 'A', "Alt-A" to 0xc0, etc.
2721 */
2722 int
2723extract_modifiers(key, modp)
2724 int key;
2725 int *modp;
2726{
2727 int modifiers = *modp;
2728
2729#ifdef MACOS
2730 /* Command-key really special, No fancynest */
2731 if (!(modifiers & MOD_MASK_CMD))
2732#endif
2733 if ((modifiers & MOD_MASK_SHIFT) && ASCII_ISALPHA(key))
2734 {
2735 key = TOUPPER_ASC(key);
2736 modifiers &= ~MOD_MASK_SHIFT;
2737 }
2738 if ((modifiers & MOD_MASK_CTRL)
2739#ifdef EBCDIC
2740 /* * TODO: EBCDIC Better use:
2741 * && (Ctrl_chr(key) || key == '?')
2742 * ??? */
2743 && strchr("?@ABCDEFGHIJKLMNOPQRSTUVWXYZ[\\]^_", key)
2744 != NULL
2745#else
2746 && ((key >= '?' && key <= '_') || ASCII_ISALPHA(key))
2747#endif
2748 )
2749 {
2750 key = Ctrl_chr(key);
2751 modifiers &= ~MOD_MASK_CTRL;
2752 /* <C-@> is <Nul> */
2753 if (key == 0)
2754 key = K_ZERO;
2755 }
2756#ifdef MACOS
2757 /* Command-key really special, No fancynest */
2758 if (!(modifiers & MOD_MASK_CMD))
2759#endif
2760 if ((modifiers & MOD_MASK_ALT) && key < 0x80
2761#ifdef FEAT_MBYTE
2762 && !enc_dbcs /* avoid creating a lead byte */
2763#endif
2764 )
2765 {
2766 key |= 0x80;
2767 modifiers &= ~MOD_MASK_ALT; /* remove the META modifier */
2768 }
2769
2770 *modp = modifiers;
2771 return key;
2772}
2773
2774/*
2775 * Try to find key "c" in the special key table.
2776 * Return the index when found, -1 when not found.
2777 */
2778 int
2779find_special_key_in_table(c)
2780 int c;
2781{
2782 int i;
2783
2784 for (i = 0; key_names_table[i].name != NULL; i++)
2785 if (c == key_names_table[i].key)
2786 break;
2787 if (key_names_table[i].name == NULL)
2788 i = -1;
2789 return i;
2790}
2791
2792/*
2793 * Find the special key with the given name (the given string does not have to
2794 * end with NUL, the name is assumed to end before the first non-idchar).
2795 * If the name starts with "t_" the next two characters are interpreted as a
2796 * termcap name.
2797 * Return the key code, or 0 if not found.
2798 */
2799 int
2800get_special_key_code(name)
2801 char_u *name;
2802{
2803 char_u *table_name;
2804 char_u string[3];
2805 int i, j;
2806
2807 /*
2808 * If it's <t_xx> we get the code for xx from the termcap
2809 */
2810 if (name[0] == 't' && name[1] == '_' && name[2] != NUL && name[3] != NUL)
2811 {
2812 string[0] = name[2];
2813 string[1] = name[3];
2814 string[2] = NUL;
2815 if (add_termcap_entry(string, FALSE) == OK)
2816 return TERMCAP2KEY(name[2], name[3]);
2817 }
2818 else
2819 for (i = 0; key_names_table[i].name != NULL; i++)
2820 {
2821 table_name = key_names_table[i].name;
2822 for (j = 0; vim_isIDc(name[j]) && table_name[j] != NUL; j++)
2823 if (TOLOWER_ASC(table_name[j]) != TOLOWER_ASC(name[j]))
2824 break;
2825 if (!vim_isIDc(name[j]) && table_name[j] == NUL)
2826 return key_names_table[i].key;
2827 }
2828 return 0;
2829}
2830
Bram Moolenaar05bb9532008-07-04 09:44:11 +00002831#if defined(FEAT_CMDL_COMPL) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002832 char_u *
2833get_key_name(i)
2834 int i;
2835{
2836 if (i >= KEY_NAMES_TABLE_LEN)
2837 return NULL;
2838 return key_names_table[i].name;
2839}
2840#endif
2841
Bram Moolenaar05bb9532008-07-04 09:44:11 +00002842#if defined(FEAT_MOUSE) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002843/*
2844 * Look up the given mouse code to return the relevant information in the other
2845 * arguments. Return which button is down or was released.
2846 */
2847 int
2848get_mouse_button(code, is_click, is_drag)
2849 int code;
2850 int *is_click;
2851 int *is_drag;
2852{
2853 int i;
2854
2855 for (i = 0; mouse_table[i].pseudo_code; i++)
2856 if (code == mouse_table[i].pseudo_code)
2857 {
2858 *is_click = mouse_table[i].is_click;
2859 *is_drag = mouse_table[i].is_drag;
2860 return mouse_table[i].button;
2861 }
2862 return 0; /* Shouldn't get here */
2863}
2864
2865/*
2866 * Return the appropriate pseudo mouse event token (KE_LEFTMOUSE etc) based on
2867 * the given information about which mouse button is down, and whether the
2868 * mouse was clicked, dragged or released.
2869 */
2870 int
2871get_pseudo_mouse_code(button, is_click, is_drag)
2872 int button; /* eg MOUSE_LEFT */
2873 int is_click;
2874 int is_drag;
2875{
2876 int i;
2877
2878 for (i = 0; mouse_table[i].pseudo_code; i++)
2879 if (button == mouse_table[i].button
2880 && is_click == mouse_table[i].is_click
2881 && is_drag == mouse_table[i].is_drag)
2882 {
2883#ifdef FEAT_GUI
Bram Moolenaarc91506a2005-04-24 22:04:21 +00002884 /* Trick: a non mappable left click and release has mouse_col -1
2885 * or added MOUSE_COLOFF. Used for 'mousefocus' in
2886 * gui_mouse_moved() */
2887 if (mouse_col < 0 || mouse_col > MOUSE_COLOFF)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002888 {
Bram Moolenaarc91506a2005-04-24 22:04:21 +00002889 if (mouse_col < 0)
2890 mouse_col = 0;
2891 else
2892 mouse_col -= MOUSE_COLOFF;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002893 if (mouse_table[i].pseudo_code == (int)KE_LEFTMOUSE)
2894 return (int)KE_LEFTMOUSE_NM;
2895 if (mouse_table[i].pseudo_code == (int)KE_LEFTRELEASE)
2896 return (int)KE_LEFTRELEASE_NM;
2897 }
2898#endif
2899 return mouse_table[i].pseudo_code;
2900 }
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00002901 return (int)KE_IGNORE; /* not recognized, ignore it */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002902}
2903#endif /* FEAT_MOUSE */
2904
2905/*
2906 * Return the current end-of-line type: EOL_DOS, EOL_UNIX or EOL_MAC.
2907 */
2908 int
2909get_fileformat(buf)
2910 buf_T *buf;
2911{
2912 int c = *buf->b_p_ff;
2913
2914 if (buf->b_p_bin || c == 'u')
2915 return EOL_UNIX;
2916 if (c == 'm')
2917 return EOL_MAC;
2918 return EOL_DOS;
2919}
2920
2921/*
2922 * Like get_fileformat(), but override 'fileformat' with "p" for "++opt=val"
2923 * argument.
2924 */
2925 int
2926get_fileformat_force(buf, eap)
2927 buf_T *buf;
2928 exarg_T *eap; /* can be NULL! */
2929{
2930 int c;
2931
2932 if (eap != NULL && eap->force_ff != 0)
2933 c = eap->cmd[eap->force_ff];
2934 else
2935 {
2936 if ((eap != NULL && eap->force_bin != 0)
2937 ? (eap->force_bin == FORCE_BIN) : buf->b_p_bin)
2938 return EOL_UNIX;
2939 c = *buf->b_p_ff;
2940 }
2941 if (c == 'u')
2942 return EOL_UNIX;
2943 if (c == 'm')
2944 return EOL_MAC;
2945 return EOL_DOS;
2946}
2947
2948/*
2949 * Set the current end-of-line type to EOL_DOS, EOL_UNIX or EOL_MAC.
2950 * Sets both 'textmode' and 'fileformat'.
2951 * Note: Does _not_ set global value of 'textmode'!
2952 */
2953 void
2954set_fileformat(t, opt_flags)
2955 int t;
2956 int opt_flags; /* OPT_LOCAL and/or OPT_GLOBAL */
2957{
2958 char *p = NULL;
2959
2960 switch (t)
2961 {
2962 case EOL_DOS:
2963 p = FF_DOS;
2964 curbuf->b_p_tx = TRUE;
2965 break;
2966 case EOL_UNIX:
2967 p = FF_UNIX;
2968 curbuf->b_p_tx = FALSE;
2969 break;
2970 case EOL_MAC:
2971 p = FF_MAC;
2972 curbuf->b_p_tx = FALSE;
2973 break;
2974 }
2975 if (p != NULL)
2976 set_string_option_direct((char_u *)"ff", -1, (char_u *)p,
Bram Moolenaar5e3cb7e2006-02-27 23:58:35 +00002977 OPT_FREE | opt_flags, 0);
2978
Bram Moolenaar071d4272004-06-13 20:20:40 +00002979#ifdef FEAT_WINDOWS
Bram Moolenaarf740b292006-02-16 22:11:02 +00002980 /* This may cause the buffer to become (un)modified. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002981 check_status(curbuf);
Bram Moolenaar997fb4b2006-02-17 21:53:23 +00002982 redraw_tabline = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002983#endif
2984#ifdef FEAT_TITLE
2985 need_maketitle = TRUE; /* set window title later */
2986#endif
2987}
2988
2989/*
2990 * Return the default fileformat from 'fileformats'.
2991 */
2992 int
2993default_fileformat()
2994{
2995 switch (*p_ffs)
2996 {
2997 case 'm': return EOL_MAC;
2998 case 'd': return EOL_DOS;
2999 }
3000 return EOL_UNIX;
3001}
3002
3003/*
3004 * Call shell. Calls mch_call_shell, with 'shellxquote' added.
3005 */
3006 int
3007call_shell(cmd, opt)
3008 char_u *cmd;
3009 int opt;
3010{
3011 char_u *ncmd;
3012 int retval;
Bram Moolenaar05159a02005-02-26 23:04:13 +00003013#ifdef FEAT_PROFILE
3014 proftime_T wait_time;
3015#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003016
3017 if (p_verbose > 3)
3018 {
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00003019 verbose_enter();
Bram Moolenaar051b7822005-05-19 21:00:46 +00003020 smsg((char_u *)_("Calling shell to execute: \"%s\""),
Bram Moolenaar071d4272004-06-13 20:20:40 +00003021 cmd == NULL ? p_sh : cmd);
3022 out_char('\n');
3023 cursor_on();
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00003024 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003025 }
3026
Bram Moolenaar05159a02005-02-26 23:04:13 +00003027#ifdef FEAT_PROFILE
Bram Moolenaar01265852006-03-20 21:50:15 +00003028 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003029 prof_child_enter(&wait_time);
3030#endif
3031
Bram Moolenaar071d4272004-06-13 20:20:40 +00003032 if (*p_sh == NUL)
3033 {
3034 EMSG(_(e_shellempty));
3035 retval = -1;
3036 }
3037 else
3038 {
3039#ifdef FEAT_GUI_MSWIN
3040 /* Don't hide the pointer while executing a shell command. */
3041 gui_mch_mousehide(FALSE);
3042#endif
3043#ifdef FEAT_GUI
3044 ++hold_gui_events;
3045#endif
3046 /* The external command may update a tags file, clear cached tags. */
3047 tag_freematch();
3048
3049 if (cmd == NULL || *p_sxq == NUL)
3050 retval = mch_call_shell(cmd, opt);
3051 else
3052 {
3053 ncmd = alloc((unsigned)(STRLEN(cmd) + STRLEN(p_sxq) * 2 + 1));
3054 if (ncmd != NULL)
3055 {
3056 STRCPY(ncmd, p_sxq);
3057 STRCAT(ncmd, cmd);
3058 STRCAT(ncmd, p_sxq);
3059 retval = mch_call_shell(ncmd, opt);
3060 vim_free(ncmd);
3061 }
3062 else
3063 retval = -1;
3064 }
3065#ifdef FEAT_GUI
3066 --hold_gui_events;
3067#endif
3068 /*
3069 * Check the window size, in case it changed while executing the
3070 * external command.
3071 */
3072 shell_resized_check();
3073 }
3074
3075#ifdef FEAT_EVAL
3076 set_vim_var_nr(VV_SHELL_ERROR, (long)retval);
Bram Moolenaar05159a02005-02-26 23:04:13 +00003077# ifdef FEAT_PROFILE
Bram Moolenaar01265852006-03-20 21:50:15 +00003078 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003079 prof_child_exit(&wait_time);
3080# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003081#endif
3082
3083 return retval;
3084}
3085
3086/*
Bram Moolenaar01265852006-03-20 21:50:15 +00003087 * VISUAL, SELECTMODE and OP_PENDING State are never set, they are equal to
3088 * NORMAL State with a condition. This function returns the real State.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003089 */
3090 int
3091get_real_state()
3092{
3093 if (State & NORMAL)
3094 {
3095#ifdef FEAT_VISUAL
3096 if (VIsual_active)
Bram Moolenaar01265852006-03-20 21:50:15 +00003097 {
3098 if (VIsual_select)
3099 return SELECTMODE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003100 return VISUAL;
Bram Moolenaar01265852006-03-20 21:50:15 +00003101 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003102 else
3103#endif
3104 if (finish_op)
Bram Moolenaar01265852006-03-20 21:50:15 +00003105 return OP_PENDING;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003106 }
3107 return State;
3108}
3109
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003110#if defined(FEAT_MBYTE) || defined(PROTO)
3111/*
3112 * Return TRUE if "p" points to just after a path separator.
3113 * Take care of multi-byte characters.
3114 * "b" must point to the start of the file name
3115 */
3116 int
3117after_pathsep(b, p)
3118 char_u *b;
3119 char_u *p;
3120{
3121 return vim_ispathsep(p[-1])
3122 && (!has_mbyte || (*mb_head_off)(b, p - 1) == 0);
3123}
3124#endif
3125
3126/*
3127 * Return TRUE if file names "f1" and "f2" are in the same directory.
3128 * "f1" may be a short name, "f2" must be a full path.
3129 */
3130 int
3131same_directory(f1, f2)
3132 char_u *f1;
3133 char_u *f2;
3134{
3135 char_u ffname[MAXPATHL];
3136 char_u *t1;
3137 char_u *t2;
3138
3139 /* safety check */
3140 if (f1 == NULL || f2 == NULL)
3141 return FALSE;
3142
3143 (void)vim_FullName(f1, ffname, MAXPATHL, FALSE);
3144 t1 = gettail_sep(ffname);
3145 t2 = gettail_sep(f2);
3146 return (t1 - ffname == t2 - f2
3147 && pathcmp((char *)ffname, (char *)f2, (int)(t1 - ffname)) == 0);
3148}
3149
Bram Moolenaar071d4272004-06-13 20:20:40 +00003150#if defined(FEAT_SESSION) || defined(MSWIN) || defined(FEAT_GUI_MAC) \
Bram Moolenaar9372a112005-12-06 19:59:18 +00003151 || ((defined(FEAT_GUI_GTK)) \
Bram Moolenaar843ee412004-06-30 16:16:41 +00003152 && ( defined(FEAT_WINDOWS) || defined(FEAT_DND)) ) \
Bram Moolenaar071d4272004-06-13 20:20:40 +00003153 || defined(FEAT_SUN_WORKSHOP) || defined(FEAT_NETBEANS_INTG) \
3154 || defined(PROTO)
3155/*
3156 * Change to a file's directory.
3157 * Caller must call shorten_fnames()!
3158 * Return OK or FAIL.
3159 */
3160 int
3161vim_chdirfile(fname)
3162 char_u *fname;
3163{
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003164 char_u dir[MAXPATHL];
Bram Moolenaar071d4272004-06-13 20:20:40 +00003165
Bram Moolenaarbbebc852005-07-18 21:47:53 +00003166 vim_strncpy(dir, fname, MAXPATHL - 1);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003167 *gettail_sep(dir) = NUL;
3168 return mch_chdir((char *)dir) == 0 ? OK : FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003169}
3170#endif
3171
3172#if defined(STAT_IGNORES_SLASH) || defined(PROTO)
3173/*
3174 * Check if "name" ends in a slash and is not a directory.
3175 * Used for systems where stat() ignores a trailing slash on a file name.
3176 * The Vim code assumes a trailing slash is only ignored for a directory.
3177 */
3178 int
3179illegal_slash(name)
3180 char *name;
3181{
3182 if (name[0] == NUL)
3183 return FALSE; /* no file name is not illegal */
3184 if (name[strlen(name) - 1] != '/')
3185 return FALSE; /* no trailing slash */
3186 if (mch_isdir((char_u *)name))
3187 return FALSE; /* trailing slash for a directory */
3188 return TRUE;
3189}
3190#endif
3191
3192#if defined(CURSOR_SHAPE) || defined(PROTO)
3193
3194/*
3195 * Handling of cursor and mouse pointer shapes in various modes.
3196 */
3197
3198cursorentry_T shape_table[SHAPE_IDX_COUNT] =
3199{
3200 /* The values will be filled in from the 'guicursor' and 'mouseshape'
3201 * defaults when Vim starts.
3202 * Adjust the SHAPE_IDX_ defines when making changes! */
3203 {0, 0, 0, 700L, 400L, 250L, 0, 0, "n", SHAPE_CURSOR+SHAPE_MOUSE},
3204 {0, 0, 0, 700L, 400L, 250L, 0, 0, "v", SHAPE_CURSOR+SHAPE_MOUSE},
3205 {0, 0, 0, 700L, 400L, 250L, 0, 0, "i", SHAPE_CURSOR+SHAPE_MOUSE},
3206 {0, 0, 0, 700L, 400L, 250L, 0, 0, "r", SHAPE_CURSOR+SHAPE_MOUSE},
3207 {0, 0, 0, 700L, 400L, 250L, 0, 0, "c", SHAPE_CURSOR+SHAPE_MOUSE},
3208 {0, 0, 0, 700L, 400L, 250L, 0, 0, "ci", SHAPE_CURSOR+SHAPE_MOUSE},
3209 {0, 0, 0, 700L, 400L, 250L, 0, 0, "cr", SHAPE_CURSOR+SHAPE_MOUSE},
3210 {0, 0, 0, 700L, 400L, 250L, 0, 0, "o", SHAPE_CURSOR+SHAPE_MOUSE},
3211 {0, 0, 0, 700L, 400L, 250L, 0, 0, "ve", SHAPE_CURSOR+SHAPE_MOUSE},
3212 {0, 0, 0, 0L, 0L, 0L, 0, 0, "e", SHAPE_MOUSE},
3213 {0, 0, 0, 0L, 0L, 0L, 0, 0, "s", SHAPE_MOUSE},
3214 {0, 0, 0, 0L, 0L, 0L, 0, 0, "sd", SHAPE_MOUSE},
3215 {0, 0, 0, 0L, 0L, 0L, 0, 0, "vs", SHAPE_MOUSE},
3216 {0, 0, 0, 0L, 0L, 0L, 0, 0, "vd", SHAPE_MOUSE},
3217 {0, 0, 0, 0L, 0L, 0L, 0, 0, "m", SHAPE_MOUSE},
3218 {0, 0, 0, 0L, 0L, 0L, 0, 0, "ml", SHAPE_MOUSE},
3219 {0, 0, 0, 100L, 100L, 100L, 0, 0, "sm", SHAPE_CURSOR},
3220};
3221
3222#ifdef FEAT_MOUSESHAPE
3223/*
3224 * Table with names for mouse shapes. Keep in sync with all the tables for
3225 * mch_set_mouse_shape()!.
3226 */
3227static char * mshape_names[] =
3228{
3229 "arrow", /* default, must be the first one */
3230 "blank", /* hidden */
3231 "beam",
3232 "updown",
3233 "udsizing",
3234 "leftright",
3235 "lrsizing",
3236 "busy",
3237 "no",
3238 "crosshair",
3239 "hand1",
3240 "hand2",
3241 "pencil",
3242 "question",
3243 "rightup-arrow",
3244 "up-arrow",
3245 NULL
3246};
3247#endif
3248
3249/*
3250 * Parse the 'guicursor' option ("what" is SHAPE_CURSOR) or 'mouseshape'
3251 * ("what" is SHAPE_MOUSE).
3252 * Returns error message for an illegal option, NULL otherwise.
3253 */
3254 char_u *
3255parse_shape_opt(what)
3256 int what;
3257{
3258 char_u *modep;
3259 char_u *colonp;
3260 char_u *commap;
3261 char_u *slashp;
3262 char_u *p, *endp;
3263 int idx = 0; /* init for GCC */
3264 int all_idx;
3265 int len;
3266 int i;
3267 long n;
3268 int found_ve = FALSE; /* found "ve" flag */
3269 int round;
3270
3271 /*
3272 * First round: check for errors; second round: do it for real.
3273 */
3274 for (round = 1; round <= 2; ++round)
3275 {
3276 /*
3277 * Repeat for all comma separated parts.
3278 */
3279#ifdef FEAT_MOUSESHAPE
3280 if (what == SHAPE_MOUSE)
3281 modep = p_mouseshape;
3282 else
3283#endif
3284 modep = p_guicursor;
3285 while (*modep != NUL)
3286 {
3287 colonp = vim_strchr(modep, ':');
3288 if (colonp == NULL)
3289 return (char_u *)N_("E545: Missing colon");
3290 if (colonp == modep)
3291 return (char_u *)N_("E546: Illegal mode");
3292 commap = vim_strchr(modep, ',');
3293
3294 /*
3295 * Repeat for all mode's before the colon.
3296 * For the 'a' mode, we loop to handle all the modes.
3297 */
3298 all_idx = -1;
3299 while (modep < colonp || all_idx >= 0)
3300 {
3301 if (all_idx < 0)
3302 {
3303 /* Find the mode. */
3304 if (modep[1] == '-' || modep[1] == ':')
3305 len = 1;
3306 else
3307 len = 2;
3308 if (len == 1 && TOLOWER_ASC(modep[0]) == 'a')
3309 all_idx = SHAPE_IDX_COUNT - 1;
3310 else
3311 {
3312 for (idx = 0; idx < SHAPE_IDX_COUNT; ++idx)
3313 if (STRNICMP(modep, shape_table[idx].name, len)
3314 == 0)
3315 break;
3316 if (idx == SHAPE_IDX_COUNT
3317 || (shape_table[idx].used_for & what) == 0)
3318 return (char_u *)N_("E546: Illegal mode");
3319 if (len == 2 && modep[0] == 'v' && modep[1] == 'e')
3320 found_ve = TRUE;
3321 }
3322 modep += len + 1;
3323 }
3324
3325 if (all_idx >= 0)
3326 idx = all_idx--;
3327 else if (round == 2)
3328 {
3329#ifdef FEAT_MOUSESHAPE
3330 if (what == SHAPE_MOUSE)
3331 {
3332 /* Set the default, for the missing parts */
3333 shape_table[idx].mshape = 0;
3334 }
3335 else
3336#endif
3337 {
3338 /* Set the defaults, for the missing parts */
3339 shape_table[idx].shape = SHAPE_BLOCK;
3340 shape_table[idx].blinkwait = 700L;
3341 shape_table[idx].blinkon = 400L;
3342 shape_table[idx].blinkoff = 250L;
3343 }
3344 }
3345
3346 /* Parse the part after the colon */
3347 for (p = colonp + 1; *p && *p != ','; )
3348 {
3349#ifdef FEAT_MOUSESHAPE
3350 if (what == SHAPE_MOUSE)
3351 {
3352 for (i = 0; ; ++i)
3353 {
3354 if (mshape_names[i] == NULL)
3355 {
3356 if (!VIM_ISDIGIT(*p))
3357 return (char_u *)N_("E547: Illegal mouseshape");
3358 if (round == 2)
3359 shape_table[idx].mshape =
3360 getdigits(&p) + MSHAPE_NUMBERED;
3361 else
3362 (void)getdigits(&p);
3363 break;
3364 }
3365 len = (int)STRLEN(mshape_names[i]);
3366 if (STRNICMP(p, mshape_names[i], len) == 0)
3367 {
3368 if (round == 2)
3369 shape_table[idx].mshape = i;
3370 p += len;
3371 break;
3372 }
3373 }
3374 }
3375 else /* if (what == SHAPE_MOUSE) */
3376#endif
3377 {
3378 /*
3379 * First handle the ones with a number argument.
3380 */
3381 i = *p;
3382 len = 0;
3383 if (STRNICMP(p, "ver", 3) == 0)
3384 len = 3;
3385 else if (STRNICMP(p, "hor", 3) == 0)
3386 len = 3;
3387 else if (STRNICMP(p, "blinkwait", 9) == 0)
3388 len = 9;
3389 else if (STRNICMP(p, "blinkon", 7) == 0)
3390 len = 7;
3391 else if (STRNICMP(p, "blinkoff", 8) == 0)
3392 len = 8;
3393 if (len != 0)
3394 {
3395 p += len;
3396 if (!VIM_ISDIGIT(*p))
3397 return (char_u *)N_("E548: digit expected");
3398 n = getdigits(&p);
3399 if (len == 3) /* "ver" or "hor" */
3400 {
3401 if (n == 0)
3402 return (char_u *)N_("E549: Illegal percentage");
3403 if (round == 2)
3404 {
3405 if (TOLOWER_ASC(i) == 'v')
3406 shape_table[idx].shape = SHAPE_VER;
3407 else
3408 shape_table[idx].shape = SHAPE_HOR;
3409 shape_table[idx].percentage = n;
3410 }
3411 }
3412 else if (round == 2)
3413 {
3414 if (len == 9)
3415 shape_table[idx].blinkwait = n;
3416 else if (len == 7)
3417 shape_table[idx].blinkon = n;
3418 else
3419 shape_table[idx].blinkoff = n;
3420 }
3421 }
3422 else if (STRNICMP(p, "block", 5) == 0)
3423 {
3424 if (round == 2)
3425 shape_table[idx].shape = SHAPE_BLOCK;
3426 p += 5;
3427 }
3428 else /* must be a highlight group name then */
3429 {
3430 endp = vim_strchr(p, '-');
3431 if (commap == NULL) /* last part */
3432 {
3433 if (endp == NULL)
3434 endp = p + STRLEN(p); /* find end of part */
3435 }
3436 else if (endp > commap || endp == NULL)
3437 endp = commap;
3438 slashp = vim_strchr(p, '/');
3439 if (slashp != NULL && slashp < endp)
3440 {
3441 /* "group/langmap_group" */
3442 i = syn_check_group(p, (int)(slashp - p));
3443 p = slashp + 1;
3444 }
3445 if (round == 2)
3446 {
3447 shape_table[idx].id = syn_check_group(p,
3448 (int)(endp - p));
3449 shape_table[idx].id_lm = shape_table[idx].id;
3450 if (slashp != NULL && slashp < endp)
3451 shape_table[idx].id = i;
3452 }
3453 p = endp;
3454 }
3455 } /* if (what != SHAPE_MOUSE) */
3456
3457 if (*p == '-')
3458 ++p;
3459 }
3460 }
3461 modep = p;
3462 if (*modep == ',')
3463 ++modep;
3464 }
3465 }
3466
3467 /* If the 's' flag is not given, use the 'v' cursor for 's' */
3468 if (!found_ve)
3469 {
3470#ifdef FEAT_MOUSESHAPE
3471 if (what == SHAPE_MOUSE)
3472 {
3473 shape_table[SHAPE_IDX_VE].mshape = shape_table[SHAPE_IDX_V].mshape;
3474 }
3475 else
3476#endif
3477 {
3478 shape_table[SHAPE_IDX_VE].shape = shape_table[SHAPE_IDX_V].shape;
3479 shape_table[SHAPE_IDX_VE].percentage =
3480 shape_table[SHAPE_IDX_V].percentage;
3481 shape_table[SHAPE_IDX_VE].blinkwait =
3482 shape_table[SHAPE_IDX_V].blinkwait;
3483 shape_table[SHAPE_IDX_VE].blinkon =
3484 shape_table[SHAPE_IDX_V].blinkon;
3485 shape_table[SHAPE_IDX_VE].blinkoff =
3486 shape_table[SHAPE_IDX_V].blinkoff;
3487 shape_table[SHAPE_IDX_VE].id = shape_table[SHAPE_IDX_V].id;
3488 shape_table[SHAPE_IDX_VE].id_lm = shape_table[SHAPE_IDX_V].id_lm;
3489 }
3490 }
3491
3492 return NULL;
3493}
3494
Bram Moolenaarac6e65f2005-08-29 22:25:38 +00003495# if defined(MCH_CURSOR_SHAPE) || defined(FEAT_GUI) \
3496 || defined(FEAT_MOUSESHAPE) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003497/*
3498 * Return the index into shape_table[] for the current mode.
3499 * When "mouse" is TRUE, consider indexes valid for the mouse pointer.
3500 */
3501 int
3502get_shape_idx(mouse)
3503 int mouse;
3504{
3505#ifdef FEAT_MOUSESHAPE
3506 if (mouse && (State == HITRETURN || State == ASKMORE))
3507 {
3508# ifdef FEAT_GUI
Bram Moolenaar9588a0f2005-01-08 21:45:39 +00003509 int x, y;
3510 gui_mch_getmouse(&x, &y);
3511 if (Y_2_ROW(y) == Rows - 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003512 return SHAPE_IDX_MOREL;
3513# endif
3514 return SHAPE_IDX_MORE;
3515 }
3516 if (mouse && drag_status_line)
3517 return SHAPE_IDX_SDRAG;
3518# ifdef FEAT_VERTSPLIT
3519 if (mouse && drag_sep_line)
3520 return SHAPE_IDX_VDRAG;
3521# endif
3522#endif
3523 if (!mouse && State == SHOWMATCH)
3524 return SHAPE_IDX_SM;
3525#ifdef FEAT_VREPLACE
3526 if (State & VREPLACE_FLAG)
3527 return SHAPE_IDX_R;
3528#endif
3529 if (State & REPLACE_FLAG)
3530 return SHAPE_IDX_R;
3531 if (State & INSERT)
3532 return SHAPE_IDX_I;
3533 if (State & CMDLINE)
3534 {
3535 if (cmdline_at_end())
3536 return SHAPE_IDX_C;
3537 if (cmdline_overstrike())
3538 return SHAPE_IDX_CR;
3539 return SHAPE_IDX_CI;
3540 }
3541 if (finish_op)
3542 return SHAPE_IDX_O;
3543#ifdef FEAT_VISUAL
3544 if (VIsual_active)
3545 {
3546 if (*p_sel == 'e')
3547 return SHAPE_IDX_VE;
3548 else
3549 return SHAPE_IDX_V;
3550 }
3551#endif
3552 return SHAPE_IDX_N;
3553}
Bram Moolenaarac6e65f2005-08-29 22:25:38 +00003554#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003555
3556# if defined(FEAT_MOUSESHAPE) || defined(PROTO)
3557static int old_mouse_shape = 0;
3558
3559/*
3560 * Set the mouse shape:
3561 * If "shape" is -1, use shape depending on the current mode,
3562 * depending on the current state.
3563 * If "shape" is -2, only update the shape when it's CLINE or STATUS (used
3564 * when the mouse moves off the status or command line).
3565 */
3566 void
3567update_mouseshape(shape_idx)
3568 int shape_idx;
3569{
3570 int new_mouse_shape;
3571
3572 /* Only works in GUI mode. */
Bram Moolenaar6bb68362005-03-22 23:03:44 +00003573 if (!gui.in_use || gui.starting)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003574 return;
3575
3576 /* Postpone the updating when more is to come. Speeds up executing of
3577 * mappings. */
3578 if (shape_idx == -1 && char_avail())
3579 {
3580 postponed_mouseshape = TRUE;
3581 return;
3582 }
3583
Bram Moolenaar14716812006-05-04 21:54:08 +00003584 /* When ignoring the mouse don't change shape on the statusline. */
3585 if (*p_mouse == NUL
3586 && (shape_idx == SHAPE_IDX_CLINE
3587 || shape_idx == SHAPE_IDX_STATUS
3588 || shape_idx == SHAPE_IDX_VSEP))
3589 shape_idx = -2;
3590
Bram Moolenaar071d4272004-06-13 20:20:40 +00003591 if (shape_idx == -2
3592 && old_mouse_shape != shape_table[SHAPE_IDX_CLINE].mshape
3593 && old_mouse_shape != shape_table[SHAPE_IDX_STATUS].mshape
3594 && old_mouse_shape != shape_table[SHAPE_IDX_VSEP].mshape)
3595 return;
3596 if (shape_idx < 0)
3597 new_mouse_shape = shape_table[get_shape_idx(TRUE)].mshape;
3598 else
3599 new_mouse_shape = shape_table[shape_idx].mshape;
3600 if (new_mouse_shape != old_mouse_shape)
3601 {
3602 mch_set_mouse_shape(new_mouse_shape);
3603 old_mouse_shape = new_mouse_shape;
3604 }
3605 postponed_mouseshape = FALSE;
3606}
3607# endif
3608
3609#endif /* CURSOR_SHAPE */
3610
3611
3612#ifdef FEAT_CRYPT
3613/*
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00003614 * Optional encryption support.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003615 * Mohsin Ahmed, mosh@sasi.com, 98-09-24
3616 * Based on zip/crypt sources.
3617 *
3618 * NOTE FOR USA: Since 2000 exporting this code from the USA is allowed to
3619 * most countries. There are a few exceptions, but that still should not be a
3620 * problem since this code was originally created in Europe and India.
3621 */
3622
3623/* from zip.h */
3624
3625typedef unsigned short ush; /* unsigned 16-bit value */
3626typedef unsigned long ulg; /* unsigned 32-bit value */
3627
3628static void make_crc_tab __ARGS((void));
3629
Bram Moolenaard6f676d2005-06-01 21:51:55 +00003630static ulg crc_32_tab[256];
Bram Moolenaar071d4272004-06-13 20:20:40 +00003631
3632/*
3633 * Fill the CRC table.
3634 */
3635 static void
3636make_crc_tab()
3637{
3638 ulg s,t,v;
3639 static int done = FALSE;
3640
3641 if (done)
3642 return;
3643 for (t = 0; t < 256; t++)
3644 {
3645 v = t;
3646 for (s = 0; s < 8; s++)
3647 v = (v >> 1) ^ ((v & 1) * (ulg)0xedb88320L);
3648 crc_32_tab[t] = v;
3649 }
3650 done = TRUE;
3651}
3652
3653#define CRC32(c, b) (crc_32_tab[((int)(c) ^ (b)) & 0xff] ^ ((c) >> 8))
3654
3655
3656static ulg keys[3]; /* keys defining the pseudo-random sequence */
3657
3658/*
3659 * Return the next byte in the pseudo-random sequence
3660 */
3661 int
3662decrypt_byte()
3663{
3664 ush temp;
3665
3666 temp = (ush)keys[2] | 2;
3667 return (int)(((unsigned)(temp * (temp ^ 1)) >> 8) & 0xff);
3668}
3669
3670/*
3671 * Update the encryption keys with the next byte of plain text
3672 */
3673 int
3674update_keys(c)
3675 int c; /* byte of plain text */
3676{
3677 keys[0] = CRC32(keys[0], c);
3678 keys[1] += keys[0] & 0xff;
3679 keys[1] = keys[1] * 134775813L + 1;
3680 keys[2] = CRC32(keys[2], (int)(keys[1] >> 24));
3681 return c;
3682}
3683
3684/*
3685 * Initialize the encryption keys and the random header according to
3686 * the given password.
3687 * If "passwd" is NULL or empty, don't do anything.
3688 */
3689 void
3690crypt_init_keys(passwd)
3691 char_u *passwd; /* password string with which to modify keys */
3692{
3693 if (passwd != NULL && *passwd != NUL)
3694 {
3695 make_crc_tab();
3696 keys[0] = 305419896L;
3697 keys[1] = 591751049L;
3698 keys[2] = 878082192L;
3699 while (*passwd != '\0')
3700 update_keys((int)*passwd++);
3701 }
3702}
3703
3704/*
3705 * Ask the user for a crypt key.
3706 * When "store" is TRUE, the new key in stored in the 'key' option, and the
3707 * 'key' option value is returned: Don't free it.
3708 * When "store" is FALSE, the typed key is returned in allocated memory.
3709 * Returns NULL on failure.
3710 */
3711 char_u *
3712get_crypt_key(store, twice)
3713 int store;
3714 int twice; /* Ask for the key twice. */
3715{
3716 char_u *p1, *p2 = NULL;
3717 int round;
3718
3719 for (round = 0; ; ++round)
3720 {
3721 cmdline_star = TRUE;
3722 cmdline_row = msg_row;
3723 p1 = getcmdline_prompt(NUL, round == 0
3724 ? (char_u *)_("Enter encryption key: ")
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +00003725 : (char_u *)_("Enter same key again: "), 0, EXPAND_NOTHING,
3726 NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003727 cmdline_star = FALSE;
3728
3729 if (p1 == NULL)
3730 break;
3731
3732 if (round == twice)
3733 {
3734 if (p2 != NULL && STRCMP(p1, p2) != 0)
3735 {
3736 MSG(_("Keys don't match!"));
3737 vim_free(p1);
3738 vim_free(p2);
3739 p2 = NULL;
3740 round = -1; /* do it again */
3741 continue;
3742 }
3743 if (store)
3744 {
3745 set_option_value((char_u *)"key", 0L, p1, OPT_LOCAL);
3746 vim_free(p1);
3747 p1 = curbuf->b_p_key;
3748 }
3749 break;
3750 }
3751 p2 = p1;
3752 }
3753
3754 /* since the user typed this, no need to wait for return */
3755 need_wait_return = FALSE;
3756 msg_didout = FALSE;
3757
3758 vim_free(p2);
3759 return p1;
3760}
3761
3762#endif /* FEAT_CRYPT */
3763
3764/* TODO: make some #ifdef for this */
3765/*--------[ file searching ]-------------------------------------------------*/
3766/*
3767 * File searching functions for 'path', 'tags' and 'cdpath' options.
3768 * External visible functions:
3769 * vim_findfile_init() creates/initialises the search context
3770 * vim_findfile_free_visited() free list of visited files/dirs of search
3771 * context
3772 * vim_findfile() find a file in the search context
3773 * vim_findfile_cleanup() cleanup/free search context created by
3774 * vim_findfile_init()
3775 *
3776 * All static functions and variables start with 'ff_'
3777 *
3778 * In general it works like this:
3779 * First you create yourself a search context by calling vim_findfile_init().
3780 * It is possible to give a search context from a previous call to
3781 * vim_findfile_init(), so it can be reused. After this you call vim_findfile()
3782 * until you are satisfied with the result or it returns NULL. On every call it
3783 * returns the next file which matches the conditions given to
3784 * vim_findfile_init(). If it doesn't find a next file it returns NULL.
3785 *
3786 * It is possible to call vim_findfile_init() again to reinitialise your search
3787 * with some new parameters. Don't forget to pass your old search context to
3788 * it, so it can reuse it and especially reuse the list of already visited
3789 * directories. If you want to delete the list of already visited directories
3790 * simply call vim_findfile_free_visited().
3791 *
3792 * When you are done call vim_findfile_cleanup() to free the search context.
3793 *
3794 * The function vim_findfile_init() has a long comment, which describes the
3795 * needed parameters.
3796 *
3797 *
3798 *
3799 * ATTENTION:
3800 * ==========
Bram Moolenaar77f66d62007-02-20 02:16:18 +00003801 * Also we use an allocated search context here, this functions are NOT
Bram Moolenaar071d4272004-06-13 20:20:40 +00003802 * thread-safe!!!!!
3803 *
3804 * To minimize parameter passing (or because I'm to lazy), only the
3805 * external visible functions get a search context as a parameter. This is
3806 * then assigned to a static global, which is used throughout the local
3807 * functions.
3808 */
3809
3810/*
3811 * type for the directory search stack
3812 */
3813typedef struct ff_stack
3814{
3815 struct ff_stack *ffs_prev;
3816
3817 /* the fix part (no wildcards) and the part containing the wildcards
3818 * of the search path
3819 */
3820 char_u *ffs_fix_path;
3821#ifdef FEAT_PATH_EXTRA
3822 char_u *ffs_wc_path;
3823#endif
3824
3825 /* files/dirs found in the above directory, matched by the first wildcard
3826 * of wc_part
3827 */
3828 char_u **ffs_filearray;
3829 int ffs_filearray_size;
3830 char_u ffs_filearray_cur; /* needed for partly handled dirs */
3831
3832 /* to store status of partly handled directories
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00003833 * 0: we work on this directory for the first time
Bram Moolenaar071d4272004-06-13 20:20:40 +00003834 * 1: this directory was partly searched in an earlier step
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00003835 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003836 int ffs_stage;
3837
3838 /* How deep are we in the directory tree?
3839 * Counts backward from value of level parameter to vim_findfile_init
3840 */
3841 int ffs_level;
3842
3843 /* Did we already expand '**' to an empty string? */
3844 int ffs_star_star_empty;
3845} ff_stack_T;
3846
3847/*
3848 * type for already visited directories or files.
3849 */
3850typedef struct ff_visited
3851{
3852 struct ff_visited *ffv_next;
3853
3854#ifdef FEAT_PATH_EXTRA
3855 /* Visited directories are different if the wildcard string are
3856 * different. So we have to save it.
3857 */
3858 char_u *ffv_wc_path;
3859#endif
3860 /* for unix use inode etc for comparison (needed because of links), else
3861 * use filename.
3862 */
3863#ifdef UNIX
3864 int ffv_dev; /* device number (-1 if not set) */
3865 ino_t ffv_ino; /* inode number */
3866#endif
3867 /* The memory for this struct is allocated according to the length of
3868 * ffv_fname.
3869 */
3870 char_u ffv_fname[1]; /* actually longer */
3871} ff_visited_T;
3872
3873/*
3874 * We might have to manage several visited lists during a search.
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00003875 * This is especially needed for the tags option. If tags is set to:
Bram Moolenaar071d4272004-06-13 20:20:40 +00003876 * "./++/tags,./++/TAGS,++/tags" (replace + with *)
3877 * So we have to do 3 searches:
3878 * 1) search from the current files directory downward for the file "tags"
3879 * 2) search from the current files directory downward for the file "TAGS"
3880 * 3) search from Vims current directory downwards for the file "tags"
3881 * As you can see, the first and the third search are for the same file, so for
3882 * the third search we can use the visited list of the first search. For the
3883 * second search we must start from a empty visited list.
3884 * The struct ff_visited_list_hdr is used to manage a linked list of already
3885 * visited lists.
3886 */
3887typedef struct ff_visited_list_hdr
3888{
3889 struct ff_visited_list_hdr *ffvl_next;
3890
3891 /* the filename the attached visited list is for */
3892 char_u *ffvl_filename;
3893
3894 ff_visited_T *ffvl_visited_list;
3895
3896} ff_visited_list_hdr_T;
3897
3898
3899/*
3900 * '**' can be expanded to several directory levels.
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00003901 * Set the default maximum depth.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003902 */
3903#define FF_MAX_STAR_STAR_EXPAND ((char_u)30)
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00003904
Bram Moolenaar071d4272004-06-13 20:20:40 +00003905/*
3906 * The search context:
3907 * ffsc_stack_ptr: the stack for the dirs to search
3908 * ffsc_visited_list: the currently active visited list
3909 * ffsc_dir_visited_list: the currently active visited list for search dirs
3910 * ffsc_visited_lists_list: the list of all visited lists
3911 * ffsc_dir_visited_lists_list: the list of all visited lists for search dirs
3912 * ffsc_file_to_search: the file to search for
3913 * ffsc_start_dir: the starting directory, if search path was relative
3914 * ffsc_fix_path: the fix part of the given path (without wildcards)
3915 * Needed for upward search.
3916 * ffsc_wc_path: the part of the given path containing wildcards
3917 * ffsc_level: how many levels of dirs to search downwards
3918 * ffsc_stopdirs_v: array of stop directories for upward search
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00003919 * ffsc_find_what: FINDFILE_BOTH, FINDFILE_DIR or FINDFILE_FILE
Bram Moolenaar071d4272004-06-13 20:20:40 +00003920 */
3921typedef struct ff_search_ctx_T
3922{
3923 ff_stack_T *ffsc_stack_ptr;
3924 ff_visited_list_hdr_T *ffsc_visited_list;
3925 ff_visited_list_hdr_T *ffsc_dir_visited_list;
3926 ff_visited_list_hdr_T *ffsc_visited_lists_list;
3927 ff_visited_list_hdr_T *ffsc_dir_visited_lists_list;
3928 char_u *ffsc_file_to_search;
3929 char_u *ffsc_start_dir;
3930 char_u *ffsc_fix_path;
3931#ifdef FEAT_PATH_EXTRA
3932 char_u *ffsc_wc_path;
3933 int ffsc_level;
3934 char_u **ffsc_stopdirs_v;
3935#endif
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00003936 int ffsc_find_what;
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00003937} ff_search_ctx_T;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003938
Bram Moolenaar071d4272004-06-13 20:20:40 +00003939/* locally needed functions */
3940#ifdef FEAT_PATH_EXTRA
3941static int ff_check_visited __ARGS((ff_visited_T **, char_u *, char_u *));
3942#else
3943static int ff_check_visited __ARGS((ff_visited_T **, char_u *));
3944#endif
3945static void vim_findfile_free_visited_list __ARGS((ff_visited_list_hdr_T **list_headp));
3946static void ff_free_visited_list __ARGS((ff_visited_T *vl));
3947static ff_visited_list_hdr_T* ff_get_visited_list __ARGS((char_u *, ff_visited_list_hdr_T **list_headp));
3948#ifdef FEAT_PATH_EXTRA
3949static int ff_wc_equal __ARGS((char_u *s1, char_u *s2));
3950#endif
3951
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00003952static void ff_push __ARGS((ff_search_ctx_T *search_ctx, ff_stack_T *stack_ptr));
3953static ff_stack_T *ff_pop __ARGS((ff_search_ctx_T *search_ctx));
3954static void ff_clear __ARGS((ff_search_ctx_T *search_ctx));
3955static void ff_free_stack_element __ARGS((ff_stack_T *stack_ptr));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003956#ifdef FEAT_PATH_EXTRA
3957static ff_stack_T *ff_create_stack_element __ARGS((char_u *, char_u *, int, int));
3958#else
3959static ff_stack_T *ff_create_stack_element __ARGS((char_u *, int, int));
3960#endif
3961#ifdef FEAT_PATH_EXTRA
3962static int ff_path_in_stoplist __ARGS((char_u *, int, char_u **));
3963#endif
3964
Bram Moolenaar071d4272004-06-13 20:20:40 +00003965#if 0
3966/*
3967 * if someone likes findfirst/findnext, here are the functions
3968 * NOT TESTED!!
3969 */
3970
3971static void *ff_fn_search_context = NULL;
3972
3973 char_u *
3974vim_findfirst(path, filename, level)
3975 char_u *path;
3976 char_u *filename;
3977 int level;
3978{
3979 ff_fn_search_context =
3980 vim_findfile_init(path, filename, NULL, level, TRUE, FALSE,
3981 ff_fn_search_context, rel_fname);
3982 if (NULL == ff_fn_search_context)
3983 return NULL;
3984 else
3985 return vim_findnext()
3986}
3987
3988 char_u *
3989vim_findnext()
3990{
3991 char_u *ret = vim_findfile(ff_fn_search_context);
3992
3993 if (NULL == ret)
3994 {
3995 vim_findfile_cleanup(ff_fn_search_context);
3996 ff_fn_search_context = NULL;
3997 }
3998 return ret;
3999}
4000#endif
4001
4002/*
4003 * Initialization routine for vim_findfile.
4004 *
4005 * Returns the newly allocated search context or NULL if an error occured.
4006 *
4007 * Don't forget to clean up by calling vim_findfile_cleanup() if you are done
4008 * with the search context.
4009 *
4010 * Find the file 'filename' in the directory 'path'.
4011 * The parameter 'path' may contain wildcards. If so only search 'level'
4012 * directories deep. The parameter 'level' is the absolute maximum and is
4013 * not related to restricts given to the '**' wildcard. If 'level' is 100
4014 * and you use '**200' vim_findfile() will stop after 100 levels.
4015 *
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004016 * 'filename' cannot contain wildcards! It is used as-is, no backslashes to
4017 * escape special characters.
4018 *
Bram Moolenaar071d4272004-06-13 20:20:40 +00004019 * If 'stopdirs' is not NULL and nothing is found downward, the search is
4020 * restarted on the next higher directory level. This is repeated until the
4021 * start-directory of a search is contained in 'stopdirs'. 'stopdirs' has the
4022 * format ";*<dirname>*\(;<dirname>\)*;\=$".
4023 *
4024 * If the 'path' is relative, the starting dir for the search is either VIM's
4025 * current dir or if the path starts with "./" the current files dir.
4026 * If the 'path' is absolut, the starting dir is that part of the path before
4027 * the first wildcard.
4028 *
4029 * Upward search is only done on the starting dir.
4030 *
4031 * If 'free_visited' is TRUE the list of already visited files/directories is
4032 * cleared. Set this to FALSE if you just want to search from another
4033 * directory, but want to be sure that no directory from a previous search is
4034 * searched again. This is useful if you search for a file at different places.
4035 * The list of visited files/dirs can also be cleared with the function
4036 * vim_findfile_free_visited().
4037 *
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004038 * Set the parameter 'find_what' to FINDFILE_DIR if you want to search for
4039 * directories only, FINDFILE_FILE for files only, FINDFILE_BOTH for both.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004040 *
4041 * A search context returned by a previous call to vim_findfile_init() can be
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004042 * passed in the parameter "search_ctx_arg". This context is reused and
4043 * reinitialized with the new parameters. The list of already visited
Bram Moolenaar071d4272004-06-13 20:20:40 +00004044 * directories from this context is only deleted if the parameter
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004045 * "free_visited" is true. Be aware that the passed "search_ctx_arg" is freed
4046 * if the reinitialization fails.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004047 *
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004048 * If you don't have a search context from a previous call "search_ctx_arg"
4049 * must be NULL.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004050 *
4051 * This function silently ignores a few errors, vim_findfile() will have
4052 * limited functionality then.
4053 */
4054/*ARGSUSED*/
4055 void *
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004056vim_findfile_init(path, filename, stopdirs, level, free_visited, find_what,
4057 search_ctx_arg, tagfile, rel_fname)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004058 char_u *path;
4059 char_u *filename;
4060 char_u *stopdirs;
4061 int level;
4062 int free_visited;
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004063 int find_what;
4064 void *search_ctx_arg;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004065 int tagfile;
4066 char_u *rel_fname; /* file name to use for "." */
4067{
4068#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004069 char_u *wc_part;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004070#endif
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004071 ff_stack_T *sptr;
4072 ff_search_ctx_T *search_ctx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004073
4074 /* If a search context is given by the caller, reuse it, else allocate a
4075 * new one.
4076 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004077 if (search_ctx_arg != NULL)
4078 search_ctx = search_ctx_arg;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004079 else
4080 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004081 search_ctx = (ff_search_ctx_T*)alloc((unsigned)sizeof(ff_search_ctx_T));
4082 if (search_ctx == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004083 goto error_return;
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004084 memset(search_ctx, 0, sizeof(ff_search_ctx_T));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004085 }
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004086 search_ctx->ffsc_find_what = find_what;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004087
4088 /* clear the search context, but NOT the visited lists */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004089 ff_clear(search_ctx);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004090
4091 /* clear visited list if wanted */
4092 if (free_visited == TRUE)
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004093 vim_findfile_free_visited(search_ctx);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004094 else
4095 {
4096 /* Reuse old visited lists. Get the visited list for the given
4097 * filename. If no list for the current filename exists, creates a new
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004098 * one. */
4099 search_ctx->ffsc_visited_list = ff_get_visited_list(filename,
4100 &search_ctx->ffsc_visited_lists_list);
4101 if (search_ctx->ffsc_visited_list == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004102 goto error_return;
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004103 search_ctx->ffsc_dir_visited_list = ff_get_visited_list(filename,
4104 &search_ctx->ffsc_dir_visited_lists_list);
4105 if (search_ctx->ffsc_dir_visited_list == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004106 goto error_return;
4107 }
4108
4109 if (ff_expand_buffer == NULL)
4110 {
4111 ff_expand_buffer = (char_u*)alloc(MAXPATHL);
4112 if (ff_expand_buffer == NULL)
4113 goto error_return;
4114 }
4115
4116 /* Store information on starting dir now if path is relative.
Bram Moolenaar5eb86f92004-07-26 12:53:41 +00004117 * If path is absolute, we do that later. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004118 if (path[0] == '.'
4119 && (vim_ispathsep(path[1]) || path[1] == NUL)
4120 && (!tagfile || vim_strchr(p_cpo, CPO_DOTTAG) == NULL)
4121 && rel_fname != NULL)
4122 {
4123 int len = (int)(gettail(rel_fname) - rel_fname);
4124
4125 if (!vim_isAbsName(rel_fname) && len + 1 < MAXPATHL)
4126 {
4127 /* Make the start dir an absolute path name. */
Bram Moolenaarbbebc852005-07-18 21:47:53 +00004128 vim_strncpy(ff_expand_buffer, rel_fname, len);
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004129 search_ctx->ffsc_start_dir = FullName_save(ff_expand_buffer, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004130 }
4131 else
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004132 search_ctx->ffsc_start_dir = vim_strnsave(rel_fname, len);
4133 if (search_ctx->ffsc_start_dir == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004134 goto error_return;
4135 if (*++path != NUL)
4136 ++path;
4137 }
4138 else if (*path == NUL || !vim_isAbsName(path))
4139 {
4140#ifdef BACKSLASH_IN_FILENAME
4141 /* "c:dir" needs "c:" to be expanded, otherwise use current dir */
4142 if (*path != NUL && path[1] == ':')
4143 {
4144 char_u drive[3];
4145
4146 drive[0] = path[0];
4147 drive[1] = ':';
4148 drive[2] = NUL;
4149 if (vim_FullName(drive, ff_expand_buffer, MAXPATHL, TRUE) == FAIL)
4150 goto error_return;
4151 path += 2;
4152 }
4153 else
4154#endif
4155 if (mch_dirname(ff_expand_buffer, MAXPATHL) == FAIL)
4156 goto error_return;
4157
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004158 search_ctx->ffsc_start_dir = vim_strsave(ff_expand_buffer);
4159 if (search_ctx->ffsc_start_dir == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004160 goto error_return;
4161
4162#ifdef BACKSLASH_IN_FILENAME
4163 /* A path that starts with "/dir" is relative to the drive, not to the
4164 * directory (but not for "//machine/dir"). Only use the drive name. */
4165 if ((*path == '/' || *path == '\\')
4166 && path[1] != path[0]
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004167 && search_ctx->ffsc_start_dir[1] == ':')
4168 search_ctx->ffsc_start_dir[2] = NUL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004169#endif
4170 }
4171
4172#ifdef FEAT_PATH_EXTRA
4173 /*
4174 * If stopdirs are given, split them into an array of pointers.
4175 * If this fails (mem allocation), there is no upward search at all or a
4176 * stop directory is not recognized -> continue silently.
4177 * If stopdirs just contains a ";" or is empty,
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004178 * search_ctx->ffsc_stopdirs_v will only contain a NULL pointer. This
Bram Moolenaar071d4272004-06-13 20:20:40 +00004179 * is handled as unlimited upward search. See function
4180 * ff_path_in_stoplist() for details.
4181 */
4182 if (stopdirs != NULL)
4183 {
4184 char_u *walker = stopdirs;
4185 int dircount;
4186
4187 while (*walker == ';')
4188 walker++;
4189
4190 dircount = 1;
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004191 search_ctx->ffsc_stopdirs_v =
4192 (char_u **)alloc((unsigned)sizeof(char_u *));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004193
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004194 if (search_ctx->ffsc_stopdirs_v != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004195 {
4196 do
4197 {
4198 char_u *helper;
4199 void *ptr;
4200
4201 helper = walker;
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004202 ptr = vim_realloc(search_ctx->ffsc_stopdirs_v,
Bram Moolenaar071d4272004-06-13 20:20:40 +00004203 (dircount + 1) * sizeof(char_u *));
4204 if (ptr)
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004205 search_ctx->ffsc_stopdirs_v = ptr;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004206 else
4207 /* ignore, keep what we have and continue */
4208 break;
4209 walker = vim_strchr(walker, ';');
4210 if (walker)
4211 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004212 search_ctx->ffsc_stopdirs_v[dircount-1] =
4213 vim_strnsave(helper, (int)(walker - helper));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004214 walker++;
4215 }
4216 else
4217 /* this might be "", which means ascent till top
4218 * of directory tree.
4219 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004220 search_ctx->ffsc_stopdirs_v[dircount-1] =
4221 vim_strsave(helper);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004222
4223 dircount++;
4224
4225 } while (walker != NULL);
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004226 search_ctx->ffsc_stopdirs_v[dircount-1] = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004227 }
4228 }
4229#endif
4230
4231#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004232 search_ctx->ffsc_level = level;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004233
4234 /* split into:
4235 * -fix path
4236 * -wildcard_stuff (might be NULL)
4237 */
4238 wc_part = vim_strchr(path, '*');
4239 if (wc_part != NULL)
4240 {
4241 int llevel;
4242 int len;
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00004243 char *errpt;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004244
4245 /* save the fix part of the path */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004246 search_ctx->ffsc_fix_path = vim_strnsave(path, (int)(wc_part - path));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004247
4248 /*
4249 * copy wc_path and add restricts to the '**' wildcard.
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00004250 * The octet after a '**' is used as a (binary) counter.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004251 * So '**3' is transposed to '**^C' ('^C' is ASCII value 3)
4252 * or '**76' is transposed to '**N'( 'N' is ASCII value 76).
4253 * For EBCDIC you get different character values.
4254 * If no restrict is given after '**' the default is used.
4255 * Due to this technic the path looks awful if you print it as a
4256 * string.
4257 */
4258 len = 0;
4259 while (*wc_part != NUL)
4260 {
4261 if (STRNCMP(wc_part, "**", 2) == 0)
4262 {
4263 ff_expand_buffer[len++] = *wc_part++;
4264 ff_expand_buffer[len++] = *wc_part++;
4265
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00004266 llevel = strtol((char *)wc_part, &errpt, 10);
4267 if ((char_u *)errpt != wc_part && llevel > 0 && llevel < 255)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004268 ff_expand_buffer[len++] = llevel;
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00004269 else if ((char_u *)errpt != wc_part && llevel == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004270 /* restrict is 0 -> remove already added '**' */
4271 len -= 2;
4272 else
4273 ff_expand_buffer[len++] = FF_MAX_STAR_STAR_EXPAND;
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00004274 wc_part = (char_u *)errpt;
Bram Moolenaar1d94f9b2005-08-04 21:29:45 +00004275 if (*wc_part != NUL && !vim_ispathsep(*wc_part))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004276 {
4277 EMSG2(_("E343: Invalid path: '**[number]' must be at the end of the path or be followed by '%s'."), PATHSEPSTR);
4278 goto error_return;
4279 }
4280 }
4281 else
4282 ff_expand_buffer[len++] = *wc_part++;
4283 }
4284 ff_expand_buffer[len] = NUL;
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004285 search_ctx->ffsc_wc_path = vim_strsave(ff_expand_buffer);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004286
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004287 if (search_ctx->ffsc_wc_path == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004288 goto error_return;
4289 }
4290 else
4291#endif
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004292 search_ctx->ffsc_fix_path = vim_strsave(path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004293
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004294 if (search_ctx->ffsc_start_dir == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004295 {
4296 /* store the fix part as startdir.
4297 * This is needed if the parameter path is fully qualified.
4298 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004299 search_ctx->ffsc_start_dir = vim_strsave(search_ctx->ffsc_fix_path);
4300 if (search_ctx->ffsc_start_dir)
4301 search_ctx->ffsc_fix_path[0] = NUL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004302 }
4303
4304 /* create an absolute path */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004305 STRCPY(ff_expand_buffer, search_ctx->ffsc_start_dir);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004306 add_pathsep(ff_expand_buffer);
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004307 STRCAT(ff_expand_buffer, search_ctx->ffsc_fix_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004308 add_pathsep(ff_expand_buffer);
4309
4310 sptr = ff_create_stack_element(ff_expand_buffer,
4311#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004312 search_ctx->ffsc_wc_path,
Bram Moolenaar071d4272004-06-13 20:20:40 +00004313#endif
4314 level, 0);
4315
4316 if (sptr == NULL)
4317 goto error_return;
4318
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004319 ff_push(search_ctx, sptr);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004320
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004321 search_ctx->ffsc_file_to_search = vim_strsave(filename);
4322 if (search_ctx->ffsc_file_to_search == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004323 goto error_return;
4324
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004325 return search_ctx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004326
4327error_return:
4328 /*
4329 * We clear the search context now!
4330 * Even when the caller gave us a (perhaps valid) context we free it here,
4331 * as we might have already destroyed it.
4332 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004333 vim_findfile_cleanup(search_ctx);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004334 return NULL;
4335}
4336
4337#if defined(FEAT_PATH_EXTRA) || defined(PROTO)
4338/*
4339 * Get the stopdir string. Check that ';' is not escaped.
4340 */
4341 char_u *
4342vim_findfile_stopdir(buf)
4343 char_u *buf;
4344{
4345 char_u *r_ptr = buf;
4346
4347 while (*r_ptr != NUL && *r_ptr != ';')
4348 {
4349 if (r_ptr[0] == '\\' && r_ptr[1] == ';')
4350 {
4351 /* overwrite the escape char,
4352 * use STRLEN(r_ptr) to move the trailing '\0'
4353 */
Bram Moolenaar446cb832008-06-24 21:56:24 +00004354 STRMOVE(r_ptr, r_ptr + 1);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004355 r_ptr++;
4356 }
4357 r_ptr++;
4358 }
4359 if (*r_ptr == ';')
4360 {
4361 *r_ptr = 0;
4362 r_ptr++;
4363 }
4364 else if (*r_ptr == NUL)
4365 r_ptr = NULL;
4366 return r_ptr;
4367}
4368#endif
4369
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004370/*
4371 * Clean up the given search context. Can handle a NULL pointer.
4372 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004373 void
4374vim_findfile_cleanup(ctx)
4375 void *ctx;
4376{
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00004377 if (ctx == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004378 return;
4379
Bram Moolenaar071d4272004-06-13 20:20:40 +00004380 vim_findfile_free_visited(ctx);
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004381 ff_clear(ctx);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004382 vim_free(ctx);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004383}
4384
4385/*
4386 * Find a file in a search context.
4387 * The search context was created with vim_findfile_init() above.
4388 * Return a pointer to an allocated file name or NULL if nothing found.
4389 * To get all matching files call this function until you get NULL.
4390 *
Bram Moolenaar5eb86f92004-07-26 12:53:41 +00004391 * If the passed search_context is NULL, NULL is returned.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004392 *
4393 * The search algorithm is depth first. To change this replace the
4394 * stack with a list (don't forget to leave partly searched directories on the
4395 * top of the list).
4396 */
4397 char_u *
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004398vim_findfile(search_ctx_arg)
4399 void *search_ctx_arg;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004400{
4401 char_u *file_path;
4402#ifdef FEAT_PATH_EXTRA
4403 char_u *rest_of_wildcards;
4404 char_u *path_end = NULL;
4405#endif
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004406 ff_stack_T *stackp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004407#if defined(FEAT_SEARCHPATH) || defined(FEAT_PATH_EXTRA)
4408 int len;
4409#endif
4410 int i;
4411 char_u *p;
4412#ifdef FEAT_SEARCHPATH
4413 char_u *suf;
4414#endif
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004415 ff_search_ctx_T *search_ctx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004416
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004417 if (search_ctx_arg == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004418 return NULL;
4419
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004420 search_ctx = (ff_search_ctx_T *)search_ctx_arg;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004421
4422 /*
4423 * filepath is used as buffer for various actions and as the storage to
4424 * return a found filename.
4425 */
4426 if ((file_path = alloc((int)MAXPATHL)) == NULL)
4427 return NULL;
4428
4429#ifdef FEAT_PATH_EXTRA
4430 /* store the end of the start dir -- needed for upward search */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004431 if (search_ctx->ffsc_start_dir != NULL)
4432 path_end = &search_ctx->ffsc_start_dir[
4433 STRLEN(search_ctx->ffsc_start_dir)];
Bram Moolenaar071d4272004-06-13 20:20:40 +00004434#endif
4435
4436#ifdef FEAT_PATH_EXTRA
4437 /* upward search loop */
4438 for (;;)
4439 {
4440#endif
4441 /* downward search loop */
4442 for (;;)
4443 {
4444 /* check if user user wants to stop the search*/
4445 ui_breakcheck();
4446 if (got_int)
4447 break;
4448
4449 /* get directory to work on from stack */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004450 stackp = ff_pop(search_ctx);
4451 if (stackp == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004452 break;
4453
4454 /*
4455 * TODO: decide if we leave this test in
4456 *
4457 * GOOD: don't search a directory(-tree) twice.
4458 * BAD: - check linked list for every new directory entered.
4459 * - check for double files also done below
4460 *
4461 * Here we check if we already searched this directory.
4462 * We already searched a directory if:
4463 * 1) The directory is the same.
4464 * 2) We would use the same wildcard string.
4465 *
4466 * Good if you have links on same directory via several ways
4467 * or you have selfreferences in directories (e.g. SuSE Linux 6.3:
4468 * /etc/rc.d/init.d is linked to /etc/rc.d -> endless loop)
4469 *
4470 * This check is only needed for directories we work on for the
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004471 * first time (hence stackp->ff_filearray == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004472 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004473 if (stackp->ffs_filearray == NULL
4474 && ff_check_visited(&search_ctx->ffsc_dir_visited_list
Bram Moolenaar071d4272004-06-13 20:20:40 +00004475 ->ffvl_visited_list,
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004476 stackp->ffs_fix_path
Bram Moolenaar071d4272004-06-13 20:20:40 +00004477#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004478 , stackp->ffs_wc_path
Bram Moolenaar071d4272004-06-13 20:20:40 +00004479#endif
4480 ) == FAIL)
4481 {
4482#ifdef FF_VERBOSE
4483 if (p_verbose >= 5)
4484 {
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004485 verbose_enter_scroll();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004486 smsg((char_u *)"Already Searched: %s (%s)",
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004487 stackp->ffs_fix_path, stackp->ffs_wc_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004488 /* don't overwrite this either */
4489 msg_puts((char_u *)"\n");
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004490 verbose_leave_scroll();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004491 }
4492#endif
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004493 ff_free_stack_element(stackp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004494 continue;
4495 }
4496#ifdef FF_VERBOSE
4497 else if (p_verbose >= 5)
4498 {
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004499 verbose_enter_scroll();
Bram Moolenaar051b7822005-05-19 21:00:46 +00004500 smsg((char_u *)"Searching: %s (%s)",
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004501 stackp->ffs_fix_path, stackp->ffs_wc_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004502 /* don't overwrite this either */
4503 msg_puts((char_u *)"\n");
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004504 verbose_leave_scroll();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004505 }
4506#endif
4507
4508 /* check depth */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004509 if (stackp->ffs_level <= 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004510 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004511 ff_free_stack_element(stackp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004512 continue;
4513 }
4514
4515 file_path[0] = NUL;
4516
4517 /*
4518 * If no filearray till now expand wildcards
4519 * The function expand_wildcards() can handle an array of paths
4520 * and all possible expands are returned in one array. We use this
4521 * to handle the expansion of '**' into an empty string.
4522 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004523 if (stackp->ffs_filearray == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004524 {
4525 char_u *dirptrs[2];
4526
4527 /* we use filepath to build the path expand_wildcards() should
4528 * expand.
4529 */
4530 dirptrs[0] = file_path;
4531 dirptrs[1] = NULL;
4532
4533 /* if we have a start dir copy it in */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004534 if (!vim_isAbsName(stackp->ffs_fix_path)
4535 && search_ctx->ffsc_start_dir)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004536 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004537 STRCPY(file_path, search_ctx->ffsc_start_dir);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004538 add_pathsep(file_path);
4539 }
4540
4541 /* append the fix part of the search path */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004542 STRCAT(file_path, stackp->ffs_fix_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004543 add_pathsep(file_path);
4544
4545#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004546 rest_of_wildcards = stackp->ffs_wc_path;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004547 if (*rest_of_wildcards != NUL)
4548 {
4549 len = (int)STRLEN(file_path);
4550 if (STRNCMP(rest_of_wildcards, "**", 2) == 0)
4551 {
4552 /* pointer to the restrict byte
4553 * The restrict byte is not a character!
4554 */
4555 p = rest_of_wildcards + 2;
4556
4557 if (*p > 0)
4558 {
4559 (*p)--;
4560 file_path[len++] = '*';
4561 }
4562
4563 if (*p == 0)
4564 {
4565 /* remove '**<numb> from wildcards */
Bram Moolenaar446cb832008-06-24 21:56:24 +00004566 STRMOVE(rest_of_wildcards, rest_of_wildcards + 3);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004567 }
4568 else
4569 rest_of_wildcards += 3;
4570
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004571 if (stackp->ffs_star_star_empty == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004572 {
4573 /* if not done before, expand '**' to empty */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004574 stackp->ffs_star_star_empty = 1;
4575 dirptrs[1] = stackp->ffs_fix_path;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004576 }
4577 }
4578
4579 /*
4580 * Here we copy until the next path separator or the end of
4581 * the path. If we stop at a path separator, there is
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00004582 * still something else left. This is handled below by
Bram Moolenaar071d4272004-06-13 20:20:40 +00004583 * pushing every directory returned from expand_wildcards()
4584 * on the stack again for further search.
4585 */
4586 while (*rest_of_wildcards
4587 && !vim_ispathsep(*rest_of_wildcards))
4588 file_path[len++] = *rest_of_wildcards++;
4589
4590 file_path[len] = NUL;
4591 if (vim_ispathsep(*rest_of_wildcards))
4592 rest_of_wildcards++;
4593 }
4594#endif
4595
4596 /*
4597 * Expand wildcards like "*" and "$VAR".
4598 * If the path is a URL don't try this.
4599 */
4600 if (path_with_url(dirptrs[0]))
4601 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004602 stackp->ffs_filearray = (char_u **)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004603 alloc((unsigned)sizeof(char *));
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004604 if (stackp->ffs_filearray != NULL
4605 && (stackp->ffs_filearray[0]
Bram Moolenaar071d4272004-06-13 20:20:40 +00004606 = vim_strsave(dirptrs[0])) != NULL)
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004607 stackp->ffs_filearray_size = 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004608 else
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004609 stackp->ffs_filearray_size = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004610 }
4611 else
4612 expand_wildcards((dirptrs[1] == NULL) ? 1 : 2, dirptrs,
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004613 &stackp->ffs_filearray_size,
4614 &stackp->ffs_filearray,
Bram Moolenaar071d4272004-06-13 20:20:40 +00004615 EW_DIR|EW_ADDSLASH|EW_SILENT);
4616
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004617 stackp->ffs_filearray_cur = 0;
4618 stackp->ffs_stage = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004619 }
4620#ifdef FEAT_PATH_EXTRA
4621 else
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004622 rest_of_wildcards = &stackp->ffs_wc_path[
4623 STRLEN(stackp->ffs_wc_path)];
Bram Moolenaar071d4272004-06-13 20:20:40 +00004624#endif
4625
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004626 if (stackp->ffs_stage == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004627 {
4628 /* this is the first time we work on this directory */
4629#ifdef FEAT_PATH_EXTRA
4630 if (*rest_of_wildcards == NUL)
4631#endif
4632 {
4633 /*
4634 * we don't have further wildcards to expand, so we have to
4635 * check for the final file now
4636 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004637 for (i = stackp->ffs_filearray_cur;
4638 i < stackp->ffs_filearray_size; ++i)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004639 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004640 if (!path_with_url(stackp->ffs_filearray[i])
4641 && !mch_isdir(stackp->ffs_filearray[i]))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004642 continue; /* not a directory */
4643
4644 /* prepare the filename to be checked for existance
4645 * below */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004646 STRCPY(file_path, stackp->ffs_filearray[i]);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004647 add_pathsep(file_path);
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004648 STRCAT(file_path, search_ctx->ffsc_file_to_search);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004649
4650 /*
4651 * Try without extra suffix and then with suffixes
4652 * from 'suffixesadd'.
4653 */
4654#ifdef FEAT_SEARCHPATH
4655 len = (int)STRLEN(file_path);
4656 suf = curbuf->b_p_sua;
4657 for (;;)
4658#endif
4659 {
4660 /* if file exists and we didn't already find it */
4661 if ((path_with_url(file_path)
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004662 || (mch_getperm(file_path) >= 0
4663 && (search_ctx->ffsc_find_what
4664 == FINDFILE_BOTH
4665 || ((search_ctx->ffsc_find_what
4666 == FINDFILE_DIR)
4667 == mch_isdir(file_path)))))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004668#ifndef FF_VERBOSE
4669 && (ff_check_visited(
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004670 &search_ctx->ffsc_visited_list->ffvl_visited_list,
Bram Moolenaar071d4272004-06-13 20:20:40 +00004671 file_path
4672#ifdef FEAT_PATH_EXTRA
4673 , (char_u *)""
4674#endif
4675 ) == OK)
4676#endif
4677 )
4678 {
4679#ifdef FF_VERBOSE
4680 if (ff_check_visited(
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004681 &search_ctx->ffsc_visited_list->ffvl_visited_list,
Bram Moolenaar071d4272004-06-13 20:20:40 +00004682 file_path
4683#ifdef FEAT_PATH_EXTRA
4684 , (char_u *)""
4685#endif
4686 ) == FAIL)
4687 {
4688 if (p_verbose >= 5)
4689 {
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004690 verbose_enter_scroll();
Bram Moolenaar051b7822005-05-19 21:00:46 +00004691 smsg((char_u *)"Already: %s",
Bram Moolenaar071d4272004-06-13 20:20:40 +00004692 file_path);
4693 /* don't overwrite this either */
4694 msg_puts((char_u *)"\n");
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004695 verbose_leave_scroll();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004696 }
4697 continue;
4698 }
4699#endif
4700
4701 /* push dir to examine rest of subdirs later */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004702 stackp->ffs_filearray_cur = i + 1;
4703 ff_push(search_ctx, stackp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004704
Bram Moolenaar6bab9fa2009-01-22 20:32:12 +00004705 if (!path_with_url(file_path))
4706 simplify_filename(file_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004707 if (mch_dirname(ff_expand_buffer, MAXPATHL)
4708 == OK)
4709 {
4710 p = shorten_fname(file_path,
4711 ff_expand_buffer);
4712 if (p != NULL)
Bram Moolenaar446cb832008-06-24 21:56:24 +00004713 STRMOVE(file_path, p);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004714 }
4715#ifdef FF_VERBOSE
4716 if (p_verbose >= 5)
4717 {
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004718 verbose_enter_scroll();
Bram Moolenaar051b7822005-05-19 21:00:46 +00004719 smsg((char_u *)"HIT: %s", file_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004720 /* don't overwrite this either */
4721 msg_puts((char_u *)"\n");
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004722 verbose_leave_scroll();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004723 }
4724#endif
4725 return file_path;
4726 }
4727
4728#ifdef FEAT_SEARCHPATH
4729 /* Not found or found already, try next suffix. */
4730 if (*suf == NUL)
4731 break;
4732 copy_option_part(&suf, file_path + len,
4733 MAXPATHL - len, ",");
4734#endif
4735 }
4736 }
4737 }
4738#ifdef FEAT_PATH_EXTRA
4739 else
4740 {
4741 /*
4742 * still wildcards left, push the directories for further
4743 * search
4744 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004745 for (i = stackp->ffs_filearray_cur;
4746 i < stackp->ffs_filearray_size; ++i)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004747 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004748 if (!mch_isdir(stackp->ffs_filearray[i]))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004749 continue; /* not a directory */
4750
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004751 ff_push(search_ctx,
4752 ff_create_stack_element(
4753 stackp->ffs_filearray[i],
4754 rest_of_wildcards,
4755 stackp->ffs_level - 1, 0));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004756 }
4757 }
4758#endif
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004759 stackp->ffs_filearray_cur = 0;
4760 stackp->ffs_stage = 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004761 }
4762
4763#ifdef FEAT_PATH_EXTRA
4764 /*
4765 * if wildcards contains '**' we have to descent till we reach the
4766 * leaves of the directory tree.
4767 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004768 if (STRNCMP(stackp->ffs_wc_path, "**", 2) == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004769 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004770 for (i = stackp->ffs_filearray_cur;
4771 i < stackp->ffs_filearray_size; ++i)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004772 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004773 if (fnamecmp(stackp->ffs_filearray[i],
4774 stackp->ffs_fix_path) == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004775 continue; /* don't repush same directory */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004776 if (!mch_isdir(stackp->ffs_filearray[i]))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004777 continue; /* not a directory */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004778 ff_push(search_ctx,
4779 ff_create_stack_element(stackp->ffs_filearray[i],
4780 stackp->ffs_wc_path, stackp->ffs_level - 1, 1));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004781 }
4782 }
4783#endif
4784
4785 /* we are done with the current directory */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004786 ff_free_stack_element(stackp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004787
4788 }
4789
4790#ifdef FEAT_PATH_EXTRA
4791 /* If we reached this, we didn't find anything downwards.
4792 * Let's check if we should do an upward search.
4793 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004794 if (search_ctx->ffsc_start_dir
4795 && search_ctx->ffsc_stopdirs_v != NULL && !got_int)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004796 {
4797 ff_stack_T *sptr;
4798
4799 /* is the last starting directory in the stop list? */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004800 if (ff_path_in_stoplist(search_ctx->ffsc_start_dir,
4801 (int)(path_end - search_ctx->ffsc_start_dir),
4802 search_ctx->ffsc_stopdirs_v) == TRUE)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004803 break;
4804
4805 /* cut of last dir */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004806 while (path_end > search_ctx->ffsc_start_dir
4807 && vim_ispathsep(*path_end))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004808 path_end--;
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004809 while (path_end > search_ctx->ffsc_start_dir
4810 && !vim_ispathsep(path_end[-1]))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004811 path_end--;
4812 *path_end = 0;
4813 path_end--;
4814
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004815 if (*search_ctx->ffsc_start_dir == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004816 break;
4817
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004818 STRCPY(file_path, search_ctx->ffsc_start_dir);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004819 add_pathsep(file_path);
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004820 STRCAT(file_path, search_ctx->ffsc_fix_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004821
4822 /* create a new stack entry */
4823 sptr = ff_create_stack_element(file_path,
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004824 search_ctx->ffsc_wc_path, search_ctx->ffsc_level, 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004825 if (sptr == NULL)
4826 break;
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004827 ff_push(search_ctx, sptr);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004828 }
4829 else
4830 break;
4831 }
4832#endif
4833
4834 vim_free(file_path);
4835 return NULL;
4836}
4837
4838/*
4839 * Free the list of lists of visited files and directories
4840 * Can handle it if the passed search_context is NULL;
4841 */
4842 void
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004843vim_findfile_free_visited(search_ctx_arg)
4844 void *search_ctx_arg;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004845{
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004846 ff_search_ctx_T *search_ctx;
4847
4848 if (search_ctx_arg == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004849 return;
4850
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004851 search_ctx = (ff_search_ctx_T *)search_ctx_arg;
4852 vim_findfile_free_visited_list(&search_ctx->ffsc_visited_lists_list);
4853 vim_findfile_free_visited_list(&search_ctx->ffsc_dir_visited_lists_list);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004854}
4855
4856 static void
4857vim_findfile_free_visited_list(list_headp)
4858 ff_visited_list_hdr_T **list_headp;
4859{
4860 ff_visited_list_hdr_T *vp;
4861
4862 while (*list_headp != NULL)
4863 {
4864 vp = (*list_headp)->ffvl_next;
4865 ff_free_visited_list((*list_headp)->ffvl_visited_list);
4866
4867 vim_free((*list_headp)->ffvl_filename);
4868 vim_free(*list_headp);
4869 *list_headp = vp;
4870 }
4871 *list_headp = NULL;
4872}
4873
4874 static void
4875ff_free_visited_list(vl)
4876 ff_visited_T *vl;
4877{
4878 ff_visited_T *vp;
4879
4880 while (vl != NULL)
4881 {
4882 vp = vl->ffv_next;
4883#ifdef FEAT_PATH_EXTRA
4884 vim_free(vl->ffv_wc_path);
4885#endif
4886 vim_free(vl);
4887 vl = vp;
4888 }
4889 vl = NULL;
4890}
4891
4892/*
4893 * Returns the already visited list for the given filename. If none is found it
4894 * allocates a new one.
4895 */
4896 static ff_visited_list_hdr_T*
4897ff_get_visited_list(filename, list_headp)
4898 char_u *filename;
4899 ff_visited_list_hdr_T **list_headp;
4900{
4901 ff_visited_list_hdr_T *retptr = NULL;
4902
4903 /* check if a visited list for the given filename exists */
4904 if (*list_headp != NULL)
4905 {
4906 retptr = *list_headp;
4907 while (retptr != NULL)
4908 {
4909 if (fnamecmp(filename, retptr->ffvl_filename) == 0)
4910 {
4911#ifdef FF_VERBOSE
4912 if (p_verbose >= 5)
4913 {
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004914 verbose_enter_scroll();
Bram Moolenaar051b7822005-05-19 21:00:46 +00004915 smsg((char_u *)"ff_get_visited_list: FOUND list for %s",
Bram Moolenaar071d4272004-06-13 20:20:40 +00004916 filename);
4917 /* don't overwrite this either */
4918 msg_puts((char_u *)"\n");
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004919 verbose_leave_scroll();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004920 }
4921#endif
4922 return retptr;
4923 }
4924 retptr = retptr->ffvl_next;
4925 }
4926 }
4927
4928#ifdef FF_VERBOSE
4929 if (p_verbose >= 5)
4930 {
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004931 verbose_enter_scroll();
Bram Moolenaar051b7822005-05-19 21:00:46 +00004932 smsg((char_u *)"ff_get_visited_list: new list for %s", filename);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004933 /* don't overwrite this either */
4934 msg_puts((char_u *)"\n");
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004935 verbose_leave_scroll();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004936 }
4937#endif
4938
4939 /*
4940 * if we reach this we didn't find a list and we have to allocate new list
4941 */
4942 retptr = (ff_visited_list_hdr_T*)alloc((unsigned)sizeof(*retptr));
4943 if (retptr == NULL)
4944 return NULL;
4945
4946 retptr->ffvl_visited_list = NULL;
4947 retptr->ffvl_filename = vim_strsave(filename);
4948 if (retptr->ffvl_filename == NULL)
4949 {
4950 vim_free(retptr);
4951 return NULL;
4952 }
4953 retptr->ffvl_next = *list_headp;
4954 *list_headp = retptr;
4955
4956 return retptr;
4957}
4958
4959#ifdef FEAT_PATH_EXTRA
4960/*
4961 * check if two wildcard paths are equal. Returns TRUE or FALSE.
4962 * They are equal if:
4963 * - both paths are NULL
4964 * - they have the same length
4965 * - char by char comparison is OK
4966 * - the only differences are in the counters behind a '**', so
4967 * '**\20' is equal to '**\24'
4968 */
4969 static int
4970ff_wc_equal(s1, s2)
4971 char_u *s1;
4972 char_u *s2;
4973{
4974 int i;
4975
4976 if (s1 == s2)
4977 return TRUE;
4978
4979 if (s1 == NULL || s2 == NULL)
4980 return FALSE;
4981
4982 if (STRLEN(s1) != STRLEN(s2))
4983 return FAIL;
4984
4985 for (i = 0; s1[i] != NUL && s2[i] != NUL; i++)
4986 {
4987 if (s1[i] != s2[i]
4988#ifdef CASE_INSENSITIVE_FILENAME
4989 && TOUPPER_LOC(s1[i]) != TOUPPER_LOC(s2[i])
4990#endif
4991 )
4992 {
4993 if (i >= 2)
4994 if (s1[i-1] == '*' && s1[i-2] == '*')
4995 continue;
4996 else
4997 return FAIL;
4998 else
4999 return FAIL;
5000 }
5001 }
5002 return TRUE;
5003}
5004#endif
5005
5006/*
5007 * maintains the list of already visited files and dirs
5008 * returns FAIL if the given file/dir is already in the list
5009 * returns OK if it is newly added
5010 *
5011 * TODO: What to do on memory allocation problems?
5012 * -> return TRUE - Better the file is found several times instead of
5013 * never.
5014 */
5015 static int
5016ff_check_visited(visited_list, fname
5017#ifdef FEAT_PATH_EXTRA
5018 , wc_path
5019#endif
5020 )
5021 ff_visited_T **visited_list;
5022 char_u *fname;
5023#ifdef FEAT_PATH_EXTRA
5024 char_u *wc_path;
5025#endif
5026{
5027 ff_visited_T *vp;
5028#ifdef UNIX
5029 struct stat st;
5030 int url = FALSE;
5031#endif
5032
5033 /* For an URL we only compare the name, otherwise we compare the
5034 * device/inode (unix) or the full path name (not Unix). */
5035 if (path_with_url(fname))
5036 {
Bram Moolenaarbbebc852005-07-18 21:47:53 +00005037 vim_strncpy(ff_expand_buffer, fname, MAXPATHL - 1);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005038#ifdef UNIX
5039 url = TRUE;
5040#endif
5041 }
5042 else
5043 {
5044 ff_expand_buffer[0] = NUL;
5045#ifdef UNIX
5046 if (mch_stat((char *)fname, &st) < 0)
5047#else
5048 if (vim_FullName(fname, ff_expand_buffer, MAXPATHL, TRUE) == FAIL)
5049#endif
5050 return FAIL;
5051 }
5052
5053 /* check against list of already visited files */
5054 for (vp = *visited_list; vp != NULL; vp = vp->ffv_next)
5055 {
5056 if (
5057#ifdef UNIX
5058 !url
5059 ? (vp->ffv_dev == st.st_dev
5060 && vp->ffv_ino == st.st_ino)
5061 :
5062#endif
5063 fnamecmp(vp->ffv_fname, ff_expand_buffer) == 0
5064 )
5065 {
5066#ifdef FEAT_PATH_EXTRA
5067 /* are the wildcard parts equal */
5068 if (ff_wc_equal(vp->ffv_wc_path, wc_path) == TRUE)
5069#endif
5070 /* already visited */
5071 return FAIL;
5072 }
5073 }
5074
5075 /*
5076 * New file/dir. Add it to the list of visited files/dirs.
5077 */
5078 vp = (ff_visited_T *)alloc((unsigned)(sizeof(ff_visited_T)
5079 + STRLEN(ff_expand_buffer)));
5080
5081 if (vp != NULL)
5082 {
5083#ifdef UNIX
5084 if (!url)
5085 {
5086 vp->ffv_ino = st.st_ino;
5087 vp->ffv_dev = st.st_dev;
5088 vp->ffv_fname[0] = NUL;
5089 }
5090 else
5091 {
5092 vp->ffv_ino = 0;
5093 vp->ffv_dev = -1;
5094#endif
5095 STRCPY(vp->ffv_fname, ff_expand_buffer);
5096#ifdef UNIX
5097 }
5098#endif
5099#ifdef FEAT_PATH_EXTRA
5100 if (wc_path != NULL)
5101 vp->ffv_wc_path = vim_strsave(wc_path);
5102 else
5103 vp->ffv_wc_path = NULL;
5104#endif
5105
5106 vp->ffv_next = *visited_list;
5107 *visited_list = vp;
5108 }
5109
5110 return OK;
5111}
5112
5113/*
5114 * create stack element from given path pieces
5115 */
5116 static ff_stack_T *
5117ff_create_stack_element(fix_part,
5118#ifdef FEAT_PATH_EXTRA
5119 wc_part,
5120#endif
5121 level, star_star_empty)
5122 char_u *fix_part;
5123#ifdef FEAT_PATH_EXTRA
5124 char_u *wc_part;
5125#endif
5126 int level;
5127 int star_star_empty;
5128{
5129 ff_stack_T *new;
5130
5131 new = (ff_stack_T *)alloc((unsigned)sizeof(ff_stack_T));
5132 if (new == NULL)
5133 return NULL;
5134
5135 new->ffs_prev = NULL;
5136 new->ffs_filearray = NULL;
5137 new->ffs_filearray_size = 0;
5138 new->ffs_filearray_cur = 0;
5139 new->ffs_stage = 0;
5140 new->ffs_level = level;
5141 new->ffs_star_star_empty = star_star_empty;;
5142
5143 /* the following saves NULL pointer checks in vim_findfile */
5144 if (fix_part == NULL)
5145 fix_part = (char_u *)"";
5146 new->ffs_fix_path = vim_strsave(fix_part);
5147
5148#ifdef FEAT_PATH_EXTRA
5149 if (wc_part == NULL)
5150 wc_part = (char_u *)"";
5151 new->ffs_wc_path = vim_strsave(wc_part);
5152#endif
5153
5154 if (new->ffs_fix_path == NULL
5155#ifdef FEAT_PATH_EXTRA
5156 || new->ffs_wc_path == NULL
5157#endif
5158 )
5159 {
5160 ff_free_stack_element(new);
5161 new = NULL;
5162 }
5163
5164 return new;
5165}
5166
5167/*
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005168 * Push a dir on the directory stack.
Bram Moolenaar071d4272004-06-13 20:20:40 +00005169 */
5170 static void
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005171ff_push(search_ctx, stack_ptr)
5172 ff_search_ctx_T *search_ctx;
5173 ff_stack_T *stack_ptr;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005174{
5175 /* check for NULL pointer, not to return an error to the user, but
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005176 * to prevent a crash */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005177 if (stack_ptr != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005178 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005179 stack_ptr->ffs_prev = search_ctx->ffsc_stack_ptr;
5180 search_ctx->ffsc_stack_ptr = stack_ptr;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005181 }
5182}
5183
5184/*
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005185 * Pop a dir from the directory stack.
5186 * Returns NULL if stack is empty.
Bram Moolenaar071d4272004-06-13 20:20:40 +00005187 */
5188 static ff_stack_T *
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005189ff_pop(search_ctx)
5190 ff_search_ctx_T *search_ctx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005191{
5192 ff_stack_T *sptr;
5193
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005194 sptr = search_ctx->ffsc_stack_ptr;
5195 if (search_ctx->ffsc_stack_ptr != NULL)
5196 search_ctx->ffsc_stack_ptr = search_ctx->ffsc_stack_ptr->ffs_prev;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005197
5198 return sptr;
5199}
5200
5201/*
5202 * free the given stack element
5203 */
5204 static void
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005205ff_free_stack_element(stack_ptr)
5206 ff_stack_T *stack_ptr;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005207{
5208 /* vim_free handles possible NULL pointers */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005209 vim_free(stack_ptr->ffs_fix_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005210#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005211 vim_free(stack_ptr->ffs_wc_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005212#endif
5213
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005214 if (stack_ptr->ffs_filearray != NULL)
5215 FreeWild(stack_ptr->ffs_filearray_size, stack_ptr->ffs_filearray);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005216
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005217 vim_free(stack_ptr);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005218}
5219
5220/*
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005221 * Clear the search context, but NOT the visited list.
Bram Moolenaar071d4272004-06-13 20:20:40 +00005222 */
5223 static void
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005224ff_clear(search_ctx)
5225 ff_search_ctx_T *search_ctx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005226{
5227 ff_stack_T *sptr;
5228
5229 /* clear up stack */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005230 while ((sptr = ff_pop(search_ctx)) != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005231 ff_free_stack_element(sptr);
5232
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005233 vim_free(search_ctx->ffsc_file_to_search);
5234 vim_free(search_ctx->ffsc_start_dir);
5235 vim_free(search_ctx->ffsc_fix_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005236#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005237 vim_free(search_ctx->ffsc_wc_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005238#endif
5239
5240#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005241 if (search_ctx->ffsc_stopdirs_v != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005242 {
5243 int i = 0;
5244
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005245 while (search_ctx->ffsc_stopdirs_v[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005246 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005247 vim_free(search_ctx->ffsc_stopdirs_v[i]);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005248 i++;
5249 }
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005250 vim_free(search_ctx->ffsc_stopdirs_v);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005251 }
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005252 search_ctx->ffsc_stopdirs_v = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005253#endif
5254
5255 /* reset everything */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005256 search_ctx->ffsc_file_to_search = NULL;
5257 search_ctx->ffsc_start_dir = NULL;
5258 search_ctx->ffsc_fix_path = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005259#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005260 search_ctx->ffsc_wc_path = NULL;
5261 search_ctx->ffsc_level = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005262#endif
5263}
5264
5265#ifdef FEAT_PATH_EXTRA
5266/*
5267 * check if the given path is in the stopdirs
5268 * returns TRUE if yes else FALSE
5269 */
5270 static int
5271ff_path_in_stoplist(path, path_len, stopdirs_v)
5272 char_u *path;
5273 int path_len;
5274 char_u **stopdirs_v;
5275{
5276 int i = 0;
5277
5278 /* eat up trailing path separators, except the first */
Bram Moolenaar1d94f9b2005-08-04 21:29:45 +00005279 while (path_len > 1 && vim_ispathsep(path[path_len - 1]))
Bram Moolenaar071d4272004-06-13 20:20:40 +00005280 path_len--;
5281
5282 /* if no path consider it as match */
5283 if (path_len == 0)
5284 return TRUE;
5285
5286 for (i = 0; stopdirs_v[i] != NULL; i++)
5287 {
5288 if ((int)STRLEN(stopdirs_v[i]) > path_len)
5289 {
5290 /* match for parent directory. So '/home' also matches
5291 * '/home/rks'. Check for PATHSEP in stopdirs_v[i], else
5292 * '/home/r' would also match '/home/rks'
5293 */
5294 if (fnamencmp(stopdirs_v[i], path, path_len) == 0
Bram Moolenaar1d94f9b2005-08-04 21:29:45 +00005295 && vim_ispathsep(stopdirs_v[i][path_len]))
Bram Moolenaar071d4272004-06-13 20:20:40 +00005296 return TRUE;
5297 }
5298 else
5299 {
5300 if (fnamecmp(stopdirs_v[i], path) == 0)
5301 return TRUE;
5302 }
5303 }
5304 return FALSE;
5305}
5306#endif
5307
5308#if defined(FEAT_SEARCHPATH) || defined(PROTO)
5309/*
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005310 * Find the file name "ptr[len]" in the path. Also finds directory names.
Bram Moolenaar071d4272004-06-13 20:20:40 +00005311 *
5312 * On the first call set the parameter 'first' to TRUE to initialize
5313 * the search. For repeating calls to FALSE.
5314 *
5315 * Repeating calls will return other files called 'ptr[len]' from the path.
5316 *
5317 * Only on the first call 'ptr' and 'len' are used. For repeating calls they
5318 * don't need valid values.
5319 *
5320 * If nothing found on the first call the option FNAME_MESS will issue the
5321 * message:
5322 * 'Can't find file "<file>" in path'
5323 * On repeating calls:
5324 * 'No more file "<file>" found in path'
5325 *
5326 * options:
5327 * FNAME_MESS give error message when not found
5328 *
5329 * Uses NameBuff[]!
5330 *
5331 * Returns an allocated string for the file name. NULL for error.
5332 *
5333 */
5334 char_u *
5335find_file_in_path(ptr, len, options, first, rel_fname)
5336 char_u *ptr; /* file name */
5337 int len; /* length of file name */
5338 int options;
5339 int first; /* use count'th matching file name */
5340 char_u *rel_fname; /* file name searching relative to */
5341{
5342 return find_file_in_path_option(ptr, len, options, first,
5343 *curbuf->b_p_path == NUL ? p_path : curbuf->b_p_path,
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005344 FINDFILE_BOTH, rel_fname, curbuf->b_p_sua);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005345}
5346
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005347static char_u *ff_file_to_find = NULL;
5348static void *fdip_search_ctx = NULL;
5349
5350#if defined(EXITFREE)
5351 static void
5352free_findfile()
5353{
5354 vim_free(ff_file_to_find);
5355 vim_findfile_cleanup(fdip_search_ctx);
5356}
5357#endif
5358
Bram Moolenaar071d4272004-06-13 20:20:40 +00005359/*
5360 * Find the directory name "ptr[len]" in the path.
5361 *
5362 * options:
5363 * FNAME_MESS give error message when not found
5364 *
5365 * Uses NameBuff[]!
5366 *
5367 * Returns an allocated string for the file name. NULL for error.
5368 */
5369 char_u *
5370find_directory_in_path(ptr, len, options, rel_fname)
5371 char_u *ptr; /* file name */
5372 int len; /* length of file name */
5373 int options;
5374 char_u *rel_fname; /* file name searching relative to */
5375{
5376 return find_file_in_path_option(ptr, len, options, TRUE, p_cdpath,
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005377 FINDFILE_DIR, rel_fname, (char_u *)"");
Bram Moolenaar071d4272004-06-13 20:20:40 +00005378}
5379
Bram Moolenaar89cb5e02004-07-19 20:55:54 +00005380 char_u *
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005381find_file_in_path_option(ptr, len, options, first, path_option, find_what, rel_fname, suffixes)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005382 char_u *ptr; /* file name */
5383 int len; /* length of file name */
5384 int options;
5385 int first; /* use count'th matching file name */
5386 char_u *path_option; /* p_path or p_cdpath */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005387 int find_what; /* FINDFILE_FILE, _DIR or _BOTH */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005388 char_u *rel_fname; /* file name we are looking relative to. */
Bram Moolenaar433f7c82006-03-21 21:29:36 +00005389 char_u *suffixes; /* list of suffixes, 'suffixesadd' option */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005390{
Bram Moolenaar071d4272004-06-13 20:20:40 +00005391 static char_u *dir;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005392 static int did_findfile_init = FALSE;
5393 char_u save_char;
5394 char_u *file_name = NULL;
5395 char_u *buf = NULL;
5396 int rel_to_curdir;
5397#ifdef AMIGA
5398 struct Process *proc = (struct Process *)FindTask(0L);
5399 APTR save_winptr = proc->pr_WindowPtr;
5400
5401 /* Avoid a requester here for a volume that doesn't exist. */
5402 proc->pr_WindowPtr = (APTR)-1L;
5403#endif
5404
5405 if (first == TRUE)
5406 {
5407 /* copy file name into NameBuff, expanding environment variables */
5408 save_char = ptr[len];
5409 ptr[len] = NUL;
5410 expand_env(ptr, NameBuff, MAXPATHL);
5411 ptr[len] = save_char;
5412
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005413 vim_free(ff_file_to_find);
5414 ff_file_to_find = vim_strsave(NameBuff);
5415 if (ff_file_to_find == NULL) /* out of memory */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005416 {
5417 file_name = NULL;
5418 goto theend;
5419 }
5420 }
5421
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005422 rel_to_curdir = (ff_file_to_find[0] == '.'
5423 && (ff_file_to_find[1] == NUL
5424 || vim_ispathsep(ff_file_to_find[1])
5425 || (ff_file_to_find[1] == '.'
5426 && (ff_file_to_find[2] == NUL
5427 || vim_ispathsep(ff_file_to_find[2])))));
5428 if (vim_isAbsName(ff_file_to_find)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005429 /* "..", "../path", "." and "./path": don't use the path_option */
5430 || rel_to_curdir
5431#if defined(MSWIN) || defined(MSDOS) || defined(OS2)
5432 /* handle "\tmp" as absolute path */
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005433 || vim_ispathsep(ff_file_to_find[0])
Bram Moolenaar071d4272004-06-13 20:20:40 +00005434 /* handle "c:name" as absulute path */
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005435 || (ff_file_to_find[0] != NUL && ff_file_to_find[1] == ':')
Bram Moolenaar071d4272004-06-13 20:20:40 +00005436#endif
5437#ifdef AMIGA
5438 /* handle ":tmp" as absolute path */
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005439 || ff_file_to_find[0] == ':'
Bram Moolenaar071d4272004-06-13 20:20:40 +00005440#endif
5441 )
5442 {
5443 /*
5444 * Absolute path, no need to use "path_option".
5445 * If this is not a first call, return NULL. We already returned a
5446 * filename on the first call.
5447 */
5448 if (first == TRUE)
5449 {
5450 int l;
5451 int run;
5452
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005453 if (path_with_url(ff_file_to_find))
Bram Moolenaar071d4272004-06-13 20:20:40 +00005454 {
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005455 file_name = vim_strsave(ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005456 goto theend;
5457 }
5458
5459 /* When FNAME_REL flag given first use the directory of the file.
5460 * Otherwise or when this fails use the current directory. */
5461 for (run = 1; run <= 2; ++run)
5462 {
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005463 l = (int)STRLEN(ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005464 if (run == 1
5465 && rel_to_curdir
5466 && (options & FNAME_REL)
5467 && rel_fname != NULL
5468 && STRLEN(rel_fname) + l < MAXPATHL)
5469 {
5470 STRCPY(NameBuff, rel_fname);
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005471 STRCPY(gettail(NameBuff), ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005472 l = (int)STRLEN(NameBuff);
5473 }
5474 else
5475 {
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005476 STRCPY(NameBuff, ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005477 run = 2;
5478 }
5479
5480 /* When the file doesn't exist, try adding parts of
5481 * 'suffixesadd'. */
Bram Moolenaar433f7c82006-03-21 21:29:36 +00005482 buf = suffixes;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005483 for (;;)
5484 {
5485 if (
5486#ifdef DJGPP
5487 /* "C:" by itself will fail for mch_getperm(),
5488 * assume it's always valid. */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005489 (find_what != FINDFILE_FILE && NameBuff[0] != NUL
Bram Moolenaar071d4272004-06-13 20:20:40 +00005490 && NameBuff[1] == ':'
5491 && NameBuff[2] == NUL) ||
5492#endif
5493 (mch_getperm(NameBuff) >= 0
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005494 && (find_what == FINDFILE_BOTH
5495 || ((find_what == FINDFILE_DIR)
5496 == mch_isdir(NameBuff)))))
Bram Moolenaar071d4272004-06-13 20:20:40 +00005497 {
5498 file_name = vim_strsave(NameBuff);
5499 goto theend;
5500 }
5501 if (*buf == NUL)
5502 break;
5503 copy_option_part(&buf, NameBuff + l, MAXPATHL - l, ",");
5504 }
5505 }
5506 }
5507 }
5508 else
5509 {
5510 /*
5511 * Loop over all paths in the 'path' or 'cdpath' option.
5512 * When "first" is set, first setup to the start of the option.
5513 * Otherwise continue to find the next match.
5514 */
5515 if (first == TRUE)
5516 {
5517 /* vim_findfile_free_visited can handle a possible NULL pointer */
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005518 vim_findfile_free_visited(fdip_search_ctx);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005519 dir = path_option;
5520 did_findfile_init = FALSE;
5521 }
5522
5523 for (;;)
5524 {
5525 if (did_findfile_init)
5526 {
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005527 file_name = vim_findfile(fdip_search_ctx);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005528 if (file_name != NULL)
5529 break;
5530
5531 did_findfile_init = FALSE;
5532 }
5533 else
5534 {
5535 char_u *r_ptr;
5536
5537 if (dir == NULL || *dir == NUL)
5538 {
5539 /* We searched all paths of the option, now we can
5540 * free the search context. */
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005541 vim_findfile_cleanup(fdip_search_ctx);
5542 fdip_search_ctx = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005543 break;
5544 }
5545
5546 if ((buf = alloc((int)(MAXPATHL))) == NULL)
5547 break;
5548
5549 /* copy next path */
5550 buf[0] = 0;
5551 copy_option_part(&dir, buf, MAXPATHL, " ,");
5552
5553#ifdef FEAT_PATH_EXTRA
5554 /* get the stopdir string */
5555 r_ptr = vim_findfile_stopdir(buf);
5556#else
5557 r_ptr = NULL;
5558#endif
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005559 fdip_search_ctx = vim_findfile_init(buf, ff_file_to_find,
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005560 r_ptr, 100, FALSE, find_what,
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005561 fdip_search_ctx, FALSE, rel_fname);
5562 if (fdip_search_ctx != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005563 did_findfile_init = TRUE;
5564 vim_free(buf);
5565 }
5566 }
5567 }
5568 if (file_name == NULL && (options & FNAME_MESS))
5569 {
5570 if (first == TRUE)
5571 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005572 if (find_what == FINDFILE_DIR)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005573 EMSG2(_("E344: Can't find directory \"%s\" in cdpath"),
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005574 ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005575 else
5576 EMSG2(_("E345: Can't find file \"%s\" in path"),
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005577 ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005578 }
5579 else
5580 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005581 if (find_what == FINDFILE_DIR)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005582 EMSG2(_("E346: No more directory \"%s\" found in cdpath"),
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005583 ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005584 else
5585 EMSG2(_("E347: No more file \"%s\" found in path"),
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005586 ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005587 }
5588 }
5589
5590theend:
5591#ifdef AMIGA
5592 proc->pr_WindowPtr = save_winptr;
5593#endif
5594 return file_name;
5595}
5596
5597#endif /* FEAT_SEARCHPATH */
5598
5599/*
5600 * Change directory to "new_dir". If FEAT_SEARCHPATH is defined, search
5601 * 'cdpath' for relative directory names, otherwise just mch_chdir().
5602 */
5603 int
5604vim_chdir(new_dir)
5605 char_u *new_dir;
5606{
5607#ifndef FEAT_SEARCHPATH
5608 return mch_chdir((char *)new_dir);
5609#else
5610 char_u *dir_name;
5611 int r;
5612
5613 dir_name = find_directory_in_path(new_dir, (int)STRLEN(new_dir),
5614 FNAME_MESS, curbuf->b_ffname);
5615 if (dir_name == NULL)
5616 return -1;
5617 r = mch_chdir((char *)dir_name);
5618 vim_free(dir_name);
5619 return r;
5620#endif
5621}
5622
5623/*
Bram Moolenaarbbebc852005-07-18 21:47:53 +00005624 * Get user name from machine-specific function.
Bram Moolenaar071d4272004-06-13 20:20:40 +00005625 * Returns the user name in "buf[len]".
Bram Moolenaarbbebc852005-07-18 21:47:53 +00005626 * Some systems are quite slow in obtaining the user name (Windows NT), thus
5627 * cache the result.
Bram Moolenaar071d4272004-06-13 20:20:40 +00005628 * Returns OK or FAIL.
5629 */
5630 int
5631get_user_name(buf, len)
5632 char_u *buf;
5633 int len;
5634{
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005635 if (username == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005636 {
5637 if (mch_get_user_name(buf, len) == FAIL)
5638 return FAIL;
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005639 username = vim_strsave(buf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005640 }
5641 else
Bram Moolenaarbbebc852005-07-18 21:47:53 +00005642 vim_strncpy(buf, username, len - 1);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005643 return OK;
5644}
5645
5646#ifndef HAVE_QSORT
5647/*
5648 * Our own qsort(), for systems that don't have it.
5649 * It's simple and slow. From the K&R C book.
5650 */
5651 void
5652qsort(base, elm_count, elm_size, cmp)
5653 void *base;
5654 size_t elm_count;
5655 size_t elm_size;
5656 int (*cmp) __ARGS((const void *, const void *));
5657{
5658 char_u *buf;
5659 char_u *p1;
5660 char_u *p2;
5661 int i, j;
5662 int gap;
5663
5664 buf = alloc((unsigned)elm_size);
5665 if (buf == NULL)
5666 return;
5667
5668 for (gap = elm_count / 2; gap > 0; gap /= 2)
5669 for (i = gap; i < elm_count; ++i)
5670 for (j = i - gap; j >= 0; j -= gap)
5671 {
5672 /* Compare the elements. */
5673 p1 = (char_u *)base + j * elm_size;
5674 p2 = (char_u *)base + (j + gap) * elm_size;
5675 if ((*cmp)((void *)p1, (void *)p2) <= 0)
5676 break;
5677 /* Exchange the elemets. */
5678 mch_memmove(buf, p1, elm_size);
5679 mch_memmove(p1, p2, elm_size);
5680 mch_memmove(p2, buf, elm_size);
5681 }
5682
5683 vim_free(buf);
5684}
5685#endif
5686
Bram Moolenaar071d4272004-06-13 20:20:40 +00005687/*
5688 * Sort an array of strings.
5689 */
5690static int
5691#ifdef __BORLANDC__
5692_RTLENTRYF
5693#endif
5694sort_compare __ARGS((const void *s1, const void *s2));
5695
5696 static int
5697#ifdef __BORLANDC__
5698_RTLENTRYF
5699#endif
5700sort_compare(s1, s2)
5701 const void *s1;
5702 const void *s2;
5703{
5704 return STRCMP(*(char **)s1, *(char **)s2);
5705}
5706
5707 void
5708sort_strings(files, count)
5709 char_u **files;
5710 int count;
5711{
5712 qsort((void *)files, (size_t)count, sizeof(char_u *), sort_compare);
5713}
Bram Moolenaar071d4272004-06-13 20:20:40 +00005714
5715#if !defined(NO_EXPANDPATH) || defined(PROTO)
5716/*
5717 * Compare path "p[]" to "q[]".
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00005718 * If "maxlen" >= 0 compare "p[maxlen]" to "q[maxlen]"
Bram Moolenaar071d4272004-06-13 20:20:40 +00005719 * Return value like strcmp(p, q), but consider path separators.
5720 */
5721 int
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00005722pathcmp(p, q, maxlen)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005723 const char *p, *q;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00005724 int maxlen;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005725{
5726 int i;
Bram Moolenaar86b68352004-12-27 21:59:20 +00005727 const char *s = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005728
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00005729 for (i = 0; maxlen < 0 || i < maxlen; ++i)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005730 {
5731 /* End of "p": check if "q" also ends or just has a slash. */
5732 if (p[i] == NUL)
5733 {
5734 if (q[i] == NUL) /* full match */
5735 return 0;
5736 s = q;
5737 break;
5738 }
5739
5740 /* End of "q": check if "p" just has a slash. */
5741 if (q[i] == NUL)
5742 {
5743 s = p;
5744 break;
5745 }
5746
5747 if (
5748#ifdef CASE_INSENSITIVE_FILENAME
5749 TOUPPER_LOC(p[i]) != TOUPPER_LOC(q[i])
5750#else
5751 p[i] != q[i]
5752#endif
5753#ifdef BACKSLASH_IN_FILENAME
5754 /* consider '/' and '\\' to be equal */
5755 && !((p[i] == '/' && q[i] == '\\')
5756 || (p[i] == '\\' && q[i] == '/'))
5757#endif
5758 )
5759 {
5760 if (vim_ispathsep(p[i]))
5761 return -1;
5762 if (vim_ispathsep(q[i]))
5763 return 1;
5764 return ((char_u *)p)[i] - ((char_u *)q)[i]; /* no match */
5765 }
5766 }
Bram Moolenaar86b68352004-12-27 21:59:20 +00005767 if (s == NULL) /* "i" ran into "maxlen" */
5768 return 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005769
5770 /* ignore a trailing slash, but not "//" or ":/" */
Bram Moolenaar86b68352004-12-27 21:59:20 +00005771 if (s[i + 1] == NUL
5772 && i > 0
5773 && !after_pathsep((char_u *)s, (char_u *)s + i)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005774#ifdef BACKSLASH_IN_FILENAME
Bram Moolenaar86b68352004-12-27 21:59:20 +00005775 && (s[i] == '/' || s[i] == '\\')
Bram Moolenaar071d4272004-06-13 20:20:40 +00005776#else
Bram Moolenaar86b68352004-12-27 21:59:20 +00005777 && s[i] == '/'
Bram Moolenaar071d4272004-06-13 20:20:40 +00005778#endif
Bram Moolenaar86b68352004-12-27 21:59:20 +00005779 )
Bram Moolenaar071d4272004-06-13 20:20:40 +00005780 return 0; /* match with trailing slash */
5781 if (s == q)
5782 return -1; /* no match */
5783 return 1;
5784}
5785#endif
5786
Bram Moolenaar071d4272004-06-13 20:20:40 +00005787/*
5788 * The putenv() implementation below comes from the "screen" program.
5789 * Included with permission from Juergen Weigert.
5790 * See pty.c for the copyright notice.
5791 */
5792
5793/*
5794 * putenv -- put value into environment
5795 *
5796 * Usage: i = putenv (string)
5797 * int i;
5798 * char *string;
5799 *
5800 * where string is of the form <name>=<value>.
5801 * Putenv returns 0 normally, -1 on error (not enough core for malloc).
5802 *
5803 * Putenv may need to add a new name into the environment, or to
5804 * associate a value longer than the current value with a particular
5805 * name. So, to make life simpler, putenv() copies your entire
5806 * environment into the heap (i.e. malloc()) from the stack
5807 * (i.e. where it resides when your process is initiated) the first
5808 * time you call it.
5809 *
5810 * (history removed, not very interesting. See the "screen" sources.)
5811 */
5812
5813#if !defined(HAVE_SETENV) && !defined(HAVE_PUTENV)
5814
5815#define EXTRASIZE 5 /* increment to add to env. size */
5816
5817static int envsize = -1; /* current size of environment */
5818#ifndef MACOS_CLASSIC
5819extern
5820#endif
5821 char **environ; /* the global which is your env. */
5822
5823static int findenv __ARGS((char *name)); /* look for a name in the env. */
5824static int newenv __ARGS((void)); /* copy env. from stack to heap */
5825static int moreenv __ARGS((void)); /* incr. size of env. */
5826
5827 int
5828putenv(string)
5829 const char *string;
5830{
5831 int i;
5832 char *p;
5833
5834 if (envsize < 0)
5835 { /* first time putenv called */
5836 if (newenv() < 0) /* copy env. to heap */
5837 return -1;
5838 }
5839
5840 i = findenv((char *)string); /* look for name in environment */
5841
5842 if (i < 0)
5843 { /* name must be added */
5844 for (i = 0; environ[i]; i++);
5845 if (i >= (envsize - 1))
5846 { /* need new slot */
5847 if (moreenv() < 0)
5848 return -1;
5849 }
5850 p = (char *)alloc((unsigned)(strlen(string) + 1));
5851 if (p == NULL) /* not enough core */
5852 return -1;
5853 environ[i + 1] = 0; /* new end of env. */
5854 }
5855 else
5856 { /* name already in env. */
5857 p = vim_realloc(environ[i], strlen(string) + 1);
5858 if (p == NULL)
5859 return -1;
5860 }
5861 sprintf(p, "%s", string); /* copy into env. */
5862 environ[i] = p;
5863
5864 return 0;
5865}
5866
5867 static int
5868findenv(name)
5869 char *name;
5870{
5871 char *namechar, *envchar;
5872 int i, found;
5873
5874 found = 0;
5875 for (i = 0; environ[i] && !found; i++)
5876 {
5877 envchar = environ[i];
5878 namechar = name;
5879 while (*namechar && *namechar != '=' && (*namechar == *envchar))
5880 {
5881 namechar++;
5882 envchar++;
5883 }
5884 found = ((*namechar == '\0' || *namechar == '=') && *envchar == '=');
5885 }
5886 return found ? i - 1 : -1;
5887}
5888
5889 static int
5890newenv()
5891{
5892 char **env, *elem;
5893 int i, esize;
5894
5895#ifdef MACOS
5896 /* for Mac a new, empty environment is created */
5897 i = 0;
5898#else
5899 for (i = 0; environ[i]; i++)
5900 ;
5901#endif
5902 esize = i + EXTRASIZE + 1;
5903 env = (char **)alloc((unsigned)(esize * sizeof (elem)));
5904 if (env == NULL)
5905 return -1;
5906
5907#ifndef MACOS
5908 for (i = 0; environ[i]; i++)
5909 {
5910 elem = (char *)alloc((unsigned)(strlen(environ[i]) + 1));
5911 if (elem == NULL)
5912 return -1;
5913 env[i] = elem;
5914 strcpy(elem, environ[i]);
5915 }
5916#endif
5917
5918 env[i] = 0;
5919 environ = env;
5920 envsize = esize;
5921 return 0;
5922}
5923
5924 static int
5925moreenv()
5926{
5927 int esize;
5928 char **env;
5929
5930 esize = envsize + EXTRASIZE;
5931 env = (char **)vim_realloc((char *)environ, esize * sizeof (*env));
5932 if (env == 0)
5933 return -1;
5934 environ = env;
5935 envsize = esize;
5936 return 0;
5937}
5938
5939# ifdef USE_VIMPTY_GETENV
5940 char_u *
5941vimpty_getenv(string)
5942 const char_u *string;
5943{
5944 int i;
5945 char_u *p;
5946
5947 if (envsize < 0)
5948 return NULL;
5949
5950 i = findenv((char *)string);
5951
5952 if (i < 0)
5953 return NULL;
5954
5955 p = vim_strchr((char_u *)environ[i], '=');
5956 return (p + 1);
5957}
5958# endif
5959
5960#endif /* !defined(HAVE_SETENV) && !defined(HAVE_PUTENV) */
Bram Moolenaarc4a06d32005-06-07 21:04:49 +00005961
Bram Moolenaarc4956c82006-03-12 21:58:43 +00005962#if defined(FEAT_EVAL) || defined(FEAT_SPELL) || defined(PROTO)
Bram Moolenaarc4a06d32005-06-07 21:04:49 +00005963/*
5964 * Return 0 for not writable, 1 for writable file, 2 for a dir which we have
5965 * rights to write into.
5966 */
5967 int
5968filewritable(fname)
5969 char_u *fname;
5970{
5971 int retval = 0;
5972#if defined(UNIX) || defined(VMS)
5973 int perm = 0;
5974#endif
5975
5976#if defined(UNIX) || defined(VMS)
5977 perm = mch_getperm(fname);
5978#endif
5979#ifndef MACOS_CLASSIC /* TODO: get either mch_writable or mch_access */
5980 if (
5981# ifdef WIN3264
5982 mch_writable(fname) &&
5983# else
5984# if defined(UNIX) || defined(VMS)
5985 (perm & 0222) &&
5986# endif
5987# endif
5988 mch_access((char *)fname, W_OK) == 0
5989 )
5990#endif
5991 {
5992 ++retval;
5993 if (mch_isdir(fname))
5994 ++retval;
5995 }
5996 return retval;
5997}
5998#endif
Bram Moolenaar6bab4d12005-06-16 21:53:56 +00005999
6000/*
6001 * Print an error message with one or two "%s" and one or two string arguments.
6002 * This is not in message.c to avoid a warning for prototypes.
6003 */
6004 int
6005emsg3(s, a1, a2)
6006 char_u *s, *a1, *a2;
6007{
Bram Moolenaareb3593b2006-04-22 22:33:57 +00006008 if (emsg_not_now())
Bram Moolenaar6bab4d12005-06-16 21:53:56 +00006009 return TRUE; /* no error messages at the moment */
Bram Moolenaar6f192452007-11-08 19:49:02 +00006010#ifdef HAVE_STDARG_H
6011 vim_snprintf((char *)IObuff, IOSIZE, (char *)s, a1, a2);
6012#else
Bram Moolenaareb3593b2006-04-22 22:33:57 +00006013 vim_snprintf((char *)IObuff, IOSIZE, (char *)s, (long_u)a1, (long_u)a2);
Bram Moolenaar6f192452007-11-08 19:49:02 +00006014#endif
Bram Moolenaar6bab4d12005-06-16 21:53:56 +00006015 return emsg(IObuff);
6016}
6017
6018/*
6019 * Print an error message with one "%ld" and one long int argument.
6020 * This is not in message.c to avoid a warning for prototypes.
6021 */
6022 int
6023emsgn(s, n)
6024 char_u *s;
6025 long n;
6026{
Bram Moolenaareb3593b2006-04-22 22:33:57 +00006027 if (emsg_not_now())
Bram Moolenaar6bab4d12005-06-16 21:53:56 +00006028 return TRUE; /* no error messages at the moment */
6029 vim_snprintf((char *)IObuff, IOSIZE, (char *)s, n);
6030 return emsg(IObuff);
6031}