Blame src/plugins/kdb/db2/libdb2/btree/bt_split.c

Packit fd8b60
/*-
Packit fd8b60
 * Copyright (c) 1990, 1993, 1994
Packit fd8b60
 *	The Regents of the University of California.  All rights reserved.
Packit fd8b60
 *
Packit fd8b60
 * This code is derived from software contributed to Berkeley by
Packit fd8b60
 * Mike Olson.
Packit fd8b60
 *
Packit fd8b60
 * Redistribution and use in source and binary forms, with or without
Packit fd8b60
 * modification, are permitted provided that the following conditions
Packit fd8b60
 * are met:
Packit fd8b60
 * 1. Redistributions of source code must retain the above copyright
Packit fd8b60
 *    notice, this list of conditions and the following disclaimer.
Packit fd8b60
 * 2. Redistributions in binary form must reproduce the above copyright
Packit fd8b60
 *    notice, this list of conditions and the following disclaimer in the
Packit fd8b60
 *    documentation and/or other materials provided with the distribution.
Packit fd8b60
 * 3. All advertising materials mentioning features or use of this software
Packit fd8b60
 *    must display the following acknowledgement:
Packit fd8b60
 *	This product includes software developed by the University of
Packit fd8b60
 *	California, Berkeley and its contributors.
Packit fd8b60
 * 4. Neither the name of the University nor the names of its contributors
Packit fd8b60
 *    may be used to endorse or promote products derived from this software
Packit fd8b60
 *    without specific prior written permission.
Packit fd8b60
 *
Packit fd8b60
 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
Packit fd8b60
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
Packit fd8b60
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
Packit fd8b60
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
Packit fd8b60
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
Packit fd8b60
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
Packit fd8b60
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
Packit fd8b60
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
Packit fd8b60
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
Packit fd8b60
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
Packit fd8b60
 * SUCH DAMAGE.
Packit fd8b60
 */
Packit fd8b60
Packit fd8b60
#if defined(LIBC_SCCS) && !defined(lint)
Packit fd8b60
static char sccsid[] = "@(#)bt_split.c	8.10 (Berkeley) 1/9/95";
Packit fd8b60
#endif /* LIBC_SCCS and not lint */
Packit fd8b60
Packit fd8b60
#include <sys/types.h>
Packit fd8b60
Packit fd8b60
#include <limits.h>
Packit fd8b60
#include <stdio.h>
Packit fd8b60
#include <stdlib.h>
Packit fd8b60
#include <string.h>
Packit fd8b60
Packit fd8b60
#include "db-int.h"
Packit fd8b60
#include "btree.h"
Packit fd8b60
Packit fd8b60
static int	 bt_broot __P((BTREE *, PAGE *, PAGE *, PAGE *));
Packit fd8b60
static PAGE	*bt_page
Packit fd8b60
		    __P((BTREE *, PAGE *, PAGE **, PAGE **, indx_t *, size_t));
Packit fd8b60
static int	 bt_preserve __P((BTREE *, db_pgno_t));
Packit fd8b60
static PAGE	*bt_psplit
Packit fd8b60
		    __P((BTREE *, PAGE *, PAGE *, PAGE *, indx_t *, size_t));
Packit fd8b60
static PAGE	*bt_root
Packit fd8b60
		    __P((BTREE *, PAGE *, PAGE **, PAGE **, indx_t *, size_t));
Packit fd8b60
static int	 bt_rroot __P((BTREE *, PAGE *, PAGE *, PAGE *));
Packit fd8b60
static recno_t	 rec_total __P((PAGE *));
Packit fd8b60
Packit fd8b60
#ifdef STATISTICS
Packit fd8b60
u_long	bt_rootsplit, bt_split, bt_sortsplit, bt_pfxsaved;
Packit fd8b60
#endif
Packit fd8b60
Packit fd8b60
/*
Packit fd8b60
 * __BT_SPLIT -- Split the tree.
Packit fd8b60
 *
Packit fd8b60
 * Parameters:
Packit fd8b60
 *	t:	tree
Packit fd8b60
 *	sp:	page to split
Packit fd8b60
 *	key:	key to insert
Packit fd8b60
 *	data:	data to insert
Packit fd8b60
 *	flags:	BIGKEY/BIGDATA flags
Packit fd8b60
 *	ilen:	insert length
Packit fd8b60
 *	skip:	index to leave open
Packit fd8b60
 *
Packit fd8b60
 * Returns:
Packit fd8b60
 *	RET_ERROR, RET_SUCCESS
Packit fd8b60
 */
Packit fd8b60
int
Packit fd8b60
__bt_split(t, sp, key, data, flags, ilen, argskip)
Packit fd8b60
	BTREE *t;
Packit fd8b60
	PAGE *sp;
Packit fd8b60
	const DBT *key, *data;
Packit fd8b60
	int flags;
Packit fd8b60
	size_t ilen;
Packit fd8b60
	u_int32_t argskip;
Packit fd8b60
{
Packit fd8b60
	BINTERNAL *bi = NULL;
Packit fd8b60
	BLEAF *bl = NULL, *tbl;
Packit fd8b60
	DBT a, b;
Packit fd8b60
	EPGNO *parent;
Packit fd8b60
	PAGE *h, *l, *r, *lchild, *rchild;
Packit fd8b60
	indx_t nxtindex;
Packit fd8b60
	u_int16_t skip;
Packit fd8b60
	u_int32_t n, nbytes, nksize = 0;
Packit fd8b60
	int parentsplit;
Packit fd8b60
	char *dest;
Packit fd8b60
Packit fd8b60
	/*
Packit fd8b60
	 * Split the page into two pages, l and r.  The split routines return
Packit fd8b60
	 * a pointer to the page into which the key should be inserted and with
Packit fd8b60
	 * skip set to the offset which should be used.  Additionally, l and r
Packit fd8b60
	 * are pinned.
Packit fd8b60
	 */
Packit fd8b60
	skip = argskip;
Packit fd8b60
	h = sp->pgno == P_ROOT ?
Packit fd8b60
	    bt_root(t, sp, &l, &r, &skip, ilen) :
Packit fd8b60
	    bt_page(t, sp, &l, &r, &skip, ilen);
Packit fd8b60
	if (h == NULL)
Packit fd8b60
		return (RET_ERROR);
Packit fd8b60
Packit fd8b60
	/*
Packit fd8b60
	 * Insert the new key/data pair into the leaf page.  (Key inserts
Packit fd8b60
	 * always cause a leaf page to split first.)
Packit fd8b60
	 */
Packit fd8b60
	h->linp[skip] = h->upper -= ilen;
Packit fd8b60
	dest = (char *)h + h->upper;
Packit fd8b60
	if (F_ISSET(t, R_RECNO))
Packit fd8b60
		WR_RLEAF(dest, data, flags)
Packit fd8b60
	else
Packit fd8b60
		WR_BLEAF(dest, key, data, flags)
Packit fd8b60
Packit fd8b60
	/* If the root page was split, make it look right. */
Packit fd8b60
	if (sp->pgno == P_ROOT &&
Packit fd8b60
	    (F_ISSET(t, R_RECNO) ?
Packit fd8b60
	    bt_rroot(t, sp, l, r) : bt_broot(t, sp, l, r)) == RET_ERROR)
Packit fd8b60
		goto err2;
Packit fd8b60
Packit fd8b60
	/*
Packit fd8b60
	 * Now we walk the parent page stack -- a LIFO stack of the pages that
Packit fd8b60
	 * were traversed when we searched for the page that split.  Each stack
Packit fd8b60
	 * entry is a page number and a page index offset.  The offset is for
Packit fd8b60
	 * the page traversed on the search.  We've just split a page, so we
Packit fd8b60
	 * have to insert a new key into the parent page.
Packit fd8b60
	 *
Packit fd8b60
	 * If the insert into the parent page causes it to split, may have to
Packit fd8b60
	 * continue splitting all the way up the tree.  We stop if the root
Packit fd8b60
	 * splits or the page inserted into didn't have to split to hold the
Packit fd8b60
	 * new key.  Some algorithms replace the key for the old page as well
Packit fd8b60
	 * as the new page.  We don't, as there's no reason to believe that the
Packit fd8b60
	 * first key on the old page is any better than the key we have, and,
Packit fd8b60
	 * in the case of a key being placed at index 0 causing the split, the
Packit fd8b60
	 * key is unavailable.
Packit fd8b60
	 *
Packit fd8b60
	 * There are a maximum of 5 pages pinned at any time.  We keep the left
Packit fd8b60
	 * and right pages pinned while working on the parent.   The 5 are the
Packit fd8b60
	 * two children, left parent and right parent (when the parent splits)
Packit fd8b60
	 * and the root page or the overflow key page when calling bt_preserve.
Packit fd8b60
	 * This code must make sure that all pins are released other than the
Packit fd8b60
	 * root page or overflow page which is unlocked elsewhere.
Packit fd8b60
	 */
Packit fd8b60
	while ((parent = BT_POP(t)) != NULL) {
Packit fd8b60
		lchild = l;
Packit fd8b60
		rchild = r;
Packit fd8b60
Packit fd8b60
		/* Get the parent page. */
Packit fd8b60
		if ((h = mpool_get(t->bt_mp, parent->pgno, 0)) == NULL)
Packit fd8b60
			goto err2;
Packit fd8b60
Packit fd8b60
	 	/*
Packit fd8b60
		 * The new key goes ONE AFTER the index, because the split
Packit fd8b60
		 * was to the right.
Packit fd8b60
		 */
Packit fd8b60
		skip = parent->index + 1;
Packit fd8b60
Packit fd8b60
		/*
Packit fd8b60
		 * Calculate the space needed on the parent page.
Packit fd8b60
		 *
Packit fd8b60
		 * Prefix trees: space hack when inserting into BINTERNAL
Packit fd8b60
		 * pages.  Retain only what's needed to distinguish between
Packit fd8b60
		 * the new entry and the LAST entry on the page to its left.
Packit fd8b60
		 * If the keys compare equal, retain the entire key.  Note,
Packit fd8b60
		 * we don't touch overflow keys, and the entire key must be
Packit fd8b60
		 * retained for the next-to-left most key on the leftmost
Packit fd8b60
		 * page of each level, or the search will fail.  Applicable
Packit fd8b60
		 * ONLY to internal pages that have leaf pages as children.
Packit fd8b60
		 * Further reduction of the key between pairs of internal
Packit fd8b60
		 * pages loses too much information.
Packit fd8b60
		 */
Packit fd8b60
		switch (rchild->flags & P_TYPE) {
Packit fd8b60
		case P_BINTERNAL:
Packit fd8b60
			bi = GETBINTERNAL(rchild, 0);
Packit fd8b60
			nbytes = NBINTERNAL(bi->ksize);
Packit fd8b60
			break;
Packit fd8b60
		case P_BLEAF:
Packit fd8b60
			bl = GETBLEAF(rchild, 0);
Packit fd8b60
			nbytes = NBINTERNAL(bl->ksize);
Packit fd8b60
			if (t->bt_pfx && !(bl->flags & P_BIGKEY) &&
Packit fd8b60
			    (h->prevpg != P_INVALID || skip > 1)) {
Packit fd8b60
				tbl = GETBLEAF(lchild, NEXTINDEX(lchild) - 1);
Packit fd8b60
				a.size = tbl->ksize;
Packit fd8b60
				a.data = tbl->bytes;
Packit fd8b60
				b.size = bl->ksize;
Packit fd8b60
				b.data = bl->bytes;
Packit fd8b60
				nksize = t->bt_pfx(&a, &b);
Packit fd8b60
				n = NBINTERNAL(nksize);
Packit fd8b60
				if (n < nbytes) {
Packit fd8b60
#ifdef STATISTICS
Packit fd8b60
					bt_pfxsaved += nbytes - n;
Packit fd8b60
#endif
Packit fd8b60
					nbytes = n;
Packit fd8b60
				} else
Packit fd8b60
					nksize = 0;
Packit fd8b60
			} else
Packit fd8b60
				nksize = 0;
Packit fd8b60
			break;
Packit fd8b60
		case P_RINTERNAL:
Packit fd8b60
		case P_RLEAF:
Packit fd8b60
			nbytes = NRINTERNAL;
Packit fd8b60
			break;
Packit fd8b60
		default:
Packit fd8b60
			abort();
Packit fd8b60
		}
Packit fd8b60
Packit fd8b60
		/* Split the parent page if necessary or shift the indices. */
Packit fd8b60
		if ((u_int32_t)h->upper - (u_int32_t)h->lower
Packit fd8b60
		    < nbytes + sizeof(indx_t)) {
Packit fd8b60
			sp = h;
Packit fd8b60
			h = h->pgno == P_ROOT ?
Packit fd8b60
			    bt_root(t, h, &l, &r, &skip, nbytes) :
Packit fd8b60
			    bt_page(t, h, &l, &r, &skip, nbytes);
Packit fd8b60
			if (h == NULL)
Packit fd8b60
				goto err1;
Packit fd8b60
			parentsplit = 1;
Packit fd8b60
		} else {
Packit fd8b60
			if (skip < (nxtindex = NEXTINDEX(h)))
Packit fd8b60
				memmove(h->linp + skip + 1, h->linp + skip,
Packit fd8b60
				    (nxtindex - skip) * sizeof(indx_t));
Packit fd8b60
			h->lower += sizeof(indx_t);
Packit fd8b60
			parentsplit = 0;
Packit fd8b60
		}
Packit fd8b60
Packit fd8b60
		/* Insert the key into the parent page. */
Packit fd8b60
		switch (rchild->flags & P_TYPE) {
Packit fd8b60
		case P_BINTERNAL:
Packit fd8b60
			h->linp[skip] = h->upper -= nbytes;
Packit fd8b60
			dest = (char *)h + h->linp[skip];
Packit fd8b60
			memmove(dest, bi, nbytes);
Packit fd8b60
			((BINTERNAL *)(void *)dest)->pgno = rchild->pgno;
Packit fd8b60
			break;
Packit fd8b60
		case P_BLEAF:
Packit fd8b60
			h->linp[skip] = h->upper -= nbytes;
Packit fd8b60
			dest = (char *)h + h->linp[skip];
Packit fd8b60
			WR_BINTERNAL(dest, nksize ? nksize : bl->ksize,
Packit fd8b60
			    rchild->pgno, bl->flags & P_BIGKEY);
Packit fd8b60
			memmove(dest, bl->bytes, nksize ? nksize : bl->ksize);
Packit fd8b60
			if (bl->flags & P_BIGKEY) {
Packit fd8b60
				db_pgno_t pgno;
Packit fd8b60
				memcpy(&pgno, bl->bytes, sizeof(pgno));
Packit fd8b60
				if (bt_preserve(t, pgno) == RET_ERROR)
Packit fd8b60
					goto err1;
Packit fd8b60
			}
Packit fd8b60
			break;
Packit fd8b60
		case P_RINTERNAL:
Packit fd8b60
			/*
Packit fd8b60
			 * Update the left page count.  If split
Packit fd8b60
			 * added at index 0, fix the correct page.
Packit fd8b60
			 */
Packit fd8b60
			if (skip > 0)
Packit fd8b60
				dest = (char *)h + h->linp[skip - 1];
Packit fd8b60
			else
Packit fd8b60
				dest = (char *)l + l->linp[NEXTINDEX(l) - 1];
Packit fd8b60
			((RINTERNAL *)(void *)dest)->nrecs = rec_total(lchild);
Packit fd8b60
			((RINTERNAL *)(void *)dest)->pgno = lchild->pgno;
Packit fd8b60
Packit fd8b60
			/* Update the right page count. */
Packit fd8b60
			h->linp[skip] = h->upper -= nbytes;
Packit fd8b60
			dest = (char *)h + h->linp[skip];
Packit fd8b60
			((RINTERNAL *)(void *)dest)->nrecs = rec_total(rchild);
Packit fd8b60
			((RINTERNAL *)(void *)dest)->pgno = rchild->pgno;
Packit fd8b60
			break;
Packit fd8b60
		case P_RLEAF:
Packit fd8b60
			/*
Packit fd8b60
			 * Update the left page count.  If split
Packit fd8b60
			 * added at index 0, fix the correct page.
Packit fd8b60
			 */
Packit fd8b60
			if (skip > 0)
Packit fd8b60
				dest = (char *)h + h->linp[skip - 1];
Packit fd8b60
			else
Packit fd8b60
				dest = (char *)l + l->linp[NEXTINDEX(l) - 1];
Packit fd8b60
			((RINTERNAL *)(void *)dest)->nrecs = NEXTINDEX(lchild);
Packit fd8b60
			((RINTERNAL *)(void *)dest)->pgno = lchild->pgno;
Packit fd8b60
Packit fd8b60
			/* Update the right page count. */
Packit fd8b60
			h->linp[skip] = h->upper -= nbytes;
Packit fd8b60
			dest = (char *)h + h->linp[skip];
Packit fd8b60
			((RINTERNAL *)(void *)dest)->nrecs = NEXTINDEX(rchild);
Packit fd8b60
			((RINTERNAL *)(void *)dest)->pgno = rchild->pgno;
Packit fd8b60
			break;
Packit fd8b60
		default:
Packit fd8b60
			abort();
Packit fd8b60
		}
Packit fd8b60
Packit fd8b60
		/* Unpin the held pages. */
Packit fd8b60
		if (!parentsplit) {
Packit fd8b60
			mpool_put(t->bt_mp, h, MPOOL_DIRTY);
Packit fd8b60
			break;
Packit fd8b60
		}
Packit fd8b60
Packit fd8b60
		/* If the root page was split, make it look right. */
Packit fd8b60
		if (sp->pgno == P_ROOT &&
Packit fd8b60
		    (F_ISSET(t, R_RECNO) ?
Packit fd8b60
		    bt_rroot(t, sp, l, r) : bt_broot(t, sp, l, r)) == RET_ERROR)
Packit fd8b60
			goto err1;
Packit fd8b60
Packit fd8b60
		mpool_put(t->bt_mp, lchild, MPOOL_DIRTY);
Packit fd8b60
		mpool_put(t->bt_mp, rchild, MPOOL_DIRTY);
Packit fd8b60
	}
Packit fd8b60
Packit fd8b60
	/* Unpin the held pages. */
Packit fd8b60
	mpool_put(t->bt_mp, l, MPOOL_DIRTY);
Packit fd8b60
	mpool_put(t->bt_mp, r, MPOOL_DIRTY);
Packit fd8b60
Packit fd8b60
	/* Clear any pages left on the stack. */
Packit fd8b60
	return (RET_SUCCESS);
Packit fd8b60
Packit fd8b60
	/*
Packit fd8b60
	 * If something fails in the above loop we were already walking back
Packit fd8b60
	 * up the tree and the tree is now inconsistent.  Nothing much we can
Packit fd8b60
	 * do about it but release any memory we're holding.
Packit fd8b60
	 */
Packit fd8b60
err1:	mpool_put(t->bt_mp, lchild, MPOOL_DIRTY);
Packit fd8b60
	mpool_put(t->bt_mp, rchild, MPOOL_DIRTY);
Packit fd8b60
Packit fd8b60
err2:	mpool_put(t->bt_mp, l, 0);
Packit fd8b60
	mpool_put(t->bt_mp, r, 0);
Packit fd8b60
	__dbpanic(t->bt_dbp);
Packit fd8b60
	return (RET_ERROR);
Packit fd8b60
}
Packit fd8b60
Packit fd8b60
/*
Packit fd8b60
 * BT_PAGE -- Split a non-root page of a btree.
Packit fd8b60
 *
Packit fd8b60
 * Parameters:
Packit fd8b60
 *	t:	tree
Packit fd8b60
 *	h:	root page
Packit fd8b60
 *	lp:	pointer to left page pointer
Packit fd8b60
 *	rp:	pointer to right page pointer
Packit fd8b60
 *	skip:	pointer to index to leave open
Packit fd8b60
 *	ilen:	insert length
Packit fd8b60
 *
Packit fd8b60
 * Returns:
Packit fd8b60
 *	Pointer to page in which to insert or NULL on error.
Packit fd8b60
 */
Packit fd8b60
static PAGE *
Packit fd8b60
bt_page(t, h, lp, rp, skip, ilen)
Packit fd8b60
	BTREE *t;
Packit fd8b60
	PAGE *h, **lp, **rp;
Packit fd8b60
	indx_t *skip;
Packit fd8b60
	size_t ilen;
Packit fd8b60
{
Packit fd8b60
	PAGE *l, *r, *tp;
Packit fd8b60
	db_pgno_t npg;
Packit fd8b60
Packit fd8b60
#ifdef STATISTICS
Packit fd8b60
	++bt_split;
Packit fd8b60
#endif
Packit fd8b60
	/* Put the new right page for the split into place. */
Packit fd8b60
	if ((r = __bt_new(t, &npg)) == NULL)
Packit fd8b60
		return (NULL);
Packit fd8b60
	r->pgno = npg;
Packit fd8b60
	r->lower = BTDATAOFF;
Packit fd8b60
	r->upper = t->bt_psize;
Packit fd8b60
	r->nextpg = h->nextpg;
Packit fd8b60
	r->prevpg = h->pgno;
Packit fd8b60
	r->flags = h->flags & P_TYPE;
Packit fd8b60
Packit fd8b60
	/*
Packit fd8b60
	 * If we're splitting the last page on a level because we're appending
Packit fd8b60
	 * a key to it (skip is NEXTINDEX()), it's likely that the data is
Packit fd8b60
	 * sorted.  Adding an empty page on the side of the level is less work
Packit fd8b60
	 * and can push the fill factor much higher than normal.  If we're
Packit fd8b60
	 * wrong it's no big deal, we'll just do the split the right way next
Packit fd8b60
	 * time.  It may look like it's equally easy to do a similar hack for
Packit fd8b60
	 * reverse sorted data, that is, split the tree left, but it's not.
Packit fd8b60
	 * Don't even try.
Packit fd8b60
	 */
Packit fd8b60
	if (h->nextpg == P_INVALID && *skip == NEXTINDEX(h)) {
Packit fd8b60
#ifdef STATISTICS
Packit fd8b60
		++bt_sortsplit;
Packit fd8b60
#endif
Packit fd8b60
		h->nextpg = r->pgno;
Packit fd8b60
		r->lower = BTDATAOFF + sizeof(indx_t);
Packit fd8b60
		*skip = 0;
Packit fd8b60
		*lp = h;
Packit fd8b60
		*rp = r;
Packit fd8b60
		return (r);
Packit fd8b60
	}
Packit fd8b60
Packit fd8b60
	/* Put the new left page for the split into place. */
Packit fd8b60
	if ((l = (PAGE *)malloc(t->bt_psize)) == NULL) {
Packit fd8b60
		mpool_put(t->bt_mp, r, 0);
Packit fd8b60
		return (NULL);
Packit fd8b60
	}
Packit fd8b60
#if 1 /* def PURIFY */
Packit fd8b60
	memset(l, 0xff, t->bt_psize);
Packit fd8b60
#endif
Packit fd8b60
	l->pgno = h->pgno;
Packit fd8b60
	l->nextpg = r->pgno;
Packit fd8b60
	l->prevpg = h->prevpg;
Packit fd8b60
	l->lower = BTDATAOFF;
Packit fd8b60
	l->upper = t->bt_psize;
Packit fd8b60
	l->flags = h->flags & P_TYPE;
Packit fd8b60
Packit fd8b60
	/* Fix up the previous pointer of the page after the split page. */
Packit fd8b60
	if (h->nextpg != P_INVALID) {
Packit fd8b60
		if ((tp = mpool_get(t->bt_mp, h->nextpg, 0)) == NULL) {
Packit fd8b60
			free(l);
Packit fd8b60
			/* XXX mpool_free(t->bt_mp, r->pgno); */
Packit fd8b60
			return (NULL);
Packit fd8b60
		}
Packit fd8b60
		tp->prevpg = r->pgno;
Packit fd8b60
		mpool_put(t->bt_mp, tp, MPOOL_DIRTY);
Packit fd8b60
	}
Packit fd8b60
Packit fd8b60
	/*
Packit fd8b60
	 * Split right.  The key/data pairs aren't sorted in the btree page so
Packit fd8b60
	 * it's simpler to copy the data from the split page onto two new pages
Packit fd8b60
	 * instead of copying half the data to the right page and compacting
Packit fd8b60
	 * the left page in place.  Since the left page can't change, we have
Packit fd8b60
	 * to swap the original and the allocated left page after the split.
Packit fd8b60
	 */
Packit fd8b60
	tp = bt_psplit(t, h, l, r, skip, ilen);
Packit fd8b60
Packit fd8b60
	/* Move the new left page onto the old left page. */
Packit fd8b60
	memmove(h, l, t->bt_psize);
Packit fd8b60
	if (tp == l)
Packit fd8b60
		tp = h;
Packit fd8b60
	free(l);
Packit fd8b60
Packit fd8b60
	*lp = h;
Packit fd8b60
	*rp = r;
Packit fd8b60
	return (tp);
Packit fd8b60
}
Packit fd8b60
Packit fd8b60
/*
Packit fd8b60
 * BT_ROOT -- Split the root page of a btree.
Packit fd8b60
 *
Packit fd8b60
 * Parameters:
Packit fd8b60
 *	t:	tree
Packit fd8b60
 *	h:	root page
Packit fd8b60
 *	lp:	pointer to left page pointer
Packit fd8b60
 *	rp:	pointer to right page pointer
Packit fd8b60
 *	skip:	pointer to index to leave open
Packit fd8b60
 *	ilen:	insert length
Packit fd8b60
 *
Packit fd8b60
 * Returns:
Packit fd8b60
 *	Pointer to page in which to insert or NULL on error.
Packit fd8b60
 */
Packit fd8b60
static PAGE *
Packit fd8b60
bt_root(t, h, lp, rp, skip, ilen)
Packit fd8b60
	BTREE *t;
Packit fd8b60
	PAGE *h, **lp, **rp;
Packit fd8b60
	indx_t *skip;
Packit fd8b60
	size_t ilen;
Packit fd8b60
{
Packit fd8b60
	PAGE *l, *r, *tp;
Packit fd8b60
	db_pgno_t lnpg, rnpg;
Packit fd8b60
Packit fd8b60
#ifdef STATISTICS
Packit fd8b60
	++bt_split;
Packit fd8b60
	++bt_rootsplit;
Packit fd8b60
#endif
Packit fd8b60
	/* Put the new left and right pages for the split into place. */
Packit fd8b60
	if ((l = __bt_new(t, &lnpg)) == NULL ||
Packit fd8b60
	    (r = __bt_new(t, &rnpg)) == NULL)
Packit fd8b60
		return (NULL);
Packit fd8b60
	l->pgno = lnpg;
Packit fd8b60
	r->pgno = rnpg;
Packit fd8b60
	l->nextpg = r->pgno;
Packit fd8b60
	r->prevpg = l->pgno;
Packit fd8b60
	l->prevpg = r->nextpg = P_INVALID;
Packit fd8b60
	l->lower = r->lower = BTDATAOFF;
Packit fd8b60
	l->upper = r->upper = t->bt_psize;
Packit fd8b60
	l->flags = r->flags = h->flags & P_TYPE;
Packit fd8b60
Packit fd8b60
	/* Split the root page. */
Packit fd8b60
	tp = bt_psplit(t, h, l, r, skip, ilen);
Packit fd8b60
Packit fd8b60
	*lp = l;
Packit fd8b60
	*rp = r;
Packit fd8b60
	return (tp);
Packit fd8b60
}
Packit fd8b60
Packit fd8b60
/*
Packit fd8b60
 * BT_RROOT -- Fix up the recno root page after it has been split.
Packit fd8b60
 *
Packit fd8b60
 * Parameters:
Packit fd8b60
 *	t:	tree
Packit fd8b60
 *	h:	root page
Packit fd8b60
 *	l:	left page
Packit fd8b60
 *	r:	right page
Packit fd8b60
 *
Packit fd8b60
 * Returns:
Packit fd8b60
 *	RET_ERROR, RET_SUCCESS
Packit fd8b60
 */
Packit fd8b60
static int
Packit fd8b60
bt_rroot(t, h, l, r)
Packit fd8b60
	BTREE *t;
Packit fd8b60
	PAGE *h, *l, *r;
Packit fd8b60
{
Packit fd8b60
	char *dest;
Packit fd8b60
Packit fd8b60
	/* Insert the left and right keys, set the header information. */
Packit fd8b60
	h->linp[0] = h->upper = t->bt_psize - NRINTERNAL;
Packit fd8b60
	dest = (char *)h + h->upper;
Packit fd8b60
	WR_RINTERNAL(dest,
Packit fd8b60
	    l->flags & P_RLEAF ? NEXTINDEX(l) : rec_total(l), l->pgno);
Packit fd8b60
Packit fd8b60
	h->linp[1] = h->upper -= NRINTERNAL;
Packit fd8b60
	dest = (char *)h + h->upper;
Packit fd8b60
	WR_RINTERNAL(dest,
Packit fd8b60
	    r->flags & P_RLEAF ? NEXTINDEX(r) : rec_total(r), r->pgno);
Packit fd8b60
Packit fd8b60
	h->lower = BTDATAOFF + 2 * sizeof(indx_t);
Packit fd8b60
Packit fd8b60
	/* Unpin the root page, set to recno internal page. */
Packit fd8b60
	h->flags &= ~P_TYPE;
Packit fd8b60
	h->flags |= P_RINTERNAL;
Packit fd8b60
	mpool_put(t->bt_mp, h, MPOOL_DIRTY);
Packit fd8b60
Packit fd8b60
	return (RET_SUCCESS);
Packit fd8b60
}
Packit fd8b60
Packit fd8b60
/*
Packit fd8b60
 * BT_BROOT -- Fix up the btree root page after it has been split.
Packit fd8b60
 *
Packit fd8b60
 * Parameters:
Packit fd8b60
 *	t:	tree
Packit fd8b60
 *	h:	root page
Packit fd8b60
 *	l:	left page
Packit fd8b60
 *	r:	right page
Packit fd8b60
 *
Packit fd8b60
 * Returns:
Packit fd8b60
 *	RET_ERROR, RET_SUCCESS
Packit fd8b60
 */
Packit fd8b60
static int
Packit fd8b60
bt_broot(t, h, l, r)
Packit fd8b60
	BTREE *t;
Packit fd8b60
	PAGE *h, *l, *r;
Packit fd8b60
{
Packit fd8b60
	BINTERNAL *bi;
Packit fd8b60
	BLEAF *bl;
Packit fd8b60
	u_int32_t nbytes;
Packit fd8b60
	char *dest;
Packit fd8b60
Packit fd8b60
	/*
Packit fd8b60
	 * If the root page was a leaf page, change it into an internal page.
Packit fd8b60
	 * We copy the key we split on (but not the key's data, in the case of
Packit fd8b60
	 * a leaf page) to the new root page.
Packit fd8b60
	 *
Packit fd8b60
	 * The btree comparison code guarantees that the left-most key on any
Packit fd8b60
	 * level of the tree is never used, so it doesn't need to be filled in.
Packit fd8b60
	 */
Packit fd8b60
	nbytes = NBINTERNAL(0);
Packit fd8b60
	h->linp[0] = h->upper = t->bt_psize - nbytes;
Packit fd8b60
	dest = (char *)h + h->upper;
Packit fd8b60
	WR_BINTERNAL(dest, 0, l->pgno, 0);
Packit fd8b60
Packit fd8b60
	switch (h->flags & P_TYPE) {
Packit fd8b60
	case P_BLEAF:
Packit fd8b60
		bl = GETBLEAF(r, 0);
Packit fd8b60
		nbytes = NBINTERNAL(bl->ksize);
Packit fd8b60
		h->linp[1] = h->upper -= nbytes;
Packit fd8b60
		dest = (char *)h + h->upper;
Packit fd8b60
		WR_BINTERNAL(dest, bl->ksize, r->pgno, 0);
Packit fd8b60
		memmove(dest, bl->bytes, bl->ksize);
Packit fd8b60
Packit fd8b60
		/*
Packit fd8b60
		 * If the key is on an overflow page, mark the overflow chain
Packit fd8b60
		 * so it isn't deleted when the leaf copy of the key is deleted.
Packit fd8b60
		 */
Packit fd8b60
		if (bl->flags & P_BIGKEY) {
Packit fd8b60
			db_pgno_t pgno;
Packit fd8b60
			memcpy(&pgno, bl->bytes, sizeof(pgno));
Packit fd8b60
			if (bt_preserve(t, pgno) == RET_ERROR)
Packit fd8b60
				return (RET_ERROR);
Packit fd8b60
		}
Packit fd8b60
		break;
Packit fd8b60
	case P_BINTERNAL:
Packit fd8b60
		bi = GETBINTERNAL(r, 0);
Packit fd8b60
		nbytes = NBINTERNAL(bi->ksize);
Packit fd8b60
		h->linp[1] = h->upper -= nbytes;
Packit fd8b60
		dest = (char *)h + h->upper;
Packit fd8b60
		memmove(dest, bi, nbytes);
Packit fd8b60
		((BINTERNAL *)(void *)dest)->pgno = r->pgno;
Packit fd8b60
		break;
Packit fd8b60
	default:
Packit fd8b60
		abort();
Packit fd8b60
	}
Packit fd8b60
Packit fd8b60
	/* There are two keys on the page. */
Packit fd8b60
	h->lower = BTDATAOFF + 2 * sizeof(indx_t);
Packit fd8b60
Packit fd8b60
	/* Unpin the root page, set to btree internal page. */
Packit fd8b60
	h->flags &= ~P_TYPE;
Packit fd8b60
	h->flags |= P_BINTERNAL;
Packit fd8b60
	mpool_put(t->bt_mp, h, MPOOL_DIRTY);
Packit fd8b60
Packit fd8b60
	return (RET_SUCCESS);
Packit fd8b60
}
Packit fd8b60
Packit fd8b60
/*
Packit fd8b60
 * BT_PSPLIT -- Do the real work of splitting the page.
Packit fd8b60
 *
Packit fd8b60
 * Parameters:
Packit fd8b60
 *	t:	tree
Packit fd8b60
 *	h:	page to be split
Packit fd8b60
 *	l:	page to put lower half of data
Packit fd8b60
 *	r:	page to put upper half of data
Packit fd8b60
 *	pskip:	pointer to index to leave open
Packit fd8b60
 *	ilen:	insert length
Packit fd8b60
 *
Packit fd8b60
 * Returns:
Packit fd8b60
 *	Pointer to page in which to insert.
Packit fd8b60
 */
Packit fd8b60
static PAGE *
Packit fd8b60
bt_psplit(t, h, l, r, pskip, ilen)
Packit fd8b60
	BTREE *t;
Packit fd8b60
	PAGE *h, *l, *r;
Packit fd8b60
	indx_t *pskip;
Packit fd8b60
	size_t ilen;
Packit fd8b60
{
Packit fd8b60
	BINTERNAL *bi;
Packit fd8b60
	BLEAF *bl;
Packit fd8b60
	CURSOR *c;
Packit fd8b60
	RLEAF *rl;
Packit fd8b60
	PAGE *rval;
Packit fd8b60
	void *src;
Packit fd8b60
	indx_t full, half, nxt, off, skip, top, used;
Packit fd8b60
	u_int32_t nbytes;
Packit fd8b60
	int bigkeycnt, isbigkey;
Packit fd8b60
Packit fd8b60
	/*
Packit fd8b60
	 * Split the data to the left and right pages.  Leave the skip index
Packit fd8b60
	 * open.  Additionally, make some effort not to split on an overflow
Packit fd8b60
	 * key.  This makes internal page processing faster and can save
Packit fd8b60
	 * space as overflow keys used by internal pages are never deleted.
Packit fd8b60
	 */
Packit fd8b60
	bigkeycnt = 0;
Packit fd8b60
	skip = *pskip;
Packit fd8b60
	full = t->bt_psize - BTDATAOFF;
Packit fd8b60
	half = full / 2;
Packit fd8b60
	used = 0;
Packit fd8b60
	src = NULL;	     /* silence gcc "uninitialized" warning */
Packit fd8b60
	for (nxt = off = 0, top = NEXTINDEX(h); nxt < top; ++off) {
Packit fd8b60
		if (skip == off) {
Packit fd8b60
			nbytes = ilen;
Packit fd8b60
			isbigkey = 0;		/* XXX: not really known. */
Packit fd8b60
		} else
Packit fd8b60
			switch (h->flags & P_TYPE) {
Packit fd8b60
			case P_BINTERNAL:
Packit fd8b60
				src = bi = GETBINTERNAL(h, nxt);
Packit fd8b60
				nbytes = NBINTERNAL(bi->ksize);
Packit fd8b60
				isbigkey = bi->flags & P_BIGKEY;
Packit fd8b60
				break;
Packit fd8b60
			case P_BLEAF:
Packit fd8b60
				src = bl = GETBLEAF(h, nxt);
Packit fd8b60
				nbytes = NBLEAF(bl);
Packit fd8b60
				isbigkey = bl->flags & P_BIGKEY;
Packit fd8b60
				break;
Packit fd8b60
			case P_RINTERNAL:
Packit fd8b60
				src = GETRINTERNAL(h, nxt);
Packit fd8b60
				nbytes = NRINTERNAL;
Packit fd8b60
				isbigkey = 0;
Packit fd8b60
				break;
Packit fd8b60
			case P_RLEAF:
Packit fd8b60
				src = rl = GETRLEAF(h, nxt);
Packit fd8b60
				nbytes = NRLEAF(rl);
Packit fd8b60
				isbigkey = 0;
Packit fd8b60
				break;
Packit fd8b60
			default:
Packit fd8b60
				abort();
Packit fd8b60
			}
Packit fd8b60
Packit fd8b60
		/*
Packit fd8b60
		 * If the key/data pairs are substantial fractions of the max
Packit fd8b60
		 * possible size for the page, it's possible to get situations
Packit fd8b60
		 * where we decide to try and copy too much onto the left page.
Packit fd8b60
		 * Make sure that doesn't happen.
Packit fd8b60
		 */
Packit fd8b60
		if ((skip <= off && used + nbytes + sizeof(indx_t) >= full)
Packit fd8b60
		    || nxt == top - 1) {
Packit fd8b60
			--off;
Packit fd8b60
			break;
Packit fd8b60
		}
Packit fd8b60
Packit fd8b60
		/* Copy the key/data pair, if not the skipped index. */
Packit fd8b60
		if (skip != off) {
Packit fd8b60
			++nxt;
Packit fd8b60
Packit fd8b60
			l->linp[off] = l->upper -= nbytes;
Packit fd8b60
			memmove((char *)l + l->upper, src, nbytes);
Packit fd8b60
		}
Packit fd8b60
Packit fd8b60
		used += nbytes + sizeof(indx_t);
Packit fd8b60
		if (used >= half) {
Packit fd8b60
			if (!isbigkey || bigkeycnt == 3)
Packit fd8b60
				break;
Packit fd8b60
			else
Packit fd8b60
				++bigkeycnt;
Packit fd8b60
		}
Packit fd8b60
	}
Packit fd8b60
Packit fd8b60
	/*
Packit fd8b60
	 * Off is the last offset that's valid for the left page.
Packit fd8b60
	 * Nxt is the first offset to be placed on the right page.
Packit fd8b60
	 */
Packit fd8b60
	l->lower += (off + 1) * sizeof(indx_t);
Packit fd8b60
Packit fd8b60
	/*
Packit fd8b60
	 * If splitting the page that the cursor was on, the cursor has to be
Packit fd8b60
	 * adjusted to point to the same record as before the split.  If the
Packit fd8b60
	 * cursor is at or past the skipped slot, the cursor is incremented by
Packit fd8b60
	 * one.  If the cursor is on the right page, it is decremented by the
Packit fd8b60
	 * number of records split to the left page.
Packit fd8b60
	 */
Packit fd8b60
	c = &t->bt_cursor;
Packit fd8b60
	if (F_ISSET(c, CURS_INIT) && c->pg.pgno == h->pgno) {
Packit fd8b60
		if (c->pg.index >= skip)
Packit fd8b60
			++c->pg.index;
Packit fd8b60
		if (c->pg.index < nxt)			/* Left page. */
Packit fd8b60
			c->pg.pgno = l->pgno;
Packit fd8b60
		else {					/* Right page. */
Packit fd8b60
			c->pg.pgno = r->pgno;
Packit fd8b60
			c->pg.index -= nxt;
Packit fd8b60
		}
Packit fd8b60
	}
Packit fd8b60
Packit fd8b60
	/*
Packit fd8b60
	 * If the skipped index was on the left page, just return that page.
Packit fd8b60
	 * Otherwise, adjust the skip index to reflect the new position on
Packit fd8b60
	 * the right page.
Packit fd8b60
	 */
Packit fd8b60
	if (skip <= off) {
Packit fd8b60
		skip = (indx_t)-1;
Packit fd8b60
		rval = l;
Packit fd8b60
	} else {
Packit fd8b60
		rval = r;
Packit fd8b60
		*pskip -= nxt;
Packit fd8b60
	}
Packit fd8b60
Packit fd8b60
	for (off = 0; nxt < top; ++off) {
Packit fd8b60
		if (skip == nxt) {
Packit fd8b60
			++off;
Packit fd8b60
			skip = (indx_t)-1;
Packit fd8b60
		}
Packit fd8b60
		switch (h->flags & P_TYPE) {
Packit fd8b60
		case P_BINTERNAL:
Packit fd8b60
			src = bi = GETBINTERNAL(h, nxt);
Packit fd8b60
			nbytes = NBINTERNAL(bi->ksize);
Packit fd8b60
			break;
Packit fd8b60
		case P_BLEAF:
Packit fd8b60
			src = bl = GETBLEAF(h, nxt);
Packit fd8b60
			nbytes = NBLEAF(bl);
Packit fd8b60
			break;
Packit fd8b60
		case P_RINTERNAL:
Packit fd8b60
			src = GETRINTERNAL(h, nxt);
Packit fd8b60
			nbytes = NRINTERNAL;
Packit fd8b60
			break;
Packit fd8b60
		case P_RLEAF:
Packit fd8b60
			src = rl = GETRLEAF(h, nxt);
Packit fd8b60
			nbytes = NRLEAF(rl);
Packit fd8b60
			break;
Packit fd8b60
		default:
Packit fd8b60
			abort();
Packit fd8b60
		}
Packit fd8b60
		++nxt;
Packit fd8b60
		r->linp[off] = r->upper -= nbytes;
Packit fd8b60
		memmove((char *)r + r->upper, src, nbytes);
Packit fd8b60
	}
Packit fd8b60
	r->lower += off * sizeof(indx_t);
Packit fd8b60
Packit fd8b60
	/* If the key is being appended to the page, adjust the index. */
Packit fd8b60
	if (skip == top)
Packit fd8b60
		r->lower += sizeof(indx_t);
Packit fd8b60
Packit fd8b60
	return (rval);
Packit fd8b60
}
Packit fd8b60
Packit fd8b60
/*
Packit fd8b60
 * BT_PRESERVE -- Mark a chain of pages as used by an internal node.
Packit fd8b60
 *
Packit fd8b60
 * Chains of indirect blocks pointed to by leaf nodes get reclaimed when the
Packit fd8b60
 * record that references them gets deleted.  Chains pointed to by internal
Packit fd8b60
 * pages never get deleted.  This routine marks a chain as pointed to by an
Packit fd8b60
 * internal page.
Packit fd8b60
 *
Packit fd8b60
 * Parameters:
Packit fd8b60
 *	t:	tree
Packit fd8b60
 *	pg:	page number of first page in the chain.
Packit fd8b60
 *
Packit fd8b60
 * Returns:
Packit fd8b60
 *	RET_SUCCESS, RET_ERROR.
Packit fd8b60
 */
Packit fd8b60
static int
Packit fd8b60
bt_preserve(t, pg)
Packit fd8b60
	BTREE *t;
Packit fd8b60
	db_pgno_t pg;
Packit fd8b60
{
Packit fd8b60
	PAGE *h;
Packit fd8b60
Packit fd8b60
	if ((h = mpool_get(t->bt_mp, pg, 0)) == NULL)
Packit fd8b60
		return (RET_ERROR);
Packit fd8b60
	h->flags |= P_PRESERVE;
Packit fd8b60
	mpool_put(t->bt_mp, h, MPOOL_DIRTY);
Packit fd8b60
	return (RET_SUCCESS);
Packit fd8b60
}
Packit fd8b60
Packit fd8b60
/*
Packit fd8b60
 * REC_TOTAL -- Return the number of recno entries below a page.
Packit fd8b60
 *
Packit fd8b60
 * Parameters:
Packit fd8b60
 *	h:	page
Packit fd8b60
 *
Packit fd8b60
 * Returns:
Packit fd8b60
 *	The number of recno entries below a page.
Packit fd8b60
 *
Packit fd8b60
 * XXX
Packit fd8b60
 * These values could be set by the bt_psplit routine.  The problem is that the
Packit fd8b60
 * entry has to be popped off of the stack etc. or the values have to be passed
Packit fd8b60
 * all the way back to bt_split/bt_rroot and it's not very clean.
Packit fd8b60
 */
Packit fd8b60
static recno_t
Packit fd8b60
rec_total(h)
Packit fd8b60
	PAGE *h;
Packit fd8b60
{
Packit fd8b60
	recno_t recs;
Packit fd8b60
	indx_t nxt, top;
Packit fd8b60
Packit fd8b60
	for (recs = 0, nxt = 0, top = NEXTINDEX(h); nxt < top; ++nxt)
Packit fd8b60
		recs += GETRINTERNAL(h, nxt)->nrecs;
Packit fd8b60
	return (recs);
Packit fd8b60
}