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  * This code is derived from software contributed to Berkeley by
67c478bd9Sstevel@tonic-gate  * Mike Olson.
77c478bd9Sstevel@tonic-gate  *
87c478bd9Sstevel@tonic-gate  * Redistribution and use in source and binary forms, with or without
97c478bd9Sstevel@tonic-gate  * modification, are permitted provided that the following conditions
107c478bd9Sstevel@tonic-gate  * are met:
117c478bd9Sstevel@tonic-gate  * 1. Redistributions of source code must retain the above copyright
127c478bd9Sstevel@tonic-gate  *    notice, this list of conditions and the following disclaimer.
137c478bd9Sstevel@tonic-gate  * 2. Redistributions in binary form must reproduce the above copyright
147c478bd9Sstevel@tonic-gate  *    notice, this list of conditions and the following disclaimer in the
157c478bd9Sstevel@tonic-gate  *    documentation and/or other materials provided with the distribution.
167c478bd9Sstevel@tonic-gate  * 3. All advertising materials mentioning features or use of this software
177c478bd9Sstevel@tonic-gate  *    must display the following acknowledgement:
187c478bd9Sstevel@tonic-gate  *	This product includes software developed by the University of
197c478bd9Sstevel@tonic-gate  *	California, Berkeley and its contributors.
207c478bd9Sstevel@tonic-gate  * 4. Neither the name of the University nor the names of its contributors
217c478bd9Sstevel@tonic-gate  *    may be used to endorse or promote products derived from this software
227c478bd9Sstevel@tonic-gate  *    without specific prior written permission.
237c478bd9Sstevel@tonic-gate  *
247c478bd9Sstevel@tonic-gate  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
257c478bd9Sstevel@tonic-gate  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
267c478bd9Sstevel@tonic-gate  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
277c478bd9Sstevel@tonic-gate  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
287c478bd9Sstevel@tonic-gate  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
297c478bd9Sstevel@tonic-gate  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
307c478bd9Sstevel@tonic-gate  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
317c478bd9Sstevel@tonic-gate  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
327c478bd9Sstevel@tonic-gate  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
337c478bd9Sstevel@tonic-gate  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
347c478bd9Sstevel@tonic-gate  * SUCH DAMAGE.
357c478bd9Sstevel@tonic-gate  */
367c478bd9Sstevel@tonic-gate 
377c478bd9Sstevel@tonic-gate #if defined(LIBC_SCCS) && !defined(lint)
387c478bd9Sstevel@tonic-gate static char sccsid[] = "@(#)bt_close.c	8.7 (Berkeley) 8/17/94";
397c478bd9Sstevel@tonic-gate #endif /* LIBC_SCCS and not lint */
407c478bd9Sstevel@tonic-gate 
417c478bd9Sstevel@tonic-gate #include <sys/param.h>
427c478bd9Sstevel@tonic-gate 
437c478bd9Sstevel@tonic-gate #include <errno.h>
447c478bd9Sstevel@tonic-gate #include <stdio.h>
457c478bd9Sstevel@tonic-gate #include <stdlib.h>
467c478bd9Sstevel@tonic-gate #include <string.h>
477c478bd9Sstevel@tonic-gate #include <unistd.h>
487c478bd9Sstevel@tonic-gate 
497c478bd9Sstevel@tonic-gate #include "db-int.h"
507c478bd9Sstevel@tonic-gate #include "btree.h"
517c478bd9Sstevel@tonic-gate 
527c478bd9Sstevel@tonic-gate static int bt_meta __P((BTREE *));
537c478bd9Sstevel@tonic-gate 
547c478bd9Sstevel@tonic-gate /*
557c478bd9Sstevel@tonic-gate  * BT_CLOSE -- Close a btree.
567c478bd9Sstevel@tonic-gate  *
577c478bd9Sstevel@tonic-gate  * Parameters:
587c478bd9Sstevel@tonic-gate  *	dbp:	pointer to access method
597c478bd9Sstevel@tonic-gate  *
607c478bd9Sstevel@tonic-gate  * Returns:
617c478bd9Sstevel@tonic-gate  *	RET_ERROR, RET_SUCCESS
627c478bd9Sstevel@tonic-gate  */
637c478bd9Sstevel@tonic-gate int
__bt_close(dbp)647c478bd9Sstevel@tonic-gate __bt_close(dbp)
657c478bd9Sstevel@tonic-gate 	DB *dbp;
667c478bd9Sstevel@tonic-gate {
677c478bd9Sstevel@tonic-gate 	BTREE *t;
687c478bd9Sstevel@tonic-gate 	int fd;
697c478bd9Sstevel@tonic-gate 
707c478bd9Sstevel@tonic-gate 	t = dbp->internal;
717c478bd9Sstevel@tonic-gate 
727c478bd9Sstevel@tonic-gate 	/* Toss any page pinned across calls. */
737c478bd9Sstevel@tonic-gate 	if (t->bt_pinned != NULL) {
747c478bd9Sstevel@tonic-gate 		mpool_put(t->bt_mp, t->bt_pinned, 0);
757c478bd9Sstevel@tonic-gate 		t->bt_pinned = NULL;
767c478bd9Sstevel@tonic-gate 	}
777c478bd9Sstevel@tonic-gate 
787c478bd9Sstevel@tonic-gate 	/* Sync the tree. */
797c478bd9Sstevel@tonic-gate 	if (__bt_sync(dbp, 0) == RET_ERROR)
807c478bd9Sstevel@tonic-gate 		return (RET_ERROR);
817c478bd9Sstevel@tonic-gate 
827c478bd9Sstevel@tonic-gate 	/* Close the memory pool. */
837c478bd9Sstevel@tonic-gate 	if (mpool_close(t->bt_mp) == RET_ERROR)
847c478bd9Sstevel@tonic-gate 		return (RET_ERROR);
857c478bd9Sstevel@tonic-gate 
867c478bd9Sstevel@tonic-gate 	/* Free random memory. */
877c478bd9Sstevel@tonic-gate 	if (t->bt_cursor.key.data != NULL) {
887c478bd9Sstevel@tonic-gate 		free(t->bt_cursor.key.data);
897c478bd9Sstevel@tonic-gate 		t->bt_cursor.key.size = 0;
907c478bd9Sstevel@tonic-gate 		t->bt_cursor.key.data = NULL;
917c478bd9Sstevel@tonic-gate 	}
927c478bd9Sstevel@tonic-gate 	if (t->bt_rkey.data) {
937c478bd9Sstevel@tonic-gate 		free(t->bt_rkey.data);
947c478bd9Sstevel@tonic-gate 		t->bt_rkey.size = 0;
957c478bd9Sstevel@tonic-gate 		t->bt_rkey.data = NULL;
967c478bd9Sstevel@tonic-gate 	}
977c478bd9Sstevel@tonic-gate 	if (t->bt_rdata.data) {
987c478bd9Sstevel@tonic-gate 		free(t->bt_rdata.data);
997c478bd9Sstevel@tonic-gate 		t->bt_rdata.size = 0;
1007c478bd9Sstevel@tonic-gate 		t->bt_rdata.data = NULL;
1017c478bd9Sstevel@tonic-gate 	}
1027c478bd9Sstevel@tonic-gate 
1037c478bd9Sstevel@tonic-gate 	fd = t->bt_fd;
1047c478bd9Sstevel@tonic-gate 	free(t);
1057c478bd9Sstevel@tonic-gate 	free(dbp);
1067c478bd9Sstevel@tonic-gate 	return (close(fd) ? RET_ERROR : RET_SUCCESS);
1077c478bd9Sstevel@tonic-gate }
1087c478bd9Sstevel@tonic-gate 
1097c478bd9Sstevel@tonic-gate /*
1107c478bd9Sstevel@tonic-gate  * BT_SYNC -- sync the btree to disk.
1117c478bd9Sstevel@tonic-gate  *
1127c478bd9Sstevel@tonic-gate  * Parameters:
1137c478bd9Sstevel@tonic-gate  *	dbp:	pointer to access method
1147c478bd9Sstevel@tonic-gate  *
1157c478bd9Sstevel@tonic-gate  * Returns:
1167c478bd9Sstevel@tonic-gate  *	RET_SUCCESS, RET_ERROR.
1177c478bd9Sstevel@tonic-gate  */
1187c478bd9Sstevel@tonic-gate int
__bt_sync(dbp,flags)1197c478bd9Sstevel@tonic-gate __bt_sync(dbp, flags)
1207c478bd9Sstevel@tonic-gate 	const DB *dbp;
1217c478bd9Sstevel@tonic-gate 	u_int flags;
1227c478bd9Sstevel@tonic-gate {
1237c478bd9Sstevel@tonic-gate 	BTREE *t;
1247c478bd9Sstevel@tonic-gate 	int status;
1257c478bd9Sstevel@tonic-gate 
1267c478bd9Sstevel@tonic-gate 	t = dbp->internal;
1277c478bd9Sstevel@tonic-gate 
1287c478bd9Sstevel@tonic-gate 	/* Toss any page pinned across calls. */
1297c478bd9Sstevel@tonic-gate 	if (t->bt_pinned != NULL) {
1307c478bd9Sstevel@tonic-gate 		mpool_put(t->bt_mp, t->bt_pinned, 0);
1317c478bd9Sstevel@tonic-gate 		t->bt_pinned = NULL;
1327c478bd9Sstevel@tonic-gate 	}
1337c478bd9Sstevel@tonic-gate 
1347c478bd9Sstevel@tonic-gate 	/* Sync doesn't currently take any flags. */
1357c478bd9Sstevel@tonic-gate 	if (flags != 0) {
1367c478bd9Sstevel@tonic-gate 		errno = EINVAL;
1377c478bd9Sstevel@tonic-gate 		return (RET_ERROR);
1387c478bd9Sstevel@tonic-gate 	}
1397c478bd9Sstevel@tonic-gate 
1407c478bd9Sstevel@tonic-gate 	if (F_ISSET(t, B_INMEM | B_RDONLY)
1417c478bd9Sstevel@tonic-gate 	    || !F_ISSET(t, B_MODIFIED | B_METADIRTY))
1427c478bd9Sstevel@tonic-gate 		return (RET_SUCCESS);
1437c478bd9Sstevel@tonic-gate 
1447c478bd9Sstevel@tonic-gate 	if (F_ISSET(t, B_METADIRTY) && bt_meta(t) == RET_ERROR)
1457c478bd9Sstevel@tonic-gate 		return (RET_ERROR);
1467c478bd9Sstevel@tonic-gate 
1477c478bd9Sstevel@tonic-gate 	if ((status = mpool_sync(t->bt_mp)) == RET_SUCCESS)
1487c478bd9Sstevel@tonic-gate 		F_CLR(t, B_MODIFIED);
1497c478bd9Sstevel@tonic-gate 
1507c478bd9Sstevel@tonic-gate 	return (status);
1517c478bd9Sstevel@tonic-gate }
1527c478bd9Sstevel@tonic-gate 
1537c478bd9Sstevel@tonic-gate /*
1547c478bd9Sstevel@tonic-gate  * BT_META -- write the tree meta data to disk.
1557c478bd9Sstevel@tonic-gate  *
1567c478bd9Sstevel@tonic-gate  * Parameters:
1577c478bd9Sstevel@tonic-gate  *	t:	tree
1587c478bd9Sstevel@tonic-gate  *
1597c478bd9Sstevel@tonic-gate  * Returns:
1607c478bd9Sstevel@tonic-gate  *	RET_ERROR, RET_SUCCESS
1617c478bd9Sstevel@tonic-gate  */
1627c478bd9Sstevel@tonic-gate static int
bt_meta(t)1637c478bd9Sstevel@tonic-gate bt_meta(t)
1647c478bd9Sstevel@tonic-gate 	BTREE *t;
1657c478bd9Sstevel@tonic-gate {
1667c478bd9Sstevel@tonic-gate 	BTMETA m;
1677c478bd9Sstevel@tonic-gate 	void *p;
1687c478bd9Sstevel@tonic-gate 
1697c478bd9Sstevel@tonic-gate 	if ((p = mpool_get(t->bt_mp, P_META, 0)) == NULL)
1707c478bd9Sstevel@tonic-gate 		return (RET_ERROR);
1717c478bd9Sstevel@tonic-gate 
1727c478bd9Sstevel@tonic-gate 	/* Fill in metadata. */
1737c478bd9Sstevel@tonic-gate 	m.magic = BTREEMAGIC;
1747c478bd9Sstevel@tonic-gate 	m.version = BTREEVERSION;
1757c478bd9Sstevel@tonic-gate 	m.psize = t->bt_psize;
1767c478bd9Sstevel@tonic-gate 	m.free = t->bt_free;
1777c478bd9Sstevel@tonic-gate 	m.nrecs = t->bt_nrecs;
1787c478bd9Sstevel@tonic-gate 	m.flags = F_ISSET(t, SAVEMETA);
1797c478bd9Sstevel@tonic-gate 
1807c478bd9Sstevel@tonic-gate 	memmove(p, &m, sizeof(BTMETA));
1817c478bd9Sstevel@tonic-gate 	mpool_put(t->bt_mp, p, MPOOL_DIRTY);
1827c478bd9Sstevel@tonic-gate 	return (RET_SUCCESS);
1837c478bd9Sstevel@tonic-gate }
184