xref: /vim-8.2.3635/src/memfile.c (revision 588ebeb7)
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 
35071d4279SBram Moolenaar #if defined MSDOS || 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 #ifdef HAVE_FCNTL_H
42071d4279SBram Moolenaar # include <fcntl.h>
43071d4279SBram Moolenaar #endif
44071d4279SBram Moolenaar 
45071d4279SBram Moolenaar /*
46071d4279SBram Moolenaar  * Some systems have the page size in statfs.f_bsize, some in stat.st_blksize
47071d4279SBram Moolenaar  */
48071d4279SBram Moolenaar #ifdef HAVE_ST_BLKSIZE
49071d4279SBram Moolenaar # define STATFS stat
50071d4279SBram Moolenaar # define F_BSIZE st_blksize
51071d4279SBram Moolenaar # define fstatfs(fd, buf, len, nul) mch_fstat((fd), (buf))
52071d4279SBram Moolenaar #else
53071d4279SBram Moolenaar # ifdef HAVE_SYS_STATFS_H
54071d4279SBram Moolenaar #  include <sys/statfs.h>
55071d4279SBram Moolenaar #  define STATFS statfs
56071d4279SBram Moolenaar #  define F_BSIZE f_bsize
57071d4279SBram Moolenaar #  ifdef __MINT__		/* do we still need this? */
58071d4279SBram Moolenaar #   define fstatfs(fd, buf, len, nul) mch_fstat((fd), (buf))
59071d4279SBram Moolenaar #  endif
60071d4279SBram Moolenaar # endif
61071d4279SBram Moolenaar #endif
62071d4279SBram Moolenaar 
63071d4279SBram Moolenaar /*
64071d4279SBram Moolenaar  * for Amiga Dos 2.0x we use Flush
65071d4279SBram Moolenaar  */
66071d4279SBram Moolenaar #ifdef AMIGA
67071d4279SBram Moolenaar # ifdef FEAT_ARP
68071d4279SBram Moolenaar extern int dos2;			/* this is in os_amiga.c */
69071d4279SBram Moolenaar # endif
70071d4279SBram Moolenaar # ifdef SASC
71071d4279SBram Moolenaar #  include <proto/dos.h>
72071d4279SBram Moolenaar #  include <ios1.h>			/* for chkufb() */
73071d4279SBram Moolenaar # endif
74071d4279SBram Moolenaar #endif
75071d4279SBram Moolenaar 
76071d4279SBram Moolenaar #define MEMFILE_PAGE_SIZE 4096		/* default page size */
77071d4279SBram Moolenaar 
78071d4279SBram Moolenaar static long_u	total_mem_used = 0;	/* total memory used for memfiles */
79071d4279SBram Moolenaar 
80071d4279SBram Moolenaar static void mf_ins_hash __ARGS((memfile_T *, bhdr_T *));
81071d4279SBram Moolenaar static void mf_rem_hash __ARGS((memfile_T *, bhdr_T *));
82071d4279SBram Moolenaar static bhdr_T *mf_find_hash __ARGS((memfile_T *, blocknr_T));
83071d4279SBram Moolenaar static void mf_ins_used __ARGS((memfile_T *, bhdr_T *));
84071d4279SBram Moolenaar static void mf_rem_used __ARGS((memfile_T *, bhdr_T *));
85071d4279SBram Moolenaar static bhdr_T *mf_release __ARGS((memfile_T *, int));
86071d4279SBram Moolenaar static bhdr_T *mf_alloc_bhdr __ARGS((memfile_T *, int));
87071d4279SBram Moolenaar static void mf_free_bhdr __ARGS((bhdr_T *));
88071d4279SBram Moolenaar static void mf_ins_free __ARGS((memfile_T *, bhdr_T *));
89071d4279SBram Moolenaar static bhdr_T *mf_rem_free __ARGS((memfile_T *));
90071d4279SBram Moolenaar static int  mf_read __ARGS((memfile_T *, bhdr_T *));
91071d4279SBram Moolenaar static int  mf_write __ARGS((memfile_T *, bhdr_T *));
92071d4279SBram Moolenaar static int  mf_trans_add __ARGS((memfile_T *, bhdr_T *));
93071d4279SBram Moolenaar static void mf_do_open __ARGS((memfile_T *, char_u *, int));
94071d4279SBram Moolenaar 
95071d4279SBram Moolenaar /*
96071d4279SBram Moolenaar  * The functions for using a memfile:
97071d4279SBram Moolenaar  *
98071d4279SBram Moolenaar  * mf_open()	    open a new or existing memfile
99071d4279SBram Moolenaar  * mf_open_file()   open a swap file for an existing memfile
100071d4279SBram Moolenaar  * mf_close()	    close (and delete) a memfile
101071d4279SBram Moolenaar  * mf_new()	    create a new block in a memfile and lock it
102071d4279SBram Moolenaar  * mf_get()	    get an existing block and lock it
103071d4279SBram Moolenaar  * mf_put()	    unlock a block, may be marked for writing
104071d4279SBram Moolenaar  * mf_free()	    remove a block
105071d4279SBram Moolenaar  * mf_sync()	    sync changed parts of memfile to disk
106071d4279SBram Moolenaar  * mf_release_all() release as much memory as possible
107071d4279SBram Moolenaar  * mf_trans_del()   may translate negative to positive block number
108071d4279SBram Moolenaar  * mf_fullname()    make file name full path (use before first :cd)
109071d4279SBram Moolenaar  */
110071d4279SBram Moolenaar 
111071d4279SBram Moolenaar /*
112071d4279SBram Moolenaar  * Open an existing or new memory block file.
113071d4279SBram Moolenaar  *
114071d4279SBram Moolenaar  *  fname:	name of file to use (NULL means no file at all)
115071d4279SBram Moolenaar  *		Note: fname must have been allocated, it is not copied!
116071d4279SBram Moolenaar  *			If opening the file fails, fname is freed.
117071d4279SBram Moolenaar  *  flags:	flags for open() call
118071d4279SBram Moolenaar  *
119071d4279SBram Moolenaar  *  If fname != NULL and file cannot be opened, fail.
120071d4279SBram Moolenaar  *
121071d4279SBram Moolenaar  * return value: identifier for this memory block file.
122071d4279SBram Moolenaar  */
123071d4279SBram Moolenaar     memfile_T *
124071d4279SBram Moolenaar mf_open(fname, flags)
125071d4279SBram Moolenaar     char_u	*fname;
126071d4279SBram Moolenaar     int		flags;
127071d4279SBram Moolenaar {
128071d4279SBram Moolenaar     memfile_T		*mfp;
129071d4279SBram Moolenaar     int			i;
130071d4279SBram Moolenaar     off_t		size;
131071d4279SBram Moolenaar #if defined(STATFS) && defined(UNIX) && !defined(__QNX__)
132071d4279SBram Moolenaar # define USE_FSTATFS
133071d4279SBram Moolenaar     struct STATFS	stf;
134071d4279SBram Moolenaar #endif
135071d4279SBram Moolenaar 
136071d4279SBram Moolenaar     if ((mfp = (memfile_T *)alloc((unsigned)sizeof(memfile_T))) == NULL)
137071d4279SBram Moolenaar 	return NULL;
138071d4279SBram Moolenaar 
139071d4279SBram Moolenaar     if (fname == NULL)	    /* no file for this memfile, use memory only */
140071d4279SBram Moolenaar     {
141071d4279SBram Moolenaar 	mfp->mf_fname = NULL;
142071d4279SBram Moolenaar 	mfp->mf_ffname = NULL;
143071d4279SBram Moolenaar 	mfp->mf_fd = -1;
144071d4279SBram Moolenaar     }
145071d4279SBram Moolenaar     else
146071d4279SBram Moolenaar     {
147071d4279SBram Moolenaar 	mf_do_open(mfp, fname, flags);	/* try to open the file */
148071d4279SBram Moolenaar 
149071d4279SBram Moolenaar 	/* if the file cannot be opened, return here */
150071d4279SBram Moolenaar 	if (mfp->mf_fd < 0)
151071d4279SBram Moolenaar 	{
152071d4279SBram Moolenaar 	    vim_free(mfp);
153071d4279SBram Moolenaar 	    return NULL;
154071d4279SBram Moolenaar 	}
155071d4279SBram Moolenaar     }
156071d4279SBram Moolenaar 
157071d4279SBram Moolenaar     mfp->mf_free_first = NULL;		/* free list is empty */
158071d4279SBram Moolenaar     mfp->mf_used_first = NULL;		/* used list is empty */
159071d4279SBram Moolenaar     mfp->mf_used_last = NULL;
160071d4279SBram Moolenaar     mfp->mf_dirty = FALSE;
161071d4279SBram Moolenaar     mfp->mf_used_count = 0;
162071d4279SBram Moolenaar     for (i = 0; i < MEMHASHSIZE; ++i)
163071d4279SBram Moolenaar     {
164071d4279SBram Moolenaar 	mfp->mf_hash[i] = NULL;		/* hash lists are empty */
165071d4279SBram Moolenaar 	mfp->mf_trans[i] = NULL;	/* trans lists are empty */
166071d4279SBram Moolenaar     }
167071d4279SBram Moolenaar     mfp->mf_page_size = MEMFILE_PAGE_SIZE;
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 /*
241071d4279SBram 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      */
422071d4279SBram Moolenaar     (void)vim_memset((char *)(hp->bh_data), 0, (size_t)mfp->mf_page_size);
423071d4279SBram Moolenaar 
424071d4279SBram Moolenaar     return hp;
425071d4279SBram Moolenaar }
426071d4279SBram Moolenaar 
427071d4279SBram Moolenaar /*
428071d4279SBram Moolenaar  * get existing block 'nr' with 'page_count' pages
429071d4279SBram Moolenaar  *
430071d4279SBram Moolenaar  * Note: The caller should first check a negative nr with mf_trans_del()
431071d4279SBram Moolenaar  */
432071d4279SBram Moolenaar     bhdr_T *
433071d4279SBram Moolenaar mf_get(mfp, nr, page_count)
434071d4279SBram Moolenaar     memfile_T	*mfp;
435071d4279SBram Moolenaar     blocknr_T	nr;
436071d4279SBram Moolenaar     int		page_count;
437071d4279SBram Moolenaar {
438071d4279SBram Moolenaar     bhdr_T    *hp;
439071d4279SBram Moolenaar 						/* doesn't exist */
440071d4279SBram Moolenaar     if (nr >= mfp->mf_blocknr_max || nr <= mfp->mf_blocknr_min)
441071d4279SBram Moolenaar 	return NULL;
442071d4279SBram Moolenaar 
443071d4279SBram Moolenaar     /*
444071d4279SBram Moolenaar      * see if it is in the cache
445071d4279SBram Moolenaar      */
446071d4279SBram Moolenaar     hp = mf_find_hash(mfp, nr);
447071d4279SBram Moolenaar     if (hp == NULL)	/* not in the hash list */
448071d4279SBram Moolenaar     {
449071d4279SBram Moolenaar 	if (nr < 0 || nr >= mfp->mf_infile_count)   /* can't be in the file */
450071d4279SBram Moolenaar 	    return NULL;
451071d4279SBram Moolenaar 
452071d4279SBram Moolenaar 	/* could check here if the block is in the free list */
453071d4279SBram Moolenaar 
454071d4279SBram Moolenaar 	/*
455071d4279SBram Moolenaar 	 * Check if we need to flush an existing block.
456071d4279SBram Moolenaar 	 * If so, use that block.
457071d4279SBram Moolenaar 	 * If not, allocate a new block.
458071d4279SBram Moolenaar 	 */
459071d4279SBram Moolenaar 	hp = mf_release(mfp, page_count);
460071d4279SBram Moolenaar 	if (hp == NULL && (hp = mf_alloc_bhdr(mfp, page_count)) == NULL)
461071d4279SBram Moolenaar 	    return NULL;
462071d4279SBram Moolenaar 
463071d4279SBram Moolenaar 	hp->bh_bnum = nr;
464071d4279SBram Moolenaar 	hp->bh_flags = 0;
465071d4279SBram Moolenaar 	hp->bh_page_count = page_count;
466071d4279SBram Moolenaar 	if (mf_read(mfp, hp) == FAIL)	    /* cannot read the block! */
467071d4279SBram Moolenaar 	{
468071d4279SBram Moolenaar 	    mf_free_bhdr(hp);
469071d4279SBram Moolenaar 	    return NULL;
470071d4279SBram Moolenaar 	}
471071d4279SBram Moolenaar     }
472071d4279SBram Moolenaar     else
473071d4279SBram Moolenaar     {
474071d4279SBram Moolenaar 	mf_rem_used(mfp, hp);	/* remove from list, insert in front below */
475071d4279SBram Moolenaar 	mf_rem_hash(mfp, hp);
476071d4279SBram Moolenaar     }
477071d4279SBram Moolenaar 
478071d4279SBram Moolenaar     hp->bh_flags |= BH_LOCKED;
479071d4279SBram Moolenaar     mf_ins_used(mfp, hp);	/* put in front of used list */
480071d4279SBram Moolenaar     mf_ins_hash(mfp, hp);	/* put in front of hash list */
481071d4279SBram Moolenaar 
482071d4279SBram Moolenaar     return hp;
483071d4279SBram Moolenaar }
484071d4279SBram Moolenaar 
485071d4279SBram Moolenaar /*
486071d4279SBram Moolenaar  * release the block *hp
487071d4279SBram Moolenaar  *
488071d4279SBram Moolenaar  *   dirty: Block must be written to file later
489071d4279SBram Moolenaar  *   infile: Block should be in file (needed for recovery)
490071d4279SBram Moolenaar  *
491071d4279SBram Moolenaar  *  no return value, function cannot fail
492071d4279SBram Moolenaar  */
493071d4279SBram Moolenaar     void
494071d4279SBram Moolenaar mf_put(mfp, hp, dirty, infile)
495071d4279SBram Moolenaar     memfile_T	*mfp;
496071d4279SBram Moolenaar     bhdr_T	*hp;
497071d4279SBram Moolenaar     int		dirty;
498071d4279SBram Moolenaar     int		infile;
499071d4279SBram Moolenaar {
500071d4279SBram Moolenaar     int		flags;
501071d4279SBram Moolenaar 
502071d4279SBram Moolenaar     flags = hp->bh_flags;
503071d4279SBram Moolenaar 
504071d4279SBram Moolenaar     if ((flags & BH_LOCKED) == 0)
505071d4279SBram Moolenaar 	EMSG(_("E293: block was not locked"));
506071d4279SBram Moolenaar     flags &= ~BH_LOCKED;
507071d4279SBram Moolenaar     if (dirty)
508071d4279SBram Moolenaar     {
509071d4279SBram Moolenaar 	flags |= BH_DIRTY;
510071d4279SBram Moolenaar 	mfp->mf_dirty = TRUE;
511071d4279SBram Moolenaar     }
512071d4279SBram Moolenaar     hp->bh_flags = flags;
513071d4279SBram Moolenaar     if (infile)
514071d4279SBram Moolenaar 	mf_trans_add(mfp, hp);	    /* may translate negative in positive nr */
515071d4279SBram Moolenaar }
516071d4279SBram Moolenaar 
517071d4279SBram Moolenaar /*
518071d4279SBram Moolenaar  * block *hp is no longer in used, may put it in the free list of memfile *mfp
519071d4279SBram Moolenaar  */
520071d4279SBram Moolenaar     void
521071d4279SBram Moolenaar mf_free(mfp, hp)
522071d4279SBram Moolenaar     memfile_T	*mfp;
523071d4279SBram Moolenaar     bhdr_T	*hp;
524071d4279SBram Moolenaar {
525071d4279SBram Moolenaar     vim_free(hp->bh_data);	/* free the memory */
526071d4279SBram Moolenaar     mf_rem_hash(mfp, hp);	/* get *hp out of the hash list */
527071d4279SBram Moolenaar     mf_rem_used(mfp, hp);	/* get *hp out of the used list */
528071d4279SBram Moolenaar     if (hp->bh_bnum < 0)
529071d4279SBram Moolenaar     {
530071d4279SBram Moolenaar 	vim_free(hp);		/* don't want negative numbers in free list */
531071d4279SBram Moolenaar 	mfp->mf_neg_count--;
532071d4279SBram Moolenaar     }
533071d4279SBram Moolenaar     else
534071d4279SBram Moolenaar 	mf_ins_free(mfp, hp);	/* put *hp in the free list */
535071d4279SBram Moolenaar }
536071d4279SBram Moolenaar 
53789f3727bSBram Moolenaar #if defined(__MORPHOS__) && defined(__libnix__)
538071d4279SBram Moolenaar /* function is missing in MorphOS libnix version */
539071d4279SBram Moolenaar extern unsigned long *__stdfiledes;
540071d4279SBram Moolenaar 
541071d4279SBram Moolenaar     static unsigned long
542071d4279SBram Moolenaar fdtofh(int filedescriptor)
543071d4279SBram Moolenaar {
544071d4279SBram Moolenaar     return __stdfiledes[filedescriptor];
545071d4279SBram Moolenaar }
546071d4279SBram Moolenaar #endif
547071d4279SBram Moolenaar 
548071d4279SBram Moolenaar /*
549071d4279SBram Moolenaar  * Sync the memory file *mfp to disk.
550071d4279SBram Moolenaar  * Flags:
551071d4279SBram Moolenaar  *  MFS_ALL	If not given, blocks with negative numbers are not synced,
552071d4279SBram Moolenaar  *		even when they are dirty!
553071d4279SBram Moolenaar  *  MFS_STOP	Stop syncing when a character becomes available, but sync at
554071d4279SBram Moolenaar  *		least one block.
555071d4279SBram Moolenaar  *  MFS_FLUSH	Make sure buffers are flushed to disk, so they will survive a
556071d4279SBram Moolenaar  *		system crash.
557071d4279SBram Moolenaar  *  MFS_ZERO	Only write block 0.
558071d4279SBram Moolenaar  *
559071d4279SBram Moolenaar  * Return FAIL for failure, OK otherwise
560071d4279SBram Moolenaar  */
561071d4279SBram Moolenaar     int
562071d4279SBram Moolenaar mf_sync(mfp, flags)
563071d4279SBram Moolenaar     memfile_T	*mfp;
564071d4279SBram Moolenaar     int		flags;
565071d4279SBram Moolenaar {
566071d4279SBram Moolenaar     int		status;
567071d4279SBram Moolenaar     bhdr_T	*hp;
568071d4279SBram Moolenaar #if defined(SYNC_DUP_CLOSE) && !defined(MSDOS)
569071d4279SBram Moolenaar     int		fd;
570071d4279SBram Moolenaar #endif
571071d4279SBram Moolenaar     int		got_int_save = got_int;
572071d4279SBram Moolenaar 
573071d4279SBram Moolenaar     if (mfp->mf_fd < 0)	    /* there is no file, nothing to do */
574071d4279SBram Moolenaar     {
575071d4279SBram Moolenaar 	mfp->mf_dirty = FALSE;
576071d4279SBram Moolenaar 	return FAIL;
577071d4279SBram Moolenaar     }
578071d4279SBram Moolenaar 
579071d4279SBram Moolenaar     /* Only a CTRL-C while writing will break us here, not one typed
580071d4279SBram Moolenaar      * previously. */
581071d4279SBram Moolenaar     got_int = FALSE;
582071d4279SBram Moolenaar 
583071d4279SBram Moolenaar     /*
584071d4279SBram Moolenaar      * sync from last to first (may reduce the probability of an inconsistent
585071d4279SBram Moolenaar      * file) If a write fails, it is very likely caused by a full filesystem.
586071d4279SBram Moolenaar      * Then we only try to write blocks within the existing file. If that also
587071d4279SBram Moolenaar      * fails then we give up.
588071d4279SBram Moolenaar      */
589071d4279SBram Moolenaar     status = OK;
590071d4279SBram Moolenaar     for (hp = mfp->mf_used_last; hp != NULL; hp = hp->bh_prev)
591071d4279SBram Moolenaar 	if (((flags & MFS_ALL) || hp->bh_bnum >= 0)
592071d4279SBram Moolenaar 		&& (hp->bh_flags & BH_DIRTY)
593071d4279SBram Moolenaar 		&& (status == OK || (hp->bh_bnum >= 0
594071d4279SBram Moolenaar 		    && hp->bh_bnum < mfp->mf_infile_count)))
595071d4279SBram Moolenaar 	{
596071d4279SBram Moolenaar 	    if ((flags & MFS_ZERO) && hp->bh_bnum != 0)
597071d4279SBram Moolenaar 		continue;
598071d4279SBram Moolenaar 	    if (mf_write(mfp, hp) == FAIL)
599071d4279SBram Moolenaar 	    {
600071d4279SBram Moolenaar 		if (status == FAIL)	/* double error: quit syncing */
601071d4279SBram Moolenaar 		    break;
602071d4279SBram Moolenaar 		status = FAIL;
603071d4279SBram Moolenaar 	    }
604071d4279SBram Moolenaar 	    if (flags & MFS_STOP)
605071d4279SBram Moolenaar 	    {
606071d4279SBram Moolenaar 		/* Stop when char available now. */
607071d4279SBram Moolenaar 		if (ui_char_avail())
608071d4279SBram Moolenaar 		    break;
609071d4279SBram Moolenaar 	    }
610071d4279SBram Moolenaar 	    else
611071d4279SBram Moolenaar 		ui_breakcheck();
612071d4279SBram Moolenaar 	    if (got_int)
613071d4279SBram Moolenaar 		break;
614071d4279SBram Moolenaar 	}
615071d4279SBram Moolenaar 
616071d4279SBram Moolenaar     /*
617071d4279SBram Moolenaar      * If the whole list is flushed, the memfile is not dirty anymore.
618071d4279SBram Moolenaar      * In case of an error this flag is also set, to avoid trying all the time.
619071d4279SBram Moolenaar      */
620071d4279SBram Moolenaar     if (hp == NULL || status == FAIL)
621071d4279SBram Moolenaar 	mfp->mf_dirty = FALSE;
622071d4279SBram Moolenaar 
623071d4279SBram Moolenaar     if ((flags & MFS_FLUSH) && *p_sws != NUL)
624071d4279SBram Moolenaar     {
625071d4279SBram Moolenaar #if defined(UNIX)
626071d4279SBram Moolenaar # ifdef HAVE_FSYNC
627071d4279SBram Moolenaar 	/*
628071d4279SBram Moolenaar 	 * most Unixes have the very useful fsync() function, just what we need.
629071d4279SBram Moolenaar 	 * However, with OS/2 and EMX it is also available, but there are
630071d4279SBram Moolenaar 	 * reports of bad problems with it (a bug in HPFS.IFS).
631071d4279SBram Moolenaar 	 * So we disable use of it here in case someone tries to be smart
632071d4279SBram Moolenaar 	 * and changes os_os2_cfg.h... (even though there is no __EMX__ test
633071d4279SBram Moolenaar 	 * in the #if, as __EMX__ does not have sync(); we hope for a timely
634071d4279SBram Moolenaar 	 * sync from the system itself).
635071d4279SBram Moolenaar 	 */
636071d4279SBram Moolenaar #  if defined(__EMX__)
637071d4279SBram Moolenaar    error "Dont use fsync with EMX! Read emxdoc.doc or emxfix01.doc for info."
638071d4279SBram Moolenaar #  endif
639071d4279SBram Moolenaar 	if (STRCMP(p_sws, "fsync") == 0)
640071d4279SBram Moolenaar 	{
641071d4279SBram Moolenaar 	    if (fsync(mfp->mf_fd))
642071d4279SBram Moolenaar 		status = FAIL;
643071d4279SBram Moolenaar 	}
644071d4279SBram Moolenaar 	else
645071d4279SBram Moolenaar # endif
646071d4279SBram Moolenaar 	    /* OpenNT is strictly POSIX (Benzinger) */
647071d4279SBram Moolenaar 	    /* Tandem/Himalaya NSK-OSS doesn't have sync() */
648071d4279SBram Moolenaar # if defined(__OPENNT) || defined(__TANDEM)
649071d4279SBram Moolenaar 	    fflush(NULL);
650071d4279SBram Moolenaar # else
651071d4279SBram Moolenaar 	    sync();
652071d4279SBram Moolenaar # endif
653071d4279SBram Moolenaar #endif
654071d4279SBram Moolenaar #ifdef VMS
655071d4279SBram Moolenaar 	if (STRCMP(p_sws, "fsync") == 0)
656071d4279SBram Moolenaar 	{
657071d4279SBram Moolenaar 	    if (fsync(mfp->mf_fd))
658071d4279SBram Moolenaar 		status = FAIL;
659071d4279SBram Moolenaar 	}
660071d4279SBram Moolenaar #endif
661071d4279SBram Moolenaar #ifdef MSDOS
662071d4279SBram Moolenaar 	if (_dos_commit(mfp->mf_fd))
663071d4279SBram Moolenaar 	    status = FAIL;
664071d4279SBram Moolenaar #else
665071d4279SBram Moolenaar # ifdef SYNC_DUP_CLOSE
666071d4279SBram Moolenaar 	/*
667071d4279SBram Moolenaar 	 * Win32 is a bit more work: Duplicate the file handle and close it.
668071d4279SBram Moolenaar 	 * This should flush the file to disk.
669071d4279SBram Moolenaar 	 */
670071d4279SBram Moolenaar 	if ((fd = dup(mfp->mf_fd)) >= 0)
671071d4279SBram Moolenaar 	    close(fd);
672071d4279SBram Moolenaar # endif
673071d4279SBram Moolenaar #endif
674071d4279SBram Moolenaar #ifdef AMIGA
6755a6404cfSBram Moolenaar # if defined(__AROS__) || defined(__amigaos4__)
676071d4279SBram Moolenaar 	if (fsync(mfp->mf_fd) != 0)
677071d4279SBram Moolenaar 	    status = FAIL;
678071d4279SBram Moolenaar # else
679071d4279SBram Moolenaar 	/*
680071d4279SBram Moolenaar 	 * Flush() only exists for AmigaDos 2.0.
681071d4279SBram Moolenaar 	 * For 1.3 it should be done with close() + open(), but then the risk
682071d4279SBram Moolenaar 	 * is that the open() may fail and lose the file....
683071d4279SBram Moolenaar 	 */
684071d4279SBram Moolenaar #  ifdef FEAT_ARP
685071d4279SBram Moolenaar 	if (dos2)
686071d4279SBram Moolenaar #  endif
687071d4279SBram Moolenaar #  ifdef SASC
688071d4279SBram Moolenaar 	{
689071d4279SBram Moolenaar 	    struct UFB *fp = chkufb(mfp->mf_fd);
690071d4279SBram Moolenaar 
691071d4279SBram Moolenaar 	    if (fp != NULL)
692071d4279SBram Moolenaar 		Flush(fp->ufbfh);
693071d4279SBram Moolenaar 	}
694071d4279SBram Moolenaar #  else
695071d4279SBram Moolenaar #   if defined(_DCC) || defined(__GNUC__) || defined(__MORPHOS__)
696071d4279SBram Moolenaar 	{
69789f3727bSBram Moolenaar #    if defined(__GNUC__) && !defined(__MORPHOS__) && defined(__libnix__)
698071d4279SBram Moolenaar 	    /* Have function (in libnix at least),
699071d4279SBram Moolenaar 	     * but ain't got no prototype anywhere. */
700071d4279SBram Moolenaar 	    extern unsigned long fdtofh(int filedescriptor);
701071d4279SBram Moolenaar #    endif
70289f3727bSBram Moolenaar #    if !defined(__libnix__)
70389f3727bSBram Moolenaar 	    fflush(NULL);
70489f3727bSBram Moolenaar #    else
705071d4279SBram Moolenaar 	    BPTR fh = (BPTR)fdtofh(mfp->mf_fd);
706071d4279SBram Moolenaar 
707071d4279SBram Moolenaar 	    if (fh != 0)
708071d4279SBram Moolenaar 		Flush(fh);
70989f3727bSBram Moolenaar #    endif
710071d4279SBram Moolenaar 	}
711071d4279SBram Moolenaar #   else /* assume Manx */
712071d4279SBram Moolenaar 	    Flush(_devtab[mfp->mf_fd].fd);
713071d4279SBram Moolenaar #   endif
714071d4279SBram Moolenaar #  endif
715071d4279SBram Moolenaar # endif
716071d4279SBram Moolenaar #endif /* AMIGA */
717071d4279SBram Moolenaar     }
718071d4279SBram Moolenaar 
719071d4279SBram Moolenaar     got_int |= got_int_save;
720071d4279SBram Moolenaar 
721071d4279SBram Moolenaar     return status;
722071d4279SBram Moolenaar }
723071d4279SBram Moolenaar 
724071d4279SBram Moolenaar /*
725c32840f2SBram Moolenaar  * For all blocks in memory file *mfp that have a positive block number set
726c32840f2SBram Moolenaar  * the dirty flag.  These are blocks that need to be written to a newly
727c32840f2SBram Moolenaar  * created swapfile.
728c32840f2SBram Moolenaar  */
729c32840f2SBram Moolenaar     void
730c32840f2SBram Moolenaar mf_set_dirty(mfp)
731c32840f2SBram Moolenaar     memfile_T	*mfp;
732c32840f2SBram Moolenaar {
733c32840f2SBram Moolenaar     bhdr_T	*hp;
734c32840f2SBram Moolenaar 
735c32840f2SBram Moolenaar     for (hp = mfp->mf_used_last; hp != NULL; hp = hp->bh_prev)
736c32840f2SBram Moolenaar 	if (hp->bh_bnum > 0)
737c32840f2SBram Moolenaar 	    hp->bh_flags |= BH_DIRTY;
738c32840f2SBram Moolenaar     mfp->mf_dirty = TRUE;
739c32840f2SBram Moolenaar }
740c32840f2SBram Moolenaar 
741c32840f2SBram Moolenaar /*
742071d4279SBram Moolenaar  * insert block *hp in front of hashlist of memfile *mfp
743071d4279SBram Moolenaar  */
744071d4279SBram Moolenaar     static void
745071d4279SBram Moolenaar mf_ins_hash(mfp, hp)
746071d4279SBram Moolenaar     memfile_T	*mfp;
747071d4279SBram Moolenaar     bhdr_T	*hp;
748071d4279SBram Moolenaar {
749071d4279SBram Moolenaar     bhdr_T	*hhp;
750071d4279SBram Moolenaar     int		hash;
751071d4279SBram Moolenaar 
752071d4279SBram Moolenaar     hash = MEMHASH(hp->bh_bnum);
753071d4279SBram Moolenaar     hhp = mfp->mf_hash[hash];
754071d4279SBram Moolenaar     hp->bh_hash_next = hhp;
755071d4279SBram Moolenaar     hp->bh_hash_prev = NULL;
756071d4279SBram Moolenaar     if (hhp != NULL)
757071d4279SBram Moolenaar 	hhp->bh_hash_prev = hp;
758071d4279SBram Moolenaar     mfp->mf_hash[hash] = hp;
759071d4279SBram Moolenaar }
760071d4279SBram Moolenaar 
761071d4279SBram Moolenaar /*
762071d4279SBram Moolenaar  * remove block *hp from hashlist of memfile list *mfp
763071d4279SBram Moolenaar  */
764071d4279SBram Moolenaar     static void
765071d4279SBram Moolenaar mf_rem_hash(mfp, hp)
766071d4279SBram Moolenaar     memfile_T	*mfp;
767071d4279SBram Moolenaar     bhdr_T	*hp;
768071d4279SBram Moolenaar {
769071d4279SBram Moolenaar     if (hp->bh_hash_prev == NULL)
770071d4279SBram Moolenaar 	mfp->mf_hash[MEMHASH(hp->bh_bnum)] = hp->bh_hash_next;
771071d4279SBram Moolenaar     else
772071d4279SBram Moolenaar 	hp->bh_hash_prev->bh_hash_next = hp->bh_hash_next;
773071d4279SBram Moolenaar 
774071d4279SBram Moolenaar     if (hp->bh_hash_next)
775071d4279SBram Moolenaar 	hp->bh_hash_next->bh_hash_prev = hp->bh_hash_prev;
776071d4279SBram Moolenaar }
777071d4279SBram Moolenaar 
778071d4279SBram Moolenaar /*
779071d4279SBram Moolenaar  * look in hash lists of memfile *mfp for block header with number 'nr'
780071d4279SBram Moolenaar  */
781071d4279SBram Moolenaar     static bhdr_T *
782071d4279SBram Moolenaar mf_find_hash(mfp, nr)
783071d4279SBram Moolenaar     memfile_T	*mfp;
784071d4279SBram Moolenaar     blocknr_T	nr;
785071d4279SBram Moolenaar {
786071d4279SBram Moolenaar     bhdr_T	*hp;
787071d4279SBram Moolenaar 
788071d4279SBram Moolenaar     for (hp = mfp->mf_hash[MEMHASH(nr)]; hp != NULL; hp = hp->bh_hash_next)
789071d4279SBram Moolenaar 	if (hp->bh_bnum == nr)
790071d4279SBram Moolenaar 	    break;
791071d4279SBram Moolenaar     return hp;
792071d4279SBram Moolenaar }
793071d4279SBram Moolenaar 
794071d4279SBram Moolenaar /*
795071d4279SBram Moolenaar  * insert block *hp in front of used list of memfile *mfp
796071d4279SBram Moolenaar  */
797071d4279SBram Moolenaar     static void
798071d4279SBram Moolenaar mf_ins_used(mfp, hp)
799071d4279SBram Moolenaar     memfile_T	*mfp;
800071d4279SBram Moolenaar     bhdr_T	*hp;
801071d4279SBram Moolenaar {
802071d4279SBram Moolenaar     hp->bh_next = mfp->mf_used_first;
803071d4279SBram Moolenaar     mfp->mf_used_first = hp;
804071d4279SBram Moolenaar     hp->bh_prev = NULL;
805071d4279SBram Moolenaar     if (hp->bh_next == NULL)	    /* list was empty, adjust last pointer */
806071d4279SBram Moolenaar 	mfp->mf_used_last = hp;
807071d4279SBram Moolenaar     else
808071d4279SBram Moolenaar 	hp->bh_next->bh_prev = hp;
809071d4279SBram Moolenaar     mfp->mf_used_count += hp->bh_page_count;
810071d4279SBram Moolenaar     total_mem_used += hp->bh_page_count * mfp->mf_page_size;
811071d4279SBram Moolenaar }
812071d4279SBram Moolenaar 
813071d4279SBram Moolenaar /*
814071d4279SBram Moolenaar  * remove block *hp from used list of memfile *mfp
815071d4279SBram Moolenaar  */
816071d4279SBram Moolenaar     static void
817071d4279SBram Moolenaar mf_rem_used(mfp, hp)
818071d4279SBram Moolenaar     memfile_T	*mfp;
819071d4279SBram Moolenaar     bhdr_T	*hp;
820071d4279SBram Moolenaar {
821071d4279SBram Moolenaar     if (hp->bh_next == NULL)	    /* last block in used list */
822071d4279SBram Moolenaar 	mfp->mf_used_last = hp->bh_prev;
823071d4279SBram Moolenaar     else
824071d4279SBram Moolenaar 	hp->bh_next->bh_prev = hp->bh_prev;
825071d4279SBram Moolenaar     if (hp->bh_prev == NULL)	    /* first block in used list */
826071d4279SBram Moolenaar 	mfp->mf_used_first = hp->bh_next;
827071d4279SBram Moolenaar     else
828071d4279SBram Moolenaar 	hp->bh_prev->bh_next = hp->bh_next;
829071d4279SBram Moolenaar     mfp->mf_used_count -= hp->bh_page_count;
830071d4279SBram Moolenaar     total_mem_used -= hp->bh_page_count * mfp->mf_page_size;
831071d4279SBram Moolenaar }
832071d4279SBram Moolenaar 
833071d4279SBram Moolenaar /*
834071d4279SBram Moolenaar  * Release the least recently used block from the used list if the number
835071d4279SBram Moolenaar  * of used memory blocks gets to big.
836071d4279SBram Moolenaar  *
837071d4279SBram Moolenaar  * Return the block header to the caller, including the memory block, so
838071d4279SBram Moolenaar  * it can be re-used. Make sure the page_count is right.
839071d4279SBram Moolenaar  */
840071d4279SBram Moolenaar     static bhdr_T *
841071d4279SBram Moolenaar mf_release(mfp, page_count)
842071d4279SBram Moolenaar     memfile_T	*mfp;
843071d4279SBram Moolenaar     int		page_count;
844071d4279SBram Moolenaar {
845071d4279SBram Moolenaar     bhdr_T	*hp;
846071d4279SBram Moolenaar     int		need_release;
847071d4279SBram Moolenaar     buf_T	*buf;
848071d4279SBram Moolenaar 
849071d4279SBram Moolenaar     /* don't release while in mf_close_file() */
85047b8b15bSBram Moolenaar     if (mf_dont_release)
851071d4279SBram Moolenaar 	return NULL;
852071d4279SBram Moolenaar 
853071d4279SBram Moolenaar     /*
854071d4279SBram Moolenaar      * Need to release a block if the number of blocks for this memfile is
855071d4279SBram Moolenaar      * higher than the maximum or total memory used is over 'maxmemtot'
856071d4279SBram Moolenaar      */
857071d4279SBram Moolenaar     need_release = ((mfp->mf_used_count >= mfp->mf_used_count_max)
858071d4279SBram Moolenaar 				  || (total_mem_used >> 10) >= (long_u)p_mmt);
859071d4279SBram Moolenaar 
860071d4279SBram Moolenaar     /*
861071d4279SBram Moolenaar      * Try to create a swap file if the amount of memory used is getting too
862071d4279SBram Moolenaar      * high.
863071d4279SBram Moolenaar      */
864071d4279SBram Moolenaar     if (mfp->mf_fd < 0 && need_release && p_uc)
865071d4279SBram Moolenaar     {
866071d4279SBram Moolenaar 	/* find for which buffer this memfile is */
867071d4279SBram Moolenaar 	for (buf = firstbuf; buf != NULL; buf = buf->b_next)
868071d4279SBram Moolenaar 	    if (buf->b_ml.ml_mfp == mfp)
869071d4279SBram Moolenaar 		break;
870071d4279SBram Moolenaar 	if (buf != NULL && buf->b_may_swap)
871071d4279SBram Moolenaar 	    ml_open_file(buf);
872071d4279SBram Moolenaar     }
873071d4279SBram Moolenaar 
874071d4279SBram Moolenaar     /*
875071d4279SBram Moolenaar      * don't release a block if
876071d4279SBram Moolenaar      *	there is no file for this memfile
877071d4279SBram Moolenaar      * or
878071d4279SBram Moolenaar      *	the number of blocks for this memfile is lower than the maximum
879071d4279SBram Moolenaar      *	  and
880071d4279SBram Moolenaar      *	total memory used is not up to 'maxmemtot'
881071d4279SBram Moolenaar      */
882071d4279SBram Moolenaar     if (mfp->mf_fd < 0 || !need_release)
883071d4279SBram Moolenaar 	return NULL;
884071d4279SBram Moolenaar 
885071d4279SBram Moolenaar     for (hp = mfp->mf_used_last; hp != NULL; hp = hp->bh_prev)
886071d4279SBram Moolenaar 	if (!(hp->bh_flags & BH_LOCKED))
887071d4279SBram Moolenaar 	    break;
888071d4279SBram Moolenaar     if (hp == NULL)	/* not a single one that can be released */
889071d4279SBram Moolenaar 	return NULL;
890071d4279SBram Moolenaar 
891071d4279SBram Moolenaar     /*
892071d4279SBram Moolenaar      * If the block is dirty, write it.
893071d4279SBram Moolenaar      * If the write fails we don't free it.
894071d4279SBram Moolenaar      */
895071d4279SBram Moolenaar     if ((hp->bh_flags & BH_DIRTY) && mf_write(mfp, hp) == FAIL)
896071d4279SBram Moolenaar 	return NULL;
897071d4279SBram Moolenaar 
898071d4279SBram Moolenaar     mf_rem_used(mfp, hp);
899071d4279SBram Moolenaar     mf_rem_hash(mfp, hp);
900071d4279SBram Moolenaar 
901071d4279SBram Moolenaar     /*
902071d4279SBram Moolenaar      * If a bhdr_T is returned, make sure that the page_count of bh_data is
903071d4279SBram Moolenaar      * right
904071d4279SBram Moolenaar      */
905071d4279SBram Moolenaar     if (hp->bh_page_count != page_count)
906071d4279SBram Moolenaar     {
907071d4279SBram Moolenaar 	vim_free(hp->bh_data);
908071d4279SBram Moolenaar 	if ((hp->bh_data = alloc(mfp->mf_page_size * page_count)) == NULL)
909071d4279SBram Moolenaar 	{
910071d4279SBram Moolenaar 	    vim_free(hp);
911071d4279SBram Moolenaar 	    return NULL;
912071d4279SBram Moolenaar 	}
913071d4279SBram Moolenaar 	hp->bh_page_count = page_count;
914071d4279SBram Moolenaar     }
915071d4279SBram Moolenaar     return hp;
916071d4279SBram Moolenaar }
917071d4279SBram Moolenaar 
918071d4279SBram Moolenaar /*
919071d4279SBram Moolenaar  * release as many blocks as possible
920071d4279SBram Moolenaar  * Used in case of out of memory
921071d4279SBram Moolenaar  *
922071d4279SBram Moolenaar  * return TRUE if any memory was released
923071d4279SBram Moolenaar  */
924071d4279SBram Moolenaar     int
925071d4279SBram Moolenaar mf_release_all()
926071d4279SBram Moolenaar {
927071d4279SBram Moolenaar     buf_T	*buf;
928071d4279SBram Moolenaar     memfile_T	*mfp;
929071d4279SBram Moolenaar     bhdr_T	*hp;
930071d4279SBram Moolenaar     int		retval = FALSE;
931071d4279SBram Moolenaar 
932071d4279SBram Moolenaar     for (buf = firstbuf; buf != NULL; buf = buf->b_next)
933071d4279SBram Moolenaar     {
934071d4279SBram Moolenaar 	mfp = buf->b_ml.ml_mfp;
935071d4279SBram Moolenaar 	if (mfp != NULL)
936071d4279SBram Moolenaar 	{
937071d4279SBram Moolenaar 	    /* If no swap file yet, may open one */
938071d4279SBram Moolenaar 	    if (mfp->mf_fd < 0 && buf->b_may_swap)
939071d4279SBram Moolenaar 		ml_open_file(buf);
940071d4279SBram Moolenaar 
941071d4279SBram Moolenaar 	    /* only if there is a swapfile */
942071d4279SBram Moolenaar 	    if (mfp->mf_fd >= 0)
943071d4279SBram Moolenaar 	    {
944071d4279SBram Moolenaar 		for (hp = mfp->mf_used_last; hp != NULL; )
945071d4279SBram Moolenaar 		{
946071d4279SBram Moolenaar 		    if (!(hp->bh_flags & BH_LOCKED)
947071d4279SBram Moolenaar 			    && (!(hp->bh_flags & BH_DIRTY)
948071d4279SBram Moolenaar 				|| mf_write(mfp, hp) != FAIL))
949071d4279SBram Moolenaar 		    {
950071d4279SBram Moolenaar 			mf_rem_used(mfp, hp);
951071d4279SBram Moolenaar 			mf_rem_hash(mfp, hp);
952071d4279SBram Moolenaar 			mf_free_bhdr(hp);
953071d4279SBram Moolenaar 			hp = mfp->mf_used_last;	/* re-start, list was changed */
954071d4279SBram Moolenaar 			retval = TRUE;
955071d4279SBram Moolenaar 		    }
956071d4279SBram Moolenaar 		    else
957071d4279SBram Moolenaar 			hp = hp->bh_prev;
958071d4279SBram Moolenaar 		}
959071d4279SBram Moolenaar 	    }
960071d4279SBram Moolenaar 	}
961071d4279SBram Moolenaar     }
962071d4279SBram Moolenaar     return retval;
963071d4279SBram Moolenaar }
964071d4279SBram Moolenaar 
965071d4279SBram Moolenaar /*
966071d4279SBram Moolenaar  * Allocate a block header and a block of memory for it
967071d4279SBram Moolenaar  */
968071d4279SBram Moolenaar     static bhdr_T *
969071d4279SBram Moolenaar mf_alloc_bhdr(mfp, page_count)
970071d4279SBram Moolenaar     memfile_T	*mfp;
971071d4279SBram Moolenaar     int		page_count;
972071d4279SBram Moolenaar {
973071d4279SBram Moolenaar     bhdr_T	*hp;
974071d4279SBram Moolenaar 
975071d4279SBram Moolenaar     if ((hp = (bhdr_T *)alloc((unsigned)sizeof(bhdr_T))) != NULL)
976071d4279SBram Moolenaar     {
977071d4279SBram Moolenaar 	if ((hp->bh_data = (char_u *)alloc(mfp->mf_page_size * page_count))
978071d4279SBram Moolenaar 								      == NULL)
979071d4279SBram Moolenaar 	{
980071d4279SBram Moolenaar 	    vim_free(hp);	    /* not enough memory */
981071d4279SBram Moolenaar 	    return NULL;
982071d4279SBram Moolenaar 	}
983071d4279SBram Moolenaar 	hp->bh_page_count = page_count;
984071d4279SBram Moolenaar     }
985071d4279SBram Moolenaar     return hp;
986071d4279SBram Moolenaar }
987071d4279SBram Moolenaar 
988071d4279SBram Moolenaar /*
989071d4279SBram Moolenaar  * Free a block header and the block of memory for it
990071d4279SBram Moolenaar  */
991071d4279SBram Moolenaar     static void
992071d4279SBram Moolenaar mf_free_bhdr(hp)
993071d4279SBram Moolenaar     bhdr_T	*hp;
994071d4279SBram Moolenaar {
995071d4279SBram Moolenaar     vim_free(hp->bh_data);
996071d4279SBram Moolenaar     vim_free(hp);
997071d4279SBram Moolenaar }
998071d4279SBram Moolenaar 
999071d4279SBram Moolenaar /*
1000071d4279SBram Moolenaar  * insert entry *hp in the free list
1001071d4279SBram Moolenaar  */
1002071d4279SBram Moolenaar     static void
1003071d4279SBram Moolenaar mf_ins_free(mfp, hp)
1004071d4279SBram Moolenaar     memfile_T	*mfp;
1005071d4279SBram Moolenaar     bhdr_T	*hp;
1006071d4279SBram Moolenaar {
1007071d4279SBram Moolenaar     hp->bh_next = mfp->mf_free_first;
1008071d4279SBram Moolenaar     mfp->mf_free_first = hp;
1009071d4279SBram Moolenaar }
1010071d4279SBram Moolenaar 
1011071d4279SBram Moolenaar /*
1012071d4279SBram Moolenaar  * remove the first entry from the free list and return a pointer to it
1013071d4279SBram Moolenaar  * Note: caller must check that mfp->mf_free_first is not NULL!
1014071d4279SBram Moolenaar  */
1015071d4279SBram Moolenaar     static bhdr_T *
1016071d4279SBram Moolenaar mf_rem_free(mfp)
1017071d4279SBram Moolenaar     memfile_T	*mfp;
1018071d4279SBram Moolenaar {
1019071d4279SBram Moolenaar     bhdr_T	*hp;
1020071d4279SBram Moolenaar 
1021071d4279SBram Moolenaar     hp = mfp->mf_free_first;
1022071d4279SBram Moolenaar     mfp->mf_free_first = hp->bh_next;
1023071d4279SBram Moolenaar     return hp;
1024071d4279SBram Moolenaar }
1025071d4279SBram Moolenaar 
1026071d4279SBram Moolenaar /*
1027071d4279SBram Moolenaar  * read a block from disk
1028071d4279SBram Moolenaar  *
1029071d4279SBram Moolenaar  * Return FAIL for failure, OK otherwise
1030071d4279SBram Moolenaar  */
1031071d4279SBram Moolenaar     static int
1032071d4279SBram Moolenaar mf_read(mfp, hp)
1033071d4279SBram Moolenaar     memfile_T	*mfp;
1034071d4279SBram Moolenaar     bhdr_T	*hp;
1035071d4279SBram Moolenaar {
1036071d4279SBram Moolenaar     off_t	offset;
1037071d4279SBram Moolenaar     unsigned	page_size;
1038071d4279SBram Moolenaar     unsigned	size;
1039071d4279SBram Moolenaar 
1040071d4279SBram Moolenaar     if (mfp->mf_fd < 0)	    /* there is no file, can't read */
1041071d4279SBram Moolenaar 	return FAIL;
1042071d4279SBram Moolenaar 
1043071d4279SBram Moolenaar     page_size = mfp->mf_page_size;
1044071d4279SBram Moolenaar     offset = (off_t)page_size * hp->bh_bnum;
1045071d4279SBram Moolenaar     size = page_size * hp->bh_page_count;
1046071d4279SBram Moolenaar     if (lseek(mfp->mf_fd, offset, SEEK_SET) != offset)
1047071d4279SBram Moolenaar     {
10483f2d9814SBram Moolenaar 	PERROR(_("E294: Seek error in swap file read"));
1049071d4279SBram Moolenaar 	return FAIL;
1050071d4279SBram Moolenaar     }
1051071d4279SBram Moolenaar     if ((unsigned)vim_read(mfp->mf_fd, hp->bh_data, size) != size)
1052071d4279SBram Moolenaar     {
10533f2d9814SBram Moolenaar 	PERROR(_("E295: Read error in swap file"));
1054071d4279SBram Moolenaar 	return FAIL;
1055071d4279SBram Moolenaar     }
1056071d4279SBram Moolenaar     return OK;
1057071d4279SBram Moolenaar }
1058071d4279SBram Moolenaar 
1059071d4279SBram Moolenaar /*
1060071d4279SBram Moolenaar  * write a block to disk
1061071d4279SBram Moolenaar  *
1062071d4279SBram Moolenaar  * Return FAIL for failure, OK otherwise
1063071d4279SBram Moolenaar  */
1064071d4279SBram Moolenaar     static int
1065071d4279SBram Moolenaar mf_write(mfp, hp)
1066071d4279SBram Moolenaar     memfile_T	*mfp;
1067071d4279SBram Moolenaar     bhdr_T	*hp;
1068071d4279SBram Moolenaar {
1069071d4279SBram Moolenaar     off_t	offset;	    /* offset in the file */
1070071d4279SBram Moolenaar     blocknr_T	nr;	    /* block nr which is being written */
1071071d4279SBram Moolenaar     bhdr_T	*hp2;
1072071d4279SBram Moolenaar     unsigned	page_size;  /* number of bytes in a page */
1073071d4279SBram Moolenaar     unsigned	page_count; /* number of pages written */
1074071d4279SBram Moolenaar     unsigned	size;	    /* number of bytes written */
1075071d4279SBram Moolenaar 
1076071d4279SBram Moolenaar     if (mfp->mf_fd < 0)	    /* there is no file, can't write */
1077071d4279SBram Moolenaar 	return FAIL;
1078071d4279SBram Moolenaar 
1079071d4279SBram Moolenaar     if (hp->bh_bnum < 0)	/* must assign file block number */
1080071d4279SBram Moolenaar 	if (mf_trans_add(mfp, hp) == FAIL)
1081071d4279SBram Moolenaar 	    return FAIL;
1082071d4279SBram Moolenaar 
1083071d4279SBram Moolenaar     page_size = mfp->mf_page_size;
1084071d4279SBram Moolenaar 
1085071d4279SBram Moolenaar     /*
1086071d4279SBram Moolenaar      * We don't want gaps in the file. Write the blocks in front of *hp
1087071d4279SBram Moolenaar      * to extend the file.
1088071d4279SBram Moolenaar      * If block 'mf_infile_count' is not in the hash list, it has been
1089071d4279SBram Moolenaar      * freed. Fill the space in the file with data from the current block.
1090071d4279SBram Moolenaar      */
1091071d4279SBram Moolenaar     for (;;)
1092071d4279SBram Moolenaar     {
1093071d4279SBram Moolenaar 	nr = hp->bh_bnum;
1094071d4279SBram Moolenaar 	if (nr > mfp->mf_infile_count)		/* beyond end of file */
1095071d4279SBram Moolenaar 	{
1096071d4279SBram Moolenaar 	    nr = mfp->mf_infile_count;
1097071d4279SBram Moolenaar 	    hp2 = mf_find_hash(mfp, nr);	/* NULL catched below */
1098071d4279SBram Moolenaar 	}
1099071d4279SBram Moolenaar 	else
1100071d4279SBram Moolenaar 	    hp2 = hp;
1101071d4279SBram Moolenaar 
1102071d4279SBram Moolenaar 	offset = (off_t)page_size * nr;
1103071d4279SBram Moolenaar 	if (lseek(mfp->mf_fd, offset, SEEK_SET) != offset)
1104071d4279SBram Moolenaar 	{
11053f2d9814SBram Moolenaar 	    PERROR(_("E296: Seek error in swap file write"));
1106071d4279SBram Moolenaar 	    return FAIL;
1107071d4279SBram Moolenaar 	}
1108071d4279SBram Moolenaar 	if (hp2 == NULL)	    /* freed block, fill with dummy data */
1109071d4279SBram Moolenaar 	    page_count = 1;
1110071d4279SBram Moolenaar 	else
1111071d4279SBram Moolenaar 	    page_count = hp2->bh_page_count;
1112071d4279SBram Moolenaar 	size = page_size * page_count;
1113071d4279SBram Moolenaar 	if ((unsigned)vim_write(mfp->mf_fd,
1114071d4279SBram Moolenaar 	     (hp2 == NULL ? hp : hp2)->bh_data, size) != size)
1115071d4279SBram Moolenaar 	{
1116071d4279SBram Moolenaar 	    /*
1117071d4279SBram Moolenaar 	     * Avoid repeating the error message, this mostly happens when the
111849325942SBram Moolenaar 	     * disk is full. We give the message again only after a successful
1119071d4279SBram Moolenaar 	     * write or when hitting a key. We keep on trying, in case some
1120071d4279SBram Moolenaar 	     * space becomes available.
1121071d4279SBram Moolenaar 	     */
1122071d4279SBram Moolenaar 	    if (!did_swapwrite_msg)
1123071d4279SBram Moolenaar 		EMSG(_("E297: Write error in swap file"));
1124071d4279SBram Moolenaar 	    did_swapwrite_msg = TRUE;
1125071d4279SBram Moolenaar 	    return FAIL;
1126071d4279SBram Moolenaar 	}
1127071d4279SBram Moolenaar 	did_swapwrite_msg = FALSE;
1128071d4279SBram Moolenaar 	if (hp2 != NULL)		    /* written a non-dummy block */
1129071d4279SBram Moolenaar 	    hp2->bh_flags &= ~BH_DIRTY;
1130071d4279SBram Moolenaar 					    /* appended to the file */
1131071d4279SBram Moolenaar 	if (nr + (blocknr_T)page_count > mfp->mf_infile_count)
1132071d4279SBram Moolenaar 	    mfp->mf_infile_count = nr + page_count;
1133071d4279SBram Moolenaar 	if (nr == hp->bh_bnum)		    /* written the desired block */
1134071d4279SBram Moolenaar 	    break;
1135071d4279SBram Moolenaar     }
1136071d4279SBram Moolenaar     return OK;
1137071d4279SBram Moolenaar }
1138071d4279SBram Moolenaar 
1139071d4279SBram Moolenaar /*
1140071d4279SBram Moolenaar  * Make block number for *hp positive and add it to the translation list
1141071d4279SBram Moolenaar  *
1142071d4279SBram Moolenaar  * Return FAIL for failure, OK otherwise
1143071d4279SBram Moolenaar  */
1144071d4279SBram Moolenaar     static int
1145071d4279SBram Moolenaar mf_trans_add(mfp, hp)
1146071d4279SBram Moolenaar     memfile_T	*mfp;
1147071d4279SBram Moolenaar     bhdr_T	*hp;
1148071d4279SBram Moolenaar {
1149071d4279SBram Moolenaar     bhdr_T	*freep;
1150071d4279SBram Moolenaar     blocknr_T	new_bnum;
1151071d4279SBram Moolenaar     int		hash;
1152071d4279SBram Moolenaar     NR_TRANS	*np;
1153071d4279SBram Moolenaar     int		page_count;
1154071d4279SBram Moolenaar 
1155071d4279SBram Moolenaar     if (hp->bh_bnum >= 0)		    /* it's already positive */
1156071d4279SBram Moolenaar 	return OK;
1157071d4279SBram Moolenaar 
1158071d4279SBram Moolenaar     if ((np = (NR_TRANS *)alloc((unsigned)sizeof(NR_TRANS))) == NULL)
1159071d4279SBram Moolenaar 	return FAIL;
1160071d4279SBram Moolenaar 
1161071d4279SBram Moolenaar /*
1162071d4279SBram Moolenaar  * get a new number for the block.
1163071d4279SBram Moolenaar  * If the first item in the free list has sufficient pages, use its number
1164071d4279SBram Moolenaar  * Otherwise use mf_blocknr_max.
1165071d4279SBram Moolenaar  */
1166071d4279SBram Moolenaar     freep = mfp->mf_free_first;
1167071d4279SBram Moolenaar     page_count = hp->bh_page_count;
1168071d4279SBram Moolenaar     if (freep != NULL && freep->bh_page_count >= page_count)
1169071d4279SBram Moolenaar     {
1170071d4279SBram Moolenaar 	new_bnum = freep->bh_bnum;
1171071d4279SBram Moolenaar 	/*
1172071d4279SBram Moolenaar 	 * If the page count of the free block was larger, recude it.
1173071d4279SBram Moolenaar 	 * If the page count matches, remove the block from the free list
1174071d4279SBram Moolenaar 	 */
1175071d4279SBram Moolenaar 	if (freep->bh_page_count > page_count)
1176071d4279SBram Moolenaar 	{
1177071d4279SBram Moolenaar 	    freep->bh_bnum += page_count;
1178071d4279SBram Moolenaar 	    freep->bh_page_count -= page_count;
1179071d4279SBram Moolenaar 	}
1180071d4279SBram Moolenaar 	else
1181071d4279SBram Moolenaar 	{
1182071d4279SBram Moolenaar 	    freep = mf_rem_free(mfp);
1183071d4279SBram Moolenaar 	    vim_free(freep);
1184071d4279SBram Moolenaar 	}
1185071d4279SBram Moolenaar     }
1186071d4279SBram Moolenaar     else
1187071d4279SBram Moolenaar     {
1188071d4279SBram Moolenaar 	new_bnum = mfp->mf_blocknr_max;
1189071d4279SBram Moolenaar 	mfp->mf_blocknr_max += page_count;
1190071d4279SBram Moolenaar     }
1191071d4279SBram Moolenaar 
1192071d4279SBram Moolenaar     np->nt_old_bnum = hp->bh_bnum;	    /* adjust number */
1193071d4279SBram Moolenaar     np->nt_new_bnum = new_bnum;
1194071d4279SBram Moolenaar 
1195071d4279SBram Moolenaar     mf_rem_hash(mfp, hp);		    /* remove from old hash list */
1196071d4279SBram Moolenaar     hp->bh_bnum = new_bnum;
1197071d4279SBram Moolenaar     mf_ins_hash(mfp, hp);		    /* insert in new hash list */
1198071d4279SBram Moolenaar 
1199071d4279SBram Moolenaar     hash = MEMHASH(np->nt_old_bnum);	    /* insert in trans list */
1200071d4279SBram Moolenaar     np->nt_next = mfp->mf_trans[hash];
1201071d4279SBram Moolenaar     mfp->mf_trans[hash] = np;
1202071d4279SBram Moolenaar     if (np->nt_next != NULL)
1203071d4279SBram Moolenaar 	np->nt_next->nt_prev = np;
1204071d4279SBram Moolenaar     np->nt_prev = NULL;
1205071d4279SBram Moolenaar 
1206071d4279SBram Moolenaar     return OK;
1207071d4279SBram Moolenaar }
1208071d4279SBram Moolenaar 
1209071d4279SBram Moolenaar /*
121049325942SBram Moolenaar  * Lookup a translation from the trans lists and delete the entry
1211071d4279SBram Moolenaar  *
1212071d4279SBram Moolenaar  * Return the positive new number when found, the old number when not found
1213071d4279SBram Moolenaar  */
1214071d4279SBram Moolenaar     blocknr_T
1215071d4279SBram Moolenaar mf_trans_del(mfp, old_nr)
1216071d4279SBram Moolenaar     memfile_T	*mfp;
1217071d4279SBram Moolenaar     blocknr_T	old_nr;
1218071d4279SBram Moolenaar {
1219071d4279SBram Moolenaar     int		hash;
1220071d4279SBram Moolenaar     NR_TRANS	*np;
1221071d4279SBram Moolenaar     blocknr_T	new_bnum;
1222071d4279SBram Moolenaar 
1223071d4279SBram Moolenaar     hash = MEMHASH(old_nr);
1224071d4279SBram Moolenaar     for (np = mfp->mf_trans[hash]; np != NULL; np = np->nt_next)
1225071d4279SBram Moolenaar 	if (np->nt_old_bnum == old_nr)
1226071d4279SBram Moolenaar 	    break;
1227071d4279SBram Moolenaar     if (np == NULL)		/* not found */
1228071d4279SBram Moolenaar 	return old_nr;
1229071d4279SBram Moolenaar 
1230071d4279SBram Moolenaar     mfp->mf_neg_count--;
1231071d4279SBram Moolenaar     new_bnum = np->nt_new_bnum;
1232071d4279SBram Moolenaar     if (np->nt_prev != NULL)		/* remove entry from the trans list */
1233071d4279SBram Moolenaar 	np->nt_prev->nt_next = np->nt_next;
1234071d4279SBram Moolenaar     else
1235071d4279SBram Moolenaar 	mfp->mf_trans[hash] = np->nt_next;
1236071d4279SBram Moolenaar     if (np->nt_next != NULL)
1237071d4279SBram Moolenaar 	np->nt_next->nt_prev = np->nt_prev;
1238071d4279SBram Moolenaar     vim_free(np);
1239071d4279SBram Moolenaar 
1240071d4279SBram Moolenaar     return new_bnum;
1241071d4279SBram Moolenaar }
1242071d4279SBram Moolenaar 
1243071d4279SBram Moolenaar /*
1244071d4279SBram Moolenaar  * Set mfp->mf_ffname according to mfp->mf_fname and some other things.
1245071d4279SBram Moolenaar  * Only called when creating or renaming the swapfile.	Either way it's a new
1246071d4279SBram Moolenaar  * name so we must work out the full path name.
1247071d4279SBram Moolenaar  */
1248071d4279SBram Moolenaar     void
1249071d4279SBram Moolenaar mf_set_ffname(mfp)
1250071d4279SBram Moolenaar     memfile_T	*mfp;
1251071d4279SBram Moolenaar {
1252071d4279SBram Moolenaar     mfp->mf_ffname = FullName_save(mfp->mf_fname, FALSE);
1253071d4279SBram Moolenaar }
1254071d4279SBram Moolenaar 
1255071d4279SBram Moolenaar /*
1256071d4279SBram Moolenaar  * Make the name of the file used for the memfile a full path.
1257071d4279SBram Moolenaar  * Used before doing a :cd
1258071d4279SBram Moolenaar  */
1259071d4279SBram Moolenaar     void
1260071d4279SBram Moolenaar mf_fullname(mfp)
1261071d4279SBram Moolenaar     memfile_T	*mfp;
1262071d4279SBram Moolenaar {
1263071d4279SBram Moolenaar     if (mfp != NULL && mfp->mf_fname != NULL && mfp->mf_ffname != NULL)
1264071d4279SBram Moolenaar     {
1265071d4279SBram Moolenaar 	vim_free(mfp->mf_fname);
1266071d4279SBram Moolenaar 	mfp->mf_fname = mfp->mf_ffname;
1267071d4279SBram Moolenaar 	mfp->mf_ffname = NULL;
1268071d4279SBram Moolenaar     }
1269071d4279SBram Moolenaar }
1270071d4279SBram Moolenaar 
1271071d4279SBram Moolenaar /*
1272071d4279SBram Moolenaar  * return TRUE if there are any translations pending for 'mfp'
1273071d4279SBram Moolenaar  */
1274071d4279SBram Moolenaar     int
1275071d4279SBram Moolenaar mf_need_trans(mfp)
1276071d4279SBram Moolenaar     memfile_T	*mfp;
1277071d4279SBram Moolenaar {
1278071d4279SBram Moolenaar     return (mfp->mf_fname != NULL && mfp->mf_neg_count > 0);
1279071d4279SBram Moolenaar }
1280071d4279SBram Moolenaar 
1281071d4279SBram Moolenaar /*
1282071d4279SBram Moolenaar  * Open a swap file for a memfile.
1283071d4279SBram Moolenaar  * The "fname" must be in allocated memory, and is consumed (also when an
1284071d4279SBram Moolenaar  * error occurs).
1285071d4279SBram Moolenaar  */
1286071d4279SBram Moolenaar     static void
1287071d4279SBram Moolenaar mf_do_open(mfp, fname, flags)
1288071d4279SBram Moolenaar     memfile_T	*mfp;
1289071d4279SBram Moolenaar     char_u	*fname;
1290071d4279SBram Moolenaar     int		flags;		/* flags for open() */
1291071d4279SBram Moolenaar {
1292071d4279SBram Moolenaar #ifdef HAVE_LSTAT
1293071d4279SBram Moolenaar     struct stat sb;
1294071d4279SBram Moolenaar #endif
1295071d4279SBram Moolenaar 
1296071d4279SBram Moolenaar     mfp->mf_fname = fname;
1297071d4279SBram Moolenaar 
1298071d4279SBram Moolenaar     /*
1299071d4279SBram Moolenaar      * Get the full path name before the open, because this is
1300071d4279SBram Moolenaar      * not possible after the open on the Amiga.
1301071d4279SBram Moolenaar      * fname cannot be NameBuff, because it must have been allocated.
1302071d4279SBram Moolenaar      */
1303071d4279SBram Moolenaar     mf_set_ffname(mfp);
1304071d4279SBram Moolenaar #if defined(MSDOS) || defined(MSWIN) || defined(RISCOS)
1305071d4279SBram Moolenaar     /*
1306071d4279SBram Moolenaar      * A ":!cd e:xxx" may change the directory without us knowning, use the
1307071d4279SBram Moolenaar      * full pathname always.  Careful: This frees fname!
1308071d4279SBram Moolenaar      */
1309071d4279SBram Moolenaar     mf_fullname(mfp);
1310071d4279SBram Moolenaar #endif
1311071d4279SBram Moolenaar 
1312071d4279SBram Moolenaar #ifdef HAVE_LSTAT
1313071d4279SBram Moolenaar     /*
1314071d4279SBram Moolenaar      * Extra security check: When creating a swap file it really shouldn't
1315071d4279SBram Moolenaar      * exist yet.  If there is a symbolic link, this is most likely an attack.
1316071d4279SBram Moolenaar      */
1317071d4279SBram Moolenaar     if ((flags & O_CREAT) && mch_lstat((char *)mfp->mf_fname, &sb) >= 0)
1318071d4279SBram Moolenaar     {
1319071d4279SBram Moolenaar 	mfp->mf_fd = -1;
1320071d4279SBram Moolenaar 	EMSG(_("E300: Swap file already exists (symlink attack?)"));
1321071d4279SBram Moolenaar     }
1322071d4279SBram Moolenaar     else
1323071d4279SBram Moolenaar #endif
1324071d4279SBram Moolenaar     {
1325071d4279SBram Moolenaar 	/*
1326071d4279SBram Moolenaar 	 * try to open the file
1327071d4279SBram Moolenaar 	 */
1328a5792f58SBram Moolenaar 	flags |= O_EXTRA | O_NOFOLLOW;
1329071d4279SBram Moolenaar #ifdef WIN32
1330071d4279SBram Moolenaar 	/* Prevent handle inheritance that cause problems with Cscope
1331071d4279SBram Moolenaar 	 * (swap file may not be deleted if cscope connection was open after
1332071d4279SBram Moolenaar 	 * the file) */
1333071d4279SBram Moolenaar 	flags |= O_NOINHERIT;
1334071d4279SBram Moolenaar #endif
1335071d4279SBram Moolenaar 	mfp->mf_fd = mch_open_rw((char *)mfp->mf_fname, flags);
1336071d4279SBram Moolenaar     }
1337071d4279SBram Moolenaar 
1338071d4279SBram Moolenaar     /*
1339071d4279SBram Moolenaar      * If the file cannot be opened, use memory only
1340071d4279SBram Moolenaar      */
1341071d4279SBram Moolenaar     if (mfp->mf_fd < 0)
1342071d4279SBram Moolenaar     {
1343071d4279SBram Moolenaar 	vim_free(mfp->mf_fname);
1344071d4279SBram Moolenaar 	vim_free(mfp->mf_ffname);
1345071d4279SBram Moolenaar 	mfp->mf_fname = NULL;
1346071d4279SBram Moolenaar 	mfp->mf_ffname = NULL;
1347071d4279SBram Moolenaar     }
1348071d4279SBram Moolenaar     else
1349*588ebeb7SBram Moolenaar     {
1350*588ebeb7SBram Moolenaar #ifdef HAVE_SELINUX
1351*588ebeb7SBram Moolenaar 	mch_copy_sec(fname, mfp->mf_fname);
1352*588ebeb7SBram Moolenaar #endif
1353071d4279SBram Moolenaar 	mch_hide(mfp->mf_fname);    /* try setting the 'hidden' flag */
1354071d4279SBram Moolenaar     }
1355*588ebeb7SBram Moolenaar }
1356