blob: e9a9a6de6fdd3fc96a6752f25dfbf047d5c3ae2a [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 Moolenaara8e93d62017-09-18 21:50:47 +02001093 static long
1094timer_time_left(timer_T *timer, proftime_T *now)
1095{
1096# ifdef WIN3264
1097 LARGE_INTEGER fr;
1098
1099 if (now->QuadPart > timer->tr_due.QuadPart)
1100 return 0;
1101 QueryPerformanceFrequency(&fr);
1102 return (long)(((double)(timer->tr_due.QuadPart - now->QuadPart)
1103 / (double)fr.QuadPart) * 1000);
1104# else
1105 if (now->tv_sec > timer->tr_due.tv_sec)
1106 return 0;
1107 return (timer->tr_due.tv_sec - now->tv_sec) * 1000
1108 + (timer->tr_due.tv_usec - now->tv_usec) / 1000;
1109# endif
1110}
Bram Moolenaar417ccd72016-09-01 21:26:20 +02001111
Bram Moolenaar975b5272016-03-15 23:10:59 +01001112/*
1113 * Insert a timer in the list of timers.
1114 */
1115 static void
1116insert_timer(timer_T *timer)
1117{
1118 timer->tr_next = first_timer;
1119 timer->tr_prev = NULL;
1120 if (first_timer != NULL)
1121 first_timer->tr_prev = timer;
1122 first_timer = timer;
Bram Moolenaar4231da42016-06-02 14:30:04 +02001123 did_add_timer = TRUE;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001124}
1125
1126/*
1127 * Take a timer out of the list of timers.
1128 */
1129 static void
1130remove_timer(timer_T *timer)
1131{
1132 if (timer->tr_prev == NULL)
1133 first_timer = timer->tr_next;
1134 else
1135 timer->tr_prev->tr_next = timer->tr_next;
1136 if (timer->tr_next != NULL)
1137 timer->tr_next->tr_prev = timer->tr_prev;
1138}
1139
1140 static void
1141free_timer(timer_T *timer)
1142{
Bram Moolenaar75537a92016-09-05 22:45:28 +02001143 free_callback(timer->tr_callback, timer->tr_partial);
1144 vim_free(timer);
Bram Moolenaar975b5272016-03-15 23:10:59 +01001145}
1146
1147/*
1148 * Create a timer and return it. NULL if out of memory.
1149 * Caller should set the callback.
1150 */
1151 timer_T *
1152create_timer(long msec, int repeat)
1153{
1154 timer_T *timer = (timer_T *)alloc_clear(sizeof(timer_T));
Bram Moolenaaree39ef02016-09-10 19:17:42 +02001155 long prev_id = last_timer_id;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001156
1157 if (timer == NULL)
1158 return NULL;
Bram Moolenaaree39ef02016-09-10 19:17:42 +02001159 if (++last_timer_id <= prev_id)
Bram Moolenaar75537a92016-09-05 22:45:28 +02001160 /* Overflow! Might cause duplicates... */
1161 last_timer_id = 0;
1162 timer->tr_id = last_timer_id;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001163 insert_timer(timer);
1164 if (repeat != 0)
Bram Moolenaar975b5272016-03-15 23:10:59 +01001165 timer->tr_repeat = repeat - 1;
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001166 timer->tr_interval = msec;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001167
1168 profile_setlimit(msec, &timer->tr_due);
1169 return timer;
1170}
1171
1172/*
1173 * Invoke the callback of "timer".
1174 */
1175 static void
1176timer_callback(timer_T *timer)
1177{
1178 typval_T rettv;
1179 int dummy;
1180 typval_T argv[2];
1181
1182 argv[0].v_type = VAR_NUMBER;
Bram Moolenaar75537a92016-09-05 22:45:28 +02001183 argv[0].vval.v_number = (varnumber_T)timer->tr_id;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001184 argv[1].v_type = VAR_UNKNOWN;
1185
1186 call_func(timer->tr_callback, (int)STRLEN(timer->tr_callback),
Bram Moolenaardf48fb42016-07-22 21:50:18 +02001187 &rettv, 1, argv, NULL, 0L, 0L, &dummy, TRUE,
Bram Moolenaar975b5272016-03-15 23:10:59 +01001188 timer->tr_partial, NULL);
1189 clear_tv(&rettv);
1190}
1191
1192/*
1193 * Call timers that are due.
1194 * Return the time in msec until the next timer is due.
Bram Moolenaarc4f83382017-07-07 14:50:44 +02001195 * Returns -1 if there are no pending timers.
Bram Moolenaar975b5272016-03-15 23:10:59 +01001196 */
1197 long
Bram Moolenaarcf089462016-06-12 21:18:43 +02001198check_due_timer(void)
Bram Moolenaar975b5272016-03-15 23:10:59 +01001199{
1200 timer_T *timer;
Bram Moolenaar75537a92016-09-05 22:45:28 +02001201 timer_T *timer_next;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001202 long this_due;
Bram Moolenaar597385a2016-03-16 23:24:43 +01001203 long next_due = -1;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001204 proftime_T now;
1205 int did_one = FALSE;
Bram Moolenaar02e177d2017-08-26 23:43:28 +02001206 int need_update_screen = FALSE;
Bram Moolenaar75537a92016-09-05 22:45:28 +02001207 long current_id = last_timer_id;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001208
Bram Moolenaarc577d812017-07-08 22:37:34 +02001209 /* Don't run any timers while exiting or dealing with an error. */
1210 if (exiting || aborting())
Bram Moolenaarc4f83382017-07-07 14:50:44 +02001211 return next_due;
1212
Bram Moolenaar75537a92016-09-05 22:45:28 +02001213 profile_start(&now);
1214 for (timer = first_timer; timer != NULL && !got_int; timer = timer_next)
Bram Moolenaar975b5272016-03-15 23:10:59 +01001215 {
Bram Moolenaar75537a92016-09-05 22:45:28 +02001216 timer_next = timer->tr_next;
Bram Moolenaar417ccd72016-09-01 21:26:20 +02001217
Bram Moolenaar75537a92016-09-05 22:45:28 +02001218 if (timer->tr_id == -1 || timer->tr_firing || timer->tr_paused)
1219 continue;
Bram Moolenaara8e93d62017-09-18 21:50:47 +02001220 this_due = timer_time_left(timer, &now);
Bram Moolenaar75537a92016-09-05 22:45:28 +02001221 if (this_due <= 1)
1222 {
Bram Moolenaar1e8e1452017-06-24 16:03:06 +02001223 int save_timer_busy = timer_busy;
1224 int save_vgetc_busy = vgetc_busy;
Bram Moolenaare723c422017-09-06 23:40:10 +02001225 int save_did_emsg = did_emsg;
1226 int save_called_emsg = called_emsg;
Bram Moolenaar02e177d2017-08-26 23:43:28 +02001227 int save_must_redraw = must_redraw;
Bram Moolenaare723c422017-09-06 23:40:10 +02001228 int save_trylevel = trylevel;
1229 int save_did_throw = did_throw;
Bram Moolenaarf5291f32017-09-14 22:55:37 +02001230 int save_ex_pressedreturn = get_pressedreturn();
Bram Moolenaare723c422017-09-06 23:40:10 +02001231 except_T *save_current_exception = current_exception;
Bram Moolenaar1e8e1452017-06-24 16:03:06 +02001232
Bram Moolenaare723c422017-09-06 23:40:10 +02001233 /* Create a scope for running the timer callback, ignoring most of
1234 * the current scope, such as being inside a try/catch. */
Bram Moolenaar1e8e1452017-06-24 16:03:06 +02001235 timer_busy = timer_busy > 0 || vgetc_busy > 0;
1236 vgetc_busy = 0;
Bram Moolenaarc577d812017-07-08 22:37:34 +02001237 called_emsg = FALSE;
Bram Moolenaare723c422017-09-06 23:40:10 +02001238 did_emsg = FALSE;
1239 did_uncaught_emsg = FALSE;
Bram Moolenaar02e177d2017-08-26 23:43:28 +02001240 must_redraw = 0;
Bram Moolenaare723c422017-09-06 23:40:10 +02001241 trylevel = 0;
1242 did_throw = FALSE;
1243 current_exception = NULL;
1244
Bram Moolenaar75537a92016-09-05 22:45:28 +02001245 timer->tr_firing = TRUE;
1246 timer_callback(timer);
1247 timer->tr_firing = FALSE;
Bram Moolenaare723c422017-09-06 23:40:10 +02001248
Bram Moolenaar75537a92016-09-05 22:45:28 +02001249 timer_next = timer->tr_next;
1250 did_one = TRUE;
Bram Moolenaar1e8e1452017-06-24 16:03:06 +02001251 timer_busy = save_timer_busy;
1252 vgetc_busy = save_vgetc_busy;
Bram Moolenaare723c422017-09-06 23:40:10 +02001253 if (did_uncaught_emsg)
Bram Moolenaarc577d812017-07-08 22:37:34 +02001254 ++timer->tr_emsg_count;
Bram Moolenaare723c422017-09-06 23:40:10 +02001255 did_emsg = save_did_emsg;
1256 called_emsg = save_called_emsg;
1257 trylevel = save_trylevel;
1258 did_throw = save_did_throw;
1259 current_exception = save_current_exception;
Bram Moolenaar02e177d2017-08-26 23:43:28 +02001260 if (must_redraw != 0)
1261 need_update_screen = TRUE;
1262 must_redraw = must_redraw > save_must_redraw
1263 ? must_redraw : save_must_redraw;
Bram Moolenaarf5291f32017-09-14 22:55:37 +02001264 set_pressedreturn(save_ex_pressedreturn);
Bram Moolenaar75537a92016-09-05 22:45:28 +02001265
1266 /* Only fire the timer again if it repeats and stop_timer() wasn't
1267 * called while inside the callback (tr_id == -1). */
Bram Moolenaarc577d812017-07-08 22:37:34 +02001268 if (timer->tr_repeat != 0 && timer->tr_id != -1
1269 && timer->tr_emsg_count < 3)
Bram Moolenaar75537a92016-09-05 22:45:28 +02001270 {
1271 profile_setlimit(timer->tr_interval, &timer->tr_due);
Bram Moolenaara8e93d62017-09-18 21:50:47 +02001272 this_due = timer_time_left(timer, &now);
Bram Moolenaar75537a92016-09-05 22:45:28 +02001273 if (this_due < 1)
1274 this_due = 1;
1275 if (timer->tr_repeat > 0)
1276 --timer->tr_repeat;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001277 }
Bram Moolenaar75537a92016-09-05 22:45:28 +02001278 else
1279 {
1280 this_due = -1;
1281 remove_timer(timer);
1282 free_timer(timer);
1283 }
Bram Moolenaar975b5272016-03-15 23:10:59 +01001284 }
Bram Moolenaar75537a92016-09-05 22:45:28 +02001285 if (this_due > 0 && (next_due == -1 || next_due > this_due))
1286 next_due = this_due;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001287 }
1288
1289 if (did_one)
Bram Moolenaar02e177d2017-08-26 23:43:28 +02001290 redraw_after_callback(need_update_screen);
Bram Moolenaar975b5272016-03-15 23:10:59 +01001291
Bram Moolenaar75537a92016-09-05 22:45:28 +02001292 return current_id != last_timer_id ? 1 : next_due;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001293}
1294
1295/*
1296 * Find a timer by ID. Returns NULL if not found;
1297 */
1298 timer_T *
Bram Moolenaar75537a92016-09-05 22:45:28 +02001299find_timer(long id)
Bram Moolenaar975b5272016-03-15 23:10:59 +01001300{
1301 timer_T *timer;
1302
Bram Moolenaar75537a92016-09-05 22:45:28 +02001303 if (id >= 0)
1304 {
1305 for (timer = first_timer; timer != NULL; timer = timer->tr_next)
1306 if (timer->tr_id == id)
1307 return timer;
1308 }
1309 return NULL;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001310}
1311
1312
1313/*
1314 * Stop a timer and delete it.
1315 */
1316 void
1317stop_timer(timer_T *timer)
1318{
Bram Moolenaar75537a92016-09-05 22:45:28 +02001319 if (timer->tr_firing)
1320 /* Free the timer after the callback returns. */
1321 timer->tr_id = -1;
1322 else
1323 {
1324 remove_timer(timer);
1325 free_timer(timer);
1326 }
Bram Moolenaar975b5272016-03-15 23:10:59 +01001327}
Bram Moolenaare3188e22016-05-31 21:13:04 +02001328
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001329 void
Bram Moolenaarb73598e2016-08-07 18:22:53 +02001330stop_all_timers(void)
1331{
Bram Moolenaar75537a92016-09-05 22:45:28 +02001332 timer_T *timer;
1333 timer_T *timer_next;
1334
1335 for (timer = first_timer; timer != NULL; timer = timer_next)
1336 {
1337 timer_next = timer->tr_next;
1338 stop_timer(timer);
1339 }
Bram Moolenaarb73598e2016-08-07 18:22:53 +02001340}
1341
1342 void
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001343add_timer_info(typval_T *rettv, timer_T *timer)
1344{
1345 list_T *list = rettv->vval.v_list;
1346 dict_T *dict = dict_alloc();
1347 dictitem_T *di;
1348 long remaining;
1349 proftime_T now;
1350
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);
Bram Moolenaara8e93d62017-09-18 21:50:47 +02001359 remaining = timer_time_left(timer, &now);
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001360 dict_add_nr_str(dict, "remaining", (long)remaining, NULL);
1361
1362 dict_add_nr_str(dict, "repeat",
1363 (long)(timer->tr_repeat < 0 ? -1 : timer->tr_repeat + 1), NULL);
Bram Moolenaarb73598e2016-08-07 18:22:53 +02001364 dict_add_nr_str(dict, "paused", (long)(timer->tr_paused), NULL);
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001365
1366 di = dictitem_alloc((char_u *)"callback");
1367 if (di != NULL)
1368 {
1369 if (dict_add(dict, di) == FAIL)
1370 vim_free(di);
1371 else if (timer->tr_partial != NULL)
1372 {
1373 di->di_tv.v_type = VAR_PARTIAL;
1374 di->di_tv.vval.v_partial = timer->tr_partial;
1375 ++timer->tr_partial->pt_refcount;
1376 }
1377 else
1378 {
1379 di->di_tv.v_type = VAR_FUNC;
1380 di->di_tv.vval.v_string = vim_strsave(timer->tr_callback);
1381 }
1382 di->di_tv.v_lock = 0;
1383 }
1384}
1385
1386 void
1387add_timer_info_all(typval_T *rettv)
1388{
1389 timer_T *timer;
1390
1391 for (timer = first_timer; timer != NULL; timer = timer->tr_next)
Bram Moolenaar75537a92016-09-05 22:45:28 +02001392 if (timer->tr_id != -1)
1393 add_timer_info(rettv, timer);
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001394}
1395
Bram Moolenaare3188e22016-05-31 21:13:04 +02001396/*
1397 * Mark references in partials of timers.
1398 */
1399 int
1400set_ref_in_timer(int copyID)
1401{
1402 int abort = FALSE;
1403 timer_T *timer;
1404 typval_T tv;
1405
1406 for (timer = first_timer; timer != NULL; timer = timer->tr_next)
1407 {
Bram Moolenaar1e96d9b2016-07-29 22:15:09 +02001408 if (timer->tr_partial != NULL)
1409 {
1410 tv.v_type = VAR_PARTIAL;
1411 tv.vval.v_partial = timer->tr_partial;
1412 }
1413 else
1414 {
1415 tv.v_type = VAR_FUNC;
1416 tv.vval.v_string = timer->tr_callback;
1417 }
Bram Moolenaare3188e22016-05-31 21:13:04 +02001418 abort = abort || set_ref_in_item(&tv, copyID, NULL, NULL);
1419 }
1420 return abort;
1421}
Bram Moolenaar623e2632016-07-30 22:47:56 +02001422
1423# if defined(EXITFREE) || defined(PROTO)
1424 void
1425timer_free_all()
1426{
1427 timer_T *timer;
1428
1429 while (first_timer != NULL)
1430 {
1431 timer = first_timer;
1432 remove_timer(timer);
1433 free_timer(timer);
1434 }
1435}
1436# endif
Bram Moolenaar975b5272016-03-15 23:10:59 +01001437# endif
1438
Bram Moolenaar8a7f5a22013-06-06 14:01:46 +02001439#if defined(FEAT_SYN_HL) && defined(FEAT_RELTIME) && defined(FEAT_FLOAT)
1440# if defined(HAVE_MATH_H)
1441# include <math.h>
1442# endif
1443
1444/*
1445 * Divide the time "tm" by "count" and store in "tm2".
1446 */
1447 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001448profile_divide(proftime_T *tm, int count, proftime_T *tm2)
Bram Moolenaar8a7f5a22013-06-06 14:01:46 +02001449{
1450 if (count == 0)
1451 profile_zero(tm2);
1452 else
1453 {
1454# ifdef WIN3264
1455 tm2->QuadPart = tm->QuadPart / count;
1456# else
1457 double usec = (tm->tv_sec * 1000000.0 + tm->tv_usec) / count;
1458
1459 tm2->tv_sec = floor(usec / 1000000.0);
Bram Moolenaara2e14fc2013-06-10 20:10:44 +02001460 tm2->tv_usec = vim_round(usec - (tm2->tv_sec * 1000000.0));
Bram Moolenaar8a7f5a22013-06-06 14:01:46 +02001461# endif
1462 }
1463}
1464#endif
1465
Bram Moolenaar76929292008-01-06 19:07:36 +00001466# if defined(FEAT_PROFILE) || defined(PROTO)
1467/*
1468 * Functions for profiling.
1469 */
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01001470static void script_do_profile(scriptitem_T *si);
1471static void script_dump_profile(FILE *fd);
Bram Moolenaar76929292008-01-06 19:07:36 +00001472static proftime_T prof_wait_time;
1473
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001474/*
Bram Moolenaar05159a02005-02-26 23:04:13 +00001475 * Add the time "tm2" to "tm".
1476 */
1477 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001478profile_add(proftime_T *tm, proftime_T *tm2)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001479{
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001480# ifdef WIN3264
1481 tm->QuadPart += tm2->QuadPart;
1482# else
Bram Moolenaar05159a02005-02-26 23:04:13 +00001483 tm->tv_usec += tm2->tv_usec;
1484 tm->tv_sec += tm2->tv_sec;
1485 if (tm->tv_usec >= 1000000)
1486 {
1487 tm->tv_usec -= 1000000;
1488 ++tm->tv_sec;
1489 }
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001490# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00001491}
1492
1493/*
Bram Moolenaar1056d982006-03-09 22:37:52 +00001494 * Add the "self" time from the total time and the children's time.
1495 */
1496 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001497profile_self(proftime_T *self, proftime_T *total, proftime_T *children)
Bram Moolenaar1056d982006-03-09 22:37:52 +00001498{
1499 /* Check that the result won't be negative. Can happen with recursive
1500 * calls. */
1501#ifdef WIN3264
1502 if (total->QuadPart <= children->QuadPart)
1503 return;
1504#else
1505 if (total->tv_sec < children->tv_sec
1506 || (total->tv_sec == children->tv_sec
1507 && total->tv_usec <= children->tv_usec))
1508 return;
1509#endif
1510 profile_add(self, total);
1511 profile_sub(self, children);
1512}
1513
1514/*
Bram Moolenaar05159a02005-02-26 23:04:13 +00001515 * Get the current waittime.
1516 */
1517 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001518profile_get_wait(proftime_T *tm)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001519{
1520 *tm = prof_wait_time;
1521}
1522
1523/*
1524 * Subtract the passed waittime since "tm" from "tma".
1525 */
1526 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001527profile_sub_wait(proftime_T *tm, proftime_T *tma)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001528{
1529 proftime_T tm3 = prof_wait_time;
1530
1531 profile_sub(&tm3, tm);
1532 profile_sub(tma, &tm3);
1533}
1534
1535/*
1536 * Return TRUE if "tm1" and "tm2" are equal.
1537 */
1538 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001539profile_equal(proftime_T *tm1, proftime_T *tm2)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001540{
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001541# ifdef WIN3264
1542 return (tm1->QuadPart == tm2->QuadPart);
1543# else
Bram Moolenaar05159a02005-02-26 23:04:13 +00001544 return (tm1->tv_usec == tm2->tv_usec && tm1->tv_sec == tm2->tv_sec);
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001545# endif
1546}
1547
1548/*
1549 * Return <0, 0 or >0 if "tm1" < "tm2", "tm1" == "tm2" or "tm1" > "tm2"
1550 */
1551 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001552profile_cmp(const proftime_T *tm1, const proftime_T *tm2)
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001553{
1554# ifdef WIN3264
1555 return (int)(tm2->QuadPart - tm1->QuadPart);
1556# else
1557 if (tm1->tv_sec == tm2->tv_sec)
1558 return tm2->tv_usec - tm1->tv_usec;
1559 return tm2->tv_sec - tm1->tv_sec;
1560# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00001561}
1562
Bram Moolenaar05159a02005-02-26 23:04:13 +00001563static char_u *profile_fname = NULL;
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001564static proftime_T pause_time;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001565
1566/*
1567 * ":profile cmd args"
1568 */
1569 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001570ex_profile(exarg_T *eap)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001571{
1572 char_u *e;
1573 int len;
1574
1575 e = skiptowhite(eap->arg);
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00001576 len = (int)(e - eap->arg);
Bram Moolenaar05159a02005-02-26 23:04:13 +00001577 e = skipwhite(e);
1578
1579 if (len == 5 && STRNCMP(eap->arg, "start", 5) == 0 && *e != NUL)
1580 {
1581 vim_free(profile_fname);
Bram Moolenaard94682f2015-04-13 15:37:56 +02001582 profile_fname = expand_env_save_opt(e, TRUE);
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001583 do_profiling = PROF_YES;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001584 profile_zero(&prof_wait_time);
1585 set_vim_var_nr(VV_PROFILING, 1L);
1586 }
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001587 else if (do_profiling == PROF_NONE)
Bram Moolenaar54c1b492010-02-24 14:01:28 +01001588 EMSG(_("E750: First use \":profile start {fname}\""));
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001589 else if (STRCMP(eap->arg, "pause") == 0)
1590 {
1591 if (do_profiling == PROF_YES)
1592 profile_start(&pause_time);
1593 do_profiling = PROF_PAUSED;
1594 }
1595 else if (STRCMP(eap->arg, "continue") == 0)
1596 {
1597 if (do_profiling == PROF_PAUSED)
1598 {
1599 profile_end(&pause_time);
1600 profile_add(&prof_wait_time, &pause_time);
1601 }
1602 do_profiling = PROF_YES;
1603 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00001604 else
1605 {
1606 /* The rest is similar to ":breakadd". */
1607 ex_breakadd(eap);
1608 }
1609}
1610
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001611/* Command line expansion for :profile. */
1612static enum
1613{
1614 PEXP_SUBCMD, /* expand :profile sub-commands */
Bram Moolenaar49789dc2011-02-25 14:46:09 +01001615 PEXP_FUNC /* expand :profile func {funcname} */
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001616} pexpand_what;
1617
1618static char *pexpand_cmds[] = {
1619 "start",
1620#define PROFCMD_START 0
1621 "pause",
1622#define PROFCMD_PAUSE 1
1623 "continue",
1624#define PROFCMD_CONTINUE 2
1625 "func",
1626#define PROFCMD_FUNC 3
1627 "file",
1628#define PROFCMD_FILE 4
1629 NULL
1630#define PROFCMD_LAST 5
1631};
1632
1633/*
1634 * Function given to ExpandGeneric() to obtain the profile command
1635 * specific expansion.
1636 */
1637 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001638get_profile_name(expand_T *xp UNUSED, int idx)
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001639{
1640 switch (pexpand_what)
1641 {
1642 case PEXP_SUBCMD:
1643 return (char_u *)pexpand_cmds[idx];
1644 /* case PEXP_FUNC: TODO */
1645 default:
1646 return NULL;
1647 }
1648}
1649
1650/*
1651 * Handle command line completion for :profile command.
1652 */
1653 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001654set_context_in_profile_cmd(expand_T *xp, char_u *arg)
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001655{
1656 char_u *end_subcmd;
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001657
1658 /* Default: expand subcommands. */
1659 xp->xp_context = EXPAND_PROFILE;
1660 pexpand_what = PEXP_SUBCMD;
1661 xp->xp_pattern = arg;
1662
1663 end_subcmd = skiptowhite(arg);
1664 if (*end_subcmd == NUL)
1665 return;
1666
Bram Moolenaar8b9c05f2010-03-02 17:54:33 +01001667 if (end_subcmd - arg == 5 && STRNCMP(arg, "start", 5) == 0)
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001668 {
1669 xp->xp_context = EXPAND_FILES;
1670 xp->xp_pattern = skipwhite(end_subcmd);
1671 return;
1672 }
1673
1674 /* TODO: expand function names after "func" */
1675 xp->xp_context = EXPAND_NOTHING;
1676}
1677
Bram Moolenaar05159a02005-02-26 23:04:13 +00001678/*
1679 * Dump the profiling info.
1680 */
1681 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001682profile_dump(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001683{
1684 FILE *fd;
1685
1686 if (profile_fname != NULL)
1687 {
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +00001688 fd = mch_fopen((char *)profile_fname, "w");
Bram Moolenaar05159a02005-02-26 23:04:13 +00001689 if (fd == NULL)
1690 EMSG2(_(e_notopen), profile_fname);
1691 else
1692 {
Bram Moolenaar05159a02005-02-26 23:04:13 +00001693 script_dump_profile(fd);
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001694 func_dump_profile(fd);
Bram Moolenaar05159a02005-02-26 23:04:13 +00001695 fclose(fd);
1696 }
1697 }
1698}
1699
1700/*
1701 * Start profiling script "fp".
1702 */
1703 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001704script_do_profile(scriptitem_T *si)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001705{
1706 si->sn_pr_count = 0;
1707 profile_zero(&si->sn_pr_total);
1708 profile_zero(&si->sn_pr_self);
1709
1710 ga_init2(&si->sn_prl_ga, sizeof(sn_prl_T), 100);
1711 si->sn_prl_idx = -1;
1712 si->sn_prof_on = TRUE;
1713 si->sn_pr_nest = 0;
1714}
1715
1716/*
Bram Moolenaar67435d92017-10-19 21:04:37 +02001717 * Save time when starting to invoke another script or function.
Bram Moolenaar05159a02005-02-26 23:04:13 +00001718 */
1719 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001720script_prof_save(
1721 proftime_T *tm) /* place to store wait time */
Bram Moolenaar05159a02005-02-26 23:04:13 +00001722{
1723 scriptitem_T *si;
1724
1725 if (current_SID > 0 && current_SID <= script_items.ga_len)
1726 {
1727 si = &SCRIPT_ITEM(current_SID);
1728 if (si->sn_prof_on && si->sn_pr_nest++ == 0)
1729 profile_start(&si->sn_pr_child);
1730 }
1731 profile_get_wait(tm);
1732}
1733
1734/*
1735 * Count time spent in children after invoking another script or function.
1736 */
1737 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001738script_prof_restore(proftime_T *tm)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001739{
1740 scriptitem_T *si;
1741
1742 if (current_SID > 0 && current_SID <= script_items.ga_len)
1743 {
1744 si = &SCRIPT_ITEM(current_SID);
1745 if (si->sn_prof_on && --si->sn_pr_nest == 0)
1746 {
1747 profile_end(&si->sn_pr_child);
1748 profile_sub_wait(tm, &si->sn_pr_child); /* don't count wait time */
1749 profile_add(&si->sn_pr_children, &si->sn_pr_child);
1750 profile_add(&si->sn_prl_children, &si->sn_pr_child);
1751 }
1752 }
1753}
1754
1755static proftime_T inchar_time;
1756
1757/*
1758 * Called when starting to wait for the user to type a character.
1759 */
1760 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001761prof_inchar_enter(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001762{
1763 profile_start(&inchar_time);
1764}
1765
1766/*
1767 * Called when finished waiting for the user to type a character.
1768 */
1769 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001770prof_inchar_exit(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001771{
1772 profile_end(&inchar_time);
1773 profile_add(&prof_wait_time, &inchar_time);
1774}
1775
1776/*
1777 * Dump the profiling results for all scripts in file "fd".
1778 */
1779 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001780script_dump_profile(FILE *fd)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001781{
1782 int id;
1783 scriptitem_T *si;
1784 int i;
1785 FILE *sfd;
1786 sn_prl_T *pp;
1787
1788 for (id = 1; id <= script_items.ga_len; ++id)
1789 {
1790 si = &SCRIPT_ITEM(id);
1791 if (si->sn_prof_on)
1792 {
1793 fprintf(fd, "SCRIPT %s\n", si->sn_name);
1794 if (si->sn_pr_count == 1)
1795 fprintf(fd, "Sourced 1 time\n");
1796 else
1797 fprintf(fd, "Sourced %d times\n", si->sn_pr_count);
1798 fprintf(fd, "Total time: %s\n", profile_msg(&si->sn_pr_total));
1799 fprintf(fd, " Self time: %s\n", profile_msg(&si->sn_pr_self));
1800 fprintf(fd, "\n");
1801 fprintf(fd, "count total (s) self (s)\n");
1802
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +00001803 sfd = mch_fopen((char *)si->sn_name, "r");
Bram Moolenaar05159a02005-02-26 23:04:13 +00001804 if (sfd == NULL)
1805 fprintf(fd, "Cannot open file!\n");
1806 else
1807 {
Bram Moolenaar67435d92017-10-19 21:04:37 +02001808 /* Keep going till the end of file, so that trailing
1809 * continuation lines are listed. */
1810 for (i = 0; ; ++i)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001811 {
1812 if (vim_fgets(IObuff, IOSIZE, sfd))
1813 break;
Bram Moolenaar67435d92017-10-19 21:04:37 +02001814 if (i < si->sn_prl_ga.ga_len
1815 && (pp = &PRL_ITEM(si, i))->snp_count > 0)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001816 {
1817 fprintf(fd, "%5d ", pp->snp_count);
1818 if (profile_equal(&pp->sn_prl_total, &pp->sn_prl_self))
1819 fprintf(fd, " ");
1820 else
1821 fprintf(fd, "%s ", profile_msg(&pp->sn_prl_total));
1822 fprintf(fd, "%s ", profile_msg(&pp->sn_prl_self));
1823 }
1824 else
1825 fprintf(fd, " ");
1826 fprintf(fd, "%s", IObuff);
1827 }
1828 fclose(sfd);
1829 }
1830 fprintf(fd, "\n");
1831 }
1832 }
1833}
1834
1835/*
1836 * Return TRUE when a function defined in the current script should be
1837 * profiled.
1838 */
1839 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001840prof_def_func(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001841{
Bram Moolenaar53180ce2005-07-05 21:48:14 +00001842 if (current_SID > 0)
1843 return SCRIPT_ITEM(current_SID).sn_pr_force;
1844 return FALSE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001845}
1846
1847# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001848#endif
1849
1850/*
1851 * If 'autowrite' option set, try to write the file.
1852 * Careful: autocommands may make "buf" invalid!
1853 *
1854 * return FAIL for failure, OK otherwise
1855 */
1856 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001857autowrite(buf_T *buf, int forceit)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001858{
Bram Moolenaar373154b2007-02-13 05:19:30 +00001859 int r;
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001860 bufref_T bufref;
Bram Moolenaar373154b2007-02-13 05:19:30 +00001861
Bram Moolenaar071d4272004-06-13 20:20:40 +00001862 if (!(p_aw || p_awa) || !p_write
1863#ifdef FEAT_QUICKFIX
Bram Moolenaar373154b2007-02-13 05:19:30 +00001864 /* never autowrite a "nofile" or "nowrite" buffer */
1865 || bt_dontwrite(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001866#endif
Bram Moolenaar373154b2007-02-13 05:19:30 +00001867 || (!forceit && buf->b_p_ro) || buf->b_ffname == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001868 return FAIL;
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001869 set_bufref(&bufref, buf);
Bram Moolenaar373154b2007-02-13 05:19:30 +00001870 r = buf_write_all(buf, forceit);
1871
1872 /* Writing may succeed but the buffer still changed, e.g., when there is a
1873 * conversion error. We do want to return FAIL then. */
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001874 if (bufref_valid(&bufref) && bufIsChanged(buf))
Bram Moolenaar373154b2007-02-13 05:19:30 +00001875 r = FAIL;
1876 return r;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001877}
1878
1879/*
1880 * flush all buffers, except the ones that are readonly
1881 */
1882 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001883autowrite_all(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001884{
1885 buf_T *buf;
1886
1887 if (!(p_aw || p_awa) || !p_write)
1888 return;
Bram Moolenaar29323592016-07-24 22:04:11 +02001889 FOR_ALL_BUFFERS(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001890 if (bufIsChanged(buf) && !buf->b_p_ro)
1891 {
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001892#ifdef FEAT_AUTOCMD
1893 bufref_T bufref;
1894
1895 set_bufref(&bufref, buf);
1896#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001897 (void)buf_write_all(buf, FALSE);
1898#ifdef FEAT_AUTOCMD
1899 /* an autocommand may have deleted the buffer */
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001900 if (!bufref_valid(&bufref))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001901 buf = firstbuf;
1902#endif
1903 }
1904}
1905
1906/*
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001907 * Return TRUE if buffer was changed and cannot be abandoned.
1908 * For flags use the CCGD_ values.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001909 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001910 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001911check_changed(buf_T *buf, int flags)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001912{
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001913 int forceit = (flags & CCGD_FORCEIT);
1914#ifdef FEAT_AUTOCMD
1915 bufref_T bufref;
1916
1917 set_bufref(&bufref, buf);
1918#endif
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001919
Bram Moolenaar071d4272004-06-13 20:20:40 +00001920 if ( !forceit
1921 && bufIsChanged(buf)
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001922 && ((flags & CCGD_MULTWIN) || buf->b_nwindows <= 1)
1923 && (!(flags & CCGD_AW) || autowrite(buf, forceit) == FAIL))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001924 {
1925#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG)
1926 if ((p_confirm || cmdmod.confirm) && p_write)
1927 {
1928 buf_T *buf2;
1929 int count = 0;
1930
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001931 if (flags & CCGD_ALLBUF)
Bram Moolenaar29323592016-07-24 22:04:11 +02001932 FOR_ALL_BUFFERS(buf2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001933 if (bufIsChanged(buf2)
1934 && (buf2->b_ffname != NULL
1935# ifdef FEAT_BROWSE
1936 || cmdmod.browse
1937# endif
1938 ))
1939 ++count;
1940# ifdef FEAT_AUTOCMD
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001941 if (!bufref_valid(&bufref))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001942 /* Autocommand deleted buffer, oops! It's not changed now. */
1943 return FALSE;
1944# endif
1945 dialog_changed(buf, count > 1);
1946# ifdef FEAT_AUTOCMD
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001947 if (!bufref_valid(&bufref))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001948 /* Autocommand deleted buffer, oops! It's not changed now. */
1949 return FALSE;
1950# endif
1951 return bufIsChanged(buf);
1952 }
1953#endif
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001954 if (flags & CCGD_EXCMD)
Bram Moolenaarf5be7cd2017-08-17 16:55:13 +02001955 no_write_message();
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001956 else
Bram Moolenaarf5be7cd2017-08-17 16:55:13 +02001957 no_write_message_nobang();
Bram Moolenaar071d4272004-06-13 20:20:40 +00001958 return TRUE;
1959 }
1960 return FALSE;
1961}
1962
1963#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG) || defined(PROTO)
1964
1965#if defined(FEAT_BROWSE) || defined(PROTO)
1966/*
1967 * When wanting to write a file without a file name, ask the user for a name.
1968 */
1969 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001970browse_save_fname(buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001971{
1972 if (buf->b_fname == NULL)
1973 {
1974 char_u *fname;
1975
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00001976 fname = do_browse(BROWSE_SAVE, (char_u *)_("Save As"),
1977 NULL, NULL, NULL, NULL, buf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001978 if (fname != NULL)
1979 {
1980 if (setfname(buf, fname, NULL, TRUE) == OK)
1981 buf->b_flags |= BF_NOTEDITED;
1982 vim_free(fname);
1983 }
1984 }
1985}
1986#endif
1987
1988/*
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02001989 * Ask the user what to do when abandoning a changed buffer.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001990 * Must check 'write' option first!
1991 */
1992 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001993dialog_changed(
1994 buf_T *buf,
1995 int checkall) /* may abandon all changed buffers */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001996{
Bram Moolenaard9462e32011-04-11 21:35:11 +02001997 char_u buff[DIALOG_MSG_SIZE];
Bram Moolenaar071d4272004-06-13 20:20:40 +00001998 int ret;
1999 buf_T *buf2;
Bram Moolenaar8218f602012-04-25 17:32:18 +02002000 exarg_T ea;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002001
Bram Moolenaar3f9a1ff2017-08-21 22:06:02 +02002002 dialog_msg(buff, _("Save changes to \"%s\"?"), buf->b_fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002003 if (checkall)
2004 ret = vim_dialog_yesnoallcancel(VIM_QUESTION, NULL, buff, 1);
2005 else
2006 ret = vim_dialog_yesnocancel(VIM_QUESTION, NULL, buff, 1);
2007
Bram Moolenaar8218f602012-04-25 17:32:18 +02002008 /* Init ea pseudo-structure, this is needed for the check_overwrite()
2009 * function. */
2010 ea.append = ea.forceit = FALSE;
2011
Bram Moolenaar071d4272004-06-13 20:20:40 +00002012 if (ret == VIM_YES)
2013 {
2014#ifdef FEAT_BROWSE
2015 /* May get file name, when there is none */
2016 browse_save_fname(buf);
2017#endif
Bram Moolenaar8218f602012-04-25 17:32:18 +02002018 if (buf->b_fname != NULL && check_overwrite(&ea, buf,
2019 buf->b_fname, buf->b_ffname, FALSE) == OK)
2020 /* didn't hit Cancel */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002021 (void)buf_write_all(buf, FALSE);
2022 }
2023 else if (ret == VIM_NO)
2024 {
2025 unchanged(buf, TRUE);
2026 }
2027 else if (ret == VIM_ALL)
2028 {
2029 /*
2030 * Write all modified files that can be written.
2031 * Skip readonly buffers, these need to be confirmed
2032 * individually.
2033 */
Bram Moolenaar29323592016-07-24 22:04:11 +02002034 FOR_ALL_BUFFERS(buf2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002035 {
2036 if (bufIsChanged(buf2)
2037 && (buf2->b_ffname != NULL
2038#ifdef FEAT_BROWSE
2039 || cmdmod.browse
2040#endif
2041 )
2042 && !buf2->b_p_ro)
2043 {
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002044#ifdef FEAT_AUTOCMD
2045 bufref_T bufref;
2046
2047 set_bufref(&bufref, buf2);
2048#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002049#ifdef FEAT_BROWSE
2050 /* May get file name, when there is none */
2051 browse_save_fname(buf2);
2052#endif
Bram Moolenaar8218f602012-04-25 17:32:18 +02002053 if (buf2->b_fname != NULL && check_overwrite(&ea, buf2,
2054 buf2->b_fname, buf2->b_ffname, FALSE) == OK)
2055 /* didn't hit Cancel */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002056 (void)buf_write_all(buf2, FALSE);
2057#ifdef FEAT_AUTOCMD
2058 /* an autocommand may have deleted the buffer */
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002059 if (!bufref_valid(&bufref))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002060 buf2 = firstbuf;
2061#endif
2062 }
2063 }
2064 }
2065 else if (ret == VIM_DISCARDALL)
2066 {
2067 /*
2068 * mark all buffers as unchanged
2069 */
Bram Moolenaar29323592016-07-24 22:04:11 +02002070 FOR_ALL_BUFFERS(buf2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002071 unchanged(buf2, TRUE);
2072 }
2073}
2074#endif
2075
2076/*
2077 * Return TRUE if the buffer "buf" can be abandoned, either by making it
2078 * hidden, autowriting it or unloading it.
2079 */
2080 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002081can_abandon(buf_T *buf, int forceit)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002082{
Bram Moolenaareb44a682017-08-03 22:44:55 +02002083 return ( buf_hide(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002084 || !bufIsChanged(buf)
2085 || buf->b_nwindows > 1
2086 || autowrite(buf, forceit) == OK
2087 || forceit);
2088}
2089
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01002090static void add_bufnum(int *bufnrs, int *bufnump, int nr);
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002091
2092/*
2093 * Add a buffer number to "bufnrs", unless it's already there.
2094 */
2095 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002096add_bufnum(int *bufnrs, int *bufnump, int nr)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002097{
2098 int i;
2099
2100 for (i = 0; i < *bufnump; ++i)
2101 if (bufnrs[i] == nr)
2102 return;
2103 bufnrs[*bufnump] = nr;
2104 *bufnump = *bufnump + 1;
2105}
2106
Bram Moolenaar071d4272004-06-13 20:20:40 +00002107/*
2108 * Return TRUE if any buffer was changed and cannot be abandoned.
2109 * That changed buffer becomes the current buffer.
Bram Moolenaar027387f2016-01-02 22:25:52 +01002110 * When "unload" is true the current buffer is unloaded instead of making it
2111 * hidden. This is used for ":q!".
Bram Moolenaar071d4272004-06-13 20:20:40 +00002112 */
2113 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002114check_changed_any(
2115 int hidden, /* Only check hidden buffers */
2116 int unload)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002117{
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002118 int ret = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002119 buf_T *buf;
2120 int save;
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002121 int i;
2122 int bufnum = 0;
2123 int bufcount = 0;
2124 int *bufnrs;
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002125 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002126 win_T *wp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002127
Bram Moolenaar29323592016-07-24 22:04:11 +02002128 FOR_ALL_BUFFERS(buf)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002129 ++bufcount;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002130
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002131 if (bufcount == 0)
2132 return FALSE;
2133
2134 bufnrs = (int *)alloc(sizeof(int) * bufcount);
2135 if (bufnrs == NULL)
2136 return FALSE;
2137
2138 /* curbuf */
2139 bufnrs[bufnum++] = curbuf->b_fnum;
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002140 /* buf in curtab */
2141 FOR_ALL_WINDOWS(wp)
2142 if (wp->w_buffer != curbuf)
2143 add_bufnum(bufnrs, &bufnum, wp->w_buffer->b_fnum);
2144
2145 /* buf in other tab */
Bram Moolenaar29323592016-07-24 22:04:11 +02002146 FOR_ALL_TABPAGES(tp)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002147 if (tp != curtab)
2148 for (wp = tp->tp_firstwin; wp != NULL; wp = wp->w_next)
2149 add_bufnum(bufnrs, &bufnum, wp->w_buffer->b_fnum);
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002150 /* any other buf */
Bram Moolenaar29323592016-07-24 22:04:11 +02002151 FOR_ALL_BUFFERS(buf)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002152 add_bufnum(bufnrs, &bufnum, buf->b_fnum);
2153
2154 for (i = 0; i < bufnum; ++i)
2155 {
2156 buf = buflist_findnr(bufnrs[i]);
2157 if (buf == NULL)
2158 continue;
2159 if ((!hidden || buf->b_nwindows == 0) && bufIsChanged(buf))
2160 {
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002161 bufref_T bufref;
2162
2163 set_bufref(&bufref, buf);
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002164 /* Try auto-writing the buffer. If this fails but the buffer no
2165 * longer exists it's not changed, that's OK. */
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002166 if (check_changed(buf, (p_awa ? CCGD_AW : 0)
2167 | CCGD_MULTWIN
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002168 | CCGD_ALLBUF) && bufref_valid(&bufref))
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002169 break; /* didn't save - still changes */
2170 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002171 }
2172
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002173 if (i >= bufnum)
2174 goto theend;
2175
2176 ret = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002177 exiting = FALSE;
2178#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG)
2179 /*
2180 * When ":confirm" used, don't give an error message.
2181 */
2182 if (!(p_confirm || cmdmod.confirm))
2183#endif
2184 {
2185 /* There must be a wait_return for this message, do_buffer()
2186 * may cause a redraw. But wait_return() is a no-op when vgetc()
2187 * is busy (Quit used from window menu), then make sure we don't
2188 * cause a scroll up. */
Bram Moolenaar61660ea2006-04-07 21:40:07 +00002189 if (vgetc_busy > 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002190 {
2191 msg_row = cmdline_row;
2192 msg_col = 0;
2193 msg_didout = FALSE;
2194 }
Bram Moolenaareb44a682017-08-03 22:44:55 +02002195 if (
2196#ifdef FEAT_TERMINAL
2197 term_job_running(buf->b_term)
2198 ? EMSG2(_("E947: Job still running in buffer \"%s\""),
2199 buf->b_fname)
2200 :
2201#endif
2202 EMSG2(_("E162: No write since last change for buffer \"%s\""),
Bram Moolenaare1704ba2012-10-03 18:25:00 +02002203 buf_spname(buf) != NULL ? buf_spname(buf) : buf->b_fname))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002204 {
2205 save = no_wait_return;
2206 no_wait_return = FALSE;
2207 wait_return(FALSE);
2208 no_wait_return = save;
2209 }
2210 }
2211
Bram Moolenaar071d4272004-06-13 20:20:40 +00002212 /* Try to find a window that contains the buffer. */
2213 if (buf != curbuf)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002214 FOR_ALL_TAB_WINDOWS(tp, wp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002215 if (wp->w_buffer == buf)
2216 {
Bram Moolenaar4033c552017-09-16 20:54:51 +02002217#ifdef FEAT_AUTOCMD
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002218 bufref_T bufref;
2219
2220 set_bufref(&bufref, buf);
Bram Moolenaar4033c552017-09-16 20:54:51 +02002221#endif
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002222 goto_tabpage_win(tp, wp);
Bram Moolenaar4033c552017-09-16 20:54:51 +02002223#ifdef FEAT_AUTOCMD
Bram Moolenaar071d4272004-06-13 20:20:40 +00002224 /* Paranoia: did autocms wipe out the buffer with changes? */
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002225 if (!bufref_valid(&bufref))
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002226 {
2227 goto theend;
2228 }
Bram Moolenaar4033c552017-09-16 20:54:51 +02002229#endif
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002230 goto buf_found;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002231 }
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002232buf_found:
Bram Moolenaar071d4272004-06-13 20:20:40 +00002233
2234 /* Open the changed buffer in the current window. */
2235 if (buf != curbuf)
Bram Moolenaar027387f2016-01-02 22:25:52 +01002236 set_curbuf(buf, unload ? DOBUF_UNLOAD : DOBUF_GOTO);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002237
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002238theend:
2239 vim_free(bufnrs);
2240 return ret;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002241}
2242
2243/*
2244 * return FAIL if there is no file name, OK if there is one
2245 * give error message for FAIL
2246 */
2247 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002248check_fname(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002249{
2250 if (curbuf->b_ffname == NULL)
2251 {
2252 EMSG(_(e_noname));
2253 return FAIL;
2254 }
2255 return OK;
2256}
2257
2258/*
2259 * flush the contents of a buffer, unless it has no file name
2260 *
2261 * return FAIL for failure, OK otherwise
2262 */
2263 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002264buf_write_all(buf_T *buf, int forceit)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002265{
2266 int retval;
2267#ifdef FEAT_AUTOCMD
2268 buf_T *old_curbuf = curbuf;
2269#endif
2270
2271 retval = (buf_write(buf, buf->b_ffname, buf->b_fname,
2272 (linenr_T)1, buf->b_ml.ml_line_count, NULL,
2273 FALSE, forceit, TRUE, FALSE));
2274#ifdef FEAT_AUTOCMD
2275 if (curbuf != old_curbuf)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00002276 {
Bram Moolenaar8820b482017-03-16 17:23:31 +01002277 msg_source(HL_ATTR(HLF_W));
Bram Moolenaar071d4272004-06-13 20:20:40 +00002278 MSG(_("Warning: Entered other buffer unexpectedly (check autocommands)"));
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00002279 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002280#endif
2281 return retval;
2282}
2283
2284/*
2285 * Code to handle the argument list.
2286 */
2287
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01002288static char_u *do_one_arg(char_u *str);
2289static int do_arglist(char_u *str, int what, int after);
2290static void alist_check_arg_idx(void);
2291static int editing_arg_idx(win_T *win);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002292#ifdef FEAT_LISTCMDS
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01002293static int alist_add_list(int count, char_u **files, int after);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002294#endif
2295#define AL_SET 1
2296#define AL_ADD 2
2297#define AL_DEL 3
2298
Bram Moolenaar071d4272004-06-13 20:20:40 +00002299/*
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002300 * Isolate one argument, taking backticks.
2301 * Changes the argument in-place, puts a NUL after it. Backticks remain.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002302 * Return a pointer to the start of the next argument.
2303 */
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002304 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002305do_one_arg(char_u *str)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002306{
2307 char_u *p;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002308 int inbacktick;
2309
Bram Moolenaar071d4272004-06-13 20:20:40 +00002310 inbacktick = FALSE;
2311 for (p = str; *str; ++str)
2312 {
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002313 /* When the backslash is used for escaping the special meaning of a
2314 * character we need to keep it until wildcard expansion. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002315 if (rem_backslash(str))
2316 {
2317 *p++ = *str++;
2318 *p++ = *str;
2319 }
2320 else
2321 {
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002322 /* An item ends at a space not in backticks */
2323 if (!inbacktick && vim_isspace(*str))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002324 break;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002325 if (*str == '`')
Bram Moolenaar071d4272004-06-13 20:20:40 +00002326 inbacktick ^= TRUE;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002327 *p++ = *str;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002328 }
2329 }
2330 str = skipwhite(str);
2331 *p = NUL;
2332
2333 return str;
2334}
2335
Bram Moolenaar86b68352004-12-27 21:59:20 +00002336/*
2337 * Separate the arguments in "str" and return a list of pointers in the
2338 * growarray "gap".
2339 */
Bram Moolenaar398ee732017-08-03 14:29:14 +02002340 static int
2341get_arglist(garray_T *gap, char_u *str, int escaped)
Bram Moolenaar86b68352004-12-27 21:59:20 +00002342{
2343 ga_init2(gap, (int)sizeof(char_u *), 20);
2344 while (*str != NUL)
2345 {
2346 if (ga_grow(gap, 1) == FAIL)
2347 {
2348 ga_clear(gap);
2349 return FAIL;
2350 }
2351 ((char_u **)gap->ga_data)[gap->ga_len++] = str;
2352
Bram Moolenaar398ee732017-08-03 14:29:14 +02002353 /* If str is escaped, don't handle backslashes or spaces */
2354 if (!escaped)
2355 return OK;
2356
Bram Moolenaar86b68352004-12-27 21:59:20 +00002357 /* Isolate one argument, change it in-place, put a NUL after it. */
2358 str = do_one_arg(str);
2359 }
2360 return OK;
2361}
2362
Bram Moolenaar7df351e2006-01-23 22:30:28 +00002363#if defined(FEAT_QUICKFIX) || defined(FEAT_SYN_HL) || defined(PROTO)
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002364/*
2365 * Parse a list of arguments (file names), expand them and return in
Bram Moolenaar8f5c6f02012-06-29 12:57:06 +02002366 * "fnames[fcountp]". When "wig" is TRUE, removes files matching 'wildignore'.
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002367 * Return FAIL or OK.
2368 */
2369 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002370get_arglist_exp(
2371 char_u *str,
2372 int *fcountp,
2373 char_u ***fnamesp,
2374 int wig)
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002375{
2376 garray_T ga;
2377 int i;
2378
Bram Moolenaar398ee732017-08-03 14:29:14 +02002379 if (get_arglist(&ga, str, TRUE) == FAIL)
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002380 return FAIL;
Bram Moolenaar8f5c6f02012-06-29 12:57:06 +02002381 if (wig == TRUE)
2382 i = expand_wildcards(ga.ga_len, (char_u **)ga.ga_data,
2383 fcountp, fnamesp, EW_FILE|EW_NOTFOUND);
2384 else
2385 i = gen_expand_wildcards(ga.ga_len, (char_u **)ga.ga_data,
2386 fcountp, fnamesp, EW_FILE|EW_NOTFOUND);
2387
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002388 ga_clear(&ga);
2389 return i;
2390}
2391#endif
2392
Bram Moolenaar071d4272004-06-13 20:20:40 +00002393#if defined(FEAT_GUI) || defined(FEAT_CLIENTSERVER) || defined(PROTO)
2394/*
2395 * Redefine the argument list.
2396 */
2397 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002398set_arglist(char_u *str)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002399{
2400 do_arglist(str, AL_SET, 0);
2401}
2402#endif
2403
2404/*
2405 * "what" == AL_SET: Redefine the argument list to 'str'.
2406 * "what" == AL_ADD: add files in 'str' to the argument list after "after".
2407 * "what" == AL_DEL: remove files in 'str' from the argument list.
2408 *
2409 * Return FAIL for failure, OK otherwise.
2410 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002411 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002412do_arglist(
2413 char_u *str,
Bram Moolenaarf1d25012016-03-03 12:22:53 +01002414 int what,
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002415 int after UNUSED) /* 0 means before first one */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002416{
2417 garray_T new_ga;
2418 int exp_count;
2419 char_u **exp_files;
2420 int i;
2421#ifdef FEAT_LISTCMDS
2422 char_u *p;
2423 int match;
2424#endif
Bram Moolenaar398ee732017-08-03 14:29:14 +02002425 int arg_escaped = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002426
2427 /*
Bram Moolenaar2faa29f2016-01-23 23:02:34 +01002428 * Set default argument for ":argadd" command.
2429 */
2430 if (what == AL_ADD && *str == NUL)
2431 {
2432 if (curbuf->b_ffname == NULL)
2433 return FAIL;
2434 str = curbuf->b_fname;
Bram Moolenaar398ee732017-08-03 14:29:14 +02002435 arg_escaped = FALSE;
Bram Moolenaar2faa29f2016-01-23 23:02:34 +01002436 }
2437
2438 /*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002439 * Collect all file name arguments in "new_ga".
2440 */
Bram Moolenaar398ee732017-08-03 14:29:14 +02002441 if (get_arglist(&new_ga, str, arg_escaped) == FAIL)
Bram Moolenaar86b68352004-12-27 21:59:20 +00002442 return FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002443
2444#ifdef FEAT_LISTCMDS
2445 if (what == AL_DEL)
2446 {
2447 regmatch_T regmatch;
2448 int didone;
2449
2450 /*
2451 * Delete the items: use each item as a regexp and find a match in the
2452 * argument list.
2453 */
Bram Moolenaar71afbfe2013-03-19 16:49:16 +01002454 regmatch.rm_ic = p_fic; /* ignore case when 'fileignorecase' is set */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002455 for (i = 0; i < new_ga.ga_len && !got_int; ++i)
2456 {
2457 p = ((char_u **)new_ga.ga_data)[i];
2458 p = file_pat_to_reg_pat(p, NULL, NULL, FALSE);
2459 if (p == NULL)
2460 break;
2461 regmatch.regprog = vim_regcomp(p, p_magic ? RE_MAGIC : 0);
2462 if (regmatch.regprog == NULL)
2463 {
2464 vim_free(p);
2465 break;
2466 }
2467
2468 didone = FALSE;
2469 for (match = 0; match < ARGCOUNT; ++match)
2470 if (vim_regexec(&regmatch, alist_name(&ARGLIST[match]),
2471 (colnr_T)0))
2472 {
2473 didone = TRUE;
2474 vim_free(ARGLIST[match].ae_fname);
2475 mch_memmove(ARGLIST + match, ARGLIST + match + 1,
2476 (ARGCOUNT - match - 1) * sizeof(aentry_T));
2477 --ALIST(curwin)->al_ga.ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002478 if (curwin->w_arg_idx > match)
2479 --curwin->w_arg_idx;
2480 --match;
2481 }
2482
Bram Moolenaar473de612013-06-08 18:19:48 +02002483 vim_regfree(regmatch.regprog);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002484 vim_free(p);
2485 if (!didone)
2486 EMSG2(_(e_nomatch2), ((char_u **)new_ga.ga_data)[i]);
2487 }
2488 ga_clear(&new_ga);
2489 }
2490 else
2491#endif
2492 {
2493 i = expand_wildcards(new_ga.ga_len, (char_u **)new_ga.ga_data,
2494 &exp_count, &exp_files, EW_DIR|EW_FILE|EW_ADDSLASH|EW_NOTFOUND);
2495 ga_clear(&new_ga);
Bram Moolenaar2db5c3b2016-01-16 22:49:34 +01002496 if (i == FAIL || exp_count == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002497 {
2498 EMSG(_(e_nomatch));
2499 return FAIL;
2500 }
2501
2502#ifdef FEAT_LISTCMDS
2503 if (what == AL_ADD)
2504 {
2505 (void)alist_add_list(exp_count, exp_files, after);
2506 vim_free(exp_files);
2507 }
2508 else /* what == AL_SET */
2509#endif
Bram Moolenaar86b68352004-12-27 21:59:20 +00002510 alist_set(ALIST(curwin), exp_count, exp_files, FALSE, NULL, 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002511 }
2512
2513 alist_check_arg_idx();
2514
2515 return OK;
2516}
2517
2518/*
2519 * Check the validity of the arg_idx for each other window.
2520 */
2521 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002522alist_check_arg_idx(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002523{
Bram Moolenaar071d4272004-06-13 20:20:40 +00002524 win_T *win;
Bram Moolenaarf740b292006-02-16 22:11:02 +00002525 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002526
Bram Moolenaarf740b292006-02-16 22:11:02 +00002527 FOR_ALL_TAB_WINDOWS(tp, win)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002528 if (win->w_alist == curwin->w_alist)
2529 check_arg_idx(win);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002530}
2531
2532/*
Bram Moolenaarb8ff1fb2012-02-04 21:59:01 +01002533 * Return TRUE if window "win" is editing the file at the current argument
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002534 * index.
2535 */
2536 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002537editing_arg_idx(win_T *win)
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002538{
2539 return !(win->w_arg_idx >= WARGCOUNT(win)
2540 || (win->w_buffer->b_fnum
2541 != WARGLIST(win)[win->w_arg_idx].ae_fnum
2542 && (win->w_buffer->b_ffname == NULL
2543 || !(fullpathcmp(
2544 alist_name(&WARGLIST(win)[win->w_arg_idx]),
2545 win->w_buffer->b_ffname, TRUE) & FPC_SAME))));
2546}
2547
2548/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002549 * Check if window "win" is editing the w_arg_idx file in its argument list.
2550 */
2551 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002552check_arg_idx(win_T *win)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002553{
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002554 if (WARGCOUNT(win) > 1 && !editing_arg_idx(win))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002555 {
2556 /* We are not editing the current entry in the argument list.
2557 * Set "arg_had_last" if we are editing the last one. */
2558 win->w_arg_idx_invalid = TRUE;
2559 if (win->w_arg_idx != WARGCOUNT(win) - 1
2560 && arg_had_last == FALSE
Bram Moolenaar071d4272004-06-13 20:20:40 +00002561 && ALIST(win) == &global_alist
Bram Moolenaar071d4272004-06-13 20:20:40 +00002562 && GARGCOUNT > 0
2563 && win->w_arg_idx < GARGCOUNT
2564 && (win->w_buffer->b_fnum == GARGLIST[GARGCOUNT - 1].ae_fnum
2565 || (win->w_buffer->b_ffname != NULL
2566 && (fullpathcmp(alist_name(&GARGLIST[GARGCOUNT - 1]),
2567 win->w_buffer->b_ffname, TRUE) & FPC_SAME))))
2568 arg_had_last = TRUE;
2569 }
2570 else
2571 {
2572 /* We are editing the current entry in the argument list.
2573 * Set "arg_had_last" if it's also the last one */
2574 win->w_arg_idx_invalid = FALSE;
2575 if (win->w_arg_idx == WARGCOUNT(win) - 1
Bram Moolenaar4033c552017-09-16 20:54:51 +02002576 && win->w_alist == &global_alist)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002577 arg_had_last = TRUE;
2578 }
2579}
2580
2581/*
2582 * ":args", ":argslocal" and ":argsglobal".
2583 */
2584 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002585ex_args(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002586{
2587 int i;
2588
2589 if (eap->cmdidx != CMD_args)
2590 {
Bram Moolenaar4033c552017-09-16 20:54:51 +02002591#if defined(FEAT_LISTCMDS)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002592 alist_unlink(ALIST(curwin));
2593 if (eap->cmdidx == CMD_argglobal)
2594 ALIST(curwin) = &global_alist;
2595 else /* eap->cmdidx == CMD_arglocal */
2596 alist_new();
2597#else
2598 ex_ni(eap);
2599 return;
2600#endif
2601 }
2602
2603 if (!ends_excmd(*eap->arg))
2604 {
2605 /*
2606 * ":args file ..": define new argument list, handle like ":next"
2607 * Also for ":argslocal file .." and ":argsglobal file ..".
2608 */
2609 ex_next(eap);
2610 }
2611 else
Bram Moolenaar4033c552017-09-16 20:54:51 +02002612#if defined(FEAT_LISTCMDS)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002613 if (eap->cmdidx == CMD_args)
2614#endif
2615 {
2616 /*
2617 * ":args": list arguments.
2618 */
2619 if (ARGCOUNT > 0)
2620 {
2621 /* Overwrite the command, for a short list there is no scrolling
2622 * required and no wait_return(). */
2623 gotocmdline(TRUE);
2624 for (i = 0; i < ARGCOUNT; ++i)
2625 {
2626 if (i == curwin->w_arg_idx)
2627 msg_putchar('[');
2628 msg_outtrans(alist_name(&ARGLIST[i]));
2629 if (i == curwin->w_arg_idx)
2630 msg_putchar(']');
2631 msg_putchar(' ');
2632 }
2633 }
2634 }
Bram Moolenaar4033c552017-09-16 20:54:51 +02002635#if defined(FEAT_LISTCMDS)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002636 else if (eap->cmdidx == CMD_arglocal)
2637 {
2638 garray_T *gap = &curwin->w_alist->al_ga;
2639
2640 /*
2641 * ":argslocal": make a local copy of the global argument list.
2642 */
2643 if (ga_grow(gap, GARGCOUNT) == OK)
2644 for (i = 0; i < GARGCOUNT; ++i)
2645 if (GARGLIST[i].ae_fname != NULL)
2646 {
2647 AARGLIST(curwin->w_alist)[gap->ga_len].ae_fname =
2648 vim_strsave(GARGLIST[i].ae_fname);
2649 AARGLIST(curwin->w_alist)[gap->ga_len].ae_fnum =
2650 GARGLIST[i].ae_fnum;
2651 ++gap->ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002652 }
2653 }
2654#endif
2655}
2656
2657/*
2658 * ":previous", ":sprevious", ":Next" and ":sNext".
2659 */
2660 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002661ex_previous(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002662{
2663 /* If past the last one already, go to the last one. */
2664 if (curwin->w_arg_idx - (int)eap->line2 >= ARGCOUNT)
2665 do_argfile(eap, ARGCOUNT - 1);
2666 else
2667 do_argfile(eap, curwin->w_arg_idx - (int)eap->line2);
2668}
2669
2670/*
2671 * ":rewind", ":first", ":sfirst" and ":srewind".
2672 */
2673 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002674ex_rewind(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002675{
2676 do_argfile(eap, 0);
2677}
2678
2679/*
2680 * ":last" and ":slast".
2681 */
2682 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002683ex_last(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002684{
2685 do_argfile(eap, ARGCOUNT - 1);
2686}
2687
2688/*
2689 * ":argument" and ":sargument".
2690 */
2691 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002692ex_argument(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002693{
2694 int i;
2695
2696 if (eap->addr_count > 0)
2697 i = eap->line2 - 1;
2698 else
2699 i = curwin->w_arg_idx;
2700 do_argfile(eap, i);
2701}
2702
2703/*
2704 * Edit file "argn" of the argument lists.
2705 */
2706 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002707do_argfile(exarg_T *eap, int argn)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002708{
2709 int other;
2710 char_u *p;
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002711 int old_arg_idx = curwin->w_arg_idx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002712
2713 if (argn < 0 || argn >= ARGCOUNT)
2714 {
2715 if (ARGCOUNT <= 1)
2716 EMSG(_("E163: There is only one file to edit"));
2717 else if (argn < 0)
2718 EMSG(_("E164: Cannot go before first file"));
2719 else
2720 EMSG(_("E165: Cannot go beyond last file"));
2721 }
2722 else
2723 {
2724 setpcmark();
2725#ifdef FEAT_GUI
2726 need_mouse_correct = TRUE;
2727#endif
2728
Bram Moolenaardf1bdc92006-02-23 21:32:16 +00002729 /* split window or create new tab page first */
2730 if (*eap->cmd == 's' || cmdmod.tab != 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002731 {
2732 if (win_split(0, 0) == FAIL)
2733 return;
Bram Moolenaar3368ea22010-09-21 16:56:35 +02002734 RESET_BINDING(curwin);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002735 }
2736 else
Bram Moolenaar071d4272004-06-13 20:20:40 +00002737 {
2738 /*
2739 * if 'hidden' set, only check for changed file when re-editing
2740 * the same buffer
2741 */
2742 other = TRUE;
Bram Moolenaareb44a682017-08-03 22:44:55 +02002743 if (buf_hide(curbuf))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002744 {
2745 p = fix_fname(alist_name(&ARGLIST[argn]));
2746 other = otherfile(p);
2747 vim_free(p);
2748 }
Bram Moolenaareb44a682017-08-03 22:44:55 +02002749 if ((!buf_hide(curbuf) || !other)
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002750 && check_changed(curbuf, CCGD_AW
2751 | (other ? 0 : CCGD_MULTWIN)
2752 | (eap->forceit ? CCGD_FORCEIT : 0)
2753 | CCGD_EXCMD))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002754 return;
2755 }
2756
2757 curwin->w_arg_idx = argn;
Bram Moolenaar4033c552017-09-16 20:54:51 +02002758 if (argn == ARGCOUNT - 1 && curwin->w_alist == &global_alist)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002759 arg_had_last = TRUE;
2760
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002761 /* Edit the file; always use the last known line number.
2762 * When it fails (e.g. Abort for already edited file) restore the
2763 * argument index. */
2764 if (do_ecmd(0, alist_name(&ARGLIST[curwin->w_arg_idx]), NULL,
Bram Moolenaar071d4272004-06-13 20:20:40 +00002765 eap, ECMD_LAST,
Bram Moolenaareb44a682017-08-03 22:44:55 +02002766 (buf_hide(curwin->w_buffer) ? ECMD_HIDE : 0)
Bram Moolenaar701f7af2008-11-15 13:12:07 +00002767 + (eap->forceit ? ECMD_FORCEIT : 0), curwin) == FAIL)
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002768 curwin->w_arg_idx = old_arg_idx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002769 /* like Vi: set the mark where the cursor is in the file. */
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002770 else if (eap->cmdidx != CMD_argdo)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002771 setmark('\'');
2772 }
2773}
2774
2775/*
2776 * ":next", and commands that behave like it.
2777 */
2778 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002779ex_next(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002780{
2781 int i;
2782
2783 /*
2784 * check for changed buffer now, if this fails the argument list is not
2785 * redefined.
2786 */
Bram Moolenaareb44a682017-08-03 22:44:55 +02002787 if ( buf_hide(curbuf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002788 || eap->cmdidx == CMD_snext
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002789 || !check_changed(curbuf, CCGD_AW
2790 | (eap->forceit ? CCGD_FORCEIT : 0)
2791 | CCGD_EXCMD))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002792 {
2793 if (*eap->arg != NUL) /* redefine file list */
2794 {
2795 if (do_arglist(eap->arg, AL_SET, 0) == FAIL)
2796 return;
2797 i = 0;
2798 }
2799 else
2800 i = curwin->w_arg_idx + (int)eap->line2;
2801 do_argfile(eap, i);
2802 }
2803}
2804
Bram Moolenaar0106e3d2016-02-23 18:55:43 +01002805#if defined(FEAT_LISTCMDS) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002806/*
2807 * ":argedit"
2808 */
2809 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002810ex_argedit(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002811{
Bram Moolenaar90305c62017-07-16 15:31:17 +02002812 int i = eap->addr_count ? (int)eap->line2 : curwin->w_arg_idx + 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002813
Bram Moolenaar90305c62017-07-16 15:31:17 +02002814 if (do_arglist(eap->arg, AL_ADD, i) == FAIL)
2815 return;
2816#ifdef FEAT_TITLE
2817 maketitle();
2818#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002819
Bram Moolenaar90305c62017-07-16 15:31:17 +02002820 if (curwin->w_arg_idx == 0 && (curbuf->b_ml.ml_flags & ML_EMPTY)
2821 && curbuf->b_ffname == NULL)
2822 i = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002823 /* Edit the argument. */
Bram Moolenaar90305c62017-07-16 15:31:17 +02002824 if (i < ARGCOUNT)
2825 do_argfile(eap, i);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002826}
2827
2828/*
2829 * ":argadd"
2830 */
2831 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002832ex_argadd(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002833{
2834 do_arglist(eap->arg, AL_ADD,
2835 eap->addr_count > 0 ? (int)eap->line2 : curwin->w_arg_idx + 1);
2836#ifdef FEAT_TITLE
2837 maketitle();
2838#endif
2839}
2840
2841/*
2842 * ":argdelete"
2843 */
2844 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002845ex_argdelete(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002846{
2847 int i;
2848 int n;
2849
2850 if (eap->addr_count > 0)
2851 {
2852 /* ":1,4argdel": Delete all arguments in the range. */
2853 if (eap->line2 > ARGCOUNT)
2854 eap->line2 = ARGCOUNT;
2855 n = eap->line2 - eap->line1 + 1;
Bram Moolenaar69a92fb2017-03-09 15:58:30 +01002856 if (*eap->arg != NUL)
2857 /* Can't have both a range and an argument. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002858 EMSG(_(e_invarg));
Bram Moolenaar69a92fb2017-03-09 15:58:30 +01002859 else if (n <= 0)
2860 {
2861 /* Don't give an error for ":%argdel" if the list is empty. */
2862 if (eap->line1 != 1 || eap->line2 != 0)
2863 EMSG(_(e_invrange));
2864 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002865 else
2866 {
2867 for (i = eap->line1; i <= eap->line2; ++i)
2868 vim_free(ARGLIST[i - 1].ae_fname);
2869 mch_memmove(ARGLIST + eap->line1 - 1, ARGLIST + eap->line2,
2870 (size_t)((ARGCOUNT - eap->line2) * sizeof(aentry_T)));
2871 ALIST(curwin)->al_ga.ga_len -= n;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002872 if (curwin->w_arg_idx >= eap->line2)
2873 curwin->w_arg_idx -= n;
2874 else if (curwin->w_arg_idx > eap->line1)
2875 curwin->w_arg_idx = eap->line1;
Bram Moolenaar72defda2016-01-17 18:04:33 +01002876 if (ARGCOUNT == 0)
2877 curwin->w_arg_idx = 0;
2878 else if (curwin->w_arg_idx >= ARGCOUNT)
2879 curwin->w_arg_idx = ARGCOUNT - 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002880 }
2881 }
2882 else if (*eap->arg == NUL)
2883 EMSG(_(e_argreq));
2884 else
2885 do_arglist(eap->arg, AL_DEL, 0);
2886#ifdef FEAT_TITLE
2887 maketitle();
2888#endif
2889}
2890
2891/*
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002892 * ":argdo", ":windo", ":bufdo", ":tabdo", ":cdo", ":ldo", ":cfdo" and ":lfdo"
Bram Moolenaar071d4272004-06-13 20:20:40 +00002893 */
2894 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002895ex_listdo(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002896{
2897 int i;
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002898 win_T *wp;
2899 tabpage_T *tp;
Bram Moolenaare25bb902015-02-27 20:33:37 +01002900 buf_T *buf = curbuf;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002901 int next_fnum = 0;
2902#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
2903 char_u *save_ei = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002904#endif
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002905 char_u *p_shm_save;
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002906#ifdef FEAT_QUICKFIX
Bram Moolenaared84b762015-09-09 22:35:29 +02002907 int qf_size = 0;
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002908 int qf_idx;
2909#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002910
Bram Moolenaar0106e3d2016-02-23 18:55:43 +01002911#ifndef FEAT_QUICKFIX
2912 if (eap->cmdidx == CMD_cdo || eap->cmdidx == CMD_ldo ||
2913 eap->cmdidx == CMD_cfdo || eap->cmdidx == CMD_lfdo)
2914 {
2915 ex_ni(eap);
2916 return;
2917 }
2918#endif
2919
Bram Moolenaar071d4272004-06-13 20:20:40 +00002920#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002921 if (eap->cmdidx != CMD_windo && eap->cmdidx != CMD_tabdo)
Bram Moolenaardcaf10e2005-01-21 11:55:25 +00002922 /* Don't do syntax HL autocommands. Skipping the syntax file is a
2923 * great speed improvement. */
2924 save_ei = au_event_disable(",Syntax");
Bram Moolenaar071d4272004-06-13 20:20:40 +00002925#endif
Bram Moolenaar6b1ee342014-08-06 18:17:11 +02002926#ifdef FEAT_CLIPBOARD
2927 start_global_changes();
2928#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002929
2930 if (eap->cmdidx == CMD_windo
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002931 || eap->cmdidx == CMD_tabdo
Bram Moolenaareb44a682017-08-03 22:44:55 +02002932 || buf_hide(curbuf)
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002933 || !check_changed(curbuf, CCGD_AW
2934 | (eap->forceit ? CCGD_FORCEIT : 0)
2935 | CCGD_EXCMD))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002936 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00002937 i = 0;
Bram Moolenaara162bc52015-01-07 16:54:21 +01002938 /* start at the eap->line1 argument/window/buffer */
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002939 wp = firstwin;
2940 tp = first_tabpage;
Bram Moolenaara162bc52015-01-07 16:54:21 +01002941 switch (eap->cmdidx)
2942 {
Bram Moolenaara162bc52015-01-07 16:54:21 +01002943 case CMD_windo:
2944 for ( ; wp != NULL && i + 1 < eap->line1; wp = wp->w_next)
2945 i++;
2946 break;
2947 case CMD_tabdo:
2948 for( ; tp != NULL && i + 1 < eap->line1; tp = tp->tp_next)
2949 i++;
2950 break;
Bram Moolenaara162bc52015-01-07 16:54:21 +01002951 case CMD_argdo:
2952 i = eap->line1 - 1;
2953 break;
Bram Moolenaara162bc52015-01-07 16:54:21 +01002954 default:
2955 break;
2956 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002957 /* set pcmark now */
2958 if (eap->cmdidx == CMD_bufdo)
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002959 {
Bram Moolenaare25bb902015-02-27 20:33:37 +01002960 /* Advance to the first listed buffer after "eap->line1". */
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002961 for (buf = firstbuf; buf != NULL && (buf->b_fnum < eap->line1
Bram Moolenaare25bb902015-02-27 20:33:37 +01002962 || !buf->b_p_bl); buf = buf->b_next)
2963 if (buf->b_fnum > eap->line2)
2964 {
2965 buf = NULL;
2966 break;
2967 }
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002968 if (buf != NULL)
Bram Moolenaare25bb902015-02-27 20:33:37 +01002969 goto_buffer(eap, DOBUF_FIRST, FORWARD, buf->b_fnum);
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002970 }
2971#ifdef FEAT_QUICKFIX
2972 else if (eap->cmdidx == CMD_cdo || eap->cmdidx == CMD_ldo
2973 || eap->cmdidx == CMD_cfdo || eap->cmdidx == CMD_lfdo)
2974 {
2975 qf_size = qf_get_size(eap);
2976 if (qf_size <= 0 || eap->line1 > qf_size)
2977 buf = NULL;
2978 else
2979 {
2980 ex_cc(eap);
2981
2982 buf = curbuf;
2983 i = eap->line1 - 1;
2984 if (eap->addr_count <= 0)
2985 /* default is all the quickfix/location list entries */
2986 eap->line2 = qf_size;
2987 }
2988 }
2989#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002990 else
2991 setpcmark();
2992 listcmd_busy = TRUE; /* avoids setting pcmark below */
2993
Bram Moolenaare25bb902015-02-27 20:33:37 +01002994 while (!got_int && buf != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002995 {
2996 if (eap->cmdidx == CMD_argdo)
2997 {
2998 /* go to argument "i" */
2999 if (i == ARGCOUNT)
3000 break;
3001 /* Don't call do_argfile() when already there, it will try
3002 * reloading the file. */
Bram Moolenaard4755bb2004-09-02 19:12:26 +00003003 if (curwin->w_arg_idx != i || !editing_arg_idx(curwin))
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003004 {
3005 /* Clear 'shm' to avoid that the file message overwrites
3006 * any output from the command. */
3007 p_shm_save = vim_strsave(p_shm);
3008 set_option_value((char_u *)"shm", 0L, (char_u *)"", 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003009 do_argfile(eap, i);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003010 set_option_value((char_u *)"shm", 0L, p_shm_save, 0);
3011 vim_free(p_shm_save);
3012 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003013 if (curwin->w_arg_idx != i)
3014 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003015 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003016 else if (eap->cmdidx == CMD_windo)
3017 {
Bram Moolenaar32466aa2006-02-24 23:53:04 +00003018 /* go to window "wp" */
3019 if (!win_valid(wp))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003020 break;
Bram Moolenaar32466aa2006-02-24 23:53:04 +00003021 win_goto(wp);
Bram Moolenaar41423242007-05-03 20:11:13 +00003022 if (curwin != wp)
3023 break; /* something must be wrong */
Bram Moolenaar32466aa2006-02-24 23:53:04 +00003024 wp = curwin->w_next;
3025 }
3026 else if (eap->cmdidx == CMD_tabdo)
3027 {
3028 /* go to window "tp" */
3029 if (!valid_tabpage(tp))
3030 break;
Bram Moolenaar49e649f2013-05-06 04:50:35 +02003031 goto_tabpage_tp(tp, TRUE, TRUE);
Bram Moolenaar32466aa2006-02-24 23:53:04 +00003032 tp = tp->tp_next;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003033 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003034 else if (eap->cmdidx == CMD_bufdo)
3035 {
3036 /* Remember the number of the next listed buffer, in case
3037 * ":bwipe" is used or autocommands do something strange. */
3038 next_fnum = -1;
3039 for (buf = curbuf->b_next; buf != NULL; buf = buf->b_next)
3040 if (buf->b_p_bl)
3041 {
3042 next_fnum = buf->b_fnum;
3043 break;
3044 }
3045 }
3046
Bram Moolenaara162bc52015-01-07 16:54:21 +01003047 ++i;
3048
Bram Moolenaar071d4272004-06-13 20:20:40 +00003049 /* execute the command */
3050 do_cmdline(eap->arg, eap->getline, eap->cookie,
3051 DOCMD_VERBOSE + DOCMD_NOWAIT);
3052
3053 if (eap->cmdidx == CMD_bufdo)
3054 {
3055 /* Done? */
Bram Moolenaara162bc52015-01-07 16:54:21 +01003056 if (next_fnum < 0 || next_fnum > eap->line2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003057 break;
3058 /* Check if the buffer still exists. */
Bram Moolenaar29323592016-07-24 22:04:11 +02003059 FOR_ALL_BUFFERS(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003060 if (buf->b_fnum == next_fnum)
3061 break;
3062 if (buf == NULL)
3063 break;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003064
3065 /* Go to the next buffer. Clear 'shm' to avoid that the file
3066 * message overwrites any output from the command. */
3067 p_shm_save = vim_strsave(p_shm);
3068 set_option_value((char_u *)"shm", 0L, (char_u *)"", 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003069 goto_buffer(eap, DOBUF_FIRST, FORWARD, next_fnum);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003070 set_option_value((char_u *)"shm", 0L, p_shm_save, 0);
3071 vim_free(p_shm_save);
3072
Bram Moolenaaraa23b372015-09-08 18:46:31 +02003073 /* If autocommands took us elsewhere, quit here. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003074 if (curbuf->b_fnum != next_fnum)
3075 break;
3076 }
3077
Bram Moolenaaraa23b372015-09-08 18:46:31 +02003078#ifdef FEAT_QUICKFIX
3079 if (eap->cmdidx == CMD_cdo || eap->cmdidx == CMD_ldo
3080 || eap->cmdidx == CMD_cfdo || eap->cmdidx == CMD_lfdo)
3081 {
3082 if (i >= qf_size || i >= eap->line2)
3083 break;
3084
3085 qf_idx = qf_get_cur_idx(eap);
3086
3087 ex_cnext(eap);
3088
3089 /* If jumping to the next quickfix entry fails, quit here */
3090 if (qf_get_cur_idx(eap) == qf_idx)
3091 break;
3092 }
3093#endif
3094
Bram Moolenaar071d4272004-06-13 20:20:40 +00003095 if (eap->cmdidx == CMD_windo)
3096 {
3097 validate_cursor(); /* cursor may have moved */
3098#ifdef FEAT_SCROLLBIND
3099 /* required when 'scrollbind' has been set */
3100 if (curwin->w_p_scb)
3101 do_check_scrollbind(TRUE);
3102#endif
3103 }
Bram Moolenaara162bc52015-01-07 16:54:21 +01003104
Bram Moolenaara162bc52015-01-07 16:54:21 +01003105 if (eap->cmdidx == CMD_windo || eap->cmdidx == CMD_tabdo)
3106 if (i+1 > eap->line2)
3107 break;
Bram Moolenaara162bc52015-01-07 16:54:21 +01003108 if (eap->cmdidx == CMD_argdo && i >= eap->line2)
3109 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003110 }
3111 listcmd_busy = FALSE;
3112 }
3113
3114#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
Bram Moolenaar6ac54292005-02-02 23:07:25 +00003115 if (save_ei != NULL)
3116 {
3117 au_event_restore(save_ei);
3118 apply_autocmds(EVENT_SYNTAX, curbuf->b_p_syn,
3119 curbuf->b_fname, TRUE, curbuf);
3120 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003121#endif
Bram Moolenaar6b1ee342014-08-06 18:17:11 +02003122#ifdef FEAT_CLIPBOARD
3123 end_global_changes();
3124#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003125}
3126
3127/*
3128 * Add files[count] to the arglist of the current window after arg "after".
3129 * The file names in files[count] must have been allocated and are taken over.
3130 * Files[] itself is not taken over.
3131 * Returns index of first added argument. Returns -1 when failed (out of mem).
3132 */
3133 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003134alist_add_list(
3135 int count,
3136 char_u **files,
3137 int after) /* where to add: 0 = before first one */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003138{
3139 int i;
Bram Moolenaara24f0a52016-01-17 19:39:00 +01003140 int old_argcount = ARGCOUNT;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003141
3142 if (ga_grow(&ALIST(curwin)->al_ga, count) == OK)
3143 {
3144 if (after < 0)
3145 after = 0;
3146 if (after > ARGCOUNT)
3147 after = ARGCOUNT;
3148 if (after < ARGCOUNT)
3149 mch_memmove(&(ARGLIST[after + count]), &(ARGLIST[after]),
3150 (ARGCOUNT - after) * sizeof(aentry_T));
3151 for (i = 0; i < count; ++i)
3152 {
3153 ARGLIST[after + i].ae_fname = files[i];
3154 ARGLIST[after + i].ae_fnum = buflist_add(files[i], BLN_LISTED);
3155 }
3156 ALIST(curwin)->al_ga.ga_len += count;
Bram Moolenaara24f0a52016-01-17 19:39:00 +01003157 if (old_argcount > 0 && curwin->w_arg_idx >= after)
3158 curwin->w_arg_idx += count;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003159 return after;
3160 }
3161
3162 for (i = 0; i < count; ++i)
3163 vim_free(files[i]);
3164 return -1;
3165}
3166
3167#endif /* FEAT_LISTCMDS */
3168
3169#ifdef FEAT_EVAL
3170/*
3171 * ":compiler[!] {name}"
3172 */
3173 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003174ex_compiler(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003175{
3176 char_u *buf;
3177 char_u *old_cur_comp = NULL;
3178 char_u *p;
3179
3180 if (*eap->arg == NUL)
3181 {
3182 /* List all compiler scripts. */
3183 do_cmdline_cmd((char_u *)"echo globpath(&rtp, 'compiler/*.vim')");
3184 /* ) keep the indenter happy... */
3185 }
3186 else
3187 {
3188 buf = alloc((unsigned)(STRLEN(eap->arg) + 14));
3189 if (buf != NULL)
3190 {
3191 if (eap->forceit)
3192 {
3193 /* ":compiler! {name}" sets global options */
3194 do_cmdline_cmd((char_u *)
3195 "command -nargs=* CompilerSet set <args>");
3196 }
3197 else
3198 {
3199 /* ":compiler! {name}" sets local options.
3200 * To remain backwards compatible "current_compiler" is always
3201 * used. A user's compiler plugin may set it, the distributed
3202 * plugin will then skip the settings. Afterwards set
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003203 * "b:current_compiler" and restore "current_compiler".
3204 * Explicitly prepend "g:" to make it work in a function. */
3205 old_cur_comp = get_var_value((char_u *)"g:current_compiler");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003206 if (old_cur_comp != NULL)
3207 old_cur_comp = vim_strsave(old_cur_comp);
3208 do_cmdline_cmd((char_u *)
3209 "command -nargs=* CompilerSet setlocal <args>");
3210 }
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003211 do_unlet((char_u *)"g:current_compiler", TRUE);
Bram Moolenaar2ce06f62005-01-31 19:19:04 +00003212 do_unlet((char_u *)"b:current_compiler", TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003213
3214 sprintf((char *)buf, "compiler/%s.vim", eap->arg);
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003215 if (source_runtime(buf, DIP_ALL) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003216 EMSG2(_("E666: compiler not supported: %s"), eap->arg);
3217 vim_free(buf);
3218
3219 do_cmdline_cmd((char_u *)":delcommand CompilerSet");
3220
3221 /* Set "b:current_compiler" from "current_compiler". */
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003222 p = get_var_value((char_u *)"g:current_compiler");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003223 if (p != NULL)
3224 set_internal_string_var((char_u *)"b:current_compiler", p);
3225
3226 /* Restore "current_compiler" for ":compiler {name}". */
3227 if (!eap->forceit)
3228 {
3229 if (old_cur_comp != NULL)
3230 {
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003231 set_internal_string_var((char_u *)"g:current_compiler",
Bram Moolenaar071d4272004-06-13 20:20:40 +00003232 old_cur_comp);
3233 vim_free(old_cur_comp);
3234 }
3235 else
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003236 do_unlet((char_u *)"g:current_compiler", TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003237 }
3238 }
3239 }
3240}
3241#endif
3242
3243/*
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003244 * ":runtime [what] {name}"
Bram Moolenaar071d4272004-06-13 20:20:40 +00003245 */
3246 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003247ex_runtime(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003248{
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003249 char_u *arg = eap->arg;
3250 char_u *p = skiptowhite(arg);
3251 int len = (int)(p - arg);
3252 int flags = eap->forceit ? DIP_ALL : 0;
3253
3254 if (STRNCMP(arg, "START", len) == 0)
3255 {
3256 flags += DIP_START + DIP_NORTP;
3257 arg = skipwhite(arg + len);
3258 }
3259 else if (STRNCMP(arg, "OPT", len) == 0)
3260 {
3261 flags += DIP_OPT + DIP_NORTP;
3262 arg = skipwhite(arg + len);
3263 }
3264 else if (STRNCMP(arg, "PACK", len) == 0)
3265 {
3266 flags += DIP_START + DIP_OPT + DIP_NORTP;
3267 arg = skipwhite(arg + len);
3268 }
3269 else if (STRNCMP(arg, "ALL", len) == 0)
3270 {
3271 flags += DIP_START + DIP_OPT;
3272 arg = skipwhite(arg + len);
3273 }
3274
3275 source_runtime(arg, flags);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003276}
3277
Bram Moolenaar071d4272004-06-13 20:20:40 +00003278 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003279source_callback(char_u *fname, void *cookie UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003280{
Bram Moolenaar910f66f2006-04-05 20:41:53 +00003281 (void)do_source(fname, FALSE, DOSO_NONE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003282}
3283
3284/*
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003285 * Find the file "name" in all directories in "path" and invoke
3286 * "callback(fname, cookie)".
3287 * "name" can contain wildcards.
3288 * When "flags" has DIP_ALL: source all files, otherwise only the first one.
3289 * When "flags" has DIP_DIR: find directories instead of files.
3290 * When "flags" has DIP_ERR: give an error message if there is no match.
3291 *
3292 * return FAIL when no file could be sourced, OK otherwise.
3293 */
Bram Moolenaar6bef5302016-03-12 21:28:26 +01003294 int
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003295do_in_path(
3296 char_u *path,
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003297 char_u *name,
Bram Moolenaar91715872016-03-03 17:13:03 +01003298 int flags,
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003299 void (*callback)(char_u *fname, void *ck),
3300 void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003301{
3302 char_u *rtp;
3303 char_u *np;
3304 char_u *buf;
3305 char_u *rtp_copy;
3306 char_u *tail;
3307 int num_files;
3308 char_u **files;
3309 int i;
3310 int did_one = FALSE;
3311#ifdef AMIGA
3312 struct Process *proc = (struct Process *)FindTask(0L);
3313 APTR save_winptr = proc->pr_WindowPtr;
3314
3315 /* Avoid a requester here for a volume that doesn't exist. */
3316 proc->pr_WindowPtr = (APTR)-1L;
3317#endif
3318
3319 /* Make a copy of 'runtimepath'. Invoking the callback may change the
3320 * value. */
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003321 rtp_copy = vim_strsave(path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003322 buf = alloc(MAXPATHL);
3323 if (buf != NULL && rtp_copy != NULL)
3324 {
Bram Moolenaarc09a6d62013-06-10 21:27:29 +02003325 if (p_verbose > 1 && name != NULL)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003326 {
3327 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00003328 smsg((char_u *)_("Searching for \"%s\" in \"%s\""),
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003329 (char *)name, (char *)path);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003330 verbose_leave();
3331 }
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00003332
Bram Moolenaar071d4272004-06-13 20:20:40 +00003333 /* Loop over all entries in 'runtimepath'. */
3334 rtp = rtp_copy;
Bram Moolenaar91715872016-03-03 17:13:03 +01003335 while (*rtp != NUL && ((flags & DIP_ALL) || !did_one))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003336 {
Bram Moolenaar66459b72016-08-06 19:01:55 +02003337 size_t buflen;
3338
Bram Moolenaar071d4272004-06-13 20:20:40 +00003339 /* Copy the path from 'runtimepath' to buf[]. */
3340 copy_option_part(&rtp, buf, MAXPATHL, ",");
Bram Moolenaar66459b72016-08-06 19:01:55 +02003341 buflen = STRLEN(buf);
3342
3343 /* Skip after or non-after directories. */
3344 if (flags & (DIP_NOAFTER | DIP_AFTER))
3345 {
3346 int is_after = buflen >= 5
3347 && STRCMP(buf + buflen - 5, "after") == 0;
3348
3349 if ((is_after && (flags & DIP_NOAFTER))
3350 || (!is_after && (flags & DIP_AFTER)))
3351 continue;
3352 }
3353
Bram Moolenaarc09a6d62013-06-10 21:27:29 +02003354 if (name == NULL)
3355 {
3356 (*callback)(buf, (void *) &cookie);
3357 if (!did_one)
3358 did_one = (cookie == NULL);
3359 }
Bram Moolenaar66459b72016-08-06 19:01:55 +02003360 else if (buflen + STRLEN(name) + 2 < MAXPATHL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003361 {
3362 add_pathsep(buf);
3363 tail = buf + STRLEN(buf);
3364
3365 /* Loop over all patterns in "name" */
3366 np = name;
Bram Moolenaar91715872016-03-03 17:13:03 +01003367 while (*np != NUL && ((flags & DIP_ALL) || !did_one))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003368 {
3369 /* Append the pattern from "name" to buf[]. */
3370 copy_option_part(&np, tail, (int)(MAXPATHL - (tail - buf)),
3371 "\t ");
3372
3373 if (p_verbose > 2)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003374 {
3375 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00003376 smsg((char_u *)_("Searching for \"%s\""), buf);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003377 verbose_leave();
3378 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003379
3380 /* Expand wildcards, invoke the callback for each match. */
3381 if (gen_expand_wildcards(1, &buf, &num_files, &files,
Bram Moolenaar91715872016-03-03 17:13:03 +01003382 (flags & DIP_DIR) ? EW_DIR : EW_FILE) == OK)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003383 {
3384 for (i = 0; i < num_files; ++i)
3385 {
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00003386 (*callback)(files[i], cookie);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003387 did_one = TRUE;
Bram Moolenaar91715872016-03-03 17:13:03 +01003388 if (!(flags & DIP_ALL))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003389 break;
3390 }
3391 FreeWild(num_files, files);
3392 }
3393 }
3394 }
3395 }
3396 }
3397 vim_free(buf);
3398 vim_free(rtp_copy);
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003399 if (!did_one && name != NULL)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003400 {
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003401 char *basepath = path == p_rtp ? "runtimepath" : "packpath";
3402
3403 if (flags & DIP_ERR)
3404 EMSG3(_(e_dirnotf), basepath, name);
3405 else if (p_verbose > 0)
3406 {
3407 verbose_enter();
3408 smsg((char_u *)_("not found in '%s': \"%s\""), basepath, name);
3409 verbose_leave();
3410 }
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003411 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003412
3413#ifdef AMIGA
3414 proc->pr_WindowPtr = save_winptr;
3415#endif
3416
3417 return did_one ? OK : FAIL;
3418}
3419
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003420/*
Bram Moolenaar07ecfa62017-06-27 14:43:55 +02003421 * Find "name" in "path". When found, invoke the callback function for
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003422 * it: callback(fname, "cookie")
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003423 * When "flags" has DIP_ALL repeat for all matches, otherwise only the first
3424 * one is used.
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003425 * Returns OK when at least one match found, FAIL otherwise.
3426 *
Bram Moolenaar07ecfa62017-06-27 14:43:55 +02003427 * If "name" is NULL calls callback for each entry in "path". Cookie is
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003428 * passed by reference in this case, setting it to NULL indicates that callback
3429 * has done its job.
3430 */
Bram Moolenaar07ecfa62017-06-27 14:43:55 +02003431 static int
3432do_in_path_and_pp(
3433 char_u *path,
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003434 char_u *name,
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003435 int flags,
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003436 void (*callback)(char_u *fname, void *ck),
3437 void *cookie)
3438{
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003439 int done = FAIL;
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003440 char_u *s;
3441 int len;
3442 char *start_dir = "pack/*/start/*/%s";
3443 char *opt_dir = "pack/*/opt/*/%s";
3444
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003445 if ((flags & DIP_NORTP) == 0)
Bram Moolenaar07ecfa62017-06-27 14:43:55 +02003446 done = do_in_path(path, name, flags, callback, cookie);
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003447
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003448 if ((done == FAIL || (flags & DIP_ALL)) && (flags & DIP_START))
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003449 {
Bram Moolenaar1c8b4ed2016-03-17 21:51:03 +01003450 len = (int)(STRLEN(start_dir) + STRLEN(name));
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003451 s = alloc(len);
3452 if (s == NULL)
3453 return FAIL;
3454 vim_snprintf((char *)s, len, start_dir, name);
3455 done = do_in_path(p_pp, s, flags, callback, cookie);
3456 vim_free(s);
3457 }
3458
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003459 if ((done == FAIL || (flags & DIP_ALL)) && (flags & DIP_OPT))
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003460 {
Bram Moolenaar1c8b4ed2016-03-17 21:51:03 +01003461 len = (int)(STRLEN(opt_dir) + STRLEN(name));
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003462 s = alloc(len);
3463 if (s == NULL)
3464 return FAIL;
3465 vim_snprintf((char *)s, len, opt_dir, name);
3466 done = do_in_path(p_pp, s, flags, callback, cookie);
3467 vim_free(s);
3468 }
3469
3470 return done;
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003471}
3472
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003473/*
Bram Moolenaar07ecfa62017-06-27 14:43:55 +02003474 * Just like do_in_path_and_pp(), using 'runtimepath' for "path".
3475 */
3476 int
3477do_in_runtimepath(
3478 char_u *name,
3479 int flags,
3480 void (*callback)(char_u *fname, void *ck),
3481 void *cookie)
3482{
3483 return do_in_path_and_pp(p_rtp, name, flags, callback, cookie);
3484}
3485
3486/*
3487 * Source the file "name" from all directories in 'runtimepath'.
3488 * "name" can contain wildcards.
3489 * When "flags" has DIP_ALL: source all files, otherwise only the first one.
3490 *
3491 * return FAIL when no file could be sourced, OK otherwise.
3492 */
3493 int
3494source_runtime(char_u *name, int flags)
3495{
3496 return source_in_path(p_rtp, name, flags);
3497}
3498
3499/*
3500 * Just like source_runtime(), but use "path" instead of 'runtimepath'.
3501 */
3502 int
3503source_in_path(char_u *path, char_u *name, int flags)
3504{
3505 return do_in_path_and_pp(path, name, flags, source_callback, NULL);
3506}
3507
3508
3509/*
Bram Moolenaarf3654822016-03-04 22:12:23 +01003510 * Expand wildcards in "pat" and invoke do_source() for each match.
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003511 */
3512 static void
Bram Moolenaarf3654822016-03-04 22:12:23 +01003513source_all_matches(char_u *pat)
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003514{
Bram Moolenaarf3654822016-03-04 22:12:23 +01003515 int num_files;
3516 char_u **files;
3517 int i;
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003518
Bram Moolenaarf3654822016-03-04 22:12:23 +01003519 if (gen_expand_wildcards(1, &pat, &num_files, &files, EW_FILE) == OK)
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003520 {
Bram Moolenaarf3654822016-03-04 22:12:23 +01003521 for (i = 0; i < num_files; ++i)
3522 (void)do_source(files[i], FALSE, DOSO_NONE);
3523 FreeWild(num_files, files);
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003524 }
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003525}
3526
Bram Moolenaar49b27322016-04-05 21:13:00 +02003527/* used for "cookie" of add_pack_plugin() */
3528static int APP_ADD_DIR;
3529static int APP_LOAD;
3530static int APP_BOTH;
3531
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003532 static void
Bram Moolenaar91715872016-03-03 17:13:03 +01003533add_pack_plugin(char_u *fname, void *cookie)
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003534{
Bram Moolenaarf3654822016-03-04 22:12:23 +01003535 char_u *p4, *p3, *p2, *p1, *p;
3536 char_u *insp;
Bram Moolenaar91715872016-03-03 17:13:03 +01003537 int c;
3538 char_u *new_rtp;
3539 int keep;
Bram Moolenaarb0550662016-05-31 21:37:36 +02003540 size_t oldlen;
3541 size_t addlen;
Bram Moolenaara5702442016-05-24 19:37:29 +02003542 char_u *afterdir;
Bram Moolenaarb0550662016-05-31 21:37:36 +02003543 size_t afterlen = 0;
Bram Moolenaar91715872016-03-03 17:13:03 +01003544 char_u *ffname = fix_fname(fname);
Bram Moolenaarfef524b2016-07-02 22:07:22 +02003545 size_t fname_len;
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003546 char_u *buf = NULL;
3547 char_u *rtp_ffname;
3548 int match;
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003549
Bram Moolenaar91715872016-03-03 17:13:03 +01003550 if (ffname == NULL)
3551 return;
Bram Moolenaar49b27322016-04-05 21:13:00 +02003552 if (cookie != &APP_LOAD && strstr((char *)p_rtp, (char *)ffname) == NULL)
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003553 {
Bram Moolenaara5702442016-05-24 19:37:29 +02003554 /* directory is not yet in 'runtimepath', add it */
Bram Moolenaarf3654822016-03-04 22:12:23 +01003555 p4 = p3 = p2 = p1 = get_past_head(ffname);
Bram Moolenaar91acfff2017-03-12 19:22:36 +01003556 for (p = p1; *p; MB_PTR_ADV(p))
Bram Moolenaarf3654822016-03-04 22:12:23 +01003557 if (vim_ispathsep_nocolon(*p))
3558 {
3559 p4 = p3; p3 = p2; p2 = p1; p1 = p;
3560 }
3561
3562 /* now we have:
Bram Moolenaaraf1a0e32016-03-09 22:19:26 +01003563 * rtp/pack/name/start/name
3564 * p4 p3 p2 p1
Bram Moolenaarf3654822016-03-04 22:12:23 +01003565 *
3566 * find the part up to "pack" in 'runtimepath' */
3567 c = *p4;
3568 *p4 = NUL;
Bram Moolenaar4c5717e2016-07-01 15:39:40 +02003569
3570 /* Find "ffname" in "p_rtp", ignoring '/' vs '\' differences. */
3571 fname_len = STRLEN(ffname);
3572 insp = p_rtp;
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003573 buf = alloc(MAXPATHL);
3574 if (buf == NULL)
3575 goto theend;
3576 while (*insp != NUL)
Bram Moolenaar4c5717e2016-07-01 15:39:40 +02003577 {
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003578 copy_option_part(&insp, buf, MAXPATHL, ",");
3579 add_pathsep(buf);
3580 rtp_ffname = fix_fname(buf);
3581 if (rtp_ffname == NULL)
3582 goto theend;
3583 match = vim_fnamencmp(rtp_ffname, ffname, fname_len) == 0;
3584 vim_free(rtp_ffname);
3585 if (match)
Bram Moolenaar4c5717e2016-07-01 15:39:40 +02003586 break;
Bram Moolenaar4c5717e2016-07-01 15:39:40 +02003587 }
3588
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003589 if (*insp == NUL)
Bram Moolenaarf3654822016-03-04 22:12:23 +01003590 /* not found, append at the end */
3591 insp = p_rtp + STRLEN(p_rtp);
3592 else
Bram Moolenaarf3654822016-03-04 22:12:23 +01003593 /* append after the matching directory. */
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003594 --insp;
Bram Moolenaarf3654822016-03-04 22:12:23 +01003595 *p4 = c;
3596
Bram Moolenaara5702442016-05-24 19:37:29 +02003597 /* check if rtp/pack/name/start/name/after exists */
3598 afterdir = concat_fnames(ffname, (char_u *)"after", TRUE);
3599 if (afterdir != NULL && mch_isdir(afterdir))
3600 afterlen = STRLEN(afterdir) + 1; /* add one for comma */
3601
Bram Moolenaarb0550662016-05-31 21:37:36 +02003602 oldlen = STRLEN(p_rtp);
3603 addlen = STRLEN(ffname) + 1; /* add one for comma */
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003604 new_rtp = alloc((int)(oldlen + addlen + afterlen + 1));
3605 /* add one for NUL */
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003606 if (new_rtp == NULL)
Bram Moolenaarf3654822016-03-04 22:12:23 +01003607 goto theend;
3608 keep = (int)(insp - p_rtp);
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003609 mch_memmove(new_rtp, p_rtp, keep);
3610 new_rtp[keep] = ',';
Bram Moolenaara5702442016-05-24 19:37:29 +02003611 mch_memmove(new_rtp + keep + 1, ffname, addlen);
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003612 if (p_rtp[keep] != NUL)
Bram Moolenaara5702442016-05-24 19:37:29 +02003613 mch_memmove(new_rtp + keep + addlen, p_rtp + keep,
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003614 oldlen - keep + 1);
Bram Moolenaara5702442016-05-24 19:37:29 +02003615 if (afterlen > 0)
3616 {
3617 STRCAT(new_rtp, ",");
3618 STRCAT(new_rtp, afterdir);
3619 }
Bram Moolenaar863c1a92016-03-03 15:47:06 +01003620 set_option_value((char_u *)"rtp", 0L, new_rtp, 0);
3621 vim_free(new_rtp);
Bram Moolenaara5702442016-05-24 19:37:29 +02003622 vim_free(afterdir);
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003623 }
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003624
Bram Moolenaar49b27322016-04-05 21:13:00 +02003625 if (cookie != &APP_ADD_DIR)
Bram Moolenaarf3654822016-03-04 22:12:23 +01003626 {
Bram Moolenaar71fb0c12016-04-02 22:44:16 +02003627 static char *plugpat = "%s/plugin/**/*.vim";
Bram Moolenaarf3654822016-03-04 22:12:23 +01003628 static char *ftpat = "%s/ftdetect/*.vim";
3629 int len;
3630 char_u *pat;
3631
3632 len = (int)STRLEN(ffname) + (int)STRLEN(ftpat);
3633 pat = alloc(len);
3634 if (pat == NULL)
3635 goto theend;
3636 vim_snprintf((char *)pat, len, plugpat, ffname);
3637 source_all_matches(pat);
3638
3639#ifdef FEAT_AUTOCMD
3640 {
3641 char_u *cmd = vim_strsave((char_u *)"g:did_load_filetypes");
3642
3643 /* If runtime/filetype.vim wasn't loaded yet, the scripts will be
3644 * found when it loads. */
3645 if (cmd != NULL && eval_to_number(cmd) > 0)
3646 {
3647 do_cmdline_cmd((char_u *)"augroup filetypedetect");
3648 vim_snprintf((char *)pat, len, ftpat, ffname);
3649 source_all_matches(pat);
3650 do_cmdline_cmd((char_u *)"augroup END");
3651 }
3652 vim_free(cmd);
3653 }
3654#endif
Bram Moolenaarba8cd122016-03-19 14:16:39 +01003655 vim_free(pat);
Bram Moolenaarf3654822016-03-04 22:12:23 +01003656 }
3657
3658theend:
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003659 vim_free(buf);
Bram Moolenaarf3654822016-03-04 22:12:23 +01003660 vim_free(ffname);
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003661}
3662
Bram Moolenaarce876aa2017-06-04 17:47:42 +02003663/*
3664 * Add all packages in the "start" directory to 'runtimepath'.
3665 */
3666 void
3667add_pack_start_dirs(void)
3668{
3669 do_in_path(p_pp, (char_u *)"pack/*/start/*", DIP_ALL + DIP_DIR,
3670 add_pack_plugin, &APP_ADD_DIR);
3671}
3672
3673/*
3674 * Load plugins from all packages in the "start" directory.
3675 */
3676 void
3677load_start_packages(void)
3678{
3679 did_source_packages = TRUE;
3680 do_in_path(p_pp, (char_u *)"pack/*/start/*", DIP_ALL + DIP_DIR,
3681 add_pack_plugin, &APP_LOAD);
3682}
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003683
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003684/*
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003685 * ":packloadall"
Bram Moolenaarf3654822016-03-04 22:12:23 +01003686 * Find plugins in the package directories and source them.
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003687 */
3688 void
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003689ex_packloadall(exarg_T *eap)
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003690{
Bram Moolenaarce876aa2017-06-04 17:47:42 +02003691 if (!did_source_packages || eap->forceit)
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003692 {
Bram Moolenaar49b27322016-04-05 21:13:00 +02003693 /* First do a round to add all directories to 'runtimepath', then load
3694 * the plugins. This allows for plugins to use an autoload directory
3695 * of another plugin. */
Bram Moolenaarce876aa2017-06-04 17:47:42 +02003696 add_pack_start_dirs();
3697 load_start_packages();
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003698 }
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003699}
3700
3701/*
Bram Moolenaarf3654822016-03-04 22:12:23 +01003702 * ":packadd[!] {name}"
Bram Moolenaar91715872016-03-03 17:13:03 +01003703 */
3704 void
3705ex_packadd(exarg_T *eap)
3706{
3707 static char *plugpat = "pack/*/opt/%s";
3708 int len;
3709 char *pat;
3710
3711 len = (int)STRLEN(plugpat) + (int)STRLEN(eap->arg);
3712 pat = (char *)alloc(len);
3713 if (pat == NULL)
3714 return;
3715 vim_snprintf(pat, len, plugpat, eap->arg);
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003716 do_in_path(p_pp, (char_u *)pat, DIP_ALL + DIP_DIR + DIP_ERR,
Bram Moolenaar49b27322016-04-05 21:13:00 +02003717 add_pack_plugin, eap->forceit ? &APP_ADD_DIR : &APP_BOTH);
Bram Moolenaar91715872016-03-03 17:13:03 +01003718 vim_free(pat);
3719}
3720
Bram Moolenaar071d4272004-06-13 20:20:40 +00003721#if defined(FEAT_EVAL) && defined(FEAT_AUTOCMD)
3722/*
3723 * ":options"
3724 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003725 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003726ex_options(
3727 exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003728{
Bram Moolenaarab6c8582017-08-11 17:15:09 +02003729 vim_setenv((char_u *)"OPTWIN_CMD", (char_u *)(cmdmod.tab ? "tab" : ""));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003730 cmd_source((char_u *)SYS_OPTWIN_FILE, NULL);
3731}
3732#endif
3733
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01003734#if defined(FEAT_PYTHON3) || defined(FEAT_PYTHON) || defined(PROTO)
3735
3736# if (defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)) || defined(PROTO)
3737/*
3738 * Detect Python 3 or 2, and initialize 'pyxversion'.
3739 */
3740 void
3741init_pyxversion(void)
3742{
3743 if (p_pyx == 0)
3744 {
3745 if (python3_enabled(FALSE))
3746 p_pyx = 3;
3747 else if (python_enabled(FALSE))
3748 p_pyx = 2;
3749 }
3750}
3751# endif
3752
3753/*
3754 * Does a file contain one of the following strings at the beginning of any
3755 * line?
3756 * "#!(any string)python2" => returns 2
3757 * "#!(any string)python3" => returns 3
3758 * "# requires python 2.x" => returns 2
3759 * "# requires python 3.x" => returns 3
3760 * otherwise return 0.
3761 */
3762 static int
3763requires_py_version(char_u *filename)
3764{
3765 FILE *file;
3766 int requires_py_version = 0;
3767 int i, lines;
3768
3769 lines = (int)p_mls;
3770 if (lines < 0)
3771 lines = 5;
3772
3773 file = mch_fopen((char *)filename, "r");
3774 if (file != NULL)
3775 {
3776 for (i = 0; i < lines; i++)
3777 {
3778 if (vim_fgets(IObuff, IOSIZE, file))
3779 break;
3780 if (i == 0 && IObuff[0] == '#' && IObuff[1] == '!')
3781 {
3782 /* Check shebang. */
3783 if (strstr((char *)IObuff + 2, "python2") != NULL)
3784 {
3785 requires_py_version = 2;
3786 break;
3787 }
3788 if (strstr((char *)IObuff + 2, "python3") != NULL)
3789 {
3790 requires_py_version = 3;
3791 break;
3792 }
3793 }
3794 IObuff[21] = '\0';
3795 if (STRCMP("# requires python 2.x", IObuff) == 0)
3796 {
3797 requires_py_version = 2;
3798 break;
3799 }
3800 if (STRCMP("# requires python 3.x", IObuff) == 0)
3801 {
3802 requires_py_version = 3;
3803 break;
3804 }
3805 }
3806 fclose(file);
3807 }
3808 return requires_py_version;
3809}
3810
3811
3812/*
3813 * Source a python file using the requested python version.
3814 */
3815 static void
3816source_pyx_file(exarg_T *eap, char_u *fname)
3817{
3818 exarg_T ex;
3819 int v = requires_py_version(fname);
3820
3821# if defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)
3822 init_pyxversion();
3823# endif
3824 if (v == 0)
3825 {
3826# if defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)
3827 /* user didn't choose a preference, 'pyx' is used */
3828 v = p_pyx;
3829# elif defined(FEAT_PYTHON)
3830 v = 2;
3831# elif defined(FEAT_PYTHON3)
3832 v = 3;
3833# endif
3834 }
3835
3836 /*
3837 * now source, if required python version is not supported show
3838 * unobtrusive message.
3839 */
3840 if (eap == NULL)
3841 vim_memset(&ex, 0, sizeof(ex));
3842 else
3843 ex = *eap;
3844 ex.arg = fname;
3845 ex.cmd = (char_u *)(v == 2 ? "pyfile" : "pyfile3");
3846
3847 if (v == 2)
3848 {
3849# ifdef FEAT_PYTHON
3850 ex_pyfile(&ex);
3851# else
3852 vim_snprintf((char *)IObuff, IOSIZE,
3853 _("W20: Required python version 2.x not supported, ignoring file: %s"),
3854 fname);
3855 MSG(IObuff);
3856# endif
3857 return;
3858 }
3859 else
3860 {
3861# ifdef FEAT_PYTHON3
3862 ex_py3file(&ex);
3863# else
3864 vim_snprintf((char *)IObuff, IOSIZE,
3865 _("W21: Required python version 3.x not supported, ignoring file: %s"),
3866 fname);
3867 MSG(IObuff);
3868# endif
3869 return;
3870 }
3871}
3872
3873/*
3874 * ":pyxfile {fname}"
3875 */
3876 void
3877ex_pyxfile(exarg_T *eap)
3878{
3879 source_pyx_file(eap, eap->arg);
3880}
3881
3882/*
3883 * ":pyx"
3884 */
3885 void
3886ex_pyx(exarg_T *eap)
3887{
3888# if defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)
3889 init_pyxversion();
3890 if (p_pyx == 2)
3891 ex_python(eap);
3892 else
3893 ex_py3(eap);
3894# elif defined(FEAT_PYTHON)
3895 ex_python(eap);
3896# elif defined(FEAT_PYTHON3)
3897 ex_py3(eap);
3898# endif
3899}
3900
3901/*
3902 * ":pyxdo"
3903 */
3904 void
3905ex_pyxdo(exarg_T *eap)
3906{
3907# if defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)
3908 init_pyxversion();
3909 if (p_pyx == 2)
3910 ex_pydo(eap);
3911 else
3912 ex_py3do(eap);
3913# elif defined(FEAT_PYTHON)
3914 ex_pydo(eap);
3915# elif defined(FEAT_PYTHON3)
3916 ex_py3do(eap);
3917# endif
3918}
3919
3920#endif
3921
Bram Moolenaar071d4272004-06-13 20:20:40 +00003922/*
3923 * ":source {fname}"
3924 */
3925 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003926ex_source(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003927{
3928#ifdef FEAT_BROWSE
3929 if (cmdmod.browse)
3930 {
3931 char_u *fname = NULL;
3932
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00003933 fname = do_browse(0, (char_u *)_("Source Vim script"), eap->arg,
Bram Moolenaar071d4272004-06-13 20:20:40 +00003934 NULL, NULL, BROWSE_FILTER_MACROS, NULL);
3935 if (fname != NULL)
3936 {
3937 cmd_source(fname, eap);
3938 vim_free(fname);
3939 }
3940 }
3941 else
3942#endif
3943 cmd_source(eap->arg, eap);
3944}
3945
3946 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003947cmd_source(char_u *fname, exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003948{
3949 if (*fname == NUL)
3950 EMSG(_(e_argreq));
3951
Bram Moolenaar071d4272004-06-13 20:20:40 +00003952 else if (eap != NULL && eap->forceit)
Bram Moolenaar84a05ac2013-05-06 04:24:17 +02003953 /* ":source!": read Normal mode commands
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00003954 * Need to execute the commands directly. This is required at least
3955 * for:
Bram Moolenaar071d4272004-06-13 20:20:40 +00003956 * - ":g" command busy
3957 * - after ":argdo", ":windo" or ":bufdo"
3958 * - another command follows
3959 * - inside a loop
3960 */
3961 openscript(fname, global_busy || listcmd_busy || eap->nextcmd != NULL
3962#ifdef FEAT_EVAL
3963 || eap->cstack->cs_idx >= 0
3964#endif
3965 );
3966
3967 /* ":source" read ex commands */
Bram Moolenaar910f66f2006-04-05 20:41:53 +00003968 else if (do_source(fname, FALSE, DOSO_NONE) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003969 EMSG2(_(e_notopen), fname);
3970}
3971
3972/*
3973 * ":source" and associated commands.
3974 */
3975/*
3976 * Structure used to store info for each sourced file.
3977 * It is shared between do_source() and getsourceline().
3978 * This is required, because it needs to be handed to do_cmdline() and
3979 * sourcing can be done recursively.
3980 */
3981struct source_cookie
3982{
3983 FILE *fp; /* opened file for sourcing */
3984 char_u *nextline; /* if not NULL: line that was read ahead */
3985 int finished; /* ":finish" used */
Bram Moolenaar860cae12010-06-05 23:22:07 +02003986#if defined(USE_CRNL) || defined(USE_CR)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003987 int fileformat; /* EOL_UNKNOWN, EOL_UNIX or EOL_DOS */
3988 int error; /* TRUE if LF found after CR-LF */
3989#endif
3990#ifdef FEAT_EVAL
3991 linenr_T breakpoint; /* next line with breakpoint or zero */
3992 char_u *fname; /* name of sourced file */
3993 int dbg_tick; /* debug_tick when breakpoint was set */
3994 int level; /* top nesting level of sourced file */
3995#endif
3996#ifdef FEAT_MBYTE
3997 vimconv_T conv; /* type of conversion */
3998#endif
3999};
4000
4001#ifdef FEAT_EVAL
4002/*
4003 * Return the address holding the next breakpoint line for a source cookie.
4004 */
4005 linenr_T *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004006source_breakpoint(void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004007{
4008 return &((struct source_cookie *)cookie)->breakpoint;
4009}
4010
4011/*
4012 * Return the address holding the debug tick for a source cookie.
4013 */
4014 int *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004015source_dbg_tick(void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004016{
4017 return &((struct source_cookie *)cookie)->dbg_tick;
4018}
4019
4020/*
4021 * Return the nesting level for a source cookie.
4022 */
4023 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004024source_level(void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004025{
4026 return ((struct source_cookie *)cookie)->level;
4027}
4028#endif
4029
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01004030static char_u *get_one_sourceline(struct source_cookie *sp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004031
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004032#if (defined(WIN32) && defined(FEAT_CSCOPE)) || defined(HAVE_FD_CLOEXEC)
4033# define USE_FOPEN_NOINH
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01004034static FILE *fopen_noinh_readbin(char *filename);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004035
4036/*
4037 * Special function to open a file without handle inheritance.
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004038 * When possible the handle is closed on exec().
Bram Moolenaar071d4272004-06-13 20:20:40 +00004039 */
4040 static FILE *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004041fopen_noinh_readbin(char *filename)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004042{
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004043# ifdef WIN32
Bram Moolenaar8d8ef0b2010-01-20 21:41:47 +01004044 int fd_tmp = mch_open(filename, O_RDONLY | O_BINARY | O_NOINHERIT, 0);
4045# else
4046 int fd_tmp = mch_open(filename, O_RDONLY, 0);
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004047# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004048
4049 if (fd_tmp == -1)
4050 return NULL;
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004051
4052# ifdef HAVE_FD_CLOEXEC
4053 {
4054 int fdflags = fcntl(fd_tmp, F_GETFD);
4055 if (fdflags >= 0 && (fdflags & FD_CLOEXEC) == 0)
Bram Moolenaarcde88542015-08-11 19:14:00 +02004056 (void)fcntl(fd_tmp, F_SETFD, fdflags | FD_CLOEXEC);
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004057 }
4058# endif
4059
Bram Moolenaar071d4272004-06-13 20:20:40 +00004060 return fdopen(fd_tmp, READBIN);
4061}
4062#endif
4063
4064
4065/*
4066 * do_source: Read the file "fname" and execute its lines as EX commands.
4067 *
4068 * This function may be called recursively!
4069 *
4070 * return FAIL if file could not be opened, OK otherwise
4071 */
4072 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004073do_source(
4074 char_u *fname,
4075 int check_other, /* check for .vimrc and _vimrc */
4076 int is_vimrc) /* DOSO_ value */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004077{
4078 struct source_cookie cookie;
4079 char_u *save_sourcing_name;
4080 linenr_T save_sourcing_lnum;
4081 char_u *p;
4082 char_u *fname_exp;
Bram Moolenaar73881402009-02-04 16:50:47 +00004083 char_u *firstline = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004084 int retval = FAIL;
4085#ifdef FEAT_EVAL
4086 scid_T save_current_SID;
4087 static scid_T last_current_SID = 0;
4088 void *save_funccalp;
4089 int save_debug_break_level = debug_break_level;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004090 scriptitem_T *si = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004091# ifdef UNIX
Bram Moolenaar8767f522016-07-01 17:17:39 +02004092 stat_T st;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004093 int stat_ok;
4094# endif
4095#endif
4096#ifdef STARTUPTIME
4097 struct timeval tv_rel;
4098 struct timeval tv_start;
4099#endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00004100#ifdef FEAT_PROFILE
4101 proftime_T wait_start;
4102#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004103
Bram Moolenaar071d4272004-06-13 20:20:40 +00004104 p = expand_env_save(fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004105 if (p == NULL)
4106 return retval;
4107 fname_exp = fix_fname(p);
4108 vim_free(p);
4109 if (fname_exp == NULL)
4110 return retval;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004111 if (mch_isdir(fname_exp))
4112 {
Bram Moolenaar555b2802005-05-19 21:08:39 +00004113 smsg((char_u *)_("Cannot source a directory: \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004114 goto theend;
4115 }
4116
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00004117#ifdef FEAT_AUTOCMD
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00004118 /* Apply SourceCmd autocommands, they should get the file and source it. */
4119 if (has_autocmd(EVENT_SOURCECMD, fname_exp, NULL)
4120 && apply_autocmds(EVENT_SOURCECMD, fname_exp, fname_exp,
4121 FALSE, curbuf))
Bram Moolenaarf33943e2008-01-15 21:17:30 +00004122 {
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00004123# ifdef FEAT_EVAL
Bram Moolenaarf33943e2008-01-15 21:17:30 +00004124 retval = aborting() ? FAIL : OK;
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00004125# else
Bram Moolenaarf33943e2008-01-15 21:17:30 +00004126 retval = OK;
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00004127# endif
Bram Moolenaarf33943e2008-01-15 21:17:30 +00004128 goto theend;
4129 }
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00004130
4131 /* Apply SourcePre autocommands, they may get the file. */
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00004132 apply_autocmds(EVENT_SOURCEPRE, fname_exp, fname_exp, FALSE, curbuf);
4133#endif
4134
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004135#ifdef USE_FOPEN_NOINH
Bram Moolenaar071d4272004-06-13 20:20:40 +00004136 cookie.fp = fopen_noinh_readbin((char *)fname_exp);
4137#else
4138 cookie.fp = mch_fopen((char *)fname_exp, READBIN);
4139#endif
4140 if (cookie.fp == NULL && check_other)
4141 {
4142 /*
4143 * Try again, replacing file name ".vimrc" by "_vimrc" or vice versa,
4144 * and ".exrc" by "_exrc" or vice versa.
4145 */
4146 p = gettail(fname_exp);
4147 if ((*p == '.' || *p == '_')
4148 && (STRICMP(p + 1, "vimrc") == 0
4149 || STRICMP(p + 1, "gvimrc") == 0
4150 || STRICMP(p + 1, "exrc") == 0))
4151 {
4152 if (*p == '_')
4153 *p = '.';
4154 else
4155 *p = '_';
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004156#ifdef USE_FOPEN_NOINH
Bram Moolenaar071d4272004-06-13 20:20:40 +00004157 cookie.fp = fopen_noinh_readbin((char *)fname_exp);
4158#else
4159 cookie.fp = mch_fopen((char *)fname_exp, READBIN);
4160#endif
4161 }
4162 }
4163
4164 if (cookie.fp == NULL)
4165 {
4166 if (p_verbose > 0)
4167 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004168 verbose_enter();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004169 if (sourcing_name == NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00004170 smsg((char_u *)_("could not source \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004171 else
4172 smsg((char_u *)_("line %ld: could not source \"%s\""),
Bram Moolenaar555b2802005-05-19 21:08:39 +00004173 sourcing_lnum, fname);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004174 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004175 }
4176 goto theend;
4177 }
4178
4179 /*
4180 * The file exists.
4181 * - In verbose mode, give a message.
4182 * - For a vimrc file, may want to set 'compatible', call vimrc_found().
4183 */
4184 if (p_verbose > 1)
4185 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004186 verbose_enter();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004187 if (sourcing_name == NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00004188 smsg((char_u *)_("sourcing \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004189 else
4190 smsg((char_u *)_("line %ld: sourcing \"%s\""),
Bram Moolenaar555b2802005-05-19 21:08:39 +00004191 sourcing_lnum, fname);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004192 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004193 }
Bram Moolenaar910f66f2006-04-05 20:41:53 +00004194 if (is_vimrc == DOSO_VIMRC)
4195 vimrc_found(fname_exp, (char_u *)"MYVIMRC");
4196 else if (is_vimrc == DOSO_GVIMRC)
4197 vimrc_found(fname_exp, (char_u *)"MYGVIMRC");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004198
4199#ifdef USE_CRNL
4200 /* If no automatic file format: Set default to CR-NL. */
4201 if (*p_ffs == NUL)
4202 cookie.fileformat = EOL_DOS;
4203 else
4204 cookie.fileformat = EOL_UNKNOWN;
4205 cookie.error = FALSE;
4206#endif
4207
4208#ifdef USE_CR
4209 /* If no automatic file format: Set default to CR. */
4210 if (*p_ffs == NUL)
4211 cookie.fileformat = EOL_MAC;
4212 else
4213 cookie.fileformat = EOL_UNKNOWN;
4214 cookie.error = FALSE;
4215#endif
4216
4217 cookie.nextline = NULL;
4218 cookie.finished = FALSE;
4219
4220#ifdef FEAT_EVAL
4221 /*
4222 * Check if this script has a breakpoint.
4223 */
4224 cookie.breakpoint = dbg_find_breakpoint(TRUE, fname_exp, (linenr_T)0);
4225 cookie.fname = fname_exp;
4226 cookie.dbg_tick = debug_tick;
4227
4228 cookie.level = ex_nesting_level;
4229#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004230
4231 /*
4232 * Keep the sourcing name/lnum, for recursive calls.
4233 */
4234 save_sourcing_name = sourcing_name;
4235 sourcing_name = fname_exp;
4236 save_sourcing_lnum = sourcing_lnum;
4237 sourcing_lnum = 0;
4238
4239#ifdef STARTUPTIME
Bram Moolenaarc4e41982010-01-19 16:31:47 +01004240 if (time_fd != NULL)
4241 time_push(&tv_rel, &tv_start);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004242#endif
4243
4244#ifdef FEAT_EVAL
Bram Moolenaar05159a02005-02-26 23:04:13 +00004245# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004246 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004247 prof_child_enter(&wait_start); /* entering a child now */
4248# endif
4249
4250 /* Don't use local function variables, if called from a function.
4251 * Also starts profiling timer for nested script. */
4252 save_funccalp = save_funccal();
4253
Bram Moolenaar071d4272004-06-13 20:20:40 +00004254 /*
4255 * Check if this script was sourced before to finds its SID.
4256 * If it's new, generate a new SID.
4257 */
4258 save_current_SID = current_SID;
4259# ifdef UNIX
4260 stat_ok = (mch_stat((char *)fname_exp, &st) >= 0);
4261# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00004262 for (current_SID = script_items.ga_len; current_SID > 0; --current_SID)
4263 {
4264 si = &SCRIPT_ITEM(current_SID);
4265 if (si->sn_name != NULL
Bram Moolenaar071d4272004-06-13 20:20:40 +00004266 && (
4267# ifdef UNIX
Bram Moolenaar7c626922005-02-07 22:01:03 +00004268 /* Compare dev/ino when possible, it catches symbolic
4269 * links. Also compare file names, the inode may change
4270 * when the file was edited. */
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00004271 ((stat_ok && si->sn_dev_valid)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004272 && (si->sn_dev == st.st_dev
4273 && si->sn_ino == st.st_ino)) ||
Bram Moolenaar071d4272004-06-13 20:20:40 +00004274# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00004275 fnamecmp(si->sn_name, fname_exp) == 0))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004276 break;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004277 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004278 if (current_SID == 0)
4279 {
4280 current_SID = ++last_current_SID;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004281 if (ga_grow(&script_items, (int)(current_SID - script_items.ga_len))
4282 == FAIL)
4283 goto almosttheend;
4284 while (script_items.ga_len < current_SID)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004285 {
Bram Moolenaar05159a02005-02-26 23:04:13 +00004286 ++script_items.ga_len;
4287 SCRIPT_ITEM(script_items.ga_len).sn_name = NULL;
4288# ifdef FEAT_PROFILE
4289 SCRIPT_ITEM(script_items.ga_len).sn_prof_on = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004290# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004291 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00004292 si = &SCRIPT_ITEM(current_SID);
4293 si->sn_name = fname_exp;
4294 fname_exp = NULL;
4295# ifdef UNIX
4296 if (stat_ok)
4297 {
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00004298 si->sn_dev_valid = TRUE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004299 si->sn_dev = st.st_dev;
4300 si->sn_ino = st.st_ino;
4301 }
4302 else
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00004303 si->sn_dev_valid = FALSE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004304# endif
4305
Bram Moolenaar071d4272004-06-13 20:20:40 +00004306 /* Allocate the local script variables to use for this script. */
4307 new_script_vars(current_SID);
4308 }
4309
Bram Moolenaar05159a02005-02-26 23:04:13 +00004310# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004311 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004312 {
4313 int forceit;
4314
4315 /* Check if we do profiling for this script. */
4316 if (!si->sn_prof_on && has_profiling(TRUE, si->sn_name, &forceit))
4317 {
4318 script_do_profile(si);
4319 si->sn_pr_force = forceit;
4320 }
4321 if (si->sn_prof_on)
4322 {
4323 ++si->sn_pr_count;
4324 profile_start(&si->sn_pr_start);
4325 profile_zero(&si->sn_pr_children);
4326 }
4327 }
4328# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004329#endif
4330
Bram Moolenaar67435d92017-10-19 21:04:37 +02004331#ifdef FEAT_MBYTE
4332 cookie.conv.vc_type = CONV_NONE; /* no conversion */
4333
4334 /* Read the first line so we can check for a UTF-8 BOM. */
4335 firstline = getsourceline(0, (void *)&cookie, 0);
4336 if (firstline != NULL && STRLEN(firstline) >= 3 && firstline[0] == 0xef
4337 && firstline[1] == 0xbb && firstline[2] == 0xbf)
4338 {
4339 /* Found BOM; setup conversion, skip over BOM and recode the line. */
4340 convert_setup(&cookie.conv, (char_u *)"utf-8", p_enc);
4341 p = string_convert(&cookie.conv, firstline + 3, NULL);
4342 if (p == NULL)
4343 p = vim_strsave(firstline + 3);
4344 if (p != NULL)
4345 {
4346 vim_free(firstline);
4347 firstline = p;
4348 }
4349 }
4350#endif
4351
Bram Moolenaar071d4272004-06-13 20:20:40 +00004352 /*
4353 * Call do_cmdline, which will call getsourceline() to get the lines.
4354 */
Bram Moolenaar73881402009-02-04 16:50:47 +00004355 do_cmdline(firstline, getsourceline, (void *)&cookie,
Bram Moolenaar071d4272004-06-13 20:20:40 +00004356 DOCMD_VERBOSE|DOCMD_NOWAIT|DOCMD_REPEAT);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004357 retval = OK;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004358
4359#ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004360 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004361 {
4362 /* Get "si" again, "script_items" may have been reallocated. */
4363 si = &SCRIPT_ITEM(current_SID);
4364 if (si->sn_prof_on)
4365 {
4366 profile_end(&si->sn_pr_start);
4367 profile_sub_wait(&wait_start, &si->sn_pr_start);
4368 profile_add(&si->sn_pr_total, &si->sn_pr_start);
Bram Moolenaar1056d982006-03-09 22:37:52 +00004369 profile_self(&si->sn_pr_self, &si->sn_pr_start,
4370 &si->sn_pr_children);
Bram Moolenaar05159a02005-02-26 23:04:13 +00004371 }
4372 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004373#endif
4374
4375 if (got_int)
4376 EMSG(_(e_interr));
4377 sourcing_name = save_sourcing_name;
4378 sourcing_lnum = save_sourcing_lnum;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004379 if (p_verbose > 1)
4380 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004381 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00004382 smsg((char_u *)_("finished sourcing %s"), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004383 if (sourcing_name != NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00004384 smsg((char_u *)_("continuing in %s"), sourcing_name);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004385 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004386 }
4387#ifdef STARTUPTIME
Bram Moolenaarc4e41982010-01-19 16:31:47 +01004388 if (time_fd != NULL)
4389 {
4390 vim_snprintf((char *)IObuff, IOSIZE, "sourcing %s", fname);
4391 time_msg((char *)IObuff, &tv_start);
4392 time_pop(&tv_rel);
4393 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004394#endif
4395
4396#ifdef FEAT_EVAL
4397 /*
4398 * After a "finish" in debug mode, need to break at first command of next
4399 * sourced file.
4400 */
4401 if (save_debug_break_level > ex_nesting_level
4402 && debug_break_level == ex_nesting_level)
4403 ++debug_break_level;
4404#endif
4405
Bram Moolenaar05159a02005-02-26 23:04:13 +00004406#ifdef FEAT_EVAL
4407almosttheend:
4408 current_SID = save_current_SID;
4409 restore_funccal(save_funccalp);
4410# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004411 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004412 prof_child_exit(&wait_start); /* leaving a child now */
4413# endif
4414#endif
4415 fclose(cookie.fp);
4416 vim_free(cookie.nextline);
Bram Moolenaar73881402009-02-04 16:50:47 +00004417 vim_free(firstline);
Bram Moolenaar05159a02005-02-26 23:04:13 +00004418#ifdef FEAT_MBYTE
4419 convert_setup(&cookie.conv, NULL, NULL);
4420#endif
4421
Bram Moolenaar071d4272004-06-13 20:20:40 +00004422theend:
4423 vim_free(fname_exp);
4424 return retval;
4425}
4426
4427#if defined(FEAT_EVAL) || defined(PROTO)
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00004428
Bram Moolenaar071d4272004-06-13 20:20:40 +00004429/*
4430 * ":scriptnames"
4431 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004432 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004433ex_scriptnames(exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004434{
4435 int i;
4436
Bram Moolenaar05159a02005-02-26 23:04:13 +00004437 for (i = 1; i <= script_items.ga_len && !got_int; ++i)
4438 if (SCRIPT_ITEM(i).sn_name != NULL)
Bram Moolenaard58ea072011-06-26 04:25:30 +02004439 {
4440 home_replace(NULL, SCRIPT_ITEM(i).sn_name,
4441 NameBuff, MAXPATHL, TRUE);
4442 smsg((char_u *)"%3d: %s", i, NameBuff);
Bram Moolenaar970a1b82012-03-23 18:39:18 +01004443 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004444}
4445
4446# if defined(BACKSLASH_IN_FILENAME) || defined(PROTO)
4447/*
4448 * Fix slashes in the list of script names for 'shellslash'.
4449 */
4450 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004451scriptnames_slash_adjust(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004452{
4453 int i;
4454
Bram Moolenaar05159a02005-02-26 23:04:13 +00004455 for (i = 1; i <= script_items.ga_len; ++i)
4456 if (SCRIPT_ITEM(i).sn_name != NULL)
4457 slash_adjust(SCRIPT_ITEM(i).sn_name);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004458}
4459# endif
4460
4461/*
4462 * Get a pointer to a script name. Used for ":verbose set".
4463 */
4464 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004465get_scriptname(scid_T id)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004466{
4467 if (id == SID_MODELINE)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00004468 return (char_u *)_("modeline");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004469 if (id == SID_CMDARG)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00004470 return (char_u *)_("--cmd argument");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004471 if (id == SID_CARG)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00004472 return (char_u *)_("-c argument");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004473 if (id == SID_ENV)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00004474 return (char_u *)_("environment variable");
4475 if (id == SID_ERROR)
4476 return (char_u *)_("error handler");
Bram Moolenaar05159a02005-02-26 23:04:13 +00004477 return SCRIPT_ITEM(id).sn_name;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004478}
Bram Moolenaar05159a02005-02-26 23:04:13 +00004479
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00004480# if defined(EXITFREE) || defined(PROTO)
4481 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004482free_scriptnames(void)
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00004483{
4484 int i;
4485
4486 for (i = script_items.ga_len; i > 0; --i)
4487 vim_free(SCRIPT_ITEM(i).sn_name);
4488 ga_clear(&script_items);
4489}
4490# endif
4491
Bram Moolenaar071d4272004-06-13 20:20:40 +00004492#endif
4493
4494#if defined(USE_CR) || defined(PROTO)
4495
4496# if defined(__MSL__) && (__MSL__ >= 22)
4497/*
4498 * Newer version of the Metrowerks library handle DOS and UNIX files
4499 * without help.
4500 * Test with earlier versions, MSL 2.2 is the library supplied with
4501 * Codewarrior Pro 2.
4502 */
4503 char *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004504fgets_cr(char *s, int n, FILE *stream)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004505{
4506 return fgets(s, n, stream);
4507}
4508# else
4509/*
4510 * Version of fgets() which also works for lines ending in a <CR> only
4511 * (Macintosh format).
4512 * For older versions of the Metrowerks library.
4513 * At least CodeWarrior 9 needed this code.
4514 */
4515 char *
Bram Moolenaard14e00e2016-01-31 17:30:51 +01004516fgets_cr(char *s, int n, FILE *stream)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004517{
4518 int c = 0;
4519 int char_read = 0;
4520
4521 while (!feof(stream) && c != '\r' && c != '\n' && char_read < n - 1)
4522 {
4523 c = fgetc(stream);
4524 s[char_read++] = c;
4525 /* If the file is in DOS format, we need to skip a NL after a CR. I
4526 * thought it was the other way around, but this appears to work... */
4527 if (c == '\n')
4528 {
4529 c = fgetc(stream);
4530 if (c != '\r')
4531 ungetc(c, stream);
4532 }
4533 }
4534
4535 s[char_read] = 0;
4536 if (char_read == 0)
4537 return NULL;
4538
4539 if (feof(stream) && char_read == 1)
4540 return NULL;
4541
4542 return s;
4543}
4544# endif
4545#endif
4546
4547/*
4548 * Get one full line from a sourced file.
4549 * Called by do_cmdline() when it's called from do_source().
4550 *
4551 * Return a pointer to the line in allocated memory.
4552 * Return NULL for end-of-file or some error.
4553 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004554 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004555getsourceline(int c UNUSED, void *cookie, int indent UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004556{
4557 struct source_cookie *sp = (struct source_cookie *)cookie;
4558 char_u *line;
Bram Moolenaarc047b9a2012-02-11 20:40:55 +01004559 char_u *p;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004560
4561#ifdef FEAT_EVAL
4562 /* If breakpoints have been added/deleted need to check for it. */
4563 if (sp->dbg_tick < debug_tick)
4564 {
4565 sp->breakpoint = dbg_find_breakpoint(TRUE, sp->fname, sourcing_lnum);
4566 sp->dbg_tick = debug_tick;
4567 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00004568# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004569 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004570 script_line_end();
4571# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004572#endif
4573 /*
4574 * Get current line. If there is a read-ahead line, use it, otherwise get
4575 * one now.
4576 */
4577 if (sp->finished)
4578 line = NULL;
4579 else if (sp->nextline == NULL)
4580 line = get_one_sourceline(sp);
4581 else
4582 {
4583 line = sp->nextline;
4584 sp->nextline = NULL;
4585 ++sourcing_lnum;
4586 }
Bram Moolenaare2cc9702005-03-15 22:43:58 +00004587#ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004588 if (line != NULL && do_profiling == PROF_YES)
Bram Moolenaare2cc9702005-03-15 22:43:58 +00004589 script_line_start();
4590#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004591
4592 /* Only concatenate lines starting with a \ when 'cpoptions' doesn't
4593 * contain the 'C' flag. */
4594 if (line != NULL && (vim_strchr(p_cpo, CPO_CONCAT) == NULL))
4595 {
4596 /* compensate for the one line read-ahead */
4597 --sourcing_lnum;
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004598
4599 /* Get the next line and concatenate it when it starts with a
4600 * backslash. We always need to read the next line, keep it in
4601 * sp->nextline. */
4602 sp->nextline = get_one_sourceline(sp);
4603 if (sp->nextline != NULL && *(p = skipwhite(sp->nextline)) == '\\')
Bram Moolenaar071d4272004-06-13 20:20:40 +00004604 {
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004605 garray_T ga;
4606
Bram Moolenaarb549a732012-02-22 18:29:33 +01004607 ga_init2(&ga, (int)sizeof(char_u), 400);
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004608 ga_concat(&ga, line);
4609 ga_concat(&ga, p + 1);
4610 for (;;)
4611 {
4612 vim_free(sp->nextline);
4613 sp->nextline = get_one_sourceline(sp);
4614 if (sp->nextline == NULL)
4615 break;
4616 p = skipwhite(sp->nextline);
4617 if (*p != '\\')
4618 break;
Bram Moolenaarb549a732012-02-22 18:29:33 +01004619 /* Adjust the growsize to the current length to speed up
4620 * concatenating many lines. */
4621 if (ga.ga_len > 400)
4622 {
4623 if (ga.ga_len > 8000)
4624 ga.ga_growsize = 8000;
4625 else
4626 ga.ga_growsize = ga.ga_len;
4627 }
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004628 ga_concat(&ga, p + 1);
4629 }
4630 ga_append(&ga, NUL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004631 vim_free(line);
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004632 line = ga.ga_data;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004633 }
4634 }
4635
4636#ifdef FEAT_MBYTE
4637 if (line != NULL && sp->conv.vc_type != CONV_NONE)
4638 {
Bram Moolenaarc047b9a2012-02-11 20:40:55 +01004639 char_u *s;
4640
Bram Moolenaar071d4272004-06-13 20:20:40 +00004641 /* Convert the encoding of the script line. */
4642 s = string_convert(&sp->conv, line, NULL);
4643 if (s != NULL)
4644 {
4645 vim_free(line);
4646 line = s;
4647 }
4648 }
4649#endif
4650
4651#ifdef FEAT_EVAL
4652 /* Did we encounter a breakpoint? */
4653 if (sp->breakpoint != 0 && sp->breakpoint <= sourcing_lnum)
4654 {
4655 dbg_breakpoint(sp->fname, sourcing_lnum);
4656 /* Find next breakpoint. */
4657 sp->breakpoint = dbg_find_breakpoint(TRUE, sp->fname, sourcing_lnum);
4658 sp->dbg_tick = debug_tick;
4659 }
4660#endif
4661
4662 return line;
4663}
4664
4665 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004666get_one_sourceline(struct source_cookie *sp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004667{
4668 garray_T ga;
4669 int len;
4670 int c;
4671 char_u *buf;
4672#ifdef USE_CRNL
4673 int has_cr; /* CR-LF found */
4674#endif
4675#ifdef USE_CR
4676 char_u *scan;
4677#endif
4678 int have_read = FALSE;
4679
4680 /* use a growarray to store the sourced line */
Bram Moolenaar6ac54292005-02-02 23:07:25 +00004681 ga_init2(&ga, 1, 250);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004682
4683 /*
4684 * Loop until there is a finished line (or end-of-file).
4685 */
4686 sourcing_lnum++;
4687 for (;;)
4688 {
Bram Moolenaar6ac54292005-02-02 23:07:25 +00004689 /* make room to read at least 120 (more) characters */
4690 if (ga_grow(&ga, 120) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004691 break;
4692 buf = (char_u *)ga.ga_data;
4693
4694#ifdef USE_CR
4695 if (sp->fileformat == EOL_MAC)
4696 {
Bram Moolenaar86b68352004-12-27 21:59:20 +00004697 if (fgets_cr((char *)buf + ga.ga_len, ga.ga_maxlen - ga.ga_len,
4698 sp->fp) == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004699 break;
4700 }
4701 else
4702#endif
Bram Moolenaar86b68352004-12-27 21:59:20 +00004703 if (fgets((char *)buf + ga.ga_len, ga.ga_maxlen - ga.ga_len,
4704 sp->fp) == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004705 break;
Bram Moolenaar6ac54292005-02-02 23:07:25 +00004706 len = ga.ga_len + (int)STRLEN(buf + ga.ga_len);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004707#ifdef USE_CRNL
4708 /* Ignore a trailing CTRL-Z, when in Dos mode. Only recognize the
4709 * CTRL-Z by its own, or after a NL. */
4710 if ( (len == 1 || (len >= 2 && buf[len - 2] == '\n'))
4711 && sp->fileformat == EOL_DOS
4712 && buf[len - 1] == Ctrl_Z)
4713 {
4714 buf[len - 1] = NUL;
4715 break;
4716 }
4717#endif
4718
4719#ifdef USE_CR
4720 /* If the read doesn't stop on a new line, and there's
4721 * some CR then we assume a Mac format */
4722 if (sp->fileformat == EOL_UNKNOWN)
4723 {
4724 if (buf[len - 1] != '\n' && vim_strchr(buf, '\r') != NULL)
4725 sp->fileformat = EOL_MAC;
4726 else
4727 sp->fileformat = EOL_UNIX;
4728 }
4729
4730 if (sp->fileformat == EOL_MAC)
4731 {
4732 scan = vim_strchr(buf, '\r');
4733
4734 if (scan != NULL)
4735 {
4736 *scan = '\n';
4737 if (*(scan + 1) != 0)
4738 {
4739 *(scan + 1) = 0;
4740 fseek(sp->fp, (long)(scan - buf - len + 1), SEEK_CUR);
4741 }
4742 }
4743 len = STRLEN(buf);
4744 }
4745#endif
4746
4747 have_read = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004748 ga.ga_len = len;
4749
4750 /* If the line was longer than the buffer, read more. */
Bram Moolenaar86b68352004-12-27 21:59:20 +00004751 if (ga.ga_maxlen - ga.ga_len == 1 && buf[len - 1] != '\n')
Bram Moolenaar071d4272004-06-13 20:20:40 +00004752 continue;
4753
4754 if (len >= 1 && buf[len - 1] == '\n') /* remove trailing NL */
4755 {
4756#ifdef USE_CRNL
4757 has_cr = (len >= 2 && buf[len - 2] == '\r');
4758 if (sp->fileformat == EOL_UNKNOWN)
4759 {
4760 if (has_cr)
4761 sp->fileformat = EOL_DOS;
4762 else
4763 sp->fileformat = EOL_UNIX;
4764 }
4765
4766 if (sp->fileformat == EOL_DOS)
4767 {
4768 if (has_cr) /* replace trailing CR */
4769 {
4770 buf[len - 2] = '\n';
4771 --len;
4772 --ga.ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004773 }
4774 else /* lines like ":map xx yy^M" will have failed */
4775 {
4776 if (!sp->error)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00004777 {
Bram Moolenaar8820b482017-03-16 17:23:31 +01004778 msg_source(HL_ATTR(HLF_W));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004779 EMSG(_("W15: Warning: Wrong line separator, ^M may be missing"));
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00004780 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004781 sp->error = TRUE;
4782 sp->fileformat = EOL_UNIX;
4783 }
4784 }
4785#endif
4786 /* The '\n' is escaped if there is an odd number of ^V's just
4787 * before it, first set "c" just before the 'V's and then check
4788 * len&c parities (is faster than ((len-c)%2 == 0)) -- Acevedo */
4789 for (c = len - 2; c >= 0 && buf[c] == Ctrl_V; c--)
4790 ;
4791 if ((len & 1) != (c & 1)) /* escaped NL, read more */
4792 {
4793 sourcing_lnum++;
4794 continue;
4795 }
4796
4797 buf[len - 1] = NUL; /* remove the NL */
4798 }
4799
4800 /*
4801 * Check for ^C here now and then, so recursive :so can be broken.
4802 */
4803 line_breakcheck();
4804 break;
4805 }
4806
4807 if (have_read)
4808 return (char_u *)ga.ga_data;
4809
4810 vim_free(ga.ga_data);
4811 return NULL;
4812}
4813
Bram Moolenaar05159a02005-02-26 23:04:13 +00004814#if defined(FEAT_PROFILE) || defined(PROTO)
4815/*
4816 * Called when starting to read a script line.
4817 * "sourcing_lnum" must be correct!
4818 * When skipping lines it may not actually be executed, but we won't find out
4819 * until later and we need to store the time now.
4820 */
4821 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004822script_line_start(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004823{
4824 scriptitem_T *si;
4825 sn_prl_T *pp;
4826
4827 if (current_SID <= 0 || current_SID > script_items.ga_len)
4828 return;
4829 si = &SCRIPT_ITEM(current_SID);
4830 if (si->sn_prof_on && sourcing_lnum >= 1)
4831 {
Bram Moolenaar8c8de832008-06-24 22:58:06 +00004832 /* Grow the array before starting the timer, so that the time spent
Bram Moolenaar05159a02005-02-26 23:04:13 +00004833 * here isn't counted. */
Bram Moolenaar67435d92017-10-19 21:04:37 +02004834 (void)ga_grow(&si->sn_prl_ga,
4835 (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/*
Bram Moolenaar67435d92017-10-19 21:04:37 +02004870 * Called when done with a script line.
Bram Moolenaar05159a02005-02-26 23:04:13 +00004871 */
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