You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

388 lines
7.5 KiB

/*-
* See the file LICENSE for redistribution information.
*
* Copyright (c) 1999
* Sleepycat Software. All rights reserved.
*/
#include "db_config.h"
#ifndef lint
static const char sccsid[] = "@(#)bt_method.c 11.8 (Sleepycat) 10/27/99";
#endif /* not lint */
#ifndef NO_SYSTEM_INCLUDES
#include <sys/types.h>
#include <errno.h>
#endif
#include "db_int.h"
#include "db_page.h"
#include "btree.h"
#include "qam.h"
static int CDB___bam_set_bt_compare __P((DB *, int (*)(const DBT *, const DBT *)));
static int CDB___bam_set_bt_maxkey __P((DB *, u_int32_t));
static int CDB___bam_set_bt_minkey __P((DB *, u_int32_t));
static int CDB___bam_set_bt_prefix __P((DB *, size_t(*)(const DBT *, const DBT *)));
static int CDB___ram_set_re_delim __P((DB *, int));
static int CDB___ram_set_re_len __P((DB *, u_int32_t));
static int CDB___ram_set_re_pad __P((DB *, int));
static int CDB___ram_set_re_source __P((DB *, const char *));
/*
* CDB___bam_db_create --
* Btree specific initialization of the DB structure.
*
* PUBLIC: int CDB___bam_db_create __P((DB *));
*/
int
CDB___bam_db_create(dbp)
DB *dbp;
{
BTREE *t;
int ret;
/* Allocate and initialize the private btree structure. */
if ((ret = CDB___os_calloc(1, sizeof(BTREE), &t)) != 0)
return (ret);
dbp->bt_internal = t;
t->bt_minkey = DEFMINKEYPAGE; /* Btree */
t->bt_compare = CDB___bam_defcmp;
t->bt_prefix = CDB___bam_defpfx;
dbp->set_bt_compare = CDB___bam_set_bt_compare;
dbp->set_bt_maxkey = CDB___bam_set_bt_maxkey;
dbp->set_bt_minkey = CDB___bam_set_bt_minkey;
dbp->set_bt_prefix = CDB___bam_set_bt_prefix;
t->re_delim = '\n'; /* Recno */
t->re_pad = ' ';
dbp->set_re_delim = CDB___ram_set_re_delim;
dbp->set_re_len = CDB___ram_set_re_len;
dbp->set_re_pad = CDB___ram_set_re_pad;
dbp->set_re_source = CDB___ram_set_re_source;
return (0);
}
/*
* CDB___bam_db_close --
* Btree specific discard of the DB structure.
*
* PUBLIC: int CDB___bam_db_close __P((DB *));
*/
int
CDB___bam_db_close(dbp)
DB *dbp;
{
BTREE *t;
t = dbp->bt_internal;
/* Recno */
/* Close any underlying mmap region. */
if (t->re_smap != NULL)
(void)CDB___os_unmapfile(dbp->dbenv, t->re_smap, t->re_msize);
/* Close any backing source file descriptor. */
if (F_ISSET(&t->re_fh, DB_FH_VALID))
(void)CDB___os_closehandle(&t->re_fh);
/* Free any backing source file name. */
if (t->re_source != NULL)
CDB___os_freestr(t->re_source);
CDB___os_free(t, sizeof(BTREE));
dbp->bt_internal = NULL;
return (0);
}
/*
* CDB___bam_set_flags --
* Set Btree specific flags.
*
* PUBLIC: int CDB___bam_set_flags __P((DB *, u_int32_t *flagsp));
*/
int
CDB___bam_set_flags(dbp, flagsp)
DB *dbp;
u_int32_t *flagsp;
{
u_int32_t flags;
flags = *flagsp;
if (LF_ISSET(DB_DUP | DB_DUPSORT | DB_RECNUM | DB_REVSPLITOFF)) {
DB_ILLEGAL_AFTER_OPEN(dbp, "DB->set_flags");
/*
* The DB_DUP and DB_DUPSORT flags are shared by the Hash
* and Btree access methods.
*/
if (LF_ISSET(DB_DUP | DB_DUPSORT))
DB_ILLEGAL_METHOD(dbp, DB_OK_BTREE | DB_OK_HASH);
if (LF_ISSET(DB_RECNUM | DB_REVSPLITOFF))
DB_ILLEGAL_METHOD(dbp, DB_OK_BTREE);
/*
* DB_DUP and DB_RECNUM are mutually incompatible. Handle
* the case where one or more flags have already been set.
*/
if ((LF_ISSET(DB_DUP) || F_ISSET(dbp, DB_AM_DUP)) &&
(LF_ISSET(DB_RECNUM) || F_ISSET(dbp, DB_BT_RECNUM)))
return (CDB___db_ferr(dbp->dbenv, "DB->set_flags", 1));
if (LF_ISSET(DB_DUP)) {
F_SET(dbp, DB_AM_DUP);
LF_CLR(DB_DUP);
}
if (LF_ISSET(DB_DUPSORT)) {
if (dbp->dup_compare == NULL)
dbp->dup_compare = CDB___bam_defcmp;
LF_CLR(DB_DUPSORT);
}
if (LF_ISSET(DB_RECNUM)) {
F_SET(dbp, DB_BT_RECNUM);
LF_CLR(DB_RECNUM);
}
if (LF_ISSET(DB_REVSPLITOFF)) {
F_SET(dbp, DB_BT_REVSPLIT);
LF_CLR(DB_REVSPLITOFF);
}
*flagsp = flags;
}
return (0);
}
/*
* CDB___bam_set_bt_compare --
* Set the comparison function.
*/
static int
CDB___bam_set_bt_compare(dbp, func)
DB *dbp;
int (*func) __P((const DBT *, const DBT *));
{
BTREE *t;
DB_ILLEGAL_AFTER_OPEN(dbp, "set_bt_compare");
DB_ILLEGAL_METHOD(dbp, DB_OK_BTREE);
t = dbp->bt_internal;
/*
* Can't default the prefix routine if the user supplies a comparison
* routine; shortening the keys can break their comparison algorithm.
*/
t->bt_compare = func;
if (t->bt_prefix == CDB___bam_defpfx)
t->bt_prefix = NULL;
return (0);
}
/*
* CDB___bam_set_bt_maxkey --
* Set the maximum keys per page.
*/
static int
CDB___bam_set_bt_maxkey(dbp, bt_maxkey)
DB *dbp;
u_int32_t bt_maxkey;
{
BTREE *t;
DB_ILLEGAL_AFTER_OPEN(dbp, "set_bt_maxkey");
DB_ILLEGAL_METHOD(dbp, DB_OK_BTREE);
t = dbp->bt_internal;
if (bt_maxkey < 1) {
CDB___db_err(dbp->dbenv, "minimum bt_maxkey value is 1");
return (EINVAL);
}
t->bt_maxkey = bt_maxkey;
return (0);
}
/*
* CDB___bam_set_bt_minkey --
* Set the minimum keys per page.
*/
static int
CDB___bam_set_bt_minkey(dbp, bt_minkey)
DB *dbp;
u_int32_t bt_minkey;
{
BTREE *t;
DB_ILLEGAL_AFTER_OPEN(dbp, "set_bt_minkey");
DB_ILLEGAL_METHOD(dbp, DB_OK_BTREE);
t = dbp->bt_internal;
if (bt_minkey < 2) {
CDB___db_err(dbp->dbenv, "minimum bt_minkey value is 2");
return (EINVAL);
}
t->bt_minkey = bt_minkey;
return (0);
}
/*
* CDB___bam_set_bt_prefix --
* Set the prefix function.
*/
static int
CDB___bam_set_bt_prefix(dbp, func)
DB *dbp;
size_t (*func) __P((const DBT *, const DBT *));
{
BTREE *t;
DB_ILLEGAL_AFTER_OPEN(dbp, "set_bt_prefix");
DB_ILLEGAL_METHOD(dbp, DB_OK_BTREE);
t = dbp->bt_internal;
t->bt_prefix = func;
return (0);
}
/*
* CDB___ram_set_flags --
* Set Recno specific flags.
*
* PUBLIC: int CDB___ram_set_flags __P((DB *, u_int32_t *flagsp));
*/
int
CDB___ram_set_flags(dbp, flagsp)
DB *dbp;
u_int32_t *flagsp;
{
u_int32_t flags;
flags = *flagsp;
if (LF_ISSET(DB_RENUMBER | DB_SNAPSHOT)) {
DB_ILLEGAL_AFTER_OPEN(dbp, "DB->set_flags");
DB_ILLEGAL_METHOD(dbp, DB_OK_RECNO);
if (LF_ISSET(DB_RENUMBER)) {
F_SET(dbp, DB_RE_RENUMBER);
LF_CLR(DB_RENUMBER);
}
if (LF_ISSET(DB_SNAPSHOT)) {
F_SET(dbp, DB_RE_SNAPSHOT);
LF_CLR(DB_SNAPSHOT);
}
*flagsp = flags;
}
return (0);
}
/*
* CDB___ram_set_re_delim --
* Set the variable-length input record delimiter.
*/
static int
CDB___ram_set_re_delim(dbp, re_delim)
DB *dbp;
int re_delim;
{
BTREE *t;
DB_ILLEGAL_AFTER_OPEN(dbp, "set_re_delim");
DB_ILLEGAL_METHOD(dbp, DB_OK_RECNO);
t = dbp->bt_internal;
t->re_delim = re_delim;
F_SET(dbp, DB_RE_DELIMITER);
return (0);
}
/*
* CDB___ram_set_re_len --
* Set the variable-length input record length.
*/
static int
CDB___ram_set_re_len(dbp, re_len)
DB *dbp;
u_int32_t re_len;
{
BTREE *t;
QUEUE *q;
DB_ILLEGAL_AFTER_OPEN(dbp, "set_re_len");
DB_ILLEGAL_METHOD(dbp, DB_OK_QUEUE | DB_OK_RECNO);
t = dbp->bt_internal;
t->re_len = re_len;
q = dbp->q_internal;
q->re_len = re_len;
F_SET(dbp, DB_RE_FIXEDLEN);
return (0);
}
/*
* CDB___ram_set_re_pad --
* Set the fixed-length record pad character.
*/
static int
CDB___ram_set_re_pad(dbp, re_pad)
DB *dbp;
int re_pad;
{
BTREE *t;
QUEUE *q;
DB_ILLEGAL_AFTER_OPEN(dbp, "set_re_pad");
DB_ILLEGAL_METHOD(dbp, DB_OK_QUEUE | DB_OK_RECNO);
t = dbp->bt_internal;
t->re_pad = re_pad;
q = dbp->q_internal;
q->re_pad = re_pad;
F_SET(dbp, DB_RE_PAD);
return (0);
}
/*
* CDB___ram_set_re_source --
* Set the backing source file name.
*/
static int
CDB___ram_set_re_source(dbp, re_source)
DB *dbp;
const char *re_source;
{
BTREE *t;
DB_ILLEGAL_AFTER_OPEN(dbp, "set_re_source");
DB_ILLEGAL_METHOD(dbp, DB_OK_RECNO);
t = dbp->bt_internal;
return (CDB___os_strdup(re_source, &t->re_source));
}