blob: 9adaea357d68c6dc9457b1f0dc2093cb3d25c17d [file] [log] [blame]
Bram Moolenaar071d4272004-06-13 20:20:40 +00001/* vi:set ts=8 sts=4 sw=4:
2 *
3 * VIM - Vi IMproved by Bram Moolenaar
4 *
5 * Do ":help uganda" in Vim to read copying and usage conditions.
6 * Do ":help credits" in Vim to see a list of people who contributed.
7 * See README.txt for an overview of the Vim source code.
8 */
9
10/*
11 * ex_cmds2.c: some more functions for command line commands
12 */
13
Bram Moolenaar071d4272004-06-13 20:20:40 +000014#include "vim.h"
Bram Moolenaar071d4272004-06-13 20:20:40 +000015#include "version.h"
16
Bram Moolenaarf28dbce2016-01-29 22:03:47 +010017static void cmd_source(char_u *fname, exarg_T *eap);
Bram Moolenaar071d4272004-06-13 20:20:40 +000018
Bram Moolenaar05159a02005-02-26 23:04:13 +000019#ifdef FEAT_EVAL
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +000020/* Growarray to store info about already sourced scripts.
Bram Moolenaar05159a02005-02-26 23:04:13 +000021 * For Unix also store the dev/ino, so that we don't have to stat() each
22 * script when going through the list. */
23typedef struct scriptitem_S
24{
25 char_u *sn_name;
26# ifdef UNIX
Bram Moolenaarbf0c4522009-05-16 19:16:33 +000027 int sn_dev_valid;
28 dev_t sn_dev;
Bram Moolenaar05159a02005-02-26 23:04:13 +000029 ino_t sn_ino;
30# endif
31# ifdef FEAT_PROFILE
32 int sn_prof_on; /* TRUE when script is/was profiled */
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +000033 int sn_pr_force; /* forceit: profile functions in this script */
Bram Moolenaar05159a02005-02-26 23:04:13 +000034 proftime_T sn_pr_child; /* time set when going into first child */
35 int sn_pr_nest; /* nesting for sn_pr_child */
36 /* profiling the script as a whole */
37 int sn_pr_count; /* nr of times sourced */
Bram Moolenaar8c8de832008-06-24 22:58:06 +000038 proftime_T sn_pr_total; /* time spent in script + children */
39 proftime_T sn_pr_self; /* time spent in script itself */
Bram Moolenaar05159a02005-02-26 23:04:13 +000040 proftime_T sn_pr_start; /* time at script start */
41 proftime_T sn_pr_children; /* time in children after script start */
42 /* profiling the script per line */
43 garray_T sn_prl_ga; /* things stored for every line */
44 proftime_T sn_prl_start; /* start time for current line */
45 proftime_T sn_prl_children; /* time spent in children for this line */
46 proftime_T sn_prl_wait; /* wait start time for current line */
47 int sn_prl_idx; /* index of line being timed; -1 if none */
48 int sn_prl_execed; /* line being timed was executed */
49# endif
50} scriptitem_T;
51
52static garray_T script_items = {0, 0, sizeof(scriptitem_T), 4, NULL};
53#define SCRIPT_ITEM(id) (((scriptitem_T *)script_items.ga_data)[(id) - 1])
54
55# ifdef FEAT_PROFILE
56/* Struct used in sn_prl_ga for every line of a script. */
57typedef struct sn_prl_S
58{
59 int snp_count; /* nr of times line was executed */
Bram Moolenaar8c8de832008-06-24 22:58:06 +000060 proftime_T sn_prl_total; /* time spent in a line + children */
61 proftime_T sn_prl_self; /* time spent in a line itself */
Bram Moolenaar05159a02005-02-26 23:04:13 +000062} sn_prl_T;
63
64# define PRL_ITEM(si, idx) (((sn_prl_T *)(si)->sn_prl_ga.ga_data)[(idx)])
65# endif
66#endif
67
Bram Moolenaar071d4272004-06-13 20:20:40 +000068#if defined(FEAT_EVAL) || defined(PROTO)
69static int debug_greedy = FALSE; /* batch mode debugging: don't save
70 and restore typeahead. */
Bram Moolenaarf1f60f82016-01-16 15:40:53 +010071static int get_maxbacktrace_level(void);
72static void do_setdebugtracelevel(char_u *arg);
73static void do_checkbacktracelevel(void);
74static void do_showbacktrace(char_u *cmd);
Bram Moolenaar071d4272004-06-13 20:20:40 +000075
76/*
77 * do_debug(): Debug mode.
78 * Repeatedly get Ex commands, until told to continue normal execution.
79 */
80 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +010081do_debug(char_u *cmd)
Bram Moolenaar071d4272004-06-13 20:20:40 +000082{
83 int save_msg_scroll = msg_scroll;
84 int save_State = State;
85 int save_did_emsg = did_emsg;
86 int save_cmd_silent = cmd_silent;
87 int save_msg_silent = msg_silent;
88 int save_emsg_silent = emsg_silent;
89 int save_redir_off = redir_off;
90 tasave_T typeaheadbuf;
Bram Moolenaaree3f7a52008-01-01 13:17:56 +000091 int typeahead_saved = FALSE;
Bram Moolenaar383c6f52008-01-04 15:01:07 +000092 int save_ignore_script = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +000093 int save_ex_normal_busy;
Bram Moolenaar071d4272004-06-13 20:20:40 +000094 int n;
95 char_u *cmdline = NULL;
96 char_u *p;
97 char *tail = NULL;
98 static int last_cmd = 0;
99#define CMD_CONT 1
100#define CMD_NEXT 2
101#define CMD_STEP 3
102#define CMD_FINISH 4
103#define CMD_QUIT 5
104#define CMD_INTERRUPT 6
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100105#define CMD_BACKTRACE 7
106#define CMD_FRAME 8
107#define CMD_UP 9
108#define CMD_DOWN 10
Bram Moolenaar071d4272004-06-13 20:20:40 +0000109
110#ifdef ALWAYS_USE_GUI
111 /* Can't do this when there is no terminal for input/output. */
112 if (!gui.in_use)
113 {
114 /* Break as soon as possible. */
115 debug_break_level = 9999;
116 return;
117 }
118#endif
119
120 /* Make sure we are in raw mode and start termcap mode. Might have side
121 * effects... */
122 settmode(TMODE_RAW);
123 starttermcap();
124
125 ++RedrawingDisabled; /* don't redisplay the window */
126 ++no_wait_return; /* don't wait for return */
127 did_emsg = FALSE; /* don't use error from debugged stuff */
128 cmd_silent = FALSE; /* display commands */
129 msg_silent = FALSE; /* display messages */
130 emsg_silent = FALSE; /* display error messages */
131 redir_off = TRUE; /* don't redirect debug commands */
132
133 State = NORMAL;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000134
135 if (!debug_did_msg)
136 MSG(_("Entering Debug mode. Type \"cont\" to continue."));
137 if (sourcing_name != NULL)
138 msg(sourcing_name);
139 if (sourcing_lnum != 0)
Bram Moolenaar555b2802005-05-19 21:08:39 +0000140 smsg((char_u *)_("line %ld: %s"), (long)sourcing_lnum, cmd);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000141 else
Bram Moolenaar555b2802005-05-19 21:08:39 +0000142 smsg((char_u *)_("cmd: %s"), cmd);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000143
144 /*
145 * Repeat getting a command and executing it.
146 */
147 for (;;)
148 {
149 msg_scroll = TRUE;
150 need_wait_return = FALSE;
Bram Moolenaar85b11762016-02-27 18:13:23 +0100151
Bram Moolenaar071d4272004-06-13 20:20:40 +0000152 /* Save the current typeahead buffer and replace it with an empty one.
153 * This makes sure we get input from the user here and don't interfere
154 * with the commands being executed. Reset "ex_normal_busy" to avoid
155 * the side effects of using ":normal". Save the stuff buffer and make
Bram Moolenaaree3f7a52008-01-01 13:17:56 +0000156 * it empty. Set ignore_script to avoid reading from script input. */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000157 save_ex_normal_busy = ex_normal_busy;
158 ex_normal_busy = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000159 if (!debug_greedy)
Bram Moolenaaree3f7a52008-01-01 13:17:56 +0000160 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000161 save_typeahead(&typeaheadbuf);
Bram Moolenaaree3f7a52008-01-01 13:17:56 +0000162 typeahead_saved = TRUE;
163 save_ignore_script = ignore_script;
164 ignore_script = TRUE;
165 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000166
Bram Moolenaardc303bc2016-05-17 17:45:38 +0200167 vim_free(cmdline);
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +0000168 cmdline = getcmdline_prompt('>', NULL, 0, EXPAND_NOTHING, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000169
Bram Moolenaaree3f7a52008-01-01 13:17:56 +0000170 if (typeahead_saved)
171 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000172 restore_typeahead(&typeaheadbuf);
Bram Moolenaaree3f7a52008-01-01 13:17:56 +0000173 ignore_script = save_ignore_script;
174 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000175 ex_normal_busy = save_ex_normal_busy;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000176
177 cmdline_row = msg_row;
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100178 msg_starthere();
Bram Moolenaar071d4272004-06-13 20:20:40 +0000179 if (cmdline != NULL)
180 {
181 /* If this is a debug command, set "last_cmd".
182 * If not, reset "last_cmd".
183 * For a blank line use previous command. */
184 p = skipwhite(cmdline);
185 if (*p != NUL)
186 {
187 switch (*p)
188 {
189 case 'c': last_cmd = CMD_CONT;
190 tail = "ont";
191 break;
192 case 'n': last_cmd = CMD_NEXT;
193 tail = "ext";
194 break;
195 case 's': last_cmd = CMD_STEP;
196 tail = "tep";
197 break;
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100198 case 'f':
199 last_cmd = 0;
200 if (p[1] == 'r')
201 {
202 last_cmd = CMD_FRAME;
203 tail = "rame";
204 }
205 else
206 {
207 last_cmd = CMD_FINISH;
208 tail = "inish";
209 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000210 break;
211 case 'q': last_cmd = CMD_QUIT;
212 tail = "uit";
213 break;
214 case 'i': last_cmd = CMD_INTERRUPT;
215 tail = "nterrupt";
216 break;
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100217 case 'b': last_cmd = CMD_BACKTRACE;
218 if (p[1] == 't')
219 tail = "t";
220 else
221 tail = "acktrace";
222 break;
223 case 'w': last_cmd = CMD_BACKTRACE;
224 tail = "here";
225 break;
226 case 'u': last_cmd = CMD_UP;
227 tail = "p";
228 break;
229 case 'd': last_cmd = CMD_DOWN;
230 tail = "own";
231 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000232 default: last_cmd = 0;
233 }
234 if (last_cmd != 0)
235 {
236 /* Check that the tail matches. */
237 ++p;
238 while (*p != NUL && *p == *tail)
239 {
240 ++p;
241 ++tail;
242 }
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100243 if (ASCII_ISALPHA(*p) && last_cmd != CMD_FRAME)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000244 last_cmd = 0;
245 }
246 }
247
248 if (last_cmd != 0)
249 {
250 /* Execute debug command: decided where to break next and
251 * return. */
252 switch (last_cmd)
253 {
254 case CMD_CONT:
255 debug_break_level = -1;
256 break;
257 case CMD_NEXT:
258 debug_break_level = ex_nesting_level;
259 break;
260 case CMD_STEP:
261 debug_break_level = 9999;
262 break;
263 case CMD_FINISH:
264 debug_break_level = ex_nesting_level - 1;
265 break;
266 case CMD_QUIT:
267 got_int = TRUE;
268 debug_break_level = -1;
269 break;
270 case CMD_INTERRUPT:
271 got_int = TRUE;
272 debug_break_level = 9999;
273 /* Do not repeat ">interrupt" cmd, continue stepping. */
274 last_cmd = CMD_STEP;
275 break;
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100276 case CMD_BACKTRACE:
277 do_showbacktrace(cmd);
278 continue;
279 case CMD_FRAME:
280 if (*p == NUL)
281 {
282 do_showbacktrace(cmd);
283 }
284 else
285 {
286 p = skipwhite(p);
287 do_setdebugtracelevel(p);
288 }
289 continue;
290 case CMD_UP:
291 debug_backtrace_level++;
292 do_checkbacktracelevel();
293 continue;
294 case CMD_DOWN:
295 debug_backtrace_level--;
296 do_checkbacktracelevel();
297 continue;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000298 }
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100299 /* Going out reset backtrace_level */
300 debug_backtrace_level = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000301 break;
302 }
303
304 /* don't debug this command */
305 n = debug_break_level;
306 debug_break_level = -1;
307 (void)do_cmdline(cmdline, getexline, NULL,
308 DOCMD_VERBOSE|DOCMD_EXCRESET);
309 debug_break_level = n;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000310 }
311 lines_left = Rows - 1;
312 }
313 vim_free(cmdline);
314
315 --RedrawingDisabled;
316 --no_wait_return;
317 redraw_all_later(NOT_VALID);
318 need_wait_return = FALSE;
319 msg_scroll = save_msg_scroll;
320 lines_left = Rows - 1;
321 State = save_State;
322 did_emsg = save_did_emsg;
323 cmd_silent = save_cmd_silent;
324 msg_silent = save_msg_silent;
325 emsg_silent = save_emsg_silent;
326 redir_off = save_redir_off;
327
328 /* Only print the message again when typing a command before coming back
329 * here. */
330 debug_did_msg = TRUE;
331}
332
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100333 static int
334get_maxbacktrace_level(void)
335{
336 char *p, *q;
Bram Moolenaardc633cf2016-04-23 14:33:19 +0200337 int maxbacktrace = 0;
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100338
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100339 if (sourcing_name != NULL)
340 {
341 p = (char *)sourcing_name;
342 while ((q = strstr(p, "..")) != NULL)
343 {
344 p = q + 2;
345 maxbacktrace++;
346 }
347 }
348 return maxbacktrace;
349}
350
351 static void
352do_setdebugtracelevel(char_u *arg)
353{
354 int level;
355
356 level = atoi((char *)arg);
357 if (*arg == '+' || level < 0)
358 debug_backtrace_level += level;
359 else
360 debug_backtrace_level = level;
361
362 do_checkbacktracelevel();
363}
364
365 static void
366do_checkbacktracelevel(void)
367{
368 if (debug_backtrace_level < 0)
369 {
370 debug_backtrace_level = 0;
371 MSG(_("frame is zero"));
372 }
373 else
374 {
375 int max = get_maxbacktrace_level();
376
377 if (debug_backtrace_level > max)
378 {
379 debug_backtrace_level = max;
380 smsg((char_u *)_("frame at highest level: %d"), max);
381 }
382 }
383}
384
385 static void
386do_showbacktrace(char_u *cmd)
387{
388 char *cur;
389 char *next;
390 int i = 0;
391 int max = get_maxbacktrace_level();
392
393 if (sourcing_name != NULL)
394 {
395 cur = (char *)sourcing_name;
396 while (!got_int)
397 {
398 next = strstr(cur, "..");
399 if (next != NULL)
400 *next = NUL;
401 if (i == max - debug_backtrace_level)
402 smsg((char_u *)"->%d %s", max - i, cur);
403 else
404 smsg((char_u *)" %d %s", max - i, cur);
405 ++i;
406 if (next == NULL)
407 break;
408 *next = '.';
409 cur = next + 2;
410 }
411 }
412 if (sourcing_lnum != 0)
413 smsg((char_u *)_("line %ld: %s"), (long)sourcing_lnum, cmd);
414 else
415 smsg((char_u *)_("cmd: %s"), cmd);
416}
417
Bram Moolenaar071d4272004-06-13 20:20:40 +0000418/*
419 * ":debug".
420 */
421 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100422ex_debug(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000423{
424 int debug_break_level_save = debug_break_level;
425
426 debug_break_level = 9999;
427 do_cmdline_cmd(eap->arg);
428 debug_break_level = debug_break_level_save;
429}
430
431static char_u *debug_breakpoint_name = NULL;
432static linenr_T debug_breakpoint_lnum;
433
434/*
435 * When debugging or a breakpoint is set on a skipped command, no debug prompt
436 * is shown by do_one_cmd(). This situation is indicated by debug_skipped, and
437 * debug_skipped_name is then set to the source name in the breakpoint case. If
438 * a skipped command decides itself that a debug prompt should be displayed, it
439 * can do so by calling dbg_check_skipped().
440 */
441static int debug_skipped;
442static char_u *debug_skipped_name;
443
444/*
445 * Go to debug mode when a breakpoint was encountered or "ex_nesting_level" is
446 * at or below the break level. But only when the line is actually
447 * executed. Return TRUE and set breakpoint_name for skipped commands that
448 * decide to execute something themselves.
449 * Called from do_one_cmd() before executing a command.
450 */
451 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100452dbg_check_breakpoint(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000453{
454 char_u *p;
455
456 debug_skipped = FALSE;
457 if (debug_breakpoint_name != NULL)
458 {
459 if (!eap->skip)
460 {
461 /* replace K_SNR with "<SNR>" */
462 if (debug_breakpoint_name[0] == K_SPECIAL
463 && debug_breakpoint_name[1] == KS_EXTRA
464 && debug_breakpoint_name[2] == (int)KE_SNR)
465 p = (char_u *)"<SNR>";
466 else
467 p = (char_u *)"";
Bram Moolenaar555b2802005-05-19 21:08:39 +0000468 smsg((char_u *)_("Breakpoint in \"%s%s\" line %ld"),
469 p,
Bram Moolenaar071d4272004-06-13 20:20:40 +0000470 debug_breakpoint_name + (*p == NUL ? 0 : 3),
471 (long)debug_breakpoint_lnum);
472 debug_breakpoint_name = NULL;
473 do_debug(eap->cmd);
474 }
475 else
476 {
477 debug_skipped = TRUE;
478 debug_skipped_name = debug_breakpoint_name;
479 debug_breakpoint_name = NULL;
480 }
481 }
482 else if (ex_nesting_level <= debug_break_level)
483 {
484 if (!eap->skip)
485 do_debug(eap->cmd);
486 else
487 {
488 debug_skipped = TRUE;
489 debug_skipped_name = NULL;
490 }
491 }
492}
493
494/*
495 * Go to debug mode if skipped by dbg_check_breakpoint() because eap->skip was
496 * set. Return TRUE when the debug mode is entered this time.
497 */
498 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100499dbg_check_skipped(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000500{
501 int prev_got_int;
502
503 if (debug_skipped)
504 {
505 /*
506 * Save the value of got_int and reset it. We don't want a previous
507 * interruption cause flushing the input buffer.
508 */
509 prev_got_int = got_int;
510 got_int = FALSE;
511 debug_breakpoint_name = debug_skipped_name;
512 /* eap->skip is TRUE */
513 eap->skip = FALSE;
514 (void)dbg_check_breakpoint(eap);
515 eap->skip = TRUE;
516 got_int |= prev_got_int;
517 return TRUE;
518 }
519 return FALSE;
520}
521
522/*
523 * The list of breakpoints: dbg_breakp.
524 * This is a grow-array of structs.
525 */
526struct debuggy
527{
528 int dbg_nr; /* breakpoint number */
529 int dbg_type; /* DBG_FUNC or DBG_FILE */
530 char_u *dbg_name; /* function or file name */
531 regprog_T *dbg_prog; /* regexp program */
532 linenr_T dbg_lnum; /* line number in function or file */
Bram Moolenaar05159a02005-02-26 23:04:13 +0000533 int dbg_forceit; /* ! used */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000534};
535
536static garray_T dbg_breakp = {0, 0, sizeof(struct debuggy), 4, NULL};
Bram Moolenaar05159a02005-02-26 23:04:13 +0000537#define BREAKP(idx) (((struct debuggy *)dbg_breakp.ga_data)[idx])
538#define DEBUGGY(gap, idx) (((struct debuggy *)gap->ga_data)[idx])
Bram Moolenaar071d4272004-06-13 20:20:40 +0000539static int last_breakp = 0; /* nr of last defined breakpoint */
540
Bram Moolenaar05159a02005-02-26 23:04:13 +0000541#ifdef FEAT_PROFILE
542/* Profiling uses file and func names similar to breakpoints. */
543static garray_T prof_ga = {0, 0, sizeof(struct debuggy), 4, NULL};
544#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000545#define DBG_FUNC 1
546#define DBG_FILE 2
547
Bram Moolenaarf28dbce2016-01-29 22:03:47 +0100548static int dbg_parsearg(char_u *arg, garray_T *gap);
549static linenr_T debuggy_find(int file,char_u *fname, linenr_T after, garray_T *gap, int *fp);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000550
551/*
Bram Moolenaar05159a02005-02-26 23:04:13 +0000552 * Parse the arguments of ":profile", ":breakadd" or ":breakdel" and put them
553 * in the entry just after the last one in dbg_breakp. Note that "dbg_name"
554 * is allocated.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000555 * Returns FAIL for failure.
556 */
557 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100558dbg_parsearg(
559 char_u *arg,
560 garray_T *gap) /* either &dbg_breakp or &prof_ga */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000561{
562 char_u *p = arg;
563 char_u *q;
564 struct debuggy *bp;
Bram Moolenaarf4b8e572004-06-24 15:53:16 +0000565 int here = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000566
Bram Moolenaar05159a02005-02-26 23:04:13 +0000567 if (ga_grow(gap, 1) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000568 return FAIL;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000569 bp = &DEBUGGY(gap, gap->ga_len);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000570
571 /* Find "func" or "file". */
572 if (STRNCMP(p, "func", 4) == 0)
573 bp->dbg_type = DBG_FUNC;
574 else if (STRNCMP(p, "file", 4) == 0)
575 bp->dbg_type = DBG_FILE;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000576 else if (
577#ifdef FEAT_PROFILE
578 gap != &prof_ga &&
579#endif
580 STRNCMP(p, "here", 4) == 0)
Bram Moolenaarf4b8e572004-06-24 15:53:16 +0000581 {
582 if (curbuf->b_ffname == NULL)
583 {
584 EMSG(_(e_noname));
585 return FAIL;
586 }
587 bp->dbg_type = DBG_FILE;
588 here = TRUE;
589 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000590 else
591 {
592 EMSG2(_(e_invarg2), p);
593 return FAIL;
594 }
595 p = skipwhite(p + 4);
596
597 /* Find optional line number. */
Bram Moolenaarf4b8e572004-06-24 15:53:16 +0000598 if (here)
599 bp->dbg_lnum = curwin->w_cursor.lnum;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000600 else if (
601#ifdef FEAT_PROFILE
602 gap != &prof_ga &&
603#endif
604 VIM_ISDIGIT(*p))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000605 {
606 bp->dbg_lnum = getdigits(&p);
607 p = skipwhite(p);
608 }
609 else
610 bp->dbg_lnum = 0;
611
612 /* Find the function or file name. Don't accept a function name with (). */
Bram Moolenaarf4b8e572004-06-24 15:53:16 +0000613 if ((!here && *p == NUL)
614 || (here && *p != NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000615 || (bp->dbg_type == DBG_FUNC && strstr((char *)p, "()") != NULL))
616 {
617 EMSG2(_(e_invarg2), arg);
618 return FAIL;
619 }
620
621 if (bp->dbg_type == DBG_FUNC)
622 bp->dbg_name = vim_strsave(p);
Bram Moolenaarf4b8e572004-06-24 15:53:16 +0000623 else if (here)
624 bp->dbg_name = vim_strsave(curbuf->b_ffname);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000625 else
626 {
627 /* Expand the file name in the same way as do_source(). This means
628 * doing it twice, so that $DIR/file gets expanded when $DIR is
629 * "~/dir". */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000630 q = expand_env_save(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000631 if (q == NULL)
632 return FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000633 p = expand_env_save(q);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000634 vim_free(q);
635 if (p == NULL)
636 return FAIL;
Bram Moolenaar843ee412004-06-30 16:16:41 +0000637 if (*p != '*')
638 {
639 bp->dbg_name = fix_fname(p);
640 vim_free(p);
641 }
642 else
643 bp->dbg_name = p;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000644 }
645
646 if (bp->dbg_name == NULL)
647 return FAIL;
648 return OK;
649}
650
651/*
652 * ":breakadd".
653 */
654 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100655ex_breakadd(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000656{
657 struct debuggy *bp;
658 char_u *pat;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000659 garray_T *gap;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000660
Bram Moolenaar05159a02005-02-26 23:04:13 +0000661 gap = &dbg_breakp;
662#ifdef FEAT_PROFILE
663 if (eap->cmdidx == CMD_profile)
664 gap = &prof_ga;
665#endif
666
667 if (dbg_parsearg(eap->arg, gap) == OK)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000668 {
Bram Moolenaar05159a02005-02-26 23:04:13 +0000669 bp = &DEBUGGY(gap, gap->ga_len);
670 bp->dbg_forceit = eap->forceit;
671
Bram Moolenaar071d4272004-06-13 20:20:40 +0000672 pat = file_pat_to_reg_pat(bp->dbg_name, NULL, NULL, FALSE);
673 if (pat != NULL)
674 {
675 bp->dbg_prog = vim_regcomp(pat, RE_MAGIC + RE_STRING);
676 vim_free(pat);
677 }
678 if (pat == NULL || bp->dbg_prog == NULL)
679 vim_free(bp->dbg_name);
680 else
681 {
682 if (bp->dbg_lnum == 0) /* default line number is 1 */
683 bp->dbg_lnum = 1;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000684#ifdef FEAT_PROFILE
685 if (eap->cmdidx != CMD_profile)
686#endif
687 {
688 DEBUGGY(gap, gap->ga_len).dbg_nr = ++last_breakp;
689 ++debug_tick;
690 }
691 ++gap->ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000692 }
693 }
694}
695
696/*
697 * ":debuggreedy".
698 */
699 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100700ex_debuggreedy(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000701{
702 if (eap->addr_count == 0 || eap->line2 != 0)
703 debug_greedy = TRUE;
704 else
705 debug_greedy = FALSE;
706}
707
708/*
Bram Moolenaard9fba312005-06-26 22:34:35 +0000709 * ":breakdel" and ":profdel".
Bram Moolenaar071d4272004-06-13 20:20:40 +0000710 */
711 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100712ex_breakdel(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000713{
714 struct debuggy *bp, *bpi;
715 int nr;
716 int todel = -1;
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000717 int del_all = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000718 int i;
719 linenr_T best_lnum = 0;
Bram Moolenaard9fba312005-06-26 22:34:35 +0000720 garray_T *gap;
721
722 gap = &dbg_breakp;
Bram Moolenaard9fba312005-06-26 22:34:35 +0000723 if (eap->cmdidx == CMD_profdel)
Bram Moolenaar38bdbd62012-06-20 15:48:57 +0200724 {
725#ifdef FEAT_PROFILE
Bram Moolenaard9fba312005-06-26 22:34:35 +0000726 gap = &prof_ga;
Bram Moolenaar38bdbd62012-06-20 15:48:57 +0200727#else
728 ex_ni(eap);
729 return;
Bram Moolenaard9fba312005-06-26 22:34:35 +0000730#endif
Bram Moolenaar38bdbd62012-06-20 15:48:57 +0200731 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000732
733 if (vim_isdigit(*eap->arg))
734 {
735 /* ":breakdel {nr}" */
736 nr = atol((char *)eap->arg);
Bram Moolenaard9fba312005-06-26 22:34:35 +0000737 for (i = 0; i < gap->ga_len; ++i)
738 if (DEBUGGY(gap, i).dbg_nr == nr)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000739 {
740 todel = i;
741 break;
742 }
743 }
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000744 else if (*eap->arg == '*')
745 {
746 todel = 0;
747 del_all = TRUE;
748 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000749 else
750 {
751 /* ":breakdel {func|file} [lnum] {name}" */
Bram Moolenaard9fba312005-06-26 22:34:35 +0000752 if (dbg_parsearg(eap->arg, gap) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000753 return;
Bram Moolenaard9fba312005-06-26 22:34:35 +0000754 bp = &DEBUGGY(gap, gap->ga_len);
755 for (i = 0; i < gap->ga_len; ++i)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000756 {
Bram Moolenaard9fba312005-06-26 22:34:35 +0000757 bpi = &DEBUGGY(gap, i);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000758 if (bp->dbg_type == bpi->dbg_type
759 && STRCMP(bp->dbg_name, bpi->dbg_name) == 0
760 && (bp->dbg_lnum == bpi->dbg_lnum
761 || (bp->dbg_lnum == 0
762 && (best_lnum == 0
763 || bpi->dbg_lnum < best_lnum))))
764 {
765 todel = i;
766 best_lnum = bpi->dbg_lnum;
767 }
768 }
769 vim_free(bp->dbg_name);
770 }
771
772 if (todel < 0)
773 EMSG2(_("E161: Breakpoint not found: %s"), eap->arg);
774 else
Bram Moolenaard9fba312005-06-26 22:34:35 +0000775 {
776 while (gap->ga_len > 0)
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000777 {
Bram Moolenaard9fba312005-06-26 22:34:35 +0000778 vim_free(DEBUGGY(gap, todel).dbg_name);
Bram Moolenaar473de612013-06-08 18:19:48 +0200779 vim_regfree(DEBUGGY(gap, todel).dbg_prog);
Bram Moolenaard9fba312005-06-26 22:34:35 +0000780 --gap->ga_len;
781 if (todel < gap->ga_len)
782 mch_memmove(&DEBUGGY(gap, todel), &DEBUGGY(gap, todel + 1),
783 (gap->ga_len - todel) * sizeof(struct debuggy));
784#ifdef FEAT_PROFILE
785 if (eap->cmdidx == CMD_breakdel)
786#endif
787 ++debug_tick;
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000788 if (!del_all)
789 break;
790 }
Bram Moolenaard9fba312005-06-26 22:34:35 +0000791
792 /* If all breakpoints were removed clear the array. */
793 if (gap->ga_len == 0)
794 ga_clear(gap);
795 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000796}
797
798/*
799 * ":breaklist".
800 */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000801 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100802ex_breaklist(exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000803{
804 struct debuggy *bp;
805 int i;
806
807 if (dbg_breakp.ga_len == 0)
808 MSG(_("No breakpoints defined"));
809 else
810 for (i = 0; i < dbg_breakp.ga_len; ++i)
811 {
812 bp = &BREAKP(i);
Bram Moolenaard58ea072011-06-26 04:25:30 +0200813 if (bp->dbg_type == DBG_FILE)
814 home_replace(NULL, bp->dbg_name, NameBuff, MAXPATHL, TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000815 smsg((char_u *)_("%3d %s %s line %ld"),
816 bp->dbg_nr,
817 bp->dbg_type == DBG_FUNC ? "func" : "file",
Bram Moolenaard58ea072011-06-26 04:25:30 +0200818 bp->dbg_type == DBG_FUNC ? bp->dbg_name : NameBuff,
Bram Moolenaar071d4272004-06-13 20:20:40 +0000819 (long)bp->dbg_lnum);
820 }
821}
822
823/*
824 * Find a breakpoint for a function or sourced file.
825 * Returns line number at which to break; zero when no matching breakpoint.
826 */
827 linenr_T
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100828dbg_find_breakpoint(
829 int file, /* TRUE for a file, FALSE for a function */
830 char_u *fname, /* file or function name */
831 linenr_T after) /* after this line number */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000832{
Bram Moolenaar05159a02005-02-26 23:04:13 +0000833 return debuggy_find(file, fname, after, &dbg_breakp, NULL);
834}
835
836#if defined(FEAT_PROFILE) || defined(PROTO)
837/*
838 * Return TRUE if profiling is on for a function or sourced file.
839 */
840 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100841has_profiling(
842 int file, /* TRUE for a file, FALSE for a function */
843 char_u *fname, /* file or function name */
844 int *fp) /* return: forceit */
Bram Moolenaar05159a02005-02-26 23:04:13 +0000845{
846 return (debuggy_find(file, fname, (linenr_T)0, &prof_ga, fp)
847 != (linenr_T)0);
848}
849#endif
850
851/*
852 * Common code for dbg_find_breakpoint() and has_profiling().
853 */
854 static linenr_T
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100855debuggy_find(
856 int file, /* TRUE for a file, FALSE for a function */
857 char_u *fname, /* file or function name */
858 linenr_T after, /* after this line number */
859 garray_T *gap, /* either &dbg_breakp or &prof_ga */
860 int *fp) /* if not NULL: return forceit */
Bram Moolenaar05159a02005-02-26 23:04:13 +0000861{
Bram Moolenaar071d4272004-06-13 20:20:40 +0000862 struct debuggy *bp;
863 int i;
864 linenr_T lnum = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000865 char_u *name = fname;
866 int prev_got_int;
867
Bram Moolenaar05159a02005-02-26 23:04:13 +0000868 /* Return quickly when there are no breakpoints. */
869 if (gap->ga_len == 0)
870 return (linenr_T)0;
871
Bram Moolenaar071d4272004-06-13 20:20:40 +0000872 /* Replace K_SNR in function name with "<SNR>". */
873 if (!file && fname[0] == K_SPECIAL)
874 {
875 name = alloc((unsigned)STRLEN(fname) + 3);
876 if (name == NULL)
877 name = fname;
878 else
879 {
880 STRCPY(name, "<SNR>");
881 STRCPY(name + 5, fname + 3);
882 }
883 }
884
Bram Moolenaar05159a02005-02-26 23:04:13 +0000885 for (i = 0; i < gap->ga_len; ++i)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000886 {
Bram Moolenaar05159a02005-02-26 23:04:13 +0000887 /* Skip entries that are not useful or are for a line that is beyond
888 * an already found breakpoint. */
889 bp = &DEBUGGY(gap, i);
890 if (((bp->dbg_type == DBG_FILE) == file && (
891#ifdef FEAT_PROFILE
892 gap == &prof_ga ||
893#endif
894 (bp->dbg_lnum > after && (lnum == 0 || bp->dbg_lnum < lnum)))))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000895 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000896 /*
Bram Moolenaar05159a02005-02-26 23:04:13 +0000897 * Save the value of got_int and reset it. We don't want a
898 * previous interruption cancel matching, only hitting CTRL-C
899 * while matching should abort it.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000900 */
901 prev_got_int = got_int;
902 got_int = FALSE;
Bram Moolenaardffa5b82014-11-19 16:38:07 +0100903 if (vim_regexec_prog(&bp->dbg_prog, FALSE, name, (colnr_T)0))
Bram Moolenaar05159a02005-02-26 23:04:13 +0000904 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000905 lnum = bp->dbg_lnum;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000906 if (fp != NULL)
907 *fp = bp->dbg_forceit;
908 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000909 got_int |= prev_got_int;
910 }
911 }
912 if (name != fname)
913 vim_free(name);
914
915 return lnum;
916}
917
918/*
919 * Called when a breakpoint was encountered.
920 */
921 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100922dbg_breakpoint(char_u *name, linenr_T lnum)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000923{
924 /* We need to check if this line is actually executed in do_one_cmd() */
925 debug_breakpoint_name = name;
926 debug_breakpoint_lnum = lnum;
927}
Bram Moolenaar05159a02005-02-26 23:04:13 +0000928
929
Bram Moolenaar433f7c82006-03-21 21:29:36 +0000930# if defined(FEAT_PROFILE) || defined(FEAT_RELTIME) || defined(PROTO)
Bram Moolenaar05159a02005-02-26 23:04:13 +0000931/*
932 * Store the current time in "tm".
933 */
934 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100935profile_start(proftime_T *tm)
Bram Moolenaar05159a02005-02-26 23:04:13 +0000936{
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000937# ifdef WIN3264
938 QueryPerformanceCounter(tm);
939# else
Bram Moolenaar05159a02005-02-26 23:04:13 +0000940 gettimeofday(tm, NULL);
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000941# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +0000942}
943
944/*
945 * Compute the elapsed time from "tm" till now and store in "tm".
946 */
947 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100948profile_end(proftime_T *tm)
Bram Moolenaar05159a02005-02-26 23:04:13 +0000949{
950 proftime_T now;
951
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000952# ifdef WIN3264
953 QueryPerformanceCounter(&now);
954 tm->QuadPart = now.QuadPart - tm->QuadPart;
955# else
Bram Moolenaar05159a02005-02-26 23:04:13 +0000956 gettimeofday(&now, NULL);
957 tm->tv_usec = now.tv_usec - tm->tv_usec;
958 tm->tv_sec = now.tv_sec - tm->tv_sec;
959 if (tm->tv_usec < 0)
960 {
961 tm->tv_usec += 1000000;
962 --tm->tv_sec;
963 }
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000964# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +0000965}
966
967/*
968 * Subtract the time "tm2" from "tm".
969 */
970 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100971profile_sub(proftime_T *tm, proftime_T *tm2)
Bram Moolenaar05159a02005-02-26 23:04:13 +0000972{
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000973# ifdef WIN3264
974 tm->QuadPart -= tm2->QuadPart;
975# else
Bram Moolenaar05159a02005-02-26 23:04:13 +0000976 tm->tv_usec -= tm2->tv_usec;
977 tm->tv_sec -= tm2->tv_sec;
978 if (tm->tv_usec < 0)
979 {
980 tm->tv_usec += 1000000;
981 --tm->tv_sec;
982 }
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000983# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +0000984}
985
986/*
Bram Moolenaar433f7c82006-03-21 21:29:36 +0000987 * Return a string that represents the time in "tm".
988 * Uses a static buffer!
989 */
990 char *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100991profile_msg(proftime_T *tm)
Bram Moolenaar433f7c82006-03-21 21:29:36 +0000992{
993 static char buf[50];
994
995# ifdef WIN3264
996 LARGE_INTEGER fr;
997
998 QueryPerformanceFrequency(&fr);
999 sprintf(buf, "%10.6lf", (double)tm->QuadPart / (double)fr.QuadPart);
1000# else
1001 sprintf(buf, "%3ld.%06ld", (long)tm->tv_sec, (long)tm->tv_usec);
Bram Moolenaar76929292008-01-06 19:07:36 +00001002# endif
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001003 return buf;
1004}
1005
Bram Moolenaar79c2c882016-02-07 21:19:28 +01001006# if defined(FEAT_FLOAT) || defined(PROTO)
1007/*
1008 * Return a float that represents the time in "tm".
1009 */
1010 float_T
1011profile_float(proftime_T *tm)
1012{
1013# ifdef WIN3264
1014 LARGE_INTEGER fr;
1015
1016 QueryPerformanceFrequency(&fr);
1017 return (float_T)tm->QuadPart / (float_T)fr.QuadPart;
1018# else
1019 return (float_T)tm->tv_sec + (float_T)tm->tv_usec / 1000000.0;
1020# endif
1021}
1022# endif
1023
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001024/*
Bram Moolenaar76929292008-01-06 19:07:36 +00001025 * Put the time "msec" past now in "tm".
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001026 */
Bram Moolenaar76929292008-01-06 19:07:36 +00001027 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001028profile_setlimit(long msec, proftime_T *tm)
Bram Moolenaar76929292008-01-06 19:07:36 +00001029{
1030 if (msec <= 0) /* no limit */
1031 profile_zero(tm);
1032 else
1033 {
1034# ifdef WIN3264
1035 LARGE_INTEGER fr;
1036
1037 QueryPerformanceCounter(tm);
1038 QueryPerformanceFrequency(&fr);
Bram Moolenaarb3c70982008-01-18 10:40:55 +00001039 tm->QuadPart += (LONGLONG)((double)msec / 1000.0 * (double)fr.QuadPart);
Bram Moolenaar76929292008-01-06 19:07:36 +00001040# else
1041 long usec;
1042
1043 gettimeofday(tm, NULL);
1044 usec = (long)tm->tv_usec + (long)msec * 1000;
1045 tm->tv_usec = usec % 1000000L;
1046 tm->tv_sec += usec / 1000000L;
1047# endif
1048 }
1049}
1050
1051/*
1052 * Return TRUE if the current time is past "tm".
1053 */
1054 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001055profile_passed_limit(proftime_T *tm)
Bram Moolenaar76929292008-01-06 19:07:36 +00001056{
1057 proftime_T now;
1058
1059# ifdef WIN3264
1060 if (tm->QuadPart == 0) /* timer was not set */
1061 return FALSE;
1062 QueryPerformanceCounter(&now);
1063 return (now.QuadPart > tm->QuadPart);
1064# else
1065 if (tm->tv_sec == 0) /* timer was not set */
1066 return FALSE;
1067 gettimeofday(&now, NULL);
1068 return (now.tv_sec > tm->tv_sec
1069 || (now.tv_sec == tm->tv_sec && now.tv_usec > tm->tv_usec));
1070# endif
1071}
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001072
1073/*
1074 * Set the time in "tm" to zero.
1075 */
1076 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001077profile_zero(proftime_T *tm)
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001078{
1079# ifdef WIN3264
1080 tm->QuadPart = 0;
1081# else
1082 tm->tv_usec = 0;
1083 tm->tv_sec = 0;
1084# endif
1085}
1086
Bram Moolenaar76929292008-01-06 19:07:36 +00001087# endif /* FEAT_PROFILE || FEAT_RELTIME */
1088
Bram Moolenaar975b5272016-03-15 23:10:59 +01001089# if defined(FEAT_TIMERS) || defined(PROTO)
1090static timer_T *first_timer = NULL;
1091static int last_timer_id = 0;
1092
1093/*
1094 * Insert a timer in the list of timers.
1095 */
1096 static void
1097insert_timer(timer_T *timer)
1098{
1099 timer->tr_next = first_timer;
1100 timer->tr_prev = NULL;
1101 if (first_timer != NULL)
1102 first_timer->tr_prev = timer;
1103 first_timer = timer;
1104}
1105
1106/*
1107 * Take a timer out of the list of timers.
1108 */
1109 static void
1110remove_timer(timer_T *timer)
1111{
1112 if (timer->tr_prev == NULL)
1113 first_timer = timer->tr_next;
1114 else
1115 timer->tr_prev->tr_next = timer->tr_next;
1116 if (timer->tr_next != NULL)
1117 timer->tr_next->tr_prev = timer->tr_prev;
1118}
1119
1120 static void
1121free_timer(timer_T *timer)
1122{
1123 vim_free(timer->tr_callback);
1124 partial_unref(timer->tr_partial);
1125 vim_free(timer);
1126}
1127
1128/*
1129 * Create a timer and return it. NULL if out of memory.
1130 * Caller should set the callback.
1131 */
1132 timer_T *
1133create_timer(long msec, int repeat)
1134{
1135 timer_T *timer = (timer_T *)alloc_clear(sizeof(timer_T));
1136
1137 if (timer == NULL)
1138 return NULL;
1139 timer->tr_id = ++last_timer_id;
1140 insert_timer(timer);
1141 if (repeat != 0)
1142 {
1143 timer->tr_repeat = repeat - 1;
1144 timer->tr_interval = msec;
1145 }
1146
1147 profile_setlimit(msec, &timer->tr_due);
1148 return timer;
1149}
1150
1151/*
1152 * Invoke the callback of "timer".
1153 */
1154 static void
1155timer_callback(timer_T *timer)
1156{
1157 typval_T rettv;
1158 int dummy;
1159 typval_T argv[2];
1160
1161 argv[0].v_type = VAR_NUMBER;
1162 argv[0].vval.v_number = timer->tr_id;
1163 argv[1].v_type = VAR_UNKNOWN;
1164
1165 call_func(timer->tr_callback, (int)STRLEN(timer->tr_callback),
1166 &rettv, 1, argv, 0L, 0L, &dummy, TRUE,
1167 timer->tr_partial, NULL);
1168 clear_tv(&rettv);
1169}
1170
1171/*
1172 * Call timers that are due.
1173 * Return the time in msec until the next timer is due.
1174 */
1175 long
1176check_due_timer()
1177{
1178 timer_T *timer;
1179 long this_due;
Bram Moolenaar597385a2016-03-16 23:24:43 +01001180 long next_due = -1;
Bram Moolenaar975b5272016-03-15 23:10:59 +01001181 proftime_T now;
1182 int did_one = FALSE;
1183# ifdef WIN3264
1184 LARGE_INTEGER fr;
1185
1186 QueryPerformanceFrequency(&fr);
1187# endif
1188 while (!got_int)
1189 {
1190 profile_start(&now);
1191 next_due = -1;
1192 for (timer = first_timer; timer != NULL; timer = timer->tr_next)
1193 {
1194# ifdef WIN3264
1195 this_due = (long)(((double)(timer->tr_due.QuadPart - now.QuadPart)
1196 / (double)fr.QuadPart) * 1000);
1197# else
1198 this_due = (timer->tr_due.tv_sec - now.tv_sec) * 1000
1199 + (timer->tr_due.tv_usec - now.tv_usec) / 1000;
1200# endif
1201 if (this_due <= 1)
1202 {
1203 remove_timer(timer);
1204 timer_callback(timer);
1205 did_one = TRUE;
1206 if (timer->tr_repeat != 0)
1207 {
1208 profile_setlimit(timer->tr_interval, &timer->tr_due);
1209 if (timer->tr_repeat > 0)
1210 --timer->tr_repeat;
1211 insert_timer(timer);
1212 }
1213 else
1214 free_timer(timer);
1215 /* the callback may do anything, start all over */
1216 break;
1217 }
1218 if (next_due == -1 || next_due > this_due)
1219 next_due = this_due;
1220 }
1221 if (timer == NULL)
1222 break;
1223 }
1224
1225 if (did_one)
1226 redraw_after_callback();
1227
1228 return next_due;
1229}
1230
1231/*
1232 * Find a timer by ID. Returns NULL if not found;
1233 */
1234 timer_T *
1235find_timer(int id)
1236{
1237 timer_T *timer;
1238
1239 for (timer = first_timer; timer != NULL; timer = timer->tr_next)
1240 if (timer->tr_id == id)
1241 break;
1242 return timer;
1243}
1244
1245
1246/*
1247 * Stop a timer and delete it.
1248 */
1249 void
1250stop_timer(timer_T *timer)
1251{
1252 remove_timer(timer);
1253 free_timer(timer);
1254}
Bram Moolenaare3188e22016-05-31 21:13:04 +02001255
1256/*
1257 * Mark references in partials of timers.
1258 */
1259 int
1260set_ref_in_timer(int copyID)
1261{
1262 int abort = FALSE;
1263 timer_T *timer;
1264 typval_T tv;
1265
1266 for (timer = first_timer; timer != NULL; timer = timer->tr_next)
1267 {
1268 tv.v_type = VAR_PARTIAL;
1269 tv.vval.v_partial = timer->tr_partial;
1270 abort = abort || set_ref_in_item(&tv, copyID, NULL, NULL);
1271 }
1272 return abort;
1273}
Bram Moolenaar975b5272016-03-15 23:10:59 +01001274# endif
1275
Bram Moolenaar8a7f5a22013-06-06 14:01:46 +02001276#if defined(FEAT_SYN_HL) && defined(FEAT_RELTIME) && defined(FEAT_FLOAT)
1277# if defined(HAVE_MATH_H)
1278# include <math.h>
1279# endif
1280
1281/*
1282 * Divide the time "tm" by "count" and store in "tm2".
1283 */
1284 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001285profile_divide(proftime_T *tm, int count, proftime_T *tm2)
Bram Moolenaar8a7f5a22013-06-06 14:01:46 +02001286{
1287 if (count == 0)
1288 profile_zero(tm2);
1289 else
1290 {
1291# ifdef WIN3264
1292 tm2->QuadPart = tm->QuadPart / count;
1293# else
1294 double usec = (tm->tv_sec * 1000000.0 + tm->tv_usec) / count;
1295
1296 tm2->tv_sec = floor(usec / 1000000.0);
Bram Moolenaara2e14fc2013-06-10 20:10:44 +02001297 tm2->tv_usec = vim_round(usec - (tm2->tv_sec * 1000000.0));
Bram Moolenaar8a7f5a22013-06-06 14:01:46 +02001298# endif
1299 }
1300}
1301#endif
1302
Bram Moolenaar76929292008-01-06 19:07:36 +00001303# if defined(FEAT_PROFILE) || defined(PROTO)
1304/*
1305 * Functions for profiling.
1306 */
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01001307static void script_do_profile(scriptitem_T *si);
1308static void script_dump_profile(FILE *fd);
Bram Moolenaar76929292008-01-06 19:07:36 +00001309static proftime_T prof_wait_time;
1310
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001311/*
Bram Moolenaar05159a02005-02-26 23:04:13 +00001312 * Add the time "tm2" to "tm".
1313 */
1314 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001315profile_add(proftime_T *tm, proftime_T *tm2)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001316{
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001317# ifdef WIN3264
1318 tm->QuadPart += tm2->QuadPart;
1319# else
Bram Moolenaar05159a02005-02-26 23:04:13 +00001320 tm->tv_usec += tm2->tv_usec;
1321 tm->tv_sec += tm2->tv_sec;
1322 if (tm->tv_usec >= 1000000)
1323 {
1324 tm->tv_usec -= 1000000;
1325 ++tm->tv_sec;
1326 }
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001327# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00001328}
1329
1330/*
Bram Moolenaar1056d982006-03-09 22:37:52 +00001331 * Add the "self" time from the total time and the children's time.
1332 */
1333 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001334profile_self(proftime_T *self, proftime_T *total, proftime_T *children)
Bram Moolenaar1056d982006-03-09 22:37:52 +00001335{
1336 /* Check that the result won't be negative. Can happen with recursive
1337 * calls. */
1338#ifdef WIN3264
1339 if (total->QuadPart <= children->QuadPart)
1340 return;
1341#else
1342 if (total->tv_sec < children->tv_sec
1343 || (total->tv_sec == children->tv_sec
1344 && total->tv_usec <= children->tv_usec))
1345 return;
1346#endif
1347 profile_add(self, total);
1348 profile_sub(self, children);
1349}
1350
1351/*
Bram Moolenaar05159a02005-02-26 23:04:13 +00001352 * Get the current waittime.
1353 */
1354 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001355profile_get_wait(proftime_T *tm)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001356{
1357 *tm = prof_wait_time;
1358}
1359
1360/*
1361 * Subtract the passed waittime since "tm" from "tma".
1362 */
1363 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001364profile_sub_wait(proftime_T *tm, proftime_T *tma)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001365{
1366 proftime_T tm3 = prof_wait_time;
1367
1368 profile_sub(&tm3, tm);
1369 profile_sub(tma, &tm3);
1370}
1371
1372/*
1373 * Return TRUE if "tm1" and "tm2" are equal.
1374 */
1375 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001376profile_equal(proftime_T *tm1, proftime_T *tm2)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001377{
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001378# ifdef WIN3264
1379 return (tm1->QuadPart == tm2->QuadPart);
1380# else
Bram Moolenaar05159a02005-02-26 23:04:13 +00001381 return (tm1->tv_usec == tm2->tv_usec && tm1->tv_sec == tm2->tv_sec);
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001382# endif
1383}
1384
1385/*
1386 * Return <0, 0 or >0 if "tm1" < "tm2", "tm1" == "tm2" or "tm1" > "tm2"
1387 */
1388 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001389profile_cmp(const proftime_T *tm1, const proftime_T *tm2)
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001390{
1391# ifdef WIN3264
1392 return (int)(tm2->QuadPart - tm1->QuadPart);
1393# else
1394 if (tm1->tv_sec == tm2->tv_sec)
1395 return tm2->tv_usec - tm1->tv_usec;
1396 return tm2->tv_sec - tm1->tv_sec;
1397# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00001398}
1399
Bram Moolenaar05159a02005-02-26 23:04:13 +00001400static char_u *profile_fname = NULL;
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001401static proftime_T pause_time;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001402
1403/*
1404 * ":profile cmd args"
1405 */
1406 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001407ex_profile(exarg_T *eap)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001408{
1409 char_u *e;
1410 int len;
1411
1412 e = skiptowhite(eap->arg);
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00001413 len = (int)(e - eap->arg);
Bram Moolenaar05159a02005-02-26 23:04:13 +00001414 e = skipwhite(e);
1415
1416 if (len == 5 && STRNCMP(eap->arg, "start", 5) == 0 && *e != NUL)
1417 {
1418 vim_free(profile_fname);
Bram Moolenaard94682f2015-04-13 15:37:56 +02001419 profile_fname = expand_env_save_opt(e, TRUE);
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001420 do_profiling = PROF_YES;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001421 profile_zero(&prof_wait_time);
1422 set_vim_var_nr(VV_PROFILING, 1L);
1423 }
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001424 else if (do_profiling == PROF_NONE)
Bram Moolenaar54c1b492010-02-24 14:01:28 +01001425 EMSG(_("E750: First use \":profile start {fname}\""));
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001426 else if (STRCMP(eap->arg, "pause") == 0)
1427 {
1428 if (do_profiling == PROF_YES)
1429 profile_start(&pause_time);
1430 do_profiling = PROF_PAUSED;
1431 }
1432 else if (STRCMP(eap->arg, "continue") == 0)
1433 {
1434 if (do_profiling == PROF_PAUSED)
1435 {
1436 profile_end(&pause_time);
1437 profile_add(&prof_wait_time, &pause_time);
1438 }
1439 do_profiling = PROF_YES;
1440 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00001441 else
1442 {
1443 /* The rest is similar to ":breakadd". */
1444 ex_breakadd(eap);
1445 }
1446}
1447
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001448/* Command line expansion for :profile. */
1449static enum
1450{
1451 PEXP_SUBCMD, /* expand :profile sub-commands */
Bram Moolenaar49789dc2011-02-25 14:46:09 +01001452 PEXP_FUNC /* expand :profile func {funcname} */
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001453} pexpand_what;
1454
1455static char *pexpand_cmds[] = {
1456 "start",
1457#define PROFCMD_START 0
1458 "pause",
1459#define PROFCMD_PAUSE 1
1460 "continue",
1461#define PROFCMD_CONTINUE 2
1462 "func",
1463#define PROFCMD_FUNC 3
1464 "file",
1465#define PROFCMD_FILE 4
1466 NULL
1467#define PROFCMD_LAST 5
1468};
1469
1470/*
1471 * Function given to ExpandGeneric() to obtain the profile command
1472 * specific expansion.
1473 */
1474 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001475get_profile_name(expand_T *xp UNUSED, int idx)
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001476{
1477 switch (pexpand_what)
1478 {
1479 case PEXP_SUBCMD:
1480 return (char_u *)pexpand_cmds[idx];
1481 /* case PEXP_FUNC: TODO */
1482 default:
1483 return NULL;
1484 }
1485}
1486
1487/*
1488 * Handle command line completion for :profile command.
1489 */
1490 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001491set_context_in_profile_cmd(expand_T *xp, char_u *arg)
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001492{
1493 char_u *end_subcmd;
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001494
1495 /* Default: expand subcommands. */
1496 xp->xp_context = EXPAND_PROFILE;
1497 pexpand_what = PEXP_SUBCMD;
1498 xp->xp_pattern = arg;
1499
1500 end_subcmd = skiptowhite(arg);
1501 if (*end_subcmd == NUL)
1502 return;
1503
Bram Moolenaar8b9c05f2010-03-02 17:54:33 +01001504 if (end_subcmd - arg == 5 && STRNCMP(arg, "start", 5) == 0)
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001505 {
1506 xp->xp_context = EXPAND_FILES;
1507 xp->xp_pattern = skipwhite(end_subcmd);
1508 return;
1509 }
1510
1511 /* TODO: expand function names after "func" */
1512 xp->xp_context = EXPAND_NOTHING;
1513}
1514
Bram Moolenaar05159a02005-02-26 23:04:13 +00001515/*
1516 * Dump the profiling info.
1517 */
1518 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001519profile_dump(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001520{
1521 FILE *fd;
1522
1523 if (profile_fname != NULL)
1524 {
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +00001525 fd = mch_fopen((char *)profile_fname, "w");
Bram Moolenaar05159a02005-02-26 23:04:13 +00001526 if (fd == NULL)
1527 EMSG2(_(e_notopen), profile_fname);
1528 else
1529 {
Bram Moolenaar05159a02005-02-26 23:04:13 +00001530 script_dump_profile(fd);
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001531 func_dump_profile(fd);
Bram Moolenaar05159a02005-02-26 23:04:13 +00001532 fclose(fd);
1533 }
1534 }
1535}
1536
1537/*
1538 * Start profiling script "fp".
1539 */
1540 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001541script_do_profile(scriptitem_T *si)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001542{
1543 si->sn_pr_count = 0;
1544 profile_zero(&si->sn_pr_total);
1545 profile_zero(&si->sn_pr_self);
1546
1547 ga_init2(&si->sn_prl_ga, sizeof(sn_prl_T), 100);
1548 si->sn_prl_idx = -1;
1549 si->sn_prof_on = TRUE;
1550 si->sn_pr_nest = 0;
1551}
1552
1553/*
1554 * save time when starting to invoke another script or function.
1555 */
1556 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001557script_prof_save(
1558 proftime_T *tm) /* place to store wait time */
Bram Moolenaar05159a02005-02-26 23:04:13 +00001559{
1560 scriptitem_T *si;
1561
1562 if (current_SID > 0 && current_SID <= script_items.ga_len)
1563 {
1564 si = &SCRIPT_ITEM(current_SID);
1565 if (si->sn_prof_on && si->sn_pr_nest++ == 0)
1566 profile_start(&si->sn_pr_child);
1567 }
1568 profile_get_wait(tm);
1569}
1570
1571/*
1572 * Count time spent in children after invoking another script or function.
1573 */
1574 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001575script_prof_restore(proftime_T *tm)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001576{
1577 scriptitem_T *si;
1578
1579 if (current_SID > 0 && current_SID <= script_items.ga_len)
1580 {
1581 si = &SCRIPT_ITEM(current_SID);
1582 if (si->sn_prof_on && --si->sn_pr_nest == 0)
1583 {
1584 profile_end(&si->sn_pr_child);
1585 profile_sub_wait(tm, &si->sn_pr_child); /* don't count wait time */
1586 profile_add(&si->sn_pr_children, &si->sn_pr_child);
1587 profile_add(&si->sn_prl_children, &si->sn_pr_child);
1588 }
1589 }
1590}
1591
1592static proftime_T inchar_time;
1593
1594/*
1595 * Called when starting to wait for the user to type a character.
1596 */
1597 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001598prof_inchar_enter(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001599{
1600 profile_start(&inchar_time);
1601}
1602
1603/*
1604 * Called when finished waiting for the user to type a character.
1605 */
1606 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001607prof_inchar_exit(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001608{
1609 profile_end(&inchar_time);
1610 profile_add(&prof_wait_time, &inchar_time);
1611}
1612
1613/*
1614 * Dump the profiling results for all scripts in file "fd".
1615 */
1616 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001617script_dump_profile(FILE *fd)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001618{
1619 int id;
1620 scriptitem_T *si;
1621 int i;
1622 FILE *sfd;
1623 sn_prl_T *pp;
1624
1625 for (id = 1; id <= script_items.ga_len; ++id)
1626 {
1627 si = &SCRIPT_ITEM(id);
1628 if (si->sn_prof_on)
1629 {
1630 fprintf(fd, "SCRIPT %s\n", si->sn_name);
1631 if (si->sn_pr_count == 1)
1632 fprintf(fd, "Sourced 1 time\n");
1633 else
1634 fprintf(fd, "Sourced %d times\n", si->sn_pr_count);
1635 fprintf(fd, "Total time: %s\n", profile_msg(&si->sn_pr_total));
1636 fprintf(fd, " Self time: %s\n", profile_msg(&si->sn_pr_self));
1637 fprintf(fd, "\n");
1638 fprintf(fd, "count total (s) self (s)\n");
1639
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +00001640 sfd = mch_fopen((char *)si->sn_name, "r");
Bram Moolenaar05159a02005-02-26 23:04:13 +00001641 if (sfd == NULL)
1642 fprintf(fd, "Cannot open file!\n");
1643 else
1644 {
1645 for (i = 0; i < si->sn_prl_ga.ga_len; ++i)
1646 {
1647 if (vim_fgets(IObuff, IOSIZE, sfd))
1648 break;
1649 pp = &PRL_ITEM(si, i);
1650 if (pp->snp_count > 0)
1651 {
1652 fprintf(fd, "%5d ", pp->snp_count);
1653 if (profile_equal(&pp->sn_prl_total, &pp->sn_prl_self))
1654 fprintf(fd, " ");
1655 else
1656 fprintf(fd, "%s ", profile_msg(&pp->sn_prl_total));
1657 fprintf(fd, "%s ", profile_msg(&pp->sn_prl_self));
1658 }
1659 else
1660 fprintf(fd, " ");
1661 fprintf(fd, "%s", IObuff);
1662 }
1663 fclose(sfd);
1664 }
1665 fprintf(fd, "\n");
1666 }
1667 }
1668}
1669
1670/*
1671 * Return TRUE when a function defined in the current script should be
1672 * profiled.
1673 */
1674 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001675prof_def_func(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001676{
Bram Moolenaar53180ce2005-07-05 21:48:14 +00001677 if (current_SID > 0)
1678 return SCRIPT_ITEM(current_SID).sn_pr_force;
1679 return FALSE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001680}
1681
1682# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001683#endif
1684
1685/*
1686 * If 'autowrite' option set, try to write the file.
1687 * Careful: autocommands may make "buf" invalid!
1688 *
1689 * return FAIL for failure, OK otherwise
1690 */
1691 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001692autowrite(buf_T *buf, int forceit)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001693{
Bram Moolenaar373154b2007-02-13 05:19:30 +00001694 int r;
1695
Bram Moolenaar071d4272004-06-13 20:20:40 +00001696 if (!(p_aw || p_awa) || !p_write
1697#ifdef FEAT_QUICKFIX
Bram Moolenaar373154b2007-02-13 05:19:30 +00001698 /* never autowrite a "nofile" or "nowrite" buffer */
1699 || bt_dontwrite(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001700#endif
Bram Moolenaar373154b2007-02-13 05:19:30 +00001701 || (!forceit && buf->b_p_ro) || buf->b_ffname == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001702 return FAIL;
Bram Moolenaar373154b2007-02-13 05:19:30 +00001703 r = buf_write_all(buf, forceit);
1704
1705 /* Writing may succeed but the buffer still changed, e.g., when there is a
1706 * conversion error. We do want to return FAIL then. */
1707 if (buf_valid(buf) && bufIsChanged(buf))
1708 r = FAIL;
1709 return r;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001710}
1711
1712/*
1713 * flush all buffers, except the ones that are readonly
1714 */
1715 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001716autowrite_all(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001717{
1718 buf_T *buf;
1719
1720 if (!(p_aw || p_awa) || !p_write)
1721 return;
1722 for (buf = firstbuf; buf; buf = buf->b_next)
1723 if (bufIsChanged(buf) && !buf->b_p_ro)
1724 {
1725 (void)buf_write_all(buf, FALSE);
1726#ifdef FEAT_AUTOCMD
1727 /* an autocommand may have deleted the buffer */
1728 if (!buf_valid(buf))
1729 buf = firstbuf;
1730#endif
1731 }
1732}
1733
1734/*
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001735 * Return TRUE if buffer was changed and cannot be abandoned.
1736 * For flags use the CCGD_ values.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001737 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001738 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001739check_changed(buf_T *buf, int flags)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001740{
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001741 int forceit = (flags & CCGD_FORCEIT);
1742
Bram Moolenaar071d4272004-06-13 20:20:40 +00001743 if ( !forceit
1744 && bufIsChanged(buf)
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001745 && ((flags & CCGD_MULTWIN) || buf->b_nwindows <= 1)
1746 && (!(flags & CCGD_AW) || autowrite(buf, forceit) == FAIL))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001747 {
1748#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG)
1749 if ((p_confirm || cmdmod.confirm) && p_write)
1750 {
1751 buf_T *buf2;
1752 int count = 0;
1753
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001754 if (flags & CCGD_ALLBUF)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001755 for (buf2 = firstbuf; buf2 != NULL; buf2 = buf2->b_next)
1756 if (bufIsChanged(buf2)
1757 && (buf2->b_ffname != NULL
1758# ifdef FEAT_BROWSE
1759 || cmdmod.browse
1760# endif
1761 ))
1762 ++count;
1763# ifdef FEAT_AUTOCMD
1764 if (!buf_valid(buf))
1765 /* Autocommand deleted buffer, oops! It's not changed now. */
1766 return FALSE;
1767# endif
1768 dialog_changed(buf, count > 1);
1769# ifdef FEAT_AUTOCMD
1770 if (!buf_valid(buf))
1771 /* Autocommand deleted buffer, oops! It's not changed now. */
1772 return FALSE;
1773# endif
1774 return bufIsChanged(buf);
1775 }
1776#endif
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001777 if (flags & CCGD_EXCMD)
1778 EMSG(_(e_nowrtmsg));
1779 else
1780 EMSG(_(e_nowrtmsg_nobang));
Bram Moolenaar071d4272004-06-13 20:20:40 +00001781 return TRUE;
1782 }
1783 return FALSE;
1784}
1785
1786#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG) || defined(PROTO)
1787
1788#if defined(FEAT_BROWSE) || defined(PROTO)
1789/*
1790 * When wanting to write a file without a file name, ask the user for a name.
1791 */
1792 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001793browse_save_fname(buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001794{
1795 if (buf->b_fname == NULL)
1796 {
1797 char_u *fname;
1798
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00001799 fname = do_browse(BROWSE_SAVE, (char_u *)_("Save As"),
1800 NULL, NULL, NULL, NULL, buf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001801 if (fname != NULL)
1802 {
1803 if (setfname(buf, fname, NULL, TRUE) == OK)
1804 buf->b_flags |= BF_NOTEDITED;
1805 vim_free(fname);
1806 }
1807 }
1808}
1809#endif
1810
1811/*
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02001812 * Ask the user what to do when abandoning a changed buffer.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001813 * Must check 'write' option first!
1814 */
1815 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001816dialog_changed(
1817 buf_T *buf,
1818 int checkall) /* may abandon all changed buffers */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001819{
Bram Moolenaard9462e32011-04-11 21:35:11 +02001820 char_u buff[DIALOG_MSG_SIZE];
Bram Moolenaar071d4272004-06-13 20:20:40 +00001821 int ret;
1822 buf_T *buf2;
Bram Moolenaar8218f602012-04-25 17:32:18 +02001823 exarg_T ea;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001824
Bram Moolenaar82cf9b62005-06-07 21:09:25 +00001825 dialog_msg(buff, _("Save changes to \"%s\"?"),
Bram Moolenaar071d4272004-06-13 20:20:40 +00001826 (buf->b_fname != NULL) ?
1827 buf->b_fname : (char_u *)_("Untitled"));
1828 if (checkall)
1829 ret = vim_dialog_yesnoallcancel(VIM_QUESTION, NULL, buff, 1);
1830 else
1831 ret = vim_dialog_yesnocancel(VIM_QUESTION, NULL, buff, 1);
1832
Bram Moolenaar8218f602012-04-25 17:32:18 +02001833 /* Init ea pseudo-structure, this is needed for the check_overwrite()
1834 * function. */
1835 ea.append = ea.forceit = FALSE;
1836
Bram Moolenaar071d4272004-06-13 20:20:40 +00001837 if (ret == VIM_YES)
1838 {
1839#ifdef FEAT_BROWSE
1840 /* May get file name, when there is none */
1841 browse_save_fname(buf);
1842#endif
Bram Moolenaar8218f602012-04-25 17:32:18 +02001843 if (buf->b_fname != NULL && check_overwrite(&ea, buf,
1844 buf->b_fname, buf->b_ffname, FALSE) == OK)
1845 /* didn't hit Cancel */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001846 (void)buf_write_all(buf, FALSE);
1847 }
1848 else if (ret == VIM_NO)
1849 {
1850 unchanged(buf, TRUE);
1851 }
1852 else if (ret == VIM_ALL)
1853 {
1854 /*
1855 * Write all modified files that can be written.
1856 * Skip readonly buffers, these need to be confirmed
1857 * individually.
1858 */
1859 for (buf2 = firstbuf; buf2 != NULL; buf2 = buf2->b_next)
1860 {
1861 if (bufIsChanged(buf2)
1862 && (buf2->b_ffname != NULL
1863#ifdef FEAT_BROWSE
1864 || cmdmod.browse
1865#endif
1866 )
1867 && !buf2->b_p_ro)
1868 {
1869#ifdef FEAT_BROWSE
1870 /* May get file name, when there is none */
1871 browse_save_fname(buf2);
1872#endif
Bram Moolenaar8218f602012-04-25 17:32:18 +02001873 if (buf2->b_fname != NULL && check_overwrite(&ea, buf2,
1874 buf2->b_fname, buf2->b_ffname, FALSE) == OK)
1875 /* didn't hit Cancel */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001876 (void)buf_write_all(buf2, FALSE);
1877#ifdef FEAT_AUTOCMD
1878 /* an autocommand may have deleted the buffer */
1879 if (!buf_valid(buf2))
1880 buf2 = firstbuf;
1881#endif
1882 }
1883 }
1884 }
1885 else if (ret == VIM_DISCARDALL)
1886 {
1887 /*
1888 * mark all buffers as unchanged
1889 */
1890 for (buf2 = firstbuf; buf2 != NULL; buf2 = buf2->b_next)
1891 unchanged(buf2, TRUE);
1892 }
1893}
1894#endif
1895
1896/*
1897 * Return TRUE if the buffer "buf" can be abandoned, either by making it
1898 * hidden, autowriting it or unloading it.
1899 */
1900 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001901can_abandon(buf_T *buf, int forceit)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001902{
1903 return ( P_HID(buf)
1904 || !bufIsChanged(buf)
1905 || buf->b_nwindows > 1
1906 || autowrite(buf, forceit) == OK
1907 || forceit);
1908}
1909
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01001910static void add_bufnum(int *bufnrs, int *bufnump, int nr);
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001911
1912/*
1913 * Add a buffer number to "bufnrs", unless it's already there.
1914 */
1915 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001916add_bufnum(int *bufnrs, int *bufnump, int nr)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001917{
1918 int i;
1919
1920 for (i = 0; i < *bufnump; ++i)
1921 if (bufnrs[i] == nr)
1922 return;
1923 bufnrs[*bufnump] = nr;
1924 *bufnump = *bufnump + 1;
1925}
1926
Bram Moolenaar071d4272004-06-13 20:20:40 +00001927/*
1928 * Return TRUE if any buffer was changed and cannot be abandoned.
1929 * That changed buffer becomes the current buffer.
Bram Moolenaar027387f2016-01-02 22:25:52 +01001930 * When "unload" is true the current buffer is unloaded instead of making it
1931 * hidden. This is used for ":q!".
Bram Moolenaar071d4272004-06-13 20:20:40 +00001932 */
1933 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001934check_changed_any(
1935 int hidden, /* Only check hidden buffers */
1936 int unload)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001937{
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001938 int ret = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001939 buf_T *buf;
1940 int save;
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001941 int i;
1942 int bufnum = 0;
1943 int bufcount = 0;
1944 int *bufnrs;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001945#ifdef FEAT_WINDOWS
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001946 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001947 win_T *wp;
1948#endif
1949
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001950 for (buf = firstbuf; buf != NULL; buf = buf->b_next)
1951 ++bufcount;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001952
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001953 if (bufcount == 0)
1954 return FALSE;
1955
1956 bufnrs = (int *)alloc(sizeof(int) * bufcount);
1957 if (bufnrs == NULL)
1958 return FALSE;
1959
1960 /* curbuf */
1961 bufnrs[bufnum++] = curbuf->b_fnum;
1962#ifdef FEAT_WINDOWS
1963 /* buf in curtab */
1964 FOR_ALL_WINDOWS(wp)
1965 if (wp->w_buffer != curbuf)
1966 add_bufnum(bufnrs, &bufnum, wp->w_buffer->b_fnum);
1967
1968 /* buf in other tab */
1969 for (tp = first_tabpage; tp != NULL; tp = tp->tp_next)
1970 if (tp != curtab)
1971 for (wp = tp->tp_firstwin; wp != NULL; wp = wp->w_next)
1972 add_bufnum(bufnrs, &bufnum, wp->w_buffer->b_fnum);
1973#endif
1974 /* any other buf */
1975 for (buf = firstbuf; buf != NULL; buf = buf->b_next)
1976 add_bufnum(bufnrs, &bufnum, buf->b_fnum);
1977
1978 for (i = 0; i < bufnum; ++i)
1979 {
1980 buf = buflist_findnr(bufnrs[i]);
1981 if (buf == NULL)
1982 continue;
1983 if ((!hidden || buf->b_nwindows == 0) && bufIsChanged(buf))
1984 {
1985 /* Try auto-writing the buffer. If this fails but the buffer no
1986 * longer exists it's not changed, that's OK. */
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001987 if (check_changed(buf, (p_awa ? CCGD_AW : 0)
1988 | CCGD_MULTWIN
1989 | CCGD_ALLBUF) && buf_valid(buf))
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001990 break; /* didn't save - still changes */
1991 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001992 }
1993
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001994 if (i >= bufnum)
1995 goto theend;
1996
1997 ret = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001998 exiting = FALSE;
1999#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG)
2000 /*
2001 * When ":confirm" used, don't give an error message.
2002 */
2003 if (!(p_confirm || cmdmod.confirm))
2004#endif
2005 {
2006 /* There must be a wait_return for this message, do_buffer()
2007 * may cause a redraw. But wait_return() is a no-op when vgetc()
2008 * is busy (Quit used from window menu), then make sure we don't
2009 * cause a scroll up. */
Bram Moolenaar61660ea2006-04-07 21:40:07 +00002010 if (vgetc_busy > 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002011 {
2012 msg_row = cmdline_row;
2013 msg_col = 0;
2014 msg_didout = FALSE;
2015 }
2016 if (EMSG2(_("E162: No write since last change for buffer \"%s\""),
Bram Moolenaare1704ba2012-10-03 18:25:00 +02002017 buf_spname(buf) != NULL ? buf_spname(buf) : buf->b_fname))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002018 {
2019 save = no_wait_return;
2020 no_wait_return = FALSE;
2021 wait_return(FALSE);
2022 no_wait_return = save;
2023 }
2024 }
2025
2026#ifdef FEAT_WINDOWS
2027 /* Try to find a window that contains the buffer. */
2028 if (buf != curbuf)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002029 FOR_ALL_TAB_WINDOWS(tp, wp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002030 if (wp->w_buffer == buf)
2031 {
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002032 goto_tabpage_win(tp, wp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002033# ifdef FEAT_AUTOCMD
2034 /* Paranoia: did autocms wipe out the buffer with changes? */
2035 if (!buf_valid(buf))
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002036 {
2037 goto theend;
2038 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002039# endif
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002040 goto buf_found;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002041 }
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002042buf_found:
Bram Moolenaar071d4272004-06-13 20:20:40 +00002043#endif
2044
2045 /* Open the changed buffer in the current window. */
2046 if (buf != curbuf)
Bram Moolenaar027387f2016-01-02 22:25:52 +01002047 set_curbuf(buf, unload ? DOBUF_UNLOAD : DOBUF_GOTO);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002048
Bram Moolenaar970a1b82012-03-23 18:39:18 +01002049theend:
2050 vim_free(bufnrs);
2051 return ret;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002052}
2053
2054/*
2055 * return FAIL if there is no file name, OK if there is one
2056 * give error message for FAIL
2057 */
2058 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002059check_fname(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002060{
2061 if (curbuf->b_ffname == NULL)
2062 {
2063 EMSG(_(e_noname));
2064 return FAIL;
2065 }
2066 return OK;
2067}
2068
2069/*
2070 * flush the contents of a buffer, unless it has no file name
2071 *
2072 * return FAIL for failure, OK otherwise
2073 */
2074 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002075buf_write_all(buf_T *buf, int forceit)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002076{
2077 int retval;
2078#ifdef FEAT_AUTOCMD
2079 buf_T *old_curbuf = curbuf;
2080#endif
2081
2082 retval = (buf_write(buf, buf->b_ffname, buf->b_fname,
2083 (linenr_T)1, buf->b_ml.ml_line_count, NULL,
2084 FALSE, forceit, TRUE, FALSE));
2085#ifdef FEAT_AUTOCMD
2086 if (curbuf != old_curbuf)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00002087 {
2088 msg_source(hl_attr(HLF_W));
Bram Moolenaar071d4272004-06-13 20:20:40 +00002089 MSG(_("Warning: Entered other buffer unexpectedly (check autocommands)"));
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00002090 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002091#endif
2092 return retval;
2093}
2094
2095/*
2096 * Code to handle the argument list.
2097 */
2098
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01002099static char_u *do_one_arg(char_u *str);
2100static int do_arglist(char_u *str, int what, int after);
2101static void alist_check_arg_idx(void);
2102static int editing_arg_idx(win_T *win);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002103#ifdef FEAT_LISTCMDS
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01002104static int alist_add_list(int count, char_u **files, int after);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002105#endif
2106#define AL_SET 1
2107#define AL_ADD 2
2108#define AL_DEL 3
2109
Bram Moolenaar071d4272004-06-13 20:20:40 +00002110/*
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002111 * Isolate one argument, taking backticks.
2112 * Changes the argument in-place, puts a NUL after it. Backticks remain.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002113 * Return a pointer to the start of the next argument.
2114 */
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002115 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002116do_one_arg(char_u *str)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002117{
2118 char_u *p;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002119 int inbacktick;
2120
Bram Moolenaar071d4272004-06-13 20:20:40 +00002121 inbacktick = FALSE;
2122 for (p = str; *str; ++str)
2123 {
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002124 /* When the backslash is used for escaping the special meaning of a
2125 * character we need to keep it until wildcard expansion. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002126 if (rem_backslash(str))
2127 {
2128 *p++ = *str++;
2129 *p++ = *str;
2130 }
2131 else
2132 {
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002133 /* An item ends at a space not in backticks */
2134 if (!inbacktick && vim_isspace(*str))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002135 break;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002136 if (*str == '`')
Bram Moolenaar071d4272004-06-13 20:20:40 +00002137 inbacktick ^= TRUE;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002138 *p++ = *str;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002139 }
2140 }
2141 str = skipwhite(str);
2142 *p = NUL;
2143
2144 return str;
2145}
2146
Bram Moolenaar86b68352004-12-27 21:59:20 +00002147/*
2148 * Separate the arguments in "str" and return a list of pointers in the
2149 * growarray "gap".
2150 */
2151 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002152get_arglist(garray_T *gap, char_u *str)
Bram Moolenaar86b68352004-12-27 21:59:20 +00002153{
2154 ga_init2(gap, (int)sizeof(char_u *), 20);
2155 while (*str != NUL)
2156 {
2157 if (ga_grow(gap, 1) == FAIL)
2158 {
2159 ga_clear(gap);
2160 return FAIL;
2161 }
2162 ((char_u **)gap->ga_data)[gap->ga_len++] = str;
2163
2164 /* Isolate one argument, change it in-place, put a NUL after it. */
2165 str = do_one_arg(str);
2166 }
2167 return OK;
2168}
2169
Bram Moolenaar7df351e2006-01-23 22:30:28 +00002170#if defined(FEAT_QUICKFIX) || defined(FEAT_SYN_HL) || defined(PROTO)
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002171/*
2172 * Parse a list of arguments (file names), expand them and return in
Bram Moolenaar8f5c6f02012-06-29 12:57:06 +02002173 * "fnames[fcountp]". When "wig" is TRUE, removes files matching 'wildignore'.
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002174 * Return FAIL or OK.
2175 */
2176 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002177get_arglist_exp(
2178 char_u *str,
2179 int *fcountp,
2180 char_u ***fnamesp,
2181 int wig)
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002182{
2183 garray_T ga;
2184 int i;
2185
2186 if (get_arglist(&ga, str) == FAIL)
2187 return FAIL;
Bram Moolenaar8f5c6f02012-06-29 12:57:06 +02002188 if (wig == TRUE)
2189 i = expand_wildcards(ga.ga_len, (char_u **)ga.ga_data,
2190 fcountp, fnamesp, EW_FILE|EW_NOTFOUND);
2191 else
2192 i = gen_expand_wildcards(ga.ga_len, (char_u **)ga.ga_data,
2193 fcountp, fnamesp, EW_FILE|EW_NOTFOUND);
2194
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002195 ga_clear(&ga);
2196 return i;
2197}
2198#endif
2199
Bram Moolenaar071d4272004-06-13 20:20:40 +00002200#if defined(FEAT_GUI) || defined(FEAT_CLIENTSERVER) || defined(PROTO)
2201/*
2202 * Redefine the argument list.
2203 */
2204 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002205set_arglist(char_u *str)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002206{
2207 do_arglist(str, AL_SET, 0);
2208}
2209#endif
2210
2211/*
2212 * "what" == AL_SET: Redefine the argument list to 'str'.
2213 * "what" == AL_ADD: add files in 'str' to the argument list after "after".
2214 * "what" == AL_DEL: remove files in 'str' from the argument list.
2215 *
2216 * Return FAIL for failure, OK otherwise.
2217 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002218 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002219do_arglist(
2220 char_u *str,
Bram Moolenaarf1d25012016-03-03 12:22:53 +01002221 int what,
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002222 int after UNUSED) /* 0 means before first one */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002223{
2224 garray_T new_ga;
2225 int exp_count;
2226 char_u **exp_files;
2227 int i;
2228#ifdef FEAT_LISTCMDS
2229 char_u *p;
2230 int match;
2231#endif
2232
2233 /*
Bram Moolenaar2faa29f2016-01-23 23:02:34 +01002234 * Set default argument for ":argadd" command.
2235 */
2236 if (what == AL_ADD && *str == NUL)
2237 {
2238 if (curbuf->b_ffname == NULL)
2239 return FAIL;
2240 str = curbuf->b_fname;
2241 }
2242
2243 /*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002244 * Collect all file name arguments in "new_ga".
2245 */
Bram Moolenaar86b68352004-12-27 21:59:20 +00002246 if (get_arglist(&new_ga, str) == FAIL)
2247 return FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002248
2249#ifdef FEAT_LISTCMDS
2250 if (what == AL_DEL)
2251 {
2252 regmatch_T regmatch;
2253 int didone;
2254
2255 /*
2256 * Delete the items: use each item as a regexp and find a match in the
2257 * argument list.
2258 */
Bram Moolenaar71afbfe2013-03-19 16:49:16 +01002259 regmatch.rm_ic = p_fic; /* ignore case when 'fileignorecase' is set */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002260 for (i = 0; i < new_ga.ga_len && !got_int; ++i)
2261 {
2262 p = ((char_u **)new_ga.ga_data)[i];
2263 p = file_pat_to_reg_pat(p, NULL, NULL, FALSE);
2264 if (p == NULL)
2265 break;
2266 regmatch.regprog = vim_regcomp(p, p_magic ? RE_MAGIC : 0);
2267 if (regmatch.regprog == NULL)
2268 {
2269 vim_free(p);
2270 break;
2271 }
2272
2273 didone = FALSE;
2274 for (match = 0; match < ARGCOUNT; ++match)
2275 if (vim_regexec(&regmatch, alist_name(&ARGLIST[match]),
2276 (colnr_T)0))
2277 {
2278 didone = TRUE;
2279 vim_free(ARGLIST[match].ae_fname);
2280 mch_memmove(ARGLIST + match, ARGLIST + match + 1,
2281 (ARGCOUNT - match - 1) * sizeof(aentry_T));
2282 --ALIST(curwin)->al_ga.ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002283 if (curwin->w_arg_idx > match)
2284 --curwin->w_arg_idx;
2285 --match;
2286 }
2287
Bram Moolenaar473de612013-06-08 18:19:48 +02002288 vim_regfree(regmatch.regprog);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002289 vim_free(p);
2290 if (!didone)
2291 EMSG2(_(e_nomatch2), ((char_u **)new_ga.ga_data)[i]);
2292 }
2293 ga_clear(&new_ga);
2294 }
2295 else
2296#endif
2297 {
2298 i = expand_wildcards(new_ga.ga_len, (char_u **)new_ga.ga_data,
2299 &exp_count, &exp_files, EW_DIR|EW_FILE|EW_ADDSLASH|EW_NOTFOUND);
2300 ga_clear(&new_ga);
Bram Moolenaar2db5c3b2016-01-16 22:49:34 +01002301 if (i == FAIL || exp_count == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002302 {
2303 EMSG(_(e_nomatch));
2304 return FAIL;
2305 }
2306
2307#ifdef FEAT_LISTCMDS
2308 if (what == AL_ADD)
2309 {
2310 (void)alist_add_list(exp_count, exp_files, after);
2311 vim_free(exp_files);
2312 }
2313 else /* what == AL_SET */
2314#endif
Bram Moolenaar86b68352004-12-27 21:59:20 +00002315 alist_set(ALIST(curwin), exp_count, exp_files, FALSE, NULL, 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002316 }
2317
2318 alist_check_arg_idx();
2319
2320 return OK;
2321}
2322
2323/*
2324 * Check the validity of the arg_idx for each other window.
2325 */
2326 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002327alist_check_arg_idx(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002328{
2329#ifdef FEAT_WINDOWS
2330 win_T *win;
Bram Moolenaarf740b292006-02-16 22:11:02 +00002331 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002332
Bram Moolenaarf740b292006-02-16 22:11:02 +00002333 FOR_ALL_TAB_WINDOWS(tp, win)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002334 if (win->w_alist == curwin->w_alist)
2335 check_arg_idx(win);
2336#else
2337 check_arg_idx(curwin);
2338#endif
2339}
2340
2341/*
Bram Moolenaarb8ff1fb2012-02-04 21:59:01 +01002342 * Return TRUE if window "win" is editing the file at the current argument
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002343 * index.
2344 */
2345 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002346editing_arg_idx(win_T *win)
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002347{
2348 return !(win->w_arg_idx >= WARGCOUNT(win)
2349 || (win->w_buffer->b_fnum
2350 != WARGLIST(win)[win->w_arg_idx].ae_fnum
2351 && (win->w_buffer->b_ffname == NULL
2352 || !(fullpathcmp(
2353 alist_name(&WARGLIST(win)[win->w_arg_idx]),
2354 win->w_buffer->b_ffname, TRUE) & FPC_SAME))));
2355}
2356
2357/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002358 * Check if window "win" is editing the w_arg_idx file in its argument list.
2359 */
2360 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002361check_arg_idx(win_T *win)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002362{
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002363 if (WARGCOUNT(win) > 1 && !editing_arg_idx(win))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002364 {
2365 /* We are not editing the current entry in the argument list.
2366 * Set "arg_had_last" if we are editing the last one. */
2367 win->w_arg_idx_invalid = TRUE;
2368 if (win->w_arg_idx != WARGCOUNT(win) - 1
2369 && arg_had_last == FALSE
2370#ifdef FEAT_WINDOWS
2371 && ALIST(win) == &global_alist
2372#endif
2373 && GARGCOUNT > 0
2374 && win->w_arg_idx < GARGCOUNT
2375 && (win->w_buffer->b_fnum == GARGLIST[GARGCOUNT - 1].ae_fnum
2376 || (win->w_buffer->b_ffname != NULL
2377 && (fullpathcmp(alist_name(&GARGLIST[GARGCOUNT - 1]),
2378 win->w_buffer->b_ffname, TRUE) & FPC_SAME))))
2379 arg_had_last = TRUE;
2380 }
2381 else
2382 {
2383 /* We are editing the current entry in the argument list.
2384 * Set "arg_had_last" if it's also the last one */
2385 win->w_arg_idx_invalid = FALSE;
2386 if (win->w_arg_idx == WARGCOUNT(win) - 1
2387#ifdef FEAT_WINDOWS
2388 && win->w_alist == &global_alist
2389#endif
2390 )
2391 arg_had_last = TRUE;
2392 }
2393}
2394
2395/*
2396 * ":args", ":argslocal" and ":argsglobal".
2397 */
2398 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002399ex_args(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002400{
2401 int i;
2402
2403 if (eap->cmdidx != CMD_args)
2404 {
2405#if defined(FEAT_WINDOWS) && defined(FEAT_LISTCMDS)
2406 alist_unlink(ALIST(curwin));
2407 if (eap->cmdidx == CMD_argglobal)
2408 ALIST(curwin) = &global_alist;
2409 else /* eap->cmdidx == CMD_arglocal */
2410 alist_new();
2411#else
2412 ex_ni(eap);
2413 return;
2414#endif
2415 }
2416
2417 if (!ends_excmd(*eap->arg))
2418 {
2419 /*
2420 * ":args file ..": define new argument list, handle like ":next"
2421 * Also for ":argslocal file .." and ":argsglobal file ..".
2422 */
2423 ex_next(eap);
2424 }
2425 else
2426#if defined(FEAT_WINDOWS) && defined(FEAT_LISTCMDS)
2427 if (eap->cmdidx == CMD_args)
2428#endif
2429 {
2430 /*
2431 * ":args": list arguments.
2432 */
2433 if (ARGCOUNT > 0)
2434 {
2435 /* Overwrite the command, for a short list there is no scrolling
2436 * required and no wait_return(). */
2437 gotocmdline(TRUE);
2438 for (i = 0; i < ARGCOUNT; ++i)
2439 {
2440 if (i == curwin->w_arg_idx)
2441 msg_putchar('[');
2442 msg_outtrans(alist_name(&ARGLIST[i]));
2443 if (i == curwin->w_arg_idx)
2444 msg_putchar(']');
2445 msg_putchar(' ');
2446 }
2447 }
2448 }
2449#if defined(FEAT_WINDOWS) && defined(FEAT_LISTCMDS)
2450 else if (eap->cmdidx == CMD_arglocal)
2451 {
2452 garray_T *gap = &curwin->w_alist->al_ga;
2453
2454 /*
2455 * ":argslocal": make a local copy of the global argument list.
2456 */
2457 if (ga_grow(gap, GARGCOUNT) == OK)
2458 for (i = 0; i < GARGCOUNT; ++i)
2459 if (GARGLIST[i].ae_fname != NULL)
2460 {
2461 AARGLIST(curwin->w_alist)[gap->ga_len].ae_fname =
2462 vim_strsave(GARGLIST[i].ae_fname);
2463 AARGLIST(curwin->w_alist)[gap->ga_len].ae_fnum =
2464 GARGLIST[i].ae_fnum;
2465 ++gap->ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002466 }
2467 }
2468#endif
2469}
2470
2471/*
2472 * ":previous", ":sprevious", ":Next" and ":sNext".
2473 */
2474 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002475ex_previous(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002476{
2477 /* If past the last one already, go to the last one. */
2478 if (curwin->w_arg_idx - (int)eap->line2 >= ARGCOUNT)
2479 do_argfile(eap, ARGCOUNT - 1);
2480 else
2481 do_argfile(eap, curwin->w_arg_idx - (int)eap->line2);
2482}
2483
2484/*
2485 * ":rewind", ":first", ":sfirst" and ":srewind".
2486 */
2487 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002488ex_rewind(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002489{
2490 do_argfile(eap, 0);
2491}
2492
2493/*
2494 * ":last" and ":slast".
2495 */
2496 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002497ex_last(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002498{
2499 do_argfile(eap, ARGCOUNT - 1);
2500}
2501
2502/*
2503 * ":argument" and ":sargument".
2504 */
2505 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002506ex_argument(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002507{
2508 int i;
2509
2510 if (eap->addr_count > 0)
2511 i = eap->line2 - 1;
2512 else
2513 i = curwin->w_arg_idx;
2514 do_argfile(eap, i);
2515}
2516
2517/*
2518 * Edit file "argn" of the argument lists.
2519 */
2520 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002521do_argfile(exarg_T *eap, int argn)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002522{
2523 int other;
2524 char_u *p;
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002525 int old_arg_idx = curwin->w_arg_idx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002526
2527 if (argn < 0 || argn >= ARGCOUNT)
2528 {
2529 if (ARGCOUNT <= 1)
2530 EMSG(_("E163: There is only one file to edit"));
2531 else if (argn < 0)
2532 EMSG(_("E164: Cannot go before first file"));
2533 else
2534 EMSG(_("E165: Cannot go beyond last file"));
2535 }
2536 else
2537 {
2538 setpcmark();
2539#ifdef FEAT_GUI
2540 need_mouse_correct = TRUE;
2541#endif
2542
2543#ifdef FEAT_WINDOWS
Bram Moolenaardf1bdc92006-02-23 21:32:16 +00002544 /* split window or create new tab page first */
2545 if (*eap->cmd == 's' || cmdmod.tab != 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002546 {
2547 if (win_split(0, 0) == FAIL)
2548 return;
Bram Moolenaar3368ea22010-09-21 16:56:35 +02002549 RESET_BINDING(curwin);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002550 }
2551 else
2552#endif
2553 {
2554 /*
2555 * if 'hidden' set, only check for changed file when re-editing
2556 * the same buffer
2557 */
2558 other = TRUE;
2559 if (P_HID(curbuf))
2560 {
2561 p = fix_fname(alist_name(&ARGLIST[argn]));
2562 other = otherfile(p);
2563 vim_free(p);
2564 }
2565 if ((!P_HID(curbuf) || !other)
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002566 && check_changed(curbuf, CCGD_AW
2567 | (other ? 0 : CCGD_MULTWIN)
2568 | (eap->forceit ? CCGD_FORCEIT : 0)
2569 | CCGD_EXCMD))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002570 return;
2571 }
2572
2573 curwin->w_arg_idx = argn;
2574 if (argn == ARGCOUNT - 1
2575#ifdef FEAT_WINDOWS
2576 && curwin->w_alist == &global_alist
2577#endif
2578 )
2579 arg_had_last = TRUE;
2580
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002581 /* Edit the file; always use the last known line number.
2582 * When it fails (e.g. Abort for already edited file) restore the
2583 * argument index. */
2584 if (do_ecmd(0, alist_name(&ARGLIST[curwin->w_arg_idx]), NULL,
Bram Moolenaar071d4272004-06-13 20:20:40 +00002585 eap, ECMD_LAST,
Bram Moolenaar701f7af2008-11-15 13:12:07 +00002586 (P_HID(curwin->w_buffer) ? ECMD_HIDE : 0)
2587 + (eap->forceit ? ECMD_FORCEIT : 0), curwin) == FAIL)
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002588 curwin->w_arg_idx = old_arg_idx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002589 /* like Vi: set the mark where the cursor is in the file. */
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002590 else if (eap->cmdidx != CMD_argdo)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002591 setmark('\'');
2592 }
2593}
2594
2595/*
2596 * ":next", and commands that behave like it.
2597 */
2598 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002599ex_next(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002600{
2601 int i;
2602
2603 /*
2604 * check for changed buffer now, if this fails the argument list is not
2605 * redefined.
2606 */
2607 if ( P_HID(curbuf)
2608 || eap->cmdidx == CMD_snext
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002609 || !check_changed(curbuf, CCGD_AW
2610 | (eap->forceit ? CCGD_FORCEIT : 0)
2611 | CCGD_EXCMD))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002612 {
2613 if (*eap->arg != NUL) /* redefine file list */
2614 {
2615 if (do_arglist(eap->arg, AL_SET, 0) == FAIL)
2616 return;
2617 i = 0;
2618 }
2619 else
2620 i = curwin->w_arg_idx + (int)eap->line2;
2621 do_argfile(eap, i);
2622 }
2623}
2624
Bram Moolenaar0106e3d2016-02-23 18:55:43 +01002625#if defined(FEAT_LISTCMDS) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002626/*
2627 * ":argedit"
2628 */
2629 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002630ex_argedit(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002631{
2632 int fnum;
2633 int i;
2634 char_u *s;
2635
2636 /* Add the argument to the buffer list and get the buffer number. */
2637 fnum = buflist_add(eap->arg, BLN_LISTED);
2638
2639 /* Check if this argument is already in the argument list. */
2640 for (i = 0; i < ARGCOUNT; ++i)
2641 if (ARGLIST[i].ae_fnum == fnum)
2642 break;
2643 if (i == ARGCOUNT)
2644 {
2645 /* Can't find it, add it to the argument list. */
2646 s = vim_strsave(eap->arg);
2647 if (s == NULL)
2648 return;
2649 i = alist_add_list(1, &s,
2650 eap->addr_count > 0 ? (int)eap->line2 : curwin->w_arg_idx + 1);
2651 if (i < 0)
2652 return;
2653 curwin->w_arg_idx = i;
2654 }
2655
2656 alist_check_arg_idx();
2657
2658 /* Edit the argument. */
2659 do_argfile(eap, i);
2660}
2661
2662/*
2663 * ":argadd"
2664 */
2665 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002666ex_argadd(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002667{
2668 do_arglist(eap->arg, AL_ADD,
2669 eap->addr_count > 0 ? (int)eap->line2 : curwin->w_arg_idx + 1);
2670#ifdef FEAT_TITLE
2671 maketitle();
2672#endif
2673}
2674
2675/*
2676 * ":argdelete"
2677 */
2678 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002679ex_argdelete(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002680{
2681 int i;
2682 int n;
2683
2684 if (eap->addr_count > 0)
2685 {
2686 /* ":1,4argdel": Delete all arguments in the range. */
2687 if (eap->line2 > ARGCOUNT)
2688 eap->line2 = ARGCOUNT;
2689 n = eap->line2 - eap->line1 + 1;
2690 if (*eap->arg != NUL || n <= 0)
2691 EMSG(_(e_invarg));
2692 else
2693 {
2694 for (i = eap->line1; i <= eap->line2; ++i)
2695 vim_free(ARGLIST[i - 1].ae_fname);
2696 mch_memmove(ARGLIST + eap->line1 - 1, ARGLIST + eap->line2,
2697 (size_t)((ARGCOUNT - eap->line2) * sizeof(aentry_T)));
2698 ALIST(curwin)->al_ga.ga_len -= n;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002699 if (curwin->w_arg_idx >= eap->line2)
2700 curwin->w_arg_idx -= n;
2701 else if (curwin->w_arg_idx > eap->line1)
2702 curwin->w_arg_idx = eap->line1;
Bram Moolenaar72defda2016-01-17 18:04:33 +01002703 if (ARGCOUNT == 0)
2704 curwin->w_arg_idx = 0;
2705 else if (curwin->w_arg_idx >= ARGCOUNT)
2706 curwin->w_arg_idx = ARGCOUNT - 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002707 }
2708 }
2709 else if (*eap->arg == NUL)
2710 EMSG(_(e_argreq));
2711 else
2712 do_arglist(eap->arg, AL_DEL, 0);
2713#ifdef FEAT_TITLE
2714 maketitle();
2715#endif
2716}
2717
2718/*
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002719 * ":argdo", ":windo", ":bufdo", ":tabdo", ":cdo", ":ldo", ":cfdo" and ":lfdo"
Bram Moolenaar071d4272004-06-13 20:20:40 +00002720 */
2721 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002722ex_listdo(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002723{
2724 int i;
2725#ifdef FEAT_WINDOWS
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002726 win_T *wp;
2727 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002728#endif
Bram Moolenaare25bb902015-02-27 20:33:37 +01002729 buf_T *buf = curbuf;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002730 int next_fnum = 0;
2731#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
2732 char_u *save_ei = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002733#endif
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002734 char_u *p_shm_save;
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002735#ifdef FEAT_QUICKFIX
Bram Moolenaared84b762015-09-09 22:35:29 +02002736 int qf_size = 0;
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002737 int qf_idx;
2738#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002739
2740#ifndef FEAT_WINDOWS
2741 if (eap->cmdidx == CMD_windo)
2742 {
2743 ex_ni(eap);
2744 return;
2745 }
2746#endif
2747
Bram Moolenaar0106e3d2016-02-23 18:55:43 +01002748#ifndef FEAT_QUICKFIX
2749 if (eap->cmdidx == CMD_cdo || eap->cmdidx == CMD_ldo ||
2750 eap->cmdidx == CMD_cfdo || eap->cmdidx == CMD_lfdo)
2751 {
2752 ex_ni(eap);
2753 return;
2754 }
2755#endif
2756
Bram Moolenaar071d4272004-06-13 20:20:40 +00002757#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002758 if (eap->cmdidx != CMD_windo && eap->cmdidx != CMD_tabdo)
Bram Moolenaardcaf10e2005-01-21 11:55:25 +00002759 /* Don't do syntax HL autocommands. Skipping the syntax file is a
2760 * great speed improvement. */
2761 save_ei = au_event_disable(",Syntax");
Bram Moolenaar071d4272004-06-13 20:20:40 +00002762#endif
Bram Moolenaar6b1ee342014-08-06 18:17:11 +02002763#ifdef FEAT_CLIPBOARD
2764 start_global_changes();
2765#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002766
2767 if (eap->cmdidx == CMD_windo
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002768 || eap->cmdidx == CMD_tabdo
Bram Moolenaar071d4272004-06-13 20:20:40 +00002769 || P_HID(curbuf)
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002770 || !check_changed(curbuf, CCGD_AW
2771 | (eap->forceit ? CCGD_FORCEIT : 0)
2772 | CCGD_EXCMD))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002773 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00002774 i = 0;
Bram Moolenaara162bc52015-01-07 16:54:21 +01002775 /* start at the eap->line1 argument/window/buffer */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002776#ifdef FEAT_WINDOWS
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002777 wp = firstwin;
2778 tp = first_tabpage;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002779#endif
Bram Moolenaara162bc52015-01-07 16:54:21 +01002780 switch (eap->cmdidx)
2781 {
2782#ifdef FEAT_WINDOWS
2783 case CMD_windo:
2784 for ( ; wp != NULL && i + 1 < eap->line1; wp = wp->w_next)
2785 i++;
2786 break;
2787 case CMD_tabdo:
2788 for( ; tp != NULL && i + 1 < eap->line1; tp = tp->tp_next)
2789 i++;
2790 break;
2791#endif
2792 case CMD_argdo:
2793 i = eap->line1 - 1;
2794 break;
Bram Moolenaara162bc52015-01-07 16:54:21 +01002795 default:
2796 break;
2797 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002798 /* set pcmark now */
2799 if (eap->cmdidx == CMD_bufdo)
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002800 {
Bram Moolenaare25bb902015-02-27 20:33:37 +01002801 /* Advance to the first listed buffer after "eap->line1". */
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002802 for (buf = firstbuf; buf != NULL && (buf->b_fnum < eap->line1
Bram Moolenaare25bb902015-02-27 20:33:37 +01002803 || !buf->b_p_bl); buf = buf->b_next)
2804 if (buf->b_fnum > eap->line2)
2805 {
2806 buf = NULL;
2807 break;
2808 }
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002809 if (buf != NULL)
Bram Moolenaare25bb902015-02-27 20:33:37 +01002810 goto_buffer(eap, DOBUF_FIRST, FORWARD, buf->b_fnum);
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002811 }
2812#ifdef FEAT_QUICKFIX
2813 else if (eap->cmdidx == CMD_cdo || eap->cmdidx == CMD_ldo
2814 || eap->cmdidx == CMD_cfdo || eap->cmdidx == CMD_lfdo)
2815 {
2816 qf_size = qf_get_size(eap);
2817 if (qf_size <= 0 || eap->line1 > qf_size)
2818 buf = NULL;
2819 else
2820 {
2821 ex_cc(eap);
2822
2823 buf = curbuf;
2824 i = eap->line1 - 1;
2825 if (eap->addr_count <= 0)
2826 /* default is all the quickfix/location list entries */
2827 eap->line2 = qf_size;
2828 }
2829 }
2830#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002831 else
2832 setpcmark();
2833 listcmd_busy = TRUE; /* avoids setting pcmark below */
2834
Bram Moolenaare25bb902015-02-27 20:33:37 +01002835 while (!got_int && buf != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002836 {
2837 if (eap->cmdidx == CMD_argdo)
2838 {
2839 /* go to argument "i" */
2840 if (i == ARGCOUNT)
2841 break;
2842 /* Don't call do_argfile() when already there, it will try
2843 * reloading the file. */
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002844 if (curwin->w_arg_idx != i || !editing_arg_idx(curwin))
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002845 {
2846 /* Clear 'shm' to avoid that the file message overwrites
2847 * any output from the command. */
2848 p_shm_save = vim_strsave(p_shm);
2849 set_option_value((char_u *)"shm", 0L, (char_u *)"", 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002850 do_argfile(eap, i);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002851 set_option_value((char_u *)"shm", 0L, p_shm_save, 0);
2852 vim_free(p_shm_save);
2853 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002854 if (curwin->w_arg_idx != i)
2855 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002856 }
2857#ifdef FEAT_WINDOWS
2858 else if (eap->cmdidx == CMD_windo)
2859 {
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002860 /* go to window "wp" */
2861 if (!win_valid(wp))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002862 break;
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002863 win_goto(wp);
Bram Moolenaar41423242007-05-03 20:11:13 +00002864 if (curwin != wp)
2865 break; /* something must be wrong */
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002866 wp = curwin->w_next;
2867 }
2868 else if (eap->cmdidx == CMD_tabdo)
2869 {
2870 /* go to window "tp" */
2871 if (!valid_tabpage(tp))
2872 break;
Bram Moolenaar49e649f2013-05-06 04:50:35 +02002873 goto_tabpage_tp(tp, TRUE, TRUE);
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002874 tp = tp->tp_next;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002875 }
2876#endif
2877 else if (eap->cmdidx == CMD_bufdo)
2878 {
2879 /* Remember the number of the next listed buffer, in case
2880 * ":bwipe" is used or autocommands do something strange. */
2881 next_fnum = -1;
2882 for (buf = curbuf->b_next; buf != NULL; buf = buf->b_next)
2883 if (buf->b_p_bl)
2884 {
2885 next_fnum = buf->b_fnum;
2886 break;
2887 }
2888 }
2889
Bram Moolenaara162bc52015-01-07 16:54:21 +01002890 ++i;
2891
Bram Moolenaar071d4272004-06-13 20:20:40 +00002892 /* execute the command */
2893 do_cmdline(eap->arg, eap->getline, eap->cookie,
2894 DOCMD_VERBOSE + DOCMD_NOWAIT);
2895
2896 if (eap->cmdidx == CMD_bufdo)
2897 {
2898 /* Done? */
Bram Moolenaara162bc52015-01-07 16:54:21 +01002899 if (next_fnum < 0 || next_fnum > eap->line2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002900 break;
2901 /* Check if the buffer still exists. */
2902 for (buf = firstbuf; buf != NULL; buf = buf->b_next)
2903 if (buf->b_fnum == next_fnum)
2904 break;
2905 if (buf == NULL)
2906 break;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002907
2908 /* Go to the next buffer. Clear 'shm' to avoid that the file
2909 * message overwrites any output from the command. */
2910 p_shm_save = vim_strsave(p_shm);
2911 set_option_value((char_u *)"shm", 0L, (char_u *)"", 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002912 goto_buffer(eap, DOBUF_FIRST, FORWARD, next_fnum);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002913 set_option_value((char_u *)"shm", 0L, p_shm_save, 0);
2914 vim_free(p_shm_save);
2915
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002916 /* If autocommands took us elsewhere, quit here. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002917 if (curbuf->b_fnum != next_fnum)
2918 break;
2919 }
2920
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002921#ifdef FEAT_QUICKFIX
2922 if (eap->cmdidx == CMD_cdo || eap->cmdidx == CMD_ldo
2923 || eap->cmdidx == CMD_cfdo || eap->cmdidx == CMD_lfdo)
2924 {
2925 if (i >= qf_size || i >= eap->line2)
2926 break;
2927
2928 qf_idx = qf_get_cur_idx(eap);
2929
2930 ex_cnext(eap);
2931
2932 /* If jumping to the next quickfix entry fails, quit here */
2933 if (qf_get_cur_idx(eap) == qf_idx)
2934 break;
2935 }
2936#endif
2937
Bram Moolenaar071d4272004-06-13 20:20:40 +00002938 if (eap->cmdidx == CMD_windo)
2939 {
2940 validate_cursor(); /* cursor may have moved */
2941#ifdef FEAT_SCROLLBIND
2942 /* required when 'scrollbind' has been set */
2943 if (curwin->w_p_scb)
2944 do_check_scrollbind(TRUE);
2945#endif
2946 }
Bram Moolenaara162bc52015-01-07 16:54:21 +01002947
2948#ifdef FEAT_WINDOWS
2949 if (eap->cmdidx == CMD_windo || eap->cmdidx == CMD_tabdo)
2950 if (i+1 > eap->line2)
2951 break;
2952#endif
2953 if (eap->cmdidx == CMD_argdo && i >= eap->line2)
2954 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002955 }
2956 listcmd_busy = FALSE;
2957 }
2958
2959#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
Bram Moolenaar6ac54292005-02-02 23:07:25 +00002960 if (save_ei != NULL)
2961 {
2962 au_event_restore(save_ei);
2963 apply_autocmds(EVENT_SYNTAX, curbuf->b_p_syn,
2964 curbuf->b_fname, TRUE, curbuf);
2965 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002966#endif
Bram Moolenaar6b1ee342014-08-06 18:17:11 +02002967#ifdef FEAT_CLIPBOARD
2968 end_global_changes();
2969#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002970}
2971
2972/*
2973 * Add files[count] to the arglist of the current window after arg "after".
2974 * The file names in files[count] must have been allocated and are taken over.
2975 * Files[] itself is not taken over.
2976 * Returns index of first added argument. Returns -1 when failed (out of mem).
2977 */
2978 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002979alist_add_list(
2980 int count,
2981 char_u **files,
2982 int after) /* where to add: 0 = before first one */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002983{
2984 int i;
Bram Moolenaara24f0a52016-01-17 19:39:00 +01002985 int old_argcount = ARGCOUNT;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002986
2987 if (ga_grow(&ALIST(curwin)->al_ga, count) == OK)
2988 {
2989 if (after < 0)
2990 after = 0;
2991 if (after > ARGCOUNT)
2992 after = ARGCOUNT;
2993 if (after < ARGCOUNT)
2994 mch_memmove(&(ARGLIST[after + count]), &(ARGLIST[after]),
2995 (ARGCOUNT - after) * sizeof(aentry_T));
2996 for (i = 0; i < count; ++i)
2997 {
2998 ARGLIST[after + i].ae_fname = files[i];
2999 ARGLIST[after + i].ae_fnum = buflist_add(files[i], BLN_LISTED);
3000 }
3001 ALIST(curwin)->al_ga.ga_len += count;
Bram Moolenaara24f0a52016-01-17 19:39:00 +01003002 if (old_argcount > 0 && curwin->w_arg_idx >= after)
3003 curwin->w_arg_idx += count;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003004 return after;
3005 }
3006
3007 for (i = 0; i < count; ++i)
3008 vim_free(files[i]);
3009 return -1;
3010}
3011
3012#endif /* FEAT_LISTCMDS */
3013
3014#ifdef FEAT_EVAL
3015/*
3016 * ":compiler[!] {name}"
3017 */
3018 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003019ex_compiler(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003020{
3021 char_u *buf;
3022 char_u *old_cur_comp = NULL;
3023 char_u *p;
3024
3025 if (*eap->arg == NUL)
3026 {
3027 /* List all compiler scripts. */
3028 do_cmdline_cmd((char_u *)"echo globpath(&rtp, 'compiler/*.vim')");
3029 /* ) keep the indenter happy... */
3030 }
3031 else
3032 {
3033 buf = alloc((unsigned)(STRLEN(eap->arg) + 14));
3034 if (buf != NULL)
3035 {
3036 if (eap->forceit)
3037 {
3038 /* ":compiler! {name}" sets global options */
3039 do_cmdline_cmd((char_u *)
3040 "command -nargs=* CompilerSet set <args>");
3041 }
3042 else
3043 {
3044 /* ":compiler! {name}" sets local options.
3045 * To remain backwards compatible "current_compiler" is always
3046 * used. A user's compiler plugin may set it, the distributed
3047 * plugin will then skip the settings. Afterwards set
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003048 * "b:current_compiler" and restore "current_compiler".
3049 * Explicitly prepend "g:" to make it work in a function. */
3050 old_cur_comp = get_var_value((char_u *)"g:current_compiler");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003051 if (old_cur_comp != NULL)
3052 old_cur_comp = vim_strsave(old_cur_comp);
3053 do_cmdline_cmd((char_u *)
3054 "command -nargs=* CompilerSet setlocal <args>");
3055 }
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003056 do_unlet((char_u *)"g:current_compiler", TRUE);
Bram Moolenaar2ce06f62005-01-31 19:19:04 +00003057 do_unlet((char_u *)"b:current_compiler", TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003058
3059 sprintf((char *)buf, "compiler/%s.vim", eap->arg);
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003060 if (source_runtime(buf, DIP_ALL) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003061 EMSG2(_("E666: compiler not supported: %s"), eap->arg);
3062 vim_free(buf);
3063
3064 do_cmdline_cmd((char_u *)":delcommand CompilerSet");
3065
3066 /* Set "b:current_compiler" from "current_compiler". */
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003067 p = get_var_value((char_u *)"g:current_compiler");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003068 if (p != NULL)
3069 set_internal_string_var((char_u *)"b:current_compiler", p);
3070
3071 /* Restore "current_compiler" for ":compiler {name}". */
3072 if (!eap->forceit)
3073 {
3074 if (old_cur_comp != NULL)
3075 {
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003076 set_internal_string_var((char_u *)"g:current_compiler",
Bram Moolenaar071d4272004-06-13 20:20:40 +00003077 old_cur_comp);
3078 vim_free(old_cur_comp);
3079 }
3080 else
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01003081 do_unlet((char_u *)"g:current_compiler", TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003082 }
3083 }
3084 }
3085}
3086#endif
3087
3088/*
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003089 * ":runtime [what] {name}"
Bram Moolenaar071d4272004-06-13 20:20:40 +00003090 */
3091 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003092ex_runtime(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003093{
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003094 char_u *arg = eap->arg;
3095 char_u *p = skiptowhite(arg);
3096 int len = (int)(p - arg);
3097 int flags = eap->forceit ? DIP_ALL : 0;
3098
3099 if (STRNCMP(arg, "START", len) == 0)
3100 {
3101 flags += DIP_START + DIP_NORTP;
3102 arg = skipwhite(arg + len);
3103 }
3104 else if (STRNCMP(arg, "OPT", len) == 0)
3105 {
3106 flags += DIP_OPT + DIP_NORTP;
3107 arg = skipwhite(arg + len);
3108 }
3109 else if (STRNCMP(arg, "PACK", len) == 0)
3110 {
3111 flags += DIP_START + DIP_OPT + DIP_NORTP;
3112 arg = skipwhite(arg + len);
3113 }
3114 else if (STRNCMP(arg, "ALL", len) == 0)
3115 {
3116 flags += DIP_START + DIP_OPT;
3117 arg = skipwhite(arg + len);
3118 }
3119
3120 source_runtime(arg, flags);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003121}
3122
Bram Moolenaar071d4272004-06-13 20:20:40 +00003123 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003124source_callback(char_u *fname, void *cookie UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003125{
Bram Moolenaar910f66f2006-04-05 20:41:53 +00003126 (void)do_source(fname, FALSE, DOSO_NONE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003127}
3128
3129/*
3130 * Source the file "name" from all directories in 'runtimepath'.
3131 * "name" can contain wildcards.
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003132 * When "flags" has DIP_ALL: source all files, otherwise only the first one.
Bram Moolenaar91715872016-03-03 17:13:03 +01003133 *
Bram Moolenaar071d4272004-06-13 20:20:40 +00003134 * return FAIL when no file could be sourced, OK otherwise.
3135 */
3136 int
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003137source_runtime(char_u *name, int flags)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003138{
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003139 return do_in_runtimepath(name, flags, source_callback, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003140}
3141
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003142/*
3143 * Find the file "name" in all directories in "path" and invoke
3144 * "callback(fname, cookie)".
3145 * "name" can contain wildcards.
3146 * When "flags" has DIP_ALL: source all files, otherwise only the first one.
3147 * When "flags" has DIP_DIR: find directories instead of files.
3148 * When "flags" has DIP_ERR: give an error message if there is no match.
3149 *
3150 * return FAIL when no file could be sourced, OK otherwise.
3151 */
Bram Moolenaar6bef5302016-03-12 21:28:26 +01003152 int
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003153do_in_path(
3154 char_u *path,
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003155 char_u *name,
Bram Moolenaar91715872016-03-03 17:13:03 +01003156 int flags,
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003157 void (*callback)(char_u *fname, void *ck),
3158 void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003159{
3160 char_u *rtp;
3161 char_u *np;
3162 char_u *buf;
3163 char_u *rtp_copy;
3164 char_u *tail;
3165 int num_files;
3166 char_u **files;
3167 int i;
3168 int did_one = FALSE;
3169#ifdef AMIGA
3170 struct Process *proc = (struct Process *)FindTask(0L);
3171 APTR save_winptr = proc->pr_WindowPtr;
3172
3173 /* Avoid a requester here for a volume that doesn't exist. */
3174 proc->pr_WindowPtr = (APTR)-1L;
3175#endif
3176
3177 /* Make a copy of 'runtimepath'. Invoking the callback may change the
3178 * value. */
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003179 rtp_copy = vim_strsave(path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003180 buf = alloc(MAXPATHL);
3181 if (buf != NULL && rtp_copy != NULL)
3182 {
Bram Moolenaarc09a6d62013-06-10 21:27:29 +02003183 if (p_verbose > 1 && name != NULL)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003184 {
3185 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00003186 smsg((char_u *)_("Searching for \"%s\" in \"%s\""),
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003187 (char *)name, (char *)path);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003188 verbose_leave();
3189 }
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00003190
Bram Moolenaar071d4272004-06-13 20:20:40 +00003191 /* Loop over all entries in 'runtimepath'. */
3192 rtp = rtp_copy;
Bram Moolenaar91715872016-03-03 17:13:03 +01003193 while (*rtp != NUL && ((flags & DIP_ALL) || !did_one))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003194 {
3195 /* Copy the path from 'runtimepath' to buf[]. */
3196 copy_option_part(&rtp, buf, MAXPATHL, ",");
Bram Moolenaarc09a6d62013-06-10 21:27:29 +02003197 if (name == NULL)
3198 {
3199 (*callback)(buf, (void *) &cookie);
3200 if (!did_one)
3201 did_one = (cookie == NULL);
3202 }
3203 else if (STRLEN(buf) + STRLEN(name) + 2 < MAXPATHL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003204 {
3205 add_pathsep(buf);
3206 tail = buf + STRLEN(buf);
3207
3208 /* Loop over all patterns in "name" */
3209 np = name;
Bram Moolenaar91715872016-03-03 17:13:03 +01003210 while (*np != NUL && ((flags & DIP_ALL) || !did_one))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003211 {
3212 /* Append the pattern from "name" to buf[]. */
3213 copy_option_part(&np, tail, (int)(MAXPATHL - (tail - buf)),
3214 "\t ");
3215
3216 if (p_verbose > 2)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003217 {
3218 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00003219 smsg((char_u *)_("Searching for \"%s\""), buf);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003220 verbose_leave();
3221 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003222
3223 /* Expand wildcards, invoke the callback for each match. */
3224 if (gen_expand_wildcards(1, &buf, &num_files, &files,
Bram Moolenaar91715872016-03-03 17:13:03 +01003225 (flags & DIP_DIR) ? EW_DIR : EW_FILE) == OK)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003226 {
3227 for (i = 0; i < num_files; ++i)
3228 {
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00003229 (*callback)(files[i], cookie);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003230 did_one = TRUE;
Bram Moolenaar91715872016-03-03 17:13:03 +01003231 if (!(flags & DIP_ALL))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003232 break;
3233 }
3234 FreeWild(num_files, files);
3235 }
3236 }
3237 }
3238 }
3239 }
3240 vim_free(buf);
3241 vim_free(rtp_copy);
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003242 if (!did_one && name != NULL)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003243 {
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003244 char *basepath = path == p_rtp ? "runtimepath" : "packpath";
3245
3246 if (flags & DIP_ERR)
3247 EMSG3(_(e_dirnotf), basepath, name);
3248 else if (p_verbose > 0)
3249 {
3250 verbose_enter();
3251 smsg((char_u *)_("not found in '%s': \"%s\""), basepath, name);
3252 verbose_leave();
3253 }
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003254 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003255
3256#ifdef AMIGA
3257 proc->pr_WindowPtr = save_winptr;
3258#endif
3259
3260 return did_one ? OK : FAIL;
3261}
3262
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003263/*
3264 * Find "name" in 'runtimepath'. When found, invoke the callback function for
3265 * it: callback(fname, "cookie")
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003266 * When "flags" has DIP_ALL repeat for all matches, otherwise only the first
3267 * one is used.
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003268 * Returns OK when at least one match found, FAIL otherwise.
3269 *
3270 * If "name" is NULL calls callback for each entry in runtimepath. Cookie is
3271 * passed by reference in this case, setting it to NULL indicates that callback
3272 * has done its job.
3273 */
3274 int
3275do_in_runtimepath(
3276 char_u *name,
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003277 int flags,
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003278 void (*callback)(char_u *fname, void *ck),
3279 void *cookie)
3280{
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003281 int done = FAIL;
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003282 char_u *s;
3283 int len;
3284 char *start_dir = "pack/*/start/*/%s";
3285 char *opt_dir = "pack/*/opt/*/%s";
3286
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003287 if ((flags & DIP_NORTP) == 0)
3288 done = do_in_path(p_rtp, name, flags, callback, cookie);
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003289
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003290 if ((done == FAIL || (flags & DIP_ALL)) && (flags & DIP_START))
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003291 {
Bram Moolenaar1c8b4ed2016-03-17 21:51:03 +01003292 len = (int)(STRLEN(start_dir) + STRLEN(name));
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003293 s = alloc(len);
3294 if (s == NULL)
3295 return FAIL;
3296 vim_snprintf((char *)s, len, start_dir, name);
3297 done = do_in_path(p_pp, s, flags, callback, cookie);
3298 vim_free(s);
3299 }
3300
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003301 if ((done == FAIL || (flags & DIP_ALL)) && (flags & DIP_OPT))
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003302 {
Bram Moolenaar1c8b4ed2016-03-17 21:51:03 +01003303 len = (int)(STRLEN(opt_dir) + STRLEN(name));
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003304 s = alloc(len);
3305 if (s == NULL)
3306 return FAIL;
3307 vim_snprintf((char *)s, len, opt_dir, name);
3308 done = do_in_path(p_pp, s, flags, callback, cookie);
3309 vim_free(s);
3310 }
3311
3312 return done;
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003313}
3314
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003315/*
Bram Moolenaarf3654822016-03-04 22:12:23 +01003316 * Expand wildcards in "pat" and invoke do_source() for each match.
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003317 */
3318 static void
Bram Moolenaarf3654822016-03-04 22:12:23 +01003319source_all_matches(char_u *pat)
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003320{
Bram Moolenaarf3654822016-03-04 22:12:23 +01003321 int num_files;
3322 char_u **files;
3323 int i;
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003324
Bram Moolenaarf3654822016-03-04 22:12:23 +01003325 if (gen_expand_wildcards(1, &pat, &num_files, &files, EW_FILE) == OK)
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003326 {
Bram Moolenaarf3654822016-03-04 22:12:23 +01003327 for (i = 0; i < num_files; ++i)
3328 (void)do_source(files[i], FALSE, DOSO_NONE);
3329 FreeWild(num_files, files);
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003330 }
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003331}
3332
Bram Moolenaar49b27322016-04-05 21:13:00 +02003333/* used for "cookie" of add_pack_plugin() */
3334static int APP_ADD_DIR;
3335static int APP_LOAD;
3336static int APP_BOTH;
3337
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003338 static void
Bram Moolenaar91715872016-03-03 17:13:03 +01003339add_pack_plugin(char_u *fname, void *cookie)
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003340{
Bram Moolenaarf3654822016-03-04 22:12:23 +01003341 char_u *p4, *p3, *p2, *p1, *p;
3342 char_u *insp;
Bram Moolenaar91715872016-03-03 17:13:03 +01003343 int c;
3344 char_u *new_rtp;
3345 int keep;
3346 int oldlen;
3347 int addlen;
Bram Moolenaara5702442016-05-24 19:37:29 +02003348 char_u *afterdir;
3349 int afterlen = 0;
Bram Moolenaar91715872016-03-03 17:13:03 +01003350 char_u *ffname = fix_fname(fname);
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003351
Bram Moolenaar91715872016-03-03 17:13:03 +01003352 if (ffname == NULL)
3353 return;
Bram Moolenaar49b27322016-04-05 21:13:00 +02003354 if (cookie != &APP_LOAD && strstr((char *)p_rtp, (char *)ffname) == NULL)
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003355 {
Bram Moolenaara5702442016-05-24 19:37:29 +02003356 /* directory is not yet in 'runtimepath', add it */
Bram Moolenaarf3654822016-03-04 22:12:23 +01003357 p4 = p3 = p2 = p1 = get_past_head(ffname);
3358 for (p = p1; *p; mb_ptr_adv(p))
3359 if (vim_ispathsep_nocolon(*p))
3360 {
3361 p4 = p3; p3 = p2; p2 = p1; p1 = p;
3362 }
3363
3364 /* now we have:
Bram Moolenaaraf1a0e32016-03-09 22:19:26 +01003365 * rtp/pack/name/start/name
3366 * p4 p3 p2 p1
Bram Moolenaarf3654822016-03-04 22:12:23 +01003367 *
3368 * find the part up to "pack" in 'runtimepath' */
3369 c = *p4;
3370 *p4 = NUL;
3371 insp = (char_u *)strstr((char *)p_rtp, (char *)ffname);
3372 if (insp == NULL)
3373 /* not found, append at the end */
3374 insp = p_rtp + STRLEN(p_rtp);
3375 else
3376 {
3377 /* append after the matching directory. */
3378 insp += STRLEN(ffname);
3379 while (*insp != NUL && *insp != ',')
3380 ++insp;
3381 }
3382 *p4 = c;
3383
Bram Moolenaara5702442016-05-24 19:37:29 +02003384 /* check if rtp/pack/name/start/name/after exists */
3385 afterdir = concat_fnames(ffname, (char_u *)"after", TRUE);
3386 if (afterdir != NULL && mch_isdir(afterdir))
3387 afterlen = STRLEN(afterdir) + 1; /* add one for comma */
3388
Bram Moolenaar1daae442016-02-22 23:25:25 +01003389 oldlen = (int)STRLEN(p_rtp);
Bram Moolenaara5702442016-05-24 19:37:29 +02003390 addlen = (int)STRLEN(ffname) + 1; /* add one for comma */
3391 new_rtp = alloc(oldlen + addlen + afterlen + 1); /* add one for NUL */
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003392 if (new_rtp == NULL)
Bram Moolenaarf3654822016-03-04 22:12:23 +01003393 goto theend;
3394 keep = (int)(insp - p_rtp);
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003395 mch_memmove(new_rtp, p_rtp, keep);
3396 new_rtp[keep] = ',';
Bram Moolenaara5702442016-05-24 19:37:29 +02003397 mch_memmove(new_rtp + keep + 1, ffname, addlen);
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003398 if (p_rtp[keep] != NUL)
Bram Moolenaara5702442016-05-24 19:37:29 +02003399 mch_memmove(new_rtp + keep + addlen, p_rtp + keep,
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003400 oldlen - keep + 1);
Bram Moolenaara5702442016-05-24 19:37:29 +02003401 if (afterlen > 0)
3402 {
3403 STRCAT(new_rtp, ",");
3404 STRCAT(new_rtp, afterdir);
3405 }
Bram Moolenaar863c1a92016-03-03 15:47:06 +01003406 set_option_value((char_u *)"rtp", 0L, new_rtp, 0);
3407 vim_free(new_rtp);
Bram Moolenaara5702442016-05-24 19:37:29 +02003408 vim_free(afterdir);
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003409 }
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003410
Bram Moolenaar49b27322016-04-05 21:13:00 +02003411 if (cookie != &APP_ADD_DIR)
Bram Moolenaarf3654822016-03-04 22:12:23 +01003412 {
Bram Moolenaar71fb0c12016-04-02 22:44:16 +02003413 static char *plugpat = "%s/plugin/**/*.vim";
Bram Moolenaarf3654822016-03-04 22:12:23 +01003414 static char *ftpat = "%s/ftdetect/*.vim";
3415 int len;
3416 char_u *pat;
3417
3418 len = (int)STRLEN(ffname) + (int)STRLEN(ftpat);
3419 pat = alloc(len);
3420 if (pat == NULL)
3421 goto theend;
3422 vim_snprintf((char *)pat, len, plugpat, ffname);
3423 source_all_matches(pat);
3424
3425#ifdef FEAT_AUTOCMD
3426 {
3427 char_u *cmd = vim_strsave((char_u *)"g:did_load_filetypes");
3428
3429 /* If runtime/filetype.vim wasn't loaded yet, the scripts will be
3430 * found when it loads. */
3431 if (cmd != NULL && eval_to_number(cmd) > 0)
3432 {
3433 do_cmdline_cmd((char_u *)"augroup filetypedetect");
3434 vim_snprintf((char *)pat, len, ftpat, ffname);
3435 source_all_matches(pat);
3436 do_cmdline_cmd((char_u *)"augroup END");
3437 }
3438 vim_free(cmd);
3439 }
3440#endif
Bram Moolenaarba8cd122016-03-19 14:16:39 +01003441 vim_free(pat);
Bram Moolenaarf3654822016-03-04 22:12:23 +01003442 }
3443
3444theend:
3445 vim_free(ffname);
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003446}
3447
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003448static int did_source_packages = FALSE;
3449
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003450/*
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003451 * ":packloadall"
Bram Moolenaarf3654822016-03-04 22:12:23 +01003452 * Find plugins in the package directories and source them.
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003453 */
3454 void
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003455ex_packloadall(exarg_T *eap)
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003456{
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003457 if (!did_source_packages || (eap != NULL && eap->forceit))
3458 {
3459 did_source_packages = TRUE;
Bram Moolenaar49b27322016-04-05 21:13:00 +02003460
3461 /* First do a round to add all directories to 'runtimepath', then load
3462 * the plugins. This allows for plugins to use an autoload directory
3463 * of another plugin. */
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003464 do_in_path(p_pp, (char_u *)"pack/*/start/*", DIP_ALL + DIP_DIR,
Bram Moolenaar49b27322016-04-05 21:13:00 +02003465 add_pack_plugin, &APP_ADD_DIR);
3466 do_in_path(p_pp, (char_u *)"pack/*/start/*", DIP_ALL + DIP_DIR,
3467 add_pack_plugin, &APP_LOAD);
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003468 }
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003469}
3470
3471/*
Bram Moolenaarf3654822016-03-04 22:12:23 +01003472 * ":packadd[!] {name}"
Bram Moolenaar91715872016-03-03 17:13:03 +01003473 */
3474 void
3475ex_packadd(exarg_T *eap)
3476{
3477 static char *plugpat = "pack/*/opt/%s";
3478 int len;
3479 char *pat;
3480
3481 len = (int)STRLEN(plugpat) + (int)STRLEN(eap->arg);
3482 pat = (char *)alloc(len);
3483 if (pat == NULL)
3484 return;
3485 vim_snprintf(pat, len, plugpat, eap->arg);
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003486 do_in_path(p_pp, (char_u *)pat, DIP_ALL + DIP_DIR + DIP_ERR,
Bram Moolenaar49b27322016-04-05 21:13:00 +02003487 add_pack_plugin, eap->forceit ? &APP_ADD_DIR : &APP_BOTH);
Bram Moolenaar91715872016-03-03 17:13:03 +01003488 vim_free(pat);
3489}
3490
Bram Moolenaar071d4272004-06-13 20:20:40 +00003491#if defined(FEAT_EVAL) && defined(FEAT_AUTOCMD)
3492/*
3493 * ":options"
3494 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003495 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003496ex_options(
3497 exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003498{
3499 cmd_source((char_u *)SYS_OPTWIN_FILE, NULL);
3500}
3501#endif
3502
3503/*
3504 * ":source {fname}"
3505 */
3506 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003507ex_source(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003508{
3509#ifdef FEAT_BROWSE
3510 if (cmdmod.browse)
3511 {
3512 char_u *fname = NULL;
3513
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00003514 fname = do_browse(0, (char_u *)_("Source Vim script"), eap->arg,
Bram Moolenaar071d4272004-06-13 20:20:40 +00003515 NULL, NULL, BROWSE_FILTER_MACROS, NULL);
3516 if (fname != NULL)
3517 {
3518 cmd_source(fname, eap);
3519 vim_free(fname);
3520 }
3521 }
3522 else
3523#endif
3524 cmd_source(eap->arg, eap);
3525}
3526
3527 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003528cmd_source(char_u *fname, exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003529{
3530 if (*fname == NUL)
3531 EMSG(_(e_argreq));
3532
Bram Moolenaar071d4272004-06-13 20:20:40 +00003533 else if (eap != NULL && eap->forceit)
Bram Moolenaar84a05ac2013-05-06 04:24:17 +02003534 /* ":source!": read Normal mode commands
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00003535 * Need to execute the commands directly. This is required at least
3536 * for:
Bram Moolenaar071d4272004-06-13 20:20:40 +00003537 * - ":g" command busy
3538 * - after ":argdo", ":windo" or ":bufdo"
3539 * - another command follows
3540 * - inside a loop
3541 */
3542 openscript(fname, global_busy || listcmd_busy || eap->nextcmd != NULL
3543#ifdef FEAT_EVAL
3544 || eap->cstack->cs_idx >= 0
3545#endif
3546 );
3547
3548 /* ":source" read ex commands */
Bram Moolenaar910f66f2006-04-05 20:41:53 +00003549 else if (do_source(fname, FALSE, DOSO_NONE) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003550 EMSG2(_(e_notopen), fname);
3551}
3552
3553/*
3554 * ":source" and associated commands.
3555 */
3556/*
3557 * Structure used to store info for each sourced file.
3558 * It is shared between do_source() and getsourceline().
3559 * This is required, because it needs to be handed to do_cmdline() and
3560 * sourcing can be done recursively.
3561 */
3562struct source_cookie
3563{
3564 FILE *fp; /* opened file for sourcing */
3565 char_u *nextline; /* if not NULL: line that was read ahead */
3566 int finished; /* ":finish" used */
Bram Moolenaar860cae12010-06-05 23:22:07 +02003567#if defined(USE_CRNL) || defined(USE_CR)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003568 int fileformat; /* EOL_UNKNOWN, EOL_UNIX or EOL_DOS */
3569 int error; /* TRUE if LF found after CR-LF */
3570#endif
3571#ifdef FEAT_EVAL
3572 linenr_T breakpoint; /* next line with breakpoint or zero */
3573 char_u *fname; /* name of sourced file */
3574 int dbg_tick; /* debug_tick when breakpoint was set */
3575 int level; /* top nesting level of sourced file */
3576#endif
3577#ifdef FEAT_MBYTE
3578 vimconv_T conv; /* type of conversion */
3579#endif
3580};
3581
3582#ifdef FEAT_EVAL
3583/*
3584 * Return the address holding the next breakpoint line for a source cookie.
3585 */
3586 linenr_T *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003587source_breakpoint(void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003588{
3589 return &((struct source_cookie *)cookie)->breakpoint;
3590}
3591
3592/*
3593 * Return the address holding the debug tick for a source cookie.
3594 */
3595 int *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003596source_dbg_tick(void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003597{
3598 return &((struct source_cookie *)cookie)->dbg_tick;
3599}
3600
3601/*
3602 * Return the nesting level for a source cookie.
3603 */
3604 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003605source_level(void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003606{
3607 return ((struct source_cookie *)cookie)->level;
3608}
3609#endif
3610
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01003611static char_u *get_one_sourceline(struct source_cookie *sp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003612
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01003613#if (defined(WIN32) && defined(FEAT_CSCOPE)) || defined(HAVE_FD_CLOEXEC)
3614# define USE_FOPEN_NOINH
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01003615static FILE *fopen_noinh_readbin(char *filename);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003616
3617/*
3618 * Special function to open a file without handle inheritance.
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01003619 * When possible the handle is closed on exec().
Bram Moolenaar071d4272004-06-13 20:20:40 +00003620 */
3621 static FILE *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003622fopen_noinh_readbin(char *filename)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003623{
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01003624# ifdef WIN32
Bram Moolenaar8d8ef0b2010-01-20 21:41:47 +01003625 int fd_tmp = mch_open(filename, O_RDONLY | O_BINARY | O_NOINHERIT, 0);
3626# else
3627 int fd_tmp = mch_open(filename, O_RDONLY, 0);
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01003628# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003629
3630 if (fd_tmp == -1)
3631 return NULL;
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01003632
3633# ifdef HAVE_FD_CLOEXEC
3634 {
3635 int fdflags = fcntl(fd_tmp, F_GETFD);
3636 if (fdflags >= 0 && (fdflags & FD_CLOEXEC) == 0)
Bram Moolenaarcde88542015-08-11 19:14:00 +02003637 (void)fcntl(fd_tmp, F_SETFD, fdflags | FD_CLOEXEC);
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01003638 }
3639# endif
3640
Bram Moolenaar071d4272004-06-13 20:20:40 +00003641 return fdopen(fd_tmp, READBIN);
3642}
3643#endif
3644
3645
3646/*
3647 * do_source: Read the file "fname" and execute its lines as EX commands.
3648 *
3649 * This function may be called recursively!
3650 *
3651 * return FAIL if file could not be opened, OK otherwise
3652 */
3653 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003654do_source(
3655 char_u *fname,
3656 int check_other, /* check for .vimrc and _vimrc */
3657 int is_vimrc) /* DOSO_ value */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003658{
3659 struct source_cookie cookie;
3660 char_u *save_sourcing_name;
3661 linenr_T save_sourcing_lnum;
3662 char_u *p;
3663 char_u *fname_exp;
Bram Moolenaar73881402009-02-04 16:50:47 +00003664 char_u *firstline = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003665 int retval = FAIL;
3666#ifdef FEAT_EVAL
3667 scid_T save_current_SID;
3668 static scid_T last_current_SID = 0;
3669 void *save_funccalp;
3670 int save_debug_break_level = debug_break_level;
Bram Moolenaar05159a02005-02-26 23:04:13 +00003671 scriptitem_T *si = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003672# ifdef UNIX
3673 struct stat st;
3674 int stat_ok;
3675# endif
3676#endif
3677#ifdef STARTUPTIME
3678 struct timeval tv_rel;
3679 struct timeval tv_start;
3680#endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00003681#ifdef FEAT_PROFILE
3682 proftime_T wait_start;
3683#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003684
Bram Moolenaar071d4272004-06-13 20:20:40 +00003685 p = expand_env_save(fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003686 if (p == NULL)
3687 return retval;
3688 fname_exp = fix_fname(p);
3689 vim_free(p);
3690 if (fname_exp == NULL)
3691 return retval;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003692 if (mch_isdir(fname_exp))
3693 {
Bram Moolenaar555b2802005-05-19 21:08:39 +00003694 smsg((char_u *)_("Cannot source a directory: \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003695 goto theend;
3696 }
3697
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00003698#ifdef FEAT_AUTOCMD
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00003699 /* Apply SourceCmd autocommands, they should get the file and source it. */
3700 if (has_autocmd(EVENT_SOURCECMD, fname_exp, NULL)
3701 && apply_autocmds(EVENT_SOURCECMD, fname_exp, fname_exp,
3702 FALSE, curbuf))
Bram Moolenaarf33943e2008-01-15 21:17:30 +00003703 {
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00003704# ifdef FEAT_EVAL
Bram Moolenaarf33943e2008-01-15 21:17:30 +00003705 retval = aborting() ? FAIL : OK;
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00003706# else
Bram Moolenaarf33943e2008-01-15 21:17:30 +00003707 retval = OK;
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00003708# endif
Bram Moolenaarf33943e2008-01-15 21:17:30 +00003709 goto theend;
3710 }
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00003711
3712 /* Apply SourcePre autocommands, they may get the file. */
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00003713 apply_autocmds(EVENT_SOURCEPRE, fname_exp, fname_exp, FALSE, curbuf);
3714#endif
3715
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01003716#ifdef USE_FOPEN_NOINH
Bram Moolenaar071d4272004-06-13 20:20:40 +00003717 cookie.fp = fopen_noinh_readbin((char *)fname_exp);
3718#else
3719 cookie.fp = mch_fopen((char *)fname_exp, READBIN);
3720#endif
3721 if (cookie.fp == NULL && check_other)
3722 {
3723 /*
3724 * Try again, replacing file name ".vimrc" by "_vimrc" or vice versa,
3725 * and ".exrc" by "_exrc" or vice versa.
3726 */
3727 p = gettail(fname_exp);
3728 if ((*p == '.' || *p == '_')
3729 && (STRICMP(p + 1, "vimrc") == 0
3730 || STRICMP(p + 1, "gvimrc") == 0
3731 || STRICMP(p + 1, "exrc") == 0))
3732 {
3733 if (*p == '_')
3734 *p = '.';
3735 else
3736 *p = '_';
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01003737#ifdef USE_FOPEN_NOINH
Bram Moolenaar071d4272004-06-13 20:20:40 +00003738 cookie.fp = fopen_noinh_readbin((char *)fname_exp);
3739#else
3740 cookie.fp = mch_fopen((char *)fname_exp, READBIN);
3741#endif
3742 }
3743 }
3744
3745 if (cookie.fp == NULL)
3746 {
3747 if (p_verbose > 0)
3748 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003749 verbose_enter();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003750 if (sourcing_name == NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00003751 smsg((char_u *)_("could not source \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003752 else
3753 smsg((char_u *)_("line %ld: could not source \"%s\""),
Bram Moolenaar555b2802005-05-19 21:08:39 +00003754 sourcing_lnum, fname);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003755 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003756 }
3757 goto theend;
3758 }
3759
3760 /*
3761 * The file exists.
3762 * - In verbose mode, give a message.
3763 * - For a vimrc file, may want to set 'compatible', call vimrc_found().
3764 */
3765 if (p_verbose > 1)
3766 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003767 verbose_enter();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003768 if (sourcing_name == NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00003769 smsg((char_u *)_("sourcing \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003770 else
3771 smsg((char_u *)_("line %ld: sourcing \"%s\""),
Bram Moolenaar555b2802005-05-19 21:08:39 +00003772 sourcing_lnum, fname);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003773 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003774 }
Bram Moolenaar910f66f2006-04-05 20:41:53 +00003775 if (is_vimrc == DOSO_VIMRC)
3776 vimrc_found(fname_exp, (char_u *)"MYVIMRC");
3777 else if (is_vimrc == DOSO_GVIMRC)
3778 vimrc_found(fname_exp, (char_u *)"MYGVIMRC");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003779
3780#ifdef USE_CRNL
3781 /* If no automatic file format: Set default to CR-NL. */
3782 if (*p_ffs == NUL)
3783 cookie.fileformat = EOL_DOS;
3784 else
3785 cookie.fileformat = EOL_UNKNOWN;
3786 cookie.error = FALSE;
3787#endif
3788
3789#ifdef USE_CR
3790 /* If no automatic file format: Set default to CR. */
3791 if (*p_ffs == NUL)
3792 cookie.fileformat = EOL_MAC;
3793 else
3794 cookie.fileformat = EOL_UNKNOWN;
3795 cookie.error = FALSE;
3796#endif
3797
3798 cookie.nextline = NULL;
3799 cookie.finished = FALSE;
3800
3801#ifdef FEAT_EVAL
3802 /*
3803 * Check if this script has a breakpoint.
3804 */
3805 cookie.breakpoint = dbg_find_breakpoint(TRUE, fname_exp, (linenr_T)0);
3806 cookie.fname = fname_exp;
3807 cookie.dbg_tick = debug_tick;
3808
3809 cookie.level = ex_nesting_level;
3810#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003811
3812 /*
3813 * Keep the sourcing name/lnum, for recursive calls.
3814 */
3815 save_sourcing_name = sourcing_name;
3816 sourcing_name = fname_exp;
3817 save_sourcing_lnum = sourcing_lnum;
3818 sourcing_lnum = 0;
3819
Bram Moolenaar73881402009-02-04 16:50:47 +00003820#ifdef FEAT_MBYTE
3821 cookie.conv.vc_type = CONV_NONE; /* no conversion */
3822
3823 /* Read the first line so we can check for a UTF-8 BOM. */
3824 firstline = getsourceline(0, (void *)&cookie, 0);
3825 if (firstline != NULL && STRLEN(firstline) >= 3 && firstline[0] == 0xef
3826 && firstline[1] == 0xbb && firstline[2] == 0xbf)
3827 {
3828 /* Found BOM; setup conversion, skip over BOM and recode the line. */
3829 convert_setup(&cookie.conv, (char_u *)"utf-8", p_enc);
3830 p = string_convert(&cookie.conv, firstline + 3, NULL);
Bram Moolenaar4e2a5952009-02-05 19:48:25 +00003831 if (p == NULL)
3832 p = vim_strsave(firstline + 3);
Bram Moolenaar73881402009-02-04 16:50:47 +00003833 if (p != NULL)
3834 {
3835 vim_free(firstline);
3836 firstline = p;
3837 }
3838 }
3839#endif
3840
Bram Moolenaar071d4272004-06-13 20:20:40 +00003841#ifdef STARTUPTIME
Bram Moolenaarc4e41982010-01-19 16:31:47 +01003842 if (time_fd != NULL)
3843 time_push(&tv_rel, &tv_start);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003844#endif
3845
3846#ifdef FEAT_EVAL
Bram Moolenaar05159a02005-02-26 23:04:13 +00003847# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00003848 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003849 prof_child_enter(&wait_start); /* entering a child now */
3850# endif
3851
3852 /* Don't use local function variables, if called from a function.
3853 * Also starts profiling timer for nested script. */
3854 save_funccalp = save_funccal();
3855
Bram Moolenaar071d4272004-06-13 20:20:40 +00003856 /*
3857 * Check if this script was sourced before to finds its SID.
3858 * If it's new, generate a new SID.
3859 */
3860 save_current_SID = current_SID;
3861# ifdef UNIX
3862 stat_ok = (mch_stat((char *)fname_exp, &st) >= 0);
3863# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00003864 for (current_SID = script_items.ga_len; current_SID > 0; --current_SID)
3865 {
3866 si = &SCRIPT_ITEM(current_SID);
3867 if (si->sn_name != NULL
Bram Moolenaar071d4272004-06-13 20:20:40 +00003868 && (
3869# ifdef UNIX
Bram Moolenaar7c626922005-02-07 22:01:03 +00003870 /* Compare dev/ino when possible, it catches symbolic
3871 * links. Also compare file names, the inode may change
3872 * when the file was edited. */
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00003873 ((stat_ok && si->sn_dev_valid)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003874 && (si->sn_dev == st.st_dev
3875 && si->sn_ino == st.st_ino)) ||
Bram Moolenaar071d4272004-06-13 20:20:40 +00003876# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00003877 fnamecmp(si->sn_name, fname_exp) == 0))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003878 break;
Bram Moolenaar05159a02005-02-26 23:04:13 +00003879 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003880 if (current_SID == 0)
3881 {
3882 current_SID = ++last_current_SID;
Bram Moolenaar05159a02005-02-26 23:04:13 +00003883 if (ga_grow(&script_items, (int)(current_SID - script_items.ga_len))
3884 == FAIL)
3885 goto almosttheend;
3886 while (script_items.ga_len < current_SID)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003887 {
Bram Moolenaar05159a02005-02-26 23:04:13 +00003888 ++script_items.ga_len;
3889 SCRIPT_ITEM(script_items.ga_len).sn_name = NULL;
3890# ifdef FEAT_PROFILE
3891 SCRIPT_ITEM(script_items.ga_len).sn_prof_on = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003892# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003893 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00003894 si = &SCRIPT_ITEM(current_SID);
3895 si->sn_name = fname_exp;
3896 fname_exp = NULL;
3897# ifdef UNIX
3898 if (stat_ok)
3899 {
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00003900 si->sn_dev_valid = TRUE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00003901 si->sn_dev = st.st_dev;
3902 si->sn_ino = st.st_ino;
3903 }
3904 else
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00003905 si->sn_dev_valid = FALSE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00003906# endif
3907
Bram Moolenaar071d4272004-06-13 20:20:40 +00003908 /* Allocate the local script variables to use for this script. */
3909 new_script_vars(current_SID);
3910 }
3911
Bram Moolenaar05159a02005-02-26 23:04:13 +00003912# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00003913 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003914 {
3915 int forceit;
3916
3917 /* Check if we do profiling for this script. */
3918 if (!si->sn_prof_on && has_profiling(TRUE, si->sn_name, &forceit))
3919 {
3920 script_do_profile(si);
3921 si->sn_pr_force = forceit;
3922 }
3923 if (si->sn_prof_on)
3924 {
3925 ++si->sn_pr_count;
3926 profile_start(&si->sn_pr_start);
3927 profile_zero(&si->sn_pr_children);
3928 }
3929 }
3930# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003931#endif
3932
3933 /*
3934 * Call do_cmdline, which will call getsourceline() to get the lines.
3935 */
Bram Moolenaar73881402009-02-04 16:50:47 +00003936 do_cmdline(firstline, getsourceline, (void *)&cookie,
Bram Moolenaar071d4272004-06-13 20:20:40 +00003937 DOCMD_VERBOSE|DOCMD_NOWAIT|DOCMD_REPEAT);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003938 retval = OK;
Bram Moolenaar05159a02005-02-26 23:04:13 +00003939
3940#ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00003941 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003942 {
3943 /* Get "si" again, "script_items" may have been reallocated. */
3944 si = &SCRIPT_ITEM(current_SID);
3945 if (si->sn_prof_on)
3946 {
3947 profile_end(&si->sn_pr_start);
3948 profile_sub_wait(&wait_start, &si->sn_pr_start);
3949 profile_add(&si->sn_pr_total, &si->sn_pr_start);
Bram Moolenaar1056d982006-03-09 22:37:52 +00003950 profile_self(&si->sn_pr_self, &si->sn_pr_start,
3951 &si->sn_pr_children);
Bram Moolenaar05159a02005-02-26 23:04:13 +00003952 }
3953 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003954#endif
3955
3956 if (got_int)
3957 EMSG(_(e_interr));
3958 sourcing_name = save_sourcing_name;
3959 sourcing_lnum = save_sourcing_lnum;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003960 if (p_verbose > 1)
3961 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003962 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00003963 smsg((char_u *)_("finished sourcing %s"), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003964 if (sourcing_name != NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00003965 smsg((char_u *)_("continuing in %s"), sourcing_name);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003966 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003967 }
3968#ifdef STARTUPTIME
Bram Moolenaarc4e41982010-01-19 16:31:47 +01003969 if (time_fd != NULL)
3970 {
3971 vim_snprintf((char *)IObuff, IOSIZE, "sourcing %s", fname);
3972 time_msg((char *)IObuff, &tv_start);
3973 time_pop(&tv_rel);
3974 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003975#endif
3976
3977#ifdef FEAT_EVAL
3978 /*
3979 * After a "finish" in debug mode, need to break at first command of next
3980 * sourced file.
3981 */
3982 if (save_debug_break_level > ex_nesting_level
3983 && debug_break_level == ex_nesting_level)
3984 ++debug_break_level;
3985#endif
3986
Bram Moolenaar05159a02005-02-26 23:04:13 +00003987#ifdef FEAT_EVAL
3988almosttheend:
3989 current_SID = save_current_SID;
3990 restore_funccal(save_funccalp);
3991# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00003992 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003993 prof_child_exit(&wait_start); /* leaving a child now */
3994# endif
3995#endif
3996 fclose(cookie.fp);
3997 vim_free(cookie.nextline);
Bram Moolenaar73881402009-02-04 16:50:47 +00003998 vim_free(firstline);
Bram Moolenaar05159a02005-02-26 23:04:13 +00003999#ifdef FEAT_MBYTE
4000 convert_setup(&cookie.conv, NULL, NULL);
4001#endif
4002
Bram Moolenaar071d4272004-06-13 20:20:40 +00004003theend:
4004 vim_free(fname_exp);
4005 return retval;
4006}
4007
4008#if defined(FEAT_EVAL) || defined(PROTO)
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00004009
Bram Moolenaar071d4272004-06-13 20:20:40 +00004010/*
4011 * ":scriptnames"
4012 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004013 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004014ex_scriptnames(exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004015{
4016 int i;
4017
Bram Moolenaar05159a02005-02-26 23:04:13 +00004018 for (i = 1; i <= script_items.ga_len && !got_int; ++i)
4019 if (SCRIPT_ITEM(i).sn_name != NULL)
Bram Moolenaard58ea072011-06-26 04:25:30 +02004020 {
4021 home_replace(NULL, SCRIPT_ITEM(i).sn_name,
4022 NameBuff, MAXPATHL, TRUE);
4023 smsg((char_u *)"%3d: %s", i, NameBuff);
Bram Moolenaar970a1b82012-03-23 18:39:18 +01004024 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004025}
4026
4027# if defined(BACKSLASH_IN_FILENAME) || defined(PROTO)
4028/*
4029 * Fix slashes in the list of script names for 'shellslash'.
4030 */
4031 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004032scriptnames_slash_adjust(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004033{
4034 int i;
4035
Bram Moolenaar05159a02005-02-26 23:04:13 +00004036 for (i = 1; i <= script_items.ga_len; ++i)
4037 if (SCRIPT_ITEM(i).sn_name != NULL)
4038 slash_adjust(SCRIPT_ITEM(i).sn_name);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004039}
4040# endif
4041
4042/*
4043 * Get a pointer to a script name. Used for ":verbose set".
4044 */
4045 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004046get_scriptname(scid_T id)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004047{
4048 if (id == SID_MODELINE)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00004049 return (char_u *)_("modeline");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004050 if (id == SID_CMDARG)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00004051 return (char_u *)_("--cmd argument");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004052 if (id == SID_CARG)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00004053 return (char_u *)_("-c argument");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004054 if (id == SID_ENV)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00004055 return (char_u *)_("environment variable");
4056 if (id == SID_ERROR)
4057 return (char_u *)_("error handler");
Bram Moolenaar05159a02005-02-26 23:04:13 +00004058 return SCRIPT_ITEM(id).sn_name;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004059}
Bram Moolenaar05159a02005-02-26 23:04:13 +00004060
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00004061# if defined(EXITFREE) || defined(PROTO)
4062 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004063free_scriptnames(void)
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00004064{
4065 int i;
4066
4067 for (i = script_items.ga_len; i > 0; --i)
4068 vim_free(SCRIPT_ITEM(i).sn_name);
4069 ga_clear(&script_items);
4070}
4071# endif
4072
Bram Moolenaar071d4272004-06-13 20:20:40 +00004073#endif
4074
4075#if defined(USE_CR) || defined(PROTO)
4076
4077# if defined(__MSL__) && (__MSL__ >= 22)
4078/*
4079 * Newer version of the Metrowerks library handle DOS and UNIX files
4080 * without help.
4081 * Test with earlier versions, MSL 2.2 is the library supplied with
4082 * Codewarrior Pro 2.
4083 */
4084 char *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004085fgets_cr(char *s, int n, FILE *stream)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004086{
4087 return fgets(s, n, stream);
4088}
4089# else
4090/*
4091 * Version of fgets() which also works for lines ending in a <CR> only
4092 * (Macintosh format).
4093 * For older versions of the Metrowerks library.
4094 * At least CodeWarrior 9 needed this code.
4095 */
4096 char *
Bram Moolenaard14e00e2016-01-31 17:30:51 +01004097fgets_cr(char *s, int n, FILE *stream)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004098{
4099 int c = 0;
4100 int char_read = 0;
4101
4102 while (!feof(stream) && c != '\r' && c != '\n' && char_read < n - 1)
4103 {
4104 c = fgetc(stream);
4105 s[char_read++] = c;
4106 /* If the file is in DOS format, we need to skip a NL after a CR. I
4107 * thought it was the other way around, but this appears to work... */
4108 if (c == '\n')
4109 {
4110 c = fgetc(stream);
4111 if (c != '\r')
4112 ungetc(c, stream);
4113 }
4114 }
4115
4116 s[char_read] = 0;
4117 if (char_read == 0)
4118 return NULL;
4119
4120 if (feof(stream) && char_read == 1)
4121 return NULL;
4122
4123 return s;
4124}
4125# endif
4126#endif
4127
4128/*
4129 * Get one full line from a sourced file.
4130 * Called by do_cmdline() when it's called from do_source().
4131 *
4132 * Return a pointer to the line in allocated memory.
4133 * Return NULL for end-of-file or some error.
4134 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004135 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004136getsourceline(int c UNUSED, void *cookie, int indent UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004137{
4138 struct source_cookie *sp = (struct source_cookie *)cookie;
4139 char_u *line;
Bram Moolenaarc047b9a2012-02-11 20:40:55 +01004140 char_u *p;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004141
4142#ifdef FEAT_EVAL
4143 /* If breakpoints have been added/deleted need to check for it. */
4144 if (sp->dbg_tick < debug_tick)
4145 {
4146 sp->breakpoint = dbg_find_breakpoint(TRUE, sp->fname, sourcing_lnum);
4147 sp->dbg_tick = debug_tick;
4148 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00004149# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004150 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004151 script_line_end();
4152# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004153#endif
4154 /*
4155 * Get current line. If there is a read-ahead line, use it, otherwise get
4156 * one now.
4157 */
4158 if (sp->finished)
4159 line = NULL;
4160 else if (sp->nextline == NULL)
4161 line = get_one_sourceline(sp);
4162 else
4163 {
4164 line = sp->nextline;
4165 sp->nextline = NULL;
4166 ++sourcing_lnum;
4167 }
Bram Moolenaare2cc9702005-03-15 22:43:58 +00004168#ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00004169 if (line != NULL && do_profiling == PROF_YES)
Bram Moolenaare2cc9702005-03-15 22:43:58 +00004170 script_line_start();
4171#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004172
4173 /* Only concatenate lines starting with a \ when 'cpoptions' doesn't
4174 * contain the 'C' flag. */
4175 if (line != NULL && (vim_strchr(p_cpo, CPO_CONCAT) == NULL))
4176 {
4177 /* compensate for the one line read-ahead */
4178 --sourcing_lnum;
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004179
4180 /* Get the next line and concatenate it when it starts with a
4181 * backslash. We always need to read the next line, keep it in
4182 * sp->nextline. */
4183 sp->nextline = get_one_sourceline(sp);
4184 if (sp->nextline != NULL && *(p = skipwhite(sp->nextline)) == '\\')
Bram Moolenaar071d4272004-06-13 20:20:40 +00004185 {
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004186 garray_T ga;
4187
Bram Moolenaarb549a732012-02-22 18:29:33 +01004188 ga_init2(&ga, (int)sizeof(char_u), 400);
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004189 ga_concat(&ga, line);
4190 ga_concat(&ga, p + 1);
4191 for (;;)
4192 {
4193 vim_free(sp->nextline);
4194 sp->nextline = get_one_sourceline(sp);
4195 if (sp->nextline == NULL)
4196 break;
4197 p = skipwhite(sp->nextline);
4198 if (*p != '\\')
4199 break;
Bram Moolenaarb549a732012-02-22 18:29:33 +01004200 /* Adjust the growsize to the current length to speed up
4201 * concatenating many lines. */
4202 if (ga.ga_len > 400)
4203 {
4204 if (ga.ga_len > 8000)
4205 ga.ga_growsize = 8000;
4206 else
4207 ga.ga_growsize = ga.ga_len;
4208 }
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004209 ga_concat(&ga, p + 1);
4210 }
4211 ga_append(&ga, NUL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004212 vim_free(line);
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004213 line = ga.ga_data;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004214 }
4215 }
4216
4217#ifdef FEAT_MBYTE
4218 if (line != NULL && sp->conv.vc_type != CONV_NONE)
4219 {
Bram Moolenaarc047b9a2012-02-11 20:40:55 +01004220 char_u *s;
4221
Bram Moolenaar071d4272004-06-13 20:20:40 +00004222 /* Convert the encoding of the script line. */
4223 s = string_convert(&sp->conv, line, NULL);
4224 if (s != NULL)
4225 {
4226 vim_free(line);
4227 line = s;
4228 }
4229 }
4230#endif
4231
4232#ifdef FEAT_EVAL
4233 /* Did we encounter a breakpoint? */
4234 if (sp->breakpoint != 0 && sp->breakpoint <= sourcing_lnum)
4235 {
4236 dbg_breakpoint(sp->fname, sourcing_lnum);
4237 /* Find next breakpoint. */
4238 sp->breakpoint = dbg_find_breakpoint(TRUE, sp->fname, sourcing_lnum);
4239 sp->dbg_tick = debug_tick;
4240 }
4241#endif
4242
4243 return line;
4244}
4245
4246 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004247get_one_sourceline(struct source_cookie *sp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004248{
4249 garray_T ga;
4250 int len;
4251 int c;
4252 char_u *buf;
4253#ifdef USE_CRNL
4254 int has_cr; /* CR-LF found */
4255#endif
4256#ifdef USE_CR
4257 char_u *scan;
4258#endif
4259 int have_read = FALSE;
4260
4261 /* use a growarray to store the sourced line */
Bram Moolenaar6ac54292005-02-02 23:07:25 +00004262 ga_init2(&ga, 1, 250);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004263
4264 /*
4265 * Loop until there is a finished line (or end-of-file).
4266 */
4267 sourcing_lnum++;
4268 for (;;)
4269 {
Bram Moolenaar6ac54292005-02-02 23:07:25 +00004270 /* make room to read at least 120 (more) characters */
4271 if (ga_grow(&ga, 120) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004272 break;
4273 buf = (char_u *)ga.ga_data;
4274
4275#ifdef USE_CR
4276 if (sp->fileformat == EOL_MAC)
4277 {
Bram Moolenaar86b68352004-12-27 21:59:20 +00004278 if (fgets_cr((char *)buf + ga.ga_len, ga.ga_maxlen - ga.ga_len,
4279 sp->fp) == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004280 break;
4281 }
4282 else
4283#endif
Bram Moolenaar86b68352004-12-27 21:59:20 +00004284 if (fgets((char *)buf + ga.ga_len, ga.ga_maxlen - ga.ga_len,
4285 sp->fp) == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004286 break;
Bram Moolenaar6ac54292005-02-02 23:07:25 +00004287 len = ga.ga_len + (int)STRLEN(buf + ga.ga_len);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004288#ifdef USE_CRNL
4289 /* Ignore a trailing CTRL-Z, when in Dos mode. Only recognize the
4290 * CTRL-Z by its own, or after a NL. */
4291 if ( (len == 1 || (len >= 2 && buf[len - 2] == '\n'))
4292 && sp->fileformat == EOL_DOS
4293 && buf[len - 1] == Ctrl_Z)
4294 {
4295 buf[len - 1] = NUL;
4296 break;
4297 }
4298#endif
4299
4300#ifdef USE_CR
4301 /* If the read doesn't stop on a new line, and there's
4302 * some CR then we assume a Mac format */
4303 if (sp->fileformat == EOL_UNKNOWN)
4304 {
4305 if (buf[len - 1] != '\n' && vim_strchr(buf, '\r') != NULL)
4306 sp->fileformat = EOL_MAC;
4307 else
4308 sp->fileformat = EOL_UNIX;
4309 }
4310
4311 if (sp->fileformat == EOL_MAC)
4312 {
4313 scan = vim_strchr(buf, '\r');
4314
4315 if (scan != NULL)
4316 {
4317 *scan = '\n';
4318 if (*(scan + 1) != 0)
4319 {
4320 *(scan + 1) = 0;
4321 fseek(sp->fp, (long)(scan - buf - len + 1), SEEK_CUR);
4322 }
4323 }
4324 len = STRLEN(buf);
4325 }
4326#endif
4327
4328 have_read = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004329 ga.ga_len = len;
4330
4331 /* If the line was longer than the buffer, read more. */
Bram Moolenaar86b68352004-12-27 21:59:20 +00004332 if (ga.ga_maxlen - ga.ga_len == 1 && buf[len - 1] != '\n')
Bram Moolenaar071d4272004-06-13 20:20:40 +00004333 continue;
4334
4335 if (len >= 1 && buf[len - 1] == '\n') /* remove trailing NL */
4336 {
4337#ifdef USE_CRNL
4338 has_cr = (len >= 2 && buf[len - 2] == '\r');
4339 if (sp->fileformat == EOL_UNKNOWN)
4340 {
4341 if (has_cr)
4342 sp->fileformat = EOL_DOS;
4343 else
4344 sp->fileformat = EOL_UNIX;
4345 }
4346
4347 if (sp->fileformat == EOL_DOS)
4348 {
4349 if (has_cr) /* replace trailing CR */
4350 {
4351 buf[len - 2] = '\n';
4352 --len;
4353 --ga.ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004354 }
4355 else /* lines like ":map xx yy^M" will have failed */
4356 {
4357 if (!sp->error)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00004358 {
4359 msg_source(hl_attr(HLF_W));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004360 EMSG(_("W15: Warning: Wrong line separator, ^M may be missing"));
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00004361 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004362 sp->error = TRUE;
4363 sp->fileformat = EOL_UNIX;
4364 }
4365 }
4366#endif
4367 /* The '\n' is escaped if there is an odd number of ^V's just
4368 * before it, first set "c" just before the 'V's and then check
4369 * len&c parities (is faster than ((len-c)%2 == 0)) -- Acevedo */
4370 for (c = len - 2; c >= 0 && buf[c] == Ctrl_V; c--)
4371 ;
4372 if ((len & 1) != (c & 1)) /* escaped NL, read more */
4373 {
4374 sourcing_lnum++;
4375 continue;
4376 }
4377
4378 buf[len - 1] = NUL; /* remove the NL */
4379 }
4380
4381 /*
4382 * Check for ^C here now and then, so recursive :so can be broken.
4383 */
4384 line_breakcheck();
4385 break;
4386 }
4387
4388 if (have_read)
4389 return (char_u *)ga.ga_data;
4390
4391 vim_free(ga.ga_data);
4392 return NULL;
4393}
4394
Bram Moolenaar05159a02005-02-26 23:04:13 +00004395#if defined(FEAT_PROFILE) || defined(PROTO)
4396/*
4397 * Called when starting to read a script line.
4398 * "sourcing_lnum" must be correct!
4399 * When skipping lines it may not actually be executed, but we won't find out
4400 * until later and we need to store the time now.
4401 */
4402 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004403script_line_start(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004404{
4405 scriptitem_T *si;
4406 sn_prl_T *pp;
4407
4408 if (current_SID <= 0 || current_SID > script_items.ga_len)
4409 return;
4410 si = &SCRIPT_ITEM(current_SID);
4411 if (si->sn_prof_on && sourcing_lnum >= 1)
4412 {
Bram Moolenaar8c8de832008-06-24 22:58:06 +00004413 /* Grow the array before starting the timer, so that the time spent
Bram Moolenaar05159a02005-02-26 23:04:13 +00004414 * here isn't counted. */
Bram Moolenaarcde88542015-08-11 19:14:00 +02004415 (void)ga_grow(&si->sn_prl_ga, (int)(sourcing_lnum - si->sn_prl_ga.ga_len));
Bram Moolenaar05159a02005-02-26 23:04:13 +00004416 si->sn_prl_idx = sourcing_lnum - 1;
4417 while (si->sn_prl_ga.ga_len <= si->sn_prl_idx
4418 && si->sn_prl_ga.ga_len < si->sn_prl_ga.ga_maxlen)
4419 {
4420 /* Zero counters for a line that was not used before. */
4421 pp = &PRL_ITEM(si, si->sn_prl_ga.ga_len);
4422 pp->snp_count = 0;
4423 profile_zero(&pp->sn_prl_total);
4424 profile_zero(&pp->sn_prl_self);
4425 ++si->sn_prl_ga.ga_len;
4426 }
4427 si->sn_prl_execed = FALSE;
4428 profile_start(&si->sn_prl_start);
4429 profile_zero(&si->sn_prl_children);
4430 profile_get_wait(&si->sn_prl_wait);
4431 }
4432}
4433
4434/*
4435 * Called when actually executing a function line.
4436 */
4437 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004438script_line_exec(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004439{
4440 scriptitem_T *si;
4441
4442 if (current_SID <= 0 || current_SID > script_items.ga_len)
4443 return;
4444 si = &SCRIPT_ITEM(current_SID);
4445 if (si->sn_prof_on && si->sn_prl_idx >= 0)
4446 si->sn_prl_execed = TRUE;
4447}
4448
4449/*
4450 * Called when done with a function line.
4451 */
4452 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004453script_line_end(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004454{
4455 scriptitem_T *si;
4456 sn_prl_T *pp;
4457
4458 if (current_SID <= 0 || current_SID > script_items.ga_len)
4459 return;
4460 si = &SCRIPT_ITEM(current_SID);
4461 if (si->sn_prof_on && si->sn_prl_idx >= 0
4462 && si->sn_prl_idx < si->sn_prl_ga.ga_len)
4463 {
4464 if (si->sn_prl_execed)
4465 {
4466 pp = &PRL_ITEM(si, si->sn_prl_idx);
4467 ++pp->snp_count;
4468 profile_end(&si->sn_prl_start);
4469 profile_sub_wait(&si->sn_prl_wait, &si->sn_prl_start);
Bram Moolenaar05159a02005-02-26 23:04:13 +00004470 profile_add(&pp->sn_prl_total, &si->sn_prl_start);
Bram Moolenaar1056d982006-03-09 22:37:52 +00004471 profile_self(&pp->sn_prl_self, &si->sn_prl_start,
4472 &si->sn_prl_children);
Bram Moolenaar05159a02005-02-26 23:04:13 +00004473 }
4474 si->sn_prl_idx = -1;
4475 }
4476}
4477#endif
4478
Bram Moolenaar071d4272004-06-13 20:20:40 +00004479/*
4480 * ":scriptencoding": Set encoding conversion for a sourced script.
4481 * Without the multi-byte feature it's simply ignored.
4482 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004483 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004484ex_scriptencoding(exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004485{
4486#ifdef FEAT_MBYTE
4487 struct source_cookie *sp;
4488 char_u *name;
4489
4490 if (!getline_equal(eap->getline, eap->cookie, getsourceline))
4491 {
4492 EMSG(_("E167: :scriptencoding used outside of a sourced file"));
4493 return;
4494 }
4495
4496 if (*eap->arg != NUL)
4497 {
4498 name = enc_canonize(eap->arg);
4499 if (name == NULL) /* out of memory */
4500 return;
4501 }
4502 else
4503 name = eap->arg;
4504
4505 /* Setup for conversion from the specified encoding to 'encoding'. */
4506 sp = (struct source_cookie *)getline_cookie(eap->getline, eap->cookie);
4507 convert_setup(&sp->conv, name, p_enc);
4508
4509 if (name != eap->arg)
4510 vim_free(name);
4511#endif
4512}
4513
4514#if defined(FEAT_EVAL) || defined(PROTO)
4515/*
4516 * ":finish": Mark a sourced file as finished.
4517 */
4518 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004519ex_finish(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004520{
4521 if (getline_equal(eap->getline, eap->cookie, getsourceline))
4522 do_finish(eap, FALSE);
4523 else
4524 EMSG(_("E168: :finish used outside of a sourced file"));
4525}
4526
4527/*
4528 * Mark a sourced file as finished. Possibly makes the ":finish" pending.
4529 * Also called for a pending finish at the ":endtry" or after returning from
4530 * an extra do_cmdline(). "reanimate" is used in the latter case.
4531 */
4532 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004533do_finish(exarg_T *eap, int reanimate)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004534{
4535 int idx;
4536
4537 if (reanimate)
4538 ((struct source_cookie *)getline_cookie(eap->getline,
4539 eap->cookie))->finished = FALSE;
4540
4541 /*
4542 * Cleanup (and inactivate) conditionals, but stop when a try conditional
4543 * not in its finally clause (which then is to be executed next) is found.
4544 * In this case, make the ":finish" pending for execution at the ":endtry".
4545 * Otherwise, finish normally.
4546 */
4547 idx = cleanup_conditionals(eap->cstack, 0, TRUE);
4548 if (idx >= 0)
4549 {
4550 eap->cstack->cs_pending[idx] = CSTP_FINISH;
4551 report_make_pending(CSTP_FINISH, NULL);
4552 }
4553 else
4554 ((struct source_cookie *)getline_cookie(eap->getline,
4555 eap->cookie))->finished = TRUE;
4556}
4557
4558
4559/*
4560 * Return TRUE when a sourced file had the ":finish" command: Don't give error
4561 * message for missing ":endif".
4562 * Return FALSE when not sourcing a file.
4563 */
4564 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004565source_finished(
4566 char_u *(*fgetline)(int, void *, int),
4567 void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004568{
Bram Moolenaar89d40322006-08-29 15:30:07 +00004569 return (getline_equal(fgetline, cookie, getsourceline)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004570 && ((struct source_cookie *)getline_cookie(
Bram Moolenaar89d40322006-08-29 15:30:07 +00004571 fgetline, cookie))->finished);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004572}
4573#endif
4574
4575#if defined(FEAT_LISTCMDS) || defined(PROTO)
4576/*
4577 * ":checktime [buffer]"
4578 */
4579 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004580ex_checktime(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004581{
4582 buf_T *buf;
4583 int save_no_check_timestamps = no_check_timestamps;
4584
4585 no_check_timestamps = 0;
4586 if (eap->addr_count == 0) /* default is all buffers */
4587 check_timestamps(FALSE);
4588 else
4589 {
4590 buf = buflist_findnr((int)eap->line2);
4591 if (buf != NULL) /* cannot happen? */
4592 (void)buf_check_timestamp(buf, FALSE);
4593 }
4594 no_check_timestamps = save_no_check_timestamps;
4595}
4596#endif
4597
Bram Moolenaar071d4272004-06-13 20:20:40 +00004598#if (defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
4599 && (defined(FEAT_EVAL) || defined(FEAT_MULTI_LANG))
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004600# define HAVE_GET_LOCALE_VAL
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004601static char_u *get_locale_val(int what);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004602
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004603 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004604get_locale_val(int what)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004605{
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004606 char_u *loc;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004607
Bram Moolenaar48e330a2016-02-23 14:53:34 +01004608 /* Obtain the locale value from the libraries. */
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004609 loc = (char_u *)setlocale(what, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004610
Bram Moolenaar61660ea2006-04-07 21:40:07 +00004611# ifdef WIN32
Bram Moolenaar071d4272004-06-13 20:20:40 +00004612 if (loc != NULL)
4613 {
4614 char_u *p;
4615
Bram Moolenaar61660ea2006-04-07 21:40:07 +00004616 /* setocale() returns something like "LC_COLLATE=<name>;LC_..." when
4617 * one of the values (e.g., LC_CTYPE) differs. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004618 p = vim_strchr(loc, '=');
4619 if (p != NULL)
4620 {
4621 loc = ++p;
4622 while (*p != NUL) /* remove trailing newline */
4623 {
Bram Moolenaar61660ea2006-04-07 21:40:07 +00004624 if (*p < ' ' || *p == ';')
Bram Moolenaar071d4272004-06-13 20:20:40 +00004625 {
4626 *p = NUL;
4627 break;
4628 }
4629 ++p;
4630 }
4631 }
4632 }
4633# endif
4634
4635 return loc;
4636}
4637#endif
4638
4639
4640#ifdef WIN32
4641/*
4642 * On MS-Windows locale names are strings like "German_Germany.1252", but
4643 * gettext expects "de". Try to translate one into another here for a few
4644 * supported languages.
4645 */
4646 static char_u *
4647gettext_lang(char_u *name)
4648{
4649 int i;
4650 static char *(mtable[]) = {
4651 "afrikaans", "af",
4652 "czech", "cs",
4653 "dutch", "nl",
4654 "german", "de",
4655 "english_united kingdom", "en_GB",
4656 "spanish", "es",
4657 "french", "fr",
4658 "italian", "it",
4659 "japanese", "ja",
4660 "korean", "ko",
4661 "norwegian", "no",
4662 "polish", "pl",
4663 "russian", "ru",
4664 "slovak", "sk",
4665 "swedish", "sv",
4666 "ukrainian", "uk",
4667 "chinese_china", "zh_CN",
4668 "chinese_taiwan", "zh_TW",
4669 NULL};
4670
4671 for (i = 0; mtable[i] != NULL; i += 2)
4672 if (STRNICMP(mtable[i], name, STRLEN(mtable[i])) == 0)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004673 return (char_u *)mtable[i + 1];
Bram Moolenaar071d4272004-06-13 20:20:40 +00004674 return name;
4675}
4676#endif
4677
4678#if defined(FEAT_MULTI_LANG) || defined(PROTO)
4679/*
4680 * Obtain the current messages language. Used to set the default for
4681 * 'helplang'. May return NULL or an empty string.
4682 */
4683 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004684get_mess_lang(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004685{
4686 char_u *p;
4687
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004688# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00004689# if defined(LC_MESSAGES)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004690 p = get_locale_val(LC_MESSAGES);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004691# else
4692 /* This is necessary for Win32, where LC_MESSAGES is not defined and $LANG
Bram Moolenaar61660ea2006-04-07 21:40:07 +00004693 * may be set to the LCID number. LC_COLLATE is the best guess, LC_TIME
4694 * and LC_MONETARY may be set differently for a Japanese working in the
4695 * US. */
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004696 p = get_locale_val(LC_COLLATE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004697# endif
4698# else
4699 p = mch_getenv((char_u *)"LC_ALL");
4700 if (p == NULL || *p == NUL)
4701 {
4702 p = mch_getenv((char_u *)"LC_MESSAGES");
4703 if (p == NULL || *p == NUL)
4704 p = mch_getenv((char_u *)"LANG");
4705 }
4706# endif
4707# ifdef WIN32
4708 p = gettext_lang(p);
4709# endif
4710 return p;
4711}
4712#endif
4713
Bram Moolenaardef9e822004-12-31 20:58:58 +00004714/* Complicated #if; matches with where get_mess_env() is used below. */
4715#if (defined(FEAT_EVAL) && !((defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
4716 && defined(LC_MESSAGES))) \
4717 || ((defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
4718 && (defined(FEAT_GETTEXT) || defined(FEAT_MBYTE)) \
4719 && !defined(LC_MESSAGES))
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01004720static char_u *get_mess_env(void);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004721
4722/*
4723 * Get the language used for messages from the environment.
4724 */
4725 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004726get_mess_env(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004727{
4728 char_u *p;
4729
4730 p = mch_getenv((char_u *)"LC_ALL");
4731 if (p == NULL || *p == NUL)
4732 {
4733 p = mch_getenv((char_u *)"LC_MESSAGES");
4734 if (p == NULL || *p == NUL)
4735 {
4736 p = mch_getenv((char_u *)"LANG");
4737 if (p != NULL && VIM_ISDIGIT(*p))
4738 p = NULL; /* ignore something like "1043" */
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004739# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00004740 if (p == NULL || *p == NUL)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004741 p = get_locale_val(LC_CTYPE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004742# endif
4743 }
4744 }
4745 return p;
4746}
4747#endif
4748
4749#if defined(FEAT_EVAL) || defined(PROTO)
4750
4751/*
4752 * Set the "v:lang" variable according to the current locale setting.
4753 * Also do "v:lc_time"and "v:ctype".
4754 */
4755 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004756set_lang_var(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004757{
4758 char_u *loc;
4759
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004760# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004761 loc = get_locale_val(LC_CTYPE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004762# else
4763 /* setlocale() not supported: use the default value */
4764 loc = (char_u *)"C";
4765# endif
4766 set_vim_var_string(VV_CTYPE, loc, -1);
4767
4768 /* When LC_MESSAGES isn't defined use the value from $LC_MESSAGES, fall
4769 * back to LC_CTYPE if it's empty. */
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004770# if defined(HAVE_GET_LOCALE_VAL) && defined(LC_MESSAGES)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004771 loc = get_locale_val(LC_MESSAGES);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004772# else
4773 loc = get_mess_env();
4774# endif
4775 set_vim_var_string(VV_LANG, loc, -1);
4776
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004777# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004778 loc = get_locale_val(LC_TIME);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004779# endif
4780 set_vim_var_string(VV_LC_TIME, loc, -1);
4781}
4782#endif
4783
4784#if (defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
4785 && (defined(FEAT_GETTEXT) || defined(FEAT_MBYTE))
4786/*
4787 * ":language": Set the language (locale).
4788 */
4789 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004790ex_language(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004791{
4792 char *loc;
4793 char_u *p;
4794 char_u *name;
4795 int what = LC_ALL;
4796 char *whatstr = "";
4797#ifdef LC_MESSAGES
4798# define VIM_LC_MESSAGES LC_MESSAGES
4799#else
4800# define VIM_LC_MESSAGES 6789
4801#endif
4802
4803 name = eap->arg;
4804
4805 /* Check for "messages {name}", "ctype {name}" or "time {name}" argument.
4806 * Allow abbreviation, but require at least 3 characters to avoid
4807 * confusion with a two letter language name "me" or "ct". */
4808 p = skiptowhite(eap->arg);
4809 if ((*p == NUL || vim_iswhite(*p)) && p - eap->arg >= 3)
4810 {
4811 if (STRNICMP(eap->arg, "messages", p - eap->arg) == 0)
4812 {
4813 what = VIM_LC_MESSAGES;
4814 name = skipwhite(p);
4815 whatstr = "messages ";
4816 }
4817 else if (STRNICMP(eap->arg, "ctype", p - eap->arg) == 0)
4818 {
4819 what = LC_CTYPE;
4820 name = skipwhite(p);
4821 whatstr = "ctype ";
4822 }
4823 else if (STRNICMP(eap->arg, "time", p - eap->arg) == 0)
4824 {
4825 what = LC_TIME;
4826 name = skipwhite(p);
4827 whatstr = "time ";
4828 }
4829 }
4830
4831 if (*name == NUL)
4832 {
4833#ifndef LC_MESSAGES
4834 if (what == VIM_LC_MESSAGES)
4835 p = get_mess_env();
4836 else
4837#endif
4838 p = (char_u *)setlocale(what, NULL);
4839 if (p == NULL || *p == NUL)
4840 p = (char_u *)"Unknown";
4841 smsg((char_u *)_("Current %slanguage: \"%s\""), whatstr, p);
4842 }
4843 else
4844 {
4845#ifndef LC_MESSAGES
4846 if (what == VIM_LC_MESSAGES)
4847 loc = "";
4848 else
4849#endif
Bram Moolenaar8c8de832008-06-24 22:58:06 +00004850 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00004851 loc = setlocale(what, (char *)name);
Bram Moolenaar8c8de832008-06-24 22:58:06 +00004852#if defined(FEAT_FLOAT) && defined(LC_NUMERIC)
4853 /* Make sure strtod() uses a decimal point, not a comma. */
4854 setlocale(LC_NUMERIC, "C");
4855#endif
4856 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004857 if (loc == NULL)
4858 EMSG2(_("E197: Cannot set language to \"%s\""), name);
4859 else
4860 {
4861#ifdef HAVE_NL_MSG_CAT_CNTR
4862 /* Need to do this for GNU gettext, otherwise cached translations
4863 * will be used again. */
4864 extern int _nl_msg_cat_cntr;
4865
4866 ++_nl_msg_cat_cntr;
4867#endif
Bram Moolenaarb8017e72007-05-10 18:59:07 +00004868 /* Reset $LC_ALL, otherwise it would overrule everything. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004869 vim_setenv((char_u *)"LC_ALL", (char_u *)"");
4870
4871 if (what != LC_TIME)
4872 {
4873 /* Tell gettext() what to translate to. It apparently doesn't
4874 * use the currently effective locale. Also do this when
4875 * FEAT_GETTEXT isn't defined, so that shell commands use this
4876 * value. */
4877 if (what == LC_ALL)
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00004878 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00004879 vim_setenv((char_u *)"LANG", name);
Bram Moolenaare3a0b532013-06-28 20:36:30 +02004880
4881 /* Clear $LANGUAGE because GNU gettext uses it. */
4882 vim_setenv((char_u *)"LANGUAGE", (char_u *)"");
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00004883# ifdef WIN32
4884 /* Apparently MS-Windows printf() may cause a crash when
4885 * we give it 8-bit text while it's expecting text in the
4886 * current locale. This call avoids that. */
4887 setlocale(LC_CTYPE, "C");
4888# endif
4889 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004890 if (what != LC_CTYPE)
4891 {
4892 char_u *mname;
4893#ifdef WIN32
4894 mname = gettext_lang(name);
4895#else
4896 mname = name;
4897#endif
4898 vim_setenv((char_u *)"LC_MESSAGES", mname);
4899#ifdef FEAT_MULTI_LANG
4900 set_helplang_default(mname);
4901#endif
4902 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004903 }
4904
4905# ifdef FEAT_EVAL
4906 /* Set v:lang, v:lc_time and v:ctype to the final result. */
4907 set_lang_var();
4908# endif
Bram Moolenaar6cc00c72011-10-20 21:41:09 +02004909# ifdef FEAT_TITLE
4910 maketitle();
4911# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004912 }
4913 }
4914}
4915
4916# if defined(FEAT_CMDL_COMPL) || defined(PROTO)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02004917
4918static char_u **locales = NULL; /* Array of all available locales */
4919static int did_init_locales = FALSE;
4920
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01004921static void init_locales(void);
4922static char_u **find_locales(void);
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02004923
4924/*
4925 * Lazy initialization of all available locales.
4926 */
4927 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004928init_locales(void)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02004929{
4930 if (!did_init_locales)
4931 {
4932 did_init_locales = TRUE;
4933 locales = find_locales();
4934 }
4935}
4936
4937/* Return an array of strings for all available locales + NULL for the
4938 * last element. Return NULL in case of error. */
4939 static char_u **
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004940find_locales(void)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02004941{
4942 garray_T locales_ga;
4943 char_u *loc;
4944
4945 /* Find all available locales by running command "locale -a". If this
4946 * doesn't work we won't have completion. */
4947 char_u *locale_a = get_cmd_output((char_u *)"locale -a",
Bram Moolenaar39c29ed2014-04-05 19:44:40 +02004948 NULL, SHELL_SILENT, NULL);
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02004949 if (locale_a == NULL)
4950 return NULL;
4951 ga_init2(&locales_ga, sizeof(char_u *), 20);
4952
4953 /* Transform locale_a string where each locale is separated by "\n"
4954 * into an array of locale strings. */
4955 loc = (char_u *)strtok((char *)locale_a, "\n");
4956
4957 while (loc != NULL)
4958 {
4959 if (ga_grow(&locales_ga, 1) == FAIL)
4960 break;
4961 loc = vim_strsave(loc);
4962 if (loc == NULL)
4963 break;
4964
4965 ((char_u **)locales_ga.ga_data)[locales_ga.ga_len++] = loc;
4966 loc = (char_u *)strtok(NULL, "\n");
4967 }
4968 vim_free(locale_a);
4969 if (ga_grow(&locales_ga, 1) == FAIL)
4970 {
4971 ga_clear(&locales_ga);
4972 return NULL;
4973 }
4974 ((char_u **)locales_ga.ga_data)[locales_ga.ga_len] = NULL;
4975 return (char_u **)locales_ga.ga_data;
4976}
4977
4978# if defined(EXITFREE) || defined(PROTO)
4979 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004980free_locales(void)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02004981{
4982 int i;
4983 if (locales != NULL)
4984 {
4985 for (i = 0; locales[i] != NULL; i++)
4986 vim_free(locales[i]);
4987 vim_free(locales);
4988 locales = NULL;
4989 }
4990}
4991# endif
4992
Bram Moolenaar071d4272004-06-13 20:20:40 +00004993/*
4994 * Function given to ExpandGeneric() to obtain the possible arguments of the
4995 * ":language" command.
4996 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004997 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004998get_lang_arg(expand_T *xp UNUSED, int idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004999{
5000 if (idx == 0)
5001 return (char_u *)"messages";
5002 if (idx == 1)
5003 return (char_u *)"ctype";
5004 if (idx == 2)
5005 return (char_u *)"time";
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005006
5007 init_locales();
5008 if (locales == NULL)
5009 return NULL;
5010 return locales[idx - 3];
5011}
5012
5013/*
5014 * Function given to ExpandGeneric() to obtain the available locales.
5015 */
5016 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005017get_locales(expand_T *xp UNUSED, int idx)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02005018{
5019 init_locales();
5020 if (locales == NULL)
5021 return NULL;
5022 return locales[idx];
Bram Moolenaar071d4272004-06-13 20:20:40 +00005023}
5024# endif
5025
5026#endif