mirror of
https://github.com/berkeleydb/libdb.git
synced 2024-11-16 09:06:25 +00:00
4050 lines
99 KiB
C
4050 lines
99 KiB
C
/*-
|
|
* See the file LICENSE for redistribution information.
|
|
*
|
|
* Copyright (c) 1999, 2011 Oracle and/or its affiliates. All rights reserved.
|
|
*
|
|
* $Id$
|
|
*/
|
|
|
|
#include "db_config.h"
|
|
|
|
#include "db_int.h"
|
|
#ifdef HAVE_SYSTEM_INCLUDE_FILES
|
|
#include <tcl.h>
|
|
#endif
|
|
#include "dbinc/db_page.h"
|
|
#include "dbinc/db_am.h"
|
|
#include "dbinc/tcl_db.h"
|
|
|
|
/*
|
|
* Prototypes for procedures defined later in this file:
|
|
*/
|
|
static int tcl_DbAssociate __P((Tcl_Interp *,
|
|
int, Tcl_Obj * CONST*, DB *));
|
|
static int tcl_DbAssociateForeign __P((Tcl_Interp *,
|
|
int, Tcl_Obj * CONST*, DB *));
|
|
static int tcl_DbClose __P((Tcl_Interp *,
|
|
int, Tcl_Obj * CONST*, DB *, DBTCL_INFO *));
|
|
static int tcl_DbDelete __P((Tcl_Interp *, int, Tcl_Obj * CONST*, DB *));
|
|
static int tcl_DbGet __P((Tcl_Interp *, int, Tcl_Obj * CONST*, DB *, int));
|
|
#ifdef CONFIG_TEST
|
|
static int tcl_DbKeyRange __P((Tcl_Interp *, int, Tcl_Obj * CONST*, DB *));
|
|
#endif
|
|
static int tcl_DbPut __P((Tcl_Interp *, int, Tcl_Obj * CONST*, DB *));
|
|
static int tcl_DbStat __P((Tcl_Interp *, int, Tcl_Obj * CONST*, DB *));
|
|
static int tcl_DbStatPrint __P((Tcl_Interp *,
|
|
int, Tcl_Obj * CONST*, DB *));
|
|
static int tcl_DbTruncate __P((Tcl_Interp *, int, Tcl_Obj * CONST*, DB *));
|
|
#ifdef CONFIG_TEST
|
|
static int tcl_DbCompact __P((Tcl_Interp *, int, Tcl_Obj * CONST*, DB *));
|
|
static int tcl_DbCompactStat __P((Tcl_Interp *,
|
|
int, Tcl_Obj * CONST*, DB *));
|
|
#endif
|
|
static int tcl_DbCursor __P((Tcl_Interp *,
|
|
int, Tcl_Obj * CONST*, DB *, DBC **));
|
|
static int tcl_DbJoin __P((Tcl_Interp *,
|
|
int, Tcl_Obj * CONST*, DB *, DBC **));
|
|
static int tcl_DbGetFlags __P((Tcl_Interp *, int, Tcl_Obj * CONST*, DB *));
|
|
static int tcl_DbGetOpenFlags __P((Tcl_Interp *,
|
|
int, Tcl_Obj * CONST*, DB *));
|
|
static int tcl_DbGetjoin __P((Tcl_Interp *, int, Tcl_Obj * CONST*, DB *));
|
|
static int tcl_DbCount __P((Tcl_Interp *, int, Tcl_Obj * CONST*, DB *));
|
|
static int tcl_second_call __P((DB *, const DBT *, const DBT *, DBT *));
|
|
static int tcl_foreign_call __P((DB *, const DBT *, DBT *,
|
|
const DBT *, int *));
|
|
|
|
/*
|
|
* _DbInfoDelete --
|
|
*
|
|
* PUBLIC: void _DbInfoDelete __P((Tcl_Interp *, DBTCL_INFO *));
|
|
*/
|
|
void
|
|
_DbInfoDelete(interp, dbip)
|
|
Tcl_Interp *interp;
|
|
DBTCL_INFO *dbip;
|
|
{
|
|
DBTCL_INFO *nextp, *p;
|
|
/*
|
|
* First we have to close any open cursors. Then we close
|
|
* our db.
|
|
*/
|
|
for (p = LIST_FIRST(&__db_infohead); p != NULL; p = nextp) {
|
|
nextp = LIST_NEXT(p, entries);
|
|
/*
|
|
* Check if this is a cursor info structure and if
|
|
* it is, if it belongs to this DB. If so, remove
|
|
* its commands and info structure.
|
|
*/
|
|
if (p->i_parent == dbip && p->i_type == I_DBC) {
|
|
(void)Tcl_DeleteCommand(interp, p->i_name);
|
|
_DeleteInfo(p);
|
|
}
|
|
}
|
|
(void)Tcl_DeleteCommand(interp, dbip->i_name);
|
|
_DeleteInfo(dbip);
|
|
}
|
|
|
|
/*
|
|
*
|
|
* PUBLIC: int db_Cmd __P((ClientData, Tcl_Interp *, int, Tcl_Obj * CONST*));
|
|
*
|
|
* db_Cmd --
|
|
* Implements the "db" widget.
|
|
*/
|
|
int
|
|
db_Cmd(clientData, interp, objc, objv)
|
|
ClientData clientData; /* DB handle */
|
|
Tcl_Interp *interp; /* Interpreter */
|
|
int objc; /* How many arguments? */
|
|
Tcl_Obj *CONST objv[]; /* The argument objects */
|
|
{
|
|
static const char *dbcmds[] = {
|
|
#ifdef CONFIG_TEST
|
|
"keyrange",
|
|
"pget",
|
|
"test",
|
|
"compact",
|
|
"compact_stat",
|
|
#endif
|
|
"associate",
|
|
"associate_foreign",
|
|
"close",
|
|
"count",
|
|
"cursor",
|
|
"del",
|
|
"get",
|
|
"get_bt_minkey",
|
|
"get_cachesize",
|
|
"get_dbname",
|
|
"get_encrypt_flags",
|
|
"get_env",
|
|
"get_errpfx",
|
|
"get_flags",
|
|
"get_h_ffactor",
|
|
"get_h_nelem",
|
|
"get_join",
|
|
"get_lorder",
|
|
"get_open_flags",
|
|
"get_pagesize",
|
|
"get_q_extentsize",
|
|
"get_re_delim",
|
|
"get_re_len",
|
|
"get_re_pad",
|
|
"get_re_source",
|
|
"get_type",
|
|
"is_byteswapped",
|
|
"join",
|
|
"put",
|
|
"stat",
|
|
"stat_print",
|
|
"sync",
|
|
"truncate",
|
|
NULL
|
|
};
|
|
enum dbcmds {
|
|
#ifdef CONFIG_TEST
|
|
DBKEYRANGE,
|
|
DBPGET,
|
|
DBTEST,
|
|
DBCOMPACT,
|
|
DBCOMPACT_STAT,
|
|
#endif
|
|
DBASSOCIATE,
|
|
DBASSOCFOREIGN,
|
|
DBCLOSE,
|
|
DBCOUNT,
|
|
DBCURSOR,
|
|
DBDELETE,
|
|
DBGET,
|
|
DBGETBTMINKEY,
|
|
DBGETCACHESIZE,
|
|
DBGETDBNAME,
|
|
DBGETENCRYPTFLAGS,
|
|
DBGETENV,
|
|
DBGETERRPFX,
|
|
DBGETFLAGS,
|
|
DBGETHFFACTOR,
|
|
DBGETHNELEM,
|
|
DBGETJOIN,
|
|
DBGETLORDER,
|
|
DBGETOPENFLAGS,
|
|
DBGETPAGESIZE,
|
|
DBGETQEXTENTSIZE,
|
|
DBGETREDELIM,
|
|
DBGETRELEN,
|
|
DBGETREPAD,
|
|
DBGETRESOURCE,
|
|
DBGETTYPE,
|
|
DBSWAPPED,
|
|
DBJOIN,
|
|
DBPUT,
|
|
DBSTAT,
|
|
DBSTATPRINT,
|
|
DBSYNC,
|
|
DBTRUNCATE
|
|
};
|
|
DB *dbp, *hrdbp, *hsdbp;
|
|
DB_ENV *dbenv;
|
|
DBC *dbc;
|
|
DBTCL_INFO *dbip, *ip;
|
|
DBTYPE type;
|
|
Tcl_Obj *res, *myobjv[3];
|
|
int cmdindex, intval, ncache, result, ret;
|
|
char newname[MSG_SIZE];
|
|
u_int32_t bytes, gbytes, value;
|
|
const char *strval, *filename, *dbname, *envid;
|
|
|
|
Tcl_ResetResult(interp);
|
|
dbp = (DB *)clientData;
|
|
dbip = _PtrToInfo((void *)dbp);
|
|
memset(newname, 0, MSG_SIZE);
|
|
result = TCL_OK;
|
|
if (objc <= 1) {
|
|
Tcl_WrongNumArgs(interp, 1, objv, "command cmdargs");
|
|
return (TCL_ERROR);
|
|
}
|
|
if (dbp == NULL) {
|
|
Tcl_SetResult(interp, "NULL db pointer", TCL_STATIC);
|
|
return (TCL_ERROR);
|
|
}
|
|
if (dbip == NULL) {
|
|
Tcl_SetResult(interp, "NULL db info pointer", TCL_STATIC);
|
|
return (TCL_ERROR);
|
|
}
|
|
|
|
/*
|
|
* Get the command name index from the object based on the dbcmds
|
|
* defined above.
|
|
*/
|
|
if (Tcl_GetIndexFromObj(interp,
|
|
objv[1], dbcmds, "command", TCL_EXACT, &cmdindex) != TCL_OK)
|
|
return (IS_HELP(objv[1]));
|
|
|
|
res = NULL;
|
|
switch ((enum dbcmds)cmdindex) {
|
|
#ifdef CONFIG_TEST
|
|
case DBKEYRANGE:
|
|
result = tcl_DbKeyRange(interp, objc, objv, dbp);
|
|
break;
|
|
case DBPGET:
|
|
result = tcl_DbGet(interp, objc, objv, dbp, 1);
|
|
break;
|
|
case DBTEST:
|
|
result = tcl_EnvTest(interp, objc, objv, dbp->dbenv);
|
|
break;
|
|
|
|
case DBCOMPACT:
|
|
result = tcl_DbCompact(interp, objc, objv, dbp);
|
|
break;
|
|
|
|
case DBCOMPACT_STAT:
|
|
result = tcl_DbCompactStat(interp, objc, objv, dbp);
|
|
break;
|
|
|
|
#endif
|
|
case DBASSOCIATE:
|
|
result = tcl_DbAssociate(interp, objc, objv, dbp);
|
|
break;
|
|
case DBASSOCFOREIGN:
|
|
result = tcl_DbAssociateForeign(interp, objc, objv, dbp);
|
|
break;
|
|
case DBCLOSE:
|
|
result = tcl_DbClose(interp, objc, objv, dbp, dbip);
|
|
break;
|
|
case DBDELETE:
|
|
result = tcl_DbDelete(interp, objc, objv, dbp);
|
|
break;
|
|
case DBGET:
|
|
result = tcl_DbGet(interp, objc, objv, dbp, 0);
|
|
break;
|
|
case DBPUT:
|
|
result = tcl_DbPut(interp, objc, objv, dbp);
|
|
break;
|
|
case DBCOUNT:
|
|
result = tcl_DbCount(interp, objc, objv, dbp);
|
|
break;
|
|
case DBSWAPPED:
|
|
/*
|
|
* No args for this. Error if there are some.
|
|
*/
|
|
if (objc > 2) {
|
|
Tcl_WrongNumArgs(interp, 2, objv, NULL);
|
|
return (TCL_ERROR);
|
|
}
|
|
_debug_check();
|
|
ret = dbp->get_byteswapped(dbp, &intval);
|
|
res = Tcl_NewIntObj(intval);
|
|
break;
|
|
case DBGETTYPE:
|
|
/*
|
|
* No args for this. Error if there are some.
|
|
*/
|
|
if (objc > 2) {
|
|
Tcl_WrongNumArgs(interp, 2, objv, NULL);
|
|
return (TCL_ERROR);
|
|
}
|
|
_debug_check();
|
|
ret = dbp->get_type(dbp, &type);
|
|
if (type == DB_BTREE)
|
|
res = NewStringObj("btree", strlen("btree"));
|
|
else if (type == DB_HASH)
|
|
res = NewStringObj("hash", strlen("hash"));
|
|
else if (type == DB_RECNO)
|
|
res = NewStringObj("recno", strlen("recno"));
|
|
else if (type == DB_QUEUE)
|
|
res = NewStringObj("queue", strlen("queue"));
|
|
else if (type == DB_HEAP)
|
|
res = NewStringObj("heap", strlen("heap"));
|
|
else {
|
|
Tcl_SetResult(interp,
|
|
"db gettype: Returned unknown type\n", TCL_STATIC);
|
|
result = TCL_ERROR;
|
|
}
|
|
break;
|
|
case DBSTAT:
|
|
result = tcl_DbStat(interp, objc, objv, dbp);
|
|
break;
|
|
case DBSTATPRINT:
|
|
result = tcl_DbStatPrint(interp, objc, objv, dbp);
|
|
break;
|
|
case DBSYNC:
|
|
/*
|
|
* No args for this. Error if there are some.
|
|
*/
|
|
if (objc > 2) {
|
|
Tcl_WrongNumArgs(interp, 2, objv, NULL);
|
|
return (TCL_ERROR);
|
|
}
|
|
_debug_check();
|
|
ret = dbp->sync(dbp, 0);
|
|
res = Tcl_NewIntObj(ret);
|
|
if (ret != 0) {
|
|
Tcl_SetObjResult(interp, res);
|
|
result = TCL_ERROR;
|
|
}
|
|
|
|
/* If we are heap, we have more work to do. */
|
|
ret = dbp->get_type(dbp, &type);
|
|
if (type == DB_HEAP) {
|
|
hrdbp = dbip->hrdbp;
|
|
hsdbp = dbip->hsdbp;
|
|
|
|
/* sync the associated dbs also */
|
|
ret = dbp->sync(hrdbp, 0);
|
|
res = Tcl_NewIntObj(ret);
|
|
if (ret != 0) {
|
|
Tcl_SetObjResult(interp, res);
|
|
result = TCL_ERROR;
|
|
}
|
|
|
|
ret = dbp->sync(hsdbp, 0);
|
|
res = Tcl_NewIntObj(ret);
|
|
if (ret != 0) {
|
|
Tcl_SetObjResult(interp, res);
|
|
result = TCL_ERROR;
|
|
}
|
|
}
|
|
break;
|
|
case DBCURSOR:
|
|
snprintf(newname, sizeof(newname),
|
|
"%s.c%d", dbip->i_name, dbip->i_dbdbcid);
|
|
ip = _NewInfo(interp, NULL, newname, I_DBC);
|
|
if (ip != NULL) {
|
|
result = tcl_DbCursor(interp, objc, objv, dbp, &dbc);
|
|
if (result == TCL_OK) {
|
|
dbip->i_dbdbcid++;
|
|
ip->i_parent = dbip;
|
|
(void)Tcl_CreateObjCommand(interp, newname,
|
|
(Tcl_ObjCmdProc *)dbc_Cmd,
|
|
(ClientData)dbc, NULL);
|
|
res = NewStringObj(newname, strlen(newname));
|
|
_SetInfoData(ip, dbc);
|
|
} else
|
|
_DeleteInfo(ip);
|
|
} else {
|
|
Tcl_SetResult(interp,
|
|
"Could not set up info", TCL_STATIC);
|
|
result = TCL_ERROR;
|
|
}
|
|
break;
|
|
case DBJOIN:
|
|
snprintf(newname, sizeof(newname),
|
|
"%s.c%d", dbip->i_name, dbip->i_dbdbcid);
|
|
ip = _NewInfo(interp, NULL, newname, I_DBC);
|
|
if (ip != NULL) {
|
|
result = tcl_DbJoin(interp, objc, objv, dbp, &dbc);
|
|
if (result == TCL_OK) {
|
|
dbip->i_dbdbcid++;
|
|
ip->i_parent = dbip;
|
|
(void)Tcl_CreateObjCommand(interp, newname,
|
|
(Tcl_ObjCmdProc *)dbc_Cmd,
|
|
(ClientData)dbc, NULL);
|
|
res = NewStringObj(newname, strlen(newname));
|
|
_SetInfoData(ip, dbc);
|
|
} else
|
|
_DeleteInfo(ip);
|
|
} else {
|
|
Tcl_SetResult(interp,
|
|
"Could not set up info", TCL_STATIC);
|
|
result = TCL_ERROR;
|
|
}
|
|
break;
|
|
case DBGETBTMINKEY:
|
|
if (objc != 2) {
|
|
Tcl_WrongNumArgs(interp, 1, objv, NULL);
|
|
return (TCL_ERROR);
|
|
}
|
|
ret = dbp->get_bt_minkey(dbp, &value);
|
|
if ((result = _ReturnSetup(interp, ret, DB_RETOK_STD(ret),
|
|
"db get_bt_minkey")) == TCL_OK)
|
|
res = Tcl_NewIntObj((int)value);
|
|
break;
|
|
case DBGETCACHESIZE:
|
|
if (objc != 2) {
|
|
Tcl_WrongNumArgs(interp, 1, objv, NULL);
|
|
return (TCL_ERROR);
|
|
}
|
|
ret = dbp->get_cachesize(dbp, &gbytes, &bytes, &ncache);
|
|
if ((result = _ReturnSetup(interp, ret, DB_RETOK_STD(ret),
|
|
"db get_cachesize")) == TCL_OK) {
|
|
myobjv[0] = Tcl_NewIntObj((int)gbytes);
|
|
myobjv[1] = Tcl_NewIntObj((int)bytes);
|
|
myobjv[2] = Tcl_NewIntObj((int)ncache);
|
|
res = Tcl_NewListObj(3, myobjv);
|
|
}
|
|
break;
|
|
case DBGETDBNAME:
|
|
if (objc != 2) {
|
|
Tcl_WrongNumArgs(interp, 1, objv, NULL);
|
|
return (TCL_ERROR);
|
|
}
|
|
ret = dbp->get_dbname(dbp, &filename, &dbname);
|
|
if ((result = _ReturnSetup(interp, ret, DB_RETOK_STD(ret),
|
|
"db get_dbname")) == TCL_OK) {
|
|
myobjv[0] = filename == NULL ? NewStringObj("", 0) :
|
|
NewStringObj(filename, strlen(filename));
|
|
myobjv[1] = dbname == NULL ? NewStringObj("", 0) :
|
|
NewStringObj(dbname, strlen(dbname));
|
|
res = Tcl_NewListObj(2, myobjv);
|
|
}
|
|
break;
|
|
case DBGETENCRYPTFLAGS:
|
|
result = tcl_EnvGetEncryptFlags(interp, objc, objv, dbp->dbenv);
|
|
break;
|
|
case DBGETENV:
|
|
if (objc != 2) {
|
|
Tcl_WrongNumArgs(interp, 1, objv, NULL);
|
|
return (TCL_ERROR);
|
|
}
|
|
dbenv = dbp->get_env(dbp);
|
|
if (dbenv != NULL && (ip = _PtrToInfo(dbenv)) != NULL) {
|
|
envid = ip->i_name;
|
|
res = NewStringObj(envid, strlen(envid));
|
|
} else
|
|
Tcl_ResetResult(interp);
|
|
break;
|
|
case DBGETERRPFX:
|
|
if (objc != 2) {
|
|
Tcl_WrongNumArgs(interp, 1, objv, NULL);
|
|
return (TCL_ERROR);
|
|
}
|
|
dbp->get_errpfx(dbp, &strval);
|
|
res = NewStringObj(strval, strlen(strval));
|
|
break;
|
|
case DBGETFLAGS:
|
|
result = tcl_DbGetFlags(interp, objc, objv, dbp);
|
|
break;
|
|
case DBGETHFFACTOR:
|
|
if (objc != 2) {
|
|
Tcl_WrongNumArgs(interp, 1, objv, NULL);
|
|
return (TCL_ERROR);
|
|
}
|
|
ret = dbp->get_h_ffactor(dbp, &value);
|
|
if ((result = _ReturnSetup(interp, ret, DB_RETOK_STD(ret),
|
|
"db get_h_ffactor")) == TCL_OK)
|
|
res = Tcl_NewIntObj((int)value);
|
|
break;
|
|
case DBGETHNELEM:
|
|
if (objc != 2) {
|
|
Tcl_WrongNumArgs(interp, 1, objv, NULL);
|
|
return (TCL_ERROR);
|
|
}
|
|
ret = dbp->get_h_nelem(dbp, &value);
|
|
if ((result = _ReturnSetup(interp, ret, DB_RETOK_STD(ret),
|
|
"db get_h_nelem")) == TCL_OK)
|
|
res = Tcl_NewIntObj((int)value);
|
|
break;
|
|
case DBGETJOIN:
|
|
result = tcl_DbGetjoin(interp, objc, objv, dbp);
|
|
break;
|
|
case DBGETLORDER:
|
|
/*
|
|
* No args for this. Error if there are some.
|
|
*/
|
|
if (objc > 2) {
|
|
Tcl_WrongNumArgs(interp, 2, objv, NULL);
|
|
return (TCL_ERROR);
|
|
}
|
|
_debug_check();
|
|
ret = dbp->get_lorder(dbp, &intval);
|
|
res = Tcl_NewIntObj(intval);
|
|
break;
|
|
case DBGETOPENFLAGS:
|
|
result = tcl_DbGetOpenFlags(interp, objc, objv, dbp);
|
|
break;
|
|
case DBGETPAGESIZE:
|
|
if (objc != 2) {
|
|
Tcl_WrongNumArgs(interp, 1, objv, NULL);
|
|
return (TCL_ERROR);
|
|
}
|
|
ret = dbp->get_pagesize(dbp, &value);
|
|
if ((result = _ReturnSetup(interp, ret, DB_RETOK_STD(ret),
|
|
"db get_pagesize")) == TCL_OK)
|
|
res = Tcl_NewIntObj((int)value);
|
|
break;
|
|
case DBGETQEXTENTSIZE:
|
|
if (objc != 2) {
|
|
Tcl_WrongNumArgs(interp, 1, objv, NULL);
|
|
return (TCL_ERROR);
|
|
}
|
|
ret = dbp->get_q_extentsize(dbp, &value);
|
|
if ((result = _ReturnSetup(interp, ret, DB_RETOK_STD(ret),
|
|
"db get_q_extentsize")) == TCL_OK)
|
|
res = Tcl_NewIntObj((int)value);
|
|
break;
|
|
case DBGETREDELIM:
|
|
if (objc != 2) {
|
|
Tcl_WrongNumArgs(interp, 1, objv, NULL);
|
|
return (TCL_ERROR);
|
|
}
|
|
ret = dbp->get_re_delim(dbp, &intval);
|
|
if ((result = _ReturnSetup(interp, ret, DB_RETOK_STD(ret),
|
|
"db get_re_delim")) == TCL_OK)
|
|
res = Tcl_NewIntObj(intval);
|
|
break;
|
|
case DBGETRELEN:
|
|
if (objc != 2) {
|
|
Tcl_WrongNumArgs(interp, 1, objv, NULL);
|
|
return (TCL_ERROR);
|
|
}
|
|
ret = dbp->get_re_len(dbp, &value);
|
|
if ((result = _ReturnSetup(interp, ret, DB_RETOK_STD(ret),
|
|
"db get_re_len")) == TCL_OK)
|
|
res = Tcl_NewIntObj((int)value);
|
|
break;
|
|
case DBGETREPAD:
|
|
if (objc != 2) {
|
|
Tcl_WrongNumArgs(interp, 1, objv, NULL);
|
|
return (TCL_ERROR);
|
|
}
|
|
ret = dbp->get_re_pad(dbp, &intval);
|
|
if ((result = _ReturnSetup(interp, ret, DB_RETOK_STD(ret),
|
|
"db get_re_pad")) == TCL_OK)
|
|
res = Tcl_NewIntObj((int)intval);
|
|
break;
|
|
case DBGETRESOURCE:
|
|
if (objc != 2) {
|
|
Tcl_WrongNumArgs(interp, 1, objv, NULL);
|
|
return (TCL_ERROR);
|
|
}
|
|
ret = dbp->get_re_source(dbp, &strval);
|
|
if ((result = _ReturnSetup(interp, ret, DB_RETOK_STD(ret),
|
|
"db get_re_source")) == TCL_OK)
|
|
res = NewStringObj(strval, strlen(strval));
|
|
break;
|
|
case DBTRUNCATE:
|
|
result = tcl_DbTruncate(interp, objc, objv, dbp);
|
|
break;
|
|
}
|
|
/*
|
|
* Only set result if we have a res. Otherwise, lower
|
|
* functions have already done so.
|
|
*/
|
|
if (result == TCL_OK && res)
|
|
Tcl_SetObjResult(interp, res);
|
|
return (result);
|
|
}
|
|
|
|
/*
|
|
* tcl_db_stat --
|
|
*/
|
|
static int
|
|
tcl_DbStat(interp, objc, objv, dbp)
|
|
Tcl_Interp *interp; /* Interpreter */
|
|
int objc; /* How many arguments? */
|
|
Tcl_Obj *CONST objv[]; /* The argument objects */
|
|
DB *dbp; /* Database pointer */
|
|
{
|
|
static const char *dbstatopts[] = {
|
|
#ifdef CONFIG_TEST
|
|
"-read_committed",
|
|
"-read_uncommitted",
|
|
#endif
|
|
"-faststat",
|
|
"-txn",
|
|
NULL
|
|
};
|
|
enum dbstatopts {
|
|
#ifdef CONFIG_TEST
|
|
DBCUR_READ_COMMITTED,
|
|
DBCUR_READ_UNCOMMITTED,
|
|
#endif
|
|
DBCUR_FASTSTAT,
|
|
DBCUR_TXN
|
|
};
|
|
DBTYPE type;
|
|
DB_BTREE_STAT *bsp;
|
|
DB_HASH_STAT *hsp;
|
|
DB_HEAP_STAT *hpsp;
|
|
DB_QUEUE_STAT *qsp;
|
|
DB_TXN *txn;
|
|
Tcl_Obj *res, *flaglist, *myobjv[2];
|
|
u_int32_t flag;
|
|
int i, optindex, result, ret;
|
|
char *arg, msg[MSG_SIZE];
|
|
void *sp;
|
|
|
|
result = TCL_OK;
|
|
flag = 0;
|
|
txn = NULL;
|
|
sp = NULL;
|
|
i = 2;
|
|
while (i < objc) {
|
|
if (Tcl_GetIndexFromObj(interp, objv[i], dbstatopts, "option",
|
|
TCL_EXACT, &optindex) != TCL_OK) {
|
|
result = IS_HELP(objv[i]);
|
|
goto error;
|
|
}
|
|
i++;
|
|
switch ((enum dbstatopts)optindex) {
|
|
#ifdef CONFIG_TEST
|
|
case DBCUR_READ_COMMITTED:
|
|
flag |= DB_READ_COMMITTED;
|
|
break;
|
|
case DBCUR_READ_UNCOMMITTED:
|
|
flag |= DB_READ_UNCOMMITTED;
|
|
break;
|
|
#endif
|
|
case DBCUR_FASTSTAT:
|
|
flag |= DB_FAST_STAT;
|
|
break;
|
|
case DBCUR_TXN:
|
|
if (i == objc) {
|
|
Tcl_WrongNumArgs(interp, 2, objv, "?-txn id?");
|
|
result = TCL_ERROR;
|
|
break;
|
|
}
|
|
arg = Tcl_GetStringFromObj(objv[i++], NULL);
|
|
txn = NAME_TO_TXN(arg);
|
|
if (txn == NULL) {
|
|
snprintf(msg, MSG_SIZE,
|
|
"Stat: Invalid txn: %s\n", arg);
|
|
Tcl_SetResult(interp, msg, TCL_VOLATILE);
|
|
result = TCL_ERROR;
|
|
}
|
|
break;
|
|
}
|
|
if (result != TCL_OK)
|
|
break;
|
|
}
|
|
if (result != TCL_OK)
|
|
goto error;
|
|
|
|
_debug_check();
|
|
ret = dbp->stat(dbp, txn, &sp, flag);
|
|
result = _ReturnSetup(interp, ret, DB_RETOK_STD(ret), "db stat");
|
|
if (result == TCL_ERROR)
|
|
return (result);
|
|
|
|
(void)dbp->get_type(dbp, &type);
|
|
/*
|
|
* Have our stats, now construct the name value
|
|
* list pairs and free up the memory.
|
|
*/
|
|
res = Tcl_NewObj();
|
|
|
|
/*
|
|
* MAKE_STAT_LIST assumes 'res' and 'error' label.
|
|
*/
|
|
if (type == DB_HASH) {
|
|
hsp = (DB_HASH_STAT *)sp;
|
|
MAKE_STAT_LIST("Magic", hsp->hash_magic);
|
|
MAKE_STAT_LIST("Version", hsp->hash_version);
|
|
MAKE_STAT_LIST("Page size", hsp->hash_pagesize);
|
|
MAKE_STAT_LIST("Page count", hsp->hash_pagecnt);
|
|
MAKE_STAT_LIST("Number of keys", hsp->hash_nkeys);
|
|
MAKE_STAT_LIST("Number of records", hsp->hash_ndata);
|
|
MAKE_STAT_LIST("Fill factor", hsp->hash_ffactor);
|
|
MAKE_STAT_LIST("Buckets", hsp->hash_buckets);
|
|
if (flag != DB_FAST_STAT) {
|
|
MAKE_STAT_LIST("Free pages", hsp->hash_free);
|
|
MAKE_WSTAT_LIST("Bytes free", hsp->hash_bfree);
|
|
MAKE_STAT_LIST("Number of big pages",
|
|
hsp->hash_bigpages);
|
|
MAKE_STAT_LIST("Big pages bytes free",
|
|
hsp->hash_big_bfree);
|
|
MAKE_STAT_LIST("Overflow pages", hsp->hash_overflows);
|
|
MAKE_STAT_LIST("Overflow bytes free",
|
|
hsp->hash_ovfl_free);
|
|
MAKE_STAT_LIST("Duplicate pages", hsp->hash_dup);
|
|
MAKE_STAT_LIST("Duplicate pages bytes free",
|
|
hsp->hash_dup_free);
|
|
}
|
|
} else if (type == DB_HEAP) {
|
|
hpsp = (DB_HEAP_STAT *)sp;
|
|
MAKE_STAT_LIST("Magic", hpsp->heap_magic);
|
|
MAKE_STAT_LIST("Version", hpsp->heap_version);
|
|
MAKE_STAT_LIST("Page size", hpsp->heap_pagesize);
|
|
MAKE_STAT_LIST("Page count", hpsp->heap_pagecnt);
|
|
MAKE_STAT_LIST("Number of records", hpsp->heap_nrecs);
|
|
} else if (type == DB_QUEUE) {
|
|
qsp = (DB_QUEUE_STAT *)sp;
|
|
MAKE_STAT_LIST("Magic", qsp->qs_magic);
|
|
MAKE_STAT_LIST("Version", qsp->qs_version);
|
|
MAKE_STAT_LIST("Page size", qsp->qs_pagesize);
|
|
MAKE_STAT_LIST("Extent size", qsp->qs_extentsize);
|
|
MAKE_STAT_LIST("Number of keys", qsp->qs_nkeys);
|
|
MAKE_STAT_LIST("Number of records", qsp->qs_ndata);
|
|
MAKE_STAT_LIST("Record length", qsp->qs_re_len);
|
|
MAKE_STAT_LIST("Record pad", qsp->qs_re_pad);
|
|
MAKE_STAT_LIST("First record number", qsp->qs_first_recno);
|
|
MAKE_STAT_LIST("Last record number", qsp->qs_cur_recno);
|
|
if (flag != DB_FAST_STAT) {
|
|
MAKE_STAT_LIST("Number of pages", qsp->qs_pages);
|
|
MAKE_WSTAT_LIST("Bytes free", qsp->qs_pgfree);
|
|
}
|
|
} else { /* BTREE and RECNO are same stats */
|
|
bsp = (DB_BTREE_STAT *)sp;
|
|
MAKE_STAT_LIST("Magic", bsp->bt_magic);
|
|
MAKE_STAT_LIST("Version", bsp->bt_version);
|
|
MAKE_STAT_LIST("Number of keys", bsp->bt_nkeys);
|
|
MAKE_STAT_LIST("Number of records", bsp->bt_ndata);
|
|
MAKE_STAT_LIST("Minimum keys per page", bsp->bt_minkey);
|
|
MAKE_STAT_LIST("Fixed record length", bsp->bt_re_len);
|
|
MAKE_STAT_LIST("Record pad", bsp->bt_re_pad);
|
|
MAKE_STAT_LIST("Page size", bsp->bt_pagesize);
|
|
MAKE_STAT_LIST("Page count", bsp->bt_pagecnt);
|
|
if (flag != DB_FAST_STAT) {
|
|
MAKE_STAT_LIST("Levels", bsp->bt_levels);
|
|
MAKE_STAT_LIST("Internal pages", bsp->bt_int_pg);
|
|
MAKE_STAT_LIST("Leaf pages", bsp->bt_leaf_pg);
|
|
MAKE_STAT_LIST("Duplicate pages", bsp->bt_dup_pg);
|
|
MAKE_STAT_LIST("Overflow pages", bsp->bt_over_pg);
|
|
MAKE_STAT_LIST("Empty pages", bsp->bt_empty_pg);
|
|
MAKE_STAT_LIST("Pages on freelist", bsp->bt_free);
|
|
MAKE_STAT_LIST("Internal pages bytes free",
|
|
bsp->bt_int_pgfree);
|
|
MAKE_STAT_LIST("Leaf pages bytes free",
|
|
bsp->bt_leaf_pgfree);
|
|
MAKE_STAT_LIST("Duplicate pages bytes free",
|
|
bsp->bt_dup_pgfree);
|
|
MAKE_STAT_LIST("Bytes free in overflow pages",
|
|
bsp->bt_over_pgfree);
|
|
}
|
|
}
|
|
|
|
/*
|
|
* Construct a {name {flag1 flag2 ... flagN}} list for the
|
|
* dbp flags. These aren't access-method dependent, but they
|
|
* include all the interesting flags, and the integer value
|
|
* isn't useful from Tcl--return the strings instead.
|
|
*/
|
|
myobjv[0] = NewStringObj("Flags", strlen("Flags"));
|
|
myobjv[1] = _GetFlagsList(interp, dbp->flags, __db_get_flags_fn());
|
|
flaglist = Tcl_NewListObj(2, myobjv);
|
|
if (flaglist == NULL) {
|
|
result = TCL_ERROR;
|
|
goto error;
|
|
}
|
|
if ((result =
|
|
Tcl_ListObjAppendElement(interp, res, flaglist)) != TCL_OK)
|
|
goto error;
|
|
|
|
Tcl_SetObjResult(interp, res);
|
|
error:
|
|
if (sp != NULL)
|
|
__os_ufree(dbp->env, sp);
|
|
return (result);
|
|
}
|
|
|
|
/*
|
|
* tcl_db_stat_print --
|
|
*/
|
|
static int
|
|
tcl_DbStatPrint(interp, objc, objv, dbp)
|
|
Tcl_Interp *interp; /* Interpreter */
|
|
int objc; /* How many arguments? */
|
|
Tcl_Obj *CONST objv[]; /* The argument objects */
|
|
DB *dbp; /* Database pointer */
|
|
{
|
|
static const char *dbstatprtopts[] = {
|
|
"-fast",
|
|
"-all",
|
|
NULL
|
|
};
|
|
enum dbstatprtopts {
|
|
DBSTATPRTFAST,
|
|
DBSTATPRTALL
|
|
};
|
|
u_int32_t flag;
|
|
int i, optindex, result, ret;
|
|
|
|
result = TCL_OK;
|
|
flag = 0;
|
|
i = 2;
|
|
|
|
while (i < objc) {
|
|
if (Tcl_GetIndexFromObj(interp, objv[i], dbstatprtopts,
|
|
"option", TCL_EXACT, &optindex) != TCL_OK) {
|
|
result = IS_HELP(objv[i]);
|
|
goto error;
|
|
}
|
|
i++;
|
|
switch ((enum dbstatprtopts)optindex) {
|
|
case DBSTATPRTFAST:
|
|
flag |= DB_FAST_STAT;
|
|
break;
|
|
case DBSTATPRTALL:
|
|
flag |= DB_STAT_ALL;
|
|
break;
|
|
}
|
|
if (result != TCL_OK)
|
|
break;
|
|
}
|
|
if (result != TCL_OK)
|
|
goto error;
|
|
|
|
_debug_check();
|
|
ret = dbp->stat_print(dbp, flag);
|
|
result = _ReturnSetup(interp, ret, DB_RETOK_STD(ret), "db stat_print");
|
|
error:
|
|
return (result);
|
|
}
|
|
|
|
/*
|
|
* tcl_db_close --
|
|
*/
|
|
static int
|
|
tcl_DbClose(interp, objc, objv, dbp, dbip)
|
|
Tcl_Interp *interp; /* Interpreter */
|
|
int objc; /* How many arguments? */
|
|
Tcl_Obj *CONST objv[]; /* The argument objects */
|
|
DB *dbp; /* Database pointer */
|
|
DBTCL_INFO *dbip; /* Info pointer */
|
|
{
|
|
static const char *dbclose[] = {
|
|
"-nosync", "--", NULL
|
|
};
|
|
enum dbclose {
|
|
TCL_DBCLOSE_NOSYNC,
|
|
TCL_DBCLOSE_ENDARG
|
|
};
|
|
DB *recdbp, *secdbp;
|
|
DBTCL_INFO *rdbip, *sdbip;
|
|
u_int32_t flag;
|
|
int endarg, i, optindex, result, ret;
|
|
char *arg;
|
|
|
|
result = TCL_OK;
|
|
endarg = 0;
|
|
flag = 0;
|
|
if (objc > 4) {
|
|
Tcl_WrongNumArgs(interp, 2, objv, "?-nosync?");
|
|
return (TCL_ERROR);
|
|
}
|
|
|
|
for (i = 2; i < objc; ++i) {
|
|
if (Tcl_GetIndexFromObj(interp, objv[i], dbclose,
|
|
"option", TCL_EXACT, &optindex) != TCL_OK) {
|
|
arg = Tcl_GetStringFromObj(objv[i], NULL);
|
|
if (arg[0] == '-')
|
|
return (IS_HELP(objv[i]));
|
|
else
|
|
Tcl_ResetResult(interp);
|
|
break;
|
|
}
|
|
switch ((enum dbclose)optindex) {
|
|
case TCL_DBCLOSE_NOSYNC:
|
|
flag = DB_NOSYNC;
|
|
break;
|
|
case TCL_DBCLOSE_ENDARG:
|
|
endarg = 1;
|
|
break;
|
|
}
|
|
/*
|
|
* If, at any time, parsing the args we get an error,
|
|
* bail out and return.
|
|
*/
|
|
if (result != TCL_OK)
|
|
return (result);
|
|
if (endarg)
|
|
break;
|
|
}
|
|
|
|
/* If it looks like there might be aux dbs, try to close them. */
|
|
recdbp = ((DBTCL_INFO *)dbp->api_internal)->hrdbp;
|
|
secdbp = ((DBTCL_INFO *)dbp->api_internal)->hsdbp;
|
|
if (recdbp != NULL && secdbp != NULL) {
|
|
rdbip = recdbp->api_internal;
|
|
_DbInfoDelete(interp, rdbip);
|
|
recdbp->api_internal = NULL;
|
|
ret = recdbp->close(recdbp, flag);
|
|
|
|
sdbip = secdbp->api_internal;
|
|
_DbInfoDelete(interp, sdbip);
|
|
secdbp->api_internal = NULL;
|
|
ret = secdbp->close(secdbp, flag);
|
|
result = _ReturnSetup(interp,
|
|
ret, DB_RETOK_STD(ret), "db close");
|
|
}
|
|
|
|
if (dbip->i_cdata != NULL)
|
|
__os_free(dbp->env, dbip->i_cdata);
|
|
_DbInfoDelete(interp, dbip);
|
|
_debug_check();
|
|
|
|
/* Paranoia. */
|
|
dbp->api_internal = NULL;
|
|
|
|
ret = (dbp)->close(dbp, flag);
|
|
|
|
/* As long as the 1st close above was ok, then we check this one. */
|
|
if (result == TCL_OK)
|
|
result = _ReturnSetup(interp,
|
|
ret, DB_RETOK_STD(ret), "db close");
|
|
|
|
return (result);
|
|
}
|
|
|
|
/*
|
|
* tcl_db_put --
|
|
*/
|
|
static int
|
|
tcl_DbPut(interp, objc, objv, dbp)
|
|
Tcl_Interp *interp; /* Interpreter */
|
|
int objc; /* How many arguments? */
|
|
Tcl_Obj *CONST objv[]; /* The argument objects */
|
|
DB *dbp; /* Database pointer */
|
|
{
|
|
static const char *dbputopts[] = {
|
|
#ifdef CONFIG_TEST
|
|
"-nodupdata",
|
|
#endif
|
|
"-append",
|
|
"-multiple",
|
|
"-multiple_key",
|
|
"-nooverwrite",
|
|
"-overwritedup",
|
|
"-partial",
|
|
"-txn",
|
|
NULL
|
|
};
|
|
enum dbputopts {
|
|
#ifdef CONFIG_TEST
|
|
DBGET_NODUPDATA,
|
|
#endif
|
|
DBPUT_APPEND,
|
|
DBPUT_MULTIPLE,
|
|
DBPUT_MULTIPLE_KEY,
|
|
DBPUT_NOOVER,
|
|
DBPUT_OVER,
|
|
DBPUT_PART,
|
|
DBPUT_TXN
|
|
};
|
|
static const char *dbputapp[] = {
|
|
"-append",
|
|
"-multiple_key",
|
|
NULL
|
|
};
|
|
enum dbputapp { DBPUT_APPEND0, DBPUT_MULTIPLE_KEY0 };
|
|
DBT data, hkey, key;
|
|
DBTYPE type;
|
|
DB *recdbp;
|
|
DB_HEAP_RID rid;
|
|
DB_TXN *txn;
|
|
Tcl_Obj **delemv, **elemv, *res;
|
|
void *dtmp, *ktmp, *ptr;
|
|
db_recno_t recno;
|
|
u_int32_t flag, hflag, multiflag;
|
|
int delemc, elemc, end, freekey, freedata, skiprecno;
|
|
int dlen, klen, i, optindex, result, ret;
|
|
char *arg, msg[MSG_SIZE];
|
|
|
|
txn = NULL;
|
|
result = TCL_OK;
|
|
flag = hflag = multiflag = 0;
|
|
if (objc <= 2) {
|
|
Tcl_WrongNumArgs(interp, 2, objv, "?-args? ?key? data");
|
|
return (TCL_ERROR);
|
|
}
|
|
|
|
dtmp = ktmp = NULL;
|
|
freekey = freedata = skiprecno = 0;
|
|
memset(&key, 0, sizeof(key));
|
|
memset(&data, 0, sizeof(data));
|
|
COMPQUIET(recno, 0);
|
|
COMPQUIET(recdbp, NULL);
|
|
|
|
/*
|
|
* If it is a QUEUE or RECNO database, the key is a record number
|
|
* and must be setup up to contain a db_recno_t. Otherwise the
|
|
* key is a "string".
|
|
*/
|
|
(void)dbp->get_type(dbp, &type);
|
|
|
|
/*
|
|
* We need to determine where the end of required args are. If we are
|
|
* using a QUEUE/RECNO/HEAP db and -append, or -multiple_key
|
|
* is specified, then there is just one req arg (data). Otherwise
|
|
* there are two (key data).
|
|
*
|
|
* We preparse the list to determine this since we need to know
|
|
* to properly check # of args for other options below.
|
|
*/
|
|
end = objc - 2;
|
|
i = 2;
|
|
while (i < objc - 1) {
|
|
if (Tcl_GetIndexFromObj(interp, objv[i++], dbputapp,
|
|
"option", TCL_EXACT, &optindex) != TCL_OK)
|
|
continue;
|
|
switch ((enum dbputapp)optindex) {
|
|
case DBPUT_APPEND0:
|
|
case DBPUT_MULTIPLE_KEY0:
|
|
end = objc - 1;
|
|
break;
|
|
}
|
|
}
|
|
Tcl_ResetResult(interp);
|
|
|
|
/*
|
|
* Get the command name index from the object based on the options
|
|
* defined above.
|
|
*/
|
|
i = 2;
|
|
while (i < end) {
|
|
if (Tcl_GetIndexFromObj(interp, objv[i],
|
|
dbputopts, "option", TCL_EXACT, &optindex) != TCL_OK)
|
|
return (IS_HELP(objv[i]));
|
|
i++;
|
|
switch ((enum dbputopts)optindex) {
|
|
#ifdef CONFIG_TEST
|
|
case DBGET_NODUPDATA:
|
|
FLAG_CHECK(flag);
|
|
flag = DB_NODUPDATA;
|
|
break;
|
|
#endif
|
|
case DBPUT_TXN:
|
|
if (i > (end - 1)) {
|
|
Tcl_WrongNumArgs(interp, 2, objv, "?-txn id?");
|
|
result = TCL_ERROR;
|
|
break;
|
|
}
|
|
arg = Tcl_GetStringFromObj(objv[i++], NULL);
|
|
txn = NAME_TO_TXN(arg);
|
|
if (txn == NULL) {
|
|
snprintf(msg, MSG_SIZE,
|
|
"Put: Invalid txn: %s\n", arg);
|
|
Tcl_SetResult(interp, msg, TCL_VOLATILE);
|
|
result = TCL_ERROR;
|
|
}
|
|
break;
|
|
case DBPUT_APPEND:
|
|
FLAG_CHECK(flag);
|
|
flag = DB_APPEND;
|
|
break;
|
|
case DBPUT_MULTIPLE:
|
|
FLAG_CHECK(multiflag);
|
|
multiflag = DB_MULTIPLE;
|
|
if (type == DB_HEAP) {
|
|
Tcl_SetResult(interp,
|
|
"-multiple not supported", TCL_STATIC);
|
|
result = TCL_ERROR;
|
|
}
|
|
break;
|
|
case DBPUT_MULTIPLE_KEY:
|
|
FLAG_CHECK(multiflag);
|
|
multiflag = DB_MULTIPLE_KEY;
|
|
if (type == DB_HEAP) {
|
|
Tcl_SetResult(interp,
|
|
"-multiple_key not supported", TCL_STATIC);
|
|
result = TCL_ERROR;
|
|
}
|
|
break;
|
|
case DBPUT_NOOVER:
|
|
FLAG_CHECK(flag);
|
|
flag = DB_NOOVERWRITE;
|
|
break;
|
|
case DBPUT_OVER:
|
|
FLAG_CHECK(flag);
|
|
flag = DB_OVERWRITE_DUP;
|
|
break;
|
|
case DBPUT_PART:
|
|
if (i > (end - 1)) {
|
|
Tcl_WrongNumArgs(interp, 2, objv,
|
|
"?-partial {offset length}?");
|
|
result = TCL_ERROR;
|
|
break;
|
|
}
|
|
/*
|
|
* Get sublist as {offset length}
|
|
*/
|
|
result = Tcl_ListObjGetElements(interp, objv[i++],
|
|
&elemc, &elemv);
|
|
if (elemc != 2) {
|
|
Tcl_SetResult(interp,
|
|
"List must be {offset length}", TCL_STATIC);
|
|
result = TCL_ERROR;
|
|
break;
|
|
}
|
|
data.flags = DB_DBT_PARTIAL;
|
|
result = _GetUInt32(interp, elemv[0], &data.doff);
|
|
if (result != TCL_OK)
|
|
break;
|
|
result = _GetUInt32(interp, elemv[1], &data.dlen);
|
|
/*
|
|
* NOTE: We don't check result here because all we'd
|
|
* do is break anyway, and we are doing that. If you
|
|
* add code here, you WILL need to add the check
|
|
* for result. (See the check for save.doff, a few
|
|
* lines above and copy that.)
|
|
*/
|
|
break;
|
|
}
|
|
if (result != TCL_OK)
|
|
break;
|
|
}
|
|
|
|
if (result == TCL_ERROR)
|
|
return (result);
|
|
|
|
if (multiflag == DB_MULTIPLE) {
|
|
/*
|
|
* To work out how big a buffer is needed, we first need to
|
|
* find out the total length of the data and the number of data
|
|
* items (elemc).
|
|
*/
|
|
ktmp = Tcl_GetByteArrayFromObj(objv[objc - 2], &klen);
|
|
result = Tcl_ListObjGetElements(interp, objv[objc - 2],
|
|
&elemc, &elemv);
|
|
if (result != TCL_OK)
|
|
return (result);
|
|
|
|
dtmp = Tcl_GetByteArrayFromObj(objv[objc - 1], &dlen);
|
|
result = Tcl_ListObjGetElements(interp, objv[objc - 1],
|
|
&delemc, &delemv);
|
|
if (result != TCL_OK)
|
|
return (result);
|
|
|
|
if (elemc < delemc)
|
|
delemc = elemc;
|
|
else
|
|
elemc = delemc;
|
|
|
|
memset(&key, 0, sizeof(key));
|
|
key.ulen = DB_ALIGN((u_int32_t)klen +
|
|
(u_int32_t)elemc * sizeof(u_int32_t) * 2, 1024UL);
|
|
key.flags = DB_DBT_USERMEM | DB_DBT_BULK;
|
|
if ((ret = __os_malloc(dbp->env, key.ulen, &key.data)) != 0)
|
|
return (ret);
|
|
freekey = 1;
|
|
|
|
memset(&data, 0, sizeof(data));
|
|
data.ulen = DB_ALIGN((u_int32_t)dlen +
|
|
(u_int32_t)delemc * sizeof(u_int32_t) * 2, 1024UL);
|
|
data.flags = DB_DBT_USERMEM | DB_DBT_BULK;
|
|
if ((ret = __os_malloc(dbp->env, data.ulen, &data.data)) != 0)
|
|
return (ret);
|
|
freedata = 1;
|
|
|
|
if (type == DB_QUEUE || type == DB_RECNO) {
|
|
DB_MULTIPLE_RECNO_WRITE_INIT(ptr, &key);
|
|
for (i = 0; i < elemc; i++) {
|
|
result = _GetUInt32(interp, elemv[i], &recno);
|
|
DB_MULTIPLE_RECNO_WRITE_NEXT(ptr, &key, recno,
|
|
dtmp, 0);
|
|
DB_ASSERT(dbp->env, ptr != NULL);
|
|
}
|
|
} else {
|
|
DB_MULTIPLE_WRITE_INIT(ptr, &key);
|
|
for (i = 0; i < elemc; i++) {
|
|
ktmp = Tcl_GetByteArrayFromObj(elemv[i], &klen);
|
|
DB_MULTIPLE_WRITE_NEXT(ptr,
|
|
&key, ktmp, (u_int32_t)klen);
|
|
DB_ASSERT(dbp->env, ptr != NULL);
|
|
}
|
|
}
|
|
DB_MULTIPLE_WRITE_INIT(ptr, &data);
|
|
for (i = 0; i < elemc; i++) {
|
|
dtmp = Tcl_GetByteArrayFromObj(delemv[i], &dlen);
|
|
DB_MULTIPLE_WRITE_NEXT(ptr,
|
|
&data, dtmp, (u_int32_t)dlen);
|
|
DB_ASSERT(dbp->env, ptr != NULL);
|
|
}
|
|
} else if (multiflag == DB_MULTIPLE_KEY) {
|
|
/*
|
|
* To work out how big a buffer is needed, we first need to
|
|
* find out the total length of the data (len) and the number
|
|
* of data items (elemc).
|
|
*/
|
|
ktmp = Tcl_GetByteArrayFromObj(objv[objc - 1], &klen);
|
|
result = Tcl_ListObjGetElements(interp, objv[objc - 1],
|
|
&elemc, &elemv);
|
|
if (result != TCL_OK)
|
|
return (result);
|
|
|
|
memset(&key, 0, sizeof(key));
|
|
key.ulen = DB_ALIGN((u_int32_t)klen +
|
|
(u_int32_t)elemc * sizeof(u_int32_t) * 2, 1024UL);
|
|
key.flags = DB_DBT_USERMEM | DB_DBT_BULK;
|
|
if ((ret = __os_malloc(dbp->env, key.ulen, &key.data)) != 0)
|
|
return (ret);
|
|
freekey = 1;
|
|
|
|
if (type == DB_QUEUE || type == DB_RECNO) {
|
|
DB_MULTIPLE_RECNO_WRITE_INIT(ptr, &key);
|
|
for (i = 0; i + 1 < elemc; i += 2) {
|
|
result = _GetUInt32(interp, elemv[i], &recno);
|
|
dtmp = Tcl_GetByteArrayFromObj(elemv[i + 1],
|
|
&dlen);
|
|
DB_MULTIPLE_RECNO_WRITE_NEXT(ptr, &key,
|
|
recno, dtmp, (u_int32_t)dlen);
|
|
DB_ASSERT(dbp->env, ptr != NULL);
|
|
}
|
|
} else {
|
|
DB_MULTIPLE_WRITE_INIT(ptr, &key);
|
|
for (i = 0; i + 1 < elemc; i += 2) {
|
|
ktmp = Tcl_GetByteArrayFromObj(elemv[i], &klen);
|
|
dtmp = Tcl_GetByteArrayFromObj(elemv[i + 1],
|
|
&dlen);
|
|
DB_MULTIPLE_KEY_WRITE_NEXT(ptr,
|
|
&key, ktmp, (u_int32_t)klen,
|
|
dtmp, (u_int32_t)dlen);
|
|
DB_ASSERT(dbp->env, ptr != NULL);
|
|
}
|
|
}
|
|
} else if (type == DB_QUEUE || type == DB_RECNO) {
|
|
/*
|
|
* If we are a recno db and we are NOT using append, then the
|
|
* 2nd last arg is the key.
|
|
*/
|
|
key.data = &recno;
|
|
key.ulen = key.size = sizeof(db_recno_t);
|
|
key.flags = DB_DBT_USERMEM;
|
|
if (flag == DB_APPEND)
|
|
recno = 0;
|
|
else {
|
|
result = _GetUInt32(interp, objv[objc-2], &recno);
|
|
if (result != TCL_OK)
|
|
return (result);
|
|
}
|
|
} else if (type == DB_HEAP) {
|
|
/*
|
|
* With heap we have 2 puts, one to the heap db and one
|
|
* to the recno db. Use the key passed in completely
|
|
* for the recno, and hkey for heap. The hkey will
|
|
* become the data for the recno db put.
|
|
*/
|
|
memset(&hkey, 0, sizeof(hkey));
|
|
hkey.data = &rid;
|
|
hkey.ulen = hkey.size = sizeof(DB_HEAP_RID);
|
|
hkey.flags = DB_DBT_USERMEM;
|
|
rid.pgno = 0;
|
|
rid.indx = 0;
|
|
|
|
/* set up key for recno auxiliary db */
|
|
key.data = &recno;
|
|
key.ulen = key.size = sizeof(db_recno_t);
|
|
key.flags = DB_DBT_USERMEM;
|
|
|
|
/* Set up the DB ptr for the recno db */
|
|
recdbp = ((DBTCL_INFO *)dbp->api_internal)->hrdbp;
|
|
|
|
/*
|
|
* If DB_APPEND is set, we want to append to the heap. If
|
|
* there's no flag set and the recno exists, we want to update
|
|
* the existing record (which means we need to find the heap rid
|
|
* in the recno db and we can skip the recno put.) If the
|
|
* recno does not exist, then we want to append to the heap.
|
|
*/
|
|
if (flag == DB_APPEND) {
|
|
recno = 0;
|
|
hflag = DB_APPEND;
|
|
} else {
|
|
result = _GetUInt32(interp, objv[objc-2], &recno);
|
|
if (result != TCL_OK)
|
|
return (result);
|
|
|
|
ret = recdbp->get(recdbp, txn, &key, &hkey, 0);
|
|
if (ret == 0) {
|
|
hflag = flag;
|
|
skiprecno = 1;
|
|
}
|
|
else if (ret == DB_NOTFOUND || ret == DB_KEYEMPTY)
|
|
hflag = DB_APPEND;
|
|
else {
|
|
result = _ReturnSetup(interp, ret,
|
|
DB_RETOK_DBPUT(ret), "db put heap");
|
|
goto out;
|
|
}
|
|
}
|
|
|
|
} else {
|
|
ret = _CopyObjBytes(interp, objv[objc-2], &ktmp,
|
|
&key.size, &freekey);
|
|
if (ret != 0) {
|
|
result = _ReturnSetup(interp, ret,
|
|
DB_RETOK_DBPUT(ret), "db put");
|
|
return (result);
|
|
}
|
|
key.data = ktmp;
|
|
}
|
|
|
|
if (multiflag == 0) {
|
|
ret = _CopyObjBytes(interp,
|
|
objv[objc-1], &dtmp, &data.size, &freedata);
|
|
if (ret != 0) {
|
|
result = _ReturnSetup(interp, ret,
|
|
DB_RETOK_DBPUT(ret), "db put");
|
|
goto out;
|
|
}
|
|
data.data = dtmp;
|
|
}
|
|
_debug_check();
|
|
if (type != DB_HEAP) {
|
|
ret = dbp->put(dbp, txn, &key, &data, flag | multiflag);
|
|
result = _ReturnSetup(interp, ret, DB_RETOK_DBPUT(ret),
|
|
"db put");
|
|
} else {
|
|
/* Do put into heap db first, then recno db. Varieties of
|
|
* -multiple are not supported in heap (checked above), so
|
|
* multiflag has been removed from the code.
|
|
*/
|
|
ret = dbp->put(dbp, txn, &hkey, &data, hflag);
|
|
result = _ReturnSetup(interp, ret,
|
|
DB_RETOK_DBPUT(ret), "db put heap");
|
|
if (ret)
|
|
goto out;
|
|
|
|
/* set up data for recno put (the heaps key) and do put
|
|
* if we have not already gotten the recno record
|
|
*/
|
|
hkey.flags = DB_DBT_USERMEM;
|
|
if (!skiprecno) {
|
|
ret = recdbp->put(recdbp, txn, &key, &hkey, flag);
|
|
result = _ReturnSetup(interp, ret, DB_RETOK_DBPUT(ret),
|
|
"db put");
|
|
}
|
|
/*
|
|
* If for some reason the recno put does not work and the
|
|
* heap one does, lets try and delete the heap record if
|
|
* we are not in a txn -- just for consistency. Ignore the
|
|
* return value, because we want to preserve ret.
|
|
*/
|
|
if (!skiprecno && txn == NULL && ret != 0)
|
|
(void)dbp->del(dbp, txn, &hkey, 0);
|
|
}
|
|
|
|
/* We may have a returned record number. */
|
|
if (ret == 0 &&
|
|
(type == DB_QUEUE || type == DB_RECNO || type == DB_HEAP) &&
|
|
flag == DB_APPEND) {
|
|
res = Tcl_NewWideIntObj((Tcl_WideInt)recno);
|
|
Tcl_SetObjResult(interp, res);
|
|
}
|
|
|
|
|
|
out: if (freedata && data.data != NULL)
|
|
__os_free(dbp->env, data.data);
|
|
if (freekey && key.data != NULL)
|
|
__os_free(dbp->env, key.data);
|
|
return (result);
|
|
}
|
|
|
|
/*
|
|
* tcl_db_get --
|
|
*/
|
|
static int
|
|
tcl_DbGet(interp, objc, objv, dbp, ispget)
|
|
Tcl_Interp *interp; /* Interpreter */
|
|
int objc; /* How many arguments? */
|
|
Tcl_Obj *CONST objv[]; /* The argument objects */
|
|
DB *dbp; /* Database pointer */
|
|
int ispget; /* 1 for pget, 0 for get */
|
|
{
|
|
static const char *dbgetopts[] = {
|
|
#ifdef CONFIG_TEST
|
|
"-data_buf_size",
|
|
"-multi",
|
|
"-nolease",
|
|
"-read_committed",
|
|
"-read_uncommitted",
|
|
#endif
|
|
"-consume",
|
|
"-consume_wait",
|
|
"-get_both",
|
|
"-glob",
|
|
"-partial",
|
|
"-recno",
|
|
"-rmw",
|
|
"-txn",
|
|
"--",
|
|
NULL
|
|
};
|
|
enum dbgetopts {
|
|
#ifdef CONFIG_TEST
|
|
DBGET_DATA_BUF_SIZE,
|
|
DBGET_MULTI,
|
|
DBGET_NOLEASE,
|
|
DBGET_READ_COMMITTED,
|
|
DBGET_READ_UNCOMMITTED,
|
|
#endif
|
|
DBGET_CONSUME,
|
|
DBGET_CONSUME_WAIT,
|
|
DBGET_BOTH,
|
|
DBGET_GLOB,
|
|
DBGET_PART,
|
|
DBGET_RECNO,
|
|
DBGET_RMW,
|
|
DBGET_TXN,
|
|
DBGET_ENDARG
|
|
};
|
|
DB *heapdbp, *recdbp;
|
|
DBC *dbc;
|
|
DBT key, hkey, pkey, data, rdata, save;
|
|
DBTYPE ptype, type;
|
|
DB_HEAP_RID rid;
|
|
DB_TXN *txn;
|
|
Tcl_Obj **elemv, *retlist;
|
|
db_recno_t precno, recno;
|
|
u_int32_t cflag, flag, hflag, isdup, mflag, rmw;
|
|
int elemc, end, endarg, freekey, freedata, i;
|
|
int optindex, result, ret, useglob, useprecno, userecno;
|
|
char *arg, *pattern, *prefix, msg[MSG_SIZE];
|
|
void *dtmp, *ktmp;
|
|
#ifdef CONFIG_TEST
|
|
int bufsize, data_buf_size;
|
|
#endif
|
|
|
|
heapdbp = recdbp = NULL;
|
|
result = TCL_OK;
|
|
freekey = freedata = 0;
|
|
cflag = endarg = flag = hflag = mflag = rmw = 0;
|
|
useglob = userecno = 0;
|
|
txn = NULL;
|
|
pattern = prefix = NULL;
|
|
dtmp = ktmp = NULL;
|
|
ptype = DB_UNKNOWN;
|
|
#ifdef CONFIG_TEST
|
|
COMPQUIET(bufsize, 0);
|
|
data_buf_size = 0;
|
|
#endif
|
|
|
|
if (objc < 3) {
|
|
Tcl_WrongNumArgs(interp, 2, objv, "?-args? key");
|
|
return (TCL_ERROR);
|
|
}
|
|
|
|
memset(&key, 0, sizeof(key));
|
|
memset(&data, 0, sizeof(data));
|
|
memset(&save, 0, sizeof(save));
|
|
|
|
/* For the primary key in a pget call. */
|
|
memset(&pkey, 0, sizeof(pkey));
|
|
|
|
/* For the key/data used in heap am. */
|
|
memset(&hkey, 0, sizeof(hkey));
|
|
memset(&rdata, 0, sizeof(rdata));
|
|
|
|
/*
|
|
* Get the command name index from the object based on the options
|
|
* defined above.
|
|
*/
|
|
i = 2;
|
|
(void)dbp->get_type(dbp, &type);
|
|
end = objc;
|
|
while (i < end) {
|
|
if (Tcl_GetIndexFromObj(interp, objv[i], dbgetopts, "option",
|
|
TCL_EXACT, &optindex) != TCL_OK) {
|
|
arg = Tcl_GetStringFromObj(objv[i], NULL);
|
|
if (arg[0] == '-') {
|
|
result = IS_HELP(objv[i]);
|
|
goto out;
|
|
} else
|
|
Tcl_ResetResult(interp);
|
|
break;
|
|
}
|
|
i++;
|
|
switch ((enum dbgetopts)optindex) {
|
|
#ifdef CONFIG_TEST
|
|
case DBGET_DATA_BUF_SIZE:
|
|
result =
|
|
Tcl_GetIntFromObj(interp, objv[i], &data_buf_size);
|
|
if (result != TCL_OK)
|
|
goto out;
|
|
i++;
|
|
break;
|
|
case DBGET_MULTI:
|
|
mflag |= DB_MULTIPLE;
|
|
result =
|
|
Tcl_GetIntFromObj(interp, objv[i], &bufsize);
|
|
if (result != TCL_OK)
|
|
goto out;
|
|
i++;
|
|
/* heap doesnt support multi yet, just catch here */
|
|
if (type == DB_HEAP) {
|
|
Tcl_SetResult(interp,
|
|
"Heap doesnt support -multi", TCL_STATIC);
|
|
result = TCL_ERROR;
|
|
goto out;
|
|
}
|
|
break;
|
|
case DBGET_NOLEASE:
|
|
rmw |= DB_IGNORE_LEASE;
|
|
break;
|
|
case DBGET_READ_COMMITTED:
|
|
rmw |= DB_READ_COMMITTED;
|
|
break;
|
|
case DBGET_READ_UNCOMMITTED:
|
|
rmw |= DB_READ_UNCOMMITTED;
|
|
break;
|
|
#endif
|
|
case DBGET_BOTH:
|
|
/*
|
|
* Change 'end' and make sure we aren't already past
|
|
* the new end.
|
|
*/
|
|
if (i > objc - 2) {
|
|
Tcl_WrongNumArgs(interp, 2, objv,
|
|
"?-get_both key data?");
|
|
result = TCL_ERROR;
|
|
break;
|
|
}
|
|
end = objc - 2;
|
|
FLAG_CHECK(flag);
|
|
flag = DB_GET_BOTH;
|
|
break;
|
|
case DBGET_TXN:
|
|
if (i >= end) {
|
|
Tcl_WrongNumArgs(interp, 2, objv, "?-txn id?");
|
|
result = TCL_ERROR;
|
|
break;
|
|
}
|
|
arg = Tcl_GetStringFromObj(objv[i++], NULL);
|
|
txn = NAME_TO_TXN(arg);
|
|
if (txn == NULL) {
|
|
snprintf(msg, MSG_SIZE,
|
|
"Get: Invalid txn: %s\n", arg);
|
|
Tcl_SetResult(interp, msg, TCL_VOLATILE);
|
|
result = TCL_ERROR;
|
|
}
|
|
break;
|
|
case DBGET_GLOB:
|
|
useglob = 1;
|
|
end = objc - 1;
|
|
break;
|
|
case DBGET_CONSUME:
|
|
FLAG_CHECK(flag);
|
|
flag = DB_CONSUME;
|
|
break;
|
|
case DBGET_CONSUME_WAIT:
|
|
FLAG_CHECK(flag);
|
|
flag = DB_CONSUME_WAIT;
|
|
break;
|
|
case DBGET_RECNO:
|
|
end = objc - 1;
|
|
userecno = 1;
|
|
if (type != DB_RECNO &&
|
|
type != DB_QUEUE && type != DB_HEAP) {
|
|
FLAG_CHECK(flag);
|
|
flag = DB_SET_RECNO;
|
|
key.flags |= DB_DBT_MALLOC;
|
|
}
|
|
break;
|
|
case DBGET_RMW:
|
|
rmw |= DB_RMW;
|
|
break;
|
|
case DBGET_PART:
|
|
end = objc - 1;
|
|
if (i == end) {
|
|
Tcl_WrongNumArgs(interp, 2, objv,
|
|
"?-partial {offset length}?");
|
|
result = TCL_ERROR;
|
|
break;
|
|
}
|
|
/*
|
|
* Get sublist as {offset length}
|
|
*/
|
|
result = Tcl_ListObjGetElements(interp, objv[i++],
|
|
&elemc, &elemv);
|
|
if (elemc != 2) {
|
|
Tcl_SetResult(interp,
|
|
"List must be {offset length}", TCL_STATIC);
|
|
result = TCL_ERROR;
|
|
break;
|
|
}
|
|
save.flags = DB_DBT_PARTIAL;
|
|
result = _GetUInt32(interp, elemv[0], &save.doff);
|
|
if (result != TCL_OK)
|
|
break;
|
|
result = _GetUInt32(interp, elemv[1], &save.dlen);
|
|
/*
|
|
* NOTE: We don't check result here because all we'd
|
|
* do is break anyway, and we are doing that. If you
|
|
* add code here, you WILL need to add the check
|
|
* for result. (See the check for save.doff, a few
|
|
* lines above and copy that.)
|
|
*/
|
|
break;
|
|
case DBGET_ENDARG:
|
|
endarg = 1;
|
|
break;
|
|
}
|
|
if (result != TCL_OK)
|
|
break;
|
|
if (endarg)
|
|
break;
|
|
}
|
|
if (result != TCL_OK)
|
|
goto out;
|
|
|
|
/* We treat heap on the tcl interface just like we do recno */
|
|
if (type == DB_RECNO || type == DB_QUEUE || type == DB_HEAP)
|
|
userecno = 1;
|
|
|
|
/*
|
|
* Check args we have left versus the flags we were given.
|
|
* We might have 0, 1 or 2 left. If we have 0, it must
|
|
* be DB_CONSUME*, if 2, then DB_GET_BOTH, all others should
|
|
* be 1.
|
|
*/
|
|
if (((flag == DB_CONSUME || flag == DB_CONSUME_WAIT) && i != objc) ||
|
|
(flag == DB_GET_BOTH && i != objc - 2)) {
|
|
Tcl_SetResult(interp,
|
|
"Wrong number of key/data given based on flags specified\n",
|
|
TCL_STATIC);
|
|
result = TCL_ERROR;
|
|
goto out;
|
|
} else if (flag == 0 && i != objc - 1) {
|
|
Tcl_SetResult(interp,
|
|
"Wrong number of key/data given\n", TCL_STATIC);
|
|
result = TCL_ERROR;
|
|
goto out;
|
|
}
|
|
|
|
/* Find out whether the primary key should also be a recno. */
|
|
if (ispget && dbp->s_primary != NULL) {
|
|
(void)dbp->s_primary->get_type(dbp->s_primary, &ptype);
|
|
useprecno = ptype == DB_RECNO ||
|
|
ptype == DB_QUEUE || ptype == DB_HEAP;
|
|
} else
|
|
useprecno = 0;
|
|
|
|
/*
|
|
* Check for illegal combos of options.
|
|
*/
|
|
if (useglob && (userecno || flag == DB_SET_RECNO ||
|
|
type == DB_RECNO || type == DB_QUEUE || type == DB_HEAP)) {
|
|
Tcl_SetResult(interp,
|
|
"Cannot use -glob and record numbers.\n",
|
|
TCL_STATIC);
|
|
result = TCL_ERROR;
|
|
goto out;
|
|
}
|
|
#ifdef CONFIG_TEST
|
|
if (data_buf_size != 0 && flag == DB_GET_BOTH) {
|
|
Tcl_SetResult(interp,
|
|
"Only one of -data_buf_size or -get_both can be specified.\n",
|
|
TCL_STATIC);
|
|
result = TCL_ERROR;
|
|
goto out;
|
|
}
|
|
if (data_buf_size != 0 && mflag != 0) {
|
|
Tcl_SetResult(interp,
|
|
"Only one of -data_buf_size or -multi can be specified.\n",
|
|
TCL_STATIC);
|
|
result = TCL_ERROR;
|
|
goto out;
|
|
}
|
|
#endif
|
|
if (useglob && flag == DB_GET_BOTH) {
|
|
Tcl_SetResult(interp,
|
|
"Only one of -glob or -get_both can be specified.\n",
|
|
TCL_STATIC);
|
|
result = TCL_ERROR;
|
|
goto out;
|
|
}
|
|
|
|
if (useglob)
|
|
pattern = Tcl_GetStringFromObj(objv[objc - 1], NULL);
|
|
|
|
/*
|
|
* This is the list we return
|
|
*/
|
|
retlist = Tcl_NewListObj(0, NULL);
|
|
save.flags |= DB_DBT_MALLOC;
|
|
|
|
/*
|
|
*If we are using heap, either as a primary db or through a
|
|
* secondary relationship, then set up rdata. The data will
|
|
* contain a key for heap.
|
|
*/
|
|
if (type == DB_HEAP || ptype == DB_HEAP) {
|
|
hflag = flag;
|
|
rdata.ulen = sizeof(DB_HEAP_RID);
|
|
rdata.flags = DB_DBT_USERMEM;
|
|
ret = __os_malloc(dbp->env, rdata.ulen, &rdata.data);
|
|
if (ret != 0) {
|
|
Tcl_SetResult(interp, db_strerror(ret), TCL_STATIC);
|
|
return(TCL_ERROR);
|
|
}
|
|
}
|
|
|
|
/*
|
|
* isdup is used to know if we support duplicates. If not, we
|
|
* can just do a db->get call and avoid using cursors.
|
|
*/
|
|
if ((ret = dbp->get_flags(dbp, &isdup)) != 0) {
|
|
result = _ReturnSetup(interp, ret, DB_RETOK_STD(ret), "db get");
|
|
goto out;
|
|
}
|
|
isdup &= DB_DUP;
|
|
|
|
/*
|
|
* If the database doesn't support duplicates or we're performing
|
|
* ops that don't require returning multiple items, use DB->get
|
|
* instead of a cursor operation.
|
|
*/
|
|
/* MJB: need to verify - do we have a case where heap could have
|
|
* a secondary of BTREE, in which case isdup could be 1.
|
|
*/
|
|
if (pattern == NULL && (isdup == 0 || mflag != 0 ||
|
|
#ifdef CONFIG_TEST
|
|
data_buf_size != 0 ||
|
|
#endif
|
|
flag == DB_SET_RECNO || flag == DB_GET_BOTH ||
|
|
flag == DB_CONSUME || flag == DB_CONSUME_WAIT)) {
|
|
#ifdef CONFIG_TEST
|
|
if (data_buf_size == 0) {
|
|
F_CLR(&save, DB_DBT_USERMEM);
|
|
F_SET(&save, DB_DBT_MALLOC);
|
|
} else {
|
|
(void)__os_malloc(
|
|
NULL, (size_t)data_buf_size, &save.data);
|
|
save.ulen = (u_int32_t)data_buf_size;
|
|
F_CLR(&save, DB_DBT_MALLOC);
|
|
F_SET(&save, DB_DBT_USERMEM);
|
|
}
|
|
#endif
|
|
if (flag == DB_GET_BOTH) {
|
|
if (userecno) {
|
|
result = _GetUInt32(interp,
|
|
objv[(objc - 2)], &recno);
|
|
if (result == TCL_OK) {
|
|
key.data = &recno;
|
|
key.size = sizeof(db_recno_t);
|
|
} else
|
|
goto out;
|
|
} else {
|
|
/*
|
|
* Some get calls (SET_*) can change the
|
|
* key pointers. So, we need to store
|
|
* the allocated key space in a tmp.
|
|
*/
|
|
ret = _CopyObjBytes(interp, objv[objc-2],
|
|
&key.data, &key.size, &freekey);
|
|
if (ret != 0) {
|
|
result = _ReturnSetup(interp, ret,
|
|
DB_RETOK_DBGET(ret), "db get");
|
|
goto out;
|
|
}
|
|
}
|
|
ktmp = key.data;
|
|
/*
|
|
* Already checked args above. Fill in key and save.
|
|
* Save is used in the dbp->get call below to fill in
|
|
* data.
|
|
*
|
|
* If the "data" here is really a primary key--that
|
|
* is, if we're in a pget--and that primary key
|
|
* is a recno, treat it appropriately as an int.
|
|
*/
|
|
if (useprecno) {
|
|
result = _GetUInt32(interp,
|
|
objv[objc - 1], &precno);
|
|
if (result == TCL_OK) {
|
|
save.data = &precno;
|
|
save.size = sizeof(db_recno_t);
|
|
} else
|
|
goto out;
|
|
} else {
|
|
ret = _CopyObjBytes(interp, objv[objc-1],
|
|
&dtmp, &save.size, &freedata);
|
|
if (ret != 0) {
|
|
result = _ReturnSetup(interp, ret,
|
|
DB_RETOK_DBGET(ret), "db get");
|
|
goto out;
|
|
}
|
|
save.data = dtmp;
|
|
}
|
|
} else if (flag != DB_CONSUME && flag != DB_CONSUME_WAIT) {
|
|
if (userecno) {
|
|
result = _GetUInt32(
|
|
interp, objv[(objc - 1)], &recno);
|
|
if (result == TCL_OK) {
|
|
key.data = &recno;
|
|
key.size = sizeof(db_recno_t);
|
|
} else
|
|
goto out;
|
|
} else {
|
|
/*
|
|
* Some get calls (SET_*) can change the
|
|
* key pointers. So, we need to store
|
|
* the allocated key space in a tmp.
|
|
*/
|
|
ret = _CopyObjBytes(interp, objv[objc-1],
|
|
&key.data, &key.size, &freekey);
|
|
if (ret != 0) {
|
|
result = _ReturnSetup(interp, ret,
|
|
DB_RETOK_DBGET(ret), "db get");
|
|
goto out;
|
|
}
|
|
}
|
|
ktmp = key.data;
|
|
#ifdef CONFIG_TEST
|
|
if (mflag & DB_MULTIPLE) {
|
|
if ((ret = __os_malloc(dbp->env,
|
|
(size_t)bufsize, &save.data)) != 0) {
|
|
Tcl_SetResult(interp,
|
|
db_strerror(ret), TCL_STATIC);
|
|
goto out;
|
|
}
|
|
save.ulen = (u_int32_t)bufsize;
|
|
F_CLR(&save, DB_DBT_MALLOC);
|
|
F_SET(&save, DB_DBT_USERMEM);
|
|
}
|
|
#endif
|
|
}
|
|
|
|
data = save;
|
|
|
|
if (ispget) {
|
|
if (flag == DB_GET_BOTH) {
|
|
pkey.data = save.data;
|
|
pkey.size = save.size;
|
|
data.data = NULL;
|
|
data.size = 0;
|
|
/*
|
|
* If the primary is heap, we need to translate
|
|
* the given recno to the RID stored in the db.
|
|
*/
|
|
if (ptype == DB_HEAP) {
|
|
/* clear all flags for heap access */
|
|
hflag = 0;
|
|
recdbp = ((DBTCL_INFO *)
|
|
dbp->s_primary->api_internal)->hrdbp;
|
|
ret = recdbp->get(recdbp, txn,
|
|
&pkey, &rdata, hflag | rmw | mflag);
|
|
if (ret != 0) {
|
|
result = _ReturnSetup(interp,
|
|
ret, DB_RETOK_DBGET(ret),
|
|
"db get");
|
|
goto out;
|
|
}
|
|
}
|
|
}
|
|
F_SET(&pkey, DB_DBT_MALLOC);
|
|
_debug_check();
|
|
/*
|
|
* In case data has DB_DBT_PARTIAL set, we want
|
|
* to use rdata if we are heap.
|
|
*/
|
|
ret = dbp->pget(dbp, txn, &key,
|
|
ptype == DB_HEAP ? &rdata : &pkey, &data, flag | rmw);
|
|
/*
|
|
* If the primary database is a heap, we need to
|
|
* translate the RID returned in rdata into a recno using
|
|
* the auxiliary database.
|
|
*/
|
|
if (ptype == DB_HEAP && flag != DB_GET_BOTH) {
|
|
rid.pgno = ((DB_HEAP_RID *)rdata.data)->pgno;
|
|
rid.indx = ((DB_HEAP_RID *)rdata.data)->indx;
|
|
hkey.data = &rid;
|
|
hkey.ulen = hkey.size = rdata.size;
|
|
hkey.flags = DB_DBT_USERMEM;
|
|
heapdbp = ((DBTCL_INFO *)
|
|
dbp->s_primary->api_internal)->hsdbp;
|
|
ret = heapdbp->get(heapdbp,
|
|
txn, &hkey, &pkey, hflag | rmw | mflag);
|
|
}
|
|
|
|
} else if (type != DB_HEAP) {
|
|
_debug_check();
|
|
ret = dbp->get(dbp,
|
|
txn, &key, &data, flag | rmw | mflag);
|
|
} else {
|
|
_debug_check();
|
|
/*
|
|
* On the recno access we want to get the entire
|
|
* data as this is the heap key. Substitute
|
|
* rdata to get this data, and ignore DB_GET_BOTH
|
|
* if set at this point by using hflag. Other
|
|
* incorrect flags will get flagged within get code.
|
|
*/
|
|
recdbp = ((DBTCL_INFO *)dbp->api_internal)->hrdbp;
|
|
FLD_CLR(hflag, DB_GET_BOTH);
|
|
ret = recdbp->get(recdbp,
|
|
txn, &key, &rdata, hflag | rmw | mflag);
|
|
if (ret == 0) {
|
|
/* The rdata will be the key for the heap get. */
|
|
rid.pgno = ((DB_HEAP_RID *)rdata.data)->pgno;
|
|
rid.indx = ((DB_HEAP_RID *)rdata.data)->indx;
|
|
hkey.data = &rid;
|
|
hkey.ulen = hkey.size = sizeof(DB_HEAP_RID);
|
|
hkey.flags = DB_DBT_USERMEM;
|
|
ret = dbp->get(dbp,
|
|
txn, &hkey, &data, flag | rmw | mflag);
|
|
}
|
|
}
|
|
result = _ReturnSetup(interp, ret, DB_RETOK_DBGET(ret),
|
|
"db get");
|
|
if (ret == 0) {
|
|
/*
|
|
* Success. Return a list of the form {name value}
|
|
* If it was a recno in key.data, we need to convert
|
|
* into a string/object representation of that recno.
|
|
*/
|
|
if (mflag & DB_MULTIPLE)
|
|
result = _SetMultiList(interp,
|
|
retlist, &key, &data, type, flag);
|
|
else if (type == DB_RECNO || type == DB_QUEUE)
|
|
if (ispget)
|
|
result = _Set3DBTList(interp,
|
|
retlist, &key, 1, &pkey,
|
|
useprecno, &data);
|
|
else
|
|
result = _SetListRecnoElem(interp,
|
|
retlist, *(db_recno_t *)key.data,
|
|
data.data, data.size);
|
|
else if (type == DB_HEAP)
|
|
/* We return the key from the recno, and
|
|
* data from the heap.
|
|
*/
|
|
if (ispget)
|
|
result = _Set3DBTList(interp,
|
|
retlist, &key, 1, &pkey,
|
|
useprecno, &data);
|
|
else
|
|
result = _SetListRecnoElem(interp,
|
|
retlist, *(db_recno_t *)key.data,
|
|
data.data, data.size);
|
|
else {
|
|
if (ispget)
|
|
result = _Set3DBTList(interp,
|
|
retlist, &key, 0, &pkey,
|
|
useprecno, &data);
|
|
else
|
|
result = _SetListElem(interp, retlist,
|
|
key.data, key.size,
|
|
data.data, data.size);
|
|
}
|
|
}
|
|
/*
|
|
* Free space from DBT.
|
|
*
|
|
* If we set DB_DBT_MALLOC, we need to free the space if and
|
|
* only if we succeeded and if DB allocated anything (the
|
|
* pointer has changed from what we passed in). If
|
|
* DB_DBT_MALLOC is not set, this is a bulk get buffer, and
|
|
* needs to be freed no matter what.
|
|
*/
|
|
if (F_ISSET(&key, DB_DBT_MALLOC) && ret == 0 &&
|
|
key.data != ktmp)
|
|
__os_ufree(dbp->env, key.data);
|
|
if (F_ISSET(&data, DB_DBT_MALLOC) && ret == 0 &&
|
|
data.data != dtmp)
|
|
__os_ufree(dbp->env, data.data);
|
|
else if (!F_ISSET(&data, DB_DBT_MALLOC))
|
|
__os_free(dbp->env, data.data);
|
|
if (ispget && ret == 0 && pkey.data != save.data)
|
|
__os_ufree(dbp->env, pkey.data);
|
|
if (F_ISSET(&rdata, DB_DBT_USERMEM))
|
|
__os_free(dbp->env, rdata.data);
|
|
if (result == TCL_OK)
|
|
Tcl_SetObjResult(interp, retlist);
|
|
goto out;
|
|
}
|
|
|
|
if (userecno) {
|
|
result = _GetUInt32(interp, objv[(objc - 1)], &recno);
|
|
if (result == TCL_OK) {
|
|
key.data = &recno;
|
|
key.size = sizeof(db_recno_t);
|
|
} else
|
|
goto out;
|
|
} else {
|
|
/*
|
|
* Some get calls (SET_*) can change the
|
|
* key pointers. So, we need to store
|
|
* the allocated key space in a tmp.
|
|
*/
|
|
ret = _CopyObjBytes(interp, objv[objc-1], &key.data,
|
|
&key.size, &freekey);
|
|
if (ret != 0) {
|
|
result = _ReturnSetup(interp, ret,
|
|
DB_RETOK_DBGET(ret), "db get");
|
|
return (result);
|
|
}
|
|
}
|
|
ktmp = key.data;
|
|
ret = dbp->cursor(dbp, txn, &dbc, 0);
|
|
result = _ReturnSetup(interp, ret, DB_RETOK_STD(ret), "db cursor");
|
|
if (result == TCL_ERROR)
|
|
goto out;
|
|
|
|
/*
|
|
* At this point, we have a cursor, if we have a pattern,
|
|
* we go to the nearest one and step forward until we don't
|
|
* have any more that match the pattern prefix. If we have
|
|
* an exact key, we go to that key position, and step through
|
|
* all the duplicates. In either case we build up a list of
|
|
* the form {{key data} {key data}...} along the way.
|
|
*/
|
|
memset(&data, 0, sizeof(data));
|
|
/*
|
|
* Restore any "partial" info we have saved.
|
|
*/
|
|
data = save;
|
|
if (pattern) {
|
|
/*
|
|
* Note, prefix is returned in new space. Must free it.
|
|
*/
|
|
ret = _GetGlobPrefix(pattern, &prefix);
|
|
if (ret) {
|
|
result = TCL_ERROR;
|
|
Tcl_SetResult(interp,
|
|
"Unable to allocate pattern space", TCL_STATIC);
|
|
goto out1;
|
|
}
|
|
key.data = prefix;
|
|
key.size = (u_int32_t)strlen(prefix);
|
|
/*
|
|
* If they give us an empty pattern string
|
|
* (i.e. -glob *), go through entire DB.
|
|
*/
|
|
if (strlen(prefix) == 0)
|
|
cflag = DB_FIRST;
|
|
else
|
|
cflag = DB_SET_RANGE;
|
|
} else
|
|
cflag = DB_SET;
|
|
if (ispget) {
|
|
_debug_check();
|
|
F_SET(&pkey, DB_DBT_MALLOC);
|
|
ret = dbc->pget(dbc, &key, &pkey, &data, cflag | rmw);
|
|
} else {
|
|
_debug_check();
|
|
ret = dbc->get(dbc, &key, &data, cflag | rmw);
|
|
}
|
|
result = _ReturnSetup(interp, ret, DB_RETOK_DBCGET(ret),
|
|
"db get (cursor)");
|
|
if (result == TCL_ERROR)
|
|
goto out1;
|
|
if (pattern) {
|
|
if (ret == 0 && prefix != NULL &&
|
|
memcmp(key.data, prefix, strlen(prefix)) != 0) {
|
|
/*
|
|
* Free space from DB_DBT_MALLOC
|
|
*/
|
|
__os_ufree(dbp->env, data.data);
|
|
goto out1;
|
|
}
|
|
cflag = DB_NEXT;
|
|
} else
|
|
cflag = DB_NEXT_DUP;
|
|
|
|
while (ret == 0 && result == TCL_OK) {
|
|
/*
|
|
* Build up our {name value} sublist
|
|
*/
|
|
if (ispget)
|
|
result = _Set3DBTList(interp, retlist, &key, 0,
|
|
&pkey, useprecno, &data);
|
|
else
|
|
result = _SetListElem(interp, retlist,
|
|
key.data, key.size, data.data, data.size);
|
|
/*
|
|
* Free space from DB_DBT_MALLOC
|
|
*/
|
|
if (ispget)
|
|
__os_ufree(dbp->env, pkey.data);
|
|
__os_ufree(dbp->env, data.data);
|
|
if (result != TCL_OK)
|
|
break;
|
|
/*
|
|
* Append {name value} to return list
|
|
*/
|
|
memset(&key, 0, sizeof(key));
|
|
memset(&pkey, 0, sizeof(pkey));
|
|
memset(&data, 0, sizeof(data));
|
|
/*
|
|
* Restore any "partial" info we have saved.
|
|
*/
|
|
data = save;
|
|
if (ispget) {
|
|
F_SET(&pkey, DB_DBT_MALLOC);
|
|
ret = dbc->pget(dbc, &key, &pkey, &data, cflag | rmw);
|
|
} else
|
|
ret = dbc->get(dbc, &key, &data, cflag | rmw);
|
|
if (ret == 0 && prefix != NULL &&
|
|
memcmp(key.data, prefix, strlen(prefix)) != 0) {
|
|
/*
|
|
* Free space from DB_DBT_MALLOC
|
|
*/
|
|
__os_ufree(dbp->env, data.data);
|
|
break;
|
|
}
|
|
}
|
|
out1:
|
|
(void)dbc->close(dbc);
|
|
if (result == TCL_OK)
|
|
Tcl_SetObjResult(interp, retlist);
|
|
out:
|
|
/*
|
|
* _GetGlobPrefix(), the function which allocates prefix, works
|
|
* by copying and condensing another string. Thus prefix may
|
|
* have multiple nuls at the end, so we free using __os_free().
|
|
*/
|
|
if (prefix != NULL)
|
|
__os_free(dbp->env, prefix);
|
|
if (dtmp != NULL && freedata)
|
|
__os_free(dbp->env, dtmp);
|
|
if (ktmp != NULL && freekey)
|
|
__os_free(dbp->env, ktmp);
|
|
return (result);
|
|
}
|
|
|
|
/*
|
|
* tcl_db_delete --
|
|
*/
|
|
static int
|
|
tcl_DbDelete(interp, objc, objv, dbp)
|
|
Tcl_Interp *interp; /* Interpreter */
|
|
int objc; /* How many arguments? */
|
|
Tcl_Obj *CONST objv[]; /* The argument objects */
|
|
DB *dbp; /* Database pointer */
|
|
{
|
|
static const char *dbdelopts[] = {
|
|
"-consume",
|
|
"-glob",
|
|
"-multiple",
|
|
"-multiple_key",
|
|
"-txn",
|
|
NULL
|
|
};
|
|
enum dbdelopts {
|
|
DBDEL_CONSUME,
|
|
DBDEL_GLOB,
|
|
DBDEL_MULTIPLE,
|
|
DBDEL_MULTIPLE_KEY,
|
|
DBDEL_TXN
|
|
};
|
|
DB *recdbp;
|
|
DBC *dbc;
|
|
DBT key, data, hkey, rdata;
|
|
DBTYPE type;
|
|
DB_HEAP_RID rid;
|
|
DB_TXN *txn;
|
|
Tcl_Obj **elemv;
|
|
void *dtmp, *ktmp, *ptr;
|
|
db_recno_t recno;
|
|
int dlen, elemc, freekey, i, j, klen, optindex, result, ret;
|
|
u_int32_t dflag, flag, multiflag;
|
|
char *arg, *pattern, *prefix, msg[MSG_SIZE];
|
|
|
|
COMPQUIET(recdbp, NULL);
|
|
result = TCL_OK;
|
|
freekey = 0;
|
|
dflag = 0;
|
|
multiflag = 0;
|
|
pattern = prefix = NULL;
|
|
txn = NULL;
|
|
if (objc < 3) {
|
|
Tcl_WrongNumArgs(interp, 2, objv, "?-args? key");
|
|
return (TCL_ERROR);
|
|
}
|
|
|
|
dtmp = ktmp = NULL;
|
|
memset(&key, 0, sizeof(key));
|
|
memset(&hkey, 0, sizeof(hkey));
|
|
memset(&rdata, 0, sizeof(rdata));
|
|
|
|
/*
|
|
* The first arg must be -glob, -txn or a list of keys.
|
|
*/
|
|
i = 2;
|
|
while (i < objc) {
|
|
if (Tcl_GetIndexFromObj(interp, objv[i], dbdelopts, "option",
|
|
TCL_EXACT, &optindex) != TCL_OK) {
|
|
/*
|
|
* If we don't have a -glob or -txn, then the remaining
|
|
* args must be exact keys. Reset the result so we
|
|
* don't get an errant error message if there is another
|
|
* error.
|
|
*/
|
|
if (IS_HELP(objv[i]) == TCL_OK)
|
|
return (TCL_OK);
|
|
Tcl_ResetResult(interp);
|
|
break;
|
|
}
|
|
i++;
|
|
switch ((enum dbdelopts)optindex) {
|
|
case DBDEL_TXN:
|
|
if (i == objc) {
|
|
/*
|
|
* Someone could conceivably have a key of
|
|
* the same name. So just break and use it.
|
|
*/
|
|
i--;
|
|
break;
|
|
}
|
|
arg = Tcl_GetStringFromObj(objv[i++], NULL);
|
|
txn = NAME_TO_TXN(arg);
|
|
if (txn == NULL) {
|
|
snprintf(msg, MSG_SIZE,
|
|
"Delete: Invalid txn: %s\n", arg);
|
|
Tcl_SetResult(interp, msg, TCL_VOLATILE);
|
|
result = TCL_ERROR;
|
|
}
|
|
break;
|
|
case DBDEL_GLOB:
|
|
/*
|
|
* Get the pattern. Get the prefix and use cursors to
|
|
* get all the data items.
|
|
*/
|
|
if (i == objc) {
|
|
/*
|
|
* Someone could conceivably have a key of
|
|
* the same name. So just break and use it.
|
|
*/
|
|
i--;
|
|
break;
|
|
}
|
|
pattern = Tcl_GetStringFromObj(objv[i++], NULL);
|
|
break;
|
|
case DBDEL_CONSUME:
|
|
FLAG_CHECK(dflag);
|
|
dflag = DB_CONSUME;
|
|
break;
|
|
case DBDEL_MULTIPLE:
|
|
FLAG_CHECK(multiflag);
|
|
multiflag |= DB_MULTIPLE;
|
|
break;
|
|
case DBDEL_MULTIPLE_KEY:
|
|
FLAG_CHECK(multiflag);
|
|
multiflag |= DB_MULTIPLE_KEY;
|
|
break;
|
|
}
|
|
if (result != TCL_OK)
|
|
break;
|
|
}
|
|
|
|
if (result != TCL_OK)
|
|
goto out;
|
|
/*
|
|
* XXX
|
|
* For consistency with get, we have decided for the moment, to
|
|
* allow -glob, or one key, not many. The code was originally
|
|
* written to take many keys and we'll leave it that way, because
|
|
* tcl_DbGet may one day accept many disjoint keys to get, rather
|
|
* than one, and at that time we'd make delete be consistent. In
|
|
* any case, the code is already here and there is no need to remove,
|
|
* just check that we only have one arg left.
|
|
*
|
|
* If we have a pattern AND more keys to process, there is an error.
|
|
* Either we have some number of exact keys, or we have a pattern.
|
|
*/
|
|
if (pattern == NULL) {
|
|
if (i != (objc - 1)) {
|
|
Tcl_WrongNumArgs(
|
|
interp, 2, objv, "?args? -glob pattern | key");
|
|
result = TCL_ERROR;
|
|
goto out;
|
|
}
|
|
} else {
|
|
if (i != objc) {
|
|
Tcl_WrongNumArgs(
|
|
interp, 2, objv, "?args? -glob pattern | key");
|
|
result = TCL_ERROR;
|
|
goto out;
|
|
}
|
|
}
|
|
|
|
/*
|
|
* If we have remaining args, they are all exact keys. Call
|
|
* DB->del on each of those keys.
|
|
*
|
|
* If it is a RECNO database, the key is a record number and must be
|
|
* setup up to contain a db_recno_t. Otherwise the key is a "string".
|
|
*/
|
|
(void)dbp->get_type(dbp, &type);
|
|
ret = 0;
|
|
while (i < objc && ret == 0) {
|
|
memset(&key, 0, sizeof(key));
|
|
if (multiflag == DB_MULTIPLE) {
|
|
/*
|
|
* Heap is not supporting bulk on initially, so
|
|
* check if type is heap and if so throw an error.
|
|
*/
|
|
if (type == DB_HEAP) {
|
|
Tcl_SetResult(interp,
|
|
"Heap doesnt support -multiple", TCL_STATIC);
|
|
result = TCL_ERROR;
|
|
goto out;
|
|
}
|
|
|
|
/*
|
|
* To work out how big a buffer is needed, we first
|
|
* need to find out the total length of the data and
|
|
* the number of data items (elemc).
|
|
*/
|
|
ktmp = Tcl_GetByteArrayFromObj(objv[i], &klen);
|
|
result = Tcl_ListObjGetElements(interp, objv[i++],
|
|
&elemc, &elemv);
|
|
if (result != TCL_OK)
|
|
return (result);
|
|
|
|
memset(&key, 0, sizeof(key));
|
|
key.ulen = DB_ALIGN((u_int32_t)klen + (u_int32_t)elemc
|
|
* sizeof(u_int32_t) * 2, 1024UL);
|
|
key.flags = DB_DBT_USERMEM | DB_DBT_BULK;
|
|
if ((ret =
|
|
__os_malloc(dbp->env, key.ulen, &key.data)) != 0)
|
|
return (ret);
|
|
freekey = 1;
|
|
|
|
if (type == DB_RECNO || type == DB_QUEUE) {
|
|
DB_MULTIPLE_RECNO_WRITE_INIT(ptr, &key);
|
|
for (j = 0; j < elemc; j++) {
|
|
result =
|
|
_GetUInt32(interp,
|
|
elemv[j], &recno);
|
|
if (result != TCL_OK)
|
|
return (result);
|
|
DB_MULTIPLE_RECNO_WRITE_NEXT(ptr,
|
|
&key, recno, dtmp, 0);
|
|
DB_ASSERT(dbp->env, ptr != NULL);
|
|
}
|
|
} else {
|
|
DB_MULTIPLE_WRITE_INIT(ptr, &key);
|
|
for (j = 0; j < elemc; j++) {
|
|
ktmp = Tcl_GetByteArrayFromObj(elemv[j],
|
|
&klen);
|
|
DB_MULTIPLE_WRITE_NEXT(ptr,
|
|
&key, ktmp, (u_int32_t)klen);
|
|
DB_ASSERT(dbp->env, ptr != NULL);
|
|
}
|
|
}
|
|
} else if (multiflag == DB_MULTIPLE_KEY) {
|
|
/*
|
|
* Heap is not supporting bulk on initially, so
|
|
* check if type is heap and if so throw an error.
|
|
*/
|
|
if (type == DB_HEAP) {
|
|
Tcl_SetResult(interp,
|
|
"Heap doesnt support -multiple_key", TCL_STATIC);
|
|
result = TCL_ERROR;
|
|
goto out;
|
|
}
|
|
|
|
/*
|
|
* To work out how big a buffer is needed, we first
|
|
* need to find out the total length of the data (len)
|
|
* and the number of data items (elemc).
|
|
*/
|
|
ktmp = Tcl_GetByteArrayFromObj(objv[i], &klen);
|
|
result = Tcl_ListObjGetElements(interp, objv[i++],
|
|
&elemc, &elemv);
|
|
if (result != TCL_OK)
|
|
return (result);
|
|
|
|
memset(&key, 0, sizeof(key));
|
|
key.ulen = DB_ALIGN((u_int32_t)klen +
|
|
(u_int32_t)elemc * sizeof(u_int32_t) * 2, 1024UL);
|
|
key.flags = DB_DBT_USERMEM | DB_DBT_BULK;
|
|
if ((ret =
|
|
__os_malloc(dbp->env, key.ulen, &key.data)) != 0)
|
|
return (ret);
|
|
freekey = 1;
|
|
|
|
if (type == DB_RECNO || type == DB_QUEUE) {
|
|
DB_MULTIPLE_RECNO_WRITE_INIT(ptr, &key);
|
|
for (j = 0; j + 1 < elemc; j += 2) {
|
|
result =
|
|
_GetUInt32(interp,
|
|
elemv[j], &recno);
|
|
if (result != TCL_OK)
|
|
return (result);
|
|
dtmp = Tcl_GetByteArrayFromObj(
|
|
elemv[j + 1], &dlen);
|
|
DB_MULTIPLE_RECNO_WRITE_NEXT(ptr,
|
|
&key, recno, dtmp, (u_int32_t)dlen);
|
|
DB_ASSERT(dbp->env, ptr != NULL);
|
|
}
|
|
} else {
|
|
DB_MULTIPLE_WRITE_INIT(ptr, &key);
|
|
for (j = 0; j + 1 < elemc; j += 2) {
|
|
ktmp = Tcl_GetByteArrayFromObj(
|
|
elemv[j], &klen);
|
|
dtmp = Tcl_GetByteArrayFromObj(
|
|
elemv[j + 1], &dlen);
|
|
DB_MULTIPLE_KEY_WRITE_NEXT(ptr,
|
|
&key, ktmp, (u_int32_t)klen,
|
|
dtmp, (u_int32_t)dlen);
|
|
DB_ASSERT(dbp->env, ptr != NULL);
|
|
}
|
|
}
|
|
} else if (type == DB_RECNO || type == DB_QUEUE) {
|
|
result = _GetUInt32(interp, objv[i++], &recno);
|
|
if (result == TCL_OK) {
|
|
key.data = &recno;
|
|
key.size = sizeof(db_recno_t);
|
|
} else
|
|
return (result);
|
|
} else if (type == DB_HEAP) {
|
|
/* For heap, the incoming key is to a recno db */
|
|
result = _GetUInt32(interp, objv[i++], &recno);
|
|
if (result == TCL_OK) {
|
|
key.data = &recno;
|
|
key.size = sizeof(db_recno_t);
|
|
} else
|
|
return (result);
|
|
} else {
|
|
ret = _CopyObjBytes(interp, objv[i++], &ktmp,
|
|
&key.size, &freekey);
|
|
if (ret != 0) {
|
|
result = _ReturnSetup(interp, ret,
|
|
DB_RETOK_DBDEL(ret), "db del");
|
|
return (result);
|
|
}
|
|
key.data = ktmp;
|
|
}
|
|
_debug_check();
|
|
if (type != DB_HEAP)
|
|
ret = dbp->del(dbp, txn, &key, dflag | multiflag);
|
|
else {
|
|
/* set up the recno data, which will be heap key */
|
|
rdata.ulen = sizeof(DB_HEAP_RID);
|
|
rdata.flags = DB_DBT_USERMEM;
|
|
if (rdata.data == NULL) {
|
|
ret = __os_malloc(dbp->env,
|
|
rdata.ulen, &rdata.data);
|
|
if (ret != 0) {
|
|
Tcl_SetResult(interp,
|
|
db_strerror(ret), TCL_STATIC);
|
|
return(TCL_ERROR);
|
|
}
|
|
}
|
|
/*
|
|
* 3 steps for delete, first get the key from the
|
|
* recno db and create heap key, second delete
|
|
* recno record, third delete heap record.
|
|
*/
|
|
recdbp = ((DBTCL_INFO *)dbp->api_internal)->hrdbp;
|
|
ret = recdbp->get(recdbp, txn, &key, &rdata, 0);
|
|
if (ret) {
|
|
result = _ReturnSetup(interp, ret,
|
|
DB_RETOK_STD(ret), "db del heap db");
|
|
goto out;
|
|
}
|
|
|
|
/* Set up the key for heap */
|
|
rid.pgno = ((DB_HEAP_RID *)rdata.data)->pgno;
|
|
rid.indx = ((DB_HEAP_RID *)rdata.data)->indx;
|
|
hkey.data = &rid;
|
|
hkey.ulen = hkey.size = sizeof(DB_HEAP_RID);
|
|
hkey.flags = DB_DBT_USERMEM;
|
|
/* 2nd: Delete from recno db */
|
|
_debug_check();
|
|
ret = recdbp->del(recdbp, txn, &key, dflag | multiflag);
|
|
if (ret) {
|
|
result = _ReturnSetup(interp, ret,
|
|
DB_RETOK_STD(ret), "db del heap db");
|
|
goto out;
|
|
}
|
|
|
|
/* 3rd: Delete from heap db */
|
|
_debug_check();
|
|
ret = dbp->del(dbp, txn, &hkey, dflag | multiflag);
|
|
if (ret) {
|
|
result = _ReturnSetup(interp, ret,
|
|
DB_RETOK_STD(ret), "db del heap db");
|
|
goto out;
|
|
}
|
|
}
|
|
/*
|
|
* If we have any error, set up return result and stop
|
|
* processing keys.
|
|
*/
|
|
if (freekey && key.data != NULL)
|
|
__os_free(dbp->env, key.data);
|
|
if (ret != 0)
|
|
break;
|
|
}
|
|
result = _ReturnSetup(interp, ret, DB_RETOK_DBDEL(ret), "db del");
|
|
|
|
/*
|
|
* At this point we've either finished or, if we have a pattern,
|
|
* we go to the nearest one and step forward until we don't
|
|
* have any more that match the pattern prefix.
|
|
*/
|
|
if (pattern) {
|
|
ret = dbp->cursor(dbp, txn, &dbc, 0);
|
|
if (ret != 0) {
|
|
result = _ReturnSetup(interp, ret, DB_RETOK_STD(ret),
|
|
"db cursor");
|
|
goto out;
|
|
}
|
|
/*
|
|
* Note, prefix is returned in new space. Must free it.
|
|
*/
|
|
memset(&key, 0, sizeof(key));
|
|
memset(&data, 0, sizeof(data));
|
|
ret = _GetGlobPrefix(pattern, &prefix);
|
|
if (ret) {
|
|
result = TCL_ERROR;
|
|
Tcl_SetResult(interp,
|
|
"Unable to allocate pattern space", TCL_STATIC);
|
|
goto out;
|
|
}
|
|
key.data = prefix;
|
|
key.size = (u_int32_t)strlen(prefix);
|
|
if (strlen(prefix) == 0)
|
|
flag = DB_FIRST;
|
|
else
|
|
flag = DB_SET_RANGE;
|
|
ret = dbc->get(dbc, &key, &data, flag);
|
|
while (ret == 0 &&
|
|
memcmp(key.data, prefix, strlen(prefix)) == 0) {
|
|
/*
|
|
* Each time through here the cursor is pointing
|
|
* at the current valid item. Delete it and
|
|
* move ahead.
|
|
*/
|
|
_debug_check();
|
|
ret = dbc->del(dbc, dflag);
|
|
if (ret != 0) {
|
|
result = _ReturnSetup(interp, ret,
|
|
DB_RETOK_DBCDEL(ret), "db c_del");
|
|
break;
|
|
}
|
|
/*
|
|
* Deleted the current, now move to the next item
|
|
* in the list, check if it matches the prefix pattern.
|
|
*/
|
|
memset(&key, 0, sizeof(key));
|
|
memset(&data, 0, sizeof(data));
|
|
ret = dbc->get(dbc, &key, &data, DB_NEXT);
|
|
}
|
|
if (ret == DB_NOTFOUND)
|
|
ret = 0;
|
|
/*
|
|
* _GetGlobPrefix(), the function which allocates prefix, works
|
|
* by copying and condensing another string. Thus prefix may
|
|
* have multiple nuls at the end, so we free using __os_free().
|
|
*/
|
|
__os_free(dbp->env, prefix);
|
|
(void)dbc->close(dbc);
|
|
result = _ReturnSetup(interp, ret, DB_RETOK_STD(ret), "db del");
|
|
}
|
|
out:
|
|
/* If we allocated memory for a heap delete, release it. */
|
|
if (rdata.data != NULL)
|
|
__os_free(dbp->env, rdata.data);
|
|
|
|
return (result);
|
|
}
|
|
|
|
/*
|
|
* tcl_db_cursor --
|
|
*/
|
|
static int
|
|
tcl_DbCursor(interp, objc, objv, dbp, dbcp)
|
|
Tcl_Interp *interp; /* Interpreter */
|
|
int objc; /* How many arguments? */
|
|
Tcl_Obj *CONST objv[]; /* The argument objects */
|
|
DB *dbp; /* Database pointer */
|
|
DBC **dbcp; /* Return cursor pointer */
|
|
{
|
|
static const char *dbcuropts[] = {
|
|
#ifdef CONFIG_TEST
|
|
"-read_committed",
|
|
"-read_uncommitted",
|
|
"-update",
|
|
#endif
|
|
"-bulk",
|
|
"-txn",
|
|
NULL
|
|
};
|
|
enum dbcuropts {
|
|
#ifdef CONFIG_TEST
|
|
DBCUR_READ_COMMITTED,
|
|
DBCUR_READ_UNCOMMITTED,
|
|
DBCUR_UPDATE,
|
|
#endif
|
|
DBCUR_BULK,
|
|
DBCUR_TXN
|
|
};
|
|
DB_TXN *txn;
|
|
u_int32_t flag;
|
|
int i, optindex, result, ret;
|
|
char *arg, msg[MSG_SIZE];
|
|
|
|
result = TCL_OK;
|
|
flag = 0;
|
|
txn = NULL;
|
|
i = 2;
|
|
while (i < objc) {
|
|
if (Tcl_GetIndexFromObj(interp, objv[i], dbcuropts, "option",
|
|
TCL_EXACT, &optindex) != TCL_OK) {
|
|
result = IS_HELP(objv[i]);
|
|
goto out;
|
|
}
|
|
i++;
|
|
switch ((enum dbcuropts)optindex) {
|
|
#ifdef CONFIG_TEST
|
|
case DBCUR_READ_COMMITTED:
|
|
flag |= DB_READ_COMMITTED;
|
|
break;
|
|
case DBCUR_READ_UNCOMMITTED:
|
|
flag |= DB_READ_UNCOMMITTED;
|
|
break;
|
|
case DBCUR_UPDATE:
|
|
flag |= DB_WRITECURSOR;
|
|
break;
|
|
#endif
|
|
case DBCUR_BULK:
|
|
flag |= DB_CURSOR_BULK;
|
|
break;
|
|
case DBCUR_TXN:
|
|
if (i == objc) {
|
|
Tcl_WrongNumArgs(interp, 2, objv, "?-txn id?");
|
|
result = TCL_ERROR;
|
|
break;
|
|
}
|
|
arg = Tcl_GetStringFromObj(objv[i++], NULL);
|
|
txn = NAME_TO_TXN(arg);
|
|
if (txn == NULL) {
|
|
snprintf(msg, MSG_SIZE,
|
|
"Cursor: Invalid txn: %s\n", arg);
|
|
Tcl_SetResult(interp, msg, TCL_VOLATILE);
|
|
result = TCL_ERROR;
|
|
}
|
|
break;
|
|
}
|
|
if (result != TCL_OK)
|
|
break;
|
|
}
|
|
if (result != TCL_OK)
|
|
goto out;
|
|
|
|
_debug_check();
|
|
ret = dbp->cursor(dbp, txn, dbcp, flag);
|
|
if (ret != 0)
|
|
result = _ErrorSetup(interp, ret, "db cursor");
|
|
out:
|
|
return (result);
|
|
}
|
|
|
|
/*
|
|
* tcl_DbAssociate --
|
|
* Call DB->associate().
|
|
*/
|
|
static int
|
|
tcl_DbAssociate(interp, objc, objv, dbp)
|
|
Tcl_Interp *interp;
|
|
int objc;
|
|
Tcl_Obj *CONST objv[];
|
|
DB *dbp;
|
|
{
|
|
static const char *dbaopts[] = {
|
|
"-create",
|
|
"-immutable_key",
|
|
"-txn",
|
|
NULL
|
|
};
|
|
enum dbaopts {
|
|
DBA_CREATE,
|
|
DBA_IMMUTABLE_KEY,
|
|
DBA_TXN
|
|
};
|
|
DB *sdbp;
|
|
DB_TXN *txn;
|
|
DBTCL_INFO *sdbip;
|
|
int i, optindex, result, ret;
|
|
char *arg, msg[MSG_SIZE];
|
|
u_int32_t flag;
|
|
|
|
txn = NULL;
|
|
result = TCL_OK;
|
|
flag = 0;
|
|
if (objc < 2) {
|
|
Tcl_WrongNumArgs(interp, 2, objv, "[callback] secondary");
|
|
return (TCL_ERROR);
|
|
}
|
|
|
|
i = 2;
|
|
while (i < objc) {
|
|
if (Tcl_GetIndexFromObj(interp, objv[i], dbaopts, "option",
|
|
TCL_EXACT, &optindex) != TCL_OK) {
|
|
result = IS_HELP(objv[i]);
|
|
if (result == TCL_OK)
|
|
return (result);
|
|
result = TCL_OK;
|
|
Tcl_ResetResult(interp);
|
|
break;
|
|
}
|
|
i++;
|
|
switch ((enum dbaopts)optindex) {
|
|
case DBA_CREATE:
|
|
flag |= DB_CREATE;
|
|
break;
|
|
case DBA_IMMUTABLE_KEY:
|
|
flag |= DB_IMMUTABLE_KEY;
|
|
break;
|
|
case DBA_TXN:
|
|
if (i > (objc - 1)) {
|
|
Tcl_WrongNumArgs(interp, 2, objv, "?-txn id?");
|
|
result = TCL_ERROR;
|
|
break;
|
|
}
|
|
arg = Tcl_GetStringFromObj(objv[i++], NULL);
|
|
txn = NAME_TO_TXN(arg);
|
|
if (txn == NULL) {
|
|
snprintf(msg, MSG_SIZE,
|
|
"Associate: Invalid txn: %s\n", arg);
|
|
Tcl_SetResult(interp, msg, TCL_VOLATILE);
|
|
result = TCL_ERROR;
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
if (result != TCL_OK)
|
|
return (result);
|
|
|
|
/*
|
|
* Better be 1 or 2 args left. The last arg must be the sdb
|
|
* handle. If 2 args then objc-2 is the callback proc, else
|
|
* we have a NULL callback.
|
|
*/
|
|
/* Get the secondary DB handle. */
|
|
arg = Tcl_GetStringFromObj(objv[objc - 1], NULL);
|
|
sdbp = NAME_TO_DB(arg);
|
|
if (sdbp == NULL) {
|
|
snprintf(msg, MSG_SIZE,
|
|
"Associate: Invalid database handle: %s\n", arg);
|
|
Tcl_SetResult(interp, msg, TCL_VOLATILE);
|
|
return (TCL_ERROR);
|
|
}
|
|
|
|
/*
|
|
* The callback is simply a Tcl object containing the name
|
|
* of the callback proc, which is the second-to-last argument.
|
|
*
|
|
* Note that the callback needs to go in the *secondary* DB handle's
|
|
* info struct; we may have multiple secondaries with different
|
|
* callbacks.
|
|
*/
|
|
sdbip = (DBTCL_INFO *)sdbp->api_internal;
|
|
|
|
#ifdef CONFIG_TEST
|
|
if (i != objc - 1) {
|
|
#else
|
|
if (i != objc - 1) {
|
|
#endif
|
|
/*
|
|
* We have 2 args, get the callback.
|
|
*/
|
|
sdbip->i_second_call = objv[objc - 2];
|
|
Tcl_IncrRefCount(sdbip->i_second_call);
|
|
|
|
/* Now call associate. */
|
|
_debug_check();
|
|
|
|
ret = dbp->associate(dbp,
|
|
txn, sdbp, tcl_second_call, flag);
|
|
} else {
|
|
/*
|
|
* We have a NULL callback.
|
|
*/
|
|
sdbip->i_second_call = NULL;
|
|
ret = dbp->associate(dbp, txn, sdbp, NULL, flag);
|
|
}
|
|
result = _ReturnSetup(interp, ret, DB_RETOK_STD(ret), "associate");
|
|
|
|
return (result);
|
|
}
|
|
|
|
/*
|
|
* tcl_DbAssociateForeign --
|
|
* Call DB->associate_foreign().
|
|
*/
|
|
static int
|
|
tcl_DbAssociateForeign(interp, objc, objv, dbp)
|
|
Tcl_Interp *interp;
|
|
int objc;
|
|
Tcl_Obj *CONST objv[];
|
|
DB *dbp;
|
|
{
|
|
static const char *dbafopts[] = {
|
|
"-abort",
|
|
"-cascade",
|
|
"-nullify",
|
|
NULL
|
|
};
|
|
enum dbafopts {
|
|
DBAF_ABORT,
|
|
DBAF_CASCADE,
|
|
DBAF_NULLIFY
|
|
};
|
|
DB *sdbp;
|
|
DBTCL_INFO *sdbip;
|
|
int i, optindex, result, ret;
|
|
char *arg, msg[MSG_SIZE];
|
|
u_int32_t flag;
|
|
int (*callback)(DB *, const DBT *, DBT *, const DBT*, int*);
|
|
|
|
result = TCL_OK;
|
|
flag = 0;
|
|
if (objc < 2) {
|
|
Tcl_WrongNumArgs(interp, 2, objv,
|
|
"?args? ?callback? secondary");
|
|
return (TCL_ERROR);
|
|
}
|
|
|
|
i = 2;
|
|
while (i < objc) {
|
|
if (Tcl_GetIndexFromObj(interp, objv[i], dbafopts, "option",
|
|
TCL_EXACT, &optindex) != TCL_OK) {
|
|
result = IS_HELP(objv[i]);
|
|
if (result == TCL_OK)
|
|
return (result);
|
|
result = TCL_OK;
|
|
Tcl_ResetResult(interp);
|
|
break;
|
|
}
|
|
i++;
|
|
switch ((enum dbafopts)optindex) {
|
|
case DBAF_ABORT:
|
|
flag |= DB_FOREIGN_ABORT;
|
|
break;
|
|
case DBAF_CASCADE:
|
|
flag |= DB_FOREIGN_CASCADE;
|
|
break;
|
|
case DBAF_NULLIFY:
|
|
if (i > (objc - 1)) {
|
|
Tcl_WrongNumArgs(interp, 2,
|
|
objv, "?-nullify ?callback?? secondary");
|
|
result = TCL_ERROR;
|
|
break;
|
|
}
|
|
flag |= DB_FOREIGN_NULLIFY;
|
|
break;
|
|
}
|
|
}
|
|
if (result != TCL_OK)
|
|
return (result);
|
|
|
|
/*
|
|
* Better be 1 or 2 args left. The last arg must be the sdb
|
|
* handle. If 2 args then objc-2 is the callback proc, else
|
|
* we have a NULL callback.
|
|
*/
|
|
/* Get the secondary DB handle. */
|
|
arg = Tcl_GetStringFromObj(objv[objc - 1], NULL);
|
|
sdbp = NAME_TO_DB(arg);
|
|
if (sdbp == NULL) {
|
|
snprintf(msg, MSG_SIZE,
|
|
"Associate_foreign: Invalid database handle: %s\n", arg);
|
|
Tcl_SetResult(interp, msg, TCL_VOLATILE);
|
|
return (TCL_ERROR);
|
|
}
|
|
|
|
/*
|
|
* The callback is simply a Tcl object containing the name
|
|
* of the callback proc, which is the second-to-last argument.
|
|
*
|
|
* Note that the callback needs to go in the *secondary* DB handle's
|
|
* info struct; we may have multiple secondaries with different
|
|
* callbacks.
|
|
*/
|
|
sdbip = (DBTCL_INFO *)sdbp->api_internal;
|
|
|
|
callback = NULL;
|
|
sdbip->i_foreign_call = NULL;
|
|
|
|
if (i != objc -1) {
|
|
/* Set the callback */
|
|
callback = tcl_foreign_call;
|
|
sdbip->i_foreign_call = objv[objc - 2];
|
|
Tcl_IncrRefCount(sdbip->i_foreign_call);
|
|
} else {
|
|
/*
|
|
* We have a NULL callback.
|
|
*/
|
|
callback = NULL;
|
|
}
|
|
|
|
_debug_check();
|
|
/* Now call associate_foreign. */
|
|
ret = dbp->associate_foreign(dbp, sdbp, callback, flag);
|
|
|
|
result = _ReturnSetup(interp, ret, DB_RETOK_STD(ret), "associate_foreign");
|
|
|
|
return (result);
|
|
}
|
|
|
|
/*
|
|
* tcl_second_call --
|
|
* Callback function for secondary indices. Get the callback
|
|
* out of ip->i_second_call and call it.
|
|
*/
|
|
static int
|
|
tcl_second_call(dbp, pkey, data, skey)
|
|
DB *dbp;
|
|
const DBT *pkey, *data;
|
|
DBT *skey;
|
|
{
|
|
DBT *tskey;
|
|
DBTCL_INFO *ip;
|
|
Tcl_Interp *interp;
|
|
Tcl_Obj *pobj, *dobj, *objv[3], *robj, **skeylist;
|
|
size_t len;
|
|
int ilen, result, ret;
|
|
u_int32_t i, nskeys;
|
|
void *retbuf, *databuf;
|
|
|
|
ip = (DBTCL_INFO *)dbp->api_internal;
|
|
interp = ip->i_interp;
|
|
objv[0] = ip->i_second_call;
|
|
|
|
/*
|
|
* Create two ByteArray objects, with the contents of the pkey
|
|
* and data DBTs that are our inputs.
|
|
*/
|
|
pobj = Tcl_NewByteArrayObj(pkey->data, (int)pkey->size);
|
|
Tcl_IncrRefCount(pobj);
|
|
dobj = Tcl_NewByteArrayObj(data->data, (int)data->size);
|
|
Tcl_IncrRefCount(dobj);
|
|
|
|
objv[1] = pobj;
|
|
objv[2] = dobj;
|
|
|
|
result = Tcl_EvalObjv(interp, 3, objv, 0);
|
|
|
|
Tcl_DecrRefCount(pobj);
|
|
Tcl_DecrRefCount(dobj);
|
|
|
|
if (result != TCL_OK) {
|
|
__db_errx(dbp->env,
|
|
"Tcl callback function failed with code %d", result);
|
|
return (EINVAL);
|
|
}
|
|
|
|
robj = Tcl_GetObjResult(interp);
|
|
if (robj->typePtr == NULL || strcmp(robj->typePtr->name, "list") != 0) {
|
|
nskeys = 1;
|
|
skeylist = &robj;
|
|
tskey = skey;
|
|
} else {
|
|
if ((result = Tcl_ListObjGetElements(interp,
|
|
robj, &ilen, &skeylist)) != TCL_OK) {
|
|
__db_errx(dbp->env,
|
|
"Could not get list elements from Tcl callback");
|
|
return (EINVAL);
|
|
}
|
|
nskeys = (u_int32_t)ilen;
|
|
|
|
/*
|
|
* It would be nice to check for nskeys == 0 and return
|
|
* DB_DONOTINDEX, but Tcl does not distinguish between an empty
|
|
* string and an empty list, so that would disallow empty
|
|
* secondary keys.
|
|
*/
|
|
if (nskeys == 0) {
|
|
nskeys = 1;
|
|
skeylist = &robj;
|
|
}
|
|
if (nskeys == 1)
|
|
tskey = skey;
|
|
else {
|
|
memset(skey, 0, sizeof(DBT));
|
|
if ((ret = __os_umalloc(dbp->env,
|
|
nskeys * sizeof(DBT), &skey->data)) != 0)
|
|
return (ret);
|
|
skey->size = nskeys;
|
|
F_SET(skey, DB_DBT_MULTIPLE | DB_DBT_APPMALLOC);
|
|
tskey = (DBT *)skey->data;
|
|
}
|
|
}
|
|
|
|
for (i = 0; i < nskeys; i++, tskey++) {
|
|
retbuf = Tcl_GetByteArrayFromObj(skeylist[i], &ilen);
|
|
len = (size_t)ilen;
|
|
|
|
/*
|
|
* retbuf is owned by Tcl; copy it into malloc'ed memory.
|
|
* We need to use __os_umalloc rather than ufree because this
|
|
* will be freed by DB using __os_ufree--the DB_DBT_APPMALLOC
|
|
* flag tells DB to free application-allocated memory.
|
|
*/
|
|
if ((ret = __os_umalloc(dbp->env, len, &databuf)) != 0)
|
|
return (ret);
|
|
memcpy(databuf, retbuf, len);
|
|
|
|
memset(tskey, 0, sizeof(DBT));
|
|
tskey->data = databuf;
|
|
tskey->size = (u_int32_t)len;
|
|
F_SET(tskey, DB_DBT_APPMALLOC);
|
|
}
|
|
|
|
return (0);
|
|
}
|
|
|
|
/*
|
|
* tcl_foreign_call --
|
|
* Foreign callback function for secondary indices. Get the callback
|
|
* out of ip->i_foreign_call and call it.
|
|
*/
|
|
static int tcl_foreign_call(sdbp, pkey, data, fkey, changed)
|
|
DB *sdbp;
|
|
const DBT *pkey, *fkey;
|
|
DBT *data;
|
|
int *changed;
|
|
{
|
|
DBTCL_INFO *ip;
|
|
Tcl_Interp *interp;
|
|
Tcl_Obj *pobj, *dobj, *fobj, *objv[4], *robj;
|
|
size_t len, orig_len;
|
|
int ilen, result, ret;
|
|
void *retbuf;
|
|
|
|
ip = (DBTCL_INFO *)sdbp->api_internal;
|
|
interp = ip->i_interp;
|
|
objv[0] = ip->i_foreign_call;
|
|
|
|
/*
|
|
* Create two ByteArray objects, with the contents of the pkey
|
|
* and data DBTs that are our inputs.
|
|
*/
|
|
pobj = Tcl_NewByteArrayObj(pkey->data, (int)pkey->size);
|
|
Tcl_IncrRefCount(pobj);
|
|
dobj = Tcl_NewByteArrayObj(data->data, (int)data->size);
|
|
Tcl_IncrRefCount(dobj);
|
|
fobj = Tcl_NewByteArrayObj(fkey->data, (int)fkey->size);
|
|
Tcl_IncrRefCount(fobj);
|
|
|
|
objv[1] = pobj;
|
|
objv[2] = dobj;
|
|
objv[3] = fobj;
|
|
|
|
result = Tcl_EvalObjv(interp, 4, objv, 0);
|
|
|
|
Tcl_DecrRefCount(pobj);
|
|
Tcl_DecrRefCount(dobj);
|
|
Tcl_DecrRefCount(fobj);
|
|
|
|
if (result != TCL_OK) {
|
|
__db_errx(sdbp->env,
|
|
"Tcl foreign callback function failed with code %d",
|
|
result);
|
|
return (EINVAL);
|
|
}
|
|
|
|
robj = Tcl_GetObjResult(interp);
|
|
retbuf = Tcl_GetByteArrayFromObj(robj, &ilen);
|
|
len = (size_t)ilen;
|
|
orig_len = (size_t)data->size;
|
|
if((len == orig_len) && (memcmp(retbuf, data->data, len) == 0)) {
|
|
*changed = 0;
|
|
return 0;
|
|
} else {
|
|
*changed = 1;
|
|
if(orig_len >= len) {
|
|
memcpy(data->data, retbuf, len);
|
|
data->size = (u_int32_t)len;
|
|
} else {
|
|
if((ret = __os_malloc(
|
|
sdbp->env, len, &data->data)) != 0)
|
|
return ret;
|
|
memcpy(data->data, retbuf, len);
|
|
data->size = (u_int32_t)len;
|
|
F_SET(data, DB_DBT_APPMALLOC);
|
|
}
|
|
}
|
|
|
|
return (0);
|
|
}
|
|
|
|
/*
|
|
* tcl_db_join --
|
|
*/
|
|
static int
|
|
tcl_DbJoin(interp, objc, objv, dbp, dbcp)
|
|
Tcl_Interp *interp; /* Interpreter */
|
|
int objc; /* How many arguments? */
|
|
Tcl_Obj *CONST objv[]; /* The argument objects */
|
|
DB *dbp; /* Database pointer */
|
|
DBC **dbcp; /* Cursor pointer */
|
|
{
|
|
static const char *dbjopts[] = {
|
|
"-nosort",
|
|
NULL
|
|
};
|
|
enum dbjopts {
|
|
DBJ_NOSORT
|
|
};
|
|
DBC **listp;
|
|
size_t size;
|
|
u_int32_t flag;
|
|
int adj, i, j, optindex, result, ret;
|
|
char *arg, msg[MSG_SIZE];
|
|
|
|
result = TCL_OK;
|
|
flag = 0;
|
|
if (objc < 3) {
|
|
Tcl_WrongNumArgs(interp, 2, objv, "curs1 curs2 ...");
|
|
return (TCL_ERROR);
|
|
}
|
|
|
|
for (adj = i = 2; i < objc; i++) {
|
|
if (Tcl_GetIndexFromObj(interp, objv[i], dbjopts, "option",
|
|
TCL_EXACT, &optindex) != TCL_OK) {
|
|
result = IS_HELP(objv[i]);
|
|
if (result == TCL_OK)
|
|
return (result);
|
|
result = TCL_OK;
|
|
Tcl_ResetResult(interp);
|
|
break;
|
|
}
|
|
switch ((enum dbjopts)optindex) {
|
|
case DBJ_NOSORT:
|
|
flag |= DB_JOIN_NOSORT;
|
|
adj++;
|
|
break;
|
|
}
|
|
}
|
|
if (result != TCL_OK)
|
|
return (result);
|
|
/*
|
|
* Allocate one more for NULL ptr at end of list.
|
|
*/
|
|
size = sizeof(DBC *) * (size_t)((objc - adj) + 1);
|
|
ret = __os_malloc(dbp->env, size, &listp);
|
|
if (ret != 0) {
|
|
Tcl_SetResult(interp, db_strerror(ret), TCL_STATIC);
|
|
return (TCL_ERROR);
|
|
}
|
|
|
|
memset(listp, 0, size);
|
|
for (j = 0, i = adj; i < objc; i++, j++) {
|
|
arg = Tcl_GetStringFromObj(objv[i], NULL);
|
|
listp[j] = NAME_TO_DBC(arg);
|
|
if (listp[j] == NULL) {
|
|
snprintf(msg, MSG_SIZE,
|
|
"Join: Invalid cursor: %s\n", arg);
|
|
Tcl_SetResult(interp, msg, TCL_VOLATILE);
|
|
result = TCL_ERROR;
|
|
goto out;
|
|
}
|
|
}
|
|
listp[j] = NULL;
|
|
_debug_check();
|
|
ret = dbp->join(dbp, listp, dbcp, flag);
|
|
result = _ReturnSetup(interp, ret, DB_RETOK_STD(ret), "db join");
|
|
|
|
out:
|
|
__os_free(dbp->env, listp);
|
|
return (result);
|
|
}
|
|
|
|
/*
|
|
* tcl_db_getjoin --
|
|
*/
|
|
static int
|
|
tcl_DbGetjoin(interp, objc, objv, dbp)
|
|
Tcl_Interp *interp; /* Interpreter */
|
|
int objc; /* How many arguments? */
|
|
Tcl_Obj *CONST objv[]; /* The argument objects */
|
|
DB *dbp; /* Database pointer */
|
|
{
|
|
static const char *dbgetjopts[] = {
|
|
#ifdef CONFIG_TEST
|
|
"-nosort",
|
|
#endif
|
|
"-txn",
|
|
NULL
|
|
};
|
|
enum dbgetjopts {
|
|
#ifdef CONFIG_TEST
|
|
DBGETJ_NOSORT,
|
|
#endif
|
|
DBGETJ_TXN
|
|
};
|
|
DB_TXN *txn;
|
|
DB *elemdbp;
|
|
DBC **listp;
|
|
DBC *dbc;
|
|
DBT key, data;
|
|
Tcl_Obj **elemv, *retlist;
|
|
void *ktmp;
|
|
size_t size;
|
|
u_int32_t flag;
|
|
int adj, elemc, freekey, i, j, optindex, result, ret;
|
|
char *arg, msg[MSG_SIZE];
|
|
|
|
result = TCL_OK;
|
|
flag = 0;
|
|
ktmp = NULL;
|
|
freekey = 0;
|
|
if (objc < 3) {
|
|
Tcl_WrongNumArgs(interp, 2, objv, "{db1 key1} {db2 key2} ...");
|
|
return (TCL_ERROR);
|
|
}
|
|
|
|
txn = NULL;
|
|
i = 2;
|
|
adj = i;
|
|
while (i < objc) {
|
|
if (Tcl_GetIndexFromObj(interp, objv[i], dbgetjopts, "option",
|
|
TCL_EXACT, &optindex) != TCL_OK) {
|
|
result = IS_HELP(objv[i]);
|
|
if (result == TCL_OK)
|
|
return (result);
|
|
result = TCL_OK;
|
|
Tcl_ResetResult(interp);
|
|
break;
|
|
}
|
|
i++;
|
|
switch ((enum dbgetjopts)optindex) {
|
|
#ifdef CONFIG_TEST
|
|
case DBGETJ_NOSORT:
|
|
flag |= DB_JOIN_NOSORT;
|
|
adj++;
|
|
break;
|
|
#endif
|
|
case DBGETJ_TXN:
|
|
if (i == objc) {
|
|
Tcl_WrongNumArgs(interp, 2, objv, "?-txn id?");
|
|
result = TCL_ERROR;
|
|
break;
|
|
}
|
|
arg = Tcl_GetStringFromObj(objv[i++], NULL);
|
|
txn = NAME_TO_TXN(arg);
|
|
adj += 2;
|
|
if (txn == NULL) {
|
|
snprintf(msg, MSG_SIZE,
|
|
"GetJoin: Invalid txn: %s\n", arg);
|
|
Tcl_SetResult(interp, msg, TCL_VOLATILE);
|
|
result = TCL_ERROR;
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
if (result != TCL_OK)
|
|
return (result);
|
|
size = sizeof(DBC *) * (size_t)((objc - adj) + 1);
|
|
ret = __os_malloc(NULL, size, &listp);
|
|
if (ret != 0) {
|
|
Tcl_SetResult(interp, db_strerror(ret), TCL_STATIC);
|
|
return (TCL_ERROR);
|
|
}
|
|
|
|
memset(listp, 0, size);
|
|
for (j = 0, i = adj; i < objc; i++, j++) {
|
|
/*
|
|
* Get each sublist as {db key}
|
|
*/
|
|
result = Tcl_ListObjGetElements(interp, objv[i],
|
|
&elemc, &elemv);
|
|
if (elemc != 2) {
|
|
Tcl_SetResult(interp, "Lists must be {db key}",
|
|
TCL_STATIC);
|
|
result = TCL_ERROR;
|
|
goto out;
|
|
}
|
|
/*
|
|
* Get a pointer to that open db. Then, open a cursor in
|
|
* that db, and go to the "key" place.
|
|
*/
|
|
elemdbp = NAME_TO_DB(Tcl_GetStringFromObj(elemv[0], NULL));
|
|
if (elemdbp == NULL) {
|
|
snprintf(msg, MSG_SIZE, "Get_join: Invalid db: %s\n",
|
|
Tcl_GetStringFromObj(elemv[0], NULL));
|
|
Tcl_SetResult(interp, msg, TCL_VOLATILE);
|
|
result = TCL_ERROR;
|
|
goto out;
|
|
}
|
|
ret = elemdbp->cursor(elemdbp, txn, &listp[j], 0);
|
|
if ((result = _ReturnSetup(interp, ret, DB_RETOK_STD(ret),
|
|
"db cursor")) == TCL_ERROR)
|
|
goto out;
|
|
memset(&key, 0, sizeof(key));
|
|
memset(&data, 0, sizeof(data));
|
|
ret = _CopyObjBytes(interp, elemv[elemc-1], &ktmp,
|
|
&key.size, &freekey);
|
|
if (ret != 0) {
|
|
result = _ReturnSetup(interp, ret,
|
|
DB_RETOK_STD(ret), "db join");
|
|
goto out;
|
|
}
|
|
key.data = ktmp;
|
|
ret = (listp[j])->get(listp[j], &key, &data, DB_SET);
|
|
if ((result = _ReturnSetup(interp, ret, DB_RETOK_DBCGET(ret),
|
|
"db cget")) == TCL_ERROR)
|
|
goto out;
|
|
}
|
|
listp[j] = NULL;
|
|
_debug_check();
|
|
ret = dbp->join(dbp, listp, &dbc, flag);
|
|
result = _ReturnSetup(interp, ret, DB_RETOK_STD(ret), "db join");
|
|
if (result == TCL_ERROR)
|
|
goto out;
|
|
|
|
retlist = Tcl_NewListObj(0, NULL);
|
|
while (ret == 0 && result == TCL_OK) {
|
|
memset(&key, 0, sizeof(key));
|
|
memset(&data, 0, sizeof(data));
|
|
key.flags |= DB_DBT_MALLOC;
|
|
data.flags |= DB_DBT_MALLOC;
|
|
ret = dbc->get(dbc, &key, &data, 0);
|
|
/*
|
|
* Build up our {name value} sublist
|
|
*/
|
|
if (ret == 0) {
|
|
result = _SetListElem(interp, retlist,
|
|
key.data, key.size,
|
|
data.data, data.size);
|
|
__os_ufree(dbp->env, key.data);
|
|
__os_ufree(dbp->env, data.data);
|
|
}
|
|
}
|
|
(void)dbc->close(dbc);
|
|
if (result == TCL_OK)
|
|
Tcl_SetObjResult(interp, retlist);
|
|
out:
|
|
if (ktmp != NULL && freekey)
|
|
__os_free(dbp->env, ktmp);
|
|
while (j) {
|
|
if (listp[j])
|
|
(void)(listp[j])->close(listp[j]);
|
|
j--;
|
|
}
|
|
__os_free(dbp->env, listp);
|
|
return (result);
|
|
}
|
|
|
|
/*
|
|
* tcl_DbGetFlags --
|
|
*/
|
|
static int
|
|
tcl_DbGetFlags(interp, objc, objv, dbp)
|
|
Tcl_Interp *interp; /* Interpreter */
|
|
int objc; /* How many arguments? */
|
|
Tcl_Obj *CONST objv[]; /* The argument objects */
|
|
DB *dbp; /* Database pointer */
|
|
{
|
|
int i, ret, result;
|
|
u_int32_t flags;
|
|
char buf[512];
|
|
Tcl_Obj *res;
|
|
|
|
static const struct {
|
|
u_int32_t flag;
|
|
char *arg;
|
|
} db_flags[] = {
|
|
{ DB_CHKSUM, "-chksum" },
|
|
{ DB_DUP, "-dup" },
|
|
{ DB_DUPSORT, "-dupsort" },
|
|
{ DB_ENCRYPT, "-encrypt" },
|
|
{ DB_INORDER, "-inorder" },
|
|
{ DB_TXN_NOT_DURABLE, "-notdurable" },
|
|
{ DB_RECNUM, "-recnum" },
|
|
{ DB_RENUMBER, "-renumber" },
|
|
{ DB_REVSPLITOFF, "-revsplitoff" },
|
|
{ DB_SNAPSHOT, "-snapshot" },
|
|
{ 0, NULL }
|
|
};
|
|
|
|
if (objc != 2) {
|
|
Tcl_WrongNumArgs(interp, 1, objv, NULL);
|
|
return (TCL_ERROR);
|
|
}
|
|
|
|
ret = dbp->get_flags(dbp, &flags);
|
|
if ((result = _ReturnSetup(
|
|
interp, ret, DB_RETOK_STD(ret), "db get_flags")) == TCL_OK) {
|
|
buf[0] = '\0';
|
|
|
|
for (i = 0; db_flags[i].flag != 0; i++)
|
|
if (LF_ISSET(db_flags[i].flag)) {
|
|
if (strlen(buf) > 0)
|
|
(void)strncat(buf, " ", sizeof(buf));
|
|
(void)strncat(
|
|
buf, db_flags[i].arg, sizeof(buf));
|
|
}
|
|
|
|
res = NewStringObj(buf, strlen(buf));
|
|
Tcl_SetObjResult(interp, res);
|
|
}
|
|
|
|
return (result);
|
|
}
|
|
|
|
/*
|
|
* tcl_DbGetOpenFlags --
|
|
*/
|
|
static int
|
|
tcl_DbGetOpenFlags(interp, objc, objv, dbp)
|
|
Tcl_Interp *interp; /* Interpreter */
|
|
int objc; /* How many arguments? */
|
|
Tcl_Obj *CONST objv[]; /* The argument objects */
|
|
DB *dbp; /* Database pointer */
|
|
{
|
|
int i, ret, result;
|
|
u_int32_t flags;
|
|
char buf[512];
|
|
Tcl_Obj *res;
|
|
|
|
static const struct {
|
|
u_int32_t flag;
|
|
char *arg;
|
|
} open_flags[] = {
|
|
{ DB_AUTO_COMMIT, "-auto_commit" },
|
|
{ DB_CREATE, "-create" },
|
|
{ DB_EXCL, "-excl" },
|
|
{ DB_MULTIVERSION, "-multiversion" },
|
|
{ DB_NOMMAP, "-nommap" },
|
|
{ DB_RDONLY, "-rdonly" },
|
|
{ DB_READ_UNCOMMITTED, "-read_uncommitted" },
|
|
{ DB_THREAD, "-thread" },
|
|
{ DB_TRUNCATE, "-truncate" },
|
|
{ 0, NULL }
|
|
};
|
|
|
|
if (objc != 2) {
|
|
Tcl_WrongNumArgs(interp, 1, objv, NULL);
|
|
return (TCL_ERROR);
|
|
}
|
|
|
|
ret = dbp->get_open_flags(dbp, &flags);
|
|
if ((result = _ReturnSetup(
|
|
interp, ret, DB_RETOK_STD(ret), "db get_open_flags")) == TCL_OK) {
|
|
buf[0] = '\0';
|
|
|
|
for (i = 0; open_flags[i].flag != 0; i++)
|
|
if (LF_ISSET(open_flags[i].flag)) {
|
|
if (strlen(buf) > 0)
|
|
(void)strncat(buf, " ", sizeof(buf));
|
|
(void)strncat(
|
|
buf, open_flags[i].arg, sizeof(buf));
|
|
}
|
|
|
|
res = NewStringObj(buf, strlen(buf));
|
|
Tcl_SetObjResult(interp, res);
|
|
}
|
|
|
|
return (result);
|
|
}
|
|
|
|
/*
|
|
* tcl_DbCount --
|
|
*/
|
|
static int
|
|
tcl_DbCount(interp, objc, objv, dbp)
|
|
Tcl_Interp *interp; /* Interpreter */
|
|
int objc; /* How many arguments? */
|
|
Tcl_Obj *CONST objv[]; /* The argument objects */
|
|
DB *dbp; /* Database pointer */
|
|
{
|
|
DB *recdbp;
|
|
DBC *dbc;
|
|
DBT key, data, hkey;
|
|
Tcl_Obj *res;
|
|
void *ktmp;
|
|
DB_HEAP_RID rid;
|
|
db_recno_t count, recno;
|
|
int freekey, result, ret;
|
|
|
|
res = NULL;
|
|
count = 0;
|
|
freekey = ret = 0;
|
|
ktmp = NULL;
|
|
result = TCL_OK;
|
|
dbc = NULL;
|
|
|
|
if (objc != 3) {
|
|
Tcl_WrongNumArgs(interp, 2, objv, "key");
|
|
return (TCL_ERROR);
|
|
}
|
|
|
|
/*
|
|
* Get the count for our key.
|
|
* We do this by getting a cursor for this DB. Moving the cursor
|
|
* to the set location, and getting a count on that cursor.
|
|
*/
|
|
memset(&key, 0, sizeof(key));
|
|
memset(&data, 0, sizeof(data));
|
|
memset(&hkey, 0, sizeof(hkey));
|
|
|
|
/*
|
|
* If it's a heap, queue or recno database, we must make sure to
|
|
* treat the key as a recno rather than as a byte string.
|
|
*/
|
|
if (dbp->type == DB_HEAP ||
|
|
dbp->type == DB_RECNO || dbp->type == DB_QUEUE) {
|
|
result = _GetUInt32(interp, objv[2], &recno);
|
|
if (result == TCL_OK) {
|
|
if (dbp->type == DB_HEAP) {
|
|
hkey.data = &recno;
|
|
hkey.size = sizeof(db_recno_t);
|
|
|
|
key.data = &rid;
|
|
key.ulen = key.size = sizeof(DB_HEAP_RID);
|
|
key.flags = DB_DBT_USERMEM;
|
|
} else {
|
|
key.data = &recno;
|
|
key.size = sizeof(db_recno_t);
|
|
}
|
|
} else
|
|
return (result);
|
|
} else {
|
|
ret = _CopyObjBytes(interp, objv[2], &ktmp,
|
|
&key.size, &freekey);
|
|
if (ret != 0) {
|
|
result = _ReturnSetup(interp, ret,
|
|
DB_RETOK_STD(ret), "db count");
|
|
return (result);
|
|
}
|
|
key.data = ktmp;
|
|
}
|
|
_debug_check();
|
|
|
|
/* If it's a heap, translate recno to rid. */
|
|
if (dbp->type == DB_HEAP) {
|
|
recdbp = ((DBTCL_INFO *)dbp->api_internal)->hrdbp;
|
|
|
|
ret = recdbp->get(recdbp, NULL, &hkey, &key, 0);
|
|
if (ret == DB_NOTFOUND || ret == DB_KEYEMPTY) {
|
|
count = 0;
|
|
goto out1;
|
|
} else if (ret != 0) {
|
|
result = _ReturnSetup(interp, ret,
|
|
DB_RETOK_DBGET(ret), "db get heap");
|
|
return (result);
|
|
}
|
|
}
|
|
|
|
ret = dbp->cursor(dbp, NULL, &dbc, 0);
|
|
if (ret != 0) {
|
|
result = _ReturnSetup(interp, ret, DB_RETOK_STD(ret),
|
|
"db cursor");
|
|
goto out;
|
|
}
|
|
/*
|
|
* Move our cursor to the key.
|
|
*/
|
|
ret = dbc->get(dbc, &key, &data, DB_SET);
|
|
if (ret == DB_KEYEMPTY || ret == DB_NOTFOUND)
|
|
count = 0;
|
|
else {
|
|
ret = dbc->count(dbc, &count, 0);
|
|
if (ret != 0) {
|
|
result = _ReturnSetup(interp, ret, DB_RETOK_STD(ret),
|
|
"db c count");
|
|
goto out;
|
|
}
|
|
}
|
|
out1: res = Tcl_NewWideIntObj((Tcl_WideInt)count);
|
|
Tcl_SetObjResult(interp, res);
|
|
|
|
out: if (ktmp != NULL && freekey)
|
|
__os_free(dbp->env, ktmp);
|
|
if (dbc != NULL)
|
|
(void)dbc->close(dbc);
|
|
return (result);
|
|
}
|
|
|
|
#ifdef CONFIG_TEST
|
|
/*
|
|
* tcl_DbKeyRange --
|
|
*/
|
|
static int
|
|
tcl_DbKeyRange(interp, objc, objv, dbp)
|
|
Tcl_Interp *interp; /* Interpreter */
|
|
int objc; /* How many arguments? */
|
|
Tcl_Obj *CONST objv[]; /* The argument objects */
|
|
DB *dbp; /* Database pointer */
|
|
{
|
|
static const char *dbkeyropts[] = {
|
|
"-txn",
|
|
NULL
|
|
};
|
|
enum dbkeyropts {
|
|
DBKEYR_TXN
|
|
};
|
|
DB_TXN *txn;
|
|
DB_KEY_RANGE range;
|
|
DBT key;
|
|
DBTYPE type;
|
|
Tcl_Obj *myobjv[3], *retlist;
|
|
void *ktmp;
|
|
db_recno_t recno;
|
|
u_int32_t flag;
|
|
int freekey, i, myobjc, optindex, result, ret;
|
|
char *arg, msg[MSG_SIZE];
|
|
|
|
ktmp = NULL;
|
|
flag = 0;
|
|
freekey = 0;
|
|
result = TCL_OK;
|
|
if (objc < 3) {
|
|
Tcl_WrongNumArgs(interp, 2, objv, "?-txn id? key");
|
|
return (TCL_ERROR);
|
|
}
|
|
|
|
txn = NULL;
|
|
for (i = 2; i < objc;) {
|
|
if (Tcl_GetIndexFromObj(interp, objv[i], dbkeyropts, "option",
|
|
TCL_EXACT, &optindex) != TCL_OK) {
|
|
result = IS_HELP(objv[i]);
|
|
if (result == TCL_OK)
|
|
return (result);
|
|
result = TCL_OK;
|
|
Tcl_ResetResult(interp);
|
|
break;
|
|
}
|
|
i++;
|
|
switch ((enum dbkeyropts)optindex) {
|
|
case DBKEYR_TXN:
|
|
if (i == objc) {
|
|
Tcl_WrongNumArgs(interp, 2, objv, "?-txn id?");
|
|
result = TCL_ERROR;
|
|
break;
|
|
}
|
|
arg = Tcl_GetStringFromObj(objv[i++], NULL);
|
|
txn = NAME_TO_TXN(arg);
|
|
if (txn == NULL) {
|
|
snprintf(msg, MSG_SIZE,
|
|
"KeyRange: Invalid txn: %s\n", arg);
|
|
Tcl_SetResult(interp, msg, TCL_VOLATILE);
|
|
result = TCL_ERROR;
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
if (result != TCL_OK)
|
|
return (result);
|
|
(void)dbp->get_type(dbp, &type);
|
|
ret = 0;
|
|
/*
|
|
* Make sure we have a key.
|
|
*/
|
|
if (i != (objc - 1)) {
|
|
Tcl_WrongNumArgs(interp, 2, objv, "?args? key");
|
|
result = TCL_ERROR;
|
|
goto out;
|
|
}
|
|
memset(&key, 0, sizeof(key));
|
|
if (type == DB_RECNO || type == DB_QUEUE) {
|
|
result = _GetUInt32(interp, objv[i], &recno);
|
|
if (result == TCL_OK) {
|
|
key.data = &recno;
|
|
key.size = sizeof(db_recno_t);
|
|
} else
|
|
return (result);
|
|
} else {
|
|
ret = _CopyObjBytes(interp, objv[i++], &ktmp,
|
|
&key.size, &freekey);
|
|
if (ret != 0) {
|
|
result = _ReturnSetup(interp, ret,
|
|
DB_RETOK_STD(ret), "db keyrange");
|
|
return (result);
|
|
}
|
|
key.data = ktmp;
|
|
}
|
|
_debug_check();
|
|
ret = dbp->key_range(dbp, txn, &key, &range, flag);
|
|
result = _ReturnSetup(interp, ret, DB_RETOK_STD(ret), "db keyrange");
|
|
if (result == TCL_ERROR)
|
|
goto out;
|
|
|
|
/*
|
|
* If we succeeded, set up return list.
|
|
*/
|
|
myobjc = 3;
|
|
myobjv[0] = Tcl_NewDoubleObj(range.less);
|
|
myobjv[1] = Tcl_NewDoubleObj(range.equal);
|
|
myobjv[2] = Tcl_NewDoubleObj(range.greater);
|
|
retlist = Tcl_NewListObj(myobjc, myobjv);
|
|
if (result == TCL_OK)
|
|
Tcl_SetObjResult(interp, retlist);
|
|
|
|
out: if (ktmp != NULL && freekey)
|
|
__os_free(dbp->env, ktmp);
|
|
return (result);
|
|
}
|
|
#endif
|
|
|
|
/*
|
|
* tcl_DbTruncate --
|
|
*/
|
|
static int
|
|
tcl_DbTruncate(interp, objc, objv, dbp)
|
|
Tcl_Interp *interp; /* Interpreter */
|
|
int objc; /* How many arguments? */
|
|
Tcl_Obj *CONST objv[]; /* The argument objects */
|
|
DB *dbp; /* Database pointer */
|
|
{
|
|
static const char *dbcuropts[] = {
|
|
"-txn",
|
|
NULL
|
|
};
|
|
enum dbcuropts {
|
|
DBTRUNC_TXN
|
|
};
|
|
DB *recdbp;
|
|
DB_TXN *txn;
|
|
Tcl_Obj *res;
|
|
u_int32_t count;
|
|
int i, optindex, result, ret;
|
|
char *arg, msg[MSG_SIZE];
|
|
|
|
txn = NULL;
|
|
result = TCL_OK;
|
|
|
|
i = 2;
|
|
while (i < objc) {
|
|
if (Tcl_GetIndexFromObj(interp, objv[i], dbcuropts, "option",
|
|
TCL_EXACT, &optindex) != TCL_OK) {
|
|
result = IS_HELP(objv[i]);
|
|
goto out;
|
|
}
|
|
i++;
|
|
switch ((enum dbcuropts)optindex) {
|
|
case DBTRUNC_TXN:
|
|
if (i == objc) {
|
|
Tcl_WrongNumArgs(interp, 2, objv, "?-txn id?");
|
|
result = TCL_ERROR;
|
|
break;
|
|
}
|
|
arg = Tcl_GetStringFromObj(objv[i++], NULL);
|
|
txn = NAME_TO_TXN(arg);
|
|
if (txn == NULL) {
|
|
snprintf(msg, MSG_SIZE,
|
|
"Truncate: Invalid txn: %s\n", arg);
|
|
Tcl_SetResult(interp, msg, TCL_VOLATILE);
|
|
result = TCL_ERROR;
|
|
}
|
|
break;
|
|
}
|
|
if (result != TCL_OK)
|
|
break;
|
|
}
|
|
if (result != TCL_OK)
|
|
goto out;
|
|
|
|
_debug_check();
|
|
ret = dbp->truncate(dbp, txn, &count, 0);
|
|
if (dbp->type == DB_HEAP && ret == 0) {
|
|
/* The truncation of hrdbp will automatically truncate hsdbp. */
|
|
recdbp = ((DBTCL_INFO *)dbp->api_internal)->hrdbp;
|
|
ret = recdbp->truncate(recdbp, txn, NULL, 0);
|
|
}
|
|
if (ret != 0)
|
|
result = _ErrorSetup(interp, ret, "db truncate");
|
|
|
|
else {
|
|
res = Tcl_NewWideIntObj((Tcl_WideInt)count);
|
|
Tcl_SetObjResult(interp, res);
|
|
}
|
|
out:
|
|
return (result);
|
|
}
|
|
|
|
#ifdef CONFIG_TEST
|
|
/*
|
|
* tcl_DbCompact --
|
|
*/
|
|
static int
|
|
tcl_DbCompact(interp, objc, objv, dbp)
|
|
Tcl_Interp *interp; /* Interpreter */
|
|
int objc; /* How many arguments? */
|
|
Tcl_Obj *CONST objv[]; /* The argument objects */
|
|
DB *dbp; /* Database pointer */
|
|
{
|
|
static const char *dbcuropts[] = {
|
|
"-fillpercent",
|
|
"-freespace",
|
|
"-freeonly",
|
|
"-pages",
|
|
"-start",
|
|
"-stop",
|
|
"-timeout",
|
|
"-txn",
|
|
NULL
|
|
};
|
|
enum dbcuropts {
|
|
DBREORG_FILLFACTOR,
|
|
DBREORG_FREESPACE,
|
|
DBREORG_FREEONLY,
|
|
DBREORG_PAGES,
|
|
DBREORG_START,
|
|
DBREORG_STOP,
|
|
DBREORG_TIMEOUT,
|
|
DBREORG_TXN
|
|
};
|
|
DBTCL_INFO *ip;
|
|
DBT *key, end, start, stop;
|
|
DBTYPE type;
|
|
DB_TXN *txn;
|
|
Tcl_Obj *myobj, *retlist;
|
|
db_recno_t recno, srecno;
|
|
u_int32_t arg, fillfactor, flags, pages, timeout;
|
|
char *carg, msg[MSG_SIZE];
|
|
int freekey, i, optindex, result, ret;
|
|
void *kp;
|
|
|
|
flags = 0;
|
|
result = TCL_OK;
|
|
txn = NULL;
|
|
(void)dbp->get_type(dbp, &type);
|
|
memset(&start, 0, sizeof(start));
|
|
memset(&stop, 0, sizeof(stop));
|
|
memset(&end, 0, sizeof(end));
|
|
ip = (DBTCL_INFO *)dbp->api_internal;
|
|
fillfactor = pages = timeout = 0;
|
|
|
|
i = 2;
|
|
while (i < objc) {
|
|
if (Tcl_GetIndexFromObj(interp, objv[i], dbcuropts, "option",
|
|
TCL_EXACT, &optindex) != TCL_OK) {
|
|
result = IS_HELP(objv[i]);
|
|
goto out;
|
|
}
|
|
i++;
|
|
switch ((enum dbcuropts)optindex) {
|
|
case DBREORG_FILLFACTOR:
|
|
if (i == objc) {
|
|
Tcl_WrongNumArgs(interp,
|
|
2, objv, "?-fillfactor number?");
|
|
result = TCL_ERROR;
|
|
break;
|
|
}
|
|
result = _GetUInt32(interp, objv[i++], &arg);
|
|
if (result != TCL_OK)
|
|
goto out;
|
|
i++;
|
|
fillfactor = arg;
|
|
break;
|
|
case DBREORG_FREESPACE:
|
|
LF_SET(DB_FREE_SPACE);
|
|
break;
|
|
|
|
case DBREORG_FREEONLY:
|
|
LF_SET(DB_FREELIST_ONLY);
|
|
break;
|
|
|
|
case DBREORG_PAGES:
|
|
if (i == objc) {
|
|
Tcl_WrongNumArgs(interp,
|
|
2, objv, "?-pages number?");
|
|
result = TCL_ERROR;
|
|
break;
|
|
}
|
|
result = _GetUInt32(interp, objv[i++], &arg);
|
|
if (result != TCL_OK)
|
|
goto out;
|
|
i++;
|
|
pages = arg;
|
|
break;
|
|
case DBREORG_TIMEOUT:
|
|
if (i == objc) {
|
|
Tcl_WrongNumArgs(interp,
|
|
2, objv, "?-timeout number?");
|
|
result = TCL_ERROR;
|
|
break;
|
|
}
|
|
result = _GetUInt32(interp, objv[i++], &arg);
|
|
if (result != TCL_OK)
|
|
goto out;
|
|
i++;
|
|
timeout = arg;
|
|
break;
|
|
|
|
case DBREORG_START:
|
|
case DBREORG_STOP:
|
|
if (i == objc) {
|
|
Tcl_WrongNumArgs(interp, 1, objv,
|
|
"?-args? -start/stop key");
|
|
result = TCL_ERROR;
|
|
goto out;
|
|
}
|
|
if ((enum dbcuropts)optindex == DBREORG_START) {
|
|
key = &start;
|
|
key->data = &recno;
|
|
} else {
|
|
key = &stop;
|
|
key->data = &srecno;
|
|
}
|
|
if (type == DB_RECNO || type == DB_QUEUE) {
|
|
result = _GetUInt32(
|
|
interp, objv[i], key->data);
|
|
if (result == TCL_OK) {
|
|
key->size = sizeof(db_recno_t);
|
|
} else
|
|
goto out;
|
|
} else {
|
|
ret = _CopyObjBytes(interp, objv[i],
|
|
&key->data, &key->size, &freekey);
|
|
if (ret != 0)
|
|
goto err;
|
|
if (freekey == 0) {
|
|
if ((ret = __os_malloc(NULL,
|
|
key->size, &kp)) != 0)
|
|
goto err;
|
|
|
|
memcpy(kp, key->data, key->size);
|
|
key->data = kp;
|
|
key->ulen = key->size;
|
|
}
|
|
}
|
|
i++;
|
|
break;
|
|
case DBREORG_TXN:
|
|
if (i == objc) {
|
|
Tcl_WrongNumArgs(interp, 2, objv, "?-txn id?");
|
|
result = TCL_ERROR;
|
|
break;
|
|
}
|
|
carg = Tcl_GetStringFromObj(objv[i++], NULL);
|
|
txn = NAME_TO_TXN(carg);
|
|
if (txn == NULL) {
|
|
snprintf(msg, MSG_SIZE,
|
|
"Compact: Invalid txn: %s\n", carg);
|
|
Tcl_SetResult(interp, msg, TCL_VOLATILE);
|
|
result = TCL_ERROR;
|
|
}
|
|
}
|
|
if (result != TCL_OK)
|
|
break;
|
|
}
|
|
if (result != TCL_OK)
|
|
goto out;
|
|
|
|
if (ip->i_cdata == NULL)
|
|
if ((ret = __os_calloc(dbp->env,
|
|
1, sizeof(DB_COMPACT), &ip->i_cdata)) != 0) {
|
|
Tcl_SetResult(interp,
|
|
db_strerror(ret), TCL_STATIC);
|
|
goto out;
|
|
}
|
|
|
|
ip->i_cdata->compact_fillpercent = fillfactor;
|
|
ip->i_cdata->compact_timeout = timeout;
|
|
ip->i_cdata->compact_pages = pages;
|
|
|
|
_debug_check();
|
|
ret = dbp->compact(dbp, txn, &start, &stop, ip->i_cdata, flags, &end);
|
|
result = _ReturnSetup(interp, ret, DB_RETOK_DBCGET(ret), "dbp compact");
|
|
if (result == TCL_ERROR)
|
|
goto out;
|
|
|
|
retlist = Tcl_NewListObj(0, NULL);
|
|
if (ret != 0)
|
|
goto out;
|
|
if (type == DB_RECNO || type == DB_QUEUE) {
|
|
if (end.size == 0)
|
|
recno = 0;
|
|
else
|
|
recno = *((db_recno_t *)end.data);
|
|
myobj = Tcl_NewWideIntObj((Tcl_WideInt)recno);
|
|
} else
|
|
myobj = Tcl_NewByteArrayObj(end.data, (int)end.size);
|
|
result = Tcl_ListObjAppendElement(interp, retlist, myobj);
|
|
if (result == TCL_OK)
|
|
Tcl_SetObjResult(interp, retlist);
|
|
|
|
if (0) {
|
|
err: result = _ReturnSetup(interp,
|
|
ret, DB_RETOK_DBCGET(ret), "dbc compact");
|
|
}
|
|
out:
|
|
if (start.data != NULL && start.data != &recno)
|
|
__os_free(NULL, start.data);
|
|
if (stop.data != NULL && stop.data != &srecno)
|
|
__os_free(NULL, stop.data);
|
|
if (end.data != NULL)
|
|
__os_free(NULL, end.data);
|
|
|
|
return (result);
|
|
}
|
|
|
|
/*
|
|
* tcl_DbCompactStat
|
|
*/
|
|
static int
|
|
tcl_DbCompactStat(interp, objc, objv, dbp)
|
|
Tcl_Interp *interp; /* Interpreter */
|
|
int objc; /* How many arguments? */
|
|
Tcl_Obj *CONST objv[]; /* The argument objects */
|
|
DB *dbp; /* Database pointer */
|
|
{
|
|
DBTCL_INFO *ip;
|
|
|
|
COMPQUIET(objc, 0);
|
|
COMPQUIET(objv, NULL);
|
|
|
|
ip = (DBTCL_INFO *)dbp->api_internal;
|
|
|
|
return (tcl_CompactStat(interp, ip));
|
|
}
|
|
|
|
/*
|
|
* PUBLIC: int tcl_CompactStat __P((Tcl_Interp *, DBTCL_INFO *));
|
|
*/
|
|
int
|
|
tcl_CompactStat(interp, ip)
|
|
Tcl_Interp *interp; /* Interpreter */
|
|
DBTCL_INFO *ip;
|
|
{
|
|
DB_COMPACT *rp;
|
|
Tcl_Obj *res;
|
|
int result;
|
|
char msg[MSG_SIZE];
|
|
|
|
result = TCL_OK;
|
|
rp = NULL;
|
|
|
|
_debug_check();
|
|
if ((rp = ip->i_cdata) == NULL) {
|
|
snprintf(msg, MSG_SIZE,
|
|
"Compact stat: No stats available\n");
|
|
Tcl_SetResult(interp, msg, TCL_VOLATILE);
|
|
result = TCL_ERROR;
|
|
goto error;
|
|
}
|
|
|
|
res = Tcl_NewObj();
|
|
|
|
MAKE_STAT_LIST("Pages freed", rp->compact_pages_free);
|
|
MAKE_STAT_LIST("Pages truncated", rp->compact_pages_truncated);
|
|
MAKE_STAT_LIST("Pages examined", rp->compact_pages_examine);
|
|
MAKE_STAT_LIST("Levels removed", rp->compact_levels);
|
|
MAKE_STAT_LIST("Deadlocks encountered", rp->compact_deadlock);
|
|
MAKE_STAT_LIST("Empty buckets", rp->compact_empty_buckets);
|
|
|
|
Tcl_SetObjResult(interp, res);
|
|
error:
|
|
return (result);
|
|
}
|
|
#endif
|