blob: e88c7b8384105d7bd2caa6e3bbb7667c3d1b6a6b [file] [log] [blame]
Bram Moolenaar473952e2019-09-28 16:30:04 +02001/* vi:set ts=8 sts=4 sw=4 noet:
2 *
3 * VIM - Vi IMproved by Bram Moolenaar
4 *
5 * Do ":help uganda" in Vim to read copying and usage conditions.
6 * Do ":help credits" in Vim to see a list of people who contributed.
7 * See README.txt for an overview of the Vim source code.
8 */
9
10/*
11 * bufwrite.c: functions for writing a buffer
12 */
13
14#include "vim.h"
15
16#if defined(HAVE_UTIME) && defined(HAVE_UTIME_H)
17# include <utime.h> // for struct utimbuf
18#endif
19
20#define SMALLBUFSIZE 256 // size of emergency write buffer
21
22/*
23 * Structure to pass arguments from buf_write() to buf_write_bytes().
24 */
25struct bw_info
26{
27 int bw_fd; // file descriptor
28 char_u *bw_buf; // buffer with data to be written
29 int bw_len; // length of data
30 int bw_flags; // FIO_ flags
31#ifdef FEAT_CRYPT
32 buf_T *bw_buffer; // buffer being written
Bram Moolenaar6ed545e2022-05-09 20:09:23 +010033 int bw_finish; // finish encrypting
Bram Moolenaar473952e2019-09-28 16:30:04 +020034#endif
35 char_u bw_rest[CONV_RESTLEN]; // not converted bytes
36 int bw_restlen; // nr of bytes in bw_rest[]
37 int bw_first; // first write call
38 char_u *bw_conv_buf; // buffer for writing converted chars
39 size_t bw_conv_buflen; // size of bw_conv_buf
40 int bw_conv_error; // set for conversion error
41 linenr_T bw_conv_error_lnum; // first line with error or zero
Bram Moolenaar6ed545e2022-05-09 20:09:23 +010042 linenr_T bw_start_lnum; // line number at start of buffer
Bram Moolenaar473952e2019-09-28 16:30:04 +020043#ifdef USE_ICONV
44 iconv_t bw_iconv_fd; // descriptor for iconv() or -1
45#endif
46};
47
48/*
49 * Convert a Unicode character to bytes.
50 * Return TRUE for an error, FALSE when it's OK.
51 */
52 static int
53ucs2bytes(
54 unsigned c, // in: character
55 char_u **pp, // in/out: pointer to result
56 int flags) // FIO_ flags
57{
58 char_u *p = *pp;
59 int error = FALSE;
60 int cc;
61
62
63 if (flags & FIO_UCS4)
64 {
65 if (flags & FIO_ENDIAN_L)
66 {
67 *p++ = c;
68 *p++ = (c >> 8);
69 *p++ = (c >> 16);
70 *p++ = (c >> 24);
71 }
72 else
73 {
74 *p++ = (c >> 24);
75 *p++ = (c >> 16);
76 *p++ = (c >> 8);
77 *p++ = c;
78 }
79 }
80 else if (flags & (FIO_UCS2 | FIO_UTF16))
81 {
82 if (c >= 0x10000)
83 {
84 if (flags & FIO_UTF16)
85 {
86 // Make two words, ten bits of the character in each. First
87 // word is 0xd800 - 0xdbff, second one 0xdc00 - 0xdfff
88 c -= 0x10000;
89 if (c >= 0x100000)
90 error = TRUE;
91 cc = ((c >> 10) & 0x3ff) + 0xd800;
92 if (flags & FIO_ENDIAN_L)
93 {
94 *p++ = cc;
95 *p++ = ((unsigned)cc >> 8);
96 }
97 else
98 {
99 *p++ = ((unsigned)cc >> 8);
100 *p++ = cc;
101 }
102 c = (c & 0x3ff) + 0xdc00;
103 }
104 else
105 error = TRUE;
106 }
107 if (flags & FIO_ENDIAN_L)
108 {
109 *p++ = c;
110 *p++ = (c >> 8);
111 }
112 else
113 {
114 *p++ = (c >> 8);
115 *p++ = c;
116 }
117 }
118 else // Latin1
119 {
120 if (c >= 0x100)
121 {
122 error = TRUE;
123 *p++ = 0xBF;
124 }
125 else
126 *p++ = c;
127 }
128
129 *pp = p;
130 return error;
131}
132
133/*
134 * Call write() to write a number of bytes to the file.
135 * Handles encryption and 'encoding' conversion.
136 *
137 * Return FAIL for failure, OK otherwise.
138 */
139 static int
140buf_write_bytes(struct bw_info *ip)
141{
142 int wlen;
143 char_u *buf = ip->bw_buf; // data to write
144 int len = ip->bw_len; // length of data
145 int flags = ip->bw_flags; // extra flags
146
147 // Skip conversion when writing the crypt magic number or the BOM.
148 if (!(flags & FIO_NOCONVERT))
149 {
150 char_u *p;
151 unsigned c;
152 int n;
153
154 if (flags & FIO_UTF8)
155 {
156 // Convert latin1 in the buffer to UTF-8 in the file.
157 p = ip->bw_conv_buf; // translate to buffer
158 for (wlen = 0; wlen < len; ++wlen)
159 p += utf_char2bytes(buf[wlen], p);
160 buf = ip->bw_conv_buf;
161 len = (int)(p - ip->bw_conv_buf);
162 }
163 else if (flags & (FIO_UCS4 | FIO_UTF16 | FIO_UCS2 | FIO_LATIN1))
164 {
165 // Convert UTF-8 bytes in the buffer to UCS-2, UCS-4, UTF-16 or
166 // Latin1 chars in the file.
167 if (flags & FIO_LATIN1)
168 p = buf; // translate in-place (can only get shorter)
169 else
170 p = ip->bw_conv_buf; // translate to buffer
171 for (wlen = 0; wlen < len; wlen += n)
172 {
173 if (wlen == 0 && ip->bw_restlen != 0)
174 {
175 int l;
176
177 // Use remainder of previous call. Append the start of
178 // buf[] to get a full sequence. Might still be too
179 // short!
180 l = CONV_RESTLEN - ip->bw_restlen;
181 if (l > len)
182 l = len;
183 mch_memmove(ip->bw_rest + ip->bw_restlen, buf, (size_t)l);
184 n = utf_ptr2len_len(ip->bw_rest, ip->bw_restlen + l);
185 if (n > ip->bw_restlen + len)
186 {
187 // We have an incomplete byte sequence at the end to
188 // be written. We can't convert it without the
189 // remaining bytes. Keep them for the next call.
190 if (ip->bw_restlen + len > CONV_RESTLEN)
191 return FAIL;
192 ip->bw_restlen += len;
193 break;
194 }
195 if (n > 1)
196 c = utf_ptr2char(ip->bw_rest);
197 else
198 c = ip->bw_rest[0];
199 if (n >= ip->bw_restlen)
200 {
201 n -= ip->bw_restlen;
202 ip->bw_restlen = 0;
203 }
204 else
205 {
206 ip->bw_restlen -= n;
207 mch_memmove(ip->bw_rest, ip->bw_rest + n,
208 (size_t)ip->bw_restlen);
209 n = 0;
210 }
211 }
212 else
213 {
214 n = utf_ptr2len_len(buf + wlen, len - wlen);
215 if (n > len - wlen)
216 {
217 // We have an incomplete byte sequence at the end to
218 // be written. We can't convert it without the
219 // remaining bytes. Keep them for the next call.
220 if (len - wlen > CONV_RESTLEN)
221 return FAIL;
222 ip->bw_restlen = len - wlen;
223 mch_memmove(ip->bw_rest, buf + wlen,
224 (size_t)ip->bw_restlen);
225 break;
226 }
227 if (n > 1)
228 c = utf_ptr2char(buf + wlen);
229 else
230 c = buf[wlen];
231 }
232
233 if (ucs2bytes(c, &p, flags) && !ip->bw_conv_error)
234 {
235 ip->bw_conv_error = TRUE;
236 ip->bw_conv_error_lnum = ip->bw_start_lnum;
237 }
238 if (c == NL)
239 ++ip->bw_start_lnum;
240 }
241 if (flags & FIO_LATIN1)
242 len = (int)(p - buf);
243 else
244 {
245 buf = ip->bw_conv_buf;
246 len = (int)(p - ip->bw_conv_buf);
247 }
248 }
249
250#ifdef MSWIN
251 else if (flags & FIO_CODEPAGE)
252 {
253 // Convert UTF-8 or codepage to UCS-2 and then to MS-Windows
254 // codepage.
255 char_u *from;
256 size_t fromlen;
257 char_u *to;
258 int u8c;
259 BOOL bad = FALSE;
260 int needed;
261
262 if (ip->bw_restlen > 0)
263 {
264 // Need to concatenate the remainder of the previous call and
265 // the bytes of the current call. Use the end of the
266 // conversion buffer for this.
267 fromlen = len + ip->bw_restlen;
268 from = ip->bw_conv_buf + ip->bw_conv_buflen - fromlen;
269 mch_memmove(from, ip->bw_rest, (size_t)ip->bw_restlen);
270 mch_memmove(from + ip->bw_restlen, buf, (size_t)len);
271 }
272 else
273 {
274 from = buf;
275 fromlen = len;
276 }
277
278 to = ip->bw_conv_buf;
279 if (enc_utf8)
280 {
281 // Convert from UTF-8 to UCS-2, to the start of the buffer.
282 // The buffer has been allocated to be big enough.
283 while (fromlen > 0)
284 {
285 n = (int)utf_ptr2len_len(from, (int)fromlen);
286 if (n > (int)fromlen) // incomplete byte sequence
287 break;
288 u8c = utf_ptr2char(from);
289 *to++ = (u8c & 0xff);
290 *to++ = (u8c >> 8);
291 fromlen -= n;
292 from += n;
293 }
294
295 // Copy remainder to ip->bw_rest[] to be used for the next
296 // call.
297 if (fromlen > CONV_RESTLEN)
298 {
299 // weird overlong sequence
300 ip->bw_conv_error = TRUE;
301 return FAIL;
302 }
303 mch_memmove(ip->bw_rest, from, fromlen);
304 ip->bw_restlen = (int)fromlen;
305 }
306 else
307 {
308 // Convert from enc_codepage to UCS-2, to the start of the
309 // buffer. The buffer has been allocated to be big enough.
310 ip->bw_restlen = 0;
311 needed = MultiByteToWideChar(enc_codepage,
312 MB_ERR_INVALID_CHARS, (LPCSTR)from, (int)fromlen,
313 NULL, 0);
314 if (needed == 0)
315 {
316 // When conversion fails there may be a trailing byte.
317 needed = MultiByteToWideChar(enc_codepage,
318 MB_ERR_INVALID_CHARS, (LPCSTR)from, (int)fromlen - 1,
319 NULL, 0);
320 if (needed == 0)
321 {
322 // Conversion doesn't work.
323 ip->bw_conv_error = TRUE;
324 return FAIL;
325 }
326 // Save the trailing byte for the next call.
327 ip->bw_rest[0] = from[fromlen - 1];
328 ip->bw_restlen = 1;
329 }
330 needed = MultiByteToWideChar(enc_codepage, MB_ERR_INVALID_CHARS,
331 (LPCSTR)from, (int)(fromlen - ip->bw_restlen),
332 (LPWSTR)to, needed);
333 if (needed == 0)
334 {
335 // Safety check: Conversion doesn't work.
336 ip->bw_conv_error = TRUE;
337 return FAIL;
338 }
339 to += needed * 2;
340 }
341
342 fromlen = to - ip->bw_conv_buf;
343 buf = to;
344# ifdef CP_UTF8 // VC 4.1 doesn't define CP_UTF8
345 if (FIO_GET_CP(flags) == CP_UTF8)
346 {
347 // Convert from UCS-2 to UTF-8, using the remainder of the
348 // conversion buffer. Fails when out of space.
349 for (from = ip->bw_conv_buf; fromlen > 1; fromlen -= 2)
350 {
351 u8c = *from++;
352 u8c += (*from++ << 8);
353 to += utf_char2bytes(u8c, to);
354 if (to + 6 >= ip->bw_conv_buf + ip->bw_conv_buflen)
355 {
356 ip->bw_conv_error = TRUE;
357 return FAIL;
358 }
359 }
360 len = (int)(to - buf);
361 }
362 else
363# endif
364 {
365 // Convert from UCS-2 to the codepage, using the remainder of
366 // the conversion buffer. If the conversion uses the default
367 // character "0", the data doesn't fit in this encoding, so
368 // fail.
369 len = WideCharToMultiByte(FIO_GET_CP(flags), 0,
370 (LPCWSTR)ip->bw_conv_buf, (int)fromlen / sizeof(WCHAR),
371 (LPSTR)to, (int)(ip->bw_conv_buflen - fromlen), 0,
372 &bad);
373 if (bad)
374 {
375 ip->bw_conv_error = TRUE;
376 return FAIL;
377 }
378 }
379 }
380#endif
381
382#ifdef MACOS_CONVERT
383 else if (flags & FIO_MACROMAN)
384 {
385 // Convert UTF-8 or latin1 to Apple MacRoman.
386 char_u *from;
387 size_t fromlen;
388
389 if (ip->bw_restlen > 0)
390 {
391 // Need to concatenate the remainder of the previous call and
392 // the bytes of the current call. Use the end of the
393 // conversion buffer for this.
394 fromlen = len + ip->bw_restlen;
395 from = ip->bw_conv_buf + ip->bw_conv_buflen - fromlen;
396 mch_memmove(from, ip->bw_rest, (size_t)ip->bw_restlen);
397 mch_memmove(from + ip->bw_restlen, buf, (size_t)len);
398 }
399 else
400 {
401 from = buf;
402 fromlen = len;
403 }
404
405 if (enc2macroman(from, fromlen,
406 ip->bw_conv_buf, &len, ip->bw_conv_buflen,
407 ip->bw_rest, &ip->bw_restlen) == FAIL)
408 {
409 ip->bw_conv_error = TRUE;
410 return FAIL;
411 }
412 buf = ip->bw_conv_buf;
413 }
414#endif
415
416#ifdef USE_ICONV
417 if (ip->bw_iconv_fd != (iconv_t)-1)
418 {
419 const char *from;
420 size_t fromlen;
421 char *to;
422 size_t tolen;
423
424 // Convert with iconv().
425 if (ip->bw_restlen > 0)
426 {
427 char *fp;
428
429 // Need to concatenate the remainder of the previous call and
430 // the bytes of the current call. Use the end of the
431 // conversion buffer for this.
432 fromlen = len + ip->bw_restlen;
433 fp = (char *)ip->bw_conv_buf + ip->bw_conv_buflen - fromlen;
434 mch_memmove(fp, ip->bw_rest, (size_t)ip->bw_restlen);
435 mch_memmove(fp + ip->bw_restlen, buf, (size_t)len);
436 from = fp;
437 tolen = ip->bw_conv_buflen - fromlen;
438 }
439 else
440 {
441 from = (const char *)buf;
442 fromlen = len;
443 tolen = ip->bw_conv_buflen;
444 }
445 to = (char *)ip->bw_conv_buf;
446
447 if (ip->bw_first)
448 {
449 size_t save_len = tolen;
450
451 // output the initial shift state sequence
452 (void)iconv(ip->bw_iconv_fd, NULL, NULL, &to, &tolen);
453
454 // There is a bug in iconv() on Linux (which appears to be
455 // wide-spread) which sets "to" to NULL and messes up "tolen".
456 if (to == NULL)
457 {
458 to = (char *)ip->bw_conv_buf;
459 tolen = save_len;
460 }
461 ip->bw_first = FALSE;
462 }
463
464 // If iconv() has an error or there is not enough room, fail.
465 if ((iconv(ip->bw_iconv_fd, (void *)&from, &fromlen, &to, &tolen)
466 == (size_t)-1 && ICONV_ERRNO != ICONV_EINVAL)
467 || fromlen > CONV_RESTLEN)
468 {
469 ip->bw_conv_error = TRUE;
470 return FAIL;
471 }
472
473 // copy remainder to ip->bw_rest[] to be used for the next call.
474 if (fromlen > 0)
475 mch_memmove(ip->bw_rest, (void *)from, fromlen);
476 ip->bw_restlen = (int)fromlen;
477
478 buf = ip->bw_conv_buf;
479 len = (int)((char_u *)to - ip->bw_conv_buf);
480 }
481#endif
482 }
483
484 if (ip->bw_fd < 0)
485 // Only checking conversion, which is OK if we get here.
486 return OK;
487
488#ifdef FEAT_CRYPT
489 if (flags & FIO_ENCRYPTED)
490 {
491 // Encrypt the data. Do it in-place if possible, otherwise use an
492 // allocated buffer.
493# ifdef CRYPT_NOT_INPLACE
494 if (crypt_works_inplace(ip->bw_buffer->b_cryptstate))
495 {
496# endif
Bram Moolenaar65aee0b2021-06-27 14:08:24 +0200497 crypt_encode_inplace(ip->bw_buffer->b_cryptstate, buf, len,
498 ip->bw_finish);
Bram Moolenaar473952e2019-09-28 16:30:04 +0200499# ifdef CRYPT_NOT_INPLACE
500 }
501 else
502 {
503 char_u *outbuf;
504
Bram Moolenaar65aee0b2021-06-27 14:08:24 +0200505 len = crypt_encode_alloc(curbuf->b_cryptstate, buf, len, &outbuf,
506 ip->bw_finish);
Bram Moolenaar473952e2019-09-28 16:30:04 +0200507 if (len == 0)
508 return OK; // Crypt layer is buffering, will flush later.
509 wlen = write_eintr(ip->bw_fd, outbuf, len);
510 vim_free(outbuf);
511 return (wlen < len) ? FAIL : OK;
512 }
513# endif
514 }
515#endif
516
517 wlen = write_eintr(ip->bw_fd, buf, len);
518 return (wlen < len) ? FAIL : OK;
519}
520
521/*
522 * Check modification time of file, before writing to it.
523 * The size isn't checked, because using a tool like "gzip" takes care of
524 * using the same timestamp but can't set the size.
525 */
526 static int
527check_mtime(buf_T *buf, stat_T *st)
528{
529 if (buf->b_mtime_read != 0
Leah Neukirchen0a7984a2021-10-14 21:27:55 +0100530 && time_differs(st, buf->b_mtime_read, buf->b_mtime_read_ns))
Bram Moolenaar473952e2019-09-28 16:30:04 +0200531 {
532 msg_scroll = TRUE; // don't overwrite messages here
533 msg_silent = 0; // must give this prompt
534 // don't use emsg() here, don't want to flush the buffers
535 msg_attr(_("WARNING: The file has been changed since reading it!!!"),
536 HL_ATTR(HLF_E));
537 if (ask_yesno((char_u *)_("Do you really want to write to it"),
538 TRUE) == 'n')
539 return FAIL;
540 msg_scroll = FALSE; // always overwrite the file message now
541 }
542 return OK;
543}
544
545/*
546 * Generate a BOM in "buf[4]" for encoding "name".
547 * Return the length of the BOM (zero when no BOM).
548 */
549 static int
550make_bom(char_u *buf, char_u *name)
551{
552 int flags;
553 char_u *p;
554
555 flags = get_fio_flags(name);
556
557 // Can't put a BOM in a non-Unicode file.
558 if (flags == FIO_LATIN1 || flags == 0)
559 return 0;
560
561 if (flags == FIO_UTF8) // UTF-8
562 {
563 buf[0] = 0xef;
564 buf[1] = 0xbb;
565 buf[2] = 0xbf;
566 return 3;
567 }
568 p = buf;
569 (void)ucs2bytes(0xfeff, &p, flags);
570 return (int)(p - buf);
571}
572
573#ifdef UNIX
574 static void
575set_file_time(
576 char_u *fname,
577 time_t atime, // access time
578 time_t mtime) // modification time
579{
580# if defined(HAVE_UTIME) && defined(HAVE_UTIME_H)
581 struct utimbuf buf;
582
583 buf.actime = atime;
584 buf.modtime = mtime;
585 (void)utime((char *)fname, &buf);
586# else
587# if defined(HAVE_UTIMES)
588 struct timeval tvp[2];
589
590 tvp[0].tv_sec = atime;
591 tvp[0].tv_usec = 0;
592 tvp[1].tv_sec = mtime;
593 tvp[1].tv_usec = 0;
594# ifdef NeXT
595 (void)utimes((char *)fname, tvp);
596# else
597 (void)utimes((char *)fname, (const struct timeval *)&tvp);
598# endif
599# endif
600# endif
601}
602#endif // UNIX
603
Bram Moolenaar722e5052020-06-12 22:31:00 +0200604 char *
605new_file_message(void)
606{
607 return shortmess(SHM_NEW) ? _("[New]") : _("[New File]");
608}
609
Bram Moolenaar473952e2019-09-28 16:30:04 +0200610/*
611 * buf_write() - write to file "fname" lines "start" through "end"
612 *
613 * We do our own buffering here because fwrite() is so slow.
614 *
615 * If "forceit" is true, we don't care for errors when attempting backups.
616 * In case of an error everything possible is done to restore the original
617 * file. But when "forceit" is TRUE, we risk losing it.
618 *
619 * When "reset_changed" is TRUE and "append" == FALSE and "start" == 1 and
620 * "end" == curbuf->b_ml.ml_line_count, reset curbuf->b_changed.
621 *
622 * This function must NOT use NameBuff (because it's called by autowrite()).
623 *
624 * return FAIL for failure, OK otherwise
625 */
626 int
627buf_write(
628 buf_T *buf,
629 char_u *fname,
630 char_u *sfname,
631 linenr_T start,
632 linenr_T end,
633 exarg_T *eap, // for forced 'ff' and 'fenc', can be
634 // NULL!
635 int append, // append to the file
636 int forceit,
637 int reset_changed,
638 int filtering)
639{
640 int fd;
641 char_u *backup = NULL;
642 int backup_copy = FALSE; // copy the original file?
643 int dobackup;
644 char_u *ffname;
645 char_u *wfname = NULL; // name of file to write to
646 char_u *s;
647 char_u *ptr;
648 char_u c;
649 int len;
650 linenr_T lnum;
651 long nchars;
652 char_u *errmsg = NULL;
653 int errmsg_allocated = FALSE;
654 char_u *errnum = NULL;
655 char_u *buffer;
656 char_u smallbuf[SMALLBUFSIZE];
657 char_u *backup_ext;
658 int bufsize;
659 long perm; // file permissions
660 int retval = OK;
661 int newfile = FALSE; // TRUE if file doesn't exist yet
662 int msg_save = msg_scroll;
663 int overwriting; // TRUE if writing over original
664 int no_eol = FALSE; // no end-of-line written
665 int device = FALSE; // writing to a device
666 stat_T st_old;
667 int prev_got_int = got_int;
668 int checking_conversion;
669 int file_readonly = FALSE; // overwritten file is read-only
Bram Moolenaar473952e2019-09-28 16:30:04 +0200670#if defined(UNIX) // XXX fix me sometime?
671 int made_writable = FALSE; // 'w' bit has been set
672#endif
673 // writing everything
674 int whole = (start == 1 && end == buf->b_ml.ml_line_count);
675 linenr_T old_line_count = buf->b_ml.ml_line_count;
676 int attr;
677 int fileformat;
678 int write_bin;
679 struct bw_info write_info; // info for buf_write_bytes()
680 int converted = FALSE;
681 int notconverted = FALSE;
682 char_u *fenc; // effective 'fileencoding'
683 char_u *fenc_tofree = NULL; // allocated "fenc"
684 int wb_flags = 0;
685#ifdef HAVE_ACL
686 vim_acl_T acl = NULL; // ACL copied from original file to
687 // backup or new file
688#endif
689#ifdef FEAT_PERSISTENT_UNDO
690 int write_undo_file = FALSE;
691 context_sha256_T sha_ctx;
692#endif
zeertzjqaa925ee2024-06-09 18:24:05 +0200693 unsigned int bkc = get_bkc_flags(buf);
Bram Moolenaarf4a1d1c2019-11-16 13:50:25 +0100694 pos_T orig_start = buf->b_op_start;
695 pos_T orig_end = buf->b_op_end;
Bram Moolenaar473952e2019-09-28 16:30:04 +0200696
697 if (fname == NULL || *fname == NUL) // safety check
698 return FAIL;
699 if (buf->b_ml.ml_mfp == NULL)
700 {
701 // This can happen during startup when there is a stray "w" in the
702 // vimrc file.
Bram Moolenaar74409f62022-01-01 15:58:22 +0000703 emsg(_(e_empty_buffer));
Bram Moolenaar473952e2019-09-28 16:30:04 +0200704 return FAIL;
705 }
706
707 // Disallow writing from .exrc and .vimrc in current directory for
708 // security reasons.
709 if (check_secure())
710 return FAIL;
711
712 // Avoid a crash for a long name.
713 if (STRLEN(fname) >= MAXPATHL)
714 {
Bram Moolenaar12f3c1b2021-12-05 21:46:34 +0000715 emsg(_(e_name_too_long));
Bram Moolenaar473952e2019-09-28 16:30:04 +0200716 return FAIL;
717 }
718
719 // must init bw_conv_buf and bw_iconv_fd before jumping to "fail"
720 write_info.bw_conv_buf = NULL;
721 write_info.bw_conv_error = FALSE;
722 write_info.bw_conv_error_lnum = 0;
723 write_info.bw_restlen = 0;
724#ifdef USE_ICONV
725 write_info.bw_iconv_fd = (iconv_t)-1;
726#endif
727#ifdef FEAT_CRYPT
728 write_info.bw_buffer = buf;
Christian Brabandtf573c6e2021-06-20 14:02:16 +0200729 write_info.bw_finish = FALSE;
Bram Moolenaar473952e2019-09-28 16:30:04 +0200730#endif
731
732 // After writing a file changedtick changes but we don't want to display
733 // the line.
734 ex_no_reprint = TRUE;
735
736 // If there is no file name yet, use the one for the written file.
737 // BF_NOTEDITED is set to reflect this (in case the write fails).
738 // Don't do this when the write is for a filter command.
739 // Don't do this when appending.
740 // Only do this when 'cpoptions' contains the 'F' flag.
741 if (buf->b_ffname == NULL
742 && reset_changed
743 && whole
744 && buf == curbuf
Bram Moolenaar473952e2019-09-28 16:30:04 +0200745 && !bt_nofilename(buf)
Bram Moolenaar473952e2019-09-28 16:30:04 +0200746 && !filtering
747 && (!append || vim_strchr(p_cpo, CPO_FNAMEAPP) != NULL)
748 && vim_strchr(p_cpo, CPO_FNAMEW) != NULL)
749 {
750 if (set_rw_fname(fname, sfname) == FAIL)
751 return FAIL;
752 buf = curbuf; // just in case autocmds made "buf" invalid
753 }
754
755 if (sfname == NULL)
756 sfname = fname;
757 // For Unix: Use the short file name whenever possible.
758 // Avoids problems with networks and when directory names are changed.
759 // Don't do this for MS-DOS, a "cd" in a sub-shell may have moved us to
760 // another directory, which we don't detect
761 ffname = fname; // remember full fname
762#ifdef UNIX
763 fname = sfname;
764#endif
765
766 if (buf->b_ffname != NULL && fnamecmp(ffname, buf->b_ffname) == 0)
767 overwriting = TRUE;
768 else
769 overwriting = FALSE;
770
771 if (exiting)
772 settmode(TMODE_COOK); // when exiting allow typeahead now
773
774 ++no_wait_return; // don't wait for return yet
775
776 // Set '[ and '] marks to the lines to be written.
777 buf->b_op_start.lnum = start;
778 buf->b_op_start.col = 0;
779 buf->b_op_end.lnum = end;
780 buf->b_op_end.col = 0;
781
782 {
783 aco_save_T aco;
784 int buf_ffname = FALSE;
785 int buf_sfname = FALSE;
786 int buf_fname_f = FALSE;
787 int buf_fname_s = FALSE;
788 int did_cmd = FALSE;
789 int nofile_err = FALSE;
790 int empty_memline = (buf->b_ml.ml_mfp == NULL);
791 bufref_T bufref;
792
793 // Apply PRE autocommands.
794 // Set curbuf to the buffer to be written.
795 // Careful: The autocommands may call buf_write() recursively!
796 if (ffname == buf->b_ffname)
797 buf_ffname = TRUE;
798 if (sfname == buf->b_sfname)
799 buf_sfname = TRUE;
800 if (fname == buf->b_ffname)
801 buf_fname_f = TRUE;
802 if (fname == buf->b_sfname)
803 buf_fname_s = TRUE;
804
Bram Moolenaare76062c2022-11-28 18:51:43 +0000805 // Set curwin/curbuf to buf and save a few things.
Bram Moolenaar473952e2019-09-28 16:30:04 +0200806 aucmd_prepbuf(&aco, buf);
Bram Moolenaare76062c2022-11-28 18:51:43 +0000807 if (curbuf != buf)
808 {
809 // Could not find a window for "buf". Doing more might cause
810 // problems, better bail out.
811 return FAIL;
812 }
813
Bram Moolenaar473952e2019-09-28 16:30:04 +0200814 set_bufref(&bufref, buf);
815
816 if (append)
817 {
818 if (!(did_cmd = apply_autocmds_exarg(EVENT_FILEAPPENDCMD,
819 sfname, sfname, FALSE, curbuf, eap)))
820 {
Bram Moolenaar473952e2019-09-28 16:30:04 +0200821 if (overwriting && bt_nofilename(curbuf))
822 nofile_err = TRUE;
823 else
Bram Moolenaar473952e2019-09-28 16:30:04 +0200824 apply_autocmds_exarg(EVENT_FILEAPPENDPRE,
825 sfname, sfname, FALSE, curbuf, eap);
826 }
827 }
828 else if (filtering)
829 {
830 apply_autocmds_exarg(EVENT_FILTERWRITEPRE,
831 NULL, sfname, FALSE, curbuf, eap);
832 }
833 else if (reset_changed && whole)
834 {
835 int was_changed = curbufIsChanged();
836
837 did_cmd = apply_autocmds_exarg(EVENT_BUFWRITECMD,
838 sfname, sfname, FALSE, curbuf, eap);
839 if (did_cmd)
840 {
841 if (was_changed && !curbufIsChanged())
842 {
843 // Written everything correctly and BufWriteCmd has reset
844 // 'modified': Correct the undo information so that an
845 // undo now sets 'modified'.
846 u_unchanged(curbuf);
847 u_update_save_nr(curbuf);
848 }
849 }
850 else
851 {
Bram Moolenaar473952e2019-09-28 16:30:04 +0200852 if (overwriting && bt_nofilename(curbuf))
853 nofile_err = TRUE;
854 else
Bram Moolenaar473952e2019-09-28 16:30:04 +0200855 apply_autocmds_exarg(EVENT_BUFWRITEPRE,
856 sfname, sfname, FALSE, curbuf, eap);
857 }
858 }
859 else
860 {
861 if (!(did_cmd = apply_autocmds_exarg(EVENT_FILEWRITECMD,
862 sfname, sfname, FALSE, curbuf, eap)))
863 {
Bram Moolenaar473952e2019-09-28 16:30:04 +0200864 if (overwriting && bt_nofilename(curbuf))
865 nofile_err = TRUE;
866 else
Bram Moolenaar473952e2019-09-28 16:30:04 +0200867 apply_autocmds_exarg(EVENT_FILEWRITEPRE,
868 sfname, sfname, FALSE, curbuf, eap);
869 }
870 }
871
872 // restore curwin/curbuf and a few other things
873 aucmd_restbuf(&aco);
874
875 // In three situations we return here and don't write the file:
876 // 1. the autocommands deleted or unloaded the buffer.
877 // 2. The autocommands abort script processing.
878 // 3. If one of the "Cmd" autocommands was executed.
879 if (!bufref_valid(&bufref))
880 buf = NULL;
881 if (buf == NULL || (buf->b_ml.ml_mfp == NULL && !empty_memline)
882 || did_cmd || nofile_err
883#ifdef FEAT_EVAL
884 || aborting()
885#endif
886 )
887 {
Bram Moolenaare1004402020-10-24 20:49:43 +0200888 if (buf != NULL && (cmdmod.cmod_flags & CMOD_LOCKMARKS))
Bram Moolenaarf4a1d1c2019-11-16 13:50:25 +0100889 {
890 // restore the original '[ and '] positions
891 buf->b_op_start = orig_start;
892 buf->b_op_end = orig_end;
893 }
894
Bram Moolenaar473952e2019-09-28 16:30:04 +0200895 --no_wait_return;
896 msg_scroll = msg_save;
897 if (nofile_err)
Bram Moolenaar500a1f92022-09-20 11:49:10 +0100898 semsg(_(e_no_matching_autocommands_for_buftype_str_buffer),
899 curbuf->b_p_bt);
Bram Moolenaar473952e2019-09-28 16:30:04 +0200900
901 if (nofile_err
902#ifdef FEAT_EVAL
903 || aborting()
904#endif
905 )
906 // An aborting error, interrupt or exception in the
907 // autocommands.
908 return FAIL;
909 if (did_cmd)
910 {
911 if (buf == NULL)
912 // The buffer was deleted. We assume it was written
913 // (can't retry anyway).
914 return OK;
915 if (overwriting)
916 {
917 // Assume the buffer was written, update the timestamp.
918 ml_timestamp(buf);
919 if (append)
920 buf->b_flags &= ~BF_NEW;
921 else
922 buf->b_flags &= ~BF_WRITE_MASK;
923 }
924 if (reset_changed && buf->b_changed && !append
925 && (overwriting || vim_strchr(p_cpo, CPO_PLUS) != NULL))
926 // Buffer still changed, the autocommands didn't work
927 // properly.
928 return FAIL;
929 return OK;
930 }
931#ifdef FEAT_EVAL
932 if (!aborting())
933#endif
Bram Moolenaar6d057012021-12-31 18:49:43 +0000934 emsg(_(e_autocommands_deleted_or_unloaded_buffer_to_be_written));
Bram Moolenaar473952e2019-09-28 16:30:04 +0200935 return FAIL;
936 }
937
938 // The autocommands may have changed the number of lines in the file.
939 // When writing the whole file, adjust the end.
940 // When writing part of the file, assume that the autocommands only
941 // changed the number of lines that are to be written (tricky!).
942 if (buf->b_ml.ml_line_count != old_line_count)
943 {
944 if (whole) // write all
945 end = buf->b_ml.ml_line_count;
946 else if (buf->b_ml.ml_line_count > old_line_count) // more lines
947 end += buf->b_ml.ml_line_count - old_line_count;
948 else // less lines
949 {
950 end -= old_line_count - buf->b_ml.ml_line_count;
951 if (end < start)
952 {
953 --no_wait_return;
954 msg_scroll = msg_save;
Bram Moolenaar6d057012021-12-31 18:49:43 +0000955 emsg(_(e_autocommands_changed_number_of_lines_in_unexpected_way));
Bram Moolenaar473952e2019-09-28 16:30:04 +0200956 return FAIL;
957 }
958 }
959 }
960
961 // The autocommands may have changed the name of the buffer, which may
962 // be kept in fname, ffname and sfname.
963 if (buf_ffname)
964 ffname = buf->b_ffname;
965 if (buf_sfname)
966 sfname = buf->b_sfname;
967 if (buf_fname_f)
968 fname = buf->b_ffname;
969 if (buf_fname_s)
970 fname = buf->b_sfname;
971 }
972
Bram Moolenaare1004402020-10-24 20:49:43 +0200973 if (cmdmod.cmod_flags & CMOD_LOCKMARKS)
Bram Moolenaarf4a1d1c2019-11-16 13:50:25 +0100974 {
975 // restore the original '[ and '] positions
976 buf->b_op_start = orig_start;
977 buf->b_op_end = orig_end;
978 }
979
Bram Moolenaar473952e2019-09-28 16:30:04 +0200980#ifdef FEAT_NETBEANS_INTG
981 if (netbeans_active() && isNetbeansBuffer(buf))
982 {
983 if (whole)
984 {
985 // b_changed can be 0 after an undo, but we still need to write
986 // the buffer to NetBeans.
987 if (buf->b_changed || isNetbeansModified(buf))
988 {
989 --no_wait_return; // may wait for return now
990 msg_scroll = msg_save;
991 netbeans_save_buffer(buf); // no error checking...
992 return retval;
993 }
994 else
995 {
996 errnum = (char_u *)"E656: ";
Bram Moolenaarf1474d82021-12-31 19:59:55 +0000997 errmsg = (char_u *)_(e_netbeans_disallows_writes_of_unmodified_buffers);
Bram Moolenaar473952e2019-09-28 16:30:04 +0200998 buffer = NULL;
999 goto fail;
1000 }
1001 }
1002 else
1003 {
1004 errnum = (char_u *)"E657: ";
Bram Moolenaarf1474d82021-12-31 19:59:55 +00001005 errmsg = (char_u *)_(e_partial_writes_disallowed_for_netbeans_buffers);
Bram Moolenaar473952e2019-09-28 16:30:04 +02001006 buffer = NULL;
1007 goto fail;
1008 }
1009 }
1010#endif
1011
1012 if (shortmess(SHM_OVER) && !exiting)
1013 msg_scroll = FALSE; // overwrite previous file message
1014 else
1015 msg_scroll = TRUE; // don't overwrite previous file message
1016 if (!filtering)
1017 filemess(buf,
1018#ifndef UNIX
1019 sfname,
1020#else
1021 fname,
1022#endif
1023 (char_u *)"", 0); // show that we are busy
1024 msg_scroll = FALSE; // always overwrite the file message now
1025
1026 buffer = alloc(WRITEBUFSIZE);
1027 if (buffer == NULL) // can't allocate big buffer, use small
1028 // one (to be able to write when out of
1029 // memory)
1030 {
1031 buffer = smallbuf;
1032 bufsize = SMALLBUFSIZE;
1033 }
1034 else
1035 bufsize = WRITEBUFSIZE;
1036
1037 // Get information about original file (if there is one).
1038#if defined(UNIX)
1039 st_old.st_dev = 0;
1040 st_old.st_ino = 0;
1041 perm = -1;
1042 if (mch_stat((char *)fname, &st_old) < 0)
1043 newfile = TRUE;
1044 else
1045 {
1046 perm = st_old.st_mode;
1047 if (!S_ISREG(st_old.st_mode)) // not a file
1048 {
1049 if (S_ISDIR(st_old.st_mode))
1050 {
1051 errnum = (char_u *)"E502: ";
Bram Moolenaarf1474d82021-12-31 19:59:55 +00001052 errmsg = (char_u *)_(e_is_a_directory);
Bram Moolenaar473952e2019-09-28 16:30:04 +02001053 goto fail;
1054 }
1055 if (mch_nodetype(fname) != NODE_WRITABLE)
1056 {
1057 errnum = (char_u *)"E503: ";
Bram Moolenaarf1474d82021-12-31 19:59:55 +00001058 errmsg = (char_u *)_(e_is_not_file_or_writable_device);
Bram Moolenaar473952e2019-09-28 16:30:04 +02001059 goto fail;
1060 }
1061 // It's a device of some kind (or a fifo) which we can write to
1062 // but for which we can't make a backup.
1063 device = TRUE;
1064 newfile = TRUE;
1065 perm = -1;
1066 }
1067 }
1068#else // !UNIX
1069 // Check for a writable device name.
1070 c = mch_nodetype(fname);
1071 if (c == NODE_OTHER)
1072 {
1073 errnum = (char_u *)"E503: ";
Bram Moolenaarf1474d82021-12-31 19:59:55 +00001074 errmsg = (char_u *)_(e_is_not_file_or_writable_device);
Bram Moolenaar473952e2019-09-28 16:30:04 +02001075 goto fail;
1076 }
1077 if (c == NODE_WRITABLE)
1078 {
1079# if defined(MSWIN)
1080 // MS-Windows allows opening a device, but we will probably get stuck
1081 // trying to write to it.
1082 if (!p_odev)
1083 {
1084 errnum = (char_u *)"E796: ";
Bram Moolenaarf1474d82021-12-31 19:59:55 +00001085 errmsg = (char_u *)_(e_writing_to_device_disabled_with_opendevice_option);
Bram Moolenaar473952e2019-09-28 16:30:04 +02001086 goto fail;
1087 }
1088# endif
1089 device = TRUE;
1090 newfile = TRUE;
1091 perm = -1;
1092 }
1093 else
1094 {
1095 perm = mch_getperm(fname);
1096 if (perm < 0)
1097 newfile = TRUE;
1098 else if (mch_isdir(fname))
1099 {
1100 errnum = (char_u *)"E502: ";
Bram Moolenaarf1474d82021-12-31 19:59:55 +00001101 errmsg = (char_u *)_(e_is_a_directory);
Bram Moolenaar473952e2019-09-28 16:30:04 +02001102 goto fail;
1103 }
1104 if (overwriting)
1105 (void)mch_stat((char *)fname, &st_old);
1106 }
1107#endif // !UNIX
1108
1109 if (!device && !newfile)
1110 {
1111 // Check if the file is really writable (when renaming the file to
1112 // make a backup we won't discover it later).
1113 file_readonly = check_file_readonly(fname, (int)perm);
1114
1115 if (!forceit && file_readonly)
1116 {
1117 if (vim_strchr(p_cpo, CPO_FWRITE) != NULL)
1118 {
1119 errnum = (char_u *)"E504: ";
Bram Moolenaarf1474d82021-12-31 19:59:55 +00001120 errmsg = (char_u *)_(e_is_read_only_cannot_override_W_in_cpoptions);
Bram Moolenaar473952e2019-09-28 16:30:04 +02001121 }
1122 else
1123 {
1124 errnum = (char_u *)"E505: ";
Bram Moolenaarf1474d82021-12-31 19:59:55 +00001125 errmsg = (char_u *)_(e_is_read_only_add_bang_to_override);
Bram Moolenaar473952e2019-09-28 16:30:04 +02001126 }
1127 goto fail;
1128 }
1129
1130 // Check if the timestamp hasn't changed since reading the file.
1131 if (overwriting)
1132 {
1133 retval = check_mtime(buf, &st_old);
1134 if (retval == FAIL)
1135 goto fail;
1136 }
1137 }
1138
1139#ifdef HAVE_ACL
1140 // For systems that support ACL: get the ACL from the original file.
1141 if (!newfile)
1142 acl = mch_get_acl(fname);
1143#endif
1144
1145 // If 'backupskip' is not empty, don't make a backup for some files.
1146 dobackup = (p_wb || p_bk || *p_pm != NUL);
Bram Moolenaar473952e2019-09-28 16:30:04 +02001147 if (dobackup && *p_bsk != NUL && match_file_list(p_bsk, sfname, ffname))
1148 dobackup = FALSE;
Bram Moolenaar473952e2019-09-28 16:30:04 +02001149
1150 // Save the value of got_int and reset it. We don't want a previous
1151 // interruption cancel writing, only hitting CTRL-C while writing should
1152 // abort it.
1153 prev_got_int = got_int;
1154 got_int = FALSE;
1155
1156 // Mark the buffer as 'being saved' to prevent changed buffer warnings
1157 buf->b_saving = TRUE;
1158
1159 // If we are not appending or filtering, the file exists, and the
1160 // 'writebackup', 'backup' or 'patchmode' option is set, need a backup.
1161 // When 'patchmode' is set also make a backup when appending.
1162 //
1163 // Do not make any backup, if 'writebackup' and 'backup' are both switched
1164 // off. This helps when editing large files on almost-full disks.
1165 if (!(append && *p_pm == NUL) && !filtering && perm >= 0 && dobackup)
1166 {
1167#if defined(UNIX) || defined(MSWIN)
1168 stat_T st;
1169#endif
1170
1171 if ((bkc & BKC_YES) || append) // "yes"
1172 backup_copy = TRUE;
1173#if defined(UNIX) || defined(MSWIN)
1174 else if ((bkc & BKC_AUTO)) // "auto"
1175 {
1176 int i;
1177
1178# ifdef UNIX
1179 // Don't rename the file when:
1180 // - it's a hard link
1181 // - it's a symbolic link
1182 // - we don't have write permission in the directory
1183 // - we can't set the owner/group of the new file
1184 if (st_old.st_nlink > 1
1185 || mch_lstat((char *)fname, &st) < 0
1186 || st.st_dev != st_old.st_dev
1187 || st.st_ino != st_old.st_ino
1188# ifndef HAVE_FCHOWN
1189 || st.st_uid != st_old.st_uid
1190 || st.st_gid != st_old.st_gid
1191# endif
1192 )
1193 backup_copy = TRUE;
1194 else
1195# else
1196# ifdef MSWIN
1197 // On NTFS file systems hard links are possible.
1198 if (mch_is_linked(fname))
1199 backup_copy = TRUE;
1200 else
1201# endif
1202# endif
1203 {
1204 // Check if we can create a file and set the owner/group to
1205 // the ones from the original file.
1206 // First find a file name that doesn't exist yet (use some
1207 // arbitrary numbers).
1208 STRCPY(IObuff, fname);
K.Takatadbdcc792022-07-31 11:50:42 +01001209 fd = -1;
Bram Moolenaar473952e2019-09-28 16:30:04 +02001210 for (i = 4913; ; i += 123)
1211 {
1212 sprintf((char *)gettail(IObuff), "%d", i);
1213 if (mch_lstat((char *)IObuff, &st) < 0)
K.Takatadbdcc792022-07-31 11:50:42 +01001214 {
1215 fd = mch_open((char *)IObuff,
Bram Moolenaar473952e2019-09-28 16:30:04 +02001216 O_CREAT|O_WRONLY|O_EXCL|O_NOFOLLOW, perm);
K.Takatadbdcc792022-07-31 11:50:42 +01001217 if (fd < 0 && errno == EEXIST)
1218 // If the same file name is created by another
1219 // process between lstat() and open(), find another
1220 // name.
1221 continue;
1222 break;
1223 }
1224 }
Bram Moolenaar473952e2019-09-28 16:30:04 +02001225 if (fd < 0) // can't write in directory
1226 backup_copy = TRUE;
1227 else
1228 {
1229# ifdef UNIX
1230# ifdef HAVE_FCHOWN
1231 vim_ignored = fchown(fd, st_old.st_uid, st_old.st_gid);
1232# endif
1233 if (mch_stat((char *)IObuff, &st) < 0
1234 || st.st_uid != st_old.st_uid
1235 || st.st_gid != st_old.st_gid
1236 || (long)st.st_mode != perm)
1237 backup_copy = TRUE;
1238# endif
1239 // Close the file before removing it, on MS-Windows we
1240 // can't delete an open file.
1241 close(fd);
1242 mch_remove(IObuff);
1243# ifdef MSWIN
1244 // MS-Windows may trigger a virus scanner to open the
1245 // file, we can't delete it then. Keep trying for half a
1246 // second.
1247 {
1248 int try;
1249
1250 for (try = 0; try < 10; ++try)
1251 {
1252 if (mch_lstat((char *)IObuff, &st) < 0)
1253 break;
1254 ui_delay(50L, TRUE); // wait 50 msec
1255 mch_remove(IObuff);
1256 }
1257 }
1258# endif
1259 }
1260 }
1261 }
1262
1263 // Break symlinks and/or hardlinks if we've been asked to.
1264 if ((bkc & BKC_BREAKSYMLINK) || (bkc & BKC_BREAKHARDLINK))
1265 {
1266# ifdef UNIX
1267 int lstat_res;
1268
1269 lstat_res = mch_lstat((char *)fname, &st);
1270
1271 // Symlinks.
1272 if ((bkc & BKC_BREAKSYMLINK)
1273 && lstat_res == 0
1274 && st.st_ino != st_old.st_ino)
1275 backup_copy = FALSE;
1276
1277 // Hardlinks.
1278 if ((bkc & BKC_BREAKHARDLINK)
1279 && st_old.st_nlink > 1
1280 && (lstat_res != 0 || st.st_ino == st_old.st_ino))
1281 backup_copy = FALSE;
1282# else
1283# if defined(MSWIN)
1284 // Symlinks.
1285 if ((bkc & BKC_BREAKSYMLINK) && mch_is_symbolic_link(fname))
1286 backup_copy = FALSE;
1287
1288 // Hardlinks.
1289 if ((bkc & BKC_BREAKHARDLINK) && mch_is_hard_link(fname))
1290 backup_copy = FALSE;
1291# endif
1292# endif
1293 }
1294
1295#endif
1296
1297 // make sure we have a valid backup extension to use
1298 if (*p_bex == NUL)
1299 backup_ext = (char_u *)".bak";
1300 else
1301 backup_ext = p_bex;
1302
1303 if (backup_copy
1304 && (fd = mch_open((char *)fname, O_RDONLY | O_EXTRA, 0)) >= 0)
1305 {
1306 int bfd;
1307 char_u *copybuf, *wp;
1308 int some_error = FALSE;
1309 stat_T st_new;
1310 char_u *dirp;
1311 char_u *rootname;
1312#if defined(UNIX) || defined(MSWIN)
1313 char_u *p;
1314#endif
1315#if defined(UNIX)
1316 int did_set_shortname;
1317 mode_t umask_save;
1318#endif
1319
1320 copybuf = alloc(WRITEBUFSIZE + 1);
1321 if (copybuf == NULL)
1322 {
1323 some_error = TRUE; // out of memory
1324 goto nobackup;
1325 }
1326
1327 // Try to make the backup in each directory in the 'bdir' option.
1328 //
1329 // Unix semantics has it, that we may have a writable file,
1330 // that cannot be recreated with a simple open(..., O_CREAT, ) e.g:
1331 // - the directory is not writable,
1332 // - the file may be a symbolic link,
1333 // - the file may belong to another user/group, etc.
1334 //
1335 // For these reasons, the existing writable file must be truncated
1336 // and reused. Creation of a backup COPY will be attempted.
1337 dirp = p_bdir;
1338 while (*dirp)
1339 {
1340#ifdef UNIX
1341 st_new.st_ino = 0;
1342 st_new.st_dev = 0;
1343 st_new.st_gid = 0;
1344#endif
1345
1346 // Isolate one directory name, using an entry in 'bdir'.
1347 (void)copy_option_part(&dirp, copybuf, WRITEBUFSIZE, ",");
1348
1349#if defined(UNIX) || defined(MSWIN)
1350 p = copybuf + STRLEN(copybuf);
1351 if (after_pathsep(copybuf, p) && p[-1] == p[-2])
1352 // Ends with '//', use full path
zeertzjq242667a2024-07-25 20:58:42 +02001353 if ((p = make_percent_swname(copybuf, p, fname)) != NULL)
Bram Moolenaar473952e2019-09-28 16:30:04 +02001354 {
1355 backup = modname(p, backup_ext, FALSE);
1356 vim_free(p);
1357 }
1358#endif
1359 rootname = get_file_in_dir(fname, copybuf);
1360 if (rootname == NULL)
1361 {
1362 some_error = TRUE; // out of memory
1363 goto nobackup;
1364 }
1365
1366#if defined(UNIX)
1367 did_set_shortname = FALSE;
1368#endif
1369
1370 // May try twice if 'shortname' not set.
1371 for (;;)
1372 {
1373 // Make the backup file name.
1374 if (backup == NULL)
1375 backup = buf_modname((buf->b_p_sn || buf->b_shortname),
1376 rootname, backup_ext, FALSE);
1377 if (backup == NULL)
1378 {
1379 vim_free(rootname);
1380 some_error = TRUE; // out of memory
1381 goto nobackup;
1382 }
1383
1384 // Check if backup file already exists.
1385 if (mch_stat((char *)backup, &st_new) >= 0)
1386 {
1387#ifdef UNIX
1388 // Check if backup file is same as original file.
1389 // May happen when modname() gave the same file back.
1390 // E.g. silly link, or file name-length reached.
1391 // If we don't check here, we either ruin the file
1392 // when copying or erase it after writing. jw.
1393 if (st_new.st_dev == st_old.st_dev
1394 && st_new.st_ino == st_old.st_ino)
1395 {
1396 VIM_CLEAR(backup); // no backup file to delete
1397 // may try again with 'shortname' set
1398 if (!(buf->b_shortname || buf->b_p_sn))
1399 {
1400 buf->b_shortname = TRUE;
1401 did_set_shortname = TRUE;
1402 continue;
1403 }
1404 // setting shortname didn't help
1405 if (did_set_shortname)
1406 buf->b_shortname = FALSE;
1407 break;
1408 }
1409#endif
1410
1411 // If we are not going to keep the backup file, don't
1412 // delete an existing one, try to use another name.
1413 // Change one character, just before the extension.
1414 if (!p_bk)
1415 {
1416 wp = backup + STRLEN(backup) - 1
1417 - STRLEN(backup_ext);
1418 if (wp < backup) // empty file name ???
1419 wp = backup;
1420 *wp = 'z';
1421 while (*wp > 'a'
1422 && mch_stat((char *)backup, &st_new) >= 0)
1423 --*wp;
1424 // They all exist??? Must be something wrong.
1425 if (*wp == 'a')
1426 VIM_CLEAR(backup);
1427 }
1428 }
1429 break;
1430 }
1431 vim_free(rootname);
1432
1433 // Try to create the backup file
1434 if (backup != NULL)
1435 {
1436 // remove old backup, if present
1437 mch_remove(backup);
1438 // Open with O_EXCL to avoid the file being created while
1439 // we were sleeping (symlink hacker attack?). Reset umask
1440 // if possible to avoid mch_setperm() below.
1441#ifdef UNIX
1442 umask_save = umask(0);
1443#endif
1444 bfd = mch_open((char *)backup,
1445 O_WRONLY|O_CREAT|O_EXTRA|O_EXCL|O_NOFOLLOW,
1446 perm & 0777);
1447#ifdef UNIX
1448 (void)umask(umask_save);
1449#endif
1450 if (bfd < 0)
1451 VIM_CLEAR(backup);
1452 else
1453 {
1454 // Set file protection same as original file, but
1455 // strip s-bit. Only needed if umask() wasn't used
1456 // above.
1457#ifndef UNIX
1458 (void)mch_setperm(backup, perm & 0777);
1459#else
1460 // Try to set the group of the backup same as the
1461 // original file. If this fails, set the protection
1462 // bits for the group same as the protection bits for
1463 // others.
1464 if (st_new.st_gid != st_old.st_gid
1465# ifdef HAVE_FCHOWN // sequent-ptx lacks fchown()
1466 && fchown(bfd, (uid_t)-1, st_old.st_gid) != 0
1467# endif
1468 )
1469 mch_setperm(backup,
1470 (perm & 0707) | ((perm & 07) << 3));
1471# if defined(HAVE_SELINUX) || defined(HAVE_SMACK)
1472 mch_copy_sec(fname, backup);
1473# endif
Christian Brabandte085dfd2023-09-30 12:49:18 +02001474# ifdef FEAT_XATTR
1475 mch_copy_xattr(fname, backup);
1476# endif
Bram Moolenaar473952e2019-09-28 16:30:04 +02001477#endif
1478
1479 // copy the file.
1480 write_info.bw_fd = bfd;
1481 write_info.bw_buf = copybuf;
1482 write_info.bw_flags = FIO_NOCONVERT;
1483 while ((write_info.bw_len = read_eintr(fd, copybuf,
1484 WRITEBUFSIZE)) > 0)
1485 {
1486 if (buf_write_bytes(&write_info) == FAIL)
1487 {
dundargocc57b5bc2022-11-02 13:30:51 +00001488 errmsg = (char_u *)_(e_cant_write_to_backup_file_add_bang_to_override);
Bram Moolenaar473952e2019-09-28 16:30:04 +02001489 break;
1490 }
1491 ui_breakcheck();
1492 if (got_int)
1493 {
Bram Moolenaar436b5ad2021-12-31 22:49:24 +00001494 errmsg = (char_u *)_(e_interrupted);
Bram Moolenaar473952e2019-09-28 16:30:04 +02001495 break;
1496 }
1497 }
1498
1499 if (close(bfd) < 0 && errmsg == NULL)
Bram Moolenaarf1474d82021-12-31 19:59:55 +00001500 errmsg = (char_u *)_(e_close_error_for_backup_file_add_bang_to_write_anyway);
Bram Moolenaar473952e2019-09-28 16:30:04 +02001501 if (write_info.bw_len < 0)
Bram Moolenaarf1474d82021-12-31 19:59:55 +00001502 errmsg = (char_u *)_(e_cant_read_file_for_backup_add_bang_to_write_anyway);
Bram Moolenaar473952e2019-09-28 16:30:04 +02001503#ifdef UNIX
1504 set_file_time(backup, st_old.st_atime, st_old.st_mtime);
1505#endif
1506#ifdef HAVE_ACL
1507 mch_set_acl(backup, acl);
1508#endif
1509#if defined(HAVE_SELINUX) || defined(HAVE_SMACK)
1510 mch_copy_sec(fname, backup);
1511#endif
Christian Brabandte085dfd2023-09-30 12:49:18 +02001512#ifdef FEAT_XATTR
1513 mch_copy_xattr(fname, backup);
1514#endif
Bram Moolenaar7781ebe2021-02-01 20:35:01 +01001515#ifdef MSWIN
1516 (void)mch_copy_file_attribute(fname, backup);
1517#endif
Bram Moolenaar473952e2019-09-28 16:30:04 +02001518 break;
1519 }
1520 }
1521 }
1522 nobackup:
1523 close(fd); // ignore errors for closing read file
1524 vim_free(copybuf);
1525
1526 if (backup == NULL && errmsg == NULL)
Bram Moolenaarf1474d82021-12-31 19:59:55 +00001527 errmsg = (char_u *)_(e_cannot_create_backup_file_add_bang_to_write_anyway);
Bram Moolenaar473952e2019-09-28 16:30:04 +02001528 // ignore errors when forceit is TRUE
1529 if ((some_error || errmsg != NULL) && !forceit)
1530 {
1531 retval = FAIL;
1532 goto fail;
1533 }
1534 errmsg = NULL;
1535 }
1536 else
1537 {
1538 char_u *dirp;
1539 char_u *p;
1540 char_u *rootname;
1541
1542 // Make a backup by renaming the original file.
1543
1544 // If 'cpoptions' includes the "W" flag, we don't want to
1545 // overwrite a read-only file. But rename may be possible
1546 // anyway, thus we need an extra check here.
1547 if (file_readonly && vim_strchr(p_cpo, CPO_FWRITE) != NULL)
1548 {
1549 errnum = (char_u *)"E504: ";
Bram Moolenaarf1474d82021-12-31 19:59:55 +00001550 errmsg = (char_u *)_(e_is_read_only_cannot_override_W_in_cpoptions);
Bram Moolenaar473952e2019-09-28 16:30:04 +02001551 goto fail;
1552 }
1553
1554 // Form the backup file name - change path/fo.o.h to
1555 // path/fo.o.h.bak Try all directories in 'backupdir', first one
1556 // that works is used.
1557 dirp = p_bdir;
1558 while (*dirp)
1559 {
1560 // Isolate one directory name and make the backup file name.
1561 (void)copy_option_part(&dirp, IObuff, IOSIZE, ",");
1562
1563#if defined(UNIX) || defined(MSWIN)
1564 p = IObuff + STRLEN(IObuff);
1565 if (after_pathsep(IObuff, p) && p[-1] == p[-2])
1566 // path ends with '//', use full path
zeertzjq242667a2024-07-25 20:58:42 +02001567 if ((p = make_percent_swname(IObuff, p, fname)) != NULL)
Bram Moolenaar473952e2019-09-28 16:30:04 +02001568 {
1569 backup = modname(p, backup_ext, FALSE);
1570 vim_free(p);
1571 }
1572#endif
1573 if (backup == NULL)
1574 {
1575 rootname = get_file_in_dir(fname, IObuff);
1576 if (rootname == NULL)
1577 backup = NULL;
1578 else
1579 {
1580 backup = buf_modname(
1581 (buf->b_p_sn || buf->b_shortname),
1582 rootname, backup_ext, FALSE);
1583 vim_free(rootname);
1584 }
1585 }
1586
1587 if (backup != NULL)
1588 {
1589 // If we are not going to keep the backup file, don't
1590 // delete an existing one, try to use another name.
1591 // Change one character, just before the extension.
1592 if (!p_bk && mch_getperm(backup) >= 0)
1593 {
1594 p = backup + STRLEN(backup) - 1 - STRLEN(backup_ext);
1595 if (p < backup) // empty file name ???
1596 p = backup;
1597 *p = 'z';
1598 while (*p > 'a' && mch_getperm(backup) >= 0)
1599 --*p;
1600 // They all exist??? Must be something wrong!
1601 if (*p == 'a')
1602 VIM_CLEAR(backup);
1603 }
1604 }
1605 if (backup != NULL)
1606 {
1607 // Delete any existing backup and move the current version
1608 // to the backup. For safety, we don't remove the backup
1609 // until the write has finished successfully. And if the
1610 // 'backup' option is set, leave it around.
1611
1612 // If the renaming of the original file to the backup file
1613 // works, quit here.
1614 if (vim_rename(fname, backup) == 0)
1615 break;
1616
1617 VIM_CLEAR(backup); // don't do the rename below
1618 }
1619 }
1620 if (backup == NULL && !forceit)
1621 {
Bram Moolenaarf1474d82021-12-31 19:59:55 +00001622 errmsg = (char_u *)_(e_cant_make_backup_file_add_bang_to_write_anyway);
Bram Moolenaar473952e2019-09-28 16:30:04 +02001623 goto fail;
1624 }
1625 }
1626 }
1627
1628#if defined(UNIX)
1629 // When using ":w!" and the file was read-only: make it writable
1630 if (forceit && perm >= 0 && !(perm & 0200) && st_old.st_uid == getuid()
1631 && vim_strchr(p_cpo, CPO_FWRITE) == NULL)
1632 {
1633 perm |= 0200;
1634 (void)mch_setperm(fname, perm);
1635 made_writable = TRUE;
1636 }
1637#endif
1638
1639 // When using ":w!" and writing to the current file, 'readonly' makes no
1640 // sense, reset it, unless 'Z' appears in 'cpoptions'.
1641 if (forceit && overwriting && vim_strchr(p_cpo, CPO_KEEPRO) == NULL)
1642 {
1643 buf->b_p_ro = FALSE;
Bram Moolenaar473952e2019-09-28 16:30:04 +02001644 need_maketitle = TRUE; // set window title later
Bram Moolenaar473952e2019-09-28 16:30:04 +02001645 status_redraw_all(); // redraw status lines later
1646 }
1647
1648 if (end > buf->b_ml.ml_line_count)
1649 end = buf->b_ml.ml_line_count;
1650 if (buf->b_ml.ml_flags & ML_EMPTY)
1651 start = end + 1;
1652
1653 // If the original file is being overwritten, there is a small chance that
1654 // we crash in the middle of writing. Therefore the file is preserved now.
1655 // This makes all block numbers positive so that recovery does not need
1656 // the original file.
1657 // Don't do this if there is a backup file and we are exiting.
1658 if (reset_changed && !newfile && overwriting
1659 && !(exiting && backup != NULL))
1660 {
1661 ml_preserve(buf, FALSE);
1662 if (got_int)
1663 {
Bram Moolenaar436b5ad2021-12-31 22:49:24 +00001664 errmsg = (char_u *)_(e_interrupted);
Bram Moolenaar473952e2019-09-28 16:30:04 +02001665 goto restore_backup;
1666 }
1667 }
1668
1669#ifdef VMS
1670 vms_remove_version(fname); // remove version
1671#endif
1672 // Default: write the file directly. May write to a temp file for
1673 // multi-byte conversion.
1674 wfname = fname;
1675
1676 // Check for forced 'fileencoding' from "++opt=val" argument.
1677 if (eap != NULL && eap->force_enc != 0)
1678 {
1679 fenc = eap->cmd + eap->force_enc;
1680 fenc = enc_canonize(fenc);
1681 fenc_tofree = fenc;
1682 }
1683 else
1684 fenc = buf->b_p_fenc;
1685
1686 // Check if the file needs to be converted.
1687 converted = need_conversion(fenc);
1688
1689 // Check if UTF-8 to UCS-2/4 or Latin1 conversion needs to be done. Or
1690 // Latin1 to Unicode conversion. This is handled in buf_write_bytes().
1691 // Prepare the flags for it and allocate bw_conv_buf when needed.
1692 if (converted && (enc_utf8 || STRCMP(p_enc, "latin1") == 0))
1693 {
1694 wb_flags = get_fio_flags(fenc);
1695 if (wb_flags & (FIO_UCS2 | FIO_UCS4 | FIO_UTF16 | FIO_UTF8))
1696 {
1697 // Need to allocate a buffer to translate into.
1698 if (wb_flags & (FIO_UCS2 | FIO_UTF16 | FIO_UTF8))
1699 write_info.bw_conv_buflen = bufsize * 2;
1700 else // FIO_UCS4
1701 write_info.bw_conv_buflen = bufsize * 4;
1702 write_info.bw_conv_buf = alloc(write_info.bw_conv_buflen);
1703 if (write_info.bw_conv_buf == NULL)
1704 end = 0;
1705 }
1706 }
1707
1708#ifdef MSWIN
1709 if (converted && wb_flags == 0 && (wb_flags = get_win_fio_flags(fenc)) != 0)
1710 {
1711 // Convert UTF-8 -> UCS-2 and UCS-2 -> DBCS. Worst-case * 4:
1712 write_info.bw_conv_buflen = bufsize * 4;
1713 write_info.bw_conv_buf = alloc(write_info.bw_conv_buflen);
1714 if (write_info.bw_conv_buf == NULL)
1715 end = 0;
1716 }
1717#endif
1718
1719#ifdef MACOS_CONVERT
1720 if (converted && wb_flags == 0 && (wb_flags = get_mac_fio_flags(fenc)) != 0)
1721 {
1722 write_info.bw_conv_buflen = bufsize * 3;
1723 write_info.bw_conv_buf = alloc(write_info.bw_conv_buflen);
1724 if (write_info.bw_conv_buf == NULL)
1725 end = 0;
1726 }
1727#endif
1728
1729#if defined(FEAT_EVAL) || defined(USE_ICONV)
1730 if (converted && wb_flags == 0)
1731 {
1732# ifdef USE_ICONV
1733 // Use iconv() conversion when conversion is needed and it's not done
1734 // internally.
1735 write_info.bw_iconv_fd = (iconv_t)my_iconv_open(fenc,
1736 enc_utf8 ? (char_u *)"utf-8" : p_enc);
1737 if (write_info.bw_iconv_fd != (iconv_t)-1)
1738 {
1739 // We're going to use iconv(), allocate a buffer to convert in.
1740 write_info.bw_conv_buflen = bufsize * ICONV_MULT;
1741 write_info.bw_conv_buf = alloc(write_info.bw_conv_buflen);
1742 if (write_info.bw_conv_buf == NULL)
1743 end = 0;
1744 write_info.bw_first = TRUE;
1745 }
1746# ifdef FEAT_EVAL
1747 else
1748# endif
1749# endif
1750
1751# ifdef FEAT_EVAL
1752 // When the file needs to be converted with 'charconvert' after
1753 // writing, write to a temp file instead and let the conversion
1754 // overwrite the original file.
1755 if (*p_ccv != NUL)
1756 {
1757 wfname = vim_tempname('w', FALSE);
1758 if (wfname == NULL) // Can't write without a tempfile!
1759 {
Bram Moolenaar6d057012021-12-31 18:49:43 +00001760 errmsg = (char_u *)_(e_cant_find_temp_file_for_writing);
Bram Moolenaar473952e2019-09-28 16:30:04 +02001761 goto restore_backup;
1762 }
1763 }
1764# endif
1765 }
1766#endif
1767 if (converted && wb_flags == 0
1768#ifdef USE_ICONV
1769 && write_info.bw_iconv_fd == (iconv_t)-1
1770# endif
1771# ifdef FEAT_EVAL
1772 && wfname == fname
1773# endif
1774 )
1775 {
1776 if (!forceit)
1777 {
Bram Moolenaar6d057012021-12-31 18:49:43 +00001778 errmsg = (char_u *)_(e_cannot_convert_add_bang_to_write_without_conversion);
Bram Moolenaar473952e2019-09-28 16:30:04 +02001779 goto restore_backup;
1780 }
1781 notconverted = TRUE;
1782 }
1783
1784 // If conversion is taking place, we may first pretend to write and check
1785 // for conversion errors. Then loop again to write for real.
1786 // When not doing conversion this writes for real right away.
1787 for (checking_conversion = TRUE; ; checking_conversion = FALSE)
1788 {
1789 // There is no need to check conversion when:
1790 // - there is no conversion
1791 // - we make a backup file, that can be restored in case of conversion
1792 // failure.
1793 if (!converted || dobackup)
1794 checking_conversion = FALSE;
1795
1796 if (checking_conversion)
1797 {
1798 // Make sure we don't write anything.
1799 fd = -1;
1800 write_info.bw_fd = fd;
1801 }
1802 else
1803 {
1804#ifdef HAVE_FTRUNCATE
1805# define TRUNC_ON_OPEN 0
1806#else
1807# define TRUNC_ON_OPEN O_TRUNC
1808#endif
1809 // Open the file "wfname" for writing.
1810 // We may try to open the file twice: If we can't write to the file
1811 // and forceit is TRUE we delete the existing file and try to
1812 // create a new one. If this still fails we may have lost the
1813 // original file! (this may happen when the user reached his
1814 // quotum for number of files).
1815 // Appending will fail if the file does not exist and forceit is
1816 // FALSE.
1817 while ((fd = mch_open((char *)wfname, O_WRONLY | O_EXTRA | (append
1818 ? (forceit ? (O_APPEND | O_CREAT) : O_APPEND)
1819 : (O_CREAT | TRUNC_ON_OPEN))
1820 , perm < 0 ? 0666 : (perm & 0777))) < 0)
1821 {
1822 // A forced write will try to create a new file if the old one
1823 // is still readonly. This may also happen when the directory
1824 // is read-only. In that case the mch_remove() will fail.
1825 if (errmsg == NULL)
1826 {
1827#ifdef UNIX
1828 stat_T st;
1829
1830 // Don't delete the file when it's a hard or symbolic link.
1831 if ((!newfile && st_old.st_nlink > 1)
1832 || (mch_lstat((char *)fname, &st) == 0
1833 && (st.st_dev != st_old.st_dev
1834 || st.st_ino != st_old.st_ino)))
Bram Moolenaar1a992222021-12-31 17:25:48 +00001835 errmsg =
1836 (char_u *)_(e_cant_open_linked_file_for_writing);
Bram Moolenaar473952e2019-09-28 16:30:04 +02001837 else
1838#endif
1839 {
Bram Moolenaar6d057012021-12-31 18:49:43 +00001840 errmsg = (char_u *)_(e_cant_open_file_for_writing);
Bram Moolenaar473952e2019-09-28 16:30:04 +02001841 if (forceit && vim_strchr(p_cpo, CPO_FWRITE) == NULL
1842 && perm >= 0)
1843 {
1844#ifdef UNIX
1845 // we write to the file, thus it should be marked
1846 // writable after all
1847 if (!(perm & 0200))
1848 made_writable = TRUE;
1849 perm |= 0200;
1850 if (st_old.st_uid != getuid()
1851 || st_old.st_gid != getgid())
1852 perm &= 0777;
1853#endif
1854 if (!append) // don't remove when appending
1855 mch_remove(wfname);
1856 continue;
1857 }
1858 }
1859 }
1860
1861restore_backup:
1862 {
1863 stat_T st;
1864
1865 // If we failed to open the file, we don't need a backup.
1866 // Throw it away. If we moved or removed the original file
1867 // try to put the backup in its place.
1868 if (backup != NULL && wfname == fname)
1869 {
1870 if (backup_copy)
1871 {
1872 // There is a small chance that we removed the
1873 // original, try to move the copy in its place.
1874 // This may not work if the vim_rename() fails.
1875 // In that case we leave the copy around.
1876
1877 // If file does not exist, put the copy in its
1878 // place
1879 if (mch_stat((char *)fname, &st) < 0)
1880 vim_rename(backup, fname);
1881 // if original file does exist throw away the copy
1882 if (mch_stat((char *)fname, &st) >= 0)
1883 mch_remove(backup);
1884 }
1885 else
1886 {
1887 // try to put the original file back
1888 vim_rename(backup, fname);
1889 }
1890 }
1891
1892 // if original file no longer exists give an extra warning
1893 if (!newfile && mch_stat((char *)fname, &st) < 0)
1894 end = 0;
1895 }
1896
1897 if (wfname != fname)
1898 vim_free(wfname);
1899 goto fail;
1900 }
1901 write_info.bw_fd = fd;
1902
1903#if defined(UNIX)
1904 {
1905 stat_T st;
1906
1907 // Double check we are writing the intended file before making
1908 // any changes.
1909 if (overwriting
1910 && (!dobackup || backup_copy)
1911 && fname == wfname
1912 && perm >= 0
1913 && mch_fstat(fd, &st) == 0
1914 && st.st_ino != st_old.st_ino)
1915 {
1916 close(fd);
Bram Moolenaarf1474d82021-12-31 19:59:55 +00001917 errmsg = (char_u *)_(e_file_changed_while_writing);
Bram Moolenaar473952e2019-09-28 16:30:04 +02001918 goto fail;
1919 }
1920 }
1921#endif
1922#ifdef HAVE_FTRUNCATE
1923 if (!append)
1924 vim_ignored = ftruncate(fd, (off_t)0);
1925#endif
1926
1927#if defined(MSWIN)
1928 if (backup != NULL && overwriting && !append)
Bram Moolenaar7781ebe2021-02-01 20:35:01 +01001929 (void)mch_copy_file_attribute(backup, wfname);
Bram Moolenaar473952e2019-09-28 16:30:04 +02001930
1931 if (!overwriting && !append)
1932 {
1933 if (buf->b_ffname != NULL)
1934 (void)mch_copy_file_attribute(buf->b_ffname, wfname);
1935 // Should copy resource fork
1936 }
1937#endif
1938
1939#ifdef FEAT_CRYPT
1940 if (*buf->b_p_key != NUL && !filtering)
1941 {
1942 char_u *header;
1943 int header_len;
1944
1945 buf->b_cryptstate = crypt_create_for_writing(
1946 crypt_get_method_nr(buf),
1947 buf->b_p_key, &header, &header_len);
1948 if (buf->b_cryptstate == NULL || header == NULL)
1949 end = 0;
1950 else
1951 {
1952 // Write magic number, so that Vim knows how this file is
1953 // encrypted when reading it back.
1954 write_info.bw_buf = header;
1955 write_info.bw_len = header_len;
1956 write_info.bw_flags = FIO_NOCONVERT;
1957 if (buf_write_bytes(&write_info) == FAIL)
1958 end = 0;
1959 wb_flags |= FIO_ENCRYPTED;
1960 vim_free(header);
1961 }
1962 }
1963#endif
1964 }
1965 errmsg = NULL;
1966
1967 write_info.bw_buf = buffer;
1968 nchars = 0;
1969
1970 // use "++bin", "++nobin" or 'binary'
1971 if (eap != NULL && eap->force_bin != 0)
1972 write_bin = (eap->force_bin == FORCE_BIN);
1973 else
1974 write_bin = buf->b_p_bin;
1975
1976 // The BOM is written just after the encryption magic number.
1977 // Skip it when appending and the file already existed, the BOM only
1978 // makes sense at the start of the file.
1979 if (buf->b_p_bomb && !write_bin && (!append || perm < 0))
1980 {
1981 write_info.bw_len = make_bom(buffer, fenc);
1982 if (write_info.bw_len > 0)
1983 {
1984 // don't convert, do encryption
1985 write_info.bw_flags = FIO_NOCONVERT | wb_flags;
1986 if (buf_write_bytes(&write_info) == FAIL)
1987 end = 0;
1988 else
1989 nchars += write_info.bw_len;
1990 }
1991 }
1992 write_info.bw_start_lnum = start;
1993
1994#ifdef FEAT_PERSISTENT_UNDO
1995 write_undo_file = (buf->b_p_udf
1996 && overwriting
1997 && !append
1998 && !filtering
Bram Moolenaar65aee0b2021-06-27 14:08:24 +02001999# ifdef CRYPT_NOT_INPLACE
2000 // writing undo file requires
2001 // crypt_encode_inplace()
Christian Brabandt8a4c8122021-07-25 14:36:05 +02002002 && (buf->b_cryptstate == NULL
2003 || crypt_works_inplace(buf->b_cryptstate))
Bram Moolenaar65aee0b2021-06-27 14:08:24 +02002004# endif
Bram Moolenaar473952e2019-09-28 16:30:04 +02002005 && reset_changed
2006 && !checking_conversion);
Christian Brabandt8a4c8122021-07-25 14:36:05 +02002007# ifdef CRYPT_NOT_INPLACE
2008 // remove undo file if encrypting it is not possible
2009 if (buf->b_p_udf
2010 && overwriting
2011 && !append
2012 && !filtering
2013 && !checking_conversion
2014 && buf->b_cryptstate != NULL
2015 && !crypt_works_inplace(buf->b_cryptstate))
2016 u_undofile_reset_and_delete(buf);
2017# endif
Bram Moolenaar473952e2019-09-28 16:30:04 +02002018 if (write_undo_file)
2019 // Prepare for computing the hash value of the text.
2020 sha256_start(&sha_ctx);
2021#endif
2022
2023 write_info.bw_len = bufsize;
2024 write_info.bw_flags = wb_flags;
2025 fileformat = get_fileformat_force(buf, eap);
2026 s = buffer;
2027 len = 0;
2028 for (lnum = start; lnum <= end; ++lnum)
2029 {
2030 // The next while loop is done once for each character written.
2031 // Keep it fast!
2032 ptr = ml_get_buf(buf, lnum, FALSE) - 1;
2033#ifdef FEAT_PERSISTENT_UNDO
2034 if (write_undo_file)
2035 sha256_update(&sha_ctx, ptr + 1,
2036 (UINT32_T)(STRLEN(ptr + 1) + 1));
2037#endif
2038 while ((c = *++ptr) != NUL)
2039 {
2040 if (c == NL)
2041 *s = NUL; // replace newlines with NULs
2042 else if (c == CAR && fileformat == EOL_MAC)
2043 *s = NL; // Mac: replace CRs with NLs
2044 else
2045 *s = c;
2046 ++s;
2047 if (++len != bufsize)
2048 continue;
Christian Brabandtf573c6e2021-06-20 14:02:16 +02002049#ifdef FEAT_CRYPT
2050 if (write_info.bw_fd > 0 && lnum == end
2051 && (write_info.bw_flags & FIO_ENCRYPTED)
2052 && *buf->b_p_key != NUL && !filtering
2053 && *ptr == NUL)
2054 write_info.bw_finish = TRUE;
2055 #endif
Bram Moolenaar473952e2019-09-28 16:30:04 +02002056 if (buf_write_bytes(&write_info) == FAIL)
2057 {
2058 end = 0; // write error: break loop
2059 break;
2060 }
2061 nchars += bufsize;
2062 s = buffer;
2063 len = 0;
2064 write_info.bw_start_lnum = lnum;
2065 }
2066 // write failed or last line has no EOL: stop here
2067 if (end == 0
2068 || (lnum == end
2069 && (write_bin || !buf->b_p_fixeol)
Bram Moolenaarb3c8b1d2020-12-23 18:54:57 +01002070 && ((write_bin && lnum == buf->b_no_eol_lnum)
Bram Moolenaar473952e2019-09-28 16:30:04 +02002071 || (lnum == buf->b_ml.ml_line_count
2072 && !buf->b_p_eol))))
2073 {
2074 ++lnum; // written the line, count it
2075 no_eol = TRUE;
2076 break;
2077 }
2078 if (fileformat == EOL_UNIX)
2079 *s++ = NL;
2080 else
2081 {
2082 *s++ = CAR; // EOL_MAC or EOL_DOS: write CR
2083 if (fileformat == EOL_DOS) // write CR-NL
2084 {
2085 if (++len == bufsize)
2086 {
2087 if (buf_write_bytes(&write_info) == FAIL)
2088 {
2089 end = 0; // write error: break loop
2090 break;
2091 }
2092 nchars += bufsize;
2093 s = buffer;
2094 len = 0;
2095 }
2096 *s++ = NL;
2097 }
2098 }
2099 if (++len == bufsize && end)
2100 {
2101 if (buf_write_bytes(&write_info) == FAIL)
2102 {
2103 end = 0; // write error: break loop
2104 break;
2105 }
2106 nchars += bufsize;
2107 s = buffer;
2108 len = 0;
2109
2110 ui_breakcheck();
2111 if (got_int)
2112 {
2113 end = 0; // Interrupted, break loop
2114 break;
2115 }
2116 }
2117#ifdef VMS
2118 // On VMS there is a problem: newlines get added when writing
2119 // blocks at a time. Fix it by writing a line at a time.
2120 // This is much slower!
2121 // Explanation: VAX/DECC RTL insists that records in some RMS
2122 // structures end with a newline (carriage return) character, and
2123 // if they don't it adds one.
2124 // With other RMS structures it works perfect without this fix.
Bram Moolenaar95f0b6e2019-12-15 12:54:18 +01002125# ifndef MIN
2126// Older DECC compiler for VAX doesn't define MIN()
2127# define MIN(a, b) ((a) < (b) ? (a) : (b))
2128# endif
Bram Moolenaar473952e2019-09-28 16:30:04 +02002129 if (buf->b_fab_rfm == FAB$C_VFC
2130 || ((buf->b_fab_rat & (FAB$M_FTN | FAB$M_CR)) != 0))
2131 {
2132 int b2write;
2133
2134 buf->b_fab_mrs = (buf->b_fab_mrs == 0
2135 ? MIN(4096, bufsize)
2136 : MIN(buf->b_fab_mrs, bufsize));
2137
2138 b2write = len;
2139 while (b2write > 0)
2140 {
2141 write_info.bw_len = MIN(b2write, buf->b_fab_mrs);
2142 if (buf_write_bytes(&write_info) == FAIL)
2143 {
2144 end = 0;
2145 break;
2146 }
2147 b2write -= MIN(b2write, buf->b_fab_mrs);
2148 }
2149 write_info.bw_len = bufsize;
2150 nchars += len;
2151 s = buffer;
2152 len = 0;
2153 }
2154#endif
2155 }
2156 if (len > 0 && end > 0)
2157 {
2158 write_info.bw_len = len;
Christian Brabandtf573c6e2021-06-20 14:02:16 +02002159#ifdef FEAT_CRYPT
2160 if (write_info.bw_fd > 0 && lnum >= end
2161 && (write_info.bw_flags & FIO_ENCRYPTED)
2162 && *buf->b_p_key != NUL && !filtering)
2163 write_info.bw_finish = TRUE;
2164 #endif
Bram Moolenaar473952e2019-09-28 16:30:04 +02002165 if (buf_write_bytes(&write_info) == FAIL)
2166 end = 0; // write error
2167 nchars += len;
2168 }
2169
K.Takata3af98212022-11-01 20:36:19 +00002170 if (!buf->b_p_fixeol && buf->b_p_eof)
2171 {
2172 // write trailing CTRL-Z
2173 (void)write_eintr(write_info.bw_fd, "\x1a", 1);
2174 nchars++;
2175 }
2176
Bram Moolenaar473952e2019-09-28 16:30:04 +02002177 // Stop when writing done or an error was encountered.
2178 if (!checking_conversion || end == 0)
2179 break;
2180
2181 // If no error happened until now, writing should be ok, so loop to
2182 // really write the buffer.
2183 }
2184
2185 // If we started writing, finish writing. Also when an error was
2186 // encountered.
2187 if (!checking_conversion)
2188 {
2189#if defined(UNIX) && defined(HAVE_FSYNC)
Bram Moolenaar8e7d6222020-12-18 19:49:56 +01002190 // On many journaling file systems there is a bug that causes both the
Bram Moolenaar473952e2019-09-28 16:30:04 +02002191 // original and the backup file to be lost when halting the system
2192 // right after writing the file. That's because only the meta-data is
2193 // journalled. Syncing the file slows down the system, but assures it
2194 // has been written to disk and we don't lose it.
2195 // For a device do try the fsync() but don't complain if it does not
2196 // work (could be a pipe).
2197 // If the 'fsync' option is FALSE, don't fsync(). Useful for laptops.
2198 if (p_fs && vim_fsync(fd) != 0 && !device)
2199 {
Bram Moolenaar460ae5d2022-01-01 14:19:49 +00002200 errmsg = (char_u *)_(e_fsync_failed);
Bram Moolenaar473952e2019-09-28 16:30:04 +02002201 end = 0;
2202 }
2203#endif
2204
Christian Brabandte085dfd2023-09-30 12:49:18 +02002205#if defined(HAVE_SELINUX) || defined(HAVE_SMACK) || defined(FEAT_XATTR)
Bram Moolenaar473952e2019-09-28 16:30:04 +02002206 // Probably need to set the security context.
2207 if (!backup_copy)
Christian Brabandte085dfd2023-09-30 12:49:18 +02002208 {
2209#if defined(HAVE_SELINUX) || defined(HAVE_SMACK)
Bram Moolenaar473952e2019-09-28 16:30:04 +02002210 mch_copy_sec(backup, wfname);
2211#endif
Christian Brabandte085dfd2023-09-30 12:49:18 +02002212#ifdef FEAT_XATTR
2213 mch_copy_xattr(backup, wfname);
2214#endif
2215 }
2216#endif
Bram Moolenaar473952e2019-09-28 16:30:04 +02002217
2218#ifdef UNIX
2219 // When creating a new file, set its owner/group to that of the
2220 // original file. Get the new device and inode number.
2221 if (backup != NULL && !backup_copy)
2222 {
2223# ifdef HAVE_FCHOWN
2224 stat_T st;
2225
2226 // Don't change the owner when it's already OK, some systems remove
2227 // permission or ACL stuff.
2228 if (mch_stat((char *)wfname, &st) < 0
2229 || st.st_uid != st_old.st_uid
2230 || st.st_gid != st_old.st_gid)
2231 {
2232 // changing owner might not be possible
2233 vim_ignored = fchown(fd, st_old.st_uid, -1);
2234 // if changing group fails clear the group permissions
2235 if (fchown(fd, -1, st_old.st_gid) == -1 && perm > 0)
2236 perm &= ~070;
2237 }
2238# endif
2239 buf_setino(buf);
2240 }
2241 else if (!buf->b_dev_valid)
2242 // Set the inode when creating a new file.
2243 buf_setino(buf);
2244#endif
2245
2246#ifdef UNIX
2247 if (made_writable)
2248 perm &= ~0200; // reset 'w' bit for security reasons
2249#endif
2250#ifdef HAVE_FCHMOD
2251 // set permission of new file same as old file
2252 if (perm >= 0)
2253 (void)mch_fsetperm(fd, perm);
2254#endif
2255 if (close(fd) != 0)
2256 {
Bram Moolenaarf1474d82021-12-31 19:59:55 +00002257 errmsg = (char_u *)_(e_close_failed);
Bram Moolenaar473952e2019-09-28 16:30:04 +02002258 end = 0;
2259 }
2260
2261#ifndef HAVE_FCHMOD
2262 // set permission of new file same as old file
2263 if (perm >= 0)
2264 (void)mch_setperm(wfname, perm);
2265#endif
2266#ifdef HAVE_ACL
2267 // Probably need to set the ACL before changing the user (can't set the
2268 // ACL on a file the user doesn't own).
2269 // On Solaris, with ZFS and the aclmode property set to "discard" (the
2270 // default), chmod() discards all part of a file's ACL that don't
2271 // represent the mode of the file. It's non-trivial for us to discover
2272 // whether we're in that situation, so we simply always re-set the ACL.
2273# ifndef HAVE_SOLARIS_ZFS_ACL
2274 if (!backup_copy)
2275# endif
2276 mch_set_acl(wfname, acl);
2277#endif
2278#ifdef FEAT_CRYPT
2279 if (buf->b_cryptstate != NULL)
2280 {
2281 crypt_free_state(buf->b_cryptstate);
2282 buf->b_cryptstate = NULL;
2283 }
2284#endif
2285
2286#if defined(FEAT_EVAL)
2287 if (wfname != fname)
2288 {
2289 // The file was written to a temp file, now it needs to be
2290 // converted with 'charconvert' to (overwrite) the output file.
2291 if (end != 0)
2292 {
2293 if (eval_charconvert(enc_utf8 ? (char_u *)"utf-8" : p_enc,
2294 fenc, wfname, fname) == FAIL)
2295 {
2296 write_info.bw_conv_error = TRUE;
2297 end = 0;
2298 }
2299 }
2300 mch_remove(wfname);
2301 vim_free(wfname);
2302 }
2303#endif
2304 }
2305
2306 if (end == 0)
2307 {
2308 // Error encountered.
2309 if (errmsg == NULL)
2310 {
2311 if (write_info.bw_conv_error)
2312 {
2313 if (write_info.bw_conv_error_lnum == 0)
Bram Moolenaarf1474d82021-12-31 19:59:55 +00002314 errmsg = (char_u *)_(e_write_error_conversion_failed_make_fenc_empty_to_override);
Bram Moolenaar473952e2019-09-28 16:30:04 +02002315 else
2316 {
2317 errmsg_allocated = TRUE;
2318 errmsg = alloc(300);
John Marriottfbe2dd72025-02-20 23:17:09 +01002319 if (errmsg == NULL)
2320 goto fail;
Bram Moolenaarf1474d82021-12-31 19:59:55 +00002321 vim_snprintf((char *)errmsg, 300, _(e_write_error_conversion_failed_in_line_nr_make_fenc_empty_to_override),
Bram Moolenaar473952e2019-09-28 16:30:04 +02002322 (long)write_info.bw_conv_error_lnum);
2323 }
2324 }
2325 else if (got_int)
Bram Moolenaar436b5ad2021-12-31 22:49:24 +00002326 errmsg = (char_u *)_(e_interrupted);
Bram Moolenaar473952e2019-09-28 16:30:04 +02002327 else
Bram Moolenaarf1474d82021-12-31 19:59:55 +00002328 errmsg = (char_u *)_(e_write_error_file_system_full);
Bram Moolenaar473952e2019-09-28 16:30:04 +02002329 }
2330
2331 // If we have a backup file, try to put it in place of the new file,
2332 // because the new file is probably corrupt. This avoids losing the
2333 // original file when trying to make a backup when writing the file a
2334 // second time.
2335 // When "backup_copy" is set we need to copy the backup over the new
2336 // file. Otherwise rename the backup file.
2337 // If this is OK, don't give the extra warning message.
2338 if (backup != NULL)
2339 {
2340 if (backup_copy)
2341 {
2342 // This may take a while, if we were interrupted let the user
2343 // know we got the message.
2344 if (got_int)
2345 {
Bram Moolenaar436b5ad2021-12-31 22:49:24 +00002346 msg(_(e_interrupted));
Bram Moolenaar473952e2019-09-28 16:30:04 +02002347 out_flush();
2348 }
2349 if ((fd = mch_open((char *)backup, O_RDONLY | O_EXTRA, 0)) >= 0)
2350 {
2351 if ((write_info.bw_fd = mch_open((char *)fname,
2352 O_WRONLY | O_CREAT | O_TRUNC | O_EXTRA,
2353 perm & 0777)) >= 0)
2354 {
2355 // copy the file.
2356 write_info.bw_buf = smallbuf;
2357 write_info.bw_flags = FIO_NOCONVERT;
2358 while ((write_info.bw_len = read_eintr(fd, smallbuf,
2359 SMALLBUFSIZE)) > 0)
2360 if (buf_write_bytes(&write_info) == FAIL)
2361 break;
2362
2363 if (close(write_info.bw_fd) >= 0
2364 && write_info.bw_len == 0)
2365 end = 1; // success
2366 }
2367 close(fd); // ignore errors for closing read file
2368 }
2369 }
2370 else
2371 {
2372 if (vim_rename(backup, fname) == 0)
2373 end = 1;
2374 }
2375 }
2376 goto fail;
2377 }
2378
2379 lnum -= start; // compute number of written lines
2380 --no_wait_return; // may wait for return now
2381
2382#if !(defined(UNIX) || defined(VMS))
2383 fname = sfname; // use shortname now, for the messages
2384#endif
2385 if (!filtering)
2386 {
2387 msg_add_fname(buf, fname); // put fname in IObuff with quotes
2388 c = FALSE;
2389 if (write_info.bw_conv_error)
2390 {
2391 STRCAT(IObuff, _(" CONVERSION ERROR"));
2392 c = TRUE;
2393 if (write_info.bw_conv_error_lnum != 0)
2394 vim_snprintf_add((char *)IObuff, IOSIZE, _(" in line %ld;"),
2395 (long)write_info.bw_conv_error_lnum);
2396 }
2397 else if (notconverted)
2398 {
2399 STRCAT(IObuff, _("[NOT converted]"));
2400 c = TRUE;
2401 }
2402 else if (converted)
2403 {
2404 STRCAT(IObuff, _("[converted]"));
2405 c = TRUE;
2406 }
2407 if (device)
2408 {
2409 STRCAT(IObuff, _("[Device]"));
2410 c = TRUE;
2411 }
2412 else if (newfile)
2413 {
Bram Moolenaar722e5052020-06-12 22:31:00 +02002414 STRCAT(IObuff, new_file_message());
Bram Moolenaar473952e2019-09-28 16:30:04 +02002415 c = TRUE;
2416 }
2417 if (no_eol)
2418 {
2419 msg_add_eol();
2420 c = TRUE;
2421 }
2422 // may add [unix/dos/mac]
2423 if (msg_add_fileformat(fileformat))
2424 c = TRUE;
2425#ifdef FEAT_CRYPT
2426 if (wb_flags & FIO_ENCRYPTED)
2427 {
2428 crypt_append_msg(buf);
2429 c = TRUE;
2430 }
2431#endif
2432 msg_add_lines(c, (long)lnum, nchars); // add line/char count
2433 if (!shortmess(SHM_WRITE))
2434 {
2435 if (append)
2436 STRCAT(IObuff, shortmess(SHM_WRI) ? _(" [a]") : _(" appended"));
2437 else
2438 STRCAT(IObuff, shortmess(SHM_WRI) ? _(" [w]") : _(" written"));
2439 }
2440
2441 set_keep_msg((char_u *)msg_trunc_attr((char *)IObuff, FALSE, 0), 0);
2442 }
2443
2444 // When written everything correctly: reset 'modified'. Unless not
2445 // writing to the original file and '+' is not in 'cpoptions'.
2446 if (reset_changed && whole && !append
2447 && !write_info.bw_conv_error
2448 && (overwriting || vim_strchr(p_cpo, CPO_PLUS) != NULL))
2449 {
2450 unchanged(buf, TRUE, FALSE);
Christian Brabandtdb3b4462021-10-16 11:58:55 +01002451 // b:changedtick may be incremented in unchanged() but that should not
2452 // trigger a TextChanged event.
Bram Moolenaar473952e2019-09-28 16:30:04 +02002453 if (buf->b_last_changedtick + 1 == CHANGEDTICK(buf))
2454 buf->b_last_changedtick = CHANGEDTICK(buf);
2455 u_unchanged(buf);
2456 u_update_save_nr(buf);
2457 }
2458
2459 // If written to the current file, update the timestamp of the swap file
2460 // and reset the BF_WRITE_MASK flags. Also sets buf->b_mtime.
2461 if (overwriting)
2462 {
2463 ml_timestamp(buf);
2464 if (append)
2465 buf->b_flags &= ~BF_NEW;
2466 else
2467 buf->b_flags &= ~BF_WRITE_MASK;
2468 }
2469
2470 // If we kept a backup until now, and we are in patch mode, then we make
2471 // the backup file our 'original' file.
2472 if (*p_pm && dobackup)
2473 {
2474 char *org = (char *)buf_modname((buf->b_p_sn || buf->b_shortname),
2475 fname, p_pm, FALSE);
2476
2477 if (backup != NULL)
2478 {
2479 stat_T st;
2480
2481 // If the original file does not exist yet
2482 // the current backup file becomes the original file
2483 if (org == NULL)
Bram Moolenaar6d057012021-12-31 18:49:43 +00002484 emsg(_(e_patchmode_cant_save_original_file));
Bram Moolenaar473952e2019-09-28 16:30:04 +02002485 else if (mch_stat(org, &st) < 0)
2486 {
2487 vim_rename(backup, (char_u *)org);
2488 VIM_CLEAR(backup); // don't delete the file
2489#ifdef UNIX
2490 set_file_time((char_u *)org, st_old.st_atime, st_old.st_mtime);
2491#endif
2492 }
2493 }
2494 // If there is no backup file, remember that a (new) file was
2495 // created.
2496 else
2497 {
2498 int empty_fd;
2499
2500 if (org == NULL
2501 || (empty_fd = mch_open(org,
2502 O_CREAT | O_EXTRA | O_EXCL | O_NOFOLLOW,
2503 perm < 0 ? 0666 : (perm & 0777))) < 0)
Bram Moolenaar6d057012021-12-31 18:49:43 +00002504 emsg(_(e_patchmode_cant_touch_empty_original_file));
Bram Moolenaar473952e2019-09-28 16:30:04 +02002505 else
2506 close(empty_fd);
2507 }
2508 if (org != NULL)
2509 {
2510 mch_setperm((char_u *)org, mch_getperm(fname) & 0777);
2511 vim_free(org);
2512 }
2513 }
2514
2515 // Remove the backup unless 'backup' option is set or there was a
2516 // conversion error.
2517 if (!p_bk && backup != NULL && !write_info.bw_conv_error
2518 && mch_remove(backup) != 0)
Bram Moolenaar6d057012021-12-31 18:49:43 +00002519 emsg(_(e_cant_delete_backup_file));
Bram Moolenaar473952e2019-09-28 16:30:04 +02002520
2521 goto nofail;
2522
2523 // Finish up. We get here either after failure or success.
2524fail:
2525 --no_wait_return; // may wait for return now
2526nofail:
2527
2528 // Done saving, we accept changed buffer warnings again
2529 buf->b_saving = FALSE;
2530
2531 vim_free(backup);
2532 if (buffer != smallbuf)
2533 vim_free(buffer);
2534 vim_free(fenc_tofree);
2535 vim_free(write_info.bw_conv_buf);
2536#ifdef USE_ICONV
2537 if (write_info.bw_iconv_fd != (iconv_t)-1)
2538 {
2539 iconv_close(write_info.bw_iconv_fd);
2540 write_info.bw_iconv_fd = (iconv_t)-1;
2541 }
2542#endif
2543#ifdef HAVE_ACL
2544 mch_free_acl(acl);
2545#endif
2546
2547 if (errmsg != NULL)
2548 {
2549 int numlen = errnum != NULL ? (int)STRLEN(errnum) : 0;
2550
2551 attr = HL_ATTR(HLF_E); // set highlight for error messages
2552 msg_add_fname(buf,
2553#ifndef UNIX
2554 sfname
2555#else
2556 fname
2557#endif
2558 ); // put file name in IObuff with quotes
2559 if (STRLEN(IObuff) + STRLEN(errmsg) + numlen >= IOSIZE)
2560 IObuff[IOSIZE - STRLEN(errmsg) - numlen - 1] = NUL;
2561 // If the error message has the form "is ...", put the error number in
2562 // front of the file name.
2563 if (errnum != NULL)
2564 {
2565 STRMOVE(IObuff + numlen, IObuff);
2566 mch_memmove(IObuff, errnum, (size_t)numlen);
2567 }
2568 STRCAT(IObuff, errmsg);
2569 emsg((char *)IObuff);
2570 if (errmsg_allocated)
2571 vim_free(errmsg);
2572
2573 retval = FAIL;
2574 if (end == 0)
2575 {
2576 msg_puts_attr(_("\nWARNING: Original file may be lost or damaged\n"),
2577 attr | MSG_HIST);
2578 msg_puts_attr(_("don't quit the editor until the file is successfully written!"),
2579 attr | MSG_HIST);
2580
2581 // Update the timestamp to avoid an "overwrite changed file"
2582 // prompt when writing again.
2583 if (mch_stat((char *)fname, &st_old) >= 0)
2584 {
2585 buf_store_time(buf, &st_old, fname);
2586 buf->b_mtime_read = buf->b_mtime;
Leah Neukirchen0a7984a2021-10-14 21:27:55 +01002587 buf->b_mtime_read_ns = buf->b_mtime_ns;
Bram Moolenaar473952e2019-09-28 16:30:04 +02002588 }
2589 }
2590 }
2591 msg_scroll = msg_save;
2592
2593#ifdef FEAT_PERSISTENT_UNDO
2594 // When writing the whole file and 'undofile' is set, also write the undo
2595 // file.
2596 if (retval == OK && write_undo_file)
2597 {
2598 char_u hash[UNDO_HASH_SIZE];
2599
2600 sha256_finish(&sha_ctx, hash);
2601 u_write_undo(NULL, FALSE, buf, hash);
2602 }
2603#endif
2604
2605#ifdef FEAT_EVAL
2606 if (!should_abort(retval))
2607#else
2608 if (!got_int)
2609#endif
2610 {
2611 aco_save_T aco;
2612
2613 curbuf->b_no_eol_lnum = 0; // in case it was set by the previous read
2614
2615 // Apply POST autocommands.
2616 // Careful: The autocommands may call buf_write() recursively!
Bram Moolenaare76062c2022-11-28 18:51:43 +00002617 // Only do this when a window was found for "buf".
Bram Moolenaar473952e2019-09-28 16:30:04 +02002618 aucmd_prepbuf(&aco, buf);
Bram Moolenaare76062c2022-11-28 18:51:43 +00002619 if (curbuf == buf)
2620 {
2621 if (append)
2622 apply_autocmds_exarg(EVENT_FILEAPPENDPOST, fname, fname,
2623 FALSE, curbuf, eap);
2624 else if (filtering)
2625 apply_autocmds_exarg(EVENT_FILTERWRITEPOST, NULL, fname,
2626 FALSE, curbuf, eap);
2627 else if (reset_changed && whole)
2628 apply_autocmds_exarg(EVENT_BUFWRITEPOST, fname, fname,
2629 FALSE, curbuf, eap);
2630 else
2631 apply_autocmds_exarg(EVENT_FILEWRITEPOST, fname, fname,
2632 FALSE, curbuf, eap);
Bram Moolenaar473952e2019-09-28 16:30:04 +02002633
Bram Moolenaare76062c2022-11-28 18:51:43 +00002634 // restore curwin/curbuf and a few other things
2635 aucmd_restbuf(&aco);
2636 }
Bram Moolenaar473952e2019-09-28 16:30:04 +02002637
2638#ifdef FEAT_EVAL
2639 if (aborting()) // autocmds may abort script processing
2640 retval = FALSE;
2641#endif
2642 }
2643
Bram Moolenaar8e6be342020-11-23 22:01:26 +01002644#ifdef FEAT_VIMINFO
2645 // Make sure marks will be written out to the viminfo file later, even when
2646 // the file is new.
2647 curbuf->b_marks_read = TRUE;
2648#endif
2649
Bram Moolenaar473952e2019-09-28 16:30:04 +02002650 got_int |= prev_got_int;
2651
2652 return retval;
2653}