blob: 6caed87bb29a40c3edb189e9139e8ec5056f45fa [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 Moolenaard99388b2017-10-26 14:28:32 +0200134 debug_mode = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000135
136 if (!debug_did_msg)
137 MSG(_("Entering Debug mode. Type \"cont\" to continue."));
138 if (sourcing_name != NULL)
139 msg(sourcing_name);
140 if (sourcing_lnum != 0)
Bram Moolenaar555b2802005-05-19 21:08:39 +0000141 smsg((char_u *)_("line %ld: %s"), (long)sourcing_lnum, cmd);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000142 else
Bram Moolenaar555b2802005-05-19 21:08:39 +0000143 smsg((char_u *)_("cmd: %s"), cmd);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000144
145 /*
146 * Repeat getting a command and executing it.
147 */
148 for (;;)
149 {
150 msg_scroll = TRUE;
151 need_wait_return = FALSE;
Bram Moolenaar85b11762016-02-27 18:13:23 +0100152
Bram Moolenaar071d4272004-06-13 20:20:40 +0000153 /* Save the current typeahead buffer and replace it with an empty one.
154 * This makes sure we get input from the user here and don't interfere
155 * with the commands being executed. Reset "ex_normal_busy" to avoid
156 * the side effects of using ":normal". Save the stuff buffer and make
Bram Moolenaaree3f7a52008-01-01 13:17:56 +0000157 * it empty. Set ignore_script to avoid reading from script input. */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000158 save_ex_normal_busy = ex_normal_busy;
159 ex_normal_busy = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000160 if (!debug_greedy)
Bram Moolenaaree3f7a52008-01-01 13:17:56 +0000161 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000162 save_typeahead(&typeaheadbuf);
Bram Moolenaaree3f7a52008-01-01 13:17:56 +0000163 typeahead_saved = TRUE;
164 save_ignore_script = ignore_script;
165 ignore_script = TRUE;
166 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000167
Bram Moolenaardc303bc2016-05-17 17:45:38 +0200168 vim_free(cmdline);
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +0000169 cmdline = getcmdline_prompt('>', NULL, 0, EXPAND_NOTHING, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000170
Bram Moolenaaree3f7a52008-01-01 13:17:56 +0000171 if (typeahead_saved)
172 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000173 restore_typeahead(&typeaheadbuf);
Bram Moolenaaree3f7a52008-01-01 13:17:56 +0000174 ignore_script = save_ignore_script;
175 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000176 ex_normal_busy = save_ex_normal_busy;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000177
178 cmdline_row = msg_row;
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100179 msg_starthere();
Bram Moolenaar071d4272004-06-13 20:20:40 +0000180 if (cmdline != NULL)
181 {
182 /* If this is a debug command, set "last_cmd".
183 * If not, reset "last_cmd".
184 * For a blank line use previous command. */
185 p = skipwhite(cmdline);
186 if (*p != NUL)
187 {
188 switch (*p)
189 {
190 case 'c': last_cmd = CMD_CONT;
191 tail = "ont";
192 break;
193 case 'n': last_cmd = CMD_NEXT;
194 tail = "ext";
195 break;
196 case 's': last_cmd = CMD_STEP;
197 tail = "tep";
198 break;
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100199 case 'f':
200 last_cmd = 0;
201 if (p[1] == 'r')
202 {
203 last_cmd = CMD_FRAME;
204 tail = "rame";
205 }
206 else
207 {
208 last_cmd = CMD_FINISH;
209 tail = "inish";
210 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000211 break;
212 case 'q': last_cmd = CMD_QUIT;
213 tail = "uit";
214 break;
215 case 'i': last_cmd = CMD_INTERRUPT;
216 tail = "nterrupt";
217 break;
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100218 case 'b': last_cmd = CMD_BACKTRACE;
219 if (p[1] == 't')
220 tail = "t";
221 else
222 tail = "acktrace";
223 break;
224 case 'w': last_cmd = CMD_BACKTRACE;
225 tail = "here";
226 break;
227 case 'u': last_cmd = CMD_UP;
228 tail = "p";
229 break;
230 case 'd': last_cmd = CMD_DOWN;
231 tail = "own";
232 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000233 default: last_cmd = 0;
234 }
235 if (last_cmd != 0)
236 {
237 /* Check that the tail matches. */
238 ++p;
239 while (*p != NUL && *p == *tail)
240 {
241 ++p;
242 ++tail;
243 }
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100244 if (ASCII_ISALPHA(*p) && last_cmd != CMD_FRAME)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000245 last_cmd = 0;
246 }
247 }
248
249 if (last_cmd != 0)
250 {
251 /* Execute debug command: decided where to break next and
252 * return. */
253 switch (last_cmd)
254 {
255 case CMD_CONT:
256 debug_break_level = -1;
257 break;
258 case CMD_NEXT:
259 debug_break_level = ex_nesting_level;
260 break;
261 case CMD_STEP:
262 debug_break_level = 9999;
263 break;
264 case CMD_FINISH:
265 debug_break_level = ex_nesting_level - 1;
266 break;
267 case CMD_QUIT:
268 got_int = TRUE;
269 debug_break_level = -1;
270 break;
271 case CMD_INTERRUPT:
272 got_int = TRUE;
273 debug_break_level = 9999;
274 /* Do not repeat ">interrupt" cmd, continue stepping. */
275 last_cmd = CMD_STEP;
276 break;
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100277 case CMD_BACKTRACE:
278 do_showbacktrace(cmd);
279 continue;
280 case CMD_FRAME:
281 if (*p == NUL)
282 {
283 do_showbacktrace(cmd);
284 }
285 else
286 {
287 p = skipwhite(p);
288 do_setdebugtracelevel(p);
289 }
290 continue;
291 case CMD_UP:
292 debug_backtrace_level++;
293 do_checkbacktracelevel();
294 continue;
295 case CMD_DOWN:
296 debug_backtrace_level--;
297 do_checkbacktracelevel();
298 continue;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000299 }
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100300 /* Going out reset backtrace_level */
301 debug_backtrace_level = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000302 break;
303 }
304
305 /* don't debug this command */
306 n = debug_break_level;
307 debug_break_level = -1;
308 (void)do_cmdline(cmdline, getexline, NULL,
309 DOCMD_VERBOSE|DOCMD_EXCRESET);
310 debug_break_level = n;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000311 }
312 lines_left = Rows - 1;
313 }
314 vim_free(cmdline);
315
316 --RedrawingDisabled;
317 --no_wait_return;
318 redraw_all_later(NOT_VALID);
319 need_wait_return = FALSE;
320 msg_scroll = save_msg_scroll;
321 lines_left = Rows - 1;
322 State = save_State;
Bram Moolenaard99388b2017-10-26 14:28:32 +0200323 debug_mode = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000324 did_emsg = save_did_emsg;
325 cmd_silent = save_cmd_silent;
326 msg_silent = save_msg_silent;
327 emsg_silent = save_emsg_silent;
328 redir_off = save_redir_off;
329
330 /* Only print the message again when typing a command before coming back
331 * here. */
332 debug_did_msg = TRUE;
333}
334
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100335 static int
336get_maxbacktrace_level(void)
337{
338 char *p, *q;
Bram Moolenaardc633cf2016-04-23 14:33:19 +0200339 int maxbacktrace = 0;
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100340
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100341 if (sourcing_name != NULL)
342 {
343 p = (char *)sourcing_name;
344 while ((q = strstr(p, "..")) != NULL)
345 {
346 p = q + 2;
347 maxbacktrace++;
348 }
349 }
350 return maxbacktrace;
351}
352
353 static void
354do_setdebugtracelevel(char_u *arg)
355{
356 int level;
357
358 level = atoi((char *)arg);
359 if (*arg == '+' || level < 0)
360 debug_backtrace_level += level;
361 else
362 debug_backtrace_level = level;
363
364 do_checkbacktracelevel();
365}
366
367 static void
368do_checkbacktracelevel(void)
369{
370 if (debug_backtrace_level < 0)
371 {
372 debug_backtrace_level = 0;
373 MSG(_("frame is zero"));
374 }
375 else
376 {
377 int max = get_maxbacktrace_level();
378
379 if (debug_backtrace_level > max)
380 {
381 debug_backtrace_level = max;
382 smsg((char_u *)_("frame at highest level: %d"), max);
383 }
384 }
385}
386
387 static void
388do_showbacktrace(char_u *cmd)
389{
390 char *cur;
391 char *next;
392 int i = 0;
393 int max = get_maxbacktrace_level();
394
395 if (sourcing_name != NULL)
396 {
397 cur = (char *)sourcing_name;
398 while (!got_int)
399 {
400 next = strstr(cur, "..");
401 if (next != NULL)
402 *next = NUL;
403 if (i == max - debug_backtrace_level)
404 smsg((char_u *)"->%d %s", max - i, cur);
405 else
406 smsg((char_u *)" %d %s", max - i, cur);
407 ++i;
408 if (next == NULL)
409 break;
410 *next = '.';
411 cur = next + 2;
412 }
413 }
414 if (sourcing_lnum != 0)
415 smsg((char_u *)_("line %ld: %s"), (long)sourcing_lnum, cmd);
416 else
417 smsg((char_u *)_("cmd: %s"), cmd);
418}
419
Bram Moolenaar071d4272004-06-13 20:20:40 +0000420/*
421 * ":debug".
422 */
423 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100424ex_debug(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000425{
426 int debug_break_level_save = debug_break_level;
427
428 debug_break_level = 9999;
429 do_cmdline_cmd(eap->arg);
430 debug_break_level = debug_break_level_save;
431}
432
433static char_u *debug_breakpoint_name = NULL;
434static linenr_T debug_breakpoint_lnum;
435
436/*
437 * When debugging or a breakpoint is set on a skipped command, no debug prompt
438 * is shown by do_one_cmd(). This situation is indicated by debug_skipped, and
439 * debug_skipped_name is then set to the source name in the breakpoint case. If
440 * a skipped command decides itself that a debug prompt should be displayed, it
441 * can do so by calling dbg_check_skipped().
442 */
443static int debug_skipped;
444static char_u *debug_skipped_name;
445
446/*
447 * Go to debug mode when a breakpoint was encountered or "ex_nesting_level" is
448 * at or below the break level. But only when the line is actually
449 * executed. Return TRUE and set breakpoint_name for skipped commands that
450 * decide to execute something themselves.
451 * Called from do_one_cmd() before executing a command.
452 */
453 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100454dbg_check_breakpoint(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000455{
456 char_u *p;
457
458 debug_skipped = FALSE;
459 if (debug_breakpoint_name != NULL)
460 {
461 if (!eap->skip)
462 {
463 /* replace K_SNR with "<SNR>" */
464 if (debug_breakpoint_name[0] == K_SPECIAL
465 && debug_breakpoint_name[1] == KS_EXTRA
466 && debug_breakpoint_name[2] == (int)KE_SNR)
467 p = (char_u *)"<SNR>";
468 else
469 p = (char_u *)"";
Bram Moolenaar555b2802005-05-19 21:08:39 +0000470 smsg((char_u *)_("Breakpoint in \"%s%s\" line %ld"),
471 p,
Bram Moolenaar071d4272004-06-13 20:20:40 +0000472 debug_breakpoint_name + (*p == NUL ? 0 : 3),
473 (long)debug_breakpoint_lnum);
474 debug_breakpoint_name = NULL;
475 do_debug(eap->cmd);
476 }
477 else
478 {
479 debug_skipped = TRUE;
480 debug_skipped_name = debug_breakpoint_name;
481 debug_breakpoint_name = NULL;
482 }
483 }
484 else if (ex_nesting_level <= debug_break_level)
485 {
486 if (!eap->skip)
487 do_debug(eap->cmd);
488 else
489 {
490 debug_skipped = TRUE;
491 debug_skipped_name = NULL;
492 }
493 }
494}
495
496/*
497 * Go to debug mode if skipped by dbg_check_breakpoint() because eap->skip was
498 * set. Return TRUE when the debug mode is entered this time.
499 */
500 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100501dbg_check_skipped(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000502{
503 int prev_got_int;
504
505 if (debug_skipped)
506 {
507 /*
508 * Save the value of got_int and reset it. We don't want a previous
509 * interruption cause flushing the input buffer.
510 */
511 prev_got_int = got_int;
512 got_int = FALSE;
513 debug_breakpoint_name = debug_skipped_name;
514 /* eap->skip is TRUE */
515 eap->skip = FALSE;
516 (void)dbg_check_breakpoint(eap);
517 eap->skip = TRUE;
518 got_int |= prev_got_int;
519 return TRUE;
520 }
521 return FALSE;
522}
523
524/*
525 * The list of breakpoints: dbg_breakp.
526 * This is a grow-array of structs.
527 */
528struct debuggy
529{
530 int dbg_nr; /* breakpoint number */
531 int dbg_type; /* DBG_FUNC or DBG_FILE */
532 char_u *dbg_name; /* function or file name */
533 regprog_T *dbg_prog; /* regexp program */
534 linenr_T dbg_lnum; /* line number in function or file */
Bram Moolenaar05159a02005-02-26 23:04:13 +0000535 int dbg_forceit; /* ! used */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000536};
537
538static garray_T dbg_breakp = {0, 0, sizeof(struct debuggy), 4, NULL};
Bram Moolenaar05159a02005-02-26 23:04:13 +0000539#define BREAKP(idx) (((struct debuggy *)dbg_breakp.ga_data)[idx])
540#define DEBUGGY(gap, idx) (((struct debuggy *)gap->ga_data)[idx])
Bram Moolenaar071d4272004-06-13 20:20:40 +0000541static int last_breakp = 0; /* nr of last defined breakpoint */
542
Bram Moolenaar05159a02005-02-26 23:04:13 +0000543#ifdef FEAT_PROFILE
544/* Profiling uses file and func names similar to breakpoints. */
545static garray_T prof_ga = {0, 0, sizeof(struct debuggy), 4, NULL};
546#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000547#define DBG_FUNC 1
548#define DBG_FILE 2
549
Bram Moolenaarf28dbce2016-01-29 22:03:47 +0100550static int dbg_parsearg(char_u *arg, garray_T *gap);
551static linenr_T debuggy_find(int file,char_u *fname, linenr_T after, garray_T *gap, int *fp);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000552
553/*
Bram Moolenaar05159a02005-02-26 23:04:13 +0000554 * Parse the arguments of ":profile", ":breakadd" or ":breakdel" and put them
555 * in the entry just after the last one in dbg_breakp. Note that "dbg_name"
556 * is allocated.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000557 * Returns FAIL for failure.
558 */
559 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100560dbg_parsearg(
561 char_u *arg,
562 garray_T *gap) /* either &dbg_breakp or &prof_ga */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000563{
564 char_u *p = arg;
565 char_u *q;
566 struct debuggy *bp;
Bram Moolenaarf4b8e572004-06-24 15:53:16 +0000567 int here = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000568
Bram Moolenaar05159a02005-02-26 23:04:13 +0000569 if (ga_grow(gap, 1) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000570 return FAIL;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000571 bp = &DEBUGGY(gap, gap->ga_len);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000572
573 /* Find "func" or "file". */
574 if (STRNCMP(p, "func", 4) == 0)
575 bp->dbg_type = DBG_FUNC;
576 else if (STRNCMP(p, "file", 4) == 0)
577 bp->dbg_type = DBG_FILE;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000578 else if (
579#ifdef FEAT_PROFILE
580 gap != &prof_ga &&
581#endif
582 STRNCMP(p, "here", 4) == 0)
Bram Moolenaarf4b8e572004-06-24 15:53:16 +0000583 {
584 if (curbuf->b_ffname == NULL)
585 {
586 EMSG(_(e_noname));
587 return FAIL;
588 }
589 bp->dbg_type = DBG_FILE;
590 here = TRUE;
591 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000592 else
593 {
594 EMSG2(_(e_invarg2), p);
595 return FAIL;
596 }
597 p = skipwhite(p + 4);
598
599 /* Find optional line number. */
Bram Moolenaarf4b8e572004-06-24 15:53:16 +0000600 if (here)
601 bp->dbg_lnum = curwin->w_cursor.lnum;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000602 else if (
603#ifdef FEAT_PROFILE
604 gap != &prof_ga &&
605#endif
606 VIM_ISDIGIT(*p))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000607 {
608 bp->dbg_lnum = getdigits(&p);
609 p = skipwhite(p);
610 }
611 else
612 bp->dbg_lnum = 0;
613
614 /* Find the function or file name. Don't accept a function name with (). */
Bram Moolenaarf4b8e572004-06-24 15:53:16 +0000615 if ((!here && *p == NUL)
616 || (here && *p != NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000617 || (bp->dbg_type == DBG_FUNC && strstr((char *)p, "()") != NULL))
618 {
619 EMSG2(_(e_invarg2), arg);
620 return FAIL;
621 }
622
623 if (bp->dbg_type == DBG_FUNC)
624 bp->dbg_name = vim_strsave(p);
Bram Moolenaarf4b8e572004-06-24 15:53:16 +0000625 else if (here)
626 bp->dbg_name = vim_strsave(curbuf->b_ffname);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000627 else
628 {
629 /* Expand the file name in the same way as do_source(). This means
630 * doing it twice, so that $DIR/file gets expanded when $DIR is
631 * "~/dir". */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000632 q = expand_env_save(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000633 if (q == NULL)
634 return FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000635 p = expand_env_save(q);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000636 vim_free(q);
637 if (p == NULL)
638 return FAIL;
Bram Moolenaar843ee412004-06-30 16:16:41 +0000639 if (*p != '*')
640 {
641 bp->dbg_name = fix_fname(p);
642 vim_free(p);
643 }
644 else
645 bp->dbg_name = p;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000646 }
647
648 if (bp->dbg_name == NULL)
649 return FAIL;
650 return OK;
651}
652
653/*
654 * ":breakadd".
655 */
656 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100657ex_breakadd(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000658{
659 struct debuggy *bp;
660 char_u *pat;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000661 garray_T *gap;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000662
Bram Moolenaar05159a02005-02-26 23:04:13 +0000663 gap = &dbg_breakp;
664#ifdef FEAT_PROFILE
665 if (eap->cmdidx == CMD_profile)
666 gap = &prof_ga;
667#endif
668
669 if (dbg_parsearg(eap->arg, gap) == OK)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000670 {
Bram Moolenaar05159a02005-02-26 23:04:13 +0000671 bp = &DEBUGGY(gap, gap->ga_len);
672 bp->dbg_forceit = eap->forceit;
673
Bram Moolenaar071d4272004-06-13 20:20:40 +0000674 pat = file_pat_to_reg_pat(bp->dbg_name, NULL, NULL, FALSE);
675 if (pat != NULL)
676 {
677 bp->dbg_prog = vim_regcomp(pat, RE_MAGIC + RE_STRING);
678 vim_free(pat);
679 }
680 if (pat == NULL || bp->dbg_prog == NULL)
681 vim_free(bp->dbg_name);
682 else
683 {
684 if (bp->dbg_lnum == 0) /* default line number is 1 */
685 bp->dbg_lnum = 1;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000686#ifdef FEAT_PROFILE
687 if (eap->cmdidx != CMD_profile)
688#endif
689 {
690 DEBUGGY(gap, gap->ga_len).dbg_nr = ++last_breakp;
691 ++debug_tick;
692 }
693 ++gap->ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000694 }
695 }
696}
697
698/*
699 * ":debuggreedy".
700 */
701 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100702ex_debuggreedy(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000703{
704 if (eap->addr_count == 0 || eap->line2 != 0)
705 debug_greedy = TRUE;
706 else
707 debug_greedy = FALSE;
708}
709
710/*
Bram Moolenaard9fba312005-06-26 22:34:35 +0000711 * ":breakdel" and ":profdel".
Bram Moolenaar071d4272004-06-13 20:20:40 +0000712 */
713 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100714ex_breakdel(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000715{
716 struct debuggy *bp, *bpi;
717 int nr;
718 int todel = -1;
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000719 int del_all = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000720 int i;
721 linenr_T best_lnum = 0;
Bram Moolenaard9fba312005-06-26 22:34:35 +0000722 garray_T *gap;
723
724 gap = &dbg_breakp;
Bram Moolenaard9fba312005-06-26 22:34:35 +0000725 if (eap->cmdidx == CMD_profdel)
Bram Moolenaar38bdbd62012-06-20 15:48:57 +0200726 {
727#ifdef FEAT_PROFILE
Bram Moolenaard9fba312005-06-26 22:34:35 +0000728 gap = &prof_ga;
Bram Moolenaar38bdbd62012-06-20 15:48:57 +0200729#else
730 ex_ni(eap);
731 return;
Bram Moolenaard9fba312005-06-26 22:34:35 +0000732#endif
Bram Moolenaar38bdbd62012-06-20 15:48:57 +0200733 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000734
735 if (vim_isdigit(*eap->arg))
736 {
737 /* ":breakdel {nr}" */
738 nr = atol((char *)eap->arg);
Bram Moolenaard9fba312005-06-26 22:34:35 +0000739 for (i = 0; i < gap->ga_len; ++i)
740 if (DEBUGGY(gap, i).dbg_nr == nr)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000741 {
742 todel = i;
743 break;
744 }
745 }
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000746 else if (*eap->arg == '*')
747 {
748 todel = 0;
749 del_all = TRUE;
750 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000751 else
752 {
753 /* ":breakdel {func|file} [lnum] {name}" */
Bram Moolenaard9fba312005-06-26 22:34:35 +0000754 if (dbg_parsearg(eap->arg, gap) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000755 return;
Bram Moolenaard9fba312005-06-26 22:34:35 +0000756 bp = &DEBUGGY(gap, gap->ga_len);
757 for (i = 0; i < gap->ga_len; ++i)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000758 {
Bram Moolenaard9fba312005-06-26 22:34:35 +0000759 bpi = &DEBUGGY(gap, i);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000760 if (bp->dbg_type == bpi->dbg_type
761 && STRCMP(bp->dbg_name, bpi->dbg_name) == 0
762 && (bp->dbg_lnum == bpi->dbg_lnum
763 || (bp->dbg_lnum == 0
764 && (best_lnum == 0
765 || bpi->dbg_lnum < best_lnum))))
766 {
767 todel = i;
768 best_lnum = bpi->dbg_lnum;
769 }
770 }
771 vim_free(bp->dbg_name);
772 }
773
774 if (todel < 0)
775 EMSG2(_("E161: Breakpoint not found: %s"), eap->arg);
776 else
Bram Moolenaard9fba312005-06-26 22:34:35 +0000777 {
778 while (gap->ga_len > 0)
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000779 {
Bram Moolenaard9fba312005-06-26 22:34:35 +0000780 vim_free(DEBUGGY(gap, todel).dbg_name);
Bram Moolenaar473de612013-06-08 18:19:48 +0200781 vim_regfree(DEBUGGY(gap, todel).dbg_prog);
Bram Moolenaard9fba312005-06-26 22:34:35 +0000782 --gap->ga_len;
783 if (todel < gap->ga_len)
784 mch_memmove(&DEBUGGY(gap, todel), &DEBUGGY(gap, todel + 1),
785 (gap->ga_len - todel) * sizeof(struct debuggy));
786#ifdef FEAT_PROFILE
787 if (eap->cmdidx == CMD_breakdel)
788#endif
789 ++debug_tick;
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000790 if (!del_all)
791 break;
792 }
Bram Moolenaard9fba312005-06-26 22:34:35 +0000793
794 /* If all breakpoints were removed clear the array. */
795 if (gap->ga_len == 0)
796 ga_clear(gap);
797 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000798}
799
800/*
801 * ":breaklist".
802 */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000803 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100804ex_breaklist(exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000805{
806 struct debuggy *bp;
807 int i;
808
809 if (dbg_breakp.ga_len == 0)
810 MSG(_("No breakpoints defined"));
811 else
812 for (i = 0; i < dbg_breakp.ga_len; ++i)
813 {
814 bp = &BREAKP(i);
Bram Moolenaard58ea072011-06-26 04:25:30 +0200815 if (bp->dbg_type == DBG_FILE)
816 home_replace(NULL, bp->dbg_name, NameBuff, MAXPATHL, TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000817 smsg((char_u *)_("%3d %s %s line %ld"),
818 bp->dbg_nr,
819 bp->dbg_type == DBG_FUNC ? "func" : "file",
Bram Moolenaard58ea072011-06-26 04:25:30 +0200820 bp->dbg_type == DBG_FUNC ? bp->dbg_name : NameBuff,
Bram Moolenaar071d4272004-06-13 20:20:40 +0000821 (long)bp->dbg_lnum);
822 }
823}
824
825/*
826 * Find a breakpoint for a function or sourced file.
827 * Returns line number at which to break; zero when no matching breakpoint.
828 */
829 linenr_T
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100830dbg_find_breakpoint(
831 int file, /* TRUE for a file, FALSE for a function */
832 char_u *fname, /* file or function name */
833 linenr_T after) /* after this line number */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000834{
Bram Moolenaar05159a02005-02-26 23:04:13 +0000835 return debuggy_find(file, fname, after, &dbg_breakp, NULL);
836}
837
838#if defined(FEAT_PROFILE) || defined(PROTO)
839/*
840 * Return TRUE if profiling is on for a function or sourced file.
841 */
842 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100843has_profiling(
844 int file, /* TRUE for a file, FALSE for a function */
845 char_u *fname, /* file or function name */
846 int *fp) /* return: forceit */
Bram Moolenaar05159a02005-02-26 23:04:13 +0000847{
848 return (debuggy_find(file, fname, (linenr_T)0, &prof_ga, fp)
849 != (linenr_T)0);
850}
851#endif
852
853/*
854 * Common code for dbg_find_breakpoint() and has_profiling().
855 */
856 static linenr_T
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100857debuggy_find(
858 int file, /* TRUE for a file, FALSE for a function */
859 char_u *fname, /* file or function name */
860 linenr_T after, /* after this line number */
861 garray_T *gap, /* either &dbg_breakp or &prof_ga */
862 int *fp) /* if not NULL: return forceit */
Bram Moolenaar05159a02005-02-26 23:04:13 +0000863{
Bram Moolenaar071d4272004-06-13 20:20:40 +0000864 struct debuggy *bp;
865 int i;
866 linenr_T lnum = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000867 char_u *name = fname;
868 int prev_got_int;
869
Bram Moolenaar05159a02005-02-26 23:04:13 +0000870 /* Return quickly when there are no breakpoints. */
871 if (gap->ga_len == 0)
872 return (linenr_T)0;
873
Bram Moolenaar071d4272004-06-13 20:20:40 +0000874 /* Replace K_SNR in function name with "<SNR>". */
875 if (!file && fname[0] == K_SPECIAL)
876 {
877 name = alloc((unsigned)STRLEN(fname) + 3);
878 if (name == NULL)
879 name = fname;
880 else
881 {
882 STRCPY(name, "<SNR>");
883 STRCPY(name + 5, fname + 3);
884 }
885 }
886
Bram Moolenaar05159a02005-02-26 23:04:13 +0000887 for (i = 0; i < gap->ga_len; ++i)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000888 {
Bram Moolenaar05159a02005-02-26 23:04:13 +0000889 /* Skip entries that are not useful or are for a line that is beyond
890 * an already found breakpoint. */
891 bp = &DEBUGGY(gap, i);
892 if (((bp->dbg_type == DBG_FILE) == file && (
893#ifdef FEAT_PROFILE
894 gap == &prof_ga ||
895#endif
896 (bp->dbg_lnum > after && (lnum == 0 || bp->dbg_lnum < lnum)))))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000897 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000898 /*
Bram Moolenaar05159a02005-02-26 23:04:13 +0000899 * Save the value of got_int and reset it. We don't want a
900 * previous interruption cancel matching, only hitting CTRL-C
901 * while matching should abort it.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000902 */
903 prev_got_int = got_int;
904 got_int = FALSE;
Bram Moolenaardffa5b82014-11-19 16:38:07 +0100905 if (vim_regexec_prog(&bp->dbg_prog, FALSE, name, (colnr_T)0))
Bram Moolenaar05159a02005-02-26 23:04:13 +0000906 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000907 lnum = bp->dbg_lnum;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000908 if (fp != NULL)
909 *fp = bp->dbg_forceit;
910 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000911 got_int |= prev_got_int;
912 }
913 }
914 if (name != fname)
915 vim_free(name);
916
917 return lnum;
918}
919
920/*
921 * Called when a breakpoint was encountered.
922 */
923 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100924dbg_breakpoint(char_u *name, linenr_T lnum)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000925{
926 /* We need to check if this line is actually executed in do_one_cmd() */
927 debug_breakpoint_name = name;
928 debug_breakpoint_lnum = lnum;
929}
Bram Moolenaar05159a02005-02-26 23:04:13 +0000930
931
Bram Moolenaar433f7c82006-03-21 21:29:36 +0000932# if defined(FEAT_PROFILE) || defined(FEAT_RELTIME) || defined(PROTO)
Bram Moolenaar05159a02005-02-26 23:04:13 +0000933/*
934 * Store the current time in "tm".
935 */
936 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100937profile_start(proftime_T *tm)
Bram Moolenaar05159a02005-02-26 23:04:13 +0000938{
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000939# ifdef WIN3264
940 QueryPerformanceCounter(tm);
941# else
Bram Moolenaar05159a02005-02-26 23:04:13 +0000942 gettimeofday(tm, NULL);
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000943# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +0000944}
945
946/*
947 * Compute the elapsed time from "tm" till now and store in "tm".
948 */
949 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100950profile_end(proftime_T *tm)
Bram Moolenaar05159a02005-02-26 23:04:13 +0000951{
952 proftime_T now;
953
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000954# ifdef WIN3264
955 QueryPerformanceCounter(&now);
956 tm->QuadPart = now.QuadPart - tm->QuadPart;
957# else
Bram Moolenaar05159a02005-02-26 23:04:13 +0000958 gettimeofday(&now, NULL);
959 tm->tv_usec = now.tv_usec - tm->tv_usec;
960 tm->tv_sec = now.tv_sec - tm->tv_sec;
961 if (tm->tv_usec < 0)
962 {
963 tm->tv_usec += 1000000;
964 --tm->tv_sec;
965 }
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000966# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +0000967}
968
969/*
970 * Subtract the time "tm2" from "tm".
971 */
972 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100973profile_sub(proftime_T *tm, proftime_T *tm2)
Bram Moolenaar05159a02005-02-26 23:04:13 +0000974{
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000975# ifdef WIN3264
976 tm->QuadPart -= tm2->QuadPart;
977# else
Bram Moolenaar05159a02005-02-26 23:04:13 +0000978 tm->tv_usec -= tm2->tv_usec;
979 tm->tv_sec -= tm2->tv_sec;
980 if (tm->tv_usec < 0)
981 {
982 tm->tv_usec += 1000000;
983 --tm->tv_sec;
984 }
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000985# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +0000986}
987
988/*
Bram Moolenaar433f7c82006-03-21 21:29:36 +0000989 * Return a string that represents the time in "tm".
990 * Uses a static buffer!
991 */
992 char *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100993profile_msg(proftime_T *tm)
Bram Moolenaar433f7c82006-03-21 21:29:36 +0000994{
995 static char buf[50];
996
997# ifdef WIN3264
998 LARGE_INTEGER fr;
999
1000 QueryPerformanceFrequency(&fr);
1001 sprintf(buf, "%10.6lf", (double)tm->QuadPart / (double)fr.QuadPart);
1002# else
1003 sprintf(buf, "%3ld.%06ld", (long)tm->tv_sec, (long)tm->tv_usec);
Bram Moolenaar76929292008-01-06 19:07:36 +00001004# endif
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001005 return buf;
1006}
1007
Bram Moolenaar79c2c882016-02-07 21:19:28 +01001008# if defined(FEAT_FLOAT) || defined(PROTO)
1009/*
1010 * Return a float that represents the time in "tm".
1011 */
1012 float_T
1013profile_float(proftime_T *tm)
1014{
1015# ifdef WIN3264
1016 LARGE_INTEGER fr;
1017
1018 QueryPerformanceFrequency(&fr);
1019 return (float_T)tm->QuadPart / (float_T)fr.QuadPart;
1020# else
1021 return (float_T)tm->tv_sec + (float_T)tm->tv_usec / 1000000.0;
1022# endif
1023}
1024# endif
1025
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001026/*
Bram Moolenaar76929292008-01-06 19:07:36 +00001027 * Put the time "msec" past now in "tm".
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001028 */
Bram Moolenaar76929292008-01-06 19:07:36 +00001029 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001030profile_setlimit(long msec, proftime_T *tm)
Bram Moolenaar76929292008-01-06 19:07:36 +00001031{
1032 if (msec <= 0) /* no limit */
1033 profile_zero(tm);
1034 else
1035 {
1036# ifdef WIN3264
1037 LARGE_INTEGER fr;
1038
1039 QueryPerformanceCounter(tm);
1040 QueryPerformanceFrequency(&fr);
Bram Moolenaarb3c70982008-01-18 10:40:55 +00001041 tm->QuadPart += (LONGLONG)((double)msec / 1000.0 * (double)fr.QuadPart);
Bram Moolenaar76929292008-01-06 19:07:36 +00001042# else
1043 long usec;
1044
1045 gettimeofday(tm, NULL);
1046 usec = (long)tm->tv_usec + (long)msec * 1000;
1047 tm->tv_usec = usec % 1000000L;
1048 tm->tv_sec += usec / 1000000L;
1049# endif
1050 }
1051}
1052
1053/*
1054 * Return TRUE if the current time is past "tm".
1055 */
1056 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001057profile_passed_limit(proftime_T *tm)
Bram Moolenaar76929292008-01-06 19:07:36 +00001058{
1059 proftime_T now;
1060
1061# ifdef WIN3264
1062 if (tm->QuadPart == 0) /* timer was not set */
1063 return FALSE;
1064 QueryPerformanceCounter(&now);
1065 return (now.QuadPart > tm->QuadPart);
1066# else
1067 if (tm->tv_sec == 0) /* timer was not set */
1068 return FALSE;
1069 gettimeofday(&now, NULL);
1070 return (now.tv_sec > tm->tv_sec
1071 || (now.tv_sec == tm->tv_sec && now.tv_usec > tm->tv_usec));
1072# endif
1073}
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001074
1075/*
1076 * Set the time in "tm" to zero.
1077 */
1078 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001079profile_zero(proftime_T *tm)
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001080{
1081# ifdef WIN3264
1082 tm->QuadPart = 0;
1083# else
1084 tm->tv_usec = 0;
1085 tm->tv_sec = 0;
1086# endif
1087}
1088
Bram Moolenaar76929292008-01-06 19:07:36 +00001089# endif /* FEAT_PROFILE || FEAT_RELTIME */
1090
Bram Moolenaar975b5272016-03-15 23:10:59 +01001091# if defined(FEAT_TIMERS) || defined(PROTO)
1092static timer_T *first_timer = NULL;
Bram Moolenaar75537a92016-09-05 22:45:28 +02001093static long last_timer_id = 0;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001094
Bram Moolenaara8e93d62017-09-18 21:50:47 +02001095 static long
Bram Moolenaar51b0f372017-11-18 18:52:04 +01001096proftime_time_left(proftime_T *due, proftime_T *now)
Bram Moolenaara8e93d62017-09-18 21:50:47 +02001097{
1098# ifdef WIN3264
1099 LARGE_INTEGER fr;
1100
Bram Moolenaar51b0f372017-11-18 18:52:04 +01001101 if (now->QuadPart > due->QuadPart)
Bram Moolenaara8e93d62017-09-18 21:50:47 +02001102 return 0;
1103 QueryPerformanceFrequency(&fr);
Bram Moolenaar51b0f372017-11-18 18:52:04 +01001104 return (long)(((double)(due->QuadPart - now->QuadPart)
Bram Moolenaara8e93d62017-09-18 21:50:47 +02001105 / (double)fr.QuadPart) * 1000);
1106# else
Bram Moolenaar51b0f372017-11-18 18:52:04 +01001107 if (now->tv_sec > due->tv_sec)
Bram Moolenaara8e93d62017-09-18 21:50:47 +02001108 return 0;
Bram Moolenaar51b0f372017-11-18 18:52:04 +01001109 return (due->tv_sec - now->tv_sec) * 1000
1110 + (due->tv_usec - now->tv_usec) / 1000;
Bram Moolenaara8e93d62017-09-18 21:50:47 +02001111# endif
1112}
Bram Moolenaar417ccd72016-09-01 21:26:20 +02001113
Bram Moolenaar975b5272016-03-15 23:10:59 +01001114/*
1115 * Insert a timer in the list of timers.
1116 */
1117 static void
1118insert_timer(timer_T *timer)
1119{
1120 timer->tr_next = first_timer;
1121 timer->tr_prev = NULL;
1122 if (first_timer != NULL)
1123 first_timer->tr_prev = timer;
1124 first_timer = timer;
Bram Moolenaar4231da42016-06-02 14:30:04 +02001125 did_add_timer = TRUE;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001126}
1127
1128/*
1129 * Take a timer out of the list of timers.
1130 */
1131 static void
1132remove_timer(timer_T *timer)
1133{
1134 if (timer->tr_prev == NULL)
1135 first_timer = timer->tr_next;
1136 else
1137 timer->tr_prev->tr_next = timer->tr_next;
1138 if (timer->tr_next != NULL)
1139 timer->tr_next->tr_prev = timer->tr_prev;
1140}
1141
1142 static void
1143free_timer(timer_T *timer)
1144{
Bram Moolenaar75537a92016-09-05 22:45:28 +02001145 free_callback(timer->tr_callback, timer->tr_partial);
1146 vim_free(timer);
Bram Moolenaar975b5272016-03-15 23:10:59 +01001147}
1148
1149/*
1150 * Create a timer and return it. NULL if out of memory.
1151 * Caller should set the callback.
1152 */
1153 timer_T *
1154create_timer(long msec, int repeat)
1155{
1156 timer_T *timer = (timer_T *)alloc_clear(sizeof(timer_T));
Bram Moolenaaree39ef02016-09-10 19:17:42 +02001157 long prev_id = last_timer_id;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001158
1159 if (timer == NULL)
1160 return NULL;
Bram Moolenaaree39ef02016-09-10 19:17:42 +02001161 if (++last_timer_id <= prev_id)
Bram Moolenaar75537a92016-09-05 22:45:28 +02001162 /* Overflow! Might cause duplicates... */
1163 last_timer_id = 0;
1164 timer->tr_id = last_timer_id;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001165 insert_timer(timer);
1166 if (repeat != 0)
Bram Moolenaar975b5272016-03-15 23:10:59 +01001167 timer->tr_repeat = repeat - 1;
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001168 timer->tr_interval = msec;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001169
1170 profile_setlimit(msec, &timer->tr_due);
1171 return timer;
1172}
1173
1174/*
1175 * Invoke the callback of "timer".
1176 */
1177 static void
1178timer_callback(timer_T *timer)
1179{
1180 typval_T rettv;
1181 int dummy;
1182 typval_T argv[2];
1183
1184 argv[0].v_type = VAR_NUMBER;
Bram Moolenaar75537a92016-09-05 22:45:28 +02001185 argv[0].vval.v_number = (varnumber_T)timer->tr_id;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001186 argv[1].v_type = VAR_UNKNOWN;
1187
1188 call_func(timer->tr_callback, (int)STRLEN(timer->tr_callback),
Bram Moolenaardf48fb42016-07-22 21:50:18 +02001189 &rettv, 1, argv, NULL, 0L, 0L, &dummy, TRUE,
Bram Moolenaar975b5272016-03-15 23:10:59 +01001190 timer->tr_partial, NULL);
1191 clear_tv(&rettv);
1192}
1193
1194/*
1195 * Call timers that are due.
1196 * Return the time in msec until the next timer is due.
Bram Moolenaarc4f83382017-07-07 14:50:44 +02001197 * Returns -1 if there are no pending timers.
Bram Moolenaar975b5272016-03-15 23:10:59 +01001198 */
1199 long
Bram Moolenaarcf089462016-06-12 21:18:43 +02001200check_due_timer(void)
Bram Moolenaar975b5272016-03-15 23:10:59 +01001201{
1202 timer_T *timer;
Bram Moolenaar75537a92016-09-05 22:45:28 +02001203 timer_T *timer_next;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001204 long this_due;
Bram Moolenaar597385a2016-03-16 23:24:43 +01001205 long next_due = -1;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001206 proftime_T now;
1207 int did_one = FALSE;
Bram Moolenaar02e177d2017-08-26 23:43:28 +02001208 int need_update_screen = FALSE;
Bram Moolenaar75537a92016-09-05 22:45:28 +02001209 long current_id = last_timer_id;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001210
Bram Moolenaarc577d812017-07-08 22:37:34 +02001211 /* Don't run any timers while exiting or dealing with an error. */
1212 if (exiting || aborting())
Bram Moolenaarc4f83382017-07-07 14:50:44 +02001213 return next_due;
1214
Bram Moolenaar75537a92016-09-05 22:45:28 +02001215 profile_start(&now);
1216 for (timer = first_timer; timer != NULL && !got_int; timer = timer_next)
Bram Moolenaar975b5272016-03-15 23:10:59 +01001217 {
Bram Moolenaar75537a92016-09-05 22:45:28 +02001218 timer_next = timer->tr_next;
Bram Moolenaar417ccd72016-09-01 21:26:20 +02001219
Bram Moolenaar75537a92016-09-05 22:45:28 +02001220 if (timer->tr_id == -1 || timer->tr_firing || timer->tr_paused)
1221 continue;
Bram Moolenaar51b0f372017-11-18 18:52:04 +01001222 this_due = proftime_time_left(&timer->tr_due, &now);
Bram Moolenaar75537a92016-09-05 22:45:28 +02001223 if (this_due <= 1)
1224 {
Bram Moolenaar1e8e1452017-06-24 16:03:06 +02001225 int save_timer_busy = timer_busy;
1226 int save_vgetc_busy = vgetc_busy;
Bram Moolenaare723c422017-09-06 23:40:10 +02001227 int save_did_emsg = did_emsg;
1228 int save_called_emsg = called_emsg;
Bram Moolenaar02e177d2017-08-26 23:43:28 +02001229 int save_must_redraw = must_redraw;
Bram Moolenaare723c422017-09-06 23:40:10 +02001230 int save_trylevel = trylevel;
1231 int save_did_throw = did_throw;
Bram Moolenaarf5291f32017-09-14 22:55:37 +02001232 int save_ex_pressedreturn = get_pressedreturn();
Bram Moolenaare723c422017-09-06 23:40:10 +02001233 except_T *save_current_exception = current_exception;
Bram Moolenaar1e8e1452017-06-24 16:03:06 +02001234
Bram Moolenaare723c422017-09-06 23:40:10 +02001235 /* Create a scope for running the timer callback, ignoring most of
1236 * the current scope, such as being inside a try/catch. */
Bram Moolenaar1e8e1452017-06-24 16:03:06 +02001237 timer_busy = timer_busy > 0 || vgetc_busy > 0;
1238 vgetc_busy = 0;
Bram Moolenaarc577d812017-07-08 22:37:34 +02001239 called_emsg = FALSE;
Bram Moolenaare723c422017-09-06 23:40:10 +02001240 did_emsg = FALSE;
1241 did_uncaught_emsg = FALSE;
Bram Moolenaar02e177d2017-08-26 23:43:28 +02001242 must_redraw = 0;
Bram Moolenaare723c422017-09-06 23:40:10 +02001243 trylevel = 0;
1244 did_throw = FALSE;
1245 current_exception = NULL;
1246
Bram Moolenaar75537a92016-09-05 22:45:28 +02001247 timer->tr_firing = TRUE;
1248 timer_callback(timer);
1249 timer->tr_firing = FALSE;
Bram Moolenaare723c422017-09-06 23:40:10 +02001250
Bram Moolenaar75537a92016-09-05 22:45:28 +02001251 timer_next = timer->tr_next;
1252 did_one = TRUE;
Bram Moolenaar1e8e1452017-06-24 16:03:06 +02001253 timer_busy = save_timer_busy;
1254 vgetc_busy = save_vgetc_busy;
Bram Moolenaare723c422017-09-06 23:40:10 +02001255 if (did_uncaught_emsg)
Bram Moolenaarc577d812017-07-08 22:37:34 +02001256 ++timer->tr_emsg_count;
Bram Moolenaare723c422017-09-06 23:40:10 +02001257 did_emsg = save_did_emsg;
1258 called_emsg = save_called_emsg;
1259 trylevel = save_trylevel;
1260 did_throw = save_did_throw;
1261 current_exception = save_current_exception;
Bram Moolenaar02e177d2017-08-26 23:43:28 +02001262 if (must_redraw != 0)
1263 need_update_screen = TRUE;
1264 must_redraw = must_redraw > save_must_redraw
1265 ? must_redraw : save_must_redraw;
Bram Moolenaarf5291f32017-09-14 22:55:37 +02001266 set_pressedreturn(save_ex_pressedreturn);
Bram Moolenaar75537a92016-09-05 22:45:28 +02001267
1268 /* Only fire the timer again if it repeats and stop_timer() wasn't
1269 * called while inside the callback (tr_id == -1). */
Bram Moolenaarc577d812017-07-08 22:37:34 +02001270 if (timer->tr_repeat != 0 && timer->tr_id != -1
1271 && timer->tr_emsg_count < 3)
Bram Moolenaar75537a92016-09-05 22:45:28 +02001272 {
1273 profile_setlimit(timer->tr_interval, &timer->tr_due);
Bram Moolenaar51b0f372017-11-18 18:52:04 +01001274 this_due = proftime_time_left(&timer->tr_due, &now);
Bram Moolenaar75537a92016-09-05 22:45:28 +02001275 if (this_due < 1)
1276 this_due = 1;
1277 if (timer->tr_repeat > 0)
1278 --timer->tr_repeat;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001279 }
Bram Moolenaar75537a92016-09-05 22:45:28 +02001280 else
1281 {
1282 this_due = -1;
1283 remove_timer(timer);
1284 free_timer(timer);
1285 }
Bram Moolenaar975b5272016-03-15 23:10:59 +01001286 }
Bram Moolenaar75537a92016-09-05 22:45:28 +02001287 if (this_due > 0 && (next_due == -1 || next_due > this_due))
1288 next_due = this_due;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001289 }
1290
1291 if (did_one)
Bram Moolenaar02e177d2017-08-26 23:43:28 +02001292 redraw_after_callback(need_update_screen);
Bram Moolenaar975b5272016-03-15 23:10:59 +01001293
Bram Moolenaarc3719bd2017-11-18 22:13:31 +01001294#ifdef FEAT_BEVAL_TERM
Bram Moolenaar51b0f372017-11-18 18:52:04 +01001295 if (bevalexpr_due_set)
1296 {
1297 this_due = proftime_time_left(&bevalexpr_due, &now);
1298 if (this_due <= 1)
1299 {
1300 bevalexpr_due_set = FALSE;
Bram Moolenaar51b0f372017-11-18 18:52:04 +01001301 if (balloonEval == NULL)
1302 {
1303 balloonEval = (BalloonEval *)alloc(sizeof(BalloonEval));
1304 balloonEvalForTerm = TRUE;
1305 }
1306 if (balloonEval != NULL)
1307 general_beval_cb(balloonEval, 0);
1308 }
1309 else if (this_due > 0 && (next_due == -1 || next_due > this_due))
1310 next_due = this_due;
1311 }
1312#endif
1313
Bram Moolenaar75537a92016-09-05 22:45:28 +02001314 return current_id != last_timer_id ? 1 : next_due;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001315}
1316
1317/*
1318 * Find a timer by ID. Returns NULL if not found;
1319 */
1320 timer_T *
Bram Moolenaar75537a92016-09-05 22:45:28 +02001321find_timer(long id)
Bram Moolenaar975b5272016-03-15 23:10:59 +01001322{
1323 timer_T *timer;
1324
Bram Moolenaar75537a92016-09-05 22:45:28 +02001325 if (id >= 0)
1326 {
1327 for (timer = first_timer; timer != NULL; timer = timer->tr_next)
1328 if (timer->tr_id == id)
1329 return timer;
1330 }
1331 return NULL;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001332}
1333
1334
1335/*
1336 * Stop a timer and delete it.
1337 */
1338 void
1339stop_timer(timer_T *timer)
1340{
Bram Moolenaar75537a92016-09-05 22:45:28 +02001341 if (timer->tr_firing)
1342 /* Free the timer after the callback returns. */
1343 timer->tr_id = -1;
1344 else
1345 {
1346 remove_timer(timer);
1347 free_timer(timer);
1348 }
Bram Moolenaar975b5272016-03-15 23:10:59 +01001349}
Bram Moolenaare3188e22016-05-31 21:13:04 +02001350
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001351 void
Bram Moolenaarb73598e2016-08-07 18:22:53 +02001352stop_all_timers(void)
1353{
Bram Moolenaar75537a92016-09-05 22:45:28 +02001354 timer_T *timer;
1355 timer_T *timer_next;
1356
1357 for (timer = first_timer; timer != NULL; timer = timer_next)
1358 {
1359 timer_next = timer->tr_next;
1360 stop_timer(timer);
1361 }
Bram Moolenaarb73598e2016-08-07 18:22:53 +02001362}
1363
1364 void
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001365add_timer_info(typval_T *rettv, timer_T *timer)
1366{
1367 list_T *list = rettv->vval.v_list;
1368 dict_T *dict = dict_alloc();
1369 dictitem_T *di;
1370 long remaining;
1371 proftime_T now;
1372
1373 if (dict == NULL)
1374 return;
1375 list_append_dict(list, dict);
1376
Bram Moolenaar75537a92016-09-05 22:45:28 +02001377 dict_add_nr_str(dict, "id", timer->tr_id, NULL);
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001378 dict_add_nr_str(dict, "time", (long)timer->tr_interval, NULL);
1379
1380 profile_start(&now);
Bram Moolenaar51b0f372017-11-18 18:52:04 +01001381 remaining = proftime_time_left(&timer->tr_due, &now);
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001382 dict_add_nr_str(dict, "remaining", (long)remaining, NULL);
1383
1384 dict_add_nr_str(dict, "repeat",
1385 (long)(timer->tr_repeat < 0 ? -1 : timer->tr_repeat + 1), NULL);
Bram Moolenaarb73598e2016-08-07 18:22:53 +02001386 dict_add_nr_str(dict, "paused", (long)(timer->tr_paused), NULL);
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001387
1388 di = dictitem_alloc((char_u *)"callback");
1389 if (di != NULL)
1390 {
1391 if (dict_add(dict, di) == FAIL)
1392 vim_free(di);
1393 else if (timer->tr_partial != NULL)
1394 {
1395 di->di_tv.v_type = VAR_PARTIAL;
1396 di->di_tv.vval.v_partial = timer->tr_partial;
1397 ++timer->tr_partial->pt_refcount;
1398 }
1399 else
1400 {
1401 di->di_tv.v_type = VAR_FUNC;
1402 di->di_tv.vval.v_string = vim_strsave(timer->tr_callback);
1403 }
1404 di->di_tv.v_lock = 0;
1405 }
1406}
1407
1408 void
1409add_timer_info_all(typval_T *rettv)
1410{
1411 timer_T *timer;
1412
1413 for (timer = first_timer; timer != NULL; timer = timer->tr_next)
Bram Moolenaar75537a92016-09-05 22:45:28 +02001414 if (timer->tr_id != -1)
1415 add_timer_info(rettv, timer);
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001416}
1417
Bram Moolenaare3188e22016-05-31 21:13:04 +02001418/*
1419 * Mark references in partials of timers.
1420 */
1421 int
1422set_ref_in_timer(int copyID)
1423{
1424 int abort = FALSE;
1425 timer_T *timer;
1426 typval_T tv;
1427
1428 for (timer = first_timer; timer != NULL; timer = timer->tr_next)
1429 {
Bram Moolenaar1e96d9b2016-07-29 22:15:09 +02001430 if (timer->tr_partial != NULL)
1431 {
1432 tv.v_type = VAR_PARTIAL;
1433 tv.vval.v_partial = timer->tr_partial;
1434 }
1435 else
1436 {
1437 tv.v_type = VAR_FUNC;
1438 tv.vval.v_string = timer->tr_callback;
1439 }
Bram Moolenaare3188e22016-05-31 21:13:04 +02001440 abort = abort || set_ref_in_item(&tv, copyID, NULL, NULL);
1441 }
1442 return abort;
1443}
Bram Moolenaar623e2632016-07-30 22:47:56 +02001444
1445# if defined(EXITFREE) || defined(PROTO)
1446 void
1447timer_free_all()
1448{
1449 timer_T *timer;
1450
1451 while (first_timer != NULL)
1452 {
1453 timer = first_timer;
1454 remove_timer(timer);
1455 free_timer(timer);
1456 }
1457}
1458# endif
Bram Moolenaar975b5272016-03-15 23:10:59 +01001459# endif
1460
Bram Moolenaar8a7f5a22013-06-06 14:01:46 +02001461#if defined(FEAT_SYN_HL) && defined(FEAT_RELTIME) && defined(FEAT_FLOAT)
1462# if defined(HAVE_MATH_H)
1463# include <math.h>
1464# endif
1465
1466/*
1467 * Divide the time "tm" by "count" and store in "tm2".
1468 */
1469 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001470profile_divide(proftime_T *tm, int count, proftime_T *tm2)
Bram Moolenaar8a7f5a22013-06-06 14:01:46 +02001471{
1472 if (count == 0)
1473 profile_zero(tm2);
1474 else
1475 {
1476# ifdef WIN3264
1477 tm2->QuadPart = tm->QuadPart / count;
1478# else
1479 double usec = (tm->tv_sec * 1000000.0 + tm->tv_usec) / count;
1480
1481 tm2->tv_sec = floor(usec / 1000000.0);
Bram Moolenaara2e14fc2013-06-10 20:10:44 +02001482 tm2->tv_usec = vim_round(usec - (tm2->tv_sec * 1000000.0));
Bram Moolenaar8a7f5a22013-06-06 14:01:46 +02001483# endif
1484 }
1485}
1486#endif
1487
Bram Moolenaar76929292008-01-06 19:07:36 +00001488# if defined(FEAT_PROFILE) || defined(PROTO)
1489/*
1490 * Functions for profiling.
1491 */
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01001492static void script_do_profile(scriptitem_T *si);
1493static void script_dump_profile(FILE *fd);
Bram Moolenaar76929292008-01-06 19:07:36 +00001494static proftime_T prof_wait_time;
1495
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001496/*
Bram Moolenaar05159a02005-02-26 23:04:13 +00001497 * Add the time "tm2" to "tm".
1498 */
1499 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001500profile_add(proftime_T *tm, proftime_T *tm2)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001501{
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001502# ifdef WIN3264
1503 tm->QuadPart += tm2->QuadPart;
1504# else
Bram Moolenaar05159a02005-02-26 23:04:13 +00001505 tm->tv_usec += tm2->tv_usec;
1506 tm->tv_sec += tm2->tv_sec;
1507 if (tm->tv_usec >= 1000000)
1508 {
1509 tm->tv_usec -= 1000000;
1510 ++tm->tv_sec;
1511 }
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001512# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00001513}
1514
1515/*
Bram Moolenaar1056d982006-03-09 22:37:52 +00001516 * Add the "self" time from the total time and the children's time.
1517 */
1518 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001519profile_self(proftime_T *self, proftime_T *total, proftime_T *children)
Bram Moolenaar1056d982006-03-09 22:37:52 +00001520{
1521 /* Check that the result won't be negative. Can happen with recursive
1522 * calls. */
1523#ifdef WIN3264
1524 if (total->QuadPart <= children->QuadPart)
1525 return;
1526#else
1527 if (total->tv_sec < children->tv_sec
1528 || (total->tv_sec == children->tv_sec
1529 && total->tv_usec <= children->tv_usec))
1530 return;
1531#endif
1532 profile_add(self, total);
1533 profile_sub(self, children);
1534}
1535
1536/*
Bram Moolenaar05159a02005-02-26 23:04:13 +00001537 * Get the current waittime.
1538 */
1539 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001540profile_get_wait(proftime_T *tm)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001541{
1542 *tm = prof_wait_time;
1543}
1544
1545/*
1546 * Subtract the passed waittime since "tm" from "tma".
1547 */
1548 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001549profile_sub_wait(proftime_T *tm, proftime_T *tma)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001550{
1551 proftime_T tm3 = prof_wait_time;
1552
1553 profile_sub(&tm3, tm);
1554 profile_sub(tma, &tm3);
1555}
1556
1557/*
1558 * Return TRUE if "tm1" and "tm2" are equal.
1559 */
1560 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001561profile_equal(proftime_T *tm1, proftime_T *tm2)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001562{
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001563# ifdef WIN3264
1564 return (tm1->QuadPart == tm2->QuadPart);
1565# else
Bram Moolenaar05159a02005-02-26 23:04:13 +00001566 return (tm1->tv_usec == tm2->tv_usec && tm1->tv_sec == tm2->tv_sec);
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001567# endif
1568}
1569
1570/*
1571 * Return <0, 0 or >0 if "tm1" < "tm2", "tm1" == "tm2" or "tm1" > "tm2"
1572 */
1573 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001574profile_cmp(const proftime_T *tm1, const proftime_T *tm2)
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001575{
1576# ifdef WIN3264
1577 return (int)(tm2->QuadPart - tm1->QuadPart);
1578# else
1579 if (tm1->tv_sec == tm2->tv_sec)
1580 return tm2->tv_usec - tm1->tv_usec;
1581 return tm2->tv_sec - tm1->tv_sec;
1582# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00001583}
1584
Bram Moolenaar05159a02005-02-26 23:04:13 +00001585static char_u *profile_fname = NULL;
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001586static proftime_T pause_time;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001587
1588/*
1589 * ":profile cmd args"
1590 */
1591 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001592ex_profile(exarg_T *eap)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001593{
1594 char_u *e;
1595 int len;
1596
1597 e = skiptowhite(eap->arg);
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00001598 len = (int)(e - eap->arg);
Bram Moolenaar05159a02005-02-26 23:04:13 +00001599 e = skipwhite(e);
1600
1601 if (len == 5 && STRNCMP(eap->arg, "start", 5) == 0 && *e != NUL)
1602 {
1603 vim_free(profile_fname);
Bram Moolenaard94682f2015-04-13 15:37:56 +02001604 profile_fname = expand_env_save_opt(e, TRUE);
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001605 do_profiling = PROF_YES;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001606 profile_zero(&prof_wait_time);
1607 set_vim_var_nr(VV_PROFILING, 1L);
1608 }
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001609 else if (do_profiling == PROF_NONE)
Bram Moolenaar54c1b492010-02-24 14:01:28 +01001610 EMSG(_("E750: First use \":profile start {fname}\""));
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001611 else if (STRCMP(eap->arg, "pause") == 0)
1612 {
1613 if (do_profiling == PROF_YES)
1614 profile_start(&pause_time);
1615 do_profiling = PROF_PAUSED;
1616 }
1617 else if (STRCMP(eap->arg, "continue") == 0)
1618 {
1619 if (do_profiling == PROF_PAUSED)
1620 {
1621 profile_end(&pause_time);
1622 profile_add(&prof_wait_time, &pause_time);
1623 }
1624 do_profiling = PROF_YES;
1625 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00001626 else
1627 {
1628 /* The rest is similar to ":breakadd". */
1629 ex_breakadd(eap);
1630 }
1631}
1632
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001633/* Command line expansion for :profile. */
1634static enum
1635{
1636 PEXP_SUBCMD, /* expand :profile sub-commands */
Bram Moolenaar49789dc2011-02-25 14:46:09 +01001637 PEXP_FUNC /* expand :profile func {funcname} */
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001638} pexpand_what;
1639
1640static char *pexpand_cmds[] = {
1641 "start",
1642#define PROFCMD_START 0
1643 "pause",
1644#define PROFCMD_PAUSE 1
1645 "continue",
1646#define PROFCMD_CONTINUE 2
1647 "func",
1648#define PROFCMD_FUNC 3
1649 "file",
1650#define PROFCMD_FILE 4
1651 NULL
1652#define PROFCMD_LAST 5
1653};
1654
1655/*
1656 * Function given to ExpandGeneric() to obtain the profile command
1657 * specific expansion.
1658 */
1659 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001660get_profile_name(expand_T *xp UNUSED, int idx)
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001661{
1662 switch (pexpand_what)
1663 {
1664 case PEXP_SUBCMD:
1665 return (char_u *)pexpand_cmds[idx];
1666 /* case PEXP_FUNC: TODO */
1667 default:
1668 return NULL;
1669 }
1670}
1671
1672/*
1673 * Handle command line completion for :profile command.
1674 */
1675 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001676set_context_in_profile_cmd(expand_T *xp, char_u *arg)
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001677{
1678 char_u *end_subcmd;
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001679
1680 /* Default: expand subcommands. */
1681 xp->xp_context = EXPAND_PROFILE;
1682 pexpand_what = PEXP_SUBCMD;
1683 xp->xp_pattern = arg;
1684
1685 end_subcmd = skiptowhite(arg);
1686 if (*end_subcmd == NUL)
1687 return;
1688
Bram Moolenaar8b9c05f2010-03-02 17:54:33 +01001689 if (end_subcmd - arg == 5 && STRNCMP(arg, "start", 5) == 0)
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001690 {
1691 xp->xp_context = EXPAND_FILES;
1692 xp->xp_pattern = skipwhite(end_subcmd);
1693 return;
1694 }
1695
1696 /* TODO: expand function names after "func" */
1697 xp->xp_context = EXPAND_NOTHING;
1698}
1699
Bram Moolenaar05159a02005-02-26 23:04:13 +00001700/*
1701 * Dump the profiling info.
1702 */
1703 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001704profile_dump(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001705{
1706 FILE *fd;
1707
1708 if (profile_fname != NULL)
1709 {
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +00001710 fd = mch_fopen((char *)profile_fname, "w");
Bram Moolenaar05159a02005-02-26 23:04:13 +00001711 if (fd == NULL)
1712 EMSG2(_(e_notopen), profile_fname);
1713 else
1714 {
Bram Moolenaar05159a02005-02-26 23:04:13 +00001715 script_dump_profile(fd);
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001716 func_dump_profile(fd);
Bram Moolenaar05159a02005-02-26 23:04:13 +00001717 fclose(fd);
1718 }
1719 }
1720}
1721
1722/*
1723 * Start profiling script "fp".
1724 */
1725 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001726script_do_profile(scriptitem_T *si)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001727{
1728 si->sn_pr_count = 0;
1729 profile_zero(&si->sn_pr_total);
1730 profile_zero(&si->sn_pr_self);
1731
1732 ga_init2(&si->sn_prl_ga, sizeof(sn_prl_T), 100);
1733 si->sn_prl_idx = -1;
1734 si->sn_prof_on = TRUE;
1735 si->sn_pr_nest = 0;
1736}
1737
1738/*
Bram Moolenaar67435d92017-10-19 21:04:37 +02001739 * Save time when starting to invoke another script or function.
Bram Moolenaar05159a02005-02-26 23:04:13 +00001740 */
1741 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001742script_prof_save(
1743 proftime_T *tm) /* place to store wait time */
Bram Moolenaar05159a02005-02-26 23:04:13 +00001744{
1745 scriptitem_T *si;
1746
1747 if (current_SID > 0 && current_SID <= script_items.ga_len)
1748 {
1749 si = &SCRIPT_ITEM(current_SID);
1750 if (si->sn_prof_on && si->sn_pr_nest++ == 0)
1751 profile_start(&si->sn_pr_child);
1752 }
1753 profile_get_wait(tm);
1754}
1755
1756/*
1757 * Count time spent in children after invoking another script or function.
1758 */
1759 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001760script_prof_restore(proftime_T *tm)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001761{
1762 scriptitem_T *si;
1763
1764 if (current_SID > 0 && current_SID <= script_items.ga_len)
1765 {
1766 si = &SCRIPT_ITEM(current_SID);
1767 if (si->sn_prof_on && --si->sn_pr_nest == 0)
1768 {
1769 profile_end(&si->sn_pr_child);
1770 profile_sub_wait(tm, &si->sn_pr_child); /* don't count wait time */
1771 profile_add(&si->sn_pr_children, &si->sn_pr_child);
1772 profile_add(&si->sn_prl_children, &si->sn_pr_child);
1773 }
1774 }
1775}
1776
1777static proftime_T inchar_time;
1778
1779/*
1780 * Called when starting to wait for the user to type a character.
1781 */
1782 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001783prof_inchar_enter(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001784{
1785 profile_start(&inchar_time);
1786}
1787
1788/*
1789 * Called when finished waiting for the user to type a character.
1790 */
1791 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001792prof_inchar_exit(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001793{
1794 profile_end(&inchar_time);
1795 profile_add(&prof_wait_time, &inchar_time);
1796}
1797
1798/*
1799 * Dump the profiling results for all scripts in file "fd".
1800 */
1801 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001802script_dump_profile(FILE *fd)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001803{
1804 int id;
1805 scriptitem_T *si;
1806 int i;
1807 FILE *sfd;
1808 sn_prl_T *pp;
1809
1810 for (id = 1; id <= script_items.ga_len; ++id)
1811 {
1812 si = &SCRIPT_ITEM(id);
1813 if (si->sn_prof_on)
1814 {
1815 fprintf(fd, "SCRIPT %s\n", si->sn_name);
1816 if (si->sn_pr_count == 1)
1817 fprintf(fd, "Sourced 1 time\n");
1818 else
1819 fprintf(fd, "Sourced %d times\n", si->sn_pr_count);
1820 fprintf(fd, "Total time: %s\n", profile_msg(&si->sn_pr_total));
1821 fprintf(fd, " Self time: %s\n", profile_msg(&si->sn_pr_self));
1822 fprintf(fd, "\n");
1823 fprintf(fd, "count total (s) self (s)\n");
1824
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +00001825 sfd = mch_fopen((char *)si->sn_name, "r");
Bram Moolenaar05159a02005-02-26 23:04:13 +00001826 if (sfd == NULL)
1827 fprintf(fd, "Cannot open file!\n");
1828 else
1829 {
Bram Moolenaar67435d92017-10-19 21:04:37 +02001830 /* Keep going till the end of file, so that trailing
1831 * continuation lines are listed. */
1832 for (i = 0; ; ++i)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001833 {
1834 if (vim_fgets(IObuff, IOSIZE, sfd))
1835 break;
Bram Moolenaarac112f02017-12-05 16:46:28 +01001836 /* When a line has been truncated, append NL, taking care
1837 * of multi-byte characters . */
1838 if (IObuff[IOSIZE - 2] != NUL && IObuff[IOSIZE - 2] != NL)
1839 {
1840 int n = IOSIZE - 2;
1841# ifdef FEAT_MBYTE
1842 if (enc_utf8)
1843 {
1844 /* Move to the first byte of this char.
1845 * utf_head_off() doesn't work, because it checks
1846 * for a truncated character. */
1847 while (n > 0 && (IObuff[n] & 0xc0) == 0x80)
1848 --n;
1849 }
1850 else if (has_mbyte)
1851 n -= mb_head_off(IObuff, IObuff + n);
1852# endif
1853 IObuff[n] = NL;
1854 IObuff[n + 1] = NUL;
1855 }
Bram Moolenaar67435d92017-10-19 21:04:37 +02001856 if (i < si->sn_prl_ga.ga_len
1857 && (pp = &PRL_ITEM(si, i))->snp_count > 0)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001858 {
1859 fprintf(fd, "%5d ", pp->snp_count);
1860 if (profile_equal(&pp->sn_prl_total, &pp->sn_prl_self))
1861 fprintf(fd, " ");
1862 else
1863 fprintf(fd, "%s ", profile_msg(&pp->sn_prl_total));
1864 fprintf(fd, "%s ", profile_msg(&pp->sn_prl_self));
1865 }
1866 else
1867 fprintf(fd, " ");
1868 fprintf(fd, "%s", IObuff);
1869 }
1870 fclose(sfd);
1871 }
1872 fprintf(fd, "\n");
1873 }
1874 }
1875}
1876
1877/*
1878 * Return TRUE when a function defined in the current script should be
1879 * profiled.
1880 */
1881 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001882prof_def_func(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001883{
Bram Moolenaar53180ce2005-07-05 21:48:14 +00001884 if (current_SID > 0)
1885 return SCRIPT_ITEM(current_SID).sn_pr_force;
1886 return FALSE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001887}
1888
1889# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001890#endif
1891
1892/*
1893 * If 'autowrite' option set, try to write the file.
1894 * Careful: autocommands may make "buf" invalid!
1895 *
1896 * return FAIL for failure, OK otherwise
1897 */
1898 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001899autowrite(buf_T *buf, int forceit)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001900{
Bram Moolenaar373154b2007-02-13 05:19:30 +00001901 int r;
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001902 bufref_T bufref;
Bram Moolenaar373154b2007-02-13 05:19:30 +00001903
Bram Moolenaar071d4272004-06-13 20:20:40 +00001904 if (!(p_aw || p_awa) || !p_write
1905#ifdef FEAT_QUICKFIX
Bram Moolenaar373154b2007-02-13 05:19:30 +00001906 /* never autowrite a "nofile" or "nowrite" buffer */
1907 || bt_dontwrite(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001908#endif
Bram Moolenaar373154b2007-02-13 05:19:30 +00001909 || (!forceit && buf->b_p_ro) || buf->b_ffname == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001910 return FAIL;
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001911 set_bufref(&bufref, buf);
Bram Moolenaar373154b2007-02-13 05:19:30 +00001912 r = buf_write_all(buf, forceit);
1913
1914 /* Writing may succeed but the buffer still changed, e.g., when there is a
1915 * conversion error. We do want to return FAIL then. */
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001916 if (bufref_valid(&bufref) && bufIsChanged(buf))
Bram Moolenaar373154b2007-02-13 05:19:30 +00001917 r = FAIL;
1918 return r;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001919}
1920
1921/*
1922 * flush all buffers, except the ones that are readonly
1923 */
1924 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001925autowrite_all(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001926{
1927 buf_T *buf;
1928
1929 if (!(p_aw || p_awa) || !p_write)
1930 return;
Bram Moolenaar29323592016-07-24 22:04:11 +02001931 FOR_ALL_BUFFERS(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001932 if (bufIsChanged(buf) && !buf->b_p_ro)
1933 {
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001934#ifdef FEAT_AUTOCMD
1935 bufref_T bufref;
1936
1937 set_bufref(&bufref, buf);
1938#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001939 (void)buf_write_all(buf, FALSE);
1940#ifdef FEAT_AUTOCMD
1941 /* an autocommand may have deleted the buffer */
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001942 if (!bufref_valid(&bufref))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001943 buf = firstbuf;
1944#endif
1945 }
1946}
1947
1948/*
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001949 * Return TRUE if buffer was changed and cannot be abandoned.
1950 * For flags use the CCGD_ values.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001951 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001952 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001953check_changed(buf_T *buf, int flags)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001954{
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001955 int forceit = (flags & CCGD_FORCEIT);
1956#ifdef FEAT_AUTOCMD
1957 bufref_T bufref;
1958
1959 set_bufref(&bufref, buf);
1960#endif
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001961
Bram Moolenaar071d4272004-06-13 20:20:40 +00001962 if ( !forceit
1963 && bufIsChanged(buf)
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001964 && ((flags & CCGD_MULTWIN) || buf->b_nwindows <= 1)
1965 && (!(flags & CCGD_AW) || autowrite(buf, forceit) == FAIL))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001966 {
1967#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG)
1968 if ((p_confirm || cmdmod.confirm) && p_write)
1969 {
1970 buf_T *buf2;
1971 int count = 0;
1972
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001973 if (flags & CCGD_ALLBUF)
Bram Moolenaar29323592016-07-24 22:04:11 +02001974 FOR_ALL_BUFFERS(buf2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001975 if (bufIsChanged(buf2)
1976 && (buf2->b_ffname != NULL
1977# ifdef FEAT_BROWSE
1978 || cmdmod.browse
1979# endif
1980 ))
1981 ++count;
1982# ifdef FEAT_AUTOCMD
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001983 if (!bufref_valid(&bufref))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001984 /* Autocommand deleted buffer, oops! It's not changed now. */
1985 return FALSE;
1986# endif
1987 dialog_changed(buf, count > 1);
1988# ifdef FEAT_AUTOCMD
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001989 if (!bufref_valid(&bufref))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001990 /* Autocommand deleted buffer, oops! It's not changed now. */
1991 return FALSE;
1992# endif
1993 return bufIsChanged(buf);
1994 }
1995#endif
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001996 if (flags & CCGD_EXCMD)
Bram Moolenaarf5be7cd2017-08-17 16:55:13 +02001997 no_write_message();
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001998 else
Bram Moolenaarf5be7cd2017-08-17 16:55:13 +02001999 no_write_message_nobang();
Bram Moolenaar071d4272004-06-13 20:20:40 +00002000 return TRUE;
2001 }
2002 return FALSE;
2003}
2004
2005#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG) || defined(PROTO)
2006
2007#if defined(FEAT_BROWSE) || defined(PROTO)
2008/*
2009 * When wanting to write a file without a file name, ask the user for a name.
2010 */
2011 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002012browse_save_fname(buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002013{
2014 if (buf->b_fname == NULL)
2015 {
2016 char_u *fname;
2017
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00002018 fname = do_browse(BROWSE_SAVE, (char_u *)_("Save As"),
2019 NULL, NULL, NULL, NULL, buf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002020 if (fname != NULL)
2021 {
2022 if (setfname(buf, fname, NULL, TRUE) == OK)
2023 buf->b_flags |= BF_NOTEDITED;
2024 vim_free(fname);
2025 }
2026 }
2027}
2028#endif
2029
2030/*
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02002031 * Ask the user what to do when abandoning a changed buffer.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002032 * Must check 'write' option first!
2033 */
2034 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002035dialog_changed(
2036 buf_T *buf,
2037 int checkall) /* may abandon all changed buffers */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002038{
Bram Moolenaard9462e32011-04-11 21:35:11 +02002039 char_u buff[DIALOG_MSG_SIZE];
Bram Moolenaar071d4272004-06-13 20:20:40 +00002040 int ret;
2041 buf_T *buf2;
Bram Moolenaar8218f602012-04-25 17:32:18 +02002042 exarg_T ea;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002043
Bram Moolenaar3f9a1ff2017-08-21 22:06:02 +02002044 dialog_msg(buff, _("Save changes to \"%s\"?"), buf->b_fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002045 if (checkall)
2046 ret = vim_dialog_yesnoallcancel(VIM_QUESTION, NULL, buff, 1);
2047 else
2048 ret = vim_dialog_yesnocancel(VIM_QUESTION, NULL, buff, 1);
2049
Bram Moolenaar8218f602012-04-25 17:32:18 +02002050 /* Init ea pseudo-structure, this is needed for the check_overwrite()
2051 * function. */
2052 ea.append = ea.forceit = FALSE;
2053
Bram Moolenaar071d4272004-06-13 20:20:40 +00002054 if (ret == VIM_YES)
2055 {
2056#ifdef FEAT_BROWSE
2057 /* May get file name, when there is none */
2058 browse_save_fname(buf);
2059#endif
Bram Moolenaar8218f602012-04-25 17:32:18 +02002060 if (buf->b_fname != NULL && check_overwrite(&ea, buf,
2061 buf->b_fname, buf->b_ffname, FALSE) == OK)
2062 /* didn't hit Cancel */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002063 (void)buf_write_all(buf, FALSE);
2064 }
2065 else if (ret == VIM_NO)
2066 {
2067 unchanged(buf, TRUE);
2068 }
2069 else if (ret == VIM_ALL)
2070 {
2071 /*
2072 * Write all modified files that can be written.
2073 * Skip readonly buffers, these need to be confirmed
2074 * individually.
2075 */
Bram Moolenaar29323592016-07-24 22:04:11 +02002076 FOR_ALL_BUFFERS(buf2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002077 {
2078 if (bufIsChanged(buf2)
2079 && (buf2->b_ffname != NULL
2080#ifdef FEAT_BROWSE
2081 || cmdmod.browse
2082#endif
2083 )
2084 && !buf2->b_p_ro)
2085 {
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002086#ifdef FEAT_AUTOCMD
2087 bufref_T bufref;
2088
2089 set_bufref(&bufref, buf2);
2090#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002091#ifdef FEAT_BROWSE
2092 /* May get file name, when there is none */
2093 browse_save_fname(buf2);
2094#endif
Bram Moolenaar8218f602012-04-25 17:32:18 +02002095 if (buf2->b_fname != NULL && check_overwrite(&ea, buf2,
2096 buf2->b_fname, buf2->b_ffname, FALSE) == OK)
2097 /* didn't hit Cancel */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002098 (void)buf_write_all(buf2, FALSE);
2099#ifdef FEAT_AUTOCMD
2100 /* an autocommand may have deleted the buffer */
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002101 if (!bufref_valid(&bufref))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002102 buf2 = firstbuf;
2103#endif
2104 }
2105 }
2106 }
2107 else if (ret == VIM_DISCARDALL)
2108 {
2109 /*
2110 * mark all buffers as unchanged
2111 */
Bram Moolenaar29323592016-07-24 22:04:11 +02002112 FOR_ALL_BUFFERS(buf2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002113 unchanged(buf2, TRUE);
2114 }
2115}
2116#endif
2117
2118/*
2119 * Return TRUE if the buffer "buf" can be abandoned, either by making it
2120 * hidden, autowriting it or unloading it.
2121 */
2122 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002123can_abandon(buf_T *buf, int forceit)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002124{
Bram Moolenaareb44a682017-08-03 22:44:55 +02002125 return ( buf_hide(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002126 || !bufIsChanged(buf)
2127 || buf->b_nwindows > 1
2128 || autowrite(buf, forceit) == OK
2129 || forceit);
2130}
2131
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01002132static void add_bufnum(int *bufnrs, int *bufnump, int nr);
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002133
2134/*
2135 * Add a buffer number to "bufnrs", unless it's already there.
2136 */
2137 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002138add_bufnum(int *bufnrs, int *bufnump, int nr)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002139{
2140 int i;
2141
2142 for (i = 0; i < *bufnump; ++i)
2143 if (bufnrs[i] == nr)
2144 return;
2145 bufnrs[*bufnump] = nr;
2146 *bufnump = *bufnump + 1;
2147}
2148
Bram Moolenaar071d4272004-06-13 20:20:40 +00002149/*
2150 * Return TRUE if any buffer was changed and cannot be abandoned.
2151 * That changed buffer becomes the current buffer.
Bram Moolenaar027387f2016-01-02 22:25:52 +01002152 * When "unload" is true the current buffer is unloaded instead of making it
2153 * hidden. This is used for ":q!".
Bram Moolenaar071d4272004-06-13 20:20:40 +00002154 */
2155 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002156check_changed_any(
2157 int hidden, /* Only check hidden buffers */
2158 int unload)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002159{
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002160 int ret = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002161 buf_T *buf;
2162 int save;
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002163 int i;
2164 int bufnum = 0;
2165 int bufcount = 0;
2166 int *bufnrs;
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002167 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002168 win_T *wp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002169
Bram Moolenaar29323592016-07-24 22:04:11 +02002170 FOR_ALL_BUFFERS(buf)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002171 ++bufcount;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002172
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002173 if (bufcount == 0)
2174 return FALSE;
2175
2176 bufnrs = (int *)alloc(sizeof(int) * bufcount);
2177 if (bufnrs == NULL)
2178 return FALSE;
2179
2180 /* curbuf */
2181 bufnrs[bufnum++] = curbuf->b_fnum;
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002182 /* buf in curtab */
2183 FOR_ALL_WINDOWS(wp)
2184 if (wp->w_buffer != curbuf)
2185 add_bufnum(bufnrs, &bufnum, wp->w_buffer->b_fnum);
2186
2187 /* buf in other tab */
Bram Moolenaar29323592016-07-24 22:04:11 +02002188 FOR_ALL_TABPAGES(tp)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002189 if (tp != curtab)
2190 for (wp = tp->tp_firstwin; wp != NULL; wp = wp->w_next)
2191 add_bufnum(bufnrs, &bufnum, wp->w_buffer->b_fnum);
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002192 /* any other buf */
Bram Moolenaar29323592016-07-24 22:04:11 +02002193 FOR_ALL_BUFFERS(buf)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002194 add_bufnum(bufnrs, &bufnum, buf->b_fnum);
2195
2196 for (i = 0; i < bufnum; ++i)
2197 {
2198 buf = buflist_findnr(bufnrs[i]);
2199 if (buf == NULL)
2200 continue;
2201 if ((!hidden || buf->b_nwindows == 0) && bufIsChanged(buf))
2202 {
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002203 bufref_T bufref;
2204
2205 set_bufref(&bufref, buf);
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002206 /* Try auto-writing the buffer. If this fails but the buffer no
2207 * longer exists it's not changed, that's OK. */
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002208 if (check_changed(buf, (p_awa ? CCGD_AW : 0)
2209 | CCGD_MULTWIN
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002210 | CCGD_ALLBUF) && bufref_valid(&bufref))
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002211 break; /* didn't save - still changes */
2212 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002213 }
2214
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002215 if (i >= bufnum)
2216 goto theend;
2217
2218 ret = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002219 exiting = FALSE;
2220#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG)
2221 /*
2222 * When ":confirm" used, don't give an error message.
2223 */
2224 if (!(p_confirm || cmdmod.confirm))
2225#endif
2226 {
2227 /* There must be a wait_return for this message, do_buffer()
2228 * may cause a redraw. But wait_return() is a no-op when vgetc()
2229 * is busy (Quit used from window menu), then make sure we don't
2230 * cause a scroll up. */
Bram Moolenaar61660ea2006-04-07 21:40:07 +00002231 if (vgetc_busy > 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002232 {
2233 msg_row = cmdline_row;
2234 msg_col = 0;
2235 msg_didout = FALSE;
2236 }
Bram Moolenaareb44a682017-08-03 22:44:55 +02002237 if (
2238#ifdef FEAT_TERMINAL
2239 term_job_running(buf->b_term)
2240 ? EMSG2(_("E947: Job still running in buffer \"%s\""),
2241 buf->b_fname)
2242 :
2243#endif
2244 EMSG2(_("E162: No write since last change for buffer \"%s\""),
Bram Moolenaare1704ba2012-10-03 18:25:00 +02002245 buf_spname(buf) != NULL ? buf_spname(buf) : buf->b_fname))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002246 {
2247 save = no_wait_return;
2248 no_wait_return = FALSE;
2249 wait_return(FALSE);
2250 no_wait_return = save;
2251 }
2252 }
2253
Bram Moolenaar071d4272004-06-13 20:20:40 +00002254 /* Try to find a window that contains the buffer. */
2255 if (buf != curbuf)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002256 FOR_ALL_TAB_WINDOWS(tp, wp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002257 if (wp->w_buffer == buf)
2258 {
Bram Moolenaar4033c552017-09-16 20:54:51 +02002259#ifdef FEAT_AUTOCMD
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002260 bufref_T bufref;
2261
2262 set_bufref(&bufref, buf);
Bram Moolenaar4033c552017-09-16 20:54:51 +02002263#endif
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002264 goto_tabpage_win(tp, wp);
Bram Moolenaar4033c552017-09-16 20:54:51 +02002265#ifdef FEAT_AUTOCMD
Bram Moolenaar071d4272004-06-13 20:20:40 +00002266 /* Paranoia: did autocms wipe out the buffer with changes? */
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002267 if (!bufref_valid(&bufref))
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002268 {
2269 goto theend;
2270 }
Bram Moolenaar4033c552017-09-16 20:54:51 +02002271#endif
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002272 goto buf_found;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002273 }
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002274buf_found:
Bram Moolenaar071d4272004-06-13 20:20:40 +00002275
2276 /* Open the changed buffer in the current window. */
2277 if (buf != curbuf)
Bram Moolenaar027387f2016-01-02 22:25:52 +01002278 set_curbuf(buf, unload ? DOBUF_UNLOAD : DOBUF_GOTO);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002279
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002280theend:
2281 vim_free(bufnrs);
2282 return ret;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002283}
2284
2285/*
2286 * return FAIL if there is no file name, OK if there is one
2287 * give error message for FAIL
2288 */
2289 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002290check_fname(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002291{
2292 if (curbuf->b_ffname == NULL)
2293 {
2294 EMSG(_(e_noname));
2295 return FAIL;
2296 }
2297 return OK;
2298}
2299
2300/*
2301 * flush the contents of a buffer, unless it has no file name
2302 *
2303 * return FAIL for failure, OK otherwise
2304 */
2305 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002306buf_write_all(buf_T *buf, int forceit)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002307{
2308 int retval;
2309#ifdef FEAT_AUTOCMD
2310 buf_T *old_curbuf = curbuf;
2311#endif
2312
2313 retval = (buf_write(buf, buf->b_ffname, buf->b_fname,
2314 (linenr_T)1, buf->b_ml.ml_line_count, NULL,
2315 FALSE, forceit, TRUE, FALSE));
2316#ifdef FEAT_AUTOCMD
2317 if (curbuf != old_curbuf)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00002318 {
Bram Moolenaar8820b482017-03-16 17:23:31 +01002319 msg_source(HL_ATTR(HLF_W));
Bram Moolenaar071d4272004-06-13 20:20:40 +00002320 MSG(_("Warning: Entered other buffer unexpectedly (check autocommands)"));
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00002321 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002322#endif
2323 return retval;
2324}
2325
2326/*
2327 * Code to handle the argument list.
2328 */
2329
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01002330static char_u *do_one_arg(char_u *str);
2331static int do_arglist(char_u *str, int what, int after);
2332static void alist_check_arg_idx(void);
2333static int editing_arg_idx(win_T *win);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002334#ifdef FEAT_LISTCMDS
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01002335static int alist_add_list(int count, char_u **files, int after);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002336#endif
2337#define AL_SET 1
2338#define AL_ADD 2
2339#define AL_DEL 3
2340
Bram Moolenaar071d4272004-06-13 20:20:40 +00002341/*
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002342 * Isolate one argument, taking backticks.
2343 * Changes the argument in-place, puts a NUL after it. Backticks remain.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002344 * Return a pointer to the start of the next argument.
2345 */
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002346 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002347do_one_arg(char_u *str)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002348{
2349 char_u *p;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002350 int inbacktick;
2351
Bram Moolenaar071d4272004-06-13 20:20:40 +00002352 inbacktick = FALSE;
2353 for (p = str; *str; ++str)
2354 {
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002355 /* When the backslash is used for escaping the special meaning of a
2356 * character we need to keep it until wildcard expansion. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002357 if (rem_backslash(str))
2358 {
2359 *p++ = *str++;
2360 *p++ = *str;
2361 }
2362 else
2363 {
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002364 /* An item ends at a space not in backticks */
2365 if (!inbacktick && vim_isspace(*str))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002366 break;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002367 if (*str == '`')
Bram Moolenaar071d4272004-06-13 20:20:40 +00002368 inbacktick ^= TRUE;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002369 *p++ = *str;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002370 }
2371 }
2372 str = skipwhite(str);
2373 *p = NUL;
2374
2375 return str;
2376}
2377
Bram Moolenaar86b68352004-12-27 21:59:20 +00002378/*
2379 * Separate the arguments in "str" and return a list of pointers in the
2380 * growarray "gap".
2381 */
Bram Moolenaar398ee732017-08-03 14:29:14 +02002382 static int
2383get_arglist(garray_T *gap, char_u *str, int escaped)
Bram Moolenaar86b68352004-12-27 21:59:20 +00002384{
2385 ga_init2(gap, (int)sizeof(char_u *), 20);
2386 while (*str != NUL)
2387 {
2388 if (ga_grow(gap, 1) == FAIL)
2389 {
2390 ga_clear(gap);
2391 return FAIL;
2392 }
2393 ((char_u **)gap->ga_data)[gap->ga_len++] = str;
2394
Bram Moolenaar398ee732017-08-03 14:29:14 +02002395 /* If str is escaped, don't handle backslashes or spaces */
2396 if (!escaped)
2397 return OK;
2398
Bram Moolenaar86b68352004-12-27 21:59:20 +00002399 /* Isolate one argument, change it in-place, put a NUL after it. */
2400 str = do_one_arg(str);
2401 }
2402 return OK;
2403}
2404
Bram Moolenaar7df351e2006-01-23 22:30:28 +00002405#if defined(FEAT_QUICKFIX) || defined(FEAT_SYN_HL) || defined(PROTO)
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002406/*
2407 * Parse a list of arguments (file names), expand them and return in
Bram Moolenaar8f5c6f02012-06-29 12:57:06 +02002408 * "fnames[fcountp]". When "wig" is TRUE, removes files matching 'wildignore'.
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002409 * Return FAIL or OK.
2410 */
2411 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002412get_arglist_exp(
2413 char_u *str,
2414 int *fcountp,
2415 char_u ***fnamesp,
2416 int wig)
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002417{
2418 garray_T ga;
2419 int i;
2420
Bram Moolenaar398ee732017-08-03 14:29:14 +02002421 if (get_arglist(&ga, str, TRUE) == FAIL)
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002422 return FAIL;
Bram Moolenaar8f5c6f02012-06-29 12:57:06 +02002423 if (wig == TRUE)
2424 i = expand_wildcards(ga.ga_len, (char_u **)ga.ga_data,
2425 fcountp, fnamesp, EW_FILE|EW_NOTFOUND);
2426 else
2427 i = gen_expand_wildcards(ga.ga_len, (char_u **)ga.ga_data,
2428 fcountp, fnamesp, EW_FILE|EW_NOTFOUND);
2429
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002430 ga_clear(&ga);
2431 return i;
2432}
2433#endif
2434
Bram Moolenaar071d4272004-06-13 20:20:40 +00002435#if defined(FEAT_GUI) || defined(FEAT_CLIENTSERVER) || defined(PROTO)
2436/*
2437 * Redefine the argument list.
2438 */
2439 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002440set_arglist(char_u *str)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002441{
2442 do_arglist(str, AL_SET, 0);
2443}
2444#endif
2445
2446/*
2447 * "what" == AL_SET: Redefine the argument list to 'str'.
2448 * "what" == AL_ADD: add files in 'str' to the argument list after "after".
2449 * "what" == AL_DEL: remove files in 'str' from the argument list.
2450 *
2451 * Return FAIL for failure, OK otherwise.
2452 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002453 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002454do_arglist(
2455 char_u *str,
Bram Moolenaarf1d25012016-03-03 12:22:53 +01002456 int what,
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002457 int after UNUSED) /* 0 means before first one */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002458{
2459 garray_T new_ga;
2460 int exp_count;
2461 char_u **exp_files;
2462 int i;
2463#ifdef FEAT_LISTCMDS
2464 char_u *p;
2465 int match;
2466#endif
Bram Moolenaar398ee732017-08-03 14:29:14 +02002467 int arg_escaped = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002468
2469 /*
Bram Moolenaar2faa29f2016-01-23 23:02:34 +01002470 * Set default argument for ":argadd" command.
2471 */
2472 if (what == AL_ADD && *str == NUL)
2473 {
2474 if (curbuf->b_ffname == NULL)
2475 return FAIL;
2476 str = curbuf->b_fname;
Bram Moolenaar398ee732017-08-03 14:29:14 +02002477 arg_escaped = FALSE;
Bram Moolenaar2faa29f2016-01-23 23:02:34 +01002478 }
2479
2480 /*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002481 * Collect all file name arguments in "new_ga".
2482 */
Bram Moolenaar398ee732017-08-03 14:29:14 +02002483 if (get_arglist(&new_ga, str, arg_escaped) == FAIL)
Bram Moolenaar86b68352004-12-27 21:59:20 +00002484 return FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002485
2486#ifdef FEAT_LISTCMDS
2487 if (what == AL_DEL)
2488 {
2489 regmatch_T regmatch;
2490 int didone;
2491
2492 /*
2493 * Delete the items: use each item as a regexp and find a match in the
2494 * argument list.
2495 */
Bram Moolenaar71afbfe2013-03-19 16:49:16 +01002496 regmatch.rm_ic = p_fic; /* ignore case when 'fileignorecase' is set */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002497 for (i = 0; i < new_ga.ga_len && !got_int; ++i)
2498 {
2499 p = ((char_u **)new_ga.ga_data)[i];
2500 p = file_pat_to_reg_pat(p, NULL, NULL, FALSE);
2501 if (p == NULL)
2502 break;
2503 regmatch.regprog = vim_regcomp(p, p_magic ? RE_MAGIC : 0);
2504 if (regmatch.regprog == NULL)
2505 {
2506 vim_free(p);
2507 break;
2508 }
2509
2510 didone = FALSE;
2511 for (match = 0; match < ARGCOUNT; ++match)
2512 if (vim_regexec(&regmatch, alist_name(&ARGLIST[match]),
2513 (colnr_T)0))
2514 {
2515 didone = TRUE;
2516 vim_free(ARGLIST[match].ae_fname);
2517 mch_memmove(ARGLIST + match, ARGLIST + match + 1,
2518 (ARGCOUNT - match - 1) * sizeof(aentry_T));
2519 --ALIST(curwin)->al_ga.ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002520 if (curwin->w_arg_idx > match)
2521 --curwin->w_arg_idx;
2522 --match;
2523 }
2524
Bram Moolenaar473de612013-06-08 18:19:48 +02002525 vim_regfree(regmatch.regprog);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002526 vim_free(p);
2527 if (!didone)
2528 EMSG2(_(e_nomatch2), ((char_u **)new_ga.ga_data)[i]);
2529 }
2530 ga_clear(&new_ga);
2531 }
2532 else
2533#endif
2534 {
2535 i = expand_wildcards(new_ga.ga_len, (char_u **)new_ga.ga_data,
2536 &exp_count, &exp_files, EW_DIR|EW_FILE|EW_ADDSLASH|EW_NOTFOUND);
2537 ga_clear(&new_ga);
Bram Moolenaar2db5c3b2016-01-16 22:49:34 +01002538 if (i == FAIL || exp_count == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002539 {
2540 EMSG(_(e_nomatch));
2541 return FAIL;
2542 }
2543
2544#ifdef FEAT_LISTCMDS
2545 if (what == AL_ADD)
2546 {
2547 (void)alist_add_list(exp_count, exp_files, after);
2548 vim_free(exp_files);
2549 }
2550 else /* what == AL_SET */
2551#endif
Bram Moolenaar86b68352004-12-27 21:59:20 +00002552 alist_set(ALIST(curwin), exp_count, exp_files, FALSE, NULL, 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002553 }
2554
2555 alist_check_arg_idx();
2556
2557 return OK;
2558}
2559
2560/*
2561 * Check the validity of the arg_idx for each other window.
2562 */
2563 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002564alist_check_arg_idx(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002565{
Bram Moolenaar071d4272004-06-13 20:20:40 +00002566 win_T *win;
Bram Moolenaarf740b292006-02-16 22:11:02 +00002567 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002568
Bram Moolenaarf740b292006-02-16 22:11:02 +00002569 FOR_ALL_TAB_WINDOWS(tp, win)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002570 if (win->w_alist == curwin->w_alist)
2571 check_arg_idx(win);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002572}
2573
2574/*
Bram Moolenaarb8ff1fb2012-02-04 21:59:01 +01002575 * Return TRUE if window "win" is editing the file at the current argument
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002576 * index.
2577 */
2578 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002579editing_arg_idx(win_T *win)
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002580{
2581 return !(win->w_arg_idx >= WARGCOUNT(win)
2582 || (win->w_buffer->b_fnum
2583 != WARGLIST(win)[win->w_arg_idx].ae_fnum
2584 && (win->w_buffer->b_ffname == NULL
2585 || !(fullpathcmp(
2586 alist_name(&WARGLIST(win)[win->w_arg_idx]),
2587 win->w_buffer->b_ffname, TRUE) & FPC_SAME))));
2588}
2589
2590/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002591 * Check if window "win" is editing the w_arg_idx file in its argument list.
2592 */
2593 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002594check_arg_idx(win_T *win)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002595{
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002596 if (WARGCOUNT(win) > 1 && !editing_arg_idx(win))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002597 {
2598 /* We are not editing the current entry in the argument list.
2599 * Set "arg_had_last" if we are editing the last one. */
2600 win->w_arg_idx_invalid = TRUE;
2601 if (win->w_arg_idx != WARGCOUNT(win) - 1
2602 && arg_had_last == FALSE
Bram Moolenaar071d4272004-06-13 20:20:40 +00002603 && ALIST(win) == &global_alist
Bram Moolenaar071d4272004-06-13 20:20:40 +00002604 && GARGCOUNT > 0
2605 && win->w_arg_idx < GARGCOUNT
2606 && (win->w_buffer->b_fnum == GARGLIST[GARGCOUNT - 1].ae_fnum
2607 || (win->w_buffer->b_ffname != NULL
2608 && (fullpathcmp(alist_name(&GARGLIST[GARGCOUNT - 1]),
2609 win->w_buffer->b_ffname, TRUE) & FPC_SAME))))
2610 arg_had_last = TRUE;
2611 }
2612 else
2613 {
2614 /* We are editing the current entry in the argument list.
2615 * Set "arg_had_last" if it's also the last one */
2616 win->w_arg_idx_invalid = FALSE;
2617 if (win->w_arg_idx == WARGCOUNT(win) - 1
Bram Moolenaar4033c552017-09-16 20:54:51 +02002618 && win->w_alist == &global_alist)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002619 arg_had_last = TRUE;
2620 }
2621}
2622
2623/*
2624 * ":args", ":argslocal" and ":argsglobal".
2625 */
2626 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002627ex_args(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002628{
2629 int i;
2630
2631 if (eap->cmdidx != CMD_args)
2632 {
Bram Moolenaar4033c552017-09-16 20:54:51 +02002633#if defined(FEAT_LISTCMDS)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002634 alist_unlink(ALIST(curwin));
2635 if (eap->cmdidx == CMD_argglobal)
2636 ALIST(curwin) = &global_alist;
2637 else /* eap->cmdidx == CMD_arglocal */
2638 alist_new();
2639#else
2640 ex_ni(eap);
2641 return;
2642#endif
2643 }
2644
2645 if (!ends_excmd(*eap->arg))
2646 {
2647 /*
2648 * ":args file ..": define new argument list, handle like ":next"
2649 * Also for ":argslocal file .." and ":argsglobal file ..".
2650 */
2651 ex_next(eap);
2652 }
2653 else
Bram Moolenaar4033c552017-09-16 20:54:51 +02002654#if defined(FEAT_LISTCMDS)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002655 if (eap->cmdidx == CMD_args)
2656#endif
2657 {
2658 /*
2659 * ":args": list arguments.
2660 */
2661 if (ARGCOUNT > 0)
2662 {
2663 /* Overwrite the command, for a short list there is no scrolling
2664 * required and no wait_return(). */
2665 gotocmdline(TRUE);
2666 for (i = 0; i < ARGCOUNT; ++i)
2667 {
2668 if (i == curwin->w_arg_idx)
2669 msg_putchar('[');
2670 msg_outtrans(alist_name(&ARGLIST[i]));
2671 if (i == curwin->w_arg_idx)
2672 msg_putchar(']');
2673 msg_putchar(' ');
2674 }
2675 }
2676 }
Bram Moolenaar4033c552017-09-16 20:54:51 +02002677#if defined(FEAT_LISTCMDS)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002678 else if (eap->cmdidx == CMD_arglocal)
2679 {
2680 garray_T *gap = &curwin->w_alist->al_ga;
2681
2682 /*
2683 * ":argslocal": make a local copy of the global argument list.
2684 */
2685 if (ga_grow(gap, GARGCOUNT) == OK)
2686 for (i = 0; i < GARGCOUNT; ++i)
2687 if (GARGLIST[i].ae_fname != NULL)
2688 {
2689 AARGLIST(curwin->w_alist)[gap->ga_len].ae_fname =
2690 vim_strsave(GARGLIST[i].ae_fname);
2691 AARGLIST(curwin->w_alist)[gap->ga_len].ae_fnum =
2692 GARGLIST[i].ae_fnum;
2693 ++gap->ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002694 }
2695 }
2696#endif
2697}
2698
2699/*
2700 * ":previous", ":sprevious", ":Next" and ":sNext".
2701 */
2702 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002703ex_previous(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002704{
2705 /* If past the last one already, go to the last one. */
2706 if (curwin->w_arg_idx - (int)eap->line2 >= ARGCOUNT)
2707 do_argfile(eap, ARGCOUNT - 1);
2708 else
2709 do_argfile(eap, curwin->w_arg_idx - (int)eap->line2);
2710}
2711
2712/*
2713 * ":rewind", ":first", ":sfirst" and ":srewind".
2714 */
2715 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002716ex_rewind(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002717{
2718 do_argfile(eap, 0);
2719}
2720
2721/*
2722 * ":last" and ":slast".
2723 */
2724 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002725ex_last(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002726{
2727 do_argfile(eap, ARGCOUNT - 1);
2728}
2729
2730/*
2731 * ":argument" and ":sargument".
2732 */
2733 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002734ex_argument(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002735{
2736 int i;
2737
2738 if (eap->addr_count > 0)
2739 i = eap->line2 - 1;
2740 else
2741 i = curwin->w_arg_idx;
2742 do_argfile(eap, i);
2743}
2744
2745/*
2746 * Edit file "argn" of the argument lists.
2747 */
2748 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002749do_argfile(exarg_T *eap, int argn)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002750{
2751 int other;
2752 char_u *p;
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002753 int old_arg_idx = curwin->w_arg_idx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002754
2755 if (argn < 0 || argn >= ARGCOUNT)
2756 {
2757 if (ARGCOUNT <= 1)
2758 EMSG(_("E163: There is only one file to edit"));
2759 else if (argn < 0)
2760 EMSG(_("E164: Cannot go before first file"));
2761 else
2762 EMSG(_("E165: Cannot go beyond last file"));
2763 }
2764 else
2765 {
2766 setpcmark();
2767#ifdef FEAT_GUI
2768 need_mouse_correct = TRUE;
2769#endif
2770
Bram Moolenaardf1bdc92006-02-23 21:32:16 +00002771 /* split window or create new tab page first */
2772 if (*eap->cmd == 's' || cmdmod.tab != 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002773 {
2774 if (win_split(0, 0) == FAIL)
2775 return;
Bram Moolenaar3368ea22010-09-21 16:56:35 +02002776 RESET_BINDING(curwin);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002777 }
2778 else
Bram Moolenaar071d4272004-06-13 20:20:40 +00002779 {
2780 /*
2781 * if 'hidden' set, only check for changed file when re-editing
2782 * the same buffer
2783 */
2784 other = TRUE;
Bram Moolenaareb44a682017-08-03 22:44:55 +02002785 if (buf_hide(curbuf))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002786 {
2787 p = fix_fname(alist_name(&ARGLIST[argn]));
2788 other = otherfile(p);
2789 vim_free(p);
2790 }
Bram Moolenaareb44a682017-08-03 22:44:55 +02002791 if ((!buf_hide(curbuf) || !other)
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002792 && check_changed(curbuf, CCGD_AW
2793 | (other ? 0 : CCGD_MULTWIN)
2794 | (eap->forceit ? CCGD_FORCEIT : 0)
2795 | CCGD_EXCMD))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002796 return;
2797 }
2798
2799 curwin->w_arg_idx = argn;
Bram Moolenaar4033c552017-09-16 20:54:51 +02002800 if (argn == ARGCOUNT - 1 && curwin->w_alist == &global_alist)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002801 arg_had_last = TRUE;
2802
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002803 /* Edit the file; always use the last known line number.
2804 * When it fails (e.g. Abort for already edited file) restore the
2805 * argument index. */
2806 if (do_ecmd(0, alist_name(&ARGLIST[curwin->w_arg_idx]), NULL,
Bram Moolenaar071d4272004-06-13 20:20:40 +00002807 eap, ECMD_LAST,
Bram Moolenaareb44a682017-08-03 22:44:55 +02002808 (buf_hide(curwin->w_buffer) ? ECMD_HIDE : 0)
Bram Moolenaar701f7af2008-11-15 13:12:07 +00002809 + (eap->forceit ? ECMD_FORCEIT : 0), curwin) == FAIL)
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002810 curwin->w_arg_idx = old_arg_idx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002811 /* like Vi: set the mark where the cursor is in the file. */
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002812 else if (eap->cmdidx != CMD_argdo)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002813 setmark('\'');
2814 }
2815}
2816
2817/*
2818 * ":next", and commands that behave like it.
2819 */
2820 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002821ex_next(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002822{
2823 int i;
2824
2825 /*
2826 * check for changed buffer now, if this fails the argument list is not
2827 * redefined.
2828 */
Bram Moolenaareb44a682017-08-03 22:44:55 +02002829 if ( buf_hide(curbuf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002830 || eap->cmdidx == CMD_snext
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002831 || !check_changed(curbuf, CCGD_AW
2832 | (eap->forceit ? CCGD_FORCEIT : 0)
2833 | CCGD_EXCMD))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002834 {
2835 if (*eap->arg != NUL) /* redefine file list */
2836 {
2837 if (do_arglist(eap->arg, AL_SET, 0) == FAIL)
2838 return;
2839 i = 0;
2840 }
2841 else
2842 i = curwin->w_arg_idx + (int)eap->line2;
2843 do_argfile(eap, i);
2844 }
2845}
2846
Bram Moolenaar0106e3d2016-02-23 18:55:43 +01002847#if defined(FEAT_LISTCMDS) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002848/*
2849 * ":argedit"
2850 */
2851 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002852ex_argedit(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002853{
Bram Moolenaar90305c62017-07-16 15:31:17 +02002854 int i = eap->addr_count ? (int)eap->line2 : curwin->w_arg_idx + 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002855
Bram Moolenaar90305c62017-07-16 15:31:17 +02002856 if (do_arglist(eap->arg, AL_ADD, i) == FAIL)
2857 return;
2858#ifdef FEAT_TITLE
2859 maketitle();
2860#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002861
Bram Moolenaar90305c62017-07-16 15:31:17 +02002862 if (curwin->w_arg_idx == 0 && (curbuf->b_ml.ml_flags & ML_EMPTY)
2863 && curbuf->b_ffname == NULL)
2864 i = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002865 /* Edit the argument. */
Bram Moolenaar90305c62017-07-16 15:31:17 +02002866 if (i < ARGCOUNT)
2867 do_argfile(eap, i);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002868}
2869
2870/*
2871 * ":argadd"
2872 */
2873 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002874ex_argadd(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002875{
2876 do_arglist(eap->arg, AL_ADD,
2877 eap->addr_count > 0 ? (int)eap->line2 : curwin->w_arg_idx + 1);
2878#ifdef FEAT_TITLE
2879 maketitle();
2880#endif
2881}
2882
2883/*
2884 * ":argdelete"
2885 */
2886 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002887ex_argdelete(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002888{
2889 int i;
2890 int n;
2891
2892 if (eap->addr_count > 0)
2893 {
2894 /* ":1,4argdel": Delete all arguments in the range. */
2895 if (eap->line2 > ARGCOUNT)
2896 eap->line2 = ARGCOUNT;
2897 n = eap->line2 - eap->line1 + 1;
Bram Moolenaar69a92fb2017-03-09 15:58:30 +01002898 if (*eap->arg != NUL)
2899 /* Can't have both a range and an argument. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002900 EMSG(_(e_invarg));
Bram Moolenaar69a92fb2017-03-09 15:58:30 +01002901 else if (n <= 0)
2902 {
2903 /* Don't give an error for ":%argdel" if the list is empty. */
2904 if (eap->line1 != 1 || eap->line2 != 0)
2905 EMSG(_(e_invrange));
2906 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002907 else
2908 {
2909 for (i = eap->line1; i <= eap->line2; ++i)
2910 vim_free(ARGLIST[i - 1].ae_fname);
2911 mch_memmove(ARGLIST + eap->line1 - 1, ARGLIST + eap->line2,
2912 (size_t)((ARGCOUNT - eap->line2) * sizeof(aentry_T)));
2913 ALIST(curwin)->al_ga.ga_len -= n;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002914 if (curwin->w_arg_idx >= eap->line2)
2915 curwin->w_arg_idx -= n;
2916 else if (curwin->w_arg_idx > eap->line1)
2917 curwin->w_arg_idx = eap->line1;
Bram Moolenaar72defda2016-01-17 18:04:33 +01002918 if (ARGCOUNT == 0)
2919 curwin->w_arg_idx = 0;
2920 else if (curwin->w_arg_idx >= ARGCOUNT)
2921 curwin->w_arg_idx = ARGCOUNT - 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002922 }
2923 }
2924 else if (*eap->arg == NUL)
2925 EMSG(_(e_argreq));
2926 else
2927 do_arglist(eap->arg, AL_DEL, 0);
2928#ifdef FEAT_TITLE
2929 maketitle();
2930#endif
2931}
2932
2933/*
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002934 * ":argdo", ":windo", ":bufdo", ":tabdo", ":cdo", ":ldo", ":cfdo" and ":lfdo"
Bram Moolenaar071d4272004-06-13 20:20:40 +00002935 */
2936 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002937ex_listdo(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002938{
2939 int i;
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002940 win_T *wp;
2941 tabpage_T *tp;
Bram Moolenaare25bb902015-02-27 20:33:37 +01002942 buf_T *buf = curbuf;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002943 int next_fnum = 0;
2944#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
2945 char_u *save_ei = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002946#endif
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002947 char_u *p_shm_save;
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002948#ifdef FEAT_QUICKFIX
Bram Moolenaared84b762015-09-09 22:35:29 +02002949 int qf_size = 0;
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002950 int qf_idx;
2951#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002952
Bram Moolenaar0106e3d2016-02-23 18:55:43 +01002953#ifndef FEAT_QUICKFIX
2954 if (eap->cmdidx == CMD_cdo || eap->cmdidx == CMD_ldo ||
2955 eap->cmdidx == CMD_cfdo || eap->cmdidx == CMD_lfdo)
2956 {
2957 ex_ni(eap);
2958 return;
2959 }
2960#endif
2961
Bram Moolenaar071d4272004-06-13 20:20:40 +00002962#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002963 if (eap->cmdidx != CMD_windo && eap->cmdidx != CMD_tabdo)
Bram Moolenaardcaf10e2005-01-21 11:55:25 +00002964 /* Don't do syntax HL autocommands. Skipping the syntax file is a
2965 * great speed improvement. */
2966 save_ei = au_event_disable(",Syntax");
Bram Moolenaar071d4272004-06-13 20:20:40 +00002967#endif
Bram Moolenaar6b1ee342014-08-06 18:17:11 +02002968#ifdef FEAT_CLIPBOARD
2969 start_global_changes();
2970#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002971
2972 if (eap->cmdidx == CMD_windo
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002973 || eap->cmdidx == CMD_tabdo
Bram Moolenaareb44a682017-08-03 22:44:55 +02002974 || buf_hide(curbuf)
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002975 || !check_changed(curbuf, CCGD_AW
2976 | (eap->forceit ? CCGD_FORCEIT : 0)
2977 | CCGD_EXCMD))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002978 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00002979 i = 0;
Bram Moolenaara162bc52015-01-07 16:54:21 +01002980 /* start at the eap->line1 argument/window/buffer */
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002981 wp = firstwin;
2982 tp = first_tabpage;
Bram Moolenaara162bc52015-01-07 16:54:21 +01002983 switch (eap->cmdidx)
2984 {
Bram Moolenaara162bc52015-01-07 16:54:21 +01002985 case CMD_windo:
2986 for ( ; wp != NULL && i + 1 < eap->line1; wp = wp->w_next)
2987 i++;
2988 break;
2989 case CMD_tabdo:
2990 for( ; tp != NULL && i + 1 < eap->line1; tp = tp->tp_next)
2991 i++;
2992 break;
Bram Moolenaara162bc52015-01-07 16:54:21 +01002993 case CMD_argdo:
2994 i = eap->line1 - 1;
2995 break;
Bram Moolenaara162bc52015-01-07 16:54:21 +01002996 default:
2997 break;
2998 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002999 /* set pcmark now */
3000 if (eap->cmdidx == CMD_bufdo)
Bram Moolenaaraa23b372015-09-08 18:46:31 +02003001 {
Bram Moolenaare25bb902015-02-27 20:33:37 +01003002 /* Advance to the first listed buffer after "eap->line1". */
Bram Moolenaaraa23b372015-09-08 18:46:31 +02003003 for (buf = firstbuf; buf != NULL && (buf->b_fnum < eap->line1
Bram Moolenaare25bb902015-02-27 20:33:37 +01003004 || !buf->b_p_bl); buf = buf->b_next)
3005 if (buf->b_fnum > eap->line2)
3006 {
3007 buf = NULL;
3008 break;
3009 }
Bram Moolenaaraa23b372015-09-08 18:46:31 +02003010 if (buf != NULL)
Bram Moolenaare25bb902015-02-27 20:33:37 +01003011 goto_buffer(eap, DOBUF_FIRST, FORWARD, buf->b_fnum);
Bram Moolenaaraa23b372015-09-08 18:46:31 +02003012 }
3013#ifdef FEAT_QUICKFIX
3014 else if (eap->cmdidx == CMD_cdo || eap->cmdidx == CMD_ldo
3015 || eap->cmdidx == CMD_cfdo || eap->cmdidx == CMD_lfdo)
3016 {
3017 qf_size = qf_get_size(eap);
3018 if (qf_size <= 0 || eap->line1 > qf_size)
3019 buf = NULL;
3020 else
3021 {
3022 ex_cc(eap);
3023
3024 buf = curbuf;
3025 i = eap->line1 - 1;
3026 if (eap->addr_count <= 0)
3027 /* default is all the quickfix/location list entries */
3028 eap->line2 = qf_size;
3029 }
3030 }
3031#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003032 else
3033 setpcmark();
3034 listcmd_busy = TRUE; /* avoids setting pcmark below */
3035
Bram Moolenaare25bb902015-02-27 20:33:37 +01003036 while (!got_int && buf != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003037 {
3038 if (eap->cmdidx == CMD_argdo)
3039 {
3040 /* go to argument "i" */
3041 if (i == ARGCOUNT)
3042 break;
3043 /* Don't call do_argfile() when already there, it will try
3044 * reloading the file. */
Bram Moolenaard4755bb2004-09-02 19:12:26 +00003045 if (curwin->w_arg_idx != i || !editing_arg_idx(curwin))
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003046 {
3047 /* Clear 'shm' to avoid that the file message overwrites
3048 * any output from the command. */
3049 p_shm_save = vim_strsave(p_shm);
3050 set_option_value((char_u *)"shm", 0L, (char_u *)"", 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003051 do_argfile(eap, i);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003052 set_option_value((char_u *)"shm", 0L, p_shm_save, 0);
3053 vim_free(p_shm_save);
3054 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003055 if (curwin->w_arg_idx != i)
3056 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003057 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003058 else if (eap->cmdidx == CMD_windo)
3059 {
Bram Moolenaar32466aa2006-02-24 23:53:04 +00003060 /* go to window "wp" */
3061 if (!win_valid(wp))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003062 break;
Bram Moolenaar32466aa2006-02-24 23:53:04 +00003063 win_goto(wp);
Bram Moolenaar41423242007-05-03 20:11:13 +00003064 if (curwin != wp)
3065 break; /* something must be wrong */
Bram Moolenaar32466aa2006-02-24 23:53:04 +00003066 wp = curwin->w_next;
3067 }
3068 else if (eap->cmdidx == CMD_tabdo)
3069 {
3070 /* go to window "tp" */
3071 if (!valid_tabpage(tp))
3072 break;
Bram Moolenaar49e649f2013-05-06 04:50:35 +02003073 goto_tabpage_tp(tp, TRUE, TRUE);
Bram Moolenaar32466aa2006-02-24 23:53:04 +00003074 tp = tp->tp_next;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003075 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003076 else if (eap->cmdidx == CMD_bufdo)
3077 {
3078 /* Remember the number of the next listed buffer, in case
3079 * ":bwipe" is used or autocommands do something strange. */
3080 next_fnum = -1;
3081 for (buf = curbuf->b_next; buf != NULL; buf = buf->b_next)
3082 if (buf->b_p_bl)
3083 {
3084 next_fnum = buf->b_fnum;
3085 break;
3086 }
3087 }
3088
Bram Moolenaara162bc52015-01-07 16:54:21 +01003089 ++i;
3090
Bram Moolenaar071d4272004-06-13 20:20:40 +00003091 /* execute the command */
3092 do_cmdline(eap->arg, eap->getline, eap->cookie,
3093 DOCMD_VERBOSE + DOCMD_NOWAIT);
3094
3095 if (eap->cmdidx == CMD_bufdo)
3096 {
3097 /* Done? */
Bram Moolenaara162bc52015-01-07 16:54:21 +01003098 if (next_fnum < 0 || next_fnum > eap->line2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003099 break;
3100 /* Check if the buffer still exists. */
Bram Moolenaar29323592016-07-24 22:04:11 +02003101 FOR_ALL_BUFFERS(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003102 if (buf->b_fnum == next_fnum)
3103 break;
3104 if (buf == NULL)
3105 break;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003106
3107 /* Go to the next buffer. Clear 'shm' to avoid that the file
3108 * message overwrites any output from the command. */
3109 p_shm_save = vim_strsave(p_shm);
3110 set_option_value((char_u *)"shm", 0L, (char_u *)"", 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003111 goto_buffer(eap, DOBUF_FIRST, FORWARD, next_fnum);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003112 set_option_value((char_u *)"shm", 0L, p_shm_save, 0);
3113 vim_free(p_shm_save);
3114
Bram Moolenaaraa23b372015-09-08 18:46:31 +02003115 /* If autocommands took us elsewhere, quit here. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003116 if (curbuf->b_fnum != next_fnum)
3117 break;
3118 }
3119
Bram Moolenaaraa23b372015-09-08 18:46:31 +02003120#ifdef FEAT_QUICKFIX
3121 if (eap->cmdidx == CMD_cdo || eap->cmdidx == CMD_ldo
3122 || eap->cmdidx == CMD_cfdo || eap->cmdidx == CMD_lfdo)
3123 {
3124 if (i >= qf_size || i >= eap->line2)
3125 break;
3126
3127 qf_idx = qf_get_cur_idx(eap);
3128
3129 ex_cnext(eap);
3130
3131 /* If jumping to the next quickfix entry fails, quit here */
3132 if (qf_get_cur_idx(eap) == qf_idx)
3133 break;
3134 }
3135#endif
3136
Bram Moolenaar071d4272004-06-13 20:20:40 +00003137 if (eap->cmdidx == CMD_windo)
3138 {
3139 validate_cursor(); /* cursor may have moved */
3140#ifdef FEAT_SCROLLBIND
3141 /* required when 'scrollbind' has been set */
3142 if (curwin->w_p_scb)
3143 do_check_scrollbind(TRUE);
3144#endif
3145 }
Bram Moolenaara162bc52015-01-07 16:54:21 +01003146
Bram Moolenaara162bc52015-01-07 16:54:21 +01003147 if (eap->cmdidx == CMD_windo || eap->cmdidx == CMD_tabdo)
3148 if (i+1 > eap->line2)
3149 break;
Bram Moolenaara162bc52015-01-07 16:54:21 +01003150 if (eap->cmdidx == CMD_argdo && i >= eap->line2)
3151 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003152 }
3153 listcmd_busy = FALSE;
3154 }
3155
3156#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
Bram Moolenaar6ac54292005-02-02 23:07:25 +00003157 if (save_ei != NULL)
3158 {
3159 au_event_restore(save_ei);
3160 apply_autocmds(EVENT_SYNTAX, curbuf->b_p_syn,
3161 curbuf->b_fname, TRUE, curbuf);
3162 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003163#endif
Bram Moolenaar6b1ee342014-08-06 18:17:11 +02003164#ifdef FEAT_CLIPBOARD
3165 end_global_changes();
3166#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003167}
3168
3169/*
3170 * Add files[count] to the arglist of the current window after arg "after".
3171 * The file names in files[count] must have been allocated and are taken over.
3172 * Files[] itself is not taken over.
3173 * Returns index of first added argument. Returns -1 when failed (out of mem).
3174 */
3175 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003176alist_add_list(
3177 int count,
3178 char_u **files,
3179 int after) /* where to add: 0 = before first one */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003180{
3181 int i;
Bram Moolenaara24f0a52016-01-17 19:39:00 +01003182 int old_argcount = ARGCOUNT;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003183
3184 if (ga_grow(&ALIST(curwin)->al_ga, count) == OK)
3185 {
3186 if (after < 0)
3187 after = 0;
3188 if (after > ARGCOUNT)
3189 after = ARGCOUNT;
3190 if (after < ARGCOUNT)
3191 mch_memmove(&(ARGLIST[after + count]), &(ARGLIST[after]),
3192 (ARGCOUNT - after) * sizeof(aentry_T));
3193 for (i = 0; i < count; ++i)
3194 {
3195 ARGLIST[after + i].ae_fname = files[i];
3196 ARGLIST[after + i].ae_fnum = buflist_add(files[i], BLN_LISTED);
3197 }
3198 ALIST(curwin)->al_ga.ga_len += count;
Bram Moolenaara24f0a52016-01-17 19:39:00 +01003199 if (old_argcount > 0 && curwin->w_arg_idx >= after)
3200 curwin->w_arg_idx += count;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003201 return after;
3202 }
3203
3204 for (i = 0; i < count; ++i)
3205 vim_free(files[i]);
3206 return -1;
3207}
3208
3209#endif /* FEAT_LISTCMDS */
3210
3211#ifdef FEAT_EVAL
3212/*
3213 * ":compiler[!] {name}"
3214 */
3215 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003216ex_compiler(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003217{
3218 char_u *buf;
3219 char_u *old_cur_comp = NULL;
3220 char_u *p;
3221
3222 if (*eap->arg == NUL)
3223 {
3224 /* List all compiler scripts. */
3225 do_cmdline_cmd((char_u *)"echo globpath(&rtp, 'compiler/*.vim')");
3226 /* ) keep the indenter happy... */
3227 }
3228 else
3229 {
3230 buf = alloc((unsigned)(STRLEN(eap->arg) + 14));
3231 if (buf != NULL)
3232 {
3233 if (eap->forceit)
3234 {
3235 /* ":compiler! {name}" sets global options */
3236 do_cmdline_cmd((char_u *)
3237 "command -nargs=* CompilerSet set <args>");
3238 }
3239 else
3240 {
3241 /* ":compiler! {name}" sets local options.
3242 * To remain backwards compatible "current_compiler" is always
3243 * used. A user's compiler plugin may set it, the distributed
3244 * plugin will then skip the settings. Afterwards set
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003245 * "b:current_compiler" and restore "current_compiler".
3246 * Explicitly prepend "g:" to make it work in a function. */
3247 old_cur_comp = get_var_value((char_u *)"g:current_compiler");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003248 if (old_cur_comp != NULL)
3249 old_cur_comp = vim_strsave(old_cur_comp);
3250 do_cmdline_cmd((char_u *)
3251 "command -nargs=* CompilerSet setlocal <args>");
3252 }
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003253 do_unlet((char_u *)"g:current_compiler", TRUE);
Bram Moolenaar2ce06f62005-01-31 19:19:04 +00003254 do_unlet((char_u *)"b:current_compiler", TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003255
3256 sprintf((char *)buf, "compiler/%s.vim", eap->arg);
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003257 if (source_runtime(buf, DIP_ALL) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003258 EMSG2(_("E666: compiler not supported: %s"), eap->arg);
3259 vim_free(buf);
3260
3261 do_cmdline_cmd((char_u *)":delcommand CompilerSet");
3262
3263 /* Set "b:current_compiler" from "current_compiler". */
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003264 p = get_var_value((char_u *)"g:current_compiler");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003265 if (p != NULL)
3266 set_internal_string_var((char_u *)"b:current_compiler", p);
3267
3268 /* Restore "current_compiler" for ":compiler {name}". */
3269 if (!eap->forceit)
3270 {
3271 if (old_cur_comp != NULL)
3272 {
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003273 set_internal_string_var((char_u *)"g:current_compiler",
Bram Moolenaar071d4272004-06-13 20:20:40 +00003274 old_cur_comp);
3275 vim_free(old_cur_comp);
3276 }
3277 else
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003278 do_unlet((char_u *)"g:current_compiler", TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003279 }
3280 }
3281 }
3282}
3283#endif
3284
3285/*
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003286 * ":runtime [what] {name}"
Bram Moolenaar071d4272004-06-13 20:20:40 +00003287 */
3288 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003289ex_runtime(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003290{
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003291 char_u *arg = eap->arg;
3292 char_u *p = skiptowhite(arg);
3293 int len = (int)(p - arg);
3294 int flags = eap->forceit ? DIP_ALL : 0;
3295
3296 if (STRNCMP(arg, "START", len) == 0)
3297 {
3298 flags += DIP_START + DIP_NORTP;
3299 arg = skipwhite(arg + len);
3300 }
3301 else if (STRNCMP(arg, "OPT", len) == 0)
3302 {
3303 flags += DIP_OPT + DIP_NORTP;
3304 arg = skipwhite(arg + len);
3305 }
3306 else if (STRNCMP(arg, "PACK", len) == 0)
3307 {
3308 flags += DIP_START + DIP_OPT + DIP_NORTP;
3309 arg = skipwhite(arg + len);
3310 }
3311 else if (STRNCMP(arg, "ALL", len) == 0)
3312 {
3313 flags += DIP_START + DIP_OPT;
3314 arg = skipwhite(arg + len);
3315 }
3316
3317 source_runtime(arg, flags);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003318}
3319
Bram Moolenaar071d4272004-06-13 20:20:40 +00003320 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003321source_callback(char_u *fname, void *cookie UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003322{
Bram Moolenaar910f66f2006-04-05 20:41:53 +00003323 (void)do_source(fname, FALSE, DOSO_NONE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003324}
3325
3326/*
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003327 * Find the file "name" in all directories in "path" and invoke
3328 * "callback(fname, cookie)".
3329 * "name" can contain wildcards.
3330 * When "flags" has DIP_ALL: source all files, otherwise only the first one.
3331 * When "flags" has DIP_DIR: find directories instead of files.
3332 * When "flags" has DIP_ERR: give an error message if there is no match.
3333 *
3334 * return FAIL when no file could be sourced, OK otherwise.
3335 */
Bram Moolenaar6bef5302016-03-12 21:28:26 +01003336 int
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003337do_in_path(
3338 char_u *path,
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003339 char_u *name,
Bram Moolenaar91715872016-03-03 17:13:03 +01003340 int flags,
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003341 void (*callback)(char_u *fname, void *ck),
3342 void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003343{
3344 char_u *rtp;
3345 char_u *np;
3346 char_u *buf;
3347 char_u *rtp_copy;
3348 char_u *tail;
3349 int num_files;
3350 char_u **files;
3351 int i;
3352 int did_one = FALSE;
3353#ifdef AMIGA
3354 struct Process *proc = (struct Process *)FindTask(0L);
3355 APTR save_winptr = proc->pr_WindowPtr;
3356
3357 /* Avoid a requester here for a volume that doesn't exist. */
3358 proc->pr_WindowPtr = (APTR)-1L;
3359#endif
3360
3361 /* Make a copy of 'runtimepath'. Invoking the callback may change the
3362 * value. */
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003363 rtp_copy = vim_strsave(path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003364 buf = alloc(MAXPATHL);
3365 if (buf != NULL && rtp_copy != NULL)
3366 {
Bram Moolenaarc09a6d62013-06-10 21:27:29 +02003367 if (p_verbose > 1 && name != NULL)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003368 {
3369 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00003370 smsg((char_u *)_("Searching for \"%s\" in \"%s\""),
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003371 (char *)name, (char *)path);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003372 verbose_leave();
3373 }
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00003374
Bram Moolenaar071d4272004-06-13 20:20:40 +00003375 /* Loop over all entries in 'runtimepath'. */
3376 rtp = rtp_copy;
Bram Moolenaar91715872016-03-03 17:13:03 +01003377 while (*rtp != NUL && ((flags & DIP_ALL) || !did_one))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003378 {
Bram Moolenaar66459b72016-08-06 19:01:55 +02003379 size_t buflen;
3380
Bram Moolenaar071d4272004-06-13 20:20:40 +00003381 /* Copy the path from 'runtimepath' to buf[]. */
3382 copy_option_part(&rtp, buf, MAXPATHL, ",");
Bram Moolenaar66459b72016-08-06 19:01:55 +02003383 buflen = STRLEN(buf);
3384
3385 /* Skip after or non-after directories. */
3386 if (flags & (DIP_NOAFTER | DIP_AFTER))
3387 {
3388 int is_after = buflen >= 5
3389 && STRCMP(buf + buflen - 5, "after") == 0;
3390
3391 if ((is_after && (flags & DIP_NOAFTER))
3392 || (!is_after && (flags & DIP_AFTER)))
3393 continue;
3394 }
3395
Bram Moolenaarc09a6d62013-06-10 21:27:29 +02003396 if (name == NULL)
3397 {
3398 (*callback)(buf, (void *) &cookie);
3399 if (!did_one)
3400 did_one = (cookie == NULL);
3401 }
Bram Moolenaar66459b72016-08-06 19:01:55 +02003402 else if (buflen + STRLEN(name) + 2 < MAXPATHL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003403 {
3404 add_pathsep(buf);
3405 tail = buf + STRLEN(buf);
3406
3407 /* Loop over all patterns in "name" */
3408 np = name;
Bram Moolenaar91715872016-03-03 17:13:03 +01003409 while (*np != NUL && ((flags & DIP_ALL) || !did_one))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003410 {
3411 /* Append the pattern from "name" to buf[]. */
3412 copy_option_part(&np, tail, (int)(MAXPATHL - (tail - buf)),
3413 "\t ");
3414
3415 if (p_verbose > 2)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003416 {
3417 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00003418 smsg((char_u *)_("Searching for \"%s\""), buf);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003419 verbose_leave();
3420 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003421
3422 /* Expand wildcards, invoke the callback for each match. */
3423 if (gen_expand_wildcards(1, &buf, &num_files, &files,
Bram Moolenaar91715872016-03-03 17:13:03 +01003424 (flags & DIP_DIR) ? EW_DIR : EW_FILE) == OK)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003425 {
3426 for (i = 0; i < num_files; ++i)
3427 {
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00003428 (*callback)(files[i], cookie);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003429 did_one = TRUE;
Bram Moolenaar91715872016-03-03 17:13:03 +01003430 if (!(flags & DIP_ALL))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003431 break;
3432 }
3433 FreeWild(num_files, files);
3434 }
3435 }
3436 }
3437 }
3438 }
3439 vim_free(buf);
3440 vim_free(rtp_copy);
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003441 if (!did_one && name != NULL)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003442 {
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003443 char *basepath = path == p_rtp ? "runtimepath" : "packpath";
3444
3445 if (flags & DIP_ERR)
3446 EMSG3(_(e_dirnotf), basepath, name);
3447 else if (p_verbose > 0)
3448 {
3449 verbose_enter();
3450 smsg((char_u *)_("not found in '%s': \"%s\""), basepath, name);
3451 verbose_leave();
3452 }
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003453 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003454
3455#ifdef AMIGA
3456 proc->pr_WindowPtr = save_winptr;
3457#endif
3458
3459 return did_one ? OK : FAIL;
3460}
3461
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003462/*
Bram Moolenaar07ecfa62017-06-27 14:43:55 +02003463 * Find "name" in "path". When found, invoke the callback function for
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003464 * it: callback(fname, "cookie")
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003465 * When "flags" has DIP_ALL repeat for all matches, otherwise only the first
3466 * one is used.
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003467 * Returns OK when at least one match found, FAIL otherwise.
3468 *
Bram Moolenaar07ecfa62017-06-27 14:43:55 +02003469 * If "name" is NULL calls callback for each entry in "path". Cookie is
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003470 * passed by reference in this case, setting it to NULL indicates that callback
3471 * has done its job.
3472 */
Bram Moolenaar07ecfa62017-06-27 14:43:55 +02003473 static int
3474do_in_path_and_pp(
3475 char_u *path,
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003476 char_u *name,
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003477 int flags,
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003478 void (*callback)(char_u *fname, void *ck),
3479 void *cookie)
3480{
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003481 int done = FAIL;
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003482 char_u *s;
3483 int len;
3484 char *start_dir = "pack/*/start/*/%s";
3485 char *opt_dir = "pack/*/opt/*/%s";
3486
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003487 if ((flags & DIP_NORTP) == 0)
Bram Moolenaar07ecfa62017-06-27 14:43:55 +02003488 done = do_in_path(path, name, flags, callback, cookie);
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003489
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003490 if ((done == FAIL || (flags & DIP_ALL)) && (flags & DIP_START))
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003491 {
Bram Moolenaar1c8b4ed2016-03-17 21:51:03 +01003492 len = (int)(STRLEN(start_dir) + STRLEN(name));
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003493 s = alloc(len);
3494 if (s == NULL)
3495 return FAIL;
3496 vim_snprintf((char *)s, len, start_dir, name);
3497 done = do_in_path(p_pp, s, flags, callback, cookie);
3498 vim_free(s);
3499 }
3500
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003501 if ((done == FAIL || (flags & DIP_ALL)) && (flags & DIP_OPT))
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003502 {
Bram Moolenaar1c8b4ed2016-03-17 21:51:03 +01003503 len = (int)(STRLEN(opt_dir) + STRLEN(name));
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003504 s = alloc(len);
3505 if (s == NULL)
3506 return FAIL;
3507 vim_snprintf((char *)s, len, opt_dir, name);
3508 done = do_in_path(p_pp, s, flags, callback, cookie);
3509 vim_free(s);
3510 }
3511
3512 return done;
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003513}
3514
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003515/*
Bram Moolenaar07ecfa62017-06-27 14:43:55 +02003516 * Just like do_in_path_and_pp(), using 'runtimepath' for "path".
3517 */
3518 int
3519do_in_runtimepath(
3520 char_u *name,
3521 int flags,
3522 void (*callback)(char_u *fname, void *ck),
3523 void *cookie)
3524{
3525 return do_in_path_and_pp(p_rtp, name, flags, callback, cookie);
3526}
3527
3528/*
3529 * Source the file "name" from all directories in 'runtimepath'.
3530 * "name" can contain wildcards.
3531 * When "flags" has DIP_ALL: source all files, otherwise only the first one.
3532 *
3533 * return FAIL when no file could be sourced, OK otherwise.
3534 */
3535 int
3536source_runtime(char_u *name, int flags)
3537{
3538 return source_in_path(p_rtp, name, flags);
3539}
3540
3541/*
3542 * Just like source_runtime(), but use "path" instead of 'runtimepath'.
3543 */
3544 int
3545source_in_path(char_u *path, char_u *name, int flags)
3546{
3547 return do_in_path_and_pp(path, name, flags, source_callback, NULL);
3548}
3549
3550
3551/*
Bram Moolenaarf3654822016-03-04 22:12:23 +01003552 * Expand wildcards in "pat" and invoke do_source() for each match.
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003553 */
3554 static void
Bram Moolenaarf3654822016-03-04 22:12:23 +01003555source_all_matches(char_u *pat)
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003556{
Bram Moolenaarf3654822016-03-04 22:12:23 +01003557 int num_files;
3558 char_u **files;
3559 int i;
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003560
Bram Moolenaarf3654822016-03-04 22:12:23 +01003561 if (gen_expand_wildcards(1, &pat, &num_files, &files, EW_FILE) == OK)
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003562 {
Bram Moolenaarf3654822016-03-04 22:12:23 +01003563 for (i = 0; i < num_files; ++i)
3564 (void)do_source(files[i], FALSE, DOSO_NONE);
3565 FreeWild(num_files, files);
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003566 }
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003567}
3568
Bram Moolenaar2374faa2018-02-04 17:47:42 +01003569/*
3570 * Add the package directory to 'runtimepath'.
3571 */
3572 static int
3573add_pack_dir_to_rtp(char_u *fname)
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003574{
Bram Moolenaarf3654822016-03-04 22:12:23 +01003575 char_u *p4, *p3, *p2, *p1, *p;
3576 char_u *insp;
Bram Moolenaar91715872016-03-03 17:13:03 +01003577 int c;
3578 char_u *new_rtp;
3579 int keep;
Bram Moolenaarb0550662016-05-31 21:37:36 +02003580 size_t oldlen;
3581 size_t addlen;
Bram Moolenaar2374faa2018-02-04 17:47:42 +01003582 char_u *afterdir = NULL;
Bram Moolenaarb0550662016-05-31 21:37:36 +02003583 size_t afterlen = 0;
Bram Moolenaar2374faa2018-02-04 17:47:42 +01003584 char_u *ffname = NULL;
Bram Moolenaarfef524b2016-07-02 22:07:22 +02003585 size_t fname_len;
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003586 char_u *buf = NULL;
3587 char_u *rtp_ffname;
3588 int match;
Bram Moolenaar2374faa2018-02-04 17:47:42 +01003589 int retval = FAIL;
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003590
Bram Moolenaar2374faa2018-02-04 17:47:42 +01003591 p4 = p3 = p2 = p1 = get_past_head(fname);
3592 for (p = p1; *p; MB_PTR_ADV(p))
3593 if (vim_ispathsep_nocolon(*p))
3594 {
3595 p4 = p3; p3 = p2; p2 = p1; p1 = p;
3596 }
3597
3598 /* now we have:
3599 * rtp/pack/name/start/name
3600 * p4 p3 p2 p1
3601 *
3602 * find the part up to "pack" in 'runtimepath' */
3603 c = *++p4; /* append pathsep in order to expand symlink */
3604 *p4 = NUL;
3605 ffname = fix_fname(fname);
3606 *p4 = c;
Bram Moolenaar91715872016-03-03 17:13:03 +01003607 if (ffname == NULL)
Bram Moolenaar2374faa2018-02-04 17:47:42 +01003608 return FAIL;
3609
3610 /* Find "ffname" in "p_rtp", ignoring '/' vs '\' differences. */
3611 fname_len = STRLEN(ffname);
3612 insp = p_rtp;
3613 buf = alloc(MAXPATHL);
3614 if (buf == NULL)
3615 goto theend;
3616 while (*insp != NUL)
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003617 {
Bram Moolenaar2374faa2018-02-04 17:47:42 +01003618 copy_option_part(&insp, buf, MAXPATHL, ",");
3619 add_pathsep(buf);
3620 rtp_ffname = fix_fname(buf);
3621 if (rtp_ffname == NULL)
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003622 goto theend;
Bram Moolenaar2374faa2018-02-04 17:47:42 +01003623 match = vim_fnamencmp(rtp_ffname, ffname, fname_len) == 0;
3624 vim_free(rtp_ffname);
3625 if (match)
3626 break;
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003627 }
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003628
Bram Moolenaar2374faa2018-02-04 17:47:42 +01003629 if (*insp == NUL)
3630 /* not found, append at the end */
3631 insp = p_rtp + STRLEN(p_rtp);
3632 else
3633 /* append after the matching directory. */
3634 --insp;
3635
3636 /* check if rtp/pack/name/start/name/after exists */
3637 afterdir = concat_fnames(fname, (char_u *)"after", TRUE);
3638 if (afterdir != NULL && mch_isdir(afterdir))
3639 afterlen = STRLEN(afterdir) + 1; /* add one for comma */
3640
3641 oldlen = STRLEN(p_rtp);
3642 addlen = STRLEN(fname) + 1; /* add one for comma */
3643 new_rtp = alloc((int)(oldlen + addlen + afterlen + 1));
3644 /* add one for NUL */
3645 if (new_rtp == NULL)
3646 goto theend;
3647 keep = (int)(insp - p_rtp);
3648 mch_memmove(new_rtp, p_rtp, keep);
3649 new_rtp[keep] = ',';
3650 mch_memmove(new_rtp + keep + 1, fname, addlen);
3651 if (p_rtp[keep] != NUL)
3652 mch_memmove(new_rtp + keep + addlen, p_rtp + keep, oldlen - keep + 1);
3653 if (afterlen > 0)
Bram Moolenaarf3654822016-03-04 22:12:23 +01003654 {
Bram Moolenaar2374faa2018-02-04 17:47:42 +01003655 STRCAT(new_rtp, ",");
3656 STRCAT(new_rtp, afterdir);
Bram Moolenaarf3654822016-03-04 22:12:23 +01003657 }
Bram Moolenaar2374faa2018-02-04 17:47:42 +01003658 set_option_value((char_u *)"rtp", 0L, new_rtp, 0);
3659 vim_free(new_rtp);
3660 retval = OK;
Bram Moolenaarf3654822016-03-04 22:12:23 +01003661
3662theend:
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003663 vim_free(buf);
Bram Moolenaarf3654822016-03-04 22:12:23 +01003664 vim_free(ffname);
Bram Moolenaar2374faa2018-02-04 17:47:42 +01003665 vim_free(afterdir);
3666 return retval;
3667}
3668
3669/*
3670 * Load scripts in "plugin" and "ftdetect" directories of the package.
3671 */
3672 static int
3673load_pack_plugin(char_u *fname)
3674{
3675 static char *plugpat = "%s/plugin/**/*.vim";
3676 static char *ftpat = "%s/ftdetect/*.vim";
3677 int len;
3678 char_u *ffname = fix_fname(fname);
3679 char_u *pat = NULL;
3680 int retval = FAIL;
3681
3682 if (ffname == NULL)
3683 return FAIL;
3684 len = (int)STRLEN(ffname) + (int)STRLEN(ftpat);
3685 pat = alloc(len);
3686 if (pat == NULL)
3687 goto theend;
3688 vim_snprintf((char *)pat, len, plugpat, ffname);
3689 source_all_matches(pat);
3690
3691#ifdef FEAT_AUTOCMD
3692 {
3693 char_u *cmd = vim_strsave((char_u *)"g:did_load_filetypes");
3694
3695 /* If runtime/filetype.vim wasn't loaded yet, the scripts will be
3696 * found when it loads. */
3697 if (cmd != NULL && eval_to_number(cmd) > 0)
3698 {
3699 do_cmdline_cmd((char_u *)"augroup filetypedetect");
3700 vim_snprintf((char *)pat, len, ftpat, ffname);
3701 source_all_matches(pat);
3702 do_cmdline_cmd((char_u *)"augroup END");
3703 }
3704 vim_free(cmd);
3705 }
3706#endif
3707 vim_free(pat);
3708 retval = OK;
3709
3710theend:
3711 vim_free(ffname);
3712 return retval;
3713}
3714
3715/* used for "cookie" of add_pack_plugin() */
3716static int APP_ADD_DIR;
3717static int APP_LOAD;
3718static int APP_BOTH;
3719
3720 static void
3721add_pack_plugin(char_u *fname, void *cookie)
3722{
3723 if (cookie != &APP_LOAD && strstr((char *)p_rtp, (char *)fname) == NULL)
3724 /* directory is not yet in 'runtimepath', add it */
3725 if (add_pack_dir_to_rtp(fname) == FAIL)
3726 return;
3727
3728 if (cookie != &APP_ADD_DIR)
3729 load_pack_plugin(fname);
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003730}
3731
Bram Moolenaarce876aa2017-06-04 17:47:42 +02003732/*
3733 * Add all packages in the "start" directory to 'runtimepath'.
3734 */
3735 void
3736add_pack_start_dirs(void)
3737{
3738 do_in_path(p_pp, (char_u *)"pack/*/start/*", DIP_ALL + DIP_DIR,
3739 add_pack_plugin, &APP_ADD_DIR);
3740}
3741
3742/*
3743 * Load plugins from all packages in the "start" directory.
3744 */
3745 void
3746load_start_packages(void)
3747{
3748 did_source_packages = TRUE;
3749 do_in_path(p_pp, (char_u *)"pack/*/start/*", DIP_ALL + DIP_DIR,
3750 add_pack_plugin, &APP_LOAD);
3751}
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003752
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003753/*
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003754 * ":packloadall"
Bram Moolenaarf3654822016-03-04 22:12:23 +01003755 * Find plugins in the package directories and source them.
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003756 */
3757 void
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003758ex_packloadall(exarg_T *eap)
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003759{
Bram Moolenaarce876aa2017-06-04 17:47:42 +02003760 if (!did_source_packages || eap->forceit)
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003761 {
Bram Moolenaar49b27322016-04-05 21:13:00 +02003762 /* First do a round to add all directories to 'runtimepath', then load
3763 * the plugins. This allows for plugins to use an autoload directory
3764 * of another plugin. */
Bram Moolenaarce876aa2017-06-04 17:47:42 +02003765 add_pack_start_dirs();
3766 load_start_packages();
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003767 }
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003768}
3769
3770/*
Bram Moolenaarf3654822016-03-04 22:12:23 +01003771 * ":packadd[!] {name}"
Bram Moolenaar91715872016-03-03 17:13:03 +01003772 */
3773 void
3774ex_packadd(exarg_T *eap)
3775{
Bram Moolenaar9e1d3992017-12-17 14:26:46 +01003776 static char *plugpat = "pack/*/%s/%s";
Bram Moolenaar91715872016-03-03 17:13:03 +01003777 int len;
3778 char *pat;
Bram Moolenaar9e1d3992017-12-17 14:26:46 +01003779 int round;
3780 int res = OK;
Bram Moolenaar91715872016-03-03 17:13:03 +01003781
Bram Moolenaar9e1d3992017-12-17 14:26:46 +01003782 /* Round 1: use "start", round 2: use "opt". */
3783 for (round = 1; round <= 2; ++round)
3784 {
3785 /* Only look under "start" when loading packages wasn't done yet. */
3786 if (round == 1 && did_source_packages)
3787 continue;
3788
3789 len = (int)STRLEN(plugpat) + (int)STRLEN(eap->arg) + 5;
3790 pat = (char *)alloc(len);
3791 if (pat == NULL)
3792 return;
3793 vim_snprintf(pat, len, plugpat, round == 1 ? "start" : "opt", eap->arg);
3794 /* The first round don't give a "not found" error, in the second round
3795 * only when nothing was found in the first round. */
3796 res = do_in_path(p_pp, (char_u *)pat,
3797 DIP_ALL + DIP_DIR + (round == 2 && res == FAIL ? DIP_ERR : 0),
3798 add_pack_plugin, eap->forceit ? &APP_ADD_DIR : &APP_BOTH);
3799 vim_free(pat);
3800 }
Bram Moolenaar91715872016-03-03 17:13:03 +01003801}
3802
Bram Moolenaar071d4272004-06-13 20:20:40 +00003803#if defined(FEAT_EVAL) && defined(FEAT_AUTOCMD)
3804/*
3805 * ":options"
3806 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003807 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003808ex_options(
3809 exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003810{
Bram Moolenaarab6c8582017-08-11 17:15:09 +02003811 vim_setenv((char_u *)"OPTWIN_CMD", (char_u *)(cmdmod.tab ? "tab" : ""));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003812 cmd_source((char_u *)SYS_OPTWIN_FILE, NULL);
3813}
3814#endif
3815
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01003816#if defined(FEAT_PYTHON3) || defined(FEAT_PYTHON) || defined(PROTO)
3817
3818# if (defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)) || defined(PROTO)
3819/*
3820 * Detect Python 3 or 2, and initialize 'pyxversion'.
3821 */
3822 void
3823init_pyxversion(void)
3824{
3825 if (p_pyx == 0)
3826 {
3827 if (python3_enabled(FALSE))
3828 p_pyx = 3;
3829 else if (python_enabled(FALSE))
3830 p_pyx = 2;
3831 }
3832}
3833# endif
3834
3835/*
3836 * Does a file contain one of the following strings at the beginning of any
3837 * line?
3838 * "#!(any string)python2" => returns 2
3839 * "#!(any string)python3" => returns 3
3840 * "# requires python 2.x" => returns 2
3841 * "# requires python 3.x" => returns 3
3842 * otherwise return 0.
3843 */
3844 static int
3845requires_py_version(char_u *filename)
3846{
3847 FILE *file;
3848 int requires_py_version = 0;
3849 int i, lines;
3850
3851 lines = (int)p_mls;
3852 if (lines < 0)
3853 lines = 5;
3854
3855 file = mch_fopen((char *)filename, "r");
3856 if (file != NULL)
3857 {
3858 for (i = 0; i < lines; i++)
3859 {
3860 if (vim_fgets(IObuff, IOSIZE, file))
3861 break;
3862 if (i == 0 && IObuff[0] == '#' && IObuff[1] == '!')
3863 {
3864 /* Check shebang. */
3865 if (strstr((char *)IObuff + 2, "python2") != NULL)
3866 {
3867 requires_py_version = 2;
3868 break;
3869 }
3870 if (strstr((char *)IObuff + 2, "python3") != NULL)
3871 {
3872 requires_py_version = 3;
3873 break;
3874 }
3875 }
3876 IObuff[21] = '\0';
3877 if (STRCMP("# requires python 2.x", IObuff) == 0)
3878 {
3879 requires_py_version = 2;
3880 break;
3881 }
3882 if (STRCMP("# requires python 3.x", IObuff) == 0)
3883 {
3884 requires_py_version = 3;
3885 break;
3886 }
3887 }
3888 fclose(file);
3889 }
3890 return requires_py_version;
3891}
3892
3893
3894/*
3895 * Source a python file using the requested python version.
3896 */
3897 static void
3898source_pyx_file(exarg_T *eap, char_u *fname)
3899{
3900 exarg_T ex;
3901 int v = requires_py_version(fname);
3902
3903# if defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)
3904 init_pyxversion();
3905# endif
3906 if (v == 0)
3907 {
3908# if defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)
3909 /* user didn't choose a preference, 'pyx' is used */
3910 v = p_pyx;
3911# elif defined(FEAT_PYTHON)
3912 v = 2;
3913# elif defined(FEAT_PYTHON3)
3914 v = 3;
3915# endif
3916 }
3917
3918 /*
3919 * now source, if required python version is not supported show
3920 * unobtrusive message.
3921 */
3922 if (eap == NULL)
3923 vim_memset(&ex, 0, sizeof(ex));
3924 else
3925 ex = *eap;
3926 ex.arg = fname;
3927 ex.cmd = (char_u *)(v == 2 ? "pyfile" : "pyfile3");
3928
3929 if (v == 2)
3930 {
3931# ifdef FEAT_PYTHON
3932 ex_pyfile(&ex);
3933# else
3934 vim_snprintf((char *)IObuff, IOSIZE,
3935 _("W20: Required python version 2.x not supported, ignoring file: %s"),
3936 fname);
3937 MSG(IObuff);
3938# endif
3939 return;
3940 }
3941 else
3942 {
3943# ifdef FEAT_PYTHON3
3944 ex_py3file(&ex);
3945# else
3946 vim_snprintf((char *)IObuff, IOSIZE,
3947 _("W21: Required python version 3.x not supported, ignoring file: %s"),
3948 fname);
3949 MSG(IObuff);
3950# endif
3951 return;
3952 }
3953}
3954
3955/*
3956 * ":pyxfile {fname}"
3957 */
3958 void
3959ex_pyxfile(exarg_T *eap)
3960{
3961 source_pyx_file(eap, eap->arg);
3962}
3963
3964/*
3965 * ":pyx"
3966 */
3967 void
3968ex_pyx(exarg_T *eap)
3969{
3970# if defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)
3971 init_pyxversion();
3972 if (p_pyx == 2)
3973 ex_python(eap);
3974 else
3975 ex_py3(eap);
3976# elif defined(FEAT_PYTHON)
3977 ex_python(eap);
3978# elif defined(FEAT_PYTHON3)
3979 ex_py3(eap);
3980# endif
3981}
3982
3983/*
3984 * ":pyxdo"
3985 */
3986 void
3987ex_pyxdo(exarg_T *eap)
3988{
3989# if defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)
3990 init_pyxversion();
3991 if (p_pyx == 2)
3992 ex_pydo(eap);
3993 else
3994 ex_py3do(eap);
3995# elif defined(FEAT_PYTHON)
3996 ex_pydo(eap);
3997# elif defined(FEAT_PYTHON3)
3998 ex_py3do(eap);
3999# endif
4000}
4001
4002#endif
4003
Bram Moolenaar071d4272004-06-13 20:20:40 +00004004/*
4005 * ":source {fname}"
4006 */
4007 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004008ex_source(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004009{
4010#ifdef FEAT_BROWSE
4011 if (cmdmod.browse)
4012 {
4013 char_u *fname = NULL;
4014
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00004015 fname = do_browse(0, (char_u *)_("Source Vim script"), eap->arg,
Bram Moolenaar071d4272004-06-13 20:20:40 +00004016 NULL, NULL, BROWSE_FILTER_MACROS, NULL);
4017 if (fname != NULL)
4018 {
4019 cmd_source(fname, eap);
4020 vim_free(fname);
4021 }
4022 }
4023 else
4024#endif
4025 cmd_source(eap->arg, eap);
4026}
4027
4028 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004029cmd_source(char_u *fname, exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004030{
4031 if (*fname == NUL)
4032 EMSG(_(e_argreq));
4033
Bram Moolenaar071d4272004-06-13 20:20:40 +00004034 else if (eap != NULL && eap->forceit)
Bram Moolenaar84a05ac2013-05-06 04:24:17 +02004035 /* ":source!": read Normal mode commands
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00004036 * Need to execute the commands directly. This is required at least
4037 * for:
Bram Moolenaar071d4272004-06-13 20:20:40 +00004038 * - ":g" command busy
4039 * - after ":argdo", ":windo" or ":bufdo"
4040 * - another command follows
4041 * - inside a loop
4042 */
4043 openscript(fname, global_busy || listcmd_busy || eap->nextcmd != NULL
4044#ifdef FEAT_EVAL
4045 || eap->cstack->cs_idx >= 0
4046#endif
4047 );
4048
4049 /* ":source" read ex commands */
Bram Moolenaar910f66f2006-04-05 20:41:53 +00004050 else if (do_source(fname, FALSE, DOSO_NONE) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004051 EMSG2(_(e_notopen), fname);
4052}
4053
4054/*
4055 * ":source" and associated commands.
4056 */
4057/*
4058 * Structure used to store info for each sourced file.
4059 * It is shared between do_source() and getsourceline().
4060 * This is required, because it needs to be handed to do_cmdline() and
4061 * sourcing can be done recursively.
4062 */
4063struct source_cookie
4064{
4065 FILE *fp; /* opened file for sourcing */
4066 char_u *nextline; /* if not NULL: line that was read ahead */
4067 int finished; /* ":finish" used */
Bram Moolenaar860cae12010-06-05 23:22:07 +02004068#if defined(USE_CRNL) || defined(USE_CR)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004069 int fileformat; /* EOL_UNKNOWN, EOL_UNIX or EOL_DOS */
4070 int error; /* TRUE if LF found after CR-LF */
4071#endif
4072#ifdef FEAT_EVAL
4073 linenr_T breakpoint; /* next line with breakpoint or zero */
4074 char_u *fname; /* name of sourced file */
4075 int dbg_tick; /* debug_tick when breakpoint was set */
4076 int level; /* top nesting level of sourced file */
4077#endif
4078#ifdef FEAT_MBYTE
4079 vimconv_T conv; /* type of conversion */
4080#endif
4081};
4082
4083#ifdef FEAT_EVAL
4084/*
4085 * Return the address holding the next breakpoint line for a source cookie.
4086 */
4087 linenr_T *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004088source_breakpoint(void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004089{
4090 return &((struct source_cookie *)cookie)->breakpoint;
4091}
4092
4093/*
4094 * Return the address holding the debug tick for a source cookie.
4095 */
4096 int *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004097source_dbg_tick(void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004098{
4099 return &((struct source_cookie *)cookie)->dbg_tick;
4100}
4101
4102/*
4103 * Return the nesting level for a source cookie.
4104 */
4105 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004106source_level(void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004107{
4108 return ((struct source_cookie *)cookie)->level;
4109}
4110#endif
4111
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01004112static char_u *get_one_sourceline(struct source_cookie *sp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004113
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004114#if (defined(WIN32) && defined(FEAT_CSCOPE)) || defined(HAVE_FD_CLOEXEC)
4115# define USE_FOPEN_NOINH
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01004116static FILE *fopen_noinh_readbin(char *filename);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004117
4118/*
4119 * Special function to open a file without handle inheritance.
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004120 * When possible the handle is closed on exec().
Bram Moolenaar071d4272004-06-13 20:20:40 +00004121 */
4122 static FILE *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004123fopen_noinh_readbin(char *filename)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004124{
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004125# ifdef WIN32
Bram Moolenaar8d8ef0b2010-01-20 21:41:47 +01004126 int fd_tmp = mch_open(filename, O_RDONLY | O_BINARY | O_NOINHERIT, 0);
4127# else
4128 int fd_tmp = mch_open(filename, O_RDONLY, 0);
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004129# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004130
4131 if (fd_tmp == -1)
4132 return NULL;
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004133
4134# ifdef HAVE_FD_CLOEXEC
4135 {
4136 int fdflags = fcntl(fd_tmp, F_GETFD);
4137 if (fdflags >= 0 && (fdflags & FD_CLOEXEC) == 0)
Bram Moolenaarcde88542015-08-11 19:14:00 +02004138 (void)fcntl(fd_tmp, F_SETFD, fdflags | FD_CLOEXEC);
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004139 }
4140# endif
4141
Bram Moolenaar071d4272004-06-13 20:20:40 +00004142 return fdopen(fd_tmp, READBIN);
4143}
4144#endif
4145
4146
4147/*
4148 * do_source: Read the file "fname" and execute its lines as EX commands.
4149 *
4150 * This function may be called recursively!
4151 *
4152 * return FAIL if file could not be opened, OK otherwise
4153 */
4154 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004155do_source(
4156 char_u *fname,
4157 int check_other, /* check for .vimrc and _vimrc */
4158 int is_vimrc) /* DOSO_ value */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004159{
4160 struct source_cookie cookie;
4161 char_u *save_sourcing_name;
4162 linenr_T save_sourcing_lnum;
4163 char_u *p;
4164 char_u *fname_exp;
Bram Moolenaar73881402009-02-04 16:50:47 +00004165 char_u *firstline = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004166 int retval = FAIL;
4167#ifdef FEAT_EVAL
4168 scid_T save_current_SID;
4169 static scid_T last_current_SID = 0;
4170 void *save_funccalp;
4171 int save_debug_break_level = debug_break_level;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004172 scriptitem_T *si = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004173# ifdef UNIX
Bram Moolenaar8767f522016-07-01 17:17:39 +02004174 stat_T st;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004175 int stat_ok;
4176# endif
4177#endif
4178#ifdef STARTUPTIME
4179 struct timeval tv_rel;
4180 struct timeval tv_start;
4181#endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00004182#ifdef FEAT_PROFILE
4183 proftime_T wait_start;
4184#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004185
Bram Moolenaar071d4272004-06-13 20:20:40 +00004186 p = expand_env_save(fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004187 if (p == NULL)
4188 return retval;
4189 fname_exp = fix_fname(p);
4190 vim_free(p);
4191 if (fname_exp == NULL)
4192 return retval;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004193 if (mch_isdir(fname_exp))
4194 {
Bram Moolenaar555b2802005-05-19 21:08:39 +00004195 smsg((char_u *)_("Cannot source a directory: \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004196 goto theend;
4197 }
4198
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00004199#ifdef FEAT_AUTOCMD
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00004200 /* Apply SourceCmd autocommands, they should get the file and source it. */
4201 if (has_autocmd(EVENT_SOURCECMD, fname_exp, NULL)
4202 && apply_autocmds(EVENT_SOURCECMD, fname_exp, fname_exp,
4203 FALSE, curbuf))
Bram Moolenaarf33943e2008-01-15 21:17:30 +00004204 {
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00004205# ifdef FEAT_EVAL
Bram Moolenaarf33943e2008-01-15 21:17:30 +00004206 retval = aborting() ? FAIL : OK;
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00004207# else
Bram Moolenaarf33943e2008-01-15 21:17:30 +00004208 retval = OK;
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00004209# endif
Bram Moolenaarf33943e2008-01-15 21:17:30 +00004210 goto theend;
4211 }
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00004212
4213 /* Apply SourcePre autocommands, they may get the file. */
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00004214 apply_autocmds(EVENT_SOURCEPRE, fname_exp, fname_exp, FALSE, curbuf);
4215#endif
4216
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004217#ifdef USE_FOPEN_NOINH
Bram Moolenaar071d4272004-06-13 20:20:40 +00004218 cookie.fp = fopen_noinh_readbin((char *)fname_exp);
4219#else
4220 cookie.fp = mch_fopen((char *)fname_exp, READBIN);
4221#endif
4222 if (cookie.fp == NULL && check_other)
4223 {
4224 /*
4225 * Try again, replacing file name ".vimrc" by "_vimrc" or vice versa,
4226 * and ".exrc" by "_exrc" or vice versa.
4227 */
4228 p = gettail(fname_exp);
4229 if ((*p == '.' || *p == '_')
4230 && (STRICMP(p + 1, "vimrc") == 0
4231 || STRICMP(p + 1, "gvimrc") == 0
4232 || STRICMP(p + 1, "exrc") == 0))
4233 {
4234 if (*p == '_')
4235 *p = '.';
4236 else
4237 *p = '_';
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004238#ifdef USE_FOPEN_NOINH
Bram Moolenaar071d4272004-06-13 20:20:40 +00004239 cookie.fp = fopen_noinh_readbin((char *)fname_exp);
4240#else
4241 cookie.fp = mch_fopen((char *)fname_exp, READBIN);
4242#endif
4243 }
4244 }
4245
4246 if (cookie.fp == NULL)
4247 {
4248 if (p_verbose > 0)
4249 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004250 verbose_enter();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004251 if (sourcing_name == NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00004252 smsg((char_u *)_("could not source \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004253 else
4254 smsg((char_u *)_("line %ld: could not source \"%s\""),
Bram Moolenaar555b2802005-05-19 21:08:39 +00004255 sourcing_lnum, fname);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004256 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004257 }
4258 goto theend;
4259 }
4260
4261 /*
4262 * The file exists.
4263 * - In verbose mode, give a message.
4264 * - For a vimrc file, may want to set 'compatible', call vimrc_found().
4265 */
4266 if (p_verbose > 1)
4267 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004268 verbose_enter();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004269 if (sourcing_name == NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00004270 smsg((char_u *)_("sourcing \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004271 else
4272 smsg((char_u *)_("line %ld: sourcing \"%s\""),
Bram Moolenaar555b2802005-05-19 21:08:39 +00004273 sourcing_lnum, fname);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004274 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004275 }
Bram Moolenaar910f66f2006-04-05 20:41:53 +00004276 if (is_vimrc == DOSO_VIMRC)
4277 vimrc_found(fname_exp, (char_u *)"MYVIMRC");
4278 else if (is_vimrc == DOSO_GVIMRC)
4279 vimrc_found(fname_exp, (char_u *)"MYGVIMRC");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004280
4281#ifdef USE_CRNL
4282 /* If no automatic file format: Set default to CR-NL. */
4283 if (*p_ffs == NUL)
4284 cookie.fileformat = EOL_DOS;
4285 else
4286 cookie.fileformat = EOL_UNKNOWN;
4287 cookie.error = FALSE;
4288#endif
4289
4290#ifdef USE_CR
4291 /* If no automatic file format: Set default to CR. */
4292 if (*p_ffs == NUL)
4293 cookie.fileformat = EOL_MAC;
4294 else
4295 cookie.fileformat = EOL_UNKNOWN;
4296 cookie.error = FALSE;
4297#endif
4298
4299 cookie.nextline = NULL;
4300 cookie.finished = FALSE;
4301
4302#ifdef FEAT_EVAL
4303 /*
4304 * Check if this script has a breakpoint.
4305 */
4306 cookie.breakpoint = dbg_find_breakpoint(TRUE, fname_exp, (linenr_T)0);
4307 cookie.fname = fname_exp;
4308 cookie.dbg_tick = debug_tick;
4309
4310 cookie.level = ex_nesting_level;
4311#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004312
4313 /*
4314 * Keep the sourcing name/lnum, for recursive calls.
4315 */
4316 save_sourcing_name = sourcing_name;
4317 sourcing_name = fname_exp;
4318 save_sourcing_lnum = sourcing_lnum;
4319 sourcing_lnum = 0;
4320
4321#ifdef STARTUPTIME
Bram Moolenaarc4e41982010-01-19 16:31:47 +01004322 if (time_fd != NULL)
4323 time_push(&tv_rel, &tv_start);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004324#endif
4325
4326#ifdef FEAT_EVAL
Bram Moolenaar05159a02005-02-26 23:04:13 +00004327# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004328 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004329 prof_child_enter(&wait_start); /* entering a child now */
4330# endif
4331
4332 /* Don't use local function variables, if called from a function.
4333 * Also starts profiling timer for nested script. */
4334 save_funccalp = save_funccal();
4335
Bram Moolenaar071d4272004-06-13 20:20:40 +00004336 /*
4337 * Check if this script was sourced before to finds its SID.
4338 * If it's new, generate a new SID.
4339 */
4340 save_current_SID = current_SID;
4341# ifdef UNIX
4342 stat_ok = (mch_stat((char *)fname_exp, &st) >= 0);
4343# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00004344 for (current_SID = script_items.ga_len; current_SID > 0; --current_SID)
4345 {
4346 si = &SCRIPT_ITEM(current_SID);
4347 if (si->sn_name != NULL
Bram Moolenaar071d4272004-06-13 20:20:40 +00004348 && (
4349# ifdef UNIX
Bram Moolenaar7c626922005-02-07 22:01:03 +00004350 /* Compare dev/ino when possible, it catches symbolic
4351 * links. Also compare file names, the inode may change
4352 * when the file was edited. */
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00004353 ((stat_ok && si->sn_dev_valid)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004354 && (si->sn_dev == st.st_dev
4355 && si->sn_ino == st.st_ino)) ||
Bram Moolenaar071d4272004-06-13 20:20:40 +00004356# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00004357 fnamecmp(si->sn_name, fname_exp) == 0))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004358 break;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004359 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004360 if (current_SID == 0)
4361 {
4362 current_SID = ++last_current_SID;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004363 if (ga_grow(&script_items, (int)(current_SID - script_items.ga_len))
4364 == FAIL)
4365 goto almosttheend;
4366 while (script_items.ga_len < current_SID)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004367 {
Bram Moolenaar05159a02005-02-26 23:04:13 +00004368 ++script_items.ga_len;
4369 SCRIPT_ITEM(script_items.ga_len).sn_name = NULL;
4370# ifdef FEAT_PROFILE
4371 SCRIPT_ITEM(script_items.ga_len).sn_prof_on = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004372# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004373 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00004374 si = &SCRIPT_ITEM(current_SID);
4375 si->sn_name = fname_exp;
4376 fname_exp = NULL;
4377# ifdef UNIX
4378 if (stat_ok)
4379 {
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00004380 si->sn_dev_valid = TRUE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004381 si->sn_dev = st.st_dev;
4382 si->sn_ino = st.st_ino;
4383 }
4384 else
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00004385 si->sn_dev_valid = FALSE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004386# endif
4387
Bram Moolenaar071d4272004-06-13 20:20:40 +00004388 /* Allocate the local script variables to use for this script. */
4389 new_script_vars(current_SID);
4390 }
4391
Bram Moolenaar05159a02005-02-26 23:04:13 +00004392# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004393 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004394 {
4395 int forceit;
4396
4397 /* Check if we do profiling for this script. */
4398 if (!si->sn_prof_on && has_profiling(TRUE, si->sn_name, &forceit))
4399 {
4400 script_do_profile(si);
4401 si->sn_pr_force = forceit;
4402 }
4403 if (si->sn_prof_on)
4404 {
4405 ++si->sn_pr_count;
4406 profile_start(&si->sn_pr_start);
4407 profile_zero(&si->sn_pr_children);
4408 }
4409 }
4410# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004411#endif
4412
Bram Moolenaar67435d92017-10-19 21:04:37 +02004413#ifdef FEAT_MBYTE
4414 cookie.conv.vc_type = CONV_NONE; /* no conversion */
4415
4416 /* Read the first line so we can check for a UTF-8 BOM. */
4417 firstline = getsourceline(0, (void *)&cookie, 0);
4418 if (firstline != NULL && STRLEN(firstline) >= 3 && firstline[0] == 0xef
4419 && firstline[1] == 0xbb && firstline[2] == 0xbf)
4420 {
4421 /* Found BOM; setup conversion, skip over BOM and recode the line. */
4422 convert_setup(&cookie.conv, (char_u *)"utf-8", p_enc);
4423 p = string_convert(&cookie.conv, firstline + 3, NULL);
4424 if (p == NULL)
4425 p = vim_strsave(firstline + 3);
4426 if (p != NULL)
4427 {
4428 vim_free(firstline);
4429 firstline = p;
4430 }
4431 }
4432#endif
4433
Bram Moolenaar071d4272004-06-13 20:20:40 +00004434 /*
4435 * Call do_cmdline, which will call getsourceline() to get the lines.
4436 */
Bram Moolenaar73881402009-02-04 16:50:47 +00004437 do_cmdline(firstline, getsourceline, (void *)&cookie,
Bram Moolenaar071d4272004-06-13 20:20:40 +00004438 DOCMD_VERBOSE|DOCMD_NOWAIT|DOCMD_REPEAT);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004439 retval = OK;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004440
4441#ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004442 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004443 {
4444 /* Get "si" again, "script_items" may have been reallocated. */
4445 si = &SCRIPT_ITEM(current_SID);
4446 if (si->sn_prof_on)
4447 {
4448 profile_end(&si->sn_pr_start);
4449 profile_sub_wait(&wait_start, &si->sn_pr_start);
4450 profile_add(&si->sn_pr_total, &si->sn_pr_start);
Bram Moolenaar1056d982006-03-09 22:37:52 +00004451 profile_self(&si->sn_pr_self, &si->sn_pr_start,
4452 &si->sn_pr_children);
Bram Moolenaar05159a02005-02-26 23:04:13 +00004453 }
4454 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004455#endif
4456
4457 if (got_int)
4458 EMSG(_(e_interr));
4459 sourcing_name = save_sourcing_name;
4460 sourcing_lnum = save_sourcing_lnum;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004461 if (p_verbose > 1)
4462 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004463 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00004464 smsg((char_u *)_("finished sourcing %s"), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004465 if (sourcing_name != NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00004466 smsg((char_u *)_("continuing in %s"), sourcing_name);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004467 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004468 }
4469#ifdef STARTUPTIME
Bram Moolenaarc4e41982010-01-19 16:31:47 +01004470 if (time_fd != NULL)
4471 {
4472 vim_snprintf((char *)IObuff, IOSIZE, "sourcing %s", fname);
4473 time_msg((char *)IObuff, &tv_start);
4474 time_pop(&tv_rel);
4475 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004476#endif
4477
4478#ifdef FEAT_EVAL
4479 /*
4480 * After a "finish" in debug mode, need to break at first command of next
4481 * sourced file.
4482 */
4483 if (save_debug_break_level > ex_nesting_level
4484 && debug_break_level == ex_nesting_level)
4485 ++debug_break_level;
4486#endif
4487
Bram Moolenaar05159a02005-02-26 23:04:13 +00004488#ifdef FEAT_EVAL
4489almosttheend:
4490 current_SID = save_current_SID;
4491 restore_funccal(save_funccalp);
4492# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004493 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004494 prof_child_exit(&wait_start); /* leaving a child now */
4495# endif
4496#endif
4497 fclose(cookie.fp);
4498 vim_free(cookie.nextline);
Bram Moolenaar73881402009-02-04 16:50:47 +00004499 vim_free(firstline);
Bram Moolenaar05159a02005-02-26 23:04:13 +00004500#ifdef FEAT_MBYTE
4501 convert_setup(&cookie.conv, NULL, NULL);
4502#endif
4503
Bram Moolenaar071d4272004-06-13 20:20:40 +00004504theend:
4505 vim_free(fname_exp);
4506 return retval;
4507}
4508
4509#if defined(FEAT_EVAL) || defined(PROTO)
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00004510
Bram Moolenaar071d4272004-06-13 20:20:40 +00004511/*
4512 * ":scriptnames"
4513 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004514 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004515ex_scriptnames(exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004516{
4517 int i;
4518
Bram Moolenaar05159a02005-02-26 23:04:13 +00004519 for (i = 1; i <= script_items.ga_len && !got_int; ++i)
4520 if (SCRIPT_ITEM(i).sn_name != NULL)
Bram Moolenaard58ea072011-06-26 04:25:30 +02004521 {
4522 home_replace(NULL, SCRIPT_ITEM(i).sn_name,
4523 NameBuff, MAXPATHL, TRUE);
4524 smsg((char_u *)"%3d: %s", i, NameBuff);
Bram Moolenaar970a1b82012-03-23 18:39:18 +01004525 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004526}
4527
4528# if defined(BACKSLASH_IN_FILENAME) || defined(PROTO)
4529/*
4530 * Fix slashes in the list of script names for 'shellslash'.
4531 */
4532 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004533scriptnames_slash_adjust(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004534{
4535 int i;
4536
Bram Moolenaar05159a02005-02-26 23:04:13 +00004537 for (i = 1; i <= script_items.ga_len; ++i)
4538 if (SCRIPT_ITEM(i).sn_name != NULL)
4539 slash_adjust(SCRIPT_ITEM(i).sn_name);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004540}
4541# endif
4542
4543/*
4544 * Get a pointer to a script name. Used for ":verbose set".
4545 */
4546 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004547get_scriptname(scid_T id)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004548{
4549 if (id == SID_MODELINE)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00004550 return (char_u *)_("modeline");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004551 if (id == SID_CMDARG)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00004552 return (char_u *)_("--cmd argument");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004553 if (id == SID_CARG)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00004554 return (char_u *)_("-c argument");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004555 if (id == SID_ENV)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00004556 return (char_u *)_("environment variable");
4557 if (id == SID_ERROR)
4558 return (char_u *)_("error handler");
Bram Moolenaar05159a02005-02-26 23:04:13 +00004559 return SCRIPT_ITEM(id).sn_name;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004560}
Bram Moolenaar05159a02005-02-26 23:04:13 +00004561
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00004562# if defined(EXITFREE) || defined(PROTO)
4563 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004564free_scriptnames(void)
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00004565{
4566 int i;
4567
4568 for (i = script_items.ga_len; i > 0; --i)
4569 vim_free(SCRIPT_ITEM(i).sn_name);
4570 ga_clear(&script_items);
4571}
4572# endif
4573
Bram Moolenaar071d4272004-06-13 20:20:40 +00004574#endif
4575
4576#if defined(USE_CR) || defined(PROTO)
4577
4578# if defined(__MSL__) && (__MSL__ >= 22)
4579/*
4580 * Newer version of the Metrowerks library handle DOS and UNIX files
4581 * without help.
4582 * Test with earlier versions, MSL 2.2 is the library supplied with
4583 * Codewarrior Pro 2.
4584 */
4585 char *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004586fgets_cr(char *s, int n, FILE *stream)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004587{
4588 return fgets(s, n, stream);
4589}
4590# else
4591/*
4592 * Version of fgets() which also works for lines ending in a <CR> only
4593 * (Macintosh format).
4594 * For older versions of the Metrowerks library.
4595 * At least CodeWarrior 9 needed this code.
4596 */
4597 char *
Bram Moolenaard14e00e2016-01-31 17:30:51 +01004598fgets_cr(char *s, int n, FILE *stream)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004599{
4600 int c = 0;
4601 int char_read = 0;
4602
4603 while (!feof(stream) && c != '\r' && c != '\n' && char_read < n - 1)
4604 {
4605 c = fgetc(stream);
4606 s[char_read++] = c;
4607 /* If the file is in DOS format, we need to skip a NL after a CR. I
4608 * thought it was the other way around, but this appears to work... */
4609 if (c == '\n')
4610 {
4611 c = fgetc(stream);
4612 if (c != '\r')
4613 ungetc(c, stream);
4614 }
4615 }
4616
4617 s[char_read] = 0;
4618 if (char_read == 0)
4619 return NULL;
4620
4621 if (feof(stream) && char_read == 1)
4622 return NULL;
4623
4624 return s;
4625}
4626# endif
4627#endif
4628
4629/*
4630 * Get one full line from a sourced file.
4631 * Called by do_cmdline() when it's called from do_source().
4632 *
4633 * Return a pointer to the line in allocated memory.
4634 * Return NULL for end-of-file or some error.
4635 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004636 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004637getsourceline(int c UNUSED, void *cookie, int indent UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004638{
4639 struct source_cookie *sp = (struct source_cookie *)cookie;
4640 char_u *line;
Bram Moolenaarc047b9a2012-02-11 20:40:55 +01004641 char_u *p;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004642
4643#ifdef FEAT_EVAL
4644 /* If breakpoints have been added/deleted need to check for it. */
4645 if (sp->dbg_tick < debug_tick)
4646 {
4647 sp->breakpoint = dbg_find_breakpoint(TRUE, sp->fname, sourcing_lnum);
4648 sp->dbg_tick = debug_tick;
4649 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00004650# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004651 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004652 script_line_end();
4653# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004654#endif
4655 /*
4656 * Get current line. If there is a read-ahead line, use it, otherwise get
4657 * one now.
4658 */
4659 if (sp->finished)
4660 line = NULL;
4661 else if (sp->nextline == NULL)
4662 line = get_one_sourceline(sp);
4663 else
4664 {
4665 line = sp->nextline;
4666 sp->nextline = NULL;
4667 ++sourcing_lnum;
4668 }
Bram Moolenaare2cc9702005-03-15 22:43:58 +00004669#ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004670 if (line != NULL && do_profiling == PROF_YES)
Bram Moolenaare2cc9702005-03-15 22:43:58 +00004671 script_line_start();
4672#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004673
4674 /* Only concatenate lines starting with a \ when 'cpoptions' doesn't
4675 * contain the 'C' flag. */
4676 if (line != NULL && (vim_strchr(p_cpo, CPO_CONCAT) == NULL))
4677 {
4678 /* compensate for the one line read-ahead */
4679 --sourcing_lnum;
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004680
4681 /* Get the next line and concatenate it when it starts with a
4682 * backslash. We always need to read the next line, keep it in
4683 * sp->nextline. */
4684 sp->nextline = get_one_sourceline(sp);
4685 if (sp->nextline != NULL && *(p = skipwhite(sp->nextline)) == '\\')
Bram Moolenaar071d4272004-06-13 20:20:40 +00004686 {
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004687 garray_T ga;
4688
Bram Moolenaarb549a732012-02-22 18:29:33 +01004689 ga_init2(&ga, (int)sizeof(char_u), 400);
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004690 ga_concat(&ga, line);
4691 ga_concat(&ga, p + 1);
4692 for (;;)
4693 {
4694 vim_free(sp->nextline);
4695 sp->nextline = get_one_sourceline(sp);
4696 if (sp->nextline == NULL)
4697 break;
4698 p = skipwhite(sp->nextline);
4699 if (*p != '\\')
4700 break;
Bram Moolenaarb549a732012-02-22 18:29:33 +01004701 /* Adjust the growsize to the current length to speed up
4702 * concatenating many lines. */
4703 if (ga.ga_len > 400)
4704 {
4705 if (ga.ga_len > 8000)
4706 ga.ga_growsize = 8000;
4707 else
4708 ga.ga_growsize = ga.ga_len;
4709 }
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004710 ga_concat(&ga, p + 1);
4711 }
4712 ga_append(&ga, NUL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004713 vim_free(line);
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004714 line = ga.ga_data;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004715 }
4716 }
4717
4718#ifdef FEAT_MBYTE
4719 if (line != NULL && sp->conv.vc_type != CONV_NONE)
4720 {
Bram Moolenaarc047b9a2012-02-11 20:40:55 +01004721 char_u *s;
4722
Bram Moolenaar071d4272004-06-13 20:20:40 +00004723 /* Convert the encoding of the script line. */
4724 s = string_convert(&sp->conv, line, NULL);
4725 if (s != NULL)
4726 {
4727 vim_free(line);
4728 line = s;
4729 }
4730 }
4731#endif
4732
4733#ifdef FEAT_EVAL
4734 /* Did we encounter a breakpoint? */
4735 if (sp->breakpoint != 0 && sp->breakpoint <= sourcing_lnum)
4736 {
4737 dbg_breakpoint(sp->fname, sourcing_lnum);
4738 /* Find next breakpoint. */
4739 sp->breakpoint = dbg_find_breakpoint(TRUE, sp->fname, sourcing_lnum);
4740 sp->dbg_tick = debug_tick;
4741 }
4742#endif
4743
4744 return line;
4745}
4746
4747 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004748get_one_sourceline(struct source_cookie *sp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004749{
4750 garray_T ga;
4751 int len;
4752 int c;
4753 char_u *buf;
4754#ifdef USE_CRNL
4755 int has_cr; /* CR-LF found */
4756#endif
4757#ifdef USE_CR
4758 char_u *scan;
4759#endif
4760 int have_read = FALSE;
4761
4762 /* use a growarray to store the sourced line */
Bram Moolenaar6ac54292005-02-02 23:07:25 +00004763 ga_init2(&ga, 1, 250);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004764
4765 /*
4766 * Loop until there is a finished line (or end-of-file).
4767 */
4768 sourcing_lnum++;
4769 for (;;)
4770 {
Bram Moolenaar6ac54292005-02-02 23:07:25 +00004771 /* make room to read at least 120 (more) characters */
4772 if (ga_grow(&ga, 120) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004773 break;
4774 buf = (char_u *)ga.ga_data;
4775
4776#ifdef USE_CR
4777 if (sp->fileformat == EOL_MAC)
4778 {
Bram Moolenaar86b68352004-12-27 21:59:20 +00004779 if (fgets_cr((char *)buf + ga.ga_len, ga.ga_maxlen - ga.ga_len,
4780 sp->fp) == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004781 break;
4782 }
4783 else
4784#endif
Bram Moolenaar86b68352004-12-27 21:59:20 +00004785 if (fgets((char *)buf + ga.ga_len, ga.ga_maxlen - ga.ga_len,
4786 sp->fp) == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004787 break;
Bram Moolenaar6ac54292005-02-02 23:07:25 +00004788 len = ga.ga_len + (int)STRLEN(buf + ga.ga_len);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004789#ifdef USE_CRNL
4790 /* Ignore a trailing CTRL-Z, when in Dos mode. Only recognize the
4791 * CTRL-Z by its own, or after a NL. */
4792 if ( (len == 1 || (len >= 2 && buf[len - 2] == '\n'))
4793 && sp->fileformat == EOL_DOS
4794 && buf[len - 1] == Ctrl_Z)
4795 {
4796 buf[len - 1] = NUL;
4797 break;
4798 }
4799#endif
4800
4801#ifdef USE_CR
4802 /* If the read doesn't stop on a new line, and there's
4803 * some CR then we assume a Mac format */
4804 if (sp->fileformat == EOL_UNKNOWN)
4805 {
4806 if (buf[len - 1] != '\n' && vim_strchr(buf, '\r') != NULL)
4807 sp->fileformat = EOL_MAC;
4808 else
4809 sp->fileformat = EOL_UNIX;
4810 }
4811
4812 if (sp->fileformat == EOL_MAC)
4813 {
4814 scan = vim_strchr(buf, '\r');
4815
4816 if (scan != NULL)
4817 {
4818 *scan = '\n';
4819 if (*(scan + 1) != 0)
4820 {
4821 *(scan + 1) = 0;
4822 fseek(sp->fp, (long)(scan - buf - len + 1), SEEK_CUR);
4823 }
4824 }
4825 len = STRLEN(buf);
4826 }
4827#endif
4828
4829 have_read = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004830 ga.ga_len = len;
4831
4832 /* If the line was longer than the buffer, read more. */
Bram Moolenaar86b68352004-12-27 21:59:20 +00004833 if (ga.ga_maxlen - ga.ga_len == 1 && buf[len - 1] != '\n')
Bram Moolenaar071d4272004-06-13 20:20:40 +00004834 continue;
4835
4836 if (len >= 1 && buf[len - 1] == '\n') /* remove trailing NL */
4837 {
4838#ifdef USE_CRNL
4839 has_cr = (len >= 2 && buf[len - 2] == '\r');
4840 if (sp->fileformat == EOL_UNKNOWN)
4841 {
4842 if (has_cr)
4843 sp->fileformat = EOL_DOS;
4844 else
4845 sp->fileformat = EOL_UNIX;
4846 }
4847
4848 if (sp->fileformat == EOL_DOS)
4849 {
4850 if (has_cr) /* replace trailing CR */
4851 {
4852 buf[len - 2] = '\n';
4853 --len;
4854 --ga.ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004855 }
4856 else /* lines like ":map xx yy^M" will have failed */
4857 {
4858 if (!sp->error)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00004859 {
Bram Moolenaar8820b482017-03-16 17:23:31 +01004860 msg_source(HL_ATTR(HLF_W));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004861 EMSG(_("W15: Warning: Wrong line separator, ^M may be missing"));
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00004862 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004863 sp->error = TRUE;
4864 sp->fileformat = EOL_UNIX;
4865 }
4866 }
4867#endif
4868 /* The '\n' is escaped if there is an odd number of ^V's just
4869 * before it, first set "c" just before the 'V's and then check
4870 * len&c parities (is faster than ((len-c)%2 == 0)) -- Acevedo */
4871 for (c = len - 2; c >= 0 && buf[c] == Ctrl_V; c--)
4872 ;
4873 if ((len & 1) != (c & 1)) /* escaped NL, read more */
4874 {
4875 sourcing_lnum++;
4876 continue;
4877 }
4878
4879 buf[len - 1] = NUL; /* remove the NL */
4880 }
4881
4882 /*
4883 * Check for ^C here now and then, so recursive :so can be broken.
4884 */
4885 line_breakcheck();
4886 break;
4887 }
4888
4889 if (have_read)
4890 return (char_u *)ga.ga_data;
4891
4892 vim_free(ga.ga_data);
4893 return NULL;
4894}
4895
Bram Moolenaar05159a02005-02-26 23:04:13 +00004896#if defined(FEAT_PROFILE) || defined(PROTO)
4897/*
4898 * Called when starting to read a script line.
4899 * "sourcing_lnum" must be correct!
4900 * When skipping lines it may not actually be executed, but we won't find out
4901 * until later and we need to store the time now.
4902 */
4903 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004904script_line_start(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004905{
4906 scriptitem_T *si;
4907 sn_prl_T *pp;
4908
4909 if (current_SID <= 0 || current_SID > script_items.ga_len)
4910 return;
4911 si = &SCRIPT_ITEM(current_SID);
4912 if (si->sn_prof_on && sourcing_lnum >= 1)
4913 {
Bram Moolenaar8c8de832008-06-24 22:58:06 +00004914 /* Grow the array before starting the timer, so that the time spent
Bram Moolenaar05159a02005-02-26 23:04:13 +00004915 * here isn't counted. */
Bram Moolenaar67435d92017-10-19 21:04:37 +02004916 (void)ga_grow(&si->sn_prl_ga,
4917 (int)(sourcing_lnum - si->sn_prl_ga.ga_len));
Bram Moolenaar05159a02005-02-26 23:04:13 +00004918 si->sn_prl_idx = sourcing_lnum - 1;
4919 while (si->sn_prl_ga.ga_len <= si->sn_prl_idx
4920 && si->sn_prl_ga.ga_len < si->sn_prl_ga.ga_maxlen)
4921 {
4922 /* Zero counters for a line that was not used before. */
4923 pp = &PRL_ITEM(si, si->sn_prl_ga.ga_len);
4924 pp->snp_count = 0;
4925 profile_zero(&pp->sn_prl_total);
4926 profile_zero(&pp->sn_prl_self);
4927 ++si->sn_prl_ga.ga_len;
4928 }
4929 si->sn_prl_execed = FALSE;
4930 profile_start(&si->sn_prl_start);
4931 profile_zero(&si->sn_prl_children);
4932 profile_get_wait(&si->sn_prl_wait);
4933 }
4934}
4935
4936/*
4937 * Called when actually executing a function line.
4938 */
4939 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004940script_line_exec(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004941{
4942 scriptitem_T *si;
4943
4944 if (current_SID <= 0 || current_SID > script_items.ga_len)
4945 return;
4946 si = &SCRIPT_ITEM(current_SID);
4947 if (si->sn_prof_on && si->sn_prl_idx >= 0)
4948 si->sn_prl_execed = TRUE;
4949}
4950
4951/*
Bram Moolenaar67435d92017-10-19 21:04:37 +02004952 * Called when done with a script line.
Bram Moolenaar05159a02005-02-26 23:04:13 +00004953 */
4954 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004955script_line_end(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004956{
4957 scriptitem_T *si;
4958 sn_prl_T *pp;
4959
4960 if (current_SID <= 0 || current_SID > script_items.ga_len)
4961 return;
4962 si = &SCRIPT_ITEM(current_SID);
4963 if (si->sn_prof_on && si->sn_prl_idx >= 0
4964 && si->sn_prl_idx < si->sn_prl_ga.ga_len)
4965 {
4966 if (si->sn_prl_execed)
4967 {
4968 pp = &PRL_ITEM(si, si->sn_prl_idx);
4969 ++pp->snp_count;
4970 profile_end(&si->sn_prl_start);
4971 profile_sub_wait(&si->sn_prl_wait, &si->sn_prl_start);
Bram Moolenaar05159a02005-02-26 23:04:13 +00004972 profile_add(&pp->sn_prl_total, &si->sn_prl_start);
Bram Moolenaar1056d982006-03-09 22:37:52 +00004973 profile_self(&pp->sn_prl_self, &si->sn_prl_start,
4974 &si->sn_prl_children);
Bram Moolenaar05159a02005-02-26 23:04:13 +00004975 }
4976 si->sn_prl_idx = -1;
4977 }
4978}
4979#endif
4980
Bram Moolenaar071d4272004-06-13 20:20:40 +00004981/*
4982 * ":scriptencoding": Set encoding conversion for a sourced script.
4983 * Without the multi-byte feature it's simply ignored.
4984 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004985 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004986ex_scriptencoding(exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004987{
4988#ifdef FEAT_MBYTE
4989 struct source_cookie *sp;
4990 char_u *name;
4991
4992 if (!getline_equal(eap->getline, eap->cookie, getsourceline))
4993 {
4994 EMSG(_("E167: :scriptencoding used outside of a sourced file"));
4995 return;
4996 }
4997
4998 if (*eap->arg != NUL)
4999 {
5000 name = enc_canonize(eap->arg);
5001 if (name == NULL) /* out of memory */
5002 return;
5003 }
5004 else
5005 name = eap->arg;
5006
5007 /* Setup for conversion from the specified encoding to 'encoding'. */
5008 sp = (struct source_cookie *)getline_cookie(eap->getline, eap->cookie);
5009 convert_setup(&sp->conv, name, p_enc);
5010
5011 if (name != eap->arg)
5012 vim_free(name);
5013#endif
5014}
5015
5016#if defined(FEAT_EVAL) || defined(PROTO)
5017/*
5018 * ":finish": Mark a sourced file as finished.
5019 */
5020 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005021ex_finish(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005022{
5023 if (getline_equal(eap->getline, eap->cookie, getsourceline))
5024 do_finish(eap, FALSE);
5025 else
5026 EMSG(_("E168: :finish used outside of a sourced file"));
5027}
5028
5029/*
5030 * Mark a sourced file as finished. Possibly makes the ":finish" pending.
5031 * Also called for a pending finish at the ":endtry" or after returning from
5032 * an extra do_cmdline(). "reanimate" is used in the latter case.
5033 */
5034 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005035do_finish(exarg_T *eap, int reanimate)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005036{
5037 int idx;
5038
5039 if (reanimate)
5040 ((struct source_cookie *)getline_cookie(eap->getline,
5041 eap->cookie))->finished = FALSE;
5042
5043 /*
5044 * Cleanup (and inactivate) conditionals, but stop when a try conditional
5045 * not in its finally clause (which then is to be executed next) is found.
5046 * In this case, make the ":finish" pending for execution at the ":endtry".
5047 * Otherwise, finish normally.
5048 */
5049 idx = cleanup_conditionals(eap->cstack, 0, TRUE);
5050 if (idx >= 0)
5051 {
5052 eap->cstack->cs_pending[idx] = CSTP_FINISH;
5053 report_make_pending(CSTP_FINISH, NULL);
5054 }
5055 else
5056 ((struct source_cookie *)getline_cookie(eap->getline,
5057 eap->cookie))->finished = TRUE;
5058}
5059
5060
5061/*
5062 * Return TRUE when a sourced file had the ":finish" command: Don't give error
5063 * message for missing ":endif".
5064 * Return FALSE when not sourcing a file.
5065 */
5066 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005067source_finished(
5068 char_u *(*fgetline)(int, void *, int),
5069 void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005070{
Bram Moolenaar89d40322006-08-29 15:30:07 +00005071 return (getline_equal(fgetline, cookie, getsourceline)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005072 && ((struct source_cookie *)getline_cookie(
Bram Moolenaar89d40322006-08-29 15:30:07 +00005073 fgetline, cookie))->finished);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005074}
5075#endif
5076
5077#if defined(FEAT_LISTCMDS) || defined(PROTO)
5078/*
5079 * ":checktime [buffer]"
5080 */
5081 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005082ex_checktime(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005083{
5084 buf_T *buf;
5085 int save_no_check_timestamps = no_check_timestamps;
5086
5087 no_check_timestamps = 0;
5088 if (eap->addr_count == 0) /* default is all buffers */
5089 check_timestamps(FALSE);
5090 else
5091 {
5092 buf = buflist_findnr((int)eap->line2);
5093 if (buf != NULL) /* cannot happen? */
5094 (void)buf_check_timestamp(buf, FALSE);
5095 }
5096 no_check_timestamps = save_no_check_timestamps;
5097}
5098#endif
5099
Bram Moolenaar071d4272004-06-13 20:20:40 +00005100#if (defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
5101 && (defined(FEAT_EVAL) || defined(FEAT_MULTI_LANG))
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02005102# define HAVE_GET_LOCALE_VAL
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005103static char_u *get_locale_val(int what);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005104
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005105 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005106get_locale_val(int what)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005107{
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005108 char_u *loc;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005109
Bram Moolenaar48e330a2016-02-23 14:53:34 +01005110 /* Obtain the locale value from the libraries. */
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005111 loc = (char_u *)setlocale(what, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005112
Bram Moolenaar61660ea2006-04-07 21:40:07 +00005113# ifdef WIN32
Bram Moolenaar071d4272004-06-13 20:20:40 +00005114 if (loc != NULL)
5115 {
5116 char_u *p;
5117
Bram Moolenaar61660ea2006-04-07 21:40:07 +00005118 /* setocale() returns something like "LC_COLLATE=<name>;LC_..." when
5119 * one of the values (e.g., LC_CTYPE) differs. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005120 p = vim_strchr(loc, '=');
5121 if (p != NULL)
5122 {
5123 loc = ++p;
5124 while (*p != NUL) /* remove trailing newline */
5125 {
Bram Moolenaar61660ea2006-04-07 21:40:07 +00005126 if (*p < ' ' || *p == ';')
Bram Moolenaar071d4272004-06-13 20:20:40 +00005127 {
5128 *p = NUL;
5129 break;
5130 }
5131 ++p;
5132 }
5133 }
5134 }
5135# endif
5136
5137 return loc;
5138}
5139#endif
5140
5141
5142#ifdef WIN32
5143/*
5144 * On MS-Windows locale names are strings like "German_Germany.1252", but
5145 * gettext expects "de". Try to translate one into another here for a few
5146 * supported languages.
5147 */
5148 static char_u *
5149gettext_lang(char_u *name)
5150{
5151 int i;
5152 static char *(mtable[]) = {
5153 "afrikaans", "af",
5154 "czech", "cs",
5155 "dutch", "nl",
5156 "german", "de",
5157 "english_united kingdom", "en_GB",
5158 "spanish", "es",
5159 "french", "fr",
5160 "italian", "it",
5161 "japanese", "ja",
5162 "korean", "ko",
5163 "norwegian", "no",
5164 "polish", "pl",
5165 "russian", "ru",
5166 "slovak", "sk",
5167 "swedish", "sv",
5168 "ukrainian", "uk",
5169 "chinese_china", "zh_CN",
5170 "chinese_taiwan", "zh_TW",
5171 NULL};
5172
5173 for (i = 0; mtable[i] != NULL; i += 2)
5174 if (STRNICMP(mtable[i], name, STRLEN(mtable[i])) == 0)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005175 return (char_u *)mtable[i + 1];
Bram Moolenaar071d4272004-06-13 20:20:40 +00005176 return name;
5177}
5178#endif
5179
5180#if defined(FEAT_MULTI_LANG) || defined(PROTO)
5181/*
5182 * Obtain the current messages language. Used to set the default for
5183 * 'helplang'. May return NULL or an empty string.
5184 */
5185 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005186get_mess_lang(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005187{
5188 char_u *p;
5189
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02005190# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00005191# if defined(LC_MESSAGES)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005192 p = get_locale_val(LC_MESSAGES);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005193# else
5194 /* This is necessary for Win32, where LC_MESSAGES is not defined and $LANG
Bram Moolenaar61660ea2006-04-07 21:40:07 +00005195 * may be set to the LCID number. LC_COLLATE is the best guess, LC_TIME
5196 * and LC_MONETARY may be set differently for a Japanese working in the
5197 * US. */
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005198 p = get_locale_val(LC_COLLATE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005199# endif
5200# else
5201 p = mch_getenv((char_u *)"LC_ALL");
5202 if (p == NULL || *p == NUL)
5203 {
5204 p = mch_getenv((char_u *)"LC_MESSAGES");
5205 if (p == NULL || *p == NUL)
5206 p = mch_getenv((char_u *)"LANG");
5207 }
5208# endif
5209# ifdef WIN32
5210 p = gettext_lang(p);
5211# endif
5212 return p;
5213}
5214#endif
5215
Bram Moolenaardef9e822004-12-31 20:58:58 +00005216/* Complicated #if; matches with where get_mess_env() is used below. */
5217#if (defined(FEAT_EVAL) && !((defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
5218 && defined(LC_MESSAGES))) \
5219 || ((defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
5220 && (defined(FEAT_GETTEXT) || defined(FEAT_MBYTE)) \
5221 && !defined(LC_MESSAGES))
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01005222static char_u *get_mess_env(void);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005223
5224/*
5225 * Get the language used for messages from the environment.
5226 */
5227 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005228get_mess_env(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005229{
5230 char_u *p;
5231
5232 p = mch_getenv((char_u *)"LC_ALL");
5233 if (p == NULL || *p == NUL)
5234 {
5235 p = mch_getenv((char_u *)"LC_MESSAGES");
5236 if (p == NULL || *p == NUL)
5237 {
5238 p = mch_getenv((char_u *)"LANG");
5239 if (p != NULL && VIM_ISDIGIT(*p))
5240 p = NULL; /* ignore something like "1043" */
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02005241# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00005242 if (p == NULL || *p == NUL)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005243 p = get_locale_val(LC_CTYPE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005244# endif
5245 }
5246 }
5247 return p;
5248}
5249#endif
5250
5251#if defined(FEAT_EVAL) || defined(PROTO)
5252
5253/*
5254 * Set the "v:lang" variable according to the current locale setting.
5255 * Also do "v:lc_time"and "v:ctype".
5256 */
5257 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005258set_lang_var(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005259{
5260 char_u *loc;
5261
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02005262# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005263 loc = get_locale_val(LC_CTYPE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005264# else
5265 /* setlocale() not supported: use the default value */
5266 loc = (char_u *)"C";
5267# endif
5268 set_vim_var_string(VV_CTYPE, loc, -1);
5269
5270 /* When LC_MESSAGES isn't defined use the value from $LC_MESSAGES, fall
5271 * back to LC_CTYPE if it's empty. */
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02005272# if defined(HAVE_GET_LOCALE_VAL) && defined(LC_MESSAGES)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005273 loc = get_locale_val(LC_MESSAGES);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005274# else
5275 loc = get_mess_env();
5276# endif
5277 set_vim_var_string(VV_LANG, loc, -1);
5278
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02005279# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005280 loc = get_locale_val(LC_TIME);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005281# endif
5282 set_vim_var_string(VV_LC_TIME, loc, -1);
5283}
5284#endif
5285
5286#if (defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
5287 && (defined(FEAT_GETTEXT) || defined(FEAT_MBYTE))
5288/*
5289 * ":language": Set the language (locale).
5290 */
5291 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005292ex_language(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005293{
5294 char *loc;
5295 char_u *p;
5296 char_u *name;
5297 int what = LC_ALL;
5298 char *whatstr = "";
5299#ifdef LC_MESSAGES
5300# define VIM_LC_MESSAGES LC_MESSAGES
5301#else
5302# define VIM_LC_MESSAGES 6789
5303#endif
5304
5305 name = eap->arg;
5306
5307 /* Check for "messages {name}", "ctype {name}" or "time {name}" argument.
5308 * Allow abbreviation, but require at least 3 characters to avoid
5309 * confusion with a two letter language name "me" or "ct". */
5310 p = skiptowhite(eap->arg);
Bram Moolenaar1c465442017-03-12 20:10:05 +01005311 if ((*p == NUL || VIM_ISWHITE(*p)) && p - eap->arg >= 3)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005312 {
5313 if (STRNICMP(eap->arg, "messages", p - eap->arg) == 0)
5314 {
5315 what = VIM_LC_MESSAGES;
5316 name = skipwhite(p);
5317 whatstr = "messages ";
5318 }
5319 else if (STRNICMP(eap->arg, "ctype", p - eap->arg) == 0)
5320 {
5321 what = LC_CTYPE;
5322 name = skipwhite(p);
5323 whatstr = "ctype ";
5324 }
5325 else if (STRNICMP(eap->arg, "time", p - eap->arg) == 0)
5326 {
5327 what = LC_TIME;
5328 name = skipwhite(p);
5329 whatstr = "time ";
5330 }
5331 }
5332
5333 if (*name == NUL)
5334 {
5335#ifndef LC_MESSAGES
5336 if (what == VIM_LC_MESSAGES)
5337 p = get_mess_env();
5338 else
5339#endif
5340 p = (char_u *)setlocale(what, NULL);
5341 if (p == NULL || *p == NUL)
5342 p = (char_u *)"Unknown";
5343 smsg((char_u *)_("Current %slanguage: \"%s\""), whatstr, p);
5344 }
5345 else
5346 {
5347#ifndef LC_MESSAGES
5348 if (what == VIM_LC_MESSAGES)
5349 loc = "";
5350 else
5351#endif
Bram Moolenaar8c8de832008-06-24 22:58:06 +00005352 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00005353 loc = setlocale(what, (char *)name);
Bram Moolenaar8c8de832008-06-24 22:58:06 +00005354#if defined(FEAT_FLOAT) && defined(LC_NUMERIC)
5355 /* Make sure strtod() uses a decimal point, not a comma. */
5356 setlocale(LC_NUMERIC, "C");
5357#endif
5358 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00005359 if (loc == NULL)
5360 EMSG2(_("E197: Cannot set language to \"%s\""), name);
5361 else
5362 {
5363#ifdef HAVE_NL_MSG_CAT_CNTR
5364 /* Need to do this for GNU gettext, otherwise cached translations
5365 * will be used again. */
5366 extern int _nl_msg_cat_cntr;
5367
5368 ++_nl_msg_cat_cntr;
5369#endif
Bram Moolenaarb8017e72007-05-10 18:59:07 +00005370 /* Reset $LC_ALL, otherwise it would overrule everything. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005371 vim_setenv((char_u *)"LC_ALL", (char_u *)"");
5372
5373 if (what != LC_TIME)
5374 {
5375 /* Tell gettext() what to translate to. It apparently doesn't
5376 * use the currently effective locale. Also do this when
5377 * FEAT_GETTEXT isn't defined, so that shell commands use this
5378 * value. */
5379 if (what == LC_ALL)
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00005380 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00005381 vim_setenv((char_u *)"LANG", name);
Bram Moolenaare3a0b532013-06-28 20:36:30 +02005382
5383 /* Clear $LANGUAGE because GNU gettext uses it. */
5384 vim_setenv((char_u *)"LANGUAGE", (char_u *)"");
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00005385# ifdef WIN32
5386 /* Apparently MS-Windows printf() may cause a crash when
5387 * we give it 8-bit text while it's expecting text in the
5388 * current locale. This call avoids that. */
5389 setlocale(LC_CTYPE, "C");
5390# endif
5391 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00005392 if (what != LC_CTYPE)
5393 {
5394 char_u *mname;
5395#ifdef WIN32
5396 mname = gettext_lang(name);
5397#else
5398 mname = name;
5399#endif
5400 vim_setenv((char_u *)"LC_MESSAGES", mname);
5401#ifdef FEAT_MULTI_LANG
5402 set_helplang_default(mname);
5403#endif
5404 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00005405 }
5406
5407# ifdef FEAT_EVAL
5408 /* Set v:lang, v:lc_time and v:ctype to the final result. */
5409 set_lang_var();
5410# endif
Bram Moolenaar6cc00c72011-10-20 21:41:09 +02005411# ifdef FEAT_TITLE
5412 maketitle();
5413# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00005414 }
5415 }
5416}
5417
5418# if defined(FEAT_CMDL_COMPL) || defined(PROTO)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005419
5420static char_u **locales = NULL; /* Array of all available locales */
Bram Moolenaarb8f7bd62017-01-12 20:28:25 +01005421
5422# ifndef WIN32
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005423static int did_init_locales = FALSE;
5424
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005425/* Return an array of strings for all available locales + NULL for the
5426 * last element. Return NULL in case of error. */
5427 static char_u **
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005428find_locales(void)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005429{
5430 garray_T locales_ga;
5431 char_u *loc;
5432
5433 /* Find all available locales by running command "locale -a". If this
5434 * doesn't work we won't have completion. */
5435 char_u *locale_a = get_cmd_output((char_u *)"locale -a",
Bram Moolenaar39c29ed2014-04-05 19:44:40 +02005436 NULL, SHELL_SILENT, NULL);
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005437 if (locale_a == NULL)
5438 return NULL;
5439 ga_init2(&locales_ga, sizeof(char_u *), 20);
5440
5441 /* Transform locale_a string where each locale is separated by "\n"
5442 * into an array of locale strings. */
5443 loc = (char_u *)strtok((char *)locale_a, "\n");
5444
5445 while (loc != NULL)
5446 {
5447 if (ga_grow(&locales_ga, 1) == FAIL)
5448 break;
5449 loc = vim_strsave(loc);
5450 if (loc == NULL)
5451 break;
5452
5453 ((char_u **)locales_ga.ga_data)[locales_ga.ga_len++] = loc;
5454 loc = (char_u *)strtok(NULL, "\n");
5455 }
5456 vim_free(locale_a);
5457 if (ga_grow(&locales_ga, 1) == FAIL)
5458 {
5459 ga_clear(&locales_ga);
5460 return NULL;
5461 }
5462 ((char_u **)locales_ga.ga_data)[locales_ga.ga_len] = NULL;
5463 return (char_u **)locales_ga.ga_data;
5464}
Bram Moolenaarb8f7bd62017-01-12 20:28:25 +01005465# endif
5466
5467/*
5468 * Lazy initialization of all available locales.
5469 */
5470 static void
5471init_locales(void)
5472{
5473# ifndef WIN32
5474 if (!did_init_locales)
5475 {
5476 did_init_locales = TRUE;
5477 locales = find_locales();
5478 }
5479# endif
5480}
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005481
5482# if defined(EXITFREE) || defined(PROTO)
5483 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005484free_locales(void)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005485{
5486 int i;
5487 if (locales != NULL)
5488 {
5489 for (i = 0; locales[i] != NULL; i++)
5490 vim_free(locales[i]);
Bram Moolenaard23a8232018-02-10 18:45:26 +01005491 VIM_CLEAR(locales);
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005492 }
5493}
5494# endif
5495
Bram Moolenaar071d4272004-06-13 20:20:40 +00005496/*
5497 * Function given to ExpandGeneric() to obtain the possible arguments of the
5498 * ":language" command.
5499 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005500 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005501get_lang_arg(expand_T *xp UNUSED, int idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005502{
5503 if (idx == 0)
5504 return (char_u *)"messages";
5505 if (idx == 1)
5506 return (char_u *)"ctype";
5507 if (idx == 2)
5508 return (char_u *)"time";
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005509
5510 init_locales();
5511 if (locales == NULL)
5512 return NULL;
5513 return locales[idx - 3];
5514}
5515
5516/*
5517 * Function given to ExpandGeneric() to obtain the available locales.
5518 */
5519 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005520get_locales(expand_T *xp UNUSED, int idx)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005521{
5522 init_locales();
5523 if (locales == NULL)
5524 return NULL;
5525 return locales[idx];
Bram Moolenaar071d4272004-06-13 20:20:40 +00005526}
5527# endif
5528
5529#endif