blob: 22a735f5597aa2e4b71427ed8d170a031ed30333 [file] [log] [blame]
Bram Moolenaaredf3f972016-08-29 22:49:24 +02001/* vi:set ts=8 sts=4 sw=4 noet:
Bram Moolenaar071d4272004-06-13 20:20:40 +00002 *
3 * VIM - Vi IMproved by Bram Moolenaar
4 *
5 * Do ":help uganda" in Vim to read copying and usage conditions.
6 * Do ":help credits" in Vim to see a list of people who contributed.
7 * See README.txt for an overview of the Vim source code.
8 */
9
10/*
11 * fileio.c: read from and write to a file
12 */
13
Bram Moolenaar071d4272004-06-13 20:20:40 +000014#include "vim.h"
15
Bram Moolenaarf4888d02009-12-02 12:31:27 +000016#if defined(__TANDEM) || defined(__MINT__)
Bram Moolenaar071d4272004-06-13 20:20:40 +000017# include <limits.h> /* for SSIZE_MAX */
18#endif
19
20#if defined(HAVE_UTIME) && defined(HAVE_UTIME_H)
21# include <utime.h> /* for struct utimbuf */
22#endif
23
24#define BUFSIZE 8192 /* size of normal write buffer */
25#define SMBUFSIZE 256 /* size of emergency write buffer */
26
Bram Moolenaar071d4272004-06-13 20:20:40 +000027/* Is there any system that doesn't have access()? */
Bram Moolenaar9372a112005-12-06 19:59:18 +000028#define USE_MCH_ACCESS
Bram Moolenaar071d4272004-06-13 20:20:40 +000029
Bram Moolenaard25c16e2016-01-29 22:13:30 +010030static char_u *next_fenc(char_u **pp);
Bram Moolenaar13505972019-01-24 15:04:48 +010031#ifdef FEAT_EVAL
Bram Moolenaard25c16e2016-01-29 22:13:30 +010032static char_u *readfile_charconvert(char_u *fname, char_u *fenc, int *fdp);
Bram Moolenaar071d4272004-06-13 20:20:40 +000033#endif
34#ifdef FEAT_VIMINFO
Bram Moolenaard25c16e2016-01-29 22:13:30 +010035static void check_marks_read(void);
Bram Moolenaar071d4272004-06-13 20:20:40 +000036#endif
37#ifdef FEAT_CRYPT
Bram Moolenaar8767f522016-07-01 17:17:39 +020038static char_u *check_for_cryptkey(char_u *cryptkey, char_u *ptr, long *sizep, off_T *filesizep, int newfile, char_u *fname, int *did_ask);
Bram Moolenaar071d4272004-06-13 20:20:40 +000039#endif
Bram Moolenaard25c16e2016-01-29 22:13:30 +010040static int set_rw_fname(char_u *fname, char_u *sfname);
41static int msg_add_fileformat(int eol_type);
42static void msg_add_eol(void);
Bram Moolenaar8767f522016-07-01 17:17:39 +020043static int check_mtime(buf_T *buf, stat_T *s);
Bram Moolenaard25c16e2016-01-29 22:13:30 +010044static int time_differs(long t1, long t2);
Bram Moolenaar071d4272004-06-13 20:20:40 +000045
Bram Moolenaar13505972019-01-24 15:04:48 +010046#define HAS_BW_FLAGS
47#define FIO_LATIN1 0x01 /* convert Latin1 */
48#define FIO_UTF8 0x02 /* convert UTF-8 */
49#define FIO_UCS2 0x04 /* convert UCS-2 */
50#define FIO_UCS4 0x08 /* convert UCS-4 */
51#define FIO_UTF16 0x10 /* convert UTF-16 */
Bram Moolenaar4f974752019-02-17 17:44:42 +010052#ifdef MSWIN
Bram Moolenaar13505972019-01-24 15:04:48 +010053# define FIO_CODEPAGE 0x20 /* convert MS-Windows codepage */
54# define FIO_PUT_CP(x) (((x) & 0xffff) << 16) /* put codepage in top word */
55# define FIO_GET_CP(x) (((x)>>16) & 0xffff) /* get codepage from top word */
Bram Moolenaar071d4272004-06-13 20:20:40 +000056#endif
Bram Moolenaar13505972019-01-24 15:04:48 +010057#ifdef MACOS_CONVERT
58# define FIO_MACROMAN 0x20 /* convert MacRoman */
59#endif
60#define FIO_ENDIAN_L 0x80 /* little endian */
61#define FIO_ENCRYPTED 0x1000 /* encrypt written bytes */
62#define FIO_NOCONVERT 0x2000 /* skip encoding conversion */
63#define FIO_UCSBOM 0x4000 /* check for BOM at start of file */
64#define FIO_ALL -1 /* allow all formats */
Bram Moolenaar071d4272004-06-13 20:20:40 +000065
66/* When converting, a read() or write() may leave some bytes to be converted
67 * for the next call. The value is guessed... */
68#define CONV_RESTLEN 30
69
70/* We have to guess how much a sequence of bytes may expand when converting
71 * with iconv() to be able to allocate a buffer. */
72#define ICONV_MULT 8
73
74/*
75 * Structure to pass arguments from buf_write() to buf_write_bytes().
76 */
77struct bw_info
78{
79 int bw_fd; /* file descriptor */
80 char_u *bw_buf; /* buffer with data to be written */
Bram Moolenaard089d9b2007-09-30 12:02:55 +000081 int bw_len; /* length of data */
Bram Moolenaar071d4272004-06-13 20:20:40 +000082#ifdef HAS_BW_FLAGS
83 int bw_flags; /* FIO_ flags */
84#endif
Bram Moolenaar8f4ac012014-08-10 13:38:34 +020085#ifdef FEAT_CRYPT
86 buf_T *bw_buffer; /* buffer being written */
87#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +000088 char_u bw_rest[CONV_RESTLEN]; /* not converted bytes */
89 int bw_restlen; /* nr of bytes in bw_rest[] */
90 int bw_first; /* first write call */
91 char_u *bw_conv_buf; /* buffer for writing converted chars */
92 int bw_conv_buflen; /* size of bw_conv_buf */
93 int bw_conv_error; /* set for conversion error */
Bram Moolenaar32b485f2009-07-29 16:06:27 +000094 linenr_T bw_conv_error_lnum; /* first line with error or zero */
95 linenr_T bw_start_lnum; /* line number at start of buffer */
Bram Moolenaar13505972019-01-24 15:04:48 +010096#ifdef USE_ICONV
Bram Moolenaar071d4272004-06-13 20:20:40 +000097 iconv_t bw_iconv_fd; /* descriptor for iconv() or -1 */
Bram Moolenaar071d4272004-06-13 20:20:40 +000098#endif
99};
100
Bram Moolenaard25c16e2016-01-29 22:13:30 +0100101static int buf_write_bytes(struct bw_info *ip);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000102
Bram Moolenaard25c16e2016-01-29 22:13:30 +0100103static linenr_T readfile_linenr(linenr_T linecnt, char_u *p, char_u *endp);
104static int ucs2bytes(unsigned c, char_u **pp, int flags);
105static int need_conversion(char_u *fenc);
106static int get_fio_flags(char_u *ptr);
107static char_u *check_for_bom(char_u *p, long size, int *lenp, int flags);
108static int make_bom(char_u *buf, char_u *name);
Bram Moolenaar4f974752019-02-17 17:44:42 +0100109#ifdef MSWIN
Bram Moolenaard25c16e2016-01-29 22:13:30 +0100110static int get_win_fio_flags(char_u *ptr);
Bram Moolenaar13505972019-01-24 15:04:48 +0100111#endif
112#ifdef MACOS_CONVERT
Bram Moolenaard25c16e2016-01-29 22:13:30 +0100113static int get_mac_fio_flags(char_u *ptr);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000114#endif
Bram Moolenaarbf1b7a72009-03-05 02:15:53 +0000115static char *e_auchangedbuf = N_("E812: Autocommands changed buffer or buffer name");
Bram Moolenaarb0bf8582005-12-13 20:02:15 +0000116
Bram Moolenaar071d4272004-06-13 20:20:40 +0000117 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100118filemess(
119 buf_T *buf,
120 char_u *name,
121 char_u *s,
122 int attr)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000123{
124 int msg_scroll_save;
125
126 if (msg_silent != 0)
127 return;
128 msg_add_fname(buf, name); /* put file name in IObuff with quotes */
129 /* If it's extremely long, truncate it. */
130 if (STRLEN(IObuff) > IOSIZE - 80)
131 IObuff[IOSIZE - 80] = NUL;
132 STRCAT(IObuff, s);
133 /*
134 * For the first message may have to start a new line.
135 * For further ones overwrite the previous one, reset msg_scroll before
136 * calling filemess().
137 */
138 msg_scroll_save = msg_scroll;
139 if (shortmess(SHM_OVERALL) && !exiting && p_verbose == 0)
140 msg_scroll = FALSE;
141 if (!msg_scroll) /* wait a bit when overwriting an error msg */
142 check_for_delay(FALSE);
143 msg_start();
144 msg_scroll = msg_scroll_save;
145 msg_scrolled_ign = TRUE;
146 /* may truncate the message to avoid a hit-return prompt */
147 msg_outtrans_attr(msg_may_trunc(FALSE, IObuff), attr);
148 msg_clr_eos();
149 out_flush();
150 msg_scrolled_ign = FALSE;
151}
152
153/*
154 * Read lines from file "fname" into the buffer after line "from".
155 *
156 * 1. We allocate blocks with lalloc, as big as possible.
157 * 2. Each block is filled with characters from the file with a single read().
158 * 3. The lines are inserted in the buffer with ml_append().
159 *
160 * (caller must check that fname != NULL, unless READ_STDIN is used)
161 *
162 * "lines_to_skip" is the number of lines that must be skipped
163 * "lines_to_read" is the number of lines that are appended
164 * When not recovering lines_to_skip is 0 and lines_to_read MAXLNUM.
165 *
166 * flags:
167 * READ_NEW starting to edit a new buffer
168 * READ_FILTER reading filter output
169 * READ_STDIN read from stdin instead of a file
170 * READ_BUFFER read from curbuf instead of a file (converting after reading
171 * stdin)
172 * READ_DUMMY read into a dummy buffer (to check if file contents changed)
Bram Moolenaar59f931e2010-07-24 20:27:03 +0200173 * READ_KEEP_UNDO don't clear undo info or read it from a file
Bram Moolenaarf71d7b92016-08-09 22:14:05 +0200174 * READ_FIFO read from fifo/socket instead of a file
Bram Moolenaar071d4272004-06-13 20:20:40 +0000175 *
Bram Moolenaare13b9af2017-01-13 22:01:02 +0100176 * return FAIL for failure, NOTDONE for directory (failure), or OK
Bram Moolenaar071d4272004-06-13 20:20:40 +0000177 */
178 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +0100179readfile(
180 char_u *fname,
181 char_u *sfname,
182 linenr_T from,
183 linenr_T lines_to_skip,
184 linenr_T lines_to_read,
185 exarg_T *eap, /* can be NULL! */
186 int flags)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000187{
188 int fd = 0;
189 int newfile = (flags & READ_NEW);
190 int check_readonly;
191 int filtering = (flags & READ_FILTER);
192 int read_stdin = (flags & READ_STDIN);
193 int read_buffer = (flags & READ_BUFFER);
Bram Moolenaarf71d7b92016-08-09 22:14:05 +0200194 int read_fifo = (flags & READ_FIFO);
Bram Moolenaar690ffc02008-01-04 15:31:21 +0000195 int set_options = newfile || read_buffer
196 || (eap != NULL && eap->read_edit);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000197 linenr_T read_buf_lnum = 1; /* next line to read from curbuf */
198 colnr_T read_buf_col = 0; /* next char to read from this line */
199 char_u c;
200 linenr_T lnum = from;
201 char_u *ptr = NULL; /* pointer into read buffer */
202 char_u *buffer = NULL; /* read buffer */
203 char_u *new_buffer = NULL; /* init to shut up gcc */
204 char_u *line_start = NULL; /* init to shut up gcc */
205 int wasempty; /* buffer was empty before reading */
206 colnr_T len;
207 long size = 0;
208 char_u *p;
Bram Moolenaar8767f522016-07-01 17:17:39 +0200209 off_T filesize = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000210 int skip_read = FALSE;
211#ifdef FEAT_CRYPT
212 char_u *cryptkey = NULL;
Bram Moolenaarf50a2532010-05-21 15:36:08 +0200213 int did_ask_for_key = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000214#endif
Bram Moolenaar55debbe2010-05-23 23:34:36 +0200215#ifdef FEAT_PERSISTENT_UNDO
216 context_sha256_T sha_ctx;
217 int read_undo_file = FALSE;
218#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000219 int split = 0; /* number of split lines */
220#define UNKNOWN 0x0fffffff /* file size is unknown */
221 linenr_T linecnt;
222 int error = FALSE; /* errors encountered */
223 int ff_error = EOL_UNKNOWN; /* file format with errors */
224 long linerest = 0; /* remaining chars in line */
225#ifdef UNIX
226 int perm = 0;
227 int swap_mode = -1; /* protection bits for swap file */
228#else
229 int perm;
230#endif
231 int fileformat = 0; /* end-of-line format */
232 int keep_fileformat = FALSE;
Bram Moolenaar8767f522016-07-01 17:17:39 +0200233 stat_T st;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000234 int file_readonly;
235 linenr_T skip_count = 0;
236 linenr_T read_count = 0;
237 int msg_save = msg_scroll;
238 linenr_T read_no_eol_lnum = 0; /* non-zero lnum when last line of
239 * last read was missing the eol */
Bram Moolenaar7a2699e2017-01-23 21:31:09 +0100240 int try_mac;
241 int try_dos;
242 int try_unix;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000243 int file_rewind = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000244 int can_retry;
Bram Moolenaarb0bf8582005-12-13 20:02:15 +0000245 linenr_T conv_error = 0; /* line nr with conversion error */
246 linenr_T illegal_byte = 0; /* line nr with illegal byte */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000247 int keep_dest_enc = FALSE; /* don't retry when char doesn't fit
248 in destination encoding */
Bram Moolenaarb0bf8582005-12-13 20:02:15 +0000249 int bad_char_behavior = BAD_REPLACE;
250 /* BAD_KEEP, BAD_DROP or character to
251 * replace with */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000252 char_u *tmpname = NULL; /* name of 'charconvert' output file */
253 int fio_flags = 0;
254 char_u *fenc; /* fileencoding to use */
255 int fenc_alloced; /* fenc_next is in allocated memory */
256 char_u *fenc_next = NULL; /* next item in 'fencs' or NULL */
257 int advance_fenc = FALSE;
258 long real_size = 0;
Bram Moolenaar13505972019-01-24 15:04:48 +0100259#ifdef USE_ICONV
Bram Moolenaar071d4272004-06-13 20:20:40 +0000260 iconv_t iconv_fd = (iconv_t)-1; /* descriptor for iconv() or -1 */
Bram Moolenaar13505972019-01-24 15:04:48 +0100261# ifdef FEAT_EVAL
Bram Moolenaar071d4272004-06-13 20:20:40 +0000262 int did_iconv = FALSE; /* TRUE when iconv() failed and trying
263 'charconvert' next */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000264# endif
Bram Moolenaar13505972019-01-24 15:04:48 +0100265#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000266 int converted = FALSE; /* TRUE if conversion done */
267 int notconverted = FALSE; /* TRUE if conversion wanted but it
268 wasn't possible */
269 char_u conv_rest[CONV_RESTLEN];
270 int conv_restlen = 0; /* nr of bytes in conv_rest[] */
Bram Moolenaarbb3d5dc2010-08-14 14:32:54 +0200271 buf_T *old_curbuf;
272 char_u *old_b_ffname;
273 char_u *old_b_fname;
274 int using_b_ffname;
275 int using_b_fname;
Bram Moolenaarbb3d5dc2010-08-14 14:32:54 +0200276
Bram Moolenaarc3691332016-04-20 12:49:49 +0200277 au_did_filetype = FALSE; /* reset before triggering any autocommands */
Bram Moolenaarc3691332016-04-20 12:49:49 +0200278
Bram Moolenaarcab35ad2011-02-15 17:39:22 +0100279 curbuf->b_no_eol_lnum = 0; /* in case it was set by the previous read */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000280
281 /*
282 * If there is no file name yet, use the one for the read file.
283 * BF_NOTEDITED is set to reflect this.
284 * Don't do this for a read from a filter.
285 * Only do this when 'cpoptions' contains the 'f' flag.
286 */
287 if (curbuf->b_ffname == NULL
288 && !filtering
289 && fname != NULL
290 && vim_strchr(p_cpo, CPO_FNAMER) != NULL
291 && !(flags & READ_DUMMY))
292 {
Bram Moolenaar2d3f4892006-01-20 23:02:51 +0000293 if (set_rw_fname(fname, sfname) == FAIL)
294 return FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000295 }
296
Bram Moolenaarbb3d5dc2010-08-14 14:32:54 +0200297 /* Remember the initial values of curbuf, curbuf->b_ffname and
298 * curbuf->b_fname to detect whether they are altered as a result of
299 * executing nasty autocommands. Also check if "fname" and "sfname"
300 * point to one of these values. */
301 old_curbuf = curbuf;
302 old_b_ffname = curbuf->b_ffname;
303 old_b_fname = curbuf->b_fname;
304 using_b_ffname = (fname == curbuf->b_ffname)
305 || (sfname == curbuf->b_ffname);
306 using_b_fname = (fname == curbuf->b_fname) || (sfname == curbuf->b_fname);
Bram Moolenaarbb3d5dc2010-08-14 14:32:54 +0200307
Bram Moolenaardf177f62005-02-22 08:39:57 +0000308 /* After reading a file the cursor line changes but we don't want to
309 * display the line. */
310 ex_no_reprint = TRUE;
311
Bram Moolenaar55b7cf82006-09-09 12:52:42 +0000312 /* don't display the file info for another buffer now */
313 need_fileinfo = FALSE;
314
Bram Moolenaar071d4272004-06-13 20:20:40 +0000315 /*
316 * For Unix: Use the short file name whenever possible.
317 * Avoids problems with networks and when directory names are changed.
318 * Don't do this for MS-DOS, a "cd" in a sub-shell may have moved us to
319 * another directory, which we don't detect.
320 */
321 if (sfname == NULL)
322 sfname = fname;
Bram Moolenaara06ecab2016-07-16 14:47:36 +0200323#if defined(UNIX)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000324 fname = sfname;
325#endif
326
Bram Moolenaar071d4272004-06-13 20:20:40 +0000327 /*
328 * The BufReadCmd and FileReadCmd events intercept the reading process by
329 * executing the associated commands instead.
330 */
331 if (!filtering && !read_stdin && !read_buffer)
332 {
333 pos_T pos;
334
335 pos = curbuf->b_op_start;
336
337 /* Set '[ mark to the line above where the lines go (line 1 if zero). */
338 curbuf->b_op_start.lnum = ((from == 0) ? 1 : from);
339 curbuf->b_op_start.col = 0;
340
341 if (newfile)
342 {
343 if (apply_autocmds_exarg(EVENT_BUFREADCMD, NULL, sfname,
344 FALSE, curbuf, eap))
345#ifdef FEAT_EVAL
346 return aborting() ? FAIL : OK;
347#else
348 return OK;
349#endif
350 }
351 else if (apply_autocmds_exarg(EVENT_FILEREADCMD, sfname, sfname,
352 FALSE, NULL, eap))
353#ifdef FEAT_EVAL
354 return aborting() ? FAIL : OK;
355#else
356 return OK;
357#endif
358
359 curbuf->b_op_start = pos;
360 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000361
362 if ((shortmess(SHM_OVER) || curbuf->b_help) && p_verbose == 0)
363 msg_scroll = FALSE; /* overwrite previous file message */
364 else
365 msg_scroll = TRUE; /* don't overwrite previous file message */
366
367 /*
368 * If the name ends in a path separator, we can't open it. Check here,
369 * because reading the file may actually work, but then creating the swap
370 * file may destroy it! Reported on MS-DOS and Win 95.
371 * If the name is too long we might crash further on, quit here.
372 */
Bram Moolenaar15d0a8c2004-09-06 17:44:46 +0000373 if (fname != NULL && *fname != NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000374 {
Bram Moolenaar1cd871b2004-12-19 22:46:22 +0000375 p = fname + STRLEN(fname);
376 if (after_pathsep(fname, p) || STRLEN(fname) >= MAXPATHL)
Bram Moolenaar15d0a8c2004-09-06 17:44:46 +0000377 {
378 filemess(curbuf, fname, (char_u *)_("Illegal file name"), 0);
379 msg_end();
380 msg_scroll = msg_save;
381 return FAIL;
382 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000383 }
384
Bram Moolenaarf71d7b92016-08-09 22:14:05 +0200385 if (!read_stdin && !read_buffer && !read_fifo)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000386 {
Bram Moolenaar4e4f5292013-08-30 17:07:01 +0200387#ifdef UNIX
388 /*
389 * On Unix it is possible to read a directory, so we have to
390 * check for it before the mch_open().
391 */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000392 perm = mch_getperm(fname);
393 if (perm >= 0 && !S_ISREG(perm) /* not a regular file ... */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000394 && !S_ISFIFO(perm) /* ... or fifo */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000395 && !S_ISSOCK(perm) /* ... or socket */
Bram Moolenaarfe1c56d2007-07-10 15:10:54 +0000396# ifdef OPEN_CHR_FILES
397 && !(S_ISCHR(perm) && is_dev_fd_file(fname))
398 /* ... or a character special file named /dev/fd/<n> */
399# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000400 )
401 {
Bram Moolenaare13b9af2017-01-13 22:01:02 +0100402 int retval = FAIL;
403
Bram Moolenaar071d4272004-06-13 20:20:40 +0000404 if (S_ISDIR(perm))
Bram Moolenaare13b9af2017-01-13 22:01:02 +0100405 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000406 filemess(curbuf, fname, (char_u *)_("is a directory"), 0);
Bram Moolenaare13b9af2017-01-13 22:01:02 +0100407 retval = NOTDONE;
408 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000409 else
410 filemess(curbuf, fname, (char_u *)_("is not a file"), 0);
411 msg_end();
412 msg_scroll = msg_save;
Bram Moolenaare13b9af2017-01-13 22:01:02 +0100413 return retval;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000414 }
Bram Moolenaar4e4f5292013-08-30 17:07:01 +0200415#endif
Bram Moolenaar48e330a2016-02-23 14:53:34 +0100416#if defined(MSWIN)
Bram Moolenaarc67764a2006-10-12 19:14:26 +0000417 /*
418 * MS-Windows allows opening a device, but we will probably get stuck
419 * trying to read it.
420 */
421 if (!p_odev && mch_nodetype(fname) == NODE_WRITABLE)
422 {
Bram Moolenaar5386a122007-06-28 20:02:32 +0000423 filemess(curbuf, fname, (char_u *)_("is a device (disabled with 'opendevice' option)"), 0);
Bram Moolenaarc67764a2006-10-12 19:14:26 +0000424 msg_end();
425 msg_scroll = msg_save;
426 return FAIL;
427 }
Bram Moolenaar043545e2006-10-10 16:44:07 +0000428#endif
Bram Moolenaar4e4f5292013-08-30 17:07:01 +0200429 }
Bram Moolenaar043545e2006-10-10 16:44:07 +0000430
Bram Moolenaarad875fb2013-07-24 15:02:03 +0200431 /* Set default or forced 'fileformat' and 'binary'. */
432 set_file_options(set_options, eap);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000433
434 /*
435 * When opening a new file we take the readonly flag from the file.
436 * Default is r/w, can be set to r/o below.
437 * Don't reset it when in readonly mode
438 * Only set/reset b_p_ro when BF_CHECK_RO is set.
439 */
440 check_readonly = (newfile && (curbuf->b_flags & BF_CHECK_RO));
Bram Moolenaar4399ef42005-02-12 14:29:27 +0000441 if (check_readonly && !readonlymode)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000442 curbuf->b_p_ro = FALSE;
443
Bram Moolenaarf71d7b92016-08-09 22:14:05 +0200444 if (newfile && !read_stdin && !read_buffer && !read_fifo)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000445 {
Bram Moolenaare60acc12011-05-10 16:41:25 +0200446 /* Remember time of file. */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000447 if (mch_stat((char *)fname, &st) >= 0)
448 {
449 buf_store_time(curbuf, &st, fname);
450 curbuf->b_mtime_read = curbuf->b_mtime;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000451#ifdef UNIX
452 /*
453 * Use the protection bits of the original file for the swap file.
454 * This makes it possible for others to read the name of the
455 * edited file from the swapfile, but only if they can read the
456 * edited file.
457 * Remove the "write" and "execute" bits for group and others
458 * (they must not write the swapfile).
459 * Add the "read" and "write" bits for the user, otherwise we may
460 * not be able to write to the file ourselves.
461 * Setting the bits is done below, after creating the swap file.
462 */
463 swap_mode = (st.st_mode & 0644) | 0600;
464#endif
465#ifdef FEAT_CW_EDITOR
466 /* Get the FSSpec on MacOS
467 * TODO: Update it properly when the buffer name changes
468 */
469 (void)GetFSSpecFromPath(curbuf->b_ffname, &curbuf->b_FSSpec);
470#endif
471#ifdef VMS
472 curbuf->b_fab_rfm = st.st_fab_rfm;
Bram Moolenaard4755bb2004-09-02 19:12:26 +0000473 curbuf->b_fab_rat = st.st_fab_rat;
474 curbuf->b_fab_mrs = st.st_fab_mrs;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000475#endif
476 }
477 else
478 {
479 curbuf->b_mtime = 0;
480 curbuf->b_mtime_read = 0;
481 curbuf->b_orig_size = 0;
482 curbuf->b_orig_mode = 0;
483 }
484
485 /* Reset the "new file" flag. It will be set again below when the
486 * file doesn't exist. */
487 curbuf->b_flags &= ~(BF_NEW | BF_NEW_W);
488 }
489
490/*
491 * for UNIX: check readonly with perm and mch_access()
Bram Moolenaar48e330a2016-02-23 14:53:34 +0100492 * for Amiga: check readonly by trying to open the file for writing
Bram Moolenaar071d4272004-06-13 20:20:40 +0000493 */
494 file_readonly = FALSE;
495 if (read_stdin)
496 {
Bram Moolenaar48e330a2016-02-23 14:53:34 +0100497#if defined(MSWIN)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000498 /* Force binary I/O on stdin to avoid CR-LF -> LF conversion. */
499 setmode(0, O_BINARY);
500#endif
501 }
502 else if (!read_buffer)
503 {
504#ifdef USE_MCH_ACCESS
505 if (
506# ifdef UNIX
507 !(perm & 0222) ||
508# endif
509 mch_access((char *)fname, W_OK))
510 file_readonly = TRUE;
511 fd = mch_open((char *)fname, O_RDONLY | O_EXTRA, 0);
512#else
513 if (!newfile
514 || readonlymode
515 || (fd = mch_open((char *)fname, O_RDWR | O_EXTRA, 0)) < 0)
516 {
517 file_readonly = TRUE;
518 /* try to open ro */
519 fd = mch_open((char *)fname, O_RDONLY | O_EXTRA, 0);
520 }
521#endif
522 }
523
524 if (fd < 0) /* cannot open at all */
525 {
526#ifndef UNIX
527 int isdir_f;
528#endif
529 msg_scroll = msg_save;
530#ifndef UNIX
531 /*
Bram Moolenaar48e330a2016-02-23 14:53:34 +0100532 * On Amiga we can't open a directory, check here.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000533 */
534 isdir_f = (mch_isdir(fname));
535 perm = mch_getperm(fname); /* check if the file exists */
536 if (isdir_f)
537 {
538 filemess(curbuf, sfname, (char_u *)_("is a directory"), 0);
539 curbuf->b_p_ro = TRUE; /* must use "w!" now */
540 }
541 else
542#endif
543 if (newfile)
544 {
Bram Moolenaar2efbc662010-05-14 18:56:38 +0200545 if (perm < 0
546#ifdef ENOENT
547 && errno == ENOENT
548#endif
549 )
Bram Moolenaar071d4272004-06-13 20:20:40 +0000550 {
551 /*
552 * Set the 'new-file' flag, so that when the file has
553 * been created by someone else, a ":w" will complain.
554 */
555 curbuf->b_flags |= BF_NEW;
556
557 /* Create a swap file now, so that other Vims are warned
558 * that we are editing this file. Don't do this for a
559 * "nofile" or "nowrite" buffer type. */
560#ifdef FEAT_QUICKFIX
561 if (!bt_dontwrite(curbuf))
562#endif
Bram Moolenaarbf1b7a72009-03-05 02:15:53 +0000563 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000564 check_need_swap(newfile);
Bram Moolenaarbf1b7a72009-03-05 02:15:53 +0000565 /* SwapExists autocommand may mess things up */
566 if (curbuf != old_curbuf
567 || (using_b_ffname
568 && (old_b_ffname != curbuf->b_ffname))
569 || (using_b_fname
570 && (old_b_fname != curbuf->b_fname)))
571 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +0100572 emsg(_(e_auchangedbuf));
Bram Moolenaarbf1b7a72009-03-05 02:15:53 +0000573 return FAIL;
574 }
Bram Moolenaarbf1b7a72009-03-05 02:15:53 +0000575 }
Bram Moolenaar5b962cf2005-12-12 21:58:40 +0000576 if (dir_of_file_exists(fname))
577 filemess(curbuf, sfname, (char_u *)_("[New File]"), 0);
578 else
579 filemess(curbuf, sfname,
580 (char_u *)_("[New DIRECTORY]"), 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000581#ifdef FEAT_VIMINFO
582 /* Even though this is a new file, it might have been
583 * edited before and deleted. Get the old marks. */
584 check_marks_read();
585#endif
Bram Moolenaarad875fb2013-07-24 15:02:03 +0200586 /* Set forced 'fileencoding'. */
587 if (eap != NULL)
588 set_forced_fenc(eap);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000589 apply_autocmds_exarg(EVENT_BUFNEWFILE, sfname, sfname,
590 FALSE, curbuf, eap);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000591 /* remember the current fileformat */
592 save_file_ff(curbuf);
593
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +0100594#if defined(FEAT_EVAL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000595 if (aborting()) /* autocmds may abort script processing */
596 return FAIL;
597#endif
598 return OK; /* a new file is not an error */
599 }
600 else
601 {
Bram Moolenaar202795b2005-10-11 20:29:39 +0000602 filemess(curbuf, sfname, (char_u *)(
603# ifdef EFBIG
604 (errno == EFBIG) ? _("[File too big]") :
605# endif
Bram Moolenaar2efbc662010-05-14 18:56:38 +0200606# ifdef EOVERFLOW
607 (errno == EOVERFLOW) ? _("[File too big]") :
608# endif
Bram Moolenaar202795b2005-10-11 20:29:39 +0000609 _("[Permission Denied]")), 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000610 curbuf->b_p_ro = TRUE; /* must use "w!" now */
611 }
612 }
613
614 return FAIL;
615 }
616
617 /*
618 * Only set the 'ro' flag for readonly files the first time they are
619 * loaded. Help files always get readonly mode
620 */
621 if ((check_readonly && file_readonly) || curbuf->b_help)
622 curbuf->b_p_ro = TRUE;
623
Bram Moolenaar910f66f2006-04-05 20:41:53 +0000624 if (set_options)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000625 {
Bram Moolenaar690ffc02008-01-04 15:31:21 +0000626 /* Don't change 'eol' if reading from buffer as it will already be
627 * correctly set when reading stdin. */
628 if (!read_buffer)
629 {
630 curbuf->b_p_eol = TRUE;
631 curbuf->b_start_eol = TRUE;
632 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000633 curbuf->b_p_bomb = FALSE;
Bram Moolenaar83eb8852007-08-12 13:51:26 +0000634 curbuf->b_start_bomb = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000635 }
636
637 /* Create a swap file now, so that other Vims are warned that we are
638 * editing this file.
639 * Don't do this for a "nofile" or "nowrite" buffer type. */
640#ifdef FEAT_QUICKFIX
641 if (!bt_dontwrite(curbuf))
642#endif
643 {
644 check_need_swap(newfile);
Bram Moolenaarbf1b7a72009-03-05 02:15:53 +0000645 if (!read_stdin && (curbuf != old_curbuf
646 || (using_b_ffname && (old_b_ffname != curbuf->b_ffname))
647 || (using_b_fname && (old_b_fname != curbuf->b_fname))))
648 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +0100649 emsg(_(e_auchangedbuf));
Bram Moolenaarbf1b7a72009-03-05 02:15:53 +0000650 if (!read_buffer)
651 close(fd);
652 return FAIL;
653 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000654#ifdef UNIX
655 /* Set swap file protection bits after creating it. */
Bram Moolenaarf061e0b2009-06-24 15:32:01 +0000656 if (swap_mode > 0 && curbuf->b_ml.ml_mfp != NULL
657 && curbuf->b_ml.ml_mfp->mf_fname != NULL)
Bram Moolenaar5a73e0c2017-11-04 21:35:01 +0100658 {
659 char_u *swap_fname = curbuf->b_ml.ml_mfp->mf_fname;
660
661 /*
662 * If the group-read bit is set but not the world-read bit, then
663 * the group must be equal to the group of the original file. If
664 * we can't make that happen then reset the group-read bit. This
665 * avoids making the swap file readable to more users when the
666 * primary group of the user is too permissive.
667 */
668 if ((swap_mode & 044) == 040)
669 {
670 stat_T swap_st;
671
672 if (mch_stat((char *)swap_fname, &swap_st) >= 0
673 && st.st_gid != swap_st.st_gid
Bram Moolenaar02e802b2018-04-19 21:15:27 +0200674# ifdef HAVE_FCHOWN
Bram Moolenaar5a73e0c2017-11-04 21:35:01 +0100675 && fchown(curbuf->b_ml.ml_mfp->mf_fd, -1, st.st_gid)
Bram Moolenaar02e802b2018-04-19 21:15:27 +0200676 == -1
Bram Moolenaar1f131ae2018-05-21 13:39:40 +0200677# endif
Bram Moolenaar02e802b2018-04-19 21:15:27 +0200678 )
Bram Moolenaar5a73e0c2017-11-04 21:35:01 +0100679 swap_mode &= 0600;
680 }
681
682 (void)mch_setperm(swap_fname, (long)swap_mode);
683 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000684#endif
685 }
686
Bram Moolenaarb815dac2005-12-07 20:59:24 +0000687#if defined(HAS_SWAP_EXISTS_ACTION)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000688 /* If "Quit" selected at ATTENTION dialog, don't load the file */
689 if (swap_exists_action == SEA_QUIT)
690 {
691 if (!read_buffer && !read_stdin)
692 close(fd);
693 return FAIL;
694 }
695#endif
696
697 ++no_wait_return; /* don't wait for return yet */
698
699 /*
700 * Set '[ mark to the line above where the lines go (line 1 if zero).
701 */
702 curbuf->b_op_start.lnum = ((from == 0) ? 1 : from);
703 curbuf->b_op_start.col = 0;
704
Bram Moolenaar7a2699e2017-01-23 21:31:09 +0100705 try_mac = (vim_strchr(p_ffs, 'm') != NULL);
706 try_dos = (vim_strchr(p_ffs, 'd') != NULL);
707 try_unix = (vim_strchr(p_ffs, 'x') != NULL);
708
Bram Moolenaar071d4272004-06-13 20:20:40 +0000709 if (!read_buffer)
710 {
711 int m = msg_scroll;
712 int n = msg_scrolled;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000713
714 /*
715 * The file must be closed again, the autocommands may want to change
716 * the file before reading it.
717 */
718 if (!read_stdin)
719 close(fd); /* ignore errors */
720
721 /*
722 * The output from the autocommands should not overwrite anything and
723 * should not be overwritten: Set msg_scroll, restore its value if no
724 * output was done.
725 */
726 msg_scroll = TRUE;
727 if (filtering)
728 apply_autocmds_exarg(EVENT_FILTERREADPRE, NULL, sfname,
729 FALSE, curbuf, eap);
730 else if (read_stdin)
731 apply_autocmds_exarg(EVENT_STDINREADPRE, NULL, sfname,
732 FALSE, curbuf, eap);
733 else if (newfile)
734 apply_autocmds_exarg(EVENT_BUFREADPRE, NULL, sfname,
735 FALSE, curbuf, eap);
736 else
737 apply_autocmds_exarg(EVENT_FILEREADPRE, sfname, sfname,
738 FALSE, NULL, eap);
Bram Moolenaar7a2699e2017-01-23 21:31:09 +0100739 /* autocommands may have changed it */
740 try_mac = (vim_strchr(p_ffs, 'm') != NULL);
741 try_dos = (vim_strchr(p_ffs, 'd') != NULL);
742 try_unix = (vim_strchr(p_ffs, 'x') != NULL);
743
Bram Moolenaar071d4272004-06-13 20:20:40 +0000744 if (msg_scrolled == n)
745 msg_scroll = m;
746
747#ifdef FEAT_EVAL
748 if (aborting()) /* autocmds may abort script processing */
749 {
750 --no_wait_return;
751 msg_scroll = msg_save;
752 curbuf->b_p_ro = TRUE; /* must use "w!" now */
753 return FAIL;
754 }
755#endif
756 /*
757 * Don't allow the autocommands to change the current buffer.
758 * Try to re-open the file.
Bram Moolenaarbf1b7a72009-03-05 02:15:53 +0000759 *
760 * Don't allow the autocommands to change the buffer name either
761 * (cd for example) if it invalidates fname or sfname.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000762 */
763 if (!read_stdin && (curbuf != old_curbuf
Bram Moolenaarbf1b7a72009-03-05 02:15:53 +0000764 || (using_b_ffname && (old_b_ffname != curbuf->b_ffname))
765 || (using_b_fname && (old_b_fname != curbuf->b_fname))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000766 || (fd = mch_open((char *)fname, O_RDONLY | O_EXTRA, 0)) < 0))
767 {
768 --no_wait_return;
769 msg_scroll = msg_save;
770 if (fd < 0)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +0100771 emsg(_("E200: *ReadPre autocommands made the file unreadable"));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000772 else
Bram Moolenaarf9e3e092019-01-13 23:38:42 +0100773 emsg(_("E201: *ReadPre autocommands must not change current buffer"));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000774 curbuf->b_p_ro = TRUE; /* must use "w!" now */
775 return FAIL;
776 }
777 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000778
779 /* Autocommands may add lines to the file, need to check if it is empty */
780 wasempty = (curbuf->b_ml.ml_flags & ML_EMPTY);
781
782 if (!recoverymode && !filtering && !(flags & READ_DUMMY))
783 {
784 /*
785 * Show the user that we are busy reading the input. Sometimes this
786 * may take a while. When reading from stdin another program may
787 * still be running, don't move the cursor to the last line, unless
788 * always using the GUI.
789 */
790 if (read_stdin)
791 {
Bram Moolenaar234d1622017-11-18 14:55:23 +0100792 if (!is_not_a_term())
793 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000794#ifndef ALWAYS_USE_GUI
Bram Moolenaar234d1622017-11-18 14:55:23 +0100795 mch_msg(_("Vim: Reading from stdin...\n"));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000796#endif
797#ifdef FEAT_GUI
Bram Moolenaar234d1622017-11-18 14:55:23 +0100798 /* Also write a message in the GUI window, if there is one. */
799 if (gui.in_use && !gui.dying && !gui.starting)
800 {
801 p = (char_u *)_("Reading from stdin...");
802 gui_write(p, (int)STRLEN(p));
803 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000804#endif
Bram Moolenaar234d1622017-11-18 14:55:23 +0100805 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000806 }
807 else if (!read_buffer)
808 filemess(curbuf, sfname, (char_u *)"", 0);
809 }
810
811 msg_scroll = FALSE; /* overwrite the file message */
812
813 /*
814 * Set linecnt now, before the "retry" caused by a wrong guess for
815 * fileformat, and after the autocommands, which may change them.
816 */
817 linecnt = curbuf->b_ml.ml_line_count;
818
Bram Moolenaarb0bf8582005-12-13 20:02:15 +0000819 /* "++bad=" argument. */
820 if (eap != NULL && eap->bad_char != 0)
Bram Moolenaar195d6352005-12-19 22:08:24 +0000821 {
Bram Moolenaarb0bf8582005-12-13 20:02:15 +0000822 bad_char_behavior = eap->bad_char;
Bram Moolenaar910f66f2006-04-05 20:41:53 +0000823 if (set_options)
Bram Moolenaar195d6352005-12-19 22:08:24 +0000824 curbuf->b_bad_char = eap->bad_char;
825 }
826 else
827 curbuf->b_bad_char = 0;
Bram Moolenaarb0bf8582005-12-13 20:02:15 +0000828
Bram Moolenaar071d4272004-06-13 20:20:40 +0000829 /*
Bram Moolenaarb0bf8582005-12-13 20:02:15 +0000830 * Decide which 'encoding' to use or use first.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000831 */
832 if (eap != NULL && eap->force_enc != 0)
833 {
834 fenc = enc_canonize(eap->cmd + eap->force_enc);
835 fenc_alloced = TRUE;
Bram Moolenaarb0bf8582005-12-13 20:02:15 +0000836 keep_dest_enc = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000837 }
838 else if (curbuf->b_p_bin)
839 {
840 fenc = (char_u *)""; /* binary: don't convert */
841 fenc_alloced = FALSE;
842 }
843 else if (curbuf->b_help)
844 {
845 char_u firstline[80];
Bram Moolenaar75c50c42005-06-04 22:06:24 +0000846 int fc;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000847
848 /* Help files are either utf-8 or latin1. Try utf-8 first, if this
849 * fails it must be latin1.
850 * Always do this when 'encoding' is "utf-8". Otherwise only do
851 * this when needed to avoid [converted] remarks all the time.
852 * It is needed when the first line contains non-ASCII characters.
853 * That is only in *.??x files. */
854 fenc = (char_u *)"latin1";
855 c = enc_utf8;
Bram Moolenaar75c50c42005-06-04 22:06:24 +0000856 if (!c && !read_stdin)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000857 {
Bram Moolenaar75c50c42005-06-04 22:06:24 +0000858 fc = fname[STRLEN(fname) - 1];
859 if (TOLOWER_ASC(fc) == 'x')
860 {
861 /* Read the first line (and a bit more). Immediately rewind to
862 * the start of the file. If the read() fails "len" is -1. */
Bram Moolenaar540fc6f2010-12-17 16:27:16 +0100863 len = read_eintr(fd, firstline, 80);
Bram Moolenaar8767f522016-07-01 17:17:39 +0200864 vim_lseek(fd, (off_T)0L, SEEK_SET);
Bram Moolenaar75c50c42005-06-04 22:06:24 +0000865 for (p = firstline; p < firstline + len; ++p)
866 if (*p >= 0x80)
867 {
868 c = TRUE;
869 break;
870 }
871 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000872 }
873
874 if (c)
875 {
876 fenc_next = fenc;
877 fenc = (char_u *)"utf-8";
878
879 /* When the file is utf-8 but a character doesn't fit in
880 * 'encoding' don't retry. In help text editing utf-8 bytes
881 * doesn't make sense. */
Bram Moolenaarf193fff2006-04-27 00:02:13 +0000882 if (!enc_utf8)
883 keep_dest_enc = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000884 }
885 fenc_alloced = FALSE;
886 }
887 else if (*p_fencs == NUL)
888 {
889 fenc = curbuf->b_p_fenc; /* use format from buffer */
890 fenc_alloced = FALSE;
891 }
892 else
893 {
894 fenc_next = p_fencs; /* try items in 'fileencodings' */
895 fenc = next_fenc(&fenc_next);
896 fenc_alloced = TRUE;
897 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000898
899 /*
900 * Jump back here to retry reading the file in different ways.
901 * Reasons to retry:
902 * - encoding conversion failed: try another one from "fenc_next"
903 * - BOM detected and fenc was set, need to setup conversion
904 * - "fileformat" check failed: try another
905 *
906 * Variables set for special retry actions:
907 * "file_rewind" Rewind the file to start reading it again.
908 * "advance_fenc" Advance "fenc" using "fenc_next".
909 * "skip_read" Re-use already read bytes (BOM detected).
910 * "did_iconv" iconv() conversion failed, try 'charconvert'.
911 * "keep_fileformat" Don't reset "fileformat".
912 *
913 * Other status indicators:
914 * "tmpname" When != NULL did conversion with 'charconvert'.
915 * Output file has to be deleted afterwards.
916 * "iconv_fd" When != -1 did conversion with iconv().
917 */
918retry:
919
920 if (file_rewind)
921 {
922 if (read_buffer)
923 {
924 read_buf_lnum = 1;
925 read_buf_col = 0;
926 }
Bram Moolenaar8767f522016-07-01 17:17:39 +0200927 else if (read_stdin || vim_lseek(fd, (off_T)0L, SEEK_SET) != 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000928 {
929 /* Can't rewind the file, give up. */
930 error = TRUE;
931 goto failed;
932 }
933 /* Delete the previously read lines. */
934 while (lnum > from)
935 ml_delete(lnum--, FALSE);
936 file_rewind = FALSE;
Bram Moolenaar910f66f2006-04-05 20:41:53 +0000937 if (set_options)
Bram Moolenaar83eb8852007-08-12 13:51:26 +0000938 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000939 curbuf->b_p_bomb = FALSE;
Bram Moolenaar83eb8852007-08-12 13:51:26 +0000940 curbuf->b_start_bomb = FALSE;
941 }
Bram Moolenaarb0bf8582005-12-13 20:02:15 +0000942 conv_error = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000943 }
944
945 /*
946 * When retrying with another "fenc" and the first time "fileformat"
947 * will be reset.
948 */
949 if (keep_fileformat)
950 keep_fileformat = FALSE;
951 else
952 {
953 if (eap != NULL && eap->force_ff != 0)
Bram Moolenaar1c860362008-11-12 15:05:21 +0000954 {
Bram Moolenaar071d4272004-06-13 20:20:40 +0000955 fileformat = get_fileformat_force(curbuf, eap);
Bram Moolenaar1c860362008-11-12 15:05:21 +0000956 try_unix = try_dos = try_mac = FALSE;
957 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000958 else if (curbuf->b_p_bin)
959 fileformat = EOL_UNIX; /* binary: use Unix format */
960 else if (*p_ffs == NUL)
961 fileformat = get_fileformat(curbuf);/* use format from buffer */
962 else
963 fileformat = EOL_UNKNOWN; /* detect from file */
964 }
965
Bram Moolenaar13505972019-01-24 15:04:48 +0100966#ifdef USE_ICONV
Bram Moolenaar071d4272004-06-13 20:20:40 +0000967 if (iconv_fd != (iconv_t)-1)
968 {
969 /* aborted conversion with iconv(), close the descriptor */
970 iconv_close(iconv_fd);
971 iconv_fd = (iconv_t)-1;
972 }
Bram Moolenaar13505972019-01-24 15:04:48 +0100973#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000974
975 if (advance_fenc)
976 {
977 /*
978 * Try the next entry in 'fileencodings'.
979 */
980 advance_fenc = FALSE;
981
982 if (eap != NULL && eap->force_enc != 0)
983 {
984 /* Conversion given with "++cc=" wasn't possible, read
985 * without conversion. */
986 notconverted = TRUE;
Bram Moolenaarb0bf8582005-12-13 20:02:15 +0000987 conv_error = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000988 if (fenc_alloced)
989 vim_free(fenc);
990 fenc = (char_u *)"";
991 fenc_alloced = FALSE;
992 }
993 else
994 {
995 if (fenc_alloced)
996 vim_free(fenc);
997 if (fenc_next != NULL)
998 {
999 fenc = next_fenc(&fenc_next);
1000 fenc_alloced = (fenc_next != NULL);
1001 }
1002 else
1003 {
1004 fenc = (char_u *)"";
1005 fenc_alloced = FALSE;
1006 }
1007 }
1008 if (tmpname != NULL)
1009 {
1010 mch_remove(tmpname); /* delete converted file */
Bram Moolenaard23a8232018-02-10 18:45:26 +01001011 VIM_CLEAR(tmpname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001012 }
1013 }
1014
1015 /*
Bram Moolenaarb5cdf2e2009-07-29 16:25:31 +00001016 * Conversion may be required when the encoding of the file is different
1017 * from 'encoding' or 'encoding' is UTF-16, UCS-2 or UCS-4.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001018 */
1019 fio_flags = 0;
Bram Moolenaarb5cdf2e2009-07-29 16:25:31 +00001020 converted = need_conversion(fenc);
1021 if (converted)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001022 {
1023
1024 /* "ucs-bom" means we need to check the first bytes of the file
1025 * for a BOM. */
1026 if (STRCMP(fenc, ENC_UCSBOM) == 0)
1027 fio_flags = FIO_UCSBOM;
1028
1029 /*
1030 * Check if UCS-2/4 or Latin1 to UTF-8 conversion needs to be
1031 * done. This is handled below after read(). Prepare the
1032 * fio_flags to avoid having to parse the string each time.
1033 * Also check for Unicode to Latin1 conversion, because iconv()
1034 * appears not to handle this correctly. This works just like
1035 * conversion to UTF-8 except how the resulting character is put in
1036 * the buffer.
1037 */
1038 else if (enc_utf8 || STRCMP(p_enc, "latin1") == 0)
1039 fio_flags = get_fio_flags(fenc);
1040
Bram Moolenaar4f974752019-02-17 17:44:42 +01001041#ifdef MSWIN
Bram Moolenaar071d4272004-06-13 20:20:40 +00001042 /*
1043 * Conversion from an MS-Windows codepage to UTF-8 or another codepage
1044 * is handled with MultiByteToWideChar().
1045 */
1046 if (fio_flags == 0)
1047 fio_flags = get_win_fio_flags(fenc);
Bram Moolenaar13505972019-01-24 15:04:48 +01001048#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001049
Bram Moolenaar13505972019-01-24 15:04:48 +01001050#ifdef MACOS_CONVERT
Bram Moolenaar071d4272004-06-13 20:20:40 +00001051 /* Conversion from Apple MacRoman to latin1 or UTF-8 */
1052 if (fio_flags == 0)
1053 fio_flags = get_mac_fio_flags(fenc);
Bram Moolenaar13505972019-01-24 15:04:48 +01001054#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001055
Bram Moolenaar13505972019-01-24 15:04:48 +01001056#ifdef USE_ICONV
Bram Moolenaar071d4272004-06-13 20:20:40 +00001057 /*
1058 * Try using iconv() if we can't convert internally.
1059 */
1060 if (fio_flags == 0
Bram Moolenaar13505972019-01-24 15:04:48 +01001061# ifdef FEAT_EVAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00001062 && !did_iconv
Bram Moolenaar13505972019-01-24 15:04:48 +01001063# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001064 )
1065 iconv_fd = (iconv_t)my_iconv_open(
1066 enc_utf8 ? (char_u *)"utf-8" : p_enc, fenc);
Bram Moolenaar13505972019-01-24 15:04:48 +01001067#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001068
Bram Moolenaar13505972019-01-24 15:04:48 +01001069#ifdef FEAT_EVAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00001070 /*
1071 * Use the 'charconvert' expression when conversion is required
1072 * and we can't do it internally or with iconv().
1073 */
1074 if (fio_flags == 0 && !read_stdin && !read_buffer && *p_ccv != NUL
Bram Moolenaarf71d7b92016-08-09 22:14:05 +02001075 && !read_fifo
Bram Moolenaar13505972019-01-24 15:04:48 +01001076# ifdef USE_ICONV
Bram Moolenaar071d4272004-06-13 20:20:40 +00001077 && iconv_fd == (iconv_t)-1
Bram Moolenaar13505972019-01-24 15:04:48 +01001078# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001079 )
1080 {
Bram Moolenaar13505972019-01-24 15:04:48 +01001081# ifdef USE_ICONV
Bram Moolenaar071d4272004-06-13 20:20:40 +00001082 did_iconv = FALSE;
Bram Moolenaar13505972019-01-24 15:04:48 +01001083# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001084 /* Skip conversion when it's already done (retry for wrong
1085 * "fileformat"). */
1086 if (tmpname == NULL)
1087 {
1088 tmpname = readfile_charconvert(fname, fenc, &fd);
1089 if (tmpname == NULL)
1090 {
1091 /* Conversion failed. Try another one. */
1092 advance_fenc = TRUE;
1093 if (fd < 0)
1094 {
1095 /* Re-opening the original file failed! */
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01001096 emsg(_("E202: Conversion made file unreadable!"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00001097 error = TRUE;
1098 goto failed;
1099 }
1100 goto retry;
1101 }
1102 }
1103 }
1104 else
Bram Moolenaar13505972019-01-24 15:04:48 +01001105#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001106 {
1107 if (fio_flags == 0
Bram Moolenaar13505972019-01-24 15:04:48 +01001108#ifdef USE_ICONV
Bram Moolenaar071d4272004-06-13 20:20:40 +00001109 && iconv_fd == (iconv_t)-1
Bram Moolenaar13505972019-01-24 15:04:48 +01001110#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001111 )
1112 {
1113 /* Conversion wanted but we can't.
1114 * Try the next conversion in 'fileencodings' */
1115 advance_fenc = TRUE;
1116 goto retry;
1117 }
1118 }
1119 }
1120
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00001121 /* Set "can_retry" when it's possible to rewind the file and try with
Bram Moolenaar071d4272004-06-13 20:20:40 +00001122 * another "fenc" value. It's FALSE when no other "fenc" to try, reading
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00001123 * stdin or fixed at a specific encoding. */
Bram Moolenaarf71d7b92016-08-09 22:14:05 +02001124 can_retry = (*fenc != NUL && !read_stdin && !read_fifo && !keep_dest_enc);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001125
1126 if (!skip_read)
1127 {
1128 linerest = 0;
1129 filesize = 0;
1130 skip_count = lines_to_skip;
1131 read_count = lines_to_read;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001132 conv_restlen = 0;
Bram Moolenaar55debbe2010-05-23 23:34:36 +02001133#ifdef FEAT_PERSISTENT_UNDO
Bram Moolenaar59f931e2010-07-24 20:27:03 +02001134 read_undo_file = (newfile && (flags & READ_KEEP_UNDO) == 0
1135 && curbuf->b_ffname != NULL
1136 && curbuf->b_p_udf
1137 && !filtering
Bram Moolenaarf71d7b92016-08-09 22:14:05 +02001138 && !read_fifo
Bram Moolenaar59f931e2010-07-24 20:27:03 +02001139 && !read_stdin
1140 && !read_buffer);
Bram Moolenaar55debbe2010-05-23 23:34:36 +02001141 if (read_undo_file)
1142 sha256_start(&sha_ctx);
1143#endif
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02001144#ifdef FEAT_CRYPT
1145 if (curbuf->b_cryptstate != NULL)
1146 {
1147 /* Need to free the state, but keep the key, don't want to ask for
1148 * it again. */
1149 crypt_free_state(curbuf->b_cryptstate);
1150 curbuf->b_cryptstate = NULL;
1151 }
1152#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001153 }
1154
1155 while (!error && !got_int)
1156 {
1157 /*
1158 * We allocate as much space for the file as we can get, plus
1159 * space for the old line plus room for one terminating NUL.
1160 * The amount is limited by the fact that read() only can read
1161 * upto max_unsigned characters (and other things).
1162 */
Bram Moolenaar13d3b052018-04-29 13:34:47 +02001163 if (!skip_read)
1164 {
Bram Moolenaar30276f22019-01-24 17:59:39 +01001165#if defined(SSIZE_MAX) && (SSIZE_MAX < 0x10000L)
Bram Moolenaar13d3b052018-04-29 13:34:47 +02001166 size = SSIZE_MAX; /* use max I/O size, 52K */
Bram Moolenaar30276f22019-01-24 17:59:39 +01001167#else
Bram Moolenaar13d3b052018-04-29 13:34:47 +02001168 /* Use buffer >= 64K. Add linerest to double the size if the
1169 * line gets very long, to avoid a lot of copying. But don't
1170 * read more than 1 Mbyte at a time, so we can be interrupted.
1171 */
1172 size = 0x10000L + linerest;
1173 if (size > 0x100000L)
1174 size = 0x100000L;
Bram Moolenaar13d3b052018-04-29 13:34:47 +02001175#endif
1176 }
1177
1178 /* Protect against the argument of lalloc() going negative. */
Bram Moolenaar30276f22019-01-24 17:59:39 +01001179 if (size < 0 || size + linerest + 1 < 0 || linerest >= MAXCOL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001180 {
1181 ++split;
1182 *ptr = NL; /* split line by inserting a NL */
1183 size = 1;
1184 }
1185 else
Bram Moolenaar071d4272004-06-13 20:20:40 +00001186 {
1187 if (!skip_read)
1188 {
Bram Moolenaarc1e37902006-04-18 21:55:01 +00001189 for ( ; size >= 10; size = (long)((long_u)size >> 1))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001190 {
1191 if ((new_buffer = lalloc((long_u)(size + linerest + 1),
1192 FALSE)) != NULL)
1193 break;
1194 }
1195 if (new_buffer == NULL)
1196 {
1197 do_outofmem_msg((long_u)(size * 2 + linerest + 1));
1198 error = TRUE;
1199 break;
1200 }
1201 if (linerest) /* copy characters from the previous buffer */
1202 mch_memmove(new_buffer, ptr - linerest, (size_t)linerest);
1203 vim_free(buffer);
1204 buffer = new_buffer;
1205 ptr = buffer + linerest;
1206 line_start = buffer;
1207
Bram Moolenaar071d4272004-06-13 20:20:40 +00001208 /* May need room to translate into.
1209 * For iconv() we don't really know the required space, use a
1210 * factor ICONV_MULT.
1211 * latin1 to utf-8: 1 byte becomes up to 2 bytes
1212 * utf-16 to utf-8: 2 bytes become up to 3 bytes, 4 bytes
1213 * become up to 4 bytes, size must be multiple of 2
1214 * ucs-2 to utf-8: 2 bytes become up to 3 bytes, size must be
1215 * multiple of 2
1216 * ucs-4 to utf-8: 4 bytes become up to 6 bytes, size must be
1217 * multiple of 4 */
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00001218 real_size = (int)size;
Bram Moolenaar13505972019-01-24 15:04:48 +01001219#ifdef USE_ICONV
Bram Moolenaar071d4272004-06-13 20:20:40 +00001220 if (iconv_fd != (iconv_t)-1)
1221 size = size / ICONV_MULT;
1222 else
Bram Moolenaar13505972019-01-24 15:04:48 +01001223#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001224 if (fio_flags & FIO_LATIN1)
1225 size = size / 2;
1226 else if (fio_flags & (FIO_UCS2 | FIO_UTF16))
1227 size = (size * 2 / 3) & ~1;
1228 else if (fio_flags & FIO_UCS4)
1229 size = (size * 2 / 3) & ~3;
1230 else if (fio_flags == FIO_UCSBOM)
1231 size = size / ICONV_MULT; /* worst case */
Bram Moolenaar4f974752019-02-17 17:44:42 +01001232#ifdef MSWIN
Bram Moolenaar071d4272004-06-13 20:20:40 +00001233 else if (fio_flags & FIO_CODEPAGE)
1234 size = size / ICONV_MULT; /* also worst case */
Bram Moolenaar13505972019-01-24 15:04:48 +01001235#endif
1236#ifdef MACOS_CONVERT
Bram Moolenaar071d4272004-06-13 20:20:40 +00001237 else if (fio_flags & FIO_MACROMAN)
1238 size = size / ICONV_MULT; /* also worst case */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001239#endif
1240
Bram Moolenaar071d4272004-06-13 20:20:40 +00001241 if (conv_restlen > 0)
1242 {
1243 /* Insert unconverted bytes from previous line. */
1244 mch_memmove(ptr, conv_rest, conv_restlen);
1245 ptr += conv_restlen;
1246 size -= conv_restlen;
1247 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001248
1249 if (read_buffer)
1250 {
1251 /*
1252 * Read bytes from curbuf. Used for converting text read
1253 * from stdin.
1254 */
1255 if (read_buf_lnum > from)
1256 size = 0;
1257 else
1258 {
1259 int n, ni;
1260 long tlen;
1261
1262 tlen = 0;
1263 for (;;)
1264 {
1265 p = ml_get(read_buf_lnum) + read_buf_col;
1266 n = (int)STRLEN(p);
1267 if ((int)tlen + n + 1 > size)
1268 {
1269 /* Filled up to "size", append partial line.
1270 * Change NL to NUL to reverse the effect done
1271 * below. */
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00001272 n = (int)(size - tlen);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001273 for (ni = 0; ni < n; ++ni)
1274 {
1275 if (p[ni] == NL)
1276 ptr[tlen++] = NUL;
1277 else
1278 ptr[tlen++] = p[ni];
1279 }
1280 read_buf_col += n;
1281 break;
1282 }
1283 else
1284 {
1285 /* Append whole line and new-line. Change NL
1286 * to NUL to reverse the effect done below. */
1287 for (ni = 0; ni < n; ++ni)
1288 {
1289 if (p[ni] == NL)
1290 ptr[tlen++] = NUL;
1291 else
1292 ptr[tlen++] = p[ni];
1293 }
1294 ptr[tlen++] = NL;
1295 read_buf_col = 0;
1296 if (++read_buf_lnum > from)
1297 {
1298 /* When the last line didn't have an
1299 * end-of-line don't add it now either. */
1300 if (!curbuf->b_p_eol)
1301 --tlen;
1302 size = tlen;
1303 break;
1304 }
1305 }
1306 }
1307 }
1308 }
1309 else
1310 {
1311 /*
1312 * Read bytes from the file.
1313 */
Bram Moolenaar540fc6f2010-12-17 16:27:16 +01001314 size = read_eintr(fd, ptr, size);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001315 }
1316
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02001317#ifdef FEAT_CRYPT
1318 /*
1319 * At start of file: Check for magic number of encryption.
1320 */
1321 if (filesize == 0 && size > 0)
1322 cryptkey = check_for_cryptkey(cryptkey, ptr, &size,
1323 &filesize, newfile, sfname,
1324 &did_ask_for_key);
1325 /*
1326 * Decrypt the read bytes. This is done before checking for
1327 * EOF because the crypt layer may be buffering.
1328 */
Bram Moolenaar829aa642017-08-23 22:32:35 +02001329 if (cryptkey != NULL && curbuf->b_cryptstate != NULL
1330 && size > 0)
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02001331 {
Bram Moolenaar987411d2019-01-18 22:48:34 +01001332# ifdef CRYPT_NOT_INPLACE
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02001333 if (crypt_works_inplace(curbuf->b_cryptstate))
1334 {
Bram Moolenaar987411d2019-01-18 22:48:34 +01001335# endif
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02001336 crypt_decode_inplace(curbuf->b_cryptstate, ptr, size);
Bram Moolenaar987411d2019-01-18 22:48:34 +01001337# ifdef CRYPT_NOT_INPLACE
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02001338 }
1339 else
1340 {
1341 char_u *newptr = NULL;
1342 int decrypted_size;
1343
1344 decrypted_size = crypt_decode_alloc(
1345 curbuf->b_cryptstate, ptr, size, &newptr);
1346
1347 /* If the crypt layer is buffering, not producing
1348 * anything yet, need to read more. */
Bram Moolenaar1c17ffa2018-04-24 15:19:04 +02001349 if (decrypted_size == 0)
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02001350 continue;
1351
1352 if (linerest == 0)
1353 {
1354 /* Simple case: reuse returned buffer (may be
1355 * NULL, checked later). */
1356 new_buffer = newptr;
1357 }
1358 else
1359 {
1360 long_u new_size;
1361
1362 /* Need new buffer to add bytes carried over. */
1363 new_size = (long_u)(decrypted_size + linerest + 1);
1364 new_buffer = lalloc(new_size, FALSE);
1365 if (new_buffer == NULL)
1366 {
1367 do_outofmem_msg(new_size);
1368 error = TRUE;
1369 break;
1370 }
1371
1372 mch_memmove(new_buffer, buffer, linerest);
1373 if (newptr != NULL)
1374 mch_memmove(new_buffer + linerest, newptr,
1375 decrypted_size);
1376 }
1377
1378 if (new_buffer != NULL)
1379 {
1380 vim_free(buffer);
1381 buffer = new_buffer;
1382 new_buffer = NULL;
1383 line_start = buffer;
1384 ptr = buffer + linerest;
1385 }
1386 size = decrypted_size;
1387 }
Bram Moolenaar987411d2019-01-18 22:48:34 +01001388# endif
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02001389 }
1390#endif
1391
Bram Moolenaar071d4272004-06-13 20:20:40 +00001392 if (size <= 0)
1393 {
1394 if (size < 0) /* read error */
1395 error = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001396 else if (conv_restlen > 0)
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00001397 {
Bram Moolenaarf453d352008-06-04 17:37:34 +00001398 /*
1399 * Reached end-of-file but some trailing bytes could
1400 * not be converted. Truncated file?
1401 */
1402
1403 /* When we did a conversion report an error. */
1404 if (fio_flags != 0
Bram Moolenaar13505972019-01-24 15:04:48 +01001405#ifdef USE_ICONV
Bram Moolenaarf453d352008-06-04 17:37:34 +00001406 || iconv_fd != (iconv_t)-1
Bram Moolenaar13505972019-01-24 15:04:48 +01001407#endif
Bram Moolenaarf453d352008-06-04 17:37:34 +00001408 )
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00001409 {
Bram Moolenaare8d95302013-04-24 16:34:02 +02001410 if (can_retry)
1411 goto rewind_retry;
Bram Moolenaarf453d352008-06-04 17:37:34 +00001412 if (conv_error == 0)
1413 conv_error = curbuf->b_ml.ml_line_count
1414 - linecnt + 1;
1415 }
1416 /* Remember the first linenr with an illegal byte */
1417 else if (illegal_byte == 0)
1418 illegal_byte = curbuf->b_ml.ml_line_count
1419 - linecnt + 1;
1420 if (bad_char_behavior == BAD_DROP)
1421 {
1422 *(ptr - conv_restlen) = NUL;
1423 conv_restlen = 0;
1424 }
1425 else
1426 {
1427 /* Replace the trailing bytes with the replacement
1428 * character if we were converting; if we weren't,
1429 * leave the UTF8 checking code to do it, as it
1430 * works slightly differently. */
1431 if (bad_char_behavior != BAD_KEEP && (fio_flags != 0
Bram Moolenaar13505972019-01-24 15:04:48 +01001432#ifdef USE_ICONV
Bram Moolenaarf453d352008-06-04 17:37:34 +00001433 || iconv_fd != (iconv_t)-1
Bram Moolenaar13505972019-01-24 15:04:48 +01001434#endif
Bram Moolenaarf453d352008-06-04 17:37:34 +00001435 ))
1436 {
1437 while (conv_restlen > 0)
1438 {
1439 *(--ptr) = bad_char_behavior;
1440 --conv_restlen;
1441 }
1442 }
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00001443 fio_flags = 0; /* don't convert this */
Bram Moolenaar13505972019-01-24 15:04:48 +01001444#ifdef USE_ICONV
Bram Moolenaarb21e5842006-04-16 18:30:08 +00001445 if (iconv_fd != (iconv_t)-1)
1446 {
1447 iconv_close(iconv_fd);
1448 iconv_fd = (iconv_t)-1;
1449 }
Bram Moolenaar13505972019-01-24 15:04:48 +01001450#endif
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00001451 }
1452 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001453 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001454 }
1455 skip_read = FALSE;
1456
Bram Moolenaar071d4272004-06-13 20:20:40 +00001457 /*
1458 * At start of file (or after crypt magic number): Check for BOM.
1459 * Also check for a BOM for other Unicode encodings, but not after
1460 * converting with 'charconvert' or when a BOM has already been
1461 * found.
1462 */
1463 if ((filesize == 0
Bram Moolenaar13505972019-01-24 15:04:48 +01001464#ifdef FEAT_CRYPT
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02001465 || (cryptkey != NULL
1466 && filesize == crypt_get_header_len(
1467 crypt_get_method_nr(curbuf)))
Bram Moolenaar13505972019-01-24 15:04:48 +01001468#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001469 )
1470 && (fio_flags == FIO_UCSBOM
1471 || (!curbuf->b_p_bomb
1472 && tmpname == NULL
1473 && (*fenc == 'u' || (*fenc == NUL && enc_utf8)))))
1474 {
1475 char_u *ccname;
1476 int blen;
1477
1478 /* no BOM detection in a short file or in binary mode */
1479 if (size < 2 || curbuf->b_p_bin)
1480 ccname = NULL;
1481 else
1482 ccname = check_for_bom(ptr, size, &blen,
1483 fio_flags == FIO_UCSBOM ? FIO_ALL : get_fio_flags(fenc));
1484 if (ccname != NULL)
1485 {
1486 /* Remove BOM from the text */
1487 filesize += blen;
1488 size -= blen;
1489 mch_memmove(ptr, ptr + blen, (size_t)size);
Bram Moolenaar910f66f2006-04-05 20:41:53 +00001490 if (set_options)
Bram Moolenaar83eb8852007-08-12 13:51:26 +00001491 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00001492 curbuf->b_p_bomb = TRUE;
Bram Moolenaar83eb8852007-08-12 13:51:26 +00001493 curbuf->b_start_bomb = TRUE;
1494 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001495 }
1496
1497 if (fio_flags == FIO_UCSBOM)
1498 {
1499 if (ccname == NULL)
1500 {
1501 /* No BOM detected: retry with next encoding. */
1502 advance_fenc = TRUE;
1503 }
1504 else
1505 {
1506 /* BOM detected: set "fenc" and jump back */
1507 if (fenc_alloced)
1508 vim_free(fenc);
1509 fenc = ccname;
1510 fenc_alloced = FALSE;
1511 }
1512 /* retry reading without getting new bytes or rewinding */
1513 skip_read = TRUE;
1514 goto retry;
1515 }
1516 }
Bram Moolenaarf453d352008-06-04 17:37:34 +00001517
1518 /* Include not converted bytes. */
1519 ptr -= conv_restlen;
1520 size += conv_restlen;
1521 conv_restlen = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001522 /*
1523 * Break here for a read error or end-of-file.
1524 */
1525 if (size <= 0)
1526 break;
1527
Bram Moolenaar071d4272004-06-13 20:20:40 +00001528
Bram Moolenaar13505972019-01-24 15:04:48 +01001529#ifdef USE_ICONV
Bram Moolenaar071d4272004-06-13 20:20:40 +00001530 if (iconv_fd != (iconv_t)-1)
1531 {
1532 /*
1533 * Attempt conversion of the read bytes to 'encoding' using
1534 * iconv().
1535 */
1536 const char *fromp;
1537 char *top;
1538 size_t from_size;
1539 size_t to_size;
1540
1541 fromp = (char *)ptr;
1542 from_size = size;
1543 ptr += size;
1544 top = (char *)ptr;
1545 to_size = real_size - size;
1546
1547 /*
1548 * If there is conversion error or not enough room try using
Bram Moolenaar19a09a12005-03-04 23:39:37 +00001549 * another conversion. Except for when there is no
1550 * alternative (help files).
Bram Moolenaar071d4272004-06-13 20:20:40 +00001551 */
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001552 while ((iconv(iconv_fd, (void *)&fromp, &from_size,
1553 &top, &to_size)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001554 == (size_t)-1 && ICONV_ERRNO != ICONV_EINVAL)
1555 || from_size > CONV_RESTLEN)
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001556 {
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00001557 if (can_retry)
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001558 goto rewind_retry;
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00001559 if (conv_error == 0)
1560 conv_error = readfile_linenr(linecnt,
1561 ptr, (char_u *)top);
Bram Moolenaar42eeac32005-06-29 22:40:58 +00001562
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00001563 /* Deal with a bad byte and continue with the next. */
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001564 ++fromp;
1565 --from_size;
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00001566 if (bad_char_behavior == BAD_KEEP)
1567 {
1568 *top++ = *(fromp - 1);
1569 --to_size;
1570 }
1571 else if (bad_char_behavior != BAD_DROP)
1572 {
1573 *top++ = bad_char_behavior;
1574 --to_size;
1575 }
Bram Moolenaar8cd06ca2005-02-28 22:44:58 +00001576 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001577
1578 if (from_size > 0)
1579 {
1580 /* Some remaining characters, keep them for the next
1581 * round. */
1582 mch_memmove(conv_rest, (char_u *)fromp, from_size);
1583 conv_restlen = (int)from_size;
1584 }
1585
1586 /* move the linerest to before the converted characters */
1587 line_start = ptr - linerest;
1588 mch_memmove(line_start, buffer, (size_t)linerest);
1589 size = (long)((char_u *)top - ptr);
1590 }
Bram Moolenaar13505972019-01-24 15:04:48 +01001591#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001592
Bram Moolenaar4f974752019-02-17 17:44:42 +01001593#ifdef MSWIN
Bram Moolenaar071d4272004-06-13 20:20:40 +00001594 if (fio_flags & FIO_CODEPAGE)
1595 {
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00001596 char_u *src, *dst;
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00001597 WCHAR ucs2buf[3];
1598 int ucs2len;
1599 int codepage = FIO_GET_CP(fio_flags);
1600 int bytelen;
1601 int found_bad;
1602 char replstr[2];
1603
Bram Moolenaar071d4272004-06-13 20:20:40 +00001604 /*
1605 * Conversion from an MS-Windows codepage or UTF-8 to UTF-8 or
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00001606 * a codepage, using standard MS-Windows functions. This
1607 * requires two steps:
1608 * 1. convert from 'fileencoding' to ucs-2
1609 * 2. convert from ucs-2 to 'encoding'
Bram Moolenaar071d4272004-06-13 20:20:40 +00001610 *
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00001611 * Because there may be illegal bytes AND an incomplete byte
1612 * sequence at the end, we may have to do the conversion one
1613 * character at a time to get it right.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001614 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001615
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00001616 /* Replacement string for WideCharToMultiByte(). */
1617 if (bad_char_behavior > 0)
1618 replstr[0] = bad_char_behavior;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001619 else
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00001620 replstr[0] = '?';
1621 replstr[1] = NUL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001622
1623 /*
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00001624 * Move the bytes to the end of the buffer, so that we have
1625 * room to put the result at the start.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001626 */
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00001627 src = ptr + real_size - size;
1628 mch_memmove(src, ptr, size);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001629
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00001630 /*
1631 * Do the conversion.
1632 */
1633 dst = ptr;
1634 size = size;
1635 while (size > 0)
1636 {
1637 found_bad = FALSE;
1638
1639# ifdef CP_UTF8 /* VC 4.1 doesn't define CP_UTF8 */
1640 if (codepage == CP_UTF8)
1641 {
1642 /* Handle CP_UTF8 input ourselves to be able to handle
1643 * trailing bytes properly.
1644 * Get one UTF-8 character from src. */
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00001645 bytelen = (int)utf_ptr2len_len(src, size);
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00001646 if (bytelen > size)
1647 {
1648 /* Only got some bytes of a character. Normally
1649 * it's put in "conv_rest", but if it's too long
1650 * deal with it as if they were illegal bytes. */
1651 if (bytelen <= CONV_RESTLEN)
1652 break;
1653
1654 /* weird overlong byte sequence */
1655 bytelen = size;
1656 found_bad = TRUE;
1657 }
1658 else
1659 {
Bram Moolenaarc01140a2006-03-24 22:21:52 +00001660 int u8c = utf_ptr2char(src);
1661
Bram Moolenaar86e01082005-12-29 22:45:34 +00001662 if (u8c > 0xffff || (*src >= 0x80 && bytelen == 1))
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00001663 found_bad = TRUE;
1664 ucs2buf[0] = u8c;
1665 ucs2len = 1;
1666 }
1667 }
1668 else
1669# endif
1670 {
1671 /* We don't know how long the byte sequence is, try
1672 * from one to three bytes. */
1673 for (bytelen = 1; bytelen <= size && bytelen <= 3;
1674 ++bytelen)
1675 {
1676 ucs2len = MultiByteToWideChar(codepage,
1677 MB_ERR_INVALID_CHARS,
1678 (LPCSTR)src, bytelen,
1679 ucs2buf, 3);
1680 if (ucs2len > 0)
1681 break;
1682 }
1683 if (ucs2len == 0)
1684 {
1685 /* If we have only one byte then it's probably an
1686 * incomplete byte sequence. Otherwise discard
1687 * one byte as a bad character. */
1688 if (size == 1)
1689 break;
1690 found_bad = TRUE;
1691 bytelen = 1;
1692 }
1693 }
1694
1695 if (!found_bad)
1696 {
1697 int i;
1698
1699 /* Convert "ucs2buf[ucs2len]" to 'enc' in "dst". */
1700 if (enc_utf8)
1701 {
1702 /* From UCS-2 to UTF-8. Cannot fail. */
1703 for (i = 0; i < ucs2len; ++i)
1704 dst += utf_char2bytes(ucs2buf[i], dst);
1705 }
1706 else
1707 {
1708 BOOL bad = FALSE;
1709 int dstlen;
1710
1711 /* From UCS-2 to "enc_codepage". If the
1712 * conversion uses the default character "?",
1713 * the data doesn't fit in this encoding. */
1714 dstlen = WideCharToMultiByte(enc_codepage, 0,
1715 (LPCWSTR)ucs2buf, ucs2len,
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00001716 (LPSTR)dst, (int)(src - dst),
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00001717 replstr, &bad);
1718 if (bad)
1719 found_bad = TRUE;
1720 else
1721 dst += dstlen;
1722 }
1723 }
1724
1725 if (found_bad)
1726 {
1727 /* Deal with bytes we can't convert. */
1728 if (can_retry)
1729 goto rewind_retry;
1730 if (conv_error == 0)
1731 conv_error = readfile_linenr(linecnt, ptr, dst);
1732 if (bad_char_behavior != BAD_DROP)
1733 {
1734 if (bad_char_behavior == BAD_KEEP)
1735 {
1736 mch_memmove(dst, src, bytelen);
1737 dst += bytelen;
1738 }
1739 else
1740 *dst++ = bad_char_behavior;
1741 }
1742 }
1743
1744 src += bytelen;
1745 size -= bytelen;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001746 }
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00001747
1748 if (size > 0)
1749 {
1750 /* An incomplete byte sequence remaining. */
1751 mch_memmove(conv_rest, src, size);
1752 conv_restlen = size;
1753 }
1754
1755 /* The new size is equal to how much "dst" was advanced. */
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00001756 size = (long)(dst - ptr);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001757 }
1758 else
Bram Moolenaar13505972019-01-24 15:04:48 +01001759#endif
1760#ifdef MACOS_CONVERT
Bram Moolenaar071d4272004-06-13 20:20:40 +00001761 if (fio_flags & FIO_MACROMAN)
1762 {
1763 /*
1764 * Conversion from Apple MacRoman char encoding to UTF-8 or
Bram Moolenaarab79bcb2004-07-18 21:34:53 +00001765 * latin1. This is in os_mac_conv.c.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001766 */
Bram Moolenaarab79bcb2004-07-18 21:34:53 +00001767 if (macroman2enc(ptr, &size, real_size) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001768 goto rewind_retry;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001769 }
1770 else
Bram Moolenaar13505972019-01-24 15:04:48 +01001771#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001772 if (fio_flags != 0)
1773 {
1774 int u8c;
1775 char_u *dest;
1776 char_u *tail = NULL;
1777
1778 /*
1779 * "enc_utf8" set: Convert Unicode or Latin1 to UTF-8.
1780 * "enc_utf8" not set: Convert Unicode to Latin1.
1781 * Go from end to start through the buffer, because the number
1782 * of bytes may increase.
1783 * "dest" points to after where the UTF-8 bytes go, "p" points
1784 * to after the next character to convert.
1785 */
1786 dest = ptr + real_size;
1787 if (fio_flags == FIO_LATIN1 || fio_flags == FIO_UTF8)
1788 {
1789 p = ptr + size;
1790 if (fio_flags == FIO_UTF8)
1791 {
1792 /* Check for a trailing incomplete UTF-8 sequence */
1793 tail = ptr + size - 1;
1794 while (tail > ptr && (*tail & 0xc0) == 0x80)
1795 --tail;
1796 if (tail + utf_byte2len(*tail) <= ptr + size)
1797 tail = NULL;
1798 else
1799 p = tail;
1800 }
1801 }
1802 else if (fio_flags & (FIO_UCS2 | FIO_UTF16))
1803 {
1804 /* Check for a trailing byte */
1805 p = ptr + (size & ~1);
1806 if (size & 1)
1807 tail = p;
1808 if ((fio_flags & FIO_UTF16) && p > ptr)
1809 {
1810 /* Check for a trailing leading word */
1811 if (fio_flags & FIO_ENDIAN_L)
1812 {
1813 u8c = (*--p << 8);
1814 u8c += *--p;
1815 }
1816 else
1817 {
1818 u8c = *--p;
1819 u8c += (*--p << 8);
1820 }
1821 if (u8c >= 0xd800 && u8c <= 0xdbff)
1822 tail = p;
1823 else
1824 p += 2;
1825 }
1826 }
1827 else /* FIO_UCS4 */
1828 {
1829 /* Check for trailing 1, 2 or 3 bytes */
1830 p = ptr + (size & ~3);
1831 if (size & 3)
1832 tail = p;
1833 }
1834
1835 /* If there is a trailing incomplete sequence move it to
1836 * conv_rest[]. */
1837 if (tail != NULL)
1838 {
1839 conv_restlen = (int)((ptr + size) - tail);
1840 mch_memmove(conv_rest, (char_u *)tail, conv_restlen);
1841 size -= conv_restlen;
1842 }
1843
1844
1845 while (p > ptr)
1846 {
1847 if (fio_flags & FIO_LATIN1)
1848 u8c = *--p;
1849 else if (fio_flags & (FIO_UCS2 | FIO_UTF16))
1850 {
1851 if (fio_flags & FIO_ENDIAN_L)
1852 {
1853 u8c = (*--p << 8);
1854 u8c += *--p;
1855 }
1856 else
1857 {
1858 u8c = *--p;
1859 u8c += (*--p << 8);
1860 }
1861 if ((fio_flags & FIO_UTF16)
1862 && u8c >= 0xdc00 && u8c <= 0xdfff)
1863 {
1864 int u16c;
1865
1866 if (p == ptr)
1867 {
1868 /* Missing leading word. */
1869 if (can_retry)
1870 goto rewind_retry;
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00001871 if (conv_error == 0)
1872 conv_error = readfile_linenr(linecnt,
1873 ptr, p);
1874 if (bad_char_behavior == BAD_DROP)
1875 continue;
1876 if (bad_char_behavior != BAD_KEEP)
1877 u8c = bad_char_behavior;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001878 }
1879
1880 /* found second word of double-word, get the first
1881 * word and compute the resulting character */
1882 if (fio_flags & FIO_ENDIAN_L)
1883 {
1884 u16c = (*--p << 8);
1885 u16c += *--p;
1886 }
1887 else
1888 {
1889 u16c = *--p;
1890 u16c += (*--p << 8);
1891 }
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00001892 u8c = 0x10000 + ((u16c & 0x3ff) << 10)
1893 + (u8c & 0x3ff);
1894
Bram Moolenaar071d4272004-06-13 20:20:40 +00001895 /* Check if the word is indeed a leading word. */
1896 if (u16c < 0xd800 || u16c > 0xdbff)
1897 {
1898 if (can_retry)
1899 goto rewind_retry;
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00001900 if (conv_error == 0)
1901 conv_error = readfile_linenr(linecnt,
1902 ptr, p);
1903 if (bad_char_behavior == BAD_DROP)
1904 continue;
1905 if (bad_char_behavior != BAD_KEEP)
1906 u8c = bad_char_behavior;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001907 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001908 }
1909 }
1910 else if (fio_flags & FIO_UCS4)
1911 {
1912 if (fio_flags & FIO_ENDIAN_L)
1913 {
Bram Moolenaardc1c9812017-10-27 22:15:24 +02001914 u8c = (unsigned)*--p << 24;
1915 u8c += (unsigned)*--p << 16;
1916 u8c += (unsigned)*--p << 8;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001917 u8c += *--p;
1918 }
1919 else /* big endian */
1920 {
1921 u8c = *--p;
Bram Moolenaardc1c9812017-10-27 22:15:24 +02001922 u8c += (unsigned)*--p << 8;
1923 u8c += (unsigned)*--p << 16;
1924 u8c += (unsigned)*--p << 24;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001925 }
1926 }
1927 else /* UTF-8 */
1928 {
1929 if (*--p < 0x80)
1930 u8c = *p;
1931 else
1932 {
1933 len = utf_head_off(ptr, p);
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00001934 p -= len;
1935 u8c = utf_ptr2char(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001936 if (len == 0)
1937 {
1938 /* Not a valid UTF-8 character, retry with
1939 * another fenc when possible, otherwise just
1940 * report the error. */
1941 if (can_retry)
1942 goto rewind_retry;
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00001943 if (conv_error == 0)
1944 conv_error = readfile_linenr(linecnt,
1945 ptr, p);
1946 if (bad_char_behavior == BAD_DROP)
1947 continue;
1948 if (bad_char_behavior != BAD_KEEP)
1949 u8c = bad_char_behavior;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001950 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001951 }
1952 }
1953 if (enc_utf8) /* produce UTF-8 */
1954 {
1955 dest -= utf_char2len(u8c);
1956 (void)utf_char2bytes(u8c, dest);
1957 }
1958 else /* produce Latin1 */
1959 {
1960 --dest;
1961 if (u8c >= 0x100)
1962 {
1963 /* character doesn't fit in latin1, retry with
1964 * another fenc when possible, otherwise just
1965 * report the error. */
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00001966 if (can_retry)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001967 goto rewind_retry;
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00001968 if (conv_error == 0)
1969 conv_error = readfile_linenr(linecnt, ptr, p);
1970 if (bad_char_behavior == BAD_DROP)
1971 ++dest;
1972 else if (bad_char_behavior == BAD_KEEP)
1973 *dest = u8c;
1974 else if (eap != NULL && eap->bad_char != 0)
1975 *dest = bad_char_behavior;
1976 else
1977 *dest = 0xBF;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001978 }
1979 else
1980 *dest = u8c;
1981 }
1982 }
1983
1984 /* move the linerest to before the converted characters */
1985 line_start = dest - linerest;
1986 mch_memmove(line_start, buffer, (size_t)linerest);
1987 size = (long)((ptr + real_size) - dest);
1988 ptr = dest;
1989 }
Bram Moolenaarf453d352008-06-04 17:37:34 +00001990 else if (enc_utf8 && !curbuf->b_p_bin)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001991 {
Bram Moolenaarf453d352008-06-04 17:37:34 +00001992 int incomplete_tail = FALSE;
1993
1994 /* Reading UTF-8: Check if the bytes are valid UTF-8. */
1995 for (p = ptr; ; ++p)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001996 {
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00001997 int todo = (int)((ptr + size) - p);
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00001998 int l;
1999
2000 if (todo <= 0)
2001 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002002 if (*p >= 0x80)
2003 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00002004 /* A length of 1 means it's an illegal byte. Accept
2005 * an incomplete character at the end though, the next
2006 * read() will get the next bytes, we'll check it
2007 * then. */
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00002008 l = utf_ptr2len_len(p, todo);
Bram Moolenaarf453d352008-06-04 17:37:34 +00002009 if (l > todo && !incomplete_tail)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002010 {
Bram Moolenaarf453d352008-06-04 17:37:34 +00002011 /* Avoid retrying with a different encoding when
2012 * a truncated file is more likely, or attempting
2013 * to read the rest of an incomplete sequence when
2014 * we have already done so. */
2015 if (p > ptr || filesize > 0)
2016 incomplete_tail = TRUE;
2017 /* Incomplete byte sequence, move it to conv_rest[]
2018 * and try to read the rest of it, unless we've
2019 * already done so. */
2020 if (p > ptr)
2021 {
2022 conv_restlen = todo;
2023 mch_memmove(conv_rest, p, conv_restlen);
2024 size -= conv_restlen;
2025 break;
2026 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002027 }
Bram Moolenaarf453d352008-06-04 17:37:34 +00002028 if (l == 1 || l > todo)
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00002029 {
2030 /* Illegal byte. If we can try another encoding
Bram Moolenaarf453d352008-06-04 17:37:34 +00002031 * do that, unless at EOF where a truncated
2032 * file is more likely than a conversion error. */
2033 if (can_retry && !incomplete_tail)
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00002034 break;
Bram Moolenaar13505972019-01-24 15:04:48 +01002035#ifdef USE_ICONV
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00002036 /* When we did a conversion report an error. */
2037 if (iconv_fd != (iconv_t)-1 && conv_error == 0)
2038 conv_error = readfile_linenr(linecnt, ptr, p);
Bram Moolenaar13505972019-01-24 15:04:48 +01002039#endif
Bram Moolenaarf453d352008-06-04 17:37:34 +00002040 /* Remember the first linenr with an illegal byte */
2041 if (conv_error == 0 && illegal_byte == 0)
2042 illegal_byte = readfile_linenr(linecnt, ptr, p);
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00002043
2044 /* Drop, keep or replace the bad byte. */
2045 if (bad_char_behavior == BAD_DROP)
2046 {
Bram Moolenaarf453d352008-06-04 17:37:34 +00002047 mch_memmove(p, p + 1, todo - 1);
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00002048 --p;
2049 --size;
2050 }
2051 else if (bad_char_behavior != BAD_KEEP)
2052 *p = bad_char_behavior;
2053 }
Bram Moolenaarf453d352008-06-04 17:37:34 +00002054 else
2055 p += l - 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002056 }
2057 }
Bram Moolenaarf453d352008-06-04 17:37:34 +00002058 if (p < ptr + size && !incomplete_tail)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002059 {
2060 /* Detected a UTF-8 error. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002061rewind_retry:
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00002062 /* Retry reading with another conversion. */
Bram Moolenaar13505972019-01-24 15:04:48 +01002063#if defined(FEAT_EVAL) && defined(USE_ICONV)
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00002064 if (*p_ccv != NUL && iconv_fd != (iconv_t)-1)
2065 /* iconv() failed, try 'charconvert' */
2066 did_iconv = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002067 else
Bram Moolenaar13505972019-01-24 15:04:48 +01002068#endif
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00002069 /* use next item from 'fileencodings' */
2070 advance_fenc = TRUE;
2071 file_rewind = TRUE;
2072 goto retry;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002073 }
2074 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002075
2076 /* count the number of characters (after conversion!) */
2077 filesize += size;
2078
2079 /*
2080 * when reading the first part of a file: guess EOL type
2081 */
2082 if (fileformat == EOL_UNKNOWN)
2083 {
2084 /* First try finding a NL, for Dos and Unix */
2085 if (try_dos || try_unix)
2086 {
Bram Moolenaarc6b72172015-02-27 17:48:09 +01002087 /* Reset the carriage return counter. */
2088 if (try_mac)
2089 try_mac = 1;
2090
Bram Moolenaar071d4272004-06-13 20:20:40 +00002091 for (p = ptr; p < ptr + size; ++p)
2092 {
2093 if (*p == NL)
2094 {
2095 if (!try_unix
2096 || (try_dos && p > ptr && p[-1] == CAR))
2097 fileformat = EOL_DOS;
2098 else
2099 fileformat = EOL_UNIX;
2100 break;
2101 }
Bram Moolenaar05eb6122015-02-17 14:15:19 +01002102 else if (*p == CAR && try_mac)
2103 try_mac++;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002104 }
2105
2106 /* Don't give in to EOL_UNIX if EOL_MAC is more likely */
2107 if (fileformat == EOL_UNIX && try_mac)
2108 {
2109 /* Need to reset the counters when retrying fenc. */
2110 try_mac = 1;
2111 try_unix = 1;
2112 for (; p >= ptr && *p != CAR; p--)
2113 ;
2114 if (p >= ptr)
2115 {
2116 for (p = ptr; p < ptr + size; ++p)
2117 {
2118 if (*p == NL)
2119 try_unix++;
2120 else if (*p == CAR)
2121 try_mac++;
2122 }
2123 if (try_mac > try_unix)
2124 fileformat = EOL_MAC;
2125 }
2126 }
Bram Moolenaar05eb6122015-02-17 14:15:19 +01002127 else if (fileformat == EOL_UNKNOWN && try_mac == 1)
2128 /* Looking for CR but found no end-of-line markers at
2129 * all: use the default format. */
2130 fileformat = default_fileformat();
Bram Moolenaar071d4272004-06-13 20:20:40 +00002131 }
2132
2133 /* No NL found: may use Mac format */
2134 if (fileformat == EOL_UNKNOWN && try_mac)
2135 fileformat = EOL_MAC;
2136
2137 /* Still nothing found? Use first format in 'ffs' */
2138 if (fileformat == EOL_UNKNOWN)
2139 fileformat = default_fileformat();
2140
2141 /* if editing a new file: may set p_tx and p_ff */
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002142 if (set_options)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002143 set_fileformat(fileformat, OPT_LOCAL);
2144 }
2145 }
2146
2147 /*
2148 * This loop is executed once for every character read.
2149 * Keep it fast!
2150 */
2151 if (fileformat == EOL_MAC)
2152 {
2153 --ptr;
2154 while (++ptr, --size >= 0)
2155 {
2156 /* catch most common case first */
2157 if ((c = *ptr) != NUL && c != CAR && c != NL)
2158 continue;
2159 if (c == NUL)
2160 *ptr = NL; /* NULs are replaced by newlines! */
2161 else if (c == NL)
2162 *ptr = CAR; /* NLs are replaced by CRs! */
2163 else
2164 {
2165 if (skip_count == 0)
2166 {
2167 *ptr = NUL; /* end of line */
2168 len = (colnr_T) (ptr - line_start + 1);
2169 if (ml_append(lnum, line_start, len, newfile) == FAIL)
2170 {
2171 error = TRUE;
2172 break;
2173 }
Bram Moolenaar55debbe2010-05-23 23:34:36 +02002174#ifdef FEAT_PERSISTENT_UNDO
2175 if (read_undo_file)
2176 sha256_update(&sha_ctx, line_start, len);
2177#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002178 ++lnum;
2179 if (--read_count == 0)
2180 {
2181 error = TRUE; /* break loop */
2182 line_start = ptr; /* nothing left to write */
2183 break;
2184 }
2185 }
2186 else
2187 --skip_count;
2188 line_start = ptr + 1;
2189 }
2190 }
2191 }
2192 else
2193 {
2194 --ptr;
2195 while (++ptr, --size >= 0)
2196 {
2197 if ((c = *ptr) != NUL && c != NL) /* catch most common case */
2198 continue;
2199 if (c == NUL)
2200 *ptr = NL; /* NULs are replaced by newlines! */
2201 else
2202 {
2203 if (skip_count == 0)
2204 {
2205 *ptr = NUL; /* end of line */
2206 len = (colnr_T)(ptr - line_start + 1);
2207 if (fileformat == EOL_DOS)
2208 {
Bram Moolenaar2aa5f692017-01-24 15:46:48 +01002209 if (ptr > line_start && ptr[-1] == CAR)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002210 {
Bram Moolenaar2aa5f692017-01-24 15:46:48 +01002211 /* remove CR before NL */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002212 ptr[-1] = NUL;
2213 --len;
2214 }
2215 /*
2216 * Reading in Dos format, but no CR-LF found!
2217 * When 'fileformats' includes "unix", delete all
2218 * the lines read so far and start all over again.
2219 * Otherwise give an error message later.
2220 */
2221 else if (ff_error != EOL_DOS)
2222 {
2223 if ( try_unix
2224 && !read_stdin
2225 && (read_buffer
Bram Moolenaar8767f522016-07-01 17:17:39 +02002226 || vim_lseek(fd, (off_T)0L, SEEK_SET)
2227 == 0))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002228 {
2229 fileformat = EOL_UNIX;
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002230 if (set_options)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002231 set_fileformat(EOL_UNIX, OPT_LOCAL);
2232 file_rewind = TRUE;
2233 keep_fileformat = TRUE;
2234 goto retry;
2235 }
2236 ff_error = EOL_DOS;
2237 }
2238 }
2239 if (ml_append(lnum, line_start, len, newfile) == FAIL)
2240 {
2241 error = TRUE;
2242 break;
2243 }
Bram Moolenaar55debbe2010-05-23 23:34:36 +02002244#ifdef FEAT_PERSISTENT_UNDO
2245 if (read_undo_file)
2246 sha256_update(&sha_ctx, line_start, len);
2247#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002248 ++lnum;
2249 if (--read_count == 0)
2250 {
2251 error = TRUE; /* break loop */
2252 line_start = ptr; /* nothing left to write */
2253 break;
2254 }
2255 }
2256 else
2257 --skip_count;
2258 line_start = ptr + 1;
2259 }
2260 }
2261 }
2262 linerest = (long)(ptr - line_start);
2263 ui_breakcheck();
2264 }
2265
2266failed:
2267 /* not an error, max. number of lines reached */
2268 if (error && read_count == 0)
2269 error = FALSE;
2270
2271 /*
2272 * If we get EOF in the middle of a line, note the fact and
2273 * complete the line ourselves.
2274 * In Dos format ignore a trailing CTRL-Z, unless 'binary' set.
2275 */
2276 if (!error
2277 && !got_int
2278 && linerest != 0
2279 && !(!curbuf->b_p_bin
2280 && fileformat == EOL_DOS
2281 && *line_start == Ctrl_Z
2282 && ptr == line_start + 1))
2283 {
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002284 /* remember for when writing */
2285 if (set_options)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002286 curbuf->b_p_eol = FALSE;
2287 *ptr = NUL;
Bram Moolenaar55debbe2010-05-23 23:34:36 +02002288 len = (colnr_T)(ptr - line_start + 1);
2289 if (ml_append(lnum, line_start, len, newfile) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002290 error = TRUE;
2291 else
Bram Moolenaar55debbe2010-05-23 23:34:36 +02002292 {
2293#ifdef FEAT_PERSISTENT_UNDO
2294 if (read_undo_file)
2295 sha256_update(&sha_ctx, line_start, len);
2296#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002297 read_no_eol_lnum = ++lnum;
Bram Moolenaar55debbe2010-05-23 23:34:36 +02002298 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002299 }
2300
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002301 if (set_options)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002302 save_file_ff(curbuf); /* remember the current file format */
2303
2304#ifdef FEAT_CRYPT
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02002305 if (curbuf->b_cryptstate != NULL)
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02002306 {
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02002307 crypt_free_state(curbuf->b_cryptstate);
2308 curbuf->b_cryptstate = NULL;
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02002309 }
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02002310 if (cryptkey != NULL && cryptkey != curbuf->b_p_key)
2311 crypt_free_key(cryptkey);
2312 /* Don't set cryptkey to NULL, it's used below as a flag that
2313 * encryption was used. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002314#endif
2315
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002316 /* If editing a new file: set 'fenc' for the current buffer.
2317 * Also for ":read ++edit file". */
2318 if (set_options)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002319 set_string_option_direct((char_u *)"fenc", -1, fenc,
Bram Moolenaar5e3cb7e2006-02-27 23:58:35 +00002320 OPT_FREE|OPT_LOCAL, 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002321 if (fenc_alloced)
2322 vim_free(fenc);
Bram Moolenaar13505972019-01-24 15:04:48 +01002323#ifdef USE_ICONV
Bram Moolenaar071d4272004-06-13 20:20:40 +00002324 if (iconv_fd != (iconv_t)-1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002325 iconv_close(iconv_fd);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002326#endif
2327
2328 if (!read_buffer && !read_stdin)
2329 close(fd); /* errors are ignored */
Bram Moolenaarf05da212009-11-17 16:13:15 +00002330#ifdef HAVE_FD_CLOEXEC
2331 else
2332 {
2333 int fdflags = fcntl(fd, F_GETFD);
2334 if (fdflags >= 0 && (fdflags & FD_CLOEXEC) == 0)
Bram Moolenaarfbc4b4d2016-02-07 15:14:01 +01002335 (void)fcntl(fd, F_SETFD, fdflags | FD_CLOEXEC);
Bram Moolenaarf05da212009-11-17 16:13:15 +00002336 }
2337#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002338 vim_free(buffer);
2339
2340#ifdef HAVE_DUP
2341 if (read_stdin)
2342 {
2343 /* Use stderr for stdin, makes shell commands work. */
2344 close(0);
Bram Moolenaar42335f52018-09-13 15:33:43 +02002345 vim_ignored = dup(2);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002346 }
2347#endif
2348
Bram Moolenaar071d4272004-06-13 20:20:40 +00002349 if (tmpname != NULL)
2350 {
2351 mch_remove(tmpname); /* delete converted file */
2352 vim_free(tmpname);
2353 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002354 --no_wait_return; /* may wait for return now */
2355
2356 /*
2357 * In recovery mode everything but autocommands is skipped.
2358 */
2359 if (!recoverymode)
2360 {
2361 /* need to delete the last line, which comes from the empty buffer */
2362 if (newfile && wasempty && !(curbuf->b_ml.ml_flags & ML_EMPTY))
2363 {
2364#ifdef FEAT_NETBEANS_INTG
2365 netbeansFireChanges = 0;
2366#endif
2367 ml_delete(curbuf->b_ml.ml_line_count, FALSE);
2368#ifdef FEAT_NETBEANS_INTG
2369 netbeansFireChanges = 1;
2370#endif
2371 --linecnt;
2372 }
2373 linecnt = curbuf->b_ml.ml_line_count - linecnt;
2374 if (filesize == 0)
2375 linecnt = 0;
2376 if (newfile || read_buffer)
Bram Moolenaar7263a772007-05-10 17:35:54 +00002377 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00002378 redraw_curbuf_later(NOT_VALID);
Bram Moolenaar7263a772007-05-10 17:35:54 +00002379#ifdef FEAT_DIFF
2380 /* After reading the text into the buffer the diff info needs to
2381 * be updated. */
2382 diff_invalidate(curbuf);
2383#endif
2384#ifdef FEAT_FOLDING
2385 /* All folds in the window are invalid now. Mark them for update
2386 * before triggering autocommands. */
2387 foldUpdateAll(curwin);
2388#endif
2389 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002390 else if (linecnt) /* appended at least one line */
2391 appended_lines_mark(from, linecnt);
2392
Bram Moolenaar071d4272004-06-13 20:20:40 +00002393#ifndef ALWAYS_USE_GUI
2394 /*
2395 * If we were reading from the same terminal as where messages go,
2396 * the screen will have been messed up.
2397 * Switch on raw mode now and clear the screen.
2398 */
2399 if (read_stdin)
2400 {
2401 settmode(TMODE_RAW); /* set to raw mode */
2402 starttermcap();
2403 screenclear();
2404 }
2405#endif
2406
2407 if (got_int)
2408 {
2409 if (!(flags & READ_DUMMY))
2410 {
2411 filemess(curbuf, sfname, (char_u *)_(e_interr), 0);
2412 if (newfile)
2413 curbuf->b_p_ro = TRUE; /* must use "w!" now */
2414 }
2415 msg_scroll = msg_save;
2416#ifdef FEAT_VIMINFO
2417 check_marks_read();
2418#endif
2419 return OK; /* an interrupt isn't really an error */
2420 }
2421
2422 if (!filtering && !(flags & READ_DUMMY))
2423 {
2424 msg_add_fname(curbuf, sfname); /* fname in IObuff with quotes */
2425 c = FALSE;
2426
2427#ifdef UNIX
Bram Moolenaard569bb02018-08-11 13:57:20 +02002428 if (S_ISFIFO(perm)) /* fifo */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002429 {
2430 STRCAT(IObuff, _("[fifo]"));
2431 c = TRUE;
2432 }
Bram Moolenaard569bb02018-08-11 13:57:20 +02002433 if (S_ISSOCK(perm)) /* or socket */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002434 {
2435 STRCAT(IObuff, _("[socket]"));
2436 c = TRUE;
2437 }
Bram Moolenaarfe1c56d2007-07-10 15:10:54 +00002438# ifdef OPEN_CHR_FILES
2439 if (S_ISCHR(perm)) /* or character special */
2440 {
2441 STRCAT(IObuff, _("[character special]"));
2442 c = TRUE;
2443 }
2444# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002445#endif
2446 if (curbuf->b_p_ro)
2447 {
2448 STRCAT(IObuff, shortmess(SHM_RO) ? _("[RO]") : _("[readonly]"));
2449 c = TRUE;
2450 }
2451 if (read_no_eol_lnum)
2452 {
2453 msg_add_eol();
2454 c = TRUE;
2455 }
2456 if (ff_error == EOL_DOS)
2457 {
2458 STRCAT(IObuff, _("[CR missing]"));
2459 c = TRUE;
2460 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002461 if (split)
2462 {
2463 STRCAT(IObuff, _("[long lines split]"));
2464 c = TRUE;
2465 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002466 if (notconverted)
2467 {
2468 STRCAT(IObuff, _("[NOT converted]"));
2469 c = TRUE;
2470 }
2471 else if (converted)
2472 {
2473 STRCAT(IObuff, _("[converted]"));
2474 c = TRUE;
2475 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002476#ifdef FEAT_CRYPT
2477 if (cryptkey != NULL)
2478 {
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02002479 crypt_append_msg(curbuf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002480 c = TRUE;
2481 }
2482#endif
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00002483 if (conv_error != 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002484 {
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00002485 sprintf((char *)IObuff + STRLEN(IObuff),
2486 _("[CONVERSION ERROR in line %ld]"), (long)conv_error);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002487 c = TRUE;
2488 }
2489 else if (illegal_byte > 0)
2490 {
2491 sprintf((char *)IObuff + STRLEN(IObuff),
2492 _("[ILLEGAL BYTE in line %ld]"), (long)illegal_byte);
2493 c = TRUE;
2494 }
Bram Moolenaar13505972019-01-24 15:04:48 +01002495 else if (error)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002496 {
2497 STRCAT(IObuff, _("[READ ERRORS]"));
2498 c = TRUE;
2499 }
2500 if (msg_add_fileformat(fileformat))
2501 c = TRUE;
2502#ifdef FEAT_CRYPT
2503 if (cryptkey != NULL)
Bram Moolenaar40e6a712010-05-16 22:32:54 +02002504 msg_add_lines(c, (long)linecnt, filesize
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02002505 - crypt_get_header_len(crypt_get_method_nr(curbuf)));
Bram Moolenaar071d4272004-06-13 20:20:40 +00002506 else
2507#endif
2508 msg_add_lines(c, (long)linecnt, filesize);
2509
Bram Moolenaard23a8232018-02-10 18:45:26 +01002510 VIM_CLEAR(keep_msg);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002511 msg_scrolled_ign = TRUE;
2512#ifdef ALWAYS_USE_GUI
2513 /* Don't show the message when reading stdin, it would end up in a
2514 * message box (which might be shown when exiting!) */
2515 if (read_stdin || read_buffer)
2516 p = msg_may_trunc(FALSE, IObuff);
2517 else
2518#endif
Bram Moolenaar32526b32019-01-19 17:43:09 +01002519 p = (char_u *)msg_trunc_attr((char *)IObuff, FALSE, 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002520 if (read_stdin || read_buffer || restart_edit != 0
Bram Moolenaar1c7715d2005-10-03 22:02:18 +00002521 || (msg_scrolled != 0 && !need_wait_return))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002522 /* Need to repeat the message after redrawing when:
2523 * - When reading from stdin (the screen will be cleared next).
2524 * - When restart_edit is set (otherwise there will be a delay
2525 * before redrawing).
2526 * - When the screen was scrolled but there is no wait-return
2527 * prompt. */
Bram Moolenaar8f7fd652006-02-21 22:04:51 +00002528 set_keep_msg(p, 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002529 msg_scrolled_ign = FALSE;
2530 }
2531
2532 /* with errors writing the file requires ":w!" */
2533 if (newfile && (error
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00002534 || conv_error != 0
Bram Moolenaar13505972019-01-24 15:04:48 +01002535 || (illegal_byte > 0 && bad_char_behavior != BAD_KEEP)))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002536 curbuf->b_p_ro = TRUE;
2537
2538 u_clearline(); /* cannot use "U" command after adding lines */
2539
2540 /*
2541 * In Ex mode: cursor at last new line.
2542 * Otherwise: cursor at first new line.
2543 */
2544 if (exmode_active)
2545 curwin->w_cursor.lnum = from + linecnt;
2546 else
2547 curwin->w_cursor.lnum = from + 1;
2548 check_cursor_lnum();
2549 beginline(BL_WHITE | BL_FIX); /* on first non-blank */
2550
2551 /*
2552 * Set '[ and '] marks to the newly read lines.
2553 */
2554 curbuf->b_op_start.lnum = from + 1;
2555 curbuf->b_op_start.col = 0;
2556 curbuf->b_op_end.lnum = from + linecnt;
2557 curbuf->b_op_end.col = 0;
Bram Moolenaar03f48552006-02-28 23:52:23 +00002558
Bram Moolenaar4f974752019-02-17 17:44:42 +01002559#ifdef MSWIN
Bram Moolenaar03f48552006-02-28 23:52:23 +00002560 /*
2561 * Work around a weird problem: When a file has two links (only
2562 * possible on NTFS) and we write through one link, then stat() it
Bram Moolenaarb23a7e82008-06-27 18:42:32 +00002563 * through the other link, the timestamp information may be wrong.
Bram Moolenaar03f48552006-02-28 23:52:23 +00002564 * It's correct again after reading the file, thus reset the timestamp
2565 * here.
2566 */
2567 if (newfile && !read_stdin && !read_buffer
2568 && mch_stat((char *)fname, &st) >= 0)
2569 {
2570 buf_store_time(curbuf, &st, fname);
2571 curbuf->b_mtime_read = curbuf->b_mtime;
2572 }
2573#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002574 }
2575 msg_scroll = msg_save;
2576
2577#ifdef FEAT_VIMINFO
2578 /*
2579 * Get the marks before executing autocommands, so they can be used there.
2580 */
2581 check_marks_read();
2582#endif
2583
Bram Moolenaar071d4272004-06-13 20:20:40 +00002584 /*
Bram Moolenaar34d72d42015-07-17 14:18:08 +02002585 * We remember if the last line of the read didn't have
2586 * an eol even when 'binary' is off, to support turning 'fixeol' off,
2587 * or writing the read again with 'binary' on. The latter is required
2588 * for ":autocmd FileReadPost *.gz set bin|'[,']!gunzip" to work.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002589 */
Bram Moolenaarcab35ad2011-02-15 17:39:22 +01002590 curbuf->b_no_eol_lnum = read_no_eol_lnum;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002591
Bram Moolenaarf9bb7342010-08-04 14:29:54 +02002592 /* When reloading a buffer put the cursor at the first line that is
2593 * different. */
2594 if (flags & READ_KEEP_UNDO)
2595 u_find_first_changed();
2596
Bram Moolenaar55debbe2010-05-23 23:34:36 +02002597#ifdef FEAT_PERSISTENT_UNDO
2598 /*
2599 * When opening a new file locate undo info and read it.
2600 */
2601 if (read_undo_file)
2602 {
2603 char_u hash[UNDO_HASH_SIZE];
2604
2605 sha256_finish(&sha_ctx, hash);
Bram Moolenaar6ed8ed82010-05-30 20:40:11 +02002606 u_read_undo(NULL, hash, fname);
Bram Moolenaar55debbe2010-05-23 23:34:36 +02002607 }
2608#endif
2609
Bram Moolenaarf71d7b92016-08-09 22:14:05 +02002610 if (!read_stdin && !read_fifo && (!read_buffer || sfname != NULL))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002611 {
2612 int m = msg_scroll;
2613 int n = msg_scrolled;
2614
2615 /* Save the fileformat now, otherwise the buffer will be considered
2616 * modified if the format/encoding was automatically detected. */
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002617 if (set_options)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002618 save_file_ff(curbuf);
2619
2620 /*
2621 * The output from the autocommands should not overwrite anything and
2622 * should not be overwritten: Set msg_scroll, restore its value if no
2623 * output was done.
2624 */
2625 msg_scroll = TRUE;
2626 if (filtering)
2627 apply_autocmds_exarg(EVENT_FILTERREADPOST, NULL, sfname,
2628 FALSE, curbuf, eap);
Bram Moolenaarf71d7b92016-08-09 22:14:05 +02002629 else if (newfile || (read_buffer && sfname != NULL))
Bram Moolenaarc3691332016-04-20 12:49:49 +02002630 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00002631 apply_autocmds_exarg(EVENT_BUFREADPOST, NULL, sfname,
2632 FALSE, curbuf, eap);
Bram Moolenaarc3691332016-04-20 12:49:49 +02002633 if (!au_did_filetype && *curbuf->b_p_ft != NUL)
2634 /*
2635 * EVENT_FILETYPE was not triggered but the buffer already has a
2636 * filetype. Trigger EVENT_FILETYPE using the existing filetype.
2637 */
2638 apply_autocmds(EVENT_FILETYPE, curbuf->b_p_ft, curbuf->b_fname,
2639 TRUE, curbuf);
2640 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002641 else
2642 apply_autocmds_exarg(EVENT_FILEREADPOST, sfname, sfname,
2643 FALSE, NULL, eap);
2644 if (msg_scrolled == n)
2645 msg_scroll = m;
Bram Moolenaarcab35ad2011-02-15 17:39:22 +01002646# ifdef FEAT_EVAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00002647 if (aborting()) /* autocmds may abort script processing */
2648 return FAIL;
Bram Moolenaarcab35ad2011-02-15 17:39:22 +01002649# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002650 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002651
2652 if (recoverymode && error)
2653 return FAIL;
2654 return OK;
2655}
2656
Bram Moolenaarf04507d2016-08-20 15:05:39 +02002657#if defined(OPEN_CHR_FILES) || defined(PROTO)
Bram Moolenaarfe1c56d2007-07-10 15:10:54 +00002658/*
2659 * Returns TRUE if the file name argument is of the form "/dev/fd/\d\+",
2660 * which is the name of files used for process substitution output by
2661 * some shells on some operating systems, e.g., bash on SunOS.
2662 * Do not accept "/dev/fd/[012]", opening these may hang Vim.
2663 */
Bram Moolenaarf04507d2016-08-20 15:05:39 +02002664 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002665is_dev_fd_file(char_u *fname)
Bram Moolenaarfe1c56d2007-07-10 15:10:54 +00002666{
2667 return (STRNCMP(fname, "/dev/fd/", 8) == 0
2668 && VIM_ISDIGIT(fname[8])
2669 && *skipdigits(fname + 9) == NUL
2670 && (fname[9] != NUL
2671 || (fname[8] != '0' && fname[8] != '1' && fname[8] != '2')));
2672}
2673#endif
2674
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00002675/*
2676 * From the current line count and characters read after that, estimate the
2677 * line number where we are now.
2678 * Used for error messages that include a line number.
2679 */
2680 static linenr_T
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002681readfile_linenr(
2682 linenr_T linecnt, /* line count before reading more bytes */
2683 char_u *p, /* start of more bytes read */
2684 char_u *endp) /* end of more bytes read */
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00002685{
2686 char_u *s;
2687 linenr_T lnum;
2688
2689 lnum = curbuf->b_ml.ml_line_count - linecnt + 1;
2690 for (s = p; s < endp; ++s)
2691 if (*s == '\n')
2692 ++lnum;
2693 return lnum;
2694}
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00002695
Bram Moolenaar071d4272004-06-13 20:20:40 +00002696/*
Bram Moolenaar195d6352005-12-19 22:08:24 +00002697 * Fill "*eap" to force the 'fileencoding', 'fileformat' and 'binary to be
2698 * equal to the buffer "buf". Used for calling readfile().
Bram Moolenaar071d4272004-06-13 20:20:40 +00002699 * Returns OK or FAIL.
2700 */
2701 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002702prep_exarg(exarg_T *eap, buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002703{
Bram Moolenaar13505972019-01-24 15:04:48 +01002704 eap->cmd = alloc(15 + (unsigned)STRLEN(buf->b_p_fenc));
Bram Moolenaar071d4272004-06-13 20:20:40 +00002705 if (eap->cmd == NULL)
2706 return FAIL;
2707
Bram Moolenaar333b80a2018-04-04 22:57:29 +02002708 sprintf((char *)eap->cmd, "e ++enc=%s", buf->b_p_fenc);
2709 eap->force_enc = 8;
Bram Moolenaar195d6352005-12-19 22:08:24 +00002710 eap->bad_char = buf->b_bad_char;
Bram Moolenaar333b80a2018-04-04 22:57:29 +02002711 eap->force_ff = *buf->b_p_ff;
Bram Moolenaar195d6352005-12-19 22:08:24 +00002712
2713 eap->force_bin = buf->b_p_bin ? FORCE_BIN : FORCE_NOBIN;
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002714 eap->read_edit = FALSE;
Bram Moolenaar195d6352005-12-19 22:08:24 +00002715 eap->forceit = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002716 return OK;
2717}
2718
Bram Moolenaarad875fb2013-07-24 15:02:03 +02002719/*
2720 * Set default or forced 'fileformat' and 'binary'.
2721 */
2722 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002723set_file_options(int set_options, exarg_T *eap)
Bram Moolenaarad875fb2013-07-24 15:02:03 +02002724{
2725 /* set default 'fileformat' */
2726 if (set_options)
2727 {
2728 if (eap != NULL && eap->force_ff != 0)
2729 set_fileformat(get_fileformat_force(curbuf, eap), OPT_LOCAL);
2730 else if (*p_ffs != NUL)
2731 set_fileformat(default_fileformat(), OPT_LOCAL);
2732 }
2733
2734 /* set or reset 'binary' */
2735 if (eap != NULL && eap->force_bin != 0)
2736 {
2737 int oldval = curbuf->b_p_bin;
2738
2739 curbuf->b_p_bin = (eap->force_bin == FORCE_BIN);
2740 set_options_bin(oldval, curbuf->b_p_bin, OPT_LOCAL);
2741 }
2742}
2743
Bram Moolenaarad875fb2013-07-24 15:02:03 +02002744/*
2745 * Set forced 'fileencoding'.
2746 */
2747 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002748set_forced_fenc(exarg_T *eap)
Bram Moolenaarad875fb2013-07-24 15:02:03 +02002749{
2750 if (eap->force_enc != 0)
2751 {
2752 char_u *fenc = enc_canonize(eap->cmd + eap->force_enc);
2753
2754 if (fenc != NULL)
2755 set_string_option_direct((char_u *)"fenc", -1,
2756 fenc, OPT_FREE|OPT_LOCAL, 0);
2757 vim_free(fenc);
2758 }
2759}
2760
Bram Moolenaar071d4272004-06-13 20:20:40 +00002761/*
2762 * Find next fileencoding to use from 'fileencodings'.
2763 * "pp" points to fenc_next. It's advanced to the next item.
2764 * When there are no more items, an empty string is returned and *pp is set to
2765 * NULL.
2766 * When *pp is not set to NULL, the result is in allocated memory.
2767 */
2768 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002769next_fenc(char_u **pp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002770{
2771 char_u *p;
2772 char_u *r;
2773
2774 if (**pp == NUL)
2775 {
2776 *pp = NULL;
2777 return (char_u *)"";
2778 }
2779 p = vim_strchr(*pp, ',');
2780 if (p == NULL)
2781 {
2782 r = enc_canonize(*pp);
2783 *pp += STRLEN(*pp);
2784 }
2785 else
2786 {
2787 r = vim_strnsave(*pp, (int)(p - *pp));
2788 *pp = p + 1;
2789 if (r != NULL)
2790 {
2791 p = enc_canonize(r);
2792 vim_free(r);
2793 r = p;
2794 }
2795 }
2796 if (r == NULL) /* out of memory */
2797 {
2798 r = (char_u *)"";
2799 *pp = NULL;
2800 }
2801 return r;
2802}
2803
Bram Moolenaar13505972019-01-24 15:04:48 +01002804#ifdef FEAT_EVAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00002805/*
2806 * Convert a file with the 'charconvert' expression.
2807 * This closes the file which is to be read, converts it and opens the
2808 * resulting file for reading.
2809 * Returns name of the resulting converted file (the caller should delete it
2810 * after reading it).
2811 * Returns NULL if the conversion failed ("*fdp" is not set) .
2812 */
2813 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002814readfile_charconvert(
2815 char_u *fname, /* name of input file */
2816 char_u *fenc, /* converted from */
2817 int *fdp) /* in/out: file descriptor of file */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002818{
2819 char_u *tmpname;
Bram Moolenaar32526b32019-01-19 17:43:09 +01002820 char *errmsg = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002821
Bram Moolenaare5c421c2015-03-31 13:33:08 +02002822 tmpname = vim_tempname('r', FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002823 if (tmpname == NULL)
Bram Moolenaar32526b32019-01-19 17:43:09 +01002824 errmsg = _("Can't find temp file for conversion");
Bram Moolenaar071d4272004-06-13 20:20:40 +00002825 else
2826 {
2827 close(*fdp); /* close the input file, ignore errors */
2828 *fdp = -1;
2829 if (eval_charconvert(fenc, enc_utf8 ? (char_u *)"utf-8" : p_enc,
2830 fname, tmpname) == FAIL)
Bram Moolenaar32526b32019-01-19 17:43:09 +01002831 errmsg = _("Conversion with 'charconvert' failed");
Bram Moolenaar071d4272004-06-13 20:20:40 +00002832 if (errmsg == NULL && (*fdp = mch_open((char *)tmpname,
2833 O_RDONLY | O_EXTRA, 0)) < 0)
Bram Moolenaar32526b32019-01-19 17:43:09 +01002834 errmsg = _("can't read output of 'charconvert'");
Bram Moolenaar071d4272004-06-13 20:20:40 +00002835 }
2836
2837 if (errmsg != NULL)
2838 {
2839 /* Don't use emsg(), it breaks mappings, the retry with
2840 * another type of conversion might still work. */
Bram Moolenaar32526b32019-01-19 17:43:09 +01002841 msg(errmsg);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002842 if (tmpname != NULL)
2843 {
2844 mch_remove(tmpname); /* delete converted file */
Bram Moolenaard23a8232018-02-10 18:45:26 +01002845 VIM_CLEAR(tmpname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002846 }
2847 }
2848
2849 /* If the input file is closed, open it (caller should check for error). */
2850 if (*fdp < 0)
2851 *fdp = mch_open((char *)fname, O_RDONLY | O_EXTRA, 0);
2852
2853 return tmpname;
2854}
Bram Moolenaar071d4272004-06-13 20:20:40 +00002855#endif
2856
Bram Moolenaar13505972019-01-24 15:04:48 +01002857
Bram Moolenaar071d4272004-06-13 20:20:40 +00002858#ifdef FEAT_VIMINFO
2859/*
2860 * Read marks for the current buffer from the viminfo file, when we support
2861 * buffer marks and the buffer has a name.
2862 */
2863 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002864check_marks_read(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002865{
2866 if (!curbuf->b_marks_read && get_viminfo_parameter('\'') > 0
2867 && curbuf->b_ffname != NULL)
Bram Moolenaard812df62008-11-09 12:46:09 +00002868 read_viminfo(NULL, VIF_WANT_MARKS);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002869
2870 /* Always set b_marks_read; needed when 'viminfo' is changed to include
2871 * the ' parameter after opening a buffer. */
2872 curbuf->b_marks_read = TRUE;
2873}
2874#endif
2875
Bram Moolenaara3ff49f2010-05-30 22:48:02 +02002876#if defined(FEAT_CRYPT) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002877/*
Bram Moolenaar40e6a712010-05-16 22:32:54 +02002878 * Check for magic number used for encryption. Applies to the current buffer.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002879 * If found, the magic number is removed from ptr[*sizep] and *sizep and
2880 * *filesizep are updated.
2881 * Return the (new) encryption key, NULL for no encryption.
2882 */
2883 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002884check_for_cryptkey(
2885 char_u *cryptkey, /* previous encryption key or NULL */
2886 char_u *ptr, /* pointer to read bytes */
2887 long *sizep, /* length of read bytes */
Bram Moolenaar8767f522016-07-01 17:17:39 +02002888 off_T *filesizep, /* nr of bytes used from file */
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002889 int newfile, /* editing a new buffer */
2890 char_u *fname, /* file name to display */
2891 int *did_ask) /* flag: whether already asked for key */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002892{
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02002893 int method = crypt_method_nr_from_magic((char *)ptr, *sizep);
Bram Moolenaarcf81aef2013-08-25 17:46:08 +02002894 int b_p_ro = curbuf->b_p_ro;
Bram Moolenaar40e6a712010-05-16 22:32:54 +02002895
2896 if (method >= 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002897 {
Bram Moolenaarcf81aef2013-08-25 17:46:08 +02002898 /* Mark the buffer as read-only until the decryption has taken place.
2899 * Avoids accidentally overwriting the file with garbage. */
2900 curbuf->b_p_ro = TRUE;
2901
Bram Moolenaar2be79502014-08-13 21:58:28 +02002902 /* Set the cryptmethod local to the buffer. */
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02002903 crypt_set_cm_option(curbuf, method);
Bram Moolenaarf50a2532010-05-21 15:36:08 +02002904 if (cryptkey == NULL && !*did_ask)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002905 {
2906 if (*curbuf->b_p_key)
2907 cryptkey = curbuf->b_p_key;
2908 else
2909 {
Bram Moolenaar40e6a712010-05-16 22:32:54 +02002910 /* When newfile is TRUE, store the typed key in the 'key'
2911 * option and don't free it. bf needs hash of the key saved.
Bram Moolenaarf50a2532010-05-21 15:36:08 +02002912 * Don't ask for the key again when first time Enter was hit.
2913 * Happens when retrying to detect encoding. */
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002914 smsg(_(need_key_msg), fname);
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02002915 msg_scroll = TRUE;
Bram Moolenaar3a0c9082014-11-12 15:15:42 +01002916 crypt_check_method(method);
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02002917 cryptkey = crypt_get_key(newfile, FALSE);
Bram Moolenaarf50a2532010-05-21 15:36:08 +02002918 *did_ask = TRUE;
2919
Bram Moolenaar071d4272004-06-13 20:20:40 +00002920 /* check if empty key entered */
2921 if (cryptkey != NULL && *cryptkey == NUL)
2922 {
2923 if (cryptkey != curbuf->b_p_key)
2924 vim_free(cryptkey);
2925 cryptkey = NULL;
2926 }
2927 }
2928 }
2929
2930 if (cryptkey != NULL)
2931 {
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02002932 int header_len;
Bram Moolenaar40e6a712010-05-16 22:32:54 +02002933
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02002934 curbuf->b_cryptstate = crypt_create_from_header(
2935 method, cryptkey, ptr);
2936 crypt_set_cm_option(curbuf, method);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002937
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02002938 /* Remove cryptmethod specific header from the text. */
2939 header_len = crypt_get_header_len(method);
Bram Moolenaar680e0152016-09-25 20:54:11 +02002940 if (*sizep <= header_len)
2941 /* invalid header, buffer can't be encrypted */
2942 return NULL;
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02002943 *filesizep += header_len;
2944 *sizep -= header_len;
2945 mch_memmove(ptr, ptr + header_len, (size_t)*sizep);
2946
Bram Moolenaarcf81aef2013-08-25 17:46:08 +02002947 /* Restore the read-only flag. */
2948 curbuf->b_p_ro = b_p_ro;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002949 }
2950 }
Bram Moolenaar40e6a712010-05-16 22:32:54 +02002951 /* When starting to edit a new file which does not have encryption, clear
2952 * the 'key' option, except when starting up (called with -x argument) */
Bram Moolenaarfa0ff9a2010-07-25 16:05:19 +02002953 else if (newfile && *curbuf->b_p_key != NUL && !starting)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002954 set_option_value((char_u *)"key", 0L, (char_u *)"", OPT_LOCAL);
2955
2956 return cryptkey;
2957}
Bram Moolenaar80794b12010-06-13 05:20:42 +02002958#endif /* FEAT_CRYPT */
2959
Bram Moolenaar071d4272004-06-13 20:20:40 +00002960#ifdef UNIX
2961 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002962set_file_time(
2963 char_u *fname,
2964 time_t atime, /* access time */
2965 time_t mtime) /* modification time */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002966{
2967# if defined(HAVE_UTIME) && defined(HAVE_UTIME_H)
2968 struct utimbuf buf;
2969
2970 buf.actime = atime;
2971 buf.modtime = mtime;
2972 (void)utime((char *)fname, &buf);
2973# else
2974# if defined(HAVE_UTIMES)
2975 struct timeval tvp[2];
2976
2977 tvp[0].tv_sec = atime;
2978 tvp[0].tv_usec = 0;
2979 tvp[1].tv_sec = mtime;
2980 tvp[1].tv_usec = 0;
2981# ifdef NeXT
2982 (void)utimes((char *)fname, tvp);
2983# else
2984 (void)utimes((char *)fname, (const struct timeval *)&tvp);
2985# endif
2986# endif
2987# endif
2988}
2989#endif /* UNIX */
2990
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002991#if defined(VMS) && !defined(MIN)
2992/* Older DECC compiler for VAX doesn't define MIN() */
2993# define MIN(a, b) ((a) < (b) ? (a) : (b))
2994#endif
2995
Bram Moolenaar071d4272004-06-13 20:20:40 +00002996/*
Bram Moolenaar5386a122007-06-28 20:02:32 +00002997 * Return TRUE if a file appears to be read-only from the file permissions.
2998 */
2999 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003000check_file_readonly(
3001 char_u *fname, /* full path to file */
3002 int perm) /* known permissions on file */
Bram Moolenaar5386a122007-06-28 20:02:32 +00003003{
3004#ifndef USE_MCH_ACCESS
3005 int fd = 0;
3006#endif
3007
3008 return (
3009#ifdef USE_MCH_ACCESS
3010# ifdef UNIX
3011 (perm & 0222) == 0 ||
3012# endif
3013 mch_access((char *)fname, W_OK)
3014#else
3015 (fd = mch_open((char *)fname, O_RDWR | O_EXTRA, 0)) < 0
3016 ? TRUE : (close(fd), FALSE)
3017#endif
3018 );
3019}
3020
3021
3022/*
Bram Moolenaar292ad192005-12-11 21:29:51 +00003023 * buf_write() - write to file "fname" lines "start" through "end"
Bram Moolenaar071d4272004-06-13 20:20:40 +00003024 *
3025 * We do our own buffering here because fwrite() is so slow.
3026 *
Bram Moolenaar292ad192005-12-11 21:29:51 +00003027 * If "forceit" is true, we don't care for errors when attempting backups.
3028 * In case of an error everything possible is done to restore the original
Bram Moolenaare37d50a2008-08-06 17:06:04 +00003029 * file. But when "forceit" is TRUE, we risk losing it.
Bram Moolenaar292ad192005-12-11 21:29:51 +00003030 *
3031 * When "reset_changed" is TRUE and "append" == FALSE and "start" == 1 and
3032 * "end" == curbuf->b_ml.ml_line_count, reset curbuf->b_changed.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003033 *
3034 * This function must NOT use NameBuff (because it's called by autowrite()).
3035 *
3036 * return FAIL for failure, OK otherwise
3037 */
3038 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003039buf_write(
3040 buf_T *buf,
3041 char_u *fname,
3042 char_u *sfname,
3043 linenr_T start,
3044 linenr_T end,
3045 exarg_T *eap, /* for forced 'ff' and 'fenc', can be
Bram Moolenaar071d4272004-06-13 20:20:40 +00003046 NULL! */
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003047 int append, /* append to the file */
3048 int forceit,
3049 int reset_changed,
3050 int filtering)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003051{
3052 int fd;
3053 char_u *backup = NULL;
3054 int backup_copy = FALSE; /* copy the original file? */
3055 int dobackup;
3056 char_u *ffname;
3057 char_u *wfname = NULL; /* name of file to write to */
3058 char_u *s;
3059 char_u *ptr;
3060 char_u c;
3061 int len;
3062 linenr_T lnum;
3063 long nchars;
3064 char_u *errmsg = NULL;
Bram Moolenaar32b485f2009-07-29 16:06:27 +00003065 int errmsg_allocated = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003066 char_u *errnum = NULL;
3067 char_u *buffer;
3068 char_u smallbuf[SMBUFSIZE];
3069 char_u *backup_ext;
3070 int bufsize;
3071 long perm; /* file permissions */
3072 int retval = OK;
3073 int newfile = FALSE; /* TRUE if file doesn't exist yet */
3074 int msg_save = msg_scroll;
3075 int overwriting; /* TRUE if writing over original */
3076 int no_eol = FALSE; /* no end-of-line written */
3077 int device = FALSE; /* writing to a device */
Bram Moolenaar8767f522016-07-01 17:17:39 +02003078 stat_T st_old;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003079 int prev_got_int = got_int;
Bram Moolenaare6bf6552017-06-27 22:11:51 +02003080 int checking_conversion;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003081 int file_readonly = FALSE; /* overwritten file is read-only */
3082 static char *err_readonly = "is read-only (cannot override: \"W\" in 'cpoptions')";
Bram Moolenaara06ecab2016-07-16 14:47:36 +02003083#if defined(UNIX) /*XXX fix me sometime? */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003084 int made_writable = FALSE; /* 'w' bit has been set */
3085#endif
3086 /* writing everything */
3087 int whole = (start == 1 && end == buf->b_ml.ml_line_count);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003088 linenr_T old_line_count = buf->b_ml.ml_line_count;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003089 int attr;
3090 int fileformat;
3091 int write_bin;
3092 struct bw_info write_info; /* info for buf_write_bytes() */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003093 int converted = FALSE;
3094 int notconverted = FALSE;
3095 char_u *fenc; /* effective 'fileencoding' */
3096 char_u *fenc_tofree = NULL; /* allocated "fenc" */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003097#ifdef HAS_BW_FLAGS
3098 int wb_flags = 0;
3099#endif
3100#ifdef HAVE_ACL
3101 vim_acl_T acl = NULL; /* ACL copied from original file to
3102 backup or new file */
3103#endif
Bram Moolenaar55debbe2010-05-23 23:34:36 +02003104#ifdef FEAT_PERSISTENT_UNDO
3105 int write_undo_file = FALSE;
3106 context_sha256_T sha_ctx;
3107#endif
Bram Moolenaarb8ee25a2014-09-23 15:45:08 +02003108 unsigned int bkc = get_bkc_value(buf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003109
3110 if (fname == NULL || *fname == NUL) /* safety check */
3111 return FAIL;
Bram Moolenaar70d60e92009-12-31 13:53:33 +00003112 if (buf->b_ml.ml_mfp == NULL)
3113 {
3114 /* This can happen during startup when there is a stray "w" in the
3115 * vimrc file. */
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003116 emsg(_(e_emptybuf));
Bram Moolenaar70d60e92009-12-31 13:53:33 +00003117 return FAIL;
3118 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003119
3120 /*
3121 * Disallow writing from .exrc and .vimrc in current directory for
3122 * security reasons.
3123 */
3124 if (check_secure())
3125 return FAIL;
3126
3127 /* Avoid a crash for a long name. */
3128 if (STRLEN(fname) >= MAXPATHL)
3129 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003130 emsg(_(e_longname));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003131 return FAIL;
3132 }
3133
Bram Moolenaar071d4272004-06-13 20:20:40 +00003134 /* must init bw_conv_buf and bw_iconv_fd before jumping to "fail" */
3135 write_info.bw_conv_buf = NULL;
3136 write_info.bw_conv_error = FALSE;
Bram Moolenaar32b485f2009-07-29 16:06:27 +00003137 write_info.bw_conv_error_lnum = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003138 write_info.bw_restlen = 0;
Bram Moolenaar13505972019-01-24 15:04:48 +01003139#ifdef USE_ICONV
Bram Moolenaar071d4272004-06-13 20:20:40 +00003140 write_info.bw_iconv_fd = (iconv_t)-1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003141#endif
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02003142#ifdef FEAT_CRYPT
3143 write_info.bw_buffer = buf;
3144#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003145
Bram Moolenaardf177f62005-02-22 08:39:57 +00003146 /* After writing a file changedtick changes but we don't want to display
3147 * the line. */
3148 ex_no_reprint = TRUE;
3149
Bram Moolenaar071d4272004-06-13 20:20:40 +00003150 /*
3151 * If there is no file name yet, use the one for the written file.
3152 * BF_NOTEDITED is set to reflect this (in case the write fails).
3153 * Don't do this when the write is for a filter command.
Bram Moolenaar292ad192005-12-11 21:29:51 +00003154 * Don't do this when appending.
3155 * Only do this when 'cpoptions' contains the 'F' flag.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003156 */
Bram Moolenaar2d3f4892006-01-20 23:02:51 +00003157 if (buf->b_ffname == NULL
3158 && reset_changed
Bram Moolenaar071d4272004-06-13 20:20:40 +00003159 && whole
3160 && buf == curbuf
Bram Moolenaar402d2fe2005-04-15 21:00:38 +00003161#ifdef FEAT_QUICKFIX
3162 && !bt_nofile(buf)
3163#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003164 && !filtering
Bram Moolenaar292ad192005-12-11 21:29:51 +00003165 && (!append || vim_strchr(p_cpo, CPO_FNAMEAPP) != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003166 && vim_strchr(p_cpo, CPO_FNAMEW) != NULL)
3167 {
Bram Moolenaar2d3f4892006-01-20 23:02:51 +00003168 if (set_rw_fname(fname, sfname) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003169 return FAIL;
Bram Moolenaar2d3f4892006-01-20 23:02:51 +00003170 buf = curbuf; /* just in case autocmds made "buf" invalid */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003171 }
3172
3173 if (sfname == NULL)
3174 sfname = fname;
3175 /*
3176 * For Unix: Use the short file name whenever possible.
3177 * Avoids problems with networks and when directory names are changed.
3178 * Don't do this for MS-DOS, a "cd" in a sub-shell may have moved us to
3179 * another directory, which we don't detect
3180 */
3181 ffname = fname; /* remember full fname */
3182#ifdef UNIX
3183 fname = sfname;
3184#endif
3185
3186 if (buf->b_ffname != NULL && fnamecmp(ffname, buf->b_ffname) == 0)
3187 overwriting = TRUE;
3188 else
3189 overwriting = FALSE;
3190
3191 if (exiting)
Bram Moolenaar84a05ac2013-05-06 04:24:17 +02003192 settmode(TMODE_COOK); /* when exiting allow typeahead now */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003193
3194 ++no_wait_return; /* don't wait for return yet */
3195
3196 /*
3197 * Set '[ and '] marks to the lines to be written.
3198 */
3199 buf->b_op_start.lnum = start;
3200 buf->b_op_start.col = 0;
3201 buf->b_op_end.lnum = end;
3202 buf->b_op_end.col = 0;
3203
Bram Moolenaar071d4272004-06-13 20:20:40 +00003204 {
3205 aco_save_T aco;
3206 int buf_ffname = FALSE;
3207 int buf_sfname = FALSE;
3208 int buf_fname_f = FALSE;
3209 int buf_fname_s = FALSE;
3210 int did_cmd = FALSE;
Bram Moolenaar21cf8232004-07-16 20:18:37 +00003211 int nofile_err = FALSE;
Bram Moolenaar7c626922005-02-07 22:01:03 +00003212 int empty_memline = (buf->b_ml.ml_mfp == NULL);
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02003213 bufref_T bufref;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003214
3215 /*
Bram Moolenaar84a05ac2013-05-06 04:24:17 +02003216 * Apply PRE autocommands.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003217 * Set curbuf to the buffer to be written.
3218 * Careful: The autocommands may call buf_write() recursively!
3219 */
3220 if (ffname == buf->b_ffname)
3221 buf_ffname = TRUE;
3222 if (sfname == buf->b_sfname)
3223 buf_sfname = TRUE;
3224 if (fname == buf->b_ffname)
3225 buf_fname_f = TRUE;
3226 if (fname == buf->b_sfname)
3227 buf_fname_s = TRUE;
3228
3229 /* set curwin/curbuf to buf and save a few things */
3230 aucmd_prepbuf(&aco, buf);
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02003231 set_bufref(&bufref, buf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003232
3233 if (append)
3234 {
3235 if (!(did_cmd = apply_autocmds_exarg(EVENT_FILEAPPENDCMD,
3236 sfname, sfname, FALSE, curbuf, eap)))
Bram Moolenaar21cf8232004-07-16 20:18:37 +00003237 {
Bram Moolenaarb1b715d2006-01-21 22:09:43 +00003238#ifdef FEAT_QUICKFIX
Bram Moolenaarab79bcb2004-07-18 21:34:53 +00003239 if (overwriting && bt_nofile(curbuf))
Bram Moolenaar21cf8232004-07-16 20:18:37 +00003240 nofile_err = TRUE;
3241 else
Bram Moolenaarb1b715d2006-01-21 22:09:43 +00003242#endif
Bram Moolenaar21cf8232004-07-16 20:18:37 +00003243 apply_autocmds_exarg(EVENT_FILEAPPENDPRE,
Bram Moolenaar071d4272004-06-13 20:20:40 +00003244 sfname, sfname, FALSE, curbuf, eap);
Bram Moolenaar21cf8232004-07-16 20:18:37 +00003245 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003246 }
3247 else if (filtering)
3248 {
3249 apply_autocmds_exarg(EVENT_FILTERWRITEPRE,
3250 NULL, sfname, FALSE, curbuf, eap);
3251 }
3252 else if (reset_changed && whole)
3253 {
Bram Moolenaar39fc42e2011-09-02 11:56:20 +02003254 int was_changed = curbufIsChanged();
3255
3256 did_cmd = apply_autocmds_exarg(EVENT_BUFWRITECMD,
3257 sfname, sfname, FALSE, curbuf, eap);
3258 if (did_cmd)
3259 {
3260 if (was_changed && !curbufIsChanged())
3261 {
3262 /* Written everything correctly and BufWriteCmd has reset
3263 * 'modified': Correct the undo information so that an
3264 * undo now sets 'modified'. */
3265 u_unchanged(curbuf);
3266 u_update_save_nr(curbuf);
3267 }
3268 }
3269 else
Bram Moolenaar21cf8232004-07-16 20:18:37 +00003270 {
Bram Moolenaarb1b715d2006-01-21 22:09:43 +00003271#ifdef FEAT_QUICKFIX
Bram Moolenaar19a09a12005-03-04 23:39:37 +00003272 if (overwriting && bt_nofile(curbuf))
Bram Moolenaar21cf8232004-07-16 20:18:37 +00003273 nofile_err = TRUE;
3274 else
Bram Moolenaarb1b715d2006-01-21 22:09:43 +00003275#endif
Bram Moolenaar21cf8232004-07-16 20:18:37 +00003276 apply_autocmds_exarg(EVENT_BUFWRITEPRE,
Bram Moolenaar071d4272004-06-13 20:20:40 +00003277 sfname, sfname, FALSE, curbuf, eap);
Bram Moolenaar21cf8232004-07-16 20:18:37 +00003278 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003279 }
3280 else
3281 {
3282 if (!(did_cmd = apply_autocmds_exarg(EVENT_FILEWRITECMD,
3283 sfname, sfname, FALSE, curbuf, eap)))
Bram Moolenaar21cf8232004-07-16 20:18:37 +00003284 {
Bram Moolenaarb1b715d2006-01-21 22:09:43 +00003285#ifdef FEAT_QUICKFIX
Bram Moolenaar19a09a12005-03-04 23:39:37 +00003286 if (overwriting && bt_nofile(curbuf))
Bram Moolenaar21cf8232004-07-16 20:18:37 +00003287 nofile_err = TRUE;
3288 else
Bram Moolenaarb1b715d2006-01-21 22:09:43 +00003289#endif
Bram Moolenaar21cf8232004-07-16 20:18:37 +00003290 apply_autocmds_exarg(EVENT_FILEWRITEPRE,
Bram Moolenaar071d4272004-06-13 20:20:40 +00003291 sfname, sfname, FALSE, curbuf, eap);
Bram Moolenaar21cf8232004-07-16 20:18:37 +00003292 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003293 }
3294
3295 /* restore curwin/curbuf and a few other things */
3296 aucmd_restbuf(&aco);
3297
3298 /*
3299 * In three situations we return here and don't write the file:
3300 * 1. the autocommands deleted or unloaded the buffer.
3301 * 2. The autocommands abort script processing.
3302 * 3. If one of the "Cmd" autocommands was executed.
3303 */
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02003304 if (!bufref_valid(&bufref))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003305 buf = NULL;
Bram Moolenaar7c626922005-02-07 22:01:03 +00003306 if (buf == NULL || (buf->b_ml.ml_mfp == NULL && !empty_memline)
Bram Moolenaar1e015462005-09-25 22:16:38 +00003307 || did_cmd || nofile_err
3308#ifdef FEAT_EVAL
3309 || aborting()
3310#endif
3311 )
Bram Moolenaar071d4272004-06-13 20:20:40 +00003312 {
3313 --no_wait_return;
3314 msg_scroll = msg_save;
Bram Moolenaar21cf8232004-07-16 20:18:37 +00003315 if (nofile_err)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003316 emsg(_("E676: No matching autocommands for acwrite buffer"));
Bram Moolenaar21cf8232004-07-16 20:18:37 +00003317
Bram Moolenaar1e015462005-09-25 22:16:38 +00003318 if (nofile_err
3319#ifdef FEAT_EVAL
3320 || aborting()
3321#endif
3322 )
Bram Moolenaar071d4272004-06-13 20:20:40 +00003323 /* An aborting error, interrupt or exception in the
3324 * autocommands. */
3325 return FAIL;
3326 if (did_cmd)
3327 {
3328 if (buf == NULL)
3329 /* The buffer was deleted. We assume it was written
3330 * (can't retry anyway). */
3331 return OK;
3332 if (overwriting)
3333 {
3334 /* Assume the buffer was written, update the timestamp. */
3335 ml_timestamp(buf);
Bram Moolenaar292ad192005-12-11 21:29:51 +00003336 if (append)
3337 buf->b_flags &= ~BF_NEW;
3338 else
3339 buf->b_flags &= ~BF_WRITE_MASK;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003340 }
Bram Moolenaar292ad192005-12-11 21:29:51 +00003341 if (reset_changed && buf->b_changed && !append
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003342 && (overwriting || vim_strchr(p_cpo, CPO_PLUS) != NULL))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003343 /* Buffer still changed, the autocommands didn't work
3344 * properly. */
3345 return FAIL;
3346 return OK;
3347 }
3348#ifdef FEAT_EVAL
3349 if (!aborting())
3350#endif
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003351 emsg(_("E203: Autocommands deleted or unloaded buffer to be written"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003352 return FAIL;
3353 }
3354
3355 /*
3356 * The autocommands may have changed the number of lines in the file.
3357 * When writing the whole file, adjust the end.
3358 * When writing part of the file, assume that the autocommands only
3359 * changed the number of lines that are to be written (tricky!).
3360 */
3361 if (buf->b_ml.ml_line_count != old_line_count)
3362 {
3363 if (whole) /* write all */
3364 end = buf->b_ml.ml_line_count;
3365 else if (buf->b_ml.ml_line_count > old_line_count) /* more lines */
3366 end += buf->b_ml.ml_line_count - old_line_count;
3367 else /* less lines */
3368 {
3369 end -= old_line_count - buf->b_ml.ml_line_count;
3370 if (end < start)
3371 {
3372 --no_wait_return;
3373 msg_scroll = msg_save;
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003374 emsg(_("E204: Autocommand changed number of lines in unexpected way"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003375 return FAIL;
3376 }
3377 }
3378 }
3379
3380 /*
3381 * The autocommands may have changed the name of the buffer, which may
3382 * be kept in fname, ffname and sfname.
3383 */
3384 if (buf_ffname)
3385 ffname = buf->b_ffname;
3386 if (buf_sfname)
3387 sfname = buf->b_sfname;
3388 if (buf_fname_f)
3389 fname = buf->b_ffname;
3390 if (buf_fname_s)
3391 fname = buf->b_sfname;
3392 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003393
3394#ifdef FEAT_NETBEANS_INTG
Bram Moolenaarb26e6322010-05-22 21:34:09 +02003395 if (netbeans_active() && isNetbeansBuffer(buf))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003396 {
3397 if (whole)
3398 {
3399 /*
3400 * b_changed can be 0 after an undo, but we still need to write
3401 * the buffer to NetBeans.
3402 */
3403 if (buf->b_changed || isNetbeansModified(buf))
3404 {
Bram Moolenaar009b2592004-10-24 19:18:58 +00003405 --no_wait_return; /* may wait for return now */
3406 msg_scroll = msg_save;
3407 netbeans_save_buffer(buf); /* no error checking... */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003408 return retval;
3409 }
3410 else
3411 {
3412 errnum = (char_u *)"E656: ";
Bram Moolenaared0e7452008-06-27 19:17:34 +00003413 errmsg = (char_u *)_("NetBeans disallows writes of unmodified buffers");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003414 buffer = NULL;
3415 goto fail;
3416 }
3417 }
3418 else
3419 {
3420 errnum = (char_u *)"E657: ";
3421 errmsg = (char_u *)_("Partial writes disallowed for NetBeans buffers");
3422 buffer = NULL;
3423 goto fail;
3424 }
3425 }
3426#endif
3427
3428 if (shortmess(SHM_OVER) && !exiting)
3429 msg_scroll = FALSE; /* overwrite previous file message */
3430 else
3431 msg_scroll = TRUE; /* don't overwrite previous file message */
3432 if (!filtering)
3433 filemess(buf,
3434#ifndef UNIX
3435 sfname,
3436#else
3437 fname,
3438#endif
3439 (char_u *)"", 0); /* show that we are busy */
3440 msg_scroll = FALSE; /* always overwrite the file message now */
3441
3442 buffer = alloc(BUFSIZE);
3443 if (buffer == NULL) /* can't allocate big buffer, use small
3444 * one (to be able to write when out of
3445 * memory) */
3446 {
3447 buffer = smallbuf;
3448 bufsize = SMBUFSIZE;
3449 }
3450 else
3451 bufsize = BUFSIZE;
3452
3453 /*
3454 * Get information about original file (if there is one).
3455 */
Bram Moolenaar53076832015-12-31 19:53:21 +01003456#if defined(UNIX)
Bram Moolenaar6f192452007-11-08 19:49:02 +00003457 st_old.st_dev = 0;
3458 st_old.st_ino = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003459 perm = -1;
3460 if (mch_stat((char *)fname, &st_old) < 0)
3461 newfile = TRUE;
3462 else
3463 {
3464 perm = st_old.st_mode;
3465 if (!S_ISREG(st_old.st_mode)) /* not a file */
3466 {
3467 if (S_ISDIR(st_old.st_mode))
3468 {
3469 errnum = (char_u *)"E502: ";
3470 errmsg = (char_u *)_("is a directory");
3471 goto fail;
3472 }
3473 if (mch_nodetype(fname) != NODE_WRITABLE)
3474 {
3475 errnum = (char_u *)"E503: ";
3476 errmsg = (char_u *)_("is not a file or writable device");
3477 goto fail;
3478 }
3479 /* It's a device of some kind (or a fifo) which we can write to
3480 * but for which we can't make a backup. */
3481 device = TRUE;
3482 newfile = TRUE;
3483 perm = -1;
3484 }
3485 }
3486#else /* !UNIX */
3487 /*
3488 * Check for a writable device name.
3489 */
3490 c = mch_nodetype(fname);
3491 if (c == NODE_OTHER)
3492 {
3493 errnum = (char_u *)"E503: ";
3494 errmsg = (char_u *)_("is not a file or writable device");
3495 goto fail;
3496 }
3497 if (c == NODE_WRITABLE)
3498 {
Bram Moolenaar48e330a2016-02-23 14:53:34 +01003499# if defined(MSWIN)
Bram Moolenaar043545e2006-10-10 16:44:07 +00003500 /* MS-Windows allows opening a device, but we will probably get stuck
3501 * trying to write to it. */
3502 if (!p_odev)
3503 {
3504 errnum = (char_u *)"E796: ";
3505 errmsg = (char_u *)_("writing to device disabled with 'opendevice' option");
3506 goto fail;
3507 }
3508# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003509 device = TRUE;
3510 newfile = TRUE;
3511 perm = -1;
3512 }
3513 else
3514 {
3515 perm = mch_getperm(fname);
3516 if (perm < 0)
3517 newfile = TRUE;
3518 else if (mch_isdir(fname))
3519 {
3520 errnum = (char_u *)"E502: ";
3521 errmsg = (char_u *)_("is a directory");
3522 goto fail;
3523 }
3524 if (overwriting)
3525 (void)mch_stat((char *)fname, &st_old);
3526 }
3527#endif /* !UNIX */
3528
3529 if (!device && !newfile)
3530 {
3531 /*
3532 * Check if the file is really writable (when renaming the file to
3533 * make a backup we won't discover it later).
3534 */
Bram Moolenaar5386a122007-06-28 20:02:32 +00003535 file_readonly = check_file_readonly(fname, (int)perm);
3536
Bram Moolenaar071d4272004-06-13 20:20:40 +00003537 if (!forceit && file_readonly)
3538 {
3539 if (vim_strchr(p_cpo, CPO_FWRITE) != NULL)
3540 {
3541 errnum = (char_u *)"E504: ";
3542 errmsg = (char_u *)_(err_readonly);
3543 }
3544 else
3545 {
3546 errnum = (char_u *)"E505: ";
3547 errmsg = (char_u *)_("is read-only (add ! to override)");
3548 }
3549 goto fail;
3550 }
3551
3552 /*
3553 * Check if the timestamp hasn't changed since reading the file.
3554 */
3555 if (overwriting)
3556 {
3557 retval = check_mtime(buf, &st_old);
3558 if (retval == FAIL)
3559 goto fail;
3560 }
3561 }
3562
3563#ifdef HAVE_ACL
3564 /*
3565 * For systems that support ACL: get the ACL from the original file.
3566 */
3567 if (!newfile)
3568 acl = mch_get_acl(fname);
3569#endif
3570
3571 /*
3572 * If 'backupskip' is not empty, don't make a backup for some files.
3573 */
3574 dobackup = (p_wb || p_bk || *p_pm != NUL);
3575#ifdef FEAT_WILDIGN
3576 if (dobackup && *p_bsk != NUL && match_file_list(p_bsk, sfname, ffname))
3577 dobackup = FALSE;
3578#endif
3579
3580 /*
3581 * Save the value of got_int and reset it. We don't want a previous
3582 * interruption cancel writing, only hitting CTRL-C while writing should
3583 * abort it.
3584 */
3585 prev_got_int = got_int;
3586 got_int = FALSE;
3587
3588 /* Mark the buffer as 'being saved' to prevent changed buffer warnings */
3589 buf->b_saving = TRUE;
3590
3591 /*
3592 * If we are not appending or filtering, the file exists, and the
3593 * 'writebackup', 'backup' or 'patchmode' option is set, need a backup.
3594 * When 'patchmode' is set also make a backup when appending.
3595 *
3596 * Do not make any backup, if 'writebackup' and 'backup' are both switched
3597 * off. This helps when editing large files on almost-full disks.
3598 */
3599 if (!(append && *p_pm == NUL) && !filtering && perm >= 0 && dobackup)
3600 {
Bram Moolenaar4f974752019-02-17 17:44:42 +01003601#if defined(UNIX) || defined(MSWIN)
Bram Moolenaar8767f522016-07-01 17:17:39 +02003602 stat_T st;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003603#endif
3604
Bram Moolenaarb8ee25a2014-09-23 15:45:08 +02003605 if ((bkc & BKC_YES) || append) /* "yes" */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003606 backup_copy = TRUE;
Bram Moolenaar4f974752019-02-17 17:44:42 +01003607#if defined(UNIX) || defined(MSWIN)
Bram Moolenaarb8ee25a2014-09-23 15:45:08 +02003608 else if ((bkc & BKC_AUTO)) /* "auto" */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003609 {
3610 int i;
3611
3612# ifdef UNIX
3613 /*
3614 * Don't rename the file when:
3615 * - it's a hard link
3616 * - it's a symbolic link
3617 * - we don't have write permission in the directory
3618 * - we can't set the owner/group of the new file
3619 */
3620 if (st_old.st_nlink > 1
3621 || mch_lstat((char *)fname, &st) < 0
3622 || st.st_dev != st_old.st_dev
Bram Moolenaara5792f52005-11-23 21:25:05 +00003623 || st.st_ino != st_old.st_ino
3624# ifndef HAVE_FCHOWN
3625 || st.st_uid != st_old.st_uid
3626 || st.st_gid != st_old.st_gid
3627# endif
3628 )
Bram Moolenaar071d4272004-06-13 20:20:40 +00003629 backup_copy = TRUE;
3630 else
Bram Moolenaar03f48552006-02-28 23:52:23 +00003631# else
Bram Moolenaar4f974752019-02-17 17:44:42 +01003632# ifdef MSWIN
Bram Moolenaar03f48552006-02-28 23:52:23 +00003633 /* On NTFS file systems hard links are possible. */
3634 if (mch_is_linked(fname))
3635 backup_copy = TRUE;
3636 else
3637# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003638# endif
3639 {
3640 /*
3641 * Check if we can create a file and set the owner/group to
3642 * the ones from the original file.
3643 * First find a file name that doesn't exist yet (use some
3644 * arbitrary numbers).
3645 */
3646 STRCPY(IObuff, fname);
3647 for (i = 4913; ; i += 123)
3648 {
3649 sprintf((char *)gettail(IObuff), "%d", i);
Bram Moolenaara5792f52005-11-23 21:25:05 +00003650 if (mch_lstat((char *)IObuff, &st) < 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003651 break;
3652 }
Bram Moolenaara5792f52005-11-23 21:25:05 +00003653 fd = mch_open((char *)IObuff,
3654 O_CREAT|O_WRONLY|O_EXCL|O_NOFOLLOW, perm);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003655 if (fd < 0) /* can't write in directory */
3656 backup_copy = TRUE;
3657 else
3658 {
3659# ifdef UNIX
Bram Moolenaara5792f52005-11-23 21:25:05 +00003660# ifdef HAVE_FCHOWN
Bram Moolenaar42335f52018-09-13 15:33:43 +02003661 vim_ignored = fchown(fd, st_old.st_uid, st_old.st_gid);
Bram Moolenaara5792f52005-11-23 21:25:05 +00003662# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003663 if (mch_stat((char *)IObuff, &st) < 0
3664 || st.st_uid != st_old.st_uid
3665 || st.st_gid != st_old.st_gid
Bram Moolenaar78a15312009-05-15 19:33:18 +00003666 || (long)st.st_mode != perm)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003667 backup_copy = TRUE;
3668# endif
Bram Moolenaar98358622005-11-28 22:58:23 +00003669 /* Close the file before removing it, on MS-Windows we
3670 * can't delete an open file. */
Bram Moolenaara5792f52005-11-23 21:25:05 +00003671 close(fd);
Bram Moolenaar98358622005-11-28 22:58:23 +00003672 mch_remove(IObuff);
Bram Moolenaar3479c5d2010-08-08 18:46:06 +02003673# ifdef MSWIN
3674 /* MS-Windows may trigger a virus scanner to open the
3675 * file, we can't delete it then. Keep trying for half a
3676 * second. */
3677 {
3678 int try;
3679
3680 for (try = 0; try < 10; ++try)
3681 {
3682 if (mch_lstat((char *)IObuff, &st) < 0)
3683 break;
3684 ui_delay(50L, TRUE); /* wait 50 msec */
3685 mch_remove(IObuff);
3686 }
3687 }
3688# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003689 }
3690 }
3691 }
3692
Bram Moolenaar071d4272004-06-13 20:20:40 +00003693 /*
3694 * Break symlinks and/or hardlinks if we've been asked to.
3695 */
Bram Moolenaarb8ee25a2014-09-23 15:45:08 +02003696 if ((bkc & BKC_BREAKSYMLINK) || (bkc & BKC_BREAKHARDLINK))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003697 {
Bram Moolenaar12b559e2013-06-12 22:41:37 +02003698# ifdef UNIX
Bram Moolenaar071d4272004-06-13 20:20:40 +00003699 int lstat_res;
3700
3701 lstat_res = mch_lstat((char *)fname, &st);
3702
3703 /* Symlinks. */
Bram Moolenaarb8ee25a2014-09-23 15:45:08 +02003704 if ((bkc & BKC_BREAKSYMLINK)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003705 && lstat_res == 0
3706 && st.st_ino != st_old.st_ino)
3707 backup_copy = FALSE;
3708
3709 /* Hardlinks. */
Bram Moolenaarb8ee25a2014-09-23 15:45:08 +02003710 if ((bkc & BKC_BREAKHARDLINK)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003711 && st_old.st_nlink > 1
3712 && (lstat_res != 0 || st.st_ino == st_old.st_ino))
3713 backup_copy = FALSE;
Bram Moolenaar12b559e2013-06-12 22:41:37 +02003714# else
Bram Moolenaar4f974752019-02-17 17:44:42 +01003715# if defined(MSWIN)
Bram Moolenaar12b559e2013-06-12 22:41:37 +02003716 /* Symlinks. */
Bram Moolenaarb8ee25a2014-09-23 15:45:08 +02003717 if ((bkc & BKC_BREAKSYMLINK) && mch_is_symbolic_link(fname))
Bram Moolenaar12b559e2013-06-12 22:41:37 +02003718 backup_copy = FALSE;
3719
3720 /* Hardlinks. */
Bram Moolenaarb8ee25a2014-09-23 15:45:08 +02003721 if ((bkc & BKC_BREAKHARDLINK) && mch_is_hard_link(fname))
Bram Moolenaar12b559e2013-06-12 22:41:37 +02003722 backup_copy = FALSE;
3723# endif
3724# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003725 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003726
3727#endif
3728
3729 /* make sure we have a valid backup extension to use */
3730 if (*p_bex == NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003731 backup_ext = (char_u *)".bak";
Bram Moolenaar071d4272004-06-13 20:20:40 +00003732 else
3733 backup_ext = p_bex;
3734
3735 if (backup_copy
3736 && (fd = mch_open((char *)fname, O_RDONLY | O_EXTRA, 0)) >= 0)
3737 {
3738 int bfd;
3739 char_u *copybuf, *wp;
3740 int some_error = FALSE;
Bram Moolenaar8767f522016-07-01 17:17:39 +02003741 stat_T st_new;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003742 char_u *dirp;
3743 char_u *rootname;
Bram Moolenaar4f974752019-02-17 17:44:42 +01003744#if defined(UNIX) || defined(MSWIN)
Bram Moolenaarb782ba42018-08-07 21:39:28 +02003745 char_u *p;
3746#endif
Bram Moolenaar48e330a2016-02-23 14:53:34 +01003747#if defined(UNIX)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003748 int did_set_shortname;
Bram Moolenaarcd142e32017-11-16 17:03:45 +01003749 mode_t umask_save;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003750#endif
3751
3752 copybuf = alloc(BUFSIZE + 1);
3753 if (copybuf == NULL)
3754 {
3755 some_error = TRUE; /* out of memory */
3756 goto nobackup;
3757 }
3758
3759 /*
3760 * Try to make the backup in each directory in the 'bdir' option.
3761 *
3762 * Unix semantics has it, that we may have a writable file,
3763 * that cannot be recreated with a simple open(..., O_CREAT, ) e.g:
3764 * - the directory is not writable,
3765 * - the file may be a symbolic link,
3766 * - the file may belong to another user/group, etc.
3767 *
3768 * For these reasons, the existing writable file must be truncated
3769 * and reused. Creation of a backup COPY will be attempted.
3770 */
3771 dirp = p_bdir;
3772 while (*dirp)
3773 {
3774#ifdef UNIX
3775 st_new.st_ino = 0;
3776 st_new.st_dev = 0;
3777 st_new.st_gid = 0;
3778#endif
3779
3780 /*
3781 * Isolate one directory name, using an entry in 'bdir'.
3782 */
3783 (void)copy_option_part(&dirp, copybuf, BUFSIZE, ",");
Bram Moolenaarb782ba42018-08-07 21:39:28 +02003784
Bram Moolenaar4f974752019-02-17 17:44:42 +01003785#if defined(UNIX) || defined(MSWIN)
Bram Moolenaarb782ba42018-08-07 21:39:28 +02003786 p = copybuf + STRLEN(copybuf);
3787 if (after_pathsep(copybuf, p) && p[-1] == p[-2])
3788 // Ends with '//', use full path
3789 if ((p = make_percent_swname(copybuf, fname)) != NULL)
3790 {
3791 backup = modname(p, backup_ext, FALSE);
3792 vim_free(p);
3793 }
3794#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003795 rootname = get_file_in_dir(fname, copybuf);
3796 if (rootname == NULL)
3797 {
3798 some_error = TRUE; /* out of memory */
3799 goto nobackup;
3800 }
3801
Bram Moolenaar48e330a2016-02-23 14:53:34 +01003802#if defined(UNIX)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003803 did_set_shortname = FALSE;
3804#endif
3805
3806 /*
3807 * May try twice if 'shortname' not set.
3808 */
3809 for (;;)
3810 {
3811 /*
Bram Moolenaarb782ba42018-08-07 21:39:28 +02003812 * Make the backup file name.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003813 */
Bram Moolenaarb782ba42018-08-07 21:39:28 +02003814 if (backup == NULL)
3815 backup = buf_modname((buf->b_p_sn || buf->b_shortname),
Bram Moolenaar071d4272004-06-13 20:20:40 +00003816 rootname, backup_ext, FALSE);
3817 if (backup == NULL)
3818 {
3819 vim_free(rootname);
3820 some_error = TRUE; /* out of memory */
3821 goto nobackup;
3822 }
3823
3824 /*
3825 * Check if backup file already exists.
3826 */
3827 if (mch_stat((char *)backup, &st_new) >= 0)
3828 {
3829#ifdef UNIX
3830 /*
3831 * Check if backup file is same as original file.
3832 * May happen when modname() gave the same file back.
3833 * E.g. silly link, or file name-length reached.
3834 * If we don't check here, we either ruin the file
3835 * when copying or erase it after writing. jw.
3836 */
3837 if (st_new.st_dev == st_old.st_dev
3838 && st_new.st_ino == st_old.st_ino)
3839 {
Bram Moolenaard23a8232018-02-10 18:45:26 +01003840 VIM_CLEAR(backup); /* no backup file to delete */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003841 /*
3842 * may try again with 'shortname' set
3843 */
3844 if (!(buf->b_shortname || buf->b_p_sn))
3845 {
3846 buf->b_shortname = TRUE;
3847 did_set_shortname = TRUE;
3848 continue;
3849 }
3850 /* setting shortname didn't help */
3851 if (did_set_shortname)
3852 buf->b_shortname = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003853 break;
3854 }
3855#endif
3856
3857 /*
3858 * If we are not going to keep the backup file, don't
3859 * delete an existing one, try to use another name.
3860 * Change one character, just before the extension.
3861 */
3862 if (!p_bk)
3863 {
3864 wp = backup + STRLEN(backup) - 1
3865 - STRLEN(backup_ext);
3866 if (wp < backup) /* empty file name ??? */
3867 wp = backup;
3868 *wp = 'z';
3869 while (*wp > 'a'
3870 && mch_stat((char *)backup, &st_new) >= 0)
3871 --*wp;
3872 /* They all exist??? Must be something wrong. */
3873 if (*wp == 'a')
Bram Moolenaard23a8232018-02-10 18:45:26 +01003874 VIM_CLEAR(backup);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003875 }
3876 }
3877 break;
3878 }
3879 vim_free(rootname);
3880
3881 /*
3882 * Try to create the backup file
3883 */
3884 if (backup != NULL)
3885 {
3886 /* remove old backup, if present */
3887 mch_remove(backup);
3888 /* Open with O_EXCL to avoid the file being created while
Bram Moolenaarcd142e32017-11-16 17:03:45 +01003889 * we were sleeping (symlink hacker attack?). Reset umask
3890 * if possible to avoid mch_setperm() below. */
3891#ifdef UNIX
3892 umask_save = umask(0);
3893#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003894 bfd = mch_open((char *)backup,
Bram Moolenaara5792f52005-11-23 21:25:05 +00003895 O_WRONLY|O_CREAT|O_EXTRA|O_EXCL|O_NOFOLLOW,
3896 perm & 0777);
Bram Moolenaarcd142e32017-11-16 17:03:45 +01003897#ifdef UNIX
3898 (void)umask(umask_save);
3899#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003900 if (bfd < 0)
Bram Moolenaard23a8232018-02-10 18:45:26 +01003901 VIM_CLEAR(backup);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003902 else
3903 {
Bram Moolenaarcd142e32017-11-16 17:03:45 +01003904 /* Set file protection same as original file, but
3905 * strip s-bit. Only needed if umask() wasn't used
3906 * above. */
3907#ifndef UNIX
Bram Moolenaar071d4272004-06-13 20:20:40 +00003908 (void)mch_setperm(backup, perm & 0777);
Bram Moolenaarcd142e32017-11-16 17:03:45 +01003909#else
Bram Moolenaar071d4272004-06-13 20:20:40 +00003910 /*
3911 * Try to set the group of the backup same as the
3912 * original file. If this fails, set the protection
3913 * bits for the group same as the protection bits for
3914 * others.
3915 */
Bram Moolenaara5792f52005-11-23 21:25:05 +00003916 if (st_new.st_gid != st_old.st_gid
Bram Moolenaar071d4272004-06-13 20:20:40 +00003917# ifdef HAVE_FCHOWN /* sequent-ptx lacks fchown() */
Bram Moolenaara5792f52005-11-23 21:25:05 +00003918 && fchown(bfd, (uid_t)-1, st_old.st_gid) != 0
Bram Moolenaar071d4272004-06-13 20:20:40 +00003919# endif
3920 )
3921 mch_setperm(backup,
3922 (perm & 0707) | ((perm & 07) << 3));
Bram Moolenaar5bd32f42014-04-02 14:05:38 +02003923# if defined(HAVE_SELINUX) || defined(HAVE_SMACK)
Bram Moolenaar588ebeb2008-05-07 17:09:24 +00003924 mch_copy_sec(fname, backup);
3925# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003926#endif
3927
3928 /*
3929 * copy the file.
3930 */
3931 write_info.bw_fd = bfd;
3932 write_info.bw_buf = copybuf;
3933#ifdef HAS_BW_FLAGS
3934 write_info.bw_flags = FIO_NOCONVERT;
3935#endif
Bram Moolenaar540fc6f2010-12-17 16:27:16 +01003936 while ((write_info.bw_len = read_eintr(fd, copybuf,
Bram Moolenaar071d4272004-06-13 20:20:40 +00003937 BUFSIZE)) > 0)
3938 {
3939 if (buf_write_bytes(&write_info) == FAIL)
3940 {
3941 errmsg = (char_u *)_("E506: Can't write to backup file (add ! to override)");
3942 break;
3943 }
3944 ui_breakcheck();
3945 if (got_int)
3946 {
3947 errmsg = (char_u *)_(e_interr);
3948 break;
3949 }
3950 }
3951
3952 if (close(bfd) < 0 && errmsg == NULL)
3953 errmsg = (char_u *)_("E507: Close error for backup file (add ! to override)");
3954 if (write_info.bw_len < 0)
3955 errmsg = (char_u *)_("E508: Can't read file for backup (add ! to override)");
3956#ifdef UNIX
3957 set_file_time(backup, st_old.st_atime, st_old.st_mtime);
3958#endif
3959#ifdef HAVE_ACL
3960 mch_set_acl(backup, acl);
3961#endif
Bram Moolenaar5bd32f42014-04-02 14:05:38 +02003962#if defined(HAVE_SELINUX) || defined(HAVE_SMACK)
Bram Moolenaar588ebeb2008-05-07 17:09:24 +00003963 mch_copy_sec(fname, backup);
3964#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003965 break;
3966 }
3967 }
3968 }
3969 nobackup:
3970 close(fd); /* ignore errors for closing read file */
3971 vim_free(copybuf);
3972
3973 if (backup == NULL && errmsg == NULL)
3974 errmsg = (char_u *)_("E509: Cannot create backup file (add ! to override)");
3975 /* ignore errors when forceit is TRUE */
3976 if ((some_error || errmsg != NULL) && !forceit)
3977 {
3978 retval = FAIL;
3979 goto fail;
3980 }
3981 errmsg = NULL;
3982 }
3983 else
3984 {
3985 char_u *dirp;
3986 char_u *p;
3987 char_u *rootname;
3988
3989 /*
3990 * Make a backup by renaming the original file.
3991 */
3992 /*
3993 * If 'cpoptions' includes the "W" flag, we don't want to
3994 * overwrite a read-only file. But rename may be possible
3995 * anyway, thus we need an extra check here.
3996 */
3997 if (file_readonly && vim_strchr(p_cpo, CPO_FWRITE) != NULL)
3998 {
3999 errnum = (char_u *)"E504: ";
4000 errmsg = (char_u *)_(err_readonly);
4001 goto fail;
4002 }
4003
4004 /*
4005 *
4006 * Form the backup file name - change path/fo.o.h to
4007 * path/fo.o.h.bak Try all directories in 'backupdir', first one
4008 * that works is used.
4009 */
4010 dirp = p_bdir;
4011 while (*dirp)
4012 {
4013 /*
4014 * Isolate one directory name and make the backup file name.
4015 */
4016 (void)copy_option_part(&dirp, IObuff, IOSIZE, ",");
Bram Moolenaarb782ba42018-08-07 21:39:28 +02004017
Bram Moolenaar4f974752019-02-17 17:44:42 +01004018#if defined(UNIX) || defined(MSWIN)
Bram Moolenaarb782ba42018-08-07 21:39:28 +02004019 p = IObuff + STRLEN(IObuff);
4020 if (after_pathsep(IObuff, p) && p[-1] == p[-2])
4021 // path ends with '//', use full path
4022 if ((p = make_percent_swname(IObuff, fname)) != NULL)
4023 {
4024 backup = modname(p, backup_ext, FALSE);
4025 vim_free(p);
4026 }
4027#endif
4028 if (backup == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004029 {
Bram Moolenaarb782ba42018-08-07 21:39:28 +02004030 rootname = get_file_in_dir(fname, IObuff);
4031 if (rootname == NULL)
4032 backup = NULL;
4033 else
4034 {
4035 backup = buf_modname(
4036 (buf->b_p_sn || buf->b_shortname),
4037 rootname, backup_ext, FALSE);
4038 vim_free(rootname);
4039 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004040 }
4041
4042 if (backup != NULL)
4043 {
4044 /*
4045 * If we are not going to keep the backup file, don't
4046 * delete an existing one, try to use another name.
4047 * Change one character, just before the extension.
4048 */
4049 if (!p_bk && mch_getperm(backup) >= 0)
4050 {
4051 p = backup + STRLEN(backup) - 1 - STRLEN(backup_ext);
4052 if (p < backup) /* empty file name ??? */
4053 p = backup;
4054 *p = 'z';
4055 while (*p > 'a' && mch_getperm(backup) >= 0)
4056 --*p;
4057 /* They all exist??? Must be something wrong! */
4058 if (*p == 'a')
Bram Moolenaard23a8232018-02-10 18:45:26 +01004059 VIM_CLEAR(backup);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004060 }
4061 }
4062 if (backup != NULL)
4063 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00004064 /*
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +00004065 * Delete any existing backup and move the current version
4066 * to the backup. For safety, we don't remove the backup
4067 * until the write has finished successfully. And if the
4068 * 'backup' option is set, leave it around.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004069 */
4070 /*
4071 * If the renaming of the original file to the backup file
4072 * works, quit here.
4073 */
4074 if (vim_rename(fname, backup) == 0)
4075 break;
4076
Bram Moolenaard23a8232018-02-10 18:45:26 +01004077 VIM_CLEAR(backup); /* don't do the rename below */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004078 }
4079 }
4080 if (backup == NULL && !forceit)
4081 {
4082 errmsg = (char_u *)_("E510: Can't make backup file (add ! to override)");
4083 goto fail;
4084 }
4085 }
4086 }
4087
Bram Moolenaar53076832015-12-31 19:53:21 +01004088#if defined(UNIX)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004089 /* When using ":w!" and the file was read-only: make it writable */
4090 if (forceit && perm >= 0 && !(perm & 0200) && st_old.st_uid == getuid()
4091 && vim_strchr(p_cpo, CPO_FWRITE) == NULL)
4092 {
4093 perm |= 0200;
4094 (void)mch_setperm(fname, perm);
4095 made_writable = TRUE;
4096 }
4097#endif
4098
Bram Moolenaar910f66f2006-04-05 20:41:53 +00004099 /* When using ":w!" and writing to the current file, 'readonly' makes no
Bram Moolenaar4399ef42005-02-12 14:29:27 +00004100 * sense, reset it, unless 'Z' appears in 'cpoptions'. */
4101 if (forceit && overwriting && vim_strchr(p_cpo, CPO_KEEPRO) == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004102 {
4103 buf->b_p_ro = FALSE;
4104#ifdef FEAT_TITLE
4105 need_maketitle = TRUE; /* set window title later */
4106#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004107 status_redraw_all(); /* redraw status lines later */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004108 }
4109
4110 if (end > buf->b_ml.ml_line_count)
4111 end = buf->b_ml.ml_line_count;
4112 if (buf->b_ml.ml_flags & ML_EMPTY)
4113 start = end + 1;
4114
4115 /*
4116 * If the original file is being overwritten, there is a small chance that
4117 * we crash in the middle of writing. Therefore the file is preserved now.
4118 * This makes all block numbers positive so that recovery does not need
4119 * the original file.
4120 * Don't do this if there is a backup file and we are exiting.
4121 */
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00004122 if (reset_changed && !newfile && overwriting
Bram Moolenaar071d4272004-06-13 20:20:40 +00004123 && !(exiting && backup != NULL))
4124 {
4125 ml_preserve(buf, FALSE);
4126 if (got_int)
4127 {
4128 errmsg = (char_u *)_(e_interr);
4129 goto restore_backup;
4130 }
4131 }
4132
Bram Moolenaar071d4272004-06-13 20:20:40 +00004133#ifdef VMS
4134 vms_remove_version(fname); /* remove version */
4135#endif
Bram Moolenaarb23a7e82008-06-27 18:42:32 +00004136 /* Default: write the file directly. May write to a temp file for
Bram Moolenaar071d4272004-06-13 20:20:40 +00004137 * multi-byte conversion. */
4138 wfname = fname;
4139
Bram Moolenaar071d4272004-06-13 20:20:40 +00004140 /* Check for forced 'fileencoding' from "++opt=val" argument. */
4141 if (eap != NULL && eap->force_enc != 0)
4142 {
4143 fenc = eap->cmd + eap->force_enc;
4144 fenc = enc_canonize(fenc);
4145 fenc_tofree = fenc;
4146 }
4147 else
4148 fenc = buf->b_p_fenc;
4149
4150 /*
Bram Moolenaarb5cdf2e2009-07-29 16:25:31 +00004151 * Check if the file needs to be converted.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004152 */
Bram Moolenaarb5cdf2e2009-07-29 16:25:31 +00004153 converted = need_conversion(fenc);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004154
4155 /*
4156 * Check if UTF-8 to UCS-2/4 or Latin1 conversion needs to be done. Or
4157 * Latin1 to Unicode conversion. This is handled in buf_write_bytes().
4158 * Prepare the flags for it and allocate bw_conv_buf when needed.
4159 */
4160 if (converted && (enc_utf8 || STRCMP(p_enc, "latin1") == 0))
4161 {
4162 wb_flags = get_fio_flags(fenc);
4163 if (wb_flags & (FIO_UCS2 | FIO_UCS4 | FIO_UTF16 | FIO_UTF8))
4164 {
4165 /* Need to allocate a buffer to translate into. */
4166 if (wb_flags & (FIO_UCS2 | FIO_UTF16 | FIO_UTF8))
4167 write_info.bw_conv_buflen = bufsize * 2;
4168 else /* FIO_UCS4 */
4169 write_info.bw_conv_buflen = bufsize * 4;
4170 write_info.bw_conv_buf
4171 = lalloc((long_u)write_info.bw_conv_buflen, TRUE);
4172 if (write_info.bw_conv_buf == NULL)
4173 end = 0;
4174 }
4175 }
4176
Bram Moolenaar4f974752019-02-17 17:44:42 +01004177#ifdef MSWIN
Bram Moolenaar071d4272004-06-13 20:20:40 +00004178 if (converted && wb_flags == 0 && (wb_flags = get_win_fio_flags(fenc)) != 0)
4179 {
4180 /* Convert UTF-8 -> UCS-2 and UCS-2 -> DBCS. Worst-case * 4: */
4181 write_info.bw_conv_buflen = bufsize * 4;
4182 write_info.bw_conv_buf
4183 = lalloc((long_u)write_info.bw_conv_buflen, TRUE);
4184 if (write_info.bw_conv_buf == NULL)
4185 end = 0;
4186 }
Bram Moolenaar13505972019-01-24 15:04:48 +01004187#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004188
Bram Moolenaar13505972019-01-24 15:04:48 +01004189#ifdef MACOS_CONVERT
Bram Moolenaar071d4272004-06-13 20:20:40 +00004190 if (converted && wb_flags == 0 && (wb_flags = get_mac_fio_flags(fenc)) != 0)
4191 {
4192 write_info.bw_conv_buflen = bufsize * 3;
4193 write_info.bw_conv_buf
4194 = lalloc((long_u)write_info.bw_conv_buflen, TRUE);
4195 if (write_info.bw_conv_buf == NULL)
4196 end = 0;
4197 }
Bram Moolenaar13505972019-01-24 15:04:48 +01004198#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004199
Bram Moolenaar13505972019-01-24 15:04:48 +01004200#if defined(FEAT_EVAL) || defined(USE_ICONV)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004201 if (converted && wb_flags == 0)
4202 {
Bram Moolenaar13505972019-01-24 15:04:48 +01004203# ifdef USE_ICONV
Bram Moolenaar071d4272004-06-13 20:20:40 +00004204 /*
4205 * Use iconv() conversion when conversion is needed and it's not done
4206 * internally.
4207 */
4208 write_info.bw_iconv_fd = (iconv_t)my_iconv_open(fenc,
4209 enc_utf8 ? (char_u *)"utf-8" : p_enc);
4210 if (write_info.bw_iconv_fd != (iconv_t)-1)
4211 {
4212 /* We're going to use iconv(), allocate a buffer to convert in. */
4213 write_info.bw_conv_buflen = bufsize * ICONV_MULT;
4214 write_info.bw_conv_buf
4215 = lalloc((long_u)write_info.bw_conv_buflen, TRUE);
4216 if (write_info.bw_conv_buf == NULL)
4217 end = 0;
4218 write_info.bw_first = TRUE;
4219 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004220# ifdef FEAT_EVAL
Bram Moolenaar13505972019-01-24 15:04:48 +01004221 else
4222# endif
4223# endif
4224
4225# ifdef FEAT_EVAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00004226 /*
4227 * When the file needs to be converted with 'charconvert' after
4228 * writing, write to a temp file instead and let the conversion
4229 * overwrite the original file.
4230 */
4231 if (*p_ccv != NUL)
4232 {
Bram Moolenaare5c421c2015-03-31 13:33:08 +02004233 wfname = vim_tempname('w', FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004234 if (wfname == NULL) /* Can't write without a tempfile! */
4235 {
4236 errmsg = (char_u *)_("E214: Can't find temp file for writing");
4237 goto restore_backup;
4238 }
4239 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004240# endif
Bram Moolenaar13505972019-01-24 15:04:48 +01004241 }
4242#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004243 if (converted && wb_flags == 0
Bram Moolenaar13505972019-01-24 15:04:48 +01004244#ifdef USE_ICONV
Bram Moolenaar071d4272004-06-13 20:20:40 +00004245 && write_info.bw_iconv_fd == (iconv_t)-1
Bram Moolenaar13505972019-01-24 15:04:48 +01004246# endif
4247# ifdef FEAT_EVAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00004248 && wfname == fname
Bram Moolenaar13505972019-01-24 15:04:48 +01004249# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004250 )
4251 {
4252 if (!forceit)
4253 {
4254 errmsg = (char_u *)_("E213: Cannot convert (add ! to write without conversion)");
4255 goto restore_backup;
4256 }
4257 notconverted = TRUE;
4258 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004259
4260 /*
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004261 * If conversion is taking place, we may first pretend to write and check
4262 * for conversion errors. Then loop again to write for real.
4263 * When not doing conversion this writes for real right away.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004264 */
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004265 for (checking_conversion = TRUE; ; checking_conversion = FALSE)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004266 {
4267 /*
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004268 * There is no need to check conversion when:
4269 * - there is no conversion
4270 * - we make a backup file, that can be restored in case of conversion
4271 * failure.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004272 */
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004273 if (!converted || dobackup)
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004274 checking_conversion = FALSE;
4275
4276 if (checking_conversion)
4277 {
4278 /* Make sure we don't write anything. */
4279 fd = -1;
4280 write_info.bw_fd = fd;
4281 }
4282 else
4283 {
Bram Moolenaarcd142e32017-11-16 17:03:45 +01004284#ifdef HAVE_FTRUNCATE
4285# define TRUNC_ON_OPEN 0
4286#else
4287# define TRUNC_ON_OPEN O_TRUNC
4288#endif
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004289 /*
4290 * Open the file "wfname" for writing.
4291 * We may try to open the file twice: If we can't write to the file
4292 * and forceit is TRUE we delete the existing file and try to
4293 * create a new one. If this still fails we may have lost the
4294 * original file! (this may happen when the user reached his
4295 * quotum for number of files).
4296 * Appending will fail if the file does not exist and forceit is
4297 * FALSE.
4298 */
4299 while ((fd = mch_open((char *)wfname, O_WRONLY | O_EXTRA | (append
4300 ? (forceit ? (O_APPEND | O_CREAT) : O_APPEND)
Bram Moolenaarcd142e32017-11-16 17:03:45 +01004301 : (O_CREAT | TRUNC_ON_OPEN))
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004302 , perm < 0 ? 0666 : (perm & 0777))) < 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004303 {
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004304 /*
4305 * A forced write will try to create a new file if the old one
4306 * is still readonly. This may also happen when the directory
4307 * is read-only. In that case the mch_remove() will fail.
4308 */
4309 if (errmsg == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004310 {
4311#ifdef UNIX
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004312 stat_T st;
4313
4314 /* Don't delete the file when it's a hard or symbolic link.
4315 */
4316 if ((!newfile && st_old.st_nlink > 1)
4317 || (mch_lstat((char *)fname, &st) == 0
4318 && (st.st_dev != st_old.st_dev
4319 || st.st_ino != st_old.st_ino)))
4320 errmsg = (char_u *)_("E166: Can't open linked file for writing");
4321 else
Bram Moolenaar071d4272004-06-13 20:20:40 +00004322#endif
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004323 {
4324 errmsg = (char_u *)_("E212: Can't open file for writing");
4325 if (forceit && vim_strchr(p_cpo, CPO_FWRITE) == NULL
4326 && perm >= 0)
4327 {
4328#ifdef UNIX
4329 /* we write to the file, thus it should be marked
4330 writable after all */
4331 if (!(perm & 0200))
4332 made_writable = TRUE;
4333 perm |= 0200;
4334 if (st_old.st_uid != getuid()
4335 || st_old.st_gid != getgid())
4336 perm &= 0777;
4337#endif
4338 if (!append) /* don't remove when appending */
4339 mch_remove(wfname);
4340 continue;
4341 }
4342 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004343 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004344
4345restore_backup:
Bram Moolenaar071d4272004-06-13 20:20:40 +00004346 {
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004347 stat_T st;
4348
Bram Moolenaar071d4272004-06-13 20:20:40 +00004349 /*
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004350 * If we failed to open the file, we don't need a backup.
4351 * Throw it away. If we moved or removed the original file
4352 * try to put the backup in its place.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004353 */
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004354 if (backup != NULL && wfname == fname)
4355 {
4356 if (backup_copy)
4357 {
4358 /*
4359 * There is a small chance that we removed the
4360 * original, try to move the copy in its place.
4361 * This may not work if the vim_rename() fails.
4362 * In that case we leave the copy around.
4363 */
4364 /* If file does not exist, put the copy in its
4365 * place */
4366 if (mch_stat((char *)fname, &st) < 0)
4367 vim_rename(backup, fname);
4368 /* if original file does exist throw away the copy
4369 */
4370 if (mch_stat((char *)fname, &st) >= 0)
4371 mch_remove(backup);
4372 }
4373 else
4374 {
4375 /* try to put the original file back */
4376 vim_rename(backup, fname);
4377 }
4378 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004379
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004380 /* if original file no longer exists give an extra warning
4381 */
4382 if (!newfile && mch_stat((char *)fname, &st) < 0)
4383 end = 0;
4384 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004385
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004386 if (wfname != fname)
4387 vim_free(wfname);
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004388 goto fail;
4389 }
4390 write_info.bw_fd = fd;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004391
Bram Moolenaarcd142e32017-11-16 17:03:45 +01004392#if defined(UNIX)
4393 {
4394 stat_T st;
4395
4396 /* Double check we are writing the intended file before making
4397 * any changes. */
4398 if (overwriting
4399 && (!dobackup || backup_copy)
4400 && fname == wfname
4401 && perm >= 0
4402 && mch_fstat(fd, &st) == 0
4403 && st.st_ino != st_old.st_ino)
4404 {
4405 close(fd);
4406 errmsg = (char_u *)_("E949: File changed while writing");
4407 goto fail;
4408 }
4409 }
4410#endif
4411#ifdef HAVE_FTRUNCATE
4412 if (!append)
Bram Moolenaar42335f52018-09-13 15:33:43 +02004413 vim_ignored = ftruncate(fd, (off_t)0);
Bram Moolenaarcd142e32017-11-16 17:03:45 +01004414#endif
4415
Bram Moolenaar4f974752019-02-17 17:44:42 +01004416#if defined(MSWIN)
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004417 if (backup != NULL && overwriting && !append)
4418 {
4419 if (backup_copy)
4420 (void)mch_copy_file_attribute(wfname, backup);
4421 else
4422 (void)mch_copy_file_attribute(backup, wfname);
4423 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004424
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004425 if (!overwriting && !append)
4426 {
4427 if (buf->b_ffname != NULL)
4428 (void)mch_copy_file_attribute(buf->b_ffname, wfname);
4429 /* Should copy resource fork */
4430 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004431#endif
4432
Bram Moolenaar071d4272004-06-13 20:20:40 +00004433#ifdef FEAT_CRYPT
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004434 if (*buf->b_p_key != NUL && !filtering)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004435 {
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004436 char_u *header;
4437 int header_len;
4438
4439 buf->b_cryptstate = crypt_create_for_writing(
4440 crypt_get_method_nr(buf),
4441 buf->b_p_key, &header, &header_len);
4442 if (buf->b_cryptstate == NULL || header == NULL)
4443 end = 0;
4444 else
Bram Moolenaar071d4272004-06-13 20:20:40 +00004445 {
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004446 /* Write magic number, so that Vim knows how this file is
4447 * encrypted when reading it back. */
4448 write_info.bw_buf = header;
4449 write_info.bw_len = header_len;
4450 write_info.bw_flags = FIO_NOCONVERT;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004451 if (buf_write_bytes(&write_info) == FAIL)
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004452 end = 0;
4453 wb_flags |= FIO_ENCRYPTED;
4454 vim_free(header);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004455 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004456 }
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004457#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004458 }
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004459 errmsg = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004460
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004461 write_info.bw_buf = buffer;
4462 nchars = 0;
4463
4464 /* use "++bin", "++nobin" or 'binary' */
4465 if (eap != NULL && eap->force_bin != 0)
4466 write_bin = (eap->force_bin == FORCE_BIN);
4467 else
4468 write_bin = buf->b_p_bin;
4469
Bram Moolenaar071d4272004-06-13 20:20:40 +00004470 /*
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004471 * The BOM is written just after the encryption magic number.
4472 * Skip it when appending and the file already existed, the BOM only
4473 * makes sense at the start of the file.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004474 */
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004475 if (buf->b_p_bomb && !write_bin && (!append || perm < 0))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004476 {
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004477 write_info.bw_len = make_bom(buffer, fenc);
4478 if (write_info.bw_len > 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004479 {
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004480 /* don't convert, do encryption */
4481 write_info.bw_flags = FIO_NOCONVERT | wb_flags;
4482 if (buf_write_bytes(&write_info) == FAIL)
4483 end = 0;
4484 else
4485 nchars += write_info.bw_len;
4486 }
4487 }
4488 write_info.bw_start_lnum = start;
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004489
4490#ifdef FEAT_PERSISTENT_UNDO
4491 write_undo_file = (buf->b_p_udf
4492 && overwriting
4493 && !append
4494 && !filtering
4495 && reset_changed
4496 && !checking_conversion);
4497 if (write_undo_file)
4498 /* Prepare for computing the hash value of the text. */
4499 sha256_start(&sha_ctx);
4500#endif
4501
4502 write_info.bw_len = bufsize;
4503#ifdef HAS_BW_FLAGS
4504 write_info.bw_flags = wb_flags;
4505#endif
4506 fileformat = get_fileformat_force(buf, eap);
4507 s = buffer;
4508 len = 0;
4509 for (lnum = start; lnum <= end; ++lnum)
4510 {
4511 /*
4512 * The next while loop is done once for each character written.
4513 * Keep it fast!
4514 */
4515 ptr = ml_get_buf(buf, lnum, FALSE) - 1;
4516#ifdef FEAT_PERSISTENT_UNDO
4517 if (write_undo_file)
4518 sha256_update(&sha_ctx, ptr + 1,
4519 (UINT32_T)(STRLEN(ptr + 1) + 1));
4520#endif
4521 while ((c = *++ptr) != NUL)
4522 {
4523 if (c == NL)
4524 *s = NUL; /* replace newlines with NULs */
4525 else if (c == CAR && fileformat == EOL_MAC)
4526 *s = NL; /* Mac: replace CRs with NLs */
4527 else
4528 *s = c;
4529 ++s;
4530 if (++len != bufsize)
4531 continue;
Bram Moolenaard4755bb2004-09-02 19:12:26 +00004532 if (buf_write_bytes(&write_info) == FAIL)
4533 {
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004534 end = 0; /* write error: break loop */
Bram Moolenaard4755bb2004-09-02 19:12:26 +00004535 break;
4536 }
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004537 nchars += bufsize;
4538 s = buffer;
4539 len = 0;
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004540 write_info.bw_start_lnum = lnum;
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004541 }
4542 /* write failed or last line has no EOL: stop here */
4543 if (end == 0
4544 || (lnum == end
4545 && (write_bin || !buf->b_p_fixeol)
4546 && (lnum == buf->b_no_eol_lnum
4547 || (lnum == buf->b_ml.ml_line_count
4548 && !buf->b_p_eol))))
4549 {
4550 ++lnum; /* written the line, count it */
4551 no_eol = TRUE;
4552 break;
4553 }
4554 if (fileformat == EOL_UNIX)
4555 *s++ = NL;
4556 else
4557 {
4558 *s++ = CAR; /* EOL_MAC or EOL_DOS: write CR */
4559 if (fileformat == EOL_DOS) /* write CR-NL */
4560 {
4561 if (++len == bufsize)
4562 {
4563 if (buf_write_bytes(&write_info) == FAIL)
4564 {
4565 end = 0; /* write error: break loop */
4566 break;
4567 }
4568 nchars += bufsize;
4569 s = buffer;
4570 len = 0;
4571 }
4572 *s++ = NL;
4573 }
4574 }
4575 if (++len == bufsize && end)
4576 {
4577 if (buf_write_bytes(&write_info) == FAIL)
4578 {
4579 end = 0; /* write error: break loop */
4580 break;
4581 }
4582 nchars += bufsize;
4583 s = buffer;
4584 len = 0;
4585
4586 ui_breakcheck();
4587 if (got_int)
4588 {
4589 end = 0; /* Interrupted, break loop */
4590 break;
4591 }
4592 }
4593#ifdef VMS
4594 /*
4595 * On VMS there is a problem: newlines get added when writing
4596 * blocks at a time. Fix it by writing a line at a time.
4597 * This is much slower!
4598 * Explanation: VAX/DECC RTL insists that records in some RMS
4599 * structures end with a newline (carriage return) character, and
4600 * if they don't it adds one.
4601 * With other RMS structures it works perfect without this fix.
4602 */
4603 if (buf->b_fab_rfm == FAB$C_VFC
4604 || ((buf->b_fab_rat & (FAB$M_FTN | FAB$M_CR)) != 0))
4605 {
4606 int b2write;
4607
4608 buf->b_fab_mrs = (buf->b_fab_mrs == 0
4609 ? MIN(4096, bufsize)
4610 : MIN(buf->b_fab_mrs, bufsize));
4611
4612 b2write = len;
4613 while (b2write > 0)
4614 {
4615 write_info.bw_len = MIN(b2write, buf->b_fab_mrs);
4616 if (buf_write_bytes(&write_info) == FAIL)
4617 {
4618 end = 0;
4619 break;
4620 }
4621 b2write -= MIN(b2write, buf->b_fab_mrs);
4622 }
4623 write_info.bw_len = bufsize;
4624 nchars += len;
4625 s = buffer;
4626 len = 0;
4627 }
4628#endif
4629 }
4630 if (len > 0 && end > 0)
4631 {
4632 write_info.bw_len = len;
4633 if (buf_write_bytes(&write_info) == FAIL)
4634 end = 0; /* write error */
4635 nchars += len;
4636 }
4637
4638 /* Stop when writing done or an error was encountered. */
4639 if (!checking_conversion || end == 0)
4640 break;
4641
4642 /* If no error happened until now, writing should be ok, so loop to
4643 * really write the buffer. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004644 }
4645
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004646 /* If we started writing, finish writing. Also when an error was
4647 * encountered. */
4648 if (!checking_conversion)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004649 {
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004650#if defined(UNIX) && defined(HAVE_FSYNC)
4651 /*
4652 * On many journalling file systems there is a bug that causes both the
4653 * original and the backup file to be lost when halting the system
4654 * right after writing the file. That's because only the meta-data is
4655 * journalled. Syncing the file slows down the system, but assures it
4656 * has been written to disk and we don't lose it.
4657 * For a device do try the fsync() but don't complain if it does not
4658 * work (could be a pipe).
4659 * If the 'fsync' option is FALSE, don't fsync(). Useful for laptops.
4660 */
Bram Moolenaara7870192019-02-14 12:56:36 +01004661 if (p_fs && vim_fsync(fd) != 0 && !device)
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004662 {
Bram Moolenaar7567d0b2017-11-16 23:04:15 +01004663 errmsg = (char_u *)_(e_fsync);
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004664 end = 0;
4665 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004666#endif
4667
Bram Moolenaar5bd32f42014-04-02 14:05:38 +02004668#if defined(HAVE_SELINUX) || defined(HAVE_SMACK)
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004669 /* Probably need to set the security context. */
4670 if (!backup_copy)
4671 mch_copy_sec(backup, wfname);
Bram Moolenaar588ebeb2008-05-07 17:09:24 +00004672#endif
4673
Bram Moolenaara5792f52005-11-23 21:25:05 +00004674#ifdef UNIX
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004675 /* When creating a new file, set its owner/group to that of the
4676 * original file. Get the new device and inode number. */
4677 if (backup != NULL && !backup_copy)
Bram Moolenaara5792f52005-11-23 21:25:05 +00004678 {
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004679# ifdef HAVE_FCHOWN
4680 stat_T st;
4681
Bram Moolenaarcd142e32017-11-16 17:03:45 +01004682 /* Don't change the owner when it's already OK, some systems remove
4683 * permission or ACL stuff. */
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004684 if (mch_stat((char *)wfname, &st) < 0
4685 || st.st_uid != st_old.st_uid
4686 || st.st_gid != st_old.st_gid)
4687 {
Bram Moolenaarcd142e32017-11-16 17:03:45 +01004688 /* changing owner might not be possible */
Bram Moolenaar42335f52018-09-13 15:33:43 +02004689 vim_ignored = fchown(fd, st_old.st_uid, -1);
Bram Moolenaarcd142e32017-11-16 17:03:45 +01004690 /* if changing group fails clear the group permissions */
4691 if (fchown(fd, -1, st_old.st_gid) == -1 && perm > 0)
4692 perm &= ~070;
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004693 }
Bram Moolenaara5792f52005-11-23 21:25:05 +00004694# endif
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004695 buf_setino(buf);
4696 }
4697 else if (!buf->b_dev_valid)
4698 /* Set the inode when creating a new file. */
4699 buf_setino(buf);
Bram Moolenaara5792f52005-11-23 21:25:05 +00004700#endif
4701
Bram Moolenaarcd142e32017-11-16 17:03:45 +01004702#ifdef UNIX
4703 if (made_writable)
4704 perm &= ~0200; /* reset 'w' bit for security reasons */
4705#endif
4706#ifdef HAVE_FCHMOD
4707 /* set permission of new file same as old file */
4708 if (perm >= 0)
4709 (void)mch_fsetperm(fd, perm);
4710#endif
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004711 if (close(fd) != 0)
4712 {
4713 errmsg = (char_u *)_("E512: Close failed");
4714 end = 0;
4715 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004716
Bram Moolenaarcd142e32017-11-16 17:03:45 +01004717#ifndef HAVE_FCHMOD
4718 /* set permission of new file same as old file */
4719 if (perm >= 0)
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004720 (void)mch_setperm(wfname, perm);
Bram Moolenaarcd142e32017-11-16 17:03:45 +01004721#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004722#ifdef HAVE_ACL
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004723 /*
4724 * Probably need to set the ACL before changing the user (can't set the
4725 * ACL on a file the user doesn't own).
4726 * On Solaris, with ZFS and the aclmode property set to "discard" (the
4727 * default), chmod() discards all part of a file's ACL that don't
4728 * represent the mode of the file. It's non-trivial for us to discover
4729 * whether we're in that situation, so we simply always re-set the ACL.
4730 */
Bram Moolenaarda412772016-07-14 20:37:07 +02004731# ifndef HAVE_SOLARIS_ZFS_ACL
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004732 if (!backup_copy)
Bram Moolenaarda412772016-07-14 20:37:07 +02004733# endif
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004734 mch_set_acl(wfname, acl);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004735#endif
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02004736#ifdef FEAT_CRYPT
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004737 if (buf->b_cryptstate != NULL)
4738 {
4739 crypt_free_state(buf->b_cryptstate);
4740 buf->b_cryptstate = NULL;
4741 }
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02004742#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004743
Bram Moolenaar13505972019-01-24 15:04:48 +01004744#if defined(FEAT_EVAL)
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004745 if (wfname != fname)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004746 {
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004747 /*
4748 * The file was written to a temp file, now it needs to be
4749 * converted with 'charconvert' to (overwrite) the output file.
4750 */
4751 if (end != 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004752 {
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004753 if (eval_charconvert(enc_utf8 ? (char_u *)"utf-8" : p_enc,
4754 fenc, wfname, fname) == FAIL)
4755 {
4756 write_info.bw_conv_error = TRUE;
4757 end = 0;
4758 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004759 }
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004760 mch_remove(wfname);
4761 vim_free(wfname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004762 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004763#endif
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004764 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004765
4766 if (end == 0)
4767 {
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004768 /*
4769 * Error encountered.
4770 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004771 if (errmsg == NULL)
4772 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00004773 if (write_info.bw_conv_error)
Bram Moolenaar32b485f2009-07-29 16:06:27 +00004774 {
4775 if (write_info.bw_conv_error_lnum == 0)
4776 errmsg = (char_u *)_("E513: write error, conversion failed (make 'fenc' empty to override)");
4777 else
4778 {
4779 errmsg_allocated = TRUE;
4780 errmsg = alloc(300);
4781 vim_snprintf((char *)errmsg, 300, _("E513: write error, conversion failed in line %ld (make 'fenc' empty to override)"),
4782 (long)write_info.bw_conv_error_lnum);
4783 }
4784 }
Bram Moolenaar13505972019-01-24 15:04:48 +01004785 else if (got_int)
4786 errmsg = (char_u *)_(e_interr);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004787 else
Bram Moolenaar13505972019-01-24 15:04:48 +01004788 errmsg = (char_u *)_("E514: write error (file system full?)");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004789 }
4790
4791 /*
4792 * If we have a backup file, try to put it in place of the new file,
Bram Moolenaare37d50a2008-08-06 17:06:04 +00004793 * because the new file is probably corrupt. This avoids losing the
Bram Moolenaar071d4272004-06-13 20:20:40 +00004794 * original file when trying to make a backup when writing the file a
4795 * second time.
4796 * When "backup_copy" is set we need to copy the backup over the new
4797 * file. Otherwise rename the backup file.
4798 * If this is OK, don't give the extra warning message.
4799 */
4800 if (backup != NULL)
4801 {
4802 if (backup_copy)
4803 {
4804 /* This may take a while, if we were interrupted let the user
4805 * know we got the message. */
4806 if (got_int)
4807 {
Bram Moolenaar32526b32019-01-19 17:43:09 +01004808 msg(_(e_interr));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004809 out_flush();
4810 }
4811 if ((fd = mch_open((char *)backup, O_RDONLY | O_EXTRA, 0)) >= 0)
4812 {
4813 if ((write_info.bw_fd = mch_open((char *)fname,
Bram Moolenaar9be038d2005-03-08 22:34:32 +00004814 O_WRONLY | O_CREAT | O_TRUNC | O_EXTRA,
4815 perm & 0777)) >= 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004816 {
4817 /* copy the file. */
4818 write_info.bw_buf = smallbuf;
4819#ifdef HAS_BW_FLAGS
4820 write_info.bw_flags = FIO_NOCONVERT;
4821#endif
Bram Moolenaar540fc6f2010-12-17 16:27:16 +01004822 while ((write_info.bw_len = read_eintr(fd, smallbuf,
Bram Moolenaar071d4272004-06-13 20:20:40 +00004823 SMBUFSIZE)) > 0)
4824 if (buf_write_bytes(&write_info) == FAIL)
4825 break;
4826
4827 if (close(write_info.bw_fd) >= 0
4828 && write_info.bw_len == 0)
4829 end = 1; /* success */
4830 }
4831 close(fd); /* ignore errors for closing read file */
4832 }
4833 }
4834 else
4835 {
4836 if (vim_rename(backup, fname) == 0)
4837 end = 1;
4838 }
4839 }
4840 goto fail;
4841 }
4842
4843 lnum -= start; /* compute number of written lines */
4844 --no_wait_return; /* may wait for return now */
4845
4846#if !(defined(UNIX) || defined(VMS))
4847 fname = sfname; /* use shortname now, for the messages */
4848#endif
4849 if (!filtering)
4850 {
4851 msg_add_fname(buf, fname); /* put fname in IObuff with quotes */
4852 c = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004853 if (write_info.bw_conv_error)
4854 {
4855 STRCAT(IObuff, _(" CONVERSION ERROR"));
4856 c = TRUE;
Bram Moolenaar32b485f2009-07-29 16:06:27 +00004857 if (write_info.bw_conv_error_lnum != 0)
Bram Moolenaara800b422010-06-27 01:15:55 +02004858 vim_snprintf_add((char *)IObuff, IOSIZE, _(" in line %ld;"),
Bram Moolenaar32b485f2009-07-29 16:06:27 +00004859 (long)write_info.bw_conv_error_lnum);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004860 }
4861 else if (notconverted)
4862 {
4863 STRCAT(IObuff, _("[NOT converted]"));
4864 c = TRUE;
4865 }
4866 else if (converted)
4867 {
4868 STRCAT(IObuff, _("[converted]"));
4869 c = TRUE;
4870 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004871 if (device)
4872 {
4873 STRCAT(IObuff, _("[Device]"));
4874 c = TRUE;
4875 }
4876 else if (newfile)
4877 {
4878 STRCAT(IObuff, shortmess(SHM_NEW) ? _("[New]") : _("[New File]"));
4879 c = TRUE;
4880 }
4881 if (no_eol)
4882 {
4883 msg_add_eol();
4884 c = TRUE;
4885 }
4886 /* may add [unix/dos/mac] */
4887 if (msg_add_fileformat(fileformat))
4888 c = TRUE;
4889#ifdef FEAT_CRYPT
4890 if (wb_flags & FIO_ENCRYPTED)
4891 {
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02004892 crypt_append_msg(buf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004893 c = TRUE;
4894 }
4895#endif
4896 msg_add_lines(c, (long)lnum, nchars); /* add line/char count */
4897 if (!shortmess(SHM_WRITE))
4898 {
4899 if (append)
4900 STRCAT(IObuff, shortmess(SHM_WRI) ? _(" [a]") : _(" appended"));
4901 else
4902 STRCAT(IObuff, shortmess(SHM_WRI) ? _(" [w]") : _(" written"));
4903 }
4904
Bram Moolenaar32526b32019-01-19 17:43:09 +01004905 set_keep_msg((char_u *)msg_trunc_attr((char *)IObuff, FALSE, 0), 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004906 }
4907
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00004908 /* When written everything correctly: reset 'modified'. Unless not
4909 * writing to the original file and '+' is not in 'cpoptions'. */
Bram Moolenaar292ad192005-12-11 21:29:51 +00004910 if (reset_changed && whole && !append
Bram Moolenaar071d4272004-06-13 20:20:40 +00004911 && !write_info.bw_conv_error
Bram Moolenaar13505972019-01-24 15:04:48 +01004912 && (overwriting || vim_strchr(p_cpo, CPO_PLUS) != NULL))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004913 {
4914 unchanged(buf, TRUE);
Bram Moolenaar95c526e2017-02-25 14:59:34 +01004915 /* b:changedtick is always incremented in unchanged() but that
Bram Moolenaar086329d2014-10-31 19:51:36 +01004916 * should not trigger a TextChanged event. */
Bram Moolenaar5a093432018-02-10 18:15:19 +01004917 if (buf->b_last_changedtick + 1 == CHANGEDTICK(buf))
4918 buf->b_last_changedtick = CHANGEDTICK(buf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004919 u_unchanged(buf);
Bram Moolenaar730cde92010-06-27 05:18:54 +02004920 u_update_save_nr(buf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004921 }
4922
4923 /*
4924 * If written to the current file, update the timestamp of the swap file
4925 * and reset the BF_WRITE_MASK flags. Also sets buf->b_mtime.
4926 */
4927 if (overwriting)
4928 {
4929 ml_timestamp(buf);
Bram Moolenaar292ad192005-12-11 21:29:51 +00004930 if (append)
4931 buf->b_flags &= ~BF_NEW;
4932 else
4933 buf->b_flags &= ~BF_WRITE_MASK;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004934 }
4935
4936 /*
4937 * If we kept a backup until now, and we are in patch mode, then we make
4938 * the backup file our 'original' file.
4939 */
4940 if (*p_pm && dobackup)
4941 {
Bram Moolenaar48e330a2016-02-23 14:53:34 +01004942 char *org = (char *)buf_modname((buf->b_p_sn || buf->b_shortname),
Bram Moolenaar071d4272004-06-13 20:20:40 +00004943 fname, p_pm, FALSE);
4944
4945 if (backup != NULL)
4946 {
Bram Moolenaar8767f522016-07-01 17:17:39 +02004947 stat_T st;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004948
4949 /*
4950 * If the original file does not exist yet
4951 * the current backup file becomes the original file
4952 */
4953 if (org == NULL)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01004954 emsg(_("E205: Patchmode: can't save original file"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004955 else if (mch_stat(org, &st) < 0)
4956 {
4957 vim_rename(backup, (char_u *)org);
Bram Moolenaard23a8232018-02-10 18:45:26 +01004958 VIM_CLEAR(backup); /* don't delete the file */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004959#ifdef UNIX
4960 set_file_time((char_u *)org, st_old.st_atime, st_old.st_mtime);
4961#endif
4962 }
4963 }
4964 /*
4965 * If there is no backup file, remember that a (new) file was
4966 * created.
4967 */
4968 else
4969 {
4970 int empty_fd;
4971
4972 if (org == NULL
Bram Moolenaara5792f52005-11-23 21:25:05 +00004973 || (empty_fd = mch_open(org,
4974 O_CREAT | O_EXTRA | O_EXCL | O_NOFOLLOW,
Bram Moolenaar4317d9b2005-03-18 20:25:31 +00004975 perm < 0 ? 0666 : (perm & 0777))) < 0)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01004976 emsg(_("E206: patchmode: can't touch empty original file"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004977 else
4978 close(empty_fd);
4979 }
4980 if (org != NULL)
4981 {
4982 mch_setperm((char_u *)org, mch_getperm(fname) & 0777);
4983 vim_free(org);
4984 }
4985 }
4986
4987 /*
4988 * Remove the backup unless 'backup' option is set
4989 */
4990 if (!p_bk && backup != NULL && mch_remove(backup) != 0)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01004991 emsg(_("E207: Can't delete backup file"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004992
Bram Moolenaar071d4272004-06-13 20:20:40 +00004993 goto nofail;
4994
4995 /*
4996 * Finish up. We get here either after failure or success.
4997 */
4998fail:
4999 --no_wait_return; /* may wait for return now */
5000nofail:
5001
5002 /* Done saving, we accept changed buffer warnings again */
5003 buf->b_saving = FALSE;
5004
5005 vim_free(backup);
5006 if (buffer != smallbuf)
5007 vim_free(buffer);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005008 vim_free(fenc_tofree);
5009 vim_free(write_info.bw_conv_buf);
Bram Moolenaar13505972019-01-24 15:04:48 +01005010#ifdef USE_ICONV
Bram Moolenaar071d4272004-06-13 20:20:40 +00005011 if (write_info.bw_iconv_fd != (iconv_t)-1)
5012 {
5013 iconv_close(write_info.bw_iconv_fd);
5014 write_info.bw_iconv_fd = (iconv_t)-1;
5015 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00005016#endif
5017#ifdef HAVE_ACL
5018 mch_free_acl(acl);
5019#endif
5020
5021 if (errmsg != NULL)
5022 {
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00005023 int numlen = errnum != NULL ? (int)STRLEN(errnum) : 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005024
Bram Moolenaar8820b482017-03-16 17:23:31 +01005025 attr = HL_ATTR(HLF_E); /* set highlight for error messages */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005026 msg_add_fname(buf,
5027#ifndef UNIX
5028 sfname
5029#else
5030 fname
5031#endif
5032 ); /* put file name in IObuff with quotes */
5033 if (STRLEN(IObuff) + STRLEN(errmsg) + numlen >= IOSIZE)
5034 IObuff[IOSIZE - STRLEN(errmsg) - numlen - 1] = NUL;
5035 /* If the error message has the form "is ...", put the error number in
5036 * front of the file name. */
5037 if (errnum != NULL)
5038 {
Bram Moolenaar3577c6f2008-06-24 21:16:56 +00005039 STRMOVE(IObuff + numlen, IObuff);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005040 mch_memmove(IObuff, errnum, (size_t)numlen);
5041 }
5042 STRCAT(IObuff, errmsg);
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01005043 emsg((char *)IObuff);
Bram Moolenaar32b485f2009-07-29 16:06:27 +00005044 if (errmsg_allocated)
5045 vim_free(errmsg);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005046
5047 retval = FAIL;
5048 if (end == 0)
5049 {
Bram Moolenaar32526b32019-01-19 17:43:09 +01005050 msg_puts_attr(_("\nWARNING: Original file may be lost or damaged\n"),
Bram Moolenaar071d4272004-06-13 20:20:40 +00005051 attr | MSG_HIST);
Bram Moolenaar32526b32019-01-19 17:43:09 +01005052 msg_puts_attr(_("don't quit the editor until the file is successfully written!"),
Bram Moolenaar071d4272004-06-13 20:20:40 +00005053 attr | MSG_HIST);
5054
5055 /* Update the timestamp to avoid an "overwrite changed file"
5056 * prompt when writing again. */
5057 if (mch_stat((char *)fname, &st_old) >= 0)
5058 {
5059 buf_store_time(buf, &st_old, fname);
5060 buf->b_mtime_read = buf->b_mtime;
5061 }
5062 }
5063 }
5064 msg_scroll = msg_save;
5065
Bram Moolenaar55debbe2010-05-23 23:34:36 +02005066#ifdef FEAT_PERSISTENT_UNDO
5067 /*
5068 * When writing the whole file and 'undofile' is set, also write the undo
5069 * file.
5070 */
5071 if (retval == OK && write_undo_file)
5072 {
5073 char_u hash[UNDO_HASH_SIZE];
5074
5075 sha256_finish(&sha_ctx, hash);
5076 u_write_undo(NULL, FALSE, buf, hash);
5077 }
5078#endif
5079
Bram Moolenaar071d4272004-06-13 20:20:40 +00005080#ifdef FEAT_EVAL
5081 if (!should_abort(retval))
5082#else
5083 if (!got_int)
5084#endif
5085 {
5086 aco_save_T aco;
5087
Bram Moolenaar68a33fc2012-04-25 16:50:48 +02005088 curbuf->b_no_eol_lnum = 0; /* in case it was set by the previous read */
5089
Bram Moolenaar071d4272004-06-13 20:20:40 +00005090 /*
5091 * Apply POST autocommands.
5092 * Careful: The autocommands may call buf_write() recursively!
5093 */
5094 aucmd_prepbuf(&aco, buf);
5095
5096 if (append)
5097 apply_autocmds_exarg(EVENT_FILEAPPENDPOST, fname, fname,
5098 FALSE, curbuf, eap);
5099 else if (filtering)
5100 apply_autocmds_exarg(EVENT_FILTERWRITEPOST, NULL, fname,
5101 FALSE, curbuf, eap);
5102 else if (reset_changed && whole)
5103 apply_autocmds_exarg(EVENT_BUFWRITEPOST, fname, fname,
5104 FALSE, curbuf, eap);
5105 else
5106 apply_autocmds_exarg(EVENT_FILEWRITEPOST, fname, fname,
5107 FALSE, curbuf, eap);
5108
5109 /* restore curwin/curbuf and a few other things */
5110 aucmd_restbuf(&aco);
5111
5112#ifdef FEAT_EVAL
5113 if (aborting()) /* autocmds may abort script processing */
5114 retval = FALSE;
5115#endif
5116 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00005117
5118 got_int |= prev_got_int;
5119
Bram Moolenaar071d4272004-06-13 20:20:40 +00005120 return retval;
5121}
5122
Bram Moolenaara7870192019-02-14 12:56:36 +01005123#if defined(HAVE_FSYNC) || defined(PROTO)
5124/*
5125 * Call fsync() with Mac-specific exception.
5126 * Return fsync() result: zero for success.
5127 */
5128 int
5129vim_fsync(int fd)
5130{
5131 int r;
5132
5133# ifdef MACOS_X
5134 r = fcntl(fd, F_FULLFSYNC);
Bram Moolenaar91668382019-02-21 12:16:12 +01005135 if (r != 0) // F_FULLFSYNC not working or not supported
Bram Moolenaara7870192019-02-14 12:56:36 +01005136# endif
5137 r = fsync(fd);
5138 return r;
5139}
5140#endif
5141
Bram Moolenaar071d4272004-06-13 20:20:40 +00005142/*
Bram Moolenaar2d3f4892006-01-20 23:02:51 +00005143 * Set the name of the current buffer. Use when the buffer doesn't have a
5144 * name and a ":r" or ":w" command with a file name is used.
5145 */
5146 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005147set_rw_fname(char_u *fname, char_u *sfname)
Bram Moolenaar2d3f4892006-01-20 23:02:51 +00005148{
Bram Moolenaar8b38e242009-06-16 13:35:20 +00005149 buf_T *buf = curbuf;
5150
Bram Moolenaar2d3f4892006-01-20 23:02:51 +00005151 /* It's like the unnamed buffer is deleted.... */
5152 if (curbuf->b_p_bl)
5153 apply_autocmds(EVENT_BUFDELETE, NULL, NULL, FALSE, curbuf);
5154 apply_autocmds(EVENT_BUFWIPEOUT, NULL, NULL, FALSE, curbuf);
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +01005155#ifdef FEAT_EVAL
Bram Moolenaar2d3f4892006-01-20 23:02:51 +00005156 if (aborting()) /* autocmds may abort script processing */
5157 return FAIL;
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +01005158#endif
Bram Moolenaar8b38e242009-06-16 13:35:20 +00005159 if (curbuf != buf)
5160 {
5161 /* We are in another buffer now, don't do the renaming. */
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01005162 emsg(_(e_auchangedbuf));
Bram Moolenaar8b38e242009-06-16 13:35:20 +00005163 return FAIL;
5164 }
Bram Moolenaar2d3f4892006-01-20 23:02:51 +00005165
5166 if (setfname(curbuf, fname, sfname, FALSE) == OK)
5167 curbuf->b_flags |= BF_NOTEDITED;
5168
Bram Moolenaar2d3f4892006-01-20 23:02:51 +00005169 /* ....and a new named one is created */
5170 apply_autocmds(EVENT_BUFNEW, NULL, NULL, FALSE, curbuf);
5171 if (curbuf->b_p_bl)
5172 apply_autocmds(EVENT_BUFADD, NULL, NULL, FALSE, curbuf);
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +01005173#ifdef FEAT_EVAL
Bram Moolenaar2d3f4892006-01-20 23:02:51 +00005174 if (aborting()) /* autocmds may abort script processing */
5175 return FAIL;
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +01005176#endif
Bram Moolenaar2d3f4892006-01-20 23:02:51 +00005177
5178 /* Do filetype detection now if 'filetype' is empty. */
5179 if (*curbuf->b_p_ft == NUL)
5180 {
Bram Moolenaar910f66f2006-04-05 20:41:53 +00005181 if (au_has_group((char_u *)"filetypedetect"))
Bram Moolenaar1610d052016-06-09 22:53:01 +02005182 (void)do_doautocmd((char_u *)"filetypedetect BufRead", FALSE, NULL);
Bram Moolenaara3227e22006-03-08 21:32:40 +00005183 do_modelines(0);
Bram Moolenaar2d3f4892006-01-20 23:02:51 +00005184 }
Bram Moolenaar2d3f4892006-01-20 23:02:51 +00005185
5186 return OK;
5187}
5188
5189/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00005190 * Put file name into IObuff with quotes.
5191 */
Bram Moolenaar009b2592004-10-24 19:18:58 +00005192 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005193msg_add_fname(buf_T *buf, char_u *fname)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005194{
5195 if (fname == NULL)
5196 fname = (char_u *)"-stdin-";
5197 home_replace(buf, fname, IObuff + 1, IOSIZE - 4, TRUE);
5198 IObuff[0] = '"';
5199 STRCAT(IObuff, "\" ");
5200}
5201
5202/*
5203 * Append message for text mode to IObuff.
5204 * Return TRUE if something appended.
5205 */
5206 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005207msg_add_fileformat(int eol_type)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005208{
5209#ifndef USE_CRNL
5210 if (eol_type == EOL_DOS)
5211 {
5212 STRCAT(IObuff, shortmess(SHM_TEXT) ? _("[dos]") : _("[dos format]"));
5213 return TRUE;
5214 }
5215#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00005216 if (eol_type == EOL_MAC)
5217 {
5218 STRCAT(IObuff, shortmess(SHM_TEXT) ? _("[mac]") : _("[mac format]"));
5219 return TRUE;
5220 }
Bram Moolenaar00590742019-02-15 21:06:09 +01005221#ifdef USE_CRNL
Bram Moolenaar071d4272004-06-13 20:20:40 +00005222 if (eol_type == EOL_UNIX)
5223 {
5224 STRCAT(IObuff, shortmess(SHM_TEXT) ? _("[unix]") : _("[unix format]"));
5225 return TRUE;
5226 }
5227#endif
5228 return FALSE;
5229}
5230
5231/*
5232 * Append line and character count to IObuff.
5233 */
Bram Moolenaar009b2592004-10-24 19:18:58 +00005234 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005235msg_add_lines(
5236 int insert_space,
5237 long lnum,
Bram Moolenaar8767f522016-07-01 17:17:39 +02005238 off_T nchars)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005239{
5240 char_u *p;
5241
5242 p = IObuff + STRLEN(IObuff);
5243
5244 if (insert_space)
5245 *p++ = ' ';
5246 if (shortmess(SHM_LINES))
Bram Moolenaarbde98102016-07-01 20:03:42 +02005247 vim_snprintf((char *)p, IOSIZE - (p - IObuff),
Bram Moolenaar88c86eb2019-01-17 17:13:30 +01005248 "%ldL, %lldC", lnum, (long_long_T)nchars);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005249 else
5250 {
Bram Moolenaarda6e8912018-08-21 15:12:14 +02005251 sprintf((char *)p, NGETTEXT("%ld line, ", "%ld lines, ", lnum), lnum);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005252 p += STRLEN(p);
Bram Moolenaarda6e8912018-08-21 15:12:14 +02005253 vim_snprintf((char *)p, IOSIZE - (p - IObuff),
5254 NGETTEXT("%lld character", "%lld characters", nchars),
Bram Moolenaar88c86eb2019-01-17 17:13:30 +01005255 (long_long_T)nchars);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005256 }
5257}
5258
5259/*
5260 * Append message for missing line separator to IObuff.
5261 */
5262 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005263msg_add_eol(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005264{
5265 STRCAT(IObuff, shortmess(SHM_LAST) ? _("[noeol]") : _("[Incomplete last line]"));
5266}
5267
5268/*
5269 * Check modification time of file, before writing to it.
5270 * The size isn't checked, because using a tool like "gzip" takes care of
5271 * using the same timestamp but can't set the size.
5272 */
5273 static int
Bram Moolenaar8767f522016-07-01 17:17:39 +02005274check_mtime(buf_T *buf, stat_T *st)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005275{
5276 if (buf->b_mtime_read != 0
5277 && time_differs((long)st->st_mtime, buf->b_mtime_read))
5278 {
5279 msg_scroll = TRUE; /* don't overwrite messages here */
5280 msg_silent = 0; /* must give this prompt */
5281 /* don't use emsg() here, don't want to flush the buffers */
Bram Moolenaar32526b32019-01-19 17:43:09 +01005282 msg_attr(_("WARNING: The file has been changed since reading it!!!"),
Bram Moolenaar8820b482017-03-16 17:23:31 +01005283 HL_ATTR(HLF_E));
Bram Moolenaar071d4272004-06-13 20:20:40 +00005284 if (ask_yesno((char_u *)_("Do you really want to write to it"),
5285 TRUE) == 'n')
5286 return FAIL;
5287 msg_scroll = FALSE; /* always overwrite the file message now */
5288 }
5289 return OK;
5290}
5291
5292 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005293time_differs(long t1, long t2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005294{
Bram Moolenaar48e330a2016-02-23 14:53:34 +01005295#if defined(__linux__) || defined(MSWIN)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005296 /* On a FAT filesystem, esp. under Linux, there are only 5 bits to store
5297 * the seconds. Since the roundoff is done when flushing the inode, the
5298 * time may change unexpectedly by one second!!! */
5299 return (t1 - t2 > 1 || t2 - t1 > 1);
5300#else
5301 return (t1 != t2);
5302#endif
5303}
5304
5305/*
5306 * Call write() to write a number of bytes to the file.
Bram Moolenaar540fc6f2010-12-17 16:27:16 +01005307 * Handles encryption and 'encoding' conversion.
Bram Moolenaar071d4272004-06-13 20:20:40 +00005308 *
5309 * Return FAIL for failure, OK otherwise.
5310 */
5311 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005312buf_write_bytes(struct bw_info *ip)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005313{
5314 int wlen;
5315 char_u *buf = ip->bw_buf; /* data to write */
5316 int len = ip->bw_len; /* length of data */
5317#ifdef HAS_BW_FLAGS
5318 int flags = ip->bw_flags; /* extra flags */
5319#endif
5320
Bram Moolenaar071d4272004-06-13 20:20:40 +00005321 /*
5322 * Skip conversion when writing the crypt magic number or the BOM.
5323 */
5324 if (!(flags & FIO_NOCONVERT))
5325 {
5326 char_u *p;
5327 unsigned c;
5328 int n;
5329
5330 if (flags & FIO_UTF8)
5331 {
5332 /*
5333 * Convert latin1 in the buffer to UTF-8 in the file.
5334 */
5335 p = ip->bw_conv_buf; /* translate to buffer */
5336 for (wlen = 0; wlen < len; ++wlen)
5337 p += utf_char2bytes(buf[wlen], p);
5338 buf = ip->bw_conv_buf;
5339 len = (int)(p - ip->bw_conv_buf);
5340 }
5341 else if (flags & (FIO_UCS4 | FIO_UTF16 | FIO_UCS2 | FIO_LATIN1))
5342 {
5343 /*
5344 * Convert UTF-8 bytes in the buffer to UCS-2, UCS-4, UTF-16 or
5345 * Latin1 chars in the file.
5346 */
5347 if (flags & FIO_LATIN1)
5348 p = buf; /* translate in-place (can only get shorter) */
5349 else
5350 p = ip->bw_conv_buf; /* translate to buffer */
5351 for (wlen = 0; wlen < len; wlen += n)
5352 {
5353 if (wlen == 0 && ip->bw_restlen != 0)
5354 {
5355 int l;
5356
5357 /* Use remainder of previous call. Append the start of
5358 * buf[] to get a full sequence. Might still be too
5359 * short! */
5360 l = CONV_RESTLEN - ip->bw_restlen;
5361 if (l > len)
5362 l = len;
5363 mch_memmove(ip->bw_rest + ip->bw_restlen, buf, (size_t)l);
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00005364 n = utf_ptr2len_len(ip->bw_rest, ip->bw_restlen + l);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005365 if (n > ip->bw_restlen + len)
5366 {
5367 /* We have an incomplete byte sequence at the end to
5368 * be written. We can't convert it without the
5369 * remaining bytes. Keep them for the next call. */
5370 if (ip->bw_restlen + len > CONV_RESTLEN)
5371 return FAIL;
5372 ip->bw_restlen += len;
5373 break;
5374 }
5375 if (n > 1)
5376 c = utf_ptr2char(ip->bw_rest);
5377 else
5378 c = ip->bw_rest[0];
5379 if (n >= ip->bw_restlen)
5380 {
5381 n -= ip->bw_restlen;
5382 ip->bw_restlen = 0;
5383 }
5384 else
5385 {
5386 ip->bw_restlen -= n;
5387 mch_memmove(ip->bw_rest, ip->bw_rest + n,
5388 (size_t)ip->bw_restlen);
5389 n = 0;
5390 }
5391 }
5392 else
5393 {
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00005394 n = utf_ptr2len_len(buf + wlen, len - wlen);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005395 if (n > len - wlen)
5396 {
5397 /* We have an incomplete byte sequence at the end to
5398 * be written. We can't convert it without the
5399 * remaining bytes. Keep them for the next call. */
5400 if (len - wlen > CONV_RESTLEN)
5401 return FAIL;
5402 ip->bw_restlen = len - wlen;
5403 mch_memmove(ip->bw_rest, buf + wlen,
5404 (size_t)ip->bw_restlen);
5405 break;
5406 }
5407 if (n > 1)
5408 c = utf_ptr2char(buf + wlen);
5409 else
5410 c = buf[wlen];
5411 }
5412
Bram Moolenaar32b485f2009-07-29 16:06:27 +00005413 if (ucs2bytes(c, &p, flags) && !ip->bw_conv_error)
5414 {
5415 ip->bw_conv_error = TRUE;
5416 ip->bw_conv_error_lnum = ip->bw_start_lnum;
5417 }
5418 if (c == NL)
5419 ++ip->bw_start_lnum;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005420 }
5421 if (flags & FIO_LATIN1)
5422 len = (int)(p - buf);
5423 else
5424 {
5425 buf = ip->bw_conv_buf;
5426 len = (int)(p - ip->bw_conv_buf);
5427 }
5428 }
5429
Bram Moolenaar4f974752019-02-17 17:44:42 +01005430#ifdef MSWIN
Bram Moolenaar071d4272004-06-13 20:20:40 +00005431 else if (flags & FIO_CODEPAGE)
5432 {
5433 /*
5434 * Convert UTF-8 or codepage to UCS-2 and then to MS-Windows
5435 * codepage.
5436 */
5437 char_u *from;
5438 size_t fromlen;
5439 char_u *to;
5440 int u8c;
5441 BOOL bad = FALSE;
5442 int needed;
5443
5444 if (ip->bw_restlen > 0)
5445 {
5446 /* Need to concatenate the remainder of the previous call and
5447 * the bytes of the current call. Use the end of the
5448 * conversion buffer for this. */
5449 fromlen = len + ip->bw_restlen;
5450 from = ip->bw_conv_buf + ip->bw_conv_buflen - fromlen;
5451 mch_memmove(from, ip->bw_rest, (size_t)ip->bw_restlen);
5452 mch_memmove(from + ip->bw_restlen, buf, (size_t)len);
5453 }
5454 else
5455 {
5456 from = buf;
5457 fromlen = len;
5458 }
5459
5460 to = ip->bw_conv_buf;
5461 if (enc_utf8)
5462 {
5463 /* Convert from UTF-8 to UCS-2, to the start of the buffer.
5464 * The buffer has been allocated to be big enough. */
5465 while (fromlen > 0)
5466 {
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00005467 n = (int)utf_ptr2len_len(from, (int)fromlen);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005468 if (n > (int)fromlen) /* incomplete byte sequence */
5469 break;
5470 u8c = utf_ptr2char(from);
5471 *to++ = (u8c & 0xff);
5472 *to++ = (u8c >> 8);
5473 fromlen -= n;
5474 from += n;
5475 }
5476
5477 /* Copy remainder to ip->bw_rest[] to be used for the next
5478 * call. */
5479 if (fromlen > CONV_RESTLEN)
5480 {
5481 /* weird overlong sequence */
5482 ip->bw_conv_error = TRUE;
5483 return FAIL;
5484 }
5485 mch_memmove(ip->bw_rest, from, fromlen);
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00005486 ip->bw_restlen = (int)fromlen;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005487 }
5488 else
5489 {
5490 /* Convert from enc_codepage to UCS-2, to the start of the
5491 * buffer. The buffer has been allocated to be big enough. */
5492 ip->bw_restlen = 0;
5493 needed = MultiByteToWideChar(enc_codepage,
Bram Moolenaar36f5ac02007-05-06 12:40:34 +00005494 MB_ERR_INVALID_CHARS, (LPCSTR)from, (int)fromlen,
Bram Moolenaar071d4272004-06-13 20:20:40 +00005495 NULL, 0);
5496 if (needed == 0)
5497 {
5498 /* When conversion fails there may be a trailing byte. */
5499 needed = MultiByteToWideChar(enc_codepage,
Bram Moolenaar36f5ac02007-05-06 12:40:34 +00005500 MB_ERR_INVALID_CHARS, (LPCSTR)from, (int)fromlen - 1,
Bram Moolenaar071d4272004-06-13 20:20:40 +00005501 NULL, 0);
5502 if (needed == 0)
5503 {
5504 /* Conversion doesn't work. */
5505 ip->bw_conv_error = TRUE;
5506 return FAIL;
5507 }
5508 /* Save the trailing byte for the next call. */
5509 ip->bw_rest[0] = from[fromlen - 1];
5510 ip->bw_restlen = 1;
5511 }
5512 needed = MultiByteToWideChar(enc_codepage, MB_ERR_INVALID_CHARS,
Bram Moolenaar36f5ac02007-05-06 12:40:34 +00005513 (LPCSTR)from, (int)(fromlen - ip->bw_restlen),
Bram Moolenaar071d4272004-06-13 20:20:40 +00005514 (LPWSTR)to, needed);
5515 if (needed == 0)
5516 {
5517 /* Safety check: Conversion doesn't work. */
5518 ip->bw_conv_error = TRUE;
5519 return FAIL;
5520 }
5521 to += needed * 2;
5522 }
5523
5524 fromlen = to - ip->bw_conv_buf;
5525 buf = to;
Bram Moolenaar13505972019-01-24 15:04:48 +01005526# ifdef CP_UTF8 /* VC 4.1 doesn't define CP_UTF8 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005527 if (FIO_GET_CP(flags) == CP_UTF8)
5528 {
5529 /* Convert from UCS-2 to UTF-8, using the remainder of the
5530 * conversion buffer. Fails when out of space. */
5531 for (from = ip->bw_conv_buf; fromlen > 1; fromlen -= 2)
5532 {
5533 u8c = *from++;
5534 u8c += (*from++ << 8);
5535 to += utf_char2bytes(u8c, to);
5536 if (to + 6 >= ip->bw_conv_buf + ip->bw_conv_buflen)
5537 {
5538 ip->bw_conv_error = TRUE;
5539 return FAIL;
5540 }
5541 }
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00005542 len = (int)(to - buf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005543 }
5544 else
Bram Moolenaar13505972019-01-24 15:04:48 +01005545# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00005546 {
5547 /* Convert from UCS-2 to the codepage, using the remainder of
5548 * the conversion buffer. If the conversion uses the default
5549 * character "0", the data doesn't fit in this encoding, so
5550 * fail. */
5551 len = WideCharToMultiByte(FIO_GET_CP(flags), 0,
5552 (LPCWSTR)ip->bw_conv_buf, (int)fromlen / sizeof(WCHAR),
Bram Moolenaar36f5ac02007-05-06 12:40:34 +00005553 (LPSTR)to, (int)(ip->bw_conv_buflen - fromlen), 0,
5554 &bad);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005555 if (bad)
5556 {
5557 ip->bw_conv_error = TRUE;
5558 return FAIL;
5559 }
5560 }
5561 }
Bram Moolenaar13505972019-01-24 15:04:48 +01005562#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00005563
Bram Moolenaar13505972019-01-24 15:04:48 +01005564#ifdef MACOS_CONVERT
Bram Moolenaar071d4272004-06-13 20:20:40 +00005565 else if (flags & FIO_MACROMAN)
5566 {
5567 /*
5568 * Convert UTF-8 or latin1 to Apple MacRoman.
5569 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005570 char_u *from;
5571 size_t fromlen;
5572
5573 if (ip->bw_restlen > 0)
5574 {
5575 /* Need to concatenate the remainder of the previous call and
5576 * the bytes of the current call. Use the end of the
5577 * conversion buffer for this. */
5578 fromlen = len + ip->bw_restlen;
5579 from = ip->bw_conv_buf + ip->bw_conv_buflen - fromlen;
5580 mch_memmove(from, ip->bw_rest, (size_t)ip->bw_restlen);
5581 mch_memmove(from + ip->bw_restlen, buf, (size_t)len);
5582 }
5583 else
5584 {
5585 from = buf;
5586 fromlen = len;
5587 }
5588
Bram Moolenaarab79bcb2004-07-18 21:34:53 +00005589 if (enc2macroman(from, fromlen,
5590 ip->bw_conv_buf, &len, ip->bw_conv_buflen,
5591 ip->bw_rest, &ip->bw_restlen) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005592 {
5593 ip->bw_conv_error = TRUE;
5594 return FAIL;
5595 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00005596 buf = ip->bw_conv_buf;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005597 }
Bram Moolenaar13505972019-01-24 15:04:48 +01005598#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00005599
Bram Moolenaar13505972019-01-24 15:04:48 +01005600#ifdef USE_ICONV
Bram Moolenaar071d4272004-06-13 20:20:40 +00005601 if (ip->bw_iconv_fd != (iconv_t)-1)
5602 {
5603 const char *from;
5604 size_t fromlen;
5605 char *to;
5606 size_t tolen;
5607
5608 /* Convert with iconv(). */
5609 if (ip->bw_restlen > 0)
5610 {
Bram Moolenaar5d294d12009-03-11 12:11:02 +00005611 char *fp;
5612
Bram Moolenaar071d4272004-06-13 20:20:40 +00005613 /* Need to concatenate the remainder of the previous call and
5614 * the bytes of the current call. Use the end of the
5615 * conversion buffer for this. */
5616 fromlen = len + ip->bw_restlen;
Bram Moolenaar5d294d12009-03-11 12:11:02 +00005617 fp = (char *)ip->bw_conv_buf + ip->bw_conv_buflen - fromlen;
5618 mch_memmove(fp, ip->bw_rest, (size_t)ip->bw_restlen);
5619 mch_memmove(fp + ip->bw_restlen, buf, (size_t)len);
5620 from = fp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005621 tolen = ip->bw_conv_buflen - fromlen;
5622 }
5623 else
5624 {
5625 from = (const char *)buf;
5626 fromlen = len;
5627 tolen = ip->bw_conv_buflen;
5628 }
5629 to = (char *)ip->bw_conv_buf;
5630
5631 if (ip->bw_first)
5632 {
5633 size_t save_len = tolen;
5634
5635 /* output the initial shift state sequence */
5636 (void)iconv(ip->bw_iconv_fd, NULL, NULL, &to, &tolen);
5637
5638 /* There is a bug in iconv() on Linux (which appears to be
5639 * wide-spread) which sets "to" to NULL and messes up "tolen".
5640 */
5641 if (to == NULL)
5642 {
5643 to = (char *)ip->bw_conv_buf;
5644 tolen = save_len;
5645 }
5646 ip->bw_first = FALSE;
5647 }
5648
5649 /*
5650 * If iconv() has an error or there is not enough room, fail.
5651 */
5652 if ((iconv(ip->bw_iconv_fd, (void *)&from, &fromlen, &to, &tolen)
5653 == (size_t)-1 && ICONV_ERRNO != ICONV_EINVAL)
5654 || fromlen > CONV_RESTLEN)
5655 {
5656 ip->bw_conv_error = TRUE;
5657 return FAIL;
5658 }
5659
5660 /* copy remainder to ip->bw_rest[] to be used for the next call. */
5661 if (fromlen > 0)
5662 mch_memmove(ip->bw_rest, (void *)from, fromlen);
5663 ip->bw_restlen = (int)fromlen;
5664
5665 buf = ip->bw_conv_buf;
5666 len = (int)((char_u *)to - ip->bw_conv_buf);
5667 }
Bram Moolenaar13505972019-01-24 15:04:48 +01005668#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00005669 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00005670
Bram Moolenaare6bf6552017-06-27 22:11:51 +02005671 if (ip->bw_fd < 0)
5672 /* Only checking conversion, which is OK if we get here. */
5673 return OK;
5674
Bram Moolenaar071d4272004-06-13 20:20:40 +00005675#ifdef FEAT_CRYPT
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02005676 if (flags & FIO_ENCRYPTED)
5677 {
5678 /* Encrypt the data. Do it in-place if possible, otherwise use an
5679 * allocated buffer. */
Bram Moolenaar987411d2019-01-18 22:48:34 +01005680# ifdef CRYPT_NOT_INPLACE
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02005681 if (crypt_works_inplace(ip->bw_buffer->b_cryptstate))
5682 {
Bram Moolenaar987411d2019-01-18 22:48:34 +01005683# endif
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02005684 crypt_encode_inplace(ip->bw_buffer->b_cryptstate, buf, len);
Bram Moolenaar987411d2019-01-18 22:48:34 +01005685# ifdef CRYPT_NOT_INPLACE
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02005686 }
5687 else
5688 {
5689 char_u *outbuf;
5690
5691 len = crypt_encode_alloc(curbuf->b_cryptstate, buf, len, &outbuf);
5692 if (len == 0)
5693 return OK; /* Crypt layer is buffering, will flush later. */
5694 wlen = write_eintr(ip->bw_fd, outbuf, len);
5695 vim_free(outbuf);
5696 return (wlen < len) ? FAIL : OK;
5697 }
Bram Moolenaar987411d2019-01-18 22:48:34 +01005698# endif
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02005699 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00005700#endif
5701
Bram Moolenaar540fc6f2010-12-17 16:27:16 +01005702 wlen = write_eintr(ip->bw_fd, buf, len);
5703 return (wlen < len) ? FAIL : OK;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005704}
5705
Bram Moolenaar071d4272004-06-13 20:20:40 +00005706/*
5707 * Convert a Unicode character to bytes.
Bram Moolenaar32b485f2009-07-29 16:06:27 +00005708 * Return TRUE for an error, FALSE when it's OK.
Bram Moolenaar071d4272004-06-13 20:20:40 +00005709 */
5710 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005711ucs2bytes(
5712 unsigned c, /* in: character */
5713 char_u **pp, /* in/out: pointer to result */
5714 int flags) /* FIO_ flags */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005715{
5716 char_u *p = *pp;
5717 int error = FALSE;
5718 int cc;
5719
5720
5721 if (flags & FIO_UCS4)
5722 {
5723 if (flags & FIO_ENDIAN_L)
5724 {
5725 *p++ = c;
5726 *p++ = (c >> 8);
5727 *p++ = (c >> 16);
5728 *p++ = (c >> 24);
5729 }
5730 else
5731 {
5732 *p++ = (c >> 24);
5733 *p++ = (c >> 16);
5734 *p++ = (c >> 8);
5735 *p++ = c;
5736 }
5737 }
5738 else if (flags & (FIO_UCS2 | FIO_UTF16))
5739 {
5740 if (c >= 0x10000)
5741 {
5742 if (flags & FIO_UTF16)
5743 {
5744 /* Make two words, ten bits of the character in each. First
5745 * word is 0xd800 - 0xdbff, second one 0xdc00 - 0xdfff */
5746 c -= 0x10000;
5747 if (c >= 0x100000)
5748 error = TRUE;
5749 cc = ((c >> 10) & 0x3ff) + 0xd800;
5750 if (flags & FIO_ENDIAN_L)
5751 {
5752 *p++ = cc;
5753 *p++ = ((unsigned)cc >> 8);
5754 }
5755 else
5756 {
5757 *p++ = ((unsigned)cc >> 8);
5758 *p++ = cc;
5759 }
5760 c = (c & 0x3ff) + 0xdc00;
5761 }
5762 else
5763 error = TRUE;
5764 }
5765 if (flags & FIO_ENDIAN_L)
5766 {
5767 *p++ = c;
5768 *p++ = (c >> 8);
5769 }
5770 else
5771 {
5772 *p++ = (c >> 8);
5773 *p++ = c;
5774 }
5775 }
5776 else /* Latin1 */
5777 {
5778 if (c >= 0x100)
5779 {
5780 error = TRUE;
5781 *p++ = 0xBF;
5782 }
5783 else
5784 *p++ = c;
5785 }
5786
5787 *pp = p;
5788 return error;
5789}
5790
5791/*
Bram Moolenaarb5cdf2e2009-07-29 16:25:31 +00005792 * Return TRUE if file encoding "fenc" requires conversion from or to
5793 * 'encoding'.
Bram Moolenaar071d4272004-06-13 20:20:40 +00005794 */
5795 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005796need_conversion(char_u *fenc)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005797{
Bram Moolenaarb5cdf2e2009-07-29 16:25:31 +00005798 int same_encoding;
5799 int enc_flags;
5800 int fenc_flags;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005801
Bram Moolenaarb5cdf2e2009-07-29 16:25:31 +00005802 if (*fenc == NUL || STRCMP(p_enc, fenc) == 0)
Bram Moolenaar442b4222010-05-24 21:34:22 +02005803 {
Bram Moolenaarb5cdf2e2009-07-29 16:25:31 +00005804 same_encoding = TRUE;
Bram Moolenaar442b4222010-05-24 21:34:22 +02005805 fenc_flags = 0;
5806 }
Bram Moolenaarb5cdf2e2009-07-29 16:25:31 +00005807 else
5808 {
5809 /* Ignore difference between "ansi" and "latin1", "ucs-4" and
5810 * "ucs-4be", etc. */
5811 enc_flags = get_fio_flags(p_enc);
5812 fenc_flags = get_fio_flags(fenc);
5813 same_encoding = (enc_flags != 0 && fenc_flags == enc_flags);
5814 }
5815 if (same_encoding)
5816 {
5817 /* Specified encoding matches with 'encoding'. This requires
5818 * conversion when 'encoding' is Unicode but not UTF-8. */
5819 return enc_unicode != 0;
5820 }
5821
5822 /* Encodings differ. However, conversion is not needed when 'enc' is any
5823 * Unicode encoding and the file is UTF-8. */
5824 return !(enc_utf8 && fenc_flags == FIO_UTF8);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005825}
5826
5827/*
5828 * Check "ptr" for a unicode encoding and return the FIO_ flags needed for the
5829 * internal conversion.
5830 * if "ptr" is an empty string, use 'encoding'.
5831 */
5832 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005833get_fio_flags(char_u *ptr)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005834{
5835 int prop;
5836
5837 if (*ptr == NUL)
5838 ptr = p_enc;
5839
5840 prop = enc_canon_props(ptr);
5841 if (prop & ENC_UNICODE)
5842 {
5843 if (prop & ENC_2BYTE)
5844 {
5845 if (prop & ENC_ENDIAN_L)
5846 return FIO_UCS2 | FIO_ENDIAN_L;
5847 return FIO_UCS2;
5848 }
5849 if (prop & ENC_4BYTE)
5850 {
5851 if (prop & ENC_ENDIAN_L)
5852 return FIO_UCS4 | FIO_ENDIAN_L;
5853 return FIO_UCS4;
5854 }
5855 if (prop & ENC_2WORD)
5856 {
5857 if (prop & ENC_ENDIAN_L)
5858 return FIO_UTF16 | FIO_ENDIAN_L;
5859 return FIO_UTF16;
5860 }
5861 return FIO_UTF8;
5862 }
5863 if (prop & ENC_LATIN1)
5864 return FIO_LATIN1;
5865 /* must be ENC_DBCS, requires iconv() */
5866 return 0;
5867}
5868
Bram Moolenaar4f974752019-02-17 17:44:42 +01005869#ifdef MSWIN
Bram Moolenaar071d4272004-06-13 20:20:40 +00005870/*
5871 * Check "ptr" for a MS-Windows codepage name and return the FIO_ flags needed
5872 * for the conversion MS-Windows can do for us. Also accept "utf-8".
5873 * Used for conversion between 'encoding' and 'fileencoding'.
5874 */
5875 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005876get_win_fio_flags(char_u *ptr)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005877{
5878 int cp;
5879
5880 /* Cannot do this when 'encoding' is not utf-8 and not a codepage. */
5881 if (!enc_utf8 && enc_codepage <= 0)
5882 return 0;
5883
5884 cp = encname2codepage(ptr);
5885 if (cp == 0)
5886 {
5887# ifdef CP_UTF8 /* VC 4.1 doesn't define CP_UTF8 */
5888 if (STRCMP(ptr, "utf-8") == 0)
5889 cp = CP_UTF8;
5890 else
5891# endif
5892 return 0;
5893 }
5894 return FIO_PUT_CP(cp) | FIO_CODEPAGE;
5895}
5896#endif
5897
Bram Moolenaard0573012017-10-28 21:11:06 +02005898#ifdef MACOS_CONVERT
Bram Moolenaar071d4272004-06-13 20:20:40 +00005899/*
5900 * Check "ptr" for a Carbon supported encoding and return the FIO_ flags
5901 * needed for the internal conversion to/from utf-8 or latin1.
5902 */
5903 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005904get_mac_fio_flags(char_u *ptr)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005905{
5906 if ((enc_utf8 || STRCMP(p_enc, "latin1") == 0)
5907 && (enc_canon_props(ptr) & ENC_MACROMAN))
5908 return FIO_MACROMAN;
5909 return 0;
5910}
5911#endif
5912
5913/*
5914 * Check for a Unicode BOM (Byte Order Mark) at the start of p[size].
5915 * "size" must be at least 2.
5916 * Return the name of the encoding and set "*lenp" to the length.
5917 * Returns NULL when no BOM found.
5918 */
5919 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005920check_for_bom(
5921 char_u *p,
5922 long size,
5923 int *lenp,
5924 int flags)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005925{
5926 char *name = NULL;
5927 int len = 2;
5928
5929 if (p[0] == 0xef && p[1] == 0xbb && size >= 3 && p[2] == 0xbf
Bram Moolenaaree0f5a62008-07-24 20:09:16 +00005930 && (flags == FIO_ALL || flags == FIO_UTF8 || flags == 0))
Bram Moolenaar071d4272004-06-13 20:20:40 +00005931 {
5932 name = "utf-8"; /* EF BB BF */
5933 len = 3;
5934 }
5935 else if (p[0] == 0xff && p[1] == 0xfe)
5936 {
5937 if (size >= 4 && p[2] == 0 && p[3] == 0
5938 && (flags == FIO_ALL || flags == (FIO_UCS4 | FIO_ENDIAN_L)))
5939 {
5940 name = "ucs-4le"; /* FF FE 00 00 */
5941 len = 4;
5942 }
Bram Moolenaar223a1892008-11-11 20:57:11 +00005943 else if (flags == (FIO_UCS2 | FIO_ENDIAN_L))
Bram Moolenaar071d4272004-06-13 20:20:40 +00005944 name = "ucs-2le"; /* FF FE */
Bram Moolenaar223a1892008-11-11 20:57:11 +00005945 else if (flags == FIO_ALL || flags == (FIO_UTF16 | FIO_ENDIAN_L))
5946 /* utf-16le is preferred, it also works for ucs-2le text */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005947 name = "utf-16le"; /* FF FE */
5948 }
5949 else if (p[0] == 0xfe && p[1] == 0xff
5950 && (flags == FIO_ALL || flags == FIO_UCS2 || flags == FIO_UTF16))
5951 {
Bram Moolenaarffd82c52008-02-20 17:15:26 +00005952 /* Default to utf-16, it works also for ucs-2 text. */
5953 if (flags == FIO_UCS2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005954 name = "ucs-2"; /* FE FF */
Bram Moolenaarffd82c52008-02-20 17:15:26 +00005955 else
5956 name = "utf-16"; /* FE FF */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005957 }
5958 else if (size >= 4 && p[0] == 0 && p[1] == 0 && p[2] == 0xfe
5959 && p[3] == 0xff && (flags == FIO_ALL || flags == FIO_UCS4))
5960 {
5961 name = "ucs-4"; /* 00 00 FE FF */
5962 len = 4;
5963 }
5964
5965 *lenp = len;
5966 return (char_u *)name;
5967}
5968
5969/*
5970 * Generate a BOM in "buf[4]" for encoding "name".
5971 * Return the length of the BOM (zero when no BOM).
5972 */
5973 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005974make_bom(char_u *buf, char_u *name)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005975{
5976 int flags;
5977 char_u *p;
5978
5979 flags = get_fio_flags(name);
5980
5981 /* Can't put a BOM in a non-Unicode file. */
5982 if (flags == FIO_LATIN1 || flags == 0)
5983 return 0;
5984
5985 if (flags == FIO_UTF8) /* UTF-8 */
5986 {
5987 buf[0] = 0xef;
5988 buf[1] = 0xbb;
5989 buf[2] = 0xbf;
5990 return 3;
5991 }
5992 p = buf;
5993 (void)ucs2bytes(0xfeff, &p, flags);
5994 return (int)(p - buf);
5995}
Bram Moolenaar071d4272004-06-13 20:20:40 +00005996
5997/*
5998 * Try to find a shortname by comparing the fullname with the current
5999 * directory.
Bram Moolenaard089d9b2007-09-30 12:02:55 +00006000 * Returns "full_path" or pointer into "full_path" if shortened.
6001 */
6002 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01006003shorten_fname1(char_u *full_path)
Bram Moolenaard089d9b2007-09-30 12:02:55 +00006004{
Bram Moolenaard9462e32011-04-11 21:35:11 +02006005 char_u *dirname;
Bram Moolenaard089d9b2007-09-30 12:02:55 +00006006 char_u *p = full_path;
6007
Bram Moolenaard9462e32011-04-11 21:35:11 +02006008 dirname = alloc(MAXPATHL);
6009 if (dirname == NULL)
6010 return full_path;
Bram Moolenaard089d9b2007-09-30 12:02:55 +00006011 if (mch_dirname(dirname, MAXPATHL) == OK)
6012 {
6013 p = shorten_fname(full_path, dirname);
6014 if (p == NULL || *p == NUL)
6015 p = full_path;
6016 }
Bram Moolenaard9462e32011-04-11 21:35:11 +02006017 vim_free(dirname);
Bram Moolenaard089d9b2007-09-30 12:02:55 +00006018 return p;
6019}
6020
6021/*
6022 * Try to find a shortname by comparing the fullname with the current
6023 * directory.
Bram Moolenaar071d4272004-06-13 20:20:40 +00006024 * Returns NULL if not shorter name possible, pointer into "full_path"
6025 * otherwise.
6026 */
6027 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01006028shorten_fname(char_u *full_path, char_u *dir_name)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006029{
6030 int len;
6031 char_u *p;
6032
6033 if (full_path == NULL)
6034 return NULL;
6035 len = (int)STRLEN(dir_name);
6036 if (fnamencmp(dir_name, full_path, len) == 0)
6037 {
6038 p = full_path + len;
Bram Moolenaar48e330a2016-02-23 14:53:34 +01006039#if defined(MSWIN)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006040 /*
Bram Moolenaar4f974752019-02-17 17:44:42 +01006041 * MS-Windows: when a file is in the root directory, dir_name will end
6042 * in a slash, since C: by itself does not define a specific dir. In
6043 * this case p may already be correct. <negri>
Bram Moolenaar071d4272004-06-13 20:20:40 +00006044 */
6045 if (!((len > 2) && (*(p - 2) == ':')))
6046#endif
6047 {
6048 if (vim_ispathsep(*p))
6049 ++p;
6050#ifndef VMS /* the path separator is always part of the path */
6051 else
6052 p = NULL;
6053#endif
6054 }
6055 }
Bram Moolenaar48e330a2016-02-23 14:53:34 +01006056#if defined(MSWIN)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006057 /*
6058 * When using a file in the current drive, remove the drive name:
6059 * "A:\dir\file" -> "\dir\file". This helps when moving a session file on
6060 * a floppy from "A:\dir" to "B:\dir".
6061 */
6062 else if (len > 3
6063 && TOUPPER_LOC(full_path[0]) == TOUPPER_LOC(dir_name[0])
6064 && full_path[1] == ':'
6065 && vim_ispathsep(full_path[2]))
6066 p = full_path + 2;
6067#endif
6068 else
6069 p = NULL;
6070 return p;
6071}
6072
6073/*
Bram Moolenaara796d462018-05-01 14:30:36 +02006074 * Shorten filename of a buffer.
Bram Moolenaar071d4272004-06-13 20:20:40 +00006075 * When "force" is TRUE: Use full path from now on for files currently being
6076 * edited, both for file name and swap file name. Try to shorten the file
6077 * names a bit, if safe to do so.
6078 * When "force" is FALSE: Only try to shorten absolute file names.
6079 * For buffers that have buftype "nofile" or "scratch": never change the file
6080 * name.
6081 */
6082 void
Bram Moolenaara796d462018-05-01 14:30:36 +02006083shorten_buf_fname(buf_T *buf, char_u *dirname, int force)
6084{
6085 char_u *p;
6086
6087 if (buf->b_fname != NULL
6088#ifdef FEAT_QUICKFIX
6089 && !bt_nofile(buf)
6090#endif
6091 && !path_with_url(buf->b_fname)
6092 && (force
6093 || buf->b_sfname == NULL
6094 || mch_isFullName(buf->b_sfname)))
6095 {
Bram Moolenaar3d6014f2018-10-11 19:27:47 +02006096 if (buf->b_sfname != buf->b_ffname)
6097 VIM_CLEAR(buf->b_sfname);
Bram Moolenaara796d462018-05-01 14:30:36 +02006098 p = shorten_fname(buf->b_ffname, dirname);
6099 if (p != NULL)
6100 {
6101 buf->b_sfname = vim_strsave(p);
6102 buf->b_fname = buf->b_sfname;
6103 }
6104 if (p == NULL || buf->b_fname == NULL)
6105 buf->b_fname = buf->b_ffname;
6106 }
6107}
6108
6109/*
6110 * Shorten filenames for all buffers.
6111 */
6112 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01006113shorten_fnames(int force)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006114{
6115 char_u dirname[MAXPATHL];
6116 buf_T *buf;
Bram Moolenaar071d4272004-06-13 20:20:40 +00006117
6118 mch_dirname(dirname, MAXPATHL);
Bram Moolenaar29323592016-07-24 22:04:11 +02006119 FOR_ALL_BUFFERS(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006120 {
Bram Moolenaara796d462018-05-01 14:30:36 +02006121 shorten_buf_fname(buf, dirname, force);
Bram Moolenaar69a7cb42004-06-20 12:51:53 +00006122
6123 /* Always make the swap file name a full path, a "nofile" buffer may
6124 * also have a swap file. */
6125 mf_fullname(buf->b_ml.ml_mfp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00006126 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00006127 status_redraw_all();
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00006128 redraw_tabline = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00006129}
6130
6131#if (defined(FEAT_DND) && defined(FEAT_GUI_GTK)) \
6132 || defined(FEAT_GUI_MSWIN) \
6133 || defined(FEAT_GUI_MAC) \
6134 || defined(PROTO)
6135/*
6136 * Shorten all filenames in "fnames[count]" by current directory.
6137 */
6138 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01006139shorten_filenames(char_u **fnames, int count)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006140{
6141 int i;
6142 char_u dirname[MAXPATHL];
6143 char_u *p;
6144
6145 if (fnames == NULL || count < 1)
6146 return;
6147 mch_dirname(dirname, sizeof(dirname));
6148 for (i = 0; i < count; ++i)
6149 {
6150 if ((p = shorten_fname(fnames[i], dirname)) != NULL)
6151 {
6152 /* shorten_fname() returns pointer in given "fnames[i]". If free
6153 * "fnames[i]" first, "p" becomes invalid. So we need to copy
6154 * "p" first then free fnames[i]. */
6155 p = vim_strsave(p);
6156 vim_free(fnames[i]);
6157 fnames[i] = p;
6158 }
6159 }
6160}
6161#endif
6162
6163/*
Bram Moolenaarb782ba42018-08-07 21:39:28 +02006164 * Add extension to file name - change path/fo.o.h to path/fo.o.h.ext or
Bram Moolenaar071d4272004-06-13 20:20:40 +00006165 * fo_o_h.ext for MSDOS or when shortname option set.
6166 *
6167 * Assumed that fname is a valid name found in the filesystem we assure that
6168 * the return value is a different name and ends in 'ext'.
6169 * "ext" MUST be at most 4 characters long if it starts with a dot, 3
6170 * characters otherwise.
6171 * Space for the returned name is allocated, must be freed later.
6172 * Returns NULL when out of memory.
6173 */
6174 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01006175modname(
6176 char_u *fname,
6177 char_u *ext,
6178 int prepend_dot) /* may prepend a '.' to file name */
Bram Moolenaar071d4272004-06-13 20:20:40 +00006179{
Bram Moolenaar48e330a2016-02-23 14:53:34 +01006180 return buf_modname((curbuf->b_p_sn || curbuf->b_shortname),
Bram Moolenaar071d4272004-06-13 20:20:40 +00006181 fname, ext, prepend_dot);
6182}
6183
6184 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01006185buf_modname(
6186 int shortname, /* use 8.3 file name */
6187 char_u *fname,
6188 char_u *ext,
6189 int prepend_dot) /* may prepend a '.' to file name */
Bram Moolenaar071d4272004-06-13 20:20:40 +00006190{
6191 char_u *retval;
6192 char_u *s;
6193 char_u *e;
6194 char_u *ptr;
6195 int fnamelen, extlen;
6196
6197 extlen = (int)STRLEN(ext);
6198
6199 /*
6200 * If there is no file name we must get the name of the current directory
6201 * (we need the full path in case :cd is used).
6202 */
6203 if (fname == NULL || *fname == NUL)
6204 {
6205 retval = alloc((unsigned)(MAXPATHL + extlen + 3));
6206 if (retval == NULL)
6207 return NULL;
6208 if (mch_dirname(retval, MAXPATHL) == FAIL ||
6209 (fnamelen = (int)STRLEN(retval)) == 0)
6210 {
6211 vim_free(retval);
6212 return NULL;
6213 }
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00006214 if (!after_pathsep(retval, retval + fnamelen))
Bram Moolenaar071d4272004-06-13 20:20:40 +00006215 {
6216 retval[fnamelen++] = PATHSEP;
6217 retval[fnamelen] = NUL;
6218 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00006219 prepend_dot = FALSE; /* nothing to prepend a dot to */
Bram Moolenaar071d4272004-06-13 20:20:40 +00006220 }
6221 else
6222 {
6223 fnamelen = (int)STRLEN(fname);
6224 retval = alloc((unsigned)(fnamelen + extlen + 3));
6225 if (retval == NULL)
6226 return NULL;
6227 STRCPY(retval, fname);
6228#ifdef VMS
6229 vms_remove_version(retval); /* we do not need versions here */
6230#endif
6231 }
6232
6233 /*
6234 * search backwards until we hit a '/', '\' or ':' replacing all '.'
6235 * by '_' for MSDOS or when shortname option set and ext starts with a dot.
6236 * Then truncate what is after the '/', '\' or ':' to 8 characters for
6237 * MSDOS and 26 characters for AMIGA, a lot more for UNIX.
6238 */
Bram Moolenaar91acfff2017-03-12 19:22:36 +01006239 for (ptr = retval + fnamelen; ptr > retval; MB_PTR_BACK(retval, ptr))
Bram Moolenaar071d4272004-06-13 20:20:40 +00006240 {
Bram Moolenaar00f148d2019-02-12 22:37:27 +01006241 if (*ext == '.' && shortname)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006242 if (*ptr == '.') /* replace '.' by '_' */
6243 *ptr = '_';
Bram Moolenaar071d4272004-06-13 20:20:40 +00006244 if (vim_ispathsep(*ptr))
Bram Moolenaar53180ce2005-07-05 21:48:14 +00006245 {
6246 ++ptr;
Bram Moolenaar071d4272004-06-13 20:20:40 +00006247 break;
Bram Moolenaar53180ce2005-07-05 21:48:14 +00006248 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00006249 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00006250
6251 /* the file name has at most BASENAMELEN characters. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00006252 if (STRLEN(ptr) > (unsigned)BASENAMELEN)
6253 ptr[BASENAMELEN] = '\0';
Bram Moolenaar071d4272004-06-13 20:20:40 +00006254
6255 s = ptr + STRLEN(ptr);
6256
6257 /*
6258 * For 8.3 file names we may have to reduce the length.
6259 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00006260 if (shortname)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006261 {
6262 /*
6263 * If there is no file name, or the file name ends in '/', and the
6264 * extension starts with '.', put a '_' before the dot, because just
6265 * ".ext" is invalid.
6266 */
6267 if (fname == NULL || *fname == NUL
6268 || vim_ispathsep(fname[STRLEN(fname) - 1]))
6269 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00006270 if (*ext == '.')
Bram Moolenaar071d4272004-06-13 20:20:40 +00006271 *s++ = '_';
6272 }
6273 /*
6274 * If the extension starts with '.', truncate the base name at 8
6275 * characters
6276 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00006277 else if (*ext == '.')
Bram Moolenaar071d4272004-06-13 20:20:40 +00006278 {
Bram Moolenaar78a15312009-05-15 19:33:18 +00006279 if ((size_t)(s - ptr) > (size_t)8)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006280 {
6281 s = ptr + 8;
6282 *s = '\0';
6283 }
6284 }
6285 /*
6286 * If the extension doesn't start with '.', and the file name
6287 * doesn't have an extension yet, append a '.'
6288 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00006289 else if ((e = vim_strchr(ptr, '.')) == NULL)
6290 *s++ = '.';
Bram Moolenaar071d4272004-06-13 20:20:40 +00006291 /*
6292 * If the extension doesn't start with '.', and there already is an
Bram Moolenaar7263a772007-05-10 17:35:54 +00006293 * extension, it may need to be truncated
Bram Moolenaar071d4272004-06-13 20:20:40 +00006294 */
6295 else if ((int)STRLEN(e) + extlen > 4)
6296 s = e + 4 - extlen;
6297 }
Bram Moolenaar4f974752019-02-17 17:44:42 +01006298#ifdef MSWIN
Bram Moolenaar071d4272004-06-13 20:20:40 +00006299 /*
6300 * If there is no file name, and the extension starts with '.', put a
6301 * '_' before the dot, because just ".ext" may be invalid if it's on a
6302 * FAT partition, and on HPFS it doesn't matter.
6303 */
6304 else if ((fname == NULL || *fname == NUL) && *ext == '.')
6305 *s++ = '_';
6306#endif
6307
6308 /*
Bram Moolenaarb23a7e82008-06-27 18:42:32 +00006309 * Append the extension.
Bram Moolenaar071d4272004-06-13 20:20:40 +00006310 * ext can start with '.' and cannot exceed 3 more characters.
6311 */
6312 STRCPY(s, ext);
6313
Bram Moolenaar071d4272004-06-13 20:20:40 +00006314 /*
6315 * Prepend the dot.
6316 */
Bram Moolenaar00f148d2019-02-12 22:37:27 +01006317 if (prepend_dot && !shortname && *(e = gettail(retval)) != '.')
Bram Moolenaar071d4272004-06-13 20:20:40 +00006318 {
Bram Moolenaar3577c6f2008-06-24 21:16:56 +00006319 STRMOVE(e + 1, e);
Bram Moolenaar071d4272004-06-13 20:20:40 +00006320 *e = '.';
Bram Moolenaar071d4272004-06-13 20:20:40 +00006321 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00006322
6323 /*
6324 * Check that, after appending the extension, the file name is really
6325 * different.
6326 */
6327 if (fname != NULL && STRCMP(fname, retval) == 0)
6328 {
6329 /* we search for a character that can be replaced by '_' */
6330 while (--s >= ptr)
6331 {
6332 if (*s != '_')
6333 {
6334 *s = '_';
6335 break;
6336 }
6337 }
6338 if (s < ptr) /* fname was "________.<ext>", how tricky! */
6339 *ptr = 'v';
6340 }
6341 return retval;
6342}
6343
6344/*
6345 * Like fgets(), but if the file line is too long, it is truncated and the
6346 * rest of the line is thrown away. Returns TRUE for end-of-file.
Bram Moolenaar7e1652c2017-12-16 18:27:02 +01006347 * If the line is truncated then buf[size - 2] will not be NUL.
Bram Moolenaar071d4272004-06-13 20:20:40 +00006348 */
6349 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01006350vim_fgets(char_u *buf, int size, FILE *fp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006351{
6352 char *eof;
6353#define FGETS_SIZE 200
6354 char tbuf[FGETS_SIZE];
6355
6356 buf[size - 2] = NUL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00006357 eof = fgets((char *)buf, size, fp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00006358 if (buf[size - 2] != NUL && buf[size - 2] != '\n')
6359 {
6360 buf[size - 1] = NUL; /* Truncate the line */
6361
6362 /* Now throw away the rest of the line: */
6363 do
6364 {
6365 tbuf[FGETS_SIZE - 2] = NUL;
Bram Moolenaar42335f52018-09-13 15:33:43 +02006366 vim_ignoredp = fgets((char *)tbuf, FGETS_SIZE, fp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00006367 } while (tbuf[FGETS_SIZE - 2] != NUL && tbuf[FGETS_SIZE - 2] != '\n');
6368 }
6369 return (eof == NULL);
6370}
6371
Bram Moolenaar071d4272004-06-13 20:20:40 +00006372/*
6373 * rename() only works if both files are on the same file system, this
6374 * function will (attempts to?) copy the file across if rename fails -- webb
6375 * Return -1 for failure, 0 for success.
6376 */
6377 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01006378vim_rename(char_u *from, char_u *to)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006379{
6380 int fd_in;
6381 int fd_out;
6382 int n;
6383 char *errmsg = NULL;
6384 char *buffer;
6385#ifdef AMIGA
6386 BPTR flock;
6387#endif
Bram Moolenaar8767f522016-07-01 17:17:39 +02006388 stat_T st;
Bram Moolenaar9be038d2005-03-08 22:34:32 +00006389 long perm;
Bram Moolenaarcd71fa32005-03-11 22:46:48 +00006390#ifdef HAVE_ACL
6391 vim_acl_T acl; /* ACL from original file */
6392#endif
Bram Moolenaare0e6f992008-12-31 15:21:32 +00006393 int use_tmp_file = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00006394
6395 /*
Bram Moolenaare0e6f992008-12-31 15:21:32 +00006396 * When the names are identical, there is nothing to do. When they refer
6397 * to the same file (ignoring case and slash/backslash differences) but
6398 * the file name differs we need to go through a temp file.
Bram Moolenaar071d4272004-06-13 20:20:40 +00006399 */
6400 if (fnamecmp(from, to) == 0)
Bram Moolenaare0e6f992008-12-31 15:21:32 +00006401 {
Bram Moolenaar71afbfe2013-03-19 16:49:16 +01006402 if (p_fic && STRCMP(gettail(from), gettail(to)) != 0)
Bram Moolenaare0e6f992008-12-31 15:21:32 +00006403 use_tmp_file = TRUE;
6404 else
Bram Moolenaare0e6f992008-12-31 15:21:32 +00006405 return 0;
6406 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00006407
6408 /*
6409 * Fail if the "from" file doesn't exist. Avoids that "to" is deleted.
6410 */
6411 if (mch_stat((char *)from, &st) < 0)
6412 return -1;
6413
Bram Moolenaar3576da72008-12-30 15:15:57 +00006414#ifdef UNIX
6415 {
Bram Moolenaar8767f522016-07-01 17:17:39 +02006416 stat_T st_to;
Bram Moolenaar3576da72008-12-30 15:15:57 +00006417
6418 /* It's possible for the source and destination to be the same file.
6419 * This happens when "from" and "to" differ in case and are on a FAT32
6420 * filesystem. In that case go through a temp file name. */
6421 if (mch_stat((char *)to, &st_to) >= 0
6422 && st.st_dev == st_to.st_dev
6423 && st.st_ino == st_to.st_ino)
Bram Moolenaare0e6f992008-12-31 15:21:32 +00006424 use_tmp_file = TRUE;
6425 }
6426#endif
Bram Moolenaar4f974752019-02-17 17:44:42 +01006427#ifdef MSWIN
Bram Moolenaar1c32dff2011-05-05 16:41:24 +02006428 {
6429 BY_HANDLE_FILE_INFORMATION info1, info2;
6430
6431 /* It's possible for the source and destination to be the same file.
6432 * In that case go through a temp file name. This makes rename("foo",
6433 * "./foo") a no-op (in a complicated way). */
6434 if (win32_fileinfo(from, &info1) == FILEINFO_OK
6435 && win32_fileinfo(to, &info2) == FILEINFO_OK
6436 && info1.dwVolumeSerialNumber == info2.dwVolumeSerialNumber
6437 && info1.nFileIndexHigh == info2.nFileIndexHigh
6438 && info1.nFileIndexLow == info2.nFileIndexLow)
6439 use_tmp_file = TRUE;
6440 }
6441#endif
Bram Moolenaare0e6f992008-12-31 15:21:32 +00006442
Bram Moolenaare0e6f992008-12-31 15:21:32 +00006443 if (use_tmp_file)
6444 {
6445 char tempname[MAXPATHL + 1];
6446
6447 /*
6448 * Find a name that doesn't exist and is in the same directory.
6449 * Rename "from" to "tempname" and then rename "tempname" to "to".
6450 */
6451 if (STRLEN(from) >= MAXPATHL - 5)
6452 return -1;
6453 STRCPY(tempname, from);
6454 for (n = 123; n < 99999; ++n)
Bram Moolenaar3576da72008-12-30 15:15:57 +00006455 {
Bram Moolenaare0e6f992008-12-31 15:21:32 +00006456 sprintf((char *)gettail((char_u *)tempname), "%d", n);
6457 if (mch_stat(tempname, &st) < 0)
Bram Moolenaar3576da72008-12-30 15:15:57 +00006458 {
Bram Moolenaare0e6f992008-12-31 15:21:32 +00006459 if (mch_rename((char *)from, tempname) == 0)
Bram Moolenaar3576da72008-12-30 15:15:57 +00006460 {
Bram Moolenaare0e6f992008-12-31 15:21:32 +00006461 if (mch_rename(tempname, (char *)to) == 0)
6462 return 0;
6463 /* Strange, the second step failed. Try moving the
6464 * file back and return failure. */
6465 mch_rename(tempname, (char *)from);
Bram Moolenaar3576da72008-12-30 15:15:57 +00006466 return -1;
6467 }
Bram Moolenaare0e6f992008-12-31 15:21:32 +00006468 /* If it fails for one temp name it will most likely fail
6469 * for any temp name, give up. */
6470 return -1;
Bram Moolenaar3576da72008-12-30 15:15:57 +00006471 }
Bram Moolenaar3576da72008-12-30 15:15:57 +00006472 }
Bram Moolenaare0e6f992008-12-31 15:21:32 +00006473 return -1;
Bram Moolenaar3576da72008-12-30 15:15:57 +00006474 }
Bram Moolenaar3576da72008-12-30 15:15:57 +00006475
Bram Moolenaar071d4272004-06-13 20:20:40 +00006476 /*
6477 * Delete the "to" file, this is required on some systems to make the
6478 * mch_rename() work, on other systems it makes sure that we don't have
6479 * two files when the mch_rename() fails.
6480 */
6481
6482#ifdef AMIGA
6483 /*
6484 * With MSDOS-compatible filesystems (crossdos, messydos) it is possible
6485 * that the name of the "to" file is the same as the "from" file, even
Bram Moolenaar7263a772007-05-10 17:35:54 +00006486 * though the names are different. To avoid the chance of accidentally
Bram Moolenaar071d4272004-06-13 20:20:40 +00006487 * deleting the "from" file (horror!) we lock it during the remove.
6488 *
6489 * When used for making a backup before writing the file: This should not
6490 * happen with ":w", because startscript() should detect this problem and
6491 * set buf->b_shortname, causing modname() to return a correct ".bak" file
6492 * name. This problem does exist with ":w filename", but then the
6493 * original file will be somewhere else so the backup isn't really
6494 * important. If autoscripting is off the rename may fail.
6495 */
6496 flock = Lock((UBYTE *)from, (long)ACCESS_READ);
6497#endif
6498 mch_remove(to);
6499#ifdef AMIGA
6500 if (flock)
6501 UnLock(flock);
6502#endif
6503
6504 /*
6505 * First try a normal rename, return if it works.
6506 */
6507 if (mch_rename((char *)from, (char *)to) == 0)
6508 return 0;
6509
6510 /*
6511 * Rename() failed, try copying the file.
6512 */
Bram Moolenaar9be038d2005-03-08 22:34:32 +00006513 perm = mch_getperm(from);
Bram Moolenaarcd71fa32005-03-11 22:46:48 +00006514#ifdef HAVE_ACL
6515 /* For systems that support ACL: get the ACL from the original file. */
6516 acl = mch_get_acl(from);
6517#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00006518 fd_in = mch_open((char *)from, O_RDONLY|O_EXTRA, 0);
6519 if (fd_in == -1)
Bram Moolenaarb23a7e82008-06-27 18:42:32 +00006520 {
6521#ifdef HAVE_ACL
6522 mch_free_acl(acl);
6523#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00006524 return -1;
Bram Moolenaarb23a7e82008-06-27 18:42:32 +00006525 }
Bram Moolenaar9be038d2005-03-08 22:34:32 +00006526
6527 /* Create the new file with same permissions as the original. */
Bram Moolenaara5792f52005-11-23 21:25:05 +00006528 fd_out = mch_open((char *)to,
6529 O_CREAT|O_EXCL|O_WRONLY|O_EXTRA|O_NOFOLLOW, (int)perm);
Bram Moolenaar071d4272004-06-13 20:20:40 +00006530 if (fd_out == -1)
6531 {
6532 close(fd_in);
Bram Moolenaarb23a7e82008-06-27 18:42:32 +00006533#ifdef HAVE_ACL
6534 mch_free_acl(acl);
6535#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00006536 return -1;
6537 }
6538
6539 buffer = (char *)alloc(BUFSIZE);
6540 if (buffer == NULL)
6541 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00006542 close(fd_out);
Bram Moolenaarb23a7e82008-06-27 18:42:32 +00006543 close(fd_in);
6544#ifdef HAVE_ACL
6545 mch_free_acl(acl);
6546#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00006547 return -1;
6548 }
6549
Bram Moolenaar540fc6f2010-12-17 16:27:16 +01006550 while ((n = read_eintr(fd_in, buffer, BUFSIZE)) > 0)
6551 if (write_eintr(fd_out, buffer, n) != n)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006552 {
6553 errmsg = _("E208: Error writing to \"%s\"");
6554 break;
6555 }
6556
6557 vim_free(buffer);
6558 close(fd_in);
6559 if (close(fd_out) < 0)
6560 errmsg = _("E209: Error closing \"%s\"");
6561 if (n < 0)
6562 {
6563 errmsg = _("E210: Error reading \"%s\"");
6564 to = from;
6565 }
Bram Moolenaar7263a772007-05-10 17:35:54 +00006566#ifndef UNIX /* for Unix mch_open() already set the permission */
Bram Moolenaar9be038d2005-03-08 22:34:32 +00006567 mch_setperm(to, perm);
Bram Moolenaarc6039d82005-12-02 00:44:04 +00006568#endif
Bram Moolenaarcd71fa32005-03-11 22:46:48 +00006569#ifdef HAVE_ACL
6570 mch_set_acl(to, acl);
Bram Moolenaarb23a7e82008-06-27 18:42:32 +00006571 mch_free_acl(acl);
Bram Moolenaarcd71fa32005-03-11 22:46:48 +00006572#endif
Bram Moolenaar5bd32f42014-04-02 14:05:38 +02006573#if defined(HAVE_SELINUX) || defined(HAVE_SMACK)
Bram Moolenaare8747442013-11-12 18:09:29 +01006574 mch_copy_sec(from, to);
Bram Moolenaar0671de32013-11-12 05:12:03 +01006575#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00006576 if (errmsg != NULL)
6577 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006578 semsg(errmsg, to);
Bram Moolenaar071d4272004-06-13 20:20:40 +00006579 return -1;
6580 }
6581 mch_remove(from);
6582 return 0;
6583}
6584
6585static int already_warned = FALSE;
6586
6587/*
6588 * Check if any not hidden buffer has been changed.
6589 * Postpone the check if there are characters in the stuff buffer, a global
6590 * command is being executed, a mapping is being executed or an autocommand is
6591 * busy.
6592 * Returns TRUE if some message was written (screen should be redrawn and
6593 * cursor positioned).
6594 */
6595 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01006596check_timestamps(
6597 int focus) /* called for GUI focus event */
Bram Moolenaar071d4272004-06-13 20:20:40 +00006598{
6599 buf_T *buf;
6600 int didit = 0;
6601 int n;
6602
6603 /* Don't check timestamps while system() or another low-level function may
6604 * cause us to lose and gain focus. */
6605 if (no_check_timestamps > 0)
6606 return FALSE;
6607
6608 /* Avoid doing a check twice. The OK/Reload dialog can cause a focus
6609 * event and we would keep on checking if the file is steadily growing.
6610 * Do check again after typing something. */
6611 if (focus && did_check_timestamps)
6612 {
6613 need_check_timestamps = TRUE;
6614 return FALSE;
6615 }
6616
6617 if (!stuff_empty() || global_busy || !typebuf_typed()
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +01006618 || autocmd_busy || curbuf_lock > 0 || allbuf_lock > 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006619 need_check_timestamps = TRUE; /* check later */
6620 else
6621 {
6622 ++no_wait_return;
6623 did_check_timestamps = TRUE;
6624 already_warned = FALSE;
Bram Moolenaar29323592016-07-24 22:04:11 +02006625 FOR_ALL_BUFFERS(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006626 {
6627 /* Only check buffers in a window. */
6628 if (buf->b_nwindows > 0)
6629 {
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02006630 bufref_T bufref;
6631
6632 set_bufref(&bufref, buf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00006633 n = buf_check_timestamp(buf, focus);
6634 if (didit < n)
6635 didit = n;
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02006636 if (n > 0 && !bufref_valid(&bufref))
Bram Moolenaar071d4272004-06-13 20:20:40 +00006637 {
6638 /* Autocommands have removed the buffer, start at the
6639 * first one again. */
6640 buf = firstbuf;
6641 continue;
6642 }
6643 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00006644 }
6645 --no_wait_return;
6646 need_check_timestamps = FALSE;
6647 if (need_wait_return && didit == 2)
6648 {
6649 /* make sure msg isn't overwritten */
Bram Moolenaar32526b32019-01-19 17:43:09 +01006650 msg_puts("\n");
Bram Moolenaar071d4272004-06-13 20:20:40 +00006651 out_flush();
6652 }
6653 }
6654 return didit;
6655}
6656
6657/*
6658 * Move all the lines from buffer "frombuf" to buffer "tobuf".
6659 * Return OK or FAIL. When FAIL "tobuf" is incomplete and/or "frombuf" is not
6660 * empty.
6661 */
6662 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01006663move_lines(buf_T *frombuf, buf_T *tobuf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006664{
6665 buf_T *tbuf = curbuf;
6666 int retval = OK;
6667 linenr_T lnum;
6668 char_u *p;
6669
6670 /* Copy the lines in "frombuf" to "tobuf". */
6671 curbuf = tobuf;
6672 for (lnum = 1; lnum <= frombuf->b_ml.ml_line_count; ++lnum)
6673 {
6674 p = vim_strsave(ml_get_buf(frombuf, lnum, FALSE));
6675 if (p == NULL || ml_append(lnum - 1, p, 0, FALSE) == FAIL)
6676 {
6677 vim_free(p);
6678 retval = FAIL;
6679 break;
6680 }
6681 vim_free(p);
6682 }
6683
6684 /* Delete all the lines in "frombuf". */
6685 if (retval != FAIL)
6686 {
6687 curbuf = frombuf;
Bram Moolenaar9460b9d2007-01-09 14:37:01 +00006688 for (lnum = curbuf->b_ml.ml_line_count; lnum > 0; --lnum)
6689 if (ml_delete(lnum, FALSE) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006690 {
6691 /* Oops! We could try putting back the saved lines, but that
6692 * might fail again... */
6693 retval = FAIL;
6694 break;
6695 }
6696 }
6697
6698 curbuf = tbuf;
6699 return retval;
6700}
6701
6702/*
6703 * Check if buffer "buf" has been changed.
6704 * Also check if the file for a new buffer unexpectedly appeared.
6705 * return 1 if a changed buffer was found.
6706 * return 2 if a message has been displayed.
6707 * return 0 otherwise.
6708 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00006709 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01006710buf_check_timestamp(
6711 buf_T *buf,
6712 int focus UNUSED) /* called for GUI focus event */
Bram Moolenaar071d4272004-06-13 20:20:40 +00006713{
Bram Moolenaar8767f522016-07-01 17:17:39 +02006714 stat_T st;
Bram Moolenaar071d4272004-06-13 20:20:40 +00006715 int stat_res;
6716 int retval = 0;
6717 char_u *path;
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006718 char *tbuf;
Bram Moolenaar071d4272004-06-13 20:20:40 +00006719 char *mesg = NULL;
Bram Moolenaar44ecf652005-03-07 23:09:59 +00006720 char *mesg2 = "";
Bram Moolenaar071d4272004-06-13 20:20:40 +00006721 int helpmesg = FALSE;
6722 int reload = FALSE;
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02006723 char *reason;
Bram Moolenaar071d4272004-06-13 20:20:40 +00006724#if defined(FEAT_CON_DIALOG) || defined(FEAT_GUI_DIALOG)
6725 int can_reload = FALSE;
6726#endif
Bram Moolenaar8767f522016-07-01 17:17:39 +02006727 off_T orig_size = buf->b_orig_size;
Bram Moolenaar071d4272004-06-13 20:20:40 +00006728 int orig_mode = buf->b_orig_mode;
6729#ifdef FEAT_GUI
6730 int save_mouse_correct = need_mouse_correct;
6731#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00006732 static int busy = FALSE;
Bram Moolenaar19a09a12005-03-04 23:39:37 +00006733 int n;
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +01006734#ifdef FEAT_EVAL
Bram Moolenaar19a09a12005-03-04 23:39:37 +00006735 char_u *s;
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +01006736#endif
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02006737 bufref_T bufref;
6738
6739 set_bufref(&bufref, buf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00006740
6741 /* If there is no file name, the buffer is not loaded, 'buftype' is
6742 * set, we are in the middle of a save or being called recursively: ignore
6743 * this buffer. */
6744 if (buf->b_ffname == NULL
6745 || buf->b_ml.ml_mfp == NULL
Bram Moolenaar91335e52018-08-01 17:53:12 +02006746 || !bt_normal(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006747 || buf->b_saving
Bram Moolenaar071d4272004-06-13 20:20:40 +00006748 || busy
Bram Moolenaar009b2592004-10-24 19:18:58 +00006749#ifdef FEAT_NETBEANS_INTG
6750 || isNetbeansBuffer(buf)
6751#endif
Bram Moolenaar8cad9302017-08-12 14:32:32 +02006752#ifdef FEAT_TERMINAL
6753 || buf->b_term != NULL
6754#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00006755 )
6756 return 0;
6757
6758 if ( !(buf->b_flags & BF_NOTEDITED)
6759 && buf->b_mtime != 0
6760 && ((stat_res = mch_stat((char *)buf->b_ffname, &st)) < 0
6761 || time_differs((long)st.st_mtime, buf->b_mtime)
Bram Moolenaara7611f62014-05-02 15:46:14 +02006762 || st.st_size != buf->b_orig_size
Bram Moolenaar071d4272004-06-13 20:20:40 +00006763#ifdef HAVE_ST_MODE
6764 || (int)st.st_mode != buf->b_orig_mode
6765#else
6766 || mch_getperm(buf->b_ffname) != buf->b_orig_mode
6767#endif
6768 ))
6769 {
6770 retval = 1;
6771
Bram Moolenaar386bc822018-07-07 18:34:12 +02006772 // set b_mtime to stop further warnings (e.g., when executing
6773 // FileChangedShell autocmd)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006774 if (stat_res < 0)
6775 {
Bram Moolenaar386bc822018-07-07 18:34:12 +02006776 // When 'autoread' is set we'll check the file again to see if it
6777 // re-appears.
6778 buf->b_mtime = buf->b_p_ar;
Bram Moolenaar071d4272004-06-13 20:20:40 +00006779 buf->b_orig_size = 0;
6780 buf->b_orig_mode = 0;
6781 }
6782 else
6783 buf_store_time(buf, &st, buf->b_ffname);
6784
6785 /* Don't do anything for a directory. Might contain the file
6786 * explorer. */
6787 if (mch_isdir(buf->b_fname))
6788 ;
6789
6790 /*
6791 * If 'autoread' is set, the buffer has no changes and the file still
6792 * exists, reload the buffer. Use the buffer-local option value if it
6793 * was set, the global option value otherwise.
6794 */
6795 else if ((buf->b_p_ar >= 0 ? buf->b_p_ar : p_ar)
6796 && !bufIsChanged(buf) && stat_res >= 0)
6797 reload = TRUE;
6798 else
6799 {
Bram Moolenaar19a09a12005-03-04 23:39:37 +00006800 if (stat_res < 0)
6801 reason = "deleted";
6802 else if (bufIsChanged(buf))
6803 reason = "conflict";
Bram Moolenaar3e460fd2019-01-26 16:21:07 +01006804 /*
6805 * Check if the file contents really changed to avoid giving a
6806 * warning when only the timestamp was set (e.g., checked out of
6807 * CVS). Always warn when the buffer was changed.
6808 */
Bram Moolenaar19a09a12005-03-04 23:39:37 +00006809 else if (orig_size != buf->b_orig_size || buf_contents_changed(buf))
6810 reason = "changed";
6811 else if (orig_mode != buf->b_orig_mode)
6812 reason = "mode";
6813 else
6814 reason = "time";
Bram Moolenaar071d4272004-06-13 20:20:40 +00006815
6816 /*
6817 * Only give the warning if there are no FileChangedShell
6818 * autocommands.
6819 * Avoid being called recursively by setting "busy".
6820 */
6821 busy = TRUE;
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +01006822#ifdef FEAT_EVAL
Bram Moolenaar19a09a12005-03-04 23:39:37 +00006823 set_vim_var_string(VV_FCS_REASON, (char_u *)reason, -1);
6824 set_vim_var_string(VV_FCS_CHOICE, (char_u *)"", -1);
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +01006825#endif
Bram Moolenaarbf1b7a72009-03-05 02:15:53 +00006826 ++allbuf_lock;
Bram Moolenaar071d4272004-06-13 20:20:40 +00006827 n = apply_autocmds(EVENT_FILECHANGEDSHELL,
6828 buf->b_fname, buf->b_fname, FALSE, buf);
Bram Moolenaarbf1b7a72009-03-05 02:15:53 +00006829 --allbuf_lock;
Bram Moolenaar071d4272004-06-13 20:20:40 +00006830 busy = FALSE;
6831 if (n)
6832 {
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02006833 if (!bufref_valid(&bufref))
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006834 emsg(_("E246: FileChangedShell autocommand deleted buffer"));
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +01006835#ifdef FEAT_EVAL
Bram Moolenaar19a09a12005-03-04 23:39:37 +00006836 s = get_vim_var_str(VV_FCS_CHOICE);
6837 if (STRCMP(s, "reload") == 0 && *reason != 'd')
6838 reload = TRUE;
6839 else if (STRCMP(s, "ask") == 0)
6840 n = FALSE;
6841 else
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +01006842#endif
Bram Moolenaar19a09a12005-03-04 23:39:37 +00006843 return 2;
Bram Moolenaar071d4272004-06-13 20:20:40 +00006844 }
Bram Moolenaar19a09a12005-03-04 23:39:37 +00006845 if (!n)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006846 {
Bram Moolenaar19a09a12005-03-04 23:39:37 +00006847 if (*reason == 'd')
6848 mesg = _("E211: File \"%s\" no longer available");
Bram Moolenaar071d4272004-06-13 20:20:40 +00006849 else
6850 {
6851 helpmesg = TRUE;
6852#if defined(FEAT_CON_DIALOG) || defined(FEAT_GUI_DIALOG)
6853 can_reload = TRUE;
6854#endif
Bram Moolenaar19a09a12005-03-04 23:39:37 +00006855 if (reason[2] == 'n')
6856 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00006857 mesg = _("W12: Warning: File \"%s\" has changed and the buffer was changed in Vim as well");
Bram Moolenaar19a09a12005-03-04 23:39:37 +00006858 mesg2 = _("See \":help W12\" for more info.");
6859 }
6860 else if (reason[1] == 'h')
6861 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00006862 mesg = _("W11: Warning: File \"%s\" has changed since editing started");
Bram Moolenaar19a09a12005-03-04 23:39:37 +00006863 mesg2 = _("See \":help W11\" for more info.");
6864 }
6865 else if (*reason == 'm')
6866 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00006867 mesg = _("W16: Warning: Mode of file \"%s\" has changed since editing started");
Bram Moolenaar19a09a12005-03-04 23:39:37 +00006868 mesg2 = _("See \":help W16\" for more info.");
6869 }
Bram Moolenaar85388b52009-06-24 09:58:32 +00006870 else
6871 /* Only timestamp changed, store it to avoid a warning
6872 * in check_mtime() later. */
6873 buf->b_mtime_read = buf->b_mtime;
Bram Moolenaar071d4272004-06-13 20:20:40 +00006874 }
6875 }
6876 }
6877
6878 }
6879 else if ((buf->b_flags & BF_NEW) && !(buf->b_flags & BF_NEW_W)
6880 && vim_fexists(buf->b_ffname))
6881 {
6882 retval = 1;
6883 mesg = _("W13: Warning: File \"%s\" has been created after editing started");
6884 buf->b_flags |= BF_NEW_W;
6885#if defined(FEAT_CON_DIALOG) || defined(FEAT_GUI_DIALOG)
6886 can_reload = TRUE;
6887#endif
6888 }
6889
6890 if (mesg != NULL)
6891 {
6892 path = home_replace_save(buf, buf->b_fname);
6893 if (path != NULL)
6894 {
Bram Moolenaar19a09a12005-03-04 23:39:37 +00006895 if (!helpmesg)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006896 mesg2 = "";
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006897 tbuf = (char *)alloc((unsigned)(STRLEN(path) + STRLEN(mesg)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006898 + STRLEN(mesg2) + 2));
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006899 sprintf(tbuf, mesg, path);
Bram Moolenaar496c5262009-03-18 14:42:00 +00006900#ifdef FEAT_EVAL
6901 /* Set warningmsg here, before the unimportant and output-specific
6902 * mesg2 has been appended. */
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006903 set_vim_var_string(VV_WARNINGMSG, (char_u *)tbuf, -1);
Bram Moolenaar496c5262009-03-18 14:42:00 +00006904#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00006905#if defined(FEAT_CON_DIALOG) || defined(FEAT_GUI_DIALOG)
6906 if (can_reload)
6907 {
6908 if (*mesg2 != NUL)
6909 {
6910 STRCAT(tbuf, "\n");
6911 STRCAT(tbuf, mesg2);
6912 }
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006913 if (do_dialog(VIM_WARNING, (char_u *)_("Warning"),
6914 (char_u *)tbuf,
Bram Moolenaard2c340a2011-01-17 20:08:11 +01006915 (char_u *)_("&OK\n&Load File"), 1, NULL, TRUE) == 2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006916 reload = TRUE;
6917 }
6918 else
6919#endif
6920 if (State > NORMAL_BUSY || (State & CMDLINE) || already_warned)
6921 {
6922 if (*mesg2 != NUL)
6923 {
6924 STRCAT(tbuf, "; ");
6925 STRCAT(tbuf, mesg2);
6926 }
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006927 emsg(tbuf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00006928 retval = 2;
6929 }
6930 else
6931 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00006932 if (!autocmd_busy)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006933 {
6934 msg_start();
Bram Moolenaar32526b32019-01-19 17:43:09 +01006935 msg_puts_attr(tbuf, HL_ATTR(HLF_E) + MSG_HIST);
Bram Moolenaar071d4272004-06-13 20:20:40 +00006936 if (*mesg2 != NUL)
Bram Moolenaar32526b32019-01-19 17:43:09 +01006937 msg_puts_attr(mesg2, HL_ATTR(HLF_W) + MSG_HIST);
Bram Moolenaar071d4272004-06-13 20:20:40 +00006938 msg_clr_eos();
6939 (void)msg_end();
6940 if (emsg_silent == 0)
6941 {
6942 out_flush();
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +01006943#ifdef FEAT_GUI
Bram Moolenaar071d4272004-06-13 20:20:40 +00006944 if (!focus)
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +01006945#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00006946 /* give the user some time to think about it */
6947 ui_delay(1000L, TRUE);
6948
6949 /* don't redraw and erase the message */
6950 redraw_cmdline = FALSE;
6951 }
6952 }
6953 already_warned = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00006954 }
6955
6956 vim_free(path);
6957 vim_free(tbuf);
6958 }
6959 }
6960
6961 if (reload)
Bram Moolenaar465748e2012-08-29 18:50:54 +02006962 {
Bram Moolenaar631d6f62005-06-07 21:02:10 +00006963 /* Reload the buffer. */
Bram Moolenaar316059c2006-01-14 21:18:42 +00006964 buf_reload(buf, orig_mode);
Bram Moolenaar465748e2012-08-29 18:50:54 +02006965#ifdef FEAT_PERSISTENT_UNDO
6966 if (buf->b_p_udf && buf->b_ffname != NULL)
6967 {
6968 char_u hash[UNDO_HASH_SIZE];
6969 buf_T *save_curbuf = curbuf;
6970
6971 /* Any existing undo file is unusable, write it now. */
6972 curbuf = buf;
6973 u_compute_hash(hash);
6974 u_write_undo(NULL, FALSE, buf, hash);
6975 curbuf = save_curbuf;
6976 }
6977#endif
6978 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00006979
Bram Moolenaar3577c6f2008-06-24 21:16:56 +00006980 /* Trigger FileChangedShell when the file was changed in any way. */
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02006981 if (bufref_valid(&bufref) && retval != 0)
Bram Moolenaar56718732006-03-15 22:53:57 +00006982 (void)apply_autocmds(EVENT_FILECHANGEDSHELLPOST,
6983 buf->b_fname, buf->b_fname, FALSE, buf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00006984#ifdef FEAT_GUI
6985 /* restore this in case an autocommand has set it; it would break
6986 * 'mousefocus' */
6987 need_mouse_correct = save_mouse_correct;
6988#endif
6989
6990 return retval;
6991}
6992
Bram Moolenaar631d6f62005-06-07 21:02:10 +00006993/*
6994 * Reload a buffer that is already loaded.
6995 * Used when the file was changed outside of Vim.
Bram Moolenaar316059c2006-01-14 21:18:42 +00006996 * "orig_mode" is buf->b_orig_mode before the need for reloading was detected.
6997 * buf->b_orig_mode may have been reset already.
Bram Moolenaar631d6f62005-06-07 21:02:10 +00006998 */
6999 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01007000buf_reload(buf_T *buf, int orig_mode)
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007001{
7002 exarg_T ea;
7003 pos_T old_cursor;
7004 linenr_T old_topline;
7005 int old_ro = buf->b_p_ro;
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007006 buf_T *savebuf;
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02007007 bufref_T bufref;
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007008 int saved = OK;
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007009 aco_save_T aco;
Bram Moolenaar59f931e2010-07-24 20:27:03 +02007010 int flags = READ_NEW;
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007011
7012 /* set curwin/curbuf for "buf" and save some things */
7013 aucmd_prepbuf(&aco, buf);
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007014
7015 /* We only want to read the text from the file, not reset the syntax
7016 * highlighting, clear marks, diff status, etc. Force the fileformat
7017 * and encoding to be the same. */
7018 if (prep_exarg(&ea, buf) == OK)
7019 {
7020 old_cursor = curwin->w_cursor;
7021 old_topline = curwin->w_topline;
7022
Bram Moolenaarf9bb7342010-08-04 14:29:54 +02007023 if (p_ur < 0 || curbuf->b_ml.ml_line_count <= p_ur)
Bram Moolenaar59f931e2010-07-24 20:27:03 +02007024 {
7025 /* Save all the text, so that the reload can be undone.
7026 * Sync first so that this is a separate undo-able action. */
7027 u_sync(FALSE);
7028 saved = u_savecommon(0, curbuf->b_ml.ml_line_count + 1, 0, TRUE);
7029 flags |= READ_KEEP_UNDO;
7030 }
7031
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007032 /*
7033 * To behave like when a new file is edited (matters for
7034 * BufReadPost autocommands) we first need to delete the current
7035 * buffer contents. But if reading the file fails we should keep
7036 * the old contents. Can't use memory only, the file might be
7037 * too big. Use a hidden buffer to move the buffer contents to.
7038 */
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01007039 if (BUFEMPTY() || saved == FAIL)
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007040 savebuf = NULL;
7041 else
7042 {
7043 /* Allocate a buffer without putting it in the buffer list. */
7044 savebuf = buflist_new(NULL, NULL, (linenr_T)1, BLN_DUMMY);
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02007045 set_bufref(&bufref, savebuf);
Bram Moolenaar8424a622006-04-19 21:23:36 +00007046 if (savebuf != NULL && buf == curbuf)
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007047 {
7048 /* Open the memline. */
7049 curbuf = savebuf;
7050 curwin->w_buffer = savebuf;
Bram Moolenaar4770d092006-01-12 23:22:24 +00007051 saved = ml_open(curbuf);
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007052 curbuf = buf;
7053 curwin->w_buffer = buf;
7054 }
Bram Moolenaar8424a622006-04-19 21:23:36 +00007055 if (savebuf == NULL || saved == FAIL || buf != curbuf
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007056 || move_lines(buf, savebuf) == FAIL)
7057 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007058 semsg(_("E462: Could not prepare for reloading \"%s\""),
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007059 buf->b_fname);
7060 saved = FAIL;
7061 }
7062 }
7063
7064 if (saved == OK)
7065 {
7066 curbuf->b_flags |= BF_CHECK_RO; /* check for RO again */
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007067 keep_filetype = TRUE; /* don't detect 'filetype' */
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007068 if (readfile(buf->b_ffname, buf->b_fname, (linenr_T)0,
7069 (linenr_T)0,
Bram Moolenaare13b9af2017-01-13 22:01:02 +01007070 (linenr_T)MAXLNUM, &ea, flags) != OK)
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007071 {
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +01007072#if defined(FEAT_EVAL)
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007073 if (!aborting())
7074#endif
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007075 semsg(_("E321: Could not reload \"%s\""), buf->b_fname);
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02007076 if (savebuf != NULL && bufref_valid(&bufref) && buf == curbuf)
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007077 {
7078 /* Put the text back from the save buffer. First
7079 * delete any lines that readfile() added. */
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01007080 while (!BUFEMPTY())
Bram Moolenaar8424a622006-04-19 21:23:36 +00007081 if (ml_delete(buf->b_ml.ml_line_count, FALSE) == FAIL)
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007082 break;
7083 (void)move_lines(savebuf, buf);
7084 }
7085 }
Bram Moolenaar59f931e2010-07-24 20:27:03 +02007086 else if (buf == curbuf) /* "buf" still valid */
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007087 {
7088 /* Mark the buffer as unmodified and free undo info. */
7089 unchanged(buf, TRUE);
Bram Moolenaar59f931e2010-07-24 20:27:03 +02007090 if ((flags & READ_KEEP_UNDO) == 0)
7091 {
7092 u_blockfree(buf);
7093 u_clearall(buf);
7094 }
7095 else
Bram Moolenaarf9bb7342010-08-04 14:29:54 +02007096 {
Bram Moolenaar59f931e2010-07-24 20:27:03 +02007097 /* Mark all undo states as changed. */
7098 u_unchanged(curbuf);
Bram Moolenaarf9bb7342010-08-04 14:29:54 +02007099 }
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007100 }
7101 }
7102 vim_free(ea.cmd);
7103
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02007104 if (savebuf != NULL && bufref_valid(&bufref))
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007105 wipe_buffer(savebuf, FALSE);
7106
7107#ifdef FEAT_DIFF
7108 /* Invalidate diff info if necessary. */
Bram Moolenaar8424a622006-04-19 21:23:36 +00007109 diff_invalidate(curbuf);
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007110#endif
7111
7112 /* Restore the topline and cursor position and check it (lines may
7113 * have been removed). */
7114 if (old_topline > curbuf->b_ml.ml_line_count)
7115 curwin->w_topline = curbuf->b_ml.ml_line_count;
7116 else
7117 curwin->w_topline = old_topline;
7118 curwin->w_cursor = old_cursor;
7119 check_cursor();
7120 update_topline();
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007121 keep_filetype = FALSE;
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007122#ifdef FEAT_FOLDING
7123 {
Bram Moolenaarbd1e5d22009-04-29 09:02:44 +00007124 win_T *wp;
7125 tabpage_T *tp;
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007126
7127 /* Update folds unless they are defined manually. */
Bram Moolenaarbd1e5d22009-04-29 09:02:44 +00007128 FOR_ALL_TAB_WINDOWS(tp, wp)
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007129 if (wp->w_buffer == curwin->w_buffer
7130 && !foldmethodIsManual(wp))
7131 foldUpdateAll(wp);
7132 }
7133#endif
7134 /* If the mode didn't change and 'readonly' was set, keep the old
7135 * value; the user probably used the ":view" command. But don't
7136 * reset it, might have had a read error. */
7137 if (orig_mode == curbuf->b_orig_mode)
7138 curbuf->b_p_ro |= old_ro;
Bram Moolenaar52f85b72013-01-30 14:13:56 +01007139
7140 /* Modelines must override settings done by autocommands. */
7141 do_modelines(0);
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007142 }
7143
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007144 /* restore curwin/curbuf and a few other things */
7145 aucmd_restbuf(&aco);
7146 /* Careful: autocommands may have made "buf" invalid! */
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007147}
7148
Bram Moolenaar071d4272004-06-13 20:20:40 +00007149 void
Bram Moolenaar8767f522016-07-01 17:17:39 +02007150buf_store_time(buf_T *buf, stat_T *st, char_u *fname UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00007151{
7152 buf->b_mtime = (long)st->st_mtime;
Bram Moolenaar914703b2010-05-31 21:59:46 +02007153 buf->b_orig_size = st->st_size;
Bram Moolenaar071d4272004-06-13 20:20:40 +00007154#ifdef HAVE_ST_MODE
7155 buf->b_orig_mode = (int)st->st_mode;
7156#else
7157 buf->b_orig_mode = mch_getperm(fname);
7158#endif
7159}
7160
7161/*
7162 * Adjust the line with missing eol, used for the next write.
7163 * Used for do_filter(), when the input lines for the filter are deleted.
7164 */
7165 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01007166write_lnum_adjust(linenr_T offset)
Bram Moolenaar071d4272004-06-13 20:20:40 +00007167{
Bram Moolenaarcab35ad2011-02-15 17:39:22 +01007168 if (curbuf->b_no_eol_lnum != 0) /* only if there is a missing eol */
7169 curbuf->b_no_eol_lnum += offset;
Bram Moolenaar071d4272004-06-13 20:20:40 +00007170}
7171
Bram Moolenaarda440d22016-01-16 21:27:23 +01007172#if defined(TEMPDIRNAMES) || defined(FEAT_EVAL) || defined(PROTO)
7173/*
7174 * Delete "name" and everything in it, recursively.
7175 * return 0 for succes, -1 if some file was not deleted.
7176 */
7177 int
7178delete_recursive(char_u *name)
7179{
7180 int result = 0;
7181 char_u **files;
7182 int file_count;
7183 int i;
7184 char_u *exp;
7185
Bram Moolenaar43a34f92016-01-17 15:56:34 +01007186 /* A symbolic link to a directory itself is deleted, not the directory it
7187 * points to. */
7188 if (
Bram Moolenaar4f974752019-02-17 17:44:42 +01007189# if defined(UNIX) || defined(MSWIN)
Bram Moolenaar43a34f92016-01-17 15:56:34 +01007190 mch_isrealdir(name)
Bram Moolenaar203258c2016-01-17 22:15:16 +01007191# else
Bram Moolenaar43a34f92016-01-17 15:56:34 +01007192 mch_isdir(name)
Bram Moolenaar43a34f92016-01-17 15:56:34 +01007193# endif
7194 )
Bram Moolenaarda440d22016-01-16 21:27:23 +01007195 {
7196 vim_snprintf((char *)NameBuff, MAXPATHL, "%s/*", name);
7197 exp = vim_strsave(NameBuff);
7198 if (exp == NULL)
7199 return -1;
7200 if (gen_expand_wildcards(1, &exp, &file_count, &files,
Bram Moolenaar336bd622016-01-17 18:23:58 +01007201 EW_DIR|EW_FILE|EW_SILENT|EW_ALLLINKS|EW_DODOT|EW_EMPTYOK) == OK)
Bram Moolenaarda440d22016-01-16 21:27:23 +01007202 {
7203 for (i = 0; i < file_count; ++i)
7204 if (delete_recursive(files[i]) != 0)
7205 result = -1;
7206 FreeWild(file_count, files);
7207 }
7208 else
7209 result = -1;
7210 vim_free(exp);
7211 (void)mch_rmdir(name);
7212 }
7213 else
7214 result = mch_remove(name) == 0 ? 0 : -1;
7215
7216 return result;
7217}
7218#endif
7219
Bram Moolenaar071d4272004-06-13 20:20:40 +00007220#if defined(TEMPDIRNAMES) || defined(PROTO)
7221static long temp_count = 0; /* Temp filename counter. */
7222
7223/*
7224 * Delete the temp directory and all files it contains.
7225 */
7226 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01007227vim_deltempdir(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00007228{
Bram Moolenaar071d4272004-06-13 20:20:40 +00007229 if (vim_tempdir != NULL)
7230 {
Bram Moolenaarda440d22016-01-16 21:27:23 +01007231 /* remove the trailing path separator */
7232 gettail(vim_tempdir)[-1] = NUL;
7233 delete_recursive(vim_tempdir);
Bram Moolenaard23a8232018-02-10 18:45:26 +01007234 VIM_CLEAR(vim_tempdir);
Bram Moolenaar071d4272004-06-13 20:20:40 +00007235 }
7236}
Bram Moolenaar071d4272004-06-13 20:20:40 +00007237
7238/*
Bram Moolenaareaf03392009-11-17 11:08:52 +00007239 * Directory "tempdir" was created. Expand this name to a full path and put
7240 * it in "vim_tempdir". This avoids that using ":cd" would confuse us.
7241 * "tempdir" must be no longer than MAXPATHL.
7242 */
7243 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01007244vim_settempdir(char_u *tempdir)
Bram Moolenaareaf03392009-11-17 11:08:52 +00007245{
7246 char_u *buf;
7247
7248 buf = alloc((unsigned)MAXPATHL + 2);
7249 if (buf != NULL)
7250 {
7251 if (vim_FullName(tempdir, buf, MAXPATHL, FALSE) == FAIL)
7252 STRCPY(buf, tempdir);
Bram Moolenaara06ecab2016-07-16 14:47:36 +02007253 add_pathsep(buf);
Bram Moolenaareaf03392009-11-17 11:08:52 +00007254 vim_tempdir = vim_strsave(buf);
7255 vim_free(buf);
7256 }
7257}
Bram Moolenaar4592dee2009-11-18 19:11:58 +00007258#endif
Bram Moolenaareaf03392009-11-17 11:08:52 +00007259
7260/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00007261 * vim_tempname(): Return a unique name that can be used for a temp file.
7262 *
Bram Moolenaar76ae22f2016-06-13 20:00:29 +02007263 * The temp file is NOT guaranteed to be created. If "keep" is FALSE it is
7264 * guaranteed to NOT be created.
Bram Moolenaar071d4272004-06-13 20:20:40 +00007265 *
7266 * The returned pointer is to allocated memory.
7267 * The returned pointer is NULL if no valid name was found.
7268 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00007269 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01007270vim_tempname(
7271 int extra_char UNUSED, /* char to use in the name instead of '?' */
7272 int keep UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00007273{
7274#ifdef USE_TMPNAM
7275 char_u itmp[L_tmpnam]; /* use tmpnam() */
Bram Moolenaar4f974752019-02-17 17:44:42 +01007276#elif defined(MSWIN)
Bram Moolenaarec0f50a2019-02-10 23:26:13 +01007277 WCHAR itmp[TEMPNAMELEN];
Bram Moolenaar071d4272004-06-13 20:20:40 +00007278#else
7279 char_u itmp[TEMPNAMELEN];
7280#endif
7281
7282#ifdef TEMPDIRNAMES
7283 static char *(tempdirs[]) = {TEMPDIRNAMES};
7284 int i;
Bram Moolenaar071d4272004-06-13 20:20:40 +00007285# ifndef EEXIST
Bram Moolenaar8767f522016-07-01 17:17:39 +02007286 stat_T st;
Bram Moolenaar071d4272004-06-13 20:20:40 +00007287# endif
7288
7289 /*
7290 * This will create a directory for private use by this instance of Vim.
7291 * This is done once, and the same directory is used for all temp files.
7292 * This method avoids security problems because of symlink attacks et al.
7293 * It's also a bit faster, because we only need to check for an existing
7294 * file when creating the directory and not for each temp file.
7295 */
7296 if (vim_tempdir == NULL)
7297 {
7298 /*
7299 * Try the entries in TEMPDIRNAMES to create the temp directory.
7300 */
Bram Moolenaar78a15312009-05-15 19:33:18 +00007301 for (i = 0; i < (int)(sizeof(tempdirs) / sizeof(char *)); ++i)
Bram Moolenaar071d4272004-06-13 20:20:40 +00007302 {
Bram Moolenaareaf03392009-11-17 11:08:52 +00007303# ifndef HAVE_MKDTEMP
Bram Moolenaar2660c0e2010-01-19 14:59:56 +01007304 size_t itmplen;
Bram Moolenaareaf03392009-11-17 11:08:52 +00007305 long nr;
7306 long off;
7307# endif
7308
Bram Moolenaare1a61992015-12-03 21:02:27 +01007309 /* Expand $TMP, leave room for "/v1100000/999999999".
7310 * Skip the directory check if the expansion fails. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00007311 expand_env((char_u *)tempdirs[i], itmp, TEMPNAMELEN - 20);
Bram Moolenaare1a61992015-12-03 21:02:27 +01007312 if (itmp[0] != '$' && mch_isdir(itmp))
Bram Moolenaar071d4272004-06-13 20:20:40 +00007313 {
Bram Moolenaare1a61992015-12-03 21:02:27 +01007314 /* directory exists */
Bram Moolenaara06ecab2016-07-16 14:47:36 +02007315 add_pathsep(itmp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00007316
Bram Moolenaareaf03392009-11-17 11:08:52 +00007317# ifdef HAVE_MKDTEMP
Bram Moolenaar35d88f42016-06-04 14:52:00 +02007318 {
7319# if defined(UNIX) || defined(VMS)
7320 /* Make sure the umask doesn't remove the executable bit.
7321 * "repl" has been reported to use "177". */
7322 mode_t umask_save = umask(077);
7323# endif
7324 /* Leave room for filename */
7325 STRCAT(itmp, "vXXXXXX");
7326 if (mkdtemp((char *)itmp) != NULL)
7327 vim_settempdir(itmp);
7328# if defined(UNIX) || defined(VMS)
7329 (void)umask(umask_save);
7330# endif
7331 }
Bram Moolenaareaf03392009-11-17 11:08:52 +00007332# else
Bram Moolenaar071d4272004-06-13 20:20:40 +00007333 /* Get an arbitrary number of up to 6 digits. When it's
7334 * unlikely that it already exists it will be faster,
7335 * otherwise it doesn't matter. The use of mkdir() avoids any
7336 * security problems because of the predictable number. */
7337 nr = (mch_get_pid() + (long)time(NULL)) % 1000000L;
Bram Moolenaar2660c0e2010-01-19 14:59:56 +01007338 itmplen = STRLEN(itmp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00007339
7340 /* Try up to 10000 different values until we find a name that
7341 * doesn't exist. */
7342 for (off = 0; off < 10000L; ++off)
7343 {
7344 int r;
Bram Moolenaareaf03392009-11-17 11:08:52 +00007345# if defined(UNIX) || defined(VMS)
Bram Moolenaar071d4272004-06-13 20:20:40 +00007346 mode_t umask_save;
Bram Moolenaareaf03392009-11-17 11:08:52 +00007347# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00007348
Bram Moolenaareaf03392009-11-17 11:08:52 +00007349 sprintf((char *)itmp + itmplen, "v%ld", nr + off);
7350# ifndef EEXIST
Bram Moolenaar071d4272004-06-13 20:20:40 +00007351 /* If mkdir() does not set errno to EEXIST, check for
7352 * existing file here. There is a race condition then,
7353 * although it's fail-safe. */
7354 if (mch_stat((char *)itmp, &st) >= 0)
7355 continue;
Bram Moolenaareaf03392009-11-17 11:08:52 +00007356# endif
7357# if defined(UNIX) || defined(VMS)
Bram Moolenaar071d4272004-06-13 20:20:40 +00007358 /* Make sure the umask doesn't remove the executable bit.
7359 * "repl" has been reported to use "177". */
7360 umask_save = umask(077);
Bram Moolenaareaf03392009-11-17 11:08:52 +00007361# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00007362 r = vim_mkdir(itmp, 0700);
Bram Moolenaareaf03392009-11-17 11:08:52 +00007363# if defined(UNIX) || defined(VMS)
Bram Moolenaar071d4272004-06-13 20:20:40 +00007364 (void)umask(umask_save);
Bram Moolenaareaf03392009-11-17 11:08:52 +00007365# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00007366 if (r == 0)
7367 {
Bram Moolenaareaf03392009-11-17 11:08:52 +00007368 vim_settempdir(itmp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00007369 break;
7370 }
Bram Moolenaareaf03392009-11-17 11:08:52 +00007371# ifdef EEXIST
Bram Moolenaar071d4272004-06-13 20:20:40 +00007372 /* If the mkdir() didn't fail because the file/dir exists,
7373 * we probably can't create any dir here, try another
7374 * place. */
7375 if (errno != EEXIST)
Bram Moolenaareaf03392009-11-17 11:08:52 +00007376# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00007377 break;
7378 }
Bram Moolenaareaf03392009-11-17 11:08:52 +00007379# endif /* HAVE_MKDTEMP */
Bram Moolenaar071d4272004-06-13 20:20:40 +00007380 if (vim_tempdir != NULL)
7381 break;
7382 }
7383 }
7384 }
7385
7386 if (vim_tempdir != NULL)
7387 {
7388 /* There is no need to check if the file exists, because we own the
7389 * directory and nobody else creates a file in it. */
7390 sprintf((char *)itmp, "%s%ld", vim_tempdir, temp_count++);
7391 return vim_strsave(itmp);
7392 }
7393
7394 return NULL;
7395
7396#else /* TEMPDIRNAMES */
7397
Bram Moolenaar4f974752019-02-17 17:44:42 +01007398# ifdef MSWIN
Bram Moolenaarec0f50a2019-02-10 23:26:13 +01007399 WCHAR wszTempFile[_MAX_PATH + 1];
7400 WCHAR buf4[4];
Bram Moolenaar071d4272004-06-13 20:20:40 +00007401 char_u *retval;
7402 char_u *p;
7403
Bram Moolenaarec0f50a2019-02-10 23:26:13 +01007404 wcscpy(itmp, L"");
7405 if (GetTempPathW(_MAX_PATH, wszTempFile) == 0)
Bram Moolenaarb1891912011-02-09 14:47:03 +01007406 {
Bram Moolenaarec0f50a2019-02-10 23:26:13 +01007407 wszTempFile[0] = L'.'; // GetTempPathW() failed, use current dir
7408 wszTempFile[1] = NUL;
Bram Moolenaarb1891912011-02-09 14:47:03 +01007409 }
Bram Moolenaarec0f50a2019-02-10 23:26:13 +01007410 wcscpy(buf4, L"VIM");
Bram Moolenaar071d4272004-06-13 20:20:40 +00007411 buf4[2] = extra_char; /* make it "VIa", "VIb", etc. */
Bram Moolenaarec0f50a2019-02-10 23:26:13 +01007412 if (GetTempFileNameW(wszTempFile, buf4, 0, itmp) == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00007413 return NULL;
Bram Moolenaare5c421c2015-03-31 13:33:08 +02007414 if (!keep)
Bram Moolenaarec0f50a2019-02-10 23:26:13 +01007415 // GetTempFileName() will create the file, we don't want that
7416 (void)DeleteFileW(itmp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00007417
Bram Moolenaarec0f50a2019-02-10 23:26:13 +01007418 // Backslashes in a temp file name cause problems when filtering with
7419 // "sh". NOTE: This also checks 'shellcmdflag' to help those people who
7420 // didn't set 'shellslash'.
7421 retval = utf16_to_enc(itmp, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00007422 if (*p_shcf == '-' || p_ssl)
7423 for (p = retval; *p; ++p)
7424 if (*p == '\\')
7425 *p = '/';
7426 return retval;
7427
Bram Moolenaar4f974752019-02-17 17:44:42 +01007428# else // MSWIN
Bram Moolenaar071d4272004-06-13 20:20:40 +00007429
7430# ifdef USE_TMPNAM
Bram Moolenaar95474ca2011-02-09 16:44:51 +01007431 char_u *p;
7432
Bram Moolenaar071d4272004-06-13 20:20:40 +00007433 /* tmpnam() will make its own name */
Bram Moolenaar95474ca2011-02-09 16:44:51 +01007434 p = tmpnam((char *)itmp);
7435 if (p == NULL || *p == NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00007436 return NULL;
7437# else
7438 char_u *p;
7439
7440# ifdef VMS_TEMPNAM
7441 /* mktemp() is not working on VMS. It seems to be
7442 * a do-nothing function. Therefore we use tempnam().
7443 */
7444 sprintf((char *)itmp, "VIM%c", extra_char);
7445 p = (char_u *)tempnam("tmp:", (char *)itmp);
7446 if (p != NULL)
7447 {
Bram Moolenaar206f0112014-03-12 16:51:55 +01007448 /* VMS will use '.LIS' if we don't explicitly specify an extension,
Bram Moolenaar071d4272004-06-13 20:20:40 +00007449 * and VIM will then be unable to find the file later */
7450 STRCPY(itmp, p);
7451 STRCAT(itmp, ".txt");
7452 free(p);
7453 }
7454 else
7455 return NULL;
7456# else
7457 STRCPY(itmp, TEMPNAME);
7458 if ((p = vim_strchr(itmp, '?')) != NULL)
7459 *p = extra_char;
7460 if (mktemp((char *)itmp) == NULL)
7461 return NULL;
7462# endif
7463# endif
7464
7465 return vim_strsave(itmp);
Bram Moolenaar4f974752019-02-17 17:44:42 +01007466# endif // MSWIN
Bram Moolenaar071d4272004-06-13 20:20:40 +00007467#endif /* TEMPDIRNAMES */
7468}
7469
7470#if defined(BACKSLASH_IN_FILENAME) || defined(PROTO)
7471/*
Bram Moolenaarb4f6a462015-10-13 19:43:17 +02007472 * Convert all backslashes in fname to forward slashes in-place, unless when
7473 * it looks like a URL.
Bram Moolenaar071d4272004-06-13 20:20:40 +00007474 */
7475 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01007476forward_slash(char_u *fname)
Bram Moolenaar071d4272004-06-13 20:20:40 +00007477{
7478 char_u *p;
7479
Bram Moolenaarb4f6a462015-10-13 19:43:17 +02007480 if (path_with_url(fname))
7481 return;
Bram Moolenaar071d4272004-06-13 20:20:40 +00007482 for (p = fname; *p != NUL; ++p)
Bram Moolenaar071d4272004-06-13 20:20:40 +00007483 /* The Big5 encoding can have '\' in the trail byte. */
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00007484 if (enc_dbcs != 0 && (*mb_ptr2len)(p) > 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00007485 ++p;
Bram Moolenaar13505972019-01-24 15:04:48 +01007486 else if (*p == '\\')
Bram Moolenaar071d4272004-06-13 20:20:40 +00007487 *p = '/';
7488}
7489#endif
7490
Bram Moolenaarf30e74c2006-08-16 17:35:00 +00007491/*
Bram Moolenaar748bf032005-02-02 23:04:36 +00007492 * Try matching a filename with a "pattern" ("prog" is NULL), or use the
7493 * precompiled regprog "prog" ("pattern" is NULL). That avoids calling
7494 * vim_regcomp() often.
Bram Moolenaar071d4272004-06-13 20:20:40 +00007495 * Used for autocommands and 'wildignore'.
7496 * Returns TRUE if there is a match, FALSE otherwise.
7497 */
Bram Moolenaar3e460fd2019-01-26 16:21:07 +01007498 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01007499match_file_pat(
7500 char_u *pattern, /* pattern to match with */
7501 regprog_T **prog, /* pre-compiled regprog or NULL */
7502 char_u *fname, /* full path of file name */
7503 char_u *sfname, /* short file name or NULL */
7504 char_u *tail, /* tail of path */
7505 int allow_dirs) /* allow matching with dir */
Bram Moolenaar071d4272004-06-13 20:20:40 +00007506{
7507 regmatch_T regmatch;
7508 int result = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00007509
Bram Moolenaar71afbfe2013-03-19 16:49:16 +01007510 regmatch.rm_ic = p_fic; /* ignore case if 'fileignorecase' is set */
Bram Moolenaar49a6ed82015-01-07 14:43:39 +01007511 if (prog != NULL)
7512 regmatch.regprog = *prog;
Bram Moolenaar071d4272004-06-13 20:20:40 +00007513 else
Bram Moolenaar49a6ed82015-01-07 14:43:39 +01007514 regmatch.regprog = vim_regcomp(pattern, RE_MAGIC);
Bram Moolenaar071d4272004-06-13 20:20:40 +00007515
7516 /*
7517 * Try for a match with the pattern with:
7518 * 1. the full file name, when the pattern has a '/'.
7519 * 2. the short file name, when the pattern has a '/'.
7520 * 3. the tail of the file name, when the pattern has no '/'.
7521 */
Bram Moolenaar49a6ed82015-01-07 14:43:39 +01007522 if (regmatch.regprog != NULL
Bram Moolenaar071d4272004-06-13 20:20:40 +00007523 && ((allow_dirs
7524 && (vim_regexec(&regmatch, fname, (colnr_T)0)
7525 || (sfname != NULL
7526 && vim_regexec(&regmatch, sfname, (colnr_T)0))))
Bram Moolenaar49a6ed82015-01-07 14:43:39 +01007527 || (!allow_dirs && vim_regexec(&regmatch, tail, (colnr_T)0))))
Bram Moolenaar071d4272004-06-13 20:20:40 +00007528 result = TRUE;
7529
Bram Moolenaardffa5b82014-11-19 16:38:07 +01007530 if (prog != NULL)
7531 *prog = regmatch.regprog;
7532 else
Bram Moolenaar473de612013-06-08 18:19:48 +02007533 vim_regfree(regmatch.regprog);
Bram Moolenaar071d4272004-06-13 20:20:40 +00007534 return result;
7535}
Bram Moolenaar071d4272004-06-13 20:20:40 +00007536
7537#if defined(FEAT_WILDIGN) || defined(PROTO)
7538/*
7539 * Return TRUE if a file matches with a pattern in "list".
7540 * "list" is a comma-separated list of patterns, like 'wildignore'.
7541 * "sfname" is the short file name or NULL, "ffname" the long file name.
7542 */
7543 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01007544match_file_list(char_u *list, char_u *sfname, char_u *ffname)
Bram Moolenaar071d4272004-06-13 20:20:40 +00007545{
7546 char_u buf[100];
7547 char_u *tail;
7548 char_u *regpat;
7549 char allow_dirs;
7550 int match;
7551 char_u *p;
7552
7553 tail = gettail(sfname);
7554
7555 /* try all patterns in 'wildignore' */
7556 p = list;
7557 while (*p)
7558 {
7559 copy_option_part(&p, buf, 100, ",");
7560 regpat = file_pat_to_reg_pat(buf, NULL, &allow_dirs, FALSE);
7561 if (regpat == NULL)
7562 break;
Bram Moolenaar748bf032005-02-02 23:04:36 +00007563 match = match_file_pat(regpat, NULL, ffname, sfname,
7564 tail, (int)allow_dirs);
Bram Moolenaar071d4272004-06-13 20:20:40 +00007565 vim_free(regpat);
7566 if (match)
7567 return TRUE;
7568 }
7569 return FALSE;
7570}
7571#endif
7572
7573/*
7574 * Convert the given pattern "pat" which has shell style wildcards in it, into
7575 * a regular expression, and return the result in allocated memory. If there
7576 * is a directory path separator to be matched, then TRUE is put in
7577 * allow_dirs, otherwise FALSE is put there -- webb.
7578 * Handle backslashes before special characters, like "\*" and "\ ".
7579 *
Bram Moolenaar071d4272004-06-13 20:20:40 +00007580 * Returns NULL when out of memory.
7581 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00007582 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01007583file_pat_to_reg_pat(
7584 char_u *pat,
7585 char_u *pat_end, /* first char after pattern or NULL */
7586 char *allow_dirs, /* Result passed back out in here */
7587 int no_bslash UNUSED) /* Don't use a backward slash as pathsep */
Bram Moolenaar071d4272004-06-13 20:20:40 +00007588{
Bram Moolenaar49a6ed82015-01-07 14:43:39 +01007589 int size = 2; /* '^' at start, '$' at end */
Bram Moolenaar071d4272004-06-13 20:20:40 +00007590 char_u *endp;
7591 char_u *reg_pat;
7592 char_u *p;
7593 int i;
7594 int nested = 0;
7595 int add_dollar = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00007596
7597 if (allow_dirs != NULL)
7598 *allow_dirs = FALSE;
7599 if (pat_end == NULL)
7600 pat_end = pat + STRLEN(pat);
7601
Bram Moolenaar071d4272004-06-13 20:20:40 +00007602 for (p = pat; p < pat_end; p++)
7603 {
7604 switch (*p)
7605 {
7606 case '*':
7607 case '.':
7608 case ',':
7609 case '{':
7610 case '}':
7611 case '~':
7612 size += 2; /* extra backslash */
7613 break;
7614#ifdef BACKSLASH_IN_FILENAME
7615 case '\\':
7616 case '/':
7617 size += 4; /* could become "[\/]" */
7618 break;
7619#endif
7620 default:
7621 size++;
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00007622 if (enc_dbcs != 0 && (*mb_ptr2len)(p) > 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00007623 {
7624 ++p;
7625 ++size;
7626 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00007627 break;
7628 }
7629 }
7630 reg_pat = alloc(size + 1);
7631 if (reg_pat == NULL)
7632 return NULL;
7633
Bram Moolenaar071d4272004-06-13 20:20:40 +00007634 i = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00007635
7636 if (pat[0] == '*')
7637 while (pat[0] == '*' && pat < pat_end - 1)
7638 pat++;
7639 else
7640 reg_pat[i++] = '^';
7641 endp = pat_end - 1;
Bram Moolenaar8fee8782015-08-11 18:45:48 +02007642 if (endp >= pat && *endp == '*')
Bram Moolenaar071d4272004-06-13 20:20:40 +00007643 {
7644 while (endp - pat > 0 && *endp == '*')
7645 endp--;
7646 add_dollar = FALSE;
7647 }
7648 for (p = pat; *p && nested >= 0 && p <= endp; p++)
7649 {
7650 switch (*p)
7651 {
7652 case '*':
7653 reg_pat[i++] = '.';
7654 reg_pat[i++] = '*';
Bram Moolenaar02743632005-07-25 20:42:36 +00007655 while (p[1] == '*') /* "**" matches like "*" */
7656 ++p;
Bram Moolenaar071d4272004-06-13 20:20:40 +00007657 break;
7658 case '.':
Bram Moolenaar071d4272004-06-13 20:20:40 +00007659 case '~':
7660 reg_pat[i++] = '\\';
7661 reg_pat[i++] = *p;
7662 break;
7663 case '?':
Bram Moolenaar071d4272004-06-13 20:20:40 +00007664 reg_pat[i++] = '.';
7665 break;
7666 case '\\':
7667 if (p[1] == NUL)
7668 break;
7669#ifdef BACKSLASH_IN_FILENAME
7670 if (!no_bslash)
7671 {
7672 /* translate:
7673 * "\x" to "\\x" e.g., "dir\file"
7674 * "\*" to "\\.*" e.g., "dir\*.c"
7675 * "\?" to "\\." e.g., "dir\??.c"
7676 * "\+" to "\+" e.g., "fileX\+.c"
7677 */
7678 if ((vim_isfilec(p[1]) || p[1] == '*' || p[1] == '?')
7679 && p[1] != '+')
7680 {
7681 reg_pat[i++] = '[';
7682 reg_pat[i++] = '\\';
7683 reg_pat[i++] = '/';
7684 reg_pat[i++] = ']';
7685 if (allow_dirs != NULL)
7686 *allow_dirs = TRUE;
7687 break;
7688 }
7689 }
7690#endif
Bram Moolenaar8cd213c2010-06-01 21:57:09 +02007691 /* Undo escaping from ExpandEscape():
7692 * foo\?bar -> foo?bar
7693 * foo\%bar -> foo%bar
7694 * foo\,bar -> foo,bar
7695 * foo\ bar -> foo bar
7696 * Don't unescape \, * and others that are also special in a
Bram Moolenaarf4e11432013-07-03 16:53:03 +02007697 * regexp.
7698 * An escaped { must be unescaped since we use magic not
Bram Moolenaara946afe2013-08-02 15:22:39 +02007699 * verymagic. Use "\\\{n,m\}"" to get "\{n,m}".
Bram Moolenaarf4e11432013-07-03 16:53:03 +02007700 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00007701 if (*++p == '?'
7702#ifdef BACKSLASH_IN_FILENAME
7703 && no_bslash
7704#endif
7705 )
7706 reg_pat[i++] = '?';
7707 else
Bram Moolenaarf4e11432013-07-03 16:53:03 +02007708 if (*p == ',' || *p == '%' || *p == '#'
Bram Moolenaar2288afe2015-08-11 16:20:05 +02007709 || vim_isspace(*p) || *p == '{' || *p == '}')
Bram Moolenaar8cd213c2010-06-01 21:57:09 +02007710 reg_pat[i++] = *p;
Bram Moolenaara946afe2013-08-02 15:22:39 +02007711 else if (*p == '\\' && p[1] == '\\' && p[2] == '{')
7712 {
7713 reg_pat[i++] = '\\';
7714 reg_pat[i++] = '{';
7715 p += 2;
7716 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00007717 else
7718 {
7719 if (allow_dirs != NULL && vim_ispathsep(*p)
7720#ifdef BACKSLASH_IN_FILENAME
7721 && (!no_bslash || *p != '\\')
7722#endif
7723 )
7724 *allow_dirs = TRUE;
7725 reg_pat[i++] = '\\';
7726 reg_pat[i++] = *p;
7727 }
7728 break;
7729#ifdef BACKSLASH_IN_FILENAME
7730 case '/':
7731 reg_pat[i++] = '[';
7732 reg_pat[i++] = '\\';
7733 reg_pat[i++] = '/';
7734 reg_pat[i++] = ']';
7735 if (allow_dirs != NULL)
7736 *allow_dirs = TRUE;
7737 break;
7738#endif
7739 case '{':
7740 reg_pat[i++] = '\\';
7741 reg_pat[i++] = '(';
7742 nested++;
7743 break;
7744 case '}':
7745 reg_pat[i++] = '\\';
7746 reg_pat[i++] = ')';
7747 --nested;
7748 break;
7749 case ',':
7750 if (nested)
7751 {
7752 reg_pat[i++] = '\\';
7753 reg_pat[i++] = '|';
7754 }
7755 else
7756 reg_pat[i++] = ',';
7757 break;
7758 default:
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00007759 if (enc_dbcs != 0 && (*mb_ptr2len)(p) > 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00007760 reg_pat[i++] = *p++;
Bram Moolenaar13505972019-01-24 15:04:48 +01007761 else if (allow_dirs != NULL && vim_ispathsep(*p))
Bram Moolenaar071d4272004-06-13 20:20:40 +00007762 *allow_dirs = TRUE;
7763 reg_pat[i++] = *p;
7764 break;
7765 }
7766 }
7767 if (add_dollar)
7768 reg_pat[i++] = '$';
7769 reg_pat[i] = NUL;
7770 if (nested != 0)
7771 {
7772 if (nested < 0)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007773 emsg(_("E219: Missing {."));
Bram Moolenaar071d4272004-06-13 20:20:40 +00007774 else
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007775 emsg(_("E220: Missing }."));
Bram Moolenaard23a8232018-02-10 18:45:26 +01007776 VIM_CLEAR(reg_pat);
Bram Moolenaar071d4272004-06-13 20:20:40 +00007777 }
7778 return reg_pat;
7779}
Bram Moolenaar540fc6f2010-12-17 16:27:16 +01007780
7781#if defined(EINTR) || defined(PROTO)
7782/*
7783 * Version of read() that retries when interrupted by EINTR (possibly
7784 * by a SIGWINCH).
7785 */
7786 long
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01007787read_eintr(int fd, void *buf, size_t bufsize)
Bram Moolenaar540fc6f2010-12-17 16:27:16 +01007788{
7789 long ret;
7790
7791 for (;;)
7792 {
7793 ret = vim_read(fd, buf, bufsize);
7794 if (ret >= 0 || errno != EINTR)
7795 break;
7796 }
7797 return ret;
7798}
7799
7800/*
7801 * Version of write() that retries when interrupted by EINTR (possibly
7802 * by a SIGWINCH).
7803 */
7804 long
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01007805write_eintr(int fd, void *buf, size_t bufsize)
Bram Moolenaar540fc6f2010-12-17 16:27:16 +01007806{
7807 long ret = 0;
7808 long wlen;
7809
7810 /* Repeat the write() so long it didn't fail, other than being interrupted
7811 * by a signal. */
7812 while (ret < (long)bufsize)
7813 {
Bram Moolenaar9c263032010-12-17 18:06:06 +01007814 wlen = vim_write(fd, (char *)buf + ret, bufsize - ret);
Bram Moolenaar540fc6f2010-12-17 16:27:16 +01007815 if (wlen < 0)
7816 {
7817 if (errno != EINTR)
7818 break;
7819 }
7820 else
7821 ret += wlen;
7822 }
7823 return ret;
7824}
7825#endif