blob: b4e02d7716ed489734a97de2172d197dbb22c3df [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/* for debugging */
Bram Moolenaar6f470022018-04-10 18:47:20 +020011/* #define CHECK(c, s) do { if (c) EMSG(s); } while (0) */
12#define CHECK(c, s) do { /**/ } while (0)
Bram Moolenaar071d4272004-06-13 20:20:40 +000013
14/*
15 * memline.c: Contains the functions for appending, deleting and changing the
Bram Moolenaar4770d092006-01-12 23:22:24 +000016 * text lines. The memfile functions are used to store the information in
17 * blocks of memory, backed up by a file. The structure of the information is
18 * a tree. The root of the tree is a pointer block. The leaves of the tree
19 * are data blocks. In between may be several layers of pointer blocks,
20 * forming branches.
Bram Moolenaar071d4272004-06-13 20:20:40 +000021 *
22 * Three types of blocks are used:
23 * - Block nr 0 contains information for recovery
24 * - Pointer blocks contain list of pointers to other blocks.
25 * - Data blocks contain the actual text.
26 *
27 * Block nr 0 contains the block0 structure (see below).
28 *
29 * Block nr 1 is the first pointer block. It is the root of the tree.
30 * Other pointer blocks are branches.
31 *
32 * If a line is too big to fit in a single page, the block containing that
33 * line is made big enough to hold the line. It may span several pages.
34 * Otherwise all blocks are one page.
35 *
36 * A data block that was filled when starting to edit a file and was not
37 * changed since then, can have a negative block number. This means that it
38 * has not yet been assigned a place in the file. When recovering, the lines
39 * in this data block can be read from the original file. When the block is
40 * changed (lines appended/deleted/changed) or when it is flushed it gets a
41 * positive number. Use mf_trans_del() to get the new number, before calling
42 * mf_get().
43 */
44
Bram Moolenaar071d4272004-06-13 20:20:40 +000045#include "vim.h"
46
Bram Moolenaar071d4272004-06-13 20:20:40 +000047#ifndef UNIX /* it's in os_unix.h for Unix */
48# include <time.h>
49#endif
50
Bram Moolenaar5a6404c2006-11-01 17:12:57 +000051#if defined(SASC) || defined(__amigaos4__)
Bram Moolenaar071d4272004-06-13 20:20:40 +000052# include <proto/dos.h> /* for Open() and Close() */
53#endif
54
55typedef struct block0 ZERO_BL; /* contents of the first block */
56typedef struct pointer_block PTR_BL; /* contents of a pointer block */
57typedef struct data_block DATA_BL; /* contents of a data block */
58typedef struct pointer_entry PTR_EN; /* block/line-count pair */
59
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +020060#define DATA_ID (('d' << 8) + 'a') /* data block id */
61#define PTR_ID (('p' << 8) + 't') /* pointer block id */
62#define BLOCK0_ID0 'b' /* block 0 id 0 */
63#define BLOCK0_ID1 '0' /* block 0 id 1 */
64#define BLOCK0_ID1_C0 'c' /* block 0 id 1 'cm' 0 */
65#define BLOCK0_ID1_C1 'C' /* block 0 id 1 'cm' 1 */
Bram Moolenaar8f4ac012014-08-10 13:38:34 +020066#define BLOCK0_ID1_C2 'd' /* block 0 id 1 'cm' 2 */
67
68#if defined(FEAT_CRYPT)
69static int id1_codes[] = {
70 BLOCK0_ID1_C0, /* CRYPT_M_ZIP */
71 BLOCK0_ID1_C1, /* CRYPT_M_BF */
72 BLOCK0_ID1_C2, /* CRYPT_M_BF2 */
73};
74#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +000075
76/*
77 * pointer to a block, used in a pointer block
78 */
79struct pointer_entry
80{
81 blocknr_T pe_bnum; /* block number */
82 linenr_T pe_line_count; /* number of lines in this branch */
83 linenr_T pe_old_lnum; /* lnum for this block (for recovery) */
84 int pe_page_count; /* number of pages in block pe_bnum */
85};
86
87/*
88 * A pointer block contains a list of branches in the tree.
89 */
90struct pointer_block
91{
92 short_u pb_id; /* ID for pointer block: PTR_ID */
Bram Moolenaar20a825a2010-05-31 21:27:30 +020093 short_u pb_count; /* number of pointers in this block */
Bram Moolenaar071d4272004-06-13 20:20:40 +000094 short_u pb_count_max; /* maximum value for pb_count */
95 PTR_EN pb_pointer[1]; /* list of pointers to blocks (actually longer)
96 * followed by empty space until end of page */
97};
98
99/*
100 * A data block is a leaf in the tree.
101 *
102 * The text of the lines is at the end of the block. The text of the first line
103 * in the block is put at the end, the text of the second line in front of it,
104 * etc. Thus the order of the lines is the opposite of the line number.
105 */
106struct data_block
107{
108 short_u db_id; /* ID for data block: DATA_ID */
109 unsigned db_free; /* free space available */
110 unsigned db_txt_start; /* byte where text starts */
111 unsigned db_txt_end; /* byte just after data block */
112 linenr_T db_line_count; /* number of lines in this block */
113 unsigned db_index[1]; /* index for start of line (actually bigger)
114 * followed by empty space upto db_txt_start
115 * followed by the text in the lines until
116 * end of page */
117};
118
119/*
120 * The low bits of db_index hold the actual index. The topmost bit is
121 * used for the global command to be able to mark a line.
122 * This method is not clean, but otherwise there would be at least one extra
123 * byte used for each line.
124 * The mark has to be in this place to keep it with the correct line when other
125 * lines are inserted or deleted.
126 */
127#define DB_MARKED ((unsigned)1 << ((sizeof(unsigned) * 8) - 1))
128#define DB_INDEX_MASK (~DB_MARKED)
129
130#define INDEX_SIZE (sizeof(unsigned)) /* size of one db_index entry */
131#define HEADER_SIZE (sizeof(DATA_BL) - INDEX_SIZE) /* size of data block header */
132
Bram Moolenaar1cd871b2004-12-19 22:46:22 +0000133#define B0_FNAME_SIZE_ORG 900 /* what it was in older versions */
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200134#define B0_FNAME_SIZE_NOCRYPT 898 /* 2 bytes used for other things */
135#define B0_FNAME_SIZE_CRYPT 890 /* 10 bytes used for other things */
Bram Moolenaar1cd871b2004-12-19 22:46:22 +0000136#define B0_UNAME_SIZE 40
137#define B0_HNAME_SIZE 40
Bram Moolenaar071d4272004-06-13 20:20:40 +0000138/*
139 * Restrict the numbers to 32 bits, otherwise most compilers will complain.
140 * This won't detect a 64 bit machine that only swaps a byte in the top 32
141 * bits, but that is crazy anyway.
142 */
143#define B0_MAGIC_LONG 0x30313233L
144#define B0_MAGIC_INT 0x20212223L
145#define B0_MAGIC_SHORT 0x10111213L
146#define B0_MAGIC_CHAR 0x55
147
148/*
149 * Block zero holds all info about the swap file.
150 *
151 * NOTE: DEFINITION OF BLOCK 0 SHOULD NOT CHANGE! It would make all existing
152 * swap files unusable!
153 *
154 * If size of block0 changes anyway, adjust MIN_SWAP_PAGE_SIZE in vim.h!!
155 *
Bram Moolenaarbae0c162007-05-10 19:30:25 +0000156 * This block is built up of single bytes, to make it portable across
Bram Moolenaar071d4272004-06-13 20:20:40 +0000157 * different machines. b0_magic_* is used to check the byte order and size of
158 * variables, because the rest of the swap file is not portable.
159 */
160struct block0
161{
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200162 char_u b0_id[2]; /* id for block 0: BLOCK0_ID0 and BLOCK0_ID1,
Bram Moolenaar8f4ac012014-08-10 13:38:34 +0200163 * BLOCK0_ID1_C0, BLOCK0_ID1_C1, etc. */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000164 char_u b0_version[10]; /* Vim version string */
165 char_u b0_page_size[4];/* number of bytes per page */
166 char_u b0_mtime[4]; /* last modification time of file */
167 char_u b0_ino[4]; /* inode of b0_fname */
168 char_u b0_pid[4]; /* process id of creator (or 0) */
169 char_u b0_uname[B0_UNAME_SIZE]; /* name of user (uid if no name) */
170 char_u b0_hname[B0_HNAME_SIZE]; /* host name (if it has a name) */
Bram Moolenaar1cd871b2004-12-19 22:46:22 +0000171 char_u b0_fname[B0_FNAME_SIZE_ORG]; /* name of file being edited */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000172 long b0_magic_long; /* check for byte order of long */
173 int b0_magic_int; /* check for byte order of int */
174 short b0_magic_short; /* check for byte order of short */
175 char_u b0_magic_char; /* check for last char */
176};
Bram Moolenaar1cd871b2004-12-19 22:46:22 +0000177
178/*
Bram Moolenaar4770d092006-01-12 23:22:24 +0000179 * Note: b0_dirty and b0_flags are put at the end of the file name. For very
Bram Moolenaar1cd871b2004-12-19 22:46:22 +0000180 * long file names in older versions of Vim they are invalid.
181 * The 'fileencoding' comes before b0_flags, with a NUL in front. But only
182 * when there is room, for very long file names it's omitted.
183 */
184#define B0_DIRTY 0x55
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200185#define b0_dirty b0_fname[B0_FNAME_SIZE_ORG - 1]
Bram Moolenaar1cd871b2004-12-19 22:46:22 +0000186
187/*
188 * The b0_flags field is new in Vim 7.0.
189 */
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200190#define b0_flags b0_fname[B0_FNAME_SIZE_ORG - 2]
191
192/*
193 * Crypt seed goes here, 8 bytes. New in Vim 7.3.
194 * Without encryption these bytes may be used for 'fenc'.
195 */
196#define b0_seed b0_fname[B0_FNAME_SIZE_ORG - 2 - MF_SEED_LEN]
Bram Moolenaar1cd871b2004-12-19 22:46:22 +0000197
198/* The lowest two bits contain the fileformat. Zero means it's not set
199 * (compatible with Vim 6.x), otherwise it's EOL_UNIX + 1, EOL_DOS + 1 or
200 * EOL_MAC + 1. */
201#define B0_FF_MASK 3
202
203/* Swap file is in directory of edited file. Used to find the file from
204 * different mount points. */
205#define B0_SAME_DIR 4
206
207/* The 'fileencoding' is at the end of b0_fname[], with a NUL in front of it.
208 * When empty there is only the NUL. */
209#define B0_HAS_FENC 8
Bram Moolenaar071d4272004-06-13 20:20:40 +0000210
211#define STACK_INCR 5 /* nr of entries added to ml_stack at a time */
212
213/*
214 * The line number where the first mark may be is remembered.
215 * If it is 0 there are no marks at all.
216 * (always used for the current buffer only, no buffer change possible while
217 * executing a global command).
218 */
219static linenr_T lowest_marked = 0;
220
221/*
222 * arguments for ml_find_line()
223 */
224#define ML_DELETE 0x11 /* delete line */
225#define ML_INSERT 0x12 /* insert line */
226#define ML_FIND 0x13 /* just find the line */
227#define ML_FLUSH 0x02 /* flush locked block */
228#define ML_SIMPLE(x) (x & 0x10) /* DEL, INS or FIND */
229
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200230/* argument for ml_upd_block0() */
231typedef enum {
232 UB_FNAME = 0 /* update timestamp and filename */
233 , UB_SAME_DIR /* update the B0_SAME_DIR flag */
234 , UB_CRYPT /* update crypt key */
235} upd_block0_T;
236
237#ifdef FEAT_CRYPT
Bram Moolenaar92b8b2d2016-01-29 22:36:45 +0100238static void ml_set_mfp_crypt(buf_T *buf);
239static void ml_set_b0_crypt(buf_T *buf, ZERO_BL *b0p);
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200240#endif
Bram Moolenaar92b8b2d2016-01-29 22:36:45 +0100241static int ml_check_b0_id(ZERO_BL *b0p);
242static void ml_upd_block0(buf_T *buf, upd_block0_T what);
243static void set_b0_fname(ZERO_BL *, buf_T *buf);
244static void set_b0_dir_flag(ZERO_BL *b0p, buf_T *buf);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +0000245#ifdef FEAT_MBYTE
Bram Moolenaar92b8b2d2016-01-29 22:36:45 +0100246static void add_b0_fenc(ZERO_BL *b0p, buf_T *buf);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +0000247#endif
Bram Moolenaar92b8b2d2016-01-29 22:36:45 +0100248static time_t swapfile_info(char_u *);
249static int recov_file_names(char_u **, char_u *, int prepend_dot);
250static int ml_append_int(buf_T *, linenr_T, char_u *, colnr_T, int, int);
251static int ml_delete_int(buf_T *, linenr_T, int);
252static char_u *findswapname(buf_T *, char_u **, char_u *);
253static void ml_flush_line(buf_T *);
254static bhdr_T *ml_new_data(memfile_T *, int, int);
255static bhdr_T *ml_new_ptr(memfile_T *);
256static bhdr_T *ml_find_line(buf_T *, linenr_T, int);
257static int ml_add_stack(buf_T *);
258static void ml_lineadd(buf_T *, int);
259static int b0_magic_wrong(ZERO_BL *);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000260#ifdef CHECK_INODE
Bram Moolenaar92b8b2d2016-01-29 22:36:45 +0100261static int fnamecmp_ino(char_u *, char_u *, long);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000262#endif
Bram Moolenaar92b8b2d2016-01-29 22:36:45 +0100263static void long_to_char(long, char_u *);
264static long char_to_long(char_u *);
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200265#ifdef FEAT_CRYPT
Bram Moolenaar8767f522016-07-01 17:17:39 +0200266static cryptstate_T *ml_crypt_prepare(memfile_T *mfp, off_T offset, int reading);
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200267#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000268#ifdef FEAT_BYTEOFF
Bram Moolenaar92b8b2d2016-01-29 22:36:45 +0100269static void ml_updatechunk(buf_T *buf, long line, long len, int updtype);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000270#endif
271
272/*
Bram Moolenaar4770d092006-01-12 23:22:24 +0000273 * Open a new memline for "buf".
Bram Moolenaar071d4272004-06-13 20:20:40 +0000274 *
Bram Moolenaar4770d092006-01-12 23:22:24 +0000275 * Return FAIL for failure, OK otherwise.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000276 */
277 int
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100278ml_open(buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000279{
280 memfile_T *mfp;
281 bhdr_T *hp = NULL;
282 ZERO_BL *b0p;
283 PTR_BL *pp;
284 DATA_BL *dp;
285
Bram Moolenaar4770d092006-01-12 23:22:24 +0000286 /*
287 * init fields in memline struct
288 */
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200289 buf->b_ml.ml_stack_size = 0; /* no stack yet */
Bram Moolenaar4770d092006-01-12 23:22:24 +0000290 buf->b_ml.ml_stack = NULL; /* no stack yet */
291 buf->b_ml.ml_stack_top = 0; /* nothing in the stack */
292 buf->b_ml.ml_locked = NULL; /* no cached block */
293 buf->b_ml.ml_line_lnum = 0; /* no cached line */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000294#ifdef FEAT_BYTEOFF
Bram Moolenaar4770d092006-01-12 23:22:24 +0000295 buf->b_ml.ml_chunksize = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000296#endif
297
Bram Moolenaar5803ae62014-03-23 16:04:02 +0100298 if (cmdmod.noswapfile)
299 buf->b_p_swf = FALSE;
300
Bram Moolenaar4770d092006-01-12 23:22:24 +0000301 /*
302 * When 'updatecount' is non-zero swap file may be opened later.
303 */
304 if (p_uc && buf->b_p_swf)
305 buf->b_may_swap = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000306 else
Bram Moolenaar4770d092006-01-12 23:22:24 +0000307 buf->b_may_swap = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000308
Bram Moolenaar4770d092006-01-12 23:22:24 +0000309 /*
310 * Open the memfile. No swap file is created yet.
311 */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000312 mfp = mf_open(NULL, 0);
313 if (mfp == NULL)
314 goto error;
315
Bram Moolenaar4770d092006-01-12 23:22:24 +0000316 buf->b_ml.ml_mfp = mfp;
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200317#ifdef FEAT_CRYPT
318 mfp->mf_buffer = buf;
319#endif
Bram Moolenaar4770d092006-01-12 23:22:24 +0000320 buf->b_ml.ml_flags = ML_EMPTY;
321 buf->b_ml.ml_line_count = 1;
Bram Moolenaar592e0a22004-07-03 16:05:59 +0000322#ifdef FEAT_LINEBREAK
323 curwin->w_nrwidth_line_count = 0;
324#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000325
Bram Moolenaar071d4272004-06-13 20:20:40 +0000326/*
327 * fill block0 struct and write page 0
328 */
329 if ((hp = mf_new(mfp, FALSE, 1)) == NULL)
330 goto error;
331 if (hp->bh_bnum != 0)
332 {
Bram Moolenaar95f09602016-11-10 20:01:45 +0100333 IEMSG(_("E298: Didn't get block nr 0?"));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000334 goto error;
335 }
336 b0p = (ZERO_BL *)(hp->bh_data);
337
338 b0p->b0_id[0] = BLOCK0_ID0;
339 b0p->b0_id[1] = BLOCK0_ID1;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000340 b0p->b0_magic_long = (long)B0_MAGIC_LONG;
341 b0p->b0_magic_int = (int)B0_MAGIC_INT;
342 b0p->b0_magic_short = (short)B0_MAGIC_SHORT;
343 b0p->b0_magic_char = B0_MAGIC_CHAR;
Bram Moolenaar22c10562018-05-26 17:35:27 +0200344 mch_memmove(b0p->b0_version, "VIM ", 4);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000345 STRNCPY(b0p->b0_version + 4, Version, 6);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000346 long_to_char((long)mfp->mf_page_size, b0p->b0_page_size);
Bram Moolenaar4770d092006-01-12 23:22:24 +0000347
Bram Moolenaar76b92b22006-03-24 22:46:53 +0000348#ifdef FEAT_SPELL
349 if (!buf->b_spell)
350#endif
Bram Moolenaar4770d092006-01-12 23:22:24 +0000351 {
352 b0p->b0_dirty = buf->b_changed ? B0_DIRTY : 0;
353 b0p->b0_flags = get_fileformat(buf) + 1;
354 set_b0_fname(b0p, buf);
355 (void)get_user_name(b0p->b0_uname, B0_UNAME_SIZE);
356 b0p->b0_uname[B0_UNAME_SIZE - 1] = NUL;
357 mch_get_host_name(b0p->b0_hname, B0_HNAME_SIZE);
358 b0p->b0_hname[B0_HNAME_SIZE - 1] = NUL;
359 long_to_char(mch_get_pid(), b0p->b0_pid);
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200360#ifdef FEAT_CRYPT
Bram Moolenaar8f4ac012014-08-10 13:38:34 +0200361 ml_set_b0_crypt(buf, b0p);
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200362#endif
Bram Moolenaar4770d092006-01-12 23:22:24 +0000363 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000364
365 /*
366 * Always sync block number 0 to disk, so we can check the file name in
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200367 * the swap file in findswapname(). Don't do this for a help files or
368 * a spell buffer though.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000369 * Only works when there's a swapfile, otherwise it's done when the file
370 * is created.
371 */
372 mf_put(mfp, hp, TRUE, FALSE);
Bram Moolenaar4770d092006-01-12 23:22:24 +0000373 if (!buf->b_help && !B_SPELL(buf))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000374 (void)mf_sync(mfp, 0);
375
Bram Moolenaar4770d092006-01-12 23:22:24 +0000376 /*
377 * Fill in root pointer block and write page 1.
378 */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000379 if ((hp = ml_new_ptr(mfp)) == NULL)
380 goto error;
381 if (hp->bh_bnum != 1)
382 {
Bram Moolenaar95f09602016-11-10 20:01:45 +0100383 IEMSG(_("E298: Didn't get block nr 1?"));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000384 goto error;
385 }
386 pp = (PTR_BL *)(hp->bh_data);
387 pp->pb_count = 1;
388 pp->pb_pointer[0].pe_bnum = 2;
389 pp->pb_pointer[0].pe_page_count = 1;
390 pp->pb_pointer[0].pe_old_lnum = 1;
391 pp->pb_pointer[0].pe_line_count = 1; /* line count after insertion */
392 mf_put(mfp, hp, TRUE, FALSE);
393
Bram Moolenaar4770d092006-01-12 23:22:24 +0000394 /*
395 * Allocate first data block and create an empty line 1.
396 */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000397 if ((hp = ml_new_data(mfp, FALSE, 1)) == NULL)
398 goto error;
399 if (hp->bh_bnum != 2)
400 {
Bram Moolenaar95f09602016-11-10 20:01:45 +0100401 IEMSG(_("E298: Didn't get block nr 2?"));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000402 goto error;
403 }
404
405 dp = (DATA_BL *)(hp->bh_data);
406 dp->db_index[0] = --dp->db_txt_start; /* at end of block */
407 dp->db_free -= 1 + INDEX_SIZE;
408 dp->db_line_count = 1;
Bram Moolenaarf05da212009-11-17 16:13:15 +0000409 *((char_u *)dp + dp->db_txt_start) = NUL; /* empty line */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000410
411 return OK;
412
413error:
414 if (mfp != NULL)
415 {
416 if (hp)
417 mf_put(mfp, hp, FALSE, FALSE);
418 mf_close(mfp, TRUE); /* will also free(mfp->mf_fname) */
419 }
Bram Moolenaar4770d092006-01-12 23:22:24 +0000420 buf->b_ml.ml_mfp = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000421 return FAIL;
422}
423
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200424#if defined(FEAT_CRYPT) || defined(PROTO)
425/*
Bram Moolenaar2be79502014-08-13 21:58:28 +0200426 * Prepare encryption for "buf" for the current key and method.
427 */
428 static void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100429ml_set_mfp_crypt(buf_T *buf)
Bram Moolenaar2be79502014-08-13 21:58:28 +0200430{
431 if (*buf->b_p_key != NUL)
432 {
433 int method_nr = crypt_get_method_nr(buf);
434
435 if (method_nr > CRYPT_M_ZIP)
436 {
437 /* Generate a seed and store it in the memfile. */
438 sha2_seed(buf->b_ml.ml_mfp->mf_seed, MF_SEED_LEN, NULL, 0);
439 }
440 }
441}
442
443/*
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200444 * Prepare encryption for "buf" with block 0 "b0p".
445 */
446 static void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100447ml_set_b0_crypt(buf_T *buf, ZERO_BL *b0p)
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200448{
449 if (*buf->b_p_key == NUL)
450 b0p->b0_id[1] = BLOCK0_ID1;
451 else
452 {
Bram Moolenaar8f4ac012014-08-10 13:38:34 +0200453 int method_nr = crypt_get_method_nr(buf);
454
455 b0p->b0_id[1] = id1_codes[method_nr];
456 if (method_nr > CRYPT_M_ZIP)
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200457 {
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200458 /* Generate a seed and store it in block 0 and in the memfile. */
459 sha2_seed(&b0p->b0_seed, MF_SEED_LEN, NULL, 0);
460 mch_memmove(buf->b_ml.ml_mfp->mf_seed, &b0p->b0_seed, MF_SEED_LEN);
461 }
462 }
463}
464
465/*
466 * Called after the crypt key or 'cryptmethod' was changed for "buf".
467 * Will apply this to the swapfile.
468 * "old_key" is the previous key. It is equal to buf->b_p_key when
469 * 'cryptmethod' is changed.
Bram Moolenaar49771f42010-07-20 17:32:38 +0200470 * "old_cm" is the previous 'cryptmethod'. It is equal to the current
471 * 'cryptmethod' when 'key' is changed.
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200472 */
473 void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100474ml_set_crypt_key(
475 buf_T *buf,
476 char_u *old_key,
477 char_u *old_cm)
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200478{
479 memfile_T *mfp = buf->b_ml.ml_mfp;
480 bhdr_T *hp;
481 int page_count;
482 int idx;
483 long error;
484 infoptr_T *ip;
485 PTR_BL *pp;
486 DATA_BL *dp;
487 blocknr_T bnum;
488 int top;
Bram Moolenaarbc563362015-06-09 18:35:25 +0200489 int old_method;
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200490
Bram Moolenaar3832c462010-08-04 15:32:46 +0200491 if (mfp == NULL)
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200492 return; /* no memfile yet, nothing to do */
Bram Moolenaarbc563362015-06-09 18:35:25 +0200493 old_method = crypt_method_nr_from_name(old_cm);
494
495 /* First make sure the swapfile is in a consistent state, using the old
496 * key and method. */
497 {
498 char_u *new_key = buf->b_p_key;
499 char_u *new_buf_cm = buf->b_p_cm;
500
501 buf->b_p_key = old_key;
502 buf->b_p_cm = old_cm;
503 ml_preserve(buf, FALSE);
504 buf->b_p_key = new_key;
505 buf->b_p_cm = new_buf_cm;
506 }
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200507
508 /* Set the key, method and seed to be used for reading, these must be the
509 * old values. */
510 mfp->mf_old_key = old_key;
Bram Moolenaarbc563362015-06-09 18:35:25 +0200511 mfp->mf_old_cm = old_method;
512 if (old_method > 0 && *old_key != NUL)
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200513 mch_memmove(mfp->mf_old_seed, mfp->mf_seed, MF_SEED_LEN);
514
515 /* Update block 0 with the crypt flag and may set a new seed. */
516 ml_upd_block0(buf, UB_CRYPT);
517
518 if (mfp->mf_infile_count > 2)
519 {
520 /*
521 * Need to read back all data blocks from disk, decrypt them with the
522 * old key/method and mark them to be written. The algorithm is
523 * similar to what happens in ml_recover(), but we skip negative block
524 * numbers.
525 */
526 ml_flush_line(buf); /* flush buffered line */
527 (void)ml_find_line(buf, (linenr_T)0, ML_FLUSH); /* flush locked block */
528
529 hp = NULL;
530 bnum = 1; /* start with block 1 */
531 page_count = 1; /* which is 1 page */
532 idx = 0; /* start with first index in block 1 */
533 error = 0;
534 buf->b_ml.ml_stack_top = 0;
Bram Moolenaard23a8232018-02-10 18:45:26 +0100535 VIM_CLEAR(buf->b_ml.ml_stack);
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200536 buf->b_ml.ml_stack_size = 0; /* no stack yet */
537
538 for ( ; !got_int; line_breakcheck())
539 {
540 if (hp != NULL)
541 mf_put(mfp, hp, FALSE, FALSE); /* release previous block */
542
543 /* get the block (pointer or data) */
544 if ((hp = mf_get(mfp, (blocknr_T)bnum, page_count)) == NULL)
545 {
546 if (bnum == 1)
547 break;
548 ++error;
549 }
550 else
551 {
552 pp = (PTR_BL *)(hp->bh_data);
553 if (pp->pb_id == PTR_ID) /* it is a pointer block */
554 {
555 if (pp->pb_count == 0)
556 {
557 /* empty block? */
558 ++error;
559 }
560 else if (idx < (int)pp->pb_count) /* go a block deeper */
561 {
562 if (pp->pb_pointer[idx].pe_bnum < 0)
563 {
Bram Moolenaarbc563362015-06-09 18:35:25 +0200564 /* Skip data block with negative block number.
565 * Should not happen, because of the ml_preserve()
566 * above. Get same block again for next index. */
567 ++idx;
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200568 continue;
569 }
570
571 /* going one block deeper in the tree, new entry in
572 * stack */
573 if ((top = ml_add_stack(buf)) < 0)
574 {
575 ++error;
576 break; /* out of memory */
577 }
578 ip = &(buf->b_ml.ml_stack[top]);
579 ip->ip_bnum = bnum;
580 ip->ip_index = idx;
581
582 bnum = pp->pb_pointer[idx].pe_bnum;
583 page_count = pp->pb_pointer[idx].pe_page_count;
Bram Moolenaarbc563362015-06-09 18:35:25 +0200584 idx = 0;
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200585 continue;
586 }
587 }
588 else /* not a pointer block */
589 {
590 dp = (DATA_BL *)(hp->bh_data);
591 if (dp->db_id != DATA_ID) /* block id wrong */
592 ++error;
593 else
594 {
595 /* It is a data block, need to write it back to disk. */
596 mf_put(mfp, hp, TRUE, FALSE);
597 hp = NULL;
598 }
599 }
600 }
601
602 if (buf->b_ml.ml_stack_top == 0) /* finished */
603 break;
604
605 /* go one block up in the tree */
606 ip = &(buf->b_ml.ml_stack[--(buf->b_ml.ml_stack_top)]);
607 bnum = ip->ip_bnum;
608 idx = ip->ip_index + 1; /* go to next index */
609 page_count = 1;
610 }
Bram Moolenaarbc563362015-06-09 18:35:25 +0200611 if (hp != NULL)
612 mf_put(mfp, hp, FALSE, FALSE); /* release previous block */
Bram Moolenaar2e2e13c2010-12-08 13:17:03 +0100613
614 if (error > 0)
615 EMSG(_("E843: Error while updating swap file crypt"));
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200616 }
617
618 mfp->mf_old_key = NULL;
619}
620#endif
621
Bram Moolenaar071d4272004-06-13 20:20:40 +0000622/*
623 * ml_setname() is called when the file name of "buf" has been changed.
624 * It may rename the swap file.
625 */
626 void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100627ml_setname(buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000628{
629 int success = FALSE;
630 memfile_T *mfp;
631 char_u *fname;
632 char_u *dirp;
Bram Moolenaar48e330a2016-02-23 14:53:34 +0100633#if defined(MSWIN)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000634 char_u *p;
635#endif
636
637 mfp = buf->b_ml.ml_mfp;
638 if (mfp->mf_fd < 0) /* there is no swap file yet */
639 {
640 /*
641 * When 'updatecount' is 0 and 'noswapfile' there is no swap file.
642 * For help files we will make a swap file now.
643 */
Bram Moolenaar5803ae62014-03-23 16:04:02 +0100644 if (p_uc != 0 && !cmdmod.noswapfile)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000645 ml_open_file(buf); /* create a swap file */
646 return;
647 }
648
649 /*
650 * Try all directories in the 'directory' option.
651 */
652 dirp = p_dir;
653 for (;;)
654 {
655 if (*dirp == NUL) /* tried all directories, fail */
656 break;
Bram Moolenaar8fc061c2004-12-29 21:03:02 +0000657 fname = findswapname(buf, &dirp, mfp->mf_fname);
658 /* alloc's fname */
Bram Moolenaarf541c362011-10-26 11:44:18 +0200659 if (dirp == NULL) /* out of memory */
660 break;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000661 if (fname == NULL) /* no file name found for this dir */
662 continue;
663
Bram Moolenaar48e330a2016-02-23 14:53:34 +0100664#if defined(MSWIN)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000665 /*
666 * Set full pathname for swap file now, because a ":!cd dir" may
667 * change directory without us knowing it.
668 */
669 p = FullName_save(fname, FALSE);
670 vim_free(fname);
671 fname = p;
672 if (fname == NULL)
673 continue;
674#endif
675 /* if the file name is the same we don't have to do anything */
676 if (fnamecmp(fname, mfp->mf_fname) == 0)
677 {
678 vim_free(fname);
679 success = TRUE;
680 break;
681 }
682 /* need to close the swap file before renaming */
683 if (mfp->mf_fd >= 0)
684 {
685 close(mfp->mf_fd);
686 mfp->mf_fd = -1;
687 }
688
689 /* try to rename the swap file */
690 if (vim_rename(mfp->mf_fname, fname) == 0)
691 {
692 success = TRUE;
693 vim_free(mfp->mf_fname);
694 mfp->mf_fname = fname;
695 vim_free(mfp->mf_ffname);
Bram Moolenaar48e330a2016-02-23 14:53:34 +0100696#if defined(MSWIN)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000697 mfp->mf_ffname = NULL; /* mf_fname is full pathname already */
698#else
699 mf_set_ffname(mfp);
700#endif
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200701 ml_upd_block0(buf, UB_SAME_DIR);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000702 break;
703 }
704 vim_free(fname); /* this fname didn't work, try another */
705 }
706
707 if (mfp->mf_fd == -1) /* need to (re)open the swap file */
708 {
709 mfp->mf_fd = mch_open((char *)mfp->mf_fname, O_RDWR | O_EXTRA, 0);
710 if (mfp->mf_fd < 0)
711 {
712 /* could not (re)open the swap file, what can we do???? */
713 EMSG(_("E301: Oops, lost the swap file!!!"));
714 return;
715 }
Bram Moolenaarf05da212009-11-17 16:13:15 +0000716#ifdef HAVE_FD_CLOEXEC
717 {
718 int fdflags = fcntl(mfp->mf_fd, F_GETFD);
719 if (fdflags >= 0 && (fdflags & FD_CLOEXEC) == 0)
Bram Moolenaarfbc4b4d2016-02-07 15:14:01 +0100720 (void)fcntl(mfp->mf_fd, F_SETFD, fdflags | FD_CLOEXEC);
Bram Moolenaarf05da212009-11-17 16:13:15 +0000721 }
722#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000723 }
724 if (!success)
725 EMSG(_("E302: Could not rename swap file"));
726}
727
728/*
729 * Open a file for the memfile for all buffers that are not readonly or have
730 * been modified.
731 * Used when 'updatecount' changes from zero to non-zero.
732 */
733 void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100734ml_open_files(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000735{
736 buf_T *buf;
737
Bram Moolenaar29323592016-07-24 22:04:11 +0200738 FOR_ALL_BUFFERS(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000739 if (!buf->b_p_ro || buf->b_changed)
740 ml_open_file(buf);
741}
742
743/*
744 * Open a swap file for an existing memfile, if there is no swap file yet.
745 * If we are unable to find a file name, mf_fname will be NULL
746 * and the memfile will be in memory only (no recovery possible).
747 */
748 void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100749ml_open_file(buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000750{
751 memfile_T *mfp;
752 char_u *fname;
753 char_u *dirp;
754
755 mfp = buf->b_ml.ml_mfp;
Bram Moolenaar5803ae62014-03-23 16:04:02 +0100756 if (mfp == NULL || mfp->mf_fd >= 0 || !buf->b_p_swf || cmdmod.noswapfile)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000757 return; /* nothing to do */
758
Bram Moolenaara1956f62006-03-12 22:18:00 +0000759#ifdef FEAT_SPELL
Bram Moolenaar4770d092006-01-12 23:22:24 +0000760 /* For a spell buffer use a temp file name. */
761 if (buf->b_spell)
762 {
Bram Moolenaare5c421c2015-03-31 13:33:08 +0200763 fname = vim_tempname('s', FALSE);
Bram Moolenaar4770d092006-01-12 23:22:24 +0000764 if (fname != NULL)
765 (void)mf_open_file(mfp, fname); /* consumes fname! */
766 buf->b_may_swap = FALSE;
767 return;
768 }
769#endif
770
Bram Moolenaar071d4272004-06-13 20:20:40 +0000771 /*
772 * Try all directories in 'directory' option.
773 */
774 dirp = p_dir;
775 for (;;)
776 {
777 if (*dirp == NUL)
778 break;
Bram Moolenaare242b832010-06-24 05:39:03 +0200779 /* There is a small chance that between choosing the swap file name
780 * and creating it, another Vim creates the file. In that case the
Bram Moolenaar071d4272004-06-13 20:20:40 +0000781 * creation will fail and we will use another directory. */
Bram Moolenaar8fc061c2004-12-29 21:03:02 +0000782 fname = findswapname(buf, &dirp, NULL); /* allocates fname */
Bram Moolenaarf541c362011-10-26 11:44:18 +0200783 if (dirp == NULL)
784 break; /* out of memory */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000785 if (fname == NULL)
786 continue;
787 if (mf_open_file(mfp, fname) == OK) /* consumes fname! */
788 {
Bram Moolenaar48e330a2016-02-23 14:53:34 +0100789#if defined(MSWIN)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000790 /*
791 * set full pathname for swap file now, because a ":!cd dir" may
792 * change directory without us knowing it.
793 */
794 mf_fullname(mfp);
795#endif
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200796 ml_upd_block0(buf, UB_SAME_DIR);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +0000797
Bram Moolenaar071d4272004-06-13 20:20:40 +0000798 /* Flush block zero, so others can read it */
799 if (mf_sync(mfp, MFS_ZERO) == OK)
Bram Moolenaarc32840f2006-01-14 21:23:38 +0000800 {
801 /* Mark all blocks that should be in the swapfile as dirty.
802 * Needed for when the 'swapfile' option was reset, so that
803 * the swap file was deleted, and then on again. */
804 mf_set_dirty(mfp);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000805 break;
Bram Moolenaarc32840f2006-01-14 21:23:38 +0000806 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000807 /* Writing block 0 failed: close the file and try another dir */
808 mf_close_file(buf, FALSE);
809 }
810 }
811
812 if (mfp->mf_fname == NULL) /* Failed! */
813 {
814 need_wait_return = TRUE; /* call wait_return later */
815 ++no_wait_return;
816 (void)EMSG2(_("E303: Unable to open swap file for \"%s\", recovery impossible"),
Bram Moolenaare1704ba2012-10-03 18:25:00 +0200817 buf_spname(buf) != NULL ? buf_spname(buf) : buf->b_fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000818 --no_wait_return;
819 }
820
821 /* don't try to open a swap file again */
822 buf->b_may_swap = FALSE;
823}
824
825/*
826 * If still need to create a swap file, and starting to edit a not-readonly
827 * file, or reading into an existing buffer, create a swap file now.
828 */
829 void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100830check_need_swap(
Bram Moolenaar2f0f8712018-08-21 18:50:18 +0200831 int newfile) // reading file into new buffer
Bram Moolenaar071d4272004-06-13 20:20:40 +0000832{
Bram Moolenaar2f0f8712018-08-21 18:50:18 +0200833 int old_msg_silent = msg_silent; // might be reset by an E325 message
834
Bram Moolenaar071d4272004-06-13 20:20:40 +0000835 if (curbuf->b_may_swap && (!curbuf->b_p_ro || !newfile))
836 ml_open_file(curbuf);
Bram Moolenaar2f0f8712018-08-21 18:50:18 +0200837 msg_silent = old_msg_silent;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000838}
839
840/*
841 * Close memline for buffer 'buf'.
842 * If 'del_file' is TRUE, delete the swap file
843 */
844 void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100845ml_close(buf_T *buf, int del_file)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000846{
847 if (buf->b_ml.ml_mfp == NULL) /* not open */
848 return;
849 mf_close(buf->b_ml.ml_mfp, del_file); /* close the .swp file */
850 if (buf->b_ml.ml_line_lnum != 0 && (buf->b_ml.ml_flags & ML_LINE_DIRTY))
851 vim_free(buf->b_ml.ml_line_ptr);
852 vim_free(buf->b_ml.ml_stack);
853#ifdef FEAT_BYTEOFF
Bram Moolenaard23a8232018-02-10 18:45:26 +0100854 VIM_CLEAR(buf->b_ml.ml_chunksize);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000855#endif
856 buf->b_ml.ml_mfp = NULL;
857
858 /* Reset the "recovered" flag, give the ATTENTION prompt the next time
859 * this buffer is loaded. */
860 buf->b_flags &= ~BF_RECOVERED;
861}
862
863/*
864 * Close all existing memlines and memfiles.
865 * Only used when exiting.
866 * When 'del_file' is TRUE, delete the memfiles.
Bram Moolenaar81bf7082005-02-12 14:31:42 +0000867 * But don't delete files that were ":preserve"d when we are POSIX compatible.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000868 */
869 void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100870ml_close_all(int del_file)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000871{
872 buf_T *buf;
873
Bram Moolenaar29323592016-07-24 22:04:11 +0200874 FOR_ALL_BUFFERS(buf)
Bram Moolenaar81bf7082005-02-12 14:31:42 +0000875 ml_close(buf, del_file && ((buf->b_flags & BF_PRESERVED) == 0
876 || vim_strchr(p_cpo, CPO_PRESERVE) == NULL));
Bram Moolenaar34b466e2013-11-28 17:41:46 +0100877#ifdef FEAT_SPELL
878 spell_delete_wordlist(); /* delete the internal wordlist */
879#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000880#ifdef TEMPDIRNAMES
Bram Moolenaar34b466e2013-11-28 17:41:46 +0100881 vim_deltempdir(); /* delete created temp directory */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000882#endif
883}
884
885/*
886 * Close all memfiles for not modified buffers.
887 * Only use just before exiting!
888 */
889 void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100890ml_close_notmod(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000891{
892 buf_T *buf;
893
Bram Moolenaar29323592016-07-24 22:04:11 +0200894 FOR_ALL_BUFFERS(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000895 if (!bufIsChanged(buf))
896 ml_close(buf, TRUE); /* close all not-modified buffers */
897}
898
899/*
900 * Update the timestamp in the .swp file.
901 * Used when the file has been written.
902 */
903 void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100904ml_timestamp(buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000905{
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200906 ml_upd_block0(buf, UB_FNAME);
907}
908
909/*
910 * Return FAIL when the ID of "b0p" is wrong.
911 */
912 static int
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100913ml_check_b0_id(ZERO_BL *b0p)
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200914{
915 if (b0p->b0_id[0] != BLOCK0_ID0
916 || (b0p->b0_id[1] != BLOCK0_ID1
917 && b0p->b0_id[1] != BLOCK0_ID1_C0
Bram Moolenaar8f4ac012014-08-10 13:38:34 +0200918 && b0p->b0_id[1] != BLOCK0_ID1_C1
919 && b0p->b0_id[1] != BLOCK0_ID1_C2)
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200920 )
921 return FAIL;
922 return OK;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +0000923}
924
925/*
926 * Update the timestamp or the B0_SAME_DIR flag of the .swp file.
927 */
928 static void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100929ml_upd_block0(buf_T *buf, upd_block0_T what)
Bram Moolenaar1cd871b2004-12-19 22:46:22 +0000930{
Bram Moolenaar071d4272004-06-13 20:20:40 +0000931 memfile_T *mfp;
932 bhdr_T *hp;
933 ZERO_BL *b0p;
934
935 mfp = buf->b_ml.ml_mfp;
Bram Moolenaar2be79502014-08-13 21:58:28 +0200936 if (mfp == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000937 return;
Bram Moolenaar2be79502014-08-13 21:58:28 +0200938 hp = mf_get(mfp, (blocknr_T)0, 1);
939 if (hp == NULL)
940 {
941#ifdef FEAT_CRYPT
942 /* Possibly update the seed in the memfile before there is a block0. */
943 if (what == UB_CRYPT)
944 ml_set_mfp_crypt(buf);
945#endif
946 return;
947 }
948
Bram Moolenaar071d4272004-06-13 20:20:40 +0000949 b0p = (ZERO_BL *)(hp->bh_data);
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200950 if (ml_check_b0_id(b0p) == FAIL)
Bram Moolenaar95f09602016-11-10 20:01:45 +0100951 IEMSG(_("E304: ml_upd_block0(): Didn't get block 0??"));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000952 else
Bram Moolenaar1cd871b2004-12-19 22:46:22 +0000953 {
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200954 if (what == UB_FNAME)
Bram Moolenaar1cd871b2004-12-19 22:46:22 +0000955 set_b0_fname(b0p, buf);
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200956#ifdef FEAT_CRYPT
957 else if (what == UB_CRYPT)
958 ml_set_b0_crypt(buf, b0p);
959#endif
960 else /* what == UB_SAME_DIR */
Bram Moolenaar1cd871b2004-12-19 22:46:22 +0000961 set_b0_dir_flag(b0p, buf);
962 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000963 mf_put(mfp, hp, TRUE, FALSE);
964}
965
966/*
967 * Write file name and timestamp into block 0 of a swap file.
968 * Also set buf->b_mtime.
969 * Don't use NameBuff[]!!!
970 */
971 static void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100972set_b0_fname(ZERO_BL *b0p, buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000973{
Bram Moolenaar8767f522016-07-01 17:17:39 +0200974 stat_T st;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000975
976 if (buf->b_ffname == NULL)
977 b0p->b0_fname[0] = NUL;
978 else
979 {
Bram Moolenaar48e330a2016-02-23 14:53:34 +0100980#if defined(MSWIN) || defined(AMIGA)
Bram Moolenaar1cd871b2004-12-19 22:46:22 +0000981 /* Systems that cannot translate "~user" back into a path: copy the
982 * file name unmodified. Do use slashes instead of backslashes for
983 * portability. */
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200984 vim_strncpy(b0p->b0_fname, buf->b_ffname, B0_FNAME_SIZE_CRYPT - 1);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +0000985# ifdef BACKSLASH_IN_FILENAME
986 forward_slash(b0p->b0_fname);
987# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000988#else
989 size_t flen, ulen;
990 char_u uname[B0_UNAME_SIZE];
991
992 /*
993 * For a file under the home directory of the current user, we try to
994 * replace the home directory path with "~user". This helps when
995 * editing the same file on different machines over a network.
996 * First replace home dir path with "~/" with home_replace().
997 * Then insert the user name to get "~user/".
998 */
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200999 home_replace(NULL, buf->b_ffname, b0p->b0_fname,
1000 B0_FNAME_SIZE_CRYPT, TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001001 if (b0p->b0_fname[0] == '~')
1002 {
1003 flen = STRLEN(b0p->b0_fname);
1004 /* If there is no user name or it is too long, don't use "~/" */
1005 if (get_user_name(uname, B0_UNAME_SIZE) == FAIL
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001006 || (ulen = STRLEN(uname)) + flen > B0_FNAME_SIZE_CRYPT - 1)
1007 vim_strncpy(b0p->b0_fname, buf->b_ffname,
1008 B0_FNAME_SIZE_CRYPT - 1);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001009 else
1010 {
1011 mch_memmove(b0p->b0_fname + ulen + 1, b0p->b0_fname + 1, flen);
1012 mch_memmove(b0p->b0_fname + 1, uname, ulen);
1013 }
1014 }
1015#endif
1016 if (mch_stat((char *)buf->b_ffname, &st) >= 0)
1017 {
1018 long_to_char((long)st.st_mtime, b0p->b0_mtime);
1019#ifdef CHECK_INODE
1020 long_to_char((long)st.st_ino, b0p->b0_ino);
1021#endif
1022 buf_store_time(buf, &st, buf->b_ffname);
1023 buf->b_mtime_read = buf->b_mtime;
1024 }
1025 else
1026 {
1027 long_to_char(0L, b0p->b0_mtime);
1028#ifdef CHECK_INODE
1029 long_to_char(0L, b0p->b0_ino);
1030#endif
1031 buf->b_mtime = 0;
1032 buf->b_mtime_read = 0;
1033 buf->b_orig_size = 0;
1034 buf->b_orig_mode = 0;
1035 }
1036 }
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001037
1038#ifdef FEAT_MBYTE
1039 /* Also add the 'fileencoding' if there is room. */
1040 add_b0_fenc(b0p, curbuf);
1041#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001042}
1043
1044/*
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001045 * Update the B0_SAME_DIR flag of the swap file. It's set if the file and the
1046 * swapfile for "buf" are in the same directory.
1047 * This is fail safe: if we are not sure the directories are equal the flag is
1048 * not set.
1049 */
1050 static void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01001051set_b0_dir_flag(ZERO_BL *b0p, buf_T *buf)
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001052{
1053 if (same_directory(buf->b_ml.ml_mfp->mf_fname, buf->b_ffname))
1054 b0p->b0_flags |= B0_SAME_DIR;
1055 else
1056 b0p->b0_flags &= ~B0_SAME_DIR;
1057}
1058
1059#ifdef FEAT_MBYTE
1060/*
1061 * When there is room, add the 'fileencoding' to block zero.
1062 */
1063 static void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01001064add_b0_fenc(
1065 ZERO_BL *b0p,
1066 buf_T *buf)
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001067{
1068 int n;
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001069 int size = B0_FNAME_SIZE_NOCRYPT;
1070
1071# ifdef FEAT_CRYPT
1072 /* Without encryption use the same offset as in Vim 7.2 to be compatible.
1073 * With encryption it's OK to move elsewhere, the swap file is not
1074 * compatible anyway. */
1075 if (*buf->b_p_key != NUL)
1076 size = B0_FNAME_SIZE_CRYPT;
1077# endif
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001078
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00001079 n = (int)STRLEN(buf->b_p_fenc);
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001080 if ((int)STRLEN(b0p->b0_fname) + n + 1 > size)
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001081 b0p->b0_flags &= ~B0_HAS_FENC;
1082 else
1083 {
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001084 mch_memmove((char *)b0p->b0_fname + size - n,
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001085 (char *)buf->b_p_fenc, (size_t)n);
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001086 *(b0p->b0_fname + size - n - 1) = NUL;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001087 b0p->b0_flags |= B0_HAS_FENC;
1088 }
1089}
1090#endif
1091
1092
1093/*
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001094 * Try to recover curbuf from the .swp file.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001095 */
1096 void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01001097ml_recover(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001098{
1099 buf_T *buf = NULL;
1100 memfile_T *mfp = NULL;
1101 char_u *fname;
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001102 char_u *fname_used = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001103 bhdr_T *hp = NULL;
1104 ZERO_BL *b0p;
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001105 int b0_ff;
1106 char_u *b0_fenc = NULL;
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001107#ifdef FEAT_CRYPT
1108 int b0_cm = -1;
1109#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001110 PTR_BL *pp;
1111 DATA_BL *dp;
1112 infoptr_T *ip;
1113 blocknr_T bnum;
1114 int page_count;
Bram Moolenaar8767f522016-07-01 17:17:39 +02001115 stat_T org_stat, swp_stat;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001116 int len;
1117 int directly;
1118 linenr_T lnum;
1119 char_u *p;
1120 int i;
1121 long error;
1122 int cannot_open;
1123 linenr_T line_count;
1124 int has_error;
1125 int idx;
1126 int top;
1127 int txt_start;
Bram Moolenaar8767f522016-07-01 17:17:39 +02001128 off_T size;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001129 int called_from_main;
1130 int serious_error = TRUE;
1131 long mtime;
1132 int attr;
Bram Moolenaarfc2d5bd2010-05-15 17:06:53 +02001133 int orig_file_status = NOTDONE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001134
1135 recoverymode = TRUE;
1136 called_from_main = (curbuf->b_ml.ml_mfp == NULL);
Bram Moolenaar8820b482017-03-16 17:23:31 +01001137 attr = HL_ATTR(HLF_E);
Bram Moolenaard0ba34a2009-11-03 12:06:23 +00001138
1139 /*
Bram Moolenaaraf903e52017-12-02 15:11:22 +01001140 * If the file name ends in ".s[a-w][a-z]" we assume this is the swap file.
Bram Moolenaard0ba34a2009-11-03 12:06:23 +00001141 * Otherwise a search is done to find the swap file(s).
1142 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001143 fname = curbuf->b_fname;
1144 if (fname == NULL) /* When there is no file name */
1145 fname = (char_u *)"";
1146 len = (int)STRLEN(fname);
1147 if (len >= 4 &&
Bram Moolenaare60acc12011-05-10 16:41:25 +02001148#if defined(VMS)
Bram Moolenaar79518e22017-02-17 16:31:35 +01001149 STRNICMP(fname + len - 4, "_s", 2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001150#else
Bram Moolenaar79518e22017-02-17 16:31:35 +01001151 STRNICMP(fname + len - 4, ".s", 2)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001152#endif
Bram Moolenaar79518e22017-02-17 16:31:35 +01001153 == 0
Bram Moolenaaraf903e52017-12-02 15:11:22 +01001154 && vim_strchr((char_u *)"abcdefghijklmnopqrstuvw",
1155 TOLOWER_ASC(fname[len - 2])) != NULL
Bram Moolenaard0ba34a2009-11-03 12:06:23 +00001156 && ASCII_ISALPHA(fname[len - 1]))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001157 {
1158 directly = TRUE;
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001159 fname_used = vim_strsave(fname); /* make a copy for mf_open() */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001160 }
1161 else
1162 {
1163 directly = FALSE;
1164
1165 /* count the number of matching swap files */
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001166 len = recover_names(fname, FALSE, 0, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001167 if (len == 0) /* no swap files found */
1168 {
1169 EMSG2(_("E305: No swap file found for %s"), fname);
1170 goto theend;
1171 }
1172 if (len == 1) /* one swap file found, use it */
1173 i = 1;
1174 else /* several swap files found, choose */
1175 {
1176 /* list the names of the swap files */
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001177 (void)recover_names(fname, TRUE, 0, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001178 msg_putchar('\n');
1179 MSG_PUTS(_("Enter number of swap file to use (0 to quit): "));
Bram Moolenaar24bbcfe2005-06-28 23:32:02 +00001180 i = get_number(FALSE, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001181 if (i < 1 || i > len)
1182 goto theend;
1183 }
1184 /* get the swap file name that will be used */
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001185 (void)recover_names(fname, FALSE, i, &fname_used);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001186 }
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001187 if (fname_used == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001188 goto theend; /* out of memory */
1189
1190 /* When called from main() still need to initialize storage structure */
Bram Moolenaar4770d092006-01-12 23:22:24 +00001191 if (called_from_main && ml_open(curbuf) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001192 getout(1);
1193
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001194 /*
1195 * Allocate a buffer structure for the swap file that is used for recovery.
Bram Moolenaar0ad014c2010-07-25 14:00:46 +02001196 * Only the memline and crypt information in it are really used.
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001197 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001198 buf = (buf_T *)alloc((unsigned)sizeof(buf_T));
1199 if (buf == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001200 goto theend;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001201
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001202 /*
1203 * init fields in memline struct
1204 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001205 buf->b_ml.ml_stack_size = 0; /* no stack yet */
1206 buf->b_ml.ml_stack = NULL; /* no stack yet */
1207 buf->b_ml.ml_stack_top = 0; /* nothing in the stack */
1208 buf->b_ml.ml_line_lnum = 0; /* no cached line */
1209 buf->b_ml.ml_locked = NULL; /* no locked block */
1210 buf->b_ml.ml_flags = 0;
Bram Moolenaar0fe849a2010-07-25 15:11:11 +02001211#ifdef FEAT_CRYPT
1212 buf->b_p_key = empty_option;
1213 buf->b_p_cm = empty_option;
1214#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001215
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001216 /*
1217 * open the memfile from the old swap file
1218 */
1219 p = vim_strsave(fname_used); /* save "fname_used" for the message:
1220 mf_open() will consume "fname_used"! */
1221 mfp = mf_open(fname_used, O_RDONLY);
1222 fname_used = p;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001223 if (mfp == NULL || mfp->mf_fd < 0)
1224 {
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001225 if (fname_used != NULL)
1226 EMSG2(_("E306: Cannot open %s"), fname_used);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001227 goto theend;
1228 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001229 buf->b_ml.ml_mfp = mfp;
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001230#ifdef FEAT_CRYPT
1231 mfp->mf_buffer = buf;
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001232#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001233
1234 /*
1235 * The page size set in mf_open() might be different from the page size
1236 * used in the swap file, we must get it from block 0. But to read block
1237 * 0 we need a page size. Use the minimal size for block 0 here, it will
1238 * be set to the real value below.
1239 */
1240 mfp->mf_page_size = MIN_SWAP_PAGE_SIZE;
1241
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001242 /*
1243 * try to read block 0
1244 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001245 if ((hp = mf_get(mfp, (blocknr_T)0, 1)) == NULL)
1246 {
1247 msg_start();
1248 MSG_PUTS_ATTR(_("Unable to read block 0 from "), attr | MSG_HIST);
1249 msg_outtrans_attr(mfp->mf_fname, attr | MSG_HIST);
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001250 MSG_PUTS_ATTR(_("\nMaybe no changes were made or Vim did not update the swap file."),
Bram Moolenaar071d4272004-06-13 20:20:40 +00001251 attr | MSG_HIST);
1252 msg_end();
1253 goto theend;
1254 }
1255 b0p = (ZERO_BL *)(hp->bh_data);
1256 if (STRNCMP(b0p->b0_version, "VIM 3.0", 7) == 0)
1257 {
1258 msg_start();
1259 msg_outtrans_attr(mfp->mf_fname, MSG_HIST);
1260 MSG_PUTS_ATTR(_(" cannot be used with this version of Vim.\n"),
1261 MSG_HIST);
1262 MSG_PUTS_ATTR(_("Use Vim version 3.0.\n"), MSG_HIST);
1263 msg_end();
1264 goto theend;
1265 }
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001266 if (ml_check_b0_id(b0p) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001267 {
1268 EMSG2(_("E307: %s does not look like a Vim swap file"), mfp->mf_fname);
1269 goto theend;
1270 }
1271 if (b0_magic_wrong(b0p))
1272 {
1273 msg_start();
1274 msg_outtrans_attr(mfp->mf_fname, attr | MSG_HIST);
Bram Moolenaar48e330a2016-02-23 14:53:34 +01001275#if defined(MSWIN)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001276 if (STRNCMP(b0p->b0_hname, "PC ", 3) == 0)
1277 MSG_PUTS_ATTR(_(" cannot be used with this version of Vim.\n"),
1278 attr | MSG_HIST);
1279 else
1280#endif
1281 MSG_PUTS_ATTR(_(" cannot be used on this computer.\n"),
1282 attr | MSG_HIST);
1283 MSG_PUTS_ATTR(_("The file was created on "), attr | MSG_HIST);
Bram Moolenaare242b832010-06-24 05:39:03 +02001284 /* avoid going past the end of a corrupted hostname */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001285 b0p->b0_fname[0] = NUL;
1286 MSG_PUTS_ATTR(b0p->b0_hname, attr | MSG_HIST);
1287 MSG_PUTS_ATTR(_(",\nor the file has been damaged."), attr | MSG_HIST);
1288 msg_end();
1289 goto theend;
1290 }
Bram Moolenaar1c536282007-04-26 15:21:56 +00001291
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001292#ifdef FEAT_CRYPT
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02001293 for (i = 0; i < (int)(sizeof(id1_codes) / sizeof(int)); ++i)
1294 if (id1_codes[i] == b0p->b0_id[1])
1295 b0_cm = i;
1296 if (b0_cm > 0)
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001297 mch_memmove(mfp->mf_seed, &b0p->b0_seed, MF_SEED_LEN);
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02001298 crypt_set_cm_option(buf, b0_cm < 0 ? 0 : b0_cm);
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001299#else
1300 if (b0p->b0_id[1] != BLOCK0_ID1)
1301 {
Bram Moolenaar996343d2010-07-04 22:20:21 +02001302 EMSG2(_("E833: %s is encrypted and this version of Vim does not support encryption"), mfp->mf_fname);
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001303 goto theend;
1304 }
1305#endif
1306
Bram Moolenaar071d4272004-06-13 20:20:40 +00001307 /*
1308 * If we guessed the wrong page size, we have to recalculate the
1309 * highest block number in the file.
1310 */
1311 if (mfp->mf_page_size != (unsigned)char_to_long(b0p->b0_page_size))
1312 {
Bram Moolenaar1c536282007-04-26 15:21:56 +00001313 unsigned previous_page_size = mfp->mf_page_size;
1314
Bram Moolenaar071d4272004-06-13 20:20:40 +00001315 mf_new_page_size(mfp, (unsigned)char_to_long(b0p->b0_page_size));
Bram Moolenaar1c536282007-04-26 15:21:56 +00001316 if (mfp->mf_page_size < previous_page_size)
1317 {
1318 msg_start();
1319 msg_outtrans_attr(mfp->mf_fname, attr | MSG_HIST);
1320 MSG_PUTS_ATTR(_(" has been damaged (page size is smaller than minimum value).\n"),
1321 attr | MSG_HIST);
1322 msg_end();
1323 goto theend;
1324 }
Bram Moolenaar8767f522016-07-01 17:17:39 +02001325 if ((size = vim_lseek(mfp->mf_fd, (off_T)0L, SEEK_END)) <= 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001326 mfp->mf_blocknr_max = 0; /* no file or empty file */
1327 else
1328 mfp->mf_blocknr_max = (blocknr_T)(size / mfp->mf_page_size);
1329 mfp->mf_infile_count = mfp->mf_blocknr_max;
Bram Moolenaar1c536282007-04-26 15:21:56 +00001330
1331 /* need to reallocate the memory used to store the data */
1332 p = alloc(mfp->mf_page_size);
1333 if (p == NULL)
1334 goto theend;
1335 mch_memmove(p, hp->bh_data, previous_page_size);
1336 vim_free(hp->bh_data);
1337 hp->bh_data = p;
1338 b0p = (ZERO_BL *)(hp->bh_data);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001339 }
1340
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001341 /*
1342 * If .swp file name given directly, use name from swap file for buffer.
1343 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001344 if (directly)
1345 {
1346 expand_env(b0p->b0_fname, NameBuff, MAXPATHL);
1347 if (setfname(curbuf, NameBuff, NULL, TRUE) == FAIL)
1348 goto theend;
1349 }
1350
1351 home_replace(NULL, mfp->mf_fname, NameBuff, MAXPATHL, TRUE);
Bram Moolenaar555b2802005-05-19 21:08:39 +00001352 smsg((char_u *)_("Using swap file \"%s\""), NameBuff);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001353
1354 if (buf_spname(curbuf) != NULL)
Bram Moolenaare1704ba2012-10-03 18:25:00 +02001355 vim_strncpy(NameBuff, buf_spname(curbuf), MAXPATHL - 1);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001356 else
1357 home_replace(NULL, curbuf->b_ffname, NameBuff, MAXPATHL, TRUE);
Bram Moolenaar555b2802005-05-19 21:08:39 +00001358 smsg((char_u *)_("Original file \"%s\""), NameBuff);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001359 msg_putchar('\n');
1360
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001361 /*
1362 * check date of swap file and original file
1363 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001364 mtime = char_to_long(b0p->b0_mtime);
1365 if (curbuf->b_ffname != NULL
1366 && mch_stat((char *)curbuf->b_ffname, &org_stat) != -1
1367 && ((mch_stat((char *)mfp->mf_fname, &swp_stat) != -1
1368 && org_stat.st_mtime > swp_stat.st_mtime)
1369 || org_stat.st_mtime != mtime))
1370 {
1371 EMSG(_("E308: Warning: Original file may have been changed"));
1372 }
1373 out_flush();
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001374
1375 /* Get the 'fileformat' and 'fileencoding' from block zero. */
1376 b0_ff = (b0p->b0_flags & B0_FF_MASK);
1377 if (b0p->b0_flags & B0_HAS_FENC)
1378 {
Bram Moolenaarf506c5b2010-06-22 06:28:58 +02001379 int fnsize = B0_FNAME_SIZE_NOCRYPT;
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001380
1381#ifdef FEAT_CRYPT
1382 /* Use the same size as in add_b0_fenc(). */
1383 if (b0p->b0_id[1] != BLOCK0_ID1)
Bram Moolenaarf506c5b2010-06-22 06:28:58 +02001384 fnsize = B0_FNAME_SIZE_CRYPT;
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001385#endif
Bram Moolenaarf506c5b2010-06-22 06:28:58 +02001386 for (p = b0p->b0_fname + fnsize; p > b0p->b0_fname && p[-1] != NUL; --p)
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001387 ;
Bram Moolenaarf506c5b2010-06-22 06:28:58 +02001388 b0_fenc = vim_strnsave(p, (int)(b0p->b0_fname + fnsize - p));
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001389 }
1390
Bram Moolenaar071d4272004-06-13 20:20:40 +00001391 mf_put(mfp, hp, FALSE, FALSE); /* release block 0 */
1392 hp = NULL;
1393
1394 /*
1395 * Now that we are sure that the file is going to be recovered, clear the
1396 * contents of the current buffer.
1397 */
1398 while (!(curbuf->b_ml.ml_flags & ML_EMPTY))
1399 ml_delete((linenr_T)1, FALSE);
1400
1401 /*
1402 * Try reading the original file to obtain the values of 'fileformat',
1403 * 'fileencoding', etc. Ignore errors. The text itself is not used.
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001404 * When the file is encrypted the user is asked to enter the key.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001405 */
1406 if (curbuf->b_ffname != NULL)
Bram Moolenaarfc2d5bd2010-05-15 17:06:53 +02001407 orig_file_status = readfile(curbuf->b_ffname, NULL, (linenr_T)0,
Bram Moolenaar071d4272004-06-13 20:20:40 +00001408 (linenr_T)0, (linenr_T)MAXLNUM, NULL, READ_NEW);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001409
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001410#ifdef FEAT_CRYPT
1411 if (b0_cm >= 0)
1412 {
1413 /* Need to ask the user for the crypt key. If this fails we continue
1414 * without a key, will probably get garbage text. */
1415 if (*curbuf->b_p_key != NUL)
1416 {
1417 smsg((char_u *)_("Swap file is encrypted: \"%s\""), fname_used);
1418 MSG_PUTS(_("\nIf you entered a new crypt key but did not write the text file,"));
1419 MSG_PUTS(_("\nenter the new crypt key."));
1420 MSG_PUTS(_("\nIf you wrote the text file after changing the crypt key press enter"));
1421 MSG_PUTS(_("\nto use the same key for text file and swap file"));
1422 }
1423 else
1424 smsg((char_u *)_(need_key_msg), fname_used);
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02001425 buf->b_p_key = crypt_get_key(FALSE, FALSE);
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001426 if (buf->b_p_key == NULL)
1427 buf->b_p_key = curbuf->b_p_key;
1428 else if (*buf->b_p_key == NUL)
1429 {
1430 vim_free(buf->b_p_key);
1431 buf->b_p_key = curbuf->b_p_key;
1432 }
1433 if (buf->b_p_key == NULL)
1434 buf->b_p_key = empty_option;
1435 }
1436#endif
1437
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00001438 /* Use the 'fileformat' and 'fileencoding' as stored in the swap file. */
1439 if (b0_ff != 0)
1440 set_fileformat(b0_ff - 1, OPT_LOCAL);
1441 if (b0_fenc != NULL)
1442 {
1443 set_option_value((char_u *)"fenc", 0L, b0_fenc, OPT_LOCAL);
1444 vim_free(b0_fenc);
1445 }
1446 unchanged(curbuf, TRUE);
1447
Bram Moolenaar071d4272004-06-13 20:20:40 +00001448 bnum = 1; /* start with block 1 */
1449 page_count = 1; /* which is 1 page */
1450 lnum = 0; /* append after line 0 in curbuf */
1451 line_count = 0;
1452 idx = 0; /* start with first index in block 1 */
1453 error = 0;
1454 buf->b_ml.ml_stack_top = 0;
1455 buf->b_ml.ml_stack = NULL;
1456 buf->b_ml.ml_stack_size = 0; /* no stack yet */
1457
1458 if (curbuf->b_ffname == NULL)
1459 cannot_open = TRUE;
1460 else
1461 cannot_open = FALSE;
1462
1463 serious_error = FALSE;
1464 for ( ; !got_int; line_breakcheck())
1465 {
1466 if (hp != NULL)
1467 mf_put(mfp, hp, FALSE, FALSE); /* release previous block */
1468
1469 /*
1470 * get block
1471 */
1472 if ((hp = mf_get(mfp, (blocknr_T)bnum, page_count)) == NULL)
1473 {
1474 if (bnum == 1)
1475 {
1476 EMSG2(_("E309: Unable to read block 1 from %s"), mfp->mf_fname);
1477 goto theend;
1478 }
1479 ++error;
1480 ml_append(lnum++, (char_u *)_("???MANY LINES MISSING"),
1481 (colnr_T)0, TRUE);
1482 }
1483 else /* there is a block */
1484 {
1485 pp = (PTR_BL *)(hp->bh_data);
1486 if (pp->pb_id == PTR_ID) /* it is a pointer block */
1487 {
1488 /* check line count when using pointer block first time */
1489 if (idx == 0 && line_count != 0)
1490 {
1491 for (i = 0; i < (int)pp->pb_count; ++i)
1492 line_count -= pp->pb_pointer[i].pe_line_count;
1493 if (line_count != 0)
1494 {
1495 ++error;
1496 ml_append(lnum++, (char_u *)_("???LINE COUNT WRONG"),
1497 (colnr_T)0, TRUE);
1498 }
1499 }
1500
1501 if (pp->pb_count == 0)
1502 {
1503 ml_append(lnum++, (char_u *)_("???EMPTY BLOCK"),
1504 (colnr_T)0, TRUE);
1505 ++error;
1506 }
1507 else if (idx < (int)pp->pb_count) /* go a block deeper */
1508 {
1509 if (pp->pb_pointer[idx].pe_bnum < 0)
1510 {
1511 /*
1512 * Data block with negative block number.
1513 * Try to read lines from the original file.
1514 * This is slow, but it works.
1515 */
1516 if (!cannot_open)
1517 {
1518 line_count = pp->pb_pointer[idx].pe_line_count;
1519 if (readfile(curbuf->b_ffname, NULL, lnum,
1520 pp->pb_pointer[idx].pe_old_lnum - 1,
Bram Moolenaare13b9af2017-01-13 22:01:02 +01001521 line_count, NULL, 0) != OK)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001522 cannot_open = TRUE;
1523 else
1524 lnum += line_count;
1525 }
1526 if (cannot_open)
1527 {
1528 ++error;
1529 ml_append(lnum++, (char_u *)_("???LINES MISSING"),
1530 (colnr_T)0, TRUE);
1531 }
1532 ++idx; /* get same block again for next index */
1533 continue;
1534 }
1535
1536 /*
1537 * going one block deeper in the tree
1538 */
1539 if ((top = ml_add_stack(buf)) < 0) /* new entry in stack */
1540 {
1541 ++error;
1542 break; /* out of memory */
1543 }
1544 ip = &(buf->b_ml.ml_stack[top]);
1545 ip->ip_bnum = bnum;
1546 ip->ip_index = idx;
1547
1548 bnum = pp->pb_pointer[idx].pe_bnum;
1549 line_count = pp->pb_pointer[idx].pe_line_count;
1550 page_count = pp->pb_pointer[idx].pe_page_count;
Bram Moolenaar986a0032011-06-13 01:07:27 +02001551 idx = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001552 continue;
1553 }
1554 }
1555 else /* not a pointer block */
1556 {
1557 dp = (DATA_BL *)(hp->bh_data);
1558 if (dp->db_id != DATA_ID) /* block id wrong */
1559 {
1560 if (bnum == 1)
1561 {
1562 EMSG2(_("E310: Block 1 ID wrong (%s not a .swp file?)"),
1563 mfp->mf_fname);
1564 goto theend;
1565 }
1566 ++error;
1567 ml_append(lnum++, (char_u *)_("???BLOCK MISSING"),
1568 (colnr_T)0, TRUE);
1569 }
1570 else
1571 {
1572 /*
1573 * it is a data block
1574 * Append all the lines in this block
1575 */
1576 has_error = FALSE;
1577 /*
1578 * check length of block
1579 * if wrong, use length in pointer block
1580 */
1581 if (page_count * mfp->mf_page_size != dp->db_txt_end)
1582 {
1583 ml_append(lnum++, (char_u *)_("??? from here until ???END lines may be messed up"),
1584 (colnr_T)0, TRUE);
1585 ++error;
1586 has_error = TRUE;
1587 dp->db_txt_end = page_count * mfp->mf_page_size;
1588 }
1589
1590 /* make sure there is a NUL at the end of the block */
1591 *((char_u *)dp + dp->db_txt_end - 1) = NUL;
1592
1593 /*
1594 * check number of lines in block
1595 * if wrong, use count in data block
1596 */
1597 if (line_count != dp->db_line_count)
1598 {
1599 ml_append(lnum++, (char_u *)_("??? from here until ???END lines may have been inserted/deleted"),
1600 (colnr_T)0, TRUE);
1601 ++error;
1602 has_error = TRUE;
1603 }
1604
1605 for (i = 0; i < dp->db_line_count; ++i)
1606 {
1607 txt_start = (dp->db_index[i] & DB_INDEX_MASK);
Bram Moolenaar740885b2009-11-03 14:33:17 +00001608 if (txt_start <= (int)HEADER_SIZE
Bram Moolenaar071d4272004-06-13 20:20:40 +00001609 || txt_start >= (int)dp->db_txt_end)
1610 {
1611 p = (char_u *)"???";
1612 ++error;
1613 }
1614 else
1615 p = (char_u *)dp + txt_start;
1616 ml_append(lnum++, p, (colnr_T)0, TRUE);
1617 }
1618 if (has_error)
Bram Moolenaar740885b2009-11-03 14:33:17 +00001619 ml_append(lnum++, (char_u *)_("???END"),
1620 (colnr_T)0, TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001621 }
1622 }
1623 }
1624
1625 if (buf->b_ml.ml_stack_top == 0) /* finished */
1626 break;
1627
1628 /*
1629 * go one block up in the tree
1630 */
1631 ip = &(buf->b_ml.ml_stack[--(buf->b_ml.ml_stack_top)]);
1632 bnum = ip->ip_bnum;
1633 idx = ip->ip_index + 1; /* go to next index */
1634 page_count = 1;
1635 }
1636
1637 /*
Bram Moolenaarfc2d5bd2010-05-15 17:06:53 +02001638 * Compare the buffer contents with the original file. When they differ
1639 * set the 'modified' flag.
1640 * Lines 1 - lnum are the new contents.
1641 * Lines lnum + 1 to ml_line_count are the original contents.
1642 * Line ml_line_count + 1 in the dummy empty line.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001643 */
Bram Moolenaarfc2d5bd2010-05-15 17:06:53 +02001644 if (orig_file_status != OK || curbuf->b_ml.ml_line_count != lnum * 2 + 1)
1645 {
1646 /* Recovering an empty file results in two lines and the first line is
1647 * empty. Don't set the modified flag then. */
1648 if (!(curbuf->b_ml.ml_line_count == 2 && *ml_get(1) == NUL))
1649 {
1650 changed_int();
Bram Moolenaar95c526e2017-02-25 14:59:34 +01001651 ++CHANGEDTICK(curbuf);
Bram Moolenaarfc2d5bd2010-05-15 17:06:53 +02001652 }
1653 }
1654 else
1655 {
1656 for (idx = 1; idx <= lnum; ++idx)
1657 {
1658 /* Need to copy one line, fetching the other one may flush it. */
1659 p = vim_strsave(ml_get(idx));
1660 i = STRCMP(p, ml_get(idx + lnum));
1661 vim_free(p);
1662 if (i != 0)
1663 {
1664 changed_int();
Bram Moolenaar95c526e2017-02-25 14:59:34 +01001665 ++CHANGEDTICK(curbuf);
Bram Moolenaarfc2d5bd2010-05-15 17:06:53 +02001666 break;
1667 }
1668 }
1669 }
1670
1671 /*
1672 * Delete the lines from the original file and the dummy line from the
1673 * empty buffer. These will now be after the last line in the buffer.
1674 */
1675 while (curbuf->b_ml.ml_line_count > lnum
1676 && !(curbuf->b_ml.ml_flags & ML_EMPTY))
1677 ml_delete(curbuf->b_ml.ml_line_count, FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001678 curbuf->b_flags |= BF_RECOVERED;
1679
1680 recoverymode = FALSE;
1681 if (got_int)
1682 EMSG(_("E311: Recovery Interrupted"));
1683 else if (error)
1684 {
1685 ++no_wait_return;
1686 MSG(">>>>>>>>>>>>>");
1687 EMSG(_("E312: Errors detected while recovering; look for lines starting with ???"));
1688 --no_wait_return;
1689 MSG(_("See \":help E312\" for more information."));
1690 MSG(">>>>>>>>>>>>>");
1691 }
1692 else
1693 {
Bram Moolenaarfc2d5bd2010-05-15 17:06:53 +02001694 if (curbuf->b_changed)
1695 {
1696 MSG(_("Recovery completed. You should check if everything is OK."));
1697 MSG_PUTS(_("\n(You might want to write out this file under another name\n"));
1698 MSG_PUTS(_("and run diff with the original file to check for changes)"));
1699 }
1700 else
1701 MSG(_("Recovery completed. Buffer contents equals file contents."));
1702 MSG_PUTS(_("\nYou may want to delete the .swp file now.\n\n"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00001703 cmdline_row = msg_row;
1704 }
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001705#ifdef FEAT_CRYPT
1706 if (*buf->b_p_key != NUL && STRCMP(curbuf->b_p_key, buf->b_p_key) != 0)
1707 {
1708 MSG_PUTS(_("Using crypt key from swap file for the text file.\n"));
1709 set_option_value((char_u *)"key", 0L, buf->b_p_key, OPT_LOCAL);
1710 }
1711#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001712 redraw_curbuf_later(NOT_VALID);
1713
1714theend:
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001715 vim_free(fname_used);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001716 recoverymode = FALSE;
1717 if (mfp != NULL)
1718 {
1719 if (hp != NULL)
1720 mf_put(mfp, hp, FALSE, FALSE);
1721 mf_close(mfp, FALSE); /* will also vim_free(mfp->mf_fname) */
1722 }
Bram Moolenaardf88dda2007-01-09 13:34:50 +00001723 if (buf != NULL)
1724 {
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001725#ifdef FEAT_CRYPT
1726 if (buf->b_p_key != curbuf->b_p_key)
1727 free_string_option(buf->b_p_key);
Bram Moolenaar0ad014c2010-07-25 14:00:46 +02001728 free_string_option(buf->b_p_cm);
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001729#endif
Bram Moolenaardf88dda2007-01-09 13:34:50 +00001730 vim_free(buf->b_ml.ml_stack);
1731 vim_free(buf);
1732 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001733 if (serious_error && called_from_main)
1734 ml_close(curbuf, TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001735 else
1736 {
1737 apply_autocmds(EVENT_BUFREADPOST, NULL, curbuf->b_fname, FALSE, curbuf);
1738 apply_autocmds(EVENT_BUFWINENTER, NULL, curbuf->b_fname, FALSE, curbuf);
1739 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001740 return;
1741}
1742
1743/*
1744 * Find the names of swap files in current directory and the directory given
1745 * with the 'directory' option.
1746 *
1747 * Used to:
1748 * - list the swap files for "vim -r"
1749 * - count the number of swap files when recovering
1750 * - list the swap files when recovering
1751 * - find the name of the n'th swap file when recovering
1752 */
1753 int
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01001754recover_names(
1755 char_u *fname, /* base for swap file name */
1756 int list, /* when TRUE, list the swap file names */
1757 int nr, /* when non-zero, return nr'th swap file name */
1758 char_u **fname_out) /* result when "nr" > 0 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001759{
1760 int num_names;
1761 char_u *(names[6]);
1762 char_u *tail;
1763 char_u *p;
1764 int num_files;
1765 int file_count = 0;
1766 char_u **files;
1767 int i;
1768 char_u *dirp;
1769 char_u *dir_name;
Bram Moolenaar64354da2010-05-25 21:37:17 +02001770 char_u *fname_res = NULL;
Bram Moolenaar9dbe4752010-05-14 17:52:42 +02001771#ifdef HAVE_READLINK
1772 char_u fname_buf[MAXPATHL];
Bram Moolenaar64354da2010-05-25 21:37:17 +02001773#endif
Bram Moolenaar9dbe4752010-05-14 17:52:42 +02001774
Bram Moolenaar64354da2010-05-25 21:37:17 +02001775 if (fname != NULL)
1776 {
1777#ifdef HAVE_READLINK
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001778 /* Expand symlink in the file name, because the swap file is created
1779 * with the actual file instead of with the symlink. */
1780 if (resolve_symlink(fname, fname_buf) == OK)
1781 fname_res = fname_buf;
1782 else
Bram Moolenaar9dbe4752010-05-14 17:52:42 +02001783#endif
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001784 fname_res = fname;
Bram Moolenaar64354da2010-05-25 21:37:17 +02001785 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001786
1787 if (list)
1788 {
Bram Moolenaar9dbe4752010-05-14 17:52:42 +02001789 /* use msg() to start the scrolling properly */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001790 msg((char_u *)_("Swap files found:"));
1791 msg_putchar('\n');
1792 }
1793
1794 /*
1795 * Do the loop for every directory in 'directory'.
1796 * First allocate some memory to put the directory name in.
1797 */
1798 dir_name = alloc((unsigned)STRLEN(p_dir) + 1);
1799 dirp = p_dir;
1800 while (dir_name != NULL && *dirp)
1801 {
1802 /*
1803 * Isolate a directory name from *dirp and put it in dir_name (we know
1804 * it is large enough, so use 31000 for length).
1805 * Advance dirp to next directory name.
1806 */
1807 (void)copy_option_part(&dirp, dir_name, 31000, ",");
1808
1809 if (dir_name[0] == '.' && dir_name[1] == NUL) /* check current dir */
1810 {
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001811 if (fname == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001812 {
1813#ifdef VMS
1814 names[0] = vim_strsave((char_u *)"*_sw%");
1815#else
Bram Moolenaar071d4272004-06-13 20:20:40 +00001816 names[0] = vim_strsave((char_u *)"*.sw?");
Bram Moolenaar071d4272004-06-13 20:20:40 +00001817#endif
Bram Moolenaar2cc93182006-10-10 19:56:03 +00001818#if defined(UNIX) || defined(WIN3264)
1819 /* For Unix names starting with a dot are special. MS-Windows
1820 * supports this too, on some file systems. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001821 names[1] = vim_strsave((char_u *)".*.sw?");
1822 names[2] = vim_strsave((char_u *)".sw?");
1823 num_names = 3;
1824#else
1825# ifdef VMS
1826 names[1] = vim_strsave((char_u *)".*_sw%");
1827 num_names = 2;
1828# else
1829 num_names = 1;
1830# endif
1831#endif
1832 }
1833 else
Bram Moolenaar9dbe4752010-05-14 17:52:42 +02001834 num_names = recov_file_names(names, fname_res, TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001835 }
1836 else /* check directory dir_name */
1837 {
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001838 if (fname == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001839 {
1840#ifdef VMS
1841 names[0] = concat_fnames(dir_name, (char_u *)"*_sw%", TRUE);
1842#else
Bram Moolenaar071d4272004-06-13 20:20:40 +00001843 names[0] = concat_fnames(dir_name, (char_u *)"*.sw?", TRUE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001844#endif
Bram Moolenaar2cc93182006-10-10 19:56:03 +00001845#if defined(UNIX) || defined(WIN3264)
1846 /* For Unix names starting with a dot are special. MS-Windows
1847 * supports this too, on some file systems. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001848 names[1] = concat_fnames(dir_name, (char_u *)".*.sw?", TRUE);
1849 names[2] = concat_fnames(dir_name, (char_u *)".sw?", TRUE);
1850 num_names = 3;
1851#else
1852# ifdef VMS
1853 names[1] = concat_fnames(dir_name, (char_u *)".*_sw%", TRUE);
1854 num_names = 2;
1855# else
1856 num_names = 1;
1857# endif
1858#endif
1859 }
1860 else
1861 {
1862#if defined(UNIX) || defined(WIN3264)
Bram Moolenaarb113c3a2017-02-28 21:26:17 +01001863 int len = (int)STRLEN(dir_name);
Bram Moolenaarc525e3a2017-02-18 16:59:02 +01001864
1865 p = dir_name + len;
1866 if (after_pathsep(dir_name, p) && len > 1 && p[-1] == p[-2])
Bram Moolenaar071d4272004-06-13 20:20:40 +00001867 {
1868 /* Ends with '//', Use Full path for swap name */
Bram Moolenaar9dbe4752010-05-14 17:52:42 +02001869 tail = make_percent_swname(dir_name, fname_res);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001870 }
1871 else
1872#endif
1873 {
Bram Moolenaar9dbe4752010-05-14 17:52:42 +02001874 tail = gettail(fname_res);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001875 tail = concat_fnames(dir_name, tail, TRUE);
1876 }
1877 if (tail == NULL)
1878 num_names = 0;
1879 else
1880 {
1881 num_names = recov_file_names(names, tail, FALSE);
1882 vim_free(tail);
1883 }
1884 }
1885 }
1886
1887 /* check for out-of-memory */
1888 for (i = 0; i < num_names; ++i)
1889 {
1890 if (names[i] == NULL)
1891 {
1892 for (i = 0; i < num_names; ++i)
1893 vim_free(names[i]);
1894 num_names = 0;
1895 }
1896 }
1897 if (num_names == 0)
1898 num_files = 0;
1899 else if (expand_wildcards(num_names, names, &num_files, &files,
1900 EW_KEEPALL|EW_FILE|EW_SILENT) == FAIL)
1901 num_files = 0;
1902
1903 /*
1904 * When no swap file found, wildcard expansion might have failed (e.g.
1905 * not able to execute the shell).
1906 * Try finding a swap file by simply adding ".swp" to the file name.
1907 */
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001908 if (*dirp == NUL && file_count + num_files == 0 && fname != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001909 {
Bram Moolenaar8767f522016-07-01 17:17:39 +02001910 stat_T st;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001911 char_u *swapname;
1912
Bram Moolenaar9dbe4752010-05-14 17:52:42 +02001913 swapname = modname(fname_res,
Bram Moolenaare60acc12011-05-10 16:41:25 +02001914#if defined(VMS)
Bram Moolenaar9dbe4752010-05-14 17:52:42 +02001915 (char_u *)"_swp", FALSE
Bram Moolenaar071d4272004-06-13 20:20:40 +00001916#else
Bram Moolenaar9dbe4752010-05-14 17:52:42 +02001917 (char_u *)".swp", TRUE
Bram Moolenaar071d4272004-06-13 20:20:40 +00001918#endif
Bram Moolenaar9dbe4752010-05-14 17:52:42 +02001919 );
Bram Moolenaar071d4272004-06-13 20:20:40 +00001920 if (swapname != NULL)
1921 {
1922 if (mch_stat((char *)swapname, &st) != -1) /* It exists! */
1923 {
1924 files = (char_u **)alloc((unsigned)sizeof(char_u *));
1925 if (files != NULL)
1926 {
1927 files[0] = swapname;
1928 swapname = NULL;
1929 num_files = 1;
1930 }
1931 }
1932 vim_free(swapname);
1933 }
1934 }
1935
1936 /*
1937 * remove swapfile name of the current buffer, it must be ignored
1938 */
1939 if (curbuf->b_ml.ml_mfp != NULL
1940 && (p = curbuf->b_ml.ml_mfp->mf_fname) != NULL)
1941 {
1942 for (i = 0; i < num_files; ++i)
1943 if (fullpathcmp(p, files[i], TRUE) & FPC_SAME)
1944 {
Bram Moolenaar9439cdd2009-04-22 13:39:36 +00001945 /* Remove the name from files[i]. Move further entries
1946 * down. When the array becomes empty free it here, since
1947 * FreeWild() won't be called below. */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001948 vim_free(files[i]);
Bram Moolenaar9439cdd2009-04-22 13:39:36 +00001949 if (--num_files == 0)
1950 vim_free(files);
1951 else
1952 for ( ; i < num_files; ++i)
1953 files[i] = files[i + 1];
Bram Moolenaar071d4272004-06-13 20:20:40 +00001954 }
1955 }
Bram Moolenaarfc1421e2006-04-20 22:17:20 +00001956 if (nr > 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001957 {
1958 file_count += num_files;
1959 if (nr <= file_count)
1960 {
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001961 *fname_out = vim_strsave(
1962 files[nr - 1 + num_files - file_count]);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001963 dirp = (char_u *)""; /* stop searching */
1964 }
1965 }
1966 else if (list)
1967 {
1968 if (dir_name[0] == '.' && dir_name[1] == NUL)
1969 {
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001970 if (fname == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001971 MSG_PUTS(_(" In current directory:\n"));
1972 else
1973 MSG_PUTS(_(" Using specified name:\n"));
1974 }
1975 else
1976 {
1977 MSG_PUTS(_(" In directory "));
1978 msg_home_replace(dir_name);
1979 MSG_PUTS(":\n");
1980 }
1981
1982 if (num_files)
1983 {
1984 for (i = 0; i < num_files; ++i)
1985 {
1986 /* print the swap file name */
1987 msg_outnum((long)++file_count);
1988 MSG_PUTS(". ");
1989 msg_puts(gettail(files[i]));
1990 msg_putchar('\n');
1991 (void)swapfile_info(files[i]);
1992 }
1993 }
1994 else
1995 MSG_PUTS(_(" -- none --\n"));
1996 out_flush();
1997 }
1998 else
1999 file_count += num_files;
2000
2001 for (i = 0; i < num_names; ++i)
2002 vim_free(names[i]);
Bram Moolenaarfc1421e2006-04-20 22:17:20 +00002003 if (num_files > 0)
2004 FreeWild(num_files, files);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002005 }
2006 vim_free(dir_name);
2007 return file_count;
2008}
2009
Bram Moolenaarb782ba42018-08-07 21:39:28 +02002010#if defined(UNIX) || defined(WIN3264) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002011/*
Bram Moolenaarb782ba42018-08-07 21:39:28 +02002012 * Need _very_ long file names.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002013 * Append the full path to name with path separators made into percent
2014 * signs, to dir. An unnamed buffer is handled as "" (<currentdir>/"")
2015 */
Bram Moolenaarb782ba42018-08-07 21:39:28 +02002016 char_u *
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01002017make_percent_swname(char_u *dir, char_u *name)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002018{
Bram Moolenaarb782ba42018-08-07 21:39:28 +02002019 char_u *d = NULL, *s, *f;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002020
Bram Moolenaarb782ba42018-08-07 21:39:28 +02002021 f = fix_fname(name != NULL ? name : (char_u *)"");
Bram Moolenaar071d4272004-06-13 20:20:40 +00002022 if (f != NULL)
2023 {
2024 s = alloc((unsigned)(STRLEN(f) + 1));
2025 if (s != NULL)
2026 {
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002027 STRCPY(s, f);
Bram Moolenaar91acfff2017-03-12 19:22:36 +01002028 for (d = s; *d != NUL; MB_PTR_ADV(d))
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00002029 if (vim_ispathsep(*d))
2030 *d = '%';
Bram Moolenaar071d4272004-06-13 20:20:40 +00002031 d = concat_fnames(dir, s, TRUE);
2032 vim_free(s);
2033 }
2034 vim_free(f);
2035 }
2036 return d;
2037}
2038#endif
2039
Bram Moolenaara06ecab2016-07-16 14:47:36 +02002040#if (defined(UNIX) || defined(VMS)) && (defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002041static int process_still_running;
2042#endif
2043
Bram Moolenaar47ad5652018-08-21 21:09:07 +02002044#if defined(FEAT_EVAL) || defined(PROTO)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002045/*
Bram Moolenaar00f123a2018-08-21 20:28:54 +02002046 * Return information found in swapfile "fname" in dictionary "d".
2047 * This is used by the swapinfo() function.
2048 */
2049 void
2050get_b0_dict(char_u *fname, dict_T *d)
2051{
2052 int fd;
2053 struct block0 b0;
2054
2055 if ((fd = mch_open((char *)fname, O_RDONLY | O_EXTRA, 0)) >= 0)
2056 {
2057 if (read_eintr(fd, &b0, sizeof(b0)) == sizeof(b0))
2058 {
Bram Moolenaar47ad5652018-08-21 21:09:07 +02002059 if (ml_check_b0_id(&b0) == FAIL)
Bram Moolenaar00f123a2018-08-21 20:28:54 +02002060 dict_add_string(d, "error",
Bram Moolenaar47ad5652018-08-21 21:09:07 +02002061 vim_strsave((char_u *)"Not a swap file"));
2062 else if (b0_magic_wrong(&b0))
2063 dict_add_string(d, "error",
2064 vim_strsave((char_u *)"Magic number mismatch"));
Bram Moolenaar00f123a2018-08-21 20:28:54 +02002065 else
2066 {
2067 /* we have swap information */
Bram Moolenaar7c605052018-08-23 23:01:27 +02002068 dict_add_string(d, "version", vim_strnsave(b0.b0_version, 10));
2069 dict_add_string(d, "user",
2070 vim_strnsave(b0.b0_uname, B0_UNAME_SIZE));
2071 dict_add_string(d, "host",
2072 vim_strnsave(b0.b0_hname, B0_HNAME_SIZE));
2073 dict_add_string(d, "fname",
2074 vim_strnsave(b0.b0_fname, B0_FNAME_SIZE_ORG));
Bram Moolenaar00f123a2018-08-21 20:28:54 +02002075
2076 dict_add_number(d, "pid", char_to_long(b0.b0_pid));
2077 dict_add_number(d, "mtime", char_to_long(b0.b0_mtime));
Bram Moolenaar47ad5652018-08-21 21:09:07 +02002078 dict_add_number(d, "dirty", b0.b0_dirty ? 1 : 0);
2079# ifdef CHECK_INODE
Bram Moolenaar00f123a2018-08-21 20:28:54 +02002080 dict_add_number(d, "inode", char_to_long(b0.b0_ino));
Bram Moolenaar47ad5652018-08-21 21:09:07 +02002081# endif
Bram Moolenaar00f123a2018-08-21 20:28:54 +02002082 }
2083 }
2084 else
2085 dict_add_string(d, "error",
2086 vim_strsave((char_u *)"Cannot read file"));
2087 close(fd);
2088 }
2089 else
2090 dict_add_string(d, "error", vim_strsave((char_u *)"Cannot open file"));
2091}
Bram Moolenaar47ad5652018-08-21 21:09:07 +02002092#endif
Bram Moolenaar00f123a2018-08-21 20:28:54 +02002093
2094/*
Bram Moolenaard5bc83f2005-12-07 21:07:59 +00002095 * Give information about an existing swap file.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002096 * Returns timestamp (0 when unknown).
2097 */
2098 static time_t
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01002099swapfile_info(char_u *fname)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002100{
Bram Moolenaar8767f522016-07-01 17:17:39 +02002101 stat_T st;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002102 int fd;
2103 struct block0 b0;
2104 time_t x = (time_t)0;
Bram Moolenaar31e97bf2006-10-10 14:20:13 +00002105 char *p;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002106#ifdef UNIX
2107 char_u uname[B0_UNAME_SIZE];
2108#endif
2109
2110 /* print the swap file date */
2111 if (mch_stat((char *)fname, &st) != -1)
2112 {
2113#ifdef UNIX
2114 /* print name of owner of the file */
2115 if (mch_get_uname(st.st_uid, uname, B0_UNAME_SIZE) == OK)
2116 {
2117 MSG_PUTS(_(" owned by: "));
2118 msg_outtrans(uname);
2119 MSG_PUTS(_(" dated: "));
2120 }
2121 else
2122#endif
2123 MSG_PUTS(_(" dated: "));
2124 x = st.st_mtime; /* Manx C can't do &st.st_mtime */
Bram Moolenaar31e97bf2006-10-10 14:20:13 +00002125 p = ctime(&x); /* includes '\n' */
2126 if (p == NULL)
2127 MSG_PUTS("(invalid)\n");
2128 else
2129 MSG_PUTS(p);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002130 }
2131
2132 /*
2133 * print the original file name
2134 */
2135 fd = mch_open((char *)fname, O_RDONLY | O_EXTRA, 0);
2136 if (fd >= 0)
2137 {
Bram Moolenaar540fc6f2010-12-17 16:27:16 +01002138 if (read_eintr(fd, &b0, sizeof(b0)) == sizeof(b0))
Bram Moolenaar071d4272004-06-13 20:20:40 +00002139 {
2140 if (STRNCMP(b0.b0_version, "VIM 3.0", 7) == 0)
2141 {
2142 MSG_PUTS(_(" [from Vim version 3.0]"));
2143 }
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02002144 else if (ml_check_b0_id(&b0) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002145 {
2146 MSG_PUTS(_(" [does not look like a Vim swap file]"));
2147 }
2148 else
2149 {
2150 MSG_PUTS(_(" file name: "));
2151 if (b0.b0_fname[0] == NUL)
Bram Moolenaar69a7cb42004-06-20 12:51:53 +00002152 MSG_PUTS(_("[No Name]"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00002153 else
2154 msg_outtrans(b0.b0_fname);
2155
2156 MSG_PUTS(_("\n modified: "));
2157 MSG_PUTS(b0.b0_dirty ? _("YES") : _("no"));
2158
2159 if (*(b0.b0_uname) != NUL)
2160 {
2161 MSG_PUTS(_("\n user name: "));
2162 msg_outtrans(b0.b0_uname);
2163 }
2164
2165 if (*(b0.b0_hname) != NUL)
2166 {
2167 if (*(b0.b0_uname) != NUL)
2168 MSG_PUTS(_(" host name: "));
2169 else
2170 MSG_PUTS(_("\n host name: "));
2171 msg_outtrans(b0.b0_hname);
2172 }
2173
2174 if (char_to_long(b0.b0_pid) != 0L)
2175 {
2176 MSG_PUTS(_("\n process ID: "));
2177 msg_outnum(char_to_long(b0.b0_pid));
Bram Moolenaara06ecab2016-07-16 14:47:36 +02002178#if defined(UNIX)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002179 /* EMX kill() not working correctly, it seems */
2180 if (kill((pid_t)char_to_long(b0.b0_pid), 0) == 0)
2181 {
2182 MSG_PUTS(_(" (still running)"));
2183# if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG)
2184 process_still_running = TRUE;
2185# endif
2186 }
2187#endif
2188 }
2189
2190 if (b0_magic_wrong(&b0))
2191 {
Bram Moolenaar48e330a2016-02-23 14:53:34 +01002192#if defined(MSWIN)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002193 if (STRNCMP(b0.b0_hname, "PC ", 3) == 0)
2194 MSG_PUTS(_("\n [not usable with this version of Vim]"));
2195 else
2196#endif
2197 MSG_PUTS(_("\n [not usable on this computer]"));
2198 }
2199 }
2200 }
2201 else
2202 MSG_PUTS(_(" [cannot be read]"));
2203 close(fd);
2204 }
2205 else
2206 MSG_PUTS(_(" [cannot be opened]"));
2207 msg_putchar('\n');
2208
2209 return x;
2210}
2211
2212 static int
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01002213recov_file_names(char_u **names, char_u *path, int prepend_dot)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002214{
2215 int num_names;
2216
Bram Moolenaar071d4272004-06-13 20:20:40 +00002217 /*
2218 * (Win32 and Win64) never short names, but do prepend a dot.
2219 * (Not MS-DOS or Win32 or Win64) maybe short name, maybe not: Try both.
2220 * Only use the short name if it is different.
2221 */
2222 char_u *p;
2223 int i;
2224# ifndef WIN3264
2225 int shortname = curbuf->b_shortname;
2226
2227 curbuf->b_shortname = FALSE;
2228# endif
2229
2230 num_names = 0;
2231
2232 /*
2233 * May also add the file name with a dot prepended, for swap file in same
2234 * dir as original file.
2235 */
2236 if (prepend_dot)
2237 {
2238 names[num_names] = modname(path, (char_u *)".sw?", TRUE);
2239 if (names[num_names] == NULL)
2240 goto end;
2241 ++num_names;
2242 }
2243
2244 /*
2245 * Form the normal swap file name pattern by appending ".sw?".
2246 */
2247#ifdef VMS
2248 names[num_names] = concat_fnames(path, (char_u *)"_sw%", FALSE);
2249#else
Bram Moolenaar071d4272004-06-13 20:20:40 +00002250 names[num_names] = concat_fnames(path, (char_u *)".sw?", FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002251#endif
2252 if (names[num_names] == NULL)
2253 goto end;
2254 if (num_names >= 1) /* check if we have the same name twice */
2255 {
2256 p = names[num_names - 1];
2257 i = (int)STRLEN(names[num_names - 1]) - (int)STRLEN(names[num_names]);
2258 if (i > 0)
2259 p += i; /* file name has been expanded to full path */
2260
2261 if (STRCMP(p, names[num_names]) != 0)
2262 ++num_names;
2263 else
2264 vim_free(names[num_names]);
2265 }
2266 else
2267 ++num_names;
2268
2269# ifndef WIN3264
2270 /*
2271 * Also try with 'shortname' set, in case the file is on a DOS filesystem.
2272 */
2273 curbuf->b_shortname = TRUE;
2274#ifdef VMS
2275 names[num_names] = modname(path, (char_u *)"_sw%", FALSE);
2276#else
Bram Moolenaar071d4272004-06-13 20:20:40 +00002277 names[num_names] = modname(path, (char_u *)".sw?", FALSE);
Bram Moolenaar071d4272004-06-13 20:20:40 +00002278#endif
2279 if (names[num_names] == NULL)
2280 goto end;
2281
2282 /*
2283 * Remove the one from 'shortname', if it's the same as with 'noshortname'.
2284 */
2285 p = names[num_names];
2286 i = STRLEN(names[num_names]) - STRLEN(names[num_names - 1]);
2287 if (i > 0)
2288 p += i; /* file name has been expanded to full path */
2289 if (STRCMP(names[num_names - 1], p) == 0)
2290 vim_free(names[num_names]);
2291 else
2292 ++num_names;
2293# endif
2294
2295end:
2296# ifndef WIN3264
2297 curbuf->b_shortname = shortname;
2298# endif
2299
Bram Moolenaar071d4272004-06-13 20:20:40 +00002300 return num_names;
2301}
2302
2303/*
2304 * sync all memlines
2305 *
2306 * If 'check_file' is TRUE, check if original file exists and was not changed.
2307 * If 'check_char' is TRUE, stop syncing when character becomes available, but
2308 * always sync at least one block.
2309 */
2310 void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01002311ml_sync_all(int check_file, int check_char)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002312{
2313 buf_T *buf;
Bram Moolenaar8767f522016-07-01 17:17:39 +02002314 stat_T st;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002315
Bram Moolenaar29323592016-07-24 22:04:11 +02002316 FOR_ALL_BUFFERS(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002317 {
2318 if (buf->b_ml.ml_mfp == NULL || buf->b_ml.ml_mfp->mf_fname == NULL)
2319 continue; /* no file */
2320
2321 ml_flush_line(buf); /* flush buffered line */
2322 /* flush locked block */
2323 (void)ml_find_line(buf, (linenr_T)0, ML_FLUSH);
2324 if (bufIsChanged(buf) && check_file && mf_need_trans(buf->b_ml.ml_mfp)
2325 && buf->b_ffname != NULL)
2326 {
2327 /*
2328 * If the original file does not exist anymore or has been changed
2329 * call ml_preserve() to get rid of all negative numbered blocks.
2330 */
2331 if (mch_stat((char *)buf->b_ffname, &st) == -1
2332 || st.st_mtime != buf->b_mtime_read
Bram Moolenaar914703b2010-05-31 21:59:46 +02002333 || st.st_size != buf->b_orig_size)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002334 {
2335 ml_preserve(buf, FALSE);
2336 did_check_timestamps = FALSE;
2337 need_check_timestamps = TRUE; /* give message later */
2338 }
2339 }
2340 if (buf->b_ml.ml_mfp->mf_dirty)
2341 {
2342 (void)mf_sync(buf->b_ml.ml_mfp, (check_char ? MFS_STOP : 0)
2343 | (bufIsChanged(buf) ? MFS_FLUSH : 0));
2344 if (check_char && ui_char_avail()) /* character available now */
2345 break;
2346 }
2347 }
2348}
2349
2350/*
2351 * sync one buffer, including negative blocks
2352 *
2353 * after this all the blocks are in the swap file
2354 *
2355 * Used for the :preserve command and when the original file has been
2356 * changed or deleted.
2357 *
2358 * when message is TRUE the success of preserving is reported
2359 */
2360 void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01002361ml_preserve(buf_T *buf, int message)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002362{
2363 bhdr_T *hp;
2364 linenr_T lnum;
2365 memfile_T *mfp = buf->b_ml.ml_mfp;
2366 int status;
2367 int got_int_save = got_int;
2368
2369 if (mfp == NULL || mfp->mf_fname == NULL)
2370 {
2371 if (message)
2372 EMSG(_("E313: Cannot preserve, there is no swap file"));
2373 return;
2374 }
2375
2376 /* We only want to stop when interrupted here, not when interrupted
2377 * before. */
2378 got_int = FALSE;
2379
2380 ml_flush_line(buf); /* flush buffered line */
2381 (void)ml_find_line(buf, (linenr_T)0, ML_FLUSH); /* flush locked block */
2382 status = mf_sync(mfp, MFS_ALL | MFS_FLUSH);
2383
2384 /* stack is invalid after mf_sync(.., MFS_ALL) */
2385 buf->b_ml.ml_stack_top = 0;
2386
2387 /*
2388 * Some of the data blocks may have been changed from negative to
2389 * positive block number. In that case the pointer blocks need to be
2390 * updated.
2391 *
2392 * We don't know in which pointer block the references are, so we visit
2393 * all data blocks until there are no more translations to be done (or
2394 * we hit the end of the file, which can only happen in case a write fails,
2395 * e.g. when file system if full).
2396 * ml_find_line() does the work by translating the negative block numbers
2397 * when getting the first line of each data block.
2398 */
2399 if (mf_need_trans(mfp) && !got_int)
2400 {
2401 lnum = 1;
2402 while (mf_need_trans(mfp) && lnum <= buf->b_ml.ml_line_count)
2403 {
2404 hp = ml_find_line(buf, lnum, ML_FIND);
2405 if (hp == NULL)
2406 {
2407 status = FAIL;
2408 goto theend;
2409 }
2410 CHECK(buf->b_ml.ml_locked_low != lnum, "low != lnum");
2411 lnum = buf->b_ml.ml_locked_high + 1;
2412 }
2413 (void)ml_find_line(buf, (linenr_T)0, ML_FLUSH); /* flush locked block */
2414 /* sync the updated pointer blocks */
2415 if (mf_sync(mfp, MFS_ALL | MFS_FLUSH) == FAIL)
2416 status = FAIL;
2417 buf->b_ml.ml_stack_top = 0; /* stack is invalid now */
2418 }
2419theend:
2420 got_int |= got_int_save;
2421
2422 if (message)
2423 {
2424 if (status == OK)
2425 MSG(_("File preserved"));
2426 else
2427 EMSG(_("E314: Preserve failed"));
2428 }
2429}
2430
2431/*
2432 * NOTE: The pointer returned by the ml_get_*() functions only remains valid
2433 * until the next call!
2434 * line1 = ml_get(1);
2435 * line2 = ml_get(2); // line1 is now invalid!
2436 * Make a copy of the line if necessary.
2437 */
2438/*
Bram Moolenaar2e2e13c2010-12-08 13:17:03 +01002439 * Return a pointer to a (read-only copy of a) line.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002440 *
2441 * On failure an error message is given and IObuff is returned (to avoid
2442 * having to check for error everywhere).
2443 */
2444 char_u *
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01002445ml_get(linenr_T lnum)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002446{
2447 return ml_get_buf(curbuf, lnum, FALSE);
2448}
2449
2450/*
Bram Moolenaar2e2e13c2010-12-08 13:17:03 +01002451 * Return pointer to position "pos".
Bram Moolenaar071d4272004-06-13 20:20:40 +00002452 */
2453 char_u *
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01002454ml_get_pos(pos_T *pos)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002455{
2456 return (ml_get_buf(curbuf, pos->lnum, FALSE) + pos->col);
2457}
2458
2459/*
Bram Moolenaar2e2e13c2010-12-08 13:17:03 +01002460 * Return pointer to cursor line.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002461 */
2462 char_u *
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01002463ml_get_curline(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002464{
2465 return ml_get_buf(curbuf, curwin->w_cursor.lnum, FALSE);
2466}
2467
2468/*
Bram Moolenaar2e2e13c2010-12-08 13:17:03 +01002469 * Return pointer to cursor position.
Bram Moolenaar071d4272004-06-13 20:20:40 +00002470 */
2471 char_u *
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01002472ml_get_cursor(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002473{
2474 return (ml_get_buf(curbuf, curwin->w_cursor.lnum, FALSE) +
2475 curwin->w_cursor.col);
2476}
2477
2478/*
Bram Moolenaar2e2e13c2010-12-08 13:17:03 +01002479 * Return a pointer to a line in a specific buffer
Bram Moolenaar071d4272004-06-13 20:20:40 +00002480 *
2481 * "will_change": if TRUE mark the buffer dirty (chars in the line will be
2482 * changed)
2483 */
2484 char_u *
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01002485ml_get_buf(
2486 buf_T *buf,
2487 linenr_T lnum,
2488 int will_change) /* line will be changed */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002489{
Bram Moolenaarad40f022007-02-13 03:01:39 +00002490 bhdr_T *hp;
2491 DATA_BL *dp;
2492 char_u *ptr;
2493 static int recursive = 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00002494
2495 if (lnum > buf->b_ml.ml_line_count) /* invalid line number */
2496 {
Bram Moolenaarad40f022007-02-13 03:01:39 +00002497 if (recursive == 0)
2498 {
2499 /* Avoid giving this message for a recursive call, may happen when
2500 * the GUI redraws part of the text. */
2501 ++recursive;
Bram Moolenaar95f09602016-11-10 20:01:45 +01002502 IEMSGN(_("E315: ml_get: invalid lnum: %ld"), lnum);
Bram Moolenaarad40f022007-02-13 03:01:39 +00002503 --recursive;
2504 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002505errorret:
2506 STRCPY(IObuff, "???");
2507 return IObuff;
2508 }
2509 if (lnum <= 0) /* pretend line 0 is line 1 */
2510 lnum = 1;
2511
2512 if (buf->b_ml.ml_mfp == NULL) /* there are no lines */
2513 return (char_u *)"";
2514
Bram Moolenaar37d619f2010-03-10 14:46:26 +01002515 /*
2516 * See if it is the same line as requested last time.
2517 * Otherwise may need to flush last used line.
2518 * Don't use the last used line when 'swapfile' is reset, need to load all
2519 * blocks.
2520 */
Bram Moolenaar47b8b152007-02-07 02:41:57 +00002521 if (buf->b_ml.ml_line_lnum != lnum || mf_dont_release)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002522 {
2523 ml_flush_line(buf);
2524
2525 /*
2526 * Find the data block containing the line.
2527 * This also fills the stack with the blocks from the root to the data
2528 * block and releases any locked block.
2529 */
2530 if ((hp = ml_find_line(buf, lnum, ML_FIND)) == NULL)
2531 {
Bram Moolenaarad40f022007-02-13 03:01:39 +00002532 if (recursive == 0)
2533 {
2534 /* Avoid giving this message for a recursive call, may happen
2535 * when the GUI redraws part of the text. */
2536 ++recursive;
Bram Moolenaar95f09602016-11-10 20:01:45 +01002537 IEMSGN(_("E316: ml_get: cannot find line %ld"), lnum);
Bram Moolenaarad40f022007-02-13 03:01:39 +00002538 --recursive;
2539 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00002540 goto errorret;
2541 }
2542
2543 dp = (DATA_BL *)(hp->bh_data);
2544
2545 ptr = (char_u *)dp + ((dp->db_index[lnum - buf->b_ml.ml_locked_low]) & DB_INDEX_MASK);
2546 buf->b_ml.ml_line_ptr = ptr;
2547 buf->b_ml.ml_line_lnum = lnum;
2548 buf->b_ml.ml_flags &= ~ML_LINE_DIRTY;
2549 }
2550 if (will_change)
2551 buf->b_ml.ml_flags |= (ML_LOCKED_DIRTY | ML_LOCKED_POS);
2552
2553 return buf->b_ml.ml_line_ptr;
2554}
2555
2556/*
2557 * Check if a line that was just obtained by a call to ml_get
2558 * is in allocated memory.
2559 */
2560 int
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01002561ml_line_alloced(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002562{
2563 return (curbuf->b_ml.ml_flags & ML_LINE_DIRTY);
2564}
2565
2566/*
2567 * Append a line after lnum (may be 0 to insert a line in front of the file).
2568 * "line" does not need to be allocated, but can't be another line in a
2569 * buffer, unlocking may make it invalid.
2570 *
2571 * newfile: TRUE when starting to edit a new file, meaning that pe_old_lnum
2572 * will be set for recovery
2573 * Check: The caller of this function should probably also call
2574 * appended_lines().
2575 *
2576 * return FAIL for failure, OK otherwise
2577 */
2578 int
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01002579ml_append(
2580 linenr_T lnum, /* append after this line (can be 0) */
2581 char_u *line, /* text of the new line */
2582 colnr_T len, /* length of new line, including NUL, or 0 */
2583 int newfile) /* flag, see above */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002584{
2585 /* When starting up, we might still need to create the memfile */
Bram Moolenaar59f931e2010-07-24 20:27:03 +02002586 if (curbuf->b_ml.ml_mfp == NULL && open_buffer(FALSE, NULL, 0) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00002587 return FAIL;
2588
2589 if (curbuf->b_ml.ml_line_lnum != 0)
2590 ml_flush_line(curbuf);
2591 return ml_append_int(curbuf, lnum, line, len, newfile, FALSE);
2592}
2593
Bram Moolenaar4033c552017-09-16 20:54:51 +02002594#if defined(FEAT_SPELL) || defined(FEAT_QUICKFIX) || defined(PROTO)
Bram Moolenaar4770d092006-01-12 23:22:24 +00002595/*
2596 * Like ml_append() but for an arbitrary buffer. The buffer must already have
2597 * a memline.
2598 */
2599 int
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01002600ml_append_buf(
2601 buf_T *buf,
2602 linenr_T lnum, /* append after this line (can be 0) */
2603 char_u *line, /* text of the new line */
2604 colnr_T len, /* length of new line, including NUL, or 0 */
2605 int newfile) /* flag, see above */
Bram Moolenaar4770d092006-01-12 23:22:24 +00002606{
2607 if (buf->b_ml.ml_mfp == NULL)
2608 return FAIL;
2609
2610 if (buf->b_ml.ml_line_lnum != 0)
2611 ml_flush_line(buf);
2612 return ml_append_int(buf, lnum, line, len, newfile, FALSE);
2613}
2614#endif
2615
Bram Moolenaar071d4272004-06-13 20:20:40 +00002616 static int
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01002617ml_append_int(
2618 buf_T *buf,
2619 linenr_T lnum, /* append after this line (can be 0) */
2620 char_u *line, /* text of the new line */
2621 colnr_T len, /* length of line, including NUL, or 0 */
2622 int newfile, /* flag, see above */
2623 int mark) /* mark the new line */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002624{
2625 int i;
2626 int line_count; /* number of indexes in current block */
2627 int offset;
2628 int from, to;
2629 int space_needed; /* space needed for new line */
2630 int page_size;
2631 int page_count;
2632 int db_idx; /* index for lnum in data block */
2633 bhdr_T *hp;
2634 memfile_T *mfp;
2635 DATA_BL *dp;
2636 PTR_BL *pp;
2637 infoptr_T *ip;
2638
2639 /* lnum out of range */
2640 if (lnum > buf->b_ml.ml_line_count || buf->b_ml.ml_mfp == NULL)
2641 return FAIL;
2642
2643 if (lowest_marked && lowest_marked > lnum)
2644 lowest_marked = lnum + 1;
2645
2646 if (len == 0)
2647 len = (colnr_T)STRLEN(line) + 1; /* space needed for the text */
2648 space_needed = len + INDEX_SIZE; /* space needed for text + index */
2649
2650 mfp = buf->b_ml.ml_mfp;
2651 page_size = mfp->mf_page_size;
2652
2653/*
2654 * find the data block containing the previous line
2655 * This also fills the stack with the blocks from the root to the data block
2656 * This also releases any locked block.
2657 */
2658 if ((hp = ml_find_line(buf, lnum == 0 ? (linenr_T)1 : lnum,
2659 ML_INSERT)) == NULL)
2660 return FAIL;
2661
2662 buf->b_ml.ml_flags &= ~ML_EMPTY;
2663
2664 if (lnum == 0) /* got line one instead, correct db_idx */
2665 db_idx = -1; /* careful, it is negative! */
2666 else
2667 db_idx = lnum - buf->b_ml.ml_locked_low;
2668 /* get line count before the insertion */
2669 line_count = buf->b_ml.ml_locked_high - buf->b_ml.ml_locked_low;
2670
2671 dp = (DATA_BL *)(hp->bh_data);
2672
2673/*
2674 * If
2675 * - there is not enough room in the current block
2676 * - appending to the last line in the block
2677 * - not appending to the last line in the file
2678 * insert in front of the next block.
2679 */
2680 if ((int)dp->db_free < space_needed && db_idx == line_count - 1
2681 && lnum < buf->b_ml.ml_line_count)
2682 {
2683 /*
2684 * Now that the line is not going to be inserted in the block that we
2685 * expected, the line count has to be adjusted in the pointer blocks
2686 * by using ml_locked_lineadd.
2687 */
2688 --(buf->b_ml.ml_locked_lineadd);
2689 --(buf->b_ml.ml_locked_high);
2690 if ((hp = ml_find_line(buf, lnum + 1, ML_INSERT)) == NULL)
2691 return FAIL;
2692
2693 db_idx = -1; /* careful, it is negative! */
2694 /* get line count before the insertion */
2695 line_count = buf->b_ml.ml_locked_high - buf->b_ml.ml_locked_low;
2696 CHECK(buf->b_ml.ml_locked_low != lnum + 1, "locked_low != lnum + 1");
2697
2698 dp = (DATA_BL *)(hp->bh_data);
2699 }
2700
2701 ++buf->b_ml.ml_line_count;
2702
2703 if ((int)dp->db_free >= space_needed) /* enough room in data block */
2704 {
2705/*
2706 * Insert new line in existing data block, or in data block allocated above.
2707 */
2708 dp->db_txt_start -= len;
2709 dp->db_free -= space_needed;
2710 ++(dp->db_line_count);
2711
2712 /*
2713 * move the text of the lines that follow to the front
2714 * adjust the indexes of the lines that follow
2715 */
2716 if (line_count > db_idx + 1) /* if there are following lines */
2717 {
2718 /*
2719 * Offset is the start of the previous line.
2720 * This will become the character just after the new line.
2721 */
2722 if (db_idx < 0)
2723 offset = dp->db_txt_end;
2724 else
2725 offset = ((dp->db_index[db_idx]) & DB_INDEX_MASK);
2726 mch_memmove((char *)dp + dp->db_txt_start,
2727 (char *)dp + dp->db_txt_start + len,
2728 (size_t)(offset - (dp->db_txt_start + len)));
2729 for (i = line_count - 1; i > db_idx; --i)
2730 dp->db_index[i + 1] = dp->db_index[i] - len;
2731 dp->db_index[db_idx + 1] = offset - len;
2732 }
2733 else /* add line at the end */
2734 dp->db_index[db_idx + 1] = dp->db_txt_start;
2735
2736 /*
2737 * copy the text into the block
2738 */
2739 mch_memmove((char *)dp + dp->db_index[db_idx + 1], line, (size_t)len);
2740 if (mark)
2741 dp->db_index[db_idx + 1] |= DB_MARKED;
2742
2743 /*
2744 * Mark the block dirty.
2745 */
2746 buf->b_ml.ml_flags |= ML_LOCKED_DIRTY;
2747 if (!newfile)
2748 buf->b_ml.ml_flags |= ML_LOCKED_POS;
2749 }
2750 else /* not enough space in data block */
2751 {
2752/*
2753 * If there is not enough room we have to create a new data block and copy some
2754 * lines into it.
2755 * Then we have to insert an entry in the pointer block.
2756 * If this pointer block also is full, we go up another block, and so on, up
2757 * to the root if necessary.
2758 * The line counts in the pointer blocks have already been adjusted by
2759 * ml_find_line().
2760 */
2761 long line_count_left, line_count_right;
2762 int page_count_left, page_count_right;
2763 bhdr_T *hp_left;
2764 bhdr_T *hp_right;
2765 bhdr_T *hp_new;
2766 int lines_moved;
2767 int data_moved = 0; /* init to shut up gcc */
2768 int total_moved = 0; /* init to shut up gcc */
2769 DATA_BL *dp_right, *dp_left;
2770 int stack_idx;
2771 int in_left;
2772 int lineadd;
2773 blocknr_T bnum_left, bnum_right;
2774 linenr_T lnum_left, lnum_right;
2775 int pb_idx;
2776 PTR_BL *pp_new;
2777
2778 /*
2779 * We are going to allocate a new data block. Depending on the
2780 * situation it will be put to the left or right of the existing
2781 * block. If possible we put the new line in the left block and move
2782 * the lines after it to the right block. Otherwise the new line is
2783 * also put in the right block. This method is more efficient when
2784 * inserting a lot of lines at one place.
2785 */
2786 if (db_idx < 0) /* left block is new, right block is existing */
2787 {
2788 lines_moved = 0;
2789 in_left = TRUE;
2790 /* space_needed does not change */
2791 }
2792 else /* left block is existing, right block is new */
2793 {
2794 lines_moved = line_count - db_idx - 1;
2795 if (lines_moved == 0)
2796 in_left = FALSE; /* put new line in right block */
2797 /* space_needed does not change */
2798 else
2799 {
2800 data_moved = ((dp->db_index[db_idx]) & DB_INDEX_MASK) -
2801 dp->db_txt_start;
2802 total_moved = data_moved + lines_moved * INDEX_SIZE;
2803 if ((int)dp->db_free + total_moved >= space_needed)
2804 {
2805 in_left = TRUE; /* put new line in left block */
2806 space_needed = total_moved;
2807 }
2808 else
2809 {
2810 in_left = FALSE; /* put new line in right block */
2811 space_needed += total_moved;
2812 }
2813 }
2814 }
2815
2816 page_count = ((space_needed + HEADER_SIZE) + page_size - 1) / page_size;
2817 if ((hp_new = ml_new_data(mfp, newfile, page_count)) == NULL)
2818 {
2819 /* correct line counts in pointer blocks */
2820 --(buf->b_ml.ml_locked_lineadd);
2821 --(buf->b_ml.ml_locked_high);
2822 return FAIL;
2823 }
2824 if (db_idx < 0) /* left block is new */
2825 {
2826 hp_left = hp_new;
2827 hp_right = hp;
2828 line_count_left = 0;
2829 line_count_right = line_count;
2830 }
2831 else /* right block is new */
2832 {
2833 hp_left = hp;
2834 hp_right = hp_new;
2835 line_count_left = line_count;
2836 line_count_right = 0;
2837 }
2838 dp_right = (DATA_BL *)(hp_right->bh_data);
2839 dp_left = (DATA_BL *)(hp_left->bh_data);
2840 bnum_left = hp_left->bh_bnum;
2841 bnum_right = hp_right->bh_bnum;
2842 page_count_left = hp_left->bh_page_count;
2843 page_count_right = hp_right->bh_page_count;
2844
2845 /*
2846 * May move the new line into the right/new block.
2847 */
2848 if (!in_left)
2849 {
2850 dp_right->db_txt_start -= len;
2851 dp_right->db_free -= len + INDEX_SIZE;
2852 dp_right->db_index[0] = dp_right->db_txt_start;
2853 if (mark)
2854 dp_right->db_index[0] |= DB_MARKED;
2855
2856 mch_memmove((char *)dp_right + dp_right->db_txt_start,
2857 line, (size_t)len);
2858 ++line_count_right;
2859 }
2860 /*
2861 * may move lines from the left/old block to the right/new one.
2862 */
2863 if (lines_moved)
2864 {
2865 /*
2866 */
2867 dp_right->db_txt_start -= data_moved;
2868 dp_right->db_free -= total_moved;
2869 mch_memmove((char *)dp_right + dp_right->db_txt_start,
2870 (char *)dp_left + dp_left->db_txt_start,
2871 (size_t)data_moved);
2872 offset = dp_right->db_txt_start - dp_left->db_txt_start;
2873 dp_left->db_txt_start += data_moved;
2874 dp_left->db_free += total_moved;
2875
2876 /*
2877 * update indexes in the new block
2878 */
2879 for (to = line_count_right, from = db_idx + 1;
2880 from < line_count_left; ++from, ++to)
2881 dp_right->db_index[to] = dp->db_index[from] + offset;
2882 line_count_right += lines_moved;
2883 line_count_left -= lines_moved;
2884 }
2885
2886 /*
2887 * May move the new line into the left (old or new) block.
2888 */
2889 if (in_left)
2890 {
2891 dp_left->db_txt_start -= len;
2892 dp_left->db_free -= len + INDEX_SIZE;
2893 dp_left->db_index[line_count_left] = dp_left->db_txt_start;
2894 if (mark)
2895 dp_left->db_index[line_count_left] |= DB_MARKED;
2896 mch_memmove((char *)dp_left + dp_left->db_txt_start,
2897 line, (size_t)len);
2898 ++line_count_left;
2899 }
2900
2901 if (db_idx < 0) /* left block is new */
2902 {
2903 lnum_left = lnum + 1;
2904 lnum_right = 0;
2905 }
2906 else /* right block is new */
2907 {
2908 lnum_left = 0;
2909 if (in_left)
2910 lnum_right = lnum + 2;
2911 else
2912 lnum_right = lnum + 1;
2913 }
2914 dp_left->db_line_count = line_count_left;
2915 dp_right->db_line_count = line_count_right;
2916
2917 /*
2918 * release the two data blocks
2919 * The new one (hp_new) already has a correct blocknumber.
2920 * The old one (hp, in ml_locked) gets a positive blocknumber if
2921 * we changed it and we are not editing a new file.
2922 */
2923 if (lines_moved || in_left)
2924 buf->b_ml.ml_flags |= ML_LOCKED_DIRTY;
2925 if (!newfile && db_idx >= 0 && in_left)
2926 buf->b_ml.ml_flags |= ML_LOCKED_POS;
2927 mf_put(mfp, hp_new, TRUE, FALSE);
2928
2929 /*
2930 * flush the old data block
2931 * set ml_locked_lineadd to 0, because the updating of the
2932 * pointer blocks is done below
2933 */
2934 lineadd = buf->b_ml.ml_locked_lineadd;
2935 buf->b_ml.ml_locked_lineadd = 0;
2936 ml_find_line(buf, (linenr_T)0, ML_FLUSH); /* flush data block */
2937
2938 /*
2939 * update pointer blocks for the new data block
2940 */
2941 for (stack_idx = buf->b_ml.ml_stack_top - 1; stack_idx >= 0;
2942 --stack_idx)
2943 {
2944 ip = &(buf->b_ml.ml_stack[stack_idx]);
2945 pb_idx = ip->ip_index;
2946 if ((hp = mf_get(mfp, ip->ip_bnum, 1)) == NULL)
2947 return FAIL;
2948 pp = (PTR_BL *)(hp->bh_data); /* must be pointer block */
2949 if (pp->pb_id != PTR_ID)
2950 {
Bram Moolenaar95f09602016-11-10 20:01:45 +01002951 IEMSG(_("E317: pointer block id wrong 3"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00002952 mf_put(mfp, hp, FALSE, FALSE);
2953 return FAIL;
2954 }
2955 /*
2956 * TODO: If the pointer block is full and we are adding at the end
2957 * try to insert in front of the next block
2958 */
2959 /* block not full, add one entry */
2960 if (pp->pb_count < pp->pb_count_max)
2961 {
2962 if (pb_idx + 1 < (int)pp->pb_count)
2963 mch_memmove(&pp->pb_pointer[pb_idx + 2],
2964 &pp->pb_pointer[pb_idx + 1],
2965 (size_t)(pp->pb_count - pb_idx - 1) * sizeof(PTR_EN));
2966 ++pp->pb_count;
2967 pp->pb_pointer[pb_idx].pe_line_count = line_count_left;
2968 pp->pb_pointer[pb_idx].pe_bnum = bnum_left;
2969 pp->pb_pointer[pb_idx].pe_page_count = page_count_left;
2970 pp->pb_pointer[pb_idx + 1].pe_line_count = line_count_right;
2971 pp->pb_pointer[pb_idx + 1].pe_bnum = bnum_right;
2972 pp->pb_pointer[pb_idx + 1].pe_page_count = page_count_right;
2973
2974 if (lnum_left != 0)
2975 pp->pb_pointer[pb_idx].pe_old_lnum = lnum_left;
2976 if (lnum_right != 0)
2977 pp->pb_pointer[pb_idx + 1].pe_old_lnum = lnum_right;
2978
2979 mf_put(mfp, hp, TRUE, FALSE);
2980 buf->b_ml.ml_stack_top = stack_idx + 1; /* truncate stack */
2981
2982 if (lineadd)
2983 {
2984 --(buf->b_ml.ml_stack_top);
Bram Moolenaar6b803a72007-05-06 14:25:46 +00002985 /* fix line count for rest of blocks in the stack */
Bram Moolenaar071d4272004-06-13 20:20:40 +00002986 ml_lineadd(buf, lineadd);
2987 /* fix stack itself */
2988 buf->b_ml.ml_stack[buf->b_ml.ml_stack_top].ip_high +=
2989 lineadd;
2990 ++(buf->b_ml.ml_stack_top);
2991 }
2992
2993 /*
2994 * We are finished, break the loop here.
2995 */
2996 break;
2997 }
2998 else /* pointer block full */
2999 {
3000 /*
3001 * split the pointer block
3002 * allocate a new pointer block
3003 * move some of the pointer into the new block
3004 * prepare for updating the parent block
3005 */
3006 for (;;) /* do this twice when splitting block 1 */
3007 {
3008 hp_new = ml_new_ptr(mfp);
3009 if (hp_new == NULL) /* TODO: try to fix tree */
3010 return FAIL;
3011 pp_new = (PTR_BL *)(hp_new->bh_data);
3012
3013 if (hp->bh_bnum != 1)
3014 break;
3015
3016 /*
3017 * if block 1 becomes full the tree is given an extra level
3018 * The pointers from block 1 are moved into the new block.
3019 * block 1 is updated to point to the new block
3020 * then continue to split the new block
3021 */
3022 mch_memmove(pp_new, pp, (size_t)page_size);
3023 pp->pb_count = 1;
3024 pp->pb_pointer[0].pe_bnum = hp_new->bh_bnum;
3025 pp->pb_pointer[0].pe_line_count = buf->b_ml.ml_line_count;
3026 pp->pb_pointer[0].pe_old_lnum = 1;
3027 pp->pb_pointer[0].pe_page_count = 1;
3028 mf_put(mfp, hp, TRUE, FALSE); /* release block 1 */
3029 hp = hp_new; /* new block is to be split */
3030 pp = pp_new;
3031 CHECK(stack_idx != 0, _("stack_idx should be 0"));
3032 ip->ip_index = 0;
3033 ++stack_idx; /* do block 1 again later */
3034 }
3035 /*
3036 * move the pointers after the current one to the new block
3037 * If there are none, the new entry will be in the new block.
3038 */
3039 total_moved = pp->pb_count - pb_idx - 1;
3040 if (total_moved)
3041 {
3042 mch_memmove(&pp_new->pb_pointer[0],
3043 &pp->pb_pointer[pb_idx + 1],
3044 (size_t)(total_moved) * sizeof(PTR_EN));
3045 pp_new->pb_count = total_moved;
3046 pp->pb_count -= total_moved - 1;
3047 pp->pb_pointer[pb_idx + 1].pe_bnum = bnum_right;
3048 pp->pb_pointer[pb_idx + 1].pe_line_count = line_count_right;
3049 pp->pb_pointer[pb_idx + 1].pe_page_count = page_count_right;
3050 if (lnum_right)
3051 pp->pb_pointer[pb_idx + 1].pe_old_lnum = lnum_right;
3052 }
3053 else
3054 {
3055 pp_new->pb_count = 1;
3056 pp_new->pb_pointer[0].pe_bnum = bnum_right;
3057 pp_new->pb_pointer[0].pe_line_count = line_count_right;
3058 pp_new->pb_pointer[0].pe_page_count = page_count_right;
3059 pp_new->pb_pointer[0].pe_old_lnum = lnum_right;
3060 }
3061 pp->pb_pointer[pb_idx].pe_bnum = bnum_left;
3062 pp->pb_pointer[pb_idx].pe_line_count = line_count_left;
3063 pp->pb_pointer[pb_idx].pe_page_count = page_count_left;
3064 if (lnum_left)
3065 pp->pb_pointer[pb_idx].pe_old_lnum = lnum_left;
3066 lnum_left = 0;
3067 lnum_right = 0;
3068
3069 /*
3070 * recompute line counts
3071 */
3072 line_count_right = 0;
3073 for (i = 0; i < (int)pp_new->pb_count; ++i)
3074 line_count_right += pp_new->pb_pointer[i].pe_line_count;
3075 line_count_left = 0;
3076 for (i = 0; i < (int)pp->pb_count; ++i)
3077 line_count_left += pp->pb_pointer[i].pe_line_count;
3078
3079 bnum_left = hp->bh_bnum;
3080 bnum_right = hp_new->bh_bnum;
3081 page_count_left = 1;
3082 page_count_right = 1;
3083 mf_put(mfp, hp, TRUE, FALSE);
3084 mf_put(mfp, hp_new, TRUE, FALSE);
3085 }
3086 }
3087
3088 /*
3089 * Safety check: fallen out of for loop?
3090 */
3091 if (stack_idx < 0)
3092 {
Bram Moolenaar95f09602016-11-10 20:01:45 +01003093 IEMSG(_("E318: Updated too many blocks?"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003094 buf->b_ml.ml_stack_top = 0; /* invalidate stack */
3095 }
3096 }
3097
3098#ifdef FEAT_BYTEOFF
3099 /* The line was inserted below 'lnum' */
3100 ml_updatechunk(buf, lnum + 1, (long)len, ML_CHNK_ADDLINE);
3101#endif
3102#ifdef FEAT_NETBEANS_INTG
Bram Moolenaarb26e6322010-05-22 21:34:09 +02003103 if (netbeans_active())
Bram Moolenaar071d4272004-06-13 20:20:40 +00003104 {
3105 if (STRLEN(line) > 0)
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00003106 netbeans_inserted(buf, lnum+1, (colnr_T)0, line, (int)STRLEN(line));
Bram Moolenaar35a9aaa2004-10-24 19:23:07 +00003107 netbeans_inserted(buf, lnum+1, (colnr_T)STRLEN(line),
Bram Moolenaar071d4272004-06-13 20:20:40 +00003108 (char_u *)"\n", 1);
3109 }
3110#endif
Bram Moolenaar509ce2a2016-03-11 22:52:15 +01003111#ifdef FEAT_JOB_CHANNEL
Bram Moolenaar99ef0622016-03-06 20:22:25 +01003112 if (buf->b_write_to_channel)
3113 channel_write_new_lines(buf);
3114#endif
3115
Bram Moolenaar071d4272004-06-13 20:20:40 +00003116 return OK;
3117}
3118
3119/*
Bram Moolenaar4770d092006-01-12 23:22:24 +00003120 * Replace line lnum, with buffering, in current buffer.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003121 *
Bram Moolenaar1056d982006-03-09 22:37:52 +00003122 * If "copy" is TRUE, make a copy of the line, otherwise the line has been
Bram Moolenaar071d4272004-06-13 20:20:40 +00003123 * copied to allocated memory already.
3124 *
3125 * Check: The caller of this function should probably also call
3126 * changed_lines(), unless update_screen(NOT_VALID) is used.
3127 *
3128 * return FAIL for failure, OK otherwise
3129 */
3130 int
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01003131ml_replace(linenr_T lnum, char_u *line, int copy)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003132{
3133 if (line == NULL) /* just checking... */
3134 return FAIL;
3135
3136 /* When starting up, we might still need to create the memfile */
Bram Moolenaar59f931e2010-07-24 20:27:03 +02003137 if (curbuf->b_ml.ml_mfp == NULL && open_buffer(FALSE, NULL, 0) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003138 return FAIL;
3139
3140 if (copy && (line = vim_strsave(line)) == NULL) /* allocate memory */
3141 return FAIL;
3142#ifdef FEAT_NETBEANS_INTG
Bram Moolenaarb26e6322010-05-22 21:34:09 +02003143 if (netbeans_active())
Bram Moolenaar071d4272004-06-13 20:20:40 +00003144 {
3145 netbeans_removed(curbuf, lnum, 0, (long)STRLEN(ml_get(lnum)));
Bram Moolenaara93fa7e2006-04-17 22:14:47 +00003146 netbeans_inserted(curbuf, lnum, 0, line, (int)STRLEN(line));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003147 }
3148#endif
3149 if (curbuf->b_ml.ml_line_lnum != lnum) /* other line buffered */
3150 ml_flush_line(curbuf); /* flush it */
3151 else if (curbuf->b_ml.ml_flags & ML_LINE_DIRTY) /* same line allocated */
3152 vim_free(curbuf->b_ml.ml_line_ptr); /* free it */
3153 curbuf->b_ml.ml_line_ptr = line;
3154 curbuf->b_ml.ml_line_lnum = lnum;
3155 curbuf->b_ml.ml_flags = (curbuf->b_ml.ml_flags | ML_LINE_DIRTY) & ~ML_EMPTY;
3156
3157 return OK;
3158}
3159
3160/*
Bram Moolenaar4033c552017-09-16 20:54:51 +02003161 * Delete line "lnum" in the current buffer.
3162 * When "message" is TRUE may give a "No lines in buffer" message.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003163 *
3164 * Check: The caller of this function should probably also call
3165 * deleted_lines() after this.
3166 *
3167 * return FAIL for failure, OK otherwise
3168 */
3169 int
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01003170ml_delete(linenr_T lnum, int message)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003171{
3172 ml_flush_line(curbuf);
3173 return ml_delete_int(curbuf, lnum, message);
3174}
3175
3176 static int
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01003177ml_delete_int(buf_T *buf, linenr_T lnum, int message)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003178{
3179 bhdr_T *hp;
3180 memfile_T *mfp;
3181 DATA_BL *dp;
3182 PTR_BL *pp;
3183 infoptr_T *ip;
3184 int count; /* number of entries in block */
3185 int idx;
3186 int stack_idx;
3187 int text_start;
3188 int line_start;
3189 long line_size;
3190 int i;
3191
3192 if (lnum < 1 || lnum > buf->b_ml.ml_line_count)
3193 return FAIL;
3194
3195 if (lowest_marked && lowest_marked > lnum)
3196 lowest_marked--;
3197
3198/*
3199 * If the file becomes empty the last line is replaced by an empty line.
3200 */
3201 if (buf->b_ml.ml_line_count == 1) /* file becomes empty */
3202 {
3203 if (message
3204#ifdef FEAT_NETBEANS_INTG
3205 && !netbeansSuppressNoLines
3206#endif
3207 )
Bram Moolenaar238a5642006-02-21 22:12:05 +00003208 set_keep_msg((char_u *)_(no_lines_msg), 0);
3209
Bram Moolenaar84a05ac2013-05-06 04:24:17 +02003210 /* FEAT_BYTEOFF already handled in there, don't worry 'bout it below */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003211 i = ml_replace((linenr_T)1, (char_u *)"", TRUE);
3212 buf->b_ml.ml_flags |= ML_EMPTY;
3213
3214 return i;
3215 }
3216
3217/*
3218 * find the data block containing the line
3219 * This also fills the stack with the blocks from the root to the data block
3220 * This also releases any locked block.
3221 */
3222 mfp = buf->b_ml.ml_mfp;
3223 if (mfp == NULL)
3224 return FAIL;
3225
3226 if ((hp = ml_find_line(buf, lnum, ML_DELETE)) == NULL)
3227 return FAIL;
3228
3229 dp = (DATA_BL *)(hp->bh_data);
3230 /* compute line count before the delete */
3231 count = (long)(buf->b_ml.ml_locked_high)
3232 - (long)(buf->b_ml.ml_locked_low) + 2;
3233 idx = lnum - buf->b_ml.ml_locked_low;
3234
3235 --buf->b_ml.ml_line_count;
3236
3237 line_start = ((dp->db_index[idx]) & DB_INDEX_MASK);
3238 if (idx == 0) /* first line in block, text at the end */
3239 line_size = dp->db_txt_end - line_start;
3240 else
3241 line_size = ((dp->db_index[idx - 1]) & DB_INDEX_MASK) - line_start;
3242
3243#ifdef FEAT_NETBEANS_INTG
Bram Moolenaarb26e6322010-05-22 21:34:09 +02003244 if (netbeans_active())
Bram Moolenaar35a9aaa2004-10-24 19:23:07 +00003245 netbeans_removed(buf, lnum, 0, (long)line_size);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003246#endif
3247
3248/*
3249 * special case: If there is only one line in the data block it becomes empty.
3250 * Then we have to remove the entry, pointing to this data block, from the
3251 * pointer block. If this pointer block also becomes empty, we go up another
3252 * block, and so on, up to the root if necessary.
3253 * The line counts in the pointer blocks have already been adjusted by
3254 * ml_find_line().
3255 */
3256 if (count == 1)
3257 {
3258 mf_free(mfp, hp); /* free the data block */
3259 buf->b_ml.ml_locked = NULL;
3260
Bram Moolenaare60acc12011-05-10 16:41:25 +02003261 for (stack_idx = buf->b_ml.ml_stack_top - 1; stack_idx >= 0;
3262 --stack_idx)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003263 {
3264 buf->b_ml.ml_stack_top = 0; /* stack is invalid when failing */
3265 ip = &(buf->b_ml.ml_stack[stack_idx]);
3266 idx = ip->ip_index;
3267 if ((hp = mf_get(mfp, ip->ip_bnum, 1)) == NULL)
3268 return FAIL;
3269 pp = (PTR_BL *)(hp->bh_data); /* must be pointer block */
3270 if (pp->pb_id != PTR_ID)
3271 {
Bram Moolenaar95f09602016-11-10 20:01:45 +01003272 IEMSG(_("E317: pointer block id wrong 4"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003273 mf_put(mfp, hp, FALSE, FALSE);
3274 return FAIL;
3275 }
3276 count = --(pp->pb_count);
3277 if (count == 0) /* the pointer block becomes empty! */
3278 mf_free(mfp, hp);
3279 else
3280 {
3281 if (count != idx) /* move entries after the deleted one */
3282 mch_memmove(&pp->pb_pointer[idx], &pp->pb_pointer[idx + 1],
3283 (size_t)(count - idx) * sizeof(PTR_EN));
3284 mf_put(mfp, hp, TRUE, FALSE);
3285
3286 buf->b_ml.ml_stack_top = stack_idx; /* truncate stack */
Bram Moolenaar6b803a72007-05-06 14:25:46 +00003287 /* fix line count for rest of blocks in the stack */
3288 if (buf->b_ml.ml_locked_lineadd != 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003289 {
3290 ml_lineadd(buf, buf->b_ml.ml_locked_lineadd);
3291 buf->b_ml.ml_stack[buf->b_ml.ml_stack_top].ip_high +=
Bram Moolenaar6b803a72007-05-06 14:25:46 +00003292 buf->b_ml.ml_locked_lineadd;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003293 }
3294 ++(buf->b_ml.ml_stack_top);
3295
3296 break;
3297 }
3298 }
3299 CHECK(stack_idx < 0, _("deleted block 1?"));
3300 }
3301 else
3302 {
3303 /*
3304 * delete the text by moving the next lines forwards
3305 */
3306 text_start = dp->db_txt_start;
3307 mch_memmove((char *)dp + text_start + line_size,
3308 (char *)dp + text_start, (size_t)(line_start - text_start));
3309
3310 /*
3311 * delete the index by moving the next indexes backwards
3312 * Adjust the indexes for the text movement.
3313 */
3314 for (i = idx; i < count - 1; ++i)
3315 dp->db_index[i] = dp->db_index[i + 1] + line_size;
3316
3317 dp->db_free += line_size + INDEX_SIZE;
3318 dp->db_txt_start += line_size;
3319 --(dp->db_line_count);
3320
3321 /*
3322 * mark the block dirty and make sure it is in the file (for recovery)
3323 */
3324 buf->b_ml.ml_flags |= (ML_LOCKED_DIRTY | ML_LOCKED_POS);
3325 }
3326
3327#ifdef FEAT_BYTEOFF
3328 ml_updatechunk(buf, lnum, line_size, ML_CHNK_DELLINE);
3329#endif
3330 return OK;
3331}
3332
3333/*
3334 * set the B_MARKED flag for line 'lnum'
3335 */
3336 void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01003337ml_setmarked(linenr_T lnum)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003338{
3339 bhdr_T *hp;
3340 DATA_BL *dp;
3341 /* invalid line number */
3342 if (lnum < 1 || lnum > curbuf->b_ml.ml_line_count
3343 || curbuf->b_ml.ml_mfp == NULL)
3344 return; /* give error message? */
3345
3346 if (lowest_marked == 0 || lowest_marked > lnum)
3347 lowest_marked = lnum;
3348
3349 /*
3350 * find the data block containing the line
3351 * This also fills the stack with the blocks from the root to the data block
3352 * This also releases any locked block.
3353 */
3354 if ((hp = ml_find_line(curbuf, lnum, ML_FIND)) == NULL)
3355 return; /* give error message? */
3356
3357 dp = (DATA_BL *)(hp->bh_data);
3358 dp->db_index[lnum - curbuf->b_ml.ml_locked_low] |= DB_MARKED;
3359 curbuf->b_ml.ml_flags |= ML_LOCKED_DIRTY;
3360}
3361
3362/*
3363 * find the first line with its B_MARKED flag set
3364 */
3365 linenr_T
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01003366ml_firstmarked(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003367{
3368 bhdr_T *hp;
3369 DATA_BL *dp;
3370 linenr_T lnum;
3371 int i;
3372
3373 if (curbuf->b_ml.ml_mfp == NULL)
3374 return (linenr_T) 0;
3375
3376 /*
3377 * The search starts with lowest_marked line. This is the last line where
3378 * a mark was found, adjusted by inserting/deleting lines.
3379 */
3380 for (lnum = lowest_marked; lnum <= curbuf->b_ml.ml_line_count; )
3381 {
3382 /*
3383 * Find the data block containing the line.
3384 * This also fills the stack with the blocks from the root to the data
3385 * block This also releases any locked block.
3386 */
3387 if ((hp = ml_find_line(curbuf, lnum, ML_FIND)) == NULL)
3388 return (linenr_T)0; /* give error message? */
3389
3390 dp = (DATA_BL *)(hp->bh_data);
3391
3392 for (i = lnum - curbuf->b_ml.ml_locked_low;
3393 lnum <= curbuf->b_ml.ml_locked_high; ++i, ++lnum)
3394 if ((dp->db_index[i]) & DB_MARKED)
3395 {
3396 (dp->db_index[i]) &= DB_INDEX_MASK;
3397 curbuf->b_ml.ml_flags |= ML_LOCKED_DIRTY;
3398 lowest_marked = lnum + 1;
3399 return lnum;
3400 }
3401 }
3402
3403 return (linenr_T) 0;
3404}
3405
Bram Moolenaar071d4272004-06-13 20:20:40 +00003406/*
3407 * clear all DB_MARKED flags
3408 */
3409 void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01003410ml_clearmarked(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003411{
3412 bhdr_T *hp;
3413 DATA_BL *dp;
3414 linenr_T lnum;
3415 int i;
3416
3417 if (curbuf->b_ml.ml_mfp == NULL) /* nothing to do */
3418 return;
3419
3420 /*
3421 * The search starts with line lowest_marked.
3422 */
3423 for (lnum = lowest_marked; lnum <= curbuf->b_ml.ml_line_count; )
3424 {
3425 /*
3426 * Find the data block containing the line.
3427 * This also fills the stack with the blocks from the root to the data
3428 * block and releases any locked block.
3429 */
3430 if ((hp = ml_find_line(curbuf, lnum, ML_FIND)) == NULL)
3431 return; /* give error message? */
3432
3433 dp = (DATA_BL *)(hp->bh_data);
3434
3435 for (i = lnum - curbuf->b_ml.ml_locked_low;
3436 lnum <= curbuf->b_ml.ml_locked_high; ++i, ++lnum)
3437 if ((dp->db_index[i]) & DB_MARKED)
3438 {
3439 (dp->db_index[i]) &= DB_INDEX_MASK;
3440 curbuf->b_ml.ml_flags |= ML_LOCKED_DIRTY;
3441 }
3442 }
3443
3444 lowest_marked = 0;
3445 return;
3446}
3447
3448/*
3449 * flush ml_line if necessary
3450 */
3451 static void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01003452ml_flush_line(buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003453{
3454 bhdr_T *hp;
3455 DATA_BL *dp;
3456 linenr_T lnum;
3457 char_u *new_line;
3458 char_u *old_line;
3459 colnr_T new_len;
3460 int old_len;
3461 int extra;
3462 int idx;
3463 int start;
3464 int count;
3465 int i;
Bram Moolenaar0ca4b352010-02-11 18:54:43 +01003466 static int entered = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003467
3468 if (buf->b_ml.ml_line_lnum == 0 || buf->b_ml.ml_mfp == NULL)
3469 return; /* nothing to do */
3470
3471 if (buf->b_ml.ml_flags & ML_LINE_DIRTY)
3472 {
Bram Moolenaar0ca4b352010-02-11 18:54:43 +01003473 /* This code doesn't work recursively, but Netbeans may call back here
3474 * when obtaining the cursor position. */
3475 if (entered)
3476 return;
3477 entered = TRUE;
3478
Bram Moolenaar071d4272004-06-13 20:20:40 +00003479 lnum = buf->b_ml.ml_line_lnum;
3480 new_line = buf->b_ml.ml_line_ptr;
3481
3482 hp = ml_find_line(buf, lnum, ML_FIND);
3483 if (hp == NULL)
Bram Moolenaar95f09602016-11-10 20:01:45 +01003484 IEMSGN(_("E320: Cannot find line %ld"), lnum);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003485 else
3486 {
3487 dp = (DATA_BL *)(hp->bh_data);
3488 idx = lnum - buf->b_ml.ml_locked_low;
3489 start = ((dp->db_index[idx]) & DB_INDEX_MASK);
3490 old_line = (char_u *)dp + start;
3491 if (idx == 0) /* line is last in block */
3492 old_len = dp->db_txt_end - start;
3493 else /* text of previous line follows */
3494 old_len = (dp->db_index[idx - 1] & DB_INDEX_MASK) - start;
3495 new_len = (colnr_T)STRLEN(new_line) + 1;
3496 extra = new_len - old_len; /* negative if lines gets smaller */
3497
3498 /*
3499 * if new line fits in data block, replace directly
3500 */
3501 if ((int)dp->db_free >= extra)
3502 {
3503 /* if the length changes and there are following lines */
3504 count = buf->b_ml.ml_locked_high - buf->b_ml.ml_locked_low + 1;
3505 if (extra != 0 && idx < count - 1)
3506 {
3507 /* move text of following lines */
3508 mch_memmove((char *)dp + dp->db_txt_start - extra,
3509 (char *)dp + dp->db_txt_start,
3510 (size_t)(start - dp->db_txt_start));
3511
3512 /* adjust pointers of this and following lines */
3513 for (i = idx + 1; i < count; ++i)
3514 dp->db_index[i] -= extra;
3515 }
3516 dp->db_index[idx] -= extra;
3517
3518 /* adjust free space */
3519 dp->db_free -= extra;
3520 dp->db_txt_start -= extra;
3521
3522 /* copy new line into the data block */
3523 mch_memmove(old_line - extra, new_line, (size_t)new_len);
3524 buf->b_ml.ml_flags |= (ML_LOCKED_DIRTY | ML_LOCKED_POS);
3525#ifdef FEAT_BYTEOFF
3526 /* The else case is already covered by the insert and delete */
3527 ml_updatechunk(buf, lnum, (long)extra, ML_CHNK_UPDLINE);
3528#endif
3529 }
3530 else
3531 {
3532 /*
3533 * Cannot do it in one data block: Delete and append.
3534 * Append first, because ml_delete_int() cannot delete the
3535 * last line in a buffer, which causes trouble for a buffer
3536 * that has only one line.
3537 * Don't forget to copy the mark!
3538 */
3539 /* How about handling errors??? */
3540 (void)ml_append_int(buf, lnum, new_line, new_len, FALSE,
3541 (dp->db_index[idx] & DB_MARKED));
3542 (void)ml_delete_int(buf, lnum, FALSE);
3543 }
3544 }
3545 vim_free(new_line);
Bram Moolenaar0ca4b352010-02-11 18:54:43 +01003546
3547 entered = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00003548 }
3549
3550 buf->b_ml.ml_line_lnum = 0;
3551}
3552
3553/*
3554 * create a new, empty, data block
3555 */
3556 static bhdr_T *
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01003557ml_new_data(memfile_T *mfp, int negative, int page_count)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003558{
3559 bhdr_T *hp;
3560 DATA_BL *dp;
3561
3562 if ((hp = mf_new(mfp, negative, page_count)) == NULL)
3563 return NULL;
3564
3565 dp = (DATA_BL *)(hp->bh_data);
3566 dp->db_id = DATA_ID;
3567 dp->db_txt_start = dp->db_txt_end = page_count * mfp->mf_page_size;
3568 dp->db_free = dp->db_txt_start - HEADER_SIZE;
3569 dp->db_line_count = 0;
3570
3571 return hp;
3572}
3573
3574/*
3575 * create a new, empty, pointer block
3576 */
3577 static bhdr_T *
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01003578ml_new_ptr(memfile_T *mfp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003579{
3580 bhdr_T *hp;
3581 PTR_BL *pp;
3582
3583 if ((hp = mf_new(mfp, FALSE, 1)) == NULL)
3584 return NULL;
3585
3586 pp = (PTR_BL *)(hp->bh_data);
3587 pp->pb_id = PTR_ID;
3588 pp->pb_count = 0;
Bram Moolenaar20a825a2010-05-31 21:27:30 +02003589 pp->pb_count_max = (short_u)((mfp->mf_page_size - sizeof(PTR_BL))
3590 / sizeof(PTR_EN) + 1);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003591
3592 return hp;
3593}
3594
3595/*
3596 * lookup line 'lnum' in a memline
3597 *
3598 * action: if ML_DELETE or ML_INSERT the line count is updated while searching
3599 * if ML_FLUSH only flush a locked block
3600 * if ML_FIND just find the line
3601 *
3602 * If the block was found it is locked and put in ml_locked.
3603 * The stack is updated to lead to the locked block. The ip_high field in
3604 * the stack is updated to reflect the last line in the block AFTER the
3605 * insert or delete, also if the pointer block has not been updated yet. But
Bram Moolenaar6b803a72007-05-06 14:25:46 +00003606 * if ml_locked != NULL ml_locked_lineadd must be added to ip_high.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003607 *
3608 * return: NULL for failure, pointer to block header otherwise
3609 */
3610 static bhdr_T *
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01003611ml_find_line(buf_T *buf, linenr_T lnum, int action)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003612{
3613 DATA_BL *dp;
3614 PTR_BL *pp;
3615 infoptr_T *ip;
3616 bhdr_T *hp;
3617 memfile_T *mfp;
3618 linenr_T t;
3619 blocknr_T bnum, bnum2;
3620 int dirty;
3621 linenr_T low, high;
3622 int top;
3623 int page_count;
3624 int idx;
3625
3626 mfp = buf->b_ml.ml_mfp;
3627
3628 /*
3629 * If there is a locked block check if the wanted line is in it.
3630 * If not, flush and release the locked block.
3631 * Don't do this for ML_INSERT_SAME, because the stack need to be updated.
3632 * Don't do this for ML_FLUSH, because we want to flush the locked block.
Bram Moolenaar47b8b152007-02-07 02:41:57 +00003633 * Don't do this when 'swapfile' is reset, we want to load all the blocks.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003634 */
3635 if (buf->b_ml.ml_locked)
3636 {
Bram Moolenaar47b8b152007-02-07 02:41:57 +00003637 if (ML_SIMPLE(action)
3638 && buf->b_ml.ml_locked_low <= lnum
3639 && buf->b_ml.ml_locked_high >= lnum
3640 && !mf_dont_release)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003641 {
Bram Moolenaar47b8b152007-02-07 02:41:57 +00003642 /* remember to update pointer blocks and stack later */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003643 if (action == ML_INSERT)
3644 {
3645 ++(buf->b_ml.ml_locked_lineadd);
3646 ++(buf->b_ml.ml_locked_high);
3647 }
3648 else if (action == ML_DELETE)
3649 {
3650 --(buf->b_ml.ml_locked_lineadd);
3651 --(buf->b_ml.ml_locked_high);
3652 }
3653 return (buf->b_ml.ml_locked);
3654 }
3655
3656 mf_put(mfp, buf->b_ml.ml_locked, buf->b_ml.ml_flags & ML_LOCKED_DIRTY,
3657 buf->b_ml.ml_flags & ML_LOCKED_POS);
3658 buf->b_ml.ml_locked = NULL;
3659
Bram Moolenaar6b803a72007-05-06 14:25:46 +00003660 /*
3661 * If lines have been added or deleted in the locked block, need to
3662 * update the line count in pointer blocks.
3663 */
3664 if (buf->b_ml.ml_locked_lineadd != 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003665 ml_lineadd(buf, buf->b_ml.ml_locked_lineadd);
3666 }
3667
3668 if (action == ML_FLUSH) /* nothing else to do */
3669 return NULL;
3670
3671 bnum = 1; /* start at the root of the tree */
3672 page_count = 1;
3673 low = 1;
3674 high = buf->b_ml.ml_line_count;
3675
3676 if (action == ML_FIND) /* first try stack entries */
3677 {
3678 for (top = buf->b_ml.ml_stack_top - 1; top >= 0; --top)
3679 {
3680 ip = &(buf->b_ml.ml_stack[top]);
3681 if (ip->ip_low <= lnum && ip->ip_high >= lnum)
3682 {
3683 bnum = ip->ip_bnum;
3684 low = ip->ip_low;
3685 high = ip->ip_high;
3686 buf->b_ml.ml_stack_top = top; /* truncate stack at prev entry */
3687 break;
3688 }
3689 }
3690 if (top < 0)
3691 buf->b_ml.ml_stack_top = 0; /* not found, start at the root */
3692 }
3693 else /* ML_DELETE or ML_INSERT */
3694 buf->b_ml.ml_stack_top = 0; /* start at the root */
3695
3696/*
3697 * search downwards in the tree until a data block is found
3698 */
3699 for (;;)
3700 {
3701 if ((hp = mf_get(mfp, bnum, page_count)) == NULL)
3702 goto error_noblock;
3703
3704 /*
3705 * update high for insert/delete
3706 */
3707 if (action == ML_INSERT)
3708 ++high;
3709 else if (action == ML_DELETE)
3710 --high;
3711
3712 dp = (DATA_BL *)(hp->bh_data);
3713 if (dp->db_id == DATA_ID) /* data block */
3714 {
3715 buf->b_ml.ml_locked = hp;
3716 buf->b_ml.ml_locked_low = low;
3717 buf->b_ml.ml_locked_high = high;
3718 buf->b_ml.ml_locked_lineadd = 0;
3719 buf->b_ml.ml_flags &= ~(ML_LOCKED_DIRTY | ML_LOCKED_POS);
3720 return hp;
3721 }
3722
3723 pp = (PTR_BL *)(dp); /* must be pointer block */
3724 if (pp->pb_id != PTR_ID)
3725 {
Bram Moolenaar95f09602016-11-10 20:01:45 +01003726 IEMSG(_("E317: pointer block id wrong"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003727 goto error_block;
3728 }
3729
3730 if ((top = ml_add_stack(buf)) < 0) /* add new entry to stack */
3731 goto error_block;
3732 ip = &(buf->b_ml.ml_stack[top]);
3733 ip->ip_bnum = bnum;
3734 ip->ip_low = low;
3735 ip->ip_high = high;
3736 ip->ip_index = -1; /* index not known yet */
3737
3738 dirty = FALSE;
3739 for (idx = 0; idx < (int)pp->pb_count; ++idx)
3740 {
3741 t = pp->pb_pointer[idx].pe_line_count;
3742 CHECK(t == 0, _("pe_line_count is zero"));
3743 if ((low += t) > lnum)
3744 {
3745 ip->ip_index = idx;
3746 bnum = pp->pb_pointer[idx].pe_bnum;
3747 page_count = pp->pb_pointer[idx].pe_page_count;
3748 high = low - 1;
3749 low -= t;
3750
3751 /*
3752 * a negative block number may have been changed
3753 */
3754 if (bnum < 0)
3755 {
3756 bnum2 = mf_trans_del(mfp, bnum);
3757 if (bnum != bnum2)
3758 {
3759 bnum = bnum2;
3760 pp->pb_pointer[idx].pe_bnum = bnum;
3761 dirty = TRUE;
3762 }
3763 }
3764
3765 break;
3766 }
3767 }
3768 if (idx >= (int)pp->pb_count) /* past the end: something wrong! */
3769 {
3770 if (lnum > buf->b_ml.ml_line_count)
Bram Moolenaar95f09602016-11-10 20:01:45 +01003771 IEMSGN(_("E322: line number out of range: %ld past the end"),
Bram Moolenaar071d4272004-06-13 20:20:40 +00003772 lnum - buf->b_ml.ml_line_count);
3773
3774 else
Bram Moolenaar95f09602016-11-10 20:01:45 +01003775 IEMSGN(_("E323: line count wrong in block %ld"), bnum);
Bram Moolenaar071d4272004-06-13 20:20:40 +00003776 goto error_block;
3777 }
3778 if (action == ML_DELETE)
3779 {
3780 pp->pb_pointer[idx].pe_line_count--;
3781 dirty = TRUE;
3782 }
3783 else if (action == ML_INSERT)
3784 {
3785 pp->pb_pointer[idx].pe_line_count++;
3786 dirty = TRUE;
3787 }
3788 mf_put(mfp, hp, dirty, FALSE);
3789 }
3790
3791error_block:
3792 mf_put(mfp, hp, FALSE, FALSE);
3793error_noblock:
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02003794 /*
3795 * If action is ML_DELETE or ML_INSERT we have to correct the tree for
3796 * the incremented/decremented line counts, because there won't be a line
3797 * inserted/deleted after all.
3798 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003799 if (action == ML_DELETE)
3800 ml_lineadd(buf, 1);
3801 else if (action == ML_INSERT)
3802 ml_lineadd(buf, -1);
3803 buf->b_ml.ml_stack_top = 0;
3804 return NULL;
3805}
3806
3807/*
3808 * add an entry to the info pointer stack
3809 *
3810 * return -1 for failure, number of the new entry otherwise
3811 */
3812 static int
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01003813ml_add_stack(buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003814{
3815 int top;
3816 infoptr_T *newstack;
3817
3818 top = buf->b_ml.ml_stack_top;
3819
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02003820 /* may have to increase the stack size */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003821 if (top == buf->b_ml.ml_stack_size)
3822 {
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02003823 CHECK(top > 0, _("Stack size increases")); /* more than 5 levels??? */
Bram Moolenaar071d4272004-06-13 20:20:40 +00003824
3825 newstack = (infoptr_T *)alloc((unsigned)sizeof(infoptr_T) *
3826 (buf->b_ml.ml_stack_size + STACK_INCR));
3827 if (newstack == NULL)
3828 return -1;
Bram Moolenaarfbd302f2015-08-08 18:23:46 +02003829 if (top > 0)
3830 mch_memmove(newstack, buf->b_ml.ml_stack,
Bram Moolenaar8c8de832008-06-24 22:58:06 +00003831 (size_t)top * sizeof(infoptr_T));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003832 vim_free(buf->b_ml.ml_stack);
3833 buf->b_ml.ml_stack = newstack;
3834 buf->b_ml.ml_stack_size += STACK_INCR;
3835 }
3836
3837 buf->b_ml.ml_stack_top++;
3838 return top;
3839}
3840
3841/*
3842 * Update the pointer blocks on the stack for inserted/deleted lines.
3843 * The stack itself is also updated.
3844 *
3845 * When a insert/delete line action fails, the line is not inserted/deleted,
3846 * but the pointer blocks have already been updated. That is fixed here by
3847 * walking through the stack.
3848 *
3849 * Count is the number of lines added, negative if lines have been deleted.
3850 */
3851 static void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01003852ml_lineadd(buf_T *buf, int count)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003853{
3854 int idx;
3855 infoptr_T *ip;
3856 PTR_BL *pp;
3857 memfile_T *mfp = buf->b_ml.ml_mfp;
3858 bhdr_T *hp;
3859
3860 for (idx = buf->b_ml.ml_stack_top - 1; idx >= 0; --idx)
3861 {
3862 ip = &(buf->b_ml.ml_stack[idx]);
3863 if ((hp = mf_get(mfp, ip->ip_bnum, 1)) == NULL)
3864 break;
3865 pp = (PTR_BL *)(hp->bh_data); /* must be pointer block */
3866 if (pp->pb_id != PTR_ID)
3867 {
3868 mf_put(mfp, hp, FALSE, FALSE);
Bram Moolenaar95f09602016-11-10 20:01:45 +01003869 IEMSG(_("E317: pointer block id wrong 2"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00003870 break;
3871 }
3872 pp->pb_pointer[ip->ip_index].pe_line_count += count;
3873 ip->ip_high += count;
3874 mf_put(mfp, hp, TRUE, FALSE);
3875 }
3876}
3877
Bram Moolenaar55debbe2010-05-23 23:34:36 +02003878#if defined(HAVE_READLINK) || defined(PROTO)
Bram Moolenaar900b4d72005-12-12 22:05:50 +00003879/*
3880 * Resolve a symlink in the last component of a file name.
3881 * Note that f_resolve() does it for every part of the path, we don't do that
3882 * here.
3883 * If it worked returns OK and the resolved link in "buf[MAXPATHL]".
3884 * Otherwise returns FAIL.
3885 */
Bram Moolenaar55debbe2010-05-23 23:34:36 +02003886 int
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01003887resolve_symlink(char_u *fname, char_u *buf)
Bram Moolenaar900b4d72005-12-12 22:05:50 +00003888{
3889 char_u tmp[MAXPATHL];
3890 int ret;
3891 int depth = 0;
3892
3893 if (fname == NULL)
3894 return FAIL;
3895
3896 /* Put the result so far in tmp[], starting with the original name. */
3897 vim_strncpy(tmp, fname, MAXPATHL - 1);
3898
3899 for (;;)
3900 {
3901 /* Limit symlink depth to 100, catch recursive loops. */
3902 if (++depth == 100)
3903 {
3904 EMSG2(_("E773: Symlink loop for \"%s\""), fname);
3905 return FAIL;
3906 }
3907
3908 ret = readlink((char *)tmp, (char *)buf, MAXPATHL - 1);
3909 if (ret <= 0)
3910 {
Bram Moolenaarcc984262005-12-23 22:19:46 +00003911 if (errno == EINVAL || errno == ENOENT)
Bram Moolenaar900b4d72005-12-12 22:05:50 +00003912 {
Bram Moolenaarcc984262005-12-23 22:19:46 +00003913 /* Found non-symlink or not existing file, stop here.
Bram Moolenaar9439cdd2009-04-22 13:39:36 +00003914 * When at the first level use the unmodified name, skip the
Bram Moolenaar900b4d72005-12-12 22:05:50 +00003915 * call to vim_FullName(). */
3916 if (depth == 1)
3917 return FAIL;
3918
3919 /* Use the resolved name in tmp[]. */
3920 break;
3921 }
3922
3923 /* There must be some error reading links, use original name. */
3924 return FAIL;
3925 }
3926 buf[ret] = NUL;
3927
3928 /*
3929 * Check whether the symlink is relative or absolute.
3930 * If it's relative, build a new path based on the directory
3931 * portion of the filename (if any) and the path the symlink
3932 * points to.
3933 */
3934 if (mch_isFullName(buf))
3935 STRCPY(tmp, buf);
3936 else
3937 {
3938 char_u *tail;
3939
3940 tail = gettail(tmp);
3941 if (STRLEN(tail) + STRLEN(buf) >= MAXPATHL)
3942 return FAIL;
3943 STRCPY(tail, buf);
3944 }
3945 }
3946
3947 /*
3948 * Try to resolve the full name of the file so that the swapfile name will
3949 * be consistent even when opening a relative symlink from different
3950 * working directories.
3951 */
3952 return vim_FullName(tmp, buf, MAXPATHL, TRUE);
3953}
3954#endif
3955
Bram Moolenaar071d4272004-06-13 20:20:40 +00003956/*
Bram Moolenaar04a09c12005-08-01 22:02:32 +00003957 * Make swap file name out of the file name and a directory name.
3958 * Returns pointer to allocated memory or NULL.
Bram Moolenaar071d4272004-06-13 20:20:40 +00003959 */
Bram Moolenaar04a09c12005-08-01 22:02:32 +00003960 char_u *
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01003961makeswapname(
3962 char_u *fname,
3963 char_u *ffname UNUSED,
3964 buf_T *buf,
3965 char_u *dir_name)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003966{
3967 char_u *r, *s;
Bram Moolenaar9dbe4752010-05-14 17:52:42 +02003968 char_u *fname_res = fname;
Bram Moolenaar900b4d72005-12-12 22:05:50 +00003969#ifdef HAVE_READLINK
3970 char_u fname_buf[MAXPATHL];
Bram Moolenaar900b4d72005-12-12 22:05:50 +00003971#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00003972
3973#if defined(UNIX) || defined(WIN3264) /* Need _very_ long file names */
Bram Moolenaarb113c3a2017-02-28 21:26:17 +01003974 int len = (int)STRLEN(dir_name);
Bram Moolenaarc525e3a2017-02-18 16:59:02 +01003975
3976 s = dir_name + len;
3977 if (after_pathsep(dir_name, s) && len > 1 && s[-1] == s[-2])
Bram Moolenaar071d4272004-06-13 20:20:40 +00003978 { /* Ends with '//', Use Full path */
3979 r = NULL;
Bram Moolenaar04a09c12005-08-01 22:02:32 +00003980 if ((s = make_percent_swname(dir_name, fname)) != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00003981 {
3982 r = modname(s, (char_u *)".swp", FALSE);
3983 vim_free(s);
3984 }
3985 return r;
3986 }
3987#endif
3988
Bram Moolenaar900b4d72005-12-12 22:05:50 +00003989#ifdef HAVE_READLINK
3990 /* Expand symlink in the file name, so that we put the swap file with the
3991 * actual file instead of with the symlink. */
3992 if (resolve_symlink(fname, fname_buf) == OK)
3993 fname_res = fname_buf;
Bram Moolenaar900b4d72005-12-12 22:05:50 +00003994#endif
3995
Bram Moolenaar071d4272004-06-13 20:20:40 +00003996 r = buf_modname(
Bram Moolenaar071d4272004-06-13 20:20:40 +00003997 (buf->b_p_sn || buf->b_shortname),
Bram Moolenaar900b4d72005-12-12 22:05:50 +00003998 fname_res,
Bram Moolenaar071d4272004-06-13 20:20:40 +00003999 (char_u *)
Bram Moolenaare60acc12011-05-10 16:41:25 +02004000#if defined(VMS)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004001 "_swp",
4002#else
4003 ".swp",
4004#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004005 /* Prepend a '.' to the swap file name for the current directory. */
Bram Moolenaar48e330a2016-02-23 14:53:34 +01004006 dir_name[0] == '.' && dir_name[1] == NUL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004007 if (r == NULL) /* out of memory */
4008 return NULL;
4009
4010 s = get_file_in_dir(r, dir_name);
4011 vim_free(r);
4012 return s;
4013}
4014
4015/*
4016 * Get file name to use for swap file or backup file.
4017 * Use the name of the edited file "fname" and an entry in the 'dir' or 'bdir'
4018 * option "dname".
4019 * - If "dname" is ".", return "fname" (swap file in dir of file).
4020 * - If "dname" starts with "./", insert "dname" in "fname" (swap file
4021 * relative to dir of file).
4022 * - Otherwise, prepend "dname" to the tail of "fname" (swap file in specific
4023 * dir).
4024 *
4025 * The return value is an allocated string and can be NULL.
4026 */
4027 char_u *
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01004028get_file_in_dir(
4029 char_u *fname,
4030 char_u *dname) /* don't use "dirname", it is a global for Alpha */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004031{
4032 char_u *t;
4033 char_u *tail;
4034 char_u *retval;
4035 int save_char;
4036
4037 tail = gettail(fname);
4038
4039 if (dname[0] == '.' && dname[1] == NUL)
4040 retval = vim_strsave(fname);
4041 else if (dname[0] == '.' && vim_ispathsep(dname[1]))
4042 {
4043 if (tail == fname) /* no path before file name */
4044 retval = concat_fnames(dname + 2, tail, TRUE);
4045 else
4046 {
4047 save_char = *tail;
4048 *tail = NUL;
4049 t = concat_fnames(fname, dname + 2, TRUE);
4050 *tail = save_char;
4051 if (t == NULL) /* out of memory */
4052 retval = NULL;
4053 else
4054 {
4055 retval = concat_fnames(t, tail, TRUE);
4056 vim_free(t);
4057 }
4058 }
4059 }
4060 else
4061 retval = concat_fnames(dname, tail, TRUE);
4062
Bram Moolenaar69c35002013-11-04 02:54:12 +01004063#ifdef WIN3264
4064 if (retval != NULL)
Bram Moolenaar91acfff2017-03-12 19:22:36 +01004065 for (t = gettail(retval); *t != NUL; MB_PTR_ADV(t))
Bram Moolenaar69c35002013-11-04 02:54:12 +01004066 if (*t == ':')
4067 *t = '%';
4068#endif
4069
Bram Moolenaar071d4272004-06-13 20:20:40 +00004070 return retval;
4071}
4072
Bram Moolenaar92b8b2d2016-01-29 22:36:45 +01004073static void attention_message(buf_T *buf, char_u *fname);
Bram Moolenaard5bc83f2005-12-07 21:07:59 +00004074
4075/*
4076 * Print the ATTENTION message: info about an existing swap file.
4077 */
4078 static void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01004079attention_message(
4080 buf_T *buf, /* buffer being edited */
4081 char_u *fname) /* swap file name */
Bram Moolenaard5bc83f2005-12-07 21:07:59 +00004082{
Bram Moolenaar8767f522016-07-01 17:17:39 +02004083 stat_T st;
Bram Moolenaard5bc83f2005-12-07 21:07:59 +00004084 time_t x, sx;
Bram Moolenaar31e97bf2006-10-10 14:20:13 +00004085 char *p;
Bram Moolenaard5bc83f2005-12-07 21:07:59 +00004086
4087 ++no_wait_return;
4088 (void)EMSG(_("E325: ATTENTION"));
4089 MSG_PUTS(_("\nFound a swap file by the name \""));
4090 msg_home_replace(fname);
4091 MSG_PUTS("\"\n");
4092 sx = swapfile_info(fname);
4093 MSG_PUTS(_("While opening file \""));
4094 msg_outtrans(buf->b_fname);
4095 MSG_PUTS("\"\n");
4096 if (mch_stat((char *)buf->b_fname, &st) != -1)
4097 {
4098 MSG_PUTS(_(" dated: "));
4099 x = st.st_mtime; /* Manx C can't do &st.st_mtime */
Bram Moolenaar31e97bf2006-10-10 14:20:13 +00004100 p = ctime(&x); /* includes '\n' */
4101 if (p == NULL)
4102 MSG_PUTS("(invalid)\n");
4103 else
4104 MSG_PUTS(p);
Bram Moolenaard5bc83f2005-12-07 21:07:59 +00004105 if (sx != 0 && x > sx)
4106 MSG_PUTS(_(" NEWER than swap file!\n"));
4107 }
4108 /* Some of these messages are long to allow translation to
4109 * other languages. */
Bram Moolenaard9ea9062016-02-02 12:38:02 +01004110 MSG_PUTS(_("\n(1) Another program may be editing the same file. If this is the case,\n be careful not to end up with two different instances of the same\n file when making changes. Quit, or continue with caution.\n"));
Bram Moolenaarc41fc712011-02-15 11:57:04 +01004111 MSG_PUTS(_("(2) An edit session for this file crashed.\n"));
Bram Moolenaard5bc83f2005-12-07 21:07:59 +00004112 MSG_PUTS(_(" If this is the case, use \":recover\" or \"vim -r "));
4113 msg_outtrans(buf->b_fname);
4114 MSG_PUTS(_("\"\n to recover the changes (see \":help recovery\").\n"));
4115 MSG_PUTS(_(" If you did this already, delete the swap file \""));
4116 msg_outtrans(fname);
4117 MSG_PUTS(_("\"\n to avoid this message.\n"));
4118 cmdline_row = msg_row;
4119 --no_wait_return;
4120}
4121
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +01004122#if defined(FEAT_EVAL)
Bram Moolenaard5bc83f2005-12-07 21:07:59 +00004123/*
4124 * Trigger the SwapExists autocommands.
4125 * Returns a value for equivalent to do_dialog() (see below):
4126 * 0: still need to ask for a choice
4127 * 1: open read-only
4128 * 2: edit anyway
4129 * 3: recover
4130 * 4: delete it
4131 * 5: quit
4132 * 6: abort
4133 */
4134 static int
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01004135do_swapexists(buf_T *buf, char_u *fname)
Bram Moolenaard5bc83f2005-12-07 21:07:59 +00004136{
4137 set_vim_var_string(VV_SWAPNAME, fname, -1);
4138 set_vim_var_string(VV_SWAPCHOICE, NULL, -1);
4139
4140 /* Trigger SwapExists autocommands with <afile> set to the file being
Bram Moolenaar12c22ce2009-04-22 13:58:46 +00004141 * edited. Disallow changing directory here. */
4142 ++allbuf_lock;
Bram Moolenaard5bc83f2005-12-07 21:07:59 +00004143 apply_autocmds(EVENT_SWAPEXISTS, buf->b_fname, NULL, FALSE, NULL);
Bram Moolenaar12c22ce2009-04-22 13:58:46 +00004144 --allbuf_lock;
Bram Moolenaard5bc83f2005-12-07 21:07:59 +00004145
4146 set_vim_var_string(VV_SWAPNAME, NULL, -1);
4147
4148 switch (*get_vim_var_str(VV_SWAPCHOICE))
4149 {
4150 case 'o': return 1;
4151 case 'e': return 2;
4152 case 'r': return 3;
4153 case 'd': return 4;
4154 case 'q': return 5;
4155 case 'a': return 6;
4156 }
4157
4158 return 0;
4159}
4160#endif
4161
Bram Moolenaar071d4272004-06-13 20:20:40 +00004162/*
4163 * Find out what name to use for the swap file for buffer 'buf'.
4164 *
4165 * Several names are tried to find one that does not exist
Bram Moolenaar04a09c12005-08-01 22:02:32 +00004166 * Returns the name in allocated memory or NULL.
Bram Moolenaarf541c362011-10-26 11:44:18 +02004167 * When out of memory "dirp" is set to NULL.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004168 *
4169 * Note: If BASENAMELEN is not correct, you will get error messages for
Bram Moolenaar55debbe2010-05-23 23:34:36 +02004170 * not being able to open the swap or undo file
Bram Moolenaar12c22ce2009-04-22 13:58:46 +00004171 * Note: May trigger SwapExists autocmd, pointers may change!
Bram Moolenaar071d4272004-06-13 20:20:40 +00004172 */
4173 static char_u *
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01004174findswapname(
4175 buf_T *buf,
4176 char_u **dirp, /* pointer to list of directories */
4177 char_u *old_fname) /* don't give warning for this file name */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004178{
4179 char_u *fname;
4180 int n;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004181 char_u *dir_name;
4182#ifdef AMIGA
4183 BPTR fh;
4184#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004185 int r;
Bram Moolenaar69c35002013-11-04 02:54:12 +01004186 char_u *buf_fname = buf->b_fname;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004187
Bram Moolenaar48e330a2016-02-23 14:53:34 +01004188#if !defined(UNIX)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004189# define CREATE_DUMMY_FILE
4190 FILE *dummyfd = NULL;
4191
Bram Moolenaar69c35002013-11-04 02:54:12 +01004192# ifdef WIN3264
4193 if (buf_fname != NULL && !mch_isFullName(buf_fname)
4194 && vim_strchr(gettail(buf_fname), ':'))
4195 {
4196 char_u *t;
4197
4198 buf_fname = vim_strsave(buf_fname);
4199 if (buf_fname == NULL)
4200 buf_fname = buf->b_fname;
4201 else
Bram Moolenaar91acfff2017-03-12 19:22:36 +01004202 for (t = gettail(buf_fname); *t != NUL; MB_PTR_ADV(t))
Bram Moolenaar69c35002013-11-04 02:54:12 +01004203 if (*t == ':')
4204 *t = '%';
4205 }
4206# endif
4207
Bram Moolenaar55debbe2010-05-23 23:34:36 +02004208 /*
4209 * If we start editing a new file, e.g. "test.doc", which resides on an
4210 * MSDOS compatible filesystem, it is possible that the file
4211 * "test.doc.swp" which we create will be exactly the same file. To avoid
4212 * this problem we temporarily create "test.doc". Don't do this when the
4213 * check below for a 8.3 file name is used.
4214 */
Bram Moolenaar69c35002013-11-04 02:54:12 +01004215 if (!(buf->b_p_sn || buf->b_shortname) && buf_fname != NULL
4216 && mch_getperm(buf_fname) < 0)
4217 dummyfd = mch_fopen((char *)buf_fname, "w");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004218#endif
4219
Bram Moolenaar55debbe2010-05-23 23:34:36 +02004220 /*
4221 * Isolate a directory name from *dirp and put it in dir_name.
4222 * First allocate some memory to put the directory name in.
4223 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004224 dir_name = alloc((unsigned)STRLEN(*dirp) + 1);
Bram Moolenaarf541c362011-10-26 11:44:18 +02004225 if (dir_name == NULL)
4226 *dirp = NULL;
4227 else
Bram Moolenaar071d4272004-06-13 20:20:40 +00004228 (void)copy_option_part(dirp, dir_name, 31000, ",");
4229
Bram Moolenaar55debbe2010-05-23 23:34:36 +02004230 /*
4231 * we try different names until we find one that does not exist yet
4232 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004233 if (dir_name == NULL) /* out of memory */
4234 fname = NULL;
4235 else
Bram Moolenaar69c35002013-11-04 02:54:12 +01004236 fname = makeswapname(buf_fname, buf->b_ffname, buf, dir_name);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004237
4238 for (;;)
4239 {
4240 if (fname == NULL) /* must be out of memory */
4241 break;
4242 if ((n = (int)STRLEN(fname)) == 0) /* safety check */
4243 {
Bram Moolenaard23a8232018-02-10 18:45:26 +01004244 VIM_CLEAR(fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004245 break;
4246 }
Bram Moolenaar48e330a2016-02-23 14:53:34 +01004247#if defined(UNIX)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004248/*
4249 * Some systems have a MS-DOS compatible filesystem that use 8.3 character
4250 * file names. If this is the first try and the swap file name does not fit in
4251 * 8.3, detect if this is the case, set shortname and try again.
4252 */
4253 if (fname[n - 2] == 'w' && fname[n - 1] == 'p'
4254 && !(buf->b_p_sn || buf->b_shortname))
4255 {
4256 char_u *tail;
4257 char_u *fname2;
Bram Moolenaar8767f522016-07-01 17:17:39 +02004258 stat_T s1, s2;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004259 int f1, f2;
4260 int created1 = FALSE, created2 = FALSE;
4261 int same = FALSE;
4262
4263 /*
4264 * Check if swapfile name does not fit in 8.3:
4265 * It either contains two dots, is longer than 8 chars, or starts
4266 * with a dot.
4267 */
Bram Moolenaar69c35002013-11-04 02:54:12 +01004268 tail = gettail(buf_fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004269 if ( vim_strchr(tail, '.') != NULL
4270 || STRLEN(tail) > (size_t)8
4271 || *gettail(fname) == '.')
4272 {
4273 fname2 = alloc(n + 2);
4274 if (fname2 != NULL)
4275 {
4276 STRCPY(fname2, fname);
4277 /* if fname == "xx.xx.swp", fname2 = "xx.xx.swx"
4278 * if fname == ".xx.swp", fname2 = ".xx.swpx"
4279 * if fname == "123456789.swp", fname2 = "12345678x.swp"
4280 */
4281 if (vim_strchr(tail, '.') != NULL)
4282 fname2[n - 1] = 'x';
4283 else if (*gettail(fname) == '.')
4284 {
4285 fname2[n] = 'x';
4286 fname2[n + 1] = NUL;
4287 }
4288 else
4289 fname2[n - 5] += 1;
4290 /*
4291 * may need to create the files to be able to use mch_stat()
4292 */
4293 f1 = mch_open((char *)fname, O_RDONLY | O_EXTRA, 0);
4294 if (f1 < 0)
4295 {
4296 f1 = mch_open_rw((char *)fname,
4297 O_RDWR|O_CREAT|O_EXCL|O_EXTRA);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004298 created1 = TRUE;
4299 }
4300 if (f1 >= 0)
4301 {
4302 f2 = mch_open((char *)fname2, O_RDONLY | O_EXTRA, 0);
4303 if (f2 < 0)
4304 {
4305 f2 = mch_open_rw((char *)fname2,
4306 O_RDWR|O_CREAT|O_EXCL|O_EXTRA);
4307 created2 = TRUE;
4308 }
4309 if (f2 >= 0)
4310 {
4311 /*
4312 * Both files exist now. If mch_stat() returns the
4313 * same device and inode they are the same file.
4314 */
4315 if (mch_fstat(f1, &s1) != -1
4316 && mch_fstat(f2, &s2) != -1
4317 && s1.st_dev == s2.st_dev
4318 && s1.st_ino == s2.st_ino)
4319 same = TRUE;
4320 close(f2);
4321 if (created2)
4322 mch_remove(fname2);
4323 }
4324 close(f1);
4325 if (created1)
4326 mch_remove(fname);
4327 }
4328 vim_free(fname2);
4329 if (same)
4330 {
4331 buf->b_shortname = TRUE;
4332 vim_free(fname);
Bram Moolenaar69c35002013-11-04 02:54:12 +01004333 fname = makeswapname(buf_fname, buf->b_ffname,
Bram Moolenaar04a09c12005-08-01 22:02:32 +00004334 buf, dir_name);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004335 continue; /* try again with b_shortname set */
4336 }
4337 }
4338 }
4339 }
4340#endif
4341 /*
4342 * check if the swapfile already exists
4343 */
4344 if (mch_getperm(fname) < 0) /* it does not exist */
4345 {
4346#ifdef HAVE_LSTAT
Bram Moolenaar8767f522016-07-01 17:17:39 +02004347 stat_T sb;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004348
4349 /*
4350 * Extra security check: When a swap file is a symbolic link, this
4351 * is most likely a symlink attack.
4352 */
4353 if (mch_lstat((char *)fname, &sb) < 0)
4354#else
4355# ifdef AMIGA
4356 fh = Open((UBYTE *)fname, (long)MODE_NEWFILE);
4357 /*
4358 * on the Amiga mch_getperm() will return -1 when the file exists
4359 * but is being used by another program. This happens if you edit
4360 * a file twice.
4361 */
4362 if (fh != (BPTR)NULL) /* can open file, OK */
4363 {
4364 Close(fh);
4365 mch_remove(fname);
4366 break;
4367 }
4368 if (IoErr() != ERROR_OBJECT_IN_USE
4369 && IoErr() != ERROR_OBJECT_EXISTS)
4370# endif
4371#endif
4372 break;
4373 }
4374
4375 /*
4376 * A file name equal to old_fname is OK to use.
4377 */
4378 if (old_fname != NULL && fnamecmp(fname, old_fname) == 0)
4379 break;
4380
4381 /*
4382 * get here when file already exists
4383 */
4384 if (fname[n - 2] == 'w' && fname[n - 1] == 'p') /* first try */
4385 {
Bram Moolenaar071d4272004-06-13 20:20:40 +00004386 /*
4387 * on MS-DOS compatible filesystems (e.g. messydos) file.doc.swp
4388 * and file.doc are the same file. To guess if this problem is
4389 * present try if file.doc.swx exists. If it does, we set
4390 * buf->b_shortname and try file_doc.swp (dots replaced by
4391 * underscores for this file), and try again. If it doesn't we
4392 * assume that "file.doc.swp" already exists.
4393 */
4394 if (!(buf->b_p_sn || buf->b_shortname)) /* not tried yet */
4395 {
4396 fname[n - 1] = 'x';
4397 r = mch_getperm(fname); /* try "file.swx" */
4398 fname[n - 1] = 'p';
4399 if (r >= 0) /* "file.swx" seems to exist */
4400 {
4401 buf->b_shortname = TRUE;
4402 vim_free(fname);
Bram Moolenaar69c35002013-11-04 02:54:12 +01004403 fname = makeswapname(buf_fname, buf->b_ffname,
Bram Moolenaar04a09c12005-08-01 22:02:32 +00004404 buf, dir_name);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004405 continue; /* try again with '.' replaced with '_' */
4406 }
4407 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004408 /*
4409 * If we get here the ".swp" file really exists.
4410 * Give an error message, unless recovering, no file name, we are
4411 * viewing a help file or when the path of the file is different
4412 * (happens when all .swp files are in one directory).
4413 */
Bram Moolenaar69c35002013-11-04 02:54:12 +01004414 if (!recoverymode && buf_fname != NULL
Bram Moolenaar8fc061c2004-12-29 21:03:02 +00004415 && !buf->b_help && !(buf->b_flags & BF_DUMMY))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004416 {
4417 int fd;
4418 struct block0 b0;
4419 int differ = FALSE;
4420
4421 /*
4422 * Try to read block 0 from the swap file to get the original
4423 * file name (and inode number).
4424 */
4425 fd = mch_open((char *)fname, O_RDONLY | O_EXTRA, 0);
4426 if (fd >= 0)
4427 {
Bram Moolenaar540fc6f2010-12-17 16:27:16 +01004428 if (read_eintr(fd, &b0, sizeof(b0)) == sizeof(b0))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004429 {
4430 /*
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00004431 * If the swapfile has the same directory as the
4432 * buffer don't compare the directory names, they can
4433 * have a different mountpoint.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004434 */
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00004435 if (b0.b0_flags & B0_SAME_DIR)
4436 {
4437 if (fnamecmp(gettail(buf->b_ffname),
4438 gettail(b0.b0_fname)) != 0
4439 || !same_directory(fname, buf->b_ffname))
Bram Moolenaar900b4d72005-12-12 22:05:50 +00004440 {
4441#ifdef CHECK_INODE
4442 /* Symlinks may point to the same file even
4443 * when the name differs, need to check the
4444 * inode too. */
4445 expand_env(b0.b0_fname, NameBuff, MAXPATHL);
4446 if (fnamecmp_ino(buf->b_ffname, NameBuff,
4447 char_to_long(b0.b0_ino)))
4448#endif
4449 differ = TRUE;
4450 }
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00004451 }
4452 else
4453 {
4454 /*
4455 * The name in the swap file may be
4456 * "~user/path/file". Expand it first.
4457 */
4458 expand_env(b0.b0_fname, NameBuff, MAXPATHL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004459#ifdef CHECK_INODE
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00004460 if (fnamecmp_ino(buf->b_ffname, NameBuff,
Bram Moolenaar900b4d72005-12-12 22:05:50 +00004461 char_to_long(b0.b0_ino)))
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00004462 differ = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004463#else
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00004464 if (fnamecmp(NameBuff, buf->b_ffname) != 0)
4465 differ = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004466#endif
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00004467 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004468 }
4469 close(fd);
4470 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004471
4472 /* give the ATTENTION message when there is an old swap file
4473 * for the current file, and the buffer was not recovered. */
4474 if (differ == FALSE && !(curbuf->b_flags & BF_RECOVERED)
4475 && vim_strchr(p_shm, SHM_ATTENTION) == NULL)
4476 {
Bram Moolenaard5bc83f2005-12-07 21:07:59 +00004477#if defined(HAS_SWAP_EXISTS_ACTION)
4478 int choice = 0;
4479#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004480#ifdef CREATE_DUMMY_FILE
4481 int did_use_dummy = FALSE;
4482
4483 /* Avoid getting a warning for the file being created
4484 * outside of Vim, it was created at the start of this
4485 * function. Delete the file now, because Vim might exit
4486 * here if the window is closed. */
4487 if (dummyfd != NULL)
4488 {
4489 fclose(dummyfd);
4490 dummyfd = NULL;
Bram Moolenaar69c35002013-11-04 02:54:12 +01004491 mch_remove(buf_fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004492 did_use_dummy = TRUE;
4493 }
4494#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004495
Bram Moolenaara06ecab2016-07-16 14:47:36 +02004496#if (defined(UNIX) || defined(VMS)) && (defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG))
Bram Moolenaar071d4272004-06-13 20:20:40 +00004497 process_still_running = FALSE;
4498#endif
Bram Moolenaarf2bd8ef2018-03-04 18:08:14 +01004499#if defined(FEAT_EVAL)
Bram Moolenaard5bc83f2005-12-07 21:07:59 +00004500 /*
4501 * If there is an SwapExists autocommand and we can handle
4502 * the response, trigger it. It may return 0 to ask the
4503 * user anyway.
4504 */
4505 if (swap_exists_action != SEA_NONE
Bram Moolenaar69c35002013-11-04 02:54:12 +01004506 && has_autocmd(EVENT_SWAPEXISTS, buf_fname, buf))
Bram Moolenaard5bc83f2005-12-07 21:07:59 +00004507 choice = do_swapexists(buf, fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004508
Bram Moolenaard5bc83f2005-12-07 21:07:59 +00004509 if (choice == 0)
4510#endif
4511 {
4512#ifdef FEAT_GUI
4513 /* If we are supposed to start the GUI but it wasn't
4514 * completely started yet, start it now. This makes
4515 * the messages displayed in the Vim window when
4516 * loading a session from the .gvimrc file. */
4517 if (gui.starting && !gui.in_use)
4518 gui_start();
4519#endif
4520 /* Show info about the existing swap file. */
4521 attention_message(buf, fname);
4522
4523 /* We don't want a 'q' typed at the more-prompt
4524 * interrupt loading a file. */
4525 got_int = FALSE;
4526 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004527
4528#if defined(FEAT_GUI_DIALOG) || defined(FEAT_CON_DIALOG)
Bram Moolenaard5bc83f2005-12-07 21:07:59 +00004529 if (swap_exists_action != SEA_NONE && choice == 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004530 {
4531 char_u *name;
4532
4533 name = alloc((unsigned)(STRLEN(fname)
4534 + STRLEN(_("Swap file \""))
4535 + STRLEN(_("\" already exists!")) + 5));
4536 if (name != NULL)
4537 {
4538 STRCPY(name, _("Swap file \""));
4539 home_replace(NULL, fname, name + STRLEN(name),
4540 1000, TRUE);
4541 STRCAT(name, _("\" already exists!"));
4542 }
Bram Moolenaard5bc83f2005-12-07 21:07:59 +00004543 choice = do_dialog(VIM_WARNING,
Bram Moolenaar071d4272004-06-13 20:20:40 +00004544 (char_u *)_("VIM - ATTENTION"),
4545 name == NULL
4546 ? (char_u *)_("Swap file already exists!")
4547 : name,
Bram Moolenaara06ecab2016-07-16 14:47:36 +02004548# if defined(UNIX) || defined(VMS)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004549 process_still_running
4550 ? (char_u *)_("&Open Read-Only\n&Edit anyway\n&Recover\n&Quit\n&Abort") :
4551# endif
Bram Moolenaard2c340a2011-01-17 20:08:11 +01004552 (char_u *)_("&Open Read-Only\n&Edit anyway\n&Recover\n&Delete it\n&Quit\n&Abort"), 1, NULL, FALSE);
Bram Moolenaard5bc83f2005-12-07 21:07:59 +00004553
Bram Moolenaara06ecab2016-07-16 14:47:36 +02004554# if defined(UNIX) || defined(VMS)
Bram Moolenaard5bc83f2005-12-07 21:07:59 +00004555 if (process_still_running && choice >= 4)
4556 choice++; /* Skip missing "Delete it" button */
4557# endif
4558 vim_free(name);
4559
4560 /* pretend screen didn't scroll, need redraw anyway */
4561 msg_scrolled = 0;
4562 redraw_all_later(NOT_VALID);
4563 }
4564#endif
4565
4566#if defined(HAS_SWAP_EXISTS_ACTION)
4567 if (choice > 0)
4568 {
4569 switch (choice)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004570 {
4571 case 1:
4572 buf->b_p_ro = TRUE;
4573 break;
4574 case 2:
4575 break;
4576 case 3:
4577 swap_exists_action = SEA_RECOVER;
4578 break;
4579 case 4:
Bram Moolenaard5bc83f2005-12-07 21:07:59 +00004580 mch_remove(fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004581 break;
4582 case 5:
4583 swap_exists_action = SEA_QUIT;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004584 break;
4585 case 6:
Bram Moolenaard5bc83f2005-12-07 21:07:59 +00004586 swap_exists_action = SEA_QUIT;
4587 got_int = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004588 break;
4589 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00004590
4591 /* If the file was deleted this fname can be used. */
4592 if (mch_getperm(fname) < 0)
4593 break;
4594 }
4595 else
4596#endif
4597 {
4598 MSG_PUTS("\n");
Bram Moolenaar4770d092006-01-12 23:22:24 +00004599 if (msg_silent == 0)
4600 /* call wait_return() later */
4601 need_wait_return = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004602 }
4603
4604#ifdef CREATE_DUMMY_FILE
4605 /* Going to try another name, need the dummy file again. */
4606 if (did_use_dummy)
Bram Moolenaar69c35002013-11-04 02:54:12 +01004607 dummyfd = mch_fopen((char *)buf_fname, "w");
Bram Moolenaar071d4272004-06-13 20:20:40 +00004608#endif
4609 }
4610 }
4611 }
4612
4613 /*
4614 * Change the ".swp" extension to find another file that can be used.
4615 * First decrement the last char: ".swo", ".swn", etc.
4616 * If that still isn't enough decrement the last but one char: ".svz"
Bram Moolenaar69a7cb42004-06-20 12:51:53 +00004617 * Can happen when editing many "No Name" buffers.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004618 */
4619 if (fname[n - 1] == 'a') /* ".s?a" */
4620 {
4621 if (fname[n - 2] == 'a') /* ".saa": tried enough, give up */
4622 {
4623 EMSG(_("E326: Too many swap files found"));
Bram Moolenaard23a8232018-02-10 18:45:26 +01004624 VIM_CLEAR(fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004625 break;
4626 }
4627 --fname[n - 2]; /* ".svz", ".suz", etc. */
4628 fname[n - 1] = 'z' + 1;
4629 }
4630 --fname[n - 1]; /* ".swo", ".swn", etc. */
4631 }
4632
4633 vim_free(dir_name);
4634#ifdef CREATE_DUMMY_FILE
4635 if (dummyfd != NULL) /* file has been created temporarily */
4636 {
4637 fclose(dummyfd);
Bram Moolenaar69c35002013-11-04 02:54:12 +01004638 mch_remove(buf_fname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00004639 }
4640#endif
Bram Moolenaar69c35002013-11-04 02:54:12 +01004641#ifdef WIN3264
4642 if (buf_fname != buf->b_fname)
4643 vim_free(buf_fname);
4644#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004645 return fname;
4646}
4647
4648 static int
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01004649b0_magic_wrong(ZERO_BL *b0p)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004650{
4651 return (b0p->b0_magic_long != (long)B0_MAGIC_LONG
4652 || b0p->b0_magic_int != (int)B0_MAGIC_INT
4653 || b0p->b0_magic_short != (short)B0_MAGIC_SHORT
4654 || b0p->b0_magic_char != B0_MAGIC_CHAR);
4655}
4656
4657#ifdef CHECK_INODE
4658/*
4659 * Compare current file name with file name from swap file.
4660 * Try to use inode numbers when possible.
4661 * Return non-zero when files are different.
4662 *
4663 * When comparing file names a few things have to be taken into consideration:
4664 * - When working over a network the full path of a file depends on the host.
4665 * We check the inode number if possible. It is not 100% reliable though,
4666 * because the device number cannot be used over a network.
4667 * - When a file does not exist yet (editing a new file) there is no inode
4668 * number.
4669 * - The file name in a swap file may not be valid on the current host. The
4670 * "~user" form is used whenever possible to avoid this.
4671 *
4672 * This is getting complicated, let's make a table:
4673 *
4674 * ino_c ino_s fname_c fname_s differ =
4675 *
4676 * both files exist -> compare inode numbers:
4677 * != 0 != 0 X X ino_c != ino_s
4678 *
4679 * inode number(s) unknown, file names available -> compare file names
4680 * == 0 X OK OK fname_c != fname_s
4681 * X == 0 OK OK fname_c != fname_s
4682 *
4683 * current file doesn't exist, file for swap file exist, file name(s) not
4684 * available -> probably different
4685 * == 0 != 0 FAIL X TRUE
4686 * == 0 != 0 X FAIL TRUE
4687 *
4688 * current file exists, inode for swap unknown, file name(s) not
4689 * available -> probably different
4690 * != 0 == 0 FAIL X TRUE
4691 * != 0 == 0 X FAIL TRUE
4692 *
4693 * current file doesn't exist, inode for swap unknown, one file name not
4694 * available -> probably different
4695 * == 0 == 0 FAIL OK TRUE
4696 * == 0 == 0 OK FAIL TRUE
4697 *
4698 * current file doesn't exist, inode for swap unknown, both file names not
Bram Moolenaar8c3169c2018-05-12 17:04:12 +02004699 * available -> compare file names
4700 * == 0 == 0 FAIL FAIL fname_c != fname_s
Bram Moolenaar071d4272004-06-13 20:20:40 +00004701 *
4702 * Note that when the ino_t is 64 bits, only the last 32 will be used. This
4703 * can't be changed without making the block 0 incompatible with 32 bit
4704 * versions.
4705 */
4706
4707 static int
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01004708fnamecmp_ino(
4709 char_u *fname_c, /* current file name */
4710 char_u *fname_s, /* file name from swap file */
4711 long ino_block0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004712{
Bram Moolenaar8767f522016-07-01 17:17:39 +02004713 stat_T st;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004714 ino_t ino_c = 0; /* ino of current file */
4715 ino_t ino_s; /* ino of file from swap file */
4716 char_u buf_c[MAXPATHL]; /* full path of fname_c */
4717 char_u buf_s[MAXPATHL]; /* full path of fname_s */
4718 int retval_c; /* flag: buf_c valid */
4719 int retval_s; /* flag: buf_s valid */
4720
4721 if (mch_stat((char *)fname_c, &st) == 0)
4722 ino_c = (ino_t)st.st_ino;
4723
4724 /*
4725 * First we try to get the inode from the file name, because the inode in
4726 * the swap file may be outdated. If that fails (e.g. this path is not
4727 * valid on this machine), use the inode from block 0.
4728 */
4729 if (mch_stat((char *)fname_s, &st) == 0)
4730 ino_s = (ino_t)st.st_ino;
4731 else
4732 ino_s = (ino_t)ino_block0;
4733
4734 if (ino_c && ino_s)
4735 return (ino_c != ino_s);
4736
4737 /*
4738 * One of the inode numbers is unknown, try a forced vim_FullName() and
4739 * compare the file names.
4740 */
4741 retval_c = vim_FullName(fname_c, buf_c, MAXPATHL, TRUE);
4742 retval_s = vim_FullName(fname_s, buf_s, MAXPATHL, TRUE);
4743 if (retval_c == OK && retval_s == OK)
Bram Moolenaar8c3169c2018-05-12 17:04:12 +02004744 return STRCMP(buf_c, buf_s) != 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004745
4746 /*
4747 * Can't compare inodes or file names, guess that the files are different,
Bram Moolenaar8c3169c2018-05-12 17:04:12 +02004748 * unless both appear not to exist at all, then compare with the file name
4749 * in the swap file.
Bram Moolenaar071d4272004-06-13 20:20:40 +00004750 */
4751 if (ino_s == 0 && ino_c == 0 && retval_c == FAIL && retval_s == FAIL)
Bram Moolenaar8c3169c2018-05-12 17:04:12 +02004752 return STRCMP(fname_c, fname_s) != 0;
Bram Moolenaar071d4272004-06-13 20:20:40 +00004753 return TRUE;
4754}
4755#endif /* CHECK_INODE */
4756
4757/*
4758 * Move a long integer into a four byte character array.
4759 * Used for machine independency in block zero.
4760 */
4761 static void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01004762long_to_char(long n, char_u *s)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004763{
4764 s[0] = (char_u)(n & 0xff);
4765 n = (unsigned)n >> 8;
4766 s[1] = (char_u)(n & 0xff);
4767 n = (unsigned)n >> 8;
4768 s[2] = (char_u)(n & 0xff);
4769 n = (unsigned)n >> 8;
4770 s[3] = (char_u)(n & 0xff);
4771}
4772
4773 static long
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01004774char_to_long(char_u *s)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004775{
4776 long retval;
4777
4778 retval = s[3];
4779 retval <<= 8;
4780 retval |= s[2];
4781 retval <<= 8;
4782 retval |= s[1];
4783 retval <<= 8;
4784 retval |= s[0];
4785
4786 return retval;
4787}
4788
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00004789/*
4790 * Set the flags in the first block of the swap file:
4791 * - file is modified or not: buf->b_changed
4792 * - 'fileformat'
4793 * - 'fileencoding'
4794 */
Bram Moolenaar071d4272004-06-13 20:20:40 +00004795 void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01004796ml_setflags(buf_T *buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004797{
4798 bhdr_T *hp;
4799 ZERO_BL *b0p;
4800
4801 if (!buf->b_ml.ml_mfp)
4802 return;
4803 for (hp = buf->b_ml.ml_mfp->mf_used_last; hp != NULL; hp = hp->bh_prev)
4804 {
4805 if (hp->bh_bnum == 0)
4806 {
4807 b0p = (ZERO_BL *)(hp->bh_data);
Bram Moolenaar1cd871b2004-12-19 22:46:22 +00004808 b0p->b0_dirty = buf->b_changed ? B0_DIRTY : 0;
4809 b0p->b0_flags = (b0p->b0_flags & ~B0_FF_MASK)
4810 | (get_fileformat(buf) + 1);
4811#ifdef FEAT_MBYTE
4812 add_b0_fenc(b0p, buf);
4813#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00004814 hp->bh_flags |= BH_DIRTY;
4815 mf_sync(buf->b_ml.ml_mfp, MFS_ZERO);
4816 break;
4817 }
4818 }
4819}
4820
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02004821#if defined(FEAT_CRYPT) || defined(PROTO)
4822/*
4823 * If "data" points to a data block encrypt the text in it and return a copy
4824 * in allocated memory. Return NULL when out of memory.
4825 * Otherwise return "data".
4826 */
4827 char_u *
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01004828ml_encrypt_data(
4829 memfile_T *mfp,
4830 char_u *data,
Bram Moolenaar8767f522016-07-01 17:17:39 +02004831 off_T offset,
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01004832 unsigned size)
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02004833{
4834 DATA_BL *dp = (DATA_BL *)data;
4835 char_u *head_end;
4836 char_u *text_start;
4837 char_u *new_data;
4838 int text_len;
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02004839 cryptstate_T *state;
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02004840
4841 if (dp->db_id != DATA_ID)
4842 return data;
4843
Bram Moolenaarbc563362015-06-09 18:35:25 +02004844 state = ml_crypt_prepare(mfp, offset, FALSE);
4845 if (state == NULL)
4846 return data;
4847
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02004848 new_data = (char_u *)alloc(size);
4849 if (new_data == NULL)
4850 return NULL;
4851 head_end = (char_u *)(&dp->db_index[dp->db_line_count]);
4852 text_start = (char_u *)dp + dp->db_txt_start;
4853 text_len = size - dp->db_txt_start;
4854
4855 /* Copy the header and the text. */
4856 mch_memmove(new_data, dp, head_end - (char_u *)dp);
4857
4858 /* Encrypt the text. */
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02004859 crypt_encode(state, text_start, text_len, new_data + dp->db_txt_start);
4860 crypt_free_state(state);
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02004861
4862 /* Clear the gap. */
4863 if (head_end < text_start)
4864 vim_memset(new_data + (head_end - data), 0, text_start - head_end);
4865
4866 return new_data;
4867}
4868
4869/*
Bram Moolenaarbc563362015-06-09 18:35:25 +02004870 * Decrypt the text in "data" if it points to an encrypted data block.
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02004871 */
4872 void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01004873ml_decrypt_data(
4874 memfile_T *mfp,
4875 char_u *data,
Bram Moolenaar8767f522016-07-01 17:17:39 +02004876 off_T offset,
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01004877 unsigned size)
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02004878{
4879 DATA_BL *dp = (DATA_BL *)data;
4880 char_u *head_end;
4881 char_u *text_start;
4882 int text_len;
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02004883 cryptstate_T *state;
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02004884
4885 if (dp->db_id == DATA_ID)
4886 {
4887 head_end = (char_u *)(&dp->db_index[dp->db_line_count]);
4888 text_start = (char_u *)dp + dp->db_txt_start;
4889 text_len = dp->db_txt_end - dp->db_txt_start;
4890
4891 if (head_end > text_start || dp->db_txt_start > size
4892 || dp->db_txt_end > size)
4893 return; /* data was messed up */
4894
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02004895 state = ml_crypt_prepare(mfp, offset, TRUE);
Bram Moolenaarbc563362015-06-09 18:35:25 +02004896 if (state != NULL)
4897 {
4898 /* Decrypt the text in place. */
4899 crypt_decode_inplace(state, text_start, text_len);
4900 crypt_free_state(state);
4901 }
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02004902 }
4903}
4904
4905/*
4906 * Prepare for encryption/decryption, using the key, seed and offset.
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02004907 * Return an allocated cryptstate_T *.
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02004908 */
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02004909 static cryptstate_T *
Bram Moolenaar8767f522016-07-01 17:17:39 +02004910ml_crypt_prepare(memfile_T *mfp, off_T offset, int reading)
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02004911{
4912 buf_T *buf = mfp->mf_buffer;
4913 char_u salt[50];
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02004914 int method_nr;
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02004915 char_u *key;
4916 char_u *seed;
4917
4918 if (reading && mfp->mf_old_key != NULL)
4919 {
4920 /* Reading back blocks with the previous key/method/seed. */
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02004921 method_nr = mfp->mf_old_cm;
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02004922 key = mfp->mf_old_key;
4923 seed = mfp->mf_old_seed;
4924 }
4925 else
4926 {
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02004927 method_nr = crypt_get_method_nr(buf);
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02004928 key = buf->b_p_key;
4929 seed = mfp->mf_seed;
4930 }
Bram Moolenaarbc563362015-06-09 18:35:25 +02004931 if (*key == NUL)
4932 return NULL;
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02004933
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02004934 if (method_nr == CRYPT_M_ZIP)
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02004935 {
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02004936 /* For PKzip: Append the offset to the key, so that we use a different
4937 * key for every block. */
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02004938 vim_snprintf((char *)salt, sizeof(salt), "%s%ld", key, (long)offset);
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02004939 return crypt_create(method_nr, salt, NULL, 0, NULL, 0);
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02004940 }
Bram Moolenaar8f4ac012014-08-10 13:38:34 +02004941
4942 /* Using blowfish or better: add salt and seed. We use the byte offset
4943 * of the block for the salt. */
4944 vim_snprintf((char *)salt, sizeof(salt), "%ld", (long)offset);
4945 return crypt_create(method_nr, key, salt, (int)STRLEN(salt),
4946 seed, MF_SEED_LEN);
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02004947}
4948
4949#endif
4950
4951
Bram Moolenaar071d4272004-06-13 20:20:40 +00004952#if defined(FEAT_BYTEOFF) || defined(PROTO)
4953
4954#define MLCS_MAXL 800 /* max no of lines in chunk */
4955#define MLCS_MINL 400 /* should be half of MLCS_MAXL */
4956
4957/*
Bram Moolenaar0ad014c2010-07-25 14:00:46 +02004958 * Keep information for finding byte offset of a line, updtype may be one of:
Bram Moolenaar071d4272004-06-13 20:20:40 +00004959 * ML_CHNK_ADDLINE: Add len to parent chunk, possibly splitting it
4960 * Careful: ML_CHNK_ADDLINE may cause ml_find_line() to be called.
4961 * ML_CHNK_DELLINE: Subtract len from parent chunk, possibly deleting it
4962 * ML_CHNK_UPDLINE: Add len to parent chunk, as a signed entity.
4963 */
4964 static void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01004965ml_updatechunk(
4966 buf_T *buf,
4967 linenr_T line,
4968 long len,
4969 int updtype)
Bram Moolenaar071d4272004-06-13 20:20:40 +00004970{
4971 static buf_T *ml_upd_lastbuf = NULL;
4972 static linenr_T ml_upd_lastline;
4973 static linenr_T ml_upd_lastcurline;
4974 static int ml_upd_lastcurix;
4975
4976 linenr_T curline = ml_upd_lastcurline;
4977 int curix = ml_upd_lastcurix;
4978 long size;
4979 chunksize_T *curchnk;
4980 int rest;
4981 bhdr_T *hp;
4982 DATA_BL *dp;
4983
4984 if (buf->b_ml.ml_usedchunks == -1 || len == 0)
4985 return;
4986 if (buf->b_ml.ml_chunksize == NULL)
4987 {
4988 buf->b_ml.ml_chunksize = (chunksize_T *)
4989 alloc((unsigned)sizeof(chunksize_T) * 100);
4990 if (buf->b_ml.ml_chunksize == NULL)
4991 {
4992 buf->b_ml.ml_usedchunks = -1;
4993 return;
4994 }
4995 buf->b_ml.ml_numchunks = 100;
4996 buf->b_ml.ml_usedchunks = 1;
4997 buf->b_ml.ml_chunksize[0].mlcs_numlines = 1;
4998 buf->b_ml.ml_chunksize[0].mlcs_totalsize = 1;
4999 }
5000
5001 if (updtype == ML_CHNK_UPDLINE && buf->b_ml.ml_line_count == 1)
5002 {
5003 /*
5004 * First line in empty buffer from ml_flush_line() -- reset
5005 */
5006 buf->b_ml.ml_usedchunks = 1;
5007 buf->b_ml.ml_chunksize[0].mlcs_numlines = 1;
5008 buf->b_ml.ml_chunksize[0].mlcs_totalsize =
5009 (long)STRLEN(buf->b_ml.ml_line_ptr) + 1;
5010 return;
5011 }
5012
5013 /*
5014 * Find chunk that our line belongs to, curline will be at start of the
5015 * chunk.
5016 */
5017 if (buf != ml_upd_lastbuf || line != ml_upd_lastline + 1
5018 || updtype != ML_CHNK_ADDLINE)
5019 {
5020 for (curline = 1, curix = 0;
5021 curix < buf->b_ml.ml_usedchunks - 1
5022 && line >= curline + buf->b_ml.ml_chunksize[curix].mlcs_numlines;
5023 curix++)
5024 {
5025 curline += buf->b_ml.ml_chunksize[curix].mlcs_numlines;
5026 }
5027 }
5028 else if (line >= curline + buf->b_ml.ml_chunksize[curix].mlcs_numlines
5029 && curix < buf->b_ml.ml_usedchunks - 1)
5030 {
5031 /* Adjust cached curix & curline */
5032 curline += buf->b_ml.ml_chunksize[curix].mlcs_numlines;
5033 curix++;
5034 }
5035 curchnk = buf->b_ml.ml_chunksize + curix;
5036
5037 if (updtype == ML_CHNK_DELLINE)
Bram Moolenaar5a6404c2006-11-01 17:12:57 +00005038 len = -len;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005039 curchnk->mlcs_totalsize += len;
5040 if (updtype == ML_CHNK_ADDLINE)
5041 {
5042 curchnk->mlcs_numlines++;
5043
5044 /* May resize here so we don't have to do it in both cases below */
5045 if (buf->b_ml.ml_usedchunks + 1 >= buf->b_ml.ml_numchunks)
5046 {
Bram Moolenaar9abd5c62015-02-10 18:34:01 +01005047 chunksize_T *t_chunksize = buf->b_ml.ml_chunksize;
5048
Bram Moolenaar071d4272004-06-13 20:20:40 +00005049 buf->b_ml.ml_numchunks = buf->b_ml.ml_numchunks * 3 / 2;
5050 buf->b_ml.ml_chunksize = (chunksize_T *)
5051 vim_realloc(buf->b_ml.ml_chunksize,
5052 sizeof(chunksize_T) * buf->b_ml.ml_numchunks);
5053 if (buf->b_ml.ml_chunksize == NULL)
5054 {
5055 /* Hmmmm, Give up on offset for this buffer */
Bram Moolenaar9abd5c62015-02-10 18:34:01 +01005056 vim_free(t_chunksize);
Bram Moolenaar071d4272004-06-13 20:20:40 +00005057 buf->b_ml.ml_usedchunks = -1;
5058 return;
5059 }
5060 }
5061
5062 if (buf->b_ml.ml_chunksize[curix].mlcs_numlines >= MLCS_MAXL)
5063 {
5064 int count; /* number of entries in block */
5065 int idx;
5066 int text_end;
5067 int linecnt;
5068
5069 mch_memmove(buf->b_ml.ml_chunksize + curix + 1,
5070 buf->b_ml.ml_chunksize + curix,
5071 (buf->b_ml.ml_usedchunks - curix) *
5072 sizeof(chunksize_T));
Bram Moolenaar9439cdd2009-04-22 13:39:36 +00005073 /* Compute length of first half of lines in the split chunk */
Bram Moolenaar071d4272004-06-13 20:20:40 +00005074 size = 0;
5075 linecnt = 0;
5076 while (curline < buf->b_ml.ml_line_count
5077 && linecnt < MLCS_MINL)
5078 {
5079 if ((hp = ml_find_line(buf, curline, ML_FIND)) == NULL)
5080 {
5081 buf->b_ml.ml_usedchunks = -1;
5082 return;
5083 }
5084 dp = (DATA_BL *)(hp->bh_data);
5085 count = (long)(buf->b_ml.ml_locked_high) -
5086 (long)(buf->b_ml.ml_locked_low) + 1;
5087 idx = curline - buf->b_ml.ml_locked_low;
5088 curline = buf->b_ml.ml_locked_high + 1;
5089 if (idx == 0)/* first line in block, text at the end */
5090 text_end = dp->db_txt_end;
5091 else
5092 text_end = ((dp->db_index[idx - 1]) & DB_INDEX_MASK);
5093 /* Compute index of last line to use in this MEMLINE */
5094 rest = count - idx;
5095 if (linecnt + rest > MLCS_MINL)
5096 {
5097 idx += MLCS_MINL - linecnt - 1;
5098 linecnt = MLCS_MINL;
5099 }
5100 else
5101 {
5102 idx = count - 1;
5103 linecnt += rest;
5104 }
5105 size += text_end - ((dp->db_index[idx]) & DB_INDEX_MASK);
5106 }
5107 buf->b_ml.ml_chunksize[curix].mlcs_numlines = linecnt;
5108 buf->b_ml.ml_chunksize[curix + 1].mlcs_numlines -= linecnt;
5109 buf->b_ml.ml_chunksize[curix].mlcs_totalsize = size;
5110 buf->b_ml.ml_chunksize[curix + 1].mlcs_totalsize -= size;
5111 buf->b_ml.ml_usedchunks++;
5112 ml_upd_lastbuf = NULL; /* Force recalc of curix & curline */
5113 return;
5114 }
5115 else if (buf->b_ml.ml_chunksize[curix].mlcs_numlines >= MLCS_MINL
5116 && curix == buf->b_ml.ml_usedchunks - 1
5117 && buf->b_ml.ml_line_count - line <= 1)
5118 {
5119 /*
5120 * We are in the last chunk and it is cheap to crate a new one
5121 * after this. Do it now to avoid the loop above later on
5122 */
5123 curchnk = buf->b_ml.ml_chunksize + curix + 1;
5124 buf->b_ml.ml_usedchunks++;
5125 if (line == buf->b_ml.ml_line_count)
5126 {
5127 curchnk->mlcs_numlines = 0;
5128 curchnk->mlcs_totalsize = 0;
5129 }
5130 else
5131 {
5132 /*
5133 * Line is just prior to last, move count for last
5134 * This is the common case when loading a new file
5135 */
5136 hp = ml_find_line(buf, buf->b_ml.ml_line_count, ML_FIND);
5137 if (hp == NULL)
5138 {
5139 buf->b_ml.ml_usedchunks = -1;
5140 return;
5141 }
5142 dp = (DATA_BL *)(hp->bh_data);
5143 if (dp->db_line_count == 1)
5144 rest = dp->db_txt_end - dp->db_txt_start;
5145 else
5146 rest =
5147 ((dp->db_index[dp->db_line_count - 2]) & DB_INDEX_MASK)
5148 - dp->db_txt_start;
5149 curchnk->mlcs_totalsize = rest;
5150 curchnk->mlcs_numlines = 1;
5151 curchnk[-1].mlcs_totalsize -= rest;
5152 curchnk[-1].mlcs_numlines -= 1;
5153 }
5154 }
5155 }
5156 else if (updtype == ML_CHNK_DELLINE)
5157 {
5158 curchnk->mlcs_numlines--;
5159 ml_upd_lastbuf = NULL; /* Force recalc of curix & curline */
5160 if (curix < (buf->b_ml.ml_usedchunks - 1)
5161 && (curchnk->mlcs_numlines + curchnk[1].mlcs_numlines)
5162 <= MLCS_MINL)
5163 {
5164 curix++;
5165 curchnk = buf->b_ml.ml_chunksize + curix;
5166 }
5167 else if (curix == 0 && curchnk->mlcs_numlines <= 0)
5168 {
5169 buf->b_ml.ml_usedchunks--;
5170 mch_memmove(buf->b_ml.ml_chunksize, buf->b_ml.ml_chunksize + 1,
5171 buf->b_ml.ml_usedchunks * sizeof(chunksize_T));
5172 return;
5173 }
5174 else if (curix == 0 || (curchnk->mlcs_numlines > 10
5175 && (curchnk->mlcs_numlines + curchnk[-1].mlcs_numlines)
5176 > MLCS_MINL))
5177 {
5178 return;
5179 }
5180
5181 /* Collapse chunks */
5182 curchnk[-1].mlcs_numlines += curchnk->mlcs_numlines;
5183 curchnk[-1].mlcs_totalsize += curchnk->mlcs_totalsize;
5184 buf->b_ml.ml_usedchunks--;
5185 if (curix < buf->b_ml.ml_usedchunks)
5186 {
5187 mch_memmove(buf->b_ml.ml_chunksize + curix,
5188 buf->b_ml.ml_chunksize + curix + 1,
5189 (buf->b_ml.ml_usedchunks - curix) *
5190 sizeof(chunksize_T));
5191 }
5192 return;
5193 }
5194 ml_upd_lastbuf = buf;
5195 ml_upd_lastline = line;
5196 ml_upd_lastcurline = curline;
5197 ml_upd_lastcurix = curix;
5198}
5199
5200/*
5201 * Find offset for line or line with offset.
Bram Moolenaar5313dcb2005-02-22 08:56:13 +00005202 * Find line with offset if "lnum" is 0; return remaining offset in offp
5203 * Find offset of line if "lnum" > 0
Bram Moolenaar071d4272004-06-13 20:20:40 +00005204 * return -1 if information is not available
5205 */
5206 long
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01005207ml_find_line_or_offset(buf_T *buf, linenr_T lnum, long *offp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005208{
5209 linenr_T curline;
5210 int curix;
5211 long size;
5212 bhdr_T *hp;
5213 DATA_BL *dp;
5214 int count; /* number of entries in block */
5215 int idx;
5216 int start_idx;
5217 int text_end;
5218 long offset;
5219 int len;
5220 int ffdos = (get_fileformat(buf) == EOL_DOS);
5221 int extra = 0;
5222
Bram Moolenaar5313dcb2005-02-22 08:56:13 +00005223 /* take care of cached line first */
5224 ml_flush_line(curbuf);
5225
Bram Moolenaar071d4272004-06-13 20:20:40 +00005226 if (buf->b_ml.ml_usedchunks == -1
5227 || buf->b_ml.ml_chunksize == NULL
Bram Moolenaar5313dcb2005-02-22 08:56:13 +00005228 || lnum < 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005229 return -1;
5230
5231 if (offp == NULL)
5232 offset = 0;
5233 else
5234 offset = *offp;
Bram Moolenaar5313dcb2005-02-22 08:56:13 +00005235 if (lnum == 0 && offset <= 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005236 return 1; /* Not a "find offset" and offset 0 _must_ be in line 1 */
5237 /*
5238 * Find the last chunk before the one containing our line. Last chunk is
5239 * special because it will never qualify
5240 */
5241 curline = 1;
5242 curix = size = 0;
5243 while (curix < buf->b_ml.ml_usedchunks - 1
Bram Moolenaar5313dcb2005-02-22 08:56:13 +00005244 && ((lnum != 0
5245 && lnum >= curline + buf->b_ml.ml_chunksize[curix].mlcs_numlines)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005246 || (offset != 0
5247 && offset > size + buf->b_ml.ml_chunksize[curix].mlcs_totalsize
5248 + ffdos * buf->b_ml.ml_chunksize[curix].mlcs_numlines)))
5249 {
5250 curline += buf->b_ml.ml_chunksize[curix].mlcs_numlines;
5251 size += buf->b_ml.ml_chunksize[curix].mlcs_totalsize;
5252 if (offset && ffdos)
5253 size += buf->b_ml.ml_chunksize[curix].mlcs_numlines;
5254 curix++;
5255 }
5256
Bram Moolenaar5313dcb2005-02-22 08:56:13 +00005257 while ((lnum != 0 && curline < lnum) || (offset != 0 && size < offset))
Bram Moolenaar071d4272004-06-13 20:20:40 +00005258 {
5259 if (curline > buf->b_ml.ml_line_count
5260 || (hp = ml_find_line(buf, curline, ML_FIND)) == NULL)
5261 return -1;
5262 dp = (DATA_BL *)(hp->bh_data);
5263 count = (long)(buf->b_ml.ml_locked_high) -
5264 (long)(buf->b_ml.ml_locked_low) + 1;
5265 start_idx = idx = curline - buf->b_ml.ml_locked_low;
5266 if (idx == 0)/* first line in block, text at the end */
5267 text_end = dp->db_txt_end;
5268 else
5269 text_end = ((dp->db_index[idx - 1]) & DB_INDEX_MASK);
5270 /* Compute index of last line to use in this MEMLINE */
Bram Moolenaar5313dcb2005-02-22 08:56:13 +00005271 if (lnum != 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005272 {
Bram Moolenaar5313dcb2005-02-22 08:56:13 +00005273 if (curline + (count - idx) >= lnum)
5274 idx += lnum - curline - 1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00005275 else
5276 idx = count - 1;
5277 }
5278 else
5279 {
5280 extra = 0;
5281 while (offset >= size
5282 + text_end - (int)((dp->db_index[idx]) & DB_INDEX_MASK)
5283 + ffdos)
5284 {
5285 if (ffdos)
5286 size++;
5287 if (idx == count - 1)
5288 {
5289 extra = 1;
5290 break;
5291 }
5292 idx++;
5293 }
5294 }
5295 len = text_end - ((dp->db_index[idx]) & DB_INDEX_MASK);
5296 size += len;
5297 if (offset != 0 && size >= offset)
5298 {
5299 if (size + ffdos == offset)
5300 *offp = 0;
5301 else if (idx == start_idx)
5302 *offp = offset - size + len;
5303 else
5304 *offp = offset - size + len
5305 - (text_end - ((dp->db_index[idx - 1]) & DB_INDEX_MASK));
5306 curline += idx - start_idx + extra;
5307 if (curline > buf->b_ml.ml_line_count)
5308 return -1; /* exactly one byte beyond the end */
5309 return curline;
5310 }
5311 curline = buf->b_ml.ml_locked_high + 1;
5312 }
5313
Bram Moolenaar5313dcb2005-02-22 08:56:13 +00005314 if (lnum != 0)
Bram Moolenaar5eb86f92004-07-26 12:53:41 +00005315 {
5316 /* Count extra CR characters. */
5317 if (ffdos)
Bram Moolenaar5313dcb2005-02-22 08:56:13 +00005318 size += lnum - 1;
Bram Moolenaar5eb86f92004-07-26 12:53:41 +00005319
Bram Moolenaar34d72d42015-07-17 14:18:08 +02005320 /* Don't count the last line break if 'noeol' and ('bin' or
5321 * 'nofixeol'). */
5322 if ((!buf->b_p_fixeol || buf->b_p_bin) && !buf->b_p_eol
Bram Moolenaarc26f7c62018-08-20 22:53:04 +02005323 && lnum > buf->b_ml.ml_line_count)
Bram Moolenaar5eb86f92004-07-26 12:53:41 +00005324 size -= ffdos + 1;
5325 }
5326
Bram Moolenaar071d4272004-06-13 20:20:40 +00005327 return size;
5328}
5329
5330/*
5331 * Goto byte in buffer with offset 'cnt'.
5332 */
5333 void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01005334goto_byte(long cnt)
Bram Moolenaar071d4272004-06-13 20:20:40 +00005335{
5336 long boff = cnt;
5337 linenr_T lnum;
5338
5339 ml_flush_line(curbuf); /* cached line may be dirty */
5340 setpcmark();
5341 if (boff)
5342 --boff;
5343 lnum = ml_find_line_or_offset(curbuf, (linenr_T)0, &boff);
5344 if (lnum < 1) /* past the end */
5345 {
5346 curwin->w_cursor.lnum = curbuf->b_ml.ml_line_count;
5347 curwin->w_curswant = MAXCOL;
5348 coladvance((colnr_T)MAXCOL);
5349 }
5350 else
5351 {
5352 curwin->w_cursor.lnum = lnum;
5353 curwin->w_cursor.col = (colnr_T)boff;
Bram Moolenaar943d2b52005-12-02 00:50:49 +00005354# ifdef FEAT_VIRTUALEDIT
5355 curwin->w_cursor.coladd = 0;
5356# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00005357 curwin->w_set_curswant = TRUE;
5358 }
5359 check_cursor();
5360
5361# ifdef FEAT_MBYTE
5362 /* Make sure the cursor is on the first byte of a multi-byte char. */
5363 if (has_mbyte)
5364 mb_adjust_cursor();
5365# endif
5366}
5367#endif