blob: 3682e797abe51c9c554dcb7ee52d6561c2e2a898 [file] [log] [blame]
Bram Moolenaaredf3f972016-08-29 22:49:24 +02001/* vi:set ts=8 sts=4 sw=4 noet:
Bram Moolenaar071d4272004-06-13 20:20:40 +00002 *
3 * VIM - Vi IMproved by Bram Moolenaar
4 *
5 * Do ":help uganda" in Vim to read copying and usage conditions.
6 * Do ":help credits" in Vim to see a list of people who contributed.
7 * See README.txt for an overview of the Vim source code.
8 */
9
10/*
11 * ex_cmds2.c: some more functions for command line commands
12 */
13
Bram Moolenaar071d4272004-06-13 20:20:40 +000014#include "vim.h"
Bram Moolenaar071d4272004-06-13 20:20:40 +000015#include "version.h"
16
Bram Moolenaarf28dbce2016-01-29 22:03:47 +010017static void cmd_source(char_u *fname, exarg_T *eap);
Bram Moolenaar071d4272004-06-13 20:20:40 +000018
Bram Moolenaar05159a02005-02-26 23:04:13 +000019#ifdef FEAT_EVAL
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +000020/* Growarray to store info about already sourced scripts.
Bram Moolenaar05159a02005-02-26 23:04:13 +000021 * For Unix also store the dev/ino, so that we don't have to stat() each
22 * script when going through the list. */
23typedef struct scriptitem_S
24{
25 char_u *sn_name;
26# ifdef UNIX
Bram Moolenaarbf0c4522009-05-16 19:16:33 +000027 int sn_dev_valid;
28 dev_t sn_dev;
Bram Moolenaar05159a02005-02-26 23:04:13 +000029 ino_t sn_ino;
30# endif
31# ifdef FEAT_PROFILE
32 int sn_prof_on; /* TRUE when script is/was profiled */
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +000033 int sn_pr_force; /* forceit: profile functions in this script */
Bram Moolenaar05159a02005-02-26 23:04:13 +000034 proftime_T sn_pr_child; /* time set when going into first child */
35 int sn_pr_nest; /* nesting for sn_pr_child */
36 /* profiling the script as a whole */
37 int sn_pr_count; /* nr of times sourced */
Bram Moolenaar8c8de832008-06-24 22:58:06 +000038 proftime_T sn_pr_total; /* time spent in script + children */
39 proftime_T sn_pr_self; /* time spent in script itself */
Bram Moolenaar05159a02005-02-26 23:04:13 +000040 proftime_T sn_pr_start; /* time at script start */
41 proftime_T sn_pr_children; /* time in children after script start */
42 /* profiling the script per line */
43 garray_T sn_prl_ga; /* things stored for every line */
44 proftime_T sn_prl_start; /* start time for current line */
45 proftime_T sn_prl_children; /* time spent in children for this line */
46 proftime_T sn_prl_wait; /* wait start time for current line */
47 int sn_prl_idx; /* index of line being timed; -1 if none */
48 int sn_prl_execed; /* line being timed was executed */
49# endif
50} scriptitem_T;
51
52static garray_T script_items = {0, 0, sizeof(scriptitem_T), 4, NULL};
53#define SCRIPT_ITEM(id) (((scriptitem_T *)script_items.ga_data)[(id) - 1])
54
55# ifdef FEAT_PROFILE
56/* Struct used in sn_prl_ga for every line of a script. */
57typedef struct sn_prl_S
58{
59 int snp_count; /* nr of times line was executed */
Bram Moolenaar8c8de832008-06-24 22:58:06 +000060 proftime_T sn_prl_total; /* time spent in a line + children */
61 proftime_T sn_prl_self; /* time spent in a line itself */
Bram Moolenaar05159a02005-02-26 23:04:13 +000062} sn_prl_T;
63
64# define PRL_ITEM(si, idx) (((sn_prl_T *)(si)->sn_prl_ga.ga_data)[(idx)])
65# endif
66#endif
67
Bram Moolenaar071d4272004-06-13 20:20:40 +000068#if defined(FEAT_EVAL) || defined(PROTO)
69static int debug_greedy = FALSE; /* batch mode debugging: don't save
70 and restore typeahead. */
Bram Moolenaarf1f60f82016-01-16 15:40:53 +010071static int get_maxbacktrace_level(void);
72static void do_setdebugtracelevel(char_u *arg);
73static void do_checkbacktracelevel(void);
74static void do_showbacktrace(char_u *cmd);
Bram Moolenaar071d4272004-06-13 20:20:40 +000075
76/*
77 * do_debug(): Debug mode.
78 * Repeatedly get Ex commands, until told to continue normal execution.
79 */
80 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +010081do_debug(char_u *cmd)
Bram Moolenaar071d4272004-06-13 20:20:40 +000082{
83 int save_msg_scroll = msg_scroll;
84 int save_State = State;
85 int save_did_emsg = did_emsg;
86 int save_cmd_silent = cmd_silent;
87 int save_msg_silent = msg_silent;
88 int save_emsg_silent = emsg_silent;
89 int save_redir_off = redir_off;
90 tasave_T typeaheadbuf;
Bram Moolenaaree3f7a52008-01-01 13:17:56 +000091 int typeahead_saved = FALSE;
Bram Moolenaar383c6f52008-01-04 15:01:07 +000092 int save_ignore_script = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +000093 int save_ex_normal_busy;
Bram Moolenaar071d4272004-06-13 20:20:40 +000094 int n;
95 char_u *cmdline = NULL;
96 char_u *p;
97 char *tail = NULL;
98 static int last_cmd = 0;
99#define CMD_CONT 1
100#define CMD_NEXT 2
101#define CMD_STEP 3
102#define CMD_FINISH 4
103#define CMD_QUIT 5
104#define CMD_INTERRUPT 6
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100105#define CMD_BACKTRACE 7
106#define CMD_FRAME 8
107#define CMD_UP 9
108#define CMD_DOWN 10
Bram Moolenaar071d4272004-06-13 20:20:40 +0000109
110#ifdef ALWAYS_USE_GUI
111 /* Can't do this when there is no terminal for input/output. */
112 if (!gui.in_use)
113 {
114 /* Break as soon as possible. */
115 debug_break_level = 9999;
116 return;
117 }
118#endif
119
120 /* Make sure we are in raw mode and start termcap mode. Might have side
121 * effects... */
122 settmode(TMODE_RAW);
123 starttermcap();
124
125 ++RedrawingDisabled; /* don't redisplay the window */
126 ++no_wait_return; /* don't wait for return */
127 did_emsg = FALSE; /* don't use error from debugged stuff */
128 cmd_silent = FALSE; /* display commands */
129 msg_silent = FALSE; /* display messages */
130 emsg_silent = FALSE; /* display error messages */
131 redir_off = TRUE; /* don't redirect debug commands */
132
133 State = NORMAL;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000134
135 if (!debug_did_msg)
136 MSG(_("Entering Debug mode. Type \"cont\" to continue."));
137 if (sourcing_name != NULL)
138 msg(sourcing_name);
139 if (sourcing_lnum != 0)
Bram Moolenaar555b2802005-05-19 21:08:39 +0000140 smsg((char_u *)_("line %ld: %s"), (long)sourcing_lnum, cmd);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000141 else
Bram Moolenaar555b2802005-05-19 21:08:39 +0000142 smsg((char_u *)_("cmd: %s"), cmd);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000143
144 /*
145 * Repeat getting a command and executing it.
146 */
147 for (;;)
148 {
149 msg_scroll = TRUE;
150 need_wait_return = FALSE;
Bram Moolenaar85b11762016-02-27 18:13:23 +0100151
Bram Moolenaar071d4272004-06-13 20:20:40 +0000152 /* Save the current typeahead buffer and replace it with an empty one.
153 * This makes sure we get input from the user here and don't interfere
154 * with the commands being executed. Reset "ex_normal_busy" to avoid
155 * the side effects of using ":normal". Save the stuff buffer and make
Bram Moolenaaree3f7a52008-01-01 13:17:56 +0000156 * it empty. Set ignore_script to avoid reading from script input. */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000157 save_ex_normal_busy = ex_normal_busy;
158 ex_normal_busy = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000159 if (!debug_greedy)
Bram Moolenaaree3f7a52008-01-01 13:17:56 +0000160 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000161 save_typeahead(&typeaheadbuf);
Bram Moolenaaree3f7a52008-01-01 13:17:56 +0000162 typeahead_saved = TRUE;
163 save_ignore_script = ignore_script;
164 ignore_script = TRUE;
165 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000166
Bram Moolenaardc303bc2016-05-17 17:45:38 +0200167 vim_free(cmdline);
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +0000168 cmdline = getcmdline_prompt('>', NULL, 0, EXPAND_NOTHING, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000169
Bram Moolenaaree3f7a52008-01-01 13:17:56 +0000170 if (typeahead_saved)
171 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000172 restore_typeahead(&typeaheadbuf);
Bram Moolenaaree3f7a52008-01-01 13:17:56 +0000173 ignore_script = save_ignore_script;
174 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000175 ex_normal_busy = save_ex_normal_busy;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000176
177 cmdline_row = msg_row;
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100178 msg_starthere();
Bram Moolenaar071d4272004-06-13 20:20:40 +0000179 if (cmdline != NULL)
180 {
181 /* If this is a debug command, set "last_cmd".
182 * If not, reset "last_cmd".
183 * For a blank line use previous command. */
184 p = skipwhite(cmdline);
185 if (*p != NUL)
186 {
187 switch (*p)
188 {
189 case 'c': last_cmd = CMD_CONT;
190 tail = "ont";
191 break;
192 case 'n': last_cmd = CMD_NEXT;
193 tail = "ext";
194 break;
195 case 's': last_cmd = CMD_STEP;
196 tail = "tep";
197 break;
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100198 case 'f':
199 last_cmd = 0;
200 if (p[1] == 'r')
201 {
202 last_cmd = CMD_FRAME;
203 tail = "rame";
204 }
205 else
206 {
207 last_cmd = CMD_FINISH;
208 tail = "inish";
209 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000210 break;
211 case 'q': last_cmd = CMD_QUIT;
212 tail = "uit";
213 break;
214 case 'i': last_cmd = CMD_INTERRUPT;
215 tail = "nterrupt";
216 break;
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100217 case 'b': last_cmd = CMD_BACKTRACE;
218 if (p[1] == 't')
219 tail = "t";
220 else
221 tail = "acktrace";
222 break;
223 case 'w': last_cmd = CMD_BACKTRACE;
224 tail = "here";
225 break;
226 case 'u': last_cmd = CMD_UP;
227 tail = "p";
228 break;
229 case 'd': last_cmd = CMD_DOWN;
230 tail = "own";
231 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000232 default: last_cmd = 0;
233 }
234 if (last_cmd != 0)
235 {
236 /* Check that the tail matches. */
237 ++p;
238 while (*p != NUL && *p == *tail)
239 {
240 ++p;
241 ++tail;
242 }
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100243 if (ASCII_ISALPHA(*p) && last_cmd != CMD_FRAME)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000244 last_cmd = 0;
245 }
246 }
247
248 if (last_cmd != 0)
249 {
250 /* Execute debug command: decided where to break next and
251 * return. */
252 switch (last_cmd)
253 {
254 case CMD_CONT:
255 debug_break_level = -1;
256 break;
257 case CMD_NEXT:
258 debug_break_level = ex_nesting_level;
259 break;
260 case CMD_STEP:
261 debug_break_level = 9999;
262 break;
263 case CMD_FINISH:
264 debug_break_level = ex_nesting_level - 1;
265 break;
266 case CMD_QUIT:
267 got_int = TRUE;
268 debug_break_level = -1;
269 break;
270 case CMD_INTERRUPT:
271 got_int = TRUE;
272 debug_break_level = 9999;
273 /* Do not repeat ">interrupt" cmd, continue stepping. */
274 last_cmd = CMD_STEP;
275 break;
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100276 case CMD_BACKTRACE:
277 do_showbacktrace(cmd);
278 continue;
279 case CMD_FRAME:
280 if (*p == NUL)
281 {
282 do_showbacktrace(cmd);
283 }
284 else
285 {
286 p = skipwhite(p);
287 do_setdebugtracelevel(p);
288 }
289 continue;
290 case CMD_UP:
291 debug_backtrace_level++;
292 do_checkbacktracelevel();
293 continue;
294 case CMD_DOWN:
295 debug_backtrace_level--;
296 do_checkbacktracelevel();
297 continue;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000298 }
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100299 /* Going out reset backtrace_level */
300 debug_backtrace_level = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000301 break;
302 }
303
304 /* don't debug this command */
305 n = debug_break_level;
306 debug_break_level = -1;
307 (void)do_cmdline(cmdline, getexline, NULL,
308 DOCMD_VERBOSE|DOCMD_EXCRESET);
309 debug_break_level = n;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000310 }
311 lines_left = Rows - 1;
312 }
313 vim_free(cmdline);
314
315 --RedrawingDisabled;
316 --no_wait_return;
317 redraw_all_later(NOT_VALID);
318 need_wait_return = FALSE;
319 msg_scroll = save_msg_scroll;
320 lines_left = Rows - 1;
321 State = save_State;
322 did_emsg = save_did_emsg;
323 cmd_silent = save_cmd_silent;
324 msg_silent = save_msg_silent;
325 emsg_silent = save_emsg_silent;
326 redir_off = save_redir_off;
327
328 /* Only print the message again when typing a command before coming back
329 * here. */
330 debug_did_msg = TRUE;
331}
332
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100333 static int
334get_maxbacktrace_level(void)
335{
336 char *p, *q;
Bram Moolenaardc633cf2016-04-23 14:33:19 +0200337 int maxbacktrace = 0;
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100338
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100339 if (sourcing_name != NULL)
340 {
341 p = (char *)sourcing_name;
342 while ((q = strstr(p, "..")) != NULL)
343 {
344 p = q + 2;
345 maxbacktrace++;
346 }
347 }
348 return maxbacktrace;
349}
350
351 static void
352do_setdebugtracelevel(char_u *arg)
353{
354 int level;
355
356 level = atoi((char *)arg);
357 if (*arg == '+' || level < 0)
358 debug_backtrace_level += level;
359 else
360 debug_backtrace_level = level;
361
362 do_checkbacktracelevel();
363}
364
365 static void
366do_checkbacktracelevel(void)
367{
368 if (debug_backtrace_level < 0)
369 {
370 debug_backtrace_level = 0;
371 MSG(_("frame is zero"));
372 }
373 else
374 {
375 int max = get_maxbacktrace_level();
376
377 if (debug_backtrace_level > max)
378 {
379 debug_backtrace_level = max;
380 smsg((char_u *)_("frame at highest level: %d"), max);
381 }
382 }
383}
384
385 static void
386do_showbacktrace(char_u *cmd)
387{
388 char *cur;
389 char *next;
390 int i = 0;
391 int max = get_maxbacktrace_level();
392
393 if (sourcing_name != NULL)
394 {
395 cur = (char *)sourcing_name;
396 while (!got_int)
397 {
398 next = strstr(cur, "..");
399 if (next != NULL)
400 *next = NUL;
401 if (i == max - debug_backtrace_level)
402 smsg((char_u *)"->%d %s", max - i, cur);
403 else
404 smsg((char_u *)" %d %s", max - i, cur);
405 ++i;
406 if (next == NULL)
407 break;
408 *next = '.';
409 cur = next + 2;
410 }
411 }
412 if (sourcing_lnum != 0)
413 smsg((char_u *)_("line %ld: %s"), (long)sourcing_lnum, cmd);
414 else
415 smsg((char_u *)_("cmd: %s"), cmd);
416}
417
Bram Moolenaar071d4272004-06-13 20:20:40 +0000418/*
419 * ":debug".
420 */
421 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100422ex_debug(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000423{
424 int debug_break_level_save = debug_break_level;
425
426 debug_break_level = 9999;
427 do_cmdline_cmd(eap->arg);
428 debug_break_level = debug_break_level_save;
429}
430
431static char_u *debug_breakpoint_name = NULL;
432static linenr_T debug_breakpoint_lnum;
433
434/*
435 * When debugging or a breakpoint is set on a skipped command, no debug prompt
436 * is shown by do_one_cmd(). This situation is indicated by debug_skipped, and
437 * debug_skipped_name is then set to the source name in the breakpoint case. If
438 * a skipped command decides itself that a debug prompt should be displayed, it
439 * can do so by calling dbg_check_skipped().
440 */
441static int debug_skipped;
442static char_u *debug_skipped_name;
443
444/*
445 * Go to debug mode when a breakpoint was encountered or "ex_nesting_level" is
446 * at or below the break level. But only when the line is actually
447 * executed. Return TRUE and set breakpoint_name for skipped commands that
448 * decide to execute something themselves.
449 * Called from do_one_cmd() before executing a command.
450 */
451 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100452dbg_check_breakpoint(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000453{
454 char_u *p;
455
456 debug_skipped = FALSE;
457 if (debug_breakpoint_name != NULL)
458 {
459 if (!eap->skip)
460 {
461 /* replace K_SNR with "<SNR>" */
462 if (debug_breakpoint_name[0] == K_SPECIAL
463 && debug_breakpoint_name[1] == KS_EXTRA
464 && debug_breakpoint_name[2] == (int)KE_SNR)
465 p = (char_u *)"<SNR>";
466 else
467 p = (char_u *)"";
Bram Moolenaar555b2802005-05-19 21:08:39 +0000468 smsg((char_u *)_("Breakpoint in \"%s%s\" line %ld"),
469 p,
Bram Moolenaar071d4272004-06-13 20:20:40 +0000470 debug_breakpoint_name + (*p == NUL ? 0 : 3),
471 (long)debug_breakpoint_lnum);
472 debug_breakpoint_name = NULL;
473 do_debug(eap->cmd);
474 }
475 else
476 {
477 debug_skipped = TRUE;
478 debug_skipped_name = debug_breakpoint_name;
479 debug_breakpoint_name = NULL;
480 }
481 }
482 else if (ex_nesting_level <= debug_break_level)
483 {
484 if (!eap->skip)
485 do_debug(eap->cmd);
486 else
487 {
488 debug_skipped = TRUE;
489 debug_skipped_name = NULL;
490 }
491 }
492}
493
494/*
495 * Go to debug mode if skipped by dbg_check_breakpoint() because eap->skip was
496 * set. Return TRUE when the debug mode is entered this time.
497 */
498 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100499dbg_check_skipped(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000500{
501 int prev_got_int;
502
503 if (debug_skipped)
504 {
505 /*
506 * Save the value of got_int and reset it. We don't want a previous
507 * interruption cause flushing the input buffer.
508 */
509 prev_got_int = got_int;
510 got_int = FALSE;
511 debug_breakpoint_name = debug_skipped_name;
512 /* eap->skip is TRUE */
513 eap->skip = FALSE;
514 (void)dbg_check_breakpoint(eap);
515 eap->skip = TRUE;
516 got_int |= prev_got_int;
517 return TRUE;
518 }
519 return FALSE;
520}
521
522/*
523 * The list of breakpoints: dbg_breakp.
524 * This is a grow-array of structs.
525 */
526struct debuggy
527{
528 int dbg_nr; /* breakpoint number */
529 int dbg_type; /* DBG_FUNC or DBG_FILE */
530 char_u *dbg_name; /* function or file name */
531 regprog_T *dbg_prog; /* regexp program */
532 linenr_T dbg_lnum; /* line number in function or file */
Bram Moolenaar05159a02005-02-26 23:04:13 +0000533 int dbg_forceit; /* ! used */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000534};
535
536static garray_T dbg_breakp = {0, 0, sizeof(struct debuggy), 4, NULL};
Bram Moolenaar05159a02005-02-26 23:04:13 +0000537#define BREAKP(idx) (((struct debuggy *)dbg_breakp.ga_data)[idx])
538#define DEBUGGY(gap, idx) (((struct debuggy *)gap->ga_data)[idx])
Bram Moolenaar071d4272004-06-13 20:20:40 +0000539static int last_breakp = 0; /* nr of last defined breakpoint */
540
Bram Moolenaar05159a02005-02-26 23:04:13 +0000541#ifdef FEAT_PROFILE
542/* Profiling uses file and func names similar to breakpoints. */
543static garray_T prof_ga = {0, 0, sizeof(struct debuggy), 4, NULL};
544#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000545#define DBG_FUNC 1
546#define DBG_FILE 2
547
Bram Moolenaarf28dbce2016-01-29 22:03:47 +0100548static int dbg_parsearg(char_u *arg, garray_T *gap);
549static linenr_T debuggy_find(int file,char_u *fname, linenr_T after, garray_T *gap, int *fp);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000550
551/*
Bram Moolenaar05159a02005-02-26 23:04:13 +0000552 * Parse the arguments of ":profile", ":breakadd" or ":breakdel" and put them
553 * in the entry just after the last one in dbg_breakp. Note that "dbg_name"
554 * is allocated.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000555 * Returns FAIL for failure.
556 */
557 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100558dbg_parsearg(
559 char_u *arg,
560 garray_T *gap) /* either &dbg_breakp or &prof_ga */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000561{
562 char_u *p = arg;
563 char_u *q;
564 struct debuggy *bp;
Bram Moolenaarf4b8e572004-06-24 15:53:16 +0000565 int here = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000566
Bram Moolenaar05159a02005-02-26 23:04:13 +0000567 if (ga_grow(gap, 1) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000568 return FAIL;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000569 bp = &DEBUGGY(gap, gap->ga_len);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000570
571 /* Find "func" or "file". */
572 if (STRNCMP(p, "func", 4) == 0)
573 bp->dbg_type = DBG_FUNC;
574 else if (STRNCMP(p, "file", 4) == 0)
575 bp->dbg_type = DBG_FILE;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000576 else if (
577#ifdef FEAT_PROFILE
578 gap != &prof_ga &&
579#endif
580 STRNCMP(p, "here", 4) == 0)
Bram Moolenaarf4b8e572004-06-24 15:53:16 +0000581 {
582 if (curbuf->b_ffname == NULL)
583 {
584 EMSG(_(e_noname));
585 return FAIL;
586 }
587 bp->dbg_type = DBG_FILE;
588 here = TRUE;
589 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000590 else
591 {
592 EMSG2(_(e_invarg2), p);
593 return FAIL;
594 }
595 p = skipwhite(p + 4);
596
597 /* Find optional line number. */
Bram Moolenaarf4b8e572004-06-24 15:53:16 +0000598 if (here)
599 bp->dbg_lnum = curwin->w_cursor.lnum;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000600 else if (
601#ifdef FEAT_PROFILE
602 gap != &prof_ga &&
603#endif
604 VIM_ISDIGIT(*p))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000605 {
606 bp->dbg_lnum = getdigits(&p);
607 p = skipwhite(p);
608 }
609 else
610 bp->dbg_lnum = 0;
611
612 /* Find the function or file name. Don't accept a function name with (). */
Bram Moolenaarf4b8e572004-06-24 15:53:16 +0000613 if ((!here && *p == NUL)
614 || (here && *p != NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000615 || (bp->dbg_type == DBG_FUNC && strstr((char *)p, "()") != NULL))
616 {
617 EMSG2(_(e_invarg2), arg);
618 return FAIL;
619 }
620
621 if (bp->dbg_type == DBG_FUNC)
622 bp->dbg_name = vim_strsave(p);
Bram Moolenaarf4b8e572004-06-24 15:53:16 +0000623 else if (here)
624 bp->dbg_name = vim_strsave(curbuf->b_ffname);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000625 else
626 {
627 /* Expand the file name in the same way as do_source(). This means
628 * doing it twice, so that $DIR/file gets expanded when $DIR is
629 * "~/dir". */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000630 q = expand_env_save(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000631 if (q == NULL)
632 return FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000633 p = expand_env_save(q);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000634 vim_free(q);
635 if (p == NULL)
636 return FAIL;
Bram Moolenaar843ee412004-06-30 16:16:41 +0000637 if (*p != '*')
638 {
639 bp->dbg_name = fix_fname(p);
640 vim_free(p);
641 }
642 else
643 bp->dbg_name = p;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000644 }
645
646 if (bp->dbg_name == NULL)
647 return FAIL;
648 return OK;
649}
650
651/*
652 * ":breakadd".
653 */
654 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100655ex_breakadd(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000656{
657 struct debuggy *bp;
658 char_u *pat;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000659 garray_T *gap;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000660
Bram Moolenaar05159a02005-02-26 23:04:13 +0000661 gap = &dbg_breakp;
662#ifdef FEAT_PROFILE
663 if (eap->cmdidx == CMD_profile)
664 gap = &prof_ga;
665#endif
666
667 if (dbg_parsearg(eap->arg, gap) == OK)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000668 {
Bram Moolenaar05159a02005-02-26 23:04:13 +0000669 bp = &DEBUGGY(gap, gap->ga_len);
670 bp->dbg_forceit = eap->forceit;
671
Bram Moolenaar071d4272004-06-13 20:20:40 +0000672 pat = file_pat_to_reg_pat(bp->dbg_name, NULL, NULL, FALSE);
673 if (pat != NULL)
674 {
675 bp->dbg_prog = vim_regcomp(pat, RE_MAGIC + RE_STRING);
676 vim_free(pat);
677 }
678 if (pat == NULL || bp->dbg_prog == NULL)
679 vim_free(bp->dbg_name);
680 else
681 {
682 if (bp->dbg_lnum == 0) /* default line number is 1 */
683 bp->dbg_lnum = 1;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000684#ifdef FEAT_PROFILE
685 if (eap->cmdidx != CMD_profile)
686#endif
687 {
688 DEBUGGY(gap, gap->ga_len).dbg_nr = ++last_breakp;
689 ++debug_tick;
690 }
691 ++gap->ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000692 }
693 }
694}
695
696/*
697 * ":debuggreedy".
698 */
699 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100700ex_debuggreedy(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000701{
702 if (eap->addr_count == 0 || eap->line2 != 0)
703 debug_greedy = TRUE;
704 else
705 debug_greedy = FALSE;
706}
707
708/*
Bram Moolenaard9fba312005-06-26 22:34:35 +0000709 * ":breakdel" and ":profdel".
Bram Moolenaar071d4272004-06-13 20:20:40 +0000710 */
711 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100712ex_breakdel(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000713{
714 struct debuggy *bp, *bpi;
715 int nr;
716 int todel = -1;
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000717 int del_all = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000718 int i;
719 linenr_T best_lnum = 0;
Bram Moolenaard9fba312005-06-26 22:34:35 +0000720 garray_T *gap;
721
722 gap = &dbg_breakp;
Bram Moolenaard9fba312005-06-26 22:34:35 +0000723 if (eap->cmdidx == CMD_profdel)
Bram Moolenaar38bdbd62012-06-20 15:48:57 +0200724 {
725#ifdef FEAT_PROFILE
Bram Moolenaard9fba312005-06-26 22:34:35 +0000726 gap = &prof_ga;
Bram Moolenaar38bdbd62012-06-20 15:48:57 +0200727#else
728 ex_ni(eap);
729 return;
Bram Moolenaard9fba312005-06-26 22:34:35 +0000730#endif
Bram Moolenaar38bdbd62012-06-20 15:48:57 +0200731 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000732
733 if (vim_isdigit(*eap->arg))
734 {
735 /* ":breakdel {nr}" */
736 nr = atol((char *)eap->arg);
Bram Moolenaard9fba312005-06-26 22:34:35 +0000737 for (i = 0; i < gap->ga_len; ++i)
738 if (DEBUGGY(gap, i).dbg_nr == nr)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000739 {
740 todel = i;
741 break;
742 }
743 }
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000744 else if (*eap->arg == '*')
745 {
746 todel = 0;
747 del_all = TRUE;
748 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000749 else
750 {
751 /* ":breakdel {func|file} [lnum] {name}" */
Bram Moolenaard9fba312005-06-26 22:34:35 +0000752 if (dbg_parsearg(eap->arg, gap) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000753 return;
Bram Moolenaard9fba312005-06-26 22:34:35 +0000754 bp = &DEBUGGY(gap, gap->ga_len);
755 for (i = 0; i < gap->ga_len; ++i)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000756 {
Bram Moolenaard9fba312005-06-26 22:34:35 +0000757 bpi = &DEBUGGY(gap, i);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000758 if (bp->dbg_type == bpi->dbg_type
759 && STRCMP(bp->dbg_name, bpi->dbg_name) == 0
760 && (bp->dbg_lnum == bpi->dbg_lnum
761 || (bp->dbg_lnum == 0
762 && (best_lnum == 0
763 || bpi->dbg_lnum < best_lnum))))
764 {
765 todel = i;
766 best_lnum = bpi->dbg_lnum;
767 }
768 }
769 vim_free(bp->dbg_name);
770 }
771
772 if (todel < 0)
773 EMSG2(_("E161: Breakpoint not found: %s"), eap->arg);
774 else
Bram Moolenaard9fba312005-06-26 22:34:35 +0000775 {
776 while (gap->ga_len > 0)
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000777 {
Bram Moolenaard9fba312005-06-26 22:34:35 +0000778 vim_free(DEBUGGY(gap, todel).dbg_name);
Bram Moolenaar473de612013-06-08 18:19:48 +0200779 vim_regfree(DEBUGGY(gap, todel).dbg_prog);
Bram Moolenaard9fba312005-06-26 22:34:35 +0000780 --gap->ga_len;
781 if (todel < gap->ga_len)
782 mch_memmove(&DEBUGGY(gap, todel), &DEBUGGY(gap, todel + 1),
783 (gap->ga_len - todel) * sizeof(struct debuggy));
784#ifdef FEAT_PROFILE
785 if (eap->cmdidx == CMD_breakdel)
786#endif
787 ++debug_tick;
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000788 if (!del_all)
789 break;
790 }
Bram Moolenaard9fba312005-06-26 22:34:35 +0000791
792 /* If all breakpoints were removed clear the array. */
793 if (gap->ga_len == 0)
794 ga_clear(gap);
795 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000796}
797
798/*
799 * ":breaklist".
800 */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000801 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100802ex_breaklist(exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000803{
804 struct debuggy *bp;
805 int i;
806
807 if (dbg_breakp.ga_len == 0)
808 MSG(_("No breakpoints defined"));
809 else
810 for (i = 0; i < dbg_breakp.ga_len; ++i)
811 {
812 bp = &BREAKP(i);
Bram Moolenaard58ea072011-06-26 04:25:30 +0200813 if (bp->dbg_type == DBG_FILE)
814 home_replace(NULL, bp->dbg_name, NameBuff, MAXPATHL, TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000815 smsg((char_u *)_("%3d %s %s line %ld"),
816 bp->dbg_nr,
817 bp->dbg_type == DBG_FUNC ? "func" : "file",
Bram Moolenaard58ea072011-06-26 04:25:30 +0200818 bp->dbg_type == DBG_FUNC ? bp->dbg_name : NameBuff,
Bram Moolenaar071d4272004-06-13 20:20:40 +0000819 (long)bp->dbg_lnum);
820 }
821}
822
823/*
824 * Find a breakpoint for a function or sourced file.
825 * Returns line number at which to break; zero when no matching breakpoint.
826 */
827 linenr_T
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100828dbg_find_breakpoint(
829 int file, /* TRUE for a file, FALSE for a function */
830 char_u *fname, /* file or function name */
831 linenr_T after) /* after this line number */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000832{
Bram Moolenaar05159a02005-02-26 23:04:13 +0000833 return debuggy_find(file, fname, after, &dbg_breakp, NULL);
834}
835
836#if defined(FEAT_PROFILE) || defined(PROTO)
837/*
838 * Return TRUE if profiling is on for a function or sourced file.
839 */
840 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100841has_profiling(
842 int file, /* TRUE for a file, FALSE for a function */
843 char_u *fname, /* file or function name */
844 int *fp) /* return: forceit */
Bram Moolenaar05159a02005-02-26 23:04:13 +0000845{
846 return (debuggy_find(file, fname, (linenr_T)0, &prof_ga, fp)
847 != (linenr_T)0);
848}
849#endif
850
851/*
852 * Common code for dbg_find_breakpoint() and has_profiling().
853 */
854 static linenr_T
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100855debuggy_find(
856 int file, /* TRUE for a file, FALSE for a function */
857 char_u *fname, /* file or function name */
858 linenr_T after, /* after this line number */
859 garray_T *gap, /* either &dbg_breakp or &prof_ga */
860 int *fp) /* if not NULL: return forceit */
Bram Moolenaar05159a02005-02-26 23:04:13 +0000861{
Bram Moolenaar071d4272004-06-13 20:20:40 +0000862 struct debuggy *bp;
863 int i;
864 linenr_T lnum = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000865 char_u *name = fname;
866 int prev_got_int;
867
Bram Moolenaar05159a02005-02-26 23:04:13 +0000868 /* Return quickly when there are no breakpoints. */
869 if (gap->ga_len == 0)
870 return (linenr_T)0;
871
Bram Moolenaar071d4272004-06-13 20:20:40 +0000872 /* Replace K_SNR in function name with "<SNR>". */
873 if (!file && fname[0] == K_SPECIAL)
874 {
875 name = alloc((unsigned)STRLEN(fname) + 3);
876 if (name == NULL)
877 name = fname;
878 else
879 {
880 STRCPY(name, "<SNR>");
881 STRCPY(name + 5, fname + 3);
882 }
883 }
884
Bram Moolenaar05159a02005-02-26 23:04:13 +0000885 for (i = 0; i < gap->ga_len; ++i)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000886 {
Bram Moolenaar05159a02005-02-26 23:04:13 +0000887 /* Skip entries that are not useful or are for a line that is beyond
888 * an already found breakpoint. */
889 bp = &DEBUGGY(gap, i);
890 if (((bp->dbg_type == DBG_FILE) == file && (
891#ifdef FEAT_PROFILE
892 gap == &prof_ga ||
893#endif
894 (bp->dbg_lnum > after && (lnum == 0 || bp->dbg_lnum < lnum)))))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000895 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000896 /*
Bram Moolenaar05159a02005-02-26 23:04:13 +0000897 * Save the value of got_int and reset it. We don't want a
898 * previous interruption cancel matching, only hitting CTRL-C
899 * while matching should abort it.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000900 */
901 prev_got_int = got_int;
902 got_int = FALSE;
Bram Moolenaardffa5b82014-11-19 16:38:07 +0100903 if (vim_regexec_prog(&bp->dbg_prog, FALSE, name, (colnr_T)0))
Bram Moolenaar05159a02005-02-26 23:04:13 +0000904 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000905 lnum = bp->dbg_lnum;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000906 if (fp != NULL)
907 *fp = bp->dbg_forceit;
908 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000909 got_int |= prev_got_int;
910 }
911 }
912 if (name != fname)
913 vim_free(name);
914
915 return lnum;
916}
917
918/*
919 * Called when a breakpoint was encountered.
920 */
921 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100922dbg_breakpoint(char_u *name, linenr_T lnum)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000923{
924 /* We need to check if this line is actually executed in do_one_cmd() */
925 debug_breakpoint_name = name;
926 debug_breakpoint_lnum = lnum;
927}
Bram Moolenaar05159a02005-02-26 23:04:13 +0000928
929
Bram Moolenaar433f7c82006-03-21 21:29:36 +0000930# if defined(FEAT_PROFILE) || defined(FEAT_RELTIME) || defined(PROTO)
Bram Moolenaar05159a02005-02-26 23:04:13 +0000931/*
932 * Store the current time in "tm".
933 */
934 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100935profile_start(proftime_T *tm)
Bram Moolenaar05159a02005-02-26 23:04:13 +0000936{
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000937# ifdef WIN3264
938 QueryPerformanceCounter(tm);
939# else
Bram Moolenaar05159a02005-02-26 23:04:13 +0000940 gettimeofday(tm, NULL);
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000941# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +0000942}
943
944/*
945 * Compute the elapsed time from "tm" till now and store in "tm".
946 */
947 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100948profile_end(proftime_T *tm)
Bram Moolenaar05159a02005-02-26 23:04:13 +0000949{
950 proftime_T now;
951
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000952# ifdef WIN3264
953 QueryPerformanceCounter(&now);
954 tm->QuadPart = now.QuadPart - tm->QuadPart;
955# else
Bram Moolenaar05159a02005-02-26 23:04:13 +0000956 gettimeofday(&now, NULL);
957 tm->tv_usec = now.tv_usec - tm->tv_usec;
958 tm->tv_sec = now.tv_sec - tm->tv_sec;
959 if (tm->tv_usec < 0)
960 {
961 tm->tv_usec += 1000000;
962 --tm->tv_sec;
963 }
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000964# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +0000965}
966
967/*
968 * Subtract the time "tm2" from "tm".
969 */
970 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100971profile_sub(proftime_T *tm, proftime_T *tm2)
Bram Moolenaar05159a02005-02-26 23:04:13 +0000972{
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000973# ifdef WIN3264
974 tm->QuadPart -= tm2->QuadPart;
975# else
Bram Moolenaar05159a02005-02-26 23:04:13 +0000976 tm->tv_usec -= tm2->tv_usec;
977 tm->tv_sec -= tm2->tv_sec;
978 if (tm->tv_usec < 0)
979 {
980 tm->tv_usec += 1000000;
981 --tm->tv_sec;
982 }
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000983# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +0000984}
985
986/*
Bram Moolenaar433f7c82006-03-21 21:29:36 +0000987 * Return a string that represents the time in "tm".
988 * Uses a static buffer!
989 */
990 char *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100991profile_msg(proftime_T *tm)
Bram Moolenaar433f7c82006-03-21 21:29:36 +0000992{
993 static char buf[50];
994
995# ifdef WIN3264
996 LARGE_INTEGER fr;
997
998 QueryPerformanceFrequency(&fr);
999 sprintf(buf, "%10.6lf", (double)tm->QuadPart / (double)fr.QuadPart);
1000# else
1001 sprintf(buf, "%3ld.%06ld", (long)tm->tv_sec, (long)tm->tv_usec);
Bram Moolenaar76929292008-01-06 19:07:36 +00001002# endif
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001003 return buf;
1004}
1005
Bram Moolenaar79c2c882016-02-07 21:19:28 +01001006# if defined(FEAT_FLOAT) || defined(PROTO)
1007/*
1008 * Return a float that represents the time in "tm".
1009 */
1010 float_T
1011profile_float(proftime_T *tm)
1012{
1013# ifdef WIN3264
1014 LARGE_INTEGER fr;
1015
1016 QueryPerformanceFrequency(&fr);
1017 return (float_T)tm->QuadPart / (float_T)fr.QuadPart;
1018# else
1019 return (float_T)tm->tv_sec + (float_T)tm->tv_usec / 1000000.0;
1020# endif
1021}
1022# endif
1023
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001024/*
Bram Moolenaar76929292008-01-06 19:07:36 +00001025 * Put the time "msec" past now in "tm".
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001026 */
Bram Moolenaar76929292008-01-06 19:07:36 +00001027 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001028profile_setlimit(long msec, proftime_T *tm)
Bram Moolenaar76929292008-01-06 19:07:36 +00001029{
1030 if (msec <= 0) /* no limit */
1031 profile_zero(tm);
1032 else
1033 {
1034# ifdef WIN3264
1035 LARGE_INTEGER fr;
1036
1037 QueryPerformanceCounter(tm);
1038 QueryPerformanceFrequency(&fr);
Bram Moolenaarb3c70982008-01-18 10:40:55 +00001039 tm->QuadPart += (LONGLONG)((double)msec / 1000.0 * (double)fr.QuadPart);
Bram Moolenaar76929292008-01-06 19:07:36 +00001040# else
1041 long usec;
1042
1043 gettimeofday(tm, NULL);
1044 usec = (long)tm->tv_usec + (long)msec * 1000;
1045 tm->tv_usec = usec % 1000000L;
1046 tm->tv_sec += usec / 1000000L;
1047# endif
1048 }
1049}
1050
1051/*
1052 * Return TRUE if the current time is past "tm".
1053 */
1054 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001055profile_passed_limit(proftime_T *tm)
Bram Moolenaar76929292008-01-06 19:07:36 +00001056{
1057 proftime_T now;
1058
1059# ifdef WIN3264
1060 if (tm->QuadPart == 0) /* timer was not set */
1061 return FALSE;
1062 QueryPerformanceCounter(&now);
1063 return (now.QuadPart > tm->QuadPart);
1064# else
1065 if (tm->tv_sec == 0) /* timer was not set */
1066 return FALSE;
1067 gettimeofday(&now, NULL);
1068 return (now.tv_sec > tm->tv_sec
1069 || (now.tv_sec == tm->tv_sec && now.tv_usec > tm->tv_usec));
1070# endif
1071}
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001072
1073/*
1074 * Set the time in "tm" to zero.
1075 */
1076 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001077profile_zero(proftime_T *tm)
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001078{
1079# ifdef WIN3264
1080 tm->QuadPart = 0;
1081# else
1082 tm->tv_usec = 0;
1083 tm->tv_sec = 0;
1084# endif
1085}
1086
Bram Moolenaar76929292008-01-06 19:07:36 +00001087# endif /* FEAT_PROFILE || FEAT_RELTIME */
1088
Bram Moolenaar975b5272016-03-15 23:10:59 +01001089# if defined(FEAT_TIMERS) || defined(PROTO)
1090static timer_T *first_timer = NULL;
Bram Moolenaar75537a92016-09-05 22:45:28 +02001091static long last_timer_id = 0;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001092
Bram Moolenaar75537a92016-09-05 22:45:28 +02001093# ifdef WIN3264
1094# define GET_TIMEDIFF(timer, now) \
1095 (long)(((double)(timer->tr_due.QuadPart - now.QuadPart) \
Bram Moolenaara7c023e2016-10-12 12:13:35 +02001096 / (double)fr.QuadPart) * 1000)
Bram Moolenaar75537a92016-09-05 22:45:28 +02001097# else
1098# define GET_TIMEDIFF(timer, now) \
1099 (timer->tr_due.tv_sec - now.tv_sec) * 1000 \
Bram Moolenaara7c023e2016-10-12 12:13:35 +02001100 + (timer->tr_due.tv_usec - now.tv_usec) / 1000
Bram Moolenaar75537a92016-09-05 22:45:28 +02001101# endif
Bram Moolenaar417ccd72016-09-01 21:26:20 +02001102
Bram Moolenaar975b5272016-03-15 23:10:59 +01001103/*
1104 * Insert a timer in the list of timers.
1105 */
1106 static void
1107insert_timer(timer_T *timer)
1108{
1109 timer->tr_next = first_timer;
1110 timer->tr_prev = NULL;
1111 if (first_timer != NULL)
1112 first_timer->tr_prev = timer;
1113 first_timer = timer;
Bram Moolenaar4231da42016-06-02 14:30:04 +02001114 did_add_timer = TRUE;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001115}
1116
1117/*
1118 * Take a timer out of the list of timers.
1119 */
1120 static void
1121remove_timer(timer_T *timer)
1122{
1123 if (timer->tr_prev == NULL)
1124 first_timer = timer->tr_next;
1125 else
1126 timer->tr_prev->tr_next = timer->tr_next;
1127 if (timer->tr_next != NULL)
1128 timer->tr_next->tr_prev = timer->tr_prev;
1129}
1130
1131 static void
1132free_timer(timer_T *timer)
1133{
Bram Moolenaar75537a92016-09-05 22:45:28 +02001134 free_callback(timer->tr_callback, timer->tr_partial);
1135 vim_free(timer);
Bram Moolenaar975b5272016-03-15 23:10:59 +01001136}
1137
1138/*
1139 * Create a timer and return it. NULL if out of memory.
1140 * Caller should set the callback.
1141 */
1142 timer_T *
1143create_timer(long msec, int repeat)
1144{
1145 timer_T *timer = (timer_T *)alloc_clear(sizeof(timer_T));
Bram Moolenaaree39ef02016-09-10 19:17:42 +02001146 long prev_id = last_timer_id;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001147
1148 if (timer == NULL)
1149 return NULL;
Bram Moolenaaree39ef02016-09-10 19:17:42 +02001150 if (++last_timer_id <= prev_id)
Bram Moolenaar75537a92016-09-05 22:45:28 +02001151 /* Overflow! Might cause duplicates... */
1152 last_timer_id = 0;
1153 timer->tr_id = last_timer_id;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001154 insert_timer(timer);
1155 if (repeat != 0)
Bram Moolenaar975b5272016-03-15 23:10:59 +01001156 timer->tr_repeat = repeat - 1;
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001157 timer->tr_interval = msec;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001158
1159 profile_setlimit(msec, &timer->tr_due);
1160 return timer;
1161}
1162
1163/*
1164 * Invoke the callback of "timer".
1165 */
1166 static void
1167timer_callback(timer_T *timer)
1168{
1169 typval_T rettv;
1170 int dummy;
1171 typval_T argv[2];
1172
1173 argv[0].v_type = VAR_NUMBER;
Bram Moolenaar75537a92016-09-05 22:45:28 +02001174 argv[0].vval.v_number = (varnumber_T)timer->tr_id;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001175 argv[1].v_type = VAR_UNKNOWN;
1176
1177 call_func(timer->tr_callback, (int)STRLEN(timer->tr_callback),
Bram Moolenaardf48fb42016-07-22 21:50:18 +02001178 &rettv, 1, argv, NULL, 0L, 0L, &dummy, TRUE,
Bram Moolenaar975b5272016-03-15 23:10:59 +01001179 timer->tr_partial, NULL);
1180 clear_tv(&rettv);
1181}
1182
1183/*
1184 * Call timers that are due.
1185 * Return the time in msec until the next timer is due.
Bram Moolenaarc4f83382017-07-07 14:50:44 +02001186 * Returns -1 if there are no pending timers.
Bram Moolenaar975b5272016-03-15 23:10:59 +01001187 */
1188 long
Bram Moolenaarcf089462016-06-12 21:18:43 +02001189check_due_timer(void)
Bram Moolenaar975b5272016-03-15 23:10:59 +01001190{
1191 timer_T *timer;
Bram Moolenaar75537a92016-09-05 22:45:28 +02001192 timer_T *timer_next;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001193 long this_due;
Bram Moolenaar597385a2016-03-16 23:24:43 +01001194 long next_due = -1;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001195 proftime_T now;
1196 int did_one = FALSE;
Bram Moolenaar75537a92016-09-05 22:45:28 +02001197 long current_id = last_timer_id;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001198# ifdef WIN3264
1199 LARGE_INTEGER fr;
Bram Moolenaarc577d812017-07-08 22:37:34 +02001200# endif
Bram Moolenaar975b5272016-03-15 23:10:59 +01001201
Bram Moolenaarc577d812017-07-08 22:37:34 +02001202 /* Don't run any timers while exiting or dealing with an error. */
1203 if (exiting || aborting())
Bram Moolenaarc4f83382017-07-07 14:50:44 +02001204 return next_due;
1205
Bram Moolenaarc577d812017-07-08 22:37:34 +02001206# ifdef WIN3264
Bram Moolenaar975b5272016-03-15 23:10:59 +01001207 QueryPerformanceFrequency(&fr);
1208# endif
Bram Moolenaar75537a92016-09-05 22:45:28 +02001209 profile_start(&now);
1210 for (timer = first_timer; timer != NULL && !got_int; timer = timer_next)
Bram Moolenaar975b5272016-03-15 23:10:59 +01001211 {
Bram Moolenaar75537a92016-09-05 22:45:28 +02001212 timer_next = timer->tr_next;
Bram Moolenaar417ccd72016-09-01 21:26:20 +02001213
Bram Moolenaar75537a92016-09-05 22:45:28 +02001214 if (timer->tr_id == -1 || timer->tr_firing || timer->tr_paused)
1215 continue;
1216 this_due = GET_TIMEDIFF(timer, now);
1217 if (this_due <= 1)
1218 {
Bram Moolenaar1e8e1452017-06-24 16:03:06 +02001219 int save_timer_busy = timer_busy;
1220 int save_vgetc_busy = vgetc_busy;
Bram Moolenaarc577d812017-07-08 22:37:34 +02001221 int did_emsg_save = did_emsg;
1222 int called_emsg_save = called_emsg;
1223 int did_throw_save = did_throw;
Bram Moolenaar1e8e1452017-06-24 16:03:06 +02001224
1225 timer_busy = timer_busy > 0 || vgetc_busy > 0;
1226 vgetc_busy = 0;
Bram Moolenaarc577d812017-07-08 22:37:34 +02001227 called_emsg = FALSE;
Bram Moolenaar75537a92016-09-05 22:45:28 +02001228 timer->tr_firing = TRUE;
1229 timer_callback(timer);
1230 timer->tr_firing = FALSE;
1231 timer_next = timer->tr_next;
1232 did_one = TRUE;
Bram Moolenaar1e8e1452017-06-24 16:03:06 +02001233 timer_busy = save_timer_busy;
1234 vgetc_busy = save_vgetc_busy;
Bram Moolenaarc577d812017-07-08 22:37:34 +02001235 if (called_emsg)
1236 {
1237 ++timer->tr_emsg_count;
Bram Moolenaarcae24be2017-07-10 22:12:10 +02001238 if (!did_throw_save && did_throw && current_exception != NULL)
Bram Moolenaarc577d812017-07-08 22:37:34 +02001239 discard_current_exception();
1240 }
1241 did_emsg = did_emsg_save;
1242 called_emsg = called_emsg_save;
Bram Moolenaar75537a92016-09-05 22:45:28 +02001243
1244 /* Only fire the timer again if it repeats and stop_timer() wasn't
1245 * called while inside the callback (tr_id == -1). */
Bram Moolenaarc577d812017-07-08 22:37:34 +02001246 if (timer->tr_repeat != 0 && timer->tr_id != -1
1247 && timer->tr_emsg_count < 3)
Bram Moolenaar75537a92016-09-05 22:45:28 +02001248 {
1249 profile_setlimit(timer->tr_interval, &timer->tr_due);
1250 this_due = GET_TIMEDIFF(timer, now);
1251 if (this_due < 1)
1252 this_due = 1;
1253 if (timer->tr_repeat > 0)
1254 --timer->tr_repeat;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001255 }
Bram Moolenaar75537a92016-09-05 22:45:28 +02001256 else
1257 {
1258 this_due = -1;
1259 remove_timer(timer);
1260 free_timer(timer);
1261 }
Bram Moolenaar975b5272016-03-15 23:10:59 +01001262 }
Bram Moolenaar75537a92016-09-05 22:45:28 +02001263 if (this_due > 0 && (next_due == -1 || next_due > this_due))
1264 next_due = this_due;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001265 }
1266
1267 if (did_one)
1268 redraw_after_callback();
1269
Bram Moolenaar75537a92016-09-05 22:45:28 +02001270 return current_id != last_timer_id ? 1 : next_due;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001271}
1272
1273/*
1274 * Find a timer by ID. Returns NULL if not found;
1275 */
1276 timer_T *
Bram Moolenaar75537a92016-09-05 22:45:28 +02001277find_timer(long id)
Bram Moolenaar975b5272016-03-15 23:10:59 +01001278{
1279 timer_T *timer;
1280
Bram Moolenaar75537a92016-09-05 22:45:28 +02001281 if (id >= 0)
1282 {
1283 for (timer = first_timer; timer != NULL; timer = timer->tr_next)
1284 if (timer->tr_id == id)
1285 return timer;
1286 }
1287 return NULL;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001288}
1289
1290
1291/*
1292 * Stop a timer and delete it.
1293 */
1294 void
1295stop_timer(timer_T *timer)
1296{
Bram Moolenaar75537a92016-09-05 22:45:28 +02001297 if (timer->tr_firing)
1298 /* Free the timer after the callback returns. */
1299 timer->tr_id = -1;
1300 else
1301 {
1302 remove_timer(timer);
1303 free_timer(timer);
1304 }
Bram Moolenaar975b5272016-03-15 23:10:59 +01001305}
Bram Moolenaare3188e22016-05-31 21:13:04 +02001306
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001307 void
Bram Moolenaarb73598e2016-08-07 18:22:53 +02001308stop_all_timers(void)
1309{
Bram Moolenaar75537a92016-09-05 22:45:28 +02001310 timer_T *timer;
1311 timer_T *timer_next;
1312
1313 for (timer = first_timer; timer != NULL; timer = timer_next)
1314 {
1315 timer_next = timer->tr_next;
1316 stop_timer(timer);
1317 }
Bram Moolenaarb73598e2016-08-07 18:22:53 +02001318}
1319
1320 void
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001321add_timer_info(typval_T *rettv, timer_T *timer)
1322{
1323 list_T *list = rettv->vval.v_list;
1324 dict_T *dict = dict_alloc();
1325 dictitem_T *di;
1326 long remaining;
1327 proftime_T now;
Bram Moolenaar00ff3802016-08-06 22:27:28 +02001328# ifdef WIN3264
1329 LARGE_INTEGER fr;
1330#endif
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001331
1332 if (dict == NULL)
1333 return;
1334 list_append_dict(list, dict);
1335
Bram Moolenaar75537a92016-09-05 22:45:28 +02001336 dict_add_nr_str(dict, "id", timer->tr_id, NULL);
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001337 dict_add_nr_str(dict, "time", (long)timer->tr_interval, NULL);
1338
1339 profile_start(&now);
1340# ifdef WIN3264
Bram Moolenaar00ff3802016-08-06 22:27:28 +02001341 QueryPerformanceFrequency(&fr);
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001342# endif
Bram Moolenaar75537a92016-09-05 22:45:28 +02001343 remaining = GET_TIMEDIFF(timer, now);
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001344 dict_add_nr_str(dict, "remaining", (long)remaining, NULL);
1345
1346 dict_add_nr_str(dict, "repeat",
1347 (long)(timer->tr_repeat < 0 ? -1 : timer->tr_repeat + 1), NULL);
Bram Moolenaarb73598e2016-08-07 18:22:53 +02001348 dict_add_nr_str(dict, "paused", (long)(timer->tr_paused), NULL);
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001349
1350 di = dictitem_alloc((char_u *)"callback");
1351 if (di != NULL)
1352 {
1353 if (dict_add(dict, di) == FAIL)
1354 vim_free(di);
1355 else if (timer->tr_partial != NULL)
1356 {
1357 di->di_tv.v_type = VAR_PARTIAL;
1358 di->di_tv.vval.v_partial = timer->tr_partial;
1359 ++timer->tr_partial->pt_refcount;
1360 }
1361 else
1362 {
1363 di->di_tv.v_type = VAR_FUNC;
1364 di->di_tv.vval.v_string = vim_strsave(timer->tr_callback);
1365 }
1366 di->di_tv.v_lock = 0;
1367 }
1368}
1369
1370 void
1371add_timer_info_all(typval_T *rettv)
1372{
1373 timer_T *timer;
1374
1375 for (timer = first_timer; timer != NULL; timer = timer->tr_next)
Bram Moolenaar75537a92016-09-05 22:45:28 +02001376 if (timer->tr_id != -1)
1377 add_timer_info(rettv, timer);
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001378}
1379
Bram Moolenaare3188e22016-05-31 21:13:04 +02001380/*
1381 * Mark references in partials of timers.
1382 */
1383 int
1384set_ref_in_timer(int copyID)
1385{
1386 int abort = FALSE;
1387 timer_T *timer;
1388 typval_T tv;
1389
1390 for (timer = first_timer; timer != NULL; timer = timer->tr_next)
1391 {
Bram Moolenaar1e96d9b2016-07-29 22:15:09 +02001392 if (timer->tr_partial != NULL)
1393 {
1394 tv.v_type = VAR_PARTIAL;
1395 tv.vval.v_partial = timer->tr_partial;
1396 }
1397 else
1398 {
1399 tv.v_type = VAR_FUNC;
1400 tv.vval.v_string = timer->tr_callback;
1401 }
Bram Moolenaare3188e22016-05-31 21:13:04 +02001402 abort = abort || set_ref_in_item(&tv, copyID, NULL, NULL);
1403 }
1404 return abort;
1405}
Bram Moolenaar623e2632016-07-30 22:47:56 +02001406
1407# if defined(EXITFREE) || defined(PROTO)
1408 void
1409timer_free_all()
1410{
1411 timer_T *timer;
1412
1413 while (first_timer != NULL)
1414 {
1415 timer = first_timer;
1416 remove_timer(timer);
1417 free_timer(timer);
1418 }
1419}
1420# endif
Bram Moolenaar975b5272016-03-15 23:10:59 +01001421# endif
1422
Bram Moolenaar8a7f5a22013-06-06 14:01:46 +02001423#if defined(FEAT_SYN_HL) && defined(FEAT_RELTIME) && defined(FEAT_FLOAT)
1424# if defined(HAVE_MATH_H)
1425# include <math.h>
1426# endif
1427
1428/*
1429 * Divide the time "tm" by "count" and store in "tm2".
1430 */
1431 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001432profile_divide(proftime_T *tm, int count, proftime_T *tm2)
Bram Moolenaar8a7f5a22013-06-06 14:01:46 +02001433{
1434 if (count == 0)
1435 profile_zero(tm2);
1436 else
1437 {
1438# ifdef WIN3264
1439 tm2->QuadPart = tm->QuadPart / count;
1440# else
1441 double usec = (tm->tv_sec * 1000000.0 + tm->tv_usec) / count;
1442
1443 tm2->tv_sec = floor(usec / 1000000.0);
Bram Moolenaara2e14fc2013-06-10 20:10:44 +02001444 tm2->tv_usec = vim_round(usec - (tm2->tv_sec * 1000000.0));
Bram Moolenaar8a7f5a22013-06-06 14:01:46 +02001445# endif
1446 }
1447}
1448#endif
1449
Bram Moolenaar76929292008-01-06 19:07:36 +00001450# if defined(FEAT_PROFILE) || defined(PROTO)
1451/*
1452 * Functions for profiling.
1453 */
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01001454static void script_do_profile(scriptitem_T *si);
1455static void script_dump_profile(FILE *fd);
Bram Moolenaar76929292008-01-06 19:07:36 +00001456static proftime_T prof_wait_time;
1457
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001458/*
Bram Moolenaar05159a02005-02-26 23:04:13 +00001459 * Add the time "tm2" to "tm".
1460 */
1461 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001462profile_add(proftime_T *tm, proftime_T *tm2)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001463{
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001464# ifdef WIN3264
1465 tm->QuadPart += tm2->QuadPart;
1466# else
Bram Moolenaar05159a02005-02-26 23:04:13 +00001467 tm->tv_usec += tm2->tv_usec;
1468 tm->tv_sec += tm2->tv_sec;
1469 if (tm->tv_usec >= 1000000)
1470 {
1471 tm->tv_usec -= 1000000;
1472 ++tm->tv_sec;
1473 }
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001474# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00001475}
1476
1477/*
Bram Moolenaar1056d982006-03-09 22:37:52 +00001478 * Add the "self" time from the total time and the children's time.
1479 */
1480 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001481profile_self(proftime_T *self, proftime_T *total, proftime_T *children)
Bram Moolenaar1056d982006-03-09 22:37:52 +00001482{
1483 /* Check that the result won't be negative. Can happen with recursive
1484 * calls. */
1485#ifdef WIN3264
1486 if (total->QuadPart <= children->QuadPart)
1487 return;
1488#else
1489 if (total->tv_sec < children->tv_sec
1490 || (total->tv_sec == children->tv_sec
1491 && total->tv_usec <= children->tv_usec))
1492 return;
1493#endif
1494 profile_add(self, total);
1495 profile_sub(self, children);
1496}
1497
1498/*
Bram Moolenaar05159a02005-02-26 23:04:13 +00001499 * Get the current waittime.
1500 */
1501 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001502profile_get_wait(proftime_T *tm)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001503{
1504 *tm = prof_wait_time;
1505}
1506
1507/*
1508 * Subtract the passed waittime since "tm" from "tma".
1509 */
1510 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001511profile_sub_wait(proftime_T *tm, proftime_T *tma)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001512{
1513 proftime_T tm3 = prof_wait_time;
1514
1515 profile_sub(&tm3, tm);
1516 profile_sub(tma, &tm3);
1517}
1518
1519/*
1520 * Return TRUE if "tm1" and "tm2" are equal.
1521 */
1522 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001523profile_equal(proftime_T *tm1, proftime_T *tm2)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001524{
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001525# ifdef WIN3264
1526 return (tm1->QuadPart == tm2->QuadPart);
1527# else
Bram Moolenaar05159a02005-02-26 23:04:13 +00001528 return (tm1->tv_usec == tm2->tv_usec && tm1->tv_sec == tm2->tv_sec);
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001529# endif
1530}
1531
1532/*
1533 * Return <0, 0 or >0 if "tm1" < "tm2", "tm1" == "tm2" or "tm1" > "tm2"
1534 */
1535 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001536profile_cmp(const proftime_T *tm1, const proftime_T *tm2)
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001537{
1538# ifdef WIN3264
1539 return (int)(tm2->QuadPart - tm1->QuadPart);
1540# else
1541 if (tm1->tv_sec == tm2->tv_sec)
1542 return tm2->tv_usec - tm1->tv_usec;
1543 return tm2->tv_sec - tm1->tv_sec;
1544# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00001545}
1546
Bram Moolenaar05159a02005-02-26 23:04:13 +00001547static char_u *profile_fname = NULL;
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001548static proftime_T pause_time;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001549
1550/*
1551 * ":profile cmd args"
1552 */
1553 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001554ex_profile(exarg_T *eap)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001555{
1556 char_u *e;
1557 int len;
1558
1559 e = skiptowhite(eap->arg);
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00001560 len = (int)(e - eap->arg);
Bram Moolenaar05159a02005-02-26 23:04:13 +00001561 e = skipwhite(e);
1562
1563 if (len == 5 && STRNCMP(eap->arg, "start", 5) == 0 && *e != NUL)
1564 {
1565 vim_free(profile_fname);
Bram Moolenaard94682f2015-04-13 15:37:56 +02001566 profile_fname = expand_env_save_opt(e, TRUE);
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001567 do_profiling = PROF_YES;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001568 profile_zero(&prof_wait_time);
1569 set_vim_var_nr(VV_PROFILING, 1L);
1570 }
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001571 else if (do_profiling == PROF_NONE)
Bram Moolenaar54c1b492010-02-24 14:01:28 +01001572 EMSG(_("E750: First use \":profile start {fname}\""));
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001573 else if (STRCMP(eap->arg, "pause") == 0)
1574 {
1575 if (do_profiling == PROF_YES)
1576 profile_start(&pause_time);
1577 do_profiling = PROF_PAUSED;
1578 }
1579 else if (STRCMP(eap->arg, "continue") == 0)
1580 {
1581 if (do_profiling == PROF_PAUSED)
1582 {
1583 profile_end(&pause_time);
1584 profile_add(&prof_wait_time, &pause_time);
1585 }
1586 do_profiling = PROF_YES;
1587 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00001588 else
1589 {
1590 /* The rest is similar to ":breakadd". */
1591 ex_breakadd(eap);
1592 }
1593}
1594
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001595/* Command line expansion for :profile. */
1596static enum
1597{
1598 PEXP_SUBCMD, /* expand :profile sub-commands */
Bram Moolenaar49789dc2011-02-25 14:46:09 +01001599 PEXP_FUNC /* expand :profile func {funcname} */
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001600} pexpand_what;
1601
1602static char *pexpand_cmds[] = {
1603 "start",
1604#define PROFCMD_START 0
1605 "pause",
1606#define PROFCMD_PAUSE 1
1607 "continue",
1608#define PROFCMD_CONTINUE 2
1609 "func",
1610#define PROFCMD_FUNC 3
1611 "file",
1612#define PROFCMD_FILE 4
1613 NULL
1614#define PROFCMD_LAST 5
1615};
1616
1617/*
1618 * Function given to ExpandGeneric() to obtain the profile command
1619 * specific expansion.
1620 */
1621 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001622get_profile_name(expand_T *xp UNUSED, int idx)
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001623{
1624 switch (pexpand_what)
1625 {
1626 case PEXP_SUBCMD:
1627 return (char_u *)pexpand_cmds[idx];
1628 /* case PEXP_FUNC: TODO */
1629 default:
1630 return NULL;
1631 }
1632}
1633
1634/*
1635 * Handle command line completion for :profile command.
1636 */
1637 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001638set_context_in_profile_cmd(expand_T *xp, char_u *arg)
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001639{
1640 char_u *end_subcmd;
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001641
1642 /* Default: expand subcommands. */
1643 xp->xp_context = EXPAND_PROFILE;
1644 pexpand_what = PEXP_SUBCMD;
1645 xp->xp_pattern = arg;
1646
1647 end_subcmd = skiptowhite(arg);
1648 if (*end_subcmd == NUL)
1649 return;
1650
Bram Moolenaar8b9c05f2010-03-02 17:54:33 +01001651 if (end_subcmd - arg == 5 && STRNCMP(arg, "start", 5) == 0)
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001652 {
1653 xp->xp_context = EXPAND_FILES;
1654 xp->xp_pattern = skipwhite(end_subcmd);
1655 return;
1656 }
1657
1658 /* TODO: expand function names after "func" */
1659 xp->xp_context = EXPAND_NOTHING;
1660}
1661
Bram Moolenaar05159a02005-02-26 23:04:13 +00001662/*
1663 * Dump the profiling info.
1664 */
1665 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001666profile_dump(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001667{
1668 FILE *fd;
1669
1670 if (profile_fname != NULL)
1671 {
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +00001672 fd = mch_fopen((char *)profile_fname, "w");
Bram Moolenaar05159a02005-02-26 23:04:13 +00001673 if (fd == NULL)
1674 EMSG2(_(e_notopen), profile_fname);
1675 else
1676 {
Bram Moolenaar05159a02005-02-26 23:04:13 +00001677 script_dump_profile(fd);
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001678 func_dump_profile(fd);
Bram Moolenaar05159a02005-02-26 23:04:13 +00001679 fclose(fd);
1680 }
1681 }
1682}
1683
1684/*
1685 * Start profiling script "fp".
1686 */
1687 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001688script_do_profile(scriptitem_T *si)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001689{
1690 si->sn_pr_count = 0;
1691 profile_zero(&si->sn_pr_total);
1692 profile_zero(&si->sn_pr_self);
1693
1694 ga_init2(&si->sn_prl_ga, sizeof(sn_prl_T), 100);
1695 si->sn_prl_idx = -1;
1696 si->sn_prof_on = TRUE;
1697 si->sn_pr_nest = 0;
1698}
1699
1700/*
1701 * save time when starting to invoke another script or function.
1702 */
1703 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001704script_prof_save(
1705 proftime_T *tm) /* place to store wait time */
Bram Moolenaar05159a02005-02-26 23:04:13 +00001706{
1707 scriptitem_T *si;
1708
1709 if (current_SID > 0 && current_SID <= script_items.ga_len)
1710 {
1711 si = &SCRIPT_ITEM(current_SID);
1712 if (si->sn_prof_on && si->sn_pr_nest++ == 0)
1713 profile_start(&si->sn_pr_child);
1714 }
1715 profile_get_wait(tm);
1716}
1717
1718/*
1719 * Count time spent in children after invoking another script or function.
1720 */
1721 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001722script_prof_restore(proftime_T *tm)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001723{
1724 scriptitem_T *si;
1725
1726 if (current_SID > 0 && current_SID <= script_items.ga_len)
1727 {
1728 si = &SCRIPT_ITEM(current_SID);
1729 if (si->sn_prof_on && --si->sn_pr_nest == 0)
1730 {
1731 profile_end(&si->sn_pr_child);
1732 profile_sub_wait(tm, &si->sn_pr_child); /* don't count wait time */
1733 profile_add(&si->sn_pr_children, &si->sn_pr_child);
1734 profile_add(&si->sn_prl_children, &si->sn_pr_child);
1735 }
1736 }
1737}
1738
1739static proftime_T inchar_time;
1740
1741/*
1742 * Called when starting to wait for the user to type a character.
1743 */
1744 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001745prof_inchar_enter(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001746{
1747 profile_start(&inchar_time);
1748}
1749
1750/*
1751 * Called when finished waiting for the user to type a character.
1752 */
1753 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001754prof_inchar_exit(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001755{
1756 profile_end(&inchar_time);
1757 profile_add(&prof_wait_time, &inchar_time);
1758}
1759
1760/*
1761 * Dump the profiling results for all scripts in file "fd".
1762 */
1763 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001764script_dump_profile(FILE *fd)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001765{
1766 int id;
1767 scriptitem_T *si;
1768 int i;
1769 FILE *sfd;
1770 sn_prl_T *pp;
1771
1772 for (id = 1; id <= script_items.ga_len; ++id)
1773 {
1774 si = &SCRIPT_ITEM(id);
1775 if (si->sn_prof_on)
1776 {
1777 fprintf(fd, "SCRIPT %s\n", si->sn_name);
1778 if (si->sn_pr_count == 1)
1779 fprintf(fd, "Sourced 1 time\n");
1780 else
1781 fprintf(fd, "Sourced %d times\n", si->sn_pr_count);
1782 fprintf(fd, "Total time: %s\n", profile_msg(&si->sn_pr_total));
1783 fprintf(fd, " Self time: %s\n", profile_msg(&si->sn_pr_self));
1784 fprintf(fd, "\n");
1785 fprintf(fd, "count total (s) self (s)\n");
1786
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +00001787 sfd = mch_fopen((char *)si->sn_name, "r");
Bram Moolenaar05159a02005-02-26 23:04:13 +00001788 if (sfd == NULL)
1789 fprintf(fd, "Cannot open file!\n");
1790 else
1791 {
1792 for (i = 0; i < si->sn_prl_ga.ga_len; ++i)
1793 {
1794 if (vim_fgets(IObuff, IOSIZE, sfd))
1795 break;
1796 pp = &PRL_ITEM(si, i);
1797 if (pp->snp_count > 0)
1798 {
1799 fprintf(fd, "%5d ", pp->snp_count);
1800 if (profile_equal(&pp->sn_prl_total, &pp->sn_prl_self))
1801 fprintf(fd, " ");
1802 else
1803 fprintf(fd, "%s ", profile_msg(&pp->sn_prl_total));
1804 fprintf(fd, "%s ", profile_msg(&pp->sn_prl_self));
1805 }
1806 else
1807 fprintf(fd, " ");
1808 fprintf(fd, "%s", IObuff);
1809 }
1810 fclose(sfd);
1811 }
1812 fprintf(fd, "\n");
1813 }
1814 }
1815}
1816
1817/*
1818 * Return TRUE when a function defined in the current script should be
1819 * profiled.
1820 */
1821 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001822prof_def_func(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001823{
Bram Moolenaar53180ce2005-07-05 21:48:14 +00001824 if (current_SID > 0)
1825 return SCRIPT_ITEM(current_SID).sn_pr_force;
1826 return FALSE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001827}
1828
1829# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001830#endif
1831
1832/*
1833 * If 'autowrite' option set, try to write the file.
1834 * Careful: autocommands may make "buf" invalid!
1835 *
1836 * return FAIL for failure, OK otherwise
1837 */
1838 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001839autowrite(buf_T *buf, int forceit)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001840{
Bram Moolenaar373154b2007-02-13 05:19:30 +00001841 int r;
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001842 bufref_T bufref;
Bram Moolenaar373154b2007-02-13 05:19:30 +00001843
Bram Moolenaar071d4272004-06-13 20:20:40 +00001844 if (!(p_aw || p_awa) || !p_write
1845#ifdef FEAT_QUICKFIX
Bram Moolenaar373154b2007-02-13 05:19:30 +00001846 /* never autowrite a "nofile" or "nowrite" buffer */
1847 || bt_dontwrite(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001848#endif
Bram Moolenaar373154b2007-02-13 05:19:30 +00001849 || (!forceit && buf->b_p_ro) || buf->b_ffname == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001850 return FAIL;
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001851 set_bufref(&bufref, buf);
Bram Moolenaar373154b2007-02-13 05:19:30 +00001852 r = buf_write_all(buf, forceit);
1853
1854 /* Writing may succeed but the buffer still changed, e.g., when there is a
1855 * conversion error. We do want to return FAIL then. */
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001856 if (bufref_valid(&bufref) && bufIsChanged(buf))
Bram Moolenaar373154b2007-02-13 05:19:30 +00001857 r = FAIL;
1858 return r;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001859}
1860
1861/*
1862 * flush all buffers, except the ones that are readonly
1863 */
1864 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001865autowrite_all(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001866{
1867 buf_T *buf;
1868
1869 if (!(p_aw || p_awa) || !p_write)
1870 return;
Bram Moolenaar29323592016-07-24 22:04:11 +02001871 FOR_ALL_BUFFERS(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001872 if (bufIsChanged(buf) && !buf->b_p_ro)
1873 {
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001874#ifdef FEAT_AUTOCMD
1875 bufref_T bufref;
1876
1877 set_bufref(&bufref, buf);
1878#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001879 (void)buf_write_all(buf, FALSE);
1880#ifdef FEAT_AUTOCMD
1881 /* an autocommand may have deleted the buffer */
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001882 if (!bufref_valid(&bufref))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001883 buf = firstbuf;
1884#endif
1885 }
1886}
1887
1888/*
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001889 * Return TRUE if buffer was changed and cannot be abandoned.
1890 * For flags use the CCGD_ values.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001891 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001892 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001893check_changed(buf_T *buf, int flags)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001894{
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001895 int forceit = (flags & CCGD_FORCEIT);
1896#ifdef FEAT_AUTOCMD
1897 bufref_T bufref;
1898
1899 set_bufref(&bufref, buf);
1900#endif
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001901
Bram Moolenaar071d4272004-06-13 20:20:40 +00001902 if ( !forceit
1903 && bufIsChanged(buf)
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001904 && ((flags & CCGD_MULTWIN) || buf->b_nwindows <= 1)
1905 && (!(flags & CCGD_AW) || autowrite(buf, forceit) == FAIL))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001906 {
1907#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG)
1908 if ((p_confirm || cmdmod.confirm) && p_write)
1909 {
1910 buf_T *buf2;
1911 int count = 0;
1912
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001913 if (flags & CCGD_ALLBUF)
Bram Moolenaar29323592016-07-24 22:04:11 +02001914 FOR_ALL_BUFFERS(buf2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001915 if (bufIsChanged(buf2)
1916 && (buf2->b_ffname != NULL
1917# ifdef FEAT_BROWSE
1918 || cmdmod.browse
1919# endif
1920 ))
1921 ++count;
1922# ifdef FEAT_AUTOCMD
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001923 if (!bufref_valid(&bufref))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001924 /* Autocommand deleted buffer, oops! It's not changed now. */
1925 return FALSE;
1926# endif
1927 dialog_changed(buf, count > 1);
1928# ifdef FEAT_AUTOCMD
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001929 if (!bufref_valid(&bufref))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001930 /* Autocommand deleted buffer, oops! It's not changed now. */
1931 return FALSE;
1932# endif
1933 return bufIsChanged(buf);
1934 }
1935#endif
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001936 if (flags & CCGD_EXCMD)
Bram Moolenaarf5be7cd2017-08-17 16:55:13 +02001937 no_write_message();
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001938 else
Bram Moolenaarf5be7cd2017-08-17 16:55:13 +02001939 no_write_message_nobang();
Bram Moolenaar071d4272004-06-13 20:20:40 +00001940 return TRUE;
1941 }
1942 return FALSE;
1943}
1944
1945#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG) || defined(PROTO)
1946
1947#if defined(FEAT_BROWSE) || defined(PROTO)
1948/*
1949 * When wanting to write a file without a file name, ask the user for a name.
1950 */
1951 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001952browse_save_fname(buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001953{
1954 if (buf->b_fname == NULL)
1955 {
1956 char_u *fname;
1957
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00001958 fname = do_browse(BROWSE_SAVE, (char_u *)_("Save As"),
1959 NULL, NULL, NULL, NULL, buf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001960 if (fname != NULL)
1961 {
1962 if (setfname(buf, fname, NULL, TRUE) == OK)
1963 buf->b_flags |= BF_NOTEDITED;
1964 vim_free(fname);
1965 }
1966 }
1967}
1968#endif
1969
1970/*
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02001971 * Ask the user what to do when abandoning a changed buffer.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001972 * Must check 'write' option first!
1973 */
1974 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001975dialog_changed(
1976 buf_T *buf,
1977 int checkall) /* may abandon all changed buffers */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001978{
Bram Moolenaard9462e32011-04-11 21:35:11 +02001979 char_u buff[DIALOG_MSG_SIZE];
Bram Moolenaar071d4272004-06-13 20:20:40 +00001980 int ret;
1981 buf_T *buf2;
Bram Moolenaar8218f602012-04-25 17:32:18 +02001982 exarg_T ea;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001983
Bram Moolenaar3f9a1ff2017-08-21 22:06:02 +02001984 dialog_msg(buff, _("Save changes to \"%s\"?"), buf->b_fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001985 if (checkall)
1986 ret = vim_dialog_yesnoallcancel(VIM_QUESTION, NULL, buff, 1);
1987 else
1988 ret = vim_dialog_yesnocancel(VIM_QUESTION, NULL, buff, 1);
1989
Bram Moolenaar8218f602012-04-25 17:32:18 +02001990 /* Init ea pseudo-structure, this is needed for the check_overwrite()
1991 * function. */
1992 ea.append = ea.forceit = FALSE;
1993
Bram Moolenaar071d4272004-06-13 20:20:40 +00001994 if (ret == VIM_YES)
1995 {
1996#ifdef FEAT_BROWSE
1997 /* May get file name, when there is none */
1998 browse_save_fname(buf);
1999#endif
Bram Moolenaar8218f602012-04-25 17:32:18 +02002000 if (buf->b_fname != NULL && check_overwrite(&ea, buf,
2001 buf->b_fname, buf->b_ffname, FALSE) == OK)
2002 /* didn't hit Cancel */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002003 (void)buf_write_all(buf, FALSE);
2004 }
2005 else if (ret == VIM_NO)
2006 {
2007 unchanged(buf, TRUE);
2008 }
2009 else if (ret == VIM_ALL)
2010 {
2011 /*
2012 * Write all modified files that can be written.
2013 * Skip readonly buffers, these need to be confirmed
2014 * individually.
2015 */
Bram Moolenaar29323592016-07-24 22:04:11 +02002016 FOR_ALL_BUFFERS(buf2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002017 {
2018 if (bufIsChanged(buf2)
2019 && (buf2->b_ffname != NULL
2020#ifdef FEAT_BROWSE
2021 || cmdmod.browse
2022#endif
2023 )
2024 && !buf2->b_p_ro)
2025 {
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002026#ifdef FEAT_AUTOCMD
2027 bufref_T bufref;
2028
2029 set_bufref(&bufref, buf2);
2030#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002031#ifdef FEAT_BROWSE
2032 /* May get file name, when there is none */
2033 browse_save_fname(buf2);
2034#endif
Bram Moolenaar8218f602012-04-25 17:32:18 +02002035 if (buf2->b_fname != NULL && check_overwrite(&ea, buf2,
2036 buf2->b_fname, buf2->b_ffname, FALSE) == OK)
2037 /* didn't hit Cancel */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002038 (void)buf_write_all(buf2, FALSE);
2039#ifdef FEAT_AUTOCMD
2040 /* an autocommand may have deleted the buffer */
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002041 if (!bufref_valid(&bufref))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002042 buf2 = firstbuf;
2043#endif
2044 }
2045 }
2046 }
2047 else if (ret == VIM_DISCARDALL)
2048 {
2049 /*
2050 * mark all buffers as unchanged
2051 */
Bram Moolenaar29323592016-07-24 22:04:11 +02002052 FOR_ALL_BUFFERS(buf2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002053 unchanged(buf2, TRUE);
2054 }
2055}
2056#endif
2057
2058/*
2059 * Return TRUE if the buffer "buf" can be abandoned, either by making it
2060 * hidden, autowriting it or unloading it.
2061 */
2062 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002063can_abandon(buf_T *buf, int forceit)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002064{
Bram Moolenaareb44a682017-08-03 22:44:55 +02002065 return ( buf_hide(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002066 || !bufIsChanged(buf)
2067 || buf->b_nwindows > 1
2068 || autowrite(buf, forceit) == OK
2069 || forceit);
2070}
2071
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01002072static void add_bufnum(int *bufnrs, int *bufnump, int nr);
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002073
2074/*
2075 * Add a buffer number to "bufnrs", unless it's already there.
2076 */
2077 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002078add_bufnum(int *bufnrs, int *bufnump, int nr)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002079{
2080 int i;
2081
2082 for (i = 0; i < *bufnump; ++i)
2083 if (bufnrs[i] == nr)
2084 return;
2085 bufnrs[*bufnump] = nr;
2086 *bufnump = *bufnump + 1;
2087}
2088
Bram Moolenaar071d4272004-06-13 20:20:40 +00002089/*
2090 * Return TRUE if any buffer was changed and cannot be abandoned.
2091 * That changed buffer becomes the current buffer.
Bram Moolenaar027387f2016-01-02 22:25:52 +01002092 * When "unload" is true the current buffer is unloaded instead of making it
2093 * hidden. This is used for ":q!".
Bram Moolenaar071d4272004-06-13 20:20:40 +00002094 */
2095 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002096check_changed_any(
2097 int hidden, /* Only check hidden buffers */
2098 int unload)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002099{
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002100 int ret = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002101 buf_T *buf;
2102 int save;
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002103 int i;
2104 int bufnum = 0;
2105 int bufcount = 0;
2106 int *bufnrs;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002107#ifdef FEAT_WINDOWS
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002108 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002109 win_T *wp;
2110#endif
2111
Bram Moolenaar29323592016-07-24 22:04:11 +02002112 FOR_ALL_BUFFERS(buf)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002113 ++bufcount;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002114
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002115 if (bufcount == 0)
2116 return FALSE;
2117
2118 bufnrs = (int *)alloc(sizeof(int) * bufcount);
2119 if (bufnrs == NULL)
2120 return FALSE;
2121
2122 /* curbuf */
2123 bufnrs[bufnum++] = curbuf->b_fnum;
2124#ifdef FEAT_WINDOWS
2125 /* buf in curtab */
2126 FOR_ALL_WINDOWS(wp)
2127 if (wp->w_buffer != curbuf)
2128 add_bufnum(bufnrs, &bufnum, wp->w_buffer->b_fnum);
2129
2130 /* buf in other tab */
Bram Moolenaar29323592016-07-24 22:04:11 +02002131 FOR_ALL_TABPAGES(tp)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002132 if (tp != curtab)
2133 for (wp = tp->tp_firstwin; wp != NULL; wp = wp->w_next)
2134 add_bufnum(bufnrs, &bufnum, wp->w_buffer->b_fnum);
2135#endif
2136 /* any other buf */
Bram Moolenaar29323592016-07-24 22:04:11 +02002137 FOR_ALL_BUFFERS(buf)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002138 add_bufnum(bufnrs, &bufnum, buf->b_fnum);
2139
2140 for (i = 0; i < bufnum; ++i)
2141 {
2142 buf = buflist_findnr(bufnrs[i]);
2143 if (buf == NULL)
2144 continue;
2145 if ((!hidden || buf->b_nwindows == 0) && bufIsChanged(buf))
2146 {
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002147 bufref_T bufref;
2148
2149 set_bufref(&bufref, buf);
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002150 /* Try auto-writing the buffer. If this fails but the buffer no
2151 * longer exists it's not changed, that's OK. */
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002152 if (check_changed(buf, (p_awa ? CCGD_AW : 0)
2153 | CCGD_MULTWIN
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002154 | CCGD_ALLBUF) && bufref_valid(&bufref))
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002155 break; /* didn't save - still changes */
2156 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002157 }
2158
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002159 if (i >= bufnum)
2160 goto theend;
2161
2162 ret = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002163 exiting = FALSE;
2164#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG)
2165 /*
2166 * When ":confirm" used, don't give an error message.
2167 */
2168 if (!(p_confirm || cmdmod.confirm))
2169#endif
2170 {
2171 /* There must be a wait_return for this message, do_buffer()
2172 * may cause a redraw. But wait_return() is a no-op when vgetc()
2173 * is busy (Quit used from window menu), then make sure we don't
2174 * cause a scroll up. */
Bram Moolenaar61660ea2006-04-07 21:40:07 +00002175 if (vgetc_busy > 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002176 {
2177 msg_row = cmdline_row;
2178 msg_col = 0;
2179 msg_didout = FALSE;
2180 }
Bram Moolenaareb44a682017-08-03 22:44:55 +02002181 if (
2182#ifdef FEAT_TERMINAL
2183 term_job_running(buf->b_term)
2184 ? EMSG2(_("E947: Job still running in buffer \"%s\""),
2185 buf->b_fname)
2186 :
2187#endif
2188 EMSG2(_("E162: No write since last change for buffer \"%s\""),
Bram Moolenaare1704ba2012-10-03 18:25:00 +02002189 buf_spname(buf) != NULL ? buf_spname(buf) : buf->b_fname))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002190 {
2191 save = no_wait_return;
2192 no_wait_return = FALSE;
2193 wait_return(FALSE);
2194 no_wait_return = save;
2195 }
2196 }
2197
2198#ifdef FEAT_WINDOWS
2199 /* Try to find a window that contains the buffer. */
2200 if (buf != curbuf)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002201 FOR_ALL_TAB_WINDOWS(tp, wp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002202 if (wp->w_buffer == buf)
2203 {
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002204# ifdef FEAT_AUTOCMD
2205 bufref_T bufref;
2206
2207 set_bufref(&bufref, buf);
2208# endif
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002209 goto_tabpage_win(tp, wp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002210# ifdef FEAT_AUTOCMD
2211 /* Paranoia: did autocms wipe out the buffer with changes? */
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002212 if (!bufref_valid(&bufref))
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002213 {
2214 goto theend;
2215 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002216# endif
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002217 goto buf_found;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002218 }
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002219buf_found:
Bram Moolenaar071d4272004-06-13 20:20:40 +00002220#endif
2221
2222 /* Open the changed buffer in the current window. */
2223 if (buf != curbuf)
Bram Moolenaar027387f2016-01-02 22:25:52 +01002224 set_curbuf(buf, unload ? DOBUF_UNLOAD : DOBUF_GOTO);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002225
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002226theend:
2227 vim_free(bufnrs);
2228 return ret;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002229}
2230
2231/*
2232 * return FAIL if there is no file name, OK if there is one
2233 * give error message for FAIL
2234 */
2235 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002236check_fname(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002237{
2238 if (curbuf->b_ffname == NULL)
2239 {
2240 EMSG(_(e_noname));
2241 return FAIL;
2242 }
2243 return OK;
2244}
2245
2246/*
2247 * flush the contents of a buffer, unless it has no file name
2248 *
2249 * return FAIL for failure, OK otherwise
2250 */
2251 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002252buf_write_all(buf_T *buf, int forceit)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002253{
2254 int retval;
2255#ifdef FEAT_AUTOCMD
2256 buf_T *old_curbuf = curbuf;
2257#endif
2258
2259 retval = (buf_write(buf, buf->b_ffname, buf->b_fname,
2260 (linenr_T)1, buf->b_ml.ml_line_count, NULL,
2261 FALSE, forceit, TRUE, FALSE));
2262#ifdef FEAT_AUTOCMD
2263 if (curbuf != old_curbuf)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00002264 {
Bram Moolenaar8820b482017-03-16 17:23:31 +01002265 msg_source(HL_ATTR(HLF_W));
Bram Moolenaar071d4272004-06-13 20:20:40 +00002266 MSG(_("Warning: Entered other buffer unexpectedly (check autocommands)"));
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00002267 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002268#endif
2269 return retval;
2270}
2271
2272/*
2273 * Code to handle the argument list.
2274 */
2275
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01002276static char_u *do_one_arg(char_u *str);
2277static int do_arglist(char_u *str, int what, int after);
2278static void alist_check_arg_idx(void);
2279static int editing_arg_idx(win_T *win);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002280#ifdef FEAT_LISTCMDS
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01002281static int alist_add_list(int count, char_u **files, int after);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002282#endif
2283#define AL_SET 1
2284#define AL_ADD 2
2285#define AL_DEL 3
2286
Bram Moolenaar071d4272004-06-13 20:20:40 +00002287/*
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002288 * Isolate one argument, taking backticks.
2289 * Changes the argument in-place, puts a NUL after it. Backticks remain.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002290 * Return a pointer to the start of the next argument.
2291 */
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002292 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002293do_one_arg(char_u *str)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002294{
2295 char_u *p;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002296 int inbacktick;
2297
Bram Moolenaar071d4272004-06-13 20:20:40 +00002298 inbacktick = FALSE;
2299 for (p = str; *str; ++str)
2300 {
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002301 /* When the backslash is used for escaping the special meaning of a
2302 * character we need to keep it until wildcard expansion. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002303 if (rem_backslash(str))
2304 {
2305 *p++ = *str++;
2306 *p++ = *str;
2307 }
2308 else
2309 {
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002310 /* An item ends at a space not in backticks */
2311 if (!inbacktick && vim_isspace(*str))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002312 break;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002313 if (*str == '`')
Bram Moolenaar071d4272004-06-13 20:20:40 +00002314 inbacktick ^= TRUE;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002315 *p++ = *str;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002316 }
2317 }
2318 str = skipwhite(str);
2319 *p = NUL;
2320
2321 return str;
2322}
2323
Bram Moolenaar86b68352004-12-27 21:59:20 +00002324/*
2325 * Separate the arguments in "str" and return a list of pointers in the
2326 * growarray "gap".
2327 */
Bram Moolenaar398ee732017-08-03 14:29:14 +02002328 static int
2329get_arglist(garray_T *gap, char_u *str, int escaped)
Bram Moolenaar86b68352004-12-27 21:59:20 +00002330{
2331 ga_init2(gap, (int)sizeof(char_u *), 20);
2332 while (*str != NUL)
2333 {
2334 if (ga_grow(gap, 1) == FAIL)
2335 {
2336 ga_clear(gap);
2337 return FAIL;
2338 }
2339 ((char_u **)gap->ga_data)[gap->ga_len++] = str;
2340
Bram Moolenaar398ee732017-08-03 14:29:14 +02002341 /* If str is escaped, don't handle backslashes or spaces */
2342 if (!escaped)
2343 return OK;
2344
Bram Moolenaar86b68352004-12-27 21:59:20 +00002345 /* Isolate one argument, change it in-place, put a NUL after it. */
2346 str = do_one_arg(str);
2347 }
2348 return OK;
2349}
2350
Bram Moolenaar7df351e2006-01-23 22:30:28 +00002351#if defined(FEAT_QUICKFIX) || defined(FEAT_SYN_HL) || defined(PROTO)
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002352/*
2353 * Parse a list of arguments (file names), expand them and return in
Bram Moolenaar8f5c6f02012-06-29 12:57:06 +02002354 * "fnames[fcountp]". When "wig" is TRUE, removes files matching 'wildignore'.
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002355 * Return FAIL or OK.
2356 */
2357 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002358get_arglist_exp(
2359 char_u *str,
2360 int *fcountp,
2361 char_u ***fnamesp,
2362 int wig)
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002363{
2364 garray_T ga;
2365 int i;
2366
Bram Moolenaar398ee732017-08-03 14:29:14 +02002367 if (get_arglist(&ga, str, TRUE) == FAIL)
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002368 return FAIL;
Bram Moolenaar8f5c6f02012-06-29 12:57:06 +02002369 if (wig == TRUE)
2370 i = expand_wildcards(ga.ga_len, (char_u **)ga.ga_data,
2371 fcountp, fnamesp, EW_FILE|EW_NOTFOUND);
2372 else
2373 i = gen_expand_wildcards(ga.ga_len, (char_u **)ga.ga_data,
2374 fcountp, fnamesp, EW_FILE|EW_NOTFOUND);
2375
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002376 ga_clear(&ga);
2377 return i;
2378}
2379#endif
2380
Bram Moolenaar071d4272004-06-13 20:20:40 +00002381#if defined(FEAT_GUI) || defined(FEAT_CLIENTSERVER) || defined(PROTO)
2382/*
2383 * Redefine the argument list.
2384 */
2385 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002386set_arglist(char_u *str)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002387{
2388 do_arglist(str, AL_SET, 0);
2389}
2390#endif
2391
2392/*
2393 * "what" == AL_SET: Redefine the argument list to 'str'.
2394 * "what" == AL_ADD: add files in 'str' to the argument list after "after".
2395 * "what" == AL_DEL: remove files in 'str' from the argument list.
2396 *
2397 * Return FAIL for failure, OK otherwise.
2398 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002399 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002400do_arglist(
2401 char_u *str,
Bram Moolenaarf1d25012016-03-03 12:22:53 +01002402 int what,
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002403 int after UNUSED) /* 0 means before first one */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002404{
2405 garray_T new_ga;
2406 int exp_count;
2407 char_u **exp_files;
2408 int i;
2409#ifdef FEAT_LISTCMDS
2410 char_u *p;
2411 int match;
2412#endif
Bram Moolenaar398ee732017-08-03 14:29:14 +02002413 int arg_escaped = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002414
2415 /*
Bram Moolenaar2faa29f2016-01-23 23:02:34 +01002416 * Set default argument for ":argadd" command.
2417 */
2418 if (what == AL_ADD && *str == NUL)
2419 {
2420 if (curbuf->b_ffname == NULL)
2421 return FAIL;
2422 str = curbuf->b_fname;
Bram Moolenaar398ee732017-08-03 14:29:14 +02002423 arg_escaped = FALSE;
Bram Moolenaar2faa29f2016-01-23 23:02:34 +01002424 }
2425
2426 /*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002427 * Collect all file name arguments in "new_ga".
2428 */
Bram Moolenaar398ee732017-08-03 14:29:14 +02002429 if (get_arglist(&new_ga, str, arg_escaped) == FAIL)
Bram Moolenaar86b68352004-12-27 21:59:20 +00002430 return FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002431
2432#ifdef FEAT_LISTCMDS
2433 if (what == AL_DEL)
2434 {
2435 regmatch_T regmatch;
2436 int didone;
2437
2438 /*
2439 * Delete the items: use each item as a regexp and find a match in the
2440 * argument list.
2441 */
Bram Moolenaar71afbfe2013-03-19 16:49:16 +01002442 regmatch.rm_ic = p_fic; /* ignore case when 'fileignorecase' is set */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002443 for (i = 0; i < new_ga.ga_len && !got_int; ++i)
2444 {
2445 p = ((char_u **)new_ga.ga_data)[i];
2446 p = file_pat_to_reg_pat(p, NULL, NULL, FALSE);
2447 if (p == NULL)
2448 break;
2449 regmatch.regprog = vim_regcomp(p, p_magic ? RE_MAGIC : 0);
2450 if (regmatch.regprog == NULL)
2451 {
2452 vim_free(p);
2453 break;
2454 }
2455
2456 didone = FALSE;
2457 for (match = 0; match < ARGCOUNT; ++match)
2458 if (vim_regexec(&regmatch, alist_name(&ARGLIST[match]),
2459 (colnr_T)0))
2460 {
2461 didone = TRUE;
2462 vim_free(ARGLIST[match].ae_fname);
2463 mch_memmove(ARGLIST + match, ARGLIST + match + 1,
2464 (ARGCOUNT - match - 1) * sizeof(aentry_T));
2465 --ALIST(curwin)->al_ga.ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002466 if (curwin->w_arg_idx > match)
2467 --curwin->w_arg_idx;
2468 --match;
2469 }
2470
Bram Moolenaar473de612013-06-08 18:19:48 +02002471 vim_regfree(regmatch.regprog);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002472 vim_free(p);
2473 if (!didone)
2474 EMSG2(_(e_nomatch2), ((char_u **)new_ga.ga_data)[i]);
2475 }
2476 ga_clear(&new_ga);
2477 }
2478 else
2479#endif
2480 {
2481 i = expand_wildcards(new_ga.ga_len, (char_u **)new_ga.ga_data,
2482 &exp_count, &exp_files, EW_DIR|EW_FILE|EW_ADDSLASH|EW_NOTFOUND);
2483 ga_clear(&new_ga);
Bram Moolenaar2db5c3b2016-01-16 22:49:34 +01002484 if (i == FAIL || exp_count == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002485 {
2486 EMSG(_(e_nomatch));
2487 return FAIL;
2488 }
2489
2490#ifdef FEAT_LISTCMDS
2491 if (what == AL_ADD)
2492 {
2493 (void)alist_add_list(exp_count, exp_files, after);
2494 vim_free(exp_files);
2495 }
2496 else /* what == AL_SET */
2497#endif
Bram Moolenaar86b68352004-12-27 21:59:20 +00002498 alist_set(ALIST(curwin), exp_count, exp_files, FALSE, NULL, 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002499 }
2500
2501 alist_check_arg_idx();
2502
2503 return OK;
2504}
2505
2506/*
2507 * Check the validity of the arg_idx for each other window.
2508 */
2509 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002510alist_check_arg_idx(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002511{
2512#ifdef FEAT_WINDOWS
2513 win_T *win;
Bram Moolenaarf740b292006-02-16 22:11:02 +00002514 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002515
Bram Moolenaarf740b292006-02-16 22:11:02 +00002516 FOR_ALL_TAB_WINDOWS(tp, win)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002517 if (win->w_alist == curwin->w_alist)
2518 check_arg_idx(win);
2519#else
2520 check_arg_idx(curwin);
2521#endif
2522}
2523
2524/*
Bram Moolenaarb8ff1fb2012-02-04 21:59:01 +01002525 * Return TRUE if window "win" is editing the file at the current argument
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002526 * index.
2527 */
2528 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002529editing_arg_idx(win_T *win)
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002530{
2531 return !(win->w_arg_idx >= WARGCOUNT(win)
2532 || (win->w_buffer->b_fnum
2533 != WARGLIST(win)[win->w_arg_idx].ae_fnum
2534 && (win->w_buffer->b_ffname == NULL
2535 || !(fullpathcmp(
2536 alist_name(&WARGLIST(win)[win->w_arg_idx]),
2537 win->w_buffer->b_ffname, TRUE) & FPC_SAME))));
2538}
2539
2540/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002541 * Check if window "win" is editing the w_arg_idx file in its argument list.
2542 */
2543 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002544check_arg_idx(win_T *win)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002545{
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002546 if (WARGCOUNT(win) > 1 && !editing_arg_idx(win))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002547 {
2548 /* We are not editing the current entry in the argument list.
2549 * Set "arg_had_last" if we are editing the last one. */
2550 win->w_arg_idx_invalid = TRUE;
2551 if (win->w_arg_idx != WARGCOUNT(win) - 1
2552 && arg_had_last == FALSE
2553#ifdef FEAT_WINDOWS
2554 && ALIST(win) == &global_alist
2555#endif
2556 && GARGCOUNT > 0
2557 && win->w_arg_idx < GARGCOUNT
2558 && (win->w_buffer->b_fnum == GARGLIST[GARGCOUNT - 1].ae_fnum
2559 || (win->w_buffer->b_ffname != NULL
2560 && (fullpathcmp(alist_name(&GARGLIST[GARGCOUNT - 1]),
2561 win->w_buffer->b_ffname, TRUE) & FPC_SAME))))
2562 arg_had_last = TRUE;
2563 }
2564 else
2565 {
2566 /* We are editing the current entry in the argument list.
2567 * Set "arg_had_last" if it's also the last one */
2568 win->w_arg_idx_invalid = FALSE;
2569 if (win->w_arg_idx == WARGCOUNT(win) - 1
2570#ifdef FEAT_WINDOWS
2571 && win->w_alist == &global_alist
2572#endif
2573 )
2574 arg_had_last = TRUE;
2575 }
2576}
2577
2578/*
2579 * ":args", ":argslocal" and ":argsglobal".
2580 */
2581 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002582ex_args(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002583{
2584 int i;
2585
2586 if (eap->cmdidx != CMD_args)
2587 {
2588#if defined(FEAT_WINDOWS) && defined(FEAT_LISTCMDS)
2589 alist_unlink(ALIST(curwin));
2590 if (eap->cmdidx == CMD_argglobal)
2591 ALIST(curwin) = &global_alist;
2592 else /* eap->cmdidx == CMD_arglocal */
2593 alist_new();
2594#else
2595 ex_ni(eap);
2596 return;
2597#endif
2598 }
2599
2600 if (!ends_excmd(*eap->arg))
2601 {
2602 /*
2603 * ":args file ..": define new argument list, handle like ":next"
2604 * Also for ":argslocal file .." and ":argsglobal file ..".
2605 */
2606 ex_next(eap);
2607 }
2608 else
2609#if defined(FEAT_WINDOWS) && defined(FEAT_LISTCMDS)
2610 if (eap->cmdidx == CMD_args)
2611#endif
2612 {
2613 /*
2614 * ":args": list arguments.
2615 */
2616 if (ARGCOUNT > 0)
2617 {
2618 /* Overwrite the command, for a short list there is no scrolling
2619 * required and no wait_return(). */
2620 gotocmdline(TRUE);
2621 for (i = 0; i < ARGCOUNT; ++i)
2622 {
2623 if (i == curwin->w_arg_idx)
2624 msg_putchar('[');
2625 msg_outtrans(alist_name(&ARGLIST[i]));
2626 if (i == curwin->w_arg_idx)
2627 msg_putchar(']');
2628 msg_putchar(' ');
2629 }
2630 }
2631 }
2632#if defined(FEAT_WINDOWS) && defined(FEAT_LISTCMDS)
2633 else if (eap->cmdidx == CMD_arglocal)
2634 {
2635 garray_T *gap = &curwin->w_alist->al_ga;
2636
2637 /*
2638 * ":argslocal": make a local copy of the global argument list.
2639 */
2640 if (ga_grow(gap, GARGCOUNT) == OK)
2641 for (i = 0; i < GARGCOUNT; ++i)
2642 if (GARGLIST[i].ae_fname != NULL)
2643 {
2644 AARGLIST(curwin->w_alist)[gap->ga_len].ae_fname =
2645 vim_strsave(GARGLIST[i].ae_fname);
2646 AARGLIST(curwin->w_alist)[gap->ga_len].ae_fnum =
2647 GARGLIST[i].ae_fnum;
2648 ++gap->ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002649 }
2650 }
2651#endif
2652}
2653
2654/*
2655 * ":previous", ":sprevious", ":Next" and ":sNext".
2656 */
2657 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002658ex_previous(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002659{
2660 /* If past the last one already, go to the last one. */
2661 if (curwin->w_arg_idx - (int)eap->line2 >= ARGCOUNT)
2662 do_argfile(eap, ARGCOUNT - 1);
2663 else
2664 do_argfile(eap, curwin->w_arg_idx - (int)eap->line2);
2665}
2666
2667/*
2668 * ":rewind", ":first", ":sfirst" and ":srewind".
2669 */
2670 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002671ex_rewind(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002672{
2673 do_argfile(eap, 0);
2674}
2675
2676/*
2677 * ":last" and ":slast".
2678 */
2679 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002680ex_last(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002681{
2682 do_argfile(eap, ARGCOUNT - 1);
2683}
2684
2685/*
2686 * ":argument" and ":sargument".
2687 */
2688 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002689ex_argument(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002690{
2691 int i;
2692
2693 if (eap->addr_count > 0)
2694 i = eap->line2 - 1;
2695 else
2696 i = curwin->w_arg_idx;
2697 do_argfile(eap, i);
2698}
2699
2700/*
2701 * Edit file "argn" of the argument lists.
2702 */
2703 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002704do_argfile(exarg_T *eap, int argn)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002705{
2706 int other;
2707 char_u *p;
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002708 int old_arg_idx = curwin->w_arg_idx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002709
2710 if (argn < 0 || argn >= ARGCOUNT)
2711 {
2712 if (ARGCOUNT <= 1)
2713 EMSG(_("E163: There is only one file to edit"));
2714 else if (argn < 0)
2715 EMSG(_("E164: Cannot go before first file"));
2716 else
2717 EMSG(_("E165: Cannot go beyond last file"));
2718 }
2719 else
2720 {
2721 setpcmark();
2722#ifdef FEAT_GUI
2723 need_mouse_correct = TRUE;
2724#endif
2725
2726#ifdef FEAT_WINDOWS
Bram Moolenaardf1bdc92006-02-23 21:32:16 +00002727 /* split window or create new tab page first */
2728 if (*eap->cmd == 's' || cmdmod.tab != 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002729 {
2730 if (win_split(0, 0) == FAIL)
2731 return;
Bram Moolenaar3368ea22010-09-21 16:56:35 +02002732 RESET_BINDING(curwin);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002733 }
2734 else
2735#endif
2736 {
2737 /*
2738 * if 'hidden' set, only check for changed file when re-editing
2739 * the same buffer
2740 */
2741 other = TRUE;
Bram Moolenaareb44a682017-08-03 22:44:55 +02002742 if (buf_hide(curbuf))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002743 {
2744 p = fix_fname(alist_name(&ARGLIST[argn]));
2745 other = otherfile(p);
2746 vim_free(p);
2747 }
Bram Moolenaareb44a682017-08-03 22:44:55 +02002748 if ((!buf_hide(curbuf) || !other)
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002749 && check_changed(curbuf, CCGD_AW
2750 | (other ? 0 : CCGD_MULTWIN)
2751 | (eap->forceit ? CCGD_FORCEIT : 0)
2752 | CCGD_EXCMD))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002753 return;
2754 }
2755
2756 curwin->w_arg_idx = argn;
2757 if (argn == ARGCOUNT - 1
2758#ifdef FEAT_WINDOWS
2759 && curwin->w_alist == &global_alist
2760#endif
2761 )
2762 arg_had_last = TRUE;
2763
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002764 /* Edit the file; always use the last known line number.
2765 * When it fails (e.g. Abort for already edited file) restore the
2766 * argument index. */
2767 if (do_ecmd(0, alist_name(&ARGLIST[curwin->w_arg_idx]), NULL,
Bram Moolenaar071d4272004-06-13 20:20:40 +00002768 eap, ECMD_LAST,
Bram Moolenaareb44a682017-08-03 22:44:55 +02002769 (buf_hide(curwin->w_buffer) ? ECMD_HIDE : 0)
Bram Moolenaar701f7af2008-11-15 13:12:07 +00002770 + (eap->forceit ? ECMD_FORCEIT : 0), curwin) == FAIL)
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002771 curwin->w_arg_idx = old_arg_idx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002772 /* like Vi: set the mark where the cursor is in the file. */
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002773 else if (eap->cmdidx != CMD_argdo)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002774 setmark('\'');
2775 }
2776}
2777
2778/*
2779 * ":next", and commands that behave like it.
2780 */
2781 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002782ex_next(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002783{
2784 int i;
2785
2786 /*
2787 * check for changed buffer now, if this fails the argument list is not
2788 * redefined.
2789 */
Bram Moolenaareb44a682017-08-03 22:44:55 +02002790 if ( buf_hide(curbuf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002791 || eap->cmdidx == CMD_snext
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002792 || !check_changed(curbuf, CCGD_AW
2793 | (eap->forceit ? CCGD_FORCEIT : 0)
2794 | CCGD_EXCMD))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002795 {
2796 if (*eap->arg != NUL) /* redefine file list */
2797 {
2798 if (do_arglist(eap->arg, AL_SET, 0) == FAIL)
2799 return;
2800 i = 0;
2801 }
2802 else
2803 i = curwin->w_arg_idx + (int)eap->line2;
2804 do_argfile(eap, i);
2805 }
2806}
2807
Bram Moolenaar0106e3d2016-02-23 18:55:43 +01002808#if defined(FEAT_LISTCMDS) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002809/*
2810 * ":argedit"
2811 */
2812 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002813ex_argedit(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002814{
Bram Moolenaar90305c62017-07-16 15:31:17 +02002815 int i = eap->addr_count ? (int)eap->line2 : curwin->w_arg_idx + 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002816
Bram Moolenaar90305c62017-07-16 15:31:17 +02002817 if (do_arglist(eap->arg, AL_ADD, i) == FAIL)
2818 return;
2819#ifdef FEAT_TITLE
2820 maketitle();
2821#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002822
Bram Moolenaar90305c62017-07-16 15:31:17 +02002823 if (curwin->w_arg_idx == 0 && (curbuf->b_ml.ml_flags & ML_EMPTY)
2824 && curbuf->b_ffname == NULL)
2825 i = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002826 /* Edit the argument. */
Bram Moolenaar90305c62017-07-16 15:31:17 +02002827 if (i < ARGCOUNT)
2828 do_argfile(eap, i);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002829}
2830
2831/*
2832 * ":argadd"
2833 */
2834 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002835ex_argadd(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002836{
2837 do_arglist(eap->arg, AL_ADD,
2838 eap->addr_count > 0 ? (int)eap->line2 : curwin->w_arg_idx + 1);
2839#ifdef FEAT_TITLE
2840 maketitle();
2841#endif
2842}
2843
2844/*
2845 * ":argdelete"
2846 */
2847 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002848ex_argdelete(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002849{
2850 int i;
2851 int n;
2852
2853 if (eap->addr_count > 0)
2854 {
2855 /* ":1,4argdel": Delete all arguments in the range. */
2856 if (eap->line2 > ARGCOUNT)
2857 eap->line2 = ARGCOUNT;
2858 n = eap->line2 - eap->line1 + 1;
Bram Moolenaar69a92fb2017-03-09 15:58:30 +01002859 if (*eap->arg != NUL)
2860 /* Can't have both a range and an argument. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002861 EMSG(_(e_invarg));
Bram Moolenaar69a92fb2017-03-09 15:58:30 +01002862 else if (n <= 0)
2863 {
2864 /* Don't give an error for ":%argdel" if the list is empty. */
2865 if (eap->line1 != 1 || eap->line2 != 0)
2866 EMSG(_(e_invrange));
2867 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002868 else
2869 {
2870 for (i = eap->line1; i <= eap->line2; ++i)
2871 vim_free(ARGLIST[i - 1].ae_fname);
2872 mch_memmove(ARGLIST + eap->line1 - 1, ARGLIST + eap->line2,
2873 (size_t)((ARGCOUNT - eap->line2) * sizeof(aentry_T)));
2874 ALIST(curwin)->al_ga.ga_len -= n;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002875 if (curwin->w_arg_idx >= eap->line2)
2876 curwin->w_arg_idx -= n;
2877 else if (curwin->w_arg_idx > eap->line1)
2878 curwin->w_arg_idx = eap->line1;
Bram Moolenaar72defda2016-01-17 18:04:33 +01002879 if (ARGCOUNT == 0)
2880 curwin->w_arg_idx = 0;
2881 else if (curwin->w_arg_idx >= ARGCOUNT)
2882 curwin->w_arg_idx = ARGCOUNT - 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002883 }
2884 }
2885 else if (*eap->arg == NUL)
2886 EMSG(_(e_argreq));
2887 else
2888 do_arglist(eap->arg, AL_DEL, 0);
2889#ifdef FEAT_TITLE
2890 maketitle();
2891#endif
2892}
2893
2894/*
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002895 * ":argdo", ":windo", ":bufdo", ":tabdo", ":cdo", ":ldo", ":cfdo" and ":lfdo"
Bram Moolenaar071d4272004-06-13 20:20:40 +00002896 */
2897 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002898ex_listdo(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002899{
2900 int i;
2901#ifdef FEAT_WINDOWS
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002902 win_T *wp;
2903 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002904#endif
Bram Moolenaare25bb902015-02-27 20:33:37 +01002905 buf_T *buf = curbuf;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002906 int next_fnum = 0;
2907#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
2908 char_u *save_ei = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002909#endif
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002910 char_u *p_shm_save;
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002911#ifdef FEAT_QUICKFIX
Bram Moolenaared84b762015-09-09 22:35:29 +02002912 int qf_size = 0;
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002913 int qf_idx;
2914#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002915
2916#ifndef FEAT_WINDOWS
2917 if (eap->cmdidx == CMD_windo)
2918 {
2919 ex_ni(eap);
2920 return;
2921 }
2922#endif
2923
Bram Moolenaar0106e3d2016-02-23 18:55:43 +01002924#ifndef FEAT_QUICKFIX
2925 if (eap->cmdidx == CMD_cdo || eap->cmdidx == CMD_ldo ||
2926 eap->cmdidx == CMD_cfdo || eap->cmdidx == CMD_lfdo)
2927 {
2928 ex_ni(eap);
2929 return;
2930 }
2931#endif
2932
Bram Moolenaar071d4272004-06-13 20:20:40 +00002933#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002934 if (eap->cmdidx != CMD_windo && eap->cmdidx != CMD_tabdo)
Bram Moolenaardcaf10e2005-01-21 11:55:25 +00002935 /* Don't do syntax HL autocommands. Skipping the syntax file is a
2936 * great speed improvement. */
2937 save_ei = au_event_disable(",Syntax");
Bram Moolenaar071d4272004-06-13 20:20:40 +00002938#endif
Bram Moolenaar6b1ee342014-08-06 18:17:11 +02002939#ifdef FEAT_CLIPBOARD
2940 start_global_changes();
2941#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002942
2943 if (eap->cmdidx == CMD_windo
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002944 || eap->cmdidx == CMD_tabdo
Bram Moolenaareb44a682017-08-03 22:44:55 +02002945 || buf_hide(curbuf)
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002946 || !check_changed(curbuf, CCGD_AW
2947 | (eap->forceit ? CCGD_FORCEIT : 0)
2948 | CCGD_EXCMD))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002949 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00002950 i = 0;
Bram Moolenaara162bc52015-01-07 16:54:21 +01002951 /* start at the eap->line1 argument/window/buffer */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002952#ifdef FEAT_WINDOWS
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002953 wp = firstwin;
2954 tp = first_tabpage;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002955#endif
Bram Moolenaara162bc52015-01-07 16:54:21 +01002956 switch (eap->cmdidx)
2957 {
2958#ifdef FEAT_WINDOWS
2959 case CMD_windo:
2960 for ( ; wp != NULL && i + 1 < eap->line1; wp = wp->w_next)
2961 i++;
2962 break;
2963 case CMD_tabdo:
2964 for( ; tp != NULL && i + 1 < eap->line1; tp = tp->tp_next)
2965 i++;
2966 break;
2967#endif
2968 case CMD_argdo:
2969 i = eap->line1 - 1;
2970 break;
Bram Moolenaara162bc52015-01-07 16:54:21 +01002971 default:
2972 break;
2973 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002974 /* set pcmark now */
2975 if (eap->cmdidx == CMD_bufdo)
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002976 {
Bram Moolenaare25bb902015-02-27 20:33:37 +01002977 /* Advance to the first listed buffer after "eap->line1". */
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002978 for (buf = firstbuf; buf != NULL && (buf->b_fnum < eap->line1
Bram Moolenaare25bb902015-02-27 20:33:37 +01002979 || !buf->b_p_bl); buf = buf->b_next)
2980 if (buf->b_fnum > eap->line2)
2981 {
2982 buf = NULL;
2983 break;
2984 }
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002985 if (buf != NULL)
Bram Moolenaare25bb902015-02-27 20:33:37 +01002986 goto_buffer(eap, DOBUF_FIRST, FORWARD, buf->b_fnum);
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002987 }
2988#ifdef FEAT_QUICKFIX
2989 else if (eap->cmdidx == CMD_cdo || eap->cmdidx == CMD_ldo
2990 || eap->cmdidx == CMD_cfdo || eap->cmdidx == CMD_lfdo)
2991 {
2992 qf_size = qf_get_size(eap);
2993 if (qf_size <= 0 || eap->line1 > qf_size)
2994 buf = NULL;
2995 else
2996 {
2997 ex_cc(eap);
2998
2999 buf = curbuf;
3000 i = eap->line1 - 1;
3001 if (eap->addr_count <= 0)
3002 /* default is all the quickfix/location list entries */
3003 eap->line2 = qf_size;
3004 }
3005 }
3006#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003007 else
3008 setpcmark();
3009 listcmd_busy = TRUE; /* avoids setting pcmark below */
3010
Bram Moolenaare25bb902015-02-27 20:33:37 +01003011 while (!got_int && buf != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003012 {
3013 if (eap->cmdidx == CMD_argdo)
3014 {
3015 /* go to argument "i" */
3016 if (i == ARGCOUNT)
3017 break;
3018 /* Don't call do_argfile() when already there, it will try
3019 * reloading the file. */
Bram Moolenaard4755bb2004-09-02 19:12:26 +00003020 if (curwin->w_arg_idx != i || !editing_arg_idx(curwin))
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003021 {
3022 /* Clear 'shm' to avoid that the file message overwrites
3023 * any output from the command. */
3024 p_shm_save = vim_strsave(p_shm);
3025 set_option_value((char_u *)"shm", 0L, (char_u *)"", 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003026 do_argfile(eap, i);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003027 set_option_value((char_u *)"shm", 0L, p_shm_save, 0);
3028 vim_free(p_shm_save);
3029 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003030 if (curwin->w_arg_idx != i)
3031 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003032 }
3033#ifdef FEAT_WINDOWS
3034 else if (eap->cmdidx == CMD_windo)
3035 {
Bram Moolenaar32466aa2006-02-24 23:53:04 +00003036 /* go to window "wp" */
3037 if (!win_valid(wp))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003038 break;
Bram Moolenaar32466aa2006-02-24 23:53:04 +00003039 win_goto(wp);
Bram Moolenaar41423242007-05-03 20:11:13 +00003040 if (curwin != wp)
3041 break; /* something must be wrong */
Bram Moolenaar32466aa2006-02-24 23:53:04 +00003042 wp = curwin->w_next;
3043 }
3044 else if (eap->cmdidx == CMD_tabdo)
3045 {
3046 /* go to window "tp" */
3047 if (!valid_tabpage(tp))
3048 break;
Bram Moolenaar49e649f2013-05-06 04:50:35 +02003049 goto_tabpage_tp(tp, TRUE, TRUE);
Bram Moolenaar32466aa2006-02-24 23:53:04 +00003050 tp = tp->tp_next;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003051 }
3052#endif
3053 else if (eap->cmdidx == CMD_bufdo)
3054 {
3055 /* Remember the number of the next listed buffer, in case
3056 * ":bwipe" is used or autocommands do something strange. */
3057 next_fnum = -1;
3058 for (buf = curbuf->b_next; buf != NULL; buf = buf->b_next)
3059 if (buf->b_p_bl)
3060 {
3061 next_fnum = buf->b_fnum;
3062 break;
3063 }
3064 }
3065
Bram Moolenaara162bc52015-01-07 16:54:21 +01003066 ++i;
3067
Bram Moolenaar071d4272004-06-13 20:20:40 +00003068 /* execute the command */
3069 do_cmdline(eap->arg, eap->getline, eap->cookie,
3070 DOCMD_VERBOSE + DOCMD_NOWAIT);
3071
3072 if (eap->cmdidx == CMD_bufdo)
3073 {
3074 /* Done? */
Bram Moolenaara162bc52015-01-07 16:54:21 +01003075 if (next_fnum < 0 || next_fnum > eap->line2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003076 break;
3077 /* Check if the buffer still exists. */
Bram Moolenaar29323592016-07-24 22:04:11 +02003078 FOR_ALL_BUFFERS(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003079 if (buf->b_fnum == next_fnum)
3080 break;
3081 if (buf == NULL)
3082 break;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003083
3084 /* Go to the next buffer. Clear 'shm' to avoid that the file
3085 * message overwrites any output from the command. */
3086 p_shm_save = vim_strsave(p_shm);
3087 set_option_value((char_u *)"shm", 0L, (char_u *)"", 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003088 goto_buffer(eap, DOBUF_FIRST, FORWARD, next_fnum);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003089 set_option_value((char_u *)"shm", 0L, p_shm_save, 0);
3090 vim_free(p_shm_save);
3091
Bram Moolenaaraa23b372015-09-08 18:46:31 +02003092 /* If autocommands took us elsewhere, quit here. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003093 if (curbuf->b_fnum != next_fnum)
3094 break;
3095 }
3096
Bram Moolenaaraa23b372015-09-08 18:46:31 +02003097#ifdef FEAT_QUICKFIX
3098 if (eap->cmdidx == CMD_cdo || eap->cmdidx == CMD_ldo
3099 || eap->cmdidx == CMD_cfdo || eap->cmdidx == CMD_lfdo)
3100 {
3101 if (i >= qf_size || i >= eap->line2)
3102 break;
3103
3104 qf_idx = qf_get_cur_idx(eap);
3105
3106 ex_cnext(eap);
3107
3108 /* If jumping to the next quickfix entry fails, quit here */
3109 if (qf_get_cur_idx(eap) == qf_idx)
3110 break;
3111 }
3112#endif
3113
Bram Moolenaar071d4272004-06-13 20:20:40 +00003114 if (eap->cmdidx == CMD_windo)
3115 {
3116 validate_cursor(); /* cursor may have moved */
3117#ifdef FEAT_SCROLLBIND
3118 /* required when 'scrollbind' has been set */
3119 if (curwin->w_p_scb)
3120 do_check_scrollbind(TRUE);
3121#endif
3122 }
Bram Moolenaara162bc52015-01-07 16:54:21 +01003123
3124#ifdef FEAT_WINDOWS
3125 if (eap->cmdidx == CMD_windo || eap->cmdidx == CMD_tabdo)
3126 if (i+1 > eap->line2)
3127 break;
3128#endif
3129 if (eap->cmdidx == CMD_argdo && i >= eap->line2)
3130 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003131 }
3132 listcmd_busy = FALSE;
3133 }
3134
3135#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
Bram Moolenaar6ac54292005-02-02 23:07:25 +00003136 if (save_ei != NULL)
3137 {
3138 au_event_restore(save_ei);
3139 apply_autocmds(EVENT_SYNTAX, curbuf->b_p_syn,
3140 curbuf->b_fname, TRUE, curbuf);
3141 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003142#endif
Bram Moolenaar6b1ee342014-08-06 18:17:11 +02003143#ifdef FEAT_CLIPBOARD
3144 end_global_changes();
3145#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003146}
3147
3148/*
3149 * Add files[count] to the arglist of the current window after arg "after".
3150 * The file names in files[count] must have been allocated and are taken over.
3151 * Files[] itself is not taken over.
3152 * Returns index of first added argument. Returns -1 when failed (out of mem).
3153 */
3154 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003155alist_add_list(
3156 int count,
3157 char_u **files,
3158 int after) /* where to add: 0 = before first one */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003159{
3160 int i;
Bram Moolenaara24f0a52016-01-17 19:39:00 +01003161 int old_argcount = ARGCOUNT;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003162
3163 if (ga_grow(&ALIST(curwin)->al_ga, count) == OK)
3164 {
3165 if (after < 0)
3166 after = 0;
3167 if (after > ARGCOUNT)
3168 after = ARGCOUNT;
3169 if (after < ARGCOUNT)
3170 mch_memmove(&(ARGLIST[after + count]), &(ARGLIST[after]),
3171 (ARGCOUNT - after) * sizeof(aentry_T));
3172 for (i = 0; i < count; ++i)
3173 {
3174 ARGLIST[after + i].ae_fname = files[i];
3175 ARGLIST[after + i].ae_fnum = buflist_add(files[i], BLN_LISTED);
3176 }
3177 ALIST(curwin)->al_ga.ga_len += count;
Bram Moolenaara24f0a52016-01-17 19:39:00 +01003178 if (old_argcount > 0 && curwin->w_arg_idx >= after)
3179 curwin->w_arg_idx += count;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003180 return after;
3181 }
3182
3183 for (i = 0; i < count; ++i)
3184 vim_free(files[i]);
3185 return -1;
3186}
3187
3188#endif /* FEAT_LISTCMDS */
3189
3190#ifdef FEAT_EVAL
3191/*
3192 * ":compiler[!] {name}"
3193 */
3194 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003195ex_compiler(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003196{
3197 char_u *buf;
3198 char_u *old_cur_comp = NULL;
3199 char_u *p;
3200
3201 if (*eap->arg == NUL)
3202 {
3203 /* List all compiler scripts. */
3204 do_cmdline_cmd((char_u *)"echo globpath(&rtp, 'compiler/*.vim')");
3205 /* ) keep the indenter happy... */
3206 }
3207 else
3208 {
3209 buf = alloc((unsigned)(STRLEN(eap->arg) + 14));
3210 if (buf != NULL)
3211 {
3212 if (eap->forceit)
3213 {
3214 /* ":compiler! {name}" sets global options */
3215 do_cmdline_cmd((char_u *)
3216 "command -nargs=* CompilerSet set <args>");
3217 }
3218 else
3219 {
3220 /* ":compiler! {name}" sets local options.
3221 * To remain backwards compatible "current_compiler" is always
3222 * used. A user's compiler plugin may set it, the distributed
3223 * plugin will then skip the settings. Afterwards set
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003224 * "b:current_compiler" and restore "current_compiler".
3225 * Explicitly prepend "g:" to make it work in a function. */
3226 old_cur_comp = get_var_value((char_u *)"g:current_compiler");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003227 if (old_cur_comp != NULL)
3228 old_cur_comp = vim_strsave(old_cur_comp);
3229 do_cmdline_cmd((char_u *)
3230 "command -nargs=* CompilerSet setlocal <args>");
3231 }
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003232 do_unlet((char_u *)"g:current_compiler", TRUE);
Bram Moolenaar2ce06f62005-01-31 19:19:04 +00003233 do_unlet((char_u *)"b:current_compiler", TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003234
3235 sprintf((char *)buf, "compiler/%s.vim", eap->arg);
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003236 if (source_runtime(buf, DIP_ALL) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003237 EMSG2(_("E666: compiler not supported: %s"), eap->arg);
3238 vim_free(buf);
3239
3240 do_cmdline_cmd((char_u *)":delcommand CompilerSet");
3241
3242 /* Set "b:current_compiler" from "current_compiler". */
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003243 p = get_var_value((char_u *)"g:current_compiler");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003244 if (p != NULL)
3245 set_internal_string_var((char_u *)"b:current_compiler", p);
3246
3247 /* Restore "current_compiler" for ":compiler {name}". */
3248 if (!eap->forceit)
3249 {
3250 if (old_cur_comp != NULL)
3251 {
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003252 set_internal_string_var((char_u *)"g:current_compiler",
Bram Moolenaar071d4272004-06-13 20:20:40 +00003253 old_cur_comp);
3254 vim_free(old_cur_comp);
3255 }
3256 else
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003257 do_unlet((char_u *)"g:current_compiler", TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003258 }
3259 }
3260 }
3261}
3262#endif
3263
3264/*
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003265 * ":runtime [what] {name}"
Bram Moolenaar071d4272004-06-13 20:20:40 +00003266 */
3267 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003268ex_runtime(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003269{
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003270 char_u *arg = eap->arg;
3271 char_u *p = skiptowhite(arg);
3272 int len = (int)(p - arg);
3273 int flags = eap->forceit ? DIP_ALL : 0;
3274
3275 if (STRNCMP(arg, "START", len) == 0)
3276 {
3277 flags += DIP_START + DIP_NORTP;
3278 arg = skipwhite(arg + len);
3279 }
3280 else if (STRNCMP(arg, "OPT", len) == 0)
3281 {
3282 flags += DIP_OPT + DIP_NORTP;
3283 arg = skipwhite(arg + len);
3284 }
3285 else if (STRNCMP(arg, "PACK", len) == 0)
3286 {
3287 flags += DIP_START + DIP_OPT + DIP_NORTP;
3288 arg = skipwhite(arg + len);
3289 }
3290 else if (STRNCMP(arg, "ALL", len) == 0)
3291 {
3292 flags += DIP_START + DIP_OPT;
3293 arg = skipwhite(arg + len);
3294 }
3295
3296 source_runtime(arg, flags);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003297}
3298
Bram Moolenaar071d4272004-06-13 20:20:40 +00003299 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003300source_callback(char_u *fname, void *cookie UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003301{
Bram Moolenaar910f66f2006-04-05 20:41:53 +00003302 (void)do_source(fname, FALSE, DOSO_NONE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003303}
3304
3305/*
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003306 * Find the file "name" in all directories in "path" and invoke
3307 * "callback(fname, cookie)".
3308 * "name" can contain wildcards.
3309 * When "flags" has DIP_ALL: source all files, otherwise only the first one.
3310 * When "flags" has DIP_DIR: find directories instead of files.
3311 * When "flags" has DIP_ERR: give an error message if there is no match.
3312 *
3313 * return FAIL when no file could be sourced, OK otherwise.
3314 */
Bram Moolenaar6bef5302016-03-12 21:28:26 +01003315 int
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003316do_in_path(
3317 char_u *path,
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003318 char_u *name,
Bram Moolenaar91715872016-03-03 17:13:03 +01003319 int flags,
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003320 void (*callback)(char_u *fname, void *ck),
3321 void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003322{
3323 char_u *rtp;
3324 char_u *np;
3325 char_u *buf;
3326 char_u *rtp_copy;
3327 char_u *tail;
3328 int num_files;
3329 char_u **files;
3330 int i;
3331 int did_one = FALSE;
3332#ifdef AMIGA
3333 struct Process *proc = (struct Process *)FindTask(0L);
3334 APTR save_winptr = proc->pr_WindowPtr;
3335
3336 /* Avoid a requester here for a volume that doesn't exist. */
3337 proc->pr_WindowPtr = (APTR)-1L;
3338#endif
3339
3340 /* Make a copy of 'runtimepath'. Invoking the callback may change the
3341 * value. */
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003342 rtp_copy = vim_strsave(path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003343 buf = alloc(MAXPATHL);
3344 if (buf != NULL && rtp_copy != NULL)
3345 {
Bram Moolenaarc09a6d62013-06-10 21:27:29 +02003346 if (p_verbose > 1 && name != NULL)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003347 {
3348 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00003349 smsg((char_u *)_("Searching for \"%s\" in \"%s\""),
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003350 (char *)name, (char *)path);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003351 verbose_leave();
3352 }
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00003353
Bram Moolenaar071d4272004-06-13 20:20:40 +00003354 /* Loop over all entries in 'runtimepath'. */
3355 rtp = rtp_copy;
Bram Moolenaar91715872016-03-03 17:13:03 +01003356 while (*rtp != NUL && ((flags & DIP_ALL) || !did_one))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003357 {
Bram Moolenaar66459b72016-08-06 19:01:55 +02003358 size_t buflen;
3359
Bram Moolenaar071d4272004-06-13 20:20:40 +00003360 /* Copy the path from 'runtimepath' to buf[]. */
3361 copy_option_part(&rtp, buf, MAXPATHL, ",");
Bram Moolenaar66459b72016-08-06 19:01:55 +02003362 buflen = STRLEN(buf);
3363
3364 /* Skip after or non-after directories. */
3365 if (flags & (DIP_NOAFTER | DIP_AFTER))
3366 {
3367 int is_after = buflen >= 5
3368 && STRCMP(buf + buflen - 5, "after") == 0;
3369
3370 if ((is_after && (flags & DIP_NOAFTER))
3371 || (!is_after && (flags & DIP_AFTER)))
3372 continue;
3373 }
3374
Bram Moolenaarc09a6d62013-06-10 21:27:29 +02003375 if (name == NULL)
3376 {
3377 (*callback)(buf, (void *) &cookie);
3378 if (!did_one)
3379 did_one = (cookie == NULL);
3380 }
Bram Moolenaar66459b72016-08-06 19:01:55 +02003381 else if (buflen + STRLEN(name) + 2 < MAXPATHL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003382 {
3383 add_pathsep(buf);
3384 tail = buf + STRLEN(buf);
3385
3386 /* Loop over all patterns in "name" */
3387 np = name;
Bram Moolenaar91715872016-03-03 17:13:03 +01003388 while (*np != NUL && ((flags & DIP_ALL) || !did_one))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003389 {
3390 /* Append the pattern from "name" to buf[]. */
3391 copy_option_part(&np, tail, (int)(MAXPATHL - (tail - buf)),
3392 "\t ");
3393
3394 if (p_verbose > 2)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003395 {
3396 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00003397 smsg((char_u *)_("Searching for \"%s\""), buf);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003398 verbose_leave();
3399 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003400
3401 /* Expand wildcards, invoke the callback for each match. */
3402 if (gen_expand_wildcards(1, &buf, &num_files, &files,
Bram Moolenaar91715872016-03-03 17:13:03 +01003403 (flags & DIP_DIR) ? EW_DIR : EW_FILE) == OK)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003404 {
3405 for (i = 0; i < num_files; ++i)
3406 {
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00003407 (*callback)(files[i], cookie);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003408 did_one = TRUE;
Bram Moolenaar91715872016-03-03 17:13:03 +01003409 if (!(flags & DIP_ALL))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003410 break;
3411 }
3412 FreeWild(num_files, files);
3413 }
3414 }
3415 }
3416 }
3417 }
3418 vim_free(buf);
3419 vim_free(rtp_copy);
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003420 if (!did_one && name != NULL)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003421 {
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003422 char *basepath = path == p_rtp ? "runtimepath" : "packpath";
3423
3424 if (flags & DIP_ERR)
3425 EMSG3(_(e_dirnotf), basepath, name);
3426 else if (p_verbose > 0)
3427 {
3428 verbose_enter();
3429 smsg((char_u *)_("not found in '%s': \"%s\""), basepath, name);
3430 verbose_leave();
3431 }
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003432 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003433
3434#ifdef AMIGA
3435 proc->pr_WindowPtr = save_winptr;
3436#endif
3437
3438 return did_one ? OK : FAIL;
3439}
3440
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003441/*
Bram Moolenaar07ecfa62017-06-27 14:43:55 +02003442 * Find "name" in "path". When found, invoke the callback function for
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003443 * it: callback(fname, "cookie")
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003444 * When "flags" has DIP_ALL repeat for all matches, otherwise only the first
3445 * one is used.
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003446 * Returns OK when at least one match found, FAIL otherwise.
3447 *
Bram Moolenaar07ecfa62017-06-27 14:43:55 +02003448 * If "name" is NULL calls callback for each entry in "path". Cookie is
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003449 * passed by reference in this case, setting it to NULL indicates that callback
3450 * has done its job.
3451 */
Bram Moolenaar07ecfa62017-06-27 14:43:55 +02003452 static int
3453do_in_path_and_pp(
3454 char_u *path,
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003455 char_u *name,
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003456 int flags,
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003457 void (*callback)(char_u *fname, void *ck),
3458 void *cookie)
3459{
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003460 int done = FAIL;
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003461 char_u *s;
3462 int len;
3463 char *start_dir = "pack/*/start/*/%s";
3464 char *opt_dir = "pack/*/opt/*/%s";
3465
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003466 if ((flags & DIP_NORTP) == 0)
Bram Moolenaar07ecfa62017-06-27 14:43:55 +02003467 done = do_in_path(path, name, flags, callback, cookie);
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003468
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003469 if ((done == FAIL || (flags & DIP_ALL)) && (flags & DIP_START))
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003470 {
Bram Moolenaar1c8b4ed2016-03-17 21:51:03 +01003471 len = (int)(STRLEN(start_dir) + STRLEN(name));
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003472 s = alloc(len);
3473 if (s == NULL)
3474 return FAIL;
3475 vim_snprintf((char *)s, len, start_dir, name);
3476 done = do_in_path(p_pp, s, flags, callback, cookie);
3477 vim_free(s);
3478 }
3479
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003480 if ((done == FAIL || (flags & DIP_ALL)) && (flags & DIP_OPT))
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003481 {
Bram Moolenaar1c8b4ed2016-03-17 21:51:03 +01003482 len = (int)(STRLEN(opt_dir) + STRLEN(name));
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003483 s = alloc(len);
3484 if (s == NULL)
3485 return FAIL;
3486 vim_snprintf((char *)s, len, opt_dir, name);
3487 done = do_in_path(p_pp, s, flags, callback, cookie);
3488 vim_free(s);
3489 }
3490
3491 return done;
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003492}
3493
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003494/*
Bram Moolenaar07ecfa62017-06-27 14:43:55 +02003495 * Just like do_in_path_and_pp(), using 'runtimepath' for "path".
3496 */
3497 int
3498do_in_runtimepath(
3499 char_u *name,
3500 int flags,
3501 void (*callback)(char_u *fname, void *ck),
3502 void *cookie)
3503{
3504 return do_in_path_and_pp(p_rtp, name, flags, callback, cookie);
3505}
3506
3507/*
3508 * Source the file "name" from all directories in 'runtimepath'.
3509 * "name" can contain wildcards.
3510 * When "flags" has DIP_ALL: source all files, otherwise only the first one.
3511 *
3512 * return FAIL when no file could be sourced, OK otherwise.
3513 */
3514 int
3515source_runtime(char_u *name, int flags)
3516{
3517 return source_in_path(p_rtp, name, flags);
3518}
3519
3520/*
3521 * Just like source_runtime(), but use "path" instead of 'runtimepath'.
3522 */
3523 int
3524source_in_path(char_u *path, char_u *name, int flags)
3525{
3526 return do_in_path_and_pp(path, name, flags, source_callback, NULL);
3527}
3528
3529
3530/*
Bram Moolenaarf3654822016-03-04 22:12:23 +01003531 * Expand wildcards in "pat" and invoke do_source() for each match.
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003532 */
3533 static void
Bram Moolenaarf3654822016-03-04 22:12:23 +01003534source_all_matches(char_u *pat)
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003535{
Bram Moolenaarf3654822016-03-04 22:12:23 +01003536 int num_files;
3537 char_u **files;
3538 int i;
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003539
Bram Moolenaarf3654822016-03-04 22:12:23 +01003540 if (gen_expand_wildcards(1, &pat, &num_files, &files, EW_FILE) == OK)
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003541 {
Bram Moolenaarf3654822016-03-04 22:12:23 +01003542 for (i = 0; i < num_files; ++i)
3543 (void)do_source(files[i], FALSE, DOSO_NONE);
3544 FreeWild(num_files, files);
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003545 }
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003546}
3547
Bram Moolenaar49b27322016-04-05 21:13:00 +02003548/* used for "cookie" of add_pack_plugin() */
3549static int APP_ADD_DIR;
3550static int APP_LOAD;
3551static int APP_BOTH;
3552
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003553 static void
Bram Moolenaar91715872016-03-03 17:13:03 +01003554add_pack_plugin(char_u *fname, void *cookie)
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003555{
Bram Moolenaarf3654822016-03-04 22:12:23 +01003556 char_u *p4, *p3, *p2, *p1, *p;
3557 char_u *insp;
Bram Moolenaar91715872016-03-03 17:13:03 +01003558 int c;
3559 char_u *new_rtp;
3560 int keep;
Bram Moolenaarb0550662016-05-31 21:37:36 +02003561 size_t oldlen;
3562 size_t addlen;
Bram Moolenaara5702442016-05-24 19:37:29 +02003563 char_u *afterdir;
Bram Moolenaarb0550662016-05-31 21:37:36 +02003564 size_t afterlen = 0;
Bram Moolenaar91715872016-03-03 17:13:03 +01003565 char_u *ffname = fix_fname(fname);
Bram Moolenaarfef524b2016-07-02 22:07:22 +02003566 size_t fname_len;
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003567 char_u *buf = NULL;
3568 char_u *rtp_ffname;
3569 int match;
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003570
Bram Moolenaar91715872016-03-03 17:13:03 +01003571 if (ffname == NULL)
3572 return;
Bram Moolenaar49b27322016-04-05 21:13:00 +02003573 if (cookie != &APP_LOAD && strstr((char *)p_rtp, (char *)ffname) == NULL)
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003574 {
Bram Moolenaara5702442016-05-24 19:37:29 +02003575 /* directory is not yet in 'runtimepath', add it */
Bram Moolenaarf3654822016-03-04 22:12:23 +01003576 p4 = p3 = p2 = p1 = get_past_head(ffname);
Bram Moolenaar91acfff2017-03-12 19:22:36 +01003577 for (p = p1; *p; MB_PTR_ADV(p))
Bram Moolenaarf3654822016-03-04 22:12:23 +01003578 if (vim_ispathsep_nocolon(*p))
3579 {
3580 p4 = p3; p3 = p2; p2 = p1; p1 = p;
3581 }
3582
3583 /* now we have:
Bram Moolenaaraf1a0e32016-03-09 22:19:26 +01003584 * rtp/pack/name/start/name
3585 * p4 p3 p2 p1
Bram Moolenaarf3654822016-03-04 22:12:23 +01003586 *
3587 * find the part up to "pack" in 'runtimepath' */
3588 c = *p4;
3589 *p4 = NUL;
Bram Moolenaar4c5717e2016-07-01 15:39:40 +02003590
3591 /* Find "ffname" in "p_rtp", ignoring '/' vs '\' differences. */
3592 fname_len = STRLEN(ffname);
3593 insp = p_rtp;
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003594 buf = alloc(MAXPATHL);
3595 if (buf == NULL)
3596 goto theend;
3597 while (*insp != NUL)
Bram Moolenaar4c5717e2016-07-01 15:39:40 +02003598 {
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003599 copy_option_part(&insp, buf, MAXPATHL, ",");
3600 add_pathsep(buf);
3601 rtp_ffname = fix_fname(buf);
3602 if (rtp_ffname == NULL)
3603 goto theend;
3604 match = vim_fnamencmp(rtp_ffname, ffname, fname_len) == 0;
3605 vim_free(rtp_ffname);
3606 if (match)
Bram Moolenaar4c5717e2016-07-01 15:39:40 +02003607 break;
Bram Moolenaar4c5717e2016-07-01 15:39:40 +02003608 }
3609
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003610 if (*insp == NUL)
Bram Moolenaarf3654822016-03-04 22:12:23 +01003611 /* not found, append at the end */
3612 insp = p_rtp + STRLEN(p_rtp);
3613 else
Bram Moolenaarf3654822016-03-04 22:12:23 +01003614 /* append after the matching directory. */
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003615 --insp;
Bram Moolenaarf3654822016-03-04 22:12:23 +01003616 *p4 = c;
3617
Bram Moolenaara5702442016-05-24 19:37:29 +02003618 /* check if rtp/pack/name/start/name/after exists */
3619 afterdir = concat_fnames(ffname, (char_u *)"after", TRUE);
3620 if (afterdir != NULL && mch_isdir(afterdir))
3621 afterlen = STRLEN(afterdir) + 1; /* add one for comma */
3622
Bram Moolenaarb0550662016-05-31 21:37:36 +02003623 oldlen = STRLEN(p_rtp);
3624 addlen = STRLEN(ffname) + 1; /* add one for comma */
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003625 new_rtp = alloc((int)(oldlen + addlen + afterlen + 1));
3626 /* add one for NUL */
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003627 if (new_rtp == NULL)
Bram Moolenaarf3654822016-03-04 22:12:23 +01003628 goto theend;
3629 keep = (int)(insp - p_rtp);
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003630 mch_memmove(new_rtp, p_rtp, keep);
3631 new_rtp[keep] = ',';
Bram Moolenaara5702442016-05-24 19:37:29 +02003632 mch_memmove(new_rtp + keep + 1, ffname, addlen);
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003633 if (p_rtp[keep] != NUL)
Bram Moolenaara5702442016-05-24 19:37:29 +02003634 mch_memmove(new_rtp + keep + addlen, p_rtp + keep,
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003635 oldlen - keep + 1);
Bram Moolenaara5702442016-05-24 19:37:29 +02003636 if (afterlen > 0)
3637 {
3638 STRCAT(new_rtp, ",");
3639 STRCAT(new_rtp, afterdir);
3640 }
Bram Moolenaar863c1a92016-03-03 15:47:06 +01003641 set_option_value((char_u *)"rtp", 0L, new_rtp, 0);
3642 vim_free(new_rtp);
Bram Moolenaara5702442016-05-24 19:37:29 +02003643 vim_free(afterdir);
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003644 }
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003645
Bram Moolenaar49b27322016-04-05 21:13:00 +02003646 if (cookie != &APP_ADD_DIR)
Bram Moolenaarf3654822016-03-04 22:12:23 +01003647 {
Bram Moolenaar71fb0c12016-04-02 22:44:16 +02003648 static char *plugpat = "%s/plugin/**/*.vim";
Bram Moolenaarf3654822016-03-04 22:12:23 +01003649 static char *ftpat = "%s/ftdetect/*.vim";
3650 int len;
3651 char_u *pat;
3652
3653 len = (int)STRLEN(ffname) + (int)STRLEN(ftpat);
3654 pat = alloc(len);
3655 if (pat == NULL)
3656 goto theend;
3657 vim_snprintf((char *)pat, len, plugpat, ffname);
3658 source_all_matches(pat);
3659
3660#ifdef FEAT_AUTOCMD
3661 {
3662 char_u *cmd = vim_strsave((char_u *)"g:did_load_filetypes");
3663
3664 /* If runtime/filetype.vim wasn't loaded yet, the scripts will be
3665 * found when it loads. */
3666 if (cmd != NULL && eval_to_number(cmd) > 0)
3667 {
3668 do_cmdline_cmd((char_u *)"augroup filetypedetect");
3669 vim_snprintf((char *)pat, len, ftpat, ffname);
3670 source_all_matches(pat);
3671 do_cmdline_cmd((char_u *)"augroup END");
3672 }
3673 vim_free(cmd);
3674 }
3675#endif
Bram Moolenaarba8cd122016-03-19 14:16:39 +01003676 vim_free(pat);
Bram Moolenaarf3654822016-03-04 22:12:23 +01003677 }
3678
3679theend:
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003680 vim_free(buf);
Bram Moolenaarf3654822016-03-04 22:12:23 +01003681 vim_free(ffname);
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003682}
3683
Bram Moolenaarce876aa2017-06-04 17:47:42 +02003684/*
3685 * Add all packages in the "start" directory to 'runtimepath'.
3686 */
3687 void
3688add_pack_start_dirs(void)
3689{
3690 do_in_path(p_pp, (char_u *)"pack/*/start/*", DIP_ALL + DIP_DIR,
3691 add_pack_plugin, &APP_ADD_DIR);
3692}
3693
3694/*
3695 * Load plugins from all packages in the "start" directory.
3696 */
3697 void
3698load_start_packages(void)
3699{
3700 did_source_packages = TRUE;
3701 do_in_path(p_pp, (char_u *)"pack/*/start/*", DIP_ALL + DIP_DIR,
3702 add_pack_plugin, &APP_LOAD);
3703}
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003704
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003705/*
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003706 * ":packloadall"
Bram Moolenaarf3654822016-03-04 22:12:23 +01003707 * Find plugins in the package directories and source them.
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003708 */
3709 void
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003710ex_packloadall(exarg_T *eap)
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003711{
Bram Moolenaarce876aa2017-06-04 17:47:42 +02003712 if (!did_source_packages || eap->forceit)
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003713 {
Bram Moolenaar49b27322016-04-05 21:13:00 +02003714 /* First do a round to add all directories to 'runtimepath', then load
3715 * the plugins. This allows for plugins to use an autoload directory
3716 * of another plugin. */
Bram Moolenaarce876aa2017-06-04 17:47:42 +02003717 add_pack_start_dirs();
3718 load_start_packages();
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003719 }
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003720}
3721
3722/*
Bram Moolenaarf3654822016-03-04 22:12:23 +01003723 * ":packadd[!] {name}"
Bram Moolenaar91715872016-03-03 17:13:03 +01003724 */
3725 void
3726ex_packadd(exarg_T *eap)
3727{
3728 static char *plugpat = "pack/*/opt/%s";
3729 int len;
3730 char *pat;
3731
3732 len = (int)STRLEN(plugpat) + (int)STRLEN(eap->arg);
3733 pat = (char *)alloc(len);
3734 if (pat == NULL)
3735 return;
3736 vim_snprintf(pat, len, plugpat, eap->arg);
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003737 do_in_path(p_pp, (char_u *)pat, DIP_ALL + DIP_DIR + DIP_ERR,
Bram Moolenaar49b27322016-04-05 21:13:00 +02003738 add_pack_plugin, eap->forceit ? &APP_ADD_DIR : &APP_BOTH);
Bram Moolenaar91715872016-03-03 17:13:03 +01003739 vim_free(pat);
3740}
3741
Bram Moolenaar071d4272004-06-13 20:20:40 +00003742#if defined(FEAT_EVAL) && defined(FEAT_AUTOCMD)
3743/*
3744 * ":options"
3745 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003746 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003747ex_options(
3748 exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003749{
Bram Moolenaarab6c8582017-08-11 17:15:09 +02003750 vim_setenv((char_u *)"OPTWIN_CMD", (char_u *)(cmdmod.tab ? "tab" : ""));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003751 cmd_source((char_u *)SYS_OPTWIN_FILE, NULL);
3752}
3753#endif
3754
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01003755#if defined(FEAT_PYTHON3) || defined(FEAT_PYTHON) || defined(PROTO)
3756
3757# if (defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)) || defined(PROTO)
3758/*
3759 * Detect Python 3 or 2, and initialize 'pyxversion'.
3760 */
3761 void
3762init_pyxversion(void)
3763{
3764 if (p_pyx == 0)
3765 {
3766 if (python3_enabled(FALSE))
3767 p_pyx = 3;
3768 else if (python_enabled(FALSE))
3769 p_pyx = 2;
3770 }
3771}
3772# endif
3773
3774/*
3775 * Does a file contain one of the following strings at the beginning of any
3776 * line?
3777 * "#!(any string)python2" => returns 2
3778 * "#!(any string)python3" => returns 3
3779 * "# requires python 2.x" => returns 2
3780 * "# requires python 3.x" => returns 3
3781 * otherwise return 0.
3782 */
3783 static int
3784requires_py_version(char_u *filename)
3785{
3786 FILE *file;
3787 int requires_py_version = 0;
3788 int i, lines;
3789
3790 lines = (int)p_mls;
3791 if (lines < 0)
3792 lines = 5;
3793
3794 file = mch_fopen((char *)filename, "r");
3795 if (file != NULL)
3796 {
3797 for (i = 0; i < lines; i++)
3798 {
3799 if (vim_fgets(IObuff, IOSIZE, file))
3800 break;
3801 if (i == 0 && IObuff[0] == '#' && IObuff[1] == '!')
3802 {
3803 /* Check shebang. */
3804 if (strstr((char *)IObuff + 2, "python2") != NULL)
3805 {
3806 requires_py_version = 2;
3807 break;
3808 }
3809 if (strstr((char *)IObuff + 2, "python3") != NULL)
3810 {
3811 requires_py_version = 3;
3812 break;
3813 }
3814 }
3815 IObuff[21] = '\0';
3816 if (STRCMP("# requires python 2.x", IObuff) == 0)
3817 {
3818 requires_py_version = 2;
3819 break;
3820 }
3821 if (STRCMP("# requires python 3.x", IObuff) == 0)
3822 {
3823 requires_py_version = 3;
3824 break;
3825 }
3826 }
3827 fclose(file);
3828 }
3829 return requires_py_version;
3830}
3831
3832
3833/*
3834 * Source a python file using the requested python version.
3835 */
3836 static void
3837source_pyx_file(exarg_T *eap, char_u *fname)
3838{
3839 exarg_T ex;
3840 int v = requires_py_version(fname);
3841
3842# if defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)
3843 init_pyxversion();
3844# endif
3845 if (v == 0)
3846 {
3847# if defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)
3848 /* user didn't choose a preference, 'pyx' is used */
3849 v = p_pyx;
3850# elif defined(FEAT_PYTHON)
3851 v = 2;
3852# elif defined(FEAT_PYTHON3)
3853 v = 3;
3854# endif
3855 }
3856
3857 /*
3858 * now source, if required python version is not supported show
3859 * unobtrusive message.
3860 */
3861 if (eap == NULL)
3862 vim_memset(&ex, 0, sizeof(ex));
3863 else
3864 ex = *eap;
3865 ex.arg = fname;
3866 ex.cmd = (char_u *)(v == 2 ? "pyfile" : "pyfile3");
3867
3868 if (v == 2)
3869 {
3870# ifdef FEAT_PYTHON
3871 ex_pyfile(&ex);
3872# else
3873 vim_snprintf((char *)IObuff, IOSIZE,
3874 _("W20: Required python version 2.x not supported, ignoring file: %s"),
3875 fname);
3876 MSG(IObuff);
3877# endif
3878 return;
3879 }
3880 else
3881 {
3882# ifdef FEAT_PYTHON3
3883 ex_py3file(&ex);
3884# else
3885 vim_snprintf((char *)IObuff, IOSIZE,
3886 _("W21: Required python version 3.x not supported, ignoring file: %s"),
3887 fname);
3888 MSG(IObuff);
3889# endif
3890 return;
3891 }
3892}
3893
3894/*
3895 * ":pyxfile {fname}"
3896 */
3897 void
3898ex_pyxfile(exarg_T *eap)
3899{
3900 source_pyx_file(eap, eap->arg);
3901}
3902
3903/*
3904 * ":pyx"
3905 */
3906 void
3907ex_pyx(exarg_T *eap)
3908{
3909# if defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)
3910 init_pyxversion();
3911 if (p_pyx == 2)
3912 ex_python(eap);
3913 else
3914 ex_py3(eap);
3915# elif defined(FEAT_PYTHON)
3916 ex_python(eap);
3917# elif defined(FEAT_PYTHON3)
3918 ex_py3(eap);
3919# endif
3920}
3921
3922/*
3923 * ":pyxdo"
3924 */
3925 void
3926ex_pyxdo(exarg_T *eap)
3927{
3928# if defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)
3929 init_pyxversion();
3930 if (p_pyx == 2)
3931 ex_pydo(eap);
3932 else
3933 ex_py3do(eap);
3934# elif defined(FEAT_PYTHON)
3935 ex_pydo(eap);
3936# elif defined(FEAT_PYTHON3)
3937 ex_py3do(eap);
3938# endif
3939}
3940
3941#endif
3942
Bram Moolenaar071d4272004-06-13 20:20:40 +00003943/*
3944 * ":source {fname}"
3945 */
3946 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003947ex_source(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003948{
3949#ifdef FEAT_BROWSE
3950 if (cmdmod.browse)
3951 {
3952 char_u *fname = NULL;
3953
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00003954 fname = do_browse(0, (char_u *)_("Source Vim script"), eap->arg,
Bram Moolenaar071d4272004-06-13 20:20:40 +00003955 NULL, NULL, BROWSE_FILTER_MACROS, NULL);
3956 if (fname != NULL)
3957 {
3958 cmd_source(fname, eap);
3959 vim_free(fname);
3960 }
3961 }
3962 else
3963#endif
3964 cmd_source(eap->arg, eap);
3965}
3966
3967 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003968cmd_source(char_u *fname, exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003969{
3970 if (*fname == NUL)
3971 EMSG(_(e_argreq));
3972
Bram Moolenaar071d4272004-06-13 20:20:40 +00003973 else if (eap != NULL && eap->forceit)
Bram Moolenaar84a05ac2013-05-06 04:24:17 +02003974 /* ":source!": read Normal mode commands
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00003975 * Need to execute the commands directly. This is required at least
3976 * for:
Bram Moolenaar071d4272004-06-13 20:20:40 +00003977 * - ":g" command busy
3978 * - after ":argdo", ":windo" or ":bufdo"
3979 * - another command follows
3980 * - inside a loop
3981 */
3982 openscript(fname, global_busy || listcmd_busy || eap->nextcmd != NULL
3983#ifdef FEAT_EVAL
3984 || eap->cstack->cs_idx >= 0
3985#endif
3986 );
3987
3988 /* ":source" read ex commands */
Bram Moolenaar910f66f2006-04-05 20:41:53 +00003989 else if (do_source(fname, FALSE, DOSO_NONE) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003990 EMSG2(_(e_notopen), fname);
3991}
3992
3993/*
3994 * ":source" and associated commands.
3995 */
3996/*
3997 * Structure used to store info for each sourced file.
3998 * It is shared between do_source() and getsourceline().
3999 * This is required, because it needs to be handed to do_cmdline() and
4000 * sourcing can be done recursively.
4001 */
4002struct source_cookie
4003{
4004 FILE *fp; /* opened file for sourcing */
4005 char_u *nextline; /* if not NULL: line that was read ahead */
4006 int finished; /* ":finish" used */
Bram Moolenaar860cae12010-06-05 23:22:07 +02004007#if defined(USE_CRNL) || defined(USE_CR)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004008 int fileformat; /* EOL_UNKNOWN, EOL_UNIX or EOL_DOS */
4009 int error; /* TRUE if LF found after CR-LF */
4010#endif
4011#ifdef FEAT_EVAL
4012 linenr_T breakpoint; /* next line with breakpoint or zero */
4013 char_u *fname; /* name of sourced file */
4014 int dbg_tick; /* debug_tick when breakpoint was set */
4015 int level; /* top nesting level of sourced file */
4016#endif
4017#ifdef FEAT_MBYTE
4018 vimconv_T conv; /* type of conversion */
4019#endif
4020};
4021
4022#ifdef FEAT_EVAL
4023/*
4024 * Return the address holding the next breakpoint line for a source cookie.
4025 */
4026 linenr_T *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004027source_breakpoint(void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004028{
4029 return &((struct source_cookie *)cookie)->breakpoint;
4030}
4031
4032/*
4033 * Return the address holding the debug tick for a source cookie.
4034 */
4035 int *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004036source_dbg_tick(void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004037{
4038 return &((struct source_cookie *)cookie)->dbg_tick;
4039}
4040
4041/*
4042 * Return the nesting level for a source cookie.
4043 */
4044 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004045source_level(void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004046{
4047 return ((struct source_cookie *)cookie)->level;
4048}
4049#endif
4050
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01004051static char_u *get_one_sourceline(struct source_cookie *sp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004052
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004053#if (defined(WIN32) && defined(FEAT_CSCOPE)) || defined(HAVE_FD_CLOEXEC)
4054# define USE_FOPEN_NOINH
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01004055static FILE *fopen_noinh_readbin(char *filename);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004056
4057/*
4058 * Special function to open a file without handle inheritance.
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004059 * When possible the handle is closed on exec().
Bram Moolenaar071d4272004-06-13 20:20:40 +00004060 */
4061 static FILE *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004062fopen_noinh_readbin(char *filename)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004063{
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004064# ifdef WIN32
Bram Moolenaar8d8ef0b2010-01-20 21:41:47 +01004065 int fd_tmp = mch_open(filename, O_RDONLY | O_BINARY | O_NOINHERIT, 0);
4066# else
4067 int fd_tmp = mch_open(filename, O_RDONLY, 0);
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004068# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004069
4070 if (fd_tmp == -1)
4071 return NULL;
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004072
4073# ifdef HAVE_FD_CLOEXEC
4074 {
4075 int fdflags = fcntl(fd_tmp, F_GETFD);
4076 if (fdflags >= 0 && (fdflags & FD_CLOEXEC) == 0)
Bram Moolenaarcde88542015-08-11 19:14:00 +02004077 (void)fcntl(fd_tmp, F_SETFD, fdflags | FD_CLOEXEC);
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004078 }
4079# endif
4080
Bram Moolenaar071d4272004-06-13 20:20:40 +00004081 return fdopen(fd_tmp, READBIN);
4082}
4083#endif
4084
4085
4086/*
4087 * do_source: Read the file "fname" and execute its lines as EX commands.
4088 *
4089 * This function may be called recursively!
4090 *
4091 * return FAIL if file could not be opened, OK otherwise
4092 */
4093 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004094do_source(
4095 char_u *fname,
4096 int check_other, /* check for .vimrc and _vimrc */
4097 int is_vimrc) /* DOSO_ value */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004098{
4099 struct source_cookie cookie;
4100 char_u *save_sourcing_name;
4101 linenr_T save_sourcing_lnum;
4102 char_u *p;
4103 char_u *fname_exp;
Bram Moolenaar73881402009-02-04 16:50:47 +00004104 char_u *firstline = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004105 int retval = FAIL;
4106#ifdef FEAT_EVAL
4107 scid_T save_current_SID;
4108 static scid_T last_current_SID = 0;
4109 void *save_funccalp;
4110 int save_debug_break_level = debug_break_level;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004111 scriptitem_T *si = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004112# ifdef UNIX
Bram Moolenaar8767f522016-07-01 17:17:39 +02004113 stat_T st;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004114 int stat_ok;
4115# endif
4116#endif
4117#ifdef STARTUPTIME
4118 struct timeval tv_rel;
4119 struct timeval tv_start;
4120#endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00004121#ifdef FEAT_PROFILE
4122 proftime_T wait_start;
4123#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004124
Bram Moolenaar071d4272004-06-13 20:20:40 +00004125 p = expand_env_save(fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004126 if (p == NULL)
4127 return retval;
4128 fname_exp = fix_fname(p);
4129 vim_free(p);
4130 if (fname_exp == NULL)
4131 return retval;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004132 if (mch_isdir(fname_exp))
4133 {
Bram Moolenaar555b2802005-05-19 21:08:39 +00004134 smsg((char_u *)_("Cannot source a directory: \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004135 goto theend;
4136 }
4137
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00004138#ifdef FEAT_AUTOCMD
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00004139 /* Apply SourceCmd autocommands, they should get the file and source it. */
4140 if (has_autocmd(EVENT_SOURCECMD, fname_exp, NULL)
4141 && apply_autocmds(EVENT_SOURCECMD, fname_exp, fname_exp,
4142 FALSE, curbuf))
Bram Moolenaarf33943e2008-01-15 21:17:30 +00004143 {
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00004144# ifdef FEAT_EVAL
Bram Moolenaarf33943e2008-01-15 21:17:30 +00004145 retval = aborting() ? FAIL : OK;
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00004146# else
Bram Moolenaarf33943e2008-01-15 21:17:30 +00004147 retval = OK;
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00004148# endif
Bram Moolenaarf33943e2008-01-15 21:17:30 +00004149 goto theend;
4150 }
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00004151
4152 /* Apply SourcePre autocommands, they may get the file. */
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00004153 apply_autocmds(EVENT_SOURCEPRE, fname_exp, fname_exp, FALSE, curbuf);
4154#endif
4155
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004156#ifdef USE_FOPEN_NOINH
Bram Moolenaar071d4272004-06-13 20:20:40 +00004157 cookie.fp = fopen_noinh_readbin((char *)fname_exp);
4158#else
4159 cookie.fp = mch_fopen((char *)fname_exp, READBIN);
4160#endif
4161 if (cookie.fp == NULL && check_other)
4162 {
4163 /*
4164 * Try again, replacing file name ".vimrc" by "_vimrc" or vice versa,
4165 * and ".exrc" by "_exrc" or vice versa.
4166 */
4167 p = gettail(fname_exp);
4168 if ((*p == '.' || *p == '_')
4169 && (STRICMP(p + 1, "vimrc") == 0
4170 || STRICMP(p + 1, "gvimrc") == 0
4171 || STRICMP(p + 1, "exrc") == 0))
4172 {
4173 if (*p == '_')
4174 *p = '.';
4175 else
4176 *p = '_';
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004177#ifdef USE_FOPEN_NOINH
Bram Moolenaar071d4272004-06-13 20:20:40 +00004178 cookie.fp = fopen_noinh_readbin((char *)fname_exp);
4179#else
4180 cookie.fp = mch_fopen((char *)fname_exp, READBIN);
4181#endif
4182 }
4183 }
4184
4185 if (cookie.fp == NULL)
4186 {
4187 if (p_verbose > 0)
4188 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004189 verbose_enter();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004190 if (sourcing_name == NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00004191 smsg((char_u *)_("could not source \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004192 else
4193 smsg((char_u *)_("line %ld: could not source \"%s\""),
Bram Moolenaar555b2802005-05-19 21:08:39 +00004194 sourcing_lnum, fname);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004195 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004196 }
4197 goto theend;
4198 }
4199
4200 /*
4201 * The file exists.
4202 * - In verbose mode, give a message.
4203 * - For a vimrc file, may want to set 'compatible', call vimrc_found().
4204 */
4205 if (p_verbose > 1)
4206 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004207 verbose_enter();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004208 if (sourcing_name == NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00004209 smsg((char_u *)_("sourcing \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004210 else
4211 smsg((char_u *)_("line %ld: sourcing \"%s\""),
Bram Moolenaar555b2802005-05-19 21:08:39 +00004212 sourcing_lnum, fname);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004213 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004214 }
Bram Moolenaar910f66f2006-04-05 20:41:53 +00004215 if (is_vimrc == DOSO_VIMRC)
4216 vimrc_found(fname_exp, (char_u *)"MYVIMRC");
4217 else if (is_vimrc == DOSO_GVIMRC)
4218 vimrc_found(fname_exp, (char_u *)"MYGVIMRC");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004219
4220#ifdef USE_CRNL
4221 /* If no automatic file format: Set default to CR-NL. */
4222 if (*p_ffs == NUL)
4223 cookie.fileformat = EOL_DOS;
4224 else
4225 cookie.fileformat = EOL_UNKNOWN;
4226 cookie.error = FALSE;
4227#endif
4228
4229#ifdef USE_CR
4230 /* If no automatic file format: Set default to CR. */
4231 if (*p_ffs == NUL)
4232 cookie.fileformat = EOL_MAC;
4233 else
4234 cookie.fileformat = EOL_UNKNOWN;
4235 cookie.error = FALSE;
4236#endif
4237
4238 cookie.nextline = NULL;
4239 cookie.finished = FALSE;
4240
4241#ifdef FEAT_EVAL
4242 /*
4243 * Check if this script has a breakpoint.
4244 */
4245 cookie.breakpoint = dbg_find_breakpoint(TRUE, fname_exp, (linenr_T)0);
4246 cookie.fname = fname_exp;
4247 cookie.dbg_tick = debug_tick;
4248
4249 cookie.level = ex_nesting_level;
4250#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004251
4252 /*
4253 * Keep the sourcing name/lnum, for recursive calls.
4254 */
4255 save_sourcing_name = sourcing_name;
4256 sourcing_name = fname_exp;
4257 save_sourcing_lnum = sourcing_lnum;
4258 sourcing_lnum = 0;
4259
Bram Moolenaar73881402009-02-04 16:50:47 +00004260#ifdef FEAT_MBYTE
4261 cookie.conv.vc_type = CONV_NONE; /* no conversion */
4262
4263 /* Read the first line so we can check for a UTF-8 BOM. */
4264 firstline = getsourceline(0, (void *)&cookie, 0);
4265 if (firstline != NULL && STRLEN(firstline) >= 3 && firstline[0] == 0xef
4266 && firstline[1] == 0xbb && firstline[2] == 0xbf)
4267 {
4268 /* Found BOM; setup conversion, skip over BOM and recode the line. */
4269 convert_setup(&cookie.conv, (char_u *)"utf-8", p_enc);
4270 p = string_convert(&cookie.conv, firstline + 3, NULL);
Bram Moolenaar4e2a5952009-02-05 19:48:25 +00004271 if (p == NULL)
4272 p = vim_strsave(firstline + 3);
Bram Moolenaar73881402009-02-04 16:50:47 +00004273 if (p != NULL)
4274 {
4275 vim_free(firstline);
4276 firstline = p;
4277 }
4278 }
4279#endif
4280
Bram Moolenaar071d4272004-06-13 20:20:40 +00004281#ifdef STARTUPTIME
Bram Moolenaarc4e41982010-01-19 16:31:47 +01004282 if (time_fd != NULL)
4283 time_push(&tv_rel, &tv_start);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004284#endif
4285
4286#ifdef FEAT_EVAL
Bram Moolenaar05159a02005-02-26 23:04:13 +00004287# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004288 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004289 prof_child_enter(&wait_start); /* entering a child now */
4290# endif
4291
4292 /* Don't use local function variables, if called from a function.
4293 * Also starts profiling timer for nested script. */
4294 save_funccalp = save_funccal();
4295
Bram Moolenaar071d4272004-06-13 20:20:40 +00004296 /*
4297 * Check if this script was sourced before to finds its SID.
4298 * If it's new, generate a new SID.
4299 */
4300 save_current_SID = current_SID;
4301# ifdef UNIX
4302 stat_ok = (mch_stat((char *)fname_exp, &st) >= 0);
4303# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00004304 for (current_SID = script_items.ga_len; current_SID > 0; --current_SID)
4305 {
4306 si = &SCRIPT_ITEM(current_SID);
4307 if (si->sn_name != NULL
Bram Moolenaar071d4272004-06-13 20:20:40 +00004308 && (
4309# ifdef UNIX
Bram Moolenaar7c626922005-02-07 22:01:03 +00004310 /* Compare dev/ino when possible, it catches symbolic
4311 * links. Also compare file names, the inode may change
4312 * when the file was edited. */
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00004313 ((stat_ok && si->sn_dev_valid)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004314 && (si->sn_dev == st.st_dev
4315 && si->sn_ino == st.st_ino)) ||
Bram Moolenaar071d4272004-06-13 20:20:40 +00004316# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00004317 fnamecmp(si->sn_name, fname_exp) == 0))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004318 break;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004319 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004320 if (current_SID == 0)
4321 {
4322 current_SID = ++last_current_SID;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004323 if (ga_grow(&script_items, (int)(current_SID - script_items.ga_len))
4324 == FAIL)
4325 goto almosttheend;
4326 while (script_items.ga_len < current_SID)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004327 {
Bram Moolenaar05159a02005-02-26 23:04:13 +00004328 ++script_items.ga_len;
4329 SCRIPT_ITEM(script_items.ga_len).sn_name = NULL;
4330# ifdef FEAT_PROFILE
4331 SCRIPT_ITEM(script_items.ga_len).sn_prof_on = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004332# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004333 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00004334 si = &SCRIPT_ITEM(current_SID);
4335 si->sn_name = fname_exp;
4336 fname_exp = NULL;
4337# ifdef UNIX
4338 if (stat_ok)
4339 {
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00004340 si->sn_dev_valid = TRUE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004341 si->sn_dev = st.st_dev;
4342 si->sn_ino = st.st_ino;
4343 }
4344 else
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00004345 si->sn_dev_valid = FALSE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004346# endif
4347
Bram Moolenaar071d4272004-06-13 20:20:40 +00004348 /* Allocate the local script variables to use for this script. */
4349 new_script_vars(current_SID);
4350 }
4351
Bram Moolenaar05159a02005-02-26 23:04:13 +00004352# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004353 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004354 {
4355 int forceit;
4356
4357 /* Check if we do profiling for this script. */
4358 if (!si->sn_prof_on && has_profiling(TRUE, si->sn_name, &forceit))
4359 {
4360 script_do_profile(si);
4361 si->sn_pr_force = forceit;
4362 }
4363 if (si->sn_prof_on)
4364 {
4365 ++si->sn_pr_count;
4366 profile_start(&si->sn_pr_start);
4367 profile_zero(&si->sn_pr_children);
4368 }
4369 }
4370# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004371#endif
4372
4373 /*
4374 * Call do_cmdline, which will call getsourceline() to get the lines.
4375 */
Bram Moolenaar73881402009-02-04 16:50:47 +00004376 do_cmdline(firstline, getsourceline, (void *)&cookie,
Bram Moolenaar071d4272004-06-13 20:20:40 +00004377 DOCMD_VERBOSE|DOCMD_NOWAIT|DOCMD_REPEAT);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004378 retval = OK;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004379
4380#ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004381 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004382 {
4383 /* Get "si" again, "script_items" may have been reallocated. */
4384 si = &SCRIPT_ITEM(current_SID);
4385 if (si->sn_prof_on)
4386 {
4387 profile_end(&si->sn_pr_start);
4388 profile_sub_wait(&wait_start, &si->sn_pr_start);
4389 profile_add(&si->sn_pr_total, &si->sn_pr_start);
Bram Moolenaar1056d982006-03-09 22:37:52 +00004390 profile_self(&si->sn_pr_self, &si->sn_pr_start,
4391 &si->sn_pr_children);
Bram Moolenaar05159a02005-02-26 23:04:13 +00004392 }
4393 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004394#endif
4395
4396 if (got_int)
4397 EMSG(_(e_interr));
4398 sourcing_name = save_sourcing_name;
4399 sourcing_lnum = save_sourcing_lnum;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004400 if (p_verbose > 1)
4401 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004402 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00004403 smsg((char_u *)_("finished sourcing %s"), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004404 if (sourcing_name != NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00004405 smsg((char_u *)_("continuing in %s"), sourcing_name);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004406 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004407 }
4408#ifdef STARTUPTIME
Bram Moolenaarc4e41982010-01-19 16:31:47 +01004409 if (time_fd != NULL)
4410 {
4411 vim_snprintf((char *)IObuff, IOSIZE, "sourcing %s", fname);
4412 time_msg((char *)IObuff, &tv_start);
4413 time_pop(&tv_rel);
4414 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004415#endif
4416
4417#ifdef FEAT_EVAL
4418 /*
4419 * After a "finish" in debug mode, need to break at first command of next
4420 * sourced file.
4421 */
4422 if (save_debug_break_level > ex_nesting_level
4423 && debug_break_level == ex_nesting_level)
4424 ++debug_break_level;
4425#endif
4426
Bram Moolenaar05159a02005-02-26 23:04:13 +00004427#ifdef FEAT_EVAL
4428almosttheend:
4429 current_SID = save_current_SID;
4430 restore_funccal(save_funccalp);
4431# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004432 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004433 prof_child_exit(&wait_start); /* leaving a child now */
4434# endif
4435#endif
4436 fclose(cookie.fp);
4437 vim_free(cookie.nextline);
Bram Moolenaar73881402009-02-04 16:50:47 +00004438 vim_free(firstline);
Bram Moolenaar05159a02005-02-26 23:04:13 +00004439#ifdef FEAT_MBYTE
4440 convert_setup(&cookie.conv, NULL, NULL);
4441#endif
4442
Bram Moolenaar071d4272004-06-13 20:20:40 +00004443theend:
4444 vim_free(fname_exp);
4445 return retval;
4446}
4447
4448#if defined(FEAT_EVAL) || defined(PROTO)
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00004449
Bram Moolenaar071d4272004-06-13 20:20:40 +00004450/*
4451 * ":scriptnames"
4452 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004453 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004454ex_scriptnames(exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004455{
4456 int i;
4457
Bram Moolenaar05159a02005-02-26 23:04:13 +00004458 for (i = 1; i <= script_items.ga_len && !got_int; ++i)
4459 if (SCRIPT_ITEM(i).sn_name != NULL)
Bram Moolenaard58ea072011-06-26 04:25:30 +02004460 {
4461 home_replace(NULL, SCRIPT_ITEM(i).sn_name,
4462 NameBuff, MAXPATHL, TRUE);
4463 smsg((char_u *)"%3d: %s", i, NameBuff);
Bram Moolenaar970a1b82012-03-23 18:39:18 +01004464 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004465}
4466
4467# if defined(BACKSLASH_IN_FILENAME) || defined(PROTO)
4468/*
4469 * Fix slashes in the list of script names for 'shellslash'.
4470 */
4471 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004472scriptnames_slash_adjust(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004473{
4474 int i;
4475
Bram Moolenaar05159a02005-02-26 23:04:13 +00004476 for (i = 1; i <= script_items.ga_len; ++i)
4477 if (SCRIPT_ITEM(i).sn_name != NULL)
4478 slash_adjust(SCRIPT_ITEM(i).sn_name);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004479}
4480# endif
4481
4482/*
4483 * Get a pointer to a script name. Used for ":verbose set".
4484 */
4485 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004486get_scriptname(scid_T id)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004487{
4488 if (id == SID_MODELINE)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00004489 return (char_u *)_("modeline");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004490 if (id == SID_CMDARG)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00004491 return (char_u *)_("--cmd argument");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004492 if (id == SID_CARG)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00004493 return (char_u *)_("-c argument");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004494 if (id == SID_ENV)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00004495 return (char_u *)_("environment variable");
4496 if (id == SID_ERROR)
4497 return (char_u *)_("error handler");
Bram Moolenaar05159a02005-02-26 23:04:13 +00004498 return SCRIPT_ITEM(id).sn_name;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004499}
Bram Moolenaar05159a02005-02-26 23:04:13 +00004500
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00004501# if defined(EXITFREE) || defined(PROTO)
4502 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004503free_scriptnames(void)
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00004504{
4505 int i;
4506
4507 for (i = script_items.ga_len; i > 0; --i)
4508 vim_free(SCRIPT_ITEM(i).sn_name);
4509 ga_clear(&script_items);
4510}
4511# endif
4512
Bram Moolenaar071d4272004-06-13 20:20:40 +00004513#endif
4514
4515#if defined(USE_CR) || defined(PROTO)
4516
4517# if defined(__MSL__) && (__MSL__ >= 22)
4518/*
4519 * Newer version of the Metrowerks library handle DOS and UNIX files
4520 * without help.
4521 * Test with earlier versions, MSL 2.2 is the library supplied with
4522 * Codewarrior Pro 2.
4523 */
4524 char *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004525fgets_cr(char *s, int n, FILE *stream)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004526{
4527 return fgets(s, n, stream);
4528}
4529# else
4530/*
4531 * Version of fgets() which also works for lines ending in a <CR> only
4532 * (Macintosh format).
4533 * For older versions of the Metrowerks library.
4534 * At least CodeWarrior 9 needed this code.
4535 */
4536 char *
Bram Moolenaard14e00e2016-01-31 17:30:51 +01004537fgets_cr(char *s, int n, FILE *stream)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004538{
4539 int c = 0;
4540 int char_read = 0;
4541
4542 while (!feof(stream) && c != '\r' && c != '\n' && char_read < n - 1)
4543 {
4544 c = fgetc(stream);
4545 s[char_read++] = c;
4546 /* If the file is in DOS format, we need to skip a NL after a CR. I
4547 * thought it was the other way around, but this appears to work... */
4548 if (c == '\n')
4549 {
4550 c = fgetc(stream);
4551 if (c != '\r')
4552 ungetc(c, stream);
4553 }
4554 }
4555
4556 s[char_read] = 0;
4557 if (char_read == 0)
4558 return NULL;
4559
4560 if (feof(stream) && char_read == 1)
4561 return NULL;
4562
4563 return s;
4564}
4565# endif
4566#endif
4567
4568/*
4569 * Get one full line from a sourced file.
4570 * Called by do_cmdline() when it's called from do_source().
4571 *
4572 * Return a pointer to the line in allocated memory.
4573 * Return NULL for end-of-file or some error.
4574 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004575 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004576getsourceline(int c UNUSED, void *cookie, int indent UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004577{
4578 struct source_cookie *sp = (struct source_cookie *)cookie;
4579 char_u *line;
Bram Moolenaarc047b9a2012-02-11 20:40:55 +01004580 char_u *p;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004581
4582#ifdef FEAT_EVAL
4583 /* If breakpoints have been added/deleted need to check for it. */
4584 if (sp->dbg_tick < debug_tick)
4585 {
4586 sp->breakpoint = dbg_find_breakpoint(TRUE, sp->fname, sourcing_lnum);
4587 sp->dbg_tick = debug_tick;
4588 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00004589# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004590 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004591 script_line_end();
4592# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004593#endif
4594 /*
4595 * Get current line. If there is a read-ahead line, use it, otherwise get
4596 * one now.
4597 */
4598 if (sp->finished)
4599 line = NULL;
4600 else if (sp->nextline == NULL)
4601 line = get_one_sourceline(sp);
4602 else
4603 {
4604 line = sp->nextline;
4605 sp->nextline = NULL;
4606 ++sourcing_lnum;
4607 }
Bram Moolenaare2cc9702005-03-15 22:43:58 +00004608#ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004609 if (line != NULL && do_profiling == PROF_YES)
Bram Moolenaare2cc9702005-03-15 22:43:58 +00004610 script_line_start();
4611#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004612
4613 /* Only concatenate lines starting with a \ when 'cpoptions' doesn't
4614 * contain the 'C' flag. */
4615 if (line != NULL && (vim_strchr(p_cpo, CPO_CONCAT) == NULL))
4616 {
4617 /* compensate for the one line read-ahead */
4618 --sourcing_lnum;
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004619
4620 /* Get the next line and concatenate it when it starts with a
4621 * backslash. We always need to read the next line, keep it in
4622 * sp->nextline. */
4623 sp->nextline = get_one_sourceline(sp);
4624 if (sp->nextline != NULL && *(p = skipwhite(sp->nextline)) == '\\')
Bram Moolenaar071d4272004-06-13 20:20:40 +00004625 {
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004626 garray_T ga;
4627
Bram Moolenaarb549a732012-02-22 18:29:33 +01004628 ga_init2(&ga, (int)sizeof(char_u), 400);
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004629 ga_concat(&ga, line);
4630 ga_concat(&ga, p + 1);
4631 for (;;)
4632 {
4633 vim_free(sp->nextline);
4634 sp->nextline = get_one_sourceline(sp);
4635 if (sp->nextline == NULL)
4636 break;
4637 p = skipwhite(sp->nextline);
4638 if (*p != '\\')
4639 break;
Bram Moolenaarb549a732012-02-22 18:29:33 +01004640 /* Adjust the growsize to the current length to speed up
4641 * concatenating many lines. */
4642 if (ga.ga_len > 400)
4643 {
4644 if (ga.ga_len > 8000)
4645 ga.ga_growsize = 8000;
4646 else
4647 ga.ga_growsize = ga.ga_len;
4648 }
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004649 ga_concat(&ga, p + 1);
4650 }
4651 ga_append(&ga, NUL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004652 vim_free(line);
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004653 line = ga.ga_data;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004654 }
4655 }
4656
4657#ifdef FEAT_MBYTE
4658 if (line != NULL && sp->conv.vc_type != CONV_NONE)
4659 {
Bram Moolenaarc047b9a2012-02-11 20:40:55 +01004660 char_u *s;
4661
Bram Moolenaar071d4272004-06-13 20:20:40 +00004662 /* Convert the encoding of the script line. */
4663 s = string_convert(&sp->conv, line, NULL);
4664 if (s != NULL)
4665 {
4666 vim_free(line);
4667 line = s;
4668 }
4669 }
4670#endif
4671
4672#ifdef FEAT_EVAL
4673 /* Did we encounter a breakpoint? */
4674 if (sp->breakpoint != 0 && sp->breakpoint <= sourcing_lnum)
4675 {
4676 dbg_breakpoint(sp->fname, sourcing_lnum);
4677 /* Find next breakpoint. */
4678 sp->breakpoint = dbg_find_breakpoint(TRUE, sp->fname, sourcing_lnum);
4679 sp->dbg_tick = debug_tick;
4680 }
4681#endif
4682
4683 return line;
4684}
4685
4686 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004687get_one_sourceline(struct source_cookie *sp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004688{
4689 garray_T ga;
4690 int len;
4691 int c;
4692 char_u *buf;
4693#ifdef USE_CRNL
4694 int has_cr; /* CR-LF found */
4695#endif
4696#ifdef USE_CR
4697 char_u *scan;
4698#endif
4699 int have_read = FALSE;
4700
4701 /* use a growarray to store the sourced line */
Bram Moolenaar6ac54292005-02-02 23:07:25 +00004702 ga_init2(&ga, 1, 250);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004703
4704 /*
4705 * Loop until there is a finished line (or end-of-file).
4706 */
4707 sourcing_lnum++;
4708 for (;;)
4709 {
Bram Moolenaar6ac54292005-02-02 23:07:25 +00004710 /* make room to read at least 120 (more) characters */
4711 if (ga_grow(&ga, 120) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004712 break;
4713 buf = (char_u *)ga.ga_data;
4714
4715#ifdef USE_CR
4716 if (sp->fileformat == EOL_MAC)
4717 {
Bram Moolenaar86b68352004-12-27 21:59:20 +00004718 if (fgets_cr((char *)buf + ga.ga_len, ga.ga_maxlen - ga.ga_len,
4719 sp->fp) == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004720 break;
4721 }
4722 else
4723#endif
Bram Moolenaar86b68352004-12-27 21:59:20 +00004724 if (fgets((char *)buf + ga.ga_len, ga.ga_maxlen - ga.ga_len,
4725 sp->fp) == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004726 break;
Bram Moolenaar6ac54292005-02-02 23:07:25 +00004727 len = ga.ga_len + (int)STRLEN(buf + ga.ga_len);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004728#ifdef USE_CRNL
4729 /* Ignore a trailing CTRL-Z, when in Dos mode. Only recognize the
4730 * CTRL-Z by its own, or after a NL. */
4731 if ( (len == 1 || (len >= 2 && buf[len - 2] == '\n'))
4732 && sp->fileformat == EOL_DOS
4733 && buf[len - 1] == Ctrl_Z)
4734 {
4735 buf[len - 1] = NUL;
4736 break;
4737 }
4738#endif
4739
4740#ifdef USE_CR
4741 /* If the read doesn't stop on a new line, and there's
4742 * some CR then we assume a Mac format */
4743 if (sp->fileformat == EOL_UNKNOWN)
4744 {
4745 if (buf[len - 1] != '\n' && vim_strchr(buf, '\r') != NULL)
4746 sp->fileformat = EOL_MAC;
4747 else
4748 sp->fileformat = EOL_UNIX;
4749 }
4750
4751 if (sp->fileformat == EOL_MAC)
4752 {
4753 scan = vim_strchr(buf, '\r');
4754
4755 if (scan != NULL)
4756 {
4757 *scan = '\n';
4758 if (*(scan + 1) != 0)
4759 {
4760 *(scan + 1) = 0;
4761 fseek(sp->fp, (long)(scan - buf - len + 1), SEEK_CUR);
4762 }
4763 }
4764 len = STRLEN(buf);
4765 }
4766#endif
4767
4768 have_read = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004769 ga.ga_len = len;
4770
4771 /* If the line was longer than the buffer, read more. */
Bram Moolenaar86b68352004-12-27 21:59:20 +00004772 if (ga.ga_maxlen - ga.ga_len == 1 && buf[len - 1] != '\n')
Bram Moolenaar071d4272004-06-13 20:20:40 +00004773 continue;
4774
4775 if (len >= 1 && buf[len - 1] == '\n') /* remove trailing NL */
4776 {
4777#ifdef USE_CRNL
4778 has_cr = (len >= 2 && buf[len - 2] == '\r');
4779 if (sp->fileformat == EOL_UNKNOWN)
4780 {
4781 if (has_cr)
4782 sp->fileformat = EOL_DOS;
4783 else
4784 sp->fileformat = EOL_UNIX;
4785 }
4786
4787 if (sp->fileformat == EOL_DOS)
4788 {
4789 if (has_cr) /* replace trailing CR */
4790 {
4791 buf[len - 2] = '\n';
4792 --len;
4793 --ga.ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004794 }
4795 else /* lines like ":map xx yy^M" will have failed */
4796 {
4797 if (!sp->error)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00004798 {
Bram Moolenaar8820b482017-03-16 17:23:31 +01004799 msg_source(HL_ATTR(HLF_W));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004800 EMSG(_("W15: Warning: Wrong line separator, ^M may be missing"));
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00004801 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004802 sp->error = TRUE;
4803 sp->fileformat = EOL_UNIX;
4804 }
4805 }
4806#endif
4807 /* The '\n' is escaped if there is an odd number of ^V's just
4808 * before it, first set "c" just before the 'V's and then check
4809 * len&c parities (is faster than ((len-c)%2 == 0)) -- Acevedo */
4810 for (c = len - 2; c >= 0 && buf[c] == Ctrl_V; c--)
4811 ;
4812 if ((len & 1) != (c & 1)) /* escaped NL, read more */
4813 {
4814 sourcing_lnum++;
4815 continue;
4816 }
4817
4818 buf[len - 1] = NUL; /* remove the NL */
4819 }
4820
4821 /*
4822 * Check for ^C here now and then, so recursive :so can be broken.
4823 */
4824 line_breakcheck();
4825 break;
4826 }
4827
4828 if (have_read)
4829 return (char_u *)ga.ga_data;
4830
4831 vim_free(ga.ga_data);
4832 return NULL;
4833}
4834
Bram Moolenaar05159a02005-02-26 23:04:13 +00004835#if defined(FEAT_PROFILE) || defined(PROTO)
4836/*
4837 * Called when starting to read a script line.
4838 * "sourcing_lnum" must be correct!
4839 * When skipping lines it may not actually be executed, but we won't find out
4840 * until later and we need to store the time now.
4841 */
4842 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004843script_line_start(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004844{
4845 scriptitem_T *si;
4846 sn_prl_T *pp;
4847
4848 if (current_SID <= 0 || current_SID > script_items.ga_len)
4849 return;
4850 si = &SCRIPT_ITEM(current_SID);
4851 if (si->sn_prof_on && sourcing_lnum >= 1)
4852 {
Bram Moolenaar8c8de832008-06-24 22:58:06 +00004853 /* Grow the array before starting the timer, so that the time spent
Bram Moolenaar05159a02005-02-26 23:04:13 +00004854 * here isn't counted. */
Bram Moolenaarcde88542015-08-11 19:14:00 +02004855 (void)ga_grow(&si->sn_prl_ga, (int)(sourcing_lnum - si->sn_prl_ga.ga_len));
Bram Moolenaar05159a02005-02-26 23:04:13 +00004856 si->sn_prl_idx = sourcing_lnum - 1;
4857 while (si->sn_prl_ga.ga_len <= si->sn_prl_idx
4858 && si->sn_prl_ga.ga_len < si->sn_prl_ga.ga_maxlen)
4859 {
4860 /* Zero counters for a line that was not used before. */
4861 pp = &PRL_ITEM(si, si->sn_prl_ga.ga_len);
4862 pp->snp_count = 0;
4863 profile_zero(&pp->sn_prl_total);
4864 profile_zero(&pp->sn_prl_self);
4865 ++si->sn_prl_ga.ga_len;
4866 }
4867 si->sn_prl_execed = FALSE;
4868 profile_start(&si->sn_prl_start);
4869 profile_zero(&si->sn_prl_children);
4870 profile_get_wait(&si->sn_prl_wait);
4871 }
4872}
4873
4874/*
4875 * Called when actually executing a function line.
4876 */
4877 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004878script_line_exec(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004879{
4880 scriptitem_T *si;
4881
4882 if (current_SID <= 0 || current_SID > script_items.ga_len)
4883 return;
4884 si = &SCRIPT_ITEM(current_SID);
4885 if (si->sn_prof_on && si->sn_prl_idx >= 0)
4886 si->sn_prl_execed = TRUE;
4887}
4888
4889/*
4890 * Called when done with a function line.
4891 */
4892 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004893script_line_end(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004894{
4895 scriptitem_T *si;
4896 sn_prl_T *pp;
4897
4898 if (current_SID <= 0 || current_SID > script_items.ga_len)
4899 return;
4900 si = &SCRIPT_ITEM(current_SID);
4901 if (si->sn_prof_on && si->sn_prl_idx >= 0
4902 && si->sn_prl_idx < si->sn_prl_ga.ga_len)
4903 {
4904 if (si->sn_prl_execed)
4905 {
4906 pp = &PRL_ITEM(si, si->sn_prl_idx);
4907 ++pp->snp_count;
4908 profile_end(&si->sn_prl_start);
4909 profile_sub_wait(&si->sn_prl_wait, &si->sn_prl_start);
Bram Moolenaar05159a02005-02-26 23:04:13 +00004910 profile_add(&pp->sn_prl_total, &si->sn_prl_start);
Bram Moolenaar1056d982006-03-09 22:37:52 +00004911 profile_self(&pp->sn_prl_self, &si->sn_prl_start,
4912 &si->sn_prl_children);
Bram Moolenaar05159a02005-02-26 23:04:13 +00004913 }
4914 si->sn_prl_idx = -1;
4915 }
4916}
4917#endif
4918
Bram Moolenaar071d4272004-06-13 20:20:40 +00004919/*
4920 * ":scriptencoding": Set encoding conversion for a sourced script.
4921 * Without the multi-byte feature it's simply ignored.
4922 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004923 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004924ex_scriptencoding(exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004925{
4926#ifdef FEAT_MBYTE
4927 struct source_cookie *sp;
4928 char_u *name;
4929
4930 if (!getline_equal(eap->getline, eap->cookie, getsourceline))
4931 {
4932 EMSG(_("E167: :scriptencoding used outside of a sourced file"));
4933 return;
4934 }
4935
4936 if (*eap->arg != NUL)
4937 {
4938 name = enc_canonize(eap->arg);
4939 if (name == NULL) /* out of memory */
4940 return;
4941 }
4942 else
4943 name = eap->arg;
4944
4945 /* Setup for conversion from the specified encoding to 'encoding'. */
4946 sp = (struct source_cookie *)getline_cookie(eap->getline, eap->cookie);
4947 convert_setup(&sp->conv, name, p_enc);
4948
4949 if (name != eap->arg)
4950 vim_free(name);
4951#endif
4952}
4953
4954#if defined(FEAT_EVAL) || defined(PROTO)
4955/*
4956 * ":finish": Mark a sourced file as finished.
4957 */
4958 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004959ex_finish(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004960{
4961 if (getline_equal(eap->getline, eap->cookie, getsourceline))
4962 do_finish(eap, FALSE);
4963 else
4964 EMSG(_("E168: :finish used outside of a sourced file"));
4965}
4966
4967/*
4968 * Mark a sourced file as finished. Possibly makes the ":finish" pending.
4969 * Also called for a pending finish at the ":endtry" or after returning from
4970 * an extra do_cmdline(). "reanimate" is used in the latter case.
4971 */
4972 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004973do_finish(exarg_T *eap, int reanimate)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004974{
4975 int idx;
4976
4977 if (reanimate)
4978 ((struct source_cookie *)getline_cookie(eap->getline,
4979 eap->cookie))->finished = FALSE;
4980
4981 /*
4982 * Cleanup (and inactivate) conditionals, but stop when a try conditional
4983 * not in its finally clause (which then is to be executed next) is found.
4984 * In this case, make the ":finish" pending for execution at the ":endtry".
4985 * Otherwise, finish normally.
4986 */
4987 idx = cleanup_conditionals(eap->cstack, 0, TRUE);
4988 if (idx >= 0)
4989 {
4990 eap->cstack->cs_pending[idx] = CSTP_FINISH;
4991 report_make_pending(CSTP_FINISH, NULL);
4992 }
4993 else
4994 ((struct source_cookie *)getline_cookie(eap->getline,
4995 eap->cookie))->finished = TRUE;
4996}
4997
4998
4999/*
5000 * Return TRUE when a sourced file had the ":finish" command: Don't give error
5001 * message for missing ":endif".
5002 * Return FALSE when not sourcing a file.
5003 */
5004 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005005source_finished(
5006 char_u *(*fgetline)(int, void *, int),
5007 void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005008{
Bram Moolenaar89d40322006-08-29 15:30:07 +00005009 return (getline_equal(fgetline, cookie, getsourceline)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005010 && ((struct source_cookie *)getline_cookie(
Bram Moolenaar89d40322006-08-29 15:30:07 +00005011 fgetline, cookie))->finished);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005012}
5013#endif
5014
5015#if defined(FEAT_LISTCMDS) || defined(PROTO)
5016/*
5017 * ":checktime [buffer]"
5018 */
5019 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005020ex_checktime(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005021{
5022 buf_T *buf;
5023 int save_no_check_timestamps = no_check_timestamps;
5024
5025 no_check_timestamps = 0;
5026 if (eap->addr_count == 0) /* default is all buffers */
5027 check_timestamps(FALSE);
5028 else
5029 {
5030 buf = buflist_findnr((int)eap->line2);
5031 if (buf != NULL) /* cannot happen? */
5032 (void)buf_check_timestamp(buf, FALSE);
5033 }
5034 no_check_timestamps = save_no_check_timestamps;
5035}
5036#endif
5037
Bram Moolenaar071d4272004-06-13 20:20:40 +00005038#if (defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
5039 && (defined(FEAT_EVAL) || defined(FEAT_MULTI_LANG))
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02005040# define HAVE_GET_LOCALE_VAL
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005041static char_u *get_locale_val(int what);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005042
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005043 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005044get_locale_val(int what)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005045{
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005046 char_u *loc;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005047
Bram Moolenaar48e330a2016-02-23 14:53:34 +01005048 /* Obtain the locale value from the libraries. */
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005049 loc = (char_u *)setlocale(what, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005050
Bram Moolenaar61660ea2006-04-07 21:40:07 +00005051# ifdef WIN32
Bram Moolenaar071d4272004-06-13 20:20:40 +00005052 if (loc != NULL)
5053 {
5054 char_u *p;
5055
Bram Moolenaar61660ea2006-04-07 21:40:07 +00005056 /* setocale() returns something like "LC_COLLATE=<name>;LC_..." when
5057 * one of the values (e.g., LC_CTYPE) differs. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005058 p = vim_strchr(loc, '=');
5059 if (p != NULL)
5060 {
5061 loc = ++p;
5062 while (*p != NUL) /* remove trailing newline */
5063 {
Bram Moolenaar61660ea2006-04-07 21:40:07 +00005064 if (*p < ' ' || *p == ';')
Bram Moolenaar071d4272004-06-13 20:20:40 +00005065 {
5066 *p = NUL;
5067 break;
5068 }
5069 ++p;
5070 }
5071 }
5072 }
5073# endif
5074
5075 return loc;
5076}
5077#endif
5078
5079
5080#ifdef WIN32
5081/*
5082 * On MS-Windows locale names are strings like "German_Germany.1252", but
5083 * gettext expects "de". Try to translate one into another here for a few
5084 * supported languages.
5085 */
5086 static char_u *
5087gettext_lang(char_u *name)
5088{
5089 int i;
5090 static char *(mtable[]) = {
5091 "afrikaans", "af",
5092 "czech", "cs",
5093 "dutch", "nl",
5094 "german", "de",
5095 "english_united kingdom", "en_GB",
5096 "spanish", "es",
5097 "french", "fr",
5098 "italian", "it",
5099 "japanese", "ja",
5100 "korean", "ko",
5101 "norwegian", "no",
5102 "polish", "pl",
5103 "russian", "ru",
5104 "slovak", "sk",
5105 "swedish", "sv",
5106 "ukrainian", "uk",
5107 "chinese_china", "zh_CN",
5108 "chinese_taiwan", "zh_TW",
5109 NULL};
5110
5111 for (i = 0; mtable[i] != NULL; i += 2)
5112 if (STRNICMP(mtable[i], name, STRLEN(mtable[i])) == 0)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005113 return (char_u *)mtable[i + 1];
Bram Moolenaar071d4272004-06-13 20:20:40 +00005114 return name;
5115}
5116#endif
5117
5118#if defined(FEAT_MULTI_LANG) || defined(PROTO)
5119/*
5120 * Obtain the current messages language. Used to set the default for
5121 * 'helplang'. May return NULL or an empty string.
5122 */
5123 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005124get_mess_lang(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005125{
5126 char_u *p;
5127
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02005128# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00005129# if defined(LC_MESSAGES)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005130 p = get_locale_val(LC_MESSAGES);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005131# else
5132 /* This is necessary for Win32, where LC_MESSAGES is not defined and $LANG
Bram Moolenaar61660ea2006-04-07 21:40:07 +00005133 * may be set to the LCID number. LC_COLLATE is the best guess, LC_TIME
5134 * and LC_MONETARY may be set differently for a Japanese working in the
5135 * US. */
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005136 p = get_locale_val(LC_COLLATE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005137# endif
5138# else
5139 p = mch_getenv((char_u *)"LC_ALL");
5140 if (p == NULL || *p == NUL)
5141 {
5142 p = mch_getenv((char_u *)"LC_MESSAGES");
5143 if (p == NULL || *p == NUL)
5144 p = mch_getenv((char_u *)"LANG");
5145 }
5146# endif
5147# ifdef WIN32
5148 p = gettext_lang(p);
5149# endif
5150 return p;
5151}
5152#endif
5153
Bram Moolenaardef9e822004-12-31 20:58:58 +00005154/* Complicated #if; matches with where get_mess_env() is used below. */
5155#if (defined(FEAT_EVAL) && !((defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
5156 && defined(LC_MESSAGES))) \
5157 || ((defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
5158 && (defined(FEAT_GETTEXT) || defined(FEAT_MBYTE)) \
5159 && !defined(LC_MESSAGES))
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01005160static char_u *get_mess_env(void);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005161
5162/*
5163 * Get the language used for messages from the environment.
5164 */
5165 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005166get_mess_env(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005167{
5168 char_u *p;
5169
5170 p = mch_getenv((char_u *)"LC_ALL");
5171 if (p == NULL || *p == NUL)
5172 {
5173 p = mch_getenv((char_u *)"LC_MESSAGES");
5174 if (p == NULL || *p == NUL)
5175 {
5176 p = mch_getenv((char_u *)"LANG");
5177 if (p != NULL && VIM_ISDIGIT(*p))
5178 p = NULL; /* ignore something like "1043" */
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02005179# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00005180 if (p == NULL || *p == NUL)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005181 p = get_locale_val(LC_CTYPE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005182# endif
5183 }
5184 }
5185 return p;
5186}
5187#endif
5188
5189#if defined(FEAT_EVAL) || defined(PROTO)
5190
5191/*
5192 * Set the "v:lang" variable according to the current locale setting.
5193 * Also do "v:lc_time"and "v:ctype".
5194 */
5195 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005196set_lang_var(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005197{
5198 char_u *loc;
5199
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02005200# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005201 loc = get_locale_val(LC_CTYPE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005202# else
5203 /* setlocale() not supported: use the default value */
5204 loc = (char_u *)"C";
5205# endif
5206 set_vim_var_string(VV_CTYPE, loc, -1);
5207
5208 /* When LC_MESSAGES isn't defined use the value from $LC_MESSAGES, fall
5209 * back to LC_CTYPE if it's empty. */
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02005210# if defined(HAVE_GET_LOCALE_VAL) && defined(LC_MESSAGES)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005211 loc = get_locale_val(LC_MESSAGES);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005212# else
5213 loc = get_mess_env();
5214# endif
5215 set_vim_var_string(VV_LANG, loc, -1);
5216
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02005217# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005218 loc = get_locale_val(LC_TIME);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005219# endif
5220 set_vim_var_string(VV_LC_TIME, loc, -1);
5221}
5222#endif
5223
5224#if (defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
5225 && (defined(FEAT_GETTEXT) || defined(FEAT_MBYTE))
5226/*
5227 * ":language": Set the language (locale).
5228 */
5229 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005230ex_language(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005231{
5232 char *loc;
5233 char_u *p;
5234 char_u *name;
5235 int what = LC_ALL;
5236 char *whatstr = "";
5237#ifdef LC_MESSAGES
5238# define VIM_LC_MESSAGES LC_MESSAGES
5239#else
5240# define VIM_LC_MESSAGES 6789
5241#endif
5242
5243 name = eap->arg;
5244
5245 /* Check for "messages {name}", "ctype {name}" or "time {name}" argument.
5246 * Allow abbreviation, but require at least 3 characters to avoid
5247 * confusion with a two letter language name "me" or "ct". */
5248 p = skiptowhite(eap->arg);
Bram Moolenaar1c465442017-03-12 20:10:05 +01005249 if ((*p == NUL || VIM_ISWHITE(*p)) && p - eap->arg >= 3)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005250 {
5251 if (STRNICMP(eap->arg, "messages", p - eap->arg) == 0)
5252 {
5253 what = VIM_LC_MESSAGES;
5254 name = skipwhite(p);
5255 whatstr = "messages ";
5256 }
5257 else if (STRNICMP(eap->arg, "ctype", p - eap->arg) == 0)
5258 {
5259 what = LC_CTYPE;
5260 name = skipwhite(p);
5261 whatstr = "ctype ";
5262 }
5263 else if (STRNICMP(eap->arg, "time", p - eap->arg) == 0)
5264 {
5265 what = LC_TIME;
5266 name = skipwhite(p);
5267 whatstr = "time ";
5268 }
5269 }
5270
5271 if (*name == NUL)
5272 {
5273#ifndef LC_MESSAGES
5274 if (what == VIM_LC_MESSAGES)
5275 p = get_mess_env();
5276 else
5277#endif
5278 p = (char_u *)setlocale(what, NULL);
5279 if (p == NULL || *p == NUL)
5280 p = (char_u *)"Unknown";
5281 smsg((char_u *)_("Current %slanguage: \"%s\""), whatstr, p);
5282 }
5283 else
5284 {
5285#ifndef LC_MESSAGES
5286 if (what == VIM_LC_MESSAGES)
5287 loc = "";
5288 else
5289#endif
Bram Moolenaar8c8de832008-06-24 22:58:06 +00005290 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00005291 loc = setlocale(what, (char *)name);
Bram Moolenaar8c8de832008-06-24 22:58:06 +00005292#if defined(FEAT_FLOAT) && defined(LC_NUMERIC)
5293 /* Make sure strtod() uses a decimal point, not a comma. */
5294 setlocale(LC_NUMERIC, "C");
5295#endif
5296 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00005297 if (loc == NULL)
5298 EMSG2(_("E197: Cannot set language to \"%s\""), name);
5299 else
5300 {
5301#ifdef HAVE_NL_MSG_CAT_CNTR
5302 /* Need to do this for GNU gettext, otherwise cached translations
5303 * will be used again. */
5304 extern int _nl_msg_cat_cntr;
5305
5306 ++_nl_msg_cat_cntr;
5307#endif
Bram Moolenaarb8017e72007-05-10 18:59:07 +00005308 /* Reset $LC_ALL, otherwise it would overrule everything. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005309 vim_setenv((char_u *)"LC_ALL", (char_u *)"");
5310
5311 if (what != LC_TIME)
5312 {
5313 /* Tell gettext() what to translate to. It apparently doesn't
5314 * use the currently effective locale. Also do this when
5315 * FEAT_GETTEXT isn't defined, so that shell commands use this
5316 * value. */
5317 if (what == LC_ALL)
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00005318 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00005319 vim_setenv((char_u *)"LANG", name);
Bram Moolenaare3a0b532013-06-28 20:36:30 +02005320
5321 /* Clear $LANGUAGE because GNU gettext uses it. */
5322 vim_setenv((char_u *)"LANGUAGE", (char_u *)"");
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00005323# ifdef WIN32
5324 /* Apparently MS-Windows printf() may cause a crash when
5325 * we give it 8-bit text while it's expecting text in the
5326 * current locale. This call avoids that. */
5327 setlocale(LC_CTYPE, "C");
5328# endif
5329 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00005330 if (what != LC_CTYPE)
5331 {
5332 char_u *mname;
5333#ifdef WIN32
5334 mname = gettext_lang(name);
5335#else
5336 mname = name;
5337#endif
5338 vim_setenv((char_u *)"LC_MESSAGES", mname);
5339#ifdef FEAT_MULTI_LANG
5340 set_helplang_default(mname);
5341#endif
5342 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00005343 }
5344
5345# ifdef FEAT_EVAL
5346 /* Set v:lang, v:lc_time and v:ctype to the final result. */
5347 set_lang_var();
5348# endif
Bram Moolenaar6cc00c72011-10-20 21:41:09 +02005349# ifdef FEAT_TITLE
5350 maketitle();
5351# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00005352 }
5353 }
5354}
5355
5356# if defined(FEAT_CMDL_COMPL) || defined(PROTO)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005357
5358static char_u **locales = NULL; /* Array of all available locales */
Bram Moolenaarb8f7bd62017-01-12 20:28:25 +01005359
5360# ifndef WIN32
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005361static int did_init_locales = FALSE;
5362
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005363/* Return an array of strings for all available locales + NULL for the
5364 * last element. Return NULL in case of error. */
5365 static char_u **
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005366find_locales(void)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005367{
5368 garray_T locales_ga;
5369 char_u *loc;
5370
5371 /* Find all available locales by running command "locale -a". If this
5372 * doesn't work we won't have completion. */
5373 char_u *locale_a = get_cmd_output((char_u *)"locale -a",
Bram Moolenaar39c29ed2014-04-05 19:44:40 +02005374 NULL, SHELL_SILENT, NULL);
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005375 if (locale_a == NULL)
5376 return NULL;
5377 ga_init2(&locales_ga, sizeof(char_u *), 20);
5378
5379 /* Transform locale_a string where each locale is separated by "\n"
5380 * into an array of locale strings. */
5381 loc = (char_u *)strtok((char *)locale_a, "\n");
5382
5383 while (loc != NULL)
5384 {
5385 if (ga_grow(&locales_ga, 1) == FAIL)
5386 break;
5387 loc = vim_strsave(loc);
5388 if (loc == NULL)
5389 break;
5390
5391 ((char_u **)locales_ga.ga_data)[locales_ga.ga_len++] = loc;
5392 loc = (char_u *)strtok(NULL, "\n");
5393 }
5394 vim_free(locale_a);
5395 if (ga_grow(&locales_ga, 1) == FAIL)
5396 {
5397 ga_clear(&locales_ga);
5398 return NULL;
5399 }
5400 ((char_u **)locales_ga.ga_data)[locales_ga.ga_len] = NULL;
5401 return (char_u **)locales_ga.ga_data;
5402}
Bram Moolenaarb8f7bd62017-01-12 20:28:25 +01005403# endif
5404
5405/*
5406 * Lazy initialization of all available locales.
5407 */
5408 static void
5409init_locales(void)
5410{
5411# ifndef WIN32
5412 if (!did_init_locales)
5413 {
5414 did_init_locales = TRUE;
5415 locales = find_locales();
5416 }
5417# endif
5418}
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005419
5420# if defined(EXITFREE) || defined(PROTO)
5421 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005422free_locales(void)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005423{
5424 int i;
5425 if (locales != NULL)
5426 {
5427 for (i = 0; locales[i] != NULL; i++)
5428 vim_free(locales[i]);
5429 vim_free(locales);
5430 locales = NULL;
5431 }
5432}
5433# endif
5434
Bram Moolenaar071d4272004-06-13 20:20:40 +00005435/*
5436 * Function given to ExpandGeneric() to obtain the possible arguments of the
5437 * ":language" command.
5438 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005439 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005440get_lang_arg(expand_T *xp UNUSED, int idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005441{
5442 if (idx == 0)
5443 return (char_u *)"messages";
5444 if (idx == 1)
5445 return (char_u *)"ctype";
5446 if (idx == 2)
5447 return (char_u *)"time";
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005448
5449 init_locales();
5450 if (locales == NULL)
5451 return NULL;
5452 return locales[idx - 3];
5453}
5454
5455/*
5456 * Function given to ExpandGeneric() to obtain the available locales.
5457 */
5458 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005459get_locales(expand_T *xp UNUSED, int idx)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005460{
5461 init_locales();
5462 if (locales == NULL)
5463 return NULL;
5464 return locales[idx];
Bram Moolenaar071d4272004-06-13 20:20:40 +00005465}
5466# endif
5467
5468#endif