gecko-dev/db/dbm/dbm.c

431 строка
8.2 KiB
C

/*-
* See the file LICENSE for redistribution information.
*
* Copyright (c) 1996, 1997, 1998
* Sleepycat Software. All rights reserved.
*/
/*
* Copyright (c) 1990, 1993
* Margo Seltzer. All rights reserved.
*/
/*
* Copyright (c) 1990, 1993
* The Regents of the University of California. All rights reserved.
*
* This code is derived from software contributed to Berkeley by
* Margo Seltzer.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
* 3. All advertising materials mentioning features or use of this software
* must display the following acknowledgement:
* This product includes software developed by the University of
* California, Berkeley and its contributors.
* 4. Neither the name of the University nor the names of its contributors
* may be used to endorse or promote products derived from this software
* without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
* SUCH DAMAGE.
*/
#include "config.h"
#ifndef lint
static const char sccsid[] = "@(#)dbm.c 10.16 (Sleepycat) 5/7/98";
#endif /* not lint */
#ifndef NO_SYSTEM_INCLUDES
#include <sys/types.h>
#include <errno.h>
#include <fcntl.h>
#include <string.h>
#endif
#define DB_DBM_HSEARCH 1
#include "db_int.h"
#include "db_page.h"
#include "hash.h"
/*
*
* This package provides dbm and ndbm compatible interfaces to DB.
*
* The DBM routines, which call the NDBM routines.
*/
static DBM *__cur_db;
static void __db_no_open __P((void));
int
__db_dbm_init(file)
char *file;
{
if (__cur_db != NULL)
(void)dbm_close(__cur_db);
if ((__cur_db =
dbm_open(file, O_CREAT | O_RDWR, __db_omode("rw----"))) != NULL)
return (0);
if ((__cur_db = dbm_open(file, O_RDONLY, 0)) != NULL)
return (0);
return (-1);
}
datum
__db_dbm_fetch(key)
datum key;
{
datum item;
if (__cur_db == NULL) {
__db_no_open();
item.dptr = 0;
return (item);
}
return (dbm_fetch(__cur_db, key));
}
datum
__db_dbm_firstkey()
{
datum item;
if (__cur_db == NULL) {
__db_no_open();
item.dptr = 0;
return (item);
}
return (dbm_firstkey(__cur_db));
}
datum
__db_dbm_nextkey(key)
datum key;
{
datum item;
COMPQUIET(key.dsize, 0);
if (__cur_db == NULL) {
__db_no_open();
item.dptr = 0;
return (item);
}
return (dbm_nextkey(__cur_db));
}
int
__db_dbm_delete(key)
datum key;
{
int ret;
if (__cur_db == NULL) {
__db_no_open();
return (-1);
}
ret = dbm_delete(__cur_db, key);
if (ret == 0)
ret = (((DB *)__cur_db)->sync)((DB *)__cur_db, 0);
return (ret);
}
int
__db_dbm_store(key, dat)
datum key, dat;
{
int ret;
if (__cur_db == NULL) {
__db_no_open();
return (-1);
}
ret = dbm_store(__cur_db, key, dat, DBM_REPLACE);
if (ret == 0)
ret = (((DB *)__cur_db)->sync)((DB *)__cur_db, 0);
return (ret);
}
static void
__db_no_open()
{
(void)fprintf(stderr, "dbm: no open database.\n");
}
/*
* This package provides dbm and ndbm compatible interfaces to DB.
*
* The NDBM routines, which call the DB routines.
*/
/*
* Returns:
* *DBM on success
* NULL on failure
*/
DBM *
__db_ndbm_open(file, oflags, mode)
const char *file;
int oflags, mode;
{
DB *dbp;
DB_INFO dbinfo;
char path[MAXPATHLEN];
memset(&dbinfo, 0, sizeof(dbinfo));
dbinfo.db_pagesize = 4096;
dbinfo.h_ffactor = 40;
dbinfo.h_nelem = 1;
/*
* XXX
* Don't use sprintf(3)/snprintf(3) -- the former is dangerous, and
* the latter isn't standard, and we're manipulating strings handed
* us by the application.
*/
if (strlen(file) + strlen(DBM_SUFFIX) + 1 > sizeof(path)) {
errno = ENAMETOOLONG;
return (NULL);
}
(void)strcpy(path, file);
(void)strcat(path, DBM_SUFFIX);
if ((errno = db_open(path,
DB_HASH, __db_oflags(oflags), mode, NULL, &dbinfo, &dbp)) != 0)
return (NULL);
return ((DBM *)dbp);
}
/*
* Returns:
* Nothing.
*/
void
__db_ndbm_close(db)
DBM *db;
{
(void)db->close(db, 0);
}
/*
* Returns:
* DATUM on success
* NULL on failure
*/
datum
__db_ndbm_fetch(db, key)
DBM *db;
datum key;
{
DBT _key, _data;
datum data;
int ret;
memset(&_key, 0, sizeof(DBT));
memset(&_data, 0, sizeof(DBT));
_key.size = key.dsize;
_key.data = key.dptr;
if ((ret = db->get((DB *)db, NULL, &_key, &_data, 0)) == 0) {
data.dptr = _data.data;
data.dsize = _data.size;
} else {
data.dptr = NULL;
data.dsize = 0;
errno = ret == DB_NOTFOUND ? ENOENT : ret;
}
return (data);
}
/*
* Returns:
* DATUM on success
* NULL on failure
*/
datum
__db_ndbm_firstkey(db)
DBM *db;
{
DBT _key, _data;
datum key;
int ret;
DBC *cp;
if ((cp = TAILQ_FIRST(&db->curs_queue)) == NULL)
if ((errno = db->cursor(db, NULL, &cp)) != 0) {
memset(&key, 0, sizeof(key));
return (key);
}
memset(&_key, 0, sizeof(DBT));
memset(&_data, 0, sizeof(DBT));
if ((ret = (cp->c_get)(cp, &_key, &_data, DB_FIRST)) == 0) {
key.dptr = _key.data;
key.dsize = _key.size;
} else {
key.dptr = NULL;
key.dsize = 0;
errno = ret == DB_NOTFOUND ? ENOENT : ret;
}
return (key);
}
/*
* Returns:
* DATUM on success
* NULL on failure
*/
datum
__db_ndbm_nextkey(db)
DBM *db;
{
DBC *cp;
DBT _key, _data;
datum key;
int ret;
if ((cp = TAILQ_FIRST(&db->curs_queue)) == NULL)
if ((errno = db->cursor(db, NULL, &cp)) != 0) {
memset(&key, 0, sizeof(key));
return (key);
}
memset(&_key, 0, sizeof(DBT));
memset(&_data, 0, sizeof(DBT));
if ((ret = (cp->c_get)(cp, &_key, &_data, DB_NEXT)) == 0) {
key.dptr = _key.data;
key.dsize = _key.size;
} else {
key.dptr = NULL;
key.dsize = 0;
errno = ret == DB_NOTFOUND ? ENOENT : ret;
}
return (key);
}
/*
* Returns:
* 0 on success
* <0 failure
*/
int
__db_ndbm_delete(db, key)
DBM *db;
datum key;
{
DBT _key;
int ret;
memset(&_key, 0, sizeof(DBT));
_key.data = key.dptr;
_key.size = key.dsize;
if ((ret = (((DB *)db)->del)((DB *)db, NULL, &_key, 0)) == 0)
return (0);
errno = ret == DB_NOTFOUND ? ENOENT : ret;
return (-1);
}
/*
* Returns:
* 0 on success
* <0 failure
* 1 if DBM_INSERT and entry exists
*/
int
__db_ndbm_store(db, key, data, flags)
DBM *db;
datum key, data;
int flags;
{
DBT _key, _data;
int ret;
memset(&_key, 0, sizeof(DBT));
memset(&_data, 0, sizeof(DBT));
_key.data = key.dptr;
_key.size = key.dsize;
_data.data = data.dptr;
_data.size = data.dsize;
if ((ret = db->put((DB *)db, NULL,
&_key, &_data, flags == DBM_INSERT ? DB_NOOVERWRITE : 0)) == 0)
return (0);
if (ret == DB_KEYEXIST)
return (1);
errno = ret;
return (-1);
}
int
__db_ndbm_error(db)
DBM *db;
{
HTAB *hp;
hp = (HTAB *)db->internal;
return (hp->local_errno);
}
int
__db_ndbm_clearerr(db)
DBM *db;
{
HTAB *hp;
hp = (HTAB *)db->internal;
hp->local_errno = 0;
return (0);
}
/*
* Returns:
* 1 if read-only
* 0 if not read-only
*/
int
__db_ndbm_rdonly(db)
DBM *db;
{
return (F_ISSET((DB *)db, DB_AM_RDONLY) ? 1 : 0);
}
/*
* XXX
* We only have a single file descriptor that we can return, not two. Return
* the same one for both files. Hopefully, the user is using it for locking
* and picked one to use at random.
*/
int
__db_ndbm_dirfno(db)
DBM *db;
{
int fd;
(void)db->fd(db, &fd);
return (fd);
}
int
__db_ndbm_pagfno(db)
DBM *db;
{
int fd;
(void)db->fd(db, &fd);
return (fd);
}