blob: de190e78c392c43193c25a6a7432f46629937d7c [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\""),
Bram Moolenaare1704ba2012-10-03 18:25:00 +02001691 buf_spname(buf) != NULL ? buf_spname(buf) : buf->b_fname))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001692 {
1693 save = no_wait_return;
1694 no_wait_return = FALSE;
1695 wait_return(FALSE);
1696 no_wait_return = save;
1697 }
1698 }
1699
1700#ifdef FEAT_WINDOWS
1701 /* Try to find a window that contains the buffer. */
1702 if (buf != curbuf)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001703 FOR_ALL_TAB_WINDOWS(tp, wp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001704 if (wp->w_buffer == buf)
1705 {
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001706 goto_tabpage_win(tp, wp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001707# ifdef FEAT_AUTOCMD
1708 /* Paranoia: did autocms wipe out the buffer with changes? */
1709 if (!buf_valid(buf))
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001710 {
1711 goto theend;
1712 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001713# endif
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001714 goto buf_found;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001715 }
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001716buf_found:
Bram Moolenaar071d4272004-06-13 20:20:40 +00001717#endif
1718
1719 /* Open the changed buffer in the current window. */
1720 if (buf != curbuf)
1721 set_curbuf(buf, DOBUF_GOTO);
1722
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001723theend:
1724 vim_free(bufnrs);
1725 return ret;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001726}
1727
1728/*
1729 * return FAIL if there is no file name, OK if there is one
1730 * give error message for FAIL
1731 */
1732 int
1733check_fname()
1734{
1735 if (curbuf->b_ffname == NULL)
1736 {
1737 EMSG(_(e_noname));
1738 return FAIL;
1739 }
1740 return OK;
1741}
1742
1743/*
1744 * flush the contents of a buffer, unless it has no file name
1745 *
1746 * return FAIL for failure, OK otherwise
1747 */
1748 int
1749buf_write_all(buf, forceit)
1750 buf_T *buf;
1751 int forceit;
1752{
1753 int retval;
1754#ifdef FEAT_AUTOCMD
1755 buf_T *old_curbuf = curbuf;
1756#endif
1757
1758 retval = (buf_write(buf, buf->b_ffname, buf->b_fname,
1759 (linenr_T)1, buf->b_ml.ml_line_count, NULL,
1760 FALSE, forceit, TRUE, FALSE));
1761#ifdef FEAT_AUTOCMD
1762 if (curbuf != old_curbuf)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001763 {
1764 msg_source(hl_attr(HLF_W));
Bram Moolenaar071d4272004-06-13 20:20:40 +00001765 MSG(_("Warning: Entered other buffer unexpectedly (check autocommands)"));
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001766 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001767#endif
1768 return retval;
1769}
1770
1771/*
1772 * Code to handle the argument list.
1773 */
1774
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001775static char_u *do_one_arg __ARGS((char_u *str));
1776static int do_arglist __ARGS((char_u *str, int what, int after));
1777static void alist_check_arg_idx __ARGS((void));
1778static int editing_arg_idx __ARGS((win_T *win));
1779#ifdef FEAT_LISTCMDS
1780static int alist_add_list __ARGS((int count, char_u **files, int after));
1781#endif
1782#define AL_SET 1
1783#define AL_ADD 2
1784#define AL_DEL 3
1785
Bram Moolenaar071d4272004-06-13 20:20:40 +00001786/*
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001787 * Isolate one argument, taking backticks.
1788 * Changes the argument in-place, puts a NUL after it. Backticks remain.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001789 * Return a pointer to the start of the next argument.
1790 */
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001791 static char_u *
Bram Moolenaar071d4272004-06-13 20:20:40 +00001792do_one_arg(str)
1793 char_u *str;
1794{
1795 char_u *p;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001796 int inbacktick;
1797
Bram Moolenaar071d4272004-06-13 20:20:40 +00001798 inbacktick = FALSE;
1799 for (p = str; *str; ++str)
1800 {
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001801 /* When the backslash is used for escaping the special meaning of a
1802 * character we need to keep it until wildcard expansion. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001803 if (rem_backslash(str))
1804 {
1805 *p++ = *str++;
1806 *p++ = *str;
1807 }
1808 else
1809 {
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001810 /* An item ends at a space not in backticks */
1811 if (!inbacktick && vim_isspace(*str))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001812 break;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001813 if (*str == '`')
Bram Moolenaar071d4272004-06-13 20:20:40 +00001814 inbacktick ^= TRUE;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001815 *p++ = *str;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001816 }
1817 }
1818 str = skipwhite(str);
1819 *p = NUL;
1820
1821 return str;
1822}
1823
Bram Moolenaar86b68352004-12-27 21:59:20 +00001824/*
1825 * Separate the arguments in "str" and return a list of pointers in the
1826 * growarray "gap".
1827 */
1828 int
1829get_arglist(gap, str)
1830 garray_T *gap;
1831 char_u *str;
1832{
1833 ga_init2(gap, (int)sizeof(char_u *), 20);
1834 while (*str != NUL)
1835 {
1836 if (ga_grow(gap, 1) == FAIL)
1837 {
1838 ga_clear(gap);
1839 return FAIL;
1840 }
1841 ((char_u **)gap->ga_data)[gap->ga_len++] = str;
1842
1843 /* Isolate one argument, change it in-place, put a NUL after it. */
1844 str = do_one_arg(str);
1845 }
1846 return OK;
1847}
1848
Bram Moolenaar7df351e2006-01-23 22:30:28 +00001849#if defined(FEAT_QUICKFIX) || defined(FEAT_SYN_HL) || defined(PROTO)
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00001850/*
1851 * Parse a list of arguments (file names), expand them and return in
Bram Moolenaar8f5c6f02012-06-29 12:57:06 +02001852 * "fnames[fcountp]". When "wig" is TRUE, removes files matching 'wildignore'.
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00001853 * Return FAIL or OK.
1854 */
1855 int
Bram Moolenaar8f5c6f02012-06-29 12:57:06 +02001856get_arglist_exp(str, fcountp, fnamesp, wig)
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00001857 char_u *str;
1858 int *fcountp;
1859 char_u ***fnamesp;
Bram Moolenaar8f5c6f02012-06-29 12:57:06 +02001860 int wig;
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00001861{
1862 garray_T ga;
1863 int i;
1864
1865 if (get_arglist(&ga, str) == FAIL)
1866 return FAIL;
Bram Moolenaar8f5c6f02012-06-29 12:57:06 +02001867 if (wig == TRUE)
1868 i = expand_wildcards(ga.ga_len, (char_u **)ga.ga_data,
1869 fcountp, fnamesp, EW_FILE|EW_NOTFOUND);
1870 else
1871 i = gen_expand_wildcards(ga.ga_len, (char_u **)ga.ga_data,
1872 fcountp, fnamesp, EW_FILE|EW_NOTFOUND);
1873
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00001874 ga_clear(&ga);
1875 return i;
1876}
1877#endif
1878
Bram Moolenaar071d4272004-06-13 20:20:40 +00001879#if defined(FEAT_GUI) || defined(FEAT_CLIENTSERVER) || defined(PROTO)
1880/*
1881 * Redefine the argument list.
1882 */
1883 void
1884set_arglist(str)
1885 char_u *str;
1886{
1887 do_arglist(str, AL_SET, 0);
1888}
1889#endif
1890
1891/*
1892 * "what" == AL_SET: Redefine the argument list to 'str'.
1893 * "what" == AL_ADD: add files in 'str' to the argument list after "after".
1894 * "what" == AL_DEL: remove files in 'str' from the argument list.
1895 *
1896 * Return FAIL for failure, OK otherwise.
1897 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001898 static int
1899do_arglist(str, what, after)
1900 char_u *str;
Bram Moolenaar0c094b92009-05-14 20:20:33 +00001901 int what UNUSED;
1902 int after UNUSED; /* 0 means before first one */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001903{
1904 garray_T new_ga;
1905 int exp_count;
1906 char_u **exp_files;
1907 int i;
1908#ifdef FEAT_LISTCMDS
1909 char_u *p;
1910 int match;
1911#endif
1912
1913 /*
1914 * Collect all file name arguments in "new_ga".
1915 */
Bram Moolenaar86b68352004-12-27 21:59:20 +00001916 if (get_arglist(&new_ga, str) == FAIL)
1917 return FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001918
1919#ifdef FEAT_LISTCMDS
1920 if (what == AL_DEL)
1921 {
1922 regmatch_T regmatch;
1923 int didone;
1924
1925 /*
1926 * Delete the items: use each item as a regexp and find a match in the
1927 * argument list.
1928 */
1929#ifdef CASE_INSENSITIVE_FILENAME
1930 regmatch.rm_ic = TRUE; /* Always ignore case */
1931#else
1932 regmatch.rm_ic = FALSE; /* Never ignore case */
1933#endif
1934 for (i = 0; i < new_ga.ga_len && !got_int; ++i)
1935 {
1936 p = ((char_u **)new_ga.ga_data)[i];
1937 p = file_pat_to_reg_pat(p, NULL, NULL, FALSE);
1938 if (p == NULL)
1939 break;
1940 regmatch.regprog = vim_regcomp(p, p_magic ? RE_MAGIC : 0);
1941 if (regmatch.regprog == NULL)
1942 {
1943 vim_free(p);
1944 break;
1945 }
1946
1947 didone = FALSE;
1948 for (match = 0; match < ARGCOUNT; ++match)
1949 if (vim_regexec(&regmatch, alist_name(&ARGLIST[match]),
1950 (colnr_T)0))
1951 {
1952 didone = TRUE;
1953 vim_free(ARGLIST[match].ae_fname);
1954 mch_memmove(ARGLIST + match, ARGLIST + match + 1,
1955 (ARGCOUNT - match - 1) * sizeof(aentry_T));
1956 --ALIST(curwin)->al_ga.ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001957 if (curwin->w_arg_idx > match)
1958 --curwin->w_arg_idx;
1959 --match;
1960 }
1961
1962 vim_free(regmatch.regprog);
1963 vim_free(p);
1964 if (!didone)
1965 EMSG2(_(e_nomatch2), ((char_u **)new_ga.ga_data)[i]);
1966 }
1967 ga_clear(&new_ga);
1968 }
1969 else
1970#endif
1971 {
1972 i = expand_wildcards(new_ga.ga_len, (char_u **)new_ga.ga_data,
1973 &exp_count, &exp_files, EW_DIR|EW_FILE|EW_ADDSLASH|EW_NOTFOUND);
1974 ga_clear(&new_ga);
1975 if (i == FAIL)
1976 return FAIL;
1977 if (exp_count == 0)
1978 {
1979 EMSG(_(e_nomatch));
1980 return FAIL;
1981 }
1982
1983#ifdef FEAT_LISTCMDS
1984 if (what == AL_ADD)
1985 {
1986 (void)alist_add_list(exp_count, exp_files, after);
1987 vim_free(exp_files);
1988 }
1989 else /* what == AL_SET */
1990#endif
Bram Moolenaar86b68352004-12-27 21:59:20 +00001991 alist_set(ALIST(curwin), exp_count, exp_files, FALSE, NULL, 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001992 }
1993
1994 alist_check_arg_idx();
1995
1996 return OK;
1997}
1998
1999/*
2000 * Check the validity of the arg_idx for each other window.
2001 */
2002 static void
2003alist_check_arg_idx()
2004{
2005#ifdef FEAT_WINDOWS
2006 win_T *win;
Bram Moolenaarf740b292006-02-16 22:11:02 +00002007 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002008
Bram Moolenaarf740b292006-02-16 22:11:02 +00002009 FOR_ALL_TAB_WINDOWS(tp, win)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002010 if (win->w_alist == curwin->w_alist)
2011 check_arg_idx(win);
2012#else
2013 check_arg_idx(curwin);
2014#endif
2015}
2016
2017/*
Bram Moolenaarb8ff1fb2012-02-04 21:59:01 +01002018 * Return TRUE if window "win" is editing the file at the current argument
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002019 * index.
2020 */
2021 static int
2022editing_arg_idx(win)
2023 win_T *win;
2024{
2025 return !(win->w_arg_idx >= WARGCOUNT(win)
2026 || (win->w_buffer->b_fnum
2027 != WARGLIST(win)[win->w_arg_idx].ae_fnum
2028 && (win->w_buffer->b_ffname == NULL
2029 || !(fullpathcmp(
2030 alist_name(&WARGLIST(win)[win->w_arg_idx]),
2031 win->w_buffer->b_ffname, TRUE) & FPC_SAME))));
2032}
2033
2034/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002035 * Check if window "win" is editing the w_arg_idx file in its argument list.
2036 */
2037 void
2038check_arg_idx(win)
2039 win_T *win;
2040{
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002041 if (WARGCOUNT(win) > 1 && !editing_arg_idx(win))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002042 {
2043 /* We are not editing the current entry in the argument list.
2044 * Set "arg_had_last" if we are editing the last one. */
2045 win->w_arg_idx_invalid = TRUE;
2046 if (win->w_arg_idx != WARGCOUNT(win) - 1
2047 && arg_had_last == FALSE
2048#ifdef FEAT_WINDOWS
2049 && ALIST(win) == &global_alist
2050#endif
2051 && GARGCOUNT > 0
2052 && win->w_arg_idx < GARGCOUNT
2053 && (win->w_buffer->b_fnum == GARGLIST[GARGCOUNT - 1].ae_fnum
2054 || (win->w_buffer->b_ffname != NULL
2055 && (fullpathcmp(alist_name(&GARGLIST[GARGCOUNT - 1]),
2056 win->w_buffer->b_ffname, TRUE) & FPC_SAME))))
2057 arg_had_last = TRUE;
2058 }
2059 else
2060 {
2061 /* We are editing the current entry in the argument list.
2062 * Set "arg_had_last" if it's also the last one */
2063 win->w_arg_idx_invalid = FALSE;
2064 if (win->w_arg_idx == WARGCOUNT(win) - 1
2065#ifdef FEAT_WINDOWS
2066 && win->w_alist == &global_alist
2067#endif
2068 )
2069 arg_had_last = TRUE;
2070 }
2071}
2072
2073/*
2074 * ":args", ":argslocal" and ":argsglobal".
2075 */
2076 void
2077ex_args(eap)
2078 exarg_T *eap;
2079{
2080 int i;
2081
2082 if (eap->cmdidx != CMD_args)
2083 {
2084#if defined(FEAT_WINDOWS) && defined(FEAT_LISTCMDS)
2085 alist_unlink(ALIST(curwin));
2086 if (eap->cmdidx == CMD_argglobal)
2087 ALIST(curwin) = &global_alist;
2088 else /* eap->cmdidx == CMD_arglocal */
2089 alist_new();
2090#else
2091 ex_ni(eap);
2092 return;
2093#endif
2094 }
2095
2096 if (!ends_excmd(*eap->arg))
2097 {
2098 /*
2099 * ":args file ..": define new argument list, handle like ":next"
2100 * Also for ":argslocal file .." and ":argsglobal file ..".
2101 */
2102 ex_next(eap);
2103 }
2104 else
2105#if defined(FEAT_WINDOWS) && defined(FEAT_LISTCMDS)
2106 if (eap->cmdidx == CMD_args)
2107#endif
2108 {
2109 /*
2110 * ":args": list arguments.
2111 */
2112 if (ARGCOUNT > 0)
2113 {
2114 /* Overwrite the command, for a short list there is no scrolling
2115 * required and no wait_return(). */
2116 gotocmdline(TRUE);
2117 for (i = 0; i < ARGCOUNT; ++i)
2118 {
2119 if (i == curwin->w_arg_idx)
2120 msg_putchar('[');
2121 msg_outtrans(alist_name(&ARGLIST[i]));
2122 if (i == curwin->w_arg_idx)
2123 msg_putchar(']');
2124 msg_putchar(' ');
2125 }
2126 }
2127 }
2128#if defined(FEAT_WINDOWS) && defined(FEAT_LISTCMDS)
2129 else if (eap->cmdidx == CMD_arglocal)
2130 {
2131 garray_T *gap = &curwin->w_alist->al_ga;
2132
2133 /*
2134 * ":argslocal": make a local copy of the global argument list.
2135 */
2136 if (ga_grow(gap, GARGCOUNT) == OK)
2137 for (i = 0; i < GARGCOUNT; ++i)
2138 if (GARGLIST[i].ae_fname != NULL)
2139 {
2140 AARGLIST(curwin->w_alist)[gap->ga_len].ae_fname =
2141 vim_strsave(GARGLIST[i].ae_fname);
2142 AARGLIST(curwin->w_alist)[gap->ga_len].ae_fnum =
2143 GARGLIST[i].ae_fnum;
2144 ++gap->ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002145 }
2146 }
2147#endif
2148}
2149
2150/*
2151 * ":previous", ":sprevious", ":Next" and ":sNext".
2152 */
2153 void
2154ex_previous(eap)
2155 exarg_T *eap;
2156{
2157 /* If past the last one already, go to the last one. */
2158 if (curwin->w_arg_idx - (int)eap->line2 >= ARGCOUNT)
2159 do_argfile(eap, ARGCOUNT - 1);
2160 else
2161 do_argfile(eap, curwin->w_arg_idx - (int)eap->line2);
2162}
2163
2164/*
2165 * ":rewind", ":first", ":sfirst" and ":srewind".
2166 */
2167 void
2168ex_rewind(eap)
2169 exarg_T *eap;
2170{
2171 do_argfile(eap, 0);
2172}
2173
2174/*
2175 * ":last" and ":slast".
2176 */
2177 void
2178ex_last(eap)
2179 exarg_T *eap;
2180{
2181 do_argfile(eap, ARGCOUNT - 1);
2182}
2183
2184/*
2185 * ":argument" and ":sargument".
2186 */
2187 void
2188ex_argument(eap)
2189 exarg_T *eap;
2190{
2191 int i;
2192
2193 if (eap->addr_count > 0)
2194 i = eap->line2 - 1;
2195 else
2196 i = curwin->w_arg_idx;
2197 do_argfile(eap, i);
2198}
2199
2200/*
2201 * Edit file "argn" of the argument lists.
2202 */
2203 void
2204do_argfile(eap, argn)
2205 exarg_T *eap;
2206 int argn;
2207{
2208 int other;
2209 char_u *p;
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002210 int old_arg_idx = curwin->w_arg_idx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002211
2212 if (argn < 0 || argn >= ARGCOUNT)
2213 {
2214 if (ARGCOUNT <= 1)
2215 EMSG(_("E163: There is only one file to edit"));
2216 else if (argn < 0)
2217 EMSG(_("E164: Cannot go before first file"));
2218 else
2219 EMSG(_("E165: Cannot go beyond last file"));
2220 }
2221 else
2222 {
2223 setpcmark();
2224#ifdef FEAT_GUI
2225 need_mouse_correct = TRUE;
2226#endif
2227
2228#ifdef FEAT_WINDOWS
Bram Moolenaardf1bdc92006-02-23 21:32:16 +00002229 /* split window or create new tab page first */
2230 if (*eap->cmd == 's' || cmdmod.tab != 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002231 {
2232 if (win_split(0, 0) == FAIL)
2233 return;
Bram Moolenaar3368ea22010-09-21 16:56:35 +02002234 RESET_BINDING(curwin);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002235 }
2236 else
2237#endif
2238 {
2239 /*
2240 * if 'hidden' set, only check for changed file when re-editing
2241 * the same buffer
2242 */
2243 other = TRUE;
2244 if (P_HID(curbuf))
2245 {
2246 p = fix_fname(alist_name(&ARGLIST[argn]));
2247 other = otherfile(p);
2248 vim_free(p);
2249 }
2250 if ((!P_HID(curbuf) || !other)
2251 && check_changed(curbuf, TRUE, !other, eap->forceit, FALSE))
2252 return;
2253 }
2254
2255 curwin->w_arg_idx = argn;
2256 if (argn == ARGCOUNT - 1
2257#ifdef FEAT_WINDOWS
2258 && curwin->w_alist == &global_alist
2259#endif
2260 )
2261 arg_had_last = TRUE;
2262
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002263 /* Edit the file; always use the last known line number.
2264 * When it fails (e.g. Abort for already edited file) restore the
2265 * argument index. */
2266 if (do_ecmd(0, alist_name(&ARGLIST[curwin->w_arg_idx]), NULL,
Bram Moolenaar071d4272004-06-13 20:20:40 +00002267 eap, ECMD_LAST,
Bram Moolenaar701f7af2008-11-15 13:12:07 +00002268 (P_HID(curwin->w_buffer) ? ECMD_HIDE : 0)
2269 + (eap->forceit ? ECMD_FORCEIT : 0), curwin) == FAIL)
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002270 curwin->w_arg_idx = old_arg_idx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002271 /* like Vi: set the mark where the cursor is in the file. */
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002272 else if (eap->cmdidx != CMD_argdo)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002273 setmark('\'');
2274 }
2275}
2276
2277/*
2278 * ":next", and commands that behave like it.
2279 */
2280 void
2281ex_next(eap)
2282 exarg_T *eap;
2283{
2284 int i;
2285
2286 /*
2287 * check for changed buffer now, if this fails the argument list is not
2288 * redefined.
2289 */
2290 if ( P_HID(curbuf)
2291 || eap->cmdidx == CMD_snext
2292 || !check_changed(curbuf, TRUE, FALSE, eap->forceit, FALSE))
2293 {
2294 if (*eap->arg != NUL) /* redefine file list */
2295 {
2296 if (do_arglist(eap->arg, AL_SET, 0) == FAIL)
2297 return;
2298 i = 0;
2299 }
2300 else
2301 i = curwin->w_arg_idx + (int)eap->line2;
2302 do_argfile(eap, i);
2303 }
2304}
2305
2306#ifdef FEAT_LISTCMDS
2307/*
2308 * ":argedit"
2309 */
2310 void
2311ex_argedit(eap)
2312 exarg_T *eap;
2313{
2314 int fnum;
2315 int i;
2316 char_u *s;
2317
2318 /* Add the argument to the buffer list and get the buffer number. */
2319 fnum = buflist_add(eap->arg, BLN_LISTED);
2320
2321 /* Check if this argument is already in the argument list. */
2322 for (i = 0; i < ARGCOUNT; ++i)
2323 if (ARGLIST[i].ae_fnum == fnum)
2324 break;
2325 if (i == ARGCOUNT)
2326 {
2327 /* Can't find it, add it to the argument list. */
2328 s = vim_strsave(eap->arg);
2329 if (s == NULL)
2330 return;
2331 i = alist_add_list(1, &s,
2332 eap->addr_count > 0 ? (int)eap->line2 : curwin->w_arg_idx + 1);
2333 if (i < 0)
2334 return;
2335 curwin->w_arg_idx = i;
2336 }
2337
2338 alist_check_arg_idx();
2339
2340 /* Edit the argument. */
2341 do_argfile(eap, i);
2342}
2343
2344/*
2345 * ":argadd"
2346 */
2347 void
2348ex_argadd(eap)
2349 exarg_T *eap;
2350{
2351 do_arglist(eap->arg, AL_ADD,
2352 eap->addr_count > 0 ? (int)eap->line2 : curwin->w_arg_idx + 1);
2353#ifdef FEAT_TITLE
2354 maketitle();
2355#endif
2356}
2357
2358/*
2359 * ":argdelete"
2360 */
2361 void
2362ex_argdelete(eap)
2363 exarg_T *eap;
2364{
2365 int i;
2366 int n;
2367
2368 if (eap->addr_count > 0)
2369 {
2370 /* ":1,4argdel": Delete all arguments in the range. */
2371 if (eap->line2 > ARGCOUNT)
2372 eap->line2 = ARGCOUNT;
2373 n = eap->line2 - eap->line1 + 1;
2374 if (*eap->arg != NUL || n <= 0)
2375 EMSG(_(e_invarg));
2376 else
2377 {
2378 for (i = eap->line1; i <= eap->line2; ++i)
2379 vim_free(ARGLIST[i - 1].ae_fname);
2380 mch_memmove(ARGLIST + eap->line1 - 1, ARGLIST + eap->line2,
2381 (size_t)((ARGCOUNT - eap->line2) * sizeof(aentry_T)));
2382 ALIST(curwin)->al_ga.ga_len -= n;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002383 if (curwin->w_arg_idx >= eap->line2)
2384 curwin->w_arg_idx -= n;
2385 else if (curwin->w_arg_idx > eap->line1)
2386 curwin->w_arg_idx = eap->line1;
2387 }
2388 }
2389 else if (*eap->arg == NUL)
2390 EMSG(_(e_argreq));
2391 else
2392 do_arglist(eap->arg, AL_DEL, 0);
2393#ifdef FEAT_TITLE
2394 maketitle();
2395#endif
2396}
2397
2398/*
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002399 * ":argdo", ":windo", ":bufdo", ":tabdo"
Bram Moolenaar071d4272004-06-13 20:20:40 +00002400 */
2401 void
2402ex_listdo(eap)
2403 exarg_T *eap;
2404{
2405 int i;
2406#ifdef FEAT_WINDOWS
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002407 win_T *wp;
2408 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002409#endif
2410 buf_T *buf;
2411 int next_fnum = 0;
2412#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
2413 char_u *save_ei = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002414#endif
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002415 char_u *p_shm_save;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002416
2417#ifndef FEAT_WINDOWS
2418 if (eap->cmdidx == CMD_windo)
2419 {
2420 ex_ni(eap);
2421 return;
2422 }
2423#endif
2424
2425#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002426 if (eap->cmdidx != CMD_windo && eap->cmdidx != CMD_tabdo)
Bram Moolenaardcaf10e2005-01-21 11:55:25 +00002427 /* Don't do syntax HL autocommands. Skipping the syntax file is a
2428 * great speed improvement. */
2429 save_ei = au_event_disable(",Syntax");
Bram Moolenaar071d4272004-06-13 20:20:40 +00002430#endif
2431
2432 if (eap->cmdidx == CMD_windo
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002433 || eap->cmdidx == CMD_tabdo
Bram Moolenaar071d4272004-06-13 20:20:40 +00002434 || P_HID(curbuf)
2435 || !check_changed(curbuf, TRUE, FALSE, eap->forceit, FALSE))
2436 {
2437 /* start at the first argument/window/buffer */
2438 i = 0;
2439#ifdef FEAT_WINDOWS
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002440 wp = firstwin;
2441 tp = first_tabpage;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002442#endif
2443 /* set pcmark now */
2444 if (eap->cmdidx == CMD_bufdo)
2445 goto_buffer(eap, DOBUF_FIRST, FORWARD, 0);
2446 else
2447 setpcmark();
2448 listcmd_busy = TRUE; /* avoids setting pcmark below */
2449
2450 while (!got_int)
2451 {
2452 if (eap->cmdidx == CMD_argdo)
2453 {
2454 /* go to argument "i" */
2455 if (i == ARGCOUNT)
2456 break;
2457 /* Don't call do_argfile() when already there, it will try
2458 * reloading the file. */
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002459 if (curwin->w_arg_idx != i || !editing_arg_idx(curwin))
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002460 {
2461 /* Clear 'shm' to avoid that the file message overwrites
2462 * any output from the command. */
2463 p_shm_save = vim_strsave(p_shm);
2464 set_option_value((char_u *)"shm", 0L, (char_u *)"", 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002465 do_argfile(eap, i);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002466 set_option_value((char_u *)"shm", 0L, p_shm_save, 0);
2467 vim_free(p_shm_save);
2468 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002469 if (curwin->w_arg_idx != i)
2470 break;
2471 ++i;
2472 }
2473#ifdef FEAT_WINDOWS
2474 else if (eap->cmdidx == CMD_windo)
2475 {
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002476 /* go to window "wp" */
2477 if (!win_valid(wp))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002478 break;
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002479 win_goto(wp);
Bram Moolenaar41423242007-05-03 20:11:13 +00002480 if (curwin != wp)
2481 break; /* something must be wrong */
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002482 wp = curwin->w_next;
2483 }
2484 else if (eap->cmdidx == CMD_tabdo)
2485 {
2486 /* go to window "tp" */
2487 if (!valid_tabpage(tp))
2488 break;
Bram Moolenaara8596c42012-06-13 14:28:20 +02002489 goto_tabpage_tp(tp, TRUE);
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002490 tp = tp->tp_next;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002491 }
2492#endif
2493 else if (eap->cmdidx == CMD_bufdo)
2494 {
2495 /* Remember the number of the next listed buffer, in case
2496 * ":bwipe" is used or autocommands do something strange. */
2497 next_fnum = -1;
2498 for (buf = curbuf->b_next; buf != NULL; buf = buf->b_next)
2499 if (buf->b_p_bl)
2500 {
2501 next_fnum = buf->b_fnum;
2502 break;
2503 }
2504 }
2505
2506 /* execute the command */
2507 do_cmdline(eap->arg, eap->getline, eap->cookie,
2508 DOCMD_VERBOSE + DOCMD_NOWAIT);
2509
2510 if (eap->cmdidx == CMD_bufdo)
2511 {
2512 /* Done? */
2513 if (next_fnum < 0)
2514 break;
2515 /* Check if the buffer still exists. */
2516 for (buf = firstbuf; buf != NULL; buf = buf->b_next)
2517 if (buf->b_fnum == next_fnum)
2518 break;
2519 if (buf == NULL)
2520 break;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002521
2522 /* Go to the next buffer. Clear 'shm' to avoid that the file
2523 * message overwrites any output from the command. */
2524 p_shm_save = vim_strsave(p_shm);
2525 set_option_value((char_u *)"shm", 0L, (char_u *)"", 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002526 goto_buffer(eap, DOBUF_FIRST, FORWARD, next_fnum);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002527 set_option_value((char_u *)"shm", 0L, p_shm_save, 0);
2528 vim_free(p_shm_save);
2529
Bram Moolenaar071d4272004-06-13 20:20:40 +00002530 /* If autocommands took us elsewhere, quit here */
2531 if (curbuf->b_fnum != next_fnum)
2532 break;
2533 }
2534
2535 if (eap->cmdidx == CMD_windo)
2536 {
2537 validate_cursor(); /* cursor may have moved */
2538#ifdef FEAT_SCROLLBIND
2539 /* required when 'scrollbind' has been set */
2540 if (curwin->w_p_scb)
2541 do_check_scrollbind(TRUE);
2542#endif
2543 }
2544 }
2545 listcmd_busy = FALSE;
2546 }
2547
2548#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
Bram Moolenaar6ac54292005-02-02 23:07:25 +00002549 if (save_ei != NULL)
2550 {
2551 au_event_restore(save_ei);
2552 apply_autocmds(EVENT_SYNTAX, curbuf->b_p_syn,
2553 curbuf->b_fname, TRUE, curbuf);
2554 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002555#endif
2556}
2557
2558/*
2559 * Add files[count] to the arglist of the current window after arg "after".
2560 * The file names in files[count] must have been allocated and are taken over.
2561 * Files[] itself is not taken over.
2562 * Returns index of first added argument. Returns -1 when failed (out of mem).
2563 */
2564 static int
2565alist_add_list(count, files, after)
2566 int count;
2567 char_u **files;
2568 int after; /* where to add: 0 = before first one */
2569{
2570 int i;
2571
2572 if (ga_grow(&ALIST(curwin)->al_ga, count) == OK)
2573 {
2574 if (after < 0)
2575 after = 0;
2576 if (after > ARGCOUNT)
2577 after = ARGCOUNT;
2578 if (after < ARGCOUNT)
2579 mch_memmove(&(ARGLIST[after + count]), &(ARGLIST[after]),
2580 (ARGCOUNT - after) * sizeof(aentry_T));
2581 for (i = 0; i < count; ++i)
2582 {
2583 ARGLIST[after + i].ae_fname = files[i];
2584 ARGLIST[after + i].ae_fnum = buflist_add(files[i], BLN_LISTED);
2585 }
2586 ALIST(curwin)->al_ga.ga_len += count;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002587 if (curwin->w_arg_idx >= after)
2588 ++curwin->w_arg_idx;
2589 return after;
2590 }
2591
2592 for (i = 0; i < count; ++i)
2593 vim_free(files[i]);
2594 return -1;
2595}
2596
2597#endif /* FEAT_LISTCMDS */
2598
2599#ifdef FEAT_EVAL
2600/*
2601 * ":compiler[!] {name}"
2602 */
2603 void
2604ex_compiler(eap)
2605 exarg_T *eap;
2606{
2607 char_u *buf;
2608 char_u *old_cur_comp = NULL;
2609 char_u *p;
2610
2611 if (*eap->arg == NUL)
2612 {
2613 /* List all compiler scripts. */
2614 do_cmdline_cmd((char_u *)"echo globpath(&rtp, 'compiler/*.vim')");
2615 /* ) keep the indenter happy... */
2616 }
2617 else
2618 {
2619 buf = alloc((unsigned)(STRLEN(eap->arg) + 14));
2620 if (buf != NULL)
2621 {
2622 if (eap->forceit)
2623 {
2624 /* ":compiler! {name}" sets global options */
2625 do_cmdline_cmd((char_u *)
2626 "command -nargs=* CompilerSet set <args>");
2627 }
2628 else
2629 {
2630 /* ":compiler! {name}" sets local options.
2631 * To remain backwards compatible "current_compiler" is always
2632 * used. A user's compiler plugin may set it, the distributed
2633 * plugin will then skip the settings. Afterwards set
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01002634 * "b:current_compiler" and restore "current_compiler".
2635 * Explicitly prepend "g:" to make it work in a function. */
2636 old_cur_comp = get_var_value((char_u *)"g:current_compiler");
Bram Moolenaar071d4272004-06-13 20:20:40 +00002637 if (old_cur_comp != NULL)
2638 old_cur_comp = vim_strsave(old_cur_comp);
2639 do_cmdline_cmd((char_u *)
2640 "command -nargs=* CompilerSet setlocal <args>");
2641 }
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01002642 do_unlet((char_u *)"g:current_compiler", TRUE);
Bram Moolenaar2ce06f62005-01-31 19:19:04 +00002643 do_unlet((char_u *)"b:current_compiler", TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002644
2645 sprintf((char *)buf, "compiler/%s.vim", eap->arg);
Bram Moolenaar90cfdbe2005-08-12 19:59:19 +00002646 if (source_runtime(buf, TRUE) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002647 EMSG2(_("E666: compiler not supported: %s"), eap->arg);
2648 vim_free(buf);
2649
2650 do_cmdline_cmd((char_u *)":delcommand CompilerSet");
2651
2652 /* Set "b:current_compiler" from "current_compiler". */
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01002653 p = get_var_value((char_u *)"g:current_compiler");
Bram Moolenaar071d4272004-06-13 20:20:40 +00002654 if (p != NULL)
2655 set_internal_string_var((char_u *)"b:current_compiler", p);
2656
2657 /* Restore "current_compiler" for ":compiler {name}". */
2658 if (!eap->forceit)
2659 {
2660 if (old_cur_comp != NULL)
2661 {
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01002662 set_internal_string_var((char_u *)"g:current_compiler",
Bram Moolenaar071d4272004-06-13 20:20:40 +00002663 old_cur_comp);
2664 vim_free(old_cur_comp);
2665 }
2666 else
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01002667 do_unlet((char_u *)"g:current_compiler", TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002668 }
2669 }
2670 }
2671}
2672#endif
2673
2674/*
2675 * ":runtime {name}"
2676 */
2677 void
2678ex_runtime(eap)
2679 exarg_T *eap;
2680{
Bram Moolenaar90cfdbe2005-08-12 19:59:19 +00002681 source_runtime(eap->arg, eap->forceit);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002682}
2683
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002684static void source_callback __ARGS((char_u *fname, void *cookie));
Bram Moolenaar071d4272004-06-13 20:20:40 +00002685
2686 static void
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002687source_callback(fname, cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002688 char_u *fname;
Bram Moolenaar0c094b92009-05-14 20:20:33 +00002689 void *cookie UNUSED;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002690{
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002691 (void)do_source(fname, FALSE, DOSO_NONE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002692}
2693
2694/*
2695 * Source the file "name" from all directories in 'runtimepath'.
2696 * "name" can contain wildcards.
2697 * When "all" is TRUE, source all files, otherwise only the first one.
2698 * return FAIL when no file could be sourced, OK otherwise.
2699 */
2700 int
Bram Moolenaar90cfdbe2005-08-12 19:59:19 +00002701source_runtime(name, all)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002702 char_u *name;
2703 int all;
2704{
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002705 return do_in_runtimepath(name, all, source_callback, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002706}
2707
2708/*
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002709 * Find "name" in 'runtimepath'. When found, invoke the callback function for
2710 * it: callback(fname, "cookie")
Bram Moolenaar071d4272004-06-13 20:20:40 +00002711 * When "all" is TRUE repeat for all matches, otherwise only the first one is
2712 * used.
2713 * Returns OK when at least one match found, FAIL otherwise.
2714 */
2715 int
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002716do_in_runtimepath(name, all, callback, cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002717 char_u *name;
2718 int all;
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002719 void (*callback)__ARGS((char_u *fname, void *ck));
2720 void *cookie;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002721{
2722 char_u *rtp;
2723 char_u *np;
2724 char_u *buf;
2725 char_u *rtp_copy;
2726 char_u *tail;
2727 int num_files;
2728 char_u **files;
2729 int i;
2730 int did_one = FALSE;
2731#ifdef AMIGA
2732 struct Process *proc = (struct Process *)FindTask(0L);
2733 APTR save_winptr = proc->pr_WindowPtr;
2734
2735 /* Avoid a requester here for a volume that doesn't exist. */
2736 proc->pr_WindowPtr = (APTR)-1L;
2737#endif
2738
2739 /* Make a copy of 'runtimepath'. Invoking the callback may change the
2740 * value. */
2741 rtp_copy = vim_strsave(p_rtp);
2742 buf = alloc(MAXPATHL);
2743 if (buf != NULL && rtp_copy != NULL)
2744 {
2745 if (p_verbose > 1)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00002746 {
2747 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00002748 smsg((char_u *)_("Searching for \"%s\" in \"%s\""),
Bram Moolenaar071d4272004-06-13 20:20:40 +00002749 (char *)name, (char *)p_rtp);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00002750 verbose_leave();
2751 }
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002752
Bram Moolenaar071d4272004-06-13 20:20:40 +00002753 /* Loop over all entries in 'runtimepath'. */
2754 rtp = rtp_copy;
2755 while (*rtp != NUL && (all || !did_one))
2756 {
2757 /* Copy the path from 'runtimepath' to buf[]. */
2758 copy_option_part(&rtp, buf, MAXPATHL, ",");
2759 if (STRLEN(buf) + STRLEN(name) + 2 < MAXPATHL)
2760 {
2761 add_pathsep(buf);
2762 tail = buf + STRLEN(buf);
2763
2764 /* Loop over all patterns in "name" */
2765 np = name;
2766 while (*np != NUL && (all || !did_one))
2767 {
2768 /* Append the pattern from "name" to buf[]. */
2769 copy_option_part(&np, tail, (int)(MAXPATHL - (tail - buf)),
2770 "\t ");
2771
2772 if (p_verbose > 2)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00002773 {
2774 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00002775 smsg((char_u *)_("Searching for \"%s\""), buf);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00002776 verbose_leave();
2777 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002778
2779 /* Expand wildcards, invoke the callback for each match. */
2780 if (gen_expand_wildcards(1, &buf, &num_files, &files,
2781 EW_FILE) == OK)
2782 {
2783 for (i = 0; i < num_files; ++i)
2784 {
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002785 (*callback)(files[i], cookie);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002786 did_one = TRUE;
2787 if (!all)
2788 break;
2789 }
2790 FreeWild(num_files, files);
2791 }
2792 }
2793 }
2794 }
2795 }
2796 vim_free(buf);
2797 vim_free(rtp_copy);
2798 if (p_verbose > 0 && !did_one)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00002799 {
2800 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00002801 smsg((char_u *)_("not found in 'runtimepath': \"%s\""), name);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00002802 verbose_leave();
2803 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002804
2805#ifdef AMIGA
2806 proc->pr_WindowPtr = save_winptr;
2807#endif
2808
2809 return did_one ? OK : FAIL;
2810}
2811
2812#if defined(FEAT_EVAL) && defined(FEAT_AUTOCMD)
2813/*
2814 * ":options"
2815 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002816 void
2817ex_options(eap)
Bram Moolenaar0c094b92009-05-14 20:20:33 +00002818 exarg_T *eap UNUSED;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002819{
2820 cmd_source((char_u *)SYS_OPTWIN_FILE, NULL);
2821}
2822#endif
2823
2824/*
2825 * ":source {fname}"
2826 */
2827 void
2828ex_source(eap)
2829 exarg_T *eap;
2830{
2831#ifdef FEAT_BROWSE
2832 if (cmdmod.browse)
2833 {
2834 char_u *fname = NULL;
2835
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00002836 fname = do_browse(0, (char_u *)_("Source Vim script"), eap->arg,
Bram Moolenaar071d4272004-06-13 20:20:40 +00002837 NULL, NULL, BROWSE_FILTER_MACROS, NULL);
2838 if (fname != NULL)
2839 {
2840 cmd_source(fname, eap);
2841 vim_free(fname);
2842 }
2843 }
2844 else
2845#endif
2846 cmd_source(eap->arg, eap);
2847}
2848
2849 static void
2850cmd_source(fname, eap)
2851 char_u *fname;
2852 exarg_T *eap;
2853{
2854 if (*fname == NUL)
2855 EMSG(_(e_argreq));
2856
Bram Moolenaar071d4272004-06-13 20:20:40 +00002857 else if (eap != NULL && eap->forceit)
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00002858 /* ":source!": read Normal mdoe commands
2859 * Need to execute the commands directly. This is required at least
2860 * for:
Bram Moolenaar071d4272004-06-13 20:20:40 +00002861 * - ":g" command busy
2862 * - after ":argdo", ":windo" or ":bufdo"
2863 * - another command follows
2864 * - inside a loop
2865 */
2866 openscript(fname, global_busy || listcmd_busy || eap->nextcmd != NULL
2867#ifdef FEAT_EVAL
2868 || eap->cstack->cs_idx >= 0
2869#endif
2870 );
2871
2872 /* ":source" read ex commands */
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002873 else if (do_source(fname, FALSE, DOSO_NONE) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002874 EMSG2(_(e_notopen), fname);
2875}
2876
2877/*
2878 * ":source" and associated commands.
2879 */
2880/*
2881 * Structure used to store info for each sourced file.
2882 * It is shared between do_source() and getsourceline().
2883 * This is required, because it needs to be handed to do_cmdline() and
2884 * sourcing can be done recursively.
2885 */
2886struct source_cookie
2887{
2888 FILE *fp; /* opened file for sourcing */
2889 char_u *nextline; /* if not NULL: line that was read ahead */
2890 int finished; /* ":finish" used */
Bram Moolenaar860cae12010-06-05 23:22:07 +02002891#if defined(USE_CRNL) || defined(USE_CR)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002892 int fileformat; /* EOL_UNKNOWN, EOL_UNIX or EOL_DOS */
2893 int error; /* TRUE if LF found after CR-LF */
2894#endif
2895#ifdef FEAT_EVAL
2896 linenr_T breakpoint; /* next line with breakpoint or zero */
2897 char_u *fname; /* name of sourced file */
2898 int dbg_tick; /* debug_tick when breakpoint was set */
2899 int level; /* top nesting level of sourced file */
2900#endif
2901#ifdef FEAT_MBYTE
2902 vimconv_T conv; /* type of conversion */
2903#endif
2904};
2905
2906#ifdef FEAT_EVAL
2907/*
2908 * Return the address holding the next breakpoint line for a source cookie.
2909 */
2910 linenr_T *
2911source_breakpoint(cookie)
2912 void *cookie;
2913{
2914 return &((struct source_cookie *)cookie)->breakpoint;
2915}
2916
2917/*
2918 * Return the address holding the debug tick for a source cookie.
2919 */
2920 int *
2921source_dbg_tick(cookie)
2922 void *cookie;
2923{
2924 return &((struct source_cookie *)cookie)->dbg_tick;
2925}
2926
2927/*
2928 * Return the nesting level for a source cookie.
2929 */
2930 int
2931source_level(cookie)
2932 void *cookie;
2933{
2934 return ((struct source_cookie *)cookie)->level;
2935}
2936#endif
2937
2938static char_u *get_one_sourceline __ARGS((struct source_cookie *sp));
2939
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01002940#if (defined(WIN32) && defined(FEAT_CSCOPE)) || defined(HAVE_FD_CLOEXEC)
2941# define USE_FOPEN_NOINH
Bram Moolenaar071d4272004-06-13 20:20:40 +00002942static FILE *fopen_noinh_readbin __ARGS((char *filename));
2943
2944/*
2945 * Special function to open a file without handle inheritance.
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01002946 * When possible the handle is closed on exec().
Bram Moolenaar071d4272004-06-13 20:20:40 +00002947 */
2948 static FILE *
2949fopen_noinh_readbin(filename)
2950 char *filename;
2951{
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01002952# ifdef WIN32
Bram Moolenaar8d8ef0b2010-01-20 21:41:47 +01002953 int fd_tmp = mch_open(filename, O_RDONLY | O_BINARY | O_NOINHERIT, 0);
2954# else
2955 int fd_tmp = mch_open(filename, O_RDONLY, 0);
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01002956# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002957
2958 if (fd_tmp == -1)
2959 return NULL;
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01002960
2961# ifdef HAVE_FD_CLOEXEC
2962 {
2963 int fdflags = fcntl(fd_tmp, F_GETFD);
2964 if (fdflags >= 0 && (fdflags & FD_CLOEXEC) == 0)
2965 fcntl(fd_tmp, F_SETFD, fdflags | FD_CLOEXEC);
2966 }
2967# endif
2968
Bram Moolenaar071d4272004-06-13 20:20:40 +00002969 return fdopen(fd_tmp, READBIN);
2970}
2971#endif
2972
2973
2974/*
2975 * do_source: Read the file "fname" and execute its lines as EX commands.
2976 *
2977 * This function may be called recursively!
2978 *
2979 * return FAIL if file could not be opened, OK otherwise
2980 */
2981 int
2982do_source(fname, check_other, is_vimrc)
2983 char_u *fname;
2984 int check_other; /* check for .vimrc and _vimrc */
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002985 int is_vimrc; /* DOSO_ value */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002986{
2987 struct source_cookie cookie;
2988 char_u *save_sourcing_name;
2989 linenr_T save_sourcing_lnum;
2990 char_u *p;
2991 char_u *fname_exp;
Bram Moolenaar73881402009-02-04 16:50:47 +00002992 char_u *firstline = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002993 int retval = FAIL;
2994#ifdef FEAT_EVAL
2995 scid_T save_current_SID;
2996 static scid_T last_current_SID = 0;
2997 void *save_funccalp;
2998 int save_debug_break_level = debug_break_level;
Bram Moolenaar05159a02005-02-26 23:04:13 +00002999 scriptitem_T *si = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003000# ifdef UNIX
3001 struct stat st;
3002 int stat_ok;
3003# endif
3004#endif
3005#ifdef STARTUPTIME
3006 struct timeval tv_rel;
3007 struct timeval tv_start;
3008#endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00003009#ifdef FEAT_PROFILE
3010 proftime_T wait_start;
3011#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003012
Bram Moolenaar071d4272004-06-13 20:20:40 +00003013 p = expand_env_save(fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003014 if (p == NULL)
3015 return retval;
3016 fname_exp = fix_fname(p);
3017 vim_free(p);
3018 if (fname_exp == NULL)
3019 return retval;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003020 if (mch_isdir(fname_exp))
3021 {
Bram Moolenaar555b2802005-05-19 21:08:39 +00003022 smsg((char_u *)_("Cannot source a directory: \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003023 goto theend;
3024 }
3025
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00003026#ifdef FEAT_AUTOCMD
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00003027 /* Apply SourceCmd autocommands, they should get the file and source it. */
3028 if (has_autocmd(EVENT_SOURCECMD, fname_exp, NULL)
3029 && apply_autocmds(EVENT_SOURCECMD, fname_exp, fname_exp,
3030 FALSE, curbuf))
Bram Moolenaarf33943e2008-01-15 21:17:30 +00003031 {
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00003032# ifdef FEAT_EVAL
Bram Moolenaarf33943e2008-01-15 21:17:30 +00003033 retval = aborting() ? FAIL : OK;
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00003034# else
Bram Moolenaarf33943e2008-01-15 21:17:30 +00003035 retval = OK;
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00003036# endif
Bram Moolenaarf33943e2008-01-15 21:17:30 +00003037 goto theend;
3038 }
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00003039
3040 /* Apply SourcePre autocommands, they may get the file. */
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00003041 apply_autocmds(EVENT_SOURCEPRE, fname_exp, fname_exp, FALSE, curbuf);
3042#endif
3043
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01003044#ifdef USE_FOPEN_NOINH
Bram Moolenaar071d4272004-06-13 20:20:40 +00003045 cookie.fp = fopen_noinh_readbin((char *)fname_exp);
3046#else
3047 cookie.fp = mch_fopen((char *)fname_exp, READBIN);
3048#endif
3049 if (cookie.fp == NULL && check_other)
3050 {
3051 /*
3052 * Try again, replacing file name ".vimrc" by "_vimrc" or vice versa,
3053 * and ".exrc" by "_exrc" or vice versa.
3054 */
3055 p = gettail(fname_exp);
3056 if ((*p == '.' || *p == '_')
3057 && (STRICMP(p + 1, "vimrc") == 0
3058 || STRICMP(p + 1, "gvimrc") == 0
3059 || STRICMP(p + 1, "exrc") == 0))
3060 {
3061 if (*p == '_')
3062 *p = '.';
3063 else
3064 *p = '_';
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01003065#ifdef USE_FOPEN_NOINH
Bram Moolenaar071d4272004-06-13 20:20:40 +00003066 cookie.fp = fopen_noinh_readbin((char *)fname_exp);
3067#else
3068 cookie.fp = mch_fopen((char *)fname_exp, READBIN);
3069#endif
3070 }
3071 }
3072
3073 if (cookie.fp == NULL)
3074 {
3075 if (p_verbose > 0)
3076 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003077 verbose_enter();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003078 if (sourcing_name == NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00003079 smsg((char_u *)_("could not source \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003080 else
3081 smsg((char_u *)_("line %ld: could not source \"%s\""),
Bram Moolenaar555b2802005-05-19 21:08:39 +00003082 sourcing_lnum, fname);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003083 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003084 }
3085 goto theend;
3086 }
3087
3088 /*
3089 * The file exists.
3090 * - In verbose mode, give a message.
3091 * - For a vimrc file, may want to set 'compatible', call vimrc_found().
3092 */
3093 if (p_verbose > 1)
3094 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003095 verbose_enter();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003096 if (sourcing_name == NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00003097 smsg((char_u *)_("sourcing \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003098 else
3099 smsg((char_u *)_("line %ld: sourcing \"%s\""),
Bram Moolenaar555b2802005-05-19 21:08:39 +00003100 sourcing_lnum, fname);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003101 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003102 }
Bram Moolenaar910f66f2006-04-05 20:41:53 +00003103 if (is_vimrc == DOSO_VIMRC)
3104 vimrc_found(fname_exp, (char_u *)"MYVIMRC");
3105 else if (is_vimrc == DOSO_GVIMRC)
3106 vimrc_found(fname_exp, (char_u *)"MYGVIMRC");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003107
3108#ifdef USE_CRNL
3109 /* If no automatic file format: Set default to CR-NL. */
3110 if (*p_ffs == NUL)
3111 cookie.fileformat = EOL_DOS;
3112 else
3113 cookie.fileformat = EOL_UNKNOWN;
3114 cookie.error = FALSE;
3115#endif
3116
3117#ifdef USE_CR
3118 /* If no automatic file format: Set default to CR. */
3119 if (*p_ffs == NUL)
3120 cookie.fileformat = EOL_MAC;
3121 else
3122 cookie.fileformat = EOL_UNKNOWN;
3123 cookie.error = FALSE;
3124#endif
3125
3126 cookie.nextline = NULL;
3127 cookie.finished = FALSE;
3128
3129#ifdef FEAT_EVAL
3130 /*
3131 * Check if this script has a breakpoint.
3132 */
3133 cookie.breakpoint = dbg_find_breakpoint(TRUE, fname_exp, (linenr_T)0);
3134 cookie.fname = fname_exp;
3135 cookie.dbg_tick = debug_tick;
3136
3137 cookie.level = ex_nesting_level;
3138#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003139
3140 /*
3141 * Keep the sourcing name/lnum, for recursive calls.
3142 */
3143 save_sourcing_name = sourcing_name;
3144 sourcing_name = fname_exp;
3145 save_sourcing_lnum = sourcing_lnum;
3146 sourcing_lnum = 0;
3147
Bram Moolenaar73881402009-02-04 16:50:47 +00003148#ifdef FEAT_MBYTE
3149 cookie.conv.vc_type = CONV_NONE; /* no conversion */
3150
3151 /* Read the first line so we can check for a UTF-8 BOM. */
3152 firstline = getsourceline(0, (void *)&cookie, 0);
3153 if (firstline != NULL && STRLEN(firstline) >= 3 && firstline[0] == 0xef
3154 && firstline[1] == 0xbb && firstline[2] == 0xbf)
3155 {
3156 /* Found BOM; setup conversion, skip over BOM and recode the line. */
3157 convert_setup(&cookie.conv, (char_u *)"utf-8", p_enc);
3158 p = string_convert(&cookie.conv, firstline + 3, NULL);
Bram Moolenaar4e2a5952009-02-05 19:48:25 +00003159 if (p == NULL)
3160 p = vim_strsave(firstline + 3);
Bram Moolenaar73881402009-02-04 16:50:47 +00003161 if (p != NULL)
3162 {
3163 vim_free(firstline);
3164 firstline = p;
3165 }
3166 }
3167#endif
3168
Bram Moolenaar071d4272004-06-13 20:20:40 +00003169#ifdef STARTUPTIME
Bram Moolenaarc4e41982010-01-19 16:31:47 +01003170 if (time_fd != NULL)
3171 time_push(&tv_rel, &tv_start);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003172#endif
3173
3174#ifdef FEAT_EVAL
Bram Moolenaar05159a02005-02-26 23:04:13 +00003175# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00003176 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003177 prof_child_enter(&wait_start); /* entering a child now */
3178# endif
3179
3180 /* Don't use local function variables, if called from a function.
3181 * Also starts profiling timer for nested script. */
3182 save_funccalp = save_funccal();
3183
Bram Moolenaar071d4272004-06-13 20:20:40 +00003184 /*
3185 * Check if this script was sourced before to finds its SID.
3186 * If it's new, generate a new SID.
3187 */
3188 save_current_SID = current_SID;
3189# ifdef UNIX
3190 stat_ok = (mch_stat((char *)fname_exp, &st) >= 0);
3191# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00003192 for (current_SID = script_items.ga_len; current_SID > 0; --current_SID)
3193 {
3194 si = &SCRIPT_ITEM(current_SID);
3195 if (si->sn_name != NULL
Bram Moolenaar071d4272004-06-13 20:20:40 +00003196 && (
3197# ifdef UNIX
Bram Moolenaar7c626922005-02-07 22:01:03 +00003198 /* Compare dev/ino when possible, it catches symbolic
3199 * links. Also compare file names, the inode may change
3200 * when the file was edited. */
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00003201 ((stat_ok && si->sn_dev_valid)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003202 && (si->sn_dev == st.st_dev
3203 && si->sn_ino == st.st_ino)) ||
Bram Moolenaar071d4272004-06-13 20:20:40 +00003204# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00003205 fnamecmp(si->sn_name, fname_exp) == 0))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003206 break;
Bram Moolenaar05159a02005-02-26 23:04:13 +00003207 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003208 if (current_SID == 0)
3209 {
3210 current_SID = ++last_current_SID;
Bram Moolenaar05159a02005-02-26 23:04:13 +00003211 if (ga_grow(&script_items, (int)(current_SID - script_items.ga_len))
3212 == FAIL)
3213 goto almosttheend;
3214 while (script_items.ga_len < current_SID)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003215 {
Bram Moolenaar05159a02005-02-26 23:04:13 +00003216 ++script_items.ga_len;
3217 SCRIPT_ITEM(script_items.ga_len).sn_name = NULL;
3218# ifdef FEAT_PROFILE
3219 SCRIPT_ITEM(script_items.ga_len).sn_prof_on = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003220# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003221 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00003222 si = &SCRIPT_ITEM(current_SID);
3223 si->sn_name = fname_exp;
3224 fname_exp = NULL;
3225# ifdef UNIX
3226 if (stat_ok)
3227 {
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00003228 si->sn_dev_valid = TRUE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00003229 si->sn_dev = st.st_dev;
3230 si->sn_ino = st.st_ino;
3231 }
3232 else
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00003233 si->sn_dev_valid = FALSE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00003234# endif
3235
Bram Moolenaar071d4272004-06-13 20:20:40 +00003236 /* Allocate the local script variables to use for this script. */
3237 new_script_vars(current_SID);
3238 }
3239
Bram Moolenaar05159a02005-02-26 23:04:13 +00003240# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00003241 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003242 {
3243 int forceit;
3244
3245 /* Check if we do profiling for this script. */
3246 if (!si->sn_prof_on && has_profiling(TRUE, si->sn_name, &forceit))
3247 {
3248 script_do_profile(si);
3249 si->sn_pr_force = forceit;
3250 }
3251 if (si->sn_prof_on)
3252 {
3253 ++si->sn_pr_count;
3254 profile_start(&si->sn_pr_start);
3255 profile_zero(&si->sn_pr_children);
3256 }
3257 }
3258# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003259#endif
3260
3261 /*
3262 * Call do_cmdline, which will call getsourceline() to get the lines.
3263 */
Bram Moolenaar73881402009-02-04 16:50:47 +00003264 do_cmdline(firstline, getsourceline, (void *)&cookie,
Bram Moolenaar071d4272004-06-13 20:20:40 +00003265 DOCMD_VERBOSE|DOCMD_NOWAIT|DOCMD_REPEAT);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003266 retval = OK;
Bram Moolenaar05159a02005-02-26 23:04:13 +00003267
3268#ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00003269 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003270 {
3271 /* Get "si" again, "script_items" may have been reallocated. */
3272 si = &SCRIPT_ITEM(current_SID);
3273 if (si->sn_prof_on)
3274 {
3275 profile_end(&si->sn_pr_start);
3276 profile_sub_wait(&wait_start, &si->sn_pr_start);
3277 profile_add(&si->sn_pr_total, &si->sn_pr_start);
Bram Moolenaar1056d982006-03-09 22:37:52 +00003278 profile_self(&si->sn_pr_self, &si->sn_pr_start,
3279 &si->sn_pr_children);
Bram Moolenaar05159a02005-02-26 23:04:13 +00003280 }
3281 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003282#endif
3283
3284 if (got_int)
3285 EMSG(_(e_interr));
3286 sourcing_name = save_sourcing_name;
3287 sourcing_lnum = save_sourcing_lnum;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003288 if (p_verbose > 1)
3289 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003290 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00003291 smsg((char_u *)_("finished sourcing %s"), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003292 if (sourcing_name != NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00003293 smsg((char_u *)_("continuing in %s"), sourcing_name);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003294 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003295 }
3296#ifdef STARTUPTIME
Bram Moolenaarc4e41982010-01-19 16:31:47 +01003297 if (time_fd != NULL)
3298 {
3299 vim_snprintf((char *)IObuff, IOSIZE, "sourcing %s", fname);
3300 time_msg((char *)IObuff, &tv_start);
3301 time_pop(&tv_rel);
3302 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003303#endif
3304
3305#ifdef FEAT_EVAL
3306 /*
3307 * After a "finish" in debug mode, need to break at first command of next
3308 * sourced file.
3309 */
3310 if (save_debug_break_level > ex_nesting_level
3311 && debug_break_level == ex_nesting_level)
3312 ++debug_break_level;
3313#endif
3314
Bram Moolenaar05159a02005-02-26 23:04:13 +00003315#ifdef FEAT_EVAL
3316almosttheend:
3317 current_SID = save_current_SID;
3318 restore_funccal(save_funccalp);
3319# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00003320 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003321 prof_child_exit(&wait_start); /* leaving a child now */
3322# endif
3323#endif
3324 fclose(cookie.fp);
3325 vim_free(cookie.nextline);
Bram Moolenaar73881402009-02-04 16:50:47 +00003326 vim_free(firstline);
Bram Moolenaar05159a02005-02-26 23:04:13 +00003327#ifdef FEAT_MBYTE
3328 convert_setup(&cookie.conv, NULL, NULL);
3329#endif
3330
Bram Moolenaar071d4272004-06-13 20:20:40 +00003331theend:
3332 vim_free(fname_exp);
3333 return retval;
3334}
3335
3336#if defined(FEAT_EVAL) || defined(PROTO)
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00003337
Bram Moolenaar071d4272004-06-13 20:20:40 +00003338/*
3339 * ":scriptnames"
3340 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003341 void
3342ex_scriptnames(eap)
Bram Moolenaar0c094b92009-05-14 20:20:33 +00003343 exarg_T *eap UNUSED;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003344{
3345 int i;
3346
Bram Moolenaar05159a02005-02-26 23:04:13 +00003347 for (i = 1; i <= script_items.ga_len && !got_int; ++i)
3348 if (SCRIPT_ITEM(i).sn_name != NULL)
Bram Moolenaard58ea072011-06-26 04:25:30 +02003349 {
3350 home_replace(NULL, SCRIPT_ITEM(i).sn_name,
3351 NameBuff, MAXPATHL, TRUE);
3352 smsg((char_u *)"%3d: %s", i, NameBuff);
Bram Moolenaar970a1b82012-03-23 18:39:18 +01003353 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003354}
3355
3356# if defined(BACKSLASH_IN_FILENAME) || defined(PROTO)
3357/*
3358 * Fix slashes in the list of script names for 'shellslash'.
3359 */
3360 void
3361scriptnames_slash_adjust()
3362{
3363 int i;
3364
Bram Moolenaar05159a02005-02-26 23:04:13 +00003365 for (i = 1; i <= script_items.ga_len; ++i)
3366 if (SCRIPT_ITEM(i).sn_name != NULL)
3367 slash_adjust(SCRIPT_ITEM(i).sn_name);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003368}
3369# endif
3370
3371/*
3372 * Get a pointer to a script name. Used for ":verbose set".
3373 */
3374 char_u *
3375get_scriptname(id)
3376 scid_T id;
3377{
3378 if (id == SID_MODELINE)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00003379 return (char_u *)_("modeline");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003380 if (id == SID_CMDARG)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00003381 return (char_u *)_("--cmd argument");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003382 if (id == SID_CARG)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00003383 return (char_u *)_("-c argument");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003384 if (id == SID_ENV)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00003385 return (char_u *)_("environment variable");
3386 if (id == SID_ERROR)
3387 return (char_u *)_("error handler");
Bram Moolenaar05159a02005-02-26 23:04:13 +00003388 return SCRIPT_ITEM(id).sn_name;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003389}
Bram Moolenaar05159a02005-02-26 23:04:13 +00003390
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00003391# if defined(EXITFREE) || defined(PROTO)
3392 void
3393free_scriptnames()
3394{
3395 int i;
3396
3397 for (i = script_items.ga_len; i > 0; --i)
3398 vim_free(SCRIPT_ITEM(i).sn_name);
3399 ga_clear(&script_items);
3400}
3401# endif
3402
Bram Moolenaar071d4272004-06-13 20:20:40 +00003403#endif
3404
3405#if defined(USE_CR) || defined(PROTO)
3406
3407# if defined(__MSL__) && (__MSL__ >= 22)
3408/*
3409 * Newer version of the Metrowerks library handle DOS and UNIX files
3410 * without help.
3411 * Test with earlier versions, MSL 2.2 is the library supplied with
3412 * Codewarrior Pro 2.
3413 */
3414 char *
3415fgets_cr(s, n, stream)
3416 char *s;
3417 int n;
3418 FILE *stream;
3419{
3420 return fgets(s, n, stream);
3421}
3422# else
3423/*
3424 * Version of fgets() which also works for lines ending in a <CR> only
3425 * (Macintosh format).
3426 * For older versions of the Metrowerks library.
3427 * At least CodeWarrior 9 needed this code.
3428 */
3429 char *
3430fgets_cr(s, n, stream)
3431 char *s;
3432 int n;
3433 FILE *stream;
3434{
3435 int c = 0;
3436 int char_read = 0;
3437
3438 while (!feof(stream) && c != '\r' && c != '\n' && char_read < n - 1)
3439 {
3440 c = fgetc(stream);
3441 s[char_read++] = c;
3442 /* If the file is in DOS format, we need to skip a NL after a CR. I
3443 * thought it was the other way around, but this appears to work... */
3444 if (c == '\n')
3445 {
3446 c = fgetc(stream);
3447 if (c != '\r')
3448 ungetc(c, stream);
3449 }
3450 }
3451
3452 s[char_read] = 0;
3453 if (char_read == 0)
3454 return NULL;
3455
3456 if (feof(stream) && char_read == 1)
3457 return NULL;
3458
3459 return s;
3460}
3461# endif
3462#endif
3463
3464/*
3465 * Get one full line from a sourced file.
3466 * Called by do_cmdline() when it's called from do_source().
3467 *
3468 * Return a pointer to the line in allocated memory.
3469 * Return NULL for end-of-file or some error.
3470 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003471 char_u *
3472getsourceline(c, cookie, indent)
Bram Moolenaar0c094b92009-05-14 20:20:33 +00003473 int c UNUSED;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003474 void *cookie;
Bram Moolenaar0c094b92009-05-14 20:20:33 +00003475 int indent UNUSED;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003476{
3477 struct source_cookie *sp = (struct source_cookie *)cookie;
3478 char_u *line;
Bram Moolenaarc047b9a2012-02-11 20:40:55 +01003479 char_u *p;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003480
3481#ifdef FEAT_EVAL
3482 /* If breakpoints have been added/deleted need to check for it. */
3483 if (sp->dbg_tick < debug_tick)
3484 {
3485 sp->breakpoint = dbg_find_breakpoint(TRUE, sp->fname, sourcing_lnum);
3486 sp->dbg_tick = debug_tick;
3487 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00003488# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00003489 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003490 script_line_end();
3491# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003492#endif
3493 /*
3494 * Get current line. If there is a read-ahead line, use it, otherwise get
3495 * one now.
3496 */
3497 if (sp->finished)
3498 line = NULL;
3499 else if (sp->nextline == NULL)
3500 line = get_one_sourceline(sp);
3501 else
3502 {
3503 line = sp->nextline;
3504 sp->nextline = NULL;
3505 ++sourcing_lnum;
3506 }
Bram Moolenaare2cc9702005-03-15 22:43:58 +00003507#ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00003508 if (line != NULL && do_profiling == PROF_YES)
Bram Moolenaare2cc9702005-03-15 22:43:58 +00003509 script_line_start();
3510#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003511
3512 /* Only concatenate lines starting with a \ when 'cpoptions' doesn't
3513 * contain the 'C' flag. */
3514 if (line != NULL && (vim_strchr(p_cpo, CPO_CONCAT) == NULL))
3515 {
3516 /* compensate for the one line read-ahead */
3517 --sourcing_lnum;
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01003518
3519 /* Get the next line and concatenate it when it starts with a
3520 * backslash. We always need to read the next line, keep it in
3521 * sp->nextline. */
3522 sp->nextline = get_one_sourceline(sp);
3523 if (sp->nextline != NULL && *(p = skipwhite(sp->nextline)) == '\\')
Bram Moolenaar071d4272004-06-13 20:20:40 +00003524 {
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01003525 garray_T ga;
3526
Bram Moolenaarb549a732012-02-22 18:29:33 +01003527 ga_init2(&ga, (int)sizeof(char_u), 400);
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01003528 ga_concat(&ga, line);
3529 ga_concat(&ga, p + 1);
3530 for (;;)
3531 {
3532 vim_free(sp->nextline);
3533 sp->nextline = get_one_sourceline(sp);
3534 if (sp->nextline == NULL)
3535 break;
3536 p = skipwhite(sp->nextline);
3537 if (*p != '\\')
3538 break;
Bram Moolenaarb549a732012-02-22 18:29:33 +01003539 /* Adjust the growsize to the current length to speed up
3540 * concatenating many lines. */
3541 if (ga.ga_len > 400)
3542 {
3543 if (ga.ga_len > 8000)
3544 ga.ga_growsize = 8000;
3545 else
3546 ga.ga_growsize = ga.ga_len;
3547 }
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01003548 ga_concat(&ga, p + 1);
3549 }
3550 ga_append(&ga, NUL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003551 vim_free(line);
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01003552 line = ga.ga_data;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003553 }
3554 }
3555
3556#ifdef FEAT_MBYTE
3557 if (line != NULL && sp->conv.vc_type != CONV_NONE)
3558 {
Bram Moolenaarc047b9a2012-02-11 20:40:55 +01003559 char_u *s;
3560
Bram Moolenaar071d4272004-06-13 20:20:40 +00003561 /* Convert the encoding of the script line. */
3562 s = string_convert(&sp->conv, line, NULL);
3563 if (s != NULL)
3564 {
3565 vim_free(line);
3566 line = s;
3567 }
3568 }
3569#endif
3570
3571#ifdef FEAT_EVAL
3572 /* Did we encounter a breakpoint? */
3573 if (sp->breakpoint != 0 && sp->breakpoint <= sourcing_lnum)
3574 {
3575 dbg_breakpoint(sp->fname, sourcing_lnum);
3576 /* Find next breakpoint. */
3577 sp->breakpoint = dbg_find_breakpoint(TRUE, sp->fname, sourcing_lnum);
3578 sp->dbg_tick = debug_tick;
3579 }
3580#endif
3581
3582 return line;
3583}
3584
3585 static char_u *
3586get_one_sourceline(sp)
3587 struct source_cookie *sp;
3588{
3589 garray_T ga;
3590 int len;
3591 int c;
3592 char_u *buf;
3593#ifdef USE_CRNL
3594 int has_cr; /* CR-LF found */
3595#endif
3596#ifdef USE_CR
3597 char_u *scan;
3598#endif
3599 int have_read = FALSE;
3600
3601 /* use a growarray to store the sourced line */
Bram Moolenaar6ac54292005-02-02 23:07:25 +00003602 ga_init2(&ga, 1, 250);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003603
3604 /*
3605 * Loop until there is a finished line (or end-of-file).
3606 */
3607 sourcing_lnum++;
3608 for (;;)
3609 {
Bram Moolenaar6ac54292005-02-02 23:07:25 +00003610 /* make room to read at least 120 (more) characters */
3611 if (ga_grow(&ga, 120) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003612 break;
3613 buf = (char_u *)ga.ga_data;
3614
3615#ifdef USE_CR
3616 if (sp->fileformat == EOL_MAC)
3617 {
Bram Moolenaar86b68352004-12-27 21:59:20 +00003618 if (fgets_cr((char *)buf + ga.ga_len, ga.ga_maxlen - ga.ga_len,
3619 sp->fp) == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003620 break;
3621 }
3622 else
3623#endif
Bram Moolenaar86b68352004-12-27 21:59:20 +00003624 if (fgets((char *)buf + ga.ga_len, ga.ga_maxlen - ga.ga_len,
3625 sp->fp) == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003626 break;
Bram Moolenaar6ac54292005-02-02 23:07:25 +00003627 len = ga.ga_len + (int)STRLEN(buf + ga.ga_len);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003628#ifdef USE_CRNL
3629 /* Ignore a trailing CTRL-Z, when in Dos mode. Only recognize the
3630 * CTRL-Z by its own, or after a NL. */
3631 if ( (len == 1 || (len >= 2 && buf[len - 2] == '\n'))
3632 && sp->fileformat == EOL_DOS
3633 && buf[len - 1] == Ctrl_Z)
3634 {
3635 buf[len - 1] = NUL;
3636 break;
3637 }
3638#endif
3639
3640#ifdef USE_CR
3641 /* If the read doesn't stop on a new line, and there's
3642 * some CR then we assume a Mac format */
3643 if (sp->fileformat == EOL_UNKNOWN)
3644 {
3645 if (buf[len - 1] != '\n' && vim_strchr(buf, '\r') != NULL)
3646 sp->fileformat = EOL_MAC;
3647 else
3648 sp->fileformat = EOL_UNIX;
3649 }
3650
3651 if (sp->fileformat == EOL_MAC)
3652 {
3653 scan = vim_strchr(buf, '\r');
3654
3655 if (scan != NULL)
3656 {
3657 *scan = '\n';
3658 if (*(scan + 1) != 0)
3659 {
3660 *(scan + 1) = 0;
3661 fseek(sp->fp, (long)(scan - buf - len + 1), SEEK_CUR);
3662 }
3663 }
3664 len = STRLEN(buf);
3665 }
3666#endif
3667
3668 have_read = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003669 ga.ga_len = len;
3670
3671 /* If the line was longer than the buffer, read more. */
Bram Moolenaar86b68352004-12-27 21:59:20 +00003672 if (ga.ga_maxlen - ga.ga_len == 1 && buf[len - 1] != '\n')
Bram Moolenaar071d4272004-06-13 20:20:40 +00003673 continue;
3674
3675 if (len >= 1 && buf[len - 1] == '\n') /* remove trailing NL */
3676 {
3677#ifdef USE_CRNL
3678 has_cr = (len >= 2 && buf[len - 2] == '\r');
3679 if (sp->fileformat == EOL_UNKNOWN)
3680 {
3681 if (has_cr)
3682 sp->fileformat = EOL_DOS;
3683 else
3684 sp->fileformat = EOL_UNIX;
3685 }
3686
3687 if (sp->fileformat == EOL_DOS)
3688 {
3689 if (has_cr) /* replace trailing CR */
3690 {
3691 buf[len - 2] = '\n';
3692 --len;
3693 --ga.ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003694 }
3695 else /* lines like ":map xx yy^M" will have failed */
3696 {
3697 if (!sp->error)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00003698 {
3699 msg_source(hl_attr(HLF_W));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003700 EMSG(_("W15: Warning: Wrong line separator, ^M may be missing"));
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00003701 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003702 sp->error = TRUE;
3703 sp->fileformat = EOL_UNIX;
3704 }
3705 }
3706#endif
3707 /* The '\n' is escaped if there is an odd number of ^V's just
3708 * before it, first set "c" just before the 'V's and then check
3709 * len&c parities (is faster than ((len-c)%2 == 0)) -- Acevedo */
3710 for (c = len - 2; c >= 0 && buf[c] == Ctrl_V; c--)
3711 ;
3712 if ((len & 1) != (c & 1)) /* escaped NL, read more */
3713 {
3714 sourcing_lnum++;
3715 continue;
3716 }
3717
3718 buf[len - 1] = NUL; /* remove the NL */
3719 }
3720
3721 /*
3722 * Check for ^C here now and then, so recursive :so can be broken.
3723 */
3724 line_breakcheck();
3725 break;
3726 }
3727
3728 if (have_read)
3729 return (char_u *)ga.ga_data;
3730
3731 vim_free(ga.ga_data);
3732 return NULL;
3733}
3734
Bram Moolenaar05159a02005-02-26 23:04:13 +00003735#if defined(FEAT_PROFILE) || defined(PROTO)
3736/*
3737 * Called when starting to read a script line.
3738 * "sourcing_lnum" must be correct!
3739 * When skipping lines it may not actually be executed, but we won't find out
3740 * until later and we need to store the time now.
3741 */
3742 void
3743script_line_start()
3744{
3745 scriptitem_T *si;
3746 sn_prl_T *pp;
3747
3748 if (current_SID <= 0 || current_SID > script_items.ga_len)
3749 return;
3750 si = &SCRIPT_ITEM(current_SID);
3751 if (si->sn_prof_on && sourcing_lnum >= 1)
3752 {
Bram Moolenaar8c8de832008-06-24 22:58:06 +00003753 /* Grow the array before starting the timer, so that the time spent
Bram Moolenaar05159a02005-02-26 23:04:13 +00003754 * here isn't counted. */
3755 ga_grow(&si->sn_prl_ga, (int)(sourcing_lnum - si->sn_prl_ga.ga_len));
3756 si->sn_prl_idx = sourcing_lnum - 1;
3757 while (si->sn_prl_ga.ga_len <= si->sn_prl_idx
3758 && si->sn_prl_ga.ga_len < si->sn_prl_ga.ga_maxlen)
3759 {
3760 /* Zero counters for a line that was not used before. */
3761 pp = &PRL_ITEM(si, si->sn_prl_ga.ga_len);
3762 pp->snp_count = 0;
3763 profile_zero(&pp->sn_prl_total);
3764 profile_zero(&pp->sn_prl_self);
3765 ++si->sn_prl_ga.ga_len;
3766 }
3767 si->sn_prl_execed = FALSE;
3768 profile_start(&si->sn_prl_start);
3769 profile_zero(&si->sn_prl_children);
3770 profile_get_wait(&si->sn_prl_wait);
3771 }
3772}
3773
3774/*
3775 * Called when actually executing a function line.
3776 */
3777 void
3778script_line_exec()
3779{
3780 scriptitem_T *si;
3781
3782 if (current_SID <= 0 || current_SID > script_items.ga_len)
3783 return;
3784 si = &SCRIPT_ITEM(current_SID);
3785 if (si->sn_prof_on && si->sn_prl_idx >= 0)
3786 si->sn_prl_execed = TRUE;
3787}
3788
3789/*
3790 * Called when done with a function line.
3791 */
3792 void
3793script_line_end()
3794{
3795 scriptitem_T *si;
3796 sn_prl_T *pp;
3797
3798 if (current_SID <= 0 || current_SID > script_items.ga_len)
3799 return;
3800 si = &SCRIPT_ITEM(current_SID);
3801 if (si->sn_prof_on && si->sn_prl_idx >= 0
3802 && si->sn_prl_idx < si->sn_prl_ga.ga_len)
3803 {
3804 if (si->sn_prl_execed)
3805 {
3806 pp = &PRL_ITEM(si, si->sn_prl_idx);
3807 ++pp->snp_count;
3808 profile_end(&si->sn_prl_start);
3809 profile_sub_wait(&si->sn_prl_wait, &si->sn_prl_start);
Bram Moolenaar05159a02005-02-26 23:04:13 +00003810 profile_add(&pp->sn_prl_total, &si->sn_prl_start);
Bram Moolenaar1056d982006-03-09 22:37:52 +00003811 profile_self(&pp->sn_prl_self, &si->sn_prl_start,
3812 &si->sn_prl_children);
Bram Moolenaar05159a02005-02-26 23:04:13 +00003813 }
3814 si->sn_prl_idx = -1;
3815 }
3816}
3817#endif
3818
Bram Moolenaar071d4272004-06-13 20:20:40 +00003819/*
3820 * ":scriptencoding": Set encoding conversion for a sourced script.
3821 * Without the multi-byte feature it's simply ignored.
3822 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003823 void
3824ex_scriptencoding(eap)
Bram Moolenaar0c094b92009-05-14 20:20:33 +00003825 exarg_T *eap UNUSED;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003826{
3827#ifdef FEAT_MBYTE
3828 struct source_cookie *sp;
3829 char_u *name;
3830
3831 if (!getline_equal(eap->getline, eap->cookie, getsourceline))
3832 {
3833 EMSG(_("E167: :scriptencoding used outside of a sourced file"));
3834 return;
3835 }
3836
3837 if (*eap->arg != NUL)
3838 {
3839 name = enc_canonize(eap->arg);
3840 if (name == NULL) /* out of memory */
3841 return;
3842 }
3843 else
3844 name = eap->arg;
3845
3846 /* Setup for conversion from the specified encoding to 'encoding'. */
3847 sp = (struct source_cookie *)getline_cookie(eap->getline, eap->cookie);
3848 convert_setup(&sp->conv, name, p_enc);
3849
3850 if (name != eap->arg)
3851 vim_free(name);
3852#endif
3853}
3854
3855#if defined(FEAT_EVAL) || defined(PROTO)
3856/*
3857 * ":finish": Mark a sourced file as finished.
3858 */
3859 void
3860ex_finish(eap)
3861 exarg_T *eap;
3862{
3863 if (getline_equal(eap->getline, eap->cookie, getsourceline))
3864 do_finish(eap, FALSE);
3865 else
3866 EMSG(_("E168: :finish used outside of a sourced file"));
3867}
3868
3869/*
3870 * Mark a sourced file as finished. Possibly makes the ":finish" pending.
3871 * Also called for a pending finish at the ":endtry" or after returning from
3872 * an extra do_cmdline(). "reanimate" is used in the latter case.
3873 */
3874 void
3875do_finish(eap, reanimate)
3876 exarg_T *eap;
3877 int reanimate;
3878{
3879 int idx;
3880
3881 if (reanimate)
3882 ((struct source_cookie *)getline_cookie(eap->getline,
3883 eap->cookie))->finished = FALSE;
3884
3885 /*
3886 * Cleanup (and inactivate) conditionals, but stop when a try conditional
3887 * not in its finally clause (which then is to be executed next) is found.
3888 * In this case, make the ":finish" pending for execution at the ":endtry".
3889 * Otherwise, finish normally.
3890 */
3891 idx = cleanup_conditionals(eap->cstack, 0, TRUE);
3892 if (idx >= 0)
3893 {
3894 eap->cstack->cs_pending[idx] = CSTP_FINISH;
3895 report_make_pending(CSTP_FINISH, NULL);
3896 }
3897 else
3898 ((struct source_cookie *)getline_cookie(eap->getline,
3899 eap->cookie))->finished = TRUE;
3900}
3901
3902
3903/*
3904 * Return TRUE when a sourced file had the ":finish" command: Don't give error
3905 * message for missing ":endif".
3906 * Return FALSE when not sourcing a file.
3907 */
3908 int
Bram Moolenaar89d40322006-08-29 15:30:07 +00003909source_finished(fgetline, cookie)
3910 char_u *(*fgetline) __ARGS((int, void *, int));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003911 void *cookie;
3912{
Bram Moolenaar89d40322006-08-29 15:30:07 +00003913 return (getline_equal(fgetline, cookie, getsourceline)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003914 && ((struct source_cookie *)getline_cookie(
Bram Moolenaar89d40322006-08-29 15:30:07 +00003915 fgetline, cookie))->finished);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003916}
3917#endif
3918
3919#if defined(FEAT_LISTCMDS) || defined(PROTO)
3920/*
3921 * ":checktime [buffer]"
3922 */
3923 void
3924ex_checktime(eap)
3925 exarg_T *eap;
3926{
3927 buf_T *buf;
3928 int save_no_check_timestamps = no_check_timestamps;
3929
3930 no_check_timestamps = 0;
3931 if (eap->addr_count == 0) /* default is all buffers */
3932 check_timestamps(FALSE);
3933 else
3934 {
3935 buf = buflist_findnr((int)eap->line2);
3936 if (buf != NULL) /* cannot happen? */
3937 (void)buf_check_timestamp(buf, FALSE);
3938 }
3939 no_check_timestamps = save_no_check_timestamps;
3940}
3941#endif
3942
Bram Moolenaar071d4272004-06-13 20:20:40 +00003943#if (defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
3944 && (defined(FEAT_EVAL) || defined(FEAT_MULTI_LANG))
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02003945# define HAVE_GET_LOCALE_VAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00003946static char *get_locale_val __ARGS((int what));
3947
3948 static char *
3949get_locale_val(what)
3950 int what;
3951{
3952 char *loc;
3953
3954 /* Obtain the locale value from the libraries. For DJGPP this is
3955 * redefined and it doesn't use the arguments. */
3956 loc = setlocale(what, NULL);
3957
Bram Moolenaar61660ea2006-04-07 21:40:07 +00003958# ifdef WIN32
Bram Moolenaar071d4272004-06-13 20:20:40 +00003959 if (loc != NULL)
3960 {
3961 char_u *p;
3962
Bram Moolenaar61660ea2006-04-07 21:40:07 +00003963 /* setocale() returns something like "LC_COLLATE=<name>;LC_..." when
3964 * one of the values (e.g., LC_CTYPE) differs. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003965 p = vim_strchr(loc, '=');
3966 if (p != NULL)
3967 {
3968 loc = ++p;
3969 while (*p != NUL) /* remove trailing newline */
3970 {
Bram Moolenaar61660ea2006-04-07 21:40:07 +00003971 if (*p < ' ' || *p == ';')
Bram Moolenaar071d4272004-06-13 20:20:40 +00003972 {
3973 *p = NUL;
3974 break;
3975 }
3976 ++p;
3977 }
3978 }
3979 }
3980# endif
3981
3982 return loc;
3983}
3984#endif
3985
3986
3987#ifdef WIN32
3988/*
3989 * On MS-Windows locale names are strings like "German_Germany.1252", but
3990 * gettext expects "de". Try to translate one into another here for a few
3991 * supported languages.
3992 */
3993 static char_u *
3994gettext_lang(char_u *name)
3995{
3996 int i;
3997 static char *(mtable[]) = {
3998 "afrikaans", "af",
3999 "czech", "cs",
4000 "dutch", "nl",
4001 "german", "de",
4002 "english_united kingdom", "en_GB",
4003 "spanish", "es",
4004 "french", "fr",
4005 "italian", "it",
4006 "japanese", "ja",
4007 "korean", "ko",
4008 "norwegian", "no",
4009 "polish", "pl",
4010 "russian", "ru",
4011 "slovak", "sk",
4012 "swedish", "sv",
4013 "ukrainian", "uk",
4014 "chinese_china", "zh_CN",
4015 "chinese_taiwan", "zh_TW",
4016 NULL};
4017
4018 for (i = 0; mtable[i] != NULL; i += 2)
4019 if (STRNICMP(mtable[i], name, STRLEN(mtable[i])) == 0)
4020 return mtable[i + 1];
4021 return name;
4022}
4023#endif
4024
4025#if defined(FEAT_MULTI_LANG) || defined(PROTO)
4026/*
4027 * Obtain the current messages language. Used to set the default for
4028 * 'helplang'. May return NULL or an empty string.
4029 */
4030 char_u *
4031get_mess_lang()
4032{
4033 char_u *p;
4034
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004035# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00004036# if defined(LC_MESSAGES)
4037 p = (char_u *)get_locale_val(LC_MESSAGES);
4038# else
4039 /* This is necessary for Win32, where LC_MESSAGES is not defined and $LANG
Bram Moolenaar61660ea2006-04-07 21:40:07 +00004040 * may be set to the LCID number. LC_COLLATE is the best guess, LC_TIME
4041 * and LC_MONETARY may be set differently for a Japanese working in the
4042 * US. */
4043 p = (char_u *)get_locale_val(LC_COLLATE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004044# endif
4045# else
4046 p = mch_getenv((char_u *)"LC_ALL");
4047 if (p == NULL || *p == NUL)
4048 {
4049 p = mch_getenv((char_u *)"LC_MESSAGES");
4050 if (p == NULL || *p == NUL)
4051 p = mch_getenv((char_u *)"LANG");
4052 }
4053# endif
4054# ifdef WIN32
4055 p = gettext_lang(p);
4056# endif
4057 return p;
4058}
4059#endif
4060
Bram Moolenaardef9e822004-12-31 20:58:58 +00004061/* Complicated #if; matches with where get_mess_env() is used below. */
4062#if (defined(FEAT_EVAL) && !((defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
4063 && defined(LC_MESSAGES))) \
4064 || ((defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
4065 && (defined(FEAT_GETTEXT) || defined(FEAT_MBYTE)) \
4066 && !defined(LC_MESSAGES))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004067static char_u *get_mess_env __ARGS((void));
4068
4069/*
4070 * Get the language used for messages from the environment.
4071 */
4072 static char_u *
4073get_mess_env()
4074{
4075 char_u *p;
4076
4077 p = mch_getenv((char_u *)"LC_ALL");
4078 if (p == NULL || *p == NUL)
4079 {
4080 p = mch_getenv((char_u *)"LC_MESSAGES");
4081 if (p == NULL || *p == NUL)
4082 {
4083 p = mch_getenv((char_u *)"LANG");
4084 if (p != NULL && VIM_ISDIGIT(*p))
4085 p = NULL; /* ignore something like "1043" */
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004086# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00004087 if (p == NULL || *p == NUL)
4088 p = (char_u *)get_locale_val(LC_CTYPE);
4089# endif
4090 }
4091 }
4092 return p;
4093}
4094#endif
4095
4096#if defined(FEAT_EVAL) || defined(PROTO)
4097
4098/*
4099 * Set the "v:lang" variable according to the current locale setting.
4100 * Also do "v:lc_time"and "v:ctype".
4101 */
4102 void
4103set_lang_var()
4104{
4105 char_u *loc;
4106
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004107# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00004108 loc = (char_u *)get_locale_val(LC_CTYPE);
4109# else
4110 /* setlocale() not supported: use the default value */
4111 loc = (char_u *)"C";
4112# endif
4113 set_vim_var_string(VV_CTYPE, loc, -1);
4114
4115 /* When LC_MESSAGES isn't defined use the value from $LC_MESSAGES, fall
4116 * back to LC_CTYPE if it's empty. */
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004117# if defined(HAVE_GET_LOCALE_VAL) && defined(LC_MESSAGES)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004118 loc = (char_u *)get_locale_val(LC_MESSAGES);
4119# else
4120 loc = get_mess_env();
4121# endif
4122 set_vim_var_string(VV_LANG, loc, -1);
4123
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004124# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00004125 loc = (char_u *)get_locale_val(LC_TIME);
4126# endif
4127 set_vim_var_string(VV_LC_TIME, loc, -1);
4128}
4129#endif
4130
4131#if (defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
4132 && (defined(FEAT_GETTEXT) || defined(FEAT_MBYTE))
4133/*
4134 * ":language": Set the language (locale).
4135 */
4136 void
4137ex_language(eap)
4138 exarg_T *eap;
4139{
4140 char *loc;
4141 char_u *p;
4142 char_u *name;
4143 int what = LC_ALL;
4144 char *whatstr = "";
4145#ifdef LC_MESSAGES
4146# define VIM_LC_MESSAGES LC_MESSAGES
4147#else
4148# define VIM_LC_MESSAGES 6789
4149#endif
4150
4151 name = eap->arg;
4152
4153 /* Check for "messages {name}", "ctype {name}" or "time {name}" argument.
4154 * Allow abbreviation, but require at least 3 characters to avoid
4155 * confusion with a two letter language name "me" or "ct". */
4156 p = skiptowhite(eap->arg);
4157 if ((*p == NUL || vim_iswhite(*p)) && p - eap->arg >= 3)
4158 {
4159 if (STRNICMP(eap->arg, "messages", p - eap->arg) == 0)
4160 {
4161 what = VIM_LC_MESSAGES;
4162 name = skipwhite(p);
4163 whatstr = "messages ";
4164 }
4165 else if (STRNICMP(eap->arg, "ctype", p - eap->arg) == 0)
4166 {
4167 what = LC_CTYPE;
4168 name = skipwhite(p);
4169 whatstr = "ctype ";
4170 }
4171 else if (STRNICMP(eap->arg, "time", p - eap->arg) == 0)
4172 {
4173 what = LC_TIME;
4174 name = skipwhite(p);
4175 whatstr = "time ";
4176 }
4177 }
4178
4179 if (*name == NUL)
4180 {
4181#ifndef LC_MESSAGES
4182 if (what == VIM_LC_MESSAGES)
4183 p = get_mess_env();
4184 else
4185#endif
4186 p = (char_u *)setlocale(what, NULL);
4187 if (p == NULL || *p == NUL)
4188 p = (char_u *)"Unknown";
4189 smsg((char_u *)_("Current %slanguage: \"%s\""), whatstr, p);
4190 }
4191 else
4192 {
4193#ifndef LC_MESSAGES
4194 if (what == VIM_LC_MESSAGES)
4195 loc = "";
4196 else
4197#endif
Bram Moolenaar8c8de832008-06-24 22:58:06 +00004198 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00004199 loc = setlocale(what, (char *)name);
Bram Moolenaar8c8de832008-06-24 22:58:06 +00004200#if defined(FEAT_FLOAT) && defined(LC_NUMERIC)
4201 /* Make sure strtod() uses a decimal point, not a comma. */
4202 setlocale(LC_NUMERIC, "C");
4203#endif
4204 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004205 if (loc == NULL)
4206 EMSG2(_("E197: Cannot set language to \"%s\""), name);
4207 else
4208 {
4209#ifdef HAVE_NL_MSG_CAT_CNTR
4210 /* Need to do this for GNU gettext, otherwise cached translations
4211 * will be used again. */
4212 extern int _nl_msg_cat_cntr;
4213
4214 ++_nl_msg_cat_cntr;
4215#endif
Bram Moolenaarb8017e72007-05-10 18:59:07 +00004216 /* Reset $LC_ALL, otherwise it would overrule everything. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004217 vim_setenv((char_u *)"LC_ALL", (char_u *)"");
4218
4219 if (what != LC_TIME)
4220 {
4221 /* Tell gettext() what to translate to. It apparently doesn't
4222 * use the currently effective locale. Also do this when
4223 * FEAT_GETTEXT isn't defined, so that shell commands use this
4224 * value. */
4225 if (what == LC_ALL)
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00004226 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00004227 vim_setenv((char_u *)"LANG", name);
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00004228# ifdef WIN32
4229 /* Apparently MS-Windows printf() may cause a crash when
4230 * we give it 8-bit text while it's expecting text in the
4231 * current locale. This call avoids that. */
4232 setlocale(LC_CTYPE, "C");
4233# endif
4234 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004235 if (what != LC_CTYPE)
4236 {
4237 char_u *mname;
4238#ifdef WIN32
4239 mname = gettext_lang(name);
4240#else
4241 mname = name;
4242#endif
4243 vim_setenv((char_u *)"LC_MESSAGES", mname);
4244#ifdef FEAT_MULTI_LANG
4245 set_helplang_default(mname);
4246#endif
4247 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004248 }
4249
4250# ifdef FEAT_EVAL
4251 /* Set v:lang, v:lc_time and v:ctype to the final result. */
4252 set_lang_var();
4253# endif
Bram Moolenaar6cc00c72011-10-20 21:41:09 +02004254# ifdef FEAT_TITLE
4255 maketitle();
4256# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004257 }
4258 }
4259}
4260
4261# if defined(FEAT_CMDL_COMPL) || defined(PROTO)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02004262
4263static char_u **locales = NULL; /* Array of all available locales */
4264static int did_init_locales = FALSE;
4265
4266static void init_locales __ARGS((void));
4267static char_u **find_locales __ARGS((void));
4268
4269/*
4270 * Lazy initialization of all available locales.
4271 */
4272 static void
4273init_locales()
4274{
4275 if (!did_init_locales)
4276 {
4277 did_init_locales = TRUE;
4278 locales = find_locales();
4279 }
4280}
4281
4282/* Return an array of strings for all available locales + NULL for the
4283 * last element. Return NULL in case of error. */
4284 static char_u **
4285find_locales()
4286{
4287 garray_T locales_ga;
4288 char_u *loc;
4289
4290 /* Find all available locales by running command "locale -a". If this
4291 * doesn't work we won't have completion. */
4292 char_u *locale_a = get_cmd_output((char_u *)"locale -a",
4293 NULL, SHELL_SILENT);
4294 if (locale_a == NULL)
4295 return NULL;
4296 ga_init2(&locales_ga, sizeof(char_u *), 20);
4297
4298 /* Transform locale_a string where each locale is separated by "\n"
4299 * into an array of locale strings. */
4300 loc = (char_u *)strtok((char *)locale_a, "\n");
4301
4302 while (loc != NULL)
4303 {
4304 if (ga_grow(&locales_ga, 1) == FAIL)
4305 break;
4306 loc = vim_strsave(loc);
4307 if (loc == NULL)
4308 break;
4309
4310 ((char_u **)locales_ga.ga_data)[locales_ga.ga_len++] = loc;
4311 loc = (char_u *)strtok(NULL, "\n");
4312 }
4313 vim_free(locale_a);
4314 if (ga_grow(&locales_ga, 1) == FAIL)
4315 {
4316 ga_clear(&locales_ga);
4317 return NULL;
4318 }
4319 ((char_u **)locales_ga.ga_data)[locales_ga.ga_len] = NULL;
4320 return (char_u **)locales_ga.ga_data;
4321}
4322
4323# if defined(EXITFREE) || defined(PROTO)
4324 void
4325free_locales()
4326{
4327 int i;
4328 if (locales != NULL)
4329 {
4330 for (i = 0; locales[i] != NULL; i++)
4331 vim_free(locales[i]);
4332 vim_free(locales);
4333 locales = NULL;
4334 }
4335}
4336# endif
4337
Bram Moolenaar071d4272004-06-13 20:20:40 +00004338/*
4339 * Function given to ExpandGeneric() to obtain the possible arguments of the
4340 * ":language" command.
4341 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004342 char_u *
4343get_lang_arg(xp, idx)
Bram Moolenaar0c094b92009-05-14 20:20:33 +00004344 expand_T *xp UNUSED;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004345 int idx;
4346{
4347 if (idx == 0)
4348 return (char_u *)"messages";
4349 if (idx == 1)
4350 return (char_u *)"ctype";
4351 if (idx == 2)
4352 return (char_u *)"time";
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02004353
4354 init_locales();
4355 if (locales == NULL)
4356 return NULL;
4357 return locales[idx - 3];
4358}
4359
4360/*
4361 * Function given to ExpandGeneric() to obtain the available locales.
4362 */
4363 char_u *
4364get_locales(xp, idx)
4365 expand_T *xp UNUSED;
4366 int idx;
4367{
4368 init_locales();
4369 if (locales == NULL)
4370 return NULL;
4371 return locales[idx];
Bram Moolenaar071d4272004-06-13 20:20:40 +00004372}
4373# endif
4374
4375#endif