blob: 353ae8981f34870273bad4caa85c3e521fc6478b [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 * ex_cmds2.c: some more functions for command line commands
12 */
13
Bram Moolenaar071d4272004-06-13 20:20:40 +000014#include "vim.h"
Bram Moolenaar071d4272004-06-13 20:20:40 +000015#include "version.h"
16
17static void cmd_source __ARGS((char_u *fname, exarg_T *eap));
18
Bram Moolenaar05159a02005-02-26 23:04:13 +000019#ifdef FEAT_EVAL
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +000020/* Growarray to store info about already sourced scripts.
Bram Moolenaar05159a02005-02-26 23:04:13 +000021 * For Unix also store the dev/ino, so that we don't have to stat() each
22 * script when going through the list. */
23typedef struct scriptitem_S
24{
25 char_u *sn_name;
26# ifdef UNIX
Bram Moolenaarbf0c4522009-05-16 19:16:33 +000027 int sn_dev_valid;
28 dev_t sn_dev;
Bram Moolenaar05159a02005-02-26 23:04:13 +000029 ino_t sn_ino;
30# endif
31# ifdef FEAT_PROFILE
32 int sn_prof_on; /* TRUE when script is/was profiled */
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +000033 int sn_pr_force; /* forceit: profile functions in this script */
Bram Moolenaar05159a02005-02-26 23:04:13 +000034 proftime_T sn_pr_child; /* time set when going into first child */
35 int sn_pr_nest; /* nesting for sn_pr_child */
36 /* profiling the script as a whole */
37 int sn_pr_count; /* nr of times sourced */
Bram Moolenaar8c8de832008-06-24 22:58:06 +000038 proftime_T sn_pr_total; /* time spent in script + children */
39 proftime_T sn_pr_self; /* time spent in script itself */
Bram Moolenaar05159a02005-02-26 23:04:13 +000040 proftime_T sn_pr_start; /* time at script start */
41 proftime_T sn_pr_children; /* time in children after script start */
42 /* profiling the script per line */
43 garray_T sn_prl_ga; /* things stored for every line */
44 proftime_T sn_prl_start; /* start time for current line */
45 proftime_T sn_prl_children; /* time spent in children for this line */
46 proftime_T sn_prl_wait; /* wait start time for current line */
47 int sn_prl_idx; /* index of line being timed; -1 if none */
48 int sn_prl_execed; /* line being timed was executed */
49# endif
50} scriptitem_T;
51
52static garray_T script_items = {0, 0, sizeof(scriptitem_T), 4, NULL};
53#define SCRIPT_ITEM(id) (((scriptitem_T *)script_items.ga_data)[(id) - 1])
54
55# ifdef FEAT_PROFILE
56/* Struct used in sn_prl_ga for every line of a script. */
57typedef struct sn_prl_S
58{
59 int snp_count; /* nr of times line was executed */
Bram Moolenaar8c8de832008-06-24 22:58:06 +000060 proftime_T sn_prl_total; /* time spent in a line + children */
61 proftime_T sn_prl_self; /* time spent in a line itself */
Bram Moolenaar05159a02005-02-26 23:04:13 +000062} sn_prl_T;
63
64# define PRL_ITEM(si, idx) (((sn_prl_T *)(si)->sn_prl_ga.ga_data)[(idx)])
65# endif
66#endif
67
Bram Moolenaar071d4272004-06-13 20:20:40 +000068#if defined(FEAT_EVAL) || defined(PROTO)
69static int debug_greedy = FALSE; /* batch mode debugging: don't save
70 and restore typeahead. */
71
72/*
73 * do_debug(): Debug mode.
74 * Repeatedly get Ex commands, until told to continue normal execution.
75 */
76 void
77do_debug(cmd)
78 char_u *cmd;
79{
80 int save_msg_scroll = msg_scroll;
81 int save_State = State;
82 int save_did_emsg = did_emsg;
83 int save_cmd_silent = cmd_silent;
84 int save_msg_silent = msg_silent;
85 int save_emsg_silent = emsg_silent;
86 int save_redir_off = redir_off;
87 tasave_T typeaheadbuf;
Bram Moolenaaree3f7a52008-01-01 13:17:56 +000088 int typeahead_saved = FALSE;
Bram Moolenaar383c6f52008-01-04 15:01:07 +000089 int save_ignore_script = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +000090# ifdef FEAT_EX_EXTRA
91 int save_ex_normal_busy;
92# endif
93 int n;
94 char_u *cmdline = NULL;
95 char_u *p;
96 char *tail = NULL;
97 static int last_cmd = 0;
98#define CMD_CONT 1
99#define CMD_NEXT 2
100#define CMD_STEP 3
101#define CMD_FINISH 4
102#define CMD_QUIT 5
103#define CMD_INTERRUPT 6
104
105#ifdef ALWAYS_USE_GUI
106 /* Can't do this when there is no terminal for input/output. */
107 if (!gui.in_use)
108 {
109 /* Break as soon as possible. */
110 debug_break_level = 9999;
111 return;
112 }
113#endif
114
115 /* Make sure we are in raw mode and start termcap mode. Might have side
116 * effects... */
117 settmode(TMODE_RAW);
118 starttermcap();
119
120 ++RedrawingDisabled; /* don't redisplay the window */
121 ++no_wait_return; /* don't wait for return */
122 did_emsg = FALSE; /* don't use error from debugged stuff */
123 cmd_silent = FALSE; /* display commands */
124 msg_silent = FALSE; /* display messages */
125 emsg_silent = FALSE; /* display error messages */
126 redir_off = TRUE; /* don't redirect debug commands */
127
128 State = NORMAL;
129#ifdef FEAT_SNIFF
130 want_sniff_request = 0; /* No K_SNIFF wanted */
131#endif
132
133 if (!debug_did_msg)
134 MSG(_("Entering Debug mode. Type \"cont\" to continue."));
135 if (sourcing_name != NULL)
136 msg(sourcing_name);
137 if (sourcing_lnum != 0)
Bram Moolenaar555b2802005-05-19 21:08:39 +0000138 smsg((char_u *)_("line %ld: %s"), (long)sourcing_lnum, cmd);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000139 else
Bram Moolenaar555b2802005-05-19 21:08:39 +0000140 smsg((char_u *)_("cmd: %s"), cmd);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000141
142 /*
143 * Repeat getting a command and executing it.
144 */
145 for (;;)
146 {
147 msg_scroll = TRUE;
148 need_wait_return = FALSE;
149#ifdef FEAT_SNIFF
150 ProcessSniffRequests();
151#endif
152 /* Save the current typeahead buffer and replace it with an empty one.
153 * This makes sure we get input from the user here and don't interfere
154 * with the commands being executed. Reset "ex_normal_busy" to avoid
155 * the side effects of using ":normal". Save the stuff buffer and make
Bram Moolenaaree3f7a52008-01-01 13:17:56 +0000156 * it empty. Set ignore_script to avoid reading from script input. */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000157# ifdef FEAT_EX_EXTRA
158 save_ex_normal_busy = ex_normal_busy;
159 ex_normal_busy = 0;
160# endif
161 if (!debug_greedy)
Bram Moolenaaree3f7a52008-01-01 13:17:56 +0000162 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000163 save_typeahead(&typeaheadbuf);
Bram Moolenaaree3f7a52008-01-01 13:17:56 +0000164 typeahead_saved = TRUE;
165 save_ignore_script = ignore_script;
166 ignore_script = TRUE;
167 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000168
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +0000169 cmdline = getcmdline_prompt('>', NULL, 0, EXPAND_NOTHING, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000170
Bram Moolenaaree3f7a52008-01-01 13:17:56 +0000171 if (typeahead_saved)
172 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000173 restore_typeahead(&typeaheadbuf);
Bram Moolenaaree3f7a52008-01-01 13:17:56 +0000174 ignore_script = save_ignore_script;
175 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000176# ifdef FEAT_EX_EXTRA
177 ex_normal_busy = save_ex_normal_busy;
178# endif
179
180 cmdline_row = msg_row;
181 if (cmdline != NULL)
182 {
183 /* If this is a debug command, set "last_cmd".
184 * If not, reset "last_cmd".
185 * For a blank line use previous command. */
186 p = skipwhite(cmdline);
187 if (*p != NUL)
188 {
189 switch (*p)
190 {
191 case 'c': last_cmd = CMD_CONT;
192 tail = "ont";
193 break;
194 case 'n': last_cmd = CMD_NEXT;
195 tail = "ext";
196 break;
197 case 's': last_cmd = CMD_STEP;
198 tail = "tep";
199 break;
200 case 'f': last_cmd = CMD_FINISH;
201 tail = "inish";
202 break;
203 case 'q': last_cmd = CMD_QUIT;
204 tail = "uit";
205 break;
206 case 'i': last_cmd = CMD_INTERRUPT;
207 tail = "nterrupt";
208 break;
209 default: last_cmd = 0;
210 }
211 if (last_cmd != 0)
212 {
213 /* Check that the tail matches. */
214 ++p;
215 while (*p != NUL && *p == *tail)
216 {
217 ++p;
218 ++tail;
219 }
220 if (ASCII_ISALPHA(*p))
221 last_cmd = 0;
222 }
223 }
224
225 if (last_cmd != 0)
226 {
227 /* Execute debug command: decided where to break next and
228 * return. */
229 switch (last_cmd)
230 {
231 case CMD_CONT:
232 debug_break_level = -1;
233 break;
234 case CMD_NEXT:
235 debug_break_level = ex_nesting_level;
236 break;
237 case CMD_STEP:
238 debug_break_level = 9999;
239 break;
240 case CMD_FINISH:
241 debug_break_level = ex_nesting_level - 1;
242 break;
243 case CMD_QUIT:
244 got_int = TRUE;
245 debug_break_level = -1;
246 break;
247 case CMD_INTERRUPT:
248 got_int = TRUE;
249 debug_break_level = 9999;
250 /* Do not repeat ">interrupt" cmd, continue stepping. */
251 last_cmd = CMD_STEP;
252 break;
253 }
254 break;
255 }
256
257 /* don't debug this command */
258 n = debug_break_level;
259 debug_break_level = -1;
260 (void)do_cmdline(cmdline, getexline, NULL,
261 DOCMD_VERBOSE|DOCMD_EXCRESET);
262 debug_break_level = n;
263
264 vim_free(cmdline);
265 }
266 lines_left = Rows - 1;
267 }
268 vim_free(cmdline);
269
270 --RedrawingDisabled;
271 --no_wait_return;
272 redraw_all_later(NOT_VALID);
273 need_wait_return = FALSE;
274 msg_scroll = save_msg_scroll;
275 lines_left = Rows - 1;
276 State = save_State;
277 did_emsg = save_did_emsg;
278 cmd_silent = save_cmd_silent;
279 msg_silent = save_msg_silent;
280 emsg_silent = save_emsg_silent;
281 redir_off = save_redir_off;
282
283 /* Only print the message again when typing a command before coming back
284 * here. */
285 debug_did_msg = TRUE;
286}
287
288/*
289 * ":debug".
290 */
291 void
292ex_debug(eap)
293 exarg_T *eap;
294{
295 int debug_break_level_save = debug_break_level;
296
297 debug_break_level = 9999;
298 do_cmdline_cmd(eap->arg);
299 debug_break_level = debug_break_level_save;
300}
301
302static char_u *debug_breakpoint_name = NULL;
303static linenr_T debug_breakpoint_lnum;
304
305/*
306 * When debugging or a breakpoint is set on a skipped command, no debug prompt
307 * is shown by do_one_cmd(). This situation is indicated by debug_skipped, and
308 * debug_skipped_name is then set to the source name in the breakpoint case. If
309 * a skipped command decides itself that a debug prompt should be displayed, it
310 * can do so by calling dbg_check_skipped().
311 */
312static int debug_skipped;
313static char_u *debug_skipped_name;
314
315/*
316 * Go to debug mode when a breakpoint was encountered or "ex_nesting_level" is
317 * at or below the break level. But only when the line is actually
318 * executed. Return TRUE and set breakpoint_name for skipped commands that
319 * decide to execute something themselves.
320 * Called from do_one_cmd() before executing a command.
321 */
322 void
323dbg_check_breakpoint(eap)
324 exarg_T *eap;
325{
326 char_u *p;
327
328 debug_skipped = FALSE;
329 if (debug_breakpoint_name != NULL)
330 {
331 if (!eap->skip)
332 {
333 /* replace K_SNR with "<SNR>" */
334 if (debug_breakpoint_name[0] == K_SPECIAL
335 && debug_breakpoint_name[1] == KS_EXTRA
336 && debug_breakpoint_name[2] == (int)KE_SNR)
337 p = (char_u *)"<SNR>";
338 else
339 p = (char_u *)"";
Bram Moolenaar555b2802005-05-19 21:08:39 +0000340 smsg((char_u *)_("Breakpoint in \"%s%s\" line %ld"),
341 p,
Bram Moolenaar071d4272004-06-13 20:20:40 +0000342 debug_breakpoint_name + (*p == NUL ? 0 : 3),
343 (long)debug_breakpoint_lnum);
344 debug_breakpoint_name = NULL;
345 do_debug(eap->cmd);
346 }
347 else
348 {
349 debug_skipped = TRUE;
350 debug_skipped_name = debug_breakpoint_name;
351 debug_breakpoint_name = NULL;
352 }
353 }
354 else if (ex_nesting_level <= debug_break_level)
355 {
356 if (!eap->skip)
357 do_debug(eap->cmd);
358 else
359 {
360 debug_skipped = TRUE;
361 debug_skipped_name = NULL;
362 }
363 }
364}
365
366/*
367 * Go to debug mode if skipped by dbg_check_breakpoint() because eap->skip was
368 * set. Return TRUE when the debug mode is entered this time.
369 */
370 int
371dbg_check_skipped(eap)
372 exarg_T *eap;
373{
374 int prev_got_int;
375
376 if (debug_skipped)
377 {
378 /*
379 * Save the value of got_int and reset it. We don't want a previous
380 * interruption cause flushing the input buffer.
381 */
382 prev_got_int = got_int;
383 got_int = FALSE;
384 debug_breakpoint_name = debug_skipped_name;
385 /* eap->skip is TRUE */
386 eap->skip = FALSE;
387 (void)dbg_check_breakpoint(eap);
388 eap->skip = TRUE;
389 got_int |= prev_got_int;
390 return TRUE;
391 }
392 return FALSE;
393}
394
395/*
396 * The list of breakpoints: dbg_breakp.
397 * This is a grow-array of structs.
398 */
399struct debuggy
400{
401 int dbg_nr; /* breakpoint number */
402 int dbg_type; /* DBG_FUNC or DBG_FILE */
403 char_u *dbg_name; /* function or file name */
404 regprog_T *dbg_prog; /* regexp program */
405 linenr_T dbg_lnum; /* line number in function or file */
Bram Moolenaar05159a02005-02-26 23:04:13 +0000406 int dbg_forceit; /* ! used */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000407};
408
409static garray_T dbg_breakp = {0, 0, sizeof(struct debuggy), 4, NULL};
Bram Moolenaar05159a02005-02-26 23:04:13 +0000410#define BREAKP(idx) (((struct debuggy *)dbg_breakp.ga_data)[idx])
411#define DEBUGGY(gap, idx) (((struct debuggy *)gap->ga_data)[idx])
Bram Moolenaar071d4272004-06-13 20:20:40 +0000412static int last_breakp = 0; /* nr of last defined breakpoint */
413
Bram Moolenaar05159a02005-02-26 23:04:13 +0000414#ifdef FEAT_PROFILE
415/* Profiling uses file and func names similar to breakpoints. */
416static garray_T prof_ga = {0, 0, sizeof(struct debuggy), 4, NULL};
417#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000418#define DBG_FUNC 1
419#define DBG_FILE 2
420
Bram Moolenaar05159a02005-02-26 23:04:13 +0000421static int dbg_parsearg __ARGS((char_u *arg, garray_T *gap));
422static linenr_T debuggy_find __ARGS((int file,char_u *fname, linenr_T after, garray_T *gap, int *fp));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000423
424/*
Bram Moolenaar05159a02005-02-26 23:04:13 +0000425 * Parse the arguments of ":profile", ":breakadd" or ":breakdel" and put them
426 * in the entry just after the last one in dbg_breakp. Note that "dbg_name"
427 * is allocated.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000428 * Returns FAIL for failure.
429 */
430 static int
Bram Moolenaar05159a02005-02-26 23:04:13 +0000431dbg_parsearg(arg, gap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000432 char_u *arg;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000433 garray_T *gap; /* either &dbg_breakp or &prof_ga */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000434{
435 char_u *p = arg;
436 char_u *q;
437 struct debuggy *bp;
Bram Moolenaarf4b8e572004-06-24 15:53:16 +0000438 int here = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000439
Bram Moolenaar05159a02005-02-26 23:04:13 +0000440 if (ga_grow(gap, 1) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000441 return FAIL;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000442 bp = &DEBUGGY(gap, gap->ga_len);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000443
444 /* Find "func" or "file". */
445 if (STRNCMP(p, "func", 4) == 0)
446 bp->dbg_type = DBG_FUNC;
447 else if (STRNCMP(p, "file", 4) == 0)
448 bp->dbg_type = DBG_FILE;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000449 else if (
450#ifdef FEAT_PROFILE
451 gap != &prof_ga &&
452#endif
453 STRNCMP(p, "here", 4) == 0)
Bram Moolenaarf4b8e572004-06-24 15:53:16 +0000454 {
455 if (curbuf->b_ffname == NULL)
456 {
457 EMSG(_(e_noname));
458 return FAIL;
459 }
460 bp->dbg_type = DBG_FILE;
461 here = TRUE;
462 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000463 else
464 {
465 EMSG2(_(e_invarg2), p);
466 return FAIL;
467 }
468 p = skipwhite(p + 4);
469
470 /* Find optional line number. */
Bram Moolenaarf4b8e572004-06-24 15:53:16 +0000471 if (here)
472 bp->dbg_lnum = curwin->w_cursor.lnum;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000473 else if (
474#ifdef FEAT_PROFILE
475 gap != &prof_ga &&
476#endif
477 VIM_ISDIGIT(*p))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000478 {
479 bp->dbg_lnum = getdigits(&p);
480 p = skipwhite(p);
481 }
482 else
483 bp->dbg_lnum = 0;
484
485 /* Find the function or file name. Don't accept a function name with (). */
Bram Moolenaarf4b8e572004-06-24 15:53:16 +0000486 if ((!here && *p == NUL)
487 || (here && *p != NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000488 || (bp->dbg_type == DBG_FUNC && strstr((char *)p, "()") != NULL))
489 {
490 EMSG2(_(e_invarg2), arg);
491 return FAIL;
492 }
493
494 if (bp->dbg_type == DBG_FUNC)
495 bp->dbg_name = vim_strsave(p);
Bram Moolenaarf4b8e572004-06-24 15:53:16 +0000496 else if (here)
497 bp->dbg_name = vim_strsave(curbuf->b_ffname);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000498 else
499 {
500 /* Expand the file name in the same way as do_source(). This means
501 * doing it twice, so that $DIR/file gets expanded when $DIR is
502 * "~/dir". */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000503 q = expand_env_save(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000504 if (q == NULL)
505 return FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000506 p = expand_env_save(q);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000507 vim_free(q);
508 if (p == NULL)
509 return FAIL;
Bram Moolenaar843ee412004-06-30 16:16:41 +0000510 if (*p != '*')
511 {
512 bp->dbg_name = fix_fname(p);
513 vim_free(p);
514 }
515 else
516 bp->dbg_name = p;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000517 }
518
519 if (bp->dbg_name == NULL)
520 return FAIL;
521 return OK;
522}
523
524/*
525 * ":breakadd".
526 */
527 void
528ex_breakadd(eap)
529 exarg_T *eap;
530{
531 struct debuggy *bp;
532 char_u *pat;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000533 garray_T *gap;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000534
Bram Moolenaar05159a02005-02-26 23:04:13 +0000535 gap = &dbg_breakp;
536#ifdef FEAT_PROFILE
537 if (eap->cmdidx == CMD_profile)
538 gap = &prof_ga;
539#endif
540
541 if (dbg_parsearg(eap->arg, gap) == OK)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000542 {
Bram Moolenaar05159a02005-02-26 23:04:13 +0000543 bp = &DEBUGGY(gap, gap->ga_len);
544 bp->dbg_forceit = eap->forceit;
545
Bram Moolenaar071d4272004-06-13 20:20:40 +0000546 pat = file_pat_to_reg_pat(bp->dbg_name, NULL, NULL, FALSE);
547 if (pat != NULL)
548 {
549 bp->dbg_prog = vim_regcomp(pat, RE_MAGIC + RE_STRING);
550 vim_free(pat);
551 }
552 if (pat == NULL || bp->dbg_prog == NULL)
553 vim_free(bp->dbg_name);
554 else
555 {
556 if (bp->dbg_lnum == 0) /* default line number is 1 */
557 bp->dbg_lnum = 1;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000558#ifdef FEAT_PROFILE
559 if (eap->cmdidx != CMD_profile)
560#endif
561 {
562 DEBUGGY(gap, gap->ga_len).dbg_nr = ++last_breakp;
563 ++debug_tick;
564 }
565 ++gap->ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000566 }
567 }
568}
569
570/*
571 * ":debuggreedy".
572 */
573 void
574ex_debuggreedy(eap)
575 exarg_T *eap;
576{
577 if (eap->addr_count == 0 || eap->line2 != 0)
578 debug_greedy = TRUE;
579 else
580 debug_greedy = FALSE;
581}
582
583/*
Bram Moolenaard9fba312005-06-26 22:34:35 +0000584 * ":breakdel" and ":profdel".
Bram Moolenaar071d4272004-06-13 20:20:40 +0000585 */
586 void
587ex_breakdel(eap)
588 exarg_T *eap;
589{
590 struct debuggy *bp, *bpi;
591 int nr;
592 int todel = -1;
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000593 int del_all = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000594 int i;
595 linenr_T best_lnum = 0;
Bram Moolenaard9fba312005-06-26 22:34:35 +0000596 garray_T *gap;
597
598 gap = &dbg_breakp;
Bram Moolenaard9fba312005-06-26 22:34:35 +0000599 if (eap->cmdidx == CMD_profdel)
Bram Moolenaar38bdbd62012-06-20 15:48:57 +0200600 {
601#ifdef FEAT_PROFILE
Bram Moolenaard9fba312005-06-26 22:34:35 +0000602 gap = &prof_ga;
Bram Moolenaar38bdbd62012-06-20 15:48:57 +0200603#else
604 ex_ni(eap);
605 return;
Bram Moolenaard9fba312005-06-26 22:34:35 +0000606#endif
Bram Moolenaar38bdbd62012-06-20 15:48:57 +0200607 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000608
609 if (vim_isdigit(*eap->arg))
610 {
611 /* ":breakdel {nr}" */
612 nr = atol((char *)eap->arg);
Bram Moolenaard9fba312005-06-26 22:34:35 +0000613 for (i = 0; i < gap->ga_len; ++i)
614 if (DEBUGGY(gap, i).dbg_nr == nr)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000615 {
616 todel = i;
617 break;
618 }
619 }
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000620 else if (*eap->arg == '*')
621 {
622 todel = 0;
623 del_all = TRUE;
624 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000625 else
626 {
627 /* ":breakdel {func|file} [lnum] {name}" */
Bram Moolenaard9fba312005-06-26 22:34:35 +0000628 if (dbg_parsearg(eap->arg, gap) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000629 return;
Bram Moolenaard9fba312005-06-26 22:34:35 +0000630 bp = &DEBUGGY(gap, gap->ga_len);
631 for (i = 0; i < gap->ga_len; ++i)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000632 {
Bram Moolenaard9fba312005-06-26 22:34:35 +0000633 bpi = &DEBUGGY(gap, i);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000634 if (bp->dbg_type == bpi->dbg_type
635 && STRCMP(bp->dbg_name, bpi->dbg_name) == 0
636 && (bp->dbg_lnum == bpi->dbg_lnum
637 || (bp->dbg_lnum == 0
638 && (best_lnum == 0
639 || bpi->dbg_lnum < best_lnum))))
640 {
641 todel = i;
642 best_lnum = bpi->dbg_lnum;
643 }
644 }
645 vim_free(bp->dbg_name);
646 }
647
648 if (todel < 0)
649 EMSG2(_("E161: Breakpoint not found: %s"), eap->arg);
650 else
Bram Moolenaard9fba312005-06-26 22:34:35 +0000651 {
652 while (gap->ga_len > 0)
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000653 {
Bram Moolenaard9fba312005-06-26 22:34:35 +0000654 vim_free(DEBUGGY(gap, todel).dbg_name);
Bram Moolenaar473de612013-06-08 18:19:48 +0200655 vim_regfree(DEBUGGY(gap, todel).dbg_prog);
Bram Moolenaard9fba312005-06-26 22:34:35 +0000656 --gap->ga_len;
657 if (todel < gap->ga_len)
658 mch_memmove(&DEBUGGY(gap, todel), &DEBUGGY(gap, todel + 1),
659 (gap->ga_len - todel) * sizeof(struct debuggy));
660#ifdef FEAT_PROFILE
661 if (eap->cmdidx == CMD_breakdel)
662#endif
663 ++debug_tick;
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000664 if (!del_all)
665 break;
666 }
Bram Moolenaard9fba312005-06-26 22:34:35 +0000667
668 /* If all breakpoints were removed clear the array. */
669 if (gap->ga_len == 0)
670 ga_clear(gap);
671 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000672}
673
674/*
675 * ":breaklist".
676 */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000677 void
678ex_breaklist(eap)
Bram Moolenaar0c094b92009-05-14 20:20:33 +0000679 exarg_T *eap UNUSED;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000680{
681 struct debuggy *bp;
682 int i;
683
684 if (dbg_breakp.ga_len == 0)
685 MSG(_("No breakpoints defined"));
686 else
687 for (i = 0; i < dbg_breakp.ga_len; ++i)
688 {
689 bp = &BREAKP(i);
Bram Moolenaard58ea072011-06-26 04:25:30 +0200690 if (bp->dbg_type == DBG_FILE)
691 home_replace(NULL, bp->dbg_name, NameBuff, MAXPATHL, TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000692 smsg((char_u *)_("%3d %s %s line %ld"),
693 bp->dbg_nr,
694 bp->dbg_type == DBG_FUNC ? "func" : "file",
Bram Moolenaard58ea072011-06-26 04:25:30 +0200695 bp->dbg_type == DBG_FUNC ? bp->dbg_name : NameBuff,
Bram Moolenaar071d4272004-06-13 20:20:40 +0000696 (long)bp->dbg_lnum);
697 }
698}
699
700/*
701 * Find a breakpoint for a function or sourced file.
702 * Returns line number at which to break; zero when no matching breakpoint.
703 */
704 linenr_T
705dbg_find_breakpoint(file, fname, after)
706 int file; /* TRUE for a file, FALSE for a function */
707 char_u *fname; /* file or function name */
708 linenr_T after; /* after this line number */
709{
Bram Moolenaar05159a02005-02-26 23:04:13 +0000710 return debuggy_find(file, fname, after, &dbg_breakp, NULL);
711}
712
713#if defined(FEAT_PROFILE) || defined(PROTO)
714/*
715 * Return TRUE if profiling is on for a function or sourced file.
716 */
717 int
718has_profiling(file, fname, fp)
719 int file; /* TRUE for a file, FALSE for a function */
720 char_u *fname; /* file or function name */
721 int *fp; /* return: forceit */
722{
723 return (debuggy_find(file, fname, (linenr_T)0, &prof_ga, fp)
724 != (linenr_T)0);
725}
726#endif
727
728/*
729 * Common code for dbg_find_breakpoint() and has_profiling().
730 */
731 static linenr_T
732debuggy_find(file, fname, after, gap, fp)
733 int file; /* TRUE for a file, FALSE for a function */
734 char_u *fname; /* file or function name */
735 linenr_T after; /* after this line number */
736 garray_T *gap; /* either &dbg_breakp or &prof_ga */
737 int *fp; /* if not NULL: return forceit */
738{
Bram Moolenaar071d4272004-06-13 20:20:40 +0000739 struct debuggy *bp;
740 int i;
741 linenr_T lnum = 0;
742 regmatch_T regmatch;
743 char_u *name = fname;
744 int prev_got_int;
745
Bram Moolenaar05159a02005-02-26 23:04:13 +0000746 /* Return quickly when there are no breakpoints. */
747 if (gap->ga_len == 0)
748 return (linenr_T)0;
749
Bram Moolenaar071d4272004-06-13 20:20:40 +0000750 /* Replace K_SNR in function name with "<SNR>". */
751 if (!file && fname[0] == K_SPECIAL)
752 {
753 name = alloc((unsigned)STRLEN(fname) + 3);
754 if (name == NULL)
755 name = fname;
756 else
757 {
758 STRCPY(name, "<SNR>");
759 STRCPY(name + 5, fname + 3);
760 }
761 }
762
Bram Moolenaar05159a02005-02-26 23:04:13 +0000763 for (i = 0; i < gap->ga_len; ++i)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000764 {
Bram Moolenaar05159a02005-02-26 23:04:13 +0000765 /* Skip entries that are not useful or are for a line that is beyond
766 * an already found breakpoint. */
767 bp = &DEBUGGY(gap, i);
768 if (((bp->dbg_type == DBG_FILE) == file && (
769#ifdef FEAT_PROFILE
770 gap == &prof_ga ||
771#endif
772 (bp->dbg_lnum > after && (lnum == 0 || bp->dbg_lnum < lnum)))))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000773 {
774 regmatch.regprog = bp->dbg_prog;
775 regmatch.rm_ic = FALSE;
776 /*
Bram Moolenaar05159a02005-02-26 23:04:13 +0000777 * Save the value of got_int and reset it. We don't want a
778 * previous interruption cancel matching, only hitting CTRL-C
779 * while matching should abort it.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000780 */
781 prev_got_int = got_int;
782 got_int = FALSE;
783 if (vim_regexec(&regmatch, name, (colnr_T)0))
Bram Moolenaar05159a02005-02-26 23:04:13 +0000784 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000785 lnum = bp->dbg_lnum;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000786 if (fp != NULL)
787 *fp = bp->dbg_forceit;
788 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000789 got_int |= prev_got_int;
790 }
791 }
792 if (name != fname)
793 vim_free(name);
794
795 return lnum;
796}
797
798/*
799 * Called when a breakpoint was encountered.
800 */
801 void
802dbg_breakpoint(name, lnum)
803 char_u *name;
804 linenr_T lnum;
805{
806 /* We need to check if this line is actually executed in do_one_cmd() */
807 debug_breakpoint_name = name;
808 debug_breakpoint_lnum = lnum;
809}
Bram Moolenaar05159a02005-02-26 23:04:13 +0000810
811
Bram Moolenaar433f7c82006-03-21 21:29:36 +0000812# if defined(FEAT_PROFILE) || defined(FEAT_RELTIME) || defined(PROTO)
Bram Moolenaar05159a02005-02-26 23:04:13 +0000813/*
814 * Store the current time in "tm".
815 */
816 void
817profile_start(tm)
818 proftime_T *tm;
819{
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000820# ifdef WIN3264
821 QueryPerformanceCounter(tm);
822# else
Bram Moolenaar05159a02005-02-26 23:04:13 +0000823 gettimeofday(tm, NULL);
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000824# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +0000825}
826
827/*
828 * Compute the elapsed time from "tm" till now and store in "tm".
829 */
830 void
831profile_end(tm)
832 proftime_T *tm;
833{
834 proftime_T now;
835
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000836# ifdef WIN3264
837 QueryPerformanceCounter(&now);
838 tm->QuadPart = now.QuadPart - tm->QuadPart;
839# else
Bram Moolenaar05159a02005-02-26 23:04:13 +0000840 gettimeofday(&now, NULL);
841 tm->tv_usec = now.tv_usec - tm->tv_usec;
842 tm->tv_sec = now.tv_sec - tm->tv_sec;
843 if (tm->tv_usec < 0)
844 {
845 tm->tv_usec += 1000000;
846 --tm->tv_sec;
847 }
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000848# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +0000849}
850
851/*
852 * Subtract the time "tm2" from "tm".
853 */
854 void
855profile_sub(tm, tm2)
856 proftime_T *tm, *tm2;
857{
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000858# ifdef WIN3264
859 tm->QuadPart -= tm2->QuadPart;
860# else
Bram Moolenaar05159a02005-02-26 23:04:13 +0000861 tm->tv_usec -= tm2->tv_usec;
862 tm->tv_sec -= tm2->tv_sec;
863 if (tm->tv_usec < 0)
864 {
865 tm->tv_usec += 1000000;
866 --tm->tv_sec;
867 }
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000868# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +0000869}
870
871/*
Bram Moolenaar433f7c82006-03-21 21:29:36 +0000872 * Return a string that represents the time in "tm".
873 * Uses a static buffer!
874 */
875 char *
876profile_msg(tm)
877 proftime_T *tm;
878{
879 static char buf[50];
880
881# ifdef WIN3264
882 LARGE_INTEGER fr;
883
884 QueryPerformanceFrequency(&fr);
885 sprintf(buf, "%10.6lf", (double)tm->QuadPart / (double)fr.QuadPart);
886# else
887 sprintf(buf, "%3ld.%06ld", (long)tm->tv_sec, (long)tm->tv_usec);
Bram Moolenaar76929292008-01-06 19:07:36 +0000888# endif
Bram Moolenaar433f7c82006-03-21 21:29:36 +0000889 return buf;
890}
891
Bram Moolenaar433f7c82006-03-21 21:29:36 +0000892/*
Bram Moolenaar76929292008-01-06 19:07:36 +0000893 * Put the time "msec" past now in "tm".
Bram Moolenaar433f7c82006-03-21 21:29:36 +0000894 */
Bram Moolenaar76929292008-01-06 19:07:36 +0000895 void
896profile_setlimit(msec, tm)
897 long msec;
898 proftime_T *tm;
899{
900 if (msec <= 0) /* no limit */
901 profile_zero(tm);
902 else
903 {
904# ifdef WIN3264
905 LARGE_INTEGER fr;
906
907 QueryPerformanceCounter(tm);
908 QueryPerformanceFrequency(&fr);
Bram Moolenaarb3c70982008-01-18 10:40:55 +0000909 tm->QuadPart += (LONGLONG)((double)msec / 1000.0 * (double)fr.QuadPart);
Bram Moolenaar76929292008-01-06 19:07:36 +0000910# else
911 long usec;
912
913 gettimeofday(tm, NULL);
914 usec = (long)tm->tv_usec + (long)msec * 1000;
915 tm->tv_usec = usec % 1000000L;
916 tm->tv_sec += usec / 1000000L;
917# endif
918 }
919}
920
921/*
922 * Return TRUE if the current time is past "tm".
923 */
924 int
925profile_passed_limit(tm)
926 proftime_T *tm;
927{
928 proftime_T now;
929
930# ifdef WIN3264
931 if (tm->QuadPart == 0) /* timer was not set */
932 return FALSE;
933 QueryPerformanceCounter(&now);
934 return (now.QuadPart > tm->QuadPart);
935# else
936 if (tm->tv_sec == 0) /* timer was not set */
937 return FALSE;
938 gettimeofday(&now, NULL);
939 return (now.tv_sec > tm->tv_sec
940 || (now.tv_sec == tm->tv_sec && now.tv_usec > tm->tv_usec));
941# endif
942}
Bram Moolenaar433f7c82006-03-21 21:29:36 +0000943
944/*
945 * Set the time in "tm" to zero.
946 */
947 void
948profile_zero(tm)
949 proftime_T *tm;
950{
951# ifdef WIN3264
952 tm->QuadPart = 0;
953# else
954 tm->tv_usec = 0;
955 tm->tv_sec = 0;
956# endif
957}
958
Bram Moolenaar76929292008-01-06 19:07:36 +0000959# endif /* FEAT_PROFILE || FEAT_RELTIME */
960
Bram Moolenaar8a7f5a22013-06-06 14:01:46 +0200961#if defined(FEAT_SYN_HL) && defined(FEAT_RELTIME) && defined(FEAT_FLOAT)
962# if defined(HAVE_MATH_H)
963# include <math.h>
964# endif
965
966/*
967 * Divide the time "tm" by "count" and store in "tm2".
968 */
969 void
970profile_divide(tm, count, tm2)
971 proftime_T *tm;
972 proftime_T *tm2;
973 int count;
974{
975 if (count == 0)
976 profile_zero(tm2);
977 else
978 {
979# ifdef WIN3264
980 tm2->QuadPart = tm->QuadPart / count;
981# else
982 double usec = (tm->tv_sec * 1000000.0 + tm->tv_usec) / count;
983
984 tm2->tv_sec = floor(usec / 1000000.0);
Bram Moolenaara2e14fc2013-06-10 20:10:44 +0200985 tm2->tv_usec = vim_round(usec - (tm2->tv_sec * 1000000.0));
Bram Moolenaar8a7f5a22013-06-06 14:01:46 +0200986# endif
987 }
988}
989#endif
990
Bram Moolenaar76929292008-01-06 19:07:36 +0000991# if defined(FEAT_PROFILE) || defined(PROTO)
992/*
993 * Functions for profiling.
994 */
995static void script_do_profile __ARGS((scriptitem_T *si));
996static void script_dump_profile __ARGS((FILE *fd));
997static proftime_T prof_wait_time;
998
Bram Moolenaar433f7c82006-03-21 21:29:36 +0000999/*
Bram Moolenaar05159a02005-02-26 23:04:13 +00001000 * Add the time "tm2" to "tm".
1001 */
1002 void
1003profile_add(tm, tm2)
1004 proftime_T *tm, *tm2;
1005{
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001006# ifdef WIN3264
1007 tm->QuadPart += tm2->QuadPart;
1008# else
Bram Moolenaar05159a02005-02-26 23:04:13 +00001009 tm->tv_usec += tm2->tv_usec;
1010 tm->tv_sec += tm2->tv_sec;
1011 if (tm->tv_usec >= 1000000)
1012 {
1013 tm->tv_usec -= 1000000;
1014 ++tm->tv_sec;
1015 }
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001016# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00001017}
1018
1019/*
Bram Moolenaar1056d982006-03-09 22:37:52 +00001020 * Add the "self" time from the total time and the children's time.
1021 */
1022 void
1023profile_self(self, total, children)
1024 proftime_T *self, *total, *children;
1025{
1026 /* Check that the result won't be negative. Can happen with recursive
1027 * calls. */
1028#ifdef WIN3264
1029 if (total->QuadPart <= children->QuadPart)
1030 return;
1031#else
1032 if (total->tv_sec < children->tv_sec
1033 || (total->tv_sec == children->tv_sec
1034 && total->tv_usec <= children->tv_usec))
1035 return;
1036#endif
1037 profile_add(self, total);
1038 profile_sub(self, children);
1039}
1040
1041/*
Bram Moolenaar05159a02005-02-26 23:04:13 +00001042 * Get the current waittime.
1043 */
1044 void
1045profile_get_wait(tm)
1046 proftime_T *tm;
1047{
1048 *tm = prof_wait_time;
1049}
1050
1051/*
1052 * Subtract the passed waittime since "tm" from "tma".
1053 */
1054 void
1055profile_sub_wait(tm, tma)
1056 proftime_T *tm, *tma;
1057{
1058 proftime_T tm3 = prof_wait_time;
1059
1060 profile_sub(&tm3, tm);
1061 profile_sub(tma, &tm3);
1062}
1063
1064/*
1065 * Return TRUE if "tm1" and "tm2" are equal.
1066 */
1067 int
1068profile_equal(tm1, tm2)
1069 proftime_T *tm1, *tm2;
1070{
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001071# ifdef WIN3264
1072 return (tm1->QuadPart == tm2->QuadPart);
1073# else
Bram Moolenaar05159a02005-02-26 23:04:13 +00001074 return (tm1->tv_usec == tm2->tv_usec && tm1->tv_sec == tm2->tv_sec);
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001075# endif
1076}
1077
1078/*
1079 * Return <0, 0 or >0 if "tm1" < "tm2", "tm1" == "tm2" or "tm1" > "tm2"
1080 */
1081 int
1082profile_cmp(tm1, tm2)
Bram Moolenaar8a7f5a22013-06-06 14:01:46 +02001083 const proftime_T *tm1, *tm2;
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001084{
1085# ifdef WIN3264
1086 return (int)(tm2->QuadPart - tm1->QuadPart);
1087# else
1088 if (tm1->tv_sec == tm2->tv_sec)
1089 return tm2->tv_usec - tm1->tv_usec;
1090 return tm2->tv_sec - tm1->tv_sec;
1091# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00001092}
1093
Bram Moolenaar05159a02005-02-26 23:04:13 +00001094static char_u *profile_fname = NULL;
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001095static proftime_T pause_time;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001096
1097/*
1098 * ":profile cmd args"
1099 */
1100 void
1101ex_profile(eap)
1102 exarg_T *eap;
1103{
1104 char_u *e;
1105 int len;
1106
1107 e = skiptowhite(eap->arg);
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00001108 len = (int)(e - eap->arg);
Bram Moolenaar05159a02005-02-26 23:04:13 +00001109 e = skipwhite(e);
1110
1111 if (len == 5 && STRNCMP(eap->arg, "start", 5) == 0 && *e != NUL)
1112 {
1113 vim_free(profile_fname);
1114 profile_fname = vim_strsave(e);
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001115 do_profiling = PROF_YES;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001116 profile_zero(&prof_wait_time);
1117 set_vim_var_nr(VV_PROFILING, 1L);
1118 }
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001119 else if (do_profiling == PROF_NONE)
Bram Moolenaar54c1b492010-02-24 14:01:28 +01001120 EMSG(_("E750: First use \":profile start {fname}\""));
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001121 else if (STRCMP(eap->arg, "pause") == 0)
1122 {
1123 if (do_profiling == PROF_YES)
1124 profile_start(&pause_time);
1125 do_profiling = PROF_PAUSED;
1126 }
1127 else if (STRCMP(eap->arg, "continue") == 0)
1128 {
1129 if (do_profiling == PROF_PAUSED)
1130 {
1131 profile_end(&pause_time);
1132 profile_add(&prof_wait_time, &pause_time);
1133 }
1134 do_profiling = PROF_YES;
1135 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00001136 else
1137 {
1138 /* The rest is similar to ":breakadd". */
1139 ex_breakadd(eap);
1140 }
1141}
1142
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001143/* Command line expansion for :profile. */
1144static enum
1145{
1146 PEXP_SUBCMD, /* expand :profile sub-commands */
Bram Moolenaar49789dc2011-02-25 14:46:09 +01001147 PEXP_FUNC /* expand :profile func {funcname} */
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001148} pexpand_what;
1149
1150static char *pexpand_cmds[] = {
1151 "start",
1152#define PROFCMD_START 0
1153 "pause",
1154#define PROFCMD_PAUSE 1
1155 "continue",
1156#define PROFCMD_CONTINUE 2
1157 "func",
1158#define PROFCMD_FUNC 3
1159 "file",
1160#define PROFCMD_FILE 4
1161 NULL
1162#define PROFCMD_LAST 5
1163};
1164
1165/*
1166 * Function given to ExpandGeneric() to obtain the profile command
1167 * specific expansion.
1168 */
1169 char_u *
1170get_profile_name(xp, idx)
1171 expand_T *xp UNUSED;
1172 int idx;
1173{
1174 switch (pexpand_what)
1175 {
1176 case PEXP_SUBCMD:
1177 return (char_u *)pexpand_cmds[idx];
1178 /* case PEXP_FUNC: TODO */
1179 default:
1180 return NULL;
1181 }
1182}
1183
1184/*
1185 * Handle command line completion for :profile command.
1186 */
1187 void
1188set_context_in_profile_cmd(xp, arg)
1189 expand_T *xp;
1190 char_u *arg;
1191{
1192 char_u *end_subcmd;
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001193
1194 /* Default: expand subcommands. */
1195 xp->xp_context = EXPAND_PROFILE;
1196 pexpand_what = PEXP_SUBCMD;
1197 xp->xp_pattern = arg;
1198
1199 end_subcmd = skiptowhite(arg);
1200 if (*end_subcmd == NUL)
1201 return;
1202
Bram Moolenaar8b9c05f2010-03-02 17:54:33 +01001203 if (end_subcmd - arg == 5 && STRNCMP(arg, "start", 5) == 0)
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001204 {
1205 xp->xp_context = EXPAND_FILES;
1206 xp->xp_pattern = skipwhite(end_subcmd);
1207 return;
1208 }
1209
1210 /* TODO: expand function names after "func" */
1211 xp->xp_context = EXPAND_NOTHING;
1212}
1213
Bram Moolenaar05159a02005-02-26 23:04:13 +00001214/*
1215 * Dump the profiling info.
1216 */
1217 void
1218profile_dump()
1219{
1220 FILE *fd;
1221
1222 if (profile_fname != NULL)
1223 {
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +00001224 fd = mch_fopen((char *)profile_fname, "w");
Bram Moolenaar05159a02005-02-26 23:04:13 +00001225 if (fd == NULL)
1226 EMSG2(_(e_notopen), profile_fname);
1227 else
1228 {
Bram Moolenaar05159a02005-02-26 23:04:13 +00001229 script_dump_profile(fd);
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001230 func_dump_profile(fd);
Bram Moolenaar05159a02005-02-26 23:04:13 +00001231 fclose(fd);
1232 }
1233 }
1234}
1235
1236/*
1237 * Start profiling script "fp".
1238 */
1239 static void
1240script_do_profile(si)
1241 scriptitem_T *si;
1242{
1243 si->sn_pr_count = 0;
1244 profile_zero(&si->sn_pr_total);
1245 profile_zero(&si->sn_pr_self);
1246
1247 ga_init2(&si->sn_prl_ga, sizeof(sn_prl_T), 100);
1248 si->sn_prl_idx = -1;
1249 si->sn_prof_on = TRUE;
1250 si->sn_pr_nest = 0;
1251}
1252
1253/*
1254 * save time when starting to invoke another script or function.
1255 */
1256 void
1257script_prof_save(tm)
1258 proftime_T *tm; /* place to store wait time */
1259{
1260 scriptitem_T *si;
1261
1262 if (current_SID > 0 && current_SID <= script_items.ga_len)
1263 {
1264 si = &SCRIPT_ITEM(current_SID);
1265 if (si->sn_prof_on && si->sn_pr_nest++ == 0)
1266 profile_start(&si->sn_pr_child);
1267 }
1268 profile_get_wait(tm);
1269}
1270
1271/*
1272 * Count time spent in children after invoking another script or function.
1273 */
1274 void
1275script_prof_restore(tm)
1276 proftime_T *tm;
1277{
1278 scriptitem_T *si;
1279
1280 if (current_SID > 0 && current_SID <= script_items.ga_len)
1281 {
1282 si = &SCRIPT_ITEM(current_SID);
1283 if (si->sn_prof_on && --si->sn_pr_nest == 0)
1284 {
1285 profile_end(&si->sn_pr_child);
1286 profile_sub_wait(tm, &si->sn_pr_child); /* don't count wait time */
1287 profile_add(&si->sn_pr_children, &si->sn_pr_child);
1288 profile_add(&si->sn_prl_children, &si->sn_pr_child);
1289 }
1290 }
1291}
1292
1293static proftime_T inchar_time;
1294
1295/*
1296 * Called when starting to wait for the user to type a character.
1297 */
1298 void
1299prof_inchar_enter()
1300{
1301 profile_start(&inchar_time);
1302}
1303
1304/*
1305 * Called when finished waiting for the user to type a character.
1306 */
1307 void
1308prof_inchar_exit()
1309{
1310 profile_end(&inchar_time);
1311 profile_add(&prof_wait_time, &inchar_time);
1312}
1313
1314/*
1315 * Dump the profiling results for all scripts in file "fd".
1316 */
1317 static void
1318script_dump_profile(fd)
1319 FILE *fd;
1320{
1321 int id;
1322 scriptitem_T *si;
1323 int i;
1324 FILE *sfd;
1325 sn_prl_T *pp;
1326
1327 for (id = 1; id <= script_items.ga_len; ++id)
1328 {
1329 si = &SCRIPT_ITEM(id);
1330 if (si->sn_prof_on)
1331 {
1332 fprintf(fd, "SCRIPT %s\n", si->sn_name);
1333 if (si->sn_pr_count == 1)
1334 fprintf(fd, "Sourced 1 time\n");
1335 else
1336 fprintf(fd, "Sourced %d times\n", si->sn_pr_count);
1337 fprintf(fd, "Total time: %s\n", profile_msg(&si->sn_pr_total));
1338 fprintf(fd, " Self time: %s\n", profile_msg(&si->sn_pr_self));
1339 fprintf(fd, "\n");
1340 fprintf(fd, "count total (s) self (s)\n");
1341
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +00001342 sfd = mch_fopen((char *)si->sn_name, "r");
Bram Moolenaar05159a02005-02-26 23:04:13 +00001343 if (sfd == NULL)
1344 fprintf(fd, "Cannot open file!\n");
1345 else
1346 {
1347 for (i = 0; i < si->sn_prl_ga.ga_len; ++i)
1348 {
1349 if (vim_fgets(IObuff, IOSIZE, sfd))
1350 break;
1351 pp = &PRL_ITEM(si, i);
1352 if (pp->snp_count > 0)
1353 {
1354 fprintf(fd, "%5d ", pp->snp_count);
1355 if (profile_equal(&pp->sn_prl_total, &pp->sn_prl_self))
1356 fprintf(fd, " ");
1357 else
1358 fprintf(fd, "%s ", profile_msg(&pp->sn_prl_total));
1359 fprintf(fd, "%s ", profile_msg(&pp->sn_prl_self));
1360 }
1361 else
1362 fprintf(fd, " ");
1363 fprintf(fd, "%s", IObuff);
1364 }
1365 fclose(sfd);
1366 }
1367 fprintf(fd, "\n");
1368 }
1369 }
1370}
1371
1372/*
1373 * Return TRUE when a function defined in the current script should be
1374 * profiled.
1375 */
1376 int
1377prof_def_func()
1378{
Bram Moolenaar53180ce2005-07-05 21:48:14 +00001379 if (current_SID > 0)
1380 return SCRIPT_ITEM(current_SID).sn_pr_force;
1381 return FALSE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001382}
1383
1384# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001385#endif
1386
1387/*
1388 * If 'autowrite' option set, try to write the file.
1389 * Careful: autocommands may make "buf" invalid!
1390 *
1391 * return FAIL for failure, OK otherwise
1392 */
1393 int
1394autowrite(buf, forceit)
1395 buf_T *buf;
1396 int forceit;
1397{
Bram Moolenaar373154b2007-02-13 05:19:30 +00001398 int r;
1399
Bram Moolenaar071d4272004-06-13 20:20:40 +00001400 if (!(p_aw || p_awa) || !p_write
1401#ifdef FEAT_QUICKFIX
Bram Moolenaar373154b2007-02-13 05:19:30 +00001402 /* never autowrite a "nofile" or "nowrite" buffer */
1403 || bt_dontwrite(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001404#endif
Bram Moolenaar373154b2007-02-13 05:19:30 +00001405 || (!forceit && buf->b_p_ro) || buf->b_ffname == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001406 return FAIL;
Bram Moolenaar373154b2007-02-13 05:19:30 +00001407 r = buf_write_all(buf, forceit);
1408
1409 /* Writing may succeed but the buffer still changed, e.g., when there is a
1410 * conversion error. We do want to return FAIL then. */
1411 if (buf_valid(buf) && bufIsChanged(buf))
1412 r = FAIL;
1413 return r;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001414}
1415
1416/*
1417 * flush all buffers, except the ones that are readonly
1418 */
1419 void
1420autowrite_all()
1421{
1422 buf_T *buf;
1423
1424 if (!(p_aw || p_awa) || !p_write)
1425 return;
1426 for (buf = firstbuf; buf; buf = buf->b_next)
1427 if (bufIsChanged(buf) && !buf->b_p_ro)
1428 {
1429 (void)buf_write_all(buf, FALSE);
1430#ifdef FEAT_AUTOCMD
1431 /* an autocommand may have deleted the buffer */
1432 if (!buf_valid(buf))
1433 buf = firstbuf;
1434#endif
1435 }
1436}
1437
1438/*
1439 * return TRUE if buffer was changed and cannot be abandoned.
1440 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001441 int
1442check_changed(buf, checkaw, mult_win, forceit, allbuf)
1443 buf_T *buf;
1444 int checkaw; /* do autowrite if buffer was changed */
1445 int mult_win; /* check also when several wins for the buf */
1446 int forceit;
Bram Moolenaar0c094b92009-05-14 20:20:33 +00001447 int allbuf UNUSED; /* may write all buffers */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001448{
1449 if ( !forceit
1450 && bufIsChanged(buf)
1451 && (mult_win || buf->b_nwindows <= 1)
1452 && (!checkaw || autowrite(buf, forceit) == FAIL))
1453 {
1454#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG)
1455 if ((p_confirm || cmdmod.confirm) && p_write)
1456 {
1457 buf_T *buf2;
1458 int count = 0;
1459
1460 if (allbuf)
1461 for (buf2 = firstbuf; buf2 != NULL; buf2 = buf2->b_next)
1462 if (bufIsChanged(buf2)
1463 && (buf2->b_ffname != NULL
1464# ifdef FEAT_BROWSE
1465 || cmdmod.browse
1466# endif
1467 ))
1468 ++count;
1469# ifdef FEAT_AUTOCMD
1470 if (!buf_valid(buf))
1471 /* Autocommand deleted buffer, oops! It's not changed now. */
1472 return FALSE;
1473# endif
1474 dialog_changed(buf, count > 1);
1475# ifdef FEAT_AUTOCMD
1476 if (!buf_valid(buf))
1477 /* Autocommand deleted buffer, oops! It's not changed now. */
1478 return FALSE;
1479# endif
1480 return bufIsChanged(buf);
1481 }
1482#endif
1483 EMSG(_(e_nowrtmsg));
1484 return TRUE;
1485 }
1486 return FALSE;
1487}
1488
1489#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG) || defined(PROTO)
1490
1491#if defined(FEAT_BROWSE) || defined(PROTO)
1492/*
1493 * When wanting to write a file without a file name, ask the user for a name.
1494 */
1495 void
1496browse_save_fname(buf)
1497 buf_T *buf;
1498{
1499 if (buf->b_fname == NULL)
1500 {
1501 char_u *fname;
1502
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00001503 fname = do_browse(BROWSE_SAVE, (char_u *)_("Save As"),
1504 NULL, NULL, NULL, NULL, buf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001505 if (fname != NULL)
1506 {
1507 if (setfname(buf, fname, NULL, TRUE) == OK)
1508 buf->b_flags |= BF_NOTEDITED;
1509 vim_free(fname);
1510 }
1511 }
1512}
1513#endif
1514
1515/*
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02001516 * Ask the user what to do when abandoning a changed buffer.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001517 * Must check 'write' option first!
1518 */
1519 void
1520dialog_changed(buf, checkall)
1521 buf_T *buf;
1522 int checkall; /* may abandon all changed buffers */
1523{
Bram Moolenaard9462e32011-04-11 21:35:11 +02001524 char_u buff[DIALOG_MSG_SIZE];
Bram Moolenaar071d4272004-06-13 20:20:40 +00001525 int ret;
1526 buf_T *buf2;
Bram Moolenaar8218f602012-04-25 17:32:18 +02001527 exarg_T ea;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001528
Bram Moolenaar82cf9b62005-06-07 21:09:25 +00001529 dialog_msg(buff, _("Save changes to \"%s\"?"),
Bram Moolenaar071d4272004-06-13 20:20:40 +00001530 (buf->b_fname != NULL) ?
1531 buf->b_fname : (char_u *)_("Untitled"));
1532 if (checkall)
1533 ret = vim_dialog_yesnoallcancel(VIM_QUESTION, NULL, buff, 1);
1534 else
1535 ret = vim_dialog_yesnocancel(VIM_QUESTION, NULL, buff, 1);
1536
Bram Moolenaar8218f602012-04-25 17:32:18 +02001537 /* Init ea pseudo-structure, this is needed for the check_overwrite()
1538 * function. */
1539 ea.append = ea.forceit = FALSE;
1540
Bram Moolenaar071d4272004-06-13 20:20:40 +00001541 if (ret == VIM_YES)
1542 {
1543#ifdef FEAT_BROWSE
1544 /* May get file name, when there is none */
1545 browse_save_fname(buf);
1546#endif
Bram Moolenaar8218f602012-04-25 17:32:18 +02001547 if (buf->b_fname != NULL && check_overwrite(&ea, buf,
1548 buf->b_fname, buf->b_ffname, FALSE) == OK)
1549 /* didn't hit Cancel */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001550 (void)buf_write_all(buf, FALSE);
1551 }
1552 else if (ret == VIM_NO)
1553 {
1554 unchanged(buf, TRUE);
1555 }
1556 else if (ret == VIM_ALL)
1557 {
1558 /*
1559 * Write all modified files that can be written.
1560 * Skip readonly buffers, these need to be confirmed
1561 * individually.
1562 */
1563 for (buf2 = firstbuf; buf2 != NULL; buf2 = buf2->b_next)
1564 {
1565 if (bufIsChanged(buf2)
1566 && (buf2->b_ffname != NULL
1567#ifdef FEAT_BROWSE
1568 || cmdmod.browse
1569#endif
1570 )
1571 && !buf2->b_p_ro)
1572 {
1573#ifdef FEAT_BROWSE
1574 /* May get file name, when there is none */
1575 browse_save_fname(buf2);
1576#endif
Bram Moolenaar8218f602012-04-25 17:32:18 +02001577 if (buf2->b_fname != NULL && check_overwrite(&ea, buf2,
1578 buf2->b_fname, buf2->b_ffname, FALSE) == OK)
1579 /* didn't hit Cancel */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001580 (void)buf_write_all(buf2, FALSE);
1581#ifdef FEAT_AUTOCMD
1582 /* an autocommand may have deleted the buffer */
1583 if (!buf_valid(buf2))
1584 buf2 = firstbuf;
1585#endif
1586 }
1587 }
1588 }
1589 else if (ret == VIM_DISCARDALL)
1590 {
1591 /*
1592 * mark all buffers as unchanged
1593 */
1594 for (buf2 = firstbuf; buf2 != NULL; buf2 = buf2->b_next)
1595 unchanged(buf2, TRUE);
1596 }
1597}
1598#endif
1599
1600/*
1601 * Return TRUE if the buffer "buf" can be abandoned, either by making it
1602 * hidden, autowriting it or unloading it.
1603 */
1604 int
1605can_abandon(buf, forceit)
1606 buf_T *buf;
1607 int forceit;
1608{
1609 return ( P_HID(buf)
1610 || !bufIsChanged(buf)
1611 || buf->b_nwindows > 1
1612 || autowrite(buf, forceit) == OK
1613 || forceit);
1614}
1615
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001616static void add_bufnum __ARGS((int *bufnrs, int *bufnump, int nr));
1617
1618/*
1619 * Add a buffer number to "bufnrs", unless it's already there.
1620 */
1621 static void
1622add_bufnum(bufnrs, bufnump, nr)
1623 int *bufnrs;
1624 int *bufnump;
1625 int nr;
1626{
1627 int i;
1628
1629 for (i = 0; i < *bufnump; ++i)
1630 if (bufnrs[i] == nr)
1631 return;
1632 bufnrs[*bufnump] = nr;
1633 *bufnump = *bufnump + 1;
1634}
1635
Bram Moolenaar071d4272004-06-13 20:20:40 +00001636/*
1637 * Return TRUE if any buffer was changed and cannot be abandoned.
1638 * That changed buffer becomes the current buffer.
1639 */
1640 int
1641check_changed_any(hidden)
1642 int hidden; /* Only check hidden buffers */
1643{
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001644 int ret = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001645 buf_T *buf;
1646 int save;
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001647 int i;
1648 int bufnum = 0;
1649 int bufcount = 0;
1650 int *bufnrs;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001651#ifdef FEAT_WINDOWS
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001652 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001653 win_T *wp;
1654#endif
1655
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001656 for (buf = firstbuf; buf != NULL; buf = buf->b_next)
1657 ++bufcount;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001658
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001659 if (bufcount == 0)
1660 return FALSE;
1661
1662 bufnrs = (int *)alloc(sizeof(int) * bufcount);
1663 if (bufnrs == NULL)
1664 return FALSE;
1665
1666 /* curbuf */
1667 bufnrs[bufnum++] = curbuf->b_fnum;
1668#ifdef FEAT_WINDOWS
1669 /* buf in curtab */
1670 FOR_ALL_WINDOWS(wp)
1671 if (wp->w_buffer != curbuf)
1672 add_bufnum(bufnrs, &bufnum, wp->w_buffer->b_fnum);
1673
1674 /* buf in other tab */
1675 for (tp = first_tabpage; tp != NULL; tp = tp->tp_next)
1676 if (tp != curtab)
1677 for (wp = tp->tp_firstwin; wp != NULL; wp = wp->w_next)
1678 add_bufnum(bufnrs, &bufnum, wp->w_buffer->b_fnum);
1679#endif
1680 /* any other buf */
1681 for (buf = firstbuf; buf != NULL; buf = buf->b_next)
1682 add_bufnum(bufnrs, &bufnum, buf->b_fnum);
1683
1684 for (i = 0; i < bufnum; ++i)
1685 {
1686 buf = buflist_findnr(bufnrs[i]);
1687 if (buf == NULL)
1688 continue;
1689 if ((!hidden || buf->b_nwindows == 0) && bufIsChanged(buf))
1690 {
1691 /* Try auto-writing the buffer. If this fails but the buffer no
1692 * longer exists it's not changed, that's OK. */
1693 if (check_changed(buf, p_awa, TRUE, FALSE, TRUE) && buf_valid(buf))
1694 break; /* didn't save - still changes */
1695 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001696 }
1697
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001698 if (i >= bufnum)
1699 goto theend;
1700
1701 ret = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001702 exiting = FALSE;
1703#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG)
1704 /*
1705 * When ":confirm" used, don't give an error message.
1706 */
1707 if (!(p_confirm || cmdmod.confirm))
1708#endif
1709 {
1710 /* There must be a wait_return for this message, do_buffer()
1711 * may cause a redraw. But wait_return() is a no-op when vgetc()
1712 * is busy (Quit used from window menu), then make sure we don't
1713 * cause a scroll up. */
Bram Moolenaar61660ea2006-04-07 21:40:07 +00001714 if (vgetc_busy > 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001715 {
1716 msg_row = cmdline_row;
1717 msg_col = 0;
1718 msg_didout = FALSE;
1719 }
1720 if (EMSG2(_("E162: No write since last change for buffer \"%s\""),
Bram Moolenaare1704ba2012-10-03 18:25:00 +02001721 buf_spname(buf) != NULL ? buf_spname(buf) : buf->b_fname))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001722 {
1723 save = no_wait_return;
1724 no_wait_return = FALSE;
1725 wait_return(FALSE);
1726 no_wait_return = save;
1727 }
1728 }
1729
1730#ifdef FEAT_WINDOWS
1731 /* Try to find a window that contains the buffer. */
1732 if (buf != curbuf)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001733 FOR_ALL_TAB_WINDOWS(tp, wp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001734 if (wp->w_buffer == buf)
1735 {
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001736 goto_tabpage_win(tp, wp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001737# ifdef FEAT_AUTOCMD
1738 /* Paranoia: did autocms wipe out the buffer with changes? */
1739 if (!buf_valid(buf))
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001740 {
1741 goto theend;
1742 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001743# endif
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001744 goto buf_found;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001745 }
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001746buf_found:
Bram Moolenaar071d4272004-06-13 20:20:40 +00001747#endif
1748
1749 /* Open the changed buffer in the current window. */
1750 if (buf != curbuf)
1751 set_curbuf(buf, DOBUF_GOTO);
1752
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001753theend:
1754 vim_free(bufnrs);
1755 return ret;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001756}
1757
1758/*
1759 * return FAIL if there is no file name, OK if there is one
1760 * give error message for FAIL
1761 */
1762 int
1763check_fname()
1764{
1765 if (curbuf->b_ffname == NULL)
1766 {
1767 EMSG(_(e_noname));
1768 return FAIL;
1769 }
1770 return OK;
1771}
1772
1773/*
1774 * flush the contents of a buffer, unless it has no file name
1775 *
1776 * return FAIL for failure, OK otherwise
1777 */
1778 int
1779buf_write_all(buf, forceit)
1780 buf_T *buf;
1781 int forceit;
1782{
1783 int retval;
1784#ifdef FEAT_AUTOCMD
1785 buf_T *old_curbuf = curbuf;
1786#endif
1787
1788 retval = (buf_write(buf, buf->b_ffname, buf->b_fname,
1789 (linenr_T)1, buf->b_ml.ml_line_count, NULL,
1790 FALSE, forceit, TRUE, FALSE));
1791#ifdef FEAT_AUTOCMD
1792 if (curbuf != old_curbuf)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001793 {
1794 msg_source(hl_attr(HLF_W));
Bram Moolenaar071d4272004-06-13 20:20:40 +00001795 MSG(_("Warning: Entered other buffer unexpectedly (check autocommands)"));
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001796 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001797#endif
1798 return retval;
1799}
1800
1801/*
1802 * Code to handle the argument list.
1803 */
1804
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001805static char_u *do_one_arg __ARGS((char_u *str));
1806static int do_arglist __ARGS((char_u *str, int what, int after));
1807static void alist_check_arg_idx __ARGS((void));
1808static int editing_arg_idx __ARGS((win_T *win));
1809#ifdef FEAT_LISTCMDS
1810static int alist_add_list __ARGS((int count, char_u **files, int after));
1811#endif
1812#define AL_SET 1
1813#define AL_ADD 2
1814#define AL_DEL 3
1815
Bram Moolenaar071d4272004-06-13 20:20:40 +00001816/*
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001817 * Isolate one argument, taking backticks.
1818 * Changes the argument in-place, puts a NUL after it. Backticks remain.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001819 * Return a pointer to the start of the next argument.
1820 */
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001821 static char_u *
Bram Moolenaar071d4272004-06-13 20:20:40 +00001822do_one_arg(str)
1823 char_u *str;
1824{
1825 char_u *p;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001826 int inbacktick;
1827
Bram Moolenaar071d4272004-06-13 20:20:40 +00001828 inbacktick = FALSE;
1829 for (p = str; *str; ++str)
1830 {
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001831 /* When the backslash is used for escaping the special meaning of a
1832 * character we need to keep it until wildcard expansion. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001833 if (rem_backslash(str))
1834 {
1835 *p++ = *str++;
1836 *p++ = *str;
1837 }
1838 else
1839 {
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001840 /* An item ends at a space not in backticks */
1841 if (!inbacktick && vim_isspace(*str))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001842 break;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001843 if (*str == '`')
Bram Moolenaar071d4272004-06-13 20:20:40 +00001844 inbacktick ^= TRUE;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001845 *p++ = *str;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001846 }
1847 }
1848 str = skipwhite(str);
1849 *p = NUL;
1850
1851 return str;
1852}
1853
Bram Moolenaar86b68352004-12-27 21:59:20 +00001854/*
1855 * Separate the arguments in "str" and return a list of pointers in the
1856 * growarray "gap".
1857 */
1858 int
1859get_arglist(gap, str)
1860 garray_T *gap;
1861 char_u *str;
1862{
1863 ga_init2(gap, (int)sizeof(char_u *), 20);
1864 while (*str != NUL)
1865 {
1866 if (ga_grow(gap, 1) == FAIL)
1867 {
1868 ga_clear(gap);
1869 return FAIL;
1870 }
1871 ((char_u **)gap->ga_data)[gap->ga_len++] = str;
1872
1873 /* Isolate one argument, change it in-place, put a NUL after it. */
1874 str = do_one_arg(str);
1875 }
1876 return OK;
1877}
1878
Bram Moolenaar7df351e2006-01-23 22:30:28 +00001879#if defined(FEAT_QUICKFIX) || defined(FEAT_SYN_HL) || defined(PROTO)
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00001880/*
1881 * Parse a list of arguments (file names), expand them and return in
Bram Moolenaar8f5c6f02012-06-29 12:57:06 +02001882 * "fnames[fcountp]". When "wig" is TRUE, removes files matching 'wildignore'.
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00001883 * Return FAIL or OK.
1884 */
1885 int
Bram Moolenaar8f5c6f02012-06-29 12:57:06 +02001886get_arglist_exp(str, fcountp, fnamesp, wig)
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00001887 char_u *str;
1888 int *fcountp;
1889 char_u ***fnamesp;
Bram Moolenaar8f5c6f02012-06-29 12:57:06 +02001890 int wig;
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00001891{
1892 garray_T ga;
1893 int i;
1894
1895 if (get_arglist(&ga, str) == FAIL)
1896 return FAIL;
Bram Moolenaar8f5c6f02012-06-29 12:57:06 +02001897 if (wig == TRUE)
1898 i = expand_wildcards(ga.ga_len, (char_u **)ga.ga_data,
1899 fcountp, fnamesp, EW_FILE|EW_NOTFOUND);
1900 else
1901 i = gen_expand_wildcards(ga.ga_len, (char_u **)ga.ga_data,
1902 fcountp, fnamesp, EW_FILE|EW_NOTFOUND);
1903
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00001904 ga_clear(&ga);
1905 return i;
1906}
1907#endif
1908
Bram Moolenaar071d4272004-06-13 20:20:40 +00001909#if defined(FEAT_GUI) || defined(FEAT_CLIENTSERVER) || defined(PROTO)
1910/*
1911 * Redefine the argument list.
1912 */
1913 void
1914set_arglist(str)
1915 char_u *str;
1916{
1917 do_arglist(str, AL_SET, 0);
1918}
1919#endif
1920
1921/*
1922 * "what" == AL_SET: Redefine the argument list to 'str'.
1923 * "what" == AL_ADD: add files in 'str' to the argument list after "after".
1924 * "what" == AL_DEL: remove files in 'str' from the argument list.
1925 *
1926 * Return FAIL for failure, OK otherwise.
1927 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001928 static int
1929do_arglist(str, what, after)
1930 char_u *str;
Bram Moolenaar0c094b92009-05-14 20:20:33 +00001931 int what UNUSED;
1932 int after UNUSED; /* 0 means before first one */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001933{
1934 garray_T new_ga;
1935 int exp_count;
1936 char_u **exp_files;
1937 int i;
1938#ifdef FEAT_LISTCMDS
1939 char_u *p;
1940 int match;
1941#endif
1942
1943 /*
1944 * Collect all file name arguments in "new_ga".
1945 */
Bram Moolenaar86b68352004-12-27 21:59:20 +00001946 if (get_arglist(&new_ga, str) == FAIL)
1947 return FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001948
1949#ifdef FEAT_LISTCMDS
1950 if (what == AL_DEL)
1951 {
1952 regmatch_T regmatch;
1953 int didone;
1954
1955 /*
1956 * Delete the items: use each item as a regexp and find a match in the
1957 * argument list.
1958 */
Bram Moolenaar71afbfe2013-03-19 16:49:16 +01001959 regmatch.rm_ic = p_fic; /* ignore case when 'fileignorecase' is set */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001960 for (i = 0; i < new_ga.ga_len && !got_int; ++i)
1961 {
1962 p = ((char_u **)new_ga.ga_data)[i];
1963 p = file_pat_to_reg_pat(p, NULL, NULL, FALSE);
1964 if (p == NULL)
1965 break;
1966 regmatch.regprog = vim_regcomp(p, p_magic ? RE_MAGIC : 0);
1967 if (regmatch.regprog == NULL)
1968 {
1969 vim_free(p);
1970 break;
1971 }
1972
1973 didone = FALSE;
1974 for (match = 0; match < ARGCOUNT; ++match)
1975 if (vim_regexec(&regmatch, alist_name(&ARGLIST[match]),
1976 (colnr_T)0))
1977 {
1978 didone = TRUE;
1979 vim_free(ARGLIST[match].ae_fname);
1980 mch_memmove(ARGLIST + match, ARGLIST + match + 1,
1981 (ARGCOUNT - match - 1) * sizeof(aentry_T));
1982 --ALIST(curwin)->al_ga.ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001983 if (curwin->w_arg_idx > match)
1984 --curwin->w_arg_idx;
1985 --match;
1986 }
1987
Bram Moolenaar473de612013-06-08 18:19:48 +02001988 vim_regfree(regmatch.regprog);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001989 vim_free(p);
1990 if (!didone)
1991 EMSG2(_(e_nomatch2), ((char_u **)new_ga.ga_data)[i]);
1992 }
1993 ga_clear(&new_ga);
1994 }
1995 else
1996#endif
1997 {
1998 i = expand_wildcards(new_ga.ga_len, (char_u **)new_ga.ga_data,
1999 &exp_count, &exp_files, EW_DIR|EW_FILE|EW_ADDSLASH|EW_NOTFOUND);
2000 ga_clear(&new_ga);
2001 if (i == FAIL)
2002 return FAIL;
2003 if (exp_count == 0)
2004 {
2005 EMSG(_(e_nomatch));
2006 return FAIL;
2007 }
2008
2009#ifdef FEAT_LISTCMDS
2010 if (what == AL_ADD)
2011 {
2012 (void)alist_add_list(exp_count, exp_files, after);
2013 vim_free(exp_files);
2014 }
2015 else /* what == AL_SET */
2016#endif
Bram Moolenaar86b68352004-12-27 21:59:20 +00002017 alist_set(ALIST(curwin), exp_count, exp_files, FALSE, NULL, 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002018 }
2019
2020 alist_check_arg_idx();
2021
2022 return OK;
2023}
2024
2025/*
2026 * Check the validity of the arg_idx for each other window.
2027 */
2028 static void
2029alist_check_arg_idx()
2030{
2031#ifdef FEAT_WINDOWS
2032 win_T *win;
Bram Moolenaarf740b292006-02-16 22:11:02 +00002033 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002034
Bram Moolenaarf740b292006-02-16 22:11:02 +00002035 FOR_ALL_TAB_WINDOWS(tp, win)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002036 if (win->w_alist == curwin->w_alist)
2037 check_arg_idx(win);
2038#else
2039 check_arg_idx(curwin);
2040#endif
2041}
2042
2043/*
Bram Moolenaarb8ff1fb2012-02-04 21:59:01 +01002044 * Return TRUE if window "win" is editing the file at the current argument
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002045 * index.
2046 */
2047 static int
2048editing_arg_idx(win)
2049 win_T *win;
2050{
2051 return !(win->w_arg_idx >= WARGCOUNT(win)
2052 || (win->w_buffer->b_fnum
2053 != WARGLIST(win)[win->w_arg_idx].ae_fnum
2054 && (win->w_buffer->b_ffname == NULL
2055 || !(fullpathcmp(
2056 alist_name(&WARGLIST(win)[win->w_arg_idx]),
2057 win->w_buffer->b_ffname, TRUE) & FPC_SAME))));
2058}
2059
2060/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002061 * Check if window "win" is editing the w_arg_idx file in its argument list.
2062 */
2063 void
2064check_arg_idx(win)
2065 win_T *win;
2066{
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002067 if (WARGCOUNT(win) > 1 && !editing_arg_idx(win))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002068 {
2069 /* We are not editing the current entry in the argument list.
2070 * Set "arg_had_last" if we are editing the last one. */
2071 win->w_arg_idx_invalid = TRUE;
2072 if (win->w_arg_idx != WARGCOUNT(win) - 1
2073 && arg_had_last == FALSE
2074#ifdef FEAT_WINDOWS
2075 && ALIST(win) == &global_alist
2076#endif
2077 && GARGCOUNT > 0
2078 && win->w_arg_idx < GARGCOUNT
2079 && (win->w_buffer->b_fnum == GARGLIST[GARGCOUNT - 1].ae_fnum
2080 || (win->w_buffer->b_ffname != NULL
2081 && (fullpathcmp(alist_name(&GARGLIST[GARGCOUNT - 1]),
2082 win->w_buffer->b_ffname, TRUE) & FPC_SAME))))
2083 arg_had_last = TRUE;
2084 }
2085 else
2086 {
2087 /* We are editing the current entry in the argument list.
2088 * Set "arg_had_last" if it's also the last one */
2089 win->w_arg_idx_invalid = FALSE;
2090 if (win->w_arg_idx == WARGCOUNT(win) - 1
2091#ifdef FEAT_WINDOWS
2092 && win->w_alist == &global_alist
2093#endif
2094 )
2095 arg_had_last = TRUE;
2096 }
2097}
2098
2099/*
2100 * ":args", ":argslocal" and ":argsglobal".
2101 */
2102 void
2103ex_args(eap)
2104 exarg_T *eap;
2105{
2106 int i;
2107
2108 if (eap->cmdidx != CMD_args)
2109 {
2110#if defined(FEAT_WINDOWS) && defined(FEAT_LISTCMDS)
2111 alist_unlink(ALIST(curwin));
2112 if (eap->cmdidx == CMD_argglobal)
2113 ALIST(curwin) = &global_alist;
2114 else /* eap->cmdidx == CMD_arglocal */
2115 alist_new();
2116#else
2117 ex_ni(eap);
2118 return;
2119#endif
2120 }
2121
2122 if (!ends_excmd(*eap->arg))
2123 {
2124 /*
2125 * ":args file ..": define new argument list, handle like ":next"
2126 * Also for ":argslocal file .." and ":argsglobal file ..".
2127 */
2128 ex_next(eap);
2129 }
2130 else
2131#if defined(FEAT_WINDOWS) && defined(FEAT_LISTCMDS)
2132 if (eap->cmdidx == CMD_args)
2133#endif
2134 {
2135 /*
2136 * ":args": list arguments.
2137 */
2138 if (ARGCOUNT > 0)
2139 {
2140 /* Overwrite the command, for a short list there is no scrolling
2141 * required and no wait_return(). */
2142 gotocmdline(TRUE);
2143 for (i = 0; i < ARGCOUNT; ++i)
2144 {
2145 if (i == curwin->w_arg_idx)
2146 msg_putchar('[');
2147 msg_outtrans(alist_name(&ARGLIST[i]));
2148 if (i == curwin->w_arg_idx)
2149 msg_putchar(']');
2150 msg_putchar(' ');
2151 }
2152 }
2153 }
2154#if defined(FEAT_WINDOWS) && defined(FEAT_LISTCMDS)
2155 else if (eap->cmdidx == CMD_arglocal)
2156 {
2157 garray_T *gap = &curwin->w_alist->al_ga;
2158
2159 /*
2160 * ":argslocal": make a local copy of the global argument list.
2161 */
2162 if (ga_grow(gap, GARGCOUNT) == OK)
2163 for (i = 0; i < GARGCOUNT; ++i)
2164 if (GARGLIST[i].ae_fname != NULL)
2165 {
2166 AARGLIST(curwin->w_alist)[gap->ga_len].ae_fname =
2167 vim_strsave(GARGLIST[i].ae_fname);
2168 AARGLIST(curwin->w_alist)[gap->ga_len].ae_fnum =
2169 GARGLIST[i].ae_fnum;
2170 ++gap->ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002171 }
2172 }
2173#endif
2174}
2175
2176/*
2177 * ":previous", ":sprevious", ":Next" and ":sNext".
2178 */
2179 void
2180ex_previous(eap)
2181 exarg_T *eap;
2182{
2183 /* If past the last one already, go to the last one. */
2184 if (curwin->w_arg_idx - (int)eap->line2 >= ARGCOUNT)
2185 do_argfile(eap, ARGCOUNT - 1);
2186 else
2187 do_argfile(eap, curwin->w_arg_idx - (int)eap->line2);
2188}
2189
2190/*
2191 * ":rewind", ":first", ":sfirst" and ":srewind".
2192 */
2193 void
2194ex_rewind(eap)
2195 exarg_T *eap;
2196{
2197 do_argfile(eap, 0);
2198}
2199
2200/*
2201 * ":last" and ":slast".
2202 */
2203 void
2204ex_last(eap)
2205 exarg_T *eap;
2206{
2207 do_argfile(eap, ARGCOUNT - 1);
2208}
2209
2210/*
2211 * ":argument" and ":sargument".
2212 */
2213 void
2214ex_argument(eap)
2215 exarg_T *eap;
2216{
2217 int i;
2218
2219 if (eap->addr_count > 0)
2220 i = eap->line2 - 1;
2221 else
2222 i = curwin->w_arg_idx;
2223 do_argfile(eap, i);
2224}
2225
2226/*
2227 * Edit file "argn" of the argument lists.
2228 */
2229 void
2230do_argfile(eap, argn)
2231 exarg_T *eap;
2232 int argn;
2233{
2234 int other;
2235 char_u *p;
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002236 int old_arg_idx = curwin->w_arg_idx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002237
2238 if (argn < 0 || argn >= ARGCOUNT)
2239 {
2240 if (ARGCOUNT <= 1)
2241 EMSG(_("E163: There is only one file to edit"));
2242 else if (argn < 0)
2243 EMSG(_("E164: Cannot go before first file"));
2244 else
2245 EMSG(_("E165: Cannot go beyond last file"));
2246 }
2247 else
2248 {
2249 setpcmark();
2250#ifdef FEAT_GUI
2251 need_mouse_correct = TRUE;
2252#endif
2253
2254#ifdef FEAT_WINDOWS
Bram Moolenaardf1bdc92006-02-23 21:32:16 +00002255 /* split window or create new tab page first */
2256 if (*eap->cmd == 's' || cmdmod.tab != 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002257 {
2258 if (win_split(0, 0) == FAIL)
2259 return;
Bram Moolenaar3368ea22010-09-21 16:56:35 +02002260 RESET_BINDING(curwin);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002261 }
2262 else
2263#endif
2264 {
2265 /*
2266 * if 'hidden' set, only check for changed file when re-editing
2267 * the same buffer
2268 */
2269 other = TRUE;
2270 if (P_HID(curbuf))
2271 {
2272 p = fix_fname(alist_name(&ARGLIST[argn]));
2273 other = otherfile(p);
2274 vim_free(p);
2275 }
2276 if ((!P_HID(curbuf) || !other)
2277 && check_changed(curbuf, TRUE, !other, eap->forceit, FALSE))
2278 return;
2279 }
2280
2281 curwin->w_arg_idx = argn;
2282 if (argn == ARGCOUNT - 1
2283#ifdef FEAT_WINDOWS
2284 && curwin->w_alist == &global_alist
2285#endif
2286 )
2287 arg_had_last = TRUE;
2288
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002289 /* Edit the file; always use the last known line number.
2290 * When it fails (e.g. Abort for already edited file) restore the
2291 * argument index. */
2292 if (do_ecmd(0, alist_name(&ARGLIST[curwin->w_arg_idx]), NULL,
Bram Moolenaar071d4272004-06-13 20:20:40 +00002293 eap, ECMD_LAST,
Bram Moolenaar701f7af2008-11-15 13:12:07 +00002294 (P_HID(curwin->w_buffer) ? ECMD_HIDE : 0)
2295 + (eap->forceit ? ECMD_FORCEIT : 0), curwin) == FAIL)
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002296 curwin->w_arg_idx = old_arg_idx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002297 /* like Vi: set the mark where the cursor is in the file. */
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002298 else if (eap->cmdidx != CMD_argdo)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002299 setmark('\'');
2300 }
2301}
2302
2303/*
2304 * ":next", and commands that behave like it.
2305 */
2306 void
2307ex_next(eap)
2308 exarg_T *eap;
2309{
2310 int i;
2311
2312 /*
2313 * check for changed buffer now, if this fails the argument list is not
2314 * redefined.
2315 */
2316 if ( P_HID(curbuf)
2317 || eap->cmdidx == CMD_snext
2318 || !check_changed(curbuf, TRUE, FALSE, eap->forceit, FALSE))
2319 {
2320 if (*eap->arg != NUL) /* redefine file list */
2321 {
2322 if (do_arglist(eap->arg, AL_SET, 0) == FAIL)
2323 return;
2324 i = 0;
2325 }
2326 else
2327 i = curwin->w_arg_idx + (int)eap->line2;
2328 do_argfile(eap, i);
2329 }
2330}
2331
2332#ifdef FEAT_LISTCMDS
2333/*
2334 * ":argedit"
2335 */
2336 void
2337ex_argedit(eap)
2338 exarg_T *eap;
2339{
2340 int fnum;
2341 int i;
2342 char_u *s;
2343
2344 /* Add the argument to the buffer list and get the buffer number. */
2345 fnum = buflist_add(eap->arg, BLN_LISTED);
2346
2347 /* Check if this argument is already in the argument list. */
2348 for (i = 0; i < ARGCOUNT; ++i)
2349 if (ARGLIST[i].ae_fnum == fnum)
2350 break;
2351 if (i == ARGCOUNT)
2352 {
2353 /* Can't find it, add it to the argument list. */
2354 s = vim_strsave(eap->arg);
2355 if (s == NULL)
2356 return;
2357 i = alist_add_list(1, &s,
2358 eap->addr_count > 0 ? (int)eap->line2 : curwin->w_arg_idx + 1);
2359 if (i < 0)
2360 return;
2361 curwin->w_arg_idx = i;
2362 }
2363
2364 alist_check_arg_idx();
2365
2366 /* Edit the argument. */
2367 do_argfile(eap, i);
2368}
2369
2370/*
2371 * ":argadd"
2372 */
2373 void
2374ex_argadd(eap)
2375 exarg_T *eap;
2376{
2377 do_arglist(eap->arg, AL_ADD,
2378 eap->addr_count > 0 ? (int)eap->line2 : curwin->w_arg_idx + 1);
2379#ifdef FEAT_TITLE
2380 maketitle();
2381#endif
2382}
2383
2384/*
2385 * ":argdelete"
2386 */
2387 void
2388ex_argdelete(eap)
2389 exarg_T *eap;
2390{
2391 int i;
2392 int n;
2393
2394 if (eap->addr_count > 0)
2395 {
2396 /* ":1,4argdel": Delete all arguments in the range. */
2397 if (eap->line2 > ARGCOUNT)
2398 eap->line2 = ARGCOUNT;
2399 n = eap->line2 - eap->line1 + 1;
2400 if (*eap->arg != NUL || n <= 0)
2401 EMSG(_(e_invarg));
2402 else
2403 {
2404 for (i = eap->line1; i <= eap->line2; ++i)
2405 vim_free(ARGLIST[i - 1].ae_fname);
2406 mch_memmove(ARGLIST + eap->line1 - 1, ARGLIST + eap->line2,
2407 (size_t)((ARGCOUNT - eap->line2) * sizeof(aentry_T)));
2408 ALIST(curwin)->al_ga.ga_len -= n;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002409 if (curwin->w_arg_idx >= eap->line2)
2410 curwin->w_arg_idx -= n;
2411 else if (curwin->w_arg_idx > eap->line1)
2412 curwin->w_arg_idx = eap->line1;
2413 }
2414 }
2415 else if (*eap->arg == NUL)
2416 EMSG(_(e_argreq));
2417 else
2418 do_arglist(eap->arg, AL_DEL, 0);
2419#ifdef FEAT_TITLE
2420 maketitle();
2421#endif
2422}
2423
2424/*
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002425 * ":argdo", ":windo", ":bufdo", ":tabdo"
Bram Moolenaar071d4272004-06-13 20:20:40 +00002426 */
2427 void
2428ex_listdo(eap)
2429 exarg_T *eap;
2430{
2431 int i;
2432#ifdef FEAT_WINDOWS
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002433 win_T *wp;
2434 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002435#endif
2436 buf_T *buf;
2437 int next_fnum = 0;
2438#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
2439 char_u *save_ei = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002440#endif
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002441 char_u *p_shm_save;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002442
2443#ifndef FEAT_WINDOWS
2444 if (eap->cmdidx == CMD_windo)
2445 {
2446 ex_ni(eap);
2447 return;
2448 }
2449#endif
2450
2451#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002452 if (eap->cmdidx != CMD_windo && eap->cmdidx != CMD_tabdo)
Bram Moolenaardcaf10e2005-01-21 11:55:25 +00002453 /* Don't do syntax HL autocommands. Skipping the syntax file is a
2454 * great speed improvement. */
2455 save_ei = au_event_disable(",Syntax");
Bram Moolenaar071d4272004-06-13 20:20:40 +00002456#endif
2457
2458 if (eap->cmdidx == CMD_windo
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002459 || eap->cmdidx == CMD_tabdo
Bram Moolenaar071d4272004-06-13 20:20:40 +00002460 || P_HID(curbuf)
2461 || !check_changed(curbuf, TRUE, FALSE, eap->forceit, FALSE))
2462 {
2463 /* start at the first argument/window/buffer */
2464 i = 0;
2465#ifdef FEAT_WINDOWS
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002466 wp = firstwin;
2467 tp = first_tabpage;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002468#endif
2469 /* set pcmark now */
2470 if (eap->cmdidx == CMD_bufdo)
2471 goto_buffer(eap, DOBUF_FIRST, FORWARD, 0);
2472 else
2473 setpcmark();
2474 listcmd_busy = TRUE; /* avoids setting pcmark below */
2475
2476 while (!got_int)
2477 {
2478 if (eap->cmdidx == CMD_argdo)
2479 {
2480 /* go to argument "i" */
2481 if (i == ARGCOUNT)
2482 break;
2483 /* Don't call do_argfile() when already there, it will try
2484 * reloading the file. */
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002485 if (curwin->w_arg_idx != i || !editing_arg_idx(curwin))
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002486 {
2487 /* Clear 'shm' to avoid that the file message overwrites
2488 * any output from the command. */
2489 p_shm_save = vim_strsave(p_shm);
2490 set_option_value((char_u *)"shm", 0L, (char_u *)"", 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002491 do_argfile(eap, i);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002492 set_option_value((char_u *)"shm", 0L, p_shm_save, 0);
2493 vim_free(p_shm_save);
2494 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002495 if (curwin->w_arg_idx != i)
2496 break;
2497 ++i;
2498 }
2499#ifdef FEAT_WINDOWS
2500 else if (eap->cmdidx == CMD_windo)
2501 {
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002502 /* go to window "wp" */
2503 if (!win_valid(wp))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002504 break;
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002505 win_goto(wp);
Bram Moolenaar41423242007-05-03 20:11:13 +00002506 if (curwin != wp)
2507 break; /* something must be wrong */
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002508 wp = curwin->w_next;
2509 }
2510 else if (eap->cmdidx == CMD_tabdo)
2511 {
2512 /* go to window "tp" */
2513 if (!valid_tabpage(tp))
2514 break;
Bram Moolenaar49e649f2013-05-06 04:50:35 +02002515 goto_tabpage_tp(tp, TRUE, TRUE);
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002516 tp = tp->tp_next;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002517 }
2518#endif
2519 else if (eap->cmdidx == CMD_bufdo)
2520 {
2521 /* Remember the number of the next listed buffer, in case
2522 * ":bwipe" is used or autocommands do something strange. */
2523 next_fnum = -1;
2524 for (buf = curbuf->b_next; buf != NULL; buf = buf->b_next)
2525 if (buf->b_p_bl)
2526 {
2527 next_fnum = buf->b_fnum;
2528 break;
2529 }
2530 }
2531
2532 /* execute the command */
2533 do_cmdline(eap->arg, eap->getline, eap->cookie,
2534 DOCMD_VERBOSE + DOCMD_NOWAIT);
2535
2536 if (eap->cmdidx == CMD_bufdo)
2537 {
2538 /* Done? */
2539 if (next_fnum < 0)
2540 break;
2541 /* Check if the buffer still exists. */
2542 for (buf = firstbuf; buf != NULL; buf = buf->b_next)
2543 if (buf->b_fnum == next_fnum)
2544 break;
2545 if (buf == NULL)
2546 break;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002547
2548 /* Go to the next buffer. Clear 'shm' to avoid that the file
2549 * message overwrites any output from the command. */
2550 p_shm_save = vim_strsave(p_shm);
2551 set_option_value((char_u *)"shm", 0L, (char_u *)"", 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002552 goto_buffer(eap, DOBUF_FIRST, FORWARD, next_fnum);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002553 set_option_value((char_u *)"shm", 0L, p_shm_save, 0);
2554 vim_free(p_shm_save);
2555
Bram Moolenaar071d4272004-06-13 20:20:40 +00002556 /* If autocommands took us elsewhere, quit here */
2557 if (curbuf->b_fnum != next_fnum)
2558 break;
2559 }
2560
2561 if (eap->cmdidx == CMD_windo)
2562 {
2563 validate_cursor(); /* cursor may have moved */
2564#ifdef FEAT_SCROLLBIND
2565 /* required when 'scrollbind' has been set */
2566 if (curwin->w_p_scb)
2567 do_check_scrollbind(TRUE);
2568#endif
2569 }
2570 }
2571 listcmd_busy = FALSE;
2572 }
2573
2574#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
Bram Moolenaar6ac54292005-02-02 23:07:25 +00002575 if (save_ei != NULL)
2576 {
2577 au_event_restore(save_ei);
2578 apply_autocmds(EVENT_SYNTAX, curbuf->b_p_syn,
2579 curbuf->b_fname, TRUE, curbuf);
2580 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002581#endif
2582}
2583
2584/*
2585 * Add files[count] to the arglist of the current window after arg "after".
2586 * The file names in files[count] must have been allocated and are taken over.
2587 * Files[] itself is not taken over.
2588 * Returns index of first added argument. Returns -1 when failed (out of mem).
2589 */
2590 static int
2591alist_add_list(count, files, after)
2592 int count;
2593 char_u **files;
2594 int after; /* where to add: 0 = before first one */
2595{
2596 int i;
2597
2598 if (ga_grow(&ALIST(curwin)->al_ga, count) == OK)
2599 {
2600 if (after < 0)
2601 after = 0;
2602 if (after > ARGCOUNT)
2603 after = ARGCOUNT;
2604 if (after < ARGCOUNT)
2605 mch_memmove(&(ARGLIST[after + count]), &(ARGLIST[after]),
2606 (ARGCOUNT - after) * sizeof(aentry_T));
2607 for (i = 0; i < count; ++i)
2608 {
2609 ARGLIST[after + i].ae_fname = files[i];
2610 ARGLIST[after + i].ae_fnum = buflist_add(files[i], BLN_LISTED);
2611 }
2612 ALIST(curwin)->al_ga.ga_len += count;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002613 if (curwin->w_arg_idx >= after)
2614 ++curwin->w_arg_idx;
2615 return after;
2616 }
2617
2618 for (i = 0; i < count; ++i)
2619 vim_free(files[i]);
2620 return -1;
2621}
2622
2623#endif /* FEAT_LISTCMDS */
2624
2625#ifdef FEAT_EVAL
2626/*
2627 * ":compiler[!] {name}"
2628 */
2629 void
2630ex_compiler(eap)
2631 exarg_T *eap;
2632{
2633 char_u *buf;
2634 char_u *old_cur_comp = NULL;
2635 char_u *p;
2636
2637 if (*eap->arg == NUL)
2638 {
2639 /* List all compiler scripts. */
2640 do_cmdline_cmd((char_u *)"echo globpath(&rtp, 'compiler/*.vim')");
2641 /* ) keep the indenter happy... */
2642 }
2643 else
2644 {
2645 buf = alloc((unsigned)(STRLEN(eap->arg) + 14));
2646 if (buf != NULL)
2647 {
2648 if (eap->forceit)
2649 {
2650 /* ":compiler! {name}" sets global options */
2651 do_cmdline_cmd((char_u *)
2652 "command -nargs=* CompilerSet set <args>");
2653 }
2654 else
2655 {
2656 /* ":compiler! {name}" sets local options.
2657 * To remain backwards compatible "current_compiler" is always
2658 * used. A user's compiler plugin may set it, the distributed
2659 * plugin will then skip the settings. Afterwards set
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01002660 * "b:current_compiler" and restore "current_compiler".
2661 * Explicitly prepend "g:" to make it work in a function. */
2662 old_cur_comp = get_var_value((char_u *)"g:current_compiler");
Bram Moolenaar071d4272004-06-13 20:20:40 +00002663 if (old_cur_comp != NULL)
2664 old_cur_comp = vim_strsave(old_cur_comp);
2665 do_cmdline_cmd((char_u *)
2666 "command -nargs=* CompilerSet setlocal <args>");
2667 }
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01002668 do_unlet((char_u *)"g:current_compiler", TRUE);
Bram Moolenaar2ce06f62005-01-31 19:19:04 +00002669 do_unlet((char_u *)"b:current_compiler", TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002670
2671 sprintf((char *)buf, "compiler/%s.vim", eap->arg);
Bram Moolenaar90cfdbe2005-08-12 19:59:19 +00002672 if (source_runtime(buf, TRUE) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002673 EMSG2(_("E666: compiler not supported: %s"), eap->arg);
2674 vim_free(buf);
2675
2676 do_cmdline_cmd((char_u *)":delcommand CompilerSet");
2677
2678 /* Set "b:current_compiler" from "current_compiler". */
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01002679 p = get_var_value((char_u *)"g:current_compiler");
Bram Moolenaar071d4272004-06-13 20:20:40 +00002680 if (p != NULL)
2681 set_internal_string_var((char_u *)"b:current_compiler", p);
2682
2683 /* Restore "current_compiler" for ":compiler {name}". */
2684 if (!eap->forceit)
2685 {
2686 if (old_cur_comp != NULL)
2687 {
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01002688 set_internal_string_var((char_u *)"g:current_compiler",
Bram Moolenaar071d4272004-06-13 20:20:40 +00002689 old_cur_comp);
2690 vim_free(old_cur_comp);
2691 }
2692 else
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01002693 do_unlet((char_u *)"g:current_compiler", TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002694 }
2695 }
2696 }
2697}
2698#endif
2699
2700/*
2701 * ":runtime {name}"
2702 */
2703 void
2704ex_runtime(eap)
2705 exarg_T *eap;
2706{
Bram Moolenaar90cfdbe2005-08-12 19:59:19 +00002707 source_runtime(eap->arg, eap->forceit);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002708}
2709
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002710static void source_callback __ARGS((char_u *fname, void *cookie));
Bram Moolenaar071d4272004-06-13 20:20:40 +00002711
2712 static void
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002713source_callback(fname, cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002714 char_u *fname;
Bram Moolenaar0c094b92009-05-14 20:20:33 +00002715 void *cookie UNUSED;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002716{
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002717 (void)do_source(fname, FALSE, DOSO_NONE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002718}
2719
2720/*
2721 * Source the file "name" from all directories in 'runtimepath'.
2722 * "name" can contain wildcards.
2723 * When "all" is TRUE, source all files, otherwise only the first one.
2724 * return FAIL when no file could be sourced, OK otherwise.
2725 */
2726 int
Bram Moolenaar90cfdbe2005-08-12 19:59:19 +00002727source_runtime(name, all)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002728 char_u *name;
2729 int all;
2730{
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002731 return do_in_runtimepath(name, all, source_callback, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002732}
2733
2734/*
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002735 * Find "name" in 'runtimepath'. When found, invoke the callback function for
2736 * it: callback(fname, "cookie")
Bram Moolenaar071d4272004-06-13 20:20:40 +00002737 * When "all" is TRUE repeat for all matches, otherwise only the first one is
2738 * used.
2739 * Returns OK when at least one match found, FAIL otherwise.
2740 */
2741 int
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002742do_in_runtimepath(name, all, callback, cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002743 char_u *name;
2744 int all;
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002745 void (*callback)__ARGS((char_u *fname, void *ck));
2746 void *cookie;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002747{
2748 char_u *rtp;
2749 char_u *np;
2750 char_u *buf;
2751 char_u *rtp_copy;
2752 char_u *tail;
2753 int num_files;
2754 char_u **files;
2755 int i;
2756 int did_one = FALSE;
2757#ifdef AMIGA
2758 struct Process *proc = (struct Process *)FindTask(0L);
2759 APTR save_winptr = proc->pr_WindowPtr;
2760
2761 /* Avoid a requester here for a volume that doesn't exist. */
2762 proc->pr_WindowPtr = (APTR)-1L;
2763#endif
2764
2765 /* Make a copy of 'runtimepath'. Invoking the callback may change the
2766 * value. */
2767 rtp_copy = vim_strsave(p_rtp);
2768 buf = alloc(MAXPATHL);
2769 if (buf != NULL && rtp_copy != NULL)
2770 {
2771 if (p_verbose > 1)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00002772 {
2773 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00002774 smsg((char_u *)_("Searching for \"%s\" in \"%s\""),
Bram Moolenaar071d4272004-06-13 20:20:40 +00002775 (char *)name, (char *)p_rtp);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00002776 verbose_leave();
2777 }
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002778
Bram Moolenaar071d4272004-06-13 20:20:40 +00002779 /* Loop over all entries in 'runtimepath'. */
2780 rtp = rtp_copy;
2781 while (*rtp != NUL && (all || !did_one))
2782 {
2783 /* Copy the path from 'runtimepath' to buf[]. */
2784 copy_option_part(&rtp, buf, MAXPATHL, ",");
2785 if (STRLEN(buf) + STRLEN(name) + 2 < MAXPATHL)
2786 {
2787 add_pathsep(buf);
2788 tail = buf + STRLEN(buf);
2789
2790 /* Loop over all patterns in "name" */
2791 np = name;
2792 while (*np != NUL && (all || !did_one))
2793 {
2794 /* Append the pattern from "name" to buf[]. */
2795 copy_option_part(&np, tail, (int)(MAXPATHL - (tail - buf)),
2796 "\t ");
2797
2798 if (p_verbose > 2)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00002799 {
2800 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00002801 smsg((char_u *)_("Searching for \"%s\""), buf);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00002802 verbose_leave();
2803 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002804
2805 /* Expand wildcards, invoke the callback for each match. */
2806 if (gen_expand_wildcards(1, &buf, &num_files, &files,
2807 EW_FILE) == OK)
2808 {
2809 for (i = 0; i < num_files; ++i)
2810 {
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002811 (*callback)(files[i], cookie);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002812 did_one = TRUE;
2813 if (!all)
2814 break;
2815 }
2816 FreeWild(num_files, files);
2817 }
2818 }
2819 }
2820 }
2821 }
2822 vim_free(buf);
2823 vim_free(rtp_copy);
2824 if (p_verbose > 0 && !did_one)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00002825 {
2826 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00002827 smsg((char_u *)_("not found in 'runtimepath': \"%s\""), name);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00002828 verbose_leave();
2829 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002830
2831#ifdef AMIGA
2832 proc->pr_WindowPtr = save_winptr;
2833#endif
2834
2835 return did_one ? OK : FAIL;
2836}
2837
2838#if defined(FEAT_EVAL) && defined(FEAT_AUTOCMD)
2839/*
2840 * ":options"
2841 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002842 void
2843ex_options(eap)
Bram Moolenaar0c094b92009-05-14 20:20:33 +00002844 exarg_T *eap UNUSED;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002845{
2846 cmd_source((char_u *)SYS_OPTWIN_FILE, NULL);
2847}
2848#endif
2849
2850/*
2851 * ":source {fname}"
2852 */
2853 void
2854ex_source(eap)
2855 exarg_T *eap;
2856{
2857#ifdef FEAT_BROWSE
2858 if (cmdmod.browse)
2859 {
2860 char_u *fname = NULL;
2861
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00002862 fname = do_browse(0, (char_u *)_("Source Vim script"), eap->arg,
Bram Moolenaar071d4272004-06-13 20:20:40 +00002863 NULL, NULL, BROWSE_FILTER_MACROS, NULL);
2864 if (fname != NULL)
2865 {
2866 cmd_source(fname, eap);
2867 vim_free(fname);
2868 }
2869 }
2870 else
2871#endif
2872 cmd_source(eap->arg, eap);
2873}
2874
2875 static void
2876cmd_source(fname, eap)
2877 char_u *fname;
2878 exarg_T *eap;
2879{
2880 if (*fname == NUL)
2881 EMSG(_(e_argreq));
2882
Bram Moolenaar071d4272004-06-13 20:20:40 +00002883 else if (eap != NULL && eap->forceit)
Bram Moolenaar84a05ac2013-05-06 04:24:17 +02002884 /* ":source!": read Normal mode commands
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00002885 * Need to execute the commands directly. This is required at least
2886 * for:
Bram Moolenaar071d4272004-06-13 20:20:40 +00002887 * - ":g" command busy
2888 * - after ":argdo", ":windo" or ":bufdo"
2889 * - another command follows
2890 * - inside a loop
2891 */
2892 openscript(fname, global_busy || listcmd_busy || eap->nextcmd != NULL
2893#ifdef FEAT_EVAL
2894 || eap->cstack->cs_idx >= 0
2895#endif
2896 );
2897
2898 /* ":source" read ex commands */
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002899 else if (do_source(fname, FALSE, DOSO_NONE) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002900 EMSG2(_(e_notopen), fname);
2901}
2902
2903/*
2904 * ":source" and associated commands.
2905 */
2906/*
2907 * Structure used to store info for each sourced file.
2908 * It is shared between do_source() and getsourceline().
2909 * This is required, because it needs to be handed to do_cmdline() and
2910 * sourcing can be done recursively.
2911 */
2912struct source_cookie
2913{
2914 FILE *fp; /* opened file for sourcing */
2915 char_u *nextline; /* if not NULL: line that was read ahead */
2916 int finished; /* ":finish" used */
Bram Moolenaar860cae12010-06-05 23:22:07 +02002917#if defined(USE_CRNL) || defined(USE_CR)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002918 int fileformat; /* EOL_UNKNOWN, EOL_UNIX or EOL_DOS */
2919 int error; /* TRUE if LF found after CR-LF */
2920#endif
2921#ifdef FEAT_EVAL
2922 linenr_T breakpoint; /* next line with breakpoint or zero */
2923 char_u *fname; /* name of sourced file */
2924 int dbg_tick; /* debug_tick when breakpoint was set */
2925 int level; /* top nesting level of sourced file */
2926#endif
2927#ifdef FEAT_MBYTE
2928 vimconv_T conv; /* type of conversion */
2929#endif
2930};
2931
2932#ifdef FEAT_EVAL
2933/*
2934 * Return the address holding the next breakpoint line for a source cookie.
2935 */
2936 linenr_T *
2937source_breakpoint(cookie)
2938 void *cookie;
2939{
2940 return &((struct source_cookie *)cookie)->breakpoint;
2941}
2942
2943/*
2944 * Return the address holding the debug tick for a source cookie.
2945 */
2946 int *
2947source_dbg_tick(cookie)
2948 void *cookie;
2949{
2950 return &((struct source_cookie *)cookie)->dbg_tick;
2951}
2952
2953/*
2954 * Return the nesting level for a source cookie.
2955 */
2956 int
2957source_level(cookie)
2958 void *cookie;
2959{
2960 return ((struct source_cookie *)cookie)->level;
2961}
2962#endif
2963
2964static char_u *get_one_sourceline __ARGS((struct source_cookie *sp));
2965
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01002966#if (defined(WIN32) && defined(FEAT_CSCOPE)) || defined(HAVE_FD_CLOEXEC)
2967# define USE_FOPEN_NOINH
Bram Moolenaar071d4272004-06-13 20:20:40 +00002968static FILE *fopen_noinh_readbin __ARGS((char *filename));
2969
2970/*
2971 * Special function to open a file without handle inheritance.
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01002972 * When possible the handle is closed on exec().
Bram Moolenaar071d4272004-06-13 20:20:40 +00002973 */
2974 static FILE *
2975fopen_noinh_readbin(filename)
2976 char *filename;
2977{
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01002978# ifdef WIN32
Bram Moolenaar8d8ef0b2010-01-20 21:41:47 +01002979 int fd_tmp = mch_open(filename, O_RDONLY | O_BINARY | O_NOINHERIT, 0);
2980# else
2981 int fd_tmp = mch_open(filename, O_RDONLY, 0);
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01002982# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002983
2984 if (fd_tmp == -1)
2985 return NULL;
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01002986
2987# ifdef HAVE_FD_CLOEXEC
2988 {
2989 int fdflags = fcntl(fd_tmp, F_GETFD);
2990 if (fdflags >= 0 && (fdflags & FD_CLOEXEC) == 0)
2991 fcntl(fd_tmp, F_SETFD, fdflags | FD_CLOEXEC);
2992 }
2993# endif
2994
Bram Moolenaar071d4272004-06-13 20:20:40 +00002995 return fdopen(fd_tmp, READBIN);
2996}
2997#endif
2998
2999
3000/*
3001 * do_source: Read the file "fname" and execute its lines as EX commands.
3002 *
3003 * This function may be called recursively!
3004 *
3005 * return FAIL if file could not be opened, OK otherwise
3006 */
3007 int
3008do_source(fname, check_other, is_vimrc)
3009 char_u *fname;
3010 int check_other; /* check for .vimrc and _vimrc */
Bram Moolenaar910f66f2006-04-05 20:41:53 +00003011 int is_vimrc; /* DOSO_ value */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003012{
3013 struct source_cookie cookie;
3014 char_u *save_sourcing_name;
3015 linenr_T save_sourcing_lnum;
3016 char_u *p;
3017 char_u *fname_exp;
Bram Moolenaar73881402009-02-04 16:50:47 +00003018 char_u *firstline = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003019 int retval = FAIL;
3020#ifdef FEAT_EVAL
3021 scid_T save_current_SID;
3022 static scid_T last_current_SID = 0;
3023 void *save_funccalp;
3024 int save_debug_break_level = debug_break_level;
Bram Moolenaar05159a02005-02-26 23:04:13 +00003025 scriptitem_T *si = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003026# ifdef UNIX
3027 struct stat st;
3028 int stat_ok;
3029# endif
3030#endif
3031#ifdef STARTUPTIME
3032 struct timeval tv_rel;
3033 struct timeval tv_start;
3034#endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00003035#ifdef FEAT_PROFILE
3036 proftime_T wait_start;
3037#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003038
Bram Moolenaar071d4272004-06-13 20:20:40 +00003039 p = expand_env_save(fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003040 if (p == NULL)
3041 return retval;
3042 fname_exp = fix_fname(p);
3043 vim_free(p);
3044 if (fname_exp == NULL)
3045 return retval;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003046 if (mch_isdir(fname_exp))
3047 {
Bram Moolenaar555b2802005-05-19 21:08:39 +00003048 smsg((char_u *)_("Cannot source a directory: \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003049 goto theend;
3050 }
3051
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00003052#ifdef FEAT_AUTOCMD
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00003053 /* Apply SourceCmd autocommands, they should get the file and source it. */
3054 if (has_autocmd(EVENT_SOURCECMD, fname_exp, NULL)
3055 && apply_autocmds(EVENT_SOURCECMD, fname_exp, fname_exp,
3056 FALSE, curbuf))
Bram Moolenaarf33943e2008-01-15 21:17:30 +00003057 {
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00003058# ifdef FEAT_EVAL
Bram Moolenaarf33943e2008-01-15 21:17:30 +00003059 retval = aborting() ? FAIL : OK;
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00003060# else
Bram Moolenaarf33943e2008-01-15 21:17:30 +00003061 retval = OK;
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00003062# endif
Bram Moolenaarf33943e2008-01-15 21:17:30 +00003063 goto theend;
3064 }
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00003065
3066 /* Apply SourcePre autocommands, they may get the file. */
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00003067 apply_autocmds(EVENT_SOURCEPRE, fname_exp, fname_exp, FALSE, curbuf);
3068#endif
3069
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01003070#ifdef USE_FOPEN_NOINH
Bram Moolenaar071d4272004-06-13 20:20:40 +00003071 cookie.fp = fopen_noinh_readbin((char *)fname_exp);
3072#else
3073 cookie.fp = mch_fopen((char *)fname_exp, READBIN);
3074#endif
3075 if (cookie.fp == NULL && check_other)
3076 {
3077 /*
3078 * Try again, replacing file name ".vimrc" by "_vimrc" or vice versa,
3079 * and ".exrc" by "_exrc" or vice versa.
3080 */
3081 p = gettail(fname_exp);
3082 if ((*p == '.' || *p == '_')
3083 && (STRICMP(p + 1, "vimrc") == 0
3084 || STRICMP(p + 1, "gvimrc") == 0
3085 || STRICMP(p + 1, "exrc") == 0))
3086 {
3087 if (*p == '_')
3088 *p = '.';
3089 else
3090 *p = '_';
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01003091#ifdef USE_FOPEN_NOINH
Bram Moolenaar071d4272004-06-13 20:20:40 +00003092 cookie.fp = fopen_noinh_readbin((char *)fname_exp);
3093#else
3094 cookie.fp = mch_fopen((char *)fname_exp, READBIN);
3095#endif
3096 }
3097 }
3098
3099 if (cookie.fp == NULL)
3100 {
3101 if (p_verbose > 0)
3102 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003103 verbose_enter();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003104 if (sourcing_name == NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00003105 smsg((char_u *)_("could not source \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003106 else
3107 smsg((char_u *)_("line %ld: could not source \"%s\""),
Bram Moolenaar555b2802005-05-19 21:08:39 +00003108 sourcing_lnum, fname);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003109 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003110 }
3111 goto theend;
3112 }
3113
3114 /*
3115 * The file exists.
3116 * - In verbose mode, give a message.
3117 * - For a vimrc file, may want to set 'compatible', call vimrc_found().
3118 */
3119 if (p_verbose > 1)
3120 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003121 verbose_enter();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003122 if (sourcing_name == NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00003123 smsg((char_u *)_("sourcing \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003124 else
3125 smsg((char_u *)_("line %ld: sourcing \"%s\""),
Bram Moolenaar555b2802005-05-19 21:08:39 +00003126 sourcing_lnum, fname);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003127 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003128 }
Bram Moolenaar910f66f2006-04-05 20:41:53 +00003129 if (is_vimrc == DOSO_VIMRC)
3130 vimrc_found(fname_exp, (char_u *)"MYVIMRC");
3131 else if (is_vimrc == DOSO_GVIMRC)
3132 vimrc_found(fname_exp, (char_u *)"MYGVIMRC");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003133
3134#ifdef USE_CRNL
3135 /* If no automatic file format: Set default to CR-NL. */
3136 if (*p_ffs == NUL)
3137 cookie.fileformat = EOL_DOS;
3138 else
3139 cookie.fileformat = EOL_UNKNOWN;
3140 cookie.error = FALSE;
3141#endif
3142
3143#ifdef USE_CR
3144 /* If no automatic file format: Set default to CR. */
3145 if (*p_ffs == NUL)
3146 cookie.fileformat = EOL_MAC;
3147 else
3148 cookie.fileformat = EOL_UNKNOWN;
3149 cookie.error = FALSE;
3150#endif
3151
3152 cookie.nextline = NULL;
3153 cookie.finished = FALSE;
3154
3155#ifdef FEAT_EVAL
3156 /*
3157 * Check if this script has a breakpoint.
3158 */
3159 cookie.breakpoint = dbg_find_breakpoint(TRUE, fname_exp, (linenr_T)0);
3160 cookie.fname = fname_exp;
3161 cookie.dbg_tick = debug_tick;
3162
3163 cookie.level = ex_nesting_level;
3164#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003165
3166 /*
3167 * Keep the sourcing name/lnum, for recursive calls.
3168 */
3169 save_sourcing_name = sourcing_name;
3170 sourcing_name = fname_exp;
3171 save_sourcing_lnum = sourcing_lnum;
3172 sourcing_lnum = 0;
3173
Bram Moolenaar73881402009-02-04 16:50:47 +00003174#ifdef FEAT_MBYTE
3175 cookie.conv.vc_type = CONV_NONE; /* no conversion */
3176
3177 /* Read the first line so we can check for a UTF-8 BOM. */
3178 firstline = getsourceline(0, (void *)&cookie, 0);
3179 if (firstline != NULL && STRLEN(firstline) >= 3 && firstline[0] == 0xef
3180 && firstline[1] == 0xbb && firstline[2] == 0xbf)
3181 {
3182 /* Found BOM; setup conversion, skip over BOM and recode the line. */
3183 convert_setup(&cookie.conv, (char_u *)"utf-8", p_enc);
3184 p = string_convert(&cookie.conv, firstline + 3, NULL);
Bram Moolenaar4e2a5952009-02-05 19:48:25 +00003185 if (p == NULL)
3186 p = vim_strsave(firstline + 3);
Bram Moolenaar73881402009-02-04 16:50:47 +00003187 if (p != NULL)
3188 {
3189 vim_free(firstline);
3190 firstline = p;
3191 }
3192 }
3193#endif
3194
Bram Moolenaar071d4272004-06-13 20:20:40 +00003195#ifdef STARTUPTIME
Bram Moolenaarc4e41982010-01-19 16:31:47 +01003196 if (time_fd != NULL)
3197 time_push(&tv_rel, &tv_start);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003198#endif
3199
3200#ifdef FEAT_EVAL
Bram Moolenaar05159a02005-02-26 23:04:13 +00003201# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00003202 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003203 prof_child_enter(&wait_start); /* entering a child now */
3204# endif
3205
3206 /* Don't use local function variables, if called from a function.
3207 * Also starts profiling timer for nested script. */
3208 save_funccalp = save_funccal();
3209
Bram Moolenaar071d4272004-06-13 20:20:40 +00003210 /*
3211 * Check if this script was sourced before to finds its SID.
3212 * If it's new, generate a new SID.
3213 */
3214 save_current_SID = current_SID;
3215# ifdef UNIX
3216 stat_ok = (mch_stat((char *)fname_exp, &st) >= 0);
3217# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00003218 for (current_SID = script_items.ga_len; current_SID > 0; --current_SID)
3219 {
3220 si = &SCRIPT_ITEM(current_SID);
3221 if (si->sn_name != NULL
Bram Moolenaar071d4272004-06-13 20:20:40 +00003222 && (
3223# ifdef UNIX
Bram Moolenaar7c626922005-02-07 22:01:03 +00003224 /* Compare dev/ino when possible, it catches symbolic
3225 * links. Also compare file names, the inode may change
3226 * when the file was edited. */
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00003227 ((stat_ok && si->sn_dev_valid)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003228 && (si->sn_dev == st.st_dev
3229 && si->sn_ino == st.st_ino)) ||
Bram Moolenaar071d4272004-06-13 20:20:40 +00003230# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00003231 fnamecmp(si->sn_name, fname_exp) == 0))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003232 break;
Bram Moolenaar05159a02005-02-26 23:04:13 +00003233 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003234 if (current_SID == 0)
3235 {
3236 current_SID = ++last_current_SID;
Bram Moolenaar05159a02005-02-26 23:04:13 +00003237 if (ga_grow(&script_items, (int)(current_SID - script_items.ga_len))
3238 == FAIL)
3239 goto almosttheend;
3240 while (script_items.ga_len < current_SID)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003241 {
Bram Moolenaar05159a02005-02-26 23:04:13 +00003242 ++script_items.ga_len;
3243 SCRIPT_ITEM(script_items.ga_len).sn_name = NULL;
3244# ifdef FEAT_PROFILE
3245 SCRIPT_ITEM(script_items.ga_len).sn_prof_on = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003246# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003247 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00003248 si = &SCRIPT_ITEM(current_SID);
3249 si->sn_name = fname_exp;
3250 fname_exp = NULL;
3251# ifdef UNIX
3252 if (stat_ok)
3253 {
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00003254 si->sn_dev_valid = TRUE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00003255 si->sn_dev = st.st_dev;
3256 si->sn_ino = st.st_ino;
3257 }
3258 else
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00003259 si->sn_dev_valid = FALSE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00003260# endif
3261
Bram Moolenaar071d4272004-06-13 20:20:40 +00003262 /* Allocate the local script variables to use for this script. */
3263 new_script_vars(current_SID);
3264 }
3265
Bram Moolenaar05159a02005-02-26 23:04:13 +00003266# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00003267 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003268 {
3269 int forceit;
3270
3271 /* Check if we do profiling for this script. */
3272 if (!si->sn_prof_on && has_profiling(TRUE, si->sn_name, &forceit))
3273 {
3274 script_do_profile(si);
3275 si->sn_pr_force = forceit;
3276 }
3277 if (si->sn_prof_on)
3278 {
3279 ++si->sn_pr_count;
3280 profile_start(&si->sn_pr_start);
3281 profile_zero(&si->sn_pr_children);
3282 }
3283 }
3284# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003285#endif
3286
3287 /*
3288 * Call do_cmdline, which will call getsourceline() to get the lines.
3289 */
Bram Moolenaar73881402009-02-04 16:50:47 +00003290 do_cmdline(firstline, getsourceline, (void *)&cookie,
Bram Moolenaar071d4272004-06-13 20:20:40 +00003291 DOCMD_VERBOSE|DOCMD_NOWAIT|DOCMD_REPEAT);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003292 retval = OK;
Bram Moolenaar05159a02005-02-26 23:04:13 +00003293
3294#ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00003295 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003296 {
3297 /* Get "si" again, "script_items" may have been reallocated. */
3298 si = &SCRIPT_ITEM(current_SID);
3299 if (si->sn_prof_on)
3300 {
3301 profile_end(&si->sn_pr_start);
3302 profile_sub_wait(&wait_start, &si->sn_pr_start);
3303 profile_add(&si->sn_pr_total, &si->sn_pr_start);
Bram Moolenaar1056d982006-03-09 22:37:52 +00003304 profile_self(&si->sn_pr_self, &si->sn_pr_start,
3305 &si->sn_pr_children);
Bram Moolenaar05159a02005-02-26 23:04:13 +00003306 }
3307 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003308#endif
3309
3310 if (got_int)
3311 EMSG(_(e_interr));
3312 sourcing_name = save_sourcing_name;
3313 sourcing_lnum = save_sourcing_lnum;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003314 if (p_verbose > 1)
3315 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003316 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00003317 smsg((char_u *)_("finished sourcing %s"), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003318 if (sourcing_name != NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00003319 smsg((char_u *)_("continuing in %s"), sourcing_name);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003320 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003321 }
3322#ifdef STARTUPTIME
Bram Moolenaarc4e41982010-01-19 16:31:47 +01003323 if (time_fd != NULL)
3324 {
3325 vim_snprintf((char *)IObuff, IOSIZE, "sourcing %s", fname);
3326 time_msg((char *)IObuff, &tv_start);
3327 time_pop(&tv_rel);
3328 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003329#endif
3330
3331#ifdef FEAT_EVAL
3332 /*
3333 * After a "finish" in debug mode, need to break at first command of next
3334 * sourced file.
3335 */
3336 if (save_debug_break_level > ex_nesting_level
3337 && debug_break_level == ex_nesting_level)
3338 ++debug_break_level;
3339#endif
3340
Bram Moolenaar05159a02005-02-26 23:04:13 +00003341#ifdef FEAT_EVAL
3342almosttheend:
3343 current_SID = save_current_SID;
3344 restore_funccal(save_funccalp);
3345# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00003346 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003347 prof_child_exit(&wait_start); /* leaving a child now */
3348# endif
3349#endif
3350 fclose(cookie.fp);
3351 vim_free(cookie.nextline);
Bram Moolenaar73881402009-02-04 16:50:47 +00003352 vim_free(firstline);
Bram Moolenaar05159a02005-02-26 23:04:13 +00003353#ifdef FEAT_MBYTE
3354 convert_setup(&cookie.conv, NULL, NULL);
3355#endif
3356
Bram Moolenaar071d4272004-06-13 20:20:40 +00003357theend:
3358 vim_free(fname_exp);
3359 return retval;
3360}
3361
3362#if defined(FEAT_EVAL) || defined(PROTO)
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00003363
Bram Moolenaar071d4272004-06-13 20:20:40 +00003364/*
3365 * ":scriptnames"
3366 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003367 void
3368ex_scriptnames(eap)
Bram Moolenaar0c094b92009-05-14 20:20:33 +00003369 exarg_T *eap UNUSED;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003370{
3371 int i;
3372
Bram Moolenaar05159a02005-02-26 23:04:13 +00003373 for (i = 1; i <= script_items.ga_len && !got_int; ++i)
3374 if (SCRIPT_ITEM(i).sn_name != NULL)
Bram Moolenaard58ea072011-06-26 04:25:30 +02003375 {
3376 home_replace(NULL, SCRIPT_ITEM(i).sn_name,
3377 NameBuff, MAXPATHL, TRUE);
3378 smsg((char_u *)"%3d: %s", i, NameBuff);
Bram Moolenaar970a1b82012-03-23 18:39:18 +01003379 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003380}
3381
3382# if defined(BACKSLASH_IN_FILENAME) || defined(PROTO)
3383/*
3384 * Fix slashes in the list of script names for 'shellslash'.
3385 */
3386 void
3387scriptnames_slash_adjust()
3388{
3389 int i;
3390
Bram Moolenaar05159a02005-02-26 23:04:13 +00003391 for (i = 1; i <= script_items.ga_len; ++i)
3392 if (SCRIPT_ITEM(i).sn_name != NULL)
3393 slash_adjust(SCRIPT_ITEM(i).sn_name);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003394}
3395# endif
3396
3397/*
3398 * Get a pointer to a script name. Used for ":verbose set".
3399 */
3400 char_u *
3401get_scriptname(id)
3402 scid_T id;
3403{
3404 if (id == SID_MODELINE)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00003405 return (char_u *)_("modeline");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003406 if (id == SID_CMDARG)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00003407 return (char_u *)_("--cmd argument");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003408 if (id == SID_CARG)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00003409 return (char_u *)_("-c argument");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003410 if (id == SID_ENV)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00003411 return (char_u *)_("environment variable");
3412 if (id == SID_ERROR)
3413 return (char_u *)_("error handler");
Bram Moolenaar05159a02005-02-26 23:04:13 +00003414 return SCRIPT_ITEM(id).sn_name;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003415}
Bram Moolenaar05159a02005-02-26 23:04:13 +00003416
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00003417# if defined(EXITFREE) || defined(PROTO)
3418 void
3419free_scriptnames()
3420{
3421 int i;
3422
3423 for (i = script_items.ga_len; i > 0; --i)
3424 vim_free(SCRIPT_ITEM(i).sn_name);
3425 ga_clear(&script_items);
3426}
3427# endif
3428
Bram Moolenaar071d4272004-06-13 20:20:40 +00003429#endif
3430
3431#if defined(USE_CR) || defined(PROTO)
3432
3433# if defined(__MSL__) && (__MSL__ >= 22)
3434/*
3435 * Newer version of the Metrowerks library handle DOS and UNIX files
3436 * without help.
3437 * Test with earlier versions, MSL 2.2 is the library supplied with
3438 * Codewarrior Pro 2.
3439 */
3440 char *
3441fgets_cr(s, n, stream)
3442 char *s;
3443 int n;
3444 FILE *stream;
3445{
3446 return fgets(s, n, stream);
3447}
3448# else
3449/*
3450 * Version of fgets() which also works for lines ending in a <CR> only
3451 * (Macintosh format).
3452 * For older versions of the Metrowerks library.
3453 * At least CodeWarrior 9 needed this code.
3454 */
3455 char *
3456fgets_cr(s, n, stream)
3457 char *s;
3458 int n;
3459 FILE *stream;
3460{
3461 int c = 0;
3462 int char_read = 0;
3463
3464 while (!feof(stream) && c != '\r' && c != '\n' && char_read < n - 1)
3465 {
3466 c = fgetc(stream);
3467 s[char_read++] = c;
3468 /* If the file is in DOS format, we need to skip a NL after a CR. I
3469 * thought it was the other way around, but this appears to work... */
3470 if (c == '\n')
3471 {
3472 c = fgetc(stream);
3473 if (c != '\r')
3474 ungetc(c, stream);
3475 }
3476 }
3477
3478 s[char_read] = 0;
3479 if (char_read == 0)
3480 return NULL;
3481
3482 if (feof(stream) && char_read == 1)
3483 return NULL;
3484
3485 return s;
3486}
3487# endif
3488#endif
3489
3490/*
3491 * Get one full line from a sourced file.
3492 * Called by do_cmdline() when it's called from do_source().
3493 *
3494 * Return a pointer to the line in allocated memory.
3495 * Return NULL for end-of-file or some error.
3496 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003497 char_u *
3498getsourceline(c, cookie, indent)
Bram Moolenaar0c094b92009-05-14 20:20:33 +00003499 int c UNUSED;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003500 void *cookie;
Bram Moolenaar0c094b92009-05-14 20:20:33 +00003501 int indent UNUSED;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003502{
3503 struct source_cookie *sp = (struct source_cookie *)cookie;
3504 char_u *line;
Bram Moolenaarc047b9a2012-02-11 20:40:55 +01003505 char_u *p;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003506
3507#ifdef FEAT_EVAL
3508 /* If breakpoints have been added/deleted need to check for it. */
3509 if (sp->dbg_tick < debug_tick)
3510 {
3511 sp->breakpoint = dbg_find_breakpoint(TRUE, sp->fname, sourcing_lnum);
3512 sp->dbg_tick = debug_tick;
3513 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00003514# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00003515 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003516 script_line_end();
3517# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003518#endif
3519 /*
3520 * Get current line. If there is a read-ahead line, use it, otherwise get
3521 * one now.
3522 */
3523 if (sp->finished)
3524 line = NULL;
3525 else if (sp->nextline == NULL)
3526 line = get_one_sourceline(sp);
3527 else
3528 {
3529 line = sp->nextline;
3530 sp->nextline = NULL;
3531 ++sourcing_lnum;
3532 }
Bram Moolenaare2cc9702005-03-15 22:43:58 +00003533#ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00003534 if (line != NULL && do_profiling == PROF_YES)
Bram Moolenaare2cc9702005-03-15 22:43:58 +00003535 script_line_start();
3536#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003537
3538 /* Only concatenate lines starting with a \ when 'cpoptions' doesn't
3539 * contain the 'C' flag. */
3540 if (line != NULL && (vim_strchr(p_cpo, CPO_CONCAT) == NULL))
3541 {
3542 /* compensate for the one line read-ahead */
3543 --sourcing_lnum;
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01003544
3545 /* Get the next line and concatenate it when it starts with a
3546 * backslash. We always need to read the next line, keep it in
3547 * sp->nextline. */
3548 sp->nextline = get_one_sourceline(sp);
3549 if (sp->nextline != NULL && *(p = skipwhite(sp->nextline)) == '\\')
Bram Moolenaar071d4272004-06-13 20:20:40 +00003550 {
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01003551 garray_T ga;
3552
Bram Moolenaarb549a732012-02-22 18:29:33 +01003553 ga_init2(&ga, (int)sizeof(char_u), 400);
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01003554 ga_concat(&ga, line);
3555 ga_concat(&ga, p + 1);
3556 for (;;)
3557 {
3558 vim_free(sp->nextline);
3559 sp->nextline = get_one_sourceline(sp);
3560 if (sp->nextline == NULL)
3561 break;
3562 p = skipwhite(sp->nextline);
3563 if (*p != '\\')
3564 break;
Bram Moolenaarb549a732012-02-22 18:29:33 +01003565 /* Adjust the growsize to the current length to speed up
3566 * concatenating many lines. */
3567 if (ga.ga_len > 400)
3568 {
3569 if (ga.ga_len > 8000)
3570 ga.ga_growsize = 8000;
3571 else
3572 ga.ga_growsize = ga.ga_len;
3573 }
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01003574 ga_concat(&ga, p + 1);
3575 }
3576 ga_append(&ga, NUL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003577 vim_free(line);
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01003578 line = ga.ga_data;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003579 }
3580 }
3581
3582#ifdef FEAT_MBYTE
3583 if (line != NULL && sp->conv.vc_type != CONV_NONE)
3584 {
Bram Moolenaarc047b9a2012-02-11 20:40:55 +01003585 char_u *s;
3586
Bram Moolenaar071d4272004-06-13 20:20:40 +00003587 /* Convert the encoding of the script line. */
3588 s = string_convert(&sp->conv, line, NULL);
3589 if (s != NULL)
3590 {
3591 vim_free(line);
3592 line = s;
3593 }
3594 }
3595#endif
3596
3597#ifdef FEAT_EVAL
3598 /* Did we encounter a breakpoint? */
3599 if (sp->breakpoint != 0 && sp->breakpoint <= sourcing_lnum)
3600 {
3601 dbg_breakpoint(sp->fname, sourcing_lnum);
3602 /* Find next breakpoint. */
3603 sp->breakpoint = dbg_find_breakpoint(TRUE, sp->fname, sourcing_lnum);
3604 sp->dbg_tick = debug_tick;
3605 }
3606#endif
3607
3608 return line;
3609}
3610
3611 static char_u *
3612get_one_sourceline(sp)
3613 struct source_cookie *sp;
3614{
3615 garray_T ga;
3616 int len;
3617 int c;
3618 char_u *buf;
3619#ifdef USE_CRNL
3620 int has_cr; /* CR-LF found */
3621#endif
3622#ifdef USE_CR
3623 char_u *scan;
3624#endif
3625 int have_read = FALSE;
3626
3627 /* use a growarray to store the sourced line */
Bram Moolenaar6ac54292005-02-02 23:07:25 +00003628 ga_init2(&ga, 1, 250);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003629
3630 /*
3631 * Loop until there is a finished line (or end-of-file).
3632 */
3633 sourcing_lnum++;
3634 for (;;)
3635 {
Bram Moolenaar6ac54292005-02-02 23:07:25 +00003636 /* make room to read at least 120 (more) characters */
3637 if (ga_grow(&ga, 120) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003638 break;
3639 buf = (char_u *)ga.ga_data;
3640
3641#ifdef USE_CR
3642 if (sp->fileformat == EOL_MAC)
3643 {
Bram Moolenaar86b68352004-12-27 21:59:20 +00003644 if (fgets_cr((char *)buf + ga.ga_len, ga.ga_maxlen - ga.ga_len,
3645 sp->fp) == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003646 break;
3647 }
3648 else
3649#endif
Bram Moolenaar86b68352004-12-27 21:59:20 +00003650 if (fgets((char *)buf + ga.ga_len, ga.ga_maxlen - ga.ga_len,
3651 sp->fp) == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003652 break;
Bram Moolenaar6ac54292005-02-02 23:07:25 +00003653 len = ga.ga_len + (int)STRLEN(buf + ga.ga_len);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003654#ifdef USE_CRNL
3655 /* Ignore a trailing CTRL-Z, when in Dos mode. Only recognize the
3656 * CTRL-Z by its own, or after a NL. */
3657 if ( (len == 1 || (len >= 2 && buf[len - 2] == '\n'))
3658 && sp->fileformat == EOL_DOS
3659 && buf[len - 1] == Ctrl_Z)
3660 {
3661 buf[len - 1] = NUL;
3662 break;
3663 }
3664#endif
3665
3666#ifdef USE_CR
3667 /* If the read doesn't stop on a new line, and there's
3668 * some CR then we assume a Mac format */
3669 if (sp->fileformat == EOL_UNKNOWN)
3670 {
3671 if (buf[len - 1] != '\n' && vim_strchr(buf, '\r') != NULL)
3672 sp->fileformat = EOL_MAC;
3673 else
3674 sp->fileformat = EOL_UNIX;
3675 }
3676
3677 if (sp->fileformat == EOL_MAC)
3678 {
3679 scan = vim_strchr(buf, '\r');
3680
3681 if (scan != NULL)
3682 {
3683 *scan = '\n';
3684 if (*(scan + 1) != 0)
3685 {
3686 *(scan + 1) = 0;
3687 fseek(sp->fp, (long)(scan - buf - len + 1), SEEK_CUR);
3688 }
3689 }
3690 len = STRLEN(buf);
3691 }
3692#endif
3693
3694 have_read = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003695 ga.ga_len = len;
3696
3697 /* If the line was longer than the buffer, read more. */
Bram Moolenaar86b68352004-12-27 21:59:20 +00003698 if (ga.ga_maxlen - ga.ga_len == 1 && buf[len - 1] != '\n')
Bram Moolenaar071d4272004-06-13 20:20:40 +00003699 continue;
3700
3701 if (len >= 1 && buf[len - 1] == '\n') /* remove trailing NL */
3702 {
3703#ifdef USE_CRNL
3704 has_cr = (len >= 2 && buf[len - 2] == '\r');
3705 if (sp->fileformat == EOL_UNKNOWN)
3706 {
3707 if (has_cr)
3708 sp->fileformat = EOL_DOS;
3709 else
3710 sp->fileformat = EOL_UNIX;
3711 }
3712
3713 if (sp->fileformat == EOL_DOS)
3714 {
3715 if (has_cr) /* replace trailing CR */
3716 {
3717 buf[len - 2] = '\n';
3718 --len;
3719 --ga.ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003720 }
3721 else /* lines like ":map xx yy^M" will have failed */
3722 {
3723 if (!sp->error)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00003724 {
3725 msg_source(hl_attr(HLF_W));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003726 EMSG(_("W15: Warning: Wrong line separator, ^M may be missing"));
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00003727 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003728 sp->error = TRUE;
3729 sp->fileformat = EOL_UNIX;
3730 }
3731 }
3732#endif
3733 /* The '\n' is escaped if there is an odd number of ^V's just
3734 * before it, first set "c" just before the 'V's and then check
3735 * len&c parities (is faster than ((len-c)%2 == 0)) -- Acevedo */
3736 for (c = len - 2; c >= 0 && buf[c] == Ctrl_V; c--)
3737 ;
3738 if ((len & 1) != (c & 1)) /* escaped NL, read more */
3739 {
3740 sourcing_lnum++;
3741 continue;
3742 }
3743
3744 buf[len - 1] = NUL; /* remove the NL */
3745 }
3746
3747 /*
3748 * Check for ^C here now and then, so recursive :so can be broken.
3749 */
3750 line_breakcheck();
3751 break;
3752 }
3753
3754 if (have_read)
3755 return (char_u *)ga.ga_data;
3756
3757 vim_free(ga.ga_data);
3758 return NULL;
3759}
3760
Bram Moolenaar05159a02005-02-26 23:04:13 +00003761#if defined(FEAT_PROFILE) || defined(PROTO)
3762/*
3763 * Called when starting to read a script line.
3764 * "sourcing_lnum" must be correct!
3765 * When skipping lines it may not actually be executed, but we won't find out
3766 * until later and we need to store the time now.
3767 */
3768 void
3769script_line_start()
3770{
3771 scriptitem_T *si;
3772 sn_prl_T *pp;
3773
3774 if (current_SID <= 0 || current_SID > script_items.ga_len)
3775 return;
3776 si = &SCRIPT_ITEM(current_SID);
3777 if (si->sn_prof_on && sourcing_lnum >= 1)
3778 {
Bram Moolenaar8c8de832008-06-24 22:58:06 +00003779 /* Grow the array before starting the timer, so that the time spent
Bram Moolenaar05159a02005-02-26 23:04:13 +00003780 * here isn't counted. */
3781 ga_grow(&si->sn_prl_ga, (int)(sourcing_lnum - si->sn_prl_ga.ga_len));
3782 si->sn_prl_idx = sourcing_lnum - 1;
3783 while (si->sn_prl_ga.ga_len <= si->sn_prl_idx
3784 && si->sn_prl_ga.ga_len < si->sn_prl_ga.ga_maxlen)
3785 {
3786 /* Zero counters for a line that was not used before. */
3787 pp = &PRL_ITEM(si, si->sn_prl_ga.ga_len);
3788 pp->snp_count = 0;
3789 profile_zero(&pp->sn_prl_total);
3790 profile_zero(&pp->sn_prl_self);
3791 ++si->sn_prl_ga.ga_len;
3792 }
3793 si->sn_prl_execed = FALSE;
3794 profile_start(&si->sn_prl_start);
3795 profile_zero(&si->sn_prl_children);
3796 profile_get_wait(&si->sn_prl_wait);
3797 }
3798}
3799
3800/*
3801 * Called when actually executing a function line.
3802 */
3803 void
3804script_line_exec()
3805{
3806 scriptitem_T *si;
3807
3808 if (current_SID <= 0 || current_SID > script_items.ga_len)
3809 return;
3810 si = &SCRIPT_ITEM(current_SID);
3811 if (si->sn_prof_on && si->sn_prl_idx >= 0)
3812 si->sn_prl_execed = TRUE;
3813}
3814
3815/*
3816 * Called when done with a function line.
3817 */
3818 void
3819script_line_end()
3820{
3821 scriptitem_T *si;
3822 sn_prl_T *pp;
3823
3824 if (current_SID <= 0 || current_SID > script_items.ga_len)
3825 return;
3826 si = &SCRIPT_ITEM(current_SID);
3827 if (si->sn_prof_on && si->sn_prl_idx >= 0
3828 && si->sn_prl_idx < si->sn_prl_ga.ga_len)
3829 {
3830 if (si->sn_prl_execed)
3831 {
3832 pp = &PRL_ITEM(si, si->sn_prl_idx);
3833 ++pp->snp_count;
3834 profile_end(&si->sn_prl_start);
3835 profile_sub_wait(&si->sn_prl_wait, &si->sn_prl_start);
Bram Moolenaar05159a02005-02-26 23:04:13 +00003836 profile_add(&pp->sn_prl_total, &si->sn_prl_start);
Bram Moolenaar1056d982006-03-09 22:37:52 +00003837 profile_self(&pp->sn_prl_self, &si->sn_prl_start,
3838 &si->sn_prl_children);
Bram Moolenaar05159a02005-02-26 23:04:13 +00003839 }
3840 si->sn_prl_idx = -1;
3841 }
3842}
3843#endif
3844
Bram Moolenaar071d4272004-06-13 20:20:40 +00003845/*
3846 * ":scriptencoding": Set encoding conversion for a sourced script.
3847 * Without the multi-byte feature it's simply ignored.
3848 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003849 void
3850ex_scriptencoding(eap)
Bram Moolenaar0c094b92009-05-14 20:20:33 +00003851 exarg_T *eap UNUSED;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003852{
3853#ifdef FEAT_MBYTE
3854 struct source_cookie *sp;
3855 char_u *name;
3856
3857 if (!getline_equal(eap->getline, eap->cookie, getsourceline))
3858 {
3859 EMSG(_("E167: :scriptencoding used outside of a sourced file"));
3860 return;
3861 }
3862
3863 if (*eap->arg != NUL)
3864 {
3865 name = enc_canonize(eap->arg);
3866 if (name == NULL) /* out of memory */
3867 return;
3868 }
3869 else
3870 name = eap->arg;
3871
3872 /* Setup for conversion from the specified encoding to 'encoding'. */
3873 sp = (struct source_cookie *)getline_cookie(eap->getline, eap->cookie);
3874 convert_setup(&sp->conv, name, p_enc);
3875
3876 if (name != eap->arg)
3877 vim_free(name);
3878#endif
3879}
3880
3881#if defined(FEAT_EVAL) || defined(PROTO)
3882/*
3883 * ":finish": Mark a sourced file as finished.
3884 */
3885 void
3886ex_finish(eap)
3887 exarg_T *eap;
3888{
3889 if (getline_equal(eap->getline, eap->cookie, getsourceline))
3890 do_finish(eap, FALSE);
3891 else
3892 EMSG(_("E168: :finish used outside of a sourced file"));
3893}
3894
3895/*
3896 * Mark a sourced file as finished. Possibly makes the ":finish" pending.
3897 * Also called for a pending finish at the ":endtry" or after returning from
3898 * an extra do_cmdline(). "reanimate" is used in the latter case.
3899 */
3900 void
3901do_finish(eap, reanimate)
3902 exarg_T *eap;
3903 int reanimate;
3904{
3905 int idx;
3906
3907 if (reanimate)
3908 ((struct source_cookie *)getline_cookie(eap->getline,
3909 eap->cookie))->finished = FALSE;
3910
3911 /*
3912 * Cleanup (and inactivate) conditionals, but stop when a try conditional
3913 * not in its finally clause (which then is to be executed next) is found.
3914 * In this case, make the ":finish" pending for execution at the ":endtry".
3915 * Otherwise, finish normally.
3916 */
3917 idx = cleanup_conditionals(eap->cstack, 0, TRUE);
3918 if (idx >= 0)
3919 {
3920 eap->cstack->cs_pending[idx] = CSTP_FINISH;
3921 report_make_pending(CSTP_FINISH, NULL);
3922 }
3923 else
3924 ((struct source_cookie *)getline_cookie(eap->getline,
3925 eap->cookie))->finished = TRUE;
3926}
3927
3928
3929/*
3930 * Return TRUE when a sourced file had the ":finish" command: Don't give error
3931 * message for missing ":endif".
3932 * Return FALSE when not sourcing a file.
3933 */
3934 int
Bram Moolenaar89d40322006-08-29 15:30:07 +00003935source_finished(fgetline, cookie)
3936 char_u *(*fgetline) __ARGS((int, void *, int));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003937 void *cookie;
3938{
Bram Moolenaar89d40322006-08-29 15:30:07 +00003939 return (getline_equal(fgetline, cookie, getsourceline)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003940 && ((struct source_cookie *)getline_cookie(
Bram Moolenaar89d40322006-08-29 15:30:07 +00003941 fgetline, cookie))->finished);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003942}
3943#endif
3944
3945#if defined(FEAT_LISTCMDS) || defined(PROTO)
3946/*
3947 * ":checktime [buffer]"
3948 */
3949 void
3950ex_checktime(eap)
3951 exarg_T *eap;
3952{
3953 buf_T *buf;
3954 int save_no_check_timestamps = no_check_timestamps;
3955
3956 no_check_timestamps = 0;
3957 if (eap->addr_count == 0) /* default is all buffers */
3958 check_timestamps(FALSE);
3959 else
3960 {
3961 buf = buflist_findnr((int)eap->line2);
3962 if (buf != NULL) /* cannot happen? */
3963 (void)buf_check_timestamp(buf, FALSE);
3964 }
3965 no_check_timestamps = save_no_check_timestamps;
3966}
3967#endif
3968
Bram Moolenaar071d4272004-06-13 20:20:40 +00003969#if (defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
3970 && (defined(FEAT_EVAL) || defined(FEAT_MULTI_LANG))
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02003971# define HAVE_GET_LOCALE_VAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00003972static char *get_locale_val __ARGS((int what));
3973
3974 static char *
3975get_locale_val(what)
3976 int what;
3977{
3978 char *loc;
3979
3980 /* Obtain the locale value from the libraries. For DJGPP this is
3981 * redefined and it doesn't use the arguments. */
3982 loc = setlocale(what, NULL);
3983
Bram Moolenaar61660ea2006-04-07 21:40:07 +00003984# ifdef WIN32
Bram Moolenaar071d4272004-06-13 20:20:40 +00003985 if (loc != NULL)
3986 {
3987 char_u *p;
3988
Bram Moolenaar61660ea2006-04-07 21:40:07 +00003989 /* setocale() returns something like "LC_COLLATE=<name>;LC_..." when
3990 * one of the values (e.g., LC_CTYPE) differs. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003991 p = vim_strchr(loc, '=');
3992 if (p != NULL)
3993 {
3994 loc = ++p;
3995 while (*p != NUL) /* remove trailing newline */
3996 {
Bram Moolenaar61660ea2006-04-07 21:40:07 +00003997 if (*p < ' ' || *p == ';')
Bram Moolenaar071d4272004-06-13 20:20:40 +00003998 {
3999 *p = NUL;
4000 break;
4001 }
4002 ++p;
4003 }
4004 }
4005 }
4006# endif
4007
4008 return loc;
4009}
4010#endif
4011
4012
4013#ifdef WIN32
4014/*
4015 * On MS-Windows locale names are strings like "German_Germany.1252", but
4016 * gettext expects "de". Try to translate one into another here for a few
4017 * supported languages.
4018 */
4019 static char_u *
4020gettext_lang(char_u *name)
4021{
4022 int i;
4023 static char *(mtable[]) = {
4024 "afrikaans", "af",
4025 "czech", "cs",
4026 "dutch", "nl",
4027 "german", "de",
4028 "english_united kingdom", "en_GB",
4029 "spanish", "es",
4030 "french", "fr",
4031 "italian", "it",
4032 "japanese", "ja",
4033 "korean", "ko",
4034 "norwegian", "no",
4035 "polish", "pl",
4036 "russian", "ru",
4037 "slovak", "sk",
4038 "swedish", "sv",
4039 "ukrainian", "uk",
4040 "chinese_china", "zh_CN",
4041 "chinese_taiwan", "zh_TW",
4042 NULL};
4043
4044 for (i = 0; mtable[i] != NULL; i += 2)
4045 if (STRNICMP(mtable[i], name, STRLEN(mtable[i])) == 0)
4046 return mtable[i + 1];
4047 return name;
4048}
4049#endif
4050
4051#if defined(FEAT_MULTI_LANG) || defined(PROTO)
4052/*
4053 * Obtain the current messages language. Used to set the default for
4054 * 'helplang'. May return NULL or an empty string.
4055 */
4056 char_u *
4057get_mess_lang()
4058{
4059 char_u *p;
4060
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004061# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00004062# if defined(LC_MESSAGES)
4063 p = (char_u *)get_locale_val(LC_MESSAGES);
4064# else
4065 /* This is necessary for Win32, where LC_MESSAGES is not defined and $LANG
Bram Moolenaar61660ea2006-04-07 21:40:07 +00004066 * may be set to the LCID number. LC_COLLATE is the best guess, LC_TIME
4067 * and LC_MONETARY may be set differently for a Japanese working in the
4068 * US. */
4069 p = (char_u *)get_locale_val(LC_COLLATE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004070# endif
4071# else
4072 p = mch_getenv((char_u *)"LC_ALL");
4073 if (p == NULL || *p == NUL)
4074 {
4075 p = mch_getenv((char_u *)"LC_MESSAGES");
4076 if (p == NULL || *p == NUL)
4077 p = mch_getenv((char_u *)"LANG");
4078 }
4079# endif
4080# ifdef WIN32
4081 p = gettext_lang(p);
4082# endif
4083 return p;
4084}
4085#endif
4086
Bram Moolenaardef9e822004-12-31 20:58:58 +00004087/* Complicated #if; matches with where get_mess_env() is used below. */
4088#if (defined(FEAT_EVAL) && !((defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
4089 && defined(LC_MESSAGES))) \
4090 || ((defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
4091 && (defined(FEAT_GETTEXT) || defined(FEAT_MBYTE)) \
4092 && !defined(LC_MESSAGES))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004093static char_u *get_mess_env __ARGS((void));
4094
4095/*
4096 * Get the language used for messages from the environment.
4097 */
4098 static char_u *
4099get_mess_env()
4100{
4101 char_u *p;
4102
4103 p = mch_getenv((char_u *)"LC_ALL");
4104 if (p == NULL || *p == NUL)
4105 {
4106 p = mch_getenv((char_u *)"LC_MESSAGES");
4107 if (p == NULL || *p == NUL)
4108 {
4109 p = mch_getenv((char_u *)"LANG");
4110 if (p != NULL && VIM_ISDIGIT(*p))
4111 p = NULL; /* ignore something like "1043" */
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004112# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00004113 if (p == NULL || *p == NUL)
4114 p = (char_u *)get_locale_val(LC_CTYPE);
4115# endif
4116 }
4117 }
4118 return p;
4119}
4120#endif
4121
4122#if defined(FEAT_EVAL) || defined(PROTO)
4123
4124/*
4125 * Set the "v:lang" variable according to the current locale setting.
4126 * Also do "v:lc_time"and "v:ctype".
4127 */
4128 void
4129set_lang_var()
4130{
4131 char_u *loc;
4132
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004133# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00004134 loc = (char_u *)get_locale_val(LC_CTYPE);
4135# else
4136 /* setlocale() not supported: use the default value */
4137 loc = (char_u *)"C";
4138# endif
4139 set_vim_var_string(VV_CTYPE, loc, -1);
4140
4141 /* When LC_MESSAGES isn't defined use the value from $LC_MESSAGES, fall
4142 * back to LC_CTYPE if it's empty. */
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004143# if defined(HAVE_GET_LOCALE_VAL) && defined(LC_MESSAGES)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004144 loc = (char_u *)get_locale_val(LC_MESSAGES);
4145# else
4146 loc = get_mess_env();
4147# endif
4148 set_vim_var_string(VV_LANG, loc, -1);
4149
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004150# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00004151 loc = (char_u *)get_locale_val(LC_TIME);
4152# endif
4153 set_vim_var_string(VV_LC_TIME, loc, -1);
4154}
4155#endif
4156
4157#if (defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
4158 && (defined(FEAT_GETTEXT) || defined(FEAT_MBYTE))
4159/*
4160 * ":language": Set the language (locale).
4161 */
4162 void
4163ex_language(eap)
4164 exarg_T *eap;
4165{
4166 char *loc;
4167 char_u *p;
4168 char_u *name;
4169 int what = LC_ALL;
4170 char *whatstr = "";
4171#ifdef LC_MESSAGES
4172# define VIM_LC_MESSAGES LC_MESSAGES
4173#else
4174# define VIM_LC_MESSAGES 6789
4175#endif
4176
4177 name = eap->arg;
4178
4179 /* Check for "messages {name}", "ctype {name}" or "time {name}" argument.
4180 * Allow abbreviation, but require at least 3 characters to avoid
4181 * confusion with a two letter language name "me" or "ct". */
4182 p = skiptowhite(eap->arg);
4183 if ((*p == NUL || vim_iswhite(*p)) && p - eap->arg >= 3)
4184 {
4185 if (STRNICMP(eap->arg, "messages", p - eap->arg) == 0)
4186 {
4187 what = VIM_LC_MESSAGES;
4188 name = skipwhite(p);
4189 whatstr = "messages ";
4190 }
4191 else if (STRNICMP(eap->arg, "ctype", p - eap->arg) == 0)
4192 {
4193 what = LC_CTYPE;
4194 name = skipwhite(p);
4195 whatstr = "ctype ";
4196 }
4197 else if (STRNICMP(eap->arg, "time", p - eap->arg) == 0)
4198 {
4199 what = LC_TIME;
4200 name = skipwhite(p);
4201 whatstr = "time ";
4202 }
4203 }
4204
4205 if (*name == NUL)
4206 {
4207#ifndef LC_MESSAGES
4208 if (what == VIM_LC_MESSAGES)
4209 p = get_mess_env();
4210 else
4211#endif
4212 p = (char_u *)setlocale(what, NULL);
4213 if (p == NULL || *p == NUL)
4214 p = (char_u *)"Unknown";
4215 smsg((char_u *)_("Current %slanguage: \"%s\""), whatstr, p);
4216 }
4217 else
4218 {
4219#ifndef LC_MESSAGES
4220 if (what == VIM_LC_MESSAGES)
4221 loc = "";
4222 else
4223#endif
Bram Moolenaar8c8de832008-06-24 22:58:06 +00004224 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00004225 loc = setlocale(what, (char *)name);
Bram Moolenaar8c8de832008-06-24 22:58:06 +00004226#if defined(FEAT_FLOAT) && defined(LC_NUMERIC)
4227 /* Make sure strtod() uses a decimal point, not a comma. */
4228 setlocale(LC_NUMERIC, "C");
4229#endif
4230 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004231 if (loc == NULL)
4232 EMSG2(_("E197: Cannot set language to \"%s\""), name);
4233 else
4234 {
4235#ifdef HAVE_NL_MSG_CAT_CNTR
4236 /* Need to do this for GNU gettext, otherwise cached translations
4237 * will be used again. */
4238 extern int _nl_msg_cat_cntr;
4239
4240 ++_nl_msg_cat_cntr;
4241#endif
Bram Moolenaarb8017e72007-05-10 18:59:07 +00004242 /* Reset $LC_ALL, otherwise it would overrule everything. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004243 vim_setenv((char_u *)"LC_ALL", (char_u *)"");
4244
4245 if (what != LC_TIME)
4246 {
4247 /* Tell gettext() what to translate to. It apparently doesn't
4248 * use the currently effective locale. Also do this when
4249 * FEAT_GETTEXT isn't defined, so that shell commands use this
4250 * value. */
4251 if (what == LC_ALL)
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00004252 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00004253 vim_setenv((char_u *)"LANG", name);
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00004254# ifdef WIN32
4255 /* Apparently MS-Windows printf() may cause a crash when
4256 * we give it 8-bit text while it's expecting text in the
4257 * current locale. This call avoids that. */
4258 setlocale(LC_CTYPE, "C");
4259# endif
4260 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004261 if (what != LC_CTYPE)
4262 {
4263 char_u *mname;
4264#ifdef WIN32
4265 mname = gettext_lang(name);
4266#else
4267 mname = name;
4268#endif
4269 vim_setenv((char_u *)"LC_MESSAGES", mname);
4270#ifdef FEAT_MULTI_LANG
4271 set_helplang_default(mname);
4272#endif
4273 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004274 }
4275
4276# ifdef FEAT_EVAL
4277 /* Set v:lang, v:lc_time and v:ctype to the final result. */
4278 set_lang_var();
4279# endif
Bram Moolenaar6cc00c72011-10-20 21:41:09 +02004280# ifdef FEAT_TITLE
4281 maketitle();
4282# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004283 }
4284 }
4285}
4286
4287# if defined(FEAT_CMDL_COMPL) || defined(PROTO)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02004288
4289static char_u **locales = NULL; /* Array of all available locales */
4290static int did_init_locales = FALSE;
4291
4292static void init_locales __ARGS((void));
4293static char_u **find_locales __ARGS((void));
4294
4295/*
4296 * Lazy initialization of all available locales.
4297 */
4298 static void
4299init_locales()
4300{
4301 if (!did_init_locales)
4302 {
4303 did_init_locales = TRUE;
4304 locales = find_locales();
4305 }
4306}
4307
4308/* Return an array of strings for all available locales + NULL for the
4309 * last element. Return NULL in case of error. */
4310 static char_u **
4311find_locales()
4312{
4313 garray_T locales_ga;
4314 char_u *loc;
4315
4316 /* Find all available locales by running command "locale -a". If this
4317 * doesn't work we won't have completion. */
4318 char_u *locale_a = get_cmd_output((char_u *)"locale -a",
4319 NULL, SHELL_SILENT);
4320 if (locale_a == NULL)
4321 return NULL;
4322 ga_init2(&locales_ga, sizeof(char_u *), 20);
4323
4324 /* Transform locale_a string where each locale is separated by "\n"
4325 * into an array of locale strings. */
4326 loc = (char_u *)strtok((char *)locale_a, "\n");
4327
4328 while (loc != NULL)
4329 {
4330 if (ga_grow(&locales_ga, 1) == FAIL)
4331 break;
4332 loc = vim_strsave(loc);
4333 if (loc == NULL)
4334 break;
4335
4336 ((char_u **)locales_ga.ga_data)[locales_ga.ga_len++] = loc;
4337 loc = (char_u *)strtok(NULL, "\n");
4338 }
4339 vim_free(locale_a);
4340 if (ga_grow(&locales_ga, 1) == FAIL)
4341 {
4342 ga_clear(&locales_ga);
4343 return NULL;
4344 }
4345 ((char_u **)locales_ga.ga_data)[locales_ga.ga_len] = NULL;
4346 return (char_u **)locales_ga.ga_data;
4347}
4348
4349# if defined(EXITFREE) || defined(PROTO)
4350 void
4351free_locales()
4352{
4353 int i;
4354 if (locales != NULL)
4355 {
4356 for (i = 0; locales[i] != NULL; i++)
4357 vim_free(locales[i]);
4358 vim_free(locales);
4359 locales = NULL;
4360 }
4361}
4362# endif
4363
Bram Moolenaar071d4272004-06-13 20:20:40 +00004364/*
4365 * Function given to ExpandGeneric() to obtain the possible arguments of the
4366 * ":language" command.
4367 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004368 char_u *
4369get_lang_arg(xp, idx)
Bram Moolenaar0c094b92009-05-14 20:20:33 +00004370 expand_T *xp UNUSED;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004371 int idx;
4372{
4373 if (idx == 0)
4374 return (char_u *)"messages";
4375 if (idx == 1)
4376 return (char_u *)"ctype";
4377 if (idx == 2)
4378 return (char_u *)"time";
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02004379
4380 init_locales();
4381 if (locales == NULL)
4382 return NULL;
4383 return locales[idx - 3];
4384}
4385
4386/*
4387 * Function given to ExpandGeneric() to obtain the available locales.
4388 */
4389 char_u *
4390get_locales(xp, idx)
4391 expand_T *xp UNUSED;
4392 int idx;
4393{
4394 init_locales();
4395 if (locales == NULL)
4396 return NULL;
4397 return locales[idx];
Bram Moolenaar071d4272004-06-13 20:20:40 +00004398}
4399# endif
4400
4401#endif