blob: 5bc82425551bb80ddf10113c01d74da40f38b236 [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);
655 vim_free(DEBUGGY(gap, todel).dbg_prog);
656 --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
961# if defined(FEAT_PROFILE) || defined(PROTO)
962/*
963 * Functions for profiling.
964 */
965static void script_do_profile __ARGS((scriptitem_T *si));
966static void script_dump_profile __ARGS((FILE *fd));
967static proftime_T prof_wait_time;
968
Bram Moolenaar433f7c82006-03-21 21:29:36 +0000969/*
Bram Moolenaar05159a02005-02-26 23:04:13 +0000970 * Add the time "tm2" to "tm".
971 */
972 void
973profile_add(tm, tm2)
974 proftime_T *tm, *tm2;
975{
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000976# ifdef WIN3264
977 tm->QuadPart += tm2->QuadPart;
978# else
Bram Moolenaar05159a02005-02-26 23:04:13 +0000979 tm->tv_usec += tm2->tv_usec;
980 tm->tv_sec += tm2->tv_sec;
981 if (tm->tv_usec >= 1000000)
982 {
983 tm->tv_usec -= 1000000;
984 ++tm->tv_sec;
985 }
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000986# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +0000987}
988
989/*
Bram Moolenaar1056d982006-03-09 22:37:52 +0000990 * Add the "self" time from the total time and the children's time.
991 */
992 void
993profile_self(self, total, children)
994 proftime_T *self, *total, *children;
995{
996 /* Check that the result won't be negative. Can happen with recursive
997 * calls. */
998#ifdef WIN3264
999 if (total->QuadPart <= children->QuadPart)
1000 return;
1001#else
1002 if (total->tv_sec < children->tv_sec
1003 || (total->tv_sec == children->tv_sec
1004 && total->tv_usec <= children->tv_usec))
1005 return;
1006#endif
1007 profile_add(self, total);
1008 profile_sub(self, children);
1009}
1010
1011/*
Bram Moolenaar05159a02005-02-26 23:04:13 +00001012 * Get the current waittime.
1013 */
1014 void
1015profile_get_wait(tm)
1016 proftime_T *tm;
1017{
1018 *tm = prof_wait_time;
1019}
1020
1021/*
1022 * Subtract the passed waittime since "tm" from "tma".
1023 */
1024 void
1025profile_sub_wait(tm, tma)
1026 proftime_T *tm, *tma;
1027{
1028 proftime_T tm3 = prof_wait_time;
1029
1030 profile_sub(&tm3, tm);
1031 profile_sub(tma, &tm3);
1032}
1033
1034/*
1035 * Return TRUE if "tm1" and "tm2" are equal.
1036 */
1037 int
1038profile_equal(tm1, tm2)
1039 proftime_T *tm1, *tm2;
1040{
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001041# ifdef WIN3264
1042 return (tm1->QuadPart == tm2->QuadPart);
1043# else
Bram Moolenaar05159a02005-02-26 23:04:13 +00001044 return (tm1->tv_usec == tm2->tv_usec && tm1->tv_sec == tm2->tv_sec);
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001045# endif
1046}
1047
1048/*
1049 * Return <0, 0 or >0 if "tm1" < "tm2", "tm1" == "tm2" or "tm1" > "tm2"
1050 */
1051 int
1052profile_cmp(tm1, tm2)
1053 proftime_T *tm1, *tm2;
1054{
1055# ifdef WIN3264
1056 return (int)(tm2->QuadPart - tm1->QuadPart);
1057# else
1058 if (tm1->tv_sec == tm2->tv_sec)
1059 return tm2->tv_usec - tm1->tv_usec;
1060 return tm2->tv_sec - tm1->tv_sec;
1061# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00001062}
1063
Bram Moolenaar05159a02005-02-26 23:04:13 +00001064static char_u *profile_fname = NULL;
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001065static proftime_T pause_time;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001066
1067/*
1068 * ":profile cmd args"
1069 */
1070 void
1071ex_profile(eap)
1072 exarg_T *eap;
1073{
1074 char_u *e;
1075 int len;
1076
1077 e = skiptowhite(eap->arg);
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00001078 len = (int)(e - eap->arg);
Bram Moolenaar05159a02005-02-26 23:04:13 +00001079 e = skipwhite(e);
1080
1081 if (len == 5 && STRNCMP(eap->arg, "start", 5) == 0 && *e != NUL)
1082 {
1083 vim_free(profile_fname);
1084 profile_fname = vim_strsave(e);
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001085 do_profiling = PROF_YES;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001086 profile_zero(&prof_wait_time);
1087 set_vim_var_nr(VV_PROFILING, 1L);
1088 }
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001089 else if (do_profiling == PROF_NONE)
Bram Moolenaar54c1b492010-02-24 14:01:28 +01001090 EMSG(_("E750: First use \":profile start {fname}\""));
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001091 else if (STRCMP(eap->arg, "pause") == 0)
1092 {
1093 if (do_profiling == PROF_YES)
1094 profile_start(&pause_time);
1095 do_profiling = PROF_PAUSED;
1096 }
1097 else if (STRCMP(eap->arg, "continue") == 0)
1098 {
1099 if (do_profiling == PROF_PAUSED)
1100 {
1101 profile_end(&pause_time);
1102 profile_add(&prof_wait_time, &pause_time);
1103 }
1104 do_profiling = PROF_YES;
1105 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00001106 else
1107 {
1108 /* The rest is similar to ":breakadd". */
1109 ex_breakadd(eap);
1110 }
1111}
1112
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001113/* Command line expansion for :profile. */
1114static enum
1115{
1116 PEXP_SUBCMD, /* expand :profile sub-commands */
Bram Moolenaar49789dc2011-02-25 14:46:09 +01001117 PEXP_FUNC /* expand :profile func {funcname} */
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001118} pexpand_what;
1119
1120static char *pexpand_cmds[] = {
1121 "start",
1122#define PROFCMD_START 0
1123 "pause",
1124#define PROFCMD_PAUSE 1
1125 "continue",
1126#define PROFCMD_CONTINUE 2
1127 "func",
1128#define PROFCMD_FUNC 3
1129 "file",
1130#define PROFCMD_FILE 4
1131 NULL
1132#define PROFCMD_LAST 5
1133};
1134
1135/*
1136 * Function given to ExpandGeneric() to obtain the profile command
1137 * specific expansion.
1138 */
1139 char_u *
1140get_profile_name(xp, idx)
1141 expand_T *xp UNUSED;
1142 int idx;
1143{
1144 switch (pexpand_what)
1145 {
1146 case PEXP_SUBCMD:
1147 return (char_u *)pexpand_cmds[idx];
1148 /* case PEXP_FUNC: TODO */
1149 default:
1150 return NULL;
1151 }
1152}
1153
1154/*
1155 * Handle command line completion for :profile command.
1156 */
1157 void
1158set_context_in_profile_cmd(xp, arg)
1159 expand_T *xp;
1160 char_u *arg;
1161{
1162 char_u *end_subcmd;
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001163
1164 /* Default: expand subcommands. */
1165 xp->xp_context = EXPAND_PROFILE;
1166 pexpand_what = PEXP_SUBCMD;
1167 xp->xp_pattern = arg;
1168
1169 end_subcmd = skiptowhite(arg);
1170 if (*end_subcmd == NUL)
1171 return;
1172
Bram Moolenaar8b9c05f2010-03-02 17:54:33 +01001173 if (end_subcmd - arg == 5 && STRNCMP(arg, "start", 5) == 0)
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001174 {
1175 xp->xp_context = EXPAND_FILES;
1176 xp->xp_pattern = skipwhite(end_subcmd);
1177 return;
1178 }
1179
1180 /* TODO: expand function names after "func" */
1181 xp->xp_context = EXPAND_NOTHING;
1182}
1183
Bram Moolenaar05159a02005-02-26 23:04:13 +00001184/*
1185 * Dump the profiling info.
1186 */
1187 void
1188profile_dump()
1189{
1190 FILE *fd;
1191
1192 if (profile_fname != NULL)
1193 {
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +00001194 fd = mch_fopen((char *)profile_fname, "w");
Bram Moolenaar05159a02005-02-26 23:04:13 +00001195 if (fd == NULL)
1196 EMSG2(_(e_notopen), profile_fname);
1197 else
1198 {
Bram Moolenaar05159a02005-02-26 23:04:13 +00001199 script_dump_profile(fd);
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001200 func_dump_profile(fd);
Bram Moolenaar05159a02005-02-26 23:04:13 +00001201 fclose(fd);
1202 }
1203 }
1204}
1205
1206/*
1207 * Start profiling script "fp".
1208 */
1209 static void
1210script_do_profile(si)
1211 scriptitem_T *si;
1212{
1213 si->sn_pr_count = 0;
1214 profile_zero(&si->sn_pr_total);
1215 profile_zero(&si->sn_pr_self);
1216
1217 ga_init2(&si->sn_prl_ga, sizeof(sn_prl_T), 100);
1218 si->sn_prl_idx = -1;
1219 si->sn_prof_on = TRUE;
1220 si->sn_pr_nest = 0;
1221}
1222
1223/*
1224 * save time when starting to invoke another script or function.
1225 */
1226 void
1227script_prof_save(tm)
1228 proftime_T *tm; /* place to store wait time */
1229{
1230 scriptitem_T *si;
1231
1232 if (current_SID > 0 && current_SID <= script_items.ga_len)
1233 {
1234 si = &SCRIPT_ITEM(current_SID);
1235 if (si->sn_prof_on && si->sn_pr_nest++ == 0)
1236 profile_start(&si->sn_pr_child);
1237 }
1238 profile_get_wait(tm);
1239}
1240
1241/*
1242 * Count time spent in children after invoking another script or function.
1243 */
1244 void
1245script_prof_restore(tm)
1246 proftime_T *tm;
1247{
1248 scriptitem_T *si;
1249
1250 if (current_SID > 0 && current_SID <= script_items.ga_len)
1251 {
1252 si = &SCRIPT_ITEM(current_SID);
1253 if (si->sn_prof_on && --si->sn_pr_nest == 0)
1254 {
1255 profile_end(&si->sn_pr_child);
1256 profile_sub_wait(tm, &si->sn_pr_child); /* don't count wait time */
1257 profile_add(&si->sn_pr_children, &si->sn_pr_child);
1258 profile_add(&si->sn_prl_children, &si->sn_pr_child);
1259 }
1260 }
1261}
1262
1263static proftime_T inchar_time;
1264
1265/*
1266 * Called when starting to wait for the user to type a character.
1267 */
1268 void
1269prof_inchar_enter()
1270{
1271 profile_start(&inchar_time);
1272}
1273
1274/*
1275 * Called when finished waiting for the user to type a character.
1276 */
1277 void
1278prof_inchar_exit()
1279{
1280 profile_end(&inchar_time);
1281 profile_add(&prof_wait_time, &inchar_time);
1282}
1283
1284/*
1285 * Dump the profiling results for all scripts in file "fd".
1286 */
1287 static void
1288script_dump_profile(fd)
1289 FILE *fd;
1290{
1291 int id;
1292 scriptitem_T *si;
1293 int i;
1294 FILE *sfd;
1295 sn_prl_T *pp;
1296
1297 for (id = 1; id <= script_items.ga_len; ++id)
1298 {
1299 si = &SCRIPT_ITEM(id);
1300 if (si->sn_prof_on)
1301 {
1302 fprintf(fd, "SCRIPT %s\n", si->sn_name);
1303 if (si->sn_pr_count == 1)
1304 fprintf(fd, "Sourced 1 time\n");
1305 else
1306 fprintf(fd, "Sourced %d times\n", si->sn_pr_count);
1307 fprintf(fd, "Total time: %s\n", profile_msg(&si->sn_pr_total));
1308 fprintf(fd, " Self time: %s\n", profile_msg(&si->sn_pr_self));
1309 fprintf(fd, "\n");
1310 fprintf(fd, "count total (s) self (s)\n");
1311
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +00001312 sfd = mch_fopen((char *)si->sn_name, "r");
Bram Moolenaar05159a02005-02-26 23:04:13 +00001313 if (sfd == NULL)
1314 fprintf(fd, "Cannot open file!\n");
1315 else
1316 {
1317 for (i = 0; i < si->sn_prl_ga.ga_len; ++i)
1318 {
1319 if (vim_fgets(IObuff, IOSIZE, sfd))
1320 break;
1321 pp = &PRL_ITEM(si, i);
1322 if (pp->snp_count > 0)
1323 {
1324 fprintf(fd, "%5d ", pp->snp_count);
1325 if (profile_equal(&pp->sn_prl_total, &pp->sn_prl_self))
1326 fprintf(fd, " ");
1327 else
1328 fprintf(fd, "%s ", profile_msg(&pp->sn_prl_total));
1329 fprintf(fd, "%s ", profile_msg(&pp->sn_prl_self));
1330 }
1331 else
1332 fprintf(fd, " ");
1333 fprintf(fd, "%s", IObuff);
1334 }
1335 fclose(sfd);
1336 }
1337 fprintf(fd, "\n");
1338 }
1339 }
1340}
1341
1342/*
1343 * Return TRUE when a function defined in the current script should be
1344 * profiled.
1345 */
1346 int
1347prof_def_func()
1348{
Bram Moolenaar53180ce2005-07-05 21:48:14 +00001349 if (current_SID > 0)
1350 return SCRIPT_ITEM(current_SID).sn_pr_force;
1351 return FALSE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001352}
1353
1354# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001355#endif
1356
1357/*
1358 * If 'autowrite' option set, try to write the file.
1359 * Careful: autocommands may make "buf" invalid!
1360 *
1361 * return FAIL for failure, OK otherwise
1362 */
1363 int
1364autowrite(buf, forceit)
1365 buf_T *buf;
1366 int forceit;
1367{
Bram Moolenaar373154b2007-02-13 05:19:30 +00001368 int r;
1369
Bram Moolenaar071d4272004-06-13 20:20:40 +00001370 if (!(p_aw || p_awa) || !p_write
1371#ifdef FEAT_QUICKFIX
Bram Moolenaar373154b2007-02-13 05:19:30 +00001372 /* never autowrite a "nofile" or "nowrite" buffer */
1373 || bt_dontwrite(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001374#endif
Bram Moolenaar373154b2007-02-13 05:19:30 +00001375 || (!forceit && buf->b_p_ro) || buf->b_ffname == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001376 return FAIL;
Bram Moolenaar373154b2007-02-13 05:19:30 +00001377 r = buf_write_all(buf, forceit);
1378
1379 /* Writing may succeed but the buffer still changed, e.g., when there is a
1380 * conversion error. We do want to return FAIL then. */
1381 if (buf_valid(buf) && bufIsChanged(buf))
1382 r = FAIL;
1383 return r;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001384}
1385
1386/*
1387 * flush all buffers, except the ones that are readonly
1388 */
1389 void
1390autowrite_all()
1391{
1392 buf_T *buf;
1393
1394 if (!(p_aw || p_awa) || !p_write)
1395 return;
1396 for (buf = firstbuf; buf; buf = buf->b_next)
1397 if (bufIsChanged(buf) && !buf->b_p_ro)
1398 {
1399 (void)buf_write_all(buf, FALSE);
1400#ifdef FEAT_AUTOCMD
1401 /* an autocommand may have deleted the buffer */
1402 if (!buf_valid(buf))
1403 buf = firstbuf;
1404#endif
1405 }
1406}
1407
1408/*
1409 * return TRUE if buffer was changed and cannot be abandoned.
1410 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001411 int
1412check_changed(buf, checkaw, mult_win, forceit, allbuf)
1413 buf_T *buf;
1414 int checkaw; /* do autowrite if buffer was changed */
1415 int mult_win; /* check also when several wins for the buf */
1416 int forceit;
Bram Moolenaar0c094b92009-05-14 20:20:33 +00001417 int allbuf UNUSED; /* may write all buffers */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001418{
1419 if ( !forceit
1420 && bufIsChanged(buf)
1421 && (mult_win || buf->b_nwindows <= 1)
1422 && (!checkaw || autowrite(buf, forceit) == FAIL))
1423 {
1424#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG)
1425 if ((p_confirm || cmdmod.confirm) && p_write)
1426 {
1427 buf_T *buf2;
1428 int count = 0;
1429
1430 if (allbuf)
1431 for (buf2 = firstbuf; buf2 != NULL; buf2 = buf2->b_next)
1432 if (bufIsChanged(buf2)
1433 && (buf2->b_ffname != NULL
1434# ifdef FEAT_BROWSE
1435 || cmdmod.browse
1436# endif
1437 ))
1438 ++count;
1439# ifdef FEAT_AUTOCMD
1440 if (!buf_valid(buf))
1441 /* Autocommand deleted buffer, oops! It's not changed now. */
1442 return FALSE;
1443# endif
1444 dialog_changed(buf, count > 1);
1445# ifdef FEAT_AUTOCMD
1446 if (!buf_valid(buf))
1447 /* Autocommand deleted buffer, oops! It's not changed now. */
1448 return FALSE;
1449# endif
1450 return bufIsChanged(buf);
1451 }
1452#endif
1453 EMSG(_(e_nowrtmsg));
1454 return TRUE;
1455 }
1456 return FALSE;
1457}
1458
1459#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG) || defined(PROTO)
1460
1461#if defined(FEAT_BROWSE) || defined(PROTO)
1462/*
1463 * When wanting to write a file without a file name, ask the user for a name.
1464 */
1465 void
1466browse_save_fname(buf)
1467 buf_T *buf;
1468{
1469 if (buf->b_fname == NULL)
1470 {
1471 char_u *fname;
1472
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00001473 fname = do_browse(BROWSE_SAVE, (char_u *)_("Save As"),
1474 NULL, NULL, NULL, NULL, buf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001475 if (fname != NULL)
1476 {
1477 if (setfname(buf, fname, NULL, TRUE) == OK)
1478 buf->b_flags |= BF_NOTEDITED;
1479 vim_free(fname);
1480 }
1481 }
1482}
1483#endif
1484
1485/*
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02001486 * Ask the user what to do when abandoning a changed buffer.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001487 * Must check 'write' option first!
1488 */
1489 void
1490dialog_changed(buf, checkall)
1491 buf_T *buf;
1492 int checkall; /* may abandon all changed buffers */
1493{
Bram Moolenaard9462e32011-04-11 21:35:11 +02001494 char_u buff[DIALOG_MSG_SIZE];
Bram Moolenaar071d4272004-06-13 20:20:40 +00001495 int ret;
1496 buf_T *buf2;
Bram Moolenaar8218f602012-04-25 17:32:18 +02001497 exarg_T ea;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001498
Bram Moolenaar82cf9b62005-06-07 21:09:25 +00001499 dialog_msg(buff, _("Save changes to \"%s\"?"),
Bram Moolenaar071d4272004-06-13 20:20:40 +00001500 (buf->b_fname != NULL) ?
1501 buf->b_fname : (char_u *)_("Untitled"));
1502 if (checkall)
1503 ret = vim_dialog_yesnoallcancel(VIM_QUESTION, NULL, buff, 1);
1504 else
1505 ret = vim_dialog_yesnocancel(VIM_QUESTION, NULL, buff, 1);
1506
Bram Moolenaar8218f602012-04-25 17:32:18 +02001507 /* Init ea pseudo-structure, this is needed for the check_overwrite()
1508 * function. */
1509 ea.append = ea.forceit = FALSE;
1510
Bram Moolenaar071d4272004-06-13 20:20:40 +00001511 if (ret == VIM_YES)
1512 {
1513#ifdef FEAT_BROWSE
1514 /* May get file name, when there is none */
1515 browse_save_fname(buf);
1516#endif
Bram Moolenaar8218f602012-04-25 17:32:18 +02001517 if (buf->b_fname != NULL && check_overwrite(&ea, buf,
1518 buf->b_fname, buf->b_ffname, FALSE) == OK)
1519 /* didn't hit Cancel */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001520 (void)buf_write_all(buf, FALSE);
1521 }
1522 else if (ret == VIM_NO)
1523 {
1524 unchanged(buf, TRUE);
1525 }
1526 else if (ret == VIM_ALL)
1527 {
1528 /*
1529 * Write all modified files that can be written.
1530 * Skip readonly buffers, these need to be confirmed
1531 * individually.
1532 */
1533 for (buf2 = firstbuf; buf2 != NULL; buf2 = buf2->b_next)
1534 {
1535 if (bufIsChanged(buf2)
1536 && (buf2->b_ffname != NULL
1537#ifdef FEAT_BROWSE
1538 || cmdmod.browse
1539#endif
1540 )
1541 && !buf2->b_p_ro)
1542 {
1543#ifdef FEAT_BROWSE
1544 /* May get file name, when there is none */
1545 browse_save_fname(buf2);
1546#endif
Bram Moolenaar8218f602012-04-25 17:32:18 +02001547 if (buf2->b_fname != NULL && check_overwrite(&ea, buf2,
1548 buf2->b_fname, buf2->b_ffname, FALSE) == OK)
1549 /* didn't hit Cancel */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001550 (void)buf_write_all(buf2, FALSE);
1551#ifdef FEAT_AUTOCMD
1552 /* an autocommand may have deleted the buffer */
1553 if (!buf_valid(buf2))
1554 buf2 = firstbuf;
1555#endif
1556 }
1557 }
1558 }
1559 else if (ret == VIM_DISCARDALL)
1560 {
1561 /*
1562 * mark all buffers as unchanged
1563 */
1564 for (buf2 = firstbuf; buf2 != NULL; buf2 = buf2->b_next)
1565 unchanged(buf2, TRUE);
1566 }
1567}
1568#endif
1569
1570/*
1571 * Return TRUE if the buffer "buf" can be abandoned, either by making it
1572 * hidden, autowriting it or unloading it.
1573 */
1574 int
1575can_abandon(buf, forceit)
1576 buf_T *buf;
1577 int forceit;
1578{
1579 return ( P_HID(buf)
1580 || !bufIsChanged(buf)
1581 || buf->b_nwindows > 1
1582 || autowrite(buf, forceit) == OK
1583 || forceit);
1584}
1585
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001586static void add_bufnum __ARGS((int *bufnrs, int *bufnump, int nr));
1587
1588/*
1589 * Add a buffer number to "bufnrs", unless it's already there.
1590 */
1591 static void
1592add_bufnum(bufnrs, bufnump, nr)
1593 int *bufnrs;
1594 int *bufnump;
1595 int nr;
1596{
1597 int i;
1598
1599 for (i = 0; i < *bufnump; ++i)
1600 if (bufnrs[i] == nr)
1601 return;
1602 bufnrs[*bufnump] = nr;
1603 *bufnump = *bufnump + 1;
1604}
1605
Bram Moolenaar071d4272004-06-13 20:20:40 +00001606/*
1607 * Return TRUE if any buffer was changed and cannot be abandoned.
1608 * That changed buffer becomes the current buffer.
1609 */
1610 int
1611check_changed_any(hidden)
1612 int hidden; /* Only check hidden buffers */
1613{
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001614 int ret = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001615 buf_T *buf;
1616 int save;
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001617 int i;
1618 int bufnum = 0;
1619 int bufcount = 0;
1620 int *bufnrs;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001621#ifdef FEAT_WINDOWS
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001622 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001623 win_T *wp;
1624#endif
1625
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001626 for (buf = firstbuf; buf != NULL; buf = buf->b_next)
1627 ++bufcount;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001628
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001629 if (bufcount == 0)
1630 return FALSE;
1631
1632 bufnrs = (int *)alloc(sizeof(int) * bufcount);
1633 if (bufnrs == NULL)
1634 return FALSE;
1635
1636 /* curbuf */
1637 bufnrs[bufnum++] = curbuf->b_fnum;
1638#ifdef FEAT_WINDOWS
1639 /* buf in curtab */
1640 FOR_ALL_WINDOWS(wp)
1641 if (wp->w_buffer != curbuf)
1642 add_bufnum(bufnrs, &bufnum, wp->w_buffer->b_fnum);
1643
1644 /* buf in other tab */
1645 for (tp = first_tabpage; tp != NULL; tp = tp->tp_next)
1646 if (tp != curtab)
1647 for (wp = tp->tp_firstwin; wp != NULL; wp = wp->w_next)
1648 add_bufnum(bufnrs, &bufnum, wp->w_buffer->b_fnum);
1649#endif
1650 /* any other buf */
1651 for (buf = firstbuf; buf != NULL; buf = buf->b_next)
1652 add_bufnum(bufnrs, &bufnum, buf->b_fnum);
1653
1654 for (i = 0; i < bufnum; ++i)
1655 {
1656 buf = buflist_findnr(bufnrs[i]);
1657 if (buf == NULL)
1658 continue;
1659 if ((!hidden || buf->b_nwindows == 0) && bufIsChanged(buf))
1660 {
1661 /* Try auto-writing the buffer. If this fails but the buffer no
1662 * longer exists it's not changed, that's OK. */
1663 if (check_changed(buf, p_awa, TRUE, FALSE, TRUE) && buf_valid(buf))
1664 break; /* didn't save - still changes */
1665 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001666 }
1667
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001668 if (i >= bufnum)
1669 goto theend;
1670
1671 ret = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001672 exiting = FALSE;
1673#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG)
1674 /*
1675 * When ":confirm" used, don't give an error message.
1676 */
1677 if (!(p_confirm || cmdmod.confirm))
1678#endif
1679 {
1680 /* There must be a wait_return for this message, do_buffer()
1681 * may cause a redraw. But wait_return() is a no-op when vgetc()
1682 * is busy (Quit used from window menu), then make sure we don't
1683 * cause a scroll up. */
Bram Moolenaar61660ea2006-04-07 21:40:07 +00001684 if (vgetc_busy > 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001685 {
1686 msg_row = cmdline_row;
1687 msg_col = 0;
1688 msg_didout = FALSE;
1689 }
1690 if (EMSG2(_("E162: No write since last change for buffer \"%s\""),
1691 buf_spname(buf) != NULL ? (char_u *)buf_spname(buf) :
1692 buf->b_fname))
1693 {
1694 save = no_wait_return;
1695 no_wait_return = FALSE;
1696 wait_return(FALSE);
1697 no_wait_return = save;
1698 }
1699 }
1700
1701#ifdef FEAT_WINDOWS
1702 /* Try to find a window that contains the buffer. */
1703 if (buf != curbuf)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001704 FOR_ALL_TAB_WINDOWS(tp, wp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001705 if (wp->w_buffer == buf)
1706 {
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001707 goto_tabpage_win(tp, wp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001708# ifdef FEAT_AUTOCMD
1709 /* Paranoia: did autocms wipe out the buffer with changes? */
1710 if (!buf_valid(buf))
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001711 {
1712 goto theend;
1713 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001714# endif
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001715 goto buf_found;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001716 }
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001717buf_found:
Bram Moolenaar071d4272004-06-13 20:20:40 +00001718#endif
1719
1720 /* Open the changed buffer in the current window. */
1721 if (buf != curbuf)
1722 set_curbuf(buf, DOBUF_GOTO);
1723
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001724theend:
1725 vim_free(bufnrs);
1726 return ret;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001727}
1728
1729/*
1730 * return FAIL if there is no file name, OK if there is one
1731 * give error message for FAIL
1732 */
1733 int
1734check_fname()
1735{
1736 if (curbuf->b_ffname == NULL)
1737 {
1738 EMSG(_(e_noname));
1739 return FAIL;
1740 }
1741 return OK;
1742}
1743
1744/*
1745 * flush the contents of a buffer, unless it has no file name
1746 *
1747 * return FAIL for failure, OK otherwise
1748 */
1749 int
1750buf_write_all(buf, forceit)
1751 buf_T *buf;
1752 int forceit;
1753{
1754 int retval;
1755#ifdef FEAT_AUTOCMD
1756 buf_T *old_curbuf = curbuf;
1757#endif
1758
1759 retval = (buf_write(buf, buf->b_ffname, buf->b_fname,
1760 (linenr_T)1, buf->b_ml.ml_line_count, NULL,
1761 FALSE, forceit, TRUE, FALSE));
1762#ifdef FEAT_AUTOCMD
1763 if (curbuf != old_curbuf)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001764 {
1765 msg_source(hl_attr(HLF_W));
Bram Moolenaar071d4272004-06-13 20:20:40 +00001766 MSG(_("Warning: Entered other buffer unexpectedly (check autocommands)"));
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001767 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001768#endif
1769 return retval;
1770}
1771
1772/*
1773 * Code to handle the argument list.
1774 */
1775
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001776static char_u *do_one_arg __ARGS((char_u *str));
1777static int do_arglist __ARGS((char_u *str, int what, int after));
1778static void alist_check_arg_idx __ARGS((void));
1779static int editing_arg_idx __ARGS((win_T *win));
1780#ifdef FEAT_LISTCMDS
1781static int alist_add_list __ARGS((int count, char_u **files, int after));
1782#endif
1783#define AL_SET 1
1784#define AL_ADD 2
1785#define AL_DEL 3
1786
Bram Moolenaar071d4272004-06-13 20:20:40 +00001787/*
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001788 * Isolate one argument, taking backticks.
1789 * Changes the argument in-place, puts a NUL after it. Backticks remain.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001790 * Return a pointer to the start of the next argument.
1791 */
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001792 static char_u *
Bram Moolenaar071d4272004-06-13 20:20:40 +00001793do_one_arg(str)
1794 char_u *str;
1795{
1796 char_u *p;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001797 int inbacktick;
1798
Bram Moolenaar071d4272004-06-13 20:20:40 +00001799 inbacktick = FALSE;
1800 for (p = str; *str; ++str)
1801 {
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001802 /* When the backslash is used for escaping the special meaning of a
1803 * character we need to keep it until wildcard expansion. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001804 if (rem_backslash(str))
1805 {
1806 *p++ = *str++;
1807 *p++ = *str;
1808 }
1809 else
1810 {
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001811 /* An item ends at a space not in backticks */
1812 if (!inbacktick && vim_isspace(*str))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001813 break;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001814 if (*str == '`')
Bram Moolenaar071d4272004-06-13 20:20:40 +00001815 inbacktick ^= TRUE;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001816 *p++ = *str;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001817 }
1818 }
1819 str = skipwhite(str);
1820 *p = NUL;
1821
1822 return str;
1823}
1824
Bram Moolenaar86b68352004-12-27 21:59:20 +00001825/*
1826 * Separate the arguments in "str" and return a list of pointers in the
1827 * growarray "gap".
1828 */
1829 int
1830get_arglist(gap, str)
1831 garray_T *gap;
1832 char_u *str;
1833{
1834 ga_init2(gap, (int)sizeof(char_u *), 20);
1835 while (*str != NUL)
1836 {
1837 if (ga_grow(gap, 1) == FAIL)
1838 {
1839 ga_clear(gap);
1840 return FAIL;
1841 }
1842 ((char_u **)gap->ga_data)[gap->ga_len++] = str;
1843
1844 /* Isolate one argument, change it in-place, put a NUL after it. */
1845 str = do_one_arg(str);
1846 }
1847 return OK;
1848}
1849
Bram Moolenaar7df351e2006-01-23 22:30:28 +00001850#if defined(FEAT_QUICKFIX) || defined(FEAT_SYN_HL) || defined(PROTO)
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00001851/*
1852 * Parse a list of arguments (file names), expand them and return in
1853 * "fnames[fcountp]".
1854 * Return FAIL or OK.
1855 */
1856 int
1857get_arglist_exp(str, fcountp, fnamesp)
1858 char_u *str;
1859 int *fcountp;
1860 char_u ***fnamesp;
1861{
1862 garray_T ga;
1863 int i;
1864
1865 if (get_arglist(&ga, str) == FAIL)
1866 return FAIL;
1867 i = gen_expand_wildcards(ga.ga_len, (char_u **)ga.ga_data,
1868 fcountp, fnamesp, EW_FILE|EW_NOTFOUND);
1869 ga_clear(&ga);
1870 return i;
1871}
1872#endif
1873
Bram Moolenaar071d4272004-06-13 20:20:40 +00001874#if defined(FEAT_GUI) || defined(FEAT_CLIENTSERVER) || defined(PROTO)
1875/*
1876 * Redefine the argument list.
1877 */
1878 void
1879set_arglist(str)
1880 char_u *str;
1881{
1882 do_arglist(str, AL_SET, 0);
1883}
1884#endif
1885
1886/*
1887 * "what" == AL_SET: Redefine the argument list to 'str'.
1888 * "what" == AL_ADD: add files in 'str' to the argument list after "after".
1889 * "what" == AL_DEL: remove files in 'str' from the argument list.
1890 *
1891 * Return FAIL for failure, OK otherwise.
1892 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001893 static int
1894do_arglist(str, what, after)
1895 char_u *str;
Bram Moolenaar0c094b92009-05-14 20:20:33 +00001896 int what UNUSED;
1897 int after UNUSED; /* 0 means before first one */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001898{
1899 garray_T new_ga;
1900 int exp_count;
1901 char_u **exp_files;
1902 int i;
1903#ifdef FEAT_LISTCMDS
1904 char_u *p;
1905 int match;
1906#endif
1907
1908 /*
1909 * Collect all file name arguments in "new_ga".
1910 */
Bram Moolenaar86b68352004-12-27 21:59:20 +00001911 if (get_arglist(&new_ga, str) == FAIL)
1912 return FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001913
1914#ifdef FEAT_LISTCMDS
1915 if (what == AL_DEL)
1916 {
1917 regmatch_T regmatch;
1918 int didone;
1919
1920 /*
1921 * Delete the items: use each item as a regexp and find a match in the
1922 * argument list.
1923 */
1924#ifdef CASE_INSENSITIVE_FILENAME
1925 regmatch.rm_ic = TRUE; /* Always ignore case */
1926#else
1927 regmatch.rm_ic = FALSE; /* Never ignore case */
1928#endif
1929 for (i = 0; i < new_ga.ga_len && !got_int; ++i)
1930 {
1931 p = ((char_u **)new_ga.ga_data)[i];
1932 p = file_pat_to_reg_pat(p, NULL, NULL, FALSE);
1933 if (p == NULL)
1934 break;
1935 regmatch.regprog = vim_regcomp(p, p_magic ? RE_MAGIC : 0);
1936 if (regmatch.regprog == NULL)
1937 {
1938 vim_free(p);
1939 break;
1940 }
1941
1942 didone = FALSE;
1943 for (match = 0; match < ARGCOUNT; ++match)
1944 if (vim_regexec(&regmatch, alist_name(&ARGLIST[match]),
1945 (colnr_T)0))
1946 {
1947 didone = TRUE;
1948 vim_free(ARGLIST[match].ae_fname);
1949 mch_memmove(ARGLIST + match, ARGLIST + match + 1,
1950 (ARGCOUNT - match - 1) * sizeof(aentry_T));
1951 --ALIST(curwin)->al_ga.ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001952 if (curwin->w_arg_idx > match)
1953 --curwin->w_arg_idx;
1954 --match;
1955 }
1956
1957 vim_free(regmatch.regprog);
1958 vim_free(p);
1959 if (!didone)
1960 EMSG2(_(e_nomatch2), ((char_u **)new_ga.ga_data)[i]);
1961 }
1962 ga_clear(&new_ga);
1963 }
1964 else
1965#endif
1966 {
1967 i = expand_wildcards(new_ga.ga_len, (char_u **)new_ga.ga_data,
1968 &exp_count, &exp_files, EW_DIR|EW_FILE|EW_ADDSLASH|EW_NOTFOUND);
1969 ga_clear(&new_ga);
1970 if (i == FAIL)
1971 return FAIL;
1972 if (exp_count == 0)
1973 {
1974 EMSG(_(e_nomatch));
1975 return FAIL;
1976 }
1977
1978#ifdef FEAT_LISTCMDS
1979 if (what == AL_ADD)
1980 {
1981 (void)alist_add_list(exp_count, exp_files, after);
1982 vim_free(exp_files);
1983 }
1984 else /* what == AL_SET */
1985#endif
Bram Moolenaar86b68352004-12-27 21:59:20 +00001986 alist_set(ALIST(curwin), exp_count, exp_files, FALSE, NULL, 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001987 }
1988
1989 alist_check_arg_idx();
1990
1991 return OK;
1992}
1993
1994/*
1995 * Check the validity of the arg_idx for each other window.
1996 */
1997 static void
1998alist_check_arg_idx()
1999{
2000#ifdef FEAT_WINDOWS
2001 win_T *win;
Bram Moolenaarf740b292006-02-16 22:11:02 +00002002 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002003
Bram Moolenaarf740b292006-02-16 22:11:02 +00002004 FOR_ALL_TAB_WINDOWS(tp, win)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002005 if (win->w_alist == curwin->w_alist)
2006 check_arg_idx(win);
2007#else
2008 check_arg_idx(curwin);
2009#endif
2010}
2011
2012/*
Bram Moolenaarb8ff1fb2012-02-04 21:59:01 +01002013 * Return TRUE if window "win" is editing the file at the current argument
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002014 * index.
2015 */
2016 static int
2017editing_arg_idx(win)
2018 win_T *win;
2019{
2020 return !(win->w_arg_idx >= WARGCOUNT(win)
2021 || (win->w_buffer->b_fnum
2022 != WARGLIST(win)[win->w_arg_idx].ae_fnum
2023 && (win->w_buffer->b_ffname == NULL
2024 || !(fullpathcmp(
2025 alist_name(&WARGLIST(win)[win->w_arg_idx]),
2026 win->w_buffer->b_ffname, TRUE) & FPC_SAME))));
2027}
2028
2029/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002030 * Check if window "win" is editing the w_arg_idx file in its argument list.
2031 */
2032 void
2033check_arg_idx(win)
2034 win_T *win;
2035{
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002036 if (WARGCOUNT(win) > 1 && !editing_arg_idx(win))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002037 {
2038 /* We are not editing the current entry in the argument list.
2039 * Set "arg_had_last" if we are editing the last one. */
2040 win->w_arg_idx_invalid = TRUE;
2041 if (win->w_arg_idx != WARGCOUNT(win) - 1
2042 && arg_had_last == FALSE
2043#ifdef FEAT_WINDOWS
2044 && ALIST(win) == &global_alist
2045#endif
2046 && GARGCOUNT > 0
2047 && win->w_arg_idx < GARGCOUNT
2048 && (win->w_buffer->b_fnum == GARGLIST[GARGCOUNT - 1].ae_fnum
2049 || (win->w_buffer->b_ffname != NULL
2050 && (fullpathcmp(alist_name(&GARGLIST[GARGCOUNT - 1]),
2051 win->w_buffer->b_ffname, TRUE) & FPC_SAME))))
2052 arg_had_last = TRUE;
2053 }
2054 else
2055 {
2056 /* We are editing the current entry in the argument list.
2057 * Set "arg_had_last" if it's also the last one */
2058 win->w_arg_idx_invalid = FALSE;
2059 if (win->w_arg_idx == WARGCOUNT(win) - 1
2060#ifdef FEAT_WINDOWS
2061 && win->w_alist == &global_alist
2062#endif
2063 )
2064 arg_had_last = TRUE;
2065 }
2066}
2067
2068/*
2069 * ":args", ":argslocal" and ":argsglobal".
2070 */
2071 void
2072ex_args(eap)
2073 exarg_T *eap;
2074{
2075 int i;
2076
2077 if (eap->cmdidx != CMD_args)
2078 {
2079#if defined(FEAT_WINDOWS) && defined(FEAT_LISTCMDS)
2080 alist_unlink(ALIST(curwin));
2081 if (eap->cmdidx == CMD_argglobal)
2082 ALIST(curwin) = &global_alist;
2083 else /* eap->cmdidx == CMD_arglocal */
2084 alist_new();
2085#else
2086 ex_ni(eap);
2087 return;
2088#endif
2089 }
2090
2091 if (!ends_excmd(*eap->arg))
2092 {
2093 /*
2094 * ":args file ..": define new argument list, handle like ":next"
2095 * Also for ":argslocal file .." and ":argsglobal file ..".
2096 */
2097 ex_next(eap);
2098 }
2099 else
2100#if defined(FEAT_WINDOWS) && defined(FEAT_LISTCMDS)
2101 if (eap->cmdidx == CMD_args)
2102#endif
2103 {
2104 /*
2105 * ":args": list arguments.
2106 */
2107 if (ARGCOUNT > 0)
2108 {
2109 /* Overwrite the command, for a short list there is no scrolling
2110 * required and no wait_return(). */
2111 gotocmdline(TRUE);
2112 for (i = 0; i < ARGCOUNT; ++i)
2113 {
2114 if (i == curwin->w_arg_idx)
2115 msg_putchar('[');
2116 msg_outtrans(alist_name(&ARGLIST[i]));
2117 if (i == curwin->w_arg_idx)
2118 msg_putchar(']');
2119 msg_putchar(' ');
2120 }
2121 }
2122 }
2123#if defined(FEAT_WINDOWS) && defined(FEAT_LISTCMDS)
2124 else if (eap->cmdidx == CMD_arglocal)
2125 {
2126 garray_T *gap = &curwin->w_alist->al_ga;
2127
2128 /*
2129 * ":argslocal": make a local copy of the global argument list.
2130 */
2131 if (ga_grow(gap, GARGCOUNT) == OK)
2132 for (i = 0; i < GARGCOUNT; ++i)
2133 if (GARGLIST[i].ae_fname != NULL)
2134 {
2135 AARGLIST(curwin->w_alist)[gap->ga_len].ae_fname =
2136 vim_strsave(GARGLIST[i].ae_fname);
2137 AARGLIST(curwin->w_alist)[gap->ga_len].ae_fnum =
2138 GARGLIST[i].ae_fnum;
2139 ++gap->ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002140 }
2141 }
2142#endif
2143}
2144
2145/*
2146 * ":previous", ":sprevious", ":Next" and ":sNext".
2147 */
2148 void
2149ex_previous(eap)
2150 exarg_T *eap;
2151{
2152 /* If past the last one already, go to the last one. */
2153 if (curwin->w_arg_idx - (int)eap->line2 >= ARGCOUNT)
2154 do_argfile(eap, ARGCOUNT - 1);
2155 else
2156 do_argfile(eap, curwin->w_arg_idx - (int)eap->line2);
2157}
2158
2159/*
2160 * ":rewind", ":first", ":sfirst" and ":srewind".
2161 */
2162 void
2163ex_rewind(eap)
2164 exarg_T *eap;
2165{
2166 do_argfile(eap, 0);
2167}
2168
2169/*
2170 * ":last" and ":slast".
2171 */
2172 void
2173ex_last(eap)
2174 exarg_T *eap;
2175{
2176 do_argfile(eap, ARGCOUNT - 1);
2177}
2178
2179/*
2180 * ":argument" and ":sargument".
2181 */
2182 void
2183ex_argument(eap)
2184 exarg_T *eap;
2185{
2186 int i;
2187
2188 if (eap->addr_count > 0)
2189 i = eap->line2 - 1;
2190 else
2191 i = curwin->w_arg_idx;
2192 do_argfile(eap, i);
2193}
2194
2195/*
2196 * Edit file "argn" of the argument lists.
2197 */
2198 void
2199do_argfile(eap, argn)
2200 exarg_T *eap;
2201 int argn;
2202{
2203 int other;
2204 char_u *p;
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002205 int old_arg_idx = curwin->w_arg_idx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002206
2207 if (argn < 0 || argn >= ARGCOUNT)
2208 {
2209 if (ARGCOUNT <= 1)
2210 EMSG(_("E163: There is only one file to edit"));
2211 else if (argn < 0)
2212 EMSG(_("E164: Cannot go before first file"));
2213 else
2214 EMSG(_("E165: Cannot go beyond last file"));
2215 }
2216 else
2217 {
2218 setpcmark();
2219#ifdef FEAT_GUI
2220 need_mouse_correct = TRUE;
2221#endif
2222
2223#ifdef FEAT_WINDOWS
Bram Moolenaardf1bdc92006-02-23 21:32:16 +00002224 /* split window or create new tab page first */
2225 if (*eap->cmd == 's' || cmdmod.tab != 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002226 {
2227 if (win_split(0, 0) == FAIL)
2228 return;
Bram Moolenaar3368ea22010-09-21 16:56:35 +02002229 RESET_BINDING(curwin);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002230 }
2231 else
2232#endif
2233 {
2234 /*
2235 * if 'hidden' set, only check for changed file when re-editing
2236 * the same buffer
2237 */
2238 other = TRUE;
2239 if (P_HID(curbuf))
2240 {
2241 p = fix_fname(alist_name(&ARGLIST[argn]));
2242 other = otherfile(p);
2243 vim_free(p);
2244 }
2245 if ((!P_HID(curbuf) || !other)
2246 && check_changed(curbuf, TRUE, !other, eap->forceit, FALSE))
2247 return;
2248 }
2249
2250 curwin->w_arg_idx = argn;
2251 if (argn == ARGCOUNT - 1
2252#ifdef FEAT_WINDOWS
2253 && curwin->w_alist == &global_alist
2254#endif
2255 )
2256 arg_had_last = TRUE;
2257
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002258 /* Edit the file; always use the last known line number.
2259 * When it fails (e.g. Abort for already edited file) restore the
2260 * argument index. */
2261 if (do_ecmd(0, alist_name(&ARGLIST[curwin->w_arg_idx]), NULL,
Bram Moolenaar071d4272004-06-13 20:20:40 +00002262 eap, ECMD_LAST,
Bram Moolenaar701f7af2008-11-15 13:12:07 +00002263 (P_HID(curwin->w_buffer) ? ECMD_HIDE : 0)
2264 + (eap->forceit ? ECMD_FORCEIT : 0), curwin) == FAIL)
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002265 curwin->w_arg_idx = old_arg_idx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002266 /* like Vi: set the mark where the cursor is in the file. */
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002267 else if (eap->cmdidx != CMD_argdo)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002268 setmark('\'');
2269 }
2270}
2271
2272/*
2273 * ":next", and commands that behave like it.
2274 */
2275 void
2276ex_next(eap)
2277 exarg_T *eap;
2278{
2279 int i;
2280
2281 /*
2282 * check for changed buffer now, if this fails the argument list is not
2283 * redefined.
2284 */
2285 if ( P_HID(curbuf)
2286 || eap->cmdidx == CMD_snext
2287 || !check_changed(curbuf, TRUE, FALSE, eap->forceit, FALSE))
2288 {
2289 if (*eap->arg != NUL) /* redefine file list */
2290 {
2291 if (do_arglist(eap->arg, AL_SET, 0) == FAIL)
2292 return;
2293 i = 0;
2294 }
2295 else
2296 i = curwin->w_arg_idx + (int)eap->line2;
2297 do_argfile(eap, i);
2298 }
2299}
2300
2301#ifdef FEAT_LISTCMDS
2302/*
2303 * ":argedit"
2304 */
2305 void
2306ex_argedit(eap)
2307 exarg_T *eap;
2308{
2309 int fnum;
2310 int i;
2311 char_u *s;
2312
2313 /* Add the argument to the buffer list and get the buffer number. */
2314 fnum = buflist_add(eap->arg, BLN_LISTED);
2315
2316 /* Check if this argument is already in the argument list. */
2317 for (i = 0; i < ARGCOUNT; ++i)
2318 if (ARGLIST[i].ae_fnum == fnum)
2319 break;
2320 if (i == ARGCOUNT)
2321 {
2322 /* Can't find it, add it to the argument list. */
2323 s = vim_strsave(eap->arg);
2324 if (s == NULL)
2325 return;
2326 i = alist_add_list(1, &s,
2327 eap->addr_count > 0 ? (int)eap->line2 : curwin->w_arg_idx + 1);
2328 if (i < 0)
2329 return;
2330 curwin->w_arg_idx = i;
2331 }
2332
2333 alist_check_arg_idx();
2334
2335 /* Edit the argument. */
2336 do_argfile(eap, i);
2337}
2338
2339/*
2340 * ":argadd"
2341 */
2342 void
2343ex_argadd(eap)
2344 exarg_T *eap;
2345{
2346 do_arglist(eap->arg, AL_ADD,
2347 eap->addr_count > 0 ? (int)eap->line2 : curwin->w_arg_idx + 1);
2348#ifdef FEAT_TITLE
2349 maketitle();
2350#endif
2351}
2352
2353/*
2354 * ":argdelete"
2355 */
2356 void
2357ex_argdelete(eap)
2358 exarg_T *eap;
2359{
2360 int i;
2361 int n;
2362
2363 if (eap->addr_count > 0)
2364 {
2365 /* ":1,4argdel": Delete all arguments in the range. */
2366 if (eap->line2 > ARGCOUNT)
2367 eap->line2 = ARGCOUNT;
2368 n = eap->line2 - eap->line1 + 1;
2369 if (*eap->arg != NUL || n <= 0)
2370 EMSG(_(e_invarg));
2371 else
2372 {
2373 for (i = eap->line1; i <= eap->line2; ++i)
2374 vim_free(ARGLIST[i - 1].ae_fname);
2375 mch_memmove(ARGLIST + eap->line1 - 1, ARGLIST + eap->line2,
2376 (size_t)((ARGCOUNT - eap->line2) * sizeof(aentry_T)));
2377 ALIST(curwin)->al_ga.ga_len -= n;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002378 if (curwin->w_arg_idx >= eap->line2)
2379 curwin->w_arg_idx -= n;
2380 else if (curwin->w_arg_idx > eap->line1)
2381 curwin->w_arg_idx = eap->line1;
2382 }
2383 }
2384 else if (*eap->arg == NUL)
2385 EMSG(_(e_argreq));
2386 else
2387 do_arglist(eap->arg, AL_DEL, 0);
2388#ifdef FEAT_TITLE
2389 maketitle();
2390#endif
2391}
2392
2393/*
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002394 * ":argdo", ":windo", ":bufdo", ":tabdo"
Bram Moolenaar071d4272004-06-13 20:20:40 +00002395 */
2396 void
2397ex_listdo(eap)
2398 exarg_T *eap;
2399{
2400 int i;
2401#ifdef FEAT_WINDOWS
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002402 win_T *wp;
2403 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002404#endif
2405 buf_T *buf;
2406 int next_fnum = 0;
2407#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
2408 char_u *save_ei = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002409#endif
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002410 char_u *p_shm_save;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002411
2412#ifndef FEAT_WINDOWS
2413 if (eap->cmdidx == CMD_windo)
2414 {
2415 ex_ni(eap);
2416 return;
2417 }
2418#endif
2419
2420#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002421 if (eap->cmdidx != CMD_windo && eap->cmdidx != CMD_tabdo)
Bram Moolenaardcaf10e2005-01-21 11:55:25 +00002422 /* Don't do syntax HL autocommands. Skipping the syntax file is a
2423 * great speed improvement. */
2424 save_ei = au_event_disable(",Syntax");
Bram Moolenaar071d4272004-06-13 20:20:40 +00002425#endif
2426
2427 if (eap->cmdidx == CMD_windo
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002428 || eap->cmdidx == CMD_tabdo
Bram Moolenaar071d4272004-06-13 20:20:40 +00002429 || P_HID(curbuf)
2430 || !check_changed(curbuf, TRUE, FALSE, eap->forceit, FALSE))
2431 {
2432 /* start at the first argument/window/buffer */
2433 i = 0;
2434#ifdef FEAT_WINDOWS
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002435 wp = firstwin;
2436 tp = first_tabpage;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002437#endif
2438 /* set pcmark now */
2439 if (eap->cmdidx == CMD_bufdo)
2440 goto_buffer(eap, DOBUF_FIRST, FORWARD, 0);
2441 else
2442 setpcmark();
2443 listcmd_busy = TRUE; /* avoids setting pcmark below */
2444
2445 while (!got_int)
2446 {
2447 if (eap->cmdidx == CMD_argdo)
2448 {
2449 /* go to argument "i" */
2450 if (i == ARGCOUNT)
2451 break;
2452 /* Don't call do_argfile() when already there, it will try
2453 * reloading the file. */
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002454 if (curwin->w_arg_idx != i || !editing_arg_idx(curwin))
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002455 {
2456 /* Clear 'shm' to avoid that the file message overwrites
2457 * any output from the command. */
2458 p_shm_save = vim_strsave(p_shm);
2459 set_option_value((char_u *)"shm", 0L, (char_u *)"", 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002460 do_argfile(eap, i);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002461 set_option_value((char_u *)"shm", 0L, p_shm_save, 0);
2462 vim_free(p_shm_save);
2463 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002464 if (curwin->w_arg_idx != i)
2465 break;
2466 ++i;
2467 }
2468#ifdef FEAT_WINDOWS
2469 else if (eap->cmdidx == CMD_windo)
2470 {
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002471 /* go to window "wp" */
2472 if (!win_valid(wp))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002473 break;
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002474 win_goto(wp);
Bram Moolenaar41423242007-05-03 20:11:13 +00002475 if (curwin != wp)
2476 break; /* something must be wrong */
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002477 wp = curwin->w_next;
2478 }
2479 else if (eap->cmdidx == CMD_tabdo)
2480 {
2481 /* go to window "tp" */
2482 if (!valid_tabpage(tp))
2483 break;
Bram Moolenaara8596c42012-06-13 14:28:20 +02002484 goto_tabpage_tp(tp, TRUE);
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002485 tp = tp->tp_next;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002486 }
2487#endif
2488 else if (eap->cmdidx == CMD_bufdo)
2489 {
2490 /* Remember the number of the next listed buffer, in case
2491 * ":bwipe" is used or autocommands do something strange. */
2492 next_fnum = -1;
2493 for (buf = curbuf->b_next; buf != NULL; buf = buf->b_next)
2494 if (buf->b_p_bl)
2495 {
2496 next_fnum = buf->b_fnum;
2497 break;
2498 }
2499 }
2500
2501 /* execute the command */
2502 do_cmdline(eap->arg, eap->getline, eap->cookie,
2503 DOCMD_VERBOSE + DOCMD_NOWAIT);
2504
2505 if (eap->cmdidx == CMD_bufdo)
2506 {
2507 /* Done? */
2508 if (next_fnum < 0)
2509 break;
2510 /* Check if the buffer still exists. */
2511 for (buf = firstbuf; buf != NULL; buf = buf->b_next)
2512 if (buf->b_fnum == next_fnum)
2513 break;
2514 if (buf == NULL)
2515 break;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002516
2517 /* Go to the next buffer. Clear 'shm' to avoid that the file
2518 * message overwrites any output from the command. */
2519 p_shm_save = vim_strsave(p_shm);
2520 set_option_value((char_u *)"shm", 0L, (char_u *)"", 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002521 goto_buffer(eap, DOBUF_FIRST, FORWARD, next_fnum);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002522 set_option_value((char_u *)"shm", 0L, p_shm_save, 0);
2523 vim_free(p_shm_save);
2524
Bram Moolenaar071d4272004-06-13 20:20:40 +00002525 /* If autocommands took us elsewhere, quit here */
2526 if (curbuf->b_fnum != next_fnum)
2527 break;
2528 }
2529
2530 if (eap->cmdidx == CMD_windo)
2531 {
2532 validate_cursor(); /* cursor may have moved */
2533#ifdef FEAT_SCROLLBIND
2534 /* required when 'scrollbind' has been set */
2535 if (curwin->w_p_scb)
2536 do_check_scrollbind(TRUE);
2537#endif
2538 }
2539 }
2540 listcmd_busy = FALSE;
2541 }
2542
2543#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
Bram Moolenaar6ac54292005-02-02 23:07:25 +00002544 if (save_ei != NULL)
2545 {
2546 au_event_restore(save_ei);
2547 apply_autocmds(EVENT_SYNTAX, curbuf->b_p_syn,
2548 curbuf->b_fname, TRUE, curbuf);
2549 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002550#endif
2551}
2552
2553/*
2554 * Add files[count] to the arglist of the current window after arg "after".
2555 * The file names in files[count] must have been allocated and are taken over.
2556 * Files[] itself is not taken over.
2557 * Returns index of first added argument. Returns -1 when failed (out of mem).
2558 */
2559 static int
2560alist_add_list(count, files, after)
2561 int count;
2562 char_u **files;
2563 int after; /* where to add: 0 = before first one */
2564{
2565 int i;
2566
2567 if (ga_grow(&ALIST(curwin)->al_ga, count) == OK)
2568 {
2569 if (after < 0)
2570 after = 0;
2571 if (after > ARGCOUNT)
2572 after = ARGCOUNT;
2573 if (after < ARGCOUNT)
2574 mch_memmove(&(ARGLIST[after + count]), &(ARGLIST[after]),
2575 (ARGCOUNT - after) * sizeof(aentry_T));
2576 for (i = 0; i < count; ++i)
2577 {
2578 ARGLIST[after + i].ae_fname = files[i];
2579 ARGLIST[after + i].ae_fnum = buflist_add(files[i], BLN_LISTED);
2580 }
2581 ALIST(curwin)->al_ga.ga_len += count;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002582 if (curwin->w_arg_idx >= after)
2583 ++curwin->w_arg_idx;
2584 return after;
2585 }
2586
2587 for (i = 0; i < count; ++i)
2588 vim_free(files[i]);
2589 return -1;
2590}
2591
2592#endif /* FEAT_LISTCMDS */
2593
2594#ifdef FEAT_EVAL
2595/*
2596 * ":compiler[!] {name}"
2597 */
2598 void
2599ex_compiler(eap)
2600 exarg_T *eap;
2601{
2602 char_u *buf;
2603 char_u *old_cur_comp = NULL;
2604 char_u *p;
2605
2606 if (*eap->arg == NUL)
2607 {
2608 /* List all compiler scripts. */
2609 do_cmdline_cmd((char_u *)"echo globpath(&rtp, 'compiler/*.vim')");
2610 /* ) keep the indenter happy... */
2611 }
2612 else
2613 {
2614 buf = alloc((unsigned)(STRLEN(eap->arg) + 14));
2615 if (buf != NULL)
2616 {
2617 if (eap->forceit)
2618 {
2619 /* ":compiler! {name}" sets global options */
2620 do_cmdline_cmd((char_u *)
2621 "command -nargs=* CompilerSet set <args>");
2622 }
2623 else
2624 {
2625 /* ":compiler! {name}" sets local options.
2626 * To remain backwards compatible "current_compiler" is always
2627 * used. A user's compiler plugin may set it, the distributed
2628 * plugin will then skip the settings. Afterwards set
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01002629 * "b:current_compiler" and restore "current_compiler".
2630 * Explicitly prepend "g:" to make it work in a function. */
2631 old_cur_comp = get_var_value((char_u *)"g:current_compiler");
Bram Moolenaar071d4272004-06-13 20:20:40 +00002632 if (old_cur_comp != NULL)
2633 old_cur_comp = vim_strsave(old_cur_comp);
2634 do_cmdline_cmd((char_u *)
2635 "command -nargs=* CompilerSet setlocal <args>");
2636 }
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01002637 do_unlet((char_u *)"g:current_compiler", TRUE);
Bram Moolenaar2ce06f62005-01-31 19:19:04 +00002638 do_unlet((char_u *)"b:current_compiler", TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002639
2640 sprintf((char *)buf, "compiler/%s.vim", eap->arg);
Bram Moolenaar90cfdbe2005-08-12 19:59:19 +00002641 if (source_runtime(buf, TRUE) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002642 EMSG2(_("E666: compiler not supported: %s"), eap->arg);
2643 vim_free(buf);
2644
2645 do_cmdline_cmd((char_u *)":delcommand CompilerSet");
2646
2647 /* Set "b:current_compiler" from "current_compiler". */
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01002648 p = get_var_value((char_u *)"g:current_compiler");
Bram Moolenaar071d4272004-06-13 20:20:40 +00002649 if (p != NULL)
2650 set_internal_string_var((char_u *)"b:current_compiler", p);
2651
2652 /* Restore "current_compiler" for ":compiler {name}". */
2653 if (!eap->forceit)
2654 {
2655 if (old_cur_comp != NULL)
2656 {
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01002657 set_internal_string_var((char_u *)"g:current_compiler",
Bram Moolenaar071d4272004-06-13 20:20:40 +00002658 old_cur_comp);
2659 vim_free(old_cur_comp);
2660 }
2661 else
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01002662 do_unlet((char_u *)"g:current_compiler", TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002663 }
2664 }
2665 }
2666}
2667#endif
2668
2669/*
2670 * ":runtime {name}"
2671 */
2672 void
2673ex_runtime(eap)
2674 exarg_T *eap;
2675{
Bram Moolenaar90cfdbe2005-08-12 19:59:19 +00002676 source_runtime(eap->arg, eap->forceit);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002677}
2678
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002679static void source_callback __ARGS((char_u *fname, void *cookie));
Bram Moolenaar071d4272004-06-13 20:20:40 +00002680
2681 static void
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002682source_callback(fname, cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002683 char_u *fname;
Bram Moolenaar0c094b92009-05-14 20:20:33 +00002684 void *cookie UNUSED;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002685{
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002686 (void)do_source(fname, FALSE, DOSO_NONE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002687}
2688
2689/*
2690 * Source the file "name" from all directories in 'runtimepath'.
2691 * "name" can contain wildcards.
2692 * When "all" is TRUE, source all files, otherwise only the first one.
2693 * return FAIL when no file could be sourced, OK otherwise.
2694 */
2695 int
Bram Moolenaar90cfdbe2005-08-12 19:59:19 +00002696source_runtime(name, all)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002697 char_u *name;
2698 int all;
2699{
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002700 return do_in_runtimepath(name, all, source_callback, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002701}
2702
2703/*
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002704 * Find "name" in 'runtimepath'. When found, invoke the callback function for
2705 * it: callback(fname, "cookie")
Bram Moolenaar071d4272004-06-13 20:20:40 +00002706 * When "all" is TRUE repeat for all matches, otherwise only the first one is
2707 * used.
2708 * Returns OK when at least one match found, FAIL otherwise.
2709 */
2710 int
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002711do_in_runtimepath(name, all, callback, cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002712 char_u *name;
2713 int all;
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002714 void (*callback)__ARGS((char_u *fname, void *ck));
2715 void *cookie;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002716{
2717 char_u *rtp;
2718 char_u *np;
2719 char_u *buf;
2720 char_u *rtp_copy;
2721 char_u *tail;
2722 int num_files;
2723 char_u **files;
2724 int i;
2725 int did_one = FALSE;
2726#ifdef AMIGA
2727 struct Process *proc = (struct Process *)FindTask(0L);
2728 APTR save_winptr = proc->pr_WindowPtr;
2729
2730 /* Avoid a requester here for a volume that doesn't exist. */
2731 proc->pr_WindowPtr = (APTR)-1L;
2732#endif
2733
2734 /* Make a copy of 'runtimepath'. Invoking the callback may change the
2735 * value. */
2736 rtp_copy = vim_strsave(p_rtp);
2737 buf = alloc(MAXPATHL);
2738 if (buf != NULL && rtp_copy != NULL)
2739 {
2740 if (p_verbose > 1)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00002741 {
2742 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00002743 smsg((char_u *)_("Searching for \"%s\" in \"%s\""),
Bram Moolenaar071d4272004-06-13 20:20:40 +00002744 (char *)name, (char *)p_rtp);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00002745 verbose_leave();
2746 }
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002747
Bram Moolenaar071d4272004-06-13 20:20:40 +00002748 /* Loop over all entries in 'runtimepath'. */
2749 rtp = rtp_copy;
2750 while (*rtp != NUL && (all || !did_one))
2751 {
2752 /* Copy the path from 'runtimepath' to buf[]. */
2753 copy_option_part(&rtp, buf, MAXPATHL, ",");
2754 if (STRLEN(buf) + STRLEN(name) + 2 < MAXPATHL)
2755 {
2756 add_pathsep(buf);
2757 tail = buf + STRLEN(buf);
2758
2759 /* Loop over all patterns in "name" */
2760 np = name;
2761 while (*np != NUL && (all || !did_one))
2762 {
2763 /* Append the pattern from "name" to buf[]. */
2764 copy_option_part(&np, tail, (int)(MAXPATHL - (tail - buf)),
2765 "\t ");
2766
2767 if (p_verbose > 2)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00002768 {
2769 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00002770 smsg((char_u *)_("Searching for \"%s\""), buf);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00002771 verbose_leave();
2772 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002773
2774 /* Expand wildcards, invoke the callback for each match. */
2775 if (gen_expand_wildcards(1, &buf, &num_files, &files,
2776 EW_FILE) == OK)
2777 {
2778 for (i = 0; i < num_files; ++i)
2779 {
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002780 (*callback)(files[i], cookie);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002781 did_one = TRUE;
2782 if (!all)
2783 break;
2784 }
2785 FreeWild(num_files, files);
2786 }
2787 }
2788 }
2789 }
2790 }
2791 vim_free(buf);
2792 vim_free(rtp_copy);
2793 if (p_verbose > 0 && !did_one)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00002794 {
2795 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00002796 smsg((char_u *)_("not found in 'runtimepath': \"%s\""), name);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00002797 verbose_leave();
2798 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002799
2800#ifdef AMIGA
2801 proc->pr_WindowPtr = save_winptr;
2802#endif
2803
2804 return did_one ? OK : FAIL;
2805}
2806
2807#if defined(FEAT_EVAL) && defined(FEAT_AUTOCMD)
2808/*
2809 * ":options"
2810 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002811 void
2812ex_options(eap)
Bram Moolenaar0c094b92009-05-14 20:20:33 +00002813 exarg_T *eap UNUSED;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002814{
2815 cmd_source((char_u *)SYS_OPTWIN_FILE, NULL);
2816}
2817#endif
2818
2819/*
2820 * ":source {fname}"
2821 */
2822 void
2823ex_source(eap)
2824 exarg_T *eap;
2825{
2826#ifdef FEAT_BROWSE
2827 if (cmdmod.browse)
2828 {
2829 char_u *fname = NULL;
2830
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00002831 fname = do_browse(0, (char_u *)_("Source Vim script"), eap->arg,
Bram Moolenaar071d4272004-06-13 20:20:40 +00002832 NULL, NULL, BROWSE_FILTER_MACROS, NULL);
2833 if (fname != NULL)
2834 {
2835 cmd_source(fname, eap);
2836 vim_free(fname);
2837 }
2838 }
2839 else
2840#endif
2841 cmd_source(eap->arg, eap);
2842}
2843
2844 static void
2845cmd_source(fname, eap)
2846 char_u *fname;
2847 exarg_T *eap;
2848{
2849 if (*fname == NUL)
2850 EMSG(_(e_argreq));
2851
Bram Moolenaar071d4272004-06-13 20:20:40 +00002852 else if (eap != NULL && eap->forceit)
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00002853 /* ":source!": read Normal mdoe commands
2854 * Need to execute the commands directly. This is required at least
2855 * for:
Bram Moolenaar071d4272004-06-13 20:20:40 +00002856 * - ":g" command busy
2857 * - after ":argdo", ":windo" or ":bufdo"
2858 * - another command follows
2859 * - inside a loop
2860 */
2861 openscript(fname, global_busy || listcmd_busy || eap->nextcmd != NULL
2862#ifdef FEAT_EVAL
2863 || eap->cstack->cs_idx >= 0
2864#endif
2865 );
2866
2867 /* ":source" read ex commands */
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002868 else if (do_source(fname, FALSE, DOSO_NONE) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002869 EMSG2(_(e_notopen), fname);
2870}
2871
2872/*
2873 * ":source" and associated commands.
2874 */
2875/*
2876 * Structure used to store info for each sourced file.
2877 * It is shared between do_source() and getsourceline().
2878 * This is required, because it needs to be handed to do_cmdline() and
2879 * sourcing can be done recursively.
2880 */
2881struct source_cookie
2882{
2883 FILE *fp; /* opened file for sourcing */
2884 char_u *nextline; /* if not NULL: line that was read ahead */
2885 int finished; /* ":finish" used */
Bram Moolenaar860cae12010-06-05 23:22:07 +02002886#if defined(USE_CRNL) || defined(USE_CR)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002887 int fileformat; /* EOL_UNKNOWN, EOL_UNIX or EOL_DOS */
2888 int error; /* TRUE if LF found after CR-LF */
2889#endif
2890#ifdef FEAT_EVAL
2891 linenr_T breakpoint; /* next line with breakpoint or zero */
2892 char_u *fname; /* name of sourced file */
2893 int dbg_tick; /* debug_tick when breakpoint was set */
2894 int level; /* top nesting level of sourced file */
2895#endif
2896#ifdef FEAT_MBYTE
2897 vimconv_T conv; /* type of conversion */
2898#endif
2899};
2900
2901#ifdef FEAT_EVAL
2902/*
2903 * Return the address holding the next breakpoint line for a source cookie.
2904 */
2905 linenr_T *
2906source_breakpoint(cookie)
2907 void *cookie;
2908{
2909 return &((struct source_cookie *)cookie)->breakpoint;
2910}
2911
2912/*
2913 * Return the address holding the debug tick for a source cookie.
2914 */
2915 int *
2916source_dbg_tick(cookie)
2917 void *cookie;
2918{
2919 return &((struct source_cookie *)cookie)->dbg_tick;
2920}
2921
2922/*
2923 * Return the nesting level for a source cookie.
2924 */
2925 int
2926source_level(cookie)
2927 void *cookie;
2928{
2929 return ((struct source_cookie *)cookie)->level;
2930}
2931#endif
2932
2933static char_u *get_one_sourceline __ARGS((struct source_cookie *sp));
2934
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01002935#if (defined(WIN32) && defined(FEAT_CSCOPE)) || defined(HAVE_FD_CLOEXEC)
2936# define USE_FOPEN_NOINH
Bram Moolenaar071d4272004-06-13 20:20:40 +00002937static FILE *fopen_noinh_readbin __ARGS((char *filename));
2938
2939/*
2940 * Special function to open a file without handle inheritance.
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01002941 * When possible the handle is closed on exec().
Bram Moolenaar071d4272004-06-13 20:20:40 +00002942 */
2943 static FILE *
2944fopen_noinh_readbin(filename)
2945 char *filename;
2946{
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01002947# ifdef WIN32
Bram Moolenaar8d8ef0b2010-01-20 21:41:47 +01002948 int fd_tmp = mch_open(filename, O_RDONLY | O_BINARY | O_NOINHERIT, 0);
2949# else
2950 int fd_tmp = mch_open(filename, O_RDONLY, 0);
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01002951# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002952
2953 if (fd_tmp == -1)
2954 return NULL;
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01002955
2956# ifdef HAVE_FD_CLOEXEC
2957 {
2958 int fdflags = fcntl(fd_tmp, F_GETFD);
2959 if (fdflags >= 0 && (fdflags & FD_CLOEXEC) == 0)
2960 fcntl(fd_tmp, F_SETFD, fdflags | FD_CLOEXEC);
2961 }
2962# endif
2963
Bram Moolenaar071d4272004-06-13 20:20:40 +00002964 return fdopen(fd_tmp, READBIN);
2965}
2966#endif
2967
2968
2969/*
2970 * do_source: Read the file "fname" and execute its lines as EX commands.
2971 *
2972 * This function may be called recursively!
2973 *
2974 * return FAIL if file could not be opened, OK otherwise
2975 */
2976 int
2977do_source(fname, check_other, is_vimrc)
2978 char_u *fname;
2979 int check_other; /* check for .vimrc and _vimrc */
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002980 int is_vimrc; /* DOSO_ value */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002981{
2982 struct source_cookie cookie;
2983 char_u *save_sourcing_name;
2984 linenr_T save_sourcing_lnum;
2985 char_u *p;
2986 char_u *fname_exp;
Bram Moolenaar73881402009-02-04 16:50:47 +00002987 char_u *firstline = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002988 int retval = FAIL;
2989#ifdef FEAT_EVAL
2990 scid_T save_current_SID;
2991 static scid_T last_current_SID = 0;
2992 void *save_funccalp;
2993 int save_debug_break_level = debug_break_level;
Bram Moolenaar05159a02005-02-26 23:04:13 +00002994 scriptitem_T *si = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002995# ifdef UNIX
2996 struct stat st;
2997 int stat_ok;
2998# endif
2999#endif
3000#ifdef STARTUPTIME
3001 struct timeval tv_rel;
3002 struct timeval tv_start;
3003#endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00003004#ifdef FEAT_PROFILE
3005 proftime_T wait_start;
3006#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003007
Bram Moolenaar071d4272004-06-13 20:20:40 +00003008 p = expand_env_save(fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003009 if (p == NULL)
3010 return retval;
3011 fname_exp = fix_fname(p);
3012 vim_free(p);
3013 if (fname_exp == NULL)
3014 return retval;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003015 if (mch_isdir(fname_exp))
3016 {
Bram Moolenaar555b2802005-05-19 21:08:39 +00003017 smsg((char_u *)_("Cannot source a directory: \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003018 goto theend;
3019 }
3020
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00003021#ifdef FEAT_AUTOCMD
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00003022 /* Apply SourceCmd autocommands, they should get the file and source it. */
3023 if (has_autocmd(EVENT_SOURCECMD, fname_exp, NULL)
3024 && apply_autocmds(EVENT_SOURCECMD, fname_exp, fname_exp,
3025 FALSE, curbuf))
Bram Moolenaarf33943e2008-01-15 21:17:30 +00003026 {
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00003027# ifdef FEAT_EVAL
Bram Moolenaarf33943e2008-01-15 21:17:30 +00003028 retval = aborting() ? FAIL : OK;
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00003029# else
Bram Moolenaarf33943e2008-01-15 21:17:30 +00003030 retval = OK;
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00003031# endif
Bram Moolenaarf33943e2008-01-15 21:17:30 +00003032 goto theend;
3033 }
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00003034
3035 /* Apply SourcePre autocommands, they may get the file. */
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00003036 apply_autocmds(EVENT_SOURCEPRE, fname_exp, fname_exp, FALSE, curbuf);
3037#endif
3038
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01003039#ifdef USE_FOPEN_NOINH
Bram Moolenaar071d4272004-06-13 20:20:40 +00003040 cookie.fp = fopen_noinh_readbin((char *)fname_exp);
3041#else
3042 cookie.fp = mch_fopen((char *)fname_exp, READBIN);
3043#endif
3044 if (cookie.fp == NULL && check_other)
3045 {
3046 /*
3047 * Try again, replacing file name ".vimrc" by "_vimrc" or vice versa,
3048 * and ".exrc" by "_exrc" or vice versa.
3049 */
3050 p = gettail(fname_exp);
3051 if ((*p == '.' || *p == '_')
3052 && (STRICMP(p + 1, "vimrc") == 0
3053 || STRICMP(p + 1, "gvimrc") == 0
3054 || STRICMP(p + 1, "exrc") == 0))
3055 {
3056 if (*p == '_')
3057 *p = '.';
3058 else
3059 *p = '_';
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01003060#ifdef USE_FOPEN_NOINH
Bram Moolenaar071d4272004-06-13 20:20:40 +00003061 cookie.fp = fopen_noinh_readbin((char *)fname_exp);
3062#else
3063 cookie.fp = mch_fopen((char *)fname_exp, READBIN);
3064#endif
3065 }
3066 }
3067
3068 if (cookie.fp == NULL)
3069 {
3070 if (p_verbose > 0)
3071 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003072 verbose_enter();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003073 if (sourcing_name == NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00003074 smsg((char_u *)_("could not source \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003075 else
3076 smsg((char_u *)_("line %ld: could not source \"%s\""),
Bram Moolenaar555b2802005-05-19 21:08:39 +00003077 sourcing_lnum, fname);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003078 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003079 }
3080 goto theend;
3081 }
3082
3083 /*
3084 * The file exists.
3085 * - In verbose mode, give a message.
3086 * - For a vimrc file, may want to set 'compatible', call vimrc_found().
3087 */
3088 if (p_verbose > 1)
3089 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003090 verbose_enter();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003091 if (sourcing_name == NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00003092 smsg((char_u *)_("sourcing \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003093 else
3094 smsg((char_u *)_("line %ld: sourcing \"%s\""),
Bram Moolenaar555b2802005-05-19 21:08:39 +00003095 sourcing_lnum, fname);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003096 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003097 }
Bram Moolenaar910f66f2006-04-05 20:41:53 +00003098 if (is_vimrc == DOSO_VIMRC)
3099 vimrc_found(fname_exp, (char_u *)"MYVIMRC");
3100 else if (is_vimrc == DOSO_GVIMRC)
3101 vimrc_found(fname_exp, (char_u *)"MYGVIMRC");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003102
3103#ifdef USE_CRNL
3104 /* If no automatic file format: Set default to CR-NL. */
3105 if (*p_ffs == NUL)
3106 cookie.fileformat = EOL_DOS;
3107 else
3108 cookie.fileformat = EOL_UNKNOWN;
3109 cookie.error = FALSE;
3110#endif
3111
3112#ifdef USE_CR
3113 /* If no automatic file format: Set default to CR. */
3114 if (*p_ffs == NUL)
3115 cookie.fileformat = EOL_MAC;
3116 else
3117 cookie.fileformat = EOL_UNKNOWN;
3118 cookie.error = FALSE;
3119#endif
3120
3121 cookie.nextline = NULL;
3122 cookie.finished = FALSE;
3123
3124#ifdef FEAT_EVAL
3125 /*
3126 * Check if this script has a breakpoint.
3127 */
3128 cookie.breakpoint = dbg_find_breakpoint(TRUE, fname_exp, (linenr_T)0);
3129 cookie.fname = fname_exp;
3130 cookie.dbg_tick = debug_tick;
3131
3132 cookie.level = ex_nesting_level;
3133#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003134
3135 /*
3136 * Keep the sourcing name/lnum, for recursive calls.
3137 */
3138 save_sourcing_name = sourcing_name;
3139 sourcing_name = fname_exp;
3140 save_sourcing_lnum = sourcing_lnum;
3141 sourcing_lnum = 0;
3142
Bram Moolenaar73881402009-02-04 16:50:47 +00003143#ifdef FEAT_MBYTE
3144 cookie.conv.vc_type = CONV_NONE; /* no conversion */
3145
3146 /* Read the first line so we can check for a UTF-8 BOM. */
3147 firstline = getsourceline(0, (void *)&cookie, 0);
3148 if (firstline != NULL && STRLEN(firstline) >= 3 && firstline[0] == 0xef
3149 && firstline[1] == 0xbb && firstline[2] == 0xbf)
3150 {
3151 /* Found BOM; setup conversion, skip over BOM and recode the line. */
3152 convert_setup(&cookie.conv, (char_u *)"utf-8", p_enc);
3153 p = string_convert(&cookie.conv, firstline + 3, NULL);
Bram Moolenaar4e2a5952009-02-05 19:48:25 +00003154 if (p == NULL)
3155 p = vim_strsave(firstline + 3);
Bram Moolenaar73881402009-02-04 16:50:47 +00003156 if (p != NULL)
3157 {
3158 vim_free(firstline);
3159 firstline = p;
3160 }
3161 }
3162#endif
3163
Bram Moolenaar071d4272004-06-13 20:20:40 +00003164#ifdef STARTUPTIME
Bram Moolenaarc4e41982010-01-19 16:31:47 +01003165 if (time_fd != NULL)
3166 time_push(&tv_rel, &tv_start);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003167#endif
3168
3169#ifdef FEAT_EVAL
Bram Moolenaar05159a02005-02-26 23:04:13 +00003170# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00003171 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003172 prof_child_enter(&wait_start); /* entering a child now */
3173# endif
3174
3175 /* Don't use local function variables, if called from a function.
3176 * Also starts profiling timer for nested script. */
3177 save_funccalp = save_funccal();
3178
Bram Moolenaar071d4272004-06-13 20:20:40 +00003179 /*
3180 * Check if this script was sourced before to finds its SID.
3181 * If it's new, generate a new SID.
3182 */
3183 save_current_SID = current_SID;
3184# ifdef UNIX
3185 stat_ok = (mch_stat((char *)fname_exp, &st) >= 0);
3186# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00003187 for (current_SID = script_items.ga_len; current_SID > 0; --current_SID)
3188 {
3189 si = &SCRIPT_ITEM(current_SID);
3190 if (si->sn_name != NULL
Bram Moolenaar071d4272004-06-13 20:20:40 +00003191 && (
3192# ifdef UNIX
Bram Moolenaar7c626922005-02-07 22:01:03 +00003193 /* Compare dev/ino when possible, it catches symbolic
3194 * links. Also compare file names, the inode may change
3195 * when the file was edited. */
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00003196 ((stat_ok && si->sn_dev_valid)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003197 && (si->sn_dev == st.st_dev
3198 && si->sn_ino == st.st_ino)) ||
Bram Moolenaar071d4272004-06-13 20:20:40 +00003199# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00003200 fnamecmp(si->sn_name, fname_exp) == 0))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003201 break;
Bram Moolenaar05159a02005-02-26 23:04:13 +00003202 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003203 if (current_SID == 0)
3204 {
3205 current_SID = ++last_current_SID;
Bram Moolenaar05159a02005-02-26 23:04:13 +00003206 if (ga_grow(&script_items, (int)(current_SID - script_items.ga_len))
3207 == FAIL)
3208 goto almosttheend;
3209 while (script_items.ga_len < current_SID)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003210 {
Bram Moolenaar05159a02005-02-26 23:04:13 +00003211 ++script_items.ga_len;
3212 SCRIPT_ITEM(script_items.ga_len).sn_name = NULL;
3213# ifdef FEAT_PROFILE
3214 SCRIPT_ITEM(script_items.ga_len).sn_prof_on = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003215# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003216 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00003217 si = &SCRIPT_ITEM(current_SID);
3218 si->sn_name = fname_exp;
3219 fname_exp = NULL;
3220# ifdef UNIX
3221 if (stat_ok)
3222 {
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00003223 si->sn_dev_valid = TRUE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00003224 si->sn_dev = st.st_dev;
3225 si->sn_ino = st.st_ino;
3226 }
3227 else
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00003228 si->sn_dev_valid = FALSE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00003229# endif
3230
Bram Moolenaar071d4272004-06-13 20:20:40 +00003231 /* Allocate the local script variables to use for this script. */
3232 new_script_vars(current_SID);
3233 }
3234
Bram Moolenaar05159a02005-02-26 23:04:13 +00003235# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00003236 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003237 {
3238 int forceit;
3239
3240 /* Check if we do profiling for this script. */
3241 if (!si->sn_prof_on && has_profiling(TRUE, si->sn_name, &forceit))
3242 {
3243 script_do_profile(si);
3244 si->sn_pr_force = forceit;
3245 }
3246 if (si->sn_prof_on)
3247 {
3248 ++si->sn_pr_count;
3249 profile_start(&si->sn_pr_start);
3250 profile_zero(&si->sn_pr_children);
3251 }
3252 }
3253# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003254#endif
3255
3256 /*
3257 * Call do_cmdline, which will call getsourceline() to get the lines.
3258 */
Bram Moolenaar73881402009-02-04 16:50:47 +00003259 do_cmdline(firstline, getsourceline, (void *)&cookie,
Bram Moolenaar071d4272004-06-13 20:20:40 +00003260 DOCMD_VERBOSE|DOCMD_NOWAIT|DOCMD_REPEAT);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003261 retval = OK;
Bram Moolenaar05159a02005-02-26 23:04:13 +00003262
3263#ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00003264 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003265 {
3266 /* Get "si" again, "script_items" may have been reallocated. */
3267 si = &SCRIPT_ITEM(current_SID);
3268 if (si->sn_prof_on)
3269 {
3270 profile_end(&si->sn_pr_start);
3271 profile_sub_wait(&wait_start, &si->sn_pr_start);
3272 profile_add(&si->sn_pr_total, &si->sn_pr_start);
Bram Moolenaar1056d982006-03-09 22:37:52 +00003273 profile_self(&si->sn_pr_self, &si->sn_pr_start,
3274 &si->sn_pr_children);
Bram Moolenaar05159a02005-02-26 23:04:13 +00003275 }
3276 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003277#endif
3278
3279 if (got_int)
3280 EMSG(_(e_interr));
3281 sourcing_name = save_sourcing_name;
3282 sourcing_lnum = save_sourcing_lnum;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003283 if (p_verbose > 1)
3284 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003285 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00003286 smsg((char_u *)_("finished sourcing %s"), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003287 if (sourcing_name != NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00003288 smsg((char_u *)_("continuing in %s"), sourcing_name);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003289 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003290 }
3291#ifdef STARTUPTIME
Bram Moolenaarc4e41982010-01-19 16:31:47 +01003292 if (time_fd != NULL)
3293 {
3294 vim_snprintf((char *)IObuff, IOSIZE, "sourcing %s", fname);
3295 time_msg((char *)IObuff, &tv_start);
3296 time_pop(&tv_rel);
3297 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003298#endif
3299
3300#ifdef FEAT_EVAL
3301 /*
3302 * After a "finish" in debug mode, need to break at first command of next
3303 * sourced file.
3304 */
3305 if (save_debug_break_level > ex_nesting_level
3306 && debug_break_level == ex_nesting_level)
3307 ++debug_break_level;
3308#endif
3309
Bram Moolenaar05159a02005-02-26 23:04:13 +00003310#ifdef FEAT_EVAL
3311almosttheend:
3312 current_SID = save_current_SID;
3313 restore_funccal(save_funccalp);
3314# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00003315 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003316 prof_child_exit(&wait_start); /* leaving a child now */
3317# endif
3318#endif
3319 fclose(cookie.fp);
3320 vim_free(cookie.nextline);
Bram Moolenaar73881402009-02-04 16:50:47 +00003321 vim_free(firstline);
Bram Moolenaar05159a02005-02-26 23:04:13 +00003322#ifdef FEAT_MBYTE
3323 convert_setup(&cookie.conv, NULL, NULL);
3324#endif
3325
Bram Moolenaar071d4272004-06-13 20:20:40 +00003326theend:
3327 vim_free(fname_exp);
3328 return retval;
3329}
3330
3331#if defined(FEAT_EVAL) || defined(PROTO)
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00003332
Bram Moolenaar071d4272004-06-13 20:20:40 +00003333/*
3334 * ":scriptnames"
3335 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003336 void
3337ex_scriptnames(eap)
Bram Moolenaar0c094b92009-05-14 20:20:33 +00003338 exarg_T *eap UNUSED;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003339{
3340 int i;
3341
Bram Moolenaar05159a02005-02-26 23:04:13 +00003342 for (i = 1; i <= script_items.ga_len && !got_int; ++i)
3343 if (SCRIPT_ITEM(i).sn_name != NULL)
Bram Moolenaard58ea072011-06-26 04:25:30 +02003344 {
3345 home_replace(NULL, SCRIPT_ITEM(i).sn_name,
3346 NameBuff, MAXPATHL, TRUE);
3347 smsg((char_u *)"%3d: %s", i, NameBuff);
Bram Moolenaar970a1b82012-03-23 18:39:18 +01003348 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003349}
3350
3351# if defined(BACKSLASH_IN_FILENAME) || defined(PROTO)
3352/*
3353 * Fix slashes in the list of script names for 'shellslash'.
3354 */
3355 void
3356scriptnames_slash_adjust()
3357{
3358 int i;
3359
Bram Moolenaar05159a02005-02-26 23:04:13 +00003360 for (i = 1; i <= script_items.ga_len; ++i)
3361 if (SCRIPT_ITEM(i).sn_name != NULL)
3362 slash_adjust(SCRIPT_ITEM(i).sn_name);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003363}
3364# endif
3365
3366/*
3367 * Get a pointer to a script name. Used for ":verbose set".
3368 */
3369 char_u *
3370get_scriptname(id)
3371 scid_T id;
3372{
3373 if (id == SID_MODELINE)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00003374 return (char_u *)_("modeline");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003375 if (id == SID_CMDARG)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00003376 return (char_u *)_("--cmd argument");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003377 if (id == SID_CARG)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00003378 return (char_u *)_("-c argument");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003379 if (id == SID_ENV)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00003380 return (char_u *)_("environment variable");
3381 if (id == SID_ERROR)
3382 return (char_u *)_("error handler");
Bram Moolenaar05159a02005-02-26 23:04:13 +00003383 return SCRIPT_ITEM(id).sn_name;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003384}
Bram Moolenaar05159a02005-02-26 23:04:13 +00003385
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00003386# if defined(EXITFREE) || defined(PROTO)
3387 void
3388free_scriptnames()
3389{
3390 int i;
3391
3392 for (i = script_items.ga_len; i > 0; --i)
3393 vim_free(SCRIPT_ITEM(i).sn_name);
3394 ga_clear(&script_items);
3395}
3396# endif
3397
Bram Moolenaar071d4272004-06-13 20:20:40 +00003398#endif
3399
3400#if defined(USE_CR) || defined(PROTO)
3401
3402# if defined(__MSL__) && (__MSL__ >= 22)
3403/*
3404 * Newer version of the Metrowerks library handle DOS and UNIX files
3405 * without help.
3406 * Test with earlier versions, MSL 2.2 is the library supplied with
3407 * Codewarrior Pro 2.
3408 */
3409 char *
3410fgets_cr(s, n, stream)
3411 char *s;
3412 int n;
3413 FILE *stream;
3414{
3415 return fgets(s, n, stream);
3416}
3417# else
3418/*
3419 * Version of fgets() which also works for lines ending in a <CR> only
3420 * (Macintosh format).
3421 * For older versions of the Metrowerks library.
3422 * At least CodeWarrior 9 needed this code.
3423 */
3424 char *
3425fgets_cr(s, n, stream)
3426 char *s;
3427 int n;
3428 FILE *stream;
3429{
3430 int c = 0;
3431 int char_read = 0;
3432
3433 while (!feof(stream) && c != '\r' && c != '\n' && char_read < n - 1)
3434 {
3435 c = fgetc(stream);
3436 s[char_read++] = c;
3437 /* If the file is in DOS format, we need to skip a NL after a CR. I
3438 * thought it was the other way around, but this appears to work... */
3439 if (c == '\n')
3440 {
3441 c = fgetc(stream);
3442 if (c != '\r')
3443 ungetc(c, stream);
3444 }
3445 }
3446
3447 s[char_read] = 0;
3448 if (char_read == 0)
3449 return NULL;
3450
3451 if (feof(stream) && char_read == 1)
3452 return NULL;
3453
3454 return s;
3455}
3456# endif
3457#endif
3458
3459/*
3460 * Get one full line from a sourced file.
3461 * Called by do_cmdline() when it's called from do_source().
3462 *
3463 * Return a pointer to the line in allocated memory.
3464 * Return NULL for end-of-file or some error.
3465 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003466 char_u *
3467getsourceline(c, cookie, indent)
Bram Moolenaar0c094b92009-05-14 20:20:33 +00003468 int c UNUSED;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003469 void *cookie;
Bram Moolenaar0c094b92009-05-14 20:20:33 +00003470 int indent UNUSED;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003471{
3472 struct source_cookie *sp = (struct source_cookie *)cookie;
3473 char_u *line;
Bram Moolenaarc047b9a2012-02-11 20:40:55 +01003474 char_u *p;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003475
3476#ifdef FEAT_EVAL
3477 /* If breakpoints have been added/deleted need to check for it. */
3478 if (sp->dbg_tick < debug_tick)
3479 {
3480 sp->breakpoint = dbg_find_breakpoint(TRUE, sp->fname, sourcing_lnum);
3481 sp->dbg_tick = debug_tick;
3482 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00003483# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00003484 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003485 script_line_end();
3486# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003487#endif
3488 /*
3489 * Get current line. If there is a read-ahead line, use it, otherwise get
3490 * one now.
3491 */
3492 if (sp->finished)
3493 line = NULL;
3494 else if (sp->nextline == NULL)
3495 line = get_one_sourceline(sp);
3496 else
3497 {
3498 line = sp->nextline;
3499 sp->nextline = NULL;
3500 ++sourcing_lnum;
3501 }
Bram Moolenaare2cc9702005-03-15 22:43:58 +00003502#ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00003503 if (line != NULL && do_profiling == PROF_YES)
Bram Moolenaare2cc9702005-03-15 22:43:58 +00003504 script_line_start();
3505#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003506
3507 /* Only concatenate lines starting with a \ when 'cpoptions' doesn't
3508 * contain the 'C' flag. */
3509 if (line != NULL && (vim_strchr(p_cpo, CPO_CONCAT) == NULL))
3510 {
3511 /* compensate for the one line read-ahead */
3512 --sourcing_lnum;
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01003513
3514 /* Get the next line and concatenate it when it starts with a
3515 * backslash. We always need to read the next line, keep it in
3516 * sp->nextline. */
3517 sp->nextline = get_one_sourceline(sp);
3518 if (sp->nextline != NULL && *(p = skipwhite(sp->nextline)) == '\\')
Bram Moolenaar071d4272004-06-13 20:20:40 +00003519 {
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01003520 garray_T ga;
3521
Bram Moolenaarb549a732012-02-22 18:29:33 +01003522 ga_init2(&ga, (int)sizeof(char_u), 400);
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01003523 ga_concat(&ga, line);
3524 ga_concat(&ga, p + 1);
3525 for (;;)
3526 {
3527 vim_free(sp->nextline);
3528 sp->nextline = get_one_sourceline(sp);
3529 if (sp->nextline == NULL)
3530 break;
3531 p = skipwhite(sp->nextline);
3532 if (*p != '\\')
3533 break;
Bram Moolenaarb549a732012-02-22 18:29:33 +01003534 /* Adjust the growsize to the current length to speed up
3535 * concatenating many lines. */
3536 if (ga.ga_len > 400)
3537 {
3538 if (ga.ga_len > 8000)
3539 ga.ga_growsize = 8000;
3540 else
3541 ga.ga_growsize = ga.ga_len;
3542 }
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01003543 ga_concat(&ga, p + 1);
3544 }
3545 ga_append(&ga, NUL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003546 vim_free(line);
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01003547 line = ga.ga_data;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003548 }
3549 }
3550
3551#ifdef FEAT_MBYTE
3552 if (line != NULL && sp->conv.vc_type != CONV_NONE)
3553 {
Bram Moolenaarc047b9a2012-02-11 20:40:55 +01003554 char_u *s;
3555
Bram Moolenaar071d4272004-06-13 20:20:40 +00003556 /* Convert the encoding of the script line. */
3557 s = string_convert(&sp->conv, line, NULL);
3558 if (s != NULL)
3559 {
3560 vim_free(line);
3561 line = s;
3562 }
3563 }
3564#endif
3565
3566#ifdef FEAT_EVAL
3567 /* Did we encounter a breakpoint? */
3568 if (sp->breakpoint != 0 && sp->breakpoint <= sourcing_lnum)
3569 {
3570 dbg_breakpoint(sp->fname, sourcing_lnum);
3571 /* Find next breakpoint. */
3572 sp->breakpoint = dbg_find_breakpoint(TRUE, sp->fname, sourcing_lnum);
3573 sp->dbg_tick = debug_tick;
3574 }
3575#endif
3576
3577 return line;
3578}
3579
3580 static char_u *
3581get_one_sourceline(sp)
3582 struct source_cookie *sp;
3583{
3584 garray_T ga;
3585 int len;
3586 int c;
3587 char_u *buf;
3588#ifdef USE_CRNL
3589 int has_cr; /* CR-LF found */
3590#endif
3591#ifdef USE_CR
3592 char_u *scan;
3593#endif
3594 int have_read = FALSE;
3595
3596 /* use a growarray to store the sourced line */
Bram Moolenaar6ac54292005-02-02 23:07:25 +00003597 ga_init2(&ga, 1, 250);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003598
3599 /*
3600 * Loop until there is a finished line (or end-of-file).
3601 */
3602 sourcing_lnum++;
3603 for (;;)
3604 {
Bram Moolenaar6ac54292005-02-02 23:07:25 +00003605 /* make room to read at least 120 (more) characters */
3606 if (ga_grow(&ga, 120) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003607 break;
3608 buf = (char_u *)ga.ga_data;
3609
3610#ifdef USE_CR
3611 if (sp->fileformat == EOL_MAC)
3612 {
Bram Moolenaar86b68352004-12-27 21:59:20 +00003613 if (fgets_cr((char *)buf + ga.ga_len, ga.ga_maxlen - ga.ga_len,
3614 sp->fp) == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003615 break;
3616 }
3617 else
3618#endif
Bram Moolenaar86b68352004-12-27 21:59:20 +00003619 if (fgets((char *)buf + ga.ga_len, ga.ga_maxlen - ga.ga_len,
3620 sp->fp) == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003621 break;
Bram Moolenaar6ac54292005-02-02 23:07:25 +00003622 len = ga.ga_len + (int)STRLEN(buf + ga.ga_len);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003623#ifdef USE_CRNL
3624 /* Ignore a trailing CTRL-Z, when in Dos mode. Only recognize the
3625 * CTRL-Z by its own, or after a NL. */
3626 if ( (len == 1 || (len >= 2 && buf[len - 2] == '\n'))
3627 && sp->fileformat == EOL_DOS
3628 && buf[len - 1] == Ctrl_Z)
3629 {
3630 buf[len - 1] = NUL;
3631 break;
3632 }
3633#endif
3634
3635#ifdef USE_CR
3636 /* If the read doesn't stop on a new line, and there's
3637 * some CR then we assume a Mac format */
3638 if (sp->fileformat == EOL_UNKNOWN)
3639 {
3640 if (buf[len - 1] != '\n' && vim_strchr(buf, '\r') != NULL)
3641 sp->fileformat = EOL_MAC;
3642 else
3643 sp->fileformat = EOL_UNIX;
3644 }
3645
3646 if (sp->fileformat == EOL_MAC)
3647 {
3648 scan = vim_strchr(buf, '\r');
3649
3650 if (scan != NULL)
3651 {
3652 *scan = '\n';
3653 if (*(scan + 1) != 0)
3654 {
3655 *(scan + 1) = 0;
3656 fseek(sp->fp, (long)(scan - buf - len + 1), SEEK_CUR);
3657 }
3658 }
3659 len = STRLEN(buf);
3660 }
3661#endif
3662
3663 have_read = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003664 ga.ga_len = len;
3665
3666 /* If the line was longer than the buffer, read more. */
Bram Moolenaar86b68352004-12-27 21:59:20 +00003667 if (ga.ga_maxlen - ga.ga_len == 1 && buf[len - 1] != '\n')
Bram Moolenaar071d4272004-06-13 20:20:40 +00003668 continue;
3669
3670 if (len >= 1 && buf[len - 1] == '\n') /* remove trailing NL */
3671 {
3672#ifdef USE_CRNL
3673 has_cr = (len >= 2 && buf[len - 2] == '\r');
3674 if (sp->fileformat == EOL_UNKNOWN)
3675 {
3676 if (has_cr)
3677 sp->fileformat = EOL_DOS;
3678 else
3679 sp->fileformat = EOL_UNIX;
3680 }
3681
3682 if (sp->fileformat == EOL_DOS)
3683 {
3684 if (has_cr) /* replace trailing CR */
3685 {
3686 buf[len - 2] = '\n';
3687 --len;
3688 --ga.ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003689 }
3690 else /* lines like ":map xx yy^M" will have failed */
3691 {
3692 if (!sp->error)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00003693 {
3694 msg_source(hl_attr(HLF_W));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003695 EMSG(_("W15: Warning: Wrong line separator, ^M may be missing"));
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00003696 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003697 sp->error = TRUE;
3698 sp->fileformat = EOL_UNIX;
3699 }
3700 }
3701#endif
3702 /* The '\n' is escaped if there is an odd number of ^V's just
3703 * before it, first set "c" just before the 'V's and then check
3704 * len&c parities (is faster than ((len-c)%2 == 0)) -- Acevedo */
3705 for (c = len - 2; c >= 0 && buf[c] == Ctrl_V; c--)
3706 ;
3707 if ((len & 1) != (c & 1)) /* escaped NL, read more */
3708 {
3709 sourcing_lnum++;
3710 continue;
3711 }
3712
3713 buf[len - 1] = NUL; /* remove the NL */
3714 }
3715
3716 /*
3717 * Check for ^C here now and then, so recursive :so can be broken.
3718 */
3719 line_breakcheck();
3720 break;
3721 }
3722
3723 if (have_read)
3724 return (char_u *)ga.ga_data;
3725
3726 vim_free(ga.ga_data);
3727 return NULL;
3728}
3729
Bram Moolenaar05159a02005-02-26 23:04:13 +00003730#if defined(FEAT_PROFILE) || defined(PROTO)
3731/*
3732 * Called when starting to read a script line.
3733 * "sourcing_lnum" must be correct!
3734 * When skipping lines it may not actually be executed, but we won't find out
3735 * until later and we need to store the time now.
3736 */
3737 void
3738script_line_start()
3739{
3740 scriptitem_T *si;
3741 sn_prl_T *pp;
3742
3743 if (current_SID <= 0 || current_SID > script_items.ga_len)
3744 return;
3745 si = &SCRIPT_ITEM(current_SID);
3746 if (si->sn_prof_on && sourcing_lnum >= 1)
3747 {
Bram Moolenaar8c8de832008-06-24 22:58:06 +00003748 /* Grow the array before starting the timer, so that the time spent
Bram Moolenaar05159a02005-02-26 23:04:13 +00003749 * here isn't counted. */
3750 ga_grow(&si->sn_prl_ga, (int)(sourcing_lnum - si->sn_prl_ga.ga_len));
3751 si->sn_prl_idx = sourcing_lnum - 1;
3752 while (si->sn_prl_ga.ga_len <= si->sn_prl_idx
3753 && si->sn_prl_ga.ga_len < si->sn_prl_ga.ga_maxlen)
3754 {
3755 /* Zero counters for a line that was not used before. */
3756 pp = &PRL_ITEM(si, si->sn_prl_ga.ga_len);
3757 pp->snp_count = 0;
3758 profile_zero(&pp->sn_prl_total);
3759 profile_zero(&pp->sn_prl_self);
3760 ++si->sn_prl_ga.ga_len;
3761 }
3762 si->sn_prl_execed = FALSE;
3763 profile_start(&si->sn_prl_start);
3764 profile_zero(&si->sn_prl_children);
3765 profile_get_wait(&si->sn_prl_wait);
3766 }
3767}
3768
3769/*
3770 * Called when actually executing a function line.
3771 */
3772 void
3773script_line_exec()
3774{
3775 scriptitem_T *si;
3776
3777 if (current_SID <= 0 || current_SID > script_items.ga_len)
3778 return;
3779 si = &SCRIPT_ITEM(current_SID);
3780 if (si->sn_prof_on && si->sn_prl_idx >= 0)
3781 si->sn_prl_execed = TRUE;
3782}
3783
3784/*
3785 * Called when done with a function line.
3786 */
3787 void
3788script_line_end()
3789{
3790 scriptitem_T *si;
3791 sn_prl_T *pp;
3792
3793 if (current_SID <= 0 || current_SID > script_items.ga_len)
3794 return;
3795 si = &SCRIPT_ITEM(current_SID);
3796 if (si->sn_prof_on && si->sn_prl_idx >= 0
3797 && si->sn_prl_idx < si->sn_prl_ga.ga_len)
3798 {
3799 if (si->sn_prl_execed)
3800 {
3801 pp = &PRL_ITEM(si, si->sn_prl_idx);
3802 ++pp->snp_count;
3803 profile_end(&si->sn_prl_start);
3804 profile_sub_wait(&si->sn_prl_wait, &si->sn_prl_start);
Bram Moolenaar05159a02005-02-26 23:04:13 +00003805 profile_add(&pp->sn_prl_total, &si->sn_prl_start);
Bram Moolenaar1056d982006-03-09 22:37:52 +00003806 profile_self(&pp->sn_prl_self, &si->sn_prl_start,
3807 &si->sn_prl_children);
Bram Moolenaar05159a02005-02-26 23:04:13 +00003808 }
3809 si->sn_prl_idx = -1;
3810 }
3811}
3812#endif
3813
Bram Moolenaar071d4272004-06-13 20:20:40 +00003814/*
3815 * ":scriptencoding": Set encoding conversion for a sourced script.
3816 * Without the multi-byte feature it's simply ignored.
3817 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003818 void
3819ex_scriptencoding(eap)
Bram Moolenaar0c094b92009-05-14 20:20:33 +00003820 exarg_T *eap UNUSED;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003821{
3822#ifdef FEAT_MBYTE
3823 struct source_cookie *sp;
3824 char_u *name;
3825
3826 if (!getline_equal(eap->getline, eap->cookie, getsourceline))
3827 {
3828 EMSG(_("E167: :scriptencoding used outside of a sourced file"));
3829 return;
3830 }
3831
3832 if (*eap->arg != NUL)
3833 {
3834 name = enc_canonize(eap->arg);
3835 if (name == NULL) /* out of memory */
3836 return;
3837 }
3838 else
3839 name = eap->arg;
3840
3841 /* Setup for conversion from the specified encoding to 'encoding'. */
3842 sp = (struct source_cookie *)getline_cookie(eap->getline, eap->cookie);
3843 convert_setup(&sp->conv, name, p_enc);
3844
3845 if (name != eap->arg)
3846 vim_free(name);
3847#endif
3848}
3849
3850#if defined(FEAT_EVAL) || defined(PROTO)
3851/*
3852 * ":finish": Mark a sourced file as finished.
3853 */
3854 void
3855ex_finish(eap)
3856 exarg_T *eap;
3857{
3858 if (getline_equal(eap->getline, eap->cookie, getsourceline))
3859 do_finish(eap, FALSE);
3860 else
3861 EMSG(_("E168: :finish used outside of a sourced file"));
3862}
3863
3864/*
3865 * Mark a sourced file as finished. Possibly makes the ":finish" pending.
3866 * Also called for a pending finish at the ":endtry" or after returning from
3867 * an extra do_cmdline(). "reanimate" is used in the latter case.
3868 */
3869 void
3870do_finish(eap, reanimate)
3871 exarg_T *eap;
3872 int reanimate;
3873{
3874 int idx;
3875
3876 if (reanimate)
3877 ((struct source_cookie *)getline_cookie(eap->getline,
3878 eap->cookie))->finished = FALSE;
3879
3880 /*
3881 * Cleanup (and inactivate) conditionals, but stop when a try conditional
3882 * not in its finally clause (which then is to be executed next) is found.
3883 * In this case, make the ":finish" pending for execution at the ":endtry".
3884 * Otherwise, finish normally.
3885 */
3886 idx = cleanup_conditionals(eap->cstack, 0, TRUE);
3887 if (idx >= 0)
3888 {
3889 eap->cstack->cs_pending[idx] = CSTP_FINISH;
3890 report_make_pending(CSTP_FINISH, NULL);
3891 }
3892 else
3893 ((struct source_cookie *)getline_cookie(eap->getline,
3894 eap->cookie))->finished = TRUE;
3895}
3896
3897
3898/*
3899 * Return TRUE when a sourced file had the ":finish" command: Don't give error
3900 * message for missing ":endif".
3901 * Return FALSE when not sourcing a file.
3902 */
3903 int
Bram Moolenaar89d40322006-08-29 15:30:07 +00003904source_finished(fgetline, cookie)
3905 char_u *(*fgetline) __ARGS((int, void *, int));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003906 void *cookie;
3907{
Bram Moolenaar89d40322006-08-29 15:30:07 +00003908 return (getline_equal(fgetline, cookie, getsourceline)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003909 && ((struct source_cookie *)getline_cookie(
Bram Moolenaar89d40322006-08-29 15:30:07 +00003910 fgetline, cookie))->finished);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003911}
3912#endif
3913
3914#if defined(FEAT_LISTCMDS) || defined(PROTO)
3915/*
3916 * ":checktime [buffer]"
3917 */
3918 void
3919ex_checktime(eap)
3920 exarg_T *eap;
3921{
3922 buf_T *buf;
3923 int save_no_check_timestamps = no_check_timestamps;
3924
3925 no_check_timestamps = 0;
3926 if (eap->addr_count == 0) /* default is all buffers */
3927 check_timestamps(FALSE);
3928 else
3929 {
3930 buf = buflist_findnr((int)eap->line2);
3931 if (buf != NULL) /* cannot happen? */
3932 (void)buf_check_timestamp(buf, FALSE);
3933 }
3934 no_check_timestamps = save_no_check_timestamps;
3935}
3936#endif
3937
Bram Moolenaar071d4272004-06-13 20:20:40 +00003938#if (defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
3939 && (defined(FEAT_EVAL) || defined(FEAT_MULTI_LANG))
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02003940# define HAVE_GET_LOCALE_VAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00003941static char *get_locale_val __ARGS((int what));
3942
3943 static char *
3944get_locale_val(what)
3945 int what;
3946{
3947 char *loc;
3948
3949 /* Obtain the locale value from the libraries. For DJGPP this is
3950 * redefined and it doesn't use the arguments. */
3951 loc = setlocale(what, NULL);
3952
Bram Moolenaar61660ea2006-04-07 21:40:07 +00003953# ifdef WIN32
Bram Moolenaar071d4272004-06-13 20:20:40 +00003954 if (loc != NULL)
3955 {
3956 char_u *p;
3957
Bram Moolenaar61660ea2006-04-07 21:40:07 +00003958 /* setocale() returns something like "LC_COLLATE=<name>;LC_..." when
3959 * one of the values (e.g., LC_CTYPE) differs. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003960 p = vim_strchr(loc, '=');
3961 if (p != NULL)
3962 {
3963 loc = ++p;
3964 while (*p != NUL) /* remove trailing newline */
3965 {
Bram Moolenaar61660ea2006-04-07 21:40:07 +00003966 if (*p < ' ' || *p == ';')
Bram Moolenaar071d4272004-06-13 20:20:40 +00003967 {
3968 *p = NUL;
3969 break;
3970 }
3971 ++p;
3972 }
3973 }
3974 }
3975# endif
3976
3977 return loc;
3978}
3979#endif
3980
3981
3982#ifdef WIN32
3983/*
3984 * On MS-Windows locale names are strings like "German_Germany.1252", but
3985 * gettext expects "de". Try to translate one into another here for a few
3986 * supported languages.
3987 */
3988 static char_u *
3989gettext_lang(char_u *name)
3990{
3991 int i;
3992 static char *(mtable[]) = {
3993 "afrikaans", "af",
3994 "czech", "cs",
3995 "dutch", "nl",
3996 "german", "de",
3997 "english_united kingdom", "en_GB",
3998 "spanish", "es",
3999 "french", "fr",
4000 "italian", "it",
4001 "japanese", "ja",
4002 "korean", "ko",
4003 "norwegian", "no",
4004 "polish", "pl",
4005 "russian", "ru",
4006 "slovak", "sk",
4007 "swedish", "sv",
4008 "ukrainian", "uk",
4009 "chinese_china", "zh_CN",
4010 "chinese_taiwan", "zh_TW",
4011 NULL};
4012
4013 for (i = 0; mtable[i] != NULL; i += 2)
4014 if (STRNICMP(mtable[i], name, STRLEN(mtable[i])) == 0)
4015 return mtable[i + 1];
4016 return name;
4017}
4018#endif
4019
4020#if defined(FEAT_MULTI_LANG) || defined(PROTO)
4021/*
4022 * Obtain the current messages language. Used to set the default for
4023 * 'helplang'. May return NULL or an empty string.
4024 */
4025 char_u *
4026get_mess_lang()
4027{
4028 char_u *p;
4029
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004030# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00004031# if defined(LC_MESSAGES)
4032 p = (char_u *)get_locale_val(LC_MESSAGES);
4033# else
4034 /* This is necessary for Win32, where LC_MESSAGES is not defined and $LANG
Bram Moolenaar61660ea2006-04-07 21:40:07 +00004035 * may be set to the LCID number. LC_COLLATE is the best guess, LC_TIME
4036 * and LC_MONETARY may be set differently for a Japanese working in the
4037 * US. */
4038 p = (char_u *)get_locale_val(LC_COLLATE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004039# endif
4040# else
4041 p = mch_getenv((char_u *)"LC_ALL");
4042 if (p == NULL || *p == NUL)
4043 {
4044 p = mch_getenv((char_u *)"LC_MESSAGES");
4045 if (p == NULL || *p == NUL)
4046 p = mch_getenv((char_u *)"LANG");
4047 }
4048# endif
4049# ifdef WIN32
4050 p = gettext_lang(p);
4051# endif
4052 return p;
4053}
4054#endif
4055
Bram Moolenaardef9e822004-12-31 20:58:58 +00004056/* Complicated #if; matches with where get_mess_env() is used below. */
4057#if (defined(FEAT_EVAL) && !((defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
4058 && defined(LC_MESSAGES))) \
4059 || ((defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
4060 && (defined(FEAT_GETTEXT) || defined(FEAT_MBYTE)) \
4061 && !defined(LC_MESSAGES))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004062static char_u *get_mess_env __ARGS((void));
4063
4064/*
4065 * Get the language used for messages from the environment.
4066 */
4067 static char_u *
4068get_mess_env()
4069{
4070 char_u *p;
4071
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 {
4078 p = mch_getenv((char_u *)"LANG");
4079 if (p != NULL && VIM_ISDIGIT(*p))
4080 p = NULL; /* ignore something like "1043" */
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004081# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00004082 if (p == NULL || *p == NUL)
4083 p = (char_u *)get_locale_val(LC_CTYPE);
4084# endif
4085 }
4086 }
4087 return p;
4088}
4089#endif
4090
4091#if defined(FEAT_EVAL) || defined(PROTO)
4092
4093/*
4094 * Set the "v:lang" variable according to the current locale setting.
4095 * Also do "v:lc_time"and "v:ctype".
4096 */
4097 void
4098set_lang_var()
4099{
4100 char_u *loc;
4101
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004102# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00004103 loc = (char_u *)get_locale_val(LC_CTYPE);
4104# else
4105 /* setlocale() not supported: use the default value */
4106 loc = (char_u *)"C";
4107# endif
4108 set_vim_var_string(VV_CTYPE, loc, -1);
4109
4110 /* When LC_MESSAGES isn't defined use the value from $LC_MESSAGES, fall
4111 * back to LC_CTYPE if it's empty. */
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004112# if defined(HAVE_GET_LOCALE_VAL) && defined(LC_MESSAGES)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004113 loc = (char_u *)get_locale_val(LC_MESSAGES);
4114# else
4115 loc = get_mess_env();
4116# endif
4117 set_vim_var_string(VV_LANG, loc, -1);
4118
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004119# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00004120 loc = (char_u *)get_locale_val(LC_TIME);
4121# endif
4122 set_vim_var_string(VV_LC_TIME, loc, -1);
4123}
4124#endif
4125
4126#if (defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
4127 && (defined(FEAT_GETTEXT) || defined(FEAT_MBYTE))
4128/*
4129 * ":language": Set the language (locale).
4130 */
4131 void
4132ex_language(eap)
4133 exarg_T *eap;
4134{
4135 char *loc;
4136 char_u *p;
4137 char_u *name;
4138 int what = LC_ALL;
4139 char *whatstr = "";
4140#ifdef LC_MESSAGES
4141# define VIM_LC_MESSAGES LC_MESSAGES
4142#else
4143# define VIM_LC_MESSAGES 6789
4144#endif
4145
4146 name = eap->arg;
4147
4148 /* Check for "messages {name}", "ctype {name}" or "time {name}" argument.
4149 * Allow abbreviation, but require at least 3 characters to avoid
4150 * confusion with a two letter language name "me" or "ct". */
4151 p = skiptowhite(eap->arg);
4152 if ((*p == NUL || vim_iswhite(*p)) && p - eap->arg >= 3)
4153 {
4154 if (STRNICMP(eap->arg, "messages", p - eap->arg) == 0)
4155 {
4156 what = VIM_LC_MESSAGES;
4157 name = skipwhite(p);
4158 whatstr = "messages ";
4159 }
4160 else if (STRNICMP(eap->arg, "ctype", p - eap->arg) == 0)
4161 {
4162 what = LC_CTYPE;
4163 name = skipwhite(p);
4164 whatstr = "ctype ";
4165 }
4166 else if (STRNICMP(eap->arg, "time", p - eap->arg) == 0)
4167 {
4168 what = LC_TIME;
4169 name = skipwhite(p);
4170 whatstr = "time ";
4171 }
4172 }
4173
4174 if (*name == NUL)
4175 {
4176#ifndef LC_MESSAGES
4177 if (what == VIM_LC_MESSAGES)
4178 p = get_mess_env();
4179 else
4180#endif
4181 p = (char_u *)setlocale(what, NULL);
4182 if (p == NULL || *p == NUL)
4183 p = (char_u *)"Unknown";
4184 smsg((char_u *)_("Current %slanguage: \"%s\""), whatstr, p);
4185 }
4186 else
4187 {
4188#ifndef LC_MESSAGES
4189 if (what == VIM_LC_MESSAGES)
4190 loc = "";
4191 else
4192#endif
Bram Moolenaar8c8de832008-06-24 22:58:06 +00004193 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00004194 loc = setlocale(what, (char *)name);
Bram Moolenaar8c8de832008-06-24 22:58:06 +00004195#if defined(FEAT_FLOAT) && defined(LC_NUMERIC)
4196 /* Make sure strtod() uses a decimal point, not a comma. */
4197 setlocale(LC_NUMERIC, "C");
4198#endif
4199 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004200 if (loc == NULL)
4201 EMSG2(_("E197: Cannot set language to \"%s\""), name);
4202 else
4203 {
4204#ifdef HAVE_NL_MSG_CAT_CNTR
4205 /* Need to do this for GNU gettext, otherwise cached translations
4206 * will be used again. */
4207 extern int _nl_msg_cat_cntr;
4208
4209 ++_nl_msg_cat_cntr;
4210#endif
Bram Moolenaarb8017e72007-05-10 18:59:07 +00004211 /* Reset $LC_ALL, otherwise it would overrule everything. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004212 vim_setenv((char_u *)"LC_ALL", (char_u *)"");
4213
4214 if (what != LC_TIME)
4215 {
4216 /* Tell gettext() what to translate to. It apparently doesn't
4217 * use the currently effective locale. Also do this when
4218 * FEAT_GETTEXT isn't defined, so that shell commands use this
4219 * value. */
4220 if (what == LC_ALL)
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00004221 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00004222 vim_setenv((char_u *)"LANG", name);
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00004223# ifdef WIN32
4224 /* Apparently MS-Windows printf() may cause a crash when
4225 * we give it 8-bit text while it's expecting text in the
4226 * current locale. This call avoids that. */
4227 setlocale(LC_CTYPE, "C");
4228# endif
4229 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004230 if (what != LC_CTYPE)
4231 {
4232 char_u *mname;
4233#ifdef WIN32
4234 mname = gettext_lang(name);
4235#else
4236 mname = name;
4237#endif
4238 vim_setenv((char_u *)"LC_MESSAGES", mname);
4239#ifdef FEAT_MULTI_LANG
4240 set_helplang_default(mname);
4241#endif
4242 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004243 }
4244
4245# ifdef FEAT_EVAL
4246 /* Set v:lang, v:lc_time and v:ctype to the final result. */
4247 set_lang_var();
4248# endif
Bram Moolenaar6cc00c72011-10-20 21:41:09 +02004249# ifdef FEAT_TITLE
4250 maketitle();
4251# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004252 }
4253 }
4254}
4255
4256# if defined(FEAT_CMDL_COMPL) || defined(PROTO)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02004257
4258static char_u **locales = NULL; /* Array of all available locales */
4259static int did_init_locales = FALSE;
4260
4261static void init_locales __ARGS((void));
4262static char_u **find_locales __ARGS((void));
4263
4264/*
4265 * Lazy initialization of all available locales.
4266 */
4267 static void
4268init_locales()
4269{
4270 if (!did_init_locales)
4271 {
4272 did_init_locales = TRUE;
4273 locales = find_locales();
4274 }
4275}
4276
4277/* Return an array of strings for all available locales + NULL for the
4278 * last element. Return NULL in case of error. */
4279 static char_u **
4280find_locales()
4281{
4282 garray_T locales_ga;
4283 char_u *loc;
4284
4285 /* Find all available locales by running command "locale -a". If this
4286 * doesn't work we won't have completion. */
4287 char_u *locale_a = get_cmd_output((char_u *)"locale -a",
4288 NULL, SHELL_SILENT);
4289 if (locale_a == NULL)
4290 return NULL;
4291 ga_init2(&locales_ga, sizeof(char_u *), 20);
4292
4293 /* Transform locale_a string where each locale is separated by "\n"
4294 * into an array of locale strings. */
4295 loc = (char_u *)strtok((char *)locale_a, "\n");
4296
4297 while (loc != NULL)
4298 {
4299 if (ga_grow(&locales_ga, 1) == FAIL)
4300 break;
4301 loc = vim_strsave(loc);
4302 if (loc == NULL)
4303 break;
4304
4305 ((char_u **)locales_ga.ga_data)[locales_ga.ga_len++] = loc;
4306 loc = (char_u *)strtok(NULL, "\n");
4307 }
4308 vim_free(locale_a);
4309 if (ga_grow(&locales_ga, 1) == FAIL)
4310 {
4311 ga_clear(&locales_ga);
4312 return NULL;
4313 }
4314 ((char_u **)locales_ga.ga_data)[locales_ga.ga_len] = NULL;
4315 return (char_u **)locales_ga.ga_data;
4316}
4317
4318# if defined(EXITFREE) || defined(PROTO)
4319 void
4320free_locales()
4321{
4322 int i;
4323 if (locales != NULL)
4324 {
4325 for (i = 0; locales[i] != NULL; i++)
4326 vim_free(locales[i]);
4327 vim_free(locales);
4328 locales = NULL;
4329 }
4330}
4331# endif
4332
Bram Moolenaar071d4272004-06-13 20:20:40 +00004333/*
4334 * Function given to ExpandGeneric() to obtain the possible arguments of the
4335 * ":language" command.
4336 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004337 char_u *
4338get_lang_arg(xp, idx)
Bram Moolenaar0c094b92009-05-14 20:20:33 +00004339 expand_T *xp UNUSED;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004340 int idx;
4341{
4342 if (idx == 0)
4343 return (char_u *)"messages";
4344 if (idx == 1)
4345 return (char_u *)"ctype";
4346 if (idx == 2)
4347 return (char_u *)"time";
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02004348
4349 init_locales();
4350 if (locales == NULL)
4351 return NULL;
4352 return locales[idx - 3];
4353}
4354
4355/*
4356 * Function given to ExpandGeneric() to obtain the available locales.
4357 */
4358 char_u *
4359get_locales(xp, idx)
4360 expand_T *xp UNUSED;
4361 int idx;
4362{
4363 init_locales();
4364 if (locales == NULL)
4365 return NULL;
4366 return locales[idx];
Bram Moolenaar071d4272004-06-13 20:20:40 +00004367}
4368# endif
4369
4370#endif