blob: 18930a6cfb26764d24effa521028387609c7077e [file] [log] [blame]
Bram Moolenaaredf3f972016-08-29 22:49:24 +02001/* vi:set ts=8 sts=4 sw=4 noet:
Bram Moolenaar071d4272004-06-13 20:20:40 +00002 *
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
Bram Moolenaarf28dbce2016-01-29 22:03:47 +010017static void cmd_source(char_u *fname, exarg_T *eap);
Bram Moolenaar071d4272004-06-13 20:20:40 +000018
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. */
Bram Moolenaarf1f60f82016-01-16 15:40:53 +010071static int get_maxbacktrace_level(void);
72static void do_setdebugtracelevel(char_u *arg);
73static void do_checkbacktracelevel(void);
74static void do_showbacktrace(char_u *cmd);
Bram Moolenaar071d4272004-06-13 20:20:40 +000075
76/*
77 * do_debug(): Debug mode.
78 * Repeatedly get Ex commands, until told to continue normal execution.
79 */
80 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +010081do_debug(char_u *cmd)
Bram Moolenaar071d4272004-06-13 20:20:40 +000082{
83 int save_msg_scroll = msg_scroll;
84 int save_State = State;
85 int save_did_emsg = did_emsg;
86 int save_cmd_silent = cmd_silent;
87 int save_msg_silent = msg_silent;
88 int save_emsg_silent = emsg_silent;
89 int save_redir_off = redir_off;
90 tasave_T typeaheadbuf;
Bram Moolenaaree3f7a52008-01-01 13:17:56 +000091 int typeahead_saved = FALSE;
Bram Moolenaar383c6f52008-01-04 15:01:07 +000092 int save_ignore_script = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +000093 int save_ex_normal_busy;
Bram Moolenaar071d4272004-06-13 20:20:40 +000094 int n;
95 char_u *cmdline = NULL;
96 char_u *p;
97 char *tail = NULL;
98 static int last_cmd = 0;
99#define CMD_CONT 1
100#define CMD_NEXT 2
101#define CMD_STEP 3
102#define CMD_FINISH 4
103#define CMD_QUIT 5
104#define CMD_INTERRUPT 6
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100105#define CMD_BACKTRACE 7
106#define CMD_FRAME 8
107#define CMD_UP 9
108#define CMD_DOWN 10
Bram Moolenaar071d4272004-06-13 20:20:40 +0000109
110#ifdef ALWAYS_USE_GUI
111 /* Can't do this when there is no terminal for input/output. */
112 if (!gui.in_use)
113 {
114 /* Break as soon as possible. */
115 debug_break_level = 9999;
116 return;
117 }
118#endif
119
120 /* Make sure we are in raw mode and start termcap mode. Might have side
121 * effects... */
122 settmode(TMODE_RAW);
123 starttermcap();
124
125 ++RedrawingDisabled; /* don't redisplay the window */
126 ++no_wait_return; /* don't wait for return */
127 did_emsg = FALSE; /* don't use error from debugged stuff */
128 cmd_silent = FALSE; /* display commands */
129 msg_silent = FALSE; /* display messages */
130 emsg_silent = FALSE; /* display error messages */
131 redir_off = TRUE; /* don't redirect debug commands */
132
133 State = NORMAL;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000134
135 if (!debug_did_msg)
136 MSG(_("Entering Debug mode. Type \"cont\" to continue."));
137 if (sourcing_name != NULL)
138 msg(sourcing_name);
139 if (sourcing_lnum != 0)
Bram Moolenaar555b2802005-05-19 21:08:39 +0000140 smsg((char_u *)_("line %ld: %s"), (long)sourcing_lnum, cmd);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000141 else
Bram Moolenaar555b2802005-05-19 21:08:39 +0000142 smsg((char_u *)_("cmd: %s"), cmd);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000143
144 /*
145 * Repeat getting a command and executing it.
146 */
147 for (;;)
148 {
149 msg_scroll = TRUE;
150 need_wait_return = FALSE;
Bram Moolenaar85b11762016-02-27 18:13:23 +0100151
Bram Moolenaar071d4272004-06-13 20:20:40 +0000152 /* 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 save_ex_normal_busy = ex_normal_busy;
158 ex_normal_busy = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000159 if (!debug_greedy)
Bram Moolenaaree3f7a52008-01-01 13:17:56 +0000160 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000161 save_typeahead(&typeaheadbuf);
Bram Moolenaaree3f7a52008-01-01 13:17:56 +0000162 typeahead_saved = TRUE;
163 save_ignore_script = ignore_script;
164 ignore_script = TRUE;
165 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000166
Bram Moolenaardc303bc2016-05-17 17:45:38 +0200167 vim_free(cmdline);
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +0000168 cmdline = getcmdline_prompt('>', NULL, 0, EXPAND_NOTHING, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000169
Bram Moolenaaree3f7a52008-01-01 13:17:56 +0000170 if (typeahead_saved)
171 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000172 restore_typeahead(&typeaheadbuf);
Bram Moolenaaree3f7a52008-01-01 13:17:56 +0000173 ignore_script = save_ignore_script;
174 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000175 ex_normal_busy = save_ex_normal_busy;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000176
177 cmdline_row = msg_row;
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100178 msg_starthere();
Bram Moolenaar071d4272004-06-13 20:20:40 +0000179 if (cmdline != NULL)
180 {
181 /* If this is a debug command, set "last_cmd".
182 * If not, reset "last_cmd".
183 * For a blank line use previous command. */
184 p = skipwhite(cmdline);
185 if (*p != NUL)
186 {
187 switch (*p)
188 {
189 case 'c': last_cmd = CMD_CONT;
190 tail = "ont";
191 break;
192 case 'n': last_cmd = CMD_NEXT;
193 tail = "ext";
194 break;
195 case 's': last_cmd = CMD_STEP;
196 tail = "tep";
197 break;
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100198 case 'f':
199 last_cmd = 0;
200 if (p[1] == 'r')
201 {
202 last_cmd = CMD_FRAME;
203 tail = "rame";
204 }
205 else
206 {
207 last_cmd = CMD_FINISH;
208 tail = "inish";
209 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000210 break;
211 case 'q': last_cmd = CMD_QUIT;
212 tail = "uit";
213 break;
214 case 'i': last_cmd = CMD_INTERRUPT;
215 tail = "nterrupt";
216 break;
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100217 case 'b': last_cmd = CMD_BACKTRACE;
218 if (p[1] == 't')
219 tail = "t";
220 else
221 tail = "acktrace";
222 break;
223 case 'w': last_cmd = CMD_BACKTRACE;
224 tail = "here";
225 break;
226 case 'u': last_cmd = CMD_UP;
227 tail = "p";
228 break;
229 case 'd': last_cmd = CMD_DOWN;
230 tail = "own";
231 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000232 default: last_cmd = 0;
233 }
234 if (last_cmd != 0)
235 {
236 /* Check that the tail matches. */
237 ++p;
238 while (*p != NUL && *p == *tail)
239 {
240 ++p;
241 ++tail;
242 }
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100243 if (ASCII_ISALPHA(*p) && last_cmd != CMD_FRAME)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000244 last_cmd = 0;
245 }
246 }
247
248 if (last_cmd != 0)
249 {
250 /* Execute debug command: decided where to break next and
251 * return. */
252 switch (last_cmd)
253 {
254 case CMD_CONT:
255 debug_break_level = -1;
256 break;
257 case CMD_NEXT:
258 debug_break_level = ex_nesting_level;
259 break;
260 case CMD_STEP:
261 debug_break_level = 9999;
262 break;
263 case CMD_FINISH:
264 debug_break_level = ex_nesting_level - 1;
265 break;
266 case CMD_QUIT:
267 got_int = TRUE;
268 debug_break_level = -1;
269 break;
270 case CMD_INTERRUPT:
271 got_int = TRUE;
272 debug_break_level = 9999;
273 /* Do not repeat ">interrupt" cmd, continue stepping. */
274 last_cmd = CMD_STEP;
275 break;
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100276 case CMD_BACKTRACE:
277 do_showbacktrace(cmd);
278 continue;
279 case CMD_FRAME:
280 if (*p == NUL)
281 {
282 do_showbacktrace(cmd);
283 }
284 else
285 {
286 p = skipwhite(p);
287 do_setdebugtracelevel(p);
288 }
289 continue;
290 case CMD_UP:
291 debug_backtrace_level++;
292 do_checkbacktracelevel();
293 continue;
294 case CMD_DOWN:
295 debug_backtrace_level--;
296 do_checkbacktracelevel();
297 continue;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000298 }
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100299 /* Going out reset backtrace_level */
300 debug_backtrace_level = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000301 break;
302 }
303
304 /* don't debug this command */
305 n = debug_break_level;
306 debug_break_level = -1;
307 (void)do_cmdline(cmdline, getexline, NULL,
308 DOCMD_VERBOSE|DOCMD_EXCRESET);
309 debug_break_level = n;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000310 }
311 lines_left = Rows - 1;
312 }
313 vim_free(cmdline);
314
315 --RedrawingDisabled;
316 --no_wait_return;
317 redraw_all_later(NOT_VALID);
318 need_wait_return = FALSE;
319 msg_scroll = save_msg_scroll;
320 lines_left = Rows - 1;
321 State = save_State;
322 did_emsg = save_did_emsg;
323 cmd_silent = save_cmd_silent;
324 msg_silent = save_msg_silent;
325 emsg_silent = save_emsg_silent;
326 redir_off = save_redir_off;
327
328 /* Only print the message again when typing a command before coming back
329 * here. */
330 debug_did_msg = TRUE;
331}
332
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100333 static int
334get_maxbacktrace_level(void)
335{
336 char *p, *q;
Bram Moolenaardc633cf2016-04-23 14:33:19 +0200337 int maxbacktrace = 0;
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100338
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100339 if (sourcing_name != NULL)
340 {
341 p = (char *)sourcing_name;
342 while ((q = strstr(p, "..")) != NULL)
343 {
344 p = q + 2;
345 maxbacktrace++;
346 }
347 }
348 return maxbacktrace;
349}
350
351 static void
352do_setdebugtracelevel(char_u *arg)
353{
354 int level;
355
356 level = atoi((char *)arg);
357 if (*arg == '+' || level < 0)
358 debug_backtrace_level += level;
359 else
360 debug_backtrace_level = level;
361
362 do_checkbacktracelevel();
363}
364
365 static void
366do_checkbacktracelevel(void)
367{
368 if (debug_backtrace_level < 0)
369 {
370 debug_backtrace_level = 0;
371 MSG(_("frame is zero"));
372 }
373 else
374 {
375 int max = get_maxbacktrace_level();
376
377 if (debug_backtrace_level > max)
378 {
379 debug_backtrace_level = max;
380 smsg((char_u *)_("frame at highest level: %d"), max);
381 }
382 }
383}
384
385 static void
386do_showbacktrace(char_u *cmd)
387{
388 char *cur;
389 char *next;
390 int i = 0;
391 int max = get_maxbacktrace_level();
392
393 if (sourcing_name != NULL)
394 {
395 cur = (char *)sourcing_name;
396 while (!got_int)
397 {
398 next = strstr(cur, "..");
399 if (next != NULL)
400 *next = NUL;
401 if (i == max - debug_backtrace_level)
402 smsg((char_u *)"->%d %s", max - i, cur);
403 else
404 smsg((char_u *)" %d %s", max - i, cur);
405 ++i;
406 if (next == NULL)
407 break;
408 *next = '.';
409 cur = next + 2;
410 }
411 }
412 if (sourcing_lnum != 0)
413 smsg((char_u *)_("line %ld: %s"), (long)sourcing_lnum, cmd);
414 else
415 smsg((char_u *)_("cmd: %s"), cmd);
416}
417
Bram Moolenaar071d4272004-06-13 20:20:40 +0000418/*
419 * ":debug".
420 */
421 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100422ex_debug(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000423{
424 int debug_break_level_save = debug_break_level;
425
426 debug_break_level = 9999;
427 do_cmdline_cmd(eap->arg);
428 debug_break_level = debug_break_level_save;
429}
430
431static char_u *debug_breakpoint_name = NULL;
432static linenr_T debug_breakpoint_lnum;
433
434/*
435 * When debugging or a breakpoint is set on a skipped command, no debug prompt
436 * is shown by do_one_cmd(). This situation is indicated by debug_skipped, and
437 * debug_skipped_name is then set to the source name in the breakpoint case. If
438 * a skipped command decides itself that a debug prompt should be displayed, it
439 * can do so by calling dbg_check_skipped().
440 */
441static int debug_skipped;
442static char_u *debug_skipped_name;
443
444/*
445 * Go to debug mode when a breakpoint was encountered or "ex_nesting_level" is
446 * at or below the break level. But only when the line is actually
447 * executed. Return TRUE and set breakpoint_name for skipped commands that
448 * decide to execute something themselves.
449 * Called from do_one_cmd() before executing a command.
450 */
451 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100452dbg_check_breakpoint(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000453{
454 char_u *p;
455
456 debug_skipped = FALSE;
457 if (debug_breakpoint_name != NULL)
458 {
459 if (!eap->skip)
460 {
461 /* replace K_SNR with "<SNR>" */
462 if (debug_breakpoint_name[0] == K_SPECIAL
463 && debug_breakpoint_name[1] == KS_EXTRA
464 && debug_breakpoint_name[2] == (int)KE_SNR)
465 p = (char_u *)"<SNR>";
466 else
467 p = (char_u *)"";
Bram Moolenaar555b2802005-05-19 21:08:39 +0000468 smsg((char_u *)_("Breakpoint in \"%s%s\" line %ld"),
469 p,
Bram Moolenaar071d4272004-06-13 20:20:40 +0000470 debug_breakpoint_name + (*p == NUL ? 0 : 3),
471 (long)debug_breakpoint_lnum);
472 debug_breakpoint_name = NULL;
473 do_debug(eap->cmd);
474 }
475 else
476 {
477 debug_skipped = TRUE;
478 debug_skipped_name = debug_breakpoint_name;
479 debug_breakpoint_name = NULL;
480 }
481 }
482 else if (ex_nesting_level <= debug_break_level)
483 {
484 if (!eap->skip)
485 do_debug(eap->cmd);
486 else
487 {
488 debug_skipped = TRUE;
489 debug_skipped_name = NULL;
490 }
491 }
492}
493
494/*
495 * Go to debug mode if skipped by dbg_check_breakpoint() because eap->skip was
496 * set. Return TRUE when the debug mode is entered this time.
497 */
498 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100499dbg_check_skipped(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000500{
501 int prev_got_int;
502
503 if (debug_skipped)
504 {
505 /*
506 * Save the value of got_int and reset it. We don't want a previous
507 * interruption cause flushing the input buffer.
508 */
509 prev_got_int = got_int;
510 got_int = FALSE;
511 debug_breakpoint_name = debug_skipped_name;
512 /* eap->skip is TRUE */
513 eap->skip = FALSE;
514 (void)dbg_check_breakpoint(eap);
515 eap->skip = TRUE;
516 got_int |= prev_got_int;
517 return TRUE;
518 }
519 return FALSE;
520}
521
522/*
523 * The list of breakpoints: dbg_breakp.
524 * This is a grow-array of structs.
525 */
526struct debuggy
527{
528 int dbg_nr; /* breakpoint number */
529 int dbg_type; /* DBG_FUNC or DBG_FILE */
530 char_u *dbg_name; /* function or file name */
531 regprog_T *dbg_prog; /* regexp program */
532 linenr_T dbg_lnum; /* line number in function or file */
Bram Moolenaar05159a02005-02-26 23:04:13 +0000533 int dbg_forceit; /* ! used */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000534};
535
536static garray_T dbg_breakp = {0, 0, sizeof(struct debuggy), 4, NULL};
Bram Moolenaar05159a02005-02-26 23:04:13 +0000537#define BREAKP(idx) (((struct debuggy *)dbg_breakp.ga_data)[idx])
538#define DEBUGGY(gap, idx) (((struct debuggy *)gap->ga_data)[idx])
Bram Moolenaar071d4272004-06-13 20:20:40 +0000539static int last_breakp = 0; /* nr of last defined breakpoint */
540
Bram Moolenaar05159a02005-02-26 23:04:13 +0000541#ifdef FEAT_PROFILE
542/* Profiling uses file and func names similar to breakpoints. */
543static garray_T prof_ga = {0, 0, sizeof(struct debuggy), 4, NULL};
544#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000545#define DBG_FUNC 1
546#define DBG_FILE 2
547
Bram Moolenaarf28dbce2016-01-29 22:03:47 +0100548static int dbg_parsearg(char_u *arg, garray_T *gap);
549static linenr_T debuggy_find(int file,char_u *fname, linenr_T after, garray_T *gap, int *fp);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000550
551/*
Bram Moolenaar05159a02005-02-26 23:04:13 +0000552 * Parse the arguments of ":profile", ":breakadd" or ":breakdel" and put them
553 * in the entry just after the last one in dbg_breakp. Note that "dbg_name"
554 * is allocated.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000555 * Returns FAIL for failure.
556 */
557 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100558dbg_parsearg(
559 char_u *arg,
560 garray_T *gap) /* either &dbg_breakp or &prof_ga */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000561{
562 char_u *p = arg;
563 char_u *q;
564 struct debuggy *bp;
Bram Moolenaarf4b8e572004-06-24 15:53:16 +0000565 int here = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000566
Bram Moolenaar05159a02005-02-26 23:04:13 +0000567 if (ga_grow(gap, 1) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000568 return FAIL;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000569 bp = &DEBUGGY(gap, gap->ga_len);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000570
571 /* Find "func" or "file". */
572 if (STRNCMP(p, "func", 4) == 0)
573 bp->dbg_type = DBG_FUNC;
574 else if (STRNCMP(p, "file", 4) == 0)
575 bp->dbg_type = DBG_FILE;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000576 else if (
577#ifdef FEAT_PROFILE
578 gap != &prof_ga &&
579#endif
580 STRNCMP(p, "here", 4) == 0)
Bram Moolenaarf4b8e572004-06-24 15:53:16 +0000581 {
582 if (curbuf->b_ffname == NULL)
583 {
584 EMSG(_(e_noname));
585 return FAIL;
586 }
587 bp->dbg_type = DBG_FILE;
588 here = TRUE;
589 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000590 else
591 {
592 EMSG2(_(e_invarg2), p);
593 return FAIL;
594 }
595 p = skipwhite(p + 4);
596
597 /* Find optional line number. */
Bram Moolenaarf4b8e572004-06-24 15:53:16 +0000598 if (here)
599 bp->dbg_lnum = curwin->w_cursor.lnum;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000600 else if (
601#ifdef FEAT_PROFILE
602 gap != &prof_ga &&
603#endif
604 VIM_ISDIGIT(*p))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000605 {
606 bp->dbg_lnum = getdigits(&p);
607 p = skipwhite(p);
608 }
609 else
610 bp->dbg_lnum = 0;
611
612 /* Find the function or file name. Don't accept a function name with (). */
Bram Moolenaarf4b8e572004-06-24 15:53:16 +0000613 if ((!here && *p == NUL)
614 || (here && *p != NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000615 || (bp->dbg_type == DBG_FUNC && strstr((char *)p, "()") != NULL))
616 {
617 EMSG2(_(e_invarg2), arg);
618 return FAIL;
619 }
620
621 if (bp->dbg_type == DBG_FUNC)
622 bp->dbg_name = vim_strsave(p);
Bram Moolenaarf4b8e572004-06-24 15:53:16 +0000623 else if (here)
624 bp->dbg_name = vim_strsave(curbuf->b_ffname);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000625 else
626 {
627 /* Expand the file name in the same way as do_source(). This means
628 * doing it twice, so that $DIR/file gets expanded when $DIR is
629 * "~/dir". */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000630 q = expand_env_save(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000631 if (q == NULL)
632 return FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000633 p = expand_env_save(q);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000634 vim_free(q);
635 if (p == NULL)
636 return FAIL;
Bram Moolenaar843ee412004-06-30 16:16:41 +0000637 if (*p != '*')
638 {
639 bp->dbg_name = fix_fname(p);
640 vim_free(p);
641 }
642 else
643 bp->dbg_name = p;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000644 }
645
646 if (bp->dbg_name == NULL)
647 return FAIL;
648 return OK;
649}
650
651/*
652 * ":breakadd".
653 */
654 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100655ex_breakadd(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000656{
657 struct debuggy *bp;
658 char_u *pat;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000659 garray_T *gap;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000660
Bram Moolenaar05159a02005-02-26 23:04:13 +0000661 gap = &dbg_breakp;
662#ifdef FEAT_PROFILE
663 if (eap->cmdidx == CMD_profile)
664 gap = &prof_ga;
665#endif
666
667 if (dbg_parsearg(eap->arg, gap) == OK)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000668 {
Bram Moolenaar05159a02005-02-26 23:04:13 +0000669 bp = &DEBUGGY(gap, gap->ga_len);
670 bp->dbg_forceit = eap->forceit;
671
Bram Moolenaar071d4272004-06-13 20:20:40 +0000672 pat = file_pat_to_reg_pat(bp->dbg_name, NULL, NULL, FALSE);
673 if (pat != NULL)
674 {
675 bp->dbg_prog = vim_regcomp(pat, RE_MAGIC + RE_STRING);
676 vim_free(pat);
677 }
678 if (pat == NULL || bp->dbg_prog == NULL)
679 vim_free(bp->dbg_name);
680 else
681 {
682 if (bp->dbg_lnum == 0) /* default line number is 1 */
683 bp->dbg_lnum = 1;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000684#ifdef FEAT_PROFILE
685 if (eap->cmdidx != CMD_profile)
686#endif
687 {
688 DEBUGGY(gap, gap->ga_len).dbg_nr = ++last_breakp;
689 ++debug_tick;
690 }
691 ++gap->ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000692 }
693 }
694}
695
696/*
697 * ":debuggreedy".
698 */
699 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100700ex_debuggreedy(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000701{
702 if (eap->addr_count == 0 || eap->line2 != 0)
703 debug_greedy = TRUE;
704 else
705 debug_greedy = FALSE;
706}
707
708/*
Bram Moolenaard9fba312005-06-26 22:34:35 +0000709 * ":breakdel" and ":profdel".
Bram Moolenaar071d4272004-06-13 20:20:40 +0000710 */
711 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100712ex_breakdel(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000713{
714 struct debuggy *bp, *bpi;
715 int nr;
716 int todel = -1;
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000717 int del_all = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000718 int i;
719 linenr_T best_lnum = 0;
Bram Moolenaard9fba312005-06-26 22:34:35 +0000720 garray_T *gap;
721
722 gap = &dbg_breakp;
Bram Moolenaard9fba312005-06-26 22:34:35 +0000723 if (eap->cmdidx == CMD_profdel)
Bram Moolenaar38bdbd62012-06-20 15:48:57 +0200724 {
725#ifdef FEAT_PROFILE
Bram Moolenaard9fba312005-06-26 22:34:35 +0000726 gap = &prof_ga;
Bram Moolenaar38bdbd62012-06-20 15:48:57 +0200727#else
728 ex_ni(eap);
729 return;
Bram Moolenaard9fba312005-06-26 22:34:35 +0000730#endif
Bram Moolenaar38bdbd62012-06-20 15:48:57 +0200731 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000732
733 if (vim_isdigit(*eap->arg))
734 {
735 /* ":breakdel {nr}" */
736 nr = atol((char *)eap->arg);
Bram Moolenaard9fba312005-06-26 22:34:35 +0000737 for (i = 0; i < gap->ga_len; ++i)
738 if (DEBUGGY(gap, i).dbg_nr == nr)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000739 {
740 todel = i;
741 break;
742 }
743 }
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000744 else if (*eap->arg == '*')
745 {
746 todel = 0;
747 del_all = TRUE;
748 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000749 else
750 {
751 /* ":breakdel {func|file} [lnum] {name}" */
Bram Moolenaard9fba312005-06-26 22:34:35 +0000752 if (dbg_parsearg(eap->arg, gap) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000753 return;
Bram Moolenaard9fba312005-06-26 22:34:35 +0000754 bp = &DEBUGGY(gap, gap->ga_len);
755 for (i = 0; i < gap->ga_len; ++i)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000756 {
Bram Moolenaard9fba312005-06-26 22:34:35 +0000757 bpi = &DEBUGGY(gap, i);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000758 if (bp->dbg_type == bpi->dbg_type
759 && STRCMP(bp->dbg_name, bpi->dbg_name) == 0
760 && (bp->dbg_lnum == bpi->dbg_lnum
761 || (bp->dbg_lnum == 0
762 && (best_lnum == 0
763 || bpi->dbg_lnum < best_lnum))))
764 {
765 todel = i;
766 best_lnum = bpi->dbg_lnum;
767 }
768 }
769 vim_free(bp->dbg_name);
770 }
771
772 if (todel < 0)
773 EMSG2(_("E161: Breakpoint not found: %s"), eap->arg);
774 else
Bram Moolenaard9fba312005-06-26 22:34:35 +0000775 {
776 while (gap->ga_len > 0)
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000777 {
Bram Moolenaard9fba312005-06-26 22:34:35 +0000778 vim_free(DEBUGGY(gap, todel).dbg_name);
Bram Moolenaar473de612013-06-08 18:19:48 +0200779 vim_regfree(DEBUGGY(gap, todel).dbg_prog);
Bram Moolenaard9fba312005-06-26 22:34:35 +0000780 --gap->ga_len;
781 if (todel < gap->ga_len)
782 mch_memmove(&DEBUGGY(gap, todel), &DEBUGGY(gap, todel + 1),
783 (gap->ga_len - todel) * sizeof(struct debuggy));
784#ifdef FEAT_PROFILE
785 if (eap->cmdidx == CMD_breakdel)
786#endif
787 ++debug_tick;
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000788 if (!del_all)
789 break;
790 }
Bram Moolenaard9fba312005-06-26 22:34:35 +0000791
792 /* If all breakpoints were removed clear the array. */
793 if (gap->ga_len == 0)
794 ga_clear(gap);
795 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000796}
797
798/*
799 * ":breaklist".
800 */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000801 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100802ex_breaklist(exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000803{
804 struct debuggy *bp;
805 int i;
806
807 if (dbg_breakp.ga_len == 0)
808 MSG(_("No breakpoints defined"));
809 else
810 for (i = 0; i < dbg_breakp.ga_len; ++i)
811 {
812 bp = &BREAKP(i);
Bram Moolenaard58ea072011-06-26 04:25:30 +0200813 if (bp->dbg_type == DBG_FILE)
814 home_replace(NULL, bp->dbg_name, NameBuff, MAXPATHL, TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000815 smsg((char_u *)_("%3d %s %s line %ld"),
816 bp->dbg_nr,
817 bp->dbg_type == DBG_FUNC ? "func" : "file",
Bram Moolenaard58ea072011-06-26 04:25:30 +0200818 bp->dbg_type == DBG_FUNC ? bp->dbg_name : NameBuff,
Bram Moolenaar071d4272004-06-13 20:20:40 +0000819 (long)bp->dbg_lnum);
820 }
821}
822
823/*
824 * Find a breakpoint for a function or sourced file.
825 * Returns line number at which to break; zero when no matching breakpoint.
826 */
827 linenr_T
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100828dbg_find_breakpoint(
829 int file, /* TRUE for a file, FALSE for a function */
830 char_u *fname, /* file or function name */
831 linenr_T after) /* after this line number */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000832{
Bram Moolenaar05159a02005-02-26 23:04:13 +0000833 return debuggy_find(file, fname, after, &dbg_breakp, NULL);
834}
835
836#if defined(FEAT_PROFILE) || defined(PROTO)
837/*
838 * Return TRUE if profiling is on for a function or sourced file.
839 */
840 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100841has_profiling(
842 int file, /* TRUE for a file, FALSE for a function */
843 char_u *fname, /* file or function name */
844 int *fp) /* return: forceit */
Bram Moolenaar05159a02005-02-26 23:04:13 +0000845{
846 return (debuggy_find(file, fname, (linenr_T)0, &prof_ga, fp)
847 != (linenr_T)0);
848}
849#endif
850
851/*
852 * Common code for dbg_find_breakpoint() and has_profiling().
853 */
854 static linenr_T
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100855debuggy_find(
856 int file, /* TRUE for a file, FALSE for a function */
857 char_u *fname, /* file or function name */
858 linenr_T after, /* after this line number */
859 garray_T *gap, /* either &dbg_breakp or &prof_ga */
860 int *fp) /* if not NULL: return forceit */
Bram Moolenaar05159a02005-02-26 23:04:13 +0000861{
Bram Moolenaar071d4272004-06-13 20:20:40 +0000862 struct debuggy *bp;
863 int i;
864 linenr_T lnum = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000865 char_u *name = fname;
866 int prev_got_int;
867
Bram Moolenaar05159a02005-02-26 23:04:13 +0000868 /* Return quickly when there are no breakpoints. */
869 if (gap->ga_len == 0)
870 return (linenr_T)0;
871
Bram Moolenaar071d4272004-06-13 20:20:40 +0000872 /* Replace K_SNR in function name with "<SNR>". */
873 if (!file && fname[0] == K_SPECIAL)
874 {
875 name = alloc((unsigned)STRLEN(fname) + 3);
876 if (name == NULL)
877 name = fname;
878 else
879 {
880 STRCPY(name, "<SNR>");
881 STRCPY(name + 5, fname + 3);
882 }
883 }
884
Bram Moolenaar05159a02005-02-26 23:04:13 +0000885 for (i = 0; i < gap->ga_len; ++i)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000886 {
Bram Moolenaar05159a02005-02-26 23:04:13 +0000887 /* Skip entries that are not useful or are for a line that is beyond
888 * an already found breakpoint. */
889 bp = &DEBUGGY(gap, i);
890 if (((bp->dbg_type == DBG_FILE) == file && (
891#ifdef FEAT_PROFILE
892 gap == &prof_ga ||
893#endif
894 (bp->dbg_lnum > after && (lnum == 0 || bp->dbg_lnum < lnum)))))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000895 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000896 /*
Bram Moolenaar05159a02005-02-26 23:04:13 +0000897 * Save the value of got_int and reset it. We don't want a
898 * previous interruption cancel matching, only hitting CTRL-C
899 * while matching should abort it.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000900 */
901 prev_got_int = got_int;
902 got_int = FALSE;
Bram Moolenaardffa5b82014-11-19 16:38:07 +0100903 if (vim_regexec_prog(&bp->dbg_prog, FALSE, name, (colnr_T)0))
Bram Moolenaar05159a02005-02-26 23:04:13 +0000904 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000905 lnum = bp->dbg_lnum;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000906 if (fp != NULL)
907 *fp = bp->dbg_forceit;
908 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000909 got_int |= prev_got_int;
910 }
911 }
912 if (name != fname)
913 vim_free(name);
914
915 return lnum;
916}
917
918/*
919 * Called when a breakpoint was encountered.
920 */
921 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100922dbg_breakpoint(char_u *name, linenr_T lnum)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000923{
924 /* We need to check if this line is actually executed in do_one_cmd() */
925 debug_breakpoint_name = name;
926 debug_breakpoint_lnum = lnum;
927}
Bram Moolenaar05159a02005-02-26 23:04:13 +0000928
929
Bram Moolenaar433f7c82006-03-21 21:29:36 +0000930# if defined(FEAT_PROFILE) || defined(FEAT_RELTIME) || defined(PROTO)
Bram Moolenaar05159a02005-02-26 23:04:13 +0000931/*
932 * Store the current time in "tm".
933 */
934 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100935profile_start(proftime_T *tm)
Bram Moolenaar05159a02005-02-26 23:04:13 +0000936{
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000937# ifdef WIN3264
938 QueryPerformanceCounter(tm);
939# else
Bram Moolenaar05159a02005-02-26 23:04:13 +0000940 gettimeofday(tm, NULL);
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000941# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +0000942}
943
944/*
945 * Compute the elapsed time from "tm" till now and store in "tm".
946 */
947 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100948profile_end(proftime_T *tm)
Bram Moolenaar05159a02005-02-26 23:04:13 +0000949{
950 proftime_T now;
951
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000952# ifdef WIN3264
953 QueryPerformanceCounter(&now);
954 tm->QuadPart = now.QuadPart - tm->QuadPart;
955# else
Bram Moolenaar05159a02005-02-26 23:04:13 +0000956 gettimeofday(&now, NULL);
957 tm->tv_usec = now.tv_usec - tm->tv_usec;
958 tm->tv_sec = now.tv_sec - tm->tv_sec;
959 if (tm->tv_usec < 0)
960 {
961 tm->tv_usec += 1000000;
962 --tm->tv_sec;
963 }
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000964# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +0000965}
966
967/*
968 * Subtract the time "tm2" from "tm".
969 */
970 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100971profile_sub(proftime_T *tm, proftime_T *tm2)
Bram Moolenaar05159a02005-02-26 23:04:13 +0000972{
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000973# ifdef WIN3264
974 tm->QuadPart -= tm2->QuadPart;
975# else
Bram Moolenaar05159a02005-02-26 23:04:13 +0000976 tm->tv_usec -= tm2->tv_usec;
977 tm->tv_sec -= tm2->tv_sec;
978 if (tm->tv_usec < 0)
979 {
980 tm->tv_usec += 1000000;
981 --tm->tv_sec;
982 }
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000983# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +0000984}
985
986/*
Bram Moolenaar433f7c82006-03-21 21:29:36 +0000987 * Return a string that represents the time in "tm".
988 * Uses a static buffer!
989 */
990 char *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100991profile_msg(proftime_T *tm)
Bram Moolenaar433f7c82006-03-21 21:29:36 +0000992{
993 static char buf[50];
994
995# ifdef WIN3264
996 LARGE_INTEGER fr;
997
998 QueryPerformanceFrequency(&fr);
999 sprintf(buf, "%10.6lf", (double)tm->QuadPart / (double)fr.QuadPart);
1000# else
1001 sprintf(buf, "%3ld.%06ld", (long)tm->tv_sec, (long)tm->tv_usec);
Bram Moolenaar76929292008-01-06 19:07:36 +00001002# endif
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001003 return buf;
1004}
1005
Bram Moolenaar79c2c882016-02-07 21:19:28 +01001006# if defined(FEAT_FLOAT) || defined(PROTO)
1007/*
1008 * Return a float that represents the time in "tm".
1009 */
1010 float_T
1011profile_float(proftime_T *tm)
1012{
1013# ifdef WIN3264
1014 LARGE_INTEGER fr;
1015
1016 QueryPerformanceFrequency(&fr);
1017 return (float_T)tm->QuadPart / (float_T)fr.QuadPart;
1018# else
1019 return (float_T)tm->tv_sec + (float_T)tm->tv_usec / 1000000.0;
1020# endif
1021}
1022# endif
1023
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001024/*
Bram Moolenaar76929292008-01-06 19:07:36 +00001025 * Put the time "msec" past now in "tm".
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001026 */
Bram Moolenaar76929292008-01-06 19:07:36 +00001027 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001028profile_setlimit(long msec, proftime_T *tm)
Bram Moolenaar76929292008-01-06 19:07:36 +00001029{
1030 if (msec <= 0) /* no limit */
1031 profile_zero(tm);
1032 else
1033 {
1034# ifdef WIN3264
1035 LARGE_INTEGER fr;
1036
1037 QueryPerformanceCounter(tm);
1038 QueryPerformanceFrequency(&fr);
Bram Moolenaarb3c70982008-01-18 10:40:55 +00001039 tm->QuadPart += (LONGLONG)((double)msec / 1000.0 * (double)fr.QuadPart);
Bram Moolenaar76929292008-01-06 19:07:36 +00001040# else
1041 long usec;
1042
1043 gettimeofday(tm, NULL);
1044 usec = (long)tm->tv_usec + (long)msec * 1000;
1045 tm->tv_usec = usec % 1000000L;
1046 tm->tv_sec += usec / 1000000L;
1047# endif
1048 }
1049}
1050
1051/*
1052 * Return TRUE if the current time is past "tm".
1053 */
1054 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001055profile_passed_limit(proftime_T *tm)
Bram Moolenaar76929292008-01-06 19:07:36 +00001056{
1057 proftime_T now;
1058
1059# ifdef WIN3264
1060 if (tm->QuadPart == 0) /* timer was not set */
1061 return FALSE;
1062 QueryPerformanceCounter(&now);
1063 return (now.QuadPart > tm->QuadPart);
1064# else
1065 if (tm->tv_sec == 0) /* timer was not set */
1066 return FALSE;
1067 gettimeofday(&now, NULL);
1068 return (now.tv_sec > tm->tv_sec
1069 || (now.tv_sec == tm->tv_sec && now.tv_usec > tm->tv_usec));
1070# endif
1071}
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001072
1073/*
1074 * Set the time in "tm" to zero.
1075 */
1076 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001077profile_zero(proftime_T *tm)
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001078{
1079# ifdef WIN3264
1080 tm->QuadPart = 0;
1081# else
1082 tm->tv_usec = 0;
1083 tm->tv_sec = 0;
1084# endif
1085}
1086
Bram Moolenaar76929292008-01-06 19:07:36 +00001087# endif /* FEAT_PROFILE || FEAT_RELTIME */
1088
Bram Moolenaar975b5272016-03-15 23:10:59 +01001089# if defined(FEAT_TIMERS) || defined(PROTO)
1090static timer_T *first_timer = NULL;
Bram Moolenaar75537a92016-09-05 22:45:28 +02001091static long last_timer_id = 0;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001092
Bram Moolenaar75537a92016-09-05 22:45:28 +02001093# ifdef WIN3264
1094# define GET_TIMEDIFF(timer, now) \
1095 (long)(((double)(timer->tr_due.QuadPart - now.QuadPart) \
Bram Moolenaara7c023e2016-10-12 12:13:35 +02001096 / (double)fr.QuadPart) * 1000)
Bram Moolenaar75537a92016-09-05 22:45:28 +02001097# else
1098# define GET_TIMEDIFF(timer, now) \
1099 (timer->tr_due.tv_sec - now.tv_sec) * 1000 \
Bram Moolenaara7c023e2016-10-12 12:13:35 +02001100 + (timer->tr_due.tv_usec - now.tv_usec) / 1000
Bram Moolenaar75537a92016-09-05 22:45:28 +02001101# endif
Bram Moolenaar417ccd72016-09-01 21:26:20 +02001102
Bram Moolenaar975b5272016-03-15 23:10:59 +01001103/*
1104 * Insert a timer in the list of timers.
1105 */
1106 static void
1107insert_timer(timer_T *timer)
1108{
1109 timer->tr_next = first_timer;
1110 timer->tr_prev = NULL;
1111 if (first_timer != NULL)
1112 first_timer->tr_prev = timer;
1113 first_timer = timer;
Bram Moolenaar4231da42016-06-02 14:30:04 +02001114 did_add_timer = TRUE;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001115}
1116
1117/*
1118 * Take a timer out of the list of timers.
1119 */
1120 static void
1121remove_timer(timer_T *timer)
1122{
1123 if (timer->tr_prev == NULL)
1124 first_timer = timer->tr_next;
1125 else
1126 timer->tr_prev->tr_next = timer->tr_next;
1127 if (timer->tr_next != NULL)
1128 timer->tr_next->tr_prev = timer->tr_prev;
1129}
1130
1131 static void
1132free_timer(timer_T *timer)
1133{
Bram Moolenaar75537a92016-09-05 22:45:28 +02001134 free_callback(timer->tr_callback, timer->tr_partial);
1135 vim_free(timer);
Bram Moolenaar975b5272016-03-15 23:10:59 +01001136}
1137
1138/*
1139 * Create a timer and return it. NULL if out of memory.
1140 * Caller should set the callback.
1141 */
1142 timer_T *
1143create_timer(long msec, int repeat)
1144{
1145 timer_T *timer = (timer_T *)alloc_clear(sizeof(timer_T));
Bram Moolenaaree39ef02016-09-10 19:17:42 +02001146 long prev_id = last_timer_id;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001147
1148 if (timer == NULL)
1149 return NULL;
Bram Moolenaaree39ef02016-09-10 19:17:42 +02001150 if (++last_timer_id <= prev_id)
Bram Moolenaar75537a92016-09-05 22:45:28 +02001151 /* Overflow! Might cause duplicates... */
1152 last_timer_id = 0;
1153 timer->tr_id = last_timer_id;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001154 insert_timer(timer);
1155 if (repeat != 0)
Bram Moolenaar975b5272016-03-15 23:10:59 +01001156 timer->tr_repeat = repeat - 1;
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001157 timer->tr_interval = msec;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001158
1159 profile_setlimit(msec, &timer->tr_due);
1160 return timer;
1161}
1162
1163/*
1164 * Invoke the callback of "timer".
1165 */
1166 static void
1167timer_callback(timer_T *timer)
1168{
1169 typval_T rettv;
1170 int dummy;
1171 typval_T argv[2];
1172
1173 argv[0].v_type = VAR_NUMBER;
Bram Moolenaar75537a92016-09-05 22:45:28 +02001174 argv[0].vval.v_number = (varnumber_T)timer->tr_id;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001175 argv[1].v_type = VAR_UNKNOWN;
1176
1177 call_func(timer->tr_callback, (int)STRLEN(timer->tr_callback),
Bram Moolenaardf48fb42016-07-22 21:50:18 +02001178 &rettv, 1, argv, NULL, 0L, 0L, &dummy, TRUE,
Bram Moolenaar975b5272016-03-15 23:10:59 +01001179 timer->tr_partial, NULL);
1180 clear_tv(&rettv);
1181}
1182
1183/*
1184 * Call timers that are due.
1185 * Return the time in msec until the next timer is due.
Bram Moolenaarc4f83382017-07-07 14:50:44 +02001186 * Returns -1 if there are no pending timers.
Bram Moolenaar975b5272016-03-15 23:10:59 +01001187 */
1188 long
Bram Moolenaarcf089462016-06-12 21:18:43 +02001189check_due_timer(void)
Bram Moolenaar975b5272016-03-15 23:10:59 +01001190{
1191 timer_T *timer;
Bram Moolenaar75537a92016-09-05 22:45:28 +02001192 timer_T *timer_next;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001193 long this_due;
Bram Moolenaar597385a2016-03-16 23:24:43 +01001194 long next_due = -1;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001195 proftime_T now;
1196 int did_one = FALSE;
Bram Moolenaar02e177d2017-08-26 23:43:28 +02001197 int need_update_screen = FALSE;
Bram Moolenaar75537a92016-09-05 22:45:28 +02001198 long current_id = last_timer_id;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001199# ifdef WIN3264
1200 LARGE_INTEGER fr;
Bram Moolenaarc577d812017-07-08 22:37:34 +02001201# endif
Bram Moolenaar975b5272016-03-15 23:10:59 +01001202
Bram Moolenaarc577d812017-07-08 22:37:34 +02001203 /* Don't run any timers while exiting or dealing with an error. */
1204 if (exiting || aborting())
Bram Moolenaarc4f83382017-07-07 14:50:44 +02001205 return next_due;
1206
Bram Moolenaarc577d812017-07-08 22:37:34 +02001207# ifdef WIN3264
Bram Moolenaar975b5272016-03-15 23:10:59 +01001208 QueryPerformanceFrequency(&fr);
1209# endif
Bram Moolenaar75537a92016-09-05 22:45:28 +02001210 profile_start(&now);
1211 for (timer = first_timer; timer != NULL && !got_int; timer = timer_next)
Bram Moolenaar975b5272016-03-15 23:10:59 +01001212 {
Bram Moolenaar75537a92016-09-05 22:45:28 +02001213 timer_next = timer->tr_next;
Bram Moolenaar417ccd72016-09-01 21:26:20 +02001214
Bram Moolenaar75537a92016-09-05 22:45:28 +02001215 if (timer->tr_id == -1 || timer->tr_firing || timer->tr_paused)
1216 continue;
1217 this_due = GET_TIMEDIFF(timer, now);
1218 if (this_due <= 1)
1219 {
Bram Moolenaar1e8e1452017-06-24 16:03:06 +02001220 int save_timer_busy = timer_busy;
1221 int save_vgetc_busy = vgetc_busy;
Bram Moolenaarc577d812017-07-08 22:37:34 +02001222 int did_emsg_save = did_emsg;
1223 int called_emsg_save = called_emsg;
1224 int did_throw_save = did_throw;
Bram Moolenaar02e177d2017-08-26 23:43:28 +02001225 int save_must_redraw = must_redraw;
Bram Moolenaar1e8e1452017-06-24 16:03:06 +02001226
1227 timer_busy = timer_busy > 0 || vgetc_busy > 0;
1228 vgetc_busy = 0;
Bram Moolenaarc577d812017-07-08 22:37:34 +02001229 called_emsg = FALSE;
Bram Moolenaar02e177d2017-08-26 23:43:28 +02001230 must_redraw = 0;
Bram Moolenaar75537a92016-09-05 22:45:28 +02001231 timer->tr_firing = TRUE;
1232 timer_callback(timer);
1233 timer->tr_firing = FALSE;
1234 timer_next = timer->tr_next;
1235 did_one = TRUE;
Bram Moolenaar1e8e1452017-06-24 16:03:06 +02001236 timer_busy = save_timer_busy;
1237 vgetc_busy = save_vgetc_busy;
Bram Moolenaarc577d812017-07-08 22:37:34 +02001238 if (called_emsg)
1239 {
1240 ++timer->tr_emsg_count;
Bram Moolenaarcae24be2017-07-10 22:12:10 +02001241 if (!did_throw_save && did_throw && current_exception != NULL)
Bram Moolenaarc577d812017-07-08 22:37:34 +02001242 discard_current_exception();
1243 }
1244 did_emsg = did_emsg_save;
1245 called_emsg = called_emsg_save;
Bram Moolenaar02e177d2017-08-26 23:43:28 +02001246 if (must_redraw != 0)
1247 need_update_screen = TRUE;
1248 must_redraw = must_redraw > save_must_redraw
1249 ? must_redraw : save_must_redraw;
Bram Moolenaar75537a92016-09-05 22:45:28 +02001250
1251 /* Only fire the timer again if it repeats and stop_timer() wasn't
1252 * called while inside the callback (tr_id == -1). */
Bram Moolenaarc577d812017-07-08 22:37:34 +02001253 if (timer->tr_repeat != 0 && timer->tr_id != -1
1254 && timer->tr_emsg_count < 3)
Bram Moolenaar75537a92016-09-05 22:45:28 +02001255 {
1256 profile_setlimit(timer->tr_interval, &timer->tr_due);
1257 this_due = GET_TIMEDIFF(timer, now);
1258 if (this_due < 1)
1259 this_due = 1;
1260 if (timer->tr_repeat > 0)
1261 --timer->tr_repeat;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001262 }
Bram Moolenaar75537a92016-09-05 22:45:28 +02001263 else
1264 {
1265 this_due = -1;
1266 remove_timer(timer);
1267 free_timer(timer);
1268 }
Bram Moolenaar975b5272016-03-15 23:10:59 +01001269 }
Bram Moolenaar75537a92016-09-05 22:45:28 +02001270 if (this_due > 0 && (next_due == -1 || next_due > this_due))
1271 next_due = this_due;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001272 }
1273
1274 if (did_one)
Bram Moolenaar02e177d2017-08-26 23:43:28 +02001275 redraw_after_callback(need_update_screen);
Bram Moolenaar975b5272016-03-15 23:10:59 +01001276
Bram Moolenaar75537a92016-09-05 22:45:28 +02001277 return current_id != last_timer_id ? 1 : next_due;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001278}
1279
1280/*
1281 * Find a timer by ID. Returns NULL if not found;
1282 */
1283 timer_T *
Bram Moolenaar75537a92016-09-05 22:45:28 +02001284find_timer(long id)
Bram Moolenaar975b5272016-03-15 23:10:59 +01001285{
1286 timer_T *timer;
1287
Bram Moolenaar75537a92016-09-05 22:45:28 +02001288 if (id >= 0)
1289 {
1290 for (timer = first_timer; timer != NULL; timer = timer->tr_next)
1291 if (timer->tr_id == id)
1292 return timer;
1293 }
1294 return NULL;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001295}
1296
1297
1298/*
1299 * Stop a timer and delete it.
1300 */
1301 void
1302stop_timer(timer_T *timer)
1303{
Bram Moolenaar75537a92016-09-05 22:45:28 +02001304 if (timer->tr_firing)
1305 /* Free the timer after the callback returns. */
1306 timer->tr_id = -1;
1307 else
1308 {
1309 remove_timer(timer);
1310 free_timer(timer);
1311 }
Bram Moolenaar975b5272016-03-15 23:10:59 +01001312}
Bram Moolenaare3188e22016-05-31 21:13:04 +02001313
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001314 void
Bram Moolenaarb73598e2016-08-07 18:22:53 +02001315stop_all_timers(void)
1316{
Bram Moolenaar75537a92016-09-05 22:45:28 +02001317 timer_T *timer;
1318 timer_T *timer_next;
1319
1320 for (timer = first_timer; timer != NULL; timer = timer_next)
1321 {
1322 timer_next = timer->tr_next;
1323 stop_timer(timer);
1324 }
Bram Moolenaarb73598e2016-08-07 18:22:53 +02001325}
1326
1327 void
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001328add_timer_info(typval_T *rettv, timer_T *timer)
1329{
1330 list_T *list = rettv->vval.v_list;
1331 dict_T *dict = dict_alloc();
1332 dictitem_T *di;
1333 long remaining;
1334 proftime_T now;
Bram Moolenaar00ff3802016-08-06 22:27:28 +02001335# ifdef WIN3264
1336 LARGE_INTEGER fr;
1337#endif
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001338
1339 if (dict == NULL)
1340 return;
1341 list_append_dict(list, dict);
1342
Bram Moolenaar75537a92016-09-05 22:45:28 +02001343 dict_add_nr_str(dict, "id", timer->tr_id, NULL);
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001344 dict_add_nr_str(dict, "time", (long)timer->tr_interval, NULL);
1345
1346 profile_start(&now);
1347# ifdef WIN3264
Bram Moolenaar00ff3802016-08-06 22:27:28 +02001348 QueryPerformanceFrequency(&fr);
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001349# endif
Bram Moolenaar75537a92016-09-05 22:45:28 +02001350 remaining = GET_TIMEDIFF(timer, now);
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001351 dict_add_nr_str(dict, "remaining", (long)remaining, NULL);
1352
1353 dict_add_nr_str(dict, "repeat",
1354 (long)(timer->tr_repeat < 0 ? -1 : timer->tr_repeat + 1), NULL);
Bram Moolenaarb73598e2016-08-07 18:22:53 +02001355 dict_add_nr_str(dict, "paused", (long)(timer->tr_paused), NULL);
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001356
1357 di = dictitem_alloc((char_u *)"callback");
1358 if (di != NULL)
1359 {
1360 if (dict_add(dict, di) == FAIL)
1361 vim_free(di);
1362 else if (timer->tr_partial != NULL)
1363 {
1364 di->di_tv.v_type = VAR_PARTIAL;
1365 di->di_tv.vval.v_partial = timer->tr_partial;
1366 ++timer->tr_partial->pt_refcount;
1367 }
1368 else
1369 {
1370 di->di_tv.v_type = VAR_FUNC;
1371 di->di_tv.vval.v_string = vim_strsave(timer->tr_callback);
1372 }
1373 di->di_tv.v_lock = 0;
1374 }
1375}
1376
1377 void
1378add_timer_info_all(typval_T *rettv)
1379{
1380 timer_T *timer;
1381
1382 for (timer = first_timer; timer != NULL; timer = timer->tr_next)
Bram Moolenaar75537a92016-09-05 22:45:28 +02001383 if (timer->tr_id != -1)
1384 add_timer_info(rettv, timer);
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001385}
1386
Bram Moolenaare3188e22016-05-31 21:13:04 +02001387/*
1388 * Mark references in partials of timers.
1389 */
1390 int
1391set_ref_in_timer(int copyID)
1392{
1393 int abort = FALSE;
1394 timer_T *timer;
1395 typval_T tv;
1396
1397 for (timer = first_timer; timer != NULL; timer = timer->tr_next)
1398 {
Bram Moolenaar1e96d9b2016-07-29 22:15:09 +02001399 if (timer->tr_partial != NULL)
1400 {
1401 tv.v_type = VAR_PARTIAL;
1402 tv.vval.v_partial = timer->tr_partial;
1403 }
1404 else
1405 {
1406 tv.v_type = VAR_FUNC;
1407 tv.vval.v_string = timer->tr_callback;
1408 }
Bram Moolenaare3188e22016-05-31 21:13:04 +02001409 abort = abort || set_ref_in_item(&tv, copyID, NULL, NULL);
1410 }
1411 return abort;
1412}
Bram Moolenaar623e2632016-07-30 22:47:56 +02001413
1414# if defined(EXITFREE) || defined(PROTO)
1415 void
1416timer_free_all()
1417{
1418 timer_T *timer;
1419
1420 while (first_timer != NULL)
1421 {
1422 timer = first_timer;
1423 remove_timer(timer);
1424 free_timer(timer);
1425 }
1426}
1427# endif
Bram Moolenaar975b5272016-03-15 23:10:59 +01001428# endif
1429
Bram Moolenaar8a7f5a22013-06-06 14:01:46 +02001430#if defined(FEAT_SYN_HL) && defined(FEAT_RELTIME) && defined(FEAT_FLOAT)
1431# if defined(HAVE_MATH_H)
1432# include <math.h>
1433# endif
1434
1435/*
1436 * Divide the time "tm" by "count" and store in "tm2".
1437 */
1438 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001439profile_divide(proftime_T *tm, int count, proftime_T *tm2)
Bram Moolenaar8a7f5a22013-06-06 14:01:46 +02001440{
1441 if (count == 0)
1442 profile_zero(tm2);
1443 else
1444 {
1445# ifdef WIN3264
1446 tm2->QuadPart = tm->QuadPart / count;
1447# else
1448 double usec = (tm->tv_sec * 1000000.0 + tm->tv_usec) / count;
1449
1450 tm2->tv_sec = floor(usec / 1000000.0);
Bram Moolenaara2e14fc2013-06-10 20:10:44 +02001451 tm2->tv_usec = vim_round(usec - (tm2->tv_sec * 1000000.0));
Bram Moolenaar8a7f5a22013-06-06 14:01:46 +02001452# endif
1453 }
1454}
1455#endif
1456
Bram Moolenaar76929292008-01-06 19:07:36 +00001457# if defined(FEAT_PROFILE) || defined(PROTO)
1458/*
1459 * Functions for profiling.
1460 */
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01001461static void script_do_profile(scriptitem_T *si);
1462static void script_dump_profile(FILE *fd);
Bram Moolenaar76929292008-01-06 19:07:36 +00001463static proftime_T prof_wait_time;
1464
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001465/*
Bram Moolenaar05159a02005-02-26 23:04:13 +00001466 * Add the time "tm2" to "tm".
1467 */
1468 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001469profile_add(proftime_T *tm, proftime_T *tm2)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001470{
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001471# ifdef WIN3264
1472 tm->QuadPart += tm2->QuadPart;
1473# else
Bram Moolenaar05159a02005-02-26 23:04:13 +00001474 tm->tv_usec += tm2->tv_usec;
1475 tm->tv_sec += tm2->tv_sec;
1476 if (tm->tv_usec >= 1000000)
1477 {
1478 tm->tv_usec -= 1000000;
1479 ++tm->tv_sec;
1480 }
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001481# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00001482}
1483
1484/*
Bram Moolenaar1056d982006-03-09 22:37:52 +00001485 * Add the "self" time from the total time and the children's time.
1486 */
1487 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001488profile_self(proftime_T *self, proftime_T *total, proftime_T *children)
Bram Moolenaar1056d982006-03-09 22:37:52 +00001489{
1490 /* Check that the result won't be negative. Can happen with recursive
1491 * calls. */
1492#ifdef WIN3264
1493 if (total->QuadPart <= children->QuadPart)
1494 return;
1495#else
1496 if (total->tv_sec < children->tv_sec
1497 || (total->tv_sec == children->tv_sec
1498 && total->tv_usec <= children->tv_usec))
1499 return;
1500#endif
1501 profile_add(self, total);
1502 profile_sub(self, children);
1503}
1504
1505/*
Bram Moolenaar05159a02005-02-26 23:04:13 +00001506 * Get the current waittime.
1507 */
1508 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001509profile_get_wait(proftime_T *tm)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001510{
1511 *tm = prof_wait_time;
1512}
1513
1514/*
1515 * Subtract the passed waittime since "tm" from "tma".
1516 */
1517 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001518profile_sub_wait(proftime_T *tm, proftime_T *tma)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001519{
1520 proftime_T tm3 = prof_wait_time;
1521
1522 profile_sub(&tm3, tm);
1523 profile_sub(tma, &tm3);
1524}
1525
1526/*
1527 * Return TRUE if "tm1" and "tm2" are equal.
1528 */
1529 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001530profile_equal(proftime_T *tm1, proftime_T *tm2)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001531{
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001532# ifdef WIN3264
1533 return (tm1->QuadPart == tm2->QuadPart);
1534# else
Bram Moolenaar05159a02005-02-26 23:04:13 +00001535 return (tm1->tv_usec == tm2->tv_usec && tm1->tv_sec == tm2->tv_sec);
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001536# endif
1537}
1538
1539/*
1540 * Return <0, 0 or >0 if "tm1" < "tm2", "tm1" == "tm2" or "tm1" > "tm2"
1541 */
1542 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001543profile_cmp(const proftime_T *tm1, const proftime_T *tm2)
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001544{
1545# ifdef WIN3264
1546 return (int)(tm2->QuadPart - tm1->QuadPart);
1547# else
1548 if (tm1->tv_sec == tm2->tv_sec)
1549 return tm2->tv_usec - tm1->tv_usec;
1550 return tm2->tv_sec - tm1->tv_sec;
1551# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00001552}
1553
Bram Moolenaar05159a02005-02-26 23:04:13 +00001554static char_u *profile_fname = NULL;
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001555static proftime_T pause_time;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001556
1557/*
1558 * ":profile cmd args"
1559 */
1560 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001561ex_profile(exarg_T *eap)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001562{
1563 char_u *e;
1564 int len;
1565
1566 e = skiptowhite(eap->arg);
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00001567 len = (int)(e - eap->arg);
Bram Moolenaar05159a02005-02-26 23:04:13 +00001568 e = skipwhite(e);
1569
1570 if (len == 5 && STRNCMP(eap->arg, "start", 5) == 0 && *e != NUL)
1571 {
1572 vim_free(profile_fname);
Bram Moolenaard94682f2015-04-13 15:37:56 +02001573 profile_fname = expand_env_save_opt(e, TRUE);
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001574 do_profiling = PROF_YES;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001575 profile_zero(&prof_wait_time);
1576 set_vim_var_nr(VV_PROFILING, 1L);
1577 }
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001578 else if (do_profiling == PROF_NONE)
Bram Moolenaar54c1b492010-02-24 14:01:28 +01001579 EMSG(_("E750: First use \":profile start {fname}\""));
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001580 else if (STRCMP(eap->arg, "pause") == 0)
1581 {
1582 if (do_profiling == PROF_YES)
1583 profile_start(&pause_time);
1584 do_profiling = PROF_PAUSED;
1585 }
1586 else if (STRCMP(eap->arg, "continue") == 0)
1587 {
1588 if (do_profiling == PROF_PAUSED)
1589 {
1590 profile_end(&pause_time);
1591 profile_add(&prof_wait_time, &pause_time);
1592 }
1593 do_profiling = PROF_YES;
1594 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00001595 else
1596 {
1597 /* The rest is similar to ":breakadd". */
1598 ex_breakadd(eap);
1599 }
1600}
1601
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001602/* Command line expansion for :profile. */
1603static enum
1604{
1605 PEXP_SUBCMD, /* expand :profile sub-commands */
Bram Moolenaar49789dc2011-02-25 14:46:09 +01001606 PEXP_FUNC /* expand :profile func {funcname} */
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001607} pexpand_what;
1608
1609static char *pexpand_cmds[] = {
1610 "start",
1611#define PROFCMD_START 0
1612 "pause",
1613#define PROFCMD_PAUSE 1
1614 "continue",
1615#define PROFCMD_CONTINUE 2
1616 "func",
1617#define PROFCMD_FUNC 3
1618 "file",
1619#define PROFCMD_FILE 4
1620 NULL
1621#define PROFCMD_LAST 5
1622};
1623
1624/*
1625 * Function given to ExpandGeneric() to obtain the profile command
1626 * specific expansion.
1627 */
1628 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001629get_profile_name(expand_T *xp UNUSED, int idx)
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001630{
1631 switch (pexpand_what)
1632 {
1633 case PEXP_SUBCMD:
1634 return (char_u *)pexpand_cmds[idx];
1635 /* case PEXP_FUNC: TODO */
1636 default:
1637 return NULL;
1638 }
1639}
1640
1641/*
1642 * Handle command line completion for :profile command.
1643 */
1644 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001645set_context_in_profile_cmd(expand_T *xp, char_u *arg)
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001646{
1647 char_u *end_subcmd;
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001648
1649 /* Default: expand subcommands. */
1650 xp->xp_context = EXPAND_PROFILE;
1651 pexpand_what = PEXP_SUBCMD;
1652 xp->xp_pattern = arg;
1653
1654 end_subcmd = skiptowhite(arg);
1655 if (*end_subcmd == NUL)
1656 return;
1657
Bram Moolenaar8b9c05f2010-03-02 17:54:33 +01001658 if (end_subcmd - arg == 5 && STRNCMP(arg, "start", 5) == 0)
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001659 {
1660 xp->xp_context = EXPAND_FILES;
1661 xp->xp_pattern = skipwhite(end_subcmd);
1662 return;
1663 }
1664
1665 /* TODO: expand function names after "func" */
1666 xp->xp_context = EXPAND_NOTHING;
1667}
1668
Bram Moolenaar05159a02005-02-26 23:04:13 +00001669/*
1670 * Dump the profiling info.
1671 */
1672 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001673profile_dump(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001674{
1675 FILE *fd;
1676
1677 if (profile_fname != NULL)
1678 {
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +00001679 fd = mch_fopen((char *)profile_fname, "w");
Bram Moolenaar05159a02005-02-26 23:04:13 +00001680 if (fd == NULL)
1681 EMSG2(_(e_notopen), profile_fname);
1682 else
1683 {
Bram Moolenaar05159a02005-02-26 23:04:13 +00001684 script_dump_profile(fd);
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001685 func_dump_profile(fd);
Bram Moolenaar05159a02005-02-26 23:04:13 +00001686 fclose(fd);
1687 }
1688 }
1689}
1690
1691/*
1692 * Start profiling script "fp".
1693 */
1694 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001695script_do_profile(scriptitem_T *si)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001696{
1697 si->sn_pr_count = 0;
1698 profile_zero(&si->sn_pr_total);
1699 profile_zero(&si->sn_pr_self);
1700
1701 ga_init2(&si->sn_prl_ga, sizeof(sn_prl_T), 100);
1702 si->sn_prl_idx = -1;
1703 si->sn_prof_on = TRUE;
1704 si->sn_pr_nest = 0;
1705}
1706
1707/*
1708 * save time when starting to invoke another script or function.
1709 */
1710 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001711script_prof_save(
1712 proftime_T *tm) /* place to store wait time */
Bram Moolenaar05159a02005-02-26 23:04:13 +00001713{
1714 scriptitem_T *si;
1715
1716 if (current_SID > 0 && current_SID <= script_items.ga_len)
1717 {
1718 si = &SCRIPT_ITEM(current_SID);
1719 if (si->sn_prof_on && si->sn_pr_nest++ == 0)
1720 profile_start(&si->sn_pr_child);
1721 }
1722 profile_get_wait(tm);
1723}
1724
1725/*
1726 * Count time spent in children after invoking another script or function.
1727 */
1728 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001729script_prof_restore(proftime_T *tm)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001730{
1731 scriptitem_T *si;
1732
1733 if (current_SID > 0 && current_SID <= script_items.ga_len)
1734 {
1735 si = &SCRIPT_ITEM(current_SID);
1736 if (si->sn_prof_on && --si->sn_pr_nest == 0)
1737 {
1738 profile_end(&si->sn_pr_child);
1739 profile_sub_wait(tm, &si->sn_pr_child); /* don't count wait time */
1740 profile_add(&si->sn_pr_children, &si->sn_pr_child);
1741 profile_add(&si->sn_prl_children, &si->sn_pr_child);
1742 }
1743 }
1744}
1745
1746static proftime_T inchar_time;
1747
1748/*
1749 * Called when starting to wait for the user to type a character.
1750 */
1751 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001752prof_inchar_enter(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001753{
1754 profile_start(&inchar_time);
1755}
1756
1757/*
1758 * Called when finished waiting for the user to type a character.
1759 */
1760 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001761prof_inchar_exit(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001762{
1763 profile_end(&inchar_time);
1764 profile_add(&prof_wait_time, &inchar_time);
1765}
1766
1767/*
1768 * Dump the profiling results for all scripts in file "fd".
1769 */
1770 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001771script_dump_profile(FILE *fd)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001772{
1773 int id;
1774 scriptitem_T *si;
1775 int i;
1776 FILE *sfd;
1777 sn_prl_T *pp;
1778
1779 for (id = 1; id <= script_items.ga_len; ++id)
1780 {
1781 si = &SCRIPT_ITEM(id);
1782 if (si->sn_prof_on)
1783 {
1784 fprintf(fd, "SCRIPT %s\n", si->sn_name);
1785 if (si->sn_pr_count == 1)
1786 fprintf(fd, "Sourced 1 time\n");
1787 else
1788 fprintf(fd, "Sourced %d times\n", si->sn_pr_count);
1789 fprintf(fd, "Total time: %s\n", profile_msg(&si->sn_pr_total));
1790 fprintf(fd, " Self time: %s\n", profile_msg(&si->sn_pr_self));
1791 fprintf(fd, "\n");
1792 fprintf(fd, "count total (s) self (s)\n");
1793
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +00001794 sfd = mch_fopen((char *)si->sn_name, "r");
Bram Moolenaar05159a02005-02-26 23:04:13 +00001795 if (sfd == NULL)
1796 fprintf(fd, "Cannot open file!\n");
1797 else
1798 {
1799 for (i = 0; i < si->sn_prl_ga.ga_len; ++i)
1800 {
1801 if (vim_fgets(IObuff, IOSIZE, sfd))
1802 break;
1803 pp = &PRL_ITEM(si, i);
1804 if (pp->snp_count > 0)
1805 {
1806 fprintf(fd, "%5d ", pp->snp_count);
1807 if (profile_equal(&pp->sn_prl_total, &pp->sn_prl_self))
1808 fprintf(fd, " ");
1809 else
1810 fprintf(fd, "%s ", profile_msg(&pp->sn_prl_total));
1811 fprintf(fd, "%s ", profile_msg(&pp->sn_prl_self));
1812 }
1813 else
1814 fprintf(fd, " ");
1815 fprintf(fd, "%s", IObuff);
1816 }
1817 fclose(sfd);
1818 }
1819 fprintf(fd, "\n");
1820 }
1821 }
1822}
1823
1824/*
1825 * Return TRUE when a function defined in the current script should be
1826 * profiled.
1827 */
1828 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001829prof_def_func(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001830{
Bram Moolenaar53180ce2005-07-05 21:48:14 +00001831 if (current_SID > 0)
1832 return SCRIPT_ITEM(current_SID).sn_pr_force;
1833 return FALSE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001834}
1835
1836# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001837#endif
1838
1839/*
1840 * If 'autowrite' option set, try to write the file.
1841 * Careful: autocommands may make "buf" invalid!
1842 *
1843 * return FAIL for failure, OK otherwise
1844 */
1845 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001846autowrite(buf_T *buf, int forceit)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001847{
Bram Moolenaar373154b2007-02-13 05:19:30 +00001848 int r;
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001849 bufref_T bufref;
Bram Moolenaar373154b2007-02-13 05:19:30 +00001850
Bram Moolenaar071d4272004-06-13 20:20:40 +00001851 if (!(p_aw || p_awa) || !p_write
1852#ifdef FEAT_QUICKFIX
Bram Moolenaar373154b2007-02-13 05:19:30 +00001853 /* never autowrite a "nofile" or "nowrite" buffer */
1854 || bt_dontwrite(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001855#endif
Bram Moolenaar373154b2007-02-13 05:19:30 +00001856 || (!forceit && buf->b_p_ro) || buf->b_ffname == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001857 return FAIL;
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001858 set_bufref(&bufref, buf);
Bram Moolenaar373154b2007-02-13 05:19:30 +00001859 r = buf_write_all(buf, forceit);
1860
1861 /* Writing may succeed but the buffer still changed, e.g., when there is a
1862 * conversion error. We do want to return FAIL then. */
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001863 if (bufref_valid(&bufref) && bufIsChanged(buf))
Bram Moolenaar373154b2007-02-13 05:19:30 +00001864 r = FAIL;
1865 return r;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001866}
1867
1868/*
1869 * flush all buffers, except the ones that are readonly
1870 */
1871 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001872autowrite_all(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001873{
1874 buf_T *buf;
1875
1876 if (!(p_aw || p_awa) || !p_write)
1877 return;
Bram Moolenaar29323592016-07-24 22:04:11 +02001878 FOR_ALL_BUFFERS(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001879 if (bufIsChanged(buf) && !buf->b_p_ro)
1880 {
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001881#ifdef FEAT_AUTOCMD
1882 bufref_T bufref;
1883
1884 set_bufref(&bufref, buf);
1885#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001886 (void)buf_write_all(buf, FALSE);
1887#ifdef FEAT_AUTOCMD
1888 /* an autocommand may have deleted the buffer */
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001889 if (!bufref_valid(&bufref))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001890 buf = firstbuf;
1891#endif
1892 }
1893}
1894
1895/*
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001896 * Return TRUE if buffer was changed and cannot be abandoned.
1897 * For flags use the CCGD_ values.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001898 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001899 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001900check_changed(buf_T *buf, int flags)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001901{
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001902 int forceit = (flags & CCGD_FORCEIT);
1903#ifdef FEAT_AUTOCMD
1904 bufref_T bufref;
1905
1906 set_bufref(&bufref, buf);
1907#endif
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001908
Bram Moolenaar071d4272004-06-13 20:20:40 +00001909 if ( !forceit
1910 && bufIsChanged(buf)
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001911 && ((flags & CCGD_MULTWIN) || buf->b_nwindows <= 1)
1912 && (!(flags & CCGD_AW) || autowrite(buf, forceit) == FAIL))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001913 {
1914#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG)
1915 if ((p_confirm || cmdmod.confirm) && p_write)
1916 {
1917 buf_T *buf2;
1918 int count = 0;
1919
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001920 if (flags & CCGD_ALLBUF)
Bram Moolenaar29323592016-07-24 22:04:11 +02001921 FOR_ALL_BUFFERS(buf2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001922 if (bufIsChanged(buf2)
1923 && (buf2->b_ffname != NULL
1924# ifdef FEAT_BROWSE
1925 || cmdmod.browse
1926# endif
1927 ))
1928 ++count;
1929# ifdef FEAT_AUTOCMD
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001930 if (!bufref_valid(&bufref))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001931 /* Autocommand deleted buffer, oops! It's not changed now. */
1932 return FALSE;
1933# endif
1934 dialog_changed(buf, count > 1);
1935# ifdef FEAT_AUTOCMD
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001936 if (!bufref_valid(&bufref))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001937 /* Autocommand deleted buffer, oops! It's not changed now. */
1938 return FALSE;
1939# endif
1940 return bufIsChanged(buf);
1941 }
1942#endif
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001943 if (flags & CCGD_EXCMD)
Bram Moolenaarf5be7cd2017-08-17 16:55:13 +02001944 no_write_message();
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001945 else
Bram Moolenaarf5be7cd2017-08-17 16:55:13 +02001946 no_write_message_nobang();
Bram Moolenaar071d4272004-06-13 20:20:40 +00001947 return TRUE;
1948 }
1949 return FALSE;
1950}
1951
1952#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG) || defined(PROTO)
1953
1954#if defined(FEAT_BROWSE) || defined(PROTO)
1955/*
1956 * When wanting to write a file without a file name, ask the user for a name.
1957 */
1958 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001959browse_save_fname(buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001960{
1961 if (buf->b_fname == NULL)
1962 {
1963 char_u *fname;
1964
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00001965 fname = do_browse(BROWSE_SAVE, (char_u *)_("Save As"),
1966 NULL, NULL, NULL, NULL, buf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001967 if (fname != NULL)
1968 {
1969 if (setfname(buf, fname, NULL, TRUE) == OK)
1970 buf->b_flags |= BF_NOTEDITED;
1971 vim_free(fname);
1972 }
1973 }
1974}
1975#endif
1976
1977/*
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02001978 * Ask the user what to do when abandoning a changed buffer.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001979 * Must check 'write' option first!
1980 */
1981 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001982dialog_changed(
1983 buf_T *buf,
1984 int checkall) /* may abandon all changed buffers */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001985{
Bram Moolenaard9462e32011-04-11 21:35:11 +02001986 char_u buff[DIALOG_MSG_SIZE];
Bram Moolenaar071d4272004-06-13 20:20:40 +00001987 int ret;
1988 buf_T *buf2;
Bram Moolenaar8218f602012-04-25 17:32:18 +02001989 exarg_T ea;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001990
Bram Moolenaar3f9a1ff2017-08-21 22:06:02 +02001991 dialog_msg(buff, _("Save changes to \"%s\"?"), buf->b_fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001992 if (checkall)
1993 ret = vim_dialog_yesnoallcancel(VIM_QUESTION, NULL, buff, 1);
1994 else
1995 ret = vim_dialog_yesnocancel(VIM_QUESTION, NULL, buff, 1);
1996
Bram Moolenaar8218f602012-04-25 17:32:18 +02001997 /* Init ea pseudo-structure, this is needed for the check_overwrite()
1998 * function. */
1999 ea.append = ea.forceit = FALSE;
2000
Bram Moolenaar071d4272004-06-13 20:20:40 +00002001 if (ret == VIM_YES)
2002 {
2003#ifdef FEAT_BROWSE
2004 /* May get file name, when there is none */
2005 browse_save_fname(buf);
2006#endif
Bram Moolenaar8218f602012-04-25 17:32:18 +02002007 if (buf->b_fname != NULL && check_overwrite(&ea, buf,
2008 buf->b_fname, buf->b_ffname, FALSE) == OK)
2009 /* didn't hit Cancel */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002010 (void)buf_write_all(buf, FALSE);
2011 }
2012 else if (ret == VIM_NO)
2013 {
2014 unchanged(buf, TRUE);
2015 }
2016 else if (ret == VIM_ALL)
2017 {
2018 /*
2019 * Write all modified files that can be written.
2020 * Skip readonly buffers, these need to be confirmed
2021 * individually.
2022 */
Bram Moolenaar29323592016-07-24 22:04:11 +02002023 FOR_ALL_BUFFERS(buf2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002024 {
2025 if (bufIsChanged(buf2)
2026 && (buf2->b_ffname != NULL
2027#ifdef FEAT_BROWSE
2028 || cmdmod.browse
2029#endif
2030 )
2031 && !buf2->b_p_ro)
2032 {
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002033#ifdef FEAT_AUTOCMD
2034 bufref_T bufref;
2035
2036 set_bufref(&bufref, buf2);
2037#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002038#ifdef FEAT_BROWSE
2039 /* May get file name, when there is none */
2040 browse_save_fname(buf2);
2041#endif
Bram Moolenaar8218f602012-04-25 17:32:18 +02002042 if (buf2->b_fname != NULL && check_overwrite(&ea, buf2,
2043 buf2->b_fname, buf2->b_ffname, FALSE) == OK)
2044 /* didn't hit Cancel */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002045 (void)buf_write_all(buf2, FALSE);
2046#ifdef FEAT_AUTOCMD
2047 /* an autocommand may have deleted the buffer */
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002048 if (!bufref_valid(&bufref))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002049 buf2 = firstbuf;
2050#endif
2051 }
2052 }
2053 }
2054 else if (ret == VIM_DISCARDALL)
2055 {
2056 /*
2057 * mark all buffers as unchanged
2058 */
Bram Moolenaar29323592016-07-24 22:04:11 +02002059 FOR_ALL_BUFFERS(buf2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002060 unchanged(buf2, TRUE);
2061 }
2062}
2063#endif
2064
2065/*
2066 * Return TRUE if the buffer "buf" can be abandoned, either by making it
2067 * hidden, autowriting it or unloading it.
2068 */
2069 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002070can_abandon(buf_T *buf, int forceit)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002071{
Bram Moolenaareb44a682017-08-03 22:44:55 +02002072 return ( buf_hide(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002073 || !bufIsChanged(buf)
2074 || buf->b_nwindows > 1
2075 || autowrite(buf, forceit) == OK
2076 || forceit);
2077}
2078
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01002079static void add_bufnum(int *bufnrs, int *bufnump, int nr);
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002080
2081/*
2082 * Add a buffer number to "bufnrs", unless it's already there.
2083 */
2084 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002085add_bufnum(int *bufnrs, int *bufnump, int nr)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002086{
2087 int i;
2088
2089 for (i = 0; i < *bufnump; ++i)
2090 if (bufnrs[i] == nr)
2091 return;
2092 bufnrs[*bufnump] = nr;
2093 *bufnump = *bufnump + 1;
2094}
2095
Bram Moolenaar071d4272004-06-13 20:20:40 +00002096/*
2097 * Return TRUE if any buffer was changed and cannot be abandoned.
2098 * That changed buffer becomes the current buffer.
Bram Moolenaar027387f2016-01-02 22:25:52 +01002099 * When "unload" is true the current buffer is unloaded instead of making it
2100 * hidden. This is used for ":q!".
Bram Moolenaar071d4272004-06-13 20:20:40 +00002101 */
2102 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002103check_changed_any(
2104 int hidden, /* Only check hidden buffers */
2105 int unload)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002106{
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002107 int ret = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002108 buf_T *buf;
2109 int save;
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002110 int i;
2111 int bufnum = 0;
2112 int bufcount = 0;
2113 int *bufnrs;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002114#ifdef FEAT_WINDOWS
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002115 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002116 win_T *wp;
2117#endif
2118
Bram Moolenaar29323592016-07-24 22:04:11 +02002119 FOR_ALL_BUFFERS(buf)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002120 ++bufcount;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002121
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002122 if (bufcount == 0)
2123 return FALSE;
2124
2125 bufnrs = (int *)alloc(sizeof(int) * bufcount);
2126 if (bufnrs == NULL)
2127 return FALSE;
2128
2129 /* curbuf */
2130 bufnrs[bufnum++] = curbuf->b_fnum;
2131#ifdef FEAT_WINDOWS
2132 /* buf in curtab */
2133 FOR_ALL_WINDOWS(wp)
2134 if (wp->w_buffer != curbuf)
2135 add_bufnum(bufnrs, &bufnum, wp->w_buffer->b_fnum);
2136
2137 /* buf in other tab */
Bram Moolenaar29323592016-07-24 22:04:11 +02002138 FOR_ALL_TABPAGES(tp)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002139 if (tp != curtab)
2140 for (wp = tp->tp_firstwin; wp != NULL; wp = wp->w_next)
2141 add_bufnum(bufnrs, &bufnum, wp->w_buffer->b_fnum);
2142#endif
2143 /* any other buf */
Bram Moolenaar29323592016-07-24 22:04:11 +02002144 FOR_ALL_BUFFERS(buf)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002145 add_bufnum(bufnrs, &bufnum, buf->b_fnum);
2146
2147 for (i = 0; i < bufnum; ++i)
2148 {
2149 buf = buflist_findnr(bufnrs[i]);
2150 if (buf == NULL)
2151 continue;
2152 if ((!hidden || buf->b_nwindows == 0) && bufIsChanged(buf))
2153 {
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002154 bufref_T bufref;
2155
2156 set_bufref(&bufref, buf);
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002157 /* Try auto-writing the buffer. If this fails but the buffer no
2158 * longer exists it's not changed, that's OK. */
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002159 if (check_changed(buf, (p_awa ? CCGD_AW : 0)
2160 | CCGD_MULTWIN
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002161 | CCGD_ALLBUF) && bufref_valid(&bufref))
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002162 break; /* didn't save - still changes */
2163 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002164 }
2165
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002166 if (i >= bufnum)
2167 goto theend;
2168
2169 ret = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002170 exiting = FALSE;
2171#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG)
2172 /*
2173 * When ":confirm" used, don't give an error message.
2174 */
2175 if (!(p_confirm || cmdmod.confirm))
2176#endif
2177 {
2178 /* There must be a wait_return for this message, do_buffer()
2179 * may cause a redraw. But wait_return() is a no-op when vgetc()
2180 * is busy (Quit used from window menu), then make sure we don't
2181 * cause a scroll up. */
Bram Moolenaar61660ea2006-04-07 21:40:07 +00002182 if (vgetc_busy > 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002183 {
2184 msg_row = cmdline_row;
2185 msg_col = 0;
2186 msg_didout = FALSE;
2187 }
Bram Moolenaareb44a682017-08-03 22:44:55 +02002188 if (
2189#ifdef FEAT_TERMINAL
2190 term_job_running(buf->b_term)
2191 ? EMSG2(_("E947: Job still running in buffer \"%s\""),
2192 buf->b_fname)
2193 :
2194#endif
2195 EMSG2(_("E162: No write since last change for buffer \"%s\""),
Bram Moolenaare1704ba2012-10-03 18:25:00 +02002196 buf_spname(buf) != NULL ? buf_spname(buf) : buf->b_fname))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002197 {
2198 save = no_wait_return;
2199 no_wait_return = FALSE;
2200 wait_return(FALSE);
2201 no_wait_return = save;
2202 }
2203 }
2204
2205#ifdef FEAT_WINDOWS
2206 /* Try to find a window that contains the buffer. */
2207 if (buf != curbuf)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002208 FOR_ALL_TAB_WINDOWS(tp, wp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002209 if (wp->w_buffer == buf)
2210 {
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002211# ifdef FEAT_AUTOCMD
2212 bufref_T bufref;
2213
2214 set_bufref(&bufref, buf);
2215# endif
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002216 goto_tabpage_win(tp, wp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002217# ifdef FEAT_AUTOCMD
2218 /* Paranoia: did autocms wipe out the buffer with changes? */
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002219 if (!bufref_valid(&bufref))
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002220 {
2221 goto theend;
2222 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002223# endif
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002224 goto buf_found;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002225 }
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002226buf_found:
Bram Moolenaar071d4272004-06-13 20:20:40 +00002227#endif
2228
2229 /* Open the changed buffer in the current window. */
2230 if (buf != curbuf)
Bram Moolenaar027387f2016-01-02 22:25:52 +01002231 set_curbuf(buf, unload ? DOBUF_UNLOAD : DOBUF_GOTO);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002232
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002233theend:
2234 vim_free(bufnrs);
2235 return ret;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002236}
2237
2238/*
2239 * return FAIL if there is no file name, OK if there is one
2240 * give error message for FAIL
2241 */
2242 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002243check_fname(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002244{
2245 if (curbuf->b_ffname == NULL)
2246 {
2247 EMSG(_(e_noname));
2248 return FAIL;
2249 }
2250 return OK;
2251}
2252
2253/*
2254 * flush the contents of a buffer, unless it has no file name
2255 *
2256 * return FAIL for failure, OK otherwise
2257 */
2258 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002259buf_write_all(buf_T *buf, int forceit)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002260{
2261 int retval;
2262#ifdef FEAT_AUTOCMD
2263 buf_T *old_curbuf = curbuf;
2264#endif
2265
2266 retval = (buf_write(buf, buf->b_ffname, buf->b_fname,
2267 (linenr_T)1, buf->b_ml.ml_line_count, NULL,
2268 FALSE, forceit, TRUE, FALSE));
2269#ifdef FEAT_AUTOCMD
2270 if (curbuf != old_curbuf)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00002271 {
Bram Moolenaar8820b482017-03-16 17:23:31 +01002272 msg_source(HL_ATTR(HLF_W));
Bram Moolenaar071d4272004-06-13 20:20:40 +00002273 MSG(_("Warning: Entered other buffer unexpectedly (check autocommands)"));
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00002274 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002275#endif
2276 return retval;
2277}
2278
2279/*
2280 * Code to handle the argument list.
2281 */
2282
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01002283static char_u *do_one_arg(char_u *str);
2284static int do_arglist(char_u *str, int what, int after);
2285static void alist_check_arg_idx(void);
2286static int editing_arg_idx(win_T *win);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002287#ifdef FEAT_LISTCMDS
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01002288static int alist_add_list(int count, char_u **files, int after);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002289#endif
2290#define AL_SET 1
2291#define AL_ADD 2
2292#define AL_DEL 3
2293
Bram Moolenaar071d4272004-06-13 20:20:40 +00002294/*
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002295 * Isolate one argument, taking backticks.
2296 * Changes the argument in-place, puts a NUL after it. Backticks remain.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002297 * Return a pointer to the start of the next argument.
2298 */
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002299 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002300do_one_arg(char_u *str)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002301{
2302 char_u *p;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002303 int inbacktick;
2304
Bram Moolenaar071d4272004-06-13 20:20:40 +00002305 inbacktick = FALSE;
2306 for (p = str; *str; ++str)
2307 {
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002308 /* When the backslash is used for escaping the special meaning of a
2309 * character we need to keep it until wildcard expansion. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002310 if (rem_backslash(str))
2311 {
2312 *p++ = *str++;
2313 *p++ = *str;
2314 }
2315 else
2316 {
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002317 /* An item ends at a space not in backticks */
2318 if (!inbacktick && vim_isspace(*str))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002319 break;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002320 if (*str == '`')
Bram Moolenaar071d4272004-06-13 20:20:40 +00002321 inbacktick ^= TRUE;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002322 *p++ = *str;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002323 }
2324 }
2325 str = skipwhite(str);
2326 *p = NUL;
2327
2328 return str;
2329}
2330
Bram Moolenaar86b68352004-12-27 21:59:20 +00002331/*
2332 * Separate the arguments in "str" and return a list of pointers in the
2333 * growarray "gap".
2334 */
Bram Moolenaar398ee732017-08-03 14:29:14 +02002335 static int
2336get_arglist(garray_T *gap, char_u *str, int escaped)
Bram Moolenaar86b68352004-12-27 21:59:20 +00002337{
2338 ga_init2(gap, (int)sizeof(char_u *), 20);
2339 while (*str != NUL)
2340 {
2341 if (ga_grow(gap, 1) == FAIL)
2342 {
2343 ga_clear(gap);
2344 return FAIL;
2345 }
2346 ((char_u **)gap->ga_data)[gap->ga_len++] = str;
2347
Bram Moolenaar398ee732017-08-03 14:29:14 +02002348 /* If str is escaped, don't handle backslashes or spaces */
2349 if (!escaped)
2350 return OK;
2351
Bram Moolenaar86b68352004-12-27 21:59:20 +00002352 /* Isolate one argument, change it in-place, put a NUL after it. */
2353 str = do_one_arg(str);
2354 }
2355 return OK;
2356}
2357
Bram Moolenaar7df351e2006-01-23 22:30:28 +00002358#if defined(FEAT_QUICKFIX) || defined(FEAT_SYN_HL) || defined(PROTO)
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002359/*
2360 * Parse a list of arguments (file names), expand them and return in
Bram Moolenaar8f5c6f02012-06-29 12:57:06 +02002361 * "fnames[fcountp]". When "wig" is TRUE, removes files matching 'wildignore'.
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002362 * Return FAIL or OK.
2363 */
2364 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002365get_arglist_exp(
2366 char_u *str,
2367 int *fcountp,
2368 char_u ***fnamesp,
2369 int wig)
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002370{
2371 garray_T ga;
2372 int i;
2373
Bram Moolenaar398ee732017-08-03 14:29:14 +02002374 if (get_arglist(&ga, str, TRUE) == FAIL)
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002375 return FAIL;
Bram Moolenaar8f5c6f02012-06-29 12:57:06 +02002376 if (wig == TRUE)
2377 i = expand_wildcards(ga.ga_len, (char_u **)ga.ga_data,
2378 fcountp, fnamesp, EW_FILE|EW_NOTFOUND);
2379 else
2380 i = gen_expand_wildcards(ga.ga_len, (char_u **)ga.ga_data,
2381 fcountp, fnamesp, EW_FILE|EW_NOTFOUND);
2382
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002383 ga_clear(&ga);
2384 return i;
2385}
2386#endif
2387
Bram Moolenaar071d4272004-06-13 20:20:40 +00002388#if defined(FEAT_GUI) || defined(FEAT_CLIENTSERVER) || defined(PROTO)
2389/*
2390 * Redefine the argument list.
2391 */
2392 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002393set_arglist(char_u *str)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002394{
2395 do_arglist(str, AL_SET, 0);
2396}
2397#endif
2398
2399/*
2400 * "what" == AL_SET: Redefine the argument list to 'str'.
2401 * "what" == AL_ADD: add files in 'str' to the argument list after "after".
2402 * "what" == AL_DEL: remove files in 'str' from the argument list.
2403 *
2404 * Return FAIL for failure, OK otherwise.
2405 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002406 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002407do_arglist(
2408 char_u *str,
Bram Moolenaarf1d25012016-03-03 12:22:53 +01002409 int what,
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002410 int after UNUSED) /* 0 means before first one */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002411{
2412 garray_T new_ga;
2413 int exp_count;
2414 char_u **exp_files;
2415 int i;
2416#ifdef FEAT_LISTCMDS
2417 char_u *p;
2418 int match;
2419#endif
Bram Moolenaar398ee732017-08-03 14:29:14 +02002420 int arg_escaped = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002421
2422 /*
Bram Moolenaar2faa29f2016-01-23 23:02:34 +01002423 * Set default argument for ":argadd" command.
2424 */
2425 if (what == AL_ADD && *str == NUL)
2426 {
2427 if (curbuf->b_ffname == NULL)
2428 return FAIL;
2429 str = curbuf->b_fname;
Bram Moolenaar398ee732017-08-03 14:29:14 +02002430 arg_escaped = FALSE;
Bram Moolenaar2faa29f2016-01-23 23:02:34 +01002431 }
2432
2433 /*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002434 * Collect all file name arguments in "new_ga".
2435 */
Bram Moolenaar398ee732017-08-03 14:29:14 +02002436 if (get_arglist(&new_ga, str, arg_escaped) == FAIL)
Bram Moolenaar86b68352004-12-27 21:59:20 +00002437 return FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002438
2439#ifdef FEAT_LISTCMDS
2440 if (what == AL_DEL)
2441 {
2442 regmatch_T regmatch;
2443 int didone;
2444
2445 /*
2446 * Delete the items: use each item as a regexp and find a match in the
2447 * argument list.
2448 */
Bram Moolenaar71afbfe2013-03-19 16:49:16 +01002449 regmatch.rm_ic = p_fic; /* ignore case when 'fileignorecase' is set */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002450 for (i = 0; i < new_ga.ga_len && !got_int; ++i)
2451 {
2452 p = ((char_u **)new_ga.ga_data)[i];
2453 p = file_pat_to_reg_pat(p, NULL, NULL, FALSE);
2454 if (p == NULL)
2455 break;
2456 regmatch.regprog = vim_regcomp(p, p_magic ? RE_MAGIC : 0);
2457 if (regmatch.regprog == NULL)
2458 {
2459 vim_free(p);
2460 break;
2461 }
2462
2463 didone = FALSE;
2464 for (match = 0; match < ARGCOUNT; ++match)
2465 if (vim_regexec(&regmatch, alist_name(&ARGLIST[match]),
2466 (colnr_T)0))
2467 {
2468 didone = TRUE;
2469 vim_free(ARGLIST[match].ae_fname);
2470 mch_memmove(ARGLIST + match, ARGLIST + match + 1,
2471 (ARGCOUNT - match - 1) * sizeof(aentry_T));
2472 --ALIST(curwin)->al_ga.ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002473 if (curwin->w_arg_idx > match)
2474 --curwin->w_arg_idx;
2475 --match;
2476 }
2477
Bram Moolenaar473de612013-06-08 18:19:48 +02002478 vim_regfree(regmatch.regprog);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002479 vim_free(p);
2480 if (!didone)
2481 EMSG2(_(e_nomatch2), ((char_u **)new_ga.ga_data)[i]);
2482 }
2483 ga_clear(&new_ga);
2484 }
2485 else
2486#endif
2487 {
2488 i = expand_wildcards(new_ga.ga_len, (char_u **)new_ga.ga_data,
2489 &exp_count, &exp_files, EW_DIR|EW_FILE|EW_ADDSLASH|EW_NOTFOUND);
2490 ga_clear(&new_ga);
Bram Moolenaar2db5c3b2016-01-16 22:49:34 +01002491 if (i == FAIL || exp_count == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002492 {
2493 EMSG(_(e_nomatch));
2494 return FAIL;
2495 }
2496
2497#ifdef FEAT_LISTCMDS
2498 if (what == AL_ADD)
2499 {
2500 (void)alist_add_list(exp_count, exp_files, after);
2501 vim_free(exp_files);
2502 }
2503 else /* what == AL_SET */
2504#endif
Bram Moolenaar86b68352004-12-27 21:59:20 +00002505 alist_set(ALIST(curwin), exp_count, exp_files, FALSE, NULL, 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002506 }
2507
2508 alist_check_arg_idx();
2509
2510 return OK;
2511}
2512
2513/*
2514 * Check the validity of the arg_idx for each other window.
2515 */
2516 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002517alist_check_arg_idx(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002518{
2519#ifdef FEAT_WINDOWS
2520 win_T *win;
Bram Moolenaarf740b292006-02-16 22:11:02 +00002521 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002522
Bram Moolenaarf740b292006-02-16 22:11:02 +00002523 FOR_ALL_TAB_WINDOWS(tp, win)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002524 if (win->w_alist == curwin->w_alist)
2525 check_arg_idx(win);
2526#else
2527 check_arg_idx(curwin);
2528#endif
2529}
2530
2531/*
Bram Moolenaarb8ff1fb2012-02-04 21:59:01 +01002532 * Return TRUE if window "win" is editing the file at the current argument
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002533 * index.
2534 */
2535 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002536editing_arg_idx(win_T *win)
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002537{
2538 return !(win->w_arg_idx >= WARGCOUNT(win)
2539 || (win->w_buffer->b_fnum
2540 != WARGLIST(win)[win->w_arg_idx].ae_fnum
2541 && (win->w_buffer->b_ffname == NULL
2542 || !(fullpathcmp(
2543 alist_name(&WARGLIST(win)[win->w_arg_idx]),
2544 win->w_buffer->b_ffname, TRUE) & FPC_SAME))));
2545}
2546
2547/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002548 * Check if window "win" is editing the w_arg_idx file in its argument list.
2549 */
2550 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002551check_arg_idx(win_T *win)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002552{
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002553 if (WARGCOUNT(win) > 1 && !editing_arg_idx(win))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002554 {
2555 /* We are not editing the current entry in the argument list.
2556 * Set "arg_had_last" if we are editing the last one. */
2557 win->w_arg_idx_invalid = TRUE;
2558 if (win->w_arg_idx != WARGCOUNT(win) - 1
2559 && arg_had_last == FALSE
2560#ifdef FEAT_WINDOWS
2561 && ALIST(win) == &global_alist
2562#endif
2563 && GARGCOUNT > 0
2564 && win->w_arg_idx < GARGCOUNT
2565 && (win->w_buffer->b_fnum == GARGLIST[GARGCOUNT - 1].ae_fnum
2566 || (win->w_buffer->b_ffname != NULL
2567 && (fullpathcmp(alist_name(&GARGLIST[GARGCOUNT - 1]),
2568 win->w_buffer->b_ffname, TRUE) & FPC_SAME))))
2569 arg_had_last = TRUE;
2570 }
2571 else
2572 {
2573 /* We are editing the current entry in the argument list.
2574 * Set "arg_had_last" if it's also the last one */
2575 win->w_arg_idx_invalid = FALSE;
2576 if (win->w_arg_idx == WARGCOUNT(win) - 1
2577#ifdef FEAT_WINDOWS
2578 && win->w_alist == &global_alist
2579#endif
2580 )
2581 arg_had_last = TRUE;
2582 }
2583}
2584
2585/*
2586 * ":args", ":argslocal" and ":argsglobal".
2587 */
2588 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002589ex_args(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002590{
2591 int i;
2592
2593 if (eap->cmdidx != CMD_args)
2594 {
2595#if defined(FEAT_WINDOWS) && defined(FEAT_LISTCMDS)
2596 alist_unlink(ALIST(curwin));
2597 if (eap->cmdidx == CMD_argglobal)
2598 ALIST(curwin) = &global_alist;
2599 else /* eap->cmdidx == CMD_arglocal */
2600 alist_new();
2601#else
2602 ex_ni(eap);
2603 return;
2604#endif
2605 }
2606
2607 if (!ends_excmd(*eap->arg))
2608 {
2609 /*
2610 * ":args file ..": define new argument list, handle like ":next"
2611 * Also for ":argslocal file .." and ":argsglobal file ..".
2612 */
2613 ex_next(eap);
2614 }
2615 else
2616#if defined(FEAT_WINDOWS) && defined(FEAT_LISTCMDS)
2617 if (eap->cmdidx == CMD_args)
2618#endif
2619 {
2620 /*
2621 * ":args": list arguments.
2622 */
2623 if (ARGCOUNT > 0)
2624 {
2625 /* Overwrite the command, for a short list there is no scrolling
2626 * required and no wait_return(). */
2627 gotocmdline(TRUE);
2628 for (i = 0; i < ARGCOUNT; ++i)
2629 {
2630 if (i == curwin->w_arg_idx)
2631 msg_putchar('[');
2632 msg_outtrans(alist_name(&ARGLIST[i]));
2633 if (i == curwin->w_arg_idx)
2634 msg_putchar(']');
2635 msg_putchar(' ');
2636 }
2637 }
2638 }
2639#if defined(FEAT_WINDOWS) && defined(FEAT_LISTCMDS)
2640 else if (eap->cmdidx == CMD_arglocal)
2641 {
2642 garray_T *gap = &curwin->w_alist->al_ga;
2643
2644 /*
2645 * ":argslocal": make a local copy of the global argument list.
2646 */
2647 if (ga_grow(gap, GARGCOUNT) == OK)
2648 for (i = 0; i < GARGCOUNT; ++i)
2649 if (GARGLIST[i].ae_fname != NULL)
2650 {
2651 AARGLIST(curwin->w_alist)[gap->ga_len].ae_fname =
2652 vim_strsave(GARGLIST[i].ae_fname);
2653 AARGLIST(curwin->w_alist)[gap->ga_len].ae_fnum =
2654 GARGLIST[i].ae_fnum;
2655 ++gap->ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002656 }
2657 }
2658#endif
2659}
2660
2661/*
2662 * ":previous", ":sprevious", ":Next" and ":sNext".
2663 */
2664 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002665ex_previous(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002666{
2667 /* If past the last one already, go to the last one. */
2668 if (curwin->w_arg_idx - (int)eap->line2 >= ARGCOUNT)
2669 do_argfile(eap, ARGCOUNT - 1);
2670 else
2671 do_argfile(eap, curwin->w_arg_idx - (int)eap->line2);
2672}
2673
2674/*
2675 * ":rewind", ":first", ":sfirst" and ":srewind".
2676 */
2677 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002678ex_rewind(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002679{
2680 do_argfile(eap, 0);
2681}
2682
2683/*
2684 * ":last" and ":slast".
2685 */
2686 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002687ex_last(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002688{
2689 do_argfile(eap, ARGCOUNT - 1);
2690}
2691
2692/*
2693 * ":argument" and ":sargument".
2694 */
2695 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002696ex_argument(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002697{
2698 int i;
2699
2700 if (eap->addr_count > 0)
2701 i = eap->line2 - 1;
2702 else
2703 i = curwin->w_arg_idx;
2704 do_argfile(eap, i);
2705}
2706
2707/*
2708 * Edit file "argn" of the argument lists.
2709 */
2710 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002711do_argfile(exarg_T *eap, int argn)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002712{
2713 int other;
2714 char_u *p;
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002715 int old_arg_idx = curwin->w_arg_idx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002716
2717 if (argn < 0 || argn >= ARGCOUNT)
2718 {
2719 if (ARGCOUNT <= 1)
2720 EMSG(_("E163: There is only one file to edit"));
2721 else if (argn < 0)
2722 EMSG(_("E164: Cannot go before first file"));
2723 else
2724 EMSG(_("E165: Cannot go beyond last file"));
2725 }
2726 else
2727 {
2728 setpcmark();
2729#ifdef FEAT_GUI
2730 need_mouse_correct = TRUE;
2731#endif
2732
2733#ifdef FEAT_WINDOWS
Bram Moolenaardf1bdc92006-02-23 21:32:16 +00002734 /* split window or create new tab page first */
2735 if (*eap->cmd == 's' || cmdmod.tab != 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002736 {
2737 if (win_split(0, 0) == FAIL)
2738 return;
Bram Moolenaar3368ea22010-09-21 16:56:35 +02002739 RESET_BINDING(curwin);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002740 }
2741 else
2742#endif
2743 {
2744 /*
2745 * if 'hidden' set, only check for changed file when re-editing
2746 * the same buffer
2747 */
2748 other = TRUE;
Bram Moolenaareb44a682017-08-03 22:44:55 +02002749 if (buf_hide(curbuf))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002750 {
2751 p = fix_fname(alist_name(&ARGLIST[argn]));
2752 other = otherfile(p);
2753 vim_free(p);
2754 }
Bram Moolenaareb44a682017-08-03 22:44:55 +02002755 if ((!buf_hide(curbuf) || !other)
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002756 && check_changed(curbuf, CCGD_AW
2757 | (other ? 0 : CCGD_MULTWIN)
2758 | (eap->forceit ? CCGD_FORCEIT : 0)
2759 | CCGD_EXCMD))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002760 return;
2761 }
2762
2763 curwin->w_arg_idx = argn;
2764 if (argn == ARGCOUNT - 1
2765#ifdef FEAT_WINDOWS
2766 && curwin->w_alist == &global_alist
2767#endif
2768 )
2769 arg_had_last = TRUE;
2770
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002771 /* Edit the file; always use the last known line number.
2772 * When it fails (e.g. Abort for already edited file) restore the
2773 * argument index. */
2774 if (do_ecmd(0, alist_name(&ARGLIST[curwin->w_arg_idx]), NULL,
Bram Moolenaar071d4272004-06-13 20:20:40 +00002775 eap, ECMD_LAST,
Bram Moolenaareb44a682017-08-03 22:44:55 +02002776 (buf_hide(curwin->w_buffer) ? ECMD_HIDE : 0)
Bram Moolenaar701f7af2008-11-15 13:12:07 +00002777 + (eap->forceit ? ECMD_FORCEIT : 0), curwin) == FAIL)
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002778 curwin->w_arg_idx = old_arg_idx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002779 /* like Vi: set the mark where the cursor is in the file. */
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002780 else if (eap->cmdidx != CMD_argdo)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002781 setmark('\'');
2782 }
2783}
2784
2785/*
2786 * ":next", and commands that behave like it.
2787 */
2788 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002789ex_next(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002790{
2791 int i;
2792
2793 /*
2794 * check for changed buffer now, if this fails the argument list is not
2795 * redefined.
2796 */
Bram Moolenaareb44a682017-08-03 22:44:55 +02002797 if ( buf_hide(curbuf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002798 || eap->cmdidx == CMD_snext
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002799 || !check_changed(curbuf, CCGD_AW
2800 | (eap->forceit ? CCGD_FORCEIT : 0)
2801 | CCGD_EXCMD))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002802 {
2803 if (*eap->arg != NUL) /* redefine file list */
2804 {
2805 if (do_arglist(eap->arg, AL_SET, 0) == FAIL)
2806 return;
2807 i = 0;
2808 }
2809 else
2810 i = curwin->w_arg_idx + (int)eap->line2;
2811 do_argfile(eap, i);
2812 }
2813}
2814
Bram Moolenaar0106e3d2016-02-23 18:55:43 +01002815#if defined(FEAT_LISTCMDS) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002816/*
2817 * ":argedit"
2818 */
2819 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002820ex_argedit(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002821{
Bram Moolenaar90305c62017-07-16 15:31:17 +02002822 int i = eap->addr_count ? (int)eap->line2 : curwin->w_arg_idx + 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002823
Bram Moolenaar90305c62017-07-16 15:31:17 +02002824 if (do_arglist(eap->arg, AL_ADD, i) == FAIL)
2825 return;
2826#ifdef FEAT_TITLE
2827 maketitle();
2828#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002829
Bram Moolenaar90305c62017-07-16 15:31:17 +02002830 if (curwin->w_arg_idx == 0 && (curbuf->b_ml.ml_flags & ML_EMPTY)
2831 && curbuf->b_ffname == NULL)
2832 i = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002833 /* Edit the argument. */
Bram Moolenaar90305c62017-07-16 15:31:17 +02002834 if (i < ARGCOUNT)
2835 do_argfile(eap, i);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002836}
2837
2838/*
2839 * ":argadd"
2840 */
2841 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002842ex_argadd(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002843{
2844 do_arglist(eap->arg, AL_ADD,
2845 eap->addr_count > 0 ? (int)eap->line2 : curwin->w_arg_idx + 1);
2846#ifdef FEAT_TITLE
2847 maketitle();
2848#endif
2849}
2850
2851/*
2852 * ":argdelete"
2853 */
2854 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002855ex_argdelete(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002856{
2857 int i;
2858 int n;
2859
2860 if (eap->addr_count > 0)
2861 {
2862 /* ":1,4argdel": Delete all arguments in the range. */
2863 if (eap->line2 > ARGCOUNT)
2864 eap->line2 = ARGCOUNT;
2865 n = eap->line2 - eap->line1 + 1;
Bram Moolenaar69a92fb2017-03-09 15:58:30 +01002866 if (*eap->arg != NUL)
2867 /* Can't have both a range and an argument. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002868 EMSG(_(e_invarg));
Bram Moolenaar69a92fb2017-03-09 15:58:30 +01002869 else if (n <= 0)
2870 {
2871 /* Don't give an error for ":%argdel" if the list is empty. */
2872 if (eap->line1 != 1 || eap->line2 != 0)
2873 EMSG(_(e_invrange));
2874 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002875 else
2876 {
2877 for (i = eap->line1; i <= eap->line2; ++i)
2878 vim_free(ARGLIST[i - 1].ae_fname);
2879 mch_memmove(ARGLIST + eap->line1 - 1, ARGLIST + eap->line2,
2880 (size_t)((ARGCOUNT - eap->line2) * sizeof(aentry_T)));
2881 ALIST(curwin)->al_ga.ga_len -= n;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002882 if (curwin->w_arg_idx >= eap->line2)
2883 curwin->w_arg_idx -= n;
2884 else if (curwin->w_arg_idx > eap->line1)
2885 curwin->w_arg_idx = eap->line1;
Bram Moolenaar72defda2016-01-17 18:04:33 +01002886 if (ARGCOUNT == 0)
2887 curwin->w_arg_idx = 0;
2888 else if (curwin->w_arg_idx >= ARGCOUNT)
2889 curwin->w_arg_idx = ARGCOUNT - 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002890 }
2891 }
2892 else if (*eap->arg == NUL)
2893 EMSG(_(e_argreq));
2894 else
2895 do_arglist(eap->arg, AL_DEL, 0);
2896#ifdef FEAT_TITLE
2897 maketitle();
2898#endif
2899}
2900
2901/*
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002902 * ":argdo", ":windo", ":bufdo", ":tabdo", ":cdo", ":ldo", ":cfdo" and ":lfdo"
Bram Moolenaar071d4272004-06-13 20:20:40 +00002903 */
2904 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002905ex_listdo(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002906{
2907 int i;
2908#ifdef FEAT_WINDOWS
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002909 win_T *wp;
2910 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002911#endif
Bram Moolenaare25bb902015-02-27 20:33:37 +01002912 buf_T *buf = curbuf;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002913 int next_fnum = 0;
2914#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
2915 char_u *save_ei = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002916#endif
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002917 char_u *p_shm_save;
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002918#ifdef FEAT_QUICKFIX
Bram Moolenaared84b762015-09-09 22:35:29 +02002919 int qf_size = 0;
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002920 int qf_idx;
2921#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002922
2923#ifndef FEAT_WINDOWS
2924 if (eap->cmdidx == CMD_windo)
2925 {
2926 ex_ni(eap);
2927 return;
2928 }
2929#endif
2930
Bram Moolenaar0106e3d2016-02-23 18:55:43 +01002931#ifndef FEAT_QUICKFIX
2932 if (eap->cmdidx == CMD_cdo || eap->cmdidx == CMD_ldo ||
2933 eap->cmdidx == CMD_cfdo || eap->cmdidx == CMD_lfdo)
2934 {
2935 ex_ni(eap);
2936 return;
2937 }
2938#endif
2939
Bram Moolenaar071d4272004-06-13 20:20:40 +00002940#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002941 if (eap->cmdidx != CMD_windo && eap->cmdidx != CMD_tabdo)
Bram Moolenaardcaf10e2005-01-21 11:55:25 +00002942 /* Don't do syntax HL autocommands. Skipping the syntax file is a
2943 * great speed improvement. */
2944 save_ei = au_event_disable(",Syntax");
Bram Moolenaar071d4272004-06-13 20:20:40 +00002945#endif
Bram Moolenaar6b1ee342014-08-06 18:17:11 +02002946#ifdef FEAT_CLIPBOARD
2947 start_global_changes();
2948#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002949
2950 if (eap->cmdidx == CMD_windo
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002951 || eap->cmdidx == CMD_tabdo
Bram Moolenaareb44a682017-08-03 22:44:55 +02002952 || buf_hide(curbuf)
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002953 || !check_changed(curbuf, CCGD_AW
2954 | (eap->forceit ? CCGD_FORCEIT : 0)
2955 | CCGD_EXCMD))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002956 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00002957 i = 0;
Bram Moolenaara162bc52015-01-07 16:54:21 +01002958 /* start at the eap->line1 argument/window/buffer */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002959#ifdef FEAT_WINDOWS
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002960 wp = firstwin;
2961 tp = first_tabpage;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002962#endif
Bram Moolenaara162bc52015-01-07 16:54:21 +01002963 switch (eap->cmdidx)
2964 {
2965#ifdef FEAT_WINDOWS
2966 case CMD_windo:
2967 for ( ; wp != NULL && i + 1 < eap->line1; wp = wp->w_next)
2968 i++;
2969 break;
2970 case CMD_tabdo:
2971 for( ; tp != NULL && i + 1 < eap->line1; tp = tp->tp_next)
2972 i++;
2973 break;
2974#endif
2975 case CMD_argdo:
2976 i = eap->line1 - 1;
2977 break;
Bram Moolenaara162bc52015-01-07 16:54:21 +01002978 default:
2979 break;
2980 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002981 /* set pcmark now */
2982 if (eap->cmdidx == CMD_bufdo)
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002983 {
Bram Moolenaare25bb902015-02-27 20:33:37 +01002984 /* Advance to the first listed buffer after "eap->line1". */
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002985 for (buf = firstbuf; buf != NULL && (buf->b_fnum < eap->line1
Bram Moolenaare25bb902015-02-27 20:33:37 +01002986 || !buf->b_p_bl); buf = buf->b_next)
2987 if (buf->b_fnum > eap->line2)
2988 {
2989 buf = NULL;
2990 break;
2991 }
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002992 if (buf != NULL)
Bram Moolenaare25bb902015-02-27 20:33:37 +01002993 goto_buffer(eap, DOBUF_FIRST, FORWARD, buf->b_fnum);
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002994 }
2995#ifdef FEAT_QUICKFIX
2996 else if (eap->cmdidx == CMD_cdo || eap->cmdidx == CMD_ldo
2997 || eap->cmdidx == CMD_cfdo || eap->cmdidx == CMD_lfdo)
2998 {
2999 qf_size = qf_get_size(eap);
3000 if (qf_size <= 0 || eap->line1 > qf_size)
3001 buf = NULL;
3002 else
3003 {
3004 ex_cc(eap);
3005
3006 buf = curbuf;
3007 i = eap->line1 - 1;
3008 if (eap->addr_count <= 0)
3009 /* default is all the quickfix/location list entries */
3010 eap->line2 = qf_size;
3011 }
3012 }
3013#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003014 else
3015 setpcmark();
3016 listcmd_busy = TRUE; /* avoids setting pcmark below */
3017
Bram Moolenaare25bb902015-02-27 20:33:37 +01003018 while (!got_int && buf != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003019 {
3020 if (eap->cmdidx == CMD_argdo)
3021 {
3022 /* go to argument "i" */
3023 if (i == ARGCOUNT)
3024 break;
3025 /* Don't call do_argfile() when already there, it will try
3026 * reloading the file. */
Bram Moolenaard4755bb2004-09-02 19:12:26 +00003027 if (curwin->w_arg_idx != i || !editing_arg_idx(curwin))
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003028 {
3029 /* Clear 'shm' to avoid that the file message overwrites
3030 * any output from the command. */
3031 p_shm_save = vim_strsave(p_shm);
3032 set_option_value((char_u *)"shm", 0L, (char_u *)"", 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003033 do_argfile(eap, i);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003034 set_option_value((char_u *)"shm", 0L, p_shm_save, 0);
3035 vim_free(p_shm_save);
3036 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003037 if (curwin->w_arg_idx != i)
3038 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003039 }
3040#ifdef FEAT_WINDOWS
3041 else if (eap->cmdidx == CMD_windo)
3042 {
Bram Moolenaar32466aa2006-02-24 23:53:04 +00003043 /* go to window "wp" */
3044 if (!win_valid(wp))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003045 break;
Bram Moolenaar32466aa2006-02-24 23:53:04 +00003046 win_goto(wp);
Bram Moolenaar41423242007-05-03 20:11:13 +00003047 if (curwin != wp)
3048 break; /* something must be wrong */
Bram Moolenaar32466aa2006-02-24 23:53:04 +00003049 wp = curwin->w_next;
3050 }
3051 else if (eap->cmdidx == CMD_tabdo)
3052 {
3053 /* go to window "tp" */
3054 if (!valid_tabpage(tp))
3055 break;
Bram Moolenaar49e649f2013-05-06 04:50:35 +02003056 goto_tabpage_tp(tp, TRUE, TRUE);
Bram Moolenaar32466aa2006-02-24 23:53:04 +00003057 tp = tp->tp_next;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003058 }
3059#endif
3060 else if (eap->cmdidx == CMD_bufdo)
3061 {
3062 /* Remember the number of the next listed buffer, in case
3063 * ":bwipe" is used or autocommands do something strange. */
3064 next_fnum = -1;
3065 for (buf = curbuf->b_next; buf != NULL; buf = buf->b_next)
3066 if (buf->b_p_bl)
3067 {
3068 next_fnum = buf->b_fnum;
3069 break;
3070 }
3071 }
3072
Bram Moolenaara162bc52015-01-07 16:54:21 +01003073 ++i;
3074
Bram Moolenaar071d4272004-06-13 20:20:40 +00003075 /* execute the command */
3076 do_cmdline(eap->arg, eap->getline, eap->cookie,
3077 DOCMD_VERBOSE + DOCMD_NOWAIT);
3078
3079 if (eap->cmdidx == CMD_bufdo)
3080 {
3081 /* Done? */
Bram Moolenaara162bc52015-01-07 16:54:21 +01003082 if (next_fnum < 0 || next_fnum > eap->line2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003083 break;
3084 /* Check if the buffer still exists. */
Bram Moolenaar29323592016-07-24 22:04:11 +02003085 FOR_ALL_BUFFERS(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003086 if (buf->b_fnum == next_fnum)
3087 break;
3088 if (buf == NULL)
3089 break;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003090
3091 /* Go to the next buffer. Clear 'shm' to avoid that the file
3092 * message overwrites any output from the command. */
3093 p_shm_save = vim_strsave(p_shm);
3094 set_option_value((char_u *)"shm", 0L, (char_u *)"", 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003095 goto_buffer(eap, DOBUF_FIRST, FORWARD, next_fnum);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003096 set_option_value((char_u *)"shm", 0L, p_shm_save, 0);
3097 vim_free(p_shm_save);
3098
Bram Moolenaaraa23b372015-09-08 18:46:31 +02003099 /* If autocommands took us elsewhere, quit here. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003100 if (curbuf->b_fnum != next_fnum)
3101 break;
3102 }
3103
Bram Moolenaaraa23b372015-09-08 18:46:31 +02003104#ifdef FEAT_QUICKFIX
3105 if (eap->cmdidx == CMD_cdo || eap->cmdidx == CMD_ldo
3106 || eap->cmdidx == CMD_cfdo || eap->cmdidx == CMD_lfdo)
3107 {
3108 if (i >= qf_size || i >= eap->line2)
3109 break;
3110
3111 qf_idx = qf_get_cur_idx(eap);
3112
3113 ex_cnext(eap);
3114
3115 /* If jumping to the next quickfix entry fails, quit here */
3116 if (qf_get_cur_idx(eap) == qf_idx)
3117 break;
3118 }
3119#endif
3120
Bram Moolenaar071d4272004-06-13 20:20:40 +00003121 if (eap->cmdidx == CMD_windo)
3122 {
3123 validate_cursor(); /* cursor may have moved */
3124#ifdef FEAT_SCROLLBIND
3125 /* required when 'scrollbind' has been set */
3126 if (curwin->w_p_scb)
3127 do_check_scrollbind(TRUE);
3128#endif
3129 }
Bram Moolenaara162bc52015-01-07 16:54:21 +01003130
3131#ifdef FEAT_WINDOWS
3132 if (eap->cmdidx == CMD_windo || eap->cmdidx == CMD_tabdo)
3133 if (i+1 > eap->line2)
3134 break;
3135#endif
3136 if (eap->cmdidx == CMD_argdo && i >= eap->line2)
3137 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003138 }
3139 listcmd_busy = FALSE;
3140 }
3141
3142#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
Bram Moolenaar6ac54292005-02-02 23:07:25 +00003143 if (save_ei != NULL)
3144 {
3145 au_event_restore(save_ei);
3146 apply_autocmds(EVENT_SYNTAX, curbuf->b_p_syn,
3147 curbuf->b_fname, TRUE, curbuf);
3148 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003149#endif
Bram Moolenaar6b1ee342014-08-06 18:17:11 +02003150#ifdef FEAT_CLIPBOARD
3151 end_global_changes();
3152#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003153}
3154
3155/*
3156 * Add files[count] to the arglist of the current window after arg "after".
3157 * The file names in files[count] must have been allocated and are taken over.
3158 * Files[] itself is not taken over.
3159 * Returns index of first added argument. Returns -1 when failed (out of mem).
3160 */
3161 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003162alist_add_list(
3163 int count,
3164 char_u **files,
3165 int after) /* where to add: 0 = before first one */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003166{
3167 int i;
Bram Moolenaara24f0a52016-01-17 19:39:00 +01003168 int old_argcount = ARGCOUNT;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003169
3170 if (ga_grow(&ALIST(curwin)->al_ga, count) == OK)
3171 {
3172 if (after < 0)
3173 after = 0;
3174 if (after > ARGCOUNT)
3175 after = ARGCOUNT;
3176 if (after < ARGCOUNT)
3177 mch_memmove(&(ARGLIST[after + count]), &(ARGLIST[after]),
3178 (ARGCOUNT - after) * sizeof(aentry_T));
3179 for (i = 0; i < count; ++i)
3180 {
3181 ARGLIST[after + i].ae_fname = files[i];
3182 ARGLIST[after + i].ae_fnum = buflist_add(files[i], BLN_LISTED);
3183 }
3184 ALIST(curwin)->al_ga.ga_len += count;
Bram Moolenaara24f0a52016-01-17 19:39:00 +01003185 if (old_argcount > 0 && curwin->w_arg_idx >= after)
3186 curwin->w_arg_idx += count;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003187 return after;
3188 }
3189
3190 for (i = 0; i < count; ++i)
3191 vim_free(files[i]);
3192 return -1;
3193}
3194
3195#endif /* FEAT_LISTCMDS */
3196
3197#ifdef FEAT_EVAL
3198/*
3199 * ":compiler[!] {name}"
3200 */
3201 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003202ex_compiler(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003203{
3204 char_u *buf;
3205 char_u *old_cur_comp = NULL;
3206 char_u *p;
3207
3208 if (*eap->arg == NUL)
3209 {
3210 /* List all compiler scripts. */
3211 do_cmdline_cmd((char_u *)"echo globpath(&rtp, 'compiler/*.vim')");
3212 /* ) keep the indenter happy... */
3213 }
3214 else
3215 {
3216 buf = alloc((unsigned)(STRLEN(eap->arg) + 14));
3217 if (buf != NULL)
3218 {
3219 if (eap->forceit)
3220 {
3221 /* ":compiler! {name}" sets global options */
3222 do_cmdline_cmd((char_u *)
3223 "command -nargs=* CompilerSet set <args>");
3224 }
3225 else
3226 {
3227 /* ":compiler! {name}" sets local options.
3228 * To remain backwards compatible "current_compiler" is always
3229 * used. A user's compiler plugin may set it, the distributed
3230 * plugin will then skip the settings. Afterwards set
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003231 * "b:current_compiler" and restore "current_compiler".
3232 * Explicitly prepend "g:" to make it work in a function. */
3233 old_cur_comp = get_var_value((char_u *)"g:current_compiler");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003234 if (old_cur_comp != NULL)
3235 old_cur_comp = vim_strsave(old_cur_comp);
3236 do_cmdline_cmd((char_u *)
3237 "command -nargs=* CompilerSet setlocal <args>");
3238 }
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003239 do_unlet((char_u *)"g:current_compiler", TRUE);
Bram Moolenaar2ce06f62005-01-31 19:19:04 +00003240 do_unlet((char_u *)"b:current_compiler", TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003241
3242 sprintf((char *)buf, "compiler/%s.vim", eap->arg);
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003243 if (source_runtime(buf, DIP_ALL) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003244 EMSG2(_("E666: compiler not supported: %s"), eap->arg);
3245 vim_free(buf);
3246
3247 do_cmdline_cmd((char_u *)":delcommand CompilerSet");
3248
3249 /* Set "b:current_compiler" from "current_compiler". */
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003250 p = get_var_value((char_u *)"g:current_compiler");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003251 if (p != NULL)
3252 set_internal_string_var((char_u *)"b:current_compiler", p);
3253
3254 /* Restore "current_compiler" for ":compiler {name}". */
3255 if (!eap->forceit)
3256 {
3257 if (old_cur_comp != NULL)
3258 {
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003259 set_internal_string_var((char_u *)"g:current_compiler",
Bram Moolenaar071d4272004-06-13 20:20:40 +00003260 old_cur_comp);
3261 vim_free(old_cur_comp);
3262 }
3263 else
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003264 do_unlet((char_u *)"g:current_compiler", TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003265 }
3266 }
3267 }
3268}
3269#endif
3270
3271/*
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003272 * ":runtime [what] {name}"
Bram Moolenaar071d4272004-06-13 20:20:40 +00003273 */
3274 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003275ex_runtime(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003276{
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003277 char_u *arg = eap->arg;
3278 char_u *p = skiptowhite(arg);
3279 int len = (int)(p - arg);
3280 int flags = eap->forceit ? DIP_ALL : 0;
3281
3282 if (STRNCMP(arg, "START", len) == 0)
3283 {
3284 flags += DIP_START + DIP_NORTP;
3285 arg = skipwhite(arg + len);
3286 }
3287 else if (STRNCMP(arg, "OPT", len) == 0)
3288 {
3289 flags += DIP_OPT + DIP_NORTP;
3290 arg = skipwhite(arg + len);
3291 }
3292 else if (STRNCMP(arg, "PACK", len) == 0)
3293 {
3294 flags += DIP_START + DIP_OPT + DIP_NORTP;
3295 arg = skipwhite(arg + len);
3296 }
3297 else if (STRNCMP(arg, "ALL", len) == 0)
3298 {
3299 flags += DIP_START + DIP_OPT;
3300 arg = skipwhite(arg + len);
3301 }
3302
3303 source_runtime(arg, flags);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003304}
3305
Bram Moolenaar071d4272004-06-13 20:20:40 +00003306 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003307source_callback(char_u *fname, void *cookie UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003308{
Bram Moolenaar910f66f2006-04-05 20:41:53 +00003309 (void)do_source(fname, FALSE, DOSO_NONE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003310}
3311
3312/*
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003313 * Find the file "name" in all directories in "path" and invoke
3314 * "callback(fname, cookie)".
3315 * "name" can contain wildcards.
3316 * When "flags" has DIP_ALL: source all files, otherwise only the first one.
3317 * When "flags" has DIP_DIR: find directories instead of files.
3318 * When "flags" has DIP_ERR: give an error message if there is no match.
3319 *
3320 * return FAIL when no file could be sourced, OK otherwise.
3321 */
Bram Moolenaar6bef5302016-03-12 21:28:26 +01003322 int
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003323do_in_path(
3324 char_u *path,
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003325 char_u *name,
Bram Moolenaar91715872016-03-03 17:13:03 +01003326 int flags,
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003327 void (*callback)(char_u *fname, void *ck),
3328 void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003329{
3330 char_u *rtp;
3331 char_u *np;
3332 char_u *buf;
3333 char_u *rtp_copy;
3334 char_u *tail;
3335 int num_files;
3336 char_u **files;
3337 int i;
3338 int did_one = FALSE;
3339#ifdef AMIGA
3340 struct Process *proc = (struct Process *)FindTask(0L);
3341 APTR save_winptr = proc->pr_WindowPtr;
3342
3343 /* Avoid a requester here for a volume that doesn't exist. */
3344 proc->pr_WindowPtr = (APTR)-1L;
3345#endif
3346
3347 /* Make a copy of 'runtimepath'. Invoking the callback may change the
3348 * value. */
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003349 rtp_copy = vim_strsave(path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003350 buf = alloc(MAXPATHL);
3351 if (buf != NULL && rtp_copy != NULL)
3352 {
Bram Moolenaarc09a6d62013-06-10 21:27:29 +02003353 if (p_verbose > 1 && name != NULL)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003354 {
3355 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00003356 smsg((char_u *)_("Searching for \"%s\" in \"%s\""),
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003357 (char *)name, (char *)path);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003358 verbose_leave();
3359 }
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00003360
Bram Moolenaar071d4272004-06-13 20:20:40 +00003361 /* Loop over all entries in 'runtimepath'. */
3362 rtp = rtp_copy;
Bram Moolenaar91715872016-03-03 17:13:03 +01003363 while (*rtp != NUL && ((flags & DIP_ALL) || !did_one))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003364 {
Bram Moolenaar66459b72016-08-06 19:01:55 +02003365 size_t buflen;
3366
Bram Moolenaar071d4272004-06-13 20:20:40 +00003367 /* Copy the path from 'runtimepath' to buf[]. */
3368 copy_option_part(&rtp, buf, MAXPATHL, ",");
Bram Moolenaar66459b72016-08-06 19:01:55 +02003369 buflen = STRLEN(buf);
3370
3371 /* Skip after or non-after directories. */
3372 if (flags & (DIP_NOAFTER | DIP_AFTER))
3373 {
3374 int is_after = buflen >= 5
3375 && STRCMP(buf + buflen - 5, "after") == 0;
3376
3377 if ((is_after && (flags & DIP_NOAFTER))
3378 || (!is_after && (flags & DIP_AFTER)))
3379 continue;
3380 }
3381
Bram Moolenaarc09a6d62013-06-10 21:27:29 +02003382 if (name == NULL)
3383 {
3384 (*callback)(buf, (void *) &cookie);
3385 if (!did_one)
3386 did_one = (cookie == NULL);
3387 }
Bram Moolenaar66459b72016-08-06 19:01:55 +02003388 else if (buflen + STRLEN(name) + 2 < MAXPATHL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003389 {
3390 add_pathsep(buf);
3391 tail = buf + STRLEN(buf);
3392
3393 /* Loop over all patterns in "name" */
3394 np = name;
Bram Moolenaar91715872016-03-03 17:13:03 +01003395 while (*np != NUL && ((flags & DIP_ALL) || !did_one))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003396 {
3397 /* Append the pattern from "name" to buf[]. */
3398 copy_option_part(&np, tail, (int)(MAXPATHL - (tail - buf)),
3399 "\t ");
3400
3401 if (p_verbose > 2)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003402 {
3403 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00003404 smsg((char_u *)_("Searching for \"%s\""), buf);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003405 verbose_leave();
3406 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003407
3408 /* Expand wildcards, invoke the callback for each match. */
3409 if (gen_expand_wildcards(1, &buf, &num_files, &files,
Bram Moolenaar91715872016-03-03 17:13:03 +01003410 (flags & DIP_DIR) ? EW_DIR : EW_FILE) == OK)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003411 {
3412 for (i = 0; i < num_files; ++i)
3413 {
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00003414 (*callback)(files[i], cookie);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003415 did_one = TRUE;
Bram Moolenaar91715872016-03-03 17:13:03 +01003416 if (!(flags & DIP_ALL))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003417 break;
3418 }
3419 FreeWild(num_files, files);
3420 }
3421 }
3422 }
3423 }
3424 }
3425 vim_free(buf);
3426 vim_free(rtp_copy);
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003427 if (!did_one && name != NULL)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003428 {
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003429 char *basepath = path == p_rtp ? "runtimepath" : "packpath";
3430
3431 if (flags & DIP_ERR)
3432 EMSG3(_(e_dirnotf), basepath, name);
3433 else if (p_verbose > 0)
3434 {
3435 verbose_enter();
3436 smsg((char_u *)_("not found in '%s': \"%s\""), basepath, name);
3437 verbose_leave();
3438 }
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003439 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003440
3441#ifdef AMIGA
3442 proc->pr_WindowPtr = save_winptr;
3443#endif
3444
3445 return did_one ? OK : FAIL;
3446}
3447
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003448/*
Bram Moolenaar07ecfa62017-06-27 14:43:55 +02003449 * Find "name" in "path". When found, invoke the callback function for
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003450 * it: callback(fname, "cookie")
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003451 * When "flags" has DIP_ALL repeat for all matches, otherwise only the first
3452 * one is used.
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003453 * Returns OK when at least one match found, FAIL otherwise.
3454 *
Bram Moolenaar07ecfa62017-06-27 14:43:55 +02003455 * If "name" is NULL calls callback for each entry in "path". Cookie is
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003456 * passed by reference in this case, setting it to NULL indicates that callback
3457 * has done its job.
3458 */
Bram Moolenaar07ecfa62017-06-27 14:43:55 +02003459 static int
3460do_in_path_and_pp(
3461 char_u *path,
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003462 char_u *name,
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003463 int flags,
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003464 void (*callback)(char_u *fname, void *ck),
3465 void *cookie)
3466{
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003467 int done = FAIL;
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003468 char_u *s;
3469 int len;
3470 char *start_dir = "pack/*/start/*/%s";
3471 char *opt_dir = "pack/*/opt/*/%s";
3472
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003473 if ((flags & DIP_NORTP) == 0)
Bram Moolenaar07ecfa62017-06-27 14:43:55 +02003474 done = do_in_path(path, name, flags, callback, cookie);
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003475
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003476 if ((done == FAIL || (flags & DIP_ALL)) && (flags & DIP_START))
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003477 {
Bram Moolenaar1c8b4ed2016-03-17 21:51:03 +01003478 len = (int)(STRLEN(start_dir) + STRLEN(name));
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003479 s = alloc(len);
3480 if (s == NULL)
3481 return FAIL;
3482 vim_snprintf((char *)s, len, start_dir, name);
3483 done = do_in_path(p_pp, s, flags, callback, cookie);
3484 vim_free(s);
3485 }
3486
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003487 if ((done == FAIL || (flags & DIP_ALL)) && (flags & DIP_OPT))
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003488 {
Bram Moolenaar1c8b4ed2016-03-17 21:51:03 +01003489 len = (int)(STRLEN(opt_dir) + STRLEN(name));
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003490 s = alloc(len);
3491 if (s == NULL)
3492 return FAIL;
3493 vim_snprintf((char *)s, len, opt_dir, name);
3494 done = do_in_path(p_pp, s, flags, callback, cookie);
3495 vim_free(s);
3496 }
3497
3498 return done;
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003499}
3500
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003501/*
Bram Moolenaar07ecfa62017-06-27 14:43:55 +02003502 * Just like do_in_path_and_pp(), using 'runtimepath' for "path".
3503 */
3504 int
3505do_in_runtimepath(
3506 char_u *name,
3507 int flags,
3508 void (*callback)(char_u *fname, void *ck),
3509 void *cookie)
3510{
3511 return do_in_path_and_pp(p_rtp, name, flags, callback, cookie);
3512}
3513
3514/*
3515 * Source the file "name" from all directories in 'runtimepath'.
3516 * "name" can contain wildcards.
3517 * When "flags" has DIP_ALL: source all files, otherwise only the first one.
3518 *
3519 * return FAIL when no file could be sourced, OK otherwise.
3520 */
3521 int
3522source_runtime(char_u *name, int flags)
3523{
3524 return source_in_path(p_rtp, name, flags);
3525}
3526
3527/*
3528 * Just like source_runtime(), but use "path" instead of 'runtimepath'.
3529 */
3530 int
3531source_in_path(char_u *path, char_u *name, int flags)
3532{
3533 return do_in_path_and_pp(path, name, flags, source_callback, NULL);
3534}
3535
3536
3537/*
Bram Moolenaarf3654822016-03-04 22:12:23 +01003538 * Expand wildcards in "pat" and invoke do_source() for each match.
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003539 */
3540 static void
Bram Moolenaarf3654822016-03-04 22:12:23 +01003541source_all_matches(char_u *pat)
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003542{
Bram Moolenaarf3654822016-03-04 22:12:23 +01003543 int num_files;
3544 char_u **files;
3545 int i;
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003546
Bram Moolenaarf3654822016-03-04 22:12:23 +01003547 if (gen_expand_wildcards(1, &pat, &num_files, &files, EW_FILE) == OK)
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003548 {
Bram Moolenaarf3654822016-03-04 22:12:23 +01003549 for (i = 0; i < num_files; ++i)
3550 (void)do_source(files[i], FALSE, DOSO_NONE);
3551 FreeWild(num_files, files);
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003552 }
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003553}
3554
Bram Moolenaar49b27322016-04-05 21:13:00 +02003555/* used for "cookie" of add_pack_plugin() */
3556static int APP_ADD_DIR;
3557static int APP_LOAD;
3558static int APP_BOTH;
3559
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003560 static void
Bram Moolenaar91715872016-03-03 17:13:03 +01003561add_pack_plugin(char_u *fname, void *cookie)
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003562{
Bram Moolenaarf3654822016-03-04 22:12:23 +01003563 char_u *p4, *p3, *p2, *p1, *p;
3564 char_u *insp;
Bram Moolenaar91715872016-03-03 17:13:03 +01003565 int c;
3566 char_u *new_rtp;
3567 int keep;
Bram Moolenaarb0550662016-05-31 21:37:36 +02003568 size_t oldlen;
3569 size_t addlen;
Bram Moolenaara5702442016-05-24 19:37:29 +02003570 char_u *afterdir;
Bram Moolenaarb0550662016-05-31 21:37:36 +02003571 size_t afterlen = 0;
Bram Moolenaar91715872016-03-03 17:13:03 +01003572 char_u *ffname = fix_fname(fname);
Bram Moolenaarfef524b2016-07-02 22:07:22 +02003573 size_t fname_len;
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003574 char_u *buf = NULL;
3575 char_u *rtp_ffname;
3576 int match;
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003577
Bram Moolenaar91715872016-03-03 17:13:03 +01003578 if (ffname == NULL)
3579 return;
Bram Moolenaar49b27322016-04-05 21:13:00 +02003580 if (cookie != &APP_LOAD && strstr((char *)p_rtp, (char *)ffname) == NULL)
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003581 {
Bram Moolenaara5702442016-05-24 19:37:29 +02003582 /* directory is not yet in 'runtimepath', add it */
Bram Moolenaarf3654822016-03-04 22:12:23 +01003583 p4 = p3 = p2 = p1 = get_past_head(ffname);
Bram Moolenaar91acfff2017-03-12 19:22:36 +01003584 for (p = p1; *p; MB_PTR_ADV(p))
Bram Moolenaarf3654822016-03-04 22:12:23 +01003585 if (vim_ispathsep_nocolon(*p))
3586 {
3587 p4 = p3; p3 = p2; p2 = p1; p1 = p;
3588 }
3589
3590 /* now we have:
Bram Moolenaaraf1a0e32016-03-09 22:19:26 +01003591 * rtp/pack/name/start/name
3592 * p4 p3 p2 p1
Bram Moolenaarf3654822016-03-04 22:12:23 +01003593 *
3594 * find the part up to "pack" in 'runtimepath' */
3595 c = *p4;
3596 *p4 = NUL;
Bram Moolenaar4c5717e2016-07-01 15:39:40 +02003597
3598 /* Find "ffname" in "p_rtp", ignoring '/' vs '\' differences. */
3599 fname_len = STRLEN(ffname);
3600 insp = p_rtp;
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003601 buf = alloc(MAXPATHL);
3602 if (buf == NULL)
3603 goto theend;
3604 while (*insp != NUL)
Bram Moolenaar4c5717e2016-07-01 15:39:40 +02003605 {
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003606 copy_option_part(&insp, buf, MAXPATHL, ",");
3607 add_pathsep(buf);
3608 rtp_ffname = fix_fname(buf);
3609 if (rtp_ffname == NULL)
3610 goto theend;
3611 match = vim_fnamencmp(rtp_ffname, ffname, fname_len) == 0;
3612 vim_free(rtp_ffname);
3613 if (match)
Bram Moolenaar4c5717e2016-07-01 15:39:40 +02003614 break;
Bram Moolenaar4c5717e2016-07-01 15:39:40 +02003615 }
3616
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003617 if (*insp == NUL)
Bram Moolenaarf3654822016-03-04 22:12:23 +01003618 /* not found, append at the end */
3619 insp = p_rtp + STRLEN(p_rtp);
3620 else
Bram Moolenaarf3654822016-03-04 22:12:23 +01003621 /* append after the matching directory. */
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003622 --insp;
Bram Moolenaarf3654822016-03-04 22:12:23 +01003623 *p4 = c;
3624
Bram Moolenaara5702442016-05-24 19:37:29 +02003625 /* check if rtp/pack/name/start/name/after exists */
3626 afterdir = concat_fnames(ffname, (char_u *)"after", TRUE);
3627 if (afterdir != NULL && mch_isdir(afterdir))
3628 afterlen = STRLEN(afterdir) + 1; /* add one for comma */
3629
Bram Moolenaarb0550662016-05-31 21:37:36 +02003630 oldlen = STRLEN(p_rtp);
3631 addlen = STRLEN(ffname) + 1; /* add one for comma */
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003632 new_rtp = alloc((int)(oldlen + addlen + afterlen + 1));
3633 /* add one for NUL */
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003634 if (new_rtp == NULL)
Bram Moolenaarf3654822016-03-04 22:12:23 +01003635 goto theend;
3636 keep = (int)(insp - p_rtp);
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003637 mch_memmove(new_rtp, p_rtp, keep);
3638 new_rtp[keep] = ',';
Bram Moolenaara5702442016-05-24 19:37:29 +02003639 mch_memmove(new_rtp + keep + 1, ffname, addlen);
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003640 if (p_rtp[keep] != NUL)
Bram Moolenaara5702442016-05-24 19:37:29 +02003641 mch_memmove(new_rtp + keep + addlen, p_rtp + keep,
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003642 oldlen - keep + 1);
Bram Moolenaara5702442016-05-24 19:37:29 +02003643 if (afterlen > 0)
3644 {
3645 STRCAT(new_rtp, ",");
3646 STRCAT(new_rtp, afterdir);
3647 }
Bram Moolenaar863c1a92016-03-03 15:47:06 +01003648 set_option_value((char_u *)"rtp", 0L, new_rtp, 0);
3649 vim_free(new_rtp);
Bram Moolenaara5702442016-05-24 19:37:29 +02003650 vim_free(afterdir);
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003651 }
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003652
Bram Moolenaar49b27322016-04-05 21:13:00 +02003653 if (cookie != &APP_ADD_DIR)
Bram Moolenaarf3654822016-03-04 22:12:23 +01003654 {
Bram Moolenaar71fb0c12016-04-02 22:44:16 +02003655 static char *plugpat = "%s/plugin/**/*.vim";
Bram Moolenaarf3654822016-03-04 22:12:23 +01003656 static char *ftpat = "%s/ftdetect/*.vim";
3657 int len;
3658 char_u *pat;
3659
3660 len = (int)STRLEN(ffname) + (int)STRLEN(ftpat);
3661 pat = alloc(len);
3662 if (pat == NULL)
3663 goto theend;
3664 vim_snprintf((char *)pat, len, plugpat, ffname);
3665 source_all_matches(pat);
3666
3667#ifdef FEAT_AUTOCMD
3668 {
3669 char_u *cmd = vim_strsave((char_u *)"g:did_load_filetypes");
3670
3671 /* If runtime/filetype.vim wasn't loaded yet, the scripts will be
3672 * found when it loads. */
3673 if (cmd != NULL && eval_to_number(cmd) > 0)
3674 {
3675 do_cmdline_cmd((char_u *)"augroup filetypedetect");
3676 vim_snprintf((char *)pat, len, ftpat, ffname);
3677 source_all_matches(pat);
3678 do_cmdline_cmd((char_u *)"augroup END");
3679 }
3680 vim_free(cmd);
3681 }
3682#endif
Bram Moolenaarba8cd122016-03-19 14:16:39 +01003683 vim_free(pat);
Bram Moolenaarf3654822016-03-04 22:12:23 +01003684 }
3685
3686theend:
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003687 vim_free(buf);
Bram Moolenaarf3654822016-03-04 22:12:23 +01003688 vim_free(ffname);
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003689}
3690
Bram Moolenaarce876aa2017-06-04 17:47:42 +02003691/*
3692 * Add all packages in the "start" directory to 'runtimepath'.
3693 */
3694 void
3695add_pack_start_dirs(void)
3696{
3697 do_in_path(p_pp, (char_u *)"pack/*/start/*", DIP_ALL + DIP_DIR,
3698 add_pack_plugin, &APP_ADD_DIR);
3699}
3700
3701/*
3702 * Load plugins from all packages in the "start" directory.
3703 */
3704 void
3705load_start_packages(void)
3706{
3707 did_source_packages = TRUE;
3708 do_in_path(p_pp, (char_u *)"pack/*/start/*", DIP_ALL + DIP_DIR,
3709 add_pack_plugin, &APP_LOAD);
3710}
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003711
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003712/*
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003713 * ":packloadall"
Bram Moolenaarf3654822016-03-04 22:12:23 +01003714 * Find plugins in the package directories and source them.
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003715 */
3716 void
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003717ex_packloadall(exarg_T *eap)
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003718{
Bram Moolenaarce876aa2017-06-04 17:47:42 +02003719 if (!did_source_packages || eap->forceit)
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003720 {
Bram Moolenaar49b27322016-04-05 21:13:00 +02003721 /* First do a round to add all directories to 'runtimepath', then load
3722 * the plugins. This allows for plugins to use an autoload directory
3723 * of another plugin. */
Bram Moolenaarce876aa2017-06-04 17:47:42 +02003724 add_pack_start_dirs();
3725 load_start_packages();
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003726 }
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003727}
3728
3729/*
Bram Moolenaarf3654822016-03-04 22:12:23 +01003730 * ":packadd[!] {name}"
Bram Moolenaar91715872016-03-03 17:13:03 +01003731 */
3732 void
3733ex_packadd(exarg_T *eap)
3734{
3735 static char *plugpat = "pack/*/opt/%s";
3736 int len;
3737 char *pat;
3738
3739 len = (int)STRLEN(plugpat) + (int)STRLEN(eap->arg);
3740 pat = (char *)alloc(len);
3741 if (pat == NULL)
3742 return;
3743 vim_snprintf(pat, len, plugpat, eap->arg);
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003744 do_in_path(p_pp, (char_u *)pat, DIP_ALL + DIP_DIR + DIP_ERR,
Bram Moolenaar49b27322016-04-05 21:13:00 +02003745 add_pack_plugin, eap->forceit ? &APP_ADD_DIR : &APP_BOTH);
Bram Moolenaar91715872016-03-03 17:13:03 +01003746 vim_free(pat);
3747}
3748
Bram Moolenaar071d4272004-06-13 20:20:40 +00003749#if defined(FEAT_EVAL) && defined(FEAT_AUTOCMD)
3750/*
3751 * ":options"
3752 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003753 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003754ex_options(
3755 exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003756{
Bram Moolenaarab6c8582017-08-11 17:15:09 +02003757 vim_setenv((char_u *)"OPTWIN_CMD", (char_u *)(cmdmod.tab ? "tab" : ""));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003758 cmd_source((char_u *)SYS_OPTWIN_FILE, NULL);
3759}
3760#endif
3761
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01003762#if defined(FEAT_PYTHON3) || defined(FEAT_PYTHON) || defined(PROTO)
3763
3764# if (defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)) || defined(PROTO)
3765/*
3766 * Detect Python 3 or 2, and initialize 'pyxversion'.
3767 */
3768 void
3769init_pyxversion(void)
3770{
3771 if (p_pyx == 0)
3772 {
3773 if (python3_enabled(FALSE))
3774 p_pyx = 3;
3775 else if (python_enabled(FALSE))
3776 p_pyx = 2;
3777 }
3778}
3779# endif
3780
3781/*
3782 * Does a file contain one of the following strings at the beginning of any
3783 * line?
3784 * "#!(any string)python2" => returns 2
3785 * "#!(any string)python3" => returns 3
3786 * "# requires python 2.x" => returns 2
3787 * "# requires python 3.x" => returns 3
3788 * otherwise return 0.
3789 */
3790 static int
3791requires_py_version(char_u *filename)
3792{
3793 FILE *file;
3794 int requires_py_version = 0;
3795 int i, lines;
3796
3797 lines = (int)p_mls;
3798 if (lines < 0)
3799 lines = 5;
3800
3801 file = mch_fopen((char *)filename, "r");
3802 if (file != NULL)
3803 {
3804 for (i = 0; i < lines; i++)
3805 {
3806 if (vim_fgets(IObuff, IOSIZE, file))
3807 break;
3808 if (i == 0 && IObuff[0] == '#' && IObuff[1] == '!')
3809 {
3810 /* Check shebang. */
3811 if (strstr((char *)IObuff + 2, "python2") != NULL)
3812 {
3813 requires_py_version = 2;
3814 break;
3815 }
3816 if (strstr((char *)IObuff + 2, "python3") != NULL)
3817 {
3818 requires_py_version = 3;
3819 break;
3820 }
3821 }
3822 IObuff[21] = '\0';
3823 if (STRCMP("# requires python 2.x", IObuff) == 0)
3824 {
3825 requires_py_version = 2;
3826 break;
3827 }
3828 if (STRCMP("# requires python 3.x", IObuff) == 0)
3829 {
3830 requires_py_version = 3;
3831 break;
3832 }
3833 }
3834 fclose(file);
3835 }
3836 return requires_py_version;
3837}
3838
3839
3840/*
3841 * Source a python file using the requested python version.
3842 */
3843 static void
3844source_pyx_file(exarg_T *eap, char_u *fname)
3845{
3846 exarg_T ex;
3847 int v = requires_py_version(fname);
3848
3849# if defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)
3850 init_pyxversion();
3851# endif
3852 if (v == 0)
3853 {
3854# if defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)
3855 /* user didn't choose a preference, 'pyx' is used */
3856 v = p_pyx;
3857# elif defined(FEAT_PYTHON)
3858 v = 2;
3859# elif defined(FEAT_PYTHON3)
3860 v = 3;
3861# endif
3862 }
3863
3864 /*
3865 * now source, if required python version is not supported show
3866 * unobtrusive message.
3867 */
3868 if (eap == NULL)
3869 vim_memset(&ex, 0, sizeof(ex));
3870 else
3871 ex = *eap;
3872 ex.arg = fname;
3873 ex.cmd = (char_u *)(v == 2 ? "pyfile" : "pyfile3");
3874
3875 if (v == 2)
3876 {
3877# ifdef FEAT_PYTHON
3878 ex_pyfile(&ex);
3879# else
3880 vim_snprintf((char *)IObuff, IOSIZE,
3881 _("W20: Required python version 2.x not supported, ignoring file: %s"),
3882 fname);
3883 MSG(IObuff);
3884# endif
3885 return;
3886 }
3887 else
3888 {
3889# ifdef FEAT_PYTHON3
3890 ex_py3file(&ex);
3891# else
3892 vim_snprintf((char *)IObuff, IOSIZE,
3893 _("W21: Required python version 3.x not supported, ignoring file: %s"),
3894 fname);
3895 MSG(IObuff);
3896# endif
3897 return;
3898 }
3899}
3900
3901/*
3902 * ":pyxfile {fname}"
3903 */
3904 void
3905ex_pyxfile(exarg_T *eap)
3906{
3907 source_pyx_file(eap, eap->arg);
3908}
3909
3910/*
3911 * ":pyx"
3912 */
3913 void
3914ex_pyx(exarg_T *eap)
3915{
3916# if defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)
3917 init_pyxversion();
3918 if (p_pyx == 2)
3919 ex_python(eap);
3920 else
3921 ex_py3(eap);
3922# elif defined(FEAT_PYTHON)
3923 ex_python(eap);
3924# elif defined(FEAT_PYTHON3)
3925 ex_py3(eap);
3926# endif
3927}
3928
3929/*
3930 * ":pyxdo"
3931 */
3932 void
3933ex_pyxdo(exarg_T *eap)
3934{
3935# if defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)
3936 init_pyxversion();
3937 if (p_pyx == 2)
3938 ex_pydo(eap);
3939 else
3940 ex_py3do(eap);
3941# elif defined(FEAT_PYTHON)
3942 ex_pydo(eap);
3943# elif defined(FEAT_PYTHON3)
3944 ex_py3do(eap);
3945# endif
3946}
3947
3948#endif
3949
Bram Moolenaar071d4272004-06-13 20:20:40 +00003950/*
3951 * ":source {fname}"
3952 */
3953 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003954ex_source(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003955{
3956#ifdef FEAT_BROWSE
3957 if (cmdmod.browse)
3958 {
3959 char_u *fname = NULL;
3960
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00003961 fname = do_browse(0, (char_u *)_("Source Vim script"), eap->arg,
Bram Moolenaar071d4272004-06-13 20:20:40 +00003962 NULL, NULL, BROWSE_FILTER_MACROS, NULL);
3963 if (fname != NULL)
3964 {
3965 cmd_source(fname, eap);
3966 vim_free(fname);
3967 }
3968 }
3969 else
3970#endif
3971 cmd_source(eap->arg, eap);
3972}
3973
3974 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003975cmd_source(char_u *fname, exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003976{
3977 if (*fname == NUL)
3978 EMSG(_(e_argreq));
3979
Bram Moolenaar071d4272004-06-13 20:20:40 +00003980 else if (eap != NULL && eap->forceit)
Bram Moolenaar84a05ac2013-05-06 04:24:17 +02003981 /* ":source!": read Normal mode commands
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00003982 * Need to execute the commands directly. This is required at least
3983 * for:
Bram Moolenaar071d4272004-06-13 20:20:40 +00003984 * - ":g" command busy
3985 * - after ":argdo", ":windo" or ":bufdo"
3986 * - another command follows
3987 * - inside a loop
3988 */
3989 openscript(fname, global_busy || listcmd_busy || eap->nextcmd != NULL
3990#ifdef FEAT_EVAL
3991 || eap->cstack->cs_idx >= 0
3992#endif
3993 );
3994
3995 /* ":source" read ex commands */
Bram Moolenaar910f66f2006-04-05 20:41:53 +00003996 else if (do_source(fname, FALSE, DOSO_NONE) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003997 EMSG2(_(e_notopen), fname);
3998}
3999
4000/*
4001 * ":source" and associated commands.
4002 */
4003/*
4004 * Structure used to store info for each sourced file.
4005 * It is shared between do_source() and getsourceline().
4006 * This is required, because it needs to be handed to do_cmdline() and
4007 * sourcing can be done recursively.
4008 */
4009struct source_cookie
4010{
4011 FILE *fp; /* opened file for sourcing */
4012 char_u *nextline; /* if not NULL: line that was read ahead */
4013 int finished; /* ":finish" used */
Bram Moolenaar860cae12010-06-05 23:22:07 +02004014#if defined(USE_CRNL) || defined(USE_CR)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004015 int fileformat; /* EOL_UNKNOWN, EOL_UNIX or EOL_DOS */
4016 int error; /* TRUE if LF found after CR-LF */
4017#endif
4018#ifdef FEAT_EVAL
4019 linenr_T breakpoint; /* next line with breakpoint or zero */
4020 char_u *fname; /* name of sourced file */
4021 int dbg_tick; /* debug_tick when breakpoint was set */
4022 int level; /* top nesting level of sourced file */
4023#endif
4024#ifdef FEAT_MBYTE
4025 vimconv_T conv; /* type of conversion */
4026#endif
4027};
4028
4029#ifdef FEAT_EVAL
4030/*
4031 * Return the address holding the next breakpoint line for a source cookie.
4032 */
4033 linenr_T *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004034source_breakpoint(void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004035{
4036 return &((struct source_cookie *)cookie)->breakpoint;
4037}
4038
4039/*
4040 * Return the address holding the debug tick for a source cookie.
4041 */
4042 int *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004043source_dbg_tick(void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004044{
4045 return &((struct source_cookie *)cookie)->dbg_tick;
4046}
4047
4048/*
4049 * Return the nesting level for a source cookie.
4050 */
4051 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004052source_level(void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004053{
4054 return ((struct source_cookie *)cookie)->level;
4055}
4056#endif
4057
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01004058static char_u *get_one_sourceline(struct source_cookie *sp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004059
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004060#if (defined(WIN32) && defined(FEAT_CSCOPE)) || defined(HAVE_FD_CLOEXEC)
4061# define USE_FOPEN_NOINH
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01004062static FILE *fopen_noinh_readbin(char *filename);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004063
4064/*
4065 * Special function to open a file without handle inheritance.
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004066 * When possible the handle is closed on exec().
Bram Moolenaar071d4272004-06-13 20:20:40 +00004067 */
4068 static FILE *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004069fopen_noinh_readbin(char *filename)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004070{
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004071# ifdef WIN32
Bram Moolenaar8d8ef0b2010-01-20 21:41:47 +01004072 int fd_tmp = mch_open(filename, O_RDONLY | O_BINARY | O_NOINHERIT, 0);
4073# else
4074 int fd_tmp = mch_open(filename, O_RDONLY, 0);
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004075# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004076
4077 if (fd_tmp == -1)
4078 return NULL;
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004079
4080# ifdef HAVE_FD_CLOEXEC
4081 {
4082 int fdflags = fcntl(fd_tmp, F_GETFD);
4083 if (fdflags >= 0 && (fdflags & FD_CLOEXEC) == 0)
Bram Moolenaarcde88542015-08-11 19:14:00 +02004084 (void)fcntl(fd_tmp, F_SETFD, fdflags | FD_CLOEXEC);
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004085 }
4086# endif
4087
Bram Moolenaar071d4272004-06-13 20:20:40 +00004088 return fdopen(fd_tmp, READBIN);
4089}
4090#endif
4091
4092
4093/*
4094 * do_source: Read the file "fname" and execute its lines as EX commands.
4095 *
4096 * This function may be called recursively!
4097 *
4098 * return FAIL if file could not be opened, OK otherwise
4099 */
4100 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004101do_source(
4102 char_u *fname,
4103 int check_other, /* check for .vimrc and _vimrc */
4104 int is_vimrc) /* DOSO_ value */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004105{
4106 struct source_cookie cookie;
4107 char_u *save_sourcing_name;
4108 linenr_T save_sourcing_lnum;
4109 char_u *p;
4110 char_u *fname_exp;
Bram Moolenaar73881402009-02-04 16:50:47 +00004111 char_u *firstline = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004112 int retval = FAIL;
4113#ifdef FEAT_EVAL
4114 scid_T save_current_SID;
4115 static scid_T last_current_SID = 0;
4116 void *save_funccalp;
4117 int save_debug_break_level = debug_break_level;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004118 scriptitem_T *si = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004119# ifdef UNIX
Bram Moolenaar8767f522016-07-01 17:17:39 +02004120 stat_T st;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004121 int stat_ok;
4122# endif
4123#endif
4124#ifdef STARTUPTIME
4125 struct timeval tv_rel;
4126 struct timeval tv_start;
4127#endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00004128#ifdef FEAT_PROFILE
4129 proftime_T wait_start;
4130#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004131
Bram Moolenaar071d4272004-06-13 20:20:40 +00004132 p = expand_env_save(fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004133 if (p == NULL)
4134 return retval;
4135 fname_exp = fix_fname(p);
4136 vim_free(p);
4137 if (fname_exp == NULL)
4138 return retval;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004139 if (mch_isdir(fname_exp))
4140 {
Bram Moolenaar555b2802005-05-19 21:08:39 +00004141 smsg((char_u *)_("Cannot source a directory: \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004142 goto theend;
4143 }
4144
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00004145#ifdef FEAT_AUTOCMD
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00004146 /* Apply SourceCmd autocommands, they should get the file and source it. */
4147 if (has_autocmd(EVENT_SOURCECMD, fname_exp, NULL)
4148 && apply_autocmds(EVENT_SOURCECMD, fname_exp, fname_exp,
4149 FALSE, curbuf))
Bram Moolenaarf33943e2008-01-15 21:17:30 +00004150 {
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00004151# ifdef FEAT_EVAL
Bram Moolenaarf33943e2008-01-15 21:17:30 +00004152 retval = aborting() ? FAIL : OK;
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00004153# else
Bram Moolenaarf33943e2008-01-15 21:17:30 +00004154 retval = OK;
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00004155# endif
Bram Moolenaarf33943e2008-01-15 21:17:30 +00004156 goto theend;
4157 }
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00004158
4159 /* Apply SourcePre autocommands, they may get the file. */
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00004160 apply_autocmds(EVENT_SOURCEPRE, fname_exp, fname_exp, FALSE, curbuf);
4161#endif
4162
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004163#ifdef USE_FOPEN_NOINH
Bram Moolenaar071d4272004-06-13 20:20:40 +00004164 cookie.fp = fopen_noinh_readbin((char *)fname_exp);
4165#else
4166 cookie.fp = mch_fopen((char *)fname_exp, READBIN);
4167#endif
4168 if (cookie.fp == NULL && check_other)
4169 {
4170 /*
4171 * Try again, replacing file name ".vimrc" by "_vimrc" or vice versa,
4172 * and ".exrc" by "_exrc" or vice versa.
4173 */
4174 p = gettail(fname_exp);
4175 if ((*p == '.' || *p == '_')
4176 && (STRICMP(p + 1, "vimrc") == 0
4177 || STRICMP(p + 1, "gvimrc") == 0
4178 || STRICMP(p + 1, "exrc") == 0))
4179 {
4180 if (*p == '_')
4181 *p = '.';
4182 else
4183 *p = '_';
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004184#ifdef USE_FOPEN_NOINH
Bram Moolenaar071d4272004-06-13 20:20:40 +00004185 cookie.fp = fopen_noinh_readbin((char *)fname_exp);
4186#else
4187 cookie.fp = mch_fopen((char *)fname_exp, READBIN);
4188#endif
4189 }
4190 }
4191
4192 if (cookie.fp == NULL)
4193 {
4194 if (p_verbose > 0)
4195 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004196 verbose_enter();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004197 if (sourcing_name == NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00004198 smsg((char_u *)_("could not source \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004199 else
4200 smsg((char_u *)_("line %ld: could not source \"%s\""),
Bram Moolenaar555b2802005-05-19 21:08:39 +00004201 sourcing_lnum, fname);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004202 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004203 }
4204 goto theend;
4205 }
4206
4207 /*
4208 * The file exists.
4209 * - In verbose mode, give a message.
4210 * - For a vimrc file, may want to set 'compatible', call vimrc_found().
4211 */
4212 if (p_verbose > 1)
4213 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004214 verbose_enter();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004215 if (sourcing_name == NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00004216 smsg((char_u *)_("sourcing \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004217 else
4218 smsg((char_u *)_("line %ld: sourcing \"%s\""),
Bram Moolenaar555b2802005-05-19 21:08:39 +00004219 sourcing_lnum, fname);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004220 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004221 }
Bram Moolenaar910f66f2006-04-05 20:41:53 +00004222 if (is_vimrc == DOSO_VIMRC)
4223 vimrc_found(fname_exp, (char_u *)"MYVIMRC");
4224 else if (is_vimrc == DOSO_GVIMRC)
4225 vimrc_found(fname_exp, (char_u *)"MYGVIMRC");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004226
4227#ifdef USE_CRNL
4228 /* If no automatic file format: Set default to CR-NL. */
4229 if (*p_ffs == NUL)
4230 cookie.fileformat = EOL_DOS;
4231 else
4232 cookie.fileformat = EOL_UNKNOWN;
4233 cookie.error = FALSE;
4234#endif
4235
4236#ifdef USE_CR
4237 /* If no automatic file format: Set default to CR. */
4238 if (*p_ffs == NUL)
4239 cookie.fileformat = EOL_MAC;
4240 else
4241 cookie.fileformat = EOL_UNKNOWN;
4242 cookie.error = FALSE;
4243#endif
4244
4245 cookie.nextline = NULL;
4246 cookie.finished = FALSE;
4247
4248#ifdef FEAT_EVAL
4249 /*
4250 * Check if this script has a breakpoint.
4251 */
4252 cookie.breakpoint = dbg_find_breakpoint(TRUE, fname_exp, (linenr_T)0);
4253 cookie.fname = fname_exp;
4254 cookie.dbg_tick = debug_tick;
4255
4256 cookie.level = ex_nesting_level;
4257#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004258
4259 /*
4260 * Keep the sourcing name/lnum, for recursive calls.
4261 */
4262 save_sourcing_name = sourcing_name;
4263 sourcing_name = fname_exp;
4264 save_sourcing_lnum = sourcing_lnum;
4265 sourcing_lnum = 0;
4266
Bram Moolenaar73881402009-02-04 16:50:47 +00004267#ifdef FEAT_MBYTE
4268 cookie.conv.vc_type = CONV_NONE; /* no conversion */
4269
4270 /* Read the first line so we can check for a UTF-8 BOM. */
4271 firstline = getsourceline(0, (void *)&cookie, 0);
4272 if (firstline != NULL && STRLEN(firstline) >= 3 && firstline[0] == 0xef
4273 && firstline[1] == 0xbb && firstline[2] == 0xbf)
4274 {
4275 /* Found BOM; setup conversion, skip over BOM and recode the line. */
4276 convert_setup(&cookie.conv, (char_u *)"utf-8", p_enc);
4277 p = string_convert(&cookie.conv, firstline + 3, NULL);
Bram Moolenaar4e2a5952009-02-05 19:48:25 +00004278 if (p == NULL)
4279 p = vim_strsave(firstline + 3);
Bram Moolenaar73881402009-02-04 16:50:47 +00004280 if (p != NULL)
4281 {
4282 vim_free(firstline);
4283 firstline = p;
4284 }
4285 }
4286#endif
4287
Bram Moolenaar071d4272004-06-13 20:20:40 +00004288#ifdef STARTUPTIME
Bram Moolenaarc4e41982010-01-19 16:31:47 +01004289 if (time_fd != NULL)
4290 time_push(&tv_rel, &tv_start);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004291#endif
4292
4293#ifdef FEAT_EVAL
Bram Moolenaar05159a02005-02-26 23:04:13 +00004294# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004295 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004296 prof_child_enter(&wait_start); /* entering a child now */
4297# endif
4298
4299 /* Don't use local function variables, if called from a function.
4300 * Also starts profiling timer for nested script. */
4301 save_funccalp = save_funccal();
4302
Bram Moolenaar071d4272004-06-13 20:20:40 +00004303 /*
4304 * Check if this script was sourced before to finds its SID.
4305 * If it's new, generate a new SID.
4306 */
4307 save_current_SID = current_SID;
4308# ifdef UNIX
4309 stat_ok = (mch_stat((char *)fname_exp, &st) >= 0);
4310# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00004311 for (current_SID = script_items.ga_len; current_SID > 0; --current_SID)
4312 {
4313 si = &SCRIPT_ITEM(current_SID);
4314 if (si->sn_name != NULL
Bram Moolenaar071d4272004-06-13 20:20:40 +00004315 && (
4316# ifdef UNIX
Bram Moolenaar7c626922005-02-07 22:01:03 +00004317 /* Compare dev/ino when possible, it catches symbolic
4318 * links. Also compare file names, the inode may change
4319 * when the file was edited. */
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00004320 ((stat_ok && si->sn_dev_valid)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004321 && (si->sn_dev == st.st_dev
4322 && si->sn_ino == st.st_ino)) ||
Bram Moolenaar071d4272004-06-13 20:20:40 +00004323# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00004324 fnamecmp(si->sn_name, fname_exp) == 0))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004325 break;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004326 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004327 if (current_SID == 0)
4328 {
4329 current_SID = ++last_current_SID;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004330 if (ga_grow(&script_items, (int)(current_SID - script_items.ga_len))
4331 == FAIL)
4332 goto almosttheend;
4333 while (script_items.ga_len < current_SID)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004334 {
Bram Moolenaar05159a02005-02-26 23:04:13 +00004335 ++script_items.ga_len;
4336 SCRIPT_ITEM(script_items.ga_len).sn_name = NULL;
4337# ifdef FEAT_PROFILE
4338 SCRIPT_ITEM(script_items.ga_len).sn_prof_on = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004339# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004340 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00004341 si = &SCRIPT_ITEM(current_SID);
4342 si->sn_name = fname_exp;
4343 fname_exp = NULL;
4344# ifdef UNIX
4345 if (stat_ok)
4346 {
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00004347 si->sn_dev_valid = TRUE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004348 si->sn_dev = st.st_dev;
4349 si->sn_ino = st.st_ino;
4350 }
4351 else
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00004352 si->sn_dev_valid = FALSE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004353# endif
4354
Bram Moolenaar071d4272004-06-13 20:20:40 +00004355 /* Allocate the local script variables to use for this script. */
4356 new_script_vars(current_SID);
4357 }
4358
Bram Moolenaar05159a02005-02-26 23:04:13 +00004359# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004360 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004361 {
4362 int forceit;
4363
4364 /* Check if we do profiling for this script. */
4365 if (!si->sn_prof_on && has_profiling(TRUE, si->sn_name, &forceit))
4366 {
4367 script_do_profile(si);
4368 si->sn_pr_force = forceit;
4369 }
4370 if (si->sn_prof_on)
4371 {
4372 ++si->sn_pr_count;
4373 profile_start(&si->sn_pr_start);
4374 profile_zero(&si->sn_pr_children);
4375 }
4376 }
4377# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004378#endif
4379
4380 /*
4381 * Call do_cmdline, which will call getsourceline() to get the lines.
4382 */
Bram Moolenaar73881402009-02-04 16:50:47 +00004383 do_cmdline(firstline, getsourceline, (void *)&cookie,
Bram Moolenaar071d4272004-06-13 20:20:40 +00004384 DOCMD_VERBOSE|DOCMD_NOWAIT|DOCMD_REPEAT);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004385 retval = OK;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004386
4387#ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004388 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004389 {
4390 /* Get "si" again, "script_items" may have been reallocated. */
4391 si = &SCRIPT_ITEM(current_SID);
4392 if (si->sn_prof_on)
4393 {
4394 profile_end(&si->sn_pr_start);
4395 profile_sub_wait(&wait_start, &si->sn_pr_start);
4396 profile_add(&si->sn_pr_total, &si->sn_pr_start);
Bram Moolenaar1056d982006-03-09 22:37:52 +00004397 profile_self(&si->sn_pr_self, &si->sn_pr_start,
4398 &si->sn_pr_children);
Bram Moolenaar05159a02005-02-26 23:04:13 +00004399 }
4400 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004401#endif
4402
4403 if (got_int)
4404 EMSG(_(e_interr));
4405 sourcing_name = save_sourcing_name;
4406 sourcing_lnum = save_sourcing_lnum;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004407 if (p_verbose > 1)
4408 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004409 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00004410 smsg((char_u *)_("finished sourcing %s"), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004411 if (sourcing_name != NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00004412 smsg((char_u *)_("continuing in %s"), sourcing_name);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004413 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004414 }
4415#ifdef STARTUPTIME
Bram Moolenaarc4e41982010-01-19 16:31:47 +01004416 if (time_fd != NULL)
4417 {
4418 vim_snprintf((char *)IObuff, IOSIZE, "sourcing %s", fname);
4419 time_msg((char *)IObuff, &tv_start);
4420 time_pop(&tv_rel);
4421 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004422#endif
4423
4424#ifdef FEAT_EVAL
4425 /*
4426 * After a "finish" in debug mode, need to break at first command of next
4427 * sourced file.
4428 */
4429 if (save_debug_break_level > ex_nesting_level
4430 && debug_break_level == ex_nesting_level)
4431 ++debug_break_level;
4432#endif
4433
Bram Moolenaar05159a02005-02-26 23:04:13 +00004434#ifdef FEAT_EVAL
4435almosttheend:
4436 current_SID = save_current_SID;
4437 restore_funccal(save_funccalp);
4438# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004439 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004440 prof_child_exit(&wait_start); /* leaving a child now */
4441# endif
4442#endif
4443 fclose(cookie.fp);
4444 vim_free(cookie.nextline);
Bram Moolenaar73881402009-02-04 16:50:47 +00004445 vim_free(firstline);
Bram Moolenaar05159a02005-02-26 23:04:13 +00004446#ifdef FEAT_MBYTE
4447 convert_setup(&cookie.conv, NULL, NULL);
4448#endif
4449
Bram Moolenaar071d4272004-06-13 20:20:40 +00004450theend:
4451 vim_free(fname_exp);
4452 return retval;
4453}
4454
4455#if defined(FEAT_EVAL) || defined(PROTO)
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00004456
Bram Moolenaar071d4272004-06-13 20:20:40 +00004457/*
4458 * ":scriptnames"
4459 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004460 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004461ex_scriptnames(exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004462{
4463 int i;
4464
Bram Moolenaar05159a02005-02-26 23:04:13 +00004465 for (i = 1; i <= script_items.ga_len && !got_int; ++i)
4466 if (SCRIPT_ITEM(i).sn_name != NULL)
Bram Moolenaard58ea072011-06-26 04:25:30 +02004467 {
4468 home_replace(NULL, SCRIPT_ITEM(i).sn_name,
4469 NameBuff, MAXPATHL, TRUE);
4470 smsg((char_u *)"%3d: %s", i, NameBuff);
Bram Moolenaar970a1b82012-03-23 18:39:18 +01004471 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004472}
4473
4474# if defined(BACKSLASH_IN_FILENAME) || defined(PROTO)
4475/*
4476 * Fix slashes in the list of script names for 'shellslash'.
4477 */
4478 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004479scriptnames_slash_adjust(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004480{
4481 int i;
4482
Bram Moolenaar05159a02005-02-26 23:04:13 +00004483 for (i = 1; i <= script_items.ga_len; ++i)
4484 if (SCRIPT_ITEM(i).sn_name != NULL)
4485 slash_adjust(SCRIPT_ITEM(i).sn_name);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004486}
4487# endif
4488
4489/*
4490 * Get a pointer to a script name. Used for ":verbose set".
4491 */
4492 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004493get_scriptname(scid_T id)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004494{
4495 if (id == SID_MODELINE)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00004496 return (char_u *)_("modeline");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004497 if (id == SID_CMDARG)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00004498 return (char_u *)_("--cmd argument");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004499 if (id == SID_CARG)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00004500 return (char_u *)_("-c argument");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004501 if (id == SID_ENV)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00004502 return (char_u *)_("environment variable");
4503 if (id == SID_ERROR)
4504 return (char_u *)_("error handler");
Bram Moolenaar05159a02005-02-26 23:04:13 +00004505 return SCRIPT_ITEM(id).sn_name;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004506}
Bram Moolenaar05159a02005-02-26 23:04:13 +00004507
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00004508# if defined(EXITFREE) || defined(PROTO)
4509 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004510free_scriptnames(void)
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00004511{
4512 int i;
4513
4514 for (i = script_items.ga_len; i > 0; --i)
4515 vim_free(SCRIPT_ITEM(i).sn_name);
4516 ga_clear(&script_items);
4517}
4518# endif
4519
Bram Moolenaar071d4272004-06-13 20:20:40 +00004520#endif
4521
4522#if defined(USE_CR) || defined(PROTO)
4523
4524# if defined(__MSL__) && (__MSL__ >= 22)
4525/*
4526 * Newer version of the Metrowerks library handle DOS and UNIX files
4527 * without help.
4528 * Test with earlier versions, MSL 2.2 is the library supplied with
4529 * Codewarrior Pro 2.
4530 */
4531 char *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004532fgets_cr(char *s, int n, FILE *stream)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004533{
4534 return fgets(s, n, stream);
4535}
4536# else
4537/*
4538 * Version of fgets() which also works for lines ending in a <CR> only
4539 * (Macintosh format).
4540 * For older versions of the Metrowerks library.
4541 * At least CodeWarrior 9 needed this code.
4542 */
4543 char *
Bram Moolenaard14e00e2016-01-31 17:30:51 +01004544fgets_cr(char *s, int n, FILE *stream)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004545{
4546 int c = 0;
4547 int char_read = 0;
4548
4549 while (!feof(stream) && c != '\r' && c != '\n' && char_read < n - 1)
4550 {
4551 c = fgetc(stream);
4552 s[char_read++] = c;
4553 /* If the file is in DOS format, we need to skip a NL after a CR. I
4554 * thought it was the other way around, but this appears to work... */
4555 if (c == '\n')
4556 {
4557 c = fgetc(stream);
4558 if (c != '\r')
4559 ungetc(c, stream);
4560 }
4561 }
4562
4563 s[char_read] = 0;
4564 if (char_read == 0)
4565 return NULL;
4566
4567 if (feof(stream) && char_read == 1)
4568 return NULL;
4569
4570 return s;
4571}
4572# endif
4573#endif
4574
4575/*
4576 * Get one full line from a sourced file.
4577 * Called by do_cmdline() when it's called from do_source().
4578 *
4579 * Return a pointer to the line in allocated memory.
4580 * Return NULL for end-of-file or some error.
4581 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004582 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004583getsourceline(int c UNUSED, void *cookie, int indent UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004584{
4585 struct source_cookie *sp = (struct source_cookie *)cookie;
4586 char_u *line;
Bram Moolenaarc047b9a2012-02-11 20:40:55 +01004587 char_u *p;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004588
4589#ifdef FEAT_EVAL
4590 /* If breakpoints have been added/deleted need to check for it. */
4591 if (sp->dbg_tick < debug_tick)
4592 {
4593 sp->breakpoint = dbg_find_breakpoint(TRUE, sp->fname, sourcing_lnum);
4594 sp->dbg_tick = debug_tick;
4595 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00004596# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004597 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004598 script_line_end();
4599# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004600#endif
4601 /*
4602 * Get current line. If there is a read-ahead line, use it, otherwise get
4603 * one now.
4604 */
4605 if (sp->finished)
4606 line = NULL;
4607 else if (sp->nextline == NULL)
4608 line = get_one_sourceline(sp);
4609 else
4610 {
4611 line = sp->nextline;
4612 sp->nextline = NULL;
4613 ++sourcing_lnum;
4614 }
Bram Moolenaare2cc9702005-03-15 22:43:58 +00004615#ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004616 if (line != NULL && do_profiling == PROF_YES)
Bram Moolenaare2cc9702005-03-15 22:43:58 +00004617 script_line_start();
4618#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004619
4620 /* Only concatenate lines starting with a \ when 'cpoptions' doesn't
4621 * contain the 'C' flag. */
4622 if (line != NULL && (vim_strchr(p_cpo, CPO_CONCAT) == NULL))
4623 {
4624 /* compensate for the one line read-ahead */
4625 --sourcing_lnum;
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004626
4627 /* Get the next line and concatenate it when it starts with a
4628 * backslash. We always need to read the next line, keep it in
4629 * sp->nextline. */
4630 sp->nextline = get_one_sourceline(sp);
4631 if (sp->nextline != NULL && *(p = skipwhite(sp->nextline)) == '\\')
Bram Moolenaar071d4272004-06-13 20:20:40 +00004632 {
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004633 garray_T ga;
4634
Bram Moolenaarb549a732012-02-22 18:29:33 +01004635 ga_init2(&ga, (int)sizeof(char_u), 400);
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004636 ga_concat(&ga, line);
4637 ga_concat(&ga, p + 1);
4638 for (;;)
4639 {
4640 vim_free(sp->nextline);
4641 sp->nextline = get_one_sourceline(sp);
4642 if (sp->nextline == NULL)
4643 break;
4644 p = skipwhite(sp->nextline);
4645 if (*p != '\\')
4646 break;
Bram Moolenaarb549a732012-02-22 18:29:33 +01004647 /* Adjust the growsize to the current length to speed up
4648 * concatenating many lines. */
4649 if (ga.ga_len > 400)
4650 {
4651 if (ga.ga_len > 8000)
4652 ga.ga_growsize = 8000;
4653 else
4654 ga.ga_growsize = ga.ga_len;
4655 }
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004656 ga_concat(&ga, p + 1);
4657 }
4658 ga_append(&ga, NUL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004659 vim_free(line);
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004660 line = ga.ga_data;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004661 }
4662 }
4663
4664#ifdef FEAT_MBYTE
4665 if (line != NULL && sp->conv.vc_type != CONV_NONE)
4666 {
Bram Moolenaarc047b9a2012-02-11 20:40:55 +01004667 char_u *s;
4668
Bram Moolenaar071d4272004-06-13 20:20:40 +00004669 /* Convert the encoding of the script line. */
4670 s = string_convert(&sp->conv, line, NULL);
4671 if (s != NULL)
4672 {
4673 vim_free(line);
4674 line = s;
4675 }
4676 }
4677#endif
4678
4679#ifdef FEAT_EVAL
4680 /* Did we encounter a breakpoint? */
4681 if (sp->breakpoint != 0 && sp->breakpoint <= sourcing_lnum)
4682 {
4683 dbg_breakpoint(sp->fname, sourcing_lnum);
4684 /* Find next breakpoint. */
4685 sp->breakpoint = dbg_find_breakpoint(TRUE, sp->fname, sourcing_lnum);
4686 sp->dbg_tick = debug_tick;
4687 }
4688#endif
4689
4690 return line;
4691}
4692
4693 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004694get_one_sourceline(struct source_cookie *sp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004695{
4696 garray_T ga;
4697 int len;
4698 int c;
4699 char_u *buf;
4700#ifdef USE_CRNL
4701 int has_cr; /* CR-LF found */
4702#endif
4703#ifdef USE_CR
4704 char_u *scan;
4705#endif
4706 int have_read = FALSE;
4707
4708 /* use a growarray to store the sourced line */
Bram Moolenaar6ac54292005-02-02 23:07:25 +00004709 ga_init2(&ga, 1, 250);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004710
4711 /*
4712 * Loop until there is a finished line (or end-of-file).
4713 */
4714 sourcing_lnum++;
4715 for (;;)
4716 {
Bram Moolenaar6ac54292005-02-02 23:07:25 +00004717 /* make room to read at least 120 (more) characters */
4718 if (ga_grow(&ga, 120) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004719 break;
4720 buf = (char_u *)ga.ga_data;
4721
4722#ifdef USE_CR
4723 if (sp->fileformat == EOL_MAC)
4724 {
Bram Moolenaar86b68352004-12-27 21:59:20 +00004725 if (fgets_cr((char *)buf + ga.ga_len, ga.ga_maxlen - ga.ga_len,
4726 sp->fp) == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004727 break;
4728 }
4729 else
4730#endif
Bram Moolenaar86b68352004-12-27 21:59:20 +00004731 if (fgets((char *)buf + ga.ga_len, ga.ga_maxlen - ga.ga_len,
4732 sp->fp) == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004733 break;
Bram Moolenaar6ac54292005-02-02 23:07:25 +00004734 len = ga.ga_len + (int)STRLEN(buf + ga.ga_len);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004735#ifdef USE_CRNL
4736 /* Ignore a trailing CTRL-Z, when in Dos mode. Only recognize the
4737 * CTRL-Z by its own, or after a NL. */
4738 if ( (len == 1 || (len >= 2 && buf[len - 2] == '\n'))
4739 && sp->fileformat == EOL_DOS
4740 && buf[len - 1] == Ctrl_Z)
4741 {
4742 buf[len - 1] = NUL;
4743 break;
4744 }
4745#endif
4746
4747#ifdef USE_CR
4748 /* If the read doesn't stop on a new line, and there's
4749 * some CR then we assume a Mac format */
4750 if (sp->fileformat == EOL_UNKNOWN)
4751 {
4752 if (buf[len - 1] != '\n' && vim_strchr(buf, '\r') != NULL)
4753 sp->fileformat = EOL_MAC;
4754 else
4755 sp->fileformat = EOL_UNIX;
4756 }
4757
4758 if (sp->fileformat == EOL_MAC)
4759 {
4760 scan = vim_strchr(buf, '\r');
4761
4762 if (scan != NULL)
4763 {
4764 *scan = '\n';
4765 if (*(scan + 1) != 0)
4766 {
4767 *(scan + 1) = 0;
4768 fseek(sp->fp, (long)(scan - buf - len + 1), SEEK_CUR);
4769 }
4770 }
4771 len = STRLEN(buf);
4772 }
4773#endif
4774
4775 have_read = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004776 ga.ga_len = len;
4777
4778 /* If the line was longer than the buffer, read more. */
Bram Moolenaar86b68352004-12-27 21:59:20 +00004779 if (ga.ga_maxlen - ga.ga_len == 1 && buf[len - 1] != '\n')
Bram Moolenaar071d4272004-06-13 20:20:40 +00004780 continue;
4781
4782 if (len >= 1 && buf[len - 1] == '\n') /* remove trailing NL */
4783 {
4784#ifdef USE_CRNL
4785 has_cr = (len >= 2 && buf[len - 2] == '\r');
4786 if (sp->fileformat == EOL_UNKNOWN)
4787 {
4788 if (has_cr)
4789 sp->fileformat = EOL_DOS;
4790 else
4791 sp->fileformat = EOL_UNIX;
4792 }
4793
4794 if (sp->fileformat == EOL_DOS)
4795 {
4796 if (has_cr) /* replace trailing CR */
4797 {
4798 buf[len - 2] = '\n';
4799 --len;
4800 --ga.ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004801 }
4802 else /* lines like ":map xx yy^M" will have failed */
4803 {
4804 if (!sp->error)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00004805 {
Bram Moolenaar8820b482017-03-16 17:23:31 +01004806 msg_source(HL_ATTR(HLF_W));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004807 EMSG(_("W15: Warning: Wrong line separator, ^M may be missing"));
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00004808 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004809 sp->error = TRUE;
4810 sp->fileformat = EOL_UNIX;
4811 }
4812 }
4813#endif
4814 /* The '\n' is escaped if there is an odd number of ^V's just
4815 * before it, first set "c" just before the 'V's and then check
4816 * len&c parities (is faster than ((len-c)%2 == 0)) -- Acevedo */
4817 for (c = len - 2; c >= 0 && buf[c] == Ctrl_V; c--)
4818 ;
4819 if ((len & 1) != (c & 1)) /* escaped NL, read more */
4820 {
4821 sourcing_lnum++;
4822 continue;
4823 }
4824
4825 buf[len - 1] = NUL; /* remove the NL */
4826 }
4827
4828 /*
4829 * Check for ^C here now and then, so recursive :so can be broken.
4830 */
4831 line_breakcheck();
4832 break;
4833 }
4834
4835 if (have_read)
4836 return (char_u *)ga.ga_data;
4837
4838 vim_free(ga.ga_data);
4839 return NULL;
4840}
4841
Bram Moolenaar05159a02005-02-26 23:04:13 +00004842#if defined(FEAT_PROFILE) || defined(PROTO)
4843/*
4844 * Called when starting to read a script line.
4845 * "sourcing_lnum" must be correct!
4846 * When skipping lines it may not actually be executed, but we won't find out
4847 * until later and we need to store the time now.
4848 */
4849 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004850script_line_start(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004851{
4852 scriptitem_T *si;
4853 sn_prl_T *pp;
4854
4855 if (current_SID <= 0 || current_SID > script_items.ga_len)
4856 return;
4857 si = &SCRIPT_ITEM(current_SID);
4858 if (si->sn_prof_on && sourcing_lnum >= 1)
4859 {
Bram Moolenaar8c8de832008-06-24 22:58:06 +00004860 /* Grow the array before starting the timer, so that the time spent
Bram Moolenaar05159a02005-02-26 23:04:13 +00004861 * here isn't counted. */
Bram Moolenaarcde88542015-08-11 19:14:00 +02004862 (void)ga_grow(&si->sn_prl_ga, (int)(sourcing_lnum - si->sn_prl_ga.ga_len));
Bram Moolenaar05159a02005-02-26 23:04:13 +00004863 si->sn_prl_idx = sourcing_lnum - 1;
4864 while (si->sn_prl_ga.ga_len <= si->sn_prl_idx
4865 && si->sn_prl_ga.ga_len < si->sn_prl_ga.ga_maxlen)
4866 {
4867 /* Zero counters for a line that was not used before. */
4868 pp = &PRL_ITEM(si, si->sn_prl_ga.ga_len);
4869 pp->snp_count = 0;
4870 profile_zero(&pp->sn_prl_total);
4871 profile_zero(&pp->sn_prl_self);
4872 ++si->sn_prl_ga.ga_len;
4873 }
4874 si->sn_prl_execed = FALSE;
4875 profile_start(&si->sn_prl_start);
4876 profile_zero(&si->sn_prl_children);
4877 profile_get_wait(&si->sn_prl_wait);
4878 }
4879}
4880
4881/*
4882 * Called when actually executing a function line.
4883 */
4884 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004885script_line_exec(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004886{
4887 scriptitem_T *si;
4888
4889 if (current_SID <= 0 || current_SID > script_items.ga_len)
4890 return;
4891 si = &SCRIPT_ITEM(current_SID);
4892 if (si->sn_prof_on && si->sn_prl_idx >= 0)
4893 si->sn_prl_execed = TRUE;
4894}
4895
4896/*
4897 * Called when done with a function line.
4898 */
4899 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004900script_line_end(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004901{
4902 scriptitem_T *si;
4903 sn_prl_T *pp;
4904
4905 if (current_SID <= 0 || current_SID > script_items.ga_len)
4906 return;
4907 si = &SCRIPT_ITEM(current_SID);
4908 if (si->sn_prof_on && si->sn_prl_idx >= 0
4909 && si->sn_prl_idx < si->sn_prl_ga.ga_len)
4910 {
4911 if (si->sn_prl_execed)
4912 {
4913 pp = &PRL_ITEM(si, si->sn_prl_idx);
4914 ++pp->snp_count;
4915 profile_end(&si->sn_prl_start);
4916 profile_sub_wait(&si->sn_prl_wait, &si->sn_prl_start);
Bram Moolenaar05159a02005-02-26 23:04:13 +00004917 profile_add(&pp->sn_prl_total, &si->sn_prl_start);
Bram Moolenaar1056d982006-03-09 22:37:52 +00004918 profile_self(&pp->sn_prl_self, &si->sn_prl_start,
4919 &si->sn_prl_children);
Bram Moolenaar05159a02005-02-26 23:04:13 +00004920 }
4921 si->sn_prl_idx = -1;
4922 }
4923}
4924#endif
4925
Bram Moolenaar071d4272004-06-13 20:20:40 +00004926/*
4927 * ":scriptencoding": Set encoding conversion for a sourced script.
4928 * Without the multi-byte feature it's simply ignored.
4929 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004930 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004931ex_scriptencoding(exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004932{
4933#ifdef FEAT_MBYTE
4934 struct source_cookie *sp;
4935 char_u *name;
4936
4937 if (!getline_equal(eap->getline, eap->cookie, getsourceline))
4938 {
4939 EMSG(_("E167: :scriptencoding used outside of a sourced file"));
4940 return;
4941 }
4942
4943 if (*eap->arg != NUL)
4944 {
4945 name = enc_canonize(eap->arg);
4946 if (name == NULL) /* out of memory */
4947 return;
4948 }
4949 else
4950 name = eap->arg;
4951
4952 /* Setup for conversion from the specified encoding to 'encoding'. */
4953 sp = (struct source_cookie *)getline_cookie(eap->getline, eap->cookie);
4954 convert_setup(&sp->conv, name, p_enc);
4955
4956 if (name != eap->arg)
4957 vim_free(name);
4958#endif
4959}
4960
4961#if defined(FEAT_EVAL) || defined(PROTO)
4962/*
4963 * ":finish": Mark a sourced file as finished.
4964 */
4965 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004966ex_finish(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004967{
4968 if (getline_equal(eap->getline, eap->cookie, getsourceline))
4969 do_finish(eap, FALSE);
4970 else
4971 EMSG(_("E168: :finish used outside of a sourced file"));
4972}
4973
4974/*
4975 * Mark a sourced file as finished. Possibly makes the ":finish" pending.
4976 * Also called for a pending finish at the ":endtry" or after returning from
4977 * an extra do_cmdline(). "reanimate" is used in the latter case.
4978 */
4979 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004980do_finish(exarg_T *eap, int reanimate)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004981{
4982 int idx;
4983
4984 if (reanimate)
4985 ((struct source_cookie *)getline_cookie(eap->getline,
4986 eap->cookie))->finished = FALSE;
4987
4988 /*
4989 * Cleanup (and inactivate) conditionals, but stop when a try conditional
4990 * not in its finally clause (which then is to be executed next) is found.
4991 * In this case, make the ":finish" pending for execution at the ":endtry".
4992 * Otherwise, finish normally.
4993 */
4994 idx = cleanup_conditionals(eap->cstack, 0, TRUE);
4995 if (idx >= 0)
4996 {
4997 eap->cstack->cs_pending[idx] = CSTP_FINISH;
4998 report_make_pending(CSTP_FINISH, NULL);
4999 }
5000 else
5001 ((struct source_cookie *)getline_cookie(eap->getline,
5002 eap->cookie))->finished = TRUE;
5003}
5004
5005
5006/*
5007 * Return TRUE when a sourced file had the ":finish" command: Don't give error
5008 * message for missing ":endif".
5009 * Return FALSE when not sourcing a file.
5010 */
5011 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005012source_finished(
5013 char_u *(*fgetline)(int, void *, int),
5014 void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005015{
Bram Moolenaar89d40322006-08-29 15:30:07 +00005016 return (getline_equal(fgetline, cookie, getsourceline)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005017 && ((struct source_cookie *)getline_cookie(
Bram Moolenaar89d40322006-08-29 15:30:07 +00005018 fgetline, cookie))->finished);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005019}
5020#endif
5021
5022#if defined(FEAT_LISTCMDS) || defined(PROTO)
5023/*
5024 * ":checktime [buffer]"
5025 */
5026 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005027ex_checktime(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005028{
5029 buf_T *buf;
5030 int save_no_check_timestamps = no_check_timestamps;
5031
5032 no_check_timestamps = 0;
5033 if (eap->addr_count == 0) /* default is all buffers */
5034 check_timestamps(FALSE);
5035 else
5036 {
5037 buf = buflist_findnr((int)eap->line2);
5038 if (buf != NULL) /* cannot happen? */
5039 (void)buf_check_timestamp(buf, FALSE);
5040 }
5041 no_check_timestamps = save_no_check_timestamps;
5042}
5043#endif
5044
Bram Moolenaar071d4272004-06-13 20:20:40 +00005045#if (defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
5046 && (defined(FEAT_EVAL) || defined(FEAT_MULTI_LANG))
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02005047# define HAVE_GET_LOCALE_VAL
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005048static char_u *get_locale_val(int what);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005049
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005050 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005051get_locale_val(int what)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005052{
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005053 char_u *loc;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005054
Bram Moolenaar48e330a2016-02-23 14:53:34 +01005055 /* Obtain the locale value from the libraries. */
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005056 loc = (char_u *)setlocale(what, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005057
Bram Moolenaar61660ea2006-04-07 21:40:07 +00005058# ifdef WIN32
Bram Moolenaar071d4272004-06-13 20:20:40 +00005059 if (loc != NULL)
5060 {
5061 char_u *p;
5062
Bram Moolenaar61660ea2006-04-07 21:40:07 +00005063 /* setocale() returns something like "LC_COLLATE=<name>;LC_..." when
5064 * one of the values (e.g., LC_CTYPE) differs. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005065 p = vim_strchr(loc, '=');
5066 if (p != NULL)
5067 {
5068 loc = ++p;
5069 while (*p != NUL) /* remove trailing newline */
5070 {
Bram Moolenaar61660ea2006-04-07 21:40:07 +00005071 if (*p < ' ' || *p == ';')
Bram Moolenaar071d4272004-06-13 20:20:40 +00005072 {
5073 *p = NUL;
5074 break;
5075 }
5076 ++p;
5077 }
5078 }
5079 }
5080# endif
5081
5082 return loc;
5083}
5084#endif
5085
5086
5087#ifdef WIN32
5088/*
5089 * On MS-Windows locale names are strings like "German_Germany.1252", but
5090 * gettext expects "de". Try to translate one into another here for a few
5091 * supported languages.
5092 */
5093 static char_u *
5094gettext_lang(char_u *name)
5095{
5096 int i;
5097 static char *(mtable[]) = {
5098 "afrikaans", "af",
5099 "czech", "cs",
5100 "dutch", "nl",
5101 "german", "de",
5102 "english_united kingdom", "en_GB",
5103 "spanish", "es",
5104 "french", "fr",
5105 "italian", "it",
5106 "japanese", "ja",
5107 "korean", "ko",
5108 "norwegian", "no",
5109 "polish", "pl",
5110 "russian", "ru",
5111 "slovak", "sk",
5112 "swedish", "sv",
5113 "ukrainian", "uk",
5114 "chinese_china", "zh_CN",
5115 "chinese_taiwan", "zh_TW",
5116 NULL};
5117
5118 for (i = 0; mtable[i] != NULL; i += 2)
5119 if (STRNICMP(mtable[i], name, STRLEN(mtable[i])) == 0)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005120 return (char_u *)mtable[i + 1];
Bram Moolenaar071d4272004-06-13 20:20:40 +00005121 return name;
5122}
5123#endif
5124
5125#if defined(FEAT_MULTI_LANG) || defined(PROTO)
5126/*
5127 * Obtain the current messages language. Used to set the default for
5128 * 'helplang'. May return NULL or an empty string.
5129 */
5130 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005131get_mess_lang(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005132{
5133 char_u *p;
5134
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02005135# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00005136# if defined(LC_MESSAGES)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005137 p = get_locale_val(LC_MESSAGES);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005138# else
5139 /* This is necessary for Win32, where LC_MESSAGES is not defined and $LANG
Bram Moolenaar61660ea2006-04-07 21:40:07 +00005140 * may be set to the LCID number. LC_COLLATE is the best guess, LC_TIME
5141 * and LC_MONETARY may be set differently for a Japanese working in the
5142 * US. */
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005143 p = get_locale_val(LC_COLLATE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005144# endif
5145# else
5146 p = mch_getenv((char_u *)"LC_ALL");
5147 if (p == NULL || *p == NUL)
5148 {
5149 p = mch_getenv((char_u *)"LC_MESSAGES");
5150 if (p == NULL || *p == NUL)
5151 p = mch_getenv((char_u *)"LANG");
5152 }
5153# endif
5154# ifdef WIN32
5155 p = gettext_lang(p);
5156# endif
5157 return p;
5158}
5159#endif
5160
Bram Moolenaardef9e822004-12-31 20:58:58 +00005161/* Complicated #if; matches with where get_mess_env() is used below. */
5162#if (defined(FEAT_EVAL) && !((defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
5163 && defined(LC_MESSAGES))) \
5164 || ((defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
5165 && (defined(FEAT_GETTEXT) || defined(FEAT_MBYTE)) \
5166 && !defined(LC_MESSAGES))
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01005167static char_u *get_mess_env(void);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005168
5169/*
5170 * Get the language used for messages from the environment.
5171 */
5172 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005173get_mess_env(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005174{
5175 char_u *p;
5176
5177 p = mch_getenv((char_u *)"LC_ALL");
5178 if (p == NULL || *p == NUL)
5179 {
5180 p = mch_getenv((char_u *)"LC_MESSAGES");
5181 if (p == NULL || *p == NUL)
5182 {
5183 p = mch_getenv((char_u *)"LANG");
5184 if (p != NULL && VIM_ISDIGIT(*p))
5185 p = NULL; /* ignore something like "1043" */
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02005186# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00005187 if (p == NULL || *p == NUL)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005188 p = get_locale_val(LC_CTYPE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005189# endif
5190 }
5191 }
5192 return p;
5193}
5194#endif
5195
5196#if defined(FEAT_EVAL) || defined(PROTO)
5197
5198/*
5199 * Set the "v:lang" variable according to the current locale setting.
5200 * Also do "v:lc_time"and "v:ctype".
5201 */
5202 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005203set_lang_var(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005204{
5205 char_u *loc;
5206
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02005207# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005208 loc = get_locale_val(LC_CTYPE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005209# else
5210 /* setlocale() not supported: use the default value */
5211 loc = (char_u *)"C";
5212# endif
5213 set_vim_var_string(VV_CTYPE, loc, -1);
5214
5215 /* When LC_MESSAGES isn't defined use the value from $LC_MESSAGES, fall
5216 * back to LC_CTYPE if it's empty. */
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02005217# if defined(HAVE_GET_LOCALE_VAL) && defined(LC_MESSAGES)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005218 loc = get_locale_val(LC_MESSAGES);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005219# else
5220 loc = get_mess_env();
5221# endif
5222 set_vim_var_string(VV_LANG, loc, -1);
5223
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02005224# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005225 loc = get_locale_val(LC_TIME);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005226# endif
5227 set_vim_var_string(VV_LC_TIME, loc, -1);
5228}
5229#endif
5230
5231#if (defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
5232 && (defined(FEAT_GETTEXT) || defined(FEAT_MBYTE))
5233/*
5234 * ":language": Set the language (locale).
5235 */
5236 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005237ex_language(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005238{
5239 char *loc;
5240 char_u *p;
5241 char_u *name;
5242 int what = LC_ALL;
5243 char *whatstr = "";
5244#ifdef LC_MESSAGES
5245# define VIM_LC_MESSAGES LC_MESSAGES
5246#else
5247# define VIM_LC_MESSAGES 6789
5248#endif
5249
5250 name = eap->arg;
5251
5252 /* Check for "messages {name}", "ctype {name}" or "time {name}" argument.
5253 * Allow abbreviation, but require at least 3 characters to avoid
5254 * confusion with a two letter language name "me" or "ct". */
5255 p = skiptowhite(eap->arg);
Bram Moolenaar1c465442017-03-12 20:10:05 +01005256 if ((*p == NUL || VIM_ISWHITE(*p)) && p - eap->arg >= 3)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005257 {
5258 if (STRNICMP(eap->arg, "messages", p - eap->arg) == 0)
5259 {
5260 what = VIM_LC_MESSAGES;
5261 name = skipwhite(p);
5262 whatstr = "messages ";
5263 }
5264 else if (STRNICMP(eap->arg, "ctype", p - eap->arg) == 0)
5265 {
5266 what = LC_CTYPE;
5267 name = skipwhite(p);
5268 whatstr = "ctype ";
5269 }
5270 else if (STRNICMP(eap->arg, "time", p - eap->arg) == 0)
5271 {
5272 what = LC_TIME;
5273 name = skipwhite(p);
5274 whatstr = "time ";
5275 }
5276 }
5277
5278 if (*name == NUL)
5279 {
5280#ifndef LC_MESSAGES
5281 if (what == VIM_LC_MESSAGES)
5282 p = get_mess_env();
5283 else
5284#endif
5285 p = (char_u *)setlocale(what, NULL);
5286 if (p == NULL || *p == NUL)
5287 p = (char_u *)"Unknown";
5288 smsg((char_u *)_("Current %slanguage: \"%s\""), whatstr, p);
5289 }
5290 else
5291 {
5292#ifndef LC_MESSAGES
5293 if (what == VIM_LC_MESSAGES)
5294 loc = "";
5295 else
5296#endif
Bram Moolenaar8c8de832008-06-24 22:58:06 +00005297 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00005298 loc = setlocale(what, (char *)name);
Bram Moolenaar8c8de832008-06-24 22:58:06 +00005299#if defined(FEAT_FLOAT) && defined(LC_NUMERIC)
5300 /* Make sure strtod() uses a decimal point, not a comma. */
5301 setlocale(LC_NUMERIC, "C");
5302#endif
5303 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00005304 if (loc == NULL)
5305 EMSG2(_("E197: Cannot set language to \"%s\""), name);
5306 else
5307 {
5308#ifdef HAVE_NL_MSG_CAT_CNTR
5309 /* Need to do this for GNU gettext, otherwise cached translations
5310 * will be used again. */
5311 extern int _nl_msg_cat_cntr;
5312
5313 ++_nl_msg_cat_cntr;
5314#endif
Bram Moolenaarb8017e72007-05-10 18:59:07 +00005315 /* Reset $LC_ALL, otherwise it would overrule everything. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005316 vim_setenv((char_u *)"LC_ALL", (char_u *)"");
5317
5318 if (what != LC_TIME)
5319 {
5320 /* Tell gettext() what to translate to. It apparently doesn't
5321 * use the currently effective locale. Also do this when
5322 * FEAT_GETTEXT isn't defined, so that shell commands use this
5323 * value. */
5324 if (what == LC_ALL)
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00005325 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00005326 vim_setenv((char_u *)"LANG", name);
Bram Moolenaare3a0b532013-06-28 20:36:30 +02005327
5328 /* Clear $LANGUAGE because GNU gettext uses it. */
5329 vim_setenv((char_u *)"LANGUAGE", (char_u *)"");
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00005330# ifdef WIN32
5331 /* Apparently MS-Windows printf() may cause a crash when
5332 * we give it 8-bit text while it's expecting text in the
5333 * current locale. This call avoids that. */
5334 setlocale(LC_CTYPE, "C");
5335# endif
5336 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00005337 if (what != LC_CTYPE)
5338 {
5339 char_u *mname;
5340#ifdef WIN32
5341 mname = gettext_lang(name);
5342#else
5343 mname = name;
5344#endif
5345 vim_setenv((char_u *)"LC_MESSAGES", mname);
5346#ifdef FEAT_MULTI_LANG
5347 set_helplang_default(mname);
5348#endif
5349 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00005350 }
5351
5352# ifdef FEAT_EVAL
5353 /* Set v:lang, v:lc_time and v:ctype to the final result. */
5354 set_lang_var();
5355# endif
Bram Moolenaar6cc00c72011-10-20 21:41:09 +02005356# ifdef FEAT_TITLE
5357 maketitle();
5358# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00005359 }
5360 }
5361}
5362
5363# if defined(FEAT_CMDL_COMPL) || defined(PROTO)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005364
5365static char_u **locales = NULL; /* Array of all available locales */
Bram Moolenaarb8f7bd62017-01-12 20:28:25 +01005366
5367# ifndef WIN32
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005368static int did_init_locales = FALSE;
5369
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005370/* Return an array of strings for all available locales + NULL for the
5371 * last element. Return NULL in case of error. */
5372 static char_u **
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005373find_locales(void)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005374{
5375 garray_T locales_ga;
5376 char_u *loc;
5377
5378 /* Find all available locales by running command "locale -a". If this
5379 * doesn't work we won't have completion. */
5380 char_u *locale_a = get_cmd_output((char_u *)"locale -a",
Bram Moolenaar39c29ed2014-04-05 19:44:40 +02005381 NULL, SHELL_SILENT, NULL);
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005382 if (locale_a == NULL)
5383 return NULL;
5384 ga_init2(&locales_ga, sizeof(char_u *), 20);
5385
5386 /* Transform locale_a string where each locale is separated by "\n"
5387 * into an array of locale strings. */
5388 loc = (char_u *)strtok((char *)locale_a, "\n");
5389
5390 while (loc != NULL)
5391 {
5392 if (ga_grow(&locales_ga, 1) == FAIL)
5393 break;
5394 loc = vim_strsave(loc);
5395 if (loc == NULL)
5396 break;
5397
5398 ((char_u **)locales_ga.ga_data)[locales_ga.ga_len++] = loc;
5399 loc = (char_u *)strtok(NULL, "\n");
5400 }
5401 vim_free(locale_a);
5402 if (ga_grow(&locales_ga, 1) == FAIL)
5403 {
5404 ga_clear(&locales_ga);
5405 return NULL;
5406 }
5407 ((char_u **)locales_ga.ga_data)[locales_ga.ga_len] = NULL;
5408 return (char_u **)locales_ga.ga_data;
5409}
Bram Moolenaarb8f7bd62017-01-12 20:28:25 +01005410# endif
5411
5412/*
5413 * Lazy initialization of all available locales.
5414 */
5415 static void
5416init_locales(void)
5417{
5418# ifndef WIN32
5419 if (!did_init_locales)
5420 {
5421 did_init_locales = TRUE;
5422 locales = find_locales();
5423 }
5424# endif
5425}
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005426
5427# if defined(EXITFREE) || defined(PROTO)
5428 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005429free_locales(void)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005430{
5431 int i;
5432 if (locales != NULL)
5433 {
5434 for (i = 0; locales[i] != NULL; i++)
5435 vim_free(locales[i]);
5436 vim_free(locales);
5437 locales = NULL;
5438 }
5439}
5440# endif
5441
Bram Moolenaar071d4272004-06-13 20:20:40 +00005442/*
5443 * Function given to ExpandGeneric() to obtain the possible arguments of the
5444 * ":language" command.
5445 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005446 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005447get_lang_arg(expand_T *xp UNUSED, int idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005448{
5449 if (idx == 0)
5450 return (char_u *)"messages";
5451 if (idx == 1)
5452 return (char_u *)"ctype";
5453 if (idx == 2)
5454 return (char_u *)"time";
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005455
5456 init_locales();
5457 if (locales == NULL)
5458 return NULL;
5459 return locales[idx - 3];
5460}
5461
5462/*
5463 * Function given to ExpandGeneric() to obtain the available locales.
5464 */
5465 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005466get_locales(expand_T *xp UNUSED, int idx)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005467{
5468 init_locales();
5469 if (locales == NULL)
5470 return NULL;
5471 return locales[idx];
Bram Moolenaar071d4272004-06-13 20:20:40 +00005472}
5473# endif
5474
5475#endif