17c478bd9Sstevel@tonic-gate /*-
27c478bd9Sstevel@tonic-gate * Copyright (c) 1990, 1993, 1994
37c478bd9Sstevel@tonic-gate * The Regents of the University of California. All rights reserved.
47c478bd9Sstevel@tonic-gate *
57c478bd9Sstevel@tonic-gate * Redistribution and use in source and binary forms, with or without
67c478bd9Sstevel@tonic-gate * modification, are permitted provided that the following conditions
77c478bd9Sstevel@tonic-gate * are met:
87c478bd9Sstevel@tonic-gate * 1. Redistributions of source code must retain the above copyright
97c478bd9Sstevel@tonic-gate * notice, this list of conditions and the following disclaimer.
107c478bd9Sstevel@tonic-gate * 2. Redistributions in binary form must reproduce the above copyright
117c478bd9Sstevel@tonic-gate * notice, this list of conditions and the following disclaimer in the
127c478bd9Sstevel@tonic-gate * documentation and/or other materials provided with the distribution.
137c478bd9Sstevel@tonic-gate * 3. All advertising materials mentioning features or use of this software
147c478bd9Sstevel@tonic-gate * must display the following acknowledgement:
157c478bd9Sstevel@tonic-gate * This product includes software developed by the University of
167c478bd9Sstevel@tonic-gate * California, Berkeley and its contributors.
177c478bd9Sstevel@tonic-gate * 4. Neither the name of the University nor the names of its contributors
187c478bd9Sstevel@tonic-gate * may be used to endorse or promote products derived from this software
197c478bd9Sstevel@tonic-gate * without specific prior written permission.
207c478bd9Sstevel@tonic-gate *
217c478bd9Sstevel@tonic-gate * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
227c478bd9Sstevel@tonic-gate * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
237c478bd9Sstevel@tonic-gate * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
247c478bd9Sstevel@tonic-gate * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
257c478bd9Sstevel@tonic-gate * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
267c478bd9Sstevel@tonic-gate * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
277c478bd9Sstevel@tonic-gate * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
287c478bd9Sstevel@tonic-gate * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
297c478bd9Sstevel@tonic-gate * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
307c478bd9Sstevel@tonic-gate * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
317c478bd9Sstevel@tonic-gate * SUCH DAMAGE.
327c478bd9Sstevel@tonic-gate */
337c478bd9Sstevel@tonic-gate
347c478bd9Sstevel@tonic-gate #if defined(LIBC_SCCS) && !defined(lint)
357c478bd9Sstevel@tonic-gate static char sccsid[] = "@(#)rec_close.c 8.9 (Berkeley) 11/18/94";
367c478bd9Sstevel@tonic-gate #endif /* LIBC_SCCS and not lint */
377c478bd9Sstevel@tonic-gate
387c478bd9Sstevel@tonic-gate #include <sys/types.h>
397c478bd9Sstevel@tonic-gate #include <sys/uio.h>
407c478bd9Sstevel@tonic-gate #ifdef RECNO_USE_MMAP
417c478bd9Sstevel@tonic-gate #include <sys/mman.h>
427c478bd9Sstevel@tonic-gate #endif
437c478bd9Sstevel@tonic-gate
447c478bd9Sstevel@tonic-gate #include <errno.h>
457c478bd9Sstevel@tonic-gate #include <limits.h>
467c478bd9Sstevel@tonic-gate #include <stdio.h>
477c478bd9Sstevel@tonic-gate #include <unistd.h>
487c478bd9Sstevel@tonic-gate
497c478bd9Sstevel@tonic-gate #include "db-int.h"
507c478bd9Sstevel@tonic-gate #include "recno.h"
517c478bd9Sstevel@tonic-gate
527c478bd9Sstevel@tonic-gate /*
537c478bd9Sstevel@tonic-gate * __REC_CLOSE -- Close a recno tree.
547c478bd9Sstevel@tonic-gate *
557c478bd9Sstevel@tonic-gate * Parameters:
567c478bd9Sstevel@tonic-gate * dbp: pointer to access method
577c478bd9Sstevel@tonic-gate *
587c478bd9Sstevel@tonic-gate * Returns:
597c478bd9Sstevel@tonic-gate * RET_ERROR, RET_SUCCESS
607c478bd9Sstevel@tonic-gate */
617c478bd9Sstevel@tonic-gate int
__rec_close(dbp)627c478bd9Sstevel@tonic-gate __rec_close(dbp)
637c478bd9Sstevel@tonic-gate DB *dbp;
647c478bd9Sstevel@tonic-gate {
657c478bd9Sstevel@tonic-gate BTREE *t;
667c478bd9Sstevel@tonic-gate int status;
677c478bd9Sstevel@tonic-gate
687c478bd9Sstevel@tonic-gate t = dbp->internal;
697c478bd9Sstevel@tonic-gate
707c478bd9Sstevel@tonic-gate /* Toss any page pinned across calls. */
717c478bd9Sstevel@tonic-gate if (t->bt_pinned != NULL) {
727c478bd9Sstevel@tonic-gate mpool_put(t->bt_mp, t->bt_pinned, 0);
737c478bd9Sstevel@tonic-gate t->bt_pinned = NULL;
747c478bd9Sstevel@tonic-gate }
757c478bd9Sstevel@tonic-gate
767c478bd9Sstevel@tonic-gate if (__rec_sync(dbp, 0) == RET_ERROR)
777c478bd9Sstevel@tonic-gate return (RET_ERROR);
787c478bd9Sstevel@tonic-gate
797c478bd9Sstevel@tonic-gate /* Committed to closing. */
807c478bd9Sstevel@tonic-gate status = RET_SUCCESS;
817c478bd9Sstevel@tonic-gate #ifdef RECNO_USE_MMAP
827c478bd9Sstevel@tonic-gate if (F_ISSET(t, R_MEMMAPPED) && munmap(t->bt_smap, t->bt_msize))
837c478bd9Sstevel@tonic-gate status = RET_ERROR;
847c478bd9Sstevel@tonic-gate #endif
857c478bd9Sstevel@tonic-gate
8656a424ccSmp if (!F_ISSET(t, R_INMEM)) {
877c478bd9Sstevel@tonic-gate if (F_ISSET(t, R_CLOSEFP)) {
887c478bd9Sstevel@tonic-gate if (fclose(t->bt_rfp))
897c478bd9Sstevel@tonic-gate status = RET_ERROR;
907c478bd9Sstevel@tonic-gate } else
917c478bd9Sstevel@tonic-gate if (close(t->bt_rfd))
927c478bd9Sstevel@tonic-gate status = RET_ERROR;
9356a424ccSmp }
947c478bd9Sstevel@tonic-gate
957c478bd9Sstevel@tonic-gate if (__bt_close(dbp) == RET_ERROR)
967c478bd9Sstevel@tonic-gate status = RET_ERROR;
977c478bd9Sstevel@tonic-gate
987c478bd9Sstevel@tonic-gate return (status);
997c478bd9Sstevel@tonic-gate }
1007c478bd9Sstevel@tonic-gate
1017c478bd9Sstevel@tonic-gate /*
1027c478bd9Sstevel@tonic-gate * __REC_SYNC -- sync the recno tree to disk.
1037c478bd9Sstevel@tonic-gate *
1047c478bd9Sstevel@tonic-gate * Parameters:
1057c478bd9Sstevel@tonic-gate * dbp: pointer to access method
1067c478bd9Sstevel@tonic-gate *
1077c478bd9Sstevel@tonic-gate * Returns:
1087c478bd9Sstevel@tonic-gate * RET_SUCCESS, RET_ERROR.
1097c478bd9Sstevel@tonic-gate */
1107c478bd9Sstevel@tonic-gate int
__rec_sync(dbp,flags)1117c478bd9Sstevel@tonic-gate __rec_sync(dbp, flags)
1127c478bd9Sstevel@tonic-gate const DB *dbp;
1137c478bd9Sstevel@tonic-gate u_int flags;
1147c478bd9Sstevel@tonic-gate {
1157c478bd9Sstevel@tonic-gate struct iovec iov[2];
1167c478bd9Sstevel@tonic-gate BTREE *t;
1177c478bd9Sstevel@tonic-gate DBT data, key;
1187c478bd9Sstevel@tonic-gate off_t off;
1197c478bd9Sstevel@tonic-gate recno_t scursor, trec;
1207c478bd9Sstevel@tonic-gate int status;
1217c478bd9Sstevel@tonic-gate
1227c478bd9Sstevel@tonic-gate t = dbp->internal;
1237c478bd9Sstevel@tonic-gate
1247c478bd9Sstevel@tonic-gate /* Toss any page pinned across calls. */
1257c478bd9Sstevel@tonic-gate if (t->bt_pinned != NULL) {
1267c478bd9Sstevel@tonic-gate mpool_put(t->bt_mp, t->bt_pinned, 0);
1277c478bd9Sstevel@tonic-gate t->bt_pinned = NULL;
1287c478bd9Sstevel@tonic-gate }
1297c478bd9Sstevel@tonic-gate
1307c478bd9Sstevel@tonic-gate if (flags == R_RECNOSYNC)
1317c478bd9Sstevel@tonic-gate return (__bt_sync(dbp, 0));
1327c478bd9Sstevel@tonic-gate
1337c478bd9Sstevel@tonic-gate if (F_ISSET(t, R_RDONLY | R_INMEM) || !F_ISSET(t, R_MODIFIED))
1347c478bd9Sstevel@tonic-gate return (RET_SUCCESS);
1357c478bd9Sstevel@tonic-gate
1367c478bd9Sstevel@tonic-gate /* Read any remaining records into the tree. */
1377c478bd9Sstevel@tonic-gate if (!F_ISSET(t, R_EOF) && t->bt_irec(t, MAX_REC_NUMBER) == RET_ERROR)
1387c478bd9Sstevel@tonic-gate return (RET_ERROR);
1397c478bd9Sstevel@tonic-gate
1407c478bd9Sstevel@tonic-gate /* Rewind the file descriptor. */
1417c478bd9Sstevel@tonic-gate if (lseek(t->bt_rfd, (off_t)0, SEEK_SET) != 0)
1427c478bd9Sstevel@tonic-gate return (RET_ERROR);
1437c478bd9Sstevel@tonic-gate
1447c478bd9Sstevel@tonic-gate /* Save the cursor. */
1457c478bd9Sstevel@tonic-gate scursor = t->bt_cursor.rcursor;
1467c478bd9Sstevel@tonic-gate
1477c478bd9Sstevel@tonic-gate key.size = sizeof(recno_t);
1487c478bd9Sstevel@tonic-gate key.data = &trec;
1497c478bd9Sstevel@tonic-gate
1507c478bd9Sstevel@tonic-gate if (F_ISSET(t, R_FIXLEN)) {
1517c478bd9Sstevel@tonic-gate /*
1527c478bd9Sstevel@tonic-gate * We assume that fixed length records are all fixed length.
1537c478bd9Sstevel@tonic-gate * Any that aren't are either EINVAL'd or corrected by the
1547c478bd9Sstevel@tonic-gate * record put code.
1557c478bd9Sstevel@tonic-gate */
1567c478bd9Sstevel@tonic-gate status = (dbp->seq)(dbp, &key, &data, R_FIRST);
1577c478bd9Sstevel@tonic-gate while (status == RET_SUCCESS) {
1587c478bd9Sstevel@tonic-gate if (write(t->bt_rfd, data.data, data.size) != data.size)
1597c478bd9Sstevel@tonic-gate return (RET_ERROR);
1607c478bd9Sstevel@tonic-gate status = (dbp->seq)(dbp, &key, &data, R_NEXT);
1617c478bd9Sstevel@tonic-gate }
1627c478bd9Sstevel@tonic-gate } else {
1637c478bd9Sstevel@tonic-gate iov[1].iov_base = (char *)&t->bt_bval;
1647c478bd9Sstevel@tonic-gate iov[1].iov_len = 1;
1657c478bd9Sstevel@tonic-gate
1667c478bd9Sstevel@tonic-gate status = (dbp->seq)(dbp, &key, &data, R_FIRST);
1677c478bd9Sstevel@tonic-gate while (status == RET_SUCCESS) {
1687c478bd9Sstevel@tonic-gate iov[0].iov_base = data.data;
1697c478bd9Sstevel@tonic-gate iov[0].iov_len = data.size;
1707c478bd9Sstevel@tonic-gate if (writev(t->bt_rfd, iov, 2) != data.size + 1)
1717c478bd9Sstevel@tonic-gate return (RET_ERROR);
1727c478bd9Sstevel@tonic-gate status = (dbp->seq)(dbp, &key, &data, R_NEXT);
1737c478bd9Sstevel@tonic-gate }
1747c478bd9Sstevel@tonic-gate }
1757c478bd9Sstevel@tonic-gate
1767c478bd9Sstevel@tonic-gate /* Restore the cursor. */
1777c478bd9Sstevel@tonic-gate t->bt_cursor.rcursor = scursor;
1787c478bd9Sstevel@tonic-gate
1797c478bd9Sstevel@tonic-gate if (status == RET_ERROR)
1807c478bd9Sstevel@tonic-gate return (RET_ERROR);
1817c478bd9Sstevel@tonic-gate if ((off = lseek(t->bt_rfd, (off_t)0, SEEK_CUR)) == -1)
1827c478bd9Sstevel@tonic-gate return (RET_ERROR);
1837c478bd9Sstevel@tonic-gate if (ftruncate(t->bt_rfd, off))
1847c478bd9Sstevel@tonic-gate return (RET_ERROR);
1857c478bd9Sstevel@tonic-gate F_CLR(t, R_MODIFIED);
1867c478bd9Sstevel@tonic-gate return (RET_SUCCESS);
1877c478bd9Sstevel@tonic-gate }
188