blob: 446ae7edf8c7e9709970010d332ed704b2f2883b [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 */
52#ifdef WIN3264
53# 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 Moolenaar13505972019-01-24 15:04:48 +0100109#ifdef WIN3264
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 Moolenaar13505972019-01-24 15:04:48 +01001041#ifdef WIN3264
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 Moolenaar13505972019-01-24 15:04:48 +01001232#ifdef WIN3264
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 Moolenaar13505972019-01-24 15:04:48 +01001593#ifdef WIN3264
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)
2325 {
2326 iconv_close(iconv_fd);
2327 iconv_fd = (iconv_t)-1;
2328 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002329#endif
2330
2331 if (!read_buffer && !read_stdin)
2332 close(fd); /* errors are ignored */
Bram Moolenaarf05da212009-11-17 16:13:15 +00002333#ifdef HAVE_FD_CLOEXEC
2334 else
2335 {
2336 int fdflags = fcntl(fd, F_GETFD);
2337 if (fdflags >= 0 && (fdflags & FD_CLOEXEC) == 0)
Bram Moolenaarfbc4b4d2016-02-07 15:14:01 +01002338 (void)fcntl(fd, F_SETFD, fdflags | FD_CLOEXEC);
Bram Moolenaarf05da212009-11-17 16:13:15 +00002339 }
2340#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002341 vim_free(buffer);
2342
2343#ifdef HAVE_DUP
2344 if (read_stdin)
2345 {
2346 /* Use stderr for stdin, makes shell commands work. */
2347 close(0);
Bram Moolenaar42335f52018-09-13 15:33:43 +02002348 vim_ignored = dup(2);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002349 }
2350#endif
2351
Bram Moolenaar071d4272004-06-13 20:20:40 +00002352 if (tmpname != NULL)
2353 {
2354 mch_remove(tmpname); /* delete converted file */
2355 vim_free(tmpname);
2356 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002357 --no_wait_return; /* may wait for return now */
2358
2359 /*
2360 * In recovery mode everything but autocommands is skipped.
2361 */
2362 if (!recoverymode)
2363 {
2364 /* need to delete the last line, which comes from the empty buffer */
2365 if (newfile && wasempty && !(curbuf->b_ml.ml_flags & ML_EMPTY))
2366 {
2367#ifdef FEAT_NETBEANS_INTG
2368 netbeansFireChanges = 0;
2369#endif
2370 ml_delete(curbuf->b_ml.ml_line_count, FALSE);
2371#ifdef FEAT_NETBEANS_INTG
2372 netbeansFireChanges = 1;
2373#endif
2374 --linecnt;
2375 }
2376 linecnt = curbuf->b_ml.ml_line_count - linecnt;
2377 if (filesize == 0)
2378 linecnt = 0;
2379 if (newfile || read_buffer)
Bram Moolenaar7263a772007-05-10 17:35:54 +00002380 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00002381 redraw_curbuf_later(NOT_VALID);
Bram Moolenaar7263a772007-05-10 17:35:54 +00002382#ifdef FEAT_DIFF
2383 /* After reading the text into the buffer the diff info needs to
2384 * be updated. */
2385 diff_invalidate(curbuf);
2386#endif
2387#ifdef FEAT_FOLDING
2388 /* All folds in the window are invalid now. Mark them for update
2389 * before triggering autocommands. */
2390 foldUpdateAll(curwin);
2391#endif
2392 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002393 else if (linecnt) /* appended at least one line */
2394 appended_lines_mark(from, linecnt);
2395
Bram Moolenaar071d4272004-06-13 20:20:40 +00002396#ifndef ALWAYS_USE_GUI
2397 /*
2398 * If we were reading from the same terminal as where messages go,
2399 * the screen will have been messed up.
2400 * Switch on raw mode now and clear the screen.
2401 */
2402 if (read_stdin)
2403 {
2404 settmode(TMODE_RAW); /* set to raw mode */
2405 starttermcap();
2406 screenclear();
2407 }
2408#endif
2409
2410 if (got_int)
2411 {
2412 if (!(flags & READ_DUMMY))
2413 {
2414 filemess(curbuf, sfname, (char_u *)_(e_interr), 0);
2415 if (newfile)
2416 curbuf->b_p_ro = TRUE; /* must use "w!" now */
2417 }
2418 msg_scroll = msg_save;
2419#ifdef FEAT_VIMINFO
2420 check_marks_read();
2421#endif
2422 return OK; /* an interrupt isn't really an error */
2423 }
2424
2425 if (!filtering && !(flags & READ_DUMMY))
2426 {
2427 msg_add_fname(curbuf, sfname); /* fname in IObuff with quotes */
2428 c = FALSE;
2429
2430#ifdef UNIX
Bram Moolenaard569bb02018-08-11 13:57:20 +02002431 if (S_ISFIFO(perm)) /* fifo */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002432 {
2433 STRCAT(IObuff, _("[fifo]"));
2434 c = TRUE;
2435 }
Bram Moolenaard569bb02018-08-11 13:57:20 +02002436 if (S_ISSOCK(perm)) /* or socket */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002437 {
2438 STRCAT(IObuff, _("[socket]"));
2439 c = TRUE;
2440 }
Bram Moolenaarfe1c56d2007-07-10 15:10:54 +00002441# ifdef OPEN_CHR_FILES
2442 if (S_ISCHR(perm)) /* or character special */
2443 {
2444 STRCAT(IObuff, _("[character special]"));
2445 c = TRUE;
2446 }
2447# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002448#endif
2449 if (curbuf->b_p_ro)
2450 {
2451 STRCAT(IObuff, shortmess(SHM_RO) ? _("[RO]") : _("[readonly]"));
2452 c = TRUE;
2453 }
2454 if (read_no_eol_lnum)
2455 {
2456 msg_add_eol();
2457 c = TRUE;
2458 }
2459 if (ff_error == EOL_DOS)
2460 {
2461 STRCAT(IObuff, _("[CR missing]"));
2462 c = TRUE;
2463 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002464 if (split)
2465 {
2466 STRCAT(IObuff, _("[long lines split]"));
2467 c = TRUE;
2468 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002469 if (notconverted)
2470 {
2471 STRCAT(IObuff, _("[NOT converted]"));
2472 c = TRUE;
2473 }
2474 else if (converted)
2475 {
2476 STRCAT(IObuff, _("[converted]"));
2477 c = TRUE;
2478 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002479#ifdef FEAT_CRYPT
2480 if (cryptkey != NULL)
2481 {
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02002482 crypt_append_msg(curbuf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002483 c = TRUE;
2484 }
2485#endif
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00002486 if (conv_error != 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002487 {
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00002488 sprintf((char *)IObuff + STRLEN(IObuff),
2489 _("[CONVERSION ERROR in line %ld]"), (long)conv_error);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002490 c = TRUE;
2491 }
2492 else if (illegal_byte > 0)
2493 {
2494 sprintf((char *)IObuff + STRLEN(IObuff),
2495 _("[ILLEGAL BYTE in line %ld]"), (long)illegal_byte);
2496 c = TRUE;
2497 }
Bram Moolenaar13505972019-01-24 15:04:48 +01002498 else if (error)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002499 {
2500 STRCAT(IObuff, _("[READ ERRORS]"));
2501 c = TRUE;
2502 }
2503 if (msg_add_fileformat(fileformat))
2504 c = TRUE;
2505#ifdef FEAT_CRYPT
2506 if (cryptkey != NULL)
Bram Moolenaar40e6a712010-05-16 22:32:54 +02002507 msg_add_lines(c, (long)linecnt, filesize
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02002508 - crypt_get_header_len(crypt_get_method_nr(curbuf)));
Bram Moolenaar071d4272004-06-13 20:20:40 +00002509 else
2510#endif
2511 msg_add_lines(c, (long)linecnt, filesize);
2512
Bram Moolenaard23a8232018-02-10 18:45:26 +01002513 VIM_CLEAR(keep_msg);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002514 msg_scrolled_ign = TRUE;
2515#ifdef ALWAYS_USE_GUI
2516 /* Don't show the message when reading stdin, it would end up in a
2517 * message box (which might be shown when exiting!) */
2518 if (read_stdin || read_buffer)
2519 p = msg_may_trunc(FALSE, IObuff);
2520 else
2521#endif
Bram Moolenaar32526b32019-01-19 17:43:09 +01002522 p = (char_u *)msg_trunc_attr((char *)IObuff, FALSE, 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002523 if (read_stdin || read_buffer || restart_edit != 0
Bram Moolenaar1c7715d2005-10-03 22:02:18 +00002524 || (msg_scrolled != 0 && !need_wait_return))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002525 /* Need to repeat the message after redrawing when:
2526 * - When reading from stdin (the screen will be cleared next).
2527 * - When restart_edit is set (otherwise there will be a delay
2528 * before redrawing).
2529 * - When the screen was scrolled but there is no wait-return
2530 * prompt. */
Bram Moolenaar8f7fd652006-02-21 22:04:51 +00002531 set_keep_msg(p, 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002532 msg_scrolled_ign = FALSE;
2533 }
2534
2535 /* with errors writing the file requires ":w!" */
2536 if (newfile && (error
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00002537 || conv_error != 0
Bram Moolenaar13505972019-01-24 15:04:48 +01002538 || (illegal_byte > 0 && bad_char_behavior != BAD_KEEP)))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002539 curbuf->b_p_ro = TRUE;
2540
2541 u_clearline(); /* cannot use "U" command after adding lines */
2542
2543 /*
2544 * In Ex mode: cursor at last new line.
2545 * Otherwise: cursor at first new line.
2546 */
2547 if (exmode_active)
2548 curwin->w_cursor.lnum = from + linecnt;
2549 else
2550 curwin->w_cursor.lnum = from + 1;
2551 check_cursor_lnum();
2552 beginline(BL_WHITE | BL_FIX); /* on first non-blank */
2553
2554 /*
2555 * Set '[ and '] marks to the newly read lines.
2556 */
2557 curbuf->b_op_start.lnum = from + 1;
2558 curbuf->b_op_start.col = 0;
2559 curbuf->b_op_end.lnum = from + linecnt;
2560 curbuf->b_op_end.col = 0;
Bram Moolenaar03f48552006-02-28 23:52:23 +00002561
2562#ifdef WIN32
2563 /*
2564 * Work around a weird problem: When a file has two links (only
2565 * possible on NTFS) and we write through one link, then stat() it
Bram Moolenaarb23a7e82008-06-27 18:42:32 +00002566 * through the other link, the timestamp information may be wrong.
Bram Moolenaar03f48552006-02-28 23:52:23 +00002567 * It's correct again after reading the file, thus reset the timestamp
2568 * here.
2569 */
2570 if (newfile && !read_stdin && !read_buffer
2571 && mch_stat((char *)fname, &st) >= 0)
2572 {
2573 buf_store_time(curbuf, &st, fname);
2574 curbuf->b_mtime_read = curbuf->b_mtime;
2575 }
2576#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002577 }
2578 msg_scroll = msg_save;
2579
2580#ifdef FEAT_VIMINFO
2581 /*
2582 * Get the marks before executing autocommands, so they can be used there.
2583 */
2584 check_marks_read();
2585#endif
2586
Bram Moolenaar071d4272004-06-13 20:20:40 +00002587 /*
Bram Moolenaar34d72d42015-07-17 14:18:08 +02002588 * We remember if the last line of the read didn't have
2589 * an eol even when 'binary' is off, to support turning 'fixeol' off,
2590 * or writing the read again with 'binary' on. The latter is required
2591 * for ":autocmd FileReadPost *.gz set bin|'[,']!gunzip" to work.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002592 */
Bram Moolenaarcab35ad2011-02-15 17:39:22 +01002593 curbuf->b_no_eol_lnum = read_no_eol_lnum;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002594
Bram Moolenaarf9bb7342010-08-04 14:29:54 +02002595 /* When reloading a buffer put the cursor at the first line that is
2596 * different. */
2597 if (flags & READ_KEEP_UNDO)
2598 u_find_first_changed();
2599
Bram Moolenaar55debbe2010-05-23 23:34:36 +02002600#ifdef FEAT_PERSISTENT_UNDO
2601 /*
2602 * When opening a new file locate undo info and read it.
2603 */
2604 if (read_undo_file)
2605 {
2606 char_u hash[UNDO_HASH_SIZE];
2607
2608 sha256_finish(&sha_ctx, hash);
Bram Moolenaar6ed8ed82010-05-30 20:40:11 +02002609 u_read_undo(NULL, hash, fname);
Bram Moolenaar55debbe2010-05-23 23:34:36 +02002610 }
2611#endif
2612
Bram Moolenaarf71d7b92016-08-09 22:14:05 +02002613 if (!read_stdin && !read_fifo && (!read_buffer || sfname != NULL))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002614 {
2615 int m = msg_scroll;
2616 int n = msg_scrolled;
2617
2618 /* Save the fileformat now, otherwise the buffer will be considered
2619 * modified if the format/encoding was automatically detected. */
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002620 if (set_options)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002621 save_file_ff(curbuf);
2622
2623 /*
2624 * The output from the autocommands should not overwrite anything and
2625 * should not be overwritten: Set msg_scroll, restore its value if no
2626 * output was done.
2627 */
2628 msg_scroll = TRUE;
2629 if (filtering)
2630 apply_autocmds_exarg(EVENT_FILTERREADPOST, NULL, sfname,
2631 FALSE, curbuf, eap);
Bram Moolenaarf71d7b92016-08-09 22:14:05 +02002632 else if (newfile || (read_buffer && sfname != NULL))
Bram Moolenaarc3691332016-04-20 12:49:49 +02002633 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00002634 apply_autocmds_exarg(EVENT_BUFREADPOST, NULL, sfname,
2635 FALSE, curbuf, eap);
Bram Moolenaarc3691332016-04-20 12:49:49 +02002636 if (!au_did_filetype && *curbuf->b_p_ft != NUL)
2637 /*
2638 * EVENT_FILETYPE was not triggered but the buffer already has a
2639 * filetype. Trigger EVENT_FILETYPE using the existing filetype.
2640 */
2641 apply_autocmds(EVENT_FILETYPE, curbuf->b_p_ft, curbuf->b_fname,
2642 TRUE, curbuf);
2643 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002644 else
2645 apply_autocmds_exarg(EVENT_FILEREADPOST, sfname, sfname,
2646 FALSE, NULL, eap);
2647 if (msg_scrolled == n)
2648 msg_scroll = m;
Bram Moolenaarcab35ad2011-02-15 17:39:22 +01002649# ifdef FEAT_EVAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00002650 if (aborting()) /* autocmds may abort script processing */
2651 return FAIL;
Bram Moolenaarcab35ad2011-02-15 17:39:22 +01002652# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00002653 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002654
2655 if (recoverymode && error)
2656 return FAIL;
2657 return OK;
2658}
2659
Bram Moolenaarf04507d2016-08-20 15:05:39 +02002660#if defined(OPEN_CHR_FILES) || defined(PROTO)
Bram Moolenaarfe1c56d2007-07-10 15:10:54 +00002661/*
2662 * Returns TRUE if the file name argument is of the form "/dev/fd/\d\+",
2663 * which is the name of files used for process substitution output by
2664 * some shells on some operating systems, e.g., bash on SunOS.
2665 * Do not accept "/dev/fd/[012]", opening these may hang Vim.
2666 */
Bram Moolenaarf04507d2016-08-20 15:05:39 +02002667 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002668is_dev_fd_file(char_u *fname)
Bram Moolenaarfe1c56d2007-07-10 15:10:54 +00002669{
2670 return (STRNCMP(fname, "/dev/fd/", 8) == 0
2671 && VIM_ISDIGIT(fname[8])
2672 && *skipdigits(fname + 9) == NUL
2673 && (fname[9] != NUL
2674 || (fname[8] != '0' && fname[8] != '1' && fname[8] != '2')));
2675}
2676#endif
2677
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00002678/*
2679 * From the current line count and characters read after that, estimate the
2680 * line number where we are now.
2681 * Used for error messages that include a line number.
2682 */
2683 static linenr_T
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002684readfile_linenr(
2685 linenr_T linecnt, /* line count before reading more bytes */
2686 char_u *p, /* start of more bytes read */
2687 char_u *endp) /* end of more bytes read */
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00002688{
2689 char_u *s;
2690 linenr_T lnum;
2691
2692 lnum = curbuf->b_ml.ml_line_count - linecnt + 1;
2693 for (s = p; s < endp; ++s)
2694 if (*s == '\n')
2695 ++lnum;
2696 return lnum;
2697}
Bram Moolenaarb0bf8582005-12-13 20:02:15 +00002698
Bram Moolenaar071d4272004-06-13 20:20:40 +00002699/*
Bram Moolenaar195d6352005-12-19 22:08:24 +00002700 * Fill "*eap" to force the 'fileencoding', 'fileformat' and 'binary to be
2701 * equal to the buffer "buf". Used for calling readfile().
Bram Moolenaar071d4272004-06-13 20:20:40 +00002702 * Returns OK or FAIL.
2703 */
2704 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002705prep_exarg(exarg_T *eap, buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002706{
Bram Moolenaar13505972019-01-24 15:04:48 +01002707 eap->cmd = alloc(15 + (unsigned)STRLEN(buf->b_p_fenc));
Bram Moolenaar071d4272004-06-13 20:20:40 +00002708 if (eap->cmd == NULL)
2709 return FAIL;
2710
Bram Moolenaar333b80a2018-04-04 22:57:29 +02002711 sprintf((char *)eap->cmd, "e ++enc=%s", buf->b_p_fenc);
2712 eap->force_enc = 8;
Bram Moolenaar195d6352005-12-19 22:08:24 +00002713 eap->bad_char = buf->b_bad_char;
Bram Moolenaar333b80a2018-04-04 22:57:29 +02002714 eap->force_ff = *buf->b_p_ff;
Bram Moolenaar195d6352005-12-19 22:08:24 +00002715
2716 eap->force_bin = buf->b_p_bin ? FORCE_BIN : FORCE_NOBIN;
Bram Moolenaar910f66f2006-04-05 20:41:53 +00002717 eap->read_edit = FALSE;
Bram Moolenaar195d6352005-12-19 22:08:24 +00002718 eap->forceit = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002719 return OK;
2720}
2721
Bram Moolenaarad875fb2013-07-24 15:02:03 +02002722/*
2723 * Set default or forced 'fileformat' and 'binary'.
2724 */
2725 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002726set_file_options(int set_options, exarg_T *eap)
Bram Moolenaarad875fb2013-07-24 15:02:03 +02002727{
2728 /* set default 'fileformat' */
2729 if (set_options)
2730 {
2731 if (eap != NULL && eap->force_ff != 0)
2732 set_fileformat(get_fileformat_force(curbuf, eap), OPT_LOCAL);
2733 else if (*p_ffs != NUL)
2734 set_fileformat(default_fileformat(), OPT_LOCAL);
2735 }
2736
2737 /* set or reset 'binary' */
2738 if (eap != NULL && eap->force_bin != 0)
2739 {
2740 int oldval = curbuf->b_p_bin;
2741
2742 curbuf->b_p_bin = (eap->force_bin == FORCE_BIN);
2743 set_options_bin(oldval, curbuf->b_p_bin, OPT_LOCAL);
2744 }
2745}
2746
Bram Moolenaarad875fb2013-07-24 15:02:03 +02002747/*
2748 * Set forced 'fileencoding'.
2749 */
2750 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002751set_forced_fenc(exarg_T *eap)
Bram Moolenaarad875fb2013-07-24 15:02:03 +02002752{
2753 if (eap->force_enc != 0)
2754 {
2755 char_u *fenc = enc_canonize(eap->cmd + eap->force_enc);
2756
2757 if (fenc != NULL)
2758 set_string_option_direct((char_u *)"fenc", -1,
2759 fenc, OPT_FREE|OPT_LOCAL, 0);
2760 vim_free(fenc);
2761 }
2762}
2763
Bram Moolenaar071d4272004-06-13 20:20:40 +00002764/*
2765 * Find next fileencoding to use from 'fileencodings'.
2766 * "pp" points to fenc_next. It's advanced to the next item.
2767 * When there are no more items, an empty string is returned and *pp is set to
2768 * NULL.
2769 * When *pp is not set to NULL, the result is in allocated memory.
2770 */
2771 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002772next_fenc(char_u **pp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002773{
2774 char_u *p;
2775 char_u *r;
2776
2777 if (**pp == NUL)
2778 {
2779 *pp = NULL;
2780 return (char_u *)"";
2781 }
2782 p = vim_strchr(*pp, ',');
2783 if (p == NULL)
2784 {
2785 r = enc_canonize(*pp);
2786 *pp += STRLEN(*pp);
2787 }
2788 else
2789 {
2790 r = vim_strnsave(*pp, (int)(p - *pp));
2791 *pp = p + 1;
2792 if (r != NULL)
2793 {
2794 p = enc_canonize(r);
2795 vim_free(r);
2796 r = p;
2797 }
2798 }
2799 if (r == NULL) /* out of memory */
2800 {
2801 r = (char_u *)"";
2802 *pp = NULL;
2803 }
2804 return r;
2805}
2806
Bram Moolenaar13505972019-01-24 15:04:48 +01002807#ifdef FEAT_EVAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00002808/*
2809 * Convert a file with the 'charconvert' expression.
2810 * This closes the file which is to be read, converts it and opens the
2811 * resulting file for reading.
2812 * Returns name of the resulting converted file (the caller should delete it
2813 * after reading it).
2814 * Returns NULL if the conversion failed ("*fdp" is not set) .
2815 */
2816 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002817readfile_charconvert(
2818 char_u *fname, /* name of input file */
2819 char_u *fenc, /* converted from */
2820 int *fdp) /* in/out: file descriptor of file */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002821{
2822 char_u *tmpname;
Bram Moolenaar32526b32019-01-19 17:43:09 +01002823 char *errmsg = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002824
Bram Moolenaare5c421c2015-03-31 13:33:08 +02002825 tmpname = vim_tempname('r', FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002826 if (tmpname == NULL)
Bram Moolenaar32526b32019-01-19 17:43:09 +01002827 errmsg = _("Can't find temp file for conversion");
Bram Moolenaar071d4272004-06-13 20:20:40 +00002828 else
2829 {
2830 close(*fdp); /* close the input file, ignore errors */
2831 *fdp = -1;
2832 if (eval_charconvert(fenc, enc_utf8 ? (char_u *)"utf-8" : p_enc,
2833 fname, tmpname) == FAIL)
Bram Moolenaar32526b32019-01-19 17:43:09 +01002834 errmsg = _("Conversion with 'charconvert' failed");
Bram Moolenaar071d4272004-06-13 20:20:40 +00002835 if (errmsg == NULL && (*fdp = mch_open((char *)tmpname,
2836 O_RDONLY | O_EXTRA, 0)) < 0)
Bram Moolenaar32526b32019-01-19 17:43:09 +01002837 errmsg = _("can't read output of 'charconvert'");
Bram Moolenaar071d4272004-06-13 20:20:40 +00002838 }
2839
2840 if (errmsg != NULL)
2841 {
2842 /* Don't use emsg(), it breaks mappings, the retry with
2843 * another type of conversion might still work. */
Bram Moolenaar32526b32019-01-19 17:43:09 +01002844 msg(errmsg);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002845 if (tmpname != NULL)
2846 {
2847 mch_remove(tmpname); /* delete converted file */
Bram Moolenaard23a8232018-02-10 18:45:26 +01002848 VIM_CLEAR(tmpname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002849 }
2850 }
2851
2852 /* If the input file is closed, open it (caller should check for error). */
2853 if (*fdp < 0)
2854 *fdp = mch_open((char *)fname, O_RDONLY | O_EXTRA, 0);
2855
2856 return tmpname;
2857}
Bram Moolenaar071d4272004-06-13 20:20:40 +00002858#endif
2859
Bram Moolenaar13505972019-01-24 15:04:48 +01002860
Bram Moolenaar071d4272004-06-13 20:20:40 +00002861#ifdef FEAT_VIMINFO
2862/*
2863 * Read marks for the current buffer from the viminfo file, when we support
2864 * buffer marks and the buffer has a name.
2865 */
2866 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002867check_marks_read(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002868{
2869 if (!curbuf->b_marks_read && get_viminfo_parameter('\'') > 0
2870 && curbuf->b_ffname != NULL)
Bram Moolenaard812df62008-11-09 12:46:09 +00002871 read_viminfo(NULL, VIF_WANT_MARKS);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002872
2873 /* Always set b_marks_read; needed when 'viminfo' is changed to include
2874 * the ' parameter after opening a buffer. */
2875 curbuf->b_marks_read = TRUE;
2876}
2877#endif
2878
Bram Moolenaara3ff49f2010-05-30 22:48:02 +02002879#if defined(FEAT_CRYPT) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002880/*
Bram Moolenaar40e6a712010-05-16 22:32:54 +02002881 * Check for magic number used for encryption. Applies to the current buffer.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002882 * If found, the magic number is removed from ptr[*sizep] and *sizep and
2883 * *filesizep are updated.
2884 * Return the (new) encryption key, NULL for no encryption.
2885 */
2886 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002887check_for_cryptkey(
2888 char_u *cryptkey, /* previous encryption key or NULL */
2889 char_u *ptr, /* pointer to read bytes */
2890 long *sizep, /* length of read bytes */
Bram Moolenaar8767f522016-07-01 17:17:39 +02002891 off_T *filesizep, /* nr of bytes used from file */
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002892 int newfile, /* editing a new buffer */
2893 char_u *fname, /* file name to display */
2894 int *did_ask) /* flag: whether already asked for key */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002895{
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02002896 int method = crypt_method_nr_from_magic((char *)ptr, *sizep);
Bram Moolenaarcf81aef2013-08-25 17:46:08 +02002897 int b_p_ro = curbuf->b_p_ro;
Bram Moolenaar40e6a712010-05-16 22:32:54 +02002898
2899 if (method >= 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002900 {
Bram Moolenaarcf81aef2013-08-25 17:46:08 +02002901 /* Mark the buffer as read-only until the decryption has taken place.
2902 * Avoids accidentally overwriting the file with garbage. */
2903 curbuf->b_p_ro = TRUE;
2904
Bram Moolenaar2be79502014-08-13 21:58:28 +02002905 /* Set the cryptmethod local to the buffer. */
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02002906 crypt_set_cm_option(curbuf, method);
Bram Moolenaarf50a2532010-05-21 15:36:08 +02002907 if (cryptkey == NULL && !*did_ask)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002908 {
2909 if (*curbuf->b_p_key)
2910 cryptkey = curbuf->b_p_key;
2911 else
2912 {
Bram Moolenaar40e6a712010-05-16 22:32:54 +02002913 /* When newfile is TRUE, store the typed key in the 'key'
2914 * option and don't free it. bf needs hash of the key saved.
Bram Moolenaarf50a2532010-05-21 15:36:08 +02002915 * Don't ask for the key again when first time Enter was hit.
2916 * Happens when retrying to detect encoding. */
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01002917 smsg(_(need_key_msg), fname);
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02002918 msg_scroll = TRUE;
Bram Moolenaar3a0c9082014-11-12 15:15:42 +01002919 crypt_check_method(method);
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02002920 cryptkey = crypt_get_key(newfile, FALSE);
Bram Moolenaarf50a2532010-05-21 15:36:08 +02002921 *did_ask = TRUE;
2922
Bram Moolenaar071d4272004-06-13 20:20:40 +00002923 /* check if empty key entered */
2924 if (cryptkey != NULL && *cryptkey == NUL)
2925 {
2926 if (cryptkey != curbuf->b_p_key)
2927 vim_free(cryptkey);
2928 cryptkey = NULL;
2929 }
2930 }
2931 }
2932
2933 if (cryptkey != NULL)
2934 {
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02002935 int header_len;
Bram Moolenaar40e6a712010-05-16 22:32:54 +02002936
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02002937 curbuf->b_cryptstate = crypt_create_from_header(
2938 method, cryptkey, ptr);
2939 crypt_set_cm_option(curbuf, method);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002940
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02002941 /* Remove cryptmethod specific header from the text. */
2942 header_len = crypt_get_header_len(method);
Bram Moolenaar680e0152016-09-25 20:54:11 +02002943 if (*sizep <= header_len)
2944 /* invalid header, buffer can't be encrypted */
2945 return NULL;
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02002946 *filesizep += header_len;
2947 *sizep -= header_len;
2948 mch_memmove(ptr, ptr + header_len, (size_t)*sizep);
2949
Bram Moolenaarcf81aef2013-08-25 17:46:08 +02002950 /* Restore the read-only flag. */
2951 curbuf->b_p_ro = b_p_ro;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002952 }
2953 }
Bram Moolenaar40e6a712010-05-16 22:32:54 +02002954 /* When starting to edit a new file which does not have encryption, clear
2955 * the 'key' option, except when starting up (called with -x argument) */
Bram Moolenaarfa0ff9a2010-07-25 16:05:19 +02002956 else if (newfile && *curbuf->b_p_key != NUL && !starting)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002957 set_option_value((char_u *)"key", 0L, (char_u *)"", OPT_LOCAL);
2958
2959 return cryptkey;
2960}
Bram Moolenaar80794b12010-06-13 05:20:42 +02002961#endif /* FEAT_CRYPT */
2962
Bram Moolenaar071d4272004-06-13 20:20:40 +00002963#ifdef UNIX
2964 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01002965set_file_time(
2966 char_u *fname,
2967 time_t atime, /* access time */
2968 time_t mtime) /* modification time */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002969{
2970# if defined(HAVE_UTIME) && defined(HAVE_UTIME_H)
2971 struct utimbuf buf;
2972
2973 buf.actime = atime;
2974 buf.modtime = mtime;
2975 (void)utime((char *)fname, &buf);
2976# else
2977# if defined(HAVE_UTIMES)
2978 struct timeval tvp[2];
2979
2980 tvp[0].tv_sec = atime;
2981 tvp[0].tv_usec = 0;
2982 tvp[1].tv_sec = mtime;
2983 tvp[1].tv_usec = 0;
2984# ifdef NeXT
2985 (void)utimes((char *)fname, tvp);
2986# else
2987 (void)utimes((char *)fname, (const struct timeval *)&tvp);
2988# endif
2989# endif
2990# endif
2991}
2992#endif /* UNIX */
2993
Bram Moolenaard4755bb2004-09-02 19:12:26 +00002994#if defined(VMS) && !defined(MIN)
2995/* Older DECC compiler for VAX doesn't define MIN() */
2996# define MIN(a, b) ((a) < (b) ? (a) : (b))
2997#endif
2998
Bram Moolenaar071d4272004-06-13 20:20:40 +00002999/*
Bram Moolenaar5386a122007-06-28 20:02:32 +00003000 * Return TRUE if a file appears to be read-only from the file permissions.
3001 */
3002 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003003check_file_readonly(
3004 char_u *fname, /* full path to file */
3005 int perm) /* known permissions on file */
Bram Moolenaar5386a122007-06-28 20:02:32 +00003006{
3007#ifndef USE_MCH_ACCESS
3008 int fd = 0;
3009#endif
3010
3011 return (
3012#ifdef USE_MCH_ACCESS
3013# ifdef UNIX
3014 (perm & 0222) == 0 ||
3015# endif
3016 mch_access((char *)fname, W_OK)
3017#else
3018 (fd = mch_open((char *)fname, O_RDWR | O_EXTRA, 0)) < 0
3019 ? TRUE : (close(fd), FALSE)
3020#endif
3021 );
3022}
3023
3024
3025/*
Bram Moolenaar292ad192005-12-11 21:29:51 +00003026 * buf_write() - write to file "fname" lines "start" through "end"
Bram Moolenaar071d4272004-06-13 20:20:40 +00003027 *
3028 * We do our own buffering here because fwrite() is so slow.
3029 *
Bram Moolenaar292ad192005-12-11 21:29:51 +00003030 * If "forceit" is true, we don't care for errors when attempting backups.
3031 * In case of an error everything possible is done to restore the original
Bram Moolenaare37d50a2008-08-06 17:06:04 +00003032 * file. But when "forceit" is TRUE, we risk losing it.
Bram Moolenaar292ad192005-12-11 21:29:51 +00003033 *
3034 * When "reset_changed" is TRUE and "append" == FALSE and "start" == 1 and
3035 * "end" == curbuf->b_ml.ml_line_count, reset curbuf->b_changed.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003036 *
3037 * This function must NOT use NameBuff (because it's called by autowrite()).
3038 *
3039 * return FAIL for failure, OK otherwise
3040 */
3041 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003042buf_write(
3043 buf_T *buf,
3044 char_u *fname,
3045 char_u *sfname,
3046 linenr_T start,
3047 linenr_T end,
3048 exarg_T *eap, /* for forced 'ff' and 'fenc', can be
Bram Moolenaar071d4272004-06-13 20:20:40 +00003049 NULL! */
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01003050 int append, /* append to the file */
3051 int forceit,
3052 int reset_changed,
3053 int filtering)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003054{
3055 int fd;
3056 char_u *backup = NULL;
3057 int backup_copy = FALSE; /* copy the original file? */
3058 int dobackup;
3059 char_u *ffname;
3060 char_u *wfname = NULL; /* name of file to write to */
3061 char_u *s;
3062 char_u *ptr;
3063 char_u c;
3064 int len;
3065 linenr_T lnum;
3066 long nchars;
3067 char_u *errmsg = NULL;
Bram Moolenaar32b485f2009-07-29 16:06:27 +00003068 int errmsg_allocated = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003069 char_u *errnum = NULL;
3070 char_u *buffer;
3071 char_u smallbuf[SMBUFSIZE];
3072 char_u *backup_ext;
3073 int bufsize;
3074 long perm; /* file permissions */
3075 int retval = OK;
3076 int newfile = FALSE; /* TRUE if file doesn't exist yet */
3077 int msg_save = msg_scroll;
3078 int overwriting; /* TRUE if writing over original */
3079 int no_eol = FALSE; /* no end-of-line written */
3080 int device = FALSE; /* writing to a device */
Bram Moolenaar8767f522016-07-01 17:17:39 +02003081 stat_T st_old;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003082 int prev_got_int = got_int;
Bram Moolenaare6bf6552017-06-27 22:11:51 +02003083 int checking_conversion;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003084 int file_readonly = FALSE; /* overwritten file is read-only */
3085 static char *err_readonly = "is read-only (cannot override: \"W\" in 'cpoptions')";
Bram Moolenaara06ecab2016-07-16 14:47:36 +02003086#if defined(UNIX) /*XXX fix me sometime? */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003087 int made_writable = FALSE; /* 'w' bit has been set */
3088#endif
3089 /* writing everything */
3090 int whole = (start == 1 && end == buf->b_ml.ml_line_count);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003091 linenr_T old_line_count = buf->b_ml.ml_line_count;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003092 int attr;
3093 int fileformat;
3094 int write_bin;
3095 struct bw_info write_info; /* info for buf_write_bytes() */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003096 int converted = FALSE;
3097 int notconverted = FALSE;
3098 char_u *fenc; /* effective 'fileencoding' */
3099 char_u *fenc_tofree = NULL; /* allocated "fenc" */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003100#ifdef HAS_BW_FLAGS
3101 int wb_flags = 0;
3102#endif
3103#ifdef HAVE_ACL
3104 vim_acl_T acl = NULL; /* ACL copied from original file to
3105 backup or new file */
3106#endif
Bram Moolenaar55debbe2010-05-23 23:34:36 +02003107#ifdef FEAT_PERSISTENT_UNDO
3108 int write_undo_file = FALSE;
3109 context_sha256_T sha_ctx;
3110#endif
Bram Moolenaarb8ee25a2014-09-23 15:45:08 +02003111 unsigned int bkc = get_bkc_value(buf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003112
3113 if (fname == NULL || *fname == NUL) /* safety check */
3114 return FAIL;
Bram Moolenaar70d60e92009-12-31 13:53:33 +00003115 if (buf->b_ml.ml_mfp == NULL)
3116 {
3117 /* This can happen during startup when there is a stray "w" in the
3118 * vimrc file. */
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003119 emsg(_(e_emptybuf));
Bram Moolenaar70d60e92009-12-31 13:53:33 +00003120 return FAIL;
3121 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003122
3123 /*
3124 * Disallow writing from .exrc and .vimrc in current directory for
3125 * security reasons.
3126 */
3127 if (check_secure())
3128 return FAIL;
3129
3130 /* Avoid a crash for a long name. */
3131 if (STRLEN(fname) >= MAXPATHL)
3132 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003133 emsg(_(e_longname));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003134 return FAIL;
3135 }
3136
Bram Moolenaar071d4272004-06-13 20:20:40 +00003137 /* must init bw_conv_buf and bw_iconv_fd before jumping to "fail" */
3138 write_info.bw_conv_buf = NULL;
3139 write_info.bw_conv_error = FALSE;
Bram Moolenaar32b485f2009-07-29 16:06:27 +00003140 write_info.bw_conv_error_lnum = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003141 write_info.bw_restlen = 0;
Bram Moolenaar13505972019-01-24 15:04:48 +01003142#ifdef USE_ICONV
Bram Moolenaar071d4272004-06-13 20:20:40 +00003143 write_info.bw_iconv_fd = (iconv_t)-1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003144#endif
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02003145#ifdef FEAT_CRYPT
3146 write_info.bw_buffer = buf;
3147#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003148
Bram Moolenaardf177f62005-02-22 08:39:57 +00003149 /* After writing a file changedtick changes but we don't want to display
3150 * the line. */
3151 ex_no_reprint = TRUE;
3152
Bram Moolenaar071d4272004-06-13 20:20:40 +00003153 /*
3154 * If there is no file name yet, use the one for the written file.
3155 * BF_NOTEDITED is set to reflect this (in case the write fails).
3156 * Don't do this when the write is for a filter command.
Bram Moolenaar292ad192005-12-11 21:29:51 +00003157 * Don't do this when appending.
3158 * Only do this when 'cpoptions' contains the 'F' flag.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003159 */
Bram Moolenaar2d3f4892006-01-20 23:02:51 +00003160 if (buf->b_ffname == NULL
3161 && reset_changed
Bram Moolenaar071d4272004-06-13 20:20:40 +00003162 && whole
3163 && buf == curbuf
Bram Moolenaar402d2fe2005-04-15 21:00:38 +00003164#ifdef FEAT_QUICKFIX
3165 && !bt_nofile(buf)
3166#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003167 && !filtering
Bram Moolenaar292ad192005-12-11 21:29:51 +00003168 && (!append || vim_strchr(p_cpo, CPO_FNAMEAPP) != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003169 && vim_strchr(p_cpo, CPO_FNAMEW) != NULL)
3170 {
Bram Moolenaar2d3f4892006-01-20 23:02:51 +00003171 if (set_rw_fname(fname, sfname) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003172 return FAIL;
Bram Moolenaar2d3f4892006-01-20 23:02:51 +00003173 buf = curbuf; /* just in case autocmds made "buf" invalid */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003174 }
3175
3176 if (sfname == NULL)
3177 sfname = fname;
3178 /*
3179 * For Unix: Use the short file name whenever possible.
3180 * Avoids problems with networks and when directory names are changed.
3181 * Don't do this for MS-DOS, a "cd" in a sub-shell may have moved us to
3182 * another directory, which we don't detect
3183 */
3184 ffname = fname; /* remember full fname */
3185#ifdef UNIX
3186 fname = sfname;
3187#endif
3188
3189 if (buf->b_ffname != NULL && fnamecmp(ffname, buf->b_ffname) == 0)
3190 overwriting = TRUE;
3191 else
3192 overwriting = FALSE;
3193
3194 if (exiting)
Bram Moolenaar84a05ac2013-05-06 04:24:17 +02003195 settmode(TMODE_COOK); /* when exiting allow typeahead now */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003196
3197 ++no_wait_return; /* don't wait for return yet */
3198
3199 /*
3200 * Set '[ and '] marks to the lines to be written.
3201 */
3202 buf->b_op_start.lnum = start;
3203 buf->b_op_start.col = 0;
3204 buf->b_op_end.lnum = end;
3205 buf->b_op_end.col = 0;
3206
Bram Moolenaar071d4272004-06-13 20:20:40 +00003207 {
3208 aco_save_T aco;
3209 int buf_ffname = FALSE;
3210 int buf_sfname = FALSE;
3211 int buf_fname_f = FALSE;
3212 int buf_fname_s = FALSE;
3213 int did_cmd = FALSE;
Bram Moolenaar21cf8232004-07-16 20:18:37 +00003214 int nofile_err = FALSE;
Bram Moolenaar7c626922005-02-07 22:01:03 +00003215 int empty_memline = (buf->b_ml.ml_mfp == NULL);
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02003216 bufref_T bufref;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003217
3218 /*
Bram Moolenaar84a05ac2013-05-06 04:24:17 +02003219 * Apply PRE autocommands.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003220 * Set curbuf to the buffer to be written.
3221 * Careful: The autocommands may call buf_write() recursively!
3222 */
3223 if (ffname == buf->b_ffname)
3224 buf_ffname = TRUE;
3225 if (sfname == buf->b_sfname)
3226 buf_sfname = TRUE;
3227 if (fname == buf->b_ffname)
3228 buf_fname_f = TRUE;
3229 if (fname == buf->b_sfname)
3230 buf_fname_s = TRUE;
3231
3232 /* set curwin/curbuf to buf and save a few things */
3233 aucmd_prepbuf(&aco, buf);
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02003234 set_bufref(&bufref, buf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003235
3236 if (append)
3237 {
3238 if (!(did_cmd = apply_autocmds_exarg(EVENT_FILEAPPENDCMD,
3239 sfname, sfname, FALSE, curbuf, eap)))
Bram Moolenaar21cf8232004-07-16 20:18:37 +00003240 {
Bram Moolenaarb1b715d2006-01-21 22:09:43 +00003241#ifdef FEAT_QUICKFIX
Bram Moolenaarab79bcb2004-07-18 21:34:53 +00003242 if (overwriting && bt_nofile(curbuf))
Bram Moolenaar21cf8232004-07-16 20:18:37 +00003243 nofile_err = TRUE;
3244 else
Bram Moolenaarb1b715d2006-01-21 22:09:43 +00003245#endif
Bram Moolenaar21cf8232004-07-16 20:18:37 +00003246 apply_autocmds_exarg(EVENT_FILEAPPENDPRE,
Bram Moolenaar071d4272004-06-13 20:20:40 +00003247 sfname, sfname, FALSE, curbuf, eap);
Bram Moolenaar21cf8232004-07-16 20:18:37 +00003248 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003249 }
3250 else if (filtering)
3251 {
3252 apply_autocmds_exarg(EVENT_FILTERWRITEPRE,
3253 NULL, sfname, FALSE, curbuf, eap);
3254 }
3255 else if (reset_changed && whole)
3256 {
Bram Moolenaar39fc42e2011-09-02 11:56:20 +02003257 int was_changed = curbufIsChanged();
3258
3259 did_cmd = apply_autocmds_exarg(EVENT_BUFWRITECMD,
3260 sfname, sfname, FALSE, curbuf, eap);
3261 if (did_cmd)
3262 {
3263 if (was_changed && !curbufIsChanged())
3264 {
3265 /* Written everything correctly and BufWriteCmd has reset
3266 * 'modified': Correct the undo information so that an
3267 * undo now sets 'modified'. */
3268 u_unchanged(curbuf);
3269 u_update_save_nr(curbuf);
3270 }
3271 }
3272 else
Bram Moolenaar21cf8232004-07-16 20:18:37 +00003273 {
Bram Moolenaarb1b715d2006-01-21 22:09:43 +00003274#ifdef FEAT_QUICKFIX
Bram Moolenaar19a09a12005-03-04 23:39:37 +00003275 if (overwriting && bt_nofile(curbuf))
Bram Moolenaar21cf8232004-07-16 20:18:37 +00003276 nofile_err = TRUE;
3277 else
Bram Moolenaarb1b715d2006-01-21 22:09:43 +00003278#endif
Bram Moolenaar21cf8232004-07-16 20:18:37 +00003279 apply_autocmds_exarg(EVENT_BUFWRITEPRE,
Bram Moolenaar071d4272004-06-13 20:20:40 +00003280 sfname, sfname, FALSE, curbuf, eap);
Bram Moolenaar21cf8232004-07-16 20:18:37 +00003281 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003282 }
3283 else
3284 {
3285 if (!(did_cmd = apply_autocmds_exarg(EVENT_FILEWRITECMD,
3286 sfname, sfname, FALSE, curbuf, eap)))
Bram Moolenaar21cf8232004-07-16 20:18:37 +00003287 {
Bram Moolenaarb1b715d2006-01-21 22:09:43 +00003288#ifdef FEAT_QUICKFIX
Bram Moolenaar19a09a12005-03-04 23:39:37 +00003289 if (overwriting && bt_nofile(curbuf))
Bram Moolenaar21cf8232004-07-16 20:18:37 +00003290 nofile_err = TRUE;
3291 else
Bram Moolenaarb1b715d2006-01-21 22:09:43 +00003292#endif
Bram Moolenaar21cf8232004-07-16 20:18:37 +00003293 apply_autocmds_exarg(EVENT_FILEWRITEPRE,
Bram Moolenaar071d4272004-06-13 20:20:40 +00003294 sfname, sfname, FALSE, curbuf, eap);
Bram Moolenaar21cf8232004-07-16 20:18:37 +00003295 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003296 }
3297
3298 /* restore curwin/curbuf and a few other things */
3299 aucmd_restbuf(&aco);
3300
3301 /*
3302 * In three situations we return here and don't write the file:
3303 * 1. the autocommands deleted or unloaded the buffer.
3304 * 2. The autocommands abort script processing.
3305 * 3. If one of the "Cmd" autocommands was executed.
3306 */
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02003307 if (!bufref_valid(&bufref))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003308 buf = NULL;
Bram Moolenaar7c626922005-02-07 22:01:03 +00003309 if (buf == NULL || (buf->b_ml.ml_mfp == NULL && !empty_memline)
Bram Moolenaar1e015462005-09-25 22:16:38 +00003310 || did_cmd || nofile_err
3311#ifdef FEAT_EVAL
3312 || aborting()
3313#endif
3314 )
Bram Moolenaar071d4272004-06-13 20:20:40 +00003315 {
3316 --no_wait_return;
3317 msg_scroll = msg_save;
Bram Moolenaar21cf8232004-07-16 20:18:37 +00003318 if (nofile_err)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003319 emsg(_("E676: No matching autocommands for acwrite buffer"));
Bram Moolenaar21cf8232004-07-16 20:18:37 +00003320
Bram Moolenaar1e015462005-09-25 22:16:38 +00003321 if (nofile_err
3322#ifdef FEAT_EVAL
3323 || aborting()
3324#endif
3325 )
Bram Moolenaar071d4272004-06-13 20:20:40 +00003326 /* An aborting error, interrupt or exception in the
3327 * autocommands. */
3328 return FAIL;
3329 if (did_cmd)
3330 {
3331 if (buf == NULL)
3332 /* The buffer was deleted. We assume it was written
3333 * (can't retry anyway). */
3334 return OK;
3335 if (overwriting)
3336 {
3337 /* Assume the buffer was written, update the timestamp. */
3338 ml_timestamp(buf);
Bram Moolenaar292ad192005-12-11 21:29:51 +00003339 if (append)
3340 buf->b_flags &= ~BF_NEW;
3341 else
3342 buf->b_flags &= ~BF_WRITE_MASK;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003343 }
Bram Moolenaar292ad192005-12-11 21:29:51 +00003344 if (reset_changed && buf->b_changed && !append
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00003345 && (overwriting || vim_strchr(p_cpo, CPO_PLUS) != NULL))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003346 /* Buffer still changed, the autocommands didn't work
3347 * properly. */
3348 return FAIL;
3349 return OK;
3350 }
3351#ifdef FEAT_EVAL
3352 if (!aborting())
3353#endif
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003354 emsg(_("E203: Autocommands deleted or unloaded buffer to be written"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003355 return FAIL;
3356 }
3357
3358 /*
3359 * The autocommands may have changed the number of lines in the file.
3360 * When writing the whole file, adjust the end.
3361 * When writing part of the file, assume that the autocommands only
3362 * changed the number of lines that are to be written (tricky!).
3363 */
3364 if (buf->b_ml.ml_line_count != old_line_count)
3365 {
3366 if (whole) /* write all */
3367 end = buf->b_ml.ml_line_count;
3368 else if (buf->b_ml.ml_line_count > old_line_count) /* more lines */
3369 end += buf->b_ml.ml_line_count - old_line_count;
3370 else /* less lines */
3371 {
3372 end -= old_line_count - buf->b_ml.ml_line_count;
3373 if (end < start)
3374 {
3375 --no_wait_return;
3376 msg_scroll = msg_save;
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01003377 emsg(_("E204: Autocommand changed number of lines in unexpected way"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003378 return FAIL;
3379 }
3380 }
3381 }
3382
3383 /*
3384 * The autocommands may have changed the name of the buffer, which may
3385 * be kept in fname, ffname and sfname.
3386 */
3387 if (buf_ffname)
3388 ffname = buf->b_ffname;
3389 if (buf_sfname)
3390 sfname = buf->b_sfname;
3391 if (buf_fname_f)
3392 fname = buf->b_ffname;
3393 if (buf_fname_s)
3394 fname = buf->b_sfname;
3395 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003396
3397#ifdef FEAT_NETBEANS_INTG
Bram Moolenaarb26e6322010-05-22 21:34:09 +02003398 if (netbeans_active() && isNetbeansBuffer(buf))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003399 {
3400 if (whole)
3401 {
3402 /*
3403 * b_changed can be 0 after an undo, but we still need to write
3404 * the buffer to NetBeans.
3405 */
3406 if (buf->b_changed || isNetbeansModified(buf))
3407 {
Bram Moolenaar009b2592004-10-24 19:18:58 +00003408 --no_wait_return; /* may wait for return now */
3409 msg_scroll = msg_save;
3410 netbeans_save_buffer(buf); /* no error checking... */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003411 return retval;
3412 }
3413 else
3414 {
3415 errnum = (char_u *)"E656: ";
Bram Moolenaared0e7452008-06-27 19:17:34 +00003416 errmsg = (char_u *)_("NetBeans disallows writes of unmodified buffers");
Bram Moolenaar071d4272004-06-13 20:20:40 +00003417 buffer = NULL;
3418 goto fail;
3419 }
3420 }
3421 else
3422 {
3423 errnum = (char_u *)"E657: ";
3424 errmsg = (char_u *)_("Partial writes disallowed for NetBeans buffers");
3425 buffer = NULL;
3426 goto fail;
3427 }
3428 }
3429#endif
3430
3431 if (shortmess(SHM_OVER) && !exiting)
3432 msg_scroll = FALSE; /* overwrite previous file message */
3433 else
3434 msg_scroll = TRUE; /* don't overwrite previous file message */
3435 if (!filtering)
3436 filemess(buf,
3437#ifndef UNIX
3438 sfname,
3439#else
3440 fname,
3441#endif
3442 (char_u *)"", 0); /* show that we are busy */
3443 msg_scroll = FALSE; /* always overwrite the file message now */
3444
3445 buffer = alloc(BUFSIZE);
3446 if (buffer == NULL) /* can't allocate big buffer, use small
3447 * one (to be able to write when out of
3448 * memory) */
3449 {
3450 buffer = smallbuf;
3451 bufsize = SMBUFSIZE;
3452 }
3453 else
3454 bufsize = BUFSIZE;
3455
3456 /*
3457 * Get information about original file (if there is one).
3458 */
Bram Moolenaar53076832015-12-31 19:53:21 +01003459#if defined(UNIX)
Bram Moolenaar6f192452007-11-08 19:49:02 +00003460 st_old.st_dev = 0;
3461 st_old.st_ino = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003462 perm = -1;
3463 if (mch_stat((char *)fname, &st_old) < 0)
3464 newfile = TRUE;
3465 else
3466 {
3467 perm = st_old.st_mode;
3468 if (!S_ISREG(st_old.st_mode)) /* not a file */
3469 {
3470 if (S_ISDIR(st_old.st_mode))
3471 {
3472 errnum = (char_u *)"E502: ";
3473 errmsg = (char_u *)_("is a directory");
3474 goto fail;
3475 }
3476 if (mch_nodetype(fname) != NODE_WRITABLE)
3477 {
3478 errnum = (char_u *)"E503: ";
3479 errmsg = (char_u *)_("is not a file or writable device");
3480 goto fail;
3481 }
3482 /* It's a device of some kind (or a fifo) which we can write to
3483 * but for which we can't make a backup. */
3484 device = TRUE;
3485 newfile = TRUE;
3486 perm = -1;
3487 }
3488 }
3489#else /* !UNIX */
3490 /*
3491 * Check for a writable device name.
3492 */
3493 c = mch_nodetype(fname);
3494 if (c == NODE_OTHER)
3495 {
3496 errnum = (char_u *)"E503: ";
3497 errmsg = (char_u *)_("is not a file or writable device");
3498 goto fail;
3499 }
3500 if (c == NODE_WRITABLE)
3501 {
Bram Moolenaar48e330a2016-02-23 14:53:34 +01003502# if defined(MSWIN)
Bram Moolenaar043545e2006-10-10 16:44:07 +00003503 /* MS-Windows allows opening a device, but we will probably get stuck
3504 * trying to write to it. */
3505 if (!p_odev)
3506 {
3507 errnum = (char_u *)"E796: ";
3508 errmsg = (char_u *)_("writing to device disabled with 'opendevice' option");
3509 goto fail;
3510 }
3511# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003512 device = TRUE;
3513 newfile = TRUE;
3514 perm = -1;
3515 }
3516 else
3517 {
3518 perm = mch_getperm(fname);
3519 if (perm < 0)
3520 newfile = TRUE;
3521 else if (mch_isdir(fname))
3522 {
3523 errnum = (char_u *)"E502: ";
3524 errmsg = (char_u *)_("is a directory");
3525 goto fail;
3526 }
3527 if (overwriting)
3528 (void)mch_stat((char *)fname, &st_old);
3529 }
3530#endif /* !UNIX */
3531
3532 if (!device && !newfile)
3533 {
3534 /*
3535 * Check if the file is really writable (when renaming the file to
3536 * make a backup we won't discover it later).
3537 */
Bram Moolenaar5386a122007-06-28 20:02:32 +00003538 file_readonly = check_file_readonly(fname, (int)perm);
3539
Bram Moolenaar071d4272004-06-13 20:20:40 +00003540 if (!forceit && file_readonly)
3541 {
3542 if (vim_strchr(p_cpo, CPO_FWRITE) != NULL)
3543 {
3544 errnum = (char_u *)"E504: ";
3545 errmsg = (char_u *)_(err_readonly);
3546 }
3547 else
3548 {
3549 errnum = (char_u *)"E505: ";
3550 errmsg = (char_u *)_("is read-only (add ! to override)");
3551 }
3552 goto fail;
3553 }
3554
3555 /*
3556 * Check if the timestamp hasn't changed since reading the file.
3557 */
3558 if (overwriting)
3559 {
3560 retval = check_mtime(buf, &st_old);
3561 if (retval == FAIL)
3562 goto fail;
3563 }
3564 }
3565
3566#ifdef HAVE_ACL
3567 /*
3568 * For systems that support ACL: get the ACL from the original file.
3569 */
3570 if (!newfile)
3571 acl = mch_get_acl(fname);
3572#endif
3573
3574 /*
3575 * If 'backupskip' is not empty, don't make a backup for some files.
3576 */
3577 dobackup = (p_wb || p_bk || *p_pm != NUL);
3578#ifdef FEAT_WILDIGN
3579 if (dobackup && *p_bsk != NUL && match_file_list(p_bsk, sfname, ffname))
3580 dobackup = FALSE;
3581#endif
3582
3583 /*
3584 * Save the value of got_int and reset it. We don't want a previous
3585 * interruption cancel writing, only hitting CTRL-C while writing should
3586 * abort it.
3587 */
3588 prev_got_int = got_int;
3589 got_int = FALSE;
3590
3591 /* Mark the buffer as 'being saved' to prevent changed buffer warnings */
3592 buf->b_saving = TRUE;
3593
3594 /*
3595 * If we are not appending or filtering, the file exists, and the
3596 * 'writebackup', 'backup' or 'patchmode' option is set, need a backup.
3597 * When 'patchmode' is set also make a backup when appending.
3598 *
3599 * Do not make any backup, if 'writebackup' and 'backup' are both switched
3600 * off. This helps when editing large files on almost-full disks.
3601 */
3602 if (!(append && *p_pm == NUL) && !filtering && perm >= 0 && dobackup)
3603 {
3604#if defined(UNIX) || defined(WIN32)
Bram Moolenaar8767f522016-07-01 17:17:39 +02003605 stat_T st;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003606#endif
3607
Bram Moolenaarb8ee25a2014-09-23 15:45:08 +02003608 if ((bkc & BKC_YES) || append) /* "yes" */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003609 backup_copy = TRUE;
3610#if defined(UNIX) || defined(WIN32)
Bram Moolenaarb8ee25a2014-09-23 15:45:08 +02003611 else if ((bkc & BKC_AUTO)) /* "auto" */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003612 {
3613 int i;
3614
3615# ifdef UNIX
3616 /*
3617 * Don't rename the file when:
3618 * - it's a hard link
3619 * - it's a symbolic link
3620 * - we don't have write permission in the directory
3621 * - we can't set the owner/group of the new file
3622 */
3623 if (st_old.st_nlink > 1
3624 || mch_lstat((char *)fname, &st) < 0
3625 || st.st_dev != st_old.st_dev
Bram Moolenaara5792f52005-11-23 21:25:05 +00003626 || st.st_ino != st_old.st_ino
3627# ifndef HAVE_FCHOWN
3628 || st.st_uid != st_old.st_uid
3629 || st.st_gid != st_old.st_gid
3630# endif
3631 )
Bram Moolenaar071d4272004-06-13 20:20:40 +00003632 backup_copy = TRUE;
3633 else
Bram Moolenaar03f48552006-02-28 23:52:23 +00003634# else
3635# ifdef WIN32
3636 /* On NTFS file systems hard links are possible. */
3637 if (mch_is_linked(fname))
3638 backup_copy = TRUE;
3639 else
3640# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003641# endif
3642 {
3643 /*
3644 * Check if we can create a file and set the owner/group to
3645 * the ones from the original file.
3646 * First find a file name that doesn't exist yet (use some
3647 * arbitrary numbers).
3648 */
3649 STRCPY(IObuff, fname);
3650 for (i = 4913; ; i += 123)
3651 {
3652 sprintf((char *)gettail(IObuff), "%d", i);
Bram Moolenaara5792f52005-11-23 21:25:05 +00003653 if (mch_lstat((char *)IObuff, &st) < 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003654 break;
3655 }
Bram Moolenaara5792f52005-11-23 21:25:05 +00003656 fd = mch_open((char *)IObuff,
3657 O_CREAT|O_WRONLY|O_EXCL|O_NOFOLLOW, perm);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003658 if (fd < 0) /* can't write in directory */
3659 backup_copy = TRUE;
3660 else
3661 {
3662# ifdef UNIX
Bram Moolenaara5792f52005-11-23 21:25:05 +00003663# ifdef HAVE_FCHOWN
Bram Moolenaar42335f52018-09-13 15:33:43 +02003664 vim_ignored = fchown(fd, st_old.st_uid, st_old.st_gid);
Bram Moolenaara5792f52005-11-23 21:25:05 +00003665# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003666 if (mch_stat((char *)IObuff, &st) < 0
3667 || st.st_uid != st_old.st_uid
3668 || st.st_gid != st_old.st_gid
Bram Moolenaar78a15312009-05-15 19:33:18 +00003669 || (long)st.st_mode != perm)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003670 backup_copy = TRUE;
3671# endif
Bram Moolenaar98358622005-11-28 22:58:23 +00003672 /* Close the file before removing it, on MS-Windows we
3673 * can't delete an open file. */
Bram Moolenaara5792f52005-11-23 21:25:05 +00003674 close(fd);
Bram Moolenaar98358622005-11-28 22:58:23 +00003675 mch_remove(IObuff);
Bram Moolenaar3479c5d2010-08-08 18:46:06 +02003676# ifdef MSWIN
3677 /* MS-Windows may trigger a virus scanner to open the
3678 * file, we can't delete it then. Keep trying for half a
3679 * second. */
3680 {
3681 int try;
3682
3683 for (try = 0; try < 10; ++try)
3684 {
3685 if (mch_lstat((char *)IObuff, &st) < 0)
3686 break;
3687 ui_delay(50L, TRUE); /* wait 50 msec */
3688 mch_remove(IObuff);
3689 }
3690 }
3691# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003692 }
3693 }
3694 }
3695
Bram Moolenaar071d4272004-06-13 20:20:40 +00003696 /*
3697 * Break symlinks and/or hardlinks if we've been asked to.
3698 */
Bram Moolenaarb8ee25a2014-09-23 15:45:08 +02003699 if ((bkc & BKC_BREAKSYMLINK) || (bkc & BKC_BREAKHARDLINK))
Bram Moolenaar071d4272004-06-13 20:20:40 +00003700 {
Bram Moolenaar12b559e2013-06-12 22:41:37 +02003701# ifdef UNIX
Bram Moolenaar071d4272004-06-13 20:20:40 +00003702 int lstat_res;
3703
3704 lstat_res = mch_lstat((char *)fname, &st);
3705
3706 /* Symlinks. */
Bram Moolenaarb8ee25a2014-09-23 15:45:08 +02003707 if ((bkc & BKC_BREAKSYMLINK)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003708 && lstat_res == 0
3709 && st.st_ino != st_old.st_ino)
3710 backup_copy = FALSE;
3711
3712 /* Hardlinks. */
Bram Moolenaarb8ee25a2014-09-23 15:45:08 +02003713 if ((bkc & BKC_BREAKHARDLINK)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003714 && st_old.st_nlink > 1
3715 && (lstat_res != 0 || st.st_ino == st_old.st_ino))
3716 backup_copy = FALSE;
Bram Moolenaar12b559e2013-06-12 22:41:37 +02003717# else
3718# if defined(WIN32)
3719 /* Symlinks. */
Bram Moolenaarb8ee25a2014-09-23 15:45:08 +02003720 if ((bkc & BKC_BREAKSYMLINK) && mch_is_symbolic_link(fname))
Bram Moolenaar12b559e2013-06-12 22:41:37 +02003721 backup_copy = FALSE;
3722
3723 /* Hardlinks. */
Bram Moolenaarb8ee25a2014-09-23 15:45:08 +02003724 if ((bkc & BKC_BREAKHARDLINK) && mch_is_hard_link(fname))
Bram Moolenaar12b559e2013-06-12 22:41:37 +02003725 backup_copy = FALSE;
3726# endif
3727# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003728 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00003729
3730#endif
3731
3732 /* make sure we have a valid backup extension to use */
3733 if (*p_bex == NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003734 backup_ext = (char_u *)".bak";
Bram Moolenaar071d4272004-06-13 20:20:40 +00003735 else
3736 backup_ext = p_bex;
3737
3738 if (backup_copy
3739 && (fd = mch_open((char *)fname, O_RDONLY | O_EXTRA, 0)) >= 0)
3740 {
3741 int bfd;
3742 char_u *copybuf, *wp;
3743 int some_error = FALSE;
Bram Moolenaar8767f522016-07-01 17:17:39 +02003744 stat_T st_new;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003745 char_u *dirp;
3746 char_u *rootname;
Bram Moolenaarb782ba42018-08-07 21:39:28 +02003747#if defined(UNIX) || defined(WIN3264)
3748 char_u *p;
3749#endif
Bram Moolenaar48e330a2016-02-23 14:53:34 +01003750#if defined(UNIX)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003751 int did_set_shortname;
Bram Moolenaarcd142e32017-11-16 17:03:45 +01003752 mode_t umask_save;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003753#endif
3754
3755 copybuf = alloc(BUFSIZE + 1);
3756 if (copybuf == NULL)
3757 {
3758 some_error = TRUE; /* out of memory */
3759 goto nobackup;
3760 }
3761
3762 /*
3763 * Try to make the backup in each directory in the 'bdir' option.
3764 *
3765 * Unix semantics has it, that we may have a writable file,
3766 * that cannot be recreated with a simple open(..., O_CREAT, ) e.g:
3767 * - the directory is not writable,
3768 * - the file may be a symbolic link,
3769 * - the file may belong to another user/group, etc.
3770 *
3771 * For these reasons, the existing writable file must be truncated
3772 * and reused. Creation of a backup COPY will be attempted.
3773 */
3774 dirp = p_bdir;
3775 while (*dirp)
3776 {
3777#ifdef UNIX
3778 st_new.st_ino = 0;
3779 st_new.st_dev = 0;
3780 st_new.st_gid = 0;
3781#endif
3782
3783 /*
3784 * Isolate one directory name, using an entry in 'bdir'.
3785 */
3786 (void)copy_option_part(&dirp, copybuf, BUFSIZE, ",");
Bram Moolenaarb782ba42018-08-07 21:39:28 +02003787
3788#if defined(UNIX) || defined(WIN3264)
3789 p = copybuf + STRLEN(copybuf);
3790 if (after_pathsep(copybuf, p) && p[-1] == p[-2])
3791 // Ends with '//', use full path
3792 if ((p = make_percent_swname(copybuf, fname)) != NULL)
3793 {
3794 backup = modname(p, backup_ext, FALSE);
3795 vim_free(p);
3796 }
3797#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003798 rootname = get_file_in_dir(fname, copybuf);
3799 if (rootname == NULL)
3800 {
3801 some_error = TRUE; /* out of memory */
3802 goto nobackup;
3803 }
3804
Bram Moolenaar48e330a2016-02-23 14:53:34 +01003805#if defined(UNIX)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003806 did_set_shortname = FALSE;
3807#endif
3808
3809 /*
3810 * May try twice if 'shortname' not set.
3811 */
3812 for (;;)
3813 {
3814 /*
Bram Moolenaarb782ba42018-08-07 21:39:28 +02003815 * Make the backup file name.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003816 */
Bram Moolenaarb782ba42018-08-07 21:39:28 +02003817 if (backup == NULL)
3818 backup = buf_modname((buf->b_p_sn || buf->b_shortname),
Bram Moolenaar071d4272004-06-13 20:20:40 +00003819 rootname, backup_ext, FALSE);
3820 if (backup == NULL)
3821 {
3822 vim_free(rootname);
3823 some_error = TRUE; /* out of memory */
3824 goto nobackup;
3825 }
3826
3827 /*
3828 * Check if backup file already exists.
3829 */
3830 if (mch_stat((char *)backup, &st_new) >= 0)
3831 {
3832#ifdef UNIX
3833 /*
3834 * Check if backup file is same as original file.
3835 * May happen when modname() gave the same file back.
3836 * E.g. silly link, or file name-length reached.
3837 * If we don't check here, we either ruin the file
3838 * when copying or erase it after writing. jw.
3839 */
3840 if (st_new.st_dev == st_old.st_dev
3841 && st_new.st_ino == st_old.st_ino)
3842 {
Bram Moolenaard23a8232018-02-10 18:45:26 +01003843 VIM_CLEAR(backup); /* no backup file to delete */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003844 /*
3845 * may try again with 'shortname' set
3846 */
3847 if (!(buf->b_shortname || buf->b_p_sn))
3848 {
3849 buf->b_shortname = TRUE;
3850 did_set_shortname = TRUE;
3851 continue;
3852 }
3853 /* setting shortname didn't help */
3854 if (did_set_shortname)
3855 buf->b_shortname = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003856 break;
3857 }
3858#endif
3859
3860 /*
3861 * If we are not going to keep the backup file, don't
3862 * delete an existing one, try to use another name.
3863 * Change one character, just before the extension.
3864 */
3865 if (!p_bk)
3866 {
3867 wp = backup + STRLEN(backup) - 1
3868 - STRLEN(backup_ext);
3869 if (wp < backup) /* empty file name ??? */
3870 wp = backup;
3871 *wp = 'z';
3872 while (*wp > 'a'
3873 && mch_stat((char *)backup, &st_new) >= 0)
3874 --*wp;
3875 /* They all exist??? Must be something wrong. */
3876 if (*wp == 'a')
Bram Moolenaard23a8232018-02-10 18:45:26 +01003877 VIM_CLEAR(backup);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003878 }
3879 }
3880 break;
3881 }
3882 vim_free(rootname);
3883
3884 /*
3885 * Try to create the backup file
3886 */
3887 if (backup != NULL)
3888 {
3889 /* remove old backup, if present */
3890 mch_remove(backup);
3891 /* Open with O_EXCL to avoid the file being created while
Bram Moolenaarcd142e32017-11-16 17:03:45 +01003892 * we were sleeping (symlink hacker attack?). Reset umask
3893 * if possible to avoid mch_setperm() below. */
3894#ifdef UNIX
3895 umask_save = umask(0);
3896#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003897 bfd = mch_open((char *)backup,
Bram Moolenaara5792f52005-11-23 21:25:05 +00003898 O_WRONLY|O_CREAT|O_EXTRA|O_EXCL|O_NOFOLLOW,
3899 perm & 0777);
Bram Moolenaarcd142e32017-11-16 17:03:45 +01003900#ifdef UNIX
3901 (void)umask(umask_save);
3902#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003903 if (bfd < 0)
Bram Moolenaard23a8232018-02-10 18:45:26 +01003904 VIM_CLEAR(backup);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003905 else
3906 {
Bram Moolenaarcd142e32017-11-16 17:03:45 +01003907 /* Set file protection same as original file, but
3908 * strip s-bit. Only needed if umask() wasn't used
3909 * above. */
3910#ifndef UNIX
Bram Moolenaar071d4272004-06-13 20:20:40 +00003911 (void)mch_setperm(backup, perm & 0777);
Bram Moolenaarcd142e32017-11-16 17:03:45 +01003912#else
Bram Moolenaar071d4272004-06-13 20:20:40 +00003913 /*
3914 * Try to set the group of the backup same as the
3915 * original file. If this fails, set the protection
3916 * bits for the group same as the protection bits for
3917 * others.
3918 */
Bram Moolenaara5792f52005-11-23 21:25:05 +00003919 if (st_new.st_gid != st_old.st_gid
Bram Moolenaar071d4272004-06-13 20:20:40 +00003920# ifdef HAVE_FCHOWN /* sequent-ptx lacks fchown() */
Bram Moolenaara5792f52005-11-23 21:25:05 +00003921 && fchown(bfd, (uid_t)-1, st_old.st_gid) != 0
Bram Moolenaar071d4272004-06-13 20:20:40 +00003922# endif
3923 )
3924 mch_setperm(backup,
3925 (perm & 0707) | ((perm & 07) << 3));
Bram Moolenaar5bd32f42014-04-02 14:05:38 +02003926# if defined(HAVE_SELINUX) || defined(HAVE_SMACK)
Bram Moolenaar588ebeb2008-05-07 17:09:24 +00003927 mch_copy_sec(fname, backup);
3928# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003929#endif
3930
3931 /*
3932 * copy the file.
3933 */
3934 write_info.bw_fd = bfd;
3935 write_info.bw_buf = copybuf;
3936#ifdef HAS_BW_FLAGS
3937 write_info.bw_flags = FIO_NOCONVERT;
3938#endif
Bram Moolenaar540fc6f2010-12-17 16:27:16 +01003939 while ((write_info.bw_len = read_eintr(fd, copybuf,
Bram Moolenaar071d4272004-06-13 20:20:40 +00003940 BUFSIZE)) > 0)
3941 {
3942 if (buf_write_bytes(&write_info) == FAIL)
3943 {
3944 errmsg = (char_u *)_("E506: Can't write to backup file (add ! to override)");
3945 break;
3946 }
3947 ui_breakcheck();
3948 if (got_int)
3949 {
3950 errmsg = (char_u *)_(e_interr);
3951 break;
3952 }
3953 }
3954
3955 if (close(bfd) < 0 && errmsg == NULL)
3956 errmsg = (char_u *)_("E507: Close error for backup file (add ! to override)");
3957 if (write_info.bw_len < 0)
3958 errmsg = (char_u *)_("E508: Can't read file for backup (add ! to override)");
3959#ifdef UNIX
3960 set_file_time(backup, st_old.st_atime, st_old.st_mtime);
3961#endif
3962#ifdef HAVE_ACL
3963 mch_set_acl(backup, acl);
3964#endif
Bram Moolenaar5bd32f42014-04-02 14:05:38 +02003965#if defined(HAVE_SELINUX) || defined(HAVE_SMACK)
Bram Moolenaar588ebeb2008-05-07 17:09:24 +00003966 mch_copy_sec(fname, backup);
3967#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003968 break;
3969 }
3970 }
3971 }
3972 nobackup:
3973 close(fd); /* ignore errors for closing read file */
3974 vim_free(copybuf);
3975
3976 if (backup == NULL && errmsg == NULL)
3977 errmsg = (char_u *)_("E509: Cannot create backup file (add ! to override)");
3978 /* ignore errors when forceit is TRUE */
3979 if ((some_error || errmsg != NULL) && !forceit)
3980 {
3981 retval = FAIL;
3982 goto fail;
3983 }
3984 errmsg = NULL;
3985 }
3986 else
3987 {
3988 char_u *dirp;
3989 char_u *p;
3990 char_u *rootname;
3991
3992 /*
3993 * Make a backup by renaming the original file.
3994 */
3995 /*
3996 * If 'cpoptions' includes the "W" flag, we don't want to
3997 * overwrite a read-only file. But rename may be possible
3998 * anyway, thus we need an extra check here.
3999 */
4000 if (file_readonly && vim_strchr(p_cpo, CPO_FWRITE) != NULL)
4001 {
4002 errnum = (char_u *)"E504: ";
4003 errmsg = (char_u *)_(err_readonly);
4004 goto fail;
4005 }
4006
4007 /*
4008 *
4009 * Form the backup file name - change path/fo.o.h to
4010 * path/fo.o.h.bak Try all directories in 'backupdir', first one
4011 * that works is used.
4012 */
4013 dirp = p_bdir;
4014 while (*dirp)
4015 {
4016 /*
4017 * Isolate one directory name and make the backup file name.
4018 */
4019 (void)copy_option_part(&dirp, IObuff, IOSIZE, ",");
Bram Moolenaarb782ba42018-08-07 21:39:28 +02004020
4021#if defined(UNIX) || defined(WIN3264)
4022 p = IObuff + STRLEN(IObuff);
4023 if (after_pathsep(IObuff, p) && p[-1] == p[-2])
4024 // path ends with '//', use full path
4025 if ((p = make_percent_swname(IObuff, fname)) != NULL)
4026 {
4027 backup = modname(p, backup_ext, FALSE);
4028 vim_free(p);
4029 }
4030#endif
4031 if (backup == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004032 {
Bram Moolenaarb782ba42018-08-07 21:39:28 +02004033 rootname = get_file_in_dir(fname, IObuff);
4034 if (rootname == NULL)
4035 backup = NULL;
4036 else
4037 {
4038 backup = buf_modname(
4039 (buf->b_p_sn || buf->b_shortname),
4040 rootname, backup_ext, FALSE);
4041 vim_free(rootname);
4042 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004043 }
4044
4045 if (backup != NULL)
4046 {
4047 /*
4048 * If we are not going to keep the backup file, don't
4049 * delete an existing one, try to use another name.
4050 * Change one character, just before the extension.
4051 */
4052 if (!p_bk && mch_getperm(backup) >= 0)
4053 {
4054 p = backup + STRLEN(backup) - 1 - STRLEN(backup_ext);
4055 if (p < backup) /* empty file name ??? */
4056 p = backup;
4057 *p = 'z';
4058 while (*p > 'a' && mch_getperm(backup) >= 0)
4059 --*p;
4060 /* They all exist??? Must be something wrong! */
4061 if (*p == 'a')
Bram Moolenaard23a8232018-02-10 18:45:26 +01004062 VIM_CLEAR(backup);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004063 }
4064 }
4065 if (backup != NULL)
4066 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00004067 /*
Bram Moolenaarbfd8fc02005-09-20 23:22:24 +00004068 * Delete any existing backup and move the current version
4069 * to the backup. For safety, we don't remove the backup
4070 * until the write has finished successfully. And if the
4071 * 'backup' option is set, leave it around.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004072 */
4073 /*
4074 * If the renaming of the original file to the backup file
4075 * works, quit here.
4076 */
4077 if (vim_rename(fname, backup) == 0)
4078 break;
4079
Bram Moolenaard23a8232018-02-10 18:45:26 +01004080 VIM_CLEAR(backup); /* don't do the rename below */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004081 }
4082 }
4083 if (backup == NULL && !forceit)
4084 {
4085 errmsg = (char_u *)_("E510: Can't make backup file (add ! to override)");
4086 goto fail;
4087 }
4088 }
4089 }
4090
Bram Moolenaar53076832015-12-31 19:53:21 +01004091#if defined(UNIX)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004092 /* When using ":w!" and the file was read-only: make it writable */
4093 if (forceit && perm >= 0 && !(perm & 0200) && st_old.st_uid == getuid()
4094 && vim_strchr(p_cpo, CPO_FWRITE) == NULL)
4095 {
4096 perm |= 0200;
4097 (void)mch_setperm(fname, perm);
4098 made_writable = TRUE;
4099 }
4100#endif
4101
Bram Moolenaar910f66f2006-04-05 20:41:53 +00004102 /* When using ":w!" and writing to the current file, 'readonly' makes no
Bram Moolenaar4399ef42005-02-12 14:29:27 +00004103 * sense, reset it, unless 'Z' appears in 'cpoptions'. */
4104 if (forceit && overwriting && vim_strchr(p_cpo, CPO_KEEPRO) == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004105 {
4106 buf->b_p_ro = FALSE;
4107#ifdef FEAT_TITLE
4108 need_maketitle = TRUE; /* set window title later */
4109#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004110 status_redraw_all(); /* redraw status lines later */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004111 }
4112
4113 if (end > buf->b_ml.ml_line_count)
4114 end = buf->b_ml.ml_line_count;
4115 if (buf->b_ml.ml_flags & ML_EMPTY)
4116 start = end + 1;
4117
4118 /*
4119 * If the original file is being overwritten, there is a small chance that
4120 * we crash in the middle of writing. Therefore the file is preserved now.
4121 * This makes all block numbers positive so that recovery does not need
4122 * the original file.
4123 * Don't do this if there is a backup file and we are exiting.
4124 */
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00004125 if (reset_changed && !newfile && overwriting
Bram Moolenaar071d4272004-06-13 20:20:40 +00004126 && !(exiting && backup != NULL))
4127 {
4128 ml_preserve(buf, FALSE);
4129 if (got_int)
4130 {
4131 errmsg = (char_u *)_(e_interr);
4132 goto restore_backup;
4133 }
4134 }
4135
Bram Moolenaar071d4272004-06-13 20:20:40 +00004136#ifdef VMS
4137 vms_remove_version(fname); /* remove version */
4138#endif
Bram Moolenaarb23a7e82008-06-27 18:42:32 +00004139 /* Default: write the file directly. May write to a temp file for
Bram Moolenaar071d4272004-06-13 20:20:40 +00004140 * multi-byte conversion. */
4141 wfname = fname;
4142
Bram Moolenaar071d4272004-06-13 20:20:40 +00004143 /* Check for forced 'fileencoding' from "++opt=val" argument. */
4144 if (eap != NULL && eap->force_enc != 0)
4145 {
4146 fenc = eap->cmd + eap->force_enc;
4147 fenc = enc_canonize(fenc);
4148 fenc_tofree = fenc;
4149 }
4150 else
4151 fenc = buf->b_p_fenc;
4152
4153 /*
Bram Moolenaarb5cdf2e2009-07-29 16:25:31 +00004154 * Check if the file needs to be converted.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004155 */
Bram Moolenaarb5cdf2e2009-07-29 16:25:31 +00004156 converted = need_conversion(fenc);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004157
4158 /*
4159 * Check if UTF-8 to UCS-2/4 or Latin1 conversion needs to be done. Or
4160 * Latin1 to Unicode conversion. This is handled in buf_write_bytes().
4161 * Prepare the flags for it and allocate bw_conv_buf when needed.
4162 */
4163 if (converted && (enc_utf8 || STRCMP(p_enc, "latin1") == 0))
4164 {
4165 wb_flags = get_fio_flags(fenc);
4166 if (wb_flags & (FIO_UCS2 | FIO_UCS4 | FIO_UTF16 | FIO_UTF8))
4167 {
4168 /* Need to allocate a buffer to translate into. */
4169 if (wb_flags & (FIO_UCS2 | FIO_UTF16 | FIO_UTF8))
4170 write_info.bw_conv_buflen = bufsize * 2;
4171 else /* FIO_UCS4 */
4172 write_info.bw_conv_buflen = bufsize * 4;
4173 write_info.bw_conv_buf
4174 = lalloc((long_u)write_info.bw_conv_buflen, TRUE);
4175 if (write_info.bw_conv_buf == NULL)
4176 end = 0;
4177 }
4178 }
4179
Bram Moolenaar13505972019-01-24 15:04:48 +01004180#ifdef WIN3264
Bram Moolenaar071d4272004-06-13 20:20:40 +00004181 if (converted && wb_flags == 0 && (wb_flags = get_win_fio_flags(fenc)) != 0)
4182 {
4183 /* Convert UTF-8 -> UCS-2 and UCS-2 -> DBCS. Worst-case * 4: */
4184 write_info.bw_conv_buflen = bufsize * 4;
4185 write_info.bw_conv_buf
4186 = lalloc((long_u)write_info.bw_conv_buflen, TRUE);
4187 if (write_info.bw_conv_buf == NULL)
4188 end = 0;
4189 }
Bram Moolenaar13505972019-01-24 15:04:48 +01004190#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004191
Bram Moolenaar13505972019-01-24 15:04:48 +01004192#ifdef MACOS_CONVERT
Bram Moolenaar071d4272004-06-13 20:20:40 +00004193 if (converted && wb_flags == 0 && (wb_flags = get_mac_fio_flags(fenc)) != 0)
4194 {
4195 write_info.bw_conv_buflen = bufsize * 3;
4196 write_info.bw_conv_buf
4197 = lalloc((long_u)write_info.bw_conv_buflen, TRUE);
4198 if (write_info.bw_conv_buf == NULL)
4199 end = 0;
4200 }
Bram Moolenaar13505972019-01-24 15:04:48 +01004201#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004202
Bram Moolenaar13505972019-01-24 15:04:48 +01004203#if defined(FEAT_EVAL) || defined(USE_ICONV)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004204 if (converted && wb_flags == 0)
4205 {
Bram Moolenaar13505972019-01-24 15:04:48 +01004206# ifdef USE_ICONV
Bram Moolenaar071d4272004-06-13 20:20:40 +00004207 /*
4208 * Use iconv() conversion when conversion is needed and it's not done
4209 * internally.
4210 */
4211 write_info.bw_iconv_fd = (iconv_t)my_iconv_open(fenc,
4212 enc_utf8 ? (char_u *)"utf-8" : p_enc);
4213 if (write_info.bw_iconv_fd != (iconv_t)-1)
4214 {
4215 /* We're going to use iconv(), allocate a buffer to convert in. */
4216 write_info.bw_conv_buflen = bufsize * ICONV_MULT;
4217 write_info.bw_conv_buf
4218 = lalloc((long_u)write_info.bw_conv_buflen, TRUE);
4219 if (write_info.bw_conv_buf == NULL)
4220 end = 0;
4221 write_info.bw_first = TRUE;
4222 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004223# ifdef FEAT_EVAL
Bram Moolenaar13505972019-01-24 15:04:48 +01004224 else
4225# endif
4226# endif
4227
4228# ifdef FEAT_EVAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00004229 /*
4230 * When the file needs to be converted with 'charconvert' after
4231 * writing, write to a temp file instead and let the conversion
4232 * overwrite the original file.
4233 */
4234 if (*p_ccv != NUL)
4235 {
Bram Moolenaare5c421c2015-03-31 13:33:08 +02004236 wfname = vim_tempname('w', FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004237 if (wfname == NULL) /* Can't write without a tempfile! */
4238 {
4239 errmsg = (char_u *)_("E214: Can't find temp file for writing");
4240 goto restore_backup;
4241 }
4242 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004243# endif
Bram Moolenaar13505972019-01-24 15:04:48 +01004244 }
4245#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004246 if (converted && wb_flags == 0
Bram Moolenaar13505972019-01-24 15:04:48 +01004247#ifdef USE_ICONV
Bram Moolenaar071d4272004-06-13 20:20:40 +00004248 && write_info.bw_iconv_fd == (iconv_t)-1
Bram Moolenaar13505972019-01-24 15:04:48 +01004249# endif
4250# ifdef FEAT_EVAL
Bram Moolenaar071d4272004-06-13 20:20:40 +00004251 && wfname == fname
Bram Moolenaar13505972019-01-24 15:04:48 +01004252# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004253 )
4254 {
4255 if (!forceit)
4256 {
4257 errmsg = (char_u *)_("E213: Cannot convert (add ! to write without conversion)");
4258 goto restore_backup;
4259 }
4260 notconverted = TRUE;
4261 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004262
4263 /*
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004264 * If conversion is taking place, we may first pretend to write and check
4265 * for conversion errors. Then loop again to write for real.
4266 * When not doing conversion this writes for real right away.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004267 */
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004268 for (checking_conversion = TRUE; ; checking_conversion = FALSE)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004269 {
4270 /*
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004271 * There is no need to check conversion when:
4272 * - there is no conversion
4273 * - we make a backup file, that can be restored in case of conversion
4274 * failure.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004275 */
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004276 if (!converted || dobackup)
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004277 checking_conversion = FALSE;
4278
4279 if (checking_conversion)
4280 {
4281 /* Make sure we don't write anything. */
4282 fd = -1;
4283 write_info.bw_fd = fd;
4284 }
4285 else
4286 {
Bram Moolenaarcd142e32017-11-16 17:03:45 +01004287#ifdef HAVE_FTRUNCATE
4288# define TRUNC_ON_OPEN 0
4289#else
4290# define TRUNC_ON_OPEN O_TRUNC
4291#endif
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004292 /*
4293 * Open the file "wfname" for writing.
4294 * We may try to open the file twice: If we can't write to the file
4295 * and forceit is TRUE we delete the existing file and try to
4296 * create a new one. If this still fails we may have lost the
4297 * original file! (this may happen when the user reached his
4298 * quotum for number of files).
4299 * Appending will fail if the file does not exist and forceit is
4300 * FALSE.
4301 */
4302 while ((fd = mch_open((char *)wfname, O_WRONLY | O_EXTRA | (append
4303 ? (forceit ? (O_APPEND | O_CREAT) : O_APPEND)
Bram Moolenaarcd142e32017-11-16 17:03:45 +01004304 : (O_CREAT | TRUNC_ON_OPEN))
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004305 , perm < 0 ? 0666 : (perm & 0777))) < 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004306 {
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004307 /*
4308 * A forced write will try to create a new file if the old one
4309 * is still readonly. This may also happen when the directory
4310 * is read-only. In that case the mch_remove() will fail.
4311 */
4312 if (errmsg == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004313 {
4314#ifdef UNIX
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004315 stat_T st;
4316
4317 /* Don't delete the file when it's a hard or symbolic link.
4318 */
4319 if ((!newfile && st_old.st_nlink > 1)
4320 || (mch_lstat((char *)fname, &st) == 0
4321 && (st.st_dev != st_old.st_dev
4322 || st.st_ino != st_old.st_ino)))
4323 errmsg = (char_u *)_("E166: Can't open linked file for writing");
4324 else
Bram Moolenaar071d4272004-06-13 20:20:40 +00004325#endif
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004326 {
4327 errmsg = (char_u *)_("E212: Can't open file for writing");
4328 if (forceit && vim_strchr(p_cpo, CPO_FWRITE) == NULL
4329 && perm >= 0)
4330 {
4331#ifdef UNIX
4332 /* we write to the file, thus it should be marked
4333 writable after all */
4334 if (!(perm & 0200))
4335 made_writable = TRUE;
4336 perm |= 0200;
4337 if (st_old.st_uid != getuid()
4338 || st_old.st_gid != getgid())
4339 perm &= 0777;
4340#endif
4341 if (!append) /* don't remove when appending */
4342 mch_remove(wfname);
4343 continue;
4344 }
4345 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004346 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004347
4348restore_backup:
Bram Moolenaar071d4272004-06-13 20:20:40 +00004349 {
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004350 stat_T st;
4351
Bram Moolenaar071d4272004-06-13 20:20:40 +00004352 /*
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004353 * If we failed to open the file, we don't need a backup.
4354 * Throw it away. If we moved or removed the original file
4355 * try to put the backup in its place.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004356 */
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004357 if (backup != NULL && wfname == fname)
4358 {
4359 if (backup_copy)
4360 {
4361 /*
4362 * There is a small chance that we removed the
4363 * original, try to move the copy in its place.
4364 * This may not work if the vim_rename() fails.
4365 * In that case we leave the copy around.
4366 */
4367 /* If file does not exist, put the copy in its
4368 * place */
4369 if (mch_stat((char *)fname, &st) < 0)
4370 vim_rename(backup, fname);
4371 /* if original file does exist throw away the copy
4372 */
4373 if (mch_stat((char *)fname, &st) >= 0)
4374 mch_remove(backup);
4375 }
4376 else
4377 {
4378 /* try to put the original file back */
4379 vim_rename(backup, fname);
4380 }
4381 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004382
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004383 /* if original file no longer exists give an extra warning
4384 */
4385 if (!newfile && mch_stat((char *)fname, &st) < 0)
4386 end = 0;
4387 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004388
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004389 if (wfname != fname)
4390 vim_free(wfname);
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004391 goto fail;
4392 }
4393 write_info.bw_fd = fd;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004394
Bram Moolenaarcd142e32017-11-16 17:03:45 +01004395#if defined(UNIX)
4396 {
4397 stat_T st;
4398
4399 /* Double check we are writing the intended file before making
4400 * any changes. */
4401 if (overwriting
4402 && (!dobackup || backup_copy)
4403 && fname == wfname
4404 && perm >= 0
4405 && mch_fstat(fd, &st) == 0
4406 && st.st_ino != st_old.st_ino)
4407 {
4408 close(fd);
4409 errmsg = (char_u *)_("E949: File changed while writing");
4410 goto fail;
4411 }
4412 }
4413#endif
4414#ifdef HAVE_FTRUNCATE
4415 if (!append)
Bram Moolenaar42335f52018-09-13 15:33:43 +02004416 vim_ignored = ftruncate(fd, (off_t)0);
Bram Moolenaarcd142e32017-11-16 17:03:45 +01004417#endif
4418
Bram Moolenaard0573012017-10-28 21:11:06 +02004419#if defined(WIN3264)
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004420 if (backup != NULL && overwriting && !append)
4421 {
4422 if (backup_copy)
4423 (void)mch_copy_file_attribute(wfname, backup);
4424 else
4425 (void)mch_copy_file_attribute(backup, wfname);
4426 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004427
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004428 if (!overwriting && !append)
4429 {
4430 if (buf->b_ffname != NULL)
4431 (void)mch_copy_file_attribute(buf->b_ffname, wfname);
4432 /* Should copy resource fork */
4433 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004434#endif
4435
Bram Moolenaar071d4272004-06-13 20:20:40 +00004436#ifdef FEAT_CRYPT
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004437 if (*buf->b_p_key != NUL && !filtering)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004438 {
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004439 char_u *header;
4440 int header_len;
4441
4442 buf->b_cryptstate = crypt_create_for_writing(
4443 crypt_get_method_nr(buf),
4444 buf->b_p_key, &header, &header_len);
4445 if (buf->b_cryptstate == NULL || header == NULL)
4446 end = 0;
4447 else
Bram Moolenaar071d4272004-06-13 20:20:40 +00004448 {
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004449 /* Write magic number, so that Vim knows how this file is
4450 * encrypted when reading it back. */
4451 write_info.bw_buf = header;
4452 write_info.bw_len = header_len;
4453 write_info.bw_flags = FIO_NOCONVERT;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004454 if (buf_write_bytes(&write_info) == FAIL)
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004455 end = 0;
4456 wb_flags |= FIO_ENCRYPTED;
4457 vim_free(header);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004458 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004459 }
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004460#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004461 }
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004462 errmsg = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004463
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004464 write_info.bw_buf = buffer;
4465 nchars = 0;
4466
4467 /* use "++bin", "++nobin" or 'binary' */
4468 if (eap != NULL && eap->force_bin != 0)
4469 write_bin = (eap->force_bin == FORCE_BIN);
4470 else
4471 write_bin = buf->b_p_bin;
4472
Bram Moolenaar071d4272004-06-13 20:20:40 +00004473 /*
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004474 * The BOM is written just after the encryption magic number.
4475 * Skip it when appending and the file already existed, the BOM only
4476 * makes sense at the start of the file.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004477 */
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004478 if (buf->b_p_bomb && !write_bin && (!append || perm < 0))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004479 {
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004480 write_info.bw_len = make_bom(buffer, fenc);
4481 if (write_info.bw_len > 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004482 {
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004483 /* don't convert, do encryption */
4484 write_info.bw_flags = FIO_NOCONVERT | wb_flags;
4485 if (buf_write_bytes(&write_info) == FAIL)
4486 end = 0;
4487 else
4488 nchars += write_info.bw_len;
4489 }
4490 }
4491 write_info.bw_start_lnum = start;
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004492
4493#ifdef FEAT_PERSISTENT_UNDO
4494 write_undo_file = (buf->b_p_udf
4495 && overwriting
4496 && !append
4497 && !filtering
4498 && reset_changed
4499 && !checking_conversion);
4500 if (write_undo_file)
4501 /* Prepare for computing the hash value of the text. */
4502 sha256_start(&sha_ctx);
4503#endif
4504
4505 write_info.bw_len = bufsize;
4506#ifdef HAS_BW_FLAGS
4507 write_info.bw_flags = wb_flags;
4508#endif
4509 fileformat = get_fileformat_force(buf, eap);
4510 s = buffer;
4511 len = 0;
4512 for (lnum = start; lnum <= end; ++lnum)
4513 {
4514 /*
4515 * The next while loop is done once for each character written.
4516 * Keep it fast!
4517 */
4518 ptr = ml_get_buf(buf, lnum, FALSE) - 1;
4519#ifdef FEAT_PERSISTENT_UNDO
4520 if (write_undo_file)
4521 sha256_update(&sha_ctx, ptr + 1,
4522 (UINT32_T)(STRLEN(ptr + 1) + 1));
4523#endif
4524 while ((c = *++ptr) != NUL)
4525 {
4526 if (c == NL)
4527 *s = NUL; /* replace newlines with NULs */
4528 else if (c == CAR && fileformat == EOL_MAC)
4529 *s = NL; /* Mac: replace CRs with NLs */
4530 else
4531 *s = c;
4532 ++s;
4533 if (++len != bufsize)
4534 continue;
Bram Moolenaard4755bb2004-09-02 19:12:26 +00004535 if (buf_write_bytes(&write_info) == FAIL)
4536 {
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004537 end = 0; /* write error: break loop */
Bram Moolenaard4755bb2004-09-02 19:12:26 +00004538 break;
4539 }
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004540 nchars += bufsize;
4541 s = buffer;
4542 len = 0;
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004543 write_info.bw_start_lnum = lnum;
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004544 }
4545 /* write failed or last line has no EOL: stop here */
4546 if (end == 0
4547 || (lnum == end
4548 && (write_bin || !buf->b_p_fixeol)
4549 && (lnum == buf->b_no_eol_lnum
4550 || (lnum == buf->b_ml.ml_line_count
4551 && !buf->b_p_eol))))
4552 {
4553 ++lnum; /* written the line, count it */
4554 no_eol = TRUE;
4555 break;
4556 }
4557 if (fileformat == EOL_UNIX)
4558 *s++ = NL;
4559 else
4560 {
4561 *s++ = CAR; /* EOL_MAC or EOL_DOS: write CR */
4562 if (fileformat == EOL_DOS) /* write CR-NL */
4563 {
4564 if (++len == bufsize)
4565 {
4566 if (buf_write_bytes(&write_info) == FAIL)
4567 {
4568 end = 0; /* write error: break loop */
4569 break;
4570 }
4571 nchars += bufsize;
4572 s = buffer;
4573 len = 0;
4574 }
4575 *s++ = NL;
4576 }
4577 }
4578 if (++len == bufsize && end)
4579 {
4580 if (buf_write_bytes(&write_info) == FAIL)
4581 {
4582 end = 0; /* write error: break loop */
4583 break;
4584 }
4585 nchars += bufsize;
4586 s = buffer;
4587 len = 0;
4588
4589 ui_breakcheck();
4590 if (got_int)
4591 {
4592 end = 0; /* Interrupted, break loop */
4593 break;
4594 }
4595 }
4596#ifdef VMS
4597 /*
4598 * On VMS there is a problem: newlines get added when writing
4599 * blocks at a time. Fix it by writing a line at a time.
4600 * This is much slower!
4601 * Explanation: VAX/DECC RTL insists that records in some RMS
4602 * structures end with a newline (carriage return) character, and
4603 * if they don't it adds one.
4604 * With other RMS structures it works perfect without this fix.
4605 */
4606 if (buf->b_fab_rfm == FAB$C_VFC
4607 || ((buf->b_fab_rat & (FAB$M_FTN | FAB$M_CR)) != 0))
4608 {
4609 int b2write;
4610
4611 buf->b_fab_mrs = (buf->b_fab_mrs == 0
4612 ? MIN(4096, bufsize)
4613 : MIN(buf->b_fab_mrs, bufsize));
4614
4615 b2write = len;
4616 while (b2write > 0)
4617 {
4618 write_info.bw_len = MIN(b2write, buf->b_fab_mrs);
4619 if (buf_write_bytes(&write_info) == FAIL)
4620 {
4621 end = 0;
4622 break;
4623 }
4624 b2write -= MIN(b2write, buf->b_fab_mrs);
4625 }
4626 write_info.bw_len = bufsize;
4627 nchars += len;
4628 s = buffer;
4629 len = 0;
4630 }
4631#endif
4632 }
4633 if (len > 0 && end > 0)
4634 {
4635 write_info.bw_len = len;
4636 if (buf_write_bytes(&write_info) == FAIL)
4637 end = 0; /* write error */
4638 nchars += len;
4639 }
4640
4641 /* Stop when writing done or an error was encountered. */
4642 if (!checking_conversion || end == 0)
4643 break;
4644
4645 /* If no error happened until now, writing should be ok, so loop to
4646 * really write the buffer. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004647 }
4648
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004649 /* If we started writing, finish writing. Also when an error was
4650 * encountered. */
4651 if (!checking_conversion)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004652 {
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004653#if defined(UNIX) && defined(HAVE_FSYNC)
4654 /*
4655 * On many journalling file systems there is a bug that causes both the
4656 * original and the backup file to be lost when halting the system
4657 * right after writing the file. That's because only the meta-data is
4658 * journalled. Syncing the file slows down the system, but assures it
4659 * has been written to disk and we don't lose it.
4660 * For a device do try the fsync() but don't complain if it does not
4661 * work (could be a pipe).
4662 * If the 'fsync' option is FALSE, don't fsync(). Useful for laptops.
4663 */
4664 if (p_fs && fsync(fd) != 0 && !device)
4665 {
Bram Moolenaar7567d0b2017-11-16 23:04:15 +01004666 errmsg = (char_u *)_(e_fsync);
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004667 end = 0;
4668 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004669#endif
4670
Bram Moolenaar5bd32f42014-04-02 14:05:38 +02004671#if defined(HAVE_SELINUX) || defined(HAVE_SMACK)
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004672 /* Probably need to set the security context. */
4673 if (!backup_copy)
4674 mch_copy_sec(backup, wfname);
Bram Moolenaar588ebeb2008-05-07 17:09:24 +00004675#endif
4676
Bram Moolenaara5792f52005-11-23 21:25:05 +00004677#ifdef UNIX
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004678 /* When creating a new file, set its owner/group to that of the
4679 * original file. Get the new device and inode number. */
4680 if (backup != NULL && !backup_copy)
Bram Moolenaara5792f52005-11-23 21:25:05 +00004681 {
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004682# ifdef HAVE_FCHOWN
4683 stat_T st;
4684
Bram Moolenaarcd142e32017-11-16 17:03:45 +01004685 /* Don't change the owner when it's already OK, some systems remove
4686 * permission or ACL stuff. */
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004687 if (mch_stat((char *)wfname, &st) < 0
4688 || st.st_uid != st_old.st_uid
4689 || st.st_gid != st_old.st_gid)
4690 {
Bram Moolenaarcd142e32017-11-16 17:03:45 +01004691 /* changing owner might not be possible */
Bram Moolenaar42335f52018-09-13 15:33:43 +02004692 vim_ignored = fchown(fd, st_old.st_uid, -1);
Bram Moolenaarcd142e32017-11-16 17:03:45 +01004693 /* if changing group fails clear the group permissions */
4694 if (fchown(fd, -1, st_old.st_gid) == -1 && perm > 0)
4695 perm &= ~070;
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004696 }
Bram Moolenaara5792f52005-11-23 21:25:05 +00004697# endif
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004698 buf_setino(buf);
4699 }
4700 else if (!buf->b_dev_valid)
4701 /* Set the inode when creating a new file. */
4702 buf_setino(buf);
Bram Moolenaara5792f52005-11-23 21:25:05 +00004703#endif
4704
Bram Moolenaarcd142e32017-11-16 17:03:45 +01004705#ifdef UNIX
4706 if (made_writable)
4707 perm &= ~0200; /* reset 'w' bit for security reasons */
4708#endif
4709#ifdef HAVE_FCHMOD
4710 /* set permission of new file same as old file */
4711 if (perm >= 0)
4712 (void)mch_fsetperm(fd, perm);
4713#endif
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004714 if (close(fd) != 0)
4715 {
4716 errmsg = (char_u *)_("E512: Close failed");
4717 end = 0;
4718 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004719
Bram Moolenaarcd142e32017-11-16 17:03:45 +01004720#ifndef HAVE_FCHMOD
4721 /* set permission of new file same as old file */
4722 if (perm >= 0)
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004723 (void)mch_setperm(wfname, perm);
Bram Moolenaarcd142e32017-11-16 17:03:45 +01004724#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004725#ifdef HAVE_ACL
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004726 /*
4727 * Probably need to set the ACL before changing the user (can't set the
4728 * ACL on a file the user doesn't own).
4729 * On Solaris, with ZFS and the aclmode property set to "discard" (the
4730 * default), chmod() discards all part of a file's ACL that don't
4731 * represent the mode of the file. It's non-trivial for us to discover
4732 * whether we're in that situation, so we simply always re-set the ACL.
4733 */
Bram Moolenaarda412772016-07-14 20:37:07 +02004734# ifndef HAVE_SOLARIS_ZFS_ACL
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004735 if (!backup_copy)
Bram Moolenaarda412772016-07-14 20:37:07 +02004736# endif
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004737 mch_set_acl(wfname, acl);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004738#endif
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02004739#ifdef FEAT_CRYPT
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004740 if (buf->b_cryptstate != NULL)
4741 {
4742 crypt_free_state(buf->b_cryptstate);
4743 buf->b_cryptstate = NULL;
4744 }
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02004745#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004746
Bram Moolenaar13505972019-01-24 15:04:48 +01004747#if defined(FEAT_EVAL)
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004748 if (wfname != fname)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004749 {
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004750 /*
4751 * The file was written to a temp file, now it needs to be
4752 * converted with 'charconvert' to (overwrite) the output file.
4753 */
4754 if (end != 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004755 {
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004756 if (eval_charconvert(enc_utf8 ? (char_u *)"utf-8" : p_enc,
4757 fenc, wfname, fname) == FAIL)
4758 {
4759 write_info.bw_conv_error = TRUE;
4760 end = 0;
4761 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004762 }
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004763 mch_remove(wfname);
4764 vim_free(wfname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004765 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004766#endif
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004767 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004768
4769 if (end == 0)
4770 {
Bram Moolenaare6bf6552017-06-27 22:11:51 +02004771 /*
4772 * Error encountered.
4773 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004774 if (errmsg == NULL)
4775 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00004776 if (write_info.bw_conv_error)
Bram Moolenaar32b485f2009-07-29 16:06:27 +00004777 {
4778 if (write_info.bw_conv_error_lnum == 0)
4779 errmsg = (char_u *)_("E513: write error, conversion failed (make 'fenc' empty to override)");
4780 else
4781 {
4782 errmsg_allocated = TRUE;
4783 errmsg = alloc(300);
4784 vim_snprintf((char *)errmsg, 300, _("E513: write error, conversion failed in line %ld (make 'fenc' empty to override)"),
4785 (long)write_info.bw_conv_error_lnum);
4786 }
4787 }
Bram Moolenaar13505972019-01-24 15:04:48 +01004788 else if (got_int)
4789 errmsg = (char_u *)_(e_interr);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004790 else
Bram Moolenaar13505972019-01-24 15:04:48 +01004791 errmsg = (char_u *)_("E514: write error (file system full?)");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004792 }
4793
4794 /*
4795 * If we have a backup file, try to put it in place of the new file,
Bram Moolenaare37d50a2008-08-06 17:06:04 +00004796 * because the new file is probably corrupt. This avoids losing the
Bram Moolenaar071d4272004-06-13 20:20:40 +00004797 * original file when trying to make a backup when writing the file a
4798 * second time.
4799 * When "backup_copy" is set we need to copy the backup over the new
4800 * file. Otherwise rename the backup file.
4801 * If this is OK, don't give the extra warning message.
4802 */
4803 if (backup != NULL)
4804 {
4805 if (backup_copy)
4806 {
4807 /* This may take a while, if we were interrupted let the user
4808 * know we got the message. */
4809 if (got_int)
4810 {
Bram Moolenaar32526b32019-01-19 17:43:09 +01004811 msg(_(e_interr));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004812 out_flush();
4813 }
4814 if ((fd = mch_open((char *)backup, O_RDONLY | O_EXTRA, 0)) >= 0)
4815 {
4816 if ((write_info.bw_fd = mch_open((char *)fname,
Bram Moolenaar9be038d2005-03-08 22:34:32 +00004817 O_WRONLY | O_CREAT | O_TRUNC | O_EXTRA,
4818 perm & 0777)) >= 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004819 {
4820 /* copy the file. */
4821 write_info.bw_buf = smallbuf;
4822#ifdef HAS_BW_FLAGS
4823 write_info.bw_flags = FIO_NOCONVERT;
4824#endif
Bram Moolenaar540fc6f2010-12-17 16:27:16 +01004825 while ((write_info.bw_len = read_eintr(fd, smallbuf,
Bram Moolenaar071d4272004-06-13 20:20:40 +00004826 SMBUFSIZE)) > 0)
4827 if (buf_write_bytes(&write_info) == FAIL)
4828 break;
4829
4830 if (close(write_info.bw_fd) >= 0
4831 && write_info.bw_len == 0)
4832 end = 1; /* success */
4833 }
4834 close(fd); /* ignore errors for closing read file */
4835 }
4836 }
4837 else
4838 {
4839 if (vim_rename(backup, fname) == 0)
4840 end = 1;
4841 }
4842 }
4843 goto fail;
4844 }
4845
4846 lnum -= start; /* compute number of written lines */
4847 --no_wait_return; /* may wait for return now */
4848
4849#if !(defined(UNIX) || defined(VMS))
4850 fname = sfname; /* use shortname now, for the messages */
4851#endif
4852 if (!filtering)
4853 {
4854 msg_add_fname(buf, fname); /* put fname in IObuff with quotes */
4855 c = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004856 if (write_info.bw_conv_error)
4857 {
4858 STRCAT(IObuff, _(" CONVERSION ERROR"));
4859 c = TRUE;
Bram Moolenaar32b485f2009-07-29 16:06:27 +00004860 if (write_info.bw_conv_error_lnum != 0)
Bram Moolenaara800b422010-06-27 01:15:55 +02004861 vim_snprintf_add((char *)IObuff, IOSIZE, _(" in line %ld;"),
Bram Moolenaar32b485f2009-07-29 16:06:27 +00004862 (long)write_info.bw_conv_error_lnum);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004863 }
4864 else if (notconverted)
4865 {
4866 STRCAT(IObuff, _("[NOT converted]"));
4867 c = TRUE;
4868 }
4869 else if (converted)
4870 {
4871 STRCAT(IObuff, _("[converted]"));
4872 c = TRUE;
4873 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004874 if (device)
4875 {
4876 STRCAT(IObuff, _("[Device]"));
4877 c = TRUE;
4878 }
4879 else if (newfile)
4880 {
4881 STRCAT(IObuff, shortmess(SHM_NEW) ? _("[New]") : _("[New File]"));
4882 c = TRUE;
4883 }
4884 if (no_eol)
4885 {
4886 msg_add_eol();
4887 c = TRUE;
4888 }
4889 /* may add [unix/dos/mac] */
4890 if (msg_add_fileformat(fileformat))
4891 c = TRUE;
4892#ifdef FEAT_CRYPT
4893 if (wb_flags & FIO_ENCRYPTED)
4894 {
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02004895 crypt_append_msg(buf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004896 c = TRUE;
4897 }
4898#endif
4899 msg_add_lines(c, (long)lnum, nchars); /* add line/char count */
4900 if (!shortmess(SHM_WRITE))
4901 {
4902 if (append)
4903 STRCAT(IObuff, shortmess(SHM_WRI) ? _(" [a]") : _(" appended"));
4904 else
4905 STRCAT(IObuff, shortmess(SHM_WRI) ? _(" [w]") : _(" written"));
4906 }
4907
Bram Moolenaar32526b32019-01-19 17:43:09 +01004908 set_keep_msg((char_u *)msg_trunc_attr((char *)IObuff, FALSE, 0), 0);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004909 }
4910
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00004911 /* When written everything correctly: reset 'modified'. Unless not
4912 * writing to the original file and '+' is not in 'cpoptions'. */
Bram Moolenaar292ad192005-12-11 21:29:51 +00004913 if (reset_changed && whole && !append
Bram Moolenaar071d4272004-06-13 20:20:40 +00004914 && !write_info.bw_conv_error
Bram Moolenaar13505972019-01-24 15:04:48 +01004915 && (overwriting || vim_strchr(p_cpo, CPO_PLUS) != NULL))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004916 {
4917 unchanged(buf, TRUE);
Bram Moolenaar95c526e2017-02-25 14:59:34 +01004918 /* b:changedtick is always incremented in unchanged() but that
Bram Moolenaar086329d2014-10-31 19:51:36 +01004919 * should not trigger a TextChanged event. */
Bram Moolenaar5a093432018-02-10 18:15:19 +01004920 if (buf->b_last_changedtick + 1 == CHANGEDTICK(buf))
4921 buf->b_last_changedtick = CHANGEDTICK(buf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004922 u_unchanged(buf);
Bram Moolenaar730cde92010-06-27 05:18:54 +02004923 u_update_save_nr(buf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004924 }
4925
4926 /*
4927 * If written to the current file, update the timestamp of the swap file
4928 * and reset the BF_WRITE_MASK flags. Also sets buf->b_mtime.
4929 */
4930 if (overwriting)
4931 {
4932 ml_timestamp(buf);
Bram Moolenaar292ad192005-12-11 21:29:51 +00004933 if (append)
4934 buf->b_flags &= ~BF_NEW;
4935 else
4936 buf->b_flags &= ~BF_WRITE_MASK;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004937 }
4938
4939 /*
4940 * If we kept a backup until now, and we are in patch mode, then we make
4941 * the backup file our 'original' file.
4942 */
4943 if (*p_pm && dobackup)
4944 {
Bram Moolenaar48e330a2016-02-23 14:53:34 +01004945 char *org = (char *)buf_modname((buf->b_p_sn || buf->b_shortname),
Bram Moolenaar071d4272004-06-13 20:20:40 +00004946 fname, p_pm, FALSE);
4947
4948 if (backup != NULL)
4949 {
Bram Moolenaar8767f522016-07-01 17:17:39 +02004950 stat_T st;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004951
4952 /*
4953 * If the original file does not exist yet
4954 * the current backup file becomes the original file
4955 */
4956 if (org == NULL)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01004957 emsg(_("E205: Patchmode: can't save original file"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004958 else if (mch_stat(org, &st) < 0)
4959 {
4960 vim_rename(backup, (char_u *)org);
Bram Moolenaard23a8232018-02-10 18:45:26 +01004961 VIM_CLEAR(backup); /* don't delete the file */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004962#ifdef UNIX
4963 set_file_time((char_u *)org, st_old.st_atime, st_old.st_mtime);
4964#endif
4965 }
4966 }
4967 /*
4968 * If there is no backup file, remember that a (new) file was
4969 * created.
4970 */
4971 else
4972 {
4973 int empty_fd;
4974
4975 if (org == NULL
Bram Moolenaara5792f52005-11-23 21:25:05 +00004976 || (empty_fd = mch_open(org,
4977 O_CREAT | O_EXTRA | O_EXCL | O_NOFOLLOW,
Bram Moolenaar4317d9b2005-03-18 20:25:31 +00004978 perm < 0 ? 0666 : (perm & 0777))) < 0)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01004979 emsg(_("E206: patchmode: can't touch empty original file"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004980 else
4981 close(empty_fd);
4982 }
4983 if (org != NULL)
4984 {
4985 mch_setperm((char_u *)org, mch_getperm(fname) & 0777);
4986 vim_free(org);
4987 }
4988 }
4989
4990 /*
4991 * Remove the backup unless 'backup' option is set
4992 */
4993 if (!p_bk && backup != NULL && mch_remove(backup) != 0)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01004994 emsg(_("E207: Can't delete backup file"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00004995
Bram Moolenaar071d4272004-06-13 20:20:40 +00004996 goto nofail;
4997
4998 /*
4999 * Finish up. We get here either after failure or success.
5000 */
5001fail:
5002 --no_wait_return; /* may wait for return now */
5003nofail:
5004
5005 /* Done saving, we accept changed buffer warnings again */
5006 buf->b_saving = FALSE;
5007
5008 vim_free(backup);
5009 if (buffer != smallbuf)
5010 vim_free(buffer);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005011 vim_free(fenc_tofree);
5012 vim_free(write_info.bw_conv_buf);
Bram Moolenaar13505972019-01-24 15:04:48 +01005013#ifdef USE_ICONV
Bram Moolenaar071d4272004-06-13 20:20:40 +00005014 if (write_info.bw_iconv_fd != (iconv_t)-1)
5015 {
5016 iconv_close(write_info.bw_iconv_fd);
5017 write_info.bw_iconv_fd = (iconv_t)-1;
5018 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00005019#endif
5020#ifdef HAVE_ACL
5021 mch_free_acl(acl);
5022#endif
5023
5024 if (errmsg != NULL)
5025 {
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00005026 int numlen = errnum != NULL ? (int)STRLEN(errnum) : 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005027
Bram Moolenaar8820b482017-03-16 17:23:31 +01005028 attr = HL_ATTR(HLF_E); /* set highlight for error messages */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005029 msg_add_fname(buf,
5030#ifndef UNIX
5031 sfname
5032#else
5033 fname
5034#endif
5035 ); /* put file name in IObuff with quotes */
5036 if (STRLEN(IObuff) + STRLEN(errmsg) + numlen >= IOSIZE)
5037 IObuff[IOSIZE - STRLEN(errmsg) - numlen - 1] = NUL;
5038 /* If the error message has the form "is ...", put the error number in
5039 * front of the file name. */
5040 if (errnum != NULL)
5041 {
Bram Moolenaar3577c6f2008-06-24 21:16:56 +00005042 STRMOVE(IObuff + numlen, IObuff);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005043 mch_memmove(IObuff, errnum, (size_t)numlen);
5044 }
5045 STRCAT(IObuff, errmsg);
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01005046 emsg((char *)IObuff);
Bram Moolenaar32b485f2009-07-29 16:06:27 +00005047 if (errmsg_allocated)
5048 vim_free(errmsg);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005049
5050 retval = FAIL;
5051 if (end == 0)
5052 {
Bram Moolenaar32526b32019-01-19 17:43:09 +01005053 msg_puts_attr(_("\nWARNING: Original file may be lost or damaged\n"),
Bram Moolenaar071d4272004-06-13 20:20:40 +00005054 attr | MSG_HIST);
Bram Moolenaar32526b32019-01-19 17:43:09 +01005055 msg_puts_attr(_("don't quit the editor until the file is successfully written!"),
Bram Moolenaar071d4272004-06-13 20:20:40 +00005056 attr | MSG_HIST);
5057
5058 /* Update the timestamp to avoid an "overwrite changed file"
5059 * prompt when writing again. */
5060 if (mch_stat((char *)fname, &st_old) >= 0)
5061 {
5062 buf_store_time(buf, &st_old, fname);
5063 buf->b_mtime_read = buf->b_mtime;
5064 }
5065 }
5066 }
5067 msg_scroll = msg_save;
5068
Bram Moolenaar55debbe2010-05-23 23:34:36 +02005069#ifdef FEAT_PERSISTENT_UNDO
5070 /*
5071 * When writing the whole file and 'undofile' is set, also write the undo
5072 * file.
5073 */
5074 if (retval == OK && write_undo_file)
5075 {
5076 char_u hash[UNDO_HASH_SIZE];
5077
5078 sha256_finish(&sha_ctx, hash);
5079 u_write_undo(NULL, FALSE, buf, hash);
5080 }
5081#endif
5082
Bram Moolenaar071d4272004-06-13 20:20:40 +00005083#ifdef FEAT_EVAL
5084 if (!should_abort(retval))
5085#else
5086 if (!got_int)
5087#endif
5088 {
5089 aco_save_T aco;
5090
Bram Moolenaar68a33fc2012-04-25 16:50:48 +02005091 curbuf->b_no_eol_lnum = 0; /* in case it was set by the previous read */
5092
Bram Moolenaar071d4272004-06-13 20:20:40 +00005093 /*
5094 * Apply POST autocommands.
5095 * Careful: The autocommands may call buf_write() recursively!
5096 */
5097 aucmd_prepbuf(&aco, buf);
5098
5099 if (append)
5100 apply_autocmds_exarg(EVENT_FILEAPPENDPOST, fname, fname,
5101 FALSE, curbuf, eap);
5102 else if (filtering)
5103 apply_autocmds_exarg(EVENT_FILTERWRITEPOST, NULL, fname,
5104 FALSE, curbuf, eap);
5105 else if (reset_changed && whole)
5106 apply_autocmds_exarg(EVENT_BUFWRITEPOST, fname, fname,
5107 FALSE, curbuf, eap);
5108 else
5109 apply_autocmds_exarg(EVENT_FILEWRITEPOST, fname, fname,
5110 FALSE, curbuf, eap);
5111
5112 /* restore curwin/curbuf and a few other things */
5113 aucmd_restbuf(&aco);
5114
5115#ifdef FEAT_EVAL
5116 if (aborting()) /* autocmds may abort script processing */
5117 retval = FALSE;
5118#endif
5119 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00005120
5121 got_int |= prev_got_int;
5122
Bram Moolenaar071d4272004-06-13 20:20:40 +00005123 return retval;
5124}
5125
5126/*
Bram Moolenaar2d3f4892006-01-20 23:02:51 +00005127 * Set the name of the current buffer. Use when the buffer doesn't have a
5128 * name and a ":r" or ":w" command with a file name is used.
5129 */
5130 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005131set_rw_fname(char_u *fname, char_u *sfname)
Bram Moolenaar2d3f4892006-01-20 23:02:51 +00005132{
Bram Moolenaar8b38e242009-06-16 13:35:20 +00005133 buf_T *buf = curbuf;
5134
Bram Moolenaar2d3f4892006-01-20 23:02:51 +00005135 /* It's like the unnamed buffer is deleted.... */
5136 if (curbuf->b_p_bl)
5137 apply_autocmds(EVENT_BUFDELETE, NULL, NULL, FALSE, curbuf);
5138 apply_autocmds(EVENT_BUFWIPEOUT, NULL, NULL, FALSE, curbuf);
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +01005139#ifdef FEAT_EVAL
Bram Moolenaar2d3f4892006-01-20 23:02:51 +00005140 if (aborting()) /* autocmds may abort script processing */
5141 return FAIL;
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +01005142#endif
Bram Moolenaar8b38e242009-06-16 13:35:20 +00005143 if (curbuf != buf)
5144 {
5145 /* We are in another buffer now, don't do the renaming. */
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01005146 emsg(_(e_auchangedbuf));
Bram Moolenaar8b38e242009-06-16 13:35:20 +00005147 return FAIL;
5148 }
Bram Moolenaar2d3f4892006-01-20 23:02:51 +00005149
5150 if (setfname(curbuf, fname, sfname, FALSE) == OK)
5151 curbuf->b_flags |= BF_NOTEDITED;
5152
Bram Moolenaar2d3f4892006-01-20 23:02:51 +00005153 /* ....and a new named one is created */
5154 apply_autocmds(EVENT_BUFNEW, NULL, NULL, FALSE, curbuf);
5155 if (curbuf->b_p_bl)
5156 apply_autocmds(EVENT_BUFADD, NULL, NULL, FALSE, curbuf);
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +01005157#ifdef FEAT_EVAL
Bram Moolenaar2d3f4892006-01-20 23:02:51 +00005158 if (aborting()) /* autocmds may abort script processing */
5159 return FAIL;
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +01005160#endif
Bram Moolenaar2d3f4892006-01-20 23:02:51 +00005161
5162 /* Do filetype detection now if 'filetype' is empty. */
5163 if (*curbuf->b_p_ft == NUL)
5164 {
Bram Moolenaar910f66f2006-04-05 20:41:53 +00005165 if (au_has_group((char_u *)"filetypedetect"))
Bram Moolenaar1610d052016-06-09 22:53:01 +02005166 (void)do_doautocmd((char_u *)"filetypedetect BufRead", FALSE, NULL);
Bram Moolenaara3227e22006-03-08 21:32:40 +00005167 do_modelines(0);
Bram Moolenaar2d3f4892006-01-20 23:02:51 +00005168 }
Bram Moolenaar2d3f4892006-01-20 23:02:51 +00005169
5170 return OK;
5171}
5172
5173/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00005174 * Put file name into IObuff with quotes.
5175 */
Bram Moolenaar009b2592004-10-24 19:18:58 +00005176 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005177msg_add_fname(buf_T *buf, char_u *fname)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005178{
5179 if (fname == NULL)
5180 fname = (char_u *)"-stdin-";
5181 home_replace(buf, fname, IObuff + 1, IOSIZE - 4, TRUE);
5182 IObuff[0] = '"';
5183 STRCAT(IObuff, "\" ");
5184}
5185
5186/*
5187 * Append message for text mode to IObuff.
5188 * Return TRUE if something appended.
5189 */
5190 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005191msg_add_fileformat(int eol_type)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005192{
5193#ifndef USE_CRNL
5194 if (eol_type == EOL_DOS)
5195 {
5196 STRCAT(IObuff, shortmess(SHM_TEXT) ? _("[dos]") : _("[dos format]"));
5197 return TRUE;
5198 }
5199#endif
5200#ifndef USE_CR
5201 if (eol_type == EOL_MAC)
5202 {
5203 STRCAT(IObuff, shortmess(SHM_TEXT) ? _("[mac]") : _("[mac format]"));
5204 return TRUE;
5205 }
5206#endif
5207#if defined(USE_CRNL) || defined(USE_CR)
5208 if (eol_type == EOL_UNIX)
5209 {
5210 STRCAT(IObuff, shortmess(SHM_TEXT) ? _("[unix]") : _("[unix format]"));
5211 return TRUE;
5212 }
5213#endif
5214 return FALSE;
5215}
5216
5217/*
5218 * Append line and character count to IObuff.
5219 */
Bram Moolenaar009b2592004-10-24 19:18:58 +00005220 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005221msg_add_lines(
5222 int insert_space,
5223 long lnum,
Bram Moolenaar8767f522016-07-01 17:17:39 +02005224 off_T nchars)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005225{
5226 char_u *p;
5227
5228 p = IObuff + STRLEN(IObuff);
5229
5230 if (insert_space)
5231 *p++ = ' ';
5232 if (shortmess(SHM_LINES))
Bram Moolenaarbde98102016-07-01 20:03:42 +02005233 vim_snprintf((char *)p, IOSIZE - (p - IObuff),
Bram Moolenaar88c86eb2019-01-17 17:13:30 +01005234 "%ldL, %lldC", lnum, (long_long_T)nchars);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005235 else
5236 {
Bram Moolenaarda6e8912018-08-21 15:12:14 +02005237 sprintf((char *)p, NGETTEXT("%ld line, ", "%ld lines, ", lnum), lnum);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005238 p += STRLEN(p);
Bram Moolenaarda6e8912018-08-21 15:12:14 +02005239 vim_snprintf((char *)p, IOSIZE - (p - IObuff),
5240 NGETTEXT("%lld character", "%lld characters", nchars),
Bram Moolenaar88c86eb2019-01-17 17:13:30 +01005241 (long_long_T)nchars);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005242 }
5243}
5244
5245/*
5246 * Append message for missing line separator to IObuff.
5247 */
5248 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005249msg_add_eol(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005250{
5251 STRCAT(IObuff, shortmess(SHM_LAST) ? _("[noeol]") : _("[Incomplete last line]"));
5252}
5253
5254/*
5255 * Check modification time of file, before writing to it.
5256 * The size isn't checked, because using a tool like "gzip" takes care of
5257 * using the same timestamp but can't set the size.
5258 */
5259 static int
Bram Moolenaar8767f522016-07-01 17:17:39 +02005260check_mtime(buf_T *buf, stat_T *st)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005261{
5262 if (buf->b_mtime_read != 0
5263 && time_differs((long)st->st_mtime, buf->b_mtime_read))
5264 {
5265 msg_scroll = TRUE; /* don't overwrite messages here */
5266 msg_silent = 0; /* must give this prompt */
5267 /* don't use emsg() here, don't want to flush the buffers */
Bram Moolenaar32526b32019-01-19 17:43:09 +01005268 msg_attr(_("WARNING: The file has been changed since reading it!!!"),
Bram Moolenaar8820b482017-03-16 17:23:31 +01005269 HL_ATTR(HLF_E));
Bram Moolenaar071d4272004-06-13 20:20:40 +00005270 if (ask_yesno((char_u *)_("Do you really want to write to it"),
5271 TRUE) == 'n')
5272 return FAIL;
5273 msg_scroll = FALSE; /* always overwrite the file message now */
5274 }
5275 return OK;
5276}
5277
5278 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005279time_differs(long t1, long t2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005280{
Bram Moolenaar48e330a2016-02-23 14:53:34 +01005281#if defined(__linux__) || defined(MSWIN)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005282 /* On a FAT filesystem, esp. under Linux, there are only 5 bits to store
5283 * the seconds. Since the roundoff is done when flushing the inode, the
5284 * time may change unexpectedly by one second!!! */
5285 return (t1 - t2 > 1 || t2 - t1 > 1);
5286#else
5287 return (t1 != t2);
5288#endif
5289}
5290
5291/*
5292 * Call write() to write a number of bytes to the file.
Bram Moolenaar540fc6f2010-12-17 16:27:16 +01005293 * Handles encryption and 'encoding' conversion.
Bram Moolenaar071d4272004-06-13 20:20:40 +00005294 *
5295 * Return FAIL for failure, OK otherwise.
5296 */
5297 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005298buf_write_bytes(struct bw_info *ip)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005299{
5300 int wlen;
5301 char_u *buf = ip->bw_buf; /* data to write */
5302 int len = ip->bw_len; /* length of data */
5303#ifdef HAS_BW_FLAGS
5304 int flags = ip->bw_flags; /* extra flags */
5305#endif
5306
Bram Moolenaar071d4272004-06-13 20:20:40 +00005307 /*
5308 * Skip conversion when writing the crypt magic number or the BOM.
5309 */
5310 if (!(flags & FIO_NOCONVERT))
5311 {
5312 char_u *p;
5313 unsigned c;
5314 int n;
5315
5316 if (flags & FIO_UTF8)
5317 {
5318 /*
5319 * Convert latin1 in the buffer to UTF-8 in the file.
5320 */
5321 p = ip->bw_conv_buf; /* translate to buffer */
5322 for (wlen = 0; wlen < len; ++wlen)
5323 p += utf_char2bytes(buf[wlen], p);
5324 buf = ip->bw_conv_buf;
5325 len = (int)(p - ip->bw_conv_buf);
5326 }
5327 else if (flags & (FIO_UCS4 | FIO_UTF16 | FIO_UCS2 | FIO_LATIN1))
5328 {
5329 /*
5330 * Convert UTF-8 bytes in the buffer to UCS-2, UCS-4, UTF-16 or
5331 * Latin1 chars in the file.
5332 */
5333 if (flags & FIO_LATIN1)
5334 p = buf; /* translate in-place (can only get shorter) */
5335 else
5336 p = ip->bw_conv_buf; /* translate to buffer */
5337 for (wlen = 0; wlen < len; wlen += n)
5338 {
5339 if (wlen == 0 && ip->bw_restlen != 0)
5340 {
5341 int l;
5342
5343 /* Use remainder of previous call. Append the start of
5344 * buf[] to get a full sequence. Might still be too
5345 * short! */
5346 l = CONV_RESTLEN - ip->bw_restlen;
5347 if (l > len)
5348 l = len;
5349 mch_memmove(ip->bw_rest + ip->bw_restlen, buf, (size_t)l);
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00005350 n = utf_ptr2len_len(ip->bw_rest, ip->bw_restlen + l);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005351 if (n > ip->bw_restlen + len)
5352 {
5353 /* We have an incomplete byte sequence at the end to
5354 * be written. We can't convert it without the
5355 * remaining bytes. Keep them for the next call. */
5356 if (ip->bw_restlen + len > CONV_RESTLEN)
5357 return FAIL;
5358 ip->bw_restlen += len;
5359 break;
5360 }
5361 if (n > 1)
5362 c = utf_ptr2char(ip->bw_rest);
5363 else
5364 c = ip->bw_rest[0];
5365 if (n >= ip->bw_restlen)
5366 {
5367 n -= ip->bw_restlen;
5368 ip->bw_restlen = 0;
5369 }
5370 else
5371 {
5372 ip->bw_restlen -= n;
5373 mch_memmove(ip->bw_rest, ip->bw_rest + n,
5374 (size_t)ip->bw_restlen);
5375 n = 0;
5376 }
5377 }
5378 else
5379 {
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00005380 n = utf_ptr2len_len(buf + wlen, len - wlen);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005381 if (n > len - wlen)
5382 {
5383 /* We have an incomplete byte sequence at the end to
5384 * be written. We can't convert it without the
5385 * remaining bytes. Keep them for the next call. */
5386 if (len - wlen > CONV_RESTLEN)
5387 return FAIL;
5388 ip->bw_restlen = len - wlen;
5389 mch_memmove(ip->bw_rest, buf + wlen,
5390 (size_t)ip->bw_restlen);
5391 break;
5392 }
5393 if (n > 1)
5394 c = utf_ptr2char(buf + wlen);
5395 else
5396 c = buf[wlen];
5397 }
5398
Bram Moolenaar32b485f2009-07-29 16:06:27 +00005399 if (ucs2bytes(c, &p, flags) && !ip->bw_conv_error)
5400 {
5401 ip->bw_conv_error = TRUE;
5402 ip->bw_conv_error_lnum = ip->bw_start_lnum;
5403 }
5404 if (c == NL)
5405 ++ip->bw_start_lnum;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005406 }
5407 if (flags & FIO_LATIN1)
5408 len = (int)(p - buf);
5409 else
5410 {
5411 buf = ip->bw_conv_buf;
5412 len = (int)(p - ip->bw_conv_buf);
5413 }
5414 }
5415
Bram Moolenaar13505972019-01-24 15:04:48 +01005416#ifdef WIN3264
Bram Moolenaar071d4272004-06-13 20:20:40 +00005417 else if (flags & FIO_CODEPAGE)
5418 {
5419 /*
5420 * Convert UTF-8 or codepage to UCS-2 and then to MS-Windows
5421 * codepage.
5422 */
5423 char_u *from;
5424 size_t fromlen;
5425 char_u *to;
5426 int u8c;
5427 BOOL bad = FALSE;
5428 int needed;
5429
5430 if (ip->bw_restlen > 0)
5431 {
5432 /* Need to concatenate the remainder of the previous call and
5433 * the bytes of the current call. Use the end of the
5434 * conversion buffer for this. */
5435 fromlen = len + ip->bw_restlen;
5436 from = ip->bw_conv_buf + ip->bw_conv_buflen - fromlen;
5437 mch_memmove(from, ip->bw_rest, (size_t)ip->bw_restlen);
5438 mch_memmove(from + ip->bw_restlen, buf, (size_t)len);
5439 }
5440 else
5441 {
5442 from = buf;
5443 fromlen = len;
5444 }
5445
5446 to = ip->bw_conv_buf;
5447 if (enc_utf8)
5448 {
5449 /* Convert from UTF-8 to UCS-2, to the start of the buffer.
5450 * The buffer has been allocated to be big enough. */
5451 while (fromlen > 0)
5452 {
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00005453 n = (int)utf_ptr2len_len(from, (int)fromlen);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005454 if (n > (int)fromlen) /* incomplete byte sequence */
5455 break;
5456 u8c = utf_ptr2char(from);
5457 *to++ = (u8c & 0xff);
5458 *to++ = (u8c >> 8);
5459 fromlen -= n;
5460 from += n;
5461 }
5462
5463 /* Copy remainder to ip->bw_rest[] to be used for the next
5464 * call. */
5465 if (fromlen > CONV_RESTLEN)
5466 {
5467 /* weird overlong sequence */
5468 ip->bw_conv_error = TRUE;
5469 return FAIL;
5470 }
5471 mch_memmove(ip->bw_rest, from, fromlen);
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00005472 ip->bw_restlen = (int)fromlen;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005473 }
5474 else
5475 {
5476 /* Convert from enc_codepage to UCS-2, to the start of the
5477 * buffer. The buffer has been allocated to be big enough. */
5478 ip->bw_restlen = 0;
5479 needed = MultiByteToWideChar(enc_codepage,
Bram Moolenaar36f5ac02007-05-06 12:40:34 +00005480 MB_ERR_INVALID_CHARS, (LPCSTR)from, (int)fromlen,
Bram Moolenaar071d4272004-06-13 20:20:40 +00005481 NULL, 0);
5482 if (needed == 0)
5483 {
5484 /* When conversion fails there may be a trailing byte. */
5485 needed = MultiByteToWideChar(enc_codepage,
Bram Moolenaar36f5ac02007-05-06 12:40:34 +00005486 MB_ERR_INVALID_CHARS, (LPCSTR)from, (int)fromlen - 1,
Bram Moolenaar071d4272004-06-13 20:20:40 +00005487 NULL, 0);
5488 if (needed == 0)
5489 {
5490 /* Conversion doesn't work. */
5491 ip->bw_conv_error = TRUE;
5492 return FAIL;
5493 }
5494 /* Save the trailing byte for the next call. */
5495 ip->bw_rest[0] = from[fromlen - 1];
5496 ip->bw_restlen = 1;
5497 }
5498 needed = MultiByteToWideChar(enc_codepage, MB_ERR_INVALID_CHARS,
Bram Moolenaar36f5ac02007-05-06 12:40:34 +00005499 (LPCSTR)from, (int)(fromlen - ip->bw_restlen),
Bram Moolenaar071d4272004-06-13 20:20:40 +00005500 (LPWSTR)to, needed);
5501 if (needed == 0)
5502 {
5503 /* Safety check: Conversion doesn't work. */
5504 ip->bw_conv_error = TRUE;
5505 return FAIL;
5506 }
5507 to += needed * 2;
5508 }
5509
5510 fromlen = to - ip->bw_conv_buf;
5511 buf = to;
Bram Moolenaar13505972019-01-24 15:04:48 +01005512# ifdef CP_UTF8 /* VC 4.1 doesn't define CP_UTF8 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005513 if (FIO_GET_CP(flags) == CP_UTF8)
5514 {
5515 /* Convert from UCS-2 to UTF-8, using the remainder of the
5516 * conversion buffer. Fails when out of space. */
5517 for (from = ip->bw_conv_buf; fromlen > 1; fromlen -= 2)
5518 {
5519 u8c = *from++;
5520 u8c += (*from++ << 8);
5521 to += utf_char2bytes(u8c, to);
5522 if (to + 6 >= ip->bw_conv_buf + ip->bw_conv_buflen)
5523 {
5524 ip->bw_conv_error = TRUE;
5525 return FAIL;
5526 }
5527 }
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00005528 len = (int)(to - buf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005529 }
5530 else
Bram Moolenaar13505972019-01-24 15:04:48 +01005531# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00005532 {
5533 /* Convert from UCS-2 to the codepage, using the remainder of
5534 * the conversion buffer. If the conversion uses the default
5535 * character "0", the data doesn't fit in this encoding, so
5536 * fail. */
5537 len = WideCharToMultiByte(FIO_GET_CP(flags), 0,
5538 (LPCWSTR)ip->bw_conv_buf, (int)fromlen / sizeof(WCHAR),
Bram Moolenaar36f5ac02007-05-06 12:40:34 +00005539 (LPSTR)to, (int)(ip->bw_conv_buflen - fromlen), 0,
5540 &bad);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005541 if (bad)
5542 {
5543 ip->bw_conv_error = TRUE;
5544 return FAIL;
5545 }
5546 }
5547 }
Bram Moolenaar13505972019-01-24 15:04:48 +01005548#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00005549
Bram Moolenaar13505972019-01-24 15:04:48 +01005550#ifdef MACOS_CONVERT
Bram Moolenaar071d4272004-06-13 20:20:40 +00005551 else if (flags & FIO_MACROMAN)
5552 {
5553 /*
5554 * Convert UTF-8 or latin1 to Apple MacRoman.
5555 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005556 char_u *from;
5557 size_t fromlen;
5558
5559 if (ip->bw_restlen > 0)
5560 {
5561 /* Need to concatenate the remainder of the previous call and
5562 * the bytes of the current call. Use the end of the
5563 * conversion buffer for this. */
5564 fromlen = len + ip->bw_restlen;
5565 from = ip->bw_conv_buf + ip->bw_conv_buflen - fromlen;
5566 mch_memmove(from, ip->bw_rest, (size_t)ip->bw_restlen);
5567 mch_memmove(from + ip->bw_restlen, buf, (size_t)len);
5568 }
5569 else
5570 {
5571 from = buf;
5572 fromlen = len;
5573 }
5574
Bram Moolenaarab79bcb2004-07-18 21:34:53 +00005575 if (enc2macroman(from, fromlen,
5576 ip->bw_conv_buf, &len, ip->bw_conv_buflen,
5577 ip->bw_rest, &ip->bw_restlen) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005578 {
5579 ip->bw_conv_error = TRUE;
5580 return FAIL;
5581 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00005582 buf = ip->bw_conv_buf;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005583 }
Bram Moolenaar13505972019-01-24 15:04:48 +01005584#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00005585
Bram Moolenaar13505972019-01-24 15:04:48 +01005586#ifdef USE_ICONV
Bram Moolenaar071d4272004-06-13 20:20:40 +00005587 if (ip->bw_iconv_fd != (iconv_t)-1)
5588 {
5589 const char *from;
5590 size_t fromlen;
5591 char *to;
5592 size_t tolen;
5593
5594 /* Convert with iconv(). */
5595 if (ip->bw_restlen > 0)
5596 {
Bram Moolenaar5d294d12009-03-11 12:11:02 +00005597 char *fp;
5598
Bram Moolenaar071d4272004-06-13 20:20:40 +00005599 /* Need to concatenate the remainder of the previous call and
5600 * the bytes of the current call. Use the end of the
5601 * conversion buffer for this. */
5602 fromlen = len + ip->bw_restlen;
Bram Moolenaar5d294d12009-03-11 12:11:02 +00005603 fp = (char *)ip->bw_conv_buf + ip->bw_conv_buflen - fromlen;
5604 mch_memmove(fp, ip->bw_rest, (size_t)ip->bw_restlen);
5605 mch_memmove(fp + ip->bw_restlen, buf, (size_t)len);
5606 from = fp;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005607 tolen = ip->bw_conv_buflen - fromlen;
5608 }
5609 else
5610 {
5611 from = (const char *)buf;
5612 fromlen = len;
5613 tolen = ip->bw_conv_buflen;
5614 }
5615 to = (char *)ip->bw_conv_buf;
5616
5617 if (ip->bw_first)
5618 {
5619 size_t save_len = tolen;
5620
5621 /* output the initial shift state sequence */
5622 (void)iconv(ip->bw_iconv_fd, NULL, NULL, &to, &tolen);
5623
5624 /* There is a bug in iconv() on Linux (which appears to be
5625 * wide-spread) which sets "to" to NULL and messes up "tolen".
5626 */
5627 if (to == NULL)
5628 {
5629 to = (char *)ip->bw_conv_buf;
5630 tolen = save_len;
5631 }
5632 ip->bw_first = FALSE;
5633 }
5634
5635 /*
5636 * If iconv() has an error or there is not enough room, fail.
5637 */
5638 if ((iconv(ip->bw_iconv_fd, (void *)&from, &fromlen, &to, &tolen)
5639 == (size_t)-1 && ICONV_ERRNO != ICONV_EINVAL)
5640 || fromlen > CONV_RESTLEN)
5641 {
5642 ip->bw_conv_error = TRUE;
5643 return FAIL;
5644 }
5645
5646 /* copy remainder to ip->bw_rest[] to be used for the next call. */
5647 if (fromlen > 0)
5648 mch_memmove(ip->bw_rest, (void *)from, fromlen);
5649 ip->bw_restlen = (int)fromlen;
5650
5651 buf = ip->bw_conv_buf;
5652 len = (int)((char_u *)to - ip->bw_conv_buf);
5653 }
Bram Moolenaar13505972019-01-24 15:04:48 +01005654#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00005655 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00005656
Bram Moolenaare6bf6552017-06-27 22:11:51 +02005657 if (ip->bw_fd < 0)
5658 /* Only checking conversion, which is OK if we get here. */
5659 return OK;
5660
Bram Moolenaar071d4272004-06-13 20:20:40 +00005661#ifdef FEAT_CRYPT
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02005662 if (flags & FIO_ENCRYPTED)
5663 {
5664 /* Encrypt the data. Do it in-place if possible, otherwise use an
5665 * allocated buffer. */
Bram Moolenaar987411d2019-01-18 22:48:34 +01005666# ifdef CRYPT_NOT_INPLACE
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02005667 if (crypt_works_inplace(ip->bw_buffer->b_cryptstate))
5668 {
Bram Moolenaar987411d2019-01-18 22:48:34 +01005669# endif
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02005670 crypt_encode_inplace(ip->bw_buffer->b_cryptstate, buf, len);
Bram Moolenaar987411d2019-01-18 22:48:34 +01005671# ifdef CRYPT_NOT_INPLACE
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02005672 }
5673 else
5674 {
5675 char_u *outbuf;
5676
5677 len = crypt_encode_alloc(curbuf->b_cryptstate, buf, len, &outbuf);
5678 if (len == 0)
5679 return OK; /* Crypt layer is buffering, will flush later. */
5680 wlen = write_eintr(ip->bw_fd, outbuf, len);
5681 vim_free(outbuf);
5682 return (wlen < len) ? FAIL : OK;
5683 }
Bram Moolenaar987411d2019-01-18 22:48:34 +01005684# endif
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02005685 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00005686#endif
5687
Bram Moolenaar540fc6f2010-12-17 16:27:16 +01005688 wlen = write_eintr(ip->bw_fd, buf, len);
5689 return (wlen < len) ? FAIL : OK;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005690}
5691
Bram Moolenaar071d4272004-06-13 20:20:40 +00005692/*
5693 * Convert a Unicode character to bytes.
Bram Moolenaar32b485f2009-07-29 16:06:27 +00005694 * Return TRUE for an error, FALSE when it's OK.
Bram Moolenaar071d4272004-06-13 20:20:40 +00005695 */
5696 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005697ucs2bytes(
5698 unsigned c, /* in: character */
5699 char_u **pp, /* in/out: pointer to result */
5700 int flags) /* FIO_ flags */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005701{
5702 char_u *p = *pp;
5703 int error = FALSE;
5704 int cc;
5705
5706
5707 if (flags & FIO_UCS4)
5708 {
5709 if (flags & FIO_ENDIAN_L)
5710 {
5711 *p++ = c;
5712 *p++ = (c >> 8);
5713 *p++ = (c >> 16);
5714 *p++ = (c >> 24);
5715 }
5716 else
5717 {
5718 *p++ = (c >> 24);
5719 *p++ = (c >> 16);
5720 *p++ = (c >> 8);
5721 *p++ = c;
5722 }
5723 }
5724 else if (flags & (FIO_UCS2 | FIO_UTF16))
5725 {
5726 if (c >= 0x10000)
5727 {
5728 if (flags & FIO_UTF16)
5729 {
5730 /* Make two words, ten bits of the character in each. First
5731 * word is 0xd800 - 0xdbff, second one 0xdc00 - 0xdfff */
5732 c -= 0x10000;
5733 if (c >= 0x100000)
5734 error = TRUE;
5735 cc = ((c >> 10) & 0x3ff) + 0xd800;
5736 if (flags & FIO_ENDIAN_L)
5737 {
5738 *p++ = cc;
5739 *p++ = ((unsigned)cc >> 8);
5740 }
5741 else
5742 {
5743 *p++ = ((unsigned)cc >> 8);
5744 *p++ = cc;
5745 }
5746 c = (c & 0x3ff) + 0xdc00;
5747 }
5748 else
5749 error = TRUE;
5750 }
5751 if (flags & FIO_ENDIAN_L)
5752 {
5753 *p++ = c;
5754 *p++ = (c >> 8);
5755 }
5756 else
5757 {
5758 *p++ = (c >> 8);
5759 *p++ = c;
5760 }
5761 }
5762 else /* Latin1 */
5763 {
5764 if (c >= 0x100)
5765 {
5766 error = TRUE;
5767 *p++ = 0xBF;
5768 }
5769 else
5770 *p++ = c;
5771 }
5772
5773 *pp = p;
5774 return error;
5775}
5776
5777/*
Bram Moolenaarb5cdf2e2009-07-29 16:25:31 +00005778 * Return TRUE if file encoding "fenc" requires conversion from or to
5779 * 'encoding'.
Bram Moolenaar071d4272004-06-13 20:20:40 +00005780 */
5781 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005782need_conversion(char_u *fenc)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005783{
Bram Moolenaarb5cdf2e2009-07-29 16:25:31 +00005784 int same_encoding;
5785 int enc_flags;
5786 int fenc_flags;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005787
Bram Moolenaarb5cdf2e2009-07-29 16:25:31 +00005788 if (*fenc == NUL || STRCMP(p_enc, fenc) == 0)
Bram Moolenaar442b4222010-05-24 21:34:22 +02005789 {
Bram Moolenaarb5cdf2e2009-07-29 16:25:31 +00005790 same_encoding = TRUE;
Bram Moolenaar442b4222010-05-24 21:34:22 +02005791 fenc_flags = 0;
5792 }
Bram Moolenaarb5cdf2e2009-07-29 16:25:31 +00005793 else
5794 {
5795 /* Ignore difference between "ansi" and "latin1", "ucs-4" and
5796 * "ucs-4be", etc. */
5797 enc_flags = get_fio_flags(p_enc);
5798 fenc_flags = get_fio_flags(fenc);
5799 same_encoding = (enc_flags != 0 && fenc_flags == enc_flags);
5800 }
5801 if (same_encoding)
5802 {
5803 /* Specified encoding matches with 'encoding'. This requires
5804 * conversion when 'encoding' is Unicode but not UTF-8. */
5805 return enc_unicode != 0;
5806 }
5807
5808 /* Encodings differ. However, conversion is not needed when 'enc' is any
5809 * Unicode encoding and the file is UTF-8. */
5810 return !(enc_utf8 && fenc_flags == FIO_UTF8);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005811}
5812
5813/*
5814 * Check "ptr" for a unicode encoding and return the FIO_ flags needed for the
5815 * internal conversion.
5816 * if "ptr" is an empty string, use 'encoding'.
5817 */
5818 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005819get_fio_flags(char_u *ptr)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005820{
5821 int prop;
5822
5823 if (*ptr == NUL)
5824 ptr = p_enc;
5825
5826 prop = enc_canon_props(ptr);
5827 if (prop & ENC_UNICODE)
5828 {
5829 if (prop & ENC_2BYTE)
5830 {
5831 if (prop & ENC_ENDIAN_L)
5832 return FIO_UCS2 | FIO_ENDIAN_L;
5833 return FIO_UCS2;
5834 }
5835 if (prop & ENC_4BYTE)
5836 {
5837 if (prop & ENC_ENDIAN_L)
5838 return FIO_UCS4 | FIO_ENDIAN_L;
5839 return FIO_UCS4;
5840 }
5841 if (prop & ENC_2WORD)
5842 {
5843 if (prop & ENC_ENDIAN_L)
5844 return FIO_UTF16 | FIO_ENDIAN_L;
5845 return FIO_UTF16;
5846 }
5847 return FIO_UTF8;
5848 }
5849 if (prop & ENC_LATIN1)
5850 return FIO_LATIN1;
5851 /* must be ENC_DBCS, requires iconv() */
5852 return 0;
5853}
5854
5855#ifdef WIN3264
5856/*
5857 * Check "ptr" for a MS-Windows codepage name and return the FIO_ flags needed
5858 * for the conversion MS-Windows can do for us. Also accept "utf-8".
5859 * Used for conversion between 'encoding' and 'fileencoding'.
5860 */
5861 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005862get_win_fio_flags(char_u *ptr)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005863{
5864 int cp;
5865
5866 /* Cannot do this when 'encoding' is not utf-8 and not a codepage. */
5867 if (!enc_utf8 && enc_codepage <= 0)
5868 return 0;
5869
5870 cp = encname2codepage(ptr);
5871 if (cp == 0)
5872 {
5873# ifdef CP_UTF8 /* VC 4.1 doesn't define CP_UTF8 */
5874 if (STRCMP(ptr, "utf-8") == 0)
5875 cp = CP_UTF8;
5876 else
5877# endif
5878 return 0;
5879 }
5880 return FIO_PUT_CP(cp) | FIO_CODEPAGE;
5881}
5882#endif
5883
Bram Moolenaard0573012017-10-28 21:11:06 +02005884#ifdef MACOS_CONVERT
Bram Moolenaar071d4272004-06-13 20:20:40 +00005885/*
5886 * Check "ptr" for a Carbon supported encoding and return the FIO_ flags
5887 * needed for the internal conversion to/from utf-8 or latin1.
5888 */
5889 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005890get_mac_fio_flags(char_u *ptr)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005891{
5892 if ((enc_utf8 || STRCMP(p_enc, "latin1") == 0)
5893 && (enc_canon_props(ptr) & ENC_MACROMAN))
5894 return FIO_MACROMAN;
5895 return 0;
5896}
5897#endif
5898
5899/*
5900 * Check for a Unicode BOM (Byte Order Mark) at the start of p[size].
5901 * "size" must be at least 2.
5902 * Return the name of the encoding and set "*lenp" to the length.
5903 * Returns NULL when no BOM found.
5904 */
5905 static char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005906check_for_bom(
5907 char_u *p,
5908 long size,
5909 int *lenp,
5910 int flags)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005911{
5912 char *name = NULL;
5913 int len = 2;
5914
5915 if (p[0] == 0xef && p[1] == 0xbb && size >= 3 && p[2] == 0xbf
Bram Moolenaaree0f5a62008-07-24 20:09:16 +00005916 && (flags == FIO_ALL || flags == FIO_UTF8 || flags == 0))
Bram Moolenaar071d4272004-06-13 20:20:40 +00005917 {
5918 name = "utf-8"; /* EF BB BF */
5919 len = 3;
5920 }
5921 else if (p[0] == 0xff && p[1] == 0xfe)
5922 {
5923 if (size >= 4 && p[2] == 0 && p[3] == 0
5924 && (flags == FIO_ALL || flags == (FIO_UCS4 | FIO_ENDIAN_L)))
5925 {
5926 name = "ucs-4le"; /* FF FE 00 00 */
5927 len = 4;
5928 }
Bram Moolenaar223a1892008-11-11 20:57:11 +00005929 else if (flags == (FIO_UCS2 | FIO_ENDIAN_L))
Bram Moolenaar071d4272004-06-13 20:20:40 +00005930 name = "ucs-2le"; /* FF FE */
Bram Moolenaar223a1892008-11-11 20:57:11 +00005931 else if (flags == FIO_ALL || flags == (FIO_UTF16 | FIO_ENDIAN_L))
5932 /* utf-16le is preferred, it also works for ucs-2le text */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005933 name = "utf-16le"; /* FF FE */
5934 }
5935 else if (p[0] == 0xfe && p[1] == 0xff
5936 && (flags == FIO_ALL || flags == FIO_UCS2 || flags == FIO_UTF16))
5937 {
Bram Moolenaarffd82c52008-02-20 17:15:26 +00005938 /* Default to utf-16, it works also for ucs-2 text. */
5939 if (flags == FIO_UCS2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005940 name = "ucs-2"; /* FE FF */
Bram Moolenaarffd82c52008-02-20 17:15:26 +00005941 else
5942 name = "utf-16"; /* FE FF */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005943 }
5944 else if (size >= 4 && p[0] == 0 && p[1] == 0 && p[2] == 0xfe
5945 && p[3] == 0xff && (flags == FIO_ALL || flags == FIO_UCS4))
5946 {
5947 name = "ucs-4"; /* 00 00 FE FF */
5948 len = 4;
5949 }
5950
5951 *lenp = len;
5952 return (char_u *)name;
5953}
5954
5955/*
5956 * Generate a BOM in "buf[4]" for encoding "name".
5957 * Return the length of the BOM (zero when no BOM).
5958 */
5959 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005960make_bom(char_u *buf, char_u *name)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005961{
5962 int flags;
5963 char_u *p;
5964
5965 flags = get_fio_flags(name);
5966
5967 /* Can't put a BOM in a non-Unicode file. */
5968 if (flags == FIO_LATIN1 || flags == 0)
5969 return 0;
5970
5971 if (flags == FIO_UTF8) /* UTF-8 */
5972 {
5973 buf[0] = 0xef;
5974 buf[1] = 0xbb;
5975 buf[2] = 0xbf;
5976 return 3;
5977 }
5978 p = buf;
5979 (void)ucs2bytes(0xfeff, &p, flags);
5980 return (int)(p - buf);
5981}
Bram Moolenaar071d4272004-06-13 20:20:40 +00005982
5983/*
5984 * Try to find a shortname by comparing the fullname with the current
5985 * directory.
Bram Moolenaard089d9b2007-09-30 12:02:55 +00005986 * Returns "full_path" or pointer into "full_path" if shortened.
5987 */
5988 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01005989shorten_fname1(char_u *full_path)
Bram Moolenaard089d9b2007-09-30 12:02:55 +00005990{
Bram Moolenaard9462e32011-04-11 21:35:11 +02005991 char_u *dirname;
Bram Moolenaard089d9b2007-09-30 12:02:55 +00005992 char_u *p = full_path;
5993
Bram Moolenaard9462e32011-04-11 21:35:11 +02005994 dirname = alloc(MAXPATHL);
5995 if (dirname == NULL)
5996 return full_path;
Bram Moolenaard089d9b2007-09-30 12:02:55 +00005997 if (mch_dirname(dirname, MAXPATHL) == OK)
5998 {
5999 p = shorten_fname(full_path, dirname);
6000 if (p == NULL || *p == NUL)
6001 p = full_path;
6002 }
Bram Moolenaard9462e32011-04-11 21:35:11 +02006003 vim_free(dirname);
Bram Moolenaard089d9b2007-09-30 12:02:55 +00006004 return p;
6005}
6006
6007/*
6008 * Try to find a shortname by comparing the fullname with the current
6009 * directory.
Bram Moolenaar071d4272004-06-13 20:20:40 +00006010 * Returns NULL if not shorter name possible, pointer into "full_path"
6011 * otherwise.
6012 */
6013 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01006014shorten_fname(char_u *full_path, char_u *dir_name)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006015{
6016 int len;
6017 char_u *p;
6018
6019 if (full_path == NULL)
6020 return NULL;
6021 len = (int)STRLEN(dir_name);
6022 if (fnamencmp(dir_name, full_path, len) == 0)
6023 {
6024 p = full_path + len;
Bram Moolenaar48e330a2016-02-23 14:53:34 +01006025#if defined(MSWIN)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006026 /*
Bram Moolenaar48e330a2016-02-23 14:53:34 +01006027 * MSWIN: when a file is in the root directory, dir_name will end in a
Bram Moolenaar071d4272004-06-13 20:20:40 +00006028 * slash, since C: by itself does not define a specific dir. In this
6029 * case p may already be correct. <negri>
6030 */
6031 if (!((len > 2) && (*(p - 2) == ':')))
6032#endif
6033 {
6034 if (vim_ispathsep(*p))
6035 ++p;
6036#ifndef VMS /* the path separator is always part of the path */
6037 else
6038 p = NULL;
6039#endif
6040 }
6041 }
Bram Moolenaar48e330a2016-02-23 14:53:34 +01006042#if defined(MSWIN)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006043 /*
6044 * When using a file in the current drive, remove the drive name:
6045 * "A:\dir\file" -> "\dir\file". This helps when moving a session file on
6046 * a floppy from "A:\dir" to "B:\dir".
6047 */
6048 else if (len > 3
6049 && TOUPPER_LOC(full_path[0]) == TOUPPER_LOC(dir_name[0])
6050 && full_path[1] == ':'
6051 && vim_ispathsep(full_path[2]))
6052 p = full_path + 2;
6053#endif
6054 else
6055 p = NULL;
6056 return p;
6057}
6058
6059/*
Bram Moolenaara796d462018-05-01 14:30:36 +02006060 * Shorten filename of a buffer.
Bram Moolenaar071d4272004-06-13 20:20:40 +00006061 * When "force" is TRUE: Use full path from now on for files currently being
6062 * edited, both for file name and swap file name. Try to shorten the file
6063 * names a bit, if safe to do so.
6064 * When "force" is FALSE: Only try to shorten absolute file names.
6065 * For buffers that have buftype "nofile" or "scratch": never change the file
6066 * name.
6067 */
6068 void
Bram Moolenaara796d462018-05-01 14:30:36 +02006069shorten_buf_fname(buf_T *buf, char_u *dirname, int force)
6070{
6071 char_u *p;
6072
6073 if (buf->b_fname != NULL
6074#ifdef FEAT_QUICKFIX
6075 && !bt_nofile(buf)
6076#endif
6077 && !path_with_url(buf->b_fname)
6078 && (force
6079 || buf->b_sfname == NULL
6080 || mch_isFullName(buf->b_sfname)))
6081 {
Bram Moolenaar3d6014f2018-10-11 19:27:47 +02006082 if (buf->b_sfname != buf->b_ffname)
6083 VIM_CLEAR(buf->b_sfname);
Bram Moolenaara796d462018-05-01 14:30:36 +02006084 p = shorten_fname(buf->b_ffname, dirname);
6085 if (p != NULL)
6086 {
6087 buf->b_sfname = vim_strsave(p);
6088 buf->b_fname = buf->b_sfname;
6089 }
6090 if (p == NULL || buf->b_fname == NULL)
6091 buf->b_fname = buf->b_ffname;
6092 }
6093}
6094
6095/*
6096 * Shorten filenames for all buffers.
6097 */
6098 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01006099shorten_fnames(int force)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006100{
6101 char_u dirname[MAXPATHL];
6102 buf_T *buf;
Bram Moolenaar071d4272004-06-13 20:20:40 +00006103
6104 mch_dirname(dirname, MAXPATHL);
Bram Moolenaar29323592016-07-24 22:04:11 +02006105 FOR_ALL_BUFFERS(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006106 {
Bram Moolenaara796d462018-05-01 14:30:36 +02006107 shorten_buf_fname(buf, dirname, force);
Bram Moolenaar69a7cb42004-06-20 12:51:53 +00006108
6109 /* Always make the swap file name a full path, a "nofile" buffer may
6110 * also have a swap file. */
6111 mf_fullname(buf->b_ml.ml_mfp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00006112 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00006113 status_redraw_all();
Bram Moolenaar49d7bf12006-02-17 21:45:41 +00006114 redraw_tabline = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00006115}
6116
6117#if (defined(FEAT_DND) && defined(FEAT_GUI_GTK)) \
6118 || defined(FEAT_GUI_MSWIN) \
6119 || defined(FEAT_GUI_MAC) \
6120 || defined(PROTO)
6121/*
6122 * Shorten all filenames in "fnames[count]" by current directory.
6123 */
6124 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01006125shorten_filenames(char_u **fnames, int count)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006126{
6127 int i;
6128 char_u dirname[MAXPATHL];
6129 char_u *p;
6130
6131 if (fnames == NULL || count < 1)
6132 return;
6133 mch_dirname(dirname, sizeof(dirname));
6134 for (i = 0; i < count; ++i)
6135 {
6136 if ((p = shorten_fname(fnames[i], dirname)) != NULL)
6137 {
6138 /* shorten_fname() returns pointer in given "fnames[i]". If free
6139 * "fnames[i]" first, "p" becomes invalid. So we need to copy
6140 * "p" first then free fnames[i]. */
6141 p = vim_strsave(p);
6142 vim_free(fnames[i]);
6143 fnames[i] = p;
6144 }
6145 }
6146}
6147#endif
6148
6149/*
Bram Moolenaarb782ba42018-08-07 21:39:28 +02006150 * Add extension to file name - change path/fo.o.h to path/fo.o.h.ext or
Bram Moolenaar071d4272004-06-13 20:20:40 +00006151 * fo_o_h.ext for MSDOS or when shortname option set.
6152 *
6153 * Assumed that fname is a valid name found in the filesystem we assure that
6154 * the return value is a different name and ends in 'ext'.
6155 * "ext" MUST be at most 4 characters long if it starts with a dot, 3
6156 * characters otherwise.
6157 * Space for the returned name is allocated, must be freed later.
6158 * Returns NULL when out of memory.
6159 */
6160 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01006161modname(
6162 char_u *fname,
6163 char_u *ext,
6164 int prepend_dot) /* may prepend a '.' to file name */
Bram Moolenaar071d4272004-06-13 20:20:40 +00006165{
Bram Moolenaar48e330a2016-02-23 14:53:34 +01006166 return buf_modname((curbuf->b_p_sn || curbuf->b_shortname),
Bram Moolenaar071d4272004-06-13 20:20:40 +00006167 fname, ext, prepend_dot);
6168}
6169
6170 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01006171buf_modname(
6172 int shortname, /* use 8.3 file name */
6173 char_u *fname,
6174 char_u *ext,
6175 int prepend_dot) /* may prepend a '.' to file name */
Bram Moolenaar071d4272004-06-13 20:20:40 +00006176{
6177 char_u *retval;
6178 char_u *s;
6179 char_u *e;
6180 char_u *ptr;
6181 int fnamelen, extlen;
6182
6183 extlen = (int)STRLEN(ext);
6184
6185 /*
6186 * If there is no file name we must get the name of the current directory
6187 * (we need the full path in case :cd is used).
6188 */
6189 if (fname == NULL || *fname == NUL)
6190 {
6191 retval = alloc((unsigned)(MAXPATHL + extlen + 3));
6192 if (retval == NULL)
6193 return NULL;
6194 if (mch_dirname(retval, MAXPATHL) == FAIL ||
6195 (fnamelen = (int)STRLEN(retval)) == 0)
6196 {
6197 vim_free(retval);
6198 return NULL;
6199 }
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00006200 if (!after_pathsep(retval, retval + fnamelen))
Bram Moolenaar071d4272004-06-13 20:20:40 +00006201 {
6202 retval[fnamelen++] = PATHSEP;
6203 retval[fnamelen] = NUL;
6204 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00006205 prepend_dot = FALSE; /* nothing to prepend a dot to */
Bram Moolenaar071d4272004-06-13 20:20:40 +00006206 }
6207 else
6208 {
6209 fnamelen = (int)STRLEN(fname);
6210 retval = alloc((unsigned)(fnamelen + extlen + 3));
6211 if (retval == NULL)
6212 return NULL;
6213 STRCPY(retval, fname);
6214#ifdef VMS
6215 vms_remove_version(retval); /* we do not need versions here */
6216#endif
6217 }
6218
6219 /*
6220 * search backwards until we hit a '/', '\' or ':' replacing all '.'
6221 * by '_' for MSDOS or when shortname option set and ext starts with a dot.
6222 * Then truncate what is after the '/', '\' or ':' to 8 characters for
6223 * MSDOS and 26 characters for AMIGA, a lot more for UNIX.
6224 */
Bram Moolenaar91acfff2017-03-12 19:22:36 +01006225 for (ptr = retval + fnamelen; ptr > retval; MB_PTR_BACK(retval, ptr))
Bram Moolenaar071d4272004-06-13 20:20:40 +00006226 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00006227 if (*ext == '.'
Bram Moolenaare60acc12011-05-10 16:41:25 +02006228#ifdef USE_LONG_FNAME
Bram Moolenaar071d4272004-06-13 20:20:40 +00006229 && (!USE_LONG_FNAME || shortname)
Bram Moolenaare60acc12011-05-10 16:41:25 +02006230#else
Bram Moolenaar071d4272004-06-13 20:20:40 +00006231 && shortname
Bram Moolenaare60acc12011-05-10 16:41:25 +02006232#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00006233 )
6234 if (*ptr == '.') /* replace '.' by '_' */
6235 *ptr = '_';
Bram Moolenaar071d4272004-06-13 20:20:40 +00006236 if (vim_ispathsep(*ptr))
Bram Moolenaar53180ce2005-07-05 21:48:14 +00006237 {
6238 ++ptr;
Bram Moolenaar071d4272004-06-13 20:20:40 +00006239 break;
Bram Moolenaar53180ce2005-07-05 21:48:14 +00006240 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00006241 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00006242
6243 /* the file name has at most BASENAMELEN characters. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00006244 if (STRLEN(ptr) > (unsigned)BASENAMELEN)
6245 ptr[BASENAMELEN] = '\0';
Bram Moolenaar071d4272004-06-13 20:20:40 +00006246
6247 s = ptr + STRLEN(ptr);
6248
6249 /*
6250 * For 8.3 file names we may have to reduce the length.
6251 */
6252#ifdef USE_LONG_FNAME
6253 if (!USE_LONG_FNAME || shortname)
6254#else
Bram Moolenaar071d4272004-06-13 20:20:40 +00006255 if (shortname)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006256#endif
6257 {
6258 /*
6259 * If there is no file name, or the file name ends in '/', and the
6260 * extension starts with '.', put a '_' before the dot, because just
6261 * ".ext" is invalid.
6262 */
6263 if (fname == NULL || *fname == NUL
6264 || vim_ispathsep(fname[STRLEN(fname) - 1]))
6265 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00006266 if (*ext == '.')
Bram Moolenaar071d4272004-06-13 20:20:40 +00006267 *s++ = '_';
6268 }
6269 /*
6270 * If the extension starts with '.', truncate the base name at 8
6271 * characters
6272 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00006273 else if (*ext == '.')
Bram Moolenaar071d4272004-06-13 20:20:40 +00006274 {
Bram Moolenaar78a15312009-05-15 19:33:18 +00006275 if ((size_t)(s - ptr) > (size_t)8)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006276 {
6277 s = ptr + 8;
6278 *s = '\0';
6279 }
6280 }
6281 /*
6282 * If the extension doesn't start with '.', and the file name
6283 * doesn't have an extension yet, append a '.'
6284 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00006285 else if ((e = vim_strchr(ptr, '.')) == NULL)
6286 *s++ = '.';
Bram Moolenaar071d4272004-06-13 20:20:40 +00006287 /*
6288 * If the extension doesn't start with '.', and there already is an
Bram Moolenaar7263a772007-05-10 17:35:54 +00006289 * extension, it may need to be truncated
Bram Moolenaar071d4272004-06-13 20:20:40 +00006290 */
6291 else if ((int)STRLEN(e) + extlen > 4)
6292 s = e + 4 - extlen;
6293 }
Bram Moolenaare7fedb62015-12-31 19:07:19 +01006294#if defined(USE_LONG_FNAME) || defined(WIN3264)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006295 /*
6296 * If there is no file name, and the extension starts with '.', put a
6297 * '_' before the dot, because just ".ext" may be invalid if it's on a
6298 * FAT partition, and on HPFS it doesn't matter.
6299 */
6300 else if ((fname == NULL || *fname == NUL) && *ext == '.')
6301 *s++ = '_';
6302#endif
6303
6304 /*
Bram Moolenaarb23a7e82008-06-27 18:42:32 +00006305 * Append the extension.
Bram Moolenaar071d4272004-06-13 20:20:40 +00006306 * ext can start with '.' and cannot exceed 3 more characters.
6307 */
6308 STRCPY(s, ext);
6309
Bram Moolenaar071d4272004-06-13 20:20:40 +00006310 /*
6311 * Prepend the dot.
6312 */
Bram Moolenaare60acc12011-05-10 16:41:25 +02006313 if (prepend_dot && !shortname && *(e = gettail(retval)) != '.'
Bram Moolenaar071d4272004-06-13 20:20:40 +00006314#ifdef USE_LONG_FNAME
6315 && USE_LONG_FNAME
6316#endif
6317 )
6318 {
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;
6357#ifdef USE_CR
6358 eof = fgets_cr((char *)buf, size, fp);
6359#else
6360 eof = fgets((char *)buf, size, fp);
6361#endif
6362 if (buf[size - 2] != NUL && buf[size - 2] != '\n')
6363 {
6364 buf[size - 1] = NUL; /* Truncate the line */
6365
6366 /* Now throw away the rest of the line: */
6367 do
6368 {
6369 tbuf[FGETS_SIZE - 2] = NUL;
6370#ifdef USE_CR
Bram Moolenaar42335f52018-09-13 15:33:43 +02006371 vim_ignoredp = fgets_cr((char *)tbuf, FGETS_SIZE, fp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00006372#else
Bram Moolenaar42335f52018-09-13 15:33:43 +02006373 vim_ignoredp = fgets((char *)tbuf, FGETS_SIZE, fp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00006374#endif
6375 } while (tbuf[FGETS_SIZE - 2] != NUL && tbuf[FGETS_SIZE - 2] != '\n');
6376 }
6377 return (eof == NULL);
6378}
6379
6380#if defined(USE_CR) || defined(PROTO)
6381/*
6382 * Like vim_fgets(), but accept any line terminator: CR, CR-LF or LF.
6383 * Returns TRUE for end-of-file.
6384 * Only used for the Mac, because it's much slower than vim_fgets().
6385 */
6386 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01006387tag_fgets(char_u *buf, int size, FILE *fp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006388{
6389 int i = 0;
6390 int c;
6391 int eof = FALSE;
6392
6393 for (;;)
6394 {
6395 c = fgetc(fp);
6396 if (c == EOF)
6397 {
6398 eof = TRUE;
6399 break;
6400 }
6401 if (c == '\r')
6402 {
6403 /* Always store a NL for end-of-line. */
6404 if (i < size - 1)
6405 buf[i++] = '\n';
6406 c = fgetc(fp);
6407 if (c != '\n') /* Macintosh format: single CR. */
6408 ungetc(c, fp);
6409 break;
6410 }
6411 if (i < size - 1)
6412 buf[i++] = c;
6413 if (c == '\n')
6414 break;
6415 }
6416 buf[i] = NUL;
6417 return eof;
6418}
6419#endif
6420
6421/*
6422 * rename() only works if both files are on the same file system, this
6423 * function will (attempts to?) copy the file across if rename fails -- webb
6424 * Return -1 for failure, 0 for success.
6425 */
6426 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01006427vim_rename(char_u *from, char_u *to)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006428{
6429 int fd_in;
6430 int fd_out;
6431 int n;
6432 char *errmsg = NULL;
6433 char *buffer;
6434#ifdef AMIGA
6435 BPTR flock;
6436#endif
Bram Moolenaar8767f522016-07-01 17:17:39 +02006437 stat_T st;
Bram Moolenaar9be038d2005-03-08 22:34:32 +00006438 long perm;
Bram Moolenaarcd71fa32005-03-11 22:46:48 +00006439#ifdef HAVE_ACL
6440 vim_acl_T acl; /* ACL from original file */
6441#endif
Bram Moolenaare0e6f992008-12-31 15:21:32 +00006442 int use_tmp_file = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00006443
6444 /*
Bram Moolenaare0e6f992008-12-31 15:21:32 +00006445 * When the names are identical, there is nothing to do. When they refer
6446 * to the same file (ignoring case and slash/backslash differences) but
6447 * the file name differs we need to go through a temp file.
Bram Moolenaar071d4272004-06-13 20:20:40 +00006448 */
6449 if (fnamecmp(from, to) == 0)
Bram Moolenaare0e6f992008-12-31 15:21:32 +00006450 {
Bram Moolenaar71afbfe2013-03-19 16:49:16 +01006451 if (p_fic && STRCMP(gettail(from), gettail(to)) != 0)
Bram Moolenaare0e6f992008-12-31 15:21:32 +00006452 use_tmp_file = TRUE;
6453 else
Bram Moolenaare0e6f992008-12-31 15:21:32 +00006454 return 0;
6455 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00006456
6457 /*
6458 * Fail if the "from" file doesn't exist. Avoids that "to" is deleted.
6459 */
6460 if (mch_stat((char *)from, &st) < 0)
6461 return -1;
6462
Bram Moolenaar3576da72008-12-30 15:15:57 +00006463#ifdef UNIX
6464 {
Bram Moolenaar8767f522016-07-01 17:17:39 +02006465 stat_T st_to;
Bram Moolenaar3576da72008-12-30 15:15:57 +00006466
6467 /* It's possible for the source and destination to be the same file.
6468 * This happens when "from" and "to" differ in case and are on a FAT32
6469 * filesystem. In that case go through a temp file name. */
6470 if (mch_stat((char *)to, &st_to) >= 0
6471 && st.st_dev == st_to.st_dev
6472 && st.st_ino == st_to.st_ino)
Bram Moolenaare0e6f992008-12-31 15:21:32 +00006473 use_tmp_file = TRUE;
6474 }
6475#endif
Bram Moolenaar1c32dff2011-05-05 16:41:24 +02006476#ifdef WIN3264
6477 {
6478 BY_HANDLE_FILE_INFORMATION info1, info2;
6479
6480 /* It's possible for the source and destination to be the same file.
6481 * In that case go through a temp file name. This makes rename("foo",
6482 * "./foo") a no-op (in a complicated way). */
6483 if (win32_fileinfo(from, &info1) == FILEINFO_OK
6484 && win32_fileinfo(to, &info2) == FILEINFO_OK
6485 && info1.dwVolumeSerialNumber == info2.dwVolumeSerialNumber
6486 && info1.nFileIndexHigh == info2.nFileIndexHigh
6487 && info1.nFileIndexLow == info2.nFileIndexLow)
6488 use_tmp_file = TRUE;
6489 }
6490#endif
Bram Moolenaare0e6f992008-12-31 15:21:32 +00006491
Bram Moolenaare0e6f992008-12-31 15:21:32 +00006492 if (use_tmp_file)
6493 {
6494 char tempname[MAXPATHL + 1];
6495
6496 /*
6497 * Find a name that doesn't exist and is in the same directory.
6498 * Rename "from" to "tempname" and then rename "tempname" to "to".
6499 */
6500 if (STRLEN(from) >= MAXPATHL - 5)
6501 return -1;
6502 STRCPY(tempname, from);
6503 for (n = 123; n < 99999; ++n)
Bram Moolenaar3576da72008-12-30 15:15:57 +00006504 {
Bram Moolenaare0e6f992008-12-31 15:21:32 +00006505 sprintf((char *)gettail((char_u *)tempname), "%d", n);
6506 if (mch_stat(tempname, &st) < 0)
Bram Moolenaar3576da72008-12-30 15:15:57 +00006507 {
Bram Moolenaare0e6f992008-12-31 15:21:32 +00006508 if (mch_rename((char *)from, tempname) == 0)
Bram Moolenaar3576da72008-12-30 15:15:57 +00006509 {
Bram Moolenaare0e6f992008-12-31 15:21:32 +00006510 if (mch_rename(tempname, (char *)to) == 0)
6511 return 0;
6512 /* Strange, the second step failed. Try moving the
6513 * file back and return failure. */
6514 mch_rename(tempname, (char *)from);
Bram Moolenaar3576da72008-12-30 15:15:57 +00006515 return -1;
6516 }
Bram Moolenaare0e6f992008-12-31 15:21:32 +00006517 /* If it fails for one temp name it will most likely fail
6518 * for any temp name, give up. */
6519 return -1;
Bram Moolenaar3576da72008-12-30 15:15:57 +00006520 }
Bram Moolenaar3576da72008-12-30 15:15:57 +00006521 }
Bram Moolenaare0e6f992008-12-31 15:21:32 +00006522 return -1;
Bram Moolenaar3576da72008-12-30 15:15:57 +00006523 }
Bram Moolenaar3576da72008-12-30 15:15:57 +00006524
Bram Moolenaar071d4272004-06-13 20:20:40 +00006525 /*
6526 * Delete the "to" file, this is required on some systems to make the
6527 * mch_rename() work, on other systems it makes sure that we don't have
6528 * two files when the mch_rename() fails.
6529 */
6530
6531#ifdef AMIGA
6532 /*
6533 * With MSDOS-compatible filesystems (crossdos, messydos) it is possible
6534 * that the name of the "to" file is the same as the "from" file, even
Bram Moolenaar7263a772007-05-10 17:35:54 +00006535 * though the names are different. To avoid the chance of accidentally
Bram Moolenaar071d4272004-06-13 20:20:40 +00006536 * deleting the "from" file (horror!) we lock it during the remove.
6537 *
6538 * When used for making a backup before writing the file: This should not
6539 * happen with ":w", because startscript() should detect this problem and
6540 * set buf->b_shortname, causing modname() to return a correct ".bak" file
6541 * name. This problem does exist with ":w filename", but then the
6542 * original file will be somewhere else so the backup isn't really
6543 * important. If autoscripting is off the rename may fail.
6544 */
6545 flock = Lock((UBYTE *)from, (long)ACCESS_READ);
6546#endif
6547 mch_remove(to);
6548#ifdef AMIGA
6549 if (flock)
6550 UnLock(flock);
6551#endif
6552
6553 /*
6554 * First try a normal rename, return if it works.
6555 */
6556 if (mch_rename((char *)from, (char *)to) == 0)
6557 return 0;
6558
6559 /*
6560 * Rename() failed, try copying the file.
6561 */
Bram Moolenaar9be038d2005-03-08 22:34:32 +00006562 perm = mch_getperm(from);
Bram Moolenaarcd71fa32005-03-11 22:46:48 +00006563#ifdef HAVE_ACL
6564 /* For systems that support ACL: get the ACL from the original file. */
6565 acl = mch_get_acl(from);
6566#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00006567 fd_in = mch_open((char *)from, O_RDONLY|O_EXTRA, 0);
6568 if (fd_in == -1)
Bram Moolenaarb23a7e82008-06-27 18:42:32 +00006569 {
6570#ifdef HAVE_ACL
6571 mch_free_acl(acl);
6572#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00006573 return -1;
Bram Moolenaarb23a7e82008-06-27 18:42:32 +00006574 }
Bram Moolenaar9be038d2005-03-08 22:34:32 +00006575
6576 /* Create the new file with same permissions as the original. */
Bram Moolenaara5792f52005-11-23 21:25:05 +00006577 fd_out = mch_open((char *)to,
6578 O_CREAT|O_EXCL|O_WRONLY|O_EXTRA|O_NOFOLLOW, (int)perm);
Bram Moolenaar071d4272004-06-13 20:20:40 +00006579 if (fd_out == -1)
6580 {
6581 close(fd_in);
Bram Moolenaarb23a7e82008-06-27 18:42:32 +00006582#ifdef HAVE_ACL
6583 mch_free_acl(acl);
6584#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00006585 return -1;
6586 }
6587
6588 buffer = (char *)alloc(BUFSIZE);
6589 if (buffer == NULL)
6590 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00006591 close(fd_out);
Bram Moolenaarb23a7e82008-06-27 18:42:32 +00006592 close(fd_in);
6593#ifdef HAVE_ACL
6594 mch_free_acl(acl);
6595#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00006596 return -1;
6597 }
6598
Bram Moolenaar540fc6f2010-12-17 16:27:16 +01006599 while ((n = read_eintr(fd_in, buffer, BUFSIZE)) > 0)
6600 if (write_eintr(fd_out, buffer, n) != n)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006601 {
6602 errmsg = _("E208: Error writing to \"%s\"");
6603 break;
6604 }
6605
6606 vim_free(buffer);
6607 close(fd_in);
6608 if (close(fd_out) < 0)
6609 errmsg = _("E209: Error closing \"%s\"");
6610 if (n < 0)
6611 {
6612 errmsg = _("E210: Error reading \"%s\"");
6613 to = from;
6614 }
Bram Moolenaar7263a772007-05-10 17:35:54 +00006615#ifndef UNIX /* for Unix mch_open() already set the permission */
Bram Moolenaar9be038d2005-03-08 22:34:32 +00006616 mch_setperm(to, perm);
Bram Moolenaarc6039d82005-12-02 00:44:04 +00006617#endif
Bram Moolenaarcd71fa32005-03-11 22:46:48 +00006618#ifdef HAVE_ACL
6619 mch_set_acl(to, acl);
Bram Moolenaarb23a7e82008-06-27 18:42:32 +00006620 mch_free_acl(acl);
Bram Moolenaarcd71fa32005-03-11 22:46:48 +00006621#endif
Bram Moolenaar5bd32f42014-04-02 14:05:38 +02006622#if defined(HAVE_SELINUX) || defined(HAVE_SMACK)
Bram Moolenaare8747442013-11-12 18:09:29 +01006623 mch_copy_sec(from, to);
Bram Moolenaar0671de32013-11-12 05:12:03 +01006624#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00006625 if (errmsg != NULL)
6626 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006627 semsg(errmsg, to);
Bram Moolenaar071d4272004-06-13 20:20:40 +00006628 return -1;
6629 }
6630 mch_remove(from);
6631 return 0;
6632}
6633
6634static int already_warned = FALSE;
6635
6636/*
6637 * Check if any not hidden buffer has been changed.
6638 * Postpone the check if there are characters in the stuff buffer, a global
6639 * command is being executed, a mapping is being executed or an autocommand is
6640 * busy.
6641 * Returns TRUE if some message was written (screen should be redrawn and
6642 * cursor positioned).
6643 */
6644 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01006645check_timestamps(
6646 int focus) /* called for GUI focus event */
Bram Moolenaar071d4272004-06-13 20:20:40 +00006647{
6648 buf_T *buf;
6649 int didit = 0;
6650 int n;
6651
6652 /* Don't check timestamps while system() or another low-level function may
6653 * cause us to lose and gain focus. */
6654 if (no_check_timestamps > 0)
6655 return FALSE;
6656
6657 /* Avoid doing a check twice. The OK/Reload dialog can cause a focus
6658 * event and we would keep on checking if the file is steadily growing.
6659 * Do check again after typing something. */
6660 if (focus && did_check_timestamps)
6661 {
6662 need_check_timestamps = TRUE;
6663 return FALSE;
6664 }
6665
6666 if (!stuff_empty() || global_busy || !typebuf_typed()
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +01006667 || autocmd_busy || curbuf_lock > 0 || allbuf_lock > 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006668 need_check_timestamps = TRUE; /* check later */
6669 else
6670 {
6671 ++no_wait_return;
6672 did_check_timestamps = TRUE;
6673 already_warned = FALSE;
Bram Moolenaar29323592016-07-24 22:04:11 +02006674 FOR_ALL_BUFFERS(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006675 {
6676 /* Only check buffers in a window. */
6677 if (buf->b_nwindows > 0)
6678 {
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02006679 bufref_T bufref;
6680
6681 set_bufref(&bufref, buf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00006682 n = buf_check_timestamp(buf, focus);
6683 if (didit < n)
6684 didit = n;
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02006685 if (n > 0 && !bufref_valid(&bufref))
Bram Moolenaar071d4272004-06-13 20:20:40 +00006686 {
6687 /* Autocommands have removed the buffer, start at the
6688 * first one again. */
6689 buf = firstbuf;
6690 continue;
6691 }
6692 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00006693 }
6694 --no_wait_return;
6695 need_check_timestamps = FALSE;
6696 if (need_wait_return && didit == 2)
6697 {
6698 /* make sure msg isn't overwritten */
Bram Moolenaar32526b32019-01-19 17:43:09 +01006699 msg_puts("\n");
Bram Moolenaar071d4272004-06-13 20:20:40 +00006700 out_flush();
6701 }
6702 }
6703 return didit;
6704}
6705
6706/*
6707 * Move all the lines from buffer "frombuf" to buffer "tobuf".
6708 * Return OK or FAIL. When FAIL "tobuf" is incomplete and/or "frombuf" is not
6709 * empty.
6710 */
6711 static int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01006712move_lines(buf_T *frombuf, buf_T *tobuf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006713{
6714 buf_T *tbuf = curbuf;
6715 int retval = OK;
6716 linenr_T lnum;
6717 char_u *p;
6718
6719 /* Copy the lines in "frombuf" to "tobuf". */
6720 curbuf = tobuf;
6721 for (lnum = 1; lnum <= frombuf->b_ml.ml_line_count; ++lnum)
6722 {
6723 p = vim_strsave(ml_get_buf(frombuf, lnum, FALSE));
6724 if (p == NULL || ml_append(lnum - 1, p, 0, FALSE) == FAIL)
6725 {
6726 vim_free(p);
6727 retval = FAIL;
6728 break;
6729 }
6730 vim_free(p);
6731 }
6732
6733 /* Delete all the lines in "frombuf". */
6734 if (retval != FAIL)
6735 {
6736 curbuf = frombuf;
Bram Moolenaar9460b9d2007-01-09 14:37:01 +00006737 for (lnum = curbuf->b_ml.ml_line_count; lnum > 0; --lnum)
6738 if (ml_delete(lnum, FALSE) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006739 {
6740 /* Oops! We could try putting back the saved lines, but that
6741 * might fail again... */
6742 retval = FAIL;
6743 break;
6744 }
6745 }
6746
6747 curbuf = tbuf;
6748 return retval;
6749}
6750
6751/*
6752 * Check if buffer "buf" has been changed.
6753 * Also check if the file for a new buffer unexpectedly appeared.
6754 * return 1 if a changed buffer was found.
6755 * return 2 if a message has been displayed.
6756 * return 0 otherwise.
6757 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00006758 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01006759buf_check_timestamp(
6760 buf_T *buf,
6761 int focus UNUSED) /* called for GUI focus event */
Bram Moolenaar071d4272004-06-13 20:20:40 +00006762{
Bram Moolenaar8767f522016-07-01 17:17:39 +02006763 stat_T st;
Bram Moolenaar071d4272004-06-13 20:20:40 +00006764 int stat_res;
6765 int retval = 0;
6766 char_u *path;
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006767 char *tbuf;
Bram Moolenaar071d4272004-06-13 20:20:40 +00006768 char *mesg = NULL;
Bram Moolenaar44ecf652005-03-07 23:09:59 +00006769 char *mesg2 = "";
Bram Moolenaar071d4272004-06-13 20:20:40 +00006770 int helpmesg = FALSE;
6771 int reload = FALSE;
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02006772 char *reason;
Bram Moolenaar071d4272004-06-13 20:20:40 +00006773#if defined(FEAT_CON_DIALOG) || defined(FEAT_GUI_DIALOG)
6774 int can_reload = FALSE;
6775#endif
Bram Moolenaar8767f522016-07-01 17:17:39 +02006776 off_T orig_size = buf->b_orig_size;
Bram Moolenaar071d4272004-06-13 20:20:40 +00006777 int orig_mode = buf->b_orig_mode;
6778#ifdef FEAT_GUI
6779 int save_mouse_correct = need_mouse_correct;
6780#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00006781 static int busy = FALSE;
Bram Moolenaar19a09a12005-03-04 23:39:37 +00006782 int n;
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +01006783#ifdef FEAT_EVAL
Bram Moolenaar19a09a12005-03-04 23:39:37 +00006784 char_u *s;
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +01006785#endif
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02006786 bufref_T bufref;
6787
6788 set_bufref(&bufref, buf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00006789
6790 /* If there is no file name, the buffer is not loaded, 'buftype' is
6791 * set, we are in the middle of a save or being called recursively: ignore
6792 * this buffer. */
6793 if (buf->b_ffname == NULL
6794 || buf->b_ml.ml_mfp == NULL
Bram Moolenaar91335e52018-08-01 17:53:12 +02006795 || !bt_normal(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006796 || buf->b_saving
Bram Moolenaar071d4272004-06-13 20:20:40 +00006797 || busy
Bram Moolenaar009b2592004-10-24 19:18:58 +00006798#ifdef FEAT_NETBEANS_INTG
6799 || isNetbeansBuffer(buf)
6800#endif
Bram Moolenaar8cad9302017-08-12 14:32:32 +02006801#ifdef FEAT_TERMINAL
6802 || buf->b_term != NULL
6803#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00006804 )
6805 return 0;
6806
6807 if ( !(buf->b_flags & BF_NOTEDITED)
6808 && buf->b_mtime != 0
6809 && ((stat_res = mch_stat((char *)buf->b_ffname, &st)) < 0
6810 || time_differs((long)st.st_mtime, buf->b_mtime)
Bram Moolenaara7611f62014-05-02 15:46:14 +02006811 || st.st_size != buf->b_orig_size
Bram Moolenaar071d4272004-06-13 20:20:40 +00006812#ifdef HAVE_ST_MODE
6813 || (int)st.st_mode != buf->b_orig_mode
6814#else
6815 || mch_getperm(buf->b_ffname) != buf->b_orig_mode
6816#endif
6817 ))
6818 {
6819 retval = 1;
6820
Bram Moolenaar386bc822018-07-07 18:34:12 +02006821 // set b_mtime to stop further warnings (e.g., when executing
6822 // FileChangedShell autocmd)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006823 if (stat_res < 0)
6824 {
Bram Moolenaar386bc822018-07-07 18:34:12 +02006825 // When 'autoread' is set we'll check the file again to see if it
6826 // re-appears.
6827 buf->b_mtime = buf->b_p_ar;
Bram Moolenaar071d4272004-06-13 20:20:40 +00006828 buf->b_orig_size = 0;
6829 buf->b_orig_mode = 0;
6830 }
6831 else
6832 buf_store_time(buf, &st, buf->b_ffname);
6833
6834 /* Don't do anything for a directory. Might contain the file
6835 * explorer. */
6836 if (mch_isdir(buf->b_fname))
6837 ;
6838
6839 /*
6840 * If 'autoread' is set, the buffer has no changes and the file still
6841 * exists, reload the buffer. Use the buffer-local option value if it
6842 * was set, the global option value otherwise.
6843 */
6844 else if ((buf->b_p_ar >= 0 ? buf->b_p_ar : p_ar)
6845 && !bufIsChanged(buf) && stat_res >= 0)
6846 reload = TRUE;
6847 else
6848 {
Bram Moolenaar19a09a12005-03-04 23:39:37 +00006849 if (stat_res < 0)
6850 reason = "deleted";
6851 else if (bufIsChanged(buf))
6852 reason = "conflict";
Bram Moolenaar3e460fd2019-01-26 16:21:07 +01006853 /*
6854 * Check if the file contents really changed to avoid giving a
6855 * warning when only the timestamp was set (e.g., checked out of
6856 * CVS). Always warn when the buffer was changed.
6857 */
Bram Moolenaar19a09a12005-03-04 23:39:37 +00006858 else if (orig_size != buf->b_orig_size || buf_contents_changed(buf))
6859 reason = "changed";
6860 else if (orig_mode != buf->b_orig_mode)
6861 reason = "mode";
6862 else
6863 reason = "time";
Bram Moolenaar071d4272004-06-13 20:20:40 +00006864
6865 /*
6866 * Only give the warning if there are no FileChangedShell
6867 * autocommands.
6868 * Avoid being called recursively by setting "busy".
6869 */
6870 busy = TRUE;
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +01006871#ifdef FEAT_EVAL
Bram Moolenaar19a09a12005-03-04 23:39:37 +00006872 set_vim_var_string(VV_FCS_REASON, (char_u *)reason, -1);
6873 set_vim_var_string(VV_FCS_CHOICE, (char_u *)"", -1);
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +01006874#endif
Bram Moolenaarbf1b7a72009-03-05 02:15:53 +00006875 ++allbuf_lock;
Bram Moolenaar071d4272004-06-13 20:20:40 +00006876 n = apply_autocmds(EVENT_FILECHANGEDSHELL,
6877 buf->b_fname, buf->b_fname, FALSE, buf);
Bram Moolenaarbf1b7a72009-03-05 02:15:53 +00006878 --allbuf_lock;
Bram Moolenaar071d4272004-06-13 20:20:40 +00006879 busy = FALSE;
6880 if (n)
6881 {
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02006882 if (!bufref_valid(&bufref))
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006883 emsg(_("E246: FileChangedShell autocommand deleted buffer"));
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +01006884#ifdef FEAT_EVAL
Bram Moolenaar19a09a12005-03-04 23:39:37 +00006885 s = get_vim_var_str(VV_FCS_CHOICE);
6886 if (STRCMP(s, "reload") == 0 && *reason != 'd')
6887 reload = TRUE;
6888 else if (STRCMP(s, "ask") == 0)
6889 n = FALSE;
6890 else
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +01006891#endif
Bram Moolenaar19a09a12005-03-04 23:39:37 +00006892 return 2;
Bram Moolenaar071d4272004-06-13 20:20:40 +00006893 }
Bram Moolenaar19a09a12005-03-04 23:39:37 +00006894 if (!n)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006895 {
Bram Moolenaar19a09a12005-03-04 23:39:37 +00006896 if (*reason == 'd')
6897 mesg = _("E211: File \"%s\" no longer available");
Bram Moolenaar071d4272004-06-13 20:20:40 +00006898 else
6899 {
6900 helpmesg = TRUE;
6901#if defined(FEAT_CON_DIALOG) || defined(FEAT_GUI_DIALOG)
6902 can_reload = TRUE;
6903#endif
Bram Moolenaar19a09a12005-03-04 23:39:37 +00006904 if (reason[2] == 'n')
6905 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00006906 mesg = _("W12: Warning: File \"%s\" has changed and the buffer was changed in Vim as well");
Bram Moolenaar19a09a12005-03-04 23:39:37 +00006907 mesg2 = _("See \":help W12\" for more info.");
6908 }
6909 else if (reason[1] == 'h')
6910 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00006911 mesg = _("W11: Warning: File \"%s\" has changed since editing started");
Bram Moolenaar19a09a12005-03-04 23:39:37 +00006912 mesg2 = _("See \":help W11\" for more info.");
6913 }
6914 else if (*reason == 'm')
6915 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00006916 mesg = _("W16: Warning: Mode of file \"%s\" has changed since editing started");
Bram Moolenaar19a09a12005-03-04 23:39:37 +00006917 mesg2 = _("See \":help W16\" for more info.");
6918 }
Bram Moolenaar85388b52009-06-24 09:58:32 +00006919 else
6920 /* Only timestamp changed, store it to avoid a warning
6921 * in check_mtime() later. */
6922 buf->b_mtime_read = buf->b_mtime;
Bram Moolenaar071d4272004-06-13 20:20:40 +00006923 }
6924 }
6925 }
6926
6927 }
6928 else if ((buf->b_flags & BF_NEW) && !(buf->b_flags & BF_NEW_W)
6929 && vim_fexists(buf->b_ffname))
6930 {
6931 retval = 1;
6932 mesg = _("W13: Warning: File \"%s\" has been created after editing started");
6933 buf->b_flags |= BF_NEW_W;
6934#if defined(FEAT_CON_DIALOG) || defined(FEAT_GUI_DIALOG)
6935 can_reload = TRUE;
6936#endif
6937 }
6938
6939 if (mesg != NULL)
6940 {
6941 path = home_replace_save(buf, buf->b_fname);
6942 if (path != NULL)
6943 {
Bram Moolenaar19a09a12005-03-04 23:39:37 +00006944 if (!helpmesg)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006945 mesg2 = "";
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006946 tbuf = (char *)alloc((unsigned)(STRLEN(path) + STRLEN(mesg)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006947 + STRLEN(mesg2) + 2));
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006948 sprintf(tbuf, mesg, path);
Bram Moolenaar496c5262009-03-18 14:42:00 +00006949#ifdef FEAT_EVAL
6950 /* Set warningmsg here, before the unimportant and output-specific
6951 * mesg2 has been appended. */
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006952 set_vim_var_string(VV_WARNINGMSG, (char_u *)tbuf, -1);
Bram Moolenaar496c5262009-03-18 14:42:00 +00006953#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00006954#if defined(FEAT_CON_DIALOG) || defined(FEAT_GUI_DIALOG)
6955 if (can_reload)
6956 {
6957 if (*mesg2 != NUL)
6958 {
6959 STRCAT(tbuf, "\n");
6960 STRCAT(tbuf, mesg2);
6961 }
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006962 if (do_dialog(VIM_WARNING, (char_u *)_("Warning"),
6963 (char_u *)tbuf,
Bram Moolenaard2c340a2011-01-17 20:08:11 +01006964 (char_u *)_("&OK\n&Load File"), 1, NULL, TRUE) == 2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006965 reload = TRUE;
6966 }
6967 else
6968#endif
6969 if (State > NORMAL_BUSY || (State & CMDLINE) || already_warned)
6970 {
6971 if (*mesg2 != NUL)
6972 {
6973 STRCAT(tbuf, "; ");
6974 STRCAT(tbuf, mesg2);
6975 }
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01006976 emsg(tbuf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00006977 retval = 2;
6978 }
6979 else
6980 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00006981 if (!autocmd_busy)
Bram Moolenaar071d4272004-06-13 20:20:40 +00006982 {
6983 msg_start();
Bram Moolenaar32526b32019-01-19 17:43:09 +01006984 msg_puts_attr(tbuf, HL_ATTR(HLF_E) + MSG_HIST);
Bram Moolenaar071d4272004-06-13 20:20:40 +00006985 if (*mesg2 != NUL)
Bram Moolenaar32526b32019-01-19 17:43:09 +01006986 msg_puts_attr(mesg2, HL_ATTR(HLF_W) + MSG_HIST);
Bram Moolenaar071d4272004-06-13 20:20:40 +00006987 msg_clr_eos();
6988 (void)msg_end();
6989 if (emsg_silent == 0)
6990 {
6991 out_flush();
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +01006992#ifdef FEAT_GUI
Bram Moolenaar071d4272004-06-13 20:20:40 +00006993 if (!focus)
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +01006994#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00006995 /* give the user some time to think about it */
6996 ui_delay(1000L, TRUE);
6997
6998 /* don't redraw and erase the message */
6999 redraw_cmdline = FALSE;
7000 }
7001 }
7002 already_warned = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00007003 }
7004
7005 vim_free(path);
7006 vim_free(tbuf);
7007 }
7008 }
7009
7010 if (reload)
Bram Moolenaar465748e2012-08-29 18:50:54 +02007011 {
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007012 /* Reload the buffer. */
Bram Moolenaar316059c2006-01-14 21:18:42 +00007013 buf_reload(buf, orig_mode);
Bram Moolenaar465748e2012-08-29 18:50:54 +02007014#ifdef FEAT_PERSISTENT_UNDO
7015 if (buf->b_p_udf && buf->b_ffname != NULL)
7016 {
7017 char_u hash[UNDO_HASH_SIZE];
7018 buf_T *save_curbuf = curbuf;
7019
7020 /* Any existing undo file is unusable, write it now. */
7021 curbuf = buf;
7022 u_compute_hash(hash);
7023 u_write_undo(NULL, FALSE, buf, hash);
7024 curbuf = save_curbuf;
7025 }
7026#endif
7027 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00007028
Bram Moolenaar3577c6f2008-06-24 21:16:56 +00007029 /* Trigger FileChangedShell when the file was changed in any way. */
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02007030 if (bufref_valid(&bufref) && retval != 0)
Bram Moolenaar56718732006-03-15 22:53:57 +00007031 (void)apply_autocmds(EVENT_FILECHANGEDSHELLPOST,
7032 buf->b_fname, buf->b_fname, FALSE, buf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00007033#ifdef FEAT_GUI
7034 /* restore this in case an autocommand has set it; it would break
7035 * 'mousefocus' */
7036 need_mouse_correct = save_mouse_correct;
7037#endif
7038
7039 return retval;
7040}
7041
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007042/*
7043 * Reload a buffer that is already loaded.
7044 * Used when the file was changed outside of Vim.
Bram Moolenaar316059c2006-01-14 21:18:42 +00007045 * "orig_mode" is buf->b_orig_mode before the need for reloading was detected.
7046 * buf->b_orig_mode may have been reset already.
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007047 */
7048 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01007049buf_reload(buf_T *buf, int orig_mode)
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007050{
7051 exarg_T ea;
7052 pos_T old_cursor;
7053 linenr_T old_topline;
7054 int old_ro = buf->b_p_ro;
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007055 buf_T *savebuf;
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02007056 bufref_T bufref;
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007057 int saved = OK;
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007058 aco_save_T aco;
Bram Moolenaar59f931e2010-07-24 20:27:03 +02007059 int flags = READ_NEW;
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007060
7061 /* set curwin/curbuf for "buf" and save some things */
7062 aucmd_prepbuf(&aco, buf);
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007063
7064 /* We only want to read the text from the file, not reset the syntax
7065 * highlighting, clear marks, diff status, etc. Force the fileformat
7066 * and encoding to be the same. */
7067 if (prep_exarg(&ea, buf) == OK)
7068 {
7069 old_cursor = curwin->w_cursor;
7070 old_topline = curwin->w_topline;
7071
Bram Moolenaarf9bb7342010-08-04 14:29:54 +02007072 if (p_ur < 0 || curbuf->b_ml.ml_line_count <= p_ur)
Bram Moolenaar59f931e2010-07-24 20:27:03 +02007073 {
7074 /* Save all the text, so that the reload can be undone.
7075 * Sync first so that this is a separate undo-able action. */
7076 u_sync(FALSE);
7077 saved = u_savecommon(0, curbuf->b_ml.ml_line_count + 1, 0, TRUE);
7078 flags |= READ_KEEP_UNDO;
7079 }
7080
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007081 /*
7082 * To behave like when a new file is edited (matters for
7083 * BufReadPost autocommands) we first need to delete the current
7084 * buffer contents. But if reading the file fails we should keep
7085 * the old contents. Can't use memory only, the file might be
7086 * too big. Use a hidden buffer to move the buffer contents to.
7087 */
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01007088 if (BUFEMPTY() || saved == FAIL)
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007089 savebuf = NULL;
7090 else
7091 {
7092 /* Allocate a buffer without putting it in the buffer list. */
7093 savebuf = buflist_new(NULL, NULL, (linenr_T)1, BLN_DUMMY);
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02007094 set_bufref(&bufref, savebuf);
Bram Moolenaar8424a622006-04-19 21:23:36 +00007095 if (savebuf != NULL && buf == curbuf)
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007096 {
7097 /* Open the memline. */
7098 curbuf = savebuf;
7099 curwin->w_buffer = savebuf;
Bram Moolenaar4770d092006-01-12 23:22:24 +00007100 saved = ml_open(curbuf);
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007101 curbuf = buf;
7102 curwin->w_buffer = buf;
7103 }
Bram Moolenaar8424a622006-04-19 21:23:36 +00007104 if (savebuf == NULL || saved == FAIL || buf != curbuf
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007105 || move_lines(buf, savebuf) == FAIL)
7106 {
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007107 semsg(_("E462: Could not prepare for reloading \"%s\""),
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007108 buf->b_fname);
7109 saved = FAIL;
7110 }
7111 }
7112
7113 if (saved == OK)
7114 {
7115 curbuf->b_flags |= BF_CHECK_RO; /* check for RO again */
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007116 keep_filetype = TRUE; /* don't detect 'filetype' */
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007117 if (readfile(buf->b_ffname, buf->b_fname, (linenr_T)0,
7118 (linenr_T)0,
Bram Moolenaare13b9af2017-01-13 22:01:02 +01007119 (linenr_T)MAXLNUM, &ea, flags) != OK)
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007120 {
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +01007121#if defined(FEAT_EVAL)
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007122 if (!aborting())
7123#endif
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007124 semsg(_("E321: Could not reload \"%s\""), buf->b_fname);
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02007125 if (savebuf != NULL && bufref_valid(&bufref) && buf == curbuf)
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007126 {
7127 /* Put the text back from the save buffer. First
7128 * delete any lines that readfile() added. */
Bram Moolenaarb5aedf32017-03-12 18:23:53 +01007129 while (!BUFEMPTY())
Bram Moolenaar8424a622006-04-19 21:23:36 +00007130 if (ml_delete(buf->b_ml.ml_line_count, FALSE) == FAIL)
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007131 break;
7132 (void)move_lines(savebuf, buf);
7133 }
7134 }
Bram Moolenaar59f931e2010-07-24 20:27:03 +02007135 else if (buf == curbuf) /* "buf" still valid */
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007136 {
7137 /* Mark the buffer as unmodified and free undo info. */
7138 unchanged(buf, TRUE);
Bram Moolenaar59f931e2010-07-24 20:27:03 +02007139 if ((flags & READ_KEEP_UNDO) == 0)
7140 {
7141 u_blockfree(buf);
7142 u_clearall(buf);
7143 }
7144 else
Bram Moolenaarf9bb7342010-08-04 14:29:54 +02007145 {
Bram Moolenaar59f931e2010-07-24 20:27:03 +02007146 /* Mark all undo states as changed. */
7147 u_unchanged(curbuf);
Bram Moolenaarf9bb7342010-08-04 14:29:54 +02007148 }
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007149 }
7150 }
7151 vim_free(ea.cmd);
7152
Bram Moolenaar7c0a2f32016-07-10 22:11:16 +02007153 if (savebuf != NULL && bufref_valid(&bufref))
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007154 wipe_buffer(savebuf, FALSE);
7155
7156#ifdef FEAT_DIFF
7157 /* Invalidate diff info if necessary. */
Bram Moolenaar8424a622006-04-19 21:23:36 +00007158 diff_invalidate(curbuf);
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007159#endif
7160
7161 /* Restore the topline and cursor position and check it (lines may
7162 * have been removed). */
7163 if (old_topline > curbuf->b_ml.ml_line_count)
7164 curwin->w_topline = curbuf->b_ml.ml_line_count;
7165 else
7166 curwin->w_topline = old_topline;
7167 curwin->w_cursor = old_cursor;
7168 check_cursor();
7169 update_topline();
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007170 keep_filetype = FALSE;
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007171#ifdef FEAT_FOLDING
7172 {
Bram Moolenaarbd1e5d22009-04-29 09:02:44 +00007173 win_T *wp;
7174 tabpage_T *tp;
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007175
7176 /* Update folds unless they are defined manually. */
Bram Moolenaarbd1e5d22009-04-29 09:02:44 +00007177 FOR_ALL_TAB_WINDOWS(tp, wp)
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007178 if (wp->w_buffer == curwin->w_buffer
7179 && !foldmethodIsManual(wp))
7180 foldUpdateAll(wp);
7181 }
7182#endif
7183 /* If the mode didn't change and 'readonly' was set, keep the old
7184 * value; the user probably used the ":view" command. But don't
7185 * reset it, might have had a read error. */
7186 if (orig_mode == curbuf->b_orig_mode)
7187 curbuf->b_p_ro |= old_ro;
Bram Moolenaar52f85b72013-01-30 14:13:56 +01007188
7189 /* Modelines must override settings done by autocommands. */
7190 do_modelines(0);
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007191 }
7192
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007193 /* restore curwin/curbuf and a few other things */
7194 aucmd_restbuf(&aco);
7195 /* Careful: autocommands may have made "buf" invalid! */
Bram Moolenaar631d6f62005-06-07 21:02:10 +00007196}
7197
Bram Moolenaar071d4272004-06-13 20:20:40 +00007198 void
Bram Moolenaar8767f522016-07-01 17:17:39 +02007199buf_store_time(buf_T *buf, stat_T *st, char_u *fname UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00007200{
7201 buf->b_mtime = (long)st->st_mtime;
Bram Moolenaar914703b2010-05-31 21:59:46 +02007202 buf->b_orig_size = st->st_size;
Bram Moolenaar071d4272004-06-13 20:20:40 +00007203#ifdef HAVE_ST_MODE
7204 buf->b_orig_mode = (int)st->st_mode;
7205#else
7206 buf->b_orig_mode = mch_getperm(fname);
7207#endif
7208}
7209
7210/*
7211 * Adjust the line with missing eol, used for the next write.
7212 * Used for do_filter(), when the input lines for the filter are deleted.
7213 */
7214 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01007215write_lnum_adjust(linenr_T offset)
Bram Moolenaar071d4272004-06-13 20:20:40 +00007216{
Bram Moolenaarcab35ad2011-02-15 17:39:22 +01007217 if (curbuf->b_no_eol_lnum != 0) /* only if there is a missing eol */
7218 curbuf->b_no_eol_lnum += offset;
Bram Moolenaar071d4272004-06-13 20:20:40 +00007219}
7220
Bram Moolenaarda440d22016-01-16 21:27:23 +01007221#if defined(TEMPDIRNAMES) || defined(FEAT_EVAL) || defined(PROTO)
7222/*
7223 * Delete "name" and everything in it, recursively.
7224 * return 0 for succes, -1 if some file was not deleted.
7225 */
7226 int
7227delete_recursive(char_u *name)
7228{
7229 int result = 0;
7230 char_u **files;
7231 int file_count;
7232 int i;
7233 char_u *exp;
7234
Bram Moolenaar43a34f92016-01-17 15:56:34 +01007235 /* A symbolic link to a directory itself is deleted, not the directory it
7236 * points to. */
7237 if (
Bram Moolenaar203258c2016-01-17 22:15:16 +01007238# if defined(UNIX) || defined(WIN32)
Bram Moolenaar43a34f92016-01-17 15:56:34 +01007239 mch_isrealdir(name)
Bram Moolenaar203258c2016-01-17 22:15:16 +01007240# else
Bram Moolenaar43a34f92016-01-17 15:56:34 +01007241 mch_isdir(name)
Bram Moolenaar43a34f92016-01-17 15:56:34 +01007242# endif
7243 )
Bram Moolenaarda440d22016-01-16 21:27:23 +01007244 {
7245 vim_snprintf((char *)NameBuff, MAXPATHL, "%s/*", name);
7246 exp = vim_strsave(NameBuff);
7247 if (exp == NULL)
7248 return -1;
7249 if (gen_expand_wildcards(1, &exp, &file_count, &files,
Bram Moolenaar336bd622016-01-17 18:23:58 +01007250 EW_DIR|EW_FILE|EW_SILENT|EW_ALLLINKS|EW_DODOT|EW_EMPTYOK) == OK)
Bram Moolenaarda440d22016-01-16 21:27:23 +01007251 {
7252 for (i = 0; i < file_count; ++i)
7253 if (delete_recursive(files[i]) != 0)
7254 result = -1;
7255 FreeWild(file_count, files);
7256 }
7257 else
7258 result = -1;
7259 vim_free(exp);
7260 (void)mch_rmdir(name);
7261 }
7262 else
7263 result = mch_remove(name) == 0 ? 0 : -1;
7264
7265 return result;
7266}
7267#endif
7268
Bram Moolenaar071d4272004-06-13 20:20:40 +00007269#if defined(TEMPDIRNAMES) || defined(PROTO)
7270static long temp_count = 0; /* Temp filename counter. */
7271
7272/*
7273 * Delete the temp directory and all files it contains.
7274 */
7275 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01007276vim_deltempdir(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00007277{
Bram Moolenaar071d4272004-06-13 20:20:40 +00007278 if (vim_tempdir != NULL)
7279 {
Bram Moolenaarda440d22016-01-16 21:27:23 +01007280 /* remove the trailing path separator */
7281 gettail(vim_tempdir)[-1] = NUL;
7282 delete_recursive(vim_tempdir);
Bram Moolenaard23a8232018-02-10 18:45:26 +01007283 VIM_CLEAR(vim_tempdir);
Bram Moolenaar071d4272004-06-13 20:20:40 +00007284 }
7285}
Bram Moolenaar071d4272004-06-13 20:20:40 +00007286
7287/*
Bram Moolenaareaf03392009-11-17 11:08:52 +00007288 * Directory "tempdir" was created. Expand this name to a full path and put
7289 * it in "vim_tempdir". This avoids that using ":cd" would confuse us.
7290 * "tempdir" must be no longer than MAXPATHL.
7291 */
7292 static void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01007293vim_settempdir(char_u *tempdir)
Bram Moolenaareaf03392009-11-17 11:08:52 +00007294{
7295 char_u *buf;
7296
7297 buf = alloc((unsigned)MAXPATHL + 2);
7298 if (buf != NULL)
7299 {
7300 if (vim_FullName(tempdir, buf, MAXPATHL, FALSE) == FAIL)
7301 STRCPY(buf, tempdir);
Bram Moolenaara06ecab2016-07-16 14:47:36 +02007302 add_pathsep(buf);
Bram Moolenaareaf03392009-11-17 11:08:52 +00007303 vim_tempdir = vim_strsave(buf);
7304 vim_free(buf);
7305 }
7306}
Bram Moolenaar4592dee2009-11-18 19:11:58 +00007307#endif
Bram Moolenaareaf03392009-11-17 11:08:52 +00007308
7309/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00007310 * vim_tempname(): Return a unique name that can be used for a temp file.
7311 *
Bram Moolenaar76ae22f2016-06-13 20:00:29 +02007312 * The temp file is NOT guaranteed to be created. If "keep" is FALSE it is
7313 * guaranteed to NOT be created.
Bram Moolenaar071d4272004-06-13 20:20:40 +00007314 *
7315 * The returned pointer is to allocated memory.
7316 * The returned pointer is NULL if no valid name was found.
7317 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00007318 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01007319vim_tempname(
7320 int extra_char UNUSED, /* char to use in the name instead of '?' */
7321 int keep UNUSED)
Bram Moolenaar071d4272004-06-13 20:20:40 +00007322{
7323#ifdef USE_TMPNAM
7324 char_u itmp[L_tmpnam]; /* use tmpnam() */
7325#else
7326 char_u itmp[TEMPNAMELEN];
7327#endif
7328
7329#ifdef TEMPDIRNAMES
7330 static char *(tempdirs[]) = {TEMPDIRNAMES};
7331 int i;
Bram Moolenaar071d4272004-06-13 20:20:40 +00007332# ifndef EEXIST
Bram Moolenaar8767f522016-07-01 17:17:39 +02007333 stat_T st;
Bram Moolenaar071d4272004-06-13 20:20:40 +00007334# endif
7335
7336 /*
7337 * This will create a directory for private use by this instance of Vim.
7338 * This is done once, and the same directory is used for all temp files.
7339 * This method avoids security problems because of symlink attacks et al.
7340 * It's also a bit faster, because we only need to check for an existing
7341 * file when creating the directory and not for each temp file.
7342 */
7343 if (vim_tempdir == NULL)
7344 {
7345 /*
7346 * Try the entries in TEMPDIRNAMES to create the temp directory.
7347 */
Bram Moolenaar78a15312009-05-15 19:33:18 +00007348 for (i = 0; i < (int)(sizeof(tempdirs) / sizeof(char *)); ++i)
Bram Moolenaar071d4272004-06-13 20:20:40 +00007349 {
Bram Moolenaareaf03392009-11-17 11:08:52 +00007350# ifndef HAVE_MKDTEMP
Bram Moolenaar2660c0e2010-01-19 14:59:56 +01007351 size_t itmplen;
Bram Moolenaareaf03392009-11-17 11:08:52 +00007352 long nr;
7353 long off;
7354# endif
7355
Bram Moolenaare1a61992015-12-03 21:02:27 +01007356 /* Expand $TMP, leave room for "/v1100000/999999999".
7357 * Skip the directory check if the expansion fails. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00007358 expand_env((char_u *)tempdirs[i], itmp, TEMPNAMELEN - 20);
Bram Moolenaare1a61992015-12-03 21:02:27 +01007359 if (itmp[0] != '$' && mch_isdir(itmp))
Bram Moolenaar071d4272004-06-13 20:20:40 +00007360 {
Bram Moolenaare1a61992015-12-03 21:02:27 +01007361 /* directory exists */
Bram Moolenaara06ecab2016-07-16 14:47:36 +02007362 add_pathsep(itmp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00007363
Bram Moolenaareaf03392009-11-17 11:08:52 +00007364# ifdef HAVE_MKDTEMP
Bram Moolenaar35d88f42016-06-04 14:52:00 +02007365 {
7366# if defined(UNIX) || defined(VMS)
7367 /* Make sure the umask doesn't remove the executable bit.
7368 * "repl" has been reported to use "177". */
7369 mode_t umask_save = umask(077);
7370# endif
7371 /* Leave room for filename */
7372 STRCAT(itmp, "vXXXXXX");
7373 if (mkdtemp((char *)itmp) != NULL)
7374 vim_settempdir(itmp);
7375# if defined(UNIX) || defined(VMS)
7376 (void)umask(umask_save);
7377# endif
7378 }
Bram Moolenaareaf03392009-11-17 11:08:52 +00007379# else
Bram Moolenaar071d4272004-06-13 20:20:40 +00007380 /* Get an arbitrary number of up to 6 digits. When it's
7381 * unlikely that it already exists it will be faster,
7382 * otherwise it doesn't matter. The use of mkdir() avoids any
7383 * security problems because of the predictable number. */
7384 nr = (mch_get_pid() + (long)time(NULL)) % 1000000L;
Bram Moolenaar2660c0e2010-01-19 14:59:56 +01007385 itmplen = STRLEN(itmp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00007386
7387 /* Try up to 10000 different values until we find a name that
7388 * doesn't exist. */
7389 for (off = 0; off < 10000L; ++off)
7390 {
7391 int r;
Bram Moolenaareaf03392009-11-17 11:08:52 +00007392# if defined(UNIX) || defined(VMS)
Bram Moolenaar071d4272004-06-13 20:20:40 +00007393 mode_t umask_save;
Bram Moolenaareaf03392009-11-17 11:08:52 +00007394# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00007395
Bram Moolenaareaf03392009-11-17 11:08:52 +00007396 sprintf((char *)itmp + itmplen, "v%ld", nr + off);
7397# ifndef EEXIST
Bram Moolenaar071d4272004-06-13 20:20:40 +00007398 /* If mkdir() does not set errno to EEXIST, check for
7399 * existing file here. There is a race condition then,
7400 * although it's fail-safe. */
7401 if (mch_stat((char *)itmp, &st) >= 0)
7402 continue;
Bram Moolenaareaf03392009-11-17 11:08:52 +00007403# endif
7404# if defined(UNIX) || defined(VMS)
Bram Moolenaar071d4272004-06-13 20:20:40 +00007405 /* Make sure the umask doesn't remove the executable bit.
7406 * "repl" has been reported to use "177". */
7407 umask_save = umask(077);
Bram Moolenaareaf03392009-11-17 11:08:52 +00007408# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00007409 r = vim_mkdir(itmp, 0700);
Bram Moolenaareaf03392009-11-17 11:08:52 +00007410# if defined(UNIX) || defined(VMS)
Bram Moolenaar071d4272004-06-13 20:20:40 +00007411 (void)umask(umask_save);
Bram Moolenaareaf03392009-11-17 11:08:52 +00007412# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00007413 if (r == 0)
7414 {
Bram Moolenaareaf03392009-11-17 11:08:52 +00007415 vim_settempdir(itmp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00007416 break;
7417 }
Bram Moolenaareaf03392009-11-17 11:08:52 +00007418# ifdef EEXIST
Bram Moolenaar071d4272004-06-13 20:20:40 +00007419 /* If the mkdir() didn't fail because the file/dir exists,
7420 * we probably can't create any dir here, try another
7421 * place. */
7422 if (errno != EEXIST)
Bram Moolenaareaf03392009-11-17 11:08:52 +00007423# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00007424 break;
7425 }
Bram Moolenaareaf03392009-11-17 11:08:52 +00007426# endif /* HAVE_MKDTEMP */
Bram Moolenaar071d4272004-06-13 20:20:40 +00007427 if (vim_tempdir != NULL)
7428 break;
7429 }
7430 }
7431 }
7432
7433 if (vim_tempdir != NULL)
7434 {
7435 /* There is no need to check if the file exists, because we own the
7436 * directory and nobody else creates a file in it. */
7437 sprintf((char *)itmp, "%s%ld", vim_tempdir, temp_count++);
7438 return vim_strsave(itmp);
7439 }
7440
7441 return NULL;
7442
7443#else /* TEMPDIRNAMES */
7444
7445# ifdef WIN3264
7446 char szTempFile[_MAX_PATH + 1];
7447 char buf4[4];
7448 char_u *retval;
7449 char_u *p;
7450
7451 STRCPY(itmp, "");
7452 if (GetTempPath(_MAX_PATH, szTempFile) == 0)
Bram Moolenaarb1891912011-02-09 14:47:03 +01007453 {
7454 szTempFile[0] = '.'; /* GetTempPath() failed, use current dir */
7455 szTempFile[1] = NUL;
7456 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00007457 strcpy(buf4, "VIM");
7458 buf4[2] = extra_char; /* make it "VIa", "VIb", etc. */
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01007459 if (GetTempFileName(szTempFile, buf4, 0, (LPSTR)itmp) == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00007460 return NULL;
Bram Moolenaare5c421c2015-03-31 13:33:08 +02007461 if (!keep)
7462 /* GetTempFileName() will create the file, we don't want that */
Bram Moolenaar6aa2cd42016-02-16 15:06:59 +01007463 (void)DeleteFile((LPSTR)itmp);
Bram Moolenaar071d4272004-06-13 20:20:40 +00007464
7465 /* Backslashes in a temp file name cause problems when filtering with
7466 * "sh". NOTE: This also checks 'shellcmdflag' to help those people who
7467 * didn't set 'shellslash'. */
7468 retval = vim_strsave(itmp);
7469 if (*p_shcf == '-' || p_ssl)
7470 for (p = retval; *p; ++p)
7471 if (*p == '\\')
7472 *p = '/';
Bram Moolenaar00362012019-02-08 12:21:30 +01007473
7474#if defined(FEAT_MBYTE) && defined(WIN3264)
7475 if (enc_utf8)
7476 {
7477 int len;
7478 char_u *pp = NULL;
7479
7480 // Convert from active codepage to UTF-8 since mch_call_shell()
7481 // converts command-line to wide string from encoding.
7482 acp_to_enc(retval, (int)STRLEN(retval), &pp, &len);
7483 if (pp != NULL)
7484 {
7485 vim_free(retval);
7486 return pp;
7487 }
7488 }
7489#endif
7490
Bram Moolenaar071d4272004-06-13 20:20:40 +00007491 return retval;
7492
7493# else /* WIN3264 */
7494
7495# ifdef USE_TMPNAM
Bram Moolenaar95474ca2011-02-09 16:44:51 +01007496 char_u *p;
7497
Bram Moolenaar071d4272004-06-13 20:20:40 +00007498 /* tmpnam() will make its own name */
Bram Moolenaar95474ca2011-02-09 16:44:51 +01007499 p = tmpnam((char *)itmp);
7500 if (p == NULL || *p == NUL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00007501 return NULL;
7502# else
7503 char_u *p;
7504
7505# ifdef VMS_TEMPNAM
7506 /* mktemp() is not working on VMS. It seems to be
7507 * a do-nothing function. Therefore we use tempnam().
7508 */
7509 sprintf((char *)itmp, "VIM%c", extra_char);
7510 p = (char_u *)tempnam("tmp:", (char *)itmp);
7511 if (p != NULL)
7512 {
Bram Moolenaar206f0112014-03-12 16:51:55 +01007513 /* VMS will use '.LIS' if we don't explicitly specify an extension,
Bram Moolenaar071d4272004-06-13 20:20:40 +00007514 * and VIM will then be unable to find the file later */
7515 STRCPY(itmp, p);
7516 STRCAT(itmp, ".txt");
7517 free(p);
7518 }
7519 else
7520 return NULL;
7521# else
7522 STRCPY(itmp, TEMPNAME);
7523 if ((p = vim_strchr(itmp, '?')) != NULL)
7524 *p = extra_char;
7525 if (mktemp((char *)itmp) == NULL)
7526 return NULL;
7527# endif
7528# endif
7529
7530 return vim_strsave(itmp);
7531# endif /* WIN3264 */
7532#endif /* TEMPDIRNAMES */
7533}
7534
7535#if defined(BACKSLASH_IN_FILENAME) || defined(PROTO)
7536/*
Bram Moolenaarb4f6a462015-10-13 19:43:17 +02007537 * Convert all backslashes in fname to forward slashes in-place, unless when
7538 * it looks like a URL.
Bram Moolenaar071d4272004-06-13 20:20:40 +00007539 */
7540 void
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01007541forward_slash(char_u *fname)
Bram Moolenaar071d4272004-06-13 20:20:40 +00007542{
7543 char_u *p;
7544
Bram Moolenaarb4f6a462015-10-13 19:43:17 +02007545 if (path_with_url(fname))
7546 return;
Bram Moolenaar071d4272004-06-13 20:20:40 +00007547 for (p = fname; *p != NUL; ++p)
Bram Moolenaar071d4272004-06-13 20:20:40 +00007548 /* The Big5 encoding can have '\' in the trail byte. */
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00007549 if (enc_dbcs != 0 && (*mb_ptr2len)(p) > 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00007550 ++p;
Bram Moolenaar13505972019-01-24 15:04:48 +01007551 else if (*p == '\\')
Bram Moolenaar071d4272004-06-13 20:20:40 +00007552 *p = '/';
7553}
7554#endif
7555
Bram Moolenaarf30e74c2006-08-16 17:35:00 +00007556/*
Bram Moolenaar748bf032005-02-02 23:04:36 +00007557 * Try matching a filename with a "pattern" ("prog" is NULL), or use the
7558 * precompiled regprog "prog" ("pattern" is NULL). That avoids calling
7559 * vim_regcomp() often.
Bram Moolenaar071d4272004-06-13 20:20:40 +00007560 * Used for autocommands and 'wildignore'.
7561 * Returns TRUE if there is a match, FALSE otherwise.
7562 */
Bram Moolenaar3e460fd2019-01-26 16:21:07 +01007563 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01007564match_file_pat(
7565 char_u *pattern, /* pattern to match with */
7566 regprog_T **prog, /* pre-compiled regprog or NULL */
7567 char_u *fname, /* full path of file name */
7568 char_u *sfname, /* short file name or NULL */
7569 char_u *tail, /* tail of path */
7570 int allow_dirs) /* allow matching with dir */
Bram Moolenaar071d4272004-06-13 20:20:40 +00007571{
7572 regmatch_T regmatch;
7573 int result = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00007574
Bram Moolenaar71afbfe2013-03-19 16:49:16 +01007575 regmatch.rm_ic = p_fic; /* ignore case if 'fileignorecase' is set */
Bram Moolenaar49a6ed82015-01-07 14:43:39 +01007576 if (prog != NULL)
7577 regmatch.regprog = *prog;
Bram Moolenaar071d4272004-06-13 20:20:40 +00007578 else
Bram Moolenaar49a6ed82015-01-07 14:43:39 +01007579 regmatch.regprog = vim_regcomp(pattern, RE_MAGIC);
Bram Moolenaar071d4272004-06-13 20:20:40 +00007580
7581 /*
7582 * Try for a match with the pattern with:
7583 * 1. the full file name, when the pattern has a '/'.
7584 * 2. the short file name, when the pattern has a '/'.
7585 * 3. the tail of the file name, when the pattern has no '/'.
7586 */
Bram Moolenaar49a6ed82015-01-07 14:43:39 +01007587 if (regmatch.regprog != NULL
Bram Moolenaar071d4272004-06-13 20:20:40 +00007588 && ((allow_dirs
7589 && (vim_regexec(&regmatch, fname, (colnr_T)0)
7590 || (sfname != NULL
7591 && vim_regexec(&regmatch, sfname, (colnr_T)0))))
Bram Moolenaar49a6ed82015-01-07 14:43:39 +01007592 || (!allow_dirs && vim_regexec(&regmatch, tail, (colnr_T)0))))
Bram Moolenaar071d4272004-06-13 20:20:40 +00007593 result = TRUE;
7594
Bram Moolenaardffa5b82014-11-19 16:38:07 +01007595 if (prog != NULL)
7596 *prog = regmatch.regprog;
7597 else
Bram Moolenaar473de612013-06-08 18:19:48 +02007598 vim_regfree(regmatch.regprog);
Bram Moolenaar071d4272004-06-13 20:20:40 +00007599 return result;
7600}
Bram Moolenaar071d4272004-06-13 20:20:40 +00007601
7602#if defined(FEAT_WILDIGN) || defined(PROTO)
7603/*
7604 * Return TRUE if a file matches with a pattern in "list".
7605 * "list" is a comma-separated list of patterns, like 'wildignore'.
7606 * "sfname" is the short file name or NULL, "ffname" the long file name.
7607 */
7608 int
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01007609match_file_list(char_u *list, char_u *sfname, char_u *ffname)
Bram Moolenaar071d4272004-06-13 20:20:40 +00007610{
7611 char_u buf[100];
7612 char_u *tail;
7613 char_u *regpat;
7614 char allow_dirs;
7615 int match;
7616 char_u *p;
7617
7618 tail = gettail(sfname);
7619
7620 /* try all patterns in 'wildignore' */
7621 p = list;
7622 while (*p)
7623 {
7624 copy_option_part(&p, buf, 100, ",");
7625 regpat = file_pat_to_reg_pat(buf, NULL, &allow_dirs, FALSE);
7626 if (regpat == NULL)
7627 break;
Bram Moolenaar748bf032005-02-02 23:04:36 +00007628 match = match_file_pat(regpat, NULL, ffname, sfname,
7629 tail, (int)allow_dirs);
Bram Moolenaar071d4272004-06-13 20:20:40 +00007630 vim_free(regpat);
7631 if (match)
7632 return TRUE;
7633 }
7634 return FALSE;
7635}
7636#endif
7637
7638/*
7639 * Convert the given pattern "pat" which has shell style wildcards in it, into
7640 * a regular expression, and return the result in allocated memory. If there
7641 * is a directory path separator to be matched, then TRUE is put in
7642 * allow_dirs, otherwise FALSE is put there -- webb.
7643 * Handle backslashes before special characters, like "\*" and "\ ".
7644 *
Bram Moolenaar071d4272004-06-13 20:20:40 +00007645 * Returns NULL when out of memory.
7646 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00007647 char_u *
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01007648file_pat_to_reg_pat(
7649 char_u *pat,
7650 char_u *pat_end, /* first char after pattern or NULL */
7651 char *allow_dirs, /* Result passed back out in here */
7652 int no_bslash UNUSED) /* Don't use a backward slash as pathsep */
Bram Moolenaar071d4272004-06-13 20:20:40 +00007653{
Bram Moolenaar49a6ed82015-01-07 14:43:39 +01007654 int size = 2; /* '^' at start, '$' at end */
Bram Moolenaar071d4272004-06-13 20:20:40 +00007655 char_u *endp;
7656 char_u *reg_pat;
7657 char_u *p;
7658 int i;
7659 int nested = 0;
7660 int add_dollar = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00007661
7662 if (allow_dirs != NULL)
7663 *allow_dirs = FALSE;
7664 if (pat_end == NULL)
7665 pat_end = pat + STRLEN(pat);
7666
Bram Moolenaar071d4272004-06-13 20:20:40 +00007667 for (p = pat; p < pat_end; p++)
7668 {
7669 switch (*p)
7670 {
7671 case '*':
7672 case '.':
7673 case ',':
7674 case '{':
7675 case '}':
7676 case '~':
7677 size += 2; /* extra backslash */
7678 break;
7679#ifdef BACKSLASH_IN_FILENAME
7680 case '\\':
7681 case '/':
7682 size += 4; /* could become "[\/]" */
7683 break;
7684#endif
7685 default:
7686 size++;
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00007687 if (enc_dbcs != 0 && (*mb_ptr2len)(p) > 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00007688 {
7689 ++p;
7690 ++size;
7691 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00007692 break;
7693 }
7694 }
7695 reg_pat = alloc(size + 1);
7696 if (reg_pat == NULL)
7697 return NULL;
7698
Bram Moolenaar071d4272004-06-13 20:20:40 +00007699 i = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00007700
7701 if (pat[0] == '*')
7702 while (pat[0] == '*' && pat < pat_end - 1)
7703 pat++;
7704 else
7705 reg_pat[i++] = '^';
7706 endp = pat_end - 1;
Bram Moolenaar8fee8782015-08-11 18:45:48 +02007707 if (endp >= pat && *endp == '*')
Bram Moolenaar071d4272004-06-13 20:20:40 +00007708 {
7709 while (endp - pat > 0 && *endp == '*')
7710 endp--;
7711 add_dollar = FALSE;
7712 }
7713 for (p = pat; *p && nested >= 0 && p <= endp; p++)
7714 {
7715 switch (*p)
7716 {
7717 case '*':
7718 reg_pat[i++] = '.';
7719 reg_pat[i++] = '*';
Bram Moolenaar02743632005-07-25 20:42:36 +00007720 while (p[1] == '*') /* "**" matches like "*" */
7721 ++p;
Bram Moolenaar071d4272004-06-13 20:20:40 +00007722 break;
7723 case '.':
Bram Moolenaar071d4272004-06-13 20:20:40 +00007724 case '~':
7725 reg_pat[i++] = '\\';
7726 reg_pat[i++] = *p;
7727 break;
7728 case '?':
Bram Moolenaar071d4272004-06-13 20:20:40 +00007729 reg_pat[i++] = '.';
7730 break;
7731 case '\\':
7732 if (p[1] == NUL)
7733 break;
7734#ifdef BACKSLASH_IN_FILENAME
7735 if (!no_bslash)
7736 {
7737 /* translate:
7738 * "\x" to "\\x" e.g., "dir\file"
7739 * "\*" to "\\.*" e.g., "dir\*.c"
7740 * "\?" to "\\." e.g., "dir\??.c"
7741 * "\+" to "\+" e.g., "fileX\+.c"
7742 */
7743 if ((vim_isfilec(p[1]) || p[1] == '*' || p[1] == '?')
7744 && p[1] != '+')
7745 {
7746 reg_pat[i++] = '[';
7747 reg_pat[i++] = '\\';
7748 reg_pat[i++] = '/';
7749 reg_pat[i++] = ']';
7750 if (allow_dirs != NULL)
7751 *allow_dirs = TRUE;
7752 break;
7753 }
7754 }
7755#endif
Bram Moolenaar8cd213c2010-06-01 21:57:09 +02007756 /* Undo escaping from ExpandEscape():
7757 * foo\?bar -> foo?bar
7758 * foo\%bar -> foo%bar
7759 * foo\,bar -> foo,bar
7760 * foo\ bar -> foo bar
7761 * Don't unescape \, * and others that are also special in a
Bram Moolenaarf4e11432013-07-03 16:53:03 +02007762 * regexp.
7763 * An escaped { must be unescaped since we use magic not
Bram Moolenaara946afe2013-08-02 15:22:39 +02007764 * verymagic. Use "\\\{n,m\}"" to get "\{n,m}".
Bram Moolenaarf4e11432013-07-03 16:53:03 +02007765 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00007766 if (*++p == '?'
7767#ifdef BACKSLASH_IN_FILENAME
7768 && no_bslash
7769#endif
7770 )
7771 reg_pat[i++] = '?';
7772 else
Bram Moolenaarf4e11432013-07-03 16:53:03 +02007773 if (*p == ',' || *p == '%' || *p == '#'
Bram Moolenaar2288afe2015-08-11 16:20:05 +02007774 || vim_isspace(*p) || *p == '{' || *p == '}')
Bram Moolenaar8cd213c2010-06-01 21:57:09 +02007775 reg_pat[i++] = *p;
Bram Moolenaara946afe2013-08-02 15:22:39 +02007776 else if (*p == '\\' && p[1] == '\\' && p[2] == '{')
7777 {
7778 reg_pat[i++] = '\\';
7779 reg_pat[i++] = '{';
7780 p += 2;
7781 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00007782 else
7783 {
7784 if (allow_dirs != NULL && vim_ispathsep(*p)
7785#ifdef BACKSLASH_IN_FILENAME
7786 && (!no_bslash || *p != '\\')
7787#endif
7788 )
7789 *allow_dirs = TRUE;
7790 reg_pat[i++] = '\\';
7791 reg_pat[i++] = *p;
7792 }
7793 break;
7794#ifdef BACKSLASH_IN_FILENAME
7795 case '/':
7796 reg_pat[i++] = '[';
7797 reg_pat[i++] = '\\';
7798 reg_pat[i++] = '/';
7799 reg_pat[i++] = ']';
7800 if (allow_dirs != NULL)
7801 *allow_dirs = TRUE;
7802 break;
7803#endif
7804 case '{':
7805 reg_pat[i++] = '\\';
7806 reg_pat[i++] = '(';
7807 nested++;
7808 break;
7809 case '}':
7810 reg_pat[i++] = '\\';
7811 reg_pat[i++] = ')';
7812 --nested;
7813 break;
7814 case ',':
7815 if (nested)
7816 {
7817 reg_pat[i++] = '\\';
7818 reg_pat[i++] = '|';
7819 }
7820 else
7821 reg_pat[i++] = ',';
7822 break;
7823 default:
Bram Moolenaar0fa313a2005-08-10 21:07:57 +00007824 if (enc_dbcs != 0 && (*mb_ptr2len)(p) > 1)
Bram Moolenaar071d4272004-06-13 20:20:40 +00007825 reg_pat[i++] = *p++;
Bram Moolenaar13505972019-01-24 15:04:48 +01007826 else if (allow_dirs != NULL && vim_ispathsep(*p))
Bram Moolenaar071d4272004-06-13 20:20:40 +00007827 *allow_dirs = TRUE;
7828 reg_pat[i++] = *p;
7829 break;
7830 }
7831 }
7832 if (add_dollar)
7833 reg_pat[i++] = '$';
7834 reg_pat[i] = NUL;
7835 if (nested != 0)
7836 {
7837 if (nested < 0)
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007838 emsg(_("E219: Missing {."));
Bram Moolenaar071d4272004-06-13 20:20:40 +00007839 else
Bram Moolenaarf9e3e092019-01-13 23:38:42 +01007840 emsg(_("E220: Missing }."));
Bram Moolenaard23a8232018-02-10 18:45:26 +01007841 VIM_CLEAR(reg_pat);
Bram Moolenaar071d4272004-06-13 20:20:40 +00007842 }
7843 return reg_pat;
7844}
Bram Moolenaar540fc6f2010-12-17 16:27:16 +01007845
7846#if defined(EINTR) || defined(PROTO)
7847/*
7848 * Version of read() that retries when interrupted by EINTR (possibly
7849 * by a SIGWINCH).
7850 */
7851 long
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01007852read_eintr(int fd, void *buf, size_t bufsize)
Bram Moolenaar540fc6f2010-12-17 16:27:16 +01007853{
7854 long ret;
7855
7856 for (;;)
7857 {
7858 ret = vim_read(fd, buf, bufsize);
7859 if (ret >= 0 || errno != EINTR)
7860 break;
7861 }
7862 return ret;
7863}
7864
7865/*
7866 * Version of write() that retries when interrupted by EINTR (possibly
7867 * by a SIGWINCH).
7868 */
7869 long
Bram Moolenaar78c0b7d2016-01-30 15:52:46 +01007870write_eintr(int fd, void *buf, size_t bufsize)
Bram Moolenaar540fc6f2010-12-17 16:27:16 +01007871{
7872 long ret = 0;
7873 long wlen;
7874
7875 /* Repeat the write() so long it didn't fail, other than being interrupted
7876 * by a signal. */
7877 while (ret < (long)bufsize)
7878 {
Bram Moolenaar9c263032010-12-17 18:06:06 +01007879 wlen = vim_write(fd, (char *)buf + ret, bufsize - ret);
Bram Moolenaar540fc6f2010-12-17 16:27:16 +01007880 if (wlen < 0)
7881 {
7882 if (errno != EINTR)
7883 break;
7884 }
7885 else
7886 ret += wlen;
7887 }
7888 return ret;
7889}
7890#endif