blob: 2b07352df34d94d4a3410fcd3fe8548f078783ac [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 Moolenaare723c422017-09-06 23:40:10 +02001222 int save_did_emsg = did_emsg;
1223 int save_called_emsg = called_emsg;
Bram Moolenaar02e177d2017-08-26 23:43:28 +02001224 int save_must_redraw = must_redraw;
Bram Moolenaare723c422017-09-06 23:40:10 +02001225 int save_trylevel = trylevel;
1226 int save_did_throw = did_throw;
Bram Moolenaarf5291f32017-09-14 22:55:37 +02001227 int save_ex_pressedreturn = get_pressedreturn();
Bram Moolenaare723c422017-09-06 23:40:10 +02001228 except_T *save_current_exception = current_exception;
Bram Moolenaar1e8e1452017-06-24 16:03:06 +02001229
Bram Moolenaare723c422017-09-06 23:40:10 +02001230 /* Create a scope for running the timer callback, ignoring most of
1231 * the current scope, such as being inside a try/catch. */
Bram Moolenaar1e8e1452017-06-24 16:03:06 +02001232 timer_busy = timer_busy > 0 || vgetc_busy > 0;
1233 vgetc_busy = 0;
Bram Moolenaarc577d812017-07-08 22:37:34 +02001234 called_emsg = FALSE;
Bram Moolenaare723c422017-09-06 23:40:10 +02001235 did_emsg = FALSE;
1236 did_uncaught_emsg = FALSE;
Bram Moolenaar02e177d2017-08-26 23:43:28 +02001237 must_redraw = 0;
Bram Moolenaare723c422017-09-06 23:40:10 +02001238 trylevel = 0;
1239 did_throw = FALSE;
1240 current_exception = NULL;
1241
Bram Moolenaar75537a92016-09-05 22:45:28 +02001242 timer->tr_firing = TRUE;
1243 timer_callback(timer);
1244 timer->tr_firing = FALSE;
Bram Moolenaare723c422017-09-06 23:40:10 +02001245
Bram Moolenaar75537a92016-09-05 22:45:28 +02001246 timer_next = timer->tr_next;
1247 did_one = TRUE;
Bram Moolenaar1e8e1452017-06-24 16:03:06 +02001248 timer_busy = save_timer_busy;
1249 vgetc_busy = save_vgetc_busy;
Bram Moolenaare723c422017-09-06 23:40:10 +02001250 if (did_uncaught_emsg)
Bram Moolenaarc577d812017-07-08 22:37:34 +02001251 ++timer->tr_emsg_count;
Bram Moolenaare723c422017-09-06 23:40:10 +02001252 did_emsg = save_did_emsg;
1253 called_emsg = save_called_emsg;
1254 trylevel = save_trylevel;
1255 did_throw = save_did_throw;
1256 current_exception = save_current_exception;
Bram Moolenaar02e177d2017-08-26 23:43:28 +02001257 if (must_redraw != 0)
1258 need_update_screen = TRUE;
1259 must_redraw = must_redraw > save_must_redraw
1260 ? must_redraw : save_must_redraw;
Bram Moolenaarf5291f32017-09-14 22:55:37 +02001261 set_pressedreturn(save_ex_pressedreturn);
Bram Moolenaar75537a92016-09-05 22:45:28 +02001262
1263 /* Only fire the timer again if it repeats and stop_timer() wasn't
1264 * called while inside the callback (tr_id == -1). */
Bram Moolenaarc577d812017-07-08 22:37:34 +02001265 if (timer->tr_repeat != 0 && timer->tr_id != -1
1266 && timer->tr_emsg_count < 3)
Bram Moolenaar75537a92016-09-05 22:45:28 +02001267 {
1268 profile_setlimit(timer->tr_interval, &timer->tr_due);
1269 this_due = GET_TIMEDIFF(timer, now);
1270 if (this_due < 1)
1271 this_due = 1;
1272 if (timer->tr_repeat > 0)
1273 --timer->tr_repeat;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001274 }
Bram Moolenaar75537a92016-09-05 22:45:28 +02001275 else
1276 {
1277 this_due = -1;
1278 remove_timer(timer);
1279 free_timer(timer);
1280 }
Bram Moolenaar975b5272016-03-15 23:10:59 +01001281 }
Bram Moolenaar75537a92016-09-05 22:45:28 +02001282 if (this_due > 0 && (next_due == -1 || next_due > this_due))
1283 next_due = this_due;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001284 }
1285
1286 if (did_one)
Bram Moolenaar02e177d2017-08-26 23:43:28 +02001287 redraw_after_callback(need_update_screen);
Bram Moolenaar975b5272016-03-15 23:10:59 +01001288
Bram Moolenaar75537a92016-09-05 22:45:28 +02001289 return current_id != last_timer_id ? 1 : next_due;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001290}
1291
1292/*
1293 * Find a timer by ID. Returns NULL if not found;
1294 */
1295 timer_T *
Bram Moolenaar75537a92016-09-05 22:45:28 +02001296find_timer(long id)
Bram Moolenaar975b5272016-03-15 23:10:59 +01001297{
1298 timer_T *timer;
1299
Bram Moolenaar75537a92016-09-05 22:45:28 +02001300 if (id >= 0)
1301 {
1302 for (timer = first_timer; timer != NULL; timer = timer->tr_next)
1303 if (timer->tr_id == id)
1304 return timer;
1305 }
1306 return NULL;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001307}
1308
1309
1310/*
1311 * Stop a timer and delete it.
1312 */
1313 void
1314stop_timer(timer_T *timer)
1315{
Bram Moolenaar75537a92016-09-05 22:45:28 +02001316 if (timer->tr_firing)
1317 /* Free the timer after the callback returns. */
1318 timer->tr_id = -1;
1319 else
1320 {
1321 remove_timer(timer);
1322 free_timer(timer);
1323 }
Bram Moolenaar975b5272016-03-15 23:10:59 +01001324}
Bram Moolenaare3188e22016-05-31 21:13:04 +02001325
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001326 void
Bram Moolenaarb73598e2016-08-07 18:22:53 +02001327stop_all_timers(void)
1328{
Bram Moolenaar75537a92016-09-05 22:45:28 +02001329 timer_T *timer;
1330 timer_T *timer_next;
1331
1332 for (timer = first_timer; timer != NULL; timer = timer_next)
1333 {
1334 timer_next = timer->tr_next;
1335 stop_timer(timer);
1336 }
Bram Moolenaarb73598e2016-08-07 18:22:53 +02001337}
1338
1339 void
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001340add_timer_info(typval_T *rettv, timer_T *timer)
1341{
1342 list_T *list = rettv->vval.v_list;
1343 dict_T *dict = dict_alloc();
1344 dictitem_T *di;
1345 long remaining;
1346 proftime_T now;
Bram Moolenaar00ff3802016-08-06 22:27:28 +02001347# ifdef WIN3264
1348 LARGE_INTEGER fr;
1349#endif
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001350
1351 if (dict == NULL)
1352 return;
1353 list_append_dict(list, dict);
1354
Bram Moolenaar75537a92016-09-05 22:45:28 +02001355 dict_add_nr_str(dict, "id", timer->tr_id, NULL);
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001356 dict_add_nr_str(dict, "time", (long)timer->tr_interval, NULL);
1357
1358 profile_start(&now);
1359# ifdef WIN3264
Bram Moolenaar00ff3802016-08-06 22:27:28 +02001360 QueryPerformanceFrequency(&fr);
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001361# endif
Bram Moolenaar75537a92016-09-05 22:45:28 +02001362 remaining = GET_TIMEDIFF(timer, now);
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001363 dict_add_nr_str(dict, "remaining", (long)remaining, NULL);
1364
1365 dict_add_nr_str(dict, "repeat",
1366 (long)(timer->tr_repeat < 0 ? -1 : timer->tr_repeat + 1), NULL);
Bram Moolenaarb73598e2016-08-07 18:22:53 +02001367 dict_add_nr_str(dict, "paused", (long)(timer->tr_paused), NULL);
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001368
1369 di = dictitem_alloc((char_u *)"callback");
1370 if (di != NULL)
1371 {
1372 if (dict_add(dict, di) == FAIL)
1373 vim_free(di);
1374 else if (timer->tr_partial != NULL)
1375 {
1376 di->di_tv.v_type = VAR_PARTIAL;
1377 di->di_tv.vval.v_partial = timer->tr_partial;
1378 ++timer->tr_partial->pt_refcount;
1379 }
1380 else
1381 {
1382 di->di_tv.v_type = VAR_FUNC;
1383 di->di_tv.vval.v_string = vim_strsave(timer->tr_callback);
1384 }
1385 di->di_tv.v_lock = 0;
1386 }
1387}
1388
1389 void
1390add_timer_info_all(typval_T *rettv)
1391{
1392 timer_T *timer;
1393
1394 for (timer = first_timer; timer != NULL; timer = timer->tr_next)
Bram Moolenaar75537a92016-09-05 22:45:28 +02001395 if (timer->tr_id != -1)
1396 add_timer_info(rettv, timer);
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001397}
1398
Bram Moolenaare3188e22016-05-31 21:13:04 +02001399/*
1400 * Mark references in partials of timers.
1401 */
1402 int
1403set_ref_in_timer(int copyID)
1404{
1405 int abort = FALSE;
1406 timer_T *timer;
1407 typval_T tv;
1408
1409 for (timer = first_timer; timer != NULL; timer = timer->tr_next)
1410 {
Bram Moolenaar1e96d9b2016-07-29 22:15:09 +02001411 if (timer->tr_partial != NULL)
1412 {
1413 tv.v_type = VAR_PARTIAL;
1414 tv.vval.v_partial = timer->tr_partial;
1415 }
1416 else
1417 {
1418 tv.v_type = VAR_FUNC;
1419 tv.vval.v_string = timer->tr_callback;
1420 }
Bram Moolenaare3188e22016-05-31 21:13:04 +02001421 abort = abort || set_ref_in_item(&tv, copyID, NULL, NULL);
1422 }
1423 return abort;
1424}
Bram Moolenaar623e2632016-07-30 22:47:56 +02001425
1426# if defined(EXITFREE) || defined(PROTO)
1427 void
1428timer_free_all()
1429{
1430 timer_T *timer;
1431
1432 while (first_timer != NULL)
1433 {
1434 timer = first_timer;
1435 remove_timer(timer);
1436 free_timer(timer);
1437 }
1438}
1439# endif
Bram Moolenaar975b5272016-03-15 23:10:59 +01001440# endif
1441
Bram Moolenaar8a7f5a22013-06-06 14:01:46 +02001442#if defined(FEAT_SYN_HL) && defined(FEAT_RELTIME) && defined(FEAT_FLOAT)
1443# if defined(HAVE_MATH_H)
1444# include <math.h>
1445# endif
1446
1447/*
1448 * Divide the time "tm" by "count" and store in "tm2".
1449 */
1450 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001451profile_divide(proftime_T *tm, int count, proftime_T *tm2)
Bram Moolenaar8a7f5a22013-06-06 14:01:46 +02001452{
1453 if (count == 0)
1454 profile_zero(tm2);
1455 else
1456 {
1457# ifdef WIN3264
1458 tm2->QuadPart = tm->QuadPart / count;
1459# else
1460 double usec = (tm->tv_sec * 1000000.0 + tm->tv_usec) / count;
1461
1462 tm2->tv_sec = floor(usec / 1000000.0);
Bram Moolenaara2e14fc2013-06-10 20:10:44 +02001463 tm2->tv_usec = vim_round(usec - (tm2->tv_sec * 1000000.0));
Bram Moolenaar8a7f5a22013-06-06 14:01:46 +02001464# endif
1465 }
1466}
1467#endif
1468
Bram Moolenaar76929292008-01-06 19:07:36 +00001469# if defined(FEAT_PROFILE) || defined(PROTO)
1470/*
1471 * Functions for profiling.
1472 */
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01001473static void script_do_profile(scriptitem_T *si);
1474static void script_dump_profile(FILE *fd);
Bram Moolenaar76929292008-01-06 19:07:36 +00001475static proftime_T prof_wait_time;
1476
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001477/*
Bram Moolenaar05159a02005-02-26 23:04:13 +00001478 * Add the time "tm2" to "tm".
1479 */
1480 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001481profile_add(proftime_T *tm, proftime_T *tm2)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001482{
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001483# ifdef WIN3264
1484 tm->QuadPart += tm2->QuadPart;
1485# else
Bram Moolenaar05159a02005-02-26 23:04:13 +00001486 tm->tv_usec += tm2->tv_usec;
1487 tm->tv_sec += tm2->tv_sec;
1488 if (tm->tv_usec >= 1000000)
1489 {
1490 tm->tv_usec -= 1000000;
1491 ++tm->tv_sec;
1492 }
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001493# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00001494}
1495
1496/*
Bram Moolenaar1056d982006-03-09 22:37:52 +00001497 * Add the "self" time from the total time and the children's time.
1498 */
1499 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001500profile_self(proftime_T *self, proftime_T *total, proftime_T *children)
Bram Moolenaar1056d982006-03-09 22:37:52 +00001501{
1502 /* Check that the result won't be negative. Can happen with recursive
1503 * calls. */
1504#ifdef WIN3264
1505 if (total->QuadPart <= children->QuadPart)
1506 return;
1507#else
1508 if (total->tv_sec < children->tv_sec
1509 || (total->tv_sec == children->tv_sec
1510 && total->tv_usec <= children->tv_usec))
1511 return;
1512#endif
1513 profile_add(self, total);
1514 profile_sub(self, children);
1515}
1516
1517/*
Bram Moolenaar05159a02005-02-26 23:04:13 +00001518 * Get the current waittime.
1519 */
1520 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001521profile_get_wait(proftime_T *tm)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001522{
1523 *tm = prof_wait_time;
1524}
1525
1526/*
1527 * Subtract the passed waittime since "tm" from "tma".
1528 */
1529 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001530profile_sub_wait(proftime_T *tm, proftime_T *tma)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001531{
1532 proftime_T tm3 = prof_wait_time;
1533
1534 profile_sub(&tm3, tm);
1535 profile_sub(tma, &tm3);
1536}
1537
1538/*
1539 * Return TRUE if "tm1" and "tm2" are equal.
1540 */
1541 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001542profile_equal(proftime_T *tm1, proftime_T *tm2)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001543{
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001544# ifdef WIN3264
1545 return (tm1->QuadPart == tm2->QuadPart);
1546# else
Bram Moolenaar05159a02005-02-26 23:04:13 +00001547 return (tm1->tv_usec == tm2->tv_usec && tm1->tv_sec == tm2->tv_sec);
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001548# endif
1549}
1550
1551/*
1552 * Return <0, 0 or >0 if "tm1" < "tm2", "tm1" == "tm2" or "tm1" > "tm2"
1553 */
1554 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001555profile_cmp(const proftime_T *tm1, const proftime_T *tm2)
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001556{
1557# ifdef WIN3264
1558 return (int)(tm2->QuadPart - tm1->QuadPart);
1559# else
1560 if (tm1->tv_sec == tm2->tv_sec)
1561 return tm2->tv_usec - tm1->tv_usec;
1562 return tm2->tv_sec - tm1->tv_sec;
1563# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00001564}
1565
Bram Moolenaar05159a02005-02-26 23:04:13 +00001566static char_u *profile_fname = NULL;
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001567static proftime_T pause_time;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001568
1569/*
1570 * ":profile cmd args"
1571 */
1572 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001573ex_profile(exarg_T *eap)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001574{
1575 char_u *e;
1576 int len;
1577
1578 e = skiptowhite(eap->arg);
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00001579 len = (int)(e - eap->arg);
Bram Moolenaar05159a02005-02-26 23:04:13 +00001580 e = skipwhite(e);
1581
1582 if (len == 5 && STRNCMP(eap->arg, "start", 5) == 0 && *e != NUL)
1583 {
1584 vim_free(profile_fname);
Bram Moolenaard94682f2015-04-13 15:37:56 +02001585 profile_fname = expand_env_save_opt(e, TRUE);
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001586 do_profiling = PROF_YES;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001587 profile_zero(&prof_wait_time);
1588 set_vim_var_nr(VV_PROFILING, 1L);
1589 }
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001590 else if (do_profiling == PROF_NONE)
Bram Moolenaar54c1b492010-02-24 14:01:28 +01001591 EMSG(_("E750: First use \":profile start {fname}\""));
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001592 else if (STRCMP(eap->arg, "pause") == 0)
1593 {
1594 if (do_profiling == PROF_YES)
1595 profile_start(&pause_time);
1596 do_profiling = PROF_PAUSED;
1597 }
1598 else if (STRCMP(eap->arg, "continue") == 0)
1599 {
1600 if (do_profiling == PROF_PAUSED)
1601 {
1602 profile_end(&pause_time);
1603 profile_add(&prof_wait_time, &pause_time);
1604 }
1605 do_profiling = PROF_YES;
1606 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00001607 else
1608 {
1609 /* The rest is similar to ":breakadd". */
1610 ex_breakadd(eap);
1611 }
1612}
1613
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001614/* Command line expansion for :profile. */
1615static enum
1616{
1617 PEXP_SUBCMD, /* expand :profile sub-commands */
Bram Moolenaar49789dc2011-02-25 14:46:09 +01001618 PEXP_FUNC /* expand :profile func {funcname} */
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001619} pexpand_what;
1620
1621static char *pexpand_cmds[] = {
1622 "start",
1623#define PROFCMD_START 0
1624 "pause",
1625#define PROFCMD_PAUSE 1
1626 "continue",
1627#define PROFCMD_CONTINUE 2
1628 "func",
1629#define PROFCMD_FUNC 3
1630 "file",
1631#define PROFCMD_FILE 4
1632 NULL
1633#define PROFCMD_LAST 5
1634};
1635
1636/*
1637 * Function given to ExpandGeneric() to obtain the profile command
1638 * specific expansion.
1639 */
1640 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001641get_profile_name(expand_T *xp UNUSED, int idx)
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001642{
1643 switch (pexpand_what)
1644 {
1645 case PEXP_SUBCMD:
1646 return (char_u *)pexpand_cmds[idx];
1647 /* case PEXP_FUNC: TODO */
1648 default:
1649 return NULL;
1650 }
1651}
1652
1653/*
1654 * Handle command line completion for :profile command.
1655 */
1656 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001657set_context_in_profile_cmd(expand_T *xp, char_u *arg)
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001658{
1659 char_u *end_subcmd;
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001660
1661 /* Default: expand subcommands. */
1662 xp->xp_context = EXPAND_PROFILE;
1663 pexpand_what = PEXP_SUBCMD;
1664 xp->xp_pattern = arg;
1665
1666 end_subcmd = skiptowhite(arg);
1667 if (*end_subcmd == NUL)
1668 return;
1669
Bram Moolenaar8b9c05f2010-03-02 17:54:33 +01001670 if (end_subcmd - arg == 5 && STRNCMP(arg, "start", 5) == 0)
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001671 {
1672 xp->xp_context = EXPAND_FILES;
1673 xp->xp_pattern = skipwhite(end_subcmd);
1674 return;
1675 }
1676
1677 /* TODO: expand function names after "func" */
1678 xp->xp_context = EXPAND_NOTHING;
1679}
1680
Bram Moolenaar05159a02005-02-26 23:04:13 +00001681/*
1682 * Dump the profiling info.
1683 */
1684 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001685profile_dump(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001686{
1687 FILE *fd;
1688
1689 if (profile_fname != NULL)
1690 {
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +00001691 fd = mch_fopen((char *)profile_fname, "w");
Bram Moolenaar05159a02005-02-26 23:04:13 +00001692 if (fd == NULL)
1693 EMSG2(_(e_notopen), profile_fname);
1694 else
1695 {
Bram Moolenaar05159a02005-02-26 23:04:13 +00001696 script_dump_profile(fd);
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001697 func_dump_profile(fd);
Bram Moolenaar05159a02005-02-26 23:04:13 +00001698 fclose(fd);
1699 }
1700 }
1701}
1702
1703/*
1704 * Start profiling script "fp".
1705 */
1706 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001707script_do_profile(scriptitem_T *si)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001708{
1709 si->sn_pr_count = 0;
1710 profile_zero(&si->sn_pr_total);
1711 profile_zero(&si->sn_pr_self);
1712
1713 ga_init2(&si->sn_prl_ga, sizeof(sn_prl_T), 100);
1714 si->sn_prl_idx = -1;
1715 si->sn_prof_on = TRUE;
1716 si->sn_pr_nest = 0;
1717}
1718
1719/*
1720 * save time when starting to invoke another script or function.
1721 */
1722 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001723script_prof_save(
1724 proftime_T *tm) /* place to store wait time */
Bram Moolenaar05159a02005-02-26 23:04:13 +00001725{
1726 scriptitem_T *si;
1727
1728 if (current_SID > 0 && current_SID <= script_items.ga_len)
1729 {
1730 si = &SCRIPT_ITEM(current_SID);
1731 if (si->sn_prof_on && si->sn_pr_nest++ == 0)
1732 profile_start(&si->sn_pr_child);
1733 }
1734 profile_get_wait(tm);
1735}
1736
1737/*
1738 * Count time spent in children after invoking another script or function.
1739 */
1740 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001741script_prof_restore(proftime_T *tm)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001742{
1743 scriptitem_T *si;
1744
1745 if (current_SID > 0 && current_SID <= script_items.ga_len)
1746 {
1747 si = &SCRIPT_ITEM(current_SID);
1748 if (si->sn_prof_on && --si->sn_pr_nest == 0)
1749 {
1750 profile_end(&si->sn_pr_child);
1751 profile_sub_wait(tm, &si->sn_pr_child); /* don't count wait time */
1752 profile_add(&si->sn_pr_children, &si->sn_pr_child);
1753 profile_add(&si->sn_prl_children, &si->sn_pr_child);
1754 }
1755 }
1756}
1757
1758static proftime_T inchar_time;
1759
1760/*
1761 * Called when starting to wait for the user to type a character.
1762 */
1763 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001764prof_inchar_enter(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001765{
1766 profile_start(&inchar_time);
1767}
1768
1769/*
1770 * Called when finished waiting for the user to type a character.
1771 */
1772 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001773prof_inchar_exit(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001774{
1775 profile_end(&inchar_time);
1776 profile_add(&prof_wait_time, &inchar_time);
1777}
1778
1779/*
1780 * Dump the profiling results for all scripts in file "fd".
1781 */
1782 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001783script_dump_profile(FILE *fd)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001784{
1785 int id;
1786 scriptitem_T *si;
1787 int i;
1788 FILE *sfd;
1789 sn_prl_T *pp;
1790
1791 for (id = 1; id <= script_items.ga_len; ++id)
1792 {
1793 si = &SCRIPT_ITEM(id);
1794 if (si->sn_prof_on)
1795 {
1796 fprintf(fd, "SCRIPT %s\n", si->sn_name);
1797 if (si->sn_pr_count == 1)
1798 fprintf(fd, "Sourced 1 time\n");
1799 else
1800 fprintf(fd, "Sourced %d times\n", si->sn_pr_count);
1801 fprintf(fd, "Total time: %s\n", profile_msg(&si->sn_pr_total));
1802 fprintf(fd, " Self time: %s\n", profile_msg(&si->sn_pr_self));
1803 fprintf(fd, "\n");
1804 fprintf(fd, "count total (s) self (s)\n");
1805
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +00001806 sfd = mch_fopen((char *)si->sn_name, "r");
Bram Moolenaar05159a02005-02-26 23:04:13 +00001807 if (sfd == NULL)
1808 fprintf(fd, "Cannot open file!\n");
1809 else
1810 {
1811 for (i = 0; i < si->sn_prl_ga.ga_len; ++i)
1812 {
1813 if (vim_fgets(IObuff, IOSIZE, sfd))
1814 break;
1815 pp = &PRL_ITEM(si, i);
1816 if (pp->snp_count > 0)
1817 {
1818 fprintf(fd, "%5d ", pp->snp_count);
1819 if (profile_equal(&pp->sn_prl_total, &pp->sn_prl_self))
1820 fprintf(fd, " ");
1821 else
1822 fprintf(fd, "%s ", profile_msg(&pp->sn_prl_total));
1823 fprintf(fd, "%s ", profile_msg(&pp->sn_prl_self));
1824 }
1825 else
1826 fprintf(fd, " ");
1827 fprintf(fd, "%s", IObuff);
1828 }
1829 fclose(sfd);
1830 }
1831 fprintf(fd, "\n");
1832 }
1833 }
1834}
1835
1836/*
1837 * Return TRUE when a function defined in the current script should be
1838 * profiled.
1839 */
1840 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001841prof_def_func(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001842{
Bram Moolenaar53180ce2005-07-05 21:48:14 +00001843 if (current_SID > 0)
1844 return SCRIPT_ITEM(current_SID).sn_pr_force;
1845 return FALSE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001846}
1847
1848# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001849#endif
1850
1851/*
1852 * If 'autowrite' option set, try to write the file.
1853 * Careful: autocommands may make "buf" invalid!
1854 *
1855 * return FAIL for failure, OK otherwise
1856 */
1857 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001858autowrite(buf_T *buf, int forceit)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001859{
Bram Moolenaar373154b2007-02-13 05:19:30 +00001860 int r;
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001861 bufref_T bufref;
Bram Moolenaar373154b2007-02-13 05:19:30 +00001862
Bram Moolenaar071d4272004-06-13 20:20:40 +00001863 if (!(p_aw || p_awa) || !p_write
1864#ifdef FEAT_QUICKFIX
Bram Moolenaar373154b2007-02-13 05:19:30 +00001865 /* never autowrite a "nofile" or "nowrite" buffer */
1866 || bt_dontwrite(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001867#endif
Bram Moolenaar373154b2007-02-13 05:19:30 +00001868 || (!forceit && buf->b_p_ro) || buf->b_ffname == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001869 return FAIL;
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001870 set_bufref(&bufref, buf);
Bram Moolenaar373154b2007-02-13 05:19:30 +00001871 r = buf_write_all(buf, forceit);
1872
1873 /* Writing may succeed but the buffer still changed, e.g., when there is a
1874 * conversion error. We do want to return FAIL then. */
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001875 if (bufref_valid(&bufref) && bufIsChanged(buf))
Bram Moolenaar373154b2007-02-13 05:19:30 +00001876 r = FAIL;
1877 return r;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001878}
1879
1880/*
1881 * flush all buffers, except the ones that are readonly
1882 */
1883 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001884autowrite_all(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001885{
1886 buf_T *buf;
1887
1888 if (!(p_aw || p_awa) || !p_write)
1889 return;
Bram Moolenaar29323592016-07-24 22:04:11 +02001890 FOR_ALL_BUFFERS(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001891 if (bufIsChanged(buf) && !buf->b_p_ro)
1892 {
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001893#ifdef FEAT_AUTOCMD
1894 bufref_T bufref;
1895
1896 set_bufref(&bufref, buf);
1897#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001898 (void)buf_write_all(buf, FALSE);
1899#ifdef FEAT_AUTOCMD
1900 /* an autocommand may have deleted the buffer */
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001901 if (!bufref_valid(&bufref))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001902 buf = firstbuf;
1903#endif
1904 }
1905}
1906
1907/*
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001908 * Return TRUE if buffer was changed and cannot be abandoned.
1909 * For flags use the CCGD_ values.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001910 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001911 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001912check_changed(buf_T *buf, int flags)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001913{
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001914 int forceit = (flags & CCGD_FORCEIT);
1915#ifdef FEAT_AUTOCMD
1916 bufref_T bufref;
1917
1918 set_bufref(&bufref, buf);
1919#endif
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001920
Bram Moolenaar071d4272004-06-13 20:20:40 +00001921 if ( !forceit
1922 && bufIsChanged(buf)
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001923 && ((flags & CCGD_MULTWIN) || buf->b_nwindows <= 1)
1924 && (!(flags & CCGD_AW) || autowrite(buf, forceit) == FAIL))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001925 {
1926#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG)
1927 if ((p_confirm || cmdmod.confirm) && p_write)
1928 {
1929 buf_T *buf2;
1930 int count = 0;
1931
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001932 if (flags & CCGD_ALLBUF)
Bram Moolenaar29323592016-07-24 22:04:11 +02001933 FOR_ALL_BUFFERS(buf2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001934 if (bufIsChanged(buf2)
1935 && (buf2->b_ffname != NULL
1936# ifdef FEAT_BROWSE
1937 || cmdmod.browse
1938# endif
1939 ))
1940 ++count;
1941# ifdef FEAT_AUTOCMD
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001942 if (!bufref_valid(&bufref))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001943 /* Autocommand deleted buffer, oops! It's not changed now. */
1944 return FALSE;
1945# endif
1946 dialog_changed(buf, count > 1);
1947# ifdef FEAT_AUTOCMD
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001948 if (!bufref_valid(&bufref))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001949 /* Autocommand deleted buffer, oops! It's not changed now. */
1950 return FALSE;
1951# endif
1952 return bufIsChanged(buf);
1953 }
1954#endif
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001955 if (flags & CCGD_EXCMD)
Bram Moolenaarf5be7cd2017-08-17 16:55:13 +02001956 no_write_message();
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001957 else
Bram Moolenaarf5be7cd2017-08-17 16:55:13 +02001958 no_write_message_nobang();
Bram Moolenaar071d4272004-06-13 20:20:40 +00001959 return TRUE;
1960 }
1961 return FALSE;
1962}
1963
1964#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG) || defined(PROTO)
1965
1966#if defined(FEAT_BROWSE) || defined(PROTO)
1967/*
1968 * When wanting to write a file without a file name, ask the user for a name.
1969 */
1970 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001971browse_save_fname(buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001972{
1973 if (buf->b_fname == NULL)
1974 {
1975 char_u *fname;
1976
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00001977 fname = do_browse(BROWSE_SAVE, (char_u *)_("Save As"),
1978 NULL, NULL, NULL, NULL, buf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001979 if (fname != NULL)
1980 {
1981 if (setfname(buf, fname, NULL, TRUE) == OK)
1982 buf->b_flags |= BF_NOTEDITED;
1983 vim_free(fname);
1984 }
1985 }
1986}
1987#endif
1988
1989/*
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02001990 * Ask the user what to do when abandoning a changed buffer.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001991 * Must check 'write' option first!
1992 */
1993 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001994dialog_changed(
1995 buf_T *buf,
1996 int checkall) /* may abandon all changed buffers */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001997{
Bram Moolenaard9462e32011-04-11 21:35:11 +02001998 char_u buff[DIALOG_MSG_SIZE];
Bram Moolenaar071d4272004-06-13 20:20:40 +00001999 int ret;
2000 buf_T *buf2;
Bram Moolenaar8218f602012-04-25 17:32:18 +02002001 exarg_T ea;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002002
Bram Moolenaar3f9a1ff2017-08-21 22:06:02 +02002003 dialog_msg(buff, _("Save changes to \"%s\"?"), buf->b_fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002004 if (checkall)
2005 ret = vim_dialog_yesnoallcancel(VIM_QUESTION, NULL, buff, 1);
2006 else
2007 ret = vim_dialog_yesnocancel(VIM_QUESTION, NULL, buff, 1);
2008
Bram Moolenaar8218f602012-04-25 17:32:18 +02002009 /* Init ea pseudo-structure, this is needed for the check_overwrite()
2010 * function. */
2011 ea.append = ea.forceit = FALSE;
2012
Bram Moolenaar071d4272004-06-13 20:20:40 +00002013 if (ret == VIM_YES)
2014 {
2015#ifdef FEAT_BROWSE
2016 /* May get file name, when there is none */
2017 browse_save_fname(buf);
2018#endif
Bram Moolenaar8218f602012-04-25 17:32:18 +02002019 if (buf->b_fname != NULL && check_overwrite(&ea, buf,
2020 buf->b_fname, buf->b_ffname, FALSE) == OK)
2021 /* didn't hit Cancel */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002022 (void)buf_write_all(buf, FALSE);
2023 }
2024 else if (ret == VIM_NO)
2025 {
2026 unchanged(buf, TRUE);
2027 }
2028 else if (ret == VIM_ALL)
2029 {
2030 /*
2031 * Write all modified files that can be written.
2032 * Skip readonly buffers, these need to be confirmed
2033 * individually.
2034 */
Bram Moolenaar29323592016-07-24 22:04:11 +02002035 FOR_ALL_BUFFERS(buf2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002036 {
2037 if (bufIsChanged(buf2)
2038 && (buf2->b_ffname != NULL
2039#ifdef FEAT_BROWSE
2040 || cmdmod.browse
2041#endif
2042 )
2043 && !buf2->b_p_ro)
2044 {
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002045#ifdef FEAT_AUTOCMD
2046 bufref_T bufref;
2047
2048 set_bufref(&bufref, buf2);
2049#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002050#ifdef FEAT_BROWSE
2051 /* May get file name, when there is none */
2052 browse_save_fname(buf2);
2053#endif
Bram Moolenaar8218f602012-04-25 17:32:18 +02002054 if (buf2->b_fname != NULL && check_overwrite(&ea, buf2,
2055 buf2->b_fname, buf2->b_ffname, FALSE) == OK)
2056 /* didn't hit Cancel */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002057 (void)buf_write_all(buf2, FALSE);
2058#ifdef FEAT_AUTOCMD
2059 /* an autocommand may have deleted the buffer */
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002060 if (!bufref_valid(&bufref))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002061 buf2 = firstbuf;
2062#endif
2063 }
2064 }
2065 }
2066 else if (ret == VIM_DISCARDALL)
2067 {
2068 /*
2069 * mark all buffers as unchanged
2070 */
Bram Moolenaar29323592016-07-24 22:04:11 +02002071 FOR_ALL_BUFFERS(buf2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002072 unchanged(buf2, TRUE);
2073 }
2074}
2075#endif
2076
2077/*
2078 * Return TRUE if the buffer "buf" can be abandoned, either by making it
2079 * hidden, autowriting it or unloading it.
2080 */
2081 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002082can_abandon(buf_T *buf, int forceit)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002083{
Bram Moolenaareb44a682017-08-03 22:44:55 +02002084 return ( buf_hide(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002085 || !bufIsChanged(buf)
2086 || buf->b_nwindows > 1
2087 || autowrite(buf, forceit) == OK
2088 || forceit);
2089}
2090
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01002091static void add_bufnum(int *bufnrs, int *bufnump, int nr);
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002092
2093/*
2094 * Add a buffer number to "bufnrs", unless it's already there.
2095 */
2096 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002097add_bufnum(int *bufnrs, int *bufnump, int nr)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002098{
2099 int i;
2100
2101 for (i = 0; i < *bufnump; ++i)
2102 if (bufnrs[i] == nr)
2103 return;
2104 bufnrs[*bufnump] = nr;
2105 *bufnump = *bufnump + 1;
2106}
2107
Bram Moolenaar071d4272004-06-13 20:20:40 +00002108/*
2109 * Return TRUE if any buffer was changed and cannot be abandoned.
2110 * That changed buffer becomes the current buffer.
Bram Moolenaar027387f2016-01-02 22:25:52 +01002111 * When "unload" is true the current buffer is unloaded instead of making it
2112 * hidden. This is used for ":q!".
Bram Moolenaar071d4272004-06-13 20:20:40 +00002113 */
2114 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002115check_changed_any(
2116 int hidden, /* Only check hidden buffers */
2117 int unload)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002118{
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002119 int ret = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002120 buf_T *buf;
2121 int save;
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002122 int i;
2123 int bufnum = 0;
2124 int bufcount = 0;
2125 int *bufnrs;
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002126 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002127 win_T *wp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002128
Bram Moolenaar29323592016-07-24 22:04:11 +02002129 FOR_ALL_BUFFERS(buf)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002130 ++bufcount;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002131
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002132 if (bufcount == 0)
2133 return FALSE;
2134
2135 bufnrs = (int *)alloc(sizeof(int) * bufcount);
2136 if (bufnrs == NULL)
2137 return FALSE;
2138
2139 /* curbuf */
2140 bufnrs[bufnum++] = curbuf->b_fnum;
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002141 /* buf in curtab */
2142 FOR_ALL_WINDOWS(wp)
2143 if (wp->w_buffer != curbuf)
2144 add_bufnum(bufnrs, &bufnum, wp->w_buffer->b_fnum);
2145
2146 /* buf in other tab */
Bram Moolenaar29323592016-07-24 22:04:11 +02002147 FOR_ALL_TABPAGES(tp)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002148 if (tp != curtab)
2149 for (wp = tp->tp_firstwin; wp != NULL; wp = wp->w_next)
2150 add_bufnum(bufnrs, &bufnum, wp->w_buffer->b_fnum);
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002151 /* any other buf */
Bram Moolenaar29323592016-07-24 22:04:11 +02002152 FOR_ALL_BUFFERS(buf)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002153 add_bufnum(bufnrs, &bufnum, buf->b_fnum);
2154
2155 for (i = 0; i < bufnum; ++i)
2156 {
2157 buf = buflist_findnr(bufnrs[i]);
2158 if (buf == NULL)
2159 continue;
2160 if ((!hidden || buf->b_nwindows == 0) && bufIsChanged(buf))
2161 {
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002162 bufref_T bufref;
2163
2164 set_bufref(&bufref, buf);
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002165 /* Try auto-writing the buffer. If this fails but the buffer no
2166 * longer exists it's not changed, that's OK. */
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002167 if (check_changed(buf, (p_awa ? CCGD_AW : 0)
2168 | CCGD_MULTWIN
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002169 | CCGD_ALLBUF) && bufref_valid(&bufref))
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002170 break; /* didn't save - still changes */
2171 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002172 }
2173
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002174 if (i >= bufnum)
2175 goto theend;
2176
2177 ret = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002178 exiting = FALSE;
2179#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG)
2180 /*
2181 * When ":confirm" used, don't give an error message.
2182 */
2183 if (!(p_confirm || cmdmod.confirm))
2184#endif
2185 {
2186 /* There must be a wait_return for this message, do_buffer()
2187 * may cause a redraw. But wait_return() is a no-op when vgetc()
2188 * is busy (Quit used from window menu), then make sure we don't
2189 * cause a scroll up. */
Bram Moolenaar61660ea2006-04-07 21:40:07 +00002190 if (vgetc_busy > 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002191 {
2192 msg_row = cmdline_row;
2193 msg_col = 0;
2194 msg_didout = FALSE;
2195 }
Bram Moolenaareb44a682017-08-03 22:44:55 +02002196 if (
2197#ifdef FEAT_TERMINAL
2198 term_job_running(buf->b_term)
2199 ? EMSG2(_("E947: Job still running in buffer \"%s\""),
2200 buf->b_fname)
2201 :
2202#endif
2203 EMSG2(_("E162: No write since last change for buffer \"%s\""),
Bram Moolenaare1704ba2012-10-03 18:25:00 +02002204 buf_spname(buf) != NULL ? buf_spname(buf) : buf->b_fname))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002205 {
2206 save = no_wait_return;
2207 no_wait_return = FALSE;
2208 wait_return(FALSE);
2209 no_wait_return = save;
2210 }
2211 }
2212
Bram Moolenaar071d4272004-06-13 20:20:40 +00002213 /* Try to find a window that contains the buffer. */
2214 if (buf != curbuf)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002215 FOR_ALL_TAB_WINDOWS(tp, wp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002216 if (wp->w_buffer == buf)
2217 {
Bram Moolenaar4033c552017-09-16 20:54:51 +02002218#ifdef FEAT_AUTOCMD
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002219 bufref_T bufref;
2220
2221 set_bufref(&bufref, buf);
Bram Moolenaar4033c552017-09-16 20:54:51 +02002222#endif
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002223 goto_tabpage_win(tp, wp);
Bram Moolenaar4033c552017-09-16 20:54:51 +02002224#ifdef FEAT_AUTOCMD
Bram Moolenaar071d4272004-06-13 20:20:40 +00002225 /* Paranoia: did autocms wipe out the buffer with changes? */
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002226 if (!bufref_valid(&bufref))
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002227 {
2228 goto theend;
2229 }
Bram Moolenaar4033c552017-09-16 20:54:51 +02002230#endif
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002231 goto buf_found;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002232 }
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002233buf_found:
Bram Moolenaar071d4272004-06-13 20:20:40 +00002234
2235 /* Open the changed buffer in the current window. */
2236 if (buf != curbuf)
Bram Moolenaar027387f2016-01-02 22:25:52 +01002237 set_curbuf(buf, unload ? DOBUF_UNLOAD : DOBUF_GOTO);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002238
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002239theend:
2240 vim_free(bufnrs);
2241 return ret;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002242}
2243
2244/*
2245 * return FAIL if there is no file name, OK if there is one
2246 * give error message for FAIL
2247 */
2248 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002249check_fname(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002250{
2251 if (curbuf->b_ffname == NULL)
2252 {
2253 EMSG(_(e_noname));
2254 return FAIL;
2255 }
2256 return OK;
2257}
2258
2259/*
2260 * flush the contents of a buffer, unless it has no file name
2261 *
2262 * return FAIL for failure, OK otherwise
2263 */
2264 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002265buf_write_all(buf_T *buf, int forceit)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002266{
2267 int retval;
2268#ifdef FEAT_AUTOCMD
2269 buf_T *old_curbuf = curbuf;
2270#endif
2271
2272 retval = (buf_write(buf, buf->b_ffname, buf->b_fname,
2273 (linenr_T)1, buf->b_ml.ml_line_count, NULL,
2274 FALSE, forceit, TRUE, FALSE));
2275#ifdef FEAT_AUTOCMD
2276 if (curbuf != old_curbuf)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00002277 {
Bram Moolenaar8820b482017-03-16 17:23:31 +01002278 msg_source(HL_ATTR(HLF_W));
Bram Moolenaar071d4272004-06-13 20:20:40 +00002279 MSG(_("Warning: Entered other buffer unexpectedly (check autocommands)"));
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00002280 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002281#endif
2282 return retval;
2283}
2284
2285/*
2286 * Code to handle the argument list.
2287 */
2288
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01002289static char_u *do_one_arg(char_u *str);
2290static int do_arglist(char_u *str, int what, int after);
2291static void alist_check_arg_idx(void);
2292static int editing_arg_idx(win_T *win);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002293#ifdef FEAT_LISTCMDS
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01002294static int alist_add_list(int count, char_u **files, int after);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002295#endif
2296#define AL_SET 1
2297#define AL_ADD 2
2298#define AL_DEL 3
2299
Bram Moolenaar071d4272004-06-13 20:20:40 +00002300/*
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002301 * Isolate one argument, taking backticks.
2302 * Changes the argument in-place, puts a NUL after it. Backticks remain.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002303 * Return a pointer to the start of the next argument.
2304 */
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002305 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002306do_one_arg(char_u *str)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002307{
2308 char_u *p;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002309 int inbacktick;
2310
Bram Moolenaar071d4272004-06-13 20:20:40 +00002311 inbacktick = FALSE;
2312 for (p = str; *str; ++str)
2313 {
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002314 /* When the backslash is used for escaping the special meaning of a
2315 * character we need to keep it until wildcard expansion. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002316 if (rem_backslash(str))
2317 {
2318 *p++ = *str++;
2319 *p++ = *str;
2320 }
2321 else
2322 {
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002323 /* An item ends at a space not in backticks */
2324 if (!inbacktick && vim_isspace(*str))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002325 break;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002326 if (*str == '`')
Bram Moolenaar071d4272004-06-13 20:20:40 +00002327 inbacktick ^= TRUE;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002328 *p++ = *str;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002329 }
2330 }
2331 str = skipwhite(str);
2332 *p = NUL;
2333
2334 return str;
2335}
2336
Bram Moolenaar86b68352004-12-27 21:59:20 +00002337/*
2338 * Separate the arguments in "str" and return a list of pointers in the
2339 * growarray "gap".
2340 */
Bram Moolenaar398ee732017-08-03 14:29:14 +02002341 static int
2342get_arglist(garray_T *gap, char_u *str, int escaped)
Bram Moolenaar86b68352004-12-27 21:59:20 +00002343{
2344 ga_init2(gap, (int)sizeof(char_u *), 20);
2345 while (*str != NUL)
2346 {
2347 if (ga_grow(gap, 1) == FAIL)
2348 {
2349 ga_clear(gap);
2350 return FAIL;
2351 }
2352 ((char_u **)gap->ga_data)[gap->ga_len++] = str;
2353
Bram Moolenaar398ee732017-08-03 14:29:14 +02002354 /* If str is escaped, don't handle backslashes or spaces */
2355 if (!escaped)
2356 return OK;
2357
Bram Moolenaar86b68352004-12-27 21:59:20 +00002358 /* Isolate one argument, change it in-place, put a NUL after it. */
2359 str = do_one_arg(str);
2360 }
2361 return OK;
2362}
2363
Bram Moolenaar7df351e2006-01-23 22:30:28 +00002364#if defined(FEAT_QUICKFIX) || defined(FEAT_SYN_HL) || defined(PROTO)
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002365/*
2366 * Parse a list of arguments (file names), expand them and return in
Bram Moolenaar8f5c6f02012-06-29 12:57:06 +02002367 * "fnames[fcountp]". When "wig" is TRUE, removes files matching 'wildignore'.
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002368 * Return FAIL or OK.
2369 */
2370 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002371get_arglist_exp(
2372 char_u *str,
2373 int *fcountp,
2374 char_u ***fnamesp,
2375 int wig)
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002376{
2377 garray_T ga;
2378 int i;
2379
Bram Moolenaar398ee732017-08-03 14:29:14 +02002380 if (get_arglist(&ga, str, TRUE) == FAIL)
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002381 return FAIL;
Bram Moolenaar8f5c6f02012-06-29 12:57:06 +02002382 if (wig == TRUE)
2383 i = expand_wildcards(ga.ga_len, (char_u **)ga.ga_data,
2384 fcountp, fnamesp, EW_FILE|EW_NOTFOUND);
2385 else
2386 i = gen_expand_wildcards(ga.ga_len, (char_u **)ga.ga_data,
2387 fcountp, fnamesp, EW_FILE|EW_NOTFOUND);
2388
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002389 ga_clear(&ga);
2390 return i;
2391}
2392#endif
2393
Bram Moolenaar071d4272004-06-13 20:20:40 +00002394#if defined(FEAT_GUI) || defined(FEAT_CLIENTSERVER) || defined(PROTO)
2395/*
2396 * Redefine the argument list.
2397 */
2398 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002399set_arglist(char_u *str)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002400{
2401 do_arglist(str, AL_SET, 0);
2402}
2403#endif
2404
2405/*
2406 * "what" == AL_SET: Redefine the argument list to 'str'.
2407 * "what" == AL_ADD: add files in 'str' to the argument list after "after".
2408 * "what" == AL_DEL: remove files in 'str' from the argument list.
2409 *
2410 * Return FAIL for failure, OK otherwise.
2411 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002412 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002413do_arglist(
2414 char_u *str,
Bram Moolenaarf1d25012016-03-03 12:22:53 +01002415 int what,
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002416 int after UNUSED) /* 0 means before first one */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002417{
2418 garray_T new_ga;
2419 int exp_count;
2420 char_u **exp_files;
2421 int i;
2422#ifdef FEAT_LISTCMDS
2423 char_u *p;
2424 int match;
2425#endif
Bram Moolenaar398ee732017-08-03 14:29:14 +02002426 int arg_escaped = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002427
2428 /*
Bram Moolenaar2faa29f2016-01-23 23:02:34 +01002429 * Set default argument for ":argadd" command.
2430 */
2431 if (what == AL_ADD && *str == NUL)
2432 {
2433 if (curbuf->b_ffname == NULL)
2434 return FAIL;
2435 str = curbuf->b_fname;
Bram Moolenaar398ee732017-08-03 14:29:14 +02002436 arg_escaped = FALSE;
Bram Moolenaar2faa29f2016-01-23 23:02:34 +01002437 }
2438
2439 /*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002440 * Collect all file name arguments in "new_ga".
2441 */
Bram Moolenaar398ee732017-08-03 14:29:14 +02002442 if (get_arglist(&new_ga, str, arg_escaped) == FAIL)
Bram Moolenaar86b68352004-12-27 21:59:20 +00002443 return FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002444
2445#ifdef FEAT_LISTCMDS
2446 if (what == AL_DEL)
2447 {
2448 regmatch_T regmatch;
2449 int didone;
2450
2451 /*
2452 * Delete the items: use each item as a regexp and find a match in the
2453 * argument list.
2454 */
Bram Moolenaar71afbfe2013-03-19 16:49:16 +01002455 regmatch.rm_ic = p_fic; /* ignore case when 'fileignorecase' is set */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002456 for (i = 0; i < new_ga.ga_len && !got_int; ++i)
2457 {
2458 p = ((char_u **)new_ga.ga_data)[i];
2459 p = file_pat_to_reg_pat(p, NULL, NULL, FALSE);
2460 if (p == NULL)
2461 break;
2462 regmatch.regprog = vim_regcomp(p, p_magic ? RE_MAGIC : 0);
2463 if (regmatch.regprog == NULL)
2464 {
2465 vim_free(p);
2466 break;
2467 }
2468
2469 didone = FALSE;
2470 for (match = 0; match < ARGCOUNT; ++match)
2471 if (vim_regexec(&regmatch, alist_name(&ARGLIST[match]),
2472 (colnr_T)0))
2473 {
2474 didone = TRUE;
2475 vim_free(ARGLIST[match].ae_fname);
2476 mch_memmove(ARGLIST + match, ARGLIST + match + 1,
2477 (ARGCOUNT - match - 1) * sizeof(aentry_T));
2478 --ALIST(curwin)->al_ga.ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002479 if (curwin->w_arg_idx > match)
2480 --curwin->w_arg_idx;
2481 --match;
2482 }
2483
Bram Moolenaar473de612013-06-08 18:19:48 +02002484 vim_regfree(regmatch.regprog);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002485 vim_free(p);
2486 if (!didone)
2487 EMSG2(_(e_nomatch2), ((char_u **)new_ga.ga_data)[i]);
2488 }
2489 ga_clear(&new_ga);
2490 }
2491 else
2492#endif
2493 {
2494 i = expand_wildcards(new_ga.ga_len, (char_u **)new_ga.ga_data,
2495 &exp_count, &exp_files, EW_DIR|EW_FILE|EW_ADDSLASH|EW_NOTFOUND);
2496 ga_clear(&new_ga);
Bram Moolenaar2db5c3b2016-01-16 22:49:34 +01002497 if (i == FAIL || exp_count == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002498 {
2499 EMSG(_(e_nomatch));
2500 return FAIL;
2501 }
2502
2503#ifdef FEAT_LISTCMDS
2504 if (what == AL_ADD)
2505 {
2506 (void)alist_add_list(exp_count, exp_files, after);
2507 vim_free(exp_files);
2508 }
2509 else /* what == AL_SET */
2510#endif
Bram Moolenaar86b68352004-12-27 21:59:20 +00002511 alist_set(ALIST(curwin), exp_count, exp_files, FALSE, NULL, 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002512 }
2513
2514 alist_check_arg_idx();
2515
2516 return OK;
2517}
2518
2519/*
2520 * Check the validity of the arg_idx for each other window.
2521 */
2522 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002523alist_check_arg_idx(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002524{
Bram Moolenaar071d4272004-06-13 20:20:40 +00002525 win_T *win;
Bram Moolenaarf740b292006-02-16 22:11:02 +00002526 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002527
Bram Moolenaarf740b292006-02-16 22:11:02 +00002528 FOR_ALL_TAB_WINDOWS(tp, win)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002529 if (win->w_alist == curwin->w_alist)
2530 check_arg_idx(win);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002531}
2532
2533/*
Bram Moolenaarb8ff1fb2012-02-04 21:59:01 +01002534 * Return TRUE if window "win" is editing the file at the current argument
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002535 * index.
2536 */
2537 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002538editing_arg_idx(win_T *win)
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002539{
2540 return !(win->w_arg_idx >= WARGCOUNT(win)
2541 || (win->w_buffer->b_fnum
2542 != WARGLIST(win)[win->w_arg_idx].ae_fnum
2543 && (win->w_buffer->b_ffname == NULL
2544 || !(fullpathcmp(
2545 alist_name(&WARGLIST(win)[win->w_arg_idx]),
2546 win->w_buffer->b_ffname, TRUE) & FPC_SAME))));
2547}
2548
2549/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002550 * Check if window "win" is editing the w_arg_idx file in its argument list.
2551 */
2552 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002553check_arg_idx(win_T *win)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002554{
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002555 if (WARGCOUNT(win) > 1 && !editing_arg_idx(win))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002556 {
2557 /* We are not editing the current entry in the argument list.
2558 * Set "arg_had_last" if we are editing the last one. */
2559 win->w_arg_idx_invalid = TRUE;
2560 if (win->w_arg_idx != WARGCOUNT(win) - 1
2561 && arg_had_last == FALSE
Bram Moolenaar071d4272004-06-13 20:20:40 +00002562 && ALIST(win) == &global_alist
Bram Moolenaar071d4272004-06-13 20:20:40 +00002563 && 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
Bram Moolenaar4033c552017-09-16 20:54:51 +02002577 && win->w_alist == &global_alist)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002578 arg_had_last = TRUE;
2579 }
2580}
2581
2582/*
2583 * ":args", ":argslocal" and ":argsglobal".
2584 */
2585 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002586ex_args(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002587{
2588 int i;
2589
2590 if (eap->cmdidx != CMD_args)
2591 {
Bram Moolenaar4033c552017-09-16 20:54:51 +02002592#if defined(FEAT_LISTCMDS)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002593 alist_unlink(ALIST(curwin));
2594 if (eap->cmdidx == CMD_argglobal)
2595 ALIST(curwin) = &global_alist;
2596 else /* eap->cmdidx == CMD_arglocal */
2597 alist_new();
2598#else
2599 ex_ni(eap);
2600 return;
2601#endif
2602 }
2603
2604 if (!ends_excmd(*eap->arg))
2605 {
2606 /*
2607 * ":args file ..": define new argument list, handle like ":next"
2608 * Also for ":argslocal file .." and ":argsglobal file ..".
2609 */
2610 ex_next(eap);
2611 }
2612 else
Bram Moolenaar4033c552017-09-16 20:54:51 +02002613#if defined(FEAT_LISTCMDS)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002614 if (eap->cmdidx == CMD_args)
2615#endif
2616 {
2617 /*
2618 * ":args": list arguments.
2619 */
2620 if (ARGCOUNT > 0)
2621 {
2622 /* Overwrite the command, for a short list there is no scrolling
2623 * required and no wait_return(). */
2624 gotocmdline(TRUE);
2625 for (i = 0; i < ARGCOUNT; ++i)
2626 {
2627 if (i == curwin->w_arg_idx)
2628 msg_putchar('[');
2629 msg_outtrans(alist_name(&ARGLIST[i]));
2630 if (i == curwin->w_arg_idx)
2631 msg_putchar(']');
2632 msg_putchar(' ');
2633 }
2634 }
2635 }
Bram Moolenaar4033c552017-09-16 20:54:51 +02002636#if defined(FEAT_LISTCMDS)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002637 else if (eap->cmdidx == CMD_arglocal)
2638 {
2639 garray_T *gap = &curwin->w_alist->al_ga;
2640
2641 /*
2642 * ":argslocal": make a local copy of the global argument list.
2643 */
2644 if (ga_grow(gap, GARGCOUNT) == OK)
2645 for (i = 0; i < GARGCOUNT; ++i)
2646 if (GARGLIST[i].ae_fname != NULL)
2647 {
2648 AARGLIST(curwin->w_alist)[gap->ga_len].ae_fname =
2649 vim_strsave(GARGLIST[i].ae_fname);
2650 AARGLIST(curwin->w_alist)[gap->ga_len].ae_fnum =
2651 GARGLIST[i].ae_fnum;
2652 ++gap->ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002653 }
2654 }
2655#endif
2656}
2657
2658/*
2659 * ":previous", ":sprevious", ":Next" and ":sNext".
2660 */
2661 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002662ex_previous(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002663{
2664 /* If past the last one already, go to the last one. */
2665 if (curwin->w_arg_idx - (int)eap->line2 >= ARGCOUNT)
2666 do_argfile(eap, ARGCOUNT - 1);
2667 else
2668 do_argfile(eap, curwin->w_arg_idx - (int)eap->line2);
2669}
2670
2671/*
2672 * ":rewind", ":first", ":sfirst" and ":srewind".
2673 */
2674 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002675ex_rewind(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002676{
2677 do_argfile(eap, 0);
2678}
2679
2680/*
2681 * ":last" and ":slast".
2682 */
2683 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002684ex_last(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002685{
2686 do_argfile(eap, ARGCOUNT - 1);
2687}
2688
2689/*
2690 * ":argument" and ":sargument".
2691 */
2692 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002693ex_argument(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002694{
2695 int i;
2696
2697 if (eap->addr_count > 0)
2698 i = eap->line2 - 1;
2699 else
2700 i = curwin->w_arg_idx;
2701 do_argfile(eap, i);
2702}
2703
2704/*
2705 * Edit file "argn" of the argument lists.
2706 */
2707 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002708do_argfile(exarg_T *eap, int argn)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002709{
2710 int other;
2711 char_u *p;
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002712 int old_arg_idx = curwin->w_arg_idx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002713
2714 if (argn < 0 || argn >= ARGCOUNT)
2715 {
2716 if (ARGCOUNT <= 1)
2717 EMSG(_("E163: There is only one file to edit"));
2718 else if (argn < 0)
2719 EMSG(_("E164: Cannot go before first file"));
2720 else
2721 EMSG(_("E165: Cannot go beyond last file"));
2722 }
2723 else
2724 {
2725 setpcmark();
2726#ifdef FEAT_GUI
2727 need_mouse_correct = TRUE;
2728#endif
2729
Bram Moolenaardf1bdc92006-02-23 21:32:16 +00002730 /* split window or create new tab page first */
2731 if (*eap->cmd == 's' || cmdmod.tab != 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002732 {
2733 if (win_split(0, 0) == FAIL)
2734 return;
Bram Moolenaar3368ea22010-09-21 16:56:35 +02002735 RESET_BINDING(curwin);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002736 }
2737 else
Bram Moolenaar071d4272004-06-13 20:20:40 +00002738 {
2739 /*
2740 * if 'hidden' set, only check for changed file when re-editing
2741 * the same buffer
2742 */
2743 other = TRUE;
Bram Moolenaareb44a682017-08-03 22:44:55 +02002744 if (buf_hide(curbuf))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002745 {
2746 p = fix_fname(alist_name(&ARGLIST[argn]));
2747 other = otherfile(p);
2748 vim_free(p);
2749 }
Bram Moolenaareb44a682017-08-03 22:44:55 +02002750 if ((!buf_hide(curbuf) || !other)
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002751 && check_changed(curbuf, CCGD_AW
2752 | (other ? 0 : CCGD_MULTWIN)
2753 | (eap->forceit ? CCGD_FORCEIT : 0)
2754 | CCGD_EXCMD))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002755 return;
2756 }
2757
2758 curwin->w_arg_idx = argn;
Bram Moolenaar4033c552017-09-16 20:54:51 +02002759 if (argn == ARGCOUNT - 1 && curwin->w_alist == &global_alist)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002760 arg_had_last = TRUE;
2761
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002762 /* Edit the file; always use the last known line number.
2763 * When it fails (e.g. Abort for already edited file) restore the
2764 * argument index. */
2765 if (do_ecmd(0, alist_name(&ARGLIST[curwin->w_arg_idx]), NULL,
Bram Moolenaar071d4272004-06-13 20:20:40 +00002766 eap, ECMD_LAST,
Bram Moolenaareb44a682017-08-03 22:44:55 +02002767 (buf_hide(curwin->w_buffer) ? ECMD_HIDE : 0)
Bram Moolenaar701f7af2008-11-15 13:12:07 +00002768 + (eap->forceit ? ECMD_FORCEIT : 0), curwin) == FAIL)
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002769 curwin->w_arg_idx = old_arg_idx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002770 /* like Vi: set the mark where the cursor is in the file. */
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002771 else if (eap->cmdidx != CMD_argdo)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002772 setmark('\'');
2773 }
2774}
2775
2776/*
2777 * ":next", and commands that behave like it.
2778 */
2779 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002780ex_next(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002781{
2782 int i;
2783
2784 /*
2785 * check for changed buffer now, if this fails the argument list is not
2786 * redefined.
2787 */
Bram Moolenaareb44a682017-08-03 22:44:55 +02002788 if ( buf_hide(curbuf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002789 || eap->cmdidx == CMD_snext
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002790 || !check_changed(curbuf, CCGD_AW
2791 | (eap->forceit ? CCGD_FORCEIT : 0)
2792 | CCGD_EXCMD))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002793 {
2794 if (*eap->arg != NUL) /* redefine file list */
2795 {
2796 if (do_arglist(eap->arg, AL_SET, 0) == FAIL)
2797 return;
2798 i = 0;
2799 }
2800 else
2801 i = curwin->w_arg_idx + (int)eap->line2;
2802 do_argfile(eap, i);
2803 }
2804}
2805
Bram Moolenaar0106e3d2016-02-23 18:55:43 +01002806#if defined(FEAT_LISTCMDS) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002807/*
2808 * ":argedit"
2809 */
2810 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002811ex_argedit(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002812{
Bram Moolenaar90305c62017-07-16 15:31:17 +02002813 int i = eap->addr_count ? (int)eap->line2 : curwin->w_arg_idx + 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002814
Bram Moolenaar90305c62017-07-16 15:31:17 +02002815 if (do_arglist(eap->arg, AL_ADD, i) == FAIL)
2816 return;
2817#ifdef FEAT_TITLE
2818 maketitle();
2819#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002820
Bram Moolenaar90305c62017-07-16 15:31:17 +02002821 if (curwin->w_arg_idx == 0 && (curbuf->b_ml.ml_flags & ML_EMPTY)
2822 && curbuf->b_ffname == NULL)
2823 i = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002824 /* Edit the argument. */
Bram Moolenaar90305c62017-07-16 15:31:17 +02002825 if (i < ARGCOUNT)
2826 do_argfile(eap, i);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002827}
2828
2829/*
2830 * ":argadd"
2831 */
2832 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002833ex_argadd(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002834{
2835 do_arglist(eap->arg, AL_ADD,
2836 eap->addr_count > 0 ? (int)eap->line2 : curwin->w_arg_idx + 1);
2837#ifdef FEAT_TITLE
2838 maketitle();
2839#endif
2840}
2841
2842/*
2843 * ":argdelete"
2844 */
2845 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002846ex_argdelete(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002847{
2848 int i;
2849 int n;
2850
2851 if (eap->addr_count > 0)
2852 {
2853 /* ":1,4argdel": Delete all arguments in the range. */
2854 if (eap->line2 > ARGCOUNT)
2855 eap->line2 = ARGCOUNT;
2856 n = eap->line2 - eap->line1 + 1;
Bram Moolenaar69a92fb2017-03-09 15:58:30 +01002857 if (*eap->arg != NUL)
2858 /* Can't have both a range and an argument. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002859 EMSG(_(e_invarg));
Bram Moolenaar69a92fb2017-03-09 15:58:30 +01002860 else if (n <= 0)
2861 {
2862 /* Don't give an error for ":%argdel" if the list is empty. */
2863 if (eap->line1 != 1 || eap->line2 != 0)
2864 EMSG(_(e_invrange));
2865 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002866 else
2867 {
2868 for (i = eap->line1; i <= eap->line2; ++i)
2869 vim_free(ARGLIST[i - 1].ae_fname);
2870 mch_memmove(ARGLIST + eap->line1 - 1, ARGLIST + eap->line2,
2871 (size_t)((ARGCOUNT - eap->line2) * sizeof(aentry_T)));
2872 ALIST(curwin)->al_ga.ga_len -= n;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002873 if (curwin->w_arg_idx >= eap->line2)
2874 curwin->w_arg_idx -= n;
2875 else if (curwin->w_arg_idx > eap->line1)
2876 curwin->w_arg_idx = eap->line1;
Bram Moolenaar72defda2016-01-17 18:04:33 +01002877 if (ARGCOUNT == 0)
2878 curwin->w_arg_idx = 0;
2879 else if (curwin->w_arg_idx >= ARGCOUNT)
2880 curwin->w_arg_idx = ARGCOUNT - 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002881 }
2882 }
2883 else if (*eap->arg == NUL)
2884 EMSG(_(e_argreq));
2885 else
2886 do_arglist(eap->arg, AL_DEL, 0);
2887#ifdef FEAT_TITLE
2888 maketitle();
2889#endif
2890}
2891
2892/*
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002893 * ":argdo", ":windo", ":bufdo", ":tabdo", ":cdo", ":ldo", ":cfdo" and ":lfdo"
Bram Moolenaar071d4272004-06-13 20:20:40 +00002894 */
2895 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002896ex_listdo(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002897{
2898 int i;
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002899 win_T *wp;
2900 tabpage_T *tp;
Bram Moolenaare25bb902015-02-27 20:33:37 +01002901 buf_T *buf = curbuf;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002902 int next_fnum = 0;
2903#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
2904 char_u *save_ei = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002905#endif
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002906 char_u *p_shm_save;
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002907#ifdef FEAT_QUICKFIX
Bram Moolenaared84b762015-09-09 22:35:29 +02002908 int qf_size = 0;
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002909 int qf_idx;
2910#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002911
Bram Moolenaar0106e3d2016-02-23 18:55:43 +01002912#ifndef FEAT_QUICKFIX
2913 if (eap->cmdidx == CMD_cdo || eap->cmdidx == CMD_ldo ||
2914 eap->cmdidx == CMD_cfdo || eap->cmdidx == CMD_lfdo)
2915 {
2916 ex_ni(eap);
2917 return;
2918 }
2919#endif
2920
Bram Moolenaar071d4272004-06-13 20:20:40 +00002921#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002922 if (eap->cmdidx != CMD_windo && eap->cmdidx != CMD_tabdo)
Bram Moolenaardcaf10e2005-01-21 11:55:25 +00002923 /* Don't do syntax HL autocommands. Skipping the syntax file is a
2924 * great speed improvement. */
2925 save_ei = au_event_disable(",Syntax");
Bram Moolenaar071d4272004-06-13 20:20:40 +00002926#endif
Bram Moolenaar6b1ee342014-08-06 18:17:11 +02002927#ifdef FEAT_CLIPBOARD
2928 start_global_changes();
2929#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002930
2931 if (eap->cmdidx == CMD_windo
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002932 || eap->cmdidx == CMD_tabdo
Bram Moolenaareb44a682017-08-03 22:44:55 +02002933 || buf_hide(curbuf)
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002934 || !check_changed(curbuf, CCGD_AW
2935 | (eap->forceit ? CCGD_FORCEIT : 0)
2936 | CCGD_EXCMD))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002937 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00002938 i = 0;
Bram Moolenaara162bc52015-01-07 16:54:21 +01002939 /* start at the eap->line1 argument/window/buffer */
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002940 wp = firstwin;
2941 tp = first_tabpage;
Bram Moolenaara162bc52015-01-07 16:54:21 +01002942 switch (eap->cmdidx)
2943 {
Bram Moolenaara162bc52015-01-07 16:54:21 +01002944 case CMD_windo:
2945 for ( ; wp != NULL && i + 1 < eap->line1; wp = wp->w_next)
2946 i++;
2947 break;
2948 case CMD_tabdo:
2949 for( ; tp != NULL && i + 1 < eap->line1; tp = tp->tp_next)
2950 i++;
2951 break;
Bram Moolenaara162bc52015-01-07 16:54:21 +01002952 case CMD_argdo:
2953 i = eap->line1 - 1;
2954 break;
Bram Moolenaara162bc52015-01-07 16:54:21 +01002955 default:
2956 break;
2957 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002958 /* set pcmark now */
2959 if (eap->cmdidx == CMD_bufdo)
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002960 {
Bram Moolenaare25bb902015-02-27 20:33:37 +01002961 /* Advance to the first listed buffer after "eap->line1". */
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002962 for (buf = firstbuf; buf != NULL && (buf->b_fnum < eap->line1
Bram Moolenaare25bb902015-02-27 20:33:37 +01002963 || !buf->b_p_bl); buf = buf->b_next)
2964 if (buf->b_fnum > eap->line2)
2965 {
2966 buf = NULL;
2967 break;
2968 }
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002969 if (buf != NULL)
Bram Moolenaare25bb902015-02-27 20:33:37 +01002970 goto_buffer(eap, DOBUF_FIRST, FORWARD, buf->b_fnum);
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002971 }
2972#ifdef FEAT_QUICKFIX
2973 else if (eap->cmdidx == CMD_cdo || eap->cmdidx == CMD_ldo
2974 || eap->cmdidx == CMD_cfdo || eap->cmdidx == CMD_lfdo)
2975 {
2976 qf_size = qf_get_size(eap);
2977 if (qf_size <= 0 || eap->line1 > qf_size)
2978 buf = NULL;
2979 else
2980 {
2981 ex_cc(eap);
2982
2983 buf = curbuf;
2984 i = eap->line1 - 1;
2985 if (eap->addr_count <= 0)
2986 /* default is all the quickfix/location list entries */
2987 eap->line2 = qf_size;
2988 }
2989 }
2990#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002991 else
2992 setpcmark();
2993 listcmd_busy = TRUE; /* avoids setting pcmark below */
2994
Bram Moolenaare25bb902015-02-27 20:33:37 +01002995 while (!got_int && buf != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002996 {
2997 if (eap->cmdidx == CMD_argdo)
2998 {
2999 /* go to argument "i" */
3000 if (i == ARGCOUNT)
3001 break;
3002 /* Don't call do_argfile() when already there, it will try
3003 * reloading the file. */
Bram Moolenaard4755bb2004-09-02 19:12:26 +00003004 if (curwin->w_arg_idx != i || !editing_arg_idx(curwin))
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003005 {
3006 /* Clear 'shm' to avoid that the file message overwrites
3007 * any output from the command. */
3008 p_shm_save = vim_strsave(p_shm);
3009 set_option_value((char_u *)"shm", 0L, (char_u *)"", 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003010 do_argfile(eap, i);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003011 set_option_value((char_u *)"shm", 0L, p_shm_save, 0);
3012 vim_free(p_shm_save);
3013 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003014 if (curwin->w_arg_idx != i)
3015 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003016 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003017 else if (eap->cmdidx == CMD_windo)
3018 {
Bram Moolenaar32466aa2006-02-24 23:53:04 +00003019 /* go to window "wp" */
3020 if (!win_valid(wp))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003021 break;
Bram Moolenaar32466aa2006-02-24 23:53:04 +00003022 win_goto(wp);
Bram Moolenaar41423242007-05-03 20:11:13 +00003023 if (curwin != wp)
3024 break; /* something must be wrong */
Bram Moolenaar32466aa2006-02-24 23:53:04 +00003025 wp = curwin->w_next;
3026 }
3027 else if (eap->cmdidx == CMD_tabdo)
3028 {
3029 /* go to window "tp" */
3030 if (!valid_tabpage(tp))
3031 break;
Bram Moolenaar49e649f2013-05-06 04:50:35 +02003032 goto_tabpage_tp(tp, TRUE, TRUE);
Bram Moolenaar32466aa2006-02-24 23:53:04 +00003033 tp = tp->tp_next;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003034 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003035 else if (eap->cmdidx == CMD_bufdo)
3036 {
3037 /* Remember the number of the next listed buffer, in case
3038 * ":bwipe" is used or autocommands do something strange. */
3039 next_fnum = -1;
3040 for (buf = curbuf->b_next; buf != NULL; buf = buf->b_next)
3041 if (buf->b_p_bl)
3042 {
3043 next_fnum = buf->b_fnum;
3044 break;
3045 }
3046 }
3047
Bram Moolenaara162bc52015-01-07 16:54:21 +01003048 ++i;
3049
Bram Moolenaar071d4272004-06-13 20:20:40 +00003050 /* execute the command */
3051 do_cmdline(eap->arg, eap->getline, eap->cookie,
3052 DOCMD_VERBOSE + DOCMD_NOWAIT);
3053
3054 if (eap->cmdidx == CMD_bufdo)
3055 {
3056 /* Done? */
Bram Moolenaara162bc52015-01-07 16:54:21 +01003057 if (next_fnum < 0 || next_fnum > eap->line2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003058 break;
3059 /* Check if the buffer still exists. */
Bram Moolenaar29323592016-07-24 22:04:11 +02003060 FOR_ALL_BUFFERS(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003061 if (buf->b_fnum == next_fnum)
3062 break;
3063 if (buf == NULL)
3064 break;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003065
3066 /* Go to the next buffer. Clear 'shm' to avoid that the file
3067 * message overwrites any output from the command. */
3068 p_shm_save = vim_strsave(p_shm);
3069 set_option_value((char_u *)"shm", 0L, (char_u *)"", 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003070 goto_buffer(eap, DOBUF_FIRST, FORWARD, next_fnum);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003071 set_option_value((char_u *)"shm", 0L, p_shm_save, 0);
3072 vim_free(p_shm_save);
3073
Bram Moolenaaraa23b372015-09-08 18:46:31 +02003074 /* If autocommands took us elsewhere, quit here. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003075 if (curbuf->b_fnum != next_fnum)
3076 break;
3077 }
3078
Bram Moolenaaraa23b372015-09-08 18:46:31 +02003079#ifdef FEAT_QUICKFIX
3080 if (eap->cmdidx == CMD_cdo || eap->cmdidx == CMD_ldo
3081 || eap->cmdidx == CMD_cfdo || eap->cmdidx == CMD_lfdo)
3082 {
3083 if (i >= qf_size || i >= eap->line2)
3084 break;
3085
3086 qf_idx = qf_get_cur_idx(eap);
3087
3088 ex_cnext(eap);
3089
3090 /* If jumping to the next quickfix entry fails, quit here */
3091 if (qf_get_cur_idx(eap) == qf_idx)
3092 break;
3093 }
3094#endif
3095
Bram Moolenaar071d4272004-06-13 20:20:40 +00003096 if (eap->cmdidx == CMD_windo)
3097 {
3098 validate_cursor(); /* cursor may have moved */
3099#ifdef FEAT_SCROLLBIND
3100 /* required when 'scrollbind' has been set */
3101 if (curwin->w_p_scb)
3102 do_check_scrollbind(TRUE);
3103#endif
3104 }
Bram Moolenaara162bc52015-01-07 16:54:21 +01003105
Bram Moolenaara162bc52015-01-07 16:54:21 +01003106 if (eap->cmdidx == CMD_windo || eap->cmdidx == CMD_tabdo)
3107 if (i+1 > eap->line2)
3108 break;
Bram Moolenaara162bc52015-01-07 16:54:21 +01003109 if (eap->cmdidx == CMD_argdo && i >= eap->line2)
3110 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003111 }
3112 listcmd_busy = FALSE;
3113 }
3114
3115#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
Bram Moolenaar6ac54292005-02-02 23:07:25 +00003116 if (save_ei != NULL)
3117 {
3118 au_event_restore(save_ei);
3119 apply_autocmds(EVENT_SYNTAX, curbuf->b_p_syn,
3120 curbuf->b_fname, TRUE, curbuf);
3121 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003122#endif
Bram Moolenaar6b1ee342014-08-06 18:17:11 +02003123#ifdef FEAT_CLIPBOARD
3124 end_global_changes();
3125#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003126}
3127
3128/*
3129 * Add files[count] to the arglist of the current window after arg "after".
3130 * The file names in files[count] must have been allocated and are taken over.
3131 * Files[] itself is not taken over.
3132 * Returns index of first added argument. Returns -1 when failed (out of mem).
3133 */
3134 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003135alist_add_list(
3136 int count,
3137 char_u **files,
3138 int after) /* where to add: 0 = before first one */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003139{
3140 int i;
Bram Moolenaara24f0a52016-01-17 19:39:00 +01003141 int old_argcount = ARGCOUNT;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003142
3143 if (ga_grow(&ALIST(curwin)->al_ga, count) == OK)
3144 {
3145 if (after < 0)
3146 after = 0;
3147 if (after > ARGCOUNT)
3148 after = ARGCOUNT;
3149 if (after < ARGCOUNT)
3150 mch_memmove(&(ARGLIST[after + count]), &(ARGLIST[after]),
3151 (ARGCOUNT - after) * sizeof(aentry_T));
3152 for (i = 0; i < count; ++i)
3153 {
3154 ARGLIST[after + i].ae_fname = files[i];
3155 ARGLIST[after + i].ae_fnum = buflist_add(files[i], BLN_LISTED);
3156 }
3157 ALIST(curwin)->al_ga.ga_len += count;
Bram Moolenaara24f0a52016-01-17 19:39:00 +01003158 if (old_argcount > 0 && curwin->w_arg_idx >= after)
3159 curwin->w_arg_idx += count;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003160 return after;
3161 }
3162
3163 for (i = 0; i < count; ++i)
3164 vim_free(files[i]);
3165 return -1;
3166}
3167
3168#endif /* FEAT_LISTCMDS */
3169
3170#ifdef FEAT_EVAL
3171/*
3172 * ":compiler[!] {name}"
3173 */
3174 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003175ex_compiler(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003176{
3177 char_u *buf;
3178 char_u *old_cur_comp = NULL;
3179 char_u *p;
3180
3181 if (*eap->arg == NUL)
3182 {
3183 /* List all compiler scripts. */
3184 do_cmdline_cmd((char_u *)"echo globpath(&rtp, 'compiler/*.vim')");
3185 /* ) keep the indenter happy... */
3186 }
3187 else
3188 {
3189 buf = alloc((unsigned)(STRLEN(eap->arg) + 14));
3190 if (buf != NULL)
3191 {
3192 if (eap->forceit)
3193 {
3194 /* ":compiler! {name}" sets global options */
3195 do_cmdline_cmd((char_u *)
3196 "command -nargs=* CompilerSet set <args>");
3197 }
3198 else
3199 {
3200 /* ":compiler! {name}" sets local options.
3201 * To remain backwards compatible "current_compiler" is always
3202 * used. A user's compiler plugin may set it, the distributed
3203 * plugin will then skip the settings. Afterwards set
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003204 * "b:current_compiler" and restore "current_compiler".
3205 * Explicitly prepend "g:" to make it work in a function. */
3206 old_cur_comp = get_var_value((char_u *)"g:current_compiler");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003207 if (old_cur_comp != NULL)
3208 old_cur_comp = vim_strsave(old_cur_comp);
3209 do_cmdline_cmd((char_u *)
3210 "command -nargs=* CompilerSet setlocal <args>");
3211 }
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003212 do_unlet((char_u *)"g:current_compiler", TRUE);
Bram Moolenaar2ce06f62005-01-31 19:19:04 +00003213 do_unlet((char_u *)"b:current_compiler", TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003214
3215 sprintf((char *)buf, "compiler/%s.vim", eap->arg);
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003216 if (source_runtime(buf, DIP_ALL) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003217 EMSG2(_("E666: compiler not supported: %s"), eap->arg);
3218 vim_free(buf);
3219
3220 do_cmdline_cmd((char_u *)":delcommand CompilerSet");
3221
3222 /* Set "b:current_compiler" from "current_compiler". */
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003223 p = get_var_value((char_u *)"g:current_compiler");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003224 if (p != NULL)
3225 set_internal_string_var((char_u *)"b:current_compiler", p);
3226
3227 /* Restore "current_compiler" for ":compiler {name}". */
3228 if (!eap->forceit)
3229 {
3230 if (old_cur_comp != NULL)
3231 {
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003232 set_internal_string_var((char_u *)"g:current_compiler",
Bram Moolenaar071d4272004-06-13 20:20:40 +00003233 old_cur_comp);
3234 vim_free(old_cur_comp);
3235 }
3236 else
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003237 do_unlet((char_u *)"g:current_compiler", TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003238 }
3239 }
3240 }
3241}
3242#endif
3243
3244/*
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003245 * ":runtime [what] {name}"
Bram Moolenaar071d4272004-06-13 20:20:40 +00003246 */
3247 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003248ex_runtime(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003249{
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003250 char_u *arg = eap->arg;
3251 char_u *p = skiptowhite(arg);
3252 int len = (int)(p - arg);
3253 int flags = eap->forceit ? DIP_ALL : 0;
3254
3255 if (STRNCMP(arg, "START", len) == 0)
3256 {
3257 flags += DIP_START + DIP_NORTP;
3258 arg = skipwhite(arg + len);
3259 }
3260 else if (STRNCMP(arg, "OPT", len) == 0)
3261 {
3262 flags += DIP_OPT + DIP_NORTP;
3263 arg = skipwhite(arg + len);
3264 }
3265 else if (STRNCMP(arg, "PACK", len) == 0)
3266 {
3267 flags += DIP_START + DIP_OPT + DIP_NORTP;
3268 arg = skipwhite(arg + len);
3269 }
3270 else if (STRNCMP(arg, "ALL", len) == 0)
3271 {
3272 flags += DIP_START + DIP_OPT;
3273 arg = skipwhite(arg + len);
3274 }
3275
3276 source_runtime(arg, flags);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003277}
3278
Bram Moolenaar071d4272004-06-13 20:20:40 +00003279 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003280source_callback(char_u *fname, void *cookie UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003281{
Bram Moolenaar910f66f2006-04-05 20:41:53 +00003282 (void)do_source(fname, FALSE, DOSO_NONE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003283}
3284
3285/*
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003286 * Find the file "name" in all directories in "path" and invoke
3287 * "callback(fname, cookie)".
3288 * "name" can contain wildcards.
3289 * When "flags" has DIP_ALL: source all files, otherwise only the first one.
3290 * When "flags" has DIP_DIR: find directories instead of files.
3291 * When "flags" has DIP_ERR: give an error message if there is no match.
3292 *
3293 * return FAIL when no file could be sourced, OK otherwise.
3294 */
Bram Moolenaar6bef5302016-03-12 21:28:26 +01003295 int
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003296do_in_path(
3297 char_u *path,
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003298 char_u *name,
Bram Moolenaar91715872016-03-03 17:13:03 +01003299 int flags,
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003300 void (*callback)(char_u *fname, void *ck),
3301 void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003302{
3303 char_u *rtp;
3304 char_u *np;
3305 char_u *buf;
3306 char_u *rtp_copy;
3307 char_u *tail;
3308 int num_files;
3309 char_u **files;
3310 int i;
3311 int did_one = FALSE;
3312#ifdef AMIGA
3313 struct Process *proc = (struct Process *)FindTask(0L);
3314 APTR save_winptr = proc->pr_WindowPtr;
3315
3316 /* Avoid a requester here for a volume that doesn't exist. */
3317 proc->pr_WindowPtr = (APTR)-1L;
3318#endif
3319
3320 /* Make a copy of 'runtimepath'. Invoking the callback may change the
3321 * value. */
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003322 rtp_copy = vim_strsave(path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003323 buf = alloc(MAXPATHL);
3324 if (buf != NULL && rtp_copy != NULL)
3325 {
Bram Moolenaarc09a6d62013-06-10 21:27:29 +02003326 if (p_verbose > 1 && name != NULL)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003327 {
3328 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00003329 smsg((char_u *)_("Searching for \"%s\" in \"%s\""),
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003330 (char *)name, (char *)path);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003331 verbose_leave();
3332 }
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00003333
Bram Moolenaar071d4272004-06-13 20:20:40 +00003334 /* Loop over all entries in 'runtimepath'. */
3335 rtp = rtp_copy;
Bram Moolenaar91715872016-03-03 17:13:03 +01003336 while (*rtp != NUL && ((flags & DIP_ALL) || !did_one))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003337 {
Bram Moolenaar66459b72016-08-06 19:01:55 +02003338 size_t buflen;
3339
Bram Moolenaar071d4272004-06-13 20:20:40 +00003340 /* Copy the path from 'runtimepath' to buf[]. */
3341 copy_option_part(&rtp, buf, MAXPATHL, ",");
Bram Moolenaar66459b72016-08-06 19:01:55 +02003342 buflen = STRLEN(buf);
3343
3344 /* Skip after or non-after directories. */
3345 if (flags & (DIP_NOAFTER | DIP_AFTER))
3346 {
3347 int is_after = buflen >= 5
3348 && STRCMP(buf + buflen - 5, "after") == 0;
3349
3350 if ((is_after && (flags & DIP_NOAFTER))
3351 || (!is_after && (flags & DIP_AFTER)))
3352 continue;
3353 }
3354
Bram Moolenaarc09a6d62013-06-10 21:27:29 +02003355 if (name == NULL)
3356 {
3357 (*callback)(buf, (void *) &cookie);
3358 if (!did_one)
3359 did_one = (cookie == NULL);
3360 }
Bram Moolenaar66459b72016-08-06 19:01:55 +02003361 else if (buflen + STRLEN(name) + 2 < MAXPATHL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003362 {
3363 add_pathsep(buf);
3364 tail = buf + STRLEN(buf);
3365
3366 /* Loop over all patterns in "name" */
3367 np = name;
Bram Moolenaar91715872016-03-03 17:13:03 +01003368 while (*np != NUL && ((flags & DIP_ALL) || !did_one))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003369 {
3370 /* Append the pattern from "name" to buf[]. */
3371 copy_option_part(&np, tail, (int)(MAXPATHL - (tail - buf)),
3372 "\t ");
3373
3374 if (p_verbose > 2)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003375 {
3376 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00003377 smsg((char_u *)_("Searching for \"%s\""), buf);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003378 verbose_leave();
3379 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003380
3381 /* Expand wildcards, invoke the callback for each match. */
3382 if (gen_expand_wildcards(1, &buf, &num_files, &files,
Bram Moolenaar91715872016-03-03 17:13:03 +01003383 (flags & DIP_DIR) ? EW_DIR : EW_FILE) == OK)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003384 {
3385 for (i = 0; i < num_files; ++i)
3386 {
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00003387 (*callback)(files[i], cookie);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003388 did_one = TRUE;
Bram Moolenaar91715872016-03-03 17:13:03 +01003389 if (!(flags & DIP_ALL))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003390 break;
3391 }
3392 FreeWild(num_files, files);
3393 }
3394 }
3395 }
3396 }
3397 }
3398 vim_free(buf);
3399 vim_free(rtp_copy);
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003400 if (!did_one && name != NULL)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003401 {
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003402 char *basepath = path == p_rtp ? "runtimepath" : "packpath";
3403
3404 if (flags & DIP_ERR)
3405 EMSG3(_(e_dirnotf), basepath, name);
3406 else if (p_verbose > 0)
3407 {
3408 verbose_enter();
3409 smsg((char_u *)_("not found in '%s': \"%s\""), basepath, name);
3410 verbose_leave();
3411 }
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003412 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003413
3414#ifdef AMIGA
3415 proc->pr_WindowPtr = save_winptr;
3416#endif
3417
3418 return did_one ? OK : FAIL;
3419}
3420
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003421/*
Bram Moolenaar07ecfa62017-06-27 14:43:55 +02003422 * Find "name" in "path". When found, invoke the callback function for
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003423 * it: callback(fname, "cookie")
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003424 * When "flags" has DIP_ALL repeat for all matches, otherwise only the first
3425 * one is used.
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003426 * Returns OK when at least one match found, FAIL otherwise.
3427 *
Bram Moolenaar07ecfa62017-06-27 14:43:55 +02003428 * If "name" is NULL calls callback for each entry in "path". Cookie is
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003429 * passed by reference in this case, setting it to NULL indicates that callback
3430 * has done its job.
3431 */
Bram Moolenaar07ecfa62017-06-27 14:43:55 +02003432 static int
3433do_in_path_and_pp(
3434 char_u *path,
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003435 char_u *name,
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003436 int flags,
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003437 void (*callback)(char_u *fname, void *ck),
3438 void *cookie)
3439{
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003440 int done = FAIL;
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003441 char_u *s;
3442 int len;
3443 char *start_dir = "pack/*/start/*/%s";
3444 char *opt_dir = "pack/*/opt/*/%s";
3445
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003446 if ((flags & DIP_NORTP) == 0)
Bram Moolenaar07ecfa62017-06-27 14:43:55 +02003447 done = do_in_path(path, name, flags, callback, cookie);
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003448
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003449 if ((done == FAIL || (flags & DIP_ALL)) && (flags & DIP_START))
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003450 {
Bram Moolenaar1c8b4ed2016-03-17 21:51:03 +01003451 len = (int)(STRLEN(start_dir) + STRLEN(name));
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003452 s = alloc(len);
3453 if (s == NULL)
3454 return FAIL;
3455 vim_snprintf((char *)s, len, start_dir, name);
3456 done = do_in_path(p_pp, s, flags, callback, cookie);
3457 vim_free(s);
3458 }
3459
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003460 if ((done == FAIL || (flags & DIP_ALL)) && (flags & DIP_OPT))
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003461 {
Bram Moolenaar1c8b4ed2016-03-17 21:51:03 +01003462 len = (int)(STRLEN(opt_dir) + STRLEN(name));
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003463 s = alloc(len);
3464 if (s == NULL)
3465 return FAIL;
3466 vim_snprintf((char *)s, len, opt_dir, name);
3467 done = do_in_path(p_pp, s, flags, callback, cookie);
3468 vim_free(s);
3469 }
3470
3471 return done;
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003472}
3473
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003474/*
Bram Moolenaar07ecfa62017-06-27 14:43:55 +02003475 * Just like do_in_path_and_pp(), using 'runtimepath' for "path".
3476 */
3477 int
3478do_in_runtimepath(
3479 char_u *name,
3480 int flags,
3481 void (*callback)(char_u *fname, void *ck),
3482 void *cookie)
3483{
3484 return do_in_path_and_pp(p_rtp, name, flags, callback, cookie);
3485}
3486
3487/*
3488 * Source the file "name" from all directories in 'runtimepath'.
3489 * "name" can contain wildcards.
3490 * When "flags" has DIP_ALL: source all files, otherwise only the first one.
3491 *
3492 * return FAIL when no file could be sourced, OK otherwise.
3493 */
3494 int
3495source_runtime(char_u *name, int flags)
3496{
3497 return source_in_path(p_rtp, name, flags);
3498}
3499
3500/*
3501 * Just like source_runtime(), but use "path" instead of 'runtimepath'.
3502 */
3503 int
3504source_in_path(char_u *path, char_u *name, int flags)
3505{
3506 return do_in_path_and_pp(path, name, flags, source_callback, NULL);
3507}
3508
3509
3510/*
Bram Moolenaarf3654822016-03-04 22:12:23 +01003511 * Expand wildcards in "pat" and invoke do_source() for each match.
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003512 */
3513 static void
Bram Moolenaarf3654822016-03-04 22:12:23 +01003514source_all_matches(char_u *pat)
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003515{
Bram Moolenaarf3654822016-03-04 22:12:23 +01003516 int num_files;
3517 char_u **files;
3518 int i;
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003519
Bram Moolenaarf3654822016-03-04 22:12:23 +01003520 if (gen_expand_wildcards(1, &pat, &num_files, &files, EW_FILE) == OK)
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003521 {
Bram Moolenaarf3654822016-03-04 22:12:23 +01003522 for (i = 0; i < num_files; ++i)
3523 (void)do_source(files[i], FALSE, DOSO_NONE);
3524 FreeWild(num_files, files);
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003525 }
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003526}
3527
Bram Moolenaar49b27322016-04-05 21:13:00 +02003528/* used for "cookie" of add_pack_plugin() */
3529static int APP_ADD_DIR;
3530static int APP_LOAD;
3531static int APP_BOTH;
3532
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003533 static void
Bram Moolenaar91715872016-03-03 17:13:03 +01003534add_pack_plugin(char_u *fname, void *cookie)
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003535{
Bram Moolenaarf3654822016-03-04 22:12:23 +01003536 char_u *p4, *p3, *p2, *p1, *p;
3537 char_u *insp;
Bram Moolenaar91715872016-03-03 17:13:03 +01003538 int c;
3539 char_u *new_rtp;
3540 int keep;
Bram Moolenaarb0550662016-05-31 21:37:36 +02003541 size_t oldlen;
3542 size_t addlen;
Bram Moolenaara5702442016-05-24 19:37:29 +02003543 char_u *afterdir;
Bram Moolenaarb0550662016-05-31 21:37:36 +02003544 size_t afterlen = 0;
Bram Moolenaar91715872016-03-03 17:13:03 +01003545 char_u *ffname = fix_fname(fname);
Bram Moolenaarfef524b2016-07-02 22:07:22 +02003546 size_t fname_len;
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003547 char_u *buf = NULL;
3548 char_u *rtp_ffname;
3549 int match;
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003550
Bram Moolenaar91715872016-03-03 17:13:03 +01003551 if (ffname == NULL)
3552 return;
Bram Moolenaar49b27322016-04-05 21:13:00 +02003553 if (cookie != &APP_LOAD && strstr((char *)p_rtp, (char *)ffname) == NULL)
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003554 {
Bram Moolenaara5702442016-05-24 19:37:29 +02003555 /* directory is not yet in 'runtimepath', add it */
Bram Moolenaarf3654822016-03-04 22:12:23 +01003556 p4 = p3 = p2 = p1 = get_past_head(ffname);
Bram Moolenaar91acfff2017-03-12 19:22:36 +01003557 for (p = p1; *p; MB_PTR_ADV(p))
Bram Moolenaarf3654822016-03-04 22:12:23 +01003558 if (vim_ispathsep_nocolon(*p))
3559 {
3560 p4 = p3; p3 = p2; p2 = p1; p1 = p;
3561 }
3562
3563 /* now we have:
Bram Moolenaaraf1a0e32016-03-09 22:19:26 +01003564 * rtp/pack/name/start/name
3565 * p4 p3 p2 p1
Bram Moolenaarf3654822016-03-04 22:12:23 +01003566 *
3567 * find the part up to "pack" in 'runtimepath' */
3568 c = *p4;
3569 *p4 = NUL;
Bram Moolenaar4c5717e2016-07-01 15:39:40 +02003570
3571 /* Find "ffname" in "p_rtp", ignoring '/' vs '\' differences. */
3572 fname_len = STRLEN(ffname);
3573 insp = p_rtp;
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003574 buf = alloc(MAXPATHL);
3575 if (buf == NULL)
3576 goto theend;
3577 while (*insp != NUL)
Bram Moolenaar4c5717e2016-07-01 15:39:40 +02003578 {
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003579 copy_option_part(&insp, buf, MAXPATHL, ",");
3580 add_pathsep(buf);
3581 rtp_ffname = fix_fname(buf);
3582 if (rtp_ffname == NULL)
3583 goto theend;
3584 match = vim_fnamencmp(rtp_ffname, ffname, fname_len) == 0;
3585 vim_free(rtp_ffname);
3586 if (match)
Bram Moolenaar4c5717e2016-07-01 15:39:40 +02003587 break;
Bram Moolenaar4c5717e2016-07-01 15:39:40 +02003588 }
3589
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003590 if (*insp == NUL)
Bram Moolenaarf3654822016-03-04 22:12:23 +01003591 /* not found, append at the end */
3592 insp = p_rtp + STRLEN(p_rtp);
3593 else
Bram Moolenaarf3654822016-03-04 22:12:23 +01003594 /* append after the matching directory. */
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003595 --insp;
Bram Moolenaarf3654822016-03-04 22:12:23 +01003596 *p4 = c;
3597
Bram Moolenaara5702442016-05-24 19:37:29 +02003598 /* check if rtp/pack/name/start/name/after exists */
3599 afterdir = concat_fnames(ffname, (char_u *)"after", TRUE);
3600 if (afterdir != NULL && mch_isdir(afterdir))
3601 afterlen = STRLEN(afterdir) + 1; /* add one for comma */
3602
Bram Moolenaarb0550662016-05-31 21:37:36 +02003603 oldlen = STRLEN(p_rtp);
3604 addlen = STRLEN(ffname) + 1; /* add one for comma */
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003605 new_rtp = alloc((int)(oldlen + addlen + afterlen + 1));
3606 /* add one for NUL */
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003607 if (new_rtp == NULL)
Bram Moolenaarf3654822016-03-04 22:12:23 +01003608 goto theend;
3609 keep = (int)(insp - p_rtp);
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003610 mch_memmove(new_rtp, p_rtp, keep);
3611 new_rtp[keep] = ',';
Bram Moolenaara5702442016-05-24 19:37:29 +02003612 mch_memmove(new_rtp + keep + 1, ffname, addlen);
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003613 if (p_rtp[keep] != NUL)
Bram Moolenaara5702442016-05-24 19:37:29 +02003614 mch_memmove(new_rtp + keep + addlen, p_rtp + keep,
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003615 oldlen - keep + 1);
Bram Moolenaara5702442016-05-24 19:37:29 +02003616 if (afterlen > 0)
3617 {
3618 STRCAT(new_rtp, ",");
3619 STRCAT(new_rtp, afterdir);
3620 }
Bram Moolenaar863c1a92016-03-03 15:47:06 +01003621 set_option_value((char_u *)"rtp", 0L, new_rtp, 0);
3622 vim_free(new_rtp);
Bram Moolenaara5702442016-05-24 19:37:29 +02003623 vim_free(afterdir);
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003624 }
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003625
Bram Moolenaar49b27322016-04-05 21:13:00 +02003626 if (cookie != &APP_ADD_DIR)
Bram Moolenaarf3654822016-03-04 22:12:23 +01003627 {
Bram Moolenaar71fb0c12016-04-02 22:44:16 +02003628 static char *plugpat = "%s/plugin/**/*.vim";
Bram Moolenaarf3654822016-03-04 22:12:23 +01003629 static char *ftpat = "%s/ftdetect/*.vim";
3630 int len;
3631 char_u *pat;
3632
3633 len = (int)STRLEN(ffname) + (int)STRLEN(ftpat);
3634 pat = alloc(len);
3635 if (pat == NULL)
3636 goto theend;
3637 vim_snprintf((char *)pat, len, plugpat, ffname);
3638 source_all_matches(pat);
3639
3640#ifdef FEAT_AUTOCMD
3641 {
3642 char_u *cmd = vim_strsave((char_u *)"g:did_load_filetypes");
3643
3644 /* If runtime/filetype.vim wasn't loaded yet, the scripts will be
3645 * found when it loads. */
3646 if (cmd != NULL && eval_to_number(cmd) > 0)
3647 {
3648 do_cmdline_cmd((char_u *)"augroup filetypedetect");
3649 vim_snprintf((char *)pat, len, ftpat, ffname);
3650 source_all_matches(pat);
3651 do_cmdline_cmd((char_u *)"augroup END");
3652 }
3653 vim_free(cmd);
3654 }
3655#endif
Bram Moolenaarba8cd122016-03-19 14:16:39 +01003656 vim_free(pat);
Bram Moolenaarf3654822016-03-04 22:12:23 +01003657 }
3658
3659theend:
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003660 vim_free(buf);
Bram Moolenaarf3654822016-03-04 22:12:23 +01003661 vim_free(ffname);
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003662}
3663
Bram Moolenaarce876aa2017-06-04 17:47:42 +02003664/*
3665 * Add all packages in the "start" directory to 'runtimepath'.
3666 */
3667 void
3668add_pack_start_dirs(void)
3669{
3670 do_in_path(p_pp, (char_u *)"pack/*/start/*", DIP_ALL + DIP_DIR,
3671 add_pack_plugin, &APP_ADD_DIR);
3672}
3673
3674/*
3675 * Load plugins from all packages in the "start" directory.
3676 */
3677 void
3678load_start_packages(void)
3679{
3680 did_source_packages = TRUE;
3681 do_in_path(p_pp, (char_u *)"pack/*/start/*", DIP_ALL + DIP_DIR,
3682 add_pack_plugin, &APP_LOAD);
3683}
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003684
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003685/*
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003686 * ":packloadall"
Bram Moolenaarf3654822016-03-04 22:12:23 +01003687 * Find plugins in the package directories and source them.
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003688 */
3689 void
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003690ex_packloadall(exarg_T *eap)
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003691{
Bram Moolenaarce876aa2017-06-04 17:47:42 +02003692 if (!did_source_packages || eap->forceit)
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003693 {
Bram Moolenaar49b27322016-04-05 21:13:00 +02003694 /* First do a round to add all directories to 'runtimepath', then load
3695 * the plugins. This allows for plugins to use an autoload directory
3696 * of another plugin. */
Bram Moolenaarce876aa2017-06-04 17:47:42 +02003697 add_pack_start_dirs();
3698 load_start_packages();
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003699 }
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003700}
3701
3702/*
Bram Moolenaarf3654822016-03-04 22:12:23 +01003703 * ":packadd[!] {name}"
Bram Moolenaar91715872016-03-03 17:13:03 +01003704 */
3705 void
3706ex_packadd(exarg_T *eap)
3707{
3708 static char *plugpat = "pack/*/opt/%s";
3709 int len;
3710 char *pat;
3711
3712 len = (int)STRLEN(plugpat) + (int)STRLEN(eap->arg);
3713 pat = (char *)alloc(len);
3714 if (pat == NULL)
3715 return;
3716 vim_snprintf(pat, len, plugpat, eap->arg);
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003717 do_in_path(p_pp, (char_u *)pat, DIP_ALL + DIP_DIR + DIP_ERR,
Bram Moolenaar49b27322016-04-05 21:13:00 +02003718 add_pack_plugin, eap->forceit ? &APP_ADD_DIR : &APP_BOTH);
Bram Moolenaar91715872016-03-03 17:13:03 +01003719 vim_free(pat);
3720}
3721
Bram Moolenaar071d4272004-06-13 20:20:40 +00003722#if defined(FEAT_EVAL) && defined(FEAT_AUTOCMD)
3723/*
3724 * ":options"
3725 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003726 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003727ex_options(
3728 exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003729{
Bram Moolenaarab6c8582017-08-11 17:15:09 +02003730 vim_setenv((char_u *)"OPTWIN_CMD", (char_u *)(cmdmod.tab ? "tab" : ""));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003731 cmd_source((char_u *)SYS_OPTWIN_FILE, NULL);
3732}
3733#endif
3734
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01003735#if defined(FEAT_PYTHON3) || defined(FEAT_PYTHON) || defined(PROTO)
3736
3737# if (defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)) || defined(PROTO)
3738/*
3739 * Detect Python 3 or 2, and initialize 'pyxversion'.
3740 */
3741 void
3742init_pyxversion(void)
3743{
3744 if (p_pyx == 0)
3745 {
3746 if (python3_enabled(FALSE))
3747 p_pyx = 3;
3748 else if (python_enabled(FALSE))
3749 p_pyx = 2;
3750 }
3751}
3752# endif
3753
3754/*
3755 * Does a file contain one of the following strings at the beginning of any
3756 * line?
3757 * "#!(any string)python2" => returns 2
3758 * "#!(any string)python3" => returns 3
3759 * "# requires python 2.x" => returns 2
3760 * "# requires python 3.x" => returns 3
3761 * otherwise return 0.
3762 */
3763 static int
3764requires_py_version(char_u *filename)
3765{
3766 FILE *file;
3767 int requires_py_version = 0;
3768 int i, lines;
3769
3770 lines = (int)p_mls;
3771 if (lines < 0)
3772 lines = 5;
3773
3774 file = mch_fopen((char *)filename, "r");
3775 if (file != NULL)
3776 {
3777 for (i = 0; i < lines; i++)
3778 {
3779 if (vim_fgets(IObuff, IOSIZE, file))
3780 break;
3781 if (i == 0 && IObuff[0] == '#' && IObuff[1] == '!')
3782 {
3783 /* Check shebang. */
3784 if (strstr((char *)IObuff + 2, "python2") != NULL)
3785 {
3786 requires_py_version = 2;
3787 break;
3788 }
3789 if (strstr((char *)IObuff + 2, "python3") != NULL)
3790 {
3791 requires_py_version = 3;
3792 break;
3793 }
3794 }
3795 IObuff[21] = '\0';
3796 if (STRCMP("# requires python 2.x", IObuff) == 0)
3797 {
3798 requires_py_version = 2;
3799 break;
3800 }
3801 if (STRCMP("# requires python 3.x", IObuff) == 0)
3802 {
3803 requires_py_version = 3;
3804 break;
3805 }
3806 }
3807 fclose(file);
3808 }
3809 return requires_py_version;
3810}
3811
3812
3813/*
3814 * Source a python file using the requested python version.
3815 */
3816 static void
3817source_pyx_file(exarg_T *eap, char_u *fname)
3818{
3819 exarg_T ex;
3820 int v = requires_py_version(fname);
3821
3822# if defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)
3823 init_pyxversion();
3824# endif
3825 if (v == 0)
3826 {
3827# if defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)
3828 /* user didn't choose a preference, 'pyx' is used */
3829 v = p_pyx;
3830# elif defined(FEAT_PYTHON)
3831 v = 2;
3832# elif defined(FEAT_PYTHON3)
3833 v = 3;
3834# endif
3835 }
3836
3837 /*
3838 * now source, if required python version is not supported show
3839 * unobtrusive message.
3840 */
3841 if (eap == NULL)
3842 vim_memset(&ex, 0, sizeof(ex));
3843 else
3844 ex = *eap;
3845 ex.arg = fname;
3846 ex.cmd = (char_u *)(v == 2 ? "pyfile" : "pyfile3");
3847
3848 if (v == 2)
3849 {
3850# ifdef FEAT_PYTHON
3851 ex_pyfile(&ex);
3852# else
3853 vim_snprintf((char *)IObuff, IOSIZE,
3854 _("W20: Required python version 2.x not supported, ignoring file: %s"),
3855 fname);
3856 MSG(IObuff);
3857# endif
3858 return;
3859 }
3860 else
3861 {
3862# ifdef FEAT_PYTHON3
3863 ex_py3file(&ex);
3864# else
3865 vim_snprintf((char *)IObuff, IOSIZE,
3866 _("W21: Required python version 3.x not supported, ignoring file: %s"),
3867 fname);
3868 MSG(IObuff);
3869# endif
3870 return;
3871 }
3872}
3873
3874/*
3875 * ":pyxfile {fname}"
3876 */
3877 void
3878ex_pyxfile(exarg_T *eap)
3879{
3880 source_pyx_file(eap, eap->arg);
3881}
3882
3883/*
3884 * ":pyx"
3885 */
3886 void
3887ex_pyx(exarg_T *eap)
3888{
3889# if defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)
3890 init_pyxversion();
3891 if (p_pyx == 2)
3892 ex_python(eap);
3893 else
3894 ex_py3(eap);
3895# elif defined(FEAT_PYTHON)
3896 ex_python(eap);
3897# elif defined(FEAT_PYTHON3)
3898 ex_py3(eap);
3899# endif
3900}
3901
3902/*
3903 * ":pyxdo"
3904 */
3905 void
3906ex_pyxdo(exarg_T *eap)
3907{
3908# if defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)
3909 init_pyxversion();
3910 if (p_pyx == 2)
3911 ex_pydo(eap);
3912 else
3913 ex_py3do(eap);
3914# elif defined(FEAT_PYTHON)
3915 ex_pydo(eap);
3916# elif defined(FEAT_PYTHON3)
3917 ex_py3do(eap);
3918# endif
3919}
3920
3921#endif
3922
Bram Moolenaar071d4272004-06-13 20:20:40 +00003923/*
3924 * ":source {fname}"
3925 */
3926 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003927ex_source(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003928{
3929#ifdef FEAT_BROWSE
3930 if (cmdmod.browse)
3931 {
3932 char_u *fname = NULL;
3933
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00003934 fname = do_browse(0, (char_u *)_("Source Vim script"), eap->arg,
Bram Moolenaar071d4272004-06-13 20:20:40 +00003935 NULL, NULL, BROWSE_FILTER_MACROS, NULL);
3936 if (fname != NULL)
3937 {
3938 cmd_source(fname, eap);
3939 vim_free(fname);
3940 }
3941 }
3942 else
3943#endif
3944 cmd_source(eap->arg, eap);
3945}
3946
3947 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003948cmd_source(char_u *fname, exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003949{
3950 if (*fname == NUL)
3951 EMSG(_(e_argreq));
3952
Bram Moolenaar071d4272004-06-13 20:20:40 +00003953 else if (eap != NULL && eap->forceit)
Bram Moolenaar84a05ac2013-05-06 04:24:17 +02003954 /* ":source!": read Normal mode commands
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00003955 * Need to execute the commands directly. This is required at least
3956 * for:
Bram Moolenaar071d4272004-06-13 20:20:40 +00003957 * - ":g" command busy
3958 * - after ":argdo", ":windo" or ":bufdo"
3959 * - another command follows
3960 * - inside a loop
3961 */
3962 openscript(fname, global_busy || listcmd_busy || eap->nextcmd != NULL
3963#ifdef FEAT_EVAL
3964 || eap->cstack->cs_idx >= 0
3965#endif
3966 );
3967
3968 /* ":source" read ex commands */
Bram Moolenaar910f66f2006-04-05 20:41:53 +00003969 else if (do_source(fname, FALSE, DOSO_NONE) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003970 EMSG2(_(e_notopen), fname);
3971}
3972
3973/*
3974 * ":source" and associated commands.
3975 */
3976/*
3977 * Structure used to store info for each sourced file.
3978 * It is shared between do_source() and getsourceline().
3979 * This is required, because it needs to be handed to do_cmdline() and
3980 * sourcing can be done recursively.
3981 */
3982struct source_cookie
3983{
3984 FILE *fp; /* opened file for sourcing */
3985 char_u *nextline; /* if not NULL: line that was read ahead */
3986 int finished; /* ":finish" used */
Bram Moolenaar860cae12010-06-05 23:22:07 +02003987#if defined(USE_CRNL) || defined(USE_CR)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003988 int fileformat; /* EOL_UNKNOWN, EOL_UNIX or EOL_DOS */
3989 int error; /* TRUE if LF found after CR-LF */
3990#endif
3991#ifdef FEAT_EVAL
3992 linenr_T breakpoint; /* next line with breakpoint or zero */
3993 char_u *fname; /* name of sourced file */
3994 int dbg_tick; /* debug_tick when breakpoint was set */
3995 int level; /* top nesting level of sourced file */
3996#endif
3997#ifdef FEAT_MBYTE
3998 vimconv_T conv; /* type of conversion */
3999#endif
4000};
4001
4002#ifdef FEAT_EVAL
4003/*
4004 * Return the address holding the next breakpoint line for a source cookie.
4005 */
4006 linenr_T *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004007source_breakpoint(void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004008{
4009 return &((struct source_cookie *)cookie)->breakpoint;
4010}
4011
4012/*
4013 * Return the address holding the debug tick for a source cookie.
4014 */
4015 int *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004016source_dbg_tick(void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004017{
4018 return &((struct source_cookie *)cookie)->dbg_tick;
4019}
4020
4021/*
4022 * Return the nesting level for a source cookie.
4023 */
4024 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004025source_level(void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004026{
4027 return ((struct source_cookie *)cookie)->level;
4028}
4029#endif
4030
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01004031static char_u *get_one_sourceline(struct source_cookie *sp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004032
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004033#if (defined(WIN32) && defined(FEAT_CSCOPE)) || defined(HAVE_FD_CLOEXEC)
4034# define USE_FOPEN_NOINH
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01004035static FILE *fopen_noinh_readbin(char *filename);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004036
4037/*
4038 * Special function to open a file without handle inheritance.
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004039 * When possible the handle is closed on exec().
Bram Moolenaar071d4272004-06-13 20:20:40 +00004040 */
4041 static FILE *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004042fopen_noinh_readbin(char *filename)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004043{
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004044# ifdef WIN32
Bram Moolenaar8d8ef0b2010-01-20 21:41:47 +01004045 int fd_tmp = mch_open(filename, O_RDONLY | O_BINARY | O_NOINHERIT, 0);
4046# else
4047 int fd_tmp = mch_open(filename, O_RDONLY, 0);
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004048# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004049
4050 if (fd_tmp == -1)
4051 return NULL;
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004052
4053# ifdef HAVE_FD_CLOEXEC
4054 {
4055 int fdflags = fcntl(fd_tmp, F_GETFD);
4056 if (fdflags >= 0 && (fdflags & FD_CLOEXEC) == 0)
Bram Moolenaarcde88542015-08-11 19:14:00 +02004057 (void)fcntl(fd_tmp, F_SETFD, fdflags | FD_CLOEXEC);
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004058 }
4059# endif
4060
Bram Moolenaar071d4272004-06-13 20:20:40 +00004061 return fdopen(fd_tmp, READBIN);
4062}
4063#endif
4064
4065
4066/*
4067 * do_source: Read the file "fname" and execute its lines as EX commands.
4068 *
4069 * This function may be called recursively!
4070 *
4071 * return FAIL if file could not be opened, OK otherwise
4072 */
4073 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004074do_source(
4075 char_u *fname,
4076 int check_other, /* check for .vimrc and _vimrc */
4077 int is_vimrc) /* DOSO_ value */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004078{
4079 struct source_cookie cookie;
4080 char_u *save_sourcing_name;
4081 linenr_T save_sourcing_lnum;
4082 char_u *p;
4083 char_u *fname_exp;
Bram Moolenaar73881402009-02-04 16:50:47 +00004084 char_u *firstline = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004085 int retval = FAIL;
4086#ifdef FEAT_EVAL
4087 scid_T save_current_SID;
4088 static scid_T last_current_SID = 0;
4089 void *save_funccalp;
4090 int save_debug_break_level = debug_break_level;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004091 scriptitem_T *si = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004092# ifdef UNIX
Bram Moolenaar8767f522016-07-01 17:17:39 +02004093 stat_T st;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004094 int stat_ok;
4095# endif
4096#endif
4097#ifdef STARTUPTIME
4098 struct timeval tv_rel;
4099 struct timeval tv_start;
4100#endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00004101#ifdef FEAT_PROFILE
4102 proftime_T wait_start;
4103#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004104
Bram Moolenaar071d4272004-06-13 20:20:40 +00004105 p = expand_env_save(fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004106 if (p == NULL)
4107 return retval;
4108 fname_exp = fix_fname(p);
4109 vim_free(p);
4110 if (fname_exp == NULL)
4111 return retval;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004112 if (mch_isdir(fname_exp))
4113 {
Bram Moolenaar555b2802005-05-19 21:08:39 +00004114 smsg((char_u *)_("Cannot source a directory: \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004115 goto theend;
4116 }
4117
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00004118#ifdef FEAT_AUTOCMD
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00004119 /* Apply SourceCmd autocommands, they should get the file and source it. */
4120 if (has_autocmd(EVENT_SOURCECMD, fname_exp, NULL)
4121 && apply_autocmds(EVENT_SOURCECMD, fname_exp, fname_exp,
4122 FALSE, curbuf))
Bram Moolenaarf33943e2008-01-15 21:17:30 +00004123 {
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00004124# ifdef FEAT_EVAL
Bram Moolenaarf33943e2008-01-15 21:17:30 +00004125 retval = aborting() ? FAIL : OK;
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00004126# else
Bram Moolenaarf33943e2008-01-15 21:17:30 +00004127 retval = OK;
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00004128# endif
Bram Moolenaarf33943e2008-01-15 21:17:30 +00004129 goto theend;
4130 }
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00004131
4132 /* Apply SourcePre autocommands, they may get the file. */
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00004133 apply_autocmds(EVENT_SOURCEPRE, fname_exp, fname_exp, FALSE, curbuf);
4134#endif
4135
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004136#ifdef USE_FOPEN_NOINH
Bram Moolenaar071d4272004-06-13 20:20:40 +00004137 cookie.fp = fopen_noinh_readbin((char *)fname_exp);
4138#else
4139 cookie.fp = mch_fopen((char *)fname_exp, READBIN);
4140#endif
4141 if (cookie.fp == NULL && check_other)
4142 {
4143 /*
4144 * Try again, replacing file name ".vimrc" by "_vimrc" or vice versa,
4145 * and ".exrc" by "_exrc" or vice versa.
4146 */
4147 p = gettail(fname_exp);
4148 if ((*p == '.' || *p == '_')
4149 && (STRICMP(p + 1, "vimrc") == 0
4150 || STRICMP(p + 1, "gvimrc") == 0
4151 || STRICMP(p + 1, "exrc") == 0))
4152 {
4153 if (*p == '_')
4154 *p = '.';
4155 else
4156 *p = '_';
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004157#ifdef USE_FOPEN_NOINH
Bram Moolenaar071d4272004-06-13 20:20:40 +00004158 cookie.fp = fopen_noinh_readbin((char *)fname_exp);
4159#else
4160 cookie.fp = mch_fopen((char *)fname_exp, READBIN);
4161#endif
4162 }
4163 }
4164
4165 if (cookie.fp == NULL)
4166 {
4167 if (p_verbose > 0)
4168 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004169 verbose_enter();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004170 if (sourcing_name == NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00004171 smsg((char_u *)_("could not source \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004172 else
4173 smsg((char_u *)_("line %ld: could not source \"%s\""),
Bram Moolenaar555b2802005-05-19 21:08:39 +00004174 sourcing_lnum, fname);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004175 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004176 }
4177 goto theend;
4178 }
4179
4180 /*
4181 * The file exists.
4182 * - In verbose mode, give a message.
4183 * - For a vimrc file, may want to set 'compatible', call vimrc_found().
4184 */
4185 if (p_verbose > 1)
4186 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004187 verbose_enter();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004188 if (sourcing_name == NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00004189 smsg((char_u *)_("sourcing \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004190 else
4191 smsg((char_u *)_("line %ld: sourcing \"%s\""),
Bram Moolenaar555b2802005-05-19 21:08:39 +00004192 sourcing_lnum, fname);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004193 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004194 }
Bram Moolenaar910f66f2006-04-05 20:41:53 +00004195 if (is_vimrc == DOSO_VIMRC)
4196 vimrc_found(fname_exp, (char_u *)"MYVIMRC");
4197 else if (is_vimrc == DOSO_GVIMRC)
4198 vimrc_found(fname_exp, (char_u *)"MYGVIMRC");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004199
4200#ifdef USE_CRNL
4201 /* If no automatic file format: Set default to CR-NL. */
4202 if (*p_ffs == NUL)
4203 cookie.fileformat = EOL_DOS;
4204 else
4205 cookie.fileformat = EOL_UNKNOWN;
4206 cookie.error = FALSE;
4207#endif
4208
4209#ifdef USE_CR
4210 /* If no automatic file format: Set default to CR. */
4211 if (*p_ffs == NUL)
4212 cookie.fileformat = EOL_MAC;
4213 else
4214 cookie.fileformat = EOL_UNKNOWN;
4215 cookie.error = FALSE;
4216#endif
4217
4218 cookie.nextline = NULL;
4219 cookie.finished = FALSE;
4220
4221#ifdef FEAT_EVAL
4222 /*
4223 * Check if this script has a breakpoint.
4224 */
4225 cookie.breakpoint = dbg_find_breakpoint(TRUE, fname_exp, (linenr_T)0);
4226 cookie.fname = fname_exp;
4227 cookie.dbg_tick = debug_tick;
4228
4229 cookie.level = ex_nesting_level;
4230#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004231
4232 /*
4233 * Keep the sourcing name/lnum, for recursive calls.
4234 */
4235 save_sourcing_name = sourcing_name;
4236 sourcing_name = fname_exp;
4237 save_sourcing_lnum = sourcing_lnum;
4238 sourcing_lnum = 0;
4239
Bram Moolenaar73881402009-02-04 16:50:47 +00004240#ifdef FEAT_MBYTE
4241 cookie.conv.vc_type = CONV_NONE; /* no conversion */
4242
4243 /* Read the first line so we can check for a UTF-8 BOM. */
4244 firstline = getsourceline(0, (void *)&cookie, 0);
4245 if (firstline != NULL && STRLEN(firstline) >= 3 && firstline[0] == 0xef
4246 && firstline[1] == 0xbb && firstline[2] == 0xbf)
4247 {
4248 /* Found BOM; setup conversion, skip over BOM and recode the line. */
4249 convert_setup(&cookie.conv, (char_u *)"utf-8", p_enc);
4250 p = string_convert(&cookie.conv, firstline + 3, NULL);
Bram Moolenaar4e2a5952009-02-05 19:48:25 +00004251 if (p == NULL)
4252 p = vim_strsave(firstline + 3);
Bram Moolenaar73881402009-02-04 16:50:47 +00004253 if (p != NULL)
4254 {
4255 vim_free(firstline);
4256 firstline = p;
4257 }
4258 }
4259#endif
4260
Bram Moolenaar071d4272004-06-13 20:20:40 +00004261#ifdef STARTUPTIME
Bram Moolenaarc4e41982010-01-19 16:31:47 +01004262 if (time_fd != NULL)
4263 time_push(&tv_rel, &tv_start);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004264#endif
4265
4266#ifdef FEAT_EVAL
Bram Moolenaar05159a02005-02-26 23:04:13 +00004267# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004268 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004269 prof_child_enter(&wait_start); /* entering a child now */
4270# endif
4271
4272 /* Don't use local function variables, if called from a function.
4273 * Also starts profiling timer for nested script. */
4274 save_funccalp = save_funccal();
4275
Bram Moolenaar071d4272004-06-13 20:20:40 +00004276 /*
4277 * Check if this script was sourced before to finds its SID.
4278 * If it's new, generate a new SID.
4279 */
4280 save_current_SID = current_SID;
4281# ifdef UNIX
4282 stat_ok = (mch_stat((char *)fname_exp, &st) >= 0);
4283# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00004284 for (current_SID = script_items.ga_len; current_SID > 0; --current_SID)
4285 {
4286 si = &SCRIPT_ITEM(current_SID);
4287 if (si->sn_name != NULL
Bram Moolenaar071d4272004-06-13 20:20:40 +00004288 && (
4289# ifdef UNIX
Bram Moolenaar7c626922005-02-07 22:01:03 +00004290 /* Compare dev/ino when possible, it catches symbolic
4291 * links. Also compare file names, the inode may change
4292 * when the file was edited. */
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00004293 ((stat_ok && si->sn_dev_valid)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004294 && (si->sn_dev == st.st_dev
4295 && si->sn_ino == st.st_ino)) ||
Bram Moolenaar071d4272004-06-13 20:20:40 +00004296# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00004297 fnamecmp(si->sn_name, fname_exp) == 0))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004298 break;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004299 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004300 if (current_SID == 0)
4301 {
4302 current_SID = ++last_current_SID;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004303 if (ga_grow(&script_items, (int)(current_SID - script_items.ga_len))
4304 == FAIL)
4305 goto almosttheend;
4306 while (script_items.ga_len < current_SID)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004307 {
Bram Moolenaar05159a02005-02-26 23:04:13 +00004308 ++script_items.ga_len;
4309 SCRIPT_ITEM(script_items.ga_len).sn_name = NULL;
4310# ifdef FEAT_PROFILE
4311 SCRIPT_ITEM(script_items.ga_len).sn_prof_on = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004312# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004313 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00004314 si = &SCRIPT_ITEM(current_SID);
4315 si->sn_name = fname_exp;
4316 fname_exp = NULL;
4317# ifdef UNIX
4318 if (stat_ok)
4319 {
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00004320 si->sn_dev_valid = TRUE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004321 si->sn_dev = st.st_dev;
4322 si->sn_ino = st.st_ino;
4323 }
4324 else
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00004325 si->sn_dev_valid = FALSE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004326# endif
4327
Bram Moolenaar071d4272004-06-13 20:20:40 +00004328 /* Allocate the local script variables to use for this script. */
4329 new_script_vars(current_SID);
4330 }
4331
Bram Moolenaar05159a02005-02-26 23:04:13 +00004332# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004333 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004334 {
4335 int forceit;
4336
4337 /* Check if we do profiling for this script. */
4338 if (!si->sn_prof_on && has_profiling(TRUE, si->sn_name, &forceit))
4339 {
4340 script_do_profile(si);
4341 si->sn_pr_force = forceit;
4342 }
4343 if (si->sn_prof_on)
4344 {
4345 ++si->sn_pr_count;
4346 profile_start(&si->sn_pr_start);
4347 profile_zero(&si->sn_pr_children);
4348 }
4349 }
4350# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004351#endif
4352
4353 /*
4354 * Call do_cmdline, which will call getsourceline() to get the lines.
4355 */
Bram Moolenaar73881402009-02-04 16:50:47 +00004356 do_cmdline(firstline, getsourceline, (void *)&cookie,
Bram Moolenaar071d4272004-06-13 20:20:40 +00004357 DOCMD_VERBOSE|DOCMD_NOWAIT|DOCMD_REPEAT);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004358 retval = OK;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004359
4360#ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004361 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004362 {
4363 /* Get "si" again, "script_items" may have been reallocated. */
4364 si = &SCRIPT_ITEM(current_SID);
4365 if (si->sn_prof_on)
4366 {
4367 profile_end(&si->sn_pr_start);
4368 profile_sub_wait(&wait_start, &si->sn_pr_start);
4369 profile_add(&si->sn_pr_total, &si->sn_pr_start);
Bram Moolenaar1056d982006-03-09 22:37:52 +00004370 profile_self(&si->sn_pr_self, &si->sn_pr_start,
4371 &si->sn_pr_children);
Bram Moolenaar05159a02005-02-26 23:04:13 +00004372 }
4373 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004374#endif
4375
4376 if (got_int)
4377 EMSG(_(e_interr));
4378 sourcing_name = save_sourcing_name;
4379 sourcing_lnum = save_sourcing_lnum;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004380 if (p_verbose > 1)
4381 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004382 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00004383 smsg((char_u *)_("finished sourcing %s"), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004384 if (sourcing_name != NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00004385 smsg((char_u *)_("continuing in %s"), sourcing_name);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004386 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004387 }
4388#ifdef STARTUPTIME
Bram Moolenaarc4e41982010-01-19 16:31:47 +01004389 if (time_fd != NULL)
4390 {
4391 vim_snprintf((char *)IObuff, IOSIZE, "sourcing %s", fname);
4392 time_msg((char *)IObuff, &tv_start);
4393 time_pop(&tv_rel);
4394 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004395#endif
4396
4397#ifdef FEAT_EVAL
4398 /*
4399 * After a "finish" in debug mode, need to break at first command of next
4400 * sourced file.
4401 */
4402 if (save_debug_break_level > ex_nesting_level
4403 && debug_break_level == ex_nesting_level)
4404 ++debug_break_level;
4405#endif
4406
Bram Moolenaar05159a02005-02-26 23:04:13 +00004407#ifdef FEAT_EVAL
4408almosttheend:
4409 current_SID = save_current_SID;
4410 restore_funccal(save_funccalp);
4411# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004412 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004413 prof_child_exit(&wait_start); /* leaving a child now */
4414# endif
4415#endif
4416 fclose(cookie.fp);
4417 vim_free(cookie.nextline);
Bram Moolenaar73881402009-02-04 16:50:47 +00004418 vim_free(firstline);
Bram Moolenaar05159a02005-02-26 23:04:13 +00004419#ifdef FEAT_MBYTE
4420 convert_setup(&cookie.conv, NULL, NULL);
4421#endif
4422
Bram Moolenaar071d4272004-06-13 20:20:40 +00004423theend:
4424 vim_free(fname_exp);
4425 return retval;
4426}
4427
4428#if defined(FEAT_EVAL) || defined(PROTO)
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00004429
Bram Moolenaar071d4272004-06-13 20:20:40 +00004430/*
4431 * ":scriptnames"
4432 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004433 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004434ex_scriptnames(exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004435{
4436 int i;
4437
Bram Moolenaar05159a02005-02-26 23:04:13 +00004438 for (i = 1; i <= script_items.ga_len && !got_int; ++i)
4439 if (SCRIPT_ITEM(i).sn_name != NULL)
Bram Moolenaard58ea072011-06-26 04:25:30 +02004440 {
4441 home_replace(NULL, SCRIPT_ITEM(i).sn_name,
4442 NameBuff, MAXPATHL, TRUE);
4443 smsg((char_u *)"%3d: %s", i, NameBuff);
Bram Moolenaar970a1b82012-03-23 18:39:18 +01004444 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004445}
4446
4447# if defined(BACKSLASH_IN_FILENAME) || defined(PROTO)
4448/*
4449 * Fix slashes in the list of script names for 'shellslash'.
4450 */
4451 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004452scriptnames_slash_adjust(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004453{
4454 int i;
4455
Bram Moolenaar05159a02005-02-26 23:04:13 +00004456 for (i = 1; i <= script_items.ga_len; ++i)
4457 if (SCRIPT_ITEM(i).sn_name != NULL)
4458 slash_adjust(SCRIPT_ITEM(i).sn_name);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004459}
4460# endif
4461
4462/*
4463 * Get a pointer to a script name. Used for ":verbose set".
4464 */
4465 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004466get_scriptname(scid_T id)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004467{
4468 if (id == SID_MODELINE)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00004469 return (char_u *)_("modeline");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004470 if (id == SID_CMDARG)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00004471 return (char_u *)_("--cmd argument");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004472 if (id == SID_CARG)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00004473 return (char_u *)_("-c argument");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004474 if (id == SID_ENV)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00004475 return (char_u *)_("environment variable");
4476 if (id == SID_ERROR)
4477 return (char_u *)_("error handler");
Bram Moolenaar05159a02005-02-26 23:04:13 +00004478 return SCRIPT_ITEM(id).sn_name;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004479}
Bram Moolenaar05159a02005-02-26 23:04:13 +00004480
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00004481# if defined(EXITFREE) || defined(PROTO)
4482 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004483free_scriptnames(void)
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00004484{
4485 int i;
4486
4487 for (i = script_items.ga_len; i > 0; --i)
4488 vim_free(SCRIPT_ITEM(i).sn_name);
4489 ga_clear(&script_items);
4490}
4491# endif
4492
Bram Moolenaar071d4272004-06-13 20:20:40 +00004493#endif
4494
4495#if defined(USE_CR) || defined(PROTO)
4496
4497# if defined(__MSL__) && (__MSL__ >= 22)
4498/*
4499 * Newer version of the Metrowerks library handle DOS and UNIX files
4500 * without help.
4501 * Test with earlier versions, MSL 2.2 is the library supplied with
4502 * Codewarrior Pro 2.
4503 */
4504 char *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004505fgets_cr(char *s, int n, FILE *stream)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004506{
4507 return fgets(s, n, stream);
4508}
4509# else
4510/*
4511 * Version of fgets() which also works for lines ending in a <CR> only
4512 * (Macintosh format).
4513 * For older versions of the Metrowerks library.
4514 * At least CodeWarrior 9 needed this code.
4515 */
4516 char *
Bram Moolenaard14e00e2016-01-31 17:30:51 +01004517fgets_cr(char *s, int n, FILE *stream)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004518{
4519 int c = 0;
4520 int char_read = 0;
4521
4522 while (!feof(stream) && c != '\r' && c != '\n' && char_read < n - 1)
4523 {
4524 c = fgetc(stream);
4525 s[char_read++] = c;
4526 /* If the file is in DOS format, we need to skip a NL after a CR. I
4527 * thought it was the other way around, but this appears to work... */
4528 if (c == '\n')
4529 {
4530 c = fgetc(stream);
4531 if (c != '\r')
4532 ungetc(c, stream);
4533 }
4534 }
4535
4536 s[char_read] = 0;
4537 if (char_read == 0)
4538 return NULL;
4539
4540 if (feof(stream) && char_read == 1)
4541 return NULL;
4542
4543 return s;
4544}
4545# endif
4546#endif
4547
4548/*
4549 * Get one full line from a sourced file.
4550 * Called by do_cmdline() when it's called from do_source().
4551 *
4552 * Return a pointer to the line in allocated memory.
4553 * Return NULL for end-of-file or some error.
4554 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004555 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004556getsourceline(int c UNUSED, void *cookie, int indent UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004557{
4558 struct source_cookie *sp = (struct source_cookie *)cookie;
4559 char_u *line;
Bram Moolenaarc047b9a2012-02-11 20:40:55 +01004560 char_u *p;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004561
4562#ifdef FEAT_EVAL
4563 /* If breakpoints have been added/deleted need to check for it. */
4564 if (sp->dbg_tick < debug_tick)
4565 {
4566 sp->breakpoint = dbg_find_breakpoint(TRUE, sp->fname, sourcing_lnum);
4567 sp->dbg_tick = debug_tick;
4568 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00004569# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004570 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004571 script_line_end();
4572# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004573#endif
4574 /*
4575 * Get current line. If there is a read-ahead line, use it, otherwise get
4576 * one now.
4577 */
4578 if (sp->finished)
4579 line = NULL;
4580 else if (sp->nextline == NULL)
4581 line = get_one_sourceline(sp);
4582 else
4583 {
4584 line = sp->nextline;
4585 sp->nextline = NULL;
4586 ++sourcing_lnum;
4587 }
Bram Moolenaare2cc9702005-03-15 22:43:58 +00004588#ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004589 if (line != NULL && do_profiling == PROF_YES)
Bram Moolenaare2cc9702005-03-15 22:43:58 +00004590 script_line_start();
4591#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004592
4593 /* Only concatenate lines starting with a \ when 'cpoptions' doesn't
4594 * contain the 'C' flag. */
4595 if (line != NULL && (vim_strchr(p_cpo, CPO_CONCAT) == NULL))
4596 {
4597 /* compensate for the one line read-ahead */
4598 --sourcing_lnum;
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004599
4600 /* Get the next line and concatenate it when it starts with a
4601 * backslash. We always need to read the next line, keep it in
4602 * sp->nextline. */
4603 sp->nextline = get_one_sourceline(sp);
4604 if (sp->nextline != NULL && *(p = skipwhite(sp->nextline)) == '\\')
Bram Moolenaar071d4272004-06-13 20:20:40 +00004605 {
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004606 garray_T ga;
4607
Bram Moolenaarb549a732012-02-22 18:29:33 +01004608 ga_init2(&ga, (int)sizeof(char_u), 400);
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004609 ga_concat(&ga, line);
4610 ga_concat(&ga, p + 1);
4611 for (;;)
4612 {
4613 vim_free(sp->nextline);
4614 sp->nextline = get_one_sourceline(sp);
4615 if (sp->nextline == NULL)
4616 break;
4617 p = skipwhite(sp->nextline);
4618 if (*p != '\\')
4619 break;
Bram Moolenaarb549a732012-02-22 18:29:33 +01004620 /* Adjust the growsize to the current length to speed up
4621 * concatenating many lines. */
4622 if (ga.ga_len > 400)
4623 {
4624 if (ga.ga_len > 8000)
4625 ga.ga_growsize = 8000;
4626 else
4627 ga.ga_growsize = ga.ga_len;
4628 }
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004629 ga_concat(&ga, p + 1);
4630 }
4631 ga_append(&ga, NUL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004632 vim_free(line);
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004633 line = ga.ga_data;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004634 }
4635 }
4636
4637#ifdef FEAT_MBYTE
4638 if (line != NULL && sp->conv.vc_type != CONV_NONE)
4639 {
Bram Moolenaarc047b9a2012-02-11 20:40:55 +01004640 char_u *s;
4641
Bram Moolenaar071d4272004-06-13 20:20:40 +00004642 /* Convert the encoding of the script line. */
4643 s = string_convert(&sp->conv, line, NULL);
4644 if (s != NULL)
4645 {
4646 vim_free(line);
4647 line = s;
4648 }
4649 }
4650#endif
4651
4652#ifdef FEAT_EVAL
4653 /* Did we encounter a breakpoint? */
4654 if (sp->breakpoint != 0 && sp->breakpoint <= sourcing_lnum)
4655 {
4656 dbg_breakpoint(sp->fname, sourcing_lnum);
4657 /* Find next breakpoint. */
4658 sp->breakpoint = dbg_find_breakpoint(TRUE, sp->fname, sourcing_lnum);
4659 sp->dbg_tick = debug_tick;
4660 }
4661#endif
4662
4663 return line;
4664}
4665
4666 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004667get_one_sourceline(struct source_cookie *sp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004668{
4669 garray_T ga;
4670 int len;
4671 int c;
4672 char_u *buf;
4673#ifdef USE_CRNL
4674 int has_cr; /* CR-LF found */
4675#endif
4676#ifdef USE_CR
4677 char_u *scan;
4678#endif
4679 int have_read = FALSE;
4680
4681 /* use a growarray to store the sourced line */
Bram Moolenaar6ac54292005-02-02 23:07:25 +00004682 ga_init2(&ga, 1, 250);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004683
4684 /*
4685 * Loop until there is a finished line (or end-of-file).
4686 */
4687 sourcing_lnum++;
4688 for (;;)
4689 {
Bram Moolenaar6ac54292005-02-02 23:07:25 +00004690 /* make room to read at least 120 (more) characters */
4691 if (ga_grow(&ga, 120) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004692 break;
4693 buf = (char_u *)ga.ga_data;
4694
4695#ifdef USE_CR
4696 if (sp->fileformat == EOL_MAC)
4697 {
Bram Moolenaar86b68352004-12-27 21:59:20 +00004698 if (fgets_cr((char *)buf + ga.ga_len, ga.ga_maxlen - ga.ga_len,
4699 sp->fp) == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004700 break;
4701 }
4702 else
4703#endif
Bram Moolenaar86b68352004-12-27 21:59:20 +00004704 if (fgets((char *)buf + ga.ga_len, ga.ga_maxlen - ga.ga_len,
4705 sp->fp) == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004706 break;
Bram Moolenaar6ac54292005-02-02 23:07:25 +00004707 len = ga.ga_len + (int)STRLEN(buf + ga.ga_len);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004708#ifdef USE_CRNL
4709 /* Ignore a trailing CTRL-Z, when in Dos mode. Only recognize the
4710 * CTRL-Z by its own, or after a NL. */
4711 if ( (len == 1 || (len >= 2 && buf[len - 2] == '\n'))
4712 && sp->fileformat == EOL_DOS
4713 && buf[len - 1] == Ctrl_Z)
4714 {
4715 buf[len - 1] = NUL;
4716 break;
4717 }
4718#endif
4719
4720#ifdef USE_CR
4721 /* If the read doesn't stop on a new line, and there's
4722 * some CR then we assume a Mac format */
4723 if (sp->fileformat == EOL_UNKNOWN)
4724 {
4725 if (buf[len - 1] != '\n' && vim_strchr(buf, '\r') != NULL)
4726 sp->fileformat = EOL_MAC;
4727 else
4728 sp->fileformat = EOL_UNIX;
4729 }
4730
4731 if (sp->fileformat == EOL_MAC)
4732 {
4733 scan = vim_strchr(buf, '\r');
4734
4735 if (scan != NULL)
4736 {
4737 *scan = '\n';
4738 if (*(scan + 1) != 0)
4739 {
4740 *(scan + 1) = 0;
4741 fseek(sp->fp, (long)(scan - buf - len + 1), SEEK_CUR);
4742 }
4743 }
4744 len = STRLEN(buf);
4745 }
4746#endif
4747
4748 have_read = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004749 ga.ga_len = len;
4750
4751 /* If the line was longer than the buffer, read more. */
Bram Moolenaar86b68352004-12-27 21:59:20 +00004752 if (ga.ga_maxlen - ga.ga_len == 1 && buf[len - 1] != '\n')
Bram Moolenaar071d4272004-06-13 20:20:40 +00004753 continue;
4754
4755 if (len >= 1 && buf[len - 1] == '\n') /* remove trailing NL */
4756 {
4757#ifdef USE_CRNL
4758 has_cr = (len >= 2 && buf[len - 2] == '\r');
4759 if (sp->fileformat == EOL_UNKNOWN)
4760 {
4761 if (has_cr)
4762 sp->fileformat = EOL_DOS;
4763 else
4764 sp->fileformat = EOL_UNIX;
4765 }
4766
4767 if (sp->fileformat == EOL_DOS)
4768 {
4769 if (has_cr) /* replace trailing CR */
4770 {
4771 buf[len - 2] = '\n';
4772 --len;
4773 --ga.ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004774 }
4775 else /* lines like ":map xx yy^M" will have failed */
4776 {
4777 if (!sp->error)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00004778 {
Bram Moolenaar8820b482017-03-16 17:23:31 +01004779 msg_source(HL_ATTR(HLF_W));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004780 EMSG(_("W15: Warning: Wrong line separator, ^M may be missing"));
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00004781 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004782 sp->error = TRUE;
4783 sp->fileformat = EOL_UNIX;
4784 }
4785 }
4786#endif
4787 /* The '\n' is escaped if there is an odd number of ^V's just
4788 * before it, first set "c" just before the 'V's and then check
4789 * len&c parities (is faster than ((len-c)%2 == 0)) -- Acevedo */
4790 for (c = len - 2; c >= 0 && buf[c] == Ctrl_V; c--)
4791 ;
4792 if ((len & 1) != (c & 1)) /* escaped NL, read more */
4793 {
4794 sourcing_lnum++;
4795 continue;
4796 }
4797
4798 buf[len - 1] = NUL; /* remove the NL */
4799 }
4800
4801 /*
4802 * Check for ^C here now and then, so recursive :so can be broken.
4803 */
4804 line_breakcheck();
4805 break;
4806 }
4807
4808 if (have_read)
4809 return (char_u *)ga.ga_data;
4810
4811 vim_free(ga.ga_data);
4812 return NULL;
4813}
4814
Bram Moolenaar05159a02005-02-26 23:04:13 +00004815#if defined(FEAT_PROFILE) || defined(PROTO)
4816/*
4817 * Called when starting to read a script line.
4818 * "sourcing_lnum" must be correct!
4819 * When skipping lines it may not actually be executed, but we won't find out
4820 * until later and we need to store the time now.
4821 */
4822 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004823script_line_start(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004824{
4825 scriptitem_T *si;
4826 sn_prl_T *pp;
4827
4828 if (current_SID <= 0 || current_SID > script_items.ga_len)
4829 return;
4830 si = &SCRIPT_ITEM(current_SID);
4831 if (si->sn_prof_on && sourcing_lnum >= 1)
4832 {
Bram Moolenaar8c8de832008-06-24 22:58:06 +00004833 /* Grow the array before starting the timer, so that the time spent
Bram Moolenaar05159a02005-02-26 23:04:13 +00004834 * here isn't counted. */
Bram Moolenaarcde88542015-08-11 19:14:00 +02004835 (void)ga_grow(&si->sn_prl_ga, (int)(sourcing_lnum - si->sn_prl_ga.ga_len));
Bram Moolenaar05159a02005-02-26 23:04:13 +00004836 si->sn_prl_idx = sourcing_lnum - 1;
4837 while (si->sn_prl_ga.ga_len <= si->sn_prl_idx
4838 && si->sn_prl_ga.ga_len < si->sn_prl_ga.ga_maxlen)
4839 {
4840 /* Zero counters for a line that was not used before. */
4841 pp = &PRL_ITEM(si, si->sn_prl_ga.ga_len);
4842 pp->snp_count = 0;
4843 profile_zero(&pp->sn_prl_total);
4844 profile_zero(&pp->sn_prl_self);
4845 ++si->sn_prl_ga.ga_len;
4846 }
4847 si->sn_prl_execed = FALSE;
4848 profile_start(&si->sn_prl_start);
4849 profile_zero(&si->sn_prl_children);
4850 profile_get_wait(&si->sn_prl_wait);
4851 }
4852}
4853
4854/*
4855 * Called when actually executing a function line.
4856 */
4857 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004858script_line_exec(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004859{
4860 scriptitem_T *si;
4861
4862 if (current_SID <= 0 || current_SID > script_items.ga_len)
4863 return;
4864 si = &SCRIPT_ITEM(current_SID);
4865 if (si->sn_prof_on && si->sn_prl_idx >= 0)
4866 si->sn_prl_execed = TRUE;
4867}
4868
4869/*
4870 * Called when done with a function line.
4871 */
4872 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004873script_line_end(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004874{
4875 scriptitem_T *si;
4876 sn_prl_T *pp;
4877
4878 if (current_SID <= 0 || current_SID > script_items.ga_len)
4879 return;
4880 si = &SCRIPT_ITEM(current_SID);
4881 if (si->sn_prof_on && si->sn_prl_idx >= 0
4882 && si->sn_prl_idx < si->sn_prl_ga.ga_len)
4883 {
4884 if (si->sn_prl_execed)
4885 {
4886 pp = &PRL_ITEM(si, si->sn_prl_idx);
4887 ++pp->snp_count;
4888 profile_end(&si->sn_prl_start);
4889 profile_sub_wait(&si->sn_prl_wait, &si->sn_prl_start);
Bram Moolenaar05159a02005-02-26 23:04:13 +00004890 profile_add(&pp->sn_prl_total, &si->sn_prl_start);
Bram Moolenaar1056d982006-03-09 22:37:52 +00004891 profile_self(&pp->sn_prl_self, &si->sn_prl_start,
4892 &si->sn_prl_children);
Bram Moolenaar05159a02005-02-26 23:04:13 +00004893 }
4894 si->sn_prl_idx = -1;
4895 }
4896}
4897#endif
4898
Bram Moolenaar071d4272004-06-13 20:20:40 +00004899/*
4900 * ":scriptencoding": Set encoding conversion for a sourced script.
4901 * Without the multi-byte feature it's simply ignored.
4902 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004903 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004904ex_scriptencoding(exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004905{
4906#ifdef FEAT_MBYTE
4907 struct source_cookie *sp;
4908 char_u *name;
4909
4910 if (!getline_equal(eap->getline, eap->cookie, getsourceline))
4911 {
4912 EMSG(_("E167: :scriptencoding used outside of a sourced file"));
4913 return;
4914 }
4915
4916 if (*eap->arg != NUL)
4917 {
4918 name = enc_canonize(eap->arg);
4919 if (name == NULL) /* out of memory */
4920 return;
4921 }
4922 else
4923 name = eap->arg;
4924
4925 /* Setup for conversion from the specified encoding to 'encoding'. */
4926 sp = (struct source_cookie *)getline_cookie(eap->getline, eap->cookie);
4927 convert_setup(&sp->conv, name, p_enc);
4928
4929 if (name != eap->arg)
4930 vim_free(name);
4931#endif
4932}
4933
4934#if defined(FEAT_EVAL) || defined(PROTO)
4935/*
4936 * ":finish": Mark a sourced file as finished.
4937 */
4938 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004939ex_finish(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004940{
4941 if (getline_equal(eap->getline, eap->cookie, getsourceline))
4942 do_finish(eap, FALSE);
4943 else
4944 EMSG(_("E168: :finish used outside of a sourced file"));
4945}
4946
4947/*
4948 * Mark a sourced file as finished. Possibly makes the ":finish" pending.
4949 * Also called for a pending finish at the ":endtry" or after returning from
4950 * an extra do_cmdline(). "reanimate" is used in the latter case.
4951 */
4952 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004953do_finish(exarg_T *eap, int reanimate)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004954{
4955 int idx;
4956
4957 if (reanimate)
4958 ((struct source_cookie *)getline_cookie(eap->getline,
4959 eap->cookie))->finished = FALSE;
4960
4961 /*
4962 * Cleanup (and inactivate) conditionals, but stop when a try conditional
4963 * not in its finally clause (which then is to be executed next) is found.
4964 * In this case, make the ":finish" pending for execution at the ":endtry".
4965 * Otherwise, finish normally.
4966 */
4967 idx = cleanup_conditionals(eap->cstack, 0, TRUE);
4968 if (idx >= 0)
4969 {
4970 eap->cstack->cs_pending[idx] = CSTP_FINISH;
4971 report_make_pending(CSTP_FINISH, NULL);
4972 }
4973 else
4974 ((struct source_cookie *)getline_cookie(eap->getline,
4975 eap->cookie))->finished = TRUE;
4976}
4977
4978
4979/*
4980 * Return TRUE when a sourced file had the ":finish" command: Don't give error
4981 * message for missing ":endif".
4982 * Return FALSE when not sourcing a file.
4983 */
4984 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004985source_finished(
4986 char_u *(*fgetline)(int, void *, int),
4987 void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004988{
Bram Moolenaar89d40322006-08-29 15:30:07 +00004989 return (getline_equal(fgetline, cookie, getsourceline)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004990 && ((struct source_cookie *)getline_cookie(
Bram Moolenaar89d40322006-08-29 15:30:07 +00004991 fgetline, cookie))->finished);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004992}
4993#endif
4994
4995#if defined(FEAT_LISTCMDS) || defined(PROTO)
4996/*
4997 * ":checktime [buffer]"
4998 */
4999 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005000ex_checktime(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005001{
5002 buf_T *buf;
5003 int save_no_check_timestamps = no_check_timestamps;
5004
5005 no_check_timestamps = 0;
5006 if (eap->addr_count == 0) /* default is all buffers */
5007 check_timestamps(FALSE);
5008 else
5009 {
5010 buf = buflist_findnr((int)eap->line2);
5011 if (buf != NULL) /* cannot happen? */
5012 (void)buf_check_timestamp(buf, FALSE);
5013 }
5014 no_check_timestamps = save_no_check_timestamps;
5015}
5016#endif
5017
Bram Moolenaar071d4272004-06-13 20:20:40 +00005018#if (defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
5019 && (defined(FEAT_EVAL) || defined(FEAT_MULTI_LANG))
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02005020# define HAVE_GET_LOCALE_VAL
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005021static char_u *get_locale_val(int what);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005022
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005023 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005024get_locale_val(int what)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005025{
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005026 char_u *loc;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005027
Bram Moolenaar48e330a2016-02-23 14:53:34 +01005028 /* Obtain the locale value from the libraries. */
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005029 loc = (char_u *)setlocale(what, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005030
Bram Moolenaar61660ea2006-04-07 21:40:07 +00005031# ifdef WIN32
Bram Moolenaar071d4272004-06-13 20:20:40 +00005032 if (loc != NULL)
5033 {
5034 char_u *p;
5035
Bram Moolenaar61660ea2006-04-07 21:40:07 +00005036 /* setocale() returns something like "LC_COLLATE=<name>;LC_..." when
5037 * one of the values (e.g., LC_CTYPE) differs. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005038 p = vim_strchr(loc, '=');
5039 if (p != NULL)
5040 {
5041 loc = ++p;
5042 while (*p != NUL) /* remove trailing newline */
5043 {
Bram Moolenaar61660ea2006-04-07 21:40:07 +00005044 if (*p < ' ' || *p == ';')
Bram Moolenaar071d4272004-06-13 20:20:40 +00005045 {
5046 *p = NUL;
5047 break;
5048 }
5049 ++p;
5050 }
5051 }
5052 }
5053# endif
5054
5055 return loc;
5056}
5057#endif
5058
5059
5060#ifdef WIN32
5061/*
5062 * On MS-Windows locale names are strings like "German_Germany.1252", but
5063 * gettext expects "de". Try to translate one into another here for a few
5064 * supported languages.
5065 */
5066 static char_u *
5067gettext_lang(char_u *name)
5068{
5069 int i;
5070 static char *(mtable[]) = {
5071 "afrikaans", "af",
5072 "czech", "cs",
5073 "dutch", "nl",
5074 "german", "de",
5075 "english_united kingdom", "en_GB",
5076 "spanish", "es",
5077 "french", "fr",
5078 "italian", "it",
5079 "japanese", "ja",
5080 "korean", "ko",
5081 "norwegian", "no",
5082 "polish", "pl",
5083 "russian", "ru",
5084 "slovak", "sk",
5085 "swedish", "sv",
5086 "ukrainian", "uk",
5087 "chinese_china", "zh_CN",
5088 "chinese_taiwan", "zh_TW",
5089 NULL};
5090
5091 for (i = 0; mtable[i] != NULL; i += 2)
5092 if (STRNICMP(mtable[i], name, STRLEN(mtable[i])) == 0)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005093 return (char_u *)mtable[i + 1];
Bram Moolenaar071d4272004-06-13 20:20:40 +00005094 return name;
5095}
5096#endif
5097
5098#if defined(FEAT_MULTI_LANG) || defined(PROTO)
5099/*
5100 * Obtain the current messages language. Used to set the default for
5101 * 'helplang'. May return NULL or an empty string.
5102 */
5103 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005104get_mess_lang(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005105{
5106 char_u *p;
5107
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02005108# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00005109# if defined(LC_MESSAGES)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005110 p = get_locale_val(LC_MESSAGES);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005111# else
5112 /* This is necessary for Win32, where LC_MESSAGES is not defined and $LANG
Bram Moolenaar61660ea2006-04-07 21:40:07 +00005113 * may be set to the LCID number. LC_COLLATE is the best guess, LC_TIME
5114 * and LC_MONETARY may be set differently for a Japanese working in the
5115 * US. */
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005116 p = get_locale_val(LC_COLLATE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005117# endif
5118# else
5119 p = mch_getenv((char_u *)"LC_ALL");
5120 if (p == NULL || *p == NUL)
5121 {
5122 p = mch_getenv((char_u *)"LC_MESSAGES");
5123 if (p == NULL || *p == NUL)
5124 p = mch_getenv((char_u *)"LANG");
5125 }
5126# endif
5127# ifdef WIN32
5128 p = gettext_lang(p);
5129# endif
5130 return p;
5131}
5132#endif
5133
Bram Moolenaardef9e822004-12-31 20:58:58 +00005134/* Complicated #if; matches with where get_mess_env() is used below. */
5135#if (defined(FEAT_EVAL) && !((defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
5136 && defined(LC_MESSAGES))) \
5137 || ((defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
5138 && (defined(FEAT_GETTEXT) || defined(FEAT_MBYTE)) \
5139 && !defined(LC_MESSAGES))
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01005140static char_u *get_mess_env(void);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005141
5142/*
5143 * Get the language used for messages from the environment.
5144 */
5145 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005146get_mess_env(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005147{
5148 char_u *p;
5149
5150 p = mch_getenv((char_u *)"LC_ALL");
5151 if (p == NULL || *p == NUL)
5152 {
5153 p = mch_getenv((char_u *)"LC_MESSAGES");
5154 if (p == NULL || *p == NUL)
5155 {
5156 p = mch_getenv((char_u *)"LANG");
5157 if (p != NULL && VIM_ISDIGIT(*p))
5158 p = NULL; /* ignore something like "1043" */
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02005159# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00005160 if (p == NULL || *p == NUL)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005161 p = get_locale_val(LC_CTYPE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005162# endif
5163 }
5164 }
5165 return p;
5166}
5167#endif
5168
5169#if defined(FEAT_EVAL) || defined(PROTO)
5170
5171/*
5172 * Set the "v:lang" variable according to the current locale setting.
5173 * Also do "v:lc_time"and "v:ctype".
5174 */
5175 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005176set_lang_var(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005177{
5178 char_u *loc;
5179
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02005180# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005181 loc = get_locale_val(LC_CTYPE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005182# else
5183 /* setlocale() not supported: use the default value */
5184 loc = (char_u *)"C";
5185# endif
5186 set_vim_var_string(VV_CTYPE, loc, -1);
5187
5188 /* When LC_MESSAGES isn't defined use the value from $LC_MESSAGES, fall
5189 * back to LC_CTYPE if it's empty. */
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02005190# if defined(HAVE_GET_LOCALE_VAL) && defined(LC_MESSAGES)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005191 loc = get_locale_val(LC_MESSAGES);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005192# else
5193 loc = get_mess_env();
5194# endif
5195 set_vim_var_string(VV_LANG, loc, -1);
5196
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02005197# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005198 loc = get_locale_val(LC_TIME);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005199# endif
5200 set_vim_var_string(VV_LC_TIME, loc, -1);
5201}
5202#endif
5203
5204#if (defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
5205 && (defined(FEAT_GETTEXT) || defined(FEAT_MBYTE))
5206/*
5207 * ":language": Set the language (locale).
5208 */
5209 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005210ex_language(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005211{
5212 char *loc;
5213 char_u *p;
5214 char_u *name;
5215 int what = LC_ALL;
5216 char *whatstr = "";
5217#ifdef LC_MESSAGES
5218# define VIM_LC_MESSAGES LC_MESSAGES
5219#else
5220# define VIM_LC_MESSAGES 6789
5221#endif
5222
5223 name = eap->arg;
5224
5225 /* Check for "messages {name}", "ctype {name}" or "time {name}" argument.
5226 * Allow abbreviation, but require at least 3 characters to avoid
5227 * confusion with a two letter language name "me" or "ct". */
5228 p = skiptowhite(eap->arg);
Bram Moolenaar1c465442017-03-12 20:10:05 +01005229 if ((*p == NUL || VIM_ISWHITE(*p)) && p - eap->arg >= 3)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005230 {
5231 if (STRNICMP(eap->arg, "messages", p - eap->arg) == 0)
5232 {
5233 what = VIM_LC_MESSAGES;
5234 name = skipwhite(p);
5235 whatstr = "messages ";
5236 }
5237 else if (STRNICMP(eap->arg, "ctype", p - eap->arg) == 0)
5238 {
5239 what = LC_CTYPE;
5240 name = skipwhite(p);
5241 whatstr = "ctype ";
5242 }
5243 else if (STRNICMP(eap->arg, "time", p - eap->arg) == 0)
5244 {
5245 what = LC_TIME;
5246 name = skipwhite(p);
5247 whatstr = "time ";
5248 }
5249 }
5250
5251 if (*name == NUL)
5252 {
5253#ifndef LC_MESSAGES
5254 if (what == VIM_LC_MESSAGES)
5255 p = get_mess_env();
5256 else
5257#endif
5258 p = (char_u *)setlocale(what, NULL);
5259 if (p == NULL || *p == NUL)
5260 p = (char_u *)"Unknown";
5261 smsg((char_u *)_("Current %slanguage: \"%s\""), whatstr, p);
5262 }
5263 else
5264 {
5265#ifndef LC_MESSAGES
5266 if (what == VIM_LC_MESSAGES)
5267 loc = "";
5268 else
5269#endif
Bram Moolenaar8c8de832008-06-24 22:58:06 +00005270 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00005271 loc = setlocale(what, (char *)name);
Bram Moolenaar8c8de832008-06-24 22:58:06 +00005272#if defined(FEAT_FLOAT) && defined(LC_NUMERIC)
5273 /* Make sure strtod() uses a decimal point, not a comma. */
5274 setlocale(LC_NUMERIC, "C");
5275#endif
5276 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00005277 if (loc == NULL)
5278 EMSG2(_("E197: Cannot set language to \"%s\""), name);
5279 else
5280 {
5281#ifdef HAVE_NL_MSG_CAT_CNTR
5282 /* Need to do this for GNU gettext, otherwise cached translations
5283 * will be used again. */
5284 extern int _nl_msg_cat_cntr;
5285
5286 ++_nl_msg_cat_cntr;
5287#endif
Bram Moolenaarb8017e72007-05-10 18:59:07 +00005288 /* Reset $LC_ALL, otherwise it would overrule everything. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005289 vim_setenv((char_u *)"LC_ALL", (char_u *)"");
5290
5291 if (what != LC_TIME)
5292 {
5293 /* Tell gettext() what to translate to. It apparently doesn't
5294 * use the currently effective locale. Also do this when
5295 * FEAT_GETTEXT isn't defined, so that shell commands use this
5296 * value. */
5297 if (what == LC_ALL)
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00005298 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00005299 vim_setenv((char_u *)"LANG", name);
Bram Moolenaare3a0b532013-06-28 20:36:30 +02005300
5301 /* Clear $LANGUAGE because GNU gettext uses it. */
5302 vim_setenv((char_u *)"LANGUAGE", (char_u *)"");
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00005303# ifdef WIN32
5304 /* Apparently MS-Windows printf() may cause a crash when
5305 * we give it 8-bit text while it's expecting text in the
5306 * current locale. This call avoids that. */
5307 setlocale(LC_CTYPE, "C");
5308# endif
5309 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00005310 if (what != LC_CTYPE)
5311 {
5312 char_u *mname;
5313#ifdef WIN32
5314 mname = gettext_lang(name);
5315#else
5316 mname = name;
5317#endif
5318 vim_setenv((char_u *)"LC_MESSAGES", mname);
5319#ifdef FEAT_MULTI_LANG
5320 set_helplang_default(mname);
5321#endif
5322 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00005323 }
5324
5325# ifdef FEAT_EVAL
5326 /* Set v:lang, v:lc_time and v:ctype to the final result. */
5327 set_lang_var();
5328# endif
Bram Moolenaar6cc00c72011-10-20 21:41:09 +02005329# ifdef FEAT_TITLE
5330 maketitle();
5331# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00005332 }
5333 }
5334}
5335
5336# if defined(FEAT_CMDL_COMPL) || defined(PROTO)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005337
5338static char_u **locales = NULL; /* Array of all available locales */
Bram Moolenaarb8f7bd62017-01-12 20:28:25 +01005339
5340# ifndef WIN32
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005341static int did_init_locales = FALSE;
5342
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005343/* Return an array of strings for all available locales + NULL for the
5344 * last element. Return NULL in case of error. */
5345 static char_u **
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005346find_locales(void)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005347{
5348 garray_T locales_ga;
5349 char_u *loc;
5350
5351 /* Find all available locales by running command "locale -a". If this
5352 * doesn't work we won't have completion. */
5353 char_u *locale_a = get_cmd_output((char_u *)"locale -a",
Bram Moolenaar39c29ed2014-04-05 19:44:40 +02005354 NULL, SHELL_SILENT, NULL);
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005355 if (locale_a == NULL)
5356 return NULL;
5357 ga_init2(&locales_ga, sizeof(char_u *), 20);
5358
5359 /* Transform locale_a string where each locale is separated by "\n"
5360 * into an array of locale strings. */
5361 loc = (char_u *)strtok((char *)locale_a, "\n");
5362
5363 while (loc != NULL)
5364 {
5365 if (ga_grow(&locales_ga, 1) == FAIL)
5366 break;
5367 loc = vim_strsave(loc);
5368 if (loc == NULL)
5369 break;
5370
5371 ((char_u **)locales_ga.ga_data)[locales_ga.ga_len++] = loc;
5372 loc = (char_u *)strtok(NULL, "\n");
5373 }
5374 vim_free(locale_a);
5375 if (ga_grow(&locales_ga, 1) == FAIL)
5376 {
5377 ga_clear(&locales_ga);
5378 return NULL;
5379 }
5380 ((char_u **)locales_ga.ga_data)[locales_ga.ga_len] = NULL;
5381 return (char_u **)locales_ga.ga_data;
5382}
Bram Moolenaarb8f7bd62017-01-12 20:28:25 +01005383# endif
5384
5385/*
5386 * Lazy initialization of all available locales.
5387 */
5388 static void
5389init_locales(void)
5390{
5391# ifndef WIN32
5392 if (!did_init_locales)
5393 {
5394 did_init_locales = TRUE;
5395 locales = find_locales();
5396 }
5397# endif
5398}
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005399
5400# if defined(EXITFREE) || defined(PROTO)
5401 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005402free_locales(void)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005403{
5404 int i;
5405 if (locales != NULL)
5406 {
5407 for (i = 0; locales[i] != NULL; i++)
5408 vim_free(locales[i]);
5409 vim_free(locales);
5410 locales = NULL;
5411 }
5412}
5413# endif
5414
Bram Moolenaar071d4272004-06-13 20:20:40 +00005415/*
5416 * Function given to ExpandGeneric() to obtain the possible arguments of the
5417 * ":language" command.
5418 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005419 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005420get_lang_arg(expand_T *xp UNUSED, int idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005421{
5422 if (idx == 0)
5423 return (char_u *)"messages";
5424 if (idx == 1)
5425 return (char_u *)"ctype";
5426 if (idx == 2)
5427 return (char_u *)"time";
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005428
5429 init_locales();
5430 if (locales == NULL)
5431 return NULL;
5432 return locales[idx - 3];
5433}
5434
5435/*
5436 * Function given to ExpandGeneric() to obtain the available locales.
5437 */
5438 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005439get_locales(expand_T *xp UNUSED, int idx)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005440{
5441 init_locales();
5442 if (locales == NULL)
5443 return NULL;
5444 return locales[idx];
Bram Moolenaar071d4272004-06-13 20:20:40 +00005445}
5446# endif
5447
5448#endif