blob: 73fe0194beb00d7963e9c89e1a138ab16085c436 [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;
1301
1302 if (balloonEval == NULL)
1303 {
1304 balloonEval = (BalloonEval *)alloc(sizeof(BalloonEval));
1305 balloonEvalForTerm = TRUE;
1306 }
1307 if (balloonEval != NULL)
1308 general_beval_cb(balloonEval, 0);
1309 }
1310 else if (this_due > 0 && (next_due == -1 || next_due > this_due))
1311 next_due = this_due;
1312 }
1313#endif
1314
Bram Moolenaar75537a92016-09-05 22:45:28 +02001315 return current_id != last_timer_id ? 1 : next_due;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001316}
1317
1318/*
1319 * Find a timer by ID. Returns NULL if not found;
1320 */
1321 timer_T *
Bram Moolenaar75537a92016-09-05 22:45:28 +02001322find_timer(long id)
Bram Moolenaar975b5272016-03-15 23:10:59 +01001323{
1324 timer_T *timer;
1325
Bram Moolenaar75537a92016-09-05 22:45:28 +02001326 if (id >= 0)
1327 {
1328 for (timer = first_timer; timer != NULL; timer = timer->tr_next)
1329 if (timer->tr_id == id)
1330 return timer;
1331 }
1332 return NULL;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001333}
1334
1335
1336/*
1337 * Stop a timer and delete it.
1338 */
1339 void
1340stop_timer(timer_T *timer)
1341{
Bram Moolenaar75537a92016-09-05 22:45:28 +02001342 if (timer->tr_firing)
1343 /* Free the timer after the callback returns. */
1344 timer->tr_id = -1;
1345 else
1346 {
1347 remove_timer(timer);
1348 free_timer(timer);
1349 }
Bram Moolenaar975b5272016-03-15 23:10:59 +01001350}
Bram Moolenaare3188e22016-05-31 21:13:04 +02001351
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001352 void
Bram Moolenaarb73598e2016-08-07 18:22:53 +02001353stop_all_timers(void)
1354{
Bram Moolenaar75537a92016-09-05 22:45:28 +02001355 timer_T *timer;
1356 timer_T *timer_next;
1357
1358 for (timer = first_timer; timer != NULL; timer = timer_next)
1359 {
1360 timer_next = timer->tr_next;
1361 stop_timer(timer);
1362 }
Bram Moolenaarb73598e2016-08-07 18:22:53 +02001363}
1364
1365 void
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001366add_timer_info(typval_T *rettv, timer_T *timer)
1367{
1368 list_T *list = rettv->vval.v_list;
1369 dict_T *dict = dict_alloc();
1370 dictitem_T *di;
1371 long remaining;
1372 proftime_T now;
1373
1374 if (dict == NULL)
1375 return;
1376 list_append_dict(list, dict);
1377
Bram Moolenaar75537a92016-09-05 22:45:28 +02001378 dict_add_nr_str(dict, "id", timer->tr_id, NULL);
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001379 dict_add_nr_str(dict, "time", (long)timer->tr_interval, NULL);
1380
1381 profile_start(&now);
Bram Moolenaar51b0f372017-11-18 18:52:04 +01001382 remaining = proftime_time_left(&timer->tr_due, &now);
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001383 dict_add_nr_str(dict, "remaining", (long)remaining, NULL);
1384
1385 dict_add_nr_str(dict, "repeat",
1386 (long)(timer->tr_repeat < 0 ? -1 : timer->tr_repeat + 1), NULL);
Bram Moolenaarb73598e2016-08-07 18:22:53 +02001387 dict_add_nr_str(dict, "paused", (long)(timer->tr_paused), NULL);
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001388
1389 di = dictitem_alloc((char_u *)"callback");
1390 if (di != NULL)
1391 {
1392 if (dict_add(dict, di) == FAIL)
1393 vim_free(di);
1394 else if (timer->tr_partial != NULL)
1395 {
1396 di->di_tv.v_type = VAR_PARTIAL;
1397 di->di_tv.vval.v_partial = timer->tr_partial;
1398 ++timer->tr_partial->pt_refcount;
1399 }
1400 else
1401 {
1402 di->di_tv.v_type = VAR_FUNC;
1403 di->di_tv.vval.v_string = vim_strsave(timer->tr_callback);
1404 }
1405 di->di_tv.v_lock = 0;
1406 }
1407}
1408
1409 void
1410add_timer_info_all(typval_T *rettv)
1411{
1412 timer_T *timer;
1413
1414 for (timer = first_timer; timer != NULL; timer = timer->tr_next)
Bram Moolenaar75537a92016-09-05 22:45:28 +02001415 if (timer->tr_id != -1)
1416 add_timer_info(rettv, timer);
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001417}
1418
Bram Moolenaare3188e22016-05-31 21:13:04 +02001419/*
1420 * Mark references in partials of timers.
1421 */
1422 int
1423set_ref_in_timer(int copyID)
1424{
1425 int abort = FALSE;
1426 timer_T *timer;
1427 typval_T tv;
1428
1429 for (timer = first_timer; timer != NULL; timer = timer->tr_next)
1430 {
Bram Moolenaar1e96d9b2016-07-29 22:15:09 +02001431 if (timer->tr_partial != NULL)
1432 {
1433 tv.v_type = VAR_PARTIAL;
1434 tv.vval.v_partial = timer->tr_partial;
1435 }
1436 else
1437 {
1438 tv.v_type = VAR_FUNC;
1439 tv.vval.v_string = timer->tr_callback;
1440 }
Bram Moolenaare3188e22016-05-31 21:13:04 +02001441 abort = abort || set_ref_in_item(&tv, copyID, NULL, NULL);
1442 }
1443 return abort;
1444}
Bram Moolenaar623e2632016-07-30 22:47:56 +02001445
1446# if defined(EXITFREE) || defined(PROTO)
1447 void
1448timer_free_all()
1449{
1450 timer_T *timer;
1451
1452 while (first_timer != NULL)
1453 {
1454 timer = first_timer;
1455 remove_timer(timer);
1456 free_timer(timer);
1457 }
1458}
1459# endif
Bram Moolenaar975b5272016-03-15 23:10:59 +01001460# endif
1461
Bram Moolenaar8a7f5a22013-06-06 14:01:46 +02001462#if defined(FEAT_SYN_HL) && defined(FEAT_RELTIME) && defined(FEAT_FLOAT)
1463# if defined(HAVE_MATH_H)
1464# include <math.h>
1465# endif
1466
1467/*
1468 * Divide the time "tm" by "count" and store in "tm2".
1469 */
1470 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001471profile_divide(proftime_T *tm, int count, proftime_T *tm2)
Bram Moolenaar8a7f5a22013-06-06 14:01:46 +02001472{
1473 if (count == 0)
1474 profile_zero(tm2);
1475 else
1476 {
1477# ifdef WIN3264
1478 tm2->QuadPart = tm->QuadPart / count;
1479# else
1480 double usec = (tm->tv_sec * 1000000.0 + tm->tv_usec) / count;
1481
1482 tm2->tv_sec = floor(usec / 1000000.0);
Bram Moolenaara2e14fc2013-06-10 20:10:44 +02001483 tm2->tv_usec = vim_round(usec - (tm2->tv_sec * 1000000.0));
Bram Moolenaar8a7f5a22013-06-06 14:01:46 +02001484# endif
1485 }
1486}
1487#endif
1488
Bram Moolenaar76929292008-01-06 19:07:36 +00001489# if defined(FEAT_PROFILE) || defined(PROTO)
1490/*
1491 * Functions for profiling.
1492 */
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01001493static void script_do_profile(scriptitem_T *si);
1494static void script_dump_profile(FILE *fd);
Bram Moolenaar76929292008-01-06 19:07:36 +00001495static proftime_T prof_wait_time;
1496
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001497/*
Bram Moolenaar05159a02005-02-26 23:04:13 +00001498 * Add the time "tm2" to "tm".
1499 */
1500 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001501profile_add(proftime_T *tm, proftime_T *tm2)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001502{
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001503# ifdef WIN3264
1504 tm->QuadPart += tm2->QuadPart;
1505# else
Bram Moolenaar05159a02005-02-26 23:04:13 +00001506 tm->tv_usec += tm2->tv_usec;
1507 tm->tv_sec += tm2->tv_sec;
1508 if (tm->tv_usec >= 1000000)
1509 {
1510 tm->tv_usec -= 1000000;
1511 ++tm->tv_sec;
1512 }
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001513# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00001514}
1515
1516/*
Bram Moolenaar1056d982006-03-09 22:37:52 +00001517 * Add the "self" time from the total time and the children's time.
1518 */
1519 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001520profile_self(proftime_T *self, proftime_T *total, proftime_T *children)
Bram Moolenaar1056d982006-03-09 22:37:52 +00001521{
1522 /* Check that the result won't be negative. Can happen with recursive
1523 * calls. */
1524#ifdef WIN3264
1525 if (total->QuadPart <= children->QuadPart)
1526 return;
1527#else
1528 if (total->tv_sec < children->tv_sec
1529 || (total->tv_sec == children->tv_sec
1530 && total->tv_usec <= children->tv_usec))
1531 return;
1532#endif
1533 profile_add(self, total);
1534 profile_sub(self, children);
1535}
1536
1537/*
Bram Moolenaar05159a02005-02-26 23:04:13 +00001538 * Get the current waittime.
1539 */
1540 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001541profile_get_wait(proftime_T *tm)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001542{
1543 *tm = prof_wait_time;
1544}
1545
1546/*
1547 * Subtract the passed waittime since "tm" from "tma".
1548 */
1549 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001550profile_sub_wait(proftime_T *tm, proftime_T *tma)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001551{
1552 proftime_T tm3 = prof_wait_time;
1553
1554 profile_sub(&tm3, tm);
1555 profile_sub(tma, &tm3);
1556}
1557
1558/*
1559 * Return TRUE if "tm1" and "tm2" are equal.
1560 */
1561 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001562profile_equal(proftime_T *tm1, proftime_T *tm2)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001563{
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001564# ifdef WIN3264
1565 return (tm1->QuadPart == tm2->QuadPart);
1566# else
Bram Moolenaar05159a02005-02-26 23:04:13 +00001567 return (tm1->tv_usec == tm2->tv_usec && tm1->tv_sec == tm2->tv_sec);
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001568# endif
1569}
1570
1571/*
1572 * Return <0, 0 or >0 if "tm1" < "tm2", "tm1" == "tm2" or "tm1" > "tm2"
1573 */
1574 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001575profile_cmp(const proftime_T *tm1, const proftime_T *tm2)
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001576{
1577# ifdef WIN3264
1578 return (int)(tm2->QuadPart - tm1->QuadPart);
1579# else
1580 if (tm1->tv_sec == tm2->tv_sec)
1581 return tm2->tv_usec - tm1->tv_usec;
1582 return tm2->tv_sec - tm1->tv_sec;
1583# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00001584}
1585
Bram Moolenaar05159a02005-02-26 23:04:13 +00001586static char_u *profile_fname = NULL;
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001587static proftime_T pause_time;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001588
1589/*
1590 * ":profile cmd args"
1591 */
1592 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001593ex_profile(exarg_T *eap)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001594{
1595 char_u *e;
1596 int len;
1597
1598 e = skiptowhite(eap->arg);
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00001599 len = (int)(e - eap->arg);
Bram Moolenaar05159a02005-02-26 23:04:13 +00001600 e = skipwhite(e);
1601
1602 if (len == 5 && STRNCMP(eap->arg, "start", 5) == 0 && *e != NUL)
1603 {
1604 vim_free(profile_fname);
Bram Moolenaard94682f2015-04-13 15:37:56 +02001605 profile_fname = expand_env_save_opt(e, TRUE);
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001606 do_profiling = PROF_YES;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001607 profile_zero(&prof_wait_time);
1608 set_vim_var_nr(VV_PROFILING, 1L);
1609 }
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001610 else if (do_profiling == PROF_NONE)
Bram Moolenaar54c1b492010-02-24 14:01:28 +01001611 EMSG(_("E750: First use \":profile start {fname}\""));
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001612 else if (STRCMP(eap->arg, "pause") == 0)
1613 {
1614 if (do_profiling == PROF_YES)
1615 profile_start(&pause_time);
1616 do_profiling = PROF_PAUSED;
1617 }
1618 else if (STRCMP(eap->arg, "continue") == 0)
1619 {
1620 if (do_profiling == PROF_PAUSED)
1621 {
1622 profile_end(&pause_time);
1623 profile_add(&prof_wait_time, &pause_time);
1624 }
1625 do_profiling = PROF_YES;
1626 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00001627 else
1628 {
1629 /* The rest is similar to ":breakadd". */
1630 ex_breakadd(eap);
1631 }
1632}
1633
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001634/* Command line expansion for :profile. */
1635static enum
1636{
1637 PEXP_SUBCMD, /* expand :profile sub-commands */
Bram Moolenaar49789dc2011-02-25 14:46:09 +01001638 PEXP_FUNC /* expand :profile func {funcname} */
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001639} pexpand_what;
1640
1641static char *pexpand_cmds[] = {
1642 "start",
1643#define PROFCMD_START 0
1644 "pause",
1645#define PROFCMD_PAUSE 1
1646 "continue",
1647#define PROFCMD_CONTINUE 2
1648 "func",
1649#define PROFCMD_FUNC 3
1650 "file",
1651#define PROFCMD_FILE 4
1652 NULL
1653#define PROFCMD_LAST 5
1654};
1655
1656/*
1657 * Function given to ExpandGeneric() to obtain the profile command
1658 * specific expansion.
1659 */
1660 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001661get_profile_name(expand_T *xp UNUSED, int idx)
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001662{
1663 switch (pexpand_what)
1664 {
1665 case PEXP_SUBCMD:
1666 return (char_u *)pexpand_cmds[idx];
1667 /* case PEXP_FUNC: TODO */
1668 default:
1669 return NULL;
1670 }
1671}
1672
1673/*
1674 * Handle command line completion for :profile command.
1675 */
1676 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001677set_context_in_profile_cmd(expand_T *xp, char_u *arg)
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001678{
1679 char_u *end_subcmd;
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001680
1681 /* Default: expand subcommands. */
1682 xp->xp_context = EXPAND_PROFILE;
1683 pexpand_what = PEXP_SUBCMD;
1684 xp->xp_pattern = arg;
1685
1686 end_subcmd = skiptowhite(arg);
1687 if (*end_subcmd == NUL)
1688 return;
1689
Bram Moolenaar8b9c05f2010-03-02 17:54:33 +01001690 if (end_subcmd - arg == 5 && STRNCMP(arg, "start", 5) == 0)
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001691 {
1692 xp->xp_context = EXPAND_FILES;
1693 xp->xp_pattern = skipwhite(end_subcmd);
1694 return;
1695 }
1696
1697 /* TODO: expand function names after "func" */
1698 xp->xp_context = EXPAND_NOTHING;
1699}
1700
Bram Moolenaar05159a02005-02-26 23:04:13 +00001701/*
1702 * Dump the profiling info.
1703 */
1704 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001705profile_dump(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001706{
1707 FILE *fd;
1708
1709 if (profile_fname != NULL)
1710 {
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +00001711 fd = mch_fopen((char *)profile_fname, "w");
Bram Moolenaar05159a02005-02-26 23:04:13 +00001712 if (fd == NULL)
1713 EMSG2(_(e_notopen), profile_fname);
1714 else
1715 {
Bram Moolenaar05159a02005-02-26 23:04:13 +00001716 script_dump_profile(fd);
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001717 func_dump_profile(fd);
Bram Moolenaar05159a02005-02-26 23:04:13 +00001718 fclose(fd);
1719 }
1720 }
1721}
1722
1723/*
1724 * Start profiling script "fp".
1725 */
1726 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001727script_do_profile(scriptitem_T *si)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001728{
1729 si->sn_pr_count = 0;
1730 profile_zero(&si->sn_pr_total);
1731 profile_zero(&si->sn_pr_self);
1732
1733 ga_init2(&si->sn_prl_ga, sizeof(sn_prl_T), 100);
1734 si->sn_prl_idx = -1;
1735 si->sn_prof_on = TRUE;
1736 si->sn_pr_nest = 0;
1737}
1738
1739/*
Bram Moolenaar67435d92017-10-19 21:04:37 +02001740 * Save time when starting to invoke another script or function.
Bram Moolenaar05159a02005-02-26 23:04:13 +00001741 */
1742 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001743script_prof_save(
1744 proftime_T *tm) /* place to store wait time */
Bram Moolenaar05159a02005-02-26 23:04:13 +00001745{
1746 scriptitem_T *si;
1747
1748 if (current_SID > 0 && current_SID <= script_items.ga_len)
1749 {
1750 si = &SCRIPT_ITEM(current_SID);
1751 if (si->sn_prof_on && si->sn_pr_nest++ == 0)
1752 profile_start(&si->sn_pr_child);
1753 }
1754 profile_get_wait(tm);
1755}
1756
1757/*
1758 * Count time spent in children after invoking another script or function.
1759 */
1760 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001761script_prof_restore(proftime_T *tm)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001762{
1763 scriptitem_T *si;
1764
1765 if (current_SID > 0 && current_SID <= script_items.ga_len)
1766 {
1767 si = &SCRIPT_ITEM(current_SID);
1768 if (si->sn_prof_on && --si->sn_pr_nest == 0)
1769 {
1770 profile_end(&si->sn_pr_child);
1771 profile_sub_wait(tm, &si->sn_pr_child); /* don't count wait time */
1772 profile_add(&si->sn_pr_children, &si->sn_pr_child);
1773 profile_add(&si->sn_prl_children, &si->sn_pr_child);
1774 }
1775 }
1776}
1777
1778static proftime_T inchar_time;
1779
1780/*
1781 * Called when starting to wait for the user to type a character.
1782 */
1783 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001784prof_inchar_enter(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001785{
1786 profile_start(&inchar_time);
1787}
1788
1789/*
1790 * Called when finished waiting for the user to type a character.
1791 */
1792 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001793prof_inchar_exit(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001794{
1795 profile_end(&inchar_time);
1796 profile_add(&prof_wait_time, &inchar_time);
1797}
1798
1799/*
1800 * Dump the profiling results for all scripts in file "fd".
1801 */
1802 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001803script_dump_profile(FILE *fd)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001804{
1805 int id;
1806 scriptitem_T *si;
1807 int i;
1808 FILE *sfd;
1809 sn_prl_T *pp;
1810
1811 for (id = 1; id <= script_items.ga_len; ++id)
1812 {
1813 si = &SCRIPT_ITEM(id);
1814 if (si->sn_prof_on)
1815 {
1816 fprintf(fd, "SCRIPT %s\n", si->sn_name);
1817 if (si->sn_pr_count == 1)
1818 fprintf(fd, "Sourced 1 time\n");
1819 else
1820 fprintf(fd, "Sourced %d times\n", si->sn_pr_count);
1821 fprintf(fd, "Total time: %s\n", profile_msg(&si->sn_pr_total));
1822 fprintf(fd, " Self time: %s\n", profile_msg(&si->sn_pr_self));
1823 fprintf(fd, "\n");
1824 fprintf(fd, "count total (s) self (s)\n");
1825
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +00001826 sfd = mch_fopen((char *)si->sn_name, "r");
Bram Moolenaar05159a02005-02-26 23:04:13 +00001827 if (sfd == NULL)
1828 fprintf(fd, "Cannot open file!\n");
1829 else
1830 {
Bram Moolenaar67435d92017-10-19 21:04:37 +02001831 /* Keep going till the end of file, so that trailing
1832 * continuation lines are listed. */
1833 for (i = 0; ; ++i)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001834 {
1835 if (vim_fgets(IObuff, IOSIZE, sfd))
1836 break;
Bram Moolenaarac112f02017-12-05 16:46:28 +01001837 /* When a line has been truncated, append NL, taking care
1838 * of multi-byte characters . */
1839 if (IObuff[IOSIZE - 2] != NUL && IObuff[IOSIZE - 2] != NL)
1840 {
1841 int n = IOSIZE - 2;
1842# ifdef FEAT_MBYTE
1843 if (enc_utf8)
1844 {
1845 /* Move to the first byte of this char.
1846 * utf_head_off() doesn't work, because it checks
1847 * for a truncated character. */
1848 while (n > 0 && (IObuff[n] & 0xc0) == 0x80)
1849 --n;
1850 }
1851 else if (has_mbyte)
1852 n -= mb_head_off(IObuff, IObuff + n);
1853# endif
1854 IObuff[n] = NL;
1855 IObuff[n + 1] = NUL;
1856 }
Bram Moolenaar67435d92017-10-19 21:04:37 +02001857 if (i < si->sn_prl_ga.ga_len
1858 && (pp = &PRL_ITEM(si, i))->snp_count > 0)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001859 {
1860 fprintf(fd, "%5d ", pp->snp_count);
1861 if (profile_equal(&pp->sn_prl_total, &pp->sn_prl_self))
1862 fprintf(fd, " ");
1863 else
1864 fprintf(fd, "%s ", profile_msg(&pp->sn_prl_total));
1865 fprintf(fd, "%s ", profile_msg(&pp->sn_prl_self));
1866 }
1867 else
1868 fprintf(fd, " ");
1869 fprintf(fd, "%s", IObuff);
1870 }
1871 fclose(sfd);
1872 }
1873 fprintf(fd, "\n");
1874 }
1875 }
1876}
1877
1878/*
1879 * Return TRUE when a function defined in the current script should be
1880 * profiled.
1881 */
1882 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001883prof_def_func(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001884{
Bram Moolenaar53180ce2005-07-05 21:48:14 +00001885 if (current_SID > 0)
1886 return SCRIPT_ITEM(current_SID).sn_pr_force;
1887 return FALSE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001888}
1889
1890# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001891#endif
1892
1893/*
1894 * If 'autowrite' option set, try to write the file.
1895 * Careful: autocommands may make "buf" invalid!
1896 *
1897 * return FAIL for failure, OK otherwise
1898 */
1899 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001900autowrite(buf_T *buf, int forceit)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001901{
Bram Moolenaar373154b2007-02-13 05:19:30 +00001902 int r;
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001903 bufref_T bufref;
Bram Moolenaar373154b2007-02-13 05:19:30 +00001904
Bram Moolenaar071d4272004-06-13 20:20:40 +00001905 if (!(p_aw || p_awa) || !p_write
1906#ifdef FEAT_QUICKFIX
Bram Moolenaar373154b2007-02-13 05:19:30 +00001907 /* never autowrite a "nofile" or "nowrite" buffer */
1908 || bt_dontwrite(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001909#endif
Bram Moolenaar373154b2007-02-13 05:19:30 +00001910 || (!forceit && buf->b_p_ro) || buf->b_ffname == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001911 return FAIL;
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001912 set_bufref(&bufref, buf);
Bram Moolenaar373154b2007-02-13 05:19:30 +00001913 r = buf_write_all(buf, forceit);
1914
1915 /* Writing may succeed but the buffer still changed, e.g., when there is a
1916 * conversion error. We do want to return FAIL then. */
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001917 if (bufref_valid(&bufref) && bufIsChanged(buf))
Bram Moolenaar373154b2007-02-13 05:19:30 +00001918 r = FAIL;
1919 return r;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001920}
1921
1922/*
1923 * flush all buffers, except the ones that are readonly
1924 */
1925 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001926autowrite_all(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001927{
1928 buf_T *buf;
1929
1930 if (!(p_aw || p_awa) || !p_write)
1931 return;
Bram Moolenaar29323592016-07-24 22:04:11 +02001932 FOR_ALL_BUFFERS(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001933 if (bufIsChanged(buf) && !buf->b_p_ro)
1934 {
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001935#ifdef FEAT_AUTOCMD
1936 bufref_T bufref;
1937
1938 set_bufref(&bufref, buf);
1939#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001940 (void)buf_write_all(buf, FALSE);
1941#ifdef FEAT_AUTOCMD
1942 /* an autocommand may have deleted the buffer */
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001943 if (!bufref_valid(&bufref))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001944 buf = firstbuf;
1945#endif
1946 }
1947}
1948
1949/*
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001950 * Return TRUE if buffer was changed and cannot be abandoned.
1951 * For flags use the CCGD_ values.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001952 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001953 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001954check_changed(buf_T *buf, int flags)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001955{
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001956 int forceit = (flags & CCGD_FORCEIT);
1957#ifdef FEAT_AUTOCMD
1958 bufref_T bufref;
1959
1960 set_bufref(&bufref, buf);
1961#endif
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001962
Bram Moolenaar071d4272004-06-13 20:20:40 +00001963 if ( !forceit
1964 && bufIsChanged(buf)
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001965 && ((flags & CCGD_MULTWIN) || buf->b_nwindows <= 1)
1966 && (!(flags & CCGD_AW) || autowrite(buf, forceit) == FAIL))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001967 {
1968#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG)
1969 if ((p_confirm || cmdmod.confirm) && p_write)
1970 {
1971 buf_T *buf2;
1972 int count = 0;
1973
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001974 if (flags & CCGD_ALLBUF)
Bram Moolenaar29323592016-07-24 22:04:11 +02001975 FOR_ALL_BUFFERS(buf2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001976 if (bufIsChanged(buf2)
1977 && (buf2->b_ffname != NULL
1978# ifdef FEAT_BROWSE
1979 || cmdmod.browse
1980# endif
1981 ))
1982 ++count;
1983# ifdef FEAT_AUTOCMD
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001984 if (!bufref_valid(&bufref))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001985 /* Autocommand deleted buffer, oops! It's not changed now. */
1986 return FALSE;
1987# endif
1988 dialog_changed(buf, count > 1);
1989# ifdef FEAT_AUTOCMD
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001990 if (!bufref_valid(&bufref))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001991 /* Autocommand deleted buffer, oops! It's not changed now. */
1992 return FALSE;
1993# endif
1994 return bufIsChanged(buf);
1995 }
1996#endif
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001997 if (flags & CCGD_EXCMD)
Bram Moolenaarf5be7cd2017-08-17 16:55:13 +02001998 no_write_message();
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001999 else
Bram Moolenaarf5be7cd2017-08-17 16:55:13 +02002000 no_write_message_nobang();
Bram Moolenaar071d4272004-06-13 20:20:40 +00002001 return TRUE;
2002 }
2003 return FALSE;
2004}
2005
2006#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG) || defined(PROTO)
2007
2008#if defined(FEAT_BROWSE) || defined(PROTO)
2009/*
2010 * When wanting to write a file without a file name, ask the user for a name.
2011 */
2012 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002013browse_save_fname(buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002014{
2015 if (buf->b_fname == NULL)
2016 {
2017 char_u *fname;
2018
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00002019 fname = do_browse(BROWSE_SAVE, (char_u *)_("Save As"),
2020 NULL, NULL, NULL, NULL, buf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002021 if (fname != NULL)
2022 {
2023 if (setfname(buf, fname, NULL, TRUE) == OK)
2024 buf->b_flags |= BF_NOTEDITED;
2025 vim_free(fname);
2026 }
2027 }
2028}
2029#endif
2030
2031/*
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02002032 * Ask the user what to do when abandoning a changed buffer.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002033 * Must check 'write' option first!
2034 */
2035 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002036dialog_changed(
2037 buf_T *buf,
2038 int checkall) /* may abandon all changed buffers */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002039{
Bram Moolenaard9462e32011-04-11 21:35:11 +02002040 char_u buff[DIALOG_MSG_SIZE];
Bram Moolenaar071d4272004-06-13 20:20:40 +00002041 int ret;
2042 buf_T *buf2;
Bram Moolenaar8218f602012-04-25 17:32:18 +02002043 exarg_T ea;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002044
Bram Moolenaar3f9a1ff2017-08-21 22:06:02 +02002045 dialog_msg(buff, _("Save changes to \"%s\"?"), buf->b_fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002046 if (checkall)
2047 ret = vim_dialog_yesnoallcancel(VIM_QUESTION, NULL, buff, 1);
2048 else
2049 ret = vim_dialog_yesnocancel(VIM_QUESTION, NULL, buff, 1);
2050
Bram Moolenaar8218f602012-04-25 17:32:18 +02002051 /* Init ea pseudo-structure, this is needed for the check_overwrite()
2052 * function. */
2053 ea.append = ea.forceit = FALSE;
2054
Bram Moolenaar071d4272004-06-13 20:20:40 +00002055 if (ret == VIM_YES)
2056 {
2057#ifdef FEAT_BROWSE
2058 /* May get file name, when there is none */
2059 browse_save_fname(buf);
2060#endif
Bram Moolenaar8218f602012-04-25 17:32:18 +02002061 if (buf->b_fname != NULL && check_overwrite(&ea, buf,
2062 buf->b_fname, buf->b_ffname, FALSE) == OK)
2063 /* didn't hit Cancel */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002064 (void)buf_write_all(buf, FALSE);
2065 }
2066 else if (ret == VIM_NO)
2067 {
2068 unchanged(buf, TRUE);
2069 }
2070 else if (ret == VIM_ALL)
2071 {
2072 /*
2073 * Write all modified files that can be written.
2074 * Skip readonly buffers, these need to be confirmed
2075 * individually.
2076 */
Bram Moolenaar29323592016-07-24 22:04:11 +02002077 FOR_ALL_BUFFERS(buf2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002078 {
2079 if (bufIsChanged(buf2)
2080 && (buf2->b_ffname != NULL
2081#ifdef FEAT_BROWSE
2082 || cmdmod.browse
2083#endif
2084 )
2085 && !buf2->b_p_ro)
2086 {
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002087#ifdef FEAT_AUTOCMD
2088 bufref_T bufref;
2089
2090 set_bufref(&bufref, buf2);
2091#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002092#ifdef FEAT_BROWSE
2093 /* May get file name, when there is none */
2094 browse_save_fname(buf2);
2095#endif
Bram Moolenaar8218f602012-04-25 17:32:18 +02002096 if (buf2->b_fname != NULL && check_overwrite(&ea, buf2,
2097 buf2->b_fname, buf2->b_ffname, FALSE) == OK)
2098 /* didn't hit Cancel */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002099 (void)buf_write_all(buf2, FALSE);
2100#ifdef FEAT_AUTOCMD
2101 /* an autocommand may have deleted the buffer */
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002102 if (!bufref_valid(&bufref))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002103 buf2 = firstbuf;
2104#endif
2105 }
2106 }
2107 }
2108 else if (ret == VIM_DISCARDALL)
2109 {
2110 /*
2111 * mark all buffers as unchanged
2112 */
Bram Moolenaar29323592016-07-24 22:04:11 +02002113 FOR_ALL_BUFFERS(buf2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002114 unchanged(buf2, TRUE);
2115 }
2116}
2117#endif
2118
2119/*
2120 * Return TRUE if the buffer "buf" can be abandoned, either by making it
2121 * hidden, autowriting it or unloading it.
2122 */
2123 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002124can_abandon(buf_T *buf, int forceit)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002125{
Bram Moolenaareb44a682017-08-03 22:44:55 +02002126 return ( buf_hide(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002127 || !bufIsChanged(buf)
2128 || buf->b_nwindows > 1
2129 || autowrite(buf, forceit) == OK
2130 || forceit);
2131}
2132
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01002133static void add_bufnum(int *bufnrs, int *bufnump, int nr);
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002134
2135/*
2136 * Add a buffer number to "bufnrs", unless it's already there.
2137 */
2138 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002139add_bufnum(int *bufnrs, int *bufnump, int nr)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002140{
2141 int i;
2142
2143 for (i = 0; i < *bufnump; ++i)
2144 if (bufnrs[i] == nr)
2145 return;
2146 bufnrs[*bufnump] = nr;
2147 *bufnump = *bufnump + 1;
2148}
2149
Bram Moolenaar071d4272004-06-13 20:20:40 +00002150/*
2151 * Return TRUE if any buffer was changed and cannot be abandoned.
2152 * That changed buffer becomes the current buffer.
Bram Moolenaar027387f2016-01-02 22:25:52 +01002153 * When "unload" is true the current buffer is unloaded instead of making it
2154 * hidden. This is used for ":q!".
Bram Moolenaar071d4272004-06-13 20:20:40 +00002155 */
2156 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002157check_changed_any(
2158 int hidden, /* Only check hidden buffers */
2159 int unload)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002160{
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002161 int ret = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002162 buf_T *buf;
2163 int save;
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002164 int i;
2165 int bufnum = 0;
2166 int bufcount = 0;
2167 int *bufnrs;
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002168 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002169 win_T *wp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002170
Bram Moolenaar29323592016-07-24 22:04:11 +02002171 FOR_ALL_BUFFERS(buf)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002172 ++bufcount;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002173
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002174 if (bufcount == 0)
2175 return FALSE;
2176
2177 bufnrs = (int *)alloc(sizeof(int) * bufcount);
2178 if (bufnrs == NULL)
2179 return FALSE;
2180
2181 /* curbuf */
2182 bufnrs[bufnum++] = curbuf->b_fnum;
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002183 /* buf in curtab */
2184 FOR_ALL_WINDOWS(wp)
2185 if (wp->w_buffer != curbuf)
2186 add_bufnum(bufnrs, &bufnum, wp->w_buffer->b_fnum);
2187
2188 /* buf in other tab */
Bram Moolenaar29323592016-07-24 22:04:11 +02002189 FOR_ALL_TABPAGES(tp)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002190 if (tp != curtab)
2191 for (wp = tp->tp_firstwin; wp != NULL; wp = wp->w_next)
2192 add_bufnum(bufnrs, &bufnum, wp->w_buffer->b_fnum);
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002193 /* any other buf */
Bram Moolenaar29323592016-07-24 22:04:11 +02002194 FOR_ALL_BUFFERS(buf)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002195 add_bufnum(bufnrs, &bufnum, buf->b_fnum);
2196
2197 for (i = 0; i < bufnum; ++i)
2198 {
2199 buf = buflist_findnr(bufnrs[i]);
2200 if (buf == NULL)
2201 continue;
2202 if ((!hidden || buf->b_nwindows == 0) && bufIsChanged(buf))
2203 {
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002204 bufref_T bufref;
2205
2206 set_bufref(&bufref, buf);
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002207 /* Try auto-writing the buffer. If this fails but the buffer no
2208 * longer exists it's not changed, that's OK. */
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002209 if (check_changed(buf, (p_awa ? CCGD_AW : 0)
2210 | CCGD_MULTWIN
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002211 | CCGD_ALLBUF) && bufref_valid(&bufref))
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002212 break; /* didn't save - still changes */
2213 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002214 }
2215
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002216 if (i >= bufnum)
2217 goto theend;
2218
2219 ret = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002220 exiting = FALSE;
2221#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG)
2222 /*
2223 * When ":confirm" used, don't give an error message.
2224 */
2225 if (!(p_confirm || cmdmod.confirm))
2226#endif
2227 {
2228 /* There must be a wait_return for this message, do_buffer()
2229 * may cause a redraw. But wait_return() is a no-op when vgetc()
2230 * is busy (Quit used from window menu), then make sure we don't
2231 * cause a scroll up. */
Bram Moolenaar61660ea2006-04-07 21:40:07 +00002232 if (vgetc_busy > 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002233 {
2234 msg_row = cmdline_row;
2235 msg_col = 0;
2236 msg_didout = FALSE;
2237 }
Bram Moolenaareb44a682017-08-03 22:44:55 +02002238 if (
2239#ifdef FEAT_TERMINAL
2240 term_job_running(buf->b_term)
2241 ? EMSG2(_("E947: Job still running in buffer \"%s\""),
2242 buf->b_fname)
2243 :
2244#endif
2245 EMSG2(_("E162: No write since last change for buffer \"%s\""),
Bram Moolenaare1704ba2012-10-03 18:25:00 +02002246 buf_spname(buf) != NULL ? buf_spname(buf) : buf->b_fname))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002247 {
2248 save = no_wait_return;
2249 no_wait_return = FALSE;
2250 wait_return(FALSE);
2251 no_wait_return = save;
2252 }
2253 }
2254
Bram Moolenaar071d4272004-06-13 20:20:40 +00002255 /* Try to find a window that contains the buffer. */
2256 if (buf != curbuf)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002257 FOR_ALL_TAB_WINDOWS(tp, wp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002258 if (wp->w_buffer == buf)
2259 {
Bram Moolenaar4033c552017-09-16 20:54:51 +02002260#ifdef FEAT_AUTOCMD
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002261 bufref_T bufref;
2262
2263 set_bufref(&bufref, buf);
Bram Moolenaar4033c552017-09-16 20:54:51 +02002264#endif
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002265 goto_tabpage_win(tp, wp);
Bram Moolenaar4033c552017-09-16 20:54:51 +02002266#ifdef FEAT_AUTOCMD
Bram Moolenaar071d4272004-06-13 20:20:40 +00002267 /* Paranoia: did autocms wipe out the buffer with changes? */
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002268 if (!bufref_valid(&bufref))
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002269 {
2270 goto theend;
2271 }
Bram Moolenaar4033c552017-09-16 20:54:51 +02002272#endif
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002273 goto buf_found;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002274 }
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002275buf_found:
Bram Moolenaar071d4272004-06-13 20:20:40 +00002276
2277 /* Open the changed buffer in the current window. */
2278 if (buf != curbuf)
Bram Moolenaar027387f2016-01-02 22:25:52 +01002279 set_curbuf(buf, unload ? DOBUF_UNLOAD : DOBUF_GOTO);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002280
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002281theend:
2282 vim_free(bufnrs);
2283 return ret;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002284}
2285
2286/*
2287 * return FAIL if there is no file name, OK if there is one
2288 * give error message for FAIL
2289 */
2290 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002291check_fname(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002292{
2293 if (curbuf->b_ffname == NULL)
2294 {
2295 EMSG(_(e_noname));
2296 return FAIL;
2297 }
2298 return OK;
2299}
2300
2301/*
2302 * flush the contents of a buffer, unless it has no file name
2303 *
2304 * return FAIL for failure, OK otherwise
2305 */
2306 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002307buf_write_all(buf_T *buf, int forceit)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002308{
2309 int retval;
2310#ifdef FEAT_AUTOCMD
2311 buf_T *old_curbuf = curbuf;
2312#endif
2313
2314 retval = (buf_write(buf, buf->b_ffname, buf->b_fname,
2315 (linenr_T)1, buf->b_ml.ml_line_count, NULL,
2316 FALSE, forceit, TRUE, FALSE));
2317#ifdef FEAT_AUTOCMD
2318 if (curbuf != old_curbuf)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00002319 {
Bram Moolenaar8820b482017-03-16 17:23:31 +01002320 msg_source(HL_ATTR(HLF_W));
Bram Moolenaar071d4272004-06-13 20:20:40 +00002321 MSG(_("Warning: Entered other buffer unexpectedly (check autocommands)"));
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00002322 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002323#endif
2324 return retval;
2325}
2326
2327/*
2328 * Code to handle the argument list.
2329 */
2330
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01002331static char_u *do_one_arg(char_u *str);
2332static int do_arglist(char_u *str, int what, int after);
2333static void alist_check_arg_idx(void);
2334static int editing_arg_idx(win_T *win);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002335#ifdef FEAT_LISTCMDS
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01002336static int alist_add_list(int count, char_u **files, int after);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002337#endif
2338#define AL_SET 1
2339#define AL_ADD 2
2340#define AL_DEL 3
2341
Bram Moolenaar071d4272004-06-13 20:20:40 +00002342/*
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002343 * Isolate one argument, taking backticks.
2344 * Changes the argument in-place, puts a NUL after it. Backticks remain.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002345 * Return a pointer to the start of the next argument.
2346 */
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002347 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002348do_one_arg(char_u *str)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002349{
2350 char_u *p;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002351 int inbacktick;
2352
Bram Moolenaar071d4272004-06-13 20:20:40 +00002353 inbacktick = FALSE;
2354 for (p = str; *str; ++str)
2355 {
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002356 /* When the backslash is used for escaping the special meaning of a
2357 * character we need to keep it until wildcard expansion. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002358 if (rem_backslash(str))
2359 {
2360 *p++ = *str++;
2361 *p++ = *str;
2362 }
2363 else
2364 {
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002365 /* An item ends at a space not in backticks */
2366 if (!inbacktick && vim_isspace(*str))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002367 break;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002368 if (*str == '`')
Bram Moolenaar071d4272004-06-13 20:20:40 +00002369 inbacktick ^= TRUE;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002370 *p++ = *str;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002371 }
2372 }
2373 str = skipwhite(str);
2374 *p = NUL;
2375
2376 return str;
2377}
2378
Bram Moolenaar86b68352004-12-27 21:59:20 +00002379/*
2380 * Separate the arguments in "str" and return a list of pointers in the
2381 * growarray "gap".
2382 */
Bram Moolenaar398ee732017-08-03 14:29:14 +02002383 static int
2384get_arglist(garray_T *gap, char_u *str, int escaped)
Bram Moolenaar86b68352004-12-27 21:59:20 +00002385{
2386 ga_init2(gap, (int)sizeof(char_u *), 20);
2387 while (*str != NUL)
2388 {
2389 if (ga_grow(gap, 1) == FAIL)
2390 {
2391 ga_clear(gap);
2392 return FAIL;
2393 }
2394 ((char_u **)gap->ga_data)[gap->ga_len++] = str;
2395
Bram Moolenaar398ee732017-08-03 14:29:14 +02002396 /* If str is escaped, don't handle backslashes or spaces */
2397 if (!escaped)
2398 return OK;
2399
Bram Moolenaar86b68352004-12-27 21:59:20 +00002400 /* Isolate one argument, change it in-place, put a NUL after it. */
2401 str = do_one_arg(str);
2402 }
2403 return OK;
2404}
2405
Bram Moolenaar7df351e2006-01-23 22:30:28 +00002406#if defined(FEAT_QUICKFIX) || defined(FEAT_SYN_HL) || defined(PROTO)
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002407/*
2408 * Parse a list of arguments (file names), expand them and return in
Bram Moolenaar8f5c6f02012-06-29 12:57:06 +02002409 * "fnames[fcountp]". When "wig" is TRUE, removes files matching 'wildignore'.
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002410 * Return FAIL or OK.
2411 */
2412 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002413get_arglist_exp(
2414 char_u *str,
2415 int *fcountp,
2416 char_u ***fnamesp,
2417 int wig)
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002418{
2419 garray_T ga;
2420 int i;
2421
Bram Moolenaar398ee732017-08-03 14:29:14 +02002422 if (get_arglist(&ga, str, TRUE) == FAIL)
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002423 return FAIL;
Bram Moolenaar8f5c6f02012-06-29 12:57:06 +02002424 if (wig == TRUE)
2425 i = expand_wildcards(ga.ga_len, (char_u **)ga.ga_data,
2426 fcountp, fnamesp, EW_FILE|EW_NOTFOUND);
2427 else
2428 i = gen_expand_wildcards(ga.ga_len, (char_u **)ga.ga_data,
2429 fcountp, fnamesp, EW_FILE|EW_NOTFOUND);
2430
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002431 ga_clear(&ga);
2432 return i;
2433}
2434#endif
2435
Bram Moolenaar071d4272004-06-13 20:20:40 +00002436#if defined(FEAT_GUI) || defined(FEAT_CLIENTSERVER) || defined(PROTO)
2437/*
2438 * Redefine the argument list.
2439 */
2440 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002441set_arglist(char_u *str)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002442{
2443 do_arglist(str, AL_SET, 0);
2444}
2445#endif
2446
2447/*
2448 * "what" == AL_SET: Redefine the argument list to 'str'.
2449 * "what" == AL_ADD: add files in 'str' to the argument list after "after".
2450 * "what" == AL_DEL: remove files in 'str' from the argument list.
2451 *
2452 * Return FAIL for failure, OK otherwise.
2453 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002454 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002455do_arglist(
2456 char_u *str,
Bram Moolenaarf1d25012016-03-03 12:22:53 +01002457 int what,
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002458 int after UNUSED) /* 0 means before first one */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002459{
2460 garray_T new_ga;
2461 int exp_count;
2462 char_u **exp_files;
2463 int i;
2464#ifdef FEAT_LISTCMDS
2465 char_u *p;
2466 int match;
2467#endif
Bram Moolenaar398ee732017-08-03 14:29:14 +02002468 int arg_escaped = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002469
2470 /*
Bram Moolenaar2faa29f2016-01-23 23:02:34 +01002471 * Set default argument for ":argadd" command.
2472 */
2473 if (what == AL_ADD && *str == NUL)
2474 {
2475 if (curbuf->b_ffname == NULL)
2476 return FAIL;
2477 str = curbuf->b_fname;
Bram Moolenaar398ee732017-08-03 14:29:14 +02002478 arg_escaped = FALSE;
Bram Moolenaar2faa29f2016-01-23 23:02:34 +01002479 }
2480
2481 /*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002482 * Collect all file name arguments in "new_ga".
2483 */
Bram Moolenaar398ee732017-08-03 14:29:14 +02002484 if (get_arglist(&new_ga, str, arg_escaped) == FAIL)
Bram Moolenaar86b68352004-12-27 21:59:20 +00002485 return FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002486
2487#ifdef FEAT_LISTCMDS
2488 if (what == AL_DEL)
2489 {
2490 regmatch_T regmatch;
2491 int didone;
2492
2493 /*
2494 * Delete the items: use each item as a regexp and find a match in the
2495 * argument list.
2496 */
Bram Moolenaar71afbfe2013-03-19 16:49:16 +01002497 regmatch.rm_ic = p_fic; /* ignore case when 'fileignorecase' is set */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002498 for (i = 0; i < new_ga.ga_len && !got_int; ++i)
2499 {
2500 p = ((char_u **)new_ga.ga_data)[i];
2501 p = file_pat_to_reg_pat(p, NULL, NULL, FALSE);
2502 if (p == NULL)
2503 break;
2504 regmatch.regprog = vim_regcomp(p, p_magic ? RE_MAGIC : 0);
2505 if (regmatch.regprog == NULL)
2506 {
2507 vim_free(p);
2508 break;
2509 }
2510
2511 didone = FALSE;
2512 for (match = 0; match < ARGCOUNT; ++match)
2513 if (vim_regexec(&regmatch, alist_name(&ARGLIST[match]),
2514 (colnr_T)0))
2515 {
2516 didone = TRUE;
2517 vim_free(ARGLIST[match].ae_fname);
2518 mch_memmove(ARGLIST + match, ARGLIST + match + 1,
2519 (ARGCOUNT - match - 1) * sizeof(aentry_T));
2520 --ALIST(curwin)->al_ga.ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002521 if (curwin->w_arg_idx > match)
2522 --curwin->w_arg_idx;
2523 --match;
2524 }
2525
Bram Moolenaar473de612013-06-08 18:19:48 +02002526 vim_regfree(regmatch.regprog);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002527 vim_free(p);
2528 if (!didone)
2529 EMSG2(_(e_nomatch2), ((char_u **)new_ga.ga_data)[i]);
2530 }
2531 ga_clear(&new_ga);
2532 }
2533 else
2534#endif
2535 {
2536 i = expand_wildcards(new_ga.ga_len, (char_u **)new_ga.ga_data,
2537 &exp_count, &exp_files, EW_DIR|EW_FILE|EW_ADDSLASH|EW_NOTFOUND);
2538 ga_clear(&new_ga);
Bram Moolenaar2db5c3b2016-01-16 22:49:34 +01002539 if (i == FAIL || exp_count == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002540 {
2541 EMSG(_(e_nomatch));
2542 return FAIL;
2543 }
2544
2545#ifdef FEAT_LISTCMDS
2546 if (what == AL_ADD)
2547 {
2548 (void)alist_add_list(exp_count, exp_files, after);
2549 vim_free(exp_files);
2550 }
2551 else /* what == AL_SET */
2552#endif
Bram Moolenaar86b68352004-12-27 21:59:20 +00002553 alist_set(ALIST(curwin), exp_count, exp_files, FALSE, NULL, 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002554 }
2555
2556 alist_check_arg_idx();
2557
2558 return OK;
2559}
2560
2561/*
2562 * Check the validity of the arg_idx for each other window.
2563 */
2564 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002565alist_check_arg_idx(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002566{
Bram Moolenaar071d4272004-06-13 20:20:40 +00002567 win_T *win;
Bram Moolenaarf740b292006-02-16 22:11:02 +00002568 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002569
Bram Moolenaarf740b292006-02-16 22:11:02 +00002570 FOR_ALL_TAB_WINDOWS(tp, win)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002571 if (win->w_alist == curwin->w_alist)
2572 check_arg_idx(win);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002573}
2574
2575/*
Bram Moolenaarb8ff1fb2012-02-04 21:59:01 +01002576 * Return TRUE if window "win" is editing the file at the current argument
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002577 * index.
2578 */
2579 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002580editing_arg_idx(win_T *win)
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002581{
2582 return !(win->w_arg_idx >= WARGCOUNT(win)
2583 || (win->w_buffer->b_fnum
2584 != WARGLIST(win)[win->w_arg_idx].ae_fnum
2585 && (win->w_buffer->b_ffname == NULL
2586 || !(fullpathcmp(
2587 alist_name(&WARGLIST(win)[win->w_arg_idx]),
2588 win->w_buffer->b_ffname, TRUE) & FPC_SAME))));
2589}
2590
2591/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002592 * Check if window "win" is editing the w_arg_idx file in its argument list.
2593 */
2594 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002595check_arg_idx(win_T *win)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002596{
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002597 if (WARGCOUNT(win) > 1 && !editing_arg_idx(win))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002598 {
2599 /* We are not editing the current entry in the argument list.
2600 * Set "arg_had_last" if we are editing the last one. */
2601 win->w_arg_idx_invalid = TRUE;
2602 if (win->w_arg_idx != WARGCOUNT(win) - 1
2603 && arg_had_last == FALSE
Bram Moolenaar071d4272004-06-13 20:20:40 +00002604 && ALIST(win) == &global_alist
Bram Moolenaar071d4272004-06-13 20:20:40 +00002605 && GARGCOUNT > 0
2606 && win->w_arg_idx < GARGCOUNT
2607 && (win->w_buffer->b_fnum == GARGLIST[GARGCOUNT - 1].ae_fnum
2608 || (win->w_buffer->b_ffname != NULL
2609 && (fullpathcmp(alist_name(&GARGLIST[GARGCOUNT - 1]),
2610 win->w_buffer->b_ffname, TRUE) & FPC_SAME))))
2611 arg_had_last = TRUE;
2612 }
2613 else
2614 {
2615 /* We are editing the current entry in the argument list.
2616 * Set "arg_had_last" if it's also the last one */
2617 win->w_arg_idx_invalid = FALSE;
2618 if (win->w_arg_idx == WARGCOUNT(win) - 1
Bram Moolenaar4033c552017-09-16 20:54:51 +02002619 && win->w_alist == &global_alist)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002620 arg_had_last = TRUE;
2621 }
2622}
2623
2624/*
2625 * ":args", ":argslocal" and ":argsglobal".
2626 */
2627 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002628ex_args(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002629{
2630 int i;
2631
2632 if (eap->cmdidx != CMD_args)
2633 {
Bram Moolenaar4033c552017-09-16 20:54:51 +02002634#if defined(FEAT_LISTCMDS)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002635 alist_unlink(ALIST(curwin));
2636 if (eap->cmdidx == CMD_argglobal)
2637 ALIST(curwin) = &global_alist;
2638 else /* eap->cmdidx == CMD_arglocal */
2639 alist_new();
2640#else
2641 ex_ni(eap);
2642 return;
2643#endif
2644 }
2645
2646 if (!ends_excmd(*eap->arg))
2647 {
2648 /*
2649 * ":args file ..": define new argument list, handle like ":next"
2650 * Also for ":argslocal file .." and ":argsglobal file ..".
2651 */
2652 ex_next(eap);
2653 }
2654 else
Bram Moolenaar4033c552017-09-16 20:54:51 +02002655#if defined(FEAT_LISTCMDS)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002656 if (eap->cmdidx == CMD_args)
2657#endif
2658 {
2659 /*
2660 * ":args": list arguments.
2661 */
2662 if (ARGCOUNT > 0)
2663 {
2664 /* Overwrite the command, for a short list there is no scrolling
2665 * required and no wait_return(). */
2666 gotocmdline(TRUE);
2667 for (i = 0; i < ARGCOUNT; ++i)
2668 {
2669 if (i == curwin->w_arg_idx)
2670 msg_putchar('[');
2671 msg_outtrans(alist_name(&ARGLIST[i]));
2672 if (i == curwin->w_arg_idx)
2673 msg_putchar(']');
2674 msg_putchar(' ');
2675 }
2676 }
2677 }
Bram Moolenaar4033c552017-09-16 20:54:51 +02002678#if defined(FEAT_LISTCMDS)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002679 else if (eap->cmdidx == CMD_arglocal)
2680 {
2681 garray_T *gap = &curwin->w_alist->al_ga;
2682
2683 /*
2684 * ":argslocal": make a local copy of the global argument list.
2685 */
2686 if (ga_grow(gap, GARGCOUNT) == OK)
2687 for (i = 0; i < GARGCOUNT; ++i)
2688 if (GARGLIST[i].ae_fname != NULL)
2689 {
2690 AARGLIST(curwin->w_alist)[gap->ga_len].ae_fname =
2691 vim_strsave(GARGLIST[i].ae_fname);
2692 AARGLIST(curwin->w_alist)[gap->ga_len].ae_fnum =
2693 GARGLIST[i].ae_fnum;
2694 ++gap->ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002695 }
2696 }
2697#endif
2698}
2699
2700/*
2701 * ":previous", ":sprevious", ":Next" and ":sNext".
2702 */
2703 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002704ex_previous(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002705{
2706 /* If past the last one already, go to the last one. */
2707 if (curwin->w_arg_idx - (int)eap->line2 >= ARGCOUNT)
2708 do_argfile(eap, ARGCOUNT - 1);
2709 else
2710 do_argfile(eap, curwin->w_arg_idx - (int)eap->line2);
2711}
2712
2713/*
2714 * ":rewind", ":first", ":sfirst" and ":srewind".
2715 */
2716 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002717ex_rewind(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002718{
2719 do_argfile(eap, 0);
2720}
2721
2722/*
2723 * ":last" and ":slast".
2724 */
2725 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002726ex_last(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002727{
2728 do_argfile(eap, ARGCOUNT - 1);
2729}
2730
2731/*
2732 * ":argument" and ":sargument".
2733 */
2734 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002735ex_argument(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002736{
2737 int i;
2738
2739 if (eap->addr_count > 0)
2740 i = eap->line2 - 1;
2741 else
2742 i = curwin->w_arg_idx;
2743 do_argfile(eap, i);
2744}
2745
2746/*
2747 * Edit file "argn" of the argument lists.
2748 */
2749 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002750do_argfile(exarg_T *eap, int argn)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002751{
2752 int other;
2753 char_u *p;
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002754 int old_arg_idx = curwin->w_arg_idx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002755
2756 if (argn < 0 || argn >= ARGCOUNT)
2757 {
2758 if (ARGCOUNT <= 1)
2759 EMSG(_("E163: There is only one file to edit"));
2760 else if (argn < 0)
2761 EMSG(_("E164: Cannot go before first file"));
2762 else
2763 EMSG(_("E165: Cannot go beyond last file"));
2764 }
2765 else
2766 {
2767 setpcmark();
2768#ifdef FEAT_GUI
2769 need_mouse_correct = TRUE;
2770#endif
2771
Bram Moolenaardf1bdc92006-02-23 21:32:16 +00002772 /* split window or create new tab page first */
2773 if (*eap->cmd == 's' || cmdmod.tab != 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002774 {
2775 if (win_split(0, 0) == FAIL)
2776 return;
Bram Moolenaar3368ea22010-09-21 16:56:35 +02002777 RESET_BINDING(curwin);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002778 }
2779 else
Bram Moolenaar071d4272004-06-13 20:20:40 +00002780 {
2781 /*
2782 * if 'hidden' set, only check for changed file when re-editing
2783 * the same buffer
2784 */
2785 other = TRUE;
Bram Moolenaareb44a682017-08-03 22:44:55 +02002786 if (buf_hide(curbuf))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002787 {
2788 p = fix_fname(alist_name(&ARGLIST[argn]));
2789 other = otherfile(p);
2790 vim_free(p);
2791 }
Bram Moolenaareb44a682017-08-03 22:44:55 +02002792 if ((!buf_hide(curbuf) || !other)
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002793 && check_changed(curbuf, CCGD_AW
2794 | (other ? 0 : CCGD_MULTWIN)
2795 | (eap->forceit ? CCGD_FORCEIT : 0)
2796 | CCGD_EXCMD))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002797 return;
2798 }
2799
2800 curwin->w_arg_idx = argn;
Bram Moolenaar4033c552017-09-16 20:54:51 +02002801 if (argn == ARGCOUNT - 1 && curwin->w_alist == &global_alist)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002802 arg_had_last = TRUE;
2803
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002804 /* Edit the file; always use the last known line number.
2805 * When it fails (e.g. Abort for already edited file) restore the
2806 * argument index. */
2807 if (do_ecmd(0, alist_name(&ARGLIST[curwin->w_arg_idx]), NULL,
Bram Moolenaar071d4272004-06-13 20:20:40 +00002808 eap, ECMD_LAST,
Bram Moolenaareb44a682017-08-03 22:44:55 +02002809 (buf_hide(curwin->w_buffer) ? ECMD_HIDE : 0)
Bram Moolenaar701f7af2008-11-15 13:12:07 +00002810 + (eap->forceit ? ECMD_FORCEIT : 0), curwin) == FAIL)
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002811 curwin->w_arg_idx = old_arg_idx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002812 /* like Vi: set the mark where the cursor is in the file. */
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002813 else if (eap->cmdidx != CMD_argdo)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002814 setmark('\'');
2815 }
2816}
2817
2818/*
2819 * ":next", and commands that behave like it.
2820 */
2821 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002822ex_next(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002823{
2824 int i;
2825
2826 /*
2827 * check for changed buffer now, if this fails the argument list is not
2828 * redefined.
2829 */
Bram Moolenaareb44a682017-08-03 22:44:55 +02002830 if ( buf_hide(curbuf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002831 || eap->cmdidx == CMD_snext
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002832 || !check_changed(curbuf, CCGD_AW
2833 | (eap->forceit ? CCGD_FORCEIT : 0)
2834 | CCGD_EXCMD))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002835 {
2836 if (*eap->arg != NUL) /* redefine file list */
2837 {
2838 if (do_arglist(eap->arg, AL_SET, 0) == FAIL)
2839 return;
2840 i = 0;
2841 }
2842 else
2843 i = curwin->w_arg_idx + (int)eap->line2;
2844 do_argfile(eap, i);
2845 }
2846}
2847
Bram Moolenaar0106e3d2016-02-23 18:55:43 +01002848#if defined(FEAT_LISTCMDS) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002849/*
2850 * ":argedit"
2851 */
2852 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002853ex_argedit(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002854{
Bram Moolenaar90305c62017-07-16 15:31:17 +02002855 int i = eap->addr_count ? (int)eap->line2 : curwin->w_arg_idx + 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002856
Bram Moolenaar90305c62017-07-16 15:31:17 +02002857 if (do_arglist(eap->arg, AL_ADD, i) == FAIL)
2858 return;
2859#ifdef FEAT_TITLE
2860 maketitle();
2861#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002862
Bram Moolenaar90305c62017-07-16 15:31:17 +02002863 if (curwin->w_arg_idx == 0 && (curbuf->b_ml.ml_flags & ML_EMPTY)
2864 && curbuf->b_ffname == NULL)
2865 i = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002866 /* Edit the argument. */
Bram Moolenaar90305c62017-07-16 15:31:17 +02002867 if (i < ARGCOUNT)
2868 do_argfile(eap, i);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002869}
2870
2871/*
2872 * ":argadd"
2873 */
2874 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002875ex_argadd(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002876{
2877 do_arglist(eap->arg, AL_ADD,
2878 eap->addr_count > 0 ? (int)eap->line2 : curwin->w_arg_idx + 1);
2879#ifdef FEAT_TITLE
2880 maketitle();
2881#endif
2882}
2883
2884/*
2885 * ":argdelete"
2886 */
2887 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002888ex_argdelete(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002889{
2890 int i;
2891 int n;
2892
2893 if (eap->addr_count > 0)
2894 {
2895 /* ":1,4argdel": Delete all arguments in the range. */
2896 if (eap->line2 > ARGCOUNT)
2897 eap->line2 = ARGCOUNT;
2898 n = eap->line2 - eap->line1 + 1;
Bram Moolenaar69a92fb2017-03-09 15:58:30 +01002899 if (*eap->arg != NUL)
2900 /* Can't have both a range and an argument. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002901 EMSG(_(e_invarg));
Bram Moolenaar69a92fb2017-03-09 15:58:30 +01002902 else if (n <= 0)
2903 {
2904 /* Don't give an error for ":%argdel" if the list is empty. */
2905 if (eap->line1 != 1 || eap->line2 != 0)
2906 EMSG(_(e_invrange));
2907 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002908 else
2909 {
2910 for (i = eap->line1; i <= eap->line2; ++i)
2911 vim_free(ARGLIST[i - 1].ae_fname);
2912 mch_memmove(ARGLIST + eap->line1 - 1, ARGLIST + eap->line2,
2913 (size_t)((ARGCOUNT - eap->line2) * sizeof(aentry_T)));
2914 ALIST(curwin)->al_ga.ga_len -= n;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002915 if (curwin->w_arg_idx >= eap->line2)
2916 curwin->w_arg_idx -= n;
2917 else if (curwin->w_arg_idx > eap->line1)
2918 curwin->w_arg_idx = eap->line1;
Bram Moolenaar72defda2016-01-17 18:04:33 +01002919 if (ARGCOUNT == 0)
2920 curwin->w_arg_idx = 0;
2921 else if (curwin->w_arg_idx >= ARGCOUNT)
2922 curwin->w_arg_idx = ARGCOUNT - 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002923 }
2924 }
2925 else if (*eap->arg == NUL)
2926 EMSG(_(e_argreq));
2927 else
2928 do_arglist(eap->arg, AL_DEL, 0);
2929#ifdef FEAT_TITLE
2930 maketitle();
2931#endif
2932}
2933
2934/*
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002935 * ":argdo", ":windo", ":bufdo", ":tabdo", ":cdo", ":ldo", ":cfdo" and ":lfdo"
Bram Moolenaar071d4272004-06-13 20:20:40 +00002936 */
2937 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002938ex_listdo(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002939{
2940 int i;
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002941 win_T *wp;
2942 tabpage_T *tp;
Bram Moolenaare25bb902015-02-27 20:33:37 +01002943 buf_T *buf = curbuf;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002944 int next_fnum = 0;
2945#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
2946 char_u *save_ei = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002947#endif
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002948 char_u *p_shm_save;
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002949#ifdef FEAT_QUICKFIX
Bram Moolenaared84b762015-09-09 22:35:29 +02002950 int qf_size = 0;
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002951 int qf_idx;
2952#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002953
Bram Moolenaar0106e3d2016-02-23 18:55:43 +01002954#ifndef FEAT_QUICKFIX
2955 if (eap->cmdidx == CMD_cdo || eap->cmdidx == CMD_ldo ||
2956 eap->cmdidx == CMD_cfdo || eap->cmdidx == CMD_lfdo)
2957 {
2958 ex_ni(eap);
2959 return;
2960 }
2961#endif
2962
Bram Moolenaar071d4272004-06-13 20:20:40 +00002963#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002964 if (eap->cmdidx != CMD_windo && eap->cmdidx != CMD_tabdo)
Bram Moolenaardcaf10e2005-01-21 11:55:25 +00002965 /* Don't do syntax HL autocommands. Skipping the syntax file is a
2966 * great speed improvement. */
2967 save_ei = au_event_disable(",Syntax");
Bram Moolenaar071d4272004-06-13 20:20:40 +00002968#endif
Bram Moolenaar6b1ee342014-08-06 18:17:11 +02002969#ifdef FEAT_CLIPBOARD
2970 start_global_changes();
2971#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002972
2973 if (eap->cmdidx == CMD_windo
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002974 || eap->cmdidx == CMD_tabdo
Bram Moolenaareb44a682017-08-03 22:44:55 +02002975 || buf_hide(curbuf)
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002976 || !check_changed(curbuf, CCGD_AW
2977 | (eap->forceit ? CCGD_FORCEIT : 0)
2978 | CCGD_EXCMD))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002979 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00002980 i = 0;
Bram Moolenaara162bc52015-01-07 16:54:21 +01002981 /* start at the eap->line1 argument/window/buffer */
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002982 wp = firstwin;
2983 tp = first_tabpage;
Bram Moolenaara162bc52015-01-07 16:54:21 +01002984 switch (eap->cmdidx)
2985 {
Bram Moolenaara162bc52015-01-07 16:54:21 +01002986 case CMD_windo:
2987 for ( ; wp != NULL && i + 1 < eap->line1; wp = wp->w_next)
2988 i++;
2989 break;
2990 case CMD_tabdo:
2991 for( ; tp != NULL && i + 1 < eap->line1; tp = tp->tp_next)
2992 i++;
2993 break;
Bram Moolenaara162bc52015-01-07 16:54:21 +01002994 case CMD_argdo:
2995 i = eap->line1 - 1;
2996 break;
Bram Moolenaara162bc52015-01-07 16:54:21 +01002997 default:
2998 break;
2999 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003000 /* set pcmark now */
3001 if (eap->cmdidx == CMD_bufdo)
Bram Moolenaaraa23b372015-09-08 18:46:31 +02003002 {
Bram Moolenaare25bb902015-02-27 20:33:37 +01003003 /* Advance to the first listed buffer after "eap->line1". */
Bram Moolenaaraa23b372015-09-08 18:46:31 +02003004 for (buf = firstbuf; buf != NULL && (buf->b_fnum < eap->line1
Bram Moolenaare25bb902015-02-27 20:33:37 +01003005 || !buf->b_p_bl); buf = buf->b_next)
3006 if (buf->b_fnum > eap->line2)
3007 {
3008 buf = NULL;
3009 break;
3010 }
Bram Moolenaaraa23b372015-09-08 18:46:31 +02003011 if (buf != NULL)
Bram Moolenaare25bb902015-02-27 20:33:37 +01003012 goto_buffer(eap, DOBUF_FIRST, FORWARD, buf->b_fnum);
Bram Moolenaaraa23b372015-09-08 18:46:31 +02003013 }
3014#ifdef FEAT_QUICKFIX
3015 else if (eap->cmdidx == CMD_cdo || eap->cmdidx == CMD_ldo
3016 || eap->cmdidx == CMD_cfdo || eap->cmdidx == CMD_lfdo)
3017 {
3018 qf_size = qf_get_size(eap);
3019 if (qf_size <= 0 || eap->line1 > qf_size)
3020 buf = NULL;
3021 else
3022 {
3023 ex_cc(eap);
3024
3025 buf = curbuf;
3026 i = eap->line1 - 1;
3027 if (eap->addr_count <= 0)
3028 /* default is all the quickfix/location list entries */
3029 eap->line2 = qf_size;
3030 }
3031 }
3032#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003033 else
3034 setpcmark();
3035 listcmd_busy = TRUE; /* avoids setting pcmark below */
3036
Bram Moolenaare25bb902015-02-27 20:33:37 +01003037 while (!got_int && buf != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003038 {
3039 if (eap->cmdidx == CMD_argdo)
3040 {
3041 /* go to argument "i" */
3042 if (i == ARGCOUNT)
3043 break;
3044 /* Don't call do_argfile() when already there, it will try
3045 * reloading the file. */
Bram Moolenaard4755bb2004-09-02 19:12:26 +00003046 if (curwin->w_arg_idx != i || !editing_arg_idx(curwin))
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003047 {
3048 /* Clear 'shm' to avoid that the file message overwrites
3049 * any output from the command. */
3050 p_shm_save = vim_strsave(p_shm);
3051 set_option_value((char_u *)"shm", 0L, (char_u *)"", 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003052 do_argfile(eap, i);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003053 set_option_value((char_u *)"shm", 0L, p_shm_save, 0);
3054 vim_free(p_shm_save);
3055 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003056 if (curwin->w_arg_idx != i)
3057 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003058 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003059 else if (eap->cmdidx == CMD_windo)
3060 {
Bram Moolenaar32466aa2006-02-24 23:53:04 +00003061 /* go to window "wp" */
3062 if (!win_valid(wp))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003063 break;
Bram Moolenaar32466aa2006-02-24 23:53:04 +00003064 win_goto(wp);
Bram Moolenaar41423242007-05-03 20:11:13 +00003065 if (curwin != wp)
3066 break; /* something must be wrong */
Bram Moolenaar32466aa2006-02-24 23:53:04 +00003067 wp = curwin->w_next;
3068 }
3069 else if (eap->cmdidx == CMD_tabdo)
3070 {
3071 /* go to window "tp" */
3072 if (!valid_tabpage(tp))
3073 break;
Bram Moolenaar49e649f2013-05-06 04:50:35 +02003074 goto_tabpage_tp(tp, TRUE, TRUE);
Bram Moolenaar32466aa2006-02-24 23:53:04 +00003075 tp = tp->tp_next;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003076 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003077 else if (eap->cmdidx == CMD_bufdo)
3078 {
3079 /* Remember the number of the next listed buffer, in case
3080 * ":bwipe" is used or autocommands do something strange. */
3081 next_fnum = -1;
3082 for (buf = curbuf->b_next; buf != NULL; buf = buf->b_next)
3083 if (buf->b_p_bl)
3084 {
3085 next_fnum = buf->b_fnum;
3086 break;
3087 }
3088 }
3089
Bram Moolenaara162bc52015-01-07 16:54:21 +01003090 ++i;
3091
Bram Moolenaar071d4272004-06-13 20:20:40 +00003092 /* execute the command */
3093 do_cmdline(eap->arg, eap->getline, eap->cookie,
3094 DOCMD_VERBOSE + DOCMD_NOWAIT);
3095
3096 if (eap->cmdidx == CMD_bufdo)
3097 {
3098 /* Done? */
Bram Moolenaara162bc52015-01-07 16:54:21 +01003099 if (next_fnum < 0 || next_fnum > eap->line2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003100 break;
3101 /* Check if the buffer still exists. */
Bram Moolenaar29323592016-07-24 22:04:11 +02003102 FOR_ALL_BUFFERS(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003103 if (buf->b_fnum == next_fnum)
3104 break;
3105 if (buf == NULL)
3106 break;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003107
3108 /* Go to the next buffer. Clear 'shm' to avoid that the file
3109 * message overwrites any output from the command. */
3110 p_shm_save = vim_strsave(p_shm);
3111 set_option_value((char_u *)"shm", 0L, (char_u *)"", 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003112 goto_buffer(eap, DOBUF_FIRST, FORWARD, next_fnum);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003113 set_option_value((char_u *)"shm", 0L, p_shm_save, 0);
3114 vim_free(p_shm_save);
3115
Bram Moolenaaraa23b372015-09-08 18:46:31 +02003116 /* If autocommands took us elsewhere, quit here. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003117 if (curbuf->b_fnum != next_fnum)
3118 break;
3119 }
3120
Bram Moolenaaraa23b372015-09-08 18:46:31 +02003121#ifdef FEAT_QUICKFIX
3122 if (eap->cmdidx == CMD_cdo || eap->cmdidx == CMD_ldo
3123 || eap->cmdidx == CMD_cfdo || eap->cmdidx == CMD_lfdo)
3124 {
3125 if (i >= qf_size || i >= eap->line2)
3126 break;
3127
3128 qf_idx = qf_get_cur_idx(eap);
3129
3130 ex_cnext(eap);
3131
3132 /* If jumping to the next quickfix entry fails, quit here */
3133 if (qf_get_cur_idx(eap) == qf_idx)
3134 break;
3135 }
3136#endif
3137
Bram Moolenaar071d4272004-06-13 20:20:40 +00003138 if (eap->cmdidx == CMD_windo)
3139 {
3140 validate_cursor(); /* cursor may have moved */
3141#ifdef FEAT_SCROLLBIND
3142 /* required when 'scrollbind' has been set */
3143 if (curwin->w_p_scb)
3144 do_check_scrollbind(TRUE);
3145#endif
3146 }
Bram Moolenaara162bc52015-01-07 16:54:21 +01003147
Bram Moolenaara162bc52015-01-07 16:54:21 +01003148 if (eap->cmdidx == CMD_windo || eap->cmdidx == CMD_tabdo)
3149 if (i+1 > eap->line2)
3150 break;
Bram Moolenaara162bc52015-01-07 16:54:21 +01003151 if (eap->cmdidx == CMD_argdo && i >= eap->line2)
3152 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003153 }
3154 listcmd_busy = FALSE;
3155 }
3156
3157#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
Bram Moolenaar6ac54292005-02-02 23:07:25 +00003158 if (save_ei != NULL)
3159 {
3160 au_event_restore(save_ei);
3161 apply_autocmds(EVENT_SYNTAX, curbuf->b_p_syn,
3162 curbuf->b_fname, TRUE, curbuf);
3163 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003164#endif
Bram Moolenaar6b1ee342014-08-06 18:17:11 +02003165#ifdef FEAT_CLIPBOARD
3166 end_global_changes();
3167#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003168}
3169
3170/*
3171 * Add files[count] to the arglist of the current window after arg "after".
3172 * The file names in files[count] must have been allocated and are taken over.
3173 * Files[] itself is not taken over.
3174 * Returns index of first added argument. Returns -1 when failed (out of mem).
3175 */
3176 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003177alist_add_list(
3178 int count,
3179 char_u **files,
3180 int after) /* where to add: 0 = before first one */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003181{
3182 int i;
Bram Moolenaara24f0a52016-01-17 19:39:00 +01003183 int old_argcount = ARGCOUNT;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003184
3185 if (ga_grow(&ALIST(curwin)->al_ga, count) == OK)
3186 {
3187 if (after < 0)
3188 after = 0;
3189 if (after > ARGCOUNT)
3190 after = ARGCOUNT;
3191 if (after < ARGCOUNT)
3192 mch_memmove(&(ARGLIST[after + count]), &(ARGLIST[after]),
3193 (ARGCOUNT - after) * sizeof(aentry_T));
3194 for (i = 0; i < count; ++i)
3195 {
3196 ARGLIST[after + i].ae_fname = files[i];
3197 ARGLIST[after + i].ae_fnum = buflist_add(files[i], BLN_LISTED);
3198 }
3199 ALIST(curwin)->al_ga.ga_len += count;
Bram Moolenaara24f0a52016-01-17 19:39:00 +01003200 if (old_argcount > 0 && curwin->w_arg_idx >= after)
3201 curwin->w_arg_idx += count;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003202 return after;
3203 }
3204
3205 for (i = 0; i < count; ++i)
3206 vim_free(files[i]);
3207 return -1;
3208}
3209
3210#endif /* FEAT_LISTCMDS */
3211
3212#ifdef FEAT_EVAL
3213/*
3214 * ":compiler[!] {name}"
3215 */
3216 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003217ex_compiler(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003218{
3219 char_u *buf;
3220 char_u *old_cur_comp = NULL;
3221 char_u *p;
3222
3223 if (*eap->arg == NUL)
3224 {
3225 /* List all compiler scripts. */
3226 do_cmdline_cmd((char_u *)"echo globpath(&rtp, 'compiler/*.vim')");
3227 /* ) keep the indenter happy... */
3228 }
3229 else
3230 {
3231 buf = alloc((unsigned)(STRLEN(eap->arg) + 14));
3232 if (buf != NULL)
3233 {
3234 if (eap->forceit)
3235 {
3236 /* ":compiler! {name}" sets global options */
3237 do_cmdline_cmd((char_u *)
3238 "command -nargs=* CompilerSet set <args>");
3239 }
3240 else
3241 {
3242 /* ":compiler! {name}" sets local options.
3243 * To remain backwards compatible "current_compiler" is always
3244 * used. A user's compiler plugin may set it, the distributed
3245 * plugin will then skip the settings. Afterwards set
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003246 * "b:current_compiler" and restore "current_compiler".
3247 * Explicitly prepend "g:" to make it work in a function. */
3248 old_cur_comp = get_var_value((char_u *)"g:current_compiler");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003249 if (old_cur_comp != NULL)
3250 old_cur_comp = vim_strsave(old_cur_comp);
3251 do_cmdline_cmd((char_u *)
3252 "command -nargs=* CompilerSet setlocal <args>");
3253 }
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003254 do_unlet((char_u *)"g:current_compiler", TRUE);
Bram Moolenaar2ce06f62005-01-31 19:19:04 +00003255 do_unlet((char_u *)"b:current_compiler", TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003256
3257 sprintf((char *)buf, "compiler/%s.vim", eap->arg);
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003258 if (source_runtime(buf, DIP_ALL) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003259 EMSG2(_("E666: compiler not supported: %s"), eap->arg);
3260 vim_free(buf);
3261
3262 do_cmdline_cmd((char_u *)":delcommand CompilerSet");
3263
3264 /* Set "b:current_compiler" from "current_compiler". */
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003265 p = get_var_value((char_u *)"g:current_compiler");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003266 if (p != NULL)
3267 set_internal_string_var((char_u *)"b:current_compiler", p);
3268
3269 /* Restore "current_compiler" for ":compiler {name}". */
3270 if (!eap->forceit)
3271 {
3272 if (old_cur_comp != NULL)
3273 {
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003274 set_internal_string_var((char_u *)"g:current_compiler",
Bram Moolenaar071d4272004-06-13 20:20:40 +00003275 old_cur_comp);
3276 vim_free(old_cur_comp);
3277 }
3278 else
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003279 do_unlet((char_u *)"g:current_compiler", TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003280 }
3281 }
3282 }
3283}
3284#endif
3285
3286/*
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003287 * ":runtime [what] {name}"
Bram Moolenaar071d4272004-06-13 20:20:40 +00003288 */
3289 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003290ex_runtime(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003291{
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003292 char_u *arg = eap->arg;
3293 char_u *p = skiptowhite(arg);
3294 int len = (int)(p - arg);
3295 int flags = eap->forceit ? DIP_ALL : 0;
3296
3297 if (STRNCMP(arg, "START", len) == 0)
3298 {
3299 flags += DIP_START + DIP_NORTP;
3300 arg = skipwhite(arg + len);
3301 }
3302 else if (STRNCMP(arg, "OPT", len) == 0)
3303 {
3304 flags += DIP_OPT + DIP_NORTP;
3305 arg = skipwhite(arg + len);
3306 }
3307 else if (STRNCMP(arg, "PACK", len) == 0)
3308 {
3309 flags += DIP_START + DIP_OPT + DIP_NORTP;
3310 arg = skipwhite(arg + len);
3311 }
3312 else if (STRNCMP(arg, "ALL", len) == 0)
3313 {
3314 flags += DIP_START + DIP_OPT;
3315 arg = skipwhite(arg + len);
3316 }
3317
3318 source_runtime(arg, flags);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003319}
3320
Bram Moolenaar071d4272004-06-13 20:20:40 +00003321 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003322source_callback(char_u *fname, void *cookie UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003323{
Bram Moolenaar910f66f2006-04-05 20:41:53 +00003324 (void)do_source(fname, FALSE, DOSO_NONE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003325}
3326
3327/*
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003328 * Find the file "name" in all directories in "path" and invoke
3329 * "callback(fname, cookie)".
3330 * "name" can contain wildcards.
3331 * When "flags" has DIP_ALL: source all files, otherwise only the first one.
3332 * When "flags" has DIP_DIR: find directories instead of files.
3333 * When "flags" has DIP_ERR: give an error message if there is no match.
3334 *
3335 * return FAIL when no file could be sourced, OK otherwise.
3336 */
Bram Moolenaar6bef5302016-03-12 21:28:26 +01003337 int
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003338do_in_path(
3339 char_u *path,
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003340 char_u *name,
Bram Moolenaar91715872016-03-03 17:13:03 +01003341 int flags,
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003342 void (*callback)(char_u *fname, void *ck),
3343 void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003344{
3345 char_u *rtp;
3346 char_u *np;
3347 char_u *buf;
3348 char_u *rtp_copy;
3349 char_u *tail;
3350 int num_files;
3351 char_u **files;
3352 int i;
3353 int did_one = FALSE;
3354#ifdef AMIGA
3355 struct Process *proc = (struct Process *)FindTask(0L);
3356 APTR save_winptr = proc->pr_WindowPtr;
3357
3358 /* Avoid a requester here for a volume that doesn't exist. */
3359 proc->pr_WindowPtr = (APTR)-1L;
3360#endif
3361
3362 /* Make a copy of 'runtimepath'. Invoking the callback may change the
3363 * value. */
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003364 rtp_copy = vim_strsave(path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003365 buf = alloc(MAXPATHL);
3366 if (buf != NULL && rtp_copy != NULL)
3367 {
Bram Moolenaarc09a6d62013-06-10 21:27:29 +02003368 if (p_verbose > 1 && name != NULL)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003369 {
3370 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00003371 smsg((char_u *)_("Searching for \"%s\" in \"%s\""),
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003372 (char *)name, (char *)path);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003373 verbose_leave();
3374 }
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00003375
Bram Moolenaar071d4272004-06-13 20:20:40 +00003376 /* Loop over all entries in 'runtimepath'. */
3377 rtp = rtp_copy;
Bram Moolenaar91715872016-03-03 17:13:03 +01003378 while (*rtp != NUL && ((flags & DIP_ALL) || !did_one))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003379 {
Bram Moolenaar66459b72016-08-06 19:01:55 +02003380 size_t buflen;
3381
Bram Moolenaar071d4272004-06-13 20:20:40 +00003382 /* Copy the path from 'runtimepath' to buf[]. */
3383 copy_option_part(&rtp, buf, MAXPATHL, ",");
Bram Moolenaar66459b72016-08-06 19:01:55 +02003384 buflen = STRLEN(buf);
3385
3386 /* Skip after or non-after directories. */
3387 if (flags & (DIP_NOAFTER | DIP_AFTER))
3388 {
3389 int is_after = buflen >= 5
3390 && STRCMP(buf + buflen - 5, "after") == 0;
3391
3392 if ((is_after && (flags & DIP_NOAFTER))
3393 || (!is_after && (flags & DIP_AFTER)))
3394 continue;
3395 }
3396
Bram Moolenaarc09a6d62013-06-10 21:27:29 +02003397 if (name == NULL)
3398 {
3399 (*callback)(buf, (void *) &cookie);
3400 if (!did_one)
3401 did_one = (cookie == NULL);
3402 }
Bram Moolenaar66459b72016-08-06 19:01:55 +02003403 else if (buflen + STRLEN(name) + 2 < MAXPATHL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003404 {
3405 add_pathsep(buf);
3406 tail = buf + STRLEN(buf);
3407
3408 /* Loop over all patterns in "name" */
3409 np = name;
Bram Moolenaar91715872016-03-03 17:13:03 +01003410 while (*np != NUL && ((flags & DIP_ALL) || !did_one))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003411 {
3412 /* Append the pattern from "name" to buf[]. */
3413 copy_option_part(&np, tail, (int)(MAXPATHL - (tail - buf)),
3414 "\t ");
3415
3416 if (p_verbose > 2)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003417 {
3418 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00003419 smsg((char_u *)_("Searching for \"%s\""), buf);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003420 verbose_leave();
3421 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003422
3423 /* Expand wildcards, invoke the callback for each match. */
3424 if (gen_expand_wildcards(1, &buf, &num_files, &files,
Bram Moolenaar91715872016-03-03 17:13:03 +01003425 (flags & DIP_DIR) ? EW_DIR : EW_FILE) == OK)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003426 {
3427 for (i = 0; i < num_files; ++i)
3428 {
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00003429 (*callback)(files[i], cookie);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003430 did_one = TRUE;
Bram Moolenaar91715872016-03-03 17:13:03 +01003431 if (!(flags & DIP_ALL))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003432 break;
3433 }
3434 FreeWild(num_files, files);
3435 }
3436 }
3437 }
3438 }
3439 }
3440 vim_free(buf);
3441 vim_free(rtp_copy);
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003442 if (!did_one && name != NULL)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003443 {
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003444 char *basepath = path == p_rtp ? "runtimepath" : "packpath";
3445
3446 if (flags & DIP_ERR)
3447 EMSG3(_(e_dirnotf), basepath, name);
3448 else if (p_verbose > 0)
3449 {
3450 verbose_enter();
3451 smsg((char_u *)_("not found in '%s': \"%s\""), basepath, name);
3452 verbose_leave();
3453 }
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003454 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003455
3456#ifdef AMIGA
3457 proc->pr_WindowPtr = save_winptr;
3458#endif
3459
3460 return did_one ? OK : FAIL;
3461}
3462
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003463/*
Bram Moolenaar07ecfa62017-06-27 14:43:55 +02003464 * Find "name" in "path". When found, invoke the callback function for
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003465 * it: callback(fname, "cookie")
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003466 * When "flags" has DIP_ALL repeat for all matches, otherwise only the first
3467 * one is used.
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003468 * Returns OK when at least one match found, FAIL otherwise.
3469 *
Bram Moolenaar07ecfa62017-06-27 14:43:55 +02003470 * If "name" is NULL calls callback for each entry in "path". Cookie is
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003471 * passed by reference in this case, setting it to NULL indicates that callback
3472 * has done its job.
3473 */
Bram Moolenaar07ecfa62017-06-27 14:43:55 +02003474 static int
3475do_in_path_and_pp(
3476 char_u *path,
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003477 char_u *name,
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003478 int flags,
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003479 void (*callback)(char_u *fname, void *ck),
3480 void *cookie)
3481{
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003482 int done = FAIL;
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003483 char_u *s;
3484 int len;
3485 char *start_dir = "pack/*/start/*/%s";
3486 char *opt_dir = "pack/*/opt/*/%s";
3487
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003488 if ((flags & DIP_NORTP) == 0)
Bram Moolenaar07ecfa62017-06-27 14:43:55 +02003489 done = do_in_path(path, name, flags, callback, cookie);
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003490
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003491 if ((done == FAIL || (flags & DIP_ALL)) && (flags & DIP_START))
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003492 {
Bram Moolenaar1c8b4ed2016-03-17 21:51:03 +01003493 len = (int)(STRLEN(start_dir) + STRLEN(name));
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003494 s = alloc(len);
3495 if (s == NULL)
3496 return FAIL;
3497 vim_snprintf((char *)s, len, start_dir, name);
3498 done = do_in_path(p_pp, s, flags, callback, cookie);
3499 vim_free(s);
3500 }
3501
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003502 if ((done == FAIL || (flags & DIP_ALL)) && (flags & DIP_OPT))
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003503 {
Bram Moolenaar1c8b4ed2016-03-17 21:51:03 +01003504 len = (int)(STRLEN(opt_dir) + STRLEN(name));
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003505 s = alloc(len);
3506 if (s == NULL)
3507 return FAIL;
3508 vim_snprintf((char *)s, len, opt_dir, name);
3509 done = do_in_path(p_pp, s, flags, callback, cookie);
3510 vim_free(s);
3511 }
3512
3513 return done;
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003514}
3515
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003516/*
Bram Moolenaar07ecfa62017-06-27 14:43:55 +02003517 * Just like do_in_path_and_pp(), using 'runtimepath' for "path".
3518 */
3519 int
3520do_in_runtimepath(
3521 char_u *name,
3522 int flags,
3523 void (*callback)(char_u *fname, void *ck),
3524 void *cookie)
3525{
3526 return do_in_path_and_pp(p_rtp, name, flags, callback, cookie);
3527}
3528
3529/*
3530 * Source the file "name" from all directories in 'runtimepath'.
3531 * "name" can contain wildcards.
3532 * When "flags" has DIP_ALL: source all files, otherwise only the first one.
3533 *
3534 * return FAIL when no file could be sourced, OK otherwise.
3535 */
3536 int
3537source_runtime(char_u *name, int flags)
3538{
3539 return source_in_path(p_rtp, name, flags);
3540}
3541
3542/*
3543 * Just like source_runtime(), but use "path" instead of 'runtimepath'.
3544 */
3545 int
3546source_in_path(char_u *path, char_u *name, int flags)
3547{
3548 return do_in_path_and_pp(path, name, flags, source_callback, NULL);
3549}
3550
3551
3552/*
Bram Moolenaarf3654822016-03-04 22:12:23 +01003553 * Expand wildcards in "pat" and invoke do_source() for each match.
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003554 */
3555 static void
Bram Moolenaarf3654822016-03-04 22:12:23 +01003556source_all_matches(char_u *pat)
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003557{
Bram Moolenaarf3654822016-03-04 22:12:23 +01003558 int num_files;
3559 char_u **files;
3560 int i;
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003561
Bram Moolenaarf3654822016-03-04 22:12:23 +01003562 if (gen_expand_wildcards(1, &pat, &num_files, &files, EW_FILE) == OK)
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003563 {
Bram Moolenaarf3654822016-03-04 22:12:23 +01003564 for (i = 0; i < num_files; ++i)
3565 (void)do_source(files[i], FALSE, DOSO_NONE);
3566 FreeWild(num_files, files);
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003567 }
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003568}
3569
Bram Moolenaar2374faa2018-02-04 17:47:42 +01003570/*
3571 * Add the package directory to 'runtimepath'.
3572 */
3573 static int
3574add_pack_dir_to_rtp(char_u *fname)
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003575{
Bram Moolenaarf3654822016-03-04 22:12:23 +01003576 char_u *p4, *p3, *p2, *p1, *p;
3577 char_u *insp;
Bram Moolenaar91715872016-03-03 17:13:03 +01003578 int c;
3579 char_u *new_rtp;
3580 int keep;
Bram Moolenaarb0550662016-05-31 21:37:36 +02003581 size_t oldlen;
3582 size_t addlen;
Bram Moolenaar2374faa2018-02-04 17:47:42 +01003583 char_u *afterdir = NULL;
Bram Moolenaarb0550662016-05-31 21:37:36 +02003584 size_t afterlen = 0;
Bram Moolenaar2374faa2018-02-04 17:47:42 +01003585 char_u *ffname = NULL;
Bram Moolenaarfef524b2016-07-02 22:07:22 +02003586 size_t fname_len;
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003587 char_u *buf = NULL;
3588 char_u *rtp_ffname;
3589 int match;
Bram Moolenaar2374faa2018-02-04 17:47:42 +01003590 int retval = FAIL;
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003591
Bram Moolenaar2374faa2018-02-04 17:47:42 +01003592 p4 = p3 = p2 = p1 = get_past_head(fname);
3593 for (p = p1; *p; MB_PTR_ADV(p))
3594 if (vim_ispathsep_nocolon(*p))
3595 {
3596 p4 = p3; p3 = p2; p2 = p1; p1 = p;
3597 }
3598
3599 /* now we have:
3600 * rtp/pack/name/start/name
3601 * p4 p3 p2 p1
3602 *
3603 * find the part up to "pack" in 'runtimepath' */
3604 c = *++p4; /* append pathsep in order to expand symlink */
3605 *p4 = NUL;
3606 ffname = fix_fname(fname);
3607 *p4 = c;
Bram Moolenaar91715872016-03-03 17:13:03 +01003608 if (ffname == NULL)
Bram Moolenaar2374faa2018-02-04 17:47:42 +01003609 return FAIL;
3610
3611 /* Find "ffname" in "p_rtp", ignoring '/' vs '\' differences. */
3612 fname_len = STRLEN(ffname);
3613 insp = p_rtp;
3614 buf = alloc(MAXPATHL);
3615 if (buf == NULL)
3616 goto theend;
3617 while (*insp != NUL)
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003618 {
Bram Moolenaar2374faa2018-02-04 17:47:42 +01003619 copy_option_part(&insp, buf, MAXPATHL, ",");
3620 add_pathsep(buf);
3621 rtp_ffname = fix_fname(buf);
3622 if (rtp_ffname == NULL)
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003623 goto theend;
Bram Moolenaar2374faa2018-02-04 17:47:42 +01003624 match = vim_fnamencmp(rtp_ffname, ffname, fname_len) == 0;
3625 vim_free(rtp_ffname);
3626 if (match)
3627 break;
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003628 }
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003629
Bram Moolenaar2374faa2018-02-04 17:47:42 +01003630 if (*insp == NUL)
3631 /* not found, append at the end */
3632 insp = p_rtp + STRLEN(p_rtp);
3633 else
3634 /* append after the matching directory. */
3635 --insp;
3636
3637 /* check if rtp/pack/name/start/name/after exists */
3638 afterdir = concat_fnames(fname, (char_u *)"after", TRUE);
3639 if (afterdir != NULL && mch_isdir(afterdir))
3640 afterlen = STRLEN(afterdir) + 1; /* add one for comma */
3641
3642 oldlen = STRLEN(p_rtp);
3643 addlen = STRLEN(fname) + 1; /* add one for comma */
3644 new_rtp = alloc((int)(oldlen + addlen + afterlen + 1));
3645 /* add one for NUL */
3646 if (new_rtp == NULL)
3647 goto theend;
3648 keep = (int)(insp - p_rtp);
3649 mch_memmove(new_rtp, p_rtp, keep);
3650 new_rtp[keep] = ',';
3651 mch_memmove(new_rtp + keep + 1, fname, addlen);
3652 if (p_rtp[keep] != NUL)
3653 mch_memmove(new_rtp + keep + addlen, p_rtp + keep, oldlen - keep + 1);
3654 if (afterlen > 0)
Bram Moolenaarf3654822016-03-04 22:12:23 +01003655 {
Bram Moolenaar2374faa2018-02-04 17:47:42 +01003656 STRCAT(new_rtp, ",");
3657 STRCAT(new_rtp, afterdir);
Bram Moolenaarf3654822016-03-04 22:12:23 +01003658 }
Bram Moolenaar2374faa2018-02-04 17:47:42 +01003659 set_option_value((char_u *)"rtp", 0L, new_rtp, 0);
3660 vim_free(new_rtp);
3661 retval = OK;
Bram Moolenaarf3654822016-03-04 22:12:23 +01003662
3663theend:
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003664 vim_free(buf);
Bram Moolenaarf3654822016-03-04 22:12:23 +01003665 vim_free(ffname);
Bram Moolenaar2374faa2018-02-04 17:47:42 +01003666 vim_free(afterdir);
3667 return retval;
3668}
3669
3670/*
3671 * Load scripts in "plugin" and "ftdetect" directories of the package.
3672 */
3673 static int
3674load_pack_plugin(char_u *fname)
3675{
3676 static char *plugpat = "%s/plugin/**/*.vim";
3677 static char *ftpat = "%s/ftdetect/*.vim";
3678 int len;
3679 char_u *ffname = fix_fname(fname);
3680 char_u *pat = NULL;
3681 int retval = FAIL;
3682
3683 if (ffname == NULL)
3684 return FAIL;
3685 len = (int)STRLEN(ffname) + (int)STRLEN(ftpat);
3686 pat = alloc(len);
3687 if (pat == NULL)
3688 goto theend;
3689 vim_snprintf((char *)pat, len, plugpat, ffname);
3690 source_all_matches(pat);
3691
3692#ifdef FEAT_AUTOCMD
3693 {
3694 char_u *cmd = vim_strsave((char_u *)"g:did_load_filetypes");
3695
3696 /* If runtime/filetype.vim wasn't loaded yet, the scripts will be
3697 * found when it loads. */
3698 if (cmd != NULL && eval_to_number(cmd) > 0)
3699 {
3700 do_cmdline_cmd((char_u *)"augroup filetypedetect");
3701 vim_snprintf((char *)pat, len, ftpat, ffname);
3702 source_all_matches(pat);
3703 do_cmdline_cmd((char_u *)"augroup END");
3704 }
3705 vim_free(cmd);
3706 }
3707#endif
3708 vim_free(pat);
3709 retval = OK;
3710
3711theend:
3712 vim_free(ffname);
3713 return retval;
3714}
3715
3716/* used for "cookie" of add_pack_plugin() */
3717static int APP_ADD_DIR;
3718static int APP_LOAD;
3719static int APP_BOTH;
3720
3721 static void
3722add_pack_plugin(char_u *fname, void *cookie)
3723{
3724 if (cookie != &APP_LOAD && strstr((char *)p_rtp, (char *)fname) == NULL)
3725 /* directory is not yet in 'runtimepath', add it */
3726 if (add_pack_dir_to_rtp(fname) == FAIL)
3727 return;
3728
3729 if (cookie != &APP_ADD_DIR)
3730 load_pack_plugin(fname);
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003731}
3732
Bram Moolenaarce876aa2017-06-04 17:47:42 +02003733/*
3734 * Add all packages in the "start" directory to 'runtimepath'.
3735 */
3736 void
3737add_pack_start_dirs(void)
3738{
3739 do_in_path(p_pp, (char_u *)"pack/*/start/*", DIP_ALL + DIP_DIR,
3740 add_pack_plugin, &APP_ADD_DIR);
3741}
3742
3743/*
3744 * Load plugins from all packages in the "start" directory.
3745 */
3746 void
3747load_start_packages(void)
3748{
3749 did_source_packages = TRUE;
3750 do_in_path(p_pp, (char_u *)"pack/*/start/*", DIP_ALL + DIP_DIR,
3751 add_pack_plugin, &APP_LOAD);
3752}
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003753
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003754/*
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003755 * ":packloadall"
Bram Moolenaarf3654822016-03-04 22:12:23 +01003756 * Find plugins in the package directories and source them.
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003757 */
3758 void
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003759ex_packloadall(exarg_T *eap)
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003760{
Bram Moolenaarce876aa2017-06-04 17:47:42 +02003761 if (!did_source_packages || eap->forceit)
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003762 {
Bram Moolenaar49b27322016-04-05 21:13:00 +02003763 /* First do a round to add all directories to 'runtimepath', then load
3764 * the plugins. This allows for plugins to use an autoload directory
3765 * of another plugin. */
Bram Moolenaarce876aa2017-06-04 17:47:42 +02003766 add_pack_start_dirs();
3767 load_start_packages();
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003768 }
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003769}
3770
3771/*
Bram Moolenaarf3654822016-03-04 22:12:23 +01003772 * ":packadd[!] {name}"
Bram Moolenaar91715872016-03-03 17:13:03 +01003773 */
3774 void
3775ex_packadd(exarg_T *eap)
3776{
Bram Moolenaar9e1d3992017-12-17 14:26:46 +01003777 static char *plugpat = "pack/*/%s/%s";
Bram Moolenaar91715872016-03-03 17:13:03 +01003778 int len;
3779 char *pat;
Bram Moolenaar9e1d3992017-12-17 14:26:46 +01003780 int round;
3781 int res = OK;
Bram Moolenaar91715872016-03-03 17:13:03 +01003782
Bram Moolenaar9e1d3992017-12-17 14:26:46 +01003783 /* Round 1: use "start", round 2: use "opt". */
3784 for (round = 1; round <= 2; ++round)
3785 {
3786 /* Only look under "start" when loading packages wasn't done yet. */
3787 if (round == 1 && did_source_packages)
3788 continue;
3789
3790 len = (int)STRLEN(plugpat) + (int)STRLEN(eap->arg) + 5;
3791 pat = (char *)alloc(len);
3792 if (pat == NULL)
3793 return;
3794 vim_snprintf(pat, len, plugpat, round == 1 ? "start" : "opt", eap->arg);
3795 /* The first round don't give a "not found" error, in the second round
3796 * only when nothing was found in the first round. */
3797 res = do_in_path(p_pp, (char_u *)pat,
3798 DIP_ALL + DIP_DIR + (round == 2 && res == FAIL ? DIP_ERR : 0),
3799 add_pack_plugin, eap->forceit ? &APP_ADD_DIR : &APP_BOTH);
3800 vim_free(pat);
3801 }
Bram Moolenaar91715872016-03-03 17:13:03 +01003802}
3803
Bram Moolenaar071d4272004-06-13 20:20:40 +00003804#if defined(FEAT_EVAL) && defined(FEAT_AUTOCMD)
3805/*
3806 * ":options"
3807 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003808 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003809ex_options(
3810 exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003811{
Bram Moolenaarab6c8582017-08-11 17:15:09 +02003812 vim_setenv((char_u *)"OPTWIN_CMD", (char_u *)(cmdmod.tab ? "tab" : ""));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003813 cmd_source((char_u *)SYS_OPTWIN_FILE, NULL);
3814}
3815#endif
3816
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01003817#if defined(FEAT_PYTHON3) || defined(FEAT_PYTHON) || defined(PROTO)
3818
3819# if (defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)) || defined(PROTO)
3820/*
3821 * Detect Python 3 or 2, and initialize 'pyxversion'.
3822 */
3823 void
3824init_pyxversion(void)
3825{
3826 if (p_pyx == 0)
3827 {
3828 if (python3_enabled(FALSE))
3829 p_pyx = 3;
3830 else if (python_enabled(FALSE))
3831 p_pyx = 2;
3832 }
3833}
3834# endif
3835
3836/*
3837 * Does a file contain one of the following strings at the beginning of any
3838 * line?
3839 * "#!(any string)python2" => returns 2
3840 * "#!(any string)python3" => returns 3
3841 * "# requires python 2.x" => returns 2
3842 * "# requires python 3.x" => returns 3
3843 * otherwise return 0.
3844 */
3845 static int
3846requires_py_version(char_u *filename)
3847{
3848 FILE *file;
3849 int requires_py_version = 0;
3850 int i, lines;
3851
3852 lines = (int)p_mls;
3853 if (lines < 0)
3854 lines = 5;
3855
3856 file = mch_fopen((char *)filename, "r");
3857 if (file != NULL)
3858 {
3859 for (i = 0; i < lines; i++)
3860 {
3861 if (vim_fgets(IObuff, IOSIZE, file))
3862 break;
3863 if (i == 0 && IObuff[0] == '#' && IObuff[1] == '!')
3864 {
3865 /* Check shebang. */
3866 if (strstr((char *)IObuff + 2, "python2") != NULL)
3867 {
3868 requires_py_version = 2;
3869 break;
3870 }
3871 if (strstr((char *)IObuff + 2, "python3") != NULL)
3872 {
3873 requires_py_version = 3;
3874 break;
3875 }
3876 }
3877 IObuff[21] = '\0';
3878 if (STRCMP("# requires python 2.x", IObuff) == 0)
3879 {
3880 requires_py_version = 2;
3881 break;
3882 }
3883 if (STRCMP("# requires python 3.x", IObuff) == 0)
3884 {
3885 requires_py_version = 3;
3886 break;
3887 }
3888 }
3889 fclose(file);
3890 }
3891 return requires_py_version;
3892}
3893
3894
3895/*
3896 * Source a python file using the requested python version.
3897 */
3898 static void
3899source_pyx_file(exarg_T *eap, char_u *fname)
3900{
3901 exarg_T ex;
3902 int v = requires_py_version(fname);
3903
3904# if defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)
3905 init_pyxversion();
3906# endif
3907 if (v == 0)
3908 {
3909# if defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)
3910 /* user didn't choose a preference, 'pyx' is used */
3911 v = p_pyx;
3912# elif defined(FEAT_PYTHON)
3913 v = 2;
3914# elif defined(FEAT_PYTHON3)
3915 v = 3;
3916# endif
3917 }
3918
3919 /*
3920 * now source, if required python version is not supported show
3921 * unobtrusive message.
3922 */
3923 if (eap == NULL)
3924 vim_memset(&ex, 0, sizeof(ex));
3925 else
3926 ex = *eap;
3927 ex.arg = fname;
3928 ex.cmd = (char_u *)(v == 2 ? "pyfile" : "pyfile3");
3929
3930 if (v == 2)
3931 {
3932# ifdef FEAT_PYTHON
3933 ex_pyfile(&ex);
3934# else
3935 vim_snprintf((char *)IObuff, IOSIZE,
3936 _("W20: Required python version 2.x not supported, ignoring file: %s"),
3937 fname);
3938 MSG(IObuff);
3939# endif
3940 return;
3941 }
3942 else
3943 {
3944# ifdef FEAT_PYTHON3
3945 ex_py3file(&ex);
3946# else
3947 vim_snprintf((char *)IObuff, IOSIZE,
3948 _("W21: Required python version 3.x not supported, ignoring file: %s"),
3949 fname);
3950 MSG(IObuff);
3951# endif
3952 return;
3953 }
3954}
3955
3956/*
3957 * ":pyxfile {fname}"
3958 */
3959 void
3960ex_pyxfile(exarg_T *eap)
3961{
3962 source_pyx_file(eap, eap->arg);
3963}
3964
3965/*
3966 * ":pyx"
3967 */
3968 void
3969ex_pyx(exarg_T *eap)
3970{
3971# if defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)
3972 init_pyxversion();
3973 if (p_pyx == 2)
3974 ex_python(eap);
3975 else
3976 ex_py3(eap);
3977# elif defined(FEAT_PYTHON)
3978 ex_python(eap);
3979# elif defined(FEAT_PYTHON3)
3980 ex_py3(eap);
3981# endif
3982}
3983
3984/*
3985 * ":pyxdo"
3986 */
3987 void
3988ex_pyxdo(exarg_T *eap)
3989{
3990# if defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)
3991 init_pyxversion();
3992 if (p_pyx == 2)
3993 ex_pydo(eap);
3994 else
3995 ex_py3do(eap);
3996# elif defined(FEAT_PYTHON)
3997 ex_pydo(eap);
3998# elif defined(FEAT_PYTHON3)
3999 ex_py3do(eap);
4000# endif
4001}
4002
4003#endif
4004
Bram Moolenaar071d4272004-06-13 20:20:40 +00004005/*
4006 * ":source {fname}"
4007 */
4008 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004009ex_source(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004010{
4011#ifdef FEAT_BROWSE
4012 if (cmdmod.browse)
4013 {
4014 char_u *fname = NULL;
4015
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00004016 fname = do_browse(0, (char_u *)_("Source Vim script"), eap->arg,
Bram Moolenaar071d4272004-06-13 20:20:40 +00004017 NULL, NULL, BROWSE_FILTER_MACROS, NULL);
4018 if (fname != NULL)
4019 {
4020 cmd_source(fname, eap);
4021 vim_free(fname);
4022 }
4023 }
4024 else
4025#endif
4026 cmd_source(eap->arg, eap);
4027}
4028
4029 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004030cmd_source(char_u *fname, exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004031{
4032 if (*fname == NUL)
4033 EMSG(_(e_argreq));
4034
Bram Moolenaar071d4272004-06-13 20:20:40 +00004035 else if (eap != NULL && eap->forceit)
Bram Moolenaar84a05ac2013-05-06 04:24:17 +02004036 /* ":source!": read Normal mode commands
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00004037 * Need to execute the commands directly. This is required at least
4038 * for:
Bram Moolenaar071d4272004-06-13 20:20:40 +00004039 * - ":g" command busy
4040 * - after ":argdo", ":windo" or ":bufdo"
4041 * - another command follows
4042 * - inside a loop
4043 */
4044 openscript(fname, global_busy || listcmd_busy || eap->nextcmd != NULL
4045#ifdef FEAT_EVAL
4046 || eap->cstack->cs_idx >= 0
4047#endif
4048 );
4049
4050 /* ":source" read ex commands */
Bram Moolenaar910f66f2006-04-05 20:41:53 +00004051 else if (do_source(fname, FALSE, DOSO_NONE) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004052 EMSG2(_(e_notopen), fname);
4053}
4054
4055/*
4056 * ":source" and associated commands.
4057 */
4058/*
4059 * Structure used to store info for each sourced file.
4060 * It is shared between do_source() and getsourceline().
4061 * This is required, because it needs to be handed to do_cmdline() and
4062 * sourcing can be done recursively.
4063 */
4064struct source_cookie
4065{
4066 FILE *fp; /* opened file for sourcing */
4067 char_u *nextline; /* if not NULL: line that was read ahead */
4068 int finished; /* ":finish" used */
Bram Moolenaar860cae12010-06-05 23:22:07 +02004069#if defined(USE_CRNL) || defined(USE_CR)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004070 int fileformat; /* EOL_UNKNOWN, EOL_UNIX or EOL_DOS */
4071 int error; /* TRUE if LF found after CR-LF */
4072#endif
4073#ifdef FEAT_EVAL
4074 linenr_T breakpoint; /* next line with breakpoint or zero */
4075 char_u *fname; /* name of sourced file */
4076 int dbg_tick; /* debug_tick when breakpoint was set */
4077 int level; /* top nesting level of sourced file */
4078#endif
4079#ifdef FEAT_MBYTE
4080 vimconv_T conv; /* type of conversion */
4081#endif
4082};
4083
4084#ifdef FEAT_EVAL
4085/*
4086 * Return the address holding the next breakpoint line for a source cookie.
4087 */
4088 linenr_T *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004089source_breakpoint(void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004090{
4091 return &((struct source_cookie *)cookie)->breakpoint;
4092}
4093
4094/*
4095 * Return the address holding the debug tick for a source cookie.
4096 */
4097 int *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004098source_dbg_tick(void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004099{
4100 return &((struct source_cookie *)cookie)->dbg_tick;
4101}
4102
4103/*
4104 * Return the nesting level for a source cookie.
4105 */
4106 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004107source_level(void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004108{
4109 return ((struct source_cookie *)cookie)->level;
4110}
4111#endif
4112
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01004113static char_u *get_one_sourceline(struct source_cookie *sp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004114
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004115#if (defined(WIN32) && defined(FEAT_CSCOPE)) || defined(HAVE_FD_CLOEXEC)
4116# define USE_FOPEN_NOINH
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01004117static FILE *fopen_noinh_readbin(char *filename);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004118
4119/*
4120 * Special function to open a file without handle inheritance.
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004121 * When possible the handle is closed on exec().
Bram Moolenaar071d4272004-06-13 20:20:40 +00004122 */
4123 static FILE *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004124fopen_noinh_readbin(char *filename)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004125{
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004126# ifdef WIN32
Bram Moolenaar8d8ef0b2010-01-20 21:41:47 +01004127 int fd_tmp = mch_open(filename, O_RDONLY | O_BINARY | O_NOINHERIT, 0);
4128# else
4129 int fd_tmp = mch_open(filename, O_RDONLY, 0);
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004130# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004131
4132 if (fd_tmp == -1)
4133 return NULL;
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004134
4135# ifdef HAVE_FD_CLOEXEC
4136 {
4137 int fdflags = fcntl(fd_tmp, F_GETFD);
4138 if (fdflags >= 0 && (fdflags & FD_CLOEXEC) == 0)
Bram Moolenaarcde88542015-08-11 19:14:00 +02004139 (void)fcntl(fd_tmp, F_SETFD, fdflags | FD_CLOEXEC);
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004140 }
4141# endif
4142
Bram Moolenaar071d4272004-06-13 20:20:40 +00004143 return fdopen(fd_tmp, READBIN);
4144}
4145#endif
4146
4147
4148/*
4149 * do_source: Read the file "fname" and execute its lines as EX commands.
4150 *
4151 * This function may be called recursively!
4152 *
4153 * return FAIL if file could not be opened, OK otherwise
4154 */
4155 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004156do_source(
4157 char_u *fname,
4158 int check_other, /* check for .vimrc and _vimrc */
4159 int is_vimrc) /* DOSO_ value */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004160{
4161 struct source_cookie cookie;
4162 char_u *save_sourcing_name;
4163 linenr_T save_sourcing_lnum;
4164 char_u *p;
4165 char_u *fname_exp;
Bram Moolenaar73881402009-02-04 16:50:47 +00004166 char_u *firstline = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004167 int retval = FAIL;
4168#ifdef FEAT_EVAL
4169 scid_T save_current_SID;
4170 static scid_T last_current_SID = 0;
4171 void *save_funccalp;
4172 int save_debug_break_level = debug_break_level;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004173 scriptitem_T *si = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004174# ifdef UNIX
Bram Moolenaar8767f522016-07-01 17:17:39 +02004175 stat_T st;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004176 int stat_ok;
4177# endif
4178#endif
4179#ifdef STARTUPTIME
4180 struct timeval tv_rel;
4181 struct timeval tv_start;
4182#endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00004183#ifdef FEAT_PROFILE
4184 proftime_T wait_start;
4185#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004186
Bram Moolenaar071d4272004-06-13 20:20:40 +00004187 p = expand_env_save(fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004188 if (p == NULL)
4189 return retval;
4190 fname_exp = fix_fname(p);
4191 vim_free(p);
4192 if (fname_exp == NULL)
4193 return retval;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004194 if (mch_isdir(fname_exp))
4195 {
Bram Moolenaar555b2802005-05-19 21:08:39 +00004196 smsg((char_u *)_("Cannot source a directory: \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004197 goto theend;
4198 }
4199
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00004200#ifdef FEAT_AUTOCMD
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00004201 /* Apply SourceCmd autocommands, they should get the file and source it. */
4202 if (has_autocmd(EVENT_SOURCECMD, fname_exp, NULL)
4203 && apply_autocmds(EVENT_SOURCECMD, fname_exp, fname_exp,
4204 FALSE, curbuf))
Bram Moolenaarf33943e2008-01-15 21:17:30 +00004205 {
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00004206# ifdef FEAT_EVAL
Bram Moolenaarf33943e2008-01-15 21:17:30 +00004207 retval = aborting() ? FAIL : OK;
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00004208# else
Bram Moolenaarf33943e2008-01-15 21:17:30 +00004209 retval = OK;
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00004210# endif
Bram Moolenaarf33943e2008-01-15 21:17:30 +00004211 goto theend;
4212 }
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00004213
4214 /* Apply SourcePre autocommands, they may get the file. */
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00004215 apply_autocmds(EVENT_SOURCEPRE, fname_exp, fname_exp, FALSE, curbuf);
4216#endif
4217
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004218#ifdef USE_FOPEN_NOINH
Bram Moolenaar071d4272004-06-13 20:20:40 +00004219 cookie.fp = fopen_noinh_readbin((char *)fname_exp);
4220#else
4221 cookie.fp = mch_fopen((char *)fname_exp, READBIN);
4222#endif
4223 if (cookie.fp == NULL && check_other)
4224 {
4225 /*
4226 * Try again, replacing file name ".vimrc" by "_vimrc" or vice versa,
4227 * and ".exrc" by "_exrc" or vice versa.
4228 */
4229 p = gettail(fname_exp);
4230 if ((*p == '.' || *p == '_')
4231 && (STRICMP(p + 1, "vimrc") == 0
4232 || STRICMP(p + 1, "gvimrc") == 0
4233 || STRICMP(p + 1, "exrc") == 0))
4234 {
4235 if (*p == '_')
4236 *p = '.';
4237 else
4238 *p = '_';
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004239#ifdef USE_FOPEN_NOINH
Bram Moolenaar071d4272004-06-13 20:20:40 +00004240 cookie.fp = fopen_noinh_readbin((char *)fname_exp);
4241#else
4242 cookie.fp = mch_fopen((char *)fname_exp, READBIN);
4243#endif
4244 }
4245 }
4246
4247 if (cookie.fp == NULL)
4248 {
4249 if (p_verbose > 0)
4250 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004251 verbose_enter();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004252 if (sourcing_name == NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00004253 smsg((char_u *)_("could not source \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004254 else
4255 smsg((char_u *)_("line %ld: could not source \"%s\""),
Bram Moolenaar555b2802005-05-19 21:08:39 +00004256 sourcing_lnum, fname);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004257 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004258 }
4259 goto theend;
4260 }
4261
4262 /*
4263 * The file exists.
4264 * - In verbose mode, give a message.
4265 * - For a vimrc file, may want to set 'compatible', call vimrc_found().
4266 */
4267 if (p_verbose > 1)
4268 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004269 verbose_enter();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004270 if (sourcing_name == NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00004271 smsg((char_u *)_("sourcing \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004272 else
4273 smsg((char_u *)_("line %ld: sourcing \"%s\""),
Bram Moolenaar555b2802005-05-19 21:08:39 +00004274 sourcing_lnum, fname);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004275 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004276 }
Bram Moolenaar910f66f2006-04-05 20:41:53 +00004277 if (is_vimrc == DOSO_VIMRC)
4278 vimrc_found(fname_exp, (char_u *)"MYVIMRC");
4279 else if (is_vimrc == DOSO_GVIMRC)
4280 vimrc_found(fname_exp, (char_u *)"MYGVIMRC");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004281
4282#ifdef USE_CRNL
4283 /* If no automatic file format: Set default to CR-NL. */
4284 if (*p_ffs == NUL)
4285 cookie.fileformat = EOL_DOS;
4286 else
4287 cookie.fileformat = EOL_UNKNOWN;
4288 cookie.error = FALSE;
4289#endif
4290
4291#ifdef USE_CR
4292 /* If no automatic file format: Set default to CR. */
4293 if (*p_ffs == NUL)
4294 cookie.fileformat = EOL_MAC;
4295 else
4296 cookie.fileformat = EOL_UNKNOWN;
4297 cookie.error = FALSE;
4298#endif
4299
4300 cookie.nextline = NULL;
4301 cookie.finished = FALSE;
4302
4303#ifdef FEAT_EVAL
4304 /*
4305 * Check if this script has a breakpoint.
4306 */
4307 cookie.breakpoint = dbg_find_breakpoint(TRUE, fname_exp, (linenr_T)0);
4308 cookie.fname = fname_exp;
4309 cookie.dbg_tick = debug_tick;
4310
4311 cookie.level = ex_nesting_level;
4312#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004313
4314 /*
4315 * Keep the sourcing name/lnum, for recursive calls.
4316 */
4317 save_sourcing_name = sourcing_name;
4318 sourcing_name = fname_exp;
4319 save_sourcing_lnum = sourcing_lnum;
4320 sourcing_lnum = 0;
4321
4322#ifdef STARTUPTIME
Bram Moolenaarc4e41982010-01-19 16:31:47 +01004323 if (time_fd != NULL)
4324 time_push(&tv_rel, &tv_start);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004325#endif
4326
4327#ifdef FEAT_EVAL
Bram Moolenaar05159a02005-02-26 23:04:13 +00004328# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004329 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004330 prof_child_enter(&wait_start); /* entering a child now */
4331# endif
4332
4333 /* Don't use local function variables, if called from a function.
4334 * Also starts profiling timer for nested script. */
4335 save_funccalp = save_funccal();
4336
Bram Moolenaar071d4272004-06-13 20:20:40 +00004337 /*
4338 * Check if this script was sourced before to finds its SID.
4339 * If it's new, generate a new SID.
4340 */
4341 save_current_SID = current_SID;
4342# ifdef UNIX
4343 stat_ok = (mch_stat((char *)fname_exp, &st) >= 0);
4344# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00004345 for (current_SID = script_items.ga_len; current_SID > 0; --current_SID)
4346 {
4347 si = &SCRIPT_ITEM(current_SID);
4348 if (si->sn_name != NULL
Bram Moolenaar071d4272004-06-13 20:20:40 +00004349 && (
4350# ifdef UNIX
Bram Moolenaar7c626922005-02-07 22:01:03 +00004351 /* Compare dev/ino when possible, it catches symbolic
4352 * links. Also compare file names, the inode may change
4353 * when the file was edited. */
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00004354 ((stat_ok && si->sn_dev_valid)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004355 && (si->sn_dev == st.st_dev
4356 && si->sn_ino == st.st_ino)) ||
Bram Moolenaar071d4272004-06-13 20:20:40 +00004357# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00004358 fnamecmp(si->sn_name, fname_exp) == 0))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004359 break;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004360 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004361 if (current_SID == 0)
4362 {
4363 current_SID = ++last_current_SID;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004364 if (ga_grow(&script_items, (int)(current_SID - script_items.ga_len))
4365 == FAIL)
4366 goto almosttheend;
4367 while (script_items.ga_len < current_SID)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004368 {
Bram Moolenaar05159a02005-02-26 23:04:13 +00004369 ++script_items.ga_len;
4370 SCRIPT_ITEM(script_items.ga_len).sn_name = NULL;
4371# ifdef FEAT_PROFILE
4372 SCRIPT_ITEM(script_items.ga_len).sn_prof_on = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004373# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004374 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00004375 si = &SCRIPT_ITEM(current_SID);
4376 si->sn_name = fname_exp;
4377 fname_exp = NULL;
4378# ifdef UNIX
4379 if (stat_ok)
4380 {
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00004381 si->sn_dev_valid = TRUE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004382 si->sn_dev = st.st_dev;
4383 si->sn_ino = st.st_ino;
4384 }
4385 else
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00004386 si->sn_dev_valid = FALSE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004387# endif
4388
Bram Moolenaar071d4272004-06-13 20:20:40 +00004389 /* Allocate the local script variables to use for this script. */
4390 new_script_vars(current_SID);
4391 }
4392
Bram Moolenaar05159a02005-02-26 23:04:13 +00004393# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004394 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004395 {
4396 int forceit;
4397
4398 /* Check if we do profiling for this script. */
4399 if (!si->sn_prof_on && has_profiling(TRUE, si->sn_name, &forceit))
4400 {
4401 script_do_profile(si);
4402 si->sn_pr_force = forceit;
4403 }
4404 if (si->sn_prof_on)
4405 {
4406 ++si->sn_pr_count;
4407 profile_start(&si->sn_pr_start);
4408 profile_zero(&si->sn_pr_children);
4409 }
4410 }
4411# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004412#endif
4413
Bram Moolenaar67435d92017-10-19 21:04:37 +02004414#ifdef FEAT_MBYTE
4415 cookie.conv.vc_type = CONV_NONE; /* no conversion */
4416
4417 /* Read the first line so we can check for a UTF-8 BOM. */
4418 firstline = getsourceline(0, (void *)&cookie, 0);
4419 if (firstline != NULL && STRLEN(firstline) >= 3 && firstline[0] == 0xef
4420 && firstline[1] == 0xbb && firstline[2] == 0xbf)
4421 {
4422 /* Found BOM; setup conversion, skip over BOM and recode the line. */
4423 convert_setup(&cookie.conv, (char_u *)"utf-8", p_enc);
4424 p = string_convert(&cookie.conv, firstline + 3, NULL);
4425 if (p == NULL)
4426 p = vim_strsave(firstline + 3);
4427 if (p != NULL)
4428 {
4429 vim_free(firstline);
4430 firstline = p;
4431 }
4432 }
4433#endif
4434
Bram Moolenaar071d4272004-06-13 20:20:40 +00004435 /*
4436 * Call do_cmdline, which will call getsourceline() to get the lines.
4437 */
Bram Moolenaar73881402009-02-04 16:50:47 +00004438 do_cmdline(firstline, getsourceline, (void *)&cookie,
Bram Moolenaar071d4272004-06-13 20:20:40 +00004439 DOCMD_VERBOSE|DOCMD_NOWAIT|DOCMD_REPEAT);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004440 retval = OK;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004441
4442#ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004443 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004444 {
4445 /* Get "si" again, "script_items" may have been reallocated. */
4446 si = &SCRIPT_ITEM(current_SID);
4447 if (si->sn_prof_on)
4448 {
4449 profile_end(&si->sn_pr_start);
4450 profile_sub_wait(&wait_start, &si->sn_pr_start);
4451 profile_add(&si->sn_pr_total, &si->sn_pr_start);
Bram Moolenaar1056d982006-03-09 22:37:52 +00004452 profile_self(&si->sn_pr_self, &si->sn_pr_start,
4453 &si->sn_pr_children);
Bram Moolenaar05159a02005-02-26 23:04:13 +00004454 }
4455 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004456#endif
4457
4458 if (got_int)
4459 EMSG(_(e_interr));
4460 sourcing_name = save_sourcing_name;
4461 sourcing_lnum = save_sourcing_lnum;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004462 if (p_verbose > 1)
4463 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004464 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00004465 smsg((char_u *)_("finished sourcing %s"), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004466 if (sourcing_name != NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00004467 smsg((char_u *)_("continuing in %s"), sourcing_name);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004468 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004469 }
4470#ifdef STARTUPTIME
Bram Moolenaarc4e41982010-01-19 16:31:47 +01004471 if (time_fd != NULL)
4472 {
4473 vim_snprintf((char *)IObuff, IOSIZE, "sourcing %s", fname);
4474 time_msg((char *)IObuff, &tv_start);
4475 time_pop(&tv_rel);
4476 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004477#endif
4478
4479#ifdef FEAT_EVAL
4480 /*
4481 * After a "finish" in debug mode, need to break at first command of next
4482 * sourced file.
4483 */
4484 if (save_debug_break_level > ex_nesting_level
4485 && debug_break_level == ex_nesting_level)
4486 ++debug_break_level;
4487#endif
4488
Bram Moolenaar05159a02005-02-26 23:04:13 +00004489#ifdef FEAT_EVAL
4490almosttheend:
4491 current_SID = save_current_SID;
4492 restore_funccal(save_funccalp);
4493# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004494 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004495 prof_child_exit(&wait_start); /* leaving a child now */
4496# endif
4497#endif
4498 fclose(cookie.fp);
4499 vim_free(cookie.nextline);
Bram Moolenaar73881402009-02-04 16:50:47 +00004500 vim_free(firstline);
Bram Moolenaar05159a02005-02-26 23:04:13 +00004501#ifdef FEAT_MBYTE
4502 convert_setup(&cookie.conv, NULL, NULL);
4503#endif
4504
Bram Moolenaar071d4272004-06-13 20:20:40 +00004505theend:
4506 vim_free(fname_exp);
4507 return retval;
4508}
4509
4510#if defined(FEAT_EVAL) || defined(PROTO)
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00004511
Bram Moolenaar071d4272004-06-13 20:20:40 +00004512/*
4513 * ":scriptnames"
4514 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004515 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004516ex_scriptnames(exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004517{
4518 int i;
4519
Bram Moolenaar05159a02005-02-26 23:04:13 +00004520 for (i = 1; i <= script_items.ga_len && !got_int; ++i)
4521 if (SCRIPT_ITEM(i).sn_name != NULL)
Bram Moolenaard58ea072011-06-26 04:25:30 +02004522 {
4523 home_replace(NULL, SCRIPT_ITEM(i).sn_name,
4524 NameBuff, MAXPATHL, TRUE);
4525 smsg((char_u *)"%3d: %s", i, NameBuff);
Bram Moolenaar970a1b82012-03-23 18:39:18 +01004526 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004527}
4528
4529# if defined(BACKSLASH_IN_FILENAME) || defined(PROTO)
4530/*
4531 * Fix slashes in the list of script names for 'shellslash'.
4532 */
4533 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004534scriptnames_slash_adjust(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004535{
4536 int i;
4537
Bram Moolenaar05159a02005-02-26 23:04:13 +00004538 for (i = 1; i <= script_items.ga_len; ++i)
4539 if (SCRIPT_ITEM(i).sn_name != NULL)
4540 slash_adjust(SCRIPT_ITEM(i).sn_name);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004541}
4542# endif
4543
4544/*
4545 * Get a pointer to a script name. Used for ":verbose set".
4546 */
4547 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004548get_scriptname(scid_T id)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004549{
4550 if (id == SID_MODELINE)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00004551 return (char_u *)_("modeline");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004552 if (id == SID_CMDARG)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00004553 return (char_u *)_("--cmd argument");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004554 if (id == SID_CARG)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00004555 return (char_u *)_("-c argument");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004556 if (id == SID_ENV)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00004557 return (char_u *)_("environment variable");
4558 if (id == SID_ERROR)
4559 return (char_u *)_("error handler");
Bram Moolenaar05159a02005-02-26 23:04:13 +00004560 return SCRIPT_ITEM(id).sn_name;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004561}
Bram Moolenaar05159a02005-02-26 23:04:13 +00004562
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00004563# if defined(EXITFREE) || defined(PROTO)
4564 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004565free_scriptnames(void)
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00004566{
4567 int i;
4568
4569 for (i = script_items.ga_len; i > 0; --i)
4570 vim_free(SCRIPT_ITEM(i).sn_name);
4571 ga_clear(&script_items);
4572}
4573# endif
4574
Bram Moolenaar071d4272004-06-13 20:20:40 +00004575#endif
4576
4577#if defined(USE_CR) || defined(PROTO)
4578
4579# if defined(__MSL__) && (__MSL__ >= 22)
4580/*
4581 * Newer version of the Metrowerks library handle DOS and UNIX files
4582 * without help.
4583 * Test with earlier versions, MSL 2.2 is the library supplied with
4584 * Codewarrior Pro 2.
4585 */
4586 char *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004587fgets_cr(char *s, int n, FILE *stream)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004588{
4589 return fgets(s, n, stream);
4590}
4591# else
4592/*
4593 * Version of fgets() which also works for lines ending in a <CR> only
4594 * (Macintosh format).
4595 * For older versions of the Metrowerks library.
4596 * At least CodeWarrior 9 needed this code.
4597 */
4598 char *
Bram Moolenaard14e00e2016-01-31 17:30:51 +01004599fgets_cr(char *s, int n, FILE *stream)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004600{
4601 int c = 0;
4602 int char_read = 0;
4603
4604 while (!feof(stream) && c != '\r' && c != '\n' && char_read < n - 1)
4605 {
4606 c = fgetc(stream);
4607 s[char_read++] = c;
4608 /* If the file is in DOS format, we need to skip a NL after a CR. I
4609 * thought it was the other way around, but this appears to work... */
4610 if (c == '\n')
4611 {
4612 c = fgetc(stream);
4613 if (c != '\r')
4614 ungetc(c, stream);
4615 }
4616 }
4617
4618 s[char_read] = 0;
4619 if (char_read == 0)
4620 return NULL;
4621
4622 if (feof(stream) && char_read == 1)
4623 return NULL;
4624
4625 return s;
4626}
4627# endif
4628#endif
4629
4630/*
4631 * Get one full line from a sourced file.
4632 * Called by do_cmdline() when it's called from do_source().
4633 *
4634 * Return a pointer to the line in allocated memory.
4635 * Return NULL for end-of-file or some error.
4636 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004637 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004638getsourceline(int c UNUSED, void *cookie, int indent UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004639{
4640 struct source_cookie *sp = (struct source_cookie *)cookie;
4641 char_u *line;
Bram Moolenaarc047b9a2012-02-11 20:40:55 +01004642 char_u *p;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004643
4644#ifdef FEAT_EVAL
4645 /* If breakpoints have been added/deleted need to check for it. */
4646 if (sp->dbg_tick < debug_tick)
4647 {
4648 sp->breakpoint = dbg_find_breakpoint(TRUE, sp->fname, sourcing_lnum);
4649 sp->dbg_tick = debug_tick;
4650 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00004651# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004652 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004653 script_line_end();
4654# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004655#endif
4656 /*
4657 * Get current line. If there is a read-ahead line, use it, otherwise get
4658 * one now.
4659 */
4660 if (sp->finished)
4661 line = NULL;
4662 else if (sp->nextline == NULL)
4663 line = get_one_sourceline(sp);
4664 else
4665 {
4666 line = sp->nextline;
4667 sp->nextline = NULL;
4668 ++sourcing_lnum;
4669 }
Bram Moolenaare2cc9702005-03-15 22:43:58 +00004670#ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004671 if (line != NULL && do_profiling == PROF_YES)
Bram Moolenaare2cc9702005-03-15 22:43:58 +00004672 script_line_start();
4673#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004674
4675 /* Only concatenate lines starting with a \ when 'cpoptions' doesn't
4676 * contain the 'C' flag. */
4677 if (line != NULL && (vim_strchr(p_cpo, CPO_CONCAT) == NULL))
4678 {
4679 /* compensate for the one line read-ahead */
4680 --sourcing_lnum;
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004681
4682 /* Get the next line and concatenate it when it starts with a
4683 * backslash. We always need to read the next line, keep it in
4684 * sp->nextline. */
4685 sp->nextline = get_one_sourceline(sp);
4686 if (sp->nextline != NULL && *(p = skipwhite(sp->nextline)) == '\\')
Bram Moolenaar071d4272004-06-13 20:20:40 +00004687 {
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004688 garray_T ga;
4689
Bram Moolenaarb549a732012-02-22 18:29:33 +01004690 ga_init2(&ga, (int)sizeof(char_u), 400);
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004691 ga_concat(&ga, line);
4692 ga_concat(&ga, p + 1);
4693 for (;;)
4694 {
4695 vim_free(sp->nextline);
4696 sp->nextline = get_one_sourceline(sp);
4697 if (sp->nextline == NULL)
4698 break;
4699 p = skipwhite(sp->nextline);
4700 if (*p != '\\')
4701 break;
Bram Moolenaarb549a732012-02-22 18:29:33 +01004702 /* Adjust the growsize to the current length to speed up
4703 * concatenating many lines. */
4704 if (ga.ga_len > 400)
4705 {
4706 if (ga.ga_len > 8000)
4707 ga.ga_growsize = 8000;
4708 else
4709 ga.ga_growsize = ga.ga_len;
4710 }
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004711 ga_concat(&ga, p + 1);
4712 }
4713 ga_append(&ga, NUL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004714 vim_free(line);
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004715 line = ga.ga_data;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004716 }
4717 }
4718
4719#ifdef FEAT_MBYTE
4720 if (line != NULL && sp->conv.vc_type != CONV_NONE)
4721 {
Bram Moolenaarc047b9a2012-02-11 20:40:55 +01004722 char_u *s;
4723
Bram Moolenaar071d4272004-06-13 20:20:40 +00004724 /* Convert the encoding of the script line. */
4725 s = string_convert(&sp->conv, line, NULL);
4726 if (s != NULL)
4727 {
4728 vim_free(line);
4729 line = s;
4730 }
4731 }
4732#endif
4733
4734#ifdef FEAT_EVAL
4735 /* Did we encounter a breakpoint? */
4736 if (sp->breakpoint != 0 && sp->breakpoint <= sourcing_lnum)
4737 {
4738 dbg_breakpoint(sp->fname, sourcing_lnum);
4739 /* Find next breakpoint. */
4740 sp->breakpoint = dbg_find_breakpoint(TRUE, sp->fname, sourcing_lnum);
4741 sp->dbg_tick = debug_tick;
4742 }
4743#endif
4744
4745 return line;
4746}
4747
4748 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004749get_one_sourceline(struct source_cookie *sp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004750{
4751 garray_T ga;
4752 int len;
4753 int c;
4754 char_u *buf;
4755#ifdef USE_CRNL
4756 int has_cr; /* CR-LF found */
4757#endif
4758#ifdef USE_CR
4759 char_u *scan;
4760#endif
4761 int have_read = FALSE;
4762
4763 /* use a growarray to store the sourced line */
Bram Moolenaar6ac54292005-02-02 23:07:25 +00004764 ga_init2(&ga, 1, 250);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004765
4766 /*
4767 * Loop until there is a finished line (or end-of-file).
4768 */
4769 sourcing_lnum++;
4770 for (;;)
4771 {
Bram Moolenaar6ac54292005-02-02 23:07:25 +00004772 /* make room to read at least 120 (more) characters */
4773 if (ga_grow(&ga, 120) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004774 break;
4775 buf = (char_u *)ga.ga_data;
4776
4777#ifdef USE_CR
4778 if (sp->fileformat == EOL_MAC)
4779 {
Bram Moolenaar86b68352004-12-27 21:59:20 +00004780 if (fgets_cr((char *)buf + ga.ga_len, ga.ga_maxlen - ga.ga_len,
4781 sp->fp) == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004782 break;
4783 }
4784 else
4785#endif
Bram Moolenaar86b68352004-12-27 21:59:20 +00004786 if (fgets((char *)buf + ga.ga_len, ga.ga_maxlen - ga.ga_len,
4787 sp->fp) == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004788 break;
Bram Moolenaar6ac54292005-02-02 23:07:25 +00004789 len = ga.ga_len + (int)STRLEN(buf + ga.ga_len);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004790#ifdef USE_CRNL
4791 /* Ignore a trailing CTRL-Z, when in Dos mode. Only recognize the
4792 * CTRL-Z by its own, or after a NL. */
4793 if ( (len == 1 || (len >= 2 && buf[len - 2] == '\n'))
4794 && sp->fileformat == EOL_DOS
4795 && buf[len - 1] == Ctrl_Z)
4796 {
4797 buf[len - 1] = NUL;
4798 break;
4799 }
4800#endif
4801
4802#ifdef USE_CR
4803 /* If the read doesn't stop on a new line, and there's
4804 * some CR then we assume a Mac format */
4805 if (sp->fileformat == EOL_UNKNOWN)
4806 {
4807 if (buf[len - 1] != '\n' && vim_strchr(buf, '\r') != NULL)
4808 sp->fileformat = EOL_MAC;
4809 else
4810 sp->fileformat = EOL_UNIX;
4811 }
4812
4813 if (sp->fileformat == EOL_MAC)
4814 {
4815 scan = vim_strchr(buf, '\r');
4816
4817 if (scan != NULL)
4818 {
4819 *scan = '\n';
4820 if (*(scan + 1) != 0)
4821 {
4822 *(scan + 1) = 0;
4823 fseek(sp->fp, (long)(scan - buf - len + 1), SEEK_CUR);
4824 }
4825 }
4826 len = STRLEN(buf);
4827 }
4828#endif
4829
4830 have_read = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004831 ga.ga_len = len;
4832
4833 /* If the line was longer than the buffer, read more. */
Bram Moolenaar86b68352004-12-27 21:59:20 +00004834 if (ga.ga_maxlen - ga.ga_len == 1 && buf[len - 1] != '\n')
Bram Moolenaar071d4272004-06-13 20:20:40 +00004835 continue;
4836
4837 if (len >= 1 && buf[len - 1] == '\n') /* remove trailing NL */
4838 {
4839#ifdef USE_CRNL
4840 has_cr = (len >= 2 && buf[len - 2] == '\r');
4841 if (sp->fileformat == EOL_UNKNOWN)
4842 {
4843 if (has_cr)
4844 sp->fileformat = EOL_DOS;
4845 else
4846 sp->fileformat = EOL_UNIX;
4847 }
4848
4849 if (sp->fileformat == EOL_DOS)
4850 {
4851 if (has_cr) /* replace trailing CR */
4852 {
4853 buf[len - 2] = '\n';
4854 --len;
4855 --ga.ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004856 }
4857 else /* lines like ":map xx yy^M" will have failed */
4858 {
4859 if (!sp->error)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00004860 {
Bram Moolenaar8820b482017-03-16 17:23:31 +01004861 msg_source(HL_ATTR(HLF_W));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004862 EMSG(_("W15: Warning: Wrong line separator, ^M may be missing"));
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00004863 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004864 sp->error = TRUE;
4865 sp->fileformat = EOL_UNIX;
4866 }
4867 }
4868#endif
4869 /* The '\n' is escaped if there is an odd number of ^V's just
4870 * before it, first set "c" just before the 'V's and then check
4871 * len&c parities (is faster than ((len-c)%2 == 0)) -- Acevedo */
4872 for (c = len - 2; c >= 0 && buf[c] == Ctrl_V; c--)
4873 ;
4874 if ((len & 1) != (c & 1)) /* escaped NL, read more */
4875 {
4876 sourcing_lnum++;
4877 continue;
4878 }
4879
4880 buf[len - 1] = NUL; /* remove the NL */
4881 }
4882
4883 /*
4884 * Check for ^C here now and then, so recursive :so can be broken.
4885 */
4886 line_breakcheck();
4887 break;
4888 }
4889
4890 if (have_read)
4891 return (char_u *)ga.ga_data;
4892
4893 vim_free(ga.ga_data);
4894 return NULL;
4895}
4896
Bram Moolenaar05159a02005-02-26 23:04:13 +00004897#if defined(FEAT_PROFILE) || defined(PROTO)
4898/*
4899 * Called when starting to read a script line.
4900 * "sourcing_lnum" must be correct!
4901 * When skipping lines it may not actually be executed, but we won't find out
4902 * until later and we need to store the time now.
4903 */
4904 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004905script_line_start(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004906{
4907 scriptitem_T *si;
4908 sn_prl_T *pp;
4909
4910 if (current_SID <= 0 || current_SID > script_items.ga_len)
4911 return;
4912 si = &SCRIPT_ITEM(current_SID);
4913 if (si->sn_prof_on && sourcing_lnum >= 1)
4914 {
Bram Moolenaar8c8de832008-06-24 22:58:06 +00004915 /* Grow the array before starting the timer, so that the time spent
Bram Moolenaar05159a02005-02-26 23:04:13 +00004916 * here isn't counted. */
Bram Moolenaar67435d92017-10-19 21:04:37 +02004917 (void)ga_grow(&si->sn_prl_ga,
4918 (int)(sourcing_lnum - si->sn_prl_ga.ga_len));
Bram Moolenaar05159a02005-02-26 23:04:13 +00004919 si->sn_prl_idx = sourcing_lnum - 1;
4920 while (si->sn_prl_ga.ga_len <= si->sn_prl_idx
4921 && si->sn_prl_ga.ga_len < si->sn_prl_ga.ga_maxlen)
4922 {
4923 /* Zero counters for a line that was not used before. */
4924 pp = &PRL_ITEM(si, si->sn_prl_ga.ga_len);
4925 pp->snp_count = 0;
4926 profile_zero(&pp->sn_prl_total);
4927 profile_zero(&pp->sn_prl_self);
4928 ++si->sn_prl_ga.ga_len;
4929 }
4930 si->sn_prl_execed = FALSE;
4931 profile_start(&si->sn_prl_start);
4932 profile_zero(&si->sn_prl_children);
4933 profile_get_wait(&si->sn_prl_wait);
4934 }
4935}
4936
4937/*
4938 * Called when actually executing a function line.
4939 */
4940 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004941script_line_exec(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004942{
4943 scriptitem_T *si;
4944
4945 if (current_SID <= 0 || current_SID > script_items.ga_len)
4946 return;
4947 si = &SCRIPT_ITEM(current_SID);
4948 if (si->sn_prof_on && si->sn_prl_idx >= 0)
4949 si->sn_prl_execed = TRUE;
4950}
4951
4952/*
Bram Moolenaar67435d92017-10-19 21:04:37 +02004953 * Called when done with a script line.
Bram Moolenaar05159a02005-02-26 23:04:13 +00004954 */
4955 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004956script_line_end(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004957{
4958 scriptitem_T *si;
4959 sn_prl_T *pp;
4960
4961 if (current_SID <= 0 || current_SID > script_items.ga_len)
4962 return;
4963 si = &SCRIPT_ITEM(current_SID);
4964 if (si->sn_prof_on && si->sn_prl_idx >= 0
4965 && si->sn_prl_idx < si->sn_prl_ga.ga_len)
4966 {
4967 if (si->sn_prl_execed)
4968 {
4969 pp = &PRL_ITEM(si, si->sn_prl_idx);
4970 ++pp->snp_count;
4971 profile_end(&si->sn_prl_start);
4972 profile_sub_wait(&si->sn_prl_wait, &si->sn_prl_start);
Bram Moolenaar05159a02005-02-26 23:04:13 +00004973 profile_add(&pp->sn_prl_total, &si->sn_prl_start);
Bram Moolenaar1056d982006-03-09 22:37:52 +00004974 profile_self(&pp->sn_prl_self, &si->sn_prl_start,
4975 &si->sn_prl_children);
Bram Moolenaar05159a02005-02-26 23:04:13 +00004976 }
4977 si->sn_prl_idx = -1;
4978 }
4979}
4980#endif
4981
Bram Moolenaar071d4272004-06-13 20:20:40 +00004982/*
4983 * ":scriptencoding": Set encoding conversion for a sourced script.
4984 * Without the multi-byte feature it's simply ignored.
4985 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004986 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004987ex_scriptencoding(exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004988{
4989#ifdef FEAT_MBYTE
4990 struct source_cookie *sp;
4991 char_u *name;
4992
4993 if (!getline_equal(eap->getline, eap->cookie, getsourceline))
4994 {
4995 EMSG(_("E167: :scriptencoding used outside of a sourced file"));
4996 return;
4997 }
4998
4999 if (*eap->arg != NUL)
5000 {
5001 name = enc_canonize(eap->arg);
5002 if (name == NULL) /* out of memory */
5003 return;
5004 }
5005 else
5006 name = eap->arg;
5007
5008 /* Setup for conversion from the specified encoding to 'encoding'. */
5009 sp = (struct source_cookie *)getline_cookie(eap->getline, eap->cookie);
5010 convert_setup(&sp->conv, name, p_enc);
5011
5012 if (name != eap->arg)
5013 vim_free(name);
5014#endif
5015}
5016
5017#if defined(FEAT_EVAL) || defined(PROTO)
5018/*
5019 * ":finish": Mark a sourced file as finished.
5020 */
5021 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005022ex_finish(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005023{
5024 if (getline_equal(eap->getline, eap->cookie, getsourceline))
5025 do_finish(eap, FALSE);
5026 else
5027 EMSG(_("E168: :finish used outside of a sourced file"));
5028}
5029
5030/*
5031 * Mark a sourced file as finished. Possibly makes the ":finish" pending.
5032 * Also called for a pending finish at the ":endtry" or after returning from
5033 * an extra do_cmdline(). "reanimate" is used in the latter case.
5034 */
5035 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005036do_finish(exarg_T *eap, int reanimate)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005037{
5038 int idx;
5039
5040 if (reanimate)
5041 ((struct source_cookie *)getline_cookie(eap->getline,
5042 eap->cookie))->finished = FALSE;
5043
5044 /*
5045 * Cleanup (and inactivate) conditionals, but stop when a try conditional
5046 * not in its finally clause (which then is to be executed next) is found.
5047 * In this case, make the ":finish" pending for execution at the ":endtry".
5048 * Otherwise, finish normally.
5049 */
5050 idx = cleanup_conditionals(eap->cstack, 0, TRUE);
5051 if (idx >= 0)
5052 {
5053 eap->cstack->cs_pending[idx] = CSTP_FINISH;
5054 report_make_pending(CSTP_FINISH, NULL);
5055 }
5056 else
5057 ((struct source_cookie *)getline_cookie(eap->getline,
5058 eap->cookie))->finished = TRUE;
5059}
5060
5061
5062/*
5063 * Return TRUE when a sourced file had the ":finish" command: Don't give error
5064 * message for missing ":endif".
5065 * Return FALSE when not sourcing a file.
5066 */
5067 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005068source_finished(
5069 char_u *(*fgetline)(int, void *, int),
5070 void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005071{
Bram Moolenaar89d40322006-08-29 15:30:07 +00005072 return (getline_equal(fgetline, cookie, getsourceline)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005073 && ((struct source_cookie *)getline_cookie(
Bram Moolenaar89d40322006-08-29 15:30:07 +00005074 fgetline, cookie))->finished);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005075}
5076#endif
5077
5078#if defined(FEAT_LISTCMDS) || defined(PROTO)
5079/*
5080 * ":checktime [buffer]"
5081 */
5082 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005083ex_checktime(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005084{
5085 buf_T *buf;
5086 int save_no_check_timestamps = no_check_timestamps;
5087
5088 no_check_timestamps = 0;
5089 if (eap->addr_count == 0) /* default is all buffers */
5090 check_timestamps(FALSE);
5091 else
5092 {
5093 buf = buflist_findnr((int)eap->line2);
5094 if (buf != NULL) /* cannot happen? */
5095 (void)buf_check_timestamp(buf, FALSE);
5096 }
5097 no_check_timestamps = save_no_check_timestamps;
5098}
5099#endif
5100
Bram Moolenaar071d4272004-06-13 20:20:40 +00005101#if (defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
5102 && (defined(FEAT_EVAL) || defined(FEAT_MULTI_LANG))
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02005103# define HAVE_GET_LOCALE_VAL
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005104static char_u *get_locale_val(int what);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005105
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005106 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005107get_locale_val(int what)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005108{
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005109 char_u *loc;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005110
Bram Moolenaar48e330a2016-02-23 14:53:34 +01005111 /* Obtain the locale value from the libraries. */
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005112 loc = (char_u *)setlocale(what, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005113
Bram Moolenaar61660ea2006-04-07 21:40:07 +00005114# ifdef WIN32
Bram Moolenaar071d4272004-06-13 20:20:40 +00005115 if (loc != NULL)
5116 {
5117 char_u *p;
5118
Bram Moolenaar61660ea2006-04-07 21:40:07 +00005119 /* setocale() returns something like "LC_COLLATE=<name>;LC_..." when
5120 * one of the values (e.g., LC_CTYPE) differs. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005121 p = vim_strchr(loc, '=');
5122 if (p != NULL)
5123 {
5124 loc = ++p;
5125 while (*p != NUL) /* remove trailing newline */
5126 {
Bram Moolenaar61660ea2006-04-07 21:40:07 +00005127 if (*p < ' ' || *p == ';')
Bram Moolenaar071d4272004-06-13 20:20:40 +00005128 {
5129 *p = NUL;
5130 break;
5131 }
5132 ++p;
5133 }
5134 }
5135 }
5136# endif
5137
5138 return loc;
5139}
5140#endif
5141
5142
5143#ifdef WIN32
5144/*
5145 * On MS-Windows locale names are strings like "German_Germany.1252", but
5146 * gettext expects "de". Try to translate one into another here for a few
5147 * supported languages.
5148 */
5149 static char_u *
5150gettext_lang(char_u *name)
5151{
5152 int i;
5153 static char *(mtable[]) = {
5154 "afrikaans", "af",
5155 "czech", "cs",
5156 "dutch", "nl",
5157 "german", "de",
5158 "english_united kingdom", "en_GB",
5159 "spanish", "es",
5160 "french", "fr",
5161 "italian", "it",
5162 "japanese", "ja",
5163 "korean", "ko",
5164 "norwegian", "no",
5165 "polish", "pl",
5166 "russian", "ru",
5167 "slovak", "sk",
5168 "swedish", "sv",
5169 "ukrainian", "uk",
5170 "chinese_china", "zh_CN",
5171 "chinese_taiwan", "zh_TW",
5172 NULL};
5173
5174 for (i = 0; mtable[i] != NULL; i += 2)
5175 if (STRNICMP(mtable[i], name, STRLEN(mtable[i])) == 0)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005176 return (char_u *)mtable[i + 1];
Bram Moolenaar071d4272004-06-13 20:20:40 +00005177 return name;
5178}
5179#endif
5180
5181#if defined(FEAT_MULTI_LANG) || defined(PROTO)
5182/*
5183 * Obtain the current messages language. Used to set the default for
5184 * 'helplang'. May return NULL or an empty string.
5185 */
5186 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005187get_mess_lang(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005188{
5189 char_u *p;
5190
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02005191# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00005192# if defined(LC_MESSAGES)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005193 p = get_locale_val(LC_MESSAGES);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005194# else
5195 /* This is necessary for Win32, where LC_MESSAGES is not defined and $LANG
Bram Moolenaar61660ea2006-04-07 21:40:07 +00005196 * may be set to the LCID number. LC_COLLATE is the best guess, LC_TIME
5197 * and LC_MONETARY may be set differently for a Japanese working in the
5198 * US. */
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005199 p = get_locale_val(LC_COLLATE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005200# endif
5201# else
5202 p = mch_getenv((char_u *)"LC_ALL");
5203 if (p == NULL || *p == NUL)
5204 {
5205 p = mch_getenv((char_u *)"LC_MESSAGES");
5206 if (p == NULL || *p == NUL)
5207 p = mch_getenv((char_u *)"LANG");
5208 }
5209# endif
5210# ifdef WIN32
5211 p = gettext_lang(p);
5212# endif
5213 return p;
5214}
5215#endif
5216
Bram Moolenaardef9e822004-12-31 20:58:58 +00005217/* Complicated #if; matches with where get_mess_env() is used below. */
5218#if (defined(FEAT_EVAL) && !((defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
5219 && defined(LC_MESSAGES))) \
5220 || ((defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
5221 && (defined(FEAT_GETTEXT) || defined(FEAT_MBYTE)) \
5222 && !defined(LC_MESSAGES))
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01005223static char_u *get_mess_env(void);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005224
5225/*
5226 * Get the language used for messages from the environment.
5227 */
5228 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005229get_mess_env(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005230{
5231 char_u *p;
5232
5233 p = mch_getenv((char_u *)"LC_ALL");
5234 if (p == NULL || *p == NUL)
5235 {
5236 p = mch_getenv((char_u *)"LC_MESSAGES");
5237 if (p == NULL || *p == NUL)
5238 {
5239 p = mch_getenv((char_u *)"LANG");
5240 if (p != NULL && VIM_ISDIGIT(*p))
5241 p = NULL; /* ignore something like "1043" */
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02005242# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00005243 if (p == NULL || *p == NUL)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005244 p = get_locale_val(LC_CTYPE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005245# endif
5246 }
5247 }
5248 return p;
5249}
5250#endif
5251
5252#if defined(FEAT_EVAL) || defined(PROTO)
5253
5254/*
5255 * Set the "v:lang" variable according to the current locale setting.
5256 * Also do "v:lc_time"and "v:ctype".
5257 */
5258 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005259set_lang_var(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005260{
5261 char_u *loc;
5262
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02005263# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005264 loc = get_locale_val(LC_CTYPE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005265# else
5266 /* setlocale() not supported: use the default value */
5267 loc = (char_u *)"C";
5268# endif
5269 set_vim_var_string(VV_CTYPE, loc, -1);
5270
5271 /* When LC_MESSAGES isn't defined use the value from $LC_MESSAGES, fall
5272 * back to LC_CTYPE if it's empty. */
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02005273# if defined(HAVE_GET_LOCALE_VAL) && defined(LC_MESSAGES)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005274 loc = get_locale_val(LC_MESSAGES);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005275# else
5276 loc = get_mess_env();
5277# endif
5278 set_vim_var_string(VV_LANG, loc, -1);
5279
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02005280# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005281 loc = get_locale_val(LC_TIME);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005282# endif
5283 set_vim_var_string(VV_LC_TIME, loc, -1);
5284}
5285#endif
5286
5287#if (defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
5288 && (defined(FEAT_GETTEXT) || defined(FEAT_MBYTE))
5289/*
5290 * ":language": Set the language (locale).
5291 */
5292 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005293ex_language(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005294{
5295 char *loc;
5296 char_u *p;
5297 char_u *name;
5298 int what = LC_ALL;
5299 char *whatstr = "";
5300#ifdef LC_MESSAGES
5301# define VIM_LC_MESSAGES LC_MESSAGES
5302#else
5303# define VIM_LC_MESSAGES 6789
5304#endif
5305
5306 name = eap->arg;
5307
5308 /* Check for "messages {name}", "ctype {name}" or "time {name}" argument.
5309 * Allow abbreviation, but require at least 3 characters to avoid
5310 * confusion with a two letter language name "me" or "ct". */
5311 p = skiptowhite(eap->arg);
Bram Moolenaar1c465442017-03-12 20:10:05 +01005312 if ((*p == NUL || VIM_ISWHITE(*p)) && p - eap->arg >= 3)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005313 {
5314 if (STRNICMP(eap->arg, "messages", p - eap->arg) == 0)
5315 {
5316 what = VIM_LC_MESSAGES;
5317 name = skipwhite(p);
5318 whatstr = "messages ";
5319 }
5320 else if (STRNICMP(eap->arg, "ctype", p - eap->arg) == 0)
5321 {
5322 what = LC_CTYPE;
5323 name = skipwhite(p);
5324 whatstr = "ctype ";
5325 }
5326 else if (STRNICMP(eap->arg, "time", p - eap->arg) == 0)
5327 {
5328 what = LC_TIME;
5329 name = skipwhite(p);
5330 whatstr = "time ";
5331 }
5332 }
5333
5334 if (*name == NUL)
5335 {
5336#ifndef LC_MESSAGES
5337 if (what == VIM_LC_MESSAGES)
5338 p = get_mess_env();
5339 else
5340#endif
5341 p = (char_u *)setlocale(what, NULL);
5342 if (p == NULL || *p == NUL)
5343 p = (char_u *)"Unknown";
5344 smsg((char_u *)_("Current %slanguage: \"%s\""), whatstr, p);
5345 }
5346 else
5347 {
5348#ifndef LC_MESSAGES
5349 if (what == VIM_LC_MESSAGES)
5350 loc = "";
5351 else
5352#endif
Bram Moolenaar8c8de832008-06-24 22:58:06 +00005353 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00005354 loc = setlocale(what, (char *)name);
Bram Moolenaar8c8de832008-06-24 22:58:06 +00005355#if defined(FEAT_FLOAT) && defined(LC_NUMERIC)
5356 /* Make sure strtod() uses a decimal point, not a comma. */
5357 setlocale(LC_NUMERIC, "C");
5358#endif
5359 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00005360 if (loc == NULL)
5361 EMSG2(_("E197: Cannot set language to \"%s\""), name);
5362 else
5363 {
5364#ifdef HAVE_NL_MSG_CAT_CNTR
5365 /* Need to do this for GNU gettext, otherwise cached translations
5366 * will be used again. */
5367 extern int _nl_msg_cat_cntr;
5368
5369 ++_nl_msg_cat_cntr;
5370#endif
Bram Moolenaarb8017e72007-05-10 18:59:07 +00005371 /* Reset $LC_ALL, otherwise it would overrule everything. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005372 vim_setenv((char_u *)"LC_ALL", (char_u *)"");
5373
5374 if (what != LC_TIME)
5375 {
5376 /* Tell gettext() what to translate to. It apparently doesn't
5377 * use the currently effective locale. Also do this when
5378 * FEAT_GETTEXT isn't defined, so that shell commands use this
5379 * value. */
5380 if (what == LC_ALL)
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00005381 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00005382 vim_setenv((char_u *)"LANG", name);
Bram Moolenaare3a0b532013-06-28 20:36:30 +02005383
5384 /* Clear $LANGUAGE because GNU gettext uses it. */
5385 vim_setenv((char_u *)"LANGUAGE", (char_u *)"");
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00005386# ifdef WIN32
5387 /* Apparently MS-Windows printf() may cause a crash when
5388 * we give it 8-bit text while it's expecting text in the
5389 * current locale. This call avoids that. */
5390 setlocale(LC_CTYPE, "C");
5391# endif
5392 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00005393 if (what != LC_CTYPE)
5394 {
5395 char_u *mname;
5396#ifdef WIN32
5397 mname = gettext_lang(name);
5398#else
5399 mname = name;
5400#endif
5401 vim_setenv((char_u *)"LC_MESSAGES", mname);
5402#ifdef FEAT_MULTI_LANG
5403 set_helplang_default(mname);
5404#endif
5405 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00005406 }
5407
5408# ifdef FEAT_EVAL
5409 /* Set v:lang, v:lc_time and v:ctype to the final result. */
5410 set_lang_var();
5411# endif
Bram Moolenaar6cc00c72011-10-20 21:41:09 +02005412# ifdef FEAT_TITLE
5413 maketitle();
5414# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00005415 }
5416 }
5417}
5418
5419# if defined(FEAT_CMDL_COMPL) || defined(PROTO)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005420
5421static char_u **locales = NULL; /* Array of all available locales */
Bram Moolenaarb8f7bd62017-01-12 20:28:25 +01005422
5423# ifndef WIN32
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005424static int did_init_locales = FALSE;
5425
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005426/* Return an array of strings for all available locales + NULL for the
5427 * last element. Return NULL in case of error. */
5428 static char_u **
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005429find_locales(void)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005430{
5431 garray_T locales_ga;
5432 char_u *loc;
5433
5434 /* Find all available locales by running command "locale -a". If this
5435 * doesn't work we won't have completion. */
5436 char_u *locale_a = get_cmd_output((char_u *)"locale -a",
Bram Moolenaar39c29ed2014-04-05 19:44:40 +02005437 NULL, SHELL_SILENT, NULL);
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005438 if (locale_a == NULL)
5439 return NULL;
5440 ga_init2(&locales_ga, sizeof(char_u *), 20);
5441
5442 /* Transform locale_a string where each locale is separated by "\n"
5443 * into an array of locale strings. */
5444 loc = (char_u *)strtok((char *)locale_a, "\n");
5445
5446 while (loc != NULL)
5447 {
5448 if (ga_grow(&locales_ga, 1) == FAIL)
5449 break;
5450 loc = vim_strsave(loc);
5451 if (loc == NULL)
5452 break;
5453
5454 ((char_u **)locales_ga.ga_data)[locales_ga.ga_len++] = loc;
5455 loc = (char_u *)strtok(NULL, "\n");
5456 }
5457 vim_free(locale_a);
5458 if (ga_grow(&locales_ga, 1) == FAIL)
5459 {
5460 ga_clear(&locales_ga);
5461 return NULL;
5462 }
5463 ((char_u **)locales_ga.ga_data)[locales_ga.ga_len] = NULL;
5464 return (char_u **)locales_ga.ga_data;
5465}
Bram Moolenaarb8f7bd62017-01-12 20:28:25 +01005466# endif
5467
5468/*
5469 * Lazy initialization of all available locales.
5470 */
5471 static void
5472init_locales(void)
5473{
5474# ifndef WIN32
5475 if (!did_init_locales)
5476 {
5477 did_init_locales = TRUE;
5478 locales = find_locales();
5479 }
5480# endif
5481}
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005482
5483# if defined(EXITFREE) || defined(PROTO)
5484 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005485free_locales(void)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005486{
5487 int i;
5488 if (locales != NULL)
5489 {
5490 for (i = 0; locales[i] != NULL; i++)
5491 vim_free(locales[i]);
5492 vim_free(locales);
5493 locales = NULL;
5494 }
5495}
5496# endif
5497
Bram Moolenaar071d4272004-06-13 20:20:40 +00005498/*
5499 * Function given to ExpandGeneric() to obtain the possible arguments of the
5500 * ":language" command.
5501 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005502 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005503get_lang_arg(expand_T *xp UNUSED, int idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005504{
5505 if (idx == 0)
5506 return (char_u *)"messages";
5507 if (idx == 1)
5508 return (char_u *)"ctype";
5509 if (idx == 2)
5510 return (char_u *)"time";
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005511
5512 init_locales();
5513 if (locales == NULL)
5514 return NULL;
5515 return locales[idx - 3];
5516}
5517
5518/*
5519 * Function given to ExpandGeneric() to obtain the available locales.
5520 */
5521 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005522get_locales(expand_T *xp UNUSED, int idx)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005523{
5524 init_locales();
5525 if (locales == NULL)
5526 return NULL;
5527 return locales[idx];
Bram Moolenaar071d4272004-06-13 20:20:40 +00005528}
5529# endif
5530
5531#endif