Merge branch 'vim-with-runtime' into feat/persistent-undo
[vim_extended.git] / src / fileio.c
blobc36796972530a7c1f9b1b8ef96fc0467e1d3e585
1 /* vi:set ts=8 sts=4 sw=4:
3 * VIM - Vi IMproved by Bram Moolenaar
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 */
11 * fileio.c: read from and write to a file
14 #if defined(MSDOS) || defined(WIN16) || defined(WIN32) || defined(_WIN64)
15 # include "vimio.h" /* for lseek(), must be before vim.h */
16 #endif
18 #if defined __EMX__
19 # include "vimio.h" /* for mktemp(), CJW 1997-12-03 */
20 #endif
22 #include "vim.h"
24 #if defined(__TANDEM) || defined(__MINT__)
25 # include <limits.h> /* for SSIZE_MAX */
26 #endif
28 #if defined(HAVE_UTIME) && defined(HAVE_UTIME_H)
29 # include <utime.h> /* for struct utimbuf */
30 #endif
32 #define BUFSIZE 8192 /* size of normal write buffer */
33 #define SMBUFSIZE 256 /* size of emergency write buffer */
35 #ifdef FEAT_CRYPT
36 # define CRYPT_MAGIC "VimCrypt~01!" /* "01" is the version nr */
37 # define CRYPT_MAGIC_LEN 12 /* must be multiple of 4! */
38 #endif
40 /* Is there any system that doesn't have access()? */
41 #define USE_MCH_ACCESS
43 #if defined(sun) && defined(S_ISCHR)
44 # define OPEN_CHR_FILES
45 static int is_dev_fd_file(char_u *fname);
46 #endif
47 #ifdef FEAT_MBYTE
48 static char_u *next_fenc __ARGS((char_u **pp));
49 # ifdef FEAT_EVAL
50 static char_u *readfile_charconvert __ARGS((char_u *fname, char_u *fenc, int *fdp));
51 # endif
52 #endif
53 #ifdef FEAT_VIMINFO
54 static void check_marks_read __ARGS((void));
55 #endif
56 #ifdef FEAT_CRYPT
57 static char_u *check_for_cryptkey __ARGS((char_u *cryptkey, char_u *ptr, long *sizep, long *filesizep, int newfile));
58 #endif
59 #ifdef UNIX
60 static void set_file_time __ARGS((char_u *fname, time_t atime, time_t mtime));
61 #endif
62 static int set_rw_fname __ARGS((char_u *fname, char_u *sfname));
63 static int msg_add_fileformat __ARGS((int eol_type));
64 static void msg_add_eol __ARGS((void));
65 static int check_mtime __ARGS((buf_T *buf, struct stat *s));
66 static int time_differs __ARGS((long t1, long t2));
67 #ifdef FEAT_AUTOCMD
68 static int apply_autocmds_exarg __ARGS((event_T event, char_u *fname, char_u *fname_io, int force, buf_T *buf, exarg_T *eap));
69 static int au_find_group __ARGS((char_u *name));
71 # define AUGROUP_DEFAULT -1 /* default autocmd group */
72 # define AUGROUP_ERROR -2 /* erroneous autocmd group */
73 # define AUGROUP_ALL -3 /* all autocmd groups */
74 #endif
76 #if defined(FEAT_CRYPT) || defined(FEAT_MBYTE)
77 # define HAS_BW_FLAGS
78 # define FIO_LATIN1 0x01 /* convert Latin1 */
79 # define FIO_UTF8 0x02 /* convert UTF-8 */
80 # define FIO_UCS2 0x04 /* convert UCS-2 */
81 # define FIO_UCS4 0x08 /* convert UCS-4 */
82 # define FIO_UTF16 0x10 /* convert UTF-16 */
83 # ifdef WIN3264
84 # define FIO_CODEPAGE 0x20 /* convert MS-Windows codepage */
85 # define FIO_PUT_CP(x) (((x) & 0xffff) << 16) /* put codepage in top word */
86 # define FIO_GET_CP(x) (((x)>>16) & 0xffff) /* get codepage from top word */
87 # endif
88 # ifdef MACOS_X
89 # define FIO_MACROMAN 0x20 /* convert MacRoman */
90 # endif
91 # define FIO_ENDIAN_L 0x80 /* little endian */
92 # define FIO_ENCRYPTED 0x1000 /* encrypt written bytes */
93 # define FIO_NOCONVERT 0x2000 /* skip encoding conversion */
94 # define FIO_UCSBOM 0x4000 /* check for BOM at start of file */
95 # define FIO_ALL -1 /* allow all formats */
96 #endif
98 /* When converting, a read() or write() may leave some bytes to be converted
99 * for the next call. The value is guessed... */
100 #define CONV_RESTLEN 30
102 /* We have to guess how much a sequence of bytes may expand when converting
103 * with iconv() to be able to allocate a buffer. */
104 #define ICONV_MULT 8
107 * Structure to pass arguments from buf_write() to buf_write_bytes().
109 struct bw_info
111 int bw_fd; /* file descriptor */
112 char_u *bw_buf; /* buffer with data to be written */
113 int bw_len; /* length of data */
114 #ifdef HAS_BW_FLAGS
115 int bw_flags; /* FIO_ flags */
116 #endif
117 #ifdef FEAT_MBYTE
118 char_u bw_rest[CONV_RESTLEN]; /* not converted bytes */
119 int bw_restlen; /* nr of bytes in bw_rest[] */
120 int bw_first; /* first write call */
121 char_u *bw_conv_buf; /* buffer for writing converted chars */
122 int bw_conv_buflen; /* size of bw_conv_buf */
123 int bw_conv_error; /* set for conversion error */
124 linenr_T bw_conv_error_lnum; /* first line with error or zero */
125 linenr_T bw_start_lnum; /* line number at start of buffer */
126 # ifdef USE_ICONV
127 iconv_t bw_iconv_fd; /* descriptor for iconv() or -1 */
128 # endif
129 #endif
132 static int buf_write_bytes __ARGS((struct bw_info *ip));
134 #ifdef FEAT_MBYTE
135 static linenr_T readfile_linenr __ARGS((linenr_T linecnt, char_u *p, char_u *endp));
136 static int ucs2bytes __ARGS((unsigned c, char_u **pp, int flags));
137 static int need_conversion __ARGS((char_u *fenc));
138 static int get_fio_flags __ARGS((char_u *ptr));
139 static char_u *check_for_bom __ARGS((char_u *p, long size, int *lenp, int flags));
140 static int make_bom __ARGS((char_u *buf, char_u *name));
141 # ifdef WIN3264
142 static int get_win_fio_flags __ARGS((char_u *ptr));
143 # endif
144 # ifdef MACOS_X
145 static int get_mac_fio_flags __ARGS((char_u *ptr));
146 # endif
147 #endif
148 static int move_lines __ARGS((buf_T *frombuf, buf_T *tobuf));
149 #ifdef TEMPDIRNAMES
150 static void vim_settempdir __ARGS((char_u *tempdir));
151 #endif
152 #ifdef FEAT_AUTOCMD
153 static char *e_auchangedbuf = N_("E812: Autocommands changed buffer or buffer name");
154 #endif
156 void
157 filemess(buf, name, s, attr)
158 buf_T *buf;
159 char_u *name;
160 char_u *s;
161 int attr;
163 int msg_scroll_save;
165 if (msg_silent != 0)
166 return;
167 msg_add_fname(buf, name); /* put file name in IObuff with quotes */
168 /* If it's extremely long, truncate it. */
169 if (STRLEN(IObuff) > IOSIZE - 80)
170 IObuff[IOSIZE - 80] = NUL;
171 STRCAT(IObuff, s);
173 * For the first message may have to start a new line.
174 * For further ones overwrite the previous one, reset msg_scroll before
175 * calling filemess().
177 msg_scroll_save = msg_scroll;
178 if (shortmess(SHM_OVERALL) && !exiting && p_verbose == 0)
179 msg_scroll = FALSE;
180 if (!msg_scroll) /* wait a bit when overwriting an error msg */
181 check_for_delay(FALSE);
182 msg_start();
183 msg_scroll = msg_scroll_save;
184 msg_scrolled_ign = TRUE;
185 /* may truncate the message to avoid a hit-return prompt */
186 msg_outtrans_attr(msg_may_trunc(FALSE, IObuff), attr);
187 msg_clr_eos();
188 out_flush();
189 msg_scrolled_ign = FALSE;
193 * Read lines from file "fname" into the buffer after line "from".
195 * 1. We allocate blocks with lalloc, as big as possible.
196 * 2. Each block is filled with characters from the file with a single read().
197 * 3. The lines are inserted in the buffer with ml_append().
199 * (caller must check that fname != NULL, unless READ_STDIN is used)
201 * "lines_to_skip" is the number of lines that must be skipped
202 * "lines_to_read" is the number of lines that are appended
203 * When not recovering lines_to_skip is 0 and lines_to_read MAXLNUM.
205 * flags:
206 * READ_NEW starting to edit a new buffer
207 * READ_FILTER reading filter output
208 * READ_STDIN read from stdin instead of a file
209 * READ_BUFFER read from curbuf instead of a file (converting after reading
210 * stdin)
211 * READ_DUMMY read into a dummy buffer (to check if file contents changed)
213 * return FAIL for failure, OK otherwise
216 readfile(fname, sfname, from, lines_to_skip, lines_to_read, eap, flags)
217 char_u *fname;
218 char_u *sfname;
219 linenr_T from;
220 linenr_T lines_to_skip;
221 linenr_T lines_to_read;
222 exarg_T *eap; /* can be NULL! */
223 int flags;
225 int fd = 0;
226 int newfile = (flags & READ_NEW);
227 int check_readonly;
228 int filtering = (flags & READ_FILTER);
229 int read_stdin = (flags & READ_STDIN);
230 int read_buffer = (flags & READ_BUFFER);
231 int set_options = newfile || read_buffer
232 || (eap != NULL && eap->read_edit);
233 linenr_T read_buf_lnum = 1; /* next line to read from curbuf */
234 colnr_T read_buf_col = 0; /* next char to read from this line */
235 char_u c;
236 linenr_T lnum = from;
237 char_u *ptr = NULL; /* pointer into read buffer */
238 char_u *buffer = NULL; /* read buffer */
239 char_u *new_buffer = NULL; /* init to shut up gcc */
240 char_u *line_start = NULL; /* init to shut up gcc */
241 int wasempty; /* buffer was empty before reading */
242 colnr_T len;
243 long size = 0;
244 char_u *p;
245 long filesize = 0;
246 int skip_read = FALSE;
247 #ifdef FEAT_CRYPT
248 char_u *cryptkey = NULL;
249 #endif
250 int split = 0; /* number of split lines */
251 #define UNKNOWN 0x0fffffff /* file size is unknown */
252 linenr_T linecnt;
253 int error = FALSE; /* errors encountered */
254 int ff_error = EOL_UNKNOWN; /* file format with errors */
255 long linerest = 0; /* remaining chars in line */
256 #ifdef UNIX
257 int perm = 0;
258 int swap_mode = -1; /* protection bits for swap file */
259 #else
260 int perm;
261 #endif
262 int fileformat = 0; /* end-of-line format */
263 int keep_fileformat = FALSE;
264 struct stat st;
265 int file_readonly;
266 linenr_T skip_count = 0;
267 linenr_T read_count = 0;
268 int msg_save = msg_scroll;
269 linenr_T read_no_eol_lnum = 0; /* non-zero lnum when last line of
270 * last read was missing the eol */
271 int try_mac = (vim_strchr(p_ffs, 'm') != NULL);
272 int try_dos = (vim_strchr(p_ffs, 'd') != NULL);
273 int try_unix = (vim_strchr(p_ffs, 'x') != NULL);
274 int file_rewind = FALSE;
275 #ifdef FEAT_MBYTE
276 int can_retry;
277 linenr_T conv_error = 0; /* line nr with conversion error */
278 linenr_T illegal_byte = 0; /* line nr with illegal byte */
279 int keep_dest_enc = FALSE; /* don't retry when char doesn't fit
280 in destination encoding */
281 int bad_char_behavior = BAD_REPLACE;
282 /* BAD_KEEP, BAD_DROP or character to
283 * replace with */
284 char_u *tmpname = NULL; /* name of 'charconvert' output file */
285 int fio_flags = 0;
286 char_u *fenc; /* fileencoding to use */
287 int fenc_alloced; /* fenc_next is in allocated memory */
288 char_u *fenc_next = NULL; /* next item in 'fencs' or NULL */
289 int advance_fenc = FALSE;
290 long real_size = 0;
291 # ifdef USE_ICONV
292 iconv_t iconv_fd = (iconv_t)-1; /* descriptor for iconv() or -1 */
293 # ifdef FEAT_EVAL
294 int did_iconv = FALSE; /* TRUE when iconv() failed and trying
295 'charconvert' next */
296 # endif
297 # endif
298 int converted = FALSE; /* TRUE if conversion done */
299 int notconverted = FALSE; /* TRUE if conversion wanted but it
300 wasn't possible */
301 char_u conv_rest[CONV_RESTLEN];
302 int conv_restlen = 0; /* nr of bytes in conv_rest[] */
303 #endif
305 #ifdef FEAT_AUTOCMD
306 /* Remember the initial values of curbuf, curbuf->b_ffname and
307 * curbuf->b_fname to detect whether they are altered as a result of
308 * executing nasty autocommands. Also check if "fname" and "sfname"
309 * point to one of these values. */
310 buf_T *old_curbuf = curbuf;
311 char_u *old_b_ffname = curbuf->b_ffname;
312 char_u *old_b_fname = curbuf->b_fname;
313 int using_b_ffname = (fname == curbuf->b_ffname)
314 || (sfname == curbuf->b_ffname);
315 int using_b_fname = (fname == curbuf->b_fname)
316 || (sfname == curbuf->b_fname);
317 #endif
318 write_no_eol_lnum = 0; /* in case it was set by the previous read */
321 * If there is no file name yet, use the one for the read file.
322 * BF_NOTEDITED is set to reflect this.
323 * Don't do this for a read from a filter.
324 * Only do this when 'cpoptions' contains the 'f' flag.
326 if (curbuf->b_ffname == NULL
327 && !filtering
328 && fname != NULL
329 && vim_strchr(p_cpo, CPO_FNAMER) != NULL
330 && !(flags & READ_DUMMY))
332 if (set_rw_fname(fname, sfname) == FAIL)
333 return FAIL;
336 /* After reading a file the cursor line changes but we don't want to
337 * display the line. */
338 ex_no_reprint = TRUE;
340 /* don't display the file info for another buffer now */
341 need_fileinfo = FALSE;
344 * For Unix: Use the short file name whenever possible.
345 * Avoids problems with networks and when directory names are changed.
346 * Don't do this for MS-DOS, a "cd" in a sub-shell may have moved us to
347 * another directory, which we don't detect.
349 if (sfname == NULL)
350 sfname = fname;
351 #if defined(UNIX) || defined(__EMX__)
352 fname = sfname;
353 #endif
355 #ifdef FEAT_AUTOCMD
357 * The BufReadCmd and FileReadCmd events intercept the reading process by
358 * executing the associated commands instead.
360 if (!filtering && !read_stdin && !read_buffer)
362 pos_T pos;
364 pos = curbuf->b_op_start;
366 /* Set '[ mark to the line above where the lines go (line 1 if zero). */
367 curbuf->b_op_start.lnum = ((from == 0) ? 1 : from);
368 curbuf->b_op_start.col = 0;
370 if (newfile)
372 if (apply_autocmds_exarg(EVENT_BUFREADCMD, NULL, sfname,
373 FALSE, curbuf, eap))
374 #ifdef FEAT_EVAL
375 return aborting() ? FAIL : OK;
376 #else
377 return OK;
378 #endif
380 else if (apply_autocmds_exarg(EVENT_FILEREADCMD, sfname, sfname,
381 FALSE, NULL, eap))
382 #ifdef FEAT_EVAL
383 return aborting() ? FAIL : OK;
384 #else
385 return OK;
386 #endif
388 curbuf->b_op_start = pos;
390 #endif
392 if ((shortmess(SHM_OVER) || curbuf->b_help) && p_verbose == 0)
393 msg_scroll = FALSE; /* overwrite previous file message */
394 else
395 msg_scroll = TRUE; /* don't overwrite previous file message */
398 * If the name ends in a path separator, we can't open it. Check here,
399 * because reading the file may actually work, but then creating the swap
400 * file may destroy it! Reported on MS-DOS and Win 95.
401 * If the name is too long we might crash further on, quit here.
403 if (fname != NULL && *fname != NUL)
405 p = fname + STRLEN(fname);
406 if (after_pathsep(fname, p) || STRLEN(fname) >= MAXPATHL)
408 filemess(curbuf, fname, (char_u *)_("Illegal file name"), 0);
409 msg_end();
410 msg_scroll = msg_save;
411 return FAIL;
415 #ifdef UNIX
417 * On Unix it is possible to read a directory, so we have to
418 * check for it before the mch_open().
420 if (!read_stdin && !read_buffer)
422 perm = mch_getperm(fname);
423 if (perm >= 0 && !S_ISREG(perm) /* not a regular file ... */
424 # ifdef S_ISFIFO
425 && !S_ISFIFO(perm) /* ... or fifo */
426 # endif
427 # ifdef S_ISSOCK
428 && !S_ISSOCK(perm) /* ... or socket */
429 # endif
430 # ifdef OPEN_CHR_FILES
431 && !(S_ISCHR(perm) && is_dev_fd_file(fname))
432 /* ... or a character special file named /dev/fd/<n> */
433 # endif
436 if (S_ISDIR(perm))
437 filemess(curbuf, fname, (char_u *)_("is a directory"), 0);
438 else
439 filemess(curbuf, fname, (char_u *)_("is not a file"), 0);
440 msg_end();
441 msg_scroll = msg_save;
442 return FAIL;
445 # if defined(MSDOS) || defined(MSWIN) || defined(OS2)
447 * MS-Windows allows opening a device, but we will probably get stuck
448 * trying to read it.
450 if (!p_odev && mch_nodetype(fname) == NODE_WRITABLE)
452 filemess(curbuf, fname, (char_u *)_("is a device (disabled with 'opendevice' option)"), 0);
453 msg_end();
454 msg_scroll = msg_save;
455 return FAIL;
457 # endif
459 #endif
461 /* set default 'fileformat' */
462 if (set_options)
464 if (eap != NULL && eap->force_ff != 0)
465 set_fileformat(get_fileformat_force(curbuf, eap), OPT_LOCAL);
466 else if (*p_ffs != NUL)
467 set_fileformat(default_fileformat(), OPT_LOCAL);
470 /* set or reset 'binary' */
471 if (eap != NULL && eap->force_bin != 0)
473 int oldval = curbuf->b_p_bin;
475 curbuf->b_p_bin = (eap->force_bin == FORCE_BIN);
476 set_options_bin(oldval, curbuf->b_p_bin, OPT_LOCAL);
480 * When opening a new file we take the readonly flag from the file.
481 * Default is r/w, can be set to r/o below.
482 * Don't reset it when in readonly mode
483 * Only set/reset b_p_ro when BF_CHECK_RO is set.
485 check_readonly = (newfile && (curbuf->b_flags & BF_CHECK_RO));
486 if (check_readonly && !readonlymode)
487 curbuf->b_p_ro = FALSE;
489 if (newfile && !read_stdin && !read_buffer)
491 /* Remember time of file.
492 * For RISCOS, also remember the filetype.
494 if (mch_stat((char *)fname, &st) >= 0)
496 buf_store_time(curbuf, &st, fname);
497 curbuf->b_mtime_read = curbuf->b_mtime;
499 #if defined(RISCOS) && defined(FEAT_OSFILETYPE)
500 /* Read the filetype into the buffer local filetype option. */
501 mch_read_filetype(fname);
502 #endif
503 #ifdef UNIX
505 * Use the protection bits of the original file for the swap file.
506 * This makes it possible for others to read the name of the
507 * edited file from the swapfile, but only if they can read the
508 * edited file.
509 * Remove the "write" and "execute" bits for group and others
510 * (they must not write the swapfile).
511 * Add the "read" and "write" bits for the user, otherwise we may
512 * not be able to write to the file ourselves.
513 * Setting the bits is done below, after creating the swap file.
515 swap_mode = (st.st_mode & 0644) | 0600;
516 #endif
517 #ifdef FEAT_CW_EDITOR
518 /* Get the FSSpec on MacOS
519 * TODO: Update it properly when the buffer name changes
521 (void)GetFSSpecFromPath(curbuf->b_ffname, &curbuf->b_FSSpec);
522 #endif
523 #ifdef VMS
524 curbuf->b_fab_rfm = st.st_fab_rfm;
525 curbuf->b_fab_rat = st.st_fab_rat;
526 curbuf->b_fab_mrs = st.st_fab_mrs;
527 #endif
529 else
531 curbuf->b_mtime = 0;
532 curbuf->b_mtime_read = 0;
533 curbuf->b_orig_size = 0;
534 curbuf->b_orig_mode = 0;
537 /* Reset the "new file" flag. It will be set again below when the
538 * file doesn't exist. */
539 curbuf->b_flags &= ~(BF_NEW | BF_NEW_W);
543 * for UNIX: check readonly with perm and mch_access()
544 * for RISCOS: same as Unix, otherwise file gets re-datestamped!
545 * for MSDOS and Amiga: check readonly by trying to open the file for writing
547 file_readonly = FALSE;
548 if (read_stdin)
550 #if defined(MSDOS) || defined(MSWIN) || defined(OS2)
551 /* Force binary I/O on stdin to avoid CR-LF -> LF conversion. */
552 setmode(0, O_BINARY);
553 #endif
555 else if (!read_buffer)
557 #ifdef USE_MCH_ACCESS
558 if (
559 # ifdef UNIX
560 !(perm & 0222) ||
561 # endif
562 mch_access((char *)fname, W_OK))
563 file_readonly = TRUE;
564 fd = mch_open((char *)fname, O_RDONLY | O_EXTRA, 0);
565 #else
566 if (!newfile
567 || readonlymode
568 || (fd = mch_open((char *)fname, O_RDWR | O_EXTRA, 0)) < 0)
570 file_readonly = TRUE;
571 /* try to open ro */
572 fd = mch_open((char *)fname, O_RDONLY | O_EXTRA, 0);
574 #endif
577 if (fd < 0) /* cannot open at all */
579 #ifndef UNIX
580 int isdir_f;
581 #endif
582 msg_scroll = msg_save;
583 #ifndef UNIX
585 * On MSDOS and Amiga we can't open a directory, check here.
587 isdir_f = (mch_isdir(fname));
588 perm = mch_getperm(fname); /* check if the file exists */
589 if (isdir_f)
591 filemess(curbuf, sfname, (char_u *)_("is a directory"), 0);
592 curbuf->b_p_ro = TRUE; /* must use "w!" now */
594 else
595 #endif
596 if (newfile)
598 if (perm < 0
599 #ifdef ENOENT
600 && errno == ENOENT
601 #endif
605 * Set the 'new-file' flag, so that when the file has
606 * been created by someone else, a ":w" will complain.
608 curbuf->b_flags |= BF_NEW;
610 /* Create a swap file now, so that other Vims are warned
611 * that we are editing this file. Don't do this for a
612 * "nofile" or "nowrite" buffer type. */
613 #ifdef FEAT_QUICKFIX
614 if (!bt_dontwrite(curbuf))
615 #endif
617 check_need_swap(newfile);
618 #ifdef FEAT_AUTOCMD
619 /* SwapExists autocommand may mess things up */
620 if (curbuf != old_curbuf
621 || (using_b_ffname
622 && (old_b_ffname != curbuf->b_ffname))
623 || (using_b_fname
624 && (old_b_fname != curbuf->b_fname)))
626 EMSG(_(e_auchangedbuf));
627 return FAIL;
629 #endif
631 if (dir_of_file_exists(fname))
632 filemess(curbuf, sfname, (char_u *)_("[New File]"), 0);
633 else
634 filemess(curbuf, sfname,
635 (char_u *)_("[New DIRECTORY]"), 0);
636 #ifdef FEAT_VIMINFO
637 /* Even though this is a new file, it might have been
638 * edited before and deleted. Get the old marks. */
639 check_marks_read();
640 #endif
641 #ifdef FEAT_MBYTE
642 if (eap != NULL && eap->force_enc != 0)
644 /* set forced 'fileencoding' */
645 fenc = enc_canonize(eap->cmd + eap->force_enc);
646 if (fenc != NULL)
647 set_string_option_direct((char_u *)"fenc", -1,
648 fenc, OPT_FREE|OPT_LOCAL, 0);
649 vim_free(fenc);
651 #endif
652 #ifdef FEAT_AUTOCMD
653 apply_autocmds_exarg(EVENT_BUFNEWFILE, sfname, sfname,
654 FALSE, curbuf, eap);
655 #endif
656 /* remember the current fileformat */
657 save_file_ff(curbuf);
659 #if defined(FEAT_AUTOCMD) && defined(FEAT_EVAL)
660 if (aborting()) /* autocmds may abort script processing */
661 return FAIL;
662 #endif
663 return OK; /* a new file is not an error */
665 else
667 filemess(curbuf, sfname, (char_u *)(
668 # ifdef EFBIG
669 (errno == EFBIG) ? _("[File too big]") :
670 # endif
671 # ifdef EOVERFLOW
672 (errno == EOVERFLOW) ? _("[File too big]") :
673 # endif
674 _("[Permission Denied]")), 0);
675 curbuf->b_p_ro = TRUE; /* must use "w!" now */
679 return FAIL;
683 * Only set the 'ro' flag for readonly files the first time they are
684 * loaded. Help files always get readonly mode
686 if ((check_readonly && file_readonly) || curbuf->b_help)
687 curbuf->b_p_ro = TRUE;
689 if (set_options)
691 /* Don't change 'eol' if reading from buffer as it will already be
692 * correctly set when reading stdin. */
693 if (!read_buffer)
695 curbuf->b_p_eol = TRUE;
696 curbuf->b_start_eol = TRUE;
698 #ifdef FEAT_MBYTE
699 curbuf->b_p_bomb = FALSE;
700 curbuf->b_start_bomb = FALSE;
701 #endif
704 /* Create a swap file now, so that other Vims are warned that we are
705 * editing this file.
706 * Don't do this for a "nofile" or "nowrite" buffer type. */
707 #ifdef FEAT_QUICKFIX
708 if (!bt_dontwrite(curbuf))
709 #endif
711 check_need_swap(newfile);
712 #ifdef FEAT_AUTOCMD
713 if (!read_stdin && (curbuf != old_curbuf
714 || (using_b_ffname && (old_b_ffname != curbuf->b_ffname))
715 || (using_b_fname && (old_b_fname != curbuf->b_fname))))
717 EMSG(_(e_auchangedbuf));
718 if (!read_buffer)
719 close(fd);
720 return FAIL;
722 #endif
723 #ifdef UNIX
724 /* Set swap file protection bits after creating it. */
725 if (swap_mode > 0 && curbuf->b_ml.ml_mfp != NULL
726 && curbuf->b_ml.ml_mfp->mf_fname != NULL)
727 (void)mch_setperm(curbuf->b_ml.ml_mfp->mf_fname, (long)swap_mode);
728 #endif
731 #if defined(HAS_SWAP_EXISTS_ACTION)
732 /* If "Quit" selected at ATTENTION dialog, don't load the file */
733 if (swap_exists_action == SEA_QUIT)
735 if (!read_buffer && !read_stdin)
736 close(fd);
737 return FAIL;
739 #endif
741 ++no_wait_return; /* don't wait for return yet */
744 * Set '[ mark to the line above where the lines go (line 1 if zero).
746 curbuf->b_op_start.lnum = ((from == 0) ? 1 : from);
747 curbuf->b_op_start.col = 0;
749 #ifdef FEAT_AUTOCMD
750 if (!read_buffer)
752 int m = msg_scroll;
753 int n = msg_scrolled;
756 * The file must be closed again, the autocommands may want to change
757 * the file before reading it.
759 if (!read_stdin)
760 close(fd); /* ignore errors */
763 * The output from the autocommands should not overwrite anything and
764 * should not be overwritten: Set msg_scroll, restore its value if no
765 * output was done.
767 msg_scroll = TRUE;
768 if (filtering)
769 apply_autocmds_exarg(EVENT_FILTERREADPRE, NULL, sfname,
770 FALSE, curbuf, eap);
771 else if (read_stdin)
772 apply_autocmds_exarg(EVENT_STDINREADPRE, NULL, sfname,
773 FALSE, curbuf, eap);
774 else if (newfile)
775 apply_autocmds_exarg(EVENT_BUFREADPRE, NULL, sfname,
776 FALSE, curbuf, eap);
777 else
778 apply_autocmds_exarg(EVENT_FILEREADPRE, sfname, sfname,
779 FALSE, NULL, eap);
780 if (msg_scrolled == n)
781 msg_scroll = m;
783 #ifdef FEAT_EVAL
784 if (aborting()) /* autocmds may abort script processing */
786 --no_wait_return;
787 msg_scroll = msg_save;
788 curbuf->b_p_ro = TRUE; /* must use "w!" now */
789 return FAIL;
791 #endif
793 * Don't allow the autocommands to change the current buffer.
794 * Try to re-open the file.
796 * Don't allow the autocommands to change the buffer name either
797 * (cd for example) if it invalidates fname or sfname.
799 if (!read_stdin && (curbuf != old_curbuf
800 || (using_b_ffname && (old_b_ffname != curbuf->b_ffname))
801 || (using_b_fname && (old_b_fname != curbuf->b_fname))
802 || (fd = mch_open((char *)fname, O_RDONLY | O_EXTRA, 0)) < 0))
804 --no_wait_return;
805 msg_scroll = msg_save;
806 if (fd < 0)
807 EMSG(_("E200: *ReadPre autocommands made the file unreadable"));
808 else
809 EMSG(_("E201: *ReadPre autocommands must not change current buffer"));
810 curbuf->b_p_ro = TRUE; /* must use "w!" now */
811 return FAIL;
814 #endif /* FEAT_AUTOCMD */
816 /* Autocommands may add lines to the file, need to check if it is empty */
817 wasempty = (curbuf->b_ml.ml_flags & ML_EMPTY);
819 if (!recoverymode && !filtering && !(flags & READ_DUMMY))
822 * Show the user that we are busy reading the input. Sometimes this
823 * may take a while. When reading from stdin another program may
824 * still be running, don't move the cursor to the last line, unless
825 * always using the GUI.
827 if (read_stdin)
829 #ifndef ALWAYS_USE_GUI
830 mch_msg(_("Vim: Reading from stdin...\n"));
831 #endif
832 #ifdef FEAT_GUI
833 /* Also write a message in the GUI window, if there is one. */
834 if (gui.in_use && !gui.dying && !gui.starting)
836 p = (char_u *)_("Reading from stdin...");
837 gui_write(p, (int)STRLEN(p));
839 #endif
841 else if (!read_buffer)
842 filemess(curbuf, sfname, (char_u *)"", 0);
845 msg_scroll = FALSE; /* overwrite the file message */
848 * Set linecnt now, before the "retry" caused by a wrong guess for
849 * fileformat, and after the autocommands, which may change them.
851 linecnt = curbuf->b_ml.ml_line_count;
853 #ifdef FEAT_MBYTE
854 /* "++bad=" argument. */
855 if (eap != NULL && eap->bad_char != 0)
857 bad_char_behavior = eap->bad_char;
858 if (set_options)
859 curbuf->b_bad_char = eap->bad_char;
861 else
862 curbuf->b_bad_char = 0;
865 * Decide which 'encoding' to use or use first.
867 if (eap != NULL && eap->force_enc != 0)
869 fenc = enc_canonize(eap->cmd + eap->force_enc);
870 fenc_alloced = TRUE;
871 keep_dest_enc = TRUE;
873 else if (curbuf->b_p_bin)
875 fenc = (char_u *)""; /* binary: don't convert */
876 fenc_alloced = FALSE;
878 else if (curbuf->b_help)
880 char_u firstline[80];
881 int fc;
883 /* Help files are either utf-8 or latin1. Try utf-8 first, if this
884 * fails it must be latin1.
885 * Always do this when 'encoding' is "utf-8". Otherwise only do
886 * this when needed to avoid [converted] remarks all the time.
887 * It is needed when the first line contains non-ASCII characters.
888 * That is only in *.??x files. */
889 fenc = (char_u *)"latin1";
890 c = enc_utf8;
891 if (!c && !read_stdin)
893 fc = fname[STRLEN(fname) - 1];
894 if (TOLOWER_ASC(fc) == 'x')
896 /* Read the first line (and a bit more). Immediately rewind to
897 * the start of the file. If the read() fails "len" is -1. */
898 len = vim_read(fd, firstline, 80);
899 lseek(fd, (off_t)0L, SEEK_SET);
900 for (p = firstline; p < firstline + len; ++p)
901 if (*p >= 0x80)
903 c = TRUE;
904 break;
909 if (c)
911 fenc_next = fenc;
912 fenc = (char_u *)"utf-8";
914 /* When the file is utf-8 but a character doesn't fit in
915 * 'encoding' don't retry. In help text editing utf-8 bytes
916 * doesn't make sense. */
917 if (!enc_utf8)
918 keep_dest_enc = TRUE;
920 fenc_alloced = FALSE;
922 else if (*p_fencs == NUL)
924 fenc = curbuf->b_p_fenc; /* use format from buffer */
925 fenc_alloced = FALSE;
927 else
929 fenc_next = p_fencs; /* try items in 'fileencodings' */
930 fenc = next_fenc(&fenc_next);
931 fenc_alloced = TRUE;
933 #endif
936 * Jump back here to retry reading the file in different ways.
937 * Reasons to retry:
938 * - encoding conversion failed: try another one from "fenc_next"
939 * - BOM detected and fenc was set, need to setup conversion
940 * - "fileformat" check failed: try another
942 * Variables set for special retry actions:
943 * "file_rewind" Rewind the file to start reading it again.
944 * "advance_fenc" Advance "fenc" using "fenc_next".
945 * "skip_read" Re-use already read bytes (BOM detected).
946 * "did_iconv" iconv() conversion failed, try 'charconvert'.
947 * "keep_fileformat" Don't reset "fileformat".
949 * Other status indicators:
950 * "tmpname" When != NULL did conversion with 'charconvert'.
951 * Output file has to be deleted afterwards.
952 * "iconv_fd" When != -1 did conversion with iconv().
954 retry:
956 if (file_rewind)
958 if (read_buffer)
960 read_buf_lnum = 1;
961 read_buf_col = 0;
963 else if (read_stdin || lseek(fd, (off_t)0L, SEEK_SET) != 0)
965 /* Can't rewind the file, give up. */
966 error = TRUE;
967 goto failed;
969 /* Delete the previously read lines. */
970 while (lnum > from)
971 ml_delete(lnum--, FALSE);
972 file_rewind = FALSE;
973 #ifdef FEAT_MBYTE
974 if (set_options)
976 curbuf->b_p_bomb = FALSE;
977 curbuf->b_start_bomb = FALSE;
979 conv_error = 0;
980 #endif
984 * When retrying with another "fenc" and the first time "fileformat"
985 * will be reset.
987 if (keep_fileformat)
988 keep_fileformat = FALSE;
989 else
991 if (eap != NULL && eap->force_ff != 0)
993 fileformat = get_fileformat_force(curbuf, eap);
994 try_unix = try_dos = try_mac = FALSE;
996 else if (curbuf->b_p_bin)
997 fileformat = EOL_UNIX; /* binary: use Unix format */
998 else if (*p_ffs == NUL)
999 fileformat = get_fileformat(curbuf);/* use format from buffer */
1000 else
1001 fileformat = EOL_UNKNOWN; /* detect from file */
1004 #ifdef FEAT_MBYTE
1005 # ifdef USE_ICONV
1006 if (iconv_fd != (iconv_t)-1)
1008 /* aborted conversion with iconv(), close the descriptor */
1009 iconv_close(iconv_fd);
1010 iconv_fd = (iconv_t)-1;
1012 # endif
1014 if (advance_fenc)
1017 * Try the next entry in 'fileencodings'.
1019 advance_fenc = FALSE;
1021 if (eap != NULL && eap->force_enc != 0)
1023 /* Conversion given with "++cc=" wasn't possible, read
1024 * without conversion. */
1025 notconverted = TRUE;
1026 conv_error = 0;
1027 if (fenc_alloced)
1028 vim_free(fenc);
1029 fenc = (char_u *)"";
1030 fenc_alloced = FALSE;
1032 else
1034 if (fenc_alloced)
1035 vim_free(fenc);
1036 if (fenc_next != NULL)
1038 fenc = next_fenc(&fenc_next);
1039 fenc_alloced = (fenc_next != NULL);
1041 else
1043 fenc = (char_u *)"";
1044 fenc_alloced = FALSE;
1047 if (tmpname != NULL)
1049 mch_remove(tmpname); /* delete converted file */
1050 vim_free(tmpname);
1051 tmpname = NULL;
1056 * Conversion may be required when the encoding of the file is different
1057 * from 'encoding' or 'encoding' is UTF-16, UCS-2 or UCS-4.
1059 fio_flags = 0;
1060 converted = need_conversion(fenc);
1061 if (converted)
1064 /* "ucs-bom" means we need to check the first bytes of the file
1065 * for a BOM. */
1066 if (STRCMP(fenc, ENC_UCSBOM) == 0)
1067 fio_flags = FIO_UCSBOM;
1070 * Check if UCS-2/4 or Latin1 to UTF-8 conversion needs to be
1071 * done. This is handled below after read(). Prepare the
1072 * fio_flags to avoid having to parse the string each time.
1073 * Also check for Unicode to Latin1 conversion, because iconv()
1074 * appears not to handle this correctly. This works just like
1075 * conversion to UTF-8 except how the resulting character is put in
1076 * the buffer.
1078 else if (enc_utf8 || STRCMP(p_enc, "latin1") == 0)
1079 fio_flags = get_fio_flags(fenc);
1081 # ifdef WIN3264
1083 * Conversion from an MS-Windows codepage to UTF-8 or another codepage
1084 * is handled with MultiByteToWideChar().
1086 if (fio_flags == 0)
1087 fio_flags = get_win_fio_flags(fenc);
1088 # endif
1090 # ifdef MACOS_X
1091 /* Conversion from Apple MacRoman to latin1 or UTF-8 */
1092 if (fio_flags == 0)
1093 fio_flags = get_mac_fio_flags(fenc);
1094 # endif
1096 # ifdef USE_ICONV
1098 * Try using iconv() if we can't convert internally.
1100 if (fio_flags == 0
1101 # ifdef FEAT_EVAL
1102 && !did_iconv
1103 # endif
1105 iconv_fd = (iconv_t)my_iconv_open(
1106 enc_utf8 ? (char_u *)"utf-8" : p_enc, fenc);
1107 # endif
1109 # ifdef FEAT_EVAL
1111 * Use the 'charconvert' expression when conversion is required
1112 * and we can't do it internally or with iconv().
1114 if (fio_flags == 0 && !read_stdin && !read_buffer && *p_ccv != NUL
1115 # ifdef USE_ICONV
1116 && iconv_fd == (iconv_t)-1
1117 # endif
1120 # ifdef USE_ICONV
1121 did_iconv = FALSE;
1122 # endif
1123 /* Skip conversion when it's already done (retry for wrong
1124 * "fileformat"). */
1125 if (tmpname == NULL)
1127 tmpname = readfile_charconvert(fname, fenc, &fd);
1128 if (tmpname == NULL)
1130 /* Conversion failed. Try another one. */
1131 advance_fenc = TRUE;
1132 if (fd < 0)
1134 /* Re-opening the original file failed! */
1135 EMSG(_("E202: Conversion made file unreadable!"));
1136 error = TRUE;
1137 goto failed;
1139 goto retry;
1143 else
1144 # endif
1146 if (fio_flags == 0
1147 # ifdef USE_ICONV
1148 && iconv_fd == (iconv_t)-1
1149 # endif
1152 /* Conversion wanted but we can't.
1153 * Try the next conversion in 'fileencodings' */
1154 advance_fenc = TRUE;
1155 goto retry;
1160 /* Set "can_retry" when it's possible to rewind the file and try with
1161 * another "fenc" value. It's FALSE when no other "fenc" to try, reading
1162 * stdin or fixed at a specific encoding. */
1163 can_retry = (*fenc != NUL && !read_stdin && !keep_dest_enc);
1164 #endif
1166 if (!skip_read)
1168 linerest = 0;
1169 filesize = 0;
1170 skip_count = lines_to_skip;
1171 read_count = lines_to_read;
1172 #ifdef FEAT_MBYTE
1173 conv_restlen = 0;
1174 #endif
1177 while (!error && !got_int)
1180 * We allocate as much space for the file as we can get, plus
1181 * space for the old line plus room for one terminating NUL.
1182 * The amount is limited by the fact that read() only can read
1183 * upto max_unsigned characters (and other things).
1185 #if SIZEOF_INT <= 2
1186 if (linerest >= 0x7ff0)
1188 ++split;
1189 *ptr = NL; /* split line by inserting a NL */
1190 size = 1;
1192 else
1193 #endif
1195 if (!skip_read)
1197 #if SIZEOF_INT > 2
1198 # if defined(SSIZE_MAX) && (SSIZE_MAX < 0x10000L)
1199 size = SSIZE_MAX; /* use max I/O size, 52K */
1200 # else
1201 size = 0x10000L; /* use buffer >= 64K */
1202 # endif
1203 #else
1204 size = 0x7ff0L - linerest; /* limit buffer to 32K */
1205 #endif
1207 for ( ; size >= 10; size = (long)((long_u)size >> 1))
1209 if ((new_buffer = lalloc((long_u)(size + linerest + 1),
1210 FALSE)) != NULL)
1211 break;
1213 if (new_buffer == NULL)
1215 do_outofmem_msg((long_u)(size * 2 + linerest + 1));
1216 error = TRUE;
1217 break;
1219 if (linerest) /* copy characters from the previous buffer */
1220 mch_memmove(new_buffer, ptr - linerest, (size_t)linerest);
1221 vim_free(buffer);
1222 buffer = new_buffer;
1223 ptr = buffer + linerest;
1224 line_start = buffer;
1226 #ifdef FEAT_MBYTE
1227 /* May need room to translate into.
1228 * For iconv() we don't really know the required space, use a
1229 * factor ICONV_MULT.
1230 * latin1 to utf-8: 1 byte becomes up to 2 bytes
1231 * utf-16 to utf-8: 2 bytes become up to 3 bytes, 4 bytes
1232 * become up to 4 bytes, size must be multiple of 2
1233 * ucs-2 to utf-8: 2 bytes become up to 3 bytes, size must be
1234 * multiple of 2
1235 * ucs-4 to utf-8: 4 bytes become up to 6 bytes, size must be
1236 * multiple of 4 */
1237 real_size = (int)size;
1238 # ifdef USE_ICONV
1239 if (iconv_fd != (iconv_t)-1)
1240 size = size / ICONV_MULT;
1241 else
1242 # endif
1243 if (fio_flags & FIO_LATIN1)
1244 size = size / 2;
1245 else if (fio_flags & (FIO_UCS2 | FIO_UTF16))
1246 size = (size * 2 / 3) & ~1;
1247 else if (fio_flags & FIO_UCS4)
1248 size = (size * 2 / 3) & ~3;
1249 else if (fio_flags == FIO_UCSBOM)
1250 size = size / ICONV_MULT; /* worst case */
1251 # ifdef WIN3264
1252 else if (fio_flags & FIO_CODEPAGE)
1253 size = size / ICONV_MULT; /* also worst case */
1254 # endif
1255 # ifdef MACOS_X
1256 else if (fio_flags & FIO_MACROMAN)
1257 size = size / ICONV_MULT; /* also worst case */
1258 # endif
1259 #endif
1261 #ifdef FEAT_MBYTE
1262 if (conv_restlen > 0)
1264 /* Insert unconverted bytes from previous line. */
1265 mch_memmove(ptr, conv_rest, conv_restlen);
1266 ptr += conv_restlen;
1267 size -= conv_restlen;
1269 #endif
1271 if (read_buffer)
1274 * Read bytes from curbuf. Used for converting text read
1275 * from stdin.
1277 if (read_buf_lnum > from)
1278 size = 0;
1279 else
1281 int n, ni;
1282 long tlen;
1284 tlen = 0;
1285 for (;;)
1287 p = ml_get(read_buf_lnum) + read_buf_col;
1288 n = (int)STRLEN(p);
1289 if ((int)tlen + n + 1 > size)
1291 /* Filled up to "size", append partial line.
1292 * Change NL to NUL to reverse the effect done
1293 * below. */
1294 n = (int)(size - tlen);
1295 for (ni = 0; ni < n; ++ni)
1297 if (p[ni] == NL)
1298 ptr[tlen++] = NUL;
1299 else
1300 ptr[tlen++] = p[ni];
1302 read_buf_col += n;
1303 break;
1305 else
1307 /* Append whole line and new-line. Change NL
1308 * to NUL to reverse the effect done below. */
1309 for (ni = 0; ni < n; ++ni)
1311 if (p[ni] == NL)
1312 ptr[tlen++] = NUL;
1313 else
1314 ptr[tlen++] = p[ni];
1316 ptr[tlen++] = NL;
1317 read_buf_col = 0;
1318 if (++read_buf_lnum > from)
1320 /* When the last line didn't have an
1321 * end-of-line don't add it now either. */
1322 if (!curbuf->b_p_eol)
1323 --tlen;
1324 size = tlen;
1325 break;
1331 else
1334 * Read bytes from the file.
1336 size = vim_read(fd, ptr, size);
1339 if (size <= 0)
1341 if (size < 0) /* read error */
1342 error = TRUE;
1343 #ifdef FEAT_MBYTE
1344 else if (conv_restlen > 0)
1347 * Reached end-of-file but some trailing bytes could
1348 * not be converted. Truncated file?
1351 /* When we did a conversion report an error. */
1352 if (fio_flags != 0
1353 # ifdef USE_ICONV
1354 || iconv_fd != (iconv_t)-1
1355 # endif
1358 if (conv_error == 0)
1359 conv_error = curbuf->b_ml.ml_line_count
1360 - linecnt + 1;
1362 /* Remember the first linenr with an illegal byte */
1363 else if (illegal_byte == 0)
1364 illegal_byte = curbuf->b_ml.ml_line_count
1365 - linecnt + 1;
1366 if (bad_char_behavior == BAD_DROP)
1368 *(ptr - conv_restlen) = NUL;
1369 conv_restlen = 0;
1371 else
1373 /* Replace the trailing bytes with the replacement
1374 * character if we were converting; if we weren't,
1375 * leave the UTF8 checking code to do it, as it
1376 * works slightly differently. */
1377 if (bad_char_behavior != BAD_KEEP && (fio_flags != 0
1378 # ifdef USE_ICONV
1379 || iconv_fd != (iconv_t)-1
1380 # endif
1383 while (conv_restlen > 0)
1385 *(--ptr) = bad_char_behavior;
1386 --conv_restlen;
1389 fio_flags = 0; /* don't convert this */
1390 # ifdef USE_ICONV
1391 if (iconv_fd != (iconv_t)-1)
1393 iconv_close(iconv_fd);
1394 iconv_fd = (iconv_t)-1;
1396 # endif
1399 #endif
1402 #ifdef FEAT_CRYPT
1404 * At start of file: Check for magic number of encryption.
1406 if (filesize == 0)
1407 cryptkey = check_for_cryptkey(cryptkey, ptr, &size,
1408 &filesize, newfile);
1410 * Decrypt the read bytes.
1412 if (cryptkey != NULL && size > 0)
1413 for (p = ptr; p < ptr + size; ++p)
1414 ZDECODE(*p);
1415 #endif
1417 skip_read = FALSE;
1419 #ifdef FEAT_MBYTE
1421 * At start of file (or after crypt magic number): Check for BOM.
1422 * Also check for a BOM for other Unicode encodings, but not after
1423 * converting with 'charconvert' or when a BOM has already been
1424 * found.
1426 if ((filesize == 0
1427 # ifdef FEAT_CRYPT
1428 || (filesize == CRYPT_MAGIC_LEN && cryptkey != NULL)
1429 # endif
1431 && (fio_flags == FIO_UCSBOM
1432 || (!curbuf->b_p_bomb
1433 && tmpname == NULL
1434 && (*fenc == 'u' || (*fenc == NUL && enc_utf8)))))
1436 char_u *ccname;
1437 int blen;
1439 /* no BOM detection in a short file or in binary mode */
1440 if (size < 2 || curbuf->b_p_bin)
1441 ccname = NULL;
1442 else
1443 ccname = check_for_bom(ptr, size, &blen,
1444 fio_flags == FIO_UCSBOM ? FIO_ALL : get_fio_flags(fenc));
1445 if (ccname != NULL)
1447 /* Remove BOM from the text */
1448 filesize += blen;
1449 size -= blen;
1450 mch_memmove(ptr, ptr + blen, (size_t)size);
1451 if (set_options)
1453 curbuf->b_p_bomb = TRUE;
1454 curbuf->b_start_bomb = TRUE;
1458 if (fio_flags == FIO_UCSBOM)
1460 if (ccname == NULL)
1462 /* No BOM detected: retry with next encoding. */
1463 advance_fenc = TRUE;
1465 else
1467 /* BOM detected: set "fenc" and jump back */
1468 if (fenc_alloced)
1469 vim_free(fenc);
1470 fenc = ccname;
1471 fenc_alloced = FALSE;
1473 /* retry reading without getting new bytes or rewinding */
1474 skip_read = TRUE;
1475 goto retry;
1479 /* Include not converted bytes. */
1480 ptr -= conv_restlen;
1481 size += conv_restlen;
1482 conv_restlen = 0;
1483 #endif
1485 * Break here for a read error or end-of-file.
1487 if (size <= 0)
1488 break;
1490 #ifdef FEAT_MBYTE
1492 # ifdef USE_ICONV
1493 if (iconv_fd != (iconv_t)-1)
1496 * Attempt conversion of the read bytes to 'encoding' using
1497 * iconv().
1499 const char *fromp;
1500 char *top;
1501 size_t from_size;
1502 size_t to_size;
1504 fromp = (char *)ptr;
1505 from_size = size;
1506 ptr += size;
1507 top = (char *)ptr;
1508 to_size = real_size - size;
1511 * If there is conversion error or not enough room try using
1512 * another conversion. Except for when there is no
1513 * alternative (help files).
1515 while ((iconv(iconv_fd, (void *)&fromp, &from_size,
1516 &top, &to_size)
1517 == (size_t)-1 && ICONV_ERRNO != ICONV_EINVAL)
1518 || from_size > CONV_RESTLEN)
1520 if (can_retry)
1521 goto rewind_retry;
1522 if (conv_error == 0)
1523 conv_error = readfile_linenr(linecnt,
1524 ptr, (char_u *)top);
1526 /* Deal with a bad byte and continue with the next. */
1527 ++fromp;
1528 --from_size;
1529 if (bad_char_behavior == BAD_KEEP)
1531 *top++ = *(fromp - 1);
1532 --to_size;
1534 else if (bad_char_behavior != BAD_DROP)
1536 *top++ = bad_char_behavior;
1537 --to_size;
1541 if (from_size > 0)
1543 /* Some remaining characters, keep them for the next
1544 * round. */
1545 mch_memmove(conv_rest, (char_u *)fromp, from_size);
1546 conv_restlen = (int)from_size;
1549 /* move the linerest to before the converted characters */
1550 line_start = ptr - linerest;
1551 mch_memmove(line_start, buffer, (size_t)linerest);
1552 size = (long)((char_u *)top - ptr);
1554 # endif
1556 # ifdef WIN3264
1557 if (fio_flags & FIO_CODEPAGE)
1559 char_u *src, *dst;
1560 WCHAR ucs2buf[3];
1561 int ucs2len;
1562 int codepage = FIO_GET_CP(fio_flags);
1563 int bytelen;
1564 int found_bad;
1565 char replstr[2];
1568 * Conversion from an MS-Windows codepage or UTF-8 to UTF-8 or
1569 * a codepage, using standard MS-Windows functions. This
1570 * requires two steps:
1571 * 1. convert from 'fileencoding' to ucs-2
1572 * 2. convert from ucs-2 to 'encoding'
1574 * Because there may be illegal bytes AND an incomplete byte
1575 * sequence at the end, we may have to do the conversion one
1576 * character at a time to get it right.
1579 /* Replacement string for WideCharToMultiByte(). */
1580 if (bad_char_behavior > 0)
1581 replstr[0] = bad_char_behavior;
1582 else
1583 replstr[0] = '?';
1584 replstr[1] = NUL;
1587 * Move the bytes to the end of the buffer, so that we have
1588 * room to put the result at the start.
1590 src = ptr + real_size - size;
1591 mch_memmove(src, ptr, size);
1594 * Do the conversion.
1596 dst = ptr;
1597 size = size;
1598 while (size > 0)
1600 found_bad = FALSE;
1602 # ifdef CP_UTF8 /* VC 4.1 doesn't define CP_UTF8 */
1603 if (codepage == CP_UTF8)
1605 /* Handle CP_UTF8 input ourselves to be able to handle
1606 * trailing bytes properly.
1607 * Get one UTF-8 character from src. */
1608 bytelen = (int)utf_ptr2len_len(src, size);
1609 if (bytelen > size)
1611 /* Only got some bytes of a character. Normally
1612 * it's put in "conv_rest", but if it's too long
1613 * deal with it as if they were illegal bytes. */
1614 if (bytelen <= CONV_RESTLEN)
1615 break;
1617 /* weird overlong byte sequence */
1618 bytelen = size;
1619 found_bad = TRUE;
1621 else
1623 int u8c = utf_ptr2char(src);
1625 if (u8c > 0xffff || (*src >= 0x80 && bytelen == 1))
1626 found_bad = TRUE;
1627 ucs2buf[0] = u8c;
1628 ucs2len = 1;
1631 else
1632 # endif
1634 /* We don't know how long the byte sequence is, try
1635 * from one to three bytes. */
1636 for (bytelen = 1; bytelen <= size && bytelen <= 3;
1637 ++bytelen)
1639 ucs2len = MultiByteToWideChar(codepage,
1640 MB_ERR_INVALID_CHARS,
1641 (LPCSTR)src, bytelen,
1642 ucs2buf, 3);
1643 if (ucs2len > 0)
1644 break;
1646 if (ucs2len == 0)
1648 /* If we have only one byte then it's probably an
1649 * incomplete byte sequence. Otherwise discard
1650 * one byte as a bad character. */
1651 if (size == 1)
1652 break;
1653 found_bad = TRUE;
1654 bytelen = 1;
1658 if (!found_bad)
1660 int i;
1662 /* Convert "ucs2buf[ucs2len]" to 'enc' in "dst". */
1663 if (enc_utf8)
1665 /* From UCS-2 to UTF-8. Cannot fail. */
1666 for (i = 0; i < ucs2len; ++i)
1667 dst += utf_char2bytes(ucs2buf[i], dst);
1669 else
1671 BOOL bad = FALSE;
1672 int dstlen;
1674 /* From UCS-2 to "enc_codepage". If the
1675 * conversion uses the default character "?",
1676 * the data doesn't fit in this encoding. */
1677 dstlen = WideCharToMultiByte(enc_codepage, 0,
1678 (LPCWSTR)ucs2buf, ucs2len,
1679 (LPSTR)dst, (int)(src - dst),
1680 replstr, &bad);
1681 if (bad)
1682 found_bad = TRUE;
1683 else
1684 dst += dstlen;
1688 if (found_bad)
1690 /* Deal with bytes we can't convert. */
1691 if (can_retry)
1692 goto rewind_retry;
1693 if (conv_error == 0)
1694 conv_error = readfile_linenr(linecnt, ptr, dst);
1695 if (bad_char_behavior != BAD_DROP)
1697 if (bad_char_behavior == BAD_KEEP)
1699 mch_memmove(dst, src, bytelen);
1700 dst += bytelen;
1702 else
1703 *dst++ = bad_char_behavior;
1707 src += bytelen;
1708 size -= bytelen;
1711 if (size > 0)
1713 /* An incomplete byte sequence remaining. */
1714 mch_memmove(conv_rest, src, size);
1715 conv_restlen = size;
1718 /* The new size is equal to how much "dst" was advanced. */
1719 size = (long)(dst - ptr);
1721 else
1722 # endif
1723 # ifdef MACOS_CONVERT
1724 if (fio_flags & FIO_MACROMAN)
1727 * Conversion from Apple MacRoman char encoding to UTF-8 or
1728 * latin1. This is in os_mac_conv.c.
1730 if (macroman2enc(ptr, &size, real_size) == FAIL)
1731 goto rewind_retry;
1733 else
1734 # endif
1735 if (fio_flags != 0)
1737 int u8c;
1738 char_u *dest;
1739 char_u *tail = NULL;
1742 * "enc_utf8" set: Convert Unicode or Latin1 to UTF-8.
1743 * "enc_utf8" not set: Convert Unicode to Latin1.
1744 * Go from end to start through the buffer, because the number
1745 * of bytes may increase.
1746 * "dest" points to after where the UTF-8 bytes go, "p" points
1747 * to after the next character to convert.
1749 dest = ptr + real_size;
1750 if (fio_flags == FIO_LATIN1 || fio_flags == FIO_UTF8)
1752 p = ptr + size;
1753 if (fio_flags == FIO_UTF8)
1755 /* Check for a trailing incomplete UTF-8 sequence */
1756 tail = ptr + size - 1;
1757 while (tail > ptr && (*tail & 0xc0) == 0x80)
1758 --tail;
1759 if (tail + utf_byte2len(*tail) <= ptr + size)
1760 tail = NULL;
1761 else
1762 p = tail;
1765 else if (fio_flags & (FIO_UCS2 | FIO_UTF16))
1767 /* Check for a trailing byte */
1768 p = ptr + (size & ~1);
1769 if (size & 1)
1770 tail = p;
1771 if ((fio_flags & FIO_UTF16) && p > ptr)
1773 /* Check for a trailing leading word */
1774 if (fio_flags & FIO_ENDIAN_L)
1776 u8c = (*--p << 8);
1777 u8c += *--p;
1779 else
1781 u8c = *--p;
1782 u8c += (*--p << 8);
1784 if (u8c >= 0xd800 && u8c <= 0xdbff)
1785 tail = p;
1786 else
1787 p += 2;
1790 else /* FIO_UCS4 */
1792 /* Check for trailing 1, 2 or 3 bytes */
1793 p = ptr + (size & ~3);
1794 if (size & 3)
1795 tail = p;
1798 /* If there is a trailing incomplete sequence move it to
1799 * conv_rest[]. */
1800 if (tail != NULL)
1802 conv_restlen = (int)((ptr + size) - tail);
1803 mch_memmove(conv_rest, (char_u *)tail, conv_restlen);
1804 size -= conv_restlen;
1808 while (p > ptr)
1810 if (fio_flags & FIO_LATIN1)
1811 u8c = *--p;
1812 else if (fio_flags & (FIO_UCS2 | FIO_UTF16))
1814 if (fio_flags & FIO_ENDIAN_L)
1816 u8c = (*--p << 8);
1817 u8c += *--p;
1819 else
1821 u8c = *--p;
1822 u8c += (*--p << 8);
1824 if ((fio_flags & FIO_UTF16)
1825 && u8c >= 0xdc00 && u8c <= 0xdfff)
1827 int u16c;
1829 if (p == ptr)
1831 /* Missing leading word. */
1832 if (can_retry)
1833 goto rewind_retry;
1834 if (conv_error == 0)
1835 conv_error = readfile_linenr(linecnt,
1836 ptr, p);
1837 if (bad_char_behavior == BAD_DROP)
1838 continue;
1839 if (bad_char_behavior != BAD_KEEP)
1840 u8c = bad_char_behavior;
1843 /* found second word of double-word, get the first
1844 * word and compute the resulting character */
1845 if (fio_flags & FIO_ENDIAN_L)
1847 u16c = (*--p << 8);
1848 u16c += *--p;
1850 else
1852 u16c = *--p;
1853 u16c += (*--p << 8);
1855 u8c = 0x10000 + ((u16c & 0x3ff) << 10)
1856 + (u8c & 0x3ff);
1858 /* Check if the word is indeed a leading word. */
1859 if (u16c < 0xd800 || u16c > 0xdbff)
1861 if (can_retry)
1862 goto rewind_retry;
1863 if (conv_error == 0)
1864 conv_error = readfile_linenr(linecnt,
1865 ptr, p);
1866 if (bad_char_behavior == BAD_DROP)
1867 continue;
1868 if (bad_char_behavior != BAD_KEEP)
1869 u8c = bad_char_behavior;
1873 else if (fio_flags & FIO_UCS4)
1875 if (fio_flags & FIO_ENDIAN_L)
1877 u8c = (*--p << 24);
1878 u8c += (*--p << 16);
1879 u8c += (*--p << 8);
1880 u8c += *--p;
1882 else /* big endian */
1884 u8c = *--p;
1885 u8c += (*--p << 8);
1886 u8c += (*--p << 16);
1887 u8c += (*--p << 24);
1890 else /* UTF-8 */
1892 if (*--p < 0x80)
1893 u8c = *p;
1894 else
1896 len = utf_head_off(ptr, p);
1897 p -= len;
1898 u8c = utf_ptr2char(p);
1899 if (len == 0)
1901 /* Not a valid UTF-8 character, retry with
1902 * another fenc when possible, otherwise just
1903 * report the error. */
1904 if (can_retry)
1905 goto rewind_retry;
1906 if (conv_error == 0)
1907 conv_error = readfile_linenr(linecnt,
1908 ptr, p);
1909 if (bad_char_behavior == BAD_DROP)
1910 continue;
1911 if (bad_char_behavior != BAD_KEEP)
1912 u8c = bad_char_behavior;
1916 if (enc_utf8) /* produce UTF-8 */
1918 dest -= utf_char2len(u8c);
1919 (void)utf_char2bytes(u8c, dest);
1921 else /* produce Latin1 */
1923 --dest;
1924 if (u8c >= 0x100)
1926 /* character doesn't fit in latin1, retry with
1927 * another fenc when possible, otherwise just
1928 * report the error. */
1929 if (can_retry)
1930 goto rewind_retry;
1931 if (conv_error == 0)
1932 conv_error = readfile_linenr(linecnt, ptr, p);
1933 if (bad_char_behavior == BAD_DROP)
1934 ++dest;
1935 else if (bad_char_behavior == BAD_KEEP)
1936 *dest = u8c;
1937 else if (eap != NULL && eap->bad_char != 0)
1938 *dest = bad_char_behavior;
1939 else
1940 *dest = 0xBF;
1942 else
1943 *dest = u8c;
1947 /* move the linerest to before the converted characters */
1948 line_start = dest - linerest;
1949 mch_memmove(line_start, buffer, (size_t)linerest);
1950 size = (long)((ptr + real_size) - dest);
1951 ptr = dest;
1953 else if (enc_utf8 && !curbuf->b_p_bin)
1955 int incomplete_tail = FALSE;
1957 /* Reading UTF-8: Check if the bytes are valid UTF-8. */
1958 for (p = ptr; ; ++p)
1960 int todo = (int)((ptr + size) - p);
1961 int l;
1963 if (todo <= 0)
1964 break;
1965 if (*p >= 0x80)
1967 /* A length of 1 means it's an illegal byte. Accept
1968 * an incomplete character at the end though, the next
1969 * read() will get the next bytes, we'll check it
1970 * then. */
1971 l = utf_ptr2len_len(p, todo);
1972 if (l > todo && !incomplete_tail)
1974 /* Avoid retrying with a different encoding when
1975 * a truncated file is more likely, or attempting
1976 * to read the rest of an incomplete sequence when
1977 * we have already done so. */
1978 if (p > ptr || filesize > 0)
1979 incomplete_tail = TRUE;
1980 /* Incomplete byte sequence, move it to conv_rest[]
1981 * and try to read the rest of it, unless we've
1982 * already done so. */
1983 if (p > ptr)
1985 conv_restlen = todo;
1986 mch_memmove(conv_rest, p, conv_restlen);
1987 size -= conv_restlen;
1988 break;
1991 if (l == 1 || l > todo)
1993 /* Illegal byte. If we can try another encoding
1994 * do that, unless at EOF where a truncated
1995 * file is more likely than a conversion error. */
1996 if (can_retry && !incomplete_tail)
1997 break;
1998 # ifdef USE_ICONV
1999 /* When we did a conversion report an error. */
2000 if (iconv_fd != (iconv_t)-1 && conv_error == 0)
2001 conv_error = readfile_linenr(linecnt, ptr, p);
2002 # endif
2003 /* Remember the first linenr with an illegal byte */
2004 if (conv_error == 0 && illegal_byte == 0)
2005 illegal_byte = readfile_linenr(linecnt, ptr, p);
2007 /* Drop, keep or replace the bad byte. */
2008 if (bad_char_behavior == BAD_DROP)
2010 mch_memmove(p, p + 1, todo - 1);
2011 --p;
2012 --size;
2014 else if (bad_char_behavior != BAD_KEEP)
2015 *p = bad_char_behavior;
2017 else
2018 p += l - 1;
2021 if (p < ptr + size && !incomplete_tail)
2023 /* Detected a UTF-8 error. */
2024 rewind_retry:
2025 /* Retry reading with another conversion. */
2026 # if defined(FEAT_EVAL) && defined(USE_ICONV)
2027 if (*p_ccv != NUL && iconv_fd != (iconv_t)-1)
2028 /* iconv() failed, try 'charconvert' */
2029 did_iconv = TRUE;
2030 else
2031 # endif
2032 /* use next item from 'fileencodings' */
2033 advance_fenc = TRUE;
2034 file_rewind = TRUE;
2035 goto retry;
2038 #endif
2040 /* count the number of characters (after conversion!) */
2041 filesize += size;
2044 * when reading the first part of a file: guess EOL type
2046 if (fileformat == EOL_UNKNOWN)
2048 /* First try finding a NL, for Dos and Unix */
2049 if (try_dos || try_unix)
2051 for (p = ptr; p < ptr + size; ++p)
2053 if (*p == NL)
2055 if (!try_unix
2056 || (try_dos && p > ptr && p[-1] == CAR))
2057 fileformat = EOL_DOS;
2058 else
2059 fileformat = EOL_UNIX;
2060 break;
2064 /* Don't give in to EOL_UNIX if EOL_MAC is more likely */
2065 if (fileformat == EOL_UNIX && try_mac)
2067 /* Need to reset the counters when retrying fenc. */
2068 try_mac = 1;
2069 try_unix = 1;
2070 for (; p >= ptr && *p != CAR; p--)
2072 if (p >= ptr)
2074 for (p = ptr; p < ptr + size; ++p)
2076 if (*p == NL)
2077 try_unix++;
2078 else if (*p == CAR)
2079 try_mac++;
2081 if (try_mac > try_unix)
2082 fileformat = EOL_MAC;
2087 /* No NL found: may use Mac format */
2088 if (fileformat == EOL_UNKNOWN && try_mac)
2089 fileformat = EOL_MAC;
2091 /* Still nothing found? Use first format in 'ffs' */
2092 if (fileformat == EOL_UNKNOWN)
2093 fileformat = default_fileformat();
2095 /* if editing a new file: may set p_tx and p_ff */
2096 if (set_options)
2097 set_fileformat(fileformat, OPT_LOCAL);
2102 * This loop is executed once for every character read.
2103 * Keep it fast!
2105 if (fileformat == EOL_MAC)
2107 --ptr;
2108 while (++ptr, --size >= 0)
2110 /* catch most common case first */
2111 if ((c = *ptr) != NUL && c != CAR && c != NL)
2112 continue;
2113 if (c == NUL)
2114 *ptr = NL; /* NULs are replaced by newlines! */
2115 else if (c == NL)
2116 *ptr = CAR; /* NLs are replaced by CRs! */
2117 else
2119 if (skip_count == 0)
2121 *ptr = NUL; /* end of line */
2122 len = (colnr_T) (ptr - line_start + 1);
2123 if (ml_append(lnum, line_start, len, newfile) == FAIL)
2125 error = TRUE;
2126 break;
2128 ++lnum;
2129 if (--read_count == 0)
2131 error = TRUE; /* break loop */
2132 line_start = ptr; /* nothing left to write */
2133 break;
2136 else
2137 --skip_count;
2138 line_start = ptr + 1;
2142 else
2144 --ptr;
2145 while (++ptr, --size >= 0)
2147 if ((c = *ptr) != NUL && c != NL) /* catch most common case */
2148 continue;
2149 if (c == NUL)
2150 *ptr = NL; /* NULs are replaced by newlines! */
2151 else
2153 if (skip_count == 0)
2155 *ptr = NUL; /* end of line */
2156 len = (colnr_T)(ptr - line_start + 1);
2157 if (fileformat == EOL_DOS)
2159 if (ptr[-1] == CAR) /* remove CR */
2161 ptr[-1] = NUL;
2162 --len;
2165 * Reading in Dos format, but no CR-LF found!
2166 * When 'fileformats' includes "unix", delete all
2167 * the lines read so far and start all over again.
2168 * Otherwise give an error message later.
2170 else if (ff_error != EOL_DOS)
2172 if ( try_unix
2173 && !read_stdin
2174 && (read_buffer
2175 || lseek(fd, (off_t)0L, SEEK_SET) == 0))
2177 fileformat = EOL_UNIX;
2178 if (set_options)
2179 set_fileformat(EOL_UNIX, OPT_LOCAL);
2180 file_rewind = TRUE;
2181 keep_fileformat = TRUE;
2182 goto retry;
2184 ff_error = EOL_DOS;
2187 if (ml_append(lnum, line_start, len, newfile) == FAIL)
2189 error = TRUE;
2190 break;
2192 ++lnum;
2193 if (--read_count == 0)
2195 error = TRUE; /* break loop */
2196 line_start = ptr; /* nothing left to write */
2197 break;
2200 else
2201 --skip_count;
2202 line_start = ptr + 1;
2206 linerest = (long)(ptr - line_start);
2207 ui_breakcheck();
2210 failed:
2211 /* not an error, max. number of lines reached */
2212 if (error && read_count == 0)
2213 error = FALSE;
2216 * If we get EOF in the middle of a line, note the fact and
2217 * complete the line ourselves.
2218 * In Dos format ignore a trailing CTRL-Z, unless 'binary' set.
2220 if (!error
2221 && !got_int
2222 && linerest != 0
2223 && !(!curbuf->b_p_bin
2224 && fileformat == EOL_DOS
2225 && *line_start == Ctrl_Z
2226 && ptr == line_start + 1))
2228 /* remember for when writing */
2229 if (set_options)
2230 curbuf->b_p_eol = FALSE;
2231 *ptr = NUL;
2232 if (ml_append(lnum, line_start,
2233 (colnr_T)(ptr - line_start + 1), newfile) == FAIL)
2234 error = TRUE;
2235 else
2236 read_no_eol_lnum = ++lnum;
2239 if (set_options)
2240 save_file_ff(curbuf); /* remember the current file format */
2242 #ifdef FEAT_CRYPT
2243 if (cryptkey != curbuf->b_p_key)
2244 vim_free(cryptkey);
2245 #endif
2247 #ifdef FEAT_MBYTE
2248 /* If editing a new file: set 'fenc' for the current buffer.
2249 * Also for ":read ++edit file". */
2250 if (set_options)
2251 set_string_option_direct((char_u *)"fenc", -1, fenc,
2252 OPT_FREE|OPT_LOCAL, 0);
2253 if (fenc_alloced)
2254 vim_free(fenc);
2255 # ifdef USE_ICONV
2256 if (iconv_fd != (iconv_t)-1)
2258 iconv_close(iconv_fd);
2259 iconv_fd = (iconv_t)-1;
2261 # endif
2262 #endif
2264 if (!read_buffer && !read_stdin)
2265 close(fd); /* errors are ignored */
2266 #ifdef HAVE_FD_CLOEXEC
2267 else
2269 int fdflags = fcntl(fd, F_GETFD);
2270 if (fdflags >= 0 && (fdflags & FD_CLOEXEC) == 0)
2271 fcntl(fd, F_SETFD, fdflags | FD_CLOEXEC);
2273 #endif
2274 vim_free(buffer);
2276 #ifdef HAVE_DUP
2277 if (read_stdin)
2279 /* Use stderr for stdin, makes shell commands work. */
2280 close(0);
2281 ignored = dup(2);
2283 #endif
2285 #ifdef FEAT_MBYTE
2286 if (tmpname != NULL)
2288 mch_remove(tmpname); /* delete converted file */
2289 vim_free(tmpname);
2291 #endif
2292 --no_wait_return; /* may wait for return now */
2295 * In recovery mode everything but autocommands is skipped.
2297 if (!recoverymode)
2299 /* need to delete the last line, which comes from the empty buffer */
2300 if (newfile && wasempty && !(curbuf->b_ml.ml_flags & ML_EMPTY))
2302 #ifdef FEAT_NETBEANS_INTG
2303 netbeansFireChanges = 0;
2304 #endif
2305 ml_delete(curbuf->b_ml.ml_line_count, FALSE);
2306 #ifdef FEAT_NETBEANS_INTG
2307 netbeansFireChanges = 1;
2308 #endif
2309 --linecnt;
2311 linecnt = curbuf->b_ml.ml_line_count - linecnt;
2312 if (filesize == 0)
2313 linecnt = 0;
2314 if (newfile || read_buffer)
2316 redraw_curbuf_later(NOT_VALID);
2317 #ifdef FEAT_DIFF
2318 /* After reading the text into the buffer the diff info needs to
2319 * be updated. */
2320 diff_invalidate(curbuf);
2321 #endif
2322 #ifdef FEAT_FOLDING
2323 /* All folds in the window are invalid now. Mark them for update
2324 * before triggering autocommands. */
2325 foldUpdateAll(curwin);
2326 #endif
2328 else if (linecnt) /* appended at least one line */
2329 appended_lines_mark(from, linecnt);
2331 #ifndef ALWAYS_USE_GUI
2333 * If we were reading from the same terminal as where messages go,
2334 * the screen will have been messed up.
2335 * Switch on raw mode now and clear the screen.
2337 if (read_stdin)
2339 settmode(TMODE_RAW); /* set to raw mode */
2340 starttermcap();
2341 screenclear();
2343 #endif
2345 if (got_int)
2347 if (!(flags & READ_DUMMY))
2349 filemess(curbuf, sfname, (char_u *)_(e_interr), 0);
2350 if (newfile)
2351 curbuf->b_p_ro = TRUE; /* must use "w!" now */
2353 msg_scroll = msg_save;
2354 #ifdef FEAT_VIMINFO
2355 check_marks_read();
2356 #endif
2357 return OK; /* an interrupt isn't really an error */
2360 if (!filtering && !(flags & READ_DUMMY))
2362 msg_add_fname(curbuf, sfname); /* fname in IObuff with quotes */
2363 c = FALSE;
2365 #ifdef UNIX
2366 # ifdef S_ISFIFO
2367 if (S_ISFIFO(perm)) /* fifo or socket */
2369 STRCAT(IObuff, _("[fifo/socket]"));
2370 c = TRUE;
2372 # else
2373 # ifdef S_IFIFO
2374 if ((perm & S_IFMT) == S_IFIFO) /* fifo */
2376 STRCAT(IObuff, _("[fifo]"));
2377 c = TRUE;
2379 # endif
2380 # ifdef S_IFSOCK
2381 if ((perm & S_IFMT) == S_IFSOCK) /* or socket */
2383 STRCAT(IObuff, _("[socket]"));
2384 c = TRUE;
2386 # endif
2387 # endif
2388 # ifdef OPEN_CHR_FILES
2389 if (S_ISCHR(perm)) /* or character special */
2391 STRCAT(IObuff, _("[character special]"));
2392 c = TRUE;
2394 # endif
2395 #endif
2396 if (curbuf->b_p_ro)
2398 STRCAT(IObuff, shortmess(SHM_RO) ? _("[RO]") : _("[readonly]"));
2399 c = TRUE;
2401 if (read_no_eol_lnum)
2403 msg_add_eol();
2404 c = TRUE;
2406 if (ff_error == EOL_DOS)
2408 STRCAT(IObuff, _("[CR missing]"));
2409 c = TRUE;
2411 if (split)
2413 STRCAT(IObuff, _("[long lines split]"));
2414 c = TRUE;
2416 #ifdef FEAT_MBYTE
2417 if (notconverted)
2419 STRCAT(IObuff, _("[NOT converted]"));
2420 c = TRUE;
2422 else if (converted)
2424 STRCAT(IObuff, _("[converted]"));
2425 c = TRUE;
2427 #endif
2428 #ifdef FEAT_CRYPT
2429 if (cryptkey != NULL)
2431 STRCAT(IObuff, _("[crypted]"));
2432 c = TRUE;
2434 #endif
2435 #ifdef FEAT_MBYTE
2436 if (conv_error != 0)
2438 sprintf((char *)IObuff + STRLEN(IObuff),
2439 _("[CONVERSION ERROR in line %ld]"), (long)conv_error);
2440 c = TRUE;
2442 else if (illegal_byte > 0)
2444 sprintf((char *)IObuff + STRLEN(IObuff),
2445 _("[ILLEGAL BYTE in line %ld]"), (long)illegal_byte);
2446 c = TRUE;
2448 else
2449 #endif
2450 if (error)
2452 STRCAT(IObuff, _("[READ ERRORS]"));
2453 c = TRUE;
2455 if (msg_add_fileformat(fileformat))
2456 c = TRUE;
2457 #ifdef FEAT_CRYPT
2458 if (cryptkey != NULL)
2459 msg_add_lines(c, (long)linecnt, filesize - CRYPT_MAGIC_LEN);
2460 else
2461 #endif
2462 msg_add_lines(c, (long)linecnt, filesize);
2464 vim_free(keep_msg);
2465 keep_msg = NULL;
2466 msg_scrolled_ign = TRUE;
2467 #ifdef ALWAYS_USE_GUI
2468 /* Don't show the message when reading stdin, it would end up in a
2469 * message box (which might be shown when exiting!) */
2470 if (read_stdin || read_buffer)
2471 p = msg_may_trunc(FALSE, IObuff);
2472 else
2473 #endif
2474 p = msg_trunc_attr(IObuff, FALSE, 0);
2475 if (read_stdin || read_buffer || restart_edit != 0
2476 || (msg_scrolled != 0 && !need_wait_return))
2477 /* Need to repeat the message after redrawing when:
2478 * - When reading from stdin (the screen will be cleared next).
2479 * - When restart_edit is set (otherwise there will be a delay
2480 * before redrawing).
2481 * - When the screen was scrolled but there is no wait-return
2482 * prompt. */
2483 set_keep_msg(p, 0);
2484 msg_scrolled_ign = FALSE;
2487 /* with errors writing the file requires ":w!" */
2488 if (newfile && (error
2489 #ifdef FEAT_MBYTE
2490 || conv_error != 0
2491 || (illegal_byte > 0 && bad_char_behavior != BAD_KEEP)
2492 #endif
2494 curbuf->b_p_ro = TRUE;
2496 u_clearline(); /* cannot use "U" command after adding lines */
2499 * In Ex mode: cursor at last new line.
2500 * Otherwise: cursor at first new line.
2502 if (exmode_active)
2503 curwin->w_cursor.lnum = from + linecnt;
2504 else
2505 curwin->w_cursor.lnum = from + 1;
2506 check_cursor_lnum();
2507 beginline(BL_WHITE | BL_FIX); /* on first non-blank */
2510 * Set '[ and '] marks to the newly read lines.
2512 curbuf->b_op_start.lnum = from + 1;
2513 curbuf->b_op_start.col = 0;
2514 curbuf->b_op_end.lnum = from + linecnt;
2515 curbuf->b_op_end.col = 0;
2517 #ifdef WIN32
2519 * Work around a weird problem: When a file has two links (only
2520 * possible on NTFS) and we write through one link, then stat() it
2521 * through the other link, the timestamp information may be wrong.
2522 * It's correct again after reading the file, thus reset the timestamp
2523 * here.
2525 if (newfile && !read_stdin && !read_buffer
2526 && mch_stat((char *)fname, &st) >= 0)
2528 buf_store_time(curbuf, &st, fname);
2529 curbuf->b_mtime_read = curbuf->b_mtime;
2531 #endif
2533 msg_scroll = msg_save;
2535 #ifdef FEAT_VIMINFO
2537 * Get the marks before executing autocommands, so they can be used there.
2539 check_marks_read();
2540 #endif
2543 * Trick: We remember if the last line of the read didn't have
2544 * an eol for when writing it again. This is required for
2545 * ":autocmd FileReadPost *.gz set bin|'[,']!gunzip" to work.
2547 write_no_eol_lnum = read_no_eol_lnum;
2549 #ifdef FEAT_AUTOCMD
2550 if (!read_stdin && !read_buffer)
2552 int m = msg_scroll;
2553 int n = msg_scrolled;
2555 /* Save the fileformat now, otherwise the buffer will be considered
2556 * modified if the format/encoding was automatically detected. */
2557 if (set_options)
2558 save_file_ff(curbuf);
2561 * The output from the autocommands should not overwrite anything and
2562 * should not be overwritten: Set msg_scroll, restore its value if no
2563 * output was done.
2565 msg_scroll = TRUE;
2566 if (filtering)
2567 apply_autocmds_exarg(EVENT_FILTERREADPOST, NULL, sfname,
2568 FALSE, curbuf, eap);
2569 else if (newfile)
2570 apply_autocmds_exarg(EVENT_BUFREADPOST, NULL, sfname,
2571 FALSE, curbuf, eap);
2572 else
2573 apply_autocmds_exarg(EVENT_FILEREADPOST, sfname, sfname,
2574 FALSE, NULL, eap);
2575 if (msg_scrolled == n)
2576 msg_scroll = m;
2577 #ifdef FEAT_EVAL
2578 if (aborting()) /* autocmds may abort script processing */
2579 return FAIL;
2580 #endif
2582 #endif
2584 if (recoverymode && error)
2585 return FAIL;
2586 return OK;
2589 #ifdef OPEN_CHR_FILES
2591 * Returns TRUE if the file name argument is of the form "/dev/fd/\d\+",
2592 * which is the name of files used for process substitution output by
2593 * some shells on some operating systems, e.g., bash on SunOS.
2594 * Do not accept "/dev/fd/[012]", opening these may hang Vim.
2596 static int
2597 is_dev_fd_file(fname)
2598 char_u *fname;
2600 return (STRNCMP(fname, "/dev/fd/", 8) == 0
2601 && VIM_ISDIGIT(fname[8])
2602 && *skipdigits(fname + 9) == NUL
2603 && (fname[9] != NUL
2604 || (fname[8] != '0' && fname[8] != '1' && fname[8] != '2')));
2606 #endif
2608 #ifdef FEAT_MBYTE
2611 * From the current line count and characters read after that, estimate the
2612 * line number where we are now.
2613 * Used for error messages that include a line number.
2615 static linenr_T
2616 readfile_linenr(linecnt, p, endp)
2617 linenr_T linecnt; /* line count before reading more bytes */
2618 char_u *p; /* start of more bytes read */
2619 char_u *endp; /* end of more bytes read */
2621 char_u *s;
2622 linenr_T lnum;
2624 lnum = curbuf->b_ml.ml_line_count - linecnt + 1;
2625 for (s = p; s < endp; ++s)
2626 if (*s == '\n')
2627 ++lnum;
2628 return lnum;
2630 #endif
2633 * Fill "*eap" to force the 'fileencoding', 'fileformat' and 'binary to be
2634 * equal to the buffer "buf". Used for calling readfile().
2635 * Returns OK or FAIL.
2638 prep_exarg(eap, buf)
2639 exarg_T *eap;
2640 buf_T *buf;
2642 eap->cmd = alloc((unsigned)(STRLEN(buf->b_p_ff)
2643 #ifdef FEAT_MBYTE
2644 + STRLEN(buf->b_p_fenc)
2645 #endif
2646 + 15));
2647 if (eap->cmd == NULL)
2648 return FAIL;
2650 #ifdef FEAT_MBYTE
2651 sprintf((char *)eap->cmd, "e ++ff=%s ++enc=%s", buf->b_p_ff, buf->b_p_fenc);
2652 eap->force_enc = 14 + (int)STRLEN(buf->b_p_ff);
2653 eap->bad_char = buf->b_bad_char;
2654 #else
2655 sprintf((char *)eap->cmd, "e ++ff=%s", buf->b_p_ff);
2656 #endif
2657 eap->force_ff = 7;
2659 eap->force_bin = buf->b_p_bin ? FORCE_BIN : FORCE_NOBIN;
2660 eap->read_edit = FALSE;
2661 eap->forceit = FALSE;
2662 return OK;
2665 #ifdef FEAT_MBYTE
2667 * Find next fileencoding to use from 'fileencodings'.
2668 * "pp" points to fenc_next. It's advanced to the next item.
2669 * When there are no more items, an empty string is returned and *pp is set to
2670 * NULL.
2671 * When *pp is not set to NULL, the result is in allocated memory.
2673 static char_u *
2674 next_fenc(pp)
2675 char_u **pp;
2677 char_u *p;
2678 char_u *r;
2680 if (**pp == NUL)
2682 *pp = NULL;
2683 return (char_u *)"";
2685 p = vim_strchr(*pp, ',');
2686 if (p == NULL)
2688 r = enc_canonize(*pp);
2689 *pp += STRLEN(*pp);
2691 else
2693 r = vim_strnsave(*pp, (int)(p - *pp));
2694 *pp = p + 1;
2695 if (r != NULL)
2697 p = enc_canonize(r);
2698 vim_free(r);
2699 r = p;
2702 if (r == NULL) /* out of memory */
2704 r = (char_u *)"";
2705 *pp = NULL;
2707 return r;
2710 # ifdef FEAT_EVAL
2712 * Convert a file with the 'charconvert' expression.
2713 * This closes the file which is to be read, converts it and opens the
2714 * resulting file for reading.
2715 * Returns name of the resulting converted file (the caller should delete it
2716 * after reading it).
2717 * Returns NULL if the conversion failed ("*fdp" is not set) .
2719 static char_u *
2720 readfile_charconvert(fname, fenc, fdp)
2721 char_u *fname; /* name of input file */
2722 char_u *fenc; /* converted from */
2723 int *fdp; /* in/out: file descriptor of file */
2725 char_u *tmpname;
2726 char_u *errmsg = NULL;
2728 tmpname = vim_tempname('r');
2729 if (tmpname == NULL)
2730 errmsg = (char_u *)_("Can't find temp file for conversion");
2731 else
2733 close(*fdp); /* close the input file, ignore errors */
2734 *fdp = -1;
2735 if (eval_charconvert(fenc, enc_utf8 ? (char_u *)"utf-8" : p_enc,
2736 fname, tmpname) == FAIL)
2737 errmsg = (char_u *)_("Conversion with 'charconvert' failed");
2738 if (errmsg == NULL && (*fdp = mch_open((char *)tmpname,
2739 O_RDONLY | O_EXTRA, 0)) < 0)
2740 errmsg = (char_u *)_("can't read output of 'charconvert'");
2743 if (errmsg != NULL)
2745 /* Don't use emsg(), it breaks mappings, the retry with
2746 * another type of conversion might still work. */
2747 MSG(errmsg);
2748 if (tmpname != NULL)
2750 mch_remove(tmpname); /* delete converted file */
2751 vim_free(tmpname);
2752 tmpname = NULL;
2756 /* If the input file is closed, open it (caller should check for error). */
2757 if (*fdp < 0)
2758 *fdp = mch_open((char *)fname, O_RDONLY | O_EXTRA, 0);
2760 return tmpname;
2762 # endif
2764 #endif
2766 #ifdef FEAT_VIMINFO
2768 * Read marks for the current buffer from the viminfo file, when we support
2769 * buffer marks and the buffer has a name.
2771 static void
2772 check_marks_read()
2774 if (!curbuf->b_marks_read && get_viminfo_parameter('\'') > 0
2775 && curbuf->b_ffname != NULL)
2776 read_viminfo(NULL, VIF_WANT_MARKS);
2778 /* Always set b_marks_read; needed when 'viminfo' is changed to include
2779 * the ' parameter after opening a buffer. */
2780 curbuf->b_marks_read = TRUE;
2782 #endif
2784 #ifdef FEAT_CRYPT
2786 * Check for magic number used for encryption.
2787 * If found, the magic number is removed from ptr[*sizep] and *sizep and
2788 * *filesizep are updated.
2789 * Return the (new) encryption key, NULL for no encryption.
2791 static char_u *
2792 check_for_cryptkey(cryptkey, ptr, sizep, filesizep, newfile)
2793 char_u *cryptkey; /* previous encryption key or NULL */
2794 char_u *ptr; /* pointer to read bytes */
2795 long *sizep; /* length of read bytes */
2796 long *filesizep; /* nr of bytes used from file */
2797 int newfile; /* editing a new buffer */
2799 if (*sizep >= CRYPT_MAGIC_LEN
2800 && STRNCMP(ptr, CRYPT_MAGIC, CRYPT_MAGIC_LEN) == 0)
2802 if (cryptkey == NULL)
2804 if (*curbuf->b_p_key)
2805 cryptkey = curbuf->b_p_key;
2806 else
2808 /* When newfile is TRUE, store the typed key
2809 * in the 'key' option and don't free it. */
2810 cryptkey = get_crypt_key(newfile, FALSE);
2811 /* check if empty key entered */
2812 if (cryptkey != NULL && *cryptkey == NUL)
2814 if (cryptkey != curbuf->b_p_key)
2815 vim_free(cryptkey);
2816 cryptkey = NULL;
2821 if (cryptkey != NULL)
2823 crypt_init_keys(cryptkey);
2825 /* Remove magic number from the text */
2826 *filesizep += CRYPT_MAGIC_LEN;
2827 *sizep -= CRYPT_MAGIC_LEN;
2828 mch_memmove(ptr, ptr + CRYPT_MAGIC_LEN, (size_t)*sizep);
2831 /* When starting to edit a new file which does not have
2832 * encryption, clear the 'key' option, except when
2833 * starting up (called with -x argument) */
2834 else if (newfile && *curbuf->b_p_key && !starting)
2835 set_option_value((char_u *)"key", 0L, (char_u *)"", OPT_LOCAL);
2837 return cryptkey;
2839 #endif
2841 #ifdef UNIX
2842 static void
2843 set_file_time(fname, atime, mtime)
2844 char_u *fname;
2845 time_t atime; /* access time */
2846 time_t mtime; /* modification time */
2848 # if defined(HAVE_UTIME) && defined(HAVE_UTIME_H)
2849 struct utimbuf buf;
2851 buf.actime = atime;
2852 buf.modtime = mtime;
2853 (void)utime((char *)fname, &buf);
2854 # else
2855 # if defined(HAVE_UTIMES)
2856 struct timeval tvp[2];
2858 tvp[0].tv_sec = atime;
2859 tvp[0].tv_usec = 0;
2860 tvp[1].tv_sec = mtime;
2861 tvp[1].tv_usec = 0;
2862 # ifdef NeXT
2863 (void)utimes((char *)fname, tvp);
2864 # else
2865 (void)utimes((char *)fname, (const struct timeval *)&tvp);
2866 # endif
2867 # endif
2868 # endif
2870 #endif /* UNIX */
2872 #if defined(VMS) && !defined(MIN)
2873 /* Older DECC compiler for VAX doesn't define MIN() */
2874 # define MIN(a, b) ((a) < (b) ? (a) : (b))
2875 #endif
2878 * Return TRUE if a file appears to be read-only from the file permissions.
2881 check_file_readonly(fname, perm)
2882 char_u *fname; /* full path to file */
2883 int perm; /* known permissions on file */
2885 #ifndef USE_MCH_ACCESS
2886 int fd = 0;
2887 #endif
2889 return (
2890 #ifdef USE_MCH_ACCESS
2891 # ifdef UNIX
2892 (perm & 0222) == 0 ||
2893 # endif
2894 mch_access((char *)fname, W_OK)
2895 #else
2896 (fd = mch_open((char *)fname, O_RDWR | O_EXTRA, 0)) < 0
2897 ? TRUE : (close(fd), FALSE)
2898 #endif
2904 * buf_write() - write to file "fname" lines "start" through "end"
2906 * We do our own buffering here because fwrite() is so slow.
2908 * If "forceit" is true, we don't care for errors when attempting backups.
2909 * In case of an error everything possible is done to restore the original
2910 * file. But when "forceit" is TRUE, we risk losing it.
2912 * When "reset_changed" is TRUE and "append" == FALSE and "start" == 1 and
2913 * "end" == curbuf->b_ml.ml_line_count, reset curbuf->b_changed.
2915 * This function must NOT use NameBuff (because it's called by autowrite()).
2917 * return FAIL for failure, OK otherwise
2920 buf_write(buf, fname, sfname, start, end, eap, append, forceit,
2921 reset_changed, filtering)
2922 buf_T *buf;
2923 char_u *fname;
2924 char_u *sfname;
2925 linenr_T start, end;
2926 exarg_T *eap; /* for forced 'ff' and 'fenc', can be
2927 NULL! */
2928 int append; /* append to the file */
2929 int forceit;
2930 int reset_changed;
2931 int filtering;
2933 int fd;
2934 char_u *backup = NULL;
2935 int backup_copy = FALSE; /* copy the original file? */
2936 int dobackup;
2937 char_u *ffname;
2938 char_u *wfname = NULL; /* name of file to write to */
2939 char_u *s;
2940 char_u *ptr;
2941 char_u c;
2942 int len;
2943 linenr_T lnum;
2944 long nchars;
2945 char_u *errmsg = NULL;
2946 int errmsg_allocated = FALSE;
2947 char_u *errnum = NULL;
2948 char_u *buffer;
2949 char_u smallbuf[SMBUFSIZE];
2950 char_u *backup_ext;
2951 int bufsize;
2952 long perm; /* file permissions */
2953 int retval = OK;
2954 int newfile = FALSE; /* TRUE if file doesn't exist yet */
2955 int msg_save = msg_scroll;
2956 int overwriting; /* TRUE if writing over original */
2957 int no_eol = FALSE; /* no end-of-line written */
2958 int device = FALSE; /* writing to a device */
2959 struct stat st_old;
2960 int prev_got_int = got_int;
2961 int file_readonly = FALSE; /* overwritten file is read-only */
2962 static char *err_readonly = "is read-only (cannot override: \"W\" in 'cpoptions')";
2963 #if defined(UNIX) || defined(__EMX__XX) /*XXX fix me sometime? */
2964 int made_writable = FALSE; /* 'w' bit has been set */
2965 #endif
2966 /* writing everything */
2967 int whole = (start == 1 && end == buf->b_ml.ml_line_count);
2968 #ifdef FEAT_AUTOCMD
2969 linenr_T old_line_count = buf->b_ml.ml_line_count;
2970 #endif
2971 int attr;
2972 int fileformat;
2973 int write_bin;
2974 struct bw_info write_info; /* info for buf_write_bytes() */
2975 #ifdef FEAT_MBYTE
2976 int converted = FALSE;
2977 int notconverted = FALSE;
2978 char_u *fenc; /* effective 'fileencoding' */
2979 char_u *fenc_tofree = NULL; /* allocated "fenc" */
2980 #endif
2981 #ifdef HAS_BW_FLAGS
2982 int wb_flags = 0;
2983 #endif
2984 #ifdef HAVE_ACL
2985 vim_acl_T acl = NULL; /* ACL copied from original file to
2986 backup or new file */
2987 #endif
2989 if (fname == NULL || *fname == NUL) /* safety check */
2990 return FAIL;
2991 if (buf->b_ml.ml_mfp == NULL)
2993 /* This can happen during startup when there is a stray "w" in the
2994 * vimrc file. */
2995 EMSG(_(e_emptybuf));
2996 return FAIL;
3000 * Disallow writing from .exrc and .vimrc in current directory for
3001 * security reasons.
3003 if (check_secure())
3004 return FAIL;
3006 /* Avoid a crash for a long name. */
3007 if (STRLEN(fname) >= MAXPATHL)
3009 EMSG(_(e_longname));
3010 return FAIL;
3013 #ifdef FEAT_MBYTE
3014 /* must init bw_conv_buf and bw_iconv_fd before jumping to "fail" */
3015 write_info.bw_conv_buf = NULL;
3016 write_info.bw_conv_error = FALSE;
3017 write_info.bw_conv_error_lnum = 0;
3018 write_info.bw_restlen = 0;
3019 # ifdef USE_ICONV
3020 write_info.bw_iconv_fd = (iconv_t)-1;
3021 # endif
3022 #endif
3024 /* After writing a file changedtick changes but we don't want to display
3025 * the line. */
3026 ex_no_reprint = TRUE;
3029 * If there is no file name yet, use the one for the written file.
3030 * BF_NOTEDITED is set to reflect this (in case the write fails).
3031 * Don't do this when the write is for a filter command.
3032 * Don't do this when appending.
3033 * Only do this when 'cpoptions' contains the 'F' flag.
3035 if (buf->b_ffname == NULL
3036 && reset_changed
3037 && whole
3038 && buf == curbuf
3039 #ifdef FEAT_QUICKFIX
3040 && !bt_nofile(buf)
3041 #endif
3042 && !filtering
3043 && (!append || vim_strchr(p_cpo, CPO_FNAMEAPP) != NULL)
3044 && vim_strchr(p_cpo, CPO_FNAMEW) != NULL)
3046 if (set_rw_fname(fname, sfname) == FAIL)
3047 return FAIL;
3048 buf = curbuf; /* just in case autocmds made "buf" invalid */
3051 if (sfname == NULL)
3052 sfname = fname;
3054 * For Unix: Use the short file name whenever possible.
3055 * Avoids problems with networks and when directory names are changed.
3056 * Don't do this for MS-DOS, a "cd" in a sub-shell may have moved us to
3057 * another directory, which we don't detect
3059 ffname = fname; /* remember full fname */
3060 #ifdef UNIX
3061 fname = sfname;
3062 #endif
3064 if (buf->b_ffname != NULL && fnamecmp(ffname, buf->b_ffname) == 0)
3065 overwriting = TRUE;
3066 else
3067 overwriting = FALSE;
3069 if (exiting)
3070 settmode(TMODE_COOK); /* when exiting allow typahead now */
3072 ++no_wait_return; /* don't wait for return yet */
3075 * Set '[ and '] marks to the lines to be written.
3077 buf->b_op_start.lnum = start;
3078 buf->b_op_start.col = 0;
3079 buf->b_op_end.lnum = end;
3080 buf->b_op_end.col = 0;
3082 #ifdef FEAT_AUTOCMD
3084 aco_save_T aco;
3085 int buf_ffname = FALSE;
3086 int buf_sfname = FALSE;
3087 int buf_fname_f = FALSE;
3088 int buf_fname_s = FALSE;
3089 int did_cmd = FALSE;
3090 int nofile_err = FALSE;
3091 int empty_memline = (buf->b_ml.ml_mfp == NULL);
3094 * Apply PRE aucocommands.
3095 * Set curbuf to the buffer to be written.
3096 * Careful: The autocommands may call buf_write() recursively!
3098 if (ffname == buf->b_ffname)
3099 buf_ffname = TRUE;
3100 if (sfname == buf->b_sfname)
3101 buf_sfname = TRUE;
3102 if (fname == buf->b_ffname)
3103 buf_fname_f = TRUE;
3104 if (fname == buf->b_sfname)
3105 buf_fname_s = TRUE;
3107 /* set curwin/curbuf to buf and save a few things */
3108 aucmd_prepbuf(&aco, buf);
3110 if (append)
3112 if (!(did_cmd = apply_autocmds_exarg(EVENT_FILEAPPENDCMD,
3113 sfname, sfname, FALSE, curbuf, eap)))
3115 #ifdef FEAT_QUICKFIX
3116 if (overwriting && bt_nofile(curbuf))
3117 nofile_err = TRUE;
3118 else
3119 #endif
3120 apply_autocmds_exarg(EVENT_FILEAPPENDPRE,
3121 sfname, sfname, FALSE, curbuf, eap);
3124 else if (filtering)
3126 apply_autocmds_exarg(EVENT_FILTERWRITEPRE,
3127 NULL, sfname, FALSE, curbuf, eap);
3129 else if (reset_changed && whole)
3131 if (!(did_cmd = apply_autocmds_exarg(EVENT_BUFWRITECMD,
3132 sfname, sfname, FALSE, curbuf, eap)))
3134 #ifdef FEAT_QUICKFIX
3135 if (overwriting && bt_nofile(curbuf))
3136 nofile_err = TRUE;
3137 else
3138 #endif
3139 apply_autocmds_exarg(EVENT_BUFWRITEPRE,
3140 sfname, sfname, FALSE, curbuf, eap);
3143 else
3145 if (!(did_cmd = apply_autocmds_exarg(EVENT_FILEWRITECMD,
3146 sfname, sfname, FALSE, curbuf, eap)))
3148 #ifdef FEAT_QUICKFIX
3149 if (overwriting && bt_nofile(curbuf))
3150 nofile_err = TRUE;
3151 else
3152 #endif
3153 apply_autocmds_exarg(EVENT_FILEWRITEPRE,
3154 sfname, sfname, FALSE, curbuf, eap);
3158 /* restore curwin/curbuf and a few other things */
3159 aucmd_restbuf(&aco);
3162 * In three situations we return here and don't write the file:
3163 * 1. the autocommands deleted or unloaded the buffer.
3164 * 2. The autocommands abort script processing.
3165 * 3. If one of the "Cmd" autocommands was executed.
3167 if (!buf_valid(buf))
3168 buf = NULL;
3169 if (buf == NULL || (buf->b_ml.ml_mfp == NULL && !empty_memline)
3170 || did_cmd || nofile_err
3171 #ifdef FEAT_EVAL
3172 || aborting()
3173 #endif
3176 --no_wait_return;
3177 msg_scroll = msg_save;
3178 if (nofile_err)
3179 EMSG(_("E676: No matching autocommands for acwrite buffer"));
3181 if (nofile_err
3182 #ifdef FEAT_EVAL
3183 || aborting()
3184 #endif
3186 /* An aborting error, interrupt or exception in the
3187 * autocommands. */
3188 return FAIL;
3189 if (did_cmd)
3191 if (buf == NULL)
3192 /* The buffer was deleted. We assume it was written
3193 * (can't retry anyway). */
3194 return OK;
3195 if (overwriting)
3197 /* Assume the buffer was written, update the timestamp. */
3198 ml_timestamp(buf);
3199 if (append)
3200 buf->b_flags &= ~BF_NEW;
3201 else
3202 buf->b_flags &= ~BF_WRITE_MASK;
3204 if (reset_changed && buf->b_changed && !append
3205 && (overwriting || vim_strchr(p_cpo, CPO_PLUS) != NULL))
3206 /* Buffer still changed, the autocommands didn't work
3207 * properly. */
3208 return FAIL;
3209 return OK;
3211 #ifdef FEAT_EVAL
3212 if (!aborting())
3213 #endif
3214 EMSG(_("E203: Autocommands deleted or unloaded buffer to be written"));
3215 return FAIL;
3219 * The autocommands may have changed the number of lines in the file.
3220 * When writing the whole file, adjust the end.
3221 * When writing part of the file, assume that the autocommands only
3222 * changed the number of lines that are to be written (tricky!).
3224 if (buf->b_ml.ml_line_count != old_line_count)
3226 if (whole) /* write all */
3227 end = buf->b_ml.ml_line_count;
3228 else if (buf->b_ml.ml_line_count > old_line_count) /* more lines */
3229 end += buf->b_ml.ml_line_count - old_line_count;
3230 else /* less lines */
3232 end -= old_line_count - buf->b_ml.ml_line_count;
3233 if (end < start)
3235 --no_wait_return;
3236 msg_scroll = msg_save;
3237 EMSG(_("E204: Autocommand changed number of lines in unexpected way"));
3238 return FAIL;
3244 * The autocommands may have changed the name of the buffer, which may
3245 * be kept in fname, ffname and sfname.
3247 if (buf_ffname)
3248 ffname = buf->b_ffname;
3249 if (buf_sfname)
3250 sfname = buf->b_sfname;
3251 if (buf_fname_f)
3252 fname = buf->b_ffname;
3253 if (buf_fname_s)
3254 fname = buf->b_sfname;
3256 #endif
3258 #ifdef FEAT_NETBEANS_INTG
3259 if (usingNetbeans && isNetbeansBuffer(buf))
3261 if (whole)
3264 * b_changed can be 0 after an undo, but we still need to write
3265 * the buffer to NetBeans.
3267 if (buf->b_changed || isNetbeansModified(buf))
3269 --no_wait_return; /* may wait for return now */
3270 msg_scroll = msg_save;
3271 netbeans_save_buffer(buf); /* no error checking... */
3272 return retval;
3274 else
3276 errnum = (char_u *)"E656: ";
3277 errmsg = (char_u *)_("NetBeans disallows writes of unmodified buffers");
3278 buffer = NULL;
3279 goto fail;
3282 else
3284 errnum = (char_u *)"E657: ";
3285 errmsg = (char_u *)_("Partial writes disallowed for NetBeans buffers");
3286 buffer = NULL;
3287 goto fail;
3290 #endif
3292 if (shortmess(SHM_OVER) && !exiting)
3293 msg_scroll = FALSE; /* overwrite previous file message */
3294 else
3295 msg_scroll = TRUE; /* don't overwrite previous file message */
3296 if (!filtering)
3297 filemess(buf,
3298 #ifndef UNIX
3299 sfname,
3300 #else
3301 fname,
3302 #endif
3303 (char_u *)"", 0); /* show that we are busy */
3304 msg_scroll = FALSE; /* always overwrite the file message now */
3306 buffer = alloc(BUFSIZE);
3307 if (buffer == NULL) /* can't allocate big buffer, use small
3308 * one (to be able to write when out of
3309 * memory) */
3311 buffer = smallbuf;
3312 bufsize = SMBUFSIZE;
3314 else
3315 bufsize = BUFSIZE;
3318 * Get information about original file (if there is one).
3320 #if defined(UNIX) && !defined(ARCHIE)
3321 st_old.st_dev = 0;
3322 st_old.st_ino = 0;
3323 perm = -1;
3324 if (mch_stat((char *)fname, &st_old) < 0)
3325 newfile = TRUE;
3326 else
3328 perm = st_old.st_mode;
3329 if (!S_ISREG(st_old.st_mode)) /* not a file */
3331 if (S_ISDIR(st_old.st_mode))
3333 errnum = (char_u *)"E502: ";
3334 errmsg = (char_u *)_("is a directory");
3335 goto fail;
3337 if (mch_nodetype(fname) != NODE_WRITABLE)
3339 errnum = (char_u *)"E503: ";
3340 errmsg = (char_u *)_("is not a file or writable device");
3341 goto fail;
3343 /* It's a device of some kind (or a fifo) which we can write to
3344 * but for which we can't make a backup. */
3345 device = TRUE;
3346 newfile = TRUE;
3347 perm = -1;
3350 #else /* !UNIX */
3352 * Check for a writable device name.
3354 c = mch_nodetype(fname);
3355 if (c == NODE_OTHER)
3357 errnum = (char_u *)"E503: ";
3358 errmsg = (char_u *)_("is not a file or writable device");
3359 goto fail;
3361 if (c == NODE_WRITABLE)
3363 # if defined(MSDOS) || defined(MSWIN) || defined(OS2)
3364 /* MS-Windows allows opening a device, but we will probably get stuck
3365 * trying to write to it. */
3366 if (!p_odev)
3368 errnum = (char_u *)"E796: ";
3369 errmsg = (char_u *)_("writing to device disabled with 'opendevice' option");
3370 goto fail;
3372 # endif
3373 device = TRUE;
3374 newfile = TRUE;
3375 perm = -1;
3377 else
3379 perm = mch_getperm(fname);
3380 if (perm < 0)
3381 newfile = TRUE;
3382 else if (mch_isdir(fname))
3384 errnum = (char_u *)"E502: ";
3385 errmsg = (char_u *)_("is a directory");
3386 goto fail;
3388 if (overwriting)
3389 (void)mch_stat((char *)fname, &st_old);
3391 #endif /* !UNIX */
3393 if (!device && !newfile)
3396 * Check if the file is really writable (when renaming the file to
3397 * make a backup we won't discover it later).
3399 file_readonly = check_file_readonly(fname, (int)perm);
3401 if (!forceit && file_readonly)
3403 if (vim_strchr(p_cpo, CPO_FWRITE) != NULL)
3405 errnum = (char_u *)"E504: ";
3406 errmsg = (char_u *)_(err_readonly);
3408 else
3410 errnum = (char_u *)"E505: ";
3411 errmsg = (char_u *)_("is read-only (add ! to override)");
3413 goto fail;
3417 * Check if the timestamp hasn't changed since reading the file.
3419 if (overwriting)
3421 retval = check_mtime(buf, &st_old);
3422 if (retval == FAIL)
3423 goto fail;
3427 #ifdef HAVE_ACL
3429 * For systems that support ACL: get the ACL from the original file.
3431 if (!newfile)
3432 acl = mch_get_acl(fname);
3433 #endif
3436 * If 'backupskip' is not empty, don't make a backup for some files.
3438 dobackup = (p_wb || p_bk || *p_pm != NUL);
3439 #ifdef FEAT_WILDIGN
3440 if (dobackup && *p_bsk != NUL && match_file_list(p_bsk, sfname, ffname))
3441 dobackup = FALSE;
3442 #endif
3445 * Save the value of got_int and reset it. We don't want a previous
3446 * interruption cancel writing, only hitting CTRL-C while writing should
3447 * abort it.
3449 prev_got_int = got_int;
3450 got_int = FALSE;
3452 /* Mark the buffer as 'being saved' to prevent changed buffer warnings */
3453 buf->b_saving = TRUE;
3456 * If we are not appending or filtering, the file exists, and the
3457 * 'writebackup', 'backup' or 'patchmode' option is set, need a backup.
3458 * When 'patchmode' is set also make a backup when appending.
3460 * Do not make any backup, if 'writebackup' and 'backup' are both switched
3461 * off. This helps when editing large files on almost-full disks.
3463 if (!(append && *p_pm == NUL) && !filtering && perm >= 0 && dobackup)
3465 #if defined(UNIX) || defined(WIN32)
3466 struct stat st;
3467 #endif
3469 if ((bkc_flags & BKC_YES) || append) /* "yes" */
3470 backup_copy = TRUE;
3471 #if defined(UNIX) || defined(WIN32)
3472 else if ((bkc_flags & BKC_AUTO)) /* "auto" */
3474 int i;
3476 # ifdef UNIX
3478 * Don't rename the file when:
3479 * - it's a hard link
3480 * - it's a symbolic link
3481 * - we don't have write permission in the directory
3482 * - we can't set the owner/group of the new file
3484 if (st_old.st_nlink > 1
3485 || mch_lstat((char *)fname, &st) < 0
3486 || st.st_dev != st_old.st_dev
3487 || st.st_ino != st_old.st_ino
3488 # ifndef HAVE_FCHOWN
3489 || st.st_uid != st_old.st_uid
3490 || st.st_gid != st_old.st_gid
3491 # endif
3493 backup_copy = TRUE;
3494 else
3495 # else
3496 # ifdef WIN32
3497 /* On NTFS file systems hard links are possible. */
3498 if (mch_is_linked(fname))
3499 backup_copy = TRUE;
3500 else
3501 # endif
3502 # endif
3505 * Check if we can create a file and set the owner/group to
3506 * the ones from the original file.
3507 * First find a file name that doesn't exist yet (use some
3508 * arbitrary numbers).
3510 STRCPY(IObuff, fname);
3511 for (i = 4913; ; i += 123)
3513 sprintf((char *)gettail(IObuff), "%d", i);
3514 if (mch_lstat((char *)IObuff, &st) < 0)
3515 break;
3517 fd = mch_open((char *)IObuff,
3518 O_CREAT|O_WRONLY|O_EXCL|O_NOFOLLOW, perm);
3519 if (fd < 0) /* can't write in directory */
3520 backup_copy = TRUE;
3521 else
3523 # ifdef UNIX
3524 # ifdef HAVE_FCHOWN
3525 ignored = fchown(fd, st_old.st_uid, st_old.st_gid);
3526 # endif
3527 if (mch_stat((char *)IObuff, &st) < 0
3528 || st.st_uid != st_old.st_uid
3529 || st.st_gid != st_old.st_gid
3530 || (long)st.st_mode != perm)
3531 backup_copy = TRUE;
3532 # endif
3533 /* Close the file before removing it, on MS-Windows we
3534 * can't delete an open file. */
3535 close(fd);
3536 mch_remove(IObuff);
3541 # ifdef UNIX
3543 * Break symlinks and/or hardlinks if we've been asked to.
3545 if ((bkc_flags & BKC_BREAKSYMLINK) || (bkc_flags & BKC_BREAKHARDLINK))
3547 int lstat_res;
3549 lstat_res = mch_lstat((char *)fname, &st);
3551 /* Symlinks. */
3552 if ((bkc_flags & BKC_BREAKSYMLINK)
3553 && lstat_res == 0
3554 && st.st_ino != st_old.st_ino)
3555 backup_copy = FALSE;
3557 /* Hardlinks. */
3558 if ((bkc_flags & BKC_BREAKHARDLINK)
3559 && st_old.st_nlink > 1
3560 && (lstat_res != 0 || st.st_ino == st_old.st_ino))
3561 backup_copy = FALSE;
3563 #endif
3565 #endif
3567 /* make sure we have a valid backup extension to use */
3568 if (*p_bex == NUL)
3570 #ifdef RISCOS
3571 backup_ext = (char_u *)"/bak";
3572 #else
3573 backup_ext = (char_u *)".bak";
3574 #endif
3576 else
3577 backup_ext = p_bex;
3579 if (backup_copy
3580 && (fd = mch_open((char *)fname, O_RDONLY | O_EXTRA, 0)) >= 0)
3582 int bfd;
3583 char_u *copybuf, *wp;
3584 int some_error = FALSE;
3585 struct stat st_new;
3586 char_u *dirp;
3587 char_u *rootname;
3588 #if defined(UNIX) && !defined(SHORT_FNAME)
3589 int did_set_shortname;
3590 #endif
3592 copybuf = alloc(BUFSIZE + 1);
3593 if (copybuf == NULL)
3595 some_error = TRUE; /* out of memory */
3596 goto nobackup;
3600 * Try to make the backup in each directory in the 'bdir' option.
3602 * Unix semantics has it, that we may have a writable file,
3603 * that cannot be recreated with a simple open(..., O_CREAT, ) e.g:
3604 * - the directory is not writable,
3605 * - the file may be a symbolic link,
3606 * - the file may belong to another user/group, etc.
3608 * For these reasons, the existing writable file must be truncated
3609 * and reused. Creation of a backup COPY will be attempted.
3611 dirp = p_bdir;
3612 while (*dirp)
3614 #ifdef UNIX
3615 st_new.st_ino = 0;
3616 st_new.st_dev = 0;
3617 st_new.st_gid = 0;
3618 #endif
3621 * Isolate one directory name, using an entry in 'bdir'.
3623 (void)copy_option_part(&dirp, copybuf, BUFSIZE, ",");
3624 rootname = get_file_in_dir(fname, copybuf);
3625 if (rootname == NULL)
3627 some_error = TRUE; /* out of memory */
3628 goto nobackup;
3631 #if defined(UNIX) && !defined(SHORT_FNAME)
3632 did_set_shortname = FALSE;
3633 #endif
3636 * May try twice if 'shortname' not set.
3638 for (;;)
3641 * Make backup file name.
3643 backup = buf_modname(
3644 #ifdef SHORT_FNAME
3645 TRUE,
3646 #else
3647 (buf->b_p_sn || buf->b_shortname),
3648 #endif
3649 rootname, backup_ext, FALSE);
3650 if (backup == NULL)
3652 vim_free(rootname);
3653 some_error = TRUE; /* out of memory */
3654 goto nobackup;
3658 * Check if backup file already exists.
3660 if (mch_stat((char *)backup, &st_new) >= 0)
3662 #ifdef UNIX
3664 * Check if backup file is same as original file.
3665 * May happen when modname() gave the same file back.
3666 * E.g. silly link, or file name-length reached.
3667 * If we don't check here, we either ruin the file
3668 * when copying or erase it after writing. jw.
3670 if (st_new.st_dev == st_old.st_dev
3671 && st_new.st_ino == st_old.st_ino)
3673 vim_free(backup);
3674 backup = NULL; /* no backup file to delete */
3675 # ifndef SHORT_FNAME
3677 * may try again with 'shortname' set
3679 if (!(buf->b_shortname || buf->b_p_sn))
3681 buf->b_shortname = TRUE;
3682 did_set_shortname = TRUE;
3683 continue;
3685 /* setting shortname didn't help */
3686 if (did_set_shortname)
3687 buf->b_shortname = FALSE;
3688 # endif
3689 break;
3691 #endif
3694 * If we are not going to keep the backup file, don't
3695 * delete an existing one, try to use another name.
3696 * Change one character, just before the extension.
3698 if (!p_bk)
3700 wp = backup + STRLEN(backup) - 1
3701 - STRLEN(backup_ext);
3702 if (wp < backup) /* empty file name ??? */
3703 wp = backup;
3704 *wp = 'z';
3705 while (*wp > 'a'
3706 && mch_stat((char *)backup, &st_new) >= 0)
3707 --*wp;
3708 /* They all exist??? Must be something wrong. */
3709 if (*wp == 'a')
3711 vim_free(backup);
3712 backup = NULL;
3716 break;
3718 vim_free(rootname);
3721 * Try to create the backup file
3723 if (backup != NULL)
3725 /* remove old backup, if present */
3726 mch_remove(backup);
3727 /* Open with O_EXCL to avoid the file being created while
3728 * we were sleeping (symlink hacker attack?) */
3729 bfd = mch_open((char *)backup,
3730 O_WRONLY|O_CREAT|O_EXTRA|O_EXCL|O_NOFOLLOW,
3731 perm & 0777);
3732 if (bfd < 0)
3734 vim_free(backup);
3735 backup = NULL;
3737 else
3739 /* set file protection same as original file, but
3740 * strip s-bit */
3741 (void)mch_setperm(backup, perm & 0777);
3743 #ifdef UNIX
3745 * Try to set the group of the backup same as the
3746 * original file. If this fails, set the protection
3747 * bits for the group same as the protection bits for
3748 * others.
3750 if (st_new.st_gid != st_old.st_gid
3751 # ifdef HAVE_FCHOWN /* sequent-ptx lacks fchown() */
3752 && fchown(bfd, (uid_t)-1, st_old.st_gid) != 0
3753 # endif
3755 mch_setperm(backup,
3756 (perm & 0707) | ((perm & 07) << 3));
3757 # ifdef HAVE_SELINUX
3758 mch_copy_sec(fname, backup);
3759 # endif
3760 #endif
3763 * copy the file.
3765 write_info.bw_fd = bfd;
3766 write_info.bw_buf = copybuf;
3767 #ifdef HAS_BW_FLAGS
3768 write_info.bw_flags = FIO_NOCONVERT;
3769 #endif
3770 while ((write_info.bw_len = vim_read(fd, copybuf,
3771 BUFSIZE)) > 0)
3773 if (buf_write_bytes(&write_info) == FAIL)
3775 errmsg = (char_u *)_("E506: Can't write to backup file (add ! to override)");
3776 break;
3778 ui_breakcheck();
3779 if (got_int)
3781 errmsg = (char_u *)_(e_interr);
3782 break;
3786 if (close(bfd) < 0 && errmsg == NULL)
3787 errmsg = (char_u *)_("E507: Close error for backup file (add ! to override)");
3788 if (write_info.bw_len < 0)
3789 errmsg = (char_u *)_("E508: Can't read file for backup (add ! to override)");
3790 #ifdef UNIX
3791 set_file_time(backup, st_old.st_atime, st_old.st_mtime);
3792 #endif
3793 #ifdef HAVE_ACL
3794 mch_set_acl(backup, acl);
3795 #endif
3796 #ifdef HAVE_SELINUX
3797 mch_copy_sec(fname, backup);
3798 #endif
3799 break;
3803 nobackup:
3804 close(fd); /* ignore errors for closing read file */
3805 vim_free(copybuf);
3807 if (backup == NULL && errmsg == NULL)
3808 errmsg = (char_u *)_("E509: Cannot create backup file (add ! to override)");
3809 /* ignore errors when forceit is TRUE */
3810 if ((some_error || errmsg != NULL) && !forceit)
3812 retval = FAIL;
3813 goto fail;
3815 errmsg = NULL;
3817 else
3819 char_u *dirp;
3820 char_u *p;
3821 char_u *rootname;
3824 * Make a backup by renaming the original file.
3827 * If 'cpoptions' includes the "W" flag, we don't want to
3828 * overwrite a read-only file. But rename may be possible
3829 * anyway, thus we need an extra check here.
3831 if (file_readonly && vim_strchr(p_cpo, CPO_FWRITE) != NULL)
3833 errnum = (char_u *)"E504: ";
3834 errmsg = (char_u *)_(err_readonly);
3835 goto fail;
3840 * Form the backup file name - change path/fo.o.h to
3841 * path/fo.o.h.bak Try all directories in 'backupdir', first one
3842 * that works is used.
3844 dirp = p_bdir;
3845 while (*dirp)
3848 * Isolate one directory name and make the backup file name.
3850 (void)copy_option_part(&dirp, IObuff, IOSIZE, ",");
3851 rootname = get_file_in_dir(fname, IObuff);
3852 if (rootname == NULL)
3853 backup = NULL;
3854 else
3856 backup = buf_modname(
3857 #ifdef SHORT_FNAME
3858 TRUE,
3859 #else
3860 (buf->b_p_sn || buf->b_shortname),
3861 #endif
3862 rootname, backup_ext, FALSE);
3863 vim_free(rootname);
3866 if (backup != NULL)
3869 * If we are not going to keep the backup file, don't
3870 * delete an existing one, try to use another name.
3871 * Change one character, just before the extension.
3873 if (!p_bk && mch_getperm(backup) >= 0)
3875 p = backup + STRLEN(backup) - 1 - STRLEN(backup_ext);
3876 if (p < backup) /* empty file name ??? */
3877 p = backup;
3878 *p = 'z';
3879 while (*p > 'a' && mch_getperm(backup) >= 0)
3880 --*p;
3881 /* They all exist??? Must be something wrong! */
3882 if (*p == 'a')
3884 vim_free(backup);
3885 backup = NULL;
3889 if (backup != NULL)
3892 * Delete any existing backup and move the current version
3893 * to the backup. For safety, we don't remove the backup
3894 * until the write has finished successfully. And if the
3895 * 'backup' option is set, leave it around.
3898 * If the renaming of the original file to the backup file
3899 * works, quit here.
3901 if (vim_rename(fname, backup) == 0)
3902 break;
3904 vim_free(backup); /* don't do the rename below */
3905 backup = NULL;
3908 if (backup == NULL && !forceit)
3910 errmsg = (char_u *)_("E510: Can't make backup file (add ! to override)");
3911 goto fail;
3916 #if defined(UNIX) && !defined(ARCHIE)
3917 /* When using ":w!" and the file was read-only: make it writable */
3918 if (forceit && perm >= 0 && !(perm & 0200) && st_old.st_uid == getuid()
3919 && vim_strchr(p_cpo, CPO_FWRITE) == NULL)
3921 perm |= 0200;
3922 (void)mch_setperm(fname, perm);
3923 made_writable = TRUE;
3925 #endif
3927 /* When using ":w!" and writing to the current file, 'readonly' makes no
3928 * sense, reset it, unless 'Z' appears in 'cpoptions'. */
3929 if (forceit && overwriting && vim_strchr(p_cpo, CPO_KEEPRO) == NULL)
3931 buf->b_p_ro = FALSE;
3932 #ifdef FEAT_TITLE
3933 need_maketitle = TRUE; /* set window title later */
3934 #endif
3935 #ifdef FEAT_WINDOWS
3936 status_redraw_all(); /* redraw status lines later */
3937 #endif
3940 if (end > buf->b_ml.ml_line_count)
3941 end = buf->b_ml.ml_line_count;
3942 if (buf->b_ml.ml_flags & ML_EMPTY)
3943 start = end + 1;
3946 * If the original file is being overwritten, there is a small chance that
3947 * we crash in the middle of writing. Therefore the file is preserved now.
3948 * This makes all block numbers positive so that recovery does not need
3949 * the original file.
3950 * Don't do this if there is a backup file and we are exiting.
3952 if (reset_changed && !newfile && overwriting
3953 && !(exiting && backup != NULL))
3955 ml_preserve(buf, FALSE);
3956 if (got_int)
3958 errmsg = (char_u *)_(e_interr);
3959 goto restore_backup;
3963 #ifdef MACOS_CLASSIC /* TODO: Is it need for MACOS_X? (Dany) */
3965 * Before risking to lose the original file verify if there's
3966 * a resource fork to preserve, and if cannot be done warn
3967 * the users. This happens when overwriting without backups.
3969 if (backup == NULL && overwriting && !append)
3970 if (mch_has_resource_fork(fname))
3972 errmsg = (char_u *)_("E460: The resource fork would be lost (add ! to override)");
3973 goto restore_backup;
3975 #endif
3977 #ifdef VMS
3978 vms_remove_version(fname); /* remove version */
3979 #endif
3980 /* Default: write the file directly. May write to a temp file for
3981 * multi-byte conversion. */
3982 wfname = fname;
3984 #ifdef FEAT_MBYTE
3985 /* Check for forced 'fileencoding' from "++opt=val" argument. */
3986 if (eap != NULL && eap->force_enc != 0)
3988 fenc = eap->cmd + eap->force_enc;
3989 fenc = enc_canonize(fenc);
3990 fenc_tofree = fenc;
3992 else
3993 fenc = buf->b_p_fenc;
3996 * Check if the file needs to be converted.
3998 converted = need_conversion(fenc);
4001 * Check if UTF-8 to UCS-2/4 or Latin1 conversion needs to be done. Or
4002 * Latin1 to Unicode conversion. This is handled in buf_write_bytes().
4003 * Prepare the flags for it and allocate bw_conv_buf when needed.
4005 if (converted && (enc_utf8 || STRCMP(p_enc, "latin1") == 0))
4007 wb_flags = get_fio_flags(fenc);
4008 if (wb_flags & (FIO_UCS2 | FIO_UCS4 | FIO_UTF16 | FIO_UTF8))
4010 /* Need to allocate a buffer to translate into. */
4011 if (wb_flags & (FIO_UCS2 | FIO_UTF16 | FIO_UTF8))
4012 write_info.bw_conv_buflen = bufsize * 2;
4013 else /* FIO_UCS4 */
4014 write_info.bw_conv_buflen = bufsize * 4;
4015 write_info.bw_conv_buf
4016 = lalloc((long_u)write_info.bw_conv_buflen, TRUE);
4017 if (write_info.bw_conv_buf == NULL)
4018 end = 0;
4022 # ifdef WIN3264
4023 if (converted && wb_flags == 0 && (wb_flags = get_win_fio_flags(fenc)) != 0)
4025 /* Convert UTF-8 -> UCS-2 and UCS-2 -> DBCS. Worst-case * 4: */
4026 write_info.bw_conv_buflen = bufsize * 4;
4027 write_info.bw_conv_buf
4028 = lalloc((long_u)write_info.bw_conv_buflen, TRUE);
4029 if (write_info.bw_conv_buf == NULL)
4030 end = 0;
4032 # endif
4034 # ifdef MACOS_X
4035 if (converted && wb_flags == 0 && (wb_flags = get_mac_fio_flags(fenc)) != 0)
4037 write_info.bw_conv_buflen = bufsize * 3;
4038 write_info.bw_conv_buf
4039 = lalloc((long_u)write_info.bw_conv_buflen, TRUE);
4040 if (write_info.bw_conv_buf == NULL)
4041 end = 0;
4043 # endif
4045 # if defined(FEAT_EVAL) || defined(USE_ICONV)
4046 if (converted && wb_flags == 0)
4048 # ifdef USE_ICONV
4050 * Use iconv() conversion when conversion is needed and it's not done
4051 * internally.
4053 write_info.bw_iconv_fd = (iconv_t)my_iconv_open(fenc,
4054 enc_utf8 ? (char_u *)"utf-8" : p_enc);
4055 if (write_info.bw_iconv_fd != (iconv_t)-1)
4057 /* We're going to use iconv(), allocate a buffer to convert in. */
4058 write_info.bw_conv_buflen = bufsize * ICONV_MULT;
4059 write_info.bw_conv_buf
4060 = lalloc((long_u)write_info.bw_conv_buflen, TRUE);
4061 if (write_info.bw_conv_buf == NULL)
4062 end = 0;
4063 write_info.bw_first = TRUE;
4065 # ifdef FEAT_EVAL
4066 else
4067 # endif
4068 # endif
4070 # ifdef FEAT_EVAL
4072 * When the file needs to be converted with 'charconvert' after
4073 * writing, write to a temp file instead and let the conversion
4074 * overwrite the original file.
4076 if (*p_ccv != NUL)
4078 wfname = vim_tempname('w');
4079 if (wfname == NULL) /* Can't write without a tempfile! */
4081 errmsg = (char_u *)_("E214: Can't find temp file for writing");
4082 goto restore_backup;
4085 # endif
4087 # endif
4088 if (converted && wb_flags == 0
4089 # ifdef USE_ICONV
4090 && write_info.bw_iconv_fd == (iconv_t)-1
4091 # endif
4092 # ifdef FEAT_EVAL
4093 && wfname == fname
4094 # endif
4097 if (!forceit)
4099 errmsg = (char_u *)_("E213: Cannot convert (add ! to write without conversion)");
4100 goto restore_backup;
4102 notconverted = TRUE;
4104 #endif
4107 * Open the file "wfname" for writing.
4108 * We may try to open the file twice: If we can't write to the
4109 * file and forceit is TRUE we delete the existing file and try to create
4110 * a new one. If this still fails we may have lost the original file!
4111 * (this may happen when the user reached his quotum for number of files).
4112 * Appending will fail if the file does not exist and forceit is FALSE.
4114 while ((fd = mch_open((char *)wfname, O_WRONLY | O_EXTRA | (append
4115 ? (forceit ? (O_APPEND | O_CREAT) : O_APPEND)
4116 : (O_CREAT | O_TRUNC))
4117 , perm < 0 ? 0666 : (perm & 0777))) < 0)
4120 * A forced write will try to create a new file if the old one is
4121 * still readonly. This may also happen when the directory is
4122 * read-only. In that case the mch_remove() will fail.
4124 if (errmsg == NULL)
4126 #ifdef UNIX
4127 struct stat st;
4129 /* Don't delete the file when it's a hard or symbolic link. */
4130 if ((!newfile && st_old.st_nlink > 1)
4131 || (mch_lstat((char *)fname, &st) == 0
4132 && (st.st_dev != st_old.st_dev
4133 || st.st_ino != st_old.st_ino)))
4134 errmsg = (char_u *)_("E166: Can't open linked file for writing");
4135 else
4136 #endif
4138 errmsg = (char_u *)_("E212: Can't open file for writing");
4139 if (forceit && vim_strchr(p_cpo, CPO_FWRITE) == NULL
4140 && perm >= 0)
4142 #ifdef UNIX
4143 /* we write to the file, thus it should be marked
4144 writable after all */
4145 if (!(perm & 0200))
4146 made_writable = TRUE;
4147 perm |= 0200;
4148 if (st_old.st_uid != getuid() || st_old.st_gid != getgid())
4149 perm &= 0777;
4150 #endif
4151 if (!append) /* don't remove when appending */
4152 mch_remove(wfname);
4153 continue;
4158 restore_backup:
4160 struct stat st;
4163 * If we failed to open the file, we don't need a backup. Throw it
4164 * away. If we moved or removed the original file try to put the
4165 * backup in its place.
4167 if (backup != NULL && wfname == fname)
4169 if (backup_copy)
4172 * There is a small chance that we removed the original,
4173 * try to move the copy in its place.
4174 * This may not work if the vim_rename() fails.
4175 * In that case we leave the copy around.
4177 /* If file does not exist, put the copy in its place */
4178 if (mch_stat((char *)fname, &st) < 0)
4179 vim_rename(backup, fname);
4180 /* if original file does exist throw away the copy */
4181 if (mch_stat((char *)fname, &st) >= 0)
4182 mch_remove(backup);
4184 else
4186 /* try to put the original file back */
4187 vim_rename(backup, fname);
4191 /* if original file no longer exists give an extra warning */
4192 if (!newfile && mch_stat((char *)fname, &st) < 0)
4193 end = 0;
4196 #ifdef FEAT_MBYTE
4197 if (wfname != fname)
4198 vim_free(wfname);
4199 #endif
4200 goto fail;
4202 errmsg = NULL;
4204 #if defined(MACOS_CLASSIC) || defined(WIN3264)
4205 /* TODO: Is it need for MACOS_X? (Dany) */
4207 * On macintosh copy the original files attributes (i.e. the backup)
4208 * This is done in order to preserve the resource fork and the
4209 * Finder attribute (label, comments, custom icons, file creator)
4211 if (backup != NULL && overwriting && !append)
4213 if (backup_copy)
4214 (void)mch_copy_file_attribute(wfname, backup);
4215 else
4216 (void)mch_copy_file_attribute(backup, wfname);
4219 if (!overwriting && !append)
4221 if (buf->b_ffname != NULL)
4222 (void)mch_copy_file_attribute(buf->b_ffname, wfname);
4223 /* Should copy resource fork */
4225 #endif
4227 write_info.bw_fd = fd;
4229 #ifdef FEAT_CRYPT
4230 if (*buf->b_p_key && !filtering)
4232 crypt_init_keys(buf->b_p_key);
4233 /* Write magic number, so that Vim knows that this file is encrypted
4234 * when reading it again. This also undergoes utf-8 to ucs-2/4
4235 * conversion when needed. */
4236 write_info.bw_buf = (char_u *)CRYPT_MAGIC;
4237 write_info.bw_len = CRYPT_MAGIC_LEN;
4238 write_info.bw_flags = FIO_NOCONVERT;
4239 if (buf_write_bytes(&write_info) == FAIL)
4240 end = 0;
4241 wb_flags |= FIO_ENCRYPTED;
4243 #endif
4245 write_info.bw_buf = buffer;
4246 nchars = 0;
4248 /* use "++bin", "++nobin" or 'binary' */
4249 if (eap != NULL && eap->force_bin != 0)
4250 write_bin = (eap->force_bin == FORCE_BIN);
4251 else
4252 write_bin = buf->b_p_bin;
4254 #ifdef FEAT_MBYTE
4256 * The BOM is written just after the encryption magic number.
4257 * Skip it when appending and the file already existed, the BOM only makes
4258 * sense at the start of the file.
4260 if (buf->b_p_bomb && !write_bin && (!append || perm < 0))
4262 write_info.bw_len = make_bom(buffer, fenc);
4263 if (write_info.bw_len > 0)
4265 /* don't convert, do encryption */
4266 write_info.bw_flags = FIO_NOCONVERT | wb_flags;
4267 if (buf_write_bytes(&write_info) == FAIL)
4268 end = 0;
4269 else
4270 nchars += write_info.bw_len;
4273 write_info.bw_start_lnum = start;
4274 #endif
4276 write_info.bw_len = bufsize;
4277 #ifdef HAS_BW_FLAGS
4278 write_info.bw_flags = wb_flags;
4279 #endif
4280 fileformat = get_fileformat_force(buf, eap);
4281 s = buffer;
4282 len = 0;
4283 for (lnum = start; lnum <= end; ++lnum)
4286 * The next while loop is done once for each character written.
4287 * Keep it fast!
4289 ptr = ml_get_buf(buf, lnum, FALSE) - 1;
4290 while ((c = *++ptr) != NUL)
4292 if (c == NL)
4293 *s = NUL; /* replace newlines with NULs */
4294 else if (c == CAR && fileformat == EOL_MAC)
4295 *s = NL; /* Mac: replace CRs with NLs */
4296 else
4297 *s = c;
4298 ++s;
4299 if (++len != bufsize)
4300 continue;
4301 if (buf_write_bytes(&write_info) == FAIL)
4303 end = 0; /* write error: break loop */
4304 break;
4306 nchars += bufsize;
4307 s = buffer;
4308 len = 0;
4309 #ifdef FEAT_MBYTE
4310 write_info.bw_start_lnum = lnum;
4311 #endif
4313 /* write failed or last line has no EOL: stop here */
4314 if (end == 0
4315 || (lnum == end
4316 && write_bin
4317 && (lnum == write_no_eol_lnum
4318 || (lnum == buf->b_ml.ml_line_count && !buf->b_p_eol))))
4320 ++lnum; /* written the line, count it */
4321 no_eol = TRUE;
4322 break;
4324 if (fileformat == EOL_UNIX)
4325 *s++ = NL;
4326 else
4328 *s++ = CAR; /* EOL_MAC or EOL_DOS: write CR */
4329 if (fileformat == EOL_DOS) /* write CR-NL */
4331 if (++len == bufsize)
4333 if (buf_write_bytes(&write_info) == FAIL)
4335 end = 0; /* write error: break loop */
4336 break;
4338 nchars += bufsize;
4339 s = buffer;
4340 len = 0;
4342 *s++ = NL;
4345 if (++len == bufsize && end)
4347 if (buf_write_bytes(&write_info) == FAIL)
4349 end = 0; /* write error: break loop */
4350 break;
4352 nchars += bufsize;
4353 s = buffer;
4354 len = 0;
4356 ui_breakcheck();
4357 if (got_int)
4359 end = 0; /* Interrupted, break loop */
4360 break;
4363 #ifdef VMS
4365 * On VMS there is a problem: newlines get added when writing blocks
4366 * at a time. Fix it by writing a line at a time.
4367 * This is much slower!
4368 * Explanation: VAX/DECC RTL insists that records in some RMS
4369 * structures end with a newline (carriage return) character, and if
4370 * they don't it adds one.
4371 * With other RMS structures it works perfect without this fix.
4373 if (buf->b_fab_rfm == FAB$C_VFC
4374 || ((buf->b_fab_rat & (FAB$M_FTN | FAB$M_CR)) != 0))
4376 int b2write;
4378 buf->b_fab_mrs = (buf->b_fab_mrs == 0
4379 ? MIN(4096, bufsize)
4380 : MIN(buf->b_fab_mrs, bufsize));
4382 b2write = len;
4383 while (b2write > 0)
4385 write_info.bw_len = MIN(b2write, buf->b_fab_mrs);
4386 if (buf_write_bytes(&write_info) == FAIL)
4388 end = 0;
4389 break;
4391 b2write -= MIN(b2write, buf->b_fab_mrs);
4393 write_info.bw_len = bufsize;
4394 nchars += len;
4395 s = buffer;
4396 len = 0;
4398 #endif
4400 if (len > 0 && end > 0)
4402 write_info.bw_len = len;
4403 if (buf_write_bytes(&write_info) == FAIL)
4404 end = 0; /* write error */
4405 nchars += len;
4408 #if defined(UNIX) && defined(HAVE_FSYNC)
4409 /* On many journalling file systems there is a bug that causes both the
4410 * original and the backup file to be lost when halting the system right
4411 * after writing the file. That's because only the meta-data is
4412 * journalled. Syncing the file slows down the system, but assures it has
4413 * been written to disk and we don't lose it.
4414 * For a device do try the fsync() but don't complain if it does not work
4415 * (could be a pipe).
4416 * If the 'fsync' option is FALSE, don't fsync(). Useful for laptops. */
4417 if (p_fs && fsync(fd) != 0 && !device)
4419 errmsg = (char_u *)_("E667: Fsync failed");
4420 end = 0;
4422 #endif
4424 #ifdef HAVE_SELINUX
4425 /* Probably need to set the security context. */
4426 if (!backup_copy)
4427 mch_copy_sec(backup, wfname);
4428 #endif
4430 #ifdef UNIX
4431 /* When creating a new file, set its owner/group to that of the original
4432 * file. Get the new device and inode number. */
4433 if (backup != NULL && !backup_copy)
4435 # ifdef HAVE_FCHOWN
4436 struct stat st;
4438 /* don't change the owner when it's already OK, some systems remove
4439 * permission or ACL stuff */
4440 if (mch_stat((char *)wfname, &st) < 0
4441 || st.st_uid != st_old.st_uid
4442 || st.st_gid != st_old.st_gid)
4444 ignored = fchown(fd, st_old.st_uid, st_old.st_gid);
4445 if (perm >= 0) /* set permission again, may have changed */
4446 (void)mch_setperm(wfname, perm);
4448 # endif
4449 buf_setino(buf);
4451 else if (!buf->b_dev_valid)
4452 /* Set the inode when creating a new file. */
4453 buf_setino(buf);
4454 #endif
4456 if (close(fd) != 0)
4458 errmsg = (char_u *)_("E512: Close failed");
4459 end = 0;
4462 #ifdef UNIX
4463 if (made_writable)
4464 perm &= ~0200; /* reset 'w' bit for security reasons */
4465 #endif
4466 if (perm >= 0) /* set perm. of new file same as old file */
4467 (void)mch_setperm(wfname, perm);
4468 #ifdef RISCOS
4469 if (!append && !filtering)
4470 /* Set the filetype after writing the file. */
4471 mch_set_filetype(wfname, buf->b_p_oft);
4472 #endif
4473 #ifdef HAVE_ACL
4474 /* Probably need to set the ACL before changing the user (can't set the
4475 * ACL on a file the user doesn't own). */
4476 if (!backup_copy)
4477 mch_set_acl(wfname, acl);
4478 #endif
4481 #if defined(FEAT_MBYTE) && defined(FEAT_EVAL)
4482 if (wfname != fname)
4485 * The file was written to a temp file, now it needs to be converted
4486 * with 'charconvert' to (overwrite) the output file.
4488 if (end != 0)
4490 if (eval_charconvert(enc_utf8 ? (char_u *)"utf-8" : p_enc, fenc,
4491 wfname, fname) == FAIL)
4493 write_info.bw_conv_error = TRUE;
4494 end = 0;
4497 mch_remove(wfname);
4498 vim_free(wfname);
4500 #endif
4502 if (end == 0)
4504 if (errmsg == NULL)
4506 #ifdef FEAT_MBYTE
4507 if (write_info.bw_conv_error)
4509 if (write_info.bw_conv_error_lnum == 0)
4510 errmsg = (char_u *)_("E513: write error, conversion failed (make 'fenc' empty to override)");
4511 else
4513 errmsg_allocated = TRUE;
4514 errmsg = alloc(300);
4515 vim_snprintf((char *)errmsg, 300, _("E513: write error, conversion failed in line %ld (make 'fenc' empty to override)"),
4516 (long)write_info.bw_conv_error_lnum);
4519 else
4520 #endif
4521 if (got_int)
4522 errmsg = (char_u *)_(e_interr);
4523 else
4524 errmsg = (char_u *)_("E514: write error (file system full?)");
4528 * If we have a backup file, try to put it in place of the new file,
4529 * because the new file is probably corrupt. This avoids losing the
4530 * original file when trying to make a backup when writing the file a
4531 * second time.
4532 * When "backup_copy" is set we need to copy the backup over the new
4533 * file. Otherwise rename the backup file.
4534 * If this is OK, don't give the extra warning message.
4536 if (backup != NULL)
4538 if (backup_copy)
4540 /* This may take a while, if we were interrupted let the user
4541 * know we got the message. */
4542 if (got_int)
4544 MSG(_(e_interr));
4545 out_flush();
4547 if ((fd = mch_open((char *)backup, O_RDONLY | O_EXTRA, 0)) >= 0)
4549 if ((write_info.bw_fd = mch_open((char *)fname,
4550 O_WRONLY | O_CREAT | O_TRUNC | O_EXTRA,
4551 perm & 0777)) >= 0)
4553 /* copy the file. */
4554 write_info.bw_buf = smallbuf;
4555 #ifdef HAS_BW_FLAGS
4556 write_info.bw_flags = FIO_NOCONVERT;
4557 #endif
4558 while ((write_info.bw_len = vim_read(fd, smallbuf,
4559 SMBUFSIZE)) > 0)
4560 if (buf_write_bytes(&write_info) == FAIL)
4561 break;
4563 if (close(write_info.bw_fd) >= 0
4564 && write_info.bw_len == 0)
4565 end = 1; /* success */
4567 close(fd); /* ignore errors for closing read file */
4570 else
4572 if (vim_rename(backup, fname) == 0)
4573 end = 1;
4576 goto fail;
4579 lnum -= start; /* compute number of written lines */
4580 --no_wait_return; /* may wait for return now */
4582 #if !(defined(UNIX) || defined(VMS))
4583 fname = sfname; /* use shortname now, for the messages */
4584 #endif
4585 if (!filtering)
4587 msg_add_fname(buf, fname); /* put fname in IObuff with quotes */
4588 c = FALSE;
4589 #ifdef FEAT_MBYTE
4590 if (write_info.bw_conv_error)
4592 STRCAT(IObuff, _(" CONVERSION ERROR"));
4593 c = TRUE;
4594 if (write_info.bw_conv_error_lnum != 0)
4596 size_t l = STRLEN(IObuff);
4597 vim_snprintf((char *)IObuff + l, IOSIZE - l, _(" in line %ld;"),
4598 (long)write_info.bw_conv_error_lnum);
4601 else if (notconverted)
4603 STRCAT(IObuff, _("[NOT converted]"));
4604 c = TRUE;
4606 else if (converted)
4608 STRCAT(IObuff, _("[converted]"));
4609 c = TRUE;
4611 #endif
4612 if (device)
4614 STRCAT(IObuff, _("[Device]"));
4615 c = TRUE;
4617 else if (newfile)
4619 STRCAT(IObuff, shortmess(SHM_NEW) ? _("[New]") : _("[New File]"));
4620 c = TRUE;
4622 if (no_eol)
4624 msg_add_eol();
4625 c = TRUE;
4627 /* may add [unix/dos/mac] */
4628 if (msg_add_fileformat(fileformat))
4629 c = TRUE;
4630 #ifdef FEAT_CRYPT
4631 if (wb_flags & FIO_ENCRYPTED)
4633 STRCAT(IObuff, _("[crypted]"));
4634 c = TRUE;
4636 #endif
4637 msg_add_lines(c, (long)lnum, nchars); /* add line/char count */
4638 if (!shortmess(SHM_WRITE))
4640 if (append)
4641 STRCAT(IObuff, shortmess(SHM_WRI) ? _(" [a]") : _(" appended"));
4642 else
4643 STRCAT(IObuff, shortmess(SHM_WRI) ? _(" [w]") : _(" written"));
4646 set_keep_msg(msg_trunc_attr(IObuff, FALSE, 0), 0);
4649 /* When written everything correctly: reset 'modified'. Unless not
4650 * writing to the original file and '+' is not in 'cpoptions'. */
4651 if (reset_changed && whole && !append
4652 #ifdef FEAT_MBYTE
4653 && !write_info.bw_conv_error
4654 #endif
4655 && (overwriting || vim_strchr(p_cpo, CPO_PLUS) != NULL)
4658 unchanged(buf, TRUE);
4659 u_unchanged(buf);
4663 * If written to the current file, update the timestamp of the swap file
4664 * and reset the BF_WRITE_MASK flags. Also sets buf->b_mtime.
4666 if (overwriting)
4668 ml_timestamp(buf);
4669 if (append)
4670 buf->b_flags &= ~BF_NEW;
4671 else
4672 buf->b_flags &= ~BF_WRITE_MASK;
4676 * If we kept a backup until now, and we are in patch mode, then we make
4677 * the backup file our 'original' file.
4679 if (*p_pm && dobackup)
4681 char *org = (char *)buf_modname(
4682 #ifdef SHORT_FNAME
4683 TRUE,
4684 #else
4685 (buf->b_p_sn || buf->b_shortname),
4686 #endif
4687 fname, p_pm, FALSE);
4689 if (backup != NULL)
4691 struct stat st;
4694 * If the original file does not exist yet
4695 * the current backup file becomes the original file
4697 if (org == NULL)
4698 EMSG(_("E205: Patchmode: can't save original file"));
4699 else if (mch_stat(org, &st) < 0)
4701 vim_rename(backup, (char_u *)org);
4702 vim_free(backup); /* don't delete the file */
4703 backup = NULL;
4704 #ifdef UNIX
4705 set_file_time((char_u *)org, st_old.st_atime, st_old.st_mtime);
4706 #endif
4710 * If there is no backup file, remember that a (new) file was
4711 * created.
4713 else
4715 int empty_fd;
4717 if (org == NULL
4718 || (empty_fd = mch_open(org,
4719 O_CREAT | O_EXTRA | O_EXCL | O_NOFOLLOW,
4720 perm < 0 ? 0666 : (perm & 0777))) < 0)
4721 EMSG(_("E206: patchmode: can't touch empty original file"));
4722 else
4723 close(empty_fd);
4725 if (org != NULL)
4727 mch_setperm((char_u *)org, mch_getperm(fname) & 0777);
4728 vim_free(org);
4733 * Remove the backup unless 'backup' option is set
4735 if (!p_bk && backup != NULL && mch_remove(backup) != 0)
4736 EMSG(_("E207: Can't delete backup file"));
4738 #ifdef FEAT_SUN_WORKSHOP
4739 if (usingSunWorkShop)
4740 workshop_file_saved((char *) ffname);
4741 #endif
4743 goto nofail;
4746 * Finish up. We get here either after failure or success.
4748 fail:
4749 --no_wait_return; /* may wait for return now */
4750 nofail:
4752 /* Done saving, we accept changed buffer warnings again */
4753 buf->b_saving = FALSE;
4755 vim_free(backup);
4756 if (buffer != smallbuf)
4757 vim_free(buffer);
4758 #ifdef FEAT_MBYTE
4759 vim_free(fenc_tofree);
4760 vim_free(write_info.bw_conv_buf);
4761 # ifdef USE_ICONV
4762 if (write_info.bw_iconv_fd != (iconv_t)-1)
4764 iconv_close(write_info.bw_iconv_fd);
4765 write_info.bw_iconv_fd = (iconv_t)-1;
4767 # endif
4768 #endif
4769 #ifdef HAVE_ACL
4770 mch_free_acl(acl);
4771 #endif
4773 if (errmsg != NULL)
4775 int numlen = errnum != NULL ? (int)STRLEN(errnum) : 0;
4777 attr = hl_attr(HLF_E); /* set highlight for error messages */
4778 msg_add_fname(buf,
4779 #ifndef UNIX
4780 sfname
4781 #else
4782 fname
4783 #endif
4784 ); /* put file name in IObuff with quotes */
4785 if (STRLEN(IObuff) + STRLEN(errmsg) + numlen >= IOSIZE)
4786 IObuff[IOSIZE - STRLEN(errmsg) - numlen - 1] = NUL;
4787 /* If the error message has the form "is ...", put the error number in
4788 * front of the file name. */
4789 if (errnum != NULL)
4791 STRMOVE(IObuff + numlen, IObuff);
4792 mch_memmove(IObuff, errnum, (size_t)numlen);
4794 STRCAT(IObuff, errmsg);
4795 emsg(IObuff);
4796 if (errmsg_allocated)
4797 vim_free(errmsg);
4799 retval = FAIL;
4800 if (end == 0)
4802 MSG_PUTS_ATTR(_("\nWARNING: Original file may be lost or damaged\n"),
4803 attr | MSG_HIST);
4804 MSG_PUTS_ATTR(_("don't quit the editor until the file is successfully written!"),
4805 attr | MSG_HIST);
4807 /* Update the timestamp to avoid an "overwrite changed file"
4808 * prompt when writing again. */
4809 if (mch_stat((char *)fname, &st_old) >= 0)
4811 buf_store_time(buf, &st_old, fname);
4812 buf->b_mtime_read = buf->b_mtime;
4816 msg_scroll = msg_save;
4818 #ifdef FEAT_AUTOCMD
4819 #ifdef FEAT_EVAL
4820 if (!should_abort(retval))
4821 #else
4822 if (!got_int)
4823 #endif
4825 aco_save_T aco;
4827 write_no_eol_lnum = 0; /* in case it was set by the previous read */
4830 * Apply POST autocommands.
4831 * Careful: The autocommands may call buf_write() recursively!
4833 aucmd_prepbuf(&aco, buf);
4835 if (append)
4836 apply_autocmds_exarg(EVENT_FILEAPPENDPOST, fname, fname,
4837 FALSE, curbuf, eap);
4838 else if (filtering)
4839 apply_autocmds_exarg(EVENT_FILTERWRITEPOST, NULL, fname,
4840 FALSE, curbuf, eap);
4841 else if (reset_changed && whole)
4842 apply_autocmds_exarg(EVENT_BUFWRITEPOST, fname, fname,
4843 FALSE, curbuf, eap);
4844 else
4845 apply_autocmds_exarg(EVENT_FILEWRITEPOST, fname, fname,
4846 FALSE, curbuf, eap);
4848 /* restore curwin/curbuf and a few other things */
4849 aucmd_restbuf(&aco);
4851 #ifdef FEAT_EVAL
4852 if (aborting()) /* autocmds may abort script processing */
4853 retval = FALSE;
4854 #endif
4856 #endif
4858 got_int |= prev_got_int;
4860 #ifdef MACOS_CLASSIC /* TODO: Is it need for MACOS_X? (Dany) */
4861 /* Update machine specific information. */
4862 mch_post_buffer_write(buf);
4863 #endif
4864 return retval;
4868 * Set the name of the current buffer. Use when the buffer doesn't have a
4869 * name and a ":r" or ":w" command with a file name is used.
4871 static int
4872 set_rw_fname(fname, sfname)
4873 char_u *fname;
4874 char_u *sfname;
4876 #ifdef FEAT_AUTOCMD
4877 buf_T *buf = curbuf;
4879 /* It's like the unnamed buffer is deleted.... */
4880 if (curbuf->b_p_bl)
4881 apply_autocmds(EVENT_BUFDELETE, NULL, NULL, FALSE, curbuf);
4882 apply_autocmds(EVENT_BUFWIPEOUT, NULL, NULL, FALSE, curbuf);
4883 # ifdef FEAT_EVAL
4884 if (aborting()) /* autocmds may abort script processing */
4885 return FAIL;
4886 # endif
4887 if (curbuf != buf)
4889 /* We are in another buffer now, don't do the renaming. */
4890 EMSG(_(e_auchangedbuf));
4891 return FAIL;
4893 #endif
4895 if (setfname(curbuf, fname, sfname, FALSE) == OK)
4896 curbuf->b_flags |= BF_NOTEDITED;
4898 #ifdef FEAT_AUTOCMD
4899 /* ....and a new named one is created */
4900 apply_autocmds(EVENT_BUFNEW, NULL, NULL, FALSE, curbuf);
4901 if (curbuf->b_p_bl)
4902 apply_autocmds(EVENT_BUFADD, NULL, NULL, FALSE, curbuf);
4903 # ifdef FEAT_EVAL
4904 if (aborting()) /* autocmds may abort script processing */
4905 return FAIL;
4906 # endif
4908 /* Do filetype detection now if 'filetype' is empty. */
4909 if (*curbuf->b_p_ft == NUL)
4911 if (au_has_group((char_u *)"filetypedetect"))
4912 (void)do_doautocmd((char_u *)"filetypedetect BufRead", FALSE);
4913 do_modelines(0);
4915 #endif
4917 return OK;
4921 * Put file name into IObuff with quotes.
4923 void
4924 msg_add_fname(buf, fname)
4925 buf_T *buf;
4926 char_u *fname;
4928 if (fname == NULL)
4929 fname = (char_u *)"-stdin-";
4930 home_replace(buf, fname, IObuff + 1, IOSIZE - 4, TRUE);
4931 IObuff[0] = '"';
4932 STRCAT(IObuff, "\" ");
4936 * Append message for text mode to IObuff.
4937 * Return TRUE if something appended.
4939 static int
4940 msg_add_fileformat(eol_type)
4941 int eol_type;
4943 #ifndef USE_CRNL
4944 if (eol_type == EOL_DOS)
4946 STRCAT(IObuff, shortmess(SHM_TEXT) ? _("[dos]") : _("[dos format]"));
4947 return TRUE;
4949 #endif
4950 #ifndef USE_CR
4951 if (eol_type == EOL_MAC)
4953 STRCAT(IObuff, shortmess(SHM_TEXT) ? _("[mac]") : _("[mac format]"));
4954 return TRUE;
4956 #endif
4957 #if defined(USE_CRNL) || defined(USE_CR)
4958 if (eol_type == EOL_UNIX)
4960 STRCAT(IObuff, shortmess(SHM_TEXT) ? _("[unix]") : _("[unix format]"));
4961 return TRUE;
4963 #endif
4964 return FALSE;
4968 * Append line and character count to IObuff.
4970 void
4971 msg_add_lines(insert_space, lnum, nchars)
4972 int insert_space;
4973 long lnum;
4974 long nchars;
4976 char_u *p;
4978 p = IObuff + STRLEN(IObuff);
4980 if (insert_space)
4981 *p++ = ' ';
4982 if (shortmess(SHM_LINES))
4983 sprintf((char *)p, "%ldL, %ldC", lnum, nchars);
4984 else
4986 if (lnum == 1)
4987 STRCPY(p, _("1 line, "));
4988 else
4989 sprintf((char *)p, _("%ld lines, "), lnum);
4990 p += STRLEN(p);
4991 if (nchars == 1)
4992 STRCPY(p, _("1 character"));
4993 else
4994 sprintf((char *)p, _("%ld characters"), nchars);
4999 * Append message for missing line separator to IObuff.
5001 static void
5002 msg_add_eol()
5004 STRCAT(IObuff, shortmess(SHM_LAST) ? _("[noeol]") : _("[Incomplete last line]"));
5008 * Check modification time of file, before writing to it.
5009 * The size isn't checked, because using a tool like "gzip" takes care of
5010 * using the same timestamp but can't set the size.
5012 static int
5013 check_mtime(buf, st)
5014 buf_T *buf;
5015 struct stat *st;
5017 if (buf->b_mtime_read != 0
5018 && time_differs((long)st->st_mtime, buf->b_mtime_read))
5020 msg_scroll = TRUE; /* don't overwrite messages here */
5021 msg_silent = 0; /* must give this prompt */
5022 /* don't use emsg() here, don't want to flush the buffers */
5023 MSG_ATTR(_("WARNING: The file has been changed since reading it!!!"),
5024 hl_attr(HLF_E));
5025 if (ask_yesno((char_u *)_("Do you really want to write to it"),
5026 TRUE) == 'n')
5027 return FAIL;
5028 msg_scroll = FALSE; /* always overwrite the file message now */
5030 return OK;
5033 static int
5034 time_differs(t1, t2)
5035 long t1, t2;
5037 #if defined(__linux__) || defined(MSDOS) || defined(MSWIN)
5038 /* On a FAT filesystem, esp. under Linux, there are only 5 bits to store
5039 * the seconds. Since the roundoff is done when flushing the inode, the
5040 * time may change unexpectedly by one second!!! */
5041 return (t1 - t2 > 1 || t2 - t1 > 1);
5042 #else
5043 return (t1 != t2);
5044 #endif
5048 * Call write() to write a number of bytes to the file.
5049 * Also handles encryption and 'encoding' conversion.
5051 * Return FAIL for failure, OK otherwise.
5053 static int
5054 buf_write_bytes(ip)
5055 struct bw_info *ip;
5057 int wlen;
5058 char_u *buf = ip->bw_buf; /* data to write */
5059 int len = ip->bw_len; /* length of data */
5060 #ifdef HAS_BW_FLAGS
5061 int flags = ip->bw_flags; /* extra flags */
5062 #endif
5064 #ifdef FEAT_MBYTE
5066 * Skip conversion when writing the crypt magic number or the BOM.
5068 if (!(flags & FIO_NOCONVERT))
5070 char_u *p;
5071 unsigned c;
5072 int n;
5074 if (flags & FIO_UTF8)
5077 * Convert latin1 in the buffer to UTF-8 in the file.
5079 p = ip->bw_conv_buf; /* translate to buffer */
5080 for (wlen = 0; wlen < len; ++wlen)
5081 p += utf_char2bytes(buf[wlen], p);
5082 buf = ip->bw_conv_buf;
5083 len = (int)(p - ip->bw_conv_buf);
5085 else if (flags & (FIO_UCS4 | FIO_UTF16 | FIO_UCS2 | FIO_LATIN1))
5088 * Convert UTF-8 bytes in the buffer to UCS-2, UCS-4, UTF-16 or
5089 * Latin1 chars in the file.
5091 if (flags & FIO_LATIN1)
5092 p = buf; /* translate in-place (can only get shorter) */
5093 else
5094 p = ip->bw_conv_buf; /* translate to buffer */
5095 for (wlen = 0; wlen < len; wlen += n)
5097 if (wlen == 0 && ip->bw_restlen != 0)
5099 int l;
5101 /* Use remainder of previous call. Append the start of
5102 * buf[] to get a full sequence. Might still be too
5103 * short! */
5104 l = CONV_RESTLEN - ip->bw_restlen;
5105 if (l > len)
5106 l = len;
5107 mch_memmove(ip->bw_rest + ip->bw_restlen, buf, (size_t)l);
5108 n = utf_ptr2len_len(ip->bw_rest, ip->bw_restlen + l);
5109 if (n > ip->bw_restlen + len)
5111 /* We have an incomplete byte sequence at the end to
5112 * be written. We can't convert it without the
5113 * remaining bytes. Keep them for the next call. */
5114 if (ip->bw_restlen + len > CONV_RESTLEN)
5115 return FAIL;
5116 ip->bw_restlen += len;
5117 break;
5119 if (n > 1)
5120 c = utf_ptr2char(ip->bw_rest);
5121 else
5122 c = ip->bw_rest[0];
5123 if (n >= ip->bw_restlen)
5125 n -= ip->bw_restlen;
5126 ip->bw_restlen = 0;
5128 else
5130 ip->bw_restlen -= n;
5131 mch_memmove(ip->bw_rest, ip->bw_rest + n,
5132 (size_t)ip->bw_restlen);
5133 n = 0;
5136 else
5138 n = utf_ptr2len_len(buf + wlen, len - wlen);
5139 if (n > len - wlen)
5141 /* We have an incomplete byte sequence at the end to
5142 * be written. We can't convert it without the
5143 * remaining bytes. Keep them for the next call. */
5144 if (len - wlen > CONV_RESTLEN)
5145 return FAIL;
5146 ip->bw_restlen = len - wlen;
5147 mch_memmove(ip->bw_rest, buf + wlen,
5148 (size_t)ip->bw_restlen);
5149 break;
5151 if (n > 1)
5152 c = utf_ptr2char(buf + wlen);
5153 else
5154 c = buf[wlen];
5157 if (ucs2bytes(c, &p, flags) && !ip->bw_conv_error)
5159 ip->bw_conv_error = TRUE;
5160 ip->bw_conv_error_lnum = ip->bw_start_lnum;
5162 if (c == NL)
5163 ++ip->bw_start_lnum;
5165 if (flags & FIO_LATIN1)
5166 len = (int)(p - buf);
5167 else
5169 buf = ip->bw_conv_buf;
5170 len = (int)(p - ip->bw_conv_buf);
5174 # ifdef WIN3264
5175 else if (flags & FIO_CODEPAGE)
5178 * Convert UTF-8 or codepage to UCS-2 and then to MS-Windows
5179 * codepage.
5181 char_u *from;
5182 size_t fromlen;
5183 char_u *to;
5184 int u8c;
5185 BOOL bad = FALSE;
5186 int needed;
5188 if (ip->bw_restlen > 0)
5190 /* Need to concatenate the remainder of the previous call and
5191 * the bytes of the current call. Use the end of the
5192 * conversion buffer for this. */
5193 fromlen = len + ip->bw_restlen;
5194 from = ip->bw_conv_buf + ip->bw_conv_buflen - fromlen;
5195 mch_memmove(from, ip->bw_rest, (size_t)ip->bw_restlen);
5196 mch_memmove(from + ip->bw_restlen, buf, (size_t)len);
5198 else
5200 from = buf;
5201 fromlen = len;
5204 to = ip->bw_conv_buf;
5205 if (enc_utf8)
5207 /* Convert from UTF-8 to UCS-2, to the start of the buffer.
5208 * The buffer has been allocated to be big enough. */
5209 while (fromlen > 0)
5211 n = (int)utf_ptr2len_len(from, (int)fromlen);
5212 if (n > (int)fromlen) /* incomplete byte sequence */
5213 break;
5214 u8c = utf_ptr2char(from);
5215 *to++ = (u8c & 0xff);
5216 *to++ = (u8c >> 8);
5217 fromlen -= n;
5218 from += n;
5221 /* Copy remainder to ip->bw_rest[] to be used for the next
5222 * call. */
5223 if (fromlen > CONV_RESTLEN)
5225 /* weird overlong sequence */
5226 ip->bw_conv_error = TRUE;
5227 return FAIL;
5229 mch_memmove(ip->bw_rest, from, fromlen);
5230 ip->bw_restlen = (int)fromlen;
5232 else
5234 /* Convert from enc_codepage to UCS-2, to the start of the
5235 * buffer. The buffer has been allocated to be big enough. */
5236 ip->bw_restlen = 0;
5237 needed = MultiByteToWideChar(enc_codepage,
5238 MB_ERR_INVALID_CHARS, (LPCSTR)from, (int)fromlen,
5239 NULL, 0);
5240 if (needed == 0)
5242 /* When conversion fails there may be a trailing byte. */
5243 needed = MultiByteToWideChar(enc_codepage,
5244 MB_ERR_INVALID_CHARS, (LPCSTR)from, (int)fromlen - 1,
5245 NULL, 0);
5246 if (needed == 0)
5248 /* Conversion doesn't work. */
5249 ip->bw_conv_error = TRUE;
5250 return FAIL;
5252 /* Save the trailing byte for the next call. */
5253 ip->bw_rest[0] = from[fromlen - 1];
5254 ip->bw_restlen = 1;
5256 needed = MultiByteToWideChar(enc_codepage, MB_ERR_INVALID_CHARS,
5257 (LPCSTR)from, (int)(fromlen - ip->bw_restlen),
5258 (LPWSTR)to, needed);
5259 if (needed == 0)
5261 /* Safety check: Conversion doesn't work. */
5262 ip->bw_conv_error = TRUE;
5263 return FAIL;
5265 to += needed * 2;
5268 fromlen = to - ip->bw_conv_buf;
5269 buf = to;
5270 # ifdef CP_UTF8 /* VC 4.1 doesn't define CP_UTF8 */
5271 if (FIO_GET_CP(flags) == CP_UTF8)
5273 /* Convert from UCS-2 to UTF-8, using the remainder of the
5274 * conversion buffer. Fails when out of space. */
5275 for (from = ip->bw_conv_buf; fromlen > 1; fromlen -= 2)
5277 u8c = *from++;
5278 u8c += (*from++ << 8);
5279 to += utf_char2bytes(u8c, to);
5280 if (to + 6 >= ip->bw_conv_buf + ip->bw_conv_buflen)
5282 ip->bw_conv_error = TRUE;
5283 return FAIL;
5286 len = (int)(to - buf);
5288 else
5289 #endif
5291 /* Convert from UCS-2 to the codepage, using the remainder of
5292 * the conversion buffer. If the conversion uses the default
5293 * character "0", the data doesn't fit in this encoding, so
5294 * fail. */
5295 len = WideCharToMultiByte(FIO_GET_CP(flags), 0,
5296 (LPCWSTR)ip->bw_conv_buf, (int)fromlen / sizeof(WCHAR),
5297 (LPSTR)to, (int)(ip->bw_conv_buflen - fromlen), 0,
5298 &bad);
5299 if (bad)
5301 ip->bw_conv_error = TRUE;
5302 return FAIL;
5306 # endif
5308 # ifdef MACOS_CONVERT
5309 else if (flags & FIO_MACROMAN)
5312 * Convert UTF-8 or latin1 to Apple MacRoman.
5314 char_u *from;
5315 size_t fromlen;
5317 if (ip->bw_restlen > 0)
5319 /* Need to concatenate the remainder of the previous call and
5320 * the bytes of the current call. Use the end of the
5321 * conversion buffer for this. */
5322 fromlen = len + ip->bw_restlen;
5323 from = ip->bw_conv_buf + ip->bw_conv_buflen - fromlen;
5324 mch_memmove(from, ip->bw_rest, (size_t)ip->bw_restlen);
5325 mch_memmove(from + ip->bw_restlen, buf, (size_t)len);
5327 else
5329 from = buf;
5330 fromlen = len;
5333 if (enc2macroman(from, fromlen,
5334 ip->bw_conv_buf, &len, ip->bw_conv_buflen,
5335 ip->bw_rest, &ip->bw_restlen) == FAIL)
5337 ip->bw_conv_error = TRUE;
5338 return FAIL;
5340 buf = ip->bw_conv_buf;
5342 # endif
5344 # ifdef USE_ICONV
5345 if (ip->bw_iconv_fd != (iconv_t)-1)
5347 const char *from;
5348 size_t fromlen;
5349 char *to;
5350 size_t tolen;
5352 /* Convert with iconv(). */
5353 if (ip->bw_restlen > 0)
5355 char *fp;
5357 /* Need to concatenate the remainder of the previous call and
5358 * the bytes of the current call. Use the end of the
5359 * conversion buffer for this. */
5360 fromlen = len + ip->bw_restlen;
5361 fp = (char *)ip->bw_conv_buf + ip->bw_conv_buflen - fromlen;
5362 mch_memmove(fp, ip->bw_rest, (size_t)ip->bw_restlen);
5363 mch_memmove(fp + ip->bw_restlen, buf, (size_t)len);
5364 from = fp;
5365 tolen = ip->bw_conv_buflen - fromlen;
5367 else
5369 from = (const char *)buf;
5370 fromlen = len;
5371 tolen = ip->bw_conv_buflen;
5373 to = (char *)ip->bw_conv_buf;
5375 if (ip->bw_first)
5377 size_t save_len = tolen;
5379 /* output the initial shift state sequence */
5380 (void)iconv(ip->bw_iconv_fd, NULL, NULL, &to, &tolen);
5382 /* There is a bug in iconv() on Linux (which appears to be
5383 * wide-spread) which sets "to" to NULL and messes up "tolen".
5385 if (to == NULL)
5387 to = (char *)ip->bw_conv_buf;
5388 tolen = save_len;
5390 ip->bw_first = FALSE;
5394 * If iconv() has an error or there is not enough room, fail.
5396 if ((iconv(ip->bw_iconv_fd, (void *)&from, &fromlen, &to, &tolen)
5397 == (size_t)-1 && ICONV_ERRNO != ICONV_EINVAL)
5398 || fromlen > CONV_RESTLEN)
5400 ip->bw_conv_error = TRUE;
5401 return FAIL;
5404 /* copy remainder to ip->bw_rest[] to be used for the next call. */
5405 if (fromlen > 0)
5406 mch_memmove(ip->bw_rest, (void *)from, fromlen);
5407 ip->bw_restlen = (int)fromlen;
5409 buf = ip->bw_conv_buf;
5410 len = (int)((char_u *)to - ip->bw_conv_buf);
5412 # endif
5414 #endif /* FEAT_MBYTE */
5416 #ifdef FEAT_CRYPT
5417 if (flags & FIO_ENCRYPTED) /* encrypt the data */
5419 int ztemp, t, i;
5421 for (i = 0; i < len; i++)
5423 ztemp = buf[i];
5424 buf[i] = ZENCODE(ztemp, t);
5427 #endif
5429 /* Repeat the write(), it may be interrupted by a signal. */
5430 while (len > 0)
5432 wlen = vim_write(ip->bw_fd, buf, len);
5433 if (wlen <= 0) /* error! */
5434 return FAIL;
5435 len -= wlen;
5436 buf += wlen;
5438 return OK;
5441 #ifdef FEAT_MBYTE
5443 * Convert a Unicode character to bytes.
5444 * Return TRUE for an error, FALSE when it's OK.
5446 static int
5447 ucs2bytes(c, pp, flags)
5448 unsigned c; /* in: character */
5449 char_u **pp; /* in/out: pointer to result */
5450 int flags; /* FIO_ flags */
5452 char_u *p = *pp;
5453 int error = FALSE;
5454 int cc;
5457 if (flags & FIO_UCS4)
5459 if (flags & FIO_ENDIAN_L)
5461 *p++ = c;
5462 *p++ = (c >> 8);
5463 *p++ = (c >> 16);
5464 *p++ = (c >> 24);
5466 else
5468 *p++ = (c >> 24);
5469 *p++ = (c >> 16);
5470 *p++ = (c >> 8);
5471 *p++ = c;
5474 else if (flags & (FIO_UCS2 | FIO_UTF16))
5476 if (c >= 0x10000)
5478 if (flags & FIO_UTF16)
5480 /* Make two words, ten bits of the character in each. First
5481 * word is 0xd800 - 0xdbff, second one 0xdc00 - 0xdfff */
5482 c -= 0x10000;
5483 if (c >= 0x100000)
5484 error = TRUE;
5485 cc = ((c >> 10) & 0x3ff) + 0xd800;
5486 if (flags & FIO_ENDIAN_L)
5488 *p++ = cc;
5489 *p++ = ((unsigned)cc >> 8);
5491 else
5493 *p++ = ((unsigned)cc >> 8);
5494 *p++ = cc;
5496 c = (c & 0x3ff) + 0xdc00;
5498 else
5499 error = TRUE;
5501 if (flags & FIO_ENDIAN_L)
5503 *p++ = c;
5504 *p++ = (c >> 8);
5506 else
5508 *p++ = (c >> 8);
5509 *p++ = c;
5512 else /* Latin1 */
5514 if (c >= 0x100)
5516 error = TRUE;
5517 *p++ = 0xBF;
5519 else
5520 *p++ = c;
5523 *pp = p;
5524 return error;
5528 * Return TRUE if file encoding "fenc" requires conversion from or to
5529 * 'encoding'.
5531 static int
5532 need_conversion(fenc)
5533 char_u *fenc;
5535 int same_encoding;
5536 int enc_flags;
5537 int fenc_flags;
5539 if (*fenc == NUL || STRCMP(p_enc, fenc) == 0)
5540 same_encoding = TRUE;
5541 else
5543 /* Ignore difference between "ansi" and "latin1", "ucs-4" and
5544 * "ucs-4be", etc. */
5545 enc_flags = get_fio_flags(p_enc);
5546 fenc_flags = get_fio_flags(fenc);
5547 same_encoding = (enc_flags != 0 && fenc_flags == enc_flags);
5549 if (same_encoding)
5551 /* Specified encoding matches with 'encoding'. This requires
5552 * conversion when 'encoding' is Unicode but not UTF-8. */
5553 return enc_unicode != 0;
5556 /* Encodings differ. However, conversion is not needed when 'enc' is any
5557 * Unicode encoding and the file is UTF-8. */
5558 return !(enc_utf8 && fenc_flags == FIO_UTF8);
5562 * Check "ptr" for a unicode encoding and return the FIO_ flags needed for the
5563 * internal conversion.
5564 * if "ptr" is an empty string, use 'encoding'.
5566 static int
5567 get_fio_flags(ptr)
5568 char_u *ptr;
5570 int prop;
5572 if (*ptr == NUL)
5573 ptr = p_enc;
5575 prop = enc_canon_props(ptr);
5576 if (prop & ENC_UNICODE)
5578 if (prop & ENC_2BYTE)
5580 if (prop & ENC_ENDIAN_L)
5581 return FIO_UCS2 | FIO_ENDIAN_L;
5582 return FIO_UCS2;
5584 if (prop & ENC_4BYTE)
5586 if (prop & ENC_ENDIAN_L)
5587 return FIO_UCS4 | FIO_ENDIAN_L;
5588 return FIO_UCS4;
5590 if (prop & ENC_2WORD)
5592 if (prop & ENC_ENDIAN_L)
5593 return FIO_UTF16 | FIO_ENDIAN_L;
5594 return FIO_UTF16;
5596 return FIO_UTF8;
5598 if (prop & ENC_LATIN1)
5599 return FIO_LATIN1;
5600 /* must be ENC_DBCS, requires iconv() */
5601 return 0;
5604 #ifdef WIN3264
5606 * Check "ptr" for a MS-Windows codepage name and return the FIO_ flags needed
5607 * for the conversion MS-Windows can do for us. Also accept "utf-8".
5608 * Used for conversion between 'encoding' and 'fileencoding'.
5610 static int
5611 get_win_fio_flags(ptr)
5612 char_u *ptr;
5614 int cp;
5616 /* Cannot do this when 'encoding' is not utf-8 and not a codepage. */
5617 if (!enc_utf8 && enc_codepage <= 0)
5618 return 0;
5620 cp = encname2codepage(ptr);
5621 if (cp == 0)
5623 # ifdef CP_UTF8 /* VC 4.1 doesn't define CP_UTF8 */
5624 if (STRCMP(ptr, "utf-8") == 0)
5625 cp = CP_UTF8;
5626 else
5627 # endif
5628 return 0;
5630 return FIO_PUT_CP(cp) | FIO_CODEPAGE;
5632 #endif
5634 #ifdef MACOS_X
5636 * Check "ptr" for a Carbon supported encoding and return the FIO_ flags
5637 * needed for the internal conversion to/from utf-8 or latin1.
5639 static int
5640 get_mac_fio_flags(ptr)
5641 char_u *ptr;
5643 if ((enc_utf8 || STRCMP(p_enc, "latin1") == 0)
5644 && (enc_canon_props(ptr) & ENC_MACROMAN))
5645 return FIO_MACROMAN;
5646 return 0;
5648 #endif
5651 * Check for a Unicode BOM (Byte Order Mark) at the start of p[size].
5652 * "size" must be at least 2.
5653 * Return the name of the encoding and set "*lenp" to the length.
5654 * Returns NULL when no BOM found.
5656 static char_u *
5657 check_for_bom(p, size, lenp, flags)
5658 char_u *p;
5659 long size;
5660 int *lenp;
5661 int flags;
5663 char *name = NULL;
5664 int len = 2;
5666 if (p[0] == 0xef && p[1] == 0xbb && size >= 3 && p[2] == 0xbf
5667 && (flags == FIO_ALL || flags == FIO_UTF8 || flags == 0))
5669 name = "utf-8"; /* EF BB BF */
5670 len = 3;
5672 else if (p[0] == 0xff && p[1] == 0xfe)
5674 if (size >= 4 && p[2] == 0 && p[3] == 0
5675 && (flags == FIO_ALL || flags == (FIO_UCS4 | FIO_ENDIAN_L)))
5677 name = "ucs-4le"; /* FF FE 00 00 */
5678 len = 4;
5680 else if (flags == (FIO_UCS2 | FIO_ENDIAN_L))
5681 name = "ucs-2le"; /* FF FE */
5682 else if (flags == FIO_ALL || flags == (FIO_UTF16 | FIO_ENDIAN_L))
5683 /* utf-16le is preferred, it also works for ucs-2le text */
5684 name = "utf-16le"; /* FF FE */
5686 else if (p[0] == 0xfe && p[1] == 0xff
5687 && (flags == FIO_ALL || flags == FIO_UCS2 || flags == FIO_UTF16))
5689 /* Default to utf-16, it works also for ucs-2 text. */
5690 if (flags == FIO_UCS2)
5691 name = "ucs-2"; /* FE FF */
5692 else
5693 name = "utf-16"; /* FE FF */
5695 else if (size >= 4 && p[0] == 0 && p[1] == 0 && p[2] == 0xfe
5696 && p[3] == 0xff && (flags == FIO_ALL || flags == FIO_UCS4))
5698 name = "ucs-4"; /* 00 00 FE FF */
5699 len = 4;
5702 *lenp = len;
5703 return (char_u *)name;
5707 * Generate a BOM in "buf[4]" for encoding "name".
5708 * Return the length of the BOM (zero when no BOM).
5710 static int
5711 make_bom(buf, name)
5712 char_u *buf;
5713 char_u *name;
5715 int flags;
5716 char_u *p;
5718 flags = get_fio_flags(name);
5720 /* Can't put a BOM in a non-Unicode file. */
5721 if (flags == FIO_LATIN1 || flags == 0)
5722 return 0;
5724 if (flags == FIO_UTF8) /* UTF-8 */
5726 buf[0] = 0xef;
5727 buf[1] = 0xbb;
5728 buf[2] = 0xbf;
5729 return 3;
5731 p = buf;
5732 (void)ucs2bytes(0xfeff, &p, flags);
5733 return (int)(p - buf);
5735 #endif
5737 #if defined(FEAT_VIMINFO) || defined(FEAT_BROWSE) || \
5738 defined(FEAT_QUICKFIX) || defined(FEAT_AUTOCMD) || defined(PROTO)
5740 * Try to find a shortname by comparing the fullname with the current
5741 * directory.
5742 * Returns "full_path" or pointer into "full_path" if shortened.
5744 char_u *
5745 shorten_fname1(full_path)
5746 char_u *full_path;
5748 char_u dirname[MAXPATHL];
5749 char_u *p = full_path;
5751 if (mch_dirname(dirname, MAXPATHL) == OK)
5753 p = shorten_fname(full_path, dirname);
5754 if (p == NULL || *p == NUL)
5755 p = full_path;
5757 return p;
5759 #endif
5762 * Try to find a shortname by comparing the fullname with the current
5763 * directory.
5764 * Returns NULL if not shorter name possible, pointer into "full_path"
5765 * otherwise.
5767 char_u *
5768 shorten_fname(full_path, dir_name)
5769 char_u *full_path;
5770 char_u *dir_name;
5772 int len;
5773 char_u *p;
5775 if (full_path == NULL)
5776 return NULL;
5777 len = (int)STRLEN(dir_name);
5778 if (fnamencmp(dir_name, full_path, len) == 0)
5780 p = full_path + len;
5781 #if defined(MSDOS) || defined(MSWIN) || defined(OS2)
5783 * MSDOS: when a file is in the root directory, dir_name will end in a
5784 * slash, since C: by itself does not define a specific dir. In this
5785 * case p may already be correct. <negri>
5787 if (!((len > 2) && (*(p - 2) == ':')))
5788 #endif
5790 if (vim_ispathsep(*p))
5791 ++p;
5792 #ifndef VMS /* the path separator is always part of the path */
5793 else
5794 p = NULL;
5795 #endif
5798 #if defined(MSDOS) || defined(MSWIN) || defined(OS2)
5800 * When using a file in the current drive, remove the drive name:
5801 * "A:\dir\file" -> "\dir\file". This helps when moving a session file on
5802 * a floppy from "A:\dir" to "B:\dir".
5804 else if (len > 3
5805 && TOUPPER_LOC(full_path[0]) == TOUPPER_LOC(dir_name[0])
5806 && full_path[1] == ':'
5807 && vim_ispathsep(full_path[2]))
5808 p = full_path + 2;
5809 #endif
5810 else
5811 p = NULL;
5812 return p;
5816 * Shorten filenames for all buffers.
5817 * When "force" is TRUE: Use full path from now on for files currently being
5818 * edited, both for file name and swap file name. Try to shorten the file
5819 * names a bit, if safe to do so.
5820 * When "force" is FALSE: Only try to shorten absolute file names.
5821 * For buffers that have buftype "nofile" or "scratch": never change the file
5822 * name.
5824 void
5825 shorten_fnames(force)
5826 int force;
5828 char_u dirname[MAXPATHL];
5829 buf_T *buf;
5830 char_u *p;
5832 mch_dirname(dirname, MAXPATHL);
5833 for (buf = firstbuf; buf != NULL; buf = buf->b_next)
5835 if (buf->b_fname != NULL
5836 #ifdef FEAT_QUICKFIX
5837 && !bt_nofile(buf)
5838 #endif
5839 && !path_with_url(buf->b_fname)
5840 && (force
5841 || buf->b_sfname == NULL
5842 || mch_isFullName(buf->b_sfname)))
5844 vim_free(buf->b_sfname);
5845 buf->b_sfname = NULL;
5846 p = shorten_fname(buf->b_ffname, dirname);
5847 if (p != NULL)
5849 buf->b_sfname = vim_strsave(p);
5850 buf->b_fname = buf->b_sfname;
5852 if (p == NULL || buf->b_fname == NULL)
5853 buf->b_fname = buf->b_ffname;
5856 /* Always make the swap file name a full path, a "nofile" buffer may
5857 * also have a swap file. */
5858 mf_fullname(buf->b_ml.ml_mfp);
5860 #ifdef FEAT_WINDOWS
5861 status_redraw_all();
5862 redraw_tabline = TRUE;
5863 #endif
5866 #if (defined(FEAT_DND) && defined(FEAT_GUI_GTK)) \
5867 || defined(FEAT_GUI_MSWIN) \
5868 || defined(FEAT_GUI_MAC) \
5869 || defined(PROTO)
5871 * Shorten all filenames in "fnames[count]" by current directory.
5873 void
5874 shorten_filenames(fnames, count)
5875 char_u **fnames;
5876 int count;
5878 int i;
5879 char_u dirname[MAXPATHL];
5880 char_u *p;
5882 if (fnames == NULL || count < 1)
5883 return;
5884 mch_dirname(dirname, sizeof(dirname));
5885 for (i = 0; i < count; ++i)
5887 if ((p = shorten_fname(fnames[i], dirname)) != NULL)
5889 /* shorten_fname() returns pointer in given "fnames[i]". If free
5890 * "fnames[i]" first, "p" becomes invalid. So we need to copy
5891 * "p" first then free fnames[i]. */
5892 p = vim_strsave(p);
5893 vim_free(fnames[i]);
5894 fnames[i] = p;
5898 #endif
5901 * add extension to file name - change path/fo.o.h to path/fo.o.h.ext or
5902 * fo_o_h.ext for MSDOS or when shortname option set.
5904 * Assumed that fname is a valid name found in the filesystem we assure that
5905 * the return value is a different name and ends in 'ext'.
5906 * "ext" MUST be at most 4 characters long if it starts with a dot, 3
5907 * characters otherwise.
5908 * Space for the returned name is allocated, must be freed later.
5909 * Returns NULL when out of memory.
5911 char_u *
5912 modname(fname, ext, prepend_dot)
5913 char_u *fname, *ext;
5914 int prepend_dot; /* may prepend a '.' to file name */
5916 return buf_modname(
5917 #ifdef SHORT_FNAME
5918 TRUE,
5919 #else
5920 (curbuf->b_p_sn || curbuf->b_shortname),
5921 #endif
5922 fname, ext, prepend_dot);
5925 char_u *
5926 buf_modname(shortname, fname, ext, prepend_dot)
5927 int shortname; /* use 8.3 file name */
5928 char_u *fname, *ext;
5929 int prepend_dot; /* may prepend a '.' to file name */
5931 char_u *retval;
5932 char_u *s;
5933 char_u *e;
5934 char_u *ptr;
5935 int fnamelen, extlen;
5937 extlen = (int)STRLEN(ext);
5940 * If there is no file name we must get the name of the current directory
5941 * (we need the full path in case :cd is used).
5943 if (fname == NULL || *fname == NUL)
5945 retval = alloc((unsigned)(MAXPATHL + extlen + 3));
5946 if (retval == NULL)
5947 return NULL;
5948 if (mch_dirname(retval, MAXPATHL) == FAIL ||
5949 (fnamelen = (int)STRLEN(retval)) == 0)
5951 vim_free(retval);
5952 return NULL;
5954 if (!after_pathsep(retval, retval + fnamelen))
5956 retval[fnamelen++] = PATHSEP;
5957 retval[fnamelen] = NUL;
5959 #ifndef SHORT_FNAME
5960 prepend_dot = FALSE; /* nothing to prepend a dot to */
5961 #endif
5963 else
5965 fnamelen = (int)STRLEN(fname);
5966 retval = alloc((unsigned)(fnamelen + extlen + 3));
5967 if (retval == NULL)
5968 return NULL;
5969 STRCPY(retval, fname);
5970 #ifdef VMS
5971 vms_remove_version(retval); /* we do not need versions here */
5972 #endif
5976 * search backwards until we hit a '/', '\' or ':' replacing all '.'
5977 * by '_' for MSDOS or when shortname option set and ext starts with a dot.
5978 * Then truncate what is after the '/', '\' or ':' to 8 characters for
5979 * MSDOS and 26 characters for AMIGA, a lot more for UNIX.
5981 for (ptr = retval + fnamelen; ptr > retval; mb_ptr_back(retval, ptr))
5983 #ifndef RISCOS
5984 if (*ext == '.'
5985 # ifdef USE_LONG_FNAME
5986 && (!USE_LONG_FNAME || shortname)
5987 # else
5988 # ifndef SHORT_FNAME
5989 && shortname
5990 # endif
5991 # endif
5993 if (*ptr == '.') /* replace '.' by '_' */
5994 *ptr = '_';
5995 #endif
5996 if (vim_ispathsep(*ptr))
5998 ++ptr;
5999 break;
6003 /* the file name has at most BASENAMELEN characters. */
6004 #ifndef SHORT_FNAME
6005 if (STRLEN(ptr) > (unsigned)BASENAMELEN)
6006 ptr[BASENAMELEN] = '\0';
6007 #endif
6009 s = ptr + STRLEN(ptr);
6012 * For 8.3 file names we may have to reduce the length.
6014 #ifdef USE_LONG_FNAME
6015 if (!USE_LONG_FNAME || shortname)
6016 #else
6017 # ifndef SHORT_FNAME
6018 if (shortname)
6019 # endif
6020 #endif
6023 * If there is no file name, or the file name ends in '/', and the
6024 * extension starts with '.', put a '_' before the dot, because just
6025 * ".ext" is invalid.
6027 if (fname == NULL || *fname == NUL
6028 || vim_ispathsep(fname[STRLEN(fname) - 1]))
6030 #ifdef RISCOS
6031 if (*ext == '/')
6032 #else
6033 if (*ext == '.')
6034 #endif
6035 *s++ = '_';
6038 * If the extension starts with '.', truncate the base name at 8
6039 * characters
6041 #ifdef RISCOS
6042 /* We normally use '/', but swap files are '_' */
6043 else if (*ext == '/' || *ext == '_')
6044 #else
6045 else if (*ext == '.')
6046 #endif
6048 if ((size_t)(s - ptr) > (size_t)8)
6050 s = ptr + 8;
6051 *s = '\0';
6055 * If the extension doesn't start with '.', and the file name
6056 * doesn't have an extension yet, append a '.'
6058 #ifdef RISCOS
6059 else if ((e = vim_strchr(ptr, '/')) == NULL)
6060 *s++ = '/';
6061 #else
6062 else if ((e = vim_strchr(ptr, '.')) == NULL)
6063 *s++ = '.';
6064 #endif
6066 * If the extension doesn't start with '.', and there already is an
6067 * extension, it may need to be truncated
6069 else if ((int)STRLEN(e) + extlen > 4)
6070 s = e + 4 - extlen;
6072 #if defined(OS2) || defined(USE_LONG_FNAME) || defined(WIN3264)
6074 * If there is no file name, and the extension starts with '.', put a
6075 * '_' before the dot, because just ".ext" may be invalid if it's on a
6076 * FAT partition, and on HPFS it doesn't matter.
6078 else if ((fname == NULL || *fname == NUL) && *ext == '.')
6079 *s++ = '_';
6080 #endif
6083 * Append the extension.
6084 * ext can start with '.' and cannot exceed 3 more characters.
6086 STRCPY(s, ext);
6088 #ifndef SHORT_FNAME
6090 * Prepend the dot.
6092 if (prepend_dot && !shortname && *(e = gettail(retval)) !=
6093 #ifdef RISCOS
6095 #else
6097 #endif
6098 #ifdef USE_LONG_FNAME
6099 && USE_LONG_FNAME
6100 #endif
6103 STRMOVE(e + 1, e);
6104 #ifdef RISCOS
6105 *e = '/';
6106 #else
6107 *e = '.';
6108 #endif
6110 #endif
6113 * Check that, after appending the extension, the file name is really
6114 * different.
6116 if (fname != NULL && STRCMP(fname, retval) == 0)
6118 /* we search for a character that can be replaced by '_' */
6119 while (--s >= ptr)
6121 if (*s != '_')
6123 *s = '_';
6124 break;
6127 if (s < ptr) /* fname was "________.<ext>", how tricky! */
6128 *ptr = 'v';
6130 return retval;
6134 * Like fgets(), but if the file line is too long, it is truncated and the
6135 * rest of the line is thrown away. Returns TRUE for end-of-file.
6138 vim_fgets(buf, size, fp)
6139 char_u *buf;
6140 int size;
6141 FILE *fp;
6143 char *eof;
6144 #define FGETS_SIZE 200
6145 char tbuf[FGETS_SIZE];
6147 buf[size - 2] = NUL;
6148 #ifdef USE_CR
6149 eof = fgets_cr((char *)buf, size, fp);
6150 #else
6151 eof = fgets((char *)buf, size, fp);
6152 #endif
6153 if (buf[size - 2] != NUL && buf[size - 2] != '\n')
6155 buf[size - 1] = NUL; /* Truncate the line */
6157 /* Now throw away the rest of the line: */
6160 tbuf[FGETS_SIZE - 2] = NUL;
6161 #ifdef USE_CR
6162 ignoredp = fgets_cr((char *)tbuf, FGETS_SIZE, fp);
6163 #else
6164 ignoredp = fgets((char *)tbuf, FGETS_SIZE, fp);
6165 #endif
6166 } while (tbuf[FGETS_SIZE - 2] != NUL && tbuf[FGETS_SIZE - 2] != '\n');
6168 return (eof == NULL);
6171 #if defined(USE_CR) || defined(PROTO)
6173 * Like vim_fgets(), but accept any line terminator: CR, CR-LF or LF.
6174 * Returns TRUE for end-of-file.
6175 * Only used for the Mac, because it's much slower than vim_fgets().
6178 tag_fgets(buf, size, fp)
6179 char_u *buf;
6180 int size;
6181 FILE *fp;
6183 int i = 0;
6184 int c;
6185 int eof = FALSE;
6187 for (;;)
6189 c = fgetc(fp);
6190 if (c == EOF)
6192 eof = TRUE;
6193 break;
6195 if (c == '\r')
6197 /* Always store a NL for end-of-line. */
6198 if (i < size - 1)
6199 buf[i++] = '\n';
6200 c = fgetc(fp);
6201 if (c != '\n') /* Macintosh format: single CR. */
6202 ungetc(c, fp);
6203 break;
6205 if (i < size - 1)
6206 buf[i++] = c;
6207 if (c == '\n')
6208 break;
6210 buf[i] = NUL;
6211 return eof;
6213 #endif
6216 * rename() only works if both files are on the same file system, this
6217 * function will (attempts to?) copy the file across if rename fails -- webb
6218 * Return -1 for failure, 0 for success.
6221 vim_rename(from, to)
6222 char_u *from;
6223 char_u *to;
6225 int fd_in;
6226 int fd_out;
6227 int n;
6228 char *errmsg = NULL;
6229 char *buffer;
6230 #ifdef AMIGA
6231 BPTR flock;
6232 #endif
6233 struct stat st;
6234 long perm;
6235 #ifdef HAVE_ACL
6236 vim_acl_T acl; /* ACL from original file */
6237 #endif
6238 #if defined(UNIX) || defined(CASE_INSENSITIVE_FILENAME)
6239 int use_tmp_file = FALSE;
6240 #endif
6243 * When the names are identical, there is nothing to do. When they refer
6244 * to the same file (ignoring case and slash/backslash differences) but
6245 * the file name differs we need to go through a temp file.
6247 if (fnamecmp(from, to) == 0)
6249 #ifdef CASE_INSENSITIVE_FILENAME
6250 if (STRCMP(gettail(from), gettail(to)) != 0)
6251 use_tmp_file = TRUE;
6252 else
6253 #endif
6254 return 0;
6258 * Fail if the "from" file doesn't exist. Avoids that "to" is deleted.
6260 if (mch_stat((char *)from, &st) < 0)
6261 return -1;
6263 #ifdef UNIX
6265 struct stat st_to;
6267 /* It's possible for the source and destination to be the same file.
6268 * This happens when "from" and "to" differ in case and are on a FAT32
6269 * filesystem. In that case go through a temp file name. */
6270 if (mch_stat((char *)to, &st_to) >= 0
6271 && st.st_dev == st_to.st_dev
6272 && st.st_ino == st_to.st_ino)
6273 use_tmp_file = TRUE;
6275 #endif
6277 #if defined(UNIX) || defined(CASE_INSENSITIVE_FILENAME)
6278 if (use_tmp_file)
6280 char tempname[MAXPATHL + 1];
6283 * Find a name that doesn't exist and is in the same directory.
6284 * Rename "from" to "tempname" and then rename "tempname" to "to".
6286 if (STRLEN(from) >= MAXPATHL - 5)
6287 return -1;
6288 STRCPY(tempname, from);
6289 for (n = 123; n < 99999; ++n)
6291 sprintf((char *)gettail((char_u *)tempname), "%d", n);
6292 if (mch_stat(tempname, &st) < 0)
6294 if (mch_rename((char *)from, tempname) == 0)
6296 if (mch_rename(tempname, (char *)to) == 0)
6297 return 0;
6298 /* Strange, the second step failed. Try moving the
6299 * file back and return failure. */
6300 mch_rename(tempname, (char *)from);
6301 return -1;
6303 /* If it fails for one temp name it will most likely fail
6304 * for any temp name, give up. */
6305 return -1;
6308 return -1;
6310 #endif
6313 * Delete the "to" file, this is required on some systems to make the
6314 * mch_rename() work, on other systems it makes sure that we don't have
6315 * two files when the mch_rename() fails.
6318 #ifdef AMIGA
6320 * With MSDOS-compatible filesystems (crossdos, messydos) it is possible
6321 * that the name of the "to" file is the same as the "from" file, even
6322 * though the names are different. To avoid the chance of accidentally
6323 * deleting the "from" file (horror!) we lock it during the remove.
6325 * When used for making a backup before writing the file: This should not
6326 * happen with ":w", because startscript() should detect this problem and
6327 * set buf->b_shortname, causing modname() to return a correct ".bak" file
6328 * name. This problem does exist with ":w filename", but then the
6329 * original file will be somewhere else so the backup isn't really
6330 * important. If autoscripting is off the rename may fail.
6332 flock = Lock((UBYTE *)from, (long)ACCESS_READ);
6333 #endif
6334 mch_remove(to);
6335 #ifdef AMIGA
6336 if (flock)
6337 UnLock(flock);
6338 #endif
6341 * First try a normal rename, return if it works.
6343 if (mch_rename((char *)from, (char *)to) == 0)
6344 return 0;
6347 * Rename() failed, try copying the file.
6349 perm = mch_getperm(from);
6350 #ifdef HAVE_ACL
6351 /* For systems that support ACL: get the ACL from the original file. */
6352 acl = mch_get_acl(from);
6353 #endif
6354 fd_in = mch_open((char *)from, O_RDONLY|O_EXTRA, 0);
6355 if (fd_in == -1)
6357 #ifdef HAVE_ACL
6358 mch_free_acl(acl);
6359 #endif
6360 return -1;
6363 /* Create the new file with same permissions as the original. */
6364 fd_out = mch_open((char *)to,
6365 O_CREAT|O_EXCL|O_WRONLY|O_EXTRA|O_NOFOLLOW, (int)perm);
6366 if (fd_out == -1)
6368 close(fd_in);
6369 #ifdef HAVE_ACL
6370 mch_free_acl(acl);
6371 #endif
6372 return -1;
6375 buffer = (char *)alloc(BUFSIZE);
6376 if (buffer == NULL)
6378 close(fd_out);
6379 close(fd_in);
6380 #ifdef HAVE_ACL
6381 mch_free_acl(acl);
6382 #endif
6383 return -1;
6386 while ((n = vim_read(fd_in, buffer, BUFSIZE)) > 0)
6387 if (vim_write(fd_out, buffer, n) != n)
6389 errmsg = _("E208: Error writing to \"%s\"");
6390 break;
6393 vim_free(buffer);
6394 close(fd_in);
6395 if (close(fd_out) < 0)
6396 errmsg = _("E209: Error closing \"%s\"");
6397 if (n < 0)
6399 errmsg = _("E210: Error reading \"%s\"");
6400 to = from;
6402 #ifndef UNIX /* for Unix mch_open() already set the permission */
6403 mch_setperm(to, perm);
6404 #endif
6405 #ifdef HAVE_ACL
6406 mch_set_acl(to, acl);
6407 mch_free_acl(acl);
6408 #endif
6409 if (errmsg != NULL)
6411 EMSG2(errmsg, to);
6412 return -1;
6414 mch_remove(from);
6415 return 0;
6418 static int already_warned = FALSE;
6421 * Check if any not hidden buffer has been changed.
6422 * Postpone the check if there are characters in the stuff buffer, a global
6423 * command is being executed, a mapping is being executed or an autocommand is
6424 * busy.
6425 * Returns TRUE if some message was written (screen should be redrawn and
6426 * cursor positioned).
6429 check_timestamps(focus)
6430 int focus; /* called for GUI focus event */
6432 buf_T *buf;
6433 int didit = 0;
6434 int n;
6436 /* Don't check timestamps while system() or another low-level function may
6437 * cause us to lose and gain focus. */
6438 if (no_check_timestamps > 0)
6439 return FALSE;
6441 /* Avoid doing a check twice. The OK/Reload dialog can cause a focus
6442 * event and we would keep on checking if the file is steadily growing.
6443 * Do check again after typing something. */
6444 if (focus && did_check_timestamps)
6446 need_check_timestamps = TRUE;
6447 return FALSE;
6450 if (!stuff_empty() || global_busy || !typebuf_typed()
6451 #ifdef FEAT_AUTOCMD
6452 || autocmd_busy || curbuf_lock > 0 || allbuf_lock > 0
6453 #endif
6455 need_check_timestamps = TRUE; /* check later */
6456 else
6458 ++no_wait_return;
6459 did_check_timestamps = TRUE;
6460 already_warned = FALSE;
6461 for (buf = firstbuf; buf != NULL; )
6463 /* Only check buffers in a window. */
6464 if (buf->b_nwindows > 0)
6466 n = buf_check_timestamp(buf, focus);
6467 if (didit < n)
6468 didit = n;
6469 if (n > 0 && !buf_valid(buf))
6471 /* Autocommands have removed the buffer, start at the
6472 * first one again. */
6473 buf = firstbuf;
6474 continue;
6477 buf = buf->b_next;
6479 --no_wait_return;
6480 need_check_timestamps = FALSE;
6481 if (need_wait_return && didit == 2)
6483 /* make sure msg isn't overwritten */
6484 msg_puts((char_u *)"\n");
6485 out_flush();
6488 return didit;
6492 * Move all the lines from buffer "frombuf" to buffer "tobuf".
6493 * Return OK or FAIL. When FAIL "tobuf" is incomplete and/or "frombuf" is not
6494 * empty.
6496 static int
6497 move_lines(frombuf, tobuf)
6498 buf_T *frombuf;
6499 buf_T *tobuf;
6501 buf_T *tbuf = curbuf;
6502 int retval = OK;
6503 linenr_T lnum;
6504 char_u *p;
6506 /* Copy the lines in "frombuf" to "tobuf". */
6507 curbuf = tobuf;
6508 for (lnum = 1; lnum <= frombuf->b_ml.ml_line_count; ++lnum)
6510 p = vim_strsave(ml_get_buf(frombuf, lnum, FALSE));
6511 if (p == NULL || ml_append(lnum - 1, p, 0, FALSE) == FAIL)
6513 vim_free(p);
6514 retval = FAIL;
6515 break;
6517 vim_free(p);
6520 /* Delete all the lines in "frombuf". */
6521 if (retval != FAIL)
6523 curbuf = frombuf;
6524 for (lnum = curbuf->b_ml.ml_line_count; lnum > 0; --lnum)
6525 if (ml_delete(lnum, FALSE) == FAIL)
6527 /* Oops! We could try putting back the saved lines, but that
6528 * might fail again... */
6529 retval = FAIL;
6530 break;
6534 curbuf = tbuf;
6535 return retval;
6539 * Check if buffer "buf" has been changed.
6540 * Also check if the file for a new buffer unexpectedly appeared.
6541 * return 1 if a changed buffer was found.
6542 * return 2 if a message has been displayed.
6543 * return 0 otherwise.
6546 buf_check_timestamp(buf, focus)
6547 buf_T *buf;
6548 int focus UNUSED; /* called for GUI focus event */
6550 struct stat st;
6551 int stat_res;
6552 int retval = 0;
6553 char_u *path;
6554 char_u *tbuf;
6555 char *mesg = NULL;
6556 char *mesg2 = "";
6557 int helpmesg = FALSE;
6558 int reload = FALSE;
6559 #if defined(FEAT_CON_DIALOG) || defined(FEAT_GUI_DIALOG)
6560 int can_reload = FALSE;
6561 #endif
6562 size_t orig_size = buf->b_orig_size;
6563 int orig_mode = buf->b_orig_mode;
6564 #ifdef FEAT_GUI
6565 int save_mouse_correct = need_mouse_correct;
6566 #endif
6567 #ifdef FEAT_AUTOCMD
6568 static int busy = FALSE;
6569 int n;
6570 char_u *s;
6571 #endif
6572 char *reason;
6574 /* If there is no file name, the buffer is not loaded, 'buftype' is
6575 * set, we are in the middle of a save or being called recursively: ignore
6576 * this buffer. */
6577 if (buf->b_ffname == NULL
6578 || buf->b_ml.ml_mfp == NULL
6579 #if defined(FEAT_QUICKFIX)
6580 || *buf->b_p_bt != NUL
6581 #endif
6582 || buf->b_saving
6583 #ifdef FEAT_AUTOCMD
6584 || busy
6585 #endif
6586 #ifdef FEAT_NETBEANS_INTG
6587 || isNetbeansBuffer(buf)
6588 #endif
6590 return 0;
6592 if ( !(buf->b_flags & BF_NOTEDITED)
6593 && buf->b_mtime != 0
6594 && ((stat_res = mch_stat((char *)buf->b_ffname, &st)) < 0
6595 || time_differs((long)st.st_mtime, buf->b_mtime)
6596 #ifdef HAVE_ST_MODE
6597 || (int)st.st_mode != buf->b_orig_mode
6598 #else
6599 || mch_getperm(buf->b_ffname) != buf->b_orig_mode
6600 #endif
6603 retval = 1;
6605 /* set b_mtime to stop further warnings (e.g., when executing
6606 * FileChangedShell autocmd) */
6607 if (stat_res < 0)
6609 buf->b_mtime = 0;
6610 buf->b_orig_size = 0;
6611 buf->b_orig_mode = 0;
6613 else
6614 buf_store_time(buf, &st, buf->b_ffname);
6616 /* Don't do anything for a directory. Might contain the file
6617 * explorer. */
6618 if (mch_isdir(buf->b_fname))
6622 * If 'autoread' is set, the buffer has no changes and the file still
6623 * exists, reload the buffer. Use the buffer-local option value if it
6624 * was set, the global option value otherwise.
6626 else if ((buf->b_p_ar >= 0 ? buf->b_p_ar : p_ar)
6627 && !bufIsChanged(buf) && stat_res >= 0)
6628 reload = TRUE;
6629 else
6631 if (stat_res < 0)
6632 reason = "deleted";
6633 else if (bufIsChanged(buf))
6634 reason = "conflict";
6635 else if (orig_size != buf->b_orig_size || buf_contents_changed(buf))
6636 reason = "changed";
6637 else if (orig_mode != buf->b_orig_mode)
6638 reason = "mode";
6639 else
6640 reason = "time";
6642 #ifdef FEAT_AUTOCMD
6644 * Only give the warning if there are no FileChangedShell
6645 * autocommands.
6646 * Avoid being called recursively by setting "busy".
6648 busy = TRUE;
6649 # ifdef FEAT_EVAL
6650 set_vim_var_string(VV_FCS_REASON, (char_u *)reason, -1);
6651 set_vim_var_string(VV_FCS_CHOICE, (char_u *)"", -1);
6652 # endif
6653 ++allbuf_lock;
6654 n = apply_autocmds(EVENT_FILECHANGEDSHELL,
6655 buf->b_fname, buf->b_fname, FALSE, buf);
6656 --allbuf_lock;
6657 busy = FALSE;
6658 if (n)
6660 if (!buf_valid(buf))
6661 EMSG(_("E246: FileChangedShell autocommand deleted buffer"));
6662 # ifdef FEAT_EVAL
6663 s = get_vim_var_str(VV_FCS_CHOICE);
6664 if (STRCMP(s, "reload") == 0 && *reason != 'd')
6665 reload = TRUE;
6666 else if (STRCMP(s, "ask") == 0)
6667 n = FALSE;
6668 else
6669 # endif
6670 return 2;
6672 if (!n)
6673 #endif
6675 if (*reason == 'd')
6676 mesg = _("E211: File \"%s\" no longer available");
6677 else
6679 helpmesg = TRUE;
6680 #if defined(FEAT_CON_DIALOG) || defined(FEAT_GUI_DIALOG)
6681 can_reload = TRUE;
6682 #endif
6684 * Check if the file contents really changed to avoid
6685 * giving a warning when only the timestamp was set (e.g.,
6686 * checked out of CVS). Always warn when the buffer was
6687 * changed.
6689 if (reason[2] == 'n')
6691 mesg = _("W12: Warning: File \"%s\" has changed and the buffer was changed in Vim as well");
6692 mesg2 = _("See \":help W12\" for more info.");
6694 else if (reason[1] == 'h')
6696 mesg = _("W11: Warning: File \"%s\" has changed since editing started");
6697 mesg2 = _("See \":help W11\" for more info.");
6699 else if (*reason == 'm')
6701 mesg = _("W16: Warning: Mode of file \"%s\" has changed since editing started");
6702 mesg2 = _("See \":help W16\" for more info.");
6704 else
6705 /* Only timestamp changed, store it to avoid a warning
6706 * in check_mtime() later. */
6707 buf->b_mtime_read = buf->b_mtime;
6713 else if ((buf->b_flags & BF_NEW) && !(buf->b_flags & BF_NEW_W)
6714 && vim_fexists(buf->b_ffname))
6716 retval = 1;
6717 mesg = _("W13: Warning: File \"%s\" has been created after editing started");
6718 buf->b_flags |= BF_NEW_W;
6719 #if defined(FEAT_CON_DIALOG) || defined(FEAT_GUI_DIALOG)
6720 can_reload = TRUE;
6721 #endif
6724 if (mesg != NULL)
6726 path = home_replace_save(buf, buf->b_fname);
6727 if (path != NULL)
6729 if (!helpmesg)
6730 mesg2 = "";
6731 tbuf = alloc((unsigned)(STRLEN(path) + STRLEN(mesg)
6732 + STRLEN(mesg2) + 2));
6733 sprintf((char *)tbuf, mesg, path);
6734 #ifdef FEAT_EVAL
6735 /* Set warningmsg here, before the unimportant and output-specific
6736 * mesg2 has been appended. */
6737 set_vim_var_string(VV_WARNINGMSG, tbuf, -1);
6738 #endif
6739 #if defined(FEAT_CON_DIALOG) || defined(FEAT_GUI_DIALOG)
6740 if (can_reload)
6742 if (*mesg2 != NUL)
6744 STRCAT(tbuf, "\n");
6745 STRCAT(tbuf, mesg2);
6747 if (do_dialog(VIM_WARNING, (char_u *)_("Warning"), tbuf,
6748 (char_u *)_("&OK\n&Load File"), 1, NULL) == 2)
6749 reload = TRUE;
6751 else
6752 #endif
6753 if (State > NORMAL_BUSY || (State & CMDLINE) || already_warned)
6755 if (*mesg2 != NUL)
6757 STRCAT(tbuf, "; ");
6758 STRCAT(tbuf, mesg2);
6760 EMSG(tbuf);
6761 retval = 2;
6763 else
6765 # ifdef FEAT_AUTOCMD
6766 if (!autocmd_busy)
6767 # endif
6769 msg_start();
6770 msg_puts_attr(tbuf, hl_attr(HLF_E) + MSG_HIST);
6771 if (*mesg2 != NUL)
6772 msg_puts_attr((char_u *)mesg2,
6773 hl_attr(HLF_W) + MSG_HIST);
6774 msg_clr_eos();
6775 (void)msg_end();
6776 if (emsg_silent == 0)
6778 out_flush();
6779 # ifdef FEAT_GUI
6780 if (!focus)
6781 # endif
6782 /* give the user some time to think about it */
6783 ui_delay(1000L, TRUE);
6785 /* don't redraw and erase the message */
6786 redraw_cmdline = FALSE;
6789 already_warned = TRUE;
6792 vim_free(path);
6793 vim_free(tbuf);
6797 if (reload)
6798 /* Reload the buffer. */
6799 buf_reload(buf, orig_mode);
6801 #ifdef FEAT_AUTOCMD
6802 /* Trigger FileChangedShell when the file was changed in any way. */
6803 if (buf_valid(buf) && retval != 0)
6804 (void)apply_autocmds(EVENT_FILECHANGEDSHELLPOST,
6805 buf->b_fname, buf->b_fname, FALSE, buf);
6806 #endif
6807 #ifdef FEAT_GUI
6808 /* restore this in case an autocommand has set it; it would break
6809 * 'mousefocus' */
6810 need_mouse_correct = save_mouse_correct;
6811 #endif
6813 return retval;
6817 * Reload a buffer that is already loaded.
6818 * Used when the file was changed outside of Vim.
6819 * "orig_mode" is buf->b_orig_mode before the need for reloading was detected.
6820 * buf->b_orig_mode may have been reset already.
6822 void
6823 buf_reload(buf, orig_mode)
6824 buf_T *buf;
6825 int orig_mode;
6827 exarg_T ea;
6828 pos_T old_cursor;
6829 linenr_T old_topline;
6830 int old_ro = buf->b_p_ro;
6831 buf_T *savebuf;
6832 int saved = OK;
6833 aco_save_T aco;
6835 /* set curwin/curbuf for "buf" and save some things */
6836 aucmd_prepbuf(&aco, buf);
6838 /* We only want to read the text from the file, not reset the syntax
6839 * highlighting, clear marks, diff status, etc. Force the fileformat
6840 * and encoding to be the same. */
6841 if (prep_exarg(&ea, buf) == OK)
6843 old_cursor = curwin->w_cursor;
6844 old_topline = curwin->w_topline;
6847 * To behave like when a new file is edited (matters for
6848 * BufReadPost autocommands) we first need to delete the current
6849 * buffer contents. But if reading the file fails we should keep
6850 * the old contents. Can't use memory only, the file might be
6851 * too big. Use a hidden buffer to move the buffer contents to.
6853 if (bufempty())
6854 savebuf = NULL;
6855 else
6857 /* Allocate a buffer without putting it in the buffer list. */
6858 savebuf = buflist_new(NULL, NULL, (linenr_T)1, BLN_DUMMY);
6859 if (savebuf != NULL && buf == curbuf)
6861 /* Open the memline. */
6862 curbuf = savebuf;
6863 curwin->w_buffer = savebuf;
6864 saved = ml_open(curbuf);
6865 curbuf = buf;
6866 curwin->w_buffer = buf;
6868 if (savebuf == NULL || saved == FAIL || buf != curbuf
6869 || move_lines(buf, savebuf) == FAIL)
6871 EMSG2(_("E462: Could not prepare for reloading \"%s\""),
6872 buf->b_fname);
6873 saved = FAIL;
6877 if (saved == OK)
6879 curbuf->b_flags |= BF_CHECK_RO; /* check for RO again */
6880 #ifdef FEAT_AUTOCMD
6881 keep_filetype = TRUE; /* don't detect 'filetype' */
6882 #endif
6883 if (readfile(buf->b_ffname, buf->b_fname, (linenr_T)0,
6884 (linenr_T)0,
6885 (linenr_T)MAXLNUM, &ea, READ_NEW) == FAIL)
6887 #if defined(FEAT_AUTOCMD) && defined(FEAT_EVAL)
6888 if (!aborting())
6889 #endif
6890 EMSG2(_("E321: Could not reload \"%s\""), buf->b_fname);
6891 if (savebuf != NULL && buf_valid(savebuf) && buf == curbuf)
6893 /* Put the text back from the save buffer. First
6894 * delete any lines that readfile() added. */
6895 while (!bufempty())
6896 if (ml_delete(buf->b_ml.ml_line_count, FALSE) == FAIL)
6897 break;
6898 (void)move_lines(savebuf, buf);
6901 else if (buf == curbuf)
6903 /* Mark the buffer as unmodified and free undo info. */
6904 unchanged(buf, TRUE);
6905 u_blockfree(buf);
6906 u_clearall(buf);
6909 vim_free(ea.cmd);
6911 if (savebuf != NULL && buf_valid(savebuf))
6912 wipe_buffer(savebuf, FALSE);
6914 #ifdef FEAT_DIFF
6915 /* Invalidate diff info if necessary. */
6916 diff_invalidate(curbuf);
6917 #endif
6919 /* Restore the topline and cursor position and check it (lines may
6920 * have been removed). */
6921 if (old_topline > curbuf->b_ml.ml_line_count)
6922 curwin->w_topline = curbuf->b_ml.ml_line_count;
6923 else
6924 curwin->w_topline = old_topline;
6925 curwin->w_cursor = old_cursor;
6926 check_cursor();
6927 update_topline();
6928 #ifdef FEAT_AUTOCMD
6929 keep_filetype = FALSE;
6930 #endif
6931 #ifdef FEAT_FOLDING
6933 win_T *wp;
6934 tabpage_T *tp;
6936 /* Update folds unless they are defined manually. */
6937 FOR_ALL_TAB_WINDOWS(tp, wp)
6938 if (wp->w_buffer == curwin->w_buffer
6939 && !foldmethodIsManual(wp))
6940 foldUpdateAll(wp);
6942 #endif
6943 /* If the mode didn't change and 'readonly' was set, keep the old
6944 * value; the user probably used the ":view" command. But don't
6945 * reset it, might have had a read error. */
6946 if (orig_mode == curbuf->b_orig_mode)
6947 curbuf->b_p_ro |= old_ro;
6950 /* restore curwin/curbuf and a few other things */
6951 aucmd_restbuf(&aco);
6952 /* Careful: autocommands may have made "buf" invalid! */
6955 void
6956 buf_store_time(buf, st, fname)
6957 buf_T *buf;
6958 struct stat *st;
6959 char_u *fname UNUSED;
6961 buf->b_mtime = (long)st->st_mtime;
6962 buf->b_orig_size = (size_t)st->st_size;
6963 #ifdef HAVE_ST_MODE
6964 buf->b_orig_mode = (int)st->st_mode;
6965 #else
6966 buf->b_orig_mode = mch_getperm(fname);
6967 #endif
6971 * Adjust the line with missing eol, used for the next write.
6972 * Used for do_filter(), when the input lines for the filter are deleted.
6974 void
6975 write_lnum_adjust(offset)
6976 linenr_T offset;
6978 if (write_no_eol_lnum != 0) /* only if there is a missing eol */
6979 write_no_eol_lnum += offset;
6982 #if defined(TEMPDIRNAMES) || defined(PROTO)
6983 static long temp_count = 0; /* Temp filename counter. */
6986 * Delete the temp directory and all files it contains.
6988 void
6989 vim_deltempdir()
6991 char_u **files;
6992 int file_count;
6993 int i;
6995 if (vim_tempdir != NULL)
6997 sprintf((char *)NameBuff, "%s*", vim_tempdir);
6998 if (gen_expand_wildcards(1, &NameBuff, &file_count, &files,
6999 EW_DIR|EW_FILE|EW_SILENT) == OK)
7001 for (i = 0; i < file_count; ++i)
7002 mch_remove(files[i]);
7003 FreeWild(file_count, files);
7005 gettail(NameBuff)[-1] = NUL;
7006 (void)mch_rmdir(NameBuff);
7008 vim_free(vim_tempdir);
7009 vim_tempdir = NULL;
7012 #endif
7014 #ifdef TEMPDIRNAMES
7016 * Directory "tempdir" was created. Expand this name to a full path and put
7017 * it in "vim_tempdir". This avoids that using ":cd" would confuse us.
7018 * "tempdir" must be no longer than MAXPATHL.
7020 static void
7021 vim_settempdir(tempdir)
7022 char_u *tempdir;
7024 char_u *buf;
7026 buf = alloc((unsigned)MAXPATHL + 2);
7027 if (buf != NULL)
7029 if (vim_FullName(tempdir, buf, MAXPATHL, FALSE) == FAIL)
7030 STRCPY(buf, tempdir);
7031 # ifdef __EMX__
7032 if (vim_strchr(buf, '/') != NULL)
7033 STRCAT(buf, "/");
7034 else
7035 # endif
7036 add_pathsep(buf);
7037 vim_tempdir = vim_strsave(buf);
7038 vim_free(buf);
7041 #endif
7044 * vim_tempname(): Return a unique name that can be used for a temp file.
7046 * The temp file is NOT created.
7048 * The returned pointer is to allocated memory.
7049 * The returned pointer is NULL if no valid name was found.
7051 char_u *
7052 vim_tempname(extra_char)
7053 int extra_char UNUSED; /* char to use in the name instead of '?' */
7055 #ifdef USE_TMPNAM
7056 char_u itmp[L_tmpnam]; /* use tmpnam() */
7057 #else
7058 char_u itmp[TEMPNAMELEN];
7059 #endif
7061 #ifdef TEMPDIRNAMES
7062 static char *(tempdirs[]) = {TEMPDIRNAMES};
7063 int i;
7064 # ifndef EEXIST
7065 struct stat st;
7066 # endif
7069 * This will create a directory for private use by this instance of Vim.
7070 * This is done once, and the same directory is used for all temp files.
7071 * This method avoids security problems because of symlink attacks et al.
7072 * It's also a bit faster, because we only need to check for an existing
7073 * file when creating the directory and not for each temp file.
7075 if (vim_tempdir == NULL)
7078 * Try the entries in TEMPDIRNAMES to create the temp directory.
7080 for (i = 0; i < (int)(sizeof(tempdirs) / sizeof(char *)); ++i)
7082 # ifndef HAVE_MKDTEMP
7083 size_t itmplen;
7084 long nr;
7085 long off;
7086 # endif
7088 /* expand $TMP, leave room for "/v1100000/999999999" */
7089 expand_env((char_u *)tempdirs[i], itmp, TEMPNAMELEN - 20);
7090 if (mch_isdir(itmp)) /* directory exists */
7092 # ifdef __EMX__
7093 /* If $TMP contains a forward slash (perhaps using bash or
7094 * tcsh), don't add a backslash, use a forward slash!
7095 * Adding 2 backslashes didn't work. */
7096 if (vim_strchr(itmp, '/') != NULL)
7097 STRCAT(itmp, "/");
7098 else
7099 # endif
7100 add_pathsep(itmp);
7102 # ifdef HAVE_MKDTEMP
7103 /* Leave room for filename */
7104 STRCAT(itmp, "vXXXXXX");
7105 if (mkdtemp((char *)itmp) != NULL)
7106 vim_settempdir(itmp);
7107 # else
7108 /* Get an arbitrary number of up to 6 digits. When it's
7109 * unlikely that it already exists it will be faster,
7110 * otherwise it doesn't matter. The use of mkdir() avoids any
7111 * security problems because of the predictable number. */
7112 nr = (mch_get_pid() + (long)time(NULL)) % 1000000L;
7113 itmplen = STRLEN(itmp);
7115 /* Try up to 10000 different values until we find a name that
7116 * doesn't exist. */
7117 for (off = 0; off < 10000L; ++off)
7119 int r;
7120 # if defined(UNIX) || defined(VMS)
7121 mode_t umask_save;
7122 # endif
7124 sprintf((char *)itmp + itmplen, "v%ld", nr + off);
7125 # ifndef EEXIST
7126 /* If mkdir() does not set errno to EEXIST, check for
7127 * existing file here. There is a race condition then,
7128 * although it's fail-safe. */
7129 if (mch_stat((char *)itmp, &st) >= 0)
7130 continue;
7131 # endif
7132 # if defined(UNIX) || defined(VMS)
7133 /* Make sure the umask doesn't remove the executable bit.
7134 * "repl" has been reported to use "177". */
7135 umask_save = umask(077);
7136 # endif
7137 r = vim_mkdir(itmp, 0700);
7138 # if defined(UNIX) || defined(VMS)
7139 (void)umask(umask_save);
7140 # endif
7141 if (r == 0)
7143 vim_settempdir(itmp);
7144 break;
7146 # ifdef EEXIST
7147 /* If the mkdir() didn't fail because the file/dir exists,
7148 * we probably can't create any dir here, try another
7149 * place. */
7150 if (errno != EEXIST)
7151 # endif
7152 break;
7154 # endif /* HAVE_MKDTEMP */
7155 if (vim_tempdir != NULL)
7156 break;
7161 if (vim_tempdir != NULL)
7163 /* There is no need to check if the file exists, because we own the
7164 * directory and nobody else creates a file in it. */
7165 sprintf((char *)itmp, "%s%ld", vim_tempdir, temp_count++);
7166 return vim_strsave(itmp);
7169 return NULL;
7171 #else /* TEMPDIRNAMES */
7173 # ifdef WIN3264
7174 char szTempFile[_MAX_PATH + 1];
7175 char buf4[4];
7176 char_u *retval;
7177 char_u *p;
7179 STRCPY(itmp, "");
7180 if (GetTempPath(_MAX_PATH, szTempFile) == 0)
7181 szTempFile[0] = NUL; /* GetTempPath() failed, use current dir */
7182 strcpy(buf4, "VIM");
7183 buf4[2] = extra_char; /* make it "VIa", "VIb", etc. */
7184 if (GetTempFileName(szTempFile, buf4, 0, itmp) == 0)
7185 return NULL;
7186 /* GetTempFileName() will create the file, we don't want that */
7187 (void)DeleteFile(itmp);
7189 /* Backslashes in a temp file name cause problems when filtering with
7190 * "sh". NOTE: This also checks 'shellcmdflag' to help those people who
7191 * didn't set 'shellslash'. */
7192 retval = vim_strsave(itmp);
7193 if (*p_shcf == '-' || p_ssl)
7194 for (p = retval; *p; ++p)
7195 if (*p == '\\')
7196 *p = '/';
7197 return retval;
7199 # else /* WIN3264 */
7201 # ifdef USE_TMPNAM
7202 /* tmpnam() will make its own name */
7203 if (*tmpnam((char *)itmp) == NUL)
7204 return NULL;
7205 # else
7206 char_u *p;
7208 # ifdef VMS_TEMPNAM
7209 /* mktemp() is not working on VMS. It seems to be
7210 * a do-nothing function. Therefore we use tempnam().
7212 sprintf((char *)itmp, "VIM%c", extra_char);
7213 p = (char_u *)tempnam("tmp:", (char *)itmp);
7214 if (p != NULL)
7216 /* VMS will use '.LOG' if we don't explicitly specify an extension,
7217 * and VIM will then be unable to find the file later */
7218 STRCPY(itmp, p);
7219 STRCAT(itmp, ".txt");
7220 free(p);
7222 else
7223 return NULL;
7224 # else
7225 STRCPY(itmp, TEMPNAME);
7226 if ((p = vim_strchr(itmp, '?')) != NULL)
7227 *p = extra_char;
7228 if (mktemp((char *)itmp) == NULL)
7229 return NULL;
7230 # endif
7231 # endif
7233 return vim_strsave(itmp);
7234 # endif /* WIN3264 */
7235 #endif /* TEMPDIRNAMES */
7238 #if defined(BACKSLASH_IN_FILENAME) || defined(PROTO)
7240 * Convert all backslashes in fname to forward slashes in-place.
7242 void
7243 forward_slash(fname)
7244 char_u *fname;
7246 char_u *p;
7248 for (p = fname; *p != NUL; ++p)
7249 # ifdef FEAT_MBYTE
7250 /* The Big5 encoding can have '\' in the trail byte. */
7251 if (enc_dbcs != 0 && (*mb_ptr2len)(p) > 1)
7252 ++p;
7253 else
7254 # endif
7255 if (*p == '\\')
7256 *p = '/';
7258 #endif
7262 * Code for automatic commands.
7264 * Only included when "FEAT_AUTOCMD" has been defined.
7267 #if defined(FEAT_AUTOCMD) || defined(PROTO)
7270 * The autocommands are stored in a list for each event.
7271 * Autocommands for the same pattern, that are consecutive, are joined
7272 * together, to avoid having to match the pattern too often.
7273 * The result is an array of Autopat lists, which point to AutoCmd lists:
7275 * first_autopat[0] --> Autopat.next --> Autopat.next --> NULL
7276 * Autopat.cmds Autopat.cmds
7277 * | |
7278 * V V
7279 * AutoCmd.next AutoCmd.next
7280 * | |
7281 * V V
7282 * AutoCmd.next NULL
7285 * NULL
7287 * first_autopat[1] --> Autopat.next --> NULL
7288 * Autopat.cmds
7291 * AutoCmd.next
7294 * NULL
7295 * etc.
7297 * The order of AutoCmds is important, this is the order in which they were
7298 * defined and will have to be executed.
7300 typedef struct AutoCmd
7302 char_u *cmd; /* The command to be executed (NULL
7303 when command has been removed) */
7304 char nested; /* If autocommands nest here */
7305 char last; /* last command in list */
7306 #ifdef FEAT_EVAL
7307 scid_T scriptID; /* script ID where defined */
7308 #endif
7309 struct AutoCmd *next; /* Next AutoCmd in list */
7310 } AutoCmd;
7312 typedef struct AutoPat
7314 int group; /* group ID */
7315 char_u *pat; /* pattern as typed (NULL when pattern
7316 has been removed) */
7317 int patlen; /* strlen() of pat */
7318 regprog_T *reg_prog; /* compiled regprog for pattern */
7319 char allow_dirs; /* Pattern may match whole path */
7320 char last; /* last pattern for apply_autocmds() */
7321 AutoCmd *cmds; /* list of commands to do */
7322 struct AutoPat *next; /* next AutoPat in AutoPat list */
7323 int buflocal_nr; /* !=0 for buffer-local AutoPat */
7324 } AutoPat;
7326 static struct event_name
7328 char *name; /* event name */
7329 event_T event; /* event number */
7330 } event_names[] =
7332 {"BufAdd", EVENT_BUFADD},
7333 {"BufCreate", EVENT_BUFADD},
7334 {"BufDelete", EVENT_BUFDELETE},
7335 {"BufEnter", EVENT_BUFENTER},
7336 {"BufFilePost", EVENT_BUFFILEPOST},
7337 {"BufFilePre", EVENT_BUFFILEPRE},
7338 {"BufHidden", EVENT_BUFHIDDEN},
7339 {"BufLeave", EVENT_BUFLEAVE},
7340 {"BufNew", EVENT_BUFNEW},
7341 {"BufNewFile", EVENT_BUFNEWFILE},
7342 {"BufRead", EVENT_BUFREADPOST},
7343 {"BufReadCmd", EVENT_BUFREADCMD},
7344 {"BufReadPost", EVENT_BUFREADPOST},
7345 {"BufReadPre", EVENT_BUFREADPRE},
7346 {"BufUnload", EVENT_BUFUNLOAD},
7347 {"BufWinEnter", EVENT_BUFWINENTER},
7348 {"BufWinLeave", EVENT_BUFWINLEAVE},
7349 {"BufWipeout", EVENT_BUFWIPEOUT},
7350 {"BufWrite", EVENT_BUFWRITEPRE},
7351 {"BufWritePost", EVENT_BUFWRITEPOST},
7352 {"BufWritePre", EVENT_BUFWRITEPRE},
7353 {"BufWriteCmd", EVENT_BUFWRITECMD},
7354 {"CmdwinEnter", EVENT_CMDWINENTER},
7355 {"CmdwinLeave", EVENT_CMDWINLEAVE},
7356 {"ColorScheme", EVENT_COLORSCHEME},
7357 {"CursorHold", EVENT_CURSORHOLD},
7358 {"CursorHoldI", EVENT_CURSORHOLDI},
7359 {"CursorMoved", EVENT_CURSORMOVED},
7360 {"CursorMovedI", EVENT_CURSORMOVEDI},
7361 {"EncodingChanged", EVENT_ENCODINGCHANGED},
7362 {"FileEncoding", EVENT_ENCODINGCHANGED},
7363 {"FileAppendPost", EVENT_FILEAPPENDPOST},
7364 {"FileAppendPre", EVENT_FILEAPPENDPRE},
7365 {"FileAppendCmd", EVENT_FILEAPPENDCMD},
7366 {"FileChangedShell",EVENT_FILECHANGEDSHELL},
7367 {"FileChangedShellPost",EVENT_FILECHANGEDSHELLPOST},
7368 {"FileChangedRO", EVENT_FILECHANGEDRO},
7369 {"FileReadPost", EVENT_FILEREADPOST},
7370 {"FileReadPre", EVENT_FILEREADPRE},
7371 {"FileReadCmd", EVENT_FILEREADCMD},
7372 {"FileType", EVENT_FILETYPE},
7373 {"FileWritePost", EVENT_FILEWRITEPOST},
7374 {"FileWritePre", EVENT_FILEWRITEPRE},
7375 {"FileWriteCmd", EVENT_FILEWRITECMD},
7376 {"FilterReadPost", EVENT_FILTERREADPOST},
7377 {"FilterReadPre", EVENT_FILTERREADPRE},
7378 {"FilterWritePost", EVENT_FILTERWRITEPOST},
7379 {"FilterWritePre", EVENT_FILTERWRITEPRE},
7380 {"FocusGained", EVENT_FOCUSGAINED},
7381 {"FocusLost", EVENT_FOCUSLOST},
7382 {"FuncUndefined", EVENT_FUNCUNDEFINED},
7383 {"GUIEnter", EVENT_GUIENTER},
7384 {"GUIFailed", EVENT_GUIFAILED},
7385 {"InsertChange", EVENT_INSERTCHANGE},
7386 {"InsertEnter", EVENT_INSERTENTER},
7387 {"InsertLeave", EVENT_INSERTLEAVE},
7388 {"MenuPopup", EVENT_MENUPOPUP},
7389 {"QuickFixCmdPost", EVENT_QUICKFIXCMDPOST},
7390 {"QuickFixCmdPre", EVENT_QUICKFIXCMDPRE},
7391 {"RemoteReply", EVENT_REMOTEREPLY},
7392 {"SessionLoadPost", EVENT_SESSIONLOADPOST},
7393 {"ShellCmdPost", EVENT_SHELLCMDPOST},
7394 {"ShellFilterPost", EVENT_SHELLFILTERPOST},
7395 {"SourcePre", EVENT_SOURCEPRE},
7396 {"SourceCmd", EVENT_SOURCECMD},
7397 {"SpellFileMissing",EVENT_SPELLFILEMISSING},
7398 {"StdinReadPost", EVENT_STDINREADPOST},
7399 {"StdinReadPre", EVENT_STDINREADPRE},
7400 {"SwapExists", EVENT_SWAPEXISTS},
7401 {"Syntax", EVENT_SYNTAX},
7402 {"TabEnter", EVENT_TABENTER},
7403 {"TabLeave", EVENT_TABLEAVE},
7404 {"TermChanged", EVENT_TERMCHANGED},
7405 {"TermResponse", EVENT_TERMRESPONSE},
7406 {"User", EVENT_USER},
7407 {"VimEnter", EVENT_VIMENTER},
7408 {"VimLeave", EVENT_VIMLEAVE},
7409 {"VimLeavePre", EVENT_VIMLEAVEPRE},
7410 {"WinEnter", EVENT_WINENTER},
7411 {"WinLeave", EVENT_WINLEAVE},
7412 {"VimResized", EVENT_VIMRESIZED},
7413 {NULL, (event_T)0}
7416 static AutoPat *first_autopat[NUM_EVENTS] =
7418 NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL,
7419 NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL,
7420 NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL,
7421 NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL,
7422 NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL,
7423 NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL
7427 * struct used to keep status while executing autocommands for an event.
7429 typedef struct AutoPatCmd
7431 AutoPat *curpat; /* next AutoPat to examine */
7432 AutoCmd *nextcmd; /* next AutoCmd to execute */
7433 int group; /* group being used */
7434 char_u *fname; /* fname to match with */
7435 char_u *sfname; /* sfname to match with */
7436 char_u *tail; /* tail of fname */
7437 event_T event; /* current event */
7438 int arg_bufnr; /* initially equal to <abuf>, set to zero when
7439 buf is deleted */
7440 struct AutoPatCmd *next; /* chain of active apc-s for auto-invalidation*/
7441 } AutoPatCmd;
7443 static AutoPatCmd *active_apc_list = NULL; /* stack of active autocommands */
7446 * augroups stores a list of autocmd group names.
7448 static garray_T augroups = {0, 0, sizeof(char_u *), 10, NULL};
7449 #define AUGROUP_NAME(i) (((char_u **)augroups.ga_data)[i])
7452 * The ID of the current group. Group 0 is the default one.
7454 static int current_augroup = AUGROUP_DEFAULT;
7456 static int au_need_clean = FALSE; /* need to delete marked patterns */
7458 static void show_autocmd __ARGS((AutoPat *ap, event_T event));
7459 static void au_remove_pat __ARGS((AutoPat *ap));
7460 static void au_remove_cmds __ARGS((AutoPat *ap));
7461 static void au_cleanup __ARGS((void));
7462 static int au_new_group __ARGS((char_u *name));
7463 static void au_del_group __ARGS((char_u *name));
7464 static event_T event_name2nr __ARGS((char_u *start, char_u **end));
7465 static char_u *event_nr2name __ARGS((event_T event));
7466 static char_u *find_end_event __ARGS((char_u *arg, int have_group));
7467 static int event_ignored __ARGS((event_T event));
7468 static int au_get_grouparg __ARGS((char_u **argp));
7469 static int do_autocmd_event __ARGS((event_T event, char_u *pat, int nested, char_u *cmd, int forceit, int group));
7470 static char_u *getnextac __ARGS((int c, void *cookie, int indent));
7471 static int apply_autocmds_group __ARGS((event_T event, char_u *fname, char_u *fname_io, int force, int group, buf_T *buf, exarg_T *eap));
7472 static void auto_next_pat __ARGS((AutoPatCmd *apc, int stop_at_last));
7475 static event_T last_event;
7476 static int last_group;
7477 static int autocmd_blocked = 0; /* block all autocmds */
7480 * Show the autocommands for one AutoPat.
7482 static void
7483 show_autocmd(ap, event)
7484 AutoPat *ap;
7485 event_T event;
7487 AutoCmd *ac;
7489 /* Check for "got_int" (here and at various places below), which is set
7490 * when "q" has been hit for the "--more--" prompt */
7491 if (got_int)
7492 return;
7493 if (ap->pat == NULL) /* pattern has been removed */
7494 return;
7496 msg_putchar('\n');
7497 if (got_int)
7498 return;
7499 if (event != last_event || ap->group != last_group)
7501 if (ap->group != AUGROUP_DEFAULT)
7503 if (AUGROUP_NAME(ap->group) == NULL)
7504 msg_puts_attr((char_u *)_("--Deleted--"), hl_attr(HLF_E));
7505 else
7506 msg_puts_attr(AUGROUP_NAME(ap->group), hl_attr(HLF_T));
7507 msg_puts((char_u *)" ");
7509 msg_puts_attr(event_nr2name(event), hl_attr(HLF_T));
7510 last_event = event;
7511 last_group = ap->group;
7512 msg_putchar('\n');
7513 if (got_int)
7514 return;
7516 msg_col = 4;
7517 msg_outtrans(ap->pat);
7519 for (ac = ap->cmds; ac != NULL; ac = ac->next)
7521 if (ac->cmd != NULL) /* skip removed commands */
7523 if (msg_col >= 14)
7524 msg_putchar('\n');
7525 msg_col = 14;
7526 if (got_int)
7527 return;
7528 msg_outtrans(ac->cmd);
7529 #ifdef FEAT_EVAL
7530 if (p_verbose > 0)
7531 last_set_msg(ac->scriptID);
7532 #endif
7533 if (got_int)
7534 return;
7535 if (ac->next != NULL)
7537 msg_putchar('\n');
7538 if (got_int)
7539 return;
7546 * Mark an autocommand pattern for deletion.
7548 static void
7549 au_remove_pat(ap)
7550 AutoPat *ap;
7552 vim_free(ap->pat);
7553 ap->pat = NULL;
7554 ap->buflocal_nr = -1;
7555 au_need_clean = TRUE;
7559 * Mark all commands for a pattern for deletion.
7561 static void
7562 au_remove_cmds(ap)
7563 AutoPat *ap;
7565 AutoCmd *ac;
7567 for (ac = ap->cmds; ac != NULL; ac = ac->next)
7569 vim_free(ac->cmd);
7570 ac->cmd = NULL;
7572 au_need_clean = TRUE;
7576 * Cleanup autocommands and patterns that have been deleted.
7577 * This is only done when not executing autocommands.
7579 static void
7580 au_cleanup()
7582 AutoPat *ap, **prev_ap;
7583 AutoCmd *ac, **prev_ac;
7584 event_T event;
7586 if (autocmd_busy || !au_need_clean)
7587 return;
7589 /* loop over all events */
7590 for (event = (event_T)0; (int)event < (int)NUM_EVENTS;
7591 event = (event_T)((int)event + 1))
7593 /* loop over all autocommand patterns */
7594 prev_ap = &(first_autopat[(int)event]);
7595 for (ap = *prev_ap; ap != NULL; ap = *prev_ap)
7597 /* loop over all commands for this pattern */
7598 prev_ac = &(ap->cmds);
7599 for (ac = *prev_ac; ac != NULL; ac = *prev_ac)
7601 /* remove the command if the pattern is to be deleted or when
7602 * the command has been marked for deletion */
7603 if (ap->pat == NULL || ac->cmd == NULL)
7605 *prev_ac = ac->next;
7606 vim_free(ac->cmd);
7607 vim_free(ac);
7609 else
7610 prev_ac = &(ac->next);
7613 /* remove the pattern if it has been marked for deletion */
7614 if (ap->pat == NULL)
7616 *prev_ap = ap->next;
7617 vim_free(ap->reg_prog);
7618 vim_free(ap);
7620 else
7621 prev_ap = &(ap->next);
7625 au_need_clean = FALSE;
7629 * Called when buffer is freed, to remove/invalidate related buffer-local
7630 * autocmds.
7632 void
7633 aubuflocal_remove(buf)
7634 buf_T *buf;
7636 AutoPat *ap;
7637 event_T event;
7638 AutoPatCmd *apc;
7640 /* invalidate currently executing autocommands */
7641 for (apc = active_apc_list; apc; apc = apc->next)
7642 if (buf->b_fnum == apc->arg_bufnr)
7643 apc->arg_bufnr = 0;
7645 /* invalidate buflocals looping through events */
7646 for (event = (event_T)0; (int)event < (int)NUM_EVENTS;
7647 event = (event_T)((int)event + 1))
7648 /* loop over all autocommand patterns */
7649 for (ap = first_autopat[(int)event]; ap != NULL; ap = ap->next)
7650 if (ap->buflocal_nr == buf->b_fnum)
7652 au_remove_pat(ap);
7653 if (p_verbose >= 6)
7655 verbose_enter();
7656 smsg((char_u *)
7657 _("auto-removing autocommand: %s <buffer=%d>"),
7658 event_nr2name(event), buf->b_fnum);
7659 verbose_leave();
7662 au_cleanup();
7666 * Add an autocmd group name.
7667 * Return it's ID. Returns AUGROUP_ERROR (< 0) for error.
7669 static int
7670 au_new_group(name)
7671 char_u *name;
7673 int i;
7675 i = au_find_group(name);
7676 if (i == AUGROUP_ERROR) /* the group doesn't exist yet, add it */
7678 /* First try using a free entry. */
7679 for (i = 0; i < augroups.ga_len; ++i)
7680 if (AUGROUP_NAME(i) == NULL)
7681 break;
7682 if (i == augroups.ga_len && ga_grow(&augroups, 1) == FAIL)
7683 return AUGROUP_ERROR;
7685 AUGROUP_NAME(i) = vim_strsave(name);
7686 if (AUGROUP_NAME(i) == NULL)
7687 return AUGROUP_ERROR;
7688 if (i == augroups.ga_len)
7689 ++augroups.ga_len;
7692 return i;
7695 static void
7696 au_del_group(name)
7697 char_u *name;
7699 int i;
7701 i = au_find_group(name);
7702 if (i == AUGROUP_ERROR) /* the group doesn't exist */
7703 EMSG2(_("E367: No such group: \"%s\""), name);
7704 else
7706 vim_free(AUGROUP_NAME(i));
7707 AUGROUP_NAME(i) = NULL;
7712 * Find the ID of an autocmd group name.
7713 * Return it's ID. Returns AUGROUP_ERROR (< 0) for error.
7715 static int
7716 au_find_group(name)
7717 char_u *name;
7719 int i;
7721 for (i = 0; i < augroups.ga_len; ++i)
7722 if (AUGROUP_NAME(i) != NULL && STRCMP(AUGROUP_NAME(i), name) == 0)
7723 return i;
7724 return AUGROUP_ERROR;
7728 * Return TRUE if augroup "name" exists.
7731 au_has_group(name)
7732 char_u *name;
7734 return au_find_group(name) != AUGROUP_ERROR;
7738 * ":augroup {name}".
7740 void
7741 do_augroup(arg, del_group)
7742 char_u *arg;
7743 int del_group;
7745 int i;
7747 if (del_group)
7749 if (*arg == NUL)
7750 EMSG(_(e_argreq));
7751 else
7752 au_del_group(arg);
7754 else if (STRICMP(arg, "end") == 0) /* ":aug end": back to group 0 */
7755 current_augroup = AUGROUP_DEFAULT;
7756 else if (*arg) /* ":aug xxx": switch to group xxx */
7758 i = au_new_group(arg);
7759 if (i != AUGROUP_ERROR)
7760 current_augroup = i;
7762 else /* ":aug": list the group names */
7764 msg_start();
7765 for (i = 0; i < augroups.ga_len; ++i)
7767 if (AUGROUP_NAME(i) != NULL)
7769 msg_puts(AUGROUP_NAME(i));
7770 msg_puts((char_u *)" ");
7773 msg_clr_eos();
7774 msg_end();
7778 #if defined(EXITFREE) || defined(PROTO)
7779 void
7780 free_all_autocmds()
7782 for (current_augroup = -1; current_augroup < augroups.ga_len;
7783 ++current_augroup)
7784 do_autocmd((char_u *)"", TRUE);
7785 ga_clear_strings(&augroups);
7787 #endif
7790 * Return the event number for event name "start".
7791 * Return NUM_EVENTS if the event name was not found.
7792 * Return a pointer to the next event name in "end".
7794 static event_T
7795 event_name2nr(start, end)
7796 char_u *start;
7797 char_u **end;
7799 char_u *p;
7800 int i;
7801 int len;
7803 /* the event name ends with end of line, a blank or a comma */
7804 for (p = start; *p && !vim_iswhite(*p) && *p != ','; ++p)
7806 for (i = 0; event_names[i].name != NULL; ++i)
7808 len = (int)STRLEN(event_names[i].name);
7809 if (len == p - start && STRNICMP(event_names[i].name, start, len) == 0)
7810 break;
7812 if (*p == ',')
7813 ++p;
7814 *end = p;
7815 if (event_names[i].name == NULL)
7816 return NUM_EVENTS;
7817 return event_names[i].event;
7821 * Return the name for event "event".
7823 static char_u *
7824 event_nr2name(event)
7825 event_T event;
7827 int i;
7829 for (i = 0; event_names[i].name != NULL; ++i)
7830 if (event_names[i].event == event)
7831 return (char_u *)event_names[i].name;
7832 return (char_u *)"Unknown";
7836 * Scan over the events. "*" stands for all events.
7838 static char_u *
7839 find_end_event(arg, have_group)
7840 char_u *arg;
7841 int have_group; /* TRUE when group name was found */
7843 char_u *pat;
7844 char_u *p;
7846 if (*arg == '*')
7848 if (arg[1] && !vim_iswhite(arg[1]))
7850 EMSG2(_("E215: Illegal character after *: %s"), arg);
7851 return NULL;
7853 pat = arg + 1;
7855 else
7857 for (pat = arg; *pat && !vim_iswhite(*pat); pat = p)
7859 if ((int)event_name2nr(pat, &p) >= (int)NUM_EVENTS)
7861 if (have_group)
7862 EMSG2(_("E216: No such event: %s"), pat);
7863 else
7864 EMSG2(_("E216: No such group or event: %s"), pat);
7865 return NULL;
7869 return pat;
7873 * Return TRUE if "event" is included in 'eventignore'.
7875 static int
7876 event_ignored(event)
7877 event_T event;
7879 char_u *p = p_ei;
7881 while (*p != NUL)
7883 if (STRNICMP(p, "all", 3) == 0 && (p[3] == NUL || p[3] == ','))
7884 return TRUE;
7885 if (event_name2nr(p, &p) == event)
7886 return TRUE;
7889 return FALSE;
7893 * Return OK when the contents of p_ei is valid, FAIL otherwise.
7896 check_ei()
7898 char_u *p = p_ei;
7900 while (*p)
7902 if (STRNICMP(p, "all", 3) == 0 && (p[3] == NUL || p[3] == ','))
7904 p += 3;
7905 if (*p == ',')
7906 ++p;
7908 else if (event_name2nr(p, &p) == NUM_EVENTS)
7909 return FAIL;
7912 return OK;
7915 # if defined(FEAT_SYN_HL) || defined(PROTO)
7918 * Add "what" to 'eventignore' to skip loading syntax highlighting for every
7919 * buffer loaded into the window. "what" must start with a comma.
7920 * Returns the old value of 'eventignore' in allocated memory.
7922 char_u *
7923 au_event_disable(what)
7924 char *what;
7926 char_u *new_ei;
7927 char_u *save_ei;
7929 save_ei = vim_strsave(p_ei);
7930 if (save_ei != NULL)
7932 new_ei = vim_strnsave(p_ei, (int)(STRLEN(p_ei) + STRLEN(what)));
7933 if (new_ei != NULL)
7935 if (*what == ',' && *p_ei == NUL)
7936 STRCPY(new_ei, what + 1);
7937 else
7938 STRCAT(new_ei, what);
7939 set_string_option_direct((char_u *)"ei", -1, new_ei,
7940 OPT_FREE, SID_NONE);
7941 vim_free(new_ei);
7944 return save_ei;
7947 void
7948 au_event_restore(old_ei)
7949 char_u *old_ei;
7951 if (old_ei != NULL)
7953 set_string_option_direct((char_u *)"ei", -1, old_ei,
7954 OPT_FREE, SID_NONE);
7955 vim_free(old_ei);
7958 # endif /* FEAT_SYN_HL */
7961 * do_autocmd() -- implements the :autocmd command. Can be used in the
7962 * following ways:
7964 * :autocmd <event> <pat> <cmd> Add <cmd> to the list of commands that
7965 * will be automatically executed for <event>
7966 * when editing a file matching <pat>, in
7967 * the current group.
7968 * :autocmd <event> <pat> Show the auto-commands associated with
7969 * <event> and <pat>.
7970 * :autocmd <event> Show the auto-commands associated with
7971 * <event>.
7972 * :autocmd Show all auto-commands.
7973 * :autocmd! <event> <pat> <cmd> Remove all auto-commands associated with
7974 * <event> and <pat>, and add the command
7975 * <cmd>, for the current group.
7976 * :autocmd! <event> <pat> Remove all auto-commands associated with
7977 * <event> and <pat> for the current group.
7978 * :autocmd! <event> Remove all auto-commands associated with
7979 * <event> for the current group.
7980 * :autocmd! Remove ALL auto-commands for the current
7981 * group.
7983 * Multiple events and patterns may be given separated by commas. Here are
7984 * some examples:
7985 * :autocmd bufread,bufenter *.c,*.h set tw=0 smartindent noic
7986 * :autocmd bufleave * set tw=79 nosmartindent ic infercase
7988 * :autocmd * *.c show all autocommands for *.c files.
7990 * Mostly a {group} argument can optionally appear before <event>.
7992 void
7993 do_autocmd(arg, forceit)
7994 char_u *arg;
7995 int forceit;
7997 char_u *pat;
7998 char_u *envpat = NULL;
7999 char_u *cmd;
8000 event_T event;
8001 int need_free = FALSE;
8002 int nested = FALSE;
8003 int group;
8006 * Check for a legal group name. If not, use AUGROUP_ALL.
8008 group = au_get_grouparg(&arg);
8009 if (arg == NULL) /* out of memory */
8010 return;
8013 * Scan over the events.
8014 * If we find an illegal name, return here, don't do anything.
8016 pat = find_end_event(arg, group != AUGROUP_ALL);
8017 if (pat == NULL)
8018 return;
8021 * Scan over the pattern. Put a NUL at the end.
8023 pat = skipwhite(pat);
8024 cmd = pat;
8025 while (*cmd && (!vim_iswhite(*cmd) || cmd[-1] == '\\'))
8026 cmd++;
8027 if (*cmd)
8028 *cmd++ = NUL;
8030 /* Expand environment variables in the pattern. Set 'shellslash', we want
8031 * forward slashes here. */
8032 if (vim_strchr(pat, '$') != NULL || vim_strchr(pat, '~') != NULL)
8034 #ifdef BACKSLASH_IN_FILENAME
8035 int p_ssl_save = p_ssl;
8037 p_ssl = TRUE;
8038 #endif
8039 envpat = expand_env_save(pat);
8040 #ifdef BACKSLASH_IN_FILENAME
8041 p_ssl = p_ssl_save;
8042 #endif
8043 if (envpat != NULL)
8044 pat = envpat;
8048 * Check for "nested" flag.
8050 cmd = skipwhite(cmd);
8051 if (*cmd != NUL && STRNCMP(cmd, "nested", 6) == 0 && vim_iswhite(cmd[6]))
8053 nested = TRUE;
8054 cmd = skipwhite(cmd + 6);
8058 * Find the start of the commands.
8059 * Expand <sfile> in it.
8061 if (*cmd != NUL)
8063 cmd = expand_sfile(cmd);
8064 if (cmd == NULL) /* some error */
8065 return;
8066 need_free = TRUE;
8070 * Print header when showing autocommands.
8072 if (!forceit && *cmd == NUL)
8074 /* Highlight title */
8075 MSG_PUTS_TITLE(_("\n--- Auto-Commands ---"));
8079 * Loop over the events.
8081 last_event = (event_T)-1; /* for listing the event name */
8082 last_group = AUGROUP_ERROR; /* for listing the group name */
8083 if (*arg == '*' || *arg == NUL)
8085 for (event = (event_T)0; (int)event < (int)NUM_EVENTS;
8086 event = (event_T)((int)event + 1))
8087 if (do_autocmd_event(event, pat,
8088 nested, cmd, forceit, group) == FAIL)
8089 break;
8091 else
8093 while (*arg && !vim_iswhite(*arg))
8094 if (do_autocmd_event(event_name2nr(arg, &arg), pat,
8095 nested, cmd, forceit, group) == FAIL)
8096 break;
8099 if (need_free)
8100 vim_free(cmd);
8101 vim_free(envpat);
8105 * Find the group ID in a ":autocmd" or ":doautocmd" argument.
8106 * The "argp" argument is advanced to the following argument.
8108 * Returns the group ID, AUGROUP_ERROR for error (out of memory).
8110 static int
8111 au_get_grouparg(argp)
8112 char_u **argp;
8114 char_u *group_name;
8115 char_u *p;
8116 char_u *arg = *argp;
8117 int group = AUGROUP_ALL;
8119 p = skiptowhite(arg);
8120 if (p > arg)
8122 group_name = vim_strnsave(arg, (int)(p - arg));
8123 if (group_name == NULL) /* out of memory */
8124 return AUGROUP_ERROR;
8125 group = au_find_group(group_name);
8126 if (group == AUGROUP_ERROR)
8127 group = AUGROUP_ALL; /* no match, use all groups */
8128 else
8129 *argp = skipwhite(p); /* match, skip over group name */
8130 vim_free(group_name);
8132 return group;
8136 * do_autocmd() for one event.
8137 * If *pat == NUL do for all patterns.
8138 * If *cmd == NUL show entries.
8139 * If forceit == TRUE delete entries.
8140 * If group is not AUGROUP_ALL, only use this group.
8142 static int
8143 do_autocmd_event(event, pat, nested, cmd, forceit, group)
8144 event_T event;
8145 char_u *pat;
8146 int nested;
8147 char_u *cmd;
8148 int forceit;
8149 int group;
8151 AutoPat *ap;
8152 AutoPat **prev_ap;
8153 AutoCmd *ac;
8154 AutoCmd **prev_ac;
8155 int brace_level;
8156 char_u *endpat;
8157 int findgroup;
8158 int allgroups;
8159 int patlen;
8160 int is_buflocal;
8161 int buflocal_nr;
8162 char_u buflocal_pat[25]; /* for "<buffer=X>" */
8164 if (group == AUGROUP_ALL)
8165 findgroup = current_augroup;
8166 else
8167 findgroup = group;
8168 allgroups = (group == AUGROUP_ALL && !forceit && *cmd == NUL);
8171 * Show or delete all patterns for an event.
8173 if (*pat == NUL)
8175 for (ap = first_autopat[(int)event]; ap != NULL; ap = ap->next)
8177 if (forceit) /* delete the AutoPat, if it's in the current group */
8179 if (ap->group == findgroup)
8180 au_remove_pat(ap);
8182 else if (group == AUGROUP_ALL || ap->group == group)
8183 show_autocmd(ap, event);
8188 * Loop through all the specified patterns.
8190 for ( ; *pat; pat = (*endpat == ',' ? endpat + 1 : endpat))
8193 * Find end of the pattern.
8194 * Watch out for a comma in braces, like "*.\{obj,o\}".
8196 brace_level = 0;
8197 for (endpat = pat; *endpat && (*endpat != ',' || brace_level
8198 || endpat[-1] == '\\'); ++endpat)
8200 if (*endpat == '{')
8201 brace_level++;
8202 else if (*endpat == '}')
8203 brace_level--;
8205 if (pat == endpat) /* ignore single comma */
8206 continue;
8207 patlen = (int)(endpat - pat);
8210 * detect special <buflocal[=X]> buffer-local patterns
8212 is_buflocal = FALSE;
8213 buflocal_nr = 0;
8215 if (patlen >= 7 && STRNCMP(pat, "<buffer", 7) == 0
8216 && pat[patlen - 1] == '>')
8218 /* Error will be printed only for addition. printing and removing
8219 * will proceed silently. */
8220 is_buflocal = TRUE;
8221 if (patlen == 8)
8222 buflocal_nr = curbuf->b_fnum;
8223 else if (patlen > 9 && pat[7] == '=')
8225 /* <buffer=abuf> */
8226 if (patlen == 13 && STRNICMP(pat, "<buffer=abuf>", 13))
8227 buflocal_nr = autocmd_bufnr;
8228 /* <buffer=123> */
8229 else if (skipdigits(pat + 8) == pat + patlen - 1)
8230 buflocal_nr = atoi((char *)pat + 8);
8234 if (is_buflocal)
8236 /* normalize pat into standard "<buffer>#N" form */
8237 sprintf((char *)buflocal_pat, "<buffer=%d>", buflocal_nr);
8238 pat = buflocal_pat; /* can modify pat and patlen */
8239 patlen = (int)STRLEN(buflocal_pat); /* but not endpat */
8243 * Find AutoPat entries with this pattern.
8245 prev_ap = &first_autopat[(int)event];
8246 while ((ap = *prev_ap) != NULL)
8248 if (ap->pat != NULL)
8250 /* Accept a pattern when:
8251 * - a group was specified and it's that group, or a group was
8252 * not specified and it's the current group, or a group was
8253 * not specified and we are listing
8254 * - the length of the pattern matches
8255 * - the pattern matches.
8256 * For <buffer[=X]>, this condition works because we normalize
8257 * all buffer-local patterns.
8259 if ((allgroups || ap->group == findgroup)
8260 && ap->patlen == patlen
8261 && STRNCMP(pat, ap->pat, patlen) == 0)
8264 * Remove existing autocommands.
8265 * If adding any new autocmd's for this AutoPat, don't
8266 * delete the pattern from the autopat list, append to
8267 * this list.
8269 if (forceit)
8271 if (*cmd != NUL && ap->next == NULL)
8273 au_remove_cmds(ap);
8274 break;
8276 au_remove_pat(ap);
8280 * Show autocmd's for this autopat, or buflocals <buffer=X>
8282 else if (*cmd == NUL)
8283 show_autocmd(ap, event);
8286 * Add autocmd to this autopat, if it's the last one.
8288 else if (ap->next == NULL)
8289 break;
8292 prev_ap = &ap->next;
8296 * Add a new command.
8298 if (*cmd != NUL)
8301 * If the pattern we want to add a command to does appear at the
8302 * end of the list (or not is not in the list at all), add the
8303 * pattern at the end of the list.
8305 if (ap == NULL)
8307 /* refuse to add buffer-local ap if buffer number is invalid */
8308 if (is_buflocal && (buflocal_nr == 0
8309 || buflist_findnr(buflocal_nr) == NULL))
8311 EMSGN(_("E680: <buffer=%d>: invalid buffer number "),
8312 buflocal_nr);
8313 return FAIL;
8316 ap = (AutoPat *)alloc((unsigned)sizeof(AutoPat));
8317 if (ap == NULL)
8318 return FAIL;
8319 ap->pat = vim_strnsave(pat, patlen);
8320 ap->patlen = patlen;
8321 if (ap->pat == NULL)
8323 vim_free(ap);
8324 return FAIL;
8327 if (is_buflocal)
8329 ap->buflocal_nr = buflocal_nr;
8330 ap->reg_prog = NULL;
8332 else
8334 char_u *reg_pat;
8336 ap->buflocal_nr = 0;
8337 reg_pat = file_pat_to_reg_pat(pat, endpat,
8338 &ap->allow_dirs, TRUE);
8339 if (reg_pat != NULL)
8340 ap->reg_prog = vim_regcomp(reg_pat, RE_MAGIC);
8341 vim_free(reg_pat);
8342 if (reg_pat == NULL || ap->reg_prog == NULL)
8344 vim_free(ap->pat);
8345 vim_free(ap);
8346 return FAIL;
8349 ap->cmds = NULL;
8350 *prev_ap = ap;
8351 ap->next = NULL;
8352 if (group == AUGROUP_ALL)
8353 ap->group = current_augroup;
8354 else
8355 ap->group = group;
8359 * Add the autocmd at the end of the AutoCmd list.
8361 prev_ac = &(ap->cmds);
8362 while ((ac = *prev_ac) != NULL)
8363 prev_ac = &ac->next;
8364 ac = (AutoCmd *)alloc((unsigned)sizeof(AutoCmd));
8365 if (ac == NULL)
8366 return FAIL;
8367 ac->cmd = vim_strsave(cmd);
8368 #ifdef FEAT_EVAL
8369 ac->scriptID = current_SID;
8370 #endif
8371 if (ac->cmd == NULL)
8373 vim_free(ac);
8374 return FAIL;
8376 ac->next = NULL;
8377 *prev_ac = ac;
8378 ac->nested = nested;
8382 au_cleanup(); /* may really delete removed patterns/commands now */
8383 return OK;
8387 * Implementation of ":doautocmd [group] event [fname]".
8388 * Return OK for success, FAIL for failure;
8391 do_doautocmd(arg, do_msg)
8392 char_u *arg;
8393 int do_msg; /* give message for no matching autocmds? */
8395 char_u *fname;
8396 int nothing_done = TRUE;
8397 int group;
8400 * Check for a legal group name. If not, use AUGROUP_ALL.
8402 group = au_get_grouparg(&arg);
8403 if (arg == NULL) /* out of memory */
8404 return FAIL;
8406 if (*arg == '*')
8408 EMSG(_("E217: Can't execute autocommands for ALL events"));
8409 return FAIL;
8413 * Scan over the events.
8414 * If we find an illegal name, return here, don't do anything.
8416 fname = find_end_event(arg, group != AUGROUP_ALL);
8417 if (fname == NULL)
8418 return FAIL;
8420 fname = skipwhite(fname);
8423 * Loop over the events.
8425 while (*arg && !vim_iswhite(*arg))
8426 if (apply_autocmds_group(event_name2nr(arg, &arg),
8427 fname, NULL, TRUE, group, curbuf, NULL))
8428 nothing_done = FALSE;
8430 if (nothing_done && do_msg)
8431 MSG(_("No matching autocommands"));
8433 #ifdef FEAT_EVAL
8434 return aborting() ? FAIL : OK;
8435 #else
8436 return OK;
8437 #endif
8441 * ":doautoall": execute autocommands for each loaded buffer.
8443 void
8444 ex_doautoall(eap)
8445 exarg_T *eap;
8447 int retval;
8448 aco_save_T aco;
8449 buf_T *buf;
8452 * This is a bit tricky: For some commands curwin->w_buffer needs to be
8453 * equal to curbuf, but for some buffers there may not be a window.
8454 * So we change the buffer for the current window for a moment. This
8455 * gives problems when the autocommands make changes to the list of
8456 * buffers or windows...
8458 for (buf = firstbuf; buf != NULL; buf = buf->b_next)
8460 if (buf->b_ml.ml_mfp != NULL)
8462 /* find a window for this buffer and save some values */
8463 aucmd_prepbuf(&aco, buf);
8465 /* execute the autocommands for this buffer */
8466 retval = do_doautocmd(eap->arg, FALSE);
8468 /* Execute the modeline settings, but don't set window-local
8469 * options if we are using the current window for another buffer. */
8470 do_modelines(curwin == aucmd_win ? OPT_NOWIN : 0);
8472 /* restore the current window */
8473 aucmd_restbuf(&aco);
8475 /* stop if there is some error or buffer was deleted */
8476 if (retval == FAIL || !buf_valid(buf))
8477 break;
8481 check_cursor(); /* just in case lines got deleted */
8485 * Prepare for executing autocommands for (hidden) buffer "buf".
8486 * Search for a visible window containing the current buffer. If there isn't
8487 * one then use "aucmd_win".
8488 * Set "curbuf" and "curwin" to match "buf".
8489 * When FEAT_AUTOCMD is not defined another version is used, see below.
8491 void
8492 aucmd_prepbuf(aco, buf)
8493 aco_save_T *aco; /* structure to save values in */
8494 buf_T *buf; /* new curbuf */
8496 win_T *win;
8497 #ifdef FEAT_WINDOWS
8498 int save_ea;
8499 #endif
8501 /* Find a window that is for the new buffer */
8502 if (buf == curbuf) /* be quick when buf is curbuf */
8503 win = curwin;
8504 else
8505 #ifdef FEAT_WINDOWS
8506 for (win = firstwin; win != NULL; win = win->w_next)
8507 if (win->w_buffer == buf)
8508 break;
8509 #else
8510 win = NULL;
8511 #endif
8513 /* Allocate "aucmd_win" when needed. If this fails (out of memory) fall
8514 * back to using the current window. */
8515 if (win == NULL && aucmd_win == NULL)
8517 win_alloc_aucmd_win();
8518 if (aucmd_win == NULL)
8519 win = curwin;
8521 if (win == NULL && aucmd_win_used)
8522 /* Strange recursive autocommand, fall back to using the current
8523 * window. Expect a few side effects... */
8524 win = curwin;
8526 aco->save_curwin = curwin;
8527 aco->save_curbuf = curbuf;
8528 if (win != NULL)
8530 /* There is a window for "buf" in the current tab page, make it the
8531 * curwin. This is preferred, it has the least side effects (esp. if
8532 * "buf" is curbuf). */
8533 aco->use_aucmd_win = FALSE;
8534 curwin = win;
8536 else
8538 /* There is no window for "buf", use "aucmd_win". To minimize the side
8539 * effects, insert it in a the current tab page.
8540 * Anything related to a window (e.g., setting folds) may have
8541 * unexpected results. */
8542 aco->use_aucmd_win = TRUE;
8543 aucmd_win_used = TRUE;
8544 aucmd_win->w_buffer = buf;
8545 ++buf->b_nwindows;
8546 win_init_empty(aucmd_win); /* set cursor and topline to safe values */
8547 vim_free(aucmd_win->w_localdir);
8548 aucmd_win->w_localdir = NULL;
8550 /* Make sure w_localdir and globaldir are NULL to avoid a chdir() in
8551 * win_enter_ext(). */
8552 aucmd_win->w_localdir = NULL;
8553 aco->globaldir = globaldir;
8554 globaldir = NULL;
8557 #ifdef FEAT_WINDOWS
8558 /* Split the current window, put the aucmd_win in the upper half.
8559 * We don't want the BufEnter or WinEnter autocommands. */
8560 block_autocmds();
8561 make_snapshot(SNAP_AUCMD_IDX);
8562 save_ea = p_ea;
8563 p_ea = FALSE;
8564 (void)win_split_ins(0, WSP_TOP, aucmd_win, 0);
8565 (void)win_comp_pos(); /* recompute window positions */
8566 p_ea = save_ea;
8567 unblock_autocmds();
8568 #endif
8569 curwin = aucmd_win;
8571 curbuf = buf;
8572 aco->new_curwin = curwin;
8573 aco->new_curbuf = curbuf;
8577 * Cleanup after executing autocommands for a (hidden) buffer.
8578 * Restore the window as it was (if possible).
8579 * When FEAT_AUTOCMD is not defined another version is used, see below.
8581 void
8582 aucmd_restbuf(aco)
8583 aco_save_T *aco; /* structure holding saved values */
8585 #ifdef FEAT_WINDOWS
8586 int dummy;
8587 #endif
8589 if (aco->use_aucmd_win)
8591 --curbuf->b_nwindows;
8592 #ifdef FEAT_WINDOWS
8593 /* Find "aucmd_win", it can't be closed, but it may be in another tab
8594 * page. Do not trigger autocommands here. */
8595 block_autocmds();
8596 if (curwin != aucmd_win)
8598 tabpage_T *tp;
8599 win_T *wp;
8601 FOR_ALL_TAB_WINDOWS(tp, wp)
8603 if (wp == aucmd_win)
8605 if (tp != curtab)
8606 goto_tabpage_tp(tp);
8607 win_goto(aucmd_win);
8608 break;
8613 /* Remove the window and frame from the tree of frames. */
8614 (void)winframe_remove(curwin, &dummy, NULL);
8615 win_remove(curwin, NULL);
8616 aucmd_win_used = FALSE;
8617 last_status(FALSE); /* may need to remove last status line */
8618 restore_snapshot(SNAP_AUCMD_IDX, FALSE);
8619 (void)win_comp_pos(); /* recompute window positions */
8620 unblock_autocmds();
8622 if (win_valid(aco->save_curwin))
8623 curwin = aco->save_curwin;
8624 else
8625 /* Hmm, original window disappeared. Just use the first one. */
8626 curwin = firstwin;
8627 # ifdef FEAT_EVAL
8628 vars_clear(&aucmd_win->w_vars.dv_hashtab); /* free all w: variables */
8629 hash_init(&aucmd_win->w_vars.dv_hashtab); /* re-use the hashtab */
8630 # endif
8631 #else
8632 curwin = aco->save_curwin;
8633 #endif
8634 curbuf = curwin->w_buffer;
8636 vim_free(globaldir);
8637 globaldir = aco->globaldir;
8639 /* the buffer contents may have changed */
8640 check_cursor();
8641 if (curwin->w_topline > curbuf->b_ml.ml_line_count)
8643 curwin->w_topline = curbuf->b_ml.ml_line_count;
8644 #ifdef FEAT_DIFF
8645 curwin->w_topfill = 0;
8646 #endif
8648 #if defined(FEAT_GUI)
8649 /* Hide the scrollbars from the aucmd_win and update. */
8650 gui_mch_enable_scrollbar(&aucmd_win->w_scrollbars[SBAR_LEFT], FALSE);
8651 gui_mch_enable_scrollbar(&aucmd_win->w_scrollbars[SBAR_RIGHT], FALSE);
8652 gui_may_update_scrollbars();
8653 #endif
8655 else
8657 /* restore curwin */
8658 #ifdef FEAT_WINDOWS
8659 if (win_valid(aco->save_curwin))
8660 #endif
8662 /* Restore the buffer which was previously edited by curwin, if
8663 * it was changed, we are still the same window and the buffer is
8664 * valid. */
8665 if (curwin == aco->new_curwin
8666 && curbuf != aco->new_curbuf
8667 && buf_valid(aco->new_curbuf)
8668 && aco->new_curbuf->b_ml.ml_mfp != NULL)
8670 --curbuf->b_nwindows;
8671 curbuf = aco->new_curbuf;
8672 curwin->w_buffer = curbuf;
8673 ++curbuf->b_nwindows;
8676 curwin = aco->save_curwin;
8677 curbuf = curwin->w_buffer;
8682 static int autocmd_nested = FALSE;
8685 * Execute autocommands for "event" and file name "fname".
8686 * Return TRUE if some commands were executed.
8689 apply_autocmds(event, fname, fname_io, force, buf)
8690 event_T event;
8691 char_u *fname; /* NULL or empty means use actual file name */
8692 char_u *fname_io; /* fname to use for <afile> on cmdline */
8693 int force; /* when TRUE, ignore autocmd_busy */
8694 buf_T *buf; /* buffer for <abuf> */
8696 return apply_autocmds_group(event, fname, fname_io, force,
8697 AUGROUP_ALL, buf, NULL);
8701 * Like apply_autocmds(), but with extra "eap" argument. This takes care of
8702 * setting v:filearg.
8704 static int
8705 apply_autocmds_exarg(event, fname, fname_io, force, buf, eap)
8706 event_T event;
8707 char_u *fname;
8708 char_u *fname_io;
8709 int force;
8710 buf_T *buf;
8711 exarg_T *eap;
8713 return apply_autocmds_group(event, fname, fname_io, force,
8714 AUGROUP_ALL, buf, eap);
8718 * Like apply_autocmds(), but handles the caller's retval. If the script
8719 * processing is being aborted or if retval is FAIL when inside a try
8720 * conditional, no autocommands are executed. If otherwise the autocommands
8721 * cause the script to be aborted, retval is set to FAIL.
8724 apply_autocmds_retval(event, fname, fname_io, force, buf, retval)
8725 event_T event;
8726 char_u *fname; /* NULL or empty means use actual file name */
8727 char_u *fname_io; /* fname to use for <afile> on cmdline */
8728 int force; /* when TRUE, ignore autocmd_busy */
8729 buf_T *buf; /* buffer for <abuf> */
8730 int *retval; /* pointer to caller's retval */
8732 int did_cmd;
8734 #ifdef FEAT_EVAL
8735 if (should_abort(*retval))
8736 return FALSE;
8737 #endif
8739 did_cmd = apply_autocmds_group(event, fname, fname_io, force,
8740 AUGROUP_ALL, buf, NULL);
8741 if (did_cmd
8742 #ifdef FEAT_EVAL
8743 && aborting()
8744 #endif
8746 *retval = FAIL;
8747 return did_cmd;
8751 * Return TRUE when there is a CursorHold autocommand defined.
8754 has_cursorhold()
8756 return (first_autopat[(int)(get_real_state() == NORMAL_BUSY
8757 ? EVENT_CURSORHOLD : EVENT_CURSORHOLDI)] != NULL);
8761 * Return TRUE if the CursorHold event can be triggered.
8764 trigger_cursorhold()
8766 int state;
8768 if (!did_cursorhold && has_cursorhold() && !Recording
8769 #ifdef FEAT_INS_EXPAND
8770 && !ins_compl_active()
8771 #endif
8774 state = get_real_state();
8775 if (state == NORMAL_BUSY || (state & INSERT) != 0)
8776 return TRUE;
8778 return FALSE;
8782 * Return TRUE when there is a CursorMoved autocommand defined.
8785 has_cursormoved()
8787 return (first_autopat[(int)EVENT_CURSORMOVED] != NULL);
8791 * Return TRUE when there is a CursorMovedI autocommand defined.
8794 has_cursormovedI()
8796 return (first_autopat[(int)EVENT_CURSORMOVEDI] != NULL);
8799 static int
8800 apply_autocmds_group(event, fname, fname_io, force, group, buf, eap)
8801 event_T event;
8802 char_u *fname; /* NULL or empty means use actual file name */
8803 char_u *fname_io; /* fname to use for <afile> on cmdline, NULL means
8804 use fname */
8805 int force; /* when TRUE, ignore autocmd_busy */
8806 int group; /* group ID, or AUGROUP_ALL */
8807 buf_T *buf; /* buffer for <abuf> */
8808 exarg_T *eap; /* command arguments */
8810 char_u *sfname = NULL; /* short file name */
8811 char_u *tail;
8812 int save_changed;
8813 buf_T *old_curbuf;
8814 int retval = FALSE;
8815 char_u *save_sourcing_name;
8816 linenr_T save_sourcing_lnum;
8817 char_u *save_autocmd_fname;
8818 int save_autocmd_fname_full;
8819 int save_autocmd_bufnr;
8820 char_u *save_autocmd_match;
8821 int save_autocmd_busy;
8822 int save_autocmd_nested;
8823 static int nesting = 0;
8824 AutoPatCmd patcmd;
8825 AutoPat *ap;
8826 #ifdef FEAT_EVAL
8827 scid_T save_current_SID;
8828 void *save_funccalp;
8829 char_u *save_cmdarg;
8830 long save_cmdbang;
8831 #endif
8832 static int filechangeshell_busy = FALSE;
8833 #ifdef FEAT_PROFILE
8834 proftime_T wait_time;
8835 #endif
8838 * Quickly return if there are no autocommands for this event or
8839 * autocommands are blocked.
8841 if (first_autopat[(int)event] == NULL || autocmd_blocked > 0)
8842 goto BYPASS_AU;
8845 * When autocommands are busy, new autocommands are only executed when
8846 * explicitly enabled with the "nested" flag.
8848 if (autocmd_busy && !(force || autocmd_nested))
8849 goto BYPASS_AU;
8851 #ifdef FEAT_EVAL
8853 * Quickly return when immediately aborting on error, or when an interrupt
8854 * occurred or an exception was thrown but not caught.
8856 if (aborting())
8857 goto BYPASS_AU;
8858 #endif
8861 * FileChangedShell never nests, because it can create an endless loop.
8863 if (filechangeshell_busy && (event == EVENT_FILECHANGEDSHELL
8864 || event == EVENT_FILECHANGEDSHELLPOST))
8865 goto BYPASS_AU;
8868 * Ignore events in 'eventignore'.
8870 if (event_ignored(event))
8871 goto BYPASS_AU;
8874 * Allow nesting of autocommands, but restrict the depth, because it's
8875 * possible to create an endless loop.
8877 if (nesting == 10)
8879 EMSG(_("E218: autocommand nesting too deep"));
8880 goto BYPASS_AU;
8884 * Check if these autocommands are disabled. Used when doing ":all" or
8885 * ":ball".
8887 if ( (autocmd_no_enter
8888 && (event == EVENT_WINENTER || event == EVENT_BUFENTER))
8889 || (autocmd_no_leave
8890 && (event == EVENT_WINLEAVE || event == EVENT_BUFLEAVE)))
8891 goto BYPASS_AU;
8894 * Save the autocmd_* variables and info about the current buffer.
8896 save_autocmd_fname = autocmd_fname;
8897 save_autocmd_fname_full = autocmd_fname_full;
8898 save_autocmd_bufnr = autocmd_bufnr;
8899 save_autocmd_match = autocmd_match;
8900 save_autocmd_busy = autocmd_busy;
8901 save_autocmd_nested = autocmd_nested;
8902 save_changed = curbuf->b_changed;
8903 old_curbuf = curbuf;
8906 * Set the file name to be used for <afile>.
8907 * Make a copy to avoid that changing a buffer name or directory makes it
8908 * invalid.
8910 if (fname_io == NULL)
8912 if (fname != NULL && *fname != NUL)
8913 autocmd_fname = fname;
8914 else if (buf != NULL)
8915 autocmd_fname = buf->b_ffname;
8916 else
8917 autocmd_fname = NULL;
8919 else
8920 autocmd_fname = fname_io;
8921 if (autocmd_fname != NULL)
8922 autocmd_fname = vim_strsave(autocmd_fname);
8923 autocmd_fname_full = FALSE; /* call FullName_save() later */
8926 * Set the buffer number to be used for <abuf>.
8928 if (buf == NULL)
8929 autocmd_bufnr = 0;
8930 else
8931 autocmd_bufnr = buf->b_fnum;
8934 * When the file name is NULL or empty, use the file name of buffer "buf".
8935 * Always use the full path of the file name to match with, in case
8936 * "allow_dirs" is set.
8938 if (fname == NULL || *fname == NUL)
8940 if (buf == NULL)
8941 fname = NULL;
8942 else
8944 #ifdef FEAT_SYN_HL
8945 if (event == EVENT_SYNTAX)
8946 fname = buf->b_p_syn;
8947 else
8948 #endif
8949 if (event == EVENT_FILETYPE)
8950 fname = buf->b_p_ft;
8951 else
8953 if (buf->b_sfname != NULL)
8954 sfname = vim_strsave(buf->b_sfname);
8955 fname = buf->b_ffname;
8958 if (fname == NULL)
8959 fname = (char_u *)"";
8960 fname = vim_strsave(fname); /* make a copy, so we can change it */
8962 else
8964 sfname = vim_strsave(fname);
8965 /* Don't try expanding FileType, Syntax, FuncUndefined, WindowID or
8966 * QuickFixCmd* */
8967 if (event == EVENT_FILETYPE
8968 || event == EVENT_SYNTAX
8969 || event == EVENT_FUNCUNDEFINED
8970 || event == EVENT_REMOTEREPLY
8971 || event == EVENT_SPELLFILEMISSING
8972 || event == EVENT_QUICKFIXCMDPRE
8973 || event == EVENT_QUICKFIXCMDPOST)
8974 fname = vim_strsave(fname);
8975 else
8976 fname = FullName_save(fname, FALSE);
8978 if (fname == NULL) /* out of memory */
8980 vim_free(sfname);
8981 retval = FALSE;
8982 goto BYPASS_AU;
8985 #ifdef BACKSLASH_IN_FILENAME
8987 * Replace all backslashes with forward slashes. This makes the
8988 * autocommand patterns portable between Unix and MS-DOS.
8990 if (sfname != NULL)
8991 forward_slash(sfname);
8992 forward_slash(fname);
8993 #endif
8995 #ifdef VMS
8996 /* remove version for correct match */
8997 if (sfname != NULL)
8998 vms_remove_version(sfname);
8999 vms_remove_version(fname);
9000 #endif
9003 * Set the name to be used for <amatch>.
9005 autocmd_match = fname;
9008 /* Don't redraw while doing auto commands. */
9009 ++RedrawingDisabled;
9010 save_sourcing_name = sourcing_name;
9011 sourcing_name = NULL; /* don't free this one */
9012 save_sourcing_lnum = sourcing_lnum;
9013 sourcing_lnum = 0; /* no line number here */
9015 #ifdef FEAT_EVAL
9016 save_current_SID = current_SID;
9018 # ifdef FEAT_PROFILE
9019 if (do_profiling == PROF_YES)
9020 prof_child_enter(&wait_time); /* doesn't count for the caller itself */
9021 # endif
9023 /* Don't use local function variables, if called from a function */
9024 save_funccalp = save_funccal();
9025 #endif
9028 * When starting to execute autocommands, save the search patterns.
9030 if (!autocmd_busy)
9032 save_search_patterns();
9033 saveRedobuff();
9034 did_filetype = keep_filetype;
9038 * Note that we are applying autocmds. Some commands need to know.
9040 autocmd_busy = TRUE;
9041 filechangeshell_busy = (event == EVENT_FILECHANGEDSHELL);
9042 ++nesting; /* see matching decrement below */
9044 /* Remember that FileType was triggered. Used for did_filetype(). */
9045 if (event == EVENT_FILETYPE)
9046 did_filetype = TRUE;
9048 tail = gettail(fname);
9050 /* Find first autocommand that matches */
9051 patcmd.curpat = first_autopat[(int)event];
9052 patcmd.nextcmd = NULL;
9053 patcmd.group = group;
9054 patcmd.fname = fname;
9055 patcmd.sfname = sfname;
9056 patcmd.tail = tail;
9057 patcmd.event = event;
9058 patcmd.arg_bufnr = autocmd_bufnr;
9059 patcmd.next = NULL;
9060 auto_next_pat(&patcmd, FALSE);
9062 /* found one, start executing the autocommands */
9063 if (patcmd.curpat != NULL)
9065 /* add to active_apc_list */
9066 patcmd.next = active_apc_list;
9067 active_apc_list = &patcmd;
9069 #ifdef FEAT_EVAL
9070 /* set v:cmdarg (only when there is a matching pattern) */
9071 save_cmdbang = get_vim_var_nr(VV_CMDBANG);
9072 if (eap != NULL)
9074 save_cmdarg = set_cmdarg(eap, NULL);
9075 set_vim_var_nr(VV_CMDBANG, (long)eap->forceit);
9077 else
9078 save_cmdarg = NULL; /* avoid gcc warning */
9079 #endif
9080 retval = TRUE;
9081 /* mark the last pattern, to avoid an endless loop when more patterns
9082 * are added when executing autocommands */
9083 for (ap = patcmd.curpat; ap->next != NULL; ap = ap->next)
9084 ap->last = FALSE;
9085 ap->last = TRUE;
9086 check_lnums(TRUE); /* make sure cursor and topline are valid */
9087 do_cmdline(NULL, getnextac, (void *)&patcmd,
9088 DOCMD_NOWAIT|DOCMD_VERBOSE|DOCMD_REPEAT);
9089 #ifdef FEAT_EVAL
9090 if (eap != NULL)
9092 (void)set_cmdarg(NULL, save_cmdarg);
9093 set_vim_var_nr(VV_CMDBANG, save_cmdbang);
9095 #endif
9096 /* delete from active_apc_list */
9097 if (active_apc_list == &patcmd) /* just in case */
9098 active_apc_list = patcmd.next;
9101 --RedrawingDisabled;
9102 autocmd_busy = save_autocmd_busy;
9103 filechangeshell_busy = FALSE;
9104 autocmd_nested = save_autocmd_nested;
9105 vim_free(sourcing_name);
9106 sourcing_name = save_sourcing_name;
9107 sourcing_lnum = save_sourcing_lnum;
9108 vim_free(autocmd_fname);
9109 autocmd_fname = save_autocmd_fname;
9110 autocmd_fname_full = save_autocmd_fname_full;
9111 autocmd_bufnr = save_autocmd_bufnr;
9112 autocmd_match = save_autocmd_match;
9113 #ifdef FEAT_EVAL
9114 current_SID = save_current_SID;
9115 restore_funccal(save_funccalp);
9116 # ifdef FEAT_PROFILE
9117 if (do_profiling == PROF_YES)
9118 prof_child_exit(&wait_time);
9119 # endif
9120 #endif
9121 vim_free(fname);
9122 vim_free(sfname);
9123 --nesting; /* see matching increment above */
9126 * When stopping to execute autocommands, restore the search patterns and
9127 * the redo buffer.
9129 if (!autocmd_busy)
9131 restore_search_patterns();
9132 restoreRedobuff();
9133 did_filetype = FALSE;
9137 * Some events don't set or reset the Changed flag.
9138 * Check if still in the same buffer!
9140 if (curbuf == old_curbuf
9141 && (event == EVENT_BUFREADPOST
9142 || event == EVENT_BUFWRITEPOST
9143 || event == EVENT_FILEAPPENDPOST
9144 || event == EVENT_VIMLEAVE
9145 || event == EVENT_VIMLEAVEPRE))
9147 #ifdef FEAT_TITLE
9148 if (curbuf->b_changed != save_changed)
9149 need_maketitle = TRUE;
9150 #endif
9151 curbuf->b_changed = save_changed;
9154 au_cleanup(); /* may really delete removed patterns/commands now */
9156 BYPASS_AU:
9157 /* When wiping out a buffer make sure all its buffer-local autocommands
9158 * are deleted. */
9159 if (event == EVENT_BUFWIPEOUT && buf != NULL)
9160 aubuflocal_remove(buf);
9162 return retval;
9165 # ifdef FEAT_EVAL
9166 static char_u *old_termresponse = NULL;
9167 # endif
9170 * Block triggering autocommands until unblock_autocmd() is called.
9171 * Can be used recursively, so long as it's symmetric.
9173 void
9174 block_autocmds()
9176 # ifdef FEAT_EVAL
9177 /* Remember the value of v:termresponse. */
9178 if (autocmd_blocked == 0)
9179 old_termresponse = get_vim_var_str(VV_TERMRESPONSE);
9180 # endif
9181 ++autocmd_blocked;
9184 void
9185 unblock_autocmds()
9187 --autocmd_blocked;
9189 # ifdef FEAT_EVAL
9190 /* When v:termresponse was set while autocommands were blocked, trigger
9191 * the autocommands now. Esp. useful when executing a shell command
9192 * during startup (vimdiff). */
9193 if (autocmd_blocked == 0
9194 && get_vim_var_str(VV_TERMRESPONSE) != old_termresponse)
9195 apply_autocmds(EVENT_TERMRESPONSE, NULL, NULL, FALSE, curbuf);
9196 # endif
9200 * Find next autocommand pattern that matches.
9202 static void
9203 auto_next_pat(apc, stop_at_last)
9204 AutoPatCmd *apc;
9205 int stop_at_last; /* stop when 'last' flag is set */
9207 AutoPat *ap;
9208 AutoCmd *cp;
9209 char_u *name;
9210 char *s;
9212 vim_free(sourcing_name);
9213 sourcing_name = NULL;
9215 for (ap = apc->curpat; ap != NULL && !got_int; ap = ap->next)
9217 apc->curpat = NULL;
9219 /* Only use a pattern when it has not been removed, has commands and
9220 * the group matches. For buffer-local autocommands only check the
9221 * buffer number. */
9222 if (ap->pat != NULL && ap->cmds != NULL
9223 && (apc->group == AUGROUP_ALL || apc->group == ap->group))
9225 /* execution-condition */
9226 if (ap->buflocal_nr == 0
9227 ? (match_file_pat(NULL, ap->reg_prog, apc->fname,
9228 apc->sfname, apc->tail, ap->allow_dirs))
9229 : ap->buflocal_nr == apc->arg_bufnr)
9231 name = event_nr2name(apc->event);
9232 s = _("%s Auto commands for \"%s\"");
9233 sourcing_name = alloc((unsigned)(STRLEN(s)
9234 + STRLEN(name) + ap->patlen + 1));
9235 if (sourcing_name != NULL)
9237 sprintf((char *)sourcing_name, s,
9238 (char *)name, (char *)ap->pat);
9239 if (p_verbose >= 8)
9241 verbose_enter();
9242 smsg((char_u *)_("Executing %s"), sourcing_name);
9243 verbose_leave();
9247 apc->curpat = ap;
9248 apc->nextcmd = ap->cmds;
9249 /* mark last command */
9250 for (cp = ap->cmds; cp->next != NULL; cp = cp->next)
9251 cp->last = FALSE;
9252 cp->last = TRUE;
9254 line_breakcheck();
9255 if (apc->curpat != NULL) /* found a match */
9256 break;
9258 if (stop_at_last && ap->last)
9259 break;
9264 * Get next autocommand command.
9265 * Called by do_cmdline() to get the next line for ":if".
9266 * Returns allocated string, or NULL for end of autocommands.
9268 static char_u *
9269 getnextac(c, cookie, indent)
9270 int c UNUSED;
9271 void *cookie;
9272 int indent UNUSED;
9274 AutoPatCmd *acp = (AutoPatCmd *)cookie;
9275 char_u *retval;
9276 AutoCmd *ac;
9278 /* Can be called again after returning the last line. */
9279 if (acp->curpat == NULL)
9280 return NULL;
9282 /* repeat until we find an autocommand to execute */
9283 for (;;)
9285 /* skip removed commands */
9286 while (acp->nextcmd != NULL && acp->nextcmd->cmd == NULL)
9287 if (acp->nextcmd->last)
9288 acp->nextcmd = NULL;
9289 else
9290 acp->nextcmd = acp->nextcmd->next;
9292 if (acp->nextcmd != NULL)
9293 break;
9295 /* at end of commands, find next pattern that matches */
9296 if (acp->curpat->last)
9297 acp->curpat = NULL;
9298 else
9299 acp->curpat = acp->curpat->next;
9300 if (acp->curpat != NULL)
9301 auto_next_pat(acp, TRUE);
9302 if (acp->curpat == NULL)
9303 return NULL;
9306 ac = acp->nextcmd;
9308 if (p_verbose >= 9)
9310 verbose_enter_scroll();
9311 smsg((char_u *)_("autocommand %s"), ac->cmd);
9312 msg_puts((char_u *)"\n"); /* don't overwrite this either */
9313 verbose_leave_scroll();
9315 retval = vim_strsave(ac->cmd);
9316 autocmd_nested = ac->nested;
9317 #ifdef FEAT_EVAL
9318 current_SID = ac->scriptID;
9319 #endif
9320 if (ac->last)
9321 acp->nextcmd = NULL;
9322 else
9323 acp->nextcmd = ac->next;
9324 return retval;
9328 * Return TRUE if there is a matching autocommand for "fname".
9329 * To account for buffer-local autocommands, function needs to know
9330 * in which buffer the file will be opened.
9333 has_autocmd(event, sfname, buf)
9334 event_T event;
9335 char_u *sfname;
9336 buf_T *buf;
9338 AutoPat *ap;
9339 char_u *fname;
9340 char_u *tail = gettail(sfname);
9341 int retval = FALSE;
9343 fname = FullName_save(sfname, FALSE);
9344 if (fname == NULL)
9345 return FALSE;
9347 #ifdef BACKSLASH_IN_FILENAME
9349 * Replace all backslashes with forward slashes. This makes the
9350 * autocommand patterns portable between Unix and MS-DOS.
9352 sfname = vim_strsave(sfname);
9353 if (sfname != NULL)
9354 forward_slash(sfname);
9355 forward_slash(fname);
9356 #endif
9358 for (ap = first_autopat[(int)event]; ap != NULL; ap = ap->next)
9359 if (ap->pat != NULL && ap->cmds != NULL
9360 && (ap->buflocal_nr == 0
9361 ? match_file_pat(NULL, ap->reg_prog,
9362 fname, sfname, tail, ap->allow_dirs)
9363 : buf != NULL && ap->buflocal_nr == buf->b_fnum
9366 retval = TRUE;
9367 break;
9370 vim_free(fname);
9371 #ifdef BACKSLASH_IN_FILENAME
9372 vim_free(sfname);
9373 #endif
9375 return retval;
9378 #if defined(FEAT_CMDL_COMPL) || defined(PROTO)
9380 * Function given to ExpandGeneric() to obtain the list of autocommand group
9381 * names.
9383 char_u *
9384 get_augroup_name(xp, idx)
9385 expand_T *xp UNUSED;
9386 int idx;
9388 if (idx == augroups.ga_len) /* add "END" add the end */
9389 return (char_u *)"END";
9390 if (idx >= augroups.ga_len) /* end of list */
9391 return NULL;
9392 if (AUGROUP_NAME(idx) == NULL) /* skip deleted entries */
9393 return (char_u *)"";
9394 return AUGROUP_NAME(idx); /* return a name */
9397 static int include_groups = FALSE;
9399 char_u *
9400 set_context_in_autocmd(xp, arg, doautocmd)
9401 expand_T *xp;
9402 char_u *arg;
9403 int doautocmd; /* TRUE for :doauto*, FALSE for :autocmd */
9405 char_u *p;
9406 int group;
9408 /* check for a group name, skip it if present */
9409 include_groups = FALSE;
9410 p = arg;
9411 group = au_get_grouparg(&arg);
9412 if (group == AUGROUP_ERROR)
9413 return NULL;
9414 /* If there only is a group name that's what we expand. */
9415 if (*arg == NUL && group != AUGROUP_ALL && !vim_iswhite(arg[-1]))
9417 arg = p;
9418 group = AUGROUP_ALL;
9421 /* skip over event name */
9422 for (p = arg; *p != NUL && !vim_iswhite(*p); ++p)
9423 if (*p == ',')
9424 arg = p + 1;
9425 if (*p == NUL)
9427 if (group == AUGROUP_ALL)
9428 include_groups = TRUE;
9429 xp->xp_context = EXPAND_EVENTS; /* expand event name */
9430 xp->xp_pattern = arg;
9431 return NULL;
9434 /* skip over pattern */
9435 arg = skipwhite(p);
9436 while (*arg && (!vim_iswhite(*arg) || arg[-1] == '\\'))
9437 arg++;
9438 if (*arg)
9439 return arg; /* expand (next) command */
9441 if (doautocmd)
9442 xp->xp_context = EXPAND_FILES; /* expand file names */
9443 else
9444 xp->xp_context = EXPAND_NOTHING; /* pattern is not expanded */
9445 return NULL;
9449 * Function given to ExpandGeneric() to obtain the list of event names.
9451 char_u *
9452 get_event_name(xp, idx)
9453 expand_T *xp UNUSED;
9454 int idx;
9456 if (idx < augroups.ga_len) /* First list group names, if wanted */
9458 if (!include_groups || AUGROUP_NAME(idx) == NULL)
9459 return (char_u *)""; /* skip deleted entries */
9460 return AUGROUP_NAME(idx); /* return a name */
9462 return (char_u *)event_names[idx - augroups.ga_len].name;
9465 #endif /* FEAT_CMDL_COMPL */
9468 * Return TRUE if autocmd is supported.
9471 autocmd_supported(name)
9472 char_u *name;
9474 char_u *p;
9476 return (event_name2nr(name, &p) != NUM_EVENTS);
9480 * Return TRUE if an autocommand is defined for a group, event and
9481 * pattern: The group can be omitted to accept any group. "event" and "pattern"
9482 * can be NULL to accept any event and pattern. "pattern" can be NULL to accept
9483 * any pattern. Buffer-local patterns <buffer> or <buffer=N> are accepted.
9484 * Used for:
9485 * exists("#Group") or
9486 * exists("#Group#Event") or
9487 * exists("#Group#Event#pat") or
9488 * exists("#Event") or
9489 * exists("#Event#pat")
9492 au_exists(arg)
9493 char_u *arg;
9495 char_u *arg_save;
9496 char_u *pattern = NULL;
9497 char_u *event_name;
9498 char_u *p;
9499 event_T event;
9500 AutoPat *ap;
9501 buf_T *buflocal_buf = NULL;
9502 int group;
9503 int retval = FALSE;
9505 /* Make a copy so that we can change the '#' chars to a NUL. */
9506 arg_save = vim_strsave(arg);
9507 if (arg_save == NULL)
9508 return FALSE;
9509 p = vim_strchr(arg_save, '#');
9510 if (p != NULL)
9511 *p++ = NUL;
9513 /* First, look for an autocmd group name */
9514 group = au_find_group(arg_save);
9515 if (group == AUGROUP_ERROR)
9517 /* Didn't match a group name, assume the first argument is an event. */
9518 group = AUGROUP_ALL;
9519 event_name = arg_save;
9521 else
9523 if (p == NULL)
9525 /* "Group": group name is present and it's recognized */
9526 retval = TRUE;
9527 goto theend;
9530 /* Must be "Group#Event" or "Group#Event#pat". */
9531 event_name = p;
9532 p = vim_strchr(event_name, '#');
9533 if (p != NULL)
9534 *p++ = NUL; /* "Group#Event#pat" */
9537 pattern = p; /* "pattern" is NULL when there is no pattern */
9539 /* find the index (enum) for the event name */
9540 event = event_name2nr(event_name, &p);
9542 /* return FALSE if the event name is not recognized */
9543 if (event == NUM_EVENTS)
9544 goto theend;
9546 /* Find the first autocommand for this event.
9547 * If there isn't any, return FALSE;
9548 * If there is one and no pattern given, return TRUE; */
9549 ap = first_autopat[(int)event];
9550 if (ap == NULL)
9551 goto theend;
9553 /* if pattern is "<buffer>", special handling is needed which uses curbuf */
9554 /* for pattern "<buffer=N>, fnamecmp() will work fine */
9555 if (pattern != NULL && STRICMP(pattern, "<buffer>") == 0)
9556 buflocal_buf = curbuf;
9558 /* Check if there is an autocommand with the given pattern. */
9559 for ( ; ap != NULL; ap = ap->next)
9560 /* only use a pattern when it has not been removed and has commands. */
9561 /* For buffer-local autocommands, fnamecmp() works fine. */
9562 if (ap->pat != NULL && ap->cmds != NULL
9563 && (group == AUGROUP_ALL || ap->group == group)
9564 && (pattern == NULL
9565 || (buflocal_buf == NULL
9566 ? fnamecmp(ap->pat, pattern) == 0
9567 : ap->buflocal_nr == buflocal_buf->b_fnum)))
9569 retval = TRUE;
9570 break;
9573 theend:
9574 vim_free(arg_save);
9575 return retval;
9578 #else /* FEAT_AUTOCMD */
9581 * Prepare for executing commands for (hidden) buffer "buf".
9582 * This is the non-autocommand version, it simply saves "curbuf" and sets
9583 * "curbuf" and "curwin" to match "buf".
9585 void
9586 aucmd_prepbuf(aco, buf)
9587 aco_save_T *aco; /* structure to save values in */
9588 buf_T *buf; /* new curbuf */
9590 aco->save_curbuf = curbuf;
9591 --curbuf->b_nwindows;
9592 curbuf = buf;
9593 curwin->w_buffer = buf;
9594 ++curbuf->b_nwindows;
9598 * Restore after executing commands for a (hidden) buffer.
9599 * This is the non-autocommand version.
9601 void
9602 aucmd_restbuf(aco)
9603 aco_save_T *aco; /* structure holding saved values */
9605 --curbuf->b_nwindows;
9606 curbuf = aco->save_curbuf;
9607 curwin->w_buffer = curbuf;
9608 ++curbuf->b_nwindows;
9611 #endif /* FEAT_AUTOCMD */
9614 #if defined(FEAT_AUTOCMD) || defined(FEAT_WILDIGN) || defined(PROTO)
9616 * Try matching a filename with a "pattern" ("prog" is NULL), or use the
9617 * precompiled regprog "prog" ("pattern" is NULL). That avoids calling
9618 * vim_regcomp() often.
9619 * Used for autocommands and 'wildignore'.
9620 * Returns TRUE if there is a match, FALSE otherwise.
9623 match_file_pat(pattern, prog, fname, sfname, tail, allow_dirs)
9624 char_u *pattern; /* pattern to match with */
9625 regprog_T *prog; /* pre-compiled regprog or NULL */
9626 char_u *fname; /* full path of file name */
9627 char_u *sfname; /* short file name or NULL */
9628 char_u *tail; /* tail of path */
9629 int allow_dirs; /* allow matching with dir */
9631 regmatch_T regmatch;
9632 int result = FALSE;
9633 #ifdef FEAT_OSFILETYPE
9634 int no_pattern = FALSE; /* TRUE if check is filetype only */
9635 char_u *type_start;
9636 char_u c;
9637 int match = FALSE;
9638 #endif
9640 #ifdef CASE_INSENSITIVE_FILENAME
9641 regmatch.rm_ic = TRUE; /* Always ignore case */
9642 #else
9643 regmatch.rm_ic = FALSE; /* Don't ever ignore case */
9644 #endif
9645 #ifdef FEAT_OSFILETYPE
9646 if (*pattern == '<')
9648 /* There is a filetype condition specified with this pattern.
9649 * Check the filetype matches first. If not, don't bother with the
9650 * pattern (set regprog to NULL).
9651 * Always use magic for the regexp.
9654 for (type_start = pattern + 1; (c = *pattern); pattern++)
9656 if ((c == ';' || c == '>') && match == FALSE)
9658 *pattern = NUL; /* Terminate the string */
9659 match = mch_check_filetype(fname, type_start);
9660 *pattern = c; /* Restore the terminator */
9661 type_start = pattern + 1;
9663 if (c == '>')
9664 break;
9667 /* (c should never be NUL, but check anyway) */
9668 if (match == FALSE || c == NUL)
9669 regmatch.regprog = NULL; /* Doesn't match - don't check pat. */
9670 else if (*pattern == NUL)
9672 regmatch.regprog = NULL; /* Vim will try to free regprog later */
9673 no_pattern = TRUE; /* Always matches - don't check pat. */
9675 else
9676 regmatch.regprog = vim_regcomp(pattern + 1, RE_MAGIC);
9678 else
9679 #endif
9681 if (prog != NULL)
9682 regmatch.regprog = prog;
9683 else
9684 regmatch.regprog = vim_regcomp(pattern, RE_MAGIC);
9688 * Try for a match with the pattern with:
9689 * 1. the full file name, when the pattern has a '/'.
9690 * 2. the short file name, when the pattern has a '/'.
9691 * 3. the tail of the file name, when the pattern has no '/'.
9693 if (
9694 #ifdef FEAT_OSFILETYPE
9695 /* If the check is for a filetype only and we don't care
9696 * about the path then skip all the regexp stuff.
9698 no_pattern ||
9699 #endif
9700 (regmatch.regprog != NULL
9701 && ((allow_dirs
9702 && (vim_regexec(&regmatch, fname, (colnr_T)0)
9703 || (sfname != NULL
9704 && vim_regexec(&regmatch, sfname, (colnr_T)0))))
9705 || (!allow_dirs && vim_regexec(&regmatch, tail, (colnr_T)0)))))
9706 result = TRUE;
9708 if (prog == NULL)
9709 vim_free(regmatch.regprog);
9710 return result;
9712 #endif
9714 #if defined(FEAT_WILDIGN) || defined(PROTO)
9716 * Return TRUE if a file matches with a pattern in "list".
9717 * "list" is a comma-separated list of patterns, like 'wildignore'.
9718 * "sfname" is the short file name or NULL, "ffname" the long file name.
9721 match_file_list(list, sfname, ffname)
9722 char_u *list;
9723 char_u *sfname;
9724 char_u *ffname;
9726 char_u buf[100];
9727 char_u *tail;
9728 char_u *regpat;
9729 char allow_dirs;
9730 int match;
9731 char_u *p;
9733 tail = gettail(sfname);
9735 /* try all patterns in 'wildignore' */
9736 p = list;
9737 while (*p)
9739 copy_option_part(&p, buf, 100, ",");
9740 regpat = file_pat_to_reg_pat(buf, NULL, &allow_dirs, FALSE);
9741 if (regpat == NULL)
9742 break;
9743 match = match_file_pat(regpat, NULL, ffname, sfname,
9744 tail, (int)allow_dirs);
9745 vim_free(regpat);
9746 if (match)
9747 return TRUE;
9749 return FALSE;
9751 #endif
9754 * Convert the given pattern "pat" which has shell style wildcards in it, into
9755 * a regular expression, and return the result in allocated memory. If there
9756 * is a directory path separator to be matched, then TRUE is put in
9757 * allow_dirs, otherwise FALSE is put there -- webb.
9758 * Handle backslashes before special characters, like "\*" and "\ ".
9760 * If FEAT_OSFILETYPE defined then pass initial <type> through unchanged. Eg:
9761 * '<html>myfile' becomes '<html>^myfile$' -- leonard.
9763 * Returns NULL when out of memory.
9765 char_u *
9766 file_pat_to_reg_pat(pat, pat_end, allow_dirs, no_bslash)
9767 char_u *pat;
9768 char_u *pat_end; /* first char after pattern or NULL */
9769 char *allow_dirs; /* Result passed back out in here */
9770 int no_bslash UNUSED; /* Don't use a backward slash as pathsep */
9772 int size;
9773 char_u *endp;
9774 char_u *reg_pat;
9775 char_u *p;
9776 int i;
9777 int nested = 0;
9778 int add_dollar = TRUE;
9779 #ifdef FEAT_OSFILETYPE
9780 int check_length = 0;
9781 #endif
9783 if (allow_dirs != NULL)
9784 *allow_dirs = FALSE;
9785 if (pat_end == NULL)
9786 pat_end = pat + STRLEN(pat);
9788 #ifdef FEAT_OSFILETYPE
9789 /* Find out how much of the string is the filetype check */
9790 if (*pat == '<')
9792 /* Count chars until the next '>' */
9793 for (p = pat + 1; p < pat_end && *p != '>'; p++)
9795 if (p < pat_end)
9797 /* Pattern is of the form <.*>.* */
9798 check_length = p - pat + 1;
9799 if (p + 1 >= pat_end)
9801 /* The 'pattern' is a filetype check ONLY */
9802 reg_pat = (char_u *)alloc(check_length + 1);
9803 if (reg_pat != NULL)
9805 mch_memmove(reg_pat, pat, (size_t)check_length);
9806 reg_pat[check_length] = NUL;
9808 return reg_pat;
9811 /* else: there was no closing '>' - assume it was a normal pattern */
9814 pat += check_length;
9815 size = 2 + check_length;
9816 #else
9817 size = 2; /* '^' at start, '$' at end */
9818 #endif
9820 for (p = pat; p < pat_end; p++)
9822 switch (*p)
9824 case '*':
9825 case '.':
9826 case ',':
9827 case '{':
9828 case '}':
9829 case '~':
9830 size += 2; /* extra backslash */
9831 break;
9832 #ifdef BACKSLASH_IN_FILENAME
9833 case '\\':
9834 case '/':
9835 size += 4; /* could become "[\/]" */
9836 break;
9837 #endif
9838 default:
9839 size++;
9840 # ifdef FEAT_MBYTE
9841 if (enc_dbcs != 0 && (*mb_ptr2len)(p) > 1)
9843 ++p;
9844 ++size;
9846 # endif
9847 break;
9850 reg_pat = alloc(size + 1);
9851 if (reg_pat == NULL)
9852 return NULL;
9854 #ifdef FEAT_OSFILETYPE
9855 /* Copy the type check in to the start. */
9856 if (check_length)
9857 mch_memmove(reg_pat, pat - check_length, (size_t)check_length);
9858 i = check_length;
9859 #else
9860 i = 0;
9861 #endif
9863 if (pat[0] == '*')
9864 while (pat[0] == '*' && pat < pat_end - 1)
9865 pat++;
9866 else
9867 reg_pat[i++] = '^';
9868 endp = pat_end - 1;
9869 if (*endp == '*')
9871 while (endp - pat > 0 && *endp == '*')
9872 endp--;
9873 add_dollar = FALSE;
9875 for (p = pat; *p && nested >= 0 && p <= endp; p++)
9877 switch (*p)
9879 case '*':
9880 reg_pat[i++] = '.';
9881 reg_pat[i++] = '*';
9882 while (p[1] == '*') /* "**" matches like "*" */
9883 ++p;
9884 break;
9885 case '.':
9886 #ifdef RISCOS
9887 if (allow_dirs != NULL)
9888 *allow_dirs = TRUE;
9889 /* FALLTHROUGH */
9890 #endif
9891 case '~':
9892 reg_pat[i++] = '\\';
9893 reg_pat[i++] = *p;
9894 break;
9895 case '?':
9896 #ifdef RISCOS
9897 case '#':
9898 #endif
9899 reg_pat[i++] = '.';
9900 break;
9901 case '\\':
9902 if (p[1] == NUL)
9903 break;
9904 #ifdef BACKSLASH_IN_FILENAME
9905 if (!no_bslash)
9907 /* translate:
9908 * "\x" to "\\x" e.g., "dir\file"
9909 * "\*" to "\\.*" e.g., "dir\*.c"
9910 * "\?" to "\\." e.g., "dir\??.c"
9911 * "\+" to "\+" e.g., "fileX\+.c"
9913 if ((vim_isfilec(p[1]) || p[1] == '*' || p[1] == '?')
9914 && p[1] != '+')
9916 reg_pat[i++] = '[';
9917 reg_pat[i++] = '\\';
9918 reg_pat[i++] = '/';
9919 reg_pat[i++] = ']';
9920 if (allow_dirs != NULL)
9921 *allow_dirs = TRUE;
9922 break;
9925 #endif
9926 if (*++p == '?'
9927 #ifdef BACKSLASH_IN_FILENAME
9928 && no_bslash
9929 #endif
9931 reg_pat[i++] = '?';
9932 else
9933 if (*p == ',')
9934 reg_pat[i++] = ',';
9935 else
9937 if (allow_dirs != NULL && vim_ispathsep(*p)
9938 #ifdef BACKSLASH_IN_FILENAME
9939 && (!no_bslash || *p != '\\')
9940 #endif
9942 *allow_dirs = TRUE;
9943 reg_pat[i++] = '\\';
9944 reg_pat[i++] = *p;
9946 break;
9947 #ifdef BACKSLASH_IN_FILENAME
9948 case '/':
9949 reg_pat[i++] = '[';
9950 reg_pat[i++] = '\\';
9951 reg_pat[i++] = '/';
9952 reg_pat[i++] = ']';
9953 if (allow_dirs != NULL)
9954 *allow_dirs = TRUE;
9955 break;
9956 #endif
9957 case '{':
9958 reg_pat[i++] = '\\';
9959 reg_pat[i++] = '(';
9960 nested++;
9961 break;
9962 case '}':
9963 reg_pat[i++] = '\\';
9964 reg_pat[i++] = ')';
9965 --nested;
9966 break;
9967 case ',':
9968 if (nested)
9970 reg_pat[i++] = '\\';
9971 reg_pat[i++] = '|';
9973 else
9974 reg_pat[i++] = ',';
9975 break;
9976 default:
9977 # ifdef FEAT_MBYTE
9978 if (enc_dbcs != 0 && (*mb_ptr2len)(p) > 1)
9979 reg_pat[i++] = *p++;
9980 else
9981 # endif
9982 if (allow_dirs != NULL && vim_ispathsep(*p))
9983 *allow_dirs = TRUE;
9984 reg_pat[i++] = *p;
9985 break;
9988 if (add_dollar)
9989 reg_pat[i++] = '$';
9990 reg_pat[i] = NUL;
9991 if (nested != 0)
9993 if (nested < 0)
9994 EMSG(_("E219: Missing {."));
9995 else
9996 EMSG(_("E220: Missing }."));
9997 vim_free(reg_pat);
9998 reg_pat = NULL;
10000 return reg_pat;