memfile.c
上传用户:gddssl
上传日期:2007-01-06
资源大小:1003k
文件大小:32k
源码类别:

编辑器/阅读器

开发平台:

DOS

  1. /* vi:set ts=8 sts=4 sw=4:
  2.  *
  3.  * VIM - Vi IMproved by Bram Moolenaar
  4.  *
  5.  * Do ":help uganda"  in Vim to read copying and usage conditions.
  6.  * Do ":help credits" in Vim to see a list of people who contributed.
  7.  */
  8. /*
  9.  * memfile.c: Contains the functions for handling blocks of memory which can
  10.  * be stored in a file. This is the implementation of a sort of virtual memory.
  11.  *
  12.  * A memfile consists of a sequence of blocks. The blocks numbered from 0
  13.  * upwards have been assigned a place in the actual file. The block number
  14.  * is equal to the page number in the file. The
  15.  * blocks with negative numbers are currently in memory only. They can be
  16.  * assigned a place in the file when too much memory is being used. At that
  17.  * moment they get a new, positive, number. A list is used for translation of
  18.  * negative to positive numbers.
  19.  *
  20.  * The size of a block is a multiple of a page size, normally the page size of
  21.  * the device the file is on. Most blocks are 1 page long. A Block of multiple
  22.  * pages is used for a line that does not fit in a single page.
  23.  *
  24.  * Each block can be in memory and/or in a file. The block stays in memory
  25.  * as long as it is locked. If it is no longer locked it can be swapped out to
  26.  * the file. It is only written to the file if it has been changed.
  27.  *
  28.  * Under normal operation the file is created when opening the memory file and
  29.  * deleted when closing the memory file. Only with recovery an existing memory
  30.  * file is opened.
  31.  */
  32. #if defined MSDOS || defined WIN32
  33. # include <io.h> /* for lseek(), must be before vim.h */
  34. #endif
  35. #include "vim.h"
  36. #ifdef HAVE_FCNTL_H
  37. # include <fcntl.h>
  38. #endif
  39. /*
  40.  * Some systems have the page size in statfs.f_bsize, some in stat.st_blksize
  41.  */
  42. #ifdef HAVE_ST_BLKSIZE
  43. # define STATFS stat
  44. # define F_BSIZE st_blksize
  45. # define fstatfs(fd, buf, len, nul) fstat((fd), (buf))
  46. #else
  47. # ifdef HAVE_SYS_STATFS_H
  48. #  include <sys/statfs.h>
  49. #  define STATFS statfs
  50. #  define F_BSIZE f_bsize
  51. #  ifdef __MINT__ /* do we still need this? */
  52. #   define fstatfs(fd, buf, len, nul) fstat((fd), (buf))
  53. #  endif
  54. # endif
  55. #endif
  56. /*
  57.  * for Amiga Dos 2.0x we use Flush
  58.  */
  59. #ifdef AMIGA
  60. # ifndef NO_ARP
  61. extern int dos2; /* this is in os_amiga.c */
  62. # endif
  63. # ifdef SASC
  64. #  include <proto/dos.h>
  65. #  include <ios1.h> /* for chkufb() */
  66. # endif
  67. #endif
  68. #define MEMFILE_PAGE_SIZE 4096 /* default page size */
  69. static long_u total_mem_used = 0; /* total memory used for memfiles */
  70. static int dont_release = FALSE; /* don't release blocks */
  71. static void mf_ins_hash __ARGS((MEMFILE *, BHDR *));
  72. static void mf_rem_hash __ARGS((MEMFILE *, BHDR *));
  73. static BHDR *mf_find_hash __ARGS((MEMFILE *, blocknr_t));
  74. static void mf_ins_used __ARGS((MEMFILE *, BHDR *));
  75. static void mf_rem_used __ARGS((MEMFILE *, BHDR *));
  76. static BHDR *mf_release __ARGS((MEMFILE *, int));
  77. static BHDR *mf_alloc_bhdr __ARGS((MEMFILE *, int));
  78. static void mf_free_bhdr __ARGS((BHDR *));
  79. static void mf_ins_free __ARGS((MEMFILE *, BHDR *));
  80. static BHDR *mf_rem_free __ARGS((MEMFILE *));
  81. static int  mf_read __ARGS((MEMFILE *, BHDR *));
  82. static int  mf_write __ARGS((MEMFILE *, BHDR *));
  83. static int  mf_trans_add __ARGS((MEMFILE *, BHDR *));
  84. static void mf_do_open __ARGS((MEMFILE *, char_u *, int));
  85. /*
  86.  * The functions for using a memfile:
  87.  *
  88.  * mf_open()     open a new or existing memfile
  89.  * mf_open_file()   open a swap file for an existing memfile
  90.  * mf_close()     close (and delete) a memfile
  91.  * mf_new()     create a new block in a memfile and lock it
  92.  * mf_get()     get an existing block and lock it
  93.  * mf_put()     unlock a block, may be marked for writing
  94.  * mf_free()     remove a block
  95.  * mf_sync()     sync changed parts of memfile to disk
  96.  * mf_release_all() release as much memory as possible
  97.  * mf_trans_del()   may translate negative to positive block number
  98.  * mf_fullname()    make file name full path (use before first :cd)
  99.  */
  100. /*
  101.  * mf_open: open an existing or new memory block file
  102.  *
  103.  *  fname: name of file to use (NULL means no file at all)
  104.  * Note: fname must have been allocated, it is not copied!
  105.  * If opening the file fails, fname is freed.
  106.  *  trunc_file:     if TRUE: file should be truncated when opening
  107.  *
  108.  *  If fname != NULL and file cannot be opened, fail.
  109.  *
  110.  * return value: identifier for this memory block file.
  111.  */
  112.     MEMFILE *
  113. mf_open(fname, trunc_file)
  114.     char_u  *fname;
  115.     int     trunc_file;
  116. {
  117.     MEMFILE     *mfp;
  118.     int     i;
  119.     off_t     size;
  120. #if defined(STATFS) && defined(UNIX) && !defined(__QNX__)
  121. # define USE_FSTATFS
  122.     struct STATFS   stf;
  123. #endif
  124.     if ((mfp = (MEMFILE *)alloc((unsigned)sizeof(MEMFILE))) == NULL)
  125. return NULL;
  126.     if (fname == NULL)     /* no file for this memfile, use memory only */
  127.     {
  128. mfp->mf_fname = NULL;
  129. mfp->mf_ffname = NULL;
  130. mfp->mf_fd = -1;
  131.     }
  132.     else
  133.     {
  134. mf_do_open(mfp, fname, trunc_file); /* try to open the file */
  135. /* if the file cannot be opened, return here */
  136. if (mfp->mf_fd < 0)
  137. {
  138.     vim_free(mfp);
  139.     return NULL;
  140. }
  141.     }
  142.     mfp->mf_free_first = NULL; /* free list is empty */
  143.     mfp->mf_used_first = NULL; /* used list is empty */
  144.     mfp->mf_used_last = NULL;
  145.     mfp->mf_dirty = FALSE;
  146.     mfp->mf_used_count = 0;
  147.     for (i = 0; i < MEMHASHSIZE; ++i)
  148.     {
  149. mfp->mf_hash[i] = NULL; /* hash lists are empty */
  150. mfp->mf_trans[i] = NULL; /* trans lists are empty */
  151.     }
  152.     mfp->mf_page_size = MEMFILE_PAGE_SIZE;
  153. #ifdef USE_FSTATFS
  154.     /*
  155.      * Try to set the page size equal to the block size of the device.
  156.      * Speeds up I/O a lot.
  157.      * NOTE: minimal block size depends on size of block 0 data! It's not done
  158.      * with a sizeof(), because block 0 is defined in memline.c (Sorry).
  159.      * The maximal block size is arbitrary.
  160.      */
  161.     if (mfp->mf_fd >= 0 &&
  162.     fstatfs(mfp->mf_fd, &stf, sizeof(struct statfs), 0) == 0 &&
  163.     stf.F_BSIZE >= 1048 && stf.F_BSIZE <= 50000)
  164. mfp->mf_page_size = stf.F_BSIZE;
  165. #endif
  166.     if (mfp->mf_fd < 0 || trunc_file ||
  167.  (size = lseek(mfp->mf_fd, (off_t)0L, SEEK_END)) <= 0)
  168. mfp->mf_blocknr_max = 0; /* no file or empty file */
  169.     else
  170. mfp->mf_blocknr_max = size / mfp->mf_page_size;
  171.     mfp->mf_blocknr_min = -1;
  172.     mfp->mf_neg_count = 0;
  173.     mfp->mf_infile_count = mfp->mf_blocknr_max;
  174.     mfp->mf_used_count_max = p_mm * 1024 / mfp->mf_page_size;
  175.     return mfp;
  176. }
  177. /*
  178.  * mf_open_file: open a file for an existing memfile. Used when updatecount
  179.  *  set from 0 to some value.
  180.  *
  181.  *  fname: name of file to use (NULL means no file at all)
  182.  * Note: fname must have been allocated, it is not copied!
  183.  * If opening the file fails, fname is freed.
  184.  *
  185.  * return value: FAIL if file could not be opened, OK otherwise
  186.  */
  187.     int
  188. mf_open_file(mfp, fname)
  189.     MEMFILE *mfp;
  190.     char_u *fname;
  191. {
  192.     mf_do_open(mfp, fname, TRUE); /* try to open the file */
  193.     if (mfp->mf_fd < 0)
  194. return FAIL;
  195.     mfp->mf_dirty = TRUE;
  196.     return OK;
  197. }
  198. /*
  199.  * close a memory file and delete the associated file if 'del_file' is TRUE
  200.  */
  201.     void
  202. mf_close(mfp, del_file)
  203.     MEMFILE *mfp;
  204.     int     del_file;
  205. {
  206.     BHDR *hp, *nextp;
  207.     NR_TRANS *tp, *tpnext;
  208.     int i;
  209.     if (mfp == NULL)     /* safety check */
  210. return;
  211.     if (mfp->mf_fd >= 0)
  212.     {
  213. if (close(mfp->mf_fd) < 0)
  214.     EMSG("Close error on swap file");
  215.     }
  216.     if (del_file && mfp->mf_fname != NULL)
  217. mch_remove(mfp->mf_fname);
  218.     /* free entries in used list */
  219.     for (hp = mfp->mf_used_first; hp != NULL; hp = nextp)
  220.     {
  221. total_mem_used -= hp->bh_page_count * mfp->mf_page_size;
  222. nextp = hp->bh_next;
  223. mf_free_bhdr(hp);
  224.     }
  225.     while (mfp->mf_free_first != NULL)     /* free entries in free list */
  226. vim_free(mf_rem_free(mfp));
  227.     for (i = 0; i < MEMHASHSIZE; ++i)     /* free entries in trans lists */
  228. for (tp = mfp->mf_trans[i]; tp != NULL; tp = tpnext)
  229. {
  230.     tpnext = tp->nt_next;
  231.     vim_free(tp);
  232. }
  233.     vim_free(mfp->mf_fname);
  234.     vim_free(mfp->mf_ffname);
  235.     vim_free(mfp);
  236. }
  237. /*
  238.  * Close the swap file for a memfile.  Used when 'swapfile' is reset.
  239.  */
  240.     void
  241. mf_close_file(buf)
  242.     BUF *buf;
  243. {
  244.     MEMFILE *mfp;
  245.     linenr_t lnum;
  246.     mfp = buf->b_ml.ml_mfp;
  247.     if (mfp == NULL || mfp->mf_fd < 0) /* nothing to close */
  248. return;
  249.     /* get all blocks in memory by accessing all lines (clumsy!) */
  250.     dont_release = TRUE;
  251.     for (lnum = 1; lnum <= buf->b_ml.ml_line_count; ++lnum)
  252. (void)ml_get_buf(buf, lnum, FALSE);
  253.     dont_release = FALSE;
  254.     /* TODO: should check if all blocks are really in core */
  255.     if (close(mfp->mf_fd) < 0) /* close the file */
  256. EMSG("Close error on swap file");
  257.     mfp->mf_fd = -1;
  258.     if (mfp->mf_fname != NULL)
  259.     {
  260. mch_remove(mfp->mf_fname); /* delete the swap file */
  261. vim_free(mfp->mf_fname);
  262. vim_free(mfp->mf_ffname);
  263. mfp->mf_fname = NULL;
  264. mfp->mf_ffname = NULL;
  265.     }
  266. }
  267. /*
  268.  * get a new block
  269.  *
  270.  *   negative: TRUE if negative block number desired (data block)
  271.  */
  272.     BHDR *
  273. mf_new(mfp, negative, page_count)
  274.     MEMFILE *mfp;
  275.     int negative;
  276.     int page_count;
  277. {
  278.     BHDR    *hp;     /* new BHDR */
  279.     BHDR    *freep;     /* first block in free list */
  280.     char_u  *p;
  281.     /*
  282.      * If we reached the maximum size for the used memory blocks, release one
  283.      * If a BHDR is returned, use it and adjust the page_count if necessary.
  284.      */
  285.     hp = mf_release(mfp, page_count);
  286. /*
  287.  * Decide on the number to use:
  288.  * If there is a free block, use its number.
  289.  * Otherwise use mf_block_min for a negative number, mf_block_max for
  290.  * a positive number.
  291.  */
  292.     freep = mfp->mf_free_first;
  293.     if (!negative && freep != NULL && freep->bh_page_count >= page_count)
  294.     {
  295. /*
  296.  * If the block in the free list has more pages, take only the number
  297.  * of pages needed and allocate a new BHDR with data
  298.  *
  299.  * If the number of pages matches and mf_release did not return a BHDR,
  300.  * use the BHDR from the free list and allocate the data
  301.  *
  302.  * If the number of pages matches and mf_release returned a BHDR,
  303.  * just use the number and free the BHDR from the free list
  304.  */
  305. if (freep->bh_page_count > page_count)
  306. {
  307.     if (hp == NULL && (hp = mf_alloc_bhdr(mfp, page_count)) == NULL)
  308. return NULL;
  309.     hp->bh_bnum = freep->bh_bnum;
  310.     freep->bh_bnum += page_count;
  311.     freep->bh_page_count -= page_count;
  312. }
  313. else if (hp == NULL)     /* need to allocate memory for this block */
  314. {
  315.     if ((p = (char_u *)alloc(mfp->mf_page_size * page_count)) == NULL)
  316. return NULL;
  317.     hp = mf_rem_free(mfp);
  318.     hp->bh_data = p;
  319. }
  320. else     /* use the number, remove entry from free list */
  321. {
  322.     freep = mf_rem_free(mfp);
  323.     hp->bh_bnum = freep->bh_bnum;
  324.     vim_free(freep);
  325. }
  326.     }
  327.     else /* get a new number */
  328.     {
  329. if (hp == NULL && (hp = mf_alloc_bhdr(mfp, page_count)) == NULL)
  330.     return NULL;
  331. if (negative)
  332. {
  333.     hp->bh_bnum = mfp->mf_blocknr_min--;
  334.     mfp->mf_neg_count++;
  335. }
  336. else
  337. {
  338.     hp->bh_bnum = mfp->mf_blocknr_max;
  339.     mfp->mf_blocknr_max += page_count;
  340. }
  341.     }
  342.     hp->bh_flags = BH_LOCKED | BH_DIRTY; /* new block is always dirty */
  343.     mfp->mf_dirty = TRUE;
  344.     hp->bh_page_count = page_count;
  345.     mf_ins_used(mfp, hp);
  346.     mf_ins_hash(mfp, hp);
  347.     /*
  348.      * Init the data to all zero, to avoid reading uninitialized data.
  349.      * This also avoids that the passwd file ends up in the swap file!
  350.      */
  351.     (void)vim_memset((char *)(hp->bh_data), 0, (size_t)mfp->mf_page_size);
  352.     return hp;
  353. }
  354. /*
  355.  * get existing block 'nr' with 'page_count' pages
  356.  *
  357.  * Note: The caller should first check a negative nr with mf_trans_del()
  358.  */
  359.     BHDR *
  360. mf_get(mfp, nr, page_count)
  361.     MEMFILE *mfp;
  362.     blocknr_t nr;
  363.     int page_count;
  364. {
  365.     BHDR    *hp;
  366. /* doesn't exist */
  367.     if (nr >= mfp->mf_blocknr_max || nr <= mfp->mf_blocknr_min)
  368. return NULL;
  369.     /*
  370.      * see if it is in the cache
  371.      */
  372.     hp = mf_find_hash(mfp, nr);
  373.     if (hp == NULL) /* not in the hash list */
  374.     {
  375. if (nr < 0 || nr >= mfp->mf_infile_count)   /* can't be in the file */
  376.     return NULL;
  377. /* could check here if the block is in the free list */
  378. /*
  379.  * Check if we need to flush an existing block.
  380.  * If so, use that block.
  381.  * If not, allocate a new block.
  382.  */
  383. hp = mf_release(mfp, page_count);
  384. if (hp == NULL && (hp = mf_alloc_bhdr(mfp, page_count)) == NULL)
  385.     return NULL;
  386. hp->bh_bnum = nr;
  387. hp->bh_flags = 0;
  388. hp->bh_page_count = page_count;
  389. if (mf_read(mfp, hp) == FAIL)     /* cannot read the block! */
  390. {
  391.     mf_free_bhdr(hp);
  392.     return NULL;
  393. }
  394.     }
  395.     else
  396.     {
  397. mf_rem_used(mfp, hp); /* remove from list, insert in front below */
  398. mf_rem_hash(mfp, hp);
  399.     }
  400.     hp->bh_flags |= BH_LOCKED;
  401.     mf_ins_used(mfp, hp); /* put in front of used list */
  402.     mf_ins_hash(mfp, hp); /* put in front of hash list */
  403.     return hp;
  404. }
  405. /*
  406.  * release the block *hp
  407.  *
  408.  *   dirty: Block must be written to file later
  409.  *   infile: Block should be in file (needed for recovery)
  410.  *
  411.  *  no return value, function cannot fail
  412.  */
  413.     void
  414. mf_put(mfp, hp, dirty, infile)
  415.     MEMFILE *mfp;
  416.     BHDR    *hp;
  417.     int     dirty;
  418.     int     infile;
  419. {
  420.     int     flags;
  421.     flags = hp->bh_flags;
  422.     if ((flags & BH_LOCKED) == 0)
  423. EMSG("block was not locked");
  424.     flags &= ~BH_LOCKED;
  425.     if (dirty)
  426.     {
  427. flags |= BH_DIRTY;
  428. mfp->mf_dirty = TRUE;
  429.     }
  430.     hp->bh_flags = flags;
  431.     if (infile)
  432. mf_trans_add(mfp, hp);     /* may translate negative in positive nr */
  433. }
  434. /*
  435.  * block *hp is no longer in used, may put it in the free list of memfile *mfp
  436.  */
  437.     void
  438. mf_free(mfp, hp)
  439.     MEMFILE *mfp;
  440.     BHDR    *hp;
  441. {
  442.     vim_free(hp->bh_data); /* free the memory */
  443.     mf_rem_hash(mfp, hp); /* get *hp out of the hash list */
  444.     mf_rem_used(mfp, hp); /* get *hp out of the used list */
  445.     if (hp->bh_bnum < 0)
  446.     {
  447. vim_free(hp); /* don't want negative numbers in free list */
  448. mfp->mf_neg_count--;
  449.     }
  450.     else
  451. mf_ins_free(mfp, hp); /* put *hp in the free list */
  452. }
  453. /*
  454.  * Sync the memory file *mfp to disk.
  455.  * Flags:
  456.  *  MFS_ALL If not given, blocks with negative numbers are not synced,
  457.  * even when they are dirty!
  458.  *  MFS_STOP Stop syncing when a character becomes available, but sync at
  459.  * least one block.
  460.  *  MFS_FLUSH Make sure buffers are flushed to disk, so they will survive a
  461.  * system crash.
  462.  *  MFS_ZERO Only write block 0.
  463.  *
  464.  * Return FAIL for failure, OK otherwise
  465.  */
  466.     int
  467. mf_sync(mfp, flags)
  468.     MEMFILE *mfp;
  469.     int     flags;
  470. {
  471.     int     status;
  472.     BHDR    *hp;
  473. #ifdef SYNC_DUP_CLOSE
  474.     int     fd;
  475. #endif
  476.     if (mfp->mf_fd < 0)     /* there is no file, nothing to do */
  477.     {
  478. mfp->mf_dirty = FALSE;
  479. return FAIL;
  480.     }
  481.     /*
  482.      * sync from last to first (may reduce the probability of an inconsistent
  483.      * file) If a write fails, it is very likely caused by a full filesystem.
  484.      * Then we only try to write blocks within the existing file. If that also
  485.      * fails then we give up.
  486.      */
  487.     status = OK;
  488.     for (hp = mfp->mf_used_last; hp != NULL; hp = hp->bh_prev)
  489. if (((flags & MFS_ALL) || hp->bh_bnum >= 0)
  490. && (hp->bh_flags & BH_DIRTY)
  491. && (status == OK || (hp->bh_bnum >= 0
  492.     && hp->bh_bnum < mfp->mf_infile_count)))
  493. {
  494.     if ((flags & MFS_ZERO) && hp->bh_bnum != 0)
  495. continue;
  496.     if (mf_write(mfp, hp) == FAIL)
  497.     {
  498. if (status == FAIL) /* double error: quit syncing */
  499.     break;
  500. status = FAIL;
  501.     }
  502.     /* Stop when char available now */
  503.     if ((flags & MFS_STOP) && ui_char_avail())
  504. break;
  505. }
  506.     /*
  507.      * If the whole list is flushed, the memfile is not dirty anymore.
  508.      * In case of an error this flag is also set, to avoid trying all the time.
  509.      */
  510.     if (hp == NULL || status == FAIL)
  511. mfp->mf_dirty = FALSE;
  512.     if ((flags & MFS_FLUSH) && *p_sws != NUL)
  513.     {
  514. #if defined(UNIX)
  515. # ifdef HAVE_FSYNC
  516. /*
  517.  * most Unixes have the very useful fsync() function, just what we need.
  518.  * However, with OS/2 and EMX it is also available, but there are
  519.  * reports of bad problems with it (a bug in HPFS.IFS).
  520.  * So we disable use of it here in case someone tries to be smart
  521.  * and changes os_os2_cfg.h... (even though there is no __EMX__ test
  522.  * in the #if, as __EMX__ does not have sync(); we hope for a timely
  523.  * sync from the system itself).
  524.  */
  525. #  if defined(__EMX__)
  526.    error "Dont use fsync with EMX! Read emxdoc.doc or emxfix01.doc for info."
  527. #  endif
  528. if (STRCMP(p_sws, "fsync") == 0)
  529. {
  530.     if (fsync(mfp->mf_fd))
  531. status = FAIL;
  532. }
  533. else
  534. # endif
  535. # ifdef __OPENNT
  536.     fflush(NULL); /* OpenNT is strictly POSIX (Benzinger) */
  537. # else
  538.     sync();
  539. # endif
  540. #endif
  541. #ifdef VMS
  542. if (STRCMP(p_sws, "fsync") == 0)
  543. {
  544.     if (fsync(mfp->mf_fd))
  545. status = FAIL;
  546. }
  547. #endif
  548. #ifdef MSDOS
  549. if (_dos_commit(mfp->mf_fd))
  550.     status = FAIL;
  551. #else
  552. # ifdef SYNC_DUP_CLOSE
  553. /*
  554.  * Win32 is a bit more work: Duplicate the file handle and close it.
  555.  * This should flush the file to disk.
  556.  */
  557. if ((fd = dup(mfp->mf_fd)) >= 0)
  558.     close(fd);
  559. # endif
  560. #endif
  561. #ifdef AMIGA
  562. /*
  563.  * Flush() only exists for AmigaDos 2.0.
  564.  * For 1.3 it should be done with close() + open(), but then the risk
  565.  * is that the open() may fail and lose the file....
  566.  */
  567. # ifndef NO_ARP
  568. if (dos2)
  569. # endif
  570. # ifdef SASC
  571. {
  572.     struct UFB *fp = chkufb(mfp->mf_fd);
  573.     if (fp != NULL)
  574. Flush(fp->ufbfh);
  575. }
  576. # else
  577. #  ifdef _DCC
  578. {
  579.     BPTR fh = (BPTR)fdtofh(mfp->mf_fd);
  580.     if (fh != 0)
  581. Flush(fh);
  582.     }
  583. #  else /* assume Manx */
  584.     Flush(_devtab[mfp->mf_fd].fd);
  585. #  endif
  586. # endif
  587. #endif /* AMIGA */
  588.     }
  589.     return status;
  590. }
  591. /*
  592.  * insert block *hp in front of hashlist of memfile *mfp
  593.  */
  594.     static void
  595. mf_ins_hash(mfp, hp)
  596.     MEMFILE *mfp;
  597.     BHDR    *hp;
  598. {
  599.     BHDR    *hhp;
  600.     int     hash;
  601.     hash = MEMHASH(hp->bh_bnum);
  602.     hhp = mfp->mf_hash[hash];
  603.     hp->bh_hash_next = hhp;
  604.     hp->bh_hash_prev = NULL;
  605.     if (hhp != NULL)
  606. hhp->bh_hash_prev = hp;
  607.     mfp->mf_hash[hash] = hp;
  608. }
  609. /*
  610.  * remove block *hp from hashlist of memfile list *mfp
  611.  */
  612.     static void
  613. mf_rem_hash(mfp, hp)
  614.     MEMFILE *mfp;
  615.     BHDR    *hp;
  616. {
  617.     if (hp->bh_hash_prev == NULL)
  618. mfp->mf_hash[MEMHASH(hp->bh_bnum)] = hp->bh_hash_next;
  619.     else
  620. hp->bh_hash_prev->bh_hash_next = hp->bh_hash_next;
  621.     if (hp->bh_hash_next)
  622. hp->bh_hash_next->bh_hash_prev = hp->bh_hash_prev;
  623. }
  624. /*
  625.  * look in hash lists of memfile *mfp for block header with number 'nr'
  626.  */
  627.     static BHDR *
  628. mf_find_hash(mfp, nr)
  629.     MEMFILE *mfp;
  630.     blocknr_t nr;
  631. {
  632.     BHDR *hp;
  633.     for (hp = mfp->mf_hash[MEMHASH(nr)]; hp != NULL; hp = hp->bh_hash_next)
  634. if (hp->bh_bnum == nr)
  635.     break;
  636.     return hp;
  637. }
  638. /*
  639.  * insert block *hp in front of used list of memfile *mfp
  640.  */
  641.     static void
  642. mf_ins_used(mfp, hp)
  643.     MEMFILE *mfp;
  644.     BHDR    *hp;
  645. {
  646.     hp->bh_next = mfp->mf_used_first;
  647.     mfp->mf_used_first = hp;
  648.     hp->bh_prev = NULL;
  649.     if (hp->bh_next == NULL)     /* list was empty, adjust last pointer */
  650. mfp->mf_used_last = hp;
  651.     else
  652. hp->bh_next->bh_prev = hp;
  653.     mfp->mf_used_count += hp->bh_page_count;
  654.     total_mem_used += hp->bh_page_count * mfp->mf_page_size;
  655. }
  656. /*
  657.  * remove block *hp from used list of memfile *mfp
  658.  */
  659.     static void
  660. mf_rem_used(mfp, hp)
  661.     MEMFILE *mfp;
  662.     BHDR    *hp;
  663. {
  664.     if (hp->bh_next == NULL)     /* last block in used list */
  665. mfp->mf_used_last = hp->bh_prev;
  666.     else
  667. hp->bh_next->bh_prev = hp->bh_prev;
  668.     if (hp->bh_prev == NULL)     /* first block in used list */
  669. mfp->mf_used_first = hp->bh_next;
  670.     else
  671. hp->bh_prev->bh_next = hp->bh_next;
  672.     mfp->mf_used_count -= hp->bh_page_count;
  673.     total_mem_used -= hp->bh_page_count * mfp->mf_page_size;
  674. }
  675. /*
  676.  * Release the least recently used block from the used list if the number
  677.  * of used memory blocks gets to big.
  678.  *
  679.  * Return the block header to the caller, including the memory block, so
  680.  * it can be re-used. Make sure the page_count is right.
  681.  */
  682.     static BHDR *
  683. mf_release(mfp, page_count)
  684.     MEMFILE *mfp;
  685.     int page_count;
  686. {
  687.     BHDR *hp;
  688.     int need_release;
  689.     BUF *buf;
  690.     /* don't release while in mf_close_file() */
  691.     if (dont_release)
  692. return NULL;
  693.     /*
  694.      * Need to release a block if the number of blocks for this memfile is
  695.      * higher than the maximum or total memory used is over 'maxmemtot'
  696.      */
  697.     need_release = ((mfp->mf_used_count >= mfp->mf_used_count_max)
  698.   || (total_mem_used >> 10) >= (long_u)p_mmt);
  699.     /*
  700.      * Try to create a swap file if the amount of memory used is getting too
  701.      * high.
  702.      */
  703.     if (mfp->mf_fd < 0 && need_release && p_uc)
  704.     {
  705. /* find for which buffer this memfile is */
  706. for (buf = firstbuf; buf != NULL; buf = buf->b_next)
  707.     if (buf->b_ml.ml_mfp == mfp)
  708. break;
  709. if (buf != NULL && buf->b_may_swap)
  710.     ml_open_file(buf);
  711.     }
  712.     /*
  713.      * don't release a block if
  714.      * there is no file for this memfile
  715.      * or
  716.      * the number of blocks for this memfile is lower than the maximum
  717.      *   and
  718.      * total memory used is not up to 'maxmemtot'
  719.      */
  720.     if (mfp->mf_fd < 0 || !need_release)
  721. return NULL;
  722.     for (hp = mfp->mf_used_last; hp != NULL; hp = hp->bh_prev)
  723. if (!(hp->bh_flags & BH_LOCKED))
  724.     break;
  725.     if (hp == NULL) /* not a single one that can be released */
  726. return NULL;
  727.     /*
  728.      * If the block is dirty, write it.
  729.      * If the write fails we don't free it.
  730.      */
  731.     if ((hp->bh_flags & BH_DIRTY) && mf_write(mfp, hp) == FAIL)
  732. return NULL;
  733.     mf_rem_used(mfp, hp);
  734.     mf_rem_hash(mfp, hp);
  735.     /*
  736.      * If a BHDR is returned, make sure that the page_count of bh_data is right
  737.      */
  738.     if (hp->bh_page_count != page_count)
  739.     {
  740. vim_free(hp->bh_data);
  741. if ((hp->bh_data = alloc(mfp->mf_page_size * page_count)) == NULL)
  742. {
  743.     vim_free(hp);
  744.     return NULL;
  745. }
  746. hp->bh_page_count = page_count;
  747.     }
  748.     return hp;
  749. }
  750. /*
  751.  * release as many blocks as possible
  752.  * Used in case of out of memory
  753.  *
  754.  * return TRUE if any memory was released
  755.  */
  756.     int
  757. mf_release_all()
  758. {
  759.     BUF *buf;
  760.     MEMFILE *mfp;
  761.     BHDR *hp;
  762.     int retval = FALSE;
  763.     for (buf = firstbuf; buf != NULL; buf = buf->b_next)
  764.     {
  765. mfp = buf->b_ml.ml_mfp;
  766. /* only if there is a memfile with a file */
  767. if (mfp != NULL && mfp->mf_fd >= 0)
  768.     for (hp = mfp->mf_used_last; hp != NULL; )
  769.     {
  770. if (!(hp->bh_flags & BH_LOCKED) && (!(hp->bh_flags & BH_DIRTY)
  771. || mf_write(mfp, hp) != FAIL))
  772. {
  773.     mf_rem_used(mfp, hp);
  774.     mf_rem_hash(mfp, hp);
  775.     mf_free_bhdr(hp);
  776.     hp = mfp->mf_used_last; /* re-start, list was changed */
  777.     retval = TRUE;
  778. }
  779. else
  780.     hp = hp->bh_prev;
  781.     }
  782.     }
  783.     return retval;
  784. }
  785. /*
  786.  * Allocate a block header and a block of memory for it
  787.  */
  788.     static BHDR *
  789. mf_alloc_bhdr(mfp, page_count)
  790.     MEMFILE *mfp;
  791.     int page_count;
  792. {
  793.     BHDR    *hp;
  794.     if ((hp = (BHDR *)alloc((unsigned)sizeof(BHDR))) != NULL)
  795.     {
  796. if ((hp->bh_data = (char_u *)alloc(mfp->mf_page_size * page_count))
  797.       == NULL)
  798. {
  799.     vim_free(hp);     /* not enough memory */
  800.     return NULL;
  801. }
  802. hp->bh_page_count = page_count;
  803.     }
  804.     return hp;
  805. }
  806. /*
  807.  * Free a block header and the block of memory for it
  808.  */
  809.     static void
  810. mf_free_bhdr(hp)
  811.     BHDR *hp;
  812. {
  813.     vim_free(hp->bh_data);
  814.     vim_free(hp);
  815. }
  816. /*
  817.  * insert entry *hp in the free list
  818.  */
  819.     static void
  820. mf_ins_free(mfp, hp)
  821.     MEMFILE *mfp;
  822.     BHDR    *hp;
  823. {
  824.     hp->bh_next = mfp->mf_free_first;
  825.     mfp->mf_free_first = hp;
  826. }
  827. /*
  828.  * remove the first entry from the free list and return a pointer to it
  829.  * Note: caller must check that mfp->mf_free_first is not NULL!
  830.  */
  831.     static BHDR *
  832. mf_rem_free(mfp)
  833.     MEMFILE *mfp;
  834. {
  835.     BHDR    *hp;
  836.     hp = mfp->mf_free_first;
  837.     mfp->mf_free_first = hp->bh_next;
  838.     return hp;
  839. }
  840. /*
  841.  * read a block from disk
  842.  *
  843.  * Return FAIL for failure, OK otherwise
  844.  */
  845.     static int
  846. mf_read(mfp, hp)
  847.     MEMFILE *mfp;
  848.     BHDR *hp;
  849. {
  850.     off_t offset;
  851.     unsigned page_size;
  852.     unsigned size;
  853.     if (mfp->mf_fd < 0)     /* there is no file, can't read */
  854. return FAIL;
  855.     page_size = mfp->mf_page_size;
  856.     offset = page_size * hp->bh_bnum;
  857.     size = page_size * hp->bh_page_count;
  858.     if (lseek(mfp->mf_fd, offset, SEEK_SET) != offset)
  859.     {
  860. EMSG("Seek error in swap file read");
  861. return FAIL;
  862.     }
  863.     if ((unsigned)read(mfp->mf_fd, (char *)hp->bh_data, (size_t)size) != size)
  864.     {
  865. EMSG("Read error in swap file");
  866. return FAIL;
  867.     }
  868.     return OK;
  869. }
  870. /*
  871.  * write a block to disk
  872.  *
  873.  * Return FAIL for failure, OK otherwise
  874.  */
  875.     static int
  876. mf_write(mfp, hp)
  877.     MEMFILE *mfp;
  878.     BHDR *hp;
  879. {
  880.     off_t offset;     /* offset in the file */
  881.     blocknr_t nr;     /* block nr which is being written */
  882.     BHDR *hp2;
  883.     unsigned page_size;  /* number of bytes in a page */
  884.     unsigned page_count; /* number of pages written */
  885.     unsigned size;     /* number of bytes written */
  886.     if (mfp->mf_fd < 0)     /* there is no file, can't write */
  887. return FAIL;
  888.     if (hp->bh_bnum < 0) /* must assign file block number */
  889. if (mf_trans_add(mfp, hp) == FAIL)
  890.     return FAIL;
  891.     page_size = mfp->mf_page_size;
  892.     /*
  893.      * We don't want gaps in the file. Write the blocks in front of *hp
  894.      * to extend the file.
  895.      * If block 'mf_infile_count' is not in the hash list, it has been
  896.      * freed. Fill the space in the file with data from the current block.
  897.      */
  898.     for (;;)
  899.     {
  900. nr = hp->bh_bnum;
  901. if (nr > mfp->mf_infile_count) /* beyond end of file */
  902. {
  903.     nr = mfp->mf_infile_count;
  904.     hp2 = mf_find_hash(mfp, nr); /* NULL catched below */
  905. }
  906. else
  907.     hp2 = hp;
  908. offset = page_size * nr;
  909. if (lseek(mfp->mf_fd, offset, SEEK_SET) != offset)
  910. {
  911.     EMSG("Seek error in swap file write");
  912.     return FAIL;
  913. }
  914. if (hp2 == NULL)     /* freed block, fill with dummy data */
  915.     page_count = 1;
  916. else
  917.     page_count = hp2->bh_page_count;
  918. size = page_size * page_count;
  919. if ((unsigned)write(mfp->mf_fd,
  920.      (char *)(hp2 == NULL ? hp : hp2)->bh_data, (size_t)size) != size)
  921. {
  922.     /*
  923.      * Avoid repeating the error message, this mostly happens when the
  924.      * disk is full. We give the message again only after a succesful
  925.      * write or when hitting a key. We keep on trying, in case some
  926.      * space becomes available.
  927.      */
  928.     if (!did_swapwrite_msg)
  929. EMSG("Write error in swap file");
  930.     did_swapwrite_msg = TRUE;
  931.     return FAIL;
  932. }
  933. did_swapwrite_msg = FALSE;
  934. if (hp2 != NULL)     /* written a non-dummy block */
  935.     hp2->bh_flags &= ~BH_DIRTY;
  936.     /* appended to the file */
  937. if (nr + (blocknr_t)page_count > mfp->mf_infile_count)
  938.     mfp->mf_infile_count = nr + page_count;
  939. if (nr == hp->bh_bnum)     /* written the desired block */
  940.     break;
  941.     }
  942.     return OK;
  943. }
  944. /*
  945.  * Make block number for *hp positive and add it to the translation list
  946.  *
  947.  * Return FAIL for failure, OK otherwise
  948.  */
  949.     static int
  950. mf_trans_add(mfp, hp)
  951.     MEMFILE *mfp;
  952.     BHDR    *hp;
  953. {
  954.     BHDR *freep;
  955.     blocknr_t new_bnum;
  956.     int hash;
  957.     NR_TRANS *np;
  958.     int page_count;
  959.     if (hp->bh_bnum >= 0)     /* it's already positive */
  960. return OK;
  961.     if ((np = (NR_TRANS *)alloc((unsigned)sizeof(NR_TRANS))) == NULL)
  962. return FAIL;
  963. /*
  964.  * get a new number for the block.
  965.  * If the first item in the free list has sufficient pages, use its number
  966.  * Otherwise use mf_blocknr_max.
  967.  */
  968.     freep = mfp->mf_free_first;
  969.     page_count = hp->bh_page_count;
  970.     if (freep != NULL && freep->bh_page_count >= page_count)
  971.     {
  972. new_bnum = freep->bh_bnum;
  973. /*
  974.  * If the page count of the free block was larger, recude it.
  975.  * If the page count matches, remove the block from the free list
  976.  */
  977. if (freep->bh_page_count > page_count)
  978. {
  979.     freep->bh_bnum += page_count;
  980.     freep->bh_page_count -= page_count;
  981. }
  982. else
  983. {
  984.     freep = mf_rem_free(mfp);
  985.     vim_free(freep);
  986. }
  987.     }
  988.     else
  989.     {
  990. new_bnum = mfp->mf_blocknr_max;
  991. mfp->mf_blocknr_max += page_count;
  992.     }
  993.     np->nt_old_bnum = hp->bh_bnum;     /* adjust number */
  994.     np->nt_new_bnum = new_bnum;
  995.     mf_rem_hash(mfp, hp);     /* remove from old hash list */
  996.     hp->bh_bnum = new_bnum;
  997.     mf_ins_hash(mfp, hp);     /* insert in new hash list */
  998.     hash = MEMHASH(np->nt_old_bnum);     /* insert in trans list */
  999.     np->nt_next = mfp->mf_trans[hash];
  1000.     mfp->mf_trans[hash] = np;
  1001.     if (np->nt_next != NULL)
  1002. np->nt_next->nt_prev = np;
  1003.     np->nt_prev = NULL;
  1004.     return OK;
  1005. }
  1006. /*
  1007.  * Lookup a tranlation from the trans lists and delete the entry
  1008.  *
  1009.  * Return the positive new number when found, the old number when not found
  1010.  */
  1011.     blocknr_t
  1012. mf_trans_del(mfp, old_nr)
  1013.     MEMFILE *mfp;
  1014.     blocknr_t old_nr;
  1015. {
  1016.     int hash;
  1017.     NR_TRANS *np;
  1018.     blocknr_t new_bnum;
  1019.     hash = MEMHASH(old_nr);
  1020.     for (np = mfp->mf_trans[hash]; np != NULL; np = np->nt_next)
  1021. if (np->nt_old_bnum == old_nr)
  1022.     break;
  1023.     if (np == NULL) /* not found */
  1024. return old_nr;
  1025.     mfp->mf_neg_count--;
  1026.     new_bnum = np->nt_new_bnum;
  1027.     if (np->nt_prev != NULL) /* remove entry from the trans list */
  1028. np->nt_prev->nt_next = np->nt_next;
  1029.     else
  1030. mfp->mf_trans[hash] = np->nt_next;
  1031.     if (np->nt_next != NULL)
  1032. np->nt_next->nt_prev = np->nt_prev;
  1033.     vim_free(np);
  1034.     return new_bnum;
  1035. }
  1036. /*
  1037.  * Set mfp->mf_ffname according to mfp->mf_fname and some other things.
  1038.  * Only called when creating or renaming the swapfile. Either way it's a new
  1039.  * name so we must work out the full path name.
  1040.  */
  1041.     void
  1042. mf_set_ffname(mfp)
  1043.     MEMFILE *mfp;
  1044. {
  1045.     mfp->mf_ffname = FullName_save(mfp->mf_fname, FALSE);
  1046. }
  1047. /*
  1048.  * Make the name of the file used for the memfile a full path.
  1049.  * Used before doing a :cd
  1050.  */
  1051.     void
  1052. mf_fullname(mfp)
  1053.     MEMFILE *mfp;
  1054. {
  1055.     if (mfp != NULL && mfp->mf_fname != NULL && mfp->mf_ffname != NULL)
  1056.     {
  1057. vim_free(mfp->mf_fname);
  1058. mfp->mf_fname = mfp->mf_ffname;
  1059. mfp->mf_ffname = NULL;
  1060.     }
  1061. }
  1062. /*
  1063.  * return TRUE if there are any translations pending for 'mfp'
  1064.  */
  1065.     int
  1066. mf_need_trans(mfp)
  1067.     MEMFILE *mfp;
  1068. {
  1069.     return (mfp->mf_fname != NULL && mfp->mf_neg_count > 0);
  1070. }
  1071. /*
  1072.  * Open a swap file for a memfile.
  1073.  * The "fname" must be in allocated memory, and is consumed (also when an
  1074.  * error occurs).
  1075.  */
  1076.     static void
  1077. mf_do_open(mfp, fname, trunc_file)
  1078.     MEMFILE *mfp;
  1079.     char_u *fname;
  1080.     int trunc_file;
  1081. {
  1082.     mfp->mf_fname = fname;
  1083.     /*
  1084.      * Get the full path name before the open, because this is
  1085.      * not possible after the open on the Amiga.
  1086.      * fname cannot be NameBuff, because it must have been allocated.
  1087.      */
  1088.     mf_set_ffname(mfp);
  1089. #if defined(MSDOS) || defined(WIN32) || defined(RISCOS)
  1090.     /*
  1091.      * A ":!cd e:xxx" may change the directory without us knowning, use the
  1092.      * full pathname always.  Careful: This frees fname!
  1093.      */
  1094.     mf_fullname(mfp);
  1095. #endif
  1096.     /*
  1097.      * try to open the file
  1098.      */
  1099.     mfp->mf_fd = open((char *)mfp->mf_fname,
  1100.     (trunc_file ? (O_CREAT | O_RDWR | O_TRUNC) : (O_RDONLY)) | O_EXTRA
  1101. #if defined(UNIX) || defined(RISCOS)  /* open in rw------- mode */
  1102.     , (mode_t)0600
  1103. #endif
  1104. #if defined(MSDOS) || defined(WIN32) || defined(__EMX__)
  1105.     , S_IREAD | S_IWRITE  /* open read/write */
  1106. #endif
  1107. #ifdef VMS     /* open in rw------- mode */
  1108.     , 0600
  1109. #endif
  1110.     );
  1111.     /*
  1112.      * If the file cannot be opened, use memory only
  1113.      */
  1114.     if (mfp->mf_fd < 0)
  1115.     {
  1116. vim_free(mfp->mf_fname);
  1117. vim_free(mfp->mf_ffname);
  1118. mfp->mf_fname = NULL;
  1119. mfp->mf_ffname = NULL;
  1120.     }
  1121.     else
  1122. mch_hide(mfp->mf_fname);    /* try setting the 'hidden' flag */
  1123. }