blob: d7bf60901e6832f439e2761e98178a07c699f2dd [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 Moolenaarbfd8fc02005-09-20 23:22:24 +0000167 cmdline = getcmdline_prompt('>', NULL, 0, EXPAND_NOTHING, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000168
Bram Moolenaaree3f7a52008-01-01 13:17:56 +0000169 if (typeahead_saved)
170 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000171 restore_typeahead(&typeaheadbuf);
Bram Moolenaaree3f7a52008-01-01 13:17:56 +0000172 ignore_script = save_ignore_script;
173 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000174 ex_normal_busy = save_ex_normal_busy;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000175
176 cmdline_row = msg_row;
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100177 msg_starthere();
Bram Moolenaar071d4272004-06-13 20:20:40 +0000178 if (cmdline != NULL)
179 {
180 /* If this is a debug command, set "last_cmd".
181 * If not, reset "last_cmd".
182 * For a blank line use previous command. */
183 p = skipwhite(cmdline);
184 if (*p != NUL)
185 {
186 switch (*p)
187 {
188 case 'c': last_cmd = CMD_CONT;
189 tail = "ont";
190 break;
191 case 'n': last_cmd = CMD_NEXT;
192 tail = "ext";
193 break;
194 case 's': last_cmd = CMD_STEP;
195 tail = "tep";
196 break;
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100197 case 'f':
198 last_cmd = 0;
199 if (p[1] == 'r')
200 {
201 last_cmd = CMD_FRAME;
202 tail = "rame";
203 }
204 else
205 {
206 last_cmd = CMD_FINISH;
207 tail = "inish";
208 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000209 break;
210 case 'q': last_cmd = CMD_QUIT;
211 tail = "uit";
212 break;
213 case 'i': last_cmd = CMD_INTERRUPT;
214 tail = "nterrupt";
215 break;
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100216 case 'b': last_cmd = CMD_BACKTRACE;
217 if (p[1] == 't')
218 tail = "t";
219 else
220 tail = "acktrace";
221 break;
222 case 'w': last_cmd = CMD_BACKTRACE;
223 tail = "here";
224 break;
225 case 'u': last_cmd = CMD_UP;
226 tail = "p";
227 break;
228 case 'd': last_cmd = CMD_DOWN;
229 tail = "own";
230 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000231 default: last_cmd = 0;
232 }
233 if (last_cmd != 0)
234 {
235 /* Check that the tail matches. */
236 ++p;
237 while (*p != NUL && *p == *tail)
238 {
239 ++p;
240 ++tail;
241 }
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100242 if (ASCII_ISALPHA(*p) && last_cmd != CMD_FRAME)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000243 last_cmd = 0;
244 }
245 }
246
247 if (last_cmd != 0)
248 {
249 /* Execute debug command: decided where to break next and
250 * return. */
251 switch (last_cmd)
252 {
253 case CMD_CONT:
254 debug_break_level = -1;
255 break;
256 case CMD_NEXT:
257 debug_break_level = ex_nesting_level;
258 break;
259 case CMD_STEP:
260 debug_break_level = 9999;
261 break;
262 case CMD_FINISH:
263 debug_break_level = ex_nesting_level - 1;
264 break;
265 case CMD_QUIT:
266 got_int = TRUE;
267 debug_break_level = -1;
268 break;
269 case CMD_INTERRUPT:
270 got_int = TRUE;
271 debug_break_level = 9999;
272 /* Do not repeat ">interrupt" cmd, continue stepping. */
273 last_cmd = CMD_STEP;
274 break;
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100275 case CMD_BACKTRACE:
276 do_showbacktrace(cmd);
277 continue;
278 case CMD_FRAME:
279 if (*p == NUL)
280 {
281 do_showbacktrace(cmd);
282 }
283 else
284 {
285 p = skipwhite(p);
286 do_setdebugtracelevel(p);
287 }
288 continue;
289 case CMD_UP:
290 debug_backtrace_level++;
291 do_checkbacktracelevel();
292 continue;
293 case CMD_DOWN:
294 debug_backtrace_level--;
295 do_checkbacktracelevel();
296 continue;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000297 }
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100298 /* Going out reset backtrace_level */
299 debug_backtrace_level = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000300 break;
301 }
302
303 /* don't debug this command */
304 n = debug_break_level;
305 debug_break_level = -1;
306 (void)do_cmdline(cmdline, getexline, NULL,
307 DOCMD_VERBOSE|DOCMD_EXCRESET);
308 debug_break_level = n;
309
310 vim_free(cmdline);
311 }
312 lines_left = Rows - 1;
313 }
314 vim_free(cmdline);
315
316 --RedrawingDisabled;
317 --no_wait_return;
318 redraw_all_later(NOT_VALID);
319 need_wait_return = FALSE;
320 msg_scroll = save_msg_scroll;
321 lines_left = Rows - 1;
322 State = save_State;
323 did_emsg = save_did_emsg;
324 cmd_silent = save_cmd_silent;
325 msg_silent = save_msg_silent;
326 emsg_silent = save_emsg_silent;
327 redir_off = save_redir_off;
328
329 /* Only print the message again when typing a command before coming back
330 * here. */
331 debug_did_msg = TRUE;
332}
333
Bram Moolenaarf1f60f82016-01-16 15:40:53 +0100334 static int
335get_maxbacktrace_level(void)
336{
337 char *p, *q;
338 int maxbacktrace = 1;
339
340 maxbacktrace = 0;
341 if (sourcing_name != NULL)
342 {
343 p = (char *)sourcing_name;
344 while ((q = strstr(p, "..")) != NULL)
345 {
346 p = q + 2;
347 maxbacktrace++;
348 }
349 }
350 return maxbacktrace;
351}
352
353 static void
354do_setdebugtracelevel(char_u *arg)
355{
356 int level;
357
358 level = atoi((char *)arg);
359 if (*arg == '+' || level < 0)
360 debug_backtrace_level += level;
361 else
362 debug_backtrace_level = level;
363
364 do_checkbacktracelevel();
365}
366
367 static void
368do_checkbacktracelevel(void)
369{
370 if (debug_backtrace_level < 0)
371 {
372 debug_backtrace_level = 0;
373 MSG(_("frame is zero"));
374 }
375 else
376 {
377 int max = get_maxbacktrace_level();
378
379 if (debug_backtrace_level > max)
380 {
381 debug_backtrace_level = max;
382 smsg((char_u *)_("frame at highest level: %d"), max);
383 }
384 }
385}
386
387 static void
388do_showbacktrace(char_u *cmd)
389{
390 char *cur;
391 char *next;
392 int i = 0;
393 int max = get_maxbacktrace_level();
394
395 if (sourcing_name != NULL)
396 {
397 cur = (char *)sourcing_name;
398 while (!got_int)
399 {
400 next = strstr(cur, "..");
401 if (next != NULL)
402 *next = NUL;
403 if (i == max - debug_backtrace_level)
404 smsg((char_u *)"->%d %s", max - i, cur);
405 else
406 smsg((char_u *)" %d %s", max - i, cur);
407 ++i;
408 if (next == NULL)
409 break;
410 *next = '.';
411 cur = next + 2;
412 }
413 }
414 if (sourcing_lnum != 0)
415 smsg((char_u *)_("line %ld: %s"), (long)sourcing_lnum, cmd);
416 else
417 smsg((char_u *)_("cmd: %s"), cmd);
418}
419
Bram Moolenaar071d4272004-06-13 20:20:40 +0000420/*
421 * ":debug".
422 */
423 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100424ex_debug(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000425{
426 int debug_break_level_save = debug_break_level;
427
428 debug_break_level = 9999;
429 do_cmdline_cmd(eap->arg);
430 debug_break_level = debug_break_level_save;
431}
432
433static char_u *debug_breakpoint_name = NULL;
434static linenr_T debug_breakpoint_lnum;
435
436/*
437 * When debugging or a breakpoint is set on a skipped command, no debug prompt
438 * is shown by do_one_cmd(). This situation is indicated by debug_skipped, and
439 * debug_skipped_name is then set to the source name in the breakpoint case. If
440 * a skipped command decides itself that a debug prompt should be displayed, it
441 * can do so by calling dbg_check_skipped().
442 */
443static int debug_skipped;
444static char_u *debug_skipped_name;
445
446/*
447 * Go to debug mode when a breakpoint was encountered or "ex_nesting_level" is
448 * at or below the break level. But only when the line is actually
449 * executed. Return TRUE and set breakpoint_name for skipped commands that
450 * decide to execute something themselves.
451 * Called from do_one_cmd() before executing a command.
452 */
453 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100454dbg_check_breakpoint(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000455{
456 char_u *p;
457
458 debug_skipped = FALSE;
459 if (debug_breakpoint_name != NULL)
460 {
461 if (!eap->skip)
462 {
463 /* replace K_SNR with "<SNR>" */
464 if (debug_breakpoint_name[0] == K_SPECIAL
465 && debug_breakpoint_name[1] == KS_EXTRA
466 && debug_breakpoint_name[2] == (int)KE_SNR)
467 p = (char_u *)"<SNR>";
468 else
469 p = (char_u *)"";
Bram Moolenaar555b2802005-05-19 21:08:39 +0000470 smsg((char_u *)_("Breakpoint in \"%s%s\" line %ld"),
471 p,
Bram Moolenaar071d4272004-06-13 20:20:40 +0000472 debug_breakpoint_name + (*p == NUL ? 0 : 3),
473 (long)debug_breakpoint_lnum);
474 debug_breakpoint_name = NULL;
475 do_debug(eap->cmd);
476 }
477 else
478 {
479 debug_skipped = TRUE;
480 debug_skipped_name = debug_breakpoint_name;
481 debug_breakpoint_name = NULL;
482 }
483 }
484 else if (ex_nesting_level <= debug_break_level)
485 {
486 if (!eap->skip)
487 do_debug(eap->cmd);
488 else
489 {
490 debug_skipped = TRUE;
491 debug_skipped_name = NULL;
492 }
493 }
494}
495
496/*
497 * Go to debug mode if skipped by dbg_check_breakpoint() because eap->skip was
498 * set. Return TRUE when the debug mode is entered this time.
499 */
500 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100501dbg_check_skipped(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000502{
503 int prev_got_int;
504
505 if (debug_skipped)
506 {
507 /*
508 * Save the value of got_int and reset it. We don't want a previous
509 * interruption cause flushing the input buffer.
510 */
511 prev_got_int = got_int;
512 got_int = FALSE;
513 debug_breakpoint_name = debug_skipped_name;
514 /* eap->skip is TRUE */
515 eap->skip = FALSE;
516 (void)dbg_check_breakpoint(eap);
517 eap->skip = TRUE;
518 got_int |= prev_got_int;
519 return TRUE;
520 }
521 return FALSE;
522}
523
524/*
525 * The list of breakpoints: dbg_breakp.
526 * This is a grow-array of structs.
527 */
528struct debuggy
529{
530 int dbg_nr; /* breakpoint number */
531 int dbg_type; /* DBG_FUNC or DBG_FILE */
532 char_u *dbg_name; /* function or file name */
533 regprog_T *dbg_prog; /* regexp program */
534 linenr_T dbg_lnum; /* line number in function or file */
Bram Moolenaar05159a02005-02-26 23:04:13 +0000535 int dbg_forceit; /* ! used */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000536};
537
538static garray_T dbg_breakp = {0, 0, sizeof(struct debuggy), 4, NULL};
Bram Moolenaar05159a02005-02-26 23:04:13 +0000539#define BREAKP(idx) (((struct debuggy *)dbg_breakp.ga_data)[idx])
540#define DEBUGGY(gap, idx) (((struct debuggy *)gap->ga_data)[idx])
Bram Moolenaar071d4272004-06-13 20:20:40 +0000541static int last_breakp = 0; /* nr of last defined breakpoint */
542
Bram Moolenaar05159a02005-02-26 23:04:13 +0000543#ifdef FEAT_PROFILE
544/* Profiling uses file and func names similar to breakpoints. */
545static garray_T prof_ga = {0, 0, sizeof(struct debuggy), 4, NULL};
546#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000547#define DBG_FUNC 1
548#define DBG_FILE 2
549
Bram Moolenaarf28dbce2016-01-29 22:03:47 +0100550static int dbg_parsearg(char_u *arg, garray_T *gap);
551static linenr_T debuggy_find(int file,char_u *fname, linenr_T after, garray_T *gap, int *fp);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000552
553/*
Bram Moolenaar05159a02005-02-26 23:04:13 +0000554 * Parse the arguments of ":profile", ":breakadd" or ":breakdel" and put them
555 * in the entry just after the last one in dbg_breakp. Note that "dbg_name"
556 * is allocated.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000557 * Returns FAIL for failure.
558 */
559 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100560dbg_parsearg(
561 char_u *arg,
562 garray_T *gap) /* either &dbg_breakp or &prof_ga */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000563{
564 char_u *p = arg;
565 char_u *q;
566 struct debuggy *bp;
Bram Moolenaarf4b8e572004-06-24 15:53:16 +0000567 int here = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000568
Bram Moolenaar05159a02005-02-26 23:04:13 +0000569 if (ga_grow(gap, 1) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000570 return FAIL;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000571 bp = &DEBUGGY(gap, gap->ga_len);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000572
573 /* Find "func" or "file". */
574 if (STRNCMP(p, "func", 4) == 0)
575 bp->dbg_type = DBG_FUNC;
576 else if (STRNCMP(p, "file", 4) == 0)
577 bp->dbg_type = DBG_FILE;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000578 else if (
579#ifdef FEAT_PROFILE
580 gap != &prof_ga &&
581#endif
582 STRNCMP(p, "here", 4) == 0)
Bram Moolenaarf4b8e572004-06-24 15:53:16 +0000583 {
584 if (curbuf->b_ffname == NULL)
585 {
586 EMSG(_(e_noname));
587 return FAIL;
588 }
589 bp->dbg_type = DBG_FILE;
590 here = TRUE;
591 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000592 else
593 {
594 EMSG2(_(e_invarg2), p);
595 return FAIL;
596 }
597 p = skipwhite(p + 4);
598
599 /* Find optional line number. */
Bram Moolenaarf4b8e572004-06-24 15:53:16 +0000600 if (here)
601 bp->dbg_lnum = curwin->w_cursor.lnum;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000602 else if (
603#ifdef FEAT_PROFILE
604 gap != &prof_ga &&
605#endif
606 VIM_ISDIGIT(*p))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000607 {
608 bp->dbg_lnum = getdigits(&p);
609 p = skipwhite(p);
610 }
611 else
612 bp->dbg_lnum = 0;
613
614 /* Find the function or file name. Don't accept a function name with (). */
Bram Moolenaarf4b8e572004-06-24 15:53:16 +0000615 if ((!here && *p == NUL)
616 || (here && *p != NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000617 || (bp->dbg_type == DBG_FUNC && strstr((char *)p, "()") != NULL))
618 {
619 EMSG2(_(e_invarg2), arg);
620 return FAIL;
621 }
622
623 if (bp->dbg_type == DBG_FUNC)
624 bp->dbg_name = vim_strsave(p);
Bram Moolenaarf4b8e572004-06-24 15:53:16 +0000625 else if (here)
626 bp->dbg_name = vim_strsave(curbuf->b_ffname);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000627 else
628 {
629 /* Expand the file name in the same way as do_source(). This means
630 * doing it twice, so that $DIR/file gets expanded when $DIR is
631 * "~/dir". */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000632 q = expand_env_save(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000633 if (q == NULL)
634 return FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000635 p = expand_env_save(q);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000636 vim_free(q);
637 if (p == NULL)
638 return FAIL;
Bram Moolenaar843ee412004-06-30 16:16:41 +0000639 if (*p != '*')
640 {
641 bp->dbg_name = fix_fname(p);
642 vim_free(p);
643 }
644 else
645 bp->dbg_name = p;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000646 }
647
648 if (bp->dbg_name == NULL)
649 return FAIL;
650 return OK;
651}
652
653/*
654 * ":breakadd".
655 */
656 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100657ex_breakadd(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000658{
659 struct debuggy *bp;
660 char_u *pat;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000661 garray_T *gap;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000662
Bram Moolenaar05159a02005-02-26 23:04:13 +0000663 gap = &dbg_breakp;
664#ifdef FEAT_PROFILE
665 if (eap->cmdidx == CMD_profile)
666 gap = &prof_ga;
667#endif
668
669 if (dbg_parsearg(eap->arg, gap) == OK)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000670 {
Bram Moolenaar05159a02005-02-26 23:04:13 +0000671 bp = &DEBUGGY(gap, gap->ga_len);
672 bp->dbg_forceit = eap->forceit;
673
Bram Moolenaar071d4272004-06-13 20:20:40 +0000674 pat = file_pat_to_reg_pat(bp->dbg_name, NULL, NULL, FALSE);
675 if (pat != NULL)
676 {
677 bp->dbg_prog = vim_regcomp(pat, RE_MAGIC + RE_STRING);
678 vim_free(pat);
679 }
680 if (pat == NULL || bp->dbg_prog == NULL)
681 vim_free(bp->dbg_name);
682 else
683 {
684 if (bp->dbg_lnum == 0) /* default line number is 1 */
685 bp->dbg_lnum = 1;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000686#ifdef FEAT_PROFILE
687 if (eap->cmdidx != CMD_profile)
688#endif
689 {
690 DEBUGGY(gap, gap->ga_len).dbg_nr = ++last_breakp;
691 ++debug_tick;
692 }
693 ++gap->ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000694 }
695 }
696}
697
698/*
699 * ":debuggreedy".
700 */
701 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100702ex_debuggreedy(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000703{
704 if (eap->addr_count == 0 || eap->line2 != 0)
705 debug_greedy = TRUE;
706 else
707 debug_greedy = FALSE;
708}
709
710/*
Bram Moolenaard9fba312005-06-26 22:34:35 +0000711 * ":breakdel" and ":profdel".
Bram Moolenaar071d4272004-06-13 20:20:40 +0000712 */
713 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100714ex_breakdel(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000715{
716 struct debuggy *bp, *bpi;
717 int nr;
718 int todel = -1;
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000719 int del_all = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000720 int i;
721 linenr_T best_lnum = 0;
Bram Moolenaard9fba312005-06-26 22:34:35 +0000722 garray_T *gap;
723
724 gap = &dbg_breakp;
Bram Moolenaard9fba312005-06-26 22:34:35 +0000725 if (eap->cmdidx == CMD_profdel)
Bram Moolenaar38bdbd62012-06-20 15:48:57 +0200726 {
727#ifdef FEAT_PROFILE
Bram Moolenaard9fba312005-06-26 22:34:35 +0000728 gap = &prof_ga;
Bram Moolenaar38bdbd62012-06-20 15:48:57 +0200729#else
730 ex_ni(eap);
731 return;
Bram Moolenaard9fba312005-06-26 22:34:35 +0000732#endif
Bram Moolenaar38bdbd62012-06-20 15:48:57 +0200733 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000734
735 if (vim_isdigit(*eap->arg))
736 {
737 /* ":breakdel {nr}" */
738 nr = atol((char *)eap->arg);
Bram Moolenaard9fba312005-06-26 22:34:35 +0000739 for (i = 0; i < gap->ga_len; ++i)
740 if (DEBUGGY(gap, i).dbg_nr == nr)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000741 {
742 todel = i;
743 break;
744 }
745 }
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000746 else if (*eap->arg == '*')
747 {
748 todel = 0;
749 del_all = TRUE;
750 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000751 else
752 {
753 /* ":breakdel {func|file} [lnum] {name}" */
Bram Moolenaard9fba312005-06-26 22:34:35 +0000754 if (dbg_parsearg(eap->arg, gap) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000755 return;
Bram Moolenaard9fba312005-06-26 22:34:35 +0000756 bp = &DEBUGGY(gap, gap->ga_len);
757 for (i = 0; i < gap->ga_len; ++i)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000758 {
Bram Moolenaard9fba312005-06-26 22:34:35 +0000759 bpi = &DEBUGGY(gap, i);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000760 if (bp->dbg_type == bpi->dbg_type
761 && STRCMP(bp->dbg_name, bpi->dbg_name) == 0
762 && (bp->dbg_lnum == bpi->dbg_lnum
763 || (bp->dbg_lnum == 0
764 && (best_lnum == 0
765 || bpi->dbg_lnum < best_lnum))))
766 {
767 todel = i;
768 best_lnum = bpi->dbg_lnum;
769 }
770 }
771 vim_free(bp->dbg_name);
772 }
773
774 if (todel < 0)
775 EMSG2(_("E161: Breakpoint not found: %s"), eap->arg);
776 else
Bram Moolenaard9fba312005-06-26 22:34:35 +0000777 {
778 while (gap->ga_len > 0)
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000779 {
Bram Moolenaard9fba312005-06-26 22:34:35 +0000780 vim_free(DEBUGGY(gap, todel).dbg_name);
Bram Moolenaar473de612013-06-08 18:19:48 +0200781 vim_regfree(DEBUGGY(gap, todel).dbg_prog);
Bram Moolenaard9fba312005-06-26 22:34:35 +0000782 --gap->ga_len;
783 if (todel < gap->ga_len)
784 mch_memmove(&DEBUGGY(gap, todel), &DEBUGGY(gap, todel + 1),
785 (gap->ga_len - todel) * sizeof(struct debuggy));
786#ifdef FEAT_PROFILE
787 if (eap->cmdidx == CMD_breakdel)
788#endif
789 ++debug_tick;
Bram Moolenaarf461c8e2005-06-25 23:04:51 +0000790 if (!del_all)
791 break;
792 }
Bram Moolenaard9fba312005-06-26 22:34:35 +0000793
794 /* If all breakpoints were removed clear the array. */
795 if (gap->ga_len == 0)
796 ga_clear(gap);
797 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000798}
799
800/*
801 * ":breaklist".
802 */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000803 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100804ex_breaklist(exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000805{
806 struct debuggy *bp;
807 int i;
808
809 if (dbg_breakp.ga_len == 0)
810 MSG(_("No breakpoints defined"));
811 else
812 for (i = 0; i < dbg_breakp.ga_len; ++i)
813 {
814 bp = &BREAKP(i);
Bram Moolenaard58ea072011-06-26 04:25:30 +0200815 if (bp->dbg_type == DBG_FILE)
816 home_replace(NULL, bp->dbg_name, NameBuff, MAXPATHL, TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000817 smsg((char_u *)_("%3d %s %s line %ld"),
818 bp->dbg_nr,
819 bp->dbg_type == DBG_FUNC ? "func" : "file",
Bram Moolenaard58ea072011-06-26 04:25:30 +0200820 bp->dbg_type == DBG_FUNC ? bp->dbg_name : NameBuff,
Bram Moolenaar071d4272004-06-13 20:20:40 +0000821 (long)bp->dbg_lnum);
822 }
823}
824
825/*
826 * Find a breakpoint for a function or sourced file.
827 * Returns line number at which to break; zero when no matching breakpoint.
828 */
829 linenr_T
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100830dbg_find_breakpoint(
831 int file, /* TRUE for a file, FALSE for a function */
832 char_u *fname, /* file or function name */
833 linenr_T after) /* after this line number */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000834{
Bram Moolenaar05159a02005-02-26 23:04:13 +0000835 return debuggy_find(file, fname, after, &dbg_breakp, NULL);
836}
837
838#if defined(FEAT_PROFILE) || defined(PROTO)
839/*
840 * Return TRUE if profiling is on for a function or sourced file.
841 */
842 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100843has_profiling(
844 int file, /* TRUE for a file, FALSE for a function */
845 char_u *fname, /* file or function name */
846 int *fp) /* return: forceit */
Bram Moolenaar05159a02005-02-26 23:04:13 +0000847{
848 return (debuggy_find(file, fname, (linenr_T)0, &prof_ga, fp)
849 != (linenr_T)0);
850}
851#endif
852
853/*
854 * Common code for dbg_find_breakpoint() and has_profiling().
855 */
856 static linenr_T
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100857debuggy_find(
858 int file, /* TRUE for a file, FALSE for a function */
859 char_u *fname, /* file or function name */
860 linenr_T after, /* after this line number */
861 garray_T *gap, /* either &dbg_breakp or &prof_ga */
862 int *fp) /* if not NULL: return forceit */
Bram Moolenaar05159a02005-02-26 23:04:13 +0000863{
Bram Moolenaar071d4272004-06-13 20:20:40 +0000864 struct debuggy *bp;
865 int i;
866 linenr_T lnum = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000867 char_u *name = fname;
868 int prev_got_int;
869
Bram Moolenaar05159a02005-02-26 23:04:13 +0000870 /* Return quickly when there are no breakpoints. */
871 if (gap->ga_len == 0)
872 return (linenr_T)0;
873
Bram Moolenaar071d4272004-06-13 20:20:40 +0000874 /* Replace K_SNR in function name with "<SNR>". */
875 if (!file && fname[0] == K_SPECIAL)
876 {
877 name = alloc((unsigned)STRLEN(fname) + 3);
878 if (name == NULL)
879 name = fname;
880 else
881 {
882 STRCPY(name, "<SNR>");
883 STRCPY(name + 5, fname + 3);
884 }
885 }
886
Bram Moolenaar05159a02005-02-26 23:04:13 +0000887 for (i = 0; i < gap->ga_len; ++i)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000888 {
Bram Moolenaar05159a02005-02-26 23:04:13 +0000889 /* Skip entries that are not useful or are for a line that is beyond
890 * an already found breakpoint. */
891 bp = &DEBUGGY(gap, i);
892 if (((bp->dbg_type == DBG_FILE) == file && (
893#ifdef FEAT_PROFILE
894 gap == &prof_ga ||
895#endif
896 (bp->dbg_lnum > after && (lnum == 0 || bp->dbg_lnum < lnum)))))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000897 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000898 /*
Bram Moolenaar05159a02005-02-26 23:04:13 +0000899 * Save the value of got_int and reset it. We don't want a
900 * previous interruption cancel matching, only hitting CTRL-C
901 * while matching should abort it.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000902 */
903 prev_got_int = got_int;
904 got_int = FALSE;
Bram Moolenaardffa5b82014-11-19 16:38:07 +0100905 if (vim_regexec_prog(&bp->dbg_prog, FALSE, name, (colnr_T)0))
Bram Moolenaar05159a02005-02-26 23:04:13 +0000906 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000907 lnum = bp->dbg_lnum;
Bram Moolenaar05159a02005-02-26 23:04:13 +0000908 if (fp != NULL)
909 *fp = bp->dbg_forceit;
910 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000911 got_int |= prev_got_int;
912 }
913 }
914 if (name != fname)
915 vim_free(name);
916
917 return lnum;
918}
919
920/*
921 * Called when a breakpoint was encountered.
922 */
923 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100924dbg_breakpoint(char_u *name, linenr_T lnum)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000925{
926 /* We need to check if this line is actually executed in do_one_cmd() */
927 debug_breakpoint_name = name;
928 debug_breakpoint_lnum = lnum;
929}
Bram Moolenaar05159a02005-02-26 23:04:13 +0000930
931
Bram Moolenaar433f7c82006-03-21 21:29:36 +0000932# if defined(FEAT_PROFILE) || defined(FEAT_RELTIME) || defined(PROTO)
Bram Moolenaar05159a02005-02-26 23:04:13 +0000933/*
934 * Store the current time in "tm".
935 */
936 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100937profile_start(proftime_T *tm)
Bram Moolenaar05159a02005-02-26 23:04:13 +0000938{
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000939# ifdef WIN3264
940 QueryPerformanceCounter(tm);
941# else
Bram Moolenaar05159a02005-02-26 23:04:13 +0000942 gettimeofday(tm, NULL);
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000943# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +0000944}
945
946/*
947 * Compute the elapsed time from "tm" till now and store in "tm".
948 */
949 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100950profile_end(proftime_T *tm)
Bram Moolenaar05159a02005-02-26 23:04:13 +0000951{
952 proftime_T now;
953
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000954# ifdef WIN3264
955 QueryPerformanceCounter(&now);
956 tm->QuadPart = now.QuadPart - tm->QuadPart;
957# else
Bram Moolenaar05159a02005-02-26 23:04:13 +0000958 gettimeofday(&now, NULL);
959 tm->tv_usec = now.tv_usec - tm->tv_usec;
960 tm->tv_sec = now.tv_sec - tm->tv_sec;
961 if (tm->tv_usec < 0)
962 {
963 tm->tv_usec += 1000000;
964 --tm->tv_sec;
965 }
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000966# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +0000967}
968
969/*
970 * Subtract the time "tm2" from "tm".
971 */
972 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100973profile_sub(proftime_T *tm, proftime_T *tm2)
Bram Moolenaar05159a02005-02-26 23:04:13 +0000974{
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000975# ifdef WIN3264
976 tm->QuadPart -= tm2->QuadPart;
977# else
Bram Moolenaar05159a02005-02-26 23:04:13 +0000978 tm->tv_usec -= tm2->tv_usec;
979 tm->tv_sec -= tm2->tv_sec;
980 if (tm->tv_usec < 0)
981 {
982 tm->tv_usec += 1000000;
983 --tm->tv_sec;
984 }
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +0000985# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +0000986}
987
988/*
Bram Moolenaar433f7c82006-03-21 21:29:36 +0000989 * Return a string that represents the time in "tm".
990 * Uses a static buffer!
991 */
992 char *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100993profile_msg(proftime_T *tm)
Bram Moolenaar433f7c82006-03-21 21:29:36 +0000994{
995 static char buf[50];
996
997# ifdef WIN3264
998 LARGE_INTEGER fr;
999
1000 QueryPerformanceFrequency(&fr);
1001 sprintf(buf, "%10.6lf", (double)tm->QuadPart / (double)fr.QuadPart);
1002# else
1003 sprintf(buf, "%3ld.%06ld", (long)tm->tv_sec, (long)tm->tv_usec);
Bram Moolenaar76929292008-01-06 19:07:36 +00001004# endif
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001005 return buf;
1006}
1007
Bram Moolenaar79c2c882016-02-07 21:19:28 +01001008# if defined(FEAT_FLOAT) || defined(PROTO)
1009/*
1010 * Return a float that represents the time in "tm".
1011 */
1012 float_T
1013profile_float(proftime_T *tm)
1014{
1015# ifdef WIN3264
1016 LARGE_INTEGER fr;
1017
1018 QueryPerformanceFrequency(&fr);
1019 return (float_T)tm->QuadPart / (float_T)fr.QuadPart;
1020# else
1021 return (float_T)tm->tv_sec + (float_T)tm->tv_usec / 1000000.0;
1022# endif
1023}
1024# endif
1025
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001026/*
Bram Moolenaar76929292008-01-06 19:07:36 +00001027 * Put the time "msec" past now in "tm".
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001028 */
Bram Moolenaar76929292008-01-06 19:07:36 +00001029 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001030profile_setlimit(long msec, proftime_T *tm)
Bram Moolenaar76929292008-01-06 19:07:36 +00001031{
1032 if (msec <= 0) /* no limit */
1033 profile_zero(tm);
1034 else
1035 {
1036# ifdef WIN3264
1037 LARGE_INTEGER fr;
1038
1039 QueryPerformanceCounter(tm);
1040 QueryPerformanceFrequency(&fr);
Bram Moolenaarb3c70982008-01-18 10:40:55 +00001041 tm->QuadPart += (LONGLONG)((double)msec / 1000.0 * (double)fr.QuadPart);
Bram Moolenaar76929292008-01-06 19:07:36 +00001042# else
1043 long usec;
1044
1045 gettimeofday(tm, NULL);
1046 usec = (long)tm->tv_usec + (long)msec * 1000;
1047 tm->tv_usec = usec % 1000000L;
1048 tm->tv_sec += usec / 1000000L;
1049# endif
1050 }
1051}
1052
1053/*
1054 * Return TRUE if the current time is past "tm".
1055 */
1056 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001057profile_passed_limit(proftime_T *tm)
Bram Moolenaar76929292008-01-06 19:07:36 +00001058{
1059 proftime_T now;
1060
1061# ifdef WIN3264
1062 if (tm->QuadPart == 0) /* timer was not set */
1063 return FALSE;
1064 QueryPerformanceCounter(&now);
1065 return (now.QuadPart > tm->QuadPart);
1066# else
1067 if (tm->tv_sec == 0) /* timer was not set */
1068 return FALSE;
1069 gettimeofday(&now, NULL);
1070 return (now.tv_sec > tm->tv_sec
1071 || (now.tv_sec == tm->tv_sec && now.tv_usec > tm->tv_usec));
1072# endif
1073}
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001074
1075/*
1076 * Set the time in "tm" to zero.
1077 */
1078 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001079profile_zero(proftime_T *tm)
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001080{
1081# ifdef WIN3264
1082 tm->QuadPart = 0;
1083# else
1084 tm->tv_usec = 0;
1085 tm->tv_sec = 0;
1086# endif
1087}
1088
Bram Moolenaar76929292008-01-06 19:07:36 +00001089# endif /* FEAT_PROFILE || FEAT_RELTIME */
1090
Bram Moolenaar8a7f5a22013-06-06 14:01:46 +02001091#if defined(FEAT_SYN_HL) && defined(FEAT_RELTIME) && defined(FEAT_FLOAT)
1092# if defined(HAVE_MATH_H)
1093# include <math.h>
1094# endif
1095
1096/*
1097 * Divide the time "tm" by "count" and store in "tm2".
1098 */
1099 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001100profile_divide(proftime_T *tm, int count, proftime_T *tm2)
Bram Moolenaar8a7f5a22013-06-06 14:01:46 +02001101{
1102 if (count == 0)
1103 profile_zero(tm2);
1104 else
1105 {
1106# ifdef WIN3264
1107 tm2->QuadPart = tm->QuadPart / count;
1108# else
1109 double usec = (tm->tv_sec * 1000000.0 + tm->tv_usec) / count;
1110
1111 tm2->tv_sec = floor(usec / 1000000.0);
Bram Moolenaara2e14fc2013-06-10 20:10:44 +02001112 tm2->tv_usec = vim_round(usec - (tm2->tv_sec * 1000000.0));
Bram Moolenaar8a7f5a22013-06-06 14:01:46 +02001113# endif
1114 }
1115}
1116#endif
1117
Bram Moolenaar76929292008-01-06 19:07:36 +00001118# if defined(FEAT_PROFILE) || defined(PROTO)
1119/*
1120 * Functions for profiling.
1121 */
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01001122static void script_do_profile(scriptitem_T *si);
1123static void script_dump_profile(FILE *fd);
Bram Moolenaar76929292008-01-06 19:07:36 +00001124static proftime_T prof_wait_time;
1125
Bram Moolenaar433f7c82006-03-21 21:29:36 +00001126/*
Bram Moolenaar05159a02005-02-26 23:04:13 +00001127 * Add the time "tm2" to "tm".
1128 */
1129 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001130profile_add(proftime_T *tm, proftime_T *tm2)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001131{
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001132# ifdef WIN3264
1133 tm->QuadPart += tm2->QuadPart;
1134# else
Bram Moolenaar05159a02005-02-26 23:04:13 +00001135 tm->tv_usec += tm2->tv_usec;
1136 tm->tv_sec += tm2->tv_sec;
1137 if (tm->tv_usec >= 1000000)
1138 {
1139 tm->tv_usec -= 1000000;
1140 ++tm->tv_sec;
1141 }
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001142# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00001143}
1144
1145/*
Bram Moolenaar1056d982006-03-09 22:37:52 +00001146 * Add the "self" time from the total time and the children's time.
1147 */
1148 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001149profile_self(proftime_T *self, proftime_T *total, proftime_T *children)
Bram Moolenaar1056d982006-03-09 22:37:52 +00001150{
1151 /* Check that the result won't be negative. Can happen with recursive
1152 * calls. */
1153#ifdef WIN3264
1154 if (total->QuadPart <= children->QuadPart)
1155 return;
1156#else
1157 if (total->tv_sec < children->tv_sec
1158 || (total->tv_sec == children->tv_sec
1159 && total->tv_usec <= children->tv_usec))
1160 return;
1161#endif
1162 profile_add(self, total);
1163 profile_sub(self, children);
1164}
1165
1166/*
Bram Moolenaar05159a02005-02-26 23:04:13 +00001167 * Get the current waittime.
1168 */
1169 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001170profile_get_wait(proftime_T *tm)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001171{
1172 *tm = prof_wait_time;
1173}
1174
1175/*
1176 * Subtract the passed waittime since "tm" from "tma".
1177 */
1178 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001179profile_sub_wait(proftime_T *tm, proftime_T *tma)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001180{
1181 proftime_T tm3 = prof_wait_time;
1182
1183 profile_sub(&tm3, tm);
1184 profile_sub(tma, &tm3);
1185}
1186
1187/*
1188 * Return TRUE if "tm1" and "tm2" are equal.
1189 */
1190 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001191profile_equal(proftime_T *tm1, proftime_T *tm2)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001192{
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001193# ifdef WIN3264
1194 return (tm1->QuadPart == tm2->QuadPart);
1195# else
Bram Moolenaar05159a02005-02-26 23:04:13 +00001196 return (tm1->tv_usec == tm2->tv_usec && tm1->tv_sec == tm2->tv_sec);
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001197# endif
1198}
1199
1200/*
1201 * Return <0, 0 or >0 if "tm1" < "tm2", "tm1" == "tm2" or "tm1" > "tm2"
1202 */
1203 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001204profile_cmp(const proftime_T *tm1, const proftime_T *tm2)
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001205{
1206# ifdef WIN3264
1207 return (int)(tm2->QuadPart - tm1->QuadPart);
1208# else
1209 if (tm1->tv_sec == tm2->tv_sec)
1210 return tm2->tv_usec - tm1->tv_usec;
1211 return tm2->tv_sec - tm1->tv_sec;
1212# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00001213}
1214
Bram Moolenaar05159a02005-02-26 23:04:13 +00001215static char_u *profile_fname = NULL;
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001216static proftime_T pause_time;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001217
1218/*
1219 * ":profile cmd args"
1220 */
1221 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001222ex_profile(exarg_T *eap)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001223{
1224 char_u *e;
1225 int len;
1226
1227 e = skiptowhite(eap->arg);
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00001228 len = (int)(e - eap->arg);
Bram Moolenaar05159a02005-02-26 23:04:13 +00001229 e = skipwhite(e);
1230
1231 if (len == 5 && STRNCMP(eap->arg, "start", 5) == 0 && *e != NUL)
1232 {
1233 vim_free(profile_fname);
Bram Moolenaard94682f2015-04-13 15:37:56 +02001234 profile_fname = expand_env_save_opt(e, TRUE);
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001235 do_profiling = PROF_YES;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001236 profile_zero(&prof_wait_time);
1237 set_vim_var_nr(VV_PROFILING, 1L);
1238 }
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001239 else if (do_profiling == PROF_NONE)
Bram Moolenaar54c1b492010-02-24 14:01:28 +01001240 EMSG(_("E750: First use \":profile start {fname}\""));
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00001241 else if (STRCMP(eap->arg, "pause") == 0)
1242 {
1243 if (do_profiling == PROF_YES)
1244 profile_start(&pause_time);
1245 do_profiling = PROF_PAUSED;
1246 }
1247 else if (STRCMP(eap->arg, "continue") == 0)
1248 {
1249 if (do_profiling == PROF_PAUSED)
1250 {
1251 profile_end(&pause_time);
1252 profile_add(&prof_wait_time, &pause_time);
1253 }
1254 do_profiling = PROF_YES;
1255 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00001256 else
1257 {
1258 /* The rest is similar to ":breakadd". */
1259 ex_breakadd(eap);
1260 }
1261}
1262
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001263/* Command line expansion for :profile. */
1264static enum
1265{
1266 PEXP_SUBCMD, /* expand :profile sub-commands */
Bram Moolenaar49789dc2011-02-25 14:46:09 +01001267 PEXP_FUNC /* expand :profile func {funcname} */
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001268} pexpand_what;
1269
1270static char *pexpand_cmds[] = {
1271 "start",
1272#define PROFCMD_START 0
1273 "pause",
1274#define PROFCMD_PAUSE 1
1275 "continue",
1276#define PROFCMD_CONTINUE 2
1277 "func",
1278#define PROFCMD_FUNC 3
1279 "file",
1280#define PROFCMD_FILE 4
1281 NULL
1282#define PROFCMD_LAST 5
1283};
1284
1285/*
1286 * Function given to ExpandGeneric() to obtain the profile command
1287 * specific expansion.
1288 */
1289 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001290get_profile_name(expand_T *xp UNUSED, int idx)
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001291{
1292 switch (pexpand_what)
1293 {
1294 case PEXP_SUBCMD:
1295 return (char_u *)pexpand_cmds[idx];
1296 /* case PEXP_FUNC: TODO */
1297 default:
1298 return NULL;
1299 }
1300}
1301
1302/*
1303 * Handle command line completion for :profile command.
1304 */
1305 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001306set_context_in_profile_cmd(expand_T *xp, char_u *arg)
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001307{
1308 char_u *end_subcmd;
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001309
1310 /* Default: expand subcommands. */
1311 xp->xp_context = EXPAND_PROFILE;
1312 pexpand_what = PEXP_SUBCMD;
1313 xp->xp_pattern = arg;
1314
1315 end_subcmd = skiptowhite(arg);
1316 if (*end_subcmd == NUL)
1317 return;
1318
Bram Moolenaar8b9c05f2010-03-02 17:54:33 +01001319 if (end_subcmd - arg == 5 && STRNCMP(arg, "start", 5) == 0)
Bram Moolenaarf86f26c2010-02-03 15:14:22 +01001320 {
1321 xp->xp_context = EXPAND_FILES;
1322 xp->xp_pattern = skipwhite(end_subcmd);
1323 return;
1324 }
1325
1326 /* TODO: expand function names after "func" */
1327 xp->xp_context = EXPAND_NOTHING;
1328}
1329
Bram Moolenaar05159a02005-02-26 23:04:13 +00001330/*
1331 * Dump the profiling info.
1332 */
1333 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001334profile_dump(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001335{
1336 FILE *fd;
1337
1338 if (profile_fname != NULL)
1339 {
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +00001340 fd = mch_fopen((char *)profile_fname, "w");
Bram Moolenaar05159a02005-02-26 23:04:13 +00001341 if (fd == NULL)
1342 EMSG2(_(e_notopen), profile_fname);
1343 else
1344 {
Bram Moolenaar05159a02005-02-26 23:04:13 +00001345 script_dump_profile(fd);
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001346 func_dump_profile(fd);
Bram Moolenaar05159a02005-02-26 23:04:13 +00001347 fclose(fd);
1348 }
1349 }
1350}
1351
1352/*
1353 * Start profiling script "fp".
1354 */
1355 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001356script_do_profile(scriptitem_T *si)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001357{
1358 si->sn_pr_count = 0;
1359 profile_zero(&si->sn_pr_total);
1360 profile_zero(&si->sn_pr_self);
1361
1362 ga_init2(&si->sn_prl_ga, sizeof(sn_prl_T), 100);
1363 si->sn_prl_idx = -1;
1364 si->sn_prof_on = TRUE;
1365 si->sn_pr_nest = 0;
1366}
1367
1368/*
1369 * save time when starting to invoke another script or function.
1370 */
1371 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001372script_prof_save(
1373 proftime_T *tm) /* place to store wait time */
Bram Moolenaar05159a02005-02-26 23:04:13 +00001374{
1375 scriptitem_T *si;
1376
1377 if (current_SID > 0 && current_SID <= script_items.ga_len)
1378 {
1379 si = &SCRIPT_ITEM(current_SID);
1380 if (si->sn_prof_on && si->sn_pr_nest++ == 0)
1381 profile_start(&si->sn_pr_child);
1382 }
1383 profile_get_wait(tm);
1384}
1385
1386/*
1387 * Count time spent in children after invoking another script or function.
1388 */
1389 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001390script_prof_restore(proftime_T *tm)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001391{
1392 scriptitem_T *si;
1393
1394 if (current_SID > 0 && current_SID <= script_items.ga_len)
1395 {
1396 si = &SCRIPT_ITEM(current_SID);
1397 if (si->sn_prof_on && --si->sn_pr_nest == 0)
1398 {
1399 profile_end(&si->sn_pr_child);
1400 profile_sub_wait(tm, &si->sn_pr_child); /* don't count wait time */
1401 profile_add(&si->sn_pr_children, &si->sn_pr_child);
1402 profile_add(&si->sn_prl_children, &si->sn_pr_child);
1403 }
1404 }
1405}
1406
1407static proftime_T inchar_time;
1408
1409/*
1410 * Called when starting to wait for the user to type a character.
1411 */
1412 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001413prof_inchar_enter(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001414{
1415 profile_start(&inchar_time);
1416}
1417
1418/*
1419 * Called when finished waiting for the user to type a character.
1420 */
1421 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001422prof_inchar_exit(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001423{
1424 profile_end(&inchar_time);
1425 profile_add(&prof_wait_time, &inchar_time);
1426}
1427
1428/*
1429 * Dump the profiling results for all scripts in file "fd".
1430 */
1431 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001432script_dump_profile(FILE *fd)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001433{
1434 int id;
1435 scriptitem_T *si;
1436 int i;
1437 FILE *sfd;
1438 sn_prl_T *pp;
1439
1440 for (id = 1; id <= script_items.ga_len; ++id)
1441 {
1442 si = &SCRIPT_ITEM(id);
1443 if (si->sn_prof_on)
1444 {
1445 fprintf(fd, "SCRIPT %s\n", si->sn_name);
1446 if (si->sn_pr_count == 1)
1447 fprintf(fd, "Sourced 1 time\n");
1448 else
1449 fprintf(fd, "Sourced %d times\n", si->sn_pr_count);
1450 fprintf(fd, "Total time: %s\n", profile_msg(&si->sn_pr_total));
1451 fprintf(fd, " Self time: %s\n", profile_msg(&si->sn_pr_self));
1452 fprintf(fd, "\n");
1453 fprintf(fd, "count total (s) self (s)\n");
1454
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +00001455 sfd = mch_fopen((char *)si->sn_name, "r");
Bram Moolenaar05159a02005-02-26 23:04:13 +00001456 if (sfd == NULL)
1457 fprintf(fd, "Cannot open file!\n");
1458 else
1459 {
1460 for (i = 0; i < si->sn_prl_ga.ga_len; ++i)
1461 {
1462 if (vim_fgets(IObuff, IOSIZE, sfd))
1463 break;
1464 pp = &PRL_ITEM(si, i);
1465 if (pp->snp_count > 0)
1466 {
1467 fprintf(fd, "%5d ", pp->snp_count);
1468 if (profile_equal(&pp->sn_prl_total, &pp->sn_prl_self))
1469 fprintf(fd, " ");
1470 else
1471 fprintf(fd, "%s ", profile_msg(&pp->sn_prl_total));
1472 fprintf(fd, "%s ", profile_msg(&pp->sn_prl_self));
1473 }
1474 else
1475 fprintf(fd, " ");
1476 fprintf(fd, "%s", IObuff);
1477 }
1478 fclose(sfd);
1479 }
1480 fprintf(fd, "\n");
1481 }
1482 }
1483}
1484
1485/*
1486 * Return TRUE when a function defined in the current script should be
1487 * profiled.
1488 */
1489 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001490prof_def_func(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00001491{
Bram Moolenaar53180ce2005-07-05 21:48:14 +00001492 if (current_SID > 0)
1493 return SCRIPT_ITEM(current_SID).sn_pr_force;
1494 return FALSE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00001495}
1496
1497# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001498#endif
1499
1500/*
1501 * If 'autowrite' option set, try to write the file.
1502 * Careful: autocommands may make "buf" invalid!
1503 *
1504 * return FAIL for failure, OK otherwise
1505 */
1506 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001507autowrite(buf_T *buf, int forceit)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001508{
Bram Moolenaar373154b2007-02-13 05:19:30 +00001509 int r;
1510
Bram Moolenaar071d4272004-06-13 20:20:40 +00001511 if (!(p_aw || p_awa) || !p_write
1512#ifdef FEAT_QUICKFIX
Bram Moolenaar373154b2007-02-13 05:19:30 +00001513 /* never autowrite a "nofile" or "nowrite" buffer */
1514 || bt_dontwrite(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001515#endif
Bram Moolenaar373154b2007-02-13 05:19:30 +00001516 || (!forceit && buf->b_p_ro) || buf->b_ffname == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001517 return FAIL;
Bram Moolenaar373154b2007-02-13 05:19:30 +00001518 r = buf_write_all(buf, forceit);
1519
1520 /* Writing may succeed but the buffer still changed, e.g., when there is a
1521 * conversion error. We do want to return FAIL then. */
1522 if (buf_valid(buf) && bufIsChanged(buf))
1523 r = FAIL;
1524 return r;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001525}
1526
1527/*
1528 * flush all buffers, except the ones that are readonly
1529 */
1530 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001531autowrite_all(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001532{
1533 buf_T *buf;
1534
1535 if (!(p_aw || p_awa) || !p_write)
1536 return;
1537 for (buf = firstbuf; buf; buf = buf->b_next)
1538 if (bufIsChanged(buf) && !buf->b_p_ro)
1539 {
1540 (void)buf_write_all(buf, FALSE);
1541#ifdef FEAT_AUTOCMD
1542 /* an autocommand may have deleted the buffer */
1543 if (!buf_valid(buf))
1544 buf = firstbuf;
1545#endif
1546 }
1547}
1548
1549/*
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001550 * Return TRUE if buffer was changed and cannot be abandoned.
1551 * For flags use the CCGD_ values.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001552 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001553 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001554check_changed(buf_T *buf, int flags)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001555{
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001556 int forceit = (flags & CCGD_FORCEIT);
1557
Bram Moolenaar071d4272004-06-13 20:20:40 +00001558 if ( !forceit
1559 && bufIsChanged(buf)
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001560 && ((flags & CCGD_MULTWIN) || buf->b_nwindows <= 1)
1561 && (!(flags & CCGD_AW) || autowrite(buf, forceit) == FAIL))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001562 {
1563#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG)
1564 if ((p_confirm || cmdmod.confirm) && p_write)
1565 {
1566 buf_T *buf2;
1567 int count = 0;
1568
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001569 if (flags & CCGD_ALLBUF)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001570 for (buf2 = firstbuf; buf2 != NULL; buf2 = buf2->b_next)
1571 if (bufIsChanged(buf2)
1572 && (buf2->b_ffname != NULL
1573# ifdef FEAT_BROWSE
1574 || cmdmod.browse
1575# endif
1576 ))
1577 ++count;
1578# ifdef FEAT_AUTOCMD
1579 if (!buf_valid(buf))
1580 /* Autocommand deleted buffer, oops! It's not changed now. */
1581 return FALSE;
1582# endif
1583 dialog_changed(buf, count > 1);
1584# ifdef FEAT_AUTOCMD
1585 if (!buf_valid(buf))
1586 /* Autocommand deleted buffer, oops! It's not changed now. */
1587 return FALSE;
1588# endif
1589 return bufIsChanged(buf);
1590 }
1591#endif
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001592 if (flags & CCGD_EXCMD)
1593 EMSG(_(e_nowrtmsg));
1594 else
1595 EMSG(_(e_nowrtmsg_nobang));
Bram Moolenaar071d4272004-06-13 20:20:40 +00001596 return TRUE;
1597 }
1598 return FALSE;
1599}
1600
1601#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG) || defined(PROTO)
1602
1603#if defined(FEAT_BROWSE) || defined(PROTO)
1604/*
1605 * When wanting to write a file without a file name, ask the user for a name.
1606 */
1607 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001608browse_save_fname(buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001609{
1610 if (buf->b_fname == NULL)
1611 {
1612 char_u *fname;
1613
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00001614 fname = do_browse(BROWSE_SAVE, (char_u *)_("Save As"),
1615 NULL, NULL, NULL, NULL, buf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001616 if (fname != NULL)
1617 {
1618 if (setfname(buf, fname, NULL, TRUE) == OK)
1619 buf->b_flags |= BF_NOTEDITED;
1620 vim_free(fname);
1621 }
1622 }
1623}
1624#endif
1625
1626/*
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02001627 * Ask the user what to do when abandoning a changed buffer.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001628 * Must check 'write' option first!
1629 */
1630 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001631dialog_changed(
1632 buf_T *buf,
1633 int checkall) /* may abandon all changed buffers */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001634{
Bram Moolenaard9462e32011-04-11 21:35:11 +02001635 char_u buff[DIALOG_MSG_SIZE];
Bram Moolenaar071d4272004-06-13 20:20:40 +00001636 int ret;
1637 buf_T *buf2;
Bram Moolenaar8218f602012-04-25 17:32:18 +02001638 exarg_T ea;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001639
Bram Moolenaar82cf9b62005-06-07 21:09:25 +00001640 dialog_msg(buff, _("Save changes to \"%s\"?"),
Bram Moolenaar071d4272004-06-13 20:20:40 +00001641 (buf->b_fname != NULL) ?
1642 buf->b_fname : (char_u *)_("Untitled"));
1643 if (checkall)
1644 ret = vim_dialog_yesnoallcancel(VIM_QUESTION, NULL, buff, 1);
1645 else
1646 ret = vim_dialog_yesnocancel(VIM_QUESTION, NULL, buff, 1);
1647
Bram Moolenaar8218f602012-04-25 17:32:18 +02001648 /* Init ea pseudo-structure, this is needed for the check_overwrite()
1649 * function. */
1650 ea.append = ea.forceit = FALSE;
1651
Bram Moolenaar071d4272004-06-13 20:20:40 +00001652 if (ret == VIM_YES)
1653 {
1654#ifdef FEAT_BROWSE
1655 /* May get file name, when there is none */
1656 browse_save_fname(buf);
1657#endif
Bram Moolenaar8218f602012-04-25 17:32:18 +02001658 if (buf->b_fname != NULL && check_overwrite(&ea, buf,
1659 buf->b_fname, buf->b_ffname, FALSE) == OK)
1660 /* didn't hit Cancel */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001661 (void)buf_write_all(buf, FALSE);
1662 }
1663 else if (ret == VIM_NO)
1664 {
1665 unchanged(buf, TRUE);
1666 }
1667 else if (ret == VIM_ALL)
1668 {
1669 /*
1670 * Write all modified files that can be written.
1671 * Skip readonly buffers, these need to be confirmed
1672 * individually.
1673 */
1674 for (buf2 = firstbuf; buf2 != NULL; buf2 = buf2->b_next)
1675 {
1676 if (bufIsChanged(buf2)
1677 && (buf2->b_ffname != NULL
1678#ifdef FEAT_BROWSE
1679 || cmdmod.browse
1680#endif
1681 )
1682 && !buf2->b_p_ro)
1683 {
1684#ifdef FEAT_BROWSE
1685 /* May get file name, when there is none */
1686 browse_save_fname(buf2);
1687#endif
Bram Moolenaar8218f602012-04-25 17:32:18 +02001688 if (buf2->b_fname != NULL && check_overwrite(&ea, buf2,
1689 buf2->b_fname, buf2->b_ffname, FALSE) == OK)
1690 /* didn't hit Cancel */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001691 (void)buf_write_all(buf2, FALSE);
1692#ifdef FEAT_AUTOCMD
1693 /* an autocommand may have deleted the buffer */
1694 if (!buf_valid(buf2))
1695 buf2 = firstbuf;
1696#endif
1697 }
1698 }
1699 }
1700 else if (ret == VIM_DISCARDALL)
1701 {
1702 /*
1703 * mark all buffers as unchanged
1704 */
1705 for (buf2 = firstbuf; buf2 != NULL; buf2 = buf2->b_next)
1706 unchanged(buf2, TRUE);
1707 }
1708}
1709#endif
1710
1711/*
1712 * Return TRUE if the buffer "buf" can be abandoned, either by making it
1713 * hidden, autowriting it or unloading it.
1714 */
1715 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001716can_abandon(buf_T *buf, int forceit)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001717{
1718 return ( P_HID(buf)
1719 || !bufIsChanged(buf)
1720 || buf->b_nwindows > 1
1721 || autowrite(buf, forceit) == OK
1722 || forceit);
1723}
1724
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01001725static void add_bufnum(int *bufnrs, int *bufnump, int nr);
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001726
1727/*
1728 * Add a buffer number to "bufnrs", unless it's already there.
1729 */
1730 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001731add_bufnum(int *bufnrs, int *bufnump, int nr)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001732{
1733 int i;
1734
1735 for (i = 0; i < *bufnump; ++i)
1736 if (bufnrs[i] == nr)
1737 return;
1738 bufnrs[*bufnump] = nr;
1739 *bufnump = *bufnump + 1;
1740}
1741
Bram Moolenaar071d4272004-06-13 20:20:40 +00001742/*
1743 * Return TRUE if any buffer was changed and cannot be abandoned.
1744 * That changed buffer becomes the current buffer.
Bram Moolenaar027387f2016-01-02 22:25:52 +01001745 * When "unload" is true the current buffer is unloaded instead of making it
1746 * hidden. This is used for ":q!".
Bram Moolenaar071d4272004-06-13 20:20:40 +00001747 */
1748 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001749check_changed_any(
1750 int hidden, /* Only check hidden buffers */
1751 int unload)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001752{
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001753 int ret = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001754 buf_T *buf;
1755 int save;
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001756 int i;
1757 int bufnum = 0;
1758 int bufcount = 0;
1759 int *bufnrs;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001760#ifdef FEAT_WINDOWS
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001761 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001762 win_T *wp;
1763#endif
1764
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001765 for (buf = firstbuf; buf != NULL; buf = buf->b_next)
1766 ++bufcount;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001767
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001768 if (bufcount == 0)
1769 return FALSE;
1770
1771 bufnrs = (int *)alloc(sizeof(int) * bufcount);
1772 if (bufnrs == NULL)
1773 return FALSE;
1774
1775 /* curbuf */
1776 bufnrs[bufnum++] = curbuf->b_fnum;
1777#ifdef FEAT_WINDOWS
1778 /* buf in curtab */
1779 FOR_ALL_WINDOWS(wp)
1780 if (wp->w_buffer != curbuf)
1781 add_bufnum(bufnrs, &bufnum, wp->w_buffer->b_fnum);
1782
1783 /* buf in other tab */
1784 for (tp = first_tabpage; tp != NULL; tp = tp->tp_next)
1785 if (tp != curtab)
1786 for (wp = tp->tp_firstwin; wp != NULL; wp = wp->w_next)
1787 add_bufnum(bufnrs, &bufnum, wp->w_buffer->b_fnum);
1788#endif
1789 /* any other buf */
1790 for (buf = firstbuf; buf != NULL; buf = buf->b_next)
1791 add_bufnum(bufnrs, &bufnum, buf->b_fnum);
1792
1793 for (i = 0; i < bufnum; ++i)
1794 {
1795 buf = buflist_findnr(bufnrs[i]);
1796 if (buf == NULL)
1797 continue;
1798 if ((!hidden || buf->b_nwindows == 0) && bufIsChanged(buf))
1799 {
1800 /* Try auto-writing the buffer. If this fails but the buffer no
1801 * longer exists it's not changed, that's OK. */
Bram Moolenaar45d3b142013-11-09 03:31:51 +01001802 if (check_changed(buf, (p_awa ? CCGD_AW : 0)
1803 | CCGD_MULTWIN
1804 | CCGD_ALLBUF) && buf_valid(buf))
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001805 break; /* didn't save - still changes */
1806 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001807 }
1808
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001809 if (i >= bufnum)
1810 goto theend;
1811
1812 ret = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001813 exiting = FALSE;
1814#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG)
1815 /*
1816 * When ":confirm" used, don't give an error message.
1817 */
1818 if (!(p_confirm || cmdmod.confirm))
1819#endif
1820 {
1821 /* There must be a wait_return for this message, do_buffer()
1822 * may cause a redraw. But wait_return() is a no-op when vgetc()
1823 * is busy (Quit used from window menu), then make sure we don't
1824 * cause a scroll up. */
Bram Moolenaar61660ea2006-04-07 21:40:07 +00001825 if (vgetc_busy > 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001826 {
1827 msg_row = cmdline_row;
1828 msg_col = 0;
1829 msg_didout = FALSE;
1830 }
1831 if (EMSG2(_("E162: No write since last change for buffer \"%s\""),
Bram Moolenaare1704ba2012-10-03 18:25:00 +02001832 buf_spname(buf) != NULL ? buf_spname(buf) : buf->b_fname))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001833 {
1834 save = no_wait_return;
1835 no_wait_return = FALSE;
1836 wait_return(FALSE);
1837 no_wait_return = save;
1838 }
1839 }
1840
1841#ifdef FEAT_WINDOWS
1842 /* Try to find a window that contains the buffer. */
1843 if (buf != curbuf)
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001844 FOR_ALL_TAB_WINDOWS(tp, wp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001845 if (wp->w_buffer == buf)
1846 {
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001847 goto_tabpage_win(tp, wp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001848# ifdef FEAT_AUTOCMD
1849 /* Paranoia: did autocms wipe out the buffer with changes? */
1850 if (!buf_valid(buf))
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001851 {
1852 goto theend;
1853 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001854# endif
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001855 goto buf_found;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001856 }
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001857buf_found:
Bram Moolenaar071d4272004-06-13 20:20:40 +00001858#endif
1859
1860 /* Open the changed buffer in the current window. */
1861 if (buf != curbuf)
Bram Moolenaar027387f2016-01-02 22:25:52 +01001862 set_curbuf(buf, unload ? DOBUF_UNLOAD : DOBUF_GOTO);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001863
Bram Moolenaar970a1b82012-03-23 18:39:18 +01001864theend:
1865 vim_free(bufnrs);
1866 return ret;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001867}
1868
1869/*
1870 * return FAIL if there is no file name, OK if there is one
1871 * give error message for FAIL
1872 */
1873 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001874check_fname(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001875{
1876 if (curbuf->b_ffname == NULL)
1877 {
1878 EMSG(_(e_noname));
1879 return FAIL;
1880 }
1881 return OK;
1882}
1883
1884/*
1885 * flush the contents of a buffer, unless it has no file name
1886 *
1887 * return FAIL for failure, OK otherwise
1888 */
1889 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001890buf_write_all(buf_T *buf, int forceit)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001891{
1892 int retval;
1893#ifdef FEAT_AUTOCMD
1894 buf_T *old_curbuf = curbuf;
1895#endif
1896
1897 retval = (buf_write(buf, buf->b_ffname, buf->b_fname,
1898 (linenr_T)1, buf->b_ml.ml_line_count, NULL,
1899 FALSE, forceit, TRUE, FALSE));
1900#ifdef FEAT_AUTOCMD
1901 if (curbuf != old_curbuf)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001902 {
1903 msg_source(hl_attr(HLF_W));
Bram Moolenaar071d4272004-06-13 20:20:40 +00001904 MSG(_("Warning: Entered other buffer unexpectedly (check autocommands)"));
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00001905 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001906#endif
1907 return retval;
1908}
1909
1910/*
1911 * Code to handle the argument list.
1912 */
1913
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01001914static char_u *do_one_arg(char_u *str);
1915static int do_arglist(char_u *str, int what, int after);
1916static void alist_check_arg_idx(void);
1917static int editing_arg_idx(win_T *win);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001918#ifdef FEAT_LISTCMDS
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01001919static int alist_add_list(int count, char_u **files, int after);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001920#endif
1921#define AL_SET 1
1922#define AL_ADD 2
1923#define AL_DEL 3
1924
Bram Moolenaar071d4272004-06-13 20:20:40 +00001925/*
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001926 * Isolate one argument, taking backticks.
1927 * Changes the argument in-place, puts a NUL after it. Backticks remain.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001928 * Return a pointer to the start of the next argument.
1929 */
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001930 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001931do_one_arg(char_u *str)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001932{
1933 char_u *p;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001934 int inbacktick;
1935
Bram Moolenaar071d4272004-06-13 20:20:40 +00001936 inbacktick = FALSE;
1937 for (p = str; *str; ++str)
1938 {
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001939 /* When the backslash is used for escaping the special meaning of a
1940 * character we need to keep it until wildcard expansion. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001941 if (rem_backslash(str))
1942 {
1943 *p++ = *str++;
1944 *p++ = *str;
1945 }
1946 else
1947 {
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001948 /* An item ends at a space not in backticks */
1949 if (!inbacktick && vim_isspace(*str))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001950 break;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001951 if (*str == '`')
Bram Moolenaar071d4272004-06-13 20:20:40 +00001952 inbacktick ^= TRUE;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001953 *p++ = *str;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001954 }
1955 }
1956 str = skipwhite(str);
1957 *p = NUL;
1958
1959 return str;
1960}
1961
Bram Moolenaar86b68352004-12-27 21:59:20 +00001962/*
1963 * Separate the arguments in "str" and return a list of pointers in the
1964 * growarray "gap".
1965 */
1966 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001967get_arglist(garray_T *gap, char_u *str)
Bram Moolenaar86b68352004-12-27 21:59:20 +00001968{
1969 ga_init2(gap, (int)sizeof(char_u *), 20);
1970 while (*str != NUL)
1971 {
1972 if (ga_grow(gap, 1) == FAIL)
1973 {
1974 ga_clear(gap);
1975 return FAIL;
1976 }
1977 ((char_u **)gap->ga_data)[gap->ga_len++] = str;
1978
1979 /* Isolate one argument, change it in-place, put a NUL after it. */
1980 str = do_one_arg(str);
1981 }
1982 return OK;
1983}
1984
Bram Moolenaar7df351e2006-01-23 22:30:28 +00001985#if defined(FEAT_QUICKFIX) || defined(FEAT_SYN_HL) || defined(PROTO)
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00001986/*
1987 * Parse a list of arguments (file names), expand them and return in
Bram Moolenaar8f5c6f02012-06-29 12:57:06 +02001988 * "fnames[fcountp]". When "wig" is TRUE, removes files matching 'wildignore'.
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00001989 * Return FAIL or OK.
1990 */
1991 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01001992get_arglist_exp(
1993 char_u *str,
1994 int *fcountp,
1995 char_u ***fnamesp,
1996 int wig)
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00001997{
1998 garray_T ga;
1999 int i;
2000
2001 if (get_arglist(&ga, str) == FAIL)
2002 return FAIL;
Bram Moolenaar8f5c6f02012-06-29 12:57:06 +02002003 if (wig == TRUE)
2004 i = expand_wildcards(ga.ga_len, (char_u **)ga.ga_data,
2005 fcountp, fnamesp, EW_FILE|EW_NOTFOUND);
2006 else
2007 i = gen_expand_wildcards(ga.ga_len, (char_u **)ga.ga_data,
2008 fcountp, fnamesp, EW_FILE|EW_NOTFOUND);
2009
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00002010 ga_clear(&ga);
2011 return i;
2012}
2013#endif
2014
Bram Moolenaar071d4272004-06-13 20:20:40 +00002015#if defined(FEAT_GUI) || defined(FEAT_CLIENTSERVER) || defined(PROTO)
2016/*
2017 * Redefine the argument list.
2018 */
2019 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002020set_arglist(char_u *str)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002021{
2022 do_arglist(str, AL_SET, 0);
2023}
2024#endif
2025
2026/*
2027 * "what" == AL_SET: Redefine the argument list to 'str'.
2028 * "what" == AL_ADD: add files in 'str' to the argument list after "after".
2029 * "what" == AL_DEL: remove files in 'str' from the argument list.
2030 *
2031 * Return FAIL for failure, OK otherwise.
2032 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002033 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002034do_arglist(
2035 char_u *str,
Bram Moolenaarf1d25012016-03-03 12:22:53 +01002036 int what,
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002037 int after UNUSED) /* 0 means before first one */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002038{
2039 garray_T new_ga;
2040 int exp_count;
2041 char_u **exp_files;
2042 int i;
2043#ifdef FEAT_LISTCMDS
2044 char_u *p;
2045 int match;
2046#endif
2047
2048 /*
Bram Moolenaar2faa29f2016-01-23 23:02:34 +01002049 * Set default argument for ":argadd" command.
2050 */
2051 if (what == AL_ADD && *str == NUL)
2052 {
2053 if (curbuf->b_ffname == NULL)
2054 return FAIL;
2055 str = curbuf->b_fname;
2056 }
2057
2058 /*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002059 * Collect all file name arguments in "new_ga".
2060 */
Bram Moolenaar86b68352004-12-27 21:59:20 +00002061 if (get_arglist(&new_ga, str) == FAIL)
2062 return FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002063
2064#ifdef FEAT_LISTCMDS
2065 if (what == AL_DEL)
2066 {
2067 regmatch_T regmatch;
2068 int didone;
2069
2070 /*
2071 * Delete the items: use each item as a regexp and find a match in the
2072 * argument list.
2073 */
Bram Moolenaar71afbfe2013-03-19 16:49:16 +01002074 regmatch.rm_ic = p_fic; /* ignore case when 'fileignorecase' is set */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002075 for (i = 0; i < new_ga.ga_len && !got_int; ++i)
2076 {
2077 p = ((char_u **)new_ga.ga_data)[i];
2078 p = file_pat_to_reg_pat(p, NULL, NULL, FALSE);
2079 if (p == NULL)
2080 break;
2081 regmatch.regprog = vim_regcomp(p, p_magic ? RE_MAGIC : 0);
2082 if (regmatch.regprog == NULL)
2083 {
2084 vim_free(p);
2085 break;
2086 }
2087
2088 didone = FALSE;
2089 for (match = 0; match < ARGCOUNT; ++match)
2090 if (vim_regexec(&regmatch, alist_name(&ARGLIST[match]),
2091 (colnr_T)0))
2092 {
2093 didone = TRUE;
2094 vim_free(ARGLIST[match].ae_fname);
2095 mch_memmove(ARGLIST + match, ARGLIST + match + 1,
2096 (ARGCOUNT - match - 1) * sizeof(aentry_T));
2097 --ALIST(curwin)->al_ga.ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002098 if (curwin->w_arg_idx > match)
2099 --curwin->w_arg_idx;
2100 --match;
2101 }
2102
Bram Moolenaar473de612013-06-08 18:19:48 +02002103 vim_regfree(regmatch.regprog);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002104 vim_free(p);
2105 if (!didone)
2106 EMSG2(_(e_nomatch2), ((char_u **)new_ga.ga_data)[i]);
2107 }
2108 ga_clear(&new_ga);
2109 }
2110 else
2111#endif
2112 {
2113 i = expand_wildcards(new_ga.ga_len, (char_u **)new_ga.ga_data,
2114 &exp_count, &exp_files, EW_DIR|EW_FILE|EW_ADDSLASH|EW_NOTFOUND);
2115 ga_clear(&new_ga);
Bram Moolenaar2db5c3b2016-01-16 22:49:34 +01002116 if (i == FAIL || exp_count == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002117 {
2118 EMSG(_(e_nomatch));
2119 return FAIL;
2120 }
2121
2122#ifdef FEAT_LISTCMDS
2123 if (what == AL_ADD)
2124 {
2125 (void)alist_add_list(exp_count, exp_files, after);
2126 vim_free(exp_files);
2127 }
2128 else /* what == AL_SET */
2129#endif
Bram Moolenaar86b68352004-12-27 21:59:20 +00002130 alist_set(ALIST(curwin), exp_count, exp_files, FALSE, NULL, 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002131 }
2132
2133 alist_check_arg_idx();
2134
2135 return OK;
2136}
2137
2138/*
2139 * Check the validity of the arg_idx for each other window.
2140 */
2141 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002142alist_check_arg_idx(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002143{
2144#ifdef FEAT_WINDOWS
2145 win_T *win;
Bram Moolenaarf740b292006-02-16 22:11:02 +00002146 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002147
Bram Moolenaarf740b292006-02-16 22:11:02 +00002148 FOR_ALL_TAB_WINDOWS(tp, win)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002149 if (win->w_alist == curwin->w_alist)
2150 check_arg_idx(win);
2151#else
2152 check_arg_idx(curwin);
2153#endif
2154}
2155
2156/*
Bram Moolenaarb8ff1fb2012-02-04 21:59:01 +01002157 * Return TRUE if window "win" is editing the file at the current argument
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002158 * index.
2159 */
2160 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002161editing_arg_idx(win_T *win)
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002162{
2163 return !(win->w_arg_idx >= WARGCOUNT(win)
2164 || (win->w_buffer->b_fnum
2165 != WARGLIST(win)[win->w_arg_idx].ae_fnum
2166 && (win->w_buffer->b_ffname == NULL
2167 || !(fullpathcmp(
2168 alist_name(&WARGLIST(win)[win->w_arg_idx]),
2169 win->w_buffer->b_ffname, TRUE) & FPC_SAME))));
2170}
2171
2172/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00002173 * Check if window "win" is editing the w_arg_idx file in its argument list.
2174 */
2175 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002176check_arg_idx(win_T *win)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002177{
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002178 if (WARGCOUNT(win) > 1 && !editing_arg_idx(win))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002179 {
2180 /* We are not editing the current entry in the argument list.
2181 * Set "arg_had_last" if we are editing the last one. */
2182 win->w_arg_idx_invalid = TRUE;
2183 if (win->w_arg_idx != WARGCOUNT(win) - 1
2184 && arg_had_last == FALSE
2185#ifdef FEAT_WINDOWS
2186 && ALIST(win) == &global_alist
2187#endif
2188 && GARGCOUNT > 0
2189 && win->w_arg_idx < GARGCOUNT
2190 && (win->w_buffer->b_fnum == GARGLIST[GARGCOUNT - 1].ae_fnum
2191 || (win->w_buffer->b_ffname != NULL
2192 && (fullpathcmp(alist_name(&GARGLIST[GARGCOUNT - 1]),
2193 win->w_buffer->b_ffname, TRUE) & FPC_SAME))))
2194 arg_had_last = TRUE;
2195 }
2196 else
2197 {
2198 /* We are editing the current entry in the argument list.
2199 * Set "arg_had_last" if it's also the last one */
2200 win->w_arg_idx_invalid = FALSE;
2201 if (win->w_arg_idx == WARGCOUNT(win) - 1
2202#ifdef FEAT_WINDOWS
2203 && win->w_alist == &global_alist
2204#endif
2205 )
2206 arg_had_last = TRUE;
2207 }
2208}
2209
2210/*
2211 * ":args", ":argslocal" and ":argsglobal".
2212 */
2213 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002214ex_args(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002215{
2216 int i;
2217
2218 if (eap->cmdidx != CMD_args)
2219 {
2220#if defined(FEAT_WINDOWS) && defined(FEAT_LISTCMDS)
2221 alist_unlink(ALIST(curwin));
2222 if (eap->cmdidx == CMD_argglobal)
2223 ALIST(curwin) = &global_alist;
2224 else /* eap->cmdidx == CMD_arglocal */
2225 alist_new();
2226#else
2227 ex_ni(eap);
2228 return;
2229#endif
2230 }
2231
2232 if (!ends_excmd(*eap->arg))
2233 {
2234 /*
2235 * ":args file ..": define new argument list, handle like ":next"
2236 * Also for ":argslocal file .." and ":argsglobal file ..".
2237 */
2238 ex_next(eap);
2239 }
2240 else
2241#if defined(FEAT_WINDOWS) && defined(FEAT_LISTCMDS)
2242 if (eap->cmdidx == CMD_args)
2243#endif
2244 {
2245 /*
2246 * ":args": list arguments.
2247 */
2248 if (ARGCOUNT > 0)
2249 {
2250 /* Overwrite the command, for a short list there is no scrolling
2251 * required and no wait_return(). */
2252 gotocmdline(TRUE);
2253 for (i = 0; i < ARGCOUNT; ++i)
2254 {
2255 if (i == curwin->w_arg_idx)
2256 msg_putchar('[');
2257 msg_outtrans(alist_name(&ARGLIST[i]));
2258 if (i == curwin->w_arg_idx)
2259 msg_putchar(']');
2260 msg_putchar(' ');
2261 }
2262 }
2263 }
2264#if defined(FEAT_WINDOWS) && defined(FEAT_LISTCMDS)
2265 else if (eap->cmdidx == CMD_arglocal)
2266 {
2267 garray_T *gap = &curwin->w_alist->al_ga;
2268
2269 /*
2270 * ":argslocal": make a local copy of the global argument list.
2271 */
2272 if (ga_grow(gap, GARGCOUNT) == OK)
2273 for (i = 0; i < GARGCOUNT; ++i)
2274 if (GARGLIST[i].ae_fname != NULL)
2275 {
2276 AARGLIST(curwin->w_alist)[gap->ga_len].ae_fname =
2277 vim_strsave(GARGLIST[i].ae_fname);
2278 AARGLIST(curwin->w_alist)[gap->ga_len].ae_fnum =
2279 GARGLIST[i].ae_fnum;
2280 ++gap->ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002281 }
2282 }
2283#endif
2284}
2285
2286/*
2287 * ":previous", ":sprevious", ":Next" and ":sNext".
2288 */
2289 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002290ex_previous(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002291{
2292 /* If past the last one already, go to the last one. */
2293 if (curwin->w_arg_idx - (int)eap->line2 >= ARGCOUNT)
2294 do_argfile(eap, ARGCOUNT - 1);
2295 else
2296 do_argfile(eap, curwin->w_arg_idx - (int)eap->line2);
2297}
2298
2299/*
2300 * ":rewind", ":first", ":sfirst" and ":srewind".
2301 */
2302 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002303ex_rewind(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002304{
2305 do_argfile(eap, 0);
2306}
2307
2308/*
2309 * ":last" and ":slast".
2310 */
2311 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002312ex_last(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002313{
2314 do_argfile(eap, ARGCOUNT - 1);
2315}
2316
2317/*
2318 * ":argument" and ":sargument".
2319 */
2320 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002321ex_argument(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002322{
2323 int i;
2324
2325 if (eap->addr_count > 0)
2326 i = eap->line2 - 1;
2327 else
2328 i = curwin->w_arg_idx;
2329 do_argfile(eap, i);
2330}
2331
2332/*
2333 * Edit file "argn" of the argument lists.
2334 */
2335 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002336do_argfile(exarg_T *eap, int argn)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002337{
2338 int other;
2339 char_u *p;
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002340 int old_arg_idx = curwin->w_arg_idx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002341
2342 if (argn < 0 || argn >= ARGCOUNT)
2343 {
2344 if (ARGCOUNT <= 1)
2345 EMSG(_("E163: There is only one file to edit"));
2346 else if (argn < 0)
2347 EMSG(_("E164: Cannot go before first file"));
2348 else
2349 EMSG(_("E165: Cannot go beyond last file"));
2350 }
2351 else
2352 {
2353 setpcmark();
2354#ifdef FEAT_GUI
2355 need_mouse_correct = TRUE;
2356#endif
2357
2358#ifdef FEAT_WINDOWS
Bram Moolenaardf1bdc92006-02-23 21:32:16 +00002359 /* split window or create new tab page first */
2360 if (*eap->cmd == 's' || cmdmod.tab != 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002361 {
2362 if (win_split(0, 0) == FAIL)
2363 return;
Bram Moolenaar3368ea22010-09-21 16:56:35 +02002364 RESET_BINDING(curwin);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002365 }
2366 else
2367#endif
2368 {
2369 /*
2370 * if 'hidden' set, only check for changed file when re-editing
2371 * the same buffer
2372 */
2373 other = TRUE;
2374 if (P_HID(curbuf))
2375 {
2376 p = fix_fname(alist_name(&ARGLIST[argn]));
2377 other = otherfile(p);
2378 vim_free(p);
2379 }
2380 if ((!P_HID(curbuf) || !other)
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002381 && check_changed(curbuf, CCGD_AW
2382 | (other ? 0 : CCGD_MULTWIN)
2383 | (eap->forceit ? CCGD_FORCEIT : 0)
2384 | CCGD_EXCMD))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002385 return;
2386 }
2387
2388 curwin->w_arg_idx = argn;
2389 if (argn == ARGCOUNT - 1
2390#ifdef FEAT_WINDOWS
2391 && curwin->w_alist == &global_alist
2392#endif
2393 )
2394 arg_had_last = TRUE;
2395
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002396 /* Edit the file; always use the last known line number.
2397 * When it fails (e.g. Abort for already edited file) restore the
2398 * argument index. */
2399 if (do_ecmd(0, alist_name(&ARGLIST[curwin->w_arg_idx]), NULL,
Bram Moolenaar071d4272004-06-13 20:20:40 +00002400 eap, ECMD_LAST,
Bram Moolenaar701f7af2008-11-15 13:12:07 +00002401 (P_HID(curwin->w_buffer) ? ECMD_HIDE : 0)
2402 + (eap->forceit ? ECMD_FORCEIT : 0), curwin) == FAIL)
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002403 curwin->w_arg_idx = old_arg_idx;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002404 /* like Vi: set the mark where the cursor is in the file. */
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00002405 else if (eap->cmdidx != CMD_argdo)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002406 setmark('\'');
2407 }
2408}
2409
2410/*
2411 * ":next", and commands that behave like it.
2412 */
2413 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002414ex_next(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002415{
2416 int i;
2417
2418 /*
2419 * check for changed buffer now, if this fails the argument list is not
2420 * redefined.
2421 */
2422 if ( P_HID(curbuf)
2423 || eap->cmdidx == CMD_snext
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002424 || !check_changed(curbuf, CCGD_AW
2425 | (eap->forceit ? CCGD_FORCEIT : 0)
2426 | CCGD_EXCMD))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002427 {
2428 if (*eap->arg != NUL) /* redefine file list */
2429 {
2430 if (do_arglist(eap->arg, AL_SET, 0) == FAIL)
2431 return;
2432 i = 0;
2433 }
2434 else
2435 i = curwin->w_arg_idx + (int)eap->line2;
2436 do_argfile(eap, i);
2437 }
2438}
2439
Bram Moolenaar0106e3d2016-02-23 18:55:43 +01002440#if defined(FEAT_LISTCMDS) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002441/*
2442 * ":argedit"
2443 */
2444 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002445ex_argedit(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002446{
2447 int fnum;
2448 int i;
2449 char_u *s;
2450
2451 /* Add the argument to the buffer list and get the buffer number. */
2452 fnum = buflist_add(eap->arg, BLN_LISTED);
2453
2454 /* Check if this argument is already in the argument list. */
2455 for (i = 0; i < ARGCOUNT; ++i)
2456 if (ARGLIST[i].ae_fnum == fnum)
2457 break;
2458 if (i == ARGCOUNT)
2459 {
2460 /* Can't find it, add it to the argument list. */
2461 s = vim_strsave(eap->arg);
2462 if (s == NULL)
2463 return;
2464 i = alist_add_list(1, &s,
2465 eap->addr_count > 0 ? (int)eap->line2 : curwin->w_arg_idx + 1);
2466 if (i < 0)
2467 return;
2468 curwin->w_arg_idx = i;
2469 }
2470
2471 alist_check_arg_idx();
2472
2473 /* Edit the argument. */
2474 do_argfile(eap, i);
2475}
2476
2477/*
2478 * ":argadd"
2479 */
2480 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002481ex_argadd(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002482{
2483 do_arglist(eap->arg, AL_ADD,
2484 eap->addr_count > 0 ? (int)eap->line2 : curwin->w_arg_idx + 1);
2485#ifdef FEAT_TITLE
2486 maketitle();
2487#endif
2488}
2489
2490/*
2491 * ":argdelete"
2492 */
2493 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002494ex_argdelete(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002495{
2496 int i;
2497 int n;
2498
2499 if (eap->addr_count > 0)
2500 {
2501 /* ":1,4argdel": Delete all arguments in the range. */
2502 if (eap->line2 > ARGCOUNT)
2503 eap->line2 = ARGCOUNT;
2504 n = eap->line2 - eap->line1 + 1;
2505 if (*eap->arg != NUL || n <= 0)
2506 EMSG(_(e_invarg));
2507 else
2508 {
2509 for (i = eap->line1; i <= eap->line2; ++i)
2510 vim_free(ARGLIST[i - 1].ae_fname);
2511 mch_memmove(ARGLIST + eap->line1 - 1, ARGLIST + eap->line2,
2512 (size_t)((ARGCOUNT - eap->line2) * sizeof(aentry_T)));
2513 ALIST(curwin)->al_ga.ga_len -= n;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002514 if (curwin->w_arg_idx >= eap->line2)
2515 curwin->w_arg_idx -= n;
2516 else if (curwin->w_arg_idx > eap->line1)
2517 curwin->w_arg_idx = eap->line1;
Bram Moolenaar72defda2016-01-17 18:04:33 +01002518 if (ARGCOUNT == 0)
2519 curwin->w_arg_idx = 0;
2520 else if (curwin->w_arg_idx >= ARGCOUNT)
2521 curwin->w_arg_idx = ARGCOUNT - 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002522 }
2523 }
2524 else if (*eap->arg == NUL)
2525 EMSG(_(e_argreq));
2526 else
2527 do_arglist(eap->arg, AL_DEL, 0);
2528#ifdef FEAT_TITLE
2529 maketitle();
2530#endif
2531}
2532
2533/*
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002534 * ":argdo", ":windo", ":bufdo", ":tabdo", ":cdo", ":ldo", ":cfdo" and ":lfdo"
Bram Moolenaar071d4272004-06-13 20:20:40 +00002535 */
2536 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002537ex_listdo(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002538{
2539 int i;
2540#ifdef FEAT_WINDOWS
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002541 win_T *wp;
2542 tabpage_T *tp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002543#endif
Bram Moolenaare25bb902015-02-27 20:33:37 +01002544 buf_T *buf = curbuf;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002545 int next_fnum = 0;
2546#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
2547 char_u *save_ei = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002548#endif
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002549 char_u *p_shm_save;
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002550#ifdef FEAT_QUICKFIX
Bram Moolenaared84b762015-09-09 22:35:29 +02002551 int qf_size = 0;
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002552 int qf_idx;
2553#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002554
2555#ifndef FEAT_WINDOWS
2556 if (eap->cmdidx == CMD_windo)
2557 {
2558 ex_ni(eap);
2559 return;
2560 }
2561#endif
2562
Bram Moolenaar0106e3d2016-02-23 18:55:43 +01002563#ifndef FEAT_QUICKFIX
2564 if (eap->cmdidx == CMD_cdo || eap->cmdidx == CMD_ldo ||
2565 eap->cmdidx == CMD_cfdo || eap->cmdidx == CMD_lfdo)
2566 {
2567 ex_ni(eap);
2568 return;
2569 }
2570#endif
2571
Bram Moolenaar071d4272004-06-13 20:20:40 +00002572#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002573 if (eap->cmdidx != CMD_windo && eap->cmdidx != CMD_tabdo)
Bram Moolenaardcaf10e2005-01-21 11:55:25 +00002574 /* Don't do syntax HL autocommands. Skipping the syntax file is a
2575 * great speed improvement. */
2576 save_ei = au_event_disable(",Syntax");
Bram Moolenaar071d4272004-06-13 20:20:40 +00002577#endif
Bram Moolenaar6b1ee342014-08-06 18:17:11 +02002578#ifdef FEAT_CLIPBOARD
2579 start_global_changes();
2580#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002581
2582 if (eap->cmdidx == CMD_windo
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002583 || eap->cmdidx == CMD_tabdo
Bram Moolenaar071d4272004-06-13 20:20:40 +00002584 || P_HID(curbuf)
Bram Moolenaar45d3b142013-11-09 03:31:51 +01002585 || !check_changed(curbuf, CCGD_AW
2586 | (eap->forceit ? CCGD_FORCEIT : 0)
2587 | CCGD_EXCMD))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002588 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00002589 i = 0;
Bram Moolenaara162bc52015-01-07 16:54:21 +01002590 /* start at the eap->line1 argument/window/buffer */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002591#ifdef FEAT_WINDOWS
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002592 wp = firstwin;
2593 tp = first_tabpage;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002594#endif
Bram Moolenaara162bc52015-01-07 16:54:21 +01002595 switch (eap->cmdidx)
2596 {
2597#ifdef FEAT_WINDOWS
2598 case CMD_windo:
2599 for ( ; wp != NULL && i + 1 < eap->line1; wp = wp->w_next)
2600 i++;
2601 break;
2602 case CMD_tabdo:
2603 for( ; tp != NULL && i + 1 < eap->line1; tp = tp->tp_next)
2604 i++;
2605 break;
2606#endif
2607 case CMD_argdo:
2608 i = eap->line1 - 1;
2609 break;
Bram Moolenaara162bc52015-01-07 16:54:21 +01002610 default:
2611 break;
2612 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002613 /* set pcmark now */
2614 if (eap->cmdidx == CMD_bufdo)
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002615 {
Bram Moolenaare25bb902015-02-27 20:33:37 +01002616 /* Advance to the first listed buffer after "eap->line1". */
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002617 for (buf = firstbuf; buf != NULL && (buf->b_fnum < eap->line1
Bram Moolenaare25bb902015-02-27 20:33:37 +01002618 || !buf->b_p_bl); buf = buf->b_next)
2619 if (buf->b_fnum > eap->line2)
2620 {
2621 buf = NULL;
2622 break;
2623 }
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002624 if (buf != NULL)
Bram Moolenaare25bb902015-02-27 20:33:37 +01002625 goto_buffer(eap, DOBUF_FIRST, FORWARD, buf->b_fnum);
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002626 }
2627#ifdef FEAT_QUICKFIX
2628 else if (eap->cmdidx == CMD_cdo || eap->cmdidx == CMD_ldo
2629 || eap->cmdidx == CMD_cfdo || eap->cmdidx == CMD_lfdo)
2630 {
2631 qf_size = qf_get_size(eap);
2632 if (qf_size <= 0 || eap->line1 > qf_size)
2633 buf = NULL;
2634 else
2635 {
2636 ex_cc(eap);
2637
2638 buf = curbuf;
2639 i = eap->line1 - 1;
2640 if (eap->addr_count <= 0)
2641 /* default is all the quickfix/location list entries */
2642 eap->line2 = qf_size;
2643 }
2644 }
2645#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002646 else
2647 setpcmark();
2648 listcmd_busy = TRUE; /* avoids setting pcmark below */
2649
Bram Moolenaare25bb902015-02-27 20:33:37 +01002650 while (!got_int && buf != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002651 {
2652 if (eap->cmdidx == CMD_argdo)
2653 {
2654 /* go to argument "i" */
2655 if (i == ARGCOUNT)
2656 break;
2657 /* Don't call do_argfile() when already there, it will try
2658 * reloading the file. */
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002659 if (curwin->w_arg_idx != i || !editing_arg_idx(curwin))
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002660 {
2661 /* Clear 'shm' to avoid that the file message overwrites
2662 * any output from the command. */
2663 p_shm_save = vim_strsave(p_shm);
2664 set_option_value((char_u *)"shm", 0L, (char_u *)"", 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002665 do_argfile(eap, i);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002666 set_option_value((char_u *)"shm", 0L, p_shm_save, 0);
2667 vim_free(p_shm_save);
2668 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002669 if (curwin->w_arg_idx != i)
2670 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002671 }
2672#ifdef FEAT_WINDOWS
2673 else if (eap->cmdidx == CMD_windo)
2674 {
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002675 /* go to window "wp" */
2676 if (!win_valid(wp))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002677 break;
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002678 win_goto(wp);
Bram Moolenaar41423242007-05-03 20:11:13 +00002679 if (curwin != wp)
2680 break; /* something must be wrong */
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002681 wp = curwin->w_next;
2682 }
2683 else if (eap->cmdidx == CMD_tabdo)
2684 {
2685 /* go to window "tp" */
2686 if (!valid_tabpage(tp))
2687 break;
Bram Moolenaar49e649f2013-05-06 04:50:35 +02002688 goto_tabpage_tp(tp, TRUE, TRUE);
Bram Moolenaar32466aa2006-02-24 23:53:04 +00002689 tp = tp->tp_next;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002690 }
2691#endif
2692 else if (eap->cmdidx == CMD_bufdo)
2693 {
2694 /* Remember the number of the next listed buffer, in case
2695 * ":bwipe" is used or autocommands do something strange. */
2696 next_fnum = -1;
2697 for (buf = curbuf->b_next; buf != NULL; buf = buf->b_next)
2698 if (buf->b_p_bl)
2699 {
2700 next_fnum = buf->b_fnum;
2701 break;
2702 }
2703 }
2704
Bram Moolenaara162bc52015-01-07 16:54:21 +01002705 ++i;
2706
Bram Moolenaar071d4272004-06-13 20:20:40 +00002707 /* execute the command */
2708 do_cmdline(eap->arg, eap->getline, eap->cookie,
2709 DOCMD_VERBOSE + DOCMD_NOWAIT);
2710
2711 if (eap->cmdidx == CMD_bufdo)
2712 {
2713 /* Done? */
Bram Moolenaara162bc52015-01-07 16:54:21 +01002714 if (next_fnum < 0 || next_fnum > eap->line2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002715 break;
2716 /* Check if the buffer still exists. */
2717 for (buf = firstbuf; buf != NULL; buf = buf->b_next)
2718 if (buf->b_fnum == next_fnum)
2719 break;
2720 if (buf == NULL)
2721 break;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002722
2723 /* Go to the next buffer. Clear 'shm' to avoid that the file
2724 * message overwrites any output from the command. */
2725 p_shm_save = vim_strsave(p_shm);
2726 set_option_value((char_u *)"shm", 0L, (char_u *)"", 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002727 goto_buffer(eap, DOBUF_FIRST, FORWARD, next_fnum);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002728 set_option_value((char_u *)"shm", 0L, p_shm_save, 0);
2729 vim_free(p_shm_save);
2730
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002731 /* If autocommands took us elsewhere, quit here. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002732 if (curbuf->b_fnum != next_fnum)
2733 break;
2734 }
2735
Bram Moolenaaraa23b372015-09-08 18:46:31 +02002736#ifdef FEAT_QUICKFIX
2737 if (eap->cmdidx == CMD_cdo || eap->cmdidx == CMD_ldo
2738 || eap->cmdidx == CMD_cfdo || eap->cmdidx == CMD_lfdo)
2739 {
2740 if (i >= qf_size || i >= eap->line2)
2741 break;
2742
2743 qf_idx = qf_get_cur_idx(eap);
2744
2745 ex_cnext(eap);
2746
2747 /* If jumping to the next quickfix entry fails, quit here */
2748 if (qf_get_cur_idx(eap) == qf_idx)
2749 break;
2750 }
2751#endif
2752
Bram Moolenaar071d4272004-06-13 20:20:40 +00002753 if (eap->cmdidx == CMD_windo)
2754 {
2755 validate_cursor(); /* cursor may have moved */
2756#ifdef FEAT_SCROLLBIND
2757 /* required when 'scrollbind' has been set */
2758 if (curwin->w_p_scb)
2759 do_check_scrollbind(TRUE);
2760#endif
2761 }
Bram Moolenaara162bc52015-01-07 16:54:21 +01002762
2763#ifdef FEAT_WINDOWS
2764 if (eap->cmdidx == CMD_windo || eap->cmdidx == CMD_tabdo)
2765 if (i+1 > eap->line2)
2766 break;
2767#endif
2768 if (eap->cmdidx == CMD_argdo && i >= eap->line2)
2769 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002770 }
2771 listcmd_busy = FALSE;
2772 }
2773
2774#if defined(FEAT_AUTOCMD) && defined(FEAT_SYN_HL)
Bram Moolenaar6ac54292005-02-02 23:07:25 +00002775 if (save_ei != NULL)
2776 {
2777 au_event_restore(save_ei);
2778 apply_autocmds(EVENT_SYNTAX, curbuf->b_p_syn,
2779 curbuf->b_fname, TRUE, curbuf);
2780 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002781#endif
Bram Moolenaar6b1ee342014-08-06 18:17:11 +02002782#ifdef FEAT_CLIPBOARD
2783 end_global_changes();
2784#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002785}
2786
2787/*
2788 * Add files[count] to the arglist of the current window after arg "after".
2789 * The file names in files[count] must have been allocated and are taken over.
2790 * Files[] itself is not taken over.
2791 * Returns index of first added argument. Returns -1 when failed (out of mem).
2792 */
2793 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002794alist_add_list(
2795 int count,
2796 char_u **files,
2797 int after) /* where to add: 0 = before first one */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002798{
2799 int i;
Bram Moolenaara24f0a52016-01-17 19:39:00 +01002800 int old_argcount = ARGCOUNT;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002801
2802 if (ga_grow(&ALIST(curwin)->al_ga, count) == OK)
2803 {
2804 if (after < 0)
2805 after = 0;
2806 if (after > ARGCOUNT)
2807 after = ARGCOUNT;
2808 if (after < ARGCOUNT)
2809 mch_memmove(&(ARGLIST[after + count]), &(ARGLIST[after]),
2810 (ARGCOUNT - after) * sizeof(aentry_T));
2811 for (i = 0; i < count; ++i)
2812 {
2813 ARGLIST[after + i].ae_fname = files[i];
2814 ARGLIST[after + i].ae_fnum = buflist_add(files[i], BLN_LISTED);
2815 }
2816 ALIST(curwin)->al_ga.ga_len += count;
Bram Moolenaara24f0a52016-01-17 19:39:00 +01002817 if (old_argcount > 0 && curwin->w_arg_idx >= after)
2818 curwin->w_arg_idx += count;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002819 return after;
2820 }
2821
2822 for (i = 0; i < count; ++i)
2823 vim_free(files[i]);
2824 return -1;
2825}
2826
2827#endif /* FEAT_LISTCMDS */
2828
2829#ifdef FEAT_EVAL
2830/*
2831 * ":compiler[!] {name}"
2832 */
2833 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002834ex_compiler(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002835{
2836 char_u *buf;
2837 char_u *old_cur_comp = NULL;
2838 char_u *p;
2839
2840 if (*eap->arg == NUL)
2841 {
2842 /* List all compiler scripts. */
2843 do_cmdline_cmd((char_u *)"echo globpath(&rtp, 'compiler/*.vim')");
2844 /* ) keep the indenter happy... */
2845 }
2846 else
2847 {
2848 buf = alloc((unsigned)(STRLEN(eap->arg) + 14));
2849 if (buf != NULL)
2850 {
2851 if (eap->forceit)
2852 {
2853 /* ":compiler! {name}" sets global options */
2854 do_cmdline_cmd((char_u *)
2855 "command -nargs=* CompilerSet set <args>");
2856 }
2857 else
2858 {
2859 /* ":compiler! {name}" sets local options.
2860 * To remain backwards compatible "current_compiler" is always
2861 * used. A user's compiler plugin may set it, the distributed
2862 * plugin will then skip the settings. Afterwards set
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01002863 * "b:current_compiler" and restore "current_compiler".
2864 * Explicitly prepend "g:" to make it work in a function. */
2865 old_cur_comp = get_var_value((char_u *)"g:current_compiler");
Bram Moolenaar071d4272004-06-13 20:20:40 +00002866 if (old_cur_comp != NULL)
2867 old_cur_comp = vim_strsave(old_cur_comp);
2868 do_cmdline_cmd((char_u *)
2869 "command -nargs=* CompilerSet setlocal <args>");
2870 }
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01002871 do_unlet((char_u *)"g:current_compiler", TRUE);
Bram Moolenaar2ce06f62005-01-31 19:19:04 +00002872 do_unlet((char_u *)"b:current_compiler", TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002873
2874 sprintf((char *)buf, "compiler/%s.vim", eap->arg);
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01002875 if (source_runtime(buf, DIP_ALL) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002876 EMSG2(_("E666: compiler not supported: %s"), eap->arg);
2877 vim_free(buf);
2878
2879 do_cmdline_cmd((char_u *)":delcommand CompilerSet");
2880
2881 /* Set "b:current_compiler" from "current_compiler". */
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01002882 p = get_var_value((char_u *)"g:current_compiler");
Bram Moolenaar071d4272004-06-13 20:20:40 +00002883 if (p != NULL)
2884 set_internal_string_var((char_u *)"b:current_compiler", p);
2885
2886 /* Restore "current_compiler" for ":compiler {name}". */
2887 if (!eap->forceit)
2888 {
2889 if (old_cur_comp != NULL)
2890 {
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01002891 set_internal_string_var((char_u *)"g:current_compiler",
Bram Moolenaar071d4272004-06-13 20:20:40 +00002892 old_cur_comp);
2893 vim_free(old_cur_comp);
2894 }
2895 else
Bram Moolenaar3d63e3f2010-01-19 16:13:50 +01002896 do_unlet((char_u *)"g:current_compiler", TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002897 }
2898 }
2899 }
2900}
2901#endif
2902
2903/*
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01002904 * ":runtime [what] {name}"
Bram Moolenaar071d4272004-06-13 20:20:40 +00002905 */
2906 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002907ex_runtime(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002908{
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01002909 char_u *arg = eap->arg;
2910 char_u *p = skiptowhite(arg);
2911 int len = (int)(p - arg);
2912 int flags = eap->forceit ? DIP_ALL : 0;
2913
2914 if (STRNCMP(arg, "START", len) == 0)
2915 {
2916 flags += DIP_START + DIP_NORTP;
2917 arg = skipwhite(arg + len);
2918 }
2919 else if (STRNCMP(arg, "OPT", len) == 0)
2920 {
2921 flags += DIP_OPT + DIP_NORTP;
2922 arg = skipwhite(arg + len);
2923 }
2924 else if (STRNCMP(arg, "PACK", len) == 0)
2925 {
2926 flags += DIP_START + DIP_OPT + DIP_NORTP;
2927 arg = skipwhite(arg + len);
2928 }
2929 else if (STRNCMP(arg, "ALL", len) == 0)
2930 {
2931 flags += DIP_START + DIP_OPT;
2932 arg = skipwhite(arg + len);
2933 }
2934
2935 source_runtime(arg, flags);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002936}
2937
Bram Moolenaar071d4272004-06-13 20:20:40 +00002938 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002939source_callback(char_u *fname, void *cookie UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002940{
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002941 (void)do_source(fname, FALSE, DOSO_NONE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002942}
2943
2944/*
2945 * Source the file "name" from all directories in 'runtimepath'.
2946 * "name" can contain wildcards.
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01002947 * When "flags" has DIP_ALL: source all files, otherwise only the first one.
Bram Moolenaar91715872016-03-03 17:13:03 +01002948 *
Bram Moolenaar071d4272004-06-13 20:20:40 +00002949 * return FAIL when no file could be sourced, OK otherwise.
2950 */
2951 int
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01002952source_runtime(char_u *name, int flags)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002953{
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01002954 return do_in_runtimepath(name, flags, source_callback, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002955}
2956
Bram Moolenaarbe82c252016-03-06 14:44:08 +01002957/*
2958 * Find the file "name" in all directories in "path" and invoke
2959 * "callback(fname, cookie)".
2960 * "name" can contain wildcards.
2961 * When "flags" has DIP_ALL: source all files, otherwise only the first one.
2962 * When "flags" has DIP_DIR: find directories instead of files.
2963 * When "flags" has DIP_ERR: give an error message if there is no match.
2964 *
2965 * return FAIL when no file could be sourced, OK otherwise.
2966 */
Bram Moolenaar6bef5302016-03-12 21:28:26 +01002967 int
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01002968do_in_path(
2969 char_u *path,
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002970 char_u *name,
Bram Moolenaar91715872016-03-03 17:13:03 +01002971 int flags,
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002972 void (*callback)(char_u *fname, void *ck),
2973 void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002974{
2975 char_u *rtp;
2976 char_u *np;
2977 char_u *buf;
2978 char_u *rtp_copy;
2979 char_u *tail;
2980 int num_files;
2981 char_u **files;
2982 int i;
2983 int did_one = FALSE;
2984#ifdef AMIGA
2985 struct Process *proc = (struct Process *)FindTask(0L);
2986 APTR save_winptr = proc->pr_WindowPtr;
2987
2988 /* Avoid a requester here for a volume that doesn't exist. */
2989 proc->pr_WindowPtr = (APTR)-1L;
2990#endif
2991
2992 /* Make a copy of 'runtimepath'. Invoking the callback may change the
2993 * value. */
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01002994 rtp_copy = vim_strsave(path);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002995 buf = alloc(MAXPATHL);
2996 if (buf != NULL && rtp_copy != NULL)
2997 {
Bram Moolenaarc09a6d62013-06-10 21:27:29 +02002998 if (p_verbose > 1 && name != NULL)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00002999 {
3000 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00003001 smsg((char_u *)_("Searching for \"%s\" in \"%s\""),
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003002 (char *)name, (char *)path);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003003 verbose_leave();
3004 }
Bram Moolenaar34cdc3e2005-05-18 22:24:46 +00003005
Bram Moolenaar071d4272004-06-13 20:20:40 +00003006 /* Loop over all entries in 'runtimepath'. */
3007 rtp = rtp_copy;
Bram Moolenaar91715872016-03-03 17:13:03 +01003008 while (*rtp != NUL && ((flags & DIP_ALL) || !did_one))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003009 {
3010 /* Copy the path from 'runtimepath' to buf[]. */
3011 copy_option_part(&rtp, buf, MAXPATHL, ",");
Bram Moolenaarc09a6d62013-06-10 21:27:29 +02003012 if (name == NULL)
3013 {
3014 (*callback)(buf, (void *) &cookie);
3015 if (!did_one)
3016 did_one = (cookie == NULL);
3017 }
3018 else if (STRLEN(buf) + STRLEN(name) + 2 < MAXPATHL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003019 {
3020 add_pathsep(buf);
3021 tail = buf + STRLEN(buf);
3022
3023 /* Loop over all patterns in "name" */
3024 np = name;
Bram Moolenaar91715872016-03-03 17:13:03 +01003025 while (*np != NUL && ((flags & DIP_ALL) || !did_one))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003026 {
3027 /* Append the pattern from "name" to buf[]. */
3028 copy_option_part(&np, tail, (int)(MAXPATHL - (tail - buf)),
3029 "\t ");
3030
3031 if (p_verbose > 2)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003032 {
3033 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00003034 smsg((char_u *)_("Searching for \"%s\""), buf);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003035 verbose_leave();
3036 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003037
3038 /* Expand wildcards, invoke the callback for each match. */
3039 if (gen_expand_wildcards(1, &buf, &num_files, &files,
Bram Moolenaar91715872016-03-03 17:13:03 +01003040 (flags & DIP_DIR) ? EW_DIR : EW_FILE) == OK)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003041 {
3042 for (i = 0; i < num_files; ++i)
3043 {
Bram Moolenaar13fcaaf2005-04-15 21:13:42 +00003044 (*callback)(files[i], cookie);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003045 did_one = TRUE;
Bram Moolenaar91715872016-03-03 17:13:03 +01003046 if (!(flags & DIP_ALL))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003047 break;
3048 }
3049 FreeWild(num_files, files);
3050 }
3051 }
3052 }
3053 }
3054 }
3055 vim_free(buf);
3056 vim_free(rtp_copy);
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003057 if (!did_one && name != NULL)
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003058 {
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003059 char *basepath = path == p_rtp ? "runtimepath" : "packpath";
3060
3061 if (flags & DIP_ERR)
3062 EMSG3(_(e_dirnotf), basepath, name);
3063 else if (p_verbose > 0)
3064 {
3065 verbose_enter();
3066 smsg((char_u *)_("not found in '%s': \"%s\""), basepath, name);
3067 verbose_leave();
3068 }
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003069 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003070
3071#ifdef AMIGA
3072 proc->pr_WindowPtr = save_winptr;
3073#endif
3074
3075 return did_one ? OK : FAIL;
3076}
3077
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003078/*
3079 * Find "name" in 'runtimepath'. When found, invoke the callback function for
3080 * it: callback(fname, "cookie")
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003081 * When "flags" has DIP_ALL repeat for all matches, otherwise only the first
3082 * one is used.
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003083 * Returns OK when at least one match found, FAIL otherwise.
3084 *
3085 * If "name" is NULL calls callback for each entry in runtimepath. Cookie is
3086 * passed by reference in this case, setting it to NULL indicates that callback
3087 * has done its job.
3088 */
3089 int
3090do_in_runtimepath(
3091 char_u *name,
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003092 int flags,
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003093 void (*callback)(char_u *fname, void *ck),
3094 void *cookie)
3095{
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003096 int done = FAIL;
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003097 char_u *s;
3098 int len;
3099 char *start_dir = "pack/*/start/*/%s";
3100 char *opt_dir = "pack/*/opt/*/%s";
3101
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003102 if ((flags & DIP_NORTP) == 0)
3103 done = do_in_path(p_rtp, name, flags, callback, cookie);
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003104
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003105 if ((done == FAIL || (flags & DIP_ALL)) && (flags & DIP_START))
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003106 {
3107 len = STRLEN(start_dir) + STRLEN(name);
3108 s = alloc(len);
3109 if (s == NULL)
3110 return FAIL;
3111 vim_snprintf((char *)s, len, start_dir, name);
3112 done = do_in_path(p_pp, s, flags, callback, cookie);
3113 vim_free(s);
3114 }
3115
Bram Moolenaar8dcf2592016-03-12 22:47:14 +01003116 if ((done == FAIL || (flags & DIP_ALL)) && (flags & DIP_OPT))
Bram Moolenaar7f8989d2016-03-12 22:11:39 +01003117 {
3118 len = STRLEN(opt_dir) + STRLEN(name);
3119 s = alloc(len);
3120 if (s == NULL)
3121 return FAIL;
3122 vim_snprintf((char *)s, len, opt_dir, name);
3123 done = do_in_path(p_pp, s, flags, callback, cookie);
3124 vim_free(s);
3125 }
3126
3127 return done;
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003128}
3129
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003130/*
Bram Moolenaarf3654822016-03-04 22:12:23 +01003131 * Expand wildcards in "pat" and invoke do_source() for each match.
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003132 */
3133 static void
Bram Moolenaarf3654822016-03-04 22:12:23 +01003134source_all_matches(char_u *pat)
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003135{
Bram Moolenaarf3654822016-03-04 22:12:23 +01003136 int num_files;
3137 char_u **files;
3138 int i;
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003139
Bram Moolenaarf3654822016-03-04 22:12:23 +01003140 if (gen_expand_wildcards(1, &pat, &num_files, &files, EW_FILE) == OK)
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003141 {
Bram Moolenaarf3654822016-03-04 22:12:23 +01003142 for (i = 0; i < num_files; ++i)
3143 (void)do_source(files[i], FALSE, DOSO_NONE);
3144 FreeWild(num_files, files);
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003145 }
Bram Moolenaar1bdd4262016-03-03 14:23:10 +01003146}
3147
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003148 static void
Bram Moolenaar91715872016-03-03 17:13:03 +01003149add_pack_plugin(char_u *fname, void *cookie)
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003150{
Bram Moolenaarf3654822016-03-04 22:12:23 +01003151 char_u *p4, *p3, *p2, *p1, *p;
3152 char_u *insp;
Bram Moolenaar91715872016-03-03 17:13:03 +01003153 int c;
3154 char_u *new_rtp;
3155 int keep;
3156 int oldlen;
3157 int addlen;
3158 char_u *ffname = fix_fname(fname);
Bram Moolenaarf3654822016-03-04 22:12:23 +01003159 int load_files = cookie != NULL;
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003160
Bram Moolenaar91715872016-03-03 17:13:03 +01003161 if (ffname == NULL)
3162 return;
Bram Moolenaar91715872016-03-03 17:13:03 +01003163 if (strstr((char *)p_rtp, (char *)ffname) == NULL)
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003164 {
3165 /* directory not in 'runtimepath', add it */
Bram Moolenaarf3654822016-03-04 22:12:23 +01003166 p4 = p3 = p2 = p1 = get_past_head(ffname);
3167 for (p = p1; *p; mb_ptr_adv(p))
3168 if (vim_ispathsep_nocolon(*p))
3169 {
3170 p4 = p3; p3 = p2; p2 = p1; p1 = p;
3171 }
3172
3173 /* now we have:
Bram Moolenaaraf1a0e32016-03-09 22:19:26 +01003174 * rtp/pack/name/start/name
3175 * p4 p3 p2 p1
Bram Moolenaarf3654822016-03-04 22:12:23 +01003176 *
3177 * find the part up to "pack" in 'runtimepath' */
3178 c = *p4;
3179 *p4 = NUL;
3180 insp = (char_u *)strstr((char *)p_rtp, (char *)ffname);
3181 if (insp == NULL)
3182 /* not found, append at the end */
3183 insp = p_rtp + STRLEN(p_rtp);
3184 else
3185 {
3186 /* append after the matching directory. */
3187 insp += STRLEN(ffname);
3188 while (*insp != NUL && *insp != ',')
3189 ++insp;
3190 }
3191 *p4 = c;
3192
Bram Moolenaar1daae442016-02-22 23:25:25 +01003193 oldlen = (int)STRLEN(p_rtp);
Bram Moolenaar91715872016-03-03 17:13:03 +01003194 addlen = (int)STRLEN(ffname);
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003195 new_rtp = alloc(oldlen + addlen + 2);
3196 if (new_rtp == NULL)
Bram Moolenaarf3654822016-03-04 22:12:23 +01003197 goto theend;
3198 keep = (int)(insp - p_rtp);
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003199 mch_memmove(new_rtp, p_rtp, keep);
3200 new_rtp[keep] = ',';
Bram Moolenaar91715872016-03-03 17:13:03 +01003201 mch_memmove(new_rtp + keep + 1, ffname, addlen + 1);
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003202 if (p_rtp[keep] != NUL)
3203 mch_memmove(new_rtp + keep + 1 + addlen, p_rtp + keep,
3204 oldlen - keep + 1);
Bram Moolenaar863c1a92016-03-03 15:47:06 +01003205 set_option_value((char_u *)"rtp", 0L, new_rtp, 0);
3206 vim_free(new_rtp);
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003207 }
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003208
Bram Moolenaarf3654822016-03-04 22:12:23 +01003209 if (load_files)
3210 {
3211 static char *plugpat = "%s/plugin/*.vim";
3212 static char *ftpat = "%s/ftdetect/*.vim";
3213 int len;
3214 char_u *pat;
3215
3216 len = (int)STRLEN(ffname) + (int)STRLEN(ftpat);
3217 pat = alloc(len);
3218 if (pat == NULL)
3219 goto theend;
3220 vim_snprintf((char *)pat, len, plugpat, ffname);
3221 source_all_matches(pat);
3222
3223#ifdef FEAT_AUTOCMD
3224 {
3225 char_u *cmd = vim_strsave((char_u *)"g:did_load_filetypes");
3226
3227 /* If runtime/filetype.vim wasn't loaded yet, the scripts will be
3228 * found when it loads. */
3229 if (cmd != NULL && eval_to_number(cmd) > 0)
3230 {
3231 do_cmdline_cmd((char_u *)"augroup filetypedetect");
3232 vim_snprintf((char *)pat, len, ftpat, ffname);
3233 source_all_matches(pat);
3234 do_cmdline_cmd((char_u *)"augroup END");
3235 }
3236 vim_free(cmd);
3237 }
3238#endif
3239 }
3240
3241theend:
3242 vim_free(ffname);
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003243}
3244
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003245static int did_source_packages = FALSE;
3246
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003247/*
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003248 * ":packloadall"
Bram Moolenaarf3654822016-03-04 22:12:23 +01003249 * Find plugins in the package directories and source them.
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003250 */
3251 void
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003252ex_packloadall(exarg_T *eap)
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003253{
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003254 if (!did_source_packages || (eap != NULL && eap->forceit))
3255 {
3256 did_source_packages = TRUE;
3257 do_in_path(p_pp, (char_u *)"pack/*/start/*", DIP_ALL + DIP_DIR,
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003258 add_pack_plugin, p_pp);
Bram Moolenaar2d8f56a2016-03-12 20:34:27 +01003259 }
Bram Moolenaarf6fee0e2016-02-21 23:02:49 +01003260}
3261
3262/*
Bram Moolenaarf3654822016-03-04 22:12:23 +01003263 * ":packadd[!] {name}"
Bram Moolenaar91715872016-03-03 17:13:03 +01003264 */
3265 void
3266ex_packadd(exarg_T *eap)
3267{
3268 static char *plugpat = "pack/*/opt/%s";
3269 int len;
3270 char *pat;
3271
3272 len = (int)STRLEN(plugpat) + (int)STRLEN(eap->arg);
3273 pat = (char *)alloc(len);
3274 if (pat == NULL)
3275 return;
3276 vim_snprintf(pat, len, plugpat, eap->arg);
Bram Moolenaarbe82c252016-03-06 14:44:08 +01003277 do_in_path(p_pp, (char_u *)pat, DIP_ALL + DIP_DIR + DIP_ERR,
3278 add_pack_plugin, eap->forceit ? NULL : p_pp);
Bram Moolenaar91715872016-03-03 17:13:03 +01003279 vim_free(pat);
3280}
3281
Bram Moolenaar071d4272004-06-13 20:20:40 +00003282#if defined(FEAT_EVAL) && defined(FEAT_AUTOCMD)
3283/*
3284 * ":options"
3285 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003286 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003287ex_options(
3288 exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003289{
3290 cmd_source((char_u *)SYS_OPTWIN_FILE, NULL);
3291}
3292#endif
3293
3294/*
3295 * ":source {fname}"
3296 */
3297 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003298ex_source(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003299{
3300#ifdef FEAT_BROWSE
3301 if (cmdmod.browse)
3302 {
3303 char_u *fname = NULL;
3304
Bram Moolenaar7b0294c2004-10-11 10:16:09 +00003305 fname = do_browse(0, (char_u *)_("Source Vim script"), eap->arg,
Bram Moolenaar071d4272004-06-13 20:20:40 +00003306 NULL, NULL, BROWSE_FILTER_MACROS, NULL);
3307 if (fname != NULL)
3308 {
3309 cmd_source(fname, eap);
3310 vim_free(fname);
3311 }
3312 }
3313 else
3314#endif
3315 cmd_source(eap->arg, eap);
3316}
3317
3318 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003319cmd_source(char_u *fname, exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003320{
3321 if (*fname == NUL)
3322 EMSG(_(e_argreq));
3323
Bram Moolenaar071d4272004-06-13 20:20:40 +00003324 else if (eap != NULL && eap->forceit)
Bram Moolenaar84a05ac2013-05-06 04:24:17 +02003325 /* ":source!": read Normal mode commands
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00003326 * Need to execute the commands directly. This is required at least
3327 * for:
Bram Moolenaar071d4272004-06-13 20:20:40 +00003328 * - ":g" command busy
3329 * - after ":argdo", ":windo" or ":bufdo"
3330 * - another command follows
3331 * - inside a loop
3332 */
3333 openscript(fname, global_busy || listcmd_busy || eap->nextcmd != NULL
3334#ifdef FEAT_EVAL
3335 || eap->cstack->cs_idx >= 0
3336#endif
3337 );
3338
3339 /* ":source" read ex commands */
Bram Moolenaar910f66f2006-04-05 20:41:53 +00003340 else if (do_source(fname, FALSE, DOSO_NONE) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003341 EMSG2(_(e_notopen), fname);
3342}
3343
3344/*
3345 * ":source" and associated commands.
3346 */
3347/*
3348 * Structure used to store info for each sourced file.
3349 * It is shared between do_source() and getsourceline().
3350 * This is required, because it needs to be handed to do_cmdline() and
3351 * sourcing can be done recursively.
3352 */
3353struct source_cookie
3354{
3355 FILE *fp; /* opened file for sourcing */
3356 char_u *nextline; /* if not NULL: line that was read ahead */
3357 int finished; /* ":finish" used */
Bram Moolenaar860cae12010-06-05 23:22:07 +02003358#if defined(USE_CRNL) || defined(USE_CR)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003359 int fileformat; /* EOL_UNKNOWN, EOL_UNIX or EOL_DOS */
3360 int error; /* TRUE if LF found after CR-LF */
3361#endif
3362#ifdef FEAT_EVAL
3363 linenr_T breakpoint; /* next line with breakpoint or zero */
3364 char_u *fname; /* name of sourced file */
3365 int dbg_tick; /* debug_tick when breakpoint was set */
3366 int level; /* top nesting level of sourced file */
3367#endif
3368#ifdef FEAT_MBYTE
3369 vimconv_T conv; /* type of conversion */
3370#endif
3371};
3372
3373#ifdef FEAT_EVAL
3374/*
3375 * Return the address holding the next breakpoint line for a source cookie.
3376 */
3377 linenr_T *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003378source_breakpoint(void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003379{
3380 return &((struct source_cookie *)cookie)->breakpoint;
3381}
3382
3383/*
3384 * Return the address holding the debug tick for a source cookie.
3385 */
3386 int *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003387source_dbg_tick(void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003388{
3389 return &((struct source_cookie *)cookie)->dbg_tick;
3390}
3391
3392/*
3393 * Return the nesting level for a source cookie.
3394 */
3395 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003396source_level(void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003397{
3398 return ((struct source_cookie *)cookie)->level;
3399}
3400#endif
3401
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01003402static char_u *get_one_sourceline(struct source_cookie *sp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003403
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01003404#if (defined(WIN32) && defined(FEAT_CSCOPE)) || defined(HAVE_FD_CLOEXEC)
3405# define USE_FOPEN_NOINH
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01003406static FILE *fopen_noinh_readbin(char *filename);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003407
3408/*
3409 * Special function to open a file without handle inheritance.
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01003410 * When possible the handle is closed on exec().
Bram Moolenaar071d4272004-06-13 20:20:40 +00003411 */
3412 static FILE *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003413fopen_noinh_readbin(char *filename)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003414{
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01003415# ifdef WIN32
Bram Moolenaar8d8ef0b2010-01-20 21:41:47 +01003416 int fd_tmp = mch_open(filename, O_RDONLY | O_BINARY | O_NOINHERIT, 0);
3417# else
3418 int fd_tmp = mch_open(filename, O_RDONLY, 0);
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01003419# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003420
3421 if (fd_tmp == -1)
3422 return NULL;
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01003423
3424# ifdef HAVE_FD_CLOEXEC
3425 {
3426 int fdflags = fcntl(fd_tmp, F_GETFD);
3427 if (fdflags >= 0 && (fdflags & FD_CLOEXEC) == 0)
Bram Moolenaarcde88542015-08-11 19:14:00 +02003428 (void)fcntl(fd_tmp, F_SETFD, fdflags | FD_CLOEXEC);
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01003429 }
3430# endif
3431
Bram Moolenaar071d4272004-06-13 20:20:40 +00003432 return fdopen(fd_tmp, READBIN);
3433}
3434#endif
3435
3436
3437/*
3438 * do_source: Read the file "fname" and execute its lines as EX commands.
3439 *
3440 * This function may be called recursively!
3441 *
3442 * return FAIL if file could not be opened, OK otherwise
3443 */
3444 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003445do_source(
3446 char_u *fname,
3447 int check_other, /* check for .vimrc and _vimrc */
3448 int is_vimrc) /* DOSO_ value */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003449{
3450 struct source_cookie cookie;
3451 char_u *save_sourcing_name;
3452 linenr_T save_sourcing_lnum;
3453 char_u *p;
3454 char_u *fname_exp;
Bram Moolenaar73881402009-02-04 16:50:47 +00003455 char_u *firstline = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003456 int retval = FAIL;
3457#ifdef FEAT_EVAL
3458 scid_T save_current_SID;
3459 static scid_T last_current_SID = 0;
3460 void *save_funccalp;
3461 int save_debug_break_level = debug_break_level;
Bram Moolenaar05159a02005-02-26 23:04:13 +00003462 scriptitem_T *si = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003463# ifdef UNIX
3464 struct stat st;
3465 int stat_ok;
3466# endif
3467#endif
3468#ifdef STARTUPTIME
3469 struct timeval tv_rel;
3470 struct timeval tv_start;
3471#endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00003472#ifdef FEAT_PROFILE
3473 proftime_T wait_start;
3474#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003475
Bram Moolenaar071d4272004-06-13 20:20:40 +00003476 p = expand_env_save(fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003477 if (p == NULL)
3478 return retval;
3479 fname_exp = fix_fname(p);
3480 vim_free(p);
3481 if (fname_exp == NULL)
3482 return retval;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003483 if (mch_isdir(fname_exp))
3484 {
Bram Moolenaar555b2802005-05-19 21:08:39 +00003485 smsg((char_u *)_("Cannot source a directory: \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003486 goto theend;
3487 }
3488
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00003489#ifdef FEAT_AUTOCMD
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00003490 /* Apply SourceCmd autocommands, they should get the file and source it. */
3491 if (has_autocmd(EVENT_SOURCECMD, fname_exp, NULL)
3492 && apply_autocmds(EVENT_SOURCECMD, fname_exp, fname_exp,
3493 FALSE, curbuf))
Bram Moolenaarf33943e2008-01-15 21:17:30 +00003494 {
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00003495# ifdef FEAT_EVAL
Bram Moolenaarf33943e2008-01-15 21:17:30 +00003496 retval = aborting() ? FAIL : OK;
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00003497# else
Bram Moolenaarf33943e2008-01-15 21:17:30 +00003498 retval = OK;
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00003499# endif
Bram Moolenaarf33943e2008-01-15 21:17:30 +00003500 goto theend;
3501 }
Bram Moolenaar8dd1aa52007-01-16 20:33:19 +00003502
3503 /* Apply SourcePre autocommands, they may get the file. */
Bram Moolenaar1f35bf92006-03-07 22:38:47 +00003504 apply_autocmds(EVENT_SOURCEPRE, fname_exp, fname_exp, FALSE, curbuf);
3505#endif
3506
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01003507#ifdef USE_FOPEN_NOINH
Bram Moolenaar071d4272004-06-13 20:20:40 +00003508 cookie.fp = fopen_noinh_readbin((char *)fname_exp);
3509#else
3510 cookie.fp = mch_fopen((char *)fname_exp, READBIN);
3511#endif
3512 if (cookie.fp == NULL && check_other)
3513 {
3514 /*
3515 * Try again, replacing file name ".vimrc" by "_vimrc" or vice versa,
3516 * and ".exrc" by "_exrc" or vice versa.
3517 */
3518 p = gettail(fname_exp);
3519 if ((*p == '.' || *p == '_')
3520 && (STRICMP(p + 1, "vimrc") == 0
3521 || STRICMP(p + 1, "gvimrc") == 0
3522 || STRICMP(p + 1, "exrc") == 0))
3523 {
3524 if (*p == '_')
3525 *p = '.';
3526 else
3527 *p = '_';
Bram Moolenaar6b29b0e2010-01-19 16:22:03 +01003528#ifdef USE_FOPEN_NOINH
Bram Moolenaar071d4272004-06-13 20:20:40 +00003529 cookie.fp = fopen_noinh_readbin((char *)fname_exp);
3530#else
3531 cookie.fp = mch_fopen((char *)fname_exp, READBIN);
3532#endif
3533 }
3534 }
3535
3536 if (cookie.fp == NULL)
3537 {
3538 if (p_verbose > 0)
3539 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003540 verbose_enter();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003541 if (sourcing_name == NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00003542 smsg((char_u *)_("could not source \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003543 else
3544 smsg((char_u *)_("line %ld: could not source \"%s\""),
Bram Moolenaar555b2802005-05-19 21:08:39 +00003545 sourcing_lnum, fname);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003546 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003547 }
3548 goto theend;
3549 }
3550
3551 /*
3552 * The file exists.
3553 * - In verbose mode, give a message.
3554 * - For a vimrc file, may want to set 'compatible', call vimrc_found().
3555 */
3556 if (p_verbose > 1)
3557 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003558 verbose_enter();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003559 if (sourcing_name == NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00003560 smsg((char_u *)_("sourcing \"%s\""), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003561 else
3562 smsg((char_u *)_("line %ld: sourcing \"%s\""),
Bram Moolenaar555b2802005-05-19 21:08:39 +00003563 sourcing_lnum, fname);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003564 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003565 }
Bram Moolenaar910f66f2006-04-05 20:41:53 +00003566 if (is_vimrc == DOSO_VIMRC)
3567 vimrc_found(fname_exp, (char_u *)"MYVIMRC");
3568 else if (is_vimrc == DOSO_GVIMRC)
3569 vimrc_found(fname_exp, (char_u *)"MYGVIMRC");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003570
3571#ifdef USE_CRNL
3572 /* If no automatic file format: Set default to CR-NL. */
3573 if (*p_ffs == NUL)
3574 cookie.fileformat = EOL_DOS;
3575 else
3576 cookie.fileformat = EOL_UNKNOWN;
3577 cookie.error = FALSE;
3578#endif
3579
3580#ifdef USE_CR
3581 /* If no automatic file format: Set default to CR. */
3582 if (*p_ffs == NUL)
3583 cookie.fileformat = EOL_MAC;
3584 else
3585 cookie.fileformat = EOL_UNKNOWN;
3586 cookie.error = FALSE;
3587#endif
3588
3589 cookie.nextline = NULL;
3590 cookie.finished = FALSE;
3591
3592#ifdef FEAT_EVAL
3593 /*
3594 * Check if this script has a breakpoint.
3595 */
3596 cookie.breakpoint = dbg_find_breakpoint(TRUE, fname_exp, (linenr_T)0);
3597 cookie.fname = fname_exp;
3598 cookie.dbg_tick = debug_tick;
3599
3600 cookie.level = ex_nesting_level;
3601#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003602
3603 /*
3604 * Keep the sourcing name/lnum, for recursive calls.
3605 */
3606 save_sourcing_name = sourcing_name;
3607 sourcing_name = fname_exp;
3608 save_sourcing_lnum = sourcing_lnum;
3609 sourcing_lnum = 0;
3610
Bram Moolenaar73881402009-02-04 16:50:47 +00003611#ifdef FEAT_MBYTE
3612 cookie.conv.vc_type = CONV_NONE; /* no conversion */
3613
3614 /* Read the first line so we can check for a UTF-8 BOM. */
3615 firstline = getsourceline(0, (void *)&cookie, 0);
3616 if (firstline != NULL && STRLEN(firstline) >= 3 && firstline[0] == 0xef
3617 && firstline[1] == 0xbb && firstline[2] == 0xbf)
3618 {
3619 /* Found BOM; setup conversion, skip over BOM and recode the line. */
3620 convert_setup(&cookie.conv, (char_u *)"utf-8", p_enc);
3621 p = string_convert(&cookie.conv, firstline + 3, NULL);
Bram Moolenaar4e2a5952009-02-05 19:48:25 +00003622 if (p == NULL)
3623 p = vim_strsave(firstline + 3);
Bram Moolenaar73881402009-02-04 16:50:47 +00003624 if (p != NULL)
3625 {
3626 vim_free(firstline);
3627 firstline = p;
3628 }
3629 }
3630#endif
3631
Bram Moolenaar071d4272004-06-13 20:20:40 +00003632#ifdef STARTUPTIME
Bram Moolenaarc4e41982010-01-19 16:31:47 +01003633 if (time_fd != NULL)
3634 time_push(&tv_rel, &tv_start);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003635#endif
3636
3637#ifdef FEAT_EVAL
Bram Moolenaar05159a02005-02-26 23:04:13 +00003638# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00003639 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003640 prof_child_enter(&wait_start); /* entering a child now */
3641# endif
3642
3643 /* Don't use local function variables, if called from a function.
3644 * Also starts profiling timer for nested script. */
3645 save_funccalp = save_funccal();
3646
Bram Moolenaar071d4272004-06-13 20:20:40 +00003647 /*
3648 * Check if this script was sourced before to finds its SID.
3649 * If it's new, generate a new SID.
3650 */
3651 save_current_SID = current_SID;
3652# ifdef UNIX
3653 stat_ok = (mch_stat((char *)fname_exp, &st) >= 0);
3654# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00003655 for (current_SID = script_items.ga_len; current_SID > 0; --current_SID)
3656 {
3657 si = &SCRIPT_ITEM(current_SID);
3658 if (si->sn_name != NULL
Bram Moolenaar071d4272004-06-13 20:20:40 +00003659 && (
3660# ifdef UNIX
Bram Moolenaar7c626922005-02-07 22:01:03 +00003661 /* Compare dev/ino when possible, it catches symbolic
3662 * links. Also compare file names, the inode may change
3663 * when the file was edited. */
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00003664 ((stat_ok && si->sn_dev_valid)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003665 && (si->sn_dev == st.st_dev
3666 && si->sn_ino == st.st_ino)) ||
Bram Moolenaar071d4272004-06-13 20:20:40 +00003667# endif
Bram Moolenaar05159a02005-02-26 23:04:13 +00003668 fnamecmp(si->sn_name, fname_exp) == 0))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003669 break;
Bram Moolenaar05159a02005-02-26 23:04:13 +00003670 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003671 if (current_SID == 0)
3672 {
3673 current_SID = ++last_current_SID;
Bram Moolenaar05159a02005-02-26 23:04:13 +00003674 if (ga_grow(&script_items, (int)(current_SID - script_items.ga_len))
3675 == FAIL)
3676 goto almosttheend;
3677 while (script_items.ga_len < current_SID)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003678 {
Bram Moolenaar05159a02005-02-26 23:04:13 +00003679 ++script_items.ga_len;
3680 SCRIPT_ITEM(script_items.ga_len).sn_name = NULL;
3681# ifdef FEAT_PROFILE
3682 SCRIPT_ITEM(script_items.ga_len).sn_prof_on = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003683# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003684 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00003685 si = &SCRIPT_ITEM(current_SID);
3686 si->sn_name = fname_exp;
3687 fname_exp = NULL;
3688# ifdef UNIX
3689 if (stat_ok)
3690 {
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00003691 si->sn_dev_valid = TRUE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00003692 si->sn_dev = st.st_dev;
3693 si->sn_ino = st.st_ino;
3694 }
3695 else
Bram Moolenaarbf0c4522009-05-16 19:16:33 +00003696 si->sn_dev_valid = FALSE;
Bram Moolenaar05159a02005-02-26 23:04:13 +00003697# endif
3698
Bram Moolenaar071d4272004-06-13 20:20:40 +00003699 /* Allocate the local script variables to use for this script. */
3700 new_script_vars(current_SID);
3701 }
3702
Bram Moolenaar05159a02005-02-26 23:04:13 +00003703# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00003704 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003705 {
3706 int forceit;
3707
3708 /* Check if we do profiling for this script. */
3709 if (!si->sn_prof_on && has_profiling(TRUE, si->sn_name, &forceit))
3710 {
3711 script_do_profile(si);
3712 si->sn_pr_force = forceit;
3713 }
3714 if (si->sn_prof_on)
3715 {
3716 ++si->sn_pr_count;
3717 profile_start(&si->sn_pr_start);
3718 profile_zero(&si->sn_pr_children);
3719 }
3720 }
3721# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003722#endif
3723
3724 /*
3725 * Call do_cmdline, which will call getsourceline() to get the lines.
3726 */
Bram Moolenaar73881402009-02-04 16:50:47 +00003727 do_cmdline(firstline, getsourceline, (void *)&cookie,
Bram Moolenaar071d4272004-06-13 20:20:40 +00003728 DOCMD_VERBOSE|DOCMD_NOWAIT|DOCMD_REPEAT);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003729 retval = OK;
Bram Moolenaar05159a02005-02-26 23:04:13 +00003730
3731#ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00003732 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003733 {
3734 /* Get "si" again, "script_items" may have been reallocated. */
3735 si = &SCRIPT_ITEM(current_SID);
3736 if (si->sn_prof_on)
3737 {
3738 profile_end(&si->sn_pr_start);
3739 profile_sub_wait(&wait_start, &si->sn_pr_start);
3740 profile_add(&si->sn_pr_total, &si->sn_pr_start);
Bram Moolenaar1056d982006-03-09 22:37:52 +00003741 profile_self(&si->sn_pr_self, &si->sn_pr_start,
3742 &si->sn_pr_children);
Bram Moolenaar05159a02005-02-26 23:04:13 +00003743 }
3744 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003745#endif
3746
3747 if (got_int)
3748 EMSG(_(e_interr));
3749 sourcing_name = save_sourcing_name;
3750 sourcing_lnum = save_sourcing_lnum;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003751 if (p_verbose > 1)
3752 {
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003753 verbose_enter();
Bram Moolenaar555b2802005-05-19 21:08:39 +00003754 smsg((char_u *)_("finished sourcing %s"), fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003755 if (sourcing_name != NULL)
Bram Moolenaar555b2802005-05-19 21:08:39 +00003756 smsg((char_u *)_("continuing in %s"), sourcing_name);
Bram Moolenaar54ee7752005-05-31 22:22:17 +00003757 verbose_leave();
Bram Moolenaar071d4272004-06-13 20:20:40 +00003758 }
3759#ifdef STARTUPTIME
Bram Moolenaarc4e41982010-01-19 16:31:47 +01003760 if (time_fd != NULL)
3761 {
3762 vim_snprintf((char *)IObuff, IOSIZE, "sourcing %s", fname);
3763 time_msg((char *)IObuff, &tv_start);
3764 time_pop(&tv_rel);
3765 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003766#endif
3767
3768#ifdef FEAT_EVAL
3769 /*
3770 * After a "finish" in debug mode, need to break at first command of next
3771 * sourced file.
3772 */
3773 if (save_debug_break_level > ex_nesting_level
3774 && debug_break_level == ex_nesting_level)
3775 ++debug_break_level;
3776#endif
3777
Bram Moolenaar05159a02005-02-26 23:04:13 +00003778#ifdef FEAT_EVAL
3779almosttheend:
3780 current_SID = save_current_SID;
3781 restore_funccal(save_funccalp);
3782# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00003783 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003784 prof_child_exit(&wait_start); /* leaving a child now */
3785# endif
3786#endif
3787 fclose(cookie.fp);
3788 vim_free(cookie.nextline);
Bram Moolenaar73881402009-02-04 16:50:47 +00003789 vim_free(firstline);
Bram Moolenaar05159a02005-02-26 23:04:13 +00003790#ifdef FEAT_MBYTE
3791 convert_setup(&cookie.conv, NULL, NULL);
3792#endif
3793
Bram Moolenaar071d4272004-06-13 20:20:40 +00003794theend:
3795 vim_free(fname_exp);
3796 return retval;
3797}
3798
3799#if defined(FEAT_EVAL) || defined(PROTO)
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00003800
Bram Moolenaar071d4272004-06-13 20:20:40 +00003801/*
3802 * ":scriptnames"
3803 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003804 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003805ex_scriptnames(exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003806{
3807 int i;
3808
Bram Moolenaar05159a02005-02-26 23:04:13 +00003809 for (i = 1; i <= script_items.ga_len && !got_int; ++i)
3810 if (SCRIPT_ITEM(i).sn_name != NULL)
Bram Moolenaard58ea072011-06-26 04:25:30 +02003811 {
3812 home_replace(NULL, SCRIPT_ITEM(i).sn_name,
3813 NameBuff, MAXPATHL, TRUE);
3814 smsg((char_u *)"%3d: %s", i, NameBuff);
Bram Moolenaar970a1b82012-03-23 18:39:18 +01003815 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003816}
3817
3818# if defined(BACKSLASH_IN_FILENAME) || defined(PROTO)
3819/*
3820 * Fix slashes in the list of script names for 'shellslash'.
3821 */
3822 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003823scriptnames_slash_adjust(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003824{
3825 int i;
3826
Bram Moolenaar05159a02005-02-26 23:04:13 +00003827 for (i = 1; i <= script_items.ga_len; ++i)
3828 if (SCRIPT_ITEM(i).sn_name != NULL)
3829 slash_adjust(SCRIPT_ITEM(i).sn_name);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003830}
3831# endif
3832
3833/*
3834 * Get a pointer to a script name. Used for ":verbose set".
3835 */
3836 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003837get_scriptname(scid_T id)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003838{
3839 if (id == SID_MODELINE)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00003840 return (char_u *)_("modeline");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003841 if (id == SID_CMDARG)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00003842 return (char_u *)_("--cmd argument");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003843 if (id == SID_CARG)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00003844 return (char_u *)_("-c argument");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003845 if (id == SID_ENV)
Bram Moolenaard1f56e62006-02-22 21:25:37 +00003846 return (char_u *)_("environment variable");
3847 if (id == SID_ERROR)
3848 return (char_u *)_("error handler");
Bram Moolenaar05159a02005-02-26 23:04:13 +00003849 return SCRIPT_ITEM(id).sn_name;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003850}
Bram Moolenaar05159a02005-02-26 23:04:13 +00003851
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00003852# if defined(EXITFREE) || defined(PROTO)
3853 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003854free_scriptnames(void)
Bram Moolenaar1ec484f2005-06-24 23:07:47 +00003855{
3856 int i;
3857
3858 for (i = script_items.ga_len; i > 0; --i)
3859 vim_free(SCRIPT_ITEM(i).sn_name);
3860 ga_clear(&script_items);
3861}
3862# endif
3863
Bram Moolenaar071d4272004-06-13 20:20:40 +00003864#endif
3865
3866#if defined(USE_CR) || defined(PROTO)
3867
3868# if defined(__MSL__) && (__MSL__ >= 22)
3869/*
3870 * Newer version of the Metrowerks library handle DOS and UNIX files
3871 * without help.
3872 * Test with earlier versions, MSL 2.2 is the library supplied with
3873 * Codewarrior Pro 2.
3874 */
3875 char *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003876fgets_cr(char *s, int n, FILE *stream)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003877{
3878 return fgets(s, n, stream);
3879}
3880# else
3881/*
3882 * Version of fgets() which also works for lines ending in a <CR> only
3883 * (Macintosh format).
3884 * For older versions of the Metrowerks library.
3885 * At least CodeWarrior 9 needed this code.
3886 */
3887 char *
Bram Moolenaard14e00e2016-01-31 17:30:51 +01003888fgets_cr(char *s, int n, FILE *stream)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003889{
3890 int c = 0;
3891 int char_read = 0;
3892
3893 while (!feof(stream) && c != '\r' && c != '\n' && char_read < n - 1)
3894 {
3895 c = fgetc(stream);
3896 s[char_read++] = c;
3897 /* If the file is in DOS format, we need to skip a NL after a CR. I
3898 * thought it was the other way around, but this appears to work... */
3899 if (c == '\n')
3900 {
3901 c = fgetc(stream);
3902 if (c != '\r')
3903 ungetc(c, stream);
3904 }
3905 }
3906
3907 s[char_read] = 0;
3908 if (char_read == 0)
3909 return NULL;
3910
3911 if (feof(stream) && char_read == 1)
3912 return NULL;
3913
3914 return s;
3915}
3916# endif
3917#endif
3918
3919/*
3920 * Get one full line from a sourced file.
3921 * Called by do_cmdline() when it's called from do_source().
3922 *
3923 * Return a pointer to the line in allocated memory.
3924 * Return NULL for end-of-file or some error.
3925 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003926 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003927getsourceline(int c UNUSED, void *cookie, int indent UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003928{
3929 struct source_cookie *sp = (struct source_cookie *)cookie;
3930 char_u *line;
Bram Moolenaarc047b9a2012-02-11 20:40:55 +01003931 char_u *p;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003932
3933#ifdef FEAT_EVAL
3934 /* If breakpoints have been added/deleted need to check for it. */
3935 if (sp->dbg_tick < debug_tick)
3936 {
3937 sp->breakpoint = dbg_find_breakpoint(TRUE, sp->fname, sourcing_lnum);
3938 sp->dbg_tick = debug_tick;
3939 }
Bram Moolenaar05159a02005-02-26 23:04:13 +00003940# ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00003941 if (do_profiling == PROF_YES)
Bram Moolenaar05159a02005-02-26 23:04:13 +00003942 script_line_end();
3943# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003944#endif
3945 /*
3946 * Get current line. If there is a read-ahead line, use it, otherwise get
3947 * one now.
3948 */
3949 if (sp->finished)
3950 line = NULL;
3951 else if (sp->nextline == NULL)
3952 line = get_one_sourceline(sp);
3953 else
3954 {
3955 line = sp->nextline;
3956 sp->nextline = NULL;
3957 ++sourcing_lnum;
3958 }
Bram Moolenaare2cc9702005-03-15 22:43:58 +00003959#ifdef FEAT_PROFILE
Bram Moolenaar9b2200a2006-03-20 21:55:45 +00003960 if (line != NULL && do_profiling == PROF_YES)
Bram Moolenaare2cc9702005-03-15 22:43:58 +00003961 script_line_start();
3962#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003963
3964 /* Only concatenate lines starting with a \ when 'cpoptions' doesn't
3965 * contain the 'C' flag. */
3966 if (line != NULL && (vim_strchr(p_cpo, CPO_CONCAT) == NULL))
3967 {
3968 /* compensate for the one line read-ahead */
3969 --sourcing_lnum;
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01003970
3971 /* Get the next line and concatenate it when it starts with a
3972 * backslash. We always need to read the next line, keep it in
3973 * sp->nextline. */
3974 sp->nextline = get_one_sourceline(sp);
3975 if (sp->nextline != NULL && *(p = skipwhite(sp->nextline)) == '\\')
Bram Moolenaar071d4272004-06-13 20:20:40 +00003976 {
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01003977 garray_T ga;
3978
Bram Moolenaarb549a732012-02-22 18:29:33 +01003979 ga_init2(&ga, (int)sizeof(char_u), 400);
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01003980 ga_concat(&ga, line);
3981 ga_concat(&ga, p + 1);
3982 for (;;)
3983 {
3984 vim_free(sp->nextline);
3985 sp->nextline = get_one_sourceline(sp);
3986 if (sp->nextline == NULL)
3987 break;
3988 p = skipwhite(sp->nextline);
3989 if (*p != '\\')
3990 break;
Bram Moolenaarb549a732012-02-22 18:29:33 +01003991 /* Adjust the growsize to the current length to speed up
3992 * concatenating many lines. */
3993 if (ga.ga_len > 400)
3994 {
3995 if (ga.ga_len > 8000)
3996 ga.ga_growsize = 8000;
3997 else
3998 ga.ga_growsize = ga.ga_len;
3999 }
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004000 ga_concat(&ga, p + 1);
4001 }
4002 ga_append(&ga, NUL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004003 vim_free(line);
Bram Moolenaarb3a6bbc2012-02-05 23:10:30 +01004004 line = ga.ga_data;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004005 }
4006 }
4007
4008#ifdef FEAT_MBYTE
4009 if (line != NULL && sp->conv.vc_type != CONV_NONE)
4010 {
Bram Moolenaarc047b9a2012-02-11 20:40:55 +01004011 char_u *s;
4012
Bram Moolenaar071d4272004-06-13 20:20:40 +00004013 /* Convert the encoding of the script line. */
4014 s = string_convert(&sp->conv, line, NULL);
4015 if (s != NULL)
4016 {
4017 vim_free(line);
4018 line = s;
4019 }
4020 }
4021#endif
4022
4023#ifdef FEAT_EVAL
4024 /* Did we encounter a breakpoint? */
4025 if (sp->breakpoint != 0 && sp->breakpoint <= sourcing_lnum)
4026 {
4027 dbg_breakpoint(sp->fname, sourcing_lnum);
4028 /* Find next breakpoint. */
4029 sp->breakpoint = dbg_find_breakpoint(TRUE, sp->fname, sourcing_lnum);
4030 sp->dbg_tick = debug_tick;
4031 }
4032#endif
4033
4034 return line;
4035}
4036
4037 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004038get_one_sourceline(struct source_cookie *sp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004039{
4040 garray_T ga;
4041 int len;
4042 int c;
4043 char_u *buf;
4044#ifdef USE_CRNL
4045 int has_cr; /* CR-LF found */
4046#endif
4047#ifdef USE_CR
4048 char_u *scan;
4049#endif
4050 int have_read = FALSE;
4051
4052 /* use a growarray to store the sourced line */
Bram Moolenaar6ac54292005-02-02 23:07:25 +00004053 ga_init2(&ga, 1, 250);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004054
4055 /*
4056 * Loop until there is a finished line (or end-of-file).
4057 */
4058 sourcing_lnum++;
4059 for (;;)
4060 {
Bram Moolenaar6ac54292005-02-02 23:07:25 +00004061 /* make room to read at least 120 (more) characters */
4062 if (ga_grow(&ga, 120) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004063 break;
4064 buf = (char_u *)ga.ga_data;
4065
4066#ifdef USE_CR
4067 if (sp->fileformat == EOL_MAC)
4068 {
Bram Moolenaar86b68352004-12-27 21:59:20 +00004069 if (fgets_cr((char *)buf + ga.ga_len, ga.ga_maxlen - ga.ga_len,
4070 sp->fp) == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004071 break;
4072 }
4073 else
4074#endif
Bram Moolenaar86b68352004-12-27 21:59:20 +00004075 if (fgets((char *)buf + ga.ga_len, ga.ga_maxlen - ga.ga_len,
4076 sp->fp) == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004077 break;
Bram Moolenaar6ac54292005-02-02 23:07:25 +00004078 len = ga.ga_len + (int)STRLEN(buf + ga.ga_len);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004079#ifdef USE_CRNL
4080 /* Ignore a trailing CTRL-Z, when in Dos mode. Only recognize the
4081 * CTRL-Z by its own, or after a NL. */
4082 if ( (len == 1 || (len >= 2 && buf[len - 2] == '\n'))
4083 && sp->fileformat == EOL_DOS
4084 && buf[len - 1] == Ctrl_Z)
4085 {
4086 buf[len - 1] = NUL;
4087 break;
4088 }
4089#endif
4090
4091#ifdef USE_CR
4092 /* If the read doesn't stop on a new line, and there's
4093 * some CR then we assume a Mac format */
4094 if (sp->fileformat == EOL_UNKNOWN)
4095 {
4096 if (buf[len - 1] != '\n' && vim_strchr(buf, '\r') != NULL)
4097 sp->fileformat = EOL_MAC;
4098 else
4099 sp->fileformat = EOL_UNIX;
4100 }
4101
4102 if (sp->fileformat == EOL_MAC)
4103 {
4104 scan = vim_strchr(buf, '\r');
4105
4106 if (scan != NULL)
4107 {
4108 *scan = '\n';
4109 if (*(scan + 1) != 0)
4110 {
4111 *(scan + 1) = 0;
4112 fseek(sp->fp, (long)(scan - buf - len + 1), SEEK_CUR);
4113 }
4114 }
4115 len = STRLEN(buf);
4116 }
4117#endif
4118
4119 have_read = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004120 ga.ga_len = len;
4121
4122 /* If the line was longer than the buffer, read more. */
Bram Moolenaar86b68352004-12-27 21:59:20 +00004123 if (ga.ga_maxlen - ga.ga_len == 1 && buf[len - 1] != '\n')
Bram Moolenaar071d4272004-06-13 20:20:40 +00004124 continue;
4125
4126 if (len >= 1 && buf[len - 1] == '\n') /* remove trailing NL */
4127 {
4128#ifdef USE_CRNL
4129 has_cr = (len >= 2 && buf[len - 2] == '\r');
4130 if (sp->fileformat == EOL_UNKNOWN)
4131 {
4132 if (has_cr)
4133 sp->fileformat = EOL_DOS;
4134 else
4135 sp->fileformat = EOL_UNIX;
4136 }
4137
4138 if (sp->fileformat == EOL_DOS)
4139 {
4140 if (has_cr) /* replace trailing CR */
4141 {
4142 buf[len - 2] = '\n';
4143 --len;
4144 --ga.ga_len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004145 }
4146 else /* lines like ":map xx yy^M" will have failed */
4147 {
4148 if (!sp->error)
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00004149 {
4150 msg_source(hl_attr(HLF_W));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004151 EMSG(_("W15: Warning: Wrong line separator, ^M may be missing"));
Bram Moolenaar2df6dcc2004-07-12 15:53:54 +00004152 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004153 sp->error = TRUE;
4154 sp->fileformat = EOL_UNIX;
4155 }
4156 }
4157#endif
4158 /* The '\n' is escaped if there is an odd number of ^V's just
4159 * before it, first set "c" just before the 'V's and then check
4160 * len&c parities (is faster than ((len-c)%2 == 0)) -- Acevedo */
4161 for (c = len - 2; c >= 0 && buf[c] == Ctrl_V; c--)
4162 ;
4163 if ((len & 1) != (c & 1)) /* escaped NL, read more */
4164 {
4165 sourcing_lnum++;
4166 continue;
4167 }
4168
4169 buf[len - 1] = NUL; /* remove the NL */
4170 }
4171
4172 /*
4173 * Check for ^C here now and then, so recursive :so can be broken.
4174 */
4175 line_breakcheck();
4176 break;
4177 }
4178
4179 if (have_read)
4180 return (char_u *)ga.ga_data;
4181
4182 vim_free(ga.ga_data);
4183 return NULL;
4184}
4185
Bram Moolenaar05159a02005-02-26 23:04:13 +00004186#if defined(FEAT_PROFILE) || defined(PROTO)
4187/*
4188 * Called when starting to read a script line.
4189 * "sourcing_lnum" must be correct!
4190 * When skipping lines it may not actually be executed, but we won't find out
4191 * until later and we need to store the time now.
4192 */
4193 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004194script_line_start(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004195{
4196 scriptitem_T *si;
4197 sn_prl_T *pp;
4198
4199 if (current_SID <= 0 || current_SID > script_items.ga_len)
4200 return;
4201 si = &SCRIPT_ITEM(current_SID);
4202 if (si->sn_prof_on && sourcing_lnum >= 1)
4203 {
Bram Moolenaar8c8de832008-06-24 22:58:06 +00004204 /* Grow the array before starting the timer, so that the time spent
Bram Moolenaar05159a02005-02-26 23:04:13 +00004205 * here isn't counted. */
Bram Moolenaarcde88542015-08-11 19:14:00 +02004206 (void)ga_grow(&si->sn_prl_ga, (int)(sourcing_lnum - si->sn_prl_ga.ga_len));
Bram Moolenaar05159a02005-02-26 23:04:13 +00004207 si->sn_prl_idx = sourcing_lnum - 1;
4208 while (si->sn_prl_ga.ga_len <= si->sn_prl_idx
4209 && si->sn_prl_ga.ga_len < si->sn_prl_ga.ga_maxlen)
4210 {
4211 /* Zero counters for a line that was not used before. */
4212 pp = &PRL_ITEM(si, si->sn_prl_ga.ga_len);
4213 pp->snp_count = 0;
4214 profile_zero(&pp->sn_prl_total);
4215 profile_zero(&pp->sn_prl_self);
4216 ++si->sn_prl_ga.ga_len;
4217 }
4218 si->sn_prl_execed = FALSE;
4219 profile_start(&si->sn_prl_start);
4220 profile_zero(&si->sn_prl_children);
4221 profile_get_wait(&si->sn_prl_wait);
4222 }
4223}
4224
4225/*
4226 * Called when actually executing a function line.
4227 */
4228 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004229script_line_exec(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004230{
4231 scriptitem_T *si;
4232
4233 if (current_SID <= 0 || current_SID > script_items.ga_len)
4234 return;
4235 si = &SCRIPT_ITEM(current_SID);
4236 if (si->sn_prof_on && si->sn_prl_idx >= 0)
4237 si->sn_prl_execed = TRUE;
4238}
4239
4240/*
4241 * Called when done with a function line.
4242 */
4243 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004244script_line_end(void)
Bram Moolenaar05159a02005-02-26 23:04:13 +00004245{
4246 scriptitem_T *si;
4247 sn_prl_T *pp;
4248
4249 if (current_SID <= 0 || current_SID > script_items.ga_len)
4250 return;
4251 si = &SCRIPT_ITEM(current_SID);
4252 if (si->sn_prof_on && si->sn_prl_idx >= 0
4253 && si->sn_prl_idx < si->sn_prl_ga.ga_len)
4254 {
4255 if (si->sn_prl_execed)
4256 {
4257 pp = &PRL_ITEM(si, si->sn_prl_idx);
4258 ++pp->snp_count;
4259 profile_end(&si->sn_prl_start);
4260 profile_sub_wait(&si->sn_prl_wait, &si->sn_prl_start);
Bram Moolenaar05159a02005-02-26 23:04:13 +00004261 profile_add(&pp->sn_prl_total, &si->sn_prl_start);
Bram Moolenaar1056d982006-03-09 22:37:52 +00004262 profile_self(&pp->sn_prl_self, &si->sn_prl_start,
4263 &si->sn_prl_children);
Bram Moolenaar05159a02005-02-26 23:04:13 +00004264 }
4265 si->sn_prl_idx = -1;
4266 }
4267}
4268#endif
4269
Bram Moolenaar071d4272004-06-13 20:20:40 +00004270/*
4271 * ":scriptencoding": Set encoding conversion for a sourced script.
4272 * Without the multi-byte feature it's simply ignored.
4273 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004274 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004275ex_scriptencoding(exarg_T *eap UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004276{
4277#ifdef FEAT_MBYTE
4278 struct source_cookie *sp;
4279 char_u *name;
4280
4281 if (!getline_equal(eap->getline, eap->cookie, getsourceline))
4282 {
4283 EMSG(_("E167: :scriptencoding used outside of a sourced file"));
4284 return;
4285 }
4286
4287 if (*eap->arg != NUL)
4288 {
4289 name = enc_canonize(eap->arg);
4290 if (name == NULL) /* out of memory */
4291 return;
4292 }
4293 else
4294 name = eap->arg;
4295
4296 /* Setup for conversion from the specified encoding to 'encoding'. */
4297 sp = (struct source_cookie *)getline_cookie(eap->getline, eap->cookie);
4298 convert_setup(&sp->conv, name, p_enc);
4299
4300 if (name != eap->arg)
4301 vim_free(name);
4302#endif
4303}
4304
4305#if defined(FEAT_EVAL) || defined(PROTO)
4306/*
4307 * ":finish": Mark a sourced file as finished.
4308 */
4309 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004310ex_finish(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004311{
4312 if (getline_equal(eap->getline, eap->cookie, getsourceline))
4313 do_finish(eap, FALSE);
4314 else
4315 EMSG(_("E168: :finish used outside of a sourced file"));
4316}
4317
4318/*
4319 * Mark a sourced file as finished. Possibly makes the ":finish" pending.
4320 * Also called for a pending finish at the ":endtry" or after returning from
4321 * an extra do_cmdline(). "reanimate" is used in the latter case.
4322 */
4323 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004324do_finish(exarg_T *eap, int reanimate)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004325{
4326 int idx;
4327
4328 if (reanimate)
4329 ((struct source_cookie *)getline_cookie(eap->getline,
4330 eap->cookie))->finished = FALSE;
4331
4332 /*
4333 * Cleanup (and inactivate) conditionals, but stop when a try conditional
4334 * not in its finally clause (which then is to be executed next) is found.
4335 * In this case, make the ":finish" pending for execution at the ":endtry".
4336 * Otherwise, finish normally.
4337 */
4338 idx = cleanup_conditionals(eap->cstack, 0, TRUE);
4339 if (idx >= 0)
4340 {
4341 eap->cstack->cs_pending[idx] = CSTP_FINISH;
4342 report_make_pending(CSTP_FINISH, NULL);
4343 }
4344 else
4345 ((struct source_cookie *)getline_cookie(eap->getline,
4346 eap->cookie))->finished = TRUE;
4347}
4348
4349
4350/*
4351 * Return TRUE when a sourced file had the ":finish" command: Don't give error
4352 * message for missing ":endif".
4353 * Return FALSE when not sourcing a file.
4354 */
4355 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004356source_finished(
4357 char_u *(*fgetline)(int, void *, int),
4358 void *cookie)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004359{
Bram Moolenaar89d40322006-08-29 15:30:07 +00004360 return (getline_equal(fgetline, cookie, getsourceline)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004361 && ((struct source_cookie *)getline_cookie(
Bram Moolenaar89d40322006-08-29 15:30:07 +00004362 fgetline, cookie))->finished);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004363}
4364#endif
4365
4366#if defined(FEAT_LISTCMDS) || defined(PROTO)
4367/*
4368 * ":checktime [buffer]"
4369 */
4370 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004371ex_checktime(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004372{
4373 buf_T *buf;
4374 int save_no_check_timestamps = no_check_timestamps;
4375
4376 no_check_timestamps = 0;
4377 if (eap->addr_count == 0) /* default is all buffers */
4378 check_timestamps(FALSE);
4379 else
4380 {
4381 buf = buflist_findnr((int)eap->line2);
4382 if (buf != NULL) /* cannot happen? */
4383 (void)buf_check_timestamp(buf, FALSE);
4384 }
4385 no_check_timestamps = save_no_check_timestamps;
4386}
4387#endif
4388
Bram Moolenaar071d4272004-06-13 20:20:40 +00004389#if (defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
4390 && (defined(FEAT_EVAL) || defined(FEAT_MULTI_LANG))
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004391# define HAVE_GET_LOCALE_VAL
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004392static char_u *get_locale_val(int what);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004393
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004394 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004395get_locale_val(int what)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004396{
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004397 char_u *loc;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004398
Bram Moolenaar48e330a2016-02-23 14:53:34 +01004399 /* Obtain the locale value from the libraries. */
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004400 loc = (char_u *)setlocale(what, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004401
Bram Moolenaar61660ea2006-04-07 21:40:07 +00004402# ifdef WIN32
Bram Moolenaar071d4272004-06-13 20:20:40 +00004403 if (loc != NULL)
4404 {
4405 char_u *p;
4406
Bram Moolenaar61660ea2006-04-07 21:40:07 +00004407 /* setocale() returns something like "LC_COLLATE=<name>;LC_..." when
4408 * one of the values (e.g., LC_CTYPE) differs. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004409 p = vim_strchr(loc, '=');
4410 if (p != NULL)
4411 {
4412 loc = ++p;
4413 while (*p != NUL) /* remove trailing newline */
4414 {
Bram Moolenaar61660ea2006-04-07 21:40:07 +00004415 if (*p < ' ' || *p == ';')
Bram Moolenaar071d4272004-06-13 20:20:40 +00004416 {
4417 *p = NUL;
4418 break;
4419 }
4420 ++p;
4421 }
4422 }
4423 }
4424# endif
4425
4426 return loc;
4427}
4428#endif
4429
4430
4431#ifdef WIN32
4432/*
4433 * On MS-Windows locale names are strings like "German_Germany.1252", but
4434 * gettext expects "de". Try to translate one into another here for a few
4435 * supported languages.
4436 */
4437 static char_u *
4438gettext_lang(char_u *name)
4439{
4440 int i;
4441 static char *(mtable[]) = {
4442 "afrikaans", "af",
4443 "czech", "cs",
4444 "dutch", "nl",
4445 "german", "de",
4446 "english_united kingdom", "en_GB",
4447 "spanish", "es",
4448 "french", "fr",
4449 "italian", "it",
4450 "japanese", "ja",
4451 "korean", "ko",
4452 "norwegian", "no",
4453 "polish", "pl",
4454 "russian", "ru",
4455 "slovak", "sk",
4456 "swedish", "sv",
4457 "ukrainian", "uk",
4458 "chinese_china", "zh_CN",
4459 "chinese_taiwan", "zh_TW",
4460 NULL};
4461
4462 for (i = 0; mtable[i] != NULL; i += 2)
4463 if (STRNICMP(mtable[i], name, STRLEN(mtable[i])) == 0)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004464 return (char_u *)mtable[i + 1];
Bram Moolenaar071d4272004-06-13 20:20:40 +00004465 return name;
4466}
4467#endif
4468
4469#if defined(FEAT_MULTI_LANG) || defined(PROTO)
4470/*
4471 * Obtain the current messages language. Used to set the default for
4472 * 'helplang'. May return NULL or an empty string.
4473 */
4474 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004475get_mess_lang(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004476{
4477 char_u *p;
4478
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004479# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00004480# if defined(LC_MESSAGES)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004481 p = get_locale_val(LC_MESSAGES);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004482# else
4483 /* This is necessary for Win32, where LC_MESSAGES is not defined and $LANG
Bram Moolenaar61660ea2006-04-07 21:40:07 +00004484 * may be set to the LCID number. LC_COLLATE is the best guess, LC_TIME
4485 * and LC_MONETARY may be set differently for a Japanese working in the
4486 * US. */
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004487 p = get_locale_val(LC_COLLATE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004488# endif
4489# else
4490 p = mch_getenv((char_u *)"LC_ALL");
4491 if (p == NULL || *p == NUL)
4492 {
4493 p = mch_getenv((char_u *)"LC_MESSAGES");
4494 if (p == NULL || *p == NUL)
4495 p = mch_getenv((char_u *)"LANG");
4496 }
4497# endif
4498# ifdef WIN32
4499 p = gettext_lang(p);
4500# endif
4501 return p;
4502}
4503#endif
4504
Bram Moolenaardef9e822004-12-31 20:58:58 +00004505/* Complicated #if; matches with where get_mess_env() is used below. */
4506#if (defined(FEAT_EVAL) && !((defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
4507 && defined(LC_MESSAGES))) \
4508 || ((defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
4509 && (defined(FEAT_GETTEXT) || defined(FEAT_MBYTE)) \
4510 && !defined(LC_MESSAGES))
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01004511static char_u *get_mess_env(void);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004512
4513/*
4514 * Get the language used for messages from the environment.
4515 */
4516 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004517get_mess_env(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004518{
4519 char_u *p;
4520
4521 p = mch_getenv((char_u *)"LC_ALL");
4522 if (p == NULL || *p == NUL)
4523 {
4524 p = mch_getenv((char_u *)"LC_MESSAGES");
4525 if (p == NULL || *p == NUL)
4526 {
4527 p = mch_getenv((char_u *)"LANG");
4528 if (p != NULL && VIM_ISDIGIT(*p))
4529 p = NULL; /* ignore something like "1043" */
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004530# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00004531 if (p == NULL || *p == NUL)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004532 p = get_locale_val(LC_CTYPE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004533# endif
4534 }
4535 }
4536 return p;
4537}
4538#endif
4539
4540#if defined(FEAT_EVAL) || defined(PROTO)
4541
4542/*
4543 * Set the "v:lang" variable according to the current locale setting.
4544 * Also do "v:lc_time"and "v:ctype".
4545 */
4546 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004547set_lang_var(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004548{
4549 char_u *loc;
4550
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004551# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004552 loc = get_locale_val(LC_CTYPE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004553# else
4554 /* setlocale() not supported: use the default value */
4555 loc = (char_u *)"C";
4556# endif
4557 set_vim_var_string(VV_CTYPE, loc, -1);
4558
4559 /* When LC_MESSAGES isn't defined use the value from $LC_MESSAGES, fall
4560 * back to LC_CTYPE if it's empty. */
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004561# if defined(HAVE_GET_LOCALE_VAL) && defined(LC_MESSAGES)
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004562 loc = get_locale_val(LC_MESSAGES);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004563# else
4564 loc = get_mess_env();
4565# endif
4566 set_vim_var_string(VV_LANG, loc, -1);
4567
Bram Moolenaar8765a4a2010-07-27 22:41:43 +02004568# ifdef HAVE_GET_LOCALE_VAL
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01004569 loc = get_locale_val(LC_TIME);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004570# endif
4571 set_vim_var_string(VV_LC_TIME, loc, -1);
4572}
4573#endif
4574
4575#if (defined(HAVE_LOCALE_H) || defined(X_LOCALE)) \
4576 && (defined(FEAT_GETTEXT) || defined(FEAT_MBYTE))
4577/*
4578 * ":language": Set the language (locale).
4579 */
4580 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004581ex_language(exarg_T *eap)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004582{
4583 char *loc;
4584 char_u *p;
4585 char_u *name;
4586 int what = LC_ALL;
4587 char *whatstr = "";
4588#ifdef LC_MESSAGES
4589# define VIM_LC_MESSAGES LC_MESSAGES
4590#else
4591# define VIM_LC_MESSAGES 6789
4592#endif
4593
4594 name = eap->arg;
4595
4596 /* Check for "messages {name}", "ctype {name}" or "time {name}" argument.
4597 * Allow abbreviation, but require at least 3 characters to avoid
4598 * confusion with a two letter language name "me" or "ct". */
4599 p = skiptowhite(eap->arg);
4600 if ((*p == NUL || vim_iswhite(*p)) && p - eap->arg >= 3)
4601 {
4602 if (STRNICMP(eap->arg, "messages", p - eap->arg) == 0)
4603 {
4604 what = VIM_LC_MESSAGES;
4605 name = skipwhite(p);
4606 whatstr = "messages ";
4607 }
4608 else if (STRNICMP(eap->arg, "ctype", p - eap->arg) == 0)
4609 {
4610 what = LC_CTYPE;
4611 name = skipwhite(p);
4612 whatstr = "ctype ";
4613 }
4614 else if (STRNICMP(eap->arg, "time", p - eap->arg) == 0)
4615 {
4616 what = LC_TIME;
4617 name = skipwhite(p);
4618 whatstr = "time ";
4619 }
4620 }
4621
4622 if (*name == NUL)
4623 {
4624#ifndef LC_MESSAGES
4625 if (what == VIM_LC_MESSAGES)
4626 p = get_mess_env();
4627 else
4628#endif
4629 p = (char_u *)setlocale(what, NULL);
4630 if (p == NULL || *p == NUL)
4631 p = (char_u *)"Unknown";
4632 smsg((char_u *)_("Current %slanguage: \"%s\""), whatstr, p);
4633 }
4634 else
4635 {
4636#ifndef LC_MESSAGES
4637 if (what == VIM_LC_MESSAGES)
4638 loc = "";
4639 else
4640#endif
Bram Moolenaar8c8de832008-06-24 22:58:06 +00004641 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00004642 loc = setlocale(what, (char *)name);
Bram Moolenaar8c8de832008-06-24 22:58:06 +00004643#if defined(FEAT_FLOAT) && defined(LC_NUMERIC)
4644 /* Make sure strtod() uses a decimal point, not a comma. */
4645 setlocale(LC_NUMERIC, "C");
4646#endif
4647 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004648 if (loc == NULL)
4649 EMSG2(_("E197: Cannot set language to \"%s\""), name);
4650 else
4651 {
4652#ifdef HAVE_NL_MSG_CAT_CNTR
4653 /* Need to do this for GNU gettext, otherwise cached translations
4654 * will be used again. */
4655 extern int _nl_msg_cat_cntr;
4656
4657 ++_nl_msg_cat_cntr;
4658#endif
Bram Moolenaarb8017e72007-05-10 18:59:07 +00004659 /* Reset $LC_ALL, otherwise it would overrule everything. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004660 vim_setenv((char_u *)"LC_ALL", (char_u *)"");
4661
4662 if (what != LC_TIME)
4663 {
4664 /* Tell gettext() what to translate to. It apparently doesn't
4665 * use the currently effective locale. Also do this when
4666 * FEAT_GETTEXT isn't defined, so that shell commands use this
4667 * value. */
4668 if (what == LC_ALL)
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00004669 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00004670 vim_setenv((char_u *)"LANG", name);
Bram Moolenaare3a0b532013-06-28 20:36:30 +02004671
4672 /* Clear $LANGUAGE because GNU gettext uses it. */
4673 vim_setenv((char_u *)"LANGUAGE", (char_u *)"");
Bram Moolenaar482aaeb2005-09-29 18:26:07 +00004674# ifdef WIN32
4675 /* Apparently MS-Windows printf() may cause a crash when
4676 * we give it 8-bit text while it's expecting text in the
4677 * current locale. This call avoids that. */
4678 setlocale(LC_CTYPE, "C");
4679# endif
4680 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004681 if (what != LC_CTYPE)
4682 {
4683 char_u *mname;
4684#ifdef WIN32
4685 mname = gettext_lang(name);
4686#else
4687 mname = name;
4688#endif
4689 vim_setenv((char_u *)"LC_MESSAGES", mname);
4690#ifdef FEAT_MULTI_LANG
4691 set_helplang_default(mname);
4692#endif
4693 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004694 }
4695
4696# ifdef FEAT_EVAL
4697 /* Set v:lang, v:lc_time and v:ctype to the final result. */
4698 set_lang_var();
4699# endif
Bram Moolenaar6cc00c72011-10-20 21:41:09 +02004700# ifdef FEAT_TITLE
4701 maketitle();
4702# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004703 }
4704 }
4705}
4706
4707# if defined(FEAT_CMDL_COMPL) || defined(PROTO)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02004708
4709static char_u **locales = NULL; /* Array of all available locales */
4710static int did_init_locales = FALSE;
4711
Bram Moolenaarf28dbce2016-01-29 22:03:47 +01004712static void init_locales(void);
4713static char_u **find_locales(void);
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02004714
4715/*
4716 * Lazy initialization of all available locales.
4717 */
4718 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004719init_locales(void)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02004720{
4721 if (!did_init_locales)
4722 {
4723 did_init_locales = TRUE;
4724 locales = find_locales();
4725 }
4726}
4727
4728/* Return an array of strings for all available locales + NULL for the
4729 * last element. Return NULL in case of error. */
4730 static char_u **
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004731find_locales(void)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02004732{
4733 garray_T locales_ga;
4734 char_u *loc;
4735
4736 /* Find all available locales by running command "locale -a". If this
4737 * doesn't work we won't have completion. */
4738 char_u *locale_a = get_cmd_output((char_u *)"locale -a",
Bram Moolenaar39c29ed2014-04-05 19:44:40 +02004739 NULL, SHELL_SILENT, NULL);
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02004740 if (locale_a == NULL)
4741 return NULL;
4742 ga_init2(&locales_ga, sizeof(char_u *), 20);
4743
4744 /* Transform locale_a string where each locale is separated by "\n"
4745 * into an array of locale strings. */
4746 loc = (char_u *)strtok((char *)locale_a, "\n");
4747
4748 while (loc != NULL)
4749 {
4750 if (ga_grow(&locales_ga, 1) == FAIL)
4751 break;
4752 loc = vim_strsave(loc);
4753 if (loc == NULL)
4754 break;
4755
4756 ((char_u **)locales_ga.ga_data)[locales_ga.ga_len++] = loc;
4757 loc = (char_u *)strtok(NULL, "\n");
4758 }
4759 vim_free(locale_a);
4760 if (ga_grow(&locales_ga, 1) == FAIL)
4761 {
4762 ga_clear(&locales_ga);
4763 return NULL;
4764 }
4765 ((char_u **)locales_ga.ga_data)[locales_ga.ga_len] = NULL;
4766 return (char_u **)locales_ga.ga_data;
4767}
4768
4769# if defined(EXITFREE) || defined(PROTO)
4770 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004771free_locales(void)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02004772{
4773 int i;
4774 if (locales != NULL)
4775 {
4776 for (i = 0; locales[i] != NULL; i++)
4777 vim_free(locales[i]);
4778 vim_free(locales);
4779 locales = NULL;
4780 }
4781}
4782# endif
4783
Bram Moolenaar071d4272004-06-13 20:20:40 +00004784/*
4785 * Function given to ExpandGeneric() to obtain the possible arguments of the
4786 * ":language" command.
4787 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004788 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004789get_lang_arg(expand_T *xp UNUSED, int idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004790{
4791 if (idx == 0)
4792 return (char_u *)"messages";
4793 if (idx == 1)
4794 return (char_u *)"ctype";
4795 if (idx == 2)
4796 return (char_u *)"time";
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02004797
4798 init_locales();
4799 if (locales == NULL)
4800 return NULL;
4801 return locales[idx - 3];
4802}
4803
4804/*
4805 * Function given to ExpandGeneric() to obtain the available locales.
4806 */
4807 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01004808get_locales(expand_T *xp UNUSED, int idx)
Bram Moolenaar9b486ca2011-05-19 18:26:40 +02004809{
4810 init_locales();
4811 if (locales == NULL)
4812 return NULL;
4813 return locales[idx];
Bram Moolenaar071d4272004-06-13 20:20:40 +00004814}
4815# endif
4816
4817#endif