blob: f6fda8d34c17fdfcb8f2ab22c7b47af64d375a1a [file] [log] [blame]
Bram Moolenaar071d4272004-06-13 20:20:40 +00001/* vi:set ts=8 sts=4 sw=4:
2 *
3 * VIM - Vi IMproved by Bram Moolenaar
4 *
5 * Do ":help uganda" in Vim to read copying and usage conditions.
6 * Do ":help credits" in Vim to see a list of people who contributed.
7 * See README.txt for an overview of the Vim source code.
8 */
9
10/*
11 * 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
49# ifdef __MINT__ /* do we still need this? */
50# define fstatfs(fd, buf, len, nul) mch_fstat((fd), (buf))
51# endif
52# endif
53#endif
54
55/*
56 * for Amiga Dos 2.0x we use Flush
57 */
58#ifdef AMIGA
59# ifdef FEAT_ARP
60extern int dos2; /* this is in os_amiga.c */
61# endif
62# ifdef SASC
63# include <proto/dos.h>
64# include <ios1.h> /* for chkufb() */
65# endif
66#endif
67
68#define MEMFILE_PAGE_SIZE 4096 /* default page size */
69
70static long_u total_mem_used = 0; /* total memory used for memfiles */
Bram Moolenaar071d4272004-06-13 20:20:40 +000071
Bram Moolenaar92b8b2d2016-01-29 22:36:45 +010072static void mf_ins_hash(memfile_T *, bhdr_T *);
73static void mf_rem_hash(memfile_T *, bhdr_T *);
74static bhdr_T *mf_find_hash(memfile_T *, blocknr_T);
75static void mf_ins_used(memfile_T *, bhdr_T *);
76static void mf_rem_used(memfile_T *, bhdr_T *);
77static bhdr_T *mf_release(memfile_T *, int);
78static bhdr_T *mf_alloc_bhdr(memfile_T *, int);
79static void mf_free_bhdr(bhdr_T *);
80static void mf_ins_free(memfile_T *, bhdr_T *);
81static bhdr_T *mf_rem_free(memfile_T *);
82static int mf_read(memfile_T *, bhdr_T *);
83static int mf_write(memfile_T *, bhdr_T *);
84static int mf_write_block(memfile_T *mfp, bhdr_T *hp, off_t offset, unsigned size);
85static int mf_trans_add(memfile_T *, bhdr_T *);
86static void mf_do_open(memfile_T *, char_u *, int);
87static void mf_hash_init(mf_hashtab_T *);
88static void mf_hash_free(mf_hashtab_T *);
89static void mf_hash_free_all(mf_hashtab_T *);
90static mf_hashitem_T *mf_hash_find(mf_hashtab_T *, blocknr_T);
91static void mf_hash_add_item(mf_hashtab_T *, mf_hashitem_T *);
92static void mf_hash_rem_item(mf_hashtab_T *, mf_hashitem_T *);
93static int mf_hash_grow(mf_hashtab_T *);
Bram Moolenaar071d4272004-06-13 20:20:40 +000094
95/*
96 * The functions for using a memfile:
97 *
98 * mf_open() open a new or existing memfile
99 * mf_open_file() open a swap file for an existing memfile
100 * mf_close() close (and delete) a memfile
101 * mf_new() create a new block in a memfile and lock it
102 * mf_get() get an existing block and lock it
103 * mf_put() unlock a block, may be marked for writing
104 * mf_free() remove a block
105 * mf_sync() sync changed parts of memfile to disk
106 * mf_release_all() release as much memory as possible
107 * mf_trans_del() may translate negative to positive block number
108 * mf_fullname() make file name full path (use before first :cd)
109 */
110
111/*
112 * Open an existing or new memory block file.
113 *
114 * fname: name of file to use (NULL means no file at all)
115 * Note: fname must have been allocated, it is not copied!
116 * If opening the file fails, fname is freed.
117 * flags: flags for open() call
118 *
119 * If fname != NULL and file cannot be opened, fail.
120 *
121 * return value: identifier for this memory block file.
122 */
123 memfile_T *
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100124mf_open(char_u *fname, int flags)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000125{
126 memfile_T *mfp;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000127 off_t size;
Bram Moolenaara03dbed2013-05-23 22:27:03 +0200128#if defined(STATFS) && defined(UNIX) && !defined(__QNX__) && !defined(__minix)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000129# define USE_FSTATFS
130 struct STATFS stf;
131#endif
132
133 if ((mfp = (memfile_T *)alloc((unsigned)sizeof(memfile_T))) == NULL)
134 return NULL;
135
136 if (fname == NULL) /* no file for this memfile, use memory only */
137 {
138 mfp->mf_fname = NULL;
139 mfp->mf_ffname = NULL;
140 mfp->mf_fd = -1;
141 }
142 else
143 {
144 mf_do_open(mfp, fname, flags); /* try to open the file */
145
146 /* if the file cannot be opened, return here */
147 if (mfp->mf_fd < 0)
148 {
149 vim_free(mfp);
150 return NULL;
151 }
152 }
153
154 mfp->mf_free_first = NULL; /* free list is empty */
155 mfp->mf_used_first = NULL; /* used list is empty */
156 mfp->mf_used_last = NULL;
157 mfp->mf_dirty = FALSE;
158 mfp->mf_used_count = 0;
Bram Moolenaarb05b10a2011-03-22 18:10:45 +0100159 mf_hash_init(&mfp->mf_hash);
160 mf_hash_init(&mfp->mf_trans);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000161 mfp->mf_page_size = MEMFILE_PAGE_SIZE;
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200162#ifdef FEAT_CRYPT
163 mfp->mf_old_key = NULL;
164#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000165
166#ifdef USE_FSTATFS
167 /*
168 * Try to set the page size equal to the block size of the device.
169 * Speeds up I/O a lot.
170 * When recovering, the actual block size will be retrieved from block 0
171 * in ml_recover(). The size used here may be wrong, therefore
172 * mf_blocknr_max must be rounded up.
173 */
174 if (mfp->mf_fd >= 0
175 && fstatfs(mfp->mf_fd, &stf, sizeof(struct statfs), 0) == 0
176 && stf.F_BSIZE >= MIN_SWAP_PAGE_SIZE
177 && stf.F_BSIZE <= MAX_SWAP_PAGE_SIZE)
178 mfp->mf_page_size = stf.F_BSIZE;
179#endif
180
181 if (mfp->mf_fd < 0 || (flags & (O_TRUNC|O_EXCL))
182 || (size = lseek(mfp->mf_fd, (off_t)0L, SEEK_END)) <= 0)
183 mfp->mf_blocknr_max = 0; /* no file or empty file */
184 else
185 mfp->mf_blocknr_max = (blocknr_T)((size + mfp->mf_page_size - 1)
186 / mfp->mf_page_size);
187 mfp->mf_blocknr_min = -1;
188 mfp->mf_neg_count = 0;
189 mfp->mf_infile_count = mfp->mf_blocknr_max;
Bram Moolenaarfe265ff2007-05-11 18:15:45 +0000190
191 /*
192 * Compute maximum number of pages ('maxmem' is in Kbyte):
193 * 'mammem' * 1Kbyte / page-size-in-bytes.
194 * Avoid overflow by first reducing page size as much as possible.
195 */
196 {
197 int shift = 10;
198 unsigned page_size = mfp->mf_page_size;
199
200 while (shift > 0 && (page_size & 1) == 0)
201 {
202 page_size = page_size >> 1;
203 --shift;
204 }
205 mfp->mf_used_count_max = (p_mm << shift) / page_size;
206 if (mfp->mf_used_count_max < 10)
207 mfp->mf_used_count_max = 10;
208 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000209
210 return mfp;
211}
212
213/*
214 * Open a file for an existing memfile. Used when updatecount set from 0 to
215 * some value.
216 * If the file already exists, this fails.
217 * "fname" is the name of file to use (NULL means no file at all)
218 * Note: "fname" must have been allocated, it is not copied! If opening the
219 * file fails, "fname" is freed.
220 *
221 * return value: FAIL if file could not be opened, OK otherwise
222 */
223 int
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100224mf_open_file(memfile_T *mfp, char_u *fname)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000225{
226 mf_do_open(mfp, fname, O_RDWR|O_CREAT|O_EXCL); /* try to open the file */
227
228 if (mfp->mf_fd < 0)
229 return FAIL;
230
231 mfp->mf_dirty = TRUE;
232 return OK;
233}
234
235/*
Bram Moolenaar945e2db2010-06-05 17:43:32 +0200236 * Close a memory file and delete the associated file if 'del_file' is TRUE.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000237 */
238 void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100239mf_close(memfile_T *mfp, int del_file)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000240{
241 bhdr_T *hp, *nextp;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000242
243 if (mfp == NULL) /* safety check */
244 return;
245 if (mfp->mf_fd >= 0)
246 {
247 if (close(mfp->mf_fd) < 0)
248 EMSG(_(e_swapclose));
249 }
250 if (del_file && mfp->mf_fname != NULL)
251 mch_remove(mfp->mf_fname);
252 /* free entries in used list */
253 for (hp = mfp->mf_used_first; hp != NULL; hp = nextp)
254 {
255 total_mem_used -= hp->bh_page_count * mfp->mf_page_size;
256 nextp = hp->bh_next;
257 mf_free_bhdr(hp);
258 }
259 while (mfp->mf_free_first != NULL) /* free entries in free list */
260 vim_free(mf_rem_free(mfp));
Bram Moolenaarb05b10a2011-03-22 18:10:45 +0100261 mf_hash_free(&mfp->mf_hash);
262 mf_hash_free_all(&mfp->mf_trans); /* free hashtable and its items */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000263 vim_free(mfp->mf_fname);
264 vim_free(mfp->mf_ffname);
265 vim_free(mfp);
266}
267
268/*
269 * Close the swap file for a memfile. Used when 'swapfile' is reset.
270 */
271 void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100272mf_close_file(
273 buf_T *buf,
274 int getlines) /* get all lines into memory? */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000275{
276 memfile_T *mfp;
277 linenr_T lnum;
278
279 mfp = buf->b_ml.ml_mfp;
280 if (mfp == NULL || mfp->mf_fd < 0) /* nothing to close */
281 return;
282
283 if (getlines)
284 {
285 /* get all blocks in memory by accessing all lines (clumsy!) */
Bram Moolenaar47b8b152007-02-07 02:41:57 +0000286 mf_dont_release = TRUE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000287 for (lnum = 1; lnum <= buf->b_ml.ml_line_count; ++lnum)
288 (void)ml_get_buf(buf, lnum, FALSE);
Bram Moolenaar47b8b152007-02-07 02:41:57 +0000289 mf_dont_release = FALSE;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000290 /* TODO: should check if all blocks are really in core */
291 }
292
293 if (close(mfp->mf_fd) < 0) /* close the file */
294 EMSG(_(e_swapclose));
295 mfp->mf_fd = -1;
296
297 if (mfp->mf_fname != NULL)
298 {
299 mch_remove(mfp->mf_fname); /* delete the swap file */
300 vim_free(mfp->mf_fname);
301 vim_free(mfp->mf_ffname);
302 mfp->mf_fname = NULL;
303 mfp->mf_ffname = NULL;
304 }
305}
306
307/*
308 * Set new size for a memfile. Used when block 0 of a swapfile has been read
309 * and the size it indicates differs from what was guessed.
310 */
311 void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100312mf_new_page_size(memfile_T *mfp, unsigned new_size)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000313{
314 /* Correct the memory used for block 0 to the new size, because it will be
315 * freed with that size later on. */
316 total_mem_used += new_size - mfp->mf_page_size;
317 mfp->mf_page_size = new_size;
318}
319
320/*
321 * get a new block
322 *
323 * negative: TRUE if negative block number desired (data block)
324 */
325 bhdr_T *
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100326mf_new(memfile_T *mfp, int negative, int page_count)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000327{
Bram Moolenaar17e79192007-05-06 14:15:36 +0000328 bhdr_T *hp; /* new bhdr_T */
329 bhdr_T *freep; /* first block in free list */
330 char_u *p;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000331
332 /*
333 * If we reached the maximum size for the used memory blocks, release one
334 * If a bhdr_T is returned, use it and adjust the page_count if necessary.
335 */
336 hp = mf_release(mfp, page_count);
337
338/*
339 * Decide on the number to use:
340 * If there is a free block, use its number.
341 * Otherwise use mf_block_min for a negative number, mf_block_max for
342 * a positive number.
343 */
344 freep = mfp->mf_free_first;
345 if (!negative && freep != NULL && freep->bh_page_count >= page_count)
346 {
347 /*
348 * If the block in the free list has more pages, take only the number
349 * of pages needed and allocate a new bhdr_T with data
350 *
Bram Moolenaar17e79192007-05-06 14:15:36 +0000351 * If the number of pages matches and mf_release() did not return a
352 * bhdr_T, use the bhdr_T from the free list and allocate the data
Bram Moolenaar071d4272004-06-13 20:20:40 +0000353 *
Bram Moolenaar17e79192007-05-06 14:15:36 +0000354 * If the number of pages matches and mf_release() returned a bhdr_T,
Bram Moolenaar071d4272004-06-13 20:20:40 +0000355 * just use the number and free the bhdr_T from the free list
356 */
357 if (freep->bh_page_count > page_count)
358 {
359 if (hp == NULL && (hp = mf_alloc_bhdr(mfp, page_count)) == NULL)
360 return NULL;
361 hp->bh_bnum = freep->bh_bnum;
362 freep->bh_bnum += page_count;
363 freep->bh_page_count -= page_count;
364 }
365 else if (hp == NULL) /* need to allocate memory for this block */
366 {
367 if ((p = (char_u *)alloc(mfp->mf_page_size * page_count)) == NULL)
368 return NULL;
369 hp = mf_rem_free(mfp);
370 hp->bh_data = p;
371 }
372 else /* use the number, remove entry from free list */
373 {
374 freep = mf_rem_free(mfp);
375 hp->bh_bnum = freep->bh_bnum;
376 vim_free(freep);
377 }
378 }
379 else /* get a new number */
380 {
381 if (hp == NULL && (hp = mf_alloc_bhdr(mfp, page_count)) == NULL)
382 return NULL;
383 if (negative)
384 {
385 hp->bh_bnum = mfp->mf_blocknr_min--;
386 mfp->mf_neg_count++;
387 }
388 else
389 {
390 hp->bh_bnum = mfp->mf_blocknr_max;
391 mfp->mf_blocknr_max += page_count;
392 }
393 }
394 hp->bh_flags = BH_LOCKED | BH_DIRTY; /* new block is always dirty */
395 mfp->mf_dirty = TRUE;
396 hp->bh_page_count = page_count;
397 mf_ins_used(mfp, hp);
398 mf_ins_hash(mfp, hp);
399
400 /*
401 * Init the data to all zero, to avoid reading uninitialized data.
402 * This also avoids that the passwd file ends up in the swap file!
403 */
Bram Moolenaar945e2db2010-06-05 17:43:32 +0200404 (void)vim_memset((char *)(hp->bh_data), 0,
405 (size_t)mfp->mf_page_size * page_count);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000406
407 return hp;
408}
409
410/*
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200411 * Get existing block "nr" with "page_count" pages.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000412 *
413 * Note: The caller should first check a negative nr with mf_trans_del()
414 */
415 bhdr_T *
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100416mf_get(memfile_T *mfp, blocknr_T nr, int page_count)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000417{
418 bhdr_T *hp;
419 /* doesn't exist */
420 if (nr >= mfp->mf_blocknr_max || nr <= mfp->mf_blocknr_min)
421 return NULL;
422
423 /*
424 * see if it is in the cache
425 */
426 hp = mf_find_hash(mfp, nr);
427 if (hp == NULL) /* not in the hash list */
428 {
429 if (nr < 0 || nr >= mfp->mf_infile_count) /* can't be in the file */
430 return NULL;
431
432 /* could check here if the block is in the free list */
433
434 /*
435 * Check if we need to flush an existing block.
436 * If so, use that block.
437 * If not, allocate a new block.
438 */
439 hp = mf_release(mfp, page_count);
440 if (hp == NULL && (hp = mf_alloc_bhdr(mfp, page_count)) == NULL)
441 return NULL;
442
443 hp->bh_bnum = nr;
444 hp->bh_flags = 0;
445 hp->bh_page_count = page_count;
446 if (mf_read(mfp, hp) == FAIL) /* cannot read the block! */
447 {
448 mf_free_bhdr(hp);
449 return NULL;
450 }
451 }
452 else
453 {
454 mf_rem_used(mfp, hp); /* remove from list, insert in front below */
455 mf_rem_hash(mfp, hp);
456 }
457
458 hp->bh_flags |= BH_LOCKED;
459 mf_ins_used(mfp, hp); /* put in front of used list */
460 mf_ins_hash(mfp, hp); /* put in front of hash list */
461
462 return hp;
463}
464
465/*
466 * release the block *hp
467 *
468 * dirty: Block must be written to file later
469 * infile: Block should be in file (needed for recovery)
470 *
471 * no return value, function cannot fail
472 */
473 void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100474mf_put(
475 memfile_T *mfp,
476 bhdr_T *hp,
477 int dirty,
478 int infile)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000479{
480 int flags;
481
482 flags = hp->bh_flags;
483
484 if ((flags & BH_LOCKED) == 0)
485 EMSG(_("E293: block was not locked"));
486 flags &= ~BH_LOCKED;
487 if (dirty)
488 {
489 flags |= BH_DIRTY;
490 mfp->mf_dirty = TRUE;
491 }
492 hp->bh_flags = flags;
493 if (infile)
494 mf_trans_add(mfp, hp); /* may translate negative in positive nr */
495}
496
497/*
498 * block *hp is no longer in used, may put it in the free list of memfile *mfp
499 */
500 void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100501mf_free(memfile_T *mfp, bhdr_T *hp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000502{
503 vim_free(hp->bh_data); /* free the memory */
504 mf_rem_hash(mfp, hp); /* get *hp out of the hash list */
505 mf_rem_used(mfp, hp); /* get *hp out of the used list */
506 if (hp->bh_bnum < 0)
507 {
508 vim_free(hp); /* don't want negative numbers in free list */
509 mfp->mf_neg_count--;
510 }
511 else
512 mf_ins_free(mfp, hp); /* put *hp in the free list */
513}
514
Bram Moolenaar89f37272006-09-26 11:48:34 +0000515#if defined(__MORPHOS__) && defined(__libnix__)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000516/* function is missing in MorphOS libnix version */
517extern unsigned long *__stdfiledes;
518
519 static unsigned long
520fdtofh(int filedescriptor)
521{
522 return __stdfiledes[filedescriptor];
523}
524#endif
525
526/*
527 * Sync the memory file *mfp to disk.
528 * Flags:
529 * MFS_ALL If not given, blocks with negative numbers are not synced,
530 * even when they are dirty!
531 * MFS_STOP Stop syncing when a character becomes available, but sync at
532 * least one block.
533 * MFS_FLUSH Make sure buffers are flushed to disk, so they will survive a
534 * system crash.
535 * MFS_ZERO Only write block 0.
536 *
537 * Return FAIL for failure, OK otherwise
538 */
539 int
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100540mf_sync(memfile_T *mfp, int flags)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000541{
542 int status;
543 bhdr_T *hp;
Bram Moolenaar48e330a2016-02-23 14:53:34 +0100544#if defined(SYNC_DUP_CLOSE)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000545 int fd;
546#endif
547 int got_int_save = got_int;
548
549 if (mfp->mf_fd < 0) /* there is no file, nothing to do */
550 {
551 mfp->mf_dirty = FALSE;
552 return FAIL;
553 }
554
555 /* Only a CTRL-C while writing will break us here, not one typed
556 * previously. */
557 got_int = FALSE;
558
559 /*
560 * sync from last to first (may reduce the probability of an inconsistent
561 * file) If a write fails, it is very likely caused by a full filesystem.
562 * Then we only try to write blocks within the existing file. If that also
563 * fails then we give up.
564 */
565 status = OK;
566 for (hp = mfp->mf_used_last; hp != NULL; hp = hp->bh_prev)
567 if (((flags & MFS_ALL) || hp->bh_bnum >= 0)
568 && (hp->bh_flags & BH_DIRTY)
569 && (status == OK || (hp->bh_bnum >= 0
570 && hp->bh_bnum < mfp->mf_infile_count)))
571 {
572 if ((flags & MFS_ZERO) && hp->bh_bnum != 0)
573 continue;
574 if (mf_write(mfp, hp) == FAIL)
575 {
576 if (status == FAIL) /* double error: quit syncing */
577 break;
578 status = FAIL;
579 }
580 if (flags & MFS_STOP)
581 {
582 /* Stop when char available now. */
583 if (ui_char_avail())
584 break;
585 }
586 else
587 ui_breakcheck();
588 if (got_int)
589 break;
590 }
591
592 /*
593 * If the whole list is flushed, the memfile is not dirty anymore.
594 * In case of an error this flag is also set, to avoid trying all the time.
595 */
596 if (hp == NULL || status == FAIL)
597 mfp->mf_dirty = FALSE;
598
599 if ((flags & MFS_FLUSH) && *p_sws != NUL)
600 {
601#if defined(UNIX)
602# ifdef HAVE_FSYNC
603 /*
604 * most Unixes have the very useful fsync() function, just what we need.
605 * However, with OS/2 and EMX it is also available, but there are
606 * reports of bad problems with it (a bug in HPFS.IFS).
607 * So we disable use of it here in case someone tries to be smart
608 * and changes os_os2_cfg.h... (even though there is no __EMX__ test
609 * in the #if, as __EMX__ does not have sync(); we hope for a timely
610 * sync from the system itself).
611 */
612# if defined(__EMX__)
Bram Moolenaar84a05ac2013-05-06 04:24:17 +0200613 error "Don't use fsync with EMX! Read emxdoc.doc or emxfix01.doc for info."
Bram Moolenaar071d4272004-06-13 20:20:40 +0000614# endif
615 if (STRCMP(p_sws, "fsync") == 0)
616 {
617 if (fsync(mfp->mf_fd))
618 status = FAIL;
619 }
620 else
621# endif
622 /* OpenNT is strictly POSIX (Benzinger) */
623 /* Tandem/Himalaya NSK-OSS doesn't have sync() */
Bram Moolenaarba17ed62015-02-27 18:25:16 +0100624 /* No sync() on Stratus VOS */
625# if defined(__OPENNT) || defined(__TANDEM) || defined(__VOS__)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000626 fflush(NULL);
627# else
628 sync();
629# endif
630#endif
631#ifdef VMS
632 if (STRCMP(p_sws, "fsync") == 0)
633 {
634 if (fsync(mfp->mf_fd))
635 status = FAIL;
636 }
637#endif
Bram Moolenaar48e330a2016-02-23 14:53:34 +0100638#ifdef SYNC_DUP_CLOSE
Bram Moolenaar071d4272004-06-13 20:20:40 +0000639 /*
640 * Win32 is a bit more work: Duplicate the file handle and close it.
641 * This should flush the file to disk.
642 */
643 if ((fd = dup(mfp->mf_fd)) >= 0)
644 close(fd);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000645#endif
646#ifdef AMIGA
Bram Moolenaar5a6404c2006-11-01 17:12:57 +0000647# if defined(__AROS__) || defined(__amigaos4__)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000648 if (fsync(mfp->mf_fd) != 0)
649 status = FAIL;
650# else
651 /*
652 * Flush() only exists for AmigaDos 2.0.
653 * For 1.3 it should be done with close() + open(), but then the risk
654 * is that the open() may fail and lose the file....
655 */
656# ifdef FEAT_ARP
657 if (dos2)
658# endif
659# ifdef SASC
660 {
661 struct UFB *fp = chkufb(mfp->mf_fd);
662
663 if (fp != NULL)
664 Flush(fp->ufbfh);
665 }
666# else
667# if defined(_DCC) || defined(__GNUC__) || defined(__MORPHOS__)
668 {
Bram Moolenaar89f37272006-09-26 11:48:34 +0000669# if defined(__GNUC__) && !defined(__MORPHOS__) && defined(__libnix__)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000670 /* Have function (in libnix at least),
671 * but ain't got no prototype anywhere. */
672 extern unsigned long fdtofh(int filedescriptor);
673# endif
Bram Moolenaar89f37272006-09-26 11:48:34 +0000674# if !defined(__libnix__)
675 fflush(NULL);
676# else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000677 BPTR fh = (BPTR)fdtofh(mfp->mf_fd);
678
679 if (fh != 0)
680 Flush(fh);
Bram Moolenaar89f37272006-09-26 11:48:34 +0000681# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000682 }
683# else /* assume Manx */
684 Flush(_devtab[mfp->mf_fd].fd);
685# endif
686# endif
687# endif
688#endif /* AMIGA */
689 }
690
691 got_int |= got_int_save;
692
693 return status;
694}
695
696/*
Bram Moolenaarc32840f2006-01-14 21:23:38 +0000697 * For all blocks in memory file *mfp that have a positive block number set
698 * the dirty flag. These are blocks that need to be written to a newly
699 * created swapfile.
700 */
701 void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100702mf_set_dirty(memfile_T *mfp)
Bram Moolenaarc32840f2006-01-14 21:23:38 +0000703{
704 bhdr_T *hp;
705
706 for (hp = mfp->mf_used_last; hp != NULL; hp = hp->bh_prev)
707 if (hp->bh_bnum > 0)
708 hp->bh_flags |= BH_DIRTY;
709 mfp->mf_dirty = TRUE;
710}
711
712/*
Bram Moolenaar071d4272004-06-13 20:20:40 +0000713 * insert block *hp in front of hashlist of memfile *mfp
714 */
715 static void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100716mf_ins_hash(memfile_T *mfp, bhdr_T *hp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000717{
Bram Moolenaarb05b10a2011-03-22 18:10:45 +0100718 mf_hash_add_item(&mfp->mf_hash, (mf_hashitem_T *)hp);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000719}
720
721/*
722 * remove block *hp from hashlist of memfile list *mfp
723 */
724 static void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100725mf_rem_hash(memfile_T *mfp, bhdr_T *hp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000726{
Bram Moolenaarb05b10a2011-03-22 18:10:45 +0100727 mf_hash_rem_item(&mfp->mf_hash, (mf_hashitem_T *)hp);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000728}
729
730/*
731 * look in hash lists of memfile *mfp for block header with number 'nr'
732 */
733 static bhdr_T *
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100734mf_find_hash(memfile_T *mfp, blocknr_T nr)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000735{
Bram Moolenaarb05b10a2011-03-22 18:10:45 +0100736 return (bhdr_T *)mf_hash_find(&mfp->mf_hash, nr);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000737}
738
739/*
740 * insert block *hp in front of used list of memfile *mfp
741 */
742 static void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100743mf_ins_used(memfile_T *mfp, bhdr_T *hp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000744{
745 hp->bh_next = mfp->mf_used_first;
746 mfp->mf_used_first = hp;
747 hp->bh_prev = NULL;
748 if (hp->bh_next == NULL) /* list was empty, adjust last pointer */
749 mfp->mf_used_last = hp;
750 else
751 hp->bh_next->bh_prev = hp;
752 mfp->mf_used_count += hp->bh_page_count;
753 total_mem_used += hp->bh_page_count * mfp->mf_page_size;
754}
755
756/*
757 * remove block *hp from used list of memfile *mfp
758 */
759 static void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100760mf_rem_used(memfile_T *mfp, bhdr_T *hp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000761{
762 if (hp->bh_next == NULL) /* last block in used list */
763 mfp->mf_used_last = hp->bh_prev;
764 else
765 hp->bh_next->bh_prev = hp->bh_prev;
766 if (hp->bh_prev == NULL) /* first block in used list */
767 mfp->mf_used_first = hp->bh_next;
768 else
769 hp->bh_prev->bh_next = hp->bh_next;
770 mfp->mf_used_count -= hp->bh_page_count;
771 total_mem_used -= hp->bh_page_count * mfp->mf_page_size;
772}
773
774/*
775 * Release the least recently used block from the used list if the number
776 * of used memory blocks gets to big.
777 *
778 * Return the block header to the caller, including the memory block, so
779 * it can be re-used. Make sure the page_count is right.
Bram Moolenaarbc563362015-06-09 18:35:25 +0200780 *
781 * Returns NULL if no block is released.
Bram Moolenaar071d4272004-06-13 20:20:40 +0000782 */
783 static bhdr_T *
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100784mf_release(memfile_T *mfp, int page_count)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000785{
786 bhdr_T *hp;
787 int need_release;
788 buf_T *buf;
789
790 /* don't release while in mf_close_file() */
Bram Moolenaar47b8b152007-02-07 02:41:57 +0000791 if (mf_dont_release)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000792 return NULL;
793
794 /*
795 * Need to release a block if the number of blocks for this memfile is
796 * higher than the maximum or total memory used is over 'maxmemtot'
797 */
798 need_release = ((mfp->mf_used_count >= mfp->mf_used_count_max)
799 || (total_mem_used >> 10) >= (long_u)p_mmt);
800
801 /*
802 * Try to create a swap file if the amount of memory used is getting too
803 * high.
804 */
805 if (mfp->mf_fd < 0 && need_release && p_uc)
806 {
807 /* find for which buffer this memfile is */
808 for (buf = firstbuf; buf != NULL; buf = buf->b_next)
809 if (buf->b_ml.ml_mfp == mfp)
810 break;
811 if (buf != NULL && buf->b_may_swap)
812 ml_open_file(buf);
813 }
814
815 /*
816 * don't release a block if
817 * there is no file for this memfile
818 * or
819 * the number of blocks for this memfile is lower than the maximum
820 * and
821 * total memory used is not up to 'maxmemtot'
822 */
823 if (mfp->mf_fd < 0 || !need_release)
824 return NULL;
825
826 for (hp = mfp->mf_used_last; hp != NULL; hp = hp->bh_prev)
827 if (!(hp->bh_flags & BH_LOCKED))
828 break;
829 if (hp == NULL) /* not a single one that can be released */
830 return NULL;
831
832 /*
833 * If the block is dirty, write it.
834 * If the write fails we don't free it.
835 */
836 if ((hp->bh_flags & BH_DIRTY) && mf_write(mfp, hp) == FAIL)
837 return NULL;
838
839 mf_rem_used(mfp, hp);
840 mf_rem_hash(mfp, hp);
841
842 /*
843 * If a bhdr_T is returned, make sure that the page_count of bh_data is
844 * right
845 */
846 if (hp->bh_page_count != page_count)
847 {
848 vim_free(hp->bh_data);
849 if ((hp->bh_data = alloc(mfp->mf_page_size * page_count)) == NULL)
850 {
851 vim_free(hp);
852 return NULL;
853 }
854 hp->bh_page_count = page_count;
855 }
856 return hp;
857}
858
859/*
860 * release as many blocks as possible
861 * Used in case of out of memory
862 *
863 * return TRUE if any memory was released
864 */
865 int
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100866mf_release_all(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000867{
868 buf_T *buf;
869 memfile_T *mfp;
870 bhdr_T *hp;
871 int retval = FALSE;
872
873 for (buf = firstbuf; buf != NULL; buf = buf->b_next)
874 {
875 mfp = buf->b_ml.ml_mfp;
876 if (mfp != NULL)
877 {
878 /* If no swap file yet, may open one */
879 if (mfp->mf_fd < 0 && buf->b_may_swap)
880 ml_open_file(buf);
881
882 /* only if there is a swapfile */
883 if (mfp->mf_fd >= 0)
884 {
885 for (hp = mfp->mf_used_last; hp != NULL; )
886 {
887 if (!(hp->bh_flags & BH_LOCKED)
888 && (!(hp->bh_flags & BH_DIRTY)
889 || mf_write(mfp, hp) != FAIL))
890 {
891 mf_rem_used(mfp, hp);
892 mf_rem_hash(mfp, hp);
893 mf_free_bhdr(hp);
894 hp = mfp->mf_used_last; /* re-start, list was changed */
895 retval = TRUE;
896 }
897 else
898 hp = hp->bh_prev;
899 }
900 }
901 }
902 }
903 return retval;
904}
905
906/*
907 * Allocate a block header and a block of memory for it
908 */
909 static bhdr_T *
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100910mf_alloc_bhdr(memfile_T *mfp, int page_count)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000911{
912 bhdr_T *hp;
913
914 if ((hp = (bhdr_T *)alloc((unsigned)sizeof(bhdr_T))) != NULL)
915 {
916 if ((hp->bh_data = (char_u *)alloc(mfp->mf_page_size * page_count))
917 == NULL)
918 {
919 vim_free(hp); /* not enough memory */
920 return NULL;
921 }
922 hp->bh_page_count = page_count;
923 }
924 return hp;
925}
926
927/*
928 * Free a block header and the block of memory for it
929 */
930 static void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100931mf_free_bhdr(bhdr_T *hp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000932{
933 vim_free(hp->bh_data);
934 vim_free(hp);
935}
936
937/*
938 * insert entry *hp in the free list
939 */
940 static void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100941mf_ins_free(memfile_T *mfp, bhdr_T *hp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000942{
943 hp->bh_next = mfp->mf_free_first;
944 mfp->mf_free_first = hp;
945}
946
947/*
948 * remove the first entry from the free list and return a pointer to it
949 * Note: caller must check that mfp->mf_free_first is not NULL!
950 */
951 static bhdr_T *
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100952mf_rem_free(memfile_T *mfp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000953{
954 bhdr_T *hp;
955
956 hp = mfp->mf_free_first;
957 mfp->mf_free_first = hp->bh_next;
958 return hp;
959}
960
961/*
962 * read a block from disk
963 *
964 * Return FAIL for failure, OK otherwise
965 */
966 static int
Bram Moolenaar52ea13d2016-01-30 18:51:09 +0100967mf_read(memfile_T *mfp, bhdr_T *hp)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000968{
969 off_t offset;
970 unsigned page_size;
971 unsigned size;
972
973 if (mfp->mf_fd < 0) /* there is no file, can't read */
974 return FAIL;
975
976 page_size = mfp->mf_page_size;
977 offset = (off_t)page_size * hp->bh_bnum;
978 size = page_size * hp->bh_page_count;
979 if (lseek(mfp->mf_fd, offset, SEEK_SET) != offset)
980 {
Bram Moolenaar3f2d9812006-11-07 17:02:30 +0000981 PERROR(_("E294: Seek error in swap file read"));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000982 return FAIL;
983 }
Bram Moolenaar540fc6f2010-12-17 16:27:16 +0100984 if ((unsigned)read_eintr(mfp->mf_fd, hp->bh_data, size) != size)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000985 {
Bram Moolenaar3f2d9812006-11-07 17:02:30 +0000986 PERROR(_("E295: Read error in swap file"));
Bram Moolenaar071d4272004-06-13 20:20:40 +0000987 return FAIL;
988 }
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200989
990#ifdef FEAT_CRYPT
Bram Moolenaar4a8c2cf2015-12-19 15:28:18 +0100991 /* Decrypt if 'key' is set and this is a data block. And when changing the
992 * key. */
993 if (*mfp->mf_buffer->b_p_key != NUL || mfp->mf_old_key != NULL)
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +0200994 ml_decrypt_data(mfp, hp->bh_data, offset, size);
995#endif
996
Bram Moolenaar071d4272004-06-13 20:20:40 +0000997 return OK;
998}
999
1000/*
1001 * write a block to disk
1002 *
1003 * Return FAIL for failure, OK otherwise
1004 */
1005 static int
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01001006mf_write(memfile_T *mfp, bhdr_T *hp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001007{
1008 off_t offset; /* offset in the file */
1009 blocknr_T nr; /* block nr which is being written */
1010 bhdr_T *hp2;
1011 unsigned page_size; /* number of bytes in a page */
1012 unsigned page_count; /* number of pages written */
1013 unsigned size; /* number of bytes written */
1014
1015 if (mfp->mf_fd < 0) /* there is no file, can't write */
1016 return FAIL;
1017
1018 if (hp->bh_bnum < 0) /* must assign file block number */
1019 if (mf_trans_add(mfp, hp) == FAIL)
1020 return FAIL;
1021
1022 page_size = mfp->mf_page_size;
1023
1024 /*
1025 * We don't want gaps in the file. Write the blocks in front of *hp
1026 * to extend the file.
1027 * If block 'mf_infile_count' is not in the hash list, it has been
1028 * freed. Fill the space in the file with data from the current block.
1029 */
1030 for (;;)
1031 {
1032 nr = hp->bh_bnum;
1033 if (nr > mfp->mf_infile_count) /* beyond end of file */
1034 {
1035 nr = mfp->mf_infile_count;
Bram Moolenaar84a05ac2013-05-06 04:24:17 +02001036 hp2 = mf_find_hash(mfp, nr); /* NULL caught below */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001037 }
1038 else
1039 hp2 = hp;
1040
1041 offset = (off_t)page_size * nr;
1042 if (lseek(mfp->mf_fd, offset, SEEK_SET) != offset)
1043 {
Bram Moolenaar3f2d9812006-11-07 17:02:30 +00001044 PERROR(_("E296: Seek error in swap file write"));
Bram Moolenaar071d4272004-06-13 20:20:40 +00001045 return FAIL;
1046 }
1047 if (hp2 == NULL) /* freed block, fill with dummy data */
1048 page_count = 1;
1049 else
1050 page_count = hp2->bh_page_count;
1051 size = page_size * page_count;
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001052 if (mf_write_block(mfp, hp2 == NULL ? hp : hp2, offset, size) == FAIL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001053 {
1054 /*
1055 * Avoid repeating the error message, this mostly happens when the
Bram Moolenaar49325942007-05-10 19:19:59 +00001056 * disk is full. We give the message again only after a successful
Bram Moolenaar071d4272004-06-13 20:20:40 +00001057 * write or when hitting a key. We keep on trying, in case some
1058 * space becomes available.
1059 */
1060 if (!did_swapwrite_msg)
1061 EMSG(_("E297: Write error in swap file"));
1062 did_swapwrite_msg = TRUE;
1063 return FAIL;
1064 }
1065 did_swapwrite_msg = FALSE;
1066 if (hp2 != NULL) /* written a non-dummy block */
1067 hp2->bh_flags &= ~BH_DIRTY;
1068 /* appended to the file */
1069 if (nr + (blocknr_T)page_count > mfp->mf_infile_count)
1070 mfp->mf_infile_count = nr + page_count;
1071 if (nr == hp->bh_bnum) /* written the desired block */
1072 break;
1073 }
1074 return OK;
1075}
1076
1077/*
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001078 * Write block "hp" with data size "size" to file "mfp->mf_fd".
1079 * Takes care of encryption.
1080 * Return FAIL or OK.
1081 */
1082 static int
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01001083mf_write_block(
1084 memfile_T *mfp,
1085 bhdr_T *hp,
1086 off_t offset UNUSED,
1087 unsigned size)
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001088{
1089 char_u *data = hp->bh_data;
1090 int result = OK;
1091
1092#ifdef FEAT_CRYPT
1093 /* Encrypt if 'key' is set and this is a data block. */
1094 if (*mfp->mf_buffer->b_p_key != NUL)
1095 {
1096 data = ml_encrypt_data(mfp, data, offset, size);
1097 if (data == NULL)
1098 return FAIL;
1099 }
1100#endif
1101
Bram Moolenaar540fc6f2010-12-17 16:27:16 +01001102 if ((unsigned)write_eintr(mfp->mf_fd, data, size) != size)
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001103 result = FAIL;
1104
1105#ifdef FEAT_CRYPT
1106 if (data != hp->bh_data)
1107 vim_free(data);
1108#endif
1109
1110 return result;
1111}
1112
1113/*
Bram Moolenaar071d4272004-06-13 20:20:40 +00001114 * Make block number for *hp positive and add it to the translation list
1115 *
1116 * Return FAIL for failure, OK otherwise
1117 */
1118 static int
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01001119mf_trans_add(memfile_T *mfp, bhdr_T *hp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001120{
1121 bhdr_T *freep;
1122 blocknr_T new_bnum;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001123 NR_TRANS *np;
1124 int page_count;
1125
1126 if (hp->bh_bnum >= 0) /* it's already positive */
1127 return OK;
1128
1129 if ((np = (NR_TRANS *)alloc((unsigned)sizeof(NR_TRANS))) == NULL)
1130 return FAIL;
1131
1132/*
Bram Moolenaara8ffcbb2010-06-21 06:15:46 +02001133 * Get a new number for the block.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001134 * If the first item in the free list has sufficient pages, use its number
1135 * Otherwise use mf_blocknr_max.
1136 */
1137 freep = mfp->mf_free_first;
1138 page_count = hp->bh_page_count;
1139 if (freep != NULL && freep->bh_page_count >= page_count)
1140 {
1141 new_bnum = freep->bh_bnum;
1142 /*
Bram Moolenaar84a05ac2013-05-06 04:24:17 +02001143 * If the page count of the free block was larger, reduce it.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001144 * If the page count matches, remove the block from the free list
1145 */
1146 if (freep->bh_page_count > page_count)
1147 {
1148 freep->bh_bnum += page_count;
1149 freep->bh_page_count -= page_count;
1150 }
1151 else
1152 {
1153 freep = mf_rem_free(mfp);
1154 vim_free(freep);
1155 }
1156 }
1157 else
1158 {
1159 new_bnum = mfp->mf_blocknr_max;
1160 mfp->mf_blocknr_max += page_count;
1161 }
1162
1163 np->nt_old_bnum = hp->bh_bnum; /* adjust number */
1164 np->nt_new_bnum = new_bnum;
1165
1166 mf_rem_hash(mfp, hp); /* remove from old hash list */
1167 hp->bh_bnum = new_bnum;
1168 mf_ins_hash(mfp, hp); /* insert in new hash list */
1169
Bram Moolenaarb05b10a2011-03-22 18:10:45 +01001170 /* Insert "np" into "mf_trans" hashtable with key "np->nt_old_bnum" */
1171 mf_hash_add_item(&mfp->mf_trans, (mf_hashitem_T *)np);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001172
1173 return OK;
1174}
1175
1176/*
Bram Moolenaarbc563362015-06-09 18:35:25 +02001177 * Lookup a translation from the trans lists and delete the entry.
Bram Moolenaar071d4272004-06-13 20:20:40 +00001178 *
1179 * Return the positive new number when found, the old number when not found
1180 */
1181 blocknr_T
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01001182mf_trans_del(memfile_T *mfp, blocknr_T old_nr)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001183{
Bram Moolenaar071d4272004-06-13 20:20:40 +00001184 NR_TRANS *np;
1185 blocknr_T new_bnum;
1186
Bram Moolenaarb05b10a2011-03-22 18:10:45 +01001187 np = (NR_TRANS *)mf_hash_find(&mfp->mf_trans, old_nr);
1188
Bram Moolenaar071d4272004-06-13 20:20:40 +00001189 if (np == NULL) /* not found */
1190 return old_nr;
1191
1192 mfp->mf_neg_count--;
1193 new_bnum = np->nt_new_bnum;
Bram Moolenaarb05b10a2011-03-22 18:10:45 +01001194
1195 /* remove entry from the trans list */
1196 mf_hash_rem_item(&mfp->mf_trans, (mf_hashitem_T *)np);
1197
Bram Moolenaar071d4272004-06-13 20:20:40 +00001198 vim_free(np);
1199
1200 return new_bnum;
1201}
1202
1203/*
1204 * Set mfp->mf_ffname according to mfp->mf_fname and some other things.
1205 * Only called when creating or renaming the swapfile. Either way it's a new
1206 * name so we must work out the full path name.
1207 */
1208 void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01001209mf_set_ffname(memfile_T *mfp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001210{
1211 mfp->mf_ffname = FullName_save(mfp->mf_fname, FALSE);
1212}
1213
1214/*
1215 * Make the name of the file used for the memfile a full path.
1216 * Used before doing a :cd
1217 */
1218 void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01001219mf_fullname(memfile_T *mfp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001220{
1221 if (mfp != NULL && mfp->mf_fname != NULL && mfp->mf_ffname != NULL)
1222 {
1223 vim_free(mfp->mf_fname);
1224 mfp->mf_fname = mfp->mf_ffname;
1225 mfp->mf_ffname = NULL;
1226 }
1227}
1228
1229/*
1230 * return TRUE if there are any translations pending for 'mfp'
1231 */
1232 int
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01001233mf_need_trans(memfile_T *mfp)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001234{
1235 return (mfp->mf_fname != NULL && mfp->mf_neg_count > 0);
1236}
1237
1238/*
1239 * Open a swap file for a memfile.
1240 * The "fname" must be in allocated memory, and is consumed (also when an
1241 * error occurs).
1242 */
1243 static void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01001244mf_do_open(
1245 memfile_T *mfp,
1246 char_u *fname,
1247 int flags) /* flags for open() */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001248{
1249#ifdef HAVE_LSTAT
1250 struct stat sb;
1251#endif
1252
1253 mfp->mf_fname = fname;
1254
1255 /*
1256 * Get the full path name before the open, because this is
1257 * not possible after the open on the Amiga.
1258 * fname cannot be NameBuff, because it must have been allocated.
1259 */
1260 mf_set_ffname(mfp);
Bram Moolenaar48e330a2016-02-23 14:53:34 +01001261#if defined(MSWIN)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001262 /*
Bram Moolenaar84a05ac2013-05-06 04:24:17 +02001263 * A ":!cd e:xxx" may change the directory without us knowing, use the
Bram Moolenaar071d4272004-06-13 20:20:40 +00001264 * full pathname always. Careful: This frees fname!
1265 */
1266 mf_fullname(mfp);
1267#endif
1268
1269#ifdef HAVE_LSTAT
1270 /*
1271 * Extra security check: When creating a swap file it really shouldn't
1272 * exist yet. If there is a symbolic link, this is most likely an attack.
1273 */
1274 if ((flags & O_CREAT) && mch_lstat((char *)mfp->mf_fname, &sb) >= 0)
1275 {
1276 mfp->mf_fd = -1;
1277 EMSG(_("E300: Swap file already exists (symlink attack?)"));
1278 }
1279 else
1280#endif
1281 {
1282 /*
1283 * try to open the file
1284 */
Bram Moolenaara5792f52005-11-23 21:25:05 +00001285 flags |= O_EXTRA | O_NOFOLLOW;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001286#ifdef WIN32
1287 /* Prevent handle inheritance that cause problems with Cscope
1288 * (swap file may not be deleted if cscope connection was open after
1289 * the file) */
1290 flags |= O_NOINHERIT;
1291#endif
1292 mfp->mf_fd = mch_open_rw((char *)mfp->mf_fname, flags);
1293 }
1294
1295 /*
1296 * If the file cannot be opened, use memory only
1297 */
1298 if (mfp->mf_fd < 0)
1299 {
1300 vim_free(mfp->mf_fname);
1301 vim_free(mfp->mf_ffname);
1302 mfp->mf_fname = NULL;
1303 mfp->mf_ffname = NULL;
1304 }
1305 else
Bram Moolenaar588ebeb2008-05-07 17:09:24 +00001306 {
Bram Moolenaarf05da212009-11-17 16:13:15 +00001307#ifdef HAVE_FD_CLOEXEC
1308 int fdflags = fcntl(mfp->mf_fd, F_GETFD);
1309 if (fdflags >= 0 && (fdflags & FD_CLOEXEC) == 0)
Bram Moolenaarfbc4b4d2016-02-07 15:14:01 +01001310 (void)fcntl(mfp->mf_fd, F_SETFD, fdflags | FD_CLOEXEC);
Bram Moolenaarf05da212009-11-17 16:13:15 +00001311#endif
Bram Moolenaar5bd32f42014-04-02 14:05:38 +02001312#if defined(HAVE_SELINUX) || defined(HAVE_SMACK)
Bram Moolenaar588ebeb2008-05-07 17:09:24 +00001313 mch_copy_sec(fname, mfp->mf_fname);
1314#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001315 mch_hide(mfp->mf_fname); /* try setting the 'hidden' flag */
Bram Moolenaar588ebeb2008-05-07 17:09:24 +00001316 }
Bram Moolenaar071d4272004-06-13 20:20:40 +00001317}
Bram Moolenaarb05b10a2011-03-22 18:10:45 +01001318
1319/*
1320 * Implementation of mf_hashtab_T follows.
1321 */
1322
1323/*
1324 * The number of buckets in the hashtable is increased by a factor of
1325 * MHT_GROWTH_FACTOR when the average number of items per bucket
1326 * exceeds 2 ^ MHT_LOG_LOAD_FACTOR.
1327 */
1328#define MHT_LOG_LOAD_FACTOR 6
1329#define MHT_GROWTH_FACTOR 2 /* must be a power of two */
1330
1331/*
1332 * Initialize an empty hash table.
1333 */
1334 static void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01001335mf_hash_init(mf_hashtab_T *mht)
Bram Moolenaarb05b10a2011-03-22 18:10:45 +01001336{
1337 vim_memset(mht, 0, sizeof(mf_hashtab_T));
1338 mht->mht_buckets = mht->mht_small_buckets;
1339 mht->mht_mask = MHT_INIT_SIZE - 1;
1340}
1341
1342/*
1343 * Free the array of a hash table. Does not free the items it contains!
1344 * The hash table must not be used again without another mf_hash_init() call.
1345 */
1346 static void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01001347mf_hash_free(mf_hashtab_T *mht)
Bram Moolenaarb05b10a2011-03-22 18:10:45 +01001348{
1349 if (mht->mht_buckets != mht->mht_small_buckets)
1350 vim_free(mht->mht_buckets);
1351}
1352
1353/*
1354 * Free the array of a hash table and all the items it contains.
1355 */
1356 static void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01001357mf_hash_free_all(mf_hashtab_T *mht)
Bram Moolenaarb05b10a2011-03-22 18:10:45 +01001358{
1359 long_u idx;
1360 mf_hashitem_T *mhi;
1361 mf_hashitem_T *next;
1362
1363 for (idx = 0; idx <= mht->mht_mask; idx++)
1364 for (mhi = mht->mht_buckets[idx]; mhi != NULL; mhi = next)
1365 {
1366 next = mhi->mhi_next;
1367 vim_free(mhi);
1368 }
1369
1370 mf_hash_free(mht);
1371}
1372
1373/*
1374 * Find "key" in hashtable "mht".
1375 * Returns a pointer to a mf_hashitem_T or NULL if the item was not found.
1376 */
1377 static mf_hashitem_T *
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01001378mf_hash_find(mf_hashtab_T *mht, blocknr_T key)
Bram Moolenaarb05b10a2011-03-22 18:10:45 +01001379{
1380 mf_hashitem_T *mhi;
1381
1382 mhi = mht->mht_buckets[key & mht->mht_mask];
1383 while (mhi != NULL && mhi->mhi_key != key)
1384 mhi = mhi->mhi_next;
1385
1386 return mhi;
1387}
1388
1389/*
1390 * Add item "mhi" to hashtable "mht".
1391 * "mhi" must not be NULL.
1392 */
1393 static void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01001394mf_hash_add_item(mf_hashtab_T *mht, mf_hashitem_T *mhi)
Bram Moolenaarb05b10a2011-03-22 18:10:45 +01001395{
1396 long_u idx;
1397
1398 idx = mhi->mhi_key & mht->mht_mask;
1399 mhi->mhi_next = mht->mht_buckets[idx];
1400 mhi->mhi_prev = NULL;
1401 if (mhi->mhi_next != NULL)
1402 mhi->mhi_next->mhi_prev = mhi;
1403 mht->mht_buckets[idx] = mhi;
1404
1405 mht->mht_count++;
1406
1407 /*
1408 * Grow hashtable when we have more thank 2^MHT_LOG_LOAD_FACTOR
1409 * items per bucket on average
1410 */
1411 if (mht->mht_fixed == 0
1412 && (mht->mht_count >> MHT_LOG_LOAD_FACTOR) > mht->mht_mask)
1413 {
1414 if (mf_hash_grow(mht) == FAIL)
1415 {
1416 /* stop trying to grow after first failure to allocate memory */
1417 mht->mht_fixed = 1;
1418 }
1419 }
1420}
1421
1422/*
1423 * Remove item "mhi" from hashtable "mht".
1424 * "mhi" must not be NULL and must have been inserted into "mht".
1425 */
1426 static void
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01001427mf_hash_rem_item(mf_hashtab_T *mht, mf_hashitem_T *mhi)
Bram Moolenaarb05b10a2011-03-22 18:10:45 +01001428{
1429 if (mhi->mhi_prev == NULL)
1430 mht->mht_buckets[mhi->mhi_key & mht->mht_mask] = mhi->mhi_next;
1431 else
1432 mhi->mhi_prev->mhi_next = mhi->mhi_next;
1433
1434 if (mhi->mhi_next != NULL)
1435 mhi->mhi_next->mhi_prev = mhi->mhi_prev;
1436
1437 mht->mht_count--;
1438
1439 /* We could shrink the table here, but it typically takes little memory,
1440 * so why bother? */
1441}
1442
1443/*
1444 * Increase number of buckets in the hashtable by MHT_GROWTH_FACTOR and
1445 * rehash items.
1446 * Returns FAIL when out of memory.
1447 */
1448 static int
Bram Moolenaar52ea13d2016-01-30 18:51:09 +01001449mf_hash_grow(mf_hashtab_T *mht)
Bram Moolenaarb05b10a2011-03-22 18:10:45 +01001450{
1451 long_u i, j;
1452 int shift;
1453 mf_hashitem_T *mhi;
1454 mf_hashitem_T *tails[MHT_GROWTH_FACTOR];
1455 mf_hashitem_T **buckets;
1456 size_t size;
1457
1458 size = (mht->mht_mask + 1) * MHT_GROWTH_FACTOR * sizeof(void *);
1459 buckets = (mf_hashitem_T **)lalloc_clear(size, FALSE);
1460 if (buckets == NULL)
1461 return FAIL;
1462
1463 shift = 0;
1464 while ((mht->mht_mask >> shift) != 0)
1465 shift++;
1466
1467 for (i = 0; i <= mht->mht_mask; i++)
1468 {
1469 /*
1470 * Traverse the items in the i-th original bucket and move them into
1471 * MHT_GROWTH_FACTOR new buckets, preserving their relative order
1472 * within each new bucket. Preserving the order is important because
1473 * mf_get() tries to keep most recently used items at the front of
1474 * each bucket.
1475 *
1476 * Here we strongly rely on the fact the hashes are computed modulo
1477 * a power of two.
1478 */
1479
1480 vim_memset(tails, 0, sizeof(tails));
1481
1482 for (mhi = mht->mht_buckets[i]; mhi != NULL; mhi = mhi->mhi_next)
1483 {
1484 j = (mhi->mhi_key >> shift) & (MHT_GROWTH_FACTOR - 1);
1485 if (tails[j] == NULL)
1486 {
1487 buckets[i + (j << shift)] = mhi;
1488 tails[j] = mhi;
1489 mhi->mhi_prev = NULL;
1490 }
1491 else
1492 {
1493 tails[j]->mhi_next = mhi;
1494 mhi->mhi_prev = tails[j];
1495 tails[j] = mhi;
1496 }
1497 }
1498
1499 for (j = 0; j < MHT_GROWTH_FACTOR; j++)
1500 if (tails[j] != NULL)
1501 tails[j]->mhi_next = NULL;
1502 }
1503
1504 if (mht->mht_buckets != mht->mht_small_buckets)
1505 vim_free(mht->mht_buckets);
1506
1507 mht->mht_buckets = buckets;
1508 mht->mht_mask = (mht->mht_mask + 1) * MHT_GROWTH_FACTOR - 1;
1509
1510 return OK;
1511}