blob: 7cef2d38aff1dc5b20f73d7cf7897aff6e830a2e [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
15#ifdef HAVE_FCNTL_H
16# include <fcntl.h> /* for chdir() */
17#endif
18
Bram Moolenaarf461c8e2005-06-25 23:04:51 +000019static char_u *username = NULL; /* cached result of mch_get_user_name() */
20
21static char_u *ff_expand_buffer = NULL; /* used for expanding filenames */
22
Bram Moolenaar071d4272004-06-13 20:20:40 +000023#if defined(FEAT_VIRTUALEDIT) || defined(PROTO)
24static int coladvance2 __ARGS((pos_T *pos, int addspaces, int finetune, colnr_T wcol));
25
26/*
27 * Return TRUE if in the current mode we need to use virtual.
28 */
29 int
30virtual_active()
31{
32 /* While an operator is being executed we return "virtual_op", because
33 * VIsual_active has already been reset, thus we can't check for "block"
34 * being used. */
35 if (virtual_op != MAYBE)
36 return virtual_op;
37 return (ve_flags == VE_ALL
38# ifdef FEAT_VISUAL
39 || ((ve_flags & VE_BLOCK) && VIsual_active && VIsual_mode == Ctrl_V)
40# endif
41 || ((ve_flags & VE_INSERT) && (State & INSERT)));
42}
43
44/*
45 * Get the screen position of the cursor.
46 */
47 int
48getviscol()
49{
50 colnr_T x;
51
52 getvvcol(curwin, &curwin->w_cursor, &x, NULL, NULL);
53 return (int)x;
54}
55
56/*
57 * Get the screen position of character col with a coladd in the cursor line.
58 */
59 int
60getviscol2(col, coladd)
61 colnr_T col;
62 colnr_T coladd;
63{
64 colnr_T x;
65 pos_T pos;
66
67 pos.lnum = curwin->w_cursor.lnum;
68 pos.col = col;
69 pos.coladd = coladd;
70 getvvcol(curwin, &pos, &x, NULL, NULL);
71 return (int)x;
72}
73
74/*
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +000075 * Go to column "wcol", and add/insert white space as necessary to get the
Bram Moolenaar071d4272004-06-13 20:20:40 +000076 * cursor in that column.
77 * The caller must have saved the cursor line for undo!
78 */
79 int
80coladvance_force(wcol)
81 colnr_T wcol;
82{
83 int rc = coladvance2(&curwin->w_cursor, TRUE, FALSE, wcol);
84
85 if (wcol == MAXCOL)
86 curwin->w_valid &= ~VALID_VIRTCOL;
87 else
88 {
89 /* Virtcol is valid */
90 curwin->w_valid |= VALID_VIRTCOL;
91 curwin->w_virtcol = wcol;
92 }
93 return rc;
94}
95#endif
96
97/*
98 * Try to advance the Cursor to the specified screen column.
99 * If virtual editing: fine tune the cursor position.
100 * Note that all virtual positions off the end of a line should share
101 * a curwin->w_cursor.col value (n.b. this is equal to STRLEN(line)),
102 * beginning at coladd 0.
103 *
104 * return OK if desired column is reached, FAIL if not
105 */
106 int
107coladvance(wcol)
108 colnr_T wcol;
109{
110 int rc = getvpos(&curwin->w_cursor, wcol);
111
112 if (wcol == MAXCOL || rc == FAIL)
113 curwin->w_valid &= ~VALID_VIRTCOL;
Bram Moolenaardfccaf02004-12-31 20:56:11 +0000114 else if (*ml_get_cursor() != TAB)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000115 {
Bram Moolenaardfccaf02004-12-31 20:56:11 +0000116 /* Virtcol is valid when not on a TAB */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000117 curwin->w_valid |= VALID_VIRTCOL;
118 curwin->w_virtcol = wcol;
119 }
120 return rc;
121}
122
123/*
124 * Return in "pos" the position of the cursor advanced to screen column "wcol".
125 * return OK if desired column is reached, FAIL if not
126 */
127 int
128getvpos(pos, wcol)
129 pos_T *pos;
130 colnr_T wcol;
131{
132#ifdef FEAT_VIRTUALEDIT
133 return coladvance2(pos, FALSE, virtual_active(), wcol);
134}
135
136 static int
137coladvance2(pos, addspaces, finetune, wcol)
138 pos_T *pos;
139 int addspaces; /* change the text to achieve our goal? */
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +0000140 int finetune; /* change char offset for the exact column */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000141 colnr_T wcol; /* column to move to */
142{
143#endif
144 int idx;
145 char_u *ptr;
146 char_u *line;
147 colnr_T col = 0;
148 int csize = 0;
149 int one_more;
150#ifdef FEAT_LINEBREAK
151 int head = 0;
152#endif
153
Bram Moolenaarefd2bf12006-03-16 21:41:35 +0000154 one_more = (State & INSERT)
155 || restart_edit != NUL
Bram Moolenaar071d4272004-06-13 20:20:40 +0000156#ifdef FEAT_VISUAL
Bram Moolenaarefd2bf12006-03-16 21:41:35 +0000157 || (VIsual_active && *p_sel != 'o')
Bram Moolenaar071d4272004-06-13 20:20:40 +0000158#endif
Bram Moolenaarefd2bf12006-03-16 21:41:35 +0000159#ifdef FEAT_VIRTUALEDIT
Bram Moolenaar97b2ad32006-03-18 21:40:56 +0000160 || ((ve_flags & VE_ONEMORE) && wcol < MAXCOL)
Bram Moolenaarefd2bf12006-03-16 21:41:35 +0000161#endif
162 ;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000163 line = ml_get_curline();
164
165 if (wcol >= MAXCOL)
166 {
167 idx = (int)STRLEN(line) - 1 + one_more;
168 col = wcol;
169
170#ifdef FEAT_VIRTUALEDIT
171 if ((addspaces || finetune) && !VIsual_active)
172 {
173 curwin->w_curswant = linetabsize(line) + one_more;
174 if (curwin->w_curswant > 0)
175 --curwin->w_curswant;
176 }
177#endif
178 }
179 else
180 {
181#ifdef FEAT_VIRTUALEDIT
182 int width = W_WIDTH(curwin) - win_col_off(curwin);
183
Bram Moolenaarebefac62005-12-28 22:39:57 +0000184 if (finetune
Bram Moolenaar071d4272004-06-13 20:20:40 +0000185 && curwin->w_p_wrap
186# ifdef FEAT_VERTSPLIT
187 && curwin->w_width != 0
188# endif
189 && wcol >= (colnr_T)width)
190 {
191 csize = linetabsize(line);
192 if (csize > 0)
193 csize--;
194
Bram Moolenaarebefac62005-12-28 22:39:57 +0000195 if (wcol / width > (colnr_T)csize / width
196 && ((State & INSERT) == 0 || (int)wcol > csize + 1))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000197 {
198 /* In case of line wrapping don't move the cursor beyond the
Bram Moolenaarebefac62005-12-28 22:39:57 +0000199 * right screen edge. In Insert mode allow going just beyond
200 * the last character (like what happens when typing and
201 * reaching the right window edge). */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000202 wcol = (csize / width + 1) * width - 1;
203 }
204 }
205#endif
206
207 idx = -1;
208 ptr = line;
209 while (col <= wcol && *ptr != NUL)
210 {
211 /* Count a tab for what it's worth (if list mode not on) */
212#ifdef FEAT_LINEBREAK
213 csize = win_lbr_chartabsize(curwin, ptr, col, &head);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +0000214 mb_ptr_adv(ptr);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000215#else
216 csize = lbr_chartabsize_adv(&ptr, col);
217#endif
218 col += csize;
219 }
220 idx = (int)(ptr - line);
221 /*
222 * Handle all the special cases. The virtual_active() check
223 * is needed to ensure that a virtual position off the end of
224 * a line has the correct indexing. The one_more comparison
225 * replaces an explicit add of one_more later on.
226 */
227 if (col > wcol || (!virtual_active() && one_more == 0))
228 {
229 idx -= 1;
230# ifdef FEAT_LINEBREAK
231 /* Don't count the chars from 'showbreak'. */
232 csize -= head;
233# endif
234 col -= csize;
235 }
236
237#ifdef FEAT_VIRTUALEDIT
238 if (virtual_active()
239 && addspaces
240 && ((col != wcol && col != wcol + 1) || csize > 1))
241 {
242 /* 'virtualedit' is set: The difference between wcol and col is
243 * filled with spaces. */
244
245 if (line[idx] == NUL)
246 {
247 /* Append spaces */
248 int correct = wcol - col;
249 char_u *newline = alloc(idx + correct + 1);
250 int t;
251
252 if (newline == NULL)
253 return FAIL;
254
255 for (t = 0; t < idx; ++t)
256 newline[t] = line[t];
257
258 for (t = 0; t < correct; ++t)
259 newline[t + idx] = ' ';
260
261 newline[idx + correct] = NUL;
262
263 ml_replace(pos->lnum, newline, FALSE);
264 changed_bytes(pos->lnum, (colnr_T)idx);
265 idx += correct;
266 col = wcol;
267 }
268 else
269 {
270 /* Break a tab */
271 int linelen = (int)STRLEN(line);
272 int correct = wcol - col - csize + 1; /* negative!! */
Bram Moolenaareb3593b2006-04-22 22:33:57 +0000273 char_u *newline;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000274 int t, s = 0;
275 int v;
276
Bram Moolenaareb3593b2006-04-22 22:33:57 +0000277 if (-correct > csize)
278 return FAIL;
279
280 newline = alloc(linelen + csize);
281 if (newline == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000282 return FAIL;
283
284 for (t = 0; t < linelen; t++)
285 {
286 if (t != idx)
287 newline[s++] = line[t];
288 else
289 for (v = 0; v < csize; v++)
290 newline[s++] = ' ';
291 }
292
293 newline[linelen + csize - 1] = NUL;
294
295 ml_replace(pos->lnum, newline, FALSE);
296 changed_bytes(pos->lnum, idx);
297 idx += (csize - 1 + correct);
298 col += correct;
299 }
300 }
301#endif
302 }
303
304 if (idx < 0)
305 pos->col = 0;
306 else
307 pos->col = idx;
308
309#ifdef FEAT_VIRTUALEDIT
310 pos->coladd = 0;
311
312 if (finetune)
313 {
314 if (wcol == MAXCOL)
315 {
316 /* The width of the last character is used to set coladd. */
317 if (!one_more)
318 {
319 colnr_T scol, ecol;
320
321 getvcol(curwin, pos, &scol, NULL, &ecol);
322 pos->coladd = ecol - scol;
323 }
324 }
325 else
326 {
327 int b = (int)wcol - (int)col;
328
329 /* The difference between wcol and col is used to set coladd. */
330 if (b > 0 && b < (MAXCOL - 2 * W_WIDTH(curwin)))
331 pos->coladd = b;
332
333 col += b;
334 }
335 }
336#endif
337
338#ifdef FEAT_MBYTE
339 /* prevent cursor from moving on the trail byte */
340 if (has_mbyte)
341 mb_adjust_cursor();
342#endif
343
344 if (col < wcol)
345 return FAIL;
346 return OK;
347}
348
349/*
350 * inc(p)
351 *
352 * Increment the line pointer 'p' crossing line boundaries as necessary.
353 * Return 1 when going to the next line.
354 * Return 2 when moving forward onto a NUL at the end of the line).
355 * Return -1 when at the end of file.
356 * Return 0 otherwise.
357 */
358 int
359inc_cursor()
360{
361 return inc(&curwin->w_cursor);
362}
363
364 int
365inc(lp)
366 pos_T *lp;
367{
368 char_u *p = ml_get_pos(lp);
369
370 if (*p != NUL) /* still within line, move to next char (may be NUL) */
371 {
372#ifdef FEAT_MBYTE
373 if (has_mbyte)
374 {
Bram Moolenaar0fa313a2005-08-10 21:07:57 +0000375 int l = (*mb_ptr2len)(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000376
377 lp->col += l;
378 return ((p[l] != NUL) ? 0 : 2);
379 }
380#endif
381 lp->col++;
382#ifdef FEAT_VIRTUALEDIT
383 lp->coladd = 0;
384#endif
385 return ((p[1] != NUL) ? 0 : 2);
386 }
387 if (lp->lnum != curbuf->b_ml.ml_line_count) /* there is a next line */
388 {
389 lp->col = 0;
390 lp->lnum++;
391#ifdef FEAT_VIRTUALEDIT
392 lp->coladd = 0;
393#endif
394 return 1;
395 }
396 return -1;
397}
398
399/*
400 * incl(lp): same as inc(), but skip the NUL at the end of non-empty lines
401 */
402 int
403incl(lp)
404 pos_T *lp;
405{
406 int r;
407
408 if ((r = inc(lp)) >= 1 && lp->col)
409 r = inc(lp);
410 return r;
411}
412
413/*
414 * dec(p)
415 *
416 * Decrement the line pointer 'p' crossing line boundaries as necessary.
417 * Return 1 when crossing a line, -1 when at start of file, 0 otherwise.
418 */
419 int
420dec_cursor()
421{
422 return dec(&curwin->w_cursor);
423}
424
425 int
426dec(lp)
427 pos_T *lp;
428{
429 char_u *p;
430
431#ifdef FEAT_VIRTUALEDIT
432 lp->coladd = 0;
433#endif
434 if (lp->col > 0) /* still within line */
435 {
436 lp->col--;
437#ifdef FEAT_MBYTE
438 if (has_mbyte)
439 {
440 p = ml_get(lp->lnum);
441 lp->col -= (*mb_head_off)(p, p + lp->col);
442 }
443#endif
444 return 0;
445 }
446 if (lp->lnum > 1) /* there is a prior line */
447 {
448 lp->lnum--;
449 p = ml_get(lp->lnum);
450 lp->col = (colnr_T)STRLEN(p);
451#ifdef FEAT_MBYTE
452 if (has_mbyte)
453 lp->col -= (*mb_head_off)(p, p + lp->col);
454#endif
455 return 1;
456 }
457 return -1; /* at start of file */
458}
459
460/*
461 * decl(lp): same as dec(), but skip the NUL at the end of non-empty lines
462 */
463 int
464decl(lp)
465 pos_T *lp;
466{
467 int r;
468
469 if ((r = dec(lp)) == 1 && lp->col)
470 r = dec(lp);
471 return r;
472}
473
474/*
475 * Make sure curwin->w_cursor.lnum is valid.
476 */
477 void
478check_cursor_lnum()
479{
480 if (curwin->w_cursor.lnum > curbuf->b_ml.ml_line_count)
481 {
482#ifdef FEAT_FOLDING
483 /* If there is a closed fold at the end of the file, put the cursor in
484 * its first line. Otherwise in the last line. */
485 if (!hasFolding(curbuf->b_ml.ml_line_count,
486 &curwin->w_cursor.lnum, NULL))
487#endif
488 curwin->w_cursor.lnum = curbuf->b_ml.ml_line_count;
489 }
490 if (curwin->w_cursor.lnum <= 0)
491 curwin->w_cursor.lnum = 1;
492}
493
494/*
495 * Make sure curwin->w_cursor.col is valid.
496 */
497 void
498check_cursor_col()
499{
500 colnr_T len;
501#ifdef FEAT_VIRTUALEDIT
502 colnr_T oldcol = curwin->w_cursor.col + curwin->w_cursor.coladd;
503#endif
504
505 len = (colnr_T)STRLEN(ml_get_curline());
506 if (len == 0)
507 curwin->w_cursor.col = 0;
508 else if (curwin->w_cursor.col >= len)
509 {
Bram Moolenaar33f54432008-01-04 20:25:44 +0000510 /* Allow cursor past end-of-line when:
511 * - in Insert mode or restarting Insert mode
512 * - in Visual mode and 'selection' isn't "old"
513 * - 'virtualedit' is set */
Bram Moolenaarebefac62005-12-28 22:39:57 +0000514 if ((State & INSERT) || restart_edit
Bram Moolenaar071d4272004-06-13 20:20:40 +0000515#ifdef FEAT_VISUAL
516 || (VIsual_active && *p_sel != 'o')
517#endif
Bram Moolenaar33f54432008-01-04 20:25:44 +0000518#ifdef FEAT_VIRTUALEDIT
519 || (ve_flags & VE_ONEMORE)
520#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000521 || virtual_active())
522 curwin->w_cursor.col = len;
523 else
Bram Moolenaar87c19962007-04-26 08:54:21 +0000524 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000525 curwin->w_cursor.col = len - 1;
Bram Moolenaar87c19962007-04-26 08:54:21 +0000526#ifdef FEAT_MBYTE
527 /* prevent cursor from moving on the trail byte */
528 if (has_mbyte)
529 mb_adjust_cursor();
530#endif
531 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000532 }
533
534#ifdef FEAT_VIRTUALEDIT
535 /* If virtual editing is on, we can leave the cursor on the old position,
536 * only we must set it to virtual. But don't do it when at the end of the
537 * line. */
538 if (oldcol == MAXCOL)
539 curwin->w_cursor.coladd = 0;
540 else if (ve_flags == VE_ALL)
541 curwin->w_cursor.coladd = oldcol - curwin->w_cursor.col;
542#endif
543}
544
545/*
546 * make sure curwin->w_cursor in on a valid character
547 */
548 void
549check_cursor()
550{
551 check_cursor_lnum();
552 check_cursor_col();
553}
554
555#if defined(FEAT_TEXTOBJ) || defined(PROTO)
556/*
557 * Make sure curwin->w_cursor is not on the NUL at the end of the line.
558 * Allow it when in Visual mode and 'selection' is not "old".
559 */
560 void
561adjust_cursor_col()
562{
563 if (curwin->w_cursor.col > 0
564# ifdef FEAT_VISUAL
565 && (!VIsual_active || *p_sel == 'o')
566# endif
567 && gchar_cursor() == NUL)
568 --curwin->w_cursor.col;
569}
570#endif
571
572/*
573 * When curwin->w_leftcol has changed, adjust the cursor position.
574 * Return TRUE if the cursor was moved.
575 */
576 int
577leftcol_changed()
578{
579 long lastcol;
580 colnr_T s, e;
581 int retval = FALSE;
582
583 changed_cline_bef_curs();
584 lastcol = curwin->w_leftcol + W_WIDTH(curwin) - curwin_col_off() - 1;
585 validate_virtcol();
586
587 /*
588 * If the cursor is right or left of the screen, move it to last or first
589 * character.
590 */
591 if (curwin->w_virtcol > (colnr_T)(lastcol - p_siso))
592 {
593 retval = TRUE;
594 coladvance((colnr_T)(lastcol - p_siso));
595 }
596 else if (curwin->w_virtcol < curwin->w_leftcol + p_siso)
597 {
598 retval = TRUE;
599 (void)coladvance((colnr_T)(curwin->w_leftcol + p_siso));
600 }
601
602 /*
603 * If the start of the character under the cursor is not on the screen,
604 * advance the cursor one more char. If this fails (last char of the
605 * line) adjust the scrolling.
606 */
607 getvvcol(curwin, &curwin->w_cursor, &s, NULL, &e);
608 if (e > (colnr_T)lastcol)
609 {
610 retval = TRUE;
611 coladvance(s - 1);
612 }
613 else if (s < curwin->w_leftcol)
614 {
615 retval = TRUE;
616 if (coladvance(e + 1) == FAIL) /* there isn't another character */
617 {
618 curwin->w_leftcol = s; /* adjust w_leftcol instead */
619 changed_cline_bef_curs();
620 }
621 }
622
623 if (retval)
624 curwin->w_set_curswant = TRUE;
625 redraw_later(NOT_VALID);
626 return retval;
627}
628
629/**********************************************************************
630 * Various routines dealing with allocation and deallocation of memory.
631 */
632
633#if defined(MEM_PROFILE) || defined(PROTO)
634
635# define MEM_SIZES 8200
636static long_u mem_allocs[MEM_SIZES];
637static long_u mem_frees[MEM_SIZES];
638static long_u mem_allocated;
639static long_u mem_freed;
640static long_u mem_peak;
641static long_u num_alloc;
642static long_u num_freed;
643
644static void mem_pre_alloc_s __ARGS((size_t *sizep));
645static void mem_pre_alloc_l __ARGS((long_u *sizep));
646static void mem_post_alloc __ARGS((void **pp, size_t size));
647static void mem_pre_free __ARGS((void **pp));
648
649 static void
650mem_pre_alloc_s(sizep)
651 size_t *sizep;
652{
653 *sizep += sizeof(size_t);
654}
655
656 static void
657mem_pre_alloc_l(sizep)
658 long_u *sizep;
659{
660 *sizep += sizeof(size_t);
661}
662
663 static void
664mem_post_alloc(pp, size)
665 void **pp;
666 size_t size;
667{
668 if (*pp == NULL)
669 return;
670 size -= sizeof(size_t);
671 *(long_u *)*pp = size;
672 if (size <= MEM_SIZES-1)
673 mem_allocs[size-1]++;
674 else
675 mem_allocs[MEM_SIZES-1]++;
676 mem_allocated += size;
677 if (mem_allocated - mem_freed > mem_peak)
678 mem_peak = mem_allocated - mem_freed;
679 num_alloc++;
680 *pp = (void *)((char *)*pp + sizeof(size_t));
681}
682
683 static void
684mem_pre_free(pp)
685 void **pp;
686{
687 long_u size;
688
689 *pp = (void *)((char *)*pp - sizeof(size_t));
690 size = *(size_t *)*pp;
691 if (size <= MEM_SIZES-1)
692 mem_frees[size-1]++;
693 else
694 mem_frees[MEM_SIZES-1]++;
695 mem_freed += size;
696 num_freed++;
697}
698
699/*
700 * called on exit via atexit()
701 */
702 void
703vim_mem_profile_dump()
704{
705 int i, j;
706
707 printf("\r\n");
708 j = 0;
709 for (i = 0; i < MEM_SIZES - 1; i++)
710 {
711 if (mem_allocs[i] || mem_frees[i])
712 {
713 if (mem_frees[i] > mem_allocs[i])
714 printf("\r\n%s", _("ERROR: "));
715 printf("[%4d / %4lu-%-4lu] ", i + 1, mem_allocs[i], mem_frees[i]);
716 j++;
717 if (j > 3)
718 {
719 j = 0;
720 printf("\r\n");
721 }
722 }
723 }
724
725 i = MEM_SIZES - 1;
726 if (mem_allocs[i])
727 {
728 printf("\r\n");
729 if (mem_frees[i] > mem_allocs[i])
730 printf(_("ERROR: "));
731 printf("[>%d / %4lu-%-4lu]", i, mem_allocs[i], mem_frees[i]);
732 }
733
734 printf(_("\n[bytes] total alloc-freed %lu-%lu, in use %lu, peak use %lu\n"),
735 mem_allocated, mem_freed, mem_allocated - mem_freed, mem_peak);
736 printf(_("[calls] total re/malloc()'s %lu, total free()'s %lu\n\n"),
737 num_alloc, num_freed);
738}
739
740#endif /* MEM_PROFILE */
741
742/*
743 * Some memory is reserved for error messages and for being able to
744 * call mf_release_all(), which needs some memory for mf_trans_add().
745 */
746#if defined(MSDOS) && !defined(DJGPP)
747# define SMALL_MEM
748# define KEEP_ROOM 8192L
749#else
750# define KEEP_ROOM (2 * 8192L)
751#endif
752
753/*
754 * Note: if unsinged is 16 bits we can only allocate up to 64K with alloc().
755 * Use lalloc for larger blocks.
756 */
757 char_u *
758alloc(size)
759 unsigned size;
760{
761 return (lalloc((long_u)size, TRUE));
762}
763
764/*
765 * Allocate memory and set all bytes to zero.
766 */
767 char_u *
768alloc_clear(size)
769 unsigned size;
770{
771 char_u *p;
772
773 p = (lalloc((long_u)size, TRUE));
774 if (p != NULL)
775 (void)vim_memset(p, 0, (size_t)size);
776 return p;
777}
778
779/*
780 * alloc() with check for maximum line length
781 */
782 char_u *
783alloc_check(size)
784 unsigned size;
785{
786#if !defined(UNIX) && !defined(__EMX__)
787 if (sizeof(int) == 2 && size > 0x7fff)
788 {
789 /* Don't hide this message */
790 emsg_silent = 0;
791 EMSG(_("E340: Line is becoming too long"));
792 return NULL;
793 }
794#endif
795 return (lalloc((long_u)size, TRUE));
796}
797
798/*
799 * Allocate memory like lalloc() and set all bytes to zero.
800 */
801 char_u *
802lalloc_clear(size, message)
803 long_u size;
804 int message;
805{
806 char_u *p;
807
808 p = (lalloc(size, message));
809 if (p != NULL)
810 (void)vim_memset(p, 0, (size_t)size);
811 return p;
812}
813
814/*
815 * Low level memory allocation function.
816 * This is used often, KEEP IT FAST!
817 */
818 char_u *
819lalloc(size, message)
820 long_u size;
821 int message;
822{
823 char_u *p; /* pointer to new storage space */
824 static int releasing = FALSE; /* don't do mf_release_all() recursive */
825 int try_again;
826#if defined(HAVE_AVAIL_MEM) && !defined(SMALL_MEM)
827 static long_u allocated = 0; /* allocated since last avail check */
828#endif
829
830 /* Safety check for allocating zero bytes */
831 if (size == 0)
832 {
833 /* Don't hide this message */
834 emsg_silent = 0;
835 EMSGN(_("E341: Internal error: lalloc(%ld, )"), size);
836 return NULL;
837 }
838
839#ifdef MEM_PROFILE
840 mem_pre_alloc_l(&size);
841#endif
842
843#if defined(MSDOS) && !defined(DJGPP)
844 if (size >= 0xfff0) /* in MSDOS we can't deal with >64K blocks */
845 p = NULL;
846 else
847#endif
848
849 /*
850 * Loop when out of memory: Try to release some memfile blocks and
851 * if some blocks are released call malloc again.
852 */
853 for (;;)
854 {
855 /*
856 * Handle three kind of systems:
857 * 1. No check for available memory: Just return.
858 * 2. Slow check for available memory: call mch_avail_mem() after
859 * allocating KEEP_ROOM amount of memory.
860 * 3. Strict check for available memory: call mch_avail_mem()
861 */
862 if ((p = (char_u *)malloc((size_t)size)) != NULL)
863 {
864#ifndef HAVE_AVAIL_MEM
865 /* 1. No check for available memory: Just return. */
866 goto theend;
867#else
868# ifndef SMALL_MEM
869 /* 2. Slow check for available memory: call mch_avail_mem() after
870 * allocating (KEEP_ROOM / 2) amount of memory. */
871 allocated += size;
872 if (allocated < KEEP_ROOM / 2)
873 goto theend;
874 allocated = 0;
875# endif
876 /* 3. check for available memory: call mch_avail_mem() */
877 if (mch_avail_mem(TRUE) < KEEP_ROOM && !releasing)
878 {
879 vim_free((char *)p); /* System is low... no go! */
880 p = NULL;
881 }
882 else
883 goto theend;
884#endif
885 }
886 /*
887 * Remember that mf_release_all() is being called to avoid an endless
888 * loop, because mf_release_all() may call alloc() recursively.
889 */
890 if (releasing)
891 break;
892 releasing = TRUE;
Bram Moolenaar661b1822005-07-28 22:36:45 +0000893
894 clear_sb_text(); /* free any scrollback text */
895 try_again = mf_release_all(); /* release as many blocks as possible */
Bram Moolenaar39a58ca2005-06-27 22:42:44 +0000896#ifdef FEAT_EVAL
Bram Moolenaar661b1822005-07-28 22:36:45 +0000897 try_again |= garbage_collect(); /* cleanup recursive lists/dicts */
Bram Moolenaar39a58ca2005-06-27 22:42:44 +0000898#endif
Bram Moolenaar661b1822005-07-28 22:36:45 +0000899
Bram Moolenaar071d4272004-06-13 20:20:40 +0000900 releasing = FALSE;
901 if (!try_again)
902 break;
903 }
904
905 if (message && p == NULL)
906 do_outofmem_msg(size);
907
908theend:
909#ifdef MEM_PROFILE
910 mem_post_alloc((void **)&p, (size_t)size);
911#endif
912 return p;
913}
914
915#if defined(MEM_PROFILE) || defined(PROTO)
916/*
917 * realloc() with memory profiling.
918 */
919 void *
920mem_realloc(ptr, size)
921 void *ptr;
922 size_t size;
923{
924 void *p;
925
926 mem_pre_free(&ptr);
927 mem_pre_alloc_s(&size);
928
929 p = realloc(ptr, size);
930
931 mem_post_alloc(&p, size);
932
933 return p;
934}
935#endif
936
937/*
938* Avoid repeating the error message many times (they take 1 second each).
939* Did_outofmem_msg is reset when a character is read.
940*/
941 void
942do_outofmem_msg(size)
943 long_u size;
944{
945 if (!did_outofmem_msg)
946 {
947 /* Don't hide this message */
948 emsg_silent = 0;
949 EMSGN(_("E342: Out of memory! (allocating %lu bytes)"), size);
950 did_outofmem_msg = TRUE;
951 }
952}
953
Bram Moolenaar1ec484f2005-06-24 23:07:47 +0000954#if defined(EXITFREE) || defined(PROTO)
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000955
956# if defined(FEAT_SEARCHPATH)
957static void free_findfile __ARGS((void));
958# endif
959
Bram Moolenaar1ec484f2005-06-24 23:07:47 +0000960/*
961 * Free everything that we allocated.
962 * Can be used to detect memory leaks, e.g., with ccmalloc.
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000963 * NOTE: This is tricky! Things are freed that functions depend on. Don't be
964 * surprised if Vim crashes...
965 * Some things can't be freed, esp. things local to a library function.
Bram Moolenaar1ec484f2005-06-24 23:07:47 +0000966 */
967 void
968free_all_mem()
969{
970 buf_T *buf, *nextbuf;
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000971 static int entered = FALSE;
972
973 /* When we cause a crash here it is caught and Vim tries to exit cleanly.
974 * Don't try freeing everything again. */
975 if (entered)
976 return;
977 entered = TRUE;
Bram Moolenaar1ec484f2005-06-24 23:07:47 +0000978
Bram Moolenaare0ca7b22007-11-24 20:28:24 +0000979# ifdef FEAT_AUTOCMD
Bram Moolenaar78ab3312007-09-29 12:16:41 +0000980 block_autocmds(); /* don't want to trigger autocommands here */
Bram Moolenaare0ca7b22007-11-24 20:28:24 +0000981# endif
Bram Moolenaar1ec484f2005-06-24 23:07:47 +0000982
Bram Moolenaare0ca7b22007-11-24 20:28:24 +0000983# ifdef FEAT_WINDOWS
Bram Moolenaar997fb4b2006-02-17 21:53:23 +0000984 /* close all tabs and windows */
Bram Moolenaar7d47b6e2006-03-15 22:59:18 +0000985 if (first_tabpage->tp_next != NULL)
986 do_cmdline_cmd((char_u *)"tabonly!");
987 if (firstwin != lastwin)
988 do_cmdline_cmd((char_u *)"only!");
Bram Moolenaare0ca7b22007-11-24 20:28:24 +0000989# endif
Bram Moolenaar997fb4b2006-02-17 21:53:23 +0000990
Bram Moolenaarc4956c82006-03-12 21:58:43 +0000991# if defined(FEAT_SPELL)
Bram Moolenaar1ec484f2005-06-24 23:07:47 +0000992 /* Free all spell info. */
993 spell_free_all();
994# endif
995
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000996# if defined(FEAT_USR_CMDS)
Bram Moolenaar1ec484f2005-06-24 23:07:47 +0000997 /* Clear user commands (before deleting buffers). */
998 ex_comclear(NULL);
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000999# endif
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001000
1001# ifdef FEAT_MENU
1002 /* Clear menus. */
1003 do_cmdline_cmd((char_u *)"aunmenu *");
1004# endif
1005
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001006 /* Clear mappings, abbreviations, breakpoints. */
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001007 do_cmdline_cmd((char_u *)"mapclear");
1008 do_cmdline_cmd((char_u *)"mapclear!");
1009 do_cmdline_cmd((char_u *)"abclear");
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001010# if defined(FEAT_EVAL)
1011 do_cmdline_cmd((char_u *)"breakdel *");
1012# endif
Bram Moolenaar1e498f52005-06-26 22:29:44 +00001013# if defined(FEAT_PROFILE)
1014 do_cmdline_cmd((char_u *)"profdel *");
1015# endif
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001016
1017# ifdef FEAT_TITLE
1018 free_titles();
1019# endif
1020# if defined(FEAT_SEARCHPATH)
1021 free_findfile();
1022# endif
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001023
1024 /* Obviously named calls. */
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001025# if defined(FEAT_AUTOCMD)
1026 free_all_autocmds();
1027# endif
1028 clear_termcodes();
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001029 free_all_options();
1030 free_all_marks();
1031 alist_clear(&global_alist);
1032 free_homedir();
1033 free_search_patterns();
1034 free_old_sub();
1035 free_last_insert();
1036 free_prev_shellcmd();
1037 free_regexp_stuff();
1038 free_tag_stuff();
1039 free_cd_dir();
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001040# ifdef FEAT_EVAL
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001041 set_expr_line(NULL);
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001042# endif
1043# ifdef FEAT_DIFF
Bram Moolenaar997fb4b2006-02-17 21:53:23 +00001044 diff_clear(curtab);
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001045# endif
Bram Moolenaara40ceaf2006-01-13 22:35:40 +00001046 clear_sb_text(); /* free any scrollback text */
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001047
1048 /* Free some global vars. */
1049 vim_free(username);
Bram Moolenaaraf92ee82007-10-07 13:45:08 +00001050# ifdef FEAT_CLIPBOARD
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001051 vim_free(clip_exclude_prog);
Bram Moolenaaraf92ee82007-10-07 13:45:08 +00001052# endif
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001053 vim_free(last_cmdline);
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001054# ifdef FEAT_CMDHIST
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001055 vim_free(new_last_cmdline);
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001056# endif
Bram Moolenaar238a5642006-02-21 22:12:05 +00001057 set_keep_msg(NULL, 0);
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001058 vim_free(ff_expand_buffer);
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001059
1060 /* Clear cmdline history. */
1061 p_hi = 0;
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001062# ifdef FEAT_CMDHIST
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001063 init_history();
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001064# endif
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001065
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001066#ifdef FEAT_QUICKFIX
Bram Moolenaare0ca7b22007-11-24 20:28:24 +00001067 {
1068 win_T *win;
1069
1070 qf_free_all(NULL);
1071 /* Free all location lists */
1072 FOR_ALL_WINDOWS(win)
1073 qf_free_all(win);
1074 }
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001075#endif
1076
1077 /* Close all script inputs. */
1078 close_all_scripts();
1079
1080#if defined(FEAT_WINDOWS)
1081 /* Destroy all windows. Must come before freeing buffers. */
1082 win_free_all();
1083#endif
1084
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001085 /* Free all buffers. */
1086 for (buf = firstbuf; buf != NULL; )
1087 {
1088 nextbuf = buf->b_next;
1089 close_buffer(NULL, buf, DOBUF_WIPE);
1090 if (buf_valid(buf))
1091 buf = nextbuf; /* didn't work, try next one */
1092 else
1093 buf = firstbuf;
1094 }
1095
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001096#ifdef FEAT_ARABIC
1097 free_cmdline_buf();
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001098#endif
1099
1100 /* Clear registers. */
1101 clear_registers();
1102 ResetRedobuff();
1103 ResetRedobuff();
1104
Bram Moolenaar85c79d32007-02-20 01:59:20 +00001105#if defined(FEAT_CLIENTSERVER) && defined(FEAT_X11)
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001106 vim_free(serverDelayedStartName);
1107#endif
1108
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001109 /* highlight info */
1110 free_highlight();
1111
Bram Moolenaar1e498f52005-06-26 22:29:44 +00001112 reset_last_sourcing();
1113
Bram Moolenaar7d47b6e2006-03-15 22:59:18 +00001114#ifdef FEAT_WINDOWS
Bram Moolenaar06a89a52006-04-29 22:01:03 +00001115 free_tabpage(first_tabpage);
1116 first_tabpage = NULL;
Bram Moolenaar7d47b6e2006-03-15 22:59:18 +00001117#endif
1118
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001119# ifdef UNIX
1120 /* Machine-specific free. */
1121 mch_free_mem();
1122# endif
1123
1124 /* message history */
1125 for (;;)
1126 if (delete_first_msg() == FAIL)
1127 break;
1128
1129# ifdef FEAT_EVAL
1130 eval_clear();
1131# endif
1132
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001133 free_termoptions();
1134
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001135 /* screenlines (can't display anything now!) */
1136 free_screenlines();
1137
1138#if defined(USE_XSMP)
1139 xsmp_close();
1140#endif
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00001141#ifdef FEAT_GUI_GTK
1142 gui_mch_free_all();
1143#endif
1144 clear_hl_tables();
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00001145
1146 vim_free(IObuff);
1147 vim_free(NameBuff);
1148}
1149#endif
1150
Bram Moolenaar071d4272004-06-13 20:20:40 +00001151/*
1152 * copy a string into newly allocated memory
1153 */
1154 char_u *
1155vim_strsave(string)
1156 char_u *string;
1157{
1158 char_u *p;
1159 unsigned len;
1160
1161 len = (unsigned)STRLEN(string) + 1;
1162 p = alloc(len);
1163 if (p != NULL)
1164 mch_memmove(p, string, (size_t)len);
1165 return p;
1166}
1167
1168 char_u *
1169vim_strnsave(string, len)
1170 char_u *string;
1171 int len;
1172{
1173 char_u *p;
1174
1175 p = alloc((unsigned)(len + 1));
1176 if (p != NULL)
1177 {
1178 STRNCPY(p, string, len);
1179 p[len] = NUL;
1180 }
1181 return p;
1182}
1183
Bram Moolenaar071d4272004-06-13 20:20:40 +00001184/*
1185 * Same as vim_strsave(), but any characters found in esc_chars are preceded
1186 * by a backslash.
1187 */
1188 char_u *
1189vim_strsave_escaped(string, esc_chars)
1190 char_u *string;
1191 char_u *esc_chars;
1192{
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001193 return vim_strsave_escaped_ext(string, esc_chars, '\\', FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001194}
1195
1196/*
1197 * Same as vim_strsave_escaped(), but when "bsl" is TRUE also escape
1198 * characters where rem_backslash() would remove the backslash.
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001199 * Escape the characters with "cc".
Bram Moolenaar071d4272004-06-13 20:20:40 +00001200 */
1201 char_u *
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001202vim_strsave_escaped_ext(string, esc_chars, cc, bsl)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001203 char_u *string;
1204 char_u *esc_chars;
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001205 int cc;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001206 int bsl;
1207{
1208 char_u *p;
1209 char_u *p2;
1210 char_u *escaped_string;
1211 unsigned length;
1212#ifdef FEAT_MBYTE
1213 int l;
1214#endif
1215
1216 /*
1217 * First count the number of backslashes required.
1218 * Then allocate the memory and insert them.
1219 */
1220 length = 1; /* count the trailing NUL */
1221 for (p = string; *p; p++)
1222 {
1223#ifdef FEAT_MBYTE
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001224 if (has_mbyte && (l = (*mb_ptr2len)(p)) > 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001225 {
1226 length += l; /* count a multibyte char */
1227 p += l - 1;
1228 continue;
1229 }
1230#endif
1231 if (vim_strchr(esc_chars, *p) != NULL || (bsl && rem_backslash(p)))
1232 ++length; /* count a backslash */
1233 ++length; /* count an ordinary char */
1234 }
1235 escaped_string = alloc(length);
1236 if (escaped_string != NULL)
1237 {
1238 p2 = escaped_string;
1239 for (p = string; *p; p++)
1240 {
1241#ifdef FEAT_MBYTE
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001242 if (has_mbyte && (l = (*mb_ptr2len)(p)) > 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001243 {
1244 mch_memmove(p2, p, (size_t)l);
1245 p2 += l;
1246 p += l - 1; /* skip multibyte char */
1247 continue;
1248 }
1249#endif
1250 if (vim_strchr(esc_chars, *p) != NULL || (bsl && rem_backslash(p)))
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001251 *p2++ = cc;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001252 *p2++ = *p;
1253 }
1254 *p2 = NUL;
1255 }
1256 return escaped_string;
1257}
1258
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001259#if defined(FEAT_EVAL) || defined(PROTO)
1260/*
1261 * Escape "string" for use as a shell argument with system().
1262 * This uses single quotes, except when we know we need to use double qoutes
1263 * (MS-DOS and MS-Windows without 'shellslash' set).
1264 * Returns the result in allocated memory, NULL if we have run out.
1265 */
1266 char_u *
1267vim_strsave_shellescape(string)
1268 char_u *string;
1269{
1270 unsigned length;
1271 char_u *p;
1272 char_u *d;
1273 char_u *escaped_string;
1274
1275 /* First count the number of extra bytes required. */
Bram Moolenaar77f66d62007-02-20 02:16:18 +00001276 length = (unsigned)STRLEN(string) + 3; /* two quotes and a trailing NUL */
Bram Moolenaar60a495f2006-10-03 12:44:42 +00001277 for (p = string; *p != NUL; mb_ptr_adv(p))
1278 {
1279# if defined(WIN32) || defined(WIN16) || defined(DOS)
1280 if (!p_ssl)
1281 {
1282 if (*p == '"')
1283 ++length; /* " -> "" */
1284 }
1285 else
1286# endif
1287 if (*p == '\'')
1288 length += 3; /* ' => '\'' */
1289 }
1290
1291 /* Allocate memory for the result and fill it. */
1292 escaped_string = alloc(length);
1293 if (escaped_string != NULL)
1294 {
1295 d = escaped_string;
1296
1297 /* add opening quote */
1298# if defined(WIN32) || defined(WIN16) || defined(DOS)
1299 if (!p_ssl)
1300 *d++ = '"';
1301 else
1302# endif
1303 *d++ = '\'';
1304
1305 for (p = string; *p != NUL; )
1306 {
1307# if defined(WIN32) || defined(WIN16) || defined(DOS)
1308 if (!p_ssl)
1309 {
1310 if (*p == '"')
1311 {
1312 *d++ = '"';
1313 *d++ = '"';
1314 ++p;
1315 continue;
1316 }
1317 }
1318 else
1319# endif
1320 if (*p == '\'')
1321 {
1322 *d++='\'';
1323 *d++='\\';
1324 *d++='\'';
1325 *d++='\'';
1326 ++p;
1327 continue;
1328 }
1329
1330 MB_COPY_CHAR(p, d);
1331 }
1332
1333 /* add terminating quote and finish with a NUL */
1334# if defined(WIN32) || defined(WIN16) || defined(DOS)
1335 if (!p_ssl)
1336 *d++ = '"';
1337 else
1338# endif
1339 *d++ = '\'';
1340 *d = NUL;
1341 }
1342
1343 return escaped_string;
1344}
1345#endif
1346
Bram Moolenaar071d4272004-06-13 20:20:40 +00001347/*
1348 * Like vim_strsave(), but make all characters uppercase.
1349 * This uses ASCII lower-to-upper case translation, language independent.
1350 */
1351 char_u *
1352vim_strsave_up(string)
1353 char_u *string;
1354{
1355 char_u *p1;
1356
1357 p1 = vim_strsave(string);
1358 vim_strup(p1);
1359 return p1;
1360}
1361
1362/*
1363 * Like vim_strnsave(), but make all characters uppercase.
1364 * This uses ASCII lower-to-upper case translation, language independent.
1365 */
1366 char_u *
1367vim_strnsave_up(string, len)
1368 char_u *string;
1369 int len;
1370{
1371 char_u *p1;
1372
1373 p1 = vim_strnsave(string, len);
1374 vim_strup(p1);
1375 return p1;
1376}
1377
1378/*
1379 * ASCII lower-to-upper case translation, language independent.
1380 */
1381 void
1382vim_strup(p)
1383 char_u *p;
1384{
1385 char_u *p2;
1386 int c;
1387
1388 if (p != NULL)
1389 {
1390 p2 = p;
1391 while ((c = *p2) != NUL)
1392#ifdef EBCDIC
1393 *p2++ = isalpha(c) ? toupper(c) : c;
1394#else
1395 *p2++ = (c < 'a' || c > 'z') ? c : (c - 0x20);
1396#endif
1397 }
1398}
1399
Bram Moolenaarc4956c82006-03-12 21:58:43 +00001400#if defined(FEAT_EVAL) || defined(FEAT_SPELL) || defined(PROTO)
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001401/*
1402 * Make string "s" all upper-case and return it in allocated memory.
1403 * Handles multi-byte characters as well as possible.
1404 * Returns NULL when out of memory.
1405 */
1406 char_u *
1407strup_save(orig)
1408 char_u *orig;
1409{
1410 char_u *p;
1411 char_u *res;
1412
1413 res = p = vim_strsave(orig);
1414
1415 if (res != NULL)
1416 while (*p != NUL)
1417 {
1418# ifdef FEAT_MBYTE
1419 int l;
1420
1421 if (enc_utf8)
1422 {
1423 int c, uc;
1424 int nl;
1425 char_u *s;
1426
1427 c = utf_ptr2char(p);
1428 uc = utf_toupper(c);
1429
1430 /* Reallocate string when byte count changes. This is rare,
1431 * thus it's OK to do another malloc()/free(). */
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001432 l = utf_ptr2len(p);
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001433 nl = utf_char2len(uc);
1434 if (nl != l)
1435 {
1436 s = alloc((unsigned)STRLEN(res) + 1 + nl - l);
1437 if (s == NULL)
1438 break;
1439 mch_memmove(s, res, p - res);
1440 STRCPY(s + (p - res) + nl, p + l);
1441 p = s + (p - res);
1442 vim_free(res);
1443 res = s;
1444 }
1445
1446 utf_char2bytes(uc, p);
1447 p += nl;
1448 }
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001449 else if (has_mbyte && (l = (*mb_ptr2len)(p)) > 1)
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001450 p += l; /* skip multi-byte character */
1451 else
1452# endif
1453 {
1454 *p = TOUPPER_LOC(*p); /* note that toupper() can be a macro */
1455 p++;
1456 }
1457 }
1458
1459 return res;
1460}
1461#endif
1462
Bram Moolenaar071d4272004-06-13 20:20:40 +00001463/*
1464 * copy a space a number of times
1465 */
1466 void
1467copy_spaces(ptr, count)
1468 char_u *ptr;
1469 size_t count;
1470{
1471 size_t i = count;
1472 char_u *p = ptr;
1473
1474 while (i--)
1475 *p++ = ' ';
1476}
1477
1478#if defined(FEAT_VISUALEXTRA) || defined(PROTO)
1479/*
1480 * Copy a character a number of times.
1481 * Does not work for multi-byte charactes!
1482 */
1483 void
1484copy_chars(ptr, count, c)
1485 char_u *ptr;
1486 size_t count;
1487 int c;
1488{
1489 size_t i = count;
1490 char_u *p = ptr;
1491
1492 while (i--)
1493 *p++ = c;
1494}
1495#endif
1496
1497/*
1498 * delete spaces at the end of a string
1499 */
1500 void
1501del_trailing_spaces(ptr)
1502 char_u *ptr;
1503{
1504 char_u *q;
1505
1506 q = ptr + STRLEN(ptr);
1507 while (--q > ptr && vim_iswhite(q[0]) && q[-1] != '\\' && q[-1] != Ctrl_V)
1508 *q = NUL;
1509}
1510
1511/*
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001512 * Like strncpy(), but always terminate the result with one NUL.
Bram Moolenaard042c562005-06-30 22:04:15 +00001513 * "to" must be "len + 1" long!
Bram Moolenaar071d4272004-06-13 20:20:40 +00001514 */
1515 void
1516vim_strncpy(to, from, len)
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001517 char_u *to;
1518 char_u *from;
Bram Moolenaarbbebc852005-07-18 21:47:53 +00001519 size_t len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001520{
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +00001521 STRNCPY(to, from, len);
1522 to[len] = NUL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001523}
1524
1525/*
1526 * Isolate one part of a string option where parts are separated with
1527 * "sep_chars".
Bram Moolenaar83bab712005-08-01 21:58:57 +00001528 * The part is copied into "buf[maxlen]".
Bram Moolenaar071d4272004-06-13 20:20:40 +00001529 * "*option" is advanced to the next part.
1530 * The length is returned.
1531 */
1532 int
1533copy_option_part(option, buf, maxlen, sep_chars)
1534 char_u **option;
1535 char_u *buf;
1536 int maxlen;
1537 char *sep_chars;
1538{
1539 int len = 0;
1540 char_u *p = *option;
1541
1542 /* skip '.' at start of option part, for 'suffixes' */
1543 if (*p == '.')
1544 buf[len++] = *p++;
1545 while (*p != NUL && vim_strchr((char_u *)sep_chars, *p) == NULL)
1546 {
1547 /*
1548 * Skip backslash before a separator character and space.
1549 */
1550 if (p[0] == '\\' && vim_strchr((char_u *)sep_chars, p[1]) != NULL)
1551 ++p;
1552 if (len < maxlen - 1)
1553 buf[len++] = *p;
1554 ++p;
1555 }
1556 buf[len] = NUL;
1557
1558 if (*p != NUL && *p != ',') /* skip non-standard separator */
1559 ++p;
1560 p = skip_to_option_part(p); /* p points to next file name */
1561
1562 *option = p;
1563 return len;
1564}
1565
1566/*
Bram Moolenaar4770d092006-01-12 23:22:24 +00001567 * Replacement for free() that ignores NULL pointers.
1568 * Also skip free() when exiting for sure, this helps when we caught a deadly
1569 * signal that was caused by a crash in free().
Bram Moolenaar071d4272004-06-13 20:20:40 +00001570 */
1571 void
1572vim_free(x)
1573 void *x;
1574{
Bram Moolenaar4770d092006-01-12 23:22:24 +00001575 if (x != NULL && !really_exiting)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001576 {
1577#ifdef MEM_PROFILE
1578 mem_pre_free(&x);
1579#endif
1580 free(x);
1581 }
1582}
1583
1584#ifndef HAVE_MEMSET
1585 void *
1586vim_memset(ptr, c, size)
1587 void *ptr;
1588 int c;
1589 size_t size;
1590{
1591 char *p = ptr;
1592
1593 while (size-- > 0)
1594 *p++ = c;
1595 return ptr;
1596}
1597#endif
1598
1599#ifdef VIM_MEMCMP
1600/*
1601 * Return zero when "b1" and "b2" are the same for "len" bytes.
1602 * Return non-zero otherwise.
1603 */
1604 int
1605vim_memcmp(b1, b2, len)
1606 void *b1;
1607 void *b2;
1608 size_t len;
1609{
1610 char_u *p1 = (char_u *)b1, *p2 = (char_u *)b2;
1611
1612 for ( ; len > 0; --len)
1613 {
1614 if (*p1 != *p2)
1615 return 1;
1616 ++p1;
1617 ++p2;
1618 }
1619 return 0;
1620}
1621#endif
1622
1623#ifdef VIM_MEMMOVE
1624/*
1625 * Version of memmove() that handles overlapping source and destination.
1626 * For systems that don't have a function that is guaranteed to do that (SYSV).
1627 */
1628 void
1629mch_memmove(dst_arg, src_arg, len)
1630 void *src_arg, *dst_arg;
1631 size_t len;
1632{
1633 /*
1634 * A void doesn't have a size, we use char pointers.
1635 */
1636 char *dst = dst_arg, *src = src_arg;
1637
1638 /* overlap, copy backwards */
1639 if (dst > src && dst < src + len)
1640 {
1641 src += len;
1642 dst += len;
1643 while (len-- > 0)
1644 *--dst = *--src;
1645 }
1646 else /* copy forwards */
1647 while (len-- > 0)
1648 *dst++ = *src++;
1649}
1650#endif
1651
1652#if (!defined(HAVE_STRCASECMP) && !defined(HAVE_STRICMP)) || defined(PROTO)
1653/*
1654 * Compare two strings, ignoring case, using current locale.
1655 * Doesn't work for multi-byte characters.
1656 * return 0 for match, < 0 for smaller, > 0 for bigger
1657 */
1658 int
1659vim_stricmp(s1, s2)
1660 char *s1;
1661 char *s2;
1662{
1663 int i;
1664
1665 for (;;)
1666 {
1667 i = (int)TOLOWER_LOC(*s1) - (int)TOLOWER_LOC(*s2);
1668 if (i != 0)
1669 return i; /* this character different */
1670 if (*s1 == NUL)
1671 break; /* strings match until NUL */
1672 ++s1;
1673 ++s2;
1674 }
1675 return 0; /* strings match */
1676}
1677#endif
1678
1679#if (!defined(HAVE_STRNCASECMP) && !defined(HAVE_STRNICMP)) || defined(PROTO)
1680/*
1681 * Compare two strings, for length "len", ignoring case, using current locale.
1682 * Doesn't work for multi-byte characters.
1683 * return 0 for match, < 0 for smaller, > 0 for bigger
1684 */
1685 int
1686vim_strnicmp(s1, s2, len)
1687 char *s1;
1688 char *s2;
1689 size_t len;
1690{
1691 int i;
1692
1693 while (len > 0)
1694 {
1695 i = (int)TOLOWER_LOC(*s1) - (int)TOLOWER_LOC(*s2);
1696 if (i != 0)
1697 return i; /* this character different */
1698 if (*s1 == NUL)
1699 break; /* strings match until NUL */
1700 ++s1;
1701 ++s2;
1702 --len;
1703 }
1704 return 0; /* strings match */
1705}
1706#endif
1707
1708#if 0 /* currently not used */
1709/*
1710 * Check if string "s2" appears somewhere in "s1" while ignoring case.
1711 * Return NULL if not, a pointer to the first occurrence if it does.
1712 */
1713 char_u *
1714vim_stristr(s1, s2)
1715 char_u *s1;
1716 char_u *s2;
1717{
1718 char_u *p;
1719 int len = STRLEN(s2);
1720 char_u *end = s1 + STRLEN(s1) - len;
1721
1722 for (p = s1; p <= end; ++p)
1723 if (STRNICMP(p, s2, len) == 0)
1724 return p;
1725 return NULL;
1726}
1727#endif
1728
1729/*
1730 * Version of strchr() and strrchr() that handle unsigned char strings
Bram Moolenaar05159a02005-02-26 23:04:13 +00001731 * with characters from 128 to 255 correctly. It also doesn't return a
1732 * pointer to the NUL at the end of the string.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001733 */
1734 char_u *
1735vim_strchr(string, c)
1736 char_u *string;
1737 int c;
1738{
1739 char_u *p;
1740 int b;
1741
1742 p = string;
1743#ifdef FEAT_MBYTE
1744 if (enc_utf8 && c >= 0x80)
1745 {
1746 while (*p != NUL)
1747 {
1748 if (utf_ptr2char(p) == c)
1749 return p;
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001750 p += (*mb_ptr2len)(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001751 }
1752 return NULL;
1753 }
1754 if (enc_dbcs != 0 && c > 255)
1755 {
1756 int n2 = c & 0xff;
1757
1758 c = ((unsigned)c >> 8) & 0xff;
1759 while ((b = *p) != NUL)
1760 {
1761 if (b == c && p[1] == n2)
1762 return p;
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001763 p += (*mb_ptr2len)(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001764 }
1765 return NULL;
1766 }
1767 if (has_mbyte)
1768 {
1769 while ((b = *p) != NUL)
1770 {
1771 if (b == c)
1772 return p;
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00001773 p += (*mb_ptr2len)(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001774 }
1775 return NULL;
1776 }
1777#endif
1778 while ((b = *p) != NUL)
1779 {
1780 if (b == c)
1781 return p;
1782 ++p;
1783 }
1784 return NULL;
1785}
1786
1787/*
Bram Moolenaar05159a02005-02-26 23:04:13 +00001788 * Version of strchr() that only works for bytes and handles unsigned char
1789 * strings with characters above 128 correctly. It also doesn't return a
1790 * pointer to the NUL at the end of the string.
1791 */
1792 char_u *
1793vim_strbyte(string, c)
1794 char_u *string;
1795 int c;
1796{
1797 char_u *p = string;
1798
1799 while (*p != NUL)
1800 {
1801 if (*p == c)
1802 return p;
1803 ++p;
1804 }
1805 return NULL;
1806}
1807
1808/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00001809 * Search for last occurrence of "c" in "string".
Bram Moolenaarac6e65f2005-08-29 22:25:38 +00001810 * Return NULL if not found.
Bram Moolenaar05159a02005-02-26 23:04:13 +00001811 * Does not handle multi-byte char for "c"!
Bram Moolenaar071d4272004-06-13 20:20:40 +00001812 */
1813 char_u *
1814vim_strrchr(string, c)
1815 char_u *string;
1816 int c;
1817{
1818 char_u *retval = NULL;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001819 char_u *p = string;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001820
Bram Moolenaar05159a02005-02-26 23:04:13 +00001821 while (*p)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001822 {
Bram Moolenaar05159a02005-02-26 23:04:13 +00001823 if (*p == c)
1824 retval = p;
1825 mb_ptr_adv(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001826 }
1827 return retval;
1828}
1829
1830/*
1831 * Vim's version of strpbrk(), in case it's missing.
1832 * Don't generate a prototype for this, causes problems when it's not used.
1833 */
1834#ifndef PROTO
1835# ifndef HAVE_STRPBRK
1836# ifdef vim_strpbrk
1837# undef vim_strpbrk
1838# endif
1839 char_u *
1840vim_strpbrk(s, charset)
1841 char_u *s;
1842 char_u *charset;
1843{
1844 while (*s)
1845 {
1846 if (vim_strchr(charset, *s) != NULL)
1847 return s;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001848 mb_ptr_adv(s);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001849 }
1850 return NULL;
1851}
1852# endif
1853#endif
1854
1855/*
1856 * Vim has its own isspace() function, because on some machines isspace()
1857 * can't handle characters above 128.
1858 */
1859 int
1860vim_isspace(x)
1861 int x;
1862{
1863 return ((x >= 9 && x <= 13) || x == ' ');
1864}
1865
1866/************************************************************************
Bram Moolenaar383f9bc2005-01-19 22:18:32 +00001867 * Functions for handling growing arrays.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001868 */
1869
1870/*
1871 * Clear an allocated growing array.
1872 */
1873 void
1874ga_clear(gap)
1875 garray_T *gap;
1876{
1877 vim_free(gap->ga_data);
1878 ga_init(gap);
1879}
1880
1881/*
1882 * Clear a growing array that contains a list of strings.
1883 */
1884 void
1885ga_clear_strings(gap)
1886 garray_T *gap;
1887{
1888 int i;
1889
1890 for (i = 0; i < gap->ga_len; ++i)
1891 vim_free(((char_u **)(gap->ga_data))[i]);
1892 ga_clear(gap);
1893}
1894
1895/*
1896 * Initialize a growing array. Don't forget to set ga_itemsize and
1897 * ga_growsize! Or use ga_init2().
1898 */
1899 void
1900ga_init(gap)
1901 garray_T *gap;
1902{
1903 gap->ga_data = NULL;
Bram Moolenaar86b68352004-12-27 21:59:20 +00001904 gap->ga_maxlen = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001905 gap->ga_len = 0;
1906}
1907
1908 void
1909ga_init2(gap, itemsize, growsize)
1910 garray_T *gap;
1911 int itemsize;
1912 int growsize;
1913{
1914 ga_init(gap);
1915 gap->ga_itemsize = itemsize;
1916 gap->ga_growsize = growsize;
1917}
1918
1919/*
1920 * Make room in growing array "gap" for at least "n" items.
1921 * Return FAIL for failure, OK otherwise.
1922 */
1923 int
1924ga_grow(gap, n)
1925 garray_T *gap;
1926 int n;
1927{
1928 size_t len;
1929 char_u *pp;
1930
Bram Moolenaar86b68352004-12-27 21:59:20 +00001931 if (gap->ga_maxlen - gap->ga_len < n)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001932 {
1933 if (n < gap->ga_growsize)
1934 n = gap->ga_growsize;
1935 len = gap->ga_itemsize * (gap->ga_len + n);
1936 pp = alloc_clear((unsigned)len);
1937 if (pp == NULL)
1938 return FAIL;
Bram Moolenaar86b68352004-12-27 21:59:20 +00001939 gap->ga_maxlen = gap->ga_len + n;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001940 if (gap->ga_data != NULL)
1941 {
1942 mch_memmove(pp, gap->ga_data,
1943 (size_t)(gap->ga_itemsize * gap->ga_len));
1944 vim_free(gap->ga_data);
1945 }
1946 gap->ga_data = pp;
1947 }
1948 return OK;
1949}
1950
1951/*
1952 * Concatenate a string to a growarray which contains characters.
1953 * Note: Does NOT copy the NUL at the end!
1954 */
1955 void
1956ga_concat(gap, s)
1957 garray_T *gap;
1958 char_u *s;
1959{
1960 int len = (int)STRLEN(s);
1961
1962 if (ga_grow(gap, len) == OK)
1963 {
1964 mch_memmove((char *)gap->ga_data + gap->ga_len, s, (size_t)len);
1965 gap->ga_len += len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001966 }
1967}
1968
1969/*
1970 * Append one byte to a growarray which contains bytes.
1971 */
1972 void
1973ga_append(gap, c)
1974 garray_T *gap;
1975 int c;
1976{
1977 if (ga_grow(gap, 1) == OK)
1978 {
1979 *((char *)gap->ga_data + gap->ga_len) = c;
1980 ++gap->ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001981 }
1982}
1983
1984/************************************************************************
1985 * functions that use lookup tables for various things, generally to do with
1986 * special key codes.
1987 */
1988
1989/*
1990 * Some useful tables.
1991 */
1992
1993static struct modmasktable
1994{
1995 short mod_mask; /* Bit-mask for particular key modifier */
1996 short mod_flag; /* Bit(s) for particular key modifier */
1997 char_u name; /* Single letter name of modifier */
1998} mod_mask_table[] =
1999{
2000 {MOD_MASK_ALT, MOD_MASK_ALT, (char_u)'M'},
Bram Moolenaar19a09a12005-03-04 23:39:37 +00002001 {MOD_MASK_META, MOD_MASK_META, (char_u)'T'},
Bram Moolenaar071d4272004-06-13 20:20:40 +00002002 {MOD_MASK_CTRL, MOD_MASK_CTRL, (char_u)'C'},
2003 {MOD_MASK_SHIFT, MOD_MASK_SHIFT, (char_u)'S'},
2004 {MOD_MASK_MULTI_CLICK, MOD_MASK_2CLICK, (char_u)'2'},
2005 {MOD_MASK_MULTI_CLICK, MOD_MASK_3CLICK, (char_u)'3'},
2006 {MOD_MASK_MULTI_CLICK, MOD_MASK_4CLICK, (char_u)'4'},
2007#ifdef MACOS
2008 {MOD_MASK_CMD, MOD_MASK_CMD, (char_u)'D'},
2009#endif
2010 /* 'A' must be the last one */
2011 {MOD_MASK_ALT, MOD_MASK_ALT, (char_u)'A'},
2012 {0, 0, NUL}
2013};
2014
2015/*
2016 * Shifted key terminal codes and their unshifted equivalent.
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00002017 * Don't add mouse codes here, they are handled separately!
Bram Moolenaar071d4272004-06-13 20:20:40 +00002018 */
2019#define MOD_KEYS_ENTRY_SIZE 5
2020
2021static char_u modifier_keys_table[] =
2022{
2023/* mod mask with modifier without modifier */
2024 MOD_MASK_SHIFT, '&', '9', '@', '1', /* begin */
2025 MOD_MASK_SHIFT, '&', '0', '@', '2', /* cancel */
2026 MOD_MASK_SHIFT, '*', '1', '@', '4', /* command */
2027 MOD_MASK_SHIFT, '*', '2', '@', '5', /* copy */
2028 MOD_MASK_SHIFT, '*', '3', '@', '6', /* create */
2029 MOD_MASK_SHIFT, '*', '4', 'k', 'D', /* delete char */
2030 MOD_MASK_SHIFT, '*', '5', 'k', 'L', /* delete line */
2031 MOD_MASK_SHIFT, '*', '7', '@', '7', /* end */
2032 MOD_MASK_CTRL, KS_EXTRA, (int)KE_C_END, '@', '7', /* end */
2033 MOD_MASK_SHIFT, '*', '9', '@', '9', /* exit */
2034 MOD_MASK_SHIFT, '*', '0', '@', '0', /* find */
2035 MOD_MASK_SHIFT, '#', '1', '%', '1', /* help */
2036 MOD_MASK_SHIFT, '#', '2', 'k', 'h', /* home */
2037 MOD_MASK_CTRL, KS_EXTRA, (int)KE_C_HOME, 'k', 'h', /* home */
2038 MOD_MASK_SHIFT, '#', '3', 'k', 'I', /* insert */
2039 MOD_MASK_SHIFT, '#', '4', 'k', 'l', /* left arrow */
2040 MOD_MASK_CTRL, KS_EXTRA, (int)KE_C_LEFT, 'k', 'l', /* left arrow */
2041 MOD_MASK_SHIFT, '%', 'a', '%', '3', /* message */
2042 MOD_MASK_SHIFT, '%', 'b', '%', '4', /* move */
2043 MOD_MASK_SHIFT, '%', 'c', '%', '5', /* next */
2044 MOD_MASK_SHIFT, '%', 'd', '%', '7', /* options */
2045 MOD_MASK_SHIFT, '%', 'e', '%', '8', /* previous */
2046 MOD_MASK_SHIFT, '%', 'f', '%', '9', /* print */
2047 MOD_MASK_SHIFT, '%', 'g', '%', '0', /* redo */
2048 MOD_MASK_SHIFT, '%', 'h', '&', '3', /* replace */
2049 MOD_MASK_SHIFT, '%', 'i', 'k', 'r', /* right arr. */
2050 MOD_MASK_CTRL, KS_EXTRA, (int)KE_C_RIGHT, 'k', 'r', /* right arr. */
2051 MOD_MASK_SHIFT, '%', 'j', '&', '5', /* resume */
2052 MOD_MASK_SHIFT, '!', '1', '&', '6', /* save */
2053 MOD_MASK_SHIFT, '!', '2', '&', '7', /* suspend */
2054 MOD_MASK_SHIFT, '!', '3', '&', '8', /* undo */
2055 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_UP, 'k', 'u', /* up arrow */
2056 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_DOWN, 'k', 'd', /* down arrow */
2057
2058 /* vt100 F1 */
2059 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_XF1, KS_EXTRA, (int)KE_XF1,
2060 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_XF2, KS_EXTRA, (int)KE_XF2,
2061 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_XF3, KS_EXTRA, (int)KE_XF3,
2062 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_XF4, KS_EXTRA, (int)KE_XF4,
2063
2064 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F1, 'k', '1', /* F1 */
2065 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F2, 'k', '2',
2066 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F3, 'k', '3',
2067 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F4, 'k', '4',
2068 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F5, 'k', '5',
2069 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F6, 'k', '6',
2070 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F7, 'k', '7',
2071 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F8, 'k', '8',
2072 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F9, 'k', '9',
2073 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F10, 'k', ';', /* F10 */
2074
2075 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F11, 'F', '1',
2076 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F12, 'F', '2',
2077 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F13, 'F', '3',
2078 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F14, 'F', '4',
2079 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F15, 'F', '5',
2080 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F16, 'F', '6',
2081 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F17, 'F', '7',
2082 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F18, 'F', '8',
2083 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F19, 'F', '9',
2084 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F20, 'F', 'A',
2085
2086 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F21, 'F', 'B',
2087 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F22, 'F', 'C',
2088 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F23, 'F', 'D',
2089 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F24, 'F', 'E',
2090 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F25, 'F', 'F',
2091 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F26, 'F', 'G',
2092 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F27, 'F', 'H',
2093 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F28, 'F', 'I',
2094 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F29, 'F', 'J',
2095 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F30, 'F', 'K',
2096
2097 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F31, 'F', 'L',
2098 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F32, 'F', 'M',
2099 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F33, 'F', 'N',
2100 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F34, 'F', 'O',
2101 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F35, 'F', 'P',
2102 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F36, 'F', 'Q',
2103 MOD_MASK_SHIFT, KS_EXTRA, (int)KE_S_F37, 'F', 'R',
2104
2105 /* TAB pseudo code*/
2106 MOD_MASK_SHIFT, 'k', 'B', KS_EXTRA, (int)KE_TAB,
2107
2108 NUL
2109};
2110
2111static struct key_name_entry
2112{
2113 int key; /* Special key code or ascii value */
2114 char_u *name; /* Name of key */
2115} key_names_table[] =
2116{
2117 {' ', (char_u *)"Space"},
2118 {TAB, (char_u *)"Tab"},
2119 {K_TAB, (char_u *)"Tab"},
2120 {NL, (char_u *)"NL"},
2121 {NL, (char_u *)"NewLine"}, /* Alternative name */
2122 {NL, (char_u *)"LineFeed"}, /* Alternative name */
2123 {NL, (char_u *)"LF"}, /* Alternative name */
2124 {CAR, (char_u *)"CR"},
2125 {CAR, (char_u *)"Return"}, /* Alternative name */
2126 {CAR, (char_u *)"Enter"}, /* Alternative name */
2127 {K_BS, (char_u *)"BS"},
2128 {K_BS, (char_u *)"BackSpace"}, /* Alternative name */
2129 {ESC, (char_u *)"Esc"},
2130 {CSI, (char_u *)"CSI"},
2131 {K_CSI, (char_u *)"xCSI"},
2132 {'|', (char_u *)"Bar"},
2133 {'\\', (char_u *)"Bslash"},
2134 {K_DEL, (char_u *)"Del"},
2135 {K_DEL, (char_u *)"Delete"}, /* Alternative name */
2136 {K_KDEL, (char_u *)"kDel"},
2137 {K_UP, (char_u *)"Up"},
2138 {K_DOWN, (char_u *)"Down"},
2139 {K_LEFT, (char_u *)"Left"},
2140 {K_RIGHT, (char_u *)"Right"},
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002141 {K_XUP, (char_u *)"xUp"},
2142 {K_XDOWN, (char_u *)"xDown"},
2143 {K_XLEFT, (char_u *)"xLeft"},
2144 {K_XRIGHT, (char_u *)"xRight"},
Bram Moolenaar071d4272004-06-13 20:20:40 +00002145
2146 {K_F1, (char_u *)"F1"},
2147 {K_F2, (char_u *)"F2"},
2148 {K_F3, (char_u *)"F3"},
2149 {K_F4, (char_u *)"F4"},
2150 {K_F5, (char_u *)"F5"},
2151 {K_F6, (char_u *)"F6"},
2152 {K_F7, (char_u *)"F7"},
2153 {K_F8, (char_u *)"F8"},
2154 {K_F9, (char_u *)"F9"},
2155 {K_F10, (char_u *)"F10"},
2156
2157 {K_F11, (char_u *)"F11"},
2158 {K_F12, (char_u *)"F12"},
2159 {K_F13, (char_u *)"F13"},
2160 {K_F14, (char_u *)"F14"},
2161 {K_F15, (char_u *)"F15"},
2162 {K_F16, (char_u *)"F16"},
2163 {K_F17, (char_u *)"F17"},
2164 {K_F18, (char_u *)"F18"},
2165 {K_F19, (char_u *)"F19"},
2166 {K_F20, (char_u *)"F20"},
2167
2168 {K_F21, (char_u *)"F21"},
2169 {K_F22, (char_u *)"F22"},
2170 {K_F23, (char_u *)"F23"},
2171 {K_F24, (char_u *)"F24"},
2172 {K_F25, (char_u *)"F25"},
2173 {K_F26, (char_u *)"F26"},
2174 {K_F27, (char_u *)"F27"},
2175 {K_F28, (char_u *)"F28"},
2176 {K_F29, (char_u *)"F29"},
2177 {K_F30, (char_u *)"F30"},
2178
2179 {K_F31, (char_u *)"F31"},
2180 {K_F32, (char_u *)"F32"},
2181 {K_F33, (char_u *)"F33"},
2182 {K_F34, (char_u *)"F34"},
2183 {K_F35, (char_u *)"F35"},
2184 {K_F36, (char_u *)"F36"},
2185 {K_F37, (char_u *)"F37"},
2186
2187 {K_XF1, (char_u *)"xF1"},
2188 {K_XF2, (char_u *)"xF2"},
2189 {K_XF3, (char_u *)"xF3"},
2190 {K_XF4, (char_u *)"xF4"},
2191
2192 {K_HELP, (char_u *)"Help"},
2193 {K_UNDO, (char_u *)"Undo"},
2194 {K_INS, (char_u *)"Insert"},
2195 {K_INS, (char_u *)"Ins"}, /* Alternative name */
2196 {K_KINS, (char_u *)"kInsert"},
2197 {K_HOME, (char_u *)"Home"},
2198 {K_KHOME, (char_u *)"kHome"},
2199 {K_XHOME, (char_u *)"xHome"},
Bram Moolenaar68b76a62005-03-25 21:53:48 +00002200 {K_ZHOME, (char_u *)"zHome"},
Bram Moolenaar071d4272004-06-13 20:20:40 +00002201 {K_END, (char_u *)"End"},
2202 {K_KEND, (char_u *)"kEnd"},
2203 {K_XEND, (char_u *)"xEnd"},
Bram Moolenaar68b76a62005-03-25 21:53:48 +00002204 {K_ZEND, (char_u *)"zEnd"},
Bram Moolenaar071d4272004-06-13 20:20:40 +00002205 {K_PAGEUP, (char_u *)"PageUp"},
2206 {K_PAGEDOWN, (char_u *)"PageDown"},
2207 {K_KPAGEUP, (char_u *)"kPageUp"},
2208 {K_KPAGEDOWN, (char_u *)"kPageDown"},
2209
2210 {K_KPLUS, (char_u *)"kPlus"},
2211 {K_KMINUS, (char_u *)"kMinus"},
2212 {K_KDIVIDE, (char_u *)"kDivide"},
2213 {K_KMULTIPLY, (char_u *)"kMultiply"},
2214 {K_KENTER, (char_u *)"kEnter"},
2215 {K_KPOINT, (char_u *)"kPoint"},
2216
2217 {K_K0, (char_u *)"k0"},
2218 {K_K1, (char_u *)"k1"},
2219 {K_K2, (char_u *)"k2"},
2220 {K_K3, (char_u *)"k3"},
2221 {K_K4, (char_u *)"k4"},
2222 {K_K5, (char_u *)"k5"},
2223 {K_K6, (char_u *)"k6"},
2224 {K_K7, (char_u *)"k7"},
2225 {K_K8, (char_u *)"k8"},
2226 {K_K9, (char_u *)"k9"},
2227
2228 {'<', (char_u *)"lt"},
2229
2230 {K_MOUSE, (char_u *)"Mouse"},
2231 {K_NETTERM_MOUSE, (char_u *)"NetMouse"},
2232 {K_DEC_MOUSE, (char_u *)"DecMouse"},
2233 {K_JSBTERM_MOUSE, (char_u *)"JsbMouse"},
2234 {K_PTERM_MOUSE, (char_u *)"PtermMouse"},
2235 {K_LEFTMOUSE, (char_u *)"LeftMouse"},
2236 {K_LEFTMOUSE_NM, (char_u *)"LeftMouseNM"},
2237 {K_LEFTDRAG, (char_u *)"LeftDrag"},
2238 {K_LEFTRELEASE, (char_u *)"LeftRelease"},
2239 {K_LEFTRELEASE_NM, (char_u *)"LeftReleaseNM"},
2240 {K_MIDDLEMOUSE, (char_u *)"MiddleMouse"},
2241 {K_MIDDLEDRAG, (char_u *)"MiddleDrag"},
2242 {K_MIDDLERELEASE, (char_u *)"MiddleRelease"},
2243 {K_RIGHTMOUSE, (char_u *)"RightMouse"},
2244 {K_RIGHTDRAG, (char_u *)"RightDrag"},
2245 {K_RIGHTRELEASE, (char_u *)"RightRelease"},
2246 {K_MOUSEDOWN, (char_u *)"MouseDown"},
2247 {K_MOUSEUP, (char_u *)"MouseUp"},
2248 {K_X1MOUSE, (char_u *)"X1Mouse"},
2249 {K_X1DRAG, (char_u *)"X1Drag"},
2250 {K_X1RELEASE, (char_u *)"X1Release"},
2251 {K_X2MOUSE, (char_u *)"X2Mouse"},
2252 {K_X2DRAG, (char_u *)"X2Drag"},
2253 {K_X2RELEASE, (char_u *)"X2Release"},
2254 {K_DROP, (char_u *)"Drop"},
2255 {K_ZERO, (char_u *)"Nul"},
2256#ifdef FEAT_EVAL
2257 {K_SNR, (char_u *)"SNR"},
2258#endif
2259 {K_PLUG, (char_u *)"Plug"},
2260 {0, NULL}
2261};
2262
2263#define KEY_NAMES_TABLE_LEN (sizeof(key_names_table) / sizeof(struct key_name_entry))
2264
2265#ifdef FEAT_MOUSE
2266static struct mousetable
2267{
2268 int pseudo_code; /* Code for pseudo mouse event */
2269 int button; /* Which mouse button is it? */
2270 int is_click; /* Is it a mouse button click event? */
2271 int is_drag; /* Is it a mouse drag event? */
2272} mouse_table[] =
2273{
2274 {(int)KE_LEFTMOUSE, MOUSE_LEFT, TRUE, FALSE},
2275#ifdef FEAT_GUI
2276 {(int)KE_LEFTMOUSE_NM, MOUSE_LEFT, TRUE, FALSE},
2277#endif
2278 {(int)KE_LEFTDRAG, MOUSE_LEFT, FALSE, TRUE},
2279 {(int)KE_LEFTRELEASE, MOUSE_LEFT, FALSE, FALSE},
2280#ifdef FEAT_GUI
2281 {(int)KE_LEFTRELEASE_NM, MOUSE_LEFT, FALSE, FALSE},
2282#endif
2283 {(int)KE_MIDDLEMOUSE, MOUSE_MIDDLE, TRUE, FALSE},
2284 {(int)KE_MIDDLEDRAG, MOUSE_MIDDLE, FALSE, TRUE},
2285 {(int)KE_MIDDLERELEASE, MOUSE_MIDDLE, FALSE, FALSE},
2286 {(int)KE_RIGHTMOUSE, MOUSE_RIGHT, TRUE, FALSE},
2287 {(int)KE_RIGHTDRAG, MOUSE_RIGHT, FALSE, TRUE},
2288 {(int)KE_RIGHTRELEASE, MOUSE_RIGHT, FALSE, FALSE},
2289 {(int)KE_X1MOUSE, MOUSE_X1, TRUE, FALSE},
2290 {(int)KE_X1DRAG, MOUSE_X1, FALSE, TRUE},
2291 {(int)KE_X1RELEASE, MOUSE_X1, FALSE, FALSE},
2292 {(int)KE_X2MOUSE, MOUSE_X2, TRUE, FALSE},
2293 {(int)KE_X2DRAG, MOUSE_X2, FALSE, TRUE},
2294 {(int)KE_X2RELEASE, MOUSE_X2, FALSE, FALSE},
2295 /* DRAG without CLICK */
2296 {(int)KE_IGNORE, MOUSE_RELEASE, FALSE, TRUE},
2297 /* RELEASE without CLICK */
2298 {(int)KE_IGNORE, MOUSE_RELEASE, FALSE, FALSE},
2299 {0, 0, 0, 0},
2300};
2301#endif /* FEAT_MOUSE */
2302
2303/*
2304 * Return the modifier mask bit (MOD_MASK_*) which corresponds to the given
2305 * modifier name ('S' for Shift, 'C' for Ctrl etc).
2306 */
2307 int
2308name_to_mod_mask(c)
2309 int c;
2310{
2311 int i;
2312
2313 c = TOUPPER_ASC(c);
2314 for (i = 0; mod_mask_table[i].mod_mask != 0; i++)
2315 if (c == mod_mask_table[i].name)
2316 return mod_mask_table[i].mod_flag;
2317 return 0;
2318}
2319
Bram Moolenaar071d4272004-06-13 20:20:40 +00002320/*
2321 * Check if if there is a special key code for "key" that includes the
2322 * modifiers specified.
2323 */
2324 int
2325simplify_key(key, modifiers)
2326 int key;
2327 int *modifiers;
2328{
2329 int i;
2330 int key0;
2331 int key1;
2332
2333 if (*modifiers & (MOD_MASK_SHIFT | MOD_MASK_CTRL | MOD_MASK_ALT))
2334 {
2335 /* TAB is a special case */
2336 if (key == TAB && (*modifiers & MOD_MASK_SHIFT))
2337 {
2338 *modifiers &= ~MOD_MASK_SHIFT;
2339 return K_S_TAB;
2340 }
2341 key0 = KEY2TERMCAP0(key);
2342 key1 = KEY2TERMCAP1(key);
2343 for (i = 0; modifier_keys_table[i] != NUL; i += MOD_KEYS_ENTRY_SIZE)
2344 if (key0 == modifier_keys_table[i + 3]
2345 && key1 == modifier_keys_table[i + 4]
2346 && (*modifiers & modifier_keys_table[i]))
2347 {
2348 *modifiers &= ~modifier_keys_table[i];
2349 return TERMCAP2KEY(modifier_keys_table[i + 1],
2350 modifier_keys_table[i + 2]);
2351 }
2352 }
2353 return key;
2354}
2355
2356/*
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002357 * Change <xHome> to <Home>, <xUp> to <Up>, etc.
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002358 */
2359 int
2360handle_x_keys(key)
2361 int key;
2362{
2363 switch (key)
2364 {
2365 case K_XUP: return K_UP;
2366 case K_XDOWN: return K_DOWN;
2367 case K_XLEFT: return K_LEFT;
2368 case K_XRIGHT: return K_RIGHT;
2369 case K_XHOME: return K_HOME;
Bram Moolenaar68b76a62005-03-25 21:53:48 +00002370 case K_ZHOME: return K_HOME;
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002371 case K_XEND: return K_END;
Bram Moolenaar68b76a62005-03-25 21:53:48 +00002372 case K_ZEND: return K_END;
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002373 case K_XF1: return K_F1;
2374 case K_XF2: return K_F2;
2375 case K_XF3: return K_F3;
2376 case K_XF4: return K_F4;
2377 case K_S_XF1: return K_S_F1;
2378 case K_S_XF2: return K_S_F2;
2379 case K_S_XF3: return K_S_F3;
2380 case K_S_XF4: return K_S_F4;
2381 }
2382 return key;
2383}
2384
2385/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002386 * Return a string which contains the name of the given key when the given
2387 * modifiers are down.
2388 */
2389 char_u *
2390get_special_key_name(c, modifiers)
2391 int c;
2392 int modifiers;
2393{
2394 static char_u string[MAX_KEY_NAME_LEN + 1];
2395
2396 int i, idx;
2397 int table_idx;
2398 char_u *s;
2399
2400 string[0] = '<';
2401 idx = 1;
2402
2403 /* Key that stands for a normal character. */
2404 if (IS_SPECIAL(c) && KEY2TERMCAP0(c) == KS_KEY)
2405 c = KEY2TERMCAP1(c);
2406
2407 /*
2408 * Translate shifted special keys into unshifted keys and set modifier.
2409 * Same for CTRL and ALT modifiers.
2410 */
2411 if (IS_SPECIAL(c))
2412 {
2413 for (i = 0; modifier_keys_table[i] != 0; i += MOD_KEYS_ENTRY_SIZE)
2414 if ( KEY2TERMCAP0(c) == (int)modifier_keys_table[i + 1]
2415 && (int)KEY2TERMCAP1(c) == (int)modifier_keys_table[i + 2])
2416 {
2417 modifiers |= modifier_keys_table[i];
2418 c = TERMCAP2KEY(modifier_keys_table[i + 3],
2419 modifier_keys_table[i + 4]);
2420 break;
2421 }
2422 }
2423
2424 /* try to find the key in the special key table */
2425 table_idx = find_special_key_in_table(c);
2426
2427 /*
2428 * When not a known special key, and not a printable character, try to
2429 * extract modifiers.
2430 */
2431 if (c > 0
2432#ifdef FEAT_MBYTE
2433 && (*mb_char2len)(c) == 1
2434#endif
2435 )
2436 {
2437 if (table_idx < 0
2438 && (!vim_isprintc(c) || (c & 0x7f) == ' ')
2439 && (c & 0x80))
2440 {
2441 c &= 0x7f;
2442 modifiers |= MOD_MASK_ALT;
2443 /* try again, to find the un-alted key in the special key table */
2444 table_idx = find_special_key_in_table(c);
2445 }
2446 if (table_idx < 0 && !vim_isprintc(c) && c < ' ')
2447 {
2448#ifdef EBCDIC
2449 c = CtrlChar(c);
2450#else
2451 c += '@';
2452#endif
2453 modifiers |= MOD_MASK_CTRL;
2454 }
2455 }
2456
2457 /* translate the modifier into a string */
2458 for (i = 0; mod_mask_table[i].name != 'A'; i++)
2459 if ((modifiers & mod_mask_table[i].mod_mask)
2460 == mod_mask_table[i].mod_flag)
2461 {
2462 string[idx++] = mod_mask_table[i].name;
2463 string[idx++] = (char_u)'-';
2464 }
2465
2466 if (table_idx < 0) /* unknown special key, may output t_xx */
2467 {
2468 if (IS_SPECIAL(c))
2469 {
2470 string[idx++] = 't';
2471 string[idx++] = '_';
2472 string[idx++] = KEY2TERMCAP0(c);
2473 string[idx++] = KEY2TERMCAP1(c);
2474 }
2475 /* Not a special key, only modifiers, output directly */
2476 else
2477 {
2478#ifdef FEAT_MBYTE
2479 if (has_mbyte && (*mb_char2len)(c) > 1)
2480 idx += (*mb_char2bytes)(c, string + idx);
2481 else
2482#endif
2483 if (vim_isprintc(c))
2484 string[idx++] = c;
2485 else
2486 {
2487 s = transchar(c);
2488 while (*s)
2489 string[idx++] = *s++;
2490 }
2491 }
2492 }
2493 else /* use name of special key */
2494 {
2495 STRCPY(string + idx, key_names_table[table_idx].name);
2496 idx = (int)STRLEN(string);
2497 }
2498 string[idx++] = '>';
2499 string[idx] = NUL;
2500 return string;
2501}
2502
2503/*
2504 * Try translating a <> name at (*srcp)[] to dst[].
2505 * Return the number of characters added to dst[], zero for no match.
2506 * If there is a match, srcp is advanced to after the <> name.
2507 * dst[] must be big enough to hold the result (up to six characters)!
2508 */
2509 int
2510trans_special(srcp, dst, keycode)
2511 char_u **srcp;
2512 char_u *dst;
2513 int keycode; /* prefer key code, e.g. K_DEL instead of DEL */
2514{
2515 int modifiers = 0;
2516 int key;
2517 int dlen = 0;
2518
2519 key = find_special_key(srcp, &modifiers, keycode);
2520 if (key == 0)
2521 return 0;
2522
2523 /* Put the appropriate modifier in a string */
2524 if (modifiers != 0)
2525 {
2526 dst[dlen++] = K_SPECIAL;
2527 dst[dlen++] = KS_MODIFIER;
2528 dst[dlen++] = modifiers;
2529 }
2530
2531 if (IS_SPECIAL(key))
2532 {
2533 dst[dlen++] = K_SPECIAL;
2534 dst[dlen++] = KEY2TERMCAP0(key);
2535 dst[dlen++] = KEY2TERMCAP1(key);
2536 }
2537#ifdef FEAT_MBYTE
2538 else if (has_mbyte && !keycode)
2539 dlen += (*mb_char2bytes)(key, dst + dlen);
2540#endif
2541 else if (keycode)
2542 dlen = (int)(add_char2buf(key, dst + dlen) - dst);
2543 else
2544 dst[dlen++] = key;
2545
2546 return dlen;
2547}
2548
2549/*
2550 * Try translating a <> name at (*srcp)[], return the key and modifiers.
2551 * srcp is advanced to after the <> name.
2552 * returns 0 if there is no match.
2553 */
2554 int
2555find_special_key(srcp, modp, keycode)
2556 char_u **srcp;
2557 int *modp;
2558 int keycode; /* prefer key code, e.g. K_DEL instead of DEL */
2559{
2560 char_u *last_dash;
2561 char_u *end_of_name;
2562 char_u *src;
2563 char_u *bp;
2564 int modifiers;
2565 int bit;
2566 int key;
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00002567 unsigned long n;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002568
2569 src = *srcp;
2570 if (src[0] != '<')
2571 return 0;
2572
2573 /* Find end of modifier list */
2574 last_dash = src;
2575 for (bp = src + 1; *bp == '-' || vim_isIDc(*bp); bp++)
2576 {
2577 if (*bp == '-')
2578 {
2579 last_dash = bp;
2580 if (bp[1] != NUL && bp[2] == '>')
2581 ++bp; /* anything accepted, like <C-?> */
2582 }
2583 if (bp[0] == 't' && bp[1] == '_' && bp[2] && bp[3])
2584 bp += 3; /* skip t_xx, xx may be '-' or '>' */
2585 }
2586
2587 if (*bp == '>') /* found matching '>' */
2588 {
2589 end_of_name = bp + 1;
2590
2591 if (STRNICMP(src + 1, "char-", 5) == 0 && VIM_ISDIGIT(src[6]))
2592 {
2593 /* <Char-123> or <Char-033> or <Char-0x33> */
2594 vim_str2nr(src + 6, NULL, NULL, TRUE, TRUE, NULL, &n);
2595 *modp = 0;
2596 *srcp = end_of_name;
2597 return (int)n;
2598 }
2599
2600 /* Which modifiers are given? */
2601 modifiers = 0x0;
2602 for (bp = src + 1; bp < last_dash; bp++)
2603 {
2604 if (*bp != '-')
2605 {
2606 bit = name_to_mod_mask(*bp);
2607 if (bit == 0x0)
2608 break; /* Illegal modifier name */
2609 modifiers |= bit;
2610 }
2611 }
2612
2613 /*
2614 * Legal modifier name.
2615 */
2616 if (bp >= last_dash)
2617 {
2618 /*
2619 * Modifier with single letter, or special key name.
2620 */
2621 if (modifiers != 0 && last_dash[2] == '>')
2622 key = last_dash[1];
2623 else
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002624 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00002625 key = get_special_key_code(last_dash + 1);
Bram Moolenaarbc7aa852005-03-06 23:38:09 +00002626 key = handle_x_keys(key);
2627 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002628
2629 /*
2630 * get_special_key_code() may return NUL for invalid
2631 * special key name.
2632 */
2633 if (key != NUL)
2634 {
2635 /*
2636 * Only use a modifier when there is no special key code that
2637 * includes the modifier.
2638 */
2639 key = simplify_key(key, &modifiers);
2640
2641 if (!keycode)
2642 {
2643 /* don't want keycode, use single byte code */
2644 if (key == K_BS)
2645 key = BS;
2646 else if (key == K_DEL || key == K_KDEL)
2647 key = DEL;
2648 }
2649
2650 /*
2651 * Normal Key with modifier: Try to make a single byte code.
2652 */
2653 if (!IS_SPECIAL(key))
2654 key = extract_modifiers(key, &modifiers);
2655
2656 *modp = modifiers;
2657 *srcp = end_of_name;
2658 return key;
2659 }
2660 }
2661 }
2662 return 0;
2663}
2664
2665/*
2666 * Try to include modifiers in the key.
2667 * Changes "Shift-a" to 'A', "Alt-A" to 0xc0, etc.
2668 */
2669 int
2670extract_modifiers(key, modp)
2671 int key;
2672 int *modp;
2673{
2674 int modifiers = *modp;
2675
2676#ifdef MACOS
2677 /* Command-key really special, No fancynest */
2678 if (!(modifiers & MOD_MASK_CMD))
2679#endif
2680 if ((modifiers & MOD_MASK_SHIFT) && ASCII_ISALPHA(key))
2681 {
2682 key = TOUPPER_ASC(key);
2683 modifiers &= ~MOD_MASK_SHIFT;
2684 }
2685 if ((modifiers & MOD_MASK_CTRL)
2686#ifdef EBCDIC
2687 /* * TODO: EBCDIC Better use:
2688 * && (Ctrl_chr(key) || key == '?')
2689 * ??? */
2690 && strchr("?@ABCDEFGHIJKLMNOPQRSTUVWXYZ[\\]^_", key)
2691 != NULL
2692#else
2693 && ((key >= '?' && key <= '_') || ASCII_ISALPHA(key))
2694#endif
2695 )
2696 {
2697 key = Ctrl_chr(key);
2698 modifiers &= ~MOD_MASK_CTRL;
2699 /* <C-@> is <Nul> */
2700 if (key == 0)
2701 key = K_ZERO;
2702 }
2703#ifdef MACOS
2704 /* Command-key really special, No fancynest */
2705 if (!(modifiers & MOD_MASK_CMD))
2706#endif
2707 if ((modifiers & MOD_MASK_ALT) && key < 0x80
2708#ifdef FEAT_MBYTE
2709 && !enc_dbcs /* avoid creating a lead byte */
2710#endif
2711 )
2712 {
2713 key |= 0x80;
2714 modifiers &= ~MOD_MASK_ALT; /* remove the META modifier */
2715 }
2716
2717 *modp = modifiers;
2718 return key;
2719}
2720
2721/*
2722 * Try to find key "c" in the special key table.
2723 * Return the index when found, -1 when not found.
2724 */
2725 int
2726find_special_key_in_table(c)
2727 int c;
2728{
2729 int i;
2730
2731 for (i = 0; key_names_table[i].name != NULL; i++)
2732 if (c == key_names_table[i].key)
2733 break;
2734 if (key_names_table[i].name == NULL)
2735 i = -1;
2736 return i;
2737}
2738
2739/*
2740 * Find the special key with the given name (the given string does not have to
2741 * end with NUL, the name is assumed to end before the first non-idchar).
2742 * If the name starts with "t_" the next two characters are interpreted as a
2743 * termcap name.
2744 * Return the key code, or 0 if not found.
2745 */
2746 int
2747get_special_key_code(name)
2748 char_u *name;
2749{
2750 char_u *table_name;
2751 char_u string[3];
2752 int i, j;
2753
2754 /*
2755 * If it's <t_xx> we get the code for xx from the termcap
2756 */
2757 if (name[0] == 't' && name[1] == '_' && name[2] != NUL && name[3] != NUL)
2758 {
2759 string[0] = name[2];
2760 string[1] = name[3];
2761 string[2] = NUL;
2762 if (add_termcap_entry(string, FALSE) == OK)
2763 return TERMCAP2KEY(name[2], name[3]);
2764 }
2765 else
2766 for (i = 0; key_names_table[i].name != NULL; i++)
2767 {
2768 table_name = key_names_table[i].name;
2769 for (j = 0; vim_isIDc(name[j]) && table_name[j] != NUL; j++)
2770 if (TOLOWER_ASC(table_name[j]) != TOLOWER_ASC(name[j]))
2771 break;
2772 if (!vim_isIDc(name[j]) && table_name[j] == NUL)
2773 return key_names_table[i].key;
2774 }
2775 return 0;
2776}
2777
2778#ifdef FEAT_CMDL_COMPL
2779 char_u *
2780get_key_name(i)
2781 int i;
2782{
2783 if (i >= KEY_NAMES_TABLE_LEN)
2784 return NULL;
2785 return key_names_table[i].name;
2786}
2787#endif
2788
2789#ifdef FEAT_MOUSE
2790/*
2791 * Look up the given mouse code to return the relevant information in the other
2792 * arguments. Return which button is down or was released.
2793 */
2794 int
2795get_mouse_button(code, is_click, is_drag)
2796 int code;
2797 int *is_click;
2798 int *is_drag;
2799{
2800 int i;
2801
2802 for (i = 0; mouse_table[i].pseudo_code; i++)
2803 if (code == mouse_table[i].pseudo_code)
2804 {
2805 *is_click = mouse_table[i].is_click;
2806 *is_drag = mouse_table[i].is_drag;
2807 return mouse_table[i].button;
2808 }
2809 return 0; /* Shouldn't get here */
2810}
2811
2812/*
2813 * Return the appropriate pseudo mouse event token (KE_LEFTMOUSE etc) based on
2814 * the given information about which mouse button is down, and whether the
2815 * mouse was clicked, dragged or released.
2816 */
2817 int
2818get_pseudo_mouse_code(button, is_click, is_drag)
2819 int button; /* eg MOUSE_LEFT */
2820 int is_click;
2821 int is_drag;
2822{
2823 int i;
2824
2825 for (i = 0; mouse_table[i].pseudo_code; i++)
2826 if (button == mouse_table[i].button
2827 && is_click == mouse_table[i].is_click
2828 && is_drag == mouse_table[i].is_drag)
2829 {
2830#ifdef FEAT_GUI
Bram Moolenaarc91506a2005-04-24 22:04:21 +00002831 /* Trick: a non mappable left click and release has mouse_col -1
2832 * or added MOUSE_COLOFF. Used for 'mousefocus' in
2833 * gui_mouse_moved() */
2834 if (mouse_col < 0 || mouse_col > MOUSE_COLOFF)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002835 {
Bram Moolenaarc91506a2005-04-24 22:04:21 +00002836 if (mouse_col < 0)
2837 mouse_col = 0;
2838 else
2839 mouse_col -= MOUSE_COLOFF;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002840 if (mouse_table[i].pseudo_code == (int)KE_LEFTMOUSE)
2841 return (int)KE_LEFTMOUSE_NM;
2842 if (mouse_table[i].pseudo_code == (int)KE_LEFTRELEASE)
2843 return (int)KE_LEFTRELEASE_NM;
2844 }
2845#endif
2846 return mouse_table[i].pseudo_code;
2847 }
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00002848 return (int)KE_IGNORE; /* not recognized, ignore it */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002849}
2850#endif /* FEAT_MOUSE */
2851
2852/*
2853 * Return the current end-of-line type: EOL_DOS, EOL_UNIX or EOL_MAC.
2854 */
2855 int
2856get_fileformat(buf)
2857 buf_T *buf;
2858{
2859 int c = *buf->b_p_ff;
2860
2861 if (buf->b_p_bin || c == 'u')
2862 return EOL_UNIX;
2863 if (c == 'm')
2864 return EOL_MAC;
2865 return EOL_DOS;
2866}
2867
2868/*
2869 * Like get_fileformat(), but override 'fileformat' with "p" for "++opt=val"
2870 * argument.
2871 */
2872 int
2873get_fileformat_force(buf, eap)
2874 buf_T *buf;
2875 exarg_T *eap; /* can be NULL! */
2876{
2877 int c;
2878
2879 if (eap != NULL && eap->force_ff != 0)
2880 c = eap->cmd[eap->force_ff];
2881 else
2882 {
2883 if ((eap != NULL && eap->force_bin != 0)
2884 ? (eap->force_bin == FORCE_BIN) : buf->b_p_bin)
2885 return EOL_UNIX;
2886 c = *buf->b_p_ff;
2887 }
2888 if (c == 'u')
2889 return EOL_UNIX;
2890 if (c == 'm')
2891 return EOL_MAC;
2892 return EOL_DOS;
2893}
2894
2895/*
2896 * Set the current end-of-line type to EOL_DOS, EOL_UNIX or EOL_MAC.
2897 * Sets both 'textmode' and 'fileformat'.
2898 * Note: Does _not_ set global value of 'textmode'!
2899 */
2900 void
2901set_fileformat(t, opt_flags)
2902 int t;
2903 int opt_flags; /* OPT_LOCAL and/or OPT_GLOBAL */
2904{
2905 char *p = NULL;
2906
2907 switch (t)
2908 {
2909 case EOL_DOS:
2910 p = FF_DOS;
2911 curbuf->b_p_tx = TRUE;
2912 break;
2913 case EOL_UNIX:
2914 p = FF_UNIX;
2915 curbuf->b_p_tx = FALSE;
2916 break;
2917 case EOL_MAC:
2918 p = FF_MAC;
2919 curbuf->b_p_tx = FALSE;
2920 break;
2921 }
2922 if (p != NULL)
2923 set_string_option_direct((char_u *)"ff", -1, (char_u *)p,
Bram Moolenaar5e3cb7e2006-02-27 23:58:35 +00002924 OPT_FREE | opt_flags, 0);
2925
Bram Moolenaar071d4272004-06-13 20:20:40 +00002926#ifdef FEAT_WINDOWS
Bram Moolenaarf740b292006-02-16 22:11:02 +00002927 /* This may cause the buffer to become (un)modified. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002928 check_status(curbuf);
Bram Moolenaar997fb4b2006-02-17 21:53:23 +00002929 redraw_tabline = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002930#endif
2931#ifdef FEAT_TITLE
2932 need_maketitle = TRUE; /* set window title later */
2933#endif
2934}
2935
2936/*
2937 * Return the default fileformat from 'fileformats'.
2938 */
2939 int
2940default_fileformat()
2941{
2942 switch (*p_ffs)
2943 {
2944 case 'm': return EOL_MAC;
2945 case 'd': return EOL_DOS;
2946 }
2947 return EOL_UNIX;
2948}
2949
2950/*
2951 * Call shell. Calls mch_call_shell, with 'shellxquote' added.
2952 */
2953 int
2954call_shell(cmd, opt)
2955 char_u *cmd;
2956 int opt;
2957{
2958 char_u *ncmd;
2959 int retval;
Bram Moolenaar05159a02005-02-26 23:04:13 +00002960#ifdef FEAT_PROFILE
2961 proftime_T wait_time;
2962#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002963
2964 if (p_verbose > 3)
2965 {
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00002966 verbose_enter();
Bram Moolenaar051b7822005-05-19 21:00:46 +00002967 smsg((char_u *)_("Calling shell to execute: \"%s\""),
Bram Moolenaar071d4272004-06-13 20:20:40 +00002968 cmd == NULL ? p_sh : cmd);
2969 out_char('\n');
2970 cursor_on();
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00002971 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00002972 }
2973
Bram Moolenaar05159a02005-02-26 23:04:13 +00002974#ifdef FEAT_PROFILE
Bram Moolenaar01265852006-03-20 21:50:15 +00002975 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00002976 prof_child_enter(&wait_time);
2977#endif
2978
Bram Moolenaar071d4272004-06-13 20:20:40 +00002979 if (*p_sh == NUL)
2980 {
2981 EMSG(_(e_shellempty));
2982 retval = -1;
2983 }
2984 else
2985 {
2986#ifdef FEAT_GUI_MSWIN
2987 /* Don't hide the pointer while executing a shell command. */
2988 gui_mch_mousehide(FALSE);
2989#endif
2990#ifdef FEAT_GUI
2991 ++hold_gui_events;
2992#endif
2993 /* The external command may update a tags file, clear cached tags. */
2994 tag_freematch();
2995
2996 if (cmd == NULL || *p_sxq == NUL)
2997 retval = mch_call_shell(cmd, opt);
2998 else
2999 {
3000 ncmd = alloc((unsigned)(STRLEN(cmd) + STRLEN(p_sxq) * 2 + 1));
3001 if (ncmd != NULL)
3002 {
3003 STRCPY(ncmd, p_sxq);
3004 STRCAT(ncmd, cmd);
3005 STRCAT(ncmd, p_sxq);
3006 retval = mch_call_shell(ncmd, opt);
3007 vim_free(ncmd);
3008 }
3009 else
3010 retval = -1;
3011 }
3012#ifdef FEAT_GUI
3013 --hold_gui_events;
3014#endif
3015 /*
3016 * Check the window size, in case it changed while executing the
3017 * external command.
3018 */
3019 shell_resized_check();
3020 }
3021
3022#ifdef FEAT_EVAL
3023 set_vim_var_nr(VV_SHELL_ERROR, (long)retval);
Bram Moolenaar05159a02005-02-26 23:04:13 +00003024# ifdef FEAT_PROFILE
Bram Moolenaar01265852006-03-20 21:50:15 +00003025 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003026 prof_child_exit(&wait_time);
3027# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003028#endif
3029
3030 return retval;
3031}
3032
3033/*
Bram Moolenaar01265852006-03-20 21:50:15 +00003034 * VISUAL, SELECTMODE and OP_PENDING State are never set, they are equal to
3035 * NORMAL State with a condition. This function returns the real State.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003036 */
3037 int
3038get_real_state()
3039{
3040 if (State & NORMAL)
3041 {
3042#ifdef FEAT_VISUAL
3043 if (VIsual_active)
Bram Moolenaar01265852006-03-20 21:50:15 +00003044 {
3045 if (VIsual_select)
3046 return SELECTMODE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003047 return VISUAL;
Bram Moolenaar01265852006-03-20 21:50:15 +00003048 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003049 else
3050#endif
3051 if (finish_op)
Bram Moolenaar01265852006-03-20 21:50:15 +00003052 return OP_PENDING;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003053 }
3054 return State;
3055}
3056
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003057#if defined(FEAT_MBYTE) || defined(PROTO)
3058/*
3059 * Return TRUE if "p" points to just after a path separator.
3060 * Take care of multi-byte characters.
3061 * "b" must point to the start of the file name
3062 */
3063 int
3064after_pathsep(b, p)
3065 char_u *b;
3066 char_u *p;
3067{
3068 return vim_ispathsep(p[-1])
3069 && (!has_mbyte || (*mb_head_off)(b, p - 1) == 0);
3070}
3071#endif
3072
3073/*
3074 * Return TRUE if file names "f1" and "f2" are in the same directory.
3075 * "f1" may be a short name, "f2" must be a full path.
3076 */
3077 int
3078same_directory(f1, f2)
3079 char_u *f1;
3080 char_u *f2;
3081{
3082 char_u ffname[MAXPATHL];
3083 char_u *t1;
3084 char_u *t2;
3085
3086 /* safety check */
3087 if (f1 == NULL || f2 == NULL)
3088 return FALSE;
3089
3090 (void)vim_FullName(f1, ffname, MAXPATHL, FALSE);
3091 t1 = gettail_sep(ffname);
3092 t2 = gettail_sep(f2);
3093 return (t1 - ffname == t2 - f2
3094 && pathcmp((char *)ffname, (char *)f2, (int)(t1 - ffname)) == 0);
3095}
3096
Bram Moolenaar071d4272004-06-13 20:20:40 +00003097#if defined(FEAT_SESSION) || defined(MSWIN) || defined(FEAT_GUI_MAC) \
Bram Moolenaar9372a112005-12-06 19:59:18 +00003098 || ((defined(FEAT_GUI_GTK)) \
Bram Moolenaar843ee412004-06-30 16:16:41 +00003099 && ( defined(FEAT_WINDOWS) || defined(FEAT_DND)) ) \
Bram Moolenaar071d4272004-06-13 20:20:40 +00003100 || defined(FEAT_SUN_WORKSHOP) || defined(FEAT_NETBEANS_INTG) \
3101 || defined(PROTO)
3102/*
3103 * Change to a file's directory.
3104 * Caller must call shorten_fnames()!
3105 * Return OK or FAIL.
3106 */
3107 int
3108vim_chdirfile(fname)
3109 char_u *fname;
3110{
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003111 char_u dir[MAXPATHL];
Bram Moolenaar071d4272004-06-13 20:20:40 +00003112
Bram Moolenaarbbebc852005-07-18 21:47:53 +00003113 vim_strncpy(dir, fname, MAXPATHL - 1);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003114 *gettail_sep(dir) = NUL;
3115 return mch_chdir((char *)dir) == 0 ? OK : FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003116}
3117#endif
3118
3119#if defined(STAT_IGNORES_SLASH) || defined(PROTO)
3120/*
3121 * Check if "name" ends in a slash and is not a directory.
3122 * Used for systems where stat() ignores a trailing slash on a file name.
3123 * The Vim code assumes a trailing slash is only ignored for a directory.
3124 */
3125 int
3126illegal_slash(name)
3127 char *name;
3128{
3129 if (name[0] == NUL)
3130 return FALSE; /* no file name is not illegal */
3131 if (name[strlen(name) - 1] != '/')
3132 return FALSE; /* no trailing slash */
3133 if (mch_isdir((char_u *)name))
3134 return FALSE; /* trailing slash for a directory */
3135 return TRUE;
3136}
3137#endif
3138
3139#if defined(CURSOR_SHAPE) || defined(PROTO)
3140
3141/*
3142 * Handling of cursor and mouse pointer shapes in various modes.
3143 */
3144
3145cursorentry_T shape_table[SHAPE_IDX_COUNT] =
3146{
3147 /* The values will be filled in from the 'guicursor' and 'mouseshape'
3148 * defaults when Vim starts.
3149 * Adjust the SHAPE_IDX_ defines when making changes! */
3150 {0, 0, 0, 700L, 400L, 250L, 0, 0, "n", SHAPE_CURSOR+SHAPE_MOUSE},
3151 {0, 0, 0, 700L, 400L, 250L, 0, 0, "v", SHAPE_CURSOR+SHAPE_MOUSE},
3152 {0, 0, 0, 700L, 400L, 250L, 0, 0, "i", SHAPE_CURSOR+SHAPE_MOUSE},
3153 {0, 0, 0, 700L, 400L, 250L, 0, 0, "r", SHAPE_CURSOR+SHAPE_MOUSE},
3154 {0, 0, 0, 700L, 400L, 250L, 0, 0, "c", SHAPE_CURSOR+SHAPE_MOUSE},
3155 {0, 0, 0, 700L, 400L, 250L, 0, 0, "ci", SHAPE_CURSOR+SHAPE_MOUSE},
3156 {0, 0, 0, 700L, 400L, 250L, 0, 0, "cr", SHAPE_CURSOR+SHAPE_MOUSE},
3157 {0, 0, 0, 700L, 400L, 250L, 0, 0, "o", SHAPE_CURSOR+SHAPE_MOUSE},
3158 {0, 0, 0, 700L, 400L, 250L, 0, 0, "ve", SHAPE_CURSOR+SHAPE_MOUSE},
3159 {0, 0, 0, 0L, 0L, 0L, 0, 0, "e", SHAPE_MOUSE},
3160 {0, 0, 0, 0L, 0L, 0L, 0, 0, "s", SHAPE_MOUSE},
3161 {0, 0, 0, 0L, 0L, 0L, 0, 0, "sd", SHAPE_MOUSE},
3162 {0, 0, 0, 0L, 0L, 0L, 0, 0, "vs", SHAPE_MOUSE},
3163 {0, 0, 0, 0L, 0L, 0L, 0, 0, "vd", SHAPE_MOUSE},
3164 {0, 0, 0, 0L, 0L, 0L, 0, 0, "m", SHAPE_MOUSE},
3165 {0, 0, 0, 0L, 0L, 0L, 0, 0, "ml", SHAPE_MOUSE},
3166 {0, 0, 0, 100L, 100L, 100L, 0, 0, "sm", SHAPE_CURSOR},
3167};
3168
3169#ifdef FEAT_MOUSESHAPE
3170/*
3171 * Table with names for mouse shapes. Keep in sync with all the tables for
3172 * mch_set_mouse_shape()!.
3173 */
3174static char * mshape_names[] =
3175{
3176 "arrow", /* default, must be the first one */
3177 "blank", /* hidden */
3178 "beam",
3179 "updown",
3180 "udsizing",
3181 "leftright",
3182 "lrsizing",
3183 "busy",
3184 "no",
3185 "crosshair",
3186 "hand1",
3187 "hand2",
3188 "pencil",
3189 "question",
3190 "rightup-arrow",
3191 "up-arrow",
3192 NULL
3193};
3194#endif
3195
3196/*
3197 * Parse the 'guicursor' option ("what" is SHAPE_CURSOR) or 'mouseshape'
3198 * ("what" is SHAPE_MOUSE).
3199 * Returns error message for an illegal option, NULL otherwise.
3200 */
3201 char_u *
3202parse_shape_opt(what)
3203 int what;
3204{
3205 char_u *modep;
3206 char_u *colonp;
3207 char_u *commap;
3208 char_u *slashp;
3209 char_u *p, *endp;
3210 int idx = 0; /* init for GCC */
3211 int all_idx;
3212 int len;
3213 int i;
3214 long n;
3215 int found_ve = FALSE; /* found "ve" flag */
3216 int round;
3217
3218 /*
3219 * First round: check for errors; second round: do it for real.
3220 */
3221 for (round = 1; round <= 2; ++round)
3222 {
3223 /*
3224 * Repeat for all comma separated parts.
3225 */
3226#ifdef FEAT_MOUSESHAPE
3227 if (what == SHAPE_MOUSE)
3228 modep = p_mouseshape;
3229 else
3230#endif
3231 modep = p_guicursor;
3232 while (*modep != NUL)
3233 {
3234 colonp = vim_strchr(modep, ':');
3235 if (colonp == NULL)
3236 return (char_u *)N_("E545: Missing colon");
3237 if (colonp == modep)
3238 return (char_u *)N_("E546: Illegal mode");
3239 commap = vim_strchr(modep, ',');
3240
3241 /*
3242 * Repeat for all mode's before the colon.
3243 * For the 'a' mode, we loop to handle all the modes.
3244 */
3245 all_idx = -1;
3246 while (modep < colonp || all_idx >= 0)
3247 {
3248 if (all_idx < 0)
3249 {
3250 /* Find the mode. */
3251 if (modep[1] == '-' || modep[1] == ':')
3252 len = 1;
3253 else
3254 len = 2;
3255 if (len == 1 && TOLOWER_ASC(modep[0]) == 'a')
3256 all_idx = SHAPE_IDX_COUNT - 1;
3257 else
3258 {
3259 for (idx = 0; idx < SHAPE_IDX_COUNT; ++idx)
3260 if (STRNICMP(modep, shape_table[idx].name, len)
3261 == 0)
3262 break;
3263 if (idx == SHAPE_IDX_COUNT
3264 || (shape_table[idx].used_for & what) == 0)
3265 return (char_u *)N_("E546: Illegal mode");
3266 if (len == 2 && modep[0] == 'v' && modep[1] == 'e')
3267 found_ve = TRUE;
3268 }
3269 modep += len + 1;
3270 }
3271
3272 if (all_idx >= 0)
3273 idx = all_idx--;
3274 else if (round == 2)
3275 {
3276#ifdef FEAT_MOUSESHAPE
3277 if (what == SHAPE_MOUSE)
3278 {
3279 /* Set the default, for the missing parts */
3280 shape_table[idx].mshape = 0;
3281 }
3282 else
3283#endif
3284 {
3285 /* Set the defaults, for the missing parts */
3286 shape_table[idx].shape = SHAPE_BLOCK;
3287 shape_table[idx].blinkwait = 700L;
3288 shape_table[idx].blinkon = 400L;
3289 shape_table[idx].blinkoff = 250L;
3290 }
3291 }
3292
3293 /* Parse the part after the colon */
3294 for (p = colonp + 1; *p && *p != ','; )
3295 {
3296#ifdef FEAT_MOUSESHAPE
3297 if (what == SHAPE_MOUSE)
3298 {
3299 for (i = 0; ; ++i)
3300 {
3301 if (mshape_names[i] == NULL)
3302 {
3303 if (!VIM_ISDIGIT(*p))
3304 return (char_u *)N_("E547: Illegal mouseshape");
3305 if (round == 2)
3306 shape_table[idx].mshape =
3307 getdigits(&p) + MSHAPE_NUMBERED;
3308 else
3309 (void)getdigits(&p);
3310 break;
3311 }
3312 len = (int)STRLEN(mshape_names[i]);
3313 if (STRNICMP(p, mshape_names[i], len) == 0)
3314 {
3315 if (round == 2)
3316 shape_table[idx].mshape = i;
3317 p += len;
3318 break;
3319 }
3320 }
3321 }
3322 else /* if (what == SHAPE_MOUSE) */
3323#endif
3324 {
3325 /*
3326 * First handle the ones with a number argument.
3327 */
3328 i = *p;
3329 len = 0;
3330 if (STRNICMP(p, "ver", 3) == 0)
3331 len = 3;
3332 else if (STRNICMP(p, "hor", 3) == 0)
3333 len = 3;
3334 else if (STRNICMP(p, "blinkwait", 9) == 0)
3335 len = 9;
3336 else if (STRNICMP(p, "blinkon", 7) == 0)
3337 len = 7;
3338 else if (STRNICMP(p, "blinkoff", 8) == 0)
3339 len = 8;
3340 if (len != 0)
3341 {
3342 p += len;
3343 if (!VIM_ISDIGIT(*p))
3344 return (char_u *)N_("E548: digit expected");
3345 n = getdigits(&p);
3346 if (len == 3) /* "ver" or "hor" */
3347 {
3348 if (n == 0)
3349 return (char_u *)N_("E549: Illegal percentage");
3350 if (round == 2)
3351 {
3352 if (TOLOWER_ASC(i) == 'v')
3353 shape_table[idx].shape = SHAPE_VER;
3354 else
3355 shape_table[idx].shape = SHAPE_HOR;
3356 shape_table[idx].percentage = n;
3357 }
3358 }
3359 else if (round == 2)
3360 {
3361 if (len == 9)
3362 shape_table[idx].blinkwait = n;
3363 else if (len == 7)
3364 shape_table[idx].blinkon = n;
3365 else
3366 shape_table[idx].blinkoff = n;
3367 }
3368 }
3369 else if (STRNICMP(p, "block", 5) == 0)
3370 {
3371 if (round == 2)
3372 shape_table[idx].shape = SHAPE_BLOCK;
3373 p += 5;
3374 }
3375 else /* must be a highlight group name then */
3376 {
3377 endp = vim_strchr(p, '-');
3378 if (commap == NULL) /* last part */
3379 {
3380 if (endp == NULL)
3381 endp = p + STRLEN(p); /* find end of part */
3382 }
3383 else if (endp > commap || endp == NULL)
3384 endp = commap;
3385 slashp = vim_strchr(p, '/');
3386 if (slashp != NULL && slashp < endp)
3387 {
3388 /* "group/langmap_group" */
3389 i = syn_check_group(p, (int)(slashp - p));
3390 p = slashp + 1;
3391 }
3392 if (round == 2)
3393 {
3394 shape_table[idx].id = syn_check_group(p,
3395 (int)(endp - p));
3396 shape_table[idx].id_lm = shape_table[idx].id;
3397 if (slashp != NULL && slashp < endp)
3398 shape_table[idx].id = i;
3399 }
3400 p = endp;
3401 }
3402 } /* if (what != SHAPE_MOUSE) */
3403
3404 if (*p == '-')
3405 ++p;
3406 }
3407 }
3408 modep = p;
3409 if (*modep == ',')
3410 ++modep;
3411 }
3412 }
3413
3414 /* If the 's' flag is not given, use the 'v' cursor for 's' */
3415 if (!found_ve)
3416 {
3417#ifdef FEAT_MOUSESHAPE
3418 if (what == SHAPE_MOUSE)
3419 {
3420 shape_table[SHAPE_IDX_VE].mshape = shape_table[SHAPE_IDX_V].mshape;
3421 }
3422 else
3423#endif
3424 {
3425 shape_table[SHAPE_IDX_VE].shape = shape_table[SHAPE_IDX_V].shape;
3426 shape_table[SHAPE_IDX_VE].percentage =
3427 shape_table[SHAPE_IDX_V].percentage;
3428 shape_table[SHAPE_IDX_VE].blinkwait =
3429 shape_table[SHAPE_IDX_V].blinkwait;
3430 shape_table[SHAPE_IDX_VE].blinkon =
3431 shape_table[SHAPE_IDX_V].blinkon;
3432 shape_table[SHAPE_IDX_VE].blinkoff =
3433 shape_table[SHAPE_IDX_V].blinkoff;
3434 shape_table[SHAPE_IDX_VE].id = shape_table[SHAPE_IDX_V].id;
3435 shape_table[SHAPE_IDX_VE].id_lm = shape_table[SHAPE_IDX_V].id_lm;
3436 }
3437 }
3438
3439 return NULL;
3440}
3441
Bram Moolenaarac6e65f2005-08-29 22:25:38 +00003442# if defined(MCH_CURSOR_SHAPE) || defined(FEAT_GUI) \
3443 || defined(FEAT_MOUSESHAPE) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003444/*
3445 * Return the index into shape_table[] for the current mode.
3446 * When "mouse" is TRUE, consider indexes valid for the mouse pointer.
3447 */
3448 int
3449get_shape_idx(mouse)
3450 int mouse;
3451{
3452#ifdef FEAT_MOUSESHAPE
3453 if (mouse && (State == HITRETURN || State == ASKMORE))
3454 {
3455# ifdef FEAT_GUI
Bram Moolenaar9588a0f2005-01-08 21:45:39 +00003456 int x, y;
3457 gui_mch_getmouse(&x, &y);
3458 if (Y_2_ROW(y) == Rows - 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003459 return SHAPE_IDX_MOREL;
3460# endif
3461 return SHAPE_IDX_MORE;
3462 }
3463 if (mouse && drag_status_line)
3464 return SHAPE_IDX_SDRAG;
3465# ifdef FEAT_VERTSPLIT
3466 if (mouse && drag_sep_line)
3467 return SHAPE_IDX_VDRAG;
3468# endif
3469#endif
3470 if (!mouse && State == SHOWMATCH)
3471 return SHAPE_IDX_SM;
3472#ifdef FEAT_VREPLACE
3473 if (State & VREPLACE_FLAG)
3474 return SHAPE_IDX_R;
3475#endif
3476 if (State & REPLACE_FLAG)
3477 return SHAPE_IDX_R;
3478 if (State & INSERT)
3479 return SHAPE_IDX_I;
3480 if (State & CMDLINE)
3481 {
3482 if (cmdline_at_end())
3483 return SHAPE_IDX_C;
3484 if (cmdline_overstrike())
3485 return SHAPE_IDX_CR;
3486 return SHAPE_IDX_CI;
3487 }
3488 if (finish_op)
3489 return SHAPE_IDX_O;
3490#ifdef FEAT_VISUAL
3491 if (VIsual_active)
3492 {
3493 if (*p_sel == 'e')
3494 return SHAPE_IDX_VE;
3495 else
3496 return SHAPE_IDX_V;
3497 }
3498#endif
3499 return SHAPE_IDX_N;
3500}
Bram Moolenaarac6e65f2005-08-29 22:25:38 +00003501#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003502
3503# if defined(FEAT_MOUSESHAPE) || defined(PROTO)
3504static int old_mouse_shape = 0;
3505
3506/*
3507 * Set the mouse shape:
3508 * If "shape" is -1, use shape depending on the current mode,
3509 * depending on the current state.
3510 * If "shape" is -2, only update the shape when it's CLINE or STATUS (used
3511 * when the mouse moves off the status or command line).
3512 */
3513 void
3514update_mouseshape(shape_idx)
3515 int shape_idx;
3516{
3517 int new_mouse_shape;
3518
3519 /* Only works in GUI mode. */
Bram Moolenaar6bb68362005-03-22 23:03:44 +00003520 if (!gui.in_use || gui.starting)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003521 return;
3522
3523 /* Postpone the updating when more is to come. Speeds up executing of
3524 * mappings. */
3525 if (shape_idx == -1 && char_avail())
3526 {
3527 postponed_mouseshape = TRUE;
3528 return;
3529 }
3530
Bram Moolenaar14716812006-05-04 21:54:08 +00003531 /* When ignoring the mouse don't change shape on the statusline. */
3532 if (*p_mouse == NUL
3533 && (shape_idx == SHAPE_IDX_CLINE
3534 || shape_idx == SHAPE_IDX_STATUS
3535 || shape_idx == SHAPE_IDX_VSEP))
3536 shape_idx = -2;
3537
Bram Moolenaar071d4272004-06-13 20:20:40 +00003538 if (shape_idx == -2
3539 && old_mouse_shape != shape_table[SHAPE_IDX_CLINE].mshape
3540 && old_mouse_shape != shape_table[SHAPE_IDX_STATUS].mshape
3541 && old_mouse_shape != shape_table[SHAPE_IDX_VSEP].mshape)
3542 return;
3543 if (shape_idx < 0)
3544 new_mouse_shape = shape_table[get_shape_idx(TRUE)].mshape;
3545 else
3546 new_mouse_shape = shape_table[shape_idx].mshape;
3547 if (new_mouse_shape != old_mouse_shape)
3548 {
3549 mch_set_mouse_shape(new_mouse_shape);
3550 old_mouse_shape = new_mouse_shape;
3551 }
3552 postponed_mouseshape = FALSE;
3553}
3554# endif
3555
3556#endif /* CURSOR_SHAPE */
3557
3558
3559#ifdef FEAT_CRYPT
3560/*
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00003561 * Optional encryption support.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003562 * Mohsin Ahmed, mosh@sasi.com, 98-09-24
3563 * Based on zip/crypt sources.
3564 *
3565 * NOTE FOR USA: Since 2000 exporting this code from the USA is allowed to
3566 * most countries. There are a few exceptions, but that still should not be a
3567 * problem since this code was originally created in Europe and India.
3568 */
3569
3570/* from zip.h */
3571
3572typedef unsigned short ush; /* unsigned 16-bit value */
3573typedef unsigned long ulg; /* unsigned 32-bit value */
3574
3575static void make_crc_tab __ARGS((void));
3576
Bram Moolenaard6f676d2005-06-01 21:51:55 +00003577static ulg crc_32_tab[256];
Bram Moolenaar071d4272004-06-13 20:20:40 +00003578
3579/*
3580 * Fill the CRC table.
3581 */
3582 static void
3583make_crc_tab()
3584{
3585 ulg s,t,v;
3586 static int done = FALSE;
3587
3588 if (done)
3589 return;
3590 for (t = 0; t < 256; t++)
3591 {
3592 v = t;
3593 for (s = 0; s < 8; s++)
3594 v = (v >> 1) ^ ((v & 1) * (ulg)0xedb88320L);
3595 crc_32_tab[t] = v;
3596 }
3597 done = TRUE;
3598}
3599
3600#define CRC32(c, b) (crc_32_tab[((int)(c) ^ (b)) & 0xff] ^ ((c) >> 8))
3601
3602
3603static ulg keys[3]; /* keys defining the pseudo-random sequence */
3604
3605/*
3606 * Return the next byte in the pseudo-random sequence
3607 */
3608 int
3609decrypt_byte()
3610{
3611 ush temp;
3612
3613 temp = (ush)keys[2] | 2;
3614 return (int)(((unsigned)(temp * (temp ^ 1)) >> 8) & 0xff);
3615}
3616
3617/*
3618 * Update the encryption keys with the next byte of plain text
3619 */
3620 int
3621update_keys(c)
3622 int c; /* byte of plain text */
3623{
3624 keys[0] = CRC32(keys[0], c);
3625 keys[1] += keys[0] & 0xff;
3626 keys[1] = keys[1] * 134775813L + 1;
3627 keys[2] = CRC32(keys[2], (int)(keys[1] >> 24));
3628 return c;
3629}
3630
3631/*
3632 * Initialize the encryption keys and the random header according to
3633 * the given password.
3634 * If "passwd" is NULL or empty, don't do anything.
3635 */
3636 void
3637crypt_init_keys(passwd)
3638 char_u *passwd; /* password string with which to modify keys */
3639{
3640 if (passwd != NULL && *passwd != NUL)
3641 {
3642 make_crc_tab();
3643 keys[0] = 305419896L;
3644 keys[1] = 591751049L;
3645 keys[2] = 878082192L;
3646 while (*passwd != '\0')
3647 update_keys((int)*passwd++);
3648 }
3649}
3650
3651/*
3652 * Ask the user for a crypt key.
3653 * When "store" is TRUE, the new key in stored in the 'key' option, and the
3654 * 'key' option value is returned: Don't free it.
3655 * When "store" is FALSE, the typed key is returned in allocated memory.
3656 * Returns NULL on failure.
3657 */
3658 char_u *
3659get_crypt_key(store, twice)
3660 int store;
3661 int twice; /* Ask for the key twice. */
3662{
3663 char_u *p1, *p2 = NULL;
3664 int round;
3665
3666 for (round = 0; ; ++round)
3667 {
3668 cmdline_star = TRUE;
3669 cmdline_row = msg_row;
3670 p1 = getcmdline_prompt(NUL, round == 0
3671 ? (char_u *)_("Enter encryption key: ")
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +00003672 : (char_u *)_("Enter same key again: "), 0, EXPAND_NOTHING,
3673 NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003674 cmdline_star = FALSE;
3675
3676 if (p1 == NULL)
3677 break;
3678
3679 if (round == twice)
3680 {
3681 if (p2 != NULL && STRCMP(p1, p2) != 0)
3682 {
3683 MSG(_("Keys don't match!"));
3684 vim_free(p1);
3685 vim_free(p2);
3686 p2 = NULL;
3687 round = -1; /* do it again */
3688 continue;
3689 }
3690 if (store)
3691 {
3692 set_option_value((char_u *)"key", 0L, p1, OPT_LOCAL);
3693 vim_free(p1);
3694 p1 = curbuf->b_p_key;
3695 }
3696 break;
3697 }
3698 p2 = p1;
3699 }
3700
3701 /* since the user typed this, no need to wait for return */
3702 need_wait_return = FALSE;
3703 msg_didout = FALSE;
3704
3705 vim_free(p2);
3706 return p1;
3707}
3708
3709#endif /* FEAT_CRYPT */
3710
3711/* TODO: make some #ifdef for this */
3712/*--------[ file searching ]-------------------------------------------------*/
3713/*
3714 * File searching functions for 'path', 'tags' and 'cdpath' options.
3715 * External visible functions:
3716 * vim_findfile_init() creates/initialises the search context
3717 * vim_findfile_free_visited() free list of visited files/dirs of search
3718 * context
3719 * vim_findfile() find a file in the search context
3720 * vim_findfile_cleanup() cleanup/free search context created by
3721 * vim_findfile_init()
3722 *
3723 * All static functions and variables start with 'ff_'
3724 *
3725 * In general it works like this:
3726 * First you create yourself a search context by calling vim_findfile_init().
3727 * It is possible to give a search context from a previous call to
3728 * vim_findfile_init(), so it can be reused. After this you call vim_findfile()
3729 * until you are satisfied with the result or it returns NULL. On every call it
3730 * returns the next file which matches the conditions given to
3731 * vim_findfile_init(). If it doesn't find a next file it returns NULL.
3732 *
3733 * It is possible to call vim_findfile_init() again to reinitialise your search
3734 * with some new parameters. Don't forget to pass your old search context to
3735 * it, so it can reuse it and especially reuse the list of already visited
3736 * directories. If you want to delete the list of already visited directories
3737 * simply call vim_findfile_free_visited().
3738 *
3739 * When you are done call vim_findfile_cleanup() to free the search context.
3740 *
3741 * The function vim_findfile_init() has a long comment, which describes the
3742 * needed parameters.
3743 *
3744 *
3745 *
3746 * ATTENTION:
3747 * ==========
Bram Moolenaar77f66d62007-02-20 02:16:18 +00003748 * Also we use an allocated search context here, this functions are NOT
Bram Moolenaar071d4272004-06-13 20:20:40 +00003749 * thread-safe!!!!!
3750 *
3751 * To minimize parameter passing (or because I'm to lazy), only the
3752 * external visible functions get a search context as a parameter. This is
3753 * then assigned to a static global, which is used throughout the local
3754 * functions.
3755 */
3756
3757/*
3758 * type for the directory search stack
3759 */
3760typedef struct ff_stack
3761{
3762 struct ff_stack *ffs_prev;
3763
3764 /* the fix part (no wildcards) and the part containing the wildcards
3765 * of the search path
3766 */
3767 char_u *ffs_fix_path;
3768#ifdef FEAT_PATH_EXTRA
3769 char_u *ffs_wc_path;
3770#endif
3771
3772 /* files/dirs found in the above directory, matched by the first wildcard
3773 * of wc_part
3774 */
3775 char_u **ffs_filearray;
3776 int ffs_filearray_size;
3777 char_u ffs_filearray_cur; /* needed for partly handled dirs */
3778
3779 /* to store status of partly handled directories
3780 * 0: we work the on this directory for the first time
3781 * 1: this directory was partly searched in an earlier step
3782 */
3783 int ffs_stage;
3784
3785 /* How deep are we in the directory tree?
3786 * Counts backward from value of level parameter to vim_findfile_init
3787 */
3788 int ffs_level;
3789
3790 /* Did we already expand '**' to an empty string? */
3791 int ffs_star_star_empty;
3792} ff_stack_T;
3793
3794/*
3795 * type for already visited directories or files.
3796 */
3797typedef struct ff_visited
3798{
3799 struct ff_visited *ffv_next;
3800
3801#ifdef FEAT_PATH_EXTRA
3802 /* Visited directories are different if the wildcard string are
3803 * different. So we have to save it.
3804 */
3805 char_u *ffv_wc_path;
3806#endif
3807 /* for unix use inode etc for comparison (needed because of links), else
3808 * use filename.
3809 */
3810#ifdef UNIX
3811 int ffv_dev; /* device number (-1 if not set) */
3812 ino_t ffv_ino; /* inode number */
3813#endif
3814 /* The memory for this struct is allocated according to the length of
3815 * ffv_fname.
3816 */
3817 char_u ffv_fname[1]; /* actually longer */
3818} ff_visited_T;
3819
3820/*
3821 * We might have to manage several visited lists during a search.
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00003822 * This is especially needed for the tags option. If tags is set to:
Bram Moolenaar071d4272004-06-13 20:20:40 +00003823 * "./++/tags,./++/TAGS,++/tags" (replace + with *)
3824 * So we have to do 3 searches:
3825 * 1) search from the current files directory downward for the file "tags"
3826 * 2) search from the current files directory downward for the file "TAGS"
3827 * 3) search from Vims current directory downwards for the file "tags"
3828 * As you can see, the first and the third search are for the same file, so for
3829 * the third search we can use the visited list of the first search. For the
3830 * second search we must start from a empty visited list.
3831 * The struct ff_visited_list_hdr is used to manage a linked list of already
3832 * visited lists.
3833 */
3834typedef struct ff_visited_list_hdr
3835{
3836 struct ff_visited_list_hdr *ffvl_next;
3837
3838 /* the filename the attached visited list is for */
3839 char_u *ffvl_filename;
3840
3841 ff_visited_T *ffvl_visited_list;
3842
3843} ff_visited_list_hdr_T;
3844
3845
3846/*
3847 * '**' can be expanded to several directory levels.
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00003848 * Set the default maximum depth.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003849 */
3850#define FF_MAX_STAR_STAR_EXPAND ((char_u)30)
3851/*
3852 * The search context:
3853 * ffsc_stack_ptr: the stack for the dirs to search
3854 * ffsc_visited_list: the currently active visited list
3855 * ffsc_dir_visited_list: the currently active visited list for search dirs
3856 * ffsc_visited_lists_list: the list of all visited lists
3857 * ffsc_dir_visited_lists_list: the list of all visited lists for search dirs
3858 * ffsc_file_to_search: the file to search for
3859 * ffsc_start_dir: the starting directory, if search path was relative
3860 * ffsc_fix_path: the fix part of the given path (without wildcards)
3861 * Needed for upward search.
3862 * ffsc_wc_path: the part of the given path containing wildcards
3863 * ffsc_level: how many levels of dirs to search downwards
3864 * ffsc_stopdirs_v: array of stop directories for upward search
3865 * ffsc_need_dir: TRUE if we search for a directory
3866 */
3867typedef struct ff_search_ctx_T
3868{
3869 ff_stack_T *ffsc_stack_ptr;
3870 ff_visited_list_hdr_T *ffsc_visited_list;
3871 ff_visited_list_hdr_T *ffsc_dir_visited_list;
3872 ff_visited_list_hdr_T *ffsc_visited_lists_list;
3873 ff_visited_list_hdr_T *ffsc_dir_visited_lists_list;
3874 char_u *ffsc_file_to_search;
3875 char_u *ffsc_start_dir;
3876 char_u *ffsc_fix_path;
3877#ifdef FEAT_PATH_EXTRA
3878 char_u *ffsc_wc_path;
3879 int ffsc_level;
3880 char_u **ffsc_stopdirs_v;
3881#endif
3882 int ffsc_need_dir;
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00003883} ff_search_ctx_T;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003884
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00003885static ff_search_ctx_T *ff_search_ctx = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003886
3887/* locally needed functions */
3888#ifdef FEAT_PATH_EXTRA
3889static int ff_check_visited __ARGS((ff_visited_T **, char_u *, char_u *));
3890#else
3891static int ff_check_visited __ARGS((ff_visited_T **, char_u *));
3892#endif
3893static void vim_findfile_free_visited_list __ARGS((ff_visited_list_hdr_T **list_headp));
3894static void ff_free_visited_list __ARGS((ff_visited_T *vl));
3895static ff_visited_list_hdr_T* ff_get_visited_list __ARGS((char_u *, ff_visited_list_hdr_T **list_headp));
3896#ifdef FEAT_PATH_EXTRA
3897static int ff_wc_equal __ARGS((char_u *s1, char_u *s2));
3898#endif
3899
3900static void ff_push __ARGS((ff_stack_T *));
3901static ff_stack_T * ff_pop __ARGS((void));
3902static void ff_clear __ARGS((void));
3903static void ff_free_stack_element __ARGS((ff_stack_T *));
3904#ifdef FEAT_PATH_EXTRA
3905static ff_stack_T *ff_create_stack_element __ARGS((char_u *, char_u *, int, int));
3906#else
3907static ff_stack_T *ff_create_stack_element __ARGS((char_u *, int, int));
3908#endif
3909#ifdef FEAT_PATH_EXTRA
3910static int ff_path_in_stoplist __ARGS((char_u *, int, char_u **));
3911#endif
3912
Bram Moolenaar071d4272004-06-13 20:20:40 +00003913#if 0
3914/*
3915 * if someone likes findfirst/findnext, here are the functions
3916 * NOT TESTED!!
3917 */
3918
3919static void *ff_fn_search_context = NULL;
3920
3921 char_u *
3922vim_findfirst(path, filename, level)
3923 char_u *path;
3924 char_u *filename;
3925 int level;
3926{
3927 ff_fn_search_context =
3928 vim_findfile_init(path, filename, NULL, level, TRUE, FALSE,
3929 ff_fn_search_context, rel_fname);
3930 if (NULL == ff_fn_search_context)
3931 return NULL;
3932 else
3933 return vim_findnext()
3934}
3935
3936 char_u *
3937vim_findnext()
3938{
3939 char_u *ret = vim_findfile(ff_fn_search_context);
3940
3941 if (NULL == ret)
3942 {
3943 vim_findfile_cleanup(ff_fn_search_context);
3944 ff_fn_search_context = NULL;
3945 }
3946 return ret;
3947}
3948#endif
3949
3950/*
3951 * Initialization routine for vim_findfile.
3952 *
3953 * Returns the newly allocated search context or NULL if an error occured.
3954 *
3955 * Don't forget to clean up by calling vim_findfile_cleanup() if you are done
3956 * with the search context.
3957 *
3958 * Find the file 'filename' in the directory 'path'.
3959 * The parameter 'path' may contain wildcards. If so only search 'level'
3960 * directories deep. The parameter 'level' is the absolute maximum and is
3961 * not related to restricts given to the '**' wildcard. If 'level' is 100
3962 * and you use '**200' vim_findfile() will stop after 100 levels.
3963 *
3964 * If 'stopdirs' is not NULL and nothing is found downward, the search is
3965 * restarted on the next higher directory level. This is repeated until the
3966 * start-directory of a search is contained in 'stopdirs'. 'stopdirs' has the
3967 * format ";*<dirname>*\(;<dirname>\)*;\=$".
3968 *
3969 * If the 'path' is relative, the starting dir for the search is either VIM's
3970 * current dir or if the path starts with "./" the current files dir.
3971 * If the 'path' is absolut, the starting dir is that part of the path before
3972 * the first wildcard.
3973 *
3974 * Upward search is only done on the starting dir.
3975 *
3976 * If 'free_visited' is TRUE the list of already visited files/directories is
3977 * cleared. Set this to FALSE if you just want to search from another
3978 * directory, but want to be sure that no directory from a previous search is
3979 * searched again. This is useful if you search for a file at different places.
3980 * The list of visited files/dirs can also be cleared with the function
3981 * vim_findfile_free_visited().
3982 *
3983 * Set the parameter 'need_dir' to TRUE if you want to search for a directory
3984 * instead of a file.
3985 *
3986 * A search context returned by a previous call to vim_findfile_init() can be
3987 * passed in the parameter 'search_ctx'. This context is than reused and
3988 * reinitialized with the new parameters. The list of already viseted
3989 * directories from this context is only deleted if the parameter
3990 * 'free_visited' is true. Be aware that the passed search_context is freed if
3991 * the reinitialization fails.
3992 *
3993 * If you don't have a search context from a previous call 'search_ctx' must be
3994 * NULL.
3995 *
3996 * This function silently ignores a few errors, vim_findfile() will have
3997 * limited functionality then.
3998 */
3999/*ARGSUSED*/
4000 void *
4001vim_findfile_init(path, filename, stopdirs, level, free_visited, need_dir,
4002 search_ctx, tagfile, rel_fname)
4003 char_u *path;
4004 char_u *filename;
4005 char_u *stopdirs;
4006 int level;
4007 int free_visited;
4008 int need_dir;
4009 void *search_ctx;
4010 int tagfile;
4011 char_u *rel_fname; /* file name to use for "." */
4012{
4013#ifdef FEAT_PATH_EXTRA
4014 char_u *wc_part;
4015#endif
4016 ff_stack_T *sptr;
4017
4018 /* If a search context is given by the caller, reuse it, else allocate a
4019 * new one.
4020 */
4021 if (search_ctx != NULL)
4022 ff_search_ctx = search_ctx;
4023 else
4024 {
4025 ff_search_ctx = (ff_search_ctx_T*)alloc(
4026 (unsigned)sizeof(ff_search_ctx_T));
4027 if (ff_search_ctx == NULL)
4028 goto error_return;
4029 memset(ff_search_ctx, 0, sizeof(ff_search_ctx_T));
4030 }
4031
4032 /* clear the search context, but NOT the visited lists */
4033 ff_clear();
4034
4035 /* clear visited list if wanted */
4036 if (free_visited == TRUE)
4037 vim_findfile_free_visited(ff_search_ctx);
4038 else
4039 {
4040 /* Reuse old visited lists. Get the visited list for the given
4041 * filename. If no list for the current filename exists, creates a new
4042 * one.
4043 */
4044 ff_search_ctx->ffsc_visited_list = ff_get_visited_list(filename,
4045 &ff_search_ctx->ffsc_visited_lists_list);
4046 if (ff_search_ctx->ffsc_visited_list == NULL)
4047 goto error_return;
4048 ff_search_ctx->ffsc_dir_visited_list = ff_get_visited_list(filename,
4049 &ff_search_ctx->ffsc_dir_visited_lists_list);
4050 if (ff_search_ctx->ffsc_dir_visited_list == NULL)
4051 goto error_return;
4052 }
4053
4054 if (ff_expand_buffer == NULL)
4055 {
4056 ff_expand_buffer = (char_u*)alloc(MAXPATHL);
4057 if (ff_expand_buffer == NULL)
4058 goto error_return;
4059 }
4060
4061 /* Store information on starting dir now if path is relative.
Bram Moolenaar5eb86f92004-07-26 12:53:41 +00004062 * If path is absolute, we do that later. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004063 if (path[0] == '.'
4064 && (vim_ispathsep(path[1]) || path[1] == NUL)
4065 && (!tagfile || vim_strchr(p_cpo, CPO_DOTTAG) == NULL)
4066 && rel_fname != NULL)
4067 {
4068 int len = (int)(gettail(rel_fname) - rel_fname);
4069
4070 if (!vim_isAbsName(rel_fname) && len + 1 < MAXPATHL)
4071 {
4072 /* Make the start dir an absolute path name. */
Bram Moolenaarbbebc852005-07-18 21:47:53 +00004073 vim_strncpy(ff_expand_buffer, rel_fname, len);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004074 ff_search_ctx->ffsc_start_dir = FullName_save(ff_expand_buffer,
4075 FALSE);
4076 }
4077 else
4078 ff_search_ctx->ffsc_start_dir = vim_strnsave(rel_fname, len);
4079 if (ff_search_ctx->ffsc_start_dir == NULL)
4080 goto error_return;
4081 if (*++path != NUL)
4082 ++path;
4083 }
4084 else if (*path == NUL || !vim_isAbsName(path))
4085 {
4086#ifdef BACKSLASH_IN_FILENAME
4087 /* "c:dir" needs "c:" to be expanded, otherwise use current dir */
4088 if (*path != NUL && path[1] == ':')
4089 {
4090 char_u drive[3];
4091
4092 drive[0] = path[0];
4093 drive[1] = ':';
4094 drive[2] = NUL;
4095 if (vim_FullName(drive, ff_expand_buffer, MAXPATHL, TRUE) == FAIL)
4096 goto error_return;
4097 path += 2;
4098 }
4099 else
4100#endif
4101 if (mch_dirname(ff_expand_buffer, MAXPATHL) == FAIL)
4102 goto error_return;
4103
4104 ff_search_ctx->ffsc_start_dir = vim_strsave(ff_expand_buffer);
4105 if (ff_search_ctx->ffsc_start_dir == NULL)
4106 goto error_return;
4107
4108#ifdef BACKSLASH_IN_FILENAME
4109 /* A path that starts with "/dir" is relative to the drive, not to the
4110 * directory (but not for "//machine/dir"). Only use the drive name. */
4111 if ((*path == '/' || *path == '\\')
4112 && path[1] != path[0]
4113 && ff_search_ctx->ffsc_start_dir[1] == ':')
4114 ff_search_ctx->ffsc_start_dir[2] = NUL;
4115#endif
4116 }
4117
4118#ifdef FEAT_PATH_EXTRA
4119 /*
4120 * If stopdirs are given, split them into an array of pointers.
4121 * If this fails (mem allocation), there is no upward search at all or a
4122 * stop directory is not recognized -> continue silently.
4123 * If stopdirs just contains a ";" or is empty,
4124 * ff_search_ctx->ffsc_stopdirs_v will only contain a NULL pointer. This
4125 * is handled as unlimited upward search. See function
4126 * ff_path_in_stoplist() for details.
4127 */
4128 if (stopdirs != NULL)
4129 {
4130 char_u *walker = stopdirs;
4131 int dircount;
4132
4133 while (*walker == ';')
4134 walker++;
4135
4136 dircount = 1;
4137 ff_search_ctx->ffsc_stopdirs_v =
4138 (char_u **)alloc((unsigned)sizeof(char_u *));
4139
4140 if (ff_search_ctx->ffsc_stopdirs_v != NULL)
4141 {
4142 do
4143 {
4144 char_u *helper;
4145 void *ptr;
4146
4147 helper = walker;
4148 ptr = vim_realloc(ff_search_ctx->ffsc_stopdirs_v,
4149 (dircount + 1) * sizeof(char_u *));
4150 if (ptr)
4151 ff_search_ctx->ffsc_stopdirs_v = ptr;
4152 else
4153 /* ignore, keep what we have and continue */
4154 break;
4155 walker = vim_strchr(walker, ';');
4156 if (walker)
4157 {
4158 ff_search_ctx->ffsc_stopdirs_v[dircount-1] =
4159 vim_strnsave(helper, (int)(walker - helper));
4160 walker++;
4161 }
4162 else
4163 /* this might be "", which means ascent till top
4164 * of directory tree.
4165 */
4166 ff_search_ctx->ffsc_stopdirs_v[dircount-1] =
4167 vim_strsave(helper);
4168
4169 dircount++;
4170
4171 } while (walker != NULL);
4172 ff_search_ctx->ffsc_stopdirs_v[dircount-1] = NULL;
4173 }
4174 }
4175#endif
4176
4177#ifdef FEAT_PATH_EXTRA
4178 ff_search_ctx->ffsc_level = level;
4179
4180 /* split into:
4181 * -fix path
4182 * -wildcard_stuff (might be NULL)
4183 */
4184 wc_part = vim_strchr(path, '*');
4185 if (wc_part != NULL)
4186 {
4187 int llevel;
4188 int len;
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00004189 char *errpt;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004190
4191 /* save the fix part of the path */
4192 ff_search_ctx->ffsc_fix_path = vim_strnsave(path,
4193 (int)(wc_part - path));
4194
4195 /*
4196 * copy wc_path and add restricts to the '**' wildcard.
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00004197 * The octet after a '**' is used as a (binary) counter.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004198 * So '**3' is transposed to '**^C' ('^C' is ASCII value 3)
4199 * or '**76' is transposed to '**N'( 'N' is ASCII value 76).
4200 * For EBCDIC you get different character values.
4201 * If no restrict is given after '**' the default is used.
4202 * Due to this technic the path looks awful if you print it as a
4203 * string.
4204 */
4205 len = 0;
4206 while (*wc_part != NUL)
4207 {
4208 if (STRNCMP(wc_part, "**", 2) == 0)
4209 {
4210 ff_expand_buffer[len++] = *wc_part++;
4211 ff_expand_buffer[len++] = *wc_part++;
4212
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00004213 llevel = strtol((char *)wc_part, &errpt, 10);
4214 if ((char_u *)errpt != wc_part && llevel > 0 && llevel < 255)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004215 ff_expand_buffer[len++] = llevel;
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00004216 else if ((char_u *)errpt != wc_part && llevel == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004217 /* restrict is 0 -> remove already added '**' */
4218 len -= 2;
4219 else
4220 ff_expand_buffer[len++] = FF_MAX_STAR_STAR_EXPAND;
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00004221 wc_part = (char_u *)errpt;
Bram Moolenaar1d94f9b2005-08-04 21:29:45 +00004222 if (*wc_part != NUL && !vim_ispathsep(*wc_part))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004223 {
4224 EMSG2(_("E343: Invalid path: '**[number]' must be at the end of the path or be followed by '%s'."), PATHSEPSTR);
4225 goto error_return;
4226 }
4227 }
4228 else
4229 ff_expand_buffer[len++] = *wc_part++;
4230 }
4231 ff_expand_buffer[len] = NUL;
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00004232 ff_search_ctx->ffsc_wc_path = vim_strsave(ff_expand_buffer);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004233
4234 if (ff_search_ctx->ffsc_wc_path == NULL)
4235 goto error_return;
4236 }
4237 else
4238#endif
4239 ff_search_ctx->ffsc_fix_path = vim_strsave(path);
4240
4241 if (ff_search_ctx->ffsc_start_dir == NULL)
4242 {
4243 /* store the fix part as startdir.
4244 * This is needed if the parameter path is fully qualified.
4245 */
4246 ff_search_ctx->ffsc_start_dir = vim_strsave(ff_search_ctx->ffsc_fix_path);
4247 if (ff_search_ctx->ffsc_start_dir)
4248 ff_search_ctx->ffsc_fix_path[0] = NUL;
4249 }
4250
4251 /* create an absolute path */
4252 STRCPY(ff_expand_buffer, ff_search_ctx->ffsc_start_dir);
4253 add_pathsep(ff_expand_buffer);
4254 STRCAT(ff_expand_buffer, ff_search_ctx->ffsc_fix_path);
4255 add_pathsep(ff_expand_buffer);
4256
4257 sptr = ff_create_stack_element(ff_expand_buffer,
4258#ifdef FEAT_PATH_EXTRA
4259 ff_search_ctx->ffsc_wc_path,
4260#endif
4261 level, 0);
4262
4263 if (sptr == NULL)
4264 goto error_return;
4265
4266 ff_push(sptr);
4267
4268 ff_search_ctx->ffsc_file_to_search = vim_strsave(filename);
4269 if (ff_search_ctx->ffsc_file_to_search == NULL)
4270 goto error_return;
4271
4272 return ff_search_ctx;
4273
4274error_return:
4275 /*
4276 * We clear the search context now!
4277 * Even when the caller gave us a (perhaps valid) context we free it here,
4278 * as we might have already destroyed it.
4279 */
4280 vim_findfile_cleanup(ff_search_ctx);
4281 return NULL;
4282}
4283
4284#if defined(FEAT_PATH_EXTRA) || defined(PROTO)
4285/*
4286 * Get the stopdir string. Check that ';' is not escaped.
4287 */
4288 char_u *
4289vim_findfile_stopdir(buf)
4290 char_u *buf;
4291{
4292 char_u *r_ptr = buf;
4293
4294 while (*r_ptr != NUL && *r_ptr != ';')
4295 {
4296 if (r_ptr[0] == '\\' && r_ptr[1] == ';')
4297 {
4298 /* overwrite the escape char,
4299 * use STRLEN(r_ptr) to move the trailing '\0'
4300 */
4301 mch_memmove(r_ptr, r_ptr + 1, STRLEN(r_ptr));
4302 r_ptr++;
4303 }
4304 r_ptr++;
4305 }
4306 if (*r_ptr == ';')
4307 {
4308 *r_ptr = 0;
4309 r_ptr++;
4310 }
4311 else if (*r_ptr == NUL)
4312 r_ptr = NULL;
4313 return r_ptr;
4314}
4315#endif
4316
4317/* Clean up the given search context. Can handle a NULL pointer */
4318 void
4319vim_findfile_cleanup(ctx)
4320 void *ctx;
4321{
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00004322 if (ctx == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004323 return;
4324
4325 ff_search_ctx = ctx;
4326
4327 vim_findfile_free_visited(ctx);
4328 ff_clear();
4329 vim_free(ctx);
4330 ff_search_ctx = NULL;
4331}
4332
4333/*
4334 * Find a file in a search context.
4335 * The search context was created with vim_findfile_init() above.
4336 * Return a pointer to an allocated file name or NULL if nothing found.
4337 * To get all matching files call this function until you get NULL.
4338 *
Bram Moolenaar5eb86f92004-07-26 12:53:41 +00004339 * If the passed search_context is NULL, NULL is returned.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004340 *
4341 * The search algorithm is depth first. To change this replace the
4342 * stack with a list (don't forget to leave partly searched directories on the
4343 * top of the list).
4344 */
4345 char_u *
4346vim_findfile(search_ctx)
4347 void *search_ctx;
4348{
4349 char_u *file_path;
4350#ifdef FEAT_PATH_EXTRA
4351 char_u *rest_of_wildcards;
4352 char_u *path_end = NULL;
4353#endif
4354 ff_stack_T *ctx;
4355#if defined(FEAT_SEARCHPATH) || defined(FEAT_PATH_EXTRA)
4356 int len;
4357#endif
4358 int i;
4359 char_u *p;
4360#ifdef FEAT_SEARCHPATH
4361 char_u *suf;
4362#endif
4363
4364 if (search_ctx == NULL)
4365 return NULL;
4366
4367 ff_search_ctx = (ff_search_ctx_T*)search_ctx;
4368
4369 /*
4370 * filepath is used as buffer for various actions and as the storage to
4371 * return a found filename.
4372 */
4373 if ((file_path = alloc((int)MAXPATHL)) == NULL)
4374 return NULL;
4375
4376#ifdef FEAT_PATH_EXTRA
4377 /* store the end of the start dir -- needed for upward search */
4378 if (ff_search_ctx->ffsc_start_dir != NULL)
4379 path_end = &ff_search_ctx->ffsc_start_dir[STRLEN(ff_search_ctx->ffsc_start_dir)];
4380#endif
4381
4382#ifdef FEAT_PATH_EXTRA
4383 /* upward search loop */
4384 for (;;)
4385 {
4386#endif
4387 /* downward search loop */
4388 for (;;)
4389 {
4390 /* check if user user wants to stop the search*/
4391 ui_breakcheck();
4392 if (got_int)
4393 break;
4394
4395 /* get directory to work on from stack */
4396 ctx = ff_pop();
4397 if (ctx == NULL)
4398 break;
4399
4400 /*
4401 * TODO: decide if we leave this test in
4402 *
4403 * GOOD: don't search a directory(-tree) twice.
4404 * BAD: - check linked list for every new directory entered.
4405 * - check for double files also done below
4406 *
4407 * Here we check if we already searched this directory.
4408 * We already searched a directory if:
4409 * 1) The directory is the same.
4410 * 2) We would use the same wildcard string.
4411 *
4412 * Good if you have links on same directory via several ways
4413 * or you have selfreferences in directories (e.g. SuSE Linux 6.3:
4414 * /etc/rc.d/init.d is linked to /etc/rc.d -> endless loop)
4415 *
4416 * This check is only needed for directories we work on for the
4417 * first time (hence ctx->ff_filearray == NULL)
4418 */
4419 if (ctx->ffs_filearray == NULL
4420 && ff_check_visited(&ff_search_ctx->ffsc_dir_visited_list
4421 ->ffvl_visited_list,
4422 ctx->ffs_fix_path
4423#ifdef FEAT_PATH_EXTRA
4424 , ctx->ffs_wc_path
4425#endif
4426 ) == FAIL)
4427 {
4428#ifdef FF_VERBOSE
4429 if (p_verbose >= 5)
4430 {
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004431 verbose_enter_scroll();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004432 smsg((char_u *)"Already Searched: %s (%s)",
4433 ctx->ffs_fix_path, ctx->ffs_wc_path);
4434 /* don't overwrite this either */
4435 msg_puts((char_u *)"\n");
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004436 verbose_leave_scroll();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004437 }
4438#endif
4439 ff_free_stack_element(ctx);
4440 continue;
4441 }
4442#ifdef FF_VERBOSE
4443 else if (p_verbose >= 5)
4444 {
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004445 verbose_enter_scroll();
Bram Moolenaar051b7822005-05-19 21:00:46 +00004446 smsg((char_u *)"Searching: %s (%s)",
4447 ctx->ffs_fix_path, ctx->ffs_wc_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004448 /* don't overwrite this either */
4449 msg_puts((char_u *)"\n");
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004450 verbose_leave_scroll();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004451 }
4452#endif
4453
4454 /* check depth */
4455 if (ctx->ffs_level <= 0)
4456 {
4457 ff_free_stack_element(ctx);
4458 continue;
4459 }
4460
4461 file_path[0] = NUL;
4462
4463 /*
4464 * If no filearray till now expand wildcards
4465 * The function expand_wildcards() can handle an array of paths
4466 * and all possible expands are returned in one array. We use this
4467 * to handle the expansion of '**' into an empty string.
4468 */
4469 if (ctx->ffs_filearray == NULL)
4470 {
4471 char_u *dirptrs[2];
4472
4473 /* we use filepath to build the path expand_wildcards() should
4474 * expand.
4475 */
4476 dirptrs[0] = file_path;
4477 dirptrs[1] = NULL;
4478
4479 /* if we have a start dir copy it in */
4480 if (!vim_isAbsName(ctx->ffs_fix_path)
4481 && ff_search_ctx->ffsc_start_dir)
4482 {
4483 STRCPY(file_path, ff_search_ctx->ffsc_start_dir);
4484 add_pathsep(file_path);
4485 }
4486
4487 /* append the fix part of the search path */
4488 STRCAT(file_path, ctx->ffs_fix_path);
4489 add_pathsep(file_path);
4490
4491#ifdef FEAT_PATH_EXTRA
4492 rest_of_wildcards = ctx->ffs_wc_path;
4493 if (*rest_of_wildcards != NUL)
4494 {
4495 len = (int)STRLEN(file_path);
4496 if (STRNCMP(rest_of_wildcards, "**", 2) == 0)
4497 {
4498 /* pointer to the restrict byte
4499 * The restrict byte is not a character!
4500 */
4501 p = rest_of_wildcards + 2;
4502
4503 if (*p > 0)
4504 {
4505 (*p)--;
4506 file_path[len++] = '*';
4507 }
4508
4509 if (*p == 0)
4510 {
4511 /* remove '**<numb> from wildcards */
4512 mch_memmove(rest_of_wildcards,
4513 rest_of_wildcards + 3,
4514 STRLEN(rest_of_wildcards + 3) + 1);
4515 }
4516 else
4517 rest_of_wildcards += 3;
4518
4519 if (ctx->ffs_star_star_empty == 0)
4520 {
4521 /* if not done before, expand '**' to empty */
4522 ctx->ffs_star_star_empty = 1;
4523 dirptrs[1] = ctx->ffs_fix_path;
4524 }
4525 }
4526
4527 /*
4528 * Here we copy until the next path separator or the end of
4529 * the path. If we stop at a path separator, there is
Bram Moolenaar7aa9f6a2007-05-10 18:00:30 +00004530 * still something else left. This is handled below by
Bram Moolenaar071d4272004-06-13 20:20:40 +00004531 * pushing every directory returned from expand_wildcards()
4532 * on the stack again for further search.
4533 */
4534 while (*rest_of_wildcards
4535 && !vim_ispathsep(*rest_of_wildcards))
4536 file_path[len++] = *rest_of_wildcards++;
4537
4538 file_path[len] = NUL;
4539 if (vim_ispathsep(*rest_of_wildcards))
4540 rest_of_wildcards++;
4541 }
4542#endif
4543
4544 /*
4545 * Expand wildcards like "*" and "$VAR".
4546 * If the path is a URL don't try this.
4547 */
4548 if (path_with_url(dirptrs[0]))
4549 {
4550 ctx->ffs_filearray = (char_u **)
4551 alloc((unsigned)sizeof(char *));
4552 if (ctx->ffs_filearray != NULL
4553 && (ctx->ffs_filearray[0]
4554 = vim_strsave(dirptrs[0])) != NULL)
4555 ctx->ffs_filearray_size = 1;
4556 else
4557 ctx->ffs_filearray_size = 0;
4558 }
4559 else
4560 expand_wildcards((dirptrs[1] == NULL) ? 1 : 2, dirptrs,
4561 &ctx->ffs_filearray_size,
4562 &ctx->ffs_filearray,
4563 EW_DIR|EW_ADDSLASH|EW_SILENT);
4564
4565 ctx->ffs_filearray_cur = 0;
4566 ctx->ffs_stage = 0;
4567 }
4568#ifdef FEAT_PATH_EXTRA
4569 else
4570 rest_of_wildcards = &ctx->ffs_wc_path[STRLEN(ctx->ffs_wc_path)];
4571#endif
4572
4573 if (ctx->ffs_stage == 0)
4574 {
4575 /* this is the first time we work on this directory */
4576#ifdef FEAT_PATH_EXTRA
4577 if (*rest_of_wildcards == NUL)
4578#endif
4579 {
4580 /*
4581 * we don't have further wildcards to expand, so we have to
4582 * check for the final file now
4583 */
4584 for (i = ctx->ffs_filearray_cur;
4585 i < ctx->ffs_filearray_size; ++i)
4586 {
4587 if (!path_with_url(ctx->ffs_filearray[i])
4588 && !mch_isdir(ctx->ffs_filearray[i]))
4589 continue; /* not a directory */
4590
4591 /* prepare the filename to be checked for existance
4592 * below */
4593 STRCPY(file_path, ctx->ffs_filearray[i]);
4594 add_pathsep(file_path);
4595 STRCAT(file_path, ff_search_ctx->ffsc_file_to_search);
4596
4597 /*
4598 * Try without extra suffix and then with suffixes
4599 * from 'suffixesadd'.
4600 */
4601#ifdef FEAT_SEARCHPATH
4602 len = (int)STRLEN(file_path);
4603 suf = curbuf->b_p_sua;
4604 for (;;)
4605#endif
4606 {
4607 /* if file exists and we didn't already find it */
4608 if ((path_with_url(file_path)
4609 || (mch_getperm(file_path) >= 0
4610 && (!ff_search_ctx->ffsc_need_dir
4611 || mch_isdir(file_path))))
4612#ifndef FF_VERBOSE
4613 && (ff_check_visited(
4614 &ff_search_ctx->ffsc_visited_list->ffvl_visited_list,
4615 file_path
4616#ifdef FEAT_PATH_EXTRA
4617 , (char_u *)""
4618#endif
4619 ) == OK)
4620#endif
4621 )
4622 {
4623#ifdef FF_VERBOSE
4624 if (ff_check_visited(
4625 &ff_search_ctx->ffsc_visited_list->ffvl_visited_list,
4626 file_path
4627#ifdef FEAT_PATH_EXTRA
4628 , (char_u *)""
4629#endif
4630 ) == FAIL)
4631 {
4632 if (p_verbose >= 5)
4633 {
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004634 verbose_enter_scroll();
Bram Moolenaar051b7822005-05-19 21:00:46 +00004635 smsg((char_u *)"Already: %s",
Bram Moolenaar071d4272004-06-13 20:20:40 +00004636 file_path);
4637 /* don't overwrite this either */
4638 msg_puts((char_u *)"\n");
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004639 verbose_leave_scroll();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004640 }
4641 continue;
4642 }
4643#endif
4644
4645 /* push dir to examine rest of subdirs later */
4646 ctx->ffs_filearray_cur = i + 1;
4647 ff_push(ctx);
4648
4649 simplify_filename(file_path);
4650 if (mch_dirname(ff_expand_buffer, MAXPATHL)
4651 == OK)
4652 {
4653 p = shorten_fname(file_path,
4654 ff_expand_buffer);
4655 if (p != NULL)
4656 mch_memmove(file_path, p,
4657 STRLEN(p) + 1);
4658 }
4659#ifdef FF_VERBOSE
4660 if (p_verbose >= 5)
4661 {
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004662 verbose_enter_scroll();
Bram Moolenaar051b7822005-05-19 21:00:46 +00004663 smsg((char_u *)"HIT: %s", file_path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004664 /* don't overwrite this either */
4665 msg_puts((char_u *)"\n");
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004666 verbose_leave_scroll();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004667 }
4668#endif
4669 return file_path;
4670 }
4671
4672#ifdef FEAT_SEARCHPATH
4673 /* Not found or found already, try next suffix. */
4674 if (*suf == NUL)
4675 break;
4676 copy_option_part(&suf, file_path + len,
4677 MAXPATHL - len, ",");
4678#endif
4679 }
4680 }
4681 }
4682#ifdef FEAT_PATH_EXTRA
4683 else
4684 {
4685 /*
4686 * still wildcards left, push the directories for further
4687 * search
4688 */
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00004689 for (i = ctx->ffs_filearray_cur;
4690 i < ctx->ffs_filearray_size; ++i)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004691 {
4692 if (!mch_isdir(ctx->ffs_filearray[i]))
4693 continue; /* not a directory */
4694
4695 ff_push(ff_create_stack_element(ctx->ffs_filearray[i],
4696 rest_of_wildcards, ctx->ffs_level - 1, 0));
4697 }
4698 }
4699#endif
4700 ctx->ffs_filearray_cur = 0;
4701 ctx->ffs_stage = 1;
4702 }
4703
4704#ifdef FEAT_PATH_EXTRA
4705 /*
4706 * if wildcards contains '**' we have to descent till we reach the
4707 * leaves of the directory tree.
4708 */
4709 if (STRNCMP(ctx->ffs_wc_path, "**", 2) == 0)
4710 {
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00004711 for (i = ctx->ffs_filearray_cur;
4712 i < ctx->ffs_filearray_size; ++i)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004713 {
4714 if (fnamecmp(ctx->ffs_filearray[i], ctx->ffs_fix_path) == 0)
4715 continue; /* don't repush same directory */
4716 if (!mch_isdir(ctx->ffs_filearray[i]))
4717 continue; /* not a directory */
4718 ff_push(ff_create_stack_element(ctx->ffs_filearray[i],
4719 ctx->ffs_wc_path, ctx->ffs_level - 1, 1));
4720 }
4721 }
4722#endif
4723
4724 /* we are done with the current directory */
4725 ff_free_stack_element(ctx);
4726
4727 }
4728
4729#ifdef FEAT_PATH_EXTRA
4730 /* If we reached this, we didn't find anything downwards.
4731 * Let's check if we should do an upward search.
4732 */
4733 if (ff_search_ctx->ffsc_start_dir
4734 && ff_search_ctx->ffsc_stopdirs_v != NULL && !got_int)
4735 {
4736 ff_stack_T *sptr;
4737
4738 /* is the last starting directory in the stop list? */
4739 if (ff_path_in_stoplist(ff_search_ctx->ffsc_start_dir,
4740 (int)(path_end - ff_search_ctx->ffsc_start_dir),
4741 ff_search_ctx->ffsc_stopdirs_v) == TRUE)
4742 break;
4743
4744 /* cut of last dir */
4745 while (path_end > ff_search_ctx->ffsc_start_dir
Bram Moolenaar1d94f9b2005-08-04 21:29:45 +00004746 && vim_ispathsep(*path_end))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004747 path_end--;
4748 while (path_end > ff_search_ctx->ffsc_start_dir
Bram Moolenaar1d94f9b2005-08-04 21:29:45 +00004749 && !vim_ispathsep(path_end[-1]))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004750 path_end--;
4751 *path_end = 0;
4752 path_end--;
4753
4754 if (*ff_search_ctx->ffsc_start_dir == 0)
4755 break;
4756
4757 STRCPY(file_path, ff_search_ctx->ffsc_start_dir);
4758 add_pathsep(file_path);
4759 STRCAT(file_path, ff_search_ctx->ffsc_fix_path);
4760
4761 /* create a new stack entry */
4762 sptr = ff_create_stack_element(file_path,
4763 ff_search_ctx->ffsc_wc_path, ff_search_ctx->ffsc_level, 0);
4764 if (sptr == NULL)
4765 break;
4766 ff_push(sptr);
4767 }
4768 else
4769 break;
4770 }
4771#endif
4772
4773 vim_free(file_path);
4774 return NULL;
4775}
4776
4777/*
4778 * Free the list of lists of visited files and directories
4779 * Can handle it if the passed search_context is NULL;
4780 */
4781 void
4782vim_findfile_free_visited(search_ctx)
4783 void *search_ctx;
4784{
4785 if (search_ctx == NULL)
4786 return;
4787
4788 ff_search_ctx = (ff_search_ctx_T *)search_ctx;
4789
4790 vim_findfile_free_visited_list(&ff_search_ctx->ffsc_visited_lists_list);
4791 vim_findfile_free_visited_list(&ff_search_ctx->ffsc_dir_visited_lists_list);
4792}
4793
4794 static void
4795vim_findfile_free_visited_list(list_headp)
4796 ff_visited_list_hdr_T **list_headp;
4797{
4798 ff_visited_list_hdr_T *vp;
4799
4800 while (*list_headp != NULL)
4801 {
4802 vp = (*list_headp)->ffvl_next;
4803 ff_free_visited_list((*list_headp)->ffvl_visited_list);
4804
4805 vim_free((*list_headp)->ffvl_filename);
4806 vim_free(*list_headp);
4807 *list_headp = vp;
4808 }
4809 *list_headp = NULL;
4810}
4811
4812 static void
4813ff_free_visited_list(vl)
4814 ff_visited_T *vl;
4815{
4816 ff_visited_T *vp;
4817
4818 while (vl != NULL)
4819 {
4820 vp = vl->ffv_next;
4821#ifdef FEAT_PATH_EXTRA
4822 vim_free(vl->ffv_wc_path);
4823#endif
4824 vim_free(vl);
4825 vl = vp;
4826 }
4827 vl = NULL;
4828}
4829
4830/*
4831 * Returns the already visited list for the given filename. If none is found it
4832 * allocates a new one.
4833 */
4834 static ff_visited_list_hdr_T*
4835ff_get_visited_list(filename, list_headp)
4836 char_u *filename;
4837 ff_visited_list_hdr_T **list_headp;
4838{
4839 ff_visited_list_hdr_T *retptr = NULL;
4840
4841 /* check if a visited list for the given filename exists */
4842 if (*list_headp != NULL)
4843 {
4844 retptr = *list_headp;
4845 while (retptr != NULL)
4846 {
4847 if (fnamecmp(filename, retptr->ffvl_filename) == 0)
4848 {
4849#ifdef FF_VERBOSE
4850 if (p_verbose >= 5)
4851 {
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004852 verbose_enter_scroll();
Bram Moolenaar051b7822005-05-19 21:00:46 +00004853 smsg((char_u *)"ff_get_visited_list: FOUND list for %s",
Bram Moolenaar071d4272004-06-13 20:20:40 +00004854 filename);
4855 /* don't overwrite this either */
4856 msg_puts((char_u *)"\n");
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004857 verbose_leave_scroll();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004858 }
4859#endif
4860 return retptr;
4861 }
4862 retptr = retptr->ffvl_next;
4863 }
4864 }
4865
4866#ifdef FF_VERBOSE
4867 if (p_verbose >= 5)
4868 {
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004869 verbose_enter_scroll();
Bram Moolenaar051b7822005-05-19 21:00:46 +00004870 smsg((char_u *)"ff_get_visited_list: new list for %s", filename);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004871 /* don't overwrite this either */
4872 msg_puts((char_u *)"\n");
Bram Moolenaar5c06f8b2005-05-31 22:14:58 +00004873 verbose_leave_scroll();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004874 }
4875#endif
4876
4877 /*
4878 * if we reach this we didn't find a list and we have to allocate new list
4879 */
4880 retptr = (ff_visited_list_hdr_T*)alloc((unsigned)sizeof(*retptr));
4881 if (retptr == NULL)
4882 return NULL;
4883
4884 retptr->ffvl_visited_list = NULL;
4885 retptr->ffvl_filename = vim_strsave(filename);
4886 if (retptr->ffvl_filename == NULL)
4887 {
4888 vim_free(retptr);
4889 return NULL;
4890 }
4891 retptr->ffvl_next = *list_headp;
4892 *list_headp = retptr;
4893
4894 return retptr;
4895}
4896
4897#ifdef FEAT_PATH_EXTRA
4898/*
4899 * check if two wildcard paths are equal. Returns TRUE or FALSE.
4900 * They are equal if:
4901 * - both paths are NULL
4902 * - they have the same length
4903 * - char by char comparison is OK
4904 * - the only differences are in the counters behind a '**', so
4905 * '**\20' is equal to '**\24'
4906 */
4907 static int
4908ff_wc_equal(s1, s2)
4909 char_u *s1;
4910 char_u *s2;
4911{
4912 int i;
4913
4914 if (s1 == s2)
4915 return TRUE;
4916
4917 if (s1 == NULL || s2 == NULL)
4918 return FALSE;
4919
4920 if (STRLEN(s1) != STRLEN(s2))
4921 return FAIL;
4922
4923 for (i = 0; s1[i] != NUL && s2[i] != NUL; i++)
4924 {
4925 if (s1[i] != s2[i]
4926#ifdef CASE_INSENSITIVE_FILENAME
4927 && TOUPPER_LOC(s1[i]) != TOUPPER_LOC(s2[i])
4928#endif
4929 )
4930 {
4931 if (i >= 2)
4932 if (s1[i-1] == '*' && s1[i-2] == '*')
4933 continue;
4934 else
4935 return FAIL;
4936 else
4937 return FAIL;
4938 }
4939 }
4940 return TRUE;
4941}
4942#endif
4943
4944/*
4945 * maintains the list of already visited files and dirs
4946 * returns FAIL if the given file/dir is already in the list
4947 * returns OK if it is newly added
4948 *
4949 * TODO: What to do on memory allocation problems?
4950 * -> return TRUE - Better the file is found several times instead of
4951 * never.
4952 */
4953 static int
4954ff_check_visited(visited_list, fname
4955#ifdef FEAT_PATH_EXTRA
4956 , wc_path
4957#endif
4958 )
4959 ff_visited_T **visited_list;
4960 char_u *fname;
4961#ifdef FEAT_PATH_EXTRA
4962 char_u *wc_path;
4963#endif
4964{
4965 ff_visited_T *vp;
4966#ifdef UNIX
4967 struct stat st;
4968 int url = FALSE;
4969#endif
4970
4971 /* For an URL we only compare the name, otherwise we compare the
4972 * device/inode (unix) or the full path name (not Unix). */
4973 if (path_with_url(fname))
4974 {
Bram Moolenaarbbebc852005-07-18 21:47:53 +00004975 vim_strncpy(ff_expand_buffer, fname, MAXPATHL - 1);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004976#ifdef UNIX
4977 url = TRUE;
4978#endif
4979 }
4980 else
4981 {
4982 ff_expand_buffer[0] = NUL;
4983#ifdef UNIX
4984 if (mch_stat((char *)fname, &st) < 0)
4985#else
4986 if (vim_FullName(fname, ff_expand_buffer, MAXPATHL, TRUE) == FAIL)
4987#endif
4988 return FAIL;
4989 }
4990
4991 /* check against list of already visited files */
4992 for (vp = *visited_list; vp != NULL; vp = vp->ffv_next)
4993 {
4994 if (
4995#ifdef UNIX
4996 !url
4997 ? (vp->ffv_dev == st.st_dev
4998 && vp->ffv_ino == st.st_ino)
4999 :
5000#endif
5001 fnamecmp(vp->ffv_fname, ff_expand_buffer) == 0
5002 )
5003 {
5004#ifdef FEAT_PATH_EXTRA
5005 /* are the wildcard parts equal */
5006 if (ff_wc_equal(vp->ffv_wc_path, wc_path) == TRUE)
5007#endif
5008 /* already visited */
5009 return FAIL;
5010 }
5011 }
5012
5013 /*
5014 * New file/dir. Add it to the list of visited files/dirs.
5015 */
5016 vp = (ff_visited_T *)alloc((unsigned)(sizeof(ff_visited_T)
5017 + STRLEN(ff_expand_buffer)));
5018
5019 if (vp != NULL)
5020 {
5021#ifdef UNIX
5022 if (!url)
5023 {
5024 vp->ffv_ino = st.st_ino;
5025 vp->ffv_dev = st.st_dev;
5026 vp->ffv_fname[0] = NUL;
5027 }
5028 else
5029 {
5030 vp->ffv_ino = 0;
5031 vp->ffv_dev = -1;
5032#endif
5033 STRCPY(vp->ffv_fname, ff_expand_buffer);
5034#ifdef UNIX
5035 }
5036#endif
5037#ifdef FEAT_PATH_EXTRA
5038 if (wc_path != NULL)
5039 vp->ffv_wc_path = vim_strsave(wc_path);
5040 else
5041 vp->ffv_wc_path = NULL;
5042#endif
5043
5044 vp->ffv_next = *visited_list;
5045 *visited_list = vp;
5046 }
5047
5048 return OK;
5049}
5050
5051/*
5052 * create stack element from given path pieces
5053 */
5054 static ff_stack_T *
5055ff_create_stack_element(fix_part,
5056#ifdef FEAT_PATH_EXTRA
5057 wc_part,
5058#endif
5059 level, star_star_empty)
5060 char_u *fix_part;
5061#ifdef FEAT_PATH_EXTRA
5062 char_u *wc_part;
5063#endif
5064 int level;
5065 int star_star_empty;
5066{
5067 ff_stack_T *new;
5068
5069 new = (ff_stack_T *)alloc((unsigned)sizeof(ff_stack_T));
5070 if (new == NULL)
5071 return NULL;
5072
5073 new->ffs_prev = NULL;
5074 new->ffs_filearray = NULL;
5075 new->ffs_filearray_size = 0;
5076 new->ffs_filearray_cur = 0;
5077 new->ffs_stage = 0;
5078 new->ffs_level = level;
5079 new->ffs_star_star_empty = star_star_empty;;
5080
5081 /* the following saves NULL pointer checks in vim_findfile */
5082 if (fix_part == NULL)
5083 fix_part = (char_u *)"";
5084 new->ffs_fix_path = vim_strsave(fix_part);
5085
5086#ifdef FEAT_PATH_EXTRA
5087 if (wc_part == NULL)
5088 wc_part = (char_u *)"";
5089 new->ffs_wc_path = vim_strsave(wc_part);
5090#endif
5091
5092 if (new->ffs_fix_path == NULL
5093#ifdef FEAT_PATH_EXTRA
5094 || new->ffs_wc_path == NULL
5095#endif
5096 )
5097 {
5098 ff_free_stack_element(new);
5099 new = NULL;
5100 }
5101
5102 return new;
5103}
5104
5105/*
5106 * push a dir on the directory stack
5107 */
5108 static void
5109ff_push(ctx)
5110 ff_stack_T *ctx;
5111{
5112 /* check for NULL pointer, not to return an error to the user, but
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005113 * to prevent a crash */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005114 if (ctx != NULL)
5115 {
5116 ctx->ffs_prev = ff_search_ctx->ffsc_stack_ptr;
5117 ff_search_ctx->ffsc_stack_ptr = ctx;
5118 }
5119}
5120
5121/*
5122 * pop a dir from the directory stack
5123 * returns NULL if stack is empty
5124 */
5125 static ff_stack_T *
5126ff_pop()
5127{
5128 ff_stack_T *sptr;
5129
5130 sptr = ff_search_ctx->ffsc_stack_ptr;
5131 if (ff_search_ctx->ffsc_stack_ptr != NULL)
5132 ff_search_ctx->ffsc_stack_ptr = ff_search_ctx->ffsc_stack_ptr->ffs_prev;
5133
5134 return sptr;
5135}
5136
5137/*
5138 * free the given stack element
5139 */
5140 static void
5141ff_free_stack_element(ctx)
5142 ff_stack_T *ctx;
5143{
5144 /* vim_free handles possible NULL pointers */
5145 vim_free(ctx->ffs_fix_path);
5146#ifdef FEAT_PATH_EXTRA
5147 vim_free(ctx->ffs_wc_path);
5148#endif
5149
5150 if (ctx->ffs_filearray != NULL)
5151 FreeWild(ctx->ffs_filearray_size, ctx->ffs_filearray);
5152
5153 vim_free(ctx);
5154}
5155
5156/*
5157 * clear the search context
5158 */
5159 static void
5160ff_clear()
5161{
5162 ff_stack_T *sptr;
5163
5164 /* clear up stack */
5165 while ((sptr = ff_pop()) != NULL)
5166 ff_free_stack_element(sptr);
5167
5168 vim_free(ff_search_ctx->ffsc_file_to_search);
5169 vim_free(ff_search_ctx->ffsc_start_dir);
5170 vim_free(ff_search_ctx->ffsc_fix_path);
5171#ifdef FEAT_PATH_EXTRA
5172 vim_free(ff_search_ctx->ffsc_wc_path);
5173#endif
5174
5175#ifdef FEAT_PATH_EXTRA
5176 if (ff_search_ctx->ffsc_stopdirs_v != NULL)
5177 {
5178 int i = 0;
5179
5180 while (ff_search_ctx->ffsc_stopdirs_v[i] != NULL)
5181 {
5182 vim_free(ff_search_ctx->ffsc_stopdirs_v[i]);
5183 i++;
5184 }
5185 vim_free(ff_search_ctx->ffsc_stopdirs_v);
5186 }
5187 ff_search_ctx->ffsc_stopdirs_v = NULL;
5188#endif
5189
5190 /* reset everything */
5191 ff_search_ctx->ffsc_file_to_search = NULL;
5192 ff_search_ctx->ffsc_start_dir = NULL;
5193 ff_search_ctx->ffsc_fix_path = NULL;
5194#ifdef FEAT_PATH_EXTRA
5195 ff_search_ctx->ffsc_wc_path = NULL;
5196 ff_search_ctx->ffsc_level = 0;
5197#endif
5198}
5199
5200#ifdef FEAT_PATH_EXTRA
5201/*
5202 * check if the given path is in the stopdirs
5203 * returns TRUE if yes else FALSE
5204 */
5205 static int
5206ff_path_in_stoplist(path, path_len, stopdirs_v)
5207 char_u *path;
5208 int path_len;
5209 char_u **stopdirs_v;
5210{
5211 int i = 0;
5212
5213 /* eat up trailing path separators, except the first */
Bram Moolenaar1d94f9b2005-08-04 21:29:45 +00005214 while (path_len > 1 && vim_ispathsep(path[path_len - 1]))
Bram Moolenaar071d4272004-06-13 20:20:40 +00005215 path_len--;
5216
5217 /* if no path consider it as match */
5218 if (path_len == 0)
5219 return TRUE;
5220
5221 for (i = 0; stopdirs_v[i] != NULL; i++)
5222 {
5223 if ((int)STRLEN(stopdirs_v[i]) > path_len)
5224 {
5225 /* match for parent directory. So '/home' also matches
5226 * '/home/rks'. Check for PATHSEP in stopdirs_v[i], else
5227 * '/home/r' would also match '/home/rks'
5228 */
5229 if (fnamencmp(stopdirs_v[i], path, path_len) == 0
Bram Moolenaar1d94f9b2005-08-04 21:29:45 +00005230 && vim_ispathsep(stopdirs_v[i][path_len]))
Bram Moolenaar071d4272004-06-13 20:20:40 +00005231 return TRUE;
5232 }
5233 else
5234 {
5235 if (fnamecmp(stopdirs_v[i], path) == 0)
5236 return TRUE;
5237 }
5238 }
5239 return FALSE;
5240}
5241#endif
5242
5243#if defined(FEAT_SEARCHPATH) || defined(PROTO)
5244/*
5245 * Find the file name "ptr[len]" in the path.
5246 *
5247 * On the first call set the parameter 'first' to TRUE to initialize
5248 * the search. For repeating calls to FALSE.
5249 *
5250 * Repeating calls will return other files called 'ptr[len]' from the path.
5251 *
5252 * Only on the first call 'ptr' and 'len' are used. For repeating calls they
5253 * don't need valid values.
5254 *
5255 * If nothing found on the first call the option FNAME_MESS will issue the
5256 * message:
5257 * 'Can't find file "<file>" in path'
5258 * On repeating calls:
5259 * 'No more file "<file>" found in path'
5260 *
5261 * options:
5262 * FNAME_MESS give error message when not found
5263 *
5264 * Uses NameBuff[]!
5265 *
5266 * Returns an allocated string for the file name. NULL for error.
5267 *
5268 */
5269 char_u *
5270find_file_in_path(ptr, len, options, first, rel_fname)
5271 char_u *ptr; /* file name */
5272 int len; /* length of file name */
5273 int options;
5274 int first; /* use count'th matching file name */
5275 char_u *rel_fname; /* file name searching relative to */
5276{
5277 return find_file_in_path_option(ptr, len, options, first,
5278 *curbuf->b_p_path == NUL ? p_path : curbuf->b_p_path,
Bram Moolenaar433f7c82006-03-21 21:29:36 +00005279 FALSE, rel_fname, curbuf->b_p_sua);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005280}
5281
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005282static char_u *ff_file_to_find = NULL;
5283static void *fdip_search_ctx = NULL;
5284
5285#if defined(EXITFREE)
5286 static void
5287free_findfile()
5288{
5289 vim_free(ff_file_to_find);
5290 vim_findfile_cleanup(fdip_search_ctx);
5291}
5292#endif
5293
Bram Moolenaar071d4272004-06-13 20:20:40 +00005294/*
5295 * Find the directory name "ptr[len]" in the path.
5296 *
5297 * options:
5298 * FNAME_MESS give error message when not found
5299 *
5300 * Uses NameBuff[]!
5301 *
5302 * Returns an allocated string for the file name. NULL for error.
5303 */
5304 char_u *
5305find_directory_in_path(ptr, len, options, rel_fname)
5306 char_u *ptr; /* file name */
5307 int len; /* length of file name */
5308 int options;
5309 char_u *rel_fname; /* file name searching relative to */
5310{
5311 return find_file_in_path_option(ptr, len, options, TRUE, p_cdpath,
Bram Moolenaar433f7c82006-03-21 21:29:36 +00005312 TRUE, rel_fname, (char_u *)"");
Bram Moolenaar071d4272004-06-13 20:20:40 +00005313}
5314
Bram Moolenaar89cb5e02004-07-19 20:55:54 +00005315 char_u *
Bram Moolenaar433f7c82006-03-21 21:29:36 +00005316find_file_in_path_option(ptr, len, options, first, path_option, need_dir, rel_fname, suffixes)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005317 char_u *ptr; /* file name */
5318 int len; /* length of file name */
5319 int options;
5320 int first; /* use count'th matching file name */
5321 char_u *path_option; /* p_path or p_cdpath */
5322 int need_dir; /* looking for directory name */
5323 char_u *rel_fname; /* file name we are looking relative to. */
Bram Moolenaar433f7c82006-03-21 21:29:36 +00005324 char_u *suffixes; /* list of suffixes, 'suffixesadd' option */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005325{
Bram Moolenaar071d4272004-06-13 20:20:40 +00005326 static char_u *dir;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005327 static int did_findfile_init = FALSE;
5328 char_u save_char;
5329 char_u *file_name = NULL;
5330 char_u *buf = NULL;
5331 int rel_to_curdir;
5332#ifdef AMIGA
5333 struct Process *proc = (struct Process *)FindTask(0L);
5334 APTR save_winptr = proc->pr_WindowPtr;
5335
5336 /* Avoid a requester here for a volume that doesn't exist. */
5337 proc->pr_WindowPtr = (APTR)-1L;
5338#endif
5339
5340 if (first == TRUE)
5341 {
5342 /* copy file name into NameBuff, expanding environment variables */
5343 save_char = ptr[len];
5344 ptr[len] = NUL;
5345 expand_env(ptr, NameBuff, MAXPATHL);
5346 ptr[len] = save_char;
5347
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005348 vim_free(ff_file_to_find);
5349 ff_file_to_find = vim_strsave(NameBuff);
5350 if (ff_file_to_find == NULL) /* out of memory */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005351 {
5352 file_name = NULL;
5353 goto theend;
5354 }
5355 }
5356
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005357 rel_to_curdir = (ff_file_to_find[0] == '.'
5358 && (ff_file_to_find[1] == NUL
5359 || vim_ispathsep(ff_file_to_find[1])
5360 || (ff_file_to_find[1] == '.'
5361 && (ff_file_to_find[2] == NUL
5362 || vim_ispathsep(ff_file_to_find[2])))));
5363 if (vim_isAbsName(ff_file_to_find)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005364 /* "..", "../path", "." and "./path": don't use the path_option */
5365 || rel_to_curdir
5366#if defined(MSWIN) || defined(MSDOS) || defined(OS2)
5367 /* handle "\tmp" as absolute path */
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005368 || vim_ispathsep(ff_file_to_find[0])
Bram Moolenaar071d4272004-06-13 20:20:40 +00005369 /* handle "c:name" as absulute path */
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005370 || (ff_file_to_find[0] != NUL && ff_file_to_find[1] == ':')
Bram Moolenaar071d4272004-06-13 20:20:40 +00005371#endif
5372#ifdef AMIGA
5373 /* handle ":tmp" as absolute path */
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005374 || ff_file_to_find[0] == ':'
Bram Moolenaar071d4272004-06-13 20:20:40 +00005375#endif
5376 )
5377 {
5378 /*
5379 * Absolute path, no need to use "path_option".
5380 * If this is not a first call, return NULL. We already returned a
5381 * filename on the first call.
5382 */
5383 if (first == TRUE)
5384 {
5385 int l;
5386 int run;
5387
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005388 if (path_with_url(ff_file_to_find))
Bram Moolenaar071d4272004-06-13 20:20:40 +00005389 {
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005390 file_name = vim_strsave(ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005391 goto theend;
5392 }
5393
5394 /* When FNAME_REL flag given first use the directory of the file.
5395 * Otherwise or when this fails use the current directory. */
5396 for (run = 1; run <= 2; ++run)
5397 {
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005398 l = (int)STRLEN(ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005399 if (run == 1
5400 && rel_to_curdir
5401 && (options & FNAME_REL)
5402 && rel_fname != NULL
5403 && STRLEN(rel_fname) + l < MAXPATHL)
5404 {
5405 STRCPY(NameBuff, rel_fname);
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005406 STRCPY(gettail(NameBuff), ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005407 l = (int)STRLEN(NameBuff);
5408 }
5409 else
5410 {
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005411 STRCPY(NameBuff, ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005412 run = 2;
5413 }
5414
5415 /* When the file doesn't exist, try adding parts of
5416 * 'suffixesadd'. */
Bram Moolenaar433f7c82006-03-21 21:29:36 +00005417 buf = suffixes;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005418 for (;;)
5419 {
5420 if (
5421#ifdef DJGPP
5422 /* "C:" by itself will fail for mch_getperm(),
5423 * assume it's always valid. */
5424 (need_dir && NameBuff[0] != NUL
5425 && NameBuff[1] == ':'
5426 && NameBuff[2] == NUL) ||
5427#endif
5428 (mch_getperm(NameBuff) >= 0
5429 && (!need_dir || mch_isdir(NameBuff))))
5430 {
5431 file_name = vim_strsave(NameBuff);
5432 goto theend;
5433 }
5434 if (*buf == NUL)
5435 break;
5436 copy_option_part(&buf, NameBuff + l, MAXPATHL - l, ",");
5437 }
5438 }
5439 }
5440 }
5441 else
5442 {
5443 /*
5444 * Loop over all paths in the 'path' or 'cdpath' option.
5445 * When "first" is set, first setup to the start of the option.
5446 * Otherwise continue to find the next match.
5447 */
5448 if (first == TRUE)
5449 {
5450 /* vim_findfile_free_visited can handle a possible NULL pointer */
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005451 vim_findfile_free_visited(fdip_search_ctx);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005452 dir = path_option;
5453 did_findfile_init = FALSE;
5454 }
5455
5456 for (;;)
5457 {
5458 if (did_findfile_init)
5459 {
5460 ff_search_ctx->ffsc_need_dir = need_dir;
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005461 file_name = vim_findfile(fdip_search_ctx);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005462 ff_search_ctx->ffsc_need_dir = FALSE;
5463 if (file_name != NULL)
5464 break;
5465
5466 did_findfile_init = FALSE;
5467 }
5468 else
5469 {
5470 char_u *r_ptr;
5471
5472 if (dir == NULL || *dir == NUL)
5473 {
5474 /* We searched all paths of the option, now we can
5475 * free the search context. */
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005476 vim_findfile_cleanup(fdip_search_ctx);
5477 fdip_search_ctx = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005478 break;
5479 }
5480
5481 if ((buf = alloc((int)(MAXPATHL))) == NULL)
5482 break;
5483
5484 /* copy next path */
5485 buf[0] = 0;
5486 copy_option_part(&dir, buf, MAXPATHL, " ,");
5487
5488#ifdef FEAT_PATH_EXTRA
5489 /* get the stopdir string */
5490 r_ptr = vim_findfile_stopdir(buf);
5491#else
5492 r_ptr = NULL;
5493#endif
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005494 fdip_search_ctx = vim_findfile_init(buf, ff_file_to_find,
5495 r_ptr, 100, FALSE, TRUE,
5496 fdip_search_ctx, FALSE, rel_fname);
5497 if (fdip_search_ctx != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005498 did_findfile_init = TRUE;
5499 vim_free(buf);
5500 }
5501 }
5502 }
5503 if (file_name == NULL && (options & FNAME_MESS))
5504 {
5505 if (first == TRUE)
5506 {
5507 if (need_dir)
5508 EMSG2(_("E344: Can't find directory \"%s\" in cdpath"),
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005509 ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005510 else
5511 EMSG2(_("E345: Can't find file \"%s\" in path"),
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005512 ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005513 }
5514 else
5515 {
5516 if (need_dir)
5517 EMSG2(_("E346: No more directory \"%s\" found in cdpath"),
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005518 ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005519 else
5520 EMSG2(_("E347: No more file \"%s\" found in path"),
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005521 ff_file_to_find);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005522 }
5523 }
5524
5525theend:
5526#ifdef AMIGA
5527 proc->pr_WindowPtr = save_winptr;
5528#endif
5529 return file_name;
5530}
5531
5532#endif /* FEAT_SEARCHPATH */
5533
5534/*
5535 * Change directory to "new_dir". If FEAT_SEARCHPATH is defined, search
5536 * 'cdpath' for relative directory names, otherwise just mch_chdir().
5537 */
5538 int
5539vim_chdir(new_dir)
5540 char_u *new_dir;
5541{
5542#ifndef FEAT_SEARCHPATH
5543 return mch_chdir((char *)new_dir);
5544#else
5545 char_u *dir_name;
5546 int r;
5547
5548 dir_name = find_directory_in_path(new_dir, (int)STRLEN(new_dir),
5549 FNAME_MESS, curbuf->b_ffname);
5550 if (dir_name == NULL)
5551 return -1;
5552 r = mch_chdir((char *)dir_name);
5553 vim_free(dir_name);
5554 return r;
5555#endif
5556}
5557
5558/*
Bram Moolenaarbbebc852005-07-18 21:47:53 +00005559 * Get user name from machine-specific function.
Bram Moolenaar071d4272004-06-13 20:20:40 +00005560 * Returns the user name in "buf[len]".
Bram Moolenaarbbebc852005-07-18 21:47:53 +00005561 * Some systems are quite slow in obtaining the user name (Windows NT), thus
5562 * cache the result.
Bram Moolenaar071d4272004-06-13 20:20:40 +00005563 * Returns OK or FAIL.
5564 */
5565 int
5566get_user_name(buf, len)
5567 char_u *buf;
5568 int len;
5569{
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005570 if (username == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005571 {
5572 if (mch_get_user_name(buf, len) == FAIL)
5573 return FAIL;
Bram Moolenaarf461c8e2005-06-25 23:04:51 +00005574 username = vim_strsave(buf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005575 }
5576 else
Bram Moolenaarbbebc852005-07-18 21:47:53 +00005577 vim_strncpy(buf, username, len - 1);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005578 return OK;
5579}
5580
5581#ifndef HAVE_QSORT
5582/*
5583 * Our own qsort(), for systems that don't have it.
5584 * It's simple and slow. From the K&R C book.
5585 */
5586 void
5587qsort(base, elm_count, elm_size, cmp)
5588 void *base;
5589 size_t elm_count;
5590 size_t elm_size;
5591 int (*cmp) __ARGS((const void *, const void *));
5592{
5593 char_u *buf;
5594 char_u *p1;
5595 char_u *p2;
5596 int i, j;
5597 int gap;
5598
5599 buf = alloc((unsigned)elm_size);
5600 if (buf == NULL)
5601 return;
5602
5603 for (gap = elm_count / 2; gap > 0; gap /= 2)
5604 for (i = gap; i < elm_count; ++i)
5605 for (j = i - gap; j >= 0; j -= gap)
5606 {
5607 /* Compare the elements. */
5608 p1 = (char_u *)base + j * elm_size;
5609 p2 = (char_u *)base + (j + gap) * elm_size;
5610 if ((*cmp)((void *)p1, (void *)p2) <= 0)
5611 break;
5612 /* Exchange the elemets. */
5613 mch_memmove(buf, p1, elm_size);
5614 mch_memmove(p1, p2, elm_size);
5615 mch_memmove(p2, buf, elm_size);
5616 }
5617
5618 vim_free(buf);
5619}
5620#endif
5621
Bram Moolenaar071d4272004-06-13 20:20:40 +00005622/*
5623 * Sort an array of strings.
5624 */
5625static int
5626#ifdef __BORLANDC__
5627_RTLENTRYF
5628#endif
5629sort_compare __ARGS((const void *s1, const void *s2));
5630
5631 static int
5632#ifdef __BORLANDC__
5633_RTLENTRYF
5634#endif
5635sort_compare(s1, s2)
5636 const void *s1;
5637 const void *s2;
5638{
5639 return STRCMP(*(char **)s1, *(char **)s2);
5640}
5641
5642 void
5643sort_strings(files, count)
5644 char_u **files;
5645 int count;
5646{
5647 qsort((void *)files, (size_t)count, sizeof(char_u *), sort_compare);
5648}
Bram Moolenaar071d4272004-06-13 20:20:40 +00005649
5650#if !defined(NO_EXPANDPATH) || defined(PROTO)
5651/*
5652 * Compare path "p[]" to "q[]".
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00005653 * If "maxlen" >= 0 compare "p[maxlen]" to "q[maxlen]"
Bram Moolenaar071d4272004-06-13 20:20:40 +00005654 * Return value like strcmp(p, q), but consider path separators.
5655 */
5656 int
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00005657pathcmp(p, q, maxlen)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005658 const char *p, *q;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00005659 int maxlen;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005660{
5661 int i;
Bram Moolenaar86b68352004-12-27 21:59:20 +00005662 const char *s = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005663
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00005664 for (i = 0; maxlen < 0 || i < maxlen; ++i)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005665 {
5666 /* End of "p": check if "q" also ends or just has a slash. */
5667 if (p[i] == NUL)
5668 {
5669 if (q[i] == NUL) /* full match */
5670 return 0;
5671 s = q;
5672 break;
5673 }
5674
5675 /* End of "q": check if "p" just has a slash. */
5676 if (q[i] == NUL)
5677 {
5678 s = p;
5679 break;
5680 }
5681
5682 if (
5683#ifdef CASE_INSENSITIVE_FILENAME
5684 TOUPPER_LOC(p[i]) != TOUPPER_LOC(q[i])
5685#else
5686 p[i] != q[i]
5687#endif
5688#ifdef BACKSLASH_IN_FILENAME
5689 /* consider '/' and '\\' to be equal */
5690 && !((p[i] == '/' && q[i] == '\\')
5691 || (p[i] == '\\' && q[i] == '/'))
5692#endif
5693 )
5694 {
5695 if (vim_ispathsep(p[i]))
5696 return -1;
5697 if (vim_ispathsep(q[i]))
5698 return 1;
5699 return ((char_u *)p)[i] - ((char_u *)q)[i]; /* no match */
5700 }
5701 }
Bram Moolenaar86b68352004-12-27 21:59:20 +00005702 if (s == NULL) /* "i" ran into "maxlen" */
5703 return 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005704
5705 /* ignore a trailing slash, but not "//" or ":/" */
Bram Moolenaar86b68352004-12-27 21:59:20 +00005706 if (s[i + 1] == NUL
5707 && i > 0
5708 && !after_pathsep((char_u *)s, (char_u *)s + i)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005709#ifdef BACKSLASH_IN_FILENAME
Bram Moolenaar86b68352004-12-27 21:59:20 +00005710 && (s[i] == '/' || s[i] == '\\')
Bram Moolenaar071d4272004-06-13 20:20:40 +00005711#else
Bram Moolenaar86b68352004-12-27 21:59:20 +00005712 && s[i] == '/'
Bram Moolenaar071d4272004-06-13 20:20:40 +00005713#endif
Bram Moolenaar86b68352004-12-27 21:59:20 +00005714 )
Bram Moolenaar071d4272004-06-13 20:20:40 +00005715 return 0; /* match with trailing slash */
5716 if (s == q)
5717 return -1; /* no match */
5718 return 1;
5719}
5720#endif
5721
Bram Moolenaar071d4272004-06-13 20:20:40 +00005722/*
5723 * The putenv() implementation below comes from the "screen" program.
5724 * Included with permission from Juergen Weigert.
5725 * See pty.c for the copyright notice.
5726 */
5727
5728/*
5729 * putenv -- put value into environment
5730 *
5731 * Usage: i = putenv (string)
5732 * int i;
5733 * char *string;
5734 *
5735 * where string is of the form <name>=<value>.
5736 * Putenv returns 0 normally, -1 on error (not enough core for malloc).
5737 *
5738 * Putenv may need to add a new name into the environment, or to
5739 * associate a value longer than the current value with a particular
5740 * name. So, to make life simpler, putenv() copies your entire
5741 * environment into the heap (i.e. malloc()) from the stack
5742 * (i.e. where it resides when your process is initiated) the first
5743 * time you call it.
5744 *
5745 * (history removed, not very interesting. See the "screen" sources.)
5746 */
5747
5748#if !defined(HAVE_SETENV) && !defined(HAVE_PUTENV)
5749
5750#define EXTRASIZE 5 /* increment to add to env. size */
5751
5752static int envsize = -1; /* current size of environment */
5753#ifndef MACOS_CLASSIC
5754extern
5755#endif
5756 char **environ; /* the global which is your env. */
5757
5758static int findenv __ARGS((char *name)); /* look for a name in the env. */
5759static int newenv __ARGS((void)); /* copy env. from stack to heap */
5760static int moreenv __ARGS((void)); /* incr. size of env. */
5761
5762 int
5763putenv(string)
5764 const char *string;
5765{
5766 int i;
5767 char *p;
5768
5769 if (envsize < 0)
5770 { /* first time putenv called */
5771 if (newenv() < 0) /* copy env. to heap */
5772 return -1;
5773 }
5774
5775 i = findenv((char *)string); /* look for name in environment */
5776
5777 if (i < 0)
5778 { /* name must be added */
5779 for (i = 0; environ[i]; i++);
5780 if (i >= (envsize - 1))
5781 { /* need new slot */
5782 if (moreenv() < 0)
5783 return -1;
5784 }
5785 p = (char *)alloc((unsigned)(strlen(string) + 1));
5786 if (p == NULL) /* not enough core */
5787 return -1;
5788 environ[i + 1] = 0; /* new end of env. */
5789 }
5790 else
5791 { /* name already in env. */
5792 p = vim_realloc(environ[i], strlen(string) + 1);
5793 if (p == NULL)
5794 return -1;
5795 }
5796 sprintf(p, "%s", string); /* copy into env. */
5797 environ[i] = p;
5798
5799 return 0;
5800}
5801
5802 static int
5803findenv(name)
5804 char *name;
5805{
5806 char *namechar, *envchar;
5807 int i, found;
5808
5809 found = 0;
5810 for (i = 0; environ[i] && !found; i++)
5811 {
5812 envchar = environ[i];
5813 namechar = name;
5814 while (*namechar && *namechar != '=' && (*namechar == *envchar))
5815 {
5816 namechar++;
5817 envchar++;
5818 }
5819 found = ((*namechar == '\0' || *namechar == '=') && *envchar == '=');
5820 }
5821 return found ? i - 1 : -1;
5822}
5823
5824 static int
5825newenv()
5826{
5827 char **env, *elem;
5828 int i, esize;
5829
5830#ifdef MACOS
5831 /* for Mac a new, empty environment is created */
5832 i = 0;
5833#else
5834 for (i = 0; environ[i]; i++)
5835 ;
5836#endif
5837 esize = i + EXTRASIZE + 1;
5838 env = (char **)alloc((unsigned)(esize * sizeof (elem)));
5839 if (env == NULL)
5840 return -1;
5841
5842#ifndef MACOS
5843 for (i = 0; environ[i]; i++)
5844 {
5845 elem = (char *)alloc((unsigned)(strlen(environ[i]) + 1));
5846 if (elem == NULL)
5847 return -1;
5848 env[i] = elem;
5849 strcpy(elem, environ[i]);
5850 }
5851#endif
5852
5853 env[i] = 0;
5854 environ = env;
5855 envsize = esize;
5856 return 0;
5857}
5858
5859 static int
5860moreenv()
5861{
5862 int esize;
5863 char **env;
5864
5865 esize = envsize + EXTRASIZE;
5866 env = (char **)vim_realloc((char *)environ, esize * sizeof (*env));
5867 if (env == 0)
5868 return -1;
5869 environ = env;
5870 envsize = esize;
5871 return 0;
5872}
5873
5874# ifdef USE_VIMPTY_GETENV
5875 char_u *
5876vimpty_getenv(string)
5877 const char_u *string;
5878{
5879 int i;
5880 char_u *p;
5881
5882 if (envsize < 0)
5883 return NULL;
5884
5885 i = findenv((char *)string);
5886
5887 if (i < 0)
5888 return NULL;
5889
5890 p = vim_strchr((char_u *)environ[i], '=');
5891 return (p + 1);
5892}
5893# endif
5894
5895#endif /* !defined(HAVE_SETENV) && !defined(HAVE_PUTENV) */
Bram Moolenaarc4a06d32005-06-07 21:04:49 +00005896
Bram Moolenaarc4956c82006-03-12 21:58:43 +00005897#if defined(FEAT_EVAL) || defined(FEAT_SPELL) || defined(PROTO)
Bram Moolenaarc4a06d32005-06-07 21:04:49 +00005898/*
5899 * Return 0 for not writable, 1 for writable file, 2 for a dir which we have
5900 * rights to write into.
5901 */
5902 int
5903filewritable(fname)
5904 char_u *fname;
5905{
5906 int retval = 0;
5907#if defined(UNIX) || defined(VMS)
5908 int perm = 0;
5909#endif
5910
5911#if defined(UNIX) || defined(VMS)
5912 perm = mch_getperm(fname);
5913#endif
5914#ifndef MACOS_CLASSIC /* TODO: get either mch_writable or mch_access */
5915 if (
5916# ifdef WIN3264
5917 mch_writable(fname) &&
5918# else
5919# if defined(UNIX) || defined(VMS)
5920 (perm & 0222) &&
5921# endif
5922# endif
5923 mch_access((char *)fname, W_OK) == 0
5924 )
5925#endif
5926 {
5927 ++retval;
5928 if (mch_isdir(fname))
5929 ++retval;
5930 }
5931 return retval;
5932}
5933#endif
Bram Moolenaar6bab4d12005-06-16 21:53:56 +00005934
5935/*
5936 * Print an error message with one or two "%s" and one or two string arguments.
5937 * This is not in message.c to avoid a warning for prototypes.
5938 */
5939 int
5940emsg3(s, a1, a2)
5941 char_u *s, *a1, *a2;
5942{
Bram Moolenaareb3593b2006-04-22 22:33:57 +00005943 if (emsg_not_now())
Bram Moolenaar6bab4d12005-06-16 21:53:56 +00005944 return TRUE; /* no error messages at the moment */
Bram Moolenaar6f192452007-11-08 19:49:02 +00005945#ifdef HAVE_STDARG_H
5946 vim_snprintf((char *)IObuff, IOSIZE, (char *)s, a1, a2);
5947#else
Bram Moolenaareb3593b2006-04-22 22:33:57 +00005948 vim_snprintf((char *)IObuff, IOSIZE, (char *)s, (long_u)a1, (long_u)a2);
Bram Moolenaar6f192452007-11-08 19:49:02 +00005949#endif
Bram Moolenaar6bab4d12005-06-16 21:53:56 +00005950 return emsg(IObuff);
5951}
5952
5953/*
5954 * Print an error message with one "%ld" and one long int argument.
5955 * This is not in message.c to avoid a warning for prototypes.
5956 */
5957 int
5958emsgn(s, n)
5959 char_u *s;
5960 long n;
5961{
Bram Moolenaareb3593b2006-04-22 22:33:57 +00005962 if (emsg_not_now())
Bram Moolenaar6bab4d12005-06-16 21:53:56 +00005963 return TRUE; /* no error messages at the moment */
5964 vim_snprintf((char *)IObuff, IOSIZE, (char *)s, n);
5965 return emsg(IObuff);
5966}