blob: 73b3a522fd680faeaddacaaac156a4e825b79098 [file] [log] [blame]
Bram Moolenaar071d4272004-06-13 20:20:40 +00001/* vi:set ts=8 sts=4 sw=4:
2 *
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;
1091static int last_timer_id = 0;
1092
1093/*
1094 * Insert a timer in the list of timers.
1095 */
1096 static void
1097insert_timer(timer_T *timer)
1098{
1099 timer->tr_next = first_timer;
1100 timer->tr_prev = NULL;
1101 if (first_timer != NULL)
1102 first_timer->tr_prev = timer;
1103 first_timer = timer;
Bram Moolenaar4231da42016-06-02 14:30:04 +02001104 did_add_timer = TRUE;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001105}
1106
1107/*
1108 * Take a timer out of the list of timers.
1109 */
1110 static void
1111remove_timer(timer_T *timer)
1112{
1113 if (timer->tr_prev == NULL)
1114 first_timer = timer->tr_next;
1115 else
1116 timer->tr_prev->tr_next = timer->tr_next;
1117 if (timer->tr_next != NULL)
1118 timer->tr_next->tr_prev = timer->tr_prev;
1119}
1120
1121 static void
1122free_timer(timer_T *timer)
1123{
Bram Moolenaar1436d8d2016-07-11 22:41:15 +02001124 free_callback(timer->tr_callback, timer->tr_partial);
Bram Moolenaar975b5272016-03-15 23:10:59 +01001125 vim_free(timer);
1126}
1127
1128/*
1129 * Create a timer and return it. NULL if out of memory.
1130 * Caller should set the callback.
1131 */
1132 timer_T *
1133create_timer(long msec, int repeat)
1134{
1135 timer_T *timer = (timer_T *)alloc_clear(sizeof(timer_T));
1136
1137 if (timer == NULL)
1138 return NULL;
1139 timer->tr_id = ++last_timer_id;
1140 insert_timer(timer);
1141 if (repeat != 0)
Bram Moolenaar975b5272016-03-15 23:10:59 +01001142 timer->tr_repeat = repeat - 1;
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001143 timer->tr_interval = msec;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001144
1145 profile_setlimit(msec, &timer->tr_due);
1146 return timer;
1147}
1148
1149/*
1150 * Invoke the callback of "timer".
1151 */
1152 static void
1153timer_callback(timer_T *timer)
1154{
1155 typval_T rettv;
1156 int dummy;
1157 typval_T argv[2];
1158
1159 argv[0].v_type = VAR_NUMBER;
1160 argv[0].vval.v_number = timer->tr_id;
1161 argv[1].v_type = VAR_UNKNOWN;
1162
1163 call_func(timer->tr_callback, (int)STRLEN(timer->tr_callback),
Bram Moolenaardf48fb42016-07-22 21:50:18 +02001164 &rettv, 1, argv, NULL, 0L, 0L, &dummy, TRUE,
Bram Moolenaar975b5272016-03-15 23:10:59 +01001165 timer->tr_partial, NULL);
1166 clear_tv(&rettv);
1167}
1168
1169/*
1170 * Call timers that are due.
1171 * Return the time in msec until the next timer is due.
1172 */
1173 long
Bram Moolenaarcf089462016-06-12 21:18:43 +02001174check_due_timer(void)
Bram Moolenaar975b5272016-03-15 23:10:59 +01001175{
1176 timer_T *timer;
1177 long this_due;
Bram Moolenaar597385a2016-03-16 23:24:43 +01001178 long next_due = -1;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001179 proftime_T now;
1180 int did_one = FALSE;
1181# ifdef WIN3264
1182 LARGE_INTEGER fr;
1183
1184 QueryPerformanceFrequency(&fr);
1185# endif
1186 while (!got_int)
1187 {
1188 profile_start(&now);
1189 next_due = -1;
1190 for (timer = first_timer; timer != NULL; timer = timer->tr_next)
1191 {
Bram Moolenaarb73598e2016-08-07 18:22:53 +02001192 if (timer->tr_paused)
1193 continue;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001194# ifdef WIN3264
1195 this_due = (long)(((double)(timer->tr_due.QuadPart - now.QuadPart)
1196 / (double)fr.QuadPart) * 1000);
1197# else
1198 this_due = (timer->tr_due.tv_sec - now.tv_sec) * 1000
1199 + (timer->tr_due.tv_usec - now.tv_usec) / 1000;
1200# endif
1201 if (this_due <= 1)
1202 {
1203 remove_timer(timer);
1204 timer_callback(timer);
1205 did_one = TRUE;
1206 if (timer->tr_repeat != 0)
1207 {
1208 profile_setlimit(timer->tr_interval, &timer->tr_due);
1209 if (timer->tr_repeat > 0)
1210 --timer->tr_repeat;
1211 insert_timer(timer);
1212 }
1213 else
1214 free_timer(timer);
1215 /* the callback may do anything, start all over */
1216 break;
1217 }
1218 if (next_due == -1 || next_due > this_due)
1219 next_due = this_due;
1220 }
1221 if (timer == NULL)
1222 break;
1223 }
1224
1225 if (did_one)
1226 redraw_after_callback();
1227
1228 return next_due;
1229}
1230
1231/*
1232 * Find a timer by ID. Returns NULL if not found;
1233 */
1234 timer_T *
1235find_timer(int id)
1236{
1237 timer_T *timer;
1238
1239 for (timer = first_timer; timer != NULL; timer = timer->tr_next)
1240 if (timer->tr_id == id)
1241 break;
1242 return timer;
1243}
1244
1245
1246/*
1247 * Stop a timer and delete it.
1248 */
1249 void
1250stop_timer(timer_T *timer)
1251{
1252 remove_timer(timer);
1253 free_timer(timer);
1254}
Bram Moolenaare3188e22016-05-31 21:13:04 +02001255
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001256 void
Bram Moolenaarb73598e2016-08-07 18:22:53 +02001257stop_all_timers(void)
1258{
1259 timer_T *timer;
1260
1261 while (first_timer != NULL)
1262 stop_timer(first_timer);
1263}
1264
1265 void
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001266add_timer_info(typval_T *rettv, timer_T *timer)
1267{
1268 list_T *list = rettv->vval.v_list;
1269 dict_T *dict = dict_alloc();
1270 dictitem_T *di;
1271 long remaining;
1272 proftime_T now;
Bram Moolenaar00ff3802016-08-06 22:27:28 +02001273# ifdef WIN3264
1274 LARGE_INTEGER fr;
1275#endif
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001276
1277 if (dict == NULL)
1278 return;
1279 list_append_dict(list, dict);
1280
1281 dict_add_nr_str(dict, "id", (long)timer->tr_id, NULL);
1282 dict_add_nr_str(dict, "time", (long)timer->tr_interval, NULL);
1283
1284 profile_start(&now);
1285# ifdef WIN3264
Bram Moolenaar00ff3802016-08-06 22:27:28 +02001286 QueryPerformanceFrequency(&fr);
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001287 remaining = (long)(((double)(timer->tr_due.QuadPart - now.QuadPart)
1288 / (double)fr.QuadPart) * 1000);
1289# else
1290 remaining = (timer->tr_due.tv_sec - now.tv_sec) * 1000
1291 + (timer->tr_due.tv_usec - now.tv_usec) / 1000;
1292# endif
1293 dict_add_nr_str(dict, "remaining", (long)remaining, NULL);
1294
1295 dict_add_nr_str(dict, "repeat",
1296 (long)(timer->tr_repeat < 0 ? -1 : timer->tr_repeat + 1), NULL);
Bram Moolenaarb73598e2016-08-07 18:22:53 +02001297 dict_add_nr_str(dict, "paused", (long)(timer->tr_paused), NULL);
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001298
1299 di = dictitem_alloc((char_u *)"callback");
1300 if (di != NULL)
1301 {
1302 if (dict_add(dict, di) == FAIL)
1303 vim_free(di);
1304 else if (timer->tr_partial != NULL)
1305 {
1306 di->di_tv.v_type = VAR_PARTIAL;
1307 di->di_tv.vval.v_partial = timer->tr_partial;
1308 ++timer->tr_partial->pt_refcount;
1309 }
1310 else
1311 {
1312 di->di_tv.v_type = VAR_FUNC;
1313 di->di_tv.vval.v_string = vim_strsave(timer->tr_callback);
1314 }
1315 di->di_tv.v_lock = 0;
1316 }
1317}
1318
1319 void
1320add_timer_info_all(typval_T *rettv)
1321{
1322 timer_T *timer;
1323
1324 for (timer = first_timer; timer != NULL; timer = timer->tr_next)
1325 add_timer_info(rettv, timer);
1326}
1327
Bram Moolenaare3188e22016-05-31 21:13:04 +02001328/*
1329 * Mark references in partials of timers.
1330 */
1331 int
1332set_ref_in_timer(int copyID)
1333{
1334 int abort = FALSE;
1335 timer_T *timer;
1336 typval_T tv;
1337
1338 for (timer = first_timer; timer != NULL; timer = timer->tr_next)
1339 {
Bram Moolenaar1e96d9b2016-07-29 22:15:09 +02001340 if (timer->tr_partial != NULL)
1341 {
1342 tv.v_type = VAR_PARTIAL;
1343 tv.vval.v_partial = timer->tr_partial;
1344 }
1345 else
1346 {
1347 tv.v_type = VAR_FUNC;
1348 tv.vval.v_string = timer->tr_callback;
1349 }
Bram Moolenaare3188e22016-05-31 21:13:04 +02001350 abort = abort || set_ref_in_item(&tv, copyID, NULL, NULL);
1351 }
1352 return abort;
1353}
Bram Moolenaar623e2632016-07-30 22:47:56 +02001354
1355# if defined(EXITFREE) || defined(PROTO)
1356 void
1357timer_free_all()
1358{
1359 timer_T *timer;
1360
1361 while (first_timer != NULL)
1362 {
1363 timer = first_timer;
1364 remove_timer(timer);
1365 free_timer(timer);
1366 }
1367}
1368# endif
Bram Moolenaar975b5272016-03-15 23:10:59 +01001369# endif
1370
Bram Moolenaar8a7f5a22013-06-06 14:01:46 +02001371#if defined(FEAT_SYN_HL) && defined(FEAT_RELTIME) && defined(FEAT_FLOAT)
1372# if defined(HAVE_MATH_H)
1373# include <math.h>
1374# endif
1375
1376/*
1377 * Divide the time "tm" by "count" and store in "tm2".
1378 */
1379 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001380profile_divide(proftime_T *tm, int count, proftime_T *tm2)
Bram Moolenaar8a7f5a22013-06-06 14:01:46 +02001381{
1382 if (count == 0)
1383 profile_zero(tm2);
1384 else
1385 {
1386# ifdef WIN3264
1387 tm2->QuadPart = tm->QuadPart / count;
1388# else
1389 double usec = (tm->tv_sec * 1000000.0 + tm->tv_usec) / count;
1390
1391 tm2->tv_sec = floor(usec / 1000000.0);
Bram Moolenaara2e14fc2013-06-10 20:10:44 +02001392 tm2->tv_usec = vim_round(usec - (tm2->tv_sec * 1000000.0));
Bram Moolenaar8a7f5a22013-06-06 14:01:46 +02001393# endif
1394 }
1395}
1396#endif
1397
Bram Moolenaar76929292008-01-06 19:07:36 +00001398# if defined(FEAT_PROFILE) || defined(PROTO)
1399/*
1400 * Functions for profiling.
1401 */
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01001402static void script_do_profile(scriptitem_T *si);
1403static void script_dump_profile(FILE *fd);
Bram Moolenaar76929292008-01-06 19:07:36 +00001404static proftime_T prof_wait_time;
1405
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001406/*
Bram Moolenaar05159a02005-02-26 23:04:13 +00001407 * Add the time "tm2" to "tm".
1408 */
1409 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001410profile_add(proftime_T *tm, proftime_T *tm2)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001411{
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001412# ifdef WIN3264
1413 tm->QuadPart += tm2->QuadPart;
1414# else
Bram Moolenaar05159a02005-02-26 23:04:13 +00001415 tm->tv_usec += tm2->tv_usec;
1416 tm->tv_sec += tm2->tv_sec;
1417 if (tm->tv_usec >= 1000000)
1418 {
1419 tm->tv_usec -= 1000000;
1420 ++tm->tv_sec;
1421 }
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001422# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00001423}
1424
1425/*
Bram Moolenaar1056d982006-03-09 22:37:52 +00001426 * Add the "self" time from the total time and the children's time.
1427 */
1428 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001429profile_self(proftime_T *self, proftime_T *total, proftime_T *children)
Bram Moolenaar1056d982006-03-09 22:37:52 +00001430{
1431 /* Check that the result won't be negative. Can happen with recursive
1432 * calls. */
1433#ifdef WIN3264
1434 if (total->QuadPart <= children->QuadPart)
1435 return;
1436#else
1437 if (total->tv_sec < children->tv_sec
1438 || (total->tv_sec == children->tv_sec
1439 && total->tv_usec <= children->tv_usec))
1440 return;
1441#endif
1442 profile_add(self, total);
1443 profile_sub(self, children);
1444}
1445
1446/*
Bram Moolenaar05159a02005-02-26 23:04:13 +00001447 * Get the current waittime.
1448 */
1449 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001450profile_get_wait(proftime_T *tm)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001451{
1452 *tm = prof_wait_time;
1453}
1454
1455/*
1456 * Subtract the passed waittime since "tm" from "tma".
1457 */
1458 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001459profile_sub_wait(proftime_T *tm, proftime_T *tma)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001460{
1461 proftime_T tm3 = prof_wait_time;
1462
1463 profile_sub(&tm3, tm);
1464 profile_sub(tma, &tm3);
1465}
1466
1467/*
1468 * Return TRUE if "tm1" and "tm2" are equal.
1469 */
1470 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001471profile_equal(proftime_T *tm1, proftime_T *tm2)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001472{
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001473# ifdef WIN3264
1474 return (tm1->QuadPart == tm2->QuadPart);
1475# else
Bram Moolenaar05159a02005-02-26 23:04:13 +00001476 return (tm1->tv_usec == tm2->tv_usec && tm1->tv_sec == tm2->tv_sec);
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001477# endif
1478}
1479
1480/*
1481 * Return <0, 0 or >0 if "tm1" < "tm2", "tm1" == "tm2" or "tm1" > "tm2"
1482 */
1483 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001484profile_cmp(const proftime_T *tm1, const proftime_T *tm2)
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001485{
1486# ifdef WIN3264
1487 return (int)(tm2->QuadPart - tm1->QuadPart);
1488# else
1489 if (tm1->tv_sec == tm2->tv_sec)
1490 return tm2->tv_usec - tm1->tv_usec;
1491 return tm2->tv_sec - tm1->tv_sec;
1492# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00001493}
1494
Bram Moolenaar05159a02005-02-26 23:04:13 +00001495static char_u *profile_fname = NULL;
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001496static proftime_T pause_time;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001497
1498/*
1499 * ":profile cmd args"
1500 */
1501 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001502ex_profile(exarg_T *eap)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001503{
1504 char_u *e;
1505 int len;
1506
1507 e = skiptowhite(eap->arg);
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00001508 len = (int)(e - eap->arg);
Bram Moolenaar05159a02005-02-26 23:04:13 +00001509 e = skipwhite(e);
1510
1511 if (len == 5 && STRNCMP(eap->arg, "start", 5) == 0 && *e != NUL)
1512 {
1513 vim_free(profile_fname);
Bram Moolenaard94682f2015-04-13 15:37:56 +02001514 profile_fname = expand_env_save_opt(e, TRUE);
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001515 do_profiling = PROF_YES;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001516 profile_zero(&prof_wait_time);
1517 set_vim_var_nr(VV_PROFILING, 1L);
1518 }
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001519 else if (do_profiling == PROF_NONE)
Bram Moolenaar54c1b492010-02-24 14:01:28 +01001520 EMSG(_("E750: First use \":profile start {fname}\""));
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001521 else if (STRCMP(eap->arg, "pause") == 0)
1522 {
1523 if (do_profiling == PROF_YES)
1524 profile_start(&pause_time);
1525 do_profiling = PROF_PAUSED;
1526 }
1527 else if (STRCMP(eap->arg, "continue") == 0)
1528 {
1529 if (do_profiling == PROF_PAUSED)
1530 {
1531 profile_end(&pause_time);
1532 profile_add(&prof_wait_time, &pause_time);
1533 }
1534 do_profiling = PROF_YES;
1535 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00001536 else
1537 {
1538 /* The rest is similar to ":breakadd". */
1539 ex_breakadd(eap);
1540 }
1541}
1542
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001543/* Command line expansion for :profile. */
1544static enum
1545{
1546 PEXP_SUBCMD, /* expand :profile sub-commands */
Bram Moolenaar49789dc2011-02-25 14:46:09 +01001547 PEXP_FUNC /* expand :profile func {funcname} */
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001548} pexpand_what;
1549
1550static char *pexpand_cmds[] = {
1551 "start",
1552#define PROFCMD_START 0
1553 "pause",
1554#define PROFCMD_PAUSE 1
1555 "continue",
1556#define PROFCMD_CONTINUE 2
1557 "func",
1558#define PROFCMD_FUNC 3
1559 "file",
1560#define PROFCMD_FILE 4
1561 NULL
1562#define PROFCMD_LAST 5
1563};
1564
1565/*
1566 * Function given to ExpandGeneric() to obtain the profile command
1567 * specific expansion.
1568 */
1569 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001570get_profile_name(expand_T *xp UNUSED, int idx)
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001571{
1572 switch (pexpand_what)
1573 {
1574 case PEXP_SUBCMD:
1575 return (char_u *)pexpand_cmds[idx];
1576 /* case PEXP_FUNC: TODO */
1577 default:
1578 return NULL;
1579 }
1580}
1581
1582/*
1583 * Handle command line completion for :profile command.
1584 */
1585 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001586set_context_in_profile_cmd(expand_T *xp, char_u *arg)
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001587{
1588 char_u *end_subcmd;
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001589
1590 /* Default: expand subcommands. */
1591 xp->xp_context = EXPAND_PROFILE;
1592 pexpand_what = PEXP_SUBCMD;
1593 xp->xp_pattern = arg;
1594
1595 end_subcmd = skiptowhite(arg);
1596 if (*end_subcmd == NUL)
1597 return;
1598
Bram Moolenaar8b9c05f2010-03-02 17:54:33 +01001599 if (end_subcmd - arg == 5 && STRNCMP(arg, "start", 5) == 0)
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001600 {
1601 xp->xp_context = EXPAND_FILES;
1602 xp->xp_pattern = skipwhite(end_subcmd);
1603 return;
1604 }
1605
1606 /* TODO: expand function names after "func" */
1607 xp->xp_context = EXPAND_NOTHING;
1608}
1609
Bram Moolenaar05159a02005-02-26 23:04:13 +00001610/*
1611 * Dump the profiling info.
1612 */
1613 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001614profile_dump(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001615{
1616 FILE *fd;
1617
1618 if (profile_fname != NULL)
1619 {
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +00001620 fd = mch_fopen((char *)profile_fname, "w");
Bram Moolenaar05159a02005-02-26 23:04:13 +00001621 if (fd == NULL)
1622 EMSG2(_(e_notopen), profile_fname);
1623 else
1624 {
Bram Moolenaar05159a02005-02-26 23:04:13 +00001625 script_dump_profile(fd);
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001626 func_dump_profile(fd);
Bram Moolenaar05159a02005-02-26 23:04:13 +00001627 fclose(fd);
1628 }
1629 }
1630}
1631
1632/*
1633 * Start profiling script "fp".
1634 */
1635 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001636script_do_profile(scriptitem_T *si)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001637{
1638 si->sn_pr_count = 0;
1639 profile_zero(&si->sn_pr_total);
1640 profile_zero(&si->sn_pr_self);
1641
1642 ga_init2(&si->sn_prl_ga, sizeof(sn_prl_T), 100);
1643 si->sn_prl_idx = -1;
1644 si->sn_prof_on = TRUE;
1645 si->sn_pr_nest = 0;
1646}
1647
1648/*
1649 * save time when starting to invoke another script or function.
1650 */
1651 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001652script_prof_save(
1653 proftime_T *tm) /* place to store wait time */
Bram Moolenaar05159a02005-02-26 23:04:13 +00001654{
1655 scriptitem_T *si;
1656
1657 if (current_SID > 0 && current_SID <= script_items.ga_len)
1658 {
1659 si = &SCRIPT_ITEM(current_SID);
1660 if (si->sn_prof_on && si->sn_pr_nest++ == 0)
1661 profile_start(&si->sn_pr_child);
1662 }
1663 profile_get_wait(tm);
1664}
1665
1666/*
1667 * Count time spent in children after invoking another script or function.
1668 */
1669 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001670script_prof_restore(proftime_T *tm)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001671{
1672 scriptitem_T *si;
1673
1674 if (current_SID > 0 && current_SID <= script_items.ga_len)
1675 {
1676 si = &SCRIPT_ITEM(current_SID);
1677 if (si->sn_prof_on && --si->sn_pr_nest == 0)
1678 {
1679 profile_end(&si->sn_pr_child);
1680 profile_sub_wait(tm, &si->sn_pr_child); /* don't count wait time */
1681 profile_add(&si->sn_pr_children, &si->sn_pr_child);
1682 profile_add(&si->sn_prl_children, &si->sn_pr_child);
1683 }
1684 }
1685}
1686
1687static proftime_T inchar_time;
1688
1689/*
1690 * Called when starting to wait for the user to type a character.
1691 */
1692 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001693prof_inchar_enter(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001694{
1695 profile_start(&inchar_time);
1696}
1697
1698/*
1699 * Called when finished waiting for the user to type a character.
1700 */
1701 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001702prof_inchar_exit(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001703{
1704 profile_end(&inchar_time);
1705 profile_add(&prof_wait_time, &inchar_time);
1706}
1707
1708/*
1709 * Dump the profiling results for all scripts in file "fd".
1710 */
1711 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001712script_dump_profile(FILE *fd)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001713{
1714 int id;
1715 scriptitem_T *si;
1716 int i;
1717 FILE *sfd;
1718 sn_prl_T *pp;
1719
1720 for (id = 1; id <= script_items.ga_len; ++id)
1721 {
1722 si = &SCRIPT_ITEM(id);
1723 if (si->sn_prof_on)
1724 {
1725 fprintf(fd, "SCRIPT %s\n", si->sn_name);
1726 if (si->sn_pr_count == 1)
1727 fprintf(fd, "Sourced 1 time\n");
1728 else
1729 fprintf(fd, "Sourced %d times\n", si->sn_pr_count);
1730 fprintf(fd, "Total time: %s\n", profile_msg(&si->sn_pr_total));
1731 fprintf(fd, " Self time: %s\n", profile_msg(&si->sn_pr_self));
1732 fprintf(fd, "\n");
1733 fprintf(fd, "count total (s) self (s)\n");
1734
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +00001735 sfd = mch_fopen((char *)si->sn_name, "r");
Bram Moolenaar05159a02005-02-26 23:04:13 +00001736 if (sfd == NULL)
1737 fprintf(fd, "Cannot open file!\n");
1738 else
1739 {
1740 for (i = 0; i < si->sn_prl_ga.ga_len; ++i)
1741 {
1742 if (vim_fgets(IObuff, IOSIZE, sfd))
1743 break;
1744 pp = &PRL_ITEM(si, i);
1745 if (pp->snp_count > 0)
1746 {
1747 fprintf(fd, "%5d ", pp->snp_count);
1748 if (profile_equal(&pp->sn_prl_total, &pp->sn_prl_self))
1749 fprintf(fd, " ");
1750 else
1751 fprintf(fd, "%s ", profile_msg(&pp->sn_prl_total));
1752 fprintf(fd, "%s ", profile_msg(&pp->sn_prl_self));
1753 }
1754 else
1755 fprintf(fd, " ");
1756 fprintf(fd, "%s", IObuff);
1757 }
1758 fclose(sfd);
1759 }
1760 fprintf(fd, "\n");
1761 }
1762 }
1763}
1764
1765/*
1766 * Return TRUE when a function defined in the current script should be
1767 * profiled.
1768 */
1769 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001770prof_def_func(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001771{
Bram Moolenaar53180ce2005-07-05 21:48:14 +00001772 if (current_SID > 0)
1773 return SCRIPT_ITEM(current_SID).sn_pr_force;
1774 return FALSE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001775}
1776
1777# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001778#endif
1779
1780/*
1781 * If 'autowrite' option set, try to write the file.
1782 * Careful: autocommands may make "buf" invalid!
1783 *
1784 * return FAIL for failure, OK otherwise
1785 */
1786 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001787autowrite(buf_T *buf, int forceit)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001788{
Bram Moolenaar373154b2007-02-13 05:19:30 +00001789 int r;
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001790 bufref_T bufref;
Bram Moolenaar373154b2007-02-13 05:19:30 +00001791
Bram Moolenaar071d4272004-06-13 20:20:40 +00001792 if (!(p_aw || p_awa) || !p_write
1793#ifdef FEAT_QUICKFIX
Bram Moolenaar373154b2007-02-13 05:19:30 +00001794 /* never autowrite a "nofile" or "nowrite" buffer */
1795 || bt_dontwrite(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001796#endif
Bram Moolenaar373154b2007-02-13 05:19:30 +00001797 || (!forceit && buf->b_p_ro) || buf->b_ffname == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001798 return FAIL;
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001799 set_bufref(&bufref, buf);
Bram Moolenaar373154b2007-02-13 05:19:30 +00001800 r = buf_write_all(buf, forceit);
1801
1802 /* Writing may succeed but the buffer still changed, e.g., when there is a
1803 * conversion error. We do want to return FAIL then. */
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001804 if (bufref_valid(&bufref) && bufIsChanged(buf))
Bram Moolenaar373154b2007-02-13 05:19:30 +00001805 r = FAIL;
1806 return r;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001807}
1808
1809/*
1810 * flush all buffers, except the ones that are readonly
1811 */
1812 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001813autowrite_all(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001814{
1815 buf_T *buf;
1816
1817 if (!(p_aw || p_awa) || !p_write)
1818 return;
Bram Moolenaar29323592016-07-24 22:04:11 +02001819 FOR_ALL_BUFFERS(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001820 if (bufIsChanged(buf) && !buf->b_p_ro)
1821 {
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001822#ifdef FEAT_AUTOCMD
1823 bufref_T bufref;
1824
1825 set_bufref(&bufref, buf);
1826#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001827 (void)buf_write_all(buf, FALSE);
1828#ifdef FEAT_AUTOCMD
1829 /* an autocommand may have deleted the buffer */
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001830 if (!bufref_valid(&bufref))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001831 buf = firstbuf;
1832#endif
1833 }
1834}
1835
1836/*
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001837 * Return TRUE if buffer was changed and cannot be abandoned.
1838 * For flags use the CCGD_ values.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001839 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001840 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001841check_changed(buf_T *buf, int flags)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001842{
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001843 int forceit = (flags & CCGD_FORCEIT);
1844#ifdef FEAT_AUTOCMD
1845 bufref_T bufref;
1846
1847 set_bufref(&bufref, buf);
1848#endif
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001849
Bram Moolenaar071d4272004-06-13 20:20:40 +00001850 if ( !forceit
1851 && bufIsChanged(buf)
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001852 && ((flags & CCGD_MULTWIN) || buf->b_nwindows <= 1)
1853 && (!(flags & CCGD_AW) || autowrite(buf, forceit) == FAIL))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001854 {
1855#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG)
1856 if ((p_confirm || cmdmod.confirm) && p_write)
1857 {
1858 buf_T *buf2;
1859 int count = 0;
1860
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001861 if (flags & CCGD_ALLBUF)
Bram Moolenaar29323592016-07-24 22:04:11 +02001862 FOR_ALL_BUFFERS(buf2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001863 if (bufIsChanged(buf2)
1864 && (buf2->b_ffname != NULL
1865# ifdef FEAT_BROWSE
1866 || cmdmod.browse
1867# endif
1868 ))
1869 ++count;
1870# ifdef FEAT_AUTOCMD
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001871 if (!bufref_valid(&bufref))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001872 /* Autocommand deleted buffer, oops! It's not changed now. */
1873 return FALSE;
1874# endif
1875 dialog_changed(buf, count > 1);
1876# ifdef FEAT_AUTOCMD
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001877 if (!bufref_valid(&bufref))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001878 /* Autocommand deleted buffer, oops! It's not changed now. */
1879 return FALSE;
1880# endif
1881 return bufIsChanged(buf);
1882 }
1883#endif
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001884 if (flags & CCGD_EXCMD)
1885 EMSG(_(e_nowrtmsg));
1886 else
1887 EMSG(_(e_nowrtmsg_nobang));
Bram Moolenaar071d4272004-06-13 20:20:40 +00001888 return TRUE;
1889 }
1890 return FALSE;
1891}
1892
1893#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG) || defined(PROTO)
1894
1895#if defined(FEAT_BROWSE) || defined(PROTO)
1896/*
1897 * When wanting to write a file without a file name, ask the user for a name.
1898 */
1899 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001900browse_save_fname(buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001901{
1902 if (buf->b_fname == NULL)
1903 {
1904 char_u *fname;
1905
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00001906 fname = do_browse(BROWSE_SAVE, (char_u *)_("Save As"),
1907 NULL, NULL, NULL, NULL, buf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001908 if (fname != NULL)
1909 {
1910 if (setfname(buf, fname, NULL, TRUE) == OK)
1911 buf->b_flags |= BF_NOTEDITED;
1912 vim_free(fname);
1913 }
1914 }
1915}
1916#endif
1917
1918/*
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02001919 * Ask the user what to do when abandoning a changed buffer.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001920 * Must check 'write' option first!
1921 */
1922 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001923dialog_changed(
1924 buf_T *buf,
1925 int checkall) /* may abandon all changed buffers */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001926{
Bram Moolenaard9462e32011-04-11 21:35:11 +02001927 char_u buff[DIALOG_MSG_SIZE];
Bram Moolenaar071d4272004-06-13 20:20:40 +00001928 int ret;
1929 buf_T *buf2;
Bram Moolenaar8218f602012-04-25 17:32:18 +02001930 exarg_T ea;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001931
Bram Moolenaar82cf9b62005-06-07 21:09:25 +00001932 dialog_msg(buff, _("Save changes to \"%s\"?"),
Bram Moolenaar071d4272004-06-13 20:20:40 +00001933 (buf->b_fname != NULL) ?
1934 buf->b_fname : (char_u *)_("Untitled"));
1935 if (checkall)
1936 ret = vim_dialog_yesnoallcancel(VIM_QUESTION, NULL, buff, 1);
1937 else
1938 ret = vim_dialog_yesnocancel(VIM_QUESTION, NULL, buff, 1);
1939
Bram Moolenaar8218f602012-04-25 17:32:18 +02001940 /* Init ea pseudo-structure, this is needed for the check_overwrite()
1941 * function. */
1942 ea.append = ea.forceit = FALSE;
1943
Bram Moolenaar071d4272004-06-13 20:20:40 +00001944 if (ret == VIM_YES)
1945 {
1946#ifdef FEAT_BROWSE
1947 /* May get file name, when there is none */
1948 browse_save_fname(buf);
1949#endif
Bram Moolenaar8218f602012-04-25 17:32:18 +02001950 if (buf->b_fname != NULL && check_overwrite(&ea, buf,
1951 buf->b_fname, buf->b_ffname, FALSE) == OK)
1952 /* didn't hit Cancel */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001953 (void)buf_write_all(buf, FALSE);
1954 }
1955 else if (ret == VIM_NO)
1956 {
1957 unchanged(buf, TRUE);
1958 }
1959 else if (ret == VIM_ALL)
1960 {
1961 /*
1962 * Write all modified files that can be written.
1963 * Skip readonly buffers, these need to be confirmed
1964 * individually.
1965 */
Bram Moolenaar29323592016-07-24 22:04:11 +02001966 FOR_ALL_BUFFERS(buf2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001967 {
1968 if (bufIsChanged(buf2)
1969 && (buf2->b_ffname != NULL
1970#ifdef FEAT_BROWSE
1971 || cmdmod.browse
1972#endif
1973 )
1974 && !buf2->b_p_ro)
1975 {
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001976#ifdef FEAT_AUTOCMD
1977 bufref_T bufref;
1978
1979 set_bufref(&bufref, buf2);
1980#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001981#ifdef FEAT_BROWSE
1982 /* May get file name, when there is none */
1983 browse_save_fname(buf2);
1984#endif
Bram Moolenaar8218f602012-04-25 17:32:18 +02001985 if (buf2->b_fname != NULL && check_overwrite(&ea, buf2,
1986 buf2->b_fname, buf2->b_ffname, FALSE) == OK)
1987 /* didn't hit Cancel */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001988 (void)buf_write_all(buf2, FALSE);
1989#ifdef FEAT_AUTOCMD
1990 /* an autocommand may have deleted the buffer */
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001991 if (!bufref_valid(&bufref))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001992 buf2 = firstbuf;
1993#endif
1994 }
1995 }
1996 }
1997 else if (ret == VIM_DISCARDALL)
1998 {
1999 /*
2000 * mark all buffers as unchanged
2001 */
Bram Moolenaar29323592016-07-24 22:04:11 +02002002 FOR_ALL_BUFFERS(buf2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002003 unchanged(buf2, TRUE);
2004 }
2005}
2006#endif
2007
2008/*
2009 * Return TRUE if the buffer "buf" can be abandoned, either by making it
2010 * hidden, autowriting it or unloading it.
2011 */
2012 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002013can_abandon(buf_T *buf, int forceit)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002014{
2015 return ( P_HID(buf)
2016 || !bufIsChanged(buf)
2017 || buf->b_nwindows > 1
2018 || autowrite(buf, forceit) == OK
2019 || forceit);
2020}
2021
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01002022static void add_bufnum(int *bufnrs, int *bufnump, int nr);
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002023
2024/*
2025 * Add a buffer number to "bufnrs", unless it's already there.
2026 */
2027 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002028add_bufnum(int *bufnrs, int *bufnump, int nr)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002029{
2030 int i;
2031
2032 for (i = 0; i < *bufnump; ++i)
2033 if (bufnrs[i] == nr)
2034 return;
2035 bufnrs[*bufnump] = nr;
2036 *bufnump = *bufnump + 1;
2037}
2038
Bram Moolenaar071d4272004-06-13 20:20:40 +00002039/*
2040 * Return TRUE if any buffer was changed and cannot be abandoned.
2041 * That changed buffer becomes the current buffer.
Bram Moolenaar027387f2016-01-02 22:25:52 +01002042 * When "unload" is true the current buffer is unloaded instead of making it
2043 * hidden. This is used for ":q!".
Bram Moolenaar071d4272004-06-13 20:20:40 +00002044 */
2045 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002046check_changed_any(
2047 int hidden, /* Only check hidden buffers */
2048 int unload)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002049{
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002050 int ret = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002051 buf_T *buf;
2052 int save;
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002053 int i;
2054 int bufnum = 0;
2055 int bufcount = 0;
2056 int *bufnrs;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002057#ifdef FEAT_WINDOWS
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002058 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002059 win_T *wp;
2060#endif
2061
Bram Moolenaar29323592016-07-24 22:04:11 +02002062 FOR_ALL_BUFFERS(buf)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002063 ++bufcount;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002064
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002065 if (bufcount == 0)
2066 return FALSE;
2067
2068 bufnrs = (int *)alloc(sizeof(int) * bufcount);
2069 if (bufnrs == NULL)
2070 return FALSE;
2071
2072 /* curbuf */
2073 bufnrs[bufnum++] = curbuf->b_fnum;
2074#ifdef FEAT_WINDOWS
2075 /* buf in curtab */
2076 FOR_ALL_WINDOWS(wp)
2077 if (wp->w_buffer != curbuf)
2078 add_bufnum(bufnrs, &bufnum, wp->w_buffer->b_fnum);
2079
2080 /* buf in other tab */
Bram Moolenaar29323592016-07-24 22:04:11 +02002081 FOR_ALL_TABPAGES(tp)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002082 if (tp != curtab)
2083 for (wp = tp->tp_firstwin; wp != NULL; wp = wp->w_next)
2084 add_bufnum(bufnrs, &bufnum, wp->w_buffer->b_fnum);
2085#endif
2086 /* any other buf */
Bram Moolenaar29323592016-07-24 22:04:11 +02002087 FOR_ALL_BUFFERS(buf)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002088 add_bufnum(bufnrs, &bufnum, buf->b_fnum);
2089
2090 for (i = 0; i < bufnum; ++i)
2091 {
2092 buf = buflist_findnr(bufnrs[i]);
2093 if (buf == NULL)
2094 continue;
2095 if ((!hidden || buf->b_nwindows == 0) && bufIsChanged(buf))
2096 {
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002097 bufref_T bufref;
2098
2099 set_bufref(&bufref, buf);
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002100 /* Try auto-writing the buffer. If this fails but the buffer no
2101 * longer exists it's not changed, that's OK. */
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002102 if (check_changed(buf, (p_awa ? CCGD_AW : 0)
2103 | CCGD_MULTWIN
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002104 | CCGD_ALLBUF) && bufref_valid(&bufref))
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002105 break; /* didn't save - still changes */
2106 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002107 }
2108
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002109 if (i >= bufnum)
2110 goto theend;
2111
2112 ret = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002113 exiting = FALSE;
2114#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG)
2115 /*
2116 * When ":confirm" used, don't give an error message.
2117 */
2118 if (!(p_confirm || cmdmod.confirm))
2119#endif
2120 {
2121 /* There must be a wait_return for this message, do_buffer()
2122 * may cause a redraw. But wait_return() is a no-op when vgetc()
2123 * is busy (Quit used from window menu), then make sure we don't
2124 * cause a scroll up. */
Bram Moolenaar61660ea2006-04-07 21:40:07 +00002125 if (vgetc_busy > 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002126 {
2127 msg_row = cmdline_row;
2128 msg_col = 0;
2129 msg_didout = FALSE;
2130 }
2131 if (EMSG2(_("E162: No write since last change for buffer \"%s\""),
Bram Moolenaare1704ba2012-10-03 18:25:00 +02002132 buf_spname(buf) != NULL ? buf_spname(buf) : buf->b_fname))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002133 {
2134 save = no_wait_return;
2135 no_wait_return = FALSE;
2136 wait_return(FALSE);
2137 no_wait_return = save;
2138 }
2139 }
2140
2141#ifdef FEAT_WINDOWS
2142 /* Try to find a window that contains the buffer. */
2143 if (buf != curbuf)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002144 FOR_ALL_TAB_WINDOWS(tp, wp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002145 if (wp->w_buffer == buf)
2146 {
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002147# ifdef FEAT_AUTOCMD
2148 bufref_T bufref;
2149
2150 set_bufref(&bufref, buf);
2151# endif
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002152 goto_tabpage_win(tp, wp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002153# ifdef FEAT_AUTOCMD
2154 /* Paranoia: did autocms wipe out the buffer with changes? */
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002155 if (!bufref_valid(&bufref))
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002156 {
2157 goto theend;
2158 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002159# endif
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002160 goto buf_found;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002161 }
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002162buf_found:
Bram Moolenaar071d4272004-06-13 20:20:40 +00002163#endif
2164
2165 /* Open the changed buffer in the current window. */
2166 if (buf != curbuf)
Bram Moolenaar027387f2016-01-02 22:25:52 +01002167 set_curbuf(buf, unload ? DOBUF_UNLOAD : DOBUF_GOTO);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002168
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002169theend:
2170 vim_free(bufnrs);
2171 return ret;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002172}
2173
2174/*
2175 * return FAIL if there is no file name, OK if there is one
2176 * give error message for FAIL
2177 */
2178 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002179check_fname(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002180{
2181 if (curbuf->b_ffname == NULL)
2182 {
2183 EMSG(_(e_noname));
2184 return FAIL;
2185 }
2186 return OK;
2187}
2188
2189/*
2190 * flush the contents of a buffer, unless it has no file name
2191 *
2192 * return FAIL for failure, OK otherwise
2193 */
2194 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002195buf_write_all(buf_T *buf, int forceit)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002196{
2197 int retval;
2198#ifdef FEAT_AUTOCMD
2199 buf_T *old_curbuf = curbuf;
2200#endif
2201
2202 retval = (buf_write(buf, buf->b_ffname, buf->b_fname,
2203 (linenr_T)1, buf->b_ml.ml_line_count, NULL,
2204 FALSE, forceit, TRUE, FALSE));
2205#ifdef FEAT_AUTOCMD
2206 if (curbuf != old_curbuf)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00002207 {
2208 msg_source(hl_attr(HLF_W));
Bram Moolenaar071d4272004-06-13 20:20:40 +00002209 MSG(_("Warning: Entered other buffer unexpectedly (check autocommands)"));
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00002210 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002211#endif
2212 return retval;
2213}
2214
2215/*
2216 * Code to handle the argument list.
2217 */
2218
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01002219static char_u *do_one_arg(char_u *str);
2220static int do_arglist(char_u *str, int what, int after);
2221static void alist_check_arg_idx(void);
2222static int editing_arg_idx(win_T *win);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002223#ifdef FEAT_LISTCMDS
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01002224static int alist_add_list(int count, char_u **files, int after);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002225#endif
2226#define AL_SET 1
2227#define AL_ADD 2
2228#define AL_DEL 3
2229
Bram Moolenaar071d4272004-06-13 20:20:40 +00002230/*
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002231 * Isolate one argument, taking backticks.
2232 * Changes the argument in-place, puts a NUL after it. Backticks remain.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002233 * Return a pointer to the start of the next argument.
2234 */
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002235 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002236do_one_arg(char_u *str)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002237{
2238 char_u *p;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002239 int inbacktick;
2240
Bram Moolenaar071d4272004-06-13 20:20:40 +00002241 inbacktick = FALSE;
2242 for (p = str; *str; ++str)
2243 {
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002244 /* When the backslash is used for escaping the special meaning of a
2245 * character we need to keep it until wildcard expansion. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002246 if (rem_backslash(str))
2247 {
2248 *p++ = *str++;
2249 *p++ = *str;
2250 }
2251 else
2252 {
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002253 /* An item ends at a space not in backticks */
2254 if (!inbacktick && vim_isspace(*str))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002255 break;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002256 if (*str == '`')
Bram Moolenaar071d4272004-06-13 20:20:40 +00002257 inbacktick ^= TRUE;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002258 *p++ = *str;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002259 }
2260 }
2261 str = skipwhite(str);
2262 *p = NUL;
2263
2264 return str;
2265}
2266
Bram Moolenaar86b68352004-12-27 21:59:20 +00002267/*
2268 * Separate the arguments in "str" and return a list of pointers in the
2269 * growarray "gap".
2270 */
2271 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002272get_arglist(garray_T *gap, char_u *str)
Bram Moolenaar86b68352004-12-27 21:59:20 +00002273{
2274 ga_init2(gap, (int)sizeof(char_u *), 20);
2275 while (*str != NUL)
2276 {
2277 if (ga_grow(gap, 1) == FAIL)
2278 {
2279 ga_clear(gap);
2280 return FAIL;
2281 }
2282 ((char_u **)gap->ga_data)[gap->ga_len++] = str;
2283
2284 /* Isolate one argument, change it in-place, put a NUL after it. */
2285 str = do_one_arg(str);
2286 }
2287 return OK;
2288}
2289
Bram Moolenaar7df351e2006-01-23 22:30:28 +00002290#if defined(FEAT_QUICKFIX) || defined(FEAT_SYN_HL) || defined(PROTO)
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002291/*
2292 * Parse a list of arguments (file names), expand them and return in
Bram Moolenaar8f5c6f02012-06-29 12:57:06 +02002293 * "fnames[fcountp]". When "wig" is TRUE, removes files matching 'wildignore'.
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002294 * Return FAIL or OK.
2295 */
2296 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002297get_arglist_exp(
2298 char_u *str,
2299 int *fcountp,
2300 char_u ***fnamesp,
2301 int wig)
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002302{
2303 garray_T ga;
2304 int i;
2305
2306 if (get_arglist(&ga, str) == FAIL)
2307 return FAIL;
Bram Moolenaar8f5c6f02012-06-29 12:57:06 +02002308 if (wig == TRUE)
2309 i = expand_wildcards(ga.ga_len, (char_u **)ga.ga_data,
2310 fcountp, fnamesp, EW_FILE|EW_NOTFOUND);
2311 else
2312 i = gen_expand_wildcards(ga.ga_len, (char_u **)ga.ga_data,
2313 fcountp, fnamesp, EW_FILE|EW_NOTFOUND);
2314
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002315 ga_clear(&ga);
2316 return i;
2317}
2318#endif
2319
Bram Moolenaar071d4272004-06-13 20:20:40 +00002320#if defined(FEAT_GUI) || defined(FEAT_CLIENTSERVER) || defined(PROTO)
2321/*
2322 * Redefine the argument list.
2323 */
2324 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002325set_arglist(char_u *str)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002326{
2327 do_arglist(str, AL_SET, 0);
2328}
2329#endif
2330
2331/*
2332 * "what" == AL_SET: Redefine the argument list to 'str'.
2333 * "what" == AL_ADD: add files in 'str' to the argument list after "after".
2334 * "what" == AL_DEL: remove files in 'str' from the argument list.
2335 *
2336 * Return FAIL for failure, OK otherwise.
2337 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002338 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002339do_arglist(
2340 char_u *str,
Bram Moolenaarf1d25012016-03-03 12:22:53 +01002341 int what,
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002342 int after UNUSED) /* 0 means before first one */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002343{
2344 garray_T new_ga;
2345 int exp_count;
2346 char_u **exp_files;
2347 int i;
2348#ifdef FEAT_LISTCMDS
2349 char_u *p;
2350 int match;
2351#endif
2352
2353 /*
Bram Moolenaar2faa29f2016-01-23 23:02:34 +01002354 * Set default argument for ":argadd" command.
2355 */
2356 if (what == AL_ADD && *str == NUL)
2357 {
2358 if (curbuf->b_ffname == NULL)
2359 return FAIL;
2360 str = curbuf->b_fname;
2361 }
2362
2363 /*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002364 * Collect all file name arguments in "new_ga".
2365 */
Bram Moolenaar86b68352004-12-27 21:59:20 +00002366 if (get_arglist(&new_ga, str) == FAIL)
2367 return FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002368
2369#ifdef FEAT_LISTCMDS
2370 if (what == AL_DEL)
2371 {
2372 regmatch_T regmatch;
2373 int didone;
2374
2375 /*
2376 * Delete the items: use each item as a regexp and find a match in the
2377 * argument list.
2378 */
Bram Moolenaar71afbfe2013-03-19 16:49:16 +01002379 regmatch.rm_ic = p_fic; /* ignore case when 'fileignorecase' is set */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002380 for (i = 0; i < new_ga.ga_len && !got_int; ++i)
2381 {
2382 p = ((char_u **)new_ga.ga_data)[i];
2383 p = file_pat_to_reg_pat(p, NULL, NULL, FALSE);
2384 if (p == NULL)
2385 break;
2386 regmatch.regprog = vim_regcomp(p, p_magic ? RE_MAGIC : 0);
2387 if (regmatch.regprog == NULL)
2388 {
2389 vim_free(p);
2390 break;
2391 }
2392
2393 didone = FALSE;
2394 for (match = 0; match < ARGCOUNT; ++match)
2395 if (vim_regexec(&regmatch, alist_name(&ARGLIST[match]),
2396 (colnr_T)0))
2397 {
2398 didone = TRUE;
2399 vim_free(ARGLIST[match].ae_fname);
2400 mch_memmove(ARGLIST + match, ARGLIST + match + 1,
2401 (ARGCOUNT - match - 1) * sizeof(aentry_T));
2402 --ALIST(curwin)->al_ga.ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002403 if (curwin->w_arg_idx > match)
2404 --curwin->w_arg_idx;
2405 --match;
2406 }
2407
Bram Moolenaar473de612013-06-08 18:19:48 +02002408 vim_regfree(regmatch.regprog);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002409 vim_free(p);
2410 if (!didone)
2411 EMSG2(_(e_nomatch2), ((char_u **)new_ga.ga_data)[i]);
2412 }
2413 ga_clear(&new_ga);
2414 }
2415 else
2416#endif
2417 {
2418 i = expand_wildcards(new_ga.ga_len, (char_u **)new_ga.ga_data,
2419 &exp_count, &exp_files, EW_DIR|EW_FILE|EW_ADDSLASH|EW_NOTFOUND);
2420 ga_clear(&new_ga);
Bram Moolenaar2db5c3b2016-01-16 22:49:34 +01002421 if (i == FAIL || exp_count == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002422 {
2423 EMSG(_(e_nomatch));
2424 return FAIL;
2425 }
2426
2427#ifdef FEAT_LISTCMDS
2428 if (what == AL_ADD)
2429 {
2430 (void)alist_add_list(exp_count, exp_files, after);
2431 vim_free(exp_files);
2432 }
2433 else /* what == AL_SET */
2434#endif
Bram Moolenaar86b68352004-12-27 21:59:20 +00002435 alist_set(ALIST(curwin), exp_count, exp_files, FALSE, NULL, 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002436 }
2437
2438 alist_check_arg_idx();
2439
2440 return OK;
2441}
2442
2443/*
2444 * Check the validity of the arg_idx for each other window.
2445 */
2446 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002447alist_check_arg_idx(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002448{
2449#ifdef FEAT_WINDOWS
2450 win_T *win;
Bram Moolenaarf740b292006-02-16 22:11:02 +00002451 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002452
Bram Moolenaarf740b292006-02-16 22:11:02 +00002453 FOR_ALL_TAB_WINDOWS(tp, win)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002454 if (win->w_alist == curwin->w_alist)
2455 check_arg_idx(win);
2456#else
2457 check_arg_idx(curwin);
2458#endif
2459}
2460
2461/*
Bram Moolenaarb8ff1fb2012-02-04 21:59:01 +01002462 * Return TRUE if window "win" is editing the file at the current argument
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002463 * index.
2464 */
2465 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002466editing_arg_idx(win_T *win)
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002467{
2468 return !(win->w_arg_idx >= WARGCOUNT(win)
2469 || (win->w_buffer->b_fnum
2470 != WARGLIST(win)[win->w_arg_idx].ae_fnum
2471 && (win->w_buffer->b_ffname == NULL
2472 || !(fullpathcmp(
2473 alist_name(&WARGLIST(win)[win->w_arg_idx]),
2474 win->w_buffer->b_ffname, TRUE) & FPC_SAME))));
2475}
2476
2477/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002478 * Check if window "win" is editing the w_arg_idx file in its argument list.
2479 */
2480 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002481check_arg_idx(win_T *win)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002482{
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002483 if (WARGCOUNT(win) > 1 && !editing_arg_idx(win))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002484 {
2485 /* We are not editing the current entry in the argument list.
2486 * Set "arg_had_last" if we are editing the last one. */
2487 win->w_arg_idx_invalid = TRUE;
2488 if (win->w_arg_idx != WARGCOUNT(win) - 1
2489 && arg_had_last == FALSE
2490#ifdef FEAT_WINDOWS
2491 && ALIST(win) == &global_alist
2492#endif
2493 && GARGCOUNT > 0
2494 && win->w_arg_idx < GARGCOUNT
2495 && (win->w_buffer->b_fnum == GARGLIST[GARGCOUNT - 1].ae_fnum
2496 || (win->w_buffer->b_ffname != NULL
2497 && (fullpathcmp(alist_name(&GARGLIST[GARGCOUNT - 1]),
2498 win->w_buffer->b_ffname, TRUE) & FPC_SAME))))
2499 arg_had_last = TRUE;
2500 }
2501 else
2502 {
2503 /* We are editing the current entry in the argument list.
2504 * Set "arg_had_last" if it's also the last one */
2505 win->w_arg_idx_invalid = FALSE;
2506 if (win->w_arg_idx == WARGCOUNT(win) - 1
2507#ifdef FEAT_WINDOWS
2508 && win->w_alist == &global_alist
2509#endif
2510 )
2511 arg_had_last = TRUE;
2512 }
2513}
2514
2515/*
2516 * ":args", ":argslocal" and ":argsglobal".
2517 */
2518 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002519ex_args(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002520{
2521 int i;
2522
2523 if (eap->cmdidx != CMD_args)
2524 {
2525#if defined(FEAT_WINDOWS) && defined(FEAT_LISTCMDS)
2526 alist_unlink(ALIST(curwin));
2527 if (eap->cmdidx == CMD_argglobal)
2528 ALIST(curwin) = &global_alist;
2529 else /* eap->cmdidx == CMD_arglocal */
2530 alist_new();
2531#else
2532 ex_ni(eap);
2533 return;
2534#endif
2535 }
2536
2537 if (!ends_excmd(*eap->arg))
2538 {
2539 /*
2540 * ":args file ..": define new argument list, handle like ":next"
2541 * Also for ":argslocal file .." and ":argsglobal file ..".
2542 */
2543 ex_next(eap);
2544 }
2545 else
2546#if defined(FEAT_WINDOWS) && defined(FEAT_LISTCMDS)
2547 if (eap->cmdidx == CMD_args)
2548#endif
2549 {
2550 /*
2551 * ":args": list arguments.
2552 */
2553 if (ARGCOUNT > 0)
2554 {
2555 /* Overwrite the command, for a short list there is no scrolling
2556 * required and no wait_return(). */
2557 gotocmdline(TRUE);
2558 for (i = 0; i < ARGCOUNT; ++i)
2559 {
2560 if (i == curwin->w_arg_idx)
2561 msg_putchar('[');
2562 msg_outtrans(alist_name(&ARGLIST[i]));
2563 if (i == curwin->w_arg_idx)
2564 msg_putchar(']');
2565 msg_putchar(' ');
2566 }
2567 }
2568 }
2569#if defined(FEAT_WINDOWS) && defined(FEAT_LISTCMDS)
2570 else if (eap->cmdidx == CMD_arglocal)
2571 {
2572 garray_T *gap = &curwin->w_alist->al_ga;
2573
2574 /*
2575 * ":argslocal": make a local copy of the global argument list.
2576 */
2577 if (ga_grow(gap, GARGCOUNT) == OK)
2578 for (i = 0; i < GARGCOUNT; ++i)
2579 if (GARGLIST[i].ae_fname != NULL)
2580 {
2581 AARGLIST(curwin->w_alist)[gap->ga_len].ae_fname =
2582 vim_strsave(GARGLIST[i].ae_fname);
2583 AARGLIST(curwin->w_alist)[gap->ga_len].ae_fnum =
2584 GARGLIST[i].ae_fnum;
2585 ++gap->ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002586 }
2587 }
2588#endif
2589}
2590
2591/*
2592 * ":previous", ":sprevious", ":Next" and ":sNext".
2593 */
2594 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002595ex_previous(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002596{
2597 /* If past the last one already, go to the last one. */
2598 if (curwin->w_arg_idx - (int)eap->line2 >= ARGCOUNT)
2599 do_argfile(eap, ARGCOUNT - 1);
2600 else
2601 do_argfile(eap, curwin->w_arg_idx - (int)eap->line2);
2602}
2603
2604/*
2605 * ":rewind", ":first", ":sfirst" and ":srewind".
2606 */
2607 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002608ex_rewind(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002609{
2610 do_argfile(eap, 0);
2611}
2612
2613/*
2614 * ":last" and ":slast".
2615 */
2616 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002617ex_last(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002618{
2619 do_argfile(eap, ARGCOUNT - 1);
2620}
2621
2622/*
2623 * ":argument" and ":sargument".
2624 */
2625 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002626ex_argument(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002627{
2628 int i;
2629
2630 if (eap->addr_count > 0)
2631 i = eap->line2 - 1;
2632 else
2633 i = curwin->w_arg_idx;
2634 do_argfile(eap, i);
2635}
2636
2637/*
2638 * Edit file "argn" of the argument lists.
2639 */
2640 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002641do_argfile(exarg_T *eap, int argn)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002642{
2643 int other;
2644 char_u *p;
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002645 int old_arg_idx = curwin->w_arg_idx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002646
2647 if (argn < 0 || argn >= ARGCOUNT)
2648 {
2649 if (ARGCOUNT <= 1)
2650 EMSG(_("E163: There is only one file to edit"));
2651 else if (argn < 0)
2652 EMSG(_("E164: Cannot go before first file"));
2653 else
2654 EMSG(_("E165: Cannot go beyond last file"));
2655 }
2656 else
2657 {
2658 setpcmark();
2659#ifdef FEAT_GUI
2660 need_mouse_correct = TRUE;
2661#endif
2662
2663#ifdef FEAT_WINDOWS
Bram Moolenaardf1bdc92006-02-23 21:32:16 +00002664 /* split window or create new tab page first */
2665 if (*eap->cmd == 's' || cmdmod.tab != 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002666 {
2667 if (win_split(0, 0) == FAIL)
2668 return;
Bram Moolenaar3368ea22010-09-21 16:56:35 +02002669 RESET_BINDING(curwin);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002670 }
2671 else
2672#endif
2673 {
2674 /*
2675 * if 'hidden' set, only check for changed file when re-editing
2676 * the same buffer
2677 */
2678 other = TRUE;
2679 if (P_HID(curbuf))
2680 {
2681 p = fix_fname(alist_name(&ARGLIST[argn]));
2682 other = otherfile(p);
2683 vim_free(p);
2684 }
2685 if ((!P_HID(curbuf) || !other)
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002686 && check_changed(curbuf, CCGD_AW
2687 | (other ? 0 : CCGD_MULTWIN)
2688 | (eap->forceit ? CCGD_FORCEIT : 0)
2689 | CCGD_EXCMD))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002690 return;
2691 }
2692
2693 curwin->w_arg_idx = argn;
2694 if (argn == ARGCOUNT - 1
2695#ifdef FEAT_WINDOWS
2696 && curwin->w_alist == &global_alist
2697#endif
2698 )
2699 arg_had_last = TRUE;
2700
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002701 /* Edit the file; always use the last known line number.
2702 * When it fails (e.g. Abort for already edited file) restore the
2703 * argument index. */
2704 if (do_ecmd(0, alist_name(&ARGLIST[curwin->w_arg_idx]), NULL,
Bram Moolenaar071d4272004-06-13 20:20:40 +00002705 eap, ECMD_LAST,
Bram Moolenaar701f7af2008-11-15 13:12:07 +00002706 (P_HID(curwin->w_buffer) ? ECMD_HIDE : 0)
2707 + (eap->forceit ? ECMD_FORCEIT : 0), curwin) == FAIL)
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002708 curwin->w_arg_idx = old_arg_idx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002709 /* like Vi: set the mark where the cursor is in the file. */
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002710 else if (eap->cmdidx != CMD_argdo)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002711 setmark('\'');
2712 }
2713}
2714
2715/*
2716 * ":next", and commands that behave like it.
2717 */
2718 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002719ex_next(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002720{
2721 int i;
2722
2723 /*
2724 * check for changed buffer now, if this fails the argument list is not
2725 * redefined.
2726 */
2727 if ( P_HID(curbuf)
2728 || eap->cmdidx == CMD_snext
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002729 || !check_changed(curbuf, CCGD_AW
2730 | (eap->forceit ? CCGD_FORCEIT : 0)
2731 | CCGD_EXCMD))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002732 {
2733 if (*eap->arg != NUL) /* redefine file list */
2734 {
2735 if (do_arglist(eap->arg, AL_SET, 0) == FAIL)
2736 return;
2737 i = 0;
2738 }
2739 else
2740 i = curwin->w_arg_idx + (int)eap->line2;
2741 do_argfile(eap, i);
2742 }
2743}
2744
Bram Moolenaar0106e3d2016-02-23 18:55:43 +01002745#if defined(FEAT_LISTCMDS) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002746/*
2747 * ":argedit"
2748 */
2749 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002750ex_argedit(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002751{
2752 int fnum;
2753 int i;
2754 char_u *s;
2755
2756 /* Add the argument to the buffer list and get the buffer number. */
2757 fnum = buflist_add(eap->arg, BLN_LISTED);
2758
2759 /* Check if this argument is already in the argument list. */
2760 for (i = 0; i < ARGCOUNT; ++i)
2761 if (ARGLIST[i].ae_fnum == fnum)
2762 break;
2763 if (i == ARGCOUNT)
2764 {
2765 /* Can't find it, add it to the argument list. */
2766 s = vim_strsave(eap->arg);
2767 if (s == NULL)
2768 return;
2769 i = alist_add_list(1, &s,
2770 eap->addr_count > 0 ? (int)eap->line2 : curwin->w_arg_idx + 1);
2771 if (i < 0)
2772 return;
2773 curwin->w_arg_idx = i;
2774 }
2775
2776 alist_check_arg_idx();
2777
2778 /* Edit the argument. */
2779 do_argfile(eap, i);
2780}
2781
2782/*
2783 * ":argadd"
2784 */
2785 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002786ex_argadd(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002787{
2788 do_arglist(eap->arg, AL_ADD,
2789 eap->addr_count > 0 ? (int)eap->line2 : curwin->w_arg_idx + 1);
2790#ifdef FEAT_TITLE
2791 maketitle();
2792#endif
2793}
2794
2795/*
2796 * ":argdelete"
2797 */
2798 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002799ex_argdelete(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002800{
2801 int i;
2802 int n;
2803
2804 if (eap->addr_count > 0)
2805 {
2806 /* ":1,4argdel": Delete all arguments in the range. */
2807 if (eap->line2 > ARGCOUNT)
2808 eap->line2 = ARGCOUNT;
2809 n = eap->line2 - eap->line1 + 1;
2810 if (*eap->arg != NUL || n <= 0)
2811 EMSG(_(e_invarg));
2812 else
2813 {
2814 for (i = eap->line1; i <= eap->line2; ++i)
2815 vim_free(ARGLIST[i - 1].ae_fname);
2816 mch_memmove(ARGLIST + eap->line1 - 1, ARGLIST + eap->line2,
2817 (size_t)((ARGCOUNT - eap->line2) * sizeof(aentry_T)));
2818 ALIST(curwin)->al_ga.ga_len -= n;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002819 if (curwin->w_arg_idx >= eap->line2)
2820 curwin->w_arg_idx -= n;
2821 else if (curwin->w_arg_idx > eap->line1)
2822 curwin->w_arg_idx = eap->line1;
Bram Moolenaar72defda2016-01-17 18:04:33 +01002823 if (ARGCOUNT == 0)
2824 curwin->w_arg_idx = 0;
2825 else if (curwin->w_arg_idx >= ARGCOUNT)
2826 curwin->w_arg_idx = ARGCOUNT - 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002827 }
2828 }
2829 else if (*eap->arg == NUL)
2830 EMSG(_(e_argreq));
2831 else
2832 do_arglist(eap->arg, AL_DEL, 0);
2833#ifdef FEAT_TITLE
2834 maketitle();
2835#endif
2836}
2837
2838/*
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002839 * ":argdo", ":windo", ":bufdo", ":tabdo", ":cdo", ":ldo", ":cfdo" and ":lfdo"
Bram Moolenaar071d4272004-06-13 20:20:40 +00002840 */
2841 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002842ex_listdo(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002843{
2844 int i;
2845#ifdef FEAT_WINDOWS
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002846 win_T *wp;
2847 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002848#endif
Bram Moolenaare25bb902015-02-27 20:33:37 +01002849 buf_T *buf = curbuf;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002850 int next_fnum = 0;
2851#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
2852 char_u *save_ei = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002853#endif
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002854 char_u *p_shm_save;
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002855#ifdef FEAT_QUICKFIX
Bram Moolenaared84b762015-09-09 22:35:29 +02002856 int qf_size = 0;
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002857 int qf_idx;
2858#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002859
2860#ifndef FEAT_WINDOWS
2861 if (eap->cmdidx == CMD_windo)
2862 {
2863 ex_ni(eap);
2864 return;
2865 }
2866#endif
2867
Bram Moolenaar0106e3d2016-02-23 18:55:43 +01002868#ifndef FEAT_QUICKFIX
2869 if (eap->cmdidx == CMD_cdo || eap->cmdidx == CMD_ldo ||
2870 eap->cmdidx == CMD_cfdo || eap->cmdidx == CMD_lfdo)
2871 {
2872 ex_ni(eap);
2873 return;
2874 }
2875#endif
2876
Bram Moolenaar071d4272004-06-13 20:20:40 +00002877#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002878 if (eap->cmdidx != CMD_windo && eap->cmdidx != CMD_tabdo)
Bram Moolenaardcaf10e2005-01-21 11:55:25 +00002879 /* Don't do syntax HL autocommands. Skipping the syntax file is a
2880 * great speed improvement. */
2881 save_ei = au_event_disable(",Syntax");
Bram Moolenaar071d4272004-06-13 20:20:40 +00002882#endif
Bram Moolenaar6b1ee342014-08-06 18:17:11 +02002883#ifdef FEAT_CLIPBOARD
2884 start_global_changes();
2885#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002886
2887 if (eap->cmdidx == CMD_windo
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002888 || eap->cmdidx == CMD_tabdo
Bram Moolenaar071d4272004-06-13 20:20:40 +00002889 || P_HID(curbuf)
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002890 || !check_changed(curbuf, CCGD_AW
2891 | (eap->forceit ? CCGD_FORCEIT : 0)
2892 | CCGD_EXCMD))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002893 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00002894 i = 0;
Bram Moolenaara162bc52015-01-07 16:54:21 +01002895 /* start at the eap->line1 argument/window/buffer */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002896#ifdef FEAT_WINDOWS
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002897 wp = firstwin;
2898 tp = first_tabpage;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002899#endif
Bram Moolenaara162bc52015-01-07 16:54:21 +01002900 switch (eap->cmdidx)
2901 {
2902#ifdef FEAT_WINDOWS
2903 case CMD_windo:
2904 for ( ; wp != NULL && i + 1 < eap->line1; wp = wp->w_next)
2905 i++;
2906 break;
2907 case CMD_tabdo:
2908 for( ; tp != NULL && i + 1 < eap->line1; tp = tp->tp_next)
2909 i++;
2910 break;
2911#endif
2912 case CMD_argdo:
2913 i = eap->line1 - 1;
2914 break;
Bram Moolenaara162bc52015-01-07 16:54:21 +01002915 default:
2916 break;
2917 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002918 /* set pcmark now */
2919 if (eap->cmdidx == CMD_bufdo)
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002920 {
Bram Moolenaare25bb902015-02-27 20:33:37 +01002921 /* Advance to the first listed buffer after "eap->line1". */
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002922 for (buf = firstbuf; buf != NULL && (buf->b_fnum < eap->line1
Bram Moolenaare25bb902015-02-27 20:33:37 +01002923 || !buf->b_p_bl); buf = buf->b_next)
2924 if (buf->b_fnum > eap->line2)
2925 {
2926 buf = NULL;
2927 break;
2928 }
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002929 if (buf != NULL)
Bram Moolenaare25bb902015-02-27 20:33:37 +01002930 goto_buffer(eap, DOBUF_FIRST, FORWARD, buf->b_fnum);
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002931 }
2932#ifdef FEAT_QUICKFIX
2933 else if (eap->cmdidx == CMD_cdo || eap->cmdidx == CMD_ldo
2934 || eap->cmdidx == CMD_cfdo || eap->cmdidx == CMD_lfdo)
2935 {
2936 qf_size = qf_get_size(eap);
2937 if (qf_size <= 0 || eap->line1 > qf_size)
2938 buf = NULL;
2939 else
2940 {
2941 ex_cc(eap);
2942
2943 buf = curbuf;
2944 i = eap->line1 - 1;
2945 if (eap->addr_count <= 0)
2946 /* default is all the quickfix/location list entries */
2947 eap->line2 = qf_size;
2948 }
2949 }
2950#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002951 else
2952 setpcmark();
2953 listcmd_busy = TRUE; /* avoids setting pcmark below */
2954
Bram Moolenaare25bb902015-02-27 20:33:37 +01002955 while (!got_int && buf != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002956 {
2957 if (eap->cmdidx == CMD_argdo)
2958 {
2959 /* go to argument "i" */
2960 if (i == ARGCOUNT)
2961 break;
2962 /* Don't call do_argfile() when already there, it will try
2963 * reloading the file. */
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002964 if (curwin->w_arg_idx != i || !editing_arg_idx(curwin))
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002965 {
2966 /* Clear 'shm' to avoid that the file message overwrites
2967 * any output from the command. */
2968 p_shm_save = vim_strsave(p_shm);
2969 set_option_value((char_u *)"shm", 0L, (char_u *)"", 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002970 do_argfile(eap, i);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002971 set_option_value((char_u *)"shm", 0L, p_shm_save, 0);
2972 vim_free(p_shm_save);
2973 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002974 if (curwin->w_arg_idx != i)
2975 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002976 }
2977#ifdef FEAT_WINDOWS
2978 else if (eap->cmdidx == CMD_windo)
2979 {
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002980 /* go to window "wp" */
2981 if (!win_valid(wp))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002982 break;
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002983 win_goto(wp);
Bram Moolenaar41423242007-05-03 20:11:13 +00002984 if (curwin != wp)
2985 break; /* something must be wrong */
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002986 wp = curwin->w_next;
2987 }
2988 else if (eap->cmdidx == CMD_tabdo)
2989 {
2990 /* go to window "tp" */
2991 if (!valid_tabpage(tp))
2992 break;
Bram Moolenaar49e649f2013-05-06 04:50:35 +02002993 goto_tabpage_tp(tp, TRUE, TRUE);
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002994 tp = tp->tp_next;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002995 }
2996#endif
2997 else if (eap->cmdidx == CMD_bufdo)
2998 {
2999 /* Remember the number of the next listed buffer, in case
3000 * ":bwipe" is used or autocommands do something strange. */
3001 next_fnum = -1;
3002 for (buf = curbuf->b_next; buf != NULL; buf = buf->b_next)
3003 if (buf->b_p_bl)
3004 {
3005 next_fnum = buf->b_fnum;
3006 break;
3007 }
3008 }
3009
Bram Moolenaara162bc52015-01-07 16:54:21 +01003010 ++i;
3011
Bram Moolenaar071d4272004-06-13 20:20:40 +00003012 /* execute the command */
3013 do_cmdline(eap->arg, eap->getline, eap->cookie,
3014 DOCMD_VERBOSE + DOCMD_NOWAIT);
3015
3016 if (eap->cmdidx == CMD_bufdo)
3017 {
3018 /* Done? */
Bram Moolenaara162bc52015-01-07 16:54:21 +01003019 if (next_fnum < 0 || next_fnum > eap->line2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003020 break;
3021 /* Check if the buffer still exists. */
Bram Moolenaar29323592016-07-24 22:04:11 +02003022 FOR_ALL_BUFFERS(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003023 if (buf->b_fnum == next_fnum)
3024 break;
3025 if (buf == NULL)
3026 break;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003027
3028 /* Go to the next buffer. Clear 'shm' to avoid that the file
3029 * message overwrites any output from the command. */
3030 p_shm_save = vim_strsave(p_shm);
3031 set_option_value((char_u *)"shm", 0L, (char_u *)"", 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003032 goto_buffer(eap, DOBUF_FIRST, FORWARD, next_fnum);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003033 set_option_value((char_u *)"shm", 0L, p_shm_save, 0);
3034 vim_free(p_shm_save);
3035
Bram Moolenaaraa23b372015-09-08 18:46:31 +02003036 /* If autocommands took us elsewhere, quit here. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003037 if (curbuf->b_fnum != next_fnum)
3038 break;
3039 }
3040
Bram Moolenaaraa23b372015-09-08 18:46:31 +02003041#ifdef FEAT_QUICKFIX
3042 if (eap->cmdidx == CMD_cdo || eap->cmdidx == CMD_ldo
3043 || eap->cmdidx == CMD_cfdo || eap->cmdidx == CMD_lfdo)
3044 {
3045 if (i >= qf_size || i >= eap->line2)
3046 break;
3047
3048 qf_idx = qf_get_cur_idx(eap);
3049
3050 ex_cnext(eap);
3051
3052 /* If jumping to the next quickfix entry fails, quit here */
3053 if (qf_get_cur_idx(eap) == qf_idx)
3054 break;
3055 }
3056#endif
3057
Bram Moolenaar071d4272004-06-13 20:20:40 +00003058 if (eap->cmdidx == CMD_windo)
3059 {
3060 validate_cursor(); /* cursor may have moved */
3061#ifdef FEAT_SCROLLBIND
3062 /* required when 'scrollbind' has been set */
3063 if (curwin->w_p_scb)
3064 do_check_scrollbind(TRUE);
3065#endif
3066 }
Bram Moolenaara162bc52015-01-07 16:54:21 +01003067
3068#ifdef FEAT_WINDOWS
3069 if (eap->cmdidx == CMD_windo || eap->cmdidx == CMD_tabdo)
3070 if (i+1 > eap->line2)
3071 break;
3072#endif
3073 if (eap->cmdidx == CMD_argdo && i >= eap->line2)
3074 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003075 }
3076 listcmd_busy = FALSE;
3077 }
3078
3079#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
Bram Moolenaar6ac54292005-02-02 23:07:25 +00003080 if (save_ei != NULL)
3081 {
3082 au_event_restore(save_ei);
3083 apply_autocmds(EVENT_SYNTAX, curbuf->b_p_syn,
3084 curbuf->b_fname, TRUE, curbuf);
3085 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003086#endif
Bram Moolenaar6b1ee342014-08-06 18:17:11 +02003087#ifdef FEAT_CLIPBOARD
3088 end_global_changes();
3089#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003090}
3091
3092/*
3093 * Add files[count] to the arglist of the current window after arg "after".
3094 * The file names in files[count] must have been allocated and are taken over.
3095 * Files[] itself is not taken over.
3096 * Returns index of first added argument. Returns -1 when failed (out of mem).
3097 */
3098 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003099alist_add_list(
3100 int count,
3101 char_u **files,
3102 int after) /* where to add: 0 = before first one */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003103{
3104 int i;
Bram Moolenaara24f0a52016-01-17 19:39:00 +01003105 int old_argcount = ARGCOUNT;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003106
3107 if (ga_grow(&ALIST(curwin)->al_ga, count) == OK)
3108 {
3109 if (after < 0)
3110 after = 0;
3111 if (after > ARGCOUNT)
3112 after = ARGCOUNT;
3113 if (after < ARGCOUNT)
3114 mch_memmove(&(ARGLIST[after + count]), &(ARGLIST[after]),
3115 (ARGCOUNT - after) * sizeof(aentry_T));
3116 for (i = 0; i < count; ++i)
3117 {
3118 ARGLIST[after + i].ae_fname = files[i];
3119 ARGLIST[after + i].ae_fnum = buflist_add(files[i], BLN_LISTED);
3120 }
3121 ALIST(curwin)->al_ga.ga_len += count;
Bram Moolenaara24f0a52016-01-17 19:39:00 +01003122 if (old_argcount > 0 && curwin->w_arg_idx >= after)
3123 curwin->w_arg_idx += count;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003124 return after;
3125 }
3126
3127 for (i = 0; i < count; ++i)
3128 vim_free(files[i]);
3129 return -1;
3130}
3131
3132#endif /* FEAT_LISTCMDS */
3133
3134#ifdef FEAT_EVAL
3135/*
3136 * ":compiler[!] {name}"
3137 */
3138 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003139ex_compiler(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003140{
3141 char_u *buf;
3142 char_u *old_cur_comp = NULL;
3143 char_u *p;
3144
3145 if (*eap->arg == NUL)
3146 {
3147 /* List all compiler scripts. */
3148 do_cmdline_cmd((char_u *)"echo globpath(&rtp, 'compiler/*.vim')");
3149 /* ) keep the indenter happy... */
3150 }
3151 else
3152 {
3153 buf = alloc((unsigned)(STRLEN(eap->arg) + 14));
3154 if (buf != NULL)
3155 {
3156 if (eap->forceit)
3157 {
3158 /* ":compiler! {name}" sets global options */
3159 do_cmdline_cmd((char_u *)
3160 "command -nargs=* CompilerSet set <args>");
3161 }
3162 else
3163 {
3164 /* ":compiler! {name}" sets local options.
3165 * To remain backwards compatible "current_compiler" is always
3166 * used. A user's compiler plugin may set it, the distributed
3167 * plugin will then skip the settings. Afterwards set
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003168 * "b:current_compiler" and restore "current_compiler".
3169 * Explicitly prepend "g:" to make it work in a function. */
3170 old_cur_comp = get_var_value((char_u *)"g:current_compiler");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003171 if (old_cur_comp != NULL)
3172 old_cur_comp = vim_strsave(old_cur_comp);
3173 do_cmdline_cmd((char_u *)
3174 "command -nargs=* CompilerSet setlocal <args>");
3175 }
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003176 do_unlet((char_u *)"g:current_compiler", TRUE);
Bram Moolenaar2ce06f62005-01-31 19:19:04 +00003177 do_unlet((char_u *)"b:current_compiler", TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003178
3179 sprintf((char *)buf, "compiler/%s.vim", eap->arg);
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003180 if (source_runtime(buf, DIP_ALL) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003181 EMSG2(_("E666: compiler not supported: %s"), eap->arg);
3182 vim_free(buf);
3183
3184 do_cmdline_cmd((char_u *)":delcommand CompilerSet");
3185
3186 /* Set "b:current_compiler" from "current_compiler". */
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003187 p = get_var_value((char_u *)"g:current_compiler");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003188 if (p != NULL)
3189 set_internal_string_var((char_u *)"b:current_compiler", p);
3190
3191 /* Restore "current_compiler" for ":compiler {name}". */
3192 if (!eap->forceit)
3193 {
3194 if (old_cur_comp != NULL)
3195 {
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003196 set_internal_string_var((char_u *)"g:current_compiler",
Bram Moolenaar071d4272004-06-13 20:20:40 +00003197 old_cur_comp);
3198 vim_free(old_cur_comp);
3199 }
3200 else
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003201 do_unlet((char_u *)"g:current_compiler", TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003202 }
3203 }
3204 }
3205}
3206#endif
3207
3208/*
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003209 * ":runtime [what] {name}"
Bram Moolenaar071d4272004-06-13 20:20:40 +00003210 */
3211 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003212ex_runtime(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003213{
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003214 char_u *arg = eap->arg;
3215 char_u *p = skiptowhite(arg);
3216 int len = (int)(p - arg);
3217 int flags = eap->forceit ? DIP_ALL : 0;
3218
3219 if (STRNCMP(arg, "START", len) == 0)
3220 {
3221 flags += DIP_START + DIP_NORTP;
3222 arg = skipwhite(arg + len);
3223 }
3224 else if (STRNCMP(arg, "OPT", len) == 0)
3225 {
3226 flags += DIP_OPT + DIP_NORTP;
3227 arg = skipwhite(arg + len);
3228 }
3229 else if (STRNCMP(arg, "PACK", len) == 0)
3230 {
3231 flags += DIP_START + DIP_OPT + DIP_NORTP;
3232 arg = skipwhite(arg + len);
3233 }
3234 else if (STRNCMP(arg, "ALL", len) == 0)
3235 {
3236 flags += DIP_START + DIP_OPT;
3237 arg = skipwhite(arg + len);
3238 }
3239
3240 source_runtime(arg, flags);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003241}
3242
Bram Moolenaar071d4272004-06-13 20:20:40 +00003243 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003244source_callback(char_u *fname, void *cookie UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003245{
Bram Moolenaar910f66f2006-04-05 20:41:53 +00003246 (void)do_source(fname, FALSE, DOSO_NONE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003247}
3248
3249/*
3250 * Source the file "name" from all directories in 'runtimepath'.
3251 * "name" can contain wildcards.
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003252 * When "flags" has DIP_ALL: source all files, otherwise only the first one.
Bram Moolenaar91715872016-03-03 17:13:03 +01003253 *
Bram Moolenaar071d4272004-06-13 20:20:40 +00003254 * return FAIL when no file could be sourced, OK otherwise.
3255 */
3256 int
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003257source_runtime(char_u *name, int flags)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003258{
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003259 return do_in_runtimepath(name, flags, source_callback, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003260}
3261
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003262/*
3263 * Find the file "name" in all directories in "path" and invoke
3264 * "callback(fname, cookie)".
3265 * "name" can contain wildcards.
3266 * When "flags" has DIP_ALL: source all files, otherwise only the first one.
3267 * When "flags" has DIP_DIR: find directories instead of files.
3268 * When "flags" has DIP_ERR: give an error message if there is no match.
3269 *
3270 * return FAIL when no file could be sourced, OK otherwise.
3271 */
Bram Moolenaar6bef5302016-03-12 21:28:26 +01003272 int
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003273do_in_path(
3274 char_u *path,
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003275 char_u *name,
Bram Moolenaar91715872016-03-03 17:13:03 +01003276 int flags,
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003277 void (*callback)(char_u *fname, void *ck),
3278 void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003279{
3280 char_u *rtp;
3281 char_u *np;
3282 char_u *buf;
3283 char_u *rtp_copy;
3284 char_u *tail;
3285 int num_files;
3286 char_u **files;
3287 int i;
3288 int did_one = FALSE;
3289#ifdef AMIGA
3290 struct Process *proc = (struct Process *)FindTask(0L);
3291 APTR save_winptr = proc->pr_WindowPtr;
3292
3293 /* Avoid a requester here for a volume that doesn't exist. */
3294 proc->pr_WindowPtr = (APTR)-1L;
3295#endif
3296
3297 /* Make a copy of 'runtimepath'. Invoking the callback may change the
3298 * value. */
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003299 rtp_copy = vim_strsave(path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003300 buf = alloc(MAXPATHL);
3301 if (buf != NULL && rtp_copy != NULL)
3302 {
Bram Moolenaarc09a6d62013-06-10 21:27:29 +02003303 if (p_verbose > 1 && name != NULL)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003304 {
3305 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00003306 smsg((char_u *)_("Searching for \"%s\" in \"%s\""),
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003307 (char *)name, (char *)path);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003308 verbose_leave();
3309 }
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00003310
Bram Moolenaar071d4272004-06-13 20:20:40 +00003311 /* Loop over all entries in 'runtimepath'. */
3312 rtp = rtp_copy;
Bram Moolenaar91715872016-03-03 17:13:03 +01003313 while (*rtp != NUL && ((flags & DIP_ALL) || !did_one))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003314 {
Bram Moolenaar66459b72016-08-06 19:01:55 +02003315 size_t buflen;
3316
Bram Moolenaar071d4272004-06-13 20:20:40 +00003317 /* Copy the path from 'runtimepath' to buf[]. */
3318 copy_option_part(&rtp, buf, MAXPATHL, ",");
Bram Moolenaar66459b72016-08-06 19:01:55 +02003319 buflen = STRLEN(buf);
3320
3321 /* Skip after or non-after directories. */
3322 if (flags & (DIP_NOAFTER | DIP_AFTER))
3323 {
3324 int is_after = buflen >= 5
3325 && STRCMP(buf + buflen - 5, "after") == 0;
3326
3327 if ((is_after && (flags & DIP_NOAFTER))
3328 || (!is_after && (flags & DIP_AFTER)))
3329 continue;
3330 }
3331
Bram Moolenaarc09a6d62013-06-10 21:27:29 +02003332 if (name == NULL)
3333 {
3334 (*callback)(buf, (void *) &cookie);
3335 if (!did_one)
3336 did_one = (cookie == NULL);
3337 }
Bram Moolenaar66459b72016-08-06 19:01:55 +02003338 else if (buflen + STRLEN(name) + 2 < MAXPATHL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003339 {
3340 add_pathsep(buf);
3341 tail = buf + STRLEN(buf);
3342
3343 /* Loop over all patterns in "name" */
3344 np = name;
Bram Moolenaar91715872016-03-03 17:13:03 +01003345 while (*np != NUL && ((flags & DIP_ALL) || !did_one))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003346 {
3347 /* Append the pattern from "name" to buf[]. */
3348 copy_option_part(&np, tail, (int)(MAXPATHL - (tail - buf)),
3349 "\t ");
3350
3351 if (p_verbose > 2)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003352 {
3353 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00003354 smsg((char_u *)_("Searching for \"%s\""), buf);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003355 verbose_leave();
3356 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003357
3358 /* Expand wildcards, invoke the callback for each match. */
3359 if (gen_expand_wildcards(1, &buf, &num_files, &files,
Bram Moolenaar91715872016-03-03 17:13:03 +01003360 (flags & DIP_DIR) ? EW_DIR : EW_FILE) == OK)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003361 {
3362 for (i = 0; i < num_files; ++i)
3363 {
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00003364 (*callback)(files[i], cookie);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003365 did_one = TRUE;
Bram Moolenaar91715872016-03-03 17:13:03 +01003366 if (!(flags & DIP_ALL))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003367 break;
3368 }
3369 FreeWild(num_files, files);
3370 }
3371 }
3372 }
3373 }
3374 }
3375 vim_free(buf);
3376 vim_free(rtp_copy);
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003377 if (!did_one && name != NULL)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003378 {
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003379 char *basepath = path == p_rtp ? "runtimepath" : "packpath";
3380
3381 if (flags & DIP_ERR)
3382 EMSG3(_(e_dirnotf), basepath, name);
3383 else if (p_verbose > 0)
3384 {
3385 verbose_enter();
3386 smsg((char_u *)_("not found in '%s': \"%s\""), basepath, name);
3387 verbose_leave();
3388 }
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003389 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003390
3391#ifdef AMIGA
3392 proc->pr_WindowPtr = save_winptr;
3393#endif
3394
3395 return did_one ? OK : FAIL;
3396}
3397
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003398/*
3399 * Find "name" in 'runtimepath'. When found, invoke the callback function for
3400 * it: callback(fname, "cookie")
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003401 * When "flags" has DIP_ALL repeat for all matches, otherwise only the first
3402 * one is used.
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003403 * Returns OK when at least one match found, FAIL otherwise.
3404 *
3405 * If "name" is NULL calls callback for each entry in runtimepath. Cookie is
3406 * passed by reference in this case, setting it to NULL indicates that callback
3407 * has done its job.
3408 */
3409 int
3410do_in_runtimepath(
3411 char_u *name,
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003412 int flags,
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003413 void (*callback)(char_u *fname, void *ck),
3414 void *cookie)
3415{
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003416 int done = FAIL;
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003417 char_u *s;
3418 int len;
3419 char *start_dir = "pack/*/start/*/%s";
3420 char *opt_dir = "pack/*/opt/*/%s";
3421
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003422 if ((flags & DIP_NORTP) == 0)
3423 done = do_in_path(p_rtp, name, flags, callback, cookie);
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003424
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003425 if ((done == FAIL || (flags & DIP_ALL)) && (flags & DIP_START))
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003426 {
Bram Moolenaar1c8b4ed2016-03-17 21:51:03 +01003427 len = (int)(STRLEN(start_dir) + STRLEN(name));
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003428 s = alloc(len);
3429 if (s == NULL)
3430 return FAIL;
3431 vim_snprintf((char *)s, len, start_dir, name);
3432 done = do_in_path(p_pp, s, flags, callback, cookie);
3433 vim_free(s);
3434 }
3435
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003436 if ((done == FAIL || (flags & DIP_ALL)) && (flags & DIP_OPT))
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003437 {
Bram Moolenaar1c8b4ed2016-03-17 21:51:03 +01003438 len = (int)(STRLEN(opt_dir) + STRLEN(name));
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003439 s = alloc(len);
3440 if (s == NULL)
3441 return FAIL;
3442 vim_snprintf((char *)s, len, opt_dir, name);
3443 done = do_in_path(p_pp, s, flags, callback, cookie);
3444 vim_free(s);
3445 }
3446
3447 return done;
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003448}
3449
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003450/*
Bram Moolenaarf3654822016-03-04 22:12:23 +01003451 * Expand wildcards in "pat" and invoke do_source() for each match.
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003452 */
3453 static void
Bram Moolenaarf3654822016-03-04 22:12:23 +01003454source_all_matches(char_u *pat)
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003455{
Bram Moolenaarf3654822016-03-04 22:12:23 +01003456 int num_files;
3457 char_u **files;
3458 int i;
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003459
Bram Moolenaarf3654822016-03-04 22:12:23 +01003460 if (gen_expand_wildcards(1, &pat, &num_files, &files, EW_FILE) == OK)
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003461 {
Bram Moolenaarf3654822016-03-04 22:12:23 +01003462 for (i = 0; i < num_files; ++i)
3463 (void)do_source(files[i], FALSE, DOSO_NONE);
3464 FreeWild(num_files, files);
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003465 }
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003466}
3467
Bram Moolenaar49b27322016-04-05 21:13:00 +02003468/* used for "cookie" of add_pack_plugin() */
3469static int APP_ADD_DIR;
3470static int APP_LOAD;
3471static int APP_BOTH;
3472
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003473 static void
Bram Moolenaar91715872016-03-03 17:13:03 +01003474add_pack_plugin(char_u *fname, void *cookie)
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003475{
Bram Moolenaarf3654822016-03-04 22:12:23 +01003476 char_u *p4, *p3, *p2, *p1, *p;
3477 char_u *insp;
Bram Moolenaar91715872016-03-03 17:13:03 +01003478 int c;
3479 char_u *new_rtp;
3480 int keep;
Bram Moolenaarb0550662016-05-31 21:37:36 +02003481 size_t oldlen;
3482 size_t addlen;
Bram Moolenaara5702442016-05-24 19:37:29 +02003483 char_u *afterdir;
Bram Moolenaarb0550662016-05-31 21:37:36 +02003484 size_t afterlen = 0;
Bram Moolenaar91715872016-03-03 17:13:03 +01003485 char_u *ffname = fix_fname(fname);
Bram Moolenaarfef524b2016-07-02 22:07:22 +02003486 size_t fname_len;
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003487
Bram Moolenaar91715872016-03-03 17:13:03 +01003488 if (ffname == NULL)
3489 return;
Bram Moolenaar49b27322016-04-05 21:13:00 +02003490 if (cookie != &APP_LOAD && strstr((char *)p_rtp, (char *)ffname) == NULL)
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003491 {
Bram Moolenaara5702442016-05-24 19:37:29 +02003492 /* directory is not yet in 'runtimepath', add it */
Bram Moolenaarf3654822016-03-04 22:12:23 +01003493 p4 = p3 = p2 = p1 = get_past_head(ffname);
3494 for (p = p1; *p; mb_ptr_adv(p))
3495 if (vim_ispathsep_nocolon(*p))
3496 {
3497 p4 = p3; p3 = p2; p2 = p1; p1 = p;
3498 }
3499
3500 /* now we have:
Bram Moolenaaraf1a0e32016-03-09 22:19:26 +01003501 * rtp/pack/name/start/name
3502 * p4 p3 p2 p1
Bram Moolenaarf3654822016-03-04 22:12:23 +01003503 *
3504 * find the part up to "pack" in 'runtimepath' */
3505 c = *p4;
3506 *p4 = NUL;
Bram Moolenaar4c5717e2016-07-01 15:39:40 +02003507
3508 /* Find "ffname" in "p_rtp", ignoring '/' vs '\' differences. */
3509 fname_len = STRLEN(ffname);
3510 insp = p_rtp;
3511 for (;;)
3512 {
3513 if (vim_fnamencmp(insp, ffname, fname_len) == 0)
3514 break;
3515 insp = vim_strchr(insp, ',');
3516 if (insp == NULL)
3517 break;
3518 ++insp;
3519 }
3520
Bram Moolenaarf3654822016-03-04 22:12:23 +01003521 if (insp == NULL)
3522 /* not found, append at the end */
3523 insp = p_rtp + STRLEN(p_rtp);
3524 else
3525 {
3526 /* append after the matching directory. */
3527 insp += STRLEN(ffname);
3528 while (*insp != NUL && *insp != ',')
3529 ++insp;
3530 }
3531 *p4 = c;
3532
Bram Moolenaara5702442016-05-24 19:37:29 +02003533 /* check if rtp/pack/name/start/name/after exists */
3534 afterdir = concat_fnames(ffname, (char_u *)"after", TRUE);
3535 if (afterdir != NULL && mch_isdir(afterdir))
3536 afterlen = STRLEN(afterdir) + 1; /* add one for comma */
3537
Bram Moolenaarb0550662016-05-31 21:37:36 +02003538 oldlen = STRLEN(p_rtp);
3539 addlen = STRLEN(ffname) + 1; /* add one for comma */
3540 new_rtp = alloc((int)(oldlen + addlen + afterlen + 1)); /* add one for NUL */
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003541 if (new_rtp == NULL)
Bram Moolenaarf3654822016-03-04 22:12:23 +01003542 goto theend;
3543 keep = (int)(insp - p_rtp);
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003544 mch_memmove(new_rtp, p_rtp, keep);
3545 new_rtp[keep] = ',';
Bram Moolenaara5702442016-05-24 19:37:29 +02003546 mch_memmove(new_rtp + keep + 1, ffname, addlen);
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003547 if (p_rtp[keep] != NUL)
Bram Moolenaara5702442016-05-24 19:37:29 +02003548 mch_memmove(new_rtp + keep + addlen, p_rtp + keep,
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003549 oldlen - keep + 1);
Bram Moolenaara5702442016-05-24 19:37:29 +02003550 if (afterlen > 0)
3551 {
3552 STRCAT(new_rtp, ",");
3553 STRCAT(new_rtp, afterdir);
3554 }
Bram Moolenaar863c1a92016-03-03 15:47:06 +01003555 set_option_value((char_u *)"rtp", 0L, new_rtp, 0);
3556 vim_free(new_rtp);
Bram Moolenaara5702442016-05-24 19:37:29 +02003557 vim_free(afterdir);
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003558 }
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003559
Bram Moolenaar49b27322016-04-05 21:13:00 +02003560 if (cookie != &APP_ADD_DIR)
Bram Moolenaarf3654822016-03-04 22:12:23 +01003561 {
Bram Moolenaar71fb0c12016-04-02 22:44:16 +02003562 static char *plugpat = "%s/plugin/**/*.vim";
Bram Moolenaarf3654822016-03-04 22:12:23 +01003563 static char *ftpat = "%s/ftdetect/*.vim";
3564 int len;
3565 char_u *pat;
3566
3567 len = (int)STRLEN(ffname) + (int)STRLEN(ftpat);
3568 pat = alloc(len);
3569 if (pat == NULL)
3570 goto theend;
3571 vim_snprintf((char *)pat, len, plugpat, ffname);
3572 source_all_matches(pat);
3573
3574#ifdef FEAT_AUTOCMD
3575 {
3576 char_u *cmd = vim_strsave((char_u *)"g:did_load_filetypes");
3577
3578 /* If runtime/filetype.vim wasn't loaded yet, the scripts will be
3579 * found when it loads. */
3580 if (cmd != NULL && eval_to_number(cmd) > 0)
3581 {
3582 do_cmdline_cmd((char_u *)"augroup filetypedetect");
3583 vim_snprintf((char *)pat, len, ftpat, ffname);
3584 source_all_matches(pat);
3585 do_cmdline_cmd((char_u *)"augroup END");
3586 }
3587 vim_free(cmd);
3588 }
3589#endif
Bram Moolenaarba8cd122016-03-19 14:16:39 +01003590 vim_free(pat);
Bram Moolenaarf3654822016-03-04 22:12:23 +01003591 }
3592
3593theend:
3594 vim_free(ffname);
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003595}
3596
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003597static int did_source_packages = FALSE;
3598
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003599/*
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003600 * ":packloadall"
Bram Moolenaarf3654822016-03-04 22:12:23 +01003601 * Find plugins in the package directories and source them.
Bram Moolenaar66459b72016-08-06 19:01:55 +02003602 * "eap" is NULL when invoked during startup.
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003603 */
3604 void
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003605ex_packloadall(exarg_T *eap)
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003606{
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003607 if (!did_source_packages || (eap != NULL && eap->forceit))
3608 {
3609 did_source_packages = TRUE;
Bram Moolenaar49b27322016-04-05 21:13:00 +02003610
3611 /* First do a round to add all directories to 'runtimepath', then load
3612 * the plugins. This allows for plugins to use an autoload directory
3613 * of another plugin. */
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003614 do_in_path(p_pp, (char_u *)"pack/*/start/*", DIP_ALL + DIP_DIR,
Bram Moolenaar49b27322016-04-05 21:13:00 +02003615 add_pack_plugin, &APP_ADD_DIR);
3616 do_in_path(p_pp, (char_u *)"pack/*/start/*", DIP_ALL + DIP_DIR,
3617 add_pack_plugin, &APP_LOAD);
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003618 }
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003619}
3620
3621/*
Bram Moolenaarf3654822016-03-04 22:12:23 +01003622 * ":packadd[!] {name}"
Bram Moolenaar91715872016-03-03 17:13:03 +01003623 */
3624 void
3625ex_packadd(exarg_T *eap)
3626{
3627 static char *plugpat = "pack/*/opt/%s";
3628 int len;
3629 char *pat;
3630
3631 len = (int)STRLEN(plugpat) + (int)STRLEN(eap->arg);
3632 pat = (char *)alloc(len);
3633 if (pat == NULL)
3634 return;
3635 vim_snprintf(pat, len, plugpat, eap->arg);
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003636 do_in_path(p_pp, (char_u *)pat, DIP_ALL + DIP_DIR + DIP_ERR,
Bram Moolenaar49b27322016-04-05 21:13:00 +02003637 add_pack_plugin, eap->forceit ? &APP_ADD_DIR : &APP_BOTH);
Bram Moolenaar91715872016-03-03 17:13:03 +01003638 vim_free(pat);
3639}
3640
Bram Moolenaar071d4272004-06-13 20:20:40 +00003641#if defined(FEAT_EVAL) && defined(FEAT_AUTOCMD)
3642/*
3643 * ":options"
3644 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003645 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003646ex_options(
3647 exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003648{
3649 cmd_source((char_u *)SYS_OPTWIN_FILE, NULL);
3650}
3651#endif
3652
3653/*
3654 * ":source {fname}"
3655 */
3656 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003657ex_source(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003658{
3659#ifdef FEAT_BROWSE
3660 if (cmdmod.browse)
3661 {
3662 char_u *fname = NULL;
3663
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00003664 fname = do_browse(0, (char_u *)_("Source Vim script"), eap->arg,
Bram Moolenaar071d4272004-06-13 20:20:40 +00003665 NULL, NULL, BROWSE_FILTER_MACROS, NULL);
3666 if (fname != NULL)
3667 {
3668 cmd_source(fname, eap);
3669 vim_free(fname);
3670 }
3671 }
3672 else
3673#endif
3674 cmd_source(eap->arg, eap);
3675}
3676
3677 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003678cmd_source(char_u *fname, exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003679{
3680 if (*fname == NUL)
3681 EMSG(_(e_argreq));
3682
Bram Moolenaar071d4272004-06-13 20:20:40 +00003683 else if (eap != NULL && eap->forceit)
Bram Moolenaar84a05ac2013-05-06 04:24:17 +02003684 /* ":source!": read Normal mode commands
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00003685 * Need to execute the commands directly. This is required at least
3686 * for:
Bram Moolenaar071d4272004-06-13 20:20:40 +00003687 * - ":g" command busy
3688 * - after ":argdo", ":windo" or ":bufdo"
3689 * - another command follows
3690 * - inside a loop
3691 */
3692 openscript(fname, global_busy || listcmd_busy || eap->nextcmd != NULL
3693#ifdef FEAT_EVAL
3694 || eap->cstack->cs_idx >= 0
3695#endif
3696 );
3697
3698 /* ":source" read ex commands */
Bram Moolenaar910f66f2006-04-05 20:41:53 +00003699 else if (do_source(fname, FALSE, DOSO_NONE) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003700 EMSG2(_(e_notopen), fname);
3701}
3702
3703/*
3704 * ":source" and associated commands.
3705 */
3706/*
3707 * Structure used to store info for each sourced file.
3708 * It is shared between do_source() and getsourceline().
3709 * This is required, because it needs to be handed to do_cmdline() and
3710 * sourcing can be done recursively.
3711 */
3712struct source_cookie
3713{
3714 FILE *fp; /* opened file for sourcing */
3715 char_u *nextline; /* if not NULL: line that was read ahead */
3716 int finished; /* ":finish" used */
Bram Moolenaar860cae12010-06-05 23:22:07 +02003717#if defined(USE_CRNL) || defined(USE_CR)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003718 int fileformat; /* EOL_UNKNOWN, EOL_UNIX or EOL_DOS */
3719 int error; /* TRUE if LF found after CR-LF */
3720#endif
3721#ifdef FEAT_EVAL
3722 linenr_T breakpoint; /* next line with breakpoint or zero */
3723 char_u *fname; /* name of sourced file */
3724 int dbg_tick; /* debug_tick when breakpoint was set */
3725 int level; /* top nesting level of sourced file */
3726#endif
3727#ifdef FEAT_MBYTE
3728 vimconv_T conv; /* type of conversion */
3729#endif
3730};
3731
3732#ifdef FEAT_EVAL
3733/*
3734 * Return the address holding the next breakpoint line for a source cookie.
3735 */
3736 linenr_T *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003737source_breakpoint(void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003738{
3739 return &((struct source_cookie *)cookie)->breakpoint;
3740}
3741
3742/*
3743 * Return the address holding the debug tick for a source cookie.
3744 */
3745 int *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003746source_dbg_tick(void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003747{
3748 return &((struct source_cookie *)cookie)->dbg_tick;
3749}
3750
3751/*
3752 * Return the nesting level for a source cookie.
3753 */
3754 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003755source_level(void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003756{
3757 return ((struct source_cookie *)cookie)->level;
3758}
3759#endif
3760
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01003761static char_u *get_one_sourceline(struct source_cookie *sp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003762
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01003763#if (defined(WIN32) && defined(FEAT_CSCOPE)) || defined(HAVE_FD_CLOEXEC)
3764# define USE_FOPEN_NOINH
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01003765static FILE *fopen_noinh_readbin(char *filename);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003766
3767/*
3768 * Special function to open a file without handle inheritance.
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01003769 * When possible the handle is closed on exec().
Bram Moolenaar071d4272004-06-13 20:20:40 +00003770 */
3771 static FILE *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003772fopen_noinh_readbin(char *filename)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003773{
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01003774# ifdef WIN32
Bram Moolenaar8d8ef0b2010-01-20 21:41:47 +01003775 int fd_tmp = mch_open(filename, O_RDONLY | O_BINARY | O_NOINHERIT, 0);
3776# else
3777 int fd_tmp = mch_open(filename, O_RDONLY, 0);
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01003778# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003779
3780 if (fd_tmp == -1)
3781 return NULL;
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01003782
3783# ifdef HAVE_FD_CLOEXEC
3784 {
3785 int fdflags = fcntl(fd_tmp, F_GETFD);
3786 if (fdflags >= 0 && (fdflags & FD_CLOEXEC) == 0)
Bram Moolenaarcde88542015-08-11 19:14:00 +02003787 (void)fcntl(fd_tmp, F_SETFD, fdflags | FD_CLOEXEC);
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01003788 }
3789# endif
3790
Bram Moolenaar071d4272004-06-13 20:20:40 +00003791 return fdopen(fd_tmp, READBIN);
3792}
3793#endif
3794
3795
3796/*
3797 * do_source: Read the file "fname" and execute its lines as EX commands.
3798 *
3799 * This function may be called recursively!
3800 *
3801 * return FAIL if file could not be opened, OK otherwise
3802 */
3803 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003804do_source(
3805 char_u *fname,
3806 int check_other, /* check for .vimrc and _vimrc */
3807 int is_vimrc) /* DOSO_ value */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003808{
3809 struct source_cookie cookie;
3810 char_u *save_sourcing_name;
3811 linenr_T save_sourcing_lnum;
3812 char_u *p;
3813 char_u *fname_exp;
Bram Moolenaar73881402009-02-04 16:50:47 +00003814 char_u *firstline = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003815 int retval = FAIL;
3816#ifdef FEAT_EVAL
3817 scid_T save_current_SID;
3818 static scid_T last_current_SID = 0;
3819 void *save_funccalp;
3820 int save_debug_break_level = debug_break_level;
Bram Moolenaar05159a02005-02-26 23:04:13 +00003821 scriptitem_T *si = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003822# ifdef UNIX
Bram Moolenaar8767f522016-07-01 17:17:39 +02003823 stat_T st;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003824 int stat_ok;
3825# endif
3826#endif
3827#ifdef STARTUPTIME
3828 struct timeval tv_rel;
3829 struct timeval tv_start;
3830#endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00003831#ifdef FEAT_PROFILE
3832 proftime_T wait_start;
3833#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003834
Bram Moolenaar071d4272004-06-13 20:20:40 +00003835 p = expand_env_save(fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003836 if (p == NULL)
3837 return retval;
3838 fname_exp = fix_fname(p);
3839 vim_free(p);
3840 if (fname_exp == NULL)
3841 return retval;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003842 if (mch_isdir(fname_exp))
3843 {
Bram Moolenaar555b2802005-05-19 21:08:39 +00003844 smsg((char_u *)_("Cannot source a directory: \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003845 goto theend;
3846 }
3847
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00003848#ifdef FEAT_AUTOCMD
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00003849 /* Apply SourceCmd autocommands, they should get the file and source it. */
3850 if (has_autocmd(EVENT_SOURCECMD, fname_exp, NULL)
3851 && apply_autocmds(EVENT_SOURCECMD, fname_exp, fname_exp,
3852 FALSE, curbuf))
Bram Moolenaarf33943e2008-01-15 21:17:30 +00003853 {
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00003854# ifdef FEAT_EVAL
Bram Moolenaarf33943e2008-01-15 21:17:30 +00003855 retval = aborting() ? FAIL : OK;
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00003856# else
Bram Moolenaarf33943e2008-01-15 21:17:30 +00003857 retval = OK;
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00003858# endif
Bram Moolenaarf33943e2008-01-15 21:17:30 +00003859 goto theend;
3860 }
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00003861
3862 /* Apply SourcePre autocommands, they may get the file. */
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00003863 apply_autocmds(EVENT_SOURCEPRE, fname_exp, fname_exp, FALSE, curbuf);
3864#endif
3865
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01003866#ifdef USE_FOPEN_NOINH
Bram Moolenaar071d4272004-06-13 20:20:40 +00003867 cookie.fp = fopen_noinh_readbin((char *)fname_exp);
3868#else
3869 cookie.fp = mch_fopen((char *)fname_exp, READBIN);
3870#endif
3871 if (cookie.fp == NULL && check_other)
3872 {
3873 /*
3874 * Try again, replacing file name ".vimrc" by "_vimrc" or vice versa,
3875 * and ".exrc" by "_exrc" or vice versa.
3876 */
3877 p = gettail(fname_exp);
3878 if ((*p == '.' || *p == '_')
3879 && (STRICMP(p + 1, "vimrc") == 0
3880 || STRICMP(p + 1, "gvimrc") == 0
3881 || STRICMP(p + 1, "exrc") == 0))
3882 {
3883 if (*p == '_')
3884 *p = '.';
3885 else
3886 *p = '_';
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01003887#ifdef USE_FOPEN_NOINH
Bram Moolenaar071d4272004-06-13 20:20:40 +00003888 cookie.fp = fopen_noinh_readbin((char *)fname_exp);
3889#else
3890 cookie.fp = mch_fopen((char *)fname_exp, READBIN);
3891#endif
3892 }
3893 }
3894
3895 if (cookie.fp == NULL)
3896 {
3897 if (p_verbose > 0)
3898 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003899 verbose_enter();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003900 if (sourcing_name == NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00003901 smsg((char_u *)_("could not source \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003902 else
3903 smsg((char_u *)_("line %ld: could not source \"%s\""),
Bram Moolenaar555b2802005-05-19 21:08:39 +00003904 sourcing_lnum, fname);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003905 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003906 }
3907 goto theend;
3908 }
3909
3910 /*
3911 * The file exists.
3912 * - In verbose mode, give a message.
3913 * - For a vimrc file, may want to set 'compatible', call vimrc_found().
3914 */
3915 if (p_verbose > 1)
3916 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003917 verbose_enter();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003918 if (sourcing_name == NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00003919 smsg((char_u *)_("sourcing \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003920 else
3921 smsg((char_u *)_("line %ld: sourcing \"%s\""),
Bram Moolenaar555b2802005-05-19 21:08:39 +00003922 sourcing_lnum, fname);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003923 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003924 }
Bram Moolenaar910f66f2006-04-05 20:41:53 +00003925 if (is_vimrc == DOSO_VIMRC)
3926 vimrc_found(fname_exp, (char_u *)"MYVIMRC");
3927 else if (is_vimrc == DOSO_GVIMRC)
3928 vimrc_found(fname_exp, (char_u *)"MYGVIMRC");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003929
3930#ifdef USE_CRNL
3931 /* If no automatic file format: Set default to CR-NL. */
3932 if (*p_ffs == NUL)
3933 cookie.fileformat = EOL_DOS;
3934 else
3935 cookie.fileformat = EOL_UNKNOWN;
3936 cookie.error = FALSE;
3937#endif
3938
3939#ifdef USE_CR
3940 /* If no automatic file format: Set default to CR. */
3941 if (*p_ffs == NUL)
3942 cookie.fileformat = EOL_MAC;
3943 else
3944 cookie.fileformat = EOL_UNKNOWN;
3945 cookie.error = FALSE;
3946#endif
3947
3948 cookie.nextline = NULL;
3949 cookie.finished = FALSE;
3950
3951#ifdef FEAT_EVAL
3952 /*
3953 * Check if this script has a breakpoint.
3954 */
3955 cookie.breakpoint = dbg_find_breakpoint(TRUE, fname_exp, (linenr_T)0);
3956 cookie.fname = fname_exp;
3957 cookie.dbg_tick = debug_tick;
3958
3959 cookie.level = ex_nesting_level;
3960#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003961
3962 /*
3963 * Keep the sourcing name/lnum, for recursive calls.
3964 */
3965 save_sourcing_name = sourcing_name;
3966 sourcing_name = fname_exp;
3967 save_sourcing_lnum = sourcing_lnum;
3968 sourcing_lnum = 0;
3969
Bram Moolenaar73881402009-02-04 16:50:47 +00003970#ifdef FEAT_MBYTE
3971 cookie.conv.vc_type = CONV_NONE; /* no conversion */
3972
3973 /* Read the first line so we can check for a UTF-8 BOM. */
3974 firstline = getsourceline(0, (void *)&cookie, 0);
3975 if (firstline != NULL && STRLEN(firstline) >= 3 && firstline[0] == 0xef
3976 && firstline[1] == 0xbb && firstline[2] == 0xbf)
3977 {
3978 /* Found BOM; setup conversion, skip over BOM and recode the line. */
3979 convert_setup(&cookie.conv, (char_u *)"utf-8", p_enc);
3980 p = string_convert(&cookie.conv, firstline + 3, NULL);
Bram Moolenaar4e2a5952009-02-05 19:48:25 +00003981 if (p == NULL)
3982 p = vim_strsave(firstline + 3);
Bram Moolenaar73881402009-02-04 16:50:47 +00003983 if (p != NULL)
3984 {
3985 vim_free(firstline);
3986 firstline = p;
3987 }
3988 }
3989#endif
3990
Bram Moolenaar071d4272004-06-13 20:20:40 +00003991#ifdef STARTUPTIME
Bram Moolenaarc4e41982010-01-19 16:31:47 +01003992 if (time_fd != NULL)
3993 time_push(&tv_rel, &tv_start);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003994#endif
3995
3996#ifdef FEAT_EVAL
Bram Moolenaar05159a02005-02-26 23:04:13 +00003997# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00003998 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003999 prof_child_enter(&wait_start); /* entering a child now */
4000# endif
4001
4002 /* Don't use local function variables, if called from a function.
4003 * Also starts profiling timer for nested script. */
4004 save_funccalp = save_funccal();
4005
Bram Moolenaar071d4272004-06-13 20:20:40 +00004006 /*
4007 * Check if this script was sourced before to finds its SID.
4008 * If it's new, generate a new SID.
4009 */
4010 save_current_SID = current_SID;
4011# ifdef UNIX
4012 stat_ok = (mch_stat((char *)fname_exp, &st) >= 0);
4013# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00004014 for (current_SID = script_items.ga_len; current_SID > 0; --current_SID)
4015 {
4016 si = &SCRIPT_ITEM(current_SID);
4017 if (si->sn_name != NULL
Bram Moolenaar071d4272004-06-13 20:20:40 +00004018 && (
4019# ifdef UNIX
Bram Moolenaar7c626922005-02-07 22:01:03 +00004020 /* Compare dev/ino when possible, it catches symbolic
4021 * links. Also compare file names, the inode may change
4022 * when the file was edited. */
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00004023 ((stat_ok && si->sn_dev_valid)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004024 && (si->sn_dev == st.st_dev
4025 && si->sn_ino == st.st_ino)) ||
Bram Moolenaar071d4272004-06-13 20:20:40 +00004026# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00004027 fnamecmp(si->sn_name, fname_exp) == 0))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004028 break;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004029 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004030 if (current_SID == 0)
4031 {
4032 current_SID = ++last_current_SID;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004033 if (ga_grow(&script_items, (int)(current_SID - script_items.ga_len))
4034 == FAIL)
4035 goto almosttheend;
4036 while (script_items.ga_len < current_SID)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004037 {
Bram Moolenaar05159a02005-02-26 23:04:13 +00004038 ++script_items.ga_len;
4039 SCRIPT_ITEM(script_items.ga_len).sn_name = NULL;
4040# ifdef FEAT_PROFILE
4041 SCRIPT_ITEM(script_items.ga_len).sn_prof_on = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004042# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004043 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00004044 si = &SCRIPT_ITEM(current_SID);
4045 si->sn_name = fname_exp;
4046 fname_exp = NULL;
4047# ifdef UNIX
4048 if (stat_ok)
4049 {
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00004050 si->sn_dev_valid = TRUE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004051 si->sn_dev = st.st_dev;
4052 si->sn_ino = st.st_ino;
4053 }
4054 else
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00004055 si->sn_dev_valid = FALSE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004056# endif
4057
Bram Moolenaar071d4272004-06-13 20:20:40 +00004058 /* Allocate the local script variables to use for this script. */
4059 new_script_vars(current_SID);
4060 }
4061
Bram Moolenaar05159a02005-02-26 23:04:13 +00004062# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004063 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004064 {
4065 int forceit;
4066
4067 /* Check if we do profiling for this script. */
4068 if (!si->sn_prof_on && has_profiling(TRUE, si->sn_name, &forceit))
4069 {
4070 script_do_profile(si);
4071 si->sn_pr_force = forceit;
4072 }
4073 if (si->sn_prof_on)
4074 {
4075 ++si->sn_pr_count;
4076 profile_start(&si->sn_pr_start);
4077 profile_zero(&si->sn_pr_children);
4078 }
4079 }
4080# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004081#endif
4082
4083 /*
4084 * Call do_cmdline, which will call getsourceline() to get the lines.
4085 */
Bram Moolenaar73881402009-02-04 16:50:47 +00004086 do_cmdline(firstline, getsourceline, (void *)&cookie,
Bram Moolenaar071d4272004-06-13 20:20:40 +00004087 DOCMD_VERBOSE|DOCMD_NOWAIT|DOCMD_REPEAT);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004088 retval = OK;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004089
4090#ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004091 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004092 {
4093 /* Get "si" again, "script_items" may have been reallocated. */
4094 si = &SCRIPT_ITEM(current_SID);
4095 if (si->sn_prof_on)
4096 {
4097 profile_end(&si->sn_pr_start);
4098 profile_sub_wait(&wait_start, &si->sn_pr_start);
4099 profile_add(&si->sn_pr_total, &si->sn_pr_start);
Bram Moolenaar1056d982006-03-09 22:37:52 +00004100 profile_self(&si->sn_pr_self, &si->sn_pr_start,
4101 &si->sn_pr_children);
Bram Moolenaar05159a02005-02-26 23:04:13 +00004102 }
4103 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004104#endif
4105
4106 if (got_int)
4107 EMSG(_(e_interr));
4108 sourcing_name = save_sourcing_name;
4109 sourcing_lnum = save_sourcing_lnum;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004110 if (p_verbose > 1)
4111 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004112 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00004113 smsg((char_u *)_("finished sourcing %s"), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004114 if (sourcing_name != NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00004115 smsg((char_u *)_("continuing in %s"), sourcing_name);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004116 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004117 }
4118#ifdef STARTUPTIME
Bram Moolenaarc4e41982010-01-19 16:31:47 +01004119 if (time_fd != NULL)
4120 {
4121 vim_snprintf((char *)IObuff, IOSIZE, "sourcing %s", fname);
4122 time_msg((char *)IObuff, &tv_start);
4123 time_pop(&tv_rel);
4124 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004125#endif
4126
4127#ifdef FEAT_EVAL
4128 /*
4129 * After a "finish" in debug mode, need to break at first command of next
4130 * sourced file.
4131 */
4132 if (save_debug_break_level > ex_nesting_level
4133 && debug_break_level == ex_nesting_level)
4134 ++debug_break_level;
4135#endif
4136
Bram Moolenaar05159a02005-02-26 23:04:13 +00004137#ifdef FEAT_EVAL
4138almosttheend:
4139 current_SID = save_current_SID;
4140 restore_funccal(save_funccalp);
4141# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004142 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004143 prof_child_exit(&wait_start); /* leaving a child now */
4144# endif
4145#endif
4146 fclose(cookie.fp);
4147 vim_free(cookie.nextline);
Bram Moolenaar73881402009-02-04 16:50:47 +00004148 vim_free(firstline);
Bram Moolenaar05159a02005-02-26 23:04:13 +00004149#ifdef FEAT_MBYTE
4150 convert_setup(&cookie.conv, NULL, NULL);
4151#endif
4152
Bram Moolenaar071d4272004-06-13 20:20:40 +00004153theend:
4154 vim_free(fname_exp);
4155 return retval;
4156}
4157
4158#if defined(FEAT_EVAL) || defined(PROTO)
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00004159
Bram Moolenaar071d4272004-06-13 20:20:40 +00004160/*
4161 * ":scriptnames"
4162 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004163 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004164ex_scriptnames(exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004165{
4166 int i;
4167
Bram Moolenaar05159a02005-02-26 23:04:13 +00004168 for (i = 1; i <= script_items.ga_len && !got_int; ++i)
4169 if (SCRIPT_ITEM(i).sn_name != NULL)
Bram Moolenaard58ea072011-06-26 04:25:30 +02004170 {
4171 home_replace(NULL, SCRIPT_ITEM(i).sn_name,
4172 NameBuff, MAXPATHL, TRUE);
4173 smsg((char_u *)"%3d: %s", i, NameBuff);
Bram Moolenaar970a1b82012-03-23 18:39:18 +01004174 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004175}
4176
4177# if defined(BACKSLASH_IN_FILENAME) || defined(PROTO)
4178/*
4179 * Fix slashes in the list of script names for 'shellslash'.
4180 */
4181 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004182scriptnames_slash_adjust(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004183{
4184 int i;
4185
Bram Moolenaar05159a02005-02-26 23:04:13 +00004186 for (i = 1; i <= script_items.ga_len; ++i)
4187 if (SCRIPT_ITEM(i).sn_name != NULL)
4188 slash_adjust(SCRIPT_ITEM(i).sn_name);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004189}
4190# endif
4191
4192/*
4193 * Get a pointer to a script name. Used for ":verbose set".
4194 */
4195 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004196get_scriptname(scid_T id)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004197{
4198 if (id == SID_MODELINE)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00004199 return (char_u *)_("modeline");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004200 if (id == SID_CMDARG)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00004201 return (char_u *)_("--cmd argument");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004202 if (id == SID_CARG)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00004203 return (char_u *)_("-c argument");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004204 if (id == SID_ENV)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00004205 return (char_u *)_("environment variable");
4206 if (id == SID_ERROR)
4207 return (char_u *)_("error handler");
Bram Moolenaar05159a02005-02-26 23:04:13 +00004208 return SCRIPT_ITEM(id).sn_name;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004209}
Bram Moolenaar05159a02005-02-26 23:04:13 +00004210
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00004211# if defined(EXITFREE) || defined(PROTO)
4212 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004213free_scriptnames(void)
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00004214{
4215 int i;
4216
4217 for (i = script_items.ga_len; i > 0; --i)
4218 vim_free(SCRIPT_ITEM(i).sn_name);
4219 ga_clear(&script_items);
4220}
4221# endif
4222
Bram Moolenaar071d4272004-06-13 20:20:40 +00004223#endif
4224
4225#if defined(USE_CR) || defined(PROTO)
4226
4227# if defined(__MSL__) && (__MSL__ >= 22)
4228/*
4229 * Newer version of the Metrowerks library handle DOS and UNIX files
4230 * without help.
4231 * Test with earlier versions, MSL 2.2 is the library supplied with
4232 * Codewarrior Pro 2.
4233 */
4234 char *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004235fgets_cr(char *s, int n, FILE *stream)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004236{
4237 return fgets(s, n, stream);
4238}
4239# else
4240/*
4241 * Version of fgets() which also works for lines ending in a <CR> only
4242 * (Macintosh format).
4243 * For older versions of the Metrowerks library.
4244 * At least CodeWarrior 9 needed this code.
4245 */
4246 char *
Bram Moolenaard14e00e2016-01-31 17:30:51 +01004247fgets_cr(char *s, int n, FILE *stream)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004248{
4249 int c = 0;
4250 int char_read = 0;
4251
4252 while (!feof(stream) && c != '\r' && c != '\n' && char_read < n - 1)
4253 {
4254 c = fgetc(stream);
4255 s[char_read++] = c;
4256 /* If the file is in DOS format, we need to skip a NL after a CR. I
4257 * thought it was the other way around, but this appears to work... */
4258 if (c == '\n')
4259 {
4260 c = fgetc(stream);
4261 if (c != '\r')
4262 ungetc(c, stream);
4263 }
4264 }
4265
4266 s[char_read] = 0;
4267 if (char_read == 0)
4268 return NULL;
4269
4270 if (feof(stream) && char_read == 1)
4271 return NULL;
4272
4273 return s;
4274}
4275# endif
4276#endif
4277
4278/*
4279 * Get one full line from a sourced file.
4280 * Called by do_cmdline() when it's called from do_source().
4281 *
4282 * Return a pointer to the line in allocated memory.
4283 * Return NULL for end-of-file or some error.
4284 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004285 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004286getsourceline(int c UNUSED, void *cookie, int indent UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004287{
4288 struct source_cookie *sp = (struct source_cookie *)cookie;
4289 char_u *line;
Bram Moolenaarc047b9a2012-02-11 20:40:55 +01004290 char_u *p;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004291
4292#ifdef FEAT_EVAL
4293 /* If breakpoints have been added/deleted need to check for it. */
4294 if (sp->dbg_tick < debug_tick)
4295 {
4296 sp->breakpoint = dbg_find_breakpoint(TRUE, sp->fname, sourcing_lnum);
4297 sp->dbg_tick = debug_tick;
4298 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00004299# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004300 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004301 script_line_end();
4302# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004303#endif
4304 /*
4305 * Get current line. If there is a read-ahead line, use it, otherwise get
4306 * one now.
4307 */
4308 if (sp->finished)
4309 line = NULL;
4310 else if (sp->nextline == NULL)
4311 line = get_one_sourceline(sp);
4312 else
4313 {
4314 line = sp->nextline;
4315 sp->nextline = NULL;
4316 ++sourcing_lnum;
4317 }
Bram Moolenaare2cc9702005-03-15 22:43:58 +00004318#ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004319 if (line != NULL && do_profiling == PROF_YES)
Bram Moolenaare2cc9702005-03-15 22:43:58 +00004320 script_line_start();
4321#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004322
4323 /* Only concatenate lines starting with a \ when 'cpoptions' doesn't
4324 * contain the 'C' flag. */
4325 if (line != NULL && (vim_strchr(p_cpo, CPO_CONCAT) == NULL))
4326 {
4327 /* compensate for the one line read-ahead */
4328 --sourcing_lnum;
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004329
4330 /* Get the next line and concatenate it when it starts with a
4331 * backslash. We always need to read the next line, keep it in
4332 * sp->nextline. */
4333 sp->nextline = get_one_sourceline(sp);
4334 if (sp->nextline != NULL && *(p = skipwhite(sp->nextline)) == '\\')
Bram Moolenaar071d4272004-06-13 20:20:40 +00004335 {
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004336 garray_T ga;
4337
Bram Moolenaarb549a732012-02-22 18:29:33 +01004338 ga_init2(&ga, (int)sizeof(char_u), 400);
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004339 ga_concat(&ga, line);
4340 ga_concat(&ga, p + 1);
4341 for (;;)
4342 {
4343 vim_free(sp->nextline);
4344 sp->nextline = get_one_sourceline(sp);
4345 if (sp->nextline == NULL)
4346 break;
4347 p = skipwhite(sp->nextline);
4348 if (*p != '\\')
4349 break;
Bram Moolenaarb549a732012-02-22 18:29:33 +01004350 /* Adjust the growsize to the current length to speed up
4351 * concatenating many lines. */
4352 if (ga.ga_len > 400)
4353 {
4354 if (ga.ga_len > 8000)
4355 ga.ga_growsize = 8000;
4356 else
4357 ga.ga_growsize = ga.ga_len;
4358 }
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004359 ga_concat(&ga, p + 1);
4360 }
4361 ga_append(&ga, NUL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004362 vim_free(line);
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004363 line = ga.ga_data;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004364 }
4365 }
4366
4367#ifdef FEAT_MBYTE
4368 if (line != NULL && sp->conv.vc_type != CONV_NONE)
4369 {
Bram Moolenaarc047b9a2012-02-11 20:40:55 +01004370 char_u *s;
4371
Bram Moolenaar071d4272004-06-13 20:20:40 +00004372 /* Convert the encoding of the script line. */
4373 s = string_convert(&sp->conv, line, NULL);
4374 if (s != NULL)
4375 {
4376 vim_free(line);
4377 line = s;
4378 }
4379 }
4380#endif
4381
4382#ifdef FEAT_EVAL
4383 /* Did we encounter a breakpoint? */
4384 if (sp->breakpoint != 0 && sp->breakpoint <= sourcing_lnum)
4385 {
4386 dbg_breakpoint(sp->fname, sourcing_lnum);
4387 /* Find next breakpoint. */
4388 sp->breakpoint = dbg_find_breakpoint(TRUE, sp->fname, sourcing_lnum);
4389 sp->dbg_tick = debug_tick;
4390 }
4391#endif
4392
4393 return line;
4394}
4395
4396 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004397get_one_sourceline(struct source_cookie *sp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004398{
4399 garray_T ga;
4400 int len;
4401 int c;
4402 char_u *buf;
4403#ifdef USE_CRNL
4404 int has_cr; /* CR-LF found */
4405#endif
4406#ifdef USE_CR
4407 char_u *scan;
4408#endif
4409 int have_read = FALSE;
4410
4411 /* use a growarray to store the sourced line */
Bram Moolenaar6ac54292005-02-02 23:07:25 +00004412 ga_init2(&ga, 1, 250);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004413
4414 /*
4415 * Loop until there is a finished line (or end-of-file).
4416 */
4417 sourcing_lnum++;
4418 for (;;)
4419 {
Bram Moolenaar6ac54292005-02-02 23:07:25 +00004420 /* make room to read at least 120 (more) characters */
4421 if (ga_grow(&ga, 120) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004422 break;
4423 buf = (char_u *)ga.ga_data;
4424
4425#ifdef USE_CR
4426 if (sp->fileformat == EOL_MAC)
4427 {
Bram Moolenaar86b68352004-12-27 21:59:20 +00004428 if (fgets_cr((char *)buf + ga.ga_len, ga.ga_maxlen - ga.ga_len,
4429 sp->fp) == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004430 break;
4431 }
4432 else
4433#endif
Bram Moolenaar86b68352004-12-27 21:59:20 +00004434 if (fgets((char *)buf + ga.ga_len, ga.ga_maxlen - ga.ga_len,
4435 sp->fp) == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004436 break;
Bram Moolenaar6ac54292005-02-02 23:07:25 +00004437 len = ga.ga_len + (int)STRLEN(buf + ga.ga_len);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004438#ifdef USE_CRNL
4439 /* Ignore a trailing CTRL-Z, when in Dos mode. Only recognize the
4440 * CTRL-Z by its own, or after a NL. */
4441 if ( (len == 1 || (len >= 2 && buf[len - 2] == '\n'))
4442 && sp->fileformat == EOL_DOS
4443 && buf[len - 1] == Ctrl_Z)
4444 {
4445 buf[len - 1] = NUL;
4446 break;
4447 }
4448#endif
4449
4450#ifdef USE_CR
4451 /* If the read doesn't stop on a new line, and there's
4452 * some CR then we assume a Mac format */
4453 if (sp->fileformat == EOL_UNKNOWN)
4454 {
4455 if (buf[len - 1] != '\n' && vim_strchr(buf, '\r') != NULL)
4456 sp->fileformat = EOL_MAC;
4457 else
4458 sp->fileformat = EOL_UNIX;
4459 }
4460
4461 if (sp->fileformat == EOL_MAC)
4462 {
4463 scan = vim_strchr(buf, '\r');
4464
4465 if (scan != NULL)
4466 {
4467 *scan = '\n';
4468 if (*(scan + 1) != 0)
4469 {
4470 *(scan + 1) = 0;
4471 fseek(sp->fp, (long)(scan - buf - len + 1), SEEK_CUR);
4472 }
4473 }
4474 len = STRLEN(buf);
4475 }
4476#endif
4477
4478 have_read = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004479 ga.ga_len = len;
4480
4481 /* If the line was longer than the buffer, read more. */
Bram Moolenaar86b68352004-12-27 21:59:20 +00004482 if (ga.ga_maxlen - ga.ga_len == 1 && buf[len - 1] != '\n')
Bram Moolenaar071d4272004-06-13 20:20:40 +00004483 continue;
4484
4485 if (len >= 1 && buf[len - 1] == '\n') /* remove trailing NL */
4486 {
4487#ifdef USE_CRNL
4488 has_cr = (len >= 2 && buf[len - 2] == '\r');
4489 if (sp->fileformat == EOL_UNKNOWN)
4490 {
4491 if (has_cr)
4492 sp->fileformat = EOL_DOS;
4493 else
4494 sp->fileformat = EOL_UNIX;
4495 }
4496
4497 if (sp->fileformat == EOL_DOS)
4498 {
4499 if (has_cr) /* replace trailing CR */
4500 {
4501 buf[len - 2] = '\n';
4502 --len;
4503 --ga.ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004504 }
4505 else /* lines like ":map xx yy^M" will have failed */
4506 {
4507 if (!sp->error)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00004508 {
4509 msg_source(hl_attr(HLF_W));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004510 EMSG(_("W15: Warning: Wrong line separator, ^M may be missing"));
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00004511 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004512 sp->error = TRUE;
4513 sp->fileformat = EOL_UNIX;
4514 }
4515 }
4516#endif
4517 /* The '\n' is escaped if there is an odd number of ^V's just
4518 * before it, first set "c" just before the 'V's and then check
4519 * len&c parities (is faster than ((len-c)%2 == 0)) -- Acevedo */
4520 for (c = len - 2; c >= 0 && buf[c] == Ctrl_V; c--)
4521 ;
4522 if ((len & 1) != (c & 1)) /* escaped NL, read more */
4523 {
4524 sourcing_lnum++;
4525 continue;
4526 }
4527
4528 buf[len - 1] = NUL; /* remove the NL */
4529 }
4530
4531 /*
4532 * Check for ^C here now and then, so recursive :so can be broken.
4533 */
4534 line_breakcheck();
4535 break;
4536 }
4537
4538 if (have_read)
4539 return (char_u *)ga.ga_data;
4540
4541 vim_free(ga.ga_data);
4542 return NULL;
4543}
4544
Bram Moolenaar05159a02005-02-26 23:04:13 +00004545#if defined(FEAT_PROFILE) || defined(PROTO)
4546/*
4547 * Called when starting to read a script line.
4548 * "sourcing_lnum" must be correct!
4549 * When skipping lines it may not actually be executed, but we won't find out
4550 * until later and we need to store the time now.
4551 */
4552 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004553script_line_start(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004554{
4555 scriptitem_T *si;
4556 sn_prl_T *pp;
4557
4558 if (current_SID <= 0 || current_SID > script_items.ga_len)
4559 return;
4560 si = &SCRIPT_ITEM(current_SID);
4561 if (si->sn_prof_on && sourcing_lnum >= 1)
4562 {
Bram Moolenaar8c8de832008-06-24 22:58:06 +00004563 /* Grow the array before starting the timer, so that the time spent
Bram Moolenaar05159a02005-02-26 23:04:13 +00004564 * here isn't counted. */
Bram Moolenaarcde88542015-08-11 19:14:00 +02004565 (void)ga_grow(&si->sn_prl_ga, (int)(sourcing_lnum - si->sn_prl_ga.ga_len));
Bram Moolenaar05159a02005-02-26 23:04:13 +00004566 si->sn_prl_idx = sourcing_lnum - 1;
4567 while (si->sn_prl_ga.ga_len <= si->sn_prl_idx
4568 && si->sn_prl_ga.ga_len < si->sn_prl_ga.ga_maxlen)
4569 {
4570 /* Zero counters for a line that was not used before. */
4571 pp = &PRL_ITEM(si, si->sn_prl_ga.ga_len);
4572 pp->snp_count = 0;
4573 profile_zero(&pp->sn_prl_total);
4574 profile_zero(&pp->sn_prl_self);
4575 ++si->sn_prl_ga.ga_len;
4576 }
4577 si->sn_prl_execed = FALSE;
4578 profile_start(&si->sn_prl_start);
4579 profile_zero(&si->sn_prl_children);
4580 profile_get_wait(&si->sn_prl_wait);
4581 }
4582}
4583
4584/*
4585 * Called when actually executing a function line.
4586 */
4587 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004588script_line_exec(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004589{
4590 scriptitem_T *si;
4591
4592 if (current_SID <= 0 || current_SID > script_items.ga_len)
4593 return;
4594 si = &SCRIPT_ITEM(current_SID);
4595 if (si->sn_prof_on && si->sn_prl_idx >= 0)
4596 si->sn_prl_execed = TRUE;
4597}
4598
4599/*
4600 * Called when done with a function line.
4601 */
4602 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004603script_line_end(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004604{
4605 scriptitem_T *si;
4606 sn_prl_T *pp;
4607
4608 if (current_SID <= 0 || current_SID > script_items.ga_len)
4609 return;
4610 si = &SCRIPT_ITEM(current_SID);
4611 if (si->sn_prof_on && si->sn_prl_idx >= 0
4612 && si->sn_prl_idx < si->sn_prl_ga.ga_len)
4613 {
4614 if (si->sn_prl_execed)
4615 {
4616 pp = &PRL_ITEM(si, si->sn_prl_idx);
4617 ++pp->snp_count;
4618 profile_end(&si->sn_prl_start);
4619 profile_sub_wait(&si->sn_prl_wait, &si->sn_prl_start);
Bram Moolenaar05159a02005-02-26 23:04:13 +00004620 profile_add(&pp->sn_prl_total, &si->sn_prl_start);
Bram Moolenaar1056d982006-03-09 22:37:52 +00004621 profile_self(&pp->sn_prl_self, &si->sn_prl_start,
4622 &si->sn_prl_children);
Bram Moolenaar05159a02005-02-26 23:04:13 +00004623 }
4624 si->sn_prl_idx = -1;
4625 }
4626}
4627#endif
4628
Bram Moolenaar071d4272004-06-13 20:20:40 +00004629/*
4630 * ":scriptencoding": Set encoding conversion for a sourced script.
4631 * Without the multi-byte feature it's simply ignored.
4632 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004633 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004634ex_scriptencoding(exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004635{
4636#ifdef FEAT_MBYTE
4637 struct source_cookie *sp;
4638 char_u *name;
4639
4640 if (!getline_equal(eap->getline, eap->cookie, getsourceline))
4641 {
4642 EMSG(_("E167: :scriptencoding used outside of a sourced file"));
4643 return;
4644 }
4645
4646 if (*eap->arg != NUL)
4647 {
4648 name = enc_canonize(eap->arg);
4649 if (name == NULL) /* out of memory */
4650 return;
4651 }
4652 else
4653 name = eap->arg;
4654
4655 /* Setup for conversion from the specified encoding to 'encoding'. */
4656 sp = (struct source_cookie *)getline_cookie(eap->getline, eap->cookie);
4657 convert_setup(&sp->conv, name, p_enc);
4658
4659 if (name != eap->arg)
4660 vim_free(name);
4661#endif
4662}
4663
4664#if defined(FEAT_EVAL) || defined(PROTO)
4665/*
4666 * ":finish": Mark a sourced file as finished.
4667 */
4668 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004669ex_finish(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004670{
4671 if (getline_equal(eap->getline, eap->cookie, getsourceline))
4672 do_finish(eap, FALSE);
4673 else
4674 EMSG(_("E168: :finish used outside of a sourced file"));
4675}
4676
4677/*
4678 * Mark a sourced file as finished. Possibly makes the ":finish" pending.
4679 * Also called for a pending finish at the ":endtry" or after returning from
4680 * an extra do_cmdline(). "reanimate" is used in the latter case.
4681 */
4682 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004683do_finish(exarg_T *eap, int reanimate)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004684{
4685 int idx;
4686
4687 if (reanimate)
4688 ((struct source_cookie *)getline_cookie(eap->getline,
4689 eap->cookie))->finished = FALSE;
4690
4691 /*
4692 * Cleanup (and inactivate) conditionals, but stop when a try conditional
4693 * not in its finally clause (which then is to be executed next) is found.
4694 * In this case, make the ":finish" pending for execution at the ":endtry".
4695 * Otherwise, finish normally.
4696 */
4697 idx = cleanup_conditionals(eap->cstack, 0, TRUE);
4698 if (idx >= 0)
4699 {
4700 eap->cstack->cs_pending[idx] = CSTP_FINISH;
4701 report_make_pending(CSTP_FINISH, NULL);
4702 }
4703 else
4704 ((struct source_cookie *)getline_cookie(eap->getline,
4705 eap->cookie))->finished = TRUE;
4706}
4707
4708
4709/*
4710 * Return TRUE when a sourced file had the ":finish" command: Don't give error
4711 * message for missing ":endif".
4712 * Return FALSE when not sourcing a file.
4713 */
4714 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004715source_finished(
4716 char_u *(*fgetline)(int, void *, int),
4717 void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004718{
Bram Moolenaar89d40322006-08-29 15:30:07 +00004719 return (getline_equal(fgetline, cookie, getsourceline)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004720 && ((struct source_cookie *)getline_cookie(
Bram Moolenaar89d40322006-08-29 15:30:07 +00004721 fgetline, cookie))->finished);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004722}
4723#endif
4724
4725#if defined(FEAT_LISTCMDS) || defined(PROTO)
4726/*
4727 * ":checktime [buffer]"
4728 */
4729 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004730ex_checktime(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004731{
4732 buf_T *buf;
4733 int save_no_check_timestamps = no_check_timestamps;
4734
4735 no_check_timestamps = 0;
4736 if (eap->addr_count == 0) /* default is all buffers */
4737 check_timestamps(FALSE);
4738 else
4739 {
4740 buf = buflist_findnr((int)eap->line2);
4741 if (buf != NULL) /* cannot happen? */
4742 (void)buf_check_timestamp(buf, FALSE);
4743 }
4744 no_check_timestamps = save_no_check_timestamps;
4745}
4746#endif
4747
Bram Moolenaar071d4272004-06-13 20:20:40 +00004748#if (defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
4749 && (defined(FEAT_EVAL) || defined(FEAT_MULTI_LANG))
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004750# define HAVE_GET_LOCALE_VAL
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004751static char_u *get_locale_val(int what);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004752
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004753 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004754get_locale_val(int what)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004755{
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004756 char_u *loc;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004757
Bram Moolenaar48e330a2016-02-23 14:53:34 +01004758 /* Obtain the locale value from the libraries. */
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004759 loc = (char_u *)setlocale(what, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004760
Bram Moolenaar61660ea2006-04-07 21:40:07 +00004761# ifdef WIN32
Bram Moolenaar071d4272004-06-13 20:20:40 +00004762 if (loc != NULL)
4763 {
4764 char_u *p;
4765
Bram Moolenaar61660ea2006-04-07 21:40:07 +00004766 /* setocale() returns something like "LC_COLLATE=<name>;LC_..." when
4767 * one of the values (e.g., LC_CTYPE) differs. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004768 p = vim_strchr(loc, '=');
4769 if (p != NULL)
4770 {
4771 loc = ++p;
4772 while (*p != NUL) /* remove trailing newline */
4773 {
Bram Moolenaar61660ea2006-04-07 21:40:07 +00004774 if (*p < ' ' || *p == ';')
Bram Moolenaar071d4272004-06-13 20:20:40 +00004775 {
4776 *p = NUL;
4777 break;
4778 }
4779 ++p;
4780 }
4781 }
4782 }
4783# endif
4784
4785 return loc;
4786}
4787#endif
4788
4789
4790#ifdef WIN32
4791/*
4792 * On MS-Windows locale names are strings like "German_Germany.1252", but
4793 * gettext expects "de". Try to translate one into another here for a few
4794 * supported languages.
4795 */
4796 static char_u *
4797gettext_lang(char_u *name)
4798{
4799 int i;
4800 static char *(mtable[]) = {
4801 "afrikaans", "af",
4802 "czech", "cs",
4803 "dutch", "nl",
4804 "german", "de",
4805 "english_united kingdom", "en_GB",
4806 "spanish", "es",
4807 "french", "fr",
4808 "italian", "it",
4809 "japanese", "ja",
4810 "korean", "ko",
4811 "norwegian", "no",
4812 "polish", "pl",
4813 "russian", "ru",
4814 "slovak", "sk",
4815 "swedish", "sv",
4816 "ukrainian", "uk",
4817 "chinese_china", "zh_CN",
4818 "chinese_taiwan", "zh_TW",
4819 NULL};
4820
4821 for (i = 0; mtable[i] != NULL; i += 2)
4822 if (STRNICMP(mtable[i], name, STRLEN(mtable[i])) == 0)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004823 return (char_u *)mtable[i + 1];
Bram Moolenaar071d4272004-06-13 20:20:40 +00004824 return name;
4825}
4826#endif
4827
4828#if defined(FEAT_MULTI_LANG) || defined(PROTO)
4829/*
4830 * Obtain the current messages language. Used to set the default for
4831 * 'helplang'. May return NULL or an empty string.
4832 */
4833 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004834get_mess_lang(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004835{
4836 char_u *p;
4837
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004838# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00004839# if defined(LC_MESSAGES)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004840 p = get_locale_val(LC_MESSAGES);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004841# else
4842 /* This is necessary for Win32, where LC_MESSAGES is not defined and $LANG
Bram Moolenaar61660ea2006-04-07 21:40:07 +00004843 * may be set to the LCID number. LC_COLLATE is the best guess, LC_TIME
4844 * and LC_MONETARY may be set differently for a Japanese working in the
4845 * US. */
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004846 p = get_locale_val(LC_COLLATE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004847# endif
4848# else
4849 p = mch_getenv((char_u *)"LC_ALL");
4850 if (p == NULL || *p == NUL)
4851 {
4852 p = mch_getenv((char_u *)"LC_MESSAGES");
4853 if (p == NULL || *p == NUL)
4854 p = mch_getenv((char_u *)"LANG");
4855 }
4856# endif
4857# ifdef WIN32
4858 p = gettext_lang(p);
4859# endif
4860 return p;
4861}
4862#endif
4863
Bram Moolenaardef9e822004-12-31 20:58:58 +00004864/* Complicated #if; matches with where get_mess_env() is used below. */
4865#if (defined(FEAT_EVAL) && !((defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
4866 && defined(LC_MESSAGES))) \
4867 || ((defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
4868 && (defined(FEAT_GETTEXT) || defined(FEAT_MBYTE)) \
4869 && !defined(LC_MESSAGES))
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01004870static char_u *get_mess_env(void);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004871
4872/*
4873 * Get the language used for messages from the environment.
4874 */
4875 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004876get_mess_env(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004877{
4878 char_u *p;
4879
4880 p = mch_getenv((char_u *)"LC_ALL");
4881 if (p == NULL || *p == NUL)
4882 {
4883 p = mch_getenv((char_u *)"LC_MESSAGES");
4884 if (p == NULL || *p == NUL)
4885 {
4886 p = mch_getenv((char_u *)"LANG");
4887 if (p != NULL && VIM_ISDIGIT(*p))
4888 p = NULL; /* ignore something like "1043" */
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004889# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00004890 if (p == NULL || *p == NUL)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004891 p = get_locale_val(LC_CTYPE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004892# endif
4893 }
4894 }
4895 return p;
4896}
4897#endif
4898
4899#if defined(FEAT_EVAL) || defined(PROTO)
4900
4901/*
4902 * Set the "v:lang" variable according to the current locale setting.
4903 * Also do "v:lc_time"and "v:ctype".
4904 */
4905 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004906set_lang_var(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004907{
4908 char_u *loc;
4909
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004910# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004911 loc = get_locale_val(LC_CTYPE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004912# else
4913 /* setlocale() not supported: use the default value */
4914 loc = (char_u *)"C";
4915# endif
4916 set_vim_var_string(VV_CTYPE, loc, -1);
4917
4918 /* When LC_MESSAGES isn't defined use the value from $LC_MESSAGES, fall
4919 * back to LC_CTYPE if it's empty. */
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004920# if defined(HAVE_GET_LOCALE_VAL) && defined(LC_MESSAGES)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004921 loc = get_locale_val(LC_MESSAGES);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004922# else
4923 loc = get_mess_env();
4924# endif
4925 set_vim_var_string(VV_LANG, loc, -1);
4926
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004927# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004928 loc = get_locale_val(LC_TIME);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004929# endif
4930 set_vim_var_string(VV_LC_TIME, loc, -1);
4931}
4932#endif
4933
4934#if (defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
4935 && (defined(FEAT_GETTEXT) || defined(FEAT_MBYTE))
4936/*
4937 * ":language": Set the language (locale).
4938 */
4939 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004940ex_language(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004941{
4942 char *loc;
4943 char_u *p;
4944 char_u *name;
4945 int what = LC_ALL;
4946 char *whatstr = "";
4947#ifdef LC_MESSAGES
4948# define VIM_LC_MESSAGES LC_MESSAGES
4949#else
4950# define VIM_LC_MESSAGES 6789
4951#endif
4952
4953 name = eap->arg;
4954
4955 /* Check for "messages {name}", "ctype {name}" or "time {name}" argument.
4956 * Allow abbreviation, but require at least 3 characters to avoid
4957 * confusion with a two letter language name "me" or "ct". */
4958 p = skiptowhite(eap->arg);
4959 if ((*p == NUL || vim_iswhite(*p)) && p - eap->arg >= 3)
4960 {
4961 if (STRNICMP(eap->arg, "messages", p - eap->arg) == 0)
4962 {
4963 what = VIM_LC_MESSAGES;
4964 name = skipwhite(p);
4965 whatstr = "messages ";
4966 }
4967 else if (STRNICMP(eap->arg, "ctype", p - eap->arg) == 0)
4968 {
4969 what = LC_CTYPE;
4970 name = skipwhite(p);
4971 whatstr = "ctype ";
4972 }
4973 else if (STRNICMP(eap->arg, "time", p - eap->arg) == 0)
4974 {
4975 what = LC_TIME;
4976 name = skipwhite(p);
4977 whatstr = "time ";
4978 }
4979 }
4980
4981 if (*name == NUL)
4982 {
4983#ifndef LC_MESSAGES
4984 if (what == VIM_LC_MESSAGES)
4985 p = get_mess_env();
4986 else
4987#endif
4988 p = (char_u *)setlocale(what, NULL);
4989 if (p == NULL || *p == NUL)
4990 p = (char_u *)"Unknown";
4991 smsg((char_u *)_("Current %slanguage: \"%s\""), whatstr, p);
4992 }
4993 else
4994 {
4995#ifndef LC_MESSAGES
4996 if (what == VIM_LC_MESSAGES)
4997 loc = "";
4998 else
4999#endif
Bram Moolenaar8c8de832008-06-24 22:58:06 +00005000 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00005001 loc = setlocale(what, (char *)name);
Bram Moolenaar8c8de832008-06-24 22:58:06 +00005002#if defined(FEAT_FLOAT) && defined(LC_NUMERIC)
5003 /* Make sure strtod() uses a decimal point, not a comma. */
5004 setlocale(LC_NUMERIC, "C");
5005#endif
5006 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00005007 if (loc == NULL)
5008 EMSG2(_("E197: Cannot set language to \"%s\""), name);
5009 else
5010 {
5011#ifdef HAVE_NL_MSG_CAT_CNTR
5012 /* Need to do this for GNU gettext, otherwise cached translations
5013 * will be used again. */
5014 extern int _nl_msg_cat_cntr;
5015
5016 ++_nl_msg_cat_cntr;
5017#endif
Bram Moolenaarb8017e72007-05-10 18:59:07 +00005018 /* Reset $LC_ALL, otherwise it would overrule everything. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005019 vim_setenv((char_u *)"LC_ALL", (char_u *)"");
5020
5021 if (what != LC_TIME)
5022 {
5023 /* Tell gettext() what to translate to. It apparently doesn't
5024 * use the currently effective locale. Also do this when
5025 * FEAT_GETTEXT isn't defined, so that shell commands use this
5026 * value. */
5027 if (what == LC_ALL)
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00005028 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00005029 vim_setenv((char_u *)"LANG", name);
Bram Moolenaare3a0b532013-06-28 20:36:30 +02005030
5031 /* Clear $LANGUAGE because GNU gettext uses it. */
5032 vim_setenv((char_u *)"LANGUAGE", (char_u *)"");
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00005033# ifdef WIN32
5034 /* Apparently MS-Windows printf() may cause a crash when
5035 * we give it 8-bit text while it's expecting text in the
5036 * current locale. This call avoids that. */
5037 setlocale(LC_CTYPE, "C");
5038# endif
5039 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00005040 if (what != LC_CTYPE)
5041 {
5042 char_u *mname;
5043#ifdef WIN32
5044 mname = gettext_lang(name);
5045#else
5046 mname = name;
5047#endif
5048 vim_setenv((char_u *)"LC_MESSAGES", mname);
5049#ifdef FEAT_MULTI_LANG
5050 set_helplang_default(mname);
5051#endif
5052 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00005053 }
5054
5055# ifdef FEAT_EVAL
5056 /* Set v:lang, v:lc_time and v:ctype to the final result. */
5057 set_lang_var();
5058# endif
Bram Moolenaar6cc00c72011-10-20 21:41:09 +02005059# ifdef FEAT_TITLE
5060 maketitle();
5061# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00005062 }
5063 }
5064}
5065
5066# if defined(FEAT_CMDL_COMPL) || defined(PROTO)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005067
5068static char_u **locales = NULL; /* Array of all available locales */
5069static int did_init_locales = FALSE;
5070
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01005071static void init_locales(void);
5072static char_u **find_locales(void);
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005073
5074/*
5075 * Lazy initialization of all available locales.
5076 */
5077 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005078init_locales(void)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005079{
5080 if (!did_init_locales)
5081 {
5082 did_init_locales = TRUE;
5083 locales = find_locales();
5084 }
5085}
5086
5087/* Return an array of strings for all available locales + NULL for the
5088 * last element. Return NULL in case of error. */
5089 static char_u **
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005090find_locales(void)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005091{
5092 garray_T locales_ga;
5093 char_u *loc;
5094
5095 /* Find all available locales by running command "locale -a". If this
5096 * doesn't work we won't have completion. */
5097 char_u *locale_a = get_cmd_output((char_u *)"locale -a",
Bram Moolenaar39c29ed2014-04-05 19:44:40 +02005098 NULL, SHELL_SILENT, NULL);
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005099 if (locale_a == NULL)
5100 return NULL;
5101 ga_init2(&locales_ga, sizeof(char_u *), 20);
5102
5103 /* Transform locale_a string where each locale is separated by "\n"
5104 * into an array of locale strings. */
5105 loc = (char_u *)strtok((char *)locale_a, "\n");
5106
5107 while (loc != NULL)
5108 {
5109 if (ga_grow(&locales_ga, 1) == FAIL)
5110 break;
5111 loc = vim_strsave(loc);
5112 if (loc == NULL)
5113 break;
5114
5115 ((char_u **)locales_ga.ga_data)[locales_ga.ga_len++] = loc;
5116 loc = (char_u *)strtok(NULL, "\n");
5117 }
5118 vim_free(locale_a);
5119 if (ga_grow(&locales_ga, 1) == FAIL)
5120 {
5121 ga_clear(&locales_ga);
5122 return NULL;
5123 }
5124 ((char_u **)locales_ga.ga_data)[locales_ga.ga_len] = NULL;
5125 return (char_u **)locales_ga.ga_data;
5126}
5127
5128# if defined(EXITFREE) || defined(PROTO)
5129 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005130free_locales(void)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005131{
5132 int i;
5133 if (locales != NULL)
5134 {
5135 for (i = 0; locales[i] != NULL; i++)
5136 vim_free(locales[i]);
5137 vim_free(locales);
5138 locales = NULL;
5139 }
5140}
5141# endif
5142
Bram Moolenaar071d4272004-06-13 20:20:40 +00005143/*
5144 * Function given to ExpandGeneric() to obtain the possible arguments of the
5145 * ":language" command.
5146 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005147 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005148get_lang_arg(expand_T *xp UNUSED, int idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005149{
5150 if (idx == 0)
5151 return (char_u *)"messages";
5152 if (idx == 1)
5153 return (char_u *)"ctype";
5154 if (idx == 2)
5155 return (char_u *)"time";
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005156
5157 init_locales();
5158 if (locales == NULL)
5159 return NULL;
5160 return locales[idx - 3];
5161}
5162
5163/*
5164 * Function given to ExpandGeneric() to obtain the available locales.
5165 */
5166 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005167get_locales(expand_T *xp UNUSED, int idx)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005168{
5169 init_locales();
5170 if (locales == NULL)
5171 return NULL;
5172 return locales[idx];
Bram Moolenaar071d4272004-06-13 20:20:40 +00005173}
5174# endif
5175
5176#endif