blob: d3bf04fade022bafa189987d5f62c417e05e796b [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
Bram Moolenaar8f5c6f02012-06-29 12:57:06 +02001853 * "fnames[fcountp]". When "wig" is TRUE, removes files matching 'wildignore'.
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00001854 * Return FAIL or OK.
1855 */
1856 int
Bram Moolenaar8f5c6f02012-06-29 12:57:06 +02001857get_arglist_exp(str, fcountp, fnamesp, wig)
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00001858 char_u *str;
1859 int *fcountp;
1860 char_u ***fnamesp;
Bram Moolenaar8f5c6f02012-06-29 12:57:06 +02001861 int wig;
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00001862{
1863 garray_T ga;
1864 int i;
1865
1866 if (get_arglist(&ga, str) == FAIL)
1867 return FAIL;
Bram Moolenaar8f5c6f02012-06-29 12:57:06 +02001868 if (wig == TRUE)
1869 i = expand_wildcards(ga.ga_len, (char_u **)ga.ga_data,
1870 fcountp, fnamesp, EW_FILE|EW_NOTFOUND);
1871 else
1872 i = gen_expand_wildcards(ga.ga_len, (char_u **)ga.ga_data,
1873 fcountp, fnamesp, EW_FILE|EW_NOTFOUND);
1874
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00001875 ga_clear(&ga);
1876 return i;
1877}
1878#endif
1879
Bram Moolenaar071d4272004-06-13 20:20:40 +00001880#if defined(FEAT_GUI) || defined(FEAT_CLIENTSERVER) || defined(PROTO)
1881/*
1882 * Redefine the argument list.
1883 */
1884 void
1885set_arglist(str)
1886 char_u *str;
1887{
1888 do_arglist(str, AL_SET, 0);
1889}
1890#endif
1891
1892/*
1893 * "what" == AL_SET: Redefine the argument list to 'str'.
1894 * "what" == AL_ADD: add files in 'str' to the argument list after "after".
1895 * "what" == AL_DEL: remove files in 'str' from the argument list.
1896 *
1897 * Return FAIL for failure, OK otherwise.
1898 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001899 static int
1900do_arglist(str, what, after)
1901 char_u *str;
Bram Moolenaar0c094b92009-05-14 20:20:33 +00001902 int what UNUSED;
1903 int after UNUSED; /* 0 means before first one */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001904{
1905 garray_T new_ga;
1906 int exp_count;
1907 char_u **exp_files;
1908 int i;
1909#ifdef FEAT_LISTCMDS
1910 char_u *p;
1911 int match;
1912#endif
1913
1914 /*
1915 * Collect all file name arguments in "new_ga".
1916 */
Bram Moolenaar86b68352004-12-27 21:59:20 +00001917 if (get_arglist(&new_ga, str) == FAIL)
1918 return FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001919
1920#ifdef FEAT_LISTCMDS
1921 if (what == AL_DEL)
1922 {
1923 regmatch_T regmatch;
1924 int didone;
1925
1926 /*
1927 * Delete the items: use each item as a regexp and find a match in the
1928 * argument list.
1929 */
1930#ifdef CASE_INSENSITIVE_FILENAME
1931 regmatch.rm_ic = TRUE; /* Always ignore case */
1932#else
1933 regmatch.rm_ic = FALSE; /* Never ignore case */
1934#endif
1935 for (i = 0; i < new_ga.ga_len && !got_int; ++i)
1936 {
1937 p = ((char_u **)new_ga.ga_data)[i];
1938 p = file_pat_to_reg_pat(p, NULL, NULL, FALSE);
1939 if (p == NULL)
1940 break;
1941 regmatch.regprog = vim_regcomp(p, p_magic ? RE_MAGIC : 0);
1942 if (regmatch.regprog == NULL)
1943 {
1944 vim_free(p);
1945 break;
1946 }
1947
1948 didone = FALSE;
1949 for (match = 0; match < ARGCOUNT; ++match)
1950 if (vim_regexec(&regmatch, alist_name(&ARGLIST[match]),
1951 (colnr_T)0))
1952 {
1953 didone = TRUE;
1954 vim_free(ARGLIST[match].ae_fname);
1955 mch_memmove(ARGLIST + match, ARGLIST + match + 1,
1956 (ARGCOUNT - match - 1) * sizeof(aentry_T));
1957 --ALIST(curwin)->al_ga.ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001958 if (curwin->w_arg_idx > match)
1959 --curwin->w_arg_idx;
1960 --match;
1961 }
1962
1963 vim_free(regmatch.regprog);
1964 vim_free(p);
1965 if (!didone)
1966 EMSG2(_(e_nomatch2), ((char_u **)new_ga.ga_data)[i]);
1967 }
1968 ga_clear(&new_ga);
1969 }
1970 else
1971#endif
1972 {
1973 i = expand_wildcards(new_ga.ga_len, (char_u **)new_ga.ga_data,
1974 &exp_count, &exp_files, EW_DIR|EW_FILE|EW_ADDSLASH|EW_NOTFOUND);
1975 ga_clear(&new_ga);
1976 if (i == FAIL)
1977 return FAIL;
1978 if (exp_count == 0)
1979 {
1980 EMSG(_(e_nomatch));
1981 return FAIL;
1982 }
1983
1984#ifdef FEAT_LISTCMDS
1985 if (what == AL_ADD)
1986 {
1987 (void)alist_add_list(exp_count, exp_files, after);
1988 vim_free(exp_files);
1989 }
1990 else /* what == AL_SET */
1991#endif
Bram Moolenaar86b68352004-12-27 21:59:20 +00001992 alist_set(ALIST(curwin), exp_count, exp_files, FALSE, NULL, 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001993 }
1994
1995 alist_check_arg_idx();
1996
1997 return OK;
1998}
1999
2000/*
2001 * Check the validity of the arg_idx for each other window.
2002 */
2003 static void
2004alist_check_arg_idx()
2005{
2006#ifdef FEAT_WINDOWS
2007 win_T *win;
Bram Moolenaarf740b292006-02-16 22:11:02 +00002008 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002009
Bram Moolenaarf740b292006-02-16 22:11:02 +00002010 FOR_ALL_TAB_WINDOWS(tp, win)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002011 if (win->w_alist == curwin->w_alist)
2012 check_arg_idx(win);
2013#else
2014 check_arg_idx(curwin);
2015#endif
2016}
2017
2018/*
Bram Moolenaarb8ff1fb2012-02-04 21:59:01 +01002019 * Return TRUE if window "win" is editing the file at the current argument
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002020 * index.
2021 */
2022 static int
2023editing_arg_idx(win)
2024 win_T *win;
2025{
2026 return !(win->w_arg_idx >= WARGCOUNT(win)
2027 || (win->w_buffer->b_fnum
2028 != WARGLIST(win)[win->w_arg_idx].ae_fnum
2029 && (win->w_buffer->b_ffname == NULL
2030 || !(fullpathcmp(
2031 alist_name(&WARGLIST(win)[win->w_arg_idx]),
2032 win->w_buffer->b_ffname, TRUE) & FPC_SAME))));
2033}
2034
2035/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002036 * Check if window "win" is editing the w_arg_idx file in its argument list.
2037 */
2038 void
2039check_arg_idx(win)
2040 win_T *win;
2041{
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002042 if (WARGCOUNT(win) > 1 && !editing_arg_idx(win))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002043 {
2044 /* We are not editing the current entry in the argument list.
2045 * Set "arg_had_last" if we are editing the last one. */
2046 win->w_arg_idx_invalid = TRUE;
2047 if (win->w_arg_idx != WARGCOUNT(win) - 1
2048 && arg_had_last == FALSE
2049#ifdef FEAT_WINDOWS
2050 && ALIST(win) == &global_alist
2051#endif
2052 && GARGCOUNT > 0
2053 && win->w_arg_idx < GARGCOUNT
2054 && (win->w_buffer->b_fnum == GARGLIST[GARGCOUNT - 1].ae_fnum
2055 || (win->w_buffer->b_ffname != NULL
2056 && (fullpathcmp(alist_name(&GARGLIST[GARGCOUNT - 1]),
2057 win->w_buffer->b_ffname, TRUE) & FPC_SAME))))
2058 arg_had_last = TRUE;
2059 }
2060 else
2061 {
2062 /* We are editing the current entry in the argument list.
2063 * Set "arg_had_last" if it's also the last one */
2064 win->w_arg_idx_invalid = FALSE;
2065 if (win->w_arg_idx == WARGCOUNT(win) - 1
2066#ifdef FEAT_WINDOWS
2067 && win->w_alist == &global_alist
2068#endif
2069 )
2070 arg_had_last = TRUE;
2071 }
2072}
2073
2074/*
2075 * ":args", ":argslocal" and ":argsglobal".
2076 */
2077 void
2078ex_args(eap)
2079 exarg_T *eap;
2080{
2081 int i;
2082
2083 if (eap->cmdidx != CMD_args)
2084 {
2085#if defined(FEAT_WINDOWS) && defined(FEAT_LISTCMDS)
2086 alist_unlink(ALIST(curwin));
2087 if (eap->cmdidx == CMD_argglobal)
2088 ALIST(curwin) = &global_alist;
2089 else /* eap->cmdidx == CMD_arglocal */
2090 alist_new();
2091#else
2092 ex_ni(eap);
2093 return;
2094#endif
2095 }
2096
2097 if (!ends_excmd(*eap->arg))
2098 {
2099 /*
2100 * ":args file ..": define new argument list, handle like ":next"
2101 * Also for ":argslocal file .." and ":argsglobal file ..".
2102 */
2103 ex_next(eap);
2104 }
2105 else
2106#if defined(FEAT_WINDOWS) && defined(FEAT_LISTCMDS)
2107 if (eap->cmdidx == CMD_args)
2108#endif
2109 {
2110 /*
2111 * ":args": list arguments.
2112 */
2113 if (ARGCOUNT > 0)
2114 {
2115 /* Overwrite the command, for a short list there is no scrolling
2116 * required and no wait_return(). */
2117 gotocmdline(TRUE);
2118 for (i = 0; i < ARGCOUNT; ++i)
2119 {
2120 if (i == curwin->w_arg_idx)
2121 msg_putchar('[');
2122 msg_outtrans(alist_name(&ARGLIST[i]));
2123 if (i == curwin->w_arg_idx)
2124 msg_putchar(']');
2125 msg_putchar(' ');
2126 }
2127 }
2128 }
2129#if defined(FEAT_WINDOWS) && defined(FEAT_LISTCMDS)
2130 else if (eap->cmdidx == CMD_arglocal)
2131 {
2132 garray_T *gap = &curwin->w_alist->al_ga;
2133
2134 /*
2135 * ":argslocal": make a local copy of the global argument list.
2136 */
2137 if (ga_grow(gap, GARGCOUNT) == OK)
2138 for (i = 0; i < GARGCOUNT; ++i)
2139 if (GARGLIST[i].ae_fname != NULL)
2140 {
2141 AARGLIST(curwin->w_alist)[gap->ga_len].ae_fname =
2142 vim_strsave(GARGLIST[i].ae_fname);
2143 AARGLIST(curwin->w_alist)[gap->ga_len].ae_fnum =
2144 GARGLIST[i].ae_fnum;
2145 ++gap->ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002146 }
2147 }
2148#endif
2149}
2150
2151/*
2152 * ":previous", ":sprevious", ":Next" and ":sNext".
2153 */
2154 void
2155ex_previous(eap)
2156 exarg_T *eap;
2157{
2158 /* If past the last one already, go to the last one. */
2159 if (curwin->w_arg_idx - (int)eap->line2 >= ARGCOUNT)
2160 do_argfile(eap, ARGCOUNT - 1);
2161 else
2162 do_argfile(eap, curwin->w_arg_idx - (int)eap->line2);
2163}
2164
2165/*
2166 * ":rewind", ":first", ":sfirst" and ":srewind".
2167 */
2168 void
2169ex_rewind(eap)
2170 exarg_T *eap;
2171{
2172 do_argfile(eap, 0);
2173}
2174
2175/*
2176 * ":last" and ":slast".
2177 */
2178 void
2179ex_last(eap)
2180 exarg_T *eap;
2181{
2182 do_argfile(eap, ARGCOUNT - 1);
2183}
2184
2185/*
2186 * ":argument" and ":sargument".
2187 */
2188 void
2189ex_argument(eap)
2190 exarg_T *eap;
2191{
2192 int i;
2193
2194 if (eap->addr_count > 0)
2195 i = eap->line2 - 1;
2196 else
2197 i = curwin->w_arg_idx;
2198 do_argfile(eap, i);
2199}
2200
2201/*
2202 * Edit file "argn" of the argument lists.
2203 */
2204 void
2205do_argfile(eap, argn)
2206 exarg_T *eap;
2207 int argn;
2208{
2209 int other;
2210 char_u *p;
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002211 int old_arg_idx = curwin->w_arg_idx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002212
2213 if (argn < 0 || argn >= ARGCOUNT)
2214 {
2215 if (ARGCOUNT <= 1)
2216 EMSG(_("E163: There is only one file to edit"));
2217 else if (argn < 0)
2218 EMSG(_("E164: Cannot go before first file"));
2219 else
2220 EMSG(_("E165: Cannot go beyond last file"));
2221 }
2222 else
2223 {
2224 setpcmark();
2225#ifdef FEAT_GUI
2226 need_mouse_correct = TRUE;
2227#endif
2228
2229#ifdef FEAT_WINDOWS
Bram Moolenaardf1bdc92006-02-23 21:32:16 +00002230 /* split window or create new tab page first */
2231 if (*eap->cmd == 's' || cmdmod.tab != 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002232 {
2233 if (win_split(0, 0) == FAIL)
2234 return;
Bram Moolenaar3368ea22010-09-21 16:56:35 +02002235 RESET_BINDING(curwin);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002236 }
2237 else
2238#endif
2239 {
2240 /*
2241 * if 'hidden' set, only check for changed file when re-editing
2242 * the same buffer
2243 */
2244 other = TRUE;
2245 if (P_HID(curbuf))
2246 {
2247 p = fix_fname(alist_name(&ARGLIST[argn]));
2248 other = otherfile(p);
2249 vim_free(p);
2250 }
2251 if ((!P_HID(curbuf) || !other)
2252 && check_changed(curbuf, TRUE, !other, eap->forceit, FALSE))
2253 return;
2254 }
2255
2256 curwin->w_arg_idx = argn;
2257 if (argn == ARGCOUNT - 1
2258#ifdef FEAT_WINDOWS
2259 && curwin->w_alist == &global_alist
2260#endif
2261 )
2262 arg_had_last = TRUE;
2263
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002264 /* Edit the file; always use the last known line number.
2265 * When it fails (e.g. Abort for already edited file) restore the
2266 * argument index. */
2267 if (do_ecmd(0, alist_name(&ARGLIST[curwin->w_arg_idx]), NULL,
Bram Moolenaar071d4272004-06-13 20:20:40 +00002268 eap, ECMD_LAST,
Bram Moolenaar701f7af2008-11-15 13:12:07 +00002269 (P_HID(curwin->w_buffer) ? ECMD_HIDE : 0)
2270 + (eap->forceit ? ECMD_FORCEIT : 0), curwin) == FAIL)
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002271 curwin->w_arg_idx = old_arg_idx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002272 /* like Vi: set the mark where the cursor is in the file. */
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002273 else if (eap->cmdidx != CMD_argdo)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002274 setmark('\'');
2275 }
2276}
2277
2278/*
2279 * ":next", and commands that behave like it.
2280 */
2281 void
2282ex_next(eap)
2283 exarg_T *eap;
2284{
2285 int i;
2286
2287 /*
2288 * check for changed buffer now, if this fails the argument list is not
2289 * redefined.
2290 */
2291 if ( P_HID(curbuf)
2292 || eap->cmdidx == CMD_snext
2293 || !check_changed(curbuf, TRUE, FALSE, eap->forceit, FALSE))
2294 {
2295 if (*eap->arg != NUL) /* redefine file list */
2296 {
2297 if (do_arglist(eap->arg, AL_SET, 0) == FAIL)
2298 return;
2299 i = 0;
2300 }
2301 else
2302 i = curwin->w_arg_idx + (int)eap->line2;
2303 do_argfile(eap, i);
2304 }
2305}
2306
2307#ifdef FEAT_LISTCMDS
2308/*
2309 * ":argedit"
2310 */
2311 void
2312ex_argedit(eap)
2313 exarg_T *eap;
2314{
2315 int fnum;
2316 int i;
2317 char_u *s;
2318
2319 /* Add the argument to the buffer list and get the buffer number. */
2320 fnum = buflist_add(eap->arg, BLN_LISTED);
2321
2322 /* Check if this argument is already in the argument list. */
2323 for (i = 0; i < ARGCOUNT; ++i)
2324 if (ARGLIST[i].ae_fnum == fnum)
2325 break;
2326 if (i == ARGCOUNT)
2327 {
2328 /* Can't find it, add it to the argument list. */
2329 s = vim_strsave(eap->arg);
2330 if (s == NULL)
2331 return;
2332 i = alist_add_list(1, &s,
2333 eap->addr_count > 0 ? (int)eap->line2 : curwin->w_arg_idx + 1);
2334 if (i < 0)
2335 return;
2336 curwin->w_arg_idx = i;
2337 }
2338
2339 alist_check_arg_idx();
2340
2341 /* Edit the argument. */
2342 do_argfile(eap, i);
2343}
2344
2345/*
2346 * ":argadd"
2347 */
2348 void
2349ex_argadd(eap)
2350 exarg_T *eap;
2351{
2352 do_arglist(eap->arg, AL_ADD,
2353 eap->addr_count > 0 ? (int)eap->line2 : curwin->w_arg_idx + 1);
2354#ifdef FEAT_TITLE
2355 maketitle();
2356#endif
2357}
2358
2359/*
2360 * ":argdelete"
2361 */
2362 void
2363ex_argdelete(eap)
2364 exarg_T *eap;
2365{
2366 int i;
2367 int n;
2368
2369 if (eap->addr_count > 0)
2370 {
2371 /* ":1,4argdel": Delete all arguments in the range. */
2372 if (eap->line2 > ARGCOUNT)
2373 eap->line2 = ARGCOUNT;
2374 n = eap->line2 - eap->line1 + 1;
2375 if (*eap->arg != NUL || n <= 0)
2376 EMSG(_(e_invarg));
2377 else
2378 {
2379 for (i = eap->line1; i <= eap->line2; ++i)
2380 vim_free(ARGLIST[i - 1].ae_fname);
2381 mch_memmove(ARGLIST + eap->line1 - 1, ARGLIST + eap->line2,
2382 (size_t)((ARGCOUNT - eap->line2) * sizeof(aentry_T)));
2383 ALIST(curwin)->al_ga.ga_len -= n;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002384 if (curwin->w_arg_idx >= eap->line2)
2385 curwin->w_arg_idx -= n;
2386 else if (curwin->w_arg_idx > eap->line1)
2387 curwin->w_arg_idx = eap->line1;
2388 }
2389 }
2390 else if (*eap->arg == NUL)
2391 EMSG(_(e_argreq));
2392 else
2393 do_arglist(eap->arg, AL_DEL, 0);
2394#ifdef FEAT_TITLE
2395 maketitle();
2396#endif
2397}
2398
2399/*
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002400 * ":argdo", ":windo", ":bufdo", ":tabdo"
Bram Moolenaar071d4272004-06-13 20:20:40 +00002401 */
2402 void
2403ex_listdo(eap)
2404 exarg_T *eap;
2405{
2406 int i;
2407#ifdef FEAT_WINDOWS
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002408 win_T *wp;
2409 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002410#endif
2411 buf_T *buf;
2412 int next_fnum = 0;
2413#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
2414 char_u *save_ei = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002415#endif
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002416 char_u *p_shm_save;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002417
2418#ifndef FEAT_WINDOWS
2419 if (eap->cmdidx == CMD_windo)
2420 {
2421 ex_ni(eap);
2422 return;
2423 }
2424#endif
2425
2426#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002427 if (eap->cmdidx != CMD_windo && eap->cmdidx != CMD_tabdo)
Bram Moolenaardcaf10e2005-01-21 11:55:25 +00002428 /* Don't do syntax HL autocommands. Skipping the syntax file is a
2429 * great speed improvement. */
2430 save_ei = au_event_disable(",Syntax");
Bram Moolenaar071d4272004-06-13 20:20:40 +00002431#endif
2432
2433 if (eap->cmdidx == CMD_windo
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002434 || eap->cmdidx == CMD_tabdo
Bram Moolenaar071d4272004-06-13 20:20:40 +00002435 || P_HID(curbuf)
2436 || !check_changed(curbuf, TRUE, FALSE, eap->forceit, FALSE))
2437 {
2438 /* start at the first argument/window/buffer */
2439 i = 0;
2440#ifdef FEAT_WINDOWS
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002441 wp = firstwin;
2442 tp = first_tabpage;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002443#endif
2444 /* set pcmark now */
2445 if (eap->cmdidx == CMD_bufdo)
2446 goto_buffer(eap, DOBUF_FIRST, FORWARD, 0);
2447 else
2448 setpcmark();
2449 listcmd_busy = TRUE; /* avoids setting pcmark below */
2450
2451 while (!got_int)
2452 {
2453 if (eap->cmdidx == CMD_argdo)
2454 {
2455 /* go to argument "i" */
2456 if (i == ARGCOUNT)
2457 break;
2458 /* Don't call do_argfile() when already there, it will try
2459 * reloading the file. */
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002460 if (curwin->w_arg_idx != i || !editing_arg_idx(curwin))
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002461 {
2462 /* Clear 'shm' to avoid that the file message overwrites
2463 * any output from the command. */
2464 p_shm_save = vim_strsave(p_shm);
2465 set_option_value((char_u *)"shm", 0L, (char_u *)"", 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002466 do_argfile(eap, i);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002467 set_option_value((char_u *)"shm", 0L, p_shm_save, 0);
2468 vim_free(p_shm_save);
2469 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002470 if (curwin->w_arg_idx != i)
2471 break;
2472 ++i;
2473 }
2474#ifdef FEAT_WINDOWS
2475 else if (eap->cmdidx == CMD_windo)
2476 {
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002477 /* go to window "wp" */
2478 if (!win_valid(wp))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002479 break;
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002480 win_goto(wp);
Bram Moolenaar41423242007-05-03 20:11:13 +00002481 if (curwin != wp)
2482 break; /* something must be wrong */
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002483 wp = curwin->w_next;
2484 }
2485 else if (eap->cmdidx == CMD_tabdo)
2486 {
2487 /* go to window "tp" */
2488 if (!valid_tabpage(tp))
2489 break;
Bram Moolenaara8596c42012-06-13 14:28:20 +02002490 goto_tabpage_tp(tp, TRUE);
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002491 tp = tp->tp_next;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002492 }
2493#endif
2494 else if (eap->cmdidx == CMD_bufdo)
2495 {
2496 /* Remember the number of the next listed buffer, in case
2497 * ":bwipe" is used or autocommands do something strange. */
2498 next_fnum = -1;
2499 for (buf = curbuf->b_next; buf != NULL; buf = buf->b_next)
2500 if (buf->b_p_bl)
2501 {
2502 next_fnum = buf->b_fnum;
2503 break;
2504 }
2505 }
2506
2507 /* execute the command */
2508 do_cmdline(eap->arg, eap->getline, eap->cookie,
2509 DOCMD_VERBOSE + DOCMD_NOWAIT);
2510
2511 if (eap->cmdidx == CMD_bufdo)
2512 {
2513 /* Done? */
2514 if (next_fnum < 0)
2515 break;
2516 /* Check if the buffer still exists. */
2517 for (buf = firstbuf; buf != NULL; buf = buf->b_next)
2518 if (buf->b_fnum == next_fnum)
2519 break;
2520 if (buf == NULL)
2521 break;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002522
2523 /* Go to the next buffer. Clear 'shm' to avoid that the file
2524 * message overwrites any output from the command. */
2525 p_shm_save = vim_strsave(p_shm);
2526 set_option_value((char_u *)"shm", 0L, (char_u *)"", 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002527 goto_buffer(eap, DOBUF_FIRST, FORWARD, next_fnum);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002528 set_option_value((char_u *)"shm", 0L, p_shm_save, 0);
2529 vim_free(p_shm_save);
2530
Bram Moolenaar071d4272004-06-13 20:20:40 +00002531 /* If autocommands took us elsewhere, quit here */
2532 if (curbuf->b_fnum != next_fnum)
2533 break;
2534 }
2535
2536 if (eap->cmdidx == CMD_windo)
2537 {
2538 validate_cursor(); /* cursor may have moved */
2539#ifdef FEAT_SCROLLBIND
2540 /* required when 'scrollbind' has been set */
2541 if (curwin->w_p_scb)
2542 do_check_scrollbind(TRUE);
2543#endif
2544 }
2545 }
2546 listcmd_busy = FALSE;
2547 }
2548
2549#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
Bram Moolenaar6ac54292005-02-02 23:07:25 +00002550 if (save_ei != NULL)
2551 {
2552 au_event_restore(save_ei);
2553 apply_autocmds(EVENT_SYNTAX, curbuf->b_p_syn,
2554 curbuf->b_fname, TRUE, curbuf);
2555 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002556#endif
2557}
2558
2559/*
2560 * Add files[count] to the arglist of the current window after arg "after".
2561 * The file names in files[count] must have been allocated and are taken over.
2562 * Files[] itself is not taken over.
2563 * Returns index of first added argument. Returns -1 when failed (out of mem).
2564 */
2565 static int
2566alist_add_list(count, files, after)
2567 int count;
2568 char_u **files;
2569 int after; /* where to add: 0 = before first one */
2570{
2571 int i;
2572
2573 if (ga_grow(&ALIST(curwin)->al_ga, count) == OK)
2574 {
2575 if (after < 0)
2576 after = 0;
2577 if (after > ARGCOUNT)
2578 after = ARGCOUNT;
2579 if (after < ARGCOUNT)
2580 mch_memmove(&(ARGLIST[after + count]), &(ARGLIST[after]),
2581 (ARGCOUNT - after) * sizeof(aentry_T));
2582 for (i = 0; i < count; ++i)
2583 {
2584 ARGLIST[after + i].ae_fname = files[i];
2585 ARGLIST[after + i].ae_fnum = buflist_add(files[i], BLN_LISTED);
2586 }
2587 ALIST(curwin)->al_ga.ga_len += count;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002588 if (curwin->w_arg_idx >= after)
2589 ++curwin->w_arg_idx;
2590 return after;
2591 }
2592
2593 for (i = 0; i < count; ++i)
2594 vim_free(files[i]);
2595 return -1;
2596}
2597
2598#endif /* FEAT_LISTCMDS */
2599
2600#ifdef FEAT_EVAL
2601/*
2602 * ":compiler[!] {name}"
2603 */
2604 void
2605ex_compiler(eap)
2606 exarg_T *eap;
2607{
2608 char_u *buf;
2609 char_u *old_cur_comp = NULL;
2610 char_u *p;
2611
2612 if (*eap->arg == NUL)
2613 {
2614 /* List all compiler scripts. */
2615 do_cmdline_cmd((char_u *)"echo globpath(&rtp, 'compiler/*.vim')");
2616 /* ) keep the indenter happy... */
2617 }
2618 else
2619 {
2620 buf = alloc((unsigned)(STRLEN(eap->arg) + 14));
2621 if (buf != NULL)
2622 {
2623 if (eap->forceit)
2624 {
2625 /* ":compiler! {name}" sets global options */
2626 do_cmdline_cmd((char_u *)
2627 "command -nargs=* CompilerSet set <args>");
2628 }
2629 else
2630 {
2631 /* ":compiler! {name}" sets local options.
2632 * To remain backwards compatible "current_compiler" is always
2633 * used. A user's compiler plugin may set it, the distributed
2634 * plugin will then skip the settings. Afterwards set
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01002635 * "b:current_compiler" and restore "current_compiler".
2636 * Explicitly prepend "g:" to make it work in a function. */
2637 old_cur_comp = get_var_value((char_u *)"g:current_compiler");
Bram Moolenaar071d4272004-06-13 20:20:40 +00002638 if (old_cur_comp != NULL)
2639 old_cur_comp = vim_strsave(old_cur_comp);
2640 do_cmdline_cmd((char_u *)
2641 "command -nargs=* CompilerSet setlocal <args>");
2642 }
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01002643 do_unlet((char_u *)"g:current_compiler", TRUE);
Bram Moolenaar2ce06f62005-01-31 19:19:04 +00002644 do_unlet((char_u *)"b:current_compiler", TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002645
2646 sprintf((char *)buf, "compiler/%s.vim", eap->arg);
Bram Moolenaar90cfdbe2005-08-12 19:59:19 +00002647 if (source_runtime(buf, TRUE) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002648 EMSG2(_("E666: compiler not supported: %s"), eap->arg);
2649 vim_free(buf);
2650
2651 do_cmdline_cmd((char_u *)":delcommand CompilerSet");
2652
2653 /* Set "b:current_compiler" from "current_compiler". */
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01002654 p = get_var_value((char_u *)"g:current_compiler");
Bram Moolenaar071d4272004-06-13 20:20:40 +00002655 if (p != NULL)
2656 set_internal_string_var((char_u *)"b:current_compiler", p);
2657
2658 /* Restore "current_compiler" for ":compiler {name}". */
2659 if (!eap->forceit)
2660 {
2661 if (old_cur_comp != NULL)
2662 {
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01002663 set_internal_string_var((char_u *)"g:current_compiler",
Bram Moolenaar071d4272004-06-13 20:20:40 +00002664 old_cur_comp);
2665 vim_free(old_cur_comp);
2666 }
2667 else
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01002668 do_unlet((char_u *)"g:current_compiler", TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002669 }
2670 }
2671 }
2672}
2673#endif
2674
2675/*
2676 * ":runtime {name}"
2677 */
2678 void
2679ex_runtime(eap)
2680 exarg_T *eap;
2681{
Bram Moolenaar90cfdbe2005-08-12 19:59:19 +00002682 source_runtime(eap->arg, eap->forceit);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002683}
2684
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002685static void source_callback __ARGS((char_u *fname, void *cookie));
Bram Moolenaar071d4272004-06-13 20:20:40 +00002686
2687 static void
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002688source_callback(fname, cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002689 char_u *fname;
Bram Moolenaar0c094b92009-05-14 20:20:33 +00002690 void *cookie UNUSED;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002691{
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002692 (void)do_source(fname, FALSE, DOSO_NONE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002693}
2694
2695/*
2696 * Source the file "name" from all directories in 'runtimepath'.
2697 * "name" can contain wildcards.
2698 * When "all" is TRUE, source all files, otherwise only the first one.
2699 * return FAIL when no file could be sourced, OK otherwise.
2700 */
2701 int
Bram Moolenaar90cfdbe2005-08-12 19:59:19 +00002702source_runtime(name, all)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002703 char_u *name;
2704 int all;
2705{
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002706 return do_in_runtimepath(name, all, source_callback, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002707}
2708
2709/*
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002710 * Find "name" in 'runtimepath'. When found, invoke the callback function for
2711 * it: callback(fname, "cookie")
Bram Moolenaar071d4272004-06-13 20:20:40 +00002712 * When "all" is TRUE repeat for all matches, otherwise only the first one is
2713 * used.
2714 * Returns OK when at least one match found, FAIL otherwise.
2715 */
2716 int
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002717do_in_runtimepath(name, all, callback, cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002718 char_u *name;
2719 int all;
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002720 void (*callback)__ARGS((char_u *fname, void *ck));
2721 void *cookie;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002722{
2723 char_u *rtp;
2724 char_u *np;
2725 char_u *buf;
2726 char_u *rtp_copy;
2727 char_u *tail;
2728 int num_files;
2729 char_u **files;
2730 int i;
2731 int did_one = FALSE;
2732#ifdef AMIGA
2733 struct Process *proc = (struct Process *)FindTask(0L);
2734 APTR save_winptr = proc->pr_WindowPtr;
2735
2736 /* Avoid a requester here for a volume that doesn't exist. */
2737 proc->pr_WindowPtr = (APTR)-1L;
2738#endif
2739
2740 /* Make a copy of 'runtimepath'. Invoking the callback may change the
2741 * value. */
2742 rtp_copy = vim_strsave(p_rtp);
2743 buf = alloc(MAXPATHL);
2744 if (buf != NULL && rtp_copy != NULL)
2745 {
2746 if (p_verbose > 1)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00002747 {
2748 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00002749 smsg((char_u *)_("Searching for \"%s\" in \"%s\""),
Bram Moolenaar071d4272004-06-13 20:20:40 +00002750 (char *)name, (char *)p_rtp);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00002751 verbose_leave();
2752 }
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002753
Bram Moolenaar071d4272004-06-13 20:20:40 +00002754 /* Loop over all entries in 'runtimepath'. */
2755 rtp = rtp_copy;
2756 while (*rtp != NUL && (all || !did_one))
2757 {
2758 /* Copy the path from 'runtimepath' to buf[]. */
2759 copy_option_part(&rtp, buf, MAXPATHL, ",");
2760 if (STRLEN(buf) + STRLEN(name) + 2 < MAXPATHL)
2761 {
2762 add_pathsep(buf);
2763 tail = buf + STRLEN(buf);
2764
2765 /* Loop over all patterns in "name" */
2766 np = name;
2767 while (*np != NUL && (all || !did_one))
2768 {
2769 /* Append the pattern from "name" to buf[]. */
2770 copy_option_part(&np, tail, (int)(MAXPATHL - (tail - buf)),
2771 "\t ");
2772
2773 if (p_verbose > 2)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00002774 {
2775 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00002776 smsg((char_u *)_("Searching for \"%s\""), buf);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00002777 verbose_leave();
2778 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002779
2780 /* Expand wildcards, invoke the callback for each match. */
2781 if (gen_expand_wildcards(1, &buf, &num_files, &files,
2782 EW_FILE) == OK)
2783 {
2784 for (i = 0; i < num_files; ++i)
2785 {
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002786 (*callback)(files[i], cookie);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002787 did_one = TRUE;
2788 if (!all)
2789 break;
2790 }
2791 FreeWild(num_files, files);
2792 }
2793 }
2794 }
2795 }
2796 }
2797 vim_free(buf);
2798 vim_free(rtp_copy);
2799 if (p_verbose > 0 && !did_one)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00002800 {
2801 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00002802 smsg((char_u *)_("not found in 'runtimepath': \"%s\""), name);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00002803 verbose_leave();
2804 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002805
2806#ifdef AMIGA
2807 proc->pr_WindowPtr = save_winptr;
2808#endif
2809
2810 return did_one ? OK : FAIL;
2811}
2812
2813#if defined(FEAT_EVAL) && defined(FEAT_AUTOCMD)
2814/*
2815 * ":options"
2816 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002817 void
2818ex_options(eap)
Bram Moolenaar0c094b92009-05-14 20:20:33 +00002819 exarg_T *eap UNUSED;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002820{
2821 cmd_source((char_u *)SYS_OPTWIN_FILE, NULL);
2822}
2823#endif
2824
2825/*
2826 * ":source {fname}"
2827 */
2828 void
2829ex_source(eap)
2830 exarg_T *eap;
2831{
2832#ifdef FEAT_BROWSE
2833 if (cmdmod.browse)
2834 {
2835 char_u *fname = NULL;
2836
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00002837 fname = do_browse(0, (char_u *)_("Source Vim script"), eap->arg,
Bram Moolenaar071d4272004-06-13 20:20:40 +00002838 NULL, NULL, BROWSE_FILTER_MACROS, NULL);
2839 if (fname != NULL)
2840 {
2841 cmd_source(fname, eap);
2842 vim_free(fname);
2843 }
2844 }
2845 else
2846#endif
2847 cmd_source(eap->arg, eap);
2848}
2849
2850 static void
2851cmd_source(fname, eap)
2852 char_u *fname;
2853 exarg_T *eap;
2854{
2855 if (*fname == NUL)
2856 EMSG(_(e_argreq));
2857
Bram Moolenaar071d4272004-06-13 20:20:40 +00002858 else if (eap != NULL && eap->forceit)
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00002859 /* ":source!": read Normal mdoe commands
2860 * Need to execute the commands directly. This is required at least
2861 * for:
Bram Moolenaar071d4272004-06-13 20:20:40 +00002862 * - ":g" command busy
2863 * - after ":argdo", ":windo" or ":bufdo"
2864 * - another command follows
2865 * - inside a loop
2866 */
2867 openscript(fname, global_busy || listcmd_busy || eap->nextcmd != NULL
2868#ifdef FEAT_EVAL
2869 || eap->cstack->cs_idx >= 0
2870#endif
2871 );
2872
2873 /* ":source" read ex commands */
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002874 else if (do_source(fname, FALSE, DOSO_NONE) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002875 EMSG2(_(e_notopen), fname);
2876}
2877
2878/*
2879 * ":source" and associated commands.
2880 */
2881/*
2882 * Structure used to store info for each sourced file.
2883 * It is shared between do_source() and getsourceline().
2884 * This is required, because it needs to be handed to do_cmdline() and
2885 * sourcing can be done recursively.
2886 */
2887struct source_cookie
2888{
2889 FILE *fp; /* opened file for sourcing */
2890 char_u *nextline; /* if not NULL: line that was read ahead */
2891 int finished; /* ":finish" used */
Bram Moolenaar860cae12010-06-05 23:22:07 +02002892#if defined(USE_CRNL) || defined(USE_CR)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002893 int fileformat; /* EOL_UNKNOWN, EOL_UNIX or EOL_DOS */
2894 int error; /* TRUE if LF found after CR-LF */
2895#endif
2896#ifdef FEAT_EVAL
2897 linenr_T breakpoint; /* next line with breakpoint or zero */
2898 char_u *fname; /* name of sourced file */
2899 int dbg_tick; /* debug_tick when breakpoint was set */
2900 int level; /* top nesting level of sourced file */
2901#endif
2902#ifdef FEAT_MBYTE
2903 vimconv_T conv; /* type of conversion */
2904#endif
2905};
2906
2907#ifdef FEAT_EVAL
2908/*
2909 * Return the address holding the next breakpoint line for a source cookie.
2910 */
2911 linenr_T *
2912source_breakpoint(cookie)
2913 void *cookie;
2914{
2915 return &((struct source_cookie *)cookie)->breakpoint;
2916}
2917
2918/*
2919 * Return the address holding the debug tick for a source cookie.
2920 */
2921 int *
2922source_dbg_tick(cookie)
2923 void *cookie;
2924{
2925 return &((struct source_cookie *)cookie)->dbg_tick;
2926}
2927
2928/*
2929 * Return the nesting level for a source cookie.
2930 */
2931 int
2932source_level(cookie)
2933 void *cookie;
2934{
2935 return ((struct source_cookie *)cookie)->level;
2936}
2937#endif
2938
2939static char_u *get_one_sourceline __ARGS((struct source_cookie *sp));
2940
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01002941#if (defined(WIN32) && defined(FEAT_CSCOPE)) || defined(HAVE_FD_CLOEXEC)
2942# define USE_FOPEN_NOINH
Bram Moolenaar071d4272004-06-13 20:20:40 +00002943static FILE *fopen_noinh_readbin __ARGS((char *filename));
2944
2945/*
2946 * Special function to open a file without handle inheritance.
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01002947 * When possible the handle is closed on exec().
Bram Moolenaar071d4272004-06-13 20:20:40 +00002948 */
2949 static FILE *
2950fopen_noinh_readbin(filename)
2951 char *filename;
2952{
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01002953# ifdef WIN32
Bram Moolenaar8d8ef0b2010-01-20 21:41:47 +01002954 int fd_tmp = mch_open(filename, O_RDONLY | O_BINARY | O_NOINHERIT, 0);
2955# else
2956 int fd_tmp = mch_open(filename, O_RDONLY, 0);
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01002957# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002958
2959 if (fd_tmp == -1)
2960 return NULL;
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01002961
2962# ifdef HAVE_FD_CLOEXEC
2963 {
2964 int fdflags = fcntl(fd_tmp, F_GETFD);
2965 if (fdflags >= 0 && (fdflags & FD_CLOEXEC) == 0)
2966 fcntl(fd_tmp, F_SETFD, fdflags | FD_CLOEXEC);
2967 }
2968# endif
2969
Bram Moolenaar071d4272004-06-13 20:20:40 +00002970 return fdopen(fd_tmp, READBIN);
2971}
2972#endif
2973
2974
2975/*
2976 * do_source: Read the file "fname" and execute its lines as EX commands.
2977 *
2978 * This function may be called recursively!
2979 *
2980 * return FAIL if file could not be opened, OK otherwise
2981 */
2982 int
2983do_source(fname, check_other, is_vimrc)
2984 char_u *fname;
2985 int check_other; /* check for .vimrc and _vimrc */
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002986 int is_vimrc; /* DOSO_ value */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002987{
2988 struct source_cookie cookie;
2989 char_u *save_sourcing_name;
2990 linenr_T save_sourcing_lnum;
2991 char_u *p;
2992 char_u *fname_exp;
Bram Moolenaar73881402009-02-04 16:50:47 +00002993 char_u *firstline = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002994 int retval = FAIL;
2995#ifdef FEAT_EVAL
2996 scid_T save_current_SID;
2997 static scid_T last_current_SID = 0;
2998 void *save_funccalp;
2999 int save_debug_break_level = debug_break_level;
Bram Moolenaar05159a02005-02-26 23:04:13 +00003000 scriptitem_T *si = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003001# ifdef UNIX
3002 struct stat st;
3003 int stat_ok;
3004# endif
3005#endif
3006#ifdef STARTUPTIME
3007 struct timeval tv_rel;
3008 struct timeval tv_start;
3009#endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00003010#ifdef FEAT_PROFILE
3011 proftime_T wait_start;
3012#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003013
Bram Moolenaar071d4272004-06-13 20:20:40 +00003014 p = expand_env_save(fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003015 if (p == NULL)
3016 return retval;
3017 fname_exp = fix_fname(p);
3018 vim_free(p);
3019 if (fname_exp == NULL)
3020 return retval;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003021 if (mch_isdir(fname_exp))
3022 {
Bram Moolenaar555b2802005-05-19 21:08:39 +00003023 smsg((char_u *)_("Cannot source a directory: \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003024 goto theend;
3025 }
3026
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00003027#ifdef FEAT_AUTOCMD
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00003028 /* Apply SourceCmd autocommands, they should get the file and source it. */
3029 if (has_autocmd(EVENT_SOURCECMD, fname_exp, NULL)
3030 && apply_autocmds(EVENT_SOURCECMD, fname_exp, fname_exp,
3031 FALSE, curbuf))
Bram Moolenaarf33943e2008-01-15 21:17:30 +00003032 {
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00003033# ifdef FEAT_EVAL
Bram Moolenaarf33943e2008-01-15 21:17:30 +00003034 retval = aborting() ? FAIL : OK;
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00003035# else
Bram Moolenaarf33943e2008-01-15 21:17:30 +00003036 retval = OK;
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00003037# endif
Bram Moolenaarf33943e2008-01-15 21:17:30 +00003038 goto theend;
3039 }
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00003040
3041 /* Apply SourcePre autocommands, they may get the file. */
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00003042 apply_autocmds(EVENT_SOURCEPRE, fname_exp, fname_exp, FALSE, curbuf);
3043#endif
3044
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01003045#ifdef USE_FOPEN_NOINH
Bram Moolenaar071d4272004-06-13 20:20:40 +00003046 cookie.fp = fopen_noinh_readbin((char *)fname_exp);
3047#else
3048 cookie.fp = mch_fopen((char *)fname_exp, READBIN);
3049#endif
3050 if (cookie.fp == NULL && check_other)
3051 {
3052 /*
3053 * Try again, replacing file name ".vimrc" by "_vimrc" or vice versa,
3054 * and ".exrc" by "_exrc" or vice versa.
3055 */
3056 p = gettail(fname_exp);
3057 if ((*p == '.' || *p == '_')
3058 && (STRICMP(p + 1, "vimrc") == 0
3059 || STRICMP(p + 1, "gvimrc") == 0
3060 || STRICMP(p + 1, "exrc") == 0))
3061 {
3062 if (*p == '_')
3063 *p = '.';
3064 else
3065 *p = '_';
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01003066#ifdef USE_FOPEN_NOINH
Bram Moolenaar071d4272004-06-13 20:20:40 +00003067 cookie.fp = fopen_noinh_readbin((char *)fname_exp);
3068#else
3069 cookie.fp = mch_fopen((char *)fname_exp, READBIN);
3070#endif
3071 }
3072 }
3073
3074 if (cookie.fp == NULL)
3075 {
3076 if (p_verbose > 0)
3077 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003078 verbose_enter();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003079 if (sourcing_name == NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00003080 smsg((char_u *)_("could not source \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003081 else
3082 smsg((char_u *)_("line %ld: could not source \"%s\""),
Bram Moolenaar555b2802005-05-19 21:08:39 +00003083 sourcing_lnum, fname);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003084 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003085 }
3086 goto theend;
3087 }
3088
3089 /*
3090 * The file exists.
3091 * - In verbose mode, give a message.
3092 * - For a vimrc file, may want to set 'compatible', call vimrc_found().
3093 */
3094 if (p_verbose > 1)
3095 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003096 verbose_enter();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003097 if (sourcing_name == NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00003098 smsg((char_u *)_("sourcing \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003099 else
3100 smsg((char_u *)_("line %ld: sourcing \"%s\""),
Bram Moolenaar555b2802005-05-19 21:08:39 +00003101 sourcing_lnum, fname);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003102 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003103 }
Bram Moolenaar910f66f2006-04-05 20:41:53 +00003104 if (is_vimrc == DOSO_VIMRC)
3105 vimrc_found(fname_exp, (char_u *)"MYVIMRC");
3106 else if (is_vimrc == DOSO_GVIMRC)
3107 vimrc_found(fname_exp, (char_u *)"MYGVIMRC");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003108
3109#ifdef USE_CRNL
3110 /* If no automatic file format: Set default to CR-NL. */
3111 if (*p_ffs == NUL)
3112 cookie.fileformat = EOL_DOS;
3113 else
3114 cookie.fileformat = EOL_UNKNOWN;
3115 cookie.error = FALSE;
3116#endif
3117
3118#ifdef USE_CR
3119 /* If no automatic file format: Set default to CR. */
3120 if (*p_ffs == NUL)
3121 cookie.fileformat = EOL_MAC;
3122 else
3123 cookie.fileformat = EOL_UNKNOWN;
3124 cookie.error = FALSE;
3125#endif
3126
3127 cookie.nextline = NULL;
3128 cookie.finished = FALSE;
3129
3130#ifdef FEAT_EVAL
3131 /*
3132 * Check if this script has a breakpoint.
3133 */
3134 cookie.breakpoint = dbg_find_breakpoint(TRUE, fname_exp, (linenr_T)0);
3135 cookie.fname = fname_exp;
3136 cookie.dbg_tick = debug_tick;
3137
3138 cookie.level = ex_nesting_level;
3139#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003140
3141 /*
3142 * Keep the sourcing name/lnum, for recursive calls.
3143 */
3144 save_sourcing_name = sourcing_name;
3145 sourcing_name = fname_exp;
3146 save_sourcing_lnum = sourcing_lnum;
3147 sourcing_lnum = 0;
3148
Bram Moolenaar73881402009-02-04 16:50:47 +00003149#ifdef FEAT_MBYTE
3150 cookie.conv.vc_type = CONV_NONE; /* no conversion */
3151
3152 /* Read the first line so we can check for a UTF-8 BOM. */
3153 firstline = getsourceline(0, (void *)&cookie, 0);
3154 if (firstline != NULL && STRLEN(firstline) >= 3 && firstline[0] == 0xef
3155 && firstline[1] == 0xbb && firstline[2] == 0xbf)
3156 {
3157 /* Found BOM; setup conversion, skip over BOM and recode the line. */
3158 convert_setup(&cookie.conv, (char_u *)"utf-8", p_enc);
3159 p = string_convert(&cookie.conv, firstline + 3, NULL);
Bram Moolenaar4e2a5952009-02-05 19:48:25 +00003160 if (p == NULL)
3161 p = vim_strsave(firstline + 3);
Bram Moolenaar73881402009-02-04 16:50:47 +00003162 if (p != NULL)
3163 {
3164 vim_free(firstline);
3165 firstline = p;
3166 }
3167 }
3168#endif
3169
Bram Moolenaar071d4272004-06-13 20:20:40 +00003170#ifdef STARTUPTIME
Bram Moolenaarc4e41982010-01-19 16:31:47 +01003171 if (time_fd != NULL)
3172 time_push(&tv_rel, &tv_start);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003173#endif
3174
3175#ifdef FEAT_EVAL
Bram Moolenaar05159a02005-02-26 23:04:13 +00003176# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00003177 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003178 prof_child_enter(&wait_start); /* entering a child now */
3179# endif
3180
3181 /* Don't use local function variables, if called from a function.
3182 * Also starts profiling timer for nested script. */
3183 save_funccalp = save_funccal();
3184
Bram Moolenaar071d4272004-06-13 20:20:40 +00003185 /*
3186 * Check if this script was sourced before to finds its SID.
3187 * If it's new, generate a new SID.
3188 */
3189 save_current_SID = current_SID;
3190# ifdef UNIX
3191 stat_ok = (mch_stat((char *)fname_exp, &st) >= 0);
3192# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00003193 for (current_SID = script_items.ga_len; current_SID > 0; --current_SID)
3194 {
3195 si = &SCRIPT_ITEM(current_SID);
3196 if (si->sn_name != NULL
Bram Moolenaar071d4272004-06-13 20:20:40 +00003197 && (
3198# ifdef UNIX
Bram Moolenaar7c626922005-02-07 22:01:03 +00003199 /* Compare dev/ino when possible, it catches symbolic
3200 * links. Also compare file names, the inode may change
3201 * when the file was edited. */
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00003202 ((stat_ok && si->sn_dev_valid)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003203 && (si->sn_dev == st.st_dev
3204 && si->sn_ino == st.st_ino)) ||
Bram Moolenaar071d4272004-06-13 20:20:40 +00003205# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00003206 fnamecmp(si->sn_name, fname_exp) == 0))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003207 break;
Bram Moolenaar05159a02005-02-26 23:04:13 +00003208 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003209 if (current_SID == 0)
3210 {
3211 current_SID = ++last_current_SID;
Bram Moolenaar05159a02005-02-26 23:04:13 +00003212 if (ga_grow(&script_items, (int)(current_SID - script_items.ga_len))
3213 == FAIL)
3214 goto almosttheend;
3215 while (script_items.ga_len < current_SID)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003216 {
Bram Moolenaar05159a02005-02-26 23:04:13 +00003217 ++script_items.ga_len;
3218 SCRIPT_ITEM(script_items.ga_len).sn_name = NULL;
3219# ifdef FEAT_PROFILE
3220 SCRIPT_ITEM(script_items.ga_len).sn_prof_on = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003221# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003222 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00003223 si = &SCRIPT_ITEM(current_SID);
3224 si->sn_name = fname_exp;
3225 fname_exp = NULL;
3226# ifdef UNIX
3227 if (stat_ok)
3228 {
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00003229 si->sn_dev_valid = TRUE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00003230 si->sn_dev = st.st_dev;
3231 si->sn_ino = st.st_ino;
3232 }
3233 else
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00003234 si->sn_dev_valid = FALSE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00003235# endif
3236
Bram Moolenaar071d4272004-06-13 20:20:40 +00003237 /* Allocate the local script variables to use for this script. */
3238 new_script_vars(current_SID);
3239 }
3240
Bram Moolenaar05159a02005-02-26 23:04:13 +00003241# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00003242 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003243 {
3244 int forceit;
3245
3246 /* Check if we do profiling for this script. */
3247 if (!si->sn_prof_on && has_profiling(TRUE, si->sn_name, &forceit))
3248 {
3249 script_do_profile(si);
3250 si->sn_pr_force = forceit;
3251 }
3252 if (si->sn_prof_on)
3253 {
3254 ++si->sn_pr_count;
3255 profile_start(&si->sn_pr_start);
3256 profile_zero(&si->sn_pr_children);
3257 }
3258 }
3259# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003260#endif
3261
3262 /*
3263 * Call do_cmdline, which will call getsourceline() to get the lines.
3264 */
Bram Moolenaar73881402009-02-04 16:50:47 +00003265 do_cmdline(firstline, getsourceline, (void *)&cookie,
Bram Moolenaar071d4272004-06-13 20:20:40 +00003266 DOCMD_VERBOSE|DOCMD_NOWAIT|DOCMD_REPEAT);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003267 retval = OK;
Bram Moolenaar05159a02005-02-26 23:04:13 +00003268
3269#ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00003270 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003271 {
3272 /* Get "si" again, "script_items" may have been reallocated. */
3273 si = &SCRIPT_ITEM(current_SID);
3274 if (si->sn_prof_on)
3275 {
3276 profile_end(&si->sn_pr_start);
3277 profile_sub_wait(&wait_start, &si->sn_pr_start);
3278 profile_add(&si->sn_pr_total, &si->sn_pr_start);
Bram Moolenaar1056d982006-03-09 22:37:52 +00003279 profile_self(&si->sn_pr_self, &si->sn_pr_start,
3280 &si->sn_pr_children);
Bram Moolenaar05159a02005-02-26 23:04:13 +00003281 }
3282 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003283#endif
3284
3285 if (got_int)
3286 EMSG(_(e_interr));
3287 sourcing_name = save_sourcing_name;
3288 sourcing_lnum = save_sourcing_lnum;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003289 if (p_verbose > 1)
3290 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003291 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00003292 smsg((char_u *)_("finished sourcing %s"), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003293 if (sourcing_name != NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00003294 smsg((char_u *)_("continuing in %s"), sourcing_name);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003295 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003296 }
3297#ifdef STARTUPTIME
Bram Moolenaarc4e41982010-01-19 16:31:47 +01003298 if (time_fd != NULL)
3299 {
3300 vim_snprintf((char *)IObuff, IOSIZE, "sourcing %s", fname);
3301 time_msg((char *)IObuff, &tv_start);
3302 time_pop(&tv_rel);
3303 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003304#endif
3305
3306#ifdef FEAT_EVAL
3307 /*
3308 * After a "finish" in debug mode, need to break at first command of next
3309 * sourced file.
3310 */
3311 if (save_debug_break_level > ex_nesting_level
3312 && debug_break_level == ex_nesting_level)
3313 ++debug_break_level;
3314#endif
3315
Bram Moolenaar05159a02005-02-26 23:04:13 +00003316#ifdef FEAT_EVAL
3317almosttheend:
3318 current_SID = save_current_SID;
3319 restore_funccal(save_funccalp);
3320# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00003321 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003322 prof_child_exit(&wait_start); /* leaving a child now */
3323# endif
3324#endif
3325 fclose(cookie.fp);
3326 vim_free(cookie.nextline);
Bram Moolenaar73881402009-02-04 16:50:47 +00003327 vim_free(firstline);
Bram Moolenaar05159a02005-02-26 23:04:13 +00003328#ifdef FEAT_MBYTE
3329 convert_setup(&cookie.conv, NULL, NULL);
3330#endif
3331
Bram Moolenaar071d4272004-06-13 20:20:40 +00003332theend:
3333 vim_free(fname_exp);
3334 return retval;
3335}
3336
3337#if defined(FEAT_EVAL) || defined(PROTO)
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00003338
Bram Moolenaar071d4272004-06-13 20:20:40 +00003339/*
3340 * ":scriptnames"
3341 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003342 void
3343ex_scriptnames(eap)
Bram Moolenaar0c094b92009-05-14 20:20:33 +00003344 exarg_T *eap UNUSED;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003345{
3346 int i;
3347
Bram Moolenaar05159a02005-02-26 23:04:13 +00003348 for (i = 1; i <= script_items.ga_len && !got_int; ++i)
3349 if (SCRIPT_ITEM(i).sn_name != NULL)
Bram Moolenaard58ea072011-06-26 04:25:30 +02003350 {
3351 home_replace(NULL, SCRIPT_ITEM(i).sn_name,
3352 NameBuff, MAXPATHL, TRUE);
3353 smsg((char_u *)"%3d: %s", i, NameBuff);
Bram Moolenaar970a1b82012-03-23 18:39:18 +01003354 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003355}
3356
3357# if defined(BACKSLASH_IN_FILENAME) || defined(PROTO)
3358/*
3359 * Fix slashes in the list of script names for 'shellslash'.
3360 */
3361 void
3362scriptnames_slash_adjust()
3363{
3364 int i;
3365
Bram Moolenaar05159a02005-02-26 23:04:13 +00003366 for (i = 1; i <= script_items.ga_len; ++i)
3367 if (SCRIPT_ITEM(i).sn_name != NULL)
3368 slash_adjust(SCRIPT_ITEM(i).sn_name);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003369}
3370# endif
3371
3372/*
3373 * Get a pointer to a script name. Used for ":verbose set".
3374 */
3375 char_u *
3376get_scriptname(id)
3377 scid_T id;
3378{
3379 if (id == SID_MODELINE)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00003380 return (char_u *)_("modeline");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003381 if (id == SID_CMDARG)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00003382 return (char_u *)_("--cmd argument");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003383 if (id == SID_CARG)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00003384 return (char_u *)_("-c argument");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003385 if (id == SID_ENV)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00003386 return (char_u *)_("environment variable");
3387 if (id == SID_ERROR)
3388 return (char_u *)_("error handler");
Bram Moolenaar05159a02005-02-26 23:04:13 +00003389 return SCRIPT_ITEM(id).sn_name;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003390}
Bram Moolenaar05159a02005-02-26 23:04:13 +00003391
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00003392# if defined(EXITFREE) || defined(PROTO)
3393 void
3394free_scriptnames()
3395{
3396 int i;
3397
3398 for (i = script_items.ga_len; i > 0; --i)
3399 vim_free(SCRIPT_ITEM(i).sn_name);
3400 ga_clear(&script_items);
3401}
3402# endif
3403
Bram Moolenaar071d4272004-06-13 20:20:40 +00003404#endif
3405
3406#if defined(USE_CR) || defined(PROTO)
3407
3408# if defined(__MSL__) && (__MSL__ >= 22)
3409/*
3410 * Newer version of the Metrowerks library handle DOS and UNIX files
3411 * without help.
3412 * Test with earlier versions, MSL 2.2 is the library supplied with
3413 * Codewarrior Pro 2.
3414 */
3415 char *
3416fgets_cr(s, n, stream)
3417 char *s;
3418 int n;
3419 FILE *stream;
3420{
3421 return fgets(s, n, stream);
3422}
3423# else
3424/*
3425 * Version of fgets() which also works for lines ending in a <CR> only
3426 * (Macintosh format).
3427 * For older versions of the Metrowerks library.
3428 * At least CodeWarrior 9 needed this code.
3429 */
3430 char *
3431fgets_cr(s, n, stream)
3432 char *s;
3433 int n;
3434 FILE *stream;
3435{
3436 int c = 0;
3437 int char_read = 0;
3438
3439 while (!feof(stream) && c != '\r' && c != '\n' && char_read < n - 1)
3440 {
3441 c = fgetc(stream);
3442 s[char_read++] = c;
3443 /* If the file is in DOS format, we need to skip a NL after a CR. I
3444 * thought it was the other way around, but this appears to work... */
3445 if (c == '\n')
3446 {
3447 c = fgetc(stream);
3448 if (c != '\r')
3449 ungetc(c, stream);
3450 }
3451 }
3452
3453 s[char_read] = 0;
3454 if (char_read == 0)
3455 return NULL;
3456
3457 if (feof(stream) && char_read == 1)
3458 return NULL;
3459
3460 return s;
3461}
3462# endif
3463#endif
3464
3465/*
3466 * Get one full line from a sourced file.
3467 * Called by do_cmdline() when it's called from do_source().
3468 *
3469 * Return a pointer to the line in allocated memory.
3470 * Return NULL for end-of-file or some error.
3471 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003472 char_u *
3473getsourceline(c, cookie, indent)
Bram Moolenaar0c094b92009-05-14 20:20:33 +00003474 int c UNUSED;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003475 void *cookie;
Bram Moolenaar0c094b92009-05-14 20:20:33 +00003476 int indent UNUSED;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003477{
3478 struct source_cookie *sp = (struct source_cookie *)cookie;
3479 char_u *line;
Bram Moolenaarc047b9a2012-02-11 20:40:55 +01003480 char_u *p;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003481
3482#ifdef FEAT_EVAL
3483 /* If breakpoints have been added/deleted need to check for it. */
3484 if (sp->dbg_tick < debug_tick)
3485 {
3486 sp->breakpoint = dbg_find_breakpoint(TRUE, sp->fname, sourcing_lnum);
3487 sp->dbg_tick = debug_tick;
3488 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00003489# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00003490 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003491 script_line_end();
3492# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003493#endif
3494 /*
3495 * Get current line. If there is a read-ahead line, use it, otherwise get
3496 * one now.
3497 */
3498 if (sp->finished)
3499 line = NULL;
3500 else if (sp->nextline == NULL)
3501 line = get_one_sourceline(sp);
3502 else
3503 {
3504 line = sp->nextline;
3505 sp->nextline = NULL;
3506 ++sourcing_lnum;
3507 }
Bram Moolenaare2cc9702005-03-15 22:43:58 +00003508#ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00003509 if (line != NULL && do_profiling == PROF_YES)
Bram Moolenaare2cc9702005-03-15 22:43:58 +00003510 script_line_start();
3511#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003512
3513 /* Only concatenate lines starting with a \ when 'cpoptions' doesn't
3514 * contain the 'C' flag. */
3515 if (line != NULL && (vim_strchr(p_cpo, CPO_CONCAT) == NULL))
3516 {
3517 /* compensate for the one line read-ahead */
3518 --sourcing_lnum;
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01003519
3520 /* Get the next line and concatenate it when it starts with a
3521 * backslash. We always need to read the next line, keep it in
3522 * sp->nextline. */
3523 sp->nextline = get_one_sourceline(sp);
3524 if (sp->nextline != NULL && *(p = skipwhite(sp->nextline)) == '\\')
Bram Moolenaar071d4272004-06-13 20:20:40 +00003525 {
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01003526 garray_T ga;
3527
Bram Moolenaarb549a732012-02-22 18:29:33 +01003528 ga_init2(&ga, (int)sizeof(char_u), 400);
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01003529 ga_concat(&ga, line);
3530 ga_concat(&ga, p + 1);
3531 for (;;)
3532 {
3533 vim_free(sp->nextline);
3534 sp->nextline = get_one_sourceline(sp);
3535 if (sp->nextline == NULL)
3536 break;
3537 p = skipwhite(sp->nextline);
3538 if (*p != '\\')
3539 break;
Bram Moolenaarb549a732012-02-22 18:29:33 +01003540 /* Adjust the growsize to the current length to speed up
3541 * concatenating many lines. */
3542 if (ga.ga_len > 400)
3543 {
3544 if (ga.ga_len > 8000)
3545 ga.ga_growsize = 8000;
3546 else
3547 ga.ga_growsize = ga.ga_len;
3548 }
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01003549 ga_concat(&ga, p + 1);
3550 }
3551 ga_append(&ga, NUL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003552 vim_free(line);
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01003553 line = ga.ga_data;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003554 }
3555 }
3556
3557#ifdef FEAT_MBYTE
3558 if (line != NULL && sp->conv.vc_type != CONV_NONE)
3559 {
Bram Moolenaarc047b9a2012-02-11 20:40:55 +01003560 char_u *s;
3561
Bram Moolenaar071d4272004-06-13 20:20:40 +00003562 /* Convert the encoding of the script line. */
3563 s = string_convert(&sp->conv, line, NULL);
3564 if (s != NULL)
3565 {
3566 vim_free(line);
3567 line = s;
3568 }
3569 }
3570#endif
3571
3572#ifdef FEAT_EVAL
3573 /* Did we encounter a breakpoint? */
3574 if (sp->breakpoint != 0 && sp->breakpoint <= sourcing_lnum)
3575 {
3576 dbg_breakpoint(sp->fname, sourcing_lnum);
3577 /* Find next breakpoint. */
3578 sp->breakpoint = dbg_find_breakpoint(TRUE, sp->fname, sourcing_lnum);
3579 sp->dbg_tick = debug_tick;
3580 }
3581#endif
3582
3583 return line;
3584}
3585
3586 static char_u *
3587get_one_sourceline(sp)
3588 struct source_cookie *sp;
3589{
3590 garray_T ga;
3591 int len;
3592 int c;
3593 char_u *buf;
3594#ifdef USE_CRNL
3595 int has_cr; /* CR-LF found */
3596#endif
3597#ifdef USE_CR
3598 char_u *scan;
3599#endif
3600 int have_read = FALSE;
3601
3602 /* use a growarray to store the sourced line */
Bram Moolenaar6ac54292005-02-02 23:07:25 +00003603 ga_init2(&ga, 1, 250);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003604
3605 /*
3606 * Loop until there is a finished line (or end-of-file).
3607 */
3608 sourcing_lnum++;
3609 for (;;)
3610 {
Bram Moolenaar6ac54292005-02-02 23:07:25 +00003611 /* make room to read at least 120 (more) characters */
3612 if (ga_grow(&ga, 120) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003613 break;
3614 buf = (char_u *)ga.ga_data;
3615
3616#ifdef USE_CR
3617 if (sp->fileformat == EOL_MAC)
3618 {
Bram Moolenaar86b68352004-12-27 21:59:20 +00003619 if (fgets_cr((char *)buf + ga.ga_len, ga.ga_maxlen - ga.ga_len,
3620 sp->fp) == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003621 break;
3622 }
3623 else
3624#endif
Bram Moolenaar86b68352004-12-27 21:59:20 +00003625 if (fgets((char *)buf + ga.ga_len, ga.ga_maxlen - ga.ga_len,
3626 sp->fp) == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003627 break;
Bram Moolenaar6ac54292005-02-02 23:07:25 +00003628 len = ga.ga_len + (int)STRLEN(buf + ga.ga_len);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003629#ifdef USE_CRNL
3630 /* Ignore a trailing CTRL-Z, when in Dos mode. Only recognize the
3631 * CTRL-Z by its own, or after a NL. */
3632 if ( (len == 1 || (len >= 2 && buf[len - 2] == '\n'))
3633 && sp->fileformat == EOL_DOS
3634 && buf[len - 1] == Ctrl_Z)
3635 {
3636 buf[len - 1] = NUL;
3637 break;
3638 }
3639#endif
3640
3641#ifdef USE_CR
3642 /* If the read doesn't stop on a new line, and there's
3643 * some CR then we assume a Mac format */
3644 if (sp->fileformat == EOL_UNKNOWN)
3645 {
3646 if (buf[len - 1] != '\n' && vim_strchr(buf, '\r') != NULL)
3647 sp->fileformat = EOL_MAC;
3648 else
3649 sp->fileformat = EOL_UNIX;
3650 }
3651
3652 if (sp->fileformat == EOL_MAC)
3653 {
3654 scan = vim_strchr(buf, '\r');
3655
3656 if (scan != NULL)
3657 {
3658 *scan = '\n';
3659 if (*(scan + 1) != 0)
3660 {
3661 *(scan + 1) = 0;
3662 fseek(sp->fp, (long)(scan - buf - len + 1), SEEK_CUR);
3663 }
3664 }
3665 len = STRLEN(buf);
3666 }
3667#endif
3668
3669 have_read = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003670 ga.ga_len = len;
3671
3672 /* If the line was longer than the buffer, read more. */
Bram Moolenaar86b68352004-12-27 21:59:20 +00003673 if (ga.ga_maxlen - ga.ga_len == 1 && buf[len - 1] != '\n')
Bram Moolenaar071d4272004-06-13 20:20:40 +00003674 continue;
3675
3676 if (len >= 1 && buf[len - 1] == '\n') /* remove trailing NL */
3677 {
3678#ifdef USE_CRNL
3679 has_cr = (len >= 2 && buf[len - 2] == '\r');
3680 if (sp->fileformat == EOL_UNKNOWN)
3681 {
3682 if (has_cr)
3683 sp->fileformat = EOL_DOS;
3684 else
3685 sp->fileformat = EOL_UNIX;
3686 }
3687
3688 if (sp->fileformat == EOL_DOS)
3689 {
3690 if (has_cr) /* replace trailing CR */
3691 {
3692 buf[len - 2] = '\n';
3693 --len;
3694 --ga.ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003695 }
3696 else /* lines like ":map xx yy^M" will have failed */
3697 {
3698 if (!sp->error)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00003699 {
3700 msg_source(hl_attr(HLF_W));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003701 EMSG(_("W15: Warning: Wrong line separator, ^M may be missing"));
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00003702 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003703 sp->error = TRUE;
3704 sp->fileformat = EOL_UNIX;
3705 }
3706 }
3707#endif
3708 /* The '\n' is escaped if there is an odd number of ^V's just
3709 * before it, first set "c" just before the 'V's and then check
3710 * len&c parities (is faster than ((len-c)%2 == 0)) -- Acevedo */
3711 for (c = len - 2; c >= 0 && buf[c] == Ctrl_V; c--)
3712 ;
3713 if ((len & 1) != (c & 1)) /* escaped NL, read more */
3714 {
3715 sourcing_lnum++;
3716 continue;
3717 }
3718
3719 buf[len - 1] = NUL; /* remove the NL */
3720 }
3721
3722 /*
3723 * Check for ^C here now and then, so recursive :so can be broken.
3724 */
3725 line_breakcheck();
3726 break;
3727 }
3728
3729 if (have_read)
3730 return (char_u *)ga.ga_data;
3731
3732 vim_free(ga.ga_data);
3733 return NULL;
3734}
3735
Bram Moolenaar05159a02005-02-26 23:04:13 +00003736#if defined(FEAT_PROFILE) || defined(PROTO)
3737/*
3738 * Called when starting to read a script line.
3739 * "sourcing_lnum" must be correct!
3740 * When skipping lines it may not actually be executed, but we won't find out
3741 * until later and we need to store the time now.
3742 */
3743 void
3744script_line_start()
3745{
3746 scriptitem_T *si;
3747 sn_prl_T *pp;
3748
3749 if (current_SID <= 0 || current_SID > script_items.ga_len)
3750 return;
3751 si = &SCRIPT_ITEM(current_SID);
3752 if (si->sn_prof_on && sourcing_lnum >= 1)
3753 {
Bram Moolenaar8c8de832008-06-24 22:58:06 +00003754 /* Grow the array before starting the timer, so that the time spent
Bram Moolenaar05159a02005-02-26 23:04:13 +00003755 * here isn't counted. */
3756 ga_grow(&si->sn_prl_ga, (int)(sourcing_lnum - si->sn_prl_ga.ga_len));
3757 si->sn_prl_idx = sourcing_lnum - 1;
3758 while (si->sn_prl_ga.ga_len <= si->sn_prl_idx
3759 && si->sn_prl_ga.ga_len < si->sn_prl_ga.ga_maxlen)
3760 {
3761 /* Zero counters for a line that was not used before. */
3762 pp = &PRL_ITEM(si, si->sn_prl_ga.ga_len);
3763 pp->snp_count = 0;
3764 profile_zero(&pp->sn_prl_total);
3765 profile_zero(&pp->sn_prl_self);
3766 ++si->sn_prl_ga.ga_len;
3767 }
3768 si->sn_prl_execed = FALSE;
3769 profile_start(&si->sn_prl_start);
3770 profile_zero(&si->sn_prl_children);
3771 profile_get_wait(&si->sn_prl_wait);
3772 }
3773}
3774
3775/*
3776 * Called when actually executing a function line.
3777 */
3778 void
3779script_line_exec()
3780{
3781 scriptitem_T *si;
3782
3783 if (current_SID <= 0 || current_SID > script_items.ga_len)
3784 return;
3785 si = &SCRIPT_ITEM(current_SID);
3786 if (si->sn_prof_on && si->sn_prl_idx >= 0)
3787 si->sn_prl_execed = TRUE;
3788}
3789
3790/*
3791 * Called when done with a function line.
3792 */
3793 void
3794script_line_end()
3795{
3796 scriptitem_T *si;
3797 sn_prl_T *pp;
3798
3799 if (current_SID <= 0 || current_SID > script_items.ga_len)
3800 return;
3801 si = &SCRIPT_ITEM(current_SID);
3802 if (si->sn_prof_on && si->sn_prl_idx >= 0
3803 && si->sn_prl_idx < si->sn_prl_ga.ga_len)
3804 {
3805 if (si->sn_prl_execed)
3806 {
3807 pp = &PRL_ITEM(si, si->sn_prl_idx);
3808 ++pp->snp_count;
3809 profile_end(&si->sn_prl_start);
3810 profile_sub_wait(&si->sn_prl_wait, &si->sn_prl_start);
Bram Moolenaar05159a02005-02-26 23:04:13 +00003811 profile_add(&pp->sn_prl_total, &si->sn_prl_start);
Bram Moolenaar1056d982006-03-09 22:37:52 +00003812 profile_self(&pp->sn_prl_self, &si->sn_prl_start,
3813 &si->sn_prl_children);
Bram Moolenaar05159a02005-02-26 23:04:13 +00003814 }
3815 si->sn_prl_idx = -1;
3816 }
3817}
3818#endif
3819
Bram Moolenaar071d4272004-06-13 20:20:40 +00003820/*
3821 * ":scriptencoding": Set encoding conversion for a sourced script.
3822 * Without the multi-byte feature it's simply ignored.
3823 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003824 void
3825ex_scriptencoding(eap)
Bram Moolenaar0c094b92009-05-14 20:20:33 +00003826 exarg_T *eap UNUSED;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003827{
3828#ifdef FEAT_MBYTE
3829 struct source_cookie *sp;
3830 char_u *name;
3831
3832 if (!getline_equal(eap->getline, eap->cookie, getsourceline))
3833 {
3834 EMSG(_("E167: :scriptencoding used outside of a sourced file"));
3835 return;
3836 }
3837
3838 if (*eap->arg != NUL)
3839 {
3840 name = enc_canonize(eap->arg);
3841 if (name == NULL) /* out of memory */
3842 return;
3843 }
3844 else
3845 name = eap->arg;
3846
3847 /* Setup for conversion from the specified encoding to 'encoding'. */
3848 sp = (struct source_cookie *)getline_cookie(eap->getline, eap->cookie);
3849 convert_setup(&sp->conv, name, p_enc);
3850
3851 if (name != eap->arg)
3852 vim_free(name);
3853#endif
3854}
3855
3856#if defined(FEAT_EVAL) || defined(PROTO)
3857/*
3858 * ":finish": Mark a sourced file as finished.
3859 */
3860 void
3861ex_finish(eap)
3862 exarg_T *eap;
3863{
3864 if (getline_equal(eap->getline, eap->cookie, getsourceline))
3865 do_finish(eap, FALSE);
3866 else
3867 EMSG(_("E168: :finish used outside of a sourced file"));
3868}
3869
3870/*
3871 * Mark a sourced file as finished. Possibly makes the ":finish" pending.
3872 * Also called for a pending finish at the ":endtry" or after returning from
3873 * an extra do_cmdline(). "reanimate" is used in the latter case.
3874 */
3875 void
3876do_finish(eap, reanimate)
3877 exarg_T *eap;
3878 int reanimate;
3879{
3880 int idx;
3881
3882 if (reanimate)
3883 ((struct source_cookie *)getline_cookie(eap->getline,
3884 eap->cookie))->finished = FALSE;
3885
3886 /*
3887 * Cleanup (and inactivate) conditionals, but stop when a try conditional
3888 * not in its finally clause (which then is to be executed next) is found.
3889 * In this case, make the ":finish" pending for execution at the ":endtry".
3890 * Otherwise, finish normally.
3891 */
3892 idx = cleanup_conditionals(eap->cstack, 0, TRUE);
3893 if (idx >= 0)
3894 {
3895 eap->cstack->cs_pending[idx] = CSTP_FINISH;
3896 report_make_pending(CSTP_FINISH, NULL);
3897 }
3898 else
3899 ((struct source_cookie *)getline_cookie(eap->getline,
3900 eap->cookie))->finished = TRUE;
3901}
3902
3903
3904/*
3905 * Return TRUE when a sourced file had the ":finish" command: Don't give error
3906 * message for missing ":endif".
3907 * Return FALSE when not sourcing a file.
3908 */
3909 int
Bram Moolenaar89d40322006-08-29 15:30:07 +00003910source_finished(fgetline, cookie)
3911 char_u *(*fgetline) __ARGS((int, void *, int));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003912 void *cookie;
3913{
Bram Moolenaar89d40322006-08-29 15:30:07 +00003914 return (getline_equal(fgetline, cookie, getsourceline)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003915 && ((struct source_cookie *)getline_cookie(
Bram Moolenaar89d40322006-08-29 15:30:07 +00003916 fgetline, cookie))->finished);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003917}
3918#endif
3919
3920#if defined(FEAT_LISTCMDS) || defined(PROTO)
3921/*
3922 * ":checktime [buffer]"
3923 */
3924 void
3925ex_checktime(eap)
3926 exarg_T *eap;
3927{
3928 buf_T *buf;
3929 int save_no_check_timestamps = no_check_timestamps;
3930
3931 no_check_timestamps = 0;
3932 if (eap->addr_count == 0) /* default is all buffers */
3933 check_timestamps(FALSE);
3934 else
3935 {
3936 buf = buflist_findnr((int)eap->line2);
3937 if (buf != NULL) /* cannot happen? */
3938 (void)buf_check_timestamp(buf, FALSE);
3939 }
3940 no_check_timestamps = save_no_check_timestamps;
3941}
3942#endif
3943
Bram Moolenaar071d4272004-06-13 20:20:40 +00003944#if (defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
3945 && (defined(FEAT_EVAL) || defined(FEAT_MULTI_LANG))
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02003946# define HAVE_GET_LOCALE_VAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00003947static char *get_locale_val __ARGS((int what));
3948
3949 static char *
3950get_locale_val(what)
3951 int what;
3952{
3953 char *loc;
3954
3955 /* Obtain the locale value from the libraries. For DJGPP this is
3956 * redefined and it doesn't use the arguments. */
3957 loc = setlocale(what, NULL);
3958
Bram Moolenaar61660ea2006-04-07 21:40:07 +00003959# ifdef WIN32
Bram Moolenaar071d4272004-06-13 20:20:40 +00003960 if (loc != NULL)
3961 {
3962 char_u *p;
3963
Bram Moolenaar61660ea2006-04-07 21:40:07 +00003964 /* setocale() returns something like "LC_COLLATE=<name>;LC_..." when
3965 * one of the values (e.g., LC_CTYPE) differs. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003966 p = vim_strchr(loc, '=');
3967 if (p != NULL)
3968 {
3969 loc = ++p;
3970 while (*p != NUL) /* remove trailing newline */
3971 {
Bram Moolenaar61660ea2006-04-07 21:40:07 +00003972 if (*p < ' ' || *p == ';')
Bram Moolenaar071d4272004-06-13 20:20:40 +00003973 {
3974 *p = NUL;
3975 break;
3976 }
3977 ++p;
3978 }
3979 }
3980 }
3981# endif
3982
3983 return loc;
3984}
3985#endif
3986
3987
3988#ifdef WIN32
3989/*
3990 * On MS-Windows locale names are strings like "German_Germany.1252", but
3991 * gettext expects "de". Try to translate one into another here for a few
3992 * supported languages.
3993 */
3994 static char_u *
3995gettext_lang(char_u *name)
3996{
3997 int i;
3998 static char *(mtable[]) = {
3999 "afrikaans", "af",
4000 "czech", "cs",
4001 "dutch", "nl",
4002 "german", "de",
4003 "english_united kingdom", "en_GB",
4004 "spanish", "es",
4005 "french", "fr",
4006 "italian", "it",
4007 "japanese", "ja",
4008 "korean", "ko",
4009 "norwegian", "no",
4010 "polish", "pl",
4011 "russian", "ru",
4012 "slovak", "sk",
4013 "swedish", "sv",
4014 "ukrainian", "uk",
4015 "chinese_china", "zh_CN",
4016 "chinese_taiwan", "zh_TW",
4017 NULL};
4018
4019 for (i = 0; mtable[i] != NULL; i += 2)
4020 if (STRNICMP(mtable[i], name, STRLEN(mtable[i])) == 0)
4021 return mtable[i + 1];
4022 return name;
4023}
4024#endif
4025
4026#if defined(FEAT_MULTI_LANG) || defined(PROTO)
4027/*
4028 * Obtain the current messages language. Used to set the default for
4029 * 'helplang'. May return NULL or an empty string.
4030 */
4031 char_u *
4032get_mess_lang()
4033{
4034 char_u *p;
4035
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004036# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00004037# if defined(LC_MESSAGES)
4038 p = (char_u *)get_locale_val(LC_MESSAGES);
4039# else
4040 /* This is necessary for Win32, where LC_MESSAGES is not defined and $LANG
Bram Moolenaar61660ea2006-04-07 21:40:07 +00004041 * may be set to the LCID number. LC_COLLATE is the best guess, LC_TIME
4042 * and LC_MONETARY may be set differently for a Japanese working in the
4043 * US. */
4044 p = (char_u *)get_locale_val(LC_COLLATE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004045# endif
4046# else
4047 p = mch_getenv((char_u *)"LC_ALL");
4048 if (p == NULL || *p == NUL)
4049 {
4050 p = mch_getenv((char_u *)"LC_MESSAGES");
4051 if (p == NULL || *p == NUL)
4052 p = mch_getenv((char_u *)"LANG");
4053 }
4054# endif
4055# ifdef WIN32
4056 p = gettext_lang(p);
4057# endif
4058 return p;
4059}
4060#endif
4061
Bram Moolenaardef9e822004-12-31 20:58:58 +00004062/* Complicated #if; matches with where get_mess_env() is used below. */
4063#if (defined(FEAT_EVAL) && !((defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
4064 && defined(LC_MESSAGES))) \
4065 || ((defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
4066 && (defined(FEAT_GETTEXT) || defined(FEAT_MBYTE)) \
4067 && !defined(LC_MESSAGES))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004068static char_u *get_mess_env __ARGS((void));
4069
4070/*
4071 * Get the language used for messages from the environment.
4072 */
4073 static char_u *
4074get_mess_env()
4075{
4076 char_u *p;
4077
4078 p = mch_getenv((char_u *)"LC_ALL");
4079 if (p == NULL || *p == NUL)
4080 {
4081 p = mch_getenv((char_u *)"LC_MESSAGES");
4082 if (p == NULL || *p == NUL)
4083 {
4084 p = mch_getenv((char_u *)"LANG");
4085 if (p != NULL && VIM_ISDIGIT(*p))
4086 p = NULL; /* ignore something like "1043" */
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004087# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00004088 if (p == NULL || *p == NUL)
4089 p = (char_u *)get_locale_val(LC_CTYPE);
4090# endif
4091 }
4092 }
4093 return p;
4094}
4095#endif
4096
4097#if defined(FEAT_EVAL) || defined(PROTO)
4098
4099/*
4100 * Set the "v:lang" variable according to the current locale setting.
4101 * Also do "v:lc_time"and "v:ctype".
4102 */
4103 void
4104set_lang_var()
4105{
4106 char_u *loc;
4107
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004108# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00004109 loc = (char_u *)get_locale_val(LC_CTYPE);
4110# else
4111 /* setlocale() not supported: use the default value */
4112 loc = (char_u *)"C";
4113# endif
4114 set_vim_var_string(VV_CTYPE, loc, -1);
4115
4116 /* When LC_MESSAGES isn't defined use the value from $LC_MESSAGES, fall
4117 * back to LC_CTYPE if it's empty. */
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004118# if defined(HAVE_GET_LOCALE_VAL) && defined(LC_MESSAGES)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004119 loc = (char_u *)get_locale_val(LC_MESSAGES);
4120# else
4121 loc = get_mess_env();
4122# endif
4123 set_vim_var_string(VV_LANG, loc, -1);
4124
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004125# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00004126 loc = (char_u *)get_locale_val(LC_TIME);
4127# endif
4128 set_vim_var_string(VV_LC_TIME, loc, -1);
4129}
4130#endif
4131
4132#if (defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
4133 && (defined(FEAT_GETTEXT) || defined(FEAT_MBYTE))
4134/*
4135 * ":language": Set the language (locale).
4136 */
4137 void
4138ex_language(eap)
4139 exarg_T *eap;
4140{
4141 char *loc;
4142 char_u *p;
4143 char_u *name;
4144 int what = LC_ALL;
4145 char *whatstr = "";
4146#ifdef LC_MESSAGES
4147# define VIM_LC_MESSAGES LC_MESSAGES
4148#else
4149# define VIM_LC_MESSAGES 6789
4150#endif
4151
4152 name = eap->arg;
4153
4154 /* Check for "messages {name}", "ctype {name}" or "time {name}" argument.
4155 * Allow abbreviation, but require at least 3 characters to avoid
4156 * confusion with a two letter language name "me" or "ct". */
4157 p = skiptowhite(eap->arg);
4158 if ((*p == NUL || vim_iswhite(*p)) && p - eap->arg >= 3)
4159 {
4160 if (STRNICMP(eap->arg, "messages", p - eap->arg) == 0)
4161 {
4162 what = VIM_LC_MESSAGES;
4163 name = skipwhite(p);
4164 whatstr = "messages ";
4165 }
4166 else if (STRNICMP(eap->arg, "ctype", p - eap->arg) == 0)
4167 {
4168 what = LC_CTYPE;
4169 name = skipwhite(p);
4170 whatstr = "ctype ";
4171 }
4172 else if (STRNICMP(eap->arg, "time", p - eap->arg) == 0)
4173 {
4174 what = LC_TIME;
4175 name = skipwhite(p);
4176 whatstr = "time ";
4177 }
4178 }
4179
4180 if (*name == NUL)
4181 {
4182#ifndef LC_MESSAGES
4183 if (what == VIM_LC_MESSAGES)
4184 p = get_mess_env();
4185 else
4186#endif
4187 p = (char_u *)setlocale(what, NULL);
4188 if (p == NULL || *p == NUL)
4189 p = (char_u *)"Unknown";
4190 smsg((char_u *)_("Current %slanguage: \"%s\""), whatstr, p);
4191 }
4192 else
4193 {
4194#ifndef LC_MESSAGES
4195 if (what == VIM_LC_MESSAGES)
4196 loc = "";
4197 else
4198#endif
Bram Moolenaar8c8de832008-06-24 22:58:06 +00004199 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00004200 loc = setlocale(what, (char *)name);
Bram Moolenaar8c8de832008-06-24 22:58:06 +00004201#if defined(FEAT_FLOAT) && defined(LC_NUMERIC)
4202 /* Make sure strtod() uses a decimal point, not a comma. */
4203 setlocale(LC_NUMERIC, "C");
4204#endif
4205 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004206 if (loc == NULL)
4207 EMSG2(_("E197: Cannot set language to \"%s\""), name);
4208 else
4209 {
4210#ifdef HAVE_NL_MSG_CAT_CNTR
4211 /* Need to do this for GNU gettext, otherwise cached translations
4212 * will be used again. */
4213 extern int _nl_msg_cat_cntr;
4214
4215 ++_nl_msg_cat_cntr;
4216#endif
Bram Moolenaarb8017e72007-05-10 18:59:07 +00004217 /* Reset $LC_ALL, otherwise it would overrule everything. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004218 vim_setenv((char_u *)"LC_ALL", (char_u *)"");
4219
4220 if (what != LC_TIME)
4221 {
4222 /* Tell gettext() what to translate to. It apparently doesn't
4223 * use the currently effective locale. Also do this when
4224 * FEAT_GETTEXT isn't defined, so that shell commands use this
4225 * value. */
4226 if (what == LC_ALL)
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00004227 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00004228 vim_setenv((char_u *)"LANG", name);
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00004229# ifdef WIN32
4230 /* Apparently MS-Windows printf() may cause a crash when
4231 * we give it 8-bit text while it's expecting text in the
4232 * current locale. This call avoids that. */
4233 setlocale(LC_CTYPE, "C");
4234# endif
4235 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004236 if (what != LC_CTYPE)
4237 {
4238 char_u *mname;
4239#ifdef WIN32
4240 mname = gettext_lang(name);
4241#else
4242 mname = name;
4243#endif
4244 vim_setenv((char_u *)"LC_MESSAGES", mname);
4245#ifdef FEAT_MULTI_LANG
4246 set_helplang_default(mname);
4247#endif
4248 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004249 }
4250
4251# ifdef FEAT_EVAL
4252 /* Set v:lang, v:lc_time and v:ctype to the final result. */
4253 set_lang_var();
4254# endif
Bram Moolenaar6cc00c72011-10-20 21:41:09 +02004255# ifdef FEAT_TITLE
4256 maketitle();
4257# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004258 }
4259 }
4260}
4261
4262# if defined(FEAT_CMDL_COMPL) || defined(PROTO)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02004263
4264static char_u **locales = NULL; /* Array of all available locales */
4265static int did_init_locales = FALSE;
4266
4267static void init_locales __ARGS((void));
4268static char_u **find_locales __ARGS((void));
4269
4270/*
4271 * Lazy initialization of all available locales.
4272 */
4273 static void
4274init_locales()
4275{
4276 if (!did_init_locales)
4277 {
4278 did_init_locales = TRUE;
4279 locales = find_locales();
4280 }
4281}
4282
4283/* Return an array of strings for all available locales + NULL for the
4284 * last element. Return NULL in case of error. */
4285 static char_u **
4286find_locales()
4287{
4288 garray_T locales_ga;
4289 char_u *loc;
4290
4291 /* Find all available locales by running command "locale -a". If this
4292 * doesn't work we won't have completion. */
4293 char_u *locale_a = get_cmd_output((char_u *)"locale -a",
4294 NULL, SHELL_SILENT);
4295 if (locale_a == NULL)
4296 return NULL;
4297 ga_init2(&locales_ga, sizeof(char_u *), 20);
4298
4299 /* Transform locale_a string where each locale is separated by "\n"
4300 * into an array of locale strings. */
4301 loc = (char_u *)strtok((char *)locale_a, "\n");
4302
4303 while (loc != NULL)
4304 {
4305 if (ga_grow(&locales_ga, 1) == FAIL)
4306 break;
4307 loc = vim_strsave(loc);
4308 if (loc == NULL)
4309 break;
4310
4311 ((char_u **)locales_ga.ga_data)[locales_ga.ga_len++] = loc;
4312 loc = (char_u *)strtok(NULL, "\n");
4313 }
4314 vim_free(locale_a);
4315 if (ga_grow(&locales_ga, 1) == FAIL)
4316 {
4317 ga_clear(&locales_ga);
4318 return NULL;
4319 }
4320 ((char_u **)locales_ga.ga_data)[locales_ga.ga_len] = NULL;
4321 return (char_u **)locales_ga.ga_data;
4322}
4323
4324# if defined(EXITFREE) || defined(PROTO)
4325 void
4326free_locales()
4327{
4328 int i;
4329 if (locales != NULL)
4330 {
4331 for (i = 0; locales[i] != NULL; i++)
4332 vim_free(locales[i]);
4333 vim_free(locales);
4334 locales = NULL;
4335 }
4336}
4337# endif
4338
Bram Moolenaar071d4272004-06-13 20:20:40 +00004339/*
4340 * Function given to ExpandGeneric() to obtain the possible arguments of the
4341 * ":language" command.
4342 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004343 char_u *
4344get_lang_arg(xp, idx)
Bram Moolenaar0c094b92009-05-14 20:20:33 +00004345 expand_T *xp UNUSED;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004346 int idx;
4347{
4348 if (idx == 0)
4349 return (char_u *)"messages";
4350 if (idx == 1)
4351 return (char_u *)"ctype";
4352 if (idx == 2)
4353 return (char_u *)"time";
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02004354
4355 init_locales();
4356 if (locales == NULL)
4357 return NULL;
4358 return locales[idx - 3];
4359}
4360
4361/*
4362 * Function given to ExpandGeneric() to obtain the available locales.
4363 */
4364 char_u *
4365get_locales(xp, idx)
4366 expand_T *xp UNUSED;
4367 int idx;
4368{
4369 init_locales();
4370 if (locales == NULL)
4371 return NULL;
4372 return locales[idx];
Bram Moolenaar071d4272004-06-13 20:20:40 +00004373}
4374# endif
4375
4376#endif