blob: 26890317752ee37999a63a0b47f92ddb8e820c0e [file] [log] [blame]
Bram Moolenaaredf3f972016-08-29 22:49:24 +02001/* vi:set ts=8 sts=4 sw=4 noet:
Bram Moolenaar071d4272004-06-13 20:20:40 +00002 *
3 * VIM - Vi IMproved by Bram Moolenaar
4 *
5 * Do ":help uganda" in Vim to read copying and usage conditions.
6 * Do ":help credits" in Vim to see a list of people who contributed.
7 * See README.txt for an overview of the Vim source code.
8 */
9
10/*
11 * memfile.c: Contains the functions for handling blocks of memory which can
12 * be stored in a file. This is the implementation of a sort of virtual memory.
13 *
14 * A memfile consists of a sequence of blocks. The blocks numbered from 0
15 * upwards have been assigned a place in the actual file. The block number
16 * is equal to the page number in the file. The
17 * blocks with negative numbers are currently in memory only. They can be
18 * assigned a place in the file when too much memory is being used. At that
19 * moment they get a new, positive, number. A list is used for translation of
20 * negative to positive numbers.
21 *
22 * The size of a block is a multiple of a page size, normally the page size of
23 * the device the file is on. Most blocks are 1 page long. A Block of multiple
24 * pages is used for a line that does not fit in a single page.
25 *
26 * Each block can be in memory and/or in a file. The block stays in memory
27 * as long as it is locked. If it is no longer locked it can be swapped out to
28 * the file. It is only written to the file if it has been changed.
29 *
30 * Under normal operation the file is created when opening the memory file and
31 * deleted when closing the memory file. Only with recovery an existing memory
32 * file is opened.
33 */
34
Bram Moolenaar071d4272004-06-13 20:20:40 +000035#include "vim.h"
36
Bram Moolenaar071d4272004-06-13 20:20:40 +000037/*
38 * Some systems have the page size in statfs.f_bsize, some in stat.st_blksize
39 */
40#ifdef HAVE_ST_BLKSIZE
41# define STATFS stat
42# define F_BSIZE st_blksize
43# define fstatfs(fd, buf, len, nul) mch_fstat((fd), (buf))
44#else
45# ifdef HAVE_SYS_STATFS_H
46# include <sys/statfs.h>
47# define STATFS statfs
48# define F_BSIZE f_bsize
Bram Moolenaar071d4272004-06-13 20:20:40 +000049# endif
50#endif
51
52/*
53 * for Amiga Dos 2.0x we use Flush
54 */
55#ifdef AMIGA
56# ifdef FEAT_ARP
Bram Moolenaar4ba37b52019-12-04 21:57:43 +010057extern int dos2; // this is in os_amiga.c
Bram Moolenaar071d4272004-06-13 20:20:40 +000058# endif
59# ifdef SASC
Bram Moolenaar4ba37b52019-12-04 21:57:43 +010060# include <ios1.h> // for chkufb()
Bram Moolenaar071d4272004-06-13 20:20:40 +000061# endif
62#endif
63
Bram Moolenaar4ba37b52019-12-04 21:57:43 +010064#define MEMFILE_PAGE_SIZE 4096 // default page size
Bram Moolenaar071d4272004-06-13 20:20:40 +000065
Bram Moolenaar4ba37b52019-12-04 21:57:43 +010066static long_u total_mem_used = 0; // total memory used for memfiles
Bram Moolenaar071d4272004-06-13 20:20:40 +000067
Bram Moolenaar92b8b2d2016-01-29 22:36:45 +010068static void mf_ins_hash(memfile_T *, bhdr_T *);
69static void mf_rem_hash(memfile_T *, bhdr_T *);
70static bhdr_T *mf_find_hash(memfile_T *, blocknr_T);
71static void mf_ins_used(memfile_T *, bhdr_T *);
72static void mf_rem_used(memfile_T *, bhdr_T *);
73static bhdr_T *mf_release(memfile_T *, int);
74static bhdr_T *mf_alloc_bhdr(memfile_T *, int);
75static void mf_free_bhdr(bhdr_T *);
76static void mf_ins_free(memfile_T *, bhdr_T *);
77static bhdr_T *mf_rem_free(memfile_T *);
78static int mf_read(memfile_T *, bhdr_T *);
79static int mf_write(memfile_T *, bhdr_T *);
Bram Moolenaar8767f522016-07-01 17:17:39 +020080static int mf_write_block(memfile_T *mfp, bhdr_T *hp, off_T offset, unsigned size);
Bram Moolenaar92b8b2d2016-01-29 22:36:45 +010081static int mf_trans_add(memfile_T *, bhdr_T *);
82static void mf_do_open(memfile_T *, char_u *, int);
83static void mf_hash_init(mf_hashtab_T *);
84static void mf_hash_free(mf_hashtab_T *);
85static void mf_hash_free_all(mf_hashtab_T *);
86static mf_hashitem_T *mf_hash_find(mf_hashtab_T *, blocknr_T);
87static void mf_hash_add_item(mf_hashtab_T *, mf_hashitem_T *);
88static void mf_hash_rem_item(mf_hashtab_T *, mf_hashitem_T *);
89static int mf_hash_grow(mf_hashtab_T *);
Bram Moolenaar071d4272004-06-13 20:20:40 +000090
91/*
92 * The functions for using a memfile:
93 *
94 * mf_open() open a new or existing memfile
95 * mf_open_file() open a swap file for an existing memfile
96 * mf_close() close (and delete) a memfile
97 * mf_new() create a new block in a memfile and lock it
98 * mf_get() get an existing block and lock it
99 * mf_put() unlock a block, may be marked for writing
100 * mf_free() remove a block
101 * mf_sync() sync changed parts of memfile to disk
102 * mf_release_all() release as much memory as possible
103 * mf_trans_del() may translate negative to positive block number
104 * mf_fullname() make file name full path (use before first :cd)
105 */
106
107/*
108 * Open an existing or new memory block file.
109 *
110 * fname: name of file to use (NULL means no file at all)
111 * Note: fname must have been allocated, it is not copied!
112 * If opening the file fails, fname is freed.
113 * flags: flags for open() call
114 *
115 * If fname != NULL and file cannot be opened, fail.
116 *
117 * return value: identifier for this memory block file.
118 */
119 memfile_T *
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100120mf_open(char_u *fname, int flags)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000121{
122 memfile_T *mfp;
Bram Moolenaar8767f522016-07-01 17:17:39 +0200123 off_T size;
Bram Moolenaara03dbed2013-05-23 22:27:03 +0200124#if defined(STATFS) && defined(UNIX) && !defined(__QNX__) && !defined(__minix)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000125# define USE_FSTATFS
126 struct STATFS stf;
127#endif
128
Bram Moolenaarc799fe22019-05-28 23:08:19 +0200129 if ((mfp = ALLOC_ONE(memfile_T)) == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000130 return NULL;
131
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100132 if (fname == NULL) // no file for this memfile, use memory only
Bram Moolenaar071d4272004-06-13 20:20:40 +0000133 {
134 mfp->mf_fname = NULL;
135 mfp->mf_ffname = NULL;
136 mfp->mf_fd = -1;
137 }
138 else
139 {
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100140 mf_do_open(mfp, fname, flags); // try to open the file
Bram Moolenaar071d4272004-06-13 20:20:40 +0000141
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100142 // if the file cannot be opened, return here
Bram Moolenaar071d4272004-06-13 20:20:40 +0000143 if (mfp->mf_fd < 0)
144 {
145 vim_free(mfp);
146 return NULL;
147 }
148 }
149
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100150 mfp->mf_free_first = NULL; // free list is empty
151 mfp->mf_used_first = NULL; // used list is empty
Bram Moolenaar071d4272004-06-13 20:20:40 +0000152 mfp->mf_used_last = NULL;
153 mfp->mf_dirty = FALSE;
154 mfp->mf_used_count = 0;
Bram Moolenaarb05b10a2011-03-22 18:10:45 +0100155 mf_hash_init(&mfp->mf_hash);
156 mf_hash_init(&mfp->mf_trans);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000157 mfp->mf_page_size = MEMFILE_PAGE_SIZE;
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200158#ifdef FEAT_CRYPT
159 mfp->mf_old_key = NULL;
160#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000161
162#ifdef USE_FSTATFS
163 /*
164 * Try to set the page size equal to the block size of the device.
165 * Speeds up I/O a lot.
166 * When recovering, the actual block size will be retrieved from block 0
167 * in ml_recover(). The size used here may be wrong, therefore
168 * mf_blocknr_max must be rounded up.
169 */
170 if (mfp->mf_fd >= 0
171 && fstatfs(mfp->mf_fd, &stf, sizeof(struct statfs), 0) == 0
172 && stf.F_BSIZE >= MIN_SWAP_PAGE_SIZE
173 && stf.F_BSIZE <= MAX_SWAP_PAGE_SIZE)
174 mfp->mf_page_size = stf.F_BSIZE;
175#endif
176
177 if (mfp->mf_fd < 0 || (flags & (O_TRUNC|O_EXCL))
Bram Moolenaar8767f522016-07-01 17:17:39 +0200178 || (size = vim_lseek(mfp->mf_fd, (off_T)0L, SEEK_END)) <= 0)
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100179 mfp->mf_blocknr_max = 0; // no file or empty file
Bram Moolenaar071d4272004-06-13 20:20:40 +0000180 else
181 mfp->mf_blocknr_max = (blocknr_T)((size + mfp->mf_page_size - 1)
182 / mfp->mf_page_size);
183 mfp->mf_blocknr_min = -1;
184 mfp->mf_neg_count = 0;
185 mfp->mf_infile_count = mfp->mf_blocknr_max;
Bram Moolenaarfe265ff2007-05-11 18:15:45 +0000186
187 /*
188 * Compute maximum number of pages ('maxmem' is in Kbyte):
189 * 'mammem' * 1Kbyte / page-size-in-bytes.
190 * Avoid overflow by first reducing page size as much as possible.
191 */
192 {
193 int shift = 10;
194 unsigned page_size = mfp->mf_page_size;
195
196 while (shift > 0 && (page_size & 1) == 0)
197 {
198 page_size = page_size >> 1;
199 --shift;
200 }
201 mfp->mf_used_count_max = (p_mm << shift) / page_size;
202 if (mfp->mf_used_count_max < 10)
203 mfp->mf_used_count_max = 10;
204 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000205
206 return mfp;
207}
208
209/*
210 * Open a file for an existing memfile. Used when updatecount set from 0 to
211 * some value.
212 * If the file already exists, this fails.
213 * "fname" is the name of file to use (NULL means no file at all)
214 * Note: "fname" must have been allocated, it is not copied! If opening the
215 * file fails, "fname" is freed.
216 *
217 * return value: FAIL if file could not be opened, OK otherwise
218 */
219 int
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100220mf_open_file(memfile_T *mfp, char_u *fname)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000221{
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100222 mf_do_open(mfp, fname, O_RDWR|O_CREAT|O_EXCL); // try to open the file
Bram Moolenaar071d4272004-06-13 20:20:40 +0000223
224 if (mfp->mf_fd < 0)
225 return FAIL;
226
227 mfp->mf_dirty = TRUE;
228 return OK;
229}
230
231/*
Bram Moolenaar945e2db2010-06-05 17:43:32 +0200232 * Close a memory file and delete the associated file if 'del_file' is TRUE.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000233 */
234 void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100235mf_close(memfile_T *mfp, int del_file)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000236{
237 bhdr_T *hp, *nextp;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000238
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100239 if (mfp == NULL) // safety check
Bram Moolenaar071d4272004-06-13 20:20:40 +0000240 return;
241 if (mfp->mf_fd >= 0)
242 {
243 if (close(mfp->mf_fd) < 0)
Bram Moolenaar12f3c1b2021-12-05 21:46:34 +0000244 emsg(_(e_close_error_on_swap_file));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000245 }
246 if (del_file && mfp->mf_fname != NULL)
247 mch_remove(mfp->mf_fname);
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100248 // free entries in used list
Bram Moolenaar071d4272004-06-13 20:20:40 +0000249 for (hp = mfp->mf_used_first; hp != NULL; hp = nextp)
250 {
=?UTF-8?q?Dundar=20G=C3=B6c?=d5cec1f2022-01-29 15:19:23 +0000251 total_mem_used -= (long_u)hp->bh_page_count * mfp->mf_page_size;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000252 nextp = hp->bh_next;
253 mf_free_bhdr(hp);
254 }
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100255 while (mfp->mf_free_first != NULL) // free entries in free list
Bram Moolenaar071d4272004-06-13 20:20:40 +0000256 vim_free(mf_rem_free(mfp));
Bram Moolenaarb05b10a2011-03-22 18:10:45 +0100257 mf_hash_free(&mfp->mf_hash);
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100258 mf_hash_free_all(&mfp->mf_trans); // free hashtable and its items
Bram Moolenaar071d4272004-06-13 20:20:40 +0000259 vim_free(mfp->mf_fname);
260 vim_free(mfp->mf_ffname);
261 vim_free(mfp);
262}
263
264/*
265 * Close the swap file for a memfile. Used when 'swapfile' is reset.
266 */
267 void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100268mf_close_file(
269 buf_T *buf,
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100270 int getlines) // get all lines into memory?
Bram Moolenaar071d4272004-06-13 20:20:40 +0000271{
272 memfile_T *mfp;
273 linenr_T lnum;
274
275 mfp = buf->b_ml.ml_mfp;
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100276 if (mfp == NULL || mfp->mf_fd < 0) // nothing to close
Bram Moolenaar071d4272004-06-13 20:20:40 +0000277 return;
278
279 if (getlines)
280 {
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100281 // get all blocks in memory by accessing all lines (clumsy!)
Bram Moolenaar47b8b152007-02-07 02:41:57 +0000282 mf_dont_release = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000283 for (lnum = 1; lnum <= buf->b_ml.ml_line_count; ++lnum)
284 (void)ml_get_buf(buf, lnum, FALSE);
Bram Moolenaar47b8b152007-02-07 02:41:57 +0000285 mf_dont_release = FALSE;
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100286 // TODO: should check if all blocks are really in core
Bram Moolenaar071d4272004-06-13 20:20:40 +0000287 }
288
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100289 if (close(mfp->mf_fd) < 0) // close the file
Bram Moolenaar12f3c1b2021-12-05 21:46:34 +0000290 emsg(_(e_close_error_on_swap_file));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000291 mfp->mf_fd = -1;
292
293 if (mfp->mf_fname != NULL)
294 {
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100295 mch_remove(mfp->mf_fname); // delete the swap file
Bram Moolenaard23a8232018-02-10 18:45:26 +0100296 VIM_CLEAR(mfp->mf_fname);
297 VIM_CLEAR(mfp->mf_ffname);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000298 }
299}
300
301/*
302 * Set new size for a memfile. Used when block 0 of a swapfile has been read
303 * and the size it indicates differs from what was guessed.
304 */
305 void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100306mf_new_page_size(memfile_T *mfp, unsigned new_size)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000307{
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100308 // Correct the memory used for block 0 to the new size, because it will be
309 // freed with that size later on.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000310 total_mem_used += new_size - mfp->mf_page_size;
311 mfp->mf_page_size = new_size;
312}
313
314/*
315 * get a new block
316 *
317 * negative: TRUE if negative block number desired (data block)
318 */
319 bhdr_T *
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100320mf_new(memfile_T *mfp, int negative, int page_count)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000321{
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100322 bhdr_T *hp; // new bhdr_T
323 bhdr_T *freep; // first block in free list
Bram Moolenaar17e79192007-05-06 14:15:36 +0000324 char_u *p;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000325
326 /*
327 * If we reached the maximum size for the used memory blocks, release one
328 * If a bhdr_T is returned, use it and adjust the page_count if necessary.
329 */
330 hp = mf_release(mfp, page_count);
331
332/*
333 * Decide on the number to use:
334 * If there is a free block, use its number.
335 * Otherwise use mf_block_min for a negative number, mf_block_max for
336 * a positive number.
337 */
338 freep = mfp->mf_free_first;
339 if (!negative && freep != NULL && freep->bh_page_count >= page_count)
340 {
341 /*
342 * If the block in the free list has more pages, take only the number
343 * of pages needed and allocate a new bhdr_T with data
344 *
Bram Moolenaar17e79192007-05-06 14:15:36 +0000345 * If the number of pages matches and mf_release() did not return a
346 * bhdr_T, use the bhdr_T from the free list and allocate the data
Bram Moolenaar071d4272004-06-13 20:20:40 +0000347 *
Bram Moolenaar17e79192007-05-06 14:15:36 +0000348 * If the number of pages matches and mf_release() returned a bhdr_T,
Bram Moolenaar071d4272004-06-13 20:20:40 +0000349 * just use the number and free the bhdr_T from the free list
350 */
351 if (freep->bh_page_count > page_count)
352 {
353 if (hp == NULL && (hp = mf_alloc_bhdr(mfp, page_count)) == NULL)
354 return NULL;
355 hp->bh_bnum = freep->bh_bnum;
356 freep->bh_bnum += page_count;
357 freep->bh_page_count -= page_count;
358 }
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100359 else if (hp == NULL) // need to allocate memory for this block
Bram Moolenaar071d4272004-06-13 20:20:40 +0000360 {
=?UTF-8?q?Dundar=20G=C3=B6c?=d5cec1f2022-01-29 15:19:23 +0000361 if ((p = alloc((size_t)mfp->mf_page_size * page_count)) == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000362 return NULL;
363 hp = mf_rem_free(mfp);
364 hp->bh_data = p;
365 }
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100366 else // use the number, remove entry from free list
Bram Moolenaar071d4272004-06-13 20:20:40 +0000367 {
368 freep = mf_rem_free(mfp);
369 hp->bh_bnum = freep->bh_bnum;
370 vim_free(freep);
371 }
372 }
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100373 else // get a new number
Bram Moolenaar071d4272004-06-13 20:20:40 +0000374 {
375 if (hp == NULL && (hp = mf_alloc_bhdr(mfp, page_count)) == NULL)
376 return NULL;
377 if (negative)
378 {
379 hp->bh_bnum = mfp->mf_blocknr_min--;
380 mfp->mf_neg_count++;
381 }
382 else
383 {
384 hp->bh_bnum = mfp->mf_blocknr_max;
385 mfp->mf_blocknr_max += page_count;
386 }
387 }
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100388 hp->bh_flags = BH_LOCKED | BH_DIRTY; // new block is always dirty
Bram Moolenaar071d4272004-06-13 20:20:40 +0000389 mfp->mf_dirty = TRUE;
390 hp->bh_page_count = page_count;
391 mf_ins_used(mfp, hp);
392 mf_ins_hash(mfp, hp);
393
394 /*
395 * Init the data to all zero, to avoid reading uninitialized data.
396 * This also avoids that the passwd file ends up in the swap file!
397 */
Bram Moolenaar945e2db2010-06-05 17:43:32 +0200398 (void)vim_memset((char *)(hp->bh_data), 0,
399 (size_t)mfp->mf_page_size * page_count);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000400
401 return hp;
402}
403
404/*
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200405 * Get existing block "nr" with "page_count" pages.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000406 *
407 * Note: The caller should first check a negative nr with mf_trans_del()
408 */
409 bhdr_T *
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100410mf_get(memfile_T *mfp, blocknr_T nr, int page_count)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000411{
412 bhdr_T *hp;
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100413 // doesn't exist
Bram Moolenaar071d4272004-06-13 20:20:40 +0000414 if (nr >= mfp->mf_blocknr_max || nr <= mfp->mf_blocknr_min)
415 return NULL;
416
417 /*
418 * see if it is in the cache
419 */
420 hp = mf_find_hash(mfp, nr);
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100421 if (hp == NULL) // not in the hash list
Bram Moolenaar071d4272004-06-13 20:20:40 +0000422 {
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100423 if (nr < 0 || nr >= mfp->mf_infile_count) // can't be in the file
Bram Moolenaar071d4272004-06-13 20:20:40 +0000424 return NULL;
425
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100426 // could check here if the block is in the free list
Bram Moolenaar071d4272004-06-13 20:20:40 +0000427
428 /*
429 * Check if we need to flush an existing block.
430 * If so, use that block.
431 * If not, allocate a new block.
432 */
433 hp = mf_release(mfp, page_count);
Bram Moolenaarb67ba032023-04-22 21:14:26 +0100434 if (hp == NULL && page_count > 0)
435 hp = mf_alloc_bhdr(mfp, page_count);
436 if (hp == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000437 return NULL;
438
439 hp->bh_bnum = nr;
440 hp->bh_flags = 0;
441 hp->bh_page_count = page_count;
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100442 if (mf_read(mfp, hp) == FAIL) // cannot read the block!
Bram Moolenaar071d4272004-06-13 20:20:40 +0000443 {
444 mf_free_bhdr(hp);
445 return NULL;
446 }
447 }
448 else
449 {
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100450 mf_rem_used(mfp, hp); // remove from list, insert in front below
Bram Moolenaar071d4272004-06-13 20:20:40 +0000451 mf_rem_hash(mfp, hp);
452 }
453
454 hp->bh_flags |= BH_LOCKED;
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100455 mf_ins_used(mfp, hp); // put in front of used list
456 mf_ins_hash(mfp, hp); // put in front of hash list
Bram Moolenaar071d4272004-06-13 20:20:40 +0000457
458 return hp;
459}
460
461/*
462 * release the block *hp
463 *
464 * dirty: Block must be written to file later
465 * infile: Block should be in file (needed for recovery)
466 *
467 * no return value, function cannot fail
468 */
469 void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100470mf_put(
471 memfile_T *mfp,
472 bhdr_T *hp,
473 int dirty,
474 int infile)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000475{
476 int flags;
477
478 flags = hp->bh_flags;
479
480 if ((flags & BH_LOCKED) == 0)
Bram Moolenaar9a846fb2022-01-01 21:59:18 +0000481 iemsg(_(e_block_was_not_locked));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000482 flags &= ~BH_LOCKED;
483 if (dirty)
484 {
485 flags |= BH_DIRTY;
486 mfp->mf_dirty = TRUE;
487 }
488 hp->bh_flags = flags;
489 if (infile)
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100490 mf_trans_add(mfp, hp); // may translate negative in positive nr
Bram Moolenaar071d4272004-06-13 20:20:40 +0000491}
492
493/*
494 * block *hp is no longer in used, may put it in the free list of memfile *mfp
495 */
496 void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100497mf_free(memfile_T *mfp, bhdr_T *hp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000498{
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100499 vim_free(hp->bh_data); // free the memory
500 mf_rem_hash(mfp, hp); // get *hp out of the hash list
501 mf_rem_used(mfp, hp); // get *hp out of the used list
Bram Moolenaar071d4272004-06-13 20:20:40 +0000502 if (hp->bh_bnum < 0)
503 {
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100504 vim_free(hp); // don't want negative numbers in free list
Bram Moolenaar071d4272004-06-13 20:20:40 +0000505 mfp->mf_neg_count--;
506 }
507 else
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100508 mf_ins_free(mfp, hp); // put *hp in the free list
Bram Moolenaar071d4272004-06-13 20:20:40 +0000509}
510
Bram Moolenaar071d4272004-06-13 20:20:40 +0000511/*
512 * Sync the memory file *mfp to disk.
513 * Flags:
514 * MFS_ALL If not given, blocks with negative numbers are not synced,
515 * even when they are dirty!
516 * MFS_STOP Stop syncing when a character becomes available, but sync at
517 * least one block.
518 * MFS_FLUSH Make sure buffers are flushed to disk, so they will survive a
519 * system crash.
520 * MFS_ZERO Only write block 0.
521 *
522 * Return FAIL for failure, OK otherwise
523 */
524 int
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100525mf_sync(memfile_T *mfp, int flags)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000526{
527 int status;
528 bhdr_T *hp;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000529 int got_int_save = got_int;
530
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100531 if (mfp->mf_fd < 0) // there is no file, nothing to do
Bram Moolenaar071d4272004-06-13 20:20:40 +0000532 {
533 mfp->mf_dirty = FALSE;
534 return FAIL;
535 }
536
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100537 // Only a CTRL-C while writing will break us here, not one typed
538 // previously.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000539 got_int = FALSE;
540
541 /*
542 * sync from last to first (may reduce the probability of an inconsistent
543 * file) If a write fails, it is very likely caused by a full filesystem.
544 * Then we only try to write blocks within the existing file. If that also
545 * fails then we give up.
546 */
547 status = OK;
548 for (hp = mfp->mf_used_last; hp != NULL; hp = hp->bh_prev)
549 if (((flags & MFS_ALL) || hp->bh_bnum >= 0)
550 && (hp->bh_flags & BH_DIRTY)
551 && (status == OK || (hp->bh_bnum >= 0
552 && hp->bh_bnum < mfp->mf_infile_count)))
553 {
554 if ((flags & MFS_ZERO) && hp->bh_bnum != 0)
555 continue;
556 if (mf_write(mfp, hp) == FAIL)
557 {
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100558 if (status == FAIL) // double error: quit syncing
Bram Moolenaar071d4272004-06-13 20:20:40 +0000559 break;
560 status = FAIL;
561 }
562 if (flags & MFS_STOP)
563 {
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100564 // Stop when char available now.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000565 if (ui_char_avail())
566 break;
567 }
568 else
569 ui_breakcheck();
570 if (got_int)
571 break;
572 }
573
574 /*
575 * If the whole list is flushed, the memfile is not dirty anymore.
576 * In case of an error this flag is also set, to avoid trying all the time.
577 */
578 if (hp == NULL || status == FAIL)
579 mfp->mf_dirty = FALSE;
580
581 if ((flags & MFS_FLUSH) && *p_sws != NUL)
582 {
583#if defined(UNIX)
584# ifdef HAVE_FSYNC
585 /*
586 * most Unixes have the very useful fsync() function, just what we need.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000587 */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000588 if (STRCMP(p_sws, "fsync") == 0)
589 {
Bram Moolenaara7870192019-02-14 12:56:36 +0100590 if (vim_fsync(mfp->mf_fd))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000591 status = FAIL;
592 }
593 else
594# endif
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100595 // OpenNT is strictly POSIX (Benzinger)
596 // Tandem/Himalaya NSK-OSS doesn't have sync()
597 // No sync() on Stratus VOS
Bram Moolenaarba17ed62015-02-27 18:25:16 +0100598# if defined(__OPENNT) || defined(__TANDEM) || defined(__VOS__)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000599 fflush(NULL);
600# else
601 sync();
602# endif
603#endif
604#ifdef VMS
605 if (STRCMP(p_sws, "fsync") == 0)
606 {
Bram Moolenaara7870192019-02-14 12:56:36 +0100607 if (vim_fsync(mfp->mf_fd))
Bram Moolenaar071d4272004-06-13 20:20:40 +0000608 status = FAIL;
609 }
610#endif
Bram Moolenaar4f974752019-02-17 17:44:42 +0100611#ifdef MSWIN
Bram Moolenaar0bd40512018-09-21 14:48:53 +0200612 if (_commit(mfp->mf_fd))
613 status = FAIL;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000614#endif
615#ifdef AMIGA
Bram Moolenaar5a6404c2006-11-01 17:12:57 +0000616# if defined(__AROS__) || defined(__amigaos4__)
Bram Moolenaara7870192019-02-14 12:56:36 +0100617 if (vim_fsync(mfp->mf_fd) != 0)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000618 status = FAIL;
619# else
620 /*
621 * Flush() only exists for AmigaDos 2.0.
622 * For 1.3 it should be done with close() + open(), but then the risk
623 * is that the open() may fail and lose the file....
624 */
625# ifdef FEAT_ARP
626 if (dos2)
627# endif
628# ifdef SASC
629 {
630 struct UFB *fp = chkufb(mfp->mf_fd);
631
632 if (fp != NULL)
633 Flush(fp->ufbfh);
634 }
635# else
636# if defined(_DCC) || defined(__GNUC__) || defined(__MORPHOS__)
637 {
Bram Moolenaar89f37272006-09-26 11:48:34 +0000638# if defined(__GNUC__) && !defined(__MORPHOS__) && defined(__libnix__)
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100639 // Have function (in libnix at least),
640 // but ain't got no prototype anywhere.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000641 extern unsigned long fdtofh(int filedescriptor);
642# endif
Bram Moolenaar89f37272006-09-26 11:48:34 +0000643# if !defined(__libnix__)
644 fflush(NULL);
645# else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000646 BPTR fh = (BPTR)fdtofh(mfp->mf_fd);
647
648 if (fh != 0)
649 Flush(fh);
Bram Moolenaar89f37272006-09-26 11:48:34 +0000650# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000651 }
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100652# else // assume Manx
Bram Moolenaar071d4272004-06-13 20:20:40 +0000653 Flush(_devtab[mfp->mf_fd].fd);
654# endif
655# endif
656# endif
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100657#endif // AMIGA
Bram Moolenaar071d4272004-06-13 20:20:40 +0000658 }
659
660 got_int |= got_int_save;
661
662 return status;
663}
664
665/*
Bram Moolenaarc32840f2006-01-14 21:23:38 +0000666 * For all blocks in memory file *mfp that have a positive block number set
667 * the dirty flag. These are blocks that need to be written to a newly
668 * created swapfile.
669 */
670 void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100671mf_set_dirty(memfile_T *mfp)
Bram Moolenaarc32840f2006-01-14 21:23:38 +0000672{
673 bhdr_T *hp;
674
675 for (hp = mfp->mf_used_last; hp != NULL; hp = hp->bh_prev)
676 if (hp->bh_bnum > 0)
677 hp->bh_flags |= BH_DIRTY;
678 mfp->mf_dirty = TRUE;
679}
680
681/*
Bram Moolenaar071d4272004-06-13 20:20:40 +0000682 * insert block *hp in front of hashlist of memfile *mfp
683 */
684 static void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100685mf_ins_hash(memfile_T *mfp, bhdr_T *hp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000686{
Bram Moolenaarb05b10a2011-03-22 18:10:45 +0100687 mf_hash_add_item(&mfp->mf_hash, (mf_hashitem_T *)hp);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000688}
689
690/*
691 * remove block *hp from hashlist of memfile list *mfp
692 */
693 static void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100694mf_rem_hash(memfile_T *mfp, bhdr_T *hp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000695{
Bram Moolenaarb05b10a2011-03-22 18:10:45 +0100696 mf_hash_rem_item(&mfp->mf_hash, (mf_hashitem_T *)hp);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000697}
698
699/*
700 * look in hash lists of memfile *mfp for block header with number 'nr'
701 */
702 static bhdr_T *
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100703mf_find_hash(memfile_T *mfp, blocknr_T nr)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000704{
Bram Moolenaarb05b10a2011-03-22 18:10:45 +0100705 return (bhdr_T *)mf_hash_find(&mfp->mf_hash, nr);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000706}
707
708/*
709 * insert block *hp in front of used list of memfile *mfp
710 */
711 static void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100712mf_ins_used(memfile_T *mfp, bhdr_T *hp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000713{
714 hp->bh_next = mfp->mf_used_first;
715 mfp->mf_used_first = hp;
716 hp->bh_prev = NULL;
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100717 if (hp->bh_next == NULL) // list was empty, adjust last pointer
Bram Moolenaar071d4272004-06-13 20:20:40 +0000718 mfp->mf_used_last = hp;
719 else
720 hp->bh_next->bh_prev = hp;
721 mfp->mf_used_count += hp->bh_page_count;
=?UTF-8?q?Dundar=20G=C3=B6c?=d5cec1f2022-01-29 15:19:23 +0000722 total_mem_used += (long_u)hp->bh_page_count * mfp->mf_page_size;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000723}
724
725/*
726 * remove block *hp from used list of memfile *mfp
727 */
728 static void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100729mf_rem_used(memfile_T *mfp, bhdr_T *hp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000730{
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100731 if (hp->bh_next == NULL) // last block in used list
Bram Moolenaar071d4272004-06-13 20:20:40 +0000732 mfp->mf_used_last = hp->bh_prev;
733 else
734 hp->bh_next->bh_prev = hp->bh_prev;
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100735 if (hp->bh_prev == NULL) // first block in used list
Bram Moolenaar071d4272004-06-13 20:20:40 +0000736 mfp->mf_used_first = hp->bh_next;
737 else
738 hp->bh_prev->bh_next = hp->bh_next;
739 mfp->mf_used_count -= hp->bh_page_count;
=?UTF-8?q?Dundar=20G=C3=B6c?=d5cec1f2022-01-29 15:19:23 +0000740 total_mem_used -= (long_u)hp->bh_page_count * mfp->mf_page_size;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000741}
742
743/*
744 * Release the least recently used block from the used list if the number
745 * of used memory blocks gets to big.
746 *
747 * Return the block header to the caller, including the memory block, so
748 * it can be re-used. Make sure the page_count is right.
Bram Moolenaarbc563362015-06-09 18:35:25 +0200749 *
750 * Returns NULL if no block is released.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000751 */
752 static bhdr_T *
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100753mf_release(memfile_T *mfp, int page_count)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000754{
755 bhdr_T *hp;
756 int need_release;
757 buf_T *buf;
758
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100759 // don't release while in mf_close_file()
Bram Moolenaar47b8b152007-02-07 02:41:57 +0000760 if (mf_dont_release)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000761 return NULL;
762
763 /*
764 * Need to release a block if the number of blocks for this memfile is
765 * higher than the maximum or total memory used is over 'maxmemtot'
766 */
767 need_release = ((mfp->mf_used_count >= mfp->mf_used_count_max)
768 || (total_mem_used >> 10) >= (long_u)p_mmt);
769
770 /*
771 * Try to create a swap file if the amount of memory used is getting too
772 * high.
773 */
774 if (mfp->mf_fd < 0 && need_release && p_uc)
775 {
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100776 // find for which buffer this memfile is
Bram Moolenaar29323592016-07-24 22:04:11 +0200777 FOR_ALL_BUFFERS(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000778 if (buf->b_ml.ml_mfp == mfp)
779 break;
780 if (buf != NULL && buf->b_may_swap)
781 ml_open_file(buf);
782 }
783
784 /*
785 * don't release a block if
786 * there is no file for this memfile
787 * or
788 * the number of blocks for this memfile is lower than the maximum
789 * and
790 * total memory used is not up to 'maxmemtot'
791 */
792 if (mfp->mf_fd < 0 || !need_release)
793 return NULL;
794
795 for (hp = mfp->mf_used_last; hp != NULL; hp = hp->bh_prev)
796 if (!(hp->bh_flags & BH_LOCKED))
797 break;
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100798 if (hp == NULL) // not a single one that can be released
Bram Moolenaar071d4272004-06-13 20:20:40 +0000799 return NULL;
800
801 /*
802 * If the block is dirty, write it.
803 * If the write fails we don't free it.
804 */
805 if ((hp->bh_flags & BH_DIRTY) && mf_write(mfp, hp) == FAIL)
806 return NULL;
807
808 mf_rem_used(mfp, hp);
809 mf_rem_hash(mfp, hp);
810
811 /*
812 * If a bhdr_T is returned, make sure that the page_count of bh_data is
813 * right
814 */
815 if (hp->bh_page_count != page_count)
816 {
Bram Moolenaarb67ba032023-04-22 21:14:26 +0100817 VIM_CLEAR(hp->bh_data);
818 if (page_count > 0)
819 hp->bh_data = alloc((size_t)mfp->mf_page_size * page_count);
820 if (hp->bh_data == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000821 {
822 vim_free(hp);
823 return NULL;
824 }
825 hp->bh_page_count = page_count;
826 }
827 return hp;
828}
829
830/*
831 * release as many blocks as possible
832 * Used in case of out of memory
833 *
834 * return TRUE if any memory was released
835 */
836 int
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100837mf_release_all(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000838{
839 buf_T *buf;
840 memfile_T *mfp;
841 bhdr_T *hp;
842 int retval = FALSE;
843
Bram Moolenaar29323592016-07-24 22:04:11 +0200844 FOR_ALL_BUFFERS(buf)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000845 {
846 mfp = buf->b_ml.ml_mfp;
847 if (mfp != NULL)
848 {
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100849 // If no swap file yet, may open one
Bram Moolenaar071d4272004-06-13 20:20:40 +0000850 if (mfp->mf_fd < 0 && buf->b_may_swap)
851 ml_open_file(buf);
852
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100853 // only if there is a swapfile
Bram Moolenaar071d4272004-06-13 20:20:40 +0000854 if (mfp->mf_fd >= 0)
855 {
856 for (hp = mfp->mf_used_last; hp != NULL; )
857 {
858 if (!(hp->bh_flags & BH_LOCKED)
859 && (!(hp->bh_flags & BH_DIRTY)
860 || mf_write(mfp, hp) != FAIL))
861 {
862 mf_rem_used(mfp, hp);
863 mf_rem_hash(mfp, hp);
864 mf_free_bhdr(hp);
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100865 hp = mfp->mf_used_last; // re-start, list was changed
Bram Moolenaar071d4272004-06-13 20:20:40 +0000866 retval = TRUE;
867 }
868 else
869 hp = hp->bh_prev;
870 }
871 }
872 }
873 }
874 return retval;
875}
876
877/*
Bram Moolenaarb67ba032023-04-22 21:14:26 +0100878 * Allocate a block header and a block of memory for it.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000879 */
880 static bhdr_T *
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100881mf_alloc_bhdr(memfile_T *mfp, int page_count)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000882{
883 bhdr_T *hp;
884
Yegappan Lakshmanane8575982023-01-14 12:32:28 +0000885 if ((hp = ALLOC_ONE(bhdr_T)) == NULL)
886 return NULL;
887
Bram Moolenaarb67ba032023-04-22 21:14:26 +0100888 if ((hp->bh_data = alloc((size_t)mfp->mf_page_size * page_count)) == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000889 {
Yegappan Lakshmanane8575982023-01-14 12:32:28 +0000890 vim_free(hp); // not enough memory
891 return NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000892 }
Yegappan Lakshmanane8575982023-01-14 12:32:28 +0000893 hp->bh_page_count = page_count;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000894 return hp;
895}
896
897/*
Bram Moolenaarb67ba032023-04-22 21:14:26 +0100898 * Free a block header and the block of memory for it.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000899 */
900 static void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100901mf_free_bhdr(bhdr_T *hp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000902{
903 vim_free(hp->bh_data);
904 vim_free(hp);
905}
906
907/*
Bram Moolenaarb67ba032023-04-22 21:14:26 +0100908 * Insert entry *hp in the free list.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000909 */
910 static void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100911mf_ins_free(memfile_T *mfp, bhdr_T *hp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000912{
913 hp->bh_next = mfp->mf_free_first;
914 mfp->mf_free_first = hp;
915}
916
917/*
918 * remove the first entry from the free list and return a pointer to it
919 * Note: caller must check that mfp->mf_free_first is not NULL!
920 */
921 static bhdr_T *
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100922mf_rem_free(memfile_T *mfp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000923{
924 bhdr_T *hp;
925
926 hp = mfp->mf_free_first;
927 mfp->mf_free_first = hp->bh_next;
928 return hp;
929}
930
931/*
932 * read a block from disk
933 *
934 * Return FAIL for failure, OK otherwise
935 */
936 static int
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100937mf_read(memfile_T *mfp, bhdr_T *hp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000938{
Bram Moolenaar8767f522016-07-01 17:17:39 +0200939 off_T offset;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000940 unsigned page_size;
941 unsigned size;
942
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100943 if (mfp->mf_fd < 0) // there is no file, can't read
Bram Moolenaar071d4272004-06-13 20:20:40 +0000944 return FAIL;
945
946 page_size = mfp->mf_page_size;
Bram Moolenaar8767f522016-07-01 17:17:39 +0200947 offset = (off_T)page_size * hp->bh_bnum;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000948 size = page_size * hp->bh_page_count;
Bram Moolenaar8767f522016-07-01 17:17:39 +0200949 if (vim_lseek(mfp->mf_fd, offset, SEEK_SET) != offset)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000950 {
Bram Moolenaar9a846fb2022-01-01 21:59:18 +0000951 PERROR(_(e_seek_error_in_swap_file_read));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000952 return FAIL;
953 }
Bram Moolenaar540fc6f2010-12-17 16:27:16 +0100954 if ((unsigned)read_eintr(mfp->mf_fd, hp->bh_data, size) != size)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000955 {
Bram Moolenaar9a846fb2022-01-01 21:59:18 +0000956 PERROR(_(e_read_error_in_swap_file));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000957 return FAIL;
958 }
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200959
960#ifdef FEAT_CRYPT
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100961 // Decrypt if 'key' is set and this is a data block. And when changing the
962 // key.
Bram Moolenaar4a8c2cf2015-12-19 15:28:18 +0100963 if (*mfp->mf_buffer->b_p_key != NUL || mfp->mf_old_key != NULL)
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200964 ml_decrypt_data(mfp, hp->bh_data, offset, size);
965#endif
966
Bram Moolenaar071d4272004-06-13 20:20:40 +0000967 return OK;
968}
969
970/*
971 * write a block to disk
972 *
973 * Return FAIL for failure, OK otherwise
974 */
975 static int
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100976mf_write(memfile_T *mfp, bhdr_T *hp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000977{
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100978 off_T offset; // offset in the file
979 blocknr_T nr; // block nr which is being written
Bram Moolenaar071d4272004-06-13 20:20:40 +0000980 bhdr_T *hp2;
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100981 unsigned page_size; // number of bytes in a page
982 unsigned page_count; // number of pages written
983 unsigned size; // number of bytes written
Bram Moolenaar071d4272004-06-13 20:20:40 +0000984
Bram Moolenaarb58a4b92019-05-27 23:36:21 +0200985 if (mfp->mf_fd < 0 && !mfp->mf_reopen)
986 // there is no file and there was no file, can't write
Bram Moolenaar071d4272004-06-13 20:20:40 +0000987 return FAIL;
988
Bram Moolenaar4ba37b52019-12-04 21:57:43 +0100989 if (hp->bh_bnum < 0) // must assign file block number
Bram Moolenaar071d4272004-06-13 20:20:40 +0000990 if (mf_trans_add(mfp, hp) == FAIL)
991 return FAIL;
992
993 page_size = mfp->mf_page_size;
994
995 /*
996 * We don't want gaps in the file. Write the blocks in front of *hp
997 * to extend the file.
998 * If block 'mf_infile_count' is not in the hash list, it has been
999 * freed. Fill the space in the file with data from the current block.
1000 */
1001 for (;;)
1002 {
Bram Moolenaarb58a4b92019-05-27 23:36:21 +02001003 int attempt;
1004
Bram Moolenaar071d4272004-06-13 20:20:40 +00001005 nr = hp->bh_bnum;
Bram Moolenaar4ba37b52019-12-04 21:57:43 +01001006 if (nr > mfp->mf_infile_count) // beyond end of file
Bram Moolenaar071d4272004-06-13 20:20:40 +00001007 {
1008 nr = mfp->mf_infile_count;
Bram Moolenaar4ba37b52019-12-04 21:57:43 +01001009 hp2 = mf_find_hash(mfp, nr); // NULL caught below
Bram Moolenaar071d4272004-06-13 20:20:40 +00001010 }
1011 else
1012 hp2 = hp;
1013
Bram Moolenaar8767f522016-07-01 17:17:39 +02001014 offset = (off_T)page_size * nr;
Bram Moolenaar4ba37b52019-12-04 21:57:43 +01001015 if (hp2 == NULL) // freed block, fill with dummy data
Bram Moolenaar071d4272004-06-13 20:20:40 +00001016 page_count = 1;
1017 else
1018 page_count = hp2->bh_page_count;
1019 size = page_size * page_count;
Bram Moolenaarb58a4b92019-05-27 23:36:21 +02001020
1021 for (attempt = 1; attempt <= 2; ++attempt)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001022 {
Bram Moolenaarb58a4b92019-05-27 23:36:21 +02001023 if (mfp->mf_fd >= 0)
1024 {
1025 if (vim_lseek(mfp->mf_fd, offset, SEEK_SET) != offset)
1026 {
Bram Moolenaar9a846fb2022-01-01 21:59:18 +00001027 PERROR(_(e_seek_error_in_swap_file_write));
Bram Moolenaarb58a4b92019-05-27 23:36:21 +02001028 return FAIL;
1029 }
1030 if (mf_write_block(mfp,
1031 hp2 == NULL ? hp : hp2, offset, size) == OK)
1032 break;
1033 }
1034
1035 if (attempt == 1)
1036 {
1037 // If the swap file is on a network drive, and the network
1038 // gets disconnected and then re-connected, we can maybe fix it
1039 // by closing and then re-opening the file.
1040 if (mfp->mf_fd >= 0)
1041 close(mfp->mf_fd);
1042 mfp->mf_fd = mch_open_rw((char *)mfp->mf_fname, mfp->mf_flags);
1043 mfp->mf_reopen = (mfp->mf_fd < 0);
1044 }
1045 if (attempt == 2 || mfp->mf_fd < 0)
1046 {
1047 // Avoid repeating the error message, this mostly happens when
1048 // the disk is full. We give the message again only after a
1049 // successful write or when hitting a key. We keep on trying,
1050 // in case some space becomes available.
1051 if (!did_swapwrite_msg)
Bram Moolenaar9a846fb2022-01-01 21:59:18 +00001052 emsg(_(e_write_error_in_swap_file));
Bram Moolenaarb58a4b92019-05-27 23:36:21 +02001053 did_swapwrite_msg = TRUE;
1054 return FAIL;
1055 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001056 }
Bram Moolenaarb58a4b92019-05-27 23:36:21 +02001057
Bram Moolenaar071d4272004-06-13 20:20:40 +00001058 did_swapwrite_msg = FALSE;
Bram Moolenaar4ba37b52019-12-04 21:57:43 +01001059 if (hp2 != NULL) // written a non-dummy block
Bram Moolenaar071d4272004-06-13 20:20:40 +00001060 hp2->bh_flags &= ~BH_DIRTY;
Bram Moolenaar4ba37b52019-12-04 21:57:43 +01001061 // appended to the file
Bram Moolenaar071d4272004-06-13 20:20:40 +00001062 if (nr + (blocknr_T)page_count > mfp->mf_infile_count)
1063 mfp->mf_infile_count = nr + page_count;
Bram Moolenaar4ba37b52019-12-04 21:57:43 +01001064 if (nr == hp->bh_bnum) // written the desired block
Bram Moolenaar071d4272004-06-13 20:20:40 +00001065 break;
1066 }
1067 return OK;
1068}
1069
1070/*
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001071 * Write block "hp" with data size "size" to file "mfp->mf_fd".
1072 * Takes care of encryption.
1073 * Return FAIL or OK.
1074 */
1075 static int
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01001076mf_write_block(
1077 memfile_T *mfp,
1078 bhdr_T *hp,
Bram Moolenaar8767f522016-07-01 17:17:39 +02001079 off_T offset UNUSED,
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01001080 unsigned size)
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001081{
1082 char_u *data = hp->bh_data;
1083 int result = OK;
1084
1085#ifdef FEAT_CRYPT
Bram Moolenaar4ba37b52019-12-04 21:57:43 +01001086 // Encrypt if 'key' is set and this is a data block.
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001087 if (*mfp->mf_buffer->b_p_key != NUL)
1088 {
1089 data = ml_encrypt_data(mfp, data, offset, size);
1090 if (data == NULL)
1091 return FAIL;
1092 }
1093#endif
1094
Bram Moolenaar540fc6f2010-12-17 16:27:16 +01001095 if ((unsigned)write_eintr(mfp->mf_fd, data, size) != size)
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001096 result = FAIL;
1097
1098#ifdef FEAT_CRYPT
1099 if (data != hp->bh_data)
1100 vim_free(data);
1101#endif
1102
1103 return result;
1104}
1105
1106/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00001107 * Make block number for *hp positive and add it to the translation list
1108 *
1109 * Return FAIL for failure, OK otherwise
1110 */
1111 static int
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01001112mf_trans_add(memfile_T *mfp, bhdr_T *hp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001113{
1114 bhdr_T *freep;
1115 blocknr_T new_bnum;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001116 NR_TRANS *np;
1117 int page_count;
1118
Bram Moolenaar4ba37b52019-12-04 21:57:43 +01001119 if (hp->bh_bnum >= 0) // it's already positive
Bram Moolenaar071d4272004-06-13 20:20:40 +00001120 return OK;
1121
Bram Moolenaarc799fe22019-05-28 23:08:19 +02001122 if ((np = ALLOC_ONE(NR_TRANS)) == NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001123 return FAIL;
1124
1125/*
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001126 * Get a new number for the block.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001127 * If the first item in the free list has sufficient pages, use its number
1128 * Otherwise use mf_blocknr_max.
1129 */
1130 freep = mfp->mf_free_first;
1131 page_count = hp->bh_page_count;
1132 if (freep != NULL && freep->bh_page_count >= page_count)
1133 {
1134 new_bnum = freep->bh_bnum;
1135 /*
Bram Moolenaar84a05ac2013-05-06 04:24:17 +02001136 * If the page count of the free block was larger, reduce it.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001137 * If the page count matches, remove the block from the free list
1138 */
1139 if (freep->bh_page_count > page_count)
1140 {
1141 freep->bh_bnum += page_count;
1142 freep->bh_page_count -= page_count;
1143 }
1144 else
1145 {
1146 freep = mf_rem_free(mfp);
1147 vim_free(freep);
1148 }
1149 }
1150 else
1151 {
1152 new_bnum = mfp->mf_blocknr_max;
1153 mfp->mf_blocknr_max += page_count;
1154 }
1155
Bram Moolenaar4ba37b52019-12-04 21:57:43 +01001156 np->nt_old_bnum = hp->bh_bnum; // adjust number
Bram Moolenaar071d4272004-06-13 20:20:40 +00001157 np->nt_new_bnum = new_bnum;
1158
Bram Moolenaar4ba37b52019-12-04 21:57:43 +01001159 mf_rem_hash(mfp, hp); // remove from old hash list
Bram Moolenaar071d4272004-06-13 20:20:40 +00001160 hp->bh_bnum = new_bnum;
Bram Moolenaar4ba37b52019-12-04 21:57:43 +01001161 mf_ins_hash(mfp, hp); // insert in new hash list
Bram Moolenaar071d4272004-06-13 20:20:40 +00001162
Bram Moolenaar4ba37b52019-12-04 21:57:43 +01001163 // Insert "np" into "mf_trans" hashtable with key "np->nt_old_bnum"
Bram Moolenaarb05b10a2011-03-22 18:10:45 +01001164 mf_hash_add_item(&mfp->mf_trans, (mf_hashitem_T *)np);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001165
1166 return OK;
1167}
1168
1169/*
Bram Moolenaarbc563362015-06-09 18:35:25 +02001170 * Lookup a translation from the trans lists and delete the entry.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001171 *
1172 * Return the positive new number when found, the old number when not found
1173 */
1174 blocknr_T
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01001175mf_trans_del(memfile_T *mfp, blocknr_T old_nr)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001176{
Bram Moolenaar071d4272004-06-13 20:20:40 +00001177 NR_TRANS *np;
1178 blocknr_T new_bnum;
1179
Bram Moolenaarb05b10a2011-03-22 18:10:45 +01001180 np = (NR_TRANS *)mf_hash_find(&mfp->mf_trans, old_nr);
1181
Bram Moolenaar4ba37b52019-12-04 21:57:43 +01001182 if (np == NULL) // not found
Bram Moolenaar071d4272004-06-13 20:20:40 +00001183 return old_nr;
1184
1185 mfp->mf_neg_count--;
1186 new_bnum = np->nt_new_bnum;
Bram Moolenaarb05b10a2011-03-22 18:10:45 +01001187
Bram Moolenaar4ba37b52019-12-04 21:57:43 +01001188 // remove entry from the trans list
Bram Moolenaarb05b10a2011-03-22 18:10:45 +01001189 mf_hash_rem_item(&mfp->mf_trans, (mf_hashitem_T *)np);
1190
Bram Moolenaar071d4272004-06-13 20:20:40 +00001191 vim_free(np);
1192
1193 return new_bnum;
1194}
1195
1196/*
1197 * Set mfp->mf_ffname according to mfp->mf_fname and some other things.
1198 * Only called when creating or renaming the swapfile. Either way it's a new
1199 * name so we must work out the full path name.
1200 */
1201 void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01001202mf_set_ffname(memfile_T *mfp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001203{
1204 mfp->mf_ffname = FullName_save(mfp->mf_fname, FALSE);
1205}
1206
1207/*
1208 * Make the name of the file used for the memfile a full path.
1209 * Used before doing a :cd
1210 */
1211 void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01001212mf_fullname(memfile_T *mfp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001213{
Yegappan Lakshmanane8575982023-01-14 12:32:28 +00001214 if (mfp == NULL || mfp->mf_fname == NULL || mfp->mf_ffname == NULL)
1215 return;
1216
1217 vim_free(mfp->mf_fname);
1218 mfp->mf_fname = mfp->mf_ffname;
1219 mfp->mf_ffname = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001220}
1221
1222/*
1223 * return TRUE if there are any translations pending for 'mfp'
1224 */
1225 int
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01001226mf_need_trans(memfile_T *mfp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001227{
1228 return (mfp->mf_fname != NULL && mfp->mf_neg_count > 0);
1229}
1230
1231/*
1232 * Open a swap file for a memfile.
1233 * The "fname" must be in allocated memory, and is consumed (also when an
1234 * error occurs).
1235 */
1236 static void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01001237mf_do_open(
1238 memfile_T *mfp,
1239 char_u *fname,
Bram Moolenaar4ba37b52019-12-04 21:57:43 +01001240 int flags) // flags for open()
Bram Moolenaar071d4272004-06-13 20:20:40 +00001241{
1242#ifdef HAVE_LSTAT
Bram Moolenaar8767f522016-07-01 17:17:39 +02001243 stat_T sb;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001244#endif
1245
1246 mfp->mf_fname = fname;
1247
1248 /*
1249 * Get the full path name before the open, because this is
1250 * not possible after the open on the Amiga.
1251 * fname cannot be NameBuff, because it must have been allocated.
1252 */
1253 mf_set_ffname(mfp);
Bram Moolenaar48e330a2016-02-23 14:53:34 +01001254#if defined(MSWIN)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001255 /*
Bram Moolenaar84a05ac2013-05-06 04:24:17 +02001256 * A ":!cd e:xxx" may change the directory without us knowing, use the
Bram Moolenaar071d4272004-06-13 20:20:40 +00001257 * full pathname always. Careful: This frees fname!
1258 */
1259 mf_fullname(mfp);
1260#endif
1261
1262#ifdef HAVE_LSTAT
1263 /*
1264 * Extra security check: When creating a swap file it really shouldn't
1265 * exist yet. If there is a symbolic link, this is most likely an attack.
1266 */
1267 if ((flags & O_CREAT) && mch_lstat((char *)mfp->mf_fname, &sb) >= 0)
1268 {
1269 mfp->mf_fd = -1;
Bram Moolenaareaaac012022-01-02 17:00:40 +00001270 emsg(_(e_swap_file_already_exists_symlink_attack));
Bram Moolenaar071d4272004-06-13 20:20:40 +00001271 }
1272 else
1273#endif
1274 {
1275 /*
1276 * try to open the file
1277 */
Bram Moolenaara5792f52005-11-23 21:25:05 +00001278 flags |= O_EXTRA | O_NOFOLLOW;
Bram Moolenaar4f974752019-02-17 17:44:42 +01001279#ifdef MSWIN
Bram Moolenaar4ba37b52019-12-04 21:57:43 +01001280 // Prevent handle inheritance that cause problems with Cscope
1281 // (swap file may not be deleted if cscope connection was open after
1282 // the file)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001283 flags |= O_NOINHERIT;
1284#endif
Bram Moolenaarb58a4b92019-05-27 23:36:21 +02001285 mfp->mf_flags = flags;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001286 mfp->mf_fd = mch_open_rw((char *)mfp->mf_fname, flags);
1287 }
1288
1289 /*
1290 * If the file cannot be opened, use memory only
1291 */
1292 if (mfp->mf_fd < 0)
1293 {
Bram Moolenaard23a8232018-02-10 18:45:26 +01001294 VIM_CLEAR(mfp->mf_fname);
1295 VIM_CLEAR(mfp->mf_ffname);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001296 }
1297 else
Bram Moolenaar588ebeb2008-05-07 17:09:24 +00001298 {
Bram Moolenaarf05da212009-11-17 16:13:15 +00001299#ifdef HAVE_FD_CLOEXEC
1300 int fdflags = fcntl(mfp->mf_fd, F_GETFD);
1301 if (fdflags >= 0 && (fdflags & FD_CLOEXEC) == 0)
Bram Moolenaarfbc4b4d2016-02-07 15:14:01 +01001302 (void)fcntl(mfp->mf_fd, F_SETFD, fdflags | FD_CLOEXEC);
Bram Moolenaarf05da212009-11-17 16:13:15 +00001303#endif
Bram Moolenaar5bd32f42014-04-02 14:05:38 +02001304#if defined(HAVE_SELINUX) || defined(HAVE_SMACK)
Bram Moolenaar588ebeb2008-05-07 17:09:24 +00001305 mch_copy_sec(fname, mfp->mf_fname);
1306#endif
Bram Moolenaar4ba37b52019-12-04 21:57:43 +01001307 mch_hide(mfp->mf_fname); // try setting the 'hidden' flag
Bram Moolenaar588ebeb2008-05-07 17:09:24 +00001308 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001309}
Bram Moolenaarb05b10a2011-03-22 18:10:45 +01001310
1311/*
1312 * Implementation of mf_hashtab_T follows.
1313 */
1314
1315/*
1316 * The number of buckets in the hashtable is increased by a factor of
1317 * MHT_GROWTH_FACTOR when the average number of items per bucket
1318 * exceeds 2 ^ MHT_LOG_LOAD_FACTOR.
1319 */
1320#define MHT_LOG_LOAD_FACTOR 6
Bram Moolenaar4ba37b52019-12-04 21:57:43 +01001321#define MHT_GROWTH_FACTOR 2 // must be a power of two
Bram Moolenaarb05b10a2011-03-22 18:10:45 +01001322
1323/*
1324 * Initialize an empty hash table.
1325 */
1326 static void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01001327mf_hash_init(mf_hashtab_T *mht)
Bram Moolenaarb05b10a2011-03-22 18:10:45 +01001328{
Bram Moolenaara80faa82020-04-12 19:37:17 +02001329 CLEAR_POINTER(mht);
Bram Moolenaarb05b10a2011-03-22 18:10:45 +01001330 mht->mht_buckets = mht->mht_small_buckets;
1331 mht->mht_mask = MHT_INIT_SIZE - 1;
1332}
1333
1334/*
1335 * Free the array of a hash table. Does not free the items it contains!
1336 * The hash table must not be used again without another mf_hash_init() call.
1337 */
1338 static void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01001339mf_hash_free(mf_hashtab_T *mht)
Bram Moolenaarb05b10a2011-03-22 18:10:45 +01001340{
1341 if (mht->mht_buckets != mht->mht_small_buckets)
1342 vim_free(mht->mht_buckets);
1343}
1344
1345/*
1346 * Free the array of a hash table and all the items it contains.
1347 */
1348 static void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01001349mf_hash_free_all(mf_hashtab_T *mht)
Bram Moolenaarb05b10a2011-03-22 18:10:45 +01001350{
1351 long_u idx;
1352 mf_hashitem_T *mhi;
1353 mf_hashitem_T *next;
1354
1355 for (idx = 0; idx <= mht->mht_mask; idx++)
1356 for (mhi = mht->mht_buckets[idx]; mhi != NULL; mhi = next)
1357 {
1358 next = mhi->mhi_next;
1359 vim_free(mhi);
1360 }
1361
1362 mf_hash_free(mht);
1363}
1364
1365/*
1366 * Find "key" in hashtable "mht".
1367 * Returns a pointer to a mf_hashitem_T or NULL if the item was not found.
1368 */
1369 static mf_hashitem_T *
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01001370mf_hash_find(mf_hashtab_T *mht, blocknr_T key)
Bram Moolenaarb05b10a2011-03-22 18:10:45 +01001371{
1372 mf_hashitem_T *mhi;
1373
1374 mhi = mht->mht_buckets[key & mht->mht_mask];
1375 while (mhi != NULL && mhi->mhi_key != key)
1376 mhi = mhi->mhi_next;
1377
1378 return mhi;
1379}
1380
1381/*
1382 * Add item "mhi" to hashtable "mht".
1383 * "mhi" must not be NULL.
1384 */
1385 static void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01001386mf_hash_add_item(mf_hashtab_T *mht, mf_hashitem_T *mhi)
Bram Moolenaarb05b10a2011-03-22 18:10:45 +01001387{
1388 long_u idx;
1389
1390 idx = mhi->mhi_key & mht->mht_mask;
1391 mhi->mhi_next = mht->mht_buckets[idx];
1392 mhi->mhi_prev = NULL;
1393 if (mhi->mhi_next != NULL)
1394 mhi->mhi_next->mhi_prev = mhi;
1395 mht->mht_buckets[idx] = mhi;
1396
1397 mht->mht_count++;
1398
1399 /*
1400 * Grow hashtable when we have more thank 2^MHT_LOG_LOAD_FACTOR
1401 * items per bucket on average
1402 */
1403 if (mht->mht_fixed == 0
1404 && (mht->mht_count >> MHT_LOG_LOAD_FACTOR) > mht->mht_mask)
1405 {
1406 if (mf_hash_grow(mht) == FAIL)
1407 {
Bram Moolenaar4ba37b52019-12-04 21:57:43 +01001408 // stop trying to grow after first failure to allocate memory
Bram Moolenaarb05b10a2011-03-22 18:10:45 +01001409 mht->mht_fixed = 1;
1410 }
1411 }
1412}
1413
1414/*
1415 * Remove item "mhi" from hashtable "mht".
1416 * "mhi" must not be NULL and must have been inserted into "mht".
1417 */
1418 static void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01001419mf_hash_rem_item(mf_hashtab_T *mht, mf_hashitem_T *mhi)
Bram Moolenaarb05b10a2011-03-22 18:10:45 +01001420{
1421 if (mhi->mhi_prev == NULL)
1422 mht->mht_buckets[mhi->mhi_key & mht->mht_mask] = mhi->mhi_next;
1423 else
1424 mhi->mhi_prev->mhi_next = mhi->mhi_next;
1425
1426 if (mhi->mhi_next != NULL)
1427 mhi->mhi_next->mhi_prev = mhi->mhi_prev;
1428
1429 mht->mht_count--;
1430
Bram Moolenaar4ba37b52019-12-04 21:57:43 +01001431 // We could shrink the table here, but it typically takes little memory,
1432 // so why bother?
Bram Moolenaarb05b10a2011-03-22 18:10:45 +01001433}
1434
1435/*
1436 * Increase number of buckets in the hashtable by MHT_GROWTH_FACTOR and
1437 * rehash items.
1438 * Returns FAIL when out of memory.
1439 */
1440 static int
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01001441mf_hash_grow(mf_hashtab_T *mht)
Bram Moolenaarb05b10a2011-03-22 18:10:45 +01001442{
1443 long_u i, j;
1444 int shift;
1445 mf_hashitem_T *mhi;
1446 mf_hashitem_T *tails[MHT_GROWTH_FACTOR];
1447 mf_hashitem_T **buckets;
1448 size_t size;
1449
1450 size = (mht->mht_mask + 1) * MHT_GROWTH_FACTOR * sizeof(void *);
Bram Moolenaarc799fe22019-05-28 23:08:19 +02001451 buckets = lalloc_clear(size, FALSE);
Bram Moolenaarb05b10a2011-03-22 18:10:45 +01001452 if (buckets == NULL)
1453 return FAIL;
1454
1455 shift = 0;
1456 while ((mht->mht_mask >> shift) != 0)
1457 shift++;
1458
1459 for (i = 0; i <= mht->mht_mask; i++)
1460 {
1461 /*
1462 * Traverse the items in the i-th original bucket and move them into
1463 * MHT_GROWTH_FACTOR new buckets, preserving their relative order
1464 * within each new bucket. Preserving the order is important because
1465 * mf_get() tries to keep most recently used items at the front of
1466 * each bucket.
1467 *
1468 * Here we strongly rely on the fact the hashes are computed modulo
1469 * a power of two.
1470 */
1471
Bram Moolenaara80faa82020-04-12 19:37:17 +02001472 CLEAR_FIELD(tails);
Bram Moolenaarb05b10a2011-03-22 18:10:45 +01001473
1474 for (mhi = mht->mht_buckets[i]; mhi != NULL; mhi = mhi->mhi_next)
1475 {
1476 j = (mhi->mhi_key >> shift) & (MHT_GROWTH_FACTOR - 1);
1477 if (tails[j] == NULL)
1478 {
1479 buckets[i + (j << shift)] = mhi;
1480 tails[j] = mhi;
1481 mhi->mhi_prev = NULL;
1482 }
1483 else
1484 {
1485 tails[j]->mhi_next = mhi;
1486 mhi->mhi_prev = tails[j];
1487 tails[j] = mhi;
1488 }
1489 }
1490
1491 for (j = 0; j < MHT_GROWTH_FACTOR; j++)
1492 if (tails[j] != NULL)
1493 tails[j]->mhi_next = NULL;
1494 }
1495
1496 if (mht->mht_buckets != mht->mht_small_buckets)
1497 vim_free(mht->mht_buckets);
1498
1499 mht->mht_buckets = buckets;
1500 mht->mht_mask = (mht->mht_mask + 1) * MHT_GROWTH_FACTOR - 1;
1501
1502 return OK;
1503}