blob: ec3849f5c41bef997a43f76c2a55cbf8b015c004 [file] [log] [blame]
Bram Moolenaar071d4272004-06-13 20:20:40 +00001/* vi:set ts=8 sts=4 sw=4:
2 *
3 * VIM - Vi IMproved by Bram Moolenaar
4 *
5 * Do ":help uganda" in Vim to read copying and usage conditions.
6 * Do ":help credits" in Vim to see a list of people who contributed.
7 * See README.txt for an overview of the Vim source code.
8 */
9
10/*
11 * misc2.c: Various functions.
12 */
13#include "vim.h"
14
Bram Moolenaarf461c8e2005-06-25 23:04:51 +000015static char_u *username = NULL; /* cached result of mch_get_user_name() */
16
17static char_u *ff_expand_buffer = NULL; /* used for expanding filenames */
18
Bram Moolenaar071d4272004-06-13 20:20:40 +000019#if defined(FEAT_VIRTUALEDIT) || defined(PROTO)
20static int coladvance2 __ARGS((pos_T *pos, int addspaces, int finetune, colnr_T wcol));
21
22/*
23 * Return TRUE if in the current mode we need to use virtual.
24 */
25 int
26virtual_active()
27{
28 /* While an operator is being executed we return "virtual_op", because
29 * VIsual_active has already been reset, thus we can't check for "block"
30 * being used. */
31 if (virtual_op != MAYBE)
32 return virtual_op;
33 return (ve_flags == VE_ALL
34# ifdef FEAT_VISUAL
35 || ((ve_flags & VE_BLOCK) && VIsual_active && VIsual_mode == Ctrl_V)
36# endif
37 || ((ve_flags & VE_INSERT) && (State & INSERT)));
38}
39
40/*
41 * Get the screen position of the cursor.
42 */
43 int
44getviscol()
45{
46 colnr_T x;
47
48 getvvcol(curwin, &curwin->w_cursor, &x, NULL, NULL);
49 return (int)x;
50}
51
52/*
53 * Get the screen position of character col with a coladd in the cursor line.
54 */
55 int
56getviscol2(col, coladd)
57 colnr_T col;
58 colnr_T coladd;
59{
60 colnr_T x;
61 pos_T pos;
62
63 pos.lnum = curwin->w_cursor.lnum;
64 pos.col = col;
65 pos.coladd = coladd;
66 getvvcol(curwin, &pos, &x, NULL, NULL);
67 return (int)x;
68}
69
70/*
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +000071 * Go to column "wcol", and add/insert white space as necessary to get the
Bram Moolenaar071d4272004-06-13 20:20:40 +000072 * cursor in that column.
73 * The caller must have saved the cursor line for undo!
74 */
75 int
76coladvance_force(wcol)
77 colnr_T wcol;
78{
79 int rc = coladvance2(&curwin->w_cursor, TRUE, FALSE, wcol);
80
81 if (wcol == MAXCOL)
82 curwin->w_valid &= ~VALID_VIRTCOL;
83 else
84 {
85 /* Virtcol is valid */
86 curwin->w_valid |= VALID_VIRTCOL;
87 curwin->w_virtcol = wcol;
88 }
89 return rc;
90}
91#endif
92
93/*
94 * Try to advance the Cursor to the specified screen column.
95 * If virtual editing: fine tune the cursor position.
96 * Note that all virtual positions off the end of a line should share
97 * a curwin->w_cursor.col value (n.b. this is equal to STRLEN(line)),
98 * beginning at coladd 0.
99 *
100 * return OK if desired column is reached, FAIL if not
101 */
102 int
103coladvance(wcol)
104 colnr_T wcol;
105{
106 int rc = getvpos(&curwin->w_cursor, wcol);
107
108 if (wcol == MAXCOL || rc == FAIL)
109 curwin->w_valid &= ~VALID_VIRTCOL;
Bram Moolenaardfccaf02004-12-31 20:56:11 +0000110 else if (*ml_get_cursor() != TAB)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000111 {
Bram Moolenaardfccaf02004-12-31 20:56:11 +0000112 /* Virtcol is valid when not on a TAB */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000113 curwin->w_valid |= VALID_VIRTCOL;
114 curwin->w_virtcol = wcol;
115 }
116 return rc;
117}
118
119/*
120 * Return in "pos" the position of the cursor advanced to screen column "wcol".
121 * return OK if desired column is reached, FAIL if not
122 */
123 int
124getvpos(pos, wcol)
125 pos_T *pos;
126 colnr_T wcol;
127{
128#ifdef FEAT_VIRTUALEDIT
129 return coladvance2(pos, FALSE, virtual_active(), wcol);
130}
131
132 static int
133coladvance2(pos, addspaces, finetune, wcol)
134 pos_T *pos;
135 int addspaces; /* change the text to achieve our goal? */
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +0000136 int finetune; /* change char offset for the exact column */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000137 colnr_T wcol; /* column to move to */
138{
139#endif
140 int idx;
141 char_u *ptr;
142 char_u *line;
143 colnr_T col = 0;
144 int csize = 0;
145 int one_more;
146#ifdef FEAT_LINEBREAK
147 int head = 0;
148#endif
149
Bram Moolenaarefd2bf12006-03-16 21:41:35 +0000150 one_more = (State & INSERT)
151 || restart_edit != NUL
Bram Moolenaar071d4272004-06-13 20:20:40 +0000152#ifdef FEAT_VISUAL
Bram Moolenaarefd2bf12006-03-16 21:41:35 +0000153 || (VIsual_active && *p_sel != 'o')
Bram Moolenaar071d4272004-06-13 20:20:40 +0000154#endif
Bram Moolenaarefd2bf12006-03-16 21:41:35 +0000155#ifdef FEAT_VIRTUALEDIT
Bram Moolenaar97b2ad32006-03-18 21:40:56 +0000156 || ((ve_flags & VE_ONEMORE) && wcol < MAXCOL)
Bram Moolenaarefd2bf12006-03-16 21:41:35 +0000157#endif
158 ;
Bram Moolenaara1381de2009-11-03 15:44:21 +0000159 line = ml_get_buf(curbuf, pos->lnum, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000160
161 if (wcol >= MAXCOL)
162 {
163 idx = (int)STRLEN(line) - 1 + one_more;
164 col = wcol;
165
166#ifdef FEAT_VIRTUALEDIT
167 if ((addspaces || finetune) && !VIsual_active)
168 {
169 curwin->w_curswant = linetabsize(line) + one_more;
170 if (curwin->w_curswant > 0)
171 --curwin->w_curswant;
172 }
173#endif
174 }
175 else
176 {
177#ifdef FEAT_VIRTUALEDIT
178 int width = W_WIDTH(curwin) - win_col_off(curwin);
179
Bram Moolenaarebefac62005-12-28 22:39:57 +0000180 if (finetune
Bram Moolenaar071d4272004-06-13 20:20:40 +0000181 && curwin->w_p_wrap
182# ifdef FEAT_VERTSPLIT
183 && curwin->w_width != 0
184# endif
185 && wcol >= (colnr_T)width)
186 {
187 csize = linetabsize(line);
188 if (csize > 0)
189 csize--;
190
Bram Moolenaarebefac62005-12-28 22:39:57 +0000191 if (wcol / width > (colnr_T)csize / width
192 && ((State & INSERT) == 0 || (int)wcol > csize + 1))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000193 {
194 /* In case of line wrapping don't move the cursor beyond the
Bram Moolenaarebefac62005-12-28 22:39:57 +0000195 * right screen edge. In Insert mode allow going just beyond
196 * the last character (like what happens when typing and
197 * reaching the right window edge). */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000198 wcol = (csize / width + 1) * width - 1;
199 }
200 }
201#endif
202
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
Bram Moolenaara1381de2009-11-03 15:44:21 +0000335 /* prevent from moving onto a trail byte */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000336 if (has_mbyte)
Bram Moolenaara1381de2009-11-03 15:44:21 +0000337 mb_adjustpos(pos);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000338#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
Bram Moolenaar552c8a52009-03-11 16:29:20 +0000499 colnr_T oldcol = curwin->w_cursor.col;
500 colnr_T oldcoladd = curwin->w_cursor.col + curwin->w_cursor.coladd;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000501#endif
502
503 len = (colnr_T)STRLEN(ml_get_curline());
504 if (len == 0)
505 curwin->w_cursor.col = 0;
506 else if (curwin->w_cursor.col >= len)
507 {
Bram Moolenaar33f54432008-01-04 20:25:44 +0000508 /* Allow cursor past end-of-line when:
509 * - in Insert mode or restarting Insert mode
510 * - in Visual mode and 'selection' isn't "old"
511 * - 'virtualedit' is set */
Bram Moolenaarebefac62005-12-28 22:39:57 +0000512 if ((State & INSERT) || restart_edit
Bram Moolenaar071d4272004-06-13 20:20:40 +0000513#ifdef FEAT_VISUAL
514 || (VIsual_active && *p_sel != 'o')
515#endif
Bram Moolenaar33f54432008-01-04 20:25:44 +0000516#ifdef FEAT_VIRTUALEDIT
517 || (ve_flags & VE_ONEMORE)
518#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000519 || virtual_active())
520 curwin->w_cursor.col = len;
521 else
Bram Moolenaar87c19962007-04-26 08:54:21 +0000522 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000523 curwin->w_cursor.col = len - 1;
Bram Moolenaar87c19962007-04-26 08:54:21 +0000524#ifdef FEAT_MBYTE
525 /* prevent cursor from moving on the trail byte */
526 if (has_mbyte)
527 mb_adjust_cursor();
528#endif
529 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000530 }
531
532#ifdef FEAT_VIRTUALEDIT
533 /* If virtual editing is on, we can leave the cursor on the old position,
534 * only we must set it to virtual. But don't do it when at the end of the
535 * line. */
536 if (oldcol == MAXCOL)
537 curwin->w_cursor.coladd = 0;
538 else if (ve_flags == VE_ALL)
Bram Moolenaar552c8a52009-03-11 16:29:20 +0000539 {
540 if (oldcoladd > curwin->w_cursor.col)
541 curwin->w_cursor.coladd = oldcoladd - curwin->w_cursor.col;
542 else
543 /* avoid weird number when there is a miscalculation or overflow */
544 curwin->w_cursor.coladd = 0;
545 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000546#endif
547}
548
549/*
550 * make sure curwin->w_cursor in on a valid character
551 */
552 void
553check_cursor()
554{
555 check_cursor_lnum();
556 check_cursor_col();
557}
558
559#if defined(FEAT_TEXTOBJ) || defined(PROTO)
560/*
561 * Make sure curwin->w_cursor is not on the NUL at the end of the line.
562 * Allow it when in Visual mode and 'selection' is not "old".
563 */
564 void
565adjust_cursor_col()
566{
567 if (curwin->w_cursor.col > 0
568# ifdef FEAT_VISUAL
569 && (!VIsual_active || *p_sel == 'o')
570# endif
571 && gchar_cursor() == NUL)
572 --curwin->w_cursor.col;
573}
574#endif
575
576/*
577 * When curwin->w_leftcol has changed, adjust the cursor position.
578 * Return TRUE if the cursor was moved.
579 */
580 int
581leftcol_changed()
582{
583 long lastcol;
584 colnr_T s, e;
585 int retval = FALSE;
586
587 changed_cline_bef_curs();
588 lastcol = curwin->w_leftcol + W_WIDTH(curwin) - curwin_col_off() - 1;
589 validate_virtcol();
590
591 /*
592 * If the cursor is right or left of the screen, move it to last or first
593 * character.
594 */
595 if (curwin->w_virtcol > (colnr_T)(lastcol - p_siso))
596 {
597 retval = TRUE;
598 coladvance((colnr_T)(lastcol - p_siso));
599 }
600 else if (curwin->w_virtcol < curwin->w_leftcol + p_siso)
601 {
602 retval = TRUE;
603 (void)coladvance((colnr_T)(curwin->w_leftcol + p_siso));
604 }
605
606 /*
607 * If the start of the character under the cursor is not on the screen,
608 * advance the cursor one more char. If this fails (last char of the
609 * line) adjust the scrolling.
610 */
611 getvvcol(curwin, &curwin->w_cursor, &s, NULL, &e);
612 if (e > (colnr_T)lastcol)
613 {
614 retval = TRUE;
615 coladvance(s - 1);
616 }
617 else if (s < curwin->w_leftcol)
618 {
619 retval = TRUE;
620 if (coladvance(e + 1) == FAIL) /* there isn't another character */
621 {
622 curwin->w_leftcol = s; /* adjust w_leftcol instead */
623 changed_cline_bef_curs();
624 }
625 }
626
627 if (retval)
628 curwin->w_set_curswant = TRUE;
629 redraw_later(NOT_VALID);
630 return retval;
631}
632
633/**********************************************************************
634 * Various routines dealing with allocation and deallocation of memory.
635 */
636
637#if defined(MEM_PROFILE) || defined(PROTO)
638
639# define MEM_SIZES 8200
640static long_u mem_allocs[MEM_SIZES];
641static long_u mem_frees[MEM_SIZES];
642static long_u mem_allocated;
643static long_u mem_freed;
644static long_u mem_peak;
645static long_u num_alloc;
646static long_u num_freed;
647
648static void mem_pre_alloc_s __ARGS((size_t *sizep));
649static void mem_pre_alloc_l __ARGS((long_u *sizep));
650static void mem_post_alloc __ARGS((void **pp, size_t size));
651static void mem_pre_free __ARGS((void **pp));
652
653 static void
654mem_pre_alloc_s(sizep)
655 size_t *sizep;
656{
657 *sizep += sizeof(size_t);
658}
659
660 static void
661mem_pre_alloc_l(sizep)
662 long_u *sizep;
663{
664 *sizep += sizeof(size_t);
665}
666
667 static void
668mem_post_alloc(pp, size)
669 void **pp;
670 size_t size;
671{
672 if (*pp == NULL)
673 return;
674 size -= sizeof(size_t);
675 *(long_u *)*pp = size;
676 if (size <= MEM_SIZES-1)
677 mem_allocs[size-1]++;
678 else
679 mem_allocs[MEM_SIZES-1]++;
680 mem_allocated += size;
681 if (mem_allocated - mem_freed > mem_peak)
682 mem_peak = mem_allocated - mem_freed;
683 num_alloc++;
684 *pp = (void *)((char *)*pp + sizeof(size_t));
685}
686
687 static void
688mem_pre_free(pp)
689 void **pp;
690{
691 long_u size;
692
693 *pp = (void *)((char *)*pp - sizeof(size_t));
694 size = *(size_t *)*pp;
695 if (size <= MEM_SIZES-1)
696 mem_frees[size-1]++;
697 else
698 mem_frees[MEM_SIZES-1]++;
699 mem_freed += size;
700 num_freed++;
701}
702
703/*
704 * called on exit via atexit()
705 */
706 void
707vim_mem_profile_dump()
708{
709 int i, j;
710
711 printf("\r\n");
712 j = 0;
713 for (i = 0; i < MEM_SIZES - 1; i++)
714 {
715 if (mem_allocs[i] || mem_frees[i])
716 {
717 if (mem_frees[i] > mem_allocs[i])
718 printf("\r\n%s", _("ERROR: "));
719 printf("[%4d / %4lu-%-4lu] ", i + 1, mem_allocs[i], mem_frees[i]);
720 j++;
721 if (j > 3)
722 {
723 j = 0;
724 printf("\r\n");
725 }
726 }
727 }
728
729 i = MEM_SIZES - 1;
730 if (mem_allocs[i])
731 {
732 printf("\r\n");
733 if (mem_frees[i] > mem_allocs[i])
734 printf(_("ERROR: "));
735 printf("[>%d / %4lu-%-4lu]", i, mem_allocs[i], mem_frees[i]);
736 }
737
738 printf(_("\n[bytes] total alloc-freed %lu-%lu, in use %lu, peak use %lu\n"),
739 mem_allocated, mem_freed, mem_allocated - mem_freed, mem_peak);
740 printf(_("[calls] total re/malloc()'s %lu, total free()'s %lu\n\n"),
741 num_alloc, num_freed);
742}
743
744#endif /* MEM_PROFILE */
745
746/*
747 * Some memory is reserved for error messages and for being able to
748 * call mf_release_all(), which needs some memory for mf_trans_add().
749 */
750#if defined(MSDOS) && !defined(DJGPP)
751# define SMALL_MEM
752# define KEEP_ROOM 8192L
753#else
754# define KEEP_ROOM (2 * 8192L)
755#endif
756
757/*
Bram Moolenaar2a329742008-04-01 12:53:43 +0000758 * Note: if unsigned is 16 bits we can only allocate up to 64K with alloc().
Bram Moolenaar071d4272004-06-13 20:20:40 +0000759 * Use lalloc for larger blocks.
760 */
761 char_u *
762alloc(size)
763 unsigned size;
764{
765 return (lalloc((long_u)size, TRUE));
766}
767
768/*
769 * Allocate memory and set all bytes to zero.
770 */
771 char_u *
772alloc_clear(size)
773 unsigned size;
774{
775 char_u *p;
776
777 p = (lalloc((long_u)size, TRUE));
778 if (p != NULL)
779 (void)vim_memset(p, 0, (size_t)size);
780 return p;
781}
782
783/*
784 * alloc() with check for maximum line length
785 */
786 char_u *
787alloc_check(size)
788 unsigned size;
789{
790#if !defined(UNIX) && !defined(__EMX__)
791 if (sizeof(int) == 2 && size > 0x7fff)
792 {
793 /* Don't hide this message */
794 emsg_silent = 0;
795 EMSG(_("E340: Line is becoming too long"));
796 return NULL;
797 }
798#endif
799 return (lalloc((long_u)size, TRUE));
800}
801
802/*
803 * Allocate memory like lalloc() and set all bytes to zero.
804 */
805 char_u *
806lalloc_clear(size, message)
807 long_u size;
808 int message;
809{
810 char_u *p;
811
812 p = (lalloc(size, message));
813 if (p != NULL)
814 (void)vim_memset(p, 0, (size_t)size);
815 return p;
816}
817
818/*
819 * Low level memory allocation function.
820 * This is used often, KEEP IT FAST!
821 */
822 char_u *
823lalloc(size, message)
824 long_u size;
825 int message;
826{
827 char_u *p; /* pointer to new storage space */
828 static int releasing = FALSE; /* don't do mf_release_all() recursive */
829 int try_again;
830#if defined(HAVE_AVAIL_MEM) && !defined(SMALL_MEM)
831 static long_u allocated = 0; /* allocated since last avail check */
832#endif
833
834 /* Safety check for allocating zero bytes */
835 if (size == 0)
836 {
837 /* Don't hide this message */
838 emsg_silent = 0;
839 EMSGN(_("E341: Internal error: lalloc(%ld, )"), size);
840 return NULL;
841 }
842
843#ifdef MEM_PROFILE
844 mem_pre_alloc_l(&size);
845#endif
846
847#if defined(MSDOS) && !defined(DJGPP)
848 if (size >= 0xfff0) /* in MSDOS we can't deal with >64K blocks */
849 p = NULL;
850 else
851#endif
852
853 /*
854 * Loop when out of memory: Try to release some memfile blocks and
855 * if some blocks are released call malloc again.
856 */
857 for (;;)
858 {
859 /*
860 * Handle three kind of systems:
861 * 1. No check for available memory: Just return.
862 * 2. Slow check for available memory: call mch_avail_mem() after
863 * allocating KEEP_ROOM amount of memory.
864 * 3. Strict check for available memory: call mch_avail_mem()
865 */
866 if ((p = (char_u *)malloc((size_t)size)) != NULL)
867 {
868#ifndef HAVE_AVAIL_MEM
869 /* 1. No check for available memory: Just return. */
870 goto theend;
871#else
872# ifndef SMALL_MEM
873 /* 2. Slow check for available memory: call mch_avail_mem() after
874 * allocating (KEEP_ROOM / 2) amount of memory. */
875 allocated += size;
876 if (allocated < KEEP_ROOM / 2)
877 goto theend;
878 allocated = 0;
879# endif
880 /* 3. check for available memory: call mch_avail_mem() */
881 if (mch_avail_mem(TRUE) < KEEP_ROOM && !releasing)
882 {
Bram Moolenaar5a221812008-11-12 12:08:45 +0000883 free((char *)p); /* System is low... no go! */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000884 p = NULL;
885 }
886 else
887 goto theend;
888#endif
889 }
890 /*
891 * Remember that mf_release_all() is being called to avoid an endless
892 * loop, because mf_release_all() may call alloc() recursively.
893 */
894 if (releasing)
895 break;
896 releasing = TRUE;
Bram Moolenaar661b1822005-07-28 22:36:45 +0000897
898 clear_sb_text(); /* free any scrollback text */
899 try_again = mf_release_all(); /* release as many blocks as possible */
Bram Moolenaar39a58ca2005-06-27 22:42:44 +0000900#ifdef FEAT_EVAL
Bram Moolenaar661b1822005-07-28 22:36:45 +0000901 try_again |= garbage_collect(); /* cleanup recursive lists/dicts */
Bram Moolenaar39a58ca2005-06-27 22:42:44 +0000902#endif
Bram Moolenaar661b1822005-07-28 22:36:45 +0000903
Bram Moolenaar071d4272004-06-13 20:20:40 +0000904 releasing = FALSE;
905 if (!try_again)
906 break;
907 }
908
909 if (message && p == NULL)
910 do_outofmem_msg(size);
911
912theend:
913#ifdef MEM_PROFILE
914 mem_post_alloc((void **)&p, (size_t)size);
915#endif
916 return p;
917}
918
919#if defined(MEM_PROFILE) || defined(PROTO)
920/*
921 * realloc() with memory profiling.
922 */
923 void *
924mem_realloc(ptr, size)
925 void *ptr;
926 size_t size;
927{
928 void *p;
929
930 mem_pre_free(&ptr);
931 mem_pre_alloc_s(&size);
932
933 p = realloc(ptr, size);
934
935 mem_post_alloc(&p, size);
936
937 return p;
938}
939#endif
940
941/*
942* Avoid repeating the error message many times (they take 1 second each).
943* Did_outofmem_msg is reset when a character is read.
944*/
945 void
946do_outofmem_msg(size)
947 long_u size;
948{
949 if (!did_outofmem_msg)
950 {
951 /* Don't hide this message */
952 emsg_silent = 0;
953 EMSGN(_("E342: Out of memory! (allocating %lu bytes)"), size);
954 did_outofmem_msg = TRUE;
955 }
956}
957
Bram Moolenaar1ec484f2005-06-24 23:07:47 +0000958#if defined(EXITFREE) || defined(PROTO)
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000959
960# if defined(FEAT_SEARCHPATH)
961static void free_findfile __ARGS((void));
962# endif
963
Bram Moolenaar1ec484f2005-06-24 23:07:47 +0000964/*
965 * Free everything that we allocated.
966 * Can be used to detect memory leaks, e.g., with ccmalloc.
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000967 * NOTE: This is tricky! Things are freed that functions depend on. Don't be
968 * surprised if Vim crashes...
969 * Some things can't be freed, esp. things local to a library function.
Bram Moolenaar1ec484f2005-06-24 23:07:47 +0000970 */
971 void
972free_all_mem()
973{
974 buf_T *buf, *nextbuf;
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000975 static int entered = FALSE;
976
977 /* When we cause a crash here it is caught and Vim tries to exit cleanly.
978 * Don't try freeing everything again. */
979 if (entered)
980 return;
981 entered = TRUE;
Bram Moolenaar1ec484f2005-06-24 23:07:47 +0000982
Bram Moolenaare0ca7b22007-11-24 20:28:24 +0000983# ifdef FEAT_AUTOCMD
Bram Moolenaar78ab3312007-09-29 12:16:41 +0000984 block_autocmds(); /* don't want to trigger autocommands here */
Bram Moolenaare0ca7b22007-11-24 20:28:24 +0000985# endif
Bram Moolenaar1ec484f2005-06-24 23:07:47 +0000986
Bram Moolenaare0ca7b22007-11-24 20:28:24 +0000987# ifdef FEAT_WINDOWS
Bram Moolenaar997fb4b2006-02-17 21:53:23 +0000988 /* close all tabs and windows */
Bram Moolenaar7d47b6e2006-03-15 22:59:18 +0000989 if (first_tabpage->tp_next != NULL)
990 do_cmdline_cmd((char_u *)"tabonly!");
991 if (firstwin != lastwin)
992 do_cmdline_cmd((char_u *)"only!");
Bram Moolenaare0ca7b22007-11-24 20:28:24 +0000993# endif
Bram Moolenaar997fb4b2006-02-17 21:53:23 +0000994
Bram Moolenaarc4956c82006-03-12 21:58:43 +0000995# if defined(FEAT_SPELL)
Bram Moolenaar1ec484f2005-06-24 23:07:47 +0000996 /* Free all spell info. */
997 spell_free_all();
998# endif
999
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001000# if defined(FEAT_USR_CMDS)
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001001 /* Clear user commands (before deleting buffers). */
1002 ex_comclear(NULL);
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001003# endif
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001004
1005# ifdef FEAT_MENU
1006 /* Clear menus. */
1007 do_cmdline_cmd((char_u *)"aunmenu *");
Bram Moolenaar21160b92009-11-11 15:56:10 +00001008# ifdef FEAT_MULTI_LANG
1009 do_cmdline_cmd((char_u *)"menutranslate clear");
1010# endif
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001011# endif
1012
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001013 /* Clear mappings, abbreviations, breakpoints. */
Bram Moolenaar21160b92009-11-11 15:56:10 +00001014 do_cmdline_cmd((char_u *)"lmapclear");
1015 do_cmdline_cmd((char_u *)"xmapclear");
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001016 do_cmdline_cmd((char_u *)"mapclear");
1017 do_cmdline_cmd((char_u *)"mapclear!");
1018 do_cmdline_cmd((char_u *)"abclear");
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001019# if defined(FEAT_EVAL)
1020 do_cmdline_cmd((char_u *)"breakdel *");
1021# endif
Bram Moolenaar1e498f52005-06-26 22:29:44 +00001022# if defined(FEAT_PROFILE)
1023 do_cmdline_cmd((char_u *)"profdel *");
1024# endif
Bram Moolenaarde0dfed2009-02-24 03:30:14 +00001025# if defined(FEAT_KEYMAP)
1026 do_cmdline_cmd((char_u *)"set keymap=");
1027#endif
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001028
1029# ifdef FEAT_TITLE
1030 free_titles();
1031# endif
1032# if defined(FEAT_SEARCHPATH)
1033 free_findfile();
1034# endif
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001035
1036 /* Obviously named calls. */
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001037# if defined(FEAT_AUTOCMD)
1038 free_all_autocmds();
1039# endif
1040 clear_termcodes();
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001041 free_all_options();
1042 free_all_marks();
1043 alist_clear(&global_alist);
1044 free_homedir();
1045 free_search_patterns();
1046 free_old_sub();
1047 free_last_insert();
1048 free_prev_shellcmd();
1049 free_regexp_stuff();
1050 free_tag_stuff();
1051 free_cd_dir();
Bram Moolenaarde0dfed2009-02-24 03:30:14 +00001052# ifdef FEAT_SIGNS
1053 free_signs();
1054# endif
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001055# ifdef FEAT_EVAL
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001056 set_expr_line(NULL);
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001057# endif
1058# ifdef FEAT_DIFF
Bram Moolenaar997fb4b2006-02-17 21:53:23 +00001059 diff_clear(curtab);
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001060# endif
Bram Moolenaara40ceaf2006-01-13 22:35:40 +00001061 clear_sb_text(); /* free any scrollback text */
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001062
1063 /* Free some global vars. */
1064 vim_free(username);
Bram Moolenaaraf92ee82007-10-07 13:45:08 +00001065# ifdef FEAT_CLIPBOARD
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001066 vim_free(clip_exclude_prog);
Bram Moolenaaraf92ee82007-10-07 13:45:08 +00001067# endif
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001068 vim_free(last_cmdline);
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001069# ifdef FEAT_CMDHIST
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001070 vim_free(new_last_cmdline);
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001071# endif
Bram Moolenaar238a5642006-02-21 22:12:05 +00001072 set_keep_msg(NULL, 0);
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001073 vim_free(ff_expand_buffer);
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001074
1075 /* Clear cmdline history. */
1076 p_hi = 0;
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001077# ifdef FEAT_CMDHIST
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001078 init_history();
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001079# endif
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001080
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001081#ifdef FEAT_QUICKFIX
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001082 {
Bram Moolenaarbd1e5d22009-04-29 09:02:44 +00001083 win_T *win;
1084 tabpage_T *tab;
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001085
1086 qf_free_all(NULL);
1087 /* Free all location lists */
Bram Moolenaarbd1e5d22009-04-29 09:02:44 +00001088 FOR_ALL_TAB_WINDOWS(tab, win)
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001089 qf_free_all(win);
1090 }
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001091#endif
1092
1093 /* Close all script inputs. */
1094 close_all_scripts();
1095
1096#if defined(FEAT_WINDOWS)
1097 /* Destroy all windows. Must come before freeing buffers. */
1098 win_free_all();
1099#endif
1100
Bram Moolenaar2a329742008-04-01 12:53:43 +00001101 /* Free all buffers. Reset 'autochdir' to avoid accessing things that
1102 * were freed already. */
1103#ifdef FEAT_AUTOCHDIR
1104 p_acd = FALSE;
1105#endif
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001106 for (buf = firstbuf; buf != NULL; )
1107 {
1108 nextbuf = buf->b_next;
1109 close_buffer(NULL, buf, DOBUF_WIPE);
1110 if (buf_valid(buf))
1111 buf = nextbuf; /* didn't work, try next one */
1112 else
1113 buf = firstbuf;
1114 }
1115
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001116#ifdef FEAT_ARABIC
1117 free_cmdline_buf();
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001118#endif
1119
1120 /* Clear registers. */
1121 clear_registers();
1122 ResetRedobuff();
1123 ResetRedobuff();
1124
Bram Moolenaar85c79d32007-02-20 01:59:20 +00001125#if defined(FEAT_CLIENTSERVER) && defined(FEAT_X11)
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001126 vim_free(serverDelayedStartName);
1127#endif
1128
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001129 /* highlight info */
1130 free_highlight();
1131
Bram Moolenaar1e498f52005-06-26 22:29:44 +00001132 reset_last_sourcing();
1133
Bram Moolenaar7d47b6e2006-03-15 22:59:18 +00001134#ifdef FEAT_WINDOWS
Bram Moolenaar06a89a52006-04-29 22:01:03 +00001135 free_tabpage(first_tabpage);
1136 first_tabpage = NULL;
Bram Moolenaar7d47b6e2006-03-15 22:59:18 +00001137#endif
1138
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001139# ifdef UNIX
1140 /* Machine-specific free. */
1141 mch_free_mem();
1142# endif
1143
1144 /* message history */
1145 for (;;)
1146 if (delete_first_msg() == FAIL)
1147 break;
1148
1149# ifdef FEAT_EVAL
1150 eval_clear();
1151# endif
1152
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001153 free_termoptions();
1154
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001155 /* screenlines (can't display anything now!) */
1156 free_screenlines();
1157
1158#if defined(USE_XSMP)
1159 xsmp_close();
1160#endif
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001161#ifdef FEAT_GUI_GTK
1162 gui_mch_free_all();
1163#endif
1164 clear_hl_tables();
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001165
1166 vim_free(IObuff);
1167 vim_free(NameBuff);
1168}
1169#endif
1170
Bram Moolenaar071d4272004-06-13 20:20:40 +00001171/*
1172 * copy a string into newly allocated memory
1173 */
1174 char_u *
1175vim_strsave(string)
1176 char_u *string;
1177{
1178 char_u *p;
1179 unsigned len;
1180
1181 len = (unsigned)STRLEN(string) + 1;
1182 p = alloc(len);
1183 if (p != NULL)
1184 mch_memmove(p, string, (size_t)len);
1185 return p;
1186}
1187
1188 char_u *
1189vim_strnsave(string, len)
1190 char_u *string;
1191 int len;
1192{
1193 char_u *p;
1194
1195 p = alloc((unsigned)(len + 1));
1196 if (p != NULL)
1197 {
1198 STRNCPY(p, string, len);
1199 p[len] = NUL;
1200 }
1201 return p;
1202}
1203
Bram Moolenaar071d4272004-06-13 20:20:40 +00001204/*
1205 * Same as vim_strsave(), but any characters found in esc_chars are preceded
1206 * by a backslash.
1207 */
1208 char_u *
1209vim_strsave_escaped(string, esc_chars)
1210 char_u *string;
1211 char_u *esc_chars;
1212{
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001213 return vim_strsave_escaped_ext(string, esc_chars, '\\', FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001214}
1215
1216/*
1217 * Same as vim_strsave_escaped(), but when "bsl" is TRUE also escape
1218 * characters where rem_backslash() would remove the backslash.
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001219 * Escape the characters with "cc".
Bram Moolenaar071d4272004-06-13 20:20:40 +00001220 */
1221 char_u *
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001222vim_strsave_escaped_ext(string, esc_chars, cc, bsl)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001223 char_u *string;
1224 char_u *esc_chars;
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001225 int cc;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001226 int bsl;
1227{
1228 char_u *p;
1229 char_u *p2;
1230 char_u *escaped_string;
1231 unsigned length;
1232#ifdef FEAT_MBYTE
1233 int l;
1234#endif
1235
1236 /*
1237 * First count the number of backslashes required.
1238 * Then allocate the memory and insert them.
1239 */
1240 length = 1; /* count the trailing NUL */
1241 for (p = string; *p; p++)
1242 {
1243#ifdef FEAT_MBYTE
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001244 if (has_mbyte && (l = (*mb_ptr2len)(p)) > 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001245 {
1246 length += l; /* count a multibyte char */
1247 p += l - 1;
1248 continue;
1249 }
1250#endif
1251 if (vim_strchr(esc_chars, *p) != NULL || (bsl && rem_backslash(p)))
1252 ++length; /* count a backslash */
1253 ++length; /* count an ordinary char */
1254 }
1255 escaped_string = alloc(length);
1256 if (escaped_string != NULL)
1257 {
1258 p2 = escaped_string;
1259 for (p = string; *p; p++)
1260 {
1261#ifdef FEAT_MBYTE
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001262 if (has_mbyte && (l = (*mb_ptr2len)(p)) > 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001263 {
1264 mch_memmove(p2, p, (size_t)l);
1265 p2 += l;
1266 p += l - 1; /* skip multibyte char */
1267 continue;
1268 }
1269#endif
1270 if (vim_strchr(esc_chars, *p) != NULL || (bsl && rem_backslash(p)))
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001271 *p2++ = cc;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001272 *p2++ = *p;
1273 }
1274 *p2 = NUL;
1275 }
1276 return escaped_string;
1277}
1278
Bram Moolenaar7693ec62008-07-24 18:29:37 +00001279/*
1280 * Return TRUE when 'shell' has "csh" in the tail.
1281 */
1282 int
1283csh_like_shell()
1284{
1285 return (strstr((char *)gettail(p_sh), "csh") != NULL);
1286}
Bram Moolenaar7693ec62008-07-24 18:29:37 +00001287
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001288/*
1289 * Escape "string" for use as a shell argument with system().
Bram Moolenaar21160b92009-11-11 15:56:10 +00001290 * This uses single quotes, except when we know we need to use double quotes
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001291 * (MS-DOS and MS-Windows without 'shellslash' set).
Bram Moolenaarffd66c42008-07-16 20:43:37 +00001292 * Escape a newline, depending on the 'shell' option.
1293 * When "do_special" is TRUE also replace "!", "%", "#" and things starting
1294 * with "<" like "<cfile>".
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001295 * Returns the result in allocated memory, NULL if we have run out.
1296 */
1297 char_u *
Bram Moolenaar05bb9532008-07-04 09:44:11 +00001298vim_strsave_shellescape(string, do_special)
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001299 char_u *string;
Bram Moolenaar05bb9532008-07-04 09:44:11 +00001300 int do_special;
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001301{
1302 unsigned length;
1303 char_u *p;
1304 char_u *d;
1305 char_u *escaped_string;
Bram Moolenaar05bb9532008-07-04 09:44:11 +00001306 int l;
Bram Moolenaarffd66c42008-07-16 20:43:37 +00001307 int csh_like;
1308
1309 /* Only csh and similar shells expand '!' within single quotes. For sh and
1310 * the like we must not put a backslash before it, it will be taken
1311 * literally. If do_special is set the '!' will be escaped twice.
1312 * Csh also needs to have "\n" escaped twice when do_special is set. */
Bram Moolenaar7693ec62008-07-24 18:29:37 +00001313 csh_like = csh_like_shell();
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001314
1315 /* First count the number of extra bytes required. */
Bram Moolenaar77f66d62007-02-20 02:16:18 +00001316 length = (unsigned)STRLEN(string) + 3; /* two quotes and a trailing NUL */
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001317 for (p = string; *p != NUL; mb_ptr_adv(p))
1318 {
1319# if defined(WIN32) || defined(WIN16) || defined(DOS)
1320 if (!p_ssl)
1321 {
1322 if (*p == '"')
1323 ++length; /* " -> "" */
1324 }
1325 else
1326# endif
1327 if (*p == '\'')
1328 length += 3; /* ' => '\'' */
Bram Moolenaarffd66c42008-07-16 20:43:37 +00001329 if (*p == '\n' || (*p == '!' && (csh_like || do_special)))
1330 {
1331 ++length; /* insert backslash */
1332 if (csh_like && do_special)
1333 ++length; /* insert backslash */
1334 }
Bram Moolenaar05bb9532008-07-04 09:44:11 +00001335 if (do_special && find_cmdline_var(p, &l) >= 0)
1336 {
1337 ++length; /* insert backslash */
1338 p += l - 1;
1339 }
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001340 }
1341
1342 /* Allocate memory for the result and fill it. */
1343 escaped_string = alloc(length);
1344 if (escaped_string != NULL)
1345 {
1346 d = escaped_string;
1347
1348 /* add opening quote */
1349# if defined(WIN32) || defined(WIN16) || defined(DOS)
1350 if (!p_ssl)
1351 *d++ = '"';
1352 else
1353# endif
1354 *d++ = '\'';
1355
1356 for (p = string; *p != NUL; )
1357 {
1358# if defined(WIN32) || defined(WIN16) || defined(DOS)
1359 if (!p_ssl)
1360 {
1361 if (*p == '"')
1362 {
1363 *d++ = '"';
1364 *d++ = '"';
1365 ++p;
1366 continue;
1367 }
1368 }
1369 else
1370# endif
1371 if (*p == '\'')
1372 {
Bram Moolenaar05bb9532008-07-04 09:44:11 +00001373 *d++ = '\'';
1374 *d++ = '\\';
1375 *d++ = '\'';
1376 *d++ = '\'';
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001377 ++p;
1378 continue;
1379 }
Bram Moolenaarffd66c42008-07-16 20:43:37 +00001380 if (*p == '\n' || (*p == '!' && (csh_like || do_special)))
1381 {
1382 *d++ = '\\';
1383 if (csh_like && do_special)
1384 *d++ = '\\';
1385 *d++ = *p++;
1386 continue;
1387 }
Bram Moolenaar05bb9532008-07-04 09:44:11 +00001388 if (do_special && find_cmdline_var(p, &l) >= 0)
1389 {
1390 *d++ = '\\'; /* insert backslash */
1391 while (--l >= 0) /* copy the var */
1392 *d++ = *p++;
1393 }
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001394
1395 MB_COPY_CHAR(p, d);
1396 }
1397
1398 /* add terminating quote and finish with a NUL */
1399# if defined(WIN32) || defined(WIN16) || defined(DOS)
1400 if (!p_ssl)
1401 *d++ = '"';
1402 else
1403# endif
1404 *d++ = '\'';
1405 *d = NUL;
1406 }
1407
1408 return escaped_string;
1409}
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001410
Bram Moolenaar071d4272004-06-13 20:20:40 +00001411/*
1412 * Like vim_strsave(), but make all characters uppercase.
1413 * This uses ASCII lower-to-upper case translation, language independent.
1414 */
1415 char_u *
1416vim_strsave_up(string)
1417 char_u *string;
1418{
1419 char_u *p1;
1420
1421 p1 = vim_strsave(string);
1422 vim_strup(p1);
1423 return p1;
1424}
1425
1426/*
1427 * Like vim_strnsave(), but make all characters uppercase.
1428 * This uses ASCII lower-to-upper case translation, language independent.
1429 */
1430 char_u *
1431vim_strnsave_up(string, len)
1432 char_u *string;
1433 int len;
1434{
1435 char_u *p1;
1436
1437 p1 = vim_strnsave(string, len);
1438 vim_strup(p1);
1439 return p1;
1440}
1441
1442/*
1443 * ASCII lower-to-upper case translation, language independent.
1444 */
1445 void
1446vim_strup(p)
1447 char_u *p;
1448{
1449 char_u *p2;
1450 int c;
1451
1452 if (p != NULL)
1453 {
1454 p2 = p;
1455 while ((c = *p2) != NUL)
1456#ifdef EBCDIC
1457 *p2++ = isalpha(c) ? toupper(c) : c;
1458#else
1459 *p2++ = (c < 'a' || c > 'z') ? c : (c - 0x20);
1460#endif
1461 }
1462}
1463
Bram Moolenaarc4956c82006-03-12 21:58:43 +00001464#if defined(FEAT_EVAL) || defined(FEAT_SPELL) || defined(PROTO)
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001465/*
1466 * Make string "s" all upper-case and return it in allocated memory.
1467 * Handles multi-byte characters as well as possible.
1468 * Returns NULL when out of memory.
1469 */
1470 char_u *
1471strup_save(orig)
1472 char_u *orig;
1473{
1474 char_u *p;
1475 char_u *res;
1476
1477 res = p = vim_strsave(orig);
1478
1479 if (res != NULL)
1480 while (*p != NUL)
1481 {
1482# ifdef FEAT_MBYTE
1483 int l;
1484
1485 if (enc_utf8)
1486 {
1487 int c, uc;
1488 int nl;
1489 char_u *s;
1490
1491 c = utf_ptr2char(p);
1492 uc = utf_toupper(c);
1493
1494 /* Reallocate string when byte count changes. This is rare,
1495 * thus it's OK to do another malloc()/free(). */
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001496 l = utf_ptr2len(p);
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001497 nl = utf_char2len(uc);
1498 if (nl != l)
1499 {
1500 s = alloc((unsigned)STRLEN(res) + 1 + nl - l);
1501 if (s == NULL)
1502 break;
1503 mch_memmove(s, res, p - res);
1504 STRCPY(s + (p - res) + nl, p + l);
1505 p = s + (p - res);
1506 vim_free(res);
1507 res = s;
1508 }
1509
1510 utf_char2bytes(uc, p);
1511 p += nl;
1512 }
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001513 else if (has_mbyte && (l = (*mb_ptr2len)(p)) > 1)
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001514 p += l; /* skip multi-byte character */
1515 else
1516# endif
1517 {
1518 *p = TOUPPER_LOC(*p); /* note that toupper() can be a macro */
1519 p++;
1520 }
1521 }
1522
1523 return res;
1524}
1525#endif
1526
Bram Moolenaar071d4272004-06-13 20:20:40 +00001527/*
1528 * copy a space a number of times
1529 */
1530 void
1531copy_spaces(ptr, count)
1532 char_u *ptr;
1533 size_t count;
1534{
1535 size_t i = count;
1536 char_u *p = ptr;
1537
1538 while (i--)
1539 *p++ = ' ';
1540}
1541
1542#if defined(FEAT_VISUALEXTRA) || defined(PROTO)
1543/*
1544 * Copy a character a number of times.
Bram Moolenaar21160b92009-11-11 15:56:10 +00001545 * Does not work for multi-byte characters!
Bram Moolenaar071d4272004-06-13 20:20:40 +00001546 */
1547 void
1548copy_chars(ptr, count, c)
1549 char_u *ptr;
1550 size_t count;
1551 int c;
1552{
1553 size_t i = count;
1554 char_u *p = ptr;
1555
1556 while (i--)
1557 *p++ = c;
1558}
1559#endif
1560
1561/*
1562 * delete spaces at the end of a string
1563 */
1564 void
1565del_trailing_spaces(ptr)
1566 char_u *ptr;
1567{
1568 char_u *q;
1569
1570 q = ptr + STRLEN(ptr);
1571 while (--q > ptr && vim_iswhite(q[0]) && q[-1] != '\\' && q[-1] != Ctrl_V)
1572 *q = NUL;
1573}
1574
1575/*
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001576 * Like strncpy(), but always terminate the result with one NUL.
Bram Moolenaard042c562005-06-30 22:04:15 +00001577 * "to" must be "len + 1" long!
Bram Moolenaar071d4272004-06-13 20:20:40 +00001578 */
1579 void
1580vim_strncpy(to, from, len)
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001581 char_u *to;
1582 char_u *from;
Bram Moolenaarbbebc852005-07-18 21:47:53 +00001583 size_t len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001584{
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001585 STRNCPY(to, from, len);
1586 to[len] = NUL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001587}
1588
1589/*
1590 * Isolate one part of a string option where parts are separated with
1591 * "sep_chars".
Bram Moolenaar83bab712005-08-01 21:58:57 +00001592 * The part is copied into "buf[maxlen]".
Bram Moolenaar071d4272004-06-13 20:20:40 +00001593 * "*option" is advanced to the next part.
1594 * The length is returned.
1595 */
1596 int
1597copy_option_part(option, buf, maxlen, sep_chars)
1598 char_u **option;
1599 char_u *buf;
1600 int maxlen;
1601 char *sep_chars;
1602{
1603 int len = 0;
1604 char_u *p = *option;
1605
1606 /* skip '.' at start of option part, for 'suffixes' */
1607 if (*p == '.')
1608 buf[len++] = *p++;
1609 while (*p != NUL && vim_strchr((char_u *)sep_chars, *p) == NULL)
1610 {
1611 /*
1612 * Skip backslash before a separator character and space.
1613 */
1614 if (p[0] == '\\' && vim_strchr((char_u *)sep_chars, p[1]) != NULL)
1615 ++p;
1616 if (len < maxlen - 1)
1617 buf[len++] = *p;
1618 ++p;
1619 }
1620 buf[len] = NUL;
1621
1622 if (*p != NUL && *p != ',') /* skip non-standard separator */
1623 ++p;
1624 p = skip_to_option_part(p); /* p points to next file name */
1625
1626 *option = p;
1627 return len;
1628}
1629
1630/*
Bram Moolenaar4770d092006-01-12 23:22:24 +00001631 * Replacement for free() that ignores NULL pointers.
1632 * Also skip free() when exiting for sure, this helps when we caught a deadly
1633 * signal that was caused by a crash in free().
Bram Moolenaar071d4272004-06-13 20:20:40 +00001634 */
1635 void
1636vim_free(x)
1637 void *x;
1638{
Bram Moolenaar4770d092006-01-12 23:22:24 +00001639 if (x != NULL && !really_exiting)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001640 {
1641#ifdef MEM_PROFILE
1642 mem_pre_free(&x);
1643#endif
1644 free(x);
1645 }
1646}
1647
1648#ifndef HAVE_MEMSET
1649 void *
1650vim_memset(ptr, c, size)
1651 void *ptr;
1652 int c;
1653 size_t size;
1654{
1655 char *p = ptr;
1656
1657 while (size-- > 0)
1658 *p++ = c;
1659 return ptr;
1660}
1661#endif
1662
1663#ifdef VIM_MEMCMP
1664/*
1665 * Return zero when "b1" and "b2" are the same for "len" bytes.
1666 * Return non-zero otherwise.
1667 */
1668 int
1669vim_memcmp(b1, b2, len)
1670 void *b1;
1671 void *b2;
1672 size_t len;
1673{
1674 char_u *p1 = (char_u *)b1, *p2 = (char_u *)b2;
1675
1676 for ( ; len > 0; --len)
1677 {
1678 if (*p1 != *p2)
1679 return 1;
1680 ++p1;
1681 ++p2;
1682 }
1683 return 0;
1684}
1685#endif
1686
1687#ifdef VIM_MEMMOVE
1688/*
1689 * Version of memmove() that handles overlapping source and destination.
1690 * For systems that don't have a function that is guaranteed to do that (SYSV).
1691 */
1692 void
1693mch_memmove(dst_arg, src_arg, len)
1694 void *src_arg, *dst_arg;
1695 size_t len;
1696{
1697 /*
1698 * A void doesn't have a size, we use char pointers.
1699 */
1700 char *dst = dst_arg, *src = src_arg;
1701
1702 /* overlap, copy backwards */
1703 if (dst > src && dst < src + len)
1704 {
1705 src += len;
1706 dst += len;
1707 while (len-- > 0)
1708 *--dst = *--src;
1709 }
1710 else /* copy forwards */
1711 while (len-- > 0)
1712 *dst++ = *src++;
1713}
1714#endif
1715
1716#if (!defined(HAVE_STRCASECMP) && !defined(HAVE_STRICMP)) || defined(PROTO)
1717/*
1718 * Compare two strings, ignoring case, using current locale.
1719 * Doesn't work for multi-byte characters.
1720 * return 0 for match, < 0 for smaller, > 0 for bigger
1721 */
1722 int
1723vim_stricmp(s1, s2)
1724 char *s1;
1725 char *s2;
1726{
1727 int i;
1728
1729 for (;;)
1730 {
1731 i = (int)TOLOWER_LOC(*s1) - (int)TOLOWER_LOC(*s2);
1732 if (i != 0)
1733 return i; /* this character different */
1734 if (*s1 == NUL)
1735 break; /* strings match until NUL */
1736 ++s1;
1737 ++s2;
1738 }
1739 return 0; /* strings match */
1740}
1741#endif
1742
1743#if (!defined(HAVE_STRNCASECMP) && !defined(HAVE_STRNICMP)) || defined(PROTO)
1744/*
1745 * Compare two strings, for length "len", ignoring case, using current locale.
1746 * Doesn't work for multi-byte characters.
1747 * return 0 for match, < 0 for smaller, > 0 for bigger
1748 */
1749 int
1750vim_strnicmp(s1, s2, len)
1751 char *s1;
1752 char *s2;
1753 size_t len;
1754{
1755 int i;
1756
1757 while (len > 0)
1758 {
1759 i = (int)TOLOWER_LOC(*s1) - (int)TOLOWER_LOC(*s2);
1760 if (i != 0)
1761 return i; /* this character different */
1762 if (*s1 == NUL)
1763 break; /* strings match until NUL */
1764 ++s1;
1765 ++s2;
1766 --len;
1767 }
1768 return 0; /* strings match */
1769}
1770#endif
1771
1772#if 0 /* currently not used */
1773/*
1774 * Check if string "s2" appears somewhere in "s1" while ignoring case.
1775 * Return NULL if not, a pointer to the first occurrence if it does.
1776 */
1777 char_u *
1778vim_stristr(s1, s2)
1779 char_u *s1;
1780 char_u *s2;
1781{
1782 char_u *p;
1783 int len = STRLEN(s2);
1784 char_u *end = s1 + STRLEN(s1) - len;
1785
1786 for (p = s1; p <= end; ++p)
1787 if (STRNICMP(p, s2, len) == 0)
1788 return p;
1789 return NULL;
1790}
1791#endif
1792
1793/*
1794 * Version of strchr() and strrchr() that handle unsigned char strings
Bram Moolenaar05159a02005-02-26 23:04:13 +00001795 * with characters from 128 to 255 correctly. It also doesn't return a
1796 * pointer to the NUL at the end of the string.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001797 */
1798 char_u *
1799vim_strchr(string, c)
1800 char_u *string;
1801 int c;
1802{
1803 char_u *p;
1804 int b;
1805
1806 p = string;
1807#ifdef FEAT_MBYTE
1808 if (enc_utf8 && c >= 0x80)
1809 {
1810 while (*p != NUL)
1811 {
1812 if (utf_ptr2char(p) == c)
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 (enc_dbcs != 0 && c > 255)
1819 {
1820 int n2 = c & 0xff;
1821
1822 c = ((unsigned)c >> 8) & 0xff;
1823 while ((b = *p) != NUL)
1824 {
1825 if (b == c && p[1] == n2)
1826 return p;
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001827 p += (*mb_ptr2len)(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001828 }
1829 return NULL;
1830 }
1831 if (has_mbyte)
1832 {
1833 while ((b = *p) != NUL)
1834 {
1835 if (b == c)
1836 return p;
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001837 p += (*mb_ptr2len)(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001838 }
1839 return NULL;
1840 }
1841#endif
1842 while ((b = *p) != NUL)
1843 {
1844 if (b == c)
1845 return p;
1846 ++p;
1847 }
1848 return NULL;
1849}
1850
1851/*
Bram Moolenaar05159a02005-02-26 23:04:13 +00001852 * Version of strchr() that only works for bytes and handles unsigned char
1853 * strings with characters above 128 correctly. It also doesn't return a
1854 * pointer to the NUL at the end of the string.
1855 */
1856 char_u *
1857vim_strbyte(string, c)
1858 char_u *string;
1859 int c;
1860{
1861 char_u *p = string;
1862
1863 while (*p != NUL)
1864 {
1865 if (*p == c)
1866 return p;
1867 ++p;
1868 }
1869 return NULL;
1870}
1871
1872/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00001873 * Search for last occurrence of "c" in "string".
Bram Moolenaarac6e65f2005-08-29 22:25:38 +00001874 * Return NULL if not found.
Bram Moolenaar05159a02005-02-26 23:04:13 +00001875 * Does not handle multi-byte char for "c"!
Bram Moolenaar071d4272004-06-13 20:20:40 +00001876 */
1877 char_u *
1878vim_strrchr(string, c)
1879 char_u *string;
1880 int c;
1881{
1882 char_u *retval = NULL;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001883 char_u *p = string;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001884
Bram Moolenaar05159a02005-02-26 23:04:13 +00001885 while (*p)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001886 {
Bram Moolenaar05159a02005-02-26 23:04:13 +00001887 if (*p == c)
1888 retval = p;
1889 mb_ptr_adv(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001890 }
1891 return retval;
1892}
1893
1894/*
1895 * Vim's version of strpbrk(), in case it's missing.
1896 * Don't generate a prototype for this, causes problems when it's not used.
1897 */
1898#ifndef PROTO
1899# ifndef HAVE_STRPBRK
1900# ifdef vim_strpbrk
1901# undef vim_strpbrk
1902# endif
1903 char_u *
1904vim_strpbrk(s, charset)
1905 char_u *s;
1906 char_u *charset;
1907{
1908 while (*s)
1909 {
1910 if (vim_strchr(charset, *s) != NULL)
1911 return s;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001912 mb_ptr_adv(s);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001913 }
1914 return NULL;
1915}
1916# endif
1917#endif
1918
1919/*
1920 * Vim has its own isspace() function, because on some machines isspace()
1921 * can't handle characters above 128.
1922 */
1923 int
1924vim_isspace(x)
1925 int x;
1926{
1927 return ((x >= 9 && x <= 13) || x == ' ');
1928}
1929
1930/************************************************************************
Bram Moolenaar383f9bc2005-01-19 22:18:32 +00001931 * Functions for handling growing arrays.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001932 */
1933
1934/*
1935 * Clear an allocated growing array.
1936 */
1937 void
1938ga_clear(gap)
1939 garray_T *gap;
1940{
1941 vim_free(gap->ga_data);
1942 ga_init(gap);
1943}
1944
1945/*
1946 * Clear a growing array that contains a list of strings.
1947 */
1948 void
1949ga_clear_strings(gap)
1950 garray_T *gap;
1951{
1952 int i;
1953
1954 for (i = 0; i < gap->ga_len; ++i)
1955 vim_free(((char_u **)(gap->ga_data))[i]);
1956 ga_clear(gap);
1957}
1958
1959/*
1960 * Initialize a growing array. Don't forget to set ga_itemsize and
1961 * ga_growsize! Or use ga_init2().
1962 */
1963 void
1964ga_init(gap)
1965 garray_T *gap;
1966{
1967 gap->ga_data = NULL;
Bram Moolenaar86b68352004-12-27 21:59:20 +00001968 gap->ga_maxlen = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001969 gap->ga_len = 0;
1970}
1971
1972 void
1973ga_init2(gap, itemsize, growsize)
1974 garray_T *gap;
1975 int itemsize;
1976 int growsize;
1977{
1978 ga_init(gap);
1979 gap->ga_itemsize = itemsize;
1980 gap->ga_growsize = growsize;
1981}
1982
1983/*
1984 * Make room in growing array "gap" for at least "n" items.
1985 * Return FAIL for failure, OK otherwise.
1986 */
1987 int
1988ga_grow(gap, n)
1989 garray_T *gap;
1990 int n;
1991{
1992 size_t len;
1993 char_u *pp;
1994
Bram Moolenaar86b68352004-12-27 21:59:20 +00001995 if (gap->ga_maxlen - gap->ga_len < n)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001996 {
1997 if (n < gap->ga_growsize)
1998 n = gap->ga_growsize;
1999 len = gap->ga_itemsize * (gap->ga_len + n);
2000 pp = alloc_clear((unsigned)len);
2001 if (pp == NULL)
2002 return FAIL;
Bram Moolenaar86b68352004-12-27 21:59:20 +00002003 gap->ga_maxlen = gap->ga_len + n;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002004 if (gap->ga_data != NULL)
2005 {
2006 mch_memmove(pp, gap->ga_data,
2007 (size_t)(gap->ga_itemsize * gap->ga_len));
2008 vim_free(gap->ga_data);
2009 }
2010 gap->ga_data = pp;
2011 }
2012 return OK;
2013}
2014
2015/*
2016 * Concatenate a string to a growarray which contains characters.
2017 * Note: Does NOT copy the NUL at the end!
2018 */
2019 void
2020ga_concat(gap, s)
2021 garray_T *gap;
2022 char_u *s;
2023{
2024 int len = (int)STRLEN(s);
2025
2026 if (ga_grow(gap, len) == OK)
2027 {
2028 mch_memmove((char *)gap->ga_data + gap->ga_len, s, (size_t)len);
2029 gap->ga_len += len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002030 }
2031}
2032
2033/*
2034 * Append one byte to a growarray which contains bytes.
2035 */
2036 void
2037ga_append(gap, c)
2038 garray_T *gap;
2039 int c;
2040{
2041 if (ga_grow(gap, 1) == OK)
2042 {
2043 *((char *)gap->ga_data + gap->ga_len) = c;
2044 ++gap->ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002045 }
2046}
2047
2048/************************************************************************
2049 * functions that use lookup tables for various things, generally to do with
2050 * special key codes.
2051 */
2052
2053/*
2054 * Some useful tables.
2055 */
2056
2057static struct modmasktable
2058{
2059 short mod_mask; /* Bit-mask for particular key modifier */
2060 short mod_flag; /* Bit(s) for particular key modifier */
2061 char_u name; /* Single letter name of modifier */
2062} mod_mask_table[] =
2063{
2064 {MOD_MASK_ALT, MOD_MASK_ALT, (char_u)'M'},
Bram Moolenaar19a09a12005-03-04 23:39:37 +00002065 {MOD_MASK_META, MOD_MASK_META, (char_u)'T'},
Bram Moolenaar071d4272004-06-13 20:20:40 +00002066 {MOD_MASK_CTRL, MOD_MASK_CTRL, (char_u)'C'},
2067 {MOD_MASK_SHIFT, MOD_MASK_SHIFT, (char_u)'S'},
2068 {MOD_MASK_MULTI_CLICK, MOD_MASK_2CLICK, (char_u)'2'},
2069 {MOD_MASK_MULTI_CLICK, MOD_MASK_3CLICK, (char_u)'3'},
2070 {MOD_MASK_MULTI_CLICK, MOD_MASK_4CLICK, (char_u)'4'},
2071#ifdef MACOS
2072 {MOD_MASK_CMD, MOD_MASK_CMD, (char_u)'D'},
2073#endif
2074 /* 'A' must be the last one */
2075 {MOD_MASK_ALT, MOD_MASK_ALT, (char_u)'A'},
2076 {0, 0, NUL}
2077};
2078
2079/*
2080 * Shifted key terminal codes and their unshifted equivalent.
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00002081 * Don't add mouse codes here, they are handled separately!
Bram Moolenaar071d4272004-06-13 20:20:40 +00002082 */
2083#define MOD_KEYS_ENTRY_SIZE 5
2084
2085static char_u modifier_keys_table[] =
2086{
2087/* mod mask with modifier without modifier */
2088 MOD_MASK_SHIFT, '&', '9', '@', '1', /* begin */
2089 MOD_MASK_SHIFT, '&', '0', '@', '2', /* cancel */
2090 MOD_MASK_SHIFT, '*', '1', '@', '4', /* command */
2091 MOD_MASK_SHIFT, '*', '2', '@', '5', /* copy */
2092 MOD_MASK_SHIFT, '*', '3', '@', '6', /* create */
2093 MOD_MASK_SHIFT, '*', '4', 'k', 'D', /* delete char */
2094 MOD_MASK_SHIFT, '*', '5', 'k', 'L', /* delete line */
2095 MOD_MASK_SHIFT, '*', '7', '@', '7', /* end */
2096 MOD_MASK_CTRL, KS_EXTRA, (int)KE_C_END, '@', '7', /* end */
2097 MOD_MASK_SHIFT, '*', '9', '@', '9', /* exit */
2098 MOD_MASK_SHIFT, '*', '0', '@', '0', /* find */
2099 MOD_MASK_SHIFT, '#', '1', '%', '1', /* help */
2100 MOD_MASK_SHIFT, '#', '2', 'k', 'h', /* home */
2101 MOD_MASK_CTRL, KS_EXTRA, (int)KE_C_HOME, 'k', 'h', /* home */
2102 MOD_MASK_SHIFT, '#', '3', 'k', 'I', /* insert */
2103 MOD_MASK_SHIFT, '#', '4', 'k', 'l', /* left arrow */
2104 MOD_MASK_CTRL, KS_EXTRA, (int)KE_C_LEFT, 'k', 'l', /* left arrow */
2105 MOD_MASK_SHIFT, '%', 'a', '%', '3', /* message */
2106 MOD_MASK_SHIFT, '%', 'b', '%', '4', /* move */
2107 MOD_MASK_SHIFT, '%', 'c', '%', '5', /* next */
2108 MOD_MASK_SHIFT, '%', 'd', '%', '7', /* options */
2109 MOD_MASK_SHIFT, '%', 'e', '%', '8', /* previous */
2110 MOD_MASK_SHIFT, '%', 'f', '%', '9', /* print */
2111 MOD_MASK_SHIFT, '%', 'g', '%', '0', /* redo */
2112 MOD_MASK_SHIFT, '%', 'h', '&', '3', /* replace */
2113 MOD_MASK_SHIFT, '%', 'i', 'k', 'r', /* right arr. */
2114 MOD_MASK_CTRL, KS_EXTRA, (int)KE_C_RIGHT, 'k', 'r', /* right arr. */
2115 MOD_MASK_SHIFT, '%', 'j', '&', '5', /* resume */
2116 MOD_MASK_SHIFT, '!', '1', '&', '6', /* save */
2117 MOD_MASK_SHIFT, '!', '2', '&', '7', /* suspend */
2118 MOD_MASK_SHIFT, '!', '3', '&', '8', /* undo */
2119 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_UP, 'k', 'u', /* up arrow */
2120 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_DOWN, 'k', 'd', /* down arrow */
2121
2122 /* vt100 F1 */
2123 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_XF1, KS_EXTRA, (int)KE_XF1,
2124 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_XF2, KS_EXTRA, (int)KE_XF2,
2125 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_XF3, KS_EXTRA, (int)KE_XF3,
2126 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_XF4, KS_EXTRA, (int)KE_XF4,
2127
2128 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F1, 'k', '1', /* F1 */
2129 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F2, 'k', '2',
2130 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F3, 'k', '3',
2131 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F4, 'k', '4',
2132 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F5, 'k', '5',
2133 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F6, 'k', '6',
2134 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F7, 'k', '7',
2135 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F8, 'k', '8',
2136 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F9, 'k', '9',
2137 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F10, 'k', ';', /* F10 */
2138
2139 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F11, 'F', '1',
2140 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F12, 'F', '2',
2141 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F13, 'F', '3',
2142 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F14, 'F', '4',
2143 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F15, 'F', '5',
2144 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F16, 'F', '6',
2145 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F17, 'F', '7',
2146 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F18, 'F', '8',
2147 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F19, 'F', '9',
2148 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F20, 'F', 'A',
2149
2150 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F21, 'F', 'B',
2151 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F22, 'F', 'C',
2152 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F23, 'F', 'D',
2153 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F24, 'F', 'E',
2154 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F25, 'F', 'F',
2155 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F26, 'F', 'G',
2156 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F27, 'F', 'H',
2157 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F28, 'F', 'I',
2158 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F29, 'F', 'J',
2159 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F30, 'F', 'K',
2160
2161 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F31, 'F', 'L',
2162 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F32, 'F', 'M',
2163 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F33, 'F', 'N',
2164 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F34, 'F', 'O',
2165 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F35, 'F', 'P',
2166 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F36, 'F', 'Q',
2167 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F37, 'F', 'R',
2168
2169 /* TAB pseudo code*/
2170 MOD_MASK_SHIFT, 'k', 'B', KS_EXTRA, (int)KE_TAB,
2171
2172 NUL
2173};
2174
2175static struct key_name_entry
2176{
2177 int key; /* Special key code or ascii value */
2178 char_u *name; /* Name of key */
2179} key_names_table[] =
2180{
2181 {' ', (char_u *)"Space"},
2182 {TAB, (char_u *)"Tab"},
2183 {K_TAB, (char_u *)"Tab"},
2184 {NL, (char_u *)"NL"},
2185 {NL, (char_u *)"NewLine"}, /* Alternative name */
2186 {NL, (char_u *)"LineFeed"}, /* Alternative name */
2187 {NL, (char_u *)"LF"}, /* Alternative name */
2188 {CAR, (char_u *)"CR"},
2189 {CAR, (char_u *)"Return"}, /* Alternative name */
2190 {CAR, (char_u *)"Enter"}, /* Alternative name */
2191 {K_BS, (char_u *)"BS"},
2192 {K_BS, (char_u *)"BackSpace"}, /* Alternative name */
2193 {ESC, (char_u *)"Esc"},
2194 {CSI, (char_u *)"CSI"},
2195 {K_CSI, (char_u *)"xCSI"},
2196 {'|', (char_u *)"Bar"},
2197 {'\\', (char_u *)"Bslash"},
2198 {K_DEL, (char_u *)"Del"},
2199 {K_DEL, (char_u *)"Delete"}, /* Alternative name */
2200 {K_KDEL, (char_u *)"kDel"},
2201 {K_UP, (char_u *)"Up"},
2202 {K_DOWN, (char_u *)"Down"},
2203 {K_LEFT, (char_u *)"Left"},
2204 {K_RIGHT, (char_u *)"Right"},
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002205 {K_XUP, (char_u *)"xUp"},
2206 {K_XDOWN, (char_u *)"xDown"},
2207 {K_XLEFT, (char_u *)"xLeft"},
2208 {K_XRIGHT, (char_u *)"xRight"},
Bram Moolenaar071d4272004-06-13 20:20:40 +00002209
2210 {K_F1, (char_u *)"F1"},
2211 {K_F2, (char_u *)"F2"},
2212 {K_F3, (char_u *)"F3"},
2213 {K_F4, (char_u *)"F4"},
2214 {K_F5, (char_u *)"F5"},
2215 {K_F6, (char_u *)"F6"},
2216 {K_F7, (char_u *)"F7"},
2217 {K_F8, (char_u *)"F8"},
2218 {K_F9, (char_u *)"F9"},
2219 {K_F10, (char_u *)"F10"},
2220
2221 {K_F11, (char_u *)"F11"},
2222 {K_F12, (char_u *)"F12"},
2223 {K_F13, (char_u *)"F13"},
2224 {K_F14, (char_u *)"F14"},
2225 {K_F15, (char_u *)"F15"},
2226 {K_F16, (char_u *)"F16"},
2227 {K_F17, (char_u *)"F17"},
2228 {K_F18, (char_u *)"F18"},
2229 {K_F19, (char_u *)"F19"},
2230 {K_F20, (char_u *)"F20"},
2231
2232 {K_F21, (char_u *)"F21"},
2233 {K_F22, (char_u *)"F22"},
2234 {K_F23, (char_u *)"F23"},
2235 {K_F24, (char_u *)"F24"},
2236 {K_F25, (char_u *)"F25"},
2237 {K_F26, (char_u *)"F26"},
2238 {K_F27, (char_u *)"F27"},
2239 {K_F28, (char_u *)"F28"},
2240 {K_F29, (char_u *)"F29"},
2241 {K_F30, (char_u *)"F30"},
2242
2243 {K_F31, (char_u *)"F31"},
2244 {K_F32, (char_u *)"F32"},
2245 {K_F33, (char_u *)"F33"},
2246 {K_F34, (char_u *)"F34"},
2247 {K_F35, (char_u *)"F35"},
2248 {K_F36, (char_u *)"F36"},
2249 {K_F37, (char_u *)"F37"},
2250
2251 {K_XF1, (char_u *)"xF1"},
2252 {K_XF2, (char_u *)"xF2"},
2253 {K_XF3, (char_u *)"xF3"},
2254 {K_XF4, (char_u *)"xF4"},
2255
2256 {K_HELP, (char_u *)"Help"},
2257 {K_UNDO, (char_u *)"Undo"},
2258 {K_INS, (char_u *)"Insert"},
2259 {K_INS, (char_u *)"Ins"}, /* Alternative name */
2260 {K_KINS, (char_u *)"kInsert"},
2261 {K_HOME, (char_u *)"Home"},
2262 {K_KHOME, (char_u *)"kHome"},
2263 {K_XHOME, (char_u *)"xHome"},
Bram Moolenaar68b76a62005-03-25 21:53:48 +00002264 {K_ZHOME, (char_u *)"zHome"},
Bram Moolenaar071d4272004-06-13 20:20:40 +00002265 {K_END, (char_u *)"End"},
2266 {K_KEND, (char_u *)"kEnd"},
2267 {K_XEND, (char_u *)"xEnd"},
Bram Moolenaar68b76a62005-03-25 21:53:48 +00002268 {K_ZEND, (char_u *)"zEnd"},
Bram Moolenaar071d4272004-06-13 20:20:40 +00002269 {K_PAGEUP, (char_u *)"PageUp"},
2270 {K_PAGEDOWN, (char_u *)"PageDown"},
2271 {K_KPAGEUP, (char_u *)"kPageUp"},
2272 {K_KPAGEDOWN, (char_u *)"kPageDown"},
2273
2274 {K_KPLUS, (char_u *)"kPlus"},
2275 {K_KMINUS, (char_u *)"kMinus"},
2276 {K_KDIVIDE, (char_u *)"kDivide"},
2277 {K_KMULTIPLY, (char_u *)"kMultiply"},
2278 {K_KENTER, (char_u *)"kEnter"},
2279 {K_KPOINT, (char_u *)"kPoint"},
2280
2281 {K_K0, (char_u *)"k0"},
2282 {K_K1, (char_u *)"k1"},
2283 {K_K2, (char_u *)"k2"},
2284 {K_K3, (char_u *)"k3"},
2285 {K_K4, (char_u *)"k4"},
2286 {K_K5, (char_u *)"k5"},
2287 {K_K6, (char_u *)"k6"},
2288 {K_K7, (char_u *)"k7"},
2289 {K_K8, (char_u *)"k8"},
2290 {K_K9, (char_u *)"k9"},
2291
2292 {'<', (char_u *)"lt"},
2293
2294 {K_MOUSE, (char_u *)"Mouse"},
2295 {K_NETTERM_MOUSE, (char_u *)"NetMouse"},
2296 {K_DEC_MOUSE, (char_u *)"DecMouse"},
2297 {K_JSBTERM_MOUSE, (char_u *)"JsbMouse"},
2298 {K_PTERM_MOUSE, (char_u *)"PtermMouse"},
2299 {K_LEFTMOUSE, (char_u *)"LeftMouse"},
2300 {K_LEFTMOUSE_NM, (char_u *)"LeftMouseNM"},
2301 {K_LEFTDRAG, (char_u *)"LeftDrag"},
2302 {K_LEFTRELEASE, (char_u *)"LeftRelease"},
2303 {K_LEFTRELEASE_NM, (char_u *)"LeftReleaseNM"},
2304 {K_MIDDLEMOUSE, (char_u *)"MiddleMouse"},
2305 {K_MIDDLEDRAG, (char_u *)"MiddleDrag"},
2306 {K_MIDDLERELEASE, (char_u *)"MiddleRelease"},
2307 {K_RIGHTMOUSE, (char_u *)"RightMouse"},
2308 {K_RIGHTDRAG, (char_u *)"RightDrag"},
2309 {K_RIGHTRELEASE, (char_u *)"RightRelease"},
2310 {K_MOUSEDOWN, (char_u *)"MouseDown"},
2311 {K_MOUSEUP, (char_u *)"MouseUp"},
2312 {K_X1MOUSE, (char_u *)"X1Mouse"},
2313 {K_X1DRAG, (char_u *)"X1Drag"},
2314 {K_X1RELEASE, (char_u *)"X1Release"},
2315 {K_X2MOUSE, (char_u *)"X2Mouse"},
2316 {K_X2DRAG, (char_u *)"X2Drag"},
2317 {K_X2RELEASE, (char_u *)"X2Release"},
2318 {K_DROP, (char_u *)"Drop"},
2319 {K_ZERO, (char_u *)"Nul"},
2320#ifdef FEAT_EVAL
2321 {K_SNR, (char_u *)"SNR"},
2322#endif
2323 {K_PLUG, (char_u *)"Plug"},
2324 {0, NULL}
2325};
2326
2327#define KEY_NAMES_TABLE_LEN (sizeof(key_names_table) / sizeof(struct key_name_entry))
2328
2329#ifdef FEAT_MOUSE
2330static struct mousetable
2331{
2332 int pseudo_code; /* Code for pseudo mouse event */
2333 int button; /* Which mouse button is it? */
2334 int is_click; /* Is it a mouse button click event? */
2335 int is_drag; /* Is it a mouse drag event? */
2336} mouse_table[] =
2337{
2338 {(int)KE_LEFTMOUSE, MOUSE_LEFT, TRUE, FALSE},
2339#ifdef FEAT_GUI
2340 {(int)KE_LEFTMOUSE_NM, MOUSE_LEFT, TRUE, FALSE},
2341#endif
2342 {(int)KE_LEFTDRAG, MOUSE_LEFT, FALSE, TRUE},
2343 {(int)KE_LEFTRELEASE, MOUSE_LEFT, FALSE, FALSE},
2344#ifdef FEAT_GUI
2345 {(int)KE_LEFTRELEASE_NM, MOUSE_LEFT, FALSE, FALSE},
2346#endif
2347 {(int)KE_MIDDLEMOUSE, MOUSE_MIDDLE, TRUE, FALSE},
2348 {(int)KE_MIDDLEDRAG, MOUSE_MIDDLE, FALSE, TRUE},
2349 {(int)KE_MIDDLERELEASE, MOUSE_MIDDLE, FALSE, FALSE},
2350 {(int)KE_RIGHTMOUSE, MOUSE_RIGHT, TRUE, FALSE},
2351 {(int)KE_RIGHTDRAG, MOUSE_RIGHT, FALSE, TRUE},
2352 {(int)KE_RIGHTRELEASE, MOUSE_RIGHT, FALSE, FALSE},
2353 {(int)KE_X1MOUSE, MOUSE_X1, TRUE, FALSE},
2354 {(int)KE_X1DRAG, MOUSE_X1, FALSE, TRUE},
2355 {(int)KE_X1RELEASE, MOUSE_X1, FALSE, FALSE},
2356 {(int)KE_X2MOUSE, MOUSE_X2, TRUE, FALSE},
2357 {(int)KE_X2DRAG, MOUSE_X2, FALSE, TRUE},
2358 {(int)KE_X2RELEASE, MOUSE_X2, FALSE, FALSE},
2359 /* DRAG without CLICK */
2360 {(int)KE_IGNORE, MOUSE_RELEASE, FALSE, TRUE},
2361 /* RELEASE without CLICK */
2362 {(int)KE_IGNORE, MOUSE_RELEASE, FALSE, FALSE},
2363 {0, 0, 0, 0},
2364};
2365#endif /* FEAT_MOUSE */
2366
2367/*
2368 * Return the modifier mask bit (MOD_MASK_*) which corresponds to the given
2369 * modifier name ('S' for Shift, 'C' for Ctrl etc).
2370 */
2371 int
2372name_to_mod_mask(c)
2373 int c;
2374{
2375 int i;
2376
2377 c = TOUPPER_ASC(c);
2378 for (i = 0; mod_mask_table[i].mod_mask != 0; i++)
2379 if (c == mod_mask_table[i].name)
2380 return mod_mask_table[i].mod_flag;
2381 return 0;
2382}
2383
Bram Moolenaar071d4272004-06-13 20:20:40 +00002384/*
2385 * Check if if there is a special key code for "key" that includes the
2386 * modifiers specified.
2387 */
2388 int
2389simplify_key(key, modifiers)
2390 int key;
2391 int *modifiers;
2392{
2393 int i;
2394 int key0;
2395 int key1;
2396
2397 if (*modifiers & (MOD_MASK_SHIFT | MOD_MASK_CTRL | MOD_MASK_ALT))
2398 {
2399 /* TAB is a special case */
2400 if (key == TAB && (*modifiers & MOD_MASK_SHIFT))
2401 {
2402 *modifiers &= ~MOD_MASK_SHIFT;
2403 return K_S_TAB;
2404 }
2405 key0 = KEY2TERMCAP0(key);
2406 key1 = KEY2TERMCAP1(key);
2407 for (i = 0; modifier_keys_table[i] != NUL; i += MOD_KEYS_ENTRY_SIZE)
2408 if (key0 == modifier_keys_table[i + 3]
2409 && key1 == modifier_keys_table[i + 4]
2410 && (*modifiers & modifier_keys_table[i]))
2411 {
2412 *modifiers &= ~modifier_keys_table[i];
2413 return TERMCAP2KEY(modifier_keys_table[i + 1],
2414 modifier_keys_table[i + 2]);
2415 }
2416 }
2417 return key;
2418}
2419
2420/*
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002421 * Change <xHome> to <Home>, <xUp> to <Up>, etc.
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002422 */
2423 int
2424handle_x_keys(key)
2425 int key;
2426{
2427 switch (key)
2428 {
2429 case K_XUP: return K_UP;
2430 case K_XDOWN: return K_DOWN;
2431 case K_XLEFT: return K_LEFT;
2432 case K_XRIGHT: return K_RIGHT;
2433 case K_XHOME: return K_HOME;
Bram Moolenaar68b76a62005-03-25 21:53:48 +00002434 case K_ZHOME: return K_HOME;
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002435 case K_XEND: return K_END;
Bram Moolenaar68b76a62005-03-25 21:53:48 +00002436 case K_ZEND: return K_END;
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002437 case K_XF1: return K_F1;
2438 case K_XF2: return K_F2;
2439 case K_XF3: return K_F3;
2440 case K_XF4: return K_F4;
2441 case K_S_XF1: return K_S_F1;
2442 case K_S_XF2: return K_S_F2;
2443 case K_S_XF3: return K_S_F3;
2444 case K_S_XF4: return K_S_F4;
2445 }
2446 return key;
2447}
2448
2449/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002450 * Return a string which contains the name of the given key when the given
2451 * modifiers are down.
2452 */
2453 char_u *
2454get_special_key_name(c, modifiers)
2455 int c;
2456 int modifiers;
2457{
2458 static char_u string[MAX_KEY_NAME_LEN + 1];
2459
2460 int i, idx;
2461 int table_idx;
2462 char_u *s;
2463
2464 string[0] = '<';
2465 idx = 1;
2466
2467 /* Key that stands for a normal character. */
2468 if (IS_SPECIAL(c) && KEY2TERMCAP0(c) == KS_KEY)
2469 c = KEY2TERMCAP1(c);
2470
2471 /*
2472 * Translate shifted special keys into unshifted keys and set modifier.
2473 * Same for CTRL and ALT modifiers.
2474 */
2475 if (IS_SPECIAL(c))
2476 {
2477 for (i = 0; modifier_keys_table[i] != 0; i += MOD_KEYS_ENTRY_SIZE)
2478 if ( KEY2TERMCAP0(c) == (int)modifier_keys_table[i + 1]
2479 && (int)KEY2TERMCAP1(c) == (int)modifier_keys_table[i + 2])
2480 {
2481 modifiers |= modifier_keys_table[i];
2482 c = TERMCAP2KEY(modifier_keys_table[i + 3],
2483 modifier_keys_table[i + 4]);
2484 break;
2485 }
2486 }
2487
2488 /* try to find the key in the special key table */
2489 table_idx = find_special_key_in_table(c);
2490
2491 /*
2492 * When not a known special key, and not a printable character, try to
2493 * extract modifiers.
2494 */
2495 if (c > 0
2496#ifdef FEAT_MBYTE
2497 && (*mb_char2len)(c) == 1
2498#endif
2499 )
2500 {
2501 if (table_idx < 0
2502 && (!vim_isprintc(c) || (c & 0x7f) == ' ')
2503 && (c & 0x80))
2504 {
2505 c &= 0x7f;
2506 modifiers |= MOD_MASK_ALT;
2507 /* try again, to find the un-alted key in the special key table */
2508 table_idx = find_special_key_in_table(c);
2509 }
2510 if (table_idx < 0 && !vim_isprintc(c) && c < ' ')
2511 {
2512#ifdef EBCDIC
2513 c = CtrlChar(c);
2514#else
2515 c += '@';
2516#endif
2517 modifiers |= MOD_MASK_CTRL;
2518 }
2519 }
2520
2521 /* translate the modifier into a string */
2522 for (i = 0; mod_mask_table[i].name != 'A'; i++)
2523 if ((modifiers & mod_mask_table[i].mod_mask)
2524 == mod_mask_table[i].mod_flag)
2525 {
2526 string[idx++] = mod_mask_table[i].name;
2527 string[idx++] = (char_u)'-';
2528 }
2529
2530 if (table_idx < 0) /* unknown special key, may output t_xx */
2531 {
2532 if (IS_SPECIAL(c))
2533 {
2534 string[idx++] = 't';
2535 string[idx++] = '_';
2536 string[idx++] = KEY2TERMCAP0(c);
2537 string[idx++] = KEY2TERMCAP1(c);
2538 }
2539 /* Not a special key, only modifiers, output directly */
2540 else
2541 {
2542#ifdef FEAT_MBYTE
2543 if (has_mbyte && (*mb_char2len)(c) > 1)
2544 idx += (*mb_char2bytes)(c, string + idx);
2545 else
2546#endif
2547 if (vim_isprintc(c))
2548 string[idx++] = c;
2549 else
2550 {
2551 s = transchar(c);
2552 while (*s)
2553 string[idx++] = *s++;
2554 }
2555 }
2556 }
2557 else /* use name of special key */
2558 {
2559 STRCPY(string + idx, key_names_table[table_idx].name);
2560 idx = (int)STRLEN(string);
2561 }
2562 string[idx++] = '>';
2563 string[idx] = NUL;
2564 return string;
2565}
2566
2567/*
2568 * Try translating a <> name at (*srcp)[] to dst[].
2569 * Return the number of characters added to dst[], zero for no match.
2570 * If there is a match, srcp is advanced to after the <> name.
2571 * dst[] must be big enough to hold the result (up to six characters)!
2572 */
2573 int
2574trans_special(srcp, dst, keycode)
2575 char_u **srcp;
2576 char_u *dst;
2577 int keycode; /* prefer key code, e.g. K_DEL instead of DEL */
2578{
2579 int modifiers = 0;
2580 int key;
2581 int dlen = 0;
2582
Bram Moolenaar2a8ced02008-12-24 11:54:31 +00002583 key = find_special_key(srcp, &modifiers, keycode, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002584 if (key == 0)
2585 return 0;
2586
2587 /* Put the appropriate modifier in a string */
2588 if (modifiers != 0)
2589 {
2590 dst[dlen++] = K_SPECIAL;
2591 dst[dlen++] = KS_MODIFIER;
2592 dst[dlen++] = modifiers;
2593 }
2594
2595 if (IS_SPECIAL(key))
2596 {
2597 dst[dlen++] = K_SPECIAL;
2598 dst[dlen++] = KEY2TERMCAP0(key);
2599 dst[dlen++] = KEY2TERMCAP1(key);
2600 }
2601#ifdef FEAT_MBYTE
2602 else if (has_mbyte && !keycode)
2603 dlen += (*mb_char2bytes)(key, dst + dlen);
2604#endif
2605 else if (keycode)
2606 dlen = (int)(add_char2buf(key, dst + dlen) - dst);
2607 else
2608 dst[dlen++] = key;
2609
2610 return dlen;
2611}
2612
2613/*
2614 * Try translating a <> name at (*srcp)[], return the key and modifiers.
2615 * srcp is advanced to after the <> name.
2616 * returns 0 if there is no match.
2617 */
2618 int
Bram Moolenaar2a8ced02008-12-24 11:54:31 +00002619find_special_key(srcp, modp, keycode, keep_x_key)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002620 char_u **srcp;
2621 int *modp;
Bram Moolenaar2a8ced02008-12-24 11:54:31 +00002622 int keycode; /* prefer key code, e.g. K_DEL instead of DEL */
2623 int keep_x_key; /* don't translate xHome to Home key */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002624{
2625 char_u *last_dash;
2626 char_u *end_of_name;
2627 char_u *src;
2628 char_u *bp;
2629 int modifiers;
2630 int bit;
2631 int key;
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00002632 unsigned long n;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002633
2634 src = *srcp;
2635 if (src[0] != '<')
2636 return 0;
2637
2638 /* Find end of modifier list */
2639 last_dash = src;
2640 for (bp = src + 1; *bp == '-' || vim_isIDc(*bp); bp++)
2641 {
2642 if (*bp == '-')
2643 {
2644 last_dash = bp;
2645 if (bp[1] != NUL && bp[2] == '>')
2646 ++bp; /* anything accepted, like <C-?> */
2647 }
2648 if (bp[0] == 't' && bp[1] == '_' && bp[2] && bp[3])
2649 bp += 3; /* skip t_xx, xx may be '-' or '>' */
2650 }
2651
2652 if (*bp == '>') /* found matching '>' */
2653 {
2654 end_of_name = bp + 1;
2655
2656 if (STRNICMP(src + 1, "char-", 5) == 0 && VIM_ISDIGIT(src[6]))
2657 {
2658 /* <Char-123> or <Char-033> or <Char-0x33> */
2659 vim_str2nr(src + 6, NULL, NULL, TRUE, TRUE, NULL, &n);
2660 *modp = 0;
2661 *srcp = end_of_name;
2662 return (int)n;
2663 }
2664
2665 /* Which modifiers are given? */
2666 modifiers = 0x0;
2667 for (bp = src + 1; bp < last_dash; bp++)
2668 {
2669 if (*bp != '-')
2670 {
2671 bit = name_to_mod_mask(*bp);
2672 if (bit == 0x0)
2673 break; /* Illegal modifier name */
2674 modifiers |= bit;
2675 }
2676 }
2677
2678 /*
2679 * Legal modifier name.
2680 */
2681 if (bp >= last_dash)
2682 {
2683 /*
2684 * Modifier with single letter, or special key name.
2685 */
2686 if (modifiers != 0 && last_dash[2] == '>')
2687 key = last_dash[1];
2688 else
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002689 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00002690 key = get_special_key_code(last_dash + 1);
Bram Moolenaar2a8ced02008-12-24 11:54:31 +00002691 if (!keep_x_key)
2692 key = handle_x_keys(key);
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002693 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002694
2695 /*
2696 * get_special_key_code() may return NUL for invalid
2697 * special key name.
2698 */
2699 if (key != NUL)
2700 {
2701 /*
2702 * Only use a modifier when there is no special key code that
2703 * includes the modifier.
2704 */
2705 key = simplify_key(key, &modifiers);
2706
2707 if (!keycode)
2708 {
2709 /* don't want keycode, use single byte code */
2710 if (key == K_BS)
2711 key = BS;
2712 else if (key == K_DEL || key == K_KDEL)
2713 key = DEL;
2714 }
2715
2716 /*
2717 * Normal Key with modifier: Try to make a single byte code.
2718 */
2719 if (!IS_SPECIAL(key))
2720 key = extract_modifiers(key, &modifiers);
2721
2722 *modp = modifiers;
2723 *srcp = end_of_name;
2724 return key;
2725 }
2726 }
2727 }
2728 return 0;
2729}
2730
2731/*
2732 * Try to include modifiers in the key.
2733 * Changes "Shift-a" to 'A', "Alt-A" to 0xc0, etc.
2734 */
2735 int
2736extract_modifiers(key, modp)
2737 int key;
2738 int *modp;
2739{
2740 int modifiers = *modp;
2741
2742#ifdef MACOS
2743 /* Command-key really special, No fancynest */
2744 if (!(modifiers & MOD_MASK_CMD))
2745#endif
2746 if ((modifiers & MOD_MASK_SHIFT) && ASCII_ISALPHA(key))
2747 {
2748 key = TOUPPER_ASC(key);
2749 modifiers &= ~MOD_MASK_SHIFT;
2750 }
2751 if ((modifiers & MOD_MASK_CTRL)
2752#ifdef EBCDIC
2753 /* * TODO: EBCDIC Better use:
2754 * && (Ctrl_chr(key) || key == '?')
2755 * ??? */
2756 && strchr("?@ABCDEFGHIJKLMNOPQRSTUVWXYZ[\\]^_", key)
2757 != NULL
2758#else
2759 && ((key >= '?' && key <= '_') || ASCII_ISALPHA(key))
2760#endif
2761 )
2762 {
2763 key = Ctrl_chr(key);
2764 modifiers &= ~MOD_MASK_CTRL;
2765 /* <C-@> is <Nul> */
2766 if (key == 0)
2767 key = K_ZERO;
2768 }
2769#ifdef MACOS
2770 /* Command-key really special, No fancynest */
2771 if (!(modifiers & MOD_MASK_CMD))
2772#endif
2773 if ((modifiers & MOD_MASK_ALT) && key < 0x80
2774#ifdef FEAT_MBYTE
2775 && !enc_dbcs /* avoid creating a lead byte */
2776#endif
2777 )
2778 {
2779 key |= 0x80;
2780 modifiers &= ~MOD_MASK_ALT; /* remove the META modifier */
2781 }
2782
2783 *modp = modifiers;
2784 return key;
2785}
2786
2787/*
2788 * Try to find key "c" in the special key table.
2789 * Return the index when found, -1 when not found.
2790 */
2791 int
2792find_special_key_in_table(c)
2793 int c;
2794{
2795 int i;
2796
2797 for (i = 0; key_names_table[i].name != NULL; i++)
2798 if (c == key_names_table[i].key)
2799 break;
2800 if (key_names_table[i].name == NULL)
2801 i = -1;
2802 return i;
2803}
2804
2805/*
2806 * Find the special key with the given name (the given string does not have to
2807 * end with NUL, the name is assumed to end before the first non-idchar).
2808 * If the name starts with "t_" the next two characters are interpreted as a
2809 * termcap name.
2810 * Return the key code, or 0 if not found.
2811 */
2812 int
2813get_special_key_code(name)
2814 char_u *name;
2815{
2816 char_u *table_name;
2817 char_u string[3];
2818 int i, j;
2819
2820 /*
2821 * If it's <t_xx> we get the code for xx from the termcap
2822 */
2823 if (name[0] == 't' && name[1] == '_' && name[2] != NUL && name[3] != NUL)
2824 {
2825 string[0] = name[2];
2826 string[1] = name[3];
2827 string[2] = NUL;
2828 if (add_termcap_entry(string, FALSE) == OK)
2829 return TERMCAP2KEY(name[2], name[3]);
2830 }
2831 else
2832 for (i = 0; key_names_table[i].name != NULL; i++)
2833 {
2834 table_name = key_names_table[i].name;
2835 for (j = 0; vim_isIDc(name[j]) && table_name[j] != NUL; j++)
2836 if (TOLOWER_ASC(table_name[j]) != TOLOWER_ASC(name[j]))
2837 break;
2838 if (!vim_isIDc(name[j]) && table_name[j] == NUL)
2839 return key_names_table[i].key;
2840 }
2841 return 0;
2842}
2843
Bram Moolenaar05bb9532008-07-04 09:44:11 +00002844#if defined(FEAT_CMDL_COMPL) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002845 char_u *
2846get_key_name(i)
2847 int i;
2848{
Bram Moolenaare1fbddc2009-05-16 19:07:03 +00002849 if (i >= (int)KEY_NAMES_TABLE_LEN)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002850 return NULL;
2851 return key_names_table[i].name;
2852}
2853#endif
2854
Bram Moolenaar05bb9532008-07-04 09:44:11 +00002855#if defined(FEAT_MOUSE) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002856/*
2857 * Look up the given mouse code to return the relevant information in the other
2858 * arguments. Return which button is down or was released.
2859 */
2860 int
2861get_mouse_button(code, is_click, is_drag)
2862 int code;
2863 int *is_click;
2864 int *is_drag;
2865{
2866 int i;
2867
2868 for (i = 0; mouse_table[i].pseudo_code; i++)
2869 if (code == mouse_table[i].pseudo_code)
2870 {
2871 *is_click = mouse_table[i].is_click;
2872 *is_drag = mouse_table[i].is_drag;
2873 return mouse_table[i].button;
2874 }
2875 return 0; /* Shouldn't get here */
2876}
2877
2878/*
2879 * Return the appropriate pseudo mouse event token (KE_LEFTMOUSE etc) based on
2880 * the given information about which mouse button is down, and whether the
2881 * mouse was clicked, dragged or released.
2882 */
2883 int
2884get_pseudo_mouse_code(button, is_click, is_drag)
2885 int button; /* eg MOUSE_LEFT */
2886 int is_click;
2887 int is_drag;
2888{
2889 int i;
2890
2891 for (i = 0; mouse_table[i].pseudo_code; i++)
2892 if (button == mouse_table[i].button
2893 && is_click == mouse_table[i].is_click
2894 && is_drag == mouse_table[i].is_drag)
2895 {
2896#ifdef FEAT_GUI
Bram Moolenaarc91506a2005-04-24 22:04:21 +00002897 /* Trick: a non mappable left click and release has mouse_col -1
2898 * or added MOUSE_COLOFF. Used for 'mousefocus' in
2899 * gui_mouse_moved() */
2900 if (mouse_col < 0 || mouse_col > MOUSE_COLOFF)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002901 {
Bram Moolenaarc91506a2005-04-24 22:04:21 +00002902 if (mouse_col < 0)
2903 mouse_col = 0;
2904 else
2905 mouse_col -= MOUSE_COLOFF;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002906 if (mouse_table[i].pseudo_code == (int)KE_LEFTMOUSE)
2907 return (int)KE_LEFTMOUSE_NM;
2908 if (mouse_table[i].pseudo_code == (int)KE_LEFTRELEASE)
2909 return (int)KE_LEFTRELEASE_NM;
2910 }
2911#endif
2912 return mouse_table[i].pseudo_code;
2913 }
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00002914 return (int)KE_IGNORE; /* not recognized, ignore it */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002915}
2916#endif /* FEAT_MOUSE */
2917
2918/*
2919 * Return the current end-of-line type: EOL_DOS, EOL_UNIX or EOL_MAC.
2920 */
2921 int
2922get_fileformat(buf)
2923 buf_T *buf;
2924{
2925 int c = *buf->b_p_ff;
2926
2927 if (buf->b_p_bin || c == 'u')
2928 return EOL_UNIX;
2929 if (c == 'm')
2930 return EOL_MAC;
2931 return EOL_DOS;
2932}
2933
2934/*
2935 * Like get_fileformat(), but override 'fileformat' with "p" for "++opt=val"
2936 * argument.
2937 */
2938 int
2939get_fileformat_force(buf, eap)
2940 buf_T *buf;
2941 exarg_T *eap; /* can be NULL! */
2942{
2943 int c;
2944
2945 if (eap != NULL && eap->force_ff != 0)
2946 c = eap->cmd[eap->force_ff];
2947 else
2948 {
2949 if ((eap != NULL && eap->force_bin != 0)
2950 ? (eap->force_bin == FORCE_BIN) : buf->b_p_bin)
2951 return EOL_UNIX;
2952 c = *buf->b_p_ff;
2953 }
2954 if (c == 'u')
2955 return EOL_UNIX;
2956 if (c == 'm')
2957 return EOL_MAC;
2958 return EOL_DOS;
2959}
2960
2961/*
2962 * Set the current end-of-line type to EOL_DOS, EOL_UNIX or EOL_MAC.
2963 * Sets both 'textmode' and 'fileformat'.
2964 * Note: Does _not_ set global value of 'textmode'!
2965 */
2966 void
2967set_fileformat(t, opt_flags)
2968 int t;
2969 int opt_flags; /* OPT_LOCAL and/or OPT_GLOBAL */
2970{
2971 char *p = NULL;
2972
2973 switch (t)
2974 {
2975 case EOL_DOS:
2976 p = FF_DOS;
2977 curbuf->b_p_tx = TRUE;
2978 break;
2979 case EOL_UNIX:
2980 p = FF_UNIX;
2981 curbuf->b_p_tx = FALSE;
2982 break;
2983 case EOL_MAC:
2984 p = FF_MAC;
2985 curbuf->b_p_tx = FALSE;
2986 break;
2987 }
2988 if (p != NULL)
2989 set_string_option_direct((char_u *)"ff", -1, (char_u *)p,
Bram Moolenaar5e3cb7e2006-02-27 23:58:35 +00002990 OPT_FREE | opt_flags, 0);
2991
Bram Moolenaar071d4272004-06-13 20:20:40 +00002992#ifdef FEAT_WINDOWS
Bram Moolenaarf740b292006-02-16 22:11:02 +00002993 /* This may cause the buffer to become (un)modified. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002994 check_status(curbuf);
Bram Moolenaar997fb4b2006-02-17 21:53:23 +00002995 redraw_tabline = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002996#endif
2997#ifdef FEAT_TITLE
2998 need_maketitle = TRUE; /* set window title later */
2999#endif
3000}
3001
3002/*
3003 * Return the default fileformat from 'fileformats'.
3004 */
3005 int
3006default_fileformat()
3007{
3008 switch (*p_ffs)
3009 {
3010 case 'm': return EOL_MAC;
3011 case 'd': return EOL_DOS;
3012 }
3013 return EOL_UNIX;
3014}
3015
3016/*
3017 * Call shell. Calls mch_call_shell, with 'shellxquote' added.
3018 */
3019 int
3020call_shell(cmd, opt)
3021 char_u *cmd;
3022 int opt;
3023{
3024 char_u *ncmd;
3025 int retval;
Bram Moolenaar05159a02005-02-26 23:04:13 +00003026#ifdef FEAT_PROFILE
3027 proftime_T wait_time;
3028#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003029
3030 if (p_verbose > 3)
3031 {
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00003032 verbose_enter();
Bram Moolenaar051b7822005-05-19 21:00:46 +00003033 smsg((char_u *)_("Calling shell to execute: \"%s\""),
Bram Moolenaar071d4272004-06-13 20:20:40 +00003034 cmd == NULL ? p_sh : cmd);
3035 out_char('\n');
3036 cursor_on();
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00003037 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003038 }
3039
Bram Moolenaar05159a02005-02-26 23:04:13 +00003040#ifdef FEAT_PROFILE
Bram Moolenaar01265852006-03-20 21:50:15 +00003041 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003042 prof_child_enter(&wait_time);
3043#endif
3044
Bram Moolenaar071d4272004-06-13 20:20:40 +00003045 if (*p_sh == NUL)
3046 {
3047 EMSG(_(e_shellempty));
3048 retval = -1;
3049 }
3050 else
3051 {
3052#ifdef FEAT_GUI_MSWIN
3053 /* Don't hide the pointer while executing a shell command. */
3054 gui_mch_mousehide(FALSE);
3055#endif
3056#ifdef FEAT_GUI
3057 ++hold_gui_events;
3058#endif
3059 /* The external command may update a tags file, clear cached tags. */
3060 tag_freematch();
3061
3062 if (cmd == NULL || *p_sxq == NUL)
3063 retval = mch_call_shell(cmd, opt);
3064 else
3065 {
3066 ncmd = alloc((unsigned)(STRLEN(cmd) + STRLEN(p_sxq) * 2 + 1));
3067 if (ncmd != NULL)
3068 {
3069 STRCPY(ncmd, p_sxq);
3070 STRCAT(ncmd, cmd);
3071 STRCAT(ncmd, p_sxq);
3072 retval = mch_call_shell(ncmd, opt);
3073 vim_free(ncmd);
3074 }
3075 else
3076 retval = -1;
3077 }
3078#ifdef FEAT_GUI
3079 --hold_gui_events;
3080#endif
3081 /*
3082 * Check the window size, in case it changed while executing the
3083 * external command.
3084 */
3085 shell_resized_check();
3086 }
3087
3088#ifdef FEAT_EVAL
3089 set_vim_var_nr(VV_SHELL_ERROR, (long)retval);
Bram Moolenaar05159a02005-02-26 23:04:13 +00003090# ifdef FEAT_PROFILE
Bram Moolenaar01265852006-03-20 21:50:15 +00003091 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003092 prof_child_exit(&wait_time);
3093# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003094#endif
3095
3096 return retval;
3097}
3098
3099/*
Bram Moolenaar01265852006-03-20 21:50:15 +00003100 * VISUAL, SELECTMODE and OP_PENDING State are never set, they are equal to
3101 * NORMAL State with a condition. This function returns the real State.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003102 */
3103 int
3104get_real_state()
3105{
3106 if (State & NORMAL)
3107 {
3108#ifdef FEAT_VISUAL
3109 if (VIsual_active)
Bram Moolenaar01265852006-03-20 21:50:15 +00003110 {
3111 if (VIsual_select)
3112 return SELECTMODE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003113 return VISUAL;
Bram Moolenaar01265852006-03-20 21:50:15 +00003114 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003115 else
3116#endif
3117 if (finish_op)
Bram Moolenaar01265852006-03-20 21:50:15 +00003118 return OP_PENDING;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003119 }
3120 return State;
3121}
3122
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003123#if defined(FEAT_MBYTE) || defined(PROTO)
3124/*
3125 * Return TRUE if "p" points to just after a path separator.
3126 * Take care of multi-byte characters.
3127 * "b" must point to the start of the file name
3128 */
3129 int
3130after_pathsep(b, p)
3131 char_u *b;
3132 char_u *p;
3133{
3134 return vim_ispathsep(p[-1])
3135 && (!has_mbyte || (*mb_head_off)(b, p - 1) == 0);
3136}
3137#endif
3138
3139/*
3140 * Return TRUE if file names "f1" and "f2" are in the same directory.
3141 * "f1" may be a short name, "f2" must be a full path.
3142 */
3143 int
3144same_directory(f1, f2)
3145 char_u *f1;
3146 char_u *f2;
3147{
3148 char_u ffname[MAXPATHL];
3149 char_u *t1;
3150 char_u *t2;
3151
3152 /* safety check */
3153 if (f1 == NULL || f2 == NULL)
3154 return FALSE;
3155
3156 (void)vim_FullName(f1, ffname, MAXPATHL, FALSE);
3157 t1 = gettail_sep(ffname);
3158 t2 = gettail_sep(f2);
3159 return (t1 - ffname == t2 - f2
3160 && pathcmp((char *)ffname, (char *)f2, (int)(t1 - ffname)) == 0);
3161}
3162
Bram Moolenaar071d4272004-06-13 20:20:40 +00003163#if defined(FEAT_SESSION) || defined(MSWIN) || defined(FEAT_GUI_MAC) \
Bram Moolenaar9372a112005-12-06 19:59:18 +00003164 || ((defined(FEAT_GUI_GTK)) \
Bram Moolenaar843ee412004-06-30 16:16:41 +00003165 && ( defined(FEAT_WINDOWS) || defined(FEAT_DND)) ) \
Bram Moolenaar071d4272004-06-13 20:20:40 +00003166 || defined(FEAT_SUN_WORKSHOP) || defined(FEAT_NETBEANS_INTG) \
3167 || defined(PROTO)
3168/*
3169 * Change to a file's directory.
3170 * Caller must call shorten_fnames()!
3171 * Return OK or FAIL.
3172 */
3173 int
3174vim_chdirfile(fname)
3175 char_u *fname;
3176{
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003177 char_u dir[MAXPATHL];
Bram Moolenaar071d4272004-06-13 20:20:40 +00003178
Bram Moolenaarbbebc852005-07-18 21:47:53 +00003179 vim_strncpy(dir, fname, MAXPATHL - 1);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003180 *gettail_sep(dir) = NUL;
3181 return mch_chdir((char *)dir) == 0 ? OK : FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003182}
3183#endif
3184
3185#if defined(STAT_IGNORES_SLASH) || defined(PROTO)
3186/*
3187 * Check if "name" ends in a slash and is not a directory.
3188 * Used for systems where stat() ignores a trailing slash on a file name.
3189 * The Vim code assumes a trailing slash is only ignored for a directory.
3190 */
3191 int
3192illegal_slash(name)
3193 char *name;
3194{
3195 if (name[0] == NUL)
3196 return FALSE; /* no file name is not illegal */
3197 if (name[strlen(name) - 1] != '/')
3198 return FALSE; /* no trailing slash */
3199 if (mch_isdir((char_u *)name))
3200 return FALSE; /* trailing slash for a directory */
3201 return TRUE;
3202}
3203#endif
3204
3205#if defined(CURSOR_SHAPE) || defined(PROTO)
3206
3207/*
3208 * Handling of cursor and mouse pointer shapes in various modes.
3209 */
3210
3211cursorentry_T shape_table[SHAPE_IDX_COUNT] =
3212{
3213 /* The values will be filled in from the 'guicursor' and 'mouseshape'
3214 * defaults when Vim starts.
3215 * Adjust the SHAPE_IDX_ defines when making changes! */
3216 {0, 0, 0, 700L, 400L, 250L, 0, 0, "n", SHAPE_CURSOR+SHAPE_MOUSE},
3217 {0, 0, 0, 700L, 400L, 250L, 0, 0, "v", SHAPE_CURSOR+SHAPE_MOUSE},
3218 {0, 0, 0, 700L, 400L, 250L, 0, 0, "i", SHAPE_CURSOR+SHAPE_MOUSE},
3219 {0, 0, 0, 700L, 400L, 250L, 0, 0, "r", SHAPE_CURSOR+SHAPE_MOUSE},
3220 {0, 0, 0, 700L, 400L, 250L, 0, 0, "c", SHAPE_CURSOR+SHAPE_MOUSE},
3221 {0, 0, 0, 700L, 400L, 250L, 0, 0, "ci", SHAPE_CURSOR+SHAPE_MOUSE},
3222 {0, 0, 0, 700L, 400L, 250L, 0, 0, "cr", SHAPE_CURSOR+SHAPE_MOUSE},
3223 {0, 0, 0, 700L, 400L, 250L, 0, 0, "o", SHAPE_CURSOR+SHAPE_MOUSE},
3224 {0, 0, 0, 700L, 400L, 250L, 0, 0, "ve", SHAPE_CURSOR+SHAPE_MOUSE},
3225 {0, 0, 0, 0L, 0L, 0L, 0, 0, "e", SHAPE_MOUSE},
3226 {0, 0, 0, 0L, 0L, 0L, 0, 0, "s", SHAPE_MOUSE},
3227 {0, 0, 0, 0L, 0L, 0L, 0, 0, "sd", SHAPE_MOUSE},
3228 {0, 0, 0, 0L, 0L, 0L, 0, 0, "vs", SHAPE_MOUSE},
3229 {0, 0, 0, 0L, 0L, 0L, 0, 0, "vd", SHAPE_MOUSE},
3230 {0, 0, 0, 0L, 0L, 0L, 0, 0, "m", SHAPE_MOUSE},
3231 {0, 0, 0, 0L, 0L, 0L, 0, 0, "ml", SHAPE_MOUSE},
3232 {0, 0, 0, 100L, 100L, 100L, 0, 0, "sm", SHAPE_CURSOR},
3233};
3234
3235#ifdef FEAT_MOUSESHAPE
3236/*
3237 * Table with names for mouse shapes. Keep in sync with all the tables for
3238 * mch_set_mouse_shape()!.
3239 */
3240static char * mshape_names[] =
3241{
3242 "arrow", /* default, must be the first one */
3243 "blank", /* hidden */
3244 "beam",
3245 "updown",
3246 "udsizing",
3247 "leftright",
3248 "lrsizing",
3249 "busy",
3250 "no",
3251 "crosshair",
3252 "hand1",
3253 "hand2",
3254 "pencil",
3255 "question",
3256 "rightup-arrow",
3257 "up-arrow",
3258 NULL
3259};
3260#endif
3261
3262/*
3263 * Parse the 'guicursor' option ("what" is SHAPE_CURSOR) or 'mouseshape'
3264 * ("what" is SHAPE_MOUSE).
3265 * Returns error message for an illegal option, NULL otherwise.
3266 */
3267 char_u *
3268parse_shape_opt(what)
3269 int what;
3270{
3271 char_u *modep;
3272 char_u *colonp;
3273 char_u *commap;
3274 char_u *slashp;
3275 char_u *p, *endp;
3276 int idx = 0; /* init for GCC */
3277 int all_idx;
3278 int len;
3279 int i;
3280 long n;
3281 int found_ve = FALSE; /* found "ve" flag */
3282 int round;
3283
3284 /*
3285 * First round: check for errors; second round: do it for real.
3286 */
3287 for (round = 1; round <= 2; ++round)
3288 {
3289 /*
3290 * Repeat for all comma separated parts.
3291 */
3292#ifdef FEAT_MOUSESHAPE
3293 if (what == SHAPE_MOUSE)
3294 modep = p_mouseshape;
3295 else
3296#endif
3297 modep = p_guicursor;
3298 while (*modep != NUL)
3299 {
3300 colonp = vim_strchr(modep, ':');
3301 if (colonp == NULL)
3302 return (char_u *)N_("E545: Missing colon");
3303 if (colonp == modep)
3304 return (char_u *)N_("E546: Illegal mode");
3305 commap = vim_strchr(modep, ',');
3306
3307 /*
3308 * Repeat for all mode's before the colon.
3309 * For the 'a' mode, we loop to handle all the modes.
3310 */
3311 all_idx = -1;
3312 while (modep < colonp || all_idx >= 0)
3313 {
3314 if (all_idx < 0)
3315 {
3316 /* Find the mode. */
3317 if (modep[1] == '-' || modep[1] == ':')
3318 len = 1;
3319 else
3320 len = 2;
3321 if (len == 1 && TOLOWER_ASC(modep[0]) == 'a')
3322 all_idx = SHAPE_IDX_COUNT - 1;
3323 else
3324 {
3325 for (idx = 0; idx < SHAPE_IDX_COUNT; ++idx)
3326 if (STRNICMP(modep, shape_table[idx].name, len)
3327 == 0)
3328 break;
3329 if (idx == SHAPE_IDX_COUNT
3330 || (shape_table[idx].used_for & what) == 0)
3331 return (char_u *)N_("E546: Illegal mode");
3332 if (len == 2 && modep[0] == 'v' && modep[1] == 'e')
3333 found_ve = TRUE;
3334 }
3335 modep += len + 1;
3336 }
3337
3338 if (all_idx >= 0)
3339 idx = all_idx--;
3340 else if (round == 2)
3341 {
3342#ifdef FEAT_MOUSESHAPE
3343 if (what == SHAPE_MOUSE)
3344 {
3345 /* Set the default, for the missing parts */
3346 shape_table[idx].mshape = 0;
3347 }
3348 else
3349#endif
3350 {
3351 /* Set the defaults, for the missing parts */
3352 shape_table[idx].shape = SHAPE_BLOCK;
3353 shape_table[idx].blinkwait = 700L;
3354 shape_table[idx].blinkon = 400L;
3355 shape_table[idx].blinkoff = 250L;
3356 }
3357 }
3358
3359 /* Parse the part after the colon */
3360 for (p = colonp + 1; *p && *p != ','; )
3361 {
3362#ifdef FEAT_MOUSESHAPE
3363 if (what == SHAPE_MOUSE)
3364 {
3365 for (i = 0; ; ++i)
3366 {
3367 if (mshape_names[i] == NULL)
3368 {
3369 if (!VIM_ISDIGIT(*p))
3370 return (char_u *)N_("E547: Illegal mouseshape");
3371 if (round == 2)
3372 shape_table[idx].mshape =
3373 getdigits(&p) + MSHAPE_NUMBERED;
3374 else
3375 (void)getdigits(&p);
3376 break;
3377 }
3378 len = (int)STRLEN(mshape_names[i]);
3379 if (STRNICMP(p, mshape_names[i], len) == 0)
3380 {
3381 if (round == 2)
3382 shape_table[idx].mshape = i;
3383 p += len;
3384 break;
3385 }
3386 }
3387 }
3388 else /* if (what == SHAPE_MOUSE) */
3389#endif
3390 {
3391 /*
3392 * First handle the ones with a number argument.
3393 */
3394 i = *p;
3395 len = 0;
3396 if (STRNICMP(p, "ver", 3) == 0)
3397 len = 3;
3398 else if (STRNICMP(p, "hor", 3) == 0)
3399 len = 3;
3400 else if (STRNICMP(p, "blinkwait", 9) == 0)
3401 len = 9;
3402 else if (STRNICMP(p, "blinkon", 7) == 0)
3403 len = 7;
3404 else if (STRNICMP(p, "blinkoff", 8) == 0)
3405 len = 8;
3406 if (len != 0)
3407 {
3408 p += len;
3409 if (!VIM_ISDIGIT(*p))
3410 return (char_u *)N_("E548: digit expected");
3411 n = getdigits(&p);
3412 if (len == 3) /* "ver" or "hor" */
3413 {
3414 if (n == 0)
3415 return (char_u *)N_("E549: Illegal percentage");
3416 if (round == 2)
3417 {
3418 if (TOLOWER_ASC(i) == 'v')
3419 shape_table[idx].shape = SHAPE_VER;
3420 else
3421 shape_table[idx].shape = SHAPE_HOR;
3422 shape_table[idx].percentage = n;
3423 }
3424 }
3425 else if (round == 2)
3426 {
3427 if (len == 9)
3428 shape_table[idx].blinkwait = n;
3429 else if (len == 7)
3430 shape_table[idx].blinkon = n;
3431 else
3432 shape_table[idx].blinkoff = n;
3433 }
3434 }
3435 else if (STRNICMP(p, "block", 5) == 0)
3436 {
3437 if (round == 2)
3438 shape_table[idx].shape = SHAPE_BLOCK;
3439 p += 5;
3440 }
3441 else /* must be a highlight group name then */
3442 {
3443 endp = vim_strchr(p, '-');
3444 if (commap == NULL) /* last part */
3445 {
3446 if (endp == NULL)
3447 endp = p + STRLEN(p); /* find end of part */
3448 }
3449 else if (endp > commap || endp == NULL)
3450 endp = commap;
3451 slashp = vim_strchr(p, '/');
3452 if (slashp != NULL && slashp < endp)
3453 {
3454 /* "group/langmap_group" */
3455 i = syn_check_group(p, (int)(slashp - p));
3456 p = slashp + 1;
3457 }
3458 if (round == 2)
3459 {
3460 shape_table[idx].id = syn_check_group(p,
3461 (int)(endp - p));
3462 shape_table[idx].id_lm = shape_table[idx].id;
3463 if (slashp != NULL && slashp < endp)
3464 shape_table[idx].id = i;
3465 }
3466 p = endp;
3467 }
3468 } /* if (what != SHAPE_MOUSE) */
3469
3470 if (*p == '-')
3471 ++p;
3472 }
3473 }
3474 modep = p;
3475 if (*modep == ',')
3476 ++modep;
3477 }
3478 }
3479
3480 /* If the 's' flag is not given, use the 'v' cursor for 's' */
3481 if (!found_ve)
3482 {
3483#ifdef FEAT_MOUSESHAPE
3484 if (what == SHAPE_MOUSE)
3485 {
3486 shape_table[SHAPE_IDX_VE].mshape = shape_table[SHAPE_IDX_V].mshape;
3487 }
3488 else
3489#endif
3490 {
3491 shape_table[SHAPE_IDX_VE].shape = shape_table[SHAPE_IDX_V].shape;
3492 shape_table[SHAPE_IDX_VE].percentage =
3493 shape_table[SHAPE_IDX_V].percentage;
3494 shape_table[SHAPE_IDX_VE].blinkwait =
3495 shape_table[SHAPE_IDX_V].blinkwait;
3496 shape_table[SHAPE_IDX_VE].blinkon =
3497 shape_table[SHAPE_IDX_V].blinkon;
3498 shape_table[SHAPE_IDX_VE].blinkoff =
3499 shape_table[SHAPE_IDX_V].blinkoff;
3500 shape_table[SHAPE_IDX_VE].id = shape_table[SHAPE_IDX_V].id;
3501 shape_table[SHAPE_IDX_VE].id_lm = shape_table[SHAPE_IDX_V].id_lm;
3502 }
3503 }
3504
3505 return NULL;
3506}
3507
Bram Moolenaarac6e65f2005-08-29 22:25:38 +00003508# if defined(MCH_CURSOR_SHAPE) || defined(FEAT_GUI) \
3509 || defined(FEAT_MOUSESHAPE) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003510/*
3511 * Return the index into shape_table[] for the current mode.
3512 * When "mouse" is TRUE, consider indexes valid for the mouse pointer.
3513 */
3514 int
3515get_shape_idx(mouse)
3516 int mouse;
3517{
3518#ifdef FEAT_MOUSESHAPE
3519 if (mouse && (State == HITRETURN || State == ASKMORE))
3520 {
3521# ifdef FEAT_GUI
Bram Moolenaar9588a0f2005-01-08 21:45:39 +00003522 int x, y;
3523 gui_mch_getmouse(&x, &y);
3524 if (Y_2_ROW(y) == Rows - 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003525 return SHAPE_IDX_MOREL;
3526# endif
3527 return SHAPE_IDX_MORE;
3528 }
3529 if (mouse && drag_status_line)
3530 return SHAPE_IDX_SDRAG;
3531# ifdef FEAT_VERTSPLIT
3532 if (mouse && drag_sep_line)
3533 return SHAPE_IDX_VDRAG;
3534# endif
3535#endif
3536 if (!mouse && State == SHOWMATCH)
3537 return SHAPE_IDX_SM;
3538#ifdef FEAT_VREPLACE
3539 if (State & VREPLACE_FLAG)
3540 return SHAPE_IDX_R;
3541#endif
3542 if (State & REPLACE_FLAG)
3543 return SHAPE_IDX_R;
3544 if (State & INSERT)
3545 return SHAPE_IDX_I;
3546 if (State & CMDLINE)
3547 {
3548 if (cmdline_at_end())
3549 return SHAPE_IDX_C;
3550 if (cmdline_overstrike())
3551 return SHAPE_IDX_CR;
3552 return SHAPE_IDX_CI;
3553 }
3554 if (finish_op)
3555 return SHAPE_IDX_O;
3556#ifdef FEAT_VISUAL
3557 if (VIsual_active)
3558 {
3559 if (*p_sel == 'e')
3560 return SHAPE_IDX_VE;
3561 else
3562 return SHAPE_IDX_V;
3563 }
3564#endif
3565 return SHAPE_IDX_N;
3566}
Bram Moolenaarac6e65f2005-08-29 22:25:38 +00003567#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003568
3569# if defined(FEAT_MOUSESHAPE) || defined(PROTO)
3570static int old_mouse_shape = 0;
3571
3572/*
3573 * Set the mouse shape:
3574 * If "shape" is -1, use shape depending on the current mode,
3575 * depending on the current state.
3576 * If "shape" is -2, only update the shape when it's CLINE or STATUS (used
3577 * when the mouse moves off the status or command line).
3578 */
3579 void
3580update_mouseshape(shape_idx)
3581 int shape_idx;
3582{
3583 int new_mouse_shape;
3584
3585 /* Only works in GUI mode. */
Bram Moolenaar6bb68362005-03-22 23:03:44 +00003586 if (!gui.in_use || gui.starting)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003587 return;
3588
3589 /* Postpone the updating when more is to come. Speeds up executing of
3590 * mappings. */
3591 if (shape_idx == -1 && char_avail())
3592 {
3593 postponed_mouseshape = TRUE;
3594 return;
3595 }
3596
Bram Moolenaar14716812006-05-04 21:54:08 +00003597 /* When ignoring the mouse don't change shape on the statusline. */
3598 if (*p_mouse == NUL
3599 && (shape_idx == SHAPE_IDX_CLINE
3600 || shape_idx == SHAPE_IDX_STATUS
3601 || shape_idx == SHAPE_IDX_VSEP))
3602 shape_idx = -2;
3603
Bram Moolenaar071d4272004-06-13 20:20:40 +00003604 if (shape_idx == -2
3605 && old_mouse_shape != shape_table[SHAPE_IDX_CLINE].mshape
3606 && old_mouse_shape != shape_table[SHAPE_IDX_STATUS].mshape
3607 && old_mouse_shape != shape_table[SHAPE_IDX_VSEP].mshape)
3608 return;
3609 if (shape_idx < 0)
3610 new_mouse_shape = shape_table[get_shape_idx(TRUE)].mshape;
3611 else
3612 new_mouse_shape = shape_table[shape_idx].mshape;
3613 if (new_mouse_shape != old_mouse_shape)
3614 {
3615 mch_set_mouse_shape(new_mouse_shape);
3616 old_mouse_shape = new_mouse_shape;
3617 }
3618 postponed_mouseshape = FALSE;
3619}
3620# endif
3621
3622#endif /* CURSOR_SHAPE */
3623
3624
3625#ifdef FEAT_CRYPT
3626/*
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00003627 * Optional encryption support.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003628 * Mohsin Ahmed, mosh@sasi.com, 98-09-24
3629 * Based on zip/crypt sources.
3630 *
3631 * NOTE FOR USA: Since 2000 exporting this code from the USA is allowed to
3632 * most countries. There are a few exceptions, but that still should not be a
3633 * problem since this code was originally created in Europe and India.
3634 */
3635
3636/* from zip.h */
3637
3638typedef unsigned short ush; /* unsigned 16-bit value */
3639typedef unsigned long ulg; /* unsigned 32-bit value */
3640
3641static void make_crc_tab __ARGS((void));
3642
Bram Moolenaard6f676d2005-06-01 21:51:55 +00003643static ulg crc_32_tab[256];
Bram Moolenaar071d4272004-06-13 20:20:40 +00003644
3645/*
3646 * Fill the CRC table.
3647 */
3648 static void
3649make_crc_tab()
3650{
3651 ulg s,t,v;
3652 static int done = FALSE;
3653
3654 if (done)
3655 return;
3656 for (t = 0; t < 256; t++)
3657 {
3658 v = t;
3659 for (s = 0; s < 8; s++)
3660 v = (v >> 1) ^ ((v & 1) * (ulg)0xedb88320L);
3661 crc_32_tab[t] = v;
3662 }
3663 done = TRUE;
3664}
3665
3666#define CRC32(c, b) (crc_32_tab[((int)(c) ^ (b)) & 0xff] ^ ((c) >> 8))
3667
3668
3669static ulg keys[3]; /* keys defining the pseudo-random sequence */
3670
3671/*
3672 * Return the next byte in the pseudo-random sequence
3673 */
3674 int
3675decrypt_byte()
3676{
3677 ush temp;
3678
3679 temp = (ush)keys[2] | 2;
3680 return (int)(((unsigned)(temp * (temp ^ 1)) >> 8) & 0xff);
3681}
3682
3683/*
3684 * Update the encryption keys with the next byte of plain text
3685 */
3686 int
3687update_keys(c)
3688 int c; /* byte of plain text */
3689{
3690 keys[0] = CRC32(keys[0], c);
3691 keys[1] += keys[0] & 0xff;
3692 keys[1] = keys[1] * 134775813L + 1;
3693 keys[2] = CRC32(keys[2], (int)(keys[1] >> 24));
3694 return c;
3695}
3696
3697/*
3698 * Initialize the encryption keys and the random header according to
3699 * the given password.
3700 * If "passwd" is NULL or empty, don't do anything.
3701 */
3702 void
3703crypt_init_keys(passwd)
3704 char_u *passwd; /* password string with which to modify keys */
3705{
3706 if (passwd != NULL && *passwd != NUL)
3707 {
3708 make_crc_tab();
3709 keys[0] = 305419896L;
3710 keys[1] = 591751049L;
3711 keys[2] = 878082192L;
3712 while (*passwd != '\0')
3713 update_keys((int)*passwd++);
3714 }
3715}
3716
3717/*
3718 * Ask the user for a crypt key.
3719 * When "store" is TRUE, the new key in stored in the 'key' option, and the
3720 * 'key' option value is returned: Don't free it.
3721 * When "store" is FALSE, the typed key is returned in allocated memory.
3722 * Returns NULL on failure.
3723 */
3724 char_u *
3725get_crypt_key(store, twice)
3726 int store;
3727 int twice; /* Ask for the key twice. */
3728{
3729 char_u *p1, *p2 = NULL;
3730 int round;
3731
3732 for (round = 0; ; ++round)
3733 {
3734 cmdline_star = TRUE;
3735 cmdline_row = msg_row;
3736 p1 = getcmdline_prompt(NUL, round == 0
3737 ? (char_u *)_("Enter encryption key: ")
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +00003738 : (char_u *)_("Enter same key again: "), 0, EXPAND_NOTHING,
3739 NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003740 cmdline_star = FALSE;
3741
3742 if (p1 == NULL)
3743 break;
3744
3745 if (round == twice)
3746 {
3747 if (p2 != NULL && STRCMP(p1, p2) != 0)
3748 {
3749 MSG(_("Keys don't match!"));
3750 vim_free(p1);
3751 vim_free(p2);
3752 p2 = NULL;
3753 round = -1; /* do it again */
3754 continue;
3755 }
3756 if (store)
3757 {
3758 set_option_value((char_u *)"key", 0L, p1, OPT_LOCAL);
3759 vim_free(p1);
3760 p1 = curbuf->b_p_key;
3761 }
3762 break;
3763 }
3764 p2 = p1;
3765 }
3766
3767 /* since the user typed this, no need to wait for return */
3768 need_wait_return = FALSE;
3769 msg_didout = FALSE;
3770
3771 vim_free(p2);
3772 return p1;
3773}
3774
3775#endif /* FEAT_CRYPT */
3776
3777/* TODO: make some #ifdef for this */
3778/*--------[ file searching ]-------------------------------------------------*/
3779/*
3780 * File searching functions for 'path', 'tags' and 'cdpath' options.
3781 * External visible functions:
3782 * vim_findfile_init() creates/initialises the search context
3783 * vim_findfile_free_visited() free list of visited files/dirs of search
3784 * context
3785 * vim_findfile() find a file in the search context
3786 * vim_findfile_cleanup() cleanup/free search context created by
3787 * vim_findfile_init()
3788 *
3789 * All static functions and variables start with 'ff_'
3790 *
3791 * In general it works like this:
3792 * First you create yourself a search context by calling vim_findfile_init().
3793 * It is possible to give a search context from a previous call to
3794 * vim_findfile_init(), so it can be reused. After this you call vim_findfile()
3795 * until you are satisfied with the result or it returns NULL. On every call it
3796 * returns the next file which matches the conditions given to
3797 * vim_findfile_init(). If it doesn't find a next file it returns NULL.
3798 *
3799 * It is possible to call vim_findfile_init() again to reinitialise your search
3800 * with some new parameters. Don't forget to pass your old search context to
3801 * it, so it can reuse it and especially reuse the list of already visited
3802 * directories. If you want to delete the list of already visited directories
3803 * simply call vim_findfile_free_visited().
3804 *
3805 * When you are done call vim_findfile_cleanup() to free the search context.
3806 *
3807 * The function vim_findfile_init() has a long comment, which describes the
3808 * needed parameters.
3809 *
3810 *
3811 *
3812 * ATTENTION:
3813 * ==========
Bram Moolenaar77f66d62007-02-20 02:16:18 +00003814 * Also we use an allocated search context here, this functions are NOT
Bram Moolenaar071d4272004-06-13 20:20:40 +00003815 * thread-safe!!!!!
3816 *
3817 * To minimize parameter passing (or because I'm to lazy), only the
3818 * external visible functions get a search context as a parameter. This is
3819 * then assigned to a static global, which is used throughout the local
3820 * functions.
3821 */
3822
3823/*
3824 * type for the directory search stack
3825 */
3826typedef struct ff_stack
3827{
3828 struct ff_stack *ffs_prev;
3829
3830 /* the fix part (no wildcards) and the part containing the wildcards
3831 * of the search path
3832 */
3833 char_u *ffs_fix_path;
3834#ifdef FEAT_PATH_EXTRA
3835 char_u *ffs_wc_path;
3836#endif
3837
3838 /* files/dirs found in the above directory, matched by the first wildcard
3839 * of wc_part
3840 */
3841 char_u **ffs_filearray;
3842 int ffs_filearray_size;
3843 char_u ffs_filearray_cur; /* needed for partly handled dirs */
3844
3845 /* to store status of partly handled directories
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00003846 * 0: we work on this directory for the first time
Bram Moolenaar071d4272004-06-13 20:20:40 +00003847 * 1: this directory was partly searched in an earlier step
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00003848 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003849 int ffs_stage;
3850
3851 /* How deep are we in the directory tree?
3852 * Counts backward from value of level parameter to vim_findfile_init
3853 */
3854 int ffs_level;
3855
3856 /* Did we already expand '**' to an empty string? */
3857 int ffs_star_star_empty;
3858} ff_stack_T;
3859
3860/*
3861 * type for already visited directories or files.
3862 */
3863typedef struct ff_visited
3864{
3865 struct ff_visited *ffv_next;
3866
3867#ifdef FEAT_PATH_EXTRA
3868 /* Visited directories are different if the wildcard string are
3869 * different. So we have to save it.
3870 */
3871 char_u *ffv_wc_path;
3872#endif
3873 /* for unix use inode etc for comparison (needed because of links), else
3874 * use filename.
3875 */
3876#ifdef UNIX
Bram Moolenaare1fbddc2009-05-16 19:07:03 +00003877 int ffv_dev_valid; /* ffv_dev and ffv_ino were set */
3878 dev_t ffv_dev; /* device number */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003879 ino_t ffv_ino; /* inode number */
3880#endif
3881 /* The memory for this struct is allocated according to the length of
3882 * ffv_fname.
3883 */
3884 char_u ffv_fname[1]; /* actually longer */
3885} ff_visited_T;
3886
3887/*
3888 * We might have to manage several visited lists during a search.
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00003889 * This is especially needed for the tags option. If tags is set to:
Bram Moolenaar071d4272004-06-13 20:20:40 +00003890 * "./++/tags,./++/TAGS,++/tags" (replace + with *)
3891 * So we have to do 3 searches:
3892 * 1) search from the current files directory downward for the file "tags"
3893 * 2) search from the current files directory downward for the file "TAGS"
3894 * 3) search from Vims current directory downwards for the file "tags"
3895 * As you can see, the first and the third search are for the same file, so for
3896 * the third search we can use the visited list of the first search. For the
3897 * second search we must start from a empty visited list.
3898 * The struct ff_visited_list_hdr is used to manage a linked list of already
3899 * visited lists.
3900 */
3901typedef struct ff_visited_list_hdr
3902{
3903 struct ff_visited_list_hdr *ffvl_next;
3904
3905 /* the filename the attached visited list is for */
3906 char_u *ffvl_filename;
3907
3908 ff_visited_T *ffvl_visited_list;
3909
3910} ff_visited_list_hdr_T;
3911
3912
3913/*
3914 * '**' can be expanded to several directory levels.
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00003915 * Set the default maximum depth.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003916 */
3917#define FF_MAX_STAR_STAR_EXPAND ((char_u)30)
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00003918
Bram Moolenaar071d4272004-06-13 20:20:40 +00003919/*
3920 * The search context:
3921 * ffsc_stack_ptr: the stack for the dirs to search
3922 * ffsc_visited_list: the currently active visited list
3923 * ffsc_dir_visited_list: the currently active visited list for search dirs
3924 * ffsc_visited_lists_list: the list of all visited lists
3925 * ffsc_dir_visited_lists_list: the list of all visited lists for search dirs
3926 * ffsc_file_to_search: the file to search for
3927 * ffsc_start_dir: the starting directory, if search path was relative
3928 * ffsc_fix_path: the fix part of the given path (without wildcards)
3929 * Needed for upward search.
3930 * ffsc_wc_path: the part of the given path containing wildcards
3931 * ffsc_level: how many levels of dirs to search downwards
3932 * ffsc_stopdirs_v: array of stop directories for upward search
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00003933 * ffsc_find_what: FINDFILE_BOTH, FINDFILE_DIR or FINDFILE_FILE
Bram Moolenaar071d4272004-06-13 20:20:40 +00003934 */
3935typedef struct ff_search_ctx_T
3936{
3937 ff_stack_T *ffsc_stack_ptr;
3938 ff_visited_list_hdr_T *ffsc_visited_list;
3939 ff_visited_list_hdr_T *ffsc_dir_visited_list;
3940 ff_visited_list_hdr_T *ffsc_visited_lists_list;
3941 ff_visited_list_hdr_T *ffsc_dir_visited_lists_list;
3942 char_u *ffsc_file_to_search;
3943 char_u *ffsc_start_dir;
3944 char_u *ffsc_fix_path;
3945#ifdef FEAT_PATH_EXTRA
3946 char_u *ffsc_wc_path;
3947 int ffsc_level;
3948 char_u **ffsc_stopdirs_v;
3949#endif
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00003950 int ffsc_find_what;
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00003951} ff_search_ctx_T;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003952
Bram Moolenaar071d4272004-06-13 20:20:40 +00003953/* locally needed functions */
3954#ifdef FEAT_PATH_EXTRA
3955static int ff_check_visited __ARGS((ff_visited_T **, char_u *, char_u *));
3956#else
3957static int ff_check_visited __ARGS((ff_visited_T **, char_u *));
3958#endif
3959static void vim_findfile_free_visited_list __ARGS((ff_visited_list_hdr_T **list_headp));
3960static void ff_free_visited_list __ARGS((ff_visited_T *vl));
3961static ff_visited_list_hdr_T* ff_get_visited_list __ARGS((char_u *, ff_visited_list_hdr_T **list_headp));
3962#ifdef FEAT_PATH_EXTRA
3963static int ff_wc_equal __ARGS((char_u *s1, char_u *s2));
3964#endif
3965
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00003966static void ff_push __ARGS((ff_search_ctx_T *search_ctx, ff_stack_T *stack_ptr));
3967static ff_stack_T *ff_pop __ARGS((ff_search_ctx_T *search_ctx));
3968static void ff_clear __ARGS((ff_search_ctx_T *search_ctx));
3969static void ff_free_stack_element __ARGS((ff_stack_T *stack_ptr));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003970#ifdef FEAT_PATH_EXTRA
3971static ff_stack_T *ff_create_stack_element __ARGS((char_u *, char_u *, int, int));
3972#else
3973static ff_stack_T *ff_create_stack_element __ARGS((char_u *, int, int));
3974#endif
3975#ifdef FEAT_PATH_EXTRA
3976static int ff_path_in_stoplist __ARGS((char_u *, int, char_u **));
3977#endif
3978
Bram Moolenaar071d4272004-06-13 20:20:40 +00003979#if 0
3980/*
3981 * if someone likes findfirst/findnext, here are the functions
3982 * NOT TESTED!!
3983 */
3984
3985static void *ff_fn_search_context = NULL;
3986
3987 char_u *
3988vim_findfirst(path, filename, level)
3989 char_u *path;
3990 char_u *filename;
3991 int level;
3992{
3993 ff_fn_search_context =
3994 vim_findfile_init(path, filename, NULL, level, TRUE, FALSE,
3995 ff_fn_search_context, rel_fname);
3996 if (NULL == ff_fn_search_context)
3997 return NULL;
3998 else
3999 return vim_findnext()
4000}
4001
4002 char_u *
4003vim_findnext()
4004{
4005 char_u *ret = vim_findfile(ff_fn_search_context);
4006
4007 if (NULL == ret)
4008 {
4009 vim_findfile_cleanup(ff_fn_search_context);
4010 ff_fn_search_context = NULL;
4011 }
4012 return ret;
4013}
4014#endif
4015
4016/*
4017 * Initialization routine for vim_findfile.
4018 *
4019 * Returns the newly allocated search context or NULL if an error occured.
4020 *
4021 * Don't forget to clean up by calling vim_findfile_cleanup() if you are done
4022 * with the search context.
4023 *
4024 * Find the file 'filename' in the directory 'path'.
4025 * The parameter 'path' may contain wildcards. If so only search 'level'
4026 * directories deep. The parameter 'level' is the absolute maximum and is
4027 * not related to restricts given to the '**' wildcard. If 'level' is 100
4028 * and you use '**200' vim_findfile() will stop after 100 levels.
4029 *
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004030 * 'filename' cannot contain wildcards! It is used as-is, no backslashes to
4031 * escape special characters.
4032 *
Bram Moolenaar071d4272004-06-13 20:20:40 +00004033 * If 'stopdirs' is not NULL and nothing is found downward, the search is
4034 * restarted on the next higher directory level. This is repeated until the
4035 * start-directory of a search is contained in 'stopdirs'. 'stopdirs' has the
4036 * format ";*<dirname>*\(;<dirname>\)*;\=$".
4037 *
4038 * If the 'path' is relative, the starting dir for the search is either VIM's
4039 * current dir or if the path starts with "./" the current files dir.
4040 * If the 'path' is absolut, the starting dir is that part of the path before
4041 * the first wildcard.
4042 *
4043 * Upward search is only done on the starting dir.
4044 *
4045 * If 'free_visited' is TRUE the list of already visited files/directories is
4046 * cleared. Set this to FALSE if you just want to search from another
4047 * directory, but want to be sure that no directory from a previous search is
4048 * searched again. This is useful if you search for a file at different places.
4049 * The list of visited files/dirs can also be cleared with the function
4050 * vim_findfile_free_visited().
4051 *
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004052 * Set the parameter 'find_what' to FINDFILE_DIR if you want to search for
4053 * directories only, FINDFILE_FILE for files only, FINDFILE_BOTH for both.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004054 *
4055 * A search context returned by a previous call to vim_findfile_init() can be
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004056 * passed in the parameter "search_ctx_arg". This context is reused and
4057 * reinitialized with the new parameters. The list of already visited
Bram Moolenaar071d4272004-06-13 20:20:40 +00004058 * directories from this context is only deleted if the parameter
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004059 * "free_visited" is true. Be aware that the passed "search_ctx_arg" is freed
4060 * if the reinitialization fails.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004061 *
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004062 * If you don't have a search context from a previous call "search_ctx_arg"
4063 * must be NULL.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004064 *
4065 * This function silently ignores a few errors, vim_findfile() will have
4066 * limited functionality then.
4067 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004068 void *
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004069vim_findfile_init(path, filename, stopdirs, level, free_visited, find_what,
4070 search_ctx_arg, tagfile, rel_fname)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004071 char_u *path;
4072 char_u *filename;
Bram Moolenaare1fbddc2009-05-16 19:07:03 +00004073 char_u *stopdirs UNUSED;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004074 int level;
4075 int free_visited;
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004076 int find_what;
4077 void *search_ctx_arg;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004078 int tagfile;
4079 char_u *rel_fname; /* file name to use for "." */
4080{
4081#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004082 char_u *wc_part;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004083#endif
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004084 ff_stack_T *sptr;
4085 ff_search_ctx_T *search_ctx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004086
4087 /* If a search context is given by the caller, reuse it, else allocate a
4088 * new one.
4089 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004090 if (search_ctx_arg != NULL)
4091 search_ctx = search_ctx_arg;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004092 else
4093 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004094 search_ctx = (ff_search_ctx_T*)alloc((unsigned)sizeof(ff_search_ctx_T));
4095 if (search_ctx == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004096 goto error_return;
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004097 memset(search_ctx, 0, sizeof(ff_search_ctx_T));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004098 }
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004099 search_ctx->ffsc_find_what = find_what;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004100
4101 /* clear the search context, but NOT the visited lists */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004102 ff_clear(search_ctx);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004103
4104 /* clear visited list if wanted */
4105 if (free_visited == TRUE)
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004106 vim_findfile_free_visited(search_ctx);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004107 else
4108 {
4109 /* Reuse old visited lists. Get the visited list for the given
4110 * filename. If no list for the current filename exists, creates a new
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004111 * one. */
4112 search_ctx->ffsc_visited_list = ff_get_visited_list(filename,
4113 &search_ctx->ffsc_visited_lists_list);
4114 if (search_ctx->ffsc_visited_list == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004115 goto error_return;
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004116 search_ctx->ffsc_dir_visited_list = ff_get_visited_list(filename,
4117 &search_ctx->ffsc_dir_visited_lists_list);
4118 if (search_ctx->ffsc_dir_visited_list == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004119 goto error_return;
4120 }
4121
4122 if (ff_expand_buffer == NULL)
4123 {
4124 ff_expand_buffer = (char_u*)alloc(MAXPATHL);
4125 if (ff_expand_buffer == NULL)
4126 goto error_return;
4127 }
4128
4129 /* Store information on starting dir now if path is relative.
Bram Moolenaar5eb86f92004-07-26 12:53:41 +00004130 * If path is absolute, we do that later. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004131 if (path[0] == '.'
4132 && (vim_ispathsep(path[1]) || path[1] == NUL)
4133 && (!tagfile || vim_strchr(p_cpo, CPO_DOTTAG) == NULL)
4134 && rel_fname != NULL)
4135 {
4136 int len = (int)(gettail(rel_fname) - rel_fname);
4137
4138 if (!vim_isAbsName(rel_fname) && len + 1 < MAXPATHL)
4139 {
4140 /* Make the start dir an absolute path name. */
Bram Moolenaarbbebc852005-07-18 21:47:53 +00004141 vim_strncpy(ff_expand_buffer, rel_fname, len);
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004142 search_ctx->ffsc_start_dir = FullName_save(ff_expand_buffer, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004143 }
4144 else
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004145 search_ctx->ffsc_start_dir = vim_strnsave(rel_fname, len);
4146 if (search_ctx->ffsc_start_dir == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004147 goto error_return;
4148 if (*++path != NUL)
4149 ++path;
4150 }
4151 else if (*path == NUL || !vim_isAbsName(path))
4152 {
4153#ifdef BACKSLASH_IN_FILENAME
4154 /* "c:dir" needs "c:" to be expanded, otherwise use current dir */
4155 if (*path != NUL && path[1] == ':')
4156 {
4157 char_u drive[3];
4158
4159 drive[0] = path[0];
4160 drive[1] = ':';
4161 drive[2] = NUL;
4162 if (vim_FullName(drive, ff_expand_buffer, MAXPATHL, TRUE) == FAIL)
4163 goto error_return;
4164 path += 2;
4165 }
4166 else
4167#endif
4168 if (mch_dirname(ff_expand_buffer, MAXPATHL) == FAIL)
4169 goto error_return;
4170
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004171 search_ctx->ffsc_start_dir = vim_strsave(ff_expand_buffer);
4172 if (search_ctx->ffsc_start_dir == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004173 goto error_return;
4174
4175#ifdef BACKSLASH_IN_FILENAME
4176 /* A path that starts with "/dir" is relative to the drive, not to the
4177 * directory (but not for "//machine/dir"). Only use the drive name. */
4178 if ((*path == '/' || *path == '\\')
4179 && path[1] != path[0]
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004180 && search_ctx->ffsc_start_dir[1] == ':')
4181 search_ctx->ffsc_start_dir[2] = NUL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004182#endif
4183 }
4184
4185#ifdef FEAT_PATH_EXTRA
4186 /*
4187 * If stopdirs are given, split them into an array of pointers.
4188 * If this fails (mem allocation), there is no upward search at all or a
4189 * stop directory is not recognized -> continue silently.
4190 * If stopdirs just contains a ";" or is empty,
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004191 * search_ctx->ffsc_stopdirs_v will only contain a NULL pointer. This
Bram Moolenaar071d4272004-06-13 20:20:40 +00004192 * is handled as unlimited upward search. See function
4193 * ff_path_in_stoplist() for details.
4194 */
4195 if (stopdirs != NULL)
4196 {
4197 char_u *walker = stopdirs;
4198 int dircount;
4199
4200 while (*walker == ';')
4201 walker++;
4202
4203 dircount = 1;
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004204 search_ctx->ffsc_stopdirs_v =
4205 (char_u **)alloc((unsigned)sizeof(char_u *));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004206
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004207 if (search_ctx->ffsc_stopdirs_v != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004208 {
4209 do
4210 {
4211 char_u *helper;
4212 void *ptr;
4213
4214 helper = walker;
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004215 ptr = vim_realloc(search_ctx->ffsc_stopdirs_v,
Bram Moolenaar071d4272004-06-13 20:20:40 +00004216 (dircount + 1) * sizeof(char_u *));
4217 if (ptr)
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004218 search_ctx->ffsc_stopdirs_v = ptr;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004219 else
4220 /* ignore, keep what we have and continue */
4221 break;
4222 walker = vim_strchr(walker, ';');
4223 if (walker)
4224 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004225 search_ctx->ffsc_stopdirs_v[dircount-1] =
4226 vim_strnsave(helper, (int)(walker - helper));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004227 walker++;
4228 }
4229 else
4230 /* this might be "", which means ascent till top
4231 * of directory tree.
4232 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004233 search_ctx->ffsc_stopdirs_v[dircount-1] =
4234 vim_strsave(helper);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004235
4236 dircount++;
4237
4238 } while (walker != NULL);
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004239 search_ctx->ffsc_stopdirs_v[dircount-1] = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004240 }
4241 }
4242#endif
4243
4244#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004245 search_ctx->ffsc_level = level;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004246
4247 /* split into:
4248 * -fix path
4249 * -wildcard_stuff (might be NULL)
4250 */
4251 wc_part = vim_strchr(path, '*');
4252 if (wc_part != NULL)
4253 {
4254 int llevel;
4255 int len;
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00004256 char *errpt;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004257
4258 /* save the fix part of the path */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004259 search_ctx->ffsc_fix_path = vim_strnsave(path, (int)(wc_part - path));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004260
4261 /*
4262 * copy wc_path and add restricts to the '**' wildcard.
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00004263 * The octet after a '**' is used as a (binary) counter.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004264 * So '**3' is transposed to '**^C' ('^C' is ASCII value 3)
4265 * or '**76' is transposed to '**N'( 'N' is ASCII value 76).
4266 * For EBCDIC you get different character values.
4267 * If no restrict is given after '**' the default is used.
Bram Moolenaar21160b92009-11-11 15:56:10 +00004268 * Due to this technique the path looks awful if you print it as a
Bram Moolenaar071d4272004-06-13 20:20:40 +00004269 * string.
4270 */
4271 len = 0;
4272 while (*wc_part != NUL)
4273 {
4274 if (STRNCMP(wc_part, "**", 2) == 0)
4275 {
4276 ff_expand_buffer[len++] = *wc_part++;
4277 ff_expand_buffer[len++] = *wc_part++;
4278
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00004279 llevel = strtol((char *)wc_part, &errpt, 10);
4280 if ((char_u *)errpt != wc_part && llevel > 0 && llevel < 255)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004281 ff_expand_buffer[len++] = llevel;
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00004282 else if ((char_u *)errpt != wc_part && llevel == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004283 /* restrict is 0 -> remove already added '**' */
4284 len -= 2;
4285 else
4286 ff_expand_buffer[len++] = FF_MAX_STAR_STAR_EXPAND;
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00004287 wc_part = (char_u *)errpt;
Bram Moolenaar1d94f9b2005-08-04 21:29:45 +00004288 if (*wc_part != NUL && !vim_ispathsep(*wc_part))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004289 {
4290 EMSG2(_("E343: Invalid path: '**[number]' must be at the end of the path or be followed by '%s'."), PATHSEPSTR);
4291 goto error_return;
4292 }
4293 }
4294 else
4295 ff_expand_buffer[len++] = *wc_part++;
4296 }
4297 ff_expand_buffer[len] = NUL;
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004298 search_ctx->ffsc_wc_path = vim_strsave(ff_expand_buffer);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004299
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004300 if (search_ctx->ffsc_wc_path == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004301 goto error_return;
4302 }
4303 else
4304#endif
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004305 search_ctx->ffsc_fix_path = vim_strsave(path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004306
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004307 if (search_ctx->ffsc_start_dir == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004308 {
4309 /* store the fix part as startdir.
4310 * This is needed if the parameter path is fully qualified.
4311 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004312 search_ctx->ffsc_start_dir = vim_strsave(search_ctx->ffsc_fix_path);
4313 if (search_ctx->ffsc_start_dir)
4314 search_ctx->ffsc_fix_path[0] = NUL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004315 }
4316
4317 /* create an absolute path */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004318 STRCPY(ff_expand_buffer, search_ctx->ffsc_start_dir);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004319 add_pathsep(ff_expand_buffer);
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004320 STRCAT(ff_expand_buffer, search_ctx->ffsc_fix_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004321 add_pathsep(ff_expand_buffer);
4322
4323 sptr = ff_create_stack_element(ff_expand_buffer,
4324#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004325 search_ctx->ffsc_wc_path,
Bram Moolenaar071d4272004-06-13 20:20:40 +00004326#endif
4327 level, 0);
4328
4329 if (sptr == NULL)
4330 goto error_return;
4331
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004332 ff_push(search_ctx, sptr);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004333
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004334 search_ctx->ffsc_file_to_search = vim_strsave(filename);
4335 if (search_ctx->ffsc_file_to_search == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004336 goto error_return;
4337
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004338 return search_ctx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004339
4340error_return:
4341 /*
4342 * We clear the search context now!
4343 * Even when the caller gave us a (perhaps valid) context we free it here,
4344 * as we might have already destroyed it.
4345 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004346 vim_findfile_cleanup(search_ctx);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004347 return NULL;
4348}
4349
4350#if defined(FEAT_PATH_EXTRA) || defined(PROTO)
4351/*
4352 * Get the stopdir string. Check that ';' is not escaped.
4353 */
4354 char_u *
4355vim_findfile_stopdir(buf)
4356 char_u *buf;
4357{
4358 char_u *r_ptr = buf;
4359
4360 while (*r_ptr != NUL && *r_ptr != ';')
4361 {
4362 if (r_ptr[0] == '\\' && r_ptr[1] == ';')
4363 {
4364 /* overwrite the escape char,
4365 * use STRLEN(r_ptr) to move the trailing '\0'
4366 */
Bram Moolenaar446cb832008-06-24 21:56:24 +00004367 STRMOVE(r_ptr, r_ptr + 1);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004368 r_ptr++;
4369 }
4370 r_ptr++;
4371 }
4372 if (*r_ptr == ';')
4373 {
4374 *r_ptr = 0;
4375 r_ptr++;
4376 }
4377 else if (*r_ptr == NUL)
4378 r_ptr = NULL;
4379 return r_ptr;
4380}
4381#endif
4382
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004383/*
4384 * Clean up the given search context. Can handle a NULL pointer.
4385 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004386 void
4387vim_findfile_cleanup(ctx)
4388 void *ctx;
4389{
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00004390 if (ctx == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004391 return;
4392
Bram Moolenaar071d4272004-06-13 20:20:40 +00004393 vim_findfile_free_visited(ctx);
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004394 ff_clear(ctx);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004395 vim_free(ctx);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004396}
4397
4398/*
4399 * Find a file in a search context.
4400 * The search context was created with vim_findfile_init() above.
4401 * Return a pointer to an allocated file name or NULL if nothing found.
4402 * To get all matching files call this function until you get NULL.
4403 *
Bram Moolenaar5eb86f92004-07-26 12:53:41 +00004404 * If the passed search_context is NULL, NULL is returned.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004405 *
4406 * The search algorithm is depth first. To change this replace the
4407 * stack with a list (don't forget to leave partly searched directories on the
4408 * top of the list).
4409 */
4410 char_u *
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004411vim_findfile(search_ctx_arg)
4412 void *search_ctx_arg;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004413{
4414 char_u *file_path;
4415#ifdef FEAT_PATH_EXTRA
4416 char_u *rest_of_wildcards;
4417 char_u *path_end = NULL;
4418#endif
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004419 ff_stack_T *stackp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004420#if defined(FEAT_SEARCHPATH) || defined(FEAT_PATH_EXTRA)
4421 int len;
4422#endif
4423 int i;
4424 char_u *p;
4425#ifdef FEAT_SEARCHPATH
4426 char_u *suf;
4427#endif
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004428 ff_search_ctx_T *search_ctx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004429
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004430 if (search_ctx_arg == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004431 return NULL;
4432
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004433 search_ctx = (ff_search_ctx_T *)search_ctx_arg;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004434
4435 /*
4436 * filepath is used as buffer for various actions and as the storage to
4437 * return a found filename.
4438 */
4439 if ((file_path = alloc((int)MAXPATHL)) == NULL)
4440 return NULL;
4441
4442#ifdef FEAT_PATH_EXTRA
4443 /* store the end of the start dir -- needed for upward search */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004444 if (search_ctx->ffsc_start_dir != NULL)
4445 path_end = &search_ctx->ffsc_start_dir[
4446 STRLEN(search_ctx->ffsc_start_dir)];
Bram Moolenaar071d4272004-06-13 20:20:40 +00004447#endif
4448
4449#ifdef FEAT_PATH_EXTRA
4450 /* upward search loop */
4451 for (;;)
4452 {
4453#endif
4454 /* downward search loop */
4455 for (;;)
4456 {
4457 /* check if user user wants to stop the search*/
4458 ui_breakcheck();
4459 if (got_int)
4460 break;
4461
4462 /* get directory to work on from stack */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004463 stackp = ff_pop(search_ctx);
4464 if (stackp == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004465 break;
4466
4467 /*
4468 * TODO: decide if we leave this test in
4469 *
4470 * GOOD: don't search a directory(-tree) twice.
4471 * BAD: - check linked list for every new directory entered.
4472 * - check for double files also done below
4473 *
4474 * Here we check if we already searched this directory.
4475 * We already searched a directory if:
4476 * 1) The directory is the same.
4477 * 2) We would use the same wildcard string.
4478 *
4479 * Good if you have links on same directory via several ways
4480 * or you have selfreferences in directories (e.g. SuSE Linux 6.3:
4481 * /etc/rc.d/init.d is linked to /etc/rc.d -> endless loop)
4482 *
4483 * This check is only needed for directories we work on for the
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004484 * first time (hence stackp->ff_filearray == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004485 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004486 if (stackp->ffs_filearray == NULL
4487 && ff_check_visited(&search_ctx->ffsc_dir_visited_list
Bram Moolenaar071d4272004-06-13 20:20:40 +00004488 ->ffvl_visited_list,
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004489 stackp->ffs_fix_path
Bram Moolenaar071d4272004-06-13 20:20:40 +00004490#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004491 , stackp->ffs_wc_path
Bram Moolenaar071d4272004-06-13 20:20:40 +00004492#endif
4493 ) == FAIL)
4494 {
4495#ifdef FF_VERBOSE
4496 if (p_verbose >= 5)
4497 {
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004498 verbose_enter_scroll();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004499 smsg((char_u *)"Already Searched: %s (%s)",
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004500 stackp->ffs_fix_path, stackp->ffs_wc_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004501 /* don't overwrite this either */
4502 msg_puts((char_u *)"\n");
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004503 verbose_leave_scroll();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004504 }
4505#endif
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004506 ff_free_stack_element(stackp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004507 continue;
4508 }
4509#ifdef FF_VERBOSE
4510 else if (p_verbose >= 5)
4511 {
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004512 verbose_enter_scroll();
Bram Moolenaar051b7822005-05-19 21:00:46 +00004513 smsg((char_u *)"Searching: %s (%s)",
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004514 stackp->ffs_fix_path, stackp->ffs_wc_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004515 /* don't overwrite this either */
4516 msg_puts((char_u *)"\n");
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004517 verbose_leave_scroll();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004518 }
4519#endif
4520
4521 /* check depth */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004522 if (stackp->ffs_level <= 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004523 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004524 ff_free_stack_element(stackp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004525 continue;
4526 }
4527
4528 file_path[0] = NUL;
4529
4530 /*
4531 * If no filearray till now expand wildcards
4532 * The function expand_wildcards() can handle an array of paths
4533 * and all possible expands are returned in one array. We use this
4534 * to handle the expansion of '**' into an empty string.
4535 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004536 if (stackp->ffs_filearray == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004537 {
4538 char_u *dirptrs[2];
4539
4540 /* we use filepath to build the path expand_wildcards() should
4541 * expand.
4542 */
4543 dirptrs[0] = file_path;
4544 dirptrs[1] = NULL;
4545
4546 /* if we have a start dir copy it in */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004547 if (!vim_isAbsName(stackp->ffs_fix_path)
4548 && search_ctx->ffsc_start_dir)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004549 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004550 STRCPY(file_path, search_ctx->ffsc_start_dir);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004551 add_pathsep(file_path);
4552 }
4553
4554 /* append the fix part of the search path */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004555 STRCAT(file_path, stackp->ffs_fix_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004556 add_pathsep(file_path);
4557
4558#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004559 rest_of_wildcards = stackp->ffs_wc_path;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004560 if (*rest_of_wildcards != NUL)
4561 {
4562 len = (int)STRLEN(file_path);
4563 if (STRNCMP(rest_of_wildcards, "**", 2) == 0)
4564 {
4565 /* pointer to the restrict byte
4566 * The restrict byte is not a character!
4567 */
4568 p = rest_of_wildcards + 2;
4569
4570 if (*p > 0)
4571 {
4572 (*p)--;
4573 file_path[len++] = '*';
4574 }
4575
4576 if (*p == 0)
4577 {
4578 /* remove '**<numb> from wildcards */
Bram Moolenaar446cb832008-06-24 21:56:24 +00004579 STRMOVE(rest_of_wildcards, rest_of_wildcards + 3);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004580 }
4581 else
4582 rest_of_wildcards += 3;
4583
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004584 if (stackp->ffs_star_star_empty == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004585 {
4586 /* if not done before, expand '**' to empty */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004587 stackp->ffs_star_star_empty = 1;
4588 dirptrs[1] = stackp->ffs_fix_path;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004589 }
4590 }
4591
4592 /*
4593 * Here we copy until the next path separator or the end of
4594 * the path. If we stop at a path separator, there is
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00004595 * still something else left. This is handled below by
Bram Moolenaar071d4272004-06-13 20:20:40 +00004596 * pushing every directory returned from expand_wildcards()
4597 * on the stack again for further search.
4598 */
4599 while (*rest_of_wildcards
4600 && !vim_ispathsep(*rest_of_wildcards))
4601 file_path[len++] = *rest_of_wildcards++;
4602
4603 file_path[len] = NUL;
4604 if (vim_ispathsep(*rest_of_wildcards))
4605 rest_of_wildcards++;
4606 }
4607#endif
4608
4609 /*
4610 * Expand wildcards like "*" and "$VAR".
4611 * If the path is a URL don't try this.
4612 */
4613 if (path_with_url(dirptrs[0]))
4614 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004615 stackp->ffs_filearray = (char_u **)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004616 alloc((unsigned)sizeof(char *));
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004617 if (stackp->ffs_filearray != NULL
4618 && (stackp->ffs_filearray[0]
Bram Moolenaar071d4272004-06-13 20:20:40 +00004619 = vim_strsave(dirptrs[0])) != NULL)
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004620 stackp->ffs_filearray_size = 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004621 else
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004622 stackp->ffs_filearray_size = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004623 }
4624 else
4625 expand_wildcards((dirptrs[1] == NULL) ? 1 : 2, dirptrs,
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004626 &stackp->ffs_filearray_size,
4627 &stackp->ffs_filearray,
Bram Moolenaar071d4272004-06-13 20:20:40 +00004628 EW_DIR|EW_ADDSLASH|EW_SILENT);
4629
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004630 stackp->ffs_filearray_cur = 0;
4631 stackp->ffs_stage = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004632 }
4633#ifdef FEAT_PATH_EXTRA
4634 else
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004635 rest_of_wildcards = &stackp->ffs_wc_path[
4636 STRLEN(stackp->ffs_wc_path)];
Bram Moolenaar071d4272004-06-13 20:20:40 +00004637#endif
4638
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004639 if (stackp->ffs_stage == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004640 {
4641 /* this is the first time we work on this directory */
4642#ifdef FEAT_PATH_EXTRA
4643 if (*rest_of_wildcards == NUL)
4644#endif
4645 {
4646 /*
4647 * we don't have further wildcards to expand, so we have to
4648 * check for the final file now
4649 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004650 for (i = stackp->ffs_filearray_cur;
4651 i < stackp->ffs_filearray_size; ++i)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004652 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004653 if (!path_with_url(stackp->ffs_filearray[i])
4654 && !mch_isdir(stackp->ffs_filearray[i]))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004655 continue; /* not a directory */
4656
Bram Moolenaar21160b92009-11-11 15:56:10 +00004657 /* prepare the filename to be checked for existence
Bram Moolenaar071d4272004-06-13 20:20:40 +00004658 * below */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004659 STRCPY(file_path, stackp->ffs_filearray[i]);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004660 add_pathsep(file_path);
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004661 STRCAT(file_path, search_ctx->ffsc_file_to_search);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004662
4663 /*
4664 * Try without extra suffix and then with suffixes
4665 * from 'suffixesadd'.
4666 */
4667#ifdef FEAT_SEARCHPATH
4668 len = (int)STRLEN(file_path);
4669 suf = curbuf->b_p_sua;
4670 for (;;)
4671#endif
4672 {
4673 /* if file exists and we didn't already find it */
4674 if ((path_with_url(file_path)
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004675 || (mch_getperm(file_path) >= 0
4676 && (search_ctx->ffsc_find_what
4677 == FINDFILE_BOTH
4678 || ((search_ctx->ffsc_find_what
4679 == FINDFILE_DIR)
4680 == mch_isdir(file_path)))))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004681#ifndef FF_VERBOSE
4682 && (ff_check_visited(
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004683 &search_ctx->ffsc_visited_list->ffvl_visited_list,
Bram Moolenaar071d4272004-06-13 20:20:40 +00004684 file_path
4685#ifdef FEAT_PATH_EXTRA
4686 , (char_u *)""
4687#endif
4688 ) == OK)
4689#endif
4690 )
4691 {
4692#ifdef FF_VERBOSE
4693 if (ff_check_visited(
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004694 &search_ctx->ffsc_visited_list->ffvl_visited_list,
Bram Moolenaar071d4272004-06-13 20:20:40 +00004695 file_path
4696#ifdef FEAT_PATH_EXTRA
4697 , (char_u *)""
4698#endif
4699 ) == FAIL)
4700 {
4701 if (p_verbose >= 5)
4702 {
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004703 verbose_enter_scroll();
Bram Moolenaar051b7822005-05-19 21:00:46 +00004704 smsg((char_u *)"Already: %s",
Bram Moolenaar071d4272004-06-13 20:20:40 +00004705 file_path);
4706 /* don't overwrite this either */
4707 msg_puts((char_u *)"\n");
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004708 verbose_leave_scroll();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004709 }
4710 continue;
4711 }
4712#endif
4713
4714 /* push dir to examine rest of subdirs later */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004715 stackp->ffs_filearray_cur = i + 1;
4716 ff_push(search_ctx, stackp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004717
Bram Moolenaar6bab9fa2009-01-22 20:32:12 +00004718 if (!path_with_url(file_path))
4719 simplify_filename(file_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004720 if (mch_dirname(ff_expand_buffer, MAXPATHL)
4721 == OK)
4722 {
4723 p = shorten_fname(file_path,
4724 ff_expand_buffer);
4725 if (p != NULL)
Bram Moolenaar446cb832008-06-24 21:56:24 +00004726 STRMOVE(file_path, p);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004727 }
4728#ifdef FF_VERBOSE
4729 if (p_verbose >= 5)
4730 {
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004731 verbose_enter_scroll();
Bram Moolenaar051b7822005-05-19 21:00:46 +00004732 smsg((char_u *)"HIT: %s", file_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004733 /* don't overwrite this either */
4734 msg_puts((char_u *)"\n");
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004735 verbose_leave_scroll();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004736 }
4737#endif
4738 return file_path;
4739 }
4740
4741#ifdef FEAT_SEARCHPATH
4742 /* Not found or found already, try next suffix. */
4743 if (*suf == NUL)
4744 break;
4745 copy_option_part(&suf, file_path + len,
4746 MAXPATHL - len, ",");
4747#endif
4748 }
4749 }
4750 }
4751#ifdef FEAT_PATH_EXTRA
4752 else
4753 {
4754 /*
4755 * still wildcards left, push the directories for further
4756 * search
4757 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004758 for (i = stackp->ffs_filearray_cur;
4759 i < stackp->ffs_filearray_size; ++i)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004760 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004761 if (!mch_isdir(stackp->ffs_filearray[i]))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004762 continue; /* not a directory */
4763
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004764 ff_push(search_ctx,
4765 ff_create_stack_element(
4766 stackp->ffs_filearray[i],
4767 rest_of_wildcards,
4768 stackp->ffs_level - 1, 0));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004769 }
4770 }
4771#endif
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004772 stackp->ffs_filearray_cur = 0;
4773 stackp->ffs_stage = 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004774 }
4775
4776#ifdef FEAT_PATH_EXTRA
4777 /*
4778 * if wildcards contains '**' we have to descent till we reach the
4779 * leaves of the directory tree.
4780 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004781 if (STRNCMP(stackp->ffs_wc_path, "**", 2) == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004782 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004783 for (i = stackp->ffs_filearray_cur;
4784 i < stackp->ffs_filearray_size; ++i)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004785 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004786 if (fnamecmp(stackp->ffs_filearray[i],
4787 stackp->ffs_fix_path) == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004788 continue; /* don't repush same directory */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004789 if (!mch_isdir(stackp->ffs_filearray[i]))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004790 continue; /* not a directory */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004791 ff_push(search_ctx,
4792 ff_create_stack_element(stackp->ffs_filearray[i],
4793 stackp->ffs_wc_path, stackp->ffs_level - 1, 1));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004794 }
4795 }
4796#endif
4797
4798 /* we are done with the current directory */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004799 ff_free_stack_element(stackp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004800
4801 }
4802
4803#ifdef FEAT_PATH_EXTRA
4804 /* If we reached this, we didn't find anything downwards.
4805 * Let's check if we should do an upward search.
4806 */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004807 if (search_ctx->ffsc_start_dir
4808 && search_ctx->ffsc_stopdirs_v != NULL && !got_int)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004809 {
4810 ff_stack_T *sptr;
4811
4812 /* is the last starting directory in the stop list? */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004813 if (ff_path_in_stoplist(search_ctx->ffsc_start_dir,
4814 (int)(path_end - search_ctx->ffsc_start_dir),
4815 search_ctx->ffsc_stopdirs_v) == TRUE)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004816 break;
4817
4818 /* cut of last dir */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004819 while (path_end > search_ctx->ffsc_start_dir
4820 && vim_ispathsep(*path_end))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004821 path_end--;
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004822 while (path_end > search_ctx->ffsc_start_dir
4823 && !vim_ispathsep(path_end[-1]))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004824 path_end--;
4825 *path_end = 0;
4826 path_end--;
4827
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004828 if (*search_ctx->ffsc_start_dir == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004829 break;
4830
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004831 STRCPY(file_path, search_ctx->ffsc_start_dir);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004832 add_pathsep(file_path);
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004833 STRCAT(file_path, search_ctx->ffsc_fix_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004834
4835 /* create a new stack entry */
4836 sptr = ff_create_stack_element(file_path,
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004837 search_ctx->ffsc_wc_path, search_ctx->ffsc_level, 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004838 if (sptr == NULL)
4839 break;
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004840 ff_push(search_ctx, sptr);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004841 }
4842 else
4843 break;
4844 }
4845#endif
4846
4847 vim_free(file_path);
4848 return NULL;
4849}
4850
4851/*
4852 * Free the list of lists of visited files and directories
4853 * Can handle it if the passed search_context is NULL;
4854 */
4855 void
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004856vim_findfile_free_visited(search_ctx_arg)
4857 void *search_ctx_arg;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004858{
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004859 ff_search_ctx_T *search_ctx;
4860
4861 if (search_ctx_arg == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004862 return;
4863
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00004864 search_ctx = (ff_search_ctx_T *)search_ctx_arg;
4865 vim_findfile_free_visited_list(&search_ctx->ffsc_visited_lists_list);
4866 vim_findfile_free_visited_list(&search_ctx->ffsc_dir_visited_lists_list);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004867}
4868
4869 static void
4870vim_findfile_free_visited_list(list_headp)
4871 ff_visited_list_hdr_T **list_headp;
4872{
4873 ff_visited_list_hdr_T *vp;
4874
4875 while (*list_headp != NULL)
4876 {
4877 vp = (*list_headp)->ffvl_next;
4878 ff_free_visited_list((*list_headp)->ffvl_visited_list);
4879
4880 vim_free((*list_headp)->ffvl_filename);
4881 vim_free(*list_headp);
4882 *list_headp = vp;
4883 }
4884 *list_headp = NULL;
4885}
4886
4887 static void
4888ff_free_visited_list(vl)
4889 ff_visited_T *vl;
4890{
4891 ff_visited_T *vp;
4892
4893 while (vl != NULL)
4894 {
4895 vp = vl->ffv_next;
4896#ifdef FEAT_PATH_EXTRA
4897 vim_free(vl->ffv_wc_path);
4898#endif
4899 vim_free(vl);
4900 vl = vp;
4901 }
4902 vl = NULL;
4903}
4904
4905/*
4906 * Returns the already visited list for the given filename. If none is found it
4907 * allocates a new one.
4908 */
4909 static ff_visited_list_hdr_T*
4910ff_get_visited_list(filename, list_headp)
4911 char_u *filename;
4912 ff_visited_list_hdr_T **list_headp;
4913{
4914 ff_visited_list_hdr_T *retptr = NULL;
4915
4916 /* check if a visited list for the given filename exists */
4917 if (*list_headp != NULL)
4918 {
4919 retptr = *list_headp;
4920 while (retptr != NULL)
4921 {
4922 if (fnamecmp(filename, retptr->ffvl_filename) == 0)
4923 {
4924#ifdef FF_VERBOSE
4925 if (p_verbose >= 5)
4926 {
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004927 verbose_enter_scroll();
Bram Moolenaar051b7822005-05-19 21:00:46 +00004928 smsg((char_u *)"ff_get_visited_list: FOUND list for %s",
Bram Moolenaar071d4272004-06-13 20:20:40 +00004929 filename);
4930 /* don't overwrite this either */
4931 msg_puts((char_u *)"\n");
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004932 verbose_leave_scroll();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004933 }
4934#endif
4935 return retptr;
4936 }
4937 retptr = retptr->ffvl_next;
4938 }
4939 }
4940
4941#ifdef FF_VERBOSE
4942 if (p_verbose >= 5)
4943 {
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004944 verbose_enter_scroll();
Bram Moolenaar051b7822005-05-19 21:00:46 +00004945 smsg((char_u *)"ff_get_visited_list: new list for %s", filename);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004946 /* don't overwrite this either */
4947 msg_puts((char_u *)"\n");
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004948 verbose_leave_scroll();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004949 }
4950#endif
4951
4952 /*
4953 * if we reach this we didn't find a list and we have to allocate new list
4954 */
4955 retptr = (ff_visited_list_hdr_T*)alloc((unsigned)sizeof(*retptr));
4956 if (retptr == NULL)
4957 return NULL;
4958
4959 retptr->ffvl_visited_list = NULL;
4960 retptr->ffvl_filename = vim_strsave(filename);
4961 if (retptr->ffvl_filename == NULL)
4962 {
4963 vim_free(retptr);
4964 return NULL;
4965 }
4966 retptr->ffvl_next = *list_headp;
4967 *list_headp = retptr;
4968
4969 return retptr;
4970}
4971
4972#ifdef FEAT_PATH_EXTRA
4973/*
4974 * check if two wildcard paths are equal. Returns TRUE or FALSE.
4975 * They are equal if:
4976 * - both paths are NULL
4977 * - they have the same length
4978 * - char by char comparison is OK
4979 * - the only differences are in the counters behind a '**', so
4980 * '**\20' is equal to '**\24'
4981 */
4982 static int
4983ff_wc_equal(s1, s2)
4984 char_u *s1;
4985 char_u *s2;
4986{
4987 int i;
4988
4989 if (s1 == s2)
4990 return TRUE;
4991
4992 if (s1 == NULL || s2 == NULL)
4993 return FALSE;
4994
4995 if (STRLEN(s1) != STRLEN(s2))
4996 return FAIL;
4997
4998 for (i = 0; s1[i] != NUL && s2[i] != NUL; i++)
4999 {
5000 if (s1[i] != s2[i]
5001#ifdef CASE_INSENSITIVE_FILENAME
5002 && TOUPPER_LOC(s1[i]) != TOUPPER_LOC(s2[i])
5003#endif
5004 )
5005 {
5006 if (i >= 2)
5007 if (s1[i-1] == '*' && s1[i-2] == '*')
5008 continue;
5009 else
5010 return FAIL;
5011 else
5012 return FAIL;
5013 }
5014 }
5015 return TRUE;
5016}
5017#endif
5018
5019/*
5020 * maintains the list of already visited files and dirs
5021 * returns FAIL if the given file/dir is already in the list
5022 * returns OK if it is newly added
5023 *
5024 * TODO: What to do on memory allocation problems?
5025 * -> return TRUE - Better the file is found several times instead of
5026 * never.
5027 */
5028 static int
5029ff_check_visited(visited_list, fname
5030#ifdef FEAT_PATH_EXTRA
5031 , wc_path
5032#endif
5033 )
5034 ff_visited_T **visited_list;
5035 char_u *fname;
5036#ifdef FEAT_PATH_EXTRA
5037 char_u *wc_path;
5038#endif
5039{
5040 ff_visited_T *vp;
5041#ifdef UNIX
5042 struct stat st;
5043 int url = FALSE;
5044#endif
5045
5046 /* For an URL we only compare the name, otherwise we compare the
5047 * device/inode (unix) or the full path name (not Unix). */
5048 if (path_with_url(fname))
5049 {
Bram Moolenaarbbebc852005-07-18 21:47:53 +00005050 vim_strncpy(ff_expand_buffer, fname, MAXPATHL - 1);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005051#ifdef UNIX
5052 url = TRUE;
5053#endif
5054 }
5055 else
5056 {
5057 ff_expand_buffer[0] = NUL;
5058#ifdef UNIX
5059 if (mch_stat((char *)fname, &st) < 0)
5060#else
5061 if (vim_FullName(fname, ff_expand_buffer, MAXPATHL, TRUE) == FAIL)
5062#endif
5063 return FAIL;
5064 }
5065
5066 /* check against list of already visited files */
5067 for (vp = *visited_list; vp != NULL; vp = vp->ffv_next)
5068 {
5069 if (
5070#ifdef UNIX
Bram Moolenaare1fbddc2009-05-16 19:07:03 +00005071 !url ? (vp->ffv_dev_valid && vp->ffv_dev == st.st_dev
5072 && vp->ffv_ino == st.st_ino)
5073 :
Bram Moolenaar071d4272004-06-13 20:20:40 +00005074#endif
5075 fnamecmp(vp->ffv_fname, ff_expand_buffer) == 0
5076 )
5077 {
5078#ifdef FEAT_PATH_EXTRA
5079 /* are the wildcard parts equal */
5080 if (ff_wc_equal(vp->ffv_wc_path, wc_path) == TRUE)
5081#endif
5082 /* already visited */
5083 return FAIL;
5084 }
5085 }
5086
5087 /*
5088 * New file/dir. Add it to the list of visited files/dirs.
5089 */
5090 vp = (ff_visited_T *)alloc((unsigned)(sizeof(ff_visited_T)
5091 + STRLEN(ff_expand_buffer)));
5092
5093 if (vp != NULL)
5094 {
5095#ifdef UNIX
5096 if (!url)
5097 {
Bram Moolenaare1fbddc2009-05-16 19:07:03 +00005098 vp->ffv_dev_valid = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005099 vp->ffv_ino = st.st_ino;
5100 vp->ffv_dev = st.st_dev;
5101 vp->ffv_fname[0] = NUL;
5102 }
5103 else
5104 {
Bram Moolenaare1fbddc2009-05-16 19:07:03 +00005105 vp->ffv_dev_valid = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005106#endif
5107 STRCPY(vp->ffv_fname, ff_expand_buffer);
5108#ifdef UNIX
5109 }
5110#endif
5111#ifdef FEAT_PATH_EXTRA
5112 if (wc_path != NULL)
5113 vp->ffv_wc_path = vim_strsave(wc_path);
5114 else
5115 vp->ffv_wc_path = NULL;
5116#endif
5117
5118 vp->ffv_next = *visited_list;
5119 *visited_list = vp;
5120 }
5121
5122 return OK;
5123}
5124
5125/*
5126 * create stack element from given path pieces
5127 */
5128 static ff_stack_T *
5129ff_create_stack_element(fix_part,
5130#ifdef FEAT_PATH_EXTRA
5131 wc_part,
5132#endif
5133 level, star_star_empty)
5134 char_u *fix_part;
5135#ifdef FEAT_PATH_EXTRA
5136 char_u *wc_part;
5137#endif
5138 int level;
5139 int star_star_empty;
5140{
5141 ff_stack_T *new;
5142
5143 new = (ff_stack_T *)alloc((unsigned)sizeof(ff_stack_T));
5144 if (new == NULL)
5145 return NULL;
5146
5147 new->ffs_prev = NULL;
5148 new->ffs_filearray = NULL;
5149 new->ffs_filearray_size = 0;
5150 new->ffs_filearray_cur = 0;
5151 new->ffs_stage = 0;
5152 new->ffs_level = level;
5153 new->ffs_star_star_empty = star_star_empty;;
5154
5155 /* the following saves NULL pointer checks in vim_findfile */
5156 if (fix_part == NULL)
5157 fix_part = (char_u *)"";
5158 new->ffs_fix_path = vim_strsave(fix_part);
5159
5160#ifdef FEAT_PATH_EXTRA
5161 if (wc_part == NULL)
5162 wc_part = (char_u *)"";
5163 new->ffs_wc_path = vim_strsave(wc_part);
5164#endif
5165
5166 if (new->ffs_fix_path == NULL
5167#ifdef FEAT_PATH_EXTRA
5168 || new->ffs_wc_path == NULL
5169#endif
5170 )
5171 {
5172 ff_free_stack_element(new);
5173 new = NULL;
5174 }
5175
5176 return new;
5177}
5178
5179/*
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005180 * Push a dir on the directory stack.
Bram Moolenaar071d4272004-06-13 20:20:40 +00005181 */
5182 static void
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005183ff_push(search_ctx, stack_ptr)
5184 ff_search_ctx_T *search_ctx;
5185 ff_stack_T *stack_ptr;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005186{
5187 /* check for NULL pointer, not to return an error to the user, but
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005188 * to prevent a crash */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005189 if (stack_ptr != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005190 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005191 stack_ptr->ffs_prev = search_ctx->ffsc_stack_ptr;
5192 search_ctx->ffsc_stack_ptr = stack_ptr;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005193 }
5194}
5195
5196/*
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005197 * Pop a dir from the directory stack.
5198 * Returns NULL if stack is empty.
Bram Moolenaar071d4272004-06-13 20:20:40 +00005199 */
5200 static ff_stack_T *
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005201ff_pop(search_ctx)
5202 ff_search_ctx_T *search_ctx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005203{
5204 ff_stack_T *sptr;
5205
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005206 sptr = search_ctx->ffsc_stack_ptr;
5207 if (search_ctx->ffsc_stack_ptr != NULL)
5208 search_ctx->ffsc_stack_ptr = search_ctx->ffsc_stack_ptr->ffs_prev;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005209
5210 return sptr;
5211}
5212
5213/*
5214 * free the given stack element
5215 */
5216 static void
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005217ff_free_stack_element(stack_ptr)
5218 ff_stack_T *stack_ptr;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005219{
5220 /* vim_free handles possible NULL pointers */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005221 vim_free(stack_ptr->ffs_fix_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005222#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005223 vim_free(stack_ptr->ffs_wc_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005224#endif
5225
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005226 if (stack_ptr->ffs_filearray != NULL)
5227 FreeWild(stack_ptr->ffs_filearray_size, stack_ptr->ffs_filearray);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005228
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005229 vim_free(stack_ptr);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005230}
5231
5232/*
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005233 * Clear the search context, but NOT the visited list.
Bram Moolenaar071d4272004-06-13 20:20:40 +00005234 */
5235 static void
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005236ff_clear(search_ctx)
5237 ff_search_ctx_T *search_ctx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005238{
5239 ff_stack_T *sptr;
5240
5241 /* clear up stack */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005242 while ((sptr = ff_pop(search_ctx)) != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005243 ff_free_stack_element(sptr);
5244
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005245 vim_free(search_ctx->ffsc_file_to_search);
5246 vim_free(search_ctx->ffsc_start_dir);
5247 vim_free(search_ctx->ffsc_fix_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005248#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005249 vim_free(search_ctx->ffsc_wc_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005250#endif
5251
5252#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005253 if (search_ctx->ffsc_stopdirs_v != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005254 {
5255 int i = 0;
5256
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005257 while (search_ctx->ffsc_stopdirs_v[i] != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005258 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005259 vim_free(search_ctx->ffsc_stopdirs_v[i]);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005260 i++;
5261 }
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005262 vim_free(search_ctx->ffsc_stopdirs_v);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005263 }
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005264 search_ctx->ffsc_stopdirs_v = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005265#endif
5266
5267 /* reset everything */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005268 search_ctx->ffsc_file_to_search = NULL;
5269 search_ctx->ffsc_start_dir = NULL;
5270 search_ctx->ffsc_fix_path = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005271#ifdef FEAT_PATH_EXTRA
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005272 search_ctx->ffsc_wc_path = NULL;
5273 search_ctx->ffsc_level = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005274#endif
5275}
5276
5277#ifdef FEAT_PATH_EXTRA
5278/*
5279 * check if the given path is in the stopdirs
5280 * returns TRUE if yes else FALSE
5281 */
5282 static int
5283ff_path_in_stoplist(path, path_len, stopdirs_v)
5284 char_u *path;
5285 int path_len;
5286 char_u **stopdirs_v;
5287{
5288 int i = 0;
5289
5290 /* eat up trailing path separators, except the first */
Bram Moolenaar1d94f9b2005-08-04 21:29:45 +00005291 while (path_len > 1 && vim_ispathsep(path[path_len - 1]))
Bram Moolenaar071d4272004-06-13 20:20:40 +00005292 path_len--;
5293
5294 /* if no path consider it as match */
5295 if (path_len == 0)
5296 return TRUE;
5297
5298 for (i = 0; stopdirs_v[i] != NULL; i++)
5299 {
5300 if ((int)STRLEN(stopdirs_v[i]) > path_len)
5301 {
5302 /* match for parent directory. So '/home' also matches
5303 * '/home/rks'. Check for PATHSEP in stopdirs_v[i], else
5304 * '/home/r' would also match '/home/rks'
5305 */
5306 if (fnamencmp(stopdirs_v[i], path, path_len) == 0
Bram Moolenaar1d94f9b2005-08-04 21:29:45 +00005307 && vim_ispathsep(stopdirs_v[i][path_len]))
Bram Moolenaar071d4272004-06-13 20:20:40 +00005308 return TRUE;
5309 }
5310 else
5311 {
5312 if (fnamecmp(stopdirs_v[i], path) == 0)
5313 return TRUE;
5314 }
5315 }
5316 return FALSE;
5317}
5318#endif
5319
5320#if defined(FEAT_SEARCHPATH) || defined(PROTO)
5321/*
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005322 * Find the file name "ptr[len]" in the path. Also finds directory names.
Bram Moolenaar071d4272004-06-13 20:20:40 +00005323 *
5324 * On the first call set the parameter 'first' to TRUE to initialize
5325 * the search. For repeating calls to FALSE.
5326 *
5327 * Repeating calls will return other files called 'ptr[len]' from the path.
5328 *
5329 * Only on the first call 'ptr' and 'len' are used. For repeating calls they
5330 * don't need valid values.
5331 *
5332 * If nothing found on the first call the option FNAME_MESS will issue the
5333 * message:
5334 * 'Can't find file "<file>" in path'
5335 * On repeating calls:
5336 * 'No more file "<file>" found in path'
5337 *
5338 * options:
5339 * FNAME_MESS give error message when not found
5340 *
5341 * Uses NameBuff[]!
5342 *
5343 * Returns an allocated string for the file name. NULL for error.
5344 *
5345 */
5346 char_u *
5347find_file_in_path(ptr, len, options, first, rel_fname)
5348 char_u *ptr; /* file name */
5349 int len; /* length of file name */
5350 int options;
5351 int first; /* use count'th matching file name */
5352 char_u *rel_fname; /* file name searching relative to */
5353{
5354 return find_file_in_path_option(ptr, len, options, first,
5355 *curbuf->b_p_path == NUL ? p_path : curbuf->b_p_path,
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005356 FINDFILE_BOTH, rel_fname, curbuf->b_p_sua);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005357}
5358
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005359static char_u *ff_file_to_find = NULL;
5360static void *fdip_search_ctx = NULL;
5361
5362#if defined(EXITFREE)
5363 static void
5364free_findfile()
5365{
5366 vim_free(ff_file_to_find);
5367 vim_findfile_cleanup(fdip_search_ctx);
5368}
5369#endif
5370
Bram Moolenaar071d4272004-06-13 20:20:40 +00005371/*
5372 * Find the directory name "ptr[len]" in the path.
5373 *
5374 * options:
5375 * FNAME_MESS give error message when not found
5376 *
5377 * Uses NameBuff[]!
5378 *
5379 * Returns an allocated string for the file name. NULL for error.
5380 */
5381 char_u *
5382find_directory_in_path(ptr, len, options, rel_fname)
5383 char_u *ptr; /* file name */
5384 int len; /* length of file name */
5385 int options;
5386 char_u *rel_fname; /* file name searching relative to */
5387{
5388 return find_file_in_path_option(ptr, len, options, TRUE, p_cdpath,
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005389 FINDFILE_DIR, rel_fname, (char_u *)"");
Bram Moolenaar071d4272004-06-13 20:20:40 +00005390}
5391
Bram Moolenaar89cb5e02004-07-19 20:55:54 +00005392 char_u *
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005393find_file_in_path_option(ptr, len, options, first, path_option, find_what, rel_fname, suffixes)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005394 char_u *ptr; /* file name */
5395 int len; /* length of file name */
5396 int options;
5397 int first; /* use count'th matching file name */
5398 char_u *path_option; /* p_path or p_cdpath */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005399 int find_what; /* FINDFILE_FILE, _DIR or _BOTH */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005400 char_u *rel_fname; /* file name we are looking relative to. */
Bram Moolenaar433f7c82006-03-21 21:29:36 +00005401 char_u *suffixes; /* list of suffixes, 'suffixesadd' option */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005402{
Bram Moolenaar071d4272004-06-13 20:20:40 +00005403 static char_u *dir;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005404 static int did_findfile_init = FALSE;
5405 char_u save_char;
5406 char_u *file_name = NULL;
5407 char_u *buf = NULL;
5408 int rel_to_curdir;
5409#ifdef AMIGA
5410 struct Process *proc = (struct Process *)FindTask(0L);
5411 APTR save_winptr = proc->pr_WindowPtr;
5412
5413 /* Avoid a requester here for a volume that doesn't exist. */
5414 proc->pr_WindowPtr = (APTR)-1L;
5415#endif
5416
5417 if (first == TRUE)
5418 {
5419 /* copy file name into NameBuff, expanding environment variables */
5420 save_char = ptr[len];
5421 ptr[len] = NUL;
5422 expand_env(ptr, NameBuff, MAXPATHL);
5423 ptr[len] = save_char;
5424
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005425 vim_free(ff_file_to_find);
5426 ff_file_to_find = vim_strsave(NameBuff);
5427 if (ff_file_to_find == NULL) /* out of memory */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005428 {
5429 file_name = NULL;
5430 goto theend;
5431 }
5432 }
5433
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005434 rel_to_curdir = (ff_file_to_find[0] == '.'
5435 && (ff_file_to_find[1] == NUL
5436 || vim_ispathsep(ff_file_to_find[1])
5437 || (ff_file_to_find[1] == '.'
5438 && (ff_file_to_find[2] == NUL
5439 || vim_ispathsep(ff_file_to_find[2])))));
5440 if (vim_isAbsName(ff_file_to_find)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005441 /* "..", "../path", "." and "./path": don't use the path_option */
5442 || rel_to_curdir
5443#if defined(MSWIN) || defined(MSDOS) || defined(OS2)
5444 /* handle "\tmp" as absolute path */
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005445 || vim_ispathsep(ff_file_to_find[0])
Bram Moolenaar21160b92009-11-11 15:56:10 +00005446 /* handle "c:name" as absolute path */
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005447 || (ff_file_to_find[0] != NUL && ff_file_to_find[1] == ':')
Bram Moolenaar071d4272004-06-13 20:20:40 +00005448#endif
5449#ifdef AMIGA
5450 /* handle ":tmp" as absolute path */
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005451 || ff_file_to_find[0] == ':'
Bram Moolenaar071d4272004-06-13 20:20:40 +00005452#endif
5453 )
5454 {
5455 /*
5456 * Absolute path, no need to use "path_option".
5457 * If this is not a first call, return NULL. We already returned a
5458 * filename on the first call.
5459 */
5460 if (first == TRUE)
5461 {
5462 int l;
5463 int run;
5464
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005465 if (path_with_url(ff_file_to_find))
Bram Moolenaar071d4272004-06-13 20:20:40 +00005466 {
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005467 file_name = vim_strsave(ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005468 goto theend;
5469 }
5470
5471 /* When FNAME_REL flag given first use the directory of the file.
5472 * Otherwise or when this fails use the current directory. */
5473 for (run = 1; run <= 2; ++run)
5474 {
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005475 l = (int)STRLEN(ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005476 if (run == 1
5477 && rel_to_curdir
5478 && (options & FNAME_REL)
5479 && rel_fname != NULL
5480 && STRLEN(rel_fname) + l < MAXPATHL)
5481 {
5482 STRCPY(NameBuff, rel_fname);
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005483 STRCPY(gettail(NameBuff), ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005484 l = (int)STRLEN(NameBuff);
5485 }
5486 else
5487 {
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005488 STRCPY(NameBuff, ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005489 run = 2;
5490 }
5491
5492 /* When the file doesn't exist, try adding parts of
5493 * 'suffixesadd'. */
Bram Moolenaar433f7c82006-03-21 21:29:36 +00005494 buf = suffixes;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005495 for (;;)
5496 {
5497 if (
5498#ifdef DJGPP
5499 /* "C:" by itself will fail for mch_getperm(),
5500 * assume it's always valid. */
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005501 (find_what != FINDFILE_FILE && NameBuff[0] != NUL
Bram Moolenaar071d4272004-06-13 20:20:40 +00005502 && NameBuff[1] == ':'
5503 && NameBuff[2] == NUL) ||
5504#endif
5505 (mch_getperm(NameBuff) >= 0
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005506 && (find_what == FINDFILE_BOTH
5507 || ((find_what == FINDFILE_DIR)
5508 == mch_isdir(NameBuff)))))
Bram Moolenaar071d4272004-06-13 20:20:40 +00005509 {
5510 file_name = vim_strsave(NameBuff);
5511 goto theend;
5512 }
5513 if (*buf == NUL)
5514 break;
5515 copy_option_part(&buf, NameBuff + l, MAXPATHL - l, ",");
5516 }
5517 }
5518 }
5519 }
5520 else
5521 {
5522 /*
5523 * Loop over all paths in the 'path' or 'cdpath' option.
5524 * When "first" is set, first setup to the start of the option.
5525 * Otherwise continue to find the next match.
5526 */
5527 if (first == TRUE)
5528 {
5529 /* vim_findfile_free_visited can handle a possible NULL pointer */
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005530 vim_findfile_free_visited(fdip_search_ctx);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005531 dir = path_option;
5532 did_findfile_init = FALSE;
5533 }
5534
5535 for (;;)
5536 {
5537 if (did_findfile_init)
5538 {
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005539 file_name = vim_findfile(fdip_search_ctx);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005540 if (file_name != NULL)
5541 break;
5542
5543 did_findfile_init = FALSE;
5544 }
5545 else
5546 {
5547 char_u *r_ptr;
5548
5549 if (dir == NULL || *dir == NUL)
5550 {
5551 /* We searched all paths of the option, now we can
5552 * free the search context. */
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005553 vim_findfile_cleanup(fdip_search_ctx);
5554 fdip_search_ctx = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005555 break;
5556 }
5557
5558 if ((buf = alloc((int)(MAXPATHL))) == NULL)
5559 break;
5560
5561 /* copy next path */
5562 buf[0] = 0;
5563 copy_option_part(&dir, buf, MAXPATHL, " ,");
5564
5565#ifdef FEAT_PATH_EXTRA
5566 /* get the stopdir string */
5567 r_ptr = vim_findfile_stopdir(buf);
5568#else
5569 r_ptr = NULL;
5570#endif
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005571 fdip_search_ctx = vim_findfile_init(buf, ff_file_to_find,
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005572 r_ptr, 100, FALSE, find_what,
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005573 fdip_search_ctx, FALSE, rel_fname);
5574 if (fdip_search_ctx != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005575 did_findfile_init = TRUE;
5576 vim_free(buf);
5577 }
5578 }
5579 }
5580 if (file_name == NULL && (options & FNAME_MESS))
5581 {
5582 if (first == TRUE)
5583 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005584 if (find_what == FINDFILE_DIR)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005585 EMSG2(_("E344: Can't find directory \"%s\" in cdpath"),
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005586 ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005587 else
5588 EMSG2(_("E345: Can't find file \"%s\" in path"),
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005589 ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005590 }
5591 else
5592 {
Bram Moolenaar4d0ec162008-02-20 11:24:52 +00005593 if (find_what == FINDFILE_DIR)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005594 EMSG2(_("E346: No more directory \"%s\" found in cdpath"),
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005595 ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005596 else
5597 EMSG2(_("E347: No more file \"%s\" found in path"),
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005598 ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005599 }
5600 }
5601
5602theend:
5603#ifdef AMIGA
5604 proc->pr_WindowPtr = save_winptr;
5605#endif
5606 return file_name;
5607}
5608
5609#endif /* FEAT_SEARCHPATH */
5610
5611/*
5612 * Change directory to "new_dir". If FEAT_SEARCHPATH is defined, search
5613 * 'cdpath' for relative directory names, otherwise just mch_chdir().
5614 */
5615 int
5616vim_chdir(new_dir)
5617 char_u *new_dir;
5618{
5619#ifndef FEAT_SEARCHPATH
5620 return mch_chdir((char *)new_dir);
5621#else
5622 char_u *dir_name;
5623 int r;
5624
5625 dir_name = find_directory_in_path(new_dir, (int)STRLEN(new_dir),
5626 FNAME_MESS, curbuf->b_ffname);
5627 if (dir_name == NULL)
5628 return -1;
5629 r = mch_chdir((char *)dir_name);
5630 vim_free(dir_name);
5631 return r;
5632#endif
5633}
5634
5635/*
Bram Moolenaarbbebc852005-07-18 21:47:53 +00005636 * Get user name from machine-specific function.
Bram Moolenaar071d4272004-06-13 20:20:40 +00005637 * Returns the user name in "buf[len]".
Bram Moolenaarbbebc852005-07-18 21:47:53 +00005638 * Some systems are quite slow in obtaining the user name (Windows NT), thus
5639 * cache the result.
Bram Moolenaar071d4272004-06-13 20:20:40 +00005640 * Returns OK or FAIL.
5641 */
5642 int
5643get_user_name(buf, len)
5644 char_u *buf;
5645 int len;
5646{
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005647 if (username == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005648 {
5649 if (mch_get_user_name(buf, len) == FAIL)
5650 return FAIL;
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005651 username = vim_strsave(buf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005652 }
5653 else
Bram Moolenaarbbebc852005-07-18 21:47:53 +00005654 vim_strncpy(buf, username, len - 1);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005655 return OK;
5656}
5657
5658#ifndef HAVE_QSORT
5659/*
5660 * Our own qsort(), for systems that don't have it.
5661 * It's simple and slow. From the K&R C book.
5662 */
5663 void
5664qsort(base, elm_count, elm_size, cmp)
5665 void *base;
5666 size_t elm_count;
5667 size_t elm_size;
5668 int (*cmp) __ARGS((const void *, const void *));
5669{
5670 char_u *buf;
5671 char_u *p1;
5672 char_u *p2;
5673 int i, j;
5674 int gap;
5675
5676 buf = alloc((unsigned)elm_size);
5677 if (buf == NULL)
5678 return;
5679
5680 for (gap = elm_count / 2; gap > 0; gap /= 2)
5681 for (i = gap; i < elm_count; ++i)
5682 for (j = i - gap; j >= 0; j -= gap)
5683 {
5684 /* Compare the elements. */
5685 p1 = (char_u *)base + j * elm_size;
5686 p2 = (char_u *)base + (j + gap) * elm_size;
5687 if ((*cmp)((void *)p1, (void *)p2) <= 0)
5688 break;
Bram Moolenaar21160b92009-11-11 15:56:10 +00005689 /* Exchange the elements. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005690 mch_memmove(buf, p1, elm_size);
5691 mch_memmove(p1, p2, elm_size);
5692 mch_memmove(p2, buf, elm_size);
5693 }
5694
5695 vim_free(buf);
5696}
5697#endif
5698
Bram Moolenaar071d4272004-06-13 20:20:40 +00005699/*
5700 * Sort an array of strings.
5701 */
5702static int
5703#ifdef __BORLANDC__
5704_RTLENTRYF
5705#endif
5706sort_compare __ARGS((const void *s1, const void *s2));
5707
5708 static int
5709#ifdef __BORLANDC__
5710_RTLENTRYF
5711#endif
5712sort_compare(s1, s2)
5713 const void *s1;
5714 const void *s2;
5715{
5716 return STRCMP(*(char **)s1, *(char **)s2);
5717}
5718
5719 void
5720sort_strings(files, count)
5721 char_u **files;
5722 int count;
5723{
5724 qsort((void *)files, (size_t)count, sizeof(char_u *), sort_compare);
5725}
Bram Moolenaar071d4272004-06-13 20:20:40 +00005726
5727#if !defined(NO_EXPANDPATH) || defined(PROTO)
5728/*
5729 * Compare path "p[]" to "q[]".
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00005730 * If "maxlen" >= 0 compare "p[maxlen]" to "q[maxlen]"
Bram Moolenaar071d4272004-06-13 20:20:40 +00005731 * Return value like strcmp(p, q), but consider path separators.
5732 */
5733 int
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00005734pathcmp(p, q, maxlen)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005735 const char *p, *q;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00005736 int maxlen;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005737{
5738 int i;
Bram Moolenaar86b68352004-12-27 21:59:20 +00005739 const char *s = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005740
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00005741 for (i = 0; maxlen < 0 || i < maxlen; ++i)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005742 {
5743 /* End of "p": check if "q" also ends or just has a slash. */
5744 if (p[i] == NUL)
5745 {
5746 if (q[i] == NUL) /* full match */
5747 return 0;
5748 s = q;
5749 break;
5750 }
5751
5752 /* End of "q": check if "p" just has a slash. */
5753 if (q[i] == NUL)
5754 {
5755 s = p;
5756 break;
5757 }
5758
5759 if (
5760#ifdef CASE_INSENSITIVE_FILENAME
5761 TOUPPER_LOC(p[i]) != TOUPPER_LOC(q[i])
5762#else
5763 p[i] != q[i]
5764#endif
5765#ifdef BACKSLASH_IN_FILENAME
5766 /* consider '/' and '\\' to be equal */
5767 && !((p[i] == '/' && q[i] == '\\')
5768 || (p[i] == '\\' && q[i] == '/'))
5769#endif
5770 )
5771 {
5772 if (vim_ispathsep(p[i]))
5773 return -1;
5774 if (vim_ispathsep(q[i]))
5775 return 1;
5776 return ((char_u *)p)[i] - ((char_u *)q)[i]; /* no match */
5777 }
5778 }
Bram Moolenaar86b68352004-12-27 21:59:20 +00005779 if (s == NULL) /* "i" ran into "maxlen" */
5780 return 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005781
5782 /* ignore a trailing slash, but not "//" or ":/" */
Bram Moolenaar86b68352004-12-27 21:59:20 +00005783 if (s[i + 1] == NUL
5784 && i > 0
5785 && !after_pathsep((char_u *)s, (char_u *)s + i)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005786#ifdef BACKSLASH_IN_FILENAME
Bram Moolenaar86b68352004-12-27 21:59:20 +00005787 && (s[i] == '/' || s[i] == '\\')
Bram Moolenaar071d4272004-06-13 20:20:40 +00005788#else
Bram Moolenaar86b68352004-12-27 21:59:20 +00005789 && s[i] == '/'
Bram Moolenaar071d4272004-06-13 20:20:40 +00005790#endif
Bram Moolenaar86b68352004-12-27 21:59:20 +00005791 )
Bram Moolenaar071d4272004-06-13 20:20:40 +00005792 return 0; /* match with trailing slash */
5793 if (s == q)
5794 return -1; /* no match */
5795 return 1;
5796}
5797#endif
5798
Bram Moolenaar071d4272004-06-13 20:20:40 +00005799/*
5800 * The putenv() implementation below comes from the "screen" program.
5801 * Included with permission from Juergen Weigert.
5802 * See pty.c for the copyright notice.
5803 */
5804
5805/*
5806 * putenv -- put value into environment
5807 *
5808 * Usage: i = putenv (string)
5809 * int i;
5810 * char *string;
5811 *
5812 * where string is of the form <name>=<value>.
5813 * Putenv returns 0 normally, -1 on error (not enough core for malloc).
5814 *
5815 * Putenv may need to add a new name into the environment, or to
5816 * associate a value longer than the current value with a particular
5817 * name. So, to make life simpler, putenv() copies your entire
5818 * environment into the heap (i.e. malloc()) from the stack
5819 * (i.e. where it resides when your process is initiated) the first
5820 * time you call it.
5821 *
5822 * (history removed, not very interesting. See the "screen" sources.)
5823 */
5824
5825#if !defined(HAVE_SETENV) && !defined(HAVE_PUTENV)
5826
5827#define EXTRASIZE 5 /* increment to add to env. size */
5828
5829static int envsize = -1; /* current size of environment */
5830#ifndef MACOS_CLASSIC
5831extern
5832#endif
5833 char **environ; /* the global which is your env. */
5834
5835static int findenv __ARGS((char *name)); /* look for a name in the env. */
5836static int newenv __ARGS((void)); /* copy env. from stack to heap */
5837static int moreenv __ARGS((void)); /* incr. size of env. */
5838
5839 int
5840putenv(string)
5841 const char *string;
5842{
5843 int i;
5844 char *p;
5845
5846 if (envsize < 0)
5847 { /* first time putenv called */
5848 if (newenv() < 0) /* copy env. to heap */
5849 return -1;
5850 }
5851
5852 i = findenv((char *)string); /* look for name in environment */
5853
5854 if (i < 0)
5855 { /* name must be added */
5856 for (i = 0; environ[i]; i++);
5857 if (i >= (envsize - 1))
5858 { /* need new slot */
5859 if (moreenv() < 0)
5860 return -1;
5861 }
5862 p = (char *)alloc((unsigned)(strlen(string) + 1));
5863 if (p == NULL) /* not enough core */
5864 return -1;
5865 environ[i + 1] = 0; /* new end of env. */
5866 }
5867 else
5868 { /* name already in env. */
5869 p = vim_realloc(environ[i], strlen(string) + 1);
5870 if (p == NULL)
5871 return -1;
5872 }
5873 sprintf(p, "%s", string); /* copy into env. */
5874 environ[i] = p;
5875
5876 return 0;
5877}
5878
5879 static int
5880findenv(name)
5881 char *name;
5882{
5883 char *namechar, *envchar;
5884 int i, found;
5885
5886 found = 0;
5887 for (i = 0; environ[i] && !found; i++)
5888 {
5889 envchar = environ[i];
5890 namechar = name;
5891 while (*namechar && *namechar != '=' && (*namechar == *envchar))
5892 {
5893 namechar++;
5894 envchar++;
5895 }
5896 found = ((*namechar == '\0' || *namechar == '=') && *envchar == '=');
5897 }
5898 return found ? i - 1 : -1;
5899}
5900
5901 static int
5902newenv()
5903{
5904 char **env, *elem;
5905 int i, esize;
5906
5907#ifdef MACOS
5908 /* for Mac a new, empty environment is created */
5909 i = 0;
5910#else
5911 for (i = 0; environ[i]; i++)
5912 ;
5913#endif
5914 esize = i + EXTRASIZE + 1;
5915 env = (char **)alloc((unsigned)(esize * sizeof (elem)));
5916 if (env == NULL)
5917 return -1;
5918
5919#ifndef MACOS
5920 for (i = 0; environ[i]; i++)
5921 {
5922 elem = (char *)alloc((unsigned)(strlen(environ[i]) + 1));
5923 if (elem == NULL)
5924 return -1;
5925 env[i] = elem;
5926 strcpy(elem, environ[i]);
5927 }
5928#endif
5929
5930 env[i] = 0;
5931 environ = env;
5932 envsize = esize;
5933 return 0;
5934}
5935
5936 static int
5937moreenv()
5938{
5939 int esize;
5940 char **env;
5941
5942 esize = envsize + EXTRASIZE;
5943 env = (char **)vim_realloc((char *)environ, esize * sizeof (*env));
5944 if (env == 0)
5945 return -1;
5946 environ = env;
5947 envsize = esize;
5948 return 0;
5949}
5950
5951# ifdef USE_VIMPTY_GETENV
5952 char_u *
5953vimpty_getenv(string)
5954 const char_u *string;
5955{
5956 int i;
5957 char_u *p;
5958
5959 if (envsize < 0)
5960 return NULL;
5961
5962 i = findenv((char *)string);
5963
5964 if (i < 0)
5965 return NULL;
5966
5967 p = vim_strchr((char_u *)environ[i], '=');
5968 return (p + 1);
5969}
5970# endif
5971
5972#endif /* !defined(HAVE_SETENV) && !defined(HAVE_PUTENV) */
Bram Moolenaarc4a06d32005-06-07 21:04:49 +00005973
Bram Moolenaarc4956c82006-03-12 21:58:43 +00005974#if defined(FEAT_EVAL) || defined(FEAT_SPELL) || defined(PROTO)
Bram Moolenaarc4a06d32005-06-07 21:04:49 +00005975/*
5976 * Return 0 for not writable, 1 for writable file, 2 for a dir which we have
5977 * rights to write into.
5978 */
5979 int
5980filewritable(fname)
5981 char_u *fname;
5982{
5983 int retval = 0;
5984#if defined(UNIX) || defined(VMS)
5985 int perm = 0;
5986#endif
5987
5988#if defined(UNIX) || defined(VMS)
5989 perm = mch_getperm(fname);
5990#endif
5991#ifndef MACOS_CLASSIC /* TODO: get either mch_writable or mch_access */
5992 if (
5993# ifdef WIN3264
5994 mch_writable(fname) &&
5995# else
5996# if defined(UNIX) || defined(VMS)
5997 (perm & 0222) &&
5998# endif
5999# endif
6000 mch_access((char *)fname, W_OK) == 0
6001 )
6002#endif
6003 {
6004 ++retval;
6005 if (mch_isdir(fname))
6006 ++retval;
6007 }
6008 return retval;
6009}
6010#endif
Bram Moolenaar6bab4d12005-06-16 21:53:56 +00006011
6012/*
6013 * Print an error message with one or two "%s" and one or two string arguments.
6014 * This is not in message.c to avoid a warning for prototypes.
6015 */
6016 int
6017emsg3(s, a1, a2)
6018 char_u *s, *a1, *a2;
6019{
Bram Moolenaareb3593b2006-04-22 22:33:57 +00006020 if (emsg_not_now())
Bram Moolenaar6bab4d12005-06-16 21:53:56 +00006021 return TRUE; /* no error messages at the moment */
Bram Moolenaar6f192452007-11-08 19:49:02 +00006022#ifdef HAVE_STDARG_H
6023 vim_snprintf((char *)IObuff, IOSIZE, (char *)s, a1, a2);
6024#else
Bram Moolenaareb3593b2006-04-22 22:33:57 +00006025 vim_snprintf((char *)IObuff, IOSIZE, (char *)s, (long_u)a1, (long_u)a2);
Bram Moolenaar6f192452007-11-08 19:49:02 +00006026#endif
Bram Moolenaar6bab4d12005-06-16 21:53:56 +00006027 return emsg(IObuff);
6028}
6029
6030/*
6031 * Print an error message with one "%ld" and one long int argument.
6032 * This is not in message.c to avoid a warning for prototypes.
6033 */
6034 int
6035emsgn(s, n)
6036 char_u *s;
6037 long n;
6038{
Bram Moolenaareb3593b2006-04-22 22:33:57 +00006039 if (emsg_not_now())
Bram Moolenaar6bab4d12005-06-16 21:53:56 +00006040 return TRUE; /* no error messages at the moment */
6041 vim_snprintf((char *)IObuff, IOSIZE, (char *)s, n);
6042 return emsg(IObuff);
6043}