blob: 225225a78c0206df453f111af18159ef46930cf5 [file] [log] [blame]
Bram Moolenaaredf3f972016-08-29 22:49:24 +02001/* vi:set ts=8 sts=4 sw=4 noet:
Bram Moolenaar071d4272004-06-13 20:20:40 +00002 *
3 * VIM - Vi IMproved by Bram Moolenaar
4 *
5 * Do ":help uganda" in Vim to read copying and usage conditions.
6 * Do ":help credits" in Vim to see a list of people who contributed.
7 * See README.txt for an overview of the Vim source code.
8 */
9
10/*
11 * ex_cmds2.c: some more functions for command line commands
12 */
13
Bram Moolenaar071d4272004-06-13 20:20:40 +000014#include "vim.h"
Bram Moolenaar071d4272004-06-13 20:20:40 +000015#include "version.h"
16
Bram Moolenaarf28dbce2016-01-29 22:03:47 +010017static void cmd_source(char_u *fname, exarg_T *eap);
Bram Moolenaar071d4272004-06-13 20:20:40 +000018
Bram Moolenaar05159a02005-02-26 23:04:13 +000019#ifdef FEAT_EVAL
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +000020/* Growarray to store info about already sourced scripts.
Bram Moolenaar05159a02005-02-26 23:04:13 +000021 * For Unix also store the dev/ino, so that we don't have to stat() each
22 * script when going through the list. */
23typedef struct scriptitem_S
24{
25 char_u *sn_name;
26# ifdef UNIX
Bram Moolenaarbf0c4522009-05-16 19:16:33 +000027 int sn_dev_valid;
28 dev_t sn_dev;
Bram Moolenaar05159a02005-02-26 23:04:13 +000029 ino_t sn_ino;
30# endif
31# ifdef FEAT_PROFILE
32 int sn_prof_on; /* TRUE when script is/was profiled */
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +000033 int sn_pr_force; /* forceit: profile functions in this script */
Bram Moolenaar05159a02005-02-26 23:04:13 +000034 proftime_T sn_pr_child; /* time set when going into first child */
35 int sn_pr_nest; /* nesting for sn_pr_child */
36 /* profiling the script as a whole */
37 int sn_pr_count; /* nr of times sourced */
Bram Moolenaar8c8de832008-06-24 22:58:06 +000038 proftime_T sn_pr_total; /* time spent in script + children */
39 proftime_T sn_pr_self; /* time spent in script itself */
Bram Moolenaar05159a02005-02-26 23:04:13 +000040 proftime_T sn_pr_start; /* time at script start */
41 proftime_T sn_pr_children; /* time in children after script start */
42 /* profiling the script per line */
43 garray_T sn_prl_ga; /* things stored for every line */
44 proftime_T sn_prl_start; /* start time for current line */
45 proftime_T sn_prl_children; /* time spent in children for this line */
46 proftime_T sn_prl_wait; /* wait start time for current line */
47 int sn_prl_idx; /* index of line being timed; -1 if none */
48 int sn_prl_execed; /* line being timed was executed */
49# endif
50} scriptitem_T;
51
52static garray_T script_items = {0, 0, sizeof(scriptitem_T), 4, NULL};
53#define SCRIPT_ITEM(id) (((scriptitem_T *)script_items.ga_data)[(id) - 1])
54
55# ifdef FEAT_PROFILE
56/* Struct used in sn_prl_ga for every line of a script. */
57typedef struct sn_prl_S
58{
59 int snp_count; /* nr of times line was executed */
Bram Moolenaar8c8de832008-06-24 22:58:06 +000060 proftime_T sn_prl_total; /* time spent in a line + children */
61 proftime_T sn_prl_self; /* time spent in a line itself */
Bram Moolenaar05159a02005-02-26 23:04:13 +000062} sn_prl_T;
63
64# define PRL_ITEM(si, idx) (((sn_prl_T *)(si)->sn_prl_ga.ga_data)[(idx)])
65# endif
66#endif
67
Bram Moolenaar071d4272004-06-13 20:20:40 +000068#if defined(FEAT_EVAL) || defined(PROTO)
69static int debug_greedy = FALSE; /* batch mode debugging: don't save
70 and restore typeahead. */
Bram Moolenaarf1f60f82016-01-16 15:40:53 +010071static int get_maxbacktrace_level(void);
72static void do_setdebugtracelevel(char_u *arg);
73static void do_checkbacktracelevel(void);
74static void do_showbacktrace(char_u *cmd);
Bram Moolenaar071d4272004-06-13 20:20:40 +000075
76/*
77 * do_debug(): Debug mode.
78 * Repeatedly get Ex commands, until told to continue normal execution.
79 */
80 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +010081do_debug(char_u *cmd)
Bram Moolenaar071d4272004-06-13 20:20:40 +000082{
83 int save_msg_scroll = msg_scroll;
84 int save_State = State;
85 int save_did_emsg = did_emsg;
86 int save_cmd_silent = cmd_silent;
87 int save_msg_silent = msg_silent;
88 int save_emsg_silent = emsg_silent;
89 int save_redir_off = redir_off;
90 tasave_T typeaheadbuf;
Bram Moolenaaree3f7a52008-01-01 13:17:56 +000091 int typeahead_saved = FALSE;
Bram Moolenaar383c6f52008-01-04 15:01:07 +000092 int save_ignore_script = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +000093 int save_ex_normal_busy;
Bram Moolenaar071d4272004-06-13 20:20:40 +000094 int n;
95 char_u *cmdline = NULL;
96 char_u *p;
97 char *tail = NULL;
98 static int last_cmd = 0;
99#define CMD_CONT 1
100#define CMD_NEXT 2
101#define CMD_STEP 3
102#define CMD_FINISH 4
103#define CMD_QUIT 5
104#define CMD_INTERRUPT 6
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100105#define CMD_BACKTRACE 7
106#define CMD_FRAME 8
107#define CMD_UP 9
108#define CMD_DOWN 10
Bram Moolenaar071d4272004-06-13 20:20:40 +0000109
110#ifdef ALWAYS_USE_GUI
111 /* Can't do this when there is no terminal for input/output. */
112 if (!gui.in_use)
113 {
114 /* Break as soon as possible. */
115 debug_break_level = 9999;
116 return;
117 }
118#endif
119
120 /* Make sure we are in raw mode and start termcap mode. Might have side
121 * effects... */
122 settmode(TMODE_RAW);
123 starttermcap();
124
125 ++RedrawingDisabled; /* don't redisplay the window */
126 ++no_wait_return; /* don't wait for return */
127 did_emsg = FALSE; /* don't use error from debugged stuff */
128 cmd_silent = FALSE; /* display commands */
129 msg_silent = FALSE; /* display messages */
130 emsg_silent = FALSE; /* display error messages */
131 redir_off = TRUE; /* don't redirect debug commands */
132
133 State = NORMAL;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000134
135 if (!debug_did_msg)
136 MSG(_("Entering Debug mode. Type \"cont\" to continue."));
137 if (sourcing_name != NULL)
138 msg(sourcing_name);
139 if (sourcing_lnum != 0)
Bram Moolenaar555b2802005-05-19 21:08:39 +0000140 smsg((char_u *)_("line %ld: %s"), (long)sourcing_lnum, cmd);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000141 else
Bram Moolenaar555b2802005-05-19 21:08:39 +0000142 smsg((char_u *)_("cmd: %s"), cmd);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000143
144 /*
145 * Repeat getting a command and executing it.
146 */
147 for (;;)
148 {
149 msg_scroll = TRUE;
150 need_wait_return = FALSE;
Bram Moolenaar85b11762016-02-27 18:13:23 +0100151
Bram Moolenaar071d4272004-06-13 20:20:40 +0000152 /* Save the current typeahead buffer and replace it with an empty one.
153 * This makes sure we get input from the user here and don't interfere
154 * with the commands being executed. Reset "ex_normal_busy" to avoid
155 * the side effects of using ":normal". Save the stuff buffer and make
Bram Moolenaaree3f7a52008-01-01 13:17:56 +0000156 * it empty. Set ignore_script to avoid reading from script input. */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000157 save_ex_normal_busy = ex_normal_busy;
158 ex_normal_busy = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000159 if (!debug_greedy)
Bram Moolenaaree3f7a52008-01-01 13:17:56 +0000160 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000161 save_typeahead(&typeaheadbuf);
Bram Moolenaaree3f7a52008-01-01 13:17:56 +0000162 typeahead_saved = TRUE;
163 save_ignore_script = ignore_script;
164 ignore_script = TRUE;
165 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000166
Bram Moolenaardc303bc2016-05-17 17:45:38 +0200167 vim_free(cmdline);
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +0000168 cmdline = getcmdline_prompt('>', NULL, 0, EXPAND_NOTHING, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000169
Bram Moolenaaree3f7a52008-01-01 13:17:56 +0000170 if (typeahead_saved)
171 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000172 restore_typeahead(&typeaheadbuf);
Bram Moolenaaree3f7a52008-01-01 13:17:56 +0000173 ignore_script = save_ignore_script;
174 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000175 ex_normal_busy = save_ex_normal_busy;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000176
177 cmdline_row = msg_row;
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100178 msg_starthere();
Bram Moolenaar071d4272004-06-13 20:20:40 +0000179 if (cmdline != NULL)
180 {
181 /* If this is a debug command, set "last_cmd".
182 * If not, reset "last_cmd".
183 * For a blank line use previous command. */
184 p = skipwhite(cmdline);
185 if (*p != NUL)
186 {
187 switch (*p)
188 {
189 case 'c': last_cmd = CMD_CONT;
190 tail = "ont";
191 break;
192 case 'n': last_cmd = CMD_NEXT;
193 tail = "ext";
194 break;
195 case 's': last_cmd = CMD_STEP;
196 tail = "tep";
197 break;
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100198 case 'f':
199 last_cmd = 0;
200 if (p[1] == 'r')
201 {
202 last_cmd = CMD_FRAME;
203 tail = "rame";
204 }
205 else
206 {
207 last_cmd = CMD_FINISH;
208 tail = "inish";
209 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000210 break;
211 case 'q': last_cmd = CMD_QUIT;
212 tail = "uit";
213 break;
214 case 'i': last_cmd = CMD_INTERRUPT;
215 tail = "nterrupt";
216 break;
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100217 case 'b': last_cmd = CMD_BACKTRACE;
218 if (p[1] == 't')
219 tail = "t";
220 else
221 tail = "acktrace";
222 break;
223 case 'w': last_cmd = CMD_BACKTRACE;
224 tail = "here";
225 break;
226 case 'u': last_cmd = CMD_UP;
227 tail = "p";
228 break;
229 case 'd': last_cmd = CMD_DOWN;
230 tail = "own";
231 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000232 default: last_cmd = 0;
233 }
234 if (last_cmd != 0)
235 {
236 /* Check that the tail matches. */
237 ++p;
238 while (*p != NUL && *p == *tail)
239 {
240 ++p;
241 ++tail;
242 }
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100243 if (ASCII_ISALPHA(*p) && last_cmd != CMD_FRAME)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000244 last_cmd = 0;
245 }
246 }
247
248 if (last_cmd != 0)
249 {
250 /* Execute debug command: decided where to break next and
251 * return. */
252 switch (last_cmd)
253 {
254 case CMD_CONT:
255 debug_break_level = -1;
256 break;
257 case CMD_NEXT:
258 debug_break_level = ex_nesting_level;
259 break;
260 case CMD_STEP:
261 debug_break_level = 9999;
262 break;
263 case CMD_FINISH:
264 debug_break_level = ex_nesting_level - 1;
265 break;
266 case CMD_QUIT:
267 got_int = TRUE;
268 debug_break_level = -1;
269 break;
270 case CMD_INTERRUPT:
271 got_int = TRUE;
272 debug_break_level = 9999;
273 /* Do not repeat ">interrupt" cmd, continue stepping. */
274 last_cmd = CMD_STEP;
275 break;
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100276 case CMD_BACKTRACE:
277 do_showbacktrace(cmd);
278 continue;
279 case CMD_FRAME:
280 if (*p == NUL)
281 {
282 do_showbacktrace(cmd);
283 }
284 else
285 {
286 p = skipwhite(p);
287 do_setdebugtracelevel(p);
288 }
289 continue;
290 case CMD_UP:
291 debug_backtrace_level++;
292 do_checkbacktracelevel();
293 continue;
294 case CMD_DOWN:
295 debug_backtrace_level--;
296 do_checkbacktracelevel();
297 continue;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000298 }
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100299 /* Going out reset backtrace_level */
300 debug_backtrace_level = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000301 break;
302 }
303
304 /* don't debug this command */
305 n = debug_break_level;
306 debug_break_level = -1;
307 (void)do_cmdline(cmdline, getexline, NULL,
308 DOCMD_VERBOSE|DOCMD_EXCRESET);
309 debug_break_level = n;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000310 }
311 lines_left = Rows - 1;
312 }
313 vim_free(cmdline);
314
315 --RedrawingDisabled;
316 --no_wait_return;
317 redraw_all_later(NOT_VALID);
318 need_wait_return = FALSE;
319 msg_scroll = save_msg_scroll;
320 lines_left = Rows - 1;
321 State = save_State;
322 did_emsg = save_did_emsg;
323 cmd_silent = save_cmd_silent;
324 msg_silent = save_msg_silent;
325 emsg_silent = save_emsg_silent;
326 redir_off = save_redir_off;
327
328 /* Only print the message again when typing a command before coming back
329 * here. */
330 debug_did_msg = TRUE;
331}
332
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100333 static int
334get_maxbacktrace_level(void)
335{
336 char *p, *q;
Bram Moolenaardc633cf2016-04-23 14:33:19 +0200337 int maxbacktrace = 0;
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100338
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100339 if (sourcing_name != NULL)
340 {
341 p = (char *)sourcing_name;
342 while ((q = strstr(p, "..")) != NULL)
343 {
344 p = q + 2;
345 maxbacktrace++;
346 }
347 }
348 return maxbacktrace;
349}
350
351 static void
352do_setdebugtracelevel(char_u *arg)
353{
354 int level;
355
356 level = atoi((char *)arg);
357 if (*arg == '+' || level < 0)
358 debug_backtrace_level += level;
359 else
360 debug_backtrace_level = level;
361
362 do_checkbacktracelevel();
363}
364
365 static void
366do_checkbacktracelevel(void)
367{
368 if (debug_backtrace_level < 0)
369 {
370 debug_backtrace_level = 0;
371 MSG(_("frame is zero"));
372 }
373 else
374 {
375 int max = get_maxbacktrace_level();
376
377 if (debug_backtrace_level > max)
378 {
379 debug_backtrace_level = max;
380 smsg((char_u *)_("frame at highest level: %d"), max);
381 }
382 }
383}
384
385 static void
386do_showbacktrace(char_u *cmd)
387{
388 char *cur;
389 char *next;
390 int i = 0;
391 int max = get_maxbacktrace_level();
392
393 if (sourcing_name != NULL)
394 {
395 cur = (char *)sourcing_name;
396 while (!got_int)
397 {
398 next = strstr(cur, "..");
399 if (next != NULL)
400 *next = NUL;
401 if (i == max - debug_backtrace_level)
402 smsg((char_u *)"->%d %s", max - i, cur);
403 else
404 smsg((char_u *)" %d %s", max - i, cur);
405 ++i;
406 if (next == NULL)
407 break;
408 *next = '.';
409 cur = next + 2;
410 }
411 }
412 if (sourcing_lnum != 0)
413 smsg((char_u *)_("line %ld: %s"), (long)sourcing_lnum, cmd);
414 else
415 smsg((char_u *)_("cmd: %s"), cmd);
416}
417
Bram Moolenaar071d4272004-06-13 20:20:40 +0000418/*
419 * ":debug".
420 */
421 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100422ex_debug(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000423{
424 int debug_break_level_save = debug_break_level;
425
426 debug_break_level = 9999;
427 do_cmdline_cmd(eap->arg);
428 debug_break_level = debug_break_level_save;
429}
430
431static char_u *debug_breakpoint_name = NULL;
432static linenr_T debug_breakpoint_lnum;
433
434/*
435 * When debugging or a breakpoint is set on a skipped command, no debug prompt
436 * is shown by do_one_cmd(). This situation is indicated by debug_skipped, and
437 * debug_skipped_name is then set to the source name in the breakpoint case. If
438 * a skipped command decides itself that a debug prompt should be displayed, it
439 * can do so by calling dbg_check_skipped().
440 */
441static int debug_skipped;
442static char_u *debug_skipped_name;
443
444/*
445 * Go to debug mode when a breakpoint was encountered or "ex_nesting_level" is
446 * at or below the break level. But only when the line is actually
447 * executed. Return TRUE and set breakpoint_name for skipped commands that
448 * decide to execute something themselves.
449 * Called from do_one_cmd() before executing a command.
450 */
451 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100452dbg_check_breakpoint(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000453{
454 char_u *p;
455
456 debug_skipped = FALSE;
457 if (debug_breakpoint_name != NULL)
458 {
459 if (!eap->skip)
460 {
461 /* replace K_SNR with "<SNR>" */
462 if (debug_breakpoint_name[0] == K_SPECIAL
463 && debug_breakpoint_name[1] == KS_EXTRA
464 && debug_breakpoint_name[2] == (int)KE_SNR)
465 p = (char_u *)"<SNR>";
466 else
467 p = (char_u *)"";
Bram Moolenaar555b2802005-05-19 21:08:39 +0000468 smsg((char_u *)_("Breakpoint in \"%s%s\" line %ld"),
469 p,
Bram Moolenaar071d4272004-06-13 20:20:40 +0000470 debug_breakpoint_name + (*p == NUL ? 0 : 3),
471 (long)debug_breakpoint_lnum);
472 debug_breakpoint_name = NULL;
473 do_debug(eap->cmd);
474 }
475 else
476 {
477 debug_skipped = TRUE;
478 debug_skipped_name = debug_breakpoint_name;
479 debug_breakpoint_name = NULL;
480 }
481 }
482 else if (ex_nesting_level <= debug_break_level)
483 {
484 if (!eap->skip)
485 do_debug(eap->cmd);
486 else
487 {
488 debug_skipped = TRUE;
489 debug_skipped_name = NULL;
490 }
491 }
492}
493
494/*
495 * Go to debug mode if skipped by dbg_check_breakpoint() because eap->skip was
496 * set. Return TRUE when the debug mode is entered this time.
497 */
498 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100499dbg_check_skipped(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000500{
501 int prev_got_int;
502
503 if (debug_skipped)
504 {
505 /*
506 * Save the value of got_int and reset it. We don't want a previous
507 * interruption cause flushing the input buffer.
508 */
509 prev_got_int = got_int;
510 got_int = FALSE;
511 debug_breakpoint_name = debug_skipped_name;
512 /* eap->skip is TRUE */
513 eap->skip = FALSE;
514 (void)dbg_check_breakpoint(eap);
515 eap->skip = TRUE;
516 got_int |= prev_got_int;
517 return TRUE;
518 }
519 return FALSE;
520}
521
522/*
523 * The list of breakpoints: dbg_breakp.
524 * This is a grow-array of structs.
525 */
526struct debuggy
527{
528 int dbg_nr; /* breakpoint number */
529 int dbg_type; /* DBG_FUNC or DBG_FILE */
530 char_u *dbg_name; /* function or file name */
531 regprog_T *dbg_prog; /* regexp program */
532 linenr_T dbg_lnum; /* line number in function or file */
Bram Moolenaar05159a02005-02-26 23:04:13 +0000533 int dbg_forceit; /* ! used */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000534};
535
536static garray_T dbg_breakp = {0, 0, sizeof(struct debuggy), 4, NULL};
Bram Moolenaar05159a02005-02-26 23:04:13 +0000537#define BREAKP(idx) (((struct debuggy *)dbg_breakp.ga_data)[idx])
538#define DEBUGGY(gap, idx) (((struct debuggy *)gap->ga_data)[idx])
Bram Moolenaar071d4272004-06-13 20:20:40 +0000539static int last_breakp = 0; /* nr of last defined breakpoint */
540
Bram Moolenaar05159a02005-02-26 23:04:13 +0000541#ifdef FEAT_PROFILE
542/* Profiling uses file and func names similar to breakpoints. */
543static garray_T prof_ga = {0, 0, sizeof(struct debuggy), 4, NULL};
544#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000545#define DBG_FUNC 1
546#define DBG_FILE 2
547
Bram Moolenaarf28dbce2016-01-29 22:03:47 +0100548static int dbg_parsearg(char_u *arg, garray_T *gap);
549static linenr_T debuggy_find(int file,char_u *fname, linenr_T after, garray_T *gap, int *fp);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000550
551/*
Bram Moolenaar05159a02005-02-26 23:04:13 +0000552 * Parse the arguments of ":profile", ":breakadd" or ":breakdel" and put them
553 * in the entry just after the last one in dbg_breakp. Note that "dbg_name"
554 * is allocated.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000555 * Returns FAIL for failure.
556 */
557 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100558dbg_parsearg(
559 char_u *arg,
560 garray_T *gap) /* either &dbg_breakp or &prof_ga */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000561{
562 char_u *p = arg;
563 char_u *q;
564 struct debuggy *bp;
Bram Moolenaarf4b8e572004-06-24 15:53:16 +0000565 int here = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000566
Bram Moolenaar05159a02005-02-26 23:04:13 +0000567 if (ga_grow(gap, 1) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000568 return FAIL;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000569 bp = &DEBUGGY(gap, gap->ga_len);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000570
571 /* Find "func" or "file". */
572 if (STRNCMP(p, "func", 4) == 0)
573 bp->dbg_type = DBG_FUNC;
574 else if (STRNCMP(p, "file", 4) == 0)
575 bp->dbg_type = DBG_FILE;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000576 else if (
577#ifdef FEAT_PROFILE
578 gap != &prof_ga &&
579#endif
580 STRNCMP(p, "here", 4) == 0)
Bram Moolenaarf4b8e572004-06-24 15:53:16 +0000581 {
582 if (curbuf->b_ffname == NULL)
583 {
584 EMSG(_(e_noname));
585 return FAIL;
586 }
587 bp->dbg_type = DBG_FILE;
588 here = TRUE;
589 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000590 else
591 {
592 EMSG2(_(e_invarg2), p);
593 return FAIL;
594 }
595 p = skipwhite(p + 4);
596
597 /* Find optional line number. */
Bram Moolenaarf4b8e572004-06-24 15:53:16 +0000598 if (here)
599 bp->dbg_lnum = curwin->w_cursor.lnum;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000600 else if (
601#ifdef FEAT_PROFILE
602 gap != &prof_ga &&
603#endif
604 VIM_ISDIGIT(*p))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000605 {
606 bp->dbg_lnum = getdigits(&p);
607 p = skipwhite(p);
608 }
609 else
610 bp->dbg_lnum = 0;
611
612 /* Find the function or file name. Don't accept a function name with (). */
Bram Moolenaarf4b8e572004-06-24 15:53:16 +0000613 if ((!here && *p == NUL)
614 || (here && *p != NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000615 || (bp->dbg_type == DBG_FUNC && strstr((char *)p, "()") != NULL))
616 {
617 EMSG2(_(e_invarg2), arg);
618 return FAIL;
619 }
620
621 if (bp->dbg_type == DBG_FUNC)
622 bp->dbg_name = vim_strsave(p);
Bram Moolenaarf4b8e572004-06-24 15:53:16 +0000623 else if (here)
624 bp->dbg_name = vim_strsave(curbuf->b_ffname);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000625 else
626 {
627 /* Expand the file name in the same way as do_source(). This means
628 * doing it twice, so that $DIR/file gets expanded when $DIR is
629 * "~/dir". */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000630 q = expand_env_save(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000631 if (q == NULL)
632 return FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000633 p = expand_env_save(q);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000634 vim_free(q);
635 if (p == NULL)
636 return FAIL;
Bram Moolenaar843ee412004-06-30 16:16:41 +0000637 if (*p != '*')
638 {
639 bp->dbg_name = fix_fname(p);
640 vim_free(p);
641 }
642 else
643 bp->dbg_name = p;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000644 }
645
646 if (bp->dbg_name == NULL)
647 return FAIL;
648 return OK;
649}
650
651/*
652 * ":breakadd".
653 */
654 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100655ex_breakadd(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000656{
657 struct debuggy *bp;
658 char_u *pat;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000659 garray_T *gap;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000660
Bram Moolenaar05159a02005-02-26 23:04:13 +0000661 gap = &dbg_breakp;
662#ifdef FEAT_PROFILE
663 if (eap->cmdidx == CMD_profile)
664 gap = &prof_ga;
665#endif
666
667 if (dbg_parsearg(eap->arg, gap) == OK)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000668 {
Bram Moolenaar05159a02005-02-26 23:04:13 +0000669 bp = &DEBUGGY(gap, gap->ga_len);
670 bp->dbg_forceit = eap->forceit;
671
Bram Moolenaar071d4272004-06-13 20:20:40 +0000672 pat = file_pat_to_reg_pat(bp->dbg_name, NULL, NULL, FALSE);
673 if (pat != NULL)
674 {
675 bp->dbg_prog = vim_regcomp(pat, RE_MAGIC + RE_STRING);
676 vim_free(pat);
677 }
678 if (pat == NULL || bp->dbg_prog == NULL)
679 vim_free(bp->dbg_name);
680 else
681 {
682 if (bp->dbg_lnum == 0) /* default line number is 1 */
683 bp->dbg_lnum = 1;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000684#ifdef FEAT_PROFILE
685 if (eap->cmdidx != CMD_profile)
686#endif
687 {
688 DEBUGGY(gap, gap->ga_len).dbg_nr = ++last_breakp;
689 ++debug_tick;
690 }
691 ++gap->ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000692 }
693 }
694}
695
696/*
697 * ":debuggreedy".
698 */
699 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100700ex_debuggreedy(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000701{
702 if (eap->addr_count == 0 || eap->line2 != 0)
703 debug_greedy = TRUE;
704 else
705 debug_greedy = FALSE;
706}
707
708/*
Bram Moolenaard9fba312005-06-26 22:34:35 +0000709 * ":breakdel" and ":profdel".
Bram Moolenaar071d4272004-06-13 20:20:40 +0000710 */
711 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100712ex_breakdel(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000713{
714 struct debuggy *bp, *bpi;
715 int nr;
716 int todel = -1;
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000717 int del_all = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000718 int i;
719 linenr_T best_lnum = 0;
Bram Moolenaard9fba312005-06-26 22:34:35 +0000720 garray_T *gap;
721
722 gap = &dbg_breakp;
Bram Moolenaard9fba312005-06-26 22:34:35 +0000723 if (eap->cmdidx == CMD_profdel)
Bram Moolenaar38bdbd62012-06-20 15:48:57 +0200724 {
725#ifdef FEAT_PROFILE
Bram Moolenaard9fba312005-06-26 22:34:35 +0000726 gap = &prof_ga;
Bram Moolenaar38bdbd62012-06-20 15:48:57 +0200727#else
728 ex_ni(eap);
729 return;
Bram Moolenaard9fba312005-06-26 22:34:35 +0000730#endif
Bram Moolenaar38bdbd62012-06-20 15:48:57 +0200731 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000732
733 if (vim_isdigit(*eap->arg))
734 {
735 /* ":breakdel {nr}" */
736 nr = atol((char *)eap->arg);
Bram Moolenaard9fba312005-06-26 22:34:35 +0000737 for (i = 0; i < gap->ga_len; ++i)
738 if (DEBUGGY(gap, i).dbg_nr == nr)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000739 {
740 todel = i;
741 break;
742 }
743 }
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000744 else if (*eap->arg == '*')
745 {
746 todel = 0;
747 del_all = TRUE;
748 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000749 else
750 {
751 /* ":breakdel {func|file} [lnum] {name}" */
Bram Moolenaard9fba312005-06-26 22:34:35 +0000752 if (dbg_parsearg(eap->arg, gap) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000753 return;
Bram Moolenaard9fba312005-06-26 22:34:35 +0000754 bp = &DEBUGGY(gap, gap->ga_len);
755 for (i = 0; i < gap->ga_len; ++i)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000756 {
Bram Moolenaard9fba312005-06-26 22:34:35 +0000757 bpi = &DEBUGGY(gap, i);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000758 if (bp->dbg_type == bpi->dbg_type
759 && STRCMP(bp->dbg_name, bpi->dbg_name) == 0
760 && (bp->dbg_lnum == bpi->dbg_lnum
761 || (bp->dbg_lnum == 0
762 && (best_lnum == 0
763 || bpi->dbg_lnum < best_lnum))))
764 {
765 todel = i;
766 best_lnum = bpi->dbg_lnum;
767 }
768 }
769 vim_free(bp->dbg_name);
770 }
771
772 if (todel < 0)
773 EMSG2(_("E161: Breakpoint not found: %s"), eap->arg);
774 else
Bram Moolenaard9fba312005-06-26 22:34:35 +0000775 {
776 while (gap->ga_len > 0)
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000777 {
Bram Moolenaard9fba312005-06-26 22:34:35 +0000778 vim_free(DEBUGGY(gap, todel).dbg_name);
Bram Moolenaar473de612013-06-08 18:19:48 +0200779 vim_regfree(DEBUGGY(gap, todel).dbg_prog);
Bram Moolenaard9fba312005-06-26 22:34:35 +0000780 --gap->ga_len;
781 if (todel < gap->ga_len)
782 mch_memmove(&DEBUGGY(gap, todel), &DEBUGGY(gap, todel + 1),
783 (gap->ga_len - todel) * sizeof(struct debuggy));
784#ifdef FEAT_PROFILE
785 if (eap->cmdidx == CMD_breakdel)
786#endif
787 ++debug_tick;
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000788 if (!del_all)
789 break;
790 }
Bram Moolenaard9fba312005-06-26 22:34:35 +0000791
792 /* If all breakpoints were removed clear the array. */
793 if (gap->ga_len == 0)
794 ga_clear(gap);
795 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000796}
797
798/*
799 * ":breaklist".
800 */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000801 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100802ex_breaklist(exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000803{
804 struct debuggy *bp;
805 int i;
806
807 if (dbg_breakp.ga_len == 0)
808 MSG(_("No breakpoints defined"));
809 else
810 for (i = 0; i < dbg_breakp.ga_len; ++i)
811 {
812 bp = &BREAKP(i);
Bram Moolenaard58ea072011-06-26 04:25:30 +0200813 if (bp->dbg_type == DBG_FILE)
814 home_replace(NULL, bp->dbg_name, NameBuff, MAXPATHL, TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000815 smsg((char_u *)_("%3d %s %s line %ld"),
816 bp->dbg_nr,
817 bp->dbg_type == DBG_FUNC ? "func" : "file",
Bram Moolenaard58ea072011-06-26 04:25:30 +0200818 bp->dbg_type == DBG_FUNC ? bp->dbg_name : NameBuff,
Bram Moolenaar071d4272004-06-13 20:20:40 +0000819 (long)bp->dbg_lnum);
820 }
821}
822
823/*
824 * Find a breakpoint for a function or sourced file.
825 * Returns line number at which to break; zero when no matching breakpoint.
826 */
827 linenr_T
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100828dbg_find_breakpoint(
829 int file, /* TRUE for a file, FALSE for a function */
830 char_u *fname, /* file or function name */
831 linenr_T after) /* after this line number */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000832{
Bram Moolenaar05159a02005-02-26 23:04:13 +0000833 return debuggy_find(file, fname, after, &dbg_breakp, NULL);
834}
835
836#if defined(FEAT_PROFILE) || defined(PROTO)
837/*
838 * Return TRUE if profiling is on for a function or sourced file.
839 */
840 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100841has_profiling(
842 int file, /* TRUE for a file, FALSE for a function */
843 char_u *fname, /* file or function name */
844 int *fp) /* return: forceit */
Bram Moolenaar05159a02005-02-26 23:04:13 +0000845{
846 return (debuggy_find(file, fname, (linenr_T)0, &prof_ga, fp)
847 != (linenr_T)0);
848}
849#endif
850
851/*
852 * Common code for dbg_find_breakpoint() and has_profiling().
853 */
854 static linenr_T
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100855debuggy_find(
856 int file, /* TRUE for a file, FALSE for a function */
857 char_u *fname, /* file or function name */
858 linenr_T after, /* after this line number */
859 garray_T *gap, /* either &dbg_breakp or &prof_ga */
860 int *fp) /* if not NULL: return forceit */
Bram Moolenaar05159a02005-02-26 23:04:13 +0000861{
Bram Moolenaar071d4272004-06-13 20:20:40 +0000862 struct debuggy *bp;
863 int i;
864 linenr_T lnum = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000865 char_u *name = fname;
866 int prev_got_int;
867
Bram Moolenaar05159a02005-02-26 23:04:13 +0000868 /* Return quickly when there are no breakpoints. */
869 if (gap->ga_len == 0)
870 return (linenr_T)0;
871
Bram Moolenaar071d4272004-06-13 20:20:40 +0000872 /* Replace K_SNR in function name with "<SNR>". */
873 if (!file && fname[0] == K_SPECIAL)
874 {
875 name = alloc((unsigned)STRLEN(fname) + 3);
876 if (name == NULL)
877 name = fname;
878 else
879 {
880 STRCPY(name, "<SNR>");
881 STRCPY(name + 5, fname + 3);
882 }
883 }
884
Bram Moolenaar05159a02005-02-26 23:04:13 +0000885 for (i = 0; i < gap->ga_len; ++i)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000886 {
Bram Moolenaar05159a02005-02-26 23:04:13 +0000887 /* Skip entries that are not useful or are for a line that is beyond
888 * an already found breakpoint. */
889 bp = &DEBUGGY(gap, i);
890 if (((bp->dbg_type == DBG_FILE) == file && (
891#ifdef FEAT_PROFILE
892 gap == &prof_ga ||
893#endif
894 (bp->dbg_lnum > after && (lnum == 0 || bp->dbg_lnum < lnum)))))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000895 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000896 /*
Bram Moolenaar05159a02005-02-26 23:04:13 +0000897 * Save the value of got_int and reset it. We don't want a
898 * previous interruption cancel matching, only hitting CTRL-C
899 * while matching should abort it.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000900 */
901 prev_got_int = got_int;
902 got_int = FALSE;
Bram Moolenaardffa5b82014-11-19 16:38:07 +0100903 if (vim_regexec_prog(&bp->dbg_prog, FALSE, name, (colnr_T)0))
Bram Moolenaar05159a02005-02-26 23:04:13 +0000904 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000905 lnum = bp->dbg_lnum;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000906 if (fp != NULL)
907 *fp = bp->dbg_forceit;
908 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000909 got_int |= prev_got_int;
910 }
911 }
912 if (name != fname)
913 vim_free(name);
914
915 return lnum;
916}
917
918/*
919 * Called when a breakpoint was encountered.
920 */
921 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100922dbg_breakpoint(char_u *name, linenr_T lnum)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000923{
924 /* We need to check if this line is actually executed in do_one_cmd() */
925 debug_breakpoint_name = name;
926 debug_breakpoint_lnum = lnum;
927}
Bram Moolenaar05159a02005-02-26 23:04:13 +0000928
929
Bram Moolenaar433f7c82006-03-21 21:29:36 +0000930# if defined(FEAT_PROFILE) || defined(FEAT_RELTIME) || defined(PROTO)
Bram Moolenaar05159a02005-02-26 23:04:13 +0000931/*
932 * Store the current time in "tm".
933 */
934 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100935profile_start(proftime_T *tm)
Bram Moolenaar05159a02005-02-26 23:04:13 +0000936{
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000937# ifdef WIN3264
938 QueryPerformanceCounter(tm);
939# else
Bram Moolenaar05159a02005-02-26 23:04:13 +0000940 gettimeofday(tm, NULL);
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000941# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +0000942}
943
944/*
945 * Compute the elapsed time from "tm" till now and store in "tm".
946 */
947 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100948profile_end(proftime_T *tm)
Bram Moolenaar05159a02005-02-26 23:04:13 +0000949{
950 proftime_T now;
951
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000952# ifdef WIN3264
953 QueryPerformanceCounter(&now);
954 tm->QuadPart = now.QuadPart - tm->QuadPart;
955# else
Bram Moolenaar05159a02005-02-26 23:04:13 +0000956 gettimeofday(&now, NULL);
957 tm->tv_usec = now.tv_usec - tm->tv_usec;
958 tm->tv_sec = now.tv_sec - tm->tv_sec;
959 if (tm->tv_usec < 0)
960 {
961 tm->tv_usec += 1000000;
962 --tm->tv_sec;
963 }
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000964# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +0000965}
966
967/*
968 * Subtract the time "tm2" from "tm".
969 */
970 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100971profile_sub(proftime_T *tm, proftime_T *tm2)
Bram Moolenaar05159a02005-02-26 23:04:13 +0000972{
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000973# ifdef WIN3264
974 tm->QuadPart -= tm2->QuadPart;
975# else
Bram Moolenaar05159a02005-02-26 23:04:13 +0000976 tm->tv_usec -= tm2->tv_usec;
977 tm->tv_sec -= tm2->tv_sec;
978 if (tm->tv_usec < 0)
979 {
980 tm->tv_usec += 1000000;
981 --tm->tv_sec;
982 }
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000983# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +0000984}
985
986/*
Bram Moolenaar433f7c82006-03-21 21:29:36 +0000987 * Return a string that represents the time in "tm".
988 * Uses a static buffer!
989 */
990 char *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100991profile_msg(proftime_T *tm)
Bram Moolenaar433f7c82006-03-21 21:29:36 +0000992{
993 static char buf[50];
994
995# ifdef WIN3264
996 LARGE_INTEGER fr;
997
998 QueryPerformanceFrequency(&fr);
999 sprintf(buf, "%10.6lf", (double)tm->QuadPart / (double)fr.QuadPart);
1000# else
1001 sprintf(buf, "%3ld.%06ld", (long)tm->tv_sec, (long)tm->tv_usec);
Bram Moolenaar76929292008-01-06 19:07:36 +00001002# endif
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001003 return buf;
1004}
1005
Bram Moolenaar79c2c882016-02-07 21:19:28 +01001006# if defined(FEAT_FLOAT) || defined(PROTO)
1007/*
1008 * Return a float that represents the time in "tm".
1009 */
1010 float_T
1011profile_float(proftime_T *tm)
1012{
1013# ifdef WIN3264
1014 LARGE_INTEGER fr;
1015
1016 QueryPerformanceFrequency(&fr);
1017 return (float_T)tm->QuadPart / (float_T)fr.QuadPart;
1018# else
1019 return (float_T)tm->tv_sec + (float_T)tm->tv_usec / 1000000.0;
1020# endif
1021}
1022# endif
1023
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001024/*
Bram Moolenaar76929292008-01-06 19:07:36 +00001025 * Put the time "msec" past now in "tm".
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001026 */
Bram Moolenaar76929292008-01-06 19:07:36 +00001027 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001028profile_setlimit(long msec, proftime_T *tm)
Bram Moolenaar76929292008-01-06 19:07:36 +00001029{
1030 if (msec <= 0) /* no limit */
1031 profile_zero(tm);
1032 else
1033 {
1034# ifdef WIN3264
1035 LARGE_INTEGER fr;
1036
1037 QueryPerformanceCounter(tm);
1038 QueryPerformanceFrequency(&fr);
Bram Moolenaarb3c70982008-01-18 10:40:55 +00001039 tm->QuadPart += (LONGLONG)((double)msec / 1000.0 * (double)fr.QuadPart);
Bram Moolenaar76929292008-01-06 19:07:36 +00001040# else
1041 long usec;
1042
1043 gettimeofday(tm, NULL);
1044 usec = (long)tm->tv_usec + (long)msec * 1000;
1045 tm->tv_usec = usec % 1000000L;
1046 tm->tv_sec += usec / 1000000L;
1047# endif
1048 }
1049}
1050
1051/*
1052 * Return TRUE if the current time is past "tm".
1053 */
1054 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001055profile_passed_limit(proftime_T *tm)
Bram Moolenaar76929292008-01-06 19:07:36 +00001056{
1057 proftime_T now;
1058
1059# ifdef WIN3264
1060 if (tm->QuadPart == 0) /* timer was not set */
1061 return FALSE;
1062 QueryPerformanceCounter(&now);
1063 return (now.QuadPart > tm->QuadPart);
1064# else
1065 if (tm->tv_sec == 0) /* timer was not set */
1066 return FALSE;
1067 gettimeofday(&now, NULL);
1068 return (now.tv_sec > tm->tv_sec
1069 || (now.tv_sec == tm->tv_sec && now.tv_usec > tm->tv_usec));
1070# endif
1071}
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001072
1073/*
1074 * Set the time in "tm" to zero.
1075 */
1076 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001077profile_zero(proftime_T *tm)
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001078{
1079# ifdef WIN3264
1080 tm->QuadPart = 0;
1081# else
1082 tm->tv_usec = 0;
1083 tm->tv_sec = 0;
1084# endif
1085}
1086
Bram Moolenaar76929292008-01-06 19:07:36 +00001087# endif /* FEAT_PROFILE || FEAT_RELTIME */
1088
Bram Moolenaar975b5272016-03-15 23:10:59 +01001089# if defined(FEAT_TIMERS) || defined(PROTO)
1090static timer_T *first_timer = NULL;
Bram Moolenaar75537a92016-09-05 22:45:28 +02001091static long last_timer_id = 0;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001092
Bram Moolenaar75537a92016-09-05 22:45:28 +02001093# ifdef WIN3264
1094# define GET_TIMEDIFF(timer, now) \
1095 (long)(((double)(timer->tr_due.QuadPart - now.QuadPart) \
Bram Moolenaara7c023e2016-10-12 12:13:35 +02001096 / (double)fr.QuadPart) * 1000)
Bram Moolenaar75537a92016-09-05 22:45:28 +02001097# else
1098# define GET_TIMEDIFF(timer, now) \
1099 (timer->tr_due.tv_sec - now.tv_sec) * 1000 \
Bram Moolenaara7c023e2016-10-12 12:13:35 +02001100 + (timer->tr_due.tv_usec - now.tv_usec) / 1000
Bram Moolenaar75537a92016-09-05 22:45:28 +02001101# endif
Bram Moolenaar417ccd72016-09-01 21:26:20 +02001102
Bram Moolenaar975b5272016-03-15 23:10:59 +01001103/*
1104 * Insert a timer in the list of timers.
1105 */
1106 static void
1107insert_timer(timer_T *timer)
1108{
1109 timer->tr_next = first_timer;
1110 timer->tr_prev = NULL;
1111 if (first_timer != NULL)
1112 first_timer->tr_prev = timer;
1113 first_timer = timer;
Bram Moolenaar4231da42016-06-02 14:30:04 +02001114 did_add_timer = TRUE;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001115}
1116
1117/*
1118 * Take a timer out of the list of timers.
1119 */
1120 static void
1121remove_timer(timer_T *timer)
1122{
1123 if (timer->tr_prev == NULL)
1124 first_timer = timer->tr_next;
1125 else
1126 timer->tr_prev->tr_next = timer->tr_next;
1127 if (timer->tr_next != NULL)
1128 timer->tr_next->tr_prev = timer->tr_prev;
1129}
1130
1131 static void
1132free_timer(timer_T *timer)
1133{
Bram Moolenaar75537a92016-09-05 22:45:28 +02001134 free_callback(timer->tr_callback, timer->tr_partial);
1135 vim_free(timer);
Bram Moolenaar975b5272016-03-15 23:10:59 +01001136}
1137
1138/*
1139 * Create a timer and return it. NULL if out of memory.
1140 * Caller should set the callback.
1141 */
1142 timer_T *
1143create_timer(long msec, int repeat)
1144{
1145 timer_T *timer = (timer_T *)alloc_clear(sizeof(timer_T));
Bram Moolenaaree39ef02016-09-10 19:17:42 +02001146 long prev_id = last_timer_id;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001147
1148 if (timer == NULL)
1149 return NULL;
Bram Moolenaaree39ef02016-09-10 19:17:42 +02001150 if (++last_timer_id <= prev_id)
Bram Moolenaar75537a92016-09-05 22:45:28 +02001151 /* Overflow! Might cause duplicates... */
1152 last_timer_id = 0;
1153 timer->tr_id = last_timer_id;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001154 insert_timer(timer);
1155 if (repeat != 0)
Bram Moolenaar975b5272016-03-15 23:10:59 +01001156 timer->tr_repeat = repeat - 1;
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001157 timer->tr_interval = msec;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001158
1159 profile_setlimit(msec, &timer->tr_due);
1160 return timer;
1161}
1162
1163/*
1164 * Invoke the callback of "timer".
1165 */
1166 static void
1167timer_callback(timer_T *timer)
1168{
1169 typval_T rettv;
1170 int dummy;
1171 typval_T argv[2];
1172
1173 argv[0].v_type = VAR_NUMBER;
Bram Moolenaar75537a92016-09-05 22:45:28 +02001174 argv[0].vval.v_number = (varnumber_T)timer->tr_id;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001175 argv[1].v_type = VAR_UNKNOWN;
1176
1177 call_func(timer->tr_callback, (int)STRLEN(timer->tr_callback),
Bram Moolenaardf48fb42016-07-22 21:50:18 +02001178 &rettv, 1, argv, NULL, 0L, 0L, &dummy, TRUE,
Bram Moolenaar975b5272016-03-15 23:10:59 +01001179 timer->tr_partial, NULL);
1180 clear_tv(&rettv);
1181}
1182
1183/*
1184 * Call timers that are due.
1185 * Return the time in msec until the next timer is due.
1186 */
1187 long
Bram Moolenaarcf089462016-06-12 21:18:43 +02001188check_due_timer(void)
Bram Moolenaar975b5272016-03-15 23:10:59 +01001189{
1190 timer_T *timer;
Bram Moolenaar75537a92016-09-05 22:45:28 +02001191 timer_T *timer_next;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001192 long this_due;
Bram Moolenaar597385a2016-03-16 23:24:43 +01001193 long next_due = -1;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001194 proftime_T now;
1195 int did_one = FALSE;
Bram Moolenaar75537a92016-09-05 22:45:28 +02001196 long current_id = last_timer_id;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001197# ifdef WIN3264
1198 LARGE_INTEGER fr;
1199
1200 QueryPerformanceFrequency(&fr);
1201# endif
Bram Moolenaar75537a92016-09-05 22:45:28 +02001202 profile_start(&now);
1203 for (timer = first_timer; timer != NULL && !got_int; timer = timer_next)
Bram Moolenaar975b5272016-03-15 23:10:59 +01001204 {
Bram Moolenaar75537a92016-09-05 22:45:28 +02001205 timer_next = timer->tr_next;
Bram Moolenaar417ccd72016-09-01 21:26:20 +02001206
Bram Moolenaar75537a92016-09-05 22:45:28 +02001207 if (timer->tr_id == -1 || timer->tr_firing || timer->tr_paused)
1208 continue;
1209 this_due = GET_TIMEDIFF(timer, now);
1210 if (this_due <= 1)
1211 {
1212 timer->tr_firing = TRUE;
1213 timer_callback(timer);
1214 timer->tr_firing = FALSE;
1215 timer_next = timer->tr_next;
1216 did_one = TRUE;
1217
1218 /* Only fire the timer again if it repeats and stop_timer() wasn't
1219 * called while inside the callback (tr_id == -1). */
1220 if (timer->tr_repeat != 0 && timer->tr_id != -1)
1221 {
1222 profile_setlimit(timer->tr_interval, &timer->tr_due);
1223 this_due = GET_TIMEDIFF(timer, now);
1224 if (this_due < 1)
1225 this_due = 1;
1226 if (timer->tr_repeat > 0)
1227 --timer->tr_repeat;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001228 }
Bram Moolenaar75537a92016-09-05 22:45:28 +02001229 else
1230 {
1231 this_due = -1;
1232 remove_timer(timer);
1233 free_timer(timer);
1234 }
Bram Moolenaar975b5272016-03-15 23:10:59 +01001235 }
Bram Moolenaar75537a92016-09-05 22:45:28 +02001236 if (this_due > 0 && (next_due == -1 || next_due > this_due))
1237 next_due = this_due;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001238 }
1239
1240 if (did_one)
1241 redraw_after_callback();
1242
Bram Moolenaar75537a92016-09-05 22:45:28 +02001243 return current_id != last_timer_id ? 1 : next_due;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001244}
1245
1246/*
1247 * Find a timer by ID. Returns NULL if not found;
1248 */
1249 timer_T *
Bram Moolenaar75537a92016-09-05 22:45:28 +02001250find_timer(long id)
Bram Moolenaar975b5272016-03-15 23:10:59 +01001251{
1252 timer_T *timer;
1253
Bram Moolenaar75537a92016-09-05 22:45:28 +02001254 if (id >= 0)
1255 {
1256 for (timer = first_timer; timer != NULL; timer = timer->tr_next)
1257 if (timer->tr_id == id)
1258 return timer;
1259 }
1260 return NULL;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001261}
1262
1263
1264/*
1265 * Stop a timer and delete it.
1266 */
1267 void
1268stop_timer(timer_T *timer)
1269{
Bram Moolenaar75537a92016-09-05 22:45:28 +02001270 if (timer->tr_firing)
1271 /* Free the timer after the callback returns. */
1272 timer->tr_id = -1;
1273 else
1274 {
1275 remove_timer(timer);
1276 free_timer(timer);
1277 }
Bram Moolenaar975b5272016-03-15 23:10:59 +01001278}
Bram Moolenaare3188e22016-05-31 21:13:04 +02001279
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001280 void
Bram Moolenaarb73598e2016-08-07 18:22:53 +02001281stop_all_timers(void)
1282{
Bram Moolenaar75537a92016-09-05 22:45:28 +02001283 timer_T *timer;
1284 timer_T *timer_next;
1285
1286 for (timer = first_timer; timer != NULL; timer = timer_next)
1287 {
1288 timer_next = timer->tr_next;
1289 stop_timer(timer);
1290 }
Bram Moolenaarb73598e2016-08-07 18:22:53 +02001291}
1292
1293 void
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001294add_timer_info(typval_T *rettv, timer_T *timer)
1295{
1296 list_T *list = rettv->vval.v_list;
1297 dict_T *dict = dict_alloc();
1298 dictitem_T *di;
1299 long remaining;
1300 proftime_T now;
Bram Moolenaar00ff3802016-08-06 22:27:28 +02001301# ifdef WIN3264
1302 LARGE_INTEGER fr;
1303#endif
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001304
1305 if (dict == NULL)
1306 return;
1307 list_append_dict(list, dict);
1308
Bram Moolenaar75537a92016-09-05 22:45:28 +02001309 dict_add_nr_str(dict, "id", timer->tr_id, NULL);
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001310 dict_add_nr_str(dict, "time", (long)timer->tr_interval, NULL);
1311
1312 profile_start(&now);
1313# ifdef WIN3264
Bram Moolenaar00ff3802016-08-06 22:27:28 +02001314 QueryPerformanceFrequency(&fr);
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001315# endif
Bram Moolenaar75537a92016-09-05 22:45:28 +02001316 remaining = GET_TIMEDIFF(timer, now);
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001317 dict_add_nr_str(dict, "remaining", (long)remaining, NULL);
1318
1319 dict_add_nr_str(dict, "repeat",
1320 (long)(timer->tr_repeat < 0 ? -1 : timer->tr_repeat + 1), NULL);
Bram Moolenaarb73598e2016-08-07 18:22:53 +02001321 dict_add_nr_str(dict, "paused", (long)(timer->tr_paused), NULL);
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001322
1323 di = dictitem_alloc((char_u *)"callback");
1324 if (di != NULL)
1325 {
1326 if (dict_add(dict, di) == FAIL)
1327 vim_free(di);
1328 else if (timer->tr_partial != NULL)
1329 {
1330 di->di_tv.v_type = VAR_PARTIAL;
1331 di->di_tv.vval.v_partial = timer->tr_partial;
1332 ++timer->tr_partial->pt_refcount;
1333 }
1334 else
1335 {
1336 di->di_tv.v_type = VAR_FUNC;
1337 di->di_tv.vval.v_string = vim_strsave(timer->tr_callback);
1338 }
1339 di->di_tv.v_lock = 0;
1340 }
1341}
1342
1343 void
1344add_timer_info_all(typval_T *rettv)
1345{
1346 timer_T *timer;
1347
1348 for (timer = first_timer; timer != NULL; timer = timer->tr_next)
Bram Moolenaar75537a92016-09-05 22:45:28 +02001349 if (timer->tr_id != -1)
1350 add_timer_info(rettv, timer);
Bram Moolenaar8e97bd72016-08-06 22:05:07 +02001351}
1352
Bram Moolenaare3188e22016-05-31 21:13:04 +02001353/*
1354 * Mark references in partials of timers.
1355 */
1356 int
1357set_ref_in_timer(int copyID)
1358{
1359 int abort = FALSE;
1360 timer_T *timer;
1361 typval_T tv;
1362
1363 for (timer = first_timer; timer != NULL; timer = timer->tr_next)
1364 {
Bram Moolenaar1e96d9b2016-07-29 22:15:09 +02001365 if (timer->tr_partial != NULL)
1366 {
1367 tv.v_type = VAR_PARTIAL;
1368 tv.vval.v_partial = timer->tr_partial;
1369 }
1370 else
1371 {
1372 tv.v_type = VAR_FUNC;
1373 tv.vval.v_string = timer->tr_callback;
1374 }
Bram Moolenaare3188e22016-05-31 21:13:04 +02001375 abort = abort || set_ref_in_item(&tv, copyID, NULL, NULL);
1376 }
1377 return abort;
1378}
Bram Moolenaar623e2632016-07-30 22:47:56 +02001379
1380# if defined(EXITFREE) || defined(PROTO)
1381 void
1382timer_free_all()
1383{
1384 timer_T *timer;
1385
1386 while (first_timer != NULL)
1387 {
1388 timer = first_timer;
1389 remove_timer(timer);
1390 free_timer(timer);
1391 }
1392}
1393# endif
Bram Moolenaar975b5272016-03-15 23:10:59 +01001394# endif
1395
Bram Moolenaar8a7f5a22013-06-06 14:01:46 +02001396#if defined(FEAT_SYN_HL) && defined(FEAT_RELTIME) && defined(FEAT_FLOAT)
1397# if defined(HAVE_MATH_H)
1398# include <math.h>
1399# endif
1400
1401/*
1402 * Divide the time "tm" by "count" and store in "tm2".
1403 */
1404 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001405profile_divide(proftime_T *tm, int count, proftime_T *tm2)
Bram Moolenaar8a7f5a22013-06-06 14:01:46 +02001406{
1407 if (count == 0)
1408 profile_zero(tm2);
1409 else
1410 {
1411# ifdef WIN3264
1412 tm2->QuadPart = tm->QuadPart / count;
1413# else
1414 double usec = (tm->tv_sec * 1000000.0 + tm->tv_usec) / count;
1415
1416 tm2->tv_sec = floor(usec / 1000000.0);
Bram Moolenaara2e14fc2013-06-10 20:10:44 +02001417 tm2->tv_usec = vim_round(usec - (tm2->tv_sec * 1000000.0));
Bram Moolenaar8a7f5a22013-06-06 14:01:46 +02001418# endif
1419 }
1420}
1421#endif
1422
Bram Moolenaar76929292008-01-06 19:07:36 +00001423# if defined(FEAT_PROFILE) || defined(PROTO)
1424/*
1425 * Functions for profiling.
1426 */
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01001427static void script_do_profile(scriptitem_T *si);
1428static void script_dump_profile(FILE *fd);
Bram Moolenaar76929292008-01-06 19:07:36 +00001429static proftime_T prof_wait_time;
1430
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001431/*
Bram Moolenaar05159a02005-02-26 23:04:13 +00001432 * Add the time "tm2" to "tm".
1433 */
1434 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001435profile_add(proftime_T *tm, proftime_T *tm2)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001436{
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001437# ifdef WIN3264
1438 tm->QuadPart += tm2->QuadPart;
1439# else
Bram Moolenaar05159a02005-02-26 23:04:13 +00001440 tm->tv_usec += tm2->tv_usec;
1441 tm->tv_sec += tm2->tv_sec;
1442 if (tm->tv_usec >= 1000000)
1443 {
1444 tm->tv_usec -= 1000000;
1445 ++tm->tv_sec;
1446 }
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001447# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00001448}
1449
1450/*
Bram Moolenaar1056d982006-03-09 22:37:52 +00001451 * Add the "self" time from the total time and the children's time.
1452 */
1453 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001454profile_self(proftime_T *self, proftime_T *total, proftime_T *children)
Bram Moolenaar1056d982006-03-09 22:37:52 +00001455{
1456 /* Check that the result won't be negative. Can happen with recursive
1457 * calls. */
1458#ifdef WIN3264
1459 if (total->QuadPart <= children->QuadPart)
1460 return;
1461#else
1462 if (total->tv_sec < children->tv_sec
1463 || (total->tv_sec == children->tv_sec
1464 && total->tv_usec <= children->tv_usec))
1465 return;
1466#endif
1467 profile_add(self, total);
1468 profile_sub(self, children);
1469}
1470
1471/*
Bram Moolenaar05159a02005-02-26 23:04:13 +00001472 * Get the current waittime.
1473 */
1474 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001475profile_get_wait(proftime_T *tm)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001476{
1477 *tm = prof_wait_time;
1478}
1479
1480/*
1481 * Subtract the passed waittime since "tm" from "tma".
1482 */
1483 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001484profile_sub_wait(proftime_T *tm, proftime_T *tma)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001485{
1486 proftime_T tm3 = prof_wait_time;
1487
1488 profile_sub(&tm3, tm);
1489 profile_sub(tma, &tm3);
1490}
1491
1492/*
1493 * Return TRUE if "tm1" and "tm2" are equal.
1494 */
1495 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001496profile_equal(proftime_T *tm1, proftime_T *tm2)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001497{
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001498# ifdef WIN3264
1499 return (tm1->QuadPart == tm2->QuadPart);
1500# else
Bram Moolenaar05159a02005-02-26 23:04:13 +00001501 return (tm1->tv_usec == tm2->tv_usec && tm1->tv_sec == tm2->tv_sec);
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001502# endif
1503}
1504
1505/*
1506 * Return <0, 0 or >0 if "tm1" < "tm2", "tm1" == "tm2" or "tm1" > "tm2"
1507 */
1508 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001509profile_cmp(const proftime_T *tm1, const proftime_T *tm2)
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001510{
1511# ifdef WIN3264
1512 return (int)(tm2->QuadPart - tm1->QuadPart);
1513# else
1514 if (tm1->tv_sec == tm2->tv_sec)
1515 return tm2->tv_usec - tm1->tv_usec;
1516 return tm2->tv_sec - tm1->tv_sec;
1517# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00001518}
1519
Bram Moolenaar05159a02005-02-26 23:04:13 +00001520static char_u *profile_fname = NULL;
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001521static proftime_T pause_time;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001522
1523/*
1524 * ":profile cmd args"
1525 */
1526 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001527ex_profile(exarg_T *eap)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001528{
1529 char_u *e;
1530 int len;
1531
1532 e = skiptowhite(eap->arg);
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00001533 len = (int)(e - eap->arg);
Bram Moolenaar05159a02005-02-26 23:04:13 +00001534 e = skipwhite(e);
1535
1536 if (len == 5 && STRNCMP(eap->arg, "start", 5) == 0 && *e != NUL)
1537 {
1538 vim_free(profile_fname);
Bram Moolenaard94682f2015-04-13 15:37:56 +02001539 profile_fname = expand_env_save_opt(e, TRUE);
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001540 do_profiling = PROF_YES;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001541 profile_zero(&prof_wait_time);
1542 set_vim_var_nr(VV_PROFILING, 1L);
1543 }
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001544 else if (do_profiling == PROF_NONE)
Bram Moolenaar54c1b492010-02-24 14:01:28 +01001545 EMSG(_("E750: First use \":profile start {fname}\""));
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001546 else if (STRCMP(eap->arg, "pause") == 0)
1547 {
1548 if (do_profiling == PROF_YES)
1549 profile_start(&pause_time);
1550 do_profiling = PROF_PAUSED;
1551 }
1552 else if (STRCMP(eap->arg, "continue") == 0)
1553 {
1554 if (do_profiling == PROF_PAUSED)
1555 {
1556 profile_end(&pause_time);
1557 profile_add(&prof_wait_time, &pause_time);
1558 }
1559 do_profiling = PROF_YES;
1560 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00001561 else
1562 {
1563 /* The rest is similar to ":breakadd". */
1564 ex_breakadd(eap);
1565 }
1566}
1567
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001568/* Command line expansion for :profile. */
1569static enum
1570{
1571 PEXP_SUBCMD, /* expand :profile sub-commands */
Bram Moolenaar49789dc2011-02-25 14:46:09 +01001572 PEXP_FUNC /* expand :profile func {funcname} */
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001573} pexpand_what;
1574
1575static char *pexpand_cmds[] = {
1576 "start",
1577#define PROFCMD_START 0
1578 "pause",
1579#define PROFCMD_PAUSE 1
1580 "continue",
1581#define PROFCMD_CONTINUE 2
1582 "func",
1583#define PROFCMD_FUNC 3
1584 "file",
1585#define PROFCMD_FILE 4
1586 NULL
1587#define PROFCMD_LAST 5
1588};
1589
1590/*
1591 * Function given to ExpandGeneric() to obtain the profile command
1592 * specific expansion.
1593 */
1594 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001595get_profile_name(expand_T *xp UNUSED, int idx)
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001596{
1597 switch (pexpand_what)
1598 {
1599 case PEXP_SUBCMD:
1600 return (char_u *)pexpand_cmds[idx];
1601 /* case PEXP_FUNC: TODO */
1602 default:
1603 return NULL;
1604 }
1605}
1606
1607/*
1608 * Handle command line completion for :profile command.
1609 */
1610 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001611set_context_in_profile_cmd(expand_T *xp, char_u *arg)
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001612{
1613 char_u *end_subcmd;
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001614
1615 /* Default: expand subcommands. */
1616 xp->xp_context = EXPAND_PROFILE;
1617 pexpand_what = PEXP_SUBCMD;
1618 xp->xp_pattern = arg;
1619
1620 end_subcmd = skiptowhite(arg);
1621 if (*end_subcmd == NUL)
1622 return;
1623
Bram Moolenaar8b9c05f2010-03-02 17:54:33 +01001624 if (end_subcmd - arg == 5 && STRNCMP(arg, "start", 5) == 0)
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001625 {
1626 xp->xp_context = EXPAND_FILES;
1627 xp->xp_pattern = skipwhite(end_subcmd);
1628 return;
1629 }
1630
1631 /* TODO: expand function names after "func" */
1632 xp->xp_context = EXPAND_NOTHING;
1633}
1634
Bram Moolenaar05159a02005-02-26 23:04:13 +00001635/*
1636 * Dump the profiling info.
1637 */
1638 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001639profile_dump(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001640{
1641 FILE *fd;
1642
1643 if (profile_fname != NULL)
1644 {
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +00001645 fd = mch_fopen((char *)profile_fname, "w");
Bram Moolenaar05159a02005-02-26 23:04:13 +00001646 if (fd == NULL)
1647 EMSG2(_(e_notopen), profile_fname);
1648 else
1649 {
Bram Moolenaar05159a02005-02-26 23:04:13 +00001650 script_dump_profile(fd);
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001651 func_dump_profile(fd);
Bram Moolenaar05159a02005-02-26 23:04:13 +00001652 fclose(fd);
1653 }
1654 }
1655}
1656
1657/*
1658 * Start profiling script "fp".
1659 */
1660 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001661script_do_profile(scriptitem_T *si)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001662{
1663 si->sn_pr_count = 0;
1664 profile_zero(&si->sn_pr_total);
1665 profile_zero(&si->sn_pr_self);
1666
1667 ga_init2(&si->sn_prl_ga, sizeof(sn_prl_T), 100);
1668 si->sn_prl_idx = -1;
1669 si->sn_prof_on = TRUE;
1670 si->sn_pr_nest = 0;
1671}
1672
1673/*
1674 * save time when starting to invoke another script or function.
1675 */
1676 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001677script_prof_save(
1678 proftime_T *tm) /* place to store wait time */
Bram Moolenaar05159a02005-02-26 23:04:13 +00001679{
1680 scriptitem_T *si;
1681
1682 if (current_SID > 0 && current_SID <= script_items.ga_len)
1683 {
1684 si = &SCRIPT_ITEM(current_SID);
1685 if (si->sn_prof_on && si->sn_pr_nest++ == 0)
1686 profile_start(&si->sn_pr_child);
1687 }
1688 profile_get_wait(tm);
1689}
1690
1691/*
1692 * Count time spent in children after invoking another script or function.
1693 */
1694 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001695script_prof_restore(proftime_T *tm)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001696{
1697 scriptitem_T *si;
1698
1699 if (current_SID > 0 && current_SID <= script_items.ga_len)
1700 {
1701 si = &SCRIPT_ITEM(current_SID);
1702 if (si->sn_prof_on && --si->sn_pr_nest == 0)
1703 {
1704 profile_end(&si->sn_pr_child);
1705 profile_sub_wait(tm, &si->sn_pr_child); /* don't count wait time */
1706 profile_add(&si->sn_pr_children, &si->sn_pr_child);
1707 profile_add(&si->sn_prl_children, &si->sn_pr_child);
1708 }
1709 }
1710}
1711
1712static proftime_T inchar_time;
1713
1714/*
1715 * Called when starting to wait for the user to type a character.
1716 */
1717 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001718prof_inchar_enter(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001719{
1720 profile_start(&inchar_time);
1721}
1722
1723/*
1724 * Called when finished waiting for the user to type a character.
1725 */
1726 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001727prof_inchar_exit(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001728{
1729 profile_end(&inchar_time);
1730 profile_add(&prof_wait_time, &inchar_time);
1731}
1732
1733/*
1734 * Dump the profiling results for all scripts in file "fd".
1735 */
1736 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001737script_dump_profile(FILE *fd)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001738{
1739 int id;
1740 scriptitem_T *si;
1741 int i;
1742 FILE *sfd;
1743 sn_prl_T *pp;
1744
1745 for (id = 1; id <= script_items.ga_len; ++id)
1746 {
1747 si = &SCRIPT_ITEM(id);
1748 if (si->sn_prof_on)
1749 {
1750 fprintf(fd, "SCRIPT %s\n", si->sn_name);
1751 if (si->sn_pr_count == 1)
1752 fprintf(fd, "Sourced 1 time\n");
1753 else
1754 fprintf(fd, "Sourced %d times\n", si->sn_pr_count);
1755 fprintf(fd, "Total time: %s\n", profile_msg(&si->sn_pr_total));
1756 fprintf(fd, " Self time: %s\n", profile_msg(&si->sn_pr_self));
1757 fprintf(fd, "\n");
1758 fprintf(fd, "count total (s) self (s)\n");
1759
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +00001760 sfd = mch_fopen((char *)si->sn_name, "r");
Bram Moolenaar05159a02005-02-26 23:04:13 +00001761 if (sfd == NULL)
1762 fprintf(fd, "Cannot open file!\n");
1763 else
1764 {
1765 for (i = 0; i < si->sn_prl_ga.ga_len; ++i)
1766 {
1767 if (vim_fgets(IObuff, IOSIZE, sfd))
1768 break;
1769 pp = &PRL_ITEM(si, i);
1770 if (pp->snp_count > 0)
1771 {
1772 fprintf(fd, "%5d ", pp->snp_count);
1773 if (profile_equal(&pp->sn_prl_total, &pp->sn_prl_self))
1774 fprintf(fd, " ");
1775 else
1776 fprintf(fd, "%s ", profile_msg(&pp->sn_prl_total));
1777 fprintf(fd, "%s ", profile_msg(&pp->sn_prl_self));
1778 }
1779 else
1780 fprintf(fd, " ");
1781 fprintf(fd, "%s", IObuff);
1782 }
1783 fclose(sfd);
1784 }
1785 fprintf(fd, "\n");
1786 }
1787 }
1788}
1789
1790/*
1791 * Return TRUE when a function defined in the current script should be
1792 * profiled.
1793 */
1794 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001795prof_def_func(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001796{
Bram Moolenaar53180ce2005-07-05 21:48:14 +00001797 if (current_SID > 0)
1798 return SCRIPT_ITEM(current_SID).sn_pr_force;
1799 return FALSE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001800}
1801
1802# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001803#endif
1804
1805/*
1806 * If 'autowrite' option set, try to write the file.
1807 * Careful: autocommands may make "buf" invalid!
1808 *
1809 * return FAIL for failure, OK otherwise
1810 */
1811 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001812autowrite(buf_T *buf, int forceit)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001813{
Bram Moolenaar373154b2007-02-13 05:19:30 +00001814 int r;
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001815 bufref_T bufref;
Bram Moolenaar373154b2007-02-13 05:19:30 +00001816
Bram Moolenaar071d4272004-06-13 20:20:40 +00001817 if (!(p_aw || p_awa) || !p_write
1818#ifdef FEAT_QUICKFIX
Bram Moolenaar373154b2007-02-13 05:19:30 +00001819 /* never autowrite a "nofile" or "nowrite" buffer */
1820 || bt_dontwrite(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001821#endif
Bram Moolenaar373154b2007-02-13 05:19:30 +00001822 || (!forceit && buf->b_p_ro) || buf->b_ffname == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001823 return FAIL;
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001824 set_bufref(&bufref, buf);
Bram Moolenaar373154b2007-02-13 05:19:30 +00001825 r = buf_write_all(buf, forceit);
1826
1827 /* Writing may succeed but the buffer still changed, e.g., when there is a
1828 * conversion error. We do want to return FAIL then. */
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001829 if (bufref_valid(&bufref) && bufIsChanged(buf))
Bram Moolenaar373154b2007-02-13 05:19:30 +00001830 r = FAIL;
1831 return r;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001832}
1833
1834/*
1835 * flush all buffers, except the ones that are readonly
1836 */
1837 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001838autowrite_all(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001839{
1840 buf_T *buf;
1841
1842 if (!(p_aw || p_awa) || !p_write)
1843 return;
Bram Moolenaar29323592016-07-24 22:04:11 +02001844 FOR_ALL_BUFFERS(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001845 if (bufIsChanged(buf) && !buf->b_p_ro)
1846 {
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001847#ifdef FEAT_AUTOCMD
1848 bufref_T bufref;
1849
1850 set_bufref(&bufref, buf);
1851#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001852 (void)buf_write_all(buf, FALSE);
1853#ifdef FEAT_AUTOCMD
1854 /* an autocommand may have deleted the buffer */
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001855 if (!bufref_valid(&bufref))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001856 buf = firstbuf;
1857#endif
1858 }
1859}
1860
1861/*
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001862 * Return TRUE if buffer was changed and cannot be abandoned.
1863 * For flags use the CCGD_ values.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001864 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001865 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001866check_changed(buf_T *buf, int flags)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001867{
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001868 int forceit = (flags & CCGD_FORCEIT);
1869#ifdef FEAT_AUTOCMD
1870 bufref_T bufref;
1871
1872 set_bufref(&bufref, buf);
1873#endif
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001874
Bram Moolenaar071d4272004-06-13 20:20:40 +00001875 if ( !forceit
1876 && bufIsChanged(buf)
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001877 && ((flags & CCGD_MULTWIN) || buf->b_nwindows <= 1)
1878 && (!(flags & CCGD_AW) || autowrite(buf, forceit) == FAIL))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001879 {
1880#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG)
1881 if ((p_confirm || cmdmod.confirm) && p_write)
1882 {
1883 buf_T *buf2;
1884 int count = 0;
1885
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001886 if (flags & CCGD_ALLBUF)
Bram Moolenaar29323592016-07-24 22:04:11 +02001887 FOR_ALL_BUFFERS(buf2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001888 if (bufIsChanged(buf2)
1889 && (buf2->b_ffname != NULL
1890# ifdef FEAT_BROWSE
1891 || cmdmod.browse
1892# endif
1893 ))
1894 ++count;
1895# ifdef FEAT_AUTOCMD
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001896 if (!bufref_valid(&bufref))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001897 /* Autocommand deleted buffer, oops! It's not changed now. */
1898 return FALSE;
1899# endif
1900 dialog_changed(buf, count > 1);
1901# ifdef FEAT_AUTOCMD
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02001902 if (!bufref_valid(&bufref))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001903 /* Autocommand deleted buffer, oops! It's not changed now. */
1904 return FALSE;
1905# endif
1906 return bufIsChanged(buf);
1907 }
1908#endif
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001909 if (flags & CCGD_EXCMD)
1910 EMSG(_(e_nowrtmsg));
1911 else
1912 EMSG(_(e_nowrtmsg_nobang));
Bram Moolenaar071d4272004-06-13 20:20:40 +00001913 return TRUE;
1914 }
1915 return FALSE;
1916}
1917
1918#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG) || defined(PROTO)
1919
1920#if defined(FEAT_BROWSE) || defined(PROTO)
1921/*
1922 * When wanting to write a file without a file name, ask the user for a name.
1923 */
1924 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001925browse_save_fname(buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001926{
1927 if (buf->b_fname == NULL)
1928 {
1929 char_u *fname;
1930
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00001931 fname = do_browse(BROWSE_SAVE, (char_u *)_("Save As"),
1932 NULL, NULL, NULL, NULL, buf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001933 if (fname != NULL)
1934 {
1935 if (setfname(buf, fname, NULL, TRUE) == OK)
1936 buf->b_flags |= BF_NOTEDITED;
1937 vim_free(fname);
1938 }
1939 }
1940}
1941#endif
1942
1943/*
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02001944 * Ask the user what to do when abandoning a changed buffer.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001945 * Must check 'write' option first!
1946 */
1947 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001948dialog_changed(
1949 buf_T *buf,
1950 int checkall) /* may abandon all changed buffers */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001951{
Bram Moolenaard9462e32011-04-11 21:35:11 +02001952 char_u buff[DIALOG_MSG_SIZE];
Bram Moolenaar071d4272004-06-13 20:20:40 +00001953 int ret;
1954 buf_T *buf2;
Bram Moolenaar8218f602012-04-25 17:32:18 +02001955 exarg_T ea;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001956
Bram Moolenaar82cf9b62005-06-07 21:09:25 +00001957 dialog_msg(buff, _("Save changes to \"%s\"?"),
Bram Moolenaar071d4272004-06-13 20:20:40 +00001958 (buf->b_fname != NULL) ?
1959 buf->b_fname : (char_u *)_("Untitled"));
1960 if (checkall)
1961 ret = vim_dialog_yesnoallcancel(VIM_QUESTION, NULL, buff, 1);
1962 else
1963 ret = vim_dialog_yesnocancel(VIM_QUESTION, NULL, buff, 1);
1964
Bram Moolenaar8218f602012-04-25 17:32:18 +02001965 /* Init ea pseudo-structure, this is needed for the check_overwrite()
1966 * function. */
1967 ea.append = ea.forceit = FALSE;
1968
Bram Moolenaar071d4272004-06-13 20:20:40 +00001969 if (ret == VIM_YES)
1970 {
1971#ifdef FEAT_BROWSE
1972 /* May get file name, when there is none */
1973 browse_save_fname(buf);
1974#endif
Bram Moolenaar8218f602012-04-25 17:32:18 +02001975 if (buf->b_fname != NULL && check_overwrite(&ea, buf,
1976 buf->b_fname, buf->b_ffname, FALSE) == OK)
1977 /* didn't hit Cancel */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001978 (void)buf_write_all(buf, FALSE);
1979 }
1980 else if (ret == VIM_NO)
1981 {
1982 unchanged(buf, TRUE);
1983 }
1984 else if (ret == VIM_ALL)
1985 {
1986 /*
1987 * Write all modified files that can be written.
1988 * Skip readonly buffers, these need to be confirmed
1989 * individually.
1990 */
Bram Moolenaar29323592016-07-24 22:04:11 +02001991 FOR_ALL_BUFFERS(buf2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001992 {
1993 if (bufIsChanged(buf2)
1994 && (buf2->b_ffname != NULL
1995#ifdef FEAT_BROWSE
1996 || cmdmod.browse
1997#endif
1998 )
1999 && !buf2->b_p_ro)
2000 {
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002001#ifdef FEAT_AUTOCMD
2002 bufref_T bufref;
2003
2004 set_bufref(&bufref, buf2);
2005#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002006#ifdef FEAT_BROWSE
2007 /* May get file name, when there is none */
2008 browse_save_fname(buf2);
2009#endif
Bram Moolenaar8218f602012-04-25 17:32:18 +02002010 if (buf2->b_fname != NULL && check_overwrite(&ea, buf2,
2011 buf2->b_fname, buf2->b_ffname, FALSE) == OK)
2012 /* didn't hit Cancel */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002013 (void)buf_write_all(buf2, FALSE);
2014#ifdef FEAT_AUTOCMD
2015 /* an autocommand may have deleted the buffer */
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002016 if (!bufref_valid(&bufref))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002017 buf2 = firstbuf;
2018#endif
2019 }
2020 }
2021 }
2022 else if (ret == VIM_DISCARDALL)
2023 {
2024 /*
2025 * mark all buffers as unchanged
2026 */
Bram Moolenaar29323592016-07-24 22:04:11 +02002027 FOR_ALL_BUFFERS(buf2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002028 unchanged(buf2, TRUE);
2029 }
2030}
2031#endif
2032
2033/*
2034 * Return TRUE if the buffer "buf" can be abandoned, either by making it
2035 * hidden, autowriting it or unloading it.
2036 */
2037 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002038can_abandon(buf_T *buf, int forceit)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002039{
2040 return ( P_HID(buf)
2041 || !bufIsChanged(buf)
2042 || buf->b_nwindows > 1
2043 || autowrite(buf, forceit) == OK
2044 || forceit);
2045}
2046
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01002047static void add_bufnum(int *bufnrs, int *bufnump, int nr);
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002048
2049/*
2050 * Add a buffer number to "bufnrs", unless it's already there.
2051 */
2052 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002053add_bufnum(int *bufnrs, int *bufnump, int nr)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002054{
2055 int i;
2056
2057 for (i = 0; i < *bufnump; ++i)
2058 if (bufnrs[i] == nr)
2059 return;
2060 bufnrs[*bufnump] = nr;
2061 *bufnump = *bufnump + 1;
2062}
2063
Bram Moolenaar071d4272004-06-13 20:20:40 +00002064/*
2065 * Return TRUE if any buffer was changed and cannot be abandoned.
2066 * That changed buffer becomes the current buffer.
Bram Moolenaar027387f2016-01-02 22:25:52 +01002067 * When "unload" is true the current buffer is unloaded instead of making it
2068 * hidden. This is used for ":q!".
Bram Moolenaar071d4272004-06-13 20:20:40 +00002069 */
2070 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002071check_changed_any(
2072 int hidden, /* Only check hidden buffers */
2073 int unload)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002074{
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002075 int ret = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002076 buf_T *buf;
2077 int save;
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002078 int i;
2079 int bufnum = 0;
2080 int bufcount = 0;
2081 int *bufnrs;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002082#ifdef FEAT_WINDOWS
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002083 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002084 win_T *wp;
2085#endif
2086
Bram Moolenaar29323592016-07-24 22:04:11 +02002087 FOR_ALL_BUFFERS(buf)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002088 ++bufcount;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002089
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002090 if (bufcount == 0)
2091 return FALSE;
2092
2093 bufnrs = (int *)alloc(sizeof(int) * bufcount);
2094 if (bufnrs == NULL)
2095 return FALSE;
2096
2097 /* curbuf */
2098 bufnrs[bufnum++] = curbuf->b_fnum;
2099#ifdef FEAT_WINDOWS
2100 /* buf in curtab */
2101 FOR_ALL_WINDOWS(wp)
2102 if (wp->w_buffer != curbuf)
2103 add_bufnum(bufnrs, &bufnum, wp->w_buffer->b_fnum);
2104
2105 /* buf in other tab */
Bram Moolenaar29323592016-07-24 22:04:11 +02002106 FOR_ALL_TABPAGES(tp)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002107 if (tp != curtab)
2108 for (wp = tp->tp_firstwin; wp != NULL; wp = wp->w_next)
2109 add_bufnum(bufnrs, &bufnum, wp->w_buffer->b_fnum);
2110#endif
2111 /* any other buf */
Bram Moolenaar29323592016-07-24 22:04:11 +02002112 FOR_ALL_BUFFERS(buf)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002113 add_bufnum(bufnrs, &bufnum, buf->b_fnum);
2114
2115 for (i = 0; i < bufnum; ++i)
2116 {
2117 buf = buflist_findnr(bufnrs[i]);
2118 if (buf == NULL)
2119 continue;
2120 if ((!hidden || buf->b_nwindows == 0) && bufIsChanged(buf))
2121 {
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002122 bufref_T bufref;
2123
2124 set_bufref(&bufref, buf);
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002125 /* Try auto-writing the buffer. If this fails but the buffer no
2126 * longer exists it's not changed, that's OK. */
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002127 if (check_changed(buf, (p_awa ? CCGD_AW : 0)
2128 | CCGD_MULTWIN
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002129 | CCGD_ALLBUF) && bufref_valid(&bufref))
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002130 break; /* didn't save - still changes */
2131 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002132 }
2133
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002134 if (i >= bufnum)
2135 goto theend;
2136
2137 ret = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002138 exiting = FALSE;
2139#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG)
2140 /*
2141 * When ":confirm" used, don't give an error message.
2142 */
2143 if (!(p_confirm || cmdmod.confirm))
2144#endif
2145 {
2146 /* There must be a wait_return for this message, do_buffer()
2147 * may cause a redraw. But wait_return() is a no-op when vgetc()
2148 * is busy (Quit used from window menu), then make sure we don't
2149 * cause a scroll up. */
Bram Moolenaar61660ea2006-04-07 21:40:07 +00002150 if (vgetc_busy > 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002151 {
2152 msg_row = cmdline_row;
2153 msg_col = 0;
2154 msg_didout = FALSE;
2155 }
2156 if (EMSG2(_("E162: No write since last change for buffer \"%s\""),
Bram Moolenaare1704ba2012-10-03 18:25:00 +02002157 buf_spname(buf) != NULL ? buf_spname(buf) : buf->b_fname))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002158 {
2159 save = no_wait_return;
2160 no_wait_return = FALSE;
2161 wait_return(FALSE);
2162 no_wait_return = save;
2163 }
2164 }
2165
2166#ifdef FEAT_WINDOWS
2167 /* Try to find a window that contains the buffer. */
2168 if (buf != curbuf)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002169 FOR_ALL_TAB_WINDOWS(tp, wp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002170 if (wp->w_buffer == buf)
2171 {
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002172# ifdef FEAT_AUTOCMD
2173 bufref_T bufref;
2174
2175 set_bufref(&bufref, buf);
2176# endif
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002177 goto_tabpage_win(tp, wp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002178# ifdef FEAT_AUTOCMD
2179 /* Paranoia: did autocms wipe out the buffer with changes? */
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02002180 if (!bufref_valid(&bufref))
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002181 {
2182 goto theend;
2183 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002184# endif
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002185 goto buf_found;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002186 }
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002187buf_found:
Bram Moolenaar071d4272004-06-13 20:20:40 +00002188#endif
2189
2190 /* Open the changed buffer in the current window. */
2191 if (buf != curbuf)
Bram Moolenaar027387f2016-01-02 22:25:52 +01002192 set_curbuf(buf, unload ? DOBUF_UNLOAD : DOBUF_GOTO);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002193
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002194theend:
2195 vim_free(bufnrs);
2196 return ret;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002197}
2198
2199/*
2200 * return FAIL if there is no file name, OK if there is one
2201 * give error message for FAIL
2202 */
2203 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002204check_fname(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002205{
2206 if (curbuf->b_ffname == NULL)
2207 {
2208 EMSG(_(e_noname));
2209 return FAIL;
2210 }
2211 return OK;
2212}
2213
2214/*
2215 * flush the contents of a buffer, unless it has no file name
2216 *
2217 * return FAIL for failure, OK otherwise
2218 */
2219 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002220buf_write_all(buf_T *buf, int forceit)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002221{
2222 int retval;
2223#ifdef FEAT_AUTOCMD
2224 buf_T *old_curbuf = curbuf;
2225#endif
2226
2227 retval = (buf_write(buf, buf->b_ffname, buf->b_fname,
2228 (linenr_T)1, buf->b_ml.ml_line_count, NULL,
2229 FALSE, forceit, TRUE, FALSE));
2230#ifdef FEAT_AUTOCMD
2231 if (curbuf != old_curbuf)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00002232 {
Bram Moolenaar8820b482017-03-16 17:23:31 +01002233 msg_source(HL_ATTR(HLF_W));
Bram Moolenaar071d4272004-06-13 20:20:40 +00002234 MSG(_("Warning: Entered other buffer unexpectedly (check autocommands)"));
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00002235 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002236#endif
2237 return retval;
2238}
2239
2240/*
2241 * Code to handle the argument list.
2242 */
2243
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01002244static char_u *do_one_arg(char_u *str);
2245static int do_arglist(char_u *str, int what, int after);
2246static void alist_check_arg_idx(void);
2247static int editing_arg_idx(win_T *win);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002248#ifdef FEAT_LISTCMDS
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01002249static int alist_add_list(int count, char_u **files, int after);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002250#endif
2251#define AL_SET 1
2252#define AL_ADD 2
2253#define AL_DEL 3
2254
Bram Moolenaar071d4272004-06-13 20:20:40 +00002255/*
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002256 * Isolate one argument, taking backticks.
2257 * Changes the argument in-place, puts a NUL after it. Backticks remain.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002258 * Return a pointer to the start of the next argument.
2259 */
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002260 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002261do_one_arg(char_u *str)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002262{
2263 char_u *p;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002264 int inbacktick;
2265
Bram Moolenaar071d4272004-06-13 20:20:40 +00002266 inbacktick = FALSE;
2267 for (p = str; *str; ++str)
2268 {
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002269 /* When the backslash is used for escaping the special meaning of a
2270 * character we need to keep it until wildcard expansion. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002271 if (rem_backslash(str))
2272 {
2273 *p++ = *str++;
2274 *p++ = *str;
2275 }
2276 else
2277 {
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002278 /* An item ends at a space not in backticks */
2279 if (!inbacktick && vim_isspace(*str))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002280 break;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002281 if (*str == '`')
Bram Moolenaar071d4272004-06-13 20:20:40 +00002282 inbacktick ^= TRUE;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002283 *p++ = *str;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002284 }
2285 }
2286 str = skipwhite(str);
2287 *p = NUL;
2288
2289 return str;
2290}
2291
Bram Moolenaar86b68352004-12-27 21:59:20 +00002292/*
2293 * Separate the arguments in "str" and return a list of pointers in the
2294 * growarray "gap".
2295 */
2296 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002297get_arglist(garray_T *gap, char_u *str)
Bram Moolenaar86b68352004-12-27 21:59:20 +00002298{
2299 ga_init2(gap, (int)sizeof(char_u *), 20);
2300 while (*str != NUL)
2301 {
2302 if (ga_grow(gap, 1) == FAIL)
2303 {
2304 ga_clear(gap);
2305 return FAIL;
2306 }
2307 ((char_u **)gap->ga_data)[gap->ga_len++] = str;
2308
2309 /* Isolate one argument, change it in-place, put a NUL after it. */
2310 str = do_one_arg(str);
2311 }
2312 return OK;
2313}
2314
Bram Moolenaar7df351e2006-01-23 22:30:28 +00002315#if defined(FEAT_QUICKFIX) || defined(FEAT_SYN_HL) || defined(PROTO)
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002316/*
2317 * Parse a list of arguments (file names), expand them and return in
Bram Moolenaar8f5c6f02012-06-29 12:57:06 +02002318 * "fnames[fcountp]". When "wig" is TRUE, removes files matching 'wildignore'.
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002319 * Return FAIL or OK.
2320 */
2321 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002322get_arglist_exp(
2323 char_u *str,
2324 int *fcountp,
2325 char_u ***fnamesp,
2326 int wig)
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002327{
2328 garray_T ga;
2329 int i;
2330
2331 if (get_arglist(&ga, str) == FAIL)
2332 return FAIL;
Bram Moolenaar8f5c6f02012-06-29 12:57:06 +02002333 if (wig == TRUE)
2334 i = expand_wildcards(ga.ga_len, (char_u **)ga.ga_data,
2335 fcountp, fnamesp, EW_FILE|EW_NOTFOUND);
2336 else
2337 i = gen_expand_wildcards(ga.ga_len, (char_u **)ga.ga_data,
2338 fcountp, fnamesp, EW_FILE|EW_NOTFOUND);
2339
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002340 ga_clear(&ga);
2341 return i;
2342}
2343#endif
2344
Bram Moolenaar071d4272004-06-13 20:20:40 +00002345#if defined(FEAT_GUI) || defined(FEAT_CLIENTSERVER) || defined(PROTO)
2346/*
2347 * Redefine the argument list.
2348 */
2349 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002350set_arglist(char_u *str)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002351{
2352 do_arglist(str, AL_SET, 0);
2353}
2354#endif
2355
2356/*
2357 * "what" == AL_SET: Redefine the argument list to 'str'.
2358 * "what" == AL_ADD: add files in 'str' to the argument list after "after".
2359 * "what" == AL_DEL: remove files in 'str' from the argument list.
2360 *
2361 * Return FAIL for failure, OK otherwise.
2362 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002363 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002364do_arglist(
2365 char_u *str,
Bram Moolenaarf1d25012016-03-03 12:22:53 +01002366 int what,
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002367 int after UNUSED) /* 0 means before first one */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002368{
2369 garray_T new_ga;
2370 int exp_count;
2371 char_u **exp_files;
2372 int i;
2373#ifdef FEAT_LISTCMDS
2374 char_u *p;
2375 int match;
2376#endif
2377
2378 /*
Bram Moolenaar2faa29f2016-01-23 23:02:34 +01002379 * Set default argument for ":argadd" command.
2380 */
2381 if (what == AL_ADD && *str == NUL)
2382 {
2383 if (curbuf->b_ffname == NULL)
2384 return FAIL;
2385 str = curbuf->b_fname;
2386 }
2387
2388 /*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002389 * Collect all file name arguments in "new_ga".
2390 */
Bram Moolenaar86b68352004-12-27 21:59:20 +00002391 if (get_arglist(&new_ga, str) == FAIL)
2392 return FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002393
2394#ifdef FEAT_LISTCMDS
2395 if (what == AL_DEL)
2396 {
2397 regmatch_T regmatch;
2398 int didone;
2399
2400 /*
2401 * Delete the items: use each item as a regexp and find a match in the
2402 * argument list.
2403 */
Bram Moolenaar71afbfe2013-03-19 16:49:16 +01002404 regmatch.rm_ic = p_fic; /* ignore case when 'fileignorecase' is set */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002405 for (i = 0; i < new_ga.ga_len && !got_int; ++i)
2406 {
2407 p = ((char_u **)new_ga.ga_data)[i];
2408 p = file_pat_to_reg_pat(p, NULL, NULL, FALSE);
2409 if (p == NULL)
2410 break;
2411 regmatch.regprog = vim_regcomp(p, p_magic ? RE_MAGIC : 0);
2412 if (regmatch.regprog == NULL)
2413 {
2414 vim_free(p);
2415 break;
2416 }
2417
2418 didone = FALSE;
2419 for (match = 0; match < ARGCOUNT; ++match)
2420 if (vim_regexec(&regmatch, alist_name(&ARGLIST[match]),
2421 (colnr_T)0))
2422 {
2423 didone = TRUE;
2424 vim_free(ARGLIST[match].ae_fname);
2425 mch_memmove(ARGLIST + match, ARGLIST + match + 1,
2426 (ARGCOUNT - match - 1) * sizeof(aentry_T));
2427 --ALIST(curwin)->al_ga.ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002428 if (curwin->w_arg_idx > match)
2429 --curwin->w_arg_idx;
2430 --match;
2431 }
2432
Bram Moolenaar473de612013-06-08 18:19:48 +02002433 vim_regfree(regmatch.regprog);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002434 vim_free(p);
2435 if (!didone)
2436 EMSG2(_(e_nomatch2), ((char_u **)new_ga.ga_data)[i]);
2437 }
2438 ga_clear(&new_ga);
2439 }
2440 else
2441#endif
2442 {
2443 i = expand_wildcards(new_ga.ga_len, (char_u **)new_ga.ga_data,
2444 &exp_count, &exp_files, EW_DIR|EW_FILE|EW_ADDSLASH|EW_NOTFOUND);
2445 ga_clear(&new_ga);
Bram Moolenaar2db5c3b2016-01-16 22:49:34 +01002446 if (i == FAIL || exp_count == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002447 {
2448 EMSG(_(e_nomatch));
2449 return FAIL;
2450 }
2451
2452#ifdef FEAT_LISTCMDS
2453 if (what == AL_ADD)
2454 {
2455 (void)alist_add_list(exp_count, exp_files, after);
2456 vim_free(exp_files);
2457 }
2458 else /* what == AL_SET */
2459#endif
Bram Moolenaar86b68352004-12-27 21:59:20 +00002460 alist_set(ALIST(curwin), exp_count, exp_files, FALSE, NULL, 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002461 }
2462
2463 alist_check_arg_idx();
2464
2465 return OK;
2466}
2467
2468/*
2469 * Check the validity of the arg_idx for each other window.
2470 */
2471 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002472alist_check_arg_idx(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002473{
2474#ifdef FEAT_WINDOWS
2475 win_T *win;
Bram Moolenaarf740b292006-02-16 22:11:02 +00002476 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002477
Bram Moolenaarf740b292006-02-16 22:11:02 +00002478 FOR_ALL_TAB_WINDOWS(tp, win)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002479 if (win->w_alist == curwin->w_alist)
2480 check_arg_idx(win);
2481#else
2482 check_arg_idx(curwin);
2483#endif
2484}
2485
2486/*
Bram Moolenaarb8ff1fb2012-02-04 21:59:01 +01002487 * Return TRUE if window "win" is editing the file at the current argument
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002488 * index.
2489 */
2490 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002491editing_arg_idx(win_T *win)
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002492{
2493 return !(win->w_arg_idx >= WARGCOUNT(win)
2494 || (win->w_buffer->b_fnum
2495 != WARGLIST(win)[win->w_arg_idx].ae_fnum
2496 && (win->w_buffer->b_ffname == NULL
2497 || !(fullpathcmp(
2498 alist_name(&WARGLIST(win)[win->w_arg_idx]),
2499 win->w_buffer->b_ffname, TRUE) & FPC_SAME))));
2500}
2501
2502/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002503 * Check if window "win" is editing the w_arg_idx file in its argument list.
2504 */
2505 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002506check_arg_idx(win_T *win)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002507{
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002508 if (WARGCOUNT(win) > 1 && !editing_arg_idx(win))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002509 {
2510 /* We are not editing the current entry in the argument list.
2511 * Set "arg_had_last" if we are editing the last one. */
2512 win->w_arg_idx_invalid = TRUE;
2513 if (win->w_arg_idx != WARGCOUNT(win) - 1
2514 && arg_had_last == FALSE
2515#ifdef FEAT_WINDOWS
2516 && ALIST(win) == &global_alist
2517#endif
2518 && GARGCOUNT > 0
2519 && win->w_arg_idx < GARGCOUNT
2520 && (win->w_buffer->b_fnum == GARGLIST[GARGCOUNT - 1].ae_fnum
2521 || (win->w_buffer->b_ffname != NULL
2522 && (fullpathcmp(alist_name(&GARGLIST[GARGCOUNT - 1]),
2523 win->w_buffer->b_ffname, TRUE) & FPC_SAME))))
2524 arg_had_last = TRUE;
2525 }
2526 else
2527 {
2528 /* We are editing the current entry in the argument list.
2529 * Set "arg_had_last" if it's also the last one */
2530 win->w_arg_idx_invalid = FALSE;
2531 if (win->w_arg_idx == WARGCOUNT(win) - 1
2532#ifdef FEAT_WINDOWS
2533 && win->w_alist == &global_alist
2534#endif
2535 )
2536 arg_had_last = TRUE;
2537 }
2538}
2539
2540/*
2541 * ":args", ":argslocal" and ":argsglobal".
2542 */
2543 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002544ex_args(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002545{
2546 int i;
2547
2548 if (eap->cmdidx != CMD_args)
2549 {
2550#if defined(FEAT_WINDOWS) && defined(FEAT_LISTCMDS)
2551 alist_unlink(ALIST(curwin));
2552 if (eap->cmdidx == CMD_argglobal)
2553 ALIST(curwin) = &global_alist;
2554 else /* eap->cmdidx == CMD_arglocal */
2555 alist_new();
2556#else
2557 ex_ni(eap);
2558 return;
2559#endif
2560 }
2561
2562 if (!ends_excmd(*eap->arg))
2563 {
2564 /*
2565 * ":args file ..": define new argument list, handle like ":next"
2566 * Also for ":argslocal file .." and ":argsglobal file ..".
2567 */
2568 ex_next(eap);
2569 }
2570 else
2571#if defined(FEAT_WINDOWS) && defined(FEAT_LISTCMDS)
2572 if (eap->cmdidx == CMD_args)
2573#endif
2574 {
2575 /*
2576 * ":args": list arguments.
2577 */
2578 if (ARGCOUNT > 0)
2579 {
2580 /* Overwrite the command, for a short list there is no scrolling
2581 * required and no wait_return(). */
2582 gotocmdline(TRUE);
2583 for (i = 0; i < ARGCOUNT; ++i)
2584 {
2585 if (i == curwin->w_arg_idx)
2586 msg_putchar('[');
2587 msg_outtrans(alist_name(&ARGLIST[i]));
2588 if (i == curwin->w_arg_idx)
2589 msg_putchar(']');
2590 msg_putchar(' ');
2591 }
2592 }
2593 }
2594#if defined(FEAT_WINDOWS) && defined(FEAT_LISTCMDS)
2595 else if (eap->cmdidx == CMD_arglocal)
2596 {
2597 garray_T *gap = &curwin->w_alist->al_ga;
2598
2599 /*
2600 * ":argslocal": make a local copy of the global argument list.
2601 */
2602 if (ga_grow(gap, GARGCOUNT) == OK)
2603 for (i = 0; i < GARGCOUNT; ++i)
2604 if (GARGLIST[i].ae_fname != NULL)
2605 {
2606 AARGLIST(curwin->w_alist)[gap->ga_len].ae_fname =
2607 vim_strsave(GARGLIST[i].ae_fname);
2608 AARGLIST(curwin->w_alist)[gap->ga_len].ae_fnum =
2609 GARGLIST[i].ae_fnum;
2610 ++gap->ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002611 }
2612 }
2613#endif
2614}
2615
2616/*
2617 * ":previous", ":sprevious", ":Next" and ":sNext".
2618 */
2619 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002620ex_previous(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002621{
2622 /* If past the last one already, go to the last one. */
2623 if (curwin->w_arg_idx - (int)eap->line2 >= ARGCOUNT)
2624 do_argfile(eap, ARGCOUNT - 1);
2625 else
2626 do_argfile(eap, curwin->w_arg_idx - (int)eap->line2);
2627}
2628
2629/*
2630 * ":rewind", ":first", ":sfirst" and ":srewind".
2631 */
2632 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002633ex_rewind(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002634{
2635 do_argfile(eap, 0);
2636}
2637
2638/*
2639 * ":last" and ":slast".
2640 */
2641 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002642ex_last(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002643{
2644 do_argfile(eap, ARGCOUNT - 1);
2645}
2646
2647/*
2648 * ":argument" and ":sargument".
2649 */
2650 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002651ex_argument(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002652{
2653 int i;
2654
2655 if (eap->addr_count > 0)
2656 i = eap->line2 - 1;
2657 else
2658 i = curwin->w_arg_idx;
2659 do_argfile(eap, i);
2660}
2661
2662/*
2663 * Edit file "argn" of the argument lists.
2664 */
2665 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002666do_argfile(exarg_T *eap, int argn)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002667{
2668 int other;
2669 char_u *p;
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002670 int old_arg_idx = curwin->w_arg_idx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002671
2672 if (argn < 0 || argn >= ARGCOUNT)
2673 {
2674 if (ARGCOUNT <= 1)
2675 EMSG(_("E163: There is only one file to edit"));
2676 else if (argn < 0)
2677 EMSG(_("E164: Cannot go before first file"));
2678 else
2679 EMSG(_("E165: Cannot go beyond last file"));
2680 }
2681 else
2682 {
2683 setpcmark();
2684#ifdef FEAT_GUI
2685 need_mouse_correct = TRUE;
2686#endif
2687
2688#ifdef FEAT_WINDOWS
Bram Moolenaardf1bdc92006-02-23 21:32:16 +00002689 /* split window or create new tab page first */
2690 if (*eap->cmd == 's' || cmdmod.tab != 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002691 {
2692 if (win_split(0, 0) == FAIL)
2693 return;
Bram Moolenaar3368ea22010-09-21 16:56:35 +02002694 RESET_BINDING(curwin);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002695 }
2696 else
2697#endif
2698 {
2699 /*
2700 * if 'hidden' set, only check for changed file when re-editing
2701 * the same buffer
2702 */
2703 other = TRUE;
2704 if (P_HID(curbuf))
2705 {
2706 p = fix_fname(alist_name(&ARGLIST[argn]));
2707 other = otherfile(p);
2708 vim_free(p);
2709 }
2710 if ((!P_HID(curbuf) || !other)
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002711 && check_changed(curbuf, CCGD_AW
2712 | (other ? 0 : CCGD_MULTWIN)
2713 | (eap->forceit ? CCGD_FORCEIT : 0)
2714 | CCGD_EXCMD))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002715 return;
2716 }
2717
2718 curwin->w_arg_idx = argn;
2719 if (argn == ARGCOUNT - 1
2720#ifdef FEAT_WINDOWS
2721 && curwin->w_alist == &global_alist
2722#endif
2723 )
2724 arg_had_last = TRUE;
2725
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002726 /* Edit the file; always use the last known line number.
2727 * When it fails (e.g. Abort for already edited file) restore the
2728 * argument index. */
2729 if (do_ecmd(0, alist_name(&ARGLIST[curwin->w_arg_idx]), NULL,
Bram Moolenaar071d4272004-06-13 20:20:40 +00002730 eap, ECMD_LAST,
Bram Moolenaar701f7af2008-11-15 13:12:07 +00002731 (P_HID(curwin->w_buffer) ? ECMD_HIDE : 0)
2732 + (eap->forceit ? ECMD_FORCEIT : 0), curwin) == FAIL)
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002733 curwin->w_arg_idx = old_arg_idx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002734 /* like Vi: set the mark where the cursor is in the file. */
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002735 else if (eap->cmdidx != CMD_argdo)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002736 setmark('\'');
2737 }
2738}
2739
2740/*
2741 * ":next", and commands that behave like it.
2742 */
2743 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002744ex_next(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002745{
2746 int i;
2747
2748 /*
2749 * check for changed buffer now, if this fails the argument list is not
2750 * redefined.
2751 */
2752 if ( P_HID(curbuf)
2753 || eap->cmdidx == CMD_snext
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002754 || !check_changed(curbuf, CCGD_AW
2755 | (eap->forceit ? CCGD_FORCEIT : 0)
2756 | CCGD_EXCMD))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002757 {
2758 if (*eap->arg != NUL) /* redefine file list */
2759 {
2760 if (do_arglist(eap->arg, AL_SET, 0) == FAIL)
2761 return;
2762 i = 0;
2763 }
2764 else
2765 i = curwin->w_arg_idx + (int)eap->line2;
2766 do_argfile(eap, i);
2767 }
2768}
2769
Bram Moolenaar0106e3d2016-02-23 18:55:43 +01002770#if defined(FEAT_LISTCMDS) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002771/*
2772 * ":argedit"
2773 */
2774 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002775ex_argedit(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002776{
2777 int fnum;
2778 int i;
2779 char_u *s;
2780
2781 /* Add the argument to the buffer list and get the buffer number. */
2782 fnum = buflist_add(eap->arg, BLN_LISTED);
2783
2784 /* Check if this argument is already in the argument list. */
2785 for (i = 0; i < ARGCOUNT; ++i)
2786 if (ARGLIST[i].ae_fnum == fnum)
2787 break;
2788 if (i == ARGCOUNT)
2789 {
2790 /* Can't find it, add it to the argument list. */
2791 s = vim_strsave(eap->arg);
2792 if (s == NULL)
2793 return;
2794 i = alist_add_list(1, &s,
2795 eap->addr_count > 0 ? (int)eap->line2 : curwin->w_arg_idx + 1);
2796 if (i < 0)
2797 return;
2798 curwin->w_arg_idx = i;
2799 }
2800
2801 alist_check_arg_idx();
2802
2803 /* Edit the argument. */
2804 do_argfile(eap, i);
2805}
2806
2807/*
2808 * ":argadd"
2809 */
2810 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002811ex_argadd(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002812{
2813 do_arglist(eap->arg, AL_ADD,
2814 eap->addr_count > 0 ? (int)eap->line2 : curwin->w_arg_idx + 1);
2815#ifdef FEAT_TITLE
2816 maketitle();
2817#endif
2818}
2819
2820/*
2821 * ":argdelete"
2822 */
2823 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002824ex_argdelete(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002825{
2826 int i;
2827 int n;
2828
2829 if (eap->addr_count > 0)
2830 {
2831 /* ":1,4argdel": Delete all arguments in the range. */
2832 if (eap->line2 > ARGCOUNT)
2833 eap->line2 = ARGCOUNT;
2834 n = eap->line2 - eap->line1 + 1;
Bram Moolenaar69a92fb2017-03-09 15:58:30 +01002835 if (*eap->arg != NUL)
2836 /* Can't have both a range and an argument. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002837 EMSG(_(e_invarg));
Bram Moolenaar69a92fb2017-03-09 15:58:30 +01002838 else if (n <= 0)
2839 {
2840 /* Don't give an error for ":%argdel" if the list is empty. */
2841 if (eap->line1 != 1 || eap->line2 != 0)
2842 EMSG(_(e_invrange));
2843 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002844 else
2845 {
2846 for (i = eap->line1; i <= eap->line2; ++i)
2847 vim_free(ARGLIST[i - 1].ae_fname);
2848 mch_memmove(ARGLIST + eap->line1 - 1, ARGLIST + eap->line2,
2849 (size_t)((ARGCOUNT - eap->line2) * sizeof(aentry_T)));
2850 ALIST(curwin)->al_ga.ga_len -= n;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002851 if (curwin->w_arg_idx >= eap->line2)
2852 curwin->w_arg_idx -= n;
2853 else if (curwin->w_arg_idx > eap->line1)
2854 curwin->w_arg_idx = eap->line1;
Bram Moolenaar72defda2016-01-17 18:04:33 +01002855 if (ARGCOUNT == 0)
2856 curwin->w_arg_idx = 0;
2857 else if (curwin->w_arg_idx >= ARGCOUNT)
2858 curwin->w_arg_idx = ARGCOUNT - 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002859 }
2860 }
2861 else if (*eap->arg == NUL)
2862 EMSG(_(e_argreq));
2863 else
2864 do_arglist(eap->arg, AL_DEL, 0);
2865#ifdef FEAT_TITLE
2866 maketitle();
2867#endif
2868}
2869
2870/*
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002871 * ":argdo", ":windo", ":bufdo", ":tabdo", ":cdo", ":ldo", ":cfdo" and ":lfdo"
Bram Moolenaar071d4272004-06-13 20:20:40 +00002872 */
2873 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002874ex_listdo(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002875{
2876 int i;
2877#ifdef FEAT_WINDOWS
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002878 win_T *wp;
2879 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002880#endif
Bram Moolenaare25bb902015-02-27 20:33:37 +01002881 buf_T *buf = curbuf;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002882 int next_fnum = 0;
2883#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
2884 char_u *save_ei = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002885#endif
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002886 char_u *p_shm_save;
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002887#ifdef FEAT_QUICKFIX
Bram Moolenaared84b762015-09-09 22:35:29 +02002888 int qf_size = 0;
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002889 int qf_idx;
2890#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002891
2892#ifndef FEAT_WINDOWS
2893 if (eap->cmdidx == CMD_windo)
2894 {
2895 ex_ni(eap);
2896 return;
2897 }
2898#endif
2899
Bram Moolenaar0106e3d2016-02-23 18:55:43 +01002900#ifndef FEAT_QUICKFIX
2901 if (eap->cmdidx == CMD_cdo || eap->cmdidx == CMD_ldo ||
2902 eap->cmdidx == CMD_cfdo || eap->cmdidx == CMD_lfdo)
2903 {
2904 ex_ni(eap);
2905 return;
2906 }
2907#endif
2908
Bram Moolenaar071d4272004-06-13 20:20:40 +00002909#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002910 if (eap->cmdidx != CMD_windo && eap->cmdidx != CMD_tabdo)
Bram Moolenaardcaf10e2005-01-21 11:55:25 +00002911 /* Don't do syntax HL autocommands. Skipping the syntax file is a
2912 * great speed improvement. */
2913 save_ei = au_event_disable(",Syntax");
Bram Moolenaar071d4272004-06-13 20:20:40 +00002914#endif
Bram Moolenaar6b1ee342014-08-06 18:17:11 +02002915#ifdef FEAT_CLIPBOARD
2916 start_global_changes();
2917#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002918
2919 if (eap->cmdidx == CMD_windo
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002920 || eap->cmdidx == CMD_tabdo
Bram Moolenaar071d4272004-06-13 20:20:40 +00002921 || P_HID(curbuf)
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002922 || !check_changed(curbuf, CCGD_AW
2923 | (eap->forceit ? CCGD_FORCEIT : 0)
2924 | CCGD_EXCMD))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002925 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00002926 i = 0;
Bram Moolenaara162bc52015-01-07 16:54:21 +01002927 /* start at the eap->line1 argument/window/buffer */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002928#ifdef FEAT_WINDOWS
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002929 wp = firstwin;
2930 tp = first_tabpage;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002931#endif
Bram Moolenaara162bc52015-01-07 16:54:21 +01002932 switch (eap->cmdidx)
2933 {
2934#ifdef FEAT_WINDOWS
2935 case CMD_windo:
2936 for ( ; wp != NULL && i + 1 < eap->line1; wp = wp->w_next)
2937 i++;
2938 break;
2939 case CMD_tabdo:
2940 for( ; tp != NULL && i + 1 < eap->line1; tp = tp->tp_next)
2941 i++;
2942 break;
2943#endif
2944 case CMD_argdo:
2945 i = eap->line1 - 1;
2946 break;
Bram Moolenaara162bc52015-01-07 16:54:21 +01002947 default:
2948 break;
2949 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002950 /* set pcmark now */
2951 if (eap->cmdidx == CMD_bufdo)
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002952 {
Bram Moolenaare25bb902015-02-27 20:33:37 +01002953 /* Advance to the first listed buffer after "eap->line1". */
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002954 for (buf = firstbuf; buf != NULL && (buf->b_fnum < eap->line1
Bram Moolenaare25bb902015-02-27 20:33:37 +01002955 || !buf->b_p_bl); buf = buf->b_next)
2956 if (buf->b_fnum > eap->line2)
2957 {
2958 buf = NULL;
2959 break;
2960 }
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002961 if (buf != NULL)
Bram Moolenaare25bb902015-02-27 20:33:37 +01002962 goto_buffer(eap, DOBUF_FIRST, FORWARD, buf->b_fnum);
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002963 }
2964#ifdef FEAT_QUICKFIX
2965 else if (eap->cmdidx == CMD_cdo || eap->cmdidx == CMD_ldo
2966 || eap->cmdidx == CMD_cfdo || eap->cmdidx == CMD_lfdo)
2967 {
2968 qf_size = qf_get_size(eap);
2969 if (qf_size <= 0 || eap->line1 > qf_size)
2970 buf = NULL;
2971 else
2972 {
2973 ex_cc(eap);
2974
2975 buf = curbuf;
2976 i = eap->line1 - 1;
2977 if (eap->addr_count <= 0)
2978 /* default is all the quickfix/location list entries */
2979 eap->line2 = qf_size;
2980 }
2981 }
2982#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002983 else
2984 setpcmark();
2985 listcmd_busy = TRUE; /* avoids setting pcmark below */
2986
Bram Moolenaare25bb902015-02-27 20:33:37 +01002987 while (!got_int && buf != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002988 {
2989 if (eap->cmdidx == CMD_argdo)
2990 {
2991 /* go to argument "i" */
2992 if (i == ARGCOUNT)
2993 break;
2994 /* Don't call do_argfile() when already there, it will try
2995 * reloading the file. */
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002996 if (curwin->w_arg_idx != i || !editing_arg_idx(curwin))
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002997 {
2998 /* Clear 'shm' to avoid that the file message overwrites
2999 * any output from the command. */
3000 p_shm_save = vim_strsave(p_shm);
3001 set_option_value((char_u *)"shm", 0L, (char_u *)"", 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003002 do_argfile(eap, i);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003003 set_option_value((char_u *)"shm", 0L, p_shm_save, 0);
3004 vim_free(p_shm_save);
3005 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003006 if (curwin->w_arg_idx != i)
3007 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003008 }
3009#ifdef FEAT_WINDOWS
3010 else if (eap->cmdidx == CMD_windo)
3011 {
Bram Moolenaar32466aa2006-02-24 23:53:04 +00003012 /* go to window "wp" */
3013 if (!win_valid(wp))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003014 break;
Bram Moolenaar32466aa2006-02-24 23:53:04 +00003015 win_goto(wp);
Bram Moolenaar41423242007-05-03 20:11:13 +00003016 if (curwin != wp)
3017 break; /* something must be wrong */
Bram Moolenaar32466aa2006-02-24 23:53:04 +00003018 wp = curwin->w_next;
3019 }
3020 else if (eap->cmdidx == CMD_tabdo)
3021 {
3022 /* go to window "tp" */
3023 if (!valid_tabpage(tp))
3024 break;
Bram Moolenaar49e649f2013-05-06 04:50:35 +02003025 goto_tabpage_tp(tp, TRUE, TRUE);
Bram Moolenaar32466aa2006-02-24 23:53:04 +00003026 tp = tp->tp_next;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003027 }
3028#endif
3029 else if (eap->cmdidx == CMD_bufdo)
3030 {
3031 /* Remember the number of the next listed buffer, in case
3032 * ":bwipe" is used or autocommands do something strange. */
3033 next_fnum = -1;
3034 for (buf = curbuf->b_next; buf != NULL; buf = buf->b_next)
3035 if (buf->b_p_bl)
3036 {
3037 next_fnum = buf->b_fnum;
3038 break;
3039 }
3040 }
3041
Bram Moolenaara162bc52015-01-07 16:54:21 +01003042 ++i;
3043
Bram Moolenaar071d4272004-06-13 20:20:40 +00003044 /* execute the command */
3045 do_cmdline(eap->arg, eap->getline, eap->cookie,
3046 DOCMD_VERBOSE + DOCMD_NOWAIT);
3047
3048 if (eap->cmdidx == CMD_bufdo)
3049 {
3050 /* Done? */
Bram Moolenaara162bc52015-01-07 16:54:21 +01003051 if (next_fnum < 0 || next_fnum > eap->line2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003052 break;
3053 /* Check if the buffer still exists. */
Bram Moolenaar29323592016-07-24 22:04:11 +02003054 FOR_ALL_BUFFERS(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003055 if (buf->b_fnum == next_fnum)
3056 break;
3057 if (buf == NULL)
3058 break;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003059
3060 /* Go to the next buffer. Clear 'shm' to avoid that the file
3061 * message overwrites any output from the command. */
3062 p_shm_save = vim_strsave(p_shm);
3063 set_option_value((char_u *)"shm", 0L, (char_u *)"", 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003064 goto_buffer(eap, DOBUF_FIRST, FORWARD, next_fnum);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003065 set_option_value((char_u *)"shm", 0L, p_shm_save, 0);
3066 vim_free(p_shm_save);
3067
Bram Moolenaaraa23b372015-09-08 18:46:31 +02003068 /* If autocommands took us elsewhere, quit here. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003069 if (curbuf->b_fnum != next_fnum)
3070 break;
3071 }
3072
Bram Moolenaaraa23b372015-09-08 18:46:31 +02003073#ifdef FEAT_QUICKFIX
3074 if (eap->cmdidx == CMD_cdo || eap->cmdidx == CMD_ldo
3075 || eap->cmdidx == CMD_cfdo || eap->cmdidx == CMD_lfdo)
3076 {
3077 if (i >= qf_size || i >= eap->line2)
3078 break;
3079
3080 qf_idx = qf_get_cur_idx(eap);
3081
3082 ex_cnext(eap);
3083
3084 /* If jumping to the next quickfix entry fails, quit here */
3085 if (qf_get_cur_idx(eap) == qf_idx)
3086 break;
3087 }
3088#endif
3089
Bram Moolenaar071d4272004-06-13 20:20:40 +00003090 if (eap->cmdidx == CMD_windo)
3091 {
3092 validate_cursor(); /* cursor may have moved */
3093#ifdef FEAT_SCROLLBIND
3094 /* required when 'scrollbind' has been set */
3095 if (curwin->w_p_scb)
3096 do_check_scrollbind(TRUE);
3097#endif
3098 }
Bram Moolenaara162bc52015-01-07 16:54:21 +01003099
3100#ifdef FEAT_WINDOWS
3101 if (eap->cmdidx == CMD_windo || eap->cmdidx == CMD_tabdo)
3102 if (i+1 > eap->line2)
3103 break;
3104#endif
3105 if (eap->cmdidx == CMD_argdo && i >= eap->line2)
3106 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003107 }
3108 listcmd_busy = FALSE;
3109 }
3110
3111#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
Bram Moolenaar6ac54292005-02-02 23:07:25 +00003112 if (save_ei != NULL)
3113 {
3114 au_event_restore(save_ei);
3115 apply_autocmds(EVENT_SYNTAX, curbuf->b_p_syn,
3116 curbuf->b_fname, TRUE, curbuf);
3117 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003118#endif
Bram Moolenaar6b1ee342014-08-06 18:17:11 +02003119#ifdef FEAT_CLIPBOARD
3120 end_global_changes();
3121#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003122}
3123
3124/*
3125 * Add files[count] to the arglist of the current window after arg "after".
3126 * The file names in files[count] must have been allocated and are taken over.
3127 * Files[] itself is not taken over.
3128 * Returns index of first added argument. Returns -1 when failed (out of mem).
3129 */
3130 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003131alist_add_list(
3132 int count,
3133 char_u **files,
3134 int after) /* where to add: 0 = before first one */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003135{
3136 int i;
Bram Moolenaara24f0a52016-01-17 19:39:00 +01003137 int old_argcount = ARGCOUNT;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003138
3139 if (ga_grow(&ALIST(curwin)->al_ga, count) == OK)
3140 {
3141 if (after < 0)
3142 after = 0;
3143 if (after > ARGCOUNT)
3144 after = ARGCOUNT;
3145 if (after < ARGCOUNT)
3146 mch_memmove(&(ARGLIST[after + count]), &(ARGLIST[after]),
3147 (ARGCOUNT - after) * sizeof(aentry_T));
3148 for (i = 0; i < count; ++i)
3149 {
3150 ARGLIST[after + i].ae_fname = files[i];
3151 ARGLIST[after + i].ae_fnum = buflist_add(files[i], BLN_LISTED);
3152 }
3153 ALIST(curwin)->al_ga.ga_len += count;
Bram Moolenaara24f0a52016-01-17 19:39:00 +01003154 if (old_argcount > 0 && curwin->w_arg_idx >= after)
3155 curwin->w_arg_idx += count;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003156 return after;
3157 }
3158
3159 for (i = 0; i < count; ++i)
3160 vim_free(files[i]);
3161 return -1;
3162}
3163
3164#endif /* FEAT_LISTCMDS */
3165
3166#ifdef FEAT_EVAL
3167/*
3168 * ":compiler[!] {name}"
3169 */
3170 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003171ex_compiler(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003172{
3173 char_u *buf;
3174 char_u *old_cur_comp = NULL;
3175 char_u *p;
3176
3177 if (*eap->arg == NUL)
3178 {
3179 /* List all compiler scripts. */
3180 do_cmdline_cmd((char_u *)"echo globpath(&rtp, 'compiler/*.vim')");
3181 /* ) keep the indenter happy... */
3182 }
3183 else
3184 {
3185 buf = alloc((unsigned)(STRLEN(eap->arg) + 14));
3186 if (buf != NULL)
3187 {
3188 if (eap->forceit)
3189 {
3190 /* ":compiler! {name}" sets global options */
3191 do_cmdline_cmd((char_u *)
3192 "command -nargs=* CompilerSet set <args>");
3193 }
3194 else
3195 {
3196 /* ":compiler! {name}" sets local options.
3197 * To remain backwards compatible "current_compiler" is always
3198 * used. A user's compiler plugin may set it, the distributed
3199 * plugin will then skip the settings. Afterwards set
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003200 * "b:current_compiler" and restore "current_compiler".
3201 * Explicitly prepend "g:" to make it work in a function. */
3202 old_cur_comp = get_var_value((char_u *)"g:current_compiler");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003203 if (old_cur_comp != NULL)
3204 old_cur_comp = vim_strsave(old_cur_comp);
3205 do_cmdline_cmd((char_u *)
3206 "command -nargs=* CompilerSet setlocal <args>");
3207 }
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003208 do_unlet((char_u *)"g:current_compiler", TRUE);
Bram Moolenaar2ce06f62005-01-31 19:19:04 +00003209 do_unlet((char_u *)"b:current_compiler", TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003210
3211 sprintf((char *)buf, "compiler/%s.vim", eap->arg);
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003212 if (source_runtime(buf, DIP_ALL) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003213 EMSG2(_("E666: compiler not supported: %s"), eap->arg);
3214 vim_free(buf);
3215
3216 do_cmdline_cmd((char_u *)":delcommand CompilerSet");
3217
3218 /* Set "b:current_compiler" from "current_compiler". */
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003219 p = get_var_value((char_u *)"g:current_compiler");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003220 if (p != NULL)
3221 set_internal_string_var((char_u *)"b:current_compiler", p);
3222
3223 /* Restore "current_compiler" for ":compiler {name}". */
3224 if (!eap->forceit)
3225 {
3226 if (old_cur_comp != NULL)
3227 {
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003228 set_internal_string_var((char_u *)"g:current_compiler",
Bram Moolenaar071d4272004-06-13 20:20:40 +00003229 old_cur_comp);
3230 vim_free(old_cur_comp);
3231 }
3232 else
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003233 do_unlet((char_u *)"g:current_compiler", TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003234 }
3235 }
3236 }
3237}
3238#endif
3239
3240/*
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003241 * ":runtime [what] {name}"
Bram Moolenaar071d4272004-06-13 20:20:40 +00003242 */
3243 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003244ex_runtime(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003245{
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003246 char_u *arg = eap->arg;
3247 char_u *p = skiptowhite(arg);
3248 int len = (int)(p - arg);
3249 int flags = eap->forceit ? DIP_ALL : 0;
3250
3251 if (STRNCMP(arg, "START", len) == 0)
3252 {
3253 flags += DIP_START + DIP_NORTP;
3254 arg = skipwhite(arg + len);
3255 }
3256 else if (STRNCMP(arg, "OPT", len) == 0)
3257 {
3258 flags += DIP_OPT + DIP_NORTP;
3259 arg = skipwhite(arg + len);
3260 }
3261 else if (STRNCMP(arg, "PACK", len) == 0)
3262 {
3263 flags += DIP_START + DIP_OPT + DIP_NORTP;
3264 arg = skipwhite(arg + len);
3265 }
3266 else if (STRNCMP(arg, "ALL", len) == 0)
3267 {
3268 flags += DIP_START + DIP_OPT;
3269 arg = skipwhite(arg + len);
3270 }
3271
3272 source_runtime(arg, flags);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003273}
3274
Bram Moolenaar071d4272004-06-13 20:20:40 +00003275 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003276source_callback(char_u *fname, void *cookie UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003277{
Bram Moolenaar910f66f2006-04-05 20:41:53 +00003278 (void)do_source(fname, FALSE, DOSO_NONE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003279}
3280
3281/*
3282 * Source the file "name" from all directories in 'runtimepath'.
3283 * "name" can contain wildcards.
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003284 * When "flags" has DIP_ALL: source all files, otherwise only the first one.
Bram Moolenaar91715872016-03-03 17:13:03 +01003285 *
Bram Moolenaar071d4272004-06-13 20:20:40 +00003286 * return FAIL when no file could be sourced, OK otherwise.
3287 */
3288 int
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003289source_runtime(char_u *name, int flags)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003290{
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003291 return do_in_runtimepath(name, flags, source_callback, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003292}
3293
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003294/*
3295 * Find the file "name" in all directories in "path" and invoke
3296 * "callback(fname, cookie)".
3297 * "name" can contain wildcards.
3298 * When "flags" has DIP_ALL: source all files, otherwise only the first one.
3299 * When "flags" has DIP_DIR: find directories instead of files.
3300 * When "flags" has DIP_ERR: give an error message if there is no match.
3301 *
3302 * return FAIL when no file could be sourced, OK otherwise.
3303 */
Bram Moolenaar6bef5302016-03-12 21:28:26 +01003304 int
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003305do_in_path(
3306 char_u *path,
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003307 char_u *name,
Bram Moolenaar91715872016-03-03 17:13:03 +01003308 int flags,
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003309 void (*callback)(char_u *fname, void *ck),
3310 void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003311{
3312 char_u *rtp;
3313 char_u *np;
3314 char_u *buf;
3315 char_u *rtp_copy;
3316 char_u *tail;
3317 int num_files;
3318 char_u **files;
3319 int i;
3320 int did_one = FALSE;
3321#ifdef AMIGA
3322 struct Process *proc = (struct Process *)FindTask(0L);
3323 APTR save_winptr = proc->pr_WindowPtr;
3324
3325 /* Avoid a requester here for a volume that doesn't exist. */
3326 proc->pr_WindowPtr = (APTR)-1L;
3327#endif
3328
3329 /* Make a copy of 'runtimepath'. Invoking the callback may change the
3330 * value. */
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003331 rtp_copy = vim_strsave(path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003332 buf = alloc(MAXPATHL);
3333 if (buf != NULL && rtp_copy != NULL)
3334 {
Bram Moolenaarc09a6d62013-06-10 21:27:29 +02003335 if (p_verbose > 1 && name != NULL)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003336 {
3337 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00003338 smsg((char_u *)_("Searching for \"%s\" in \"%s\""),
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003339 (char *)name, (char *)path);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003340 verbose_leave();
3341 }
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00003342
Bram Moolenaar071d4272004-06-13 20:20:40 +00003343 /* Loop over all entries in 'runtimepath'. */
3344 rtp = rtp_copy;
Bram Moolenaar91715872016-03-03 17:13:03 +01003345 while (*rtp != NUL && ((flags & DIP_ALL) || !did_one))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003346 {
Bram Moolenaar66459b72016-08-06 19:01:55 +02003347 size_t buflen;
3348
Bram Moolenaar071d4272004-06-13 20:20:40 +00003349 /* Copy the path from 'runtimepath' to buf[]. */
3350 copy_option_part(&rtp, buf, MAXPATHL, ",");
Bram Moolenaar66459b72016-08-06 19:01:55 +02003351 buflen = STRLEN(buf);
3352
3353 /* Skip after or non-after directories. */
3354 if (flags & (DIP_NOAFTER | DIP_AFTER))
3355 {
3356 int is_after = buflen >= 5
3357 && STRCMP(buf + buflen - 5, "after") == 0;
3358
3359 if ((is_after && (flags & DIP_NOAFTER))
3360 || (!is_after && (flags & DIP_AFTER)))
3361 continue;
3362 }
3363
Bram Moolenaarc09a6d62013-06-10 21:27:29 +02003364 if (name == NULL)
3365 {
3366 (*callback)(buf, (void *) &cookie);
3367 if (!did_one)
3368 did_one = (cookie == NULL);
3369 }
Bram Moolenaar66459b72016-08-06 19:01:55 +02003370 else if (buflen + STRLEN(name) + 2 < MAXPATHL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003371 {
3372 add_pathsep(buf);
3373 tail = buf + STRLEN(buf);
3374
3375 /* Loop over all patterns in "name" */
3376 np = name;
Bram Moolenaar91715872016-03-03 17:13:03 +01003377 while (*np != NUL && ((flags & DIP_ALL) || !did_one))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003378 {
3379 /* Append the pattern from "name" to buf[]. */
3380 copy_option_part(&np, tail, (int)(MAXPATHL - (tail - buf)),
3381 "\t ");
3382
3383 if (p_verbose > 2)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003384 {
3385 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00003386 smsg((char_u *)_("Searching for \"%s\""), buf);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003387 verbose_leave();
3388 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003389
3390 /* Expand wildcards, invoke the callback for each match. */
3391 if (gen_expand_wildcards(1, &buf, &num_files, &files,
Bram Moolenaar91715872016-03-03 17:13:03 +01003392 (flags & DIP_DIR) ? EW_DIR : EW_FILE) == OK)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003393 {
3394 for (i = 0; i < num_files; ++i)
3395 {
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00003396 (*callback)(files[i], cookie);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003397 did_one = TRUE;
Bram Moolenaar91715872016-03-03 17:13:03 +01003398 if (!(flags & DIP_ALL))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003399 break;
3400 }
3401 FreeWild(num_files, files);
3402 }
3403 }
3404 }
3405 }
3406 }
3407 vim_free(buf);
3408 vim_free(rtp_copy);
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003409 if (!did_one && name != NULL)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003410 {
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003411 char *basepath = path == p_rtp ? "runtimepath" : "packpath";
3412
3413 if (flags & DIP_ERR)
3414 EMSG3(_(e_dirnotf), basepath, name);
3415 else if (p_verbose > 0)
3416 {
3417 verbose_enter();
3418 smsg((char_u *)_("not found in '%s': \"%s\""), basepath, name);
3419 verbose_leave();
3420 }
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003421 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003422
3423#ifdef AMIGA
3424 proc->pr_WindowPtr = save_winptr;
3425#endif
3426
3427 return did_one ? OK : FAIL;
3428}
3429
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003430/*
3431 * Find "name" in 'runtimepath'. When found, invoke the callback function for
3432 * it: callback(fname, "cookie")
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003433 * When "flags" has DIP_ALL repeat for all matches, otherwise only the first
3434 * one is used.
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003435 * Returns OK when at least one match found, FAIL otherwise.
3436 *
3437 * If "name" is NULL calls callback for each entry in runtimepath. Cookie is
3438 * passed by reference in this case, setting it to NULL indicates that callback
3439 * has done its job.
3440 */
3441 int
3442do_in_runtimepath(
3443 char_u *name,
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003444 int flags,
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003445 void (*callback)(char_u *fname, void *ck),
3446 void *cookie)
3447{
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003448 int done = FAIL;
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003449 char_u *s;
3450 int len;
3451 char *start_dir = "pack/*/start/*/%s";
3452 char *opt_dir = "pack/*/opt/*/%s";
3453
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003454 if ((flags & DIP_NORTP) == 0)
3455 done = do_in_path(p_rtp, name, flags, callback, cookie);
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003456
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003457 if ((done == FAIL || (flags & DIP_ALL)) && (flags & DIP_START))
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003458 {
Bram Moolenaar1c8b4ed2016-03-17 21:51:03 +01003459 len = (int)(STRLEN(start_dir) + STRLEN(name));
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003460 s = alloc(len);
3461 if (s == NULL)
3462 return FAIL;
3463 vim_snprintf((char *)s, len, start_dir, name);
3464 done = do_in_path(p_pp, s, flags, callback, cookie);
3465 vim_free(s);
3466 }
3467
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003468 if ((done == FAIL || (flags & DIP_ALL)) && (flags & DIP_OPT))
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003469 {
Bram Moolenaar1c8b4ed2016-03-17 21:51:03 +01003470 len = (int)(STRLEN(opt_dir) + STRLEN(name));
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003471 s = alloc(len);
3472 if (s == NULL)
3473 return FAIL;
3474 vim_snprintf((char *)s, len, opt_dir, name);
3475 done = do_in_path(p_pp, s, flags, callback, cookie);
3476 vim_free(s);
3477 }
3478
3479 return done;
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003480}
3481
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003482/*
Bram Moolenaarf3654822016-03-04 22:12:23 +01003483 * Expand wildcards in "pat" and invoke do_source() for each match.
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003484 */
3485 static void
Bram Moolenaarf3654822016-03-04 22:12:23 +01003486source_all_matches(char_u *pat)
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003487{
Bram Moolenaarf3654822016-03-04 22:12:23 +01003488 int num_files;
3489 char_u **files;
3490 int i;
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003491
Bram Moolenaarf3654822016-03-04 22:12:23 +01003492 if (gen_expand_wildcards(1, &pat, &num_files, &files, EW_FILE) == OK)
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003493 {
Bram Moolenaarf3654822016-03-04 22:12:23 +01003494 for (i = 0; i < num_files; ++i)
3495 (void)do_source(files[i], FALSE, DOSO_NONE);
3496 FreeWild(num_files, files);
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003497 }
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003498}
3499
Bram Moolenaar49b27322016-04-05 21:13:00 +02003500/* used for "cookie" of add_pack_plugin() */
3501static int APP_ADD_DIR;
3502static int APP_LOAD;
3503static int APP_BOTH;
3504
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003505 static void
Bram Moolenaar91715872016-03-03 17:13:03 +01003506add_pack_plugin(char_u *fname, void *cookie)
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003507{
Bram Moolenaarf3654822016-03-04 22:12:23 +01003508 char_u *p4, *p3, *p2, *p1, *p;
3509 char_u *insp;
Bram Moolenaar91715872016-03-03 17:13:03 +01003510 int c;
3511 char_u *new_rtp;
3512 int keep;
Bram Moolenaarb0550662016-05-31 21:37:36 +02003513 size_t oldlen;
3514 size_t addlen;
Bram Moolenaara5702442016-05-24 19:37:29 +02003515 char_u *afterdir;
Bram Moolenaarb0550662016-05-31 21:37:36 +02003516 size_t afterlen = 0;
Bram Moolenaar91715872016-03-03 17:13:03 +01003517 char_u *ffname = fix_fname(fname);
Bram Moolenaarfef524b2016-07-02 22:07:22 +02003518 size_t fname_len;
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003519 char_u *buf = NULL;
3520 char_u *rtp_ffname;
3521 int match;
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003522
Bram Moolenaar91715872016-03-03 17:13:03 +01003523 if (ffname == NULL)
3524 return;
Bram Moolenaar49b27322016-04-05 21:13:00 +02003525 if (cookie != &APP_LOAD && strstr((char *)p_rtp, (char *)ffname) == NULL)
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003526 {
Bram Moolenaara5702442016-05-24 19:37:29 +02003527 /* directory is not yet in 'runtimepath', add it */
Bram Moolenaarf3654822016-03-04 22:12:23 +01003528 p4 = p3 = p2 = p1 = get_past_head(ffname);
Bram Moolenaar91acfff2017-03-12 19:22:36 +01003529 for (p = p1; *p; MB_PTR_ADV(p))
Bram Moolenaarf3654822016-03-04 22:12:23 +01003530 if (vim_ispathsep_nocolon(*p))
3531 {
3532 p4 = p3; p3 = p2; p2 = p1; p1 = p;
3533 }
3534
3535 /* now we have:
Bram Moolenaaraf1a0e32016-03-09 22:19:26 +01003536 * rtp/pack/name/start/name
3537 * p4 p3 p2 p1
Bram Moolenaarf3654822016-03-04 22:12:23 +01003538 *
3539 * find the part up to "pack" in 'runtimepath' */
3540 c = *p4;
3541 *p4 = NUL;
Bram Moolenaar4c5717e2016-07-01 15:39:40 +02003542
3543 /* Find "ffname" in "p_rtp", ignoring '/' vs '\' differences. */
3544 fname_len = STRLEN(ffname);
3545 insp = p_rtp;
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003546 buf = alloc(MAXPATHL);
3547 if (buf == NULL)
3548 goto theend;
3549 while (*insp != NUL)
Bram Moolenaar4c5717e2016-07-01 15:39:40 +02003550 {
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003551 copy_option_part(&insp, buf, MAXPATHL, ",");
3552 add_pathsep(buf);
3553 rtp_ffname = fix_fname(buf);
3554 if (rtp_ffname == NULL)
3555 goto theend;
3556 match = vim_fnamencmp(rtp_ffname, ffname, fname_len) == 0;
3557 vim_free(rtp_ffname);
3558 if (match)
Bram Moolenaar4c5717e2016-07-01 15:39:40 +02003559 break;
Bram Moolenaar4c5717e2016-07-01 15:39:40 +02003560 }
3561
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003562 if (*insp == NUL)
Bram Moolenaarf3654822016-03-04 22:12:23 +01003563 /* not found, append at the end */
3564 insp = p_rtp + STRLEN(p_rtp);
3565 else
Bram Moolenaarf3654822016-03-04 22:12:23 +01003566 /* append after the matching directory. */
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003567 --insp;
Bram Moolenaarf3654822016-03-04 22:12:23 +01003568 *p4 = c;
3569
Bram Moolenaara5702442016-05-24 19:37:29 +02003570 /* check if rtp/pack/name/start/name/after exists */
3571 afterdir = concat_fnames(ffname, (char_u *)"after", TRUE);
3572 if (afterdir != NULL && mch_isdir(afterdir))
3573 afterlen = STRLEN(afterdir) + 1; /* add one for comma */
3574
Bram Moolenaarb0550662016-05-31 21:37:36 +02003575 oldlen = STRLEN(p_rtp);
3576 addlen = STRLEN(ffname) + 1; /* add one for comma */
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003577 new_rtp = alloc((int)(oldlen + addlen + afterlen + 1));
3578 /* add one for NUL */
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003579 if (new_rtp == NULL)
Bram Moolenaarf3654822016-03-04 22:12:23 +01003580 goto theend;
3581 keep = (int)(insp - p_rtp);
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003582 mch_memmove(new_rtp, p_rtp, keep);
3583 new_rtp[keep] = ',';
Bram Moolenaara5702442016-05-24 19:37:29 +02003584 mch_memmove(new_rtp + keep + 1, ffname, addlen);
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003585 if (p_rtp[keep] != NUL)
Bram Moolenaara5702442016-05-24 19:37:29 +02003586 mch_memmove(new_rtp + keep + addlen, p_rtp + keep,
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003587 oldlen - keep + 1);
Bram Moolenaara5702442016-05-24 19:37:29 +02003588 if (afterlen > 0)
3589 {
3590 STRCAT(new_rtp, ",");
3591 STRCAT(new_rtp, afterdir);
3592 }
Bram Moolenaar863c1a92016-03-03 15:47:06 +01003593 set_option_value((char_u *)"rtp", 0L, new_rtp, 0);
3594 vim_free(new_rtp);
Bram Moolenaara5702442016-05-24 19:37:29 +02003595 vim_free(afterdir);
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003596 }
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003597
Bram Moolenaar49b27322016-04-05 21:13:00 +02003598 if (cookie != &APP_ADD_DIR)
Bram Moolenaarf3654822016-03-04 22:12:23 +01003599 {
Bram Moolenaar71fb0c12016-04-02 22:44:16 +02003600 static char *plugpat = "%s/plugin/**/*.vim";
Bram Moolenaarf3654822016-03-04 22:12:23 +01003601 static char *ftpat = "%s/ftdetect/*.vim";
3602 int len;
3603 char_u *pat;
3604
3605 len = (int)STRLEN(ffname) + (int)STRLEN(ftpat);
3606 pat = alloc(len);
3607 if (pat == NULL)
3608 goto theend;
3609 vim_snprintf((char *)pat, len, plugpat, ffname);
3610 source_all_matches(pat);
3611
3612#ifdef FEAT_AUTOCMD
3613 {
3614 char_u *cmd = vim_strsave((char_u *)"g:did_load_filetypes");
3615
3616 /* If runtime/filetype.vim wasn't loaded yet, the scripts will be
3617 * found when it loads. */
3618 if (cmd != NULL && eval_to_number(cmd) > 0)
3619 {
3620 do_cmdline_cmd((char_u *)"augroup filetypedetect");
3621 vim_snprintf((char *)pat, len, ftpat, ffname);
3622 source_all_matches(pat);
3623 do_cmdline_cmd((char_u *)"augroup END");
3624 }
3625 vim_free(cmd);
3626 }
3627#endif
Bram Moolenaarba8cd122016-03-19 14:16:39 +01003628 vim_free(pat);
Bram Moolenaarf3654822016-03-04 22:12:23 +01003629 }
3630
3631theend:
Bram Moolenaar2f9e5752017-02-05 16:07:54 +01003632 vim_free(buf);
Bram Moolenaarf3654822016-03-04 22:12:23 +01003633 vim_free(ffname);
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003634}
3635
Bram Moolenaarce876aa2017-06-04 17:47:42 +02003636/*
3637 * Add all packages in the "start" directory to 'runtimepath'.
3638 */
3639 void
3640add_pack_start_dirs(void)
3641{
3642 do_in_path(p_pp, (char_u *)"pack/*/start/*", DIP_ALL + DIP_DIR,
3643 add_pack_plugin, &APP_ADD_DIR);
3644}
3645
3646/*
3647 * Load plugins from all packages in the "start" directory.
3648 */
3649 void
3650load_start_packages(void)
3651{
3652 did_source_packages = TRUE;
3653 do_in_path(p_pp, (char_u *)"pack/*/start/*", DIP_ALL + DIP_DIR,
3654 add_pack_plugin, &APP_LOAD);
3655}
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003656
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003657/*
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003658 * ":packloadall"
Bram Moolenaarf3654822016-03-04 22:12:23 +01003659 * Find plugins in the package directories and source them.
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003660 */
3661 void
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003662ex_packloadall(exarg_T *eap)
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003663{
Bram Moolenaarce876aa2017-06-04 17:47:42 +02003664 if (!did_source_packages || eap->forceit)
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003665 {
Bram Moolenaar49b27322016-04-05 21:13:00 +02003666 /* First do a round to add all directories to 'runtimepath', then load
3667 * the plugins. This allows for plugins to use an autoload directory
3668 * of another plugin. */
Bram Moolenaarce876aa2017-06-04 17:47:42 +02003669 add_pack_start_dirs();
3670 load_start_packages();
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003671 }
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003672}
3673
3674/*
Bram Moolenaarf3654822016-03-04 22:12:23 +01003675 * ":packadd[!] {name}"
Bram Moolenaar91715872016-03-03 17:13:03 +01003676 */
3677 void
3678ex_packadd(exarg_T *eap)
3679{
3680 static char *plugpat = "pack/*/opt/%s";
3681 int len;
3682 char *pat;
3683
3684 len = (int)STRLEN(plugpat) + (int)STRLEN(eap->arg);
3685 pat = (char *)alloc(len);
3686 if (pat == NULL)
3687 return;
3688 vim_snprintf(pat, len, plugpat, eap->arg);
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003689 do_in_path(p_pp, (char_u *)pat, DIP_ALL + DIP_DIR + DIP_ERR,
Bram Moolenaar49b27322016-04-05 21:13:00 +02003690 add_pack_plugin, eap->forceit ? &APP_ADD_DIR : &APP_BOTH);
Bram Moolenaar91715872016-03-03 17:13:03 +01003691 vim_free(pat);
3692}
3693
Bram Moolenaar071d4272004-06-13 20:20:40 +00003694#if defined(FEAT_EVAL) && defined(FEAT_AUTOCMD)
3695/*
3696 * ":options"
3697 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003698 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003699ex_options(
3700 exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003701{
3702 cmd_source((char_u *)SYS_OPTWIN_FILE, NULL);
3703}
3704#endif
3705
Bram Moolenaarf42dd3c2017-01-28 16:06:38 +01003706#if defined(FEAT_PYTHON3) || defined(FEAT_PYTHON) || defined(PROTO)
3707
3708# if (defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)) || defined(PROTO)
3709/*
3710 * Detect Python 3 or 2, and initialize 'pyxversion'.
3711 */
3712 void
3713init_pyxversion(void)
3714{
3715 if (p_pyx == 0)
3716 {
3717 if (python3_enabled(FALSE))
3718 p_pyx = 3;
3719 else if (python_enabled(FALSE))
3720 p_pyx = 2;
3721 }
3722}
3723# endif
3724
3725/*
3726 * Does a file contain one of the following strings at the beginning of any
3727 * line?
3728 * "#!(any string)python2" => returns 2
3729 * "#!(any string)python3" => returns 3
3730 * "# requires python 2.x" => returns 2
3731 * "# requires python 3.x" => returns 3
3732 * otherwise return 0.
3733 */
3734 static int
3735requires_py_version(char_u *filename)
3736{
3737 FILE *file;
3738 int requires_py_version = 0;
3739 int i, lines;
3740
3741 lines = (int)p_mls;
3742 if (lines < 0)
3743 lines = 5;
3744
3745 file = mch_fopen((char *)filename, "r");
3746 if (file != NULL)
3747 {
3748 for (i = 0; i < lines; i++)
3749 {
3750 if (vim_fgets(IObuff, IOSIZE, file))
3751 break;
3752 if (i == 0 && IObuff[0] == '#' && IObuff[1] == '!')
3753 {
3754 /* Check shebang. */
3755 if (strstr((char *)IObuff + 2, "python2") != NULL)
3756 {
3757 requires_py_version = 2;
3758 break;
3759 }
3760 if (strstr((char *)IObuff + 2, "python3") != NULL)
3761 {
3762 requires_py_version = 3;
3763 break;
3764 }
3765 }
3766 IObuff[21] = '\0';
3767 if (STRCMP("# requires python 2.x", IObuff) == 0)
3768 {
3769 requires_py_version = 2;
3770 break;
3771 }
3772 if (STRCMP("# requires python 3.x", IObuff) == 0)
3773 {
3774 requires_py_version = 3;
3775 break;
3776 }
3777 }
3778 fclose(file);
3779 }
3780 return requires_py_version;
3781}
3782
3783
3784/*
3785 * Source a python file using the requested python version.
3786 */
3787 static void
3788source_pyx_file(exarg_T *eap, char_u *fname)
3789{
3790 exarg_T ex;
3791 int v = requires_py_version(fname);
3792
3793# if defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)
3794 init_pyxversion();
3795# endif
3796 if (v == 0)
3797 {
3798# if defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)
3799 /* user didn't choose a preference, 'pyx' is used */
3800 v = p_pyx;
3801# elif defined(FEAT_PYTHON)
3802 v = 2;
3803# elif defined(FEAT_PYTHON3)
3804 v = 3;
3805# endif
3806 }
3807
3808 /*
3809 * now source, if required python version is not supported show
3810 * unobtrusive message.
3811 */
3812 if (eap == NULL)
3813 vim_memset(&ex, 0, sizeof(ex));
3814 else
3815 ex = *eap;
3816 ex.arg = fname;
3817 ex.cmd = (char_u *)(v == 2 ? "pyfile" : "pyfile3");
3818
3819 if (v == 2)
3820 {
3821# ifdef FEAT_PYTHON
3822 ex_pyfile(&ex);
3823# else
3824 vim_snprintf((char *)IObuff, IOSIZE,
3825 _("W20: Required python version 2.x not supported, ignoring file: %s"),
3826 fname);
3827 MSG(IObuff);
3828# endif
3829 return;
3830 }
3831 else
3832 {
3833# ifdef FEAT_PYTHON3
3834 ex_py3file(&ex);
3835# else
3836 vim_snprintf((char *)IObuff, IOSIZE,
3837 _("W21: Required python version 3.x not supported, ignoring file: %s"),
3838 fname);
3839 MSG(IObuff);
3840# endif
3841 return;
3842 }
3843}
3844
3845/*
3846 * ":pyxfile {fname}"
3847 */
3848 void
3849ex_pyxfile(exarg_T *eap)
3850{
3851 source_pyx_file(eap, eap->arg);
3852}
3853
3854/*
3855 * ":pyx"
3856 */
3857 void
3858ex_pyx(exarg_T *eap)
3859{
3860# if defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)
3861 init_pyxversion();
3862 if (p_pyx == 2)
3863 ex_python(eap);
3864 else
3865 ex_py3(eap);
3866# elif defined(FEAT_PYTHON)
3867 ex_python(eap);
3868# elif defined(FEAT_PYTHON3)
3869 ex_py3(eap);
3870# endif
3871}
3872
3873/*
3874 * ":pyxdo"
3875 */
3876 void
3877ex_pyxdo(exarg_T *eap)
3878{
3879# if defined(FEAT_PYTHON) && defined(FEAT_PYTHON3)
3880 init_pyxversion();
3881 if (p_pyx == 2)
3882 ex_pydo(eap);
3883 else
3884 ex_py3do(eap);
3885# elif defined(FEAT_PYTHON)
3886 ex_pydo(eap);
3887# elif defined(FEAT_PYTHON3)
3888 ex_py3do(eap);
3889# endif
3890}
3891
3892#endif
3893
Bram Moolenaar071d4272004-06-13 20:20:40 +00003894/*
3895 * ":source {fname}"
3896 */
3897 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003898ex_source(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003899{
3900#ifdef FEAT_BROWSE
3901 if (cmdmod.browse)
3902 {
3903 char_u *fname = NULL;
3904
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00003905 fname = do_browse(0, (char_u *)_("Source Vim script"), eap->arg,
Bram Moolenaar071d4272004-06-13 20:20:40 +00003906 NULL, NULL, BROWSE_FILTER_MACROS, NULL);
3907 if (fname != NULL)
3908 {
3909 cmd_source(fname, eap);
3910 vim_free(fname);
3911 }
3912 }
3913 else
3914#endif
3915 cmd_source(eap->arg, eap);
3916}
3917
3918 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003919cmd_source(char_u *fname, exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003920{
3921 if (*fname == NUL)
3922 EMSG(_(e_argreq));
3923
Bram Moolenaar071d4272004-06-13 20:20:40 +00003924 else if (eap != NULL && eap->forceit)
Bram Moolenaar84a05ac2013-05-06 04:24:17 +02003925 /* ":source!": read Normal mode commands
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00003926 * Need to execute the commands directly. This is required at least
3927 * for:
Bram Moolenaar071d4272004-06-13 20:20:40 +00003928 * - ":g" command busy
3929 * - after ":argdo", ":windo" or ":bufdo"
3930 * - another command follows
3931 * - inside a loop
3932 */
3933 openscript(fname, global_busy || listcmd_busy || eap->nextcmd != NULL
3934#ifdef FEAT_EVAL
3935 || eap->cstack->cs_idx >= 0
3936#endif
3937 );
3938
3939 /* ":source" read ex commands */
Bram Moolenaar910f66f2006-04-05 20:41:53 +00003940 else if (do_source(fname, FALSE, DOSO_NONE) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003941 EMSG2(_(e_notopen), fname);
3942}
3943
3944/*
3945 * ":source" and associated commands.
3946 */
3947/*
3948 * Structure used to store info for each sourced file.
3949 * It is shared between do_source() and getsourceline().
3950 * This is required, because it needs to be handed to do_cmdline() and
3951 * sourcing can be done recursively.
3952 */
3953struct source_cookie
3954{
3955 FILE *fp; /* opened file for sourcing */
3956 char_u *nextline; /* if not NULL: line that was read ahead */
3957 int finished; /* ":finish" used */
Bram Moolenaar860cae12010-06-05 23:22:07 +02003958#if defined(USE_CRNL) || defined(USE_CR)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003959 int fileformat; /* EOL_UNKNOWN, EOL_UNIX or EOL_DOS */
3960 int error; /* TRUE if LF found after CR-LF */
3961#endif
3962#ifdef FEAT_EVAL
3963 linenr_T breakpoint; /* next line with breakpoint or zero */
3964 char_u *fname; /* name of sourced file */
3965 int dbg_tick; /* debug_tick when breakpoint was set */
3966 int level; /* top nesting level of sourced file */
3967#endif
3968#ifdef FEAT_MBYTE
3969 vimconv_T conv; /* type of conversion */
3970#endif
3971};
3972
3973#ifdef FEAT_EVAL
3974/*
3975 * Return the address holding the next breakpoint line for a source cookie.
3976 */
3977 linenr_T *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003978source_breakpoint(void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003979{
3980 return &((struct source_cookie *)cookie)->breakpoint;
3981}
3982
3983/*
3984 * Return the address holding the debug tick for a source cookie.
3985 */
3986 int *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003987source_dbg_tick(void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003988{
3989 return &((struct source_cookie *)cookie)->dbg_tick;
3990}
3991
3992/*
3993 * Return the nesting level for a source cookie.
3994 */
3995 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003996source_level(void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003997{
3998 return ((struct source_cookie *)cookie)->level;
3999}
4000#endif
4001
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01004002static char_u *get_one_sourceline(struct source_cookie *sp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004003
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004004#if (defined(WIN32) && defined(FEAT_CSCOPE)) || defined(HAVE_FD_CLOEXEC)
4005# define USE_FOPEN_NOINH
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01004006static FILE *fopen_noinh_readbin(char *filename);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004007
4008/*
4009 * Special function to open a file without handle inheritance.
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004010 * When possible the handle is closed on exec().
Bram Moolenaar071d4272004-06-13 20:20:40 +00004011 */
4012 static FILE *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004013fopen_noinh_readbin(char *filename)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004014{
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004015# ifdef WIN32
Bram Moolenaar8d8ef0b2010-01-20 21:41:47 +01004016 int fd_tmp = mch_open(filename, O_RDONLY | O_BINARY | O_NOINHERIT, 0);
4017# else
4018 int fd_tmp = mch_open(filename, O_RDONLY, 0);
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004019# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004020
4021 if (fd_tmp == -1)
4022 return NULL;
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004023
4024# ifdef HAVE_FD_CLOEXEC
4025 {
4026 int fdflags = fcntl(fd_tmp, F_GETFD);
4027 if (fdflags >= 0 && (fdflags & FD_CLOEXEC) == 0)
Bram Moolenaarcde88542015-08-11 19:14:00 +02004028 (void)fcntl(fd_tmp, F_SETFD, fdflags | FD_CLOEXEC);
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004029 }
4030# endif
4031
Bram Moolenaar071d4272004-06-13 20:20:40 +00004032 return fdopen(fd_tmp, READBIN);
4033}
4034#endif
4035
4036
4037/*
4038 * do_source: Read the file "fname" and execute its lines as EX commands.
4039 *
4040 * This function may be called recursively!
4041 *
4042 * return FAIL if file could not be opened, OK otherwise
4043 */
4044 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004045do_source(
4046 char_u *fname,
4047 int check_other, /* check for .vimrc and _vimrc */
4048 int is_vimrc) /* DOSO_ value */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004049{
4050 struct source_cookie cookie;
4051 char_u *save_sourcing_name;
4052 linenr_T save_sourcing_lnum;
4053 char_u *p;
4054 char_u *fname_exp;
Bram Moolenaar73881402009-02-04 16:50:47 +00004055 char_u *firstline = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004056 int retval = FAIL;
4057#ifdef FEAT_EVAL
4058 scid_T save_current_SID;
4059 static scid_T last_current_SID = 0;
4060 void *save_funccalp;
4061 int save_debug_break_level = debug_break_level;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004062 scriptitem_T *si = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004063# ifdef UNIX
Bram Moolenaar8767f522016-07-01 17:17:39 +02004064 stat_T st;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004065 int stat_ok;
4066# endif
4067#endif
4068#ifdef STARTUPTIME
4069 struct timeval tv_rel;
4070 struct timeval tv_start;
4071#endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00004072#ifdef FEAT_PROFILE
4073 proftime_T wait_start;
4074#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004075
Bram Moolenaar071d4272004-06-13 20:20:40 +00004076 p = expand_env_save(fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004077 if (p == NULL)
4078 return retval;
4079 fname_exp = fix_fname(p);
4080 vim_free(p);
4081 if (fname_exp == NULL)
4082 return retval;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004083 if (mch_isdir(fname_exp))
4084 {
Bram Moolenaar555b2802005-05-19 21:08:39 +00004085 smsg((char_u *)_("Cannot source a directory: \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004086 goto theend;
4087 }
4088
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00004089#ifdef FEAT_AUTOCMD
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00004090 /* Apply SourceCmd autocommands, they should get the file and source it. */
4091 if (has_autocmd(EVENT_SOURCECMD, fname_exp, NULL)
4092 && apply_autocmds(EVENT_SOURCECMD, fname_exp, fname_exp,
4093 FALSE, curbuf))
Bram Moolenaarf33943e2008-01-15 21:17:30 +00004094 {
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00004095# ifdef FEAT_EVAL
Bram Moolenaarf33943e2008-01-15 21:17:30 +00004096 retval = aborting() ? FAIL : OK;
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00004097# else
Bram Moolenaarf33943e2008-01-15 21:17:30 +00004098 retval = OK;
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00004099# endif
Bram Moolenaarf33943e2008-01-15 21:17:30 +00004100 goto theend;
4101 }
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00004102
4103 /* Apply SourcePre autocommands, they may get the file. */
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00004104 apply_autocmds(EVENT_SOURCEPRE, fname_exp, fname_exp, FALSE, curbuf);
4105#endif
4106
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004107#ifdef USE_FOPEN_NOINH
Bram Moolenaar071d4272004-06-13 20:20:40 +00004108 cookie.fp = fopen_noinh_readbin((char *)fname_exp);
4109#else
4110 cookie.fp = mch_fopen((char *)fname_exp, READBIN);
4111#endif
4112 if (cookie.fp == NULL && check_other)
4113 {
4114 /*
4115 * Try again, replacing file name ".vimrc" by "_vimrc" or vice versa,
4116 * and ".exrc" by "_exrc" or vice versa.
4117 */
4118 p = gettail(fname_exp);
4119 if ((*p == '.' || *p == '_')
4120 && (STRICMP(p + 1, "vimrc") == 0
4121 || STRICMP(p + 1, "gvimrc") == 0
4122 || STRICMP(p + 1, "exrc") == 0))
4123 {
4124 if (*p == '_')
4125 *p = '.';
4126 else
4127 *p = '_';
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01004128#ifdef USE_FOPEN_NOINH
Bram Moolenaar071d4272004-06-13 20:20:40 +00004129 cookie.fp = fopen_noinh_readbin((char *)fname_exp);
4130#else
4131 cookie.fp = mch_fopen((char *)fname_exp, READBIN);
4132#endif
4133 }
4134 }
4135
4136 if (cookie.fp == NULL)
4137 {
4138 if (p_verbose > 0)
4139 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004140 verbose_enter();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004141 if (sourcing_name == NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00004142 smsg((char_u *)_("could not source \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004143 else
4144 smsg((char_u *)_("line %ld: could not source \"%s\""),
Bram Moolenaar555b2802005-05-19 21:08:39 +00004145 sourcing_lnum, fname);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004146 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004147 }
4148 goto theend;
4149 }
4150
4151 /*
4152 * The file exists.
4153 * - In verbose mode, give a message.
4154 * - For a vimrc file, may want to set 'compatible', call vimrc_found().
4155 */
4156 if (p_verbose > 1)
4157 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004158 verbose_enter();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004159 if (sourcing_name == NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00004160 smsg((char_u *)_("sourcing \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004161 else
4162 smsg((char_u *)_("line %ld: sourcing \"%s\""),
Bram Moolenaar555b2802005-05-19 21:08:39 +00004163 sourcing_lnum, fname);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004164 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004165 }
Bram Moolenaar910f66f2006-04-05 20:41:53 +00004166 if (is_vimrc == DOSO_VIMRC)
4167 vimrc_found(fname_exp, (char_u *)"MYVIMRC");
4168 else if (is_vimrc == DOSO_GVIMRC)
4169 vimrc_found(fname_exp, (char_u *)"MYGVIMRC");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004170
4171#ifdef USE_CRNL
4172 /* If no automatic file format: Set default to CR-NL. */
4173 if (*p_ffs == NUL)
4174 cookie.fileformat = EOL_DOS;
4175 else
4176 cookie.fileformat = EOL_UNKNOWN;
4177 cookie.error = FALSE;
4178#endif
4179
4180#ifdef USE_CR
4181 /* If no automatic file format: Set default to CR. */
4182 if (*p_ffs == NUL)
4183 cookie.fileformat = EOL_MAC;
4184 else
4185 cookie.fileformat = EOL_UNKNOWN;
4186 cookie.error = FALSE;
4187#endif
4188
4189 cookie.nextline = NULL;
4190 cookie.finished = FALSE;
4191
4192#ifdef FEAT_EVAL
4193 /*
4194 * Check if this script has a breakpoint.
4195 */
4196 cookie.breakpoint = dbg_find_breakpoint(TRUE, fname_exp, (linenr_T)0);
4197 cookie.fname = fname_exp;
4198 cookie.dbg_tick = debug_tick;
4199
4200 cookie.level = ex_nesting_level;
4201#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004202
4203 /*
4204 * Keep the sourcing name/lnum, for recursive calls.
4205 */
4206 save_sourcing_name = sourcing_name;
4207 sourcing_name = fname_exp;
4208 save_sourcing_lnum = sourcing_lnum;
4209 sourcing_lnum = 0;
4210
Bram Moolenaar73881402009-02-04 16:50:47 +00004211#ifdef FEAT_MBYTE
4212 cookie.conv.vc_type = CONV_NONE; /* no conversion */
4213
4214 /* Read the first line so we can check for a UTF-8 BOM. */
4215 firstline = getsourceline(0, (void *)&cookie, 0);
4216 if (firstline != NULL && STRLEN(firstline) >= 3 && firstline[0] == 0xef
4217 && firstline[1] == 0xbb && firstline[2] == 0xbf)
4218 {
4219 /* Found BOM; setup conversion, skip over BOM and recode the line. */
4220 convert_setup(&cookie.conv, (char_u *)"utf-8", p_enc);
4221 p = string_convert(&cookie.conv, firstline + 3, NULL);
Bram Moolenaar4e2a5952009-02-05 19:48:25 +00004222 if (p == NULL)
4223 p = vim_strsave(firstline + 3);
Bram Moolenaar73881402009-02-04 16:50:47 +00004224 if (p != NULL)
4225 {
4226 vim_free(firstline);
4227 firstline = p;
4228 }
4229 }
4230#endif
4231
Bram Moolenaar071d4272004-06-13 20:20:40 +00004232#ifdef STARTUPTIME
Bram Moolenaarc4e41982010-01-19 16:31:47 +01004233 if (time_fd != NULL)
4234 time_push(&tv_rel, &tv_start);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004235#endif
4236
4237#ifdef FEAT_EVAL
Bram Moolenaar05159a02005-02-26 23:04:13 +00004238# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004239 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004240 prof_child_enter(&wait_start); /* entering a child now */
4241# endif
4242
4243 /* Don't use local function variables, if called from a function.
4244 * Also starts profiling timer for nested script. */
4245 save_funccalp = save_funccal();
4246
Bram Moolenaar071d4272004-06-13 20:20:40 +00004247 /*
4248 * Check if this script was sourced before to finds its SID.
4249 * If it's new, generate a new SID.
4250 */
4251 save_current_SID = current_SID;
4252# ifdef UNIX
4253 stat_ok = (mch_stat((char *)fname_exp, &st) >= 0);
4254# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00004255 for (current_SID = script_items.ga_len; current_SID > 0; --current_SID)
4256 {
4257 si = &SCRIPT_ITEM(current_SID);
4258 if (si->sn_name != NULL
Bram Moolenaar071d4272004-06-13 20:20:40 +00004259 && (
4260# ifdef UNIX
Bram Moolenaar7c626922005-02-07 22:01:03 +00004261 /* Compare dev/ino when possible, it catches symbolic
4262 * links. Also compare file names, the inode may change
4263 * when the file was edited. */
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00004264 ((stat_ok && si->sn_dev_valid)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004265 && (si->sn_dev == st.st_dev
4266 && si->sn_ino == st.st_ino)) ||
Bram Moolenaar071d4272004-06-13 20:20:40 +00004267# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00004268 fnamecmp(si->sn_name, fname_exp) == 0))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004269 break;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004270 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004271 if (current_SID == 0)
4272 {
4273 current_SID = ++last_current_SID;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004274 if (ga_grow(&script_items, (int)(current_SID - script_items.ga_len))
4275 == FAIL)
4276 goto almosttheend;
4277 while (script_items.ga_len < current_SID)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004278 {
Bram Moolenaar05159a02005-02-26 23:04:13 +00004279 ++script_items.ga_len;
4280 SCRIPT_ITEM(script_items.ga_len).sn_name = NULL;
4281# ifdef FEAT_PROFILE
4282 SCRIPT_ITEM(script_items.ga_len).sn_prof_on = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004283# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004284 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00004285 si = &SCRIPT_ITEM(current_SID);
4286 si->sn_name = fname_exp;
4287 fname_exp = NULL;
4288# ifdef UNIX
4289 if (stat_ok)
4290 {
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00004291 si->sn_dev_valid = TRUE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004292 si->sn_dev = st.st_dev;
4293 si->sn_ino = st.st_ino;
4294 }
4295 else
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00004296 si->sn_dev_valid = FALSE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004297# endif
4298
Bram Moolenaar071d4272004-06-13 20:20:40 +00004299 /* Allocate the local script variables to use for this script. */
4300 new_script_vars(current_SID);
4301 }
4302
Bram Moolenaar05159a02005-02-26 23:04:13 +00004303# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004304 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004305 {
4306 int forceit;
4307
4308 /* Check if we do profiling for this script. */
4309 if (!si->sn_prof_on && has_profiling(TRUE, si->sn_name, &forceit))
4310 {
4311 script_do_profile(si);
4312 si->sn_pr_force = forceit;
4313 }
4314 if (si->sn_prof_on)
4315 {
4316 ++si->sn_pr_count;
4317 profile_start(&si->sn_pr_start);
4318 profile_zero(&si->sn_pr_children);
4319 }
4320 }
4321# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004322#endif
4323
4324 /*
4325 * Call do_cmdline, which will call getsourceline() to get the lines.
4326 */
Bram Moolenaar73881402009-02-04 16:50:47 +00004327 do_cmdline(firstline, getsourceline, (void *)&cookie,
Bram Moolenaar071d4272004-06-13 20:20:40 +00004328 DOCMD_VERBOSE|DOCMD_NOWAIT|DOCMD_REPEAT);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004329 retval = OK;
Bram Moolenaar05159a02005-02-26 23:04:13 +00004330
4331#ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004332 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004333 {
4334 /* Get "si" again, "script_items" may have been reallocated. */
4335 si = &SCRIPT_ITEM(current_SID);
4336 if (si->sn_prof_on)
4337 {
4338 profile_end(&si->sn_pr_start);
4339 profile_sub_wait(&wait_start, &si->sn_pr_start);
4340 profile_add(&si->sn_pr_total, &si->sn_pr_start);
Bram Moolenaar1056d982006-03-09 22:37:52 +00004341 profile_self(&si->sn_pr_self, &si->sn_pr_start,
4342 &si->sn_pr_children);
Bram Moolenaar05159a02005-02-26 23:04:13 +00004343 }
4344 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004345#endif
4346
4347 if (got_int)
4348 EMSG(_(e_interr));
4349 sourcing_name = save_sourcing_name;
4350 sourcing_lnum = save_sourcing_lnum;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004351 if (p_verbose > 1)
4352 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004353 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00004354 smsg((char_u *)_("finished sourcing %s"), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004355 if (sourcing_name != NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00004356 smsg((char_u *)_("continuing in %s"), sourcing_name);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00004357 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00004358 }
4359#ifdef STARTUPTIME
Bram Moolenaarc4e41982010-01-19 16:31:47 +01004360 if (time_fd != NULL)
4361 {
4362 vim_snprintf((char *)IObuff, IOSIZE, "sourcing %s", fname);
4363 time_msg((char *)IObuff, &tv_start);
4364 time_pop(&tv_rel);
4365 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004366#endif
4367
4368#ifdef FEAT_EVAL
4369 /*
4370 * After a "finish" in debug mode, need to break at first command of next
4371 * sourced file.
4372 */
4373 if (save_debug_break_level > ex_nesting_level
4374 && debug_break_level == ex_nesting_level)
4375 ++debug_break_level;
4376#endif
4377
Bram Moolenaar05159a02005-02-26 23:04:13 +00004378#ifdef FEAT_EVAL
4379almosttheend:
4380 current_SID = save_current_SID;
4381 restore_funccal(save_funccalp);
4382# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004383 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004384 prof_child_exit(&wait_start); /* leaving a child now */
4385# endif
4386#endif
4387 fclose(cookie.fp);
4388 vim_free(cookie.nextline);
Bram Moolenaar73881402009-02-04 16:50:47 +00004389 vim_free(firstline);
Bram Moolenaar05159a02005-02-26 23:04:13 +00004390#ifdef FEAT_MBYTE
4391 convert_setup(&cookie.conv, NULL, NULL);
4392#endif
4393
Bram Moolenaar071d4272004-06-13 20:20:40 +00004394theend:
4395 vim_free(fname_exp);
4396 return retval;
4397}
4398
4399#if defined(FEAT_EVAL) || defined(PROTO)
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00004400
Bram Moolenaar071d4272004-06-13 20:20:40 +00004401/*
4402 * ":scriptnames"
4403 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004404 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004405ex_scriptnames(exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004406{
4407 int i;
4408
Bram Moolenaar05159a02005-02-26 23:04:13 +00004409 for (i = 1; i <= script_items.ga_len && !got_int; ++i)
4410 if (SCRIPT_ITEM(i).sn_name != NULL)
Bram Moolenaard58ea072011-06-26 04:25:30 +02004411 {
4412 home_replace(NULL, SCRIPT_ITEM(i).sn_name,
4413 NameBuff, MAXPATHL, TRUE);
4414 smsg((char_u *)"%3d: %s", i, NameBuff);
Bram Moolenaar970a1b82012-03-23 18:39:18 +01004415 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004416}
4417
4418# if defined(BACKSLASH_IN_FILENAME) || defined(PROTO)
4419/*
4420 * Fix slashes in the list of script names for 'shellslash'.
4421 */
4422 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004423scriptnames_slash_adjust(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004424{
4425 int i;
4426
Bram Moolenaar05159a02005-02-26 23:04:13 +00004427 for (i = 1; i <= script_items.ga_len; ++i)
4428 if (SCRIPT_ITEM(i).sn_name != NULL)
4429 slash_adjust(SCRIPT_ITEM(i).sn_name);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004430}
4431# endif
4432
4433/*
4434 * Get a pointer to a script name. Used for ":verbose set".
4435 */
4436 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004437get_scriptname(scid_T id)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004438{
4439 if (id == SID_MODELINE)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00004440 return (char_u *)_("modeline");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004441 if (id == SID_CMDARG)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00004442 return (char_u *)_("--cmd argument");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004443 if (id == SID_CARG)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00004444 return (char_u *)_("-c argument");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004445 if (id == SID_ENV)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00004446 return (char_u *)_("environment variable");
4447 if (id == SID_ERROR)
4448 return (char_u *)_("error handler");
Bram Moolenaar05159a02005-02-26 23:04:13 +00004449 return SCRIPT_ITEM(id).sn_name;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004450}
Bram Moolenaar05159a02005-02-26 23:04:13 +00004451
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00004452# if defined(EXITFREE) || defined(PROTO)
4453 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004454free_scriptnames(void)
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00004455{
4456 int i;
4457
4458 for (i = script_items.ga_len; i > 0; --i)
4459 vim_free(SCRIPT_ITEM(i).sn_name);
4460 ga_clear(&script_items);
4461}
4462# endif
4463
Bram Moolenaar071d4272004-06-13 20:20:40 +00004464#endif
4465
4466#if defined(USE_CR) || defined(PROTO)
4467
4468# if defined(__MSL__) && (__MSL__ >= 22)
4469/*
4470 * Newer version of the Metrowerks library handle DOS and UNIX files
4471 * without help.
4472 * Test with earlier versions, MSL 2.2 is the library supplied with
4473 * Codewarrior Pro 2.
4474 */
4475 char *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004476fgets_cr(char *s, int n, FILE *stream)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004477{
4478 return fgets(s, n, stream);
4479}
4480# else
4481/*
4482 * Version of fgets() which also works for lines ending in a <CR> only
4483 * (Macintosh format).
4484 * For older versions of the Metrowerks library.
4485 * At least CodeWarrior 9 needed this code.
4486 */
4487 char *
Bram Moolenaard14e00e2016-01-31 17:30:51 +01004488fgets_cr(char *s, int n, FILE *stream)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004489{
4490 int c = 0;
4491 int char_read = 0;
4492
4493 while (!feof(stream) && c != '\r' && c != '\n' && char_read < n - 1)
4494 {
4495 c = fgetc(stream);
4496 s[char_read++] = c;
4497 /* If the file is in DOS format, we need to skip a NL after a CR. I
4498 * thought it was the other way around, but this appears to work... */
4499 if (c == '\n')
4500 {
4501 c = fgetc(stream);
4502 if (c != '\r')
4503 ungetc(c, stream);
4504 }
4505 }
4506
4507 s[char_read] = 0;
4508 if (char_read == 0)
4509 return NULL;
4510
4511 if (feof(stream) && char_read == 1)
4512 return NULL;
4513
4514 return s;
4515}
4516# endif
4517#endif
4518
4519/*
4520 * Get one full line from a sourced file.
4521 * Called by do_cmdline() when it's called from do_source().
4522 *
4523 * Return a pointer to the line in allocated memory.
4524 * Return NULL for end-of-file or some error.
4525 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004526 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004527getsourceline(int c UNUSED, void *cookie, int indent UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004528{
4529 struct source_cookie *sp = (struct source_cookie *)cookie;
4530 char_u *line;
Bram Moolenaarc047b9a2012-02-11 20:40:55 +01004531 char_u *p;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004532
4533#ifdef FEAT_EVAL
4534 /* If breakpoints have been added/deleted need to check for it. */
4535 if (sp->dbg_tick < debug_tick)
4536 {
4537 sp->breakpoint = dbg_find_breakpoint(TRUE, sp->fname, sourcing_lnum);
4538 sp->dbg_tick = debug_tick;
4539 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00004540# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004541 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004542 script_line_end();
4543# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004544#endif
4545 /*
4546 * Get current line. If there is a read-ahead line, use it, otherwise get
4547 * one now.
4548 */
4549 if (sp->finished)
4550 line = NULL;
4551 else if (sp->nextline == NULL)
4552 line = get_one_sourceline(sp);
4553 else
4554 {
4555 line = sp->nextline;
4556 sp->nextline = NULL;
4557 ++sourcing_lnum;
4558 }
Bram Moolenaare2cc9702005-03-15 22:43:58 +00004559#ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004560 if (line != NULL && do_profiling == PROF_YES)
Bram Moolenaare2cc9702005-03-15 22:43:58 +00004561 script_line_start();
4562#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004563
4564 /* Only concatenate lines starting with a \ when 'cpoptions' doesn't
4565 * contain the 'C' flag. */
4566 if (line != NULL && (vim_strchr(p_cpo, CPO_CONCAT) == NULL))
4567 {
4568 /* compensate for the one line read-ahead */
4569 --sourcing_lnum;
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004570
4571 /* Get the next line and concatenate it when it starts with a
4572 * backslash. We always need to read the next line, keep it in
4573 * sp->nextline. */
4574 sp->nextline = get_one_sourceline(sp);
4575 if (sp->nextline != NULL && *(p = skipwhite(sp->nextline)) == '\\')
Bram Moolenaar071d4272004-06-13 20:20:40 +00004576 {
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004577 garray_T ga;
4578
Bram Moolenaarb549a732012-02-22 18:29:33 +01004579 ga_init2(&ga, (int)sizeof(char_u), 400);
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004580 ga_concat(&ga, line);
4581 ga_concat(&ga, p + 1);
4582 for (;;)
4583 {
4584 vim_free(sp->nextline);
4585 sp->nextline = get_one_sourceline(sp);
4586 if (sp->nextline == NULL)
4587 break;
4588 p = skipwhite(sp->nextline);
4589 if (*p != '\\')
4590 break;
Bram Moolenaarb549a732012-02-22 18:29:33 +01004591 /* Adjust the growsize to the current length to speed up
4592 * concatenating many lines. */
4593 if (ga.ga_len > 400)
4594 {
4595 if (ga.ga_len > 8000)
4596 ga.ga_growsize = 8000;
4597 else
4598 ga.ga_growsize = ga.ga_len;
4599 }
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004600 ga_concat(&ga, p + 1);
4601 }
4602 ga_append(&ga, NUL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004603 vim_free(line);
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004604 line = ga.ga_data;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004605 }
4606 }
4607
4608#ifdef FEAT_MBYTE
4609 if (line != NULL && sp->conv.vc_type != CONV_NONE)
4610 {
Bram Moolenaarc047b9a2012-02-11 20:40:55 +01004611 char_u *s;
4612
Bram Moolenaar071d4272004-06-13 20:20:40 +00004613 /* Convert the encoding of the script line. */
4614 s = string_convert(&sp->conv, line, NULL);
4615 if (s != NULL)
4616 {
4617 vim_free(line);
4618 line = s;
4619 }
4620 }
4621#endif
4622
4623#ifdef FEAT_EVAL
4624 /* Did we encounter a breakpoint? */
4625 if (sp->breakpoint != 0 && sp->breakpoint <= sourcing_lnum)
4626 {
4627 dbg_breakpoint(sp->fname, sourcing_lnum);
4628 /* Find next breakpoint. */
4629 sp->breakpoint = dbg_find_breakpoint(TRUE, sp->fname, sourcing_lnum);
4630 sp->dbg_tick = debug_tick;
4631 }
4632#endif
4633
4634 return line;
4635}
4636
4637 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004638get_one_sourceline(struct source_cookie *sp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004639{
4640 garray_T ga;
4641 int len;
4642 int c;
4643 char_u *buf;
4644#ifdef USE_CRNL
4645 int has_cr; /* CR-LF found */
4646#endif
4647#ifdef USE_CR
4648 char_u *scan;
4649#endif
4650 int have_read = FALSE;
4651
4652 /* use a growarray to store the sourced line */
Bram Moolenaar6ac54292005-02-02 23:07:25 +00004653 ga_init2(&ga, 1, 250);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004654
4655 /*
4656 * Loop until there is a finished line (or end-of-file).
4657 */
4658 sourcing_lnum++;
4659 for (;;)
4660 {
Bram Moolenaar6ac54292005-02-02 23:07:25 +00004661 /* make room to read at least 120 (more) characters */
4662 if (ga_grow(&ga, 120) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004663 break;
4664 buf = (char_u *)ga.ga_data;
4665
4666#ifdef USE_CR
4667 if (sp->fileformat == EOL_MAC)
4668 {
Bram Moolenaar86b68352004-12-27 21:59:20 +00004669 if (fgets_cr((char *)buf + ga.ga_len, ga.ga_maxlen - ga.ga_len,
4670 sp->fp) == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004671 break;
4672 }
4673 else
4674#endif
Bram Moolenaar86b68352004-12-27 21:59:20 +00004675 if (fgets((char *)buf + ga.ga_len, ga.ga_maxlen - ga.ga_len,
4676 sp->fp) == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004677 break;
Bram Moolenaar6ac54292005-02-02 23:07:25 +00004678 len = ga.ga_len + (int)STRLEN(buf + ga.ga_len);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004679#ifdef USE_CRNL
4680 /* Ignore a trailing CTRL-Z, when in Dos mode. Only recognize the
4681 * CTRL-Z by its own, or after a NL. */
4682 if ( (len == 1 || (len >= 2 && buf[len - 2] == '\n'))
4683 && sp->fileformat == EOL_DOS
4684 && buf[len - 1] == Ctrl_Z)
4685 {
4686 buf[len - 1] = NUL;
4687 break;
4688 }
4689#endif
4690
4691#ifdef USE_CR
4692 /* If the read doesn't stop on a new line, and there's
4693 * some CR then we assume a Mac format */
4694 if (sp->fileformat == EOL_UNKNOWN)
4695 {
4696 if (buf[len - 1] != '\n' && vim_strchr(buf, '\r') != NULL)
4697 sp->fileformat = EOL_MAC;
4698 else
4699 sp->fileformat = EOL_UNIX;
4700 }
4701
4702 if (sp->fileformat == EOL_MAC)
4703 {
4704 scan = vim_strchr(buf, '\r');
4705
4706 if (scan != NULL)
4707 {
4708 *scan = '\n';
4709 if (*(scan + 1) != 0)
4710 {
4711 *(scan + 1) = 0;
4712 fseek(sp->fp, (long)(scan - buf - len + 1), SEEK_CUR);
4713 }
4714 }
4715 len = STRLEN(buf);
4716 }
4717#endif
4718
4719 have_read = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004720 ga.ga_len = len;
4721
4722 /* If the line was longer than the buffer, read more. */
Bram Moolenaar86b68352004-12-27 21:59:20 +00004723 if (ga.ga_maxlen - ga.ga_len == 1 && buf[len - 1] != '\n')
Bram Moolenaar071d4272004-06-13 20:20:40 +00004724 continue;
4725
4726 if (len >= 1 && buf[len - 1] == '\n') /* remove trailing NL */
4727 {
4728#ifdef USE_CRNL
4729 has_cr = (len >= 2 && buf[len - 2] == '\r');
4730 if (sp->fileformat == EOL_UNKNOWN)
4731 {
4732 if (has_cr)
4733 sp->fileformat = EOL_DOS;
4734 else
4735 sp->fileformat = EOL_UNIX;
4736 }
4737
4738 if (sp->fileformat == EOL_DOS)
4739 {
4740 if (has_cr) /* replace trailing CR */
4741 {
4742 buf[len - 2] = '\n';
4743 --len;
4744 --ga.ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004745 }
4746 else /* lines like ":map xx yy^M" will have failed */
4747 {
4748 if (!sp->error)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00004749 {
Bram Moolenaar8820b482017-03-16 17:23:31 +01004750 msg_source(HL_ATTR(HLF_W));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004751 EMSG(_("W15: Warning: Wrong line separator, ^M may be missing"));
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00004752 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004753 sp->error = TRUE;
4754 sp->fileformat = EOL_UNIX;
4755 }
4756 }
4757#endif
4758 /* The '\n' is escaped if there is an odd number of ^V's just
4759 * before it, first set "c" just before the 'V's and then check
4760 * len&c parities (is faster than ((len-c)%2 == 0)) -- Acevedo */
4761 for (c = len - 2; c >= 0 && buf[c] == Ctrl_V; c--)
4762 ;
4763 if ((len & 1) != (c & 1)) /* escaped NL, read more */
4764 {
4765 sourcing_lnum++;
4766 continue;
4767 }
4768
4769 buf[len - 1] = NUL; /* remove the NL */
4770 }
4771
4772 /*
4773 * Check for ^C here now and then, so recursive :so can be broken.
4774 */
4775 line_breakcheck();
4776 break;
4777 }
4778
4779 if (have_read)
4780 return (char_u *)ga.ga_data;
4781
4782 vim_free(ga.ga_data);
4783 return NULL;
4784}
4785
Bram Moolenaar05159a02005-02-26 23:04:13 +00004786#if defined(FEAT_PROFILE) || defined(PROTO)
4787/*
4788 * Called when starting to read a script line.
4789 * "sourcing_lnum" must be correct!
4790 * When skipping lines it may not actually be executed, but we won't find out
4791 * until later and we need to store the time now.
4792 */
4793 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004794script_line_start(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004795{
4796 scriptitem_T *si;
4797 sn_prl_T *pp;
4798
4799 if (current_SID <= 0 || current_SID > script_items.ga_len)
4800 return;
4801 si = &SCRIPT_ITEM(current_SID);
4802 if (si->sn_prof_on && sourcing_lnum >= 1)
4803 {
Bram Moolenaar8c8de832008-06-24 22:58:06 +00004804 /* Grow the array before starting the timer, so that the time spent
Bram Moolenaar05159a02005-02-26 23:04:13 +00004805 * here isn't counted. */
Bram Moolenaarcde88542015-08-11 19:14:00 +02004806 (void)ga_grow(&si->sn_prl_ga, (int)(sourcing_lnum - si->sn_prl_ga.ga_len));
Bram Moolenaar05159a02005-02-26 23:04:13 +00004807 si->sn_prl_idx = sourcing_lnum - 1;
4808 while (si->sn_prl_ga.ga_len <= si->sn_prl_idx
4809 && si->sn_prl_ga.ga_len < si->sn_prl_ga.ga_maxlen)
4810 {
4811 /* Zero counters for a line that was not used before. */
4812 pp = &PRL_ITEM(si, si->sn_prl_ga.ga_len);
4813 pp->snp_count = 0;
4814 profile_zero(&pp->sn_prl_total);
4815 profile_zero(&pp->sn_prl_self);
4816 ++si->sn_prl_ga.ga_len;
4817 }
4818 si->sn_prl_execed = FALSE;
4819 profile_start(&si->sn_prl_start);
4820 profile_zero(&si->sn_prl_children);
4821 profile_get_wait(&si->sn_prl_wait);
4822 }
4823}
4824
4825/*
4826 * Called when actually executing a function line.
4827 */
4828 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004829script_line_exec(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004830{
4831 scriptitem_T *si;
4832
4833 if (current_SID <= 0 || current_SID > script_items.ga_len)
4834 return;
4835 si = &SCRIPT_ITEM(current_SID);
4836 if (si->sn_prof_on && si->sn_prl_idx >= 0)
4837 si->sn_prl_execed = TRUE;
4838}
4839
4840/*
4841 * Called when done with a function line.
4842 */
4843 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004844script_line_end(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004845{
4846 scriptitem_T *si;
4847 sn_prl_T *pp;
4848
4849 if (current_SID <= 0 || current_SID > script_items.ga_len)
4850 return;
4851 si = &SCRIPT_ITEM(current_SID);
4852 if (si->sn_prof_on && si->sn_prl_idx >= 0
4853 && si->sn_prl_idx < si->sn_prl_ga.ga_len)
4854 {
4855 if (si->sn_prl_execed)
4856 {
4857 pp = &PRL_ITEM(si, si->sn_prl_idx);
4858 ++pp->snp_count;
4859 profile_end(&si->sn_prl_start);
4860 profile_sub_wait(&si->sn_prl_wait, &si->sn_prl_start);
Bram Moolenaar05159a02005-02-26 23:04:13 +00004861 profile_add(&pp->sn_prl_total, &si->sn_prl_start);
Bram Moolenaar1056d982006-03-09 22:37:52 +00004862 profile_self(&pp->sn_prl_self, &si->sn_prl_start,
4863 &si->sn_prl_children);
Bram Moolenaar05159a02005-02-26 23:04:13 +00004864 }
4865 si->sn_prl_idx = -1;
4866 }
4867}
4868#endif
4869
Bram Moolenaar071d4272004-06-13 20:20:40 +00004870/*
4871 * ":scriptencoding": Set encoding conversion for a sourced script.
4872 * Without the multi-byte feature it's simply ignored.
4873 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004874 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004875ex_scriptencoding(exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004876{
4877#ifdef FEAT_MBYTE
4878 struct source_cookie *sp;
4879 char_u *name;
4880
4881 if (!getline_equal(eap->getline, eap->cookie, getsourceline))
4882 {
4883 EMSG(_("E167: :scriptencoding used outside of a sourced file"));
4884 return;
4885 }
4886
4887 if (*eap->arg != NUL)
4888 {
4889 name = enc_canonize(eap->arg);
4890 if (name == NULL) /* out of memory */
4891 return;
4892 }
4893 else
4894 name = eap->arg;
4895
4896 /* Setup for conversion from the specified encoding to 'encoding'. */
4897 sp = (struct source_cookie *)getline_cookie(eap->getline, eap->cookie);
4898 convert_setup(&sp->conv, name, p_enc);
4899
4900 if (name != eap->arg)
4901 vim_free(name);
4902#endif
4903}
4904
4905#if defined(FEAT_EVAL) || defined(PROTO)
4906/*
4907 * ":finish": Mark a sourced file as finished.
4908 */
4909 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004910ex_finish(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004911{
4912 if (getline_equal(eap->getline, eap->cookie, getsourceline))
4913 do_finish(eap, FALSE);
4914 else
4915 EMSG(_("E168: :finish used outside of a sourced file"));
4916}
4917
4918/*
4919 * Mark a sourced file as finished. Possibly makes the ":finish" pending.
4920 * Also called for a pending finish at the ":endtry" or after returning from
4921 * an extra do_cmdline(). "reanimate" is used in the latter case.
4922 */
4923 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004924do_finish(exarg_T *eap, int reanimate)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004925{
4926 int idx;
4927
4928 if (reanimate)
4929 ((struct source_cookie *)getline_cookie(eap->getline,
4930 eap->cookie))->finished = FALSE;
4931
4932 /*
4933 * Cleanup (and inactivate) conditionals, but stop when a try conditional
4934 * not in its finally clause (which then is to be executed next) is found.
4935 * In this case, make the ":finish" pending for execution at the ":endtry".
4936 * Otherwise, finish normally.
4937 */
4938 idx = cleanup_conditionals(eap->cstack, 0, TRUE);
4939 if (idx >= 0)
4940 {
4941 eap->cstack->cs_pending[idx] = CSTP_FINISH;
4942 report_make_pending(CSTP_FINISH, NULL);
4943 }
4944 else
4945 ((struct source_cookie *)getline_cookie(eap->getline,
4946 eap->cookie))->finished = TRUE;
4947}
4948
4949
4950/*
4951 * Return TRUE when a sourced file had the ":finish" command: Don't give error
4952 * message for missing ":endif".
4953 * Return FALSE when not sourcing a file.
4954 */
4955 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004956source_finished(
4957 char_u *(*fgetline)(int, void *, int),
4958 void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004959{
Bram Moolenaar89d40322006-08-29 15:30:07 +00004960 return (getline_equal(fgetline, cookie, getsourceline)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004961 && ((struct source_cookie *)getline_cookie(
Bram Moolenaar89d40322006-08-29 15:30:07 +00004962 fgetline, cookie))->finished);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004963}
4964#endif
4965
4966#if defined(FEAT_LISTCMDS) || defined(PROTO)
4967/*
4968 * ":checktime [buffer]"
4969 */
4970 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004971ex_checktime(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004972{
4973 buf_T *buf;
4974 int save_no_check_timestamps = no_check_timestamps;
4975
4976 no_check_timestamps = 0;
4977 if (eap->addr_count == 0) /* default is all buffers */
4978 check_timestamps(FALSE);
4979 else
4980 {
4981 buf = buflist_findnr((int)eap->line2);
4982 if (buf != NULL) /* cannot happen? */
4983 (void)buf_check_timestamp(buf, FALSE);
4984 }
4985 no_check_timestamps = save_no_check_timestamps;
4986}
4987#endif
4988
Bram Moolenaar071d4272004-06-13 20:20:40 +00004989#if (defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
4990 && (defined(FEAT_EVAL) || defined(FEAT_MULTI_LANG))
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004991# define HAVE_GET_LOCALE_VAL
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004992static char_u *get_locale_val(int what);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004993
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004994 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004995get_locale_val(int what)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004996{
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004997 char_u *loc;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004998
Bram Moolenaar48e330a2016-02-23 14:53:34 +01004999 /* Obtain the locale value from the libraries. */
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005000 loc = (char_u *)setlocale(what, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005001
Bram Moolenaar61660ea2006-04-07 21:40:07 +00005002# ifdef WIN32
Bram Moolenaar071d4272004-06-13 20:20:40 +00005003 if (loc != NULL)
5004 {
5005 char_u *p;
5006
Bram Moolenaar61660ea2006-04-07 21:40:07 +00005007 /* setocale() returns something like "LC_COLLATE=<name>;LC_..." when
5008 * one of the values (e.g., LC_CTYPE) differs. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005009 p = vim_strchr(loc, '=');
5010 if (p != NULL)
5011 {
5012 loc = ++p;
5013 while (*p != NUL) /* remove trailing newline */
5014 {
Bram Moolenaar61660ea2006-04-07 21:40:07 +00005015 if (*p < ' ' || *p == ';')
Bram Moolenaar071d4272004-06-13 20:20:40 +00005016 {
5017 *p = NUL;
5018 break;
5019 }
5020 ++p;
5021 }
5022 }
5023 }
5024# endif
5025
5026 return loc;
5027}
5028#endif
5029
5030
5031#ifdef WIN32
5032/*
5033 * On MS-Windows locale names are strings like "German_Germany.1252", but
5034 * gettext expects "de". Try to translate one into another here for a few
5035 * supported languages.
5036 */
5037 static char_u *
5038gettext_lang(char_u *name)
5039{
5040 int i;
5041 static char *(mtable[]) = {
5042 "afrikaans", "af",
5043 "czech", "cs",
5044 "dutch", "nl",
5045 "german", "de",
5046 "english_united kingdom", "en_GB",
5047 "spanish", "es",
5048 "french", "fr",
5049 "italian", "it",
5050 "japanese", "ja",
5051 "korean", "ko",
5052 "norwegian", "no",
5053 "polish", "pl",
5054 "russian", "ru",
5055 "slovak", "sk",
5056 "swedish", "sv",
5057 "ukrainian", "uk",
5058 "chinese_china", "zh_CN",
5059 "chinese_taiwan", "zh_TW",
5060 NULL};
5061
5062 for (i = 0; mtable[i] != NULL; i += 2)
5063 if (STRNICMP(mtable[i], name, STRLEN(mtable[i])) == 0)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005064 return (char_u *)mtable[i + 1];
Bram Moolenaar071d4272004-06-13 20:20:40 +00005065 return name;
5066}
5067#endif
5068
5069#if defined(FEAT_MULTI_LANG) || defined(PROTO)
5070/*
5071 * Obtain the current messages language. Used to set the default for
5072 * 'helplang'. May return NULL or an empty string.
5073 */
5074 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005075get_mess_lang(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005076{
5077 char_u *p;
5078
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02005079# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00005080# if defined(LC_MESSAGES)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005081 p = get_locale_val(LC_MESSAGES);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005082# else
5083 /* This is necessary for Win32, where LC_MESSAGES is not defined and $LANG
Bram Moolenaar61660ea2006-04-07 21:40:07 +00005084 * may be set to the LCID number. LC_COLLATE is the best guess, LC_TIME
5085 * and LC_MONETARY may be set differently for a Japanese working in the
5086 * US. */
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005087 p = get_locale_val(LC_COLLATE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005088# endif
5089# else
5090 p = mch_getenv((char_u *)"LC_ALL");
5091 if (p == NULL || *p == NUL)
5092 {
5093 p = mch_getenv((char_u *)"LC_MESSAGES");
5094 if (p == NULL || *p == NUL)
5095 p = mch_getenv((char_u *)"LANG");
5096 }
5097# endif
5098# ifdef WIN32
5099 p = gettext_lang(p);
5100# endif
5101 return p;
5102}
5103#endif
5104
Bram Moolenaardef9e822004-12-31 20:58:58 +00005105/* Complicated #if; matches with where get_mess_env() is used below. */
5106#if (defined(FEAT_EVAL) && !((defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
5107 && defined(LC_MESSAGES))) \
5108 || ((defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
5109 && (defined(FEAT_GETTEXT) || defined(FEAT_MBYTE)) \
5110 && !defined(LC_MESSAGES))
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01005111static char_u *get_mess_env(void);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005112
5113/*
5114 * Get the language used for messages from the environment.
5115 */
5116 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005117get_mess_env(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005118{
5119 char_u *p;
5120
5121 p = mch_getenv((char_u *)"LC_ALL");
5122 if (p == NULL || *p == NUL)
5123 {
5124 p = mch_getenv((char_u *)"LC_MESSAGES");
5125 if (p == NULL || *p == NUL)
5126 {
5127 p = mch_getenv((char_u *)"LANG");
5128 if (p != NULL && VIM_ISDIGIT(*p))
5129 p = NULL; /* ignore something like "1043" */
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02005130# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00005131 if (p == NULL || *p == NUL)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005132 p = get_locale_val(LC_CTYPE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005133# endif
5134 }
5135 }
5136 return p;
5137}
5138#endif
5139
5140#if defined(FEAT_EVAL) || defined(PROTO)
5141
5142/*
5143 * Set the "v:lang" variable according to the current locale setting.
5144 * Also do "v:lc_time"and "v:ctype".
5145 */
5146 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005147set_lang_var(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005148{
5149 char_u *loc;
5150
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02005151# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005152 loc = get_locale_val(LC_CTYPE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005153# else
5154 /* setlocale() not supported: use the default value */
5155 loc = (char_u *)"C";
5156# endif
5157 set_vim_var_string(VV_CTYPE, loc, -1);
5158
5159 /* When LC_MESSAGES isn't defined use the value from $LC_MESSAGES, fall
5160 * back to LC_CTYPE if it's empty. */
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02005161# if defined(HAVE_GET_LOCALE_VAL) && defined(LC_MESSAGES)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005162 loc = get_locale_val(LC_MESSAGES);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005163# else
5164 loc = get_mess_env();
5165# endif
5166 set_vim_var_string(VV_LANG, loc, -1);
5167
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02005168# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01005169 loc = get_locale_val(LC_TIME);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005170# endif
5171 set_vim_var_string(VV_LC_TIME, loc, -1);
5172}
5173#endif
5174
5175#if (defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
5176 && (defined(FEAT_GETTEXT) || defined(FEAT_MBYTE))
5177/*
5178 * ":language": Set the language (locale).
5179 */
5180 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005181ex_language(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005182{
5183 char *loc;
5184 char_u *p;
5185 char_u *name;
5186 int what = LC_ALL;
5187 char *whatstr = "";
5188#ifdef LC_MESSAGES
5189# define VIM_LC_MESSAGES LC_MESSAGES
5190#else
5191# define VIM_LC_MESSAGES 6789
5192#endif
5193
5194 name = eap->arg;
5195
5196 /* Check for "messages {name}", "ctype {name}" or "time {name}" argument.
5197 * Allow abbreviation, but require at least 3 characters to avoid
5198 * confusion with a two letter language name "me" or "ct". */
5199 p = skiptowhite(eap->arg);
Bram Moolenaar1c465442017-03-12 20:10:05 +01005200 if ((*p == NUL || VIM_ISWHITE(*p)) && p - eap->arg >= 3)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005201 {
5202 if (STRNICMP(eap->arg, "messages", p - eap->arg) == 0)
5203 {
5204 what = VIM_LC_MESSAGES;
5205 name = skipwhite(p);
5206 whatstr = "messages ";
5207 }
5208 else if (STRNICMP(eap->arg, "ctype", p - eap->arg) == 0)
5209 {
5210 what = LC_CTYPE;
5211 name = skipwhite(p);
5212 whatstr = "ctype ";
5213 }
5214 else if (STRNICMP(eap->arg, "time", p - eap->arg) == 0)
5215 {
5216 what = LC_TIME;
5217 name = skipwhite(p);
5218 whatstr = "time ";
5219 }
5220 }
5221
5222 if (*name == NUL)
5223 {
5224#ifndef LC_MESSAGES
5225 if (what == VIM_LC_MESSAGES)
5226 p = get_mess_env();
5227 else
5228#endif
5229 p = (char_u *)setlocale(what, NULL);
5230 if (p == NULL || *p == NUL)
5231 p = (char_u *)"Unknown";
5232 smsg((char_u *)_("Current %slanguage: \"%s\""), whatstr, p);
5233 }
5234 else
5235 {
5236#ifndef LC_MESSAGES
5237 if (what == VIM_LC_MESSAGES)
5238 loc = "";
5239 else
5240#endif
Bram Moolenaar8c8de832008-06-24 22:58:06 +00005241 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00005242 loc = setlocale(what, (char *)name);
Bram Moolenaar8c8de832008-06-24 22:58:06 +00005243#if defined(FEAT_FLOAT) && defined(LC_NUMERIC)
5244 /* Make sure strtod() uses a decimal point, not a comma. */
5245 setlocale(LC_NUMERIC, "C");
5246#endif
5247 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00005248 if (loc == NULL)
5249 EMSG2(_("E197: Cannot set language to \"%s\""), name);
5250 else
5251 {
5252#ifdef HAVE_NL_MSG_CAT_CNTR
5253 /* Need to do this for GNU gettext, otherwise cached translations
5254 * will be used again. */
5255 extern int _nl_msg_cat_cntr;
5256
5257 ++_nl_msg_cat_cntr;
5258#endif
Bram Moolenaarb8017e72007-05-10 18:59:07 +00005259 /* Reset $LC_ALL, otherwise it would overrule everything. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005260 vim_setenv((char_u *)"LC_ALL", (char_u *)"");
5261
5262 if (what != LC_TIME)
5263 {
5264 /* Tell gettext() what to translate to. It apparently doesn't
5265 * use the currently effective locale. Also do this when
5266 * FEAT_GETTEXT isn't defined, so that shell commands use this
5267 * value. */
5268 if (what == LC_ALL)
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00005269 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00005270 vim_setenv((char_u *)"LANG", name);
Bram Moolenaare3a0b532013-06-28 20:36:30 +02005271
5272 /* Clear $LANGUAGE because GNU gettext uses it. */
5273 vim_setenv((char_u *)"LANGUAGE", (char_u *)"");
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00005274# ifdef WIN32
5275 /* Apparently MS-Windows printf() may cause a crash when
5276 * we give it 8-bit text while it's expecting text in the
5277 * current locale. This call avoids that. */
5278 setlocale(LC_CTYPE, "C");
5279# endif
5280 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00005281 if (what != LC_CTYPE)
5282 {
5283 char_u *mname;
5284#ifdef WIN32
5285 mname = gettext_lang(name);
5286#else
5287 mname = name;
5288#endif
5289 vim_setenv((char_u *)"LC_MESSAGES", mname);
5290#ifdef FEAT_MULTI_LANG
5291 set_helplang_default(mname);
5292#endif
5293 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00005294 }
5295
5296# ifdef FEAT_EVAL
5297 /* Set v:lang, v:lc_time and v:ctype to the final result. */
5298 set_lang_var();
5299# endif
Bram Moolenaar6cc00c72011-10-20 21:41:09 +02005300# ifdef FEAT_TITLE
5301 maketitle();
5302# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00005303 }
5304 }
5305}
5306
5307# if defined(FEAT_CMDL_COMPL) || defined(PROTO)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005308
5309static char_u **locales = NULL; /* Array of all available locales */
Bram Moolenaarb8f7bd62017-01-12 20:28:25 +01005310
5311# ifndef WIN32
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005312static int did_init_locales = FALSE;
5313
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005314/* Return an array of strings for all available locales + NULL for the
5315 * last element. Return NULL in case of error. */
5316 static char_u **
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005317find_locales(void)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005318{
5319 garray_T locales_ga;
5320 char_u *loc;
5321
5322 /* Find all available locales by running command "locale -a". If this
5323 * doesn't work we won't have completion. */
5324 char_u *locale_a = get_cmd_output((char_u *)"locale -a",
Bram Moolenaar39c29ed2014-04-05 19:44:40 +02005325 NULL, SHELL_SILENT, NULL);
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005326 if (locale_a == NULL)
5327 return NULL;
5328 ga_init2(&locales_ga, sizeof(char_u *), 20);
5329
5330 /* Transform locale_a string where each locale is separated by "\n"
5331 * into an array of locale strings. */
5332 loc = (char_u *)strtok((char *)locale_a, "\n");
5333
5334 while (loc != NULL)
5335 {
5336 if (ga_grow(&locales_ga, 1) == FAIL)
5337 break;
5338 loc = vim_strsave(loc);
5339 if (loc == NULL)
5340 break;
5341
5342 ((char_u **)locales_ga.ga_data)[locales_ga.ga_len++] = loc;
5343 loc = (char_u *)strtok(NULL, "\n");
5344 }
5345 vim_free(locale_a);
5346 if (ga_grow(&locales_ga, 1) == FAIL)
5347 {
5348 ga_clear(&locales_ga);
5349 return NULL;
5350 }
5351 ((char_u **)locales_ga.ga_data)[locales_ga.ga_len] = NULL;
5352 return (char_u **)locales_ga.ga_data;
5353}
Bram Moolenaarb8f7bd62017-01-12 20:28:25 +01005354# endif
5355
5356/*
5357 * Lazy initialization of all available locales.
5358 */
5359 static void
5360init_locales(void)
5361{
5362# ifndef WIN32
5363 if (!did_init_locales)
5364 {
5365 did_init_locales = TRUE;
5366 locales = find_locales();
5367 }
5368# endif
5369}
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005370
5371# if defined(EXITFREE) || defined(PROTO)
5372 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005373free_locales(void)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005374{
5375 int i;
5376 if (locales != NULL)
5377 {
5378 for (i = 0; locales[i] != NULL; i++)
5379 vim_free(locales[i]);
5380 vim_free(locales);
5381 locales = NULL;
5382 }
5383}
5384# endif
5385
Bram Moolenaar071d4272004-06-13 20:20:40 +00005386/*
5387 * Function given to ExpandGeneric() to obtain the possible arguments of the
5388 * ":language" command.
5389 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005390 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005391get_lang_arg(expand_T *xp UNUSED, int idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005392{
5393 if (idx == 0)
5394 return (char_u *)"messages";
5395 if (idx == 1)
5396 return (char_u *)"ctype";
5397 if (idx == 2)
5398 return (char_u *)"time";
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005399
5400 init_locales();
5401 if (locales == NULL)
5402 return NULL;
5403 return locales[idx - 3];
5404}
5405
5406/*
5407 * Function given to ExpandGeneric() to obtain the available locales.
5408 */
5409 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005410get_locales(expand_T *xp UNUSED, int idx)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005411{
5412 init_locales();
5413 if (locales == NULL)
5414 return NULL;
5415 return locales[idx];
Bram Moolenaar071d4272004-06-13 20:20:40 +00005416}
5417# endif
5418
5419#endif