xref: /vim-8.2.3635/src/memfile.c (revision 540fc6fb)
1071d4279SBram Moolenaar /* vi:set ts=8 sts=4 sw=4:
2071d4279SBram Moolenaar  *
3071d4279SBram Moolenaar  * VIM - Vi IMproved	by Bram Moolenaar
4071d4279SBram Moolenaar  *
5071d4279SBram Moolenaar  * Do ":help uganda"  in Vim to read copying and usage conditions.
6071d4279SBram Moolenaar  * Do ":help credits" in Vim to see a list of people who contributed.
7071d4279SBram Moolenaar  * See README.txt for an overview of the Vim source code.
8071d4279SBram Moolenaar  */
9071d4279SBram Moolenaar 
10071d4279SBram Moolenaar /*
11071d4279SBram Moolenaar  * memfile.c: Contains the functions for handling blocks of memory which can
12071d4279SBram Moolenaar  * be stored in a file. This is the implementation of a sort of virtual memory.
13071d4279SBram Moolenaar  *
14071d4279SBram Moolenaar  * A memfile consists of a sequence of blocks. The blocks numbered from 0
15071d4279SBram Moolenaar  * upwards have been assigned a place in the actual file. The block number
16071d4279SBram Moolenaar  * is equal to the page number in the file. The
17071d4279SBram Moolenaar  * blocks with negative numbers are currently in memory only. They can be
18071d4279SBram Moolenaar  * assigned a place in the file when too much memory is being used. At that
19071d4279SBram Moolenaar  * moment they get a new, positive, number. A list is used for translation of
20071d4279SBram Moolenaar  * negative to positive numbers.
21071d4279SBram Moolenaar  *
22071d4279SBram Moolenaar  * The size of a block is a multiple of a page size, normally the page size of
23071d4279SBram Moolenaar  * the device the file is on. Most blocks are 1 page long. A Block of multiple
24071d4279SBram Moolenaar  * pages is used for a line that does not fit in a single page.
25071d4279SBram Moolenaar  *
26071d4279SBram Moolenaar  * Each block can be in memory and/or in a file. The block stays in memory
27071d4279SBram Moolenaar  * as long as it is locked. If it is no longer locked it can be swapped out to
28071d4279SBram Moolenaar  * the file. It is only written to the file if it has been changed.
29071d4279SBram Moolenaar  *
30071d4279SBram Moolenaar  * Under normal operation the file is created when opening the memory file and
31071d4279SBram Moolenaar  * deleted when closing the memory file. Only with recovery an existing memory
32071d4279SBram Moolenaar  * file is opened.
33071d4279SBram Moolenaar  */
34071d4279SBram Moolenaar 
35c236c16dSBram Moolenaar #if defined(MSDOS) || defined(WIN16) || defined(WIN32) || defined(_WIN64)
36362e1a30SBram Moolenaar # include "vimio.h"	/* for lseek(), must be before vim.h */
37071d4279SBram Moolenaar #endif
38071d4279SBram Moolenaar 
39071d4279SBram Moolenaar #include "vim.h"
40071d4279SBram Moolenaar 
41071d4279SBram Moolenaar /*
42071d4279SBram Moolenaar  * Some systems have the page size in statfs.f_bsize, some in stat.st_blksize
43071d4279SBram Moolenaar  */
44071d4279SBram Moolenaar #ifdef HAVE_ST_BLKSIZE
45071d4279SBram Moolenaar # define STATFS stat
46071d4279SBram Moolenaar # define F_BSIZE st_blksize
47071d4279SBram Moolenaar # define fstatfs(fd, buf, len, nul) mch_fstat((fd), (buf))
48071d4279SBram Moolenaar #else
49071d4279SBram Moolenaar # ifdef HAVE_SYS_STATFS_H
50071d4279SBram Moolenaar #  include <sys/statfs.h>
51071d4279SBram Moolenaar #  define STATFS statfs
52071d4279SBram Moolenaar #  define F_BSIZE f_bsize
53071d4279SBram Moolenaar #  ifdef __MINT__		/* do we still need this? */
54071d4279SBram Moolenaar #   define fstatfs(fd, buf, len, nul) mch_fstat((fd), (buf))
55071d4279SBram Moolenaar #  endif
56071d4279SBram Moolenaar # endif
57071d4279SBram Moolenaar #endif
58071d4279SBram Moolenaar 
59071d4279SBram Moolenaar /*
60071d4279SBram Moolenaar  * for Amiga Dos 2.0x we use Flush
61071d4279SBram Moolenaar  */
62071d4279SBram Moolenaar #ifdef AMIGA
63071d4279SBram Moolenaar # ifdef FEAT_ARP
64071d4279SBram Moolenaar extern int dos2;			/* this is in os_amiga.c */
65071d4279SBram Moolenaar # endif
66071d4279SBram Moolenaar # ifdef SASC
67071d4279SBram Moolenaar #  include <proto/dos.h>
68071d4279SBram Moolenaar #  include <ios1.h>			/* for chkufb() */
69071d4279SBram Moolenaar # endif
70071d4279SBram Moolenaar #endif
71071d4279SBram Moolenaar 
72071d4279SBram Moolenaar #define MEMFILE_PAGE_SIZE 4096		/* default page size */
73071d4279SBram Moolenaar 
74071d4279SBram Moolenaar static long_u	total_mem_used = 0;	/* total memory used for memfiles */
75071d4279SBram Moolenaar 
76071d4279SBram Moolenaar static void mf_ins_hash __ARGS((memfile_T *, bhdr_T *));
77071d4279SBram Moolenaar static void mf_rem_hash __ARGS((memfile_T *, bhdr_T *));
78071d4279SBram Moolenaar static bhdr_T *mf_find_hash __ARGS((memfile_T *, blocknr_T));
79071d4279SBram Moolenaar static void mf_ins_used __ARGS((memfile_T *, bhdr_T *));
80071d4279SBram Moolenaar static void mf_rem_used __ARGS((memfile_T *, bhdr_T *));
81071d4279SBram Moolenaar static bhdr_T *mf_release __ARGS((memfile_T *, int));
82071d4279SBram Moolenaar static bhdr_T *mf_alloc_bhdr __ARGS((memfile_T *, int));
83071d4279SBram Moolenaar static void mf_free_bhdr __ARGS((bhdr_T *));
84071d4279SBram Moolenaar static void mf_ins_free __ARGS((memfile_T *, bhdr_T *));
85071d4279SBram Moolenaar static bhdr_T *mf_rem_free __ARGS((memfile_T *));
86071d4279SBram Moolenaar static int  mf_read __ARGS((memfile_T *, bhdr_T *));
87071d4279SBram Moolenaar static int  mf_write __ARGS((memfile_T *, bhdr_T *));
88a8ffcbbfSBram Moolenaar static int  mf_write_block __ARGS((memfile_T *mfp, bhdr_T *hp, off_t offset, unsigned size));
89071d4279SBram Moolenaar static int  mf_trans_add __ARGS((memfile_T *, bhdr_T *));
90071d4279SBram Moolenaar static void mf_do_open __ARGS((memfile_T *, char_u *, int));
91071d4279SBram Moolenaar 
92071d4279SBram Moolenaar /*
93071d4279SBram Moolenaar  * The functions for using a memfile:
94071d4279SBram Moolenaar  *
95071d4279SBram Moolenaar  * mf_open()	    open a new or existing memfile
96071d4279SBram Moolenaar  * mf_open_file()   open a swap file for an existing memfile
97071d4279SBram Moolenaar  * mf_close()	    close (and delete) a memfile
98071d4279SBram Moolenaar  * mf_new()	    create a new block in a memfile and lock it
99071d4279SBram Moolenaar  * mf_get()	    get an existing block and lock it
100071d4279SBram Moolenaar  * mf_put()	    unlock a block, may be marked for writing
101071d4279SBram Moolenaar  * mf_free()	    remove a block
102071d4279SBram Moolenaar  * mf_sync()	    sync changed parts of memfile to disk
103071d4279SBram Moolenaar  * mf_release_all() release as much memory as possible
104071d4279SBram Moolenaar  * mf_trans_del()   may translate negative to positive block number
105071d4279SBram Moolenaar  * mf_fullname()    make file name full path (use before first :cd)
106071d4279SBram Moolenaar  */
107071d4279SBram Moolenaar 
108071d4279SBram Moolenaar /*
109071d4279SBram Moolenaar  * Open an existing or new memory block file.
110071d4279SBram Moolenaar  *
111071d4279SBram Moolenaar  *  fname:	name of file to use (NULL means no file at all)
112071d4279SBram Moolenaar  *		Note: fname must have been allocated, it is not copied!
113071d4279SBram Moolenaar  *			If opening the file fails, fname is freed.
114071d4279SBram Moolenaar  *  flags:	flags for open() call
115071d4279SBram Moolenaar  *
116071d4279SBram Moolenaar  *  If fname != NULL and file cannot be opened, fail.
117071d4279SBram Moolenaar  *
118071d4279SBram Moolenaar  * return value: identifier for this memory block file.
119071d4279SBram Moolenaar  */
120071d4279SBram Moolenaar     memfile_T *
121071d4279SBram Moolenaar mf_open(fname, flags)
122071d4279SBram Moolenaar     char_u	*fname;
123071d4279SBram Moolenaar     int		flags;
124071d4279SBram Moolenaar {
125071d4279SBram Moolenaar     memfile_T		*mfp;
126071d4279SBram Moolenaar     int			i;
127071d4279SBram Moolenaar     off_t		size;
128071d4279SBram Moolenaar #if defined(STATFS) && defined(UNIX) && !defined(__QNX__)
129071d4279SBram Moolenaar # define USE_FSTATFS
130071d4279SBram Moolenaar     struct STATFS	stf;
131071d4279SBram Moolenaar #endif
132071d4279SBram Moolenaar 
133071d4279SBram Moolenaar     if ((mfp = (memfile_T *)alloc((unsigned)sizeof(memfile_T))) == NULL)
134071d4279SBram Moolenaar 	return NULL;
135071d4279SBram Moolenaar 
136071d4279SBram Moolenaar     if (fname == NULL)	    /* no file for this memfile, use memory only */
137071d4279SBram Moolenaar     {
138071d4279SBram Moolenaar 	mfp->mf_fname = NULL;
139071d4279SBram Moolenaar 	mfp->mf_ffname = NULL;
140071d4279SBram Moolenaar 	mfp->mf_fd = -1;
141071d4279SBram Moolenaar     }
142071d4279SBram Moolenaar     else
143071d4279SBram Moolenaar     {
144071d4279SBram Moolenaar 	mf_do_open(mfp, fname, flags);	/* try to open the file */
145071d4279SBram Moolenaar 
146071d4279SBram Moolenaar 	/* if the file cannot be opened, return here */
147071d4279SBram Moolenaar 	if (mfp->mf_fd < 0)
148071d4279SBram Moolenaar 	{
149071d4279SBram Moolenaar 	    vim_free(mfp);
150071d4279SBram Moolenaar 	    return NULL;
151071d4279SBram Moolenaar 	}
152071d4279SBram Moolenaar     }
153071d4279SBram Moolenaar 
154071d4279SBram Moolenaar     mfp->mf_free_first = NULL;		/* free list is empty */
155071d4279SBram Moolenaar     mfp->mf_used_first = NULL;		/* used list is empty */
156071d4279SBram Moolenaar     mfp->mf_used_last = NULL;
157071d4279SBram Moolenaar     mfp->mf_dirty = FALSE;
158071d4279SBram Moolenaar     mfp->mf_used_count = 0;
159071d4279SBram Moolenaar     for (i = 0; i < MEMHASHSIZE; ++i)
160071d4279SBram Moolenaar     {
161071d4279SBram Moolenaar 	mfp->mf_hash[i] = NULL;		/* hash lists are empty */
162071d4279SBram Moolenaar 	mfp->mf_trans[i] = NULL;	/* trans lists are empty */
163071d4279SBram Moolenaar     }
164071d4279SBram Moolenaar     mfp->mf_page_size = MEMFILE_PAGE_SIZE;
165a8ffcbbfSBram Moolenaar #ifdef FEAT_CRYPT
166a8ffcbbfSBram Moolenaar     mfp->mf_old_key = NULL;
167a8ffcbbfSBram Moolenaar #endif
168071d4279SBram Moolenaar 
169071d4279SBram Moolenaar #ifdef USE_FSTATFS
170071d4279SBram Moolenaar     /*
171071d4279SBram Moolenaar      * Try to set the page size equal to the block size of the device.
172071d4279SBram Moolenaar      * Speeds up I/O a lot.
173071d4279SBram Moolenaar      * When recovering, the actual block size will be retrieved from block 0
174071d4279SBram Moolenaar      * in ml_recover().  The size used here may be wrong, therefore
175071d4279SBram Moolenaar      * mf_blocknr_max must be rounded up.
176071d4279SBram Moolenaar      */
177071d4279SBram Moolenaar     if (mfp->mf_fd >= 0
178071d4279SBram Moolenaar 	    && fstatfs(mfp->mf_fd, &stf, sizeof(struct statfs), 0) == 0
179071d4279SBram Moolenaar 	    && stf.F_BSIZE >= MIN_SWAP_PAGE_SIZE
180071d4279SBram Moolenaar 	    && stf.F_BSIZE <= MAX_SWAP_PAGE_SIZE)
181071d4279SBram Moolenaar 	mfp->mf_page_size = stf.F_BSIZE;
182071d4279SBram Moolenaar #endif
183071d4279SBram Moolenaar 
184071d4279SBram Moolenaar     if (mfp->mf_fd < 0 || (flags & (O_TRUNC|O_EXCL))
185071d4279SBram Moolenaar 		      || (size = lseek(mfp->mf_fd, (off_t)0L, SEEK_END)) <= 0)
186071d4279SBram Moolenaar 	mfp->mf_blocknr_max = 0;	/* no file or empty file */
187071d4279SBram Moolenaar     else
188071d4279SBram Moolenaar 	mfp->mf_blocknr_max = (blocknr_T)((size + mfp->mf_page_size - 1)
189071d4279SBram Moolenaar 							 / mfp->mf_page_size);
190071d4279SBram Moolenaar     mfp->mf_blocknr_min = -1;
191071d4279SBram Moolenaar     mfp->mf_neg_count = 0;
192071d4279SBram Moolenaar     mfp->mf_infile_count = mfp->mf_blocknr_max;
193fe265ff5SBram Moolenaar 
194fe265ff5SBram Moolenaar     /*
195fe265ff5SBram Moolenaar      * Compute maximum number of pages ('maxmem' is in Kbyte):
196fe265ff5SBram Moolenaar      *	'mammem' * 1Kbyte / page-size-in-bytes.
197fe265ff5SBram Moolenaar      * Avoid overflow by first reducing page size as much as possible.
198fe265ff5SBram Moolenaar      */
199fe265ff5SBram Moolenaar     {
200fe265ff5SBram Moolenaar 	int	    shift = 10;
201fe265ff5SBram Moolenaar 	unsigned    page_size = mfp->mf_page_size;
202fe265ff5SBram Moolenaar 
203fe265ff5SBram Moolenaar 	while (shift > 0 && (page_size & 1) == 0)
204fe265ff5SBram Moolenaar 	{
205fe265ff5SBram Moolenaar 	    page_size = page_size >> 1;
206fe265ff5SBram Moolenaar 	    --shift;
207fe265ff5SBram Moolenaar 	}
208fe265ff5SBram Moolenaar 	mfp->mf_used_count_max = (p_mm << shift) / page_size;
209fe265ff5SBram Moolenaar 	if (mfp->mf_used_count_max < 10)
210fe265ff5SBram Moolenaar 	    mfp->mf_used_count_max = 10;
211fe265ff5SBram Moolenaar     }
212071d4279SBram Moolenaar 
213071d4279SBram Moolenaar     return mfp;
214071d4279SBram Moolenaar }
215071d4279SBram Moolenaar 
216071d4279SBram Moolenaar /*
217071d4279SBram Moolenaar  * Open a file for an existing memfile.  Used when updatecount set from 0 to
218071d4279SBram Moolenaar  * some value.
219071d4279SBram Moolenaar  * If the file already exists, this fails.
220071d4279SBram Moolenaar  * "fname" is the name of file to use (NULL means no file at all)
221071d4279SBram Moolenaar  * Note: "fname" must have been allocated, it is not copied!  If opening the
222071d4279SBram Moolenaar  * file fails, "fname" is freed.
223071d4279SBram Moolenaar  *
224071d4279SBram Moolenaar  * return value: FAIL if file could not be opened, OK otherwise
225071d4279SBram Moolenaar  */
226071d4279SBram Moolenaar     int
227071d4279SBram Moolenaar mf_open_file(mfp, fname)
228071d4279SBram Moolenaar     memfile_T	*mfp;
229071d4279SBram Moolenaar     char_u	*fname;
230071d4279SBram Moolenaar {
231071d4279SBram Moolenaar     mf_do_open(mfp, fname, O_RDWR|O_CREAT|O_EXCL); /* try to open the file */
232071d4279SBram Moolenaar 
233071d4279SBram Moolenaar     if (mfp->mf_fd < 0)
234071d4279SBram Moolenaar 	return FAIL;
235071d4279SBram Moolenaar 
236071d4279SBram Moolenaar     mfp->mf_dirty = TRUE;
237071d4279SBram Moolenaar     return OK;
238071d4279SBram Moolenaar }
239071d4279SBram Moolenaar 
240071d4279SBram Moolenaar /*
241945e2dbbSBram Moolenaar  * Close a memory file and delete the associated file if 'del_file' is TRUE.
242071d4279SBram Moolenaar  */
243071d4279SBram Moolenaar     void
244071d4279SBram Moolenaar mf_close(mfp, del_file)
245071d4279SBram Moolenaar     memfile_T	*mfp;
246071d4279SBram Moolenaar     int		del_file;
247071d4279SBram Moolenaar {
248071d4279SBram Moolenaar     bhdr_T	*hp, *nextp;
249071d4279SBram Moolenaar     NR_TRANS	*tp, *tpnext;
250071d4279SBram Moolenaar     int		i;
251071d4279SBram Moolenaar 
252071d4279SBram Moolenaar     if (mfp == NULL)		    /* safety check */
253071d4279SBram Moolenaar 	return;
254071d4279SBram Moolenaar     if (mfp->mf_fd >= 0)
255071d4279SBram Moolenaar     {
256071d4279SBram Moolenaar 	if (close(mfp->mf_fd) < 0)
257071d4279SBram Moolenaar 	    EMSG(_(e_swapclose));
258071d4279SBram Moolenaar     }
259071d4279SBram Moolenaar     if (del_file && mfp->mf_fname != NULL)
260071d4279SBram Moolenaar 	mch_remove(mfp->mf_fname);
261071d4279SBram Moolenaar 					    /* free entries in used list */
262071d4279SBram Moolenaar     for (hp = mfp->mf_used_first; hp != NULL; hp = nextp)
263071d4279SBram Moolenaar     {
264071d4279SBram Moolenaar 	total_mem_used -= hp->bh_page_count * mfp->mf_page_size;
265071d4279SBram Moolenaar 	nextp = hp->bh_next;
266071d4279SBram Moolenaar 	mf_free_bhdr(hp);
267071d4279SBram Moolenaar     }
268071d4279SBram Moolenaar     while (mfp->mf_free_first != NULL)	    /* free entries in free list */
269071d4279SBram Moolenaar 	vim_free(mf_rem_free(mfp));
270071d4279SBram Moolenaar     for (i = 0; i < MEMHASHSIZE; ++i)	    /* free entries in trans lists */
271071d4279SBram Moolenaar 	for (tp = mfp->mf_trans[i]; tp != NULL; tp = tpnext)
272071d4279SBram Moolenaar 	{
273071d4279SBram Moolenaar 	    tpnext = tp->nt_next;
274071d4279SBram Moolenaar 	    vim_free(tp);
275071d4279SBram Moolenaar 	}
276071d4279SBram Moolenaar     vim_free(mfp->mf_fname);
277071d4279SBram Moolenaar     vim_free(mfp->mf_ffname);
278071d4279SBram Moolenaar     vim_free(mfp);
279071d4279SBram Moolenaar }
280071d4279SBram Moolenaar 
281071d4279SBram Moolenaar /*
282071d4279SBram Moolenaar  * Close the swap file for a memfile.  Used when 'swapfile' is reset.
283071d4279SBram Moolenaar  */
284071d4279SBram Moolenaar     void
285071d4279SBram Moolenaar mf_close_file(buf, getlines)
286071d4279SBram Moolenaar     buf_T	*buf;
287071d4279SBram Moolenaar     int		getlines;	/* get all lines into memory? */
288071d4279SBram Moolenaar {
289071d4279SBram Moolenaar     memfile_T	*mfp;
290071d4279SBram Moolenaar     linenr_T	lnum;
291071d4279SBram Moolenaar 
292071d4279SBram Moolenaar     mfp = buf->b_ml.ml_mfp;
293071d4279SBram Moolenaar     if (mfp == NULL || mfp->mf_fd < 0)		/* nothing to close */
294071d4279SBram Moolenaar 	return;
295071d4279SBram Moolenaar 
296071d4279SBram Moolenaar     if (getlines)
297071d4279SBram Moolenaar     {
298071d4279SBram Moolenaar 	/* get all blocks in memory by accessing all lines (clumsy!) */
29947b8b15bSBram Moolenaar 	mf_dont_release = TRUE;
300071d4279SBram Moolenaar 	for (lnum = 1; lnum <= buf->b_ml.ml_line_count; ++lnum)
301071d4279SBram Moolenaar 	    (void)ml_get_buf(buf, lnum, FALSE);
30247b8b15bSBram Moolenaar 	mf_dont_release = FALSE;
303071d4279SBram Moolenaar 	/* TODO: should check if all blocks are really in core */
304071d4279SBram Moolenaar     }
305071d4279SBram Moolenaar 
306071d4279SBram Moolenaar     if (close(mfp->mf_fd) < 0)			/* close the file */
307071d4279SBram Moolenaar 	EMSG(_(e_swapclose));
308071d4279SBram Moolenaar     mfp->mf_fd = -1;
309071d4279SBram Moolenaar 
310071d4279SBram Moolenaar     if (mfp->mf_fname != NULL)
311071d4279SBram Moolenaar     {
312071d4279SBram Moolenaar 	mch_remove(mfp->mf_fname);		/* delete the swap file */
313071d4279SBram Moolenaar 	vim_free(mfp->mf_fname);
314071d4279SBram Moolenaar 	vim_free(mfp->mf_ffname);
315071d4279SBram Moolenaar 	mfp->mf_fname = NULL;
316071d4279SBram Moolenaar 	mfp->mf_ffname = NULL;
317071d4279SBram Moolenaar     }
318071d4279SBram Moolenaar }
319071d4279SBram Moolenaar 
320071d4279SBram Moolenaar /*
321071d4279SBram Moolenaar  * Set new size for a memfile.  Used when block 0 of a swapfile has been read
322071d4279SBram Moolenaar  * and the size it indicates differs from what was guessed.
323071d4279SBram Moolenaar  */
324071d4279SBram Moolenaar     void
325071d4279SBram Moolenaar mf_new_page_size(mfp, new_size)
326071d4279SBram Moolenaar     memfile_T	*mfp;
327071d4279SBram Moolenaar     unsigned	new_size;
328071d4279SBram Moolenaar {
329071d4279SBram Moolenaar     /* Correct the memory used for block 0 to the new size, because it will be
330071d4279SBram Moolenaar      * freed with that size later on. */
331071d4279SBram Moolenaar     total_mem_used += new_size - mfp->mf_page_size;
332071d4279SBram Moolenaar     mfp->mf_page_size = new_size;
333071d4279SBram Moolenaar }
334071d4279SBram Moolenaar 
335071d4279SBram Moolenaar /*
336071d4279SBram Moolenaar  * get a new block
337071d4279SBram Moolenaar  *
338071d4279SBram Moolenaar  *   negative: TRUE if negative block number desired (data block)
339071d4279SBram Moolenaar  */
340071d4279SBram Moolenaar     bhdr_T *
341071d4279SBram Moolenaar mf_new(mfp, negative, page_count)
342071d4279SBram Moolenaar     memfile_T	*mfp;
343071d4279SBram Moolenaar     int		negative;
344071d4279SBram Moolenaar     int		page_count;
345071d4279SBram Moolenaar {
346071d4279SBram Moolenaar     bhdr_T	*hp;	/* new bhdr_T */
347071d4279SBram Moolenaar     bhdr_T	*freep;	/* first block in free list */
348071d4279SBram Moolenaar     char_u	*p;
349071d4279SBram Moolenaar 
350071d4279SBram Moolenaar     /*
351071d4279SBram Moolenaar      * If we reached the maximum size for the used memory blocks, release one
352071d4279SBram Moolenaar      * If a bhdr_T is returned, use it and adjust the page_count if necessary.
353071d4279SBram Moolenaar      */
354071d4279SBram Moolenaar     hp = mf_release(mfp, page_count);
355071d4279SBram Moolenaar 
356071d4279SBram Moolenaar /*
357071d4279SBram Moolenaar  * Decide on the number to use:
358071d4279SBram Moolenaar  * If there is a free block, use its number.
359071d4279SBram Moolenaar  * Otherwise use mf_block_min for a negative number, mf_block_max for
360071d4279SBram Moolenaar  * a positive number.
361071d4279SBram Moolenaar  */
362071d4279SBram Moolenaar     freep = mfp->mf_free_first;
363071d4279SBram Moolenaar     if (!negative && freep != NULL && freep->bh_page_count >= page_count)
364071d4279SBram Moolenaar     {
365071d4279SBram Moolenaar 	/*
366071d4279SBram Moolenaar 	 * If the block in the free list has more pages, take only the number
367071d4279SBram Moolenaar 	 * of pages needed and allocate a new bhdr_T with data
368071d4279SBram Moolenaar 	 *
36917e79197SBram Moolenaar 	 * If the number of pages matches and mf_release() did not return a
37017e79197SBram Moolenaar 	 * bhdr_T, use the bhdr_T from the free list and allocate the data
371071d4279SBram Moolenaar 	 *
37217e79197SBram Moolenaar 	 * If the number of pages matches and mf_release() returned a bhdr_T,
373071d4279SBram Moolenaar 	 * just use the number and free the bhdr_T from the free list
374071d4279SBram Moolenaar 	 */
375071d4279SBram Moolenaar 	if (freep->bh_page_count > page_count)
376071d4279SBram Moolenaar 	{
377071d4279SBram Moolenaar 	    if (hp == NULL && (hp = mf_alloc_bhdr(mfp, page_count)) == NULL)
378071d4279SBram Moolenaar 		return NULL;
379071d4279SBram Moolenaar 	    hp->bh_bnum = freep->bh_bnum;
380071d4279SBram Moolenaar 	    freep->bh_bnum += page_count;
381071d4279SBram Moolenaar 	    freep->bh_page_count -= page_count;
382071d4279SBram Moolenaar 	}
383071d4279SBram Moolenaar 	else if (hp == NULL)	    /* need to allocate memory for this block */
384071d4279SBram Moolenaar 	{
385071d4279SBram Moolenaar 	    if ((p = (char_u *)alloc(mfp->mf_page_size * page_count)) == NULL)
386071d4279SBram Moolenaar 		return NULL;
387071d4279SBram Moolenaar 	    hp = mf_rem_free(mfp);
388071d4279SBram Moolenaar 	    hp->bh_data = p;
389071d4279SBram Moolenaar 	}
390071d4279SBram Moolenaar 	else		    /* use the number, remove entry from free list */
391071d4279SBram Moolenaar 	{
392071d4279SBram Moolenaar 	    freep = mf_rem_free(mfp);
393071d4279SBram Moolenaar 	    hp->bh_bnum = freep->bh_bnum;
394071d4279SBram Moolenaar 	    vim_free(freep);
395071d4279SBram Moolenaar 	}
396071d4279SBram Moolenaar     }
397071d4279SBram Moolenaar     else	/* get a new number */
398071d4279SBram Moolenaar     {
399071d4279SBram Moolenaar 	if (hp == NULL && (hp = mf_alloc_bhdr(mfp, page_count)) == NULL)
400071d4279SBram Moolenaar 	    return NULL;
401071d4279SBram Moolenaar 	if (negative)
402071d4279SBram Moolenaar 	{
403071d4279SBram Moolenaar 	    hp->bh_bnum = mfp->mf_blocknr_min--;
404071d4279SBram Moolenaar 	    mfp->mf_neg_count++;
405071d4279SBram Moolenaar 	}
406071d4279SBram Moolenaar 	else
407071d4279SBram Moolenaar 	{
408071d4279SBram Moolenaar 	    hp->bh_bnum = mfp->mf_blocknr_max;
409071d4279SBram Moolenaar 	    mfp->mf_blocknr_max += page_count;
410071d4279SBram Moolenaar 	}
411071d4279SBram Moolenaar     }
412071d4279SBram Moolenaar     hp->bh_flags = BH_LOCKED | BH_DIRTY;	/* new block is always dirty */
413071d4279SBram Moolenaar     mfp->mf_dirty = TRUE;
414071d4279SBram Moolenaar     hp->bh_page_count = page_count;
415071d4279SBram Moolenaar     mf_ins_used(mfp, hp);
416071d4279SBram Moolenaar     mf_ins_hash(mfp, hp);
417071d4279SBram Moolenaar 
418071d4279SBram Moolenaar     /*
419071d4279SBram Moolenaar      * Init the data to all zero, to avoid reading uninitialized data.
420071d4279SBram Moolenaar      * This also avoids that the passwd file ends up in the swap file!
421071d4279SBram Moolenaar      */
422945e2dbbSBram Moolenaar     (void)vim_memset((char *)(hp->bh_data), 0,
423945e2dbbSBram Moolenaar 				      (size_t)mfp->mf_page_size * page_count);
424071d4279SBram Moolenaar 
425071d4279SBram Moolenaar     return hp;
426071d4279SBram Moolenaar }
427071d4279SBram Moolenaar 
428071d4279SBram Moolenaar /*
429a8ffcbbfSBram Moolenaar  * Get existing block "nr" with "page_count" pages.
430071d4279SBram Moolenaar  *
431071d4279SBram Moolenaar  * Note: The caller should first check a negative nr with mf_trans_del()
432071d4279SBram Moolenaar  */
433071d4279SBram Moolenaar     bhdr_T *
434071d4279SBram Moolenaar mf_get(mfp, nr, page_count)
435071d4279SBram Moolenaar     memfile_T	*mfp;
436071d4279SBram Moolenaar     blocknr_T	nr;
437071d4279SBram Moolenaar     int		page_count;
438071d4279SBram Moolenaar {
439071d4279SBram Moolenaar     bhdr_T    *hp;
440071d4279SBram Moolenaar 						/* doesn't exist */
441071d4279SBram Moolenaar     if (nr >= mfp->mf_blocknr_max || nr <= mfp->mf_blocknr_min)
442071d4279SBram Moolenaar 	return NULL;
443071d4279SBram Moolenaar 
444071d4279SBram Moolenaar     /*
445071d4279SBram Moolenaar      * see if it is in the cache
446071d4279SBram Moolenaar      */
447071d4279SBram Moolenaar     hp = mf_find_hash(mfp, nr);
448071d4279SBram Moolenaar     if (hp == NULL)	/* not in the hash list */
449071d4279SBram Moolenaar     {
450071d4279SBram Moolenaar 	if (nr < 0 || nr >= mfp->mf_infile_count)   /* can't be in the file */
451071d4279SBram Moolenaar 	    return NULL;
452071d4279SBram Moolenaar 
453071d4279SBram Moolenaar 	/* could check here if the block is in the free list */
454071d4279SBram Moolenaar 
455071d4279SBram Moolenaar 	/*
456071d4279SBram Moolenaar 	 * Check if we need to flush an existing block.
457071d4279SBram Moolenaar 	 * If so, use that block.
458071d4279SBram Moolenaar 	 * If not, allocate a new block.
459071d4279SBram Moolenaar 	 */
460071d4279SBram Moolenaar 	hp = mf_release(mfp, page_count);
461071d4279SBram Moolenaar 	if (hp == NULL && (hp = mf_alloc_bhdr(mfp, page_count)) == NULL)
462071d4279SBram Moolenaar 	    return NULL;
463071d4279SBram Moolenaar 
464071d4279SBram Moolenaar 	hp->bh_bnum = nr;
465071d4279SBram Moolenaar 	hp->bh_flags = 0;
466071d4279SBram Moolenaar 	hp->bh_page_count = page_count;
467071d4279SBram Moolenaar 	if (mf_read(mfp, hp) == FAIL)	    /* cannot read the block! */
468071d4279SBram Moolenaar 	{
469071d4279SBram Moolenaar 	    mf_free_bhdr(hp);
470071d4279SBram Moolenaar 	    return NULL;
471071d4279SBram Moolenaar 	}
472071d4279SBram Moolenaar     }
473071d4279SBram Moolenaar     else
474071d4279SBram Moolenaar     {
475071d4279SBram Moolenaar 	mf_rem_used(mfp, hp);	/* remove from list, insert in front below */
476071d4279SBram Moolenaar 	mf_rem_hash(mfp, hp);
477071d4279SBram Moolenaar     }
478071d4279SBram Moolenaar 
479071d4279SBram Moolenaar     hp->bh_flags |= BH_LOCKED;
480071d4279SBram Moolenaar     mf_ins_used(mfp, hp);	/* put in front of used list */
481071d4279SBram Moolenaar     mf_ins_hash(mfp, hp);	/* put in front of hash list */
482071d4279SBram Moolenaar 
483071d4279SBram Moolenaar     return hp;
484071d4279SBram Moolenaar }
485071d4279SBram Moolenaar 
486071d4279SBram Moolenaar /*
487071d4279SBram Moolenaar  * release the block *hp
488071d4279SBram Moolenaar  *
489071d4279SBram Moolenaar  *   dirty: Block must be written to file later
490071d4279SBram Moolenaar  *   infile: Block should be in file (needed for recovery)
491071d4279SBram Moolenaar  *
492071d4279SBram Moolenaar  *  no return value, function cannot fail
493071d4279SBram Moolenaar  */
494071d4279SBram Moolenaar     void
495071d4279SBram Moolenaar mf_put(mfp, hp, dirty, infile)
496071d4279SBram Moolenaar     memfile_T	*mfp;
497071d4279SBram Moolenaar     bhdr_T	*hp;
498071d4279SBram Moolenaar     int		dirty;
499071d4279SBram Moolenaar     int		infile;
500071d4279SBram Moolenaar {
501071d4279SBram Moolenaar     int		flags;
502071d4279SBram Moolenaar 
503071d4279SBram Moolenaar     flags = hp->bh_flags;
504071d4279SBram Moolenaar 
505071d4279SBram Moolenaar     if ((flags & BH_LOCKED) == 0)
506071d4279SBram Moolenaar 	EMSG(_("E293: block was not locked"));
507071d4279SBram Moolenaar     flags &= ~BH_LOCKED;
508071d4279SBram Moolenaar     if (dirty)
509071d4279SBram Moolenaar     {
510071d4279SBram Moolenaar 	flags |= BH_DIRTY;
511071d4279SBram Moolenaar 	mfp->mf_dirty = TRUE;
512071d4279SBram Moolenaar     }
513071d4279SBram Moolenaar     hp->bh_flags = flags;
514071d4279SBram Moolenaar     if (infile)
515071d4279SBram Moolenaar 	mf_trans_add(mfp, hp);	    /* may translate negative in positive nr */
516071d4279SBram Moolenaar }
517071d4279SBram Moolenaar 
518071d4279SBram Moolenaar /*
519071d4279SBram Moolenaar  * block *hp is no longer in used, may put it in the free list of memfile *mfp
520071d4279SBram Moolenaar  */
521071d4279SBram Moolenaar     void
522071d4279SBram Moolenaar mf_free(mfp, hp)
523071d4279SBram Moolenaar     memfile_T	*mfp;
524071d4279SBram Moolenaar     bhdr_T	*hp;
525071d4279SBram Moolenaar {
526071d4279SBram Moolenaar     vim_free(hp->bh_data);	/* free the memory */
527071d4279SBram Moolenaar     mf_rem_hash(mfp, hp);	/* get *hp out of the hash list */
528071d4279SBram Moolenaar     mf_rem_used(mfp, hp);	/* get *hp out of the used list */
529071d4279SBram Moolenaar     if (hp->bh_bnum < 0)
530071d4279SBram Moolenaar     {
531071d4279SBram Moolenaar 	vim_free(hp);		/* don't want negative numbers in free list */
532071d4279SBram Moolenaar 	mfp->mf_neg_count--;
533071d4279SBram Moolenaar     }
534071d4279SBram Moolenaar     else
535071d4279SBram Moolenaar 	mf_ins_free(mfp, hp);	/* put *hp in the free list */
536071d4279SBram Moolenaar }
537071d4279SBram Moolenaar 
53889f3727bSBram Moolenaar #if defined(__MORPHOS__) && defined(__libnix__)
539071d4279SBram Moolenaar /* function is missing in MorphOS libnix version */
540071d4279SBram Moolenaar extern unsigned long *__stdfiledes;
541071d4279SBram Moolenaar 
542071d4279SBram Moolenaar     static unsigned long
543071d4279SBram Moolenaar fdtofh(int filedescriptor)
544071d4279SBram Moolenaar {
545071d4279SBram Moolenaar     return __stdfiledes[filedescriptor];
546071d4279SBram Moolenaar }
547071d4279SBram Moolenaar #endif
548071d4279SBram Moolenaar 
549071d4279SBram Moolenaar /*
550071d4279SBram Moolenaar  * Sync the memory file *mfp to disk.
551071d4279SBram Moolenaar  * Flags:
552071d4279SBram Moolenaar  *  MFS_ALL	If not given, blocks with negative numbers are not synced,
553071d4279SBram Moolenaar  *		even when they are dirty!
554071d4279SBram Moolenaar  *  MFS_STOP	Stop syncing when a character becomes available, but sync at
555071d4279SBram Moolenaar  *		least one block.
556071d4279SBram Moolenaar  *  MFS_FLUSH	Make sure buffers are flushed to disk, so they will survive a
557071d4279SBram Moolenaar  *		system crash.
558071d4279SBram Moolenaar  *  MFS_ZERO	Only write block 0.
559071d4279SBram Moolenaar  *
560071d4279SBram Moolenaar  * Return FAIL for failure, OK otherwise
561071d4279SBram Moolenaar  */
562071d4279SBram Moolenaar     int
563071d4279SBram Moolenaar mf_sync(mfp, flags)
564071d4279SBram Moolenaar     memfile_T	*mfp;
565071d4279SBram Moolenaar     int		flags;
566071d4279SBram Moolenaar {
567071d4279SBram Moolenaar     int		status;
568071d4279SBram Moolenaar     bhdr_T	*hp;
569071d4279SBram Moolenaar #if defined(SYNC_DUP_CLOSE) && !defined(MSDOS)
570071d4279SBram Moolenaar     int		fd;
571071d4279SBram Moolenaar #endif
572071d4279SBram Moolenaar     int		got_int_save = got_int;
573071d4279SBram Moolenaar 
574071d4279SBram Moolenaar     if (mfp->mf_fd < 0)	    /* there is no file, nothing to do */
575071d4279SBram Moolenaar     {
576071d4279SBram Moolenaar 	mfp->mf_dirty = FALSE;
577071d4279SBram Moolenaar 	return FAIL;
578071d4279SBram Moolenaar     }
579071d4279SBram Moolenaar 
580071d4279SBram Moolenaar     /* Only a CTRL-C while writing will break us here, not one typed
581071d4279SBram Moolenaar      * previously. */
582071d4279SBram Moolenaar     got_int = FALSE;
583071d4279SBram Moolenaar 
584071d4279SBram Moolenaar     /*
585071d4279SBram Moolenaar      * sync from last to first (may reduce the probability of an inconsistent
586071d4279SBram Moolenaar      * file) If a write fails, it is very likely caused by a full filesystem.
587071d4279SBram Moolenaar      * Then we only try to write blocks within the existing file. If that also
588071d4279SBram Moolenaar      * fails then we give up.
589071d4279SBram Moolenaar      */
590071d4279SBram Moolenaar     status = OK;
591071d4279SBram Moolenaar     for (hp = mfp->mf_used_last; hp != NULL; hp = hp->bh_prev)
592071d4279SBram Moolenaar 	if (((flags & MFS_ALL) || hp->bh_bnum >= 0)
593071d4279SBram Moolenaar 		&& (hp->bh_flags & BH_DIRTY)
594071d4279SBram Moolenaar 		&& (status == OK || (hp->bh_bnum >= 0
595071d4279SBram Moolenaar 		    && hp->bh_bnum < mfp->mf_infile_count)))
596071d4279SBram Moolenaar 	{
597071d4279SBram Moolenaar 	    if ((flags & MFS_ZERO) && hp->bh_bnum != 0)
598071d4279SBram Moolenaar 		continue;
599071d4279SBram Moolenaar 	    if (mf_write(mfp, hp) == FAIL)
600071d4279SBram Moolenaar 	    {
601071d4279SBram Moolenaar 		if (status == FAIL)	/* double error: quit syncing */
602071d4279SBram Moolenaar 		    break;
603071d4279SBram Moolenaar 		status = FAIL;
604071d4279SBram Moolenaar 	    }
605071d4279SBram Moolenaar 	    if (flags & MFS_STOP)
606071d4279SBram Moolenaar 	    {
607071d4279SBram Moolenaar 		/* Stop when char available now. */
608071d4279SBram Moolenaar 		if (ui_char_avail())
609071d4279SBram Moolenaar 		    break;
610071d4279SBram Moolenaar 	    }
611071d4279SBram Moolenaar 	    else
612071d4279SBram Moolenaar 		ui_breakcheck();
613071d4279SBram Moolenaar 	    if (got_int)
614071d4279SBram Moolenaar 		break;
615071d4279SBram Moolenaar 	}
616071d4279SBram Moolenaar 
617071d4279SBram Moolenaar     /*
618071d4279SBram Moolenaar      * If the whole list is flushed, the memfile is not dirty anymore.
619071d4279SBram Moolenaar      * In case of an error this flag is also set, to avoid trying all the time.
620071d4279SBram Moolenaar      */
621071d4279SBram Moolenaar     if (hp == NULL || status == FAIL)
622071d4279SBram Moolenaar 	mfp->mf_dirty = FALSE;
623071d4279SBram Moolenaar 
624071d4279SBram Moolenaar     if ((flags & MFS_FLUSH) && *p_sws != NUL)
625071d4279SBram Moolenaar     {
626071d4279SBram Moolenaar #if defined(UNIX)
627071d4279SBram Moolenaar # ifdef HAVE_FSYNC
628071d4279SBram Moolenaar 	/*
629071d4279SBram Moolenaar 	 * most Unixes have the very useful fsync() function, just what we need.
630071d4279SBram Moolenaar 	 * However, with OS/2 and EMX it is also available, but there are
631071d4279SBram Moolenaar 	 * reports of bad problems with it (a bug in HPFS.IFS).
632071d4279SBram Moolenaar 	 * So we disable use of it here in case someone tries to be smart
633071d4279SBram Moolenaar 	 * and changes os_os2_cfg.h... (even though there is no __EMX__ test
634071d4279SBram Moolenaar 	 * in the #if, as __EMX__ does not have sync(); we hope for a timely
635071d4279SBram Moolenaar 	 * sync from the system itself).
636071d4279SBram Moolenaar 	 */
637071d4279SBram Moolenaar #  if defined(__EMX__)
638071d4279SBram Moolenaar    error "Dont use fsync with EMX! Read emxdoc.doc or emxfix01.doc for info."
639071d4279SBram Moolenaar #  endif
640071d4279SBram Moolenaar 	if (STRCMP(p_sws, "fsync") == 0)
641071d4279SBram Moolenaar 	{
642071d4279SBram Moolenaar 	    if (fsync(mfp->mf_fd))
643071d4279SBram Moolenaar 		status = FAIL;
644071d4279SBram Moolenaar 	}
645071d4279SBram Moolenaar 	else
646071d4279SBram Moolenaar # endif
647071d4279SBram Moolenaar 	    /* OpenNT is strictly POSIX (Benzinger) */
648071d4279SBram Moolenaar 	    /* Tandem/Himalaya NSK-OSS doesn't have sync() */
649071d4279SBram Moolenaar # if defined(__OPENNT) || defined(__TANDEM)
650071d4279SBram Moolenaar 	    fflush(NULL);
651071d4279SBram Moolenaar # else
652071d4279SBram Moolenaar 	    sync();
653071d4279SBram Moolenaar # endif
654071d4279SBram Moolenaar #endif
655071d4279SBram Moolenaar #ifdef VMS
656071d4279SBram Moolenaar 	if (STRCMP(p_sws, "fsync") == 0)
657071d4279SBram Moolenaar 	{
658071d4279SBram Moolenaar 	    if (fsync(mfp->mf_fd))
659071d4279SBram Moolenaar 		status = FAIL;
660071d4279SBram Moolenaar 	}
661071d4279SBram Moolenaar #endif
662071d4279SBram Moolenaar #ifdef MSDOS
663071d4279SBram Moolenaar 	if (_dos_commit(mfp->mf_fd))
664071d4279SBram Moolenaar 	    status = FAIL;
665071d4279SBram Moolenaar #else
666071d4279SBram Moolenaar # ifdef SYNC_DUP_CLOSE
667071d4279SBram Moolenaar 	/*
668071d4279SBram Moolenaar 	 * Win32 is a bit more work: Duplicate the file handle and close it.
669071d4279SBram Moolenaar 	 * This should flush the file to disk.
670071d4279SBram Moolenaar 	 */
671071d4279SBram Moolenaar 	if ((fd = dup(mfp->mf_fd)) >= 0)
672071d4279SBram Moolenaar 	    close(fd);
673071d4279SBram Moolenaar # endif
674071d4279SBram Moolenaar #endif
675071d4279SBram Moolenaar #ifdef AMIGA
6765a6404cfSBram Moolenaar # if defined(__AROS__) || defined(__amigaos4__)
677071d4279SBram Moolenaar 	if (fsync(mfp->mf_fd) != 0)
678071d4279SBram Moolenaar 	    status = FAIL;
679071d4279SBram Moolenaar # else
680071d4279SBram Moolenaar 	/*
681071d4279SBram Moolenaar 	 * Flush() only exists for AmigaDos 2.0.
682071d4279SBram Moolenaar 	 * For 1.3 it should be done with close() + open(), but then the risk
683071d4279SBram Moolenaar 	 * is that the open() may fail and lose the file....
684071d4279SBram Moolenaar 	 */
685071d4279SBram Moolenaar #  ifdef FEAT_ARP
686071d4279SBram Moolenaar 	if (dos2)
687071d4279SBram Moolenaar #  endif
688071d4279SBram Moolenaar #  ifdef SASC
689071d4279SBram Moolenaar 	{
690071d4279SBram Moolenaar 	    struct UFB *fp = chkufb(mfp->mf_fd);
691071d4279SBram Moolenaar 
692071d4279SBram Moolenaar 	    if (fp != NULL)
693071d4279SBram Moolenaar 		Flush(fp->ufbfh);
694071d4279SBram Moolenaar 	}
695071d4279SBram Moolenaar #  else
696071d4279SBram Moolenaar #   if defined(_DCC) || defined(__GNUC__) || defined(__MORPHOS__)
697071d4279SBram Moolenaar 	{
69889f3727bSBram Moolenaar #    if defined(__GNUC__) && !defined(__MORPHOS__) && defined(__libnix__)
699071d4279SBram Moolenaar 	    /* Have function (in libnix at least),
700071d4279SBram Moolenaar 	     * but ain't got no prototype anywhere. */
701071d4279SBram Moolenaar 	    extern unsigned long fdtofh(int filedescriptor);
702071d4279SBram Moolenaar #    endif
70389f3727bSBram Moolenaar #    if !defined(__libnix__)
70489f3727bSBram Moolenaar 	    fflush(NULL);
70589f3727bSBram Moolenaar #    else
706071d4279SBram Moolenaar 	    BPTR fh = (BPTR)fdtofh(mfp->mf_fd);
707071d4279SBram Moolenaar 
708071d4279SBram Moolenaar 	    if (fh != 0)
709071d4279SBram Moolenaar 		Flush(fh);
71089f3727bSBram Moolenaar #    endif
711071d4279SBram Moolenaar 	}
712071d4279SBram Moolenaar #   else /* assume Manx */
713071d4279SBram Moolenaar 	    Flush(_devtab[mfp->mf_fd].fd);
714071d4279SBram Moolenaar #   endif
715071d4279SBram Moolenaar #  endif
716071d4279SBram Moolenaar # endif
717071d4279SBram Moolenaar #endif /* AMIGA */
718071d4279SBram Moolenaar     }
719071d4279SBram Moolenaar 
720071d4279SBram Moolenaar     got_int |= got_int_save;
721071d4279SBram Moolenaar 
722071d4279SBram Moolenaar     return status;
723071d4279SBram Moolenaar }
724071d4279SBram Moolenaar 
725071d4279SBram Moolenaar /*
726c32840f2SBram Moolenaar  * For all blocks in memory file *mfp that have a positive block number set
727c32840f2SBram Moolenaar  * the dirty flag.  These are blocks that need to be written to a newly
728c32840f2SBram Moolenaar  * created swapfile.
729c32840f2SBram Moolenaar  */
730c32840f2SBram Moolenaar     void
731c32840f2SBram Moolenaar mf_set_dirty(mfp)
732c32840f2SBram Moolenaar     memfile_T	*mfp;
733c32840f2SBram Moolenaar {
734c32840f2SBram Moolenaar     bhdr_T	*hp;
735c32840f2SBram Moolenaar 
736c32840f2SBram Moolenaar     for (hp = mfp->mf_used_last; hp != NULL; hp = hp->bh_prev)
737c32840f2SBram Moolenaar 	if (hp->bh_bnum > 0)
738c32840f2SBram Moolenaar 	    hp->bh_flags |= BH_DIRTY;
739c32840f2SBram Moolenaar     mfp->mf_dirty = TRUE;
740c32840f2SBram Moolenaar }
741c32840f2SBram Moolenaar 
742c32840f2SBram Moolenaar /*
743071d4279SBram Moolenaar  * insert block *hp in front of hashlist of memfile *mfp
744071d4279SBram Moolenaar  */
745071d4279SBram Moolenaar     static void
746071d4279SBram Moolenaar mf_ins_hash(mfp, hp)
747071d4279SBram Moolenaar     memfile_T	*mfp;
748071d4279SBram Moolenaar     bhdr_T	*hp;
749071d4279SBram Moolenaar {
750071d4279SBram Moolenaar     bhdr_T	*hhp;
751071d4279SBram Moolenaar     int		hash;
752071d4279SBram Moolenaar 
753071d4279SBram Moolenaar     hash = MEMHASH(hp->bh_bnum);
754071d4279SBram Moolenaar     hhp = mfp->mf_hash[hash];
755071d4279SBram Moolenaar     hp->bh_hash_next = hhp;
756071d4279SBram Moolenaar     hp->bh_hash_prev = NULL;
757071d4279SBram Moolenaar     if (hhp != NULL)
758071d4279SBram Moolenaar 	hhp->bh_hash_prev = hp;
759071d4279SBram Moolenaar     mfp->mf_hash[hash] = hp;
760071d4279SBram Moolenaar }
761071d4279SBram Moolenaar 
762071d4279SBram Moolenaar /*
763071d4279SBram Moolenaar  * remove block *hp from hashlist of memfile list *mfp
764071d4279SBram Moolenaar  */
765071d4279SBram Moolenaar     static void
766071d4279SBram Moolenaar mf_rem_hash(mfp, hp)
767071d4279SBram Moolenaar     memfile_T	*mfp;
768071d4279SBram Moolenaar     bhdr_T	*hp;
769071d4279SBram Moolenaar {
770071d4279SBram Moolenaar     if (hp->bh_hash_prev == NULL)
771071d4279SBram Moolenaar 	mfp->mf_hash[MEMHASH(hp->bh_bnum)] = hp->bh_hash_next;
772071d4279SBram Moolenaar     else
773071d4279SBram Moolenaar 	hp->bh_hash_prev->bh_hash_next = hp->bh_hash_next;
774071d4279SBram Moolenaar 
775071d4279SBram Moolenaar     if (hp->bh_hash_next)
776071d4279SBram Moolenaar 	hp->bh_hash_next->bh_hash_prev = hp->bh_hash_prev;
777071d4279SBram Moolenaar }
778071d4279SBram Moolenaar 
779071d4279SBram Moolenaar /*
780071d4279SBram Moolenaar  * look in hash lists of memfile *mfp for block header with number 'nr'
781071d4279SBram Moolenaar  */
782071d4279SBram Moolenaar     static bhdr_T *
783071d4279SBram Moolenaar mf_find_hash(mfp, nr)
784071d4279SBram Moolenaar     memfile_T	*mfp;
785071d4279SBram Moolenaar     blocknr_T	nr;
786071d4279SBram Moolenaar {
787071d4279SBram Moolenaar     bhdr_T	*hp;
788071d4279SBram Moolenaar 
789071d4279SBram Moolenaar     for (hp = mfp->mf_hash[MEMHASH(nr)]; hp != NULL; hp = hp->bh_hash_next)
790071d4279SBram Moolenaar 	if (hp->bh_bnum == nr)
791071d4279SBram Moolenaar 	    break;
792071d4279SBram Moolenaar     return hp;
793071d4279SBram Moolenaar }
794071d4279SBram Moolenaar 
795071d4279SBram Moolenaar /*
796071d4279SBram Moolenaar  * insert block *hp in front of used list of memfile *mfp
797071d4279SBram Moolenaar  */
798071d4279SBram Moolenaar     static void
799071d4279SBram Moolenaar mf_ins_used(mfp, hp)
800071d4279SBram Moolenaar     memfile_T	*mfp;
801071d4279SBram Moolenaar     bhdr_T	*hp;
802071d4279SBram Moolenaar {
803071d4279SBram Moolenaar     hp->bh_next = mfp->mf_used_first;
804071d4279SBram Moolenaar     mfp->mf_used_first = hp;
805071d4279SBram Moolenaar     hp->bh_prev = NULL;
806071d4279SBram Moolenaar     if (hp->bh_next == NULL)	    /* list was empty, adjust last pointer */
807071d4279SBram Moolenaar 	mfp->mf_used_last = hp;
808071d4279SBram Moolenaar     else
809071d4279SBram Moolenaar 	hp->bh_next->bh_prev = hp;
810071d4279SBram Moolenaar     mfp->mf_used_count += hp->bh_page_count;
811071d4279SBram Moolenaar     total_mem_used += hp->bh_page_count * mfp->mf_page_size;
812071d4279SBram Moolenaar }
813071d4279SBram Moolenaar 
814071d4279SBram Moolenaar /*
815071d4279SBram Moolenaar  * remove block *hp from used list of memfile *mfp
816071d4279SBram Moolenaar  */
817071d4279SBram Moolenaar     static void
818071d4279SBram Moolenaar mf_rem_used(mfp, hp)
819071d4279SBram Moolenaar     memfile_T	*mfp;
820071d4279SBram Moolenaar     bhdr_T	*hp;
821071d4279SBram Moolenaar {
822071d4279SBram Moolenaar     if (hp->bh_next == NULL)	    /* last block in used list */
823071d4279SBram Moolenaar 	mfp->mf_used_last = hp->bh_prev;
824071d4279SBram Moolenaar     else
825071d4279SBram Moolenaar 	hp->bh_next->bh_prev = hp->bh_prev;
826071d4279SBram Moolenaar     if (hp->bh_prev == NULL)	    /* first block in used list */
827071d4279SBram Moolenaar 	mfp->mf_used_first = hp->bh_next;
828071d4279SBram Moolenaar     else
829071d4279SBram Moolenaar 	hp->bh_prev->bh_next = hp->bh_next;
830071d4279SBram Moolenaar     mfp->mf_used_count -= hp->bh_page_count;
831071d4279SBram Moolenaar     total_mem_used -= hp->bh_page_count * mfp->mf_page_size;
832071d4279SBram Moolenaar }
833071d4279SBram Moolenaar 
834071d4279SBram Moolenaar /*
835071d4279SBram Moolenaar  * Release the least recently used block from the used list if the number
836071d4279SBram Moolenaar  * of used memory blocks gets to big.
837071d4279SBram Moolenaar  *
838071d4279SBram Moolenaar  * Return the block header to the caller, including the memory block, so
839071d4279SBram Moolenaar  * it can be re-used. Make sure the page_count is right.
840071d4279SBram Moolenaar  */
841071d4279SBram Moolenaar     static bhdr_T *
842071d4279SBram Moolenaar mf_release(mfp, page_count)
843071d4279SBram Moolenaar     memfile_T	*mfp;
844071d4279SBram Moolenaar     int		page_count;
845071d4279SBram Moolenaar {
846071d4279SBram Moolenaar     bhdr_T	*hp;
847071d4279SBram Moolenaar     int		need_release;
848071d4279SBram Moolenaar     buf_T	*buf;
849071d4279SBram Moolenaar 
850071d4279SBram Moolenaar     /* don't release while in mf_close_file() */
85147b8b15bSBram Moolenaar     if (mf_dont_release)
852071d4279SBram Moolenaar 	return NULL;
853071d4279SBram Moolenaar 
854071d4279SBram Moolenaar     /*
855071d4279SBram Moolenaar      * Need to release a block if the number of blocks for this memfile is
856071d4279SBram Moolenaar      * higher than the maximum or total memory used is over 'maxmemtot'
857071d4279SBram Moolenaar      */
858071d4279SBram Moolenaar     need_release = ((mfp->mf_used_count >= mfp->mf_used_count_max)
859071d4279SBram Moolenaar 				  || (total_mem_used >> 10) >= (long_u)p_mmt);
860071d4279SBram Moolenaar 
861071d4279SBram Moolenaar     /*
862071d4279SBram Moolenaar      * Try to create a swap file if the amount of memory used is getting too
863071d4279SBram Moolenaar      * high.
864071d4279SBram Moolenaar      */
865071d4279SBram Moolenaar     if (mfp->mf_fd < 0 && need_release && p_uc)
866071d4279SBram Moolenaar     {
867071d4279SBram Moolenaar 	/* find for which buffer this memfile is */
868071d4279SBram Moolenaar 	for (buf = firstbuf; buf != NULL; buf = buf->b_next)
869071d4279SBram Moolenaar 	    if (buf->b_ml.ml_mfp == mfp)
870071d4279SBram Moolenaar 		break;
871071d4279SBram Moolenaar 	if (buf != NULL && buf->b_may_swap)
872071d4279SBram Moolenaar 	    ml_open_file(buf);
873071d4279SBram Moolenaar     }
874071d4279SBram Moolenaar 
875071d4279SBram Moolenaar     /*
876071d4279SBram Moolenaar      * don't release a block if
877071d4279SBram Moolenaar      *	there is no file for this memfile
878071d4279SBram Moolenaar      * or
879071d4279SBram Moolenaar      *	the number of blocks for this memfile is lower than the maximum
880071d4279SBram Moolenaar      *	  and
881071d4279SBram Moolenaar      *	total memory used is not up to 'maxmemtot'
882071d4279SBram Moolenaar      */
883071d4279SBram Moolenaar     if (mfp->mf_fd < 0 || !need_release)
884071d4279SBram Moolenaar 	return NULL;
885071d4279SBram Moolenaar 
886071d4279SBram Moolenaar     for (hp = mfp->mf_used_last; hp != NULL; hp = hp->bh_prev)
887071d4279SBram Moolenaar 	if (!(hp->bh_flags & BH_LOCKED))
888071d4279SBram Moolenaar 	    break;
889071d4279SBram Moolenaar     if (hp == NULL)	/* not a single one that can be released */
890071d4279SBram Moolenaar 	return NULL;
891071d4279SBram Moolenaar 
892071d4279SBram Moolenaar     /*
893071d4279SBram Moolenaar      * If the block is dirty, write it.
894071d4279SBram Moolenaar      * If the write fails we don't free it.
895071d4279SBram Moolenaar      */
896071d4279SBram Moolenaar     if ((hp->bh_flags & BH_DIRTY) && mf_write(mfp, hp) == FAIL)
897071d4279SBram Moolenaar 	return NULL;
898071d4279SBram Moolenaar 
899071d4279SBram Moolenaar     mf_rem_used(mfp, hp);
900071d4279SBram Moolenaar     mf_rem_hash(mfp, hp);
901071d4279SBram Moolenaar 
902071d4279SBram Moolenaar     /*
903071d4279SBram Moolenaar      * If a bhdr_T is returned, make sure that the page_count of bh_data is
904071d4279SBram Moolenaar      * right
905071d4279SBram Moolenaar      */
906071d4279SBram Moolenaar     if (hp->bh_page_count != page_count)
907071d4279SBram Moolenaar     {
908071d4279SBram Moolenaar 	vim_free(hp->bh_data);
909071d4279SBram Moolenaar 	if ((hp->bh_data = alloc(mfp->mf_page_size * page_count)) == NULL)
910071d4279SBram Moolenaar 	{
911071d4279SBram Moolenaar 	    vim_free(hp);
912071d4279SBram Moolenaar 	    return NULL;
913071d4279SBram Moolenaar 	}
914071d4279SBram Moolenaar 	hp->bh_page_count = page_count;
915071d4279SBram Moolenaar     }
916071d4279SBram Moolenaar     return hp;
917071d4279SBram Moolenaar }
918071d4279SBram Moolenaar 
919071d4279SBram Moolenaar /*
920071d4279SBram Moolenaar  * release as many blocks as possible
921071d4279SBram Moolenaar  * Used in case of out of memory
922071d4279SBram Moolenaar  *
923071d4279SBram Moolenaar  * return TRUE if any memory was released
924071d4279SBram Moolenaar  */
925071d4279SBram Moolenaar     int
926071d4279SBram Moolenaar mf_release_all()
927071d4279SBram Moolenaar {
928071d4279SBram Moolenaar     buf_T	*buf;
929071d4279SBram Moolenaar     memfile_T	*mfp;
930071d4279SBram Moolenaar     bhdr_T	*hp;
931071d4279SBram Moolenaar     int		retval = FALSE;
932071d4279SBram Moolenaar 
933071d4279SBram Moolenaar     for (buf = firstbuf; buf != NULL; buf = buf->b_next)
934071d4279SBram Moolenaar     {
935071d4279SBram Moolenaar 	mfp = buf->b_ml.ml_mfp;
936071d4279SBram Moolenaar 	if (mfp != NULL)
937071d4279SBram Moolenaar 	{
938071d4279SBram Moolenaar 	    /* If no swap file yet, may open one */
939071d4279SBram Moolenaar 	    if (mfp->mf_fd < 0 && buf->b_may_swap)
940071d4279SBram Moolenaar 		ml_open_file(buf);
941071d4279SBram Moolenaar 
942071d4279SBram Moolenaar 	    /* only if there is a swapfile */
943071d4279SBram Moolenaar 	    if (mfp->mf_fd >= 0)
944071d4279SBram Moolenaar 	    {
945071d4279SBram Moolenaar 		for (hp = mfp->mf_used_last; hp != NULL; )
946071d4279SBram Moolenaar 		{
947071d4279SBram Moolenaar 		    if (!(hp->bh_flags & BH_LOCKED)
948071d4279SBram Moolenaar 			    && (!(hp->bh_flags & BH_DIRTY)
949071d4279SBram Moolenaar 				|| mf_write(mfp, hp) != FAIL))
950071d4279SBram Moolenaar 		    {
951071d4279SBram Moolenaar 			mf_rem_used(mfp, hp);
952071d4279SBram Moolenaar 			mf_rem_hash(mfp, hp);
953071d4279SBram Moolenaar 			mf_free_bhdr(hp);
954071d4279SBram Moolenaar 			hp = mfp->mf_used_last;	/* re-start, list was changed */
955071d4279SBram Moolenaar 			retval = TRUE;
956071d4279SBram Moolenaar 		    }
957071d4279SBram Moolenaar 		    else
958071d4279SBram Moolenaar 			hp = hp->bh_prev;
959071d4279SBram Moolenaar 		}
960071d4279SBram Moolenaar 	    }
961071d4279SBram Moolenaar 	}
962071d4279SBram Moolenaar     }
963071d4279SBram Moolenaar     return retval;
964071d4279SBram Moolenaar }
965071d4279SBram Moolenaar 
966071d4279SBram Moolenaar /*
967071d4279SBram Moolenaar  * Allocate a block header and a block of memory for it
968071d4279SBram Moolenaar  */
969071d4279SBram Moolenaar     static bhdr_T *
970071d4279SBram Moolenaar mf_alloc_bhdr(mfp, page_count)
971071d4279SBram Moolenaar     memfile_T	*mfp;
972071d4279SBram Moolenaar     int		page_count;
973071d4279SBram Moolenaar {
974071d4279SBram Moolenaar     bhdr_T	*hp;
975071d4279SBram Moolenaar 
976071d4279SBram Moolenaar     if ((hp = (bhdr_T *)alloc((unsigned)sizeof(bhdr_T))) != NULL)
977071d4279SBram Moolenaar     {
978071d4279SBram Moolenaar 	if ((hp->bh_data = (char_u *)alloc(mfp->mf_page_size * page_count))
979071d4279SBram Moolenaar 								      == NULL)
980071d4279SBram Moolenaar 	{
981071d4279SBram Moolenaar 	    vim_free(hp);	    /* not enough memory */
982071d4279SBram Moolenaar 	    return NULL;
983071d4279SBram Moolenaar 	}
984071d4279SBram Moolenaar 	hp->bh_page_count = page_count;
985071d4279SBram Moolenaar     }
986071d4279SBram Moolenaar     return hp;
987071d4279SBram Moolenaar }
988071d4279SBram Moolenaar 
989071d4279SBram Moolenaar /*
990071d4279SBram Moolenaar  * Free a block header and the block of memory for it
991071d4279SBram Moolenaar  */
992071d4279SBram Moolenaar     static void
993071d4279SBram Moolenaar mf_free_bhdr(hp)
994071d4279SBram Moolenaar     bhdr_T	*hp;
995071d4279SBram Moolenaar {
996071d4279SBram Moolenaar     vim_free(hp->bh_data);
997071d4279SBram Moolenaar     vim_free(hp);
998071d4279SBram Moolenaar }
999071d4279SBram Moolenaar 
1000071d4279SBram Moolenaar /*
1001071d4279SBram Moolenaar  * insert entry *hp in the free list
1002071d4279SBram Moolenaar  */
1003071d4279SBram Moolenaar     static void
1004071d4279SBram Moolenaar mf_ins_free(mfp, hp)
1005071d4279SBram Moolenaar     memfile_T	*mfp;
1006071d4279SBram Moolenaar     bhdr_T	*hp;
1007071d4279SBram Moolenaar {
1008071d4279SBram Moolenaar     hp->bh_next = mfp->mf_free_first;
1009071d4279SBram Moolenaar     mfp->mf_free_first = hp;
1010071d4279SBram Moolenaar }
1011071d4279SBram Moolenaar 
1012071d4279SBram Moolenaar /*
1013071d4279SBram Moolenaar  * remove the first entry from the free list and return a pointer to it
1014071d4279SBram Moolenaar  * Note: caller must check that mfp->mf_free_first is not NULL!
1015071d4279SBram Moolenaar  */
1016071d4279SBram Moolenaar     static bhdr_T *
1017071d4279SBram Moolenaar mf_rem_free(mfp)
1018071d4279SBram Moolenaar     memfile_T	*mfp;
1019071d4279SBram Moolenaar {
1020071d4279SBram Moolenaar     bhdr_T	*hp;
1021071d4279SBram Moolenaar 
1022071d4279SBram Moolenaar     hp = mfp->mf_free_first;
1023071d4279SBram Moolenaar     mfp->mf_free_first = hp->bh_next;
1024071d4279SBram Moolenaar     return hp;
1025071d4279SBram Moolenaar }
1026071d4279SBram Moolenaar 
1027071d4279SBram Moolenaar /*
1028071d4279SBram Moolenaar  * read a block from disk
1029071d4279SBram Moolenaar  *
1030071d4279SBram Moolenaar  * Return FAIL for failure, OK otherwise
1031071d4279SBram Moolenaar  */
1032071d4279SBram Moolenaar     static int
1033071d4279SBram Moolenaar mf_read(mfp, hp)
1034071d4279SBram Moolenaar     memfile_T	*mfp;
1035071d4279SBram Moolenaar     bhdr_T	*hp;
1036071d4279SBram Moolenaar {
1037071d4279SBram Moolenaar     off_t	offset;
1038071d4279SBram Moolenaar     unsigned	page_size;
1039071d4279SBram Moolenaar     unsigned	size;
1040071d4279SBram Moolenaar 
1041071d4279SBram Moolenaar     if (mfp->mf_fd < 0)	    /* there is no file, can't read */
1042071d4279SBram Moolenaar 	return FAIL;
1043071d4279SBram Moolenaar 
1044071d4279SBram Moolenaar     page_size = mfp->mf_page_size;
1045071d4279SBram Moolenaar     offset = (off_t)page_size * hp->bh_bnum;
1046071d4279SBram Moolenaar     size = page_size * hp->bh_page_count;
1047071d4279SBram Moolenaar     if (lseek(mfp->mf_fd, offset, SEEK_SET) != offset)
1048071d4279SBram Moolenaar     {
10493f2d9814SBram Moolenaar 	PERROR(_("E294: Seek error in swap file read"));
1050071d4279SBram Moolenaar 	return FAIL;
1051071d4279SBram Moolenaar     }
1052*540fc6fbSBram Moolenaar     if ((unsigned)read_eintr(mfp->mf_fd, hp->bh_data, size) != size)
1053071d4279SBram Moolenaar     {
10543f2d9814SBram Moolenaar 	PERROR(_("E295: Read error in swap file"));
1055071d4279SBram Moolenaar 	return FAIL;
1056071d4279SBram Moolenaar     }
1057a8ffcbbfSBram Moolenaar 
1058a8ffcbbfSBram Moolenaar #ifdef FEAT_CRYPT
1059a8ffcbbfSBram Moolenaar     /* Decrypt if 'key' is set and this is a data block. */
1060a8ffcbbfSBram Moolenaar     if (*mfp->mf_buffer->b_p_key != NUL)
1061a8ffcbbfSBram Moolenaar 	ml_decrypt_data(mfp, hp->bh_data, offset, size);
1062a8ffcbbfSBram Moolenaar #endif
1063a8ffcbbfSBram Moolenaar 
1064071d4279SBram Moolenaar     return OK;
1065071d4279SBram Moolenaar }
1066071d4279SBram Moolenaar 
1067071d4279SBram Moolenaar /*
1068071d4279SBram Moolenaar  * write a block to disk
1069071d4279SBram Moolenaar  *
1070071d4279SBram Moolenaar  * Return FAIL for failure, OK otherwise
1071071d4279SBram Moolenaar  */
1072071d4279SBram Moolenaar     static int
1073071d4279SBram Moolenaar mf_write(mfp, hp)
1074071d4279SBram Moolenaar     memfile_T	*mfp;
1075071d4279SBram Moolenaar     bhdr_T	*hp;
1076071d4279SBram Moolenaar {
1077071d4279SBram Moolenaar     off_t	offset;	    /* offset in the file */
1078071d4279SBram Moolenaar     blocknr_T	nr;	    /* block nr which is being written */
1079071d4279SBram Moolenaar     bhdr_T	*hp2;
1080071d4279SBram Moolenaar     unsigned	page_size;  /* number of bytes in a page */
1081071d4279SBram Moolenaar     unsigned	page_count; /* number of pages written */
1082071d4279SBram Moolenaar     unsigned	size;	    /* number of bytes written */
1083071d4279SBram Moolenaar 
1084071d4279SBram Moolenaar     if (mfp->mf_fd < 0)	    /* there is no file, can't write */
1085071d4279SBram Moolenaar 	return FAIL;
1086071d4279SBram Moolenaar 
1087071d4279SBram Moolenaar     if (hp->bh_bnum < 0)	/* must assign file block number */
1088071d4279SBram Moolenaar 	if (mf_trans_add(mfp, hp) == FAIL)
1089071d4279SBram Moolenaar 	    return FAIL;
1090071d4279SBram Moolenaar 
1091071d4279SBram Moolenaar     page_size = mfp->mf_page_size;
1092071d4279SBram Moolenaar 
1093071d4279SBram Moolenaar     /*
1094071d4279SBram Moolenaar      * We don't want gaps in the file. Write the blocks in front of *hp
1095071d4279SBram Moolenaar      * to extend the file.
1096071d4279SBram Moolenaar      * If block 'mf_infile_count' is not in the hash list, it has been
1097071d4279SBram Moolenaar      * freed. Fill the space in the file with data from the current block.
1098071d4279SBram Moolenaar      */
1099071d4279SBram Moolenaar     for (;;)
1100071d4279SBram Moolenaar     {
1101071d4279SBram Moolenaar 	nr = hp->bh_bnum;
1102071d4279SBram Moolenaar 	if (nr > mfp->mf_infile_count)		/* beyond end of file */
1103071d4279SBram Moolenaar 	{
1104071d4279SBram Moolenaar 	    nr = mfp->mf_infile_count;
1105071d4279SBram Moolenaar 	    hp2 = mf_find_hash(mfp, nr);	/* NULL catched below */
1106071d4279SBram Moolenaar 	}
1107071d4279SBram Moolenaar 	else
1108071d4279SBram Moolenaar 	    hp2 = hp;
1109071d4279SBram Moolenaar 
1110071d4279SBram Moolenaar 	offset = (off_t)page_size * nr;
1111071d4279SBram Moolenaar 	if (lseek(mfp->mf_fd, offset, SEEK_SET) != offset)
1112071d4279SBram Moolenaar 	{
11133f2d9814SBram Moolenaar 	    PERROR(_("E296: Seek error in swap file write"));
1114071d4279SBram Moolenaar 	    return FAIL;
1115071d4279SBram Moolenaar 	}
1116071d4279SBram Moolenaar 	if (hp2 == NULL)	    /* freed block, fill with dummy data */
1117071d4279SBram Moolenaar 	    page_count = 1;
1118071d4279SBram Moolenaar 	else
1119071d4279SBram Moolenaar 	    page_count = hp2->bh_page_count;
1120071d4279SBram Moolenaar 	size = page_size * page_count;
1121a8ffcbbfSBram Moolenaar 	if (mf_write_block(mfp, hp2 == NULL ? hp : hp2, offset, size) == FAIL)
1122071d4279SBram Moolenaar 	{
1123071d4279SBram Moolenaar 	    /*
1124071d4279SBram Moolenaar 	     * Avoid repeating the error message, this mostly happens when the
112549325942SBram Moolenaar 	     * disk is full. We give the message again only after a successful
1126071d4279SBram Moolenaar 	     * write or when hitting a key. We keep on trying, in case some
1127071d4279SBram Moolenaar 	     * space becomes available.
1128071d4279SBram Moolenaar 	     */
1129071d4279SBram Moolenaar 	    if (!did_swapwrite_msg)
1130071d4279SBram Moolenaar 		EMSG(_("E297: Write error in swap file"));
1131071d4279SBram Moolenaar 	    did_swapwrite_msg = TRUE;
1132071d4279SBram Moolenaar 	    return FAIL;
1133071d4279SBram Moolenaar 	}
1134071d4279SBram Moolenaar 	did_swapwrite_msg = FALSE;
1135071d4279SBram Moolenaar 	if (hp2 != NULL)		    /* written a non-dummy block */
1136071d4279SBram Moolenaar 	    hp2->bh_flags &= ~BH_DIRTY;
1137071d4279SBram Moolenaar 					    /* appended to the file */
1138071d4279SBram Moolenaar 	if (nr + (blocknr_T)page_count > mfp->mf_infile_count)
1139071d4279SBram Moolenaar 	    mfp->mf_infile_count = nr + page_count;
1140071d4279SBram Moolenaar 	if (nr == hp->bh_bnum)		    /* written the desired block */
1141071d4279SBram Moolenaar 	    break;
1142071d4279SBram Moolenaar     }
1143071d4279SBram Moolenaar     return OK;
1144071d4279SBram Moolenaar }
1145071d4279SBram Moolenaar 
1146071d4279SBram Moolenaar /*
1147a8ffcbbfSBram Moolenaar  * Write block "hp" with data size "size" to file "mfp->mf_fd".
1148a8ffcbbfSBram Moolenaar  * Takes care of encryption.
1149a8ffcbbfSBram Moolenaar  * Return FAIL or OK.
1150a8ffcbbfSBram Moolenaar  */
1151a8ffcbbfSBram Moolenaar     static int
1152a8ffcbbfSBram Moolenaar mf_write_block(mfp, hp, offset, size)
1153a8ffcbbfSBram Moolenaar     memfile_T	*mfp;
1154a8ffcbbfSBram Moolenaar     bhdr_T	*hp;
1155a8ffcbbfSBram Moolenaar     off_t	offset UNUSED;
1156a8ffcbbfSBram Moolenaar     unsigned	size;
1157a8ffcbbfSBram Moolenaar {
1158a8ffcbbfSBram Moolenaar     char_u	*data = hp->bh_data;
1159a8ffcbbfSBram Moolenaar     int		result = OK;
1160a8ffcbbfSBram Moolenaar 
1161a8ffcbbfSBram Moolenaar #ifdef FEAT_CRYPT
1162a8ffcbbfSBram Moolenaar     /* Encrypt if 'key' is set and this is a data block. */
1163a8ffcbbfSBram Moolenaar     if (*mfp->mf_buffer->b_p_key != NUL)
1164a8ffcbbfSBram Moolenaar     {
1165a8ffcbbfSBram Moolenaar 	data = ml_encrypt_data(mfp, data, offset, size);
1166a8ffcbbfSBram Moolenaar 	if (data == NULL)
1167a8ffcbbfSBram Moolenaar 	    return FAIL;
1168a8ffcbbfSBram Moolenaar     }
1169a8ffcbbfSBram Moolenaar #endif
1170a8ffcbbfSBram Moolenaar 
1171*540fc6fbSBram Moolenaar     if ((unsigned)write_eintr(mfp->mf_fd, data, size) != size)
1172a8ffcbbfSBram Moolenaar 	result = FAIL;
1173a8ffcbbfSBram Moolenaar 
1174a8ffcbbfSBram Moolenaar #ifdef FEAT_CRYPT
1175a8ffcbbfSBram Moolenaar     if (data != hp->bh_data)
1176a8ffcbbfSBram Moolenaar 	vim_free(data);
1177a8ffcbbfSBram Moolenaar #endif
1178a8ffcbbfSBram Moolenaar 
1179a8ffcbbfSBram Moolenaar     return result;
1180a8ffcbbfSBram Moolenaar }
1181a8ffcbbfSBram Moolenaar 
1182a8ffcbbfSBram Moolenaar /*
1183071d4279SBram Moolenaar  * Make block number for *hp positive and add it to the translation list
1184071d4279SBram Moolenaar  *
1185071d4279SBram Moolenaar  * Return FAIL for failure, OK otherwise
1186071d4279SBram Moolenaar  */
1187071d4279SBram Moolenaar     static int
1188071d4279SBram Moolenaar mf_trans_add(mfp, hp)
1189071d4279SBram Moolenaar     memfile_T	*mfp;
1190071d4279SBram Moolenaar     bhdr_T	*hp;
1191071d4279SBram Moolenaar {
1192071d4279SBram Moolenaar     bhdr_T	*freep;
1193071d4279SBram Moolenaar     blocknr_T	new_bnum;
1194071d4279SBram Moolenaar     int		hash;
1195071d4279SBram Moolenaar     NR_TRANS	*np;
1196071d4279SBram Moolenaar     int		page_count;
1197071d4279SBram Moolenaar 
1198071d4279SBram Moolenaar     if (hp->bh_bnum >= 0)		    /* it's already positive */
1199071d4279SBram Moolenaar 	return OK;
1200071d4279SBram Moolenaar 
1201071d4279SBram Moolenaar     if ((np = (NR_TRANS *)alloc((unsigned)sizeof(NR_TRANS))) == NULL)
1202071d4279SBram Moolenaar 	return FAIL;
1203071d4279SBram Moolenaar 
1204071d4279SBram Moolenaar /*
1205a8ffcbbfSBram Moolenaar  * Get a new number for the block.
1206071d4279SBram Moolenaar  * If the first item in the free list has sufficient pages, use its number
1207071d4279SBram Moolenaar  * Otherwise use mf_blocknr_max.
1208071d4279SBram Moolenaar  */
1209071d4279SBram Moolenaar     freep = mfp->mf_free_first;
1210071d4279SBram Moolenaar     page_count = hp->bh_page_count;
1211071d4279SBram Moolenaar     if (freep != NULL && freep->bh_page_count >= page_count)
1212071d4279SBram Moolenaar     {
1213071d4279SBram Moolenaar 	new_bnum = freep->bh_bnum;
1214071d4279SBram Moolenaar 	/*
1215071d4279SBram Moolenaar 	 * If the page count of the free block was larger, recude it.
1216071d4279SBram Moolenaar 	 * If the page count matches, remove the block from the free list
1217071d4279SBram Moolenaar 	 */
1218071d4279SBram Moolenaar 	if (freep->bh_page_count > page_count)
1219071d4279SBram Moolenaar 	{
1220071d4279SBram Moolenaar 	    freep->bh_bnum += page_count;
1221071d4279SBram Moolenaar 	    freep->bh_page_count -= page_count;
1222071d4279SBram Moolenaar 	}
1223071d4279SBram Moolenaar 	else
1224071d4279SBram Moolenaar 	{
1225071d4279SBram Moolenaar 	    freep = mf_rem_free(mfp);
1226071d4279SBram Moolenaar 	    vim_free(freep);
1227071d4279SBram Moolenaar 	}
1228071d4279SBram Moolenaar     }
1229071d4279SBram Moolenaar     else
1230071d4279SBram Moolenaar     {
1231071d4279SBram Moolenaar 	new_bnum = mfp->mf_blocknr_max;
1232071d4279SBram Moolenaar 	mfp->mf_blocknr_max += page_count;
1233071d4279SBram Moolenaar     }
1234071d4279SBram Moolenaar 
1235071d4279SBram Moolenaar     np->nt_old_bnum = hp->bh_bnum;	    /* adjust number */
1236071d4279SBram Moolenaar     np->nt_new_bnum = new_bnum;
1237071d4279SBram Moolenaar 
1238071d4279SBram Moolenaar     mf_rem_hash(mfp, hp);		    /* remove from old hash list */
1239071d4279SBram Moolenaar     hp->bh_bnum = new_bnum;
1240071d4279SBram Moolenaar     mf_ins_hash(mfp, hp);		    /* insert in new hash list */
1241071d4279SBram Moolenaar 
1242071d4279SBram Moolenaar     hash = MEMHASH(np->nt_old_bnum);	    /* insert in trans list */
1243071d4279SBram Moolenaar     np->nt_next = mfp->mf_trans[hash];
1244071d4279SBram Moolenaar     mfp->mf_trans[hash] = np;
1245071d4279SBram Moolenaar     if (np->nt_next != NULL)
1246071d4279SBram Moolenaar 	np->nt_next->nt_prev = np;
1247071d4279SBram Moolenaar     np->nt_prev = NULL;
1248071d4279SBram Moolenaar 
1249071d4279SBram Moolenaar     return OK;
1250071d4279SBram Moolenaar }
1251071d4279SBram Moolenaar 
1252071d4279SBram Moolenaar /*
125349325942SBram Moolenaar  * Lookup a translation from the trans lists and delete the entry
1254071d4279SBram Moolenaar  *
1255071d4279SBram Moolenaar  * Return the positive new number when found, the old number when not found
1256071d4279SBram Moolenaar  */
1257071d4279SBram Moolenaar     blocknr_T
1258071d4279SBram Moolenaar mf_trans_del(mfp, old_nr)
1259071d4279SBram Moolenaar     memfile_T	*mfp;
1260071d4279SBram Moolenaar     blocknr_T	old_nr;
1261071d4279SBram Moolenaar {
1262071d4279SBram Moolenaar     int		hash;
1263071d4279SBram Moolenaar     NR_TRANS	*np;
1264071d4279SBram Moolenaar     blocknr_T	new_bnum;
1265071d4279SBram Moolenaar 
1266071d4279SBram Moolenaar     hash = MEMHASH(old_nr);
1267071d4279SBram Moolenaar     for (np = mfp->mf_trans[hash]; np != NULL; np = np->nt_next)
1268071d4279SBram Moolenaar 	if (np->nt_old_bnum == old_nr)
1269071d4279SBram Moolenaar 	    break;
1270071d4279SBram Moolenaar     if (np == NULL)		/* not found */
1271071d4279SBram Moolenaar 	return old_nr;
1272071d4279SBram Moolenaar 
1273071d4279SBram Moolenaar     mfp->mf_neg_count--;
1274071d4279SBram Moolenaar     new_bnum = np->nt_new_bnum;
1275071d4279SBram Moolenaar     if (np->nt_prev != NULL)		/* remove entry from the trans list */
1276071d4279SBram Moolenaar 	np->nt_prev->nt_next = np->nt_next;
1277071d4279SBram Moolenaar     else
1278071d4279SBram Moolenaar 	mfp->mf_trans[hash] = np->nt_next;
1279071d4279SBram Moolenaar     if (np->nt_next != NULL)
1280071d4279SBram Moolenaar 	np->nt_next->nt_prev = np->nt_prev;
1281071d4279SBram Moolenaar     vim_free(np);
1282071d4279SBram Moolenaar 
1283071d4279SBram Moolenaar     return new_bnum;
1284071d4279SBram Moolenaar }
1285071d4279SBram Moolenaar 
1286071d4279SBram Moolenaar /*
1287071d4279SBram Moolenaar  * Set mfp->mf_ffname according to mfp->mf_fname and some other things.
1288071d4279SBram Moolenaar  * Only called when creating or renaming the swapfile.	Either way it's a new
1289071d4279SBram Moolenaar  * name so we must work out the full path name.
1290071d4279SBram Moolenaar  */
1291071d4279SBram Moolenaar     void
1292071d4279SBram Moolenaar mf_set_ffname(mfp)
1293071d4279SBram Moolenaar     memfile_T	*mfp;
1294071d4279SBram Moolenaar {
1295071d4279SBram Moolenaar     mfp->mf_ffname = FullName_save(mfp->mf_fname, FALSE);
1296071d4279SBram Moolenaar }
1297071d4279SBram Moolenaar 
1298071d4279SBram Moolenaar /*
1299071d4279SBram Moolenaar  * Make the name of the file used for the memfile a full path.
1300071d4279SBram Moolenaar  * Used before doing a :cd
1301071d4279SBram Moolenaar  */
1302071d4279SBram Moolenaar     void
1303071d4279SBram Moolenaar mf_fullname(mfp)
1304071d4279SBram Moolenaar     memfile_T	*mfp;
1305071d4279SBram Moolenaar {
1306071d4279SBram Moolenaar     if (mfp != NULL && mfp->mf_fname != NULL && mfp->mf_ffname != NULL)
1307071d4279SBram Moolenaar     {
1308071d4279SBram Moolenaar 	vim_free(mfp->mf_fname);
1309071d4279SBram Moolenaar 	mfp->mf_fname = mfp->mf_ffname;
1310071d4279SBram Moolenaar 	mfp->mf_ffname = NULL;
1311071d4279SBram Moolenaar     }
1312071d4279SBram Moolenaar }
1313071d4279SBram Moolenaar 
1314071d4279SBram Moolenaar /*
1315071d4279SBram Moolenaar  * return TRUE if there are any translations pending for 'mfp'
1316071d4279SBram Moolenaar  */
1317071d4279SBram Moolenaar     int
1318071d4279SBram Moolenaar mf_need_trans(mfp)
1319071d4279SBram Moolenaar     memfile_T	*mfp;
1320071d4279SBram Moolenaar {
1321071d4279SBram Moolenaar     return (mfp->mf_fname != NULL && mfp->mf_neg_count > 0);
1322071d4279SBram Moolenaar }
1323071d4279SBram Moolenaar 
1324071d4279SBram Moolenaar /*
1325071d4279SBram Moolenaar  * Open a swap file for a memfile.
1326071d4279SBram Moolenaar  * The "fname" must be in allocated memory, and is consumed (also when an
1327071d4279SBram Moolenaar  * error occurs).
1328071d4279SBram Moolenaar  */
1329071d4279SBram Moolenaar     static void
1330071d4279SBram Moolenaar mf_do_open(mfp, fname, flags)
1331071d4279SBram Moolenaar     memfile_T	*mfp;
1332071d4279SBram Moolenaar     char_u	*fname;
1333071d4279SBram Moolenaar     int		flags;		/* flags for open() */
1334071d4279SBram Moolenaar {
1335071d4279SBram Moolenaar #ifdef HAVE_LSTAT
1336071d4279SBram Moolenaar     struct stat sb;
1337071d4279SBram Moolenaar #endif
1338071d4279SBram Moolenaar 
1339071d4279SBram Moolenaar     mfp->mf_fname = fname;
1340071d4279SBram Moolenaar 
1341071d4279SBram Moolenaar     /*
1342071d4279SBram Moolenaar      * Get the full path name before the open, because this is
1343071d4279SBram Moolenaar      * not possible after the open on the Amiga.
1344071d4279SBram Moolenaar      * fname cannot be NameBuff, because it must have been allocated.
1345071d4279SBram Moolenaar      */
1346071d4279SBram Moolenaar     mf_set_ffname(mfp);
1347071d4279SBram Moolenaar #if defined(MSDOS) || defined(MSWIN) || defined(RISCOS)
1348071d4279SBram Moolenaar     /*
1349071d4279SBram Moolenaar      * A ":!cd e:xxx" may change the directory without us knowning, use the
1350071d4279SBram Moolenaar      * full pathname always.  Careful: This frees fname!
1351071d4279SBram Moolenaar      */
1352071d4279SBram Moolenaar     mf_fullname(mfp);
1353071d4279SBram Moolenaar #endif
1354071d4279SBram Moolenaar 
1355071d4279SBram Moolenaar #ifdef HAVE_LSTAT
1356071d4279SBram Moolenaar     /*
1357071d4279SBram Moolenaar      * Extra security check: When creating a swap file it really shouldn't
1358071d4279SBram Moolenaar      * exist yet.  If there is a symbolic link, this is most likely an attack.
1359071d4279SBram Moolenaar      */
1360071d4279SBram Moolenaar     if ((flags & O_CREAT) && mch_lstat((char *)mfp->mf_fname, &sb) >= 0)
1361071d4279SBram Moolenaar     {
1362071d4279SBram Moolenaar 	mfp->mf_fd = -1;
1363071d4279SBram Moolenaar 	EMSG(_("E300: Swap file already exists (symlink attack?)"));
1364071d4279SBram Moolenaar     }
1365071d4279SBram Moolenaar     else
1366071d4279SBram Moolenaar #endif
1367071d4279SBram Moolenaar     {
1368071d4279SBram Moolenaar 	/*
1369071d4279SBram Moolenaar 	 * try to open the file
1370071d4279SBram Moolenaar 	 */
1371a5792f58SBram Moolenaar 	flags |= O_EXTRA | O_NOFOLLOW;
1372071d4279SBram Moolenaar #ifdef WIN32
1373071d4279SBram Moolenaar 	/* Prevent handle inheritance that cause problems with Cscope
1374071d4279SBram Moolenaar 	 * (swap file may not be deleted if cscope connection was open after
1375071d4279SBram Moolenaar 	 * the file) */
1376071d4279SBram Moolenaar 	flags |= O_NOINHERIT;
1377071d4279SBram Moolenaar #endif
1378071d4279SBram Moolenaar 	mfp->mf_fd = mch_open_rw((char *)mfp->mf_fname, flags);
1379071d4279SBram Moolenaar     }
1380071d4279SBram Moolenaar 
1381071d4279SBram Moolenaar     /*
1382071d4279SBram Moolenaar      * If the file cannot be opened, use memory only
1383071d4279SBram Moolenaar      */
1384071d4279SBram Moolenaar     if (mfp->mf_fd < 0)
1385071d4279SBram Moolenaar     {
1386071d4279SBram Moolenaar 	vim_free(mfp->mf_fname);
1387071d4279SBram Moolenaar 	vim_free(mfp->mf_ffname);
1388071d4279SBram Moolenaar 	mfp->mf_fname = NULL;
1389071d4279SBram Moolenaar 	mfp->mf_ffname = NULL;
1390071d4279SBram Moolenaar     }
1391071d4279SBram Moolenaar     else
1392588ebeb7SBram Moolenaar     {
1393f05da219SBram Moolenaar #ifdef HAVE_FD_CLOEXEC
1394f05da219SBram Moolenaar 	int fdflags = fcntl(mfp->mf_fd, F_GETFD);
1395f05da219SBram Moolenaar 	if (fdflags >= 0 && (fdflags & FD_CLOEXEC) == 0)
1396f05da219SBram Moolenaar 	    fcntl(mfp->mf_fd, F_SETFD, fdflags | FD_CLOEXEC);
1397f05da219SBram Moolenaar #endif
1398588ebeb7SBram Moolenaar #ifdef HAVE_SELINUX
1399588ebeb7SBram Moolenaar 	mch_copy_sec(fname, mfp->mf_fname);
1400588ebeb7SBram Moolenaar #endif
1401071d4279SBram Moolenaar 	mch_hide(mfp->mf_fname);    /* try setting the 'hidden' flag */
1402071d4279SBram Moolenaar     }
1403588ebeb7SBram Moolenaar }
1404