blob: 710a2789e8b4221c8074534bf3feac89657ba05f [file] [log] [blame]
Bram Moolenaaredf3f972016-08-29 22:49:24 +02001/* vi:set ts=8 sts=4 sw=4 noet:
Bram Moolenaar071d4272004-06-13 20:20:40 +00002 *
3 * VIM - Vi IMproved by Bram Moolenaar
4 *
5 * Do ":help uganda" in Vim to read copying and usage conditions.
6 * Do ":help credits" in Vim to see a list of people who contributed.
7 * See README.txt for an overview of the Vim source code.
8 */
9
10/*
11 * ex_cmds2.c: some more functions for command line commands
12 */
13
Bram Moolenaar071d4272004-06-13 20:20:40 +000014#include "vim.h"
Bram Moolenaar071d4272004-06-13 20:20:40 +000015#include "version.h"
16
Bram Moolenaarf28dbce2016-01-29 22:03:47 +010017static void cmd_source(char_u *fname, exarg_T *eap);
Bram Moolenaar071d4272004-06-13 20:20:40 +000018
Bram Moolenaar05159a02005-02-26 23:04:13 +000019#ifdef FEAT_EVAL
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +000020/* Growarray to store info about already sourced scripts.
Bram Moolenaar05159a02005-02-26 23:04:13 +000021 * For Unix also store the dev/ino, so that we don't have to stat() each
22 * script when going through the list. */
23typedef struct scriptitem_S
24{
25 char_u *sn_name;
26# ifdef UNIX
Bram Moolenaarbf0c4522009-05-16 19:16:33 +000027 int sn_dev_valid;
28 dev_t sn_dev;
Bram Moolenaar05159a02005-02-26 23:04:13 +000029 ino_t sn_ino;
30# endif
31# ifdef FEAT_PROFILE
32 int sn_prof_on; /* TRUE when script is/was profiled */
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +000033 int sn_pr_force; /* forceit: profile functions in this script */
Bram Moolenaar05159a02005-02-26 23:04:13 +000034 proftime_T sn_pr_child; /* time set when going into first child */
35 int sn_pr_nest; /* nesting for sn_pr_child */
36 /* profiling the script as a whole */
37 int sn_pr_count; /* nr of times sourced */
Bram Moolenaar8c8de832008-06-24 22:58:06 +000038 proftime_T sn_pr_total; /* time spent in script + children */
39 proftime_T sn_pr_self; /* time spent in script itself */
Bram Moolenaar05159a02005-02-26 23:04:13 +000040 proftime_T sn_pr_start; /* time at script start */
41 proftime_T sn_pr_children; /* time in children after script start */
42 /* profiling the script per line */
43 garray_T sn_prl_ga; /* things stored for every line */
44 proftime_T sn_prl_start; /* start time for current line */
45 proftime_T sn_prl_children; /* time spent in children for this line */
46 proftime_T sn_prl_wait; /* wait start time for current line */
47 int sn_prl_idx; /* index of line being timed; -1 if none */
48 int sn_prl_execed; /* line being timed was executed */
49# endif
50} scriptitem_T;
51
52static garray_T script_items = {0, 0, sizeof(scriptitem_T), 4, NULL};
53#define SCRIPT_ITEM(id) (((scriptitem_T *)script_items.ga_data)[(id) - 1])
54
55# ifdef FEAT_PROFILE
56/* Struct used in sn_prl_ga for every line of a script. */
57typedef struct sn_prl_S
58{
59 int snp_count; /* nr of times line was executed */
Bram Moolenaar8c8de832008-06-24 22:58:06 +000060 proftime_T sn_prl_total; /* time spent in a line + children */
61 proftime_T sn_prl_self; /* time spent in a line itself */
Bram Moolenaar05159a02005-02-26 23:04:13 +000062} sn_prl_T;
63
64# define PRL_ITEM(si, idx) (((sn_prl_T *)(si)->sn_prl_ga.ga_data)[(idx)])
65# endif
66#endif
67
Bram Moolenaar071d4272004-06-13 20:20:40 +000068#if defined(FEAT_EVAL) || defined(PROTO)
69static int debug_greedy = FALSE; /* batch mode debugging: don't save
70 and restore typeahead. */
Bram Moolenaarf1f60f82016-01-16 15:40:53 +010071static int get_maxbacktrace_level(void);
72static void do_setdebugtracelevel(char_u *arg);
73static void do_checkbacktracelevel(void);
74static void do_showbacktrace(char_u *cmd);
Bram Moolenaar071d4272004-06-13 20:20:40 +000075
76/*
77 * do_debug(): Debug mode.
78 * Repeatedly get Ex commands, until told to continue normal execution.
79 */
80 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +010081do_debug(char_u *cmd)
Bram Moolenaar071d4272004-06-13 20:20:40 +000082{
83 int save_msg_scroll = msg_scroll;
84 int save_State = State;
85 int save_did_emsg = did_emsg;
86 int save_cmd_silent = cmd_silent;
87 int save_msg_silent = msg_silent;
88 int save_emsg_silent = emsg_silent;
89 int save_redir_off = redir_off;
90 tasave_T typeaheadbuf;
Bram Moolenaaree3f7a52008-01-01 13:17:56 +000091 int typeahead_saved = FALSE;
Bram Moolenaar383c6f52008-01-04 15:01:07 +000092 int save_ignore_script = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +000093 int save_ex_normal_busy;
Bram Moolenaar071d4272004-06-13 20:20:40 +000094 int n;
95 char_u *cmdline = NULL;
96 char_u *p;
97 char *tail = NULL;
98 static int last_cmd = 0;
99#define CMD_CONT 1
100#define CMD_NEXT 2
101#define CMD_STEP 3
102#define CMD_FINISH 4
103#define CMD_QUIT 5
104#define CMD_INTERRUPT 6
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100105#define CMD_BACKTRACE 7
106#define CMD_FRAME 8
107#define CMD_UP 9
108#define CMD_DOWN 10
Bram Moolenaar071d4272004-06-13 20:20:40 +0000109
110#ifdef ALWAYS_USE_GUI
111 /* Can't do this when there is no terminal for input/output. */
112 if (!gui.in_use)
113 {
114 /* Break as soon as possible. */
115 debug_break_level = 9999;
116 return;
117 }
118#endif
119
120 /* Make sure we are in raw mode and start termcap mode. Might have side
121 * effects... */
122 settmode(TMODE_RAW);
123 starttermcap();
124
125 ++RedrawingDisabled; /* don't redisplay the window */
126 ++no_wait_return; /* don't wait for return */
127 did_emsg = FALSE; /* don't use error from debugged stuff */
128 cmd_silent = FALSE; /* display commands */
129 msg_silent = FALSE; /* display messages */
130 emsg_silent = FALSE; /* display error messages */
131 redir_off = TRUE; /* don't redirect debug commands */
132
133 State = NORMAL;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000134
135 if (!debug_did_msg)
136 MSG(_("Entering Debug mode. Type \"cont\" to continue."));
137 if (sourcing_name != NULL)
138 msg(sourcing_name);
139 if (sourcing_lnum != 0)
Bram Moolenaar555b2802005-05-19 21:08:39 +0000140 smsg((char_u *)_("line %ld: %s"), (long)sourcing_lnum, cmd);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000141 else
Bram Moolenaar555b2802005-05-19 21:08:39 +0000142 smsg((char_u *)_("cmd: %s"), cmd);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000143
144 /*
145 * Repeat getting a command and executing it.
146 */
147 for (;;)
148 {
149 msg_scroll = TRUE;
150 need_wait_return = FALSE;
Bram Moolenaar85b11762016-02-27 18:13:23 +0100151
Bram Moolenaar071d4272004-06-13 20:20:40 +0000152 /* Save the current typeahead buffer and replace it with an empty one.
153 * This makes sure we get input from the user here and don't interfere
154 * with the commands being executed. Reset "ex_normal_busy" to avoid
155 * the side effects of using ":normal". Save the stuff buffer and make
Bram Moolenaaree3f7a52008-01-01 13:17:56 +0000156 * it empty. Set ignore_script to avoid reading from script input. */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000157 save_ex_normal_busy = ex_normal_busy;
158 ex_normal_busy = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000159 if (!debug_greedy)
Bram Moolenaaree3f7a52008-01-01 13:17:56 +0000160 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000161 save_typeahead(&typeaheadbuf);
Bram Moolenaaree3f7a52008-01-01 13:17:56 +0000162 typeahead_saved = TRUE;
163 save_ignore_script = ignore_script;
164 ignore_script = TRUE;
165 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000166
Bram Moolenaardc303bc2016-05-17 17:45:38 +0200167 vim_free(cmdline);
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +0000168 cmdline = getcmdline_prompt('>', NULL, 0, EXPAND_NOTHING, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000169
Bram Moolenaaree3f7a52008-01-01 13:17:56 +0000170 if (typeahead_saved)
171 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000172 restore_typeahead(&typeaheadbuf);
Bram Moolenaaree3f7a52008-01-01 13:17:56 +0000173 ignore_script = save_ignore_script;
174 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000175 ex_normal_busy = save_ex_normal_busy;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000176
177 cmdline_row = msg_row;
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100178 msg_starthere();
Bram Moolenaar071d4272004-06-13 20:20:40 +0000179 if (cmdline != NULL)
180 {
181 /* If this is a debug command, set "last_cmd".
182 * If not, reset "last_cmd".
183 * For a blank line use previous command. */
184 p = skipwhite(cmdline);
185 if (*p != NUL)
186 {
187 switch (*p)
188 {
189 case 'c': last_cmd = CMD_CONT;
190 tail = "ont";
191 break;
192 case 'n': last_cmd = CMD_NEXT;
193 tail = "ext";
194 break;
195 case 's': last_cmd = CMD_STEP;
196 tail = "tep";
197 break;
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100198 case 'f':
199 last_cmd = 0;
200 if (p[1] == 'r')
201 {
202 last_cmd = CMD_FRAME;
203 tail = "rame";
204 }
205 else
206 {
207 last_cmd = CMD_FINISH;
208 tail = "inish";
209 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000210 break;
211 case 'q': last_cmd = CMD_QUIT;
212 tail = "uit";
213 break;
214 case 'i': last_cmd = CMD_INTERRUPT;
215 tail = "nterrupt";
216 break;
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100217 case 'b': last_cmd = CMD_BACKTRACE;
218 if (p[1] == 't')
219 tail = "t";
220 else
221 tail = "acktrace";
222 break;
223 case 'w': last_cmd = CMD_BACKTRACE;
224 tail = "here";
225 break;
226 case 'u': last_cmd = CMD_UP;
227 tail = "p";
228 break;
229 case 'd': last_cmd = CMD_DOWN;
230 tail = "own";
231 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000232 default: last_cmd = 0;
233 }
234 if (last_cmd != 0)
235 {
236 /* Check that the tail matches. */
237 ++p;
238 while (*p != NUL && *p == *tail)
239 {
240 ++p;
241 ++tail;
242 }
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100243 if (ASCII_ISALPHA(*p) && last_cmd != CMD_FRAME)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000244 last_cmd = 0;
245 }
246 }
247
248 if (last_cmd != 0)
249 {
250 /* Execute debug command: decided where to break next and
251 * return. */
252 switch (last_cmd)
253 {
254 case CMD_CONT:
255 debug_break_level = -1;
256 break;
257 case CMD_NEXT:
258 debug_break_level = ex_nesting_level;
259 break;
260 case CMD_STEP:
261 debug_break_level = 9999;
262 break;
263 case CMD_FINISH:
264 debug_break_level = ex_nesting_level - 1;
265 break;
266 case CMD_QUIT:
267 got_int = TRUE;
268 debug_break_level = -1;
269 break;
270 case CMD_INTERRUPT:
271 got_int = TRUE;
272 debug_break_level = 9999;
273 /* Do not repeat ">interrupt" cmd, continue stepping. */
274 last_cmd = CMD_STEP;
275 break;
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100276 case CMD_BACKTRACE:
277 do_showbacktrace(cmd);
278 continue;
279 case CMD_FRAME:
280 if (*p == NUL)
281 {
282 do_showbacktrace(cmd);
283 }
284 else
285 {
286 p = skipwhite(p);
287 do_setdebugtracelevel(p);
288 }
289 continue;
290 case CMD_UP:
291 debug_backtrace_level++;
292 do_checkbacktracelevel();
293 continue;
294 case CMD_DOWN:
295 debug_backtrace_level--;
296 do_checkbacktracelevel();
297 continue;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000298 }
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100299 /* Going out reset backtrace_level */
300 debug_backtrace_level = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000301 break;
302 }
303
304 /* don't debug this command */
305 n = debug_break_level;
306 debug_break_level = -1;
307 (void)do_cmdline(cmdline, getexline, NULL,
308 DOCMD_VERBOSE|DOCMD_EXCRESET);
309 debug_break_level = n;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000310 }
311 lines_left = Rows - 1;
312 }
313 vim_free(cmdline);
314
315 --RedrawingDisabled;
316 --no_wait_return;
317 redraw_all_later(NOT_VALID);
318 need_wait_return = FALSE;
319 msg_scroll = save_msg_scroll;
320 lines_left = Rows - 1;
321 State = save_State;
322 did_emsg = save_did_emsg;
323 cmd_silent = save_cmd_silent;
324 msg_silent = save_msg_silent;
325 emsg_silent = save_emsg_silent;
326 redir_off = save_redir_off;
327
328 /* Only print the message again when typing a command before coming back
329 * here. */
330 debug_did_msg = TRUE;
331}
332
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100333 static int
334get_maxbacktrace_level(void)
335{
336 char *p, *q;
Bram Moolenaardc633cf2016-04-23 14:33:19 +0200337 int maxbacktrace = 0;
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100338
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100339 if (sourcing_name != NULL)
340 {
341 p = (char *)sourcing_name;
342 while ((q = strstr(p, "..")) != NULL)
343 {
344 p = q + 2;
345 maxbacktrace++;
346 }
347 }
348 return maxbacktrace;
349}
350
351 static void
352do_setdebugtracelevel(char_u *arg)
353{
354 int level;
355
356 level = atoi((char *)arg);
357 if (*arg == '+' || level < 0)
358 debug_backtrace_level += level;
359 else
360 debug_backtrace_level = level;
361
362 do_checkbacktracelevel();
363}
364
365 static void
366do_checkbacktracelevel(void)
367{
368 if (debug_backtrace_level < 0)
369 {
370 debug_backtrace_level = 0;
371 MSG(_("frame is zero"));
372 }
373 else
374 {
375 int max = get_maxbacktrace_level();
376
377 if (debug_backtrace_level > max)
378 {
379 debug_backtrace_level = max;
380 smsg((char_u *)_("frame at highest level: %d"), max);
381 }
382 }
383}
384
385 static void
386do_showbacktrace(char_u *cmd)
387{
388 char *cur;
389 char *next;
390 int i = 0;
391 int max = get_maxbacktrace_level();
392
393 if (sourcing_name != NULL)
394 {
395 cur = (char *)sourcing_name;
396 while (!got_int)
397 {
398 next = strstr(cur, "..");
399 if (next != NULL)
400 *next = NUL;
401 if (i == max - debug_backtrace_level)
402 smsg((char_u *)"->%d %s", max - i, cur);
403 else
404 smsg((char_u *)" %d %s", max - i, cur);
405 ++i;
406 if (next == NULL)
407 break;
408 *next = '.';
409 cur = next + 2;
410 }
411 }
412 if (sourcing_lnum != 0)
413 smsg((char_u *)_("line %ld: %s"), (long)sourcing_lnum, cmd);
414 else
415 smsg((char_u *)_("cmd: %s"), cmd);
416}
417
Bram Moolenaar071d4272004-06-13 20:20:40 +0000418/*
419 * ":debug".
420 */
421 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100422ex_debug(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000423{
424 int debug_break_level_save = debug_break_level;
425
426 debug_break_level = 9999;
427 do_cmdline_cmd(eap->arg);
428 debug_break_level = debug_break_level_save;
429}
430
431static char_u *debug_breakpoint_name = NULL;
432static linenr_T debug_breakpoint_lnum;
433
434/*
435 * When debugging or a breakpoint is set on a skipped command, no debug prompt
436 * is shown by do_one_cmd(). This situation is indicated by debug_skipped, and
437 * debug_skipped_name is then set to the source name in the breakpoint case. If
438 * a skipped command decides itself that a debug prompt should be displayed, it
439 * can do so by calling dbg_check_skipped().
440 */
441static int debug_skipped;
442static char_u *debug_skipped_name;
443
444/*
445 * Go to debug mode when a breakpoint was encountered or "ex_nesting_level" is
446 * at or below the break level. But only when the line is actually
447 * executed. Return TRUE and set breakpoint_name for skipped commands that
448 * decide to execute something themselves.
449 * Called from do_one_cmd() before executing a command.
450 */
451 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100452dbg_check_breakpoint(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000453{
454 char_u *p;
455
456 debug_skipped = FALSE;
457 if (debug_breakpoint_name != NULL)
458 {
459 if (!eap->skip)
460 {
461 /* replace K_SNR with "<SNR>" */
462 if (debug_breakpoint_name[0] == K_SPECIAL
463 && debug_breakpoint_name[1] == KS_EXTRA
464 && debug_breakpoint_name[2] == (int)KE_SNR)
465 p = (char_u *)"<SNR>";
466 else
467 p = (char_u *)"";
Bram Moolenaar555b2802005-05-19 21:08:39 +0000468 smsg((char_u *)_("Breakpoint in \"%s%s\" line %ld"),
469 p,
Bram Moolenaar071d4272004-06-13 20:20:40 +0000470 debug_breakpoint_name + (*p == NUL ? 0 : 3),
471 (long)debug_breakpoint_lnum);
472 debug_breakpoint_name = NULL;
473 do_debug(eap->cmd);
474 }
475 else
476 {
477 debug_skipped = TRUE;
478 debug_skipped_name = debug_breakpoint_name;
479 debug_breakpoint_name = NULL;
480 }
481 }
482 else if (ex_nesting_level <= debug_break_level)
483 {
484 if (!eap->skip)
485 do_debug(eap->cmd);
486 else
487 {
488 debug_skipped = TRUE;
489 debug_skipped_name = NULL;
490 }
491 }
492}
493
494/*
495 * Go to debug mode if skipped by dbg_check_breakpoint() because eap->skip was
496 * set. Return TRUE when the debug mode is entered this time.
497 */
498 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100499dbg_check_skipped(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000500{
501 int prev_got_int;
502
503 if (debug_skipped)
504 {
505 /*
506 * Save the value of got_int and reset it. We don't want a previous
507 * interruption cause flushing the input buffer.
508 */
509 prev_got_int = got_int;
510 got_int = FALSE;
511 debug_breakpoint_name = debug_skipped_name;
512 /* eap->skip is TRUE */
513 eap->skip = FALSE;
514 (void)dbg_check_breakpoint(eap);
515 eap->skip = TRUE;
516 got_int |= prev_got_int;
517 return TRUE;
518 }
519 return FALSE;
520}
521
522/*
523 * The list of breakpoints: dbg_breakp.
524 * This is a grow-array of structs.
525 */
526struct debuggy
527{
528 int dbg_nr; /* breakpoint number */
529 int dbg_type; /* DBG_FUNC or DBG_FILE */
530 char_u *dbg_name; /* function or file name */
531 regprog_T *dbg_prog; /* regexp program */
532 linenr_T dbg_lnum; /* line number in function or file */
Bram Moolenaar05159a02005-02-26 23:04:13 +0000533 int dbg_forceit; /* ! used */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000534};
535
536static garray_T dbg_breakp = {0, 0, sizeof(struct debuggy), 4, NULL};
Bram Moolenaar05159a02005-02-26 23:04:13 +0000537#define BREAKP(idx) (((struct debuggy *)dbg_breakp.ga_data)[idx])
538#define DEBUGGY(gap, idx) (((struct debuggy *)gap->ga_data)[idx])
Bram Moolenaar071d4272004-06-13 20:20:40 +0000539static int last_breakp = 0; /* nr of last defined breakpoint */
540
Bram Moolenaar05159a02005-02-26 23:04:13 +0000541#ifdef FEAT_PROFILE
542/* Profiling uses file and func names similar to breakpoints. */
543static garray_T prof_ga = {0, 0, sizeof(struct debuggy), 4, NULL};
544#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000545#define DBG_FUNC 1
546#define DBG_FILE 2
547
Bram Moolenaarf28dbce2016-01-29 22:03:47 +0100548static int dbg_parsearg(char_u *arg, garray_T *gap);
549static linenr_T debuggy_find(int file,char_u *fname, linenr_T after, garray_T *gap, int *fp);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000550
551/*
Bram Moolenaar05159a02005-02-26 23:04:13 +0000552 * Parse the arguments of ":profile", ":breakadd" or ":breakdel" and put them
553 * in the entry just after the last one in dbg_breakp. Note that "dbg_name"
554 * is allocated.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000555 * Returns FAIL for failure.
556 */
557 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100558dbg_parsearg(
559 char_u *arg,
560 garray_T *gap) /* either &dbg_breakp or &prof_ga */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000561{
562 char_u *p = arg;
563 char_u *q;
564 struct debuggy *bp;
Bram Moolenaarf4b8e572004-06-24 15:53:16 +0000565 int here = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000566
Bram Moolenaar05159a02005-02-26 23:04:13 +0000567 if (ga_grow(gap, 1) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000568 return FAIL;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000569 bp = &DEBUGGY(gap, gap->ga_len);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000570
571 /* Find "func" or "file". */
572 if (STRNCMP(p, "func", 4) == 0)
573 bp->dbg_type = DBG_FUNC;
574 else if (STRNCMP(p, "file", 4) == 0)
575 bp->dbg_type = DBG_FILE;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000576 else if (
577#ifdef FEAT_PROFILE
578 gap != &prof_ga &&
579#endif
580 STRNCMP(p, "here", 4) == 0)
Bram Moolenaarf4b8e572004-06-24 15:53:16 +0000581 {
582 if (curbuf->b_ffname == NULL)
583 {
584 EMSG(_(e_noname));
585 return FAIL;
586 }
587 bp->dbg_type = DBG_FILE;
588 here = TRUE;
589 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000590 else
591 {
592 EMSG2(_(e_invarg2), p);
593 return FAIL;
594 }
595 p = skipwhite(p + 4);
596
597 /* Find optional line number. */
Bram Moolenaarf4b8e572004-06-24 15:53:16 +0000598 if (here)
599 bp->dbg_lnum = curwin->w_cursor.lnum;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000600 else if (
601#ifdef FEAT_PROFILE
602 gap != &prof_ga &&
603#endif
604 VIM_ISDIGIT(*p))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000605 {
606 bp->dbg_lnum = getdigits(&p);
607 p = skipwhite(p);
608 }
609 else
610 bp->dbg_lnum = 0;
611
612 /* Find the function or file name. Don't accept a function name with (). */
Bram Moolenaarf4b8e572004-06-24 15:53:16 +0000613 if ((!here && *p == NUL)
614 || (here && *p != NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000615 || (bp->dbg_type == DBG_FUNC && strstr((char *)p, "()") != NULL))
616 {
617 EMSG2(_(e_invarg2), arg);
618 return FAIL;
619 }
620
621 if (bp->dbg_type == DBG_FUNC)
622 bp->dbg_name = vim_strsave(p);
Bram Moolenaarf4b8e572004-06-24 15:53:16 +0000623 else if (here)
624 bp->dbg_name = vim_strsave(curbuf->b_ffname);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000625 else
626 {
627 /* Expand the file name in the same way as do_source(). This means
628 * doing it twice, so that $DIR/file gets expanded when $DIR is
629 * "~/dir". */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000630 q = expand_env_save(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000631 if (q == NULL)
632 return FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000633 p = expand_env_save(q);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000634 vim_free(q);
635 if (p == NULL)
636 return FAIL;
Bram Moolenaar843ee412004-06-30 16:16:41 +0000637 if (*p != '*')
638 {
639 bp->dbg_name = fix_fname(p);
640 vim_free(p);
641 }
642 else
643 bp->dbg_name = p;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000644 }
645
646 if (bp->dbg_name == NULL)
647 return FAIL;
648 return OK;
649}
650
651/*
652 * ":breakadd".
653 */
654 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100655ex_breakadd(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000656{
657 struct debuggy *bp;
658 char_u *pat;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000659 garray_T *gap;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000660
Bram Moolenaar05159a02005-02-26 23:04:13 +0000661 gap = &dbg_breakp;
662#ifdef FEAT_PROFILE
663 if (eap->cmdidx == CMD_profile)
664 gap = &prof_ga;
665#endif
666
667 if (dbg_parsearg(eap->arg, gap) == OK)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000668 {
Bram Moolenaar05159a02005-02-26 23:04:13 +0000669 bp = &DEBUGGY(gap, gap->ga_len);
670 bp->dbg_forceit = eap->forceit;
671
Bram Moolenaar071d4272004-06-13 20:20:40 +0000672 pat = file_pat_to_reg_pat(bp->dbg_name, NULL, NULL, FALSE);
673 if (pat != NULL)
674 {
675 bp->dbg_prog = vim_regcomp(pat, RE_MAGIC + RE_STRING);
676 vim_free(pat);
677 }
678 if (pat == NULL || bp->dbg_prog == NULL)
679 vim_free(bp->dbg_name);
680 else
681 {
682 if (bp->dbg_lnum == 0) /* default line number is 1 */
683 bp->dbg_lnum = 1;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000684#ifdef FEAT_PROFILE
685 if (eap->cmdidx != CMD_profile)
686#endif
687 {
688 DEBUGGY(gap, gap->ga_len).dbg_nr = ++last_breakp;
689 ++debug_tick;
690 }
691 ++gap->ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000692 }
693 }
694}
695
696/*
697 * ":debuggreedy".
698 */
699 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100700ex_debuggreedy(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000701{
702 if (eap->addr_count == 0 || eap->line2 != 0)
703 debug_greedy = TRUE;
704 else
705 debug_greedy = FALSE;
706}
707
708/*
Bram Moolenaard9fba312005-06-26 22:34:35 +0000709 * ":breakdel" and ":profdel".
Bram Moolenaar071d4272004-06-13 20:20:40 +0000710 */
711 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100712ex_breakdel(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000713{
714 struct debuggy *bp, *bpi;
715 int nr;
716 int todel = -1;
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000717 int del_all = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000718 int i;
719 linenr_T best_lnum = 0;
Bram Moolenaard9fba312005-06-26 22:34:35 +0000720 garray_T *gap;
721
722 gap = &dbg_breakp;
Bram Moolenaard9fba312005-06-26 22:34:35 +0000723 if (eap->cmdidx == CMD_profdel)
Bram Moolenaar38bdbd62012-06-20 15:48:57 +0200724 {
725#ifdef FEAT_PROFILE
Bram Moolenaard9fba312005-06-26 22:34:35 +0000726 gap = &prof_ga;
Bram Moolenaar38bdbd62012-06-20 15:48:57 +0200727#else
728 ex_ni(eap);
729 return;
Bram Moolenaard9fba312005-06-26 22:34:35 +0000730#endif
Bram Moolenaar38bdbd62012-06-20 15:48:57 +0200731 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000732
733 if (vim_isdigit(*eap->arg))
734 {
735 /* ":breakdel {nr}" */
736 nr = atol((char *)eap->arg);
Bram Moolenaard9fba312005-06-26 22:34:35 +0000737 for (i = 0; i < gap->ga_len; ++i)
738 if (DEBUGGY(gap, i).dbg_nr == nr)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000739 {
740 todel = i;
741 break;
742 }
743 }
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000744 else if (*eap->arg == '*')
745 {
746 todel = 0;
747 del_all = TRUE;
748 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000749 else
750 {
751 /* ":breakdel {func|file} [lnum] {name}" */
Bram Moolenaard9fba312005-06-26 22:34:35 +0000752 if (dbg_parsearg(eap->arg, gap) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000753 return;
Bram Moolenaard9fba312005-06-26 22:34:35 +0000754 bp = &DEBUGGY(gap, gap->ga_len);
755 for (i = 0; i < gap->ga_len; ++i)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000756 {
Bram Moolenaard9fba312005-06-26 22:34:35 +0000757 bpi = &DEBUGGY(gap, i);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000758 if (bp->dbg_type == bpi->dbg_type
759 && STRCMP(bp->dbg_name, bpi->dbg_name) == 0
760 && (bp->dbg_lnum == bpi->dbg_lnum
761 || (bp->dbg_lnum == 0
762 && (best_lnum == 0
763 || bpi->dbg_lnum < best_lnum))))
764 {
765 todel = i;
766 best_lnum = bpi->dbg_lnum;
767 }
768 }
769 vim_free(bp->dbg_name);
770 }
771
772 if (todel < 0)
773 EMSG2(_("E161: Breakpoint not found: %s"), eap->arg);
774 else
Bram Moolenaard9fba312005-06-26 22:34:35 +0000775 {
776 while (gap->ga_len > 0)
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000777 {
Bram Moolenaard9fba312005-06-26 22:34:35 +0000778 vim_free(DEBUGGY(gap, todel).dbg_name);
Bram Moolenaar473de612013-06-08 18:19:48 +0200779 vim_regfree(DEBUGGY(gap, todel).dbg_prog);
Bram Moolenaard9fba312005-06-26 22:34:35 +0000780 --gap->ga_len;
781 if (todel < gap->ga_len)
782 mch_memmove(&DEBUGGY(gap, todel), &DEBUGGY(gap, todel + 1),
783 (gap->ga_len - todel) * sizeof(struct debuggy));
784#ifdef FEAT_PROFILE
785 if (eap->cmdidx == CMD_breakdel)
786#endif
787 ++debug_tick;
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000788 if (!del_all)
789 break;
790 }
Bram Moolenaard9fba312005-06-26 22:34:35 +0000791
792 /* If all breakpoints were removed clear the array. */
793 if (gap->ga_len == 0)
794 ga_clear(gap);
795 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000796}
797
798/*
799 * ":breaklist".
800 */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000801 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100802ex_breaklist(exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000803{
804 struct debuggy *bp;
805 int i;
806
807 if (dbg_breakp.ga_len == 0)
808 MSG(_("No breakpoints defined"));
809 else
810 for (i = 0; i < dbg_breakp.ga_len; ++i)
811 {
812 bp = &BREAKP(i);
Bram Moolenaard58ea072011-06-26 04:25:30 +0200813 if (bp->dbg_type == DBG_FILE)
814 home_replace(NULL, bp->dbg_name, NameBuff, MAXPATHL, TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000815 smsg((char_u *)_("%3d %s %s line %ld"),
816 bp->dbg_nr,
817 bp->dbg_type == DBG_FUNC ? "func" : "file",
Bram Moolenaard58ea072011-06-26 04:25:30 +0200818 bp->dbg_type == DBG_FUNC ? bp->dbg_name : NameBuff,
Bram Moolenaar071d4272004-06-13 20:20:40 +0000819 (long)bp->dbg_lnum);
820 }
821}
822
823/*
824 * Find a breakpoint for a function or sourced file.
825 * Returns line number at which to break; zero when no matching breakpoint.
826 */
827 linenr_T
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100828dbg_find_breakpoint(
829 int file, /* TRUE for a file, FALSE for a function */
830 char_u *fname, /* file or function name */
831 linenr_T after) /* after this line number */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000832{
Bram Moolenaar05159a02005-02-26 23:04:13 +0000833 return debuggy_find(file, fname, after, &dbg_breakp, NULL);
834}
835
836#if defined(FEAT_PROFILE) || defined(PROTO)
837/*
838 * Return TRUE if profiling is on for a function or sourced file.
839 */
840 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100841has_profiling(
842 int file, /* TRUE for a file, FALSE for a function */
843 char_u *fname, /* file or function name */
844 int *fp) /* return: forceit */
Bram Moolenaar05159a02005-02-26 23:04:13 +0000845{
846 return (debuggy_find(file, fname, (linenr_T)0, &prof_ga, fp)
847 != (linenr_T)0);
848}
849#endif
850
851/*
852 * Common code for dbg_find_breakpoint() and has_profiling().
853 */
854 static linenr_T
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100855debuggy_find(
856 int file, /* TRUE for a file, FALSE for a function */
857 char_u *fname, /* file or function name */
858 linenr_T after, /* after this line number */
859 garray_T *gap, /* either &dbg_breakp or &prof_ga */
860 int *fp) /* if not NULL: return forceit */
Bram Moolenaar05159a02005-02-26 23:04:13 +0000861{
Bram Moolenaar071d4272004-06-13 20:20:40 +0000862 struct debuggy *bp;
863 int i;
864 linenr_T lnum = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000865 char_u *name = fname;
866 int prev_got_int;
867
Bram Moolenaar05159a02005-02-26 23:04:13 +0000868 /* Return quickly when there are no breakpoints. */
869 if (gap->ga_len == 0)
870 return (linenr_T)0;
871
Bram Moolenaar071d4272004-06-13 20:20:40 +0000872 /* Replace K_SNR in function name with "<SNR>". */
873 if (!file && fname[0] == K_SPECIAL)
874 {
875 name = alloc((unsigned)STRLEN(fname) + 3);
876 if (name == NULL)
877 name = fname;
878 else
879 {
880 STRCPY(name, "<SNR>");
881 STRCPY(name + 5, fname + 3);
882 }
883 }
884
Bram Moolenaar05159a02005-02-26 23:04:13 +0000885 for (i = 0; i < gap->ga_len; ++i)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000886 {
Bram Moolenaar05159a02005-02-26 23:04:13 +0000887 /* Skip entries that are not useful or are for a line that is beyond
888 * an already found breakpoint. */
889 bp = &DEBUGGY(gap, i);
890 if (((bp->dbg_type == DBG_FILE) == file && (
891#ifdef FEAT_PROFILE
892 gap == &prof_ga ||
893#endif
894 (bp->dbg_lnum > after && (lnum == 0 || bp->dbg_lnum < lnum)))))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000895 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000896 /*
Bram Moolenaar05159a02005-02-26 23:04:13 +0000897 * Save the value of got_int and reset it. We don't want a
898 * previous interruption cancel matching, only hitting CTRL-C
899 * while matching should abort it.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000900 */
901 prev_got_int = got_int;
902 got_int = FALSE;
Bram Moolenaardffa5b82014-11-19 16:38:07 +0100903 if (vim_regexec_prog(&bp->dbg_prog, FALSE, name, (colnr_T)0))
Bram Moolenaar05159a02005-02-26 23:04:13 +0000904 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000905 lnum = bp->dbg_lnum;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000906 if (fp != NULL)
907 *fp = bp->dbg_forceit;
908 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000909 got_int |= prev_got_int;
910 }
911 }
912 if (name != fname)
913 vim_free(name);
914
915 return lnum;
916}
917
918/*
919 * Called when a breakpoint was encountered.
920 */
921 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100922dbg_breakpoint(char_u *name, linenr_T lnum)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000923{
924 /* We need to check if this line is actually executed in do_one_cmd() */
925 debug_breakpoint_name = name;
926 debug_breakpoint_lnum = lnum;
927}
Bram Moolenaar05159a02005-02-26 23:04:13 +0000928
929
Bram Moolenaar433f7c82006-03-21 21:29:36 +0000930# if defined(FEAT_PROFILE) || defined(FEAT_RELTIME) || defined(PROTO)
Bram Moolenaar05159a02005-02-26 23:04:13 +0000931/*
932 * Store the current time in "tm".
933 */
934 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100935profile_start(proftime_T *tm)
Bram Moolenaar05159a02005-02-26 23:04:13 +0000936{
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000937# ifdef WIN3264
938 QueryPerformanceCounter(tm);
939# else
Bram Moolenaar05159a02005-02-26 23:04:13 +0000940 gettimeofday(tm, NULL);
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000941# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +0000942}
943
944/*
945 * Compute the elapsed time from "tm" till now and store in "tm".
946 */
947 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100948profile_end(proftime_T *tm)
Bram Moolenaar05159a02005-02-26 23:04:13 +0000949{
950 proftime_T now;
951
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000952# ifdef WIN3264
953 QueryPerformanceCounter(&now);
954 tm->QuadPart = now.QuadPart - tm->QuadPart;
955# else
Bram Moolenaar05159a02005-02-26 23:04:13 +0000956 gettimeofday(&now, NULL);
957 tm->tv_usec = now.tv_usec - tm->tv_usec;
958 tm->tv_sec = now.tv_sec - tm->tv_sec;
959 if (tm->tv_usec < 0)
960 {
961 tm->tv_usec += 1000000;
962 --tm->tv_sec;
963 }
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000964# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +0000965}
966
967/*
968 * Subtract the time "tm2" from "tm".
969 */
970 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100971profile_sub(proftime_T *tm, proftime_T *tm2)
Bram Moolenaar05159a02005-02-26 23:04:13 +0000972{
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000973# ifdef WIN3264
974 tm->QuadPart -= tm2->QuadPart;
975# else
Bram Moolenaar05159a02005-02-26 23:04:13 +0000976 tm->tv_usec -= tm2->tv_usec;
977 tm->tv_sec -= tm2->tv_sec;
978 if (tm->tv_usec < 0)
979 {
980 tm->tv_usec += 1000000;
981 --tm->tv_sec;
982 }
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000983# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +0000984}
985
986/*
Bram Moolenaar433f7c82006-03-21 21:29:36 +0000987 * Return a string that represents the time in "tm".
988 * Uses a static buffer!
989 */
990 char *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100991profile_msg(proftime_T *tm)
Bram Moolenaar433f7c82006-03-21 21:29:36 +0000992{
993 static char buf[50];
994
995# ifdef WIN3264
996 LARGE_INTEGER fr;
997
998 QueryPerformanceFrequency(&fr);
999 sprintf(buf, "%10.6lf", (double)tm->QuadPart / (double)fr.QuadPart);
1000# else
1001 sprintf(buf, "%3ld.%06ld", (long)tm->tv_sec, (long)tm->tv_usec);
Bram Moolenaar76929292008-01-06 19:07:36 +00001002# endif
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001003 return buf;
1004}
1005
Bram Moolenaar79c2c882016-02-07 21:19:28 +01001006# if defined(FEAT_FLOAT) || defined(PROTO)
1007/*
1008 * Return a float that represents the time in "tm".
1009 */
1010 float_T
1011profile_float(proftime_T *tm)
1012{
1013# ifdef WIN3264
1014 LARGE_INTEGER fr;
1015
1016 QueryPerformanceFrequency(&fr);
1017 return (float_T)tm->QuadPart / (float_T)fr.QuadPart;
1018# else
1019 return (float_T)tm->tv_sec + (float_T)tm->tv_usec / 1000000.0;
1020# endif
1021}
1022# endif
1023
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001024/*
Bram Moolenaar76929292008-01-06 19:07:36 +00001025 * Put the time "msec" past now in "tm".
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001026 */
Bram Moolenaar76929292008-01-06 19:07:36 +00001027 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001028profile_setlimit(long msec, proftime_T *tm)
Bram Moolenaar76929292008-01-06 19:07:36 +00001029{
1030 if (msec <= 0) /* no limit */
1031 profile_zero(tm);
1032 else
1033 {
1034# ifdef WIN3264
1035 LARGE_INTEGER fr;
1036
1037 QueryPerformanceCounter(tm);
1038 QueryPerformanceFrequency(&fr);
Bram Moolenaarb3c70982008-01-18 10:40:55 +00001039 tm->QuadPart += (LONGLONG)((double)msec / 1000.0 * (double)fr.QuadPart);
Bram Moolenaar76929292008-01-06 19:07:36 +00001040# else
1041 long usec;
1042
1043 gettimeofday(tm, NULL);
1044 usec = (long)tm->tv_usec + (long)msec * 1000;
1045 tm->tv_usec = usec % 1000000L;
1046 tm->tv_sec += usec / 1000000L;
1047# endif
1048 }
1049}
1050
1051/*
1052 * Return TRUE if the current time is past "tm".
1053 */
1054 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001055profile_passed_limit(proftime_T *tm)
Bram Moolenaar76929292008-01-06 19:07:36 +00001056{
1057 proftime_T now;
1058
1059# ifdef WIN3264
1060 if (tm->QuadPart == 0) /* timer was not set */
1061 return FALSE;
1062 QueryPerformanceCounter(&now);
1063 return (now.QuadPart > tm->QuadPart);
1064# else
1065 if (tm->tv_sec == 0) /* timer was not set */
1066 return FALSE;
1067 gettimeofday(&now, NULL);
1068 return (now.tv_sec > tm->tv_sec
1069 || (now.tv_sec == tm->tv_sec && now.tv_usec > tm->tv_usec));
1070# endif
1071}
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001072
1073/*
1074 * Set the time in "tm" to zero.
1075 */
1076 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001077profile_zero(proftime_T *tm)
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001078{
1079# ifdef WIN3264
1080 tm->QuadPart = 0;
1081# else
1082 tm->tv_usec = 0;
1083 tm->tv_sec = 0;
1084# endif
1085}
1086
Bram Moolenaar76929292008-01-06 19:07:36 +00001087# endif /* FEAT_PROFILE || FEAT_RELTIME */
1088
Bram Moolenaar975b5272016-03-15 23:10:59 +01001089# if defined(FEAT_TIMERS) || defined(PROTO)
1090static timer_T *first_timer = NULL;
Bram Moolenaar75537a92016-09-05 22:45:28 +02001091static long last_timer_id = 0;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001092
Bram Moolenaar75537a92016-09-05 22:45:28 +02001093# ifdef WIN3264
1094# define GET_TIMEDIFF(timer, now) \
1095 (long)(((double)(timer->tr_due.QuadPart - now.QuadPart) \
Bram Moolenaara7c023e2016-10-12 12:13:35 +02001096 / (double)fr.QuadPart) * 1000)
Bram Moolenaar75537a92016-09-05 22:45:28 +02001097# else
1098# define GET_TIMEDIFF(timer, now) \
1099 (timer->tr_due.tv_sec - now.tv_sec) * 1000 \
Bram Moolenaara7c023e2016-10-12 12:13:35 +02001100 + (timer->tr_due.tv_usec - now.tv_usec) / 1000
Bram Moolenaar75537a92016-09-05 22:45:28 +02001101# endif
Bram Moolenaar417ccd72016-09-01 21:26:20 +02001102
Bram Moolenaar975b5272016-03-15 23:10:59 +01001103/*
1104 * Insert a timer in the list of timers.
1105 */
1106 static void
1107insert_timer(timer_T *timer)
1108{
1109 timer->tr_next = first_timer;
1110 timer->tr_prev = NULL;
1111 if (first_timer != NULL)
1112 first_timer->tr_prev = timer;
1113 first_timer = timer;
Bram Moolenaar4231da42016-06-02 14:30:04 +02001114 did_add_timer = TRUE;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001115}
1116
1117/*
1118 * Take a timer out of the list of timers.
1119 */
1120 static void
1121remove_timer(timer_T *timer)
1122{
1123 if (timer->tr_prev == NULL)
1124 first_timer = timer->tr_next;
1125 else
1126 timer->tr_prev->tr_next = timer->tr_next;
1127 if (timer->tr_next != NULL)
1128 timer->tr_next->tr_prev = timer->tr_prev;
1129}
1130
1131 static void
1132free_timer(timer_T *timer)
1133{
Bram Moolenaar75537a92016-09-05 22:45:28 +02001134 free_callback(timer->tr_callback, timer->tr_partial);
1135 vim_free(timer);
Bram Moolenaar975b5272016-03-15 23:10:59 +01001136}
1137
1138/*
1139 * Create a timer and return it. NULL if out of memory.
1140 * Caller should set the callback.
1141 */
1142 timer_T *
1143create_timer(long msec, int repeat)
1144{
1145 timer_T *timer = (timer_T *)alloc_clear(sizeof(timer_T));
Bram Moolenaaree39ef02016-09-10 19:17:42 +02001146 long prev_id = last_timer_id;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001147
1148 if (timer == NULL)
1149 return NULL;
Bram Moolenaaree39ef02016-09-10 19:17:42 +02001150 if (++last_timer_id <= prev_id)
Bram Moolenaar75537a92016-09-05 22:45:28 +02001151 /* Overflow! Might cause duplicates... */
1152 last_timer_id = 0;
1153 timer->tr_id = last_timer_id;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001154 insert_timer(timer);
1155 if (repeat != 0)
Bram Moolenaar975b5272016-03-15 23:10:59 +01001156 timer->tr_repeat = repeat - 1;
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001157 timer->tr_interval = msec;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001158
1159 profile_setlimit(msec, &timer->tr_due);
1160 return timer;
1161}
1162
1163/*
1164 * Invoke the callback of "timer".
1165 */
1166 static void
1167timer_callback(timer_T *timer)
1168{
1169 typval_T rettv;
1170 int dummy;
1171 typval_T argv[2];
1172
1173 argv[0].v_type = VAR_NUMBER;
Bram Moolenaar75537a92016-09-05 22:45:28 +02001174 argv[0].vval.v_number = (varnumber_T)timer->tr_id;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001175 argv[1].v_type = VAR_UNKNOWN;
1176
1177 call_func(timer->tr_callback, (int)STRLEN(timer->tr_callback),
Bram Moolenaardf48fb42016-07-22 21:50:18 +02001178 &rettv, 1, argv, NULL, 0L, 0L, &dummy, TRUE,
Bram Moolenaar975b5272016-03-15 23:10:59 +01001179 timer->tr_partial, NULL);
1180 clear_tv(&rettv);
1181}
1182
1183/*
1184 * Call timers that are due.
1185 * Return the time in msec until the next timer is due.
Bram Moolenaarc4f83382017-07-07 14:50:44 +02001186 * Returns -1 if there are no pending timers.
Bram Moolenaar975b5272016-03-15 23:10:59 +01001187 */
1188 long
Bram Moolenaarcf089462016-06-12 21:18:43 +02001189check_due_timer(void)
Bram Moolenaar975b5272016-03-15 23:10:59 +01001190{
1191 timer_T *timer;
Bram Moolenaar75537a92016-09-05 22:45:28 +02001192 timer_T *timer_next;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001193 long this_due;
Bram Moolenaar597385a2016-03-16 23:24:43 +01001194 long next_due = -1;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001195 proftime_T now;
1196 int did_one = FALSE;
Bram Moolenaar02e177d2017-08-26 23:43:28 +02001197 int need_update_screen = FALSE;
Bram Moolenaar75537a92016-09-05 22:45:28 +02001198 long current_id = last_timer_id;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001199# ifdef WIN3264
1200 LARGE_INTEGER fr;
Bram Moolenaarc577d812017-07-08 22:37:34 +02001201# endif
Bram Moolenaar975b5272016-03-15 23:10:59 +01001202
Bram Moolenaarc577d812017-07-08 22:37:34 +02001203 /* Don't run any timers while exiting or dealing with an error. */
1204 if (exiting || aborting())
Bram Moolenaarc4f83382017-07-07 14:50:44 +02001205 return next_due;
1206
Bram Moolenaarc577d812017-07-08 22:37:34 +02001207# ifdef WIN3264
Bram Moolenaar975b5272016-03-15 23:10:59 +01001208 QueryPerformanceFrequency(&fr);
1209# endif
Bram Moolenaar75537a92016-09-05 22:45:28 +02001210 profile_start(&now);
1211 for (timer = first_timer; timer != NULL && !got_int; timer = timer_next)
Bram Moolenaar975b5272016-03-15 23:10:59 +01001212 {
Bram Moolenaar75537a92016-09-05 22:45:28 +02001213 timer_next = timer->tr_next;
Bram Moolenaar417ccd72016-09-01 21:26:20 +02001214
Bram Moolenaar75537a92016-09-05 22:45:28 +02001215 if (timer->tr_id == -1 || timer->tr_firing || timer->tr_paused)
1216 continue;
1217 this_due = GET_TIMEDIFF(timer, now);
1218 if (this_due <= 1)
1219 {
Bram Moolenaar1e8e1452017-06-24 16:03:06 +02001220 int save_timer_busy = timer_busy;
1221 int save_vgetc_busy = vgetc_busy;
Bram Moolenaare723c422017-09-06 23:40:10 +02001222 int save_did_emsg = did_emsg;
1223 int save_called_emsg = called_emsg;
Bram Moolenaar02e177d2017-08-26 23:43:28 +02001224 int save_must_redraw = must_redraw;
Bram Moolenaare723c422017-09-06 23:40:10 +02001225 int save_trylevel = trylevel;
1226 int save_did_throw = did_throw;
1227 except_T *save_current_exception = current_exception;
Bram Moolenaar1e8e1452017-06-24 16:03:06 +02001228
Bram Moolenaare723c422017-09-06 23:40:10 +02001229 /* Create a scope for running the timer callback, ignoring most of
1230 * the current scope, such as being inside a try/catch. */
Bram Moolenaar1e8e1452017-06-24 16:03:06 +02001231 timer_busy = timer_busy > 0 || vgetc_busy > 0;
1232 vgetc_busy = 0;
Bram Moolenaarc577d812017-07-08 22:37:34 +02001233 called_emsg = FALSE;
Bram Moolenaare723c422017-09-06 23:40:10 +02001234 did_emsg = FALSE;
1235 did_uncaught_emsg = FALSE;
Bram Moolenaar02e177d2017-08-26 23:43:28 +02001236 must_redraw = 0;
Bram Moolenaare723c422017-09-06 23:40:10 +02001237 trylevel = 0;
1238 did_throw = FALSE;
1239 current_exception = NULL;
1240
Bram Moolenaar75537a92016-09-05 22:45:28 +02001241 timer->tr_firing = TRUE;
1242 timer_callback(timer);
1243 timer->tr_firing = FALSE;
Bram Moolenaare723c422017-09-06 23:40:10 +02001244
Bram Moolenaar75537a92016-09-05 22:45:28 +02001245 timer_next = timer->tr_next;
1246 did_one = TRUE;
Bram Moolenaar1e8e1452017-06-24 16:03:06 +02001247 timer_busy = save_timer_busy;
1248 vgetc_busy = save_vgetc_busy;
Bram Moolenaare723c422017-09-06 23:40:10 +02001249 if (did_uncaught_emsg)
Bram Moolenaarc577d812017-07-08 22:37:34 +02001250 ++timer->tr_emsg_count;
Bram Moolenaare723c422017-09-06 23:40:10 +02001251 did_emsg = save_did_emsg;
1252 called_emsg = save_called_emsg;
1253 trylevel = save_trylevel;
1254 did_throw = save_did_throw;
1255 current_exception = save_current_exception;
Bram Moolenaar02e177d2017-08-26 23:43:28 +02001256 if (must_redraw != 0)
1257 need_update_screen = TRUE;
1258 must_redraw = must_redraw > save_must_redraw
1259 ? must_redraw : save_must_redraw;
Bram Moolenaar75537a92016-09-05 22:45:28 +02001260
1261 /* Only fire the timer again if it repeats and stop_timer() wasn't
1262 * called while inside the callback (tr_id == -1). */
Bram Moolenaarc577d812017-07-08 22:37:34 +02001263 if (timer->tr_repeat != 0 && timer->tr_id != -1
1264 && timer->tr_emsg_count < 3)
Bram Moolenaar75537a92016-09-05 22:45:28 +02001265 {
1266 profile_setlimit(timer->tr_interval, &timer->tr_due);
1267 this_due = GET_TIMEDIFF(timer, now);
1268 if (this_due < 1)
1269 this_due = 1;
1270 if (timer->tr_repeat > 0)
1271 --timer->tr_repeat;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001272 }
Bram Moolenaar75537a92016-09-05 22:45:28 +02001273 else
1274 {
1275 this_due = -1;
1276 remove_timer(timer);
1277 free_timer(timer);
1278 }
Bram Moolenaar975b5272016-03-15 23:10:59 +01001279 }
Bram Moolenaar75537a92016-09-05 22:45:28 +02001280 if (this_due > 0 && (next_due == -1 || next_due > this_due))
1281 next_due = this_due;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001282 }
1283
1284 if (did_one)
Bram Moolenaar02e177d2017-08-26 23:43:28 +02001285 redraw_after_callback(need_update_screen);
Bram Moolenaar975b5272016-03-15 23:10:59 +01001286
Bram Moolenaar75537a92016-09-05 22:45:28 +02001287 return current_id != last_timer_id ? 1 : next_due;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001288}
1289
1290/*
1291 * Find a timer by ID. Returns NULL if not found;
1292 */
1293 timer_T *
Bram Moolenaar75537a92016-09-05 22:45:28 +02001294find_timer(long id)
Bram Moolenaar975b5272016-03-15 23:10:59 +01001295{
1296 timer_T *timer;
1297
Bram Moolenaar75537a92016-09-05 22:45:28 +02001298 if (id >= 0)
1299 {
1300 for (timer = first_timer; timer != NULL; timer = timer->tr_next)
1301 if (timer->tr_id == id)
1302 return timer;
1303 }
1304 return NULL;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001305}
1306
1307
1308/*
1309 * Stop a timer and delete it.
1310 */
1311 void
1312stop_timer(timer_T *timer)
1313{
Bram Moolenaar75537a92016-09-05 22:45:28 +02001314 if (timer->tr_firing)
1315 /* Free the timer after the callback returns. */
1316 timer->tr_id = -1;
1317 else
1318 {
1319 remove_timer(timer);
1320 free_timer(timer);
1321 }
Bram Moolenaar975b5272016-03-15 23:10:59 +01001322}
Bram Moolenaare3188e22016-05-31 21:13:04 +02001323
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001324 void
Bram Moolenaarb73598e2016-08-07 18:22:53 +02001325stop_all_timers(void)
1326{
Bram Moolenaar75537a92016-09-05 22:45:28 +02001327 timer_T *timer;
1328 timer_T *timer_next;
1329
1330 for (timer = first_timer; timer != NULL; timer = timer_next)
1331 {
1332 timer_next = timer->tr_next;
1333 stop_timer(timer);
1334 }
Bram Moolenaarb73598e2016-08-07 18:22:53 +02001335}
1336
1337 void
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001338add_timer_info(typval_T *rettv, timer_T *timer)
1339{
1340 list_T *list = rettv->vval.v_list;
1341 dict_T *dict = dict_alloc();
1342 dictitem_T *di;
1343 long remaining;
1344 proftime_T now;
Bram Moolenaar00ff3802016-08-06 22:27:28 +02001345# ifdef WIN3264
1346 LARGE_INTEGER fr;
1347#endif
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001348
1349 if (dict == NULL)
1350 return;
1351 list_append_dict(list, dict);
1352
Bram Moolenaar75537a92016-09-05 22:45:28 +02001353 dict_add_nr_str(dict, "id", timer->tr_id, NULL);
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001354 dict_add_nr_str(dict, "time", (long)timer->tr_interval, NULL);
1355
1356 profile_start(&now);
1357# ifdef WIN3264
Bram Moolenaar00ff3802016-08-06 22:27:28 +02001358 QueryPerformanceFrequency(&fr);
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001359# endif
Bram Moolenaar75537a92016-09-05 22:45:28 +02001360 remaining = GET_TIMEDIFF(timer, now);
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001361 dict_add_nr_str(dict, "remaining", (long)remaining, NULL);
1362
1363 dict_add_nr_str(dict, "repeat",
1364 (long)(timer->tr_repeat < 0 ? -1 : timer->tr_repeat + 1), NULL);
Bram Moolenaarb73598e2016-08-07 18:22:53 +02001365 dict_add_nr_str(dict, "paused", (long)(timer->tr_paused), NULL);
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001366
1367 di = dictitem_alloc((char_u *)"callback");
1368 if (di != NULL)
1369 {
1370 if (dict_add(dict, di) == FAIL)
1371 vim_free(di);
1372 else if (timer->tr_partial != NULL)
1373 {
1374 di->di_tv.v_type = VAR_PARTIAL;
1375 di->di_tv.vval.v_partial = timer->tr_partial;
1376 ++timer->tr_partial->pt_refcount;
1377 }
1378 else
1379 {
1380 di->di_tv.v_type = VAR_FUNC;
1381 di->di_tv.vval.v_string = vim_strsave(timer->tr_callback);
1382 }
1383 di->di_tv.v_lock = 0;
1384 }
1385}
1386
1387 void
1388add_timer_info_all(typval_T *rettv)
1389{
1390 timer_T *timer;
1391
1392 for (timer = first_timer; timer != NULL; timer = timer->tr_next)
Bram Moolenaar75537a92016-09-05 22:45:28 +02001393 if (timer->tr_id != -1)
1394 add_timer_info(rettv, timer);
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001395}
1396
Bram Moolenaare3188e22016-05-31 21:13:04 +02001397/*
1398 * Mark references in partials of timers.
1399 */
1400 int
1401set_ref_in_timer(int copyID)
1402{
1403 int abort = FALSE;
1404 timer_T *timer;
1405 typval_T tv;
1406
1407 for (timer = first_timer; timer != NULL; timer = timer->tr_next)
1408 {
Bram Moolenaar1e96d9b2016-07-29 22:15:09 +02001409 if (timer->tr_partial != NULL)
1410 {
1411 tv.v_type = VAR_PARTIAL;
1412 tv.vval.v_partial = timer->tr_partial;
1413 }
1414 else
1415 {
1416 tv.v_type = VAR_FUNC;
1417 tv.vval.v_string = timer->tr_callback;
1418 }
Bram Moolenaare3188e22016-05-31 21:13:04 +02001419 abort = abort || set_ref_in_item(&tv, copyID, NULL, NULL);
1420 }
1421 return abort;
1422}
Bram Moolenaar623e2632016-07-30 22:47:56 +02001423
1424# if defined(EXITFREE) || defined(PROTO)
1425 void
1426timer_free_all()
1427{
1428 timer_T *timer;
1429
1430 while (first_timer != NULL)
1431 {
1432 timer = first_timer;
1433 remove_timer(timer);
1434 free_timer(timer);
1435 }
1436}
1437# endif
Bram Moolenaar975b5272016-03-15 23:10:59 +01001438# endif
1439
Bram Moolenaar8a7f5a22013-06-06 14:01:46 +02001440#if defined(FEAT_SYN_HL) && defined(FEAT_RELTIME) && defined(FEAT_FLOAT)
1441# if defined(HAVE_MATH_H)
1442# include <math.h>
1443# endif
1444
1445/*
1446 * Divide the time "tm" by "count" and store in "tm2".
1447 */
1448 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001449profile_divide(proftime_T *tm, int count, proftime_T *tm2)
Bram Moolenaar8a7f5a22013-06-06 14:01:46 +02001450{
1451 if (count == 0)
1452 profile_zero(tm2);
1453 else
1454 {
1455# ifdef WIN3264
1456 tm2->QuadPart = tm->QuadPart / count;
1457# else
1458 double usec = (tm->tv_sec * 1000000.0 + tm->tv_usec) / count;
1459
1460 tm2->tv_sec = floor(usec / 1000000.0);
Bram Moolenaara2e14fc2013-06-10 20:10:44 +02001461 tm2->tv_usec = vim_round(usec - (tm2->tv_sec * 1000000.0));
Bram Moolenaar8a7f5a22013-06-06 14:01:46 +02001462# endif
1463 }
1464}
1465#endif
1466
Bram Moolenaar76929292008-01-06 19:07:36 +00001467# if defined(FEAT_PROFILE) || defined(PROTO)
1468/*
1469 * Functions for profiling.
1470 */
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01001471static void script_do_profile(scriptitem_T *si);
1472static void script_dump_profile(FILE *fd);
Bram Moolenaar76929292008-01-06 19:07:36 +00001473static proftime_T prof_wait_time;
1474
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001475/*
Bram Moolenaar05159a02005-02-26 23:04:13 +00001476 * Add the time "tm2" to "tm".
1477 */
1478 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001479profile_add(proftime_T *tm, proftime_T *tm2)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001480{
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001481# ifdef WIN3264
1482 tm->QuadPart += tm2->QuadPart;
1483# else
Bram Moolenaar05159a02005-02-26 23:04:13 +00001484 tm->tv_usec += tm2->tv_usec;
1485 tm->tv_sec += tm2->tv_sec;
1486 if (tm->tv_usec >= 1000000)
1487 {
1488 tm->tv_usec -= 1000000;
1489 ++tm->tv_sec;
1490 }
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001491# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00001492}
1493
1494/*
Bram Moolenaar1056d982006-03-09 22:37:52 +00001495 * Add the "self" time from the total time and the children's time.
1496 */
1497 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001498profile_self(proftime_T *self, proftime_T *total, proftime_T *children)
Bram Moolenaar1056d982006-03-09 22:37:52 +00001499{
1500 /* Check that the result won't be negative. Can happen with recursive
1501 * calls. */
1502#ifdef WIN3264
1503 if (total->QuadPart <= children->QuadPart)
1504 return;
1505#else
1506 if (total->tv_sec < children->tv_sec
1507 || (total->tv_sec == children->tv_sec
1508 && total->tv_usec <= children->tv_usec))
1509 return;
1510#endif
1511 profile_add(self, total);
1512 profile_sub(self, children);
1513}
1514
1515/*
Bram Moolenaar05159a02005-02-26 23:04:13 +00001516 * Get the current waittime.
1517 */
1518 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001519profile_get_wait(proftime_T *tm)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001520{
1521 *tm = prof_wait_time;
1522}
1523
1524/*
1525 * Subtract the passed waittime since "tm" from "tma".
1526 */
1527 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001528profile_sub_wait(proftime_T *tm, proftime_T *tma)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001529{
1530 proftime_T tm3 = prof_wait_time;
1531
1532 profile_sub(&tm3, tm);
1533 profile_sub(tma, &tm3);
1534}
1535
1536/*
1537 * Return TRUE if "tm1" and "tm2" are equal.
1538 */
1539 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001540profile_equal(proftime_T *tm1, proftime_T *tm2)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001541{
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001542# ifdef WIN3264
1543 return (tm1->QuadPart == tm2->QuadPart);
1544# else
Bram Moolenaar05159a02005-02-26 23:04:13 +00001545 return (tm1->tv_usec == tm2->tv_usec && tm1->tv_sec == tm2->tv_sec);
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001546# endif
1547}
1548
1549/*
1550 * Return <0, 0 or >0 if "tm1" < "tm2", "tm1" == "tm2" or "tm1" > "tm2"
1551 */
1552 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001553profile_cmp(const proftime_T *tm1, const proftime_T *tm2)
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001554{
1555# ifdef WIN3264
1556 return (int)(tm2->QuadPart - tm1->QuadPart);
1557# else
1558 if (tm1->tv_sec == tm2->tv_sec)
1559 return tm2->tv_usec - tm1->tv_usec;
1560 return tm2->tv_sec - tm1->tv_sec;
1561# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00001562}
1563
Bram Moolenaar05159a02005-02-26 23:04:13 +00001564static char_u *profile_fname = NULL;
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001565static proftime_T pause_time;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001566
1567/*
1568 * ":profile cmd args"
1569 */
1570 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001571ex_profile(exarg_T *eap)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001572{
1573 char_u *e;
1574 int len;
1575
1576 e = skiptowhite(eap->arg);
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00001577 len = (int)(e - eap->arg);
Bram Moolenaar05159a02005-02-26 23:04:13 +00001578 e = skipwhite(e);
1579
1580 if (len == 5 && STRNCMP(eap->arg, "start", 5) == 0 && *e != NUL)
1581 {
1582 vim_free(profile_fname);
Bram Moolenaard94682f2015-04-13 15:37:56 +02001583 profile_fname = expand_env_save_opt(e, TRUE);
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001584 do_profiling = PROF_YES;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001585 profile_zero(&prof_wait_time);
1586 set_vim_var_nr(VV_PROFILING, 1L);
1587 }
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001588 else if (do_profiling == PROF_NONE)
Bram Moolenaar54c1b492010-02-24 14:01:28 +01001589 EMSG(_("E750: First use \":profile start {fname}\""));
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001590 else if (STRCMP(eap->arg, "pause") == 0)
1591 {
1592 if (do_profiling == PROF_YES)
1593 profile_start(&pause_time);
1594 do_profiling = PROF_PAUSED;
1595 }
1596 else if (STRCMP(eap->arg, "continue") == 0)
1597 {
1598 if (do_profiling == PROF_PAUSED)
1599 {
1600 profile_end(&pause_time);
1601 profile_add(&prof_wait_time, &pause_time);
1602 }
1603 do_profiling = PROF_YES;
1604 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00001605 else
1606 {
1607 /* The rest is similar to ":breakadd". */
1608 ex_breakadd(eap);
1609 }
1610}
1611
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001612/* Command line expansion for :profile. */
1613static enum
1614{
1615 PEXP_SUBCMD, /* expand :profile sub-commands */
Bram Moolenaar49789dc2011-02-25 14:46:09 +01001616 PEXP_FUNC /* expand :profile func {funcname} */
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001617} pexpand_what;
1618
1619static char *pexpand_cmds[] = {
1620 "start",
1621#define PROFCMD_START 0
1622 "pause",
1623#define PROFCMD_PAUSE 1
1624 "continue",
1625#define PROFCMD_CONTINUE 2
1626 "func",
1627#define PROFCMD_FUNC 3
1628 "file",
1629#define PROFCMD_FILE 4
1630 NULL
1631#define PROFCMD_LAST 5
1632};
1633
1634/*
1635 * Function given to ExpandGeneric() to obtain the profile command
1636 * specific expansion.
1637 */
1638 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001639get_profile_name(expand_T *xp UNUSED, int idx)
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001640{
1641 switch (pexpand_what)
1642 {
1643 case PEXP_SUBCMD:
1644 return (char_u *)pexpand_cmds[idx];
1645 /* case PEXP_FUNC: TODO */
1646 default:
1647 return NULL;
1648 }
1649}
1650
1651/*
1652 * Handle command line completion for :profile command.
1653 */
1654 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001655set_context_in_profile_cmd(expand_T *xp, char_u *arg)
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001656{
1657 char_u *end_subcmd;
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001658
1659 /* Default: expand subcommands. */
1660 xp->xp_context = EXPAND_PROFILE;
1661 pexpand_what = PEXP_SUBCMD;
1662 xp->xp_pattern = arg;
1663
1664 end_subcmd = skiptowhite(arg);
1665 if (*end_subcmd == NUL)
1666 return;
1667
Bram Moolenaar8b9c05f2010-03-02 17:54:33 +01001668 if (end_subcmd - arg == 5 && STRNCMP(arg, "start", 5) == 0)
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001669 {
1670 xp->xp_context = EXPAND_FILES;
1671 xp->xp_pattern = skipwhite(end_subcmd);
1672 return;
1673 }
1674
1675 /* TODO: expand function names after "func" */
1676 xp->xp_context = EXPAND_NOTHING;
1677}
1678
Bram Moolenaar05159a02005-02-26 23:04:13 +00001679/*
1680 * Dump the profiling info.
1681 */
1682 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001683profile_dump(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001684{
1685 FILE *fd;
1686
1687 if (profile_fname != NULL)
1688 {
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +00001689 fd = mch_fopen((char *)profile_fname, "w");
Bram Moolenaar05159a02005-02-26 23:04:13 +00001690 if (fd == NULL)
1691 EMSG2(_(e_notopen), profile_fname);
1692 else
1693 {
Bram Moolenaar05159a02005-02-26 23:04:13 +00001694 script_dump_profile(fd);
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001695 func_dump_profile(fd);
Bram Moolenaar05159a02005-02-26 23:04:13 +00001696 fclose(fd);
1697 }
1698 }
1699}
1700
1701/*
1702 * Start profiling script "fp".
1703 */
1704 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001705script_do_profile(scriptitem_T *si)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001706{
1707 si->sn_pr_count = 0;
1708 profile_zero(&si->sn_pr_total);
1709 profile_zero(&si->sn_pr_self);
1710
1711 ga_init2(&si->sn_prl_ga, sizeof(sn_prl_T), 100);
1712 si->sn_prl_idx = -1;
1713 si->sn_prof_on = TRUE;
1714 si->sn_pr_nest = 0;
1715}
1716
1717/*
1718 * save time when starting to invoke another script or function.
1719 */
1720 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001721script_prof_save(
1722 proftime_T *tm) /* place to store wait time */
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 profile_start(&si->sn_pr_child);
1731 }
1732 profile_get_wait(tm);
1733}
1734
1735/*
1736 * Count time spent in children after invoking another script or function.
1737 */
1738 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001739script_prof_restore(proftime_T *tm)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001740{
1741 scriptitem_T *si;
1742
1743 if (current_SID > 0 && current_SID <= script_items.ga_len)
1744 {
1745 si = &SCRIPT_ITEM(current_SID);
1746 if (si->sn_prof_on && --si->sn_pr_nest == 0)
1747 {
1748 profile_end(&si->sn_pr_child);
1749 profile_sub_wait(tm, &si->sn_pr_child); /* don't count wait time */
1750 profile_add(&si->sn_pr_children, &si->sn_pr_child);
1751 profile_add(&si->sn_prl_children, &si->sn_pr_child);
1752 }
1753 }
1754}
1755
1756static proftime_T inchar_time;
1757
1758/*
1759 * Called when starting to wait for the user to type a character.
1760 */
1761 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001762prof_inchar_enter(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001763{
1764 profile_start(&inchar_time);
1765}
1766
1767/*
1768 * Called when finished waiting for the user to type a character.
1769 */
1770 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001771prof_inchar_exit(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001772{
1773 profile_end(&inchar_time);
1774 profile_add(&prof_wait_time, &inchar_time);
1775}
1776
1777/*
1778 * Dump the profiling results for all scripts in file "fd".
1779 */
1780 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001781script_dump_profile(FILE *fd)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001782{
1783 int id;
1784 scriptitem_T *si;
1785 int i;
1786 FILE *sfd;
1787 sn_prl_T *pp;
1788
1789 for (id = 1; id <= script_items.ga_len; ++id)
1790 {
1791 si = &SCRIPT_ITEM(id);
1792 if (si->sn_prof_on)
1793 {
1794 fprintf(fd, "SCRIPT %s\n", si->sn_name);
1795 if (si->sn_pr_count == 1)
1796 fprintf(fd, "Sourced 1 time\n");
1797 else
1798 fprintf(fd, "Sourced %d times\n", si->sn_pr_count);
1799 fprintf(fd, "Total time: %s\n", profile_msg(&si->sn_pr_total));
1800 fprintf(fd, " Self time: %s\n", profile_msg(&si->sn_pr_self));
1801 fprintf(fd, "\n");
1802 fprintf(fd, "count total (s) self (s)\n");
1803
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +00001804 sfd = mch_fopen((char *)si->sn_name, "r");
Bram Moolenaar05159a02005-02-26 23:04:13 +00001805 if (sfd == NULL)
1806 fprintf(fd, "Cannot open file!\n");
1807 else
1808 {
1809 for (i = 0; i < si->sn_prl_ga.ga_len; ++i)
1810 {
1811 if (vim_fgets(IObuff, IOSIZE, sfd))
1812 break;
1813 pp = &PRL_ITEM(si, i);
1814 if (pp->snp_count > 0)
1815 {
1816 fprintf(fd, "%5d ", pp->snp_count);
1817 if (profile_equal(&pp->sn_prl_total, &pp->sn_prl_self))
1818 fprintf(fd, " ");
1819 else
1820 fprintf(fd, "%s ", profile_msg(&pp->sn_prl_total));
1821 fprintf(fd, "%s ", profile_msg(&pp->sn_prl_self));
1822 }
1823 else
1824 fprintf(fd, " ");
1825 fprintf(fd, "%s", IObuff);
1826 }
1827 fclose(sfd);
1828 }
1829 fprintf(fd, "\n");
1830 }
1831 }
1832}
1833
1834/*
1835 * Return TRUE when a function defined in the current script should be
1836 * profiled.
1837 */
1838 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001839prof_def_func(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001840{
Bram Moolenaar53180ce2005-07-05 21:48:14 +00001841 if (current_SID > 0)
1842 return SCRIPT_ITEM(current_SID).sn_pr_force;
1843 return FALSE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001844}
1845
1846# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001847#endif
1848
1849/*
1850 * If 'autowrite' option set, try to write the file.
1851 * Careful: autocommands may make "buf" invalid!
1852 *
1853 * return FAIL for failure, OK otherwise
1854 */
1855 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001856autowrite(buf_T *buf, int forceit)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001857{
Bram Moolenaar373154b2007-02-13 05:19:30 +00001858 int r;
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001859 bufref_T bufref;
Bram Moolenaar373154b2007-02-13 05:19:30 +00001860
Bram Moolenaar071d4272004-06-13 20:20:40 +00001861 if (!(p_aw || p_awa) || !p_write
1862#ifdef FEAT_QUICKFIX
Bram Moolenaar373154b2007-02-13 05:19:30 +00001863 /* never autowrite a "nofile" or "nowrite" buffer */
1864 || bt_dontwrite(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001865#endif
Bram Moolenaar373154b2007-02-13 05:19:30 +00001866 || (!forceit && buf->b_p_ro) || buf->b_ffname == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001867 return FAIL;
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001868 set_bufref(&bufref, buf);
Bram Moolenaar373154b2007-02-13 05:19:30 +00001869 r = buf_write_all(buf, forceit);
1870
1871 /* Writing may succeed but the buffer still changed, e.g., when there is a
1872 * conversion error. We do want to return FAIL then. */
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001873 if (bufref_valid(&bufref) && bufIsChanged(buf))
Bram Moolenaar373154b2007-02-13 05:19:30 +00001874 r = FAIL;
1875 return r;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001876}
1877
1878/*
1879 * flush all buffers, except the ones that are readonly
1880 */
1881 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001882autowrite_all(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001883{
1884 buf_T *buf;
1885
1886 if (!(p_aw || p_awa) || !p_write)
1887 return;
Bram Moolenaar29323592016-07-24 22:04:11 +02001888 FOR_ALL_BUFFERS(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001889 if (bufIsChanged(buf) && !buf->b_p_ro)
1890 {
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001891#ifdef FEAT_AUTOCMD
1892 bufref_T bufref;
1893
1894 set_bufref(&bufref, buf);
1895#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001896 (void)buf_write_all(buf, FALSE);
1897#ifdef FEAT_AUTOCMD
1898 /* an autocommand may have deleted the buffer */
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001899 if (!bufref_valid(&bufref))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001900 buf = firstbuf;
1901#endif
1902 }
1903}
1904
1905/*
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001906 * Return TRUE if buffer was changed and cannot be abandoned.
1907 * For flags use the CCGD_ values.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001908 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001909 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001910check_changed(buf_T *buf, int flags)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001911{
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001912 int forceit = (flags & CCGD_FORCEIT);
1913#ifdef FEAT_AUTOCMD
1914 bufref_T bufref;
1915
1916 set_bufref(&bufref, buf);
1917#endif
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001918
Bram Moolenaar071d4272004-06-13 20:20:40 +00001919 if ( !forceit
1920 && bufIsChanged(buf)
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001921 && ((flags & CCGD_MULTWIN) || buf->b_nwindows <= 1)
1922 && (!(flags & CCGD_AW) || autowrite(buf, forceit) == FAIL))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001923 {
1924#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG)
1925 if ((p_confirm || cmdmod.confirm) && p_write)
1926 {
1927 buf_T *buf2;
1928 int count = 0;
1929
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001930 if (flags & CCGD_ALLBUF)
Bram Moolenaar29323592016-07-24 22:04:11 +02001931 FOR_ALL_BUFFERS(buf2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001932 if (bufIsChanged(buf2)
1933 && (buf2->b_ffname != NULL
1934# ifdef FEAT_BROWSE
1935 || cmdmod.browse
1936# endif
1937 ))
1938 ++count;
1939# ifdef FEAT_AUTOCMD
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001940 if (!bufref_valid(&bufref))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001941 /* Autocommand deleted buffer, oops! It's not changed now. */
1942 return FALSE;
1943# endif
1944 dialog_changed(buf, count > 1);
1945# ifdef FEAT_AUTOCMD
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001946 if (!bufref_valid(&bufref))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001947 /* Autocommand deleted buffer, oops! It's not changed now. */
1948 return FALSE;
1949# endif
1950 return bufIsChanged(buf);
1951 }
1952#endif
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001953 if (flags & CCGD_EXCMD)
Bram Moolenaarf5be7cd2017-08-17 16:55:13 +02001954 no_write_message();
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001955 else
Bram Moolenaarf5be7cd2017-08-17 16:55:13 +02001956 no_write_message_nobang();
Bram Moolenaar071d4272004-06-13 20:20:40 +00001957 return TRUE;
1958 }
1959 return FALSE;
1960}
1961
1962#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG) || defined(PROTO)
1963
1964#if defined(FEAT_BROWSE) || defined(PROTO)
1965/*
1966 * When wanting to write a file without a file name, ask the user for a name.
1967 */
1968 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001969browse_save_fname(buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001970{
1971 if (buf->b_fname == NULL)
1972 {
1973 char_u *fname;
1974
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00001975 fname = do_browse(BROWSE_SAVE, (char_u *)_("Save As"),
1976 NULL, NULL, NULL, NULL, buf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001977 if (fname != NULL)
1978 {
1979 if (setfname(buf, fname, NULL, TRUE) == OK)
1980 buf->b_flags |= BF_NOTEDITED;
1981 vim_free(fname);
1982 }
1983 }
1984}
1985#endif
1986
1987/*
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02001988 * Ask the user what to do when abandoning a changed buffer.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001989 * Must check 'write' option first!
1990 */
1991 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001992dialog_changed(
1993 buf_T *buf,
1994 int checkall) /* may abandon all changed buffers */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001995{
Bram Moolenaard9462e32011-04-11 21:35:11 +02001996 char_u buff[DIALOG_MSG_SIZE];
Bram Moolenaar071d4272004-06-13 20:20:40 +00001997 int ret;
1998 buf_T *buf2;
Bram Moolenaar8218f602012-04-25 17:32:18 +02001999 exarg_T ea;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002000
Bram Moolenaar3f9a1ff2017-08-21 22:06:02 +02002001 dialog_msg(buff, _("Save changes to \"%s\"?"), buf->b_fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002002 if (checkall)
2003 ret = vim_dialog_yesnoallcancel(VIM_QUESTION, NULL, buff, 1);
2004 else
2005 ret = vim_dialog_yesnocancel(VIM_QUESTION, NULL, buff, 1);
2006
Bram Moolenaar8218f602012-04-25 17:32:18 +02002007 /* Init ea pseudo-structure, this is needed for the check_overwrite()
2008 * function. */
2009 ea.append = ea.forceit = FALSE;
2010
Bram Moolenaar071d4272004-06-13 20:20:40 +00002011 if (ret == VIM_YES)
2012 {
2013#ifdef FEAT_BROWSE
2014 /* May get file name, when there is none */
2015 browse_save_fname(buf);
2016#endif
Bram Moolenaar8218f602012-04-25 17:32:18 +02002017 if (buf->b_fname != NULL && check_overwrite(&ea, buf,
2018 buf->b_fname, buf->b_ffname, FALSE) == OK)
2019 /* didn't hit Cancel */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002020 (void)buf_write_all(buf, FALSE);
2021 }
2022 else if (ret == VIM_NO)
2023 {
2024 unchanged(buf, TRUE);
2025 }
2026 else if (ret == VIM_ALL)
2027 {
2028 /*
2029 * Write all modified files that can be written.
2030 * Skip readonly buffers, these need to be confirmed
2031 * individually.
2032 */
Bram Moolenaar29323592016-07-24 22:04:11 +02002033 FOR_ALL_BUFFERS(buf2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002034 {
2035 if (bufIsChanged(buf2)
2036 && (buf2->b_ffname != NULL
2037#ifdef FEAT_BROWSE
2038 || cmdmod.browse
2039#endif
2040 )
2041 && !buf2->b_p_ro)
2042 {
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002043#ifdef FEAT_AUTOCMD
2044 bufref_T bufref;
2045
2046 set_bufref(&bufref, buf2);
2047#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002048#ifdef FEAT_BROWSE
2049 /* May get file name, when there is none */
2050 browse_save_fname(buf2);
2051#endif
Bram Moolenaar8218f602012-04-25 17:32:18 +02002052 if (buf2->b_fname != NULL && check_overwrite(&ea, buf2,
2053 buf2->b_fname, buf2->b_ffname, FALSE) == OK)
2054 /* didn't hit Cancel */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002055 (void)buf_write_all(buf2, FALSE);
2056#ifdef FEAT_AUTOCMD
2057 /* an autocommand may have deleted the buffer */
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002058 if (!bufref_valid(&bufref))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002059 buf2 = firstbuf;
2060#endif
2061 }
2062 }
2063 }
2064 else if (ret == VIM_DISCARDALL)
2065 {
2066 /*
2067 * mark all buffers as unchanged
2068 */
Bram Moolenaar29323592016-07-24 22:04:11 +02002069 FOR_ALL_BUFFERS(buf2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002070 unchanged(buf2, TRUE);
2071 }
2072}
2073#endif
2074
2075/*
2076 * Return TRUE if the buffer "buf" can be abandoned, either by making it
2077 * hidden, autowriting it or unloading it.
2078 */
2079 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002080can_abandon(buf_T *buf, int forceit)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002081{
Bram Moolenaareb44a682017-08-03 22:44:55 +02002082 return ( buf_hide(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002083 || !bufIsChanged(buf)
2084 || buf->b_nwindows > 1
2085 || autowrite(buf, forceit) == OK
2086 || forceit);
2087}
2088
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01002089static void add_bufnum(int *bufnrs, int *bufnump, int nr);
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002090
2091/*
2092 * Add a buffer number to "bufnrs", unless it's already there.
2093 */
2094 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002095add_bufnum(int *bufnrs, int *bufnump, int nr)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002096{
2097 int i;
2098
2099 for (i = 0; i < *bufnump; ++i)
2100 if (bufnrs[i] == nr)
2101 return;
2102 bufnrs[*bufnump] = nr;
2103 *bufnump = *bufnump + 1;
2104}
2105
Bram Moolenaar071d4272004-06-13 20:20:40 +00002106/*
2107 * Return TRUE if any buffer was changed and cannot be abandoned.
2108 * That changed buffer becomes the current buffer.
Bram Moolenaar027387f2016-01-02 22:25:52 +01002109 * When "unload" is true the current buffer is unloaded instead of making it
2110 * hidden. This is used for ":q!".
Bram Moolenaar071d4272004-06-13 20:20:40 +00002111 */
2112 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002113check_changed_any(
2114 int hidden, /* Only check hidden buffers */
2115 int unload)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002116{
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002117 int ret = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002118 buf_T *buf;
2119 int save;
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002120 int i;
2121 int bufnum = 0;
2122 int bufcount = 0;
2123 int *bufnrs;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002124#ifdef FEAT_WINDOWS
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002125 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002126 win_T *wp;
2127#endif
2128
Bram Moolenaar29323592016-07-24 22:04:11 +02002129 FOR_ALL_BUFFERS(buf)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002130 ++bufcount;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002131
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002132 if (bufcount == 0)
2133 return FALSE;
2134
2135 bufnrs = (int *)alloc(sizeof(int) * bufcount);
2136 if (bufnrs == NULL)
2137 return FALSE;
2138
2139 /* curbuf */
2140 bufnrs[bufnum++] = curbuf->b_fnum;
2141#ifdef FEAT_WINDOWS
2142 /* buf in curtab */
2143 FOR_ALL_WINDOWS(wp)
2144 if (wp->w_buffer != curbuf)
2145 add_bufnum(bufnrs, &bufnum, wp->w_buffer->b_fnum);
2146
2147 /* buf in other tab */
Bram Moolenaar29323592016-07-24 22:04:11 +02002148 FOR_ALL_TABPAGES(tp)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002149 if (tp != curtab)
2150 for (wp = tp->tp_firstwin; wp != NULL; wp = wp->w_next)
2151 add_bufnum(bufnrs, &bufnum, wp->w_buffer->b_fnum);
2152#endif
2153 /* any other buf */
Bram Moolenaar29323592016-07-24 22:04:11 +02002154 FOR_ALL_BUFFERS(buf)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002155 add_bufnum(bufnrs, &bufnum, buf->b_fnum);
2156
2157 for (i = 0; i < bufnum; ++i)
2158 {
2159 buf = buflist_findnr(bufnrs[i]);
2160 if (buf == NULL)
2161 continue;
2162 if ((!hidden || buf->b_nwindows == 0) && bufIsChanged(buf))
2163 {
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002164 bufref_T bufref;
2165
2166 set_bufref(&bufref, buf);
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002167 /* Try auto-writing the buffer. If this fails but the buffer no
2168 * longer exists it's not changed, that's OK. */
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002169 if (check_changed(buf, (p_awa ? CCGD_AW : 0)
2170 | CCGD_MULTWIN
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002171 | CCGD_ALLBUF) && bufref_valid(&bufref))
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002172 break; /* didn't save - still changes */
2173 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002174 }
2175
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002176 if (i >= bufnum)
2177 goto theend;
2178
2179 ret = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002180 exiting = FALSE;
2181#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG)
2182 /*
2183 * When ":confirm" used, don't give an error message.
2184 */
2185 if (!(p_confirm || cmdmod.confirm))
2186#endif
2187 {
2188 /* There must be a wait_return for this message, do_buffer()
2189 * may cause a redraw. But wait_return() is a no-op when vgetc()
2190 * is busy (Quit used from window menu), then make sure we don't
2191 * cause a scroll up. */
Bram Moolenaar61660ea2006-04-07 21:40:07 +00002192 if (vgetc_busy > 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002193 {
2194 msg_row = cmdline_row;
2195 msg_col = 0;
2196 msg_didout = FALSE;
2197 }
Bram Moolenaareb44a682017-08-03 22:44:55 +02002198 if (
2199#ifdef FEAT_TERMINAL
2200 term_job_running(buf->b_term)
2201 ? EMSG2(_("E947: Job still running in buffer \"%s\""),
2202 buf->b_fname)
2203 :
2204#endif
2205 EMSG2(_("E162: No write since last change for buffer \"%s\""),
Bram Moolenaare1704ba2012-10-03 18:25:00 +02002206 buf_spname(buf) != NULL ? buf_spname(buf) : buf->b_fname))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002207 {
2208 save = no_wait_return;
2209 no_wait_return = FALSE;
2210 wait_return(FALSE);
2211 no_wait_return = save;
2212 }
2213 }
2214
2215#ifdef FEAT_WINDOWS
2216 /* Try to find a window that contains the buffer. */
2217 if (buf != curbuf)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002218 FOR_ALL_TAB_WINDOWS(tp, wp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002219 if (wp->w_buffer == buf)
2220 {
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002221# ifdef FEAT_AUTOCMD
2222 bufref_T bufref;
2223
2224 set_bufref(&bufref, buf);
2225# endif
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002226 goto_tabpage_win(tp, wp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002227# ifdef FEAT_AUTOCMD
2228 /* Paranoia: did autocms wipe out the buffer with changes? */
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002229 if (!bufref_valid(&bufref))
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002230 {
2231 goto theend;
2232 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002233# endif
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002234 goto buf_found;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002235 }
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002236buf_found:
Bram Moolenaar071d4272004-06-13 20:20:40 +00002237#endif
2238
2239 /* Open the changed buffer in the current window. */
2240 if (buf != curbuf)
Bram Moolenaar027387f2016-01-02 22:25:52 +01002241 set_curbuf(buf, unload ? DOBUF_UNLOAD : DOBUF_GOTO);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002242
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002243theend:
2244 vim_free(bufnrs);
2245 return ret;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002246}
2247
2248/*
2249 * return FAIL if there is no file name, OK if there is one
2250 * give error message for FAIL
2251 */
2252 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002253check_fname(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002254{
2255 if (curbuf->b_ffname == NULL)
2256 {
2257 EMSG(_(e_noname));
2258 return FAIL;
2259 }
2260 return OK;
2261}
2262
2263/*
2264 * flush the contents of a buffer, unless it has no file name
2265 *
2266 * return FAIL for failure, OK otherwise
2267 */
2268 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002269buf_write_all(buf_T *buf, int forceit)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002270{
2271 int retval;
2272#ifdef FEAT_AUTOCMD
2273 buf_T *old_curbuf = curbuf;
2274#endif
2275
2276 retval = (buf_write(buf, buf->b_ffname, buf->b_fname,
2277 (linenr_T)1, buf->b_ml.ml_line_count, NULL,
2278 FALSE, forceit, TRUE, FALSE));
2279#ifdef FEAT_AUTOCMD
2280 if (curbuf != old_curbuf)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00002281 {
Bram Moolenaar8820b482017-03-16 17:23:31 +01002282 msg_source(HL_ATTR(HLF_W));
Bram Moolenaar071d4272004-06-13 20:20:40 +00002283 MSG(_("Warning: Entered other buffer unexpectedly (check autocommands)"));
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00002284 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002285#endif
2286 return retval;
2287}
2288
2289/*
2290 * Code to handle the argument list.
2291 */
2292
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01002293static char_u *do_one_arg(char_u *str);
2294static int do_arglist(char_u *str, int what, int after);
2295static void alist_check_arg_idx(void);
2296static int editing_arg_idx(win_T *win);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002297#ifdef FEAT_LISTCMDS
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01002298static int alist_add_list(int count, char_u **files, int after);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002299#endif
2300#define AL_SET 1
2301#define AL_ADD 2
2302#define AL_DEL 3
2303
Bram Moolenaar071d4272004-06-13 20:20:40 +00002304/*
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002305 * Isolate one argument, taking backticks.
2306 * Changes the argument in-place, puts a NUL after it. Backticks remain.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002307 * Return a pointer to the start of the next argument.
2308 */
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002309 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002310do_one_arg(char_u *str)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002311{
2312 char_u *p;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002313 int inbacktick;
2314
Bram Moolenaar071d4272004-06-13 20:20:40 +00002315 inbacktick = FALSE;
2316 for (p = str; *str; ++str)
2317 {
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002318 /* When the backslash is used for escaping the special meaning of a
2319 * character we need to keep it until wildcard expansion. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002320 if (rem_backslash(str))
2321 {
2322 *p++ = *str++;
2323 *p++ = *str;
2324 }
2325 else
2326 {
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002327 /* An item ends at a space not in backticks */
2328 if (!inbacktick && vim_isspace(*str))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002329 break;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002330 if (*str == '`')
Bram Moolenaar071d4272004-06-13 20:20:40 +00002331 inbacktick ^= TRUE;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002332 *p++ = *str;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002333 }
2334 }
2335 str = skipwhite(str);
2336 *p = NUL;
2337
2338 return str;
2339}
2340
Bram Moolenaar86b68352004-12-27 21:59:20 +00002341/*
2342 * Separate the arguments in "str" and return a list of pointers in the
2343 * growarray "gap".
2344 */
Bram Moolenaar398ee732017-08-03 14:29:14 +02002345 static int
2346get_arglist(garray_T *gap, char_u *str, int escaped)
Bram Moolenaar86b68352004-12-27 21:59:20 +00002347{
2348 ga_init2(gap, (int)sizeof(char_u *), 20);
2349 while (*str != NUL)
2350 {
2351 if (ga_grow(gap, 1) == FAIL)
2352 {
2353 ga_clear(gap);
2354 return FAIL;
2355 }
2356 ((char_u **)gap->ga_data)[gap->ga_len++] = str;
2357
Bram Moolenaar398ee732017-08-03 14:29:14 +02002358 /* If str is escaped, don't handle backslashes or spaces */
2359 if (!escaped)
2360 return OK;
2361
Bram Moolenaar86b68352004-12-27 21:59:20 +00002362 /* Isolate one argument, change it in-place, put a NUL after it. */
2363 str = do_one_arg(str);
2364 }
2365 return OK;
2366}
2367
Bram Moolenaar7df351e2006-01-23 22:30:28 +00002368#if defined(FEAT_QUICKFIX) || defined(FEAT_SYN_HL) || defined(PROTO)
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002369/*
2370 * Parse a list of arguments (file names), expand them and return in
Bram Moolenaar8f5c6f02012-06-29 12:57:06 +02002371 * "fnames[fcountp]". When "wig" is TRUE, removes files matching 'wildignore'.
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002372 * Return FAIL or OK.
2373 */
2374 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002375get_arglist_exp(
2376 char_u *str,
2377 int *fcountp,
2378 char_u ***fnamesp,
2379 int wig)
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002380{
2381 garray_T ga;
2382 int i;
2383
Bram Moolenaar398ee732017-08-03 14:29:14 +02002384 if (get_arglist(&ga, str, TRUE) == FAIL)
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002385 return FAIL;
Bram Moolenaar8f5c6f02012-06-29 12:57:06 +02002386 if (wig == TRUE)
2387 i = expand_wildcards(ga.ga_len, (char_u **)ga.ga_data,
2388 fcountp, fnamesp, EW_FILE|EW_NOTFOUND);
2389 else
2390 i = gen_expand_wildcards(ga.ga_len, (char_u **)ga.ga_data,
2391 fcountp, fnamesp, EW_FILE|EW_NOTFOUND);
2392
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002393 ga_clear(&ga);
2394 return i;
2395}
2396#endif
2397
Bram Moolenaar071d4272004-06-13 20:20:40 +00002398#if defined(FEAT_GUI) || defined(FEAT_CLIENTSERVER) || defined(PROTO)
2399/*
2400 * Redefine the argument list.
2401 */
2402 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002403set_arglist(char_u *str)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002404{
2405 do_arglist(str, AL_SET, 0);
2406}
2407#endif
2408
2409/*
2410 * "what" == AL_SET: Redefine the argument list to 'str'.
2411 * "what" == AL_ADD: add files in 'str' to the argument list after "after".
2412 * "what" == AL_DEL: remove files in 'str' from the argument list.
2413 *
2414 * Return FAIL for failure, OK otherwise.
2415 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002416 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002417do_arglist(
2418 char_u *str,
Bram Moolenaarf1d25012016-03-03 12:22:53 +01002419 int what,
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002420 int after UNUSED) /* 0 means before first one */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002421{
2422 garray_T new_ga;
2423 int exp_count;
2424 char_u **exp_files;
2425 int i;
2426#ifdef FEAT_LISTCMDS
2427 char_u *p;
2428 int match;
2429#endif
Bram Moolenaar398ee732017-08-03 14:29:14 +02002430 int arg_escaped = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002431
2432 /*
Bram Moolenaar2faa29f2016-01-23 23:02:34 +01002433 * Set default argument for ":argadd" command.
2434 */
2435 if (what == AL_ADD && *str == NUL)
2436 {
2437 if (curbuf->b_ffname == NULL)
2438 return FAIL;
2439 str = curbuf->b_fname;
Bram Moolenaar398ee732017-08-03 14:29:14 +02002440 arg_escaped = FALSE;
Bram Moolenaar2faa29f2016-01-23 23:02:34 +01002441 }
2442
2443 /*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002444 * Collect all file name arguments in "new_ga".
2445 */
Bram Moolenaar398ee732017-08-03 14:29:14 +02002446 if (get_arglist(&new_ga, str, arg_escaped) == FAIL)
Bram Moolenaar86b68352004-12-27 21:59:20 +00002447 return FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002448
2449#ifdef FEAT_LISTCMDS
2450 if (what == AL_DEL)
2451 {
2452 regmatch_T regmatch;
2453 int didone;
2454
2455 /*
2456 * Delete the items: use each item as a regexp and find a match in the
2457 * argument list.
2458 */
Bram Moolenaar71afbfe2013-03-19 16:49:16 +01002459 regmatch.rm_ic = p_fic; /* ignore case when 'fileignorecase' is set */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002460 for (i = 0; i < new_ga.ga_len && !got_int; ++i)
2461 {
2462 p = ((char_u **)new_ga.ga_data)[i];
2463 p = file_pat_to_reg_pat(p, NULL, NULL, FALSE);
2464 if (p == NULL)
2465 break;
2466 regmatch.regprog = vim_regcomp(p, p_magic ? RE_MAGIC : 0);
2467 if (regmatch.regprog == NULL)
2468 {
2469 vim_free(p);
2470 break;
2471 }
2472
2473 didone = FALSE;
2474 for (match = 0; match < ARGCOUNT; ++match)
2475 if (vim_regexec(&regmatch, alist_name(&ARGLIST[match]),
2476 (colnr_T)0))
2477 {
2478 didone = TRUE;
2479 vim_free(ARGLIST[match].ae_fname);
2480 mch_memmove(ARGLIST + match, ARGLIST + match + 1,
2481 (ARGCOUNT - match - 1) * sizeof(aentry_T));
2482 --ALIST(curwin)->al_ga.ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002483 if (curwin->w_arg_idx > match)
2484 --curwin->w_arg_idx;
2485 --match;
2486 }
2487
Bram Moolenaar473de612013-06-08 18:19:48 +02002488 vim_regfree(regmatch.regprog);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002489 vim_free(p);
2490 if (!didone)
2491 EMSG2(_(e_nomatch2), ((char_u **)new_ga.ga_data)[i]);
2492 }
2493 ga_clear(&new_ga);
2494 }
2495 else
2496#endif
2497 {
2498 i = expand_wildcards(new_ga.ga_len, (char_u **)new_ga.ga_data,
2499 &exp_count, &exp_files, EW_DIR|EW_FILE|EW_ADDSLASH|EW_NOTFOUND);
2500 ga_clear(&new_ga);
Bram Moolenaar2db5c3b2016-01-16 22:49:34 +01002501 if (i == FAIL || exp_count == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002502 {
2503 EMSG(_(e_nomatch));
2504 return FAIL;
2505 }
2506
2507#ifdef FEAT_LISTCMDS
2508 if (what == AL_ADD)
2509 {
2510 (void)alist_add_list(exp_count, exp_files, after);
2511 vim_free(exp_files);
2512 }
2513 else /* what == AL_SET */
2514#endif
Bram Moolenaar86b68352004-12-27 21:59:20 +00002515 alist_set(ALIST(curwin), exp_count, exp_files, FALSE, NULL, 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002516 }
2517
2518 alist_check_arg_idx();
2519
2520 return OK;
2521}
2522
2523/*
2524 * Check the validity of the arg_idx for each other window.
2525 */
2526 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002527alist_check_arg_idx(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002528{
2529#ifdef FEAT_WINDOWS
2530 win_T *win;
Bram Moolenaarf740b292006-02-16 22:11:02 +00002531 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002532
Bram Moolenaarf740b292006-02-16 22:11:02 +00002533 FOR_ALL_TAB_WINDOWS(tp, win)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002534 if (win->w_alist == curwin->w_alist)
2535 check_arg_idx(win);
2536#else
2537 check_arg_idx(curwin);
2538#endif
2539}
2540
2541/*
Bram Moolenaarb8ff1fb2012-02-04 21:59:01 +01002542 * Return TRUE if window "win" is editing the file at the current argument
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002543 * index.
2544 */
2545 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002546editing_arg_idx(win_T *win)
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002547{
2548 return !(win->w_arg_idx >= WARGCOUNT(win)
2549 || (win->w_buffer->b_fnum
2550 != WARGLIST(win)[win->w_arg_idx].ae_fnum
2551 && (win->w_buffer->b_ffname == NULL
2552 || !(fullpathcmp(
2553 alist_name(&WARGLIST(win)[win->w_arg_idx]),
2554 win->w_buffer->b_ffname, TRUE) & FPC_SAME))));
2555}
2556
2557/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002558 * Check if window "win" is editing the w_arg_idx file in its argument list.
2559 */
2560 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002561check_arg_idx(win_T *win)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002562{
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002563 if (WARGCOUNT(win) > 1 && !editing_arg_idx(win))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002564 {
2565 /* We are not editing the current entry in the argument list.
2566 * Set "arg_had_last" if we are editing the last one. */
2567 win->w_arg_idx_invalid = TRUE;
2568 if (win->w_arg_idx != WARGCOUNT(win) - 1
2569 && arg_had_last == FALSE
2570#ifdef FEAT_WINDOWS
2571 && ALIST(win) == &global_alist
2572#endif
2573 && GARGCOUNT > 0
2574 && win->w_arg_idx < GARGCOUNT
2575 && (win->w_buffer->b_fnum == GARGLIST[GARGCOUNT - 1].ae_fnum
2576 || (win->w_buffer->b_ffname != NULL
2577 && (fullpathcmp(alist_name(&GARGLIST[GARGCOUNT - 1]),
2578 win->w_buffer->b_ffname, TRUE) & FPC_SAME))))
2579 arg_had_last = TRUE;
2580 }
2581 else
2582 {
2583 /* We are editing the current entry in the argument list.
2584 * Set "arg_had_last" if it's also the last one */
2585 win->w_arg_idx_invalid = FALSE;
2586 if (win->w_arg_idx == WARGCOUNT(win) - 1
2587#ifdef FEAT_WINDOWS
2588 && win->w_alist == &global_alist
2589#endif
2590 )
2591 arg_had_last = TRUE;
2592 }
2593}
2594
2595/*
2596 * ":args", ":argslocal" and ":argsglobal".
2597 */
2598 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002599ex_args(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002600{
2601 int i;
2602
2603 if (eap->cmdidx != CMD_args)
2604 {
2605#if defined(FEAT_WINDOWS) && defined(FEAT_LISTCMDS)
2606 alist_unlink(ALIST(curwin));
2607 if (eap->cmdidx == CMD_argglobal)
2608 ALIST(curwin) = &global_alist;
2609 else /* eap->cmdidx == CMD_arglocal */
2610 alist_new();
2611#else
2612 ex_ni(eap);
2613 return;
2614#endif
2615 }
2616
2617 if (!ends_excmd(*eap->arg))
2618 {
2619 /*
2620 * ":args file ..": define new argument list, handle like ":next"
2621 * Also for ":argslocal file .." and ":argsglobal file ..".
2622 */
2623 ex_next(eap);
2624 }
2625 else
2626#if defined(FEAT_WINDOWS) && defined(FEAT_LISTCMDS)
2627 if (eap->cmdidx == CMD_args)
2628#endif
2629 {
2630 /*
2631 * ":args": list arguments.
2632 */
2633 if (ARGCOUNT > 0)
2634 {
2635 /* Overwrite the command, for a short list there is no scrolling
2636 * required and no wait_return(). */
2637 gotocmdline(TRUE);
2638 for (i = 0; i < ARGCOUNT; ++i)
2639 {
2640 if (i == curwin->w_arg_idx)
2641 msg_putchar('[');
2642 msg_outtrans(alist_name(&ARGLIST[i]));
2643 if (i == curwin->w_arg_idx)
2644 msg_putchar(']');
2645 msg_putchar(' ');
2646 }
2647 }
2648 }
2649#if defined(FEAT_WINDOWS) && defined(FEAT_LISTCMDS)
2650 else if (eap->cmdidx == CMD_arglocal)
2651 {
2652 garray_T *gap = &curwin->w_alist->al_ga;
2653
2654 /*
2655 * ":argslocal": make a local copy of the global argument list.
2656 */
2657 if (ga_grow(gap, GARGCOUNT) == OK)
2658 for (i = 0; i < GARGCOUNT; ++i)
2659 if (GARGLIST[i].ae_fname != NULL)
2660 {
2661 AARGLIST(curwin->w_alist)[gap->ga_len].ae_fname =
2662 vim_strsave(GARGLIST[i].ae_fname);
2663 AARGLIST(curwin->w_alist)[gap->ga_len].ae_fnum =
2664 GARGLIST[i].ae_fnum;
2665 ++gap->ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002666 }
2667 }
2668#endif
2669}
2670
2671/*
2672 * ":previous", ":sprevious", ":Next" and ":sNext".
2673 */
2674 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002675ex_previous(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002676{
2677 /* If past the last one already, go to the last one. */
2678 if (curwin->w_arg_idx - (int)eap->line2 >= ARGCOUNT)
2679 do_argfile(eap, ARGCOUNT - 1);
2680 else
2681 do_argfile(eap, curwin->w_arg_idx - (int)eap->line2);
2682}
2683
2684/*
2685 * ":rewind", ":first", ":sfirst" and ":srewind".
2686 */
2687 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002688ex_rewind(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002689{
2690 do_argfile(eap, 0);
2691}
2692
2693/*
2694 * ":last" and ":slast".
2695 */
2696 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002697ex_last(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002698{
2699 do_argfile(eap, ARGCOUNT - 1);
2700}
2701
2702/*
2703 * ":argument" and ":sargument".
2704 */
2705 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002706ex_argument(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002707{
2708 int i;
2709
2710 if (eap->addr_count > 0)
2711 i = eap->line2 - 1;
2712 else
2713 i = curwin->w_arg_idx;
2714 do_argfile(eap, i);
2715}
2716
2717/*
2718 * Edit file "argn" of the argument lists.
2719 */
2720 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002721do_argfile(exarg_T *eap, int argn)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002722{
2723 int other;
2724 char_u *p;
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002725 int old_arg_idx = curwin->w_arg_idx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002726
2727 if (argn < 0 || argn >= ARGCOUNT)
2728 {
2729 if (ARGCOUNT <= 1)
2730 EMSG(_("E163: There is only one file to edit"));
2731 else if (argn < 0)
2732 EMSG(_("E164: Cannot go before first file"));
2733 else
2734 EMSG(_("E165: Cannot go beyond last file"));
2735 }
2736 else
2737 {
2738 setpcmark();
2739#ifdef FEAT_GUI
2740 need_mouse_correct = TRUE;
2741#endif
2742
2743#ifdef FEAT_WINDOWS
Bram Moolenaardf1bdc92006-02-23 21:32:16 +00002744 /* split window or create new tab page first */
2745 if (*eap->cmd == 's' || cmdmod.tab != 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002746 {
2747 if (win_split(0, 0) == FAIL)
2748 return;
Bram Moolenaar3368ea22010-09-21 16:56:35 +02002749 RESET_BINDING(curwin);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002750 }
2751 else
2752#endif
2753 {
2754 /*
2755 * if 'hidden' set, only check for changed file when re-editing
2756 * the same buffer
2757 */
2758 other = TRUE;
Bram Moolenaareb44a682017-08-03 22:44:55 +02002759 if (buf_hide(curbuf))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002760 {
2761 p = fix_fname(alist_name(&ARGLIST[argn]));
2762 other = otherfile(p);
2763 vim_free(p);
2764 }
Bram Moolenaareb44a682017-08-03 22:44:55 +02002765 if ((!buf_hide(curbuf) || !other)
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002766 && check_changed(curbuf, CCGD_AW
2767 | (other ? 0 : CCGD_MULTWIN)
2768 | (eap->forceit ? CCGD_FORCEIT : 0)
2769 | CCGD_EXCMD))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002770 return;
2771 }
2772
2773 curwin->w_arg_idx = argn;
2774 if (argn == ARGCOUNT - 1
2775#ifdef FEAT_WINDOWS
2776 && curwin->w_alist == &global_alist
2777#endif
2778 )
2779 arg_had_last = TRUE;
2780
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002781 /* Edit the file; always use the last known line number.
2782 * When it fails (e.g. Abort for already edited file) restore the
2783 * argument index. */
2784 if (do_ecmd(0, alist_name(&ARGLIST[curwin->w_arg_idx]), NULL,
Bram Moolenaar071d4272004-06-13 20:20:40 +00002785 eap, ECMD_LAST,
Bram Moolenaareb44a682017-08-03 22:44:55 +02002786 (buf_hide(curwin->w_buffer) ? ECMD_HIDE : 0)
Bram Moolenaar701f7af2008-11-15 13:12:07 +00002787 + (eap->forceit ? ECMD_FORCEIT : 0), curwin) == FAIL)
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002788 curwin->w_arg_idx = old_arg_idx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002789 /* like Vi: set the mark where the cursor is in the file. */
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002790 else if (eap->cmdidx != CMD_argdo)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002791 setmark('\'');
2792 }
2793}
2794
2795/*
2796 * ":next", and commands that behave like it.
2797 */
2798 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002799ex_next(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002800{
2801 int i;
2802
2803 /*
2804 * check for changed buffer now, if this fails the argument list is not
2805 * redefined.
2806 */
Bram Moolenaareb44a682017-08-03 22:44:55 +02002807 if ( buf_hide(curbuf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002808 || eap->cmdidx == CMD_snext
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002809 || !check_changed(curbuf, CCGD_AW
2810 | (eap->forceit ? CCGD_FORCEIT : 0)
2811 | CCGD_EXCMD))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002812 {
2813 if (*eap->arg != NUL) /* redefine file list */
2814 {
2815 if (do_arglist(eap->arg, AL_SET, 0) == FAIL)
2816 return;
2817 i = 0;
2818 }
2819 else
2820 i = curwin->w_arg_idx + (int)eap->line2;
2821 do_argfile(eap, i);
2822 }
2823}
2824
Bram Moolenaar0106e3d2016-02-23 18:55:43 +01002825#if defined(FEAT_LISTCMDS) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002826/*
2827 * ":argedit"
2828 */
2829 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002830ex_argedit(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002831{
Bram Moolenaar90305c62017-07-16 15:31:17 +02002832 int i = eap->addr_count ? (int)eap->line2 : curwin->w_arg_idx + 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002833
Bram Moolenaar90305c62017-07-16 15:31:17 +02002834 if (do_arglist(eap->arg, AL_ADD, i) == FAIL)
2835 return;
2836#ifdef FEAT_TITLE
2837 maketitle();
2838#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002839
Bram Moolenaar90305c62017-07-16 15:31:17 +02002840 if (curwin->w_arg_idx == 0 && (curbuf->b_ml.ml_flags & ML_EMPTY)
2841 && curbuf->b_ffname == NULL)
2842 i = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002843 /* Edit the argument. */
Bram Moolenaar90305c62017-07-16 15:31:17 +02002844 if (i < ARGCOUNT)
2845 do_argfile(eap, i);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002846}
2847
2848/*
2849 * ":argadd"
2850 */
2851 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002852ex_argadd(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002853{
2854 do_arglist(eap->arg, AL_ADD,
2855 eap->addr_count > 0 ? (int)eap->line2 : curwin->w_arg_idx + 1);
2856#ifdef FEAT_TITLE
2857 maketitle();
2858#endif
2859}
2860
2861/*
2862 * ":argdelete"
2863 */
2864 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002865ex_argdelete(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002866{
2867 int i;
2868 int n;
2869
2870 if (eap->addr_count > 0)
2871 {
2872 /* ":1,4argdel": Delete all arguments in the range. */
2873 if (eap->line2 > ARGCOUNT)
2874 eap->line2 = ARGCOUNT;
2875 n = eap->line2 - eap->line1 + 1;
Bram Moolenaar69a92fb2017-03-09 15:58:30 +01002876 if (*eap->arg != NUL)
2877 /* Can't have both a range and an argument. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002878 EMSG(_(e_invarg));
Bram Moolenaar69a92fb2017-03-09 15:58:30 +01002879 else if (n <= 0)
2880 {
2881 /* Don't give an error for ":%argdel" if the list is empty. */
2882 if (eap->line1 != 1 || eap->line2 != 0)
2883 EMSG(_(e_invrange));
2884 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002885 else
2886 {
2887 for (i = eap->line1; i <= eap->line2; ++i)
2888 vim_free(ARGLIST[i - 1].ae_fname);
2889 mch_memmove(ARGLIST + eap->line1 - 1, ARGLIST + eap->line2,
2890 (size_t)((ARGCOUNT - eap->line2) * sizeof(aentry_T)));
2891 ALIST(curwin)->al_ga.ga_len -= n;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002892 if (curwin->w_arg_idx >= eap->line2)
2893 curwin->w_arg_idx -= n;
2894 else if (curwin->w_arg_idx > eap->line1)
2895 curwin->w_arg_idx = eap->line1;
Bram Moolenaar72defda2016-01-17 18:04:33 +01002896 if (ARGCOUNT == 0)
2897 curwin->w_arg_idx = 0;
2898 else if (curwin->w_arg_idx >= ARGCOUNT)
2899 curwin->w_arg_idx = ARGCOUNT - 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002900 }
2901 }
2902 else if (*eap->arg == NUL)
2903 EMSG(_(e_argreq));
2904 else
2905 do_arglist(eap->arg, AL_DEL, 0);
2906#ifdef FEAT_TITLE
2907 maketitle();
2908#endif
2909}
2910
2911/*
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002912 * ":argdo", ":windo", ":bufdo", ":tabdo", ":cdo", ":ldo", ":cfdo" and ":lfdo"
Bram Moolenaar071d4272004-06-13 20:20:40 +00002913 */
2914 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002915ex_listdo(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002916{
2917 int i;
2918#ifdef FEAT_WINDOWS
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002919 win_T *wp;
2920 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002921#endif
Bram Moolenaare25bb902015-02-27 20:33:37 +01002922 buf_T *buf = curbuf;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002923 int next_fnum = 0;
2924#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
2925 char_u *save_ei = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002926#endif
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002927 char_u *p_shm_save;
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002928#ifdef FEAT_QUICKFIX
Bram Moolenaared84b762015-09-09 22:35:29 +02002929 int qf_size = 0;
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002930 int qf_idx;
2931#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002932
2933#ifndef FEAT_WINDOWS
2934 if (eap->cmdidx == CMD_windo)
2935 {
2936 ex_ni(eap);
2937 return;
2938 }
2939#endif
2940
Bram Moolenaar0106e3d2016-02-23 18:55:43 +01002941#ifndef FEAT_QUICKFIX
2942 if (eap->cmdidx == CMD_cdo || eap->cmdidx == CMD_ldo ||
2943 eap->cmdidx == CMD_cfdo || eap->cmdidx == CMD_lfdo)
2944 {
2945 ex_ni(eap);
2946 return;
2947 }
2948#endif
2949
Bram Moolenaar071d4272004-06-13 20:20:40 +00002950#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002951 if (eap->cmdidx != CMD_windo && eap->cmdidx != CMD_tabdo)
Bram Moolenaardcaf10e2005-01-21 11:55:25 +00002952 /* Don't do syntax HL autocommands. Skipping the syntax file is a
2953 * great speed improvement. */
2954 save_ei = au_event_disable(",Syntax");
Bram Moolenaar071d4272004-06-13 20:20:40 +00002955#endif
Bram Moolenaar6b1ee342014-08-06 18:17:11 +02002956#ifdef FEAT_CLIPBOARD
2957 start_global_changes();
2958#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002959
2960 if (eap->cmdidx == CMD_windo
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002961 || eap->cmdidx == CMD_tabdo
Bram Moolenaareb44a682017-08-03 22:44:55 +02002962 || buf_hide(curbuf)
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002963 || !check_changed(curbuf, CCGD_AW
2964 | (eap->forceit ? CCGD_FORCEIT : 0)
2965 | CCGD_EXCMD))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002966 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00002967 i = 0;
Bram Moolenaara162bc52015-01-07 16:54:21 +01002968 /* start at the eap->line1 argument/window/buffer */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002969#ifdef FEAT_WINDOWS
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002970 wp = firstwin;
2971 tp = first_tabpage;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002972#endif
Bram Moolenaara162bc52015-01-07 16:54:21 +01002973 switch (eap->cmdidx)
2974 {
2975#ifdef FEAT_WINDOWS
2976 case CMD_windo:
2977 for ( ; wp != NULL && i + 1 < eap->line1; wp = wp->w_next)
2978 i++;
2979 break;
2980 case CMD_tabdo:
2981 for( ; tp != NULL && i + 1 < eap->line1; tp = tp->tp_next)
2982 i++;
2983 break;
2984#endif
2985 case CMD_argdo:
2986 i = eap->line1 - 1;
2987 break;
Bram Moolenaara162bc52015-01-07 16:54:21 +01002988 default:
2989 break;
2990 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002991 /* set pcmark now */
2992 if (eap->cmdidx == CMD_bufdo)
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002993 {
Bram Moolenaare25bb902015-02-27 20:33:37 +01002994 /* Advance to the first listed buffer after "eap->line1". */
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002995 for (buf = firstbuf; buf != NULL && (buf->b_fnum < eap->line1
Bram Moolenaare25bb902015-02-27 20:33:37 +01002996 || !buf->b_p_bl); buf = buf->b_next)
2997 if (buf->b_fnum > eap->line2)
2998 {
2999 buf = NULL;
3000 break;
3001 }
Bram Moolenaaraa23b372015-09-08 18:46:31 +02003002 if (buf != NULL)
Bram Moolenaare25bb902015-02-27 20:33:37 +01003003 goto_buffer(eap, DOBUF_FIRST, FORWARD, buf->b_fnum);
Bram Moolenaaraa23b372015-09-08 18:46:31 +02003004 }
3005#ifdef FEAT_QUICKFIX
3006 else if (eap->cmdidx == CMD_cdo || eap->cmdidx == CMD_ldo
3007 || eap->cmdidx == CMD_cfdo || eap->cmdidx == CMD_lfdo)
3008 {
3009 qf_size = qf_get_size(eap);
3010 if (qf_size <= 0 || eap->line1 > qf_size)
3011 buf = NULL;
3012 else
3013 {
3014 ex_cc(eap);
3015
3016 buf = curbuf;
3017 i = eap->line1 - 1;
3018 if (eap->addr_count <= 0)
3019 /* default is all the quickfix/location list entries */
3020 eap->line2 = qf_size;
3021 }
3022 }
3023#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003024 else
3025 setpcmark();
3026 listcmd_busy = TRUE; /* avoids setting pcmark below */
3027
Bram Moolenaare25bb902015-02-27 20:33:37 +01003028 while (!got_int && buf != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003029 {
3030 if (eap->cmdidx == CMD_argdo)
3031 {
3032 /* go to argument "i" */
3033 if (i == ARGCOUNT)
3034 break;
3035 /* Don't call do_argfile() when already there, it will try
3036 * reloading the file. */
Bram Moolenaard4755bb2004-09-02 19:12:26 +00003037 if (curwin->w_arg_idx != i || !editing_arg_idx(curwin))
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003038 {
3039 /* Clear 'shm' to avoid that the file message overwrites
3040 * any output from the command. */
3041 p_shm_save = vim_strsave(p_shm);
3042 set_option_value((char_u *)"shm", 0L, (char_u *)"", 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003043 do_argfile(eap, i);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003044 set_option_value((char_u *)"shm", 0L, p_shm_save, 0);
3045 vim_free(p_shm_save);
3046 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003047 if (curwin->w_arg_idx != i)
3048 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003049 }
3050#ifdef FEAT_WINDOWS
3051 else if (eap->cmdidx == CMD_windo)
3052 {
Bram Moolenaar32466aa2006-02-24 23:53:04 +00003053 /* go to window "wp" */
3054 if (!win_valid(wp))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003055 break;
Bram Moolenaar32466aa2006-02-24 23:53:04 +00003056 win_goto(wp);
Bram Moolenaar41423242007-05-03 20:11:13 +00003057 if (curwin != wp)
3058 break; /* something must be wrong */
Bram Moolenaar32466aa2006-02-24 23:53:04 +00003059 wp = curwin->w_next;
3060 }
3061 else if (eap->cmdidx == CMD_tabdo)
3062 {
3063 /* go to window "tp" */
3064 if (!valid_tabpage(tp))
3065 break;
Bram Moolenaar49e649f2013-05-06 04:50:35 +02003066 goto_tabpage_tp(tp, TRUE, TRUE);
Bram Moolenaar32466aa2006-02-24 23:53:04 +00003067 tp = tp->tp_next;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003068 }
3069#endif
3070 else if (eap->cmdidx == CMD_bufdo)
3071 {
3072 /* Remember the number of the next listed buffer, in case
3073 * ":bwipe" is used or autocommands do something strange. */
3074 next_fnum = -1;
3075 for (buf = curbuf->b_next; buf != NULL; buf = buf->b_next)
3076 if (buf->b_p_bl)
3077 {
3078 next_fnum = buf->b_fnum;
3079 break;
3080 }
3081 }
3082
Bram Moolenaara162bc52015-01-07 16:54:21 +01003083 ++i;
3084
Bram Moolenaar071d4272004-06-13 20:20:40 +00003085 /* execute the command */
3086 do_cmdline(eap->arg, eap->getline, eap->cookie,
3087 DOCMD_VERBOSE + DOCMD_NOWAIT);
3088
3089 if (eap->cmdidx == CMD_bufdo)
3090 {
3091 /* Done? */
Bram Moolenaara162bc52015-01-07 16:54:21 +01003092 if (next_fnum < 0 || next_fnum > eap->line2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003093 break;
3094 /* Check if the buffer still exists. */
Bram Moolenaar29323592016-07-24 22:04:11 +02003095 FOR_ALL_BUFFERS(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003096 if (buf->b_fnum == next_fnum)
3097 break;
3098 if (buf == NULL)
3099 break;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003100
3101 /* Go to the next buffer. Clear 'shm' to avoid that the file
3102 * message overwrites any output from the command. */
3103 p_shm_save = vim_strsave(p_shm);
3104 set_option_value((char_u *)"shm", 0L, (char_u *)"", 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003105 goto_buffer(eap, DOBUF_FIRST, FORWARD, next_fnum);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003106 set_option_value((char_u *)"shm", 0L, p_shm_save, 0);
3107 vim_free(p_shm_save);
3108
Bram Moolenaaraa23b372015-09-08 18:46:31 +02003109 /* If autocommands took us elsewhere, quit here. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003110 if (curbuf->b_fnum != next_fnum)
3111 break;
3112 }
3113
Bram Moolenaaraa23b372015-09-08 18:46:31 +02003114#ifdef FEAT_QUICKFIX
3115 if (eap->cmdidx == CMD_cdo || eap->cmdidx == CMD_ldo
3116 || eap->cmdidx == CMD_cfdo || eap->cmdidx == CMD_lfdo)
3117 {
3118 if (i >= qf_size || i >= eap->line2)
3119 break;
3120
3121 qf_idx = qf_get_cur_idx(eap);
3122
3123 ex_cnext(eap);
3124
3125 /* If jumping to the next quickfix entry fails, quit here */
3126 if (qf_get_cur_idx(eap) == qf_idx)
3127 break;
3128 }
3129#endif
3130
Bram Moolenaar071d4272004-06-13 20:20:40 +00003131 if (eap->cmdidx == CMD_windo)
3132 {
3133 validate_cursor(); /* cursor may have moved */
3134#ifdef FEAT_SCROLLBIND
3135 /* required when 'scrollbind' has been set */
3136 if (curwin->w_p_scb)
3137 do_check_scrollbind(TRUE);
3138#endif
3139 }
Bram Moolenaara162bc52015-01-07 16:54:21 +01003140
3141#ifdef FEAT_WINDOWS
3142 if (eap->cmdidx == CMD_windo || eap->cmdidx == CMD_tabdo)
3143 if (i+1 > eap->line2)
3144 break;
3145#endif
3146 if (eap->cmdidx == CMD_argdo && i >= eap->line2)
3147 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003148 }
3149 listcmd_busy = FALSE;
3150 }
3151
3152#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
Bram Moolenaar6ac54292005-02-02 23:07:25 +00003153 if (save_ei != NULL)
3154 {
3155 au_event_restore(save_ei);
3156 apply_autocmds(EVENT_SYNTAX, curbuf->b_p_syn,
3157 curbuf->b_fname, TRUE, curbuf);
3158 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003159#endif
Bram Moolenaar6b1ee342014-08-06 18:17:11 +02003160#ifdef FEAT_CLIPBOARD
3161 end_global_changes();
3162#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003163}
3164
3165/*
3166 * Add files[count] to the arglist of the current window after arg "after".
3167 * The file names in files[count] must have been allocated and are taken over.
3168 * Files[] itself is not taken over.
3169 * Returns index of first added argument. Returns -1 when failed (out of mem).
3170 */
3171 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003172alist_add_list(
3173 int count,
3174 char_u **files,
3175 int after) /* where to add: 0 = before first one */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003176{
3177 int i;
Bram Moolenaara24f0a52016-01-17 19:39:00 +01003178 int old_argcount = ARGCOUNT;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003179
3180 if (ga_grow(&ALIST(curwin)->al_ga, count) == OK)
3181 {
3182 if (after < 0)
3183 after = 0;
3184 if (after > ARGCOUNT)
3185 after = ARGCOUNT;
3186 if (after < ARGCOUNT)
3187 mch_memmove(&(ARGLIST[after + count]), &(ARGLIST[after]),
3188 (ARGCOUNT - after) * sizeof(aentry_T));
3189 for (i = 0; i < count; ++i)
3190 {
3191 ARGLIST[after + i].ae_fname = files[i];
3192 ARGLIST[after + i].ae_fnum = buflist_add(files[i], BLN_LISTED);
3193 }
3194 ALIST(curwin)->al_ga.ga_len += count;
Bram Moolenaara24f0a52016-01-17 19:39:00 +01003195 if (old_argcount > 0 && curwin->w_arg_idx >= after)
3196 curwin->w_arg_idx += count;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003197 return after;
3198 }
3199
3200 for (i = 0; i < count; ++i)
3201 vim_free(files[i]);
3202 return -1;
3203}
3204
3205#endif /* FEAT_LISTCMDS */
3206
3207#ifdef FEAT_EVAL
3208/*
3209 * ":compiler[!] {name}"
3210 */
3211 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003212ex_compiler(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003213{
3214 char_u *buf;
3215 char_u *old_cur_comp = NULL;
3216 char_u *p;
3217
3218 if (*eap->arg == NUL)
3219 {
3220 /* List all compiler scripts. */
3221 do_cmdline_cmd((char_u *)"echo globpath(&rtp, 'compiler/*.vim')");
3222 /* ) keep the indenter happy... */
3223 }
3224 else
3225 {
3226 buf = alloc((unsigned)(STRLEN(eap->arg) + 14));
3227 if (buf != NULL)
3228 {
3229 if (eap->forceit)
3230 {
3231 /* ":compiler! {name}" sets global options */
3232 do_cmdline_cmd((char_u *)
3233 "command -nargs=* CompilerSet set <args>");
3234 }
3235 else
3236 {
3237 /* ":compiler! {name}" sets local options.
3238 * To remain backwards compatible "current_compiler" is always
3239 * used. A user's compiler plugin may set it, the distributed
3240 * plugin will then skip the settings. Afterwards set
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003241 * "b:current_compiler" and restore "current_compiler".
3242 * Explicitly prepend "g:" to make it work in a function. */
3243 old_cur_comp = get_var_value((char_u *)"g:current_compiler");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003244 if (old_cur_comp != NULL)
3245 old_cur_comp = vim_strsave(old_cur_comp);
3246 do_cmdline_cmd((char_u *)
3247 "command -nargs=* CompilerSet setlocal <args>");
3248 }
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003249 do_unlet((char_u *)"g:current_compiler", TRUE);
Bram Moolenaar2ce06f62005-01-31 19:19:04 +00003250 do_unlet((char_u *)"b:current_compiler", TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003251
3252 sprintf((char *)buf, "compiler/%s.vim", eap->arg);
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003253 if (source_runtime(buf, DIP_ALL) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003254 EMSG2(_("E666: compiler not supported: %s"), eap->arg);
3255 vim_free(buf);
3256
3257 do_cmdline_cmd((char_u *)":delcommand CompilerSet");
3258
3259 /* Set "b:current_compiler" from "current_compiler". */
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003260 p = get_var_value((char_u *)"g:current_compiler");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003261 if (p != NULL)
3262 set_internal_string_var((char_u *)"b:current_compiler", p);
3263
3264 /* Restore "current_compiler" for ":compiler {name}". */
3265 if (!eap->forceit)
3266 {
3267 if (old_cur_comp != NULL)
3268 {
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003269 set_internal_string_var((char_u *)"g:current_compiler",
Bram Moolenaar071d4272004-06-13 20:20:40 +00003270 old_cur_comp);
3271 vim_free(old_cur_comp);
3272 }
3273 else
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003274 do_unlet((char_u *)"g:current_compiler", TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003275 }
3276 }
3277 }
3278}
3279#endif
3280
3281/*
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003282 * ":runtime [what] {name}"
Bram Moolenaar071d4272004-06-13 20:20:40 +00003283 */
3284 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003285ex_runtime(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003286{
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003287 char_u *arg = eap->arg;
3288 char_u *p = skiptowhite(arg);
3289 int len = (int)(p - arg);
3290 int flags = eap->forceit ? DIP_ALL : 0;
3291
3292 if (STRNCMP(arg, "START", len) == 0)
3293 {
3294 flags += DIP_START + DIP_NORTP;
3295 arg = skipwhite(arg + len);
3296 }
3297 else if (STRNCMP(arg, "OPT", len) == 0)
3298 {
3299 flags += DIP_OPT + DIP_NORTP;
3300 arg = skipwhite(arg + len);
3301 }
3302 else if (STRNCMP(arg, "PACK", len) == 0)
3303 {
3304 flags += DIP_START + DIP_OPT + DIP_NORTP;
3305 arg = skipwhite(arg + len);
3306 }
3307 else if (STRNCMP(arg, "ALL", len) == 0)
3308 {
3309 flags += DIP_START + DIP_OPT;
3310 arg = skipwhite(arg + len);
3311 }
3312
3313 source_runtime(arg, flags);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003314}
3315
Bram Moolenaar071d4272004-06-13 20:20:40 +00003316 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003317source_callback(char_u *fname, void *cookie UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003318{
Bram Moolenaar910f66f2006-04-05 20:41:53 +00003319 (void)do_source(fname, FALSE, DOSO_NONE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003320}
3321
3322/*
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003323 * Find the file "name" in all directories in "path" and invoke
3324 * "callback(fname, cookie)".
3325 * "name" can contain wildcards.
3326 * When "flags" has DIP_ALL: source all files, otherwise only the first one.
3327 * When "flags" has DIP_DIR: find directories instead of files.
3328 * When "flags" has DIP_ERR: give an error message if there is no match.
3329 *
3330 * return FAIL when no file could be sourced, OK otherwise.
3331 */
Bram Moolenaar6bef5302016-03-12 21:28:26 +01003332 int
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003333do_in_path(
3334 char_u *path,
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003335 char_u *name,
Bram Moolenaar91715872016-03-03 17:13:03 +01003336 int flags,
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003337 void (*callback)(char_u *fname, void *ck),
3338 void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003339{
3340 char_u *rtp;
3341 char_u *np;
3342 char_u *buf;
3343 char_u *rtp_copy;
3344 char_u *tail;
3345 int num_files;
3346 char_u **files;
3347 int i;
3348 int did_one = FALSE;
3349#ifdef AMIGA
3350 struct Process *proc = (struct Process *)FindTask(0L);
3351 APTR save_winptr = proc->pr_WindowPtr;
3352
3353 /* Avoid a requester here for a volume that doesn't exist. */
3354 proc->pr_WindowPtr = (APTR)-1L;
3355#endif
3356
3357 /* Make a copy of 'runtimepath'. Invoking the callback may change the
3358 * value. */
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003359 rtp_copy = vim_strsave(path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003360 buf = alloc(MAXPATHL);
3361 if (buf != NULL && rtp_copy != NULL)
3362 {
Bram Moolenaarc09a6d62013-06-10 21:27:29 +02003363 if (p_verbose > 1 && name != NULL)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003364 {
3365 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00003366 smsg((char_u *)_("Searching for \"%s\" in \"%s\""),
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003367 (char *)name, (char *)path);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003368 verbose_leave();
3369 }
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00003370
Bram Moolenaar071d4272004-06-13 20:20:40 +00003371 /* Loop over all entries in 'runtimepath'. */
3372 rtp = rtp_copy;
Bram Moolenaar91715872016-03-03 17:13:03 +01003373 while (*rtp != NUL && ((flags & DIP_ALL) || !did_one))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003374 {
Bram Moolenaar66459b72016-08-06 19:01:55 +02003375 size_t buflen;
3376
Bram Moolenaar071d4272004-06-13 20:20:40 +00003377 /* Copy the path from 'runtimepath' to buf[]. */
3378 copy_option_part(&rtp, buf, MAXPATHL, ",");
Bram Moolenaar66459b72016-08-06 19:01:55 +02003379 buflen = STRLEN(buf);
3380
3381 /* Skip after or non-after directories. */
3382 if (flags & (DIP_NOAFTER | DIP_AFTER))
3383 {
3384 int is_after = buflen >= 5
3385 && STRCMP(buf + buflen - 5, "after") == 0;
3386
3387 if ((is_after && (flags & DIP_NOAFTER))
3388 || (!is_after && (flags & DIP_AFTER)))
3389 continue;
3390 }
3391
Bram Moolenaarc09a6d62013-06-10 21:27:29 +02003392 if (name == NULL)
3393 {
3394 (*callback)(buf, (void *) &cookie);
3395 if (!did_one)
3396 did_one = (cookie == NULL);
3397 }
Bram Moolenaar66459b72016-08-06 19:01:55 +02003398 else if (buflen + STRLEN(name) + 2 < MAXPATHL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003399 {
3400 add_pathsep(buf);
3401 tail = buf + STRLEN(buf);
3402
3403 /* Loop over all patterns in "name" */
3404 np = name;
Bram Moolenaar91715872016-03-03 17:13:03 +01003405 while (*np != NUL && ((flags & DIP_ALL) || !did_one))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003406 {
3407 /* Append the pattern from "name" to buf[]. */
3408 copy_option_part(&np, tail, (int)(MAXPATHL - (tail - buf)),
3409 "\t ");
3410
3411 if (p_verbose > 2)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003412 {
3413 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00003414 smsg((char_u *)_("Searching for \"%s\""), buf);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003415 verbose_leave();
3416 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003417
3418 /* Expand wildcards, invoke the callback for each match. */
3419 if (gen_expand_wildcards(1, &buf, &num_files, &files,
Bram Moolenaar91715872016-03-03 17:13:03 +01003420 (flags & DIP_DIR) ? EW_DIR : EW_FILE) == OK)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003421 {
3422 for (i = 0; i < num_files; ++i)
3423 {
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00003424 (*callback)(files[i], cookie);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003425 did_one = TRUE;
Bram Moolenaar91715872016-03-03 17:13:03 +01003426 if (!(flags & DIP_ALL))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003427 break;
3428 }
3429 FreeWild(num_files, files);
3430 }
3431 }
3432 }
3433 }
3434 }
3435 vim_free(buf);
3436 vim_free(rtp_copy);
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003437 if (!did_one && name != NULL)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003438 {
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003439 char *basepath = path == p_rtp ? "runtimepath" : "packpath";
3440
3441 if (flags & DIP_ERR)
3442 EMSG3(_(e_dirnotf), basepath, name);
3443 else if (p_verbose > 0)
3444 {
3445 verbose_enter();
3446 smsg((char_u *)_("not found in '%s': \"%s\""), basepath, name);
3447 verbose_leave();
3448 }
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003449 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003450
3451#ifdef AMIGA
3452 proc->pr_WindowPtr = save_winptr;
3453#endif
3454
3455 return did_one ? OK : FAIL;
3456}
3457
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003458/*
Bram Moolenaar07ecfa62017-06-27 14:43:55 +02003459 * Find "name" in "path". When found, invoke the callback function for
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003460 * it: callback(fname, "cookie")
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003461 * When "flags" has DIP_ALL repeat for all matches, otherwise only the first
3462 * one is used.
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003463 * Returns OK when at least one match found, FAIL otherwise.
3464 *
Bram Moolenaar07ecfa62017-06-27 14:43:55 +02003465 * If "name" is NULL calls callback for each entry in "path". Cookie is
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003466 * passed by reference in this case, setting it to NULL indicates that callback
3467 * has done its job.
3468 */
Bram Moolenaar07ecfa62017-06-27 14:43:55 +02003469 static int
3470do_in_path_and_pp(
3471 char_u *path,
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003472 char_u *name,
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003473 int flags,
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003474 void (*callback)(char_u *fname, void *ck),
3475 void *cookie)
3476{
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003477 int done = FAIL;
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003478 char_u *s;
3479 int len;
3480 char *start_dir = "pack/*/start/*/%s";
3481 char *opt_dir = "pack/*/opt/*/%s";
3482
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003483 if ((flags & DIP_NORTP) == 0)
Bram Moolenaar07ecfa62017-06-27 14:43:55 +02003484 done = do_in_path(path, name, flags, callback, cookie);
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003485
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003486 if ((done == FAIL || (flags & DIP_ALL)) && (flags & DIP_START))
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003487 {
Bram Moolenaar1c8b4ed2016-03-17 21:51:03 +01003488 len = (int)(STRLEN(start_dir) + STRLEN(name));
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003489 s = alloc(len);
3490 if (s == NULL)
3491 return FAIL;
3492 vim_snprintf((char *)s, len, start_dir, name);
3493 done = do_in_path(p_pp, s, flags, callback, cookie);
3494 vim_free(s);
3495 }
3496
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003497 if ((done == FAIL || (flags & DIP_ALL)) && (flags & DIP_OPT))
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003498 {
Bram Moolenaar1c8b4ed2016-03-17 21:51:03 +01003499 len = (int)(STRLEN(opt_dir) + STRLEN(name));
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003500 s = alloc(len);
3501 if (s == NULL)
3502 return FAIL;
3503 vim_snprintf((char *)s, len, opt_dir, name);
3504 done = do_in_path(p_pp, s, flags, callback, cookie);
3505 vim_free(s);
3506 }
3507
3508 return done;
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003509}
3510
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003511/*
Bram Moolenaar07ecfa62017-06-27 14:43:55 +02003512 * Just like do_in_path_and_pp(), using 'runtimepath' for "path".
3513 */
3514 int
3515do_in_runtimepath(
3516 char_u *name,
3517 int flags,
3518 void (*callback)(char_u *fname, void *ck),
3519 void *cookie)
3520{
3521 return do_in_path_and_pp(p_rtp, name, flags, callback, cookie);
3522}
3523
3524/*
3525 * Source the file "name" from all directories in 'runtimepath'.
3526 * "name" can contain wildcards.
3527 * When "flags" has DIP_ALL: source all files, otherwise only the first one.
3528 *
3529 * return FAIL when no file could be sourced, OK otherwise.
3530 */
3531 int
3532source_runtime(char_u *name, int flags)
3533{
3534 return source_in_path(p_rtp, name, flags);
3535}
3536
3537/*
3538 * Just like source_runtime(), but use "path" instead of 'runtimepath'.
3539 */
3540 int
3541source_in_path(char_u *path, char_u *name, int flags)
3542{
3543 return do_in_path_and_pp(path, name, flags, source_callback, NULL);
3544}
3545
3546
3547/*
Bram Moolenaarf3654822016-03-04 22:12:23 +01003548 * Expand wildcards in "pat" and invoke do_source() for each match.
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003549 */
3550 static void
Bram Moolenaarf3654822016-03-04 22:12:23 +01003551source_all_matches(char_u *pat)
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003552{
Bram Moolenaarf3654822016-03-04 22:12:23 +01003553 int num_files;
3554 char_u **files;
3555 int i;
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003556
Bram Moolenaarf3654822016-03-04 22:12:23 +01003557 if (gen_expand_wildcards(1, &pat, &num_files, &files, EW_FILE) == OK)
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003558 {
Bram Moolenaarf3654822016-03-04 22:12:23 +01003559 for (i = 0; i < num_files; ++i)
3560 (void)do_source(files[i], FALSE, DOSO_NONE);
3561 FreeWild(num_files, files);
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003562 }
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003563}
3564
Bram Moolenaar49b27322016-04-05 21:13:00 +02003565/* used for "cookie" of add_pack_plugin() */
3566static int APP_ADD_DIR;
3567static int APP_LOAD;
3568static int APP_BOTH;
3569
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003570 static void
Bram Moolenaar91715872016-03-03 17:13:03 +01003571add_pack_plugin(char_u *fname, void *cookie)
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003572{
Bram Moolenaarf3654822016-03-04 22:12:23 +01003573 char_u *p4, *p3, *p2, *p1, *p;
3574 char_u *insp;
Bram Moolenaar91715872016-03-03 17:13:03 +01003575 int c;
3576 char_u *new_rtp;
3577 int keep;
Bram Moolenaarb0550662016-05-31 21:37:36 +02003578 size_t oldlen;
3579 size_t addlen;
Bram Moolenaara5702442016-05-24 19:37:29 +02003580 char_u *afterdir;
Bram Moolenaarb0550662016-05-31 21:37:36 +02003581 size_t afterlen = 0;
Bram Moolenaar91715872016-03-03 17:13:03 +01003582 char_u *ffname = fix_fname(fname);
Bram Moolenaarfef524b2016-07-02 22:07:22 +02003583 size_t fname_len;
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003584 char_u *buf = NULL;
3585 char_u *rtp_ffname;
3586 int match;
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003587
Bram Moolenaar91715872016-03-03 17:13:03 +01003588 if (ffname == NULL)
3589 return;
Bram Moolenaar49b27322016-04-05 21:13:00 +02003590 if (cookie != &APP_LOAD && strstr((char *)p_rtp, (char *)ffname) == NULL)
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003591 {
Bram Moolenaara5702442016-05-24 19:37:29 +02003592 /* directory is not yet in 'runtimepath', add it */
Bram Moolenaarf3654822016-03-04 22:12:23 +01003593 p4 = p3 = p2 = p1 = get_past_head(ffname);
Bram Moolenaar91acfff2017-03-12 19:22:36 +01003594 for (p = p1; *p; MB_PTR_ADV(p))
Bram Moolenaarf3654822016-03-04 22:12:23 +01003595 if (vim_ispathsep_nocolon(*p))
3596 {
3597 p4 = p3; p3 = p2; p2 = p1; p1 = p;
3598 }
3599
3600 /* now we have:
Bram Moolenaaraf1a0e32016-03-09 22:19:26 +01003601 * rtp/pack/name/start/name
3602 * p4 p3 p2 p1
Bram Moolenaarf3654822016-03-04 22:12:23 +01003603 *
3604 * find the part up to "pack" in 'runtimepath' */
3605 c = *p4;
3606 *p4 = NUL;
Bram Moolenaar4c5717e2016-07-01 15:39:40 +02003607
3608 /* Find "ffname" in "p_rtp", ignoring '/' vs '\' differences. */
3609 fname_len = STRLEN(ffname);
3610 insp = p_rtp;
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003611 buf = alloc(MAXPATHL);
3612 if (buf == NULL)
3613 goto theend;
3614 while (*insp != NUL)
Bram Moolenaar4c5717e2016-07-01 15:39:40 +02003615 {
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003616 copy_option_part(&insp, buf, MAXPATHL, ",");
3617 add_pathsep(buf);
3618 rtp_ffname = fix_fname(buf);
3619 if (rtp_ffname == NULL)
3620 goto theend;
3621 match = vim_fnamencmp(rtp_ffname, ffname, fname_len) == 0;
3622 vim_free(rtp_ffname);
3623 if (match)
Bram Moolenaar4c5717e2016-07-01 15:39:40 +02003624 break;
Bram Moolenaar4c5717e2016-07-01 15:39:40 +02003625 }
3626
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003627 if (*insp == NUL)
Bram Moolenaarf3654822016-03-04 22:12:23 +01003628 /* not found, append at the end */
3629 insp = p_rtp + STRLEN(p_rtp);
3630 else
Bram Moolenaarf3654822016-03-04 22:12:23 +01003631 /* append after the matching directory. */
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003632 --insp;
Bram Moolenaarf3654822016-03-04 22:12:23 +01003633 *p4 = c;
3634
Bram Moolenaara5702442016-05-24 19:37:29 +02003635 /* check if rtp/pack/name/start/name/after exists */
3636 afterdir = concat_fnames(ffname, (char_u *)"after", TRUE);
3637 if (afterdir != NULL && mch_isdir(afterdir))
3638 afterlen = STRLEN(afterdir) + 1; /* add one for comma */
3639
Bram Moolenaarb0550662016-05-31 21:37:36 +02003640 oldlen = STRLEN(p_rtp);
3641 addlen = STRLEN(ffname) + 1; /* add one for comma */
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003642 new_rtp = alloc((int)(oldlen + addlen + afterlen + 1));
3643 /* add one for NUL */
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003644 if (new_rtp == NULL)
Bram Moolenaarf3654822016-03-04 22:12:23 +01003645 goto theend;
3646 keep = (int)(insp - p_rtp);
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003647 mch_memmove(new_rtp, p_rtp, keep);
3648 new_rtp[keep] = ',';
Bram Moolenaara5702442016-05-24 19:37:29 +02003649 mch_memmove(new_rtp + keep + 1, ffname, addlen);
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003650 if (p_rtp[keep] != NUL)
Bram Moolenaara5702442016-05-24 19:37:29 +02003651 mch_memmove(new_rtp + keep + addlen, p_rtp + keep,
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003652 oldlen - keep + 1);
Bram Moolenaara5702442016-05-24 19:37:29 +02003653 if (afterlen > 0)
3654 {
3655 STRCAT(new_rtp, ",");
3656 STRCAT(new_rtp, afterdir);
3657 }
Bram Moolenaar863c1a92016-03-03 15:47:06 +01003658 set_option_value((char_u *)"rtp", 0L, new_rtp, 0);
3659 vim_free(new_rtp);
Bram Moolenaara5702442016-05-24 19:37:29 +02003660 vim_free(afterdir);
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003661 }
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003662
Bram Moolenaar49b27322016-04-05 21:13:00 +02003663 if (cookie != &APP_ADD_DIR)
Bram Moolenaarf3654822016-03-04 22:12:23 +01003664 {
Bram Moolenaar71fb0c12016-04-02 22:44:16 +02003665 static char *plugpat = "%s/plugin/**/*.vim";
Bram Moolenaarf3654822016-03-04 22:12:23 +01003666 static char *ftpat = "%s/ftdetect/*.vim";
3667 int len;
3668 char_u *pat;
3669
3670 len = (int)STRLEN(ffname) + (int)STRLEN(ftpat);
3671 pat = alloc(len);
3672 if (pat == NULL)
3673 goto theend;
3674 vim_snprintf((char *)pat, len, plugpat, ffname);
3675 source_all_matches(pat);
3676
3677#ifdef FEAT_AUTOCMD
3678 {
3679 char_u *cmd = vim_strsave((char_u *)"g:did_load_filetypes");
3680
3681 /* If runtime/filetype.vim wasn't loaded yet, the scripts will be
3682 * found when it loads. */
3683 if (cmd != NULL && eval_to_number(cmd) > 0)
3684 {
3685 do_cmdline_cmd((char_u *)"augroup filetypedetect");
3686 vim_snprintf((char *)pat, len, ftpat, ffname);
3687 source_all_matches(pat);
3688 do_cmdline_cmd((char_u *)"augroup END");
3689 }
3690 vim_free(cmd);
3691 }
3692#endif
Bram Moolenaarba8cd122016-03-19 14:16:39 +01003693 vim_free(pat);
Bram Moolenaarf3654822016-03-04 22:12:23 +01003694 }
3695
3696theend:
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003697 vim_free(buf);
Bram Moolenaarf3654822016-03-04 22:12:23 +01003698 vim_free(ffname);
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003699}
3700
Bram Moolenaarce876aa2017-06-04 17:47:42 +02003701/*
3702 * Add all packages in the "start" directory to 'runtimepath'.
3703 */
3704 void
3705add_pack_start_dirs(void)
3706{
3707 do_in_path(p_pp, (char_u *)"pack/*/start/*", DIP_ALL + DIP_DIR,
3708 add_pack_plugin, &APP_ADD_DIR);
3709}
3710
3711/*
3712 * Load plugins from all packages in the "start" directory.
3713 */
3714 void
3715load_start_packages(void)
3716{
3717 did_source_packages = TRUE;
3718 do_in_path(p_pp, (char_u *)"pack/*/start/*", DIP_ALL + DIP_DIR,
3719 add_pack_plugin, &APP_LOAD);
3720}
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003721
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003722/*
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003723 * ":packloadall"
Bram Moolenaarf3654822016-03-04 22:12:23 +01003724 * Find plugins in the package directories and source them.
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003725 */
3726 void
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003727ex_packloadall(exarg_T *eap)
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003728{
Bram Moolenaarce876aa2017-06-04 17:47:42 +02003729 if (!did_source_packages || eap->forceit)
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003730 {
Bram Moolenaar49b27322016-04-05 21:13:00 +02003731 /* First do a round to add all directories to 'runtimepath', then load
3732 * the plugins. This allows for plugins to use an autoload directory
3733 * of another plugin. */
Bram Moolenaarce876aa2017-06-04 17:47:42 +02003734 add_pack_start_dirs();
3735 load_start_packages();
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003736 }
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003737}
3738
3739/*
Bram Moolenaarf3654822016-03-04 22:12:23 +01003740 * ":packadd[!] {name}"
Bram Moolenaar91715872016-03-03 17:13:03 +01003741 */
3742 void
3743ex_packadd(exarg_T *eap)
3744{
3745 static char *plugpat = "pack/*/opt/%s";
3746 int len;
3747 char *pat;
3748
3749 len = (int)STRLEN(plugpat) + (int)STRLEN(eap->arg);
3750 pat = (char *)alloc(len);
3751 if (pat == NULL)
3752 return;
3753 vim_snprintf(pat, len, plugpat, eap->arg);
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003754 do_in_path(p_pp, (char_u *)pat, DIP_ALL + DIP_DIR + DIP_ERR,
Bram Moolenaar49b27322016-04-05 21:13:00 +02003755 add_pack_plugin, eap->forceit ? &APP_ADD_DIR : &APP_BOTH);
Bram Moolenaar91715872016-03-03 17:13:03 +01003756 vim_free(pat);
3757}
3758
Bram Moolenaar071d4272004-06-13 20:20:40 +00003759#if defined(FEAT_EVAL) && defined(FEAT_AUTOCMD)
3760/*
3761 * ":options"
3762 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003763 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003764ex_options(
3765 exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003766{
Bram Moolenaarab6c8582017-08-11 17:15:09 +02003767 vim_setenv((char_u *)"OPTWIN_CMD", (char_u *)(cmdmod.tab ? "tab" : ""));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003768 cmd_source((char_u *)SYS_OPTWIN_FILE, NULL);
3769}
3770#endif
3771
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01003772#if defined(FEAT_PYTHON3) || defined(FEAT_PYTHON) || defined(PROTO)
3773
3774# if (defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)) || defined(PROTO)
3775/*
3776 * Detect Python 3 or 2, and initialize 'pyxversion'.
3777 */
3778 void
3779init_pyxversion(void)
3780{
3781 if (p_pyx == 0)
3782 {
3783 if (python3_enabled(FALSE))
3784 p_pyx = 3;
3785 else if (python_enabled(FALSE))
3786 p_pyx = 2;
3787 }
3788}
3789# endif
3790
3791/*
3792 * Does a file contain one of the following strings at the beginning of any
3793 * line?
3794 * "#!(any string)python2" => returns 2
3795 * "#!(any string)python3" => returns 3
3796 * "# requires python 2.x" => returns 2
3797 * "# requires python 3.x" => returns 3
3798 * otherwise return 0.
3799 */
3800 static int
3801requires_py_version(char_u *filename)
3802{
3803 FILE *file;
3804 int requires_py_version = 0;
3805 int i, lines;
3806
3807 lines = (int)p_mls;
3808 if (lines < 0)
3809 lines = 5;
3810
3811 file = mch_fopen((char *)filename, "r");
3812 if (file != NULL)
3813 {
3814 for (i = 0; i < lines; i++)
3815 {
3816 if (vim_fgets(IObuff, IOSIZE, file))
3817 break;
3818 if (i == 0 && IObuff[0] == '#' && IObuff[1] == '!')
3819 {
3820 /* Check shebang. */
3821 if (strstr((char *)IObuff + 2, "python2") != NULL)
3822 {
3823 requires_py_version = 2;
3824 break;
3825 }
3826 if (strstr((char *)IObuff + 2, "python3") != NULL)
3827 {
3828 requires_py_version = 3;
3829 break;
3830 }
3831 }
3832 IObuff[21] = '\0';
3833 if (STRCMP("# requires python 2.x", IObuff) == 0)
3834 {
3835 requires_py_version = 2;
3836 break;
3837 }
3838 if (STRCMP("# requires python 3.x", IObuff) == 0)
3839 {
3840 requires_py_version = 3;
3841 break;
3842 }
3843 }
3844 fclose(file);
3845 }
3846 return requires_py_version;
3847}
3848
3849
3850/*
3851 * Source a python file using the requested python version.
3852 */
3853 static void
3854source_pyx_file(exarg_T *eap, char_u *fname)
3855{
3856 exarg_T ex;
3857 int v = requires_py_version(fname);
3858
3859# if defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)
3860 init_pyxversion();
3861# endif
3862 if (v == 0)
3863 {
3864# if defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)
3865 /* user didn't choose a preference, 'pyx' is used */
3866 v = p_pyx;
3867# elif defined(FEAT_PYTHON)
3868 v = 2;
3869# elif defined(FEAT_PYTHON3)
3870 v = 3;
3871# endif
3872 }
3873
3874 /*
3875 * now source, if required python version is not supported show
3876 * unobtrusive message.
3877 */
3878 if (eap == NULL)
3879 vim_memset(&ex, 0, sizeof(ex));
3880 else
3881 ex = *eap;
3882 ex.arg = fname;
3883 ex.cmd = (char_u *)(v == 2 ? "pyfile" : "pyfile3");
3884
3885 if (v == 2)
3886 {
3887# ifdef FEAT_PYTHON
3888 ex_pyfile(&ex);
3889# else
3890 vim_snprintf((char *)IObuff, IOSIZE,
3891 _("W20: Required python version 2.x not supported, ignoring file: %s"),
3892 fname);
3893 MSG(IObuff);
3894# endif
3895 return;
3896 }
3897 else
3898 {
3899# ifdef FEAT_PYTHON3
3900 ex_py3file(&ex);
3901# else
3902 vim_snprintf((char *)IObuff, IOSIZE,
3903 _("W21: Required python version 3.x not supported, ignoring file: %s"),
3904 fname);
3905 MSG(IObuff);
3906# endif
3907 return;
3908 }
3909}
3910
3911/*
3912 * ":pyxfile {fname}"
3913 */
3914 void
3915ex_pyxfile(exarg_T *eap)
3916{
3917 source_pyx_file(eap, eap->arg);
3918}
3919
3920/*
3921 * ":pyx"
3922 */
3923 void
3924ex_pyx(exarg_T *eap)
3925{
3926# if defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)
3927 init_pyxversion();
3928 if (p_pyx == 2)
3929 ex_python(eap);
3930 else
3931 ex_py3(eap);
3932# elif defined(FEAT_PYTHON)
3933 ex_python(eap);
3934# elif defined(FEAT_PYTHON3)
3935 ex_py3(eap);
3936# endif
3937}
3938
3939/*
3940 * ":pyxdo"
3941 */
3942 void
3943ex_pyxdo(exarg_T *eap)
3944{
3945# if defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)
3946 init_pyxversion();
3947 if (p_pyx == 2)
3948 ex_pydo(eap);
3949 else
3950 ex_py3do(eap);
3951# elif defined(FEAT_PYTHON)
3952 ex_pydo(eap);
3953# elif defined(FEAT_PYTHON3)
3954 ex_py3do(eap);
3955# endif
3956}
3957
3958#endif
3959
Bram Moolenaar071d4272004-06-13 20:20:40 +00003960/*
3961 * ":source {fname}"
3962 */
3963 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003964ex_source(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003965{
3966#ifdef FEAT_BROWSE
3967 if (cmdmod.browse)
3968 {
3969 char_u *fname = NULL;
3970
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00003971 fname = do_browse(0, (char_u *)_("Source Vim script"), eap->arg,
Bram Moolenaar071d4272004-06-13 20:20:40 +00003972 NULL, NULL, BROWSE_FILTER_MACROS, NULL);
3973 if (fname != NULL)
3974 {
3975 cmd_source(fname, eap);
3976 vim_free(fname);
3977 }
3978 }
3979 else
3980#endif
3981 cmd_source(eap->arg, eap);
3982}
3983
3984 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003985cmd_source(char_u *fname, exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003986{
3987 if (*fname == NUL)
3988 EMSG(_(e_argreq));
3989
Bram Moolenaar071d4272004-06-13 20:20:40 +00003990 else if (eap != NULL && eap->forceit)
Bram Moolenaar84a05ac2013-05-06 04:24:17 +02003991 /* ":source!": read Normal mode commands
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00003992 * Need to execute the commands directly. This is required at least
3993 * for:
Bram Moolenaar071d4272004-06-13 20:20:40 +00003994 * - ":g" command busy
3995 * - after ":argdo", ":windo" or ":bufdo"
3996 * - another command follows
3997 * - inside a loop
3998 */
3999 openscript(fname, global_busy || listcmd_busy || eap->nextcmd != NULL
4000#ifdef FEAT_EVAL
4001 || eap->cstack->cs_idx >= 0
4002#endif
4003 );
4004
4005 /* ":source" read ex commands */
Bram Moolenaar910f66f2006-04-05 20:41:53 +00004006 else if (do_source(fname, FALSE, DOSO_NONE) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004007 EMSG2(_(e_notopen), fname);
4008}
4009
4010/*
4011 * ":source" and associated commands.
4012 */
4013/*
4014 * Structure used to store info for each sourced file.
4015 * It is shared between do_source() and getsourceline().
4016 * This is required, because it needs to be handed to do_cmdline() and
4017 * sourcing can be done recursively.
4018 */
4019struct source_cookie
4020{
4021 FILE *fp; /* opened file for sourcing */
4022 char_u *nextline; /* if not NULL: line that was read ahead */
4023 int finished; /* ":finish" used */
Bram Moolenaar860cae12010-06-05 23:22:07 +02004024#if defined(USE_CRNL) || defined(USE_CR)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004025 int fileformat; /* EOL_UNKNOWN, EOL_UNIX or EOL_DOS */
4026 int error; /* TRUE if LF found after CR-LF */
4027#endif
4028#ifdef FEAT_EVAL
4029 linenr_T breakpoint; /* next line with breakpoint or zero */
4030 char_u *fname; /* name of sourced file */
4031 int dbg_tick; /* debug_tick when breakpoint was set */
4032 int level; /* top nesting level of sourced file */
4033#endif
4034#ifdef FEAT_MBYTE
4035 vimconv_T conv; /* type of conversion */
4036#endif
4037};
4038
4039#ifdef FEAT_EVAL
4040/*
4041 * Return the address holding the next breakpoint line for a source cookie.
4042 */
4043 linenr_T *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004044source_breakpoint(void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004045{
4046 return &((struct source_cookie *)cookie)->breakpoint;
4047}
4048
4049/*
4050 * Return the address holding the debug tick for a source cookie.
4051 */
4052 int *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004053source_dbg_tick(void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004054{
4055 return &((struct source_cookie *)cookie)->dbg_tick;
4056}
4057
4058/*
4059 * Return the nesting level for a source cookie.
4060 */
4061 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004062source_level(void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004063{
4064 return ((struct source_cookie *)cookie)->level;
4065}
4066#endif
4067
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01004068static char_u *get_one_sourceline(struct source_cookie *sp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004069
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004070#if (defined(WIN32) && defined(FEAT_CSCOPE)) || defined(HAVE_FD_CLOEXEC)
4071# define USE_FOPEN_NOINH
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01004072static FILE *fopen_noinh_readbin(char *filename);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004073
4074/*
4075 * Special function to open a file without handle inheritance.
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004076 * When possible the handle is closed on exec().
Bram Moolenaar071d4272004-06-13 20:20:40 +00004077 */
4078 static FILE *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004079fopen_noinh_readbin(char *filename)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004080{
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004081# ifdef WIN32
Bram Moolenaar8d8ef0b2010-01-20 21:41:47 +01004082 int fd_tmp = mch_open(filename, O_RDONLY | O_BINARY | O_NOINHERIT, 0);
4083# else
4084 int fd_tmp = mch_open(filename, O_RDONLY, 0);
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004085# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004086
4087 if (fd_tmp == -1)
4088 return NULL;
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004089
4090# ifdef HAVE_FD_CLOEXEC
4091 {
4092 int fdflags = fcntl(fd_tmp, F_GETFD);
4093 if (fdflags >= 0 && (fdflags & FD_CLOEXEC) == 0)
Bram Moolenaarcde88542015-08-11 19:14:00 +02004094 (void)fcntl(fd_tmp, F_SETFD, fdflags | FD_CLOEXEC);
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004095 }
4096# endif
4097
Bram Moolenaar071d4272004-06-13 20:20:40 +00004098 return fdopen(fd_tmp, READBIN);
4099}
4100#endif
4101
4102
4103/*
4104 * do_source: Read the file "fname" and execute its lines as EX commands.
4105 *
4106 * This function may be called recursively!
4107 *
4108 * return FAIL if file could not be opened, OK otherwise
4109 */
4110 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004111do_source(
4112 char_u *fname,
4113 int check_other, /* check for .vimrc and _vimrc */
4114 int is_vimrc) /* DOSO_ value */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004115{
4116 struct source_cookie cookie;
4117 char_u *save_sourcing_name;
4118 linenr_T save_sourcing_lnum;
4119 char_u *p;
4120 char_u *fname_exp;
Bram Moolenaar73881402009-02-04 16:50:47 +00004121 char_u *firstline = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004122 int retval = FAIL;
4123#ifdef FEAT_EVAL
4124 scid_T save_current_SID;
4125 static scid_T last_current_SID = 0;
4126 void *save_funccalp;
4127 int save_debug_break_level = debug_break_level;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004128 scriptitem_T *si = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004129# ifdef UNIX
Bram Moolenaar8767f522016-07-01 17:17:39 +02004130 stat_T st;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004131 int stat_ok;
4132# endif
4133#endif
4134#ifdef STARTUPTIME
4135 struct timeval tv_rel;
4136 struct timeval tv_start;
4137#endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00004138#ifdef FEAT_PROFILE
4139 proftime_T wait_start;
4140#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004141
Bram Moolenaar071d4272004-06-13 20:20:40 +00004142 p = expand_env_save(fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004143 if (p == NULL)
4144 return retval;
4145 fname_exp = fix_fname(p);
4146 vim_free(p);
4147 if (fname_exp == NULL)
4148 return retval;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004149 if (mch_isdir(fname_exp))
4150 {
Bram Moolenaar555b2802005-05-19 21:08:39 +00004151 smsg((char_u *)_("Cannot source a directory: \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004152 goto theend;
4153 }
4154
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00004155#ifdef FEAT_AUTOCMD
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00004156 /* Apply SourceCmd autocommands, they should get the file and source it. */
4157 if (has_autocmd(EVENT_SOURCECMD, fname_exp, NULL)
4158 && apply_autocmds(EVENT_SOURCECMD, fname_exp, fname_exp,
4159 FALSE, curbuf))
Bram Moolenaarf33943e2008-01-15 21:17:30 +00004160 {
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00004161# ifdef FEAT_EVAL
Bram Moolenaarf33943e2008-01-15 21:17:30 +00004162 retval = aborting() ? FAIL : OK;
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00004163# else
Bram Moolenaarf33943e2008-01-15 21:17:30 +00004164 retval = OK;
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00004165# endif
Bram Moolenaarf33943e2008-01-15 21:17:30 +00004166 goto theend;
4167 }
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00004168
4169 /* Apply SourcePre autocommands, they may get the file. */
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00004170 apply_autocmds(EVENT_SOURCEPRE, fname_exp, fname_exp, FALSE, curbuf);
4171#endif
4172
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004173#ifdef USE_FOPEN_NOINH
Bram Moolenaar071d4272004-06-13 20:20:40 +00004174 cookie.fp = fopen_noinh_readbin((char *)fname_exp);
4175#else
4176 cookie.fp = mch_fopen((char *)fname_exp, READBIN);
4177#endif
4178 if (cookie.fp == NULL && check_other)
4179 {
4180 /*
4181 * Try again, replacing file name ".vimrc" by "_vimrc" or vice versa,
4182 * and ".exrc" by "_exrc" or vice versa.
4183 */
4184 p = gettail(fname_exp);
4185 if ((*p == '.' || *p == '_')
4186 && (STRICMP(p + 1, "vimrc") == 0
4187 || STRICMP(p + 1, "gvimrc") == 0
4188 || STRICMP(p + 1, "exrc") == 0))
4189 {
4190 if (*p == '_')
4191 *p = '.';
4192 else
4193 *p = '_';
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004194#ifdef USE_FOPEN_NOINH
Bram Moolenaar071d4272004-06-13 20:20:40 +00004195 cookie.fp = fopen_noinh_readbin((char *)fname_exp);
4196#else
4197 cookie.fp = mch_fopen((char *)fname_exp, READBIN);
4198#endif
4199 }
4200 }
4201
4202 if (cookie.fp == NULL)
4203 {
4204 if (p_verbose > 0)
4205 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004206 verbose_enter();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004207 if (sourcing_name == NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00004208 smsg((char_u *)_("could not source \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004209 else
4210 smsg((char_u *)_("line %ld: could not source \"%s\""),
Bram Moolenaar555b2802005-05-19 21:08:39 +00004211 sourcing_lnum, fname);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004212 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004213 }
4214 goto theend;
4215 }
4216
4217 /*
4218 * The file exists.
4219 * - In verbose mode, give a message.
4220 * - For a vimrc file, may want to set 'compatible', call vimrc_found().
4221 */
4222 if (p_verbose > 1)
4223 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004224 verbose_enter();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004225 if (sourcing_name == NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00004226 smsg((char_u *)_("sourcing \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004227 else
4228 smsg((char_u *)_("line %ld: sourcing \"%s\""),
Bram Moolenaar555b2802005-05-19 21:08:39 +00004229 sourcing_lnum, fname);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004230 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004231 }
Bram Moolenaar910f66f2006-04-05 20:41:53 +00004232 if (is_vimrc == DOSO_VIMRC)
4233 vimrc_found(fname_exp, (char_u *)"MYVIMRC");
4234 else if (is_vimrc == DOSO_GVIMRC)
4235 vimrc_found(fname_exp, (char_u *)"MYGVIMRC");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004236
4237#ifdef USE_CRNL
4238 /* If no automatic file format: Set default to CR-NL. */
4239 if (*p_ffs == NUL)
4240 cookie.fileformat = EOL_DOS;
4241 else
4242 cookie.fileformat = EOL_UNKNOWN;
4243 cookie.error = FALSE;
4244#endif
4245
4246#ifdef USE_CR
4247 /* If no automatic file format: Set default to CR. */
4248 if (*p_ffs == NUL)
4249 cookie.fileformat = EOL_MAC;
4250 else
4251 cookie.fileformat = EOL_UNKNOWN;
4252 cookie.error = FALSE;
4253#endif
4254
4255 cookie.nextline = NULL;
4256 cookie.finished = FALSE;
4257
4258#ifdef FEAT_EVAL
4259 /*
4260 * Check if this script has a breakpoint.
4261 */
4262 cookie.breakpoint = dbg_find_breakpoint(TRUE, fname_exp, (linenr_T)0);
4263 cookie.fname = fname_exp;
4264 cookie.dbg_tick = debug_tick;
4265
4266 cookie.level = ex_nesting_level;
4267#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004268
4269 /*
4270 * Keep the sourcing name/lnum, for recursive calls.
4271 */
4272 save_sourcing_name = sourcing_name;
4273 sourcing_name = fname_exp;
4274 save_sourcing_lnum = sourcing_lnum;
4275 sourcing_lnum = 0;
4276
Bram Moolenaar73881402009-02-04 16:50:47 +00004277#ifdef FEAT_MBYTE
4278 cookie.conv.vc_type = CONV_NONE; /* no conversion */
4279
4280 /* Read the first line so we can check for a UTF-8 BOM. */
4281 firstline = getsourceline(0, (void *)&cookie, 0);
4282 if (firstline != NULL && STRLEN(firstline) >= 3 && firstline[0] == 0xef
4283 && firstline[1] == 0xbb && firstline[2] == 0xbf)
4284 {
4285 /* Found BOM; setup conversion, skip over BOM and recode the line. */
4286 convert_setup(&cookie.conv, (char_u *)"utf-8", p_enc);
4287 p = string_convert(&cookie.conv, firstline + 3, NULL);
Bram Moolenaar4e2a5952009-02-05 19:48:25 +00004288 if (p == NULL)
4289 p = vim_strsave(firstline + 3);
Bram Moolenaar73881402009-02-04 16:50:47 +00004290 if (p != NULL)
4291 {
4292 vim_free(firstline);
4293 firstline = p;
4294 }
4295 }
4296#endif
4297
Bram Moolenaar071d4272004-06-13 20:20:40 +00004298#ifdef STARTUPTIME
Bram Moolenaarc4e41982010-01-19 16:31:47 +01004299 if (time_fd != NULL)
4300 time_push(&tv_rel, &tv_start);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004301#endif
4302
4303#ifdef FEAT_EVAL
Bram Moolenaar05159a02005-02-26 23:04:13 +00004304# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004305 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004306 prof_child_enter(&wait_start); /* entering a child now */
4307# endif
4308
4309 /* Don't use local function variables, if called from a function.
4310 * Also starts profiling timer for nested script. */
4311 save_funccalp = save_funccal();
4312
Bram Moolenaar071d4272004-06-13 20:20:40 +00004313 /*
4314 * Check if this script was sourced before to finds its SID.
4315 * If it's new, generate a new SID.
4316 */
4317 save_current_SID = current_SID;
4318# ifdef UNIX
4319 stat_ok = (mch_stat((char *)fname_exp, &st) >= 0);
4320# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00004321 for (current_SID = script_items.ga_len; current_SID > 0; --current_SID)
4322 {
4323 si = &SCRIPT_ITEM(current_SID);
4324 if (si->sn_name != NULL
Bram Moolenaar071d4272004-06-13 20:20:40 +00004325 && (
4326# ifdef UNIX
Bram Moolenaar7c626922005-02-07 22:01:03 +00004327 /* Compare dev/ino when possible, it catches symbolic
4328 * links. Also compare file names, the inode may change
4329 * when the file was edited. */
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00004330 ((stat_ok && si->sn_dev_valid)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004331 && (si->sn_dev == st.st_dev
4332 && si->sn_ino == st.st_ino)) ||
Bram Moolenaar071d4272004-06-13 20:20:40 +00004333# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00004334 fnamecmp(si->sn_name, fname_exp) == 0))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004335 break;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004336 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004337 if (current_SID == 0)
4338 {
4339 current_SID = ++last_current_SID;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004340 if (ga_grow(&script_items, (int)(current_SID - script_items.ga_len))
4341 == FAIL)
4342 goto almosttheend;
4343 while (script_items.ga_len < current_SID)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004344 {
Bram Moolenaar05159a02005-02-26 23:04:13 +00004345 ++script_items.ga_len;
4346 SCRIPT_ITEM(script_items.ga_len).sn_name = NULL;
4347# ifdef FEAT_PROFILE
4348 SCRIPT_ITEM(script_items.ga_len).sn_prof_on = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004349# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004350 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00004351 si = &SCRIPT_ITEM(current_SID);
4352 si->sn_name = fname_exp;
4353 fname_exp = NULL;
4354# ifdef UNIX
4355 if (stat_ok)
4356 {
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00004357 si->sn_dev_valid = TRUE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004358 si->sn_dev = st.st_dev;
4359 si->sn_ino = st.st_ino;
4360 }
4361 else
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00004362 si->sn_dev_valid = FALSE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004363# endif
4364
Bram Moolenaar071d4272004-06-13 20:20:40 +00004365 /* Allocate the local script variables to use for this script. */
4366 new_script_vars(current_SID);
4367 }
4368
Bram Moolenaar05159a02005-02-26 23:04:13 +00004369# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004370 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004371 {
4372 int forceit;
4373
4374 /* Check if we do profiling for this script. */
4375 if (!si->sn_prof_on && has_profiling(TRUE, si->sn_name, &forceit))
4376 {
4377 script_do_profile(si);
4378 si->sn_pr_force = forceit;
4379 }
4380 if (si->sn_prof_on)
4381 {
4382 ++si->sn_pr_count;
4383 profile_start(&si->sn_pr_start);
4384 profile_zero(&si->sn_pr_children);
4385 }
4386 }
4387# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004388#endif
4389
4390 /*
4391 * Call do_cmdline, which will call getsourceline() to get the lines.
4392 */
Bram Moolenaar73881402009-02-04 16:50:47 +00004393 do_cmdline(firstline, getsourceline, (void *)&cookie,
Bram Moolenaar071d4272004-06-13 20:20:40 +00004394 DOCMD_VERBOSE|DOCMD_NOWAIT|DOCMD_REPEAT);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004395 retval = OK;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004396
4397#ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004398 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004399 {
4400 /* Get "si" again, "script_items" may have been reallocated. */
4401 si = &SCRIPT_ITEM(current_SID);
4402 if (si->sn_prof_on)
4403 {
4404 profile_end(&si->sn_pr_start);
4405 profile_sub_wait(&wait_start, &si->sn_pr_start);
4406 profile_add(&si->sn_pr_total, &si->sn_pr_start);
Bram Moolenaar1056d982006-03-09 22:37:52 +00004407 profile_self(&si->sn_pr_self, &si->sn_pr_start,
4408 &si->sn_pr_children);
Bram Moolenaar05159a02005-02-26 23:04:13 +00004409 }
4410 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004411#endif
4412
4413 if (got_int)
4414 EMSG(_(e_interr));
4415 sourcing_name = save_sourcing_name;
4416 sourcing_lnum = save_sourcing_lnum;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004417 if (p_verbose > 1)
4418 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004419 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00004420 smsg((char_u *)_("finished sourcing %s"), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004421 if (sourcing_name != NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00004422 smsg((char_u *)_("continuing in %s"), sourcing_name);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004423 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004424 }
4425#ifdef STARTUPTIME
Bram Moolenaarc4e41982010-01-19 16:31:47 +01004426 if (time_fd != NULL)
4427 {
4428 vim_snprintf((char *)IObuff, IOSIZE, "sourcing %s", fname);
4429 time_msg((char *)IObuff, &tv_start);
4430 time_pop(&tv_rel);
4431 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004432#endif
4433
4434#ifdef FEAT_EVAL
4435 /*
4436 * After a "finish" in debug mode, need to break at first command of next
4437 * sourced file.
4438 */
4439 if (save_debug_break_level > ex_nesting_level
4440 && debug_break_level == ex_nesting_level)
4441 ++debug_break_level;
4442#endif
4443
Bram Moolenaar05159a02005-02-26 23:04:13 +00004444#ifdef FEAT_EVAL
4445almosttheend:
4446 current_SID = save_current_SID;
4447 restore_funccal(save_funccalp);
4448# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004449 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004450 prof_child_exit(&wait_start); /* leaving a child now */
4451# endif
4452#endif
4453 fclose(cookie.fp);
4454 vim_free(cookie.nextline);
Bram Moolenaar73881402009-02-04 16:50:47 +00004455 vim_free(firstline);
Bram Moolenaar05159a02005-02-26 23:04:13 +00004456#ifdef FEAT_MBYTE
4457 convert_setup(&cookie.conv, NULL, NULL);
4458#endif
4459
Bram Moolenaar071d4272004-06-13 20:20:40 +00004460theend:
4461 vim_free(fname_exp);
4462 return retval;
4463}
4464
4465#if defined(FEAT_EVAL) || defined(PROTO)
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00004466
Bram Moolenaar071d4272004-06-13 20:20:40 +00004467/*
4468 * ":scriptnames"
4469 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004470 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004471ex_scriptnames(exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004472{
4473 int i;
4474
Bram Moolenaar05159a02005-02-26 23:04:13 +00004475 for (i = 1; i <= script_items.ga_len && !got_int; ++i)
4476 if (SCRIPT_ITEM(i).sn_name != NULL)
Bram Moolenaard58ea072011-06-26 04:25:30 +02004477 {
4478 home_replace(NULL, SCRIPT_ITEM(i).sn_name,
4479 NameBuff, MAXPATHL, TRUE);
4480 smsg((char_u *)"%3d: %s", i, NameBuff);
Bram Moolenaar970a1b82012-03-23 18:39:18 +01004481 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004482}
4483
4484# if defined(BACKSLASH_IN_FILENAME) || defined(PROTO)
4485/*
4486 * Fix slashes in the list of script names for 'shellslash'.
4487 */
4488 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004489scriptnames_slash_adjust(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004490{
4491 int i;
4492
Bram Moolenaar05159a02005-02-26 23:04:13 +00004493 for (i = 1; i <= script_items.ga_len; ++i)
4494 if (SCRIPT_ITEM(i).sn_name != NULL)
4495 slash_adjust(SCRIPT_ITEM(i).sn_name);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004496}
4497# endif
4498
4499/*
4500 * Get a pointer to a script name. Used for ":verbose set".
4501 */
4502 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004503get_scriptname(scid_T id)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004504{
4505 if (id == SID_MODELINE)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00004506 return (char_u *)_("modeline");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004507 if (id == SID_CMDARG)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00004508 return (char_u *)_("--cmd argument");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004509 if (id == SID_CARG)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00004510 return (char_u *)_("-c argument");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004511 if (id == SID_ENV)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00004512 return (char_u *)_("environment variable");
4513 if (id == SID_ERROR)
4514 return (char_u *)_("error handler");
Bram Moolenaar05159a02005-02-26 23:04:13 +00004515 return SCRIPT_ITEM(id).sn_name;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004516}
Bram Moolenaar05159a02005-02-26 23:04:13 +00004517
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00004518# if defined(EXITFREE) || defined(PROTO)
4519 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004520free_scriptnames(void)
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00004521{
4522 int i;
4523
4524 for (i = script_items.ga_len; i > 0; --i)
4525 vim_free(SCRIPT_ITEM(i).sn_name);
4526 ga_clear(&script_items);
4527}
4528# endif
4529
Bram Moolenaar071d4272004-06-13 20:20:40 +00004530#endif
4531
4532#if defined(USE_CR) || defined(PROTO)
4533
4534# if defined(__MSL__) && (__MSL__ >= 22)
4535/*
4536 * Newer version of the Metrowerks library handle DOS and UNIX files
4537 * without help.
4538 * Test with earlier versions, MSL 2.2 is the library supplied with
4539 * Codewarrior Pro 2.
4540 */
4541 char *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004542fgets_cr(char *s, int n, FILE *stream)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004543{
4544 return fgets(s, n, stream);
4545}
4546# else
4547/*
4548 * Version of fgets() which also works for lines ending in a <CR> only
4549 * (Macintosh format).
4550 * For older versions of the Metrowerks library.
4551 * At least CodeWarrior 9 needed this code.
4552 */
4553 char *
Bram Moolenaard14e00e2016-01-31 17:30:51 +01004554fgets_cr(char *s, int n, FILE *stream)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004555{
4556 int c = 0;
4557 int char_read = 0;
4558
4559 while (!feof(stream) && c != '\r' && c != '\n' && char_read < n - 1)
4560 {
4561 c = fgetc(stream);
4562 s[char_read++] = c;
4563 /* If the file is in DOS format, we need to skip a NL after a CR. I
4564 * thought it was the other way around, but this appears to work... */
4565 if (c == '\n')
4566 {
4567 c = fgetc(stream);
4568 if (c != '\r')
4569 ungetc(c, stream);
4570 }
4571 }
4572
4573 s[char_read] = 0;
4574 if (char_read == 0)
4575 return NULL;
4576
4577 if (feof(stream) && char_read == 1)
4578 return NULL;
4579
4580 return s;
4581}
4582# endif
4583#endif
4584
4585/*
4586 * Get one full line from a sourced file.
4587 * Called by do_cmdline() when it's called from do_source().
4588 *
4589 * Return a pointer to the line in allocated memory.
4590 * Return NULL for end-of-file or some error.
4591 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004592 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004593getsourceline(int c UNUSED, void *cookie, int indent UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004594{
4595 struct source_cookie *sp = (struct source_cookie *)cookie;
4596 char_u *line;
Bram Moolenaarc047b9a2012-02-11 20:40:55 +01004597 char_u *p;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004598
4599#ifdef FEAT_EVAL
4600 /* If breakpoints have been added/deleted need to check for it. */
4601 if (sp->dbg_tick < debug_tick)
4602 {
4603 sp->breakpoint = dbg_find_breakpoint(TRUE, sp->fname, sourcing_lnum);
4604 sp->dbg_tick = debug_tick;
4605 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00004606# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004607 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004608 script_line_end();
4609# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004610#endif
4611 /*
4612 * Get current line. If there is a read-ahead line, use it, otherwise get
4613 * one now.
4614 */
4615 if (sp->finished)
4616 line = NULL;
4617 else if (sp->nextline == NULL)
4618 line = get_one_sourceline(sp);
4619 else
4620 {
4621 line = sp->nextline;
4622 sp->nextline = NULL;
4623 ++sourcing_lnum;
4624 }
Bram Moolenaare2cc9702005-03-15 22:43:58 +00004625#ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004626 if (line != NULL && do_profiling == PROF_YES)
Bram Moolenaare2cc9702005-03-15 22:43:58 +00004627 script_line_start();
4628#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004629
4630 /* Only concatenate lines starting with a \ when 'cpoptions' doesn't
4631 * contain the 'C' flag. */
4632 if (line != NULL && (vim_strchr(p_cpo, CPO_CONCAT) == NULL))
4633 {
4634 /* compensate for the one line read-ahead */
4635 --sourcing_lnum;
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004636
4637 /* Get the next line and concatenate it when it starts with a
4638 * backslash. We always need to read the next line, keep it in
4639 * sp->nextline. */
4640 sp->nextline = get_one_sourceline(sp);
4641 if (sp->nextline != NULL && *(p = skipwhite(sp->nextline)) == '\\')
Bram Moolenaar071d4272004-06-13 20:20:40 +00004642 {
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004643 garray_T ga;
4644
Bram Moolenaarb549a732012-02-22 18:29:33 +01004645 ga_init2(&ga, (int)sizeof(char_u), 400);
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004646 ga_concat(&ga, line);
4647 ga_concat(&ga, p + 1);
4648 for (;;)
4649 {
4650 vim_free(sp->nextline);
4651 sp->nextline = get_one_sourceline(sp);
4652 if (sp->nextline == NULL)
4653 break;
4654 p = skipwhite(sp->nextline);
4655 if (*p != '\\')
4656 break;
Bram Moolenaarb549a732012-02-22 18:29:33 +01004657 /* Adjust the growsize to the current length to speed up
4658 * concatenating many lines. */
4659 if (ga.ga_len > 400)
4660 {
4661 if (ga.ga_len > 8000)
4662 ga.ga_growsize = 8000;
4663 else
4664 ga.ga_growsize = ga.ga_len;
4665 }
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004666 ga_concat(&ga, p + 1);
4667 }
4668 ga_append(&ga, NUL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004669 vim_free(line);
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004670 line = ga.ga_data;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004671 }
4672 }
4673
4674#ifdef FEAT_MBYTE
4675 if (line != NULL && sp->conv.vc_type != CONV_NONE)
4676 {
Bram Moolenaarc047b9a2012-02-11 20:40:55 +01004677 char_u *s;
4678
Bram Moolenaar071d4272004-06-13 20:20:40 +00004679 /* Convert the encoding of the script line. */
4680 s = string_convert(&sp->conv, line, NULL);
4681 if (s != NULL)
4682 {
4683 vim_free(line);
4684 line = s;
4685 }
4686 }
4687#endif
4688
4689#ifdef FEAT_EVAL
4690 /* Did we encounter a breakpoint? */
4691 if (sp->breakpoint != 0 && sp->breakpoint <= sourcing_lnum)
4692 {
4693 dbg_breakpoint(sp->fname, sourcing_lnum);
4694 /* Find next breakpoint. */
4695 sp->breakpoint = dbg_find_breakpoint(TRUE, sp->fname, sourcing_lnum);
4696 sp->dbg_tick = debug_tick;
4697 }
4698#endif
4699
4700 return line;
4701}
4702
4703 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004704get_one_sourceline(struct source_cookie *sp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004705{
4706 garray_T ga;
4707 int len;
4708 int c;
4709 char_u *buf;
4710#ifdef USE_CRNL
4711 int has_cr; /* CR-LF found */
4712#endif
4713#ifdef USE_CR
4714 char_u *scan;
4715#endif
4716 int have_read = FALSE;
4717
4718 /* use a growarray to store the sourced line */
Bram Moolenaar6ac54292005-02-02 23:07:25 +00004719 ga_init2(&ga, 1, 250);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004720
4721 /*
4722 * Loop until there is a finished line (or end-of-file).
4723 */
4724 sourcing_lnum++;
4725 for (;;)
4726 {
Bram Moolenaar6ac54292005-02-02 23:07:25 +00004727 /* make room to read at least 120 (more) characters */
4728 if (ga_grow(&ga, 120) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004729 break;
4730 buf = (char_u *)ga.ga_data;
4731
4732#ifdef USE_CR
4733 if (sp->fileformat == EOL_MAC)
4734 {
Bram Moolenaar86b68352004-12-27 21:59:20 +00004735 if (fgets_cr((char *)buf + ga.ga_len, ga.ga_maxlen - ga.ga_len,
4736 sp->fp) == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004737 break;
4738 }
4739 else
4740#endif
Bram Moolenaar86b68352004-12-27 21:59:20 +00004741 if (fgets((char *)buf + ga.ga_len, ga.ga_maxlen - ga.ga_len,
4742 sp->fp) == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004743 break;
Bram Moolenaar6ac54292005-02-02 23:07:25 +00004744 len = ga.ga_len + (int)STRLEN(buf + ga.ga_len);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004745#ifdef USE_CRNL
4746 /* Ignore a trailing CTRL-Z, when in Dos mode. Only recognize the
4747 * CTRL-Z by its own, or after a NL. */
4748 if ( (len == 1 || (len >= 2 && buf[len - 2] == '\n'))
4749 && sp->fileformat == EOL_DOS
4750 && buf[len - 1] == Ctrl_Z)
4751 {
4752 buf[len - 1] = NUL;
4753 break;
4754 }
4755#endif
4756
4757#ifdef USE_CR
4758 /* If the read doesn't stop on a new line, and there's
4759 * some CR then we assume a Mac format */
4760 if (sp->fileformat == EOL_UNKNOWN)
4761 {
4762 if (buf[len - 1] != '\n' && vim_strchr(buf, '\r') != NULL)
4763 sp->fileformat = EOL_MAC;
4764 else
4765 sp->fileformat = EOL_UNIX;
4766 }
4767
4768 if (sp->fileformat == EOL_MAC)
4769 {
4770 scan = vim_strchr(buf, '\r');
4771
4772 if (scan != NULL)
4773 {
4774 *scan = '\n';
4775 if (*(scan + 1) != 0)
4776 {
4777 *(scan + 1) = 0;
4778 fseek(sp->fp, (long)(scan - buf - len + 1), SEEK_CUR);
4779 }
4780 }
4781 len = STRLEN(buf);
4782 }
4783#endif
4784
4785 have_read = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004786 ga.ga_len = len;
4787
4788 /* If the line was longer than the buffer, read more. */
Bram Moolenaar86b68352004-12-27 21:59:20 +00004789 if (ga.ga_maxlen - ga.ga_len == 1 && buf[len - 1] != '\n')
Bram Moolenaar071d4272004-06-13 20:20:40 +00004790 continue;
4791
4792 if (len >= 1 && buf[len - 1] == '\n') /* remove trailing NL */
4793 {
4794#ifdef USE_CRNL
4795 has_cr = (len >= 2 && buf[len - 2] == '\r');
4796 if (sp->fileformat == EOL_UNKNOWN)
4797 {
4798 if (has_cr)
4799 sp->fileformat = EOL_DOS;
4800 else
4801 sp->fileformat = EOL_UNIX;
4802 }
4803
4804 if (sp->fileformat == EOL_DOS)
4805 {
4806 if (has_cr) /* replace trailing CR */
4807 {
4808 buf[len - 2] = '\n';
4809 --len;
4810 --ga.ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004811 }
4812 else /* lines like ":map xx yy^M" will have failed */
4813 {
4814 if (!sp->error)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00004815 {
Bram Moolenaar8820b482017-03-16 17:23:31 +01004816 msg_source(HL_ATTR(HLF_W));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004817 EMSG(_("W15: Warning: Wrong line separator, ^M may be missing"));
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00004818 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004819 sp->error = TRUE;
4820 sp->fileformat = EOL_UNIX;
4821 }
4822 }
4823#endif
4824 /* The '\n' is escaped if there is an odd number of ^V's just
4825 * before it, first set "c" just before the 'V's and then check
4826 * len&c parities (is faster than ((len-c)%2 == 0)) -- Acevedo */
4827 for (c = len - 2; c >= 0 && buf[c] == Ctrl_V; c--)
4828 ;
4829 if ((len & 1) != (c & 1)) /* escaped NL, read more */
4830 {
4831 sourcing_lnum++;
4832 continue;
4833 }
4834
4835 buf[len - 1] = NUL; /* remove the NL */
4836 }
4837
4838 /*
4839 * Check for ^C here now and then, so recursive :so can be broken.
4840 */
4841 line_breakcheck();
4842 break;
4843 }
4844
4845 if (have_read)
4846 return (char_u *)ga.ga_data;
4847
4848 vim_free(ga.ga_data);
4849 return NULL;
4850}
4851
Bram Moolenaar05159a02005-02-26 23:04:13 +00004852#if defined(FEAT_PROFILE) || defined(PROTO)
4853/*
4854 * Called when starting to read a script line.
4855 * "sourcing_lnum" must be correct!
4856 * When skipping lines it may not actually be executed, but we won't find out
4857 * until later and we need to store the time now.
4858 */
4859 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004860script_line_start(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004861{
4862 scriptitem_T *si;
4863 sn_prl_T *pp;
4864
4865 if (current_SID <= 0 || current_SID > script_items.ga_len)
4866 return;
4867 si = &SCRIPT_ITEM(current_SID);
4868 if (si->sn_prof_on && sourcing_lnum >= 1)
4869 {
Bram Moolenaar8c8de832008-06-24 22:58:06 +00004870 /* Grow the array before starting the timer, so that the time spent
Bram Moolenaar05159a02005-02-26 23:04:13 +00004871 * here isn't counted. */
Bram Moolenaarcde88542015-08-11 19:14:00 +02004872 (void)ga_grow(&si->sn_prl_ga, (int)(sourcing_lnum - si->sn_prl_ga.ga_len));
Bram Moolenaar05159a02005-02-26 23:04:13 +00004873 si->sn_prl_idx = sourcing_lnum - 1;
4874 while (si->sn_prl_ga.ga_len <= si->sn_prl_idx
4875 && si->sn_prl_ga.ga_len < si->sn_prl_ga.ga_maxlen)
4876 {
4877 /* Zero counters for a line that was not used before. */
4878 pp = &PRL_ITEM(si, si->sn_prl_ga.ga_len);
4879 pp->snp_count = 0;
4880 profile_zero(&pp->sn_prl_total);
4881 profile_zero(&pp->sn_prl_self);
4882 ++si->sn_prl_ga.ga_len;
4883 }
4884 si->sn_prl_execed = FALSE;
4885 profile_start(&si->sn_prl_start);
4886 profile_zero(&si->sn_prl_children);
4887 profile_get_wait(&si->sn_prl_wait);
4888 }
4889}
4890
4891/*
4892 * Called when actually executing a function line.
4893 */
4894 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004895script_line_exec(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004896{
4897 scriptitem_T *si;
4898
4899 if (current_SID <= 0 || current_SID > script_items.ga_len)
4900 return;
4901 si = &SCRIPT_ITEM(current_SID);
4902 if (si->sn_prof_on && si->sn_prl_idx >= 0)
4903 si->sn_prl_execed = TRUE;
4904}
4905
4906/*
4907 * Called when done with a function line.
4908 */
4909 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004910script_line_end(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004911{
4912 scriptitem_T *si;
4913 sn_prl_T *pp;
4914
4915 if (current_SID <= 0 || current_SID > script_items.ga_len)
4916 return;
4917 si = &SCRIPT_ITEM(current_SID);
4918 if (si->sn_prof_on && si->sn_prl_idx >= 0
4919 && si->sn_prl_idx < si->sn_prl_ga.ga_len)
4920 {
4921 if (si->sn_prl_execed)
4922 {
4923 pp = &PRL_ITEM(si, si->sn_prl_idx);
4924 ++pp->snp_count;
4925 profile_end(&si->sn_prl_start);
4926 profile_sub_wait(&si->sn_prl_wait, &si->sn_prl_start);
Bram Moolenaar05159a02005-02-26 23:04:13 +00004927 profile_add(&pp->sn_prl_total, &si->sn_prl_start);
Bram Moolenaar1056d982006-03-09 22:37:52 +00004928 profile_self(&pp->sn_prl_self, &si->sn_prl_start,
4929 &si->sn_prl_children);
Bram Moolenaar05159a02005-02-26 23:04:13 +00004930 }
4931 si->sn_prl_idx = -1;
4932 }
4933}
4934#endif
4935
Bram Moolenaar071d4272004-06-13 20:20:40 +00004936/*
4937 * ":scriptencoding": Set encoding conversion for a sourced script.
4938 * Without the multi-byte feature it's simply ignored.
4939 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004940 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004941ex_scriptencoding(exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004942{
4943#ifdef FEAT_MBYTE
4944 struct source_cookie *sp;
4945 char_u *name;
4946
4947 if (!getline_equal(eap->getline, eap->cookie, getsourceline))
4948 {
4949 EMSG(_("E167: :scriptencoding used outside of a sourced file"));
4950 return;
4951 }
4952
4953 if (*eap->arg != NUL)
4954 {
4955 name = enc_canonize(eap->arg);
4956 if (name == NULL) /* out of memory */
4957 return;
4958 }
4959 else
4960 name = eap->arg;
4961
4962 /* Setup for conversion from the specified encoding to 'encoding'. */
4963 sp = (struct source_cookie *)getline_cookie(eap->getline, eap->cookie);
4964 convert_setup(&sp->conv, name, p_enc);
4965
4966 if (name != eap->arg)
4967 vim_free(name);
4968#endif
4969}
4970
4971#if defined(FEAT_EVAL) || defined(PROTO)
4972/*
4973 * ":finish": Mark a sourced file as finished.
4974 */
4975 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004976ex_finish(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004977{
4978 if (getline_equal(eap->getline, eap->cookie, getsourceline))
4979 do_finish(eap, FALSE);
4980 else
4981 EMSG(_("E168: :finish used outside of a sourced file"));
4982}
4983
4984/*
4985 * Mark a sourced file as finished. Possibly makes the ":finish" pending.
4986 * Also called for a pending finish at the ":endtry" or after returning from
4987 * an extra do_cmdline(). "reanimate" is used in the latter case.
4988 */
4989 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004990do_finish(exarg_T *eap, int reanimate)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004991{
4992 int idx;
4993
4994 if (reanimate)
4995 ((struct source_cookie *)getline_cookie(eap->getline,
4996 eap->cookie))->finished = FALSE;
4997
4998 /*
4999 * Cleanup (and inactivate) conditionals, but stop when a try conditional
5000 * not in its finally clause (which then is to be executed next) is found.
5001 * In this case, make the ":finish" pending for execution at the ":endtry".
5002 * Otherwise, finish normally.
5003 */
5004 idx = cleanup_conditionals(eap->cstack, 0, TRUE);
5005 if (idx >= 0)
5006 {
5007 eap->cstack->cs_pending[idx] = CSTP_FINISH;
5008 report_make_pending(CSTP_FINISH, NULL);
5009 }
5010 else
5011 ((struct source_cookie *)getline_cookie(eap->getline,
5012 eap->cookie))->finished = TRUE;
5013}
5014
5015
5016/*
5017 * Return TRUE when a sourced file had the ":finish" command: Don't give error
5018 * message for missing ":endif".
5019 * Return FALSE when not sourcing a file.
5020 */
5021 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005022source_finished(
5023 char_u *(*fgetline)(int, void *, int),
5024 void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005025{
Bram Moolenaar89d40322006-08-29 15:30:07 +00005026 return (getline_equal(fgetline, cookie, getsourceline)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005027 && ((struct source_cookie *)getline_cookie(
Bram Moolenaar89d40322006-08-29 15:30:07 +00005028 fgetline, cookie))->finished);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005029}
5030#endif
5031
5032#if defined(FEAT_LISTCMDS) || defined(PROTO)
5033/*
5034 * ":checktime [buffer]"
5035 */
5036 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005037ex_checktime(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005038{
5039 buf_T *buf;
5040 int save_no_check_timestamps = no_check_timestamps;
5041
5042 no_check_timestamps = 0;
5043 if (eap->addr_count == 0) /* default is all buffers */
5044 check_timestamps(FALSE);
5045 else
5046 {
5047 buf = buflist_findnr((int)eap->line2);
5048 if (buf != NULL) /* cannot happen? */
5049 (void)buf_check_timestamp(buf, FALSE);
5050 }
5051 no_check_timestamps = save_no_check_timestamps;
5052}
5053#endif
5054
Bram Moolenaar071d4272004-06-13 20:20:40 +00005055#if (defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
5056 && (defined(FEAT_EVAL) || defined(FEAT_MULTI_LANG))
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02005057# define HAVE_GET_LOCALE_VAL
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005058static char_u *get_locale_val(int what);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005059
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005060 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005061get_locale_val(int what)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005062{
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005063 char_u *loc;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005064
Bram Moolenaar48e330a2016-02-23 14:53:34 +01005065 /* Obtain the locale value from the libraries. */
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005066 loc = (char_u *)setlocale(what, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005067
Bram Moolenaar61660ea2006-04-07 21:40:07 +00005068# ifdef WIN32
Bram Moolenaar071d4272004-06-13 20:20:40 +00005069 if (loc != NULL)
5070 {
5071 char_u *p;
5072
Bram Moolenaar61660ea2006-04-07 21:40:07 +00005073 /* setocale() returns something like "LC_COLLATE=<name>;LC_..." when
5074 * one of the values (e.g., LC_CTYPE) differs. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005075 p = vim_strchr(loc, '=');
5076 if (p != NULL)
5077 {
5078 loc = ++p;
5079 while (*p != NUL) /* remove trailing newline */
5080 {
Bram Moolenaar61660ea2006-04-07 21:40:07 +00005081 if (*p < ' ' || *p == ';')
Bram Moolenaar071d4272004-06-13 20:20:40 +00005082 {
5083 *p = NUL;
5084 break;
5085 }
5086 ++p;
5087 }
5088 }
5089 }
5090# endif
5091
5092 return loc;
5093}
5094#endif
5095
5096
5097#ifdef WIN32
5098/*
5099 * On MS-Windows locale names are strings like "German_Germany.1252", but
5100 * gettext expects "de". Try to translate one into another here for a few
5101 * supported languages.
5102 */
5103 static char_u *
5104gettext_lang(char_u *name)
5105{
5106 int i;
5107 static char *(mtable[]) = {
5108 "afrikaans", "af",
5109 "czech", "cs",
5110 "dutch", "nl",
5111 "german", "de",
5112 "english_united kingdom", "en_GB",
5113 "spanish", "es",
5114 "french", "fr",
5115 "italian", "it",
5116 "japanese", "ja",
5117 "korean", "ko",
5118 "norwegian", "no",
5119 "polish", "pl",
5120 "russian", "ru",
5121 "slovak", "sk",
5122 "swedish", "sv",
5123 "ukrainian", "uk",
5124 "chinese_china", "zh_CN",
5125 "chinese_taiwan", "zh_TW",
5126 NULL};
5127
5128 for (i = 0; mtable[i] != NULL; i += 2)
5129 if (STRNICMP(mtable[i], name, STRLEN(mtable[i])) == 0)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005130 return (char_u *)mtable[i + 1];
Bram Moolenaar071d4272004-06-13 20:20:40 +00005131 return name;
5132}
5133#endif
5134
5135#if defined(FEAT_MULTI_LANG) || defined(PROTO)
5136/*
5137 * Obtain the current messages language. Used to set the default for
5138 * 'helplang'. May return NULL or an empty string.
5139 */
5140 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005141get_mess_lang(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005142{
5143 char_u *p;
5144
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02005145# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00005146# if defined(LC_MESSAGES)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005147 p = get_locale_val(LC_MESSAGES);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005148# else
5149 /* This is necessary for Win32, where LC_MESSAGES is not defined and $LANG
Bram Moolenaar61660ea2006-04-07 21:40:07 +00005150 * may be set to the LCID number. LC_COLLATE is the best guess, LC_TIME
5151 * and LC_MONETARY may be set differently for a Japanese working in the
5152 * US. */
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005153 p = get_locale_val(LC_COLLATE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005154# endif
5155# else
5156 p = mch_getenv((char_u *)"LC_ALL");
5157 if (p == NULL || *p == NUL)
5158 {
5159 p = mch_getenv((char_u *)"LC_MESSAGES");
5160 if (p == NULL || *p == NUL)
5161 p = mch_getenv((char_u *)"LANG");
5162 }
5163# endif
5164# ifdef WIN32
5165 p = gettext_lang(p);
5166# endif
5167 return p;
5168}
5169#endif
5170
Bram Moolenaardef9e822004-12-31 20:58:58 +00005171/* Complicated #if; matches with where get_mess_env() is used below. */
5172#if (defined(FEAT_EVAL) && !((defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
5173 && defined(LC_MESSAGES))) \
5174 || ((defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
5175 && (defined(FEAT_GETTEXT) || defined(FEAT_MBYTE)) \
5176 && !defined(LC_MESSAGES))
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01005177static char_u *get_mess_env(void);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005178
5179/*
5180 * Get the language used for messages from the environment.
5181 */
5182 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005183get_mess_env(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005184{
5185 char_u *p;
5186
5187 p = mch_getenv((char_u *)"LC_ALL");
5188 if (p == NULL || *p == NUL)
5189 {
5190 p = mch_getenv((char_u *)"LC_MESSAGES");
5191 if (p == NULL || *p == NUL)
5192 {
5193 p = mch_getenv((char_u *)"LANG");
5194 if (p != NULL && VIM_ISDIGIT(*p))
5195 p = NULL; /* ignore something like "1043" */
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02005196# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00005197 if (p == NULL || *p == NUL)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005198 p = get_locale_val(LC_CTYPE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005199# endif
5200 }
5201 }
5202 return p;
5203}
5204#endif
5205
5206#if defined(FEAT_EVAL) || defined(PROTO)
5207
5208/*
5209 * Set the "v:lang" variable according to the current locale setting.
5210 * Also do "v:lc_time"and "v:ctype".
5211 */
5212 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005213set_lang_var(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005214{
5215 char_u *loc;
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_CTYPE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005219# else
5220 /* setlocale() not supported: use the default value */
5221 loc = (char_u *)"C";
5222# endif
5223 set_vim_var_string(VV_CTYPE, loc, -1);
5224
5225 /* When LC_MESSAGES isn't defined use the value from $LC_MESSAGES, fall
5226 * back to LC_CTYPE if it's empty. */
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02005227# if defined(HAVE_GET_LOCALE_VAL) && defined(LC_MESSAGES)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005228 loc = get_locale_val(LC_MESSAGES);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005229# else
5230 loc = get_mess_env();
5231# endif
5232 set_vim_var_string(VV_LANG, loc, -1);
5233
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02005234# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005235 loc = get_locale_val(LC_TIME);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005236# endif
5237 set_vim_var_string(VV_LC_TIME, loc, -1);
5238}
5239#endif
5240
5241#if (defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
5242 && (defined(FEAT_GETTEXT) || defined(FEAT_MBYTE))
5243/*
5244 * ":language": Set the language (locale).
5245 */
5246 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005247ex_language(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005248{
5249 char *loc;
5250 char_u *p;
5251 char_u *name;
5252 int what = LC_ALL;
5253 char *whatstr = "";
5254#ifdef LC_MESSAGES
5255# define VIM_LC_MESSAGES LC_MESSAGES
5256#else
5257# define VIM_LC_MESSAGES 6789
5258#endif
5259
5260 name = eap->arg;
5261
5262 /* Check for "messages {name}", "ctype {name}" or "time {name}" argument.
5263 * Allow abbreviation, but require at least 3 characters to avoid
5264 * confusion with a two letter language name "me" or "ct". */
5265 p = skiptowhite(eap->arg);
Bram Moolenaar1c465442017-03-12 20:10:05 +01005266 if ((*p == NUL || VIM_ISWHITE(*p)) && p - eap->arg >= 3)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005267 {
5268 if (STRNICMP(eap->arg, "messages", p - eap->arg) == 0)
5269 {
5270 what = VIM_LC_MESSAGES;
5271 name = skipwhite(p);
5272 whatstr = "messages ";
5273 }
5274 else if (STRNICMP(eap->arg, "ctype", p - eap->arg) == 0)
5275 {
5276 what = LC_CTYPE;
5277 name = skipwhite(p);
5278 whatstr = "ctype ";
5279 }
5280 else if (STRNICMP(eap->arg, "time", p - eap->arg) == 0)
5281 {
5282 what = LC_TIME;
5283 name = skipwhite(p);
5284 whatstr = "time ";
5285 }
5286 }
5287
5288 if (*name == NUL)
5289 {
5290#ifndef LC_MESSAGES
5291 if (what == VIM_LC_MESSAGES)
5292 p = get_mess_env();
5293 else
5294#endif
5295 p = (char_u *)setlocale(what, NULL);
5296 if (p == NULL || *p == NUL)
5297 p = (char_u *)"Unknown";
5298 smsg((char_u *)_("Current %slanguage: \"%s\""), whatstr, p);
5299 }
5300 else
5301 {
5302#ifndef LC_MESSAGES
5303 if (what == VIM_LC_MESSAGES)
5304 loc = "";
5305 else
5306#endif
Bram Moolenaar8c8de832008-06-24 22:58:06 +00005307 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00005308 loc = setlocale(what, (char *)name);
Bram Moolenaar8c8de832008-06-24 22:58:06 +00005309#if defined(FEAT_FLOAT) && defined(LC_NUMERIC)
5310 /* Make sure strtod() uses a decimal point, not a comma. */
5311 setlocale(LC_NUMERIC, "C");
5312#endif
5313 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00005314 if (loc == NULL)
5315 EMSG2(_("E197: Cannot set language to \"%s\""), name);
5316 else
5317 {
5318#ifdef HAVE_NL_MSG_CAT_CNTR
5319 /* Need to do this for GNU gettext, otherwise cached translations
5320 * will be used again. */
5321 extern int _nl_msg_cat_cntr;
5322
5323 ++_nl_msg_cat_cntr;
5324#endif
Bram Moolenaarb8017e72007-05-10 18:59:07 +00005325 /* Reset $LC_ALL, otherwise it would overrule everything. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005326 vim_setenv((char_u *)"LC_ALL", (char_u *)"");
5327
5328 if (what != LC_TIME)
5329 {
5330 /* Tell gettext() what to translate to. It apparently doesn't
5331 * use the currently effective locale. Also do this when
5332 * FEAT_GETTEXT isn't defined, so that shell commands use this
5333 * value. */
5334 if (what == LC_ALL)
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00005335 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00005336 vim_setenv((char_u *)"LANG", name);
Bram Moolenaare3a0b532013-06-28 20:36:30 +02005337
5338 /* Clear $LANGUAGE because GNU gettext uses it. */
5339 vim_setenv((char_u *)"LANGUAGE", (char_u *)"");
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00005340# ifdef WIN32
5341 /* Apparently MS-Windows printf() may cause a crash when
5342 * we give it 8-bit text while it's expecting text in the
5343 * current locale. This call avoids that. */
5344 setlocale(LC_CTYPE, "C");
5345# endif
5346 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00005347 if (what != LC_CTYPE)
5348 {
5349 char_u *mname;
5350#ifdef WIN32
5351 mname = gettext_lang(name);
5352#else
5353 mname = name;
5354#endif
5355 vim_setenv((char_u *)"LC_MESSAGES", mname);
5356#ifdef FEAT_MULTI_LANG
5357 set_helplang_default(mname);
5358#endif
5359 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00005360 }
5361
5362# ifdef FEAT_EVAL
5363 /* Set v:lang, v:lc_time and v:ctype to the final result. */
5364 set_lang_var();
5365# endif
Bram Moolenaar6cc00c72011-10-20 21:41:09 +02005366# ifdef FEAT_TITLE
5367 maketitle();
5368# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00005369 }
5370 }
5371}
5372
5373# if defined(FEAT_CMDL_COMPL) || defined(PROTO)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005374
5375static char_u **locales = NULL; /* Array of all available locales */
Bram Moolenaarb8f7bd62017-01-12 20:28:25 +01005376
5377# ifndef WIN32
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005378static int did_init_locales = FALSE;
5379
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005380/* Return an array of strings for all available locales + NULL for the
5381 * last element. Return NULL in case of error. */
5382 static char_u **
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005383find_locales(void)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005384{
5385 garray_T locales_ga;
5386 char_u *loc;
5387
5388 /* Find all available locales by running command "locale -a". If this
5389 * doesn't work we won't have completion. */
5390 char_u *locale_a = get_cmd_output((char_u *)"locale -a",
Bram Moolenaar39c29ed2014-04-05 19:44:40 +02005391 NULL, SHELL_SILENT, NULL);
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005392 if (locale_a == NULL)
5393 return NULL;
5394 ga_init2(&locales_ga, sizeof(char_u *), 20);
5395
5396 /* Transform locale_a string where each locale is separated by "\n"
5397 * into an array of locale strings. */
5398 loc = (char_u *)strtok((char *)locale_a, "\n");
5399
5400 while (loc != NULL)
5401 {
5402 if (ga_grow(&locales_ga, 1) == FAIL)
5403 break;
5404 loc = vim_strsave(loc);
5405 if (loc == NULL)
5406 break;
5407
5408 ((char_u **)locales_ga.ga_data)[locales_ga.ga_len++] = loc;
5409 loc = (char_u *)strtok(NULL, "\n");
5410 }
5411 vim_free(locale_a);
5412 if (ga_grow(&locales_ga, 1) == FAIL)
5413 {
5414 ga_clear(&locales_ga);
5415 return NULL;
5416 }
5417 ((char_u **)locales_ga.ga_data)[locales_ga.ga_len] = NULL;
5418 return (char_u **)locales_ga.ga_data;
5419}
Bram Moolenaarb8f7bd62017-01-12 20:28:25 +01005420# endif
5421
5422/*
5423 * Lazy initialization of all available locales.
5424 */
5425 static void
5426init_locales(void)
5427{
5428# ifndef WIN32
5429 if (!did_init_locales)
5430 {
5431 did_init_locales = TRUE;
5432 locales = find_locales();
5433 }
5434# endif
5435}
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005436
5437# if defined(EXITFREE) || defined(PROTO)
5438 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005439free_locales(void)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005440{
5441 int i;
5442 if (locales != NULL)
5443 {
5444 for (i = 0; locales[i] != NULL; i++)
5445 vim_free(locales[i]);
5446 vim_free(locales);
5447 locales = NULL;
5448 }
5449}
5450# endif
5451
Bram Moolenaar071d4272004-06-13 20:20:40 +00005452/*
5453 * Function given to ExpandGeneric() to obtain the possible arguments of the
5454 * ":language" command.
5455 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005456 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005457get_lang_arg(expand_T *xp UNUSED, int idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005458{
5459 if (idx == 0)
5460 return (char_u *)"messages";
5461 if (idx == 1)
5462 return (char_u *)"ctype";
5463 if (idx == 2)
5464 return (char_u *)"time";
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005465
5466 init_locales();
5467 if (locales == NULL)
5468 return NULL;
5469 return locales[idx - 3];
5470}
5471
5472/*
5473 * Function given to ExpandGeneric() to obtain the available locales.
5474 */
5475 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005476get_locales(expand_T *xp UNUSED, int idx)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005477{
5478 init_locales();
5479 if (locales == NULL)
5480 return NULL;
5481 return locales[idx];
Bram Moolenaar071d4272004-06-13 20:20:40 +00005482}
5483# endif
5484
5485#endif