blob: 6cf26e9931ce447ccfb4c6aba7cfe9d53ed6c35d [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 {
1192# ifdef WIN3264
1193 this_due = (long)(((double)(timer->tr_due.QuadPart - now.QuadPart)
1194 / (double)fr.QuadPart) * 1000);
1195# else
1196 this_due = (timer->tr_due.tv_sec - now.tv_sec) * 1000
1197 + (timer->tr_due.tv_usec - now.tv_usec) / 1000;
1198# endif
1199 if (this_due <= 1)
1200 {
1201 remove_timer(timer);
1202 timer_callback(timer);
1203 did_one = TRUE;
1204 if (timer->tr_repeat != 0)
1205 {
1206 profile_setlimit(timer->tr_interval, &timer->tr_due);
1207 if (timer->tr_repeat > 0)
1208 --timer->tr_repeat;
1209 insert_timer(timer);
1210 }
1211 else
1212 free_timer(timer);
1213 /* the callback may do anything, start all over */
1214 break;
1215 }
1216 if (next_due == -1 || next_due > this_due)
1217 next_due = this_due;
1218 }
1219 if (timer == NULL)
1220 break;
1221 }
1222
1223 if (did_one)
1224 redraw_after_callback();
1225
1226 return next_due;
1227}
1228
1229/*
1230 * Find a timer by ID. Returns NULL if not found;
1231 */
1232 timer_T *
1233find_timer(int id)
1234{
1235 timer_T *timer;
1236
1237 for (timer = first_timer; timer != NULL; timer = timer->tr_next)
1238 if (timer->tr_id == id)
1239 break;
1240 return timer;
1241}
1242
1243
1244/*
1245 * Stop a timer and delete it.
1246 */
1247 void
1248stop_timer(timer_T *timer)
1249{
1250 remove_timer(timer);
1251 free_timer(timer);
1252}
Bram Moolenaare3188e22016-05-31 21:13:04 +02001253
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001254 void
1255add_timer_info(typval_T *rettv, timer_T *timer)
1256{
1257 list_T *list = rettv->vval.v_list;
1258 dict_T *dict = dict_alloc();
1259 dictitem_T *di;
1260 long remaining;
1261 proftime_T now;
Bram Moolenaar00ff3802016-08-06 22:27:28 +02001262# ifdef WIN3264
1263 LARGE_INTEGER fr;
1264#endif
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001265
1266 if (dict == NULL)
1267 return;
1268 list_append_dict(list, dict);
1269
1270 dict_add_nr_str(dict, "id", (long)timer->tr_id, NULL);
1271 dict_add_nr_str(dict, "time", (long)timer->tr_interval, NULL);
1272
1273 profile_start(&now);
1274# ifdef WIN3264
Bram Moolenaar00ff3802016-08-06 22:27:28 +02001275 QueryPerformanceFrequency(&fr);
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001276 remaining = (long)(((double)(timer->tr_due.QuadPart - now.QuadPart)
1277 / (double)fr.QuadPart) * 1000);
1278# else
1279 remaining = (timer->tr_due.tv_sec - now.tv_sec) * 1000
1280 + (timer->tr_due.tv_usec - now.tv_usec) / 1000;
1281# endif
1282 dict_add_nr_str(dict, "remaining", (long)remaining, NULL);
1283
1284 dict_add_nr_str(dict, "repeat",
1285 (long)(timer->tr_repeat < 0 ? -1 : timer->tr_repeat + 1), NULL);
1286
1287 di = dictitem_alloc((char_u *)"callback");
1288 if (di != NULL)
1289 {
1290 if (dict_add(dict, di) == FAIL)
1291 vim_free(di);
1292 else if (timer->tr_partial != NULL)
1293 {
1294 di->di_tv.v_type = VAR_PARTIAL;
1295 di->di_tv.vval.v_partial = timer->tr_partial;
1296 ++timer->tr_partial->pt_refcount;
1297 }
1298 else
1299 {
1300 di->di_tv.v_type = VAR_FUNC;
1301 di->di_tv.vval.v_string = vim_strsave(timer->tr_callback);
1302 }
1303 di->di_tv.v_lock = 0;
1304 }
1305}
1306
1307 void
1308add_timer_info_all(typval_T *rettv)
1309{
1310 timer_T *timer;
1311
1312 for (timer = first_timer; timer != NULL; timer = timer->tr_next)
1313 add_timer_info(rettv, timer);
1314}
1315
Bram Moolenaare3188e22016-05-31 21:13:04 +02001316/*
1317 * Mark references in partials of timers.
1318 */
1319 int
1320set_ref_in_timer(int copyID)
1321{
1322 int abort = FALSE;
1323 timer_T *timer;
1324 typval_T tv;
1325
1326 for (timer = first_timer; timer != NULL; timer = timer->tr_next)
1327 {
Bram Moolenaar1e96d9b2016-07-29 22:15:09 +02001328 if (timer->tr_partial != NULL)
1329 {
1330 tv.v_type = VAR_PARTIAL;
1331 tv.vval.v_partial = timer->tr_partial;
1332 }
1333 else
1334 {
1335 tv.v_type = VAR_FUNC;
1336 tv.vval.v_string = timer->tr_callback;
1337 }
Bram Moolenaare3188e22016-05-31 21:13:04 +02001338 abort = abort || set_ref_in_item(&tv, copyID, NULL, NULL);
1339 }
1340 return abort;
1341}
Bram Moolenaar623e2632016-07-30 22:47:56 +02001342
1343# if defined(EXITFREE) || defined(PROTO)
1344 void
1345timer_free_all()
1346{
1347 timer_T *timer;
1348
1349 while (first_timer != NULL)
1350 {
1351 timer = first_timer;
1352 remove_timer(timer);
1353 free_timer(timer);
1354 }
1355}
1356# endif
Bram Moolenaar975b5272016-03-15 23:10:59 +01001357# endif
1358
Bram Moolenaar8a7f5a22013-06-06 14:01:46 +02001359#if defined(FEAT_SYN_HL) && defined(FEAT_RELTIME) && defined(FEAT_FLOAT)
1360# if defined(HAVE_MATH_H)
1361# include <math.h>
1362# endif
1363
1364/*
1365 * Divide the time "tm" by "count" and store in "tm2".
1366 */
1367 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001368profile_divide(proftime_T *tm, int count, proftime_T *tm2)
Bram Moolenaar8a7f5a22013-06-06 14:01:46 +02001369{
1370 if (count == 0)
1371 profile_zero(tm2);
1372 else
1373 {
1374# ifdef WIN3264
1375 tm2->QuadPart = tm->QuadPart / count;
1376# else
1377 double usec = (tm->tv_sec * 1000000.0 + tm->tv_usec) / count;
1378
1379 tm2->tv_sec = floor(usec / 1000000.0);
Bram Moolenaara2e14fc2013-06-10 20:10:44 +02001380 tm2->tv_usec = vim_round(usec - (tm2->tv_sec * 1000000.0));
Bram Moolenaar8a7f5a22013-06-06 14:01:46 +02001381# endif
1382 }
1383}
1384#endif
1385
Bram Moolenaar76929292008-01-06 19:07:36 +00001386# if defined(FEAT_PROFILE) || defined(PROTO)
1387/*
1388 * Functions for profiling.
1389 */
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01001390static void script_do_profile(scriptitem_T *si);
1391static void script_dump_profile(FILE *fd);
Bram Moolenaar76929292008-01-06 19:07:36 +00001392static proftime_T prof_wait_time;
1393
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001394/*
Bram Moolenaar05159a02005-02-26 23:04:13 +00001395 * Add the time "tm2" to "tm".
1396 */
1397 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001398profile_add(proftime_T *tm, proftime_T *tm2)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001399{
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001400# ifdef WIN3264
1401 tm->QuadPart += tm2->QuadPart;
1402# else
Bram Moolenaar05159a02005-02-26 23:04:13 +00001403 tm->tv_usec += tm2->tv_usec;
1404 tm->tv_sec += tm2->tv_sec;
1405 if (tm->tv_usec >= 1000000)
1406 {
1407 tm->tv_usec -= 1000000;
1408 ++tm->tv_sec;
1409 }
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001410# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00001411}
1412
1413/*
Bram Moolenaar1056d982006-03-09 22:37:52 +00001414 * Add the "self" time from the total time and the children's time.
1415 */
1416 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001417profile_self(proftime_T *self, proftime_T *total, proftime_T *children)
Bram Moolenaar1056d982006-03-09 22:37:52 +00001418{
1419 /* Check that the result won't be negative. Can happen with recursive
1420 * calls. */
1421#ifdef WIN3264
1422 if (total->QuadPart <= children->QuadPart)
1423 return;
1424#else
1425 if (total->tv_sec < children->tv_sec
1426 || (total->tv_sec == children->tv_sec
1427 && total->tv_usec <= children->tv_usec))
1428 return;
1429#endif
1430 profile_add(self, total);
1431 profile_sub(self, children);
1432}
1433
1434/*
Bram Moolenaar05159a02005-02-26 23:04:13 +00001435 * Get the current waittime.
1436 */
1437 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001438profile_get_wait(proftime_T *tm)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001439{
1440 *tm = prof_wait_time;
1441}
1442
1443/*
1444 * Subtract the passed waittime since "tm" from "tma".
1445 */
1446 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001447profile_sub_wait(proftime_T *tm, proftime_T *tma)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001448{
1449 proftime_T tm3 = prof_wait_time;
1450
1451 profile_sub(&tm3, tm);
1452 profile_sub(tma, &tm3);
1453}
1454
1455/*
1456 * Return TRUE if "tm1" and "tm2" are equal.
1457 */
1458 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001459profile_equal(proftime_T *tm1, proftime_T *tm2)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001460{
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001461# ifdef WIN3264
1462 return (tm1->QuadPart == tm2->QuadPart);
1463# else
Bram Moolenaar05159a02005-02-26 23:04:13 +00001464 return (tm1->tv_usec == tm2->tv_usec && tm1->tv_sec == tm2->tv_sec);
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001465# endif
1466}
1467
1468/*
1469 * Return <0, 0 or >0 if "tm1" < "tm2", "tm1" == "tm2" or "tm1" > "tm2"
1470 */
1471 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001472profile_cmp(const proftime_T *tm1, const proftime_T *tm2)
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001473{
1474# ifdef WIN3264
1475 return (int)(tm2->QuadPart - tm1->QuadPart);
1476# else
1477 if (tm1->tv_sec == tm2->tv_sec)
1478 return tm2->tv_usec - tm1->tv_usec;
1479 return tm2->tv_sec - tm1->tv_sec;
1480# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00001481}
1482
Bram Moolenaar05159a02005-02-26 23:04:13 +00001483static char_u *profile_fname = NULL;
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001484static proftime_T pause_time;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001485
1486/*
1487 * ":profile cmd args"
1488 */
1489 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001490ex_profile(exarg_T *eap)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001491{
1492 char_u *e;
1493 int len;
1494
1495 e = skiptowhite(eap->arg);
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00001496 len = (int)(e - eap->arg);
Bram Moolenaar05159a02005-02-26 23:04:13 +00001497 e = skipwhite(e);
1498
1499 if (len == 5 && STRNCMP(eap->arg, "start", 5) == 0 && *e != NUL)
1500 {
1501 vim_free(profile_fname);
Bram Moolenaard94682f2015-04-13 15:37:56 +02001502 profile_fname = expand_env_save_opt(e, TRUE);
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001503 do_profiling = PROF_YES;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001504 profile_zero(&prof_wait_time);
1505 set_vim_var_nr(VV_PROFILING, 1L);
1506 }
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001507 else if (do_profiling == PROF_NONE)
Bram Moolenaar54c1b492010-02-24 14:01:28 +01001508 EMSG(_("E750: First use \":profile start {fname}\""));
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001509 else if (STRCMP(eap->arg, "pause") == 0)
1510 {
1511 if (do_profiling == PROF_YES)
1512 profile_start(&pause_time);
1513 do_profiling = PROF_PAUSED;
1514 }
1515 else if (STRCMP(eap->arg, "continue") == 0)
1516 {
1517 if (do_profiling == PROF_PAUSED)
1518 {
1519 profile_end(&pause_time);
1520 profile_add(&prof_wait_time, &pause_time);
1521 }
1522 do_profiling = PROF_YES;
1523 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00001524 else
1525 {
1526 /* The rest is similar to ":breakadd". */
1527 ex_breakadd(eap);
1528 }
1529}
1530
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001531/* Command line expansion for :profile. */
1532static enum
1533{
1534 PEXP_SUBCMD, /* expand :profile sub-commands */
Bram Moolenaar49789dc2011-02-25 14:46:09 +01001535 PEXP_FUNC /* expand :profile func {funcname} */
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001536} pexpand_what;
1537
1538static char *pexpand_cmds[] = {
1539 "start",
1540#define PROFCMD_START 0
1541 "pause",
1542#define PROFCMD_PAUSE 1
1543 "continue",
1544#define PROFCMD_CONTINUE 2
1545 "func",
1546#define PROFCMD_FUNC 3
1547 "file",
1548#define PROFCMD_FILE 4
1549 NULL
1550#define PROFCMD_LAST 5
1551};
1552
1553/*
1554 * Function given to ExpandGeneric() to obtain the profile command
1555 * specific expansion.
1556 */
1557 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001558get_profile_name(expand_T *xp UNUSED, int idx)
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001559{
1560 switch (pexpand_what)
1561 {
1562 case PEXP_SUBCMD:
1563 return (char_u *)pexpand_cmds[idx];
1564 /* case PEXP_FUNC: TODO */
1565 default:
1566 return NULL;
1567 }
1568}
1569
1570/*
1571 * Handle command line completion for :profile command.
1572 */
1573 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001574set_context_in_profile_cmd(expand_T *xp, char_u *arg)
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001575{
1576 char_u *end_subcmd;
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001577
1578 /* Default: expand subcommands. */
1579 xp->xp_context = EXPAND_PROFILE;
1580 pexpand_what = PEXP_SUBCMD;
1581 xp->xp_pattern = arg;
1582
1583 end_subcmd = skiptowhite(arg);
1584 if (*end_subcmd == NUL)
1585 return;
1586
Bram Moolenaar8b9c05f2010-03-02 17:54:33 +01001587 if (end_subcmd - arg == 5 && STRNCMP(arg, "start", 5) == 0)
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001588 {
1589 xp->xp_context = EXPAND_FILES;
1590 xp->xp_pattern = skipwhite(end_subcmd);
1591 return;
1592 }
1593
1594 /* TODO: expand function names after "func" */
1595 xp->xp_context = EXPAND_NOTHING;
1596}
1597
Bram Moolenaar05159a02005-02-26 23:04:13 +00001598/*
1599 * Dump the profiling info.
1600 */
1601 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001602profile_dump(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001603{
1604 FILE *fd;
1605
1606 if (profile_fname != NULL)
1607 {
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +00001608 fd = mch_fopen((char *)profile_fname, "w");
Bram Moolenaar05159a02005-02-26 23:04:13 +00001609 if (fd == NULL)
1610 EMSG2(_(e_notopen), profile_fname);
1611 else
1612 {
Bram Moolenaar05159a02005-02-26 23:04:13 +00001613 script_dump_profile(fd);
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001614 func_dump_profile(fd);
Bram Moolenaar05159a02005-02-26 23:04:13 +00001615 fclose(fd);
1616 }
1617 }
1618}
1619
1620/*
1621 * Start profiling script "fp".
1622 */
1623 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001624script_do_profile(scriptitem_T *si)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001625{
1626 si->sn_pr_count = 0;
1627 profile_zero(&si->sn_pr_total);
1628 profile_zero(&si->sn_pr_self);
1629
1630 ga_init2(&si->sn_prl_ga, sizeof(sn_prl_T), 100);
1631 si->sn_prl_idx = -1;
1632 si->sn_prof_on = TRUE;
1633 si->sn_pr_nest = 0;
1634}
1635
1636/*
1637 * save time when starting to invoke another script or function.
1638 */
1639 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001640script_prof_save(
1641 proftime_T *tm) /* place to store wait time */
Bram Moolenaar05159a02005-02-26 23:04:13 +00001642{
1643 scriptitem_T *si;
1644
1645 if (current_SID > 0 && current_SID <= script_items.ga_len)
1646 {
1647 si = &SCRIPT_ITEM(current_SID);
1648 if (si->sn_prof_on && si->sn_pr_nest++ == 0)
1649 profile_start(&si->sn_pr_child);
1650 }
1651 profile_get_wait(tm);
1652}
1653
1654/*
1655 * Count time spent in children after invoking another script or function.
1656 */
1657 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001658script_prof_restore(proftime_T *tm)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001659{
1660 scriptitem_T *si;
1661
1662 if (current_SID > 0 && current_SID <= script_items.ga_len)
1663 {
1664 si = &SCRIPT_ITEM(current_SID);
1665 if (si->sn_prof_on && --si->sn_pr_nest == 0)
1666 {
1667 profile_end(&si->sn_pr_child);
1668 profile_sub_wait(tm, &si->sn_pr_child); /* don't count wait time */
1669 profile_add(&si->sn_pr_children, &si->sn_pr_child);
1670 profile_add(&si->sn_prl_children, &si->sn_pr_child);
1671 }
1672 }
1673}
1674
1675static proftime_T inchar_time;
1676
1677/*
1678 * Called when starting to wait for the user to type a character.
1679 */
1680 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001681prof_inchar_enter(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001682{
1683 profile_start(&inchar_time);
1684}
1685
1686/*
1687 * Called when finished waiting for the user to type a character.
1688 */
1689 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001690prof_inchar_exit(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001691{
1692 profile_end(&inchar_time);
1693 profile_add(&prof_wait_time, &inchar_time);
1694}
1695
1696/*
1697 * Dump the profiling results for all scripts in file "fd".
1698 */
1699 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001700script_dump_profile(FILE *fd)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001701{
1702 int id;
1703 scriptitem_T *si;
1704 int i;
1705 FILE *sfd;
1706 sn_prl_T *pp;
1707
1708 for (id = 1; id <= script_items.ga_len; ++id)
1709 {
1710 si = &SCRIPT_ITEM(id);
1711 if (si->sn_prof_on)
1712 {
1713 fprintf(fd, "SCRIPT %s\n", si->sn_name);
1714 if (si->sn_pr_count == 1)
1715 fprintf(fd, "Sourced 1 time\n");
1716 else
1717 fprintf(fd, "Sourced %d times\n", si->sn_pr_count);
1718 fprintf(fd, "Total time: %s\n", profile_msg(&si->sn_pr_total));
1719 fprintf(fd, " Self time: %s\n", profile_msg(&si->sn_pr_self));
1720 fprintf(fd, "\n");
1721 fprintf(fd, "count total (s) self (s)\n");
1722
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +00001723 sfd = mch_fopen((char *)si->sn_name, "r");
Bram Moolenaar05159a02005-02-26 23:04:13 +00001724 if (sfd == NULL)
1725 fprintf(fd, "Cannot open file!\n");
1726 else
1727 {
1728 for (i = 0; i < si->sn_prl_ga.ga_len; ++i)
1729 {
1730 if (vim_fgets(IObuff, IOSIZE, sfd))
1731 break;
1732 pp = &PRL_ITEM(si, i);
1733 if (pp->snp_count > 0)
1734 {
1735 fprintf(fd, "%5d ", pp->snp_count);
1736 if (profile_equal(&pp->sn_prl_total, &pp->sn_prl_self))
1737 fprintf(fd, " ");
1738 else
1739 fprintf(fd, "%s ", profile_msg(&pp->sn_prl_total));
1740 fprintf(fd, "%s ", profile_msg(&pp->sn_prl_self));
1741 }
1742 else
1743 fprintf(fd, " ");
1744 fprintf(fd, "%s", IObuff);
1745 }
1746 fclose(sfd);
1747 }
1748 fprintf(fd, "\n");
1749 }
1750 }
1751}
1752
1753/*
1754 * Return TRUE when a function defined in the current script should be
1755 * profiled.
1756 */
1757 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001758prof_def_func(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001759{
Bram Moolenaar53180ce2005-07-05 21:48:14 +00001760 if (current_SID > 0)
1761 return SCRIPT_ITEM(current_SID).sn_pr_force;
1762 return FALSE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001763}
1764
1765# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001766#endif
1767
1768/*
1769 * If 'autowrite' option set, try to write the file.
1770 * Careful: autocommands may make "buf" invalid!
1771 *
1772 * return FAIL for failure, OK otherwise
1773 */
1774 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001775autowrite(buf_T *buf, int forceit)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001776{
Bram Moolenaar373154b2007-02-13 05:19:30 +00001777 int r;
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001778 bufref_T bufref;
Bram Moolenaar373154b2007-02-13 05:19:30 +00001779
Bram Moolenaar071d4272004-06-13 20:20:40 +00001780 if (!(p_aw || p_awa) || !p_write
1781#ifdef FEAT_QUICKFIX
Bram Moolenaar373154b2007-02-13 05:19:30 +00001782 /* never autowrite a "nofile" or "nowrite" buffer */
1783 || bt_dontwrite(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001784#endif
Bram Moolenaar373154b2007-02-13 05:19:30 +00001785 || (!forceit && buf->b_p_ro) || buf->b_ffname == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001786 return FAIL;
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001787 set_bufref(&bufref, buf);
Bram Moolenaar373154b2007-02-13 05:19:30 +00001788 r = buf_write_all(buf, forceit);
1789
1790 /* Writing may succeed but the buffer still changed, e.g., when there is a
1791 * conversion error. We do want to return FAIL then. */
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001792 if (bufref_valid(&bufref) && bufIsChanged(buf))
Bram Moolenaar373154b2007-02-13 05:19:30 +00001793 r = FAIL;
1794 return r;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001795}
1796
1797/*
1798 * flush all buffers, except the ones that are readonly
1799 */
1800 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001801autowrite_all(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001802{
1803 buf_T *buf;
1804
1805 if (!(p_aw || p_awa) || !p_write)
1806 return;
Bram Moolenaar29323592016-07-24 22:04:11 +02001807 FOR_ALL_BUFFERS(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001808 if (bufIsChanged(buf) && !buf->b_p_ro)
1809 {
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001810#ifdef FEAT_AUTOCMD
1811 bufref_T bufref;
1812
1813 set_bufref(&bufref, buf);
1814#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001815 (void)buf_write_all(buf, FALSE);
1816#ifdef FEAT_AUTOCMD
1817 /* an autocommand may have deleted the buffer */
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001818 if (!bufref_valid(&bufref))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001819 buf = firstbuf;
1820#endif
1821 }
1822}
1823
1824/*
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001825 * Return TRUE if buffer was changed and cannot be abandoned.
1826 * For flags use the CCGD_ values.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001827 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001828 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001829check_changed(buf_T *buf, int flags)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001830{
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001831 int forceit = (flags & CCGD_FORCEIT);
1832#ifdef FEAT_AUTOCMD
1833 bufref_T bufref;
1834
1835 set_bufref(&bufref, buf);
1836#endif
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001837
Bram Moolenaar071d4272004-06-13 20:20:40 +00001838 if ( !forceit
1839 && bufIsChanged(buf)
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001840 && ((flags & CCGD_MULTWIN) || buf->b_nwindows <= 1)
1841 && (!(flags & CCGD_AW) || autowrite(buf, forceit) == FAIL))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001842 {
1843#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG)
1844 if ((p_confirm || cmdmod.confirm) && p_write)
1845 {
1846 buf_T *buf2;
1847 int count = 0;
1848
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001849 if (flags & CCGD_ALLBUF)
Bram Moolenaar29323592016-07-24 22:04:11 +02001850 FOR_ALL_BUFFERS(buf2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001851 if (bufIsChanged(buf2)
1852 && (buf2->b_ffname != NULL
1853# ifdef FEAT_BROWSE
1854 || cmdmod.browse
1855# endif
1856 ))
1857 ++count;
1858# ifdef FEAT_AUTOCMD
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001859 if (!bufref_valid(&bufref))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001860 /* Autocommand deleted buffer, oops! It's not changed now. */
1861 return FALSE;
1862# endif
1863 dialog_changed(buf, count > 1);
1864# ifdef FEAT_AUTOCMD
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001865 if (!bufref_valid(&bufref))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001866 /* Autocommand deleted buffer, oops! It's not changed now. */
1867 return FALSE;
1868# endif
1869 return bufIsChanged(buf);
1870 }
1871#endif
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001872 if (flags & CCGD_EXCMD)
1873 EMSG(_(e_nowrtmsg));
1874 else
1875 EMSG(_(e_nowrtmsg_nobang));
Bram Moolenaar071d4272004-06-13 20:20:40 +00001876 return TRUE;
1877 }
1878 return FALSE;
1879}
1880
1881#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG) || defined(PROTO)
1882
1883#if defined(FEAT_BROWSE) || defined(PROTO)
1884/*
1885 * When wanting to write a file without a file name, ask the user for a name.
1886 */
1887 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001888browse_save_fname(buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001889{
1890 if (buf->b_fname == NULL)
1891 {
1892 char_u *fname;
1893
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00001894 fname = do_browse(BROWSE_SAVE, (char_u *)_("Save As"),
1895 NULL, NULL, NULL, NULL, buf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001896 if (fname != NULL)
1897 {
1898 if (setfname(buf, fname, NULL, TRUE) == OK)
1899 buf->b_flags |= BF_NOTEDITED;
1900 vim_free(fname);
1901 }
1902 }
1903}
1904#endif
1905
1906/*
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02001907 * Ask the user what to do when abandoning a changed buffer.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001908 * Must check 'write' option first!
1909 */
1910 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001911dialog_changed(
1912 buf_T *buf,
1913 int checkall) /* may abandon all changed buffers */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001914{
Bram Moolenaard9462e32011-04-11 21:35:11 +02001915 char_u buff[DIALOG_MSG_SIZE];
Bram Moolenaar071d4272004-06-13 20:20:40 +00001916 int ret;
1917 buf_T *buf2;
Bram Moolenaar8218f602012-04-25 17:32:18 +02001918 exarg_T ea;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001919
Bram Moolenaar82cf9b62005-06-07 21:09:25 +00001920 dialog_msg(buff, _("Save changes to \"%s\"?"),
Bram Moolenaar071d4272004-06-13 20:20:40 +00001921 (buf->b_fname != NULL) ?
1922 buf->b_fname : (char_u *)_("Untitled"));
1923 if (checkall)
1924 ret = vim_dialog_yesnoallcancel(VIM_QUESTION, NULL, buff, 1);
1925 else
1926 ret = vim_dialog_yesnocancel(VIM_QUESTION, NULL, buff, 1);
1927
Bram Moolenaar8218f602012-04-25 17:32:18 +02001928 /* Init ea pseudo-structure, this is needed for the check_overwrite()
1929 * function. */
1930 ea.append = ea.forceit = FALSE;
1931
Bram Moolenaar071d4272004-06-13 20:20:40 +00001932 if (ret == VIM_YES)
1933 {
1934#ifdef FEAT_BROWSE
1935 /* May get file name, when there is none */
1936 browse_save_fname(buf);
1937#endif
Bram Moolenaar8218f602012-04-25 17:32:18 +02001938 if (buf->b_fname != NULL && check_overwrite(&ea, buf,
1939 buf->b_fname, buf->b_ffname, FALSE) == OK)
1940 /* didn't hit Cancel */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001941 (void)buf_write_all(buf, FALSE);
1942 }
1943 else if (ret == VIM_NO)
1944 {
1945 unchanged(buf, TRUE);
1946 }
1947 else if (ret == VIM_ALL)
1948 {
1949 /*
1950 * Write all modified files that can be written.
1951 * Skip readonly buffers, these need to be confirmed
1952 * individually.
1953 */
Bram Moolenaar29323592016-07-24 22:04:11 +02001954 FOR_ALL_BUFFERS(buf2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001955 {
1956 if (bufIsChanged(buf2)
1957 && (buf2->b_ffname != NULL
1958#ifdef FEAT_BROWSE
1959 || cmdmod.browse
1960#endif
1961 )
1962 && !buf2->b_p_ro)
1963 {
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001964#ifdef FEAT_AUTOCMD
1965 bufref_T bufref;
1966
1967 set_bufref(&bufref, buf2);
1968#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001969#ifdef FEAT_BROWSE
1970 /* May get file name, when there is none */
1971 browse_save_fname(buf2);
1972#endif
Bram Moolenaar8218f602012-04-25 17:32:18 +02001973 if (buf2->b_fname != NULL && check_overwrite(&ea, buf2,
1974 buf2->b_fname, buf2->b_ffname, FALSE) == OK)
1975 /* didn't hit Cancel */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001976 (void)buf_write_all(buf2, FALSE);
1977#ifdef FEAT_AUTOCMD
1978 /* an autocommand may have deleted the buffer */
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001979 if (!bufref_valid(&bufref))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001980 buf2 = firstbuf;
1981#endif
1982 }
1983 }
1984 }
1985 else if (ret == VIM_DISCARDALL)
1986 {
1987 /*
1988 * mark all buffers as unchanged
1989 */
Bram Moolenaar29323592016-07-24 22:04:11 +02001990 FOR_ALL_BUFFERS(buf2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001991 unchanged(buf2, TRUE);
1992 }
1993}
1994#endif
1995
1996/*
1997 * Return TRUE if the buffer "buf" can be abandoned, either by making it
1998 * hidden, autowriting it or unloading it.
1999 */
2000 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002001can_abandon(buf_T *buf, int forceit)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002002{
2003 return ( P_HID(buf)
2004 || !bufIsChanged(buf)
2005 || buf->b_nwindows > 1
2006 || autowrite(buf, forceit) == OK
2007 || forceit);
2008}
2009
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01002010static void add_bufnum(int *bufnrs, int *bufnump, int nr);
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002011
2012/*
2013 * Add a buffer number to "bufnrs", unless it's already there.
2014 */
2015 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002016add_bufnum(int *bufnrs, int *bufnump, int nr)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002017{
2018 int i;
2019
2020 for (i = 0; i < *bufnump; ++i)
2021 if (bufnrs[i] == nr)
2022 return;
2023 bufnrs[*bufnump] = nr;
2024 *bufnump = *bufnump + 1;
2025}
2026
Bram Moolenaar071d4272004-06-13 20:20:40 +00002027/*
2028 * Return TRUE if any buffer was changed and cannot be abandoned.
2029 * That changed buffer becomes the current buffer.
Bram Moolenaar027387f2016-01-02 22:25:52 +01002030 * When "unload" is true the current buffer is unloaded instead of making it
2031 * hidden. This is used for ":q!".
Bram Moolenaar071d4272004-06-13 20:20:40 +00002032 */
2033 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002034check_changed_any(
2035 int hidden, /* Only check hidden buffers */
2036 int unload)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002037{
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002038 int ret = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002039 buf_T *buf;
2040 int save;
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002041 int i;
2042 int bufnum = 0;
2043 int bufcount = 0;
2044 int *bufnrs;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002045#ifdef FEAT_WINDOWS
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002046 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002047 win_T *wp;
2048#endif
2049
Bram Moolenaar29323592016-07-24 22:04:11 +02002050 FOR_ALL_BUFFERS(buf)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002051 ++bufcount;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002052
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002053 if (bufcount == 0)
2054 return FALSE;
2055
2056 bufnrs = (int *)alloc(sizeof(int) * bufcount);
2057 if (bufnrs == NULL)
2058 return FALSE;
2059
2060 /* curbuf */
2061 bufnrs[bufnum++] = curbuf->b_fnum;
2062#ifdef FEAT_WINDOWS
2063 /* buf in curtab */
2064 FOR_ALL_WINDOWS(wp)
2065 if (wp->w_buffer != curbuf)
2066 add_bufnum(bufnrs, &bufnum, wp->w_buffer->b_fnum);
2067
2068 /* buf in other tab */
Bram Moolenaar29323592016-07-24 22:04:11 +02002069 FOR_ALL_TABPAGES(tp)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002070 if (tp != curtab)
2071 for (wp = tp->tp_firstwin; wp != NULL; wp = wp->w_next)
2072 add_bufnum(bufnrs, &bufnum, wp->w_buffer->b_fnum);
2073#endif
2074 /* any other buf */
Bram Moolenaar29323592016-07-24 22:04:11 +02002075 FOR_ALL_BUFFERS(buf)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002076 add_bufnum(bufnrs, &bufnum, buf->b_fnum);
2077
2078 for (i = 0; i < bufnum; ++i)
2079 {
2080 buf = buflist_findnr(bufnrs[i]);
2081 if (buf == NULL)
2082 continue;
2083 if ((!hidden || buf->b_nwindows == 0) && bufIsChanged(buf))
2084 {
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002085 bufref_T bufref;
2086
2087 set_bufref(&bufref, buf);
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002088 /* Try auto-writing the buffer. If this fails but the buffer no
2089 * longer exists it's not changed, that's OK. */
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002090 if (check_changed(buf, (p_awa ? CCGD_AW : 0)
2091 | CCGD_MULTWIN
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002092 | CCGD_ALLBUF) && bufref_valid(&bufref))
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002093 break; /* didn't save - still changes */
2094 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002095 }
2096
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002097 if (i >= bufnum)
2098 goto theend;
2099
2100 ret = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002101 exiting = FALSE;
2102#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG)
2103 /*
2104 * When ":confirm" used, don't give an error message.
2105 */
2106 if (!(p_confirm || cmdmod.confirm))
2107#endif
2108 {
2109 /* There must be a wait_return for this message, do_buffer()
2110 * may cause a redraw. But wait_return() is a no-op when vgetc()
2111 * is busy (Quit used from window menu), then make sure we don't
2112 * cause a scroll up. */
Bram Moolenaar61660ea2006-04-07 21:40:07 +00002113 if (vgetc_busy > 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002114 {
2115 msg_row = cmdline_row;
2116 msg_col = 0;
2117 msg_didout = FALSE;
2118 }
2119 if (EMSG2(_("E162: No write since last change for buffer \"%s\""),
Bram Moolenaare1704ba2012-10-03 18:25:00 +02002120 buf_spname(buf) != NULL ? buf_spname(buf) : buf->b_fname))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002121 {
2122 save = no_wait_return;
2123 no_wait_return = FALSE;
2124 wait_return(FALSE);
2125 no_wait_return = save;
2126 }
2127 }
2128
2129#ifdef FEAT_WINDOWS
2130 /* Try to find a window that contains the buffer. */
2131 if (buf != curbuf)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002132 FOR_ALL_TAB_WINDOWS(tp, wp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002133 if (wp->w_buffer == buf)
2134 {
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002135# ifdef FEAT_AUTOCMD
2136 bufref_T bufref;
2137
2138 set_bufref(&bufref, buf);
2139# endif
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002140 goto_tabpage_win(tp, wp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002141# ifdef FEAT_AUTOCMD
2142 /* Paranoia: did autocms wipe out the buffer with changes? */
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002143 if (!bufref_valid(&bufref))
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002144 {
2145 goto theend;
2146 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002147# endif
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002148 goto buf_found;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002149 }
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002150buf_found:
Bram Moolenaar071d4272004-06-13 20:20:40 +00002151#endif
2152
2153 /* Open the changed buffer in the current window. */
2154 if (buf != curbuf)
Bram Moolenaar027387f2016-01-02 22:25:52 +01002155 set_curbuf(buf, unload ? DOBUF_UNLOAD : DOBUF_GOTO);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002156
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002157theend:
2158 vim_free(bufnrs);
2159 return ret;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002160}
2161
2162/*
2163 * return FAIL if there is no file name, OK if there is one
2164 * give error message for FAIL
2165 */
2166 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002167check_fname(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002168{
2169 if (curbuf->b_ffname == NULL)
2170 {
2171 EMSG(_(e_noname));
2172 return FAIL;
2173 }
2174 return OK;
2175}
2176
2177/*
2178 * flush the contents of a buffer, unless it has no file name
2179 *
2180 * return FAIL for failure, OK otherwise
2181 */
2182 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002183buf_write_all(buf_T *buf, int forceit)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002184{
2185 int retval;
2186#ifdef FEAT_AUTOCMD
2187 buf_T *old_curbuf = curbuf;
2188#endif
2189
2190 retval = (buf_write(buf, buf->b_ffname, buf->b_fname,
2191 (linenr_T)1, buf->b_ml.ml_line_count, NULL,
2192 FALSE, forceit, TRUE, FALSE));
2193#ifdef FEAT_AUTOCMD
2194 if (curbuf != old_curbuf)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00002195 {
2196 msg_source(hl_attr(HLF_W));
Bram Moolenaar071d4272004-06-13 20:20:40 +00002197 MSG(_("Warning: Entered other buffer unexpectedly (check autocommands)"));
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00002198 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002199#endif
2200 return retval;
2201}
2202
2203/*
2204 * Code to handle the argument list.
2205 */
2206
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01002207static char_u *do_one_arg(char_u *str);
2208static int do_arglist(char_u *str, int what, int after);
2209static void alist_check_arg_idx(void);
2210static int editing_arg_idx(win_T *win);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002211#ifdef FEAT_LISTCMDS
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01002212static int alist_add_list(int count, char_u **files, int after);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002213#endif
2214#define AL_SET 1
2215#define AL_ADD 2
2216#define AL_DEL 3
2217
Bram Moolenaar071d4272004-06-13 20:20:40 +00002218/*
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002219 * Isolate one argument, taking backticks.
2220 * Changes the argument in-place, puts a NUL after it. Backticks remain.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002221 * Return a pointer to the start of the next argument.
2222 */
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002223 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002224do_one_arg(char_u *str)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002225{
2226 char_u *p;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002227 int inbacktick;
2228
Bram Moolenaar071d4272004-06-13 20:20:40 +00002229 inbacktick = FALSE;
2230 for (p = str; *str; ++str)
2231 {
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002232 /* When the backslash is used for escaping the special meaning of a
2233 * character we need to keep it until wildcard expansion. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002234 if (rem_backslash(str))
2235 {
2236 *p++ = *str++;
2237 *p++ = *str;
2238 }
2239 else
2240 {
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002241 /* An item ends at a space not in backticks */
2242 if (!inbacktick && vim_isspace(*str))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002243 break;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002244 if (*str == '`')
Bram Moolenaar071d4272004-06-13 20:20:40 +00002245 inbacktick ^= TRUE;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002246 *p++ = *str;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002247 }
2248 }
2249 str = skipwhite(str);
2250 *p = NUL;
2251
2252 return str;
2253}
2254
Bram Moolenaar86b68352004-12-27 21:59:20 +00002255/*
2256 * Separate the arguments in "str" and return a list of pointers in the
2257 * growarray "gap".
2258 */
2259 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002260get_arglist(garray_T *gap, char_u *str)
Bram Moolenaar86b68352004-12-27 21:59:20 +00002261{
2262 ga_init2(gap, (int)sizeof(char_u *), 20);
2263 while (*str != NUL)
2264 {
2265 if (ga_grow(gap, 1) == FAIL)
2266 {
2267 ga_clear(gap);
2268 return FAIL;
2269 }
2270 ((char_u **)gap->ga_data)[gap->ga_len++] = str;
2271
2272 /* Isolate one argument, change it in-place, put a NUL after it. */
2273 str = do_one_arg(str);
2274 }
2275 return OK;
2276}
2277
Bram Moolenaar7df351e2006-01-23 22:30:28 +00002278#if defined(FEAT_QUICKFIX) || defined(FEAT_SYN_HL) || defined(PROTO)
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002279/*
2280 * Parse a list of arguments (file names), expand them and return in
Bram Moolenaar8f5c6f02012-06-29 12:57:06 +02002281 * "fnames[fcountp]". When "wig" is TRUE, removes files matching 'wildignore'.
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002282 * Return FAIL or OK.
2283 */
2284 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002285get_arglist_exp(
2286 char_u *str,
2287 int *fcountp,
2288 char_u ***fnamesp,
2289 int wig)
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002290{
2291 garray_T ga;
2292 int i;
2293
2294 if (get_arglist(&ga, str) == FAIL)
2295 return FAIL;
Bram Moolenaar8f5c6f02012-06-29 12:57:06 +02002296 if (wig == TRUE)
2297 i = expand_wildcards(ga.ga_len, (char_u **)ga.ga_data,
2298 fcountp, fnamesp, EW_FILE|EW_NOTFOUND);
2299 else
2300 i = gen_expand_wildcards(ga.ga_len, (char_u **)ga.ga_data,
2301 fcountp, fnamesp, EW_FILE|EW_NOTFOUND);
2302
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002303 ga_clear(&ga);
2304 return i;
2305}
2306#endif
2307
Bram Moolenaar071d4272004-06-13 20:20:40 +00002308#if defined(FEAT_GUI) || defined(FEAT_CLIENTSERVER) || defined(PROTO)
2309/*
2310 * Redefine the argument list.
2311 */
2312 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002313set_arglist(char_u *str)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002314{
2315 do_arglist(str, AL_SET, 0);
2316}
2317#endif
2318
2319/*
2320 * "what" == AL_SET: Redefine the argument list to 'str'.
2321 * "what" == AL_ADD: add files in 'str' to the argument list after "after".
2322 * "what" == AL_DEL: remove files in 'str' from the argument list.
2323 *
2324 * Return FAIL for failure, OK otherwise.
2325 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002326 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002327do_arglist(
2328 char_u *str,
Bram Moolenaarf1d25012016-03-03 12:22:53 +01002329 int what,
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002330 int after UNUSED) /* 0 means before first one */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002331{
2332 garray_T new_ga;
2333 int exp_count;
2334 char_u **exp_files;
2335 int i;
2336#ifdef FEAT_LISTCMDS
2337 char_u *p;
2338 int match;
2339#endif
2340
2341 /*
Bram Moolenaar2faa29f2016-01-23 23:02:34 +01002342 * Set default argument for ":argadd" command.
2343 */
2344 if (what == AL_ADD && *str == NUL)
2345 {
2346 if (curbuf->b_ffname == NULL)
2347 return FAIL;
2348 str = curbuf->b_fname;
2349 }
2350
2351 /*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002352 * Collect all file name arguments in "new_ga".
2353 */
Bram Moolenaar86b68352004-12-27 21:59:20 +00002354 if (get_arglist(&new_ga, str) == FAIL)
2355 return FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002356
2357#ifdef FEAT_LISTCMDS
2358 if (what == AL_DEL)
2359 {
2360 regmatch_T regmatch;
2361 int didone;
2362
2363 /*
2364 * Delete the items: use each item as a regexp and find a match in the
2365 * argument list.
2366 */
Bram Moolenaar71afbfe2013-03-19 16:49:16 +01002367 regmatch.rm_ic = p_fic; /* ignore case when 'fileignorecase' is set */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002368 for (i = 0; i < new_ga.ga_len && !got_int; ++i)
2369 {
2370 p = ((char_u **)new_ga.ga_data)[i];
2371 p = file_pat_to_reg_pat(p, NULL, NULL, FALSE);
2372 if (p == NULL)
2373 break;
2374 regmatch.regprog = vim_regcomp(p, p_magic ? RE_MAGIC : 0);
2375 if (regmatch.regprog == NULL)
2376 {
2377 vim_free(p);
2378 break;
2379 }
2380
2381 didone = FALSE;
2382 for (match = 0; match < ARGCOUNT; ++match)
2383 if (vim_regexec(&regmatch, alist_name(&ARGLIST[match]),
2384 (colnr_T)0))
2385 {
2386 didone = TRUE;
2387 vim_free(ARGLIST[match].ae_fname);
2388 mch_memmove(ARGLIST + match, ARGLIST + match + 1,
2389 (ARGCOUNT - match - 1) * sizeof(aentry_T));
2390 --ALIST(curwin)->al_ga.ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002391 if (curwin->w_arg_idx > match)
2392 --curwin->w_arg_idx;
2393 --match;
2394 }
2395
Bram Moolenaar473de612013-06-08 18:19:48 +02002396 vim_regfree(regmatch.regprog);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002397 vim_free(p);
2398 if (!didone)
2399 EMSG2(_(e_nomatch2), ((char_u **)new_ga.ga_data)[i]);
2400 }
2401 ga_clear(&new_ga);
2402 }
2403 else
2404#endif
2405 {
2406 i = expand_wildcards(new_ga.ga_len, (char_u **)new_ga.ga_data,
2407 &exp_count, &exp_files, EW_DIR|EW_FILE|EW_ADDSLASH|EW_NOTFOUND);
2408 ga_clear(&new_ga);
Bram Moolenaar2db5c3b2016-01-16 22:49:34 +01002409 if (i == FAIL || exp_count == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002410 {
2411 EMSG(_(e_nomatch));
2412 return FAIL;
2413 }
2414
2415#ifdef FEAT_LISTCMDS
2416 if (what == AL_ADD)
2417 {
2418 (void)alist_add_list(exp_count, exp_files, after);
2419 vim_free(exp_files);
2420 }
2421 else /* what == AL_SET */
2422#endif
Bram Moolenaar86b68352004-12-27 21:59:20 +00002423 alist_set(ALIST(curwin), exp_count, exp_files, FALSE, NULL, 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002424 }
2425
2426 alist_check_arg_idx();
2427
2428 return OK;
2429}
2430
2431/*
2432 * Check the validity of the arg_idx for each other window.
2433 */
2434 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002435alist_check_arg_idx(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002436{
2437#ifdef FEAT_WINDOWS
2438 win_T *win;
Bram Moolenaarf740b292006-02-16 22:11:02 +00002439 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002440
Bram Moolenaarf740b292006-02-16 22:11:02 +00002441 FOR_ALL_TAB_WINDOWS(tp, win)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002442 if (win->w_alist == curwin->w_alist)
2443 check_arg_idx(win);
2444#else
2445 check_arg_idx(curwin);
2446#endif
2447}
2448
2449/*
Bram Moolenaarb8ff1fb2012-02-04 21:59:01 +01002450 * Return TRUE if window "win" is editing the file at the current argument
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002451 * index.
2452 */
2453 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002454editing_arg_idx(win_T *win)
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002455{
2456 return !(win->w_arg_idx >= WARGCOUNT(win)
2457 || (win->w_buffer->b_fnum
2458 != WARGLIST(win)[win->w_arg_idx].ae_fnum
2459 && (win->w_buffer->b_ffname == NULL
2460 || !(fullpathcmp(
2461 alist_name(&WARGLIST(win)[win->w_arg_idx]),
2462 win->w_buffer->b_ffname, TRUE) & FPC_SAME))));
2463}
2464
2465/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002466 * Check if window "win" is editing the w_arg_idx file in its argument list.
2467 */
2468 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002469check_arg_idx(win_T *win)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002470{
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002471 if (WARGCOUNT(win) > 1 && !editing_arg_idx(win))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002472 {
2473 /* We are not editing the current entry in the argument list.
2474 * Set "arg_had_last" if we are editing the last one. */
2475 win->w_arg_idx_invalid = TRUE;
2476 if (win->w_arg_idx != WARGCOUNT(win) - 1
2477 && arg_had_last == FALSE
2478#ifdef FEAT_WINDOWS
2479 && ALIST(win) == &global_alist
2480#endif
2481 && GARGCOUNT > 0
2482 && win->w_arg_idx < GARGCOUNT
2483 && (win->w_buffer->b_fnum == GARGLIST[GARGCOUNT - 1].ae_fnum
2484 || (win->w_buffer->b_ffname != NULL
2485 && (fullpathcmp(alist_name(&GARGLIST[GARGCOUNT - 1]),
2486 win->w_buffer->b_ffname, TRUE) & FPC_SAME))))
2487 arg_had_last = TRUE;
2488 }
2489 else
2490 {
2491 /* We are editing the current entry in the argument list.
2492 * Set "arg_had_last" if it's also the last one */
2493 win->w_arg_idx_invalid = FALSE;
2494 if (win->w_arg_idx == WARGCOUNT(win) - 1
2495#ifdef FEAT_WINDOWS
2496 && win->w_alist == &global_alist
2497#endif
2498 )
2499 arg_had_last = TRUE;
2500 }
2501}
2502
2503/*
2504 * ":args", ":argslocal" and ":argsglobal".
2505 */
2506 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002507ex_args(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002508{
2509 int i;
2510
2511 if (eap->cmdidx != CMD_args)
2512 {
2513#if defined(FEAT_WINDOWS) && defined(FEAT_LISTCMDS)
2514 alist_unlink(ALIST(curwin));
2515 if (eap->cmdidx == CMD_argglobal)
2516 ALIST(curwin) = &global_alist;
2517 else /* eap->cmdidx == CMD_arglocal */
2518 alist_new();
2519#else
2520 ex_ni(eap);
2521 return;
2522#endif
2523 }
2524
2525 if (!ends_excmd(*eap->arg))
2526 {
2527 /*
2528 * ":args file ..": define new argument list, handle like ":next"
2529 * Also for ":argslocal file .." and ":argsglobal file ..".
2530 */
2531 ex_next(eap);
2532 }
2533 else
2534#if defined(FEAT_WINDOWS) && defined(FEAT_LISTCMDS)
2535 if (eap->cmdidx == CMD_args)
2536#endif
2537 {
2538 /*
2539 * ":args": list arguments.
2540 */
2541 if (ARGCOUNT > 0)
2542 {
2543 /* Overwrite the command, for a short list there is no scrolling
2544 * required and no wait_return(). */
2545 gotocmdline(TRUE);
2546 for (i = 0; i < ARGCOUNT; ++i)
2547 {
2548 if (i == curwin->w_arg_idx)
2549 msg_putchar('[');
2550 msg_outtrans(alist_name(&ARGLIST[i]));
2551 if (i == curwin->w_arg_idx)
2552 msg_putchar(']');
2553 msg_putchar(' ');
2554 }
2555 }
2556 }
2557#if defined(FEAT_WINDOWS) && defined(FEAT_LISTCMDS)
2558 else if (eap->cmdidx == CMD_arglocal)
2559 {
2560 garray_T *gap = &curwin->w_alist->al_ga;
2561
2562 /*
2563 * ":argslocal": make a local copy of the global argument list.
2564 */
2565 if (ga_grow(gap, GARGCOUNT) == OK)
2566 for (i = 0; i < GARGCOUNT; ++i)
2567 if (GARGLIST[i].ae_fname != NULL)
2568 {
2569 AARGLIST(curwin->w_alist)[gap->ga_len].ae_fname =
2570 vim_strsave(GARGLIST[i].ae_fname);
2571 AARGLIST(curwin->w_alist)[gap->ga_len].ae_fnum =
2572 GARGLIST[i].ae_fnum;
2573 ++gap->ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002574 }
2575 }
2576#endif
2577}
2578
2579/*
2580 * ":previous", ":sprevious", ":Next" and ":sNext".
2581 */
2582 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002583ex_previous(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002584{
2585 /* If past the last one already, go to the last one. */
2586 if (curwin->w_arg_idx - (int)eap->line2 >= ARGCOUNT)
2587 do_argfile(eap, ARGCOUNT - 1);
2588 else
2589 do_argfile(eap, curwin->w_arg_idx - (int)eap->line2);
2590}
2591
2592/*
2593 * ":rewind", ":first", ":sfirst" and ":srewind".
2594 */
2595 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002596ex_rewind(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002597{
2598 do_argfile(eap, 0);
2599}
2600
2601/*
2602 * ":last" and ":slast".
2603 */
2604 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002605ex_last(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002606{
2607 do_argfile(eap, ARGCOUNT - 1);
2608}
2609
2610/*
2611 * ":argument" and ":sargument".
2612 */
2613 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002614ex_argument(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002615{
2616 int i;
2617
2618 if (eap->addr_count > 0)
2619 i = eap->line2 - 1;
2620 else
2621 i = curwin->w_arg_idx;
2622 do_argfile(eap, i);
2623}
2624
2625/*
2626 * Edit file "argn" of the argument lists.
2627 */
2628 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002629do_argfile(exarg_T *eap, int argn)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002630{
2631 int other;
2632 char_u *p;
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002633 int old_arg_idx = curwin->w_arg_idx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002634
2635 if (argn < 0 || argn >= ARGCOUNT)
2636 {
2637 if (ARGCOUNT <= 1)
2638 EMSG(_("E163: There is only one file to edit"));
2639 else if (argn < 0)
2640 EMSG(_("E164: Cannot go before first file"));
2641 else
2642 EMSG(_("E165: Cannot go beyond last file"));
2643 }
2644 else
2645 {
2646 setpcmark();
2647#ifdef FEAT_GUI
2648 need_mouse_correct = TRUE;
2649#endif
2650
2651#ifdef FEAT_WINDOWS
Bram Moolenaardf1bdc92006-02-23 21:32:16 +00002652 /* split window or create new tab page first */
2653 if (*eap->cmd == 's' || cmdmod.tab != 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002654 {
2655 if (win_split(0, 0) == FAIL)
2656 return;
Bram Moolenaar3368ea22010-09-21 16:56:35 +02002657 RESET_BINDING(curwin);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002658 }
2659 else
2660#endif
2661 {
2662 /*
2663 * if 'hidden' set, only check for changed file when re-editing
2664 * the same buffer
2665 */
2666 other = TRUE;
2667 if (P_HID(curbuf))
2668 {
2669 p = fix_fname(alist_name(&ARGLIST[argn]));
2670 other = otherfile(p);
2671 vim_free(p);
2672 }
2673 if ((!P_HID(curbuf) || !other)
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002674 && check_changed(curbuf, CCGD_AW
2675 | (other ? 0 : CCGD_MULTWIN)
2676 | (eap->forceit ? CCGD_FORCEIT : 0)
2677 | CCGD_EXCMD))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002678 return;
2679 }
2680
2681 curwin->w_arg_idx = argn;
2682 if (argn == ARGCOUNT - 1
2683#ifdef FEAT_WINDOWS
2684 && curwin->w_alist == &global_alist
2685#endif
2686 )
2687 arg_had_last = TRUE;
2688
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002689 /* Edit the file; always use the last known line number.
2690 * When it fails (e.g. Abort for already edited file) restore the
2691 * argument index. */
2692 if (do_ecmd(0, alist_name(&ARGLIST[curwin->w_arg_idx]), NULL,
Bram Moolenaar071d4272004-06-13 20:20:40 +00002693 eap, ECMD_LAST,
Bram Moolenaar701f7af2008-11-15 13:12:07 +00002694 (P_HID(curwin->w_buffer) ? ECMD_HIDE : 0)
2695 + (eap->forceit ? ECMD_FORCEIT : 0), curwin) == FAIL)
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002696 curwin->w_arg_idx = old_arg_idx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002697 /* like Vi: set the mark where the cursor is in the file. */
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002698 else if (eap->cmdidx != CMD_argdo)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002699 setmark('\'');
2700 }
2701}
2702
2703/*
2704 * ":next", and commands that behave like it.
2705 */
2706 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002707ex_next(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002708{
2709 int i;
2710
2711 /*
2712 * check for changed buffer now, if this fails the argument list is not
2713 * redefined.
2714 */
2715 if ( P_HID(curbuf)
2716 || eap->cmdidx == CMD_snext
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002717 || !check_changed(curbuf, CCGD_AW
2718 | (eap->forceit ? CCGD_FORCEIT : 0)
2719 | CCGD_EXCMD))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002720 {
2721 if (*eap->arg != NUL) /* redefine file list */
2722 {
2723 if (do_arglist(eap->arg, AL_SET, 0) == FAIL)
2724 return;
2725 i = 0;
2726 }
2727 else
2728 i = curwin->w_arg_idx + (int)eap->line2;
2729 do_argfile(eap, i);
2730 }
2731}
2732
Bram Moolenaar0106e3d2016-02-23 18:55:43 +01002733#if defined(FEAT_LISTCMDS) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002734/*
2735 * ":argedit"
2736 */
2737 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002738ex_argedit(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002739{
2740 int fnum;
2741 int i;
2742 char_u *s;
2743
2744 /* Add the argument to the buffer list and get the buffer number. */
2745 fnum = buflist_add(eap->arg, BLN_LISTED);
2746
2747 /* Check if this argument is already in the argument list. */
2748 for (i = 0; i < ARGCOUNT; ++i)
2749 if (ARGLIST[i].ae_fnum == fnum)
2750 break;
2751 if (i == ARGCOUNT)
2752 {
2753 /* Can't find it, add it to the argument list. */
2754 s = vim_strsave(eap->arg);
2755 if (s == NULL)
2756 return;
2757 i = alist_add_list(1, &s,
2758 eap->addr_count > 0 ? (int)eap->line2 : curwin->w_arg_idx + 1);
2759 if (i < 0)
2760 return;
2761 curwin->w_arg_idx = i;
2762 }
2763
2764 alist_check_arg_idx();
2765
2766 /* Edit the argument. */
2767 do_argfile(eap, i);
2768}
2769
2770/*
2771 * ":argadd"
2772 */
2773 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002774ex_argadd(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002775{
2776 do_arglist(eap->arg, AL_ADD,
2777 eap->addr_count > 0 ? (int)eap->line2 : curwin->w_arg_idx + 1);
2778#ifdef FEAT_TITLE
2779 maketitle();
2780#endif
2781}
2782
2783/*
2784 * ":argdelete"
2785 */
2786 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002787ex_argdelete(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002788{
2789 int i;
2790 int n;
2791
2792 if (eap->addr_count > 0)
2793 {
2794 /* ":1,4argdel": Delete all arguments in the range. */
2795 if (eap->line2 > ARGCOUNT)
2796 eap->line2 = ARGCOUNT;
2797 n = eap->line2 - eap->line1 + 1;
2798 if (*eap->arg != NUL || n <= 0)
2799 EMSG(_(e_invarg));
2800 else
2801 {
2802 for (i = eap->line1; i <= eap->line2; ++i)
2803 vim_free(ARGLIST[i - 1].ae_fname);
2804 mch_memmove(ARGLIST + eap->line1 - 1, ARGLIST + eap->line2,
2805 (size_t)((ARGCOUNT - eap->line2) * sizeof(aentry_T)));
2806 ALIST(curwin)->al_ga.ga_len -= n;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002807 if (curwin->w_arg_idx >= eap->line2)
2808 curwin->w_arg_idx -= n;
2809 else if (curwin->w_arg_idx > eap->line1)
2810 curwin->w_arg_idx = eap->line1;
Bram Moolenaar72defda2016-01-17 18:04:33 +01002811 if (ARGCOUNT == 0)
2812 curwin->w_arg_idx = 0;
2813 else if (curwin->w_arg_idx >= ARGCOUNT)
2814 curwin->w_arg_idx = ARGCOUNT - 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002815 }
2816 }
2817 else if (*eap->arg == NUL)
2818 EMSG(_(e_argreq));
2819 else
2820 do_arglist(eap->arg, AL_DEL, 0);
2821#ifdef FEAT_TITLE
2822 maketitle();
2823#endif
2824}
2825
2826/*
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002827 * ":argdo", ":windo", ":bufdo", ":tabdo", ":cdo", ":ldo", ":cfdo" and ":lfdo"
Bram Moolenaar071d4272004-06-13 20:20:40 +00002828 */
2829 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002830ex_listdo(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002831{
2832 int i;
2833#ifdef FEAT_WINDOWS
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002834 win_T *wp;
2835 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002836#endif
Bram Moolenaare25bb902015-02-27 20:33:37 +01002837 buf_T *buf = curbuf;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002838 int next_fnum = 0;
2839#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
2840 char_u *save_ei = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002841#endif
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002842 char_u *p_shm_save;
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002843#ifdef FEAT_QUICKFIX
Bram Moolenaared84b762015-09-09 22:35:29 +02002844 int qf_size = 0;
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002845 int qf_idx;
2846#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002847
2848#ifndef FEAT_WINDOWS
2849 if (eap->cmdidx == CMD_windo)
2850 {
2851 ex_ni(eap);
2852 return;
2853 }
2854#endif
2855
Bram Moolenaar0106e3d2016-02-23 18:55:43 +01002856#ifndef FEAT_QUICKFIX
2857 if (eap->cmdidx == CMD_cdo || eap->cmdidx == CMD_ldo ||
2858 eap->cmdidx == CMD_cfdo || eap->cmdidx == CMD_lfdo)
2859 {
2860 ex_ni(eap);
2861 return;
2862 }
2863#endif
2864
Bram Moolenaar071d4272004-06-13 20:20:40 +00002865#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002866 if (eap->cmdidx != CMD_windo && eap->cmdidx != CMD_tabdo)
Bram Moolenaardcaf10e2005-01-21 11:55:25 +00002867 /* Don't do syntax HL autocommands. Skipping the syntax file is a
2868 * great speed improvement. */
2869 save_ei = au_event_disable(",Syntax");
Bram Moolenaar071d4272004-06-13 20:20:40 +00002870#endif
Bram Moolenaar6b1ee342014-08-06 18:17:11 +02002871#ifdef FEAT_CLIPBOARD
2872 start_global_changes();
2873#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002874
2875 if (eap->cmdidx == CMD_windo
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002876 || eap->cmdidx == CMD_tabdo
Bram Moolenaar071d4272004-06-13 20:20:40 +00002877 || P_HID(curbuf)
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002878 || !check_changed(curbuf, CCGD_AW
2879 | (eap->forceit ? CCGD_FORCEIT : 0)
2880 | CCGD_EXCMD))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002881 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00002882 i = 0;
Bram Moolenaara162bc52015-01-07 16:54:21 +01002883 /* start at the eap->line1 argument/window/buffer */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002884#ifdef FEAT_WINDOWS
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002885 wp = firstwin;
2886 tp = first_tabpage;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002887#endif
Bram Moolenaara162bc52015-01-07 16:54:21 +01002888 switch (eap->cmdidx)
2889 {
2890#ifdef FEAT_WINDOWS
2891 case CMD_windo:
2892 for ( ; wp != NULL && i + 1 < eap->line1; wp = wp->w_next)
2893 i++;
2894 break;
2895 case CMD_tabdo:
2896 for( ; tp != NULL && i + 1 < eap->line1; tp = tp->tp_next)
2897 i++;
2898 break;
2899#endif
2900 case CMD_argdo:
2901 i = eap->line1 - 1;
2902 break;
Bram Moolenaara162bc52015-01-07 16:54:21 +01002903 default:
2904 break;
2905 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002906 /* set pcmark now */
2907 if (eap->cmdidx == CMD_bufdo)
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002908 {
Bram Moolenaare25bb902015-02-27 20:33:37 +01002909 /* Advance to the first listed buffer after "eap->line1". */
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002910 for (buf = firstbuf; buf != NULL && (buf->b_fnum < eap->line1
Bram Moolenaare25bb902015-02-27 20:33:37 +01002911 || !buf->b_p_bl); buf = buf->b_next)
2912 if (buf->b_fnum > eap->line2)
2913 {
2914 buf = NULL;
2915 break;
2916 }
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002917 if (buf != NULL)
Bram Moolenaare25bb902015-02-27 20:33:37 +01002918 goto_buffer(eap, DOBUF_FIRST, FORWARD, buf->b_fnum);
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002919 }
2920#ifdef FEAT_QUICKFIX
2921 else if (eap->cmdidx == CMD_cdo || eap->cmdidx == CMD_ldo
2922 || eap->cmdidx == CMD_cfdo || eap->cmdidx == CMD_lfdo)
2923 {
2924 qf_size = qf_get_size(eap);
2925 if (qf_size <= 0 || eap->line1 > qf_size)
2926 buf = NULL;
2927 else
2928 {
2929 ex_cc(eap);
2930
2931 buf = curbuf;
2932 i = eap->line1 - 1;
2933 if (eap->addr_count <= 0)
2934 /* default is all the quickfix/location list entries */
2935 eap->line2 = qf_size;
2936 }
2937 }
2938#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002939 else
2940 setpcmark();
2941 listcmd_busy = TRUE; /* avoids setting pcmark below */
2942
Bram Moolenaare25bb902015-02-27 20:33:37 +01002943 while (!got_int && buf != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002944 {
2945 if (eap->cmdidx == CMD_argdo)
2946 {
2947 /* go to argument "i" */
2948 if (i == ARGCOUNT)
2949 break;
2950 /* Don't call do_argfile() when already there, it will try
2951 * reloading the file. */
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002952 if (curwin->w_arg_idx != i || !editing_arg_idx(curwin))
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002953 {
2954 /* Clear 'shm' to avoid that the file message overwrites
2955 * any output from the command. */
2956 p_shm_save = vim_strsave(p_shm);
2957 set_option_value((char_u *)"shm", 0L, (char_u *)"", 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002958 do_argfile(eap, i);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002959 set_option_value((char_u *)"shm", 0L, p_shm_save, 0);
2960 vim_free(p_shm_save);
2961 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002962 if (curwin->w_arg_idx != i)
2963 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002964 }
2965#ifdef FEAT_WINDOWS
2966 else if (eap->cmdidx == CMD_windo)
2967 {
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002968 /* go to window "wp" */
2969 if (!win_valid(wp))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002970 break;
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002971 win_goto(wp);
Bram Moolenaar41423242007-05-03 20:11:13 +00002972 if (curwin != wp)
2973 break; /* something must be wrong */
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002974 wp = curwin->w_next;
2975 }
2976 else if (eap->cmdidx == CMD_tabdo)
2977 {
2978 /* go to window "tp" */
2979 if (!valid_tabpage(tp))
2980 break;
Bram Moolenaar49e649f2013-05-06 04:50:35 +02002981 goto_tabpage_tp(tp, TRUE, TRUE);
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002982 tp = tp->tp_next;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002983 }
2984#endif
2985 else if (eap->cmdidx == CMD_bufdo)
2986 {
2987 /* Remember the number of the next listed buffer, in case
2988 * ":bwipe" is used or autocommands do something strange. */
2989 next_fnum = -1;
2990 for (buf = curbuf->b_next; buf != NULL; buf = buf->b_next)
2991 if (buf->b_p_bl)
2992 {
2993 next_fnum = buf->b_fnum;
2994 break;
2995 }
2996 }
2997
Bram Moolenaara162bc52015-01-07 16:54:21 +01002998 ++i;
2999
Bram Moolenaar071d4272004-06-13 20:20:40 +00003000 /* execute the command */
3001 do_cmdline(eap->arg, eap->getline, eap->cookie,
3002 DOCMD_VERBOSE + DOCMD_NOWAIT);
3003
3004 if (eap->cmdidx == CMD_bufdo)
3005 {
3006 /* Done? */
Bram Moolenaara162bc52015-01-07 16:54:21 +01003007 if (next_fnum < 0 || next_fnum > eap->line2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003008 break;
3009 /* Check if the buffer still exists. */
Bram Moolenaar29323592016-07-24 22:04:11 +02003010 FOR_ALL_BUFFERS(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003011 if (buf->b_fnum == next_fnum)
3012 break;
3013 if (buf == NULL)
3014 break;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003015
3016 /* Go to the next buffer. Clear 'shm' to avoid that the file
3017 * message overwrites any output from the command. */
3018 p_shm_save = vim_strsave(p_shm);
3019 set_option_value((char_u *)"shm", 0L, (char_u *)"", 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003020 goto_buffer(eap, DOBUF_FIRST, FORWARD, next_fnum);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003021 set_option_value((char_u *)"shm", 0L, p_shm_save, 0);
3022 vim_free(p_shm_save);
3023
Bram Moolenaaraa23b372015-09-08 18:46:31 +02003024 /* If autocommands took us elsewhere, quit here. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003025 if (curbuf->b_fnum != next_fnum)
3026 break;
3027 }
3028
Bram Moolenaaraa23b372015-09-08 18:46:31 +02003029#ifdef FEAT_QUICKFIX
3030 if (eap->cmdidx == CMD_cdo || eap->cmdidx == CMD_ldo
3031 || eap->cmdidx == CMD_cfdo || eap->cmdidx == CMD_lfdo)
3032 {
3033 if (i >= qf_size || i >= eap->line2)
3034 break;
3035
3036 qf_idx = qf_get_cur_idx(eap);
3037
3038 ex_cnext(eap);
3039
3040 /* If jumping to the next quickfix entry fails, quit here */
3041 if (qf_get_cur_idx(eap) == qf_idx)
3042 break;
3043 }
3044#endif
3045
Bram Moolenaar071d4272004-06-13 20:20:40 +00003046 if (eap->cmdidx == CMD_windo)
3047 {
3048 validate_cursor(); /* cursor may have moved */
3049#ifdef FEAT_SCROLLBIND
3050 /* required when 'scrollbind' has been set */
3051 if (curwin->w_p_scb)
3052 do_check_scrollbind(TRUE);
3053#endif
3054 }
Bram Moolenaara162bc52015-01-07 16:54:21 +01003055
3056#ifdef FEAT_WINDOWS
3057 if (eap->cmdidx == CMD_windo || eap->cmdidx == CMD_tabdo)
3058 if (i+1 > eap->line2)
3059 break;
3060#endif
3061 if (eap->cmdidx == CMD_argdo && i >= eap->line2)
3062 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003063 }
3064 listcmd_busy = FALSE;
3065 }
3066
3067#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
Bram Moolenaar6ac54292005-02-02 23:07:25 +00003068 if (save_ei != NULL)
3069 {
3070 au_event_restore(save_ei);
3071 apply_autocmds(EVENT_SYNTAX, curbuf->b_p_syn,
3072 curbuf->b_fname, TRUE, curbuf);
3073 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003074#endif
Bram Moolenaar6b1ee342014-08-06 18:17:11 +02003075#ifdef FEAT_CLIPBOARD
3076 end_global_changes();
3077#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003078}
3079
3080/*
3081 * Add files[count] to the arglist of the current window after arg "after".
3082 * The file names in files[count] must have been allocated and are taken over.
3083 * Files[] itself is not taken over.
3084 * Returns index of first added argument. Returns -1 when failed (out of mem).
3085 */
3086 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003087alist_add_list(
3088 int count,
3089 char_u **files,
3090 int after) /* where to add: 0 = before first one */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003091{
3092 int i;
Bram Moolenaara24f0a52016-01-17 19:39:00 +01003093 int old_argcount = ARGCOUNT;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003094
3095 if (ga_grow(&ALIST(curwin)->al_ga, count) == OK)
3096 {
3097 if (after < 0)
3098 after = 0;
3099 if (after > ARGCOUNT)
3100 after = ARGCOUNT;
3101 if (after < ARGCOUNT)
3102 mch_memmove(&(ARGLIST[after + count]), &(ARGLIST[after]),
3103 (ARGCOUNT - after) * sizeof(aentry_T));
3104 for (i = 0; i < count; ++i)
3105 {
3106 ARGLIST[after + i].ae_fname = files[i];
3107 ARGLIST[after + i].ae_fnum = buflist_add(files[i], BLN_LISTED);
3108 }
3109 ALIST(curwin)->al_ga.ga_len += count;
Bram Moolenaara24f0a52016-01-17 19:39:00 +01003110 if (old_argcount > 0 && curwin->w_arg_idx >= after)
3111 curwin->w_arg_idx += count;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003112 return after;
3113 }
3114
3115 for (i = 0; i < count; ++i)
3116 vim_free(files[i]);
3117 return -1;
3118}
3119
3120#endif /* FEAT_LISTCMDS */
3121
3122#ifdef FEAT_EVAL
3123/*
3124 * ":compiler[!] {name}"
3125 */
3126 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003127ex_compiler(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003128{
3129 char_u *buf;
3130 char_u *old_cur_comp = NULL;
3131 char_u *p;
3132
3133 if (*eap->arg == NUL)
3134 {
3135 /* List all compiler scripts. */
3136 do_cmdline_cmd((char_u *)"echo globpath(&rtp, 'compiler/*.vim')");
3137 /* ) keep the indenter happy... */
3138 }
3139 else
3140 {
3141 buf = alloc((unsigned)(STRLEN(eap->arg) + 14));
3142 if (buf != NULL)
3143 {
3144 if (eap->forceit)
3145 {
3146 /* ":compiler! {name}" sets global options */
3147 do_cmdline_cmd((char_u *)
3148 "command -nargs=* CompilerSet set <args>");
3149 }
3150 else
3151 {
3152 /* ":compiler! {name}" sets local options.
3153 * To remain backwards compatible "current_compiler" is always
3154 * used. A user's compiler plugin may set it, the distributed
3155 * plugin will then skip the settings. Afterwards set
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003156 * "b:current_compiler" and restore "current_compiler".
3157 * Explicitly prepend "g:" to make it work in a function. */
3158 old_cur_comp = get_var_value((char_u *)"g:current_compiler");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003159 if (old_cur_comp != NULL)
3160 old_cur_comp = vim_strsave(old_cur_comp);
3161 do_cmdline_cmd((char_u *)
3162 "command -nargs=* CompilerSet setlocal <args>");
3163 }
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003164 do_unlet((char_u *)"g:current_compiler", TRUE);
Bram Moolenaar2ce06f62005-01-31 19:19:04 +00003165 do_unlet((char_u *)"b:current_compiler", TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003166
3167 sprintf((char *)buf, "compiler/%s.vim", eap->arg);
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003168 if (source_runtime(buf, DIP_ALL) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003169 EMSG2(_("E666: compiler not supported: %s"), eap->arg);
3170 vim_free(buf);
3171
3172 do_cmdline_cmd((char_u *)":delcommand CompilerSet");
3173
3174 /* Set "b:current_compiler" from "current_compiler". */
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003175 p = get_var_value((char_u *)"g:current_compiler");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003176 if (p != NULL)
3177 set_internal_string_var((char_u *)"b:current_compiler", p);
3178
3179 /* Restore "current_compiler" for ":compiler {name}". */
3180 if (!eap->forceit)
3181 {
3182 if (old_cur_comp != NULL)
3183 {
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003184 set_internal_string_var((char_u *)"g:current_compiler",
Bram Moolenaar071d4272004-06-13 20:20:40 +00003185 old_cur_comp);
3186 vim_free(old_cur_comp);
3187 }
3188 else
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003189 do_unlet((char_u *)"g:current_compiler", TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003190 }
3191 }
3192 }
3193}
3194#endif
3195
3196/*
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003197 * ":runtime [what] {name}"
Bram Moolenaar071d4272004-06-13 20:20:40 +00003198 */
3199 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003200ex_runtime(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003201{
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003202 char_u *arg = eap->arg;
3203 char_u *p = skiptowhite(arg);
3204 int len = (int)(p - arg);
3205 int flags = eap->forceit ? DIP_ALL : 0;
3206
3207 if (STRNCMP(arg, "START", len) == 0)
3208 {
3209 flags += DIP_START + DIP_NORTP;
3210 arg = skipwhite(arg + len);
3211 }
3212 else if (STRNCMP(arg, "OPT", len) == 0)
3213 {
3214 flags += DIP_OPT + DIP_NORTP;
3215 arg = skipwhite(arg + len);
3216 }
3217 else if (STRNCMP(arg, "PACK", len) == 0)
3218 {
3219 flags += DIP_START + DIP_OPT + DIP_NORTP;
3220 arg = skipwhite(arg + len);
3221 }
3222 else if (STRNCMP(arg, "ALL", len) == 0)
3223 {
3224 flags += DIP_START + DIP_OPT;
3225 arg = skipwhite(arg + len);
3226 }
3227
3228 source_runtime(arg, flags);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003229}
3230
Bram Moolenaar071d4272004-06-13 20:20:40 +00003231 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003232source_callback(char_u *fname, void *cookie UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003233{
Bram Moolenaar910f66f2006-04-05 20:41:53 +00003234 (void)do_source(fname, FALSE, DOSO_NONE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003235}
3236
3237/*
3238 * Source the file "name" from all directories in 'runtimepath'.
3239 * "name" can contain wildcards.
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003240 * When "flags" has DIP_ALL: source all files, otherwise only the first one.
Bram Moolenaar91715872016-03-03 17:13:03 +01003241 *
Bram Moolenaar071d4272004-06-13 20:20:40 +00003242 * return FAIL when no file could be sourced, OK otherwise.
3243 */
3244 int
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003245source_runtime(char_u *name, int flags)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003246{
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003247 return do_in_runtimepath(name, flags, source_callback, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003248}
3249
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003250/*
3251 * Find the file "name" in all directories in "path" and invoke
3252 * "callback(fname, cookie)".
3253 * "name" can contain wildcards.
3254 * When "flags" has DIP_ALL: source all files, otherwise only the first one.
3255 * When "flags" has DIP_DIR: find directories instead of files.
3256 * When "flags" has DIP_ERR: give an error message if there is no match.
3257 *
3258 * return FAIL when no file could be sourced, OK otherwise.
3259 */
Bram Moolenaar6bef5302016-03-12 21:28:26 +01003260 int
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003261do_in_path(
3262 char_u *path,
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003263 char_u *name,
Bram Moolenaar91715872016-03-03 17:13:03 +01003264 int flags,
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003265 void (*callback)(char_u *fname, void *ck),
3266 void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003267{
3268 char_u *rtp;
3269 char_u *np;
3270 char_u *buf;
3271 char_u *rtp_copy;
3272 char_u *tail;
3273 int num_files;
3274 char_u **files;
3275 int i;
3276 int did_one = FALSE;
3277#ifdef AMIGA
3278 struct Process *proc = (struct Process *)FindTask(0L);
3279 APTR save_winptr = proc->pr_WindowPtr;
3280
3281 /* Avoid a requester here for a volume that doesn't exist. */
3282 proc->pr_WindowPtr = (APTR)-1L;
3283#endif
3284
3285 /* Make a copy of 'runtimepath'. Invoking the callback may change the
3286 * value. */
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003287 rtp_copy = vim_strsave(path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003288 buf = alloc(MAXPATHL);
3289 if (buf != NULL && rtp_copy != NULL)
3290 {
Bram Moolenaarc09a6d62013-06-10 21:27:29 +02003291 if (p_verbose > 1 && name != NULL)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003292 {
3293 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00003294 smsg((char_u *)_("Searching for \"%s\" in \"%s\""),
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003295 (char *)name, (char *)path);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003296 verbose_leave();
3297 }
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00003298
Bram Moolenaar071d4272004-06-13 20:20:40 +00003299 /* Loop over all entries in 'runtimepath'. */
3300 rtp = rtp_copy;
Bram Moolenaar91715872016-03-03 17:13:03 +01003301 while (*rtp != NUL && ((flags & DIP_ALL) || !did_one))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003302 {
Bram Moolenaar66459b72016-08-06 19:01:55 +02003303 size_t buflen;
3304
Bram Moolenaar071d4272004-06-13 20:20:40 +00003305 /* Copy the path from 'runtimepath' to buf[]. */
3306 copy_option_part(&rtp, buf, MAXPATHL, ",");
Bram Moolenaar66459b72016-08-06 19:01:55 +02003307 buflen = STRLEN(buf);
3308
3309 /* Skip after or non-after directories. */
3310 if (flags & (DIP_NOAFTER | DIP_AFTER))
3311 {
3312 int is_after = buflen >= 5
3313 && STRCMP(buf + buflen - 5, "after") == 0;
3314
3315 if ((is_after && (flags & DIP_NOAFTER))
3316 || (!is_after && (flags & DIP_AFTER)))
3317 continue;
3318 }
3319
Bram Moolenaarc09a6d62013-06-10 21:27:29 +02003320 if (name == NULL)
3321 {
3322 (*callback)(buf, (void *) &cookie);
3323 if (!did_one)
3324 did_one = (cookie == NULL);
3325 }
Bram Moolenaar66459b72016-08-06 19:01:55 +02003326 else if (buflen + STRLEN(name) + 2 < MAXPATHL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003327 {
3328 add_pathsep(buf);
3329 tail = buf + STRLEN(buf);
3330
3331 /* Loop over all patterns in "name" */
3332 np = name;
Bram Moolenaar91715872016-03-03 17:13:03 +01003333 while (*np != NUL && ((flags & DIP_ALL) || !did_one))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003334 {
3335 /* Append the pattern from "name" to buf[]. */
3336 copy_option_part(&np, tail, (int)(MAXPATHL - (tail - buf)),
3337 "\t ");
3338
3339 if (p_verbose > 2)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003340 {
3341 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00003342 smsg((char_u *)_("Searching for \"%s\""), buf);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003343 verbose_leave();
3344 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003345
3346 /* Expand wildcards, invoke the callback for each match. */
3347 if (gen_expand_wildcards(1, &buf, &num_files, &files,
Bram Moolenaar91715872016-03-03 17:13:03 +01003348 (flags & DIP_DIR) ? EW_DIR : EW_FILE) == OK)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003349 {
3350 for (i = 0; i < num_files; ++i)
3351 {
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00003352 (*callback)(files[i], cookie);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003353 did_one = TRUE;
Bram Moolenaar91715872016-03-03 17:13:03 +01003354 if (!(flags & DIP_ALL))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003355 break;
3356 }
3357 FreeWild(num_files, files);
3358 }
3359 }
3360 }
3361 }
3362 }
3363 vim_free(buf);
3364 vim_free(rtp_copy);
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003365 if (!did_one && name != NULL)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003366 {
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003367 char *basepath = path == p_rtp ? "runtimepath" : "packpath";
3368
3369 if (flags & DIP_ERR)
3370 EMSG3(_(e_dirnotf), basepath, name);
3371 else if (p_verbose > 0)
3372 {
3373 verbose_enter();
3374 smsg((char_u *)_("not found in '%s': \"%s\""), basepath, name);
3375 verbose_leave();
3376 }
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003377 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003378
3379#ifdef AMIGA
3380 proc->pr_WindowPtr = save_winptr;
3381#endif
3382
3383 return did_one ? OK : FAIL;
3384}
3385
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003386/*
3387 * Find "name" in 'runtimepath'. When found, invoke the callback function for
3388 * it: callback(fname, "cookie")
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003389 * When "flags" has DIP_ALL repeat for all matches, otherwise only the first
3390 * one is used.
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003391 * Returns OK when at least one match found, FAIL otherwise.
3392 *
3393 * If "name" is NULL calls callback for each entry in runtimepath. Cookie is
3394 * passed by reference in this case, setting it to NULL indicates that callback
3395 * has done its job.
3396 */
3397 int
3398do_in_runtimepath(
3399 char_u *name,
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003400 int flags,
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003401 void (*callback)(char_u *fname, void *ck),
3402 void *cookie)
3403{
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003404 int done = FAIL;
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003405 char_u *s;
3406 int len;
3407 char *start_dir = "pack/*/start/*/%s";
3408 char *opt_dir = "pack/*/opt/*/%s";
3409
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003410 if ((flags & DIP_NORTP) == 0)
3411 done = do_in_path(p_rtp, name, flags, callback, cookie);
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003412
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003413 if ((done == FAIL || (flags & DIP_ALL)) && (flags & DIP_START))
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003414 {
Bram Moolenaar1c8b4ed2016-03-17 21:51:03 +01003415 len = (int)(STRLEN(start_dir) + STRLEN(name));
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003416 s = alloc(len);
3417 if (s == NULL)
3418 return FAIL;
3419 vim_snprintf((char *)s, len, start_dir, name);
3420 done = do_in_path(p_pp, s, flags, callback, cookie);
3421 vim_free(s);
3422 }
3423
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003424 if ((done == FAIL || (flags & DIP_ALL)) && (flags & DIP_OPT))
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003425 {
Bram Moolenaar1c8b4ed2016-03-17 21:51:03 +01003426 len = (int)(STRLEN(opt_dir) + STRLEN(name));
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003427 s = alloc(len);
3428 if (s == NULL)
3429 return FAIL;
3430 vim_snprintf((char *)s, len, opt_dir, name);
3431 done = do_in_path(p_pp, s, flags, callback, cookie);
3432 vim_free(s);
3433 }
3434
3435 return done;
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003436}
3437
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003438/*
Bram Moolenaarf3654822016-03-04 22:12:23 +01003439 * Expand wildcards in "pat" and invoke do_source() for each match.
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003440 */
3441 static void
Bram Moolenaarf3654822016-03-04 22:12:23 +01003442source_all_matches(char_u *pat)
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003443{
Bram Moolenaarf3654822016-03-04 22:12:23 +01003444 int num_files;
3445 char_u **files;
3446 int i;
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003447
Bram Moolenaarf3654822016-03-04 22:12:23 +01003448 if (gen_expand_wildcards(1, &pat, &num_files, &files, EW_FILE) == OK)
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003449 {
Bram Moolenaarf3654822016-03-04 22:12:23 +01003450 for (i = 0; i < num_files; ++i)
3451 (void)do_source(files[i], FALSE, DOSO_NONE);
3452 FreeWild(num_files, files);
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003453 }
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003454}
3455
Bram Moolenaar49b27322016-04-05 21:13:00 +02003456/* used for "cookie" of add_pack_plugin() */
3457static int APP_ADD_DIR;
3458static int APP_LOAD;
3459static int APP_BOTH;
3460
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003461 static void
Bram Moolenaar91715872016-03-03 17:13:03 +01003462add_pack_plugin(char_u *fname, void *cookie)
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003463{
Bram Moolenaarf3654822016-03-04 22:12:23 +01003464 char_u *p4, *p3, *p2, *p1, *p;
3465 char_u *insp;
Bram Moolenaar91715872016-03-03 17:13:03 +01003466 int c;
3467 char_u *new_rtp;
3468 int keep;
Bram Moolenaarb0550662016-05-31 21:37:36 +02003469 size_t oldlen;
3470 size_t addlen;
Bram Moolenaara5702442016-05-24 19:37:29 +02003471 char_u *afterdir;
Bram Moolenaarb0550662016-05-31 21:37:36 +02003472 size_t afterlen = 0;
Bram Moolenaar91715872016-03-03 17:13:03 +01003473 char_u *ffname = fix_fname(fname);
Bram Moolenaarfef524b2016-07-02 22:07:22 +02003474 size_t fname_len;
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003475
Bram Moolenaar91715872016-03-03 17:13:03 +01003476 if (ffname == NULL)
3477 return;
Bram Moolenaar49b27322016-04-05 21:13:00 +02003478 if (cookie != &APP_LOAD && strstr((char *)p_rtp, (char *)ffname) == NULL)
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003479 {
Bram Moolenaara5702442016-05-24 19:37:29 +02003480 /* directory is not yet in 'runtimepath', add it */
Bram Moolenaarf3654822016-03-04 22:12:23 +01003481 p4 = p3 = p2 = p1 = get_past_head(ffname);
3482 for (p = p1; *p; mb_ptr_adv(p))
3483 if (vim_ispathsep_nocolon(*p))
3484 {
3485 p4 = p3; p3 = p2; p2 = p1; p1 = p;
3486 }
3487
3488 /* now we have:
Bram Moolenaaraf1a0e32016-03-09 22:19:26 +01003489 * rtp/pack/name/start/name
3490 * p4 p3 p2 p1
Bram Moolenaarf3654822016-03-04 22:12:23 +01003491 *
3492 * find the part up to "pack" in 'runtimepath' */
3493 c = *p4;
3494 *p4 = NUL;
Bram Moolenaar4c5717e2016-07-01 15:39:40 +02003495
3496 /* Find "ffname" in "p_rtp", ignoring '/' vs '\' differences. */
3497 fname_len = STRLEN(ffname);
3498 insp = p_rtp;
3499 for (;;)
3500 {
3501 if (vim_fnamencmp(insp, ffname, fname_len) == 0)
3502 break;
3503 insp = vim_strchr(insp, ',');
3504 if (insp == NULL)
3505 break;
3506 ++insp;
3507 }
3508
Bram Moolenaarf3654822016-03-04 22:12:23 +01003509 if (insp == NULL)
3510 /* not found, append at the end */
3511 insp = p_rtp + STRLEN(p_rtp);
3512 else
3513 {
3514 /* append after the matching directory. */
3515 insp += STRLEN(ffname);
3516 while (*insp != NUL && *insp != ',')
3517 ++insp;
3518 }
3519 *p4 = c;
3520
Bram Moolenaara5702442016-05-24 19:37:29 +02003521 /* check if rtp/pack/name/start/name/after exists */
3522 afterdir = concat_fnames(ffname, (char_u *)"after", TRUE);
3523 if (afterdir != NULL && mch_isdir(afterdir))
3524 afterlen = STRLEN(afterdir) + 1; /* add one for comma */
3525
Bram Moolenaarb0550662016-05-31 21:37:36 +02003526 oldlen = STRLEN(p_rtp);
3527 addlen = STRLEN(ffname) + 1; /* add one for comma */
3528 new_rtp = alloc((int)(oldlen + addlen + afterlen + 1)); /* add one for NUL */
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003529 if (new_rtp == NULL)
Bram Moolenaarf3654822016-03-04 22:12:23 +01003530 goto theend;
3531 keep = (int)(insp - p_rtp);
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003532 mch_memmove(new_rtp, p_rtp, keep);
3533 new_rtp[keep] = ',';
Bram Moolenaara5702442016-05-24 19:37:29 +02003534 mch_memmove(new_rtp + keep + 1, ffname, addlen);
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003535 if (p_rtp[keep] != NUL)
Bram Moolenaara5702442016-05-24 19:37:29 +02003536 mch_memmove(new_rtp + keep + addlen, p_rtp + keep,
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003537 oldlen - keep + 1);
Bram Moolenaara5702442016-05-24 19:37:29 +02003538 if (afterlen > 0)
3539 {
3540 STRCAT(new_rtp, ",");
3541 STRCAT(new_rtp, afterdir);
3542 }
Bram Moolenaar863c1a92016-03-03 15:47:06 +01003543 set_option_value((char_u *)"rtp", 0L, new_rtp, 0);
3544 vim_free(new_rtp);
Bram Moolenaara5702442016-05-24 19:37:29 +02003545 vim_free(afterdir);
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003546 }
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003547
Bram Moolenaar49b27322016-04-05 21:13:00 +02003548 if (cookie != &APP_ADD_DIR)
Bram Moolenaarf3654822016-03-04 22:12:23 +01003549 {
Bram Moolenaar71fb0c12016-04-02 22:44:16 +02003550 static char *plugpat = "%s/plugin/**/*.vim";
Bram Moolenaarf3654822016-03-04 22:12:23 +01003551 static char *ftpat = "%s/ftdetect/*.vim";
3552 int len;
3553 char_u *pat;
3554
3555 len = (int)STRLEN(ffname) + (int)STRLEN(ftpat);
3556 pat = alloc(len);
3557 if (pat == NULL)
3558 goto theend;
3559 vim_snprintf((char *)pat, len, plugpat, ffname);
3560 source_all_matches(pat);
3561
3562#ifdef FEAT_AUTOCMD
3563 {
3564 char_u *cmd = vim_strsave((char_u *)"g:did_load_filetypes");
3565
3566 /* If runtime/filetype.vim wasn't loaded yet, the scripts will be
3567 * found when it loads. */
3568 if (cmd != NULL && eval_to_number(cmd) > 0)
3569 {
3570 do_cmdline_cmd((char_u *)"augroup filetypedetect");
3571 vim_snprintf((char *)pat, len, ftpat, ffname);
3572 source_all_matches(pat);
3573 do_cmdline_cmd((char_u *)"augroup END");
3574 }
3575 vim_free(cmd);
3576 }
3577#endif
Bram Moolenaarba8cd122016-03-19 14:16:39 +01003578 vim_free(pat);
Bram Moolenaarf3654822016-03-04 22:12:23 +01003579 }
3580
3581theend:
3582 vim_free(ffname);
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003583}
3584
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003585static int did_source_packages = FALSE;
3586
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003587/*
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003588 * ":packloadall"
Bram Moolenaarf3654822016-03-04 22:12:23 +01003589 * Find plugins in the package directories and source them.
Bram Moolenaar66459b72016-08-06 19:01:55 +02003590 * "eap" is NULL when invoked during startup.
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003591 */
3592 void
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003593ex_packloadall(exarg_T *eap)
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003594{
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003595 if (!did_source_packages || (eap != NULL && eap->forceit))
3596 {
3597 did_source_packages = TRUE;
Bram Moolenaar49b27322016-04-05 21:13:00 +02003598
3599 /* First do a round to add all directories to 'runtimepath', then load
3600 * the plugins. This allows for plugins to use an autoload directory
3601 * of another plugin. */
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003602 do_in_path(p_pp, (char_u *)"pack/*/start/*", DIP_ALL + DIP_DIR,
Bram Moolenaar49b27322016-04-05 21:13:00 +02003603 add_pack_plugin, &APP_ADD_DIR);
3604 do_in_path(p_pp, (char_u *)"pack/*/start/*", DIP_ALL + DIP_DIR,
3605 add_pack_plugin, &APP_LOAD);
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003606 }
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003607}
3608
3609/*
Bram Moolenaarf3654822016-03-04 22:12:23 +01003610 * ":packadd[!] {name}"
Bram Moolenaar91715872016-03-03 17:13:03 +01003611 */
3612 void
3613ex_packadd(exarg_T *eap)
3614{
3615 static char *plugpat = "pack/*/opt/%s";
3616 int len;
3617 char *pat;
3618
3619 len = (int)STRLEN(plugpat) + (int)STRLEN(eap->arg);
3620 pat = (char *)alloc(len);
3621 if (pat == NULL)
3622 return;
3623 vim_snprintf(pat, len, plugpat, eap->arg);
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003624 do_in_path(p_pp, (char_u *)pat, DIP_ALL + DIP_DIR + DIP_ERR,
Bram Moolenaar49b27322016-04-05 21:13:00 +02003625 add_pack_plugin, eap->forceit ? &APP_ADD_DIR : &APP_BOTH);
Bram Moolenaar91715872016-03-03 17:13:03 +01003626 vim_free(pat);
3627}
3628
Bram Moolenaar071d4272004-06-13 20:20:40 +00003629#if defined(FEAT_EVAL) && defined(FEAT_AUTOCMD)
3630/*
3631 * ":options"
3632 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003633 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003634ex_options(
3635 exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003636{
3637 cmd_source((char_u *)SYS_OPTWIN_FILE, NULL);
3638}
3639#endif
3640
3641/*
3642 * ":source {fname}"
3643 */
3644 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003645ex_source(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003646{
3647#ifdef FEAT_BROWSE
3648 if (cmdmod.browse)
3649 {
3650 char_u *fname = NULL;
3651
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00003652 fname = do_browse(0, (char_u *)_("Source Vim script"), eap->arg,
Bram Moolenaar071d4272004-06-13 20:20:40 +00003653 NULL, NULL, BROWSE_FILTER_MACROS, NULL);
3654 if (fname != NULL)
3655 {
3656 cmd_source(fname, eap);
3657 vim_free(fname);
3658 }
3659 }
3660 else
3661#endif
3662 cmd_source(eap->arg, eap);
3663}
3664
3665 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003666cmd_source(char_u *fname, exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003667{
3668 if (*fname == NUL)
3669 EMSG(_(e_argreq));
3670
Bram Moolenaar071d4272004-06-13 20:20:40 +00003671 else if (eap != NULL && eap->forceit)
Bram Moolenaar84a05ac2013-05-06 04:24:17 +02003672 /* ":source!": read Normal mode commands
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00003673 * Need to execute the commands directly. This is required at least
3674 * for:
Bram Moolenaar071d4272004-06-13 20:20:40 +00003675 * - ":g" command busy
3676 * - after ":argdo", ":windo" or ":bufdo"
3677 * - another command follows
3678 * - inside a loop
3679 */
3680 openscript(fname, global_busy || listcmd_busy || eap->nextcmd != NULL
3681#ifdef FEAT_EVAL
3682 || eap->cstack->cs_idx >= 0
3683#endif
3684 );
3685
3686 /* ":source" read ex commands */
Bram Moolenaar910f66f2006-04-05 20:41:53 +00003687 else if (do_source(fname, FALSE, DOSO_NONE) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003688 EMSG2(_(e_notopen), fname);
3689}
3690
3691/*
3692 * ":source" and associated commands.
3693 */
3694/*
3695 * Structure used to store info for each sourced file.
3696 * It is shared between do_source() and getsourceline().
3697 * This is required, because it needs to be handed to do_cmdline() and
3698 * sourcing can be done recursively.
3699 */
3700struct source_cookie
3701{
3702 FILE *fp; /* opened file for sourcing */
3703 char_u *nextline; /* if not NULL: line that was read ahead */
3704 int finished; /* ":finish" used */
Bram Moolenaar860cae12010-06-05 23:22:07 +02003705#if defined(USE_CRNL) || defined(USE_CR)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003706 int fileformat; /* EOL_UNKNOWN, EOL_UNIX or EOL_DOS */
3707 int error; /* TRUE if LF found after CR-LF */
3708#endif
3709#ifdef FEAT_EVAL
3710 linenr_T breakpoint; /* next line with breakpoint or zero */
3711 char_u *fname; /* name of sourced file */
3712 int dbg_tick; /* debug_tick when breakpoint was set */
3713 int level; /* top nesting level of sourced file */
3714#endif
3715#ifdef FEAT_MBYTE
3716 vimconv_T conv; /* type of conversion */
3717#endif
3718};
3719
3720#ifdef FEAT_EVAL
3721/*
3722 * Return the address holding the next breakpoint line for a source cookie.
3723 */
3724 linenr_T *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003725source_breakpoint(void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003726{
3727 return &((struct source_cookie *)cookie)->breakpoint;
3728}
3729
3730/*
3731 * Return the address holding the debug tick for a source cookie.
3732 */
3733 int *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003734source_dbg_tick(void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003735{
3736 return &((struct source_cookie *)cookie)->dbg_tick;
3737}
3738
3739/*
3740 * Return the nesting level for a source cookie.
3741 */
3742 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003743source_level(void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003744{
3745 return ((struct source_cookie *)cookie)->level;
3746}
3747#endif
3748
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01003749static char_u *get_one_sourceline(struct source_cookie *sp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003750
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01003751#if (defined(WIN32) && defined(FEAT_CSCOPE)) || defined(HAVE_FD_CLOEXEC)
3752# define USE_FOPEN_NOINH
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01003753static FILE *fopen_noinh_readbin(char *filename);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003754
3755/*
3756 * Special function to open a file without handle inheritance.
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01003757 * When possible the handle is closed on exec().
Bram Moolenaar071d4272004-06-13 20:20:40 +00003758 */
3759 static FILE *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003760fopen_noinh_readbin(char *filename)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003761{
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01003762# ifdef WIN32
Bram Moolenaar8d8ef0b2010-01-20 21:41:47 +01003763 int fd_tmp = mch_open(filename, O_RDONLY | O_BINARY | O_NOINHERIT, 0);
3764# else
3765 int fd_tmp = mch_open(filename, O_RDONLY, 0);
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01003766# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003767
3768 if (fd_tmp == -1)
3769 return NULL;
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01003770
3771# ifdef HAVE_FD_CLOEXEC
3772 {
3773 int fdflags = fcntl(fd_tmp, F_GETFD);
3774 if (fdflags >= 0 && (fdflags & FD_CLOEXEC) == 0)
Bram Moolenaarcde88542015-08-11 19:14:00 +02003775 (void)fcntl(fd_tmp, F_SETFD, fdflags | FD_CLOEXEC);
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01003776 }
3777# endif
3778
Bram Moolenaar071d4272004-06-13 20:20:40 +00003779 return fdopen(fd_tmp, READBIN);
3780}
3781#endif
3782
3783
3784/*
3785 * do_source: Read the file "fname" and execute its lines as EX commands.
3786 *
3787 * This function may be called recursively!
3788 *
3789 * return FAIL if file could not be opened, OK otherwise
3790 */
3791 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003792do_source(
3793 char_u *fname,
3794 int check_other, /* check for .vimrc and _vimrc */
3795 int is_vimrc) /* DOSO_ value */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003796{
3797 struct source_cookie cookie;
3798 char_u *save_sourcing_name;
3799 linenr_T save_sourcing_lnum;
3800 char_u *p;
3801 char_u *fname_exp;
Bram Moolenaar73881402009-02-04 16:50:47 +00003802 char_u *firstline = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003803 int retval = FAIL;
3804#ifdef FEAT_EVAL
3805 scid_T save_current_SID;
3806 static scid_T last_current_SID = 0;
3807 void *save_funccalp;
3808 int save_debug_break_level = debug_break_level;
Bram Moolenaar05159a02005-02-26 23:04:13 +00003809 scriptitem_T *si = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003810# ifdef UNIX
Bram Moolenaar8767f522016-07-01 17:17:39 +02003811 stat_T st;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003812 int stat_ok;
3813# endif
3814#endif
3815#ifdef STARTUPTIME
3816 struct timeval tv_rel;
3817 struct timeval tv_start;
3818#endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00003819#ifdef FEAT_PROFILE
3820 proftime_T wait_start;
3821#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003822
Bram Moolenaar071d4272004-06-13 20:20:40 +00003823 p = expand_env_save(fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003824 if (p == NULL)
3825 return retval;
3826 fname_exp = fix_fname(p);
3827 vim_free(p);
3828 if (fname_exp == NULL)
3829 return retval;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003830 if (mch_isdir(fname_exp))
3831 {
Bram Moolenaar555b2802005-05-19 21:08:39 +00003832 smsg((char_u *)_("Cannot source a directory: \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003833 goto theend;
3834 }
3835
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00003836#ifdef FEAT_AUTOCMD
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00003837 /* Apply SourceCmd autocommands, they should get the file and source it. */
3838 if (has_autocmd(EVENT_SOURCECMD, fname_exp, NULL)
3839 && apply_autocmds(EVENT_SOURCECMD, fname_exp, fname_exp,
3840 FALSE, curbuf))
Bram Moolenaarf33943e2008-01-15 21:17:30 +00003841 {
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00003842# ifdef FEAT_EVAL
Bram Moolenaarf33943e2008-01-15 21:17:30 +00003843 retval = aborting() ? FAIL : OK;
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00003844# else
Bram Moolenaarf33943e2008-01-15 21:17:30 +00003845 retval = OK;
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00003846# endif
Bram Moolenaarf33943e2008-01-15 21:17:30 +00003847 goto theend;
3848 }
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00003849
3850 /* Apply SourcePre autocommands, they may get the file. */
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00003851 apply_autocmds(EVENT_SOURCEPRE, fname_exp, fname_exp, FALSE, curbuf);
3852#endif
3853
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01003854#ifdef USE_FOPEN_NOINH
Bram Moolenaar071d4272004-06-13 20:20:40 +00003855 cookie.fp = fopen_noinh_readbin((char *)fname_exp);
3856#else
3857 cookie.fp = mch_fopen((char *)fname_exp, READBIN);
3858#endif
3859 if (cookie.fp == NULL && check_other)
3860 {
3861 /*
3862 * Try again, replacing file name ".vimrc" by "_vimrc" or vice versa,
3863 * and ".exrc" by "_exrc" or vice versa.
3864 */
3865 p = gettail(fname_exp);
3866 if ((*p == '.' || *p == '_')
3867 && (STRICMP(p + 1, "vimrc") == 0
3868 || STRICMP(p + 1, "gvimrc") == 0
3869 || STRICMP(p + 1, "exrc") == 0))
3870 {
3871 if (*p == '_')
3872 *p = '.';
3873 else
3874 *p = '_';
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01003875#ifdef USE_FOPEN_NOINH
Bram Moolenaar071d4272004-06-13 20:20:40 +00003876 cookie.fp = fopen_noinh_readbin((char *)fname_exp);
3877#else
3878 cookie.fp = mch_fopen((char *)fname_exp, READBIN);
3879#endif
3880 }
3881 }
3882
3883 if (cookie.fp == NULL)
3884 {
3885 if (p_verbose > 0)
3886 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003887 verbose_enter();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003888 if (sourcing_name == NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00003889 smsg((char_u *)_("could not source \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003890 else
3891 smsg((char_u *)_("line %ld: could not source \"%s\""),
Bram Moolenaar555b2802005-05-19 21:08:39 +00003892 sourcing_lnum, fname);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003893 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003894 }
3895 goto theend;
3896 }
3897
3898 /*
3899 * The file exists.
3900 * - In verbose mode, give a message.
3901 * - For a vimrc file, may want to set 'compatible', call vimrc_found().
3902 */
3903 if (p_verbose > 1)
3904 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003905 verbose_enter();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003906 if (sourcing_name == NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00003907 smsg((char_u *)_("sourcing \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003908 else
3909 smsg((char_u *)_("line %ld: sourcing \"%s\""),
Bram Moolenaar555b2802005-05-19 21:08:39 +00003910 sourcing_lnum, fname);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003911 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003912 }
Bram Moolenaar910f66f2006-04-05 20:41:53 +00003913 if (is_vimrc == DOSO_VIMRC)
3914 vimrc_found(fname_exp, (char_u *)"MYVIMRC");
3915 else if (is_vimrc == DOSO_GVIMRC)
3916 vimrc_found(fname_exp, (char_u *)"MYGVIMRC");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003917
3918#ifdef USE_CRNL
3919 /* If no automatic file format: Set default to CR-NL. */
3920 if (*p_ffs == NUL)
3921 cookie.fileformat = EOL_DOS;
3922 else
3923 cookie.fileformat = EOL_UNKNOWN;
3924 cookie.error = FALSE;
3925#endif
3926
3927#ifdef USE_CR
3928 /* If no automatic file format: Set default to CR. */
3929 if (*p_ffs == NUL)
3930 cookie.fileformat = EOL_MAC;
3931 else
3932 cookie.fileformat = EOL_UNKNOWN;
3933 cookie.error = FALSE;
3934#endif
3935
3936 cookie.nextline = NULL;
3937 cookie.finished = FALSE;
3938
3939#ifdef FEAT_EVAL
3940 /*
3941 * Check if this script has a breakpoint.
3942 */
3943 cookie.breakpoint = dbg_find_breakpoint(TRUE, fname_exp, (linenr_T)0);
3944 cookie.fname = fname_exp;
3945 cookie.dbg_tick = debug_tick;
3946
3947 cookie.level = ex_nesting_level;
3948#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003949
3950 /*
3951 * Keep the sourcing name/lnum, for recursive calls.
3952 */
3953 save_sourcing_name = sourcing_name;
3954 sourcing_name = fname_exp;
3955 save_sourcing_lnum = sourcing_lnum;
3956 sourcing_lnum = 0;
3957
Bram Moolenaar73881402009-02-04 16:50:47 +00003958#ifdef FEAT_MBYTE
3959 cookie.conv.vc_type = CONV_NONE; /* no conversion */
3960
3961 /* Read the first line so we can check for a UTF-8 BOM. */
3962 firstline = getsourceline(0, (void *)&cookie, 0);
3963 if (firstline != NULL && STRLEN(firstline) >= 3 && firstline[0] == 0xef
3964 && firstline[1] == 0xbb && firstline[2] == 0xbf)
3965 {
3966 /* Found BOM; setup conversion, skip over BOM and recode the line. */
3967 convert_setup(&cookie.conv, (char_u *)"utf-8", p_enc);
3968 p = string_convert(&cookie.conv, firstline + 3, NULL);
Bram Moolenaar4e2a5952009-02-05 19:48:25 +00003969 if (p == NULL)
3970 p = vim_strsave(firstline + 3);
Bram Moolenaar73881402009-02-04 16:50:47 +00003971 if (p != NULL)
3972 {
3973 vim_free(firstline);
3974 firstline = p;
3975 }
3976 }
3977#endif
3978
Bram Moolenaar071d4272004-06-13 20:20:40 +00003979#ifdef STARTUPTIME
Bram Moolenaarc4e41982010-01-19 16:31:47 +01003980 if (time_fd != NULL)
3981 time_push(&tv_rel, &tv_start);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003982#endif
3983
3984#ifdef FEAT_EVAL
Bram Moolenaar05159a02005-02-26 23:04:13 +00003985# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00003986 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003987 prof_child_enter(&wait_start); /* entering a child now */
3988# endif
3989
3990 /* Don't use local function variables, if called from a function.
3991 * Also starts profiling timer for nested script. */
3992 save_funccalp = save_funccal();
3993
Bram Moolenaar071d4272004-06-13 20:20:40 +00003994 /*
3995 * Check if this script was sourced before to finds its SID.
3996 * If it's new, generate a new SID.
3997 */
3998 save_current_SID = current_SID;
3999# ifdef UNIX
4000 stat_ok = (mch_stat((char *)fname_exp, &st) >= 0);
4001# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00004002 for (current_SID = script_items.ga_len; current_SID > 0; --current_SID)
4003 {
4004 si = &SCRIPT_ITEM(current_SID);
4005 if (si->sn_name != NULL
Bram Moolenaar071d4272004-06-13 20:20:40 +00004006 && (
4007# ifdef UNIX
Bram Moolenaar7c626922005-02-07 22:01:03 +00004008 /* Compare dev/ino when possible, it catches symbolic
4009 * links. Also compare file names, the inode may change
4010 * when the file was edited. */
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00004011 ((stat_ok && si->sn_dev_valid)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004012 && (si->sn_dev == st.st_dev
4013 && si->sn_ino == st.st_ino)) ||
Bram Moolenaar071d4272004-06-13 20:20:40 +00004014# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00004015 fnamecmp(si->sn_name, fname_exp) == 0))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004016 break;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004017 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004018 if (current_SID == 0)
4019 {
4020 current_SID = ++last_current_SID;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004021 if (ga_grow(&script_items, (int)(current_SID - script_items.ga_len))
4022 == FAIL)
4023 goto almosttheend;
4024 while (script_items.ga_len < current_SID)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004025 {
Bram Moolenaar05159a02005-02-26 23:04:13 +00004026 ++script_items.ga_len;
4027 SCRIPT_ITEM(script_items.ga_len).sn_name = NULL;
4028# ifdef FEAT_PROFILE
4029 SCRIPT_ITEM(script_items.ga_len).sn_prof_on = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004030# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004031 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00004032 si = &SCRIPT_ITEM(current_SID);
4033 si->sn_name = fname_exp;
4034 fname_exp = NULL;
4035# ifdef UNIX
4036 if (stat_ok)
4037 {
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00004038 si->sn_dev_valid = TRUE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004039 si->sn_dev = st.st_dev;
4040 si->sn_ino = st.st_ino;
4041 }
4042 else
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00004043 si->sn_dev_valid = FALSE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004044# endif
4045
Bram Moolenaar071d4272004-06-13 20:20:40 +00004046 /* Allocate the local script variables to use for this script. */
4047 new_script_vars(current_SID);
4048 }
4049
Bram Moolenaar05159a02005-02-26 23:04:13 +00004050# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004051 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004052 {
4053 int forceit;
4054
4055 /* Check if we do profiling for this script. */
4056 if (!si->sn_prof_on && has_profiling(TRUE, si->sn_name, &forceit))
4057 {
4058 script_do_profile(si);
4059 si->sn_pr_force = forceit;
4060 }
4061 if (si->sn_prof_on)
4062 {
4063 ++si->sn_pr_count;
4064 profile_start(&si->sn_pr_start);
4065 profile_zero(&si->sn_pr_children);
4066 }
4067 }
4068# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004069#endif
4070
4071 /*
4072 * Call do_cmdline, which will call getsourceline() to get the lines.
4073 */
Bram Moolenaar73881402009-02-04 16:50:47 +00004074 do_cmdline(firstline, getsourceline, (void *)&cookie,
Bram Moolenaar071d4272004-06-13 20:20:40 +00004075 DOCMD_VERBOSE|DOCMD_NOWAIT|DOCMD_REPEAT);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004076 retval = OK;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004077
4078#ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004079 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004080 {
4081 /* Get "si" again, "script_items" may have been reallocated. */
4082 si = &SCRIPT_ITEM(current_SID);
4083 if (si->sn_prof_on)
4084 {
4085 profile_end(&si->sn_pr_start);
4086 profile_sub_wait(&wait_start, &si->sn_pr_start);
4087 profile_add(&si->sn_pr_total, &si->sn_pr_start);
Bram Moolenaar1056d982006-03-09 22:37:52 +00004088 profile_self(&si->sn_pr_self, &si->sn_pr_start,
4089 &si->sn_pr_children);
Bram Moolenaar05159a02005-02-26 23:04:13 +00004090 }
4091 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004092#endif
4093
4094 if (got_int)
4095 EMSG(_(e_interr));
4096 sourcing_name = save_sourcing_name;
4097 sourcing_lnum = save_sourcing_lnum;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004098 if (p_verbose > 1)
4099 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004100 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00004101 smsg((char_u *)_("finished sourcing %s"), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004102 if (sourcing_name != NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00004103 smsg((char_u *)_("continuing in %s"), sourcing_name);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004104 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004105 }
4106#ifdef STARTUPTIME
Bram Moolenaarc4e41982010-01-19 16:31:47 +01004107 if (time_fd != NULL)
4108 {
4109 vim_snprintf((char *)IObuff, IOSIZE, "sourcing %s", fname);
4110 time_msg((char *)IObuff, &tv_start);
4111 time_pop(&tv_rel);
4112 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004113#endif
4114
4115#ifdef FEAT_EVAL
4116 /*
4117 * After a "finish" in debug mode, need to break at first command of next
4118 * sourced file.
4119 */
4120 if (save_debug_break_level > ex_nesting_level
4121 && debug_break_level == ex_nesting_level)
4122 ++debug_break_level;
4123#endif
4124
Bram Moolenaar05159a02005-02-26 23:04:13 +00004125#ifdef FEAT_EVAL
4126almosttheend:
4127 current_SID = save_current_SID;
4128 restore_funccal(save_funccalp);
4129# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004130 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004131 prof_child_exit(&wait_start); /* leaving a child now */
4132# endif
4133#endif
4134 fclose(cookie.fp);
4135 vim_free(cookie.nextline);
Bram Moolenaar73881402009-02-04 16:50:47 +00004136 vim_free(firstline);
Bram Moolenaar05159a02005-02-26 23:04:13 +00004137#ifdef FEAT_MBYTE
4138 convert_setup(&cookie.conv, NULL, NULL);
4139#endif
4140
Bram Moolenaar071d4272004-06-13 20:20:40 +00004141theend:
4142 vim_free(fname_exp);
4143 return retval;
4144}
4145
4146#if defined(FEAT_EVAL) || defined(PROTO)
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00004147
Bram Moolenaar071d4272004-06-13 20:20:40 +00004148/*
4149 * ":scriptnames"
4150 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004151 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004152ex_scriptnames(exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004153{
4154 int i;
4155
Bram Moolenaar05159a02005-02-26 23:04:13 +00004156 for (i = 1; i <= script_items.ga_len && !got_int; ++i)
4157 if (SCRIPT_ITEM(i).sn_name != NULL)
Bram Moolenaard58ea072011-06-26 04:25:30 +02004158 {
4159 home_replace(NULL, SCRIPT_ITEM(i).sn_name,
4160 NameBuff, MAXPATHL, TRUE);
4161 smsg((char_u *)"%3d: %s", i, NameBuff);
Bram Moolenaar970a1b82012-03-23 18:39:18 +01004162 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004163}
4164
4165# if defined(BACKSLASH_IN_FILENAME) || defined(PROTO)
4166/*
4167 * Fix slashes in the list of script names for 'shellslash'.
4168 */
4169 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004170scriptnames_slash_adjust(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004171{
4172 int i;
4173
Bram Moolenaar05159a02005-02-26 23:04:13 +00004174 for (i = 1; i <= script_items.ga_len; ++i)
4175 if (SCRIPT_ITEM(i).sn_name != NULL)
4176 slash_adjust(SCRIPT_ITEM(i).sn_name);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004177}
4178# endif
4179
4180/*
4181 * Get a pointer to a script name. Used for ":verbose set".
4182 */
4183 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004184get_scriptname(scid_T id)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004185{
4186 if (id == SID_MODELINE)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00004187 return (char_u *)_("modeline");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004188 if (id == SID_CMDARG)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00004189 return (char_u *)_("--cmd argument");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004190 if (id == SID_CARG)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00004191 return (char_u *)_("-c argument");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004192 if (id == SID_ENV)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00004193 return (char_u *)_("environment variable");
4194 if (id == SID_ERROR)
4195 return (char_u *)_("error handler");
Bram Moolenaar05159a02005-02-26 23:04:13 +00004196 return SCRIPT_ITEM(id).sn_name;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004197}
Bram Moolenaar05159a02005-02-26 23:04:13 +00004198
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00004199# if defined(EXITFREE) || defined(PROTO)
4200 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004201free_scriptnames(void)
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00004202{
4203 int i;
4204
4205 for (i = script_items.ga_len; i > 0; --i)
4206 vim_free(SCRIPT_ITEM(i).sn_name);
4207 ga_clear(&script_items);
4208}
4209# endif
4210
Bram Moolenaar071d4272004-06-13 20:20:40 +00004211#endif
4212
4213#if defined(USE_CR) || defined(PROTO)
4214
4215# if defined(__MSL__) && (__MSL__ >= 22)
4216/*
4217 * Newer version of the Metrowerks library handle DOS and UNIX files
4218 * without help.
4219 * Test with earlier versions, MSL 2.2 is the library supplied with
4220 * Codewarrior Pro 2.
4221 */
4222 char *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004223fgets_cr(char *s, int n, FILE *stream)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004224{
4225 return fgets(s, n, stream);
4226}
4227# else
4228/*
4229 * Version of fgets() which also works for lines ending in a <CR> only
4230 * (Macintosh format).
4231 * For older versions of the Metrowerks library.
4232 * At least CodeWarrior 9 needed this code.
4233 */
4234 char *
Bram Moolenaard14e00e2016-01-31 17:30:51 +01004235fgets_cr(char *s, int n, FILE *stream)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004236{
4237 int c = 0;
4238 int char_read = 0;
4239
4240 while (!feof(stream) && c != '\r' && c != '\n' && char_read < n - 1)
4241 {
4242 c = fgetc(stream);
4243 s[char_read++] = c;
4244 /* If the file is in DOS format, we need to skip a NL after a CR. I
4245 * thought it was the other way around, but this appears to work... */
4246 if (c == '\n')
4247 {
4248 c = fgetc(stream);
4249 if (c != '\r')
4250 ungetc(c, stream);
4251 }
4252 }
4253
4254 s[char_read] = 0;
4255 if (char_read == 0)
4256 return NULL;
4257
4258 if (feof(stream) && char_read == 1)
4259 return NULL;
4260
4261 return s;
4262}
4263# endif
4264#endif
4265
4266/*
4267 * Get one full line from a sourced file.
4268 * Called by do_cmdline() when it's called from do_source().
4269 *
4270 * Return a pointer to the line in allocated memory.
4271 * Return NULL for end-of-file or some error.
4272 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004273 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004274getsourceline(int c UNUSED, void *cookie, int indent UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004275{
4276 struct source_cookie *sp = (struct source_cookie *)cookie;
4277 char_u *line;
Bram Moolenaarc047b9a2012-02-11 20:40:55 +01004278 char_u *p;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004279
4280#ifdef FEAT_EVAL
4281 /* If breakpoints have been added/deleted need to check for it. */
4282 if (sp->dbg_tick < debug_tick)
4283 {
4284 sp->breakpoint = dbg_find_breakpoint(TRUE, sp->fname, sourcing_lnum);
4285 sp->dbg_tick = debug_tick;
4286 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00004287# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004288 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004289 script_line_end();
4290# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004291#endif
4292 /*
4293 * Get current line. If there is a read-ahead line, use it, otherwise get
4294 * one now.
4295 */
4296 if (sp->finished)
4297 line = NULL;
4298 else if (sp->nextline == NULL)
4299 line = get_one_sourceline(sp);
4300 else
4301 {
4302 line = sp->nextline;
4303 sp->nextline = NULL;
4304 ++sourcing_lnum;
4305 }
Bram Moolenaare2cc9702005-03-15 22:43:58 +00004306#ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004307 if (line != NULL && do_profiling == PROF_YES)
Bram Moolenaare2cc9702005-03-15 22:43:58 +00004308 script_line_start();
4309#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004310
4311 /* Only concatenate lines starting with a \ when 'cpoptions' doesn't
4312 * contain the 'C' flag. */
4313 if (line != NULL && (vim_strchr(p_cpo, CPO_CONCAT) == NULL))
4314 {
4315 /* compensate for the one line read-ahead */
4316 --sourcing_lnum;
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004317
4318 /* Get the next line and concatenate it when it starts with a
4319 * backslash. We always need to read the next line, keep it in
4320 * sp->nextline. */
4321 sp->nextline = get_one_sourceline(sp);
4322 if (sp->nextline != NULL && *(p = skipwhite(sp->nextline)) == '\\')
Bram Moolenaar071d4272004-06-13 20:20:40 +00004323 {
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004324 garray_T ga;
4325
Bram Moolenaarb549a732012-02-22 18:29:33 +01004326 ga_init2(&ga, (int)sizeof(char_u), 400);
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004327 ga_concat(&ga, line);
4328 ga_concat(&ga, p + 1);
4329 for (;;)
4330 {
4331 vim_free(sp->nextline);
4332 sp->nextline = get_one_sourceline(sp);
4333 if (sp->nextline == NULL)
4334 break;
4335 p = skipwhite(sp->nextline);
4336 if (*p != '\\')
4337 break;
Bram Moolenaarb549a732012-02-22 18:29:33 +01004338 /* Adjust the growsize to the current length to speed up
4339 * concatenating many lines. */
4340 if (ga.ga_len > 400)
4341 {
4342 if (ga.ga_len > 8000)
4343 ga.ga_growsize = 8000;
4344 else
4345 ga.ga_growsize = ga.ga_len;
4346 }
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004347 ga_concat(&ga, p + 1);
4348 }
4349 ga_append(&ga, NUL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004350 vim_free(line);
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004351 line = ga.ga_data;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004352 }
4353 }
4354
4355#ifdef FEAT_MBYTE
4356 if (line != NULL && sp->conv.vc_type != CONV_NONE)
4357 {
Bram Moolenaarc047b9a2012-02-11 20:40:55 +01004358 char_u *s;
4359
Bram Moolenaar071d4272004-06-13 20:20:40 +00004360 /* Convert the encoding of the script line. */
4361 s = string_convert(&sp->conv, line, NULL);
4362 if (s != NULL)
4363 {
4364 vim_free(line);
4365 line = s;
4366 }
4367 }
4368#endif
4369
4370#ifdef FEAT_EVAL
4371 /* Did we encounter a breakpoint? */
4372 if (sp->breakpoint != 0 && sp->breakpoint <= sourcing_lnum)
4373 {
4374 dbg_breakpoint(sp->fname, sourcing_lnum);
4375 /* Find next breakpoint. */
4376 sp->breakpoint = dbg_find_breakpoint(TRUE, sp->fname, sourcing_lnum);
4377 sp->dbg_tick = debug_tick;
4378 }
4379#endif
4380
4381 return line;
4382}
4383
4384 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004385get_one_sourceline(struct source_cookie *sp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004386{
4387 garray_T ga;
4388 int len;
4389 int c;
4390 char_u *buf;
4391#ifdef USE_CRNL
4392 int has_cr; /* CR-LF found */
4393#endif
4394#ifdef USE_CR
4395 char_u *scan;
4396#endif
4397 int have_read = FALSE;
4398
4399 /* use a growarray to store the sourced line */
Bram Moolenaar6ac54292005-02-02 23:07:25 +00004400 ga_init2(&ga, 1, 250);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004401
4402 /*
4403 * Loop until there is a finished line (or end-of-file).
4404 */
4405 sourcing_lnum++;
4406 for (;;)
4407 {
Bram Moolenaar6ac54292005-02-02 23:07:25 +00004408 /* make room to read at least 120 (more) characters */
4409 if (ga_grow(&ga, 120) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004410 break;
4411 buf = (char_u *)ga.ga_data;
4412
4413#ifdef USE_CR
4414 if (sp->fileformat == EOL_MAC)
4415 {
Bram Moolenaar86b68352004-12-27 21:59:20 +00004416 if (fgets_cr((char *)buf + ga.ga_len, ga.ga_maxlen - ga.ga_len,
4417 sp->fp) == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004418 break;
4419 }
4420 else
4421#endif
Bram Moolenaar86b68352004-12-27 21:59:20 +00004422 if (fgets((char *)buf + ga.ga_len, ga.ga_maxlen - ga.ga_len,
4423 sp->fp) == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004424 break;
Bram Moolenaar6ac54292005-02-02 23:07:25 +00004425 len = ga.ga_len + (int)STRLEN(buf + ga.ga_len);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004426#ifdef USE_CRNL
4427 /* Ignore a trailing CTRL-Z, when in Dos mode. Only recognize the
4428 * CTRL-Z by its own, or after a NL. */
4429 if ( (len == 1 || (len >= 2 && buf[len - 2] == '\n'))
4430 && sp->fileformat == EOL_DOS
4431 && buf[len - 1] == Ctrl_Z)
4432 {
4433 buf[len - 1] = NUL;
4434 break;
4435 }
4436#endif
4437
4438#ifdef USE_CR
4439 /* If the read doesn't stop on a new line, and there's
4440 * some CR then we assume a Mac format */
4441 if (sp->fileformat == EOL_UNKNOWN)
4442 {
4443 if (buf[len - 1] != '\n' && vim_strchr(buf, '\r') != NULL)
4444 sp->fileformat = EOL_MAC;
4445 else
4446 sp->fileformat = EOL_UNIX;
4447 }
4448
4449 if (sp->fileformat == EOL_MAC)
4450 {
4451 scan = vim_strchr(buf, '\r');
4452
4453 if (scan != NULL)
4454 {
4455 *scan = '\n';
4456 if (*(scan + 1) != 0)
4457 {
4458 *(scan + 1) = 0;
4459 fseek(sp->fp, (long)(scan - buf - len + 1), SEEK_CUR);
4460 }
4461 }
4462 len = STRLEN(buf);
4463 }
4464#endif
4465
4466 have_read = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004467 ga.ga_len = len;
4468
4469 /* If the line was longer than the buffer, read more. */
Bram Moolenaar86b68352004-12-27 21:59:20 +00004470 if (ga.ga_maxlen - ga.ga_len == 1 && buf[len - 1] != '\n')
Bram Moolenaar071d4272004-06-13 20:20:40 +00004471 continue;
4472
4473 if (len >= 1 && buf[len - 1] == '\n') /* remove trailing NL */
4474 {
4475#ifdef USE_CRNL
4476 has_cr = (len >= 2 && buf[len - 2] == '\r');
4477 if (sp->fileformat == EOL_UNKNOWN)
4478 {
4479 if (has_cr)
4480 sp->fileformat = EOL_DOS;
4481 else
4482 sp->fileformat = EOL_UNIX;
4483 }
4484
4485 if (sp->fileformat == EOL_DOS)
4486 {
4487 if (has_cr) /* replace trailing CR */
4488 {
4489 buf[len - 2] = '\n';
4490 --len;
4491 --ga.ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004492 }
4493 else /* lines like ":map xx yy^M" will have failed */
4494 {
4495 if (!sp->error)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00004496 {
4497 msg_source(hl_attr(HLF_W));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004498 EMSG(_("W15: Warning: Wrong line separator, ^M may be missing"));
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00004499 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004500 sp->error = TRUE;
4501 sp->fileformat = EOL_UNIX;
4502 }
4503 }
4504#endif
4505 /* The '\n' is escaped if there is an odd number of ^V's just
4506 * before it, first set "c" just before the 'V's and then check
4507 * len&c parities (is faster than ((len-c)%2 == 0)) -- Acevedo */
4508 for (c = len - 2; c >= 0 && buf[c] == Ctrl_V; c--)
4509 ;
4510 if ((len & 1) != (c & 1)) /* escaped NL, read more */
4511 {
4512 sourcing_lnum++;
4513 continue;
4514 }
4515
4516 buf[len - 1] = NUL; /* remove the NL */
4517 }
4518
4519 /*
4520 * Check for ^C here now and then, so recursive :so can be broken.
4521 */
4522 line_breakcheck();
4523 break;
4524 }
4525
4526 if (have_read)
4527 return (char_u *)ga.ga_data;
4528
4529 vim_free(ga.ga_data);
4530 return NULL;
4531}
4532
Bram Moolenaar05159a02005-02-26 23:04:13 +00004533#if defined(FEAT_PROFILE) || defined(PROTO)
4534/*
4535 * Called when starting to read a script line.
4536 * "sourcing_lnum" must be correct!
4537 * When skipping lines it may not actually be executed, but we won't find out
4538 * until later and we need to store the time now.
4539 */
4540 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004541script_line_start(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004542{
4543 scriptitem_T *si;
4544 sn_prl_T *pp;
4545
4546 if (current_SID <= 0 || current_SID > script_items.ga_len)
4547 return;
4548 si = &SCRIPT_ITEM(current_SID);
4549 if (si->sn_prof_on && sourcing_lnum >= 1)
4550 {
Bram Moolenaar8c8de832008-06-24 22:58:06 +00004551 /* Grow the array before starting the timer, so that the time spent
Bram Moolenaar05159a02005-02-26 23:04:13 +00004552 * here isn't counted. */
Bram Moolenaarcde88542015-08-11 19:14:00 +02004553 (void)ga_grow(&si->sn_prl_ga, (int)(sourcing_lnum - si->sn_prl_ga.ga_len));
Bram Moolenaar05159a02005-02-26 23:04:13 +00004554 si->sn_prl_idx = sourcing_lnum - 1;
4555 while (si->sn_prl_ga.ga_len <= si->sn_prl_idx
4556 && si->sn_prl_ga.ga_len < si->sn_prl_ga.ga_maxlen)
4557 {
4558 /* Zero counters for a line that was not used before. */
4559 pp = &PRL_ITEM(si, si->sn_prl_ga.ga_len);
4560 pp->snp_count = 0;
4561 profile_zero(&pp->sn_prl_total);
4562 profile_zero(&pp->sn_prl_self);
4563 ++si->sn_prl_ga.ga_len;
4564 }
4565 si->sn_prl_execed = FALSE;
4566 profile_start(&si->sn_prl_start);
4567 profile_zero(&si->sn_prl_children);
4568 profile_get_wait(&si->sn_prl_wait);
4569 }
4570}
4571
4572/*
4573 * Called when actually executing a function line.
4574 */
4575 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004576script_line_exec(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004577{
4578 scriptitem_T *si;
4579
4580 if (current_SID <= 0 || current_SID > script_items.ga_len)
4581 return;
4582 si = &SCRIPT_ITEM(current_SID);
4583 if (si->sn_prof_on && si->sn_prl_idx >= 0)
4584 si->sn_prl_execed = TRUE;
4585}
4586
4587/*
4588 * Called when done with a function line.
4589 */
4590 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004591script_line_end(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004592{
4593 scriptitem_T *si;
4594 sn_prl_T *pp;
4595
4596 if (current_SID <= 0 || current_SID > script_items.ga_len)
4597 return;
4598 si = &SCRIPT_ITEM(current_SID);
4599 if (si->sn_prof_on && si->sn_prl_idx >= 0
4600 && si->sn_prl_idx < si->sn_prl_ga.ga_len)
4601 {
4602 if (si->sn_prl_execed)
4603 {
4604 pp = &PRL_ITEM(si, si->sn_prl_idx);
4605 ++pp->snp_count;
4606 profile_end(&si->sn_prl_start);
4607 profile_sub_wait(&si->sn_prl_wait, &si->sn_prl_start);
Bram Moolenaar05159a02005-02-26 23:04:13 +00004608 profile_add(&pp->sn_prl_total, &si->sn_prl_start);
Bram Moolenaar1056d982006-03-09 22:37:52 +00004609 profile_self(&pp->sn_prl_self, &si->sn_prl_start,
4610 &si->sn_prl_children);
Bram Moolenaar05159a02005-02-26 23:04:13 +00004611 }
4612 si->sn_prl_idx = -1;
4613 }
4614}
4615#endif
4616
Bram Moolenaar071d4272004-06-13 20:20:40 +00004617/*
4618 * ":scriptencoding": Set encoding conversion for a sourced script.
4619 * Without the multi-byte feature it's simply ignored.
4620 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004621 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004622ex_scriptencoding(exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004623{
4624#ifdef FEAT_MBYTE
4625 struct source_cookie *sp;
4626 char_u *name;
4627
4628 if (!getline_equal(eap->getline, eap->cookie, getsourceline))
4629 {
4630 EMSG(_("E167: :scriptencoding used outside of a sourced file"));
4631 return;
4632 }
4633
4634 if (*eap->arg != NUL)
4635 {
4636 name = enc_canonize(eap->arg);
4637 if (name == NULL) /* out of memory */
4638 return;
4639 }
4640 else
4641 name = eap->arg;
4642
4643 /* Setup for conversion from the specified encoding to 'encoding'. */
4644 sp = (struct source_cookie *)getline_cookie(eap->getline, eap->cookie);
4645 convert_setup(&sp->conv, name, p_enc);
4646
4647 if (name != eap->arg)
4648 vim_free(name);
4649#endif
4650}
4651
4652#if defined(FEAT_EVAL) || defined(PROTO)
4653/*
4654 * ":finish": Mark a sourced file as finished.
4655 */
4656 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004657ex_finish(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004658{
4659 if (getline_equal(eap->getline, eap->cookie, getsourceline))
4660 do_finish(eap, FALSE);
4661 else
4662 EMSG(_("E168: :finish used outside of a sourced file"));
4663}
4664
4665/*
4666 * Mark a sourced file as finished. Possibly makes the ":finish" pending.
4667 * Also called for a pending finish at the ":endtry" or after returning from
4668 * an extra do_cmdline(). "reanimate" is used in the latter case.
4669 */
4670 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004671do_finish(exarg_T *eap, int reanimate)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004672{
4673 int idx;
4674
4675 if (reanimate)
4676 ((struct source_cookie *)getline_cookie(eap->getline,
4677 eap->cookie))->finished = FALSE;
4678
4679 /*
4680 * Cleanup (and inactivate) conditionals, but stop when a try conditional
4681 * not in its finally clause (which then is to be executed next) is found.
4682 * In this case, make the ":finish" pending for execution at the ":endtry".
4683 * Otherwise, finish normally.
4684 */
4685 idx = cleanup_conditionals(eap->cstack, 0, TRUE);
4686 if (idx >= 0)
4687 {
4688 eap->cstack->cs_pending[idx] = CSTP_FINISH;
4689 report_make_pending(CSTP_FINISH, NULL);
4690 }
4691 else
4692 ((struct source_cookie *)getline_cookie(eap->getline,
4693 eap->cookie))->finished = TRUE;
4694}
4695
4696
4697/*
4698 * Return TRUE when a sourced file had the ":finish" command: Don't give error
4699 * message for missing ":endif".
4700 * Return FALSE when not sourcing a file.
4701 */
4702 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004703source_finished(
4704 char_u *(*fgetline)(int, void *, int),
4705 void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004706{
Bram Moolenaar89d40322006-08-29 15:30:07 +00004707 return (getline_equal(fgetline, cookie, getsourceline)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004708 && ((struct source_cookie *)getline_cookie(
Bram Moolenaar89d40322006-08-29 15:30:07 +00004709 fgetline, cookie))->finished);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004710}
4711#endif
4712
4713#if defined(FEAT_LISTCMDS) || defined(PROTO)
4714/*
4715 * ":checktime [buffer]"
4716 */
4717 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004718ex_checktime(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004719{
4720 buf_T *buf;
4721 int save_no_check_timestamps = no_check_timestamps;
4722
4723 no_check_timestamps = 0;
4724 if (eap->addr_count == 0) /* default is all buffers */
4725 check_timestamps(FALSE);
4726 else
4727 {
4728 buf = buflist_findnr((int)eap->line2);
4729 if (buf != NULL) /* cannot happen? */
4730 (void)buf_check_timestamp(buf, FALSE);
4731 }
4732 no_check_timestamps = save_no_check_timestamps;
4733}
4734#endif
4735
Bram Moolenaar071d4272004-06-13 20:20:40 +00004736#if (defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
4737 && (defined(FEAT_EVAL) || defined(FEAT_MULTI_LANG))
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004738# define HAVE_GET_LOCALE_VAL
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004739static char_u *get_locale_val(int what);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004740
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004741 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004742get_locale_val(int what)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004743{
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004744 char_u *loc;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004745
Bram Moolenaar48e330a2016-02-23 14:53:34 +01004746 /* Obtain the locale value from the libraries. */
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004747 loc = (char_u *)setlocale(what, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004748
Bram Moolenaar61660ea2006-04-07 21:40:07 +00004749# ifdef WIN32
Bram Moolenaar071d4272004-06-13 20:20:40 +00004750 if (loc != NULL)
4751 {
4752 char_u *p;
4753
Bram Moolenaar61660ea2006-04-07 21:40:07 +00004754 /* setocale() returns something like "LC_COLLATE=<name>;LC_..." when
4755 * one of the values (e.g., LC_CTYPE) differs. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004756 p = vim_strchr(loc, '=');
4757 if (p != NULL)
4758 {
4759 loc = ++p;
4760 while (*p != NUL) /* remove trailing newline */
4761 {
Bram Moolenaar61660ea2006-04-07 21:40:07 +00004762 if (*p < ' ' || *p == ';')
Bram Moolenaar071d4272004-06-13 20:20:40 +00004763 {
4764 *p = NUL;
4765 break;
4766 }
4767 ++p;
4768 }
4769 }
4770 }
4771# endif
4772
4773 return loc;
4774}
4775#endif
4776
4777
4778#ifdef WIN32
4779/*
4780 * On MS-Windows locale names are strings like "German_Germany.1252", but
4781 * gettext expects "de". Try to translate one into another here for a few
4782 * supported languages.
4783 */
4784 static char_u *
4785gettext_lang(char_u *name)
4786{
4787 int i;
4788 static char *(mtable[]) = {
4789 "afrikaans", "af",
4790 "czech", "cs",
4791 "dutch", "nl",
4792 "german", "de",
4793 "english_united kingdom", "en_GB",
4794 "spanish", "es",
4795 "french", "fr",
4796 "italian", "it",
4797 "japanese", "ja",
4798 "korean", "ko",
4799 "norwegian", "no",
4800 "polish", "pl",
4801 "russian", "ru",
4802 "slovak", "sk",
4803 "swedish", "sv",
4804 "ukrainian", "uk",
4805 "chinese_china", "zh_CN",
4806 "chinese_taiwan", "zh_TW",
4807 NULL};
4808
4809 for (i = 0; mtable[i] != NULL; i += 2)
4810 if (STRNICMP(mtable[i], name, STRLEN(mtable[i])) == 0)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004811 return (char_u *)mtable[i + 1];
Bram Moolenaar071d4272004-06-13 20:20:40 +00004812 return name;
4813}
4814#endif
4815
4816#if defined(FEAT_MULTI_LANG) || defined(PROTO)
4817/*
4818 * Obtain the current messages language. Used to set the default for
4819 * 'helplang'. May return NULL or an empty string.
4820 */
4821 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004822get_mess_lang(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004823{
4824 char_u *p;
4825
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004826# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00004827# if defined(LC_MESSAGES)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004828 p = get_locale_val(LC_MESSAGES);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004829# else
4830 /* This is necessary for Win32, where LC_MESSAGES is not defined and $LANG
Bram Moolenaar61660ea2006-04-07 21:40:07 +00004831 * may be set to the LCID number. LC_COLLATE is the best guess, LC_TIME
4832 * and LC_MONETARY may be set differently for a Japanese working in the
4833 * US. */
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004834 p = get_locale_val(LC_COLLATE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004835# endif
4836# else
4837 p = mch_getenv((char_u *)"LC_ALL");
4838 if (p == NULL || *p == NUL)
4839 {
4840 p = mch_getenv((char_u *)"LC_MESSAGES");
4841 if (p == NULL || *p == NUL)
4842 p = mch_getenv((char_u *)"LANG");
4843 }
4844# endif
4845# ifdef WIN32
4846 p = gettext_lang(p);
4847# endif
4848 return p;
4849}
4850#endif
4851
Bram Moolenaardef9e822004-12-31 20:58:58 +00004852/* Complicated #if; matches with where get_mess_env() is used below. */
4853#if (defined(FEAT_EVAL) && !((defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
4854 && defined(LC_MESSAGES))) \
4855 || ((defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
4856 && (defined(FEAT_GETTEXT) || defined(FEAT_MBYTE)) \
4857 && !defined(LC_MESSAGES))
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01004858static char_u *get_mess_env(void);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004859
4860/*
4861 * Get the language used for messages from the environment.
4862 */
4863 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004864get_mess_env(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004865{
4866 char_u *p;
4867
4868 p = mch_getenv((char_u *)"LC_ALL");
4869 if (p == NULL || *p == NUL)
4870 {
4871 p = mch_getenv((char_u *)"LC_MESSAGES");
4872 if (p == NULL || *p == NUL)
4873 {
4874 p = mch_getenv((char_u *)"LANG");
4875 if (p != NULL && VIM_ISDIGIT(*p))
4876 p = NULL; /* ignore something like "1043" */
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004877# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00004878 if (p == NULL || *p == NUL)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004879 p = get_locale_val(LC_CTYPE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004880# endif
4881 }
4882 }
4883 return p;
4884}
4885#endif
4886
4887#if defined(FEAT_EVAL) || defined(PROTO)
4888
4889/*
4890 * Set the "v:lang" variable according to the current locale setting.
4891 * Also do "v:lc_time"and "v:ctype".
4892 */
4893 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004894set_lang_var(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004895{
4896 char_u *loc;
4897
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004898# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004899 loc = get_locale_val(LC_CTYPE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004900# else
4901 /* setlocale() not supported: use the default value */
4902 loc = (char_u *)"C";
4903# endif
4904 set_vim_var_string(VV_CTYPE, loc, -1);
4905
4906 /* When LC_MESSAGES isn't defined use the value from $LC_MESSAGES, fall
4907 * back to LC_CTYPE if it's empty. */
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004908# if defined(HAVE_GET_LOCALE_VAL) && defined(LC_MESSAGES)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004909 loc = get_locale_val(LC_MESSAGES);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004910# else
4911 loc = get_mess_env();
4912# endif
4913 set_vim_var_string(VV_LANG, loc, -1);
4914
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004915# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004916 loc = get_locale_val(LC_TIME);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004917# endif
4918 set_vim_var_string(VV_LC_TIME, loc, -1);
4919}
4920#endif
4921
4922#if (defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
4923 && (defined(FEAT_GETTEXT) || defined(FEAT_MBYTE))
4924/*
4925 * ":language": Set the language (locale).
4926 */
4927 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004928ex_language(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004929{
4930 char *loc;
4931 char_u *p;
4932 char_u *name;
4933 int what = LC_ALL;
4934 char *whatstr = "";
4935#ifdef LC_MESSAGES
4936# define VIM_LC_MESSAGES LC_MESSAGES
4937#else
4938# define VIM_LC_MESSAGES 6789
4939#endif
4940
4941 name = eap->arg;
4942
4943 /* Check for "messages {name}", "ctype {name}" or "time {name}" argument.
4944 * Allow abbreviation, but require at least 3 characters to avoid
4945 * confusion with a two letter language name "me" or "ct". */
4946 p = skiptowhite(eap->arg);
4947 if ((*p == NUL || vim_iswhite(*p)) && p - eap->arg >= 3)
4948 {
4949 if (STRNICMP(eap->arg, "messages", p - eap->arg) == 0)
4950 {
4951 what = VIM_LC_MESSAGES;
4952 name = skipwhite(p);
4953 whatstr = "messages ";
4954 }
4955 else if (STRNICMP(eap->arg, "ctype", p - eap->arg) == 0)
4956 {
4957 what = LC_CTYPE;
4958 name = skipwhite(p);
4959 whatstr = "ctype ";
4960 }
4961 else if (STRNICMP(eap->arg, "time", p - eap->arg) == 0)
4962 {
4963 what = LC_TIME;
4964 name = skipwhite(p);
4965 whatstr = "time ";
4966 }
4967 }
4968
4969 if (*name == NUL)
4970 {
4971#ifndef LC_MESSAGES
4972 if (what == VIM_LC_MESSAGES)
4973 p = get_mess_env();
4974 else
4975#endif
4976 p = (char_u *)setlocale(what, NULL);
4977 if (p == NULL || *p == NUL)
4978 p = (char_u *)"Unknown";
4979 smsg((char_u *)_("Current %slanguage: \"%s\""), whatstr, p);
4980 }
4981 else
4982 {
4983#ifndef LC_MESSAGES
4984 if (what == VIM_LC_MESSAGES)
4985 loc = "";
4986 else
4987#endif
Bram Moolenaar8c8de832008-06-24 22:58:06 +00004988 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00004989 loc = setlocale(what, (char *)name);
Bram Moolenaar8c8de832008-06-24 22:58:06 +00004990#if defined(FEAT_FLOAT) && defined(LC_NUMERIC)
4991 /* Make sure strtod() uses a decimal point, not a comma. */
4992 setlocale(LC_NUMERIC, "C");
4993#endif
4994 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004995 if (loc == NULL)
4996 EMSG2(_("E197: Cannot set language to \"%s\""), name);
4997 else
4998 {
4999#ifdef HAVE_NL_MSG_CAT_CNTR
5000 /* Need to do this for GNU gettext, otherwise cached translations
5001 * will be used again. */
5002 extern int _nl_msg_cat_cntr;
5003
5004 ++_nl_msg_cat_cntr;
5005#endif
Bram Moolenaarb8017e72007-05-10 18:59:07 +00005006 /* Reset $LC_ALL, otherwise it would overrule everything. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005007 vim_setenv((char_u *)"LC_ALL", (char_u *)"");
5008
5009 if (what != LC_TIME)
5010 {
5011 /* Tell gettext() what to translate to. It apparently doesn't
5012 * use the currently effective locale. Also do this when
5013 * FEAT_GETTEXT isn't defined, so that shell commands use this
5014 * value. */
5015 if (what == LC_ALL)
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00005016 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00005017 vim_setenv((char_u *)"LANG", name);
Bram Moolenaare3a0b532013-06-28 20:36:30 +02005018
5019 /* Clear $LANGUAGE because GNU gettext uses it. */
5020 vim_setenv((char_u *)"LANGUAGE", (char_u *)"");
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00005021# ifdef WIN32
5022 /* Apparently MS-Windows printf() may cause a crash when
5023 * we give it 8-bit text while it's expecting text in the
5024 * current locale. This call avoids that. */
5025 setlocale(LC_CTYPE, "C");
5026# endif
5027 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00005028 if (what != LC_CTYPE)
5029 {
5030 char_u *mname;
5031#ifdef WIN32
5032 mname = gettext_lang(name);
5033#else
5034 mname = name;
5035#endif
5036 vim_setenv((char_u *)"LC_MESSAGES", mname);
5037#ifdef FEAT_MULTI_LANG
5038 set_helplang_default(mname);
5039#endif
5040 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00005041 }
5042
5043# ifdef FEAT_EVAL
5044 /* Set v:lang, v:lc_time and v:ctype to the final result. */
5045 set_lang_var();
5046# endif
Bram Moolenaar6cc00c72011-10-20 21:41:09 +02005047# ifdef FEAT_TITLE
5048 maketitle();
5049# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00005050 }
5051 }
5052}
5053
5054# if defined(FEAT_CMDL_COMPL) || defined(PROTO)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005055
5056static char_u **locales = NULL; /* Array of all available locales */
5057static int did_init_locales = FALSE;
5058
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01005059static void init_locales(void);
5060static char_u **find_locales(void);
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005061
5062/*
5063 * Lazy initialization of all available locales.
5064 */
5065 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005066init_locales(void)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005067{
5068 if (!did_init_locales)
5069 {
5070 did_init_locales = TRUE;
5071 locales = find_locales();
5072 }
5073}
5074
5075/* Return an array of strings for all available locales + NULL for the
5076 * last element. Return NULL in case of error. */
5077 static char_u **
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005078find_locales(void)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005079{
5080 garray_T locales_ga;
5081 char_u *loc;
5082
5083 /* Find all available locales by running command "locale -a". If this
5084 * doesn't work we won't have completion. */
5085 char_u *locale_a = get_cmd_output((char_u *)"locale -a",
Bram Moolenaar39c29ed2014-04-05 19:44:40 +02005086 NULL, SHELL_SILENT, NULL);
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005087 if (locale_a == NULL)
5088 return NULL;
5089 ga_init2(&locales_ga, sizeof(char_u *), 20);
5090
5091 /* Transform locale_a string where each locale is separated by "\n"
5092 * into an array of locale strings. */
5093 loc = (char_u *)strtok((char *)locale_a, "\n");
5094
5095 while (loc != NULL)
5096 {
5097 if (ga_grow(&locales_ga, 1) == FAIL)
5098 break;
5099 loc = vim_strsave(loc);
5100 if (loc == NULL)
5101 break;
5102
5103 ((char_u **)locales_ga.ga_data)[locales_ga.ga_len++] = loc;
5104 loc = (char_u *)strtok(NULL, "\n");
5105 }
5106 vim_free(locale_a);
5107 if (ga_grow(&locales_ga, 1) == FAIL)
5108 {
5109 ga_clear(&locales_ga);
5110 return NULL;
5111 }
5112 ((char_u **)locales_ga.ga_data)[locales_ga.ga_len] = NULL;
5113 return (char_u **)locales_ga.ga_data;
5114}
5115
5116# if defined(EXITFREE) || defined(PROTO)
5117 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005118free_locales(void)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005119{
5120 int i;
5121 if (locales != NULL)
5122 {
5123 for (i = 0; locales[i] != NULL; i++)
5124 vim_free(locales[i]);
5125 vim_free(locales);
5126 locales = NULL;
5127 }
5128}
5129# endif
5130
Bram Moolenaar071d4272004-06-13 20:20:40 +00005131/*
5132 * Function given to ExpandGeneric() to obtain the possible arguments of the
5133 * ":language" command.
5134 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005135 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005136get_lang_arg(expand_T *xp UNUSED, int idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005137{
5138 if (idx == 0)
5139 return (char_u *)"messages";
5140 if (idx == 1)
5141 return (char_u *)"ctype";
5142 if (idx == 2)
5143 return (char_u *)"time";
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005144
5145 init_locales();
5146 if (locales == NULL)
5147 return NULL;
5148 return locales[idx - 3];
5149}
5150
5151/*
5152 * Function given to ExpandGeneric() to obtain the available locales.
5153 */
5154 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005155get_locales(expand_T *xp UNUSED, int idx)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005156{
5157 init_locales();
5158 if (locales == NULL)
5159 return NULL;
5160 return locales[idx];
Bram Moolenaar071d4272004-06-13 20:20:40 +00005161}
5162# endif
5163
5164#endif