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
db_eget(SCR * sp,db_recno_t lno,CHAR_T ** pp,size_t * lenp,int * isemptyp)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
db_get(SCR * sp,db_recno_t lno,u_int32_t flags,CHAR_T ** pp,size_t * lenp)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
db_delete(SCR * sp,db_recno_t lno)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
append(SCR * sp,db_recno_t lno,const CHAR_T * p,size_t len,lnop_t op,int update)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
db_append(SCR * sp,int update,db_recno_t lno,const CHAR_T * p,size_t len)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
db_insert(SCR * sp,db_recno_t lno,CHAR_T * p,size_t len)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
db_set(SCR * sp,db_recno_t lno,CHAR_T * p,size_t len)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
db_exist(SCR * sp,db_recno_t lno)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
db_last(SCR * sp,db_recno_t * lnop)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
db_err(SCR * sp,db_recno_t lno)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
scr_update(SCR * sp,db_recno_t lno,lnop_t op,int current)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
update_cache(SCR * sp,lnop_t op,db_recno_t lno)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
line_insdel(SCR * sp,lnop_t op,db_recno_t lno)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
db_setup(SCR * sp,EXF * ep)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 */
round_up(size_t v)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 */
db_msg_open(SCR * sp,const char * file,DB ** dbp)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
db_init(SCR * sp,EXF * ep,char * rcv_name,char * oname,size_t psize,int * open_err)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 }
846