external/bsd/nvi/dist/common/vi_db.c

   1 /*-
   2  * Copyright (c) 1992, 1993, 1994
   3  *      The Regents of the University of California.  All rights reserved.
   4  * Copyright (c) 1992, 1993, 1994, 1995, 1996
   5  *      Keith Bostic.  All rights reserved.
   6  *
   7  * See the LICENSE file for redistribution information.
   8  */
   9
  10 #include "config.h"
  11
  12 #include <sys/cdefs.h>
  13 #if 0
  14 #ifndef lint
  15 static const char sccsid[] = "Id: db.c,v 10.48 2002/06/08 19:32:52 skimo Exp  (Berkeley) Date: 2002/06/08 19:32:52 ";
  16 #endif /* not lint */
  17 #else
  18 __RCSID("$NetBSD: vi_db.c,v 1.6 2014/01/26 21:43:45 christos Exp $");
  19 #endif
  20
  21 #include <sys/types.h>
  22 #include <sys/queue.h>
  23 #include <sys/time.h>
  24 #include <sys/stat.h>
  25
  26 #include <bitstring.h>
  27 #include <errno.h>
  28 #include <limits.h>
  29 #include <stdio.h>
  30 #include <string.h>
  31 #include <stdlib.h>
  32 #include <unistd.h>
  33
  34 #include "common.h"
  35 #include "dbinternal.h"
  36 #include "../vi/vi.h"
  37
  38 static int append __P((SCR*, db_recno_t, const CHAR_T*, size_t, lnop_t, int));
  39
  40 /*
  41  * db_eget --
  42  *      Front-end to db_get, special case handling for empty files.
  43  *
  44  * PUBLIC: int db_eget __P((SCR *, db_recno_t, CHAR_T **, size_t *, int *));
  45  */
  46 int
  47 db_eget(SCR *sp, db_recno_t lno, CHAR_T **pp, size_t *lenp, int *isemptyp)
  48
  49                                                 /* Line number. */
  50                                                 /* Pointer store. */
  51                                                 /* Length store. */
  52
  53 {
  54         db_recno_t l1;
  55
  56         if (isemptyp != NULL)
  57                 *isemptyp = 0;
  58
  59         /* If the line exists, simply return it. */
  60         if (!db_get(sp, lno, 0, pp, lenp))
  61                 return (0);
  62
  63         /*
  64          * If the user asked for line 0 or line 1, i.e. the only possible
  65          * line in an empty file, find the last line of the file; db_last
  66          * fails loudly.
  67          */
  68         if ((lno == 0 || lno == 1) && db_last(sp, &l1))
  69                 return (1);
  70
  71         /* If the file isn't empty, fail loudly. */
  72         if ((lno != 0 && lno != 1) || l1 != 0) {
  73                 db_err(sp, lno);
  74                 return (1);
  75         }
  76
  77         if (isemptyp != NULL)
  78                 *isemptyp = 1;
  79
  80         return (1);
  81 }
  82
  83 /*
  84  * db_get --
  85  *      Look in the text buffers for a line, followed by the cache, followed
  86  *      by the database.
  87  *
  88  * PUBLIC: int db_get __P((SCR *, db_recno_t, u_int32_t, CHAR_T **, size_t *));
  89  */
  90 int
  91 db_get(SCR *sp, db_recno_t lno, u_int32_t flags, CHAR_T **pp, size_t *lenp)
  92                 /* Line number. */ /* Pointer store. */ /* Length store. */
  93 {
  94         DBT data, key;
  95         EXF *ep;
  96         TEXT *tp;
  97         db_recno_t l1, l2;
  98         const CHAR_T *wp;
  99         size_t wlen;
 100         size_t nlen;
 101
 102         /*
 103          * The underlying recno stuff handles zero by returning NULL, but
 104          * have to have an OOB condition for the look-aside into the input
 105          * buffer anyway.
 106          */
 107         if (lno == 0)
 108                 goto err1;
 109
 110         /* Check for no underlying file. */
 111         if ((ep = sp->ep) == NULL) {
 112                 ex_emsg(sp, NULL, EXM_NOFILEYET);
 113                 goto err3;
 114         }
 115
 116         if (LF_ISSET(DBG_NOCACHE))
 117                 goto nocache;
 118
 119         /*
 120          * Look-aside into the TEXT buffers and see if the line we want
 121          * is there.
 122          */
 123         if (F_ISSET(sp, SC_TINPUT)) {
 124                 l1 = TAILQ_FIRST(&sp->tiq)->lno;
 125                 l2 = TAILQ_LAST(&sp->tiq, _texth)->lno;
 126                 if (l1 <= lno && l2 >= lno) {
 127 #if defined(DEBUG) && 0
 128                         vtrace(sp,
 129                             "retrieve TEXT buffer line %lu\n", (u_long)lno);
 130 #endif
 131                         for (tp = TAILQ_FIRST(&sp->tiq);
 132                             tp->lno != lno; tp = TAILQ_NEXT(tp, q));
 133                         if (lenp != NULL)
 134                                 *lenp = tp->len;
 135                         if (pp != NULL)
 136                                 *pp = tp->lb;
 137                         return (0);
 138                 }
 139                 /*
 140                  * Adjust the line number for the number of lines used
 141                  * by the text input buffers.
 142                  */
 143                 if (lno > l2)
 144                         lno -= l2 - l1;
 145         }
 146
 147         /* Look-aside into the cache, and see if the line we want is there. */
 148         if (lno == sp->c_lno) {
 149 #if defined(DEBUG) && 0
 150                 vtrace(sp, "retrieve cached line %lu\n", (u_long)lno);
 151 #endif
 152                 if (lenp != NULL)
 153                         *lenp = sp->c_len;
 154                 if (pp != NULL)
 155                         *pp = sp->c_lp;
 156                 return (0);
 157         }
 158         sp->c_lno = OOBLNO;
 159
 160 nocache:
 161         nlen = 1024;
 162 retry:
 163         /* data.size contains length in bytes */
 164         BINC_GOTO(sp, CHAR_T, sp->c_lp, sp->c_blen, nlen);
 165
 166         /* Get the line from the underlying database. */
 167         memset(&key, 0, sizeof(key));
 168         key.data = &lno;
 169         key.size = sizeof(lno);
 170         memset(&data, 0, sizeof(data));
 171         data.data = sp->c_lp;
 172         data.ulen = sp->c_blen;
 173         data.flags = DB_DBT_USERMEM;
 174         switch (ep->db->get(ep->db, NULL, &key, &data, 0)) {
 175         case DB_BUFFER_SMALL:
 176                 nlen = data.size;
 177                 goto retry;
 178         default:
 179                 goto err2;
 180         case DB_NOTFOUND:
 181 err1:           if (LF_ISSET(DBG_FATAL))
 182 err2:                   db_err(sp, lno);
 183 alloc_err:
 184 err3:           if (lenp != NULL)
 185                         *lenp = 0;
 186                 if (pp != NULL)
 187                         *pp = NULL;
 188                 return (1);
 189         case 0:
 190                 ;
 191         }
 192
 193         if (FILE2INT(sp, data.data, data.size, wp, wlen)) {
 194             if (!F_ISSET(sp, SC_CONV_ERROR)) {
 195                 F_SET(sp, SC_CONV_ERROR);
 196                 msgq(sp, M_ERR, "324|Conversion error on line %d", lno);
 197             }
 198             goto err3;
 199         }
 200
 201         /* Reset the cache. */
 202         if (wp != data.data) {
 203             BINC_GOTOW(sp, sp->c_lp, sp->c_blen, wlen);
 204             MEMCPYW(sp->c_lp, wp, wlen);
 205         }
 206         sp->c_lno = lno;
 207         sp->c_len = wlen;
 208
 209 #if defined(DEBUG) && 0
 210         vtrace(sp, "retrieve DB line %lu\n", (u_long)lno);
 211 #endif
 212         if (lenp != NULL)
 213                 *lenp = wlen;
 214         if (pp != NULL)
 215                 *pp = sp->c_lp;
 216         return (0);
 217 }
 218
 219 /*
 220  * db_delete --
 221  *      Delete a line from the file.
 222  *
 223  * PUBLIC: int db_delete __P((SCR *, db_recno_t));
 224  */
 225 int
 226 db_delete(SCR *sp, db_recno_t lno)
 227 {
 228         DBT key;
 229         EXF *ep;
 230
 231 #if defined(DEBUG) && 0
 232         vtrace(sp, "delete line %lu\n", (u_long)lno);
 233 #endif
 234         /* Check for no underlying file. */
 235         if ((ep = sp->ep) == NULL) {
 236                 ex_emsg(sp, NULL, EXM_NOFILEYET);
 237                 return (1);
 238         }
 239         if (ep->l_win && ep->l_win != sp->wp) {
 240                 ex_emsg(sp, NULL, EXM_LOCKED);
 241                 return 1;
 242         }
 243
 244         /* Update marks, @ and global commands. */
 245         if (line_insdel(sp, LINE_DELETE, lno))
 246                 return 1;
 247
 248         /* Log before change. */
 249         log_line(sp, lno, LOG_LINE_DELETE_B);
 250
 251         /* Update file. */
 252         memset(&key, 0, sizeof(key));
 253         key.data = &lno;
 254         key.size = sizeof(lno);
 255         if ((sp->db_error = ep->db->del(ep->db, NULL, &key, 0)) != 0) {
 256                 msgq(sp, M_DBERR, "003|unable to delete line %lu",
 257                     (u_long)lno);
 258                 return (1);
 259         }
 260
 261         /* Flush the cache, update line count, before screen update. */
 262         update_cache(sp, LINE_DELETE, lno);
 263
 264         /* File now modified. */
 265         if (F_ISSET(ep, F_FIRSTMODIFY))
 266                 (void)rcv_init(sp);
 267         F_SET(ep, F_MODIFIED);
 268
 269         /* Log after change. */
 270         log_line(sp, lno, LOG_LINE_DELETE_F);
 271
 272         /* Update screen. */
 273         return (scr_update(sp, lno, LINE_DELETE, 1));
 274 }
 275
 276 /* maybe this could be simpler
 277  *
 278  * DB3 behaves differently from DB1
 279  *
 280  * if lno != 0 just go to lno and put the new line after it
 281  * if lno == 0 then if there are any record, put in front of the first
 282  *                  otherwise just append to the end thus creating the first
 283  *                              line
 284  */
 285 static int
 286 append(SCR *sp, db_recno_t lno, const CHAR_T *p, size_t len, lnop_t op, int update)
 287 {
 288         DBT data, key;
 289         DBC *dbcp_put;
 290         EXF *ep;
 291         const char *fp;
 292         size_t flen;
 293         int rval;
 294
 295         /* Check for no underlying file. */
 296         if ((ep = sp->ep) == NULL) {
 297                 ex_emsg(sp, NULL, EXM_NOFILEYET);
 298                 return (1);
 299         }
 300         if (ep->l_win && ep->l_win != sp->wp) {
 301                 ex_emsg(sp, NULL, EXM_LOCKED);
 302                 return 1;
 303         }
 304
 305         /* Log before change. */
 306         log_line(sp, lno + 1, LOG_LINE_APPEND_B);
 307
 308         /* Update file. */
 309         memset(&key, 0, sizeof(key));
 310         key.data = &lno;
 311         key.size = sizeof(lno);
 312         memset(&data, 0, sizeof(data));
 313
 314         if ((sp->db_error = ep->db->cursor(ep->db, NULL, &dbcp_put, 0)) != 0)
 315             return 1;
 316
 317         INT2FILE(sp, p, len, fp, flen);
 318
 319         if (lno != 0) {
 320             if ((sp->db_error = dbcp_put->c_get(dbcp_put, &key, &data, DB_SET)) != 0)
 321                 goto err2;
 322
 323             data.data = __UNCONST(fp);
 324             data.size = flen;
 325             if ((sp->db_error = dbcp_put->c_put(dbcp_put, &key, &data, DB_AFTER)) != 0) {
 326 err2:
 327                 (void)dbcp_put->c_close(dbcp_put);
 328                 msgq(sp, M_DBERR,
 329                         (op == LINE_APPEND)
 330                             ? "004|unable to append to line %lu"
 331                             : "005|unable to insert at line %lu",
 332                         (u_long)lno);
 333                 return (1);
 334             }
 335         } else {
 336             if ((sp->db_error = dbcp_put->c_get(dbcp_put, &key, &data, DB_FIRST)) != 0) {
 337                 if (sp->db_error != DB_NOTFOUND)
 338                     goto err2;
 339
 340                 data.data = __UNCONST(fp);
 341                 data.size = flen;
 342                 if ((sp->db_error = ep->db->put(ep->db, NULL, &key, &data, DB_APPEND)) != 0) {
 343                     goto err2;
 344                 }
 345             } else {
 346                 key.data = &lno;
 347                 key.size = sizeof(lno);
 348                 data.data = __UNCONST(fp);
 349                 data.size = flen;
 350                 if ((sp->db_error = dbcp_put->c_put(dbcp_put, &key, &data, DB_BEFORE)) != 0) {
 351                     goto err2;
 352                 }
 353             }
 354         }
 355
 356         (void)dbcp_put->c_close(dbcp_put);
 357
 358         /* Flush the cache, update line count, before screen update. */
 359         update_cache(sp, LINE_INSERT, lno);
 360
 361         /* File now dirty. */
 362         if (F_ISSET(ep, F_FIRSTMODIFY))
 363                 (void)rcv_init(sp);
 364         F_SET(ep, F_MODIFIED);
 365
 366         /* Log after change. */
 367         log_line(sp, lno + 1, LOG_LINE_APPEND_F);
 368
 369         /* Update marks, @ and global commands. */
 370         rval = line_insdel(sp, LINE_INSERT, lno + 1);
 371
 372         /*
 373          * Update screen.
 374          *
 375          * comment copied from db_append
 376          * XXX
 377          * Nasty hack.  If multiple lines are input by the user, they aren't
 378          * committed until an <ESC> is entered.  The problem is the screen was
 379          * updated/scrolled as each line was entered.  So, when this routine
 380          * is called to copy the new lines from the cut buffer into the file,
 381          * it has to know not to update the screen again.
 382          */
 383         return (scr_update(sp, lno + 1, LINE_INSERT, update) || rval);
 384 }
 385
 386 /*
 387  * db_append --
 388  *      Append a line into the file.
 389  *
 390  * PUBLIC: int db_append __P((SCR *, int, db_recno_t, const CHAR_T *, size_t));
 391  */
 392 int
 393 db_append(SCR *sp, int update, db_recno_t lno, const CHAR_T *p, size_t len)
 394 {
 395 #if defined(DEBUG) && 0
 396         vtrace(sp, "append to %lu: len %u {%.*s}\n", lno, len, MIN(len, 20), p);
 397 #endif
 398
 399         /* Update file. */
 400         return append(sp, lno, p, len, LINE_APPEND, update);
 401 }
 402
 403 /*
 404  * db_insert --
 405  *      Insert a line into the file.
 406  *
 407  * PUBLIC: int db_insert __P((SCR *, db_recno_t, CHAR_T *, size_t));
 408  */
 409 int
 410 db_insert(SCR *sp, db_recno_t lno, CHAR_T *p, size_t len)
 411 {
 412 #if defined(DEBUG) && 0
 413         vtrace(sp, "insert before %lu: len %lu {%.*s}\n",
 414             (u_long)lno, (u_long)len, MIN(len, 20), p);
 415 #endif
 416         return append(sp, lno - 1, p, len, LINE_INSERT, 1);
 417 }
 418
 419 /*
 420  * db_set --
 421  *      Store a line in the file.
 422  *
 423  * PUBLIC: int db_set __P((SCR *, db_recno_t, CHAR_T *, size_t));
 424  */
 425 int
 426 db_set(SCR *sp, db_recno_t lno, CHAR_T *p, size_t len)
 427 {
 428         DBT data, key;
 429         EXF *ep;
 430         const char *fp;
 431         size_t flen;
 432
 433 #if defined(DEBUG) && 0
 434         vtrace(sp, "replace line %lu: len %lu {%.*s}\n",
 435             (u_long)lno, (u_long)len, MIN(len, 20), p);
 436 #endif
 437         /* Check for no underlying file. */
 438         if ((ep = sp->ep) == NULL) {
 439                 ex_emsg(sp, NULL, EXM_NOFILEYET);
 440                 return (1);
 441         }
 442         if (ep->l_win && ep->l_win != sp->wp) {
 443                 ex_emsg(sp, NULL, EXM_LOCKED);
 444                 return 1;
 445         }
 446
 447         /* Log before change. */
 448         log_line(sp, lno, LOG_LINE_RESET_B);
 449
 450         INT2FILE(sp, p, len, fp, flen);
 451
 452         /* Update file. */
 453         memset(&key, 0, sizeof(key));
 454         key.data = &lno;
 455         key.size = sizeof(lno);
 456         memset(&data, 0, sizeof(data));
 457         data.data = __UNCONST(fp);
 458         data.size = flen;
 459         if ((sp->db_error = ep->db->put(ep->db, NULL, &key, &data, 0)) != 0) {
 460                 msgq(sp, M_DBERR, "006|unable to store line %lu", (u_long)lno);
 461                 return (1);
 462         }
 463
 464         /* Flush the cache, update line count, before screen update. */
 465         update_cache(sp, LINE_RESET, lno);
 466
 467         /* File now dirty. */
 468         if (F_ISSET(ep, F_FIRSTMODIFY))
 469                 (void)rcv_init(sp);
 470         F_SET(ep, F_MODIFIED);
 471
 472         /* Log after change. */
 473         log_line(sp, lno, LOG_LINE_RESET_F);
 474
 475         /* Update screen. */
 476         return (scr_update(sp, lno, LINE_RESET, 1));
 477 }
 478
 479 /*
 480  * db_exist --
 481  *      Return if a line exists.
 482  *
 483  * PUBLIC: int db_exist __P((SCR *, db_recno_t));
 484  */
 485 int
 486 db_exist(SCR *sp, db_recno_t lno)
 487 {
 488         EXF *ep;
 489
 490         /* Check for no underlying file. */
 491         if ((ep = sp->ep) == NULL) {
 492                 ex_emsg(sp, NULL, EXM_NOFILEYET);
 493                 return (1);
 494         }
 495
 496         if (lno == OOBLNO)
 497                 return (0);
 498
 499         /*
 500          * Check the last-line number cache.  Adjust the cached line
 501          * number for the lines used by the text input buffers.
 502          */
 503         if (ep->c_nlines != OOBLNO)
 504                 return (lno <= (F_ISSET(sp, SC_TINPUT) ?
 505                     ep->c_nlines + TAILQ_LAST(&sp->tiq, _texth)->lno -
 506                     TAILQ_FIRST(&sp->tiq)->lno : ep->c_nlines));
 507
 508         /* Go get the line. */
 509         return (!db_get(sp, lno, 0, NULL, NULL));
 510 }
 511
 512 /*
 513  * db_last --
 514  *      Return the number of lines in the file.
 515  *
 516  * PUBLIC: int db_last __P((SCR *, db_recno_t *));
 517  */
 518 int
 519 db_last(SCR *sp, db_recno_t *lnop)
 520 {
 521         DBT data, key;
 522         DBC *dbcp;
 523         EXF *ep;
 524         db_recno_t lno;
 525         const CHAR_T *wp;
 526         size_t wlen;
 527
 528         /* Check for no underlying file. */
 529         if ((ep = sp->ep) == NULL) {
 530                 ex_emsg(sp, NULL, EXM_NOFILEYET);
 531                 return (1);
 532         }
 533
 534         /*
 535          * Check the last-line number cache.  Adjust the cached line
 536          * number for the lines used by the text input buffers.
 537          */
 538         if (ep->c_nlines != OOBLNO) {
 539                 *lnop = ep->c_nlines;
 540                 if (F_ISSET(sp, SC_TINPUT))
 541                         *lnop += TAILQ_LAST(&sp->tiq, _texth)->lno -
 542                             TAILQ_FIRST(&sp->tiq)->lno;
 543                 return (0);
 544         }
 545
 546         memset(&key, 0, sizeof(key));
 547         key.data = &lno;
 548         key.size = sizeof(lno);
 549         memset(&data, 0, sizeof(data));
 550
 551         if ((sp->db_error = ep->db->cursor(ep->db, NULL, &dbcp, 0)) != 0)
 552             goto err1;
 553         switch (sp->db_error = dbcp->c_get(dbcp, &key, &data, DB_LAST)) {
 554         case DB_NOTFOUND:
 555                 *lnop = 0;
 556                 return (0);
 557         default:
 558                 (void)dbcp->c_close(dbcp);
 559 alloc_err:
 560 err1:
 561                 msgq(sp, M_DBERR, "007|unable to get last line");
 562                 *lnop = 0;
 563                 return (1);
 564         case 0:
 565                 ;
 566         }
 567
 568         memcpy(&lno, key.data, sizeof(lno));
 569
 570         if (lno != sp->c_lno) {
 571             FILE2INT(sp, data.data, data.size, wp, wlen);
 572
 573             /* Fill the cache. */
 574             BINC_GOTOW(sp, sp->c_lp, sp->c_blen, wlen);
 575             MEMCPYW(sp->c_lp, wp, wlen);
 576             sp->c_lno = lno;
 577             sp->c_len = wlen;
 578         }
 579         ep->c_nlines = lno;
 580
 581         (void)dbcp->c_close(dbcp);
 582
 583         /* Return the value. */
 584         *lnop = (F_ISSET(sp, SC_TINPUT) &&
 585             TAILQ_LAST(&sp->tiq, _texth)->lno > lno ?
 586             TAILQ_LAST(&sp->tiq, _texth)->lno : lno);
 587         return (0);
 588 }
 589
 590 /*
 591  * db_err --
 592  *      Report a line error.
 593  *
 594  * PUBLIC: void db_err __P((SCR *, db_recno_t));
 595  */
 596 void
 597 db_err(SCR *sp, db_recno_t lno)
 598 {
 599         msgq(sp, M_ERR,
 600             "008|Error: unable to retrieve line %lu", (u_long)lno);
 601 }
 602
 603 /*
 604  * scr_update --
 605  *      Update all of the screens that are backed by the file that
 606  *      just changed.
 607  *
 608  * PUBLIC: int scr_update __P((SCR *sp, db_recno_t lno,
 609  * PUBLIC:                      lnop_t op, int current));
 610  */
 611 int
 612 scr_update(SCR *sp, db_recno_t lno, lnop_t op, int current)
 613 {
 614         EXF *ep;
 615         SCR *tsp;
 616         WIN *wp;
 617
 618         if (F_ISSET(sp, SC_EX))
 619                 return (0);
 620
 621         /* XXXX goes outside of window */
 622         ep = sp->ep;
 623         if (ep->refcnt != 1)
 624                 TAILQ_FOREACH(wp, &sp->gp->dq, q)
 625                         TAILQ_FOREACH(tsp, &wp->scrq, q)
 626                                 if (sp != tsp && tsp->ep == ep)
 627                                         if (vs_change(tsp, lno, op))
 628                                                 return (1);
 629         return (current ? vs_change(sp, lno, op) : 0);
 630 }
 631
 632 /*
 633  * PUBLIC: void update_cache __P((SCR *sp, lnop_t op, db_recno_t lno));
 634  */
 635 void
 636 update_cache(SCR *sp, lnop_t op, db_recno_t lno)
 637 {
 638         SCR* scrp;
 639         EXF *ep;
 640
 641         ep = sp->ep;
 642
 643         /* Flush the cache, update line count, before screen update. */
 644         /* The flushing is probably not needed, since it was incorrect
 645          * for db_insert.  It might be better to adjust it, like
 646          * marks, @ and global
 647          */
 648         TAILQ_FOREACH(scrp, &ep->scrq, eq)
 649                 switch (op) {
 650                 case LINE_INSERT:
 651                 case LINE_DELETE:
 652                         if (lno <= scrp->c_lno)
 653                                 scrp->c_lno = OOBLNO;
 654                         break;
 655                 case LINE_RESET:
 656                         if (lno == scrp->c_lno)
 657                                 scrp->c_lno = OOBLNO;
 658                         break;
 659                 case LINE_APPEND:
 660                         break;
 661                 }
 662
 663         if (ep->c_nlines != OOBLNO)
 664                 switch (op) {
 665                 case LINE_INSERT:
 666                         ++ep->c_nlines;
 667                         break;
 668                 case LINE_DELETE:
 669                         --ep->c_nlines;
 670                         break;
 671                 case LINE_APPEND:
 672                 case LINE_RESET:
 673                         break;
 674                 }
 675 }
 676
 677 /*
 678  * PUBLIC: int line_insdel __P((SCR *sp, lnop_t op, db_recno_t lno));
 679  */
 680 int
 681 line_insdel(SCR *sp, lnop_t op, db_recno_t lno)
 682 {
 683         int rval;
 684
 685         /* Update marks, @ and global commands. */
 686         rval = 0;
 687         if (mark_insdel(sp, op, lno))
 688                 rval = 1;
 689         if (ex_g_insdel(sp, op, lno))
 690                 rval = 1;
 691
 692         return rval;
 693 }
 694
 695 #ifdef USE_DB4_LOGGING
 696 #define VI_DB_INIT_LOG  DB_INIT_LOG
 697 #else
 698 #define VI_DB_INIT_LOG  0
 699 #endif
 700
 701 /*
 702  * PUBLIC: int db_setup __P((SCR *, EXF *));
 703  */
 704 int
 705 db_setup(SCR *sp, EXF *ep)
 706 {
 707         char path[MAXPATHLEN];
 708         int fd;
 709         DB_ENV  *env;
 710
 711         (void)snprintf(path, sizeof(path), "%s/vi.XXXXXX", O_STR(sp, O_RECDIR));
 712         if ((fd = mkstemp(path)) == -1) {
 713                 msgq(sp, M_SYSERR, "%s", path);
 714                 goto err;
 715         }
 716         (void)close(fd);
 717         (void)unlink(path);
 718         if (mkdir(path, S_IRWXU)) {
 719                 msgq(sp, M_SYSERR, "%s", path);
 720                 goto err;
 721         }
 722         if (db_env_create(&env, 0)) {
 723                 msgq(sp, M_ERR, "env_create");
 724                 goto err;
 725         }
 726 #ifdef USE_DB4_LOGGING
 727         if ((sp->db_error = vi_db_init_recover(env))) {
 728                 msgq(sp, M_DBERR, "init_recover");
 729                 goto err;
 730         }
 731         if ((sp->db_error = __vi_init_recover(env))) {
 732                 msgq(sp, M_DBERR, "init_recover");
 733                 goto err;
 734         }
 735 #endif
 736         if ((sp->db_error = db_env_open(env, path,
 737             DB_PRIVATE | DB_CREATE | DB_INIT_MPOOL | VI_DB_THREAD
 738             | VI_DB_INIT_LOG, 0)) != 0) {
 739                 msgq(sp, M_DBERR, "env->open");
 740                 goto err;
 741         }
 742
 743         if ((ep->env_path = strdup(path)) == NULL) {
 744                 msgq(sp, M_SYSERR, NULL);
 745                 (void)rmdir(path);
 746                 goto err;
 747         }
 748         ep->env = env;
 749         return 0;
 750 err:
 751         return 1;
 752 }
 753
 754 /* Round up v to the nearest power of 2 */
 755 static size_t round_up(size_t v)
 756 {
 757         ssize_t old_v = v;
 758
 759         while (v) {
 760                 old_v = v;
 761                 v ^= v & -v;
 762         }
 763         return old_v << 1;
 764 }
 765
 766 /*
 767  * PUBLIC: int db_msg_open __P((SCR *, const char *, DB **));
 768  */
 769 int db_msg_open(SCR *sp, const char *file, DB **dbp)
 770 {
 771         return  (sp->db_error = db_create(dbp, 0, 0)) != 0 ||
 772                 (sp->db_error = (*dbp)->set_re_source(*dbp, file)) != 0 ||
 773                 (sp->db_error = db_open(*dbp, NULL, DB_RECNO, 0, 0)) != 0;
 774 }
 775
 776 /*
 777  * PUBLIC: int db_init __P((SCR *, EXF *, char *, char *, size_t, int *));
 778  */
 779 int
 780 db_init(SCR *sp, EXF *ep, char *rcv_name, char *oname, size_t psize, int *open_err)
 781 {
 782         if (db_setup(sp, ep))
 783                 return 1;
 784
 785         /* Open a db structure. */
 786         if ((sp->db_error = db_create(&ep->db, 0, 0)) != 0) {
 787                 msgq(sp, M_DBERR, "db_create");
 788                 return 1;
 789         }
 790
 791         ep->db->set_re_delim(ep->db, '\n');             /* Always set. */
 792         ep->db->set_pagesize(ep->db, round_up(psize));
 793         ep->db->set_flags(ep->db, DB_RENUMBER | DB_SNAPSHOT);
 794         if (rcv_name == NULL)
 795                 ep->db->set_re_source(ep->db, oname);
 796
 797 /*
 798  * Don't let db use mmap when using fcntl for locking
 799  */
 800 #ifdef HAVE_LOCK_FCNTL
 801 #define NOMMAPIFFCNTL DB_NOMMAP
 802 #else
 803 #define NOMMAPIFFCNTL 0
 804 #endif
 805
 806 #define _DB_OPEN_MODE   S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP | S_IROTH | S_IWOTH
 807
 808         if ((sp->db_error = db_open(ep->db, ep->rcv_path, DB_RECNO,
 809             ((rcv_name == 0) ? DB_TRUNCATE : 0) | VI_DB_THREAD | NOMMAPIFFCNTL,
 810             _DB_OPEN_MODE)) != 0) {
 811                 msgq_str(sp,
 812                     M_DBERR, rcv_name == NULL ? oname : rcv_name, "%s");
 813                 /*
 814                  * !!!
 815                  * Historically, vi permitted users to edit files that couldn't
 816                  * be read.  This isn't useful for single files from a command
 817                  * line, but it's quite useful for "vi *.c", since you can skip
 818                  * past files that you can't read.
 819                  */
 820                 ep->db = NULL; /* Don't close it; it wasn't opened */
 821
 822                 *open_err = 1;
 823                 return 1;
 824         }
 825
 826         /* re_source is loaded into the database.
 827          * Close it and reopen it in the environment.
 828          */
 829         if ((sp->db_error = ep->db->close(ep->db, 0))) {
 830                 msgq(sp, M_DBERR, "close");
 831                 return 1;
 832         }
 833         if ((sp->db_error = db_create(&ep->db, ep->env, 0)) != 0) {
 834                 msgq(sp, M_DBERR, "db_create 2");
 835                 return 1;
 836         }
 837         if ((sp->db_error = db_open(ep->db, ep->rcv_path, DB_RECNO,
 838             VI_DB_THREAD | NOMMAPIFFCNTL, _DB_OPEN_MODE)) != 0) {
 839                 msgq_str(sp,
 840                     M_DBERR, ep->rcv_path, "%s");
 841                 return 1;
 842         }
 843
 844         return 0;
 845 }