mirror of
https://github.com/berkeleydb/libdb.git
synced 2024-11-16 17:16:25 +00:00
1333 lines
32 KiB
C
1333 lines
32 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/tcl_db.h"
|
||
|
|
||
|
/*
|
||
|
* Prototypes for procedures defined later in this file:
|
||
|
*/
|
||
|
static int tcl_DbcDel __P((Tcl_Interp *, int, Tcl_Obj * CONST*, DBC *));
|
||
|
static int tcl_DbcDup __P((Tcl_Interp *, int, Tcl_Obj * CONST*, DBC *));
|
||
|
static int tcl_DbcCompare __P((Tcl_Interp *, int, Tcl_Obj * CONST*, DBC *));
|
||
|
static int tcl_DbcGet __P((Tcl_Interp *, int, Tcl_Obj * CONST*, DBC *, int));
|
||
|
static int tcl_DbcHeapDel __P((Tcl_Interp *, DBC *));
|
||
|
static int tcl_DbcPut __P((Tcl_Interp *, int, Tcl_Obj * CONST*, DBC *));
|
||
|
|
||
|
/*
|
||
|
* PUBLIC: int dbc_Cmd __P((ClientData, Tcl_Interp *, int, Tcl_Obj * CONST*));
|
||
|
*
|
||
|
* dbc_cmd --
|
||
|
* Implements the cursor command.
|
||
|
*/
|
||
|
int
|
||
|
dbc_Cmd(clientData, interp, objc, objv)
|
||
|
ClientData clientData; /* Cursor handle */
|
||
|
Tcl_Interp *interp; /* Interpreter */
|
||
|
int objc; /* How many arguments? */
|
||
|
Tcl_Obj *CONST objv[]; /* The argument objects */
|
||
|
{
|
||
|
static const char *dbccmds[] = {
|
||
|
#ifdef CONFIG_TEST
|
||
|
"pget",
|
||
|
#endif
|
||
|
"close",
|
||
|
"cmp",
|
||
|
"del",
|
||
|
"dup",
|
||
|
"get",
|
||
|
"put",
|
||
|
NULL
|
||
|
};
|
||
|
enum dbccmds {
|
||
|
#ifdef CONFIG_TEST
|
||
|
DBCPGET,
|
||
|
#endif
|
||
|
DBCCLOSE,
|
||
|
DBCCOMPARE,
|
||
|
DBCDELETE,
|
||
|
DBCDUP,
|
||
|
DBCGET,
|
||
|
DBCPUT
|
||
|
};
|
||
|
DBC *dbc;
|
||
|
DBTCL_INFO *dbip;
|
||
|
int cmdindex, result, ret;
|
||
|
|
||
|
Tcl_ResetResult(interp);
|
||
|
dbc = (DBC *)clientData;
|
||
|
dbip = _PtrToInfo((void *)dbc);
|
||
|
result = TCL_OK;
|
||
|
|
||
|
if (objc <= 1) {
|
||
|
Tcl_WrongNumArgs(interp, 1, objv, "command cmdargs");
|
||
|
return (TCL_ERROR);
|
||
|
}
|
||
|
if (dbc == NULL) {
|
||
|
Tcl_SetResult(interp, "NULL dbc pointer", TCL_STATIC);
|
||
|
return (TCL_ERROR);
|
||
|
}
|
||
|
if (dbip == NULL) {
|
||
|
Tcl_SetResult(interp, "NULL dbc info pointer", TCL_STATIC);
|
||
|
return (TCL_ERROR);
|
||
|
}
|
||
|
|
||
|
/*
|
||
|
* Get the command name index from the object based on the berkdbcmds
|
||
|
* defined above.
|
||
|
*/
|
||
|
if (Tcl_GetIndexFromObj(interp, objv[1], dbccmds, "command",
|
||
|
TCL_EXACT, &cmdindex) != TCL_OK)
|
||
|
return (IS_HELP(objv[1]));
|
||
|
switch ((enum dbccmds)cmdindex) {
|
||
|
#ifdef CONFIG_TEST
|
||
|
case DBCPGET:
|
||
|
result = tcl_DbcGet(interp, objc, objv, dbc, 1);
|
||
|
break;
|
||
|
#endif
|
||
|
case DBCCLOSE:
|
||
|
/*
|
||
|
* No args for this. Error if there are some.
|
||
|
*/
|
||
|
if (objc > 2) {
|
||
|
Tcl_WrongNumArgs(interp, 2, objv, NULL);
|
||
|
return (TCL_ERROR);
|
||
|
}
|
||
|
_debug_check();
|
||
|
ret = dbc->close(dbc);
|
||
|
result = _ReturnSetup(interp, ret, DB_RETOK_STD(ret),
|
||
|
"dbc close");
|
||
|
if (result == TCL_OK) {
|
||
|
(void)Tcl_DeleteCommand(interp, dbip->i_name);
|
||
|
_DeleteInfo(dbip);
|
||
|
}
|
||
|
break;
|
||
|
case DBCCOMPARE:
|
||
|
if (objc > 3) {
|
||
|
Tcl_WrongNumArgs(interp, 3, objv, NULL);
|
||
|
return (TCL_ERROR);
|
||
|
}
|
||
|
_debug_check();
|
||
|
result = tcl_DbcCompare(interp, objc, objv, dbc);
|
||
|
break;
|
||
|
case DBCDELETE:
|
||
|
result = tcl_DbcDel(interp, objc, objv, dbc);
|
||
|
break;
|
||
|
case DBCDUP:
|
||
|
result = tcl_DbcDup(interp, objc, objv, dbc);
|
||
|
break;
|
||
|
case DBCGET:
|
||
|
result = tcl_DbcGet(interp, objc, objv, dbc, 0);
|
||
|
break;
|
||
|
case DBCPUT:
|
||
|
result = tcl_DbcPut(interp, objc, objv, dbc);
|
||
|
break;
|
||
|
}
|
||
|
return (result);
|
||
|
}
|
||
|
|
||
|
/*
|
||
|
* tcl_DbcHeapDel --
|
||
|
*/
|
||
|
static int
|
||
|
tcl_DbcHeapDel(interp, dbc)
|
||
|
Tcl_Interp *interp;
|
||
|
DBC *dbc;
|
||
|
{
|
||
|
DB *dbp, *hrdbp, *hsdbp;
|
||
|
DBT hkey, key, tmpdata;
|
||
|
DB_HEAP_RID rid;
|
||
|
db_recno_t recno;
|
||
|
int result, ret, t_ret;
|
||
|
|
||
|
dbp = dbc->dbp;
|
||
|
hrdbp = ((DBTCL_INFO *)dbp->api_internal)->hrdbp;
|
||
|
hsdbp = ((DBTCL_INFO *)dbp->api_internal)->hsdbp;
|
||
|
|
||
|
memset(&hkey, 0, sizeof(DBT));
|
||
|
hkey.data = &rid;
|
||
|
hkey.size = hkey.ulen = sizeof(DB_HEAP_RID);
|
||
|
hkey.flags = DB_DBT_USERMEM;
|
||
|
memset(&tmpdata, 0, sizeof(DBT));
|
||
|
tmpdata.flags = DB_DBT_PARTIAL | DB_DBT_USERMEM;
|
||
|
if ((t_ret = dbc->get(dbc, &hkey, &tmpdata, DB_CURRENT)) != 0) {
|
||
|
ret = t_ret;
|
||
|
goto err;
|
||
|
}
|
||
|
|
||
|
memset(&key, 0, sizeof(DBT));
|
||
|
key.data = &recno;
|
||
|
key.size = key.ulen = sizeof(db_recno_t);
|
||
|
key.flags = DB_DBT_USERMEM;
|
||
|
if ((t_ret = hsdbp->pget(
|
||
|
hsdbp, dbc->txn, &hkey, &key, &tmpdata, 0)) != 0) {
|
||
|
ret = t_ret;
|
||
|
goto err;
|
||
|
}
|
||
|
|
||
|
ret = dbc->del(dbc, 0);
|
||
|
if ((t_ret = hrdbp->del(hrdbp, dbc->txn, &key, 0)) != 0 && ret == 0)
|
||
|
ret = t_ret;
|
||
|
|
||
|
err: result = _ReturnSetup(
|
||
|
interp, ret, DB_RETOK_DBCDEL(ret), "dbc delete");
|
||
|
return result;
|
||
|
}
|
||
|
|
||
|
/*
|
||
|
* tcl_DbcPut --
|
||
|
*/
|
||
|
static int
|
||
|
tcl_DbcPut(interp, objc, objv, dbc)
|
||
|
Tcl_Interp *interp; /* Interpreter */
|
||
|
int objc; /* How many arguments? */
|
||
|
Tcl_Obj *CONST objv[]; /* The argument objects */
|
||
|
DBC *dbc; /* Cursor pointer */
|
||
|
{
|
||
|
static const char *dbcutopts[] = {
|
||
|
#ifdef CONFIG_TEST
|
||
|
"-nodupdata",
|
||
|
#endif
|
||
|
"-after",
|
||
|
"-before",
|
||
|
"-current",
|
||
|
"-keyfirst",
|
||
|
"-keylast",
|
||
|
"-overwritedup",
|
||
|
"-partial",
|
||
|
NULL
|
||
|
};
|
||
|
enum dbcutopts {
|
||
|
#ifdef CONFIG_TEST
|
||
|
DBCPUT_NODUPDATA,
|
||
|
#endif
|
||
|
DBCPUT_AFTER,
|
||
|
DBCPUT_BEFORE,
|
||
|
DBCPUT_CURRENT,
|
||
|
DBCPUT_KEYFIRST,
|
||
|
DBCPUT_KEYLAST,
|
||
|
DBCPUT_OVERWRITE_DUP,
|
||
|
DBCPUT_PART
|
||
|
};
|
||
|
DB *thisdbp, *hrdbp, *hsdbp;
|
||
|
DBT data, hkey, key, tmpdata;
|
||
|
DBTCL_INFO *dbcip, *dbip;
|
||
|
DBTYPE type;
|
||
|
DB_HEAP_RID rid;
|
||
|
Tcl_Obj **elemv, *res;
|
||
|
void *dtmp, *ktmp;
|
||
|
db_recno_t recno;
|
||
|
u_int32_t flag;
|
||
|
int elemc, freekey, freedata, i, optindex, result, ret;
|
||
|
|
||
|
COMPQUIET(dtmp, NULL);
|
||
|
COMPQUIET(ktmp, NULL);
|
||
|
|
||
|
result = TCL_OK;
|
||
|
flag = 0;
|
||
|
freekey = freedata = 0;
|
||
|
|
||
|
if (objc < 2) {
|
||
|
Tcl_WrongNumArgs(interp, 2, objv, "?-args? ?key?");
|
||
|
return (TCL_ERROR);
|
||
|
}
|
||
|
|
||
|
memset(&key, 0, sizeof(key));
|
||
|
memset(&data, 0, sizeof(data));
|
||
|
memset(&hkey, 0, sizeof(hkey));
|
||
|
|
||
|
/*
|
||
|
* Get the command name index from the object based on the options
|
||
|
* defined above.
|
||
|
*/
|
||
|
i = 2;
|
||
|
while (i < (objc - 1)) {
|
||
|
if (Tcl_GetIndexFromObj(interp, objv[i], dbcutopts, "option",
|
||
|
TCL_EXACT, &optindex) != TCL_OK) {
|
||
|
/*
|
||
|
* Reset the result so we don't get
|
||
|
* an errant error message if there is another error.
|
||
|
*/
|
||
|
if (IS_HELP(objv[i]) == TCL_OK) {
|
||
|
result = TCL_OK;
|
||
|
goto out;
|
||
|
}
|
||
|
Tcl_ResetResult(interp);
|
||
|
break;
|
||
|
}
|
||
|
i++;
|
||
|
switch ((enum dbcutopts)optindex) {
|
||
|
#ifdef CONFIG_TEST
|
||
|
case DBCPUT_NODUPDATA:
|
||
|
FLAG_CHECK(flag);
|
||
|
flag = DB_NODUPDATA;
|
||
|
break;
|
||
|
#endif
|
||
|
case DBCPUT_AFTER:
|
||
|
FLAG_CHECK(flag);
|
||
|
flag = DB_AFTER;
|
||
|
break;
|
||
|
case DBCPUT_BEFORE:
|
||
|
FLAG_CHECK(flag);
|
||
|
flag = DB_BEFORE;
|
||
|
break;
|
||
|
case DBCPUT_CURRENT:
|
||
|
FLAG_CHECK(flag);
|
||
|
flag = DB_CURRENT;
|
||
|
break;
|
||
|
case DBCPUT_KEYFIRST:
|
||
|
FLAG_CHECK(flag);
|
||
|
flag = DB_KEYFIRST;
|
||
|
break;
|
||
|
case DBCPUT_KEYLAST:
|
||
|
FLAG_CHECK(flag);
|
||
|
flag = DB_KEYLAST;
|
||
|
break;
|
||
|
case DBCPUT_OVERWRITE_DUP:
|
||
|
FLAG_CHECK(flag);
|
||
|
flag = DB_OVERWRITE_DUP;
|
||
|
break;
|
||
|
case DBCPUT_PART:
|
||
|
if (i > (objc - 2)) {
|
||
|
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.)
|
||
|
*/
|
||
|
}
|
||
|
if (result != TCL_OK)
|
||
|
break;
|
||
|
}
|
||
|
if (result != TCL_OK)
|
||
|
goto out;
|
||
|
|
||
|
/*
|
||
|
* We need to determine if we are a recno database or not. If we are,
|
||
|
* then key.data is a recno, not a string.
|
||
|
*/
|
||
|
dbcip = _PtrToInfo(dbc);
|
||
|
if (dbcip == NULL) {
|
||
|
type = DB_UNKNOWN;
|
||
|
thisdbp = NULL;
|
||
|
} else {
|
||
|
dbip = dbcip->i_parent;
|
||
|
if (dbip == NULL) {
|
||
|
Tcl_SetResult(interp, "Cursor without parent database",
|
||
|
TCL_STATIC);
|
||
|
result = TCL_ERROR;
|
||
|
return (result);
|
||
|
}
|
||
|
thisdbp = dbip->i_dbp;
|
||
|
(void)thisdbp->get_type(thisdbp, &type);
|
||
|
}
|
||
|
/*
|
||
|
* When we get here, we better have:
|
||
|
* 1 arg if -after, -before or -current
|
||
|
* 2 args in all other cases
|
||
|
*/
|
||
|
if (flag == DB_AFTER || flag == DB_BEFORE || flag == DB_CURRENT) {
|
||
|
if (i != (objc - 1)) {
|
||
|
Tcl_WrongNumArgs(interp, 2, objv,
|
||
|
"?-args? data");
|
||
|
result = TCL_ERROR;
|
||
|
goto out;
|
||
|
}
|
||
|
/*
|
||
|
* We want to get the key back, so we need to set
|
||
|
* up the location to get it back in.
|
||
|
*/
|
||
|
if (type == DB_RECNO || type == DB_QUEUE) {
|
||
|
recno = 0;
|
||
|
key.data = &recno;
|
||
|
key.size = sizeof(db_recno_t);
|
||
|
}
|
||
|
} else {
|
||
|
if (i != (objc - 2)) {
|
||
|
Tcl_WrongNumArgs(interp, 2, objv,
|
||
|
"?-args? key data");
|
||
|
result = TCL_ERROR;
|
||
|
goto out;
|
||
|
}
|
||
|
if (type == DB_HEAP || type == DB_RECNO || type == DB_QUEUE) {
|
||
|
result = _GetUInt32(interp, objv[objc-2], &recno);
|
||
|
if (result == TCL_OK) {
|
||
|
key.data = &recno;
|
||
|
key.size = sizeof(db_recno_t);
|
||
|
} else
|
||
|
return (result);
|
||
|
} else {
|
||
|
ret = _CopyObjBytes(interp, objv[objc-2], &ktmp,
|
||
|
&key.size, &freekey);
|
||
|
if (ret != 0) {
|
||
|
result = _ReturnSetup(interp, ret,
|
||
|
DB_RETOK_DBCPUT(ret), "dbc put");
|
||
|
return (result);
|
||
|
}
|
||
|
key.data = ktmp;
|
||
|
}
|
||
|
}
|
||
|
ret = _CopyObjBytes(interp, objv[objc-1], &dtmp,
|
||
|
&data.size, &freedata);
|
||
|
data.data = dtmp;
|
||
|
if (ret != 0) {
|
||
|
result = _ReturnSetup(interp, ret,
|
||
|
DB_RETOK_DBCPUT(ret), "dbc put");
|
||
|
goto out;
|
||
|
}
|
||
|
_debug_check();
|
||
|
if (type != DB_HEAP) {
|
||
|
ret = dbc->put(dbc, &key, &data, flag);
|
||
|
result = _ReturnSetup(interp, ret, DB_RETOK_DBCPUT(ret),
|
||
|
"dbc put");
|
||
|
} else {
|
||
|
hkey.data = &rid;
|
||
|
hkey.ulen = hkey.size = sizeof(DB_HEAP_RID);
|
||
|
hkey.flags = DB_DBT_USERMEM;
|
||
|
hrdbp = ((DBTCL_INFO *)thisdbp->api_internal)->hrdbp;
|
||
|
if (flag != DB_CURRENT) {
|
||
|
/* Given a recno, need to find the associated RID. */
|
||
|
ret = hrdbp->get(hrdbp, dbc->txn, &key, &hkey, 0);
|
||
|
result = _ReturnSetup(interp,
|
||
|
ret, DB_RETOK_DBGET(ret), "db get recno");
|
||
|
} else {
|
||
|
/* We have neither RID nor recno, but need both. */
|
||
|
memset(&tmpdata, 0, sizeof(DBT));
|
||
|
tmpdata.dlen = 0;
|
||
|
tmpdata.flags = DB_DBT_PARTIAL | DB_DBT_USERMEM;
|
||
|
ret = dbc->get(dbc, &hkey, &tmpdata, DB_CURRENT);
|
||
|
result = _ReturnSetup(interp,
|
||
|
ret, DB_RETOK_DBGET(ret), "dbc get");
|
||
|
|
||
|
hsdbp = ((DBTCL_INFO *)thisdbp->api_internal)->hsdbp;
|
||
|
key.data = &recno;
|
||
|
key.ulen = sizeof(db_recno_t);
|
||
|
key.flags = DB_DBT_USERMEM;
|
||
|
ret = hsdbp->pget(hsdbp,
|
||
|
dbc->txn, &hkey, &key, &tmpdata, 0);
|
||
|
result = _ReturnSetup(interp,
|
||
|
ret, DB_RETOK_DBGET(ret), "db pget rid");
|
||
|
}
|
||
|
|
||
|
/* Do the put in the heap db first */
|
||
|
ret = dbc->put(dbc, &hkey, &data, flag);
|
||
|
if (ret) {
|
||
|
result = _ReturnSetup(interp,
|
||
|
ret, DB_RETOK_DBCPUT(ret), "dbc put");
|
||
|
goto out;
|
||
|
}
|
||
|
|
||
|
hkey.flags = DB_DBT_USERMEM;
|
||
|
ret = hrdbp->put(hrdbp, dbc->txn, &key, &hkey, 0);
|
||
|
result = _ReturnSetup(interp,
|
||
|
ret, DB_RETOK_DBCPUT(ret), "dbc put recno");
|
||
|
|
||
|
/*
|
||
|
* To keep the consistency, if the put in recno db fails,
|
||
|
* the current key and data will be removed from the heap db.
|
||
|
*/
|
||
|
if (dbc->txn == NULL && ret != 0)
|
||
|
(void)thisdbp->del(thisdbp, NULL, &hkey, 0);
|
||
|
}
|
||
|
if (ret == 0 &&
|
||
|
(flag == DB_AFTER || flag == DB_BEFORE) &&
|
||
|
(type == DB_RECNO || type == DB_HEAP)) {
|
||
|
res = Tcl_NewWideIntObj((Tcl_WideInt)*(db_recno_t *)key.data);
|
||
|
Tcl_SetObjResult(interp, res);
|
||
|
}
|
||
|
out:
|
||
|
if (freedata)
|
||
|
__os_free(NULL, dtmp);
|
||
|
if (freekey)
|
||
|
__os_free(NULL, ktmp);
|
||
|
return (result);
|
||
|
}
|
||
|
|
||
|
/*
|
||
|
* tcl_dbc_get --
|
||
|
*/
|
||
|
static int
|
||
|
tcl_DbcGet(interp, objc, objv, dbc, ispget)
|
||
|
Tcl_Interp *interp; /* Interpreter */
|
||
|
int objc; /* How many arguments? */
|
||
|
Tcl_Obj *CONST objv[]; /* The argument objects */
|
||
|
DBC *dbc; /* Cursor pointer */
|
||
|
int ispget; /* 1 for pget, 0 for get */
|
||
|
{
|
||
|
static const char *dbcgetopts[] = {
|
||
|
#ifdef CONFIG_TEST
|
||
|
"-data_buf_size",
|
||
|
"-get_both_range",
|
||
|
"-key_buf_size",
|
||
|
"-multi",
|
||
|
"-multi_key",
|
||
|
"-nolease",
|
||
|
"-read_committed",
|
||
|
"-read_uncommitted",
|
||
|
#endif
|
||
|
"-current",
|
||
|
"-first",
|
||
|
"-get_both",
|
||
|
"-get_recno",
|
||
|
"-join_item",
|
||
|
"-last",
|
||
|
"-next",
|
||
|
"-nextdup",
|
||
|
"-nextnodup",
|
||
|
"-partial",
|
||
|
"-prev",
|
||
|
"-prevdup",
|
||
|
"-prevnodup",
|
||
|
"-rmw",
|
||
|
"-set",
|
||
|
"-set_range",
|
||
|
"-set_recno",
|
||
|
NULL
|
||
|
};
|
||
|
enum dbcgetopts {
|
||
|
#ifdef CONFIG_TEST
|
||
|
DBCGET_DATA_BUF_SIZE,
|
||
|
DBCGET_BOTH_RANGE,
|
||
|
DBCGET_KEY_BUF_SIZE,
|
||
|
DBCGET_MULTI,
|
||
|
DBCGET_MULTI_KEY,
|
||
|
DBCGET_NOLEASE,
|
||
|
DBCGET_READ_COMMITTED,
|
||
|
DBCGET_READ_UNCOMMITTED,
|
||
|
#endif
|
||
|
DBCGET_CURRENT,
|
||
|
DBCGET_FIRST,
|
||
|
DBCGET_BOTH,
|
||
|
DBCGET_RECNO,
|
||
|
DBCGET_JOIN,
|
||
|
DBCGET_LAST,
|
||
|
DBCGET_NEXT,
|
||
|
DBCGET_NEXTDUP,
|
||
|
DBCGET_NEXTNODUP,
|
||
|
DBCGET_PART,
|
||
|
DBCGET_PREV,
|
||
|
DBCGET_PREVDUP,
|
||
|
DBCGET_PREVNODUP,
|
||
|
DBCGET_RMW,
|
||
|
DBCGET_SET,
|
||
|
DBCGET_SETRANGE,
|
||
|
DBCGET_SETRECNO
|
||
|
};
|
||
|
DB *hrdbp, *hsdbp, *thisdbp;
|
||
|
DB_HEAP_RID rid;
|
||
|
DBT hkey, key, data, pdata, rkey, rdata, tmpdata;
|
||
|
DBTCL_INFO *dbcip, *dbip;
|
||
|
DBTYPE ptype, type;
|
||
|
Tcl_Obj **elemv, *myobj, *retlist;
|
||
|
void *dtmp, *ktmp;
|
||
|
db_recno_t precno, recno;
|
||
|
u_int32_t flag, heapflag, op;
|
||
|
int elemc, freekey, freedata, i, optindex, result, ret;
|
||
|
#ifdef CONFIG_TEST
|
||
|
int data_buf_size, key_buf_size;
|
||
|
|
||
|
data_buf_size = key_buf_size = 0;
|
||
|
#endif
|
||
|
COMPQUIET(dtmp, NULL);
|
||
|
COMPQUIET(ktmp, NULL);
|
||
|
result = TCL_OK;
|
||
|
flag = heapflag = 0;
|
||
|
freekey = freedata = 0;
|
||
|
hrdbp = hsdbp = NULL;
|
||
|
type = ptype = DB_UNKNOWN;
|
||
|
memset(&hkey, 0, sizeof(hkey));
|
||
|
memset(&key, 0, sizeof(key));
|
||
|
memset(&data, 0, sizeof(data));
|
||
|
memset(&pdata, 0, sizeof(DBT));
|
||
|
memset(&rkey, 0, sizeof(DBT));
|
||
|
memset(&rdata, 0, sizeof(DBT));
|
||
|
memset(&tmpdata, 0, sizeof(DBT));
|
||
|
tmpdata.flags = DB_DBT_PARTIAL | DB_DBT_USERMEM;
|
||
|
|
||
|
if (objc < 2) {
|
||
|
Tcl_WrongNumArgs(interp, 2, objv, "?-args? ?key?");
|
||
|
return (TCL_ERROR);
|
||
|
}
|
||
|
|
||
|
/*
|
||
|
* Get the command name index from the object based on the options
|
||
|
* defined above.
|
||
|
*/
|
||
|
i = 2;
|
||
|
while (i < objc) {
|
||
|
if (Tcl_GetIndexFromObj(interp, objv[i], dbcgetopts,
|
||
|
"option", TCL_EXACT, &optindex) != TCL_OK) {
|
||
|
/*
|
||
|
* Reset the result so we don't get
|
||
|
* an errant error message if there is another error.
|
||
|
*/
|
||
|
if (IS_HELP(objv[i]) == TCL_OK) {
|
||
|
result = TCL_OK;
|
||
|
goto out;
|
||
|
}
|
||
|
Tcl_ResetResult(interp);
|
||
|
break;
|
||
|
}
|
||
|
i++;
|
||
|
|
||
|
#define FLAG_CHECK2_STDARG \
|
||
|
(DB_RMW | DB_MULTIPLE | DB_MULTIPLE_KEY | DB_IGNORE_LEASE | \
|
||
|
DB_READ_UNCOMMITTED | DB_READ_COMMITTED)
|
||
|
|
||
|
switch ((enum dbcgetopts)optindex) {
|
||
|
#ifdef CONFIG_TEST
|
||
|
case DBCGET_DATA_BUF_SIZE:
|
||
|
result =
|
||
|
Tcl_GetIntFromObj(interp, objv[i], &data_buf_size);
|
||
|
if (result != TCL_OK)
|
||
|
goto out;
|
||
|
i++;
|
||
|
break;
|
||
|
case DBCGET_BOTH_RANGE:
|
||
|
FLAG_CHECK2(flag, FLAG_CHECK2_STDARG);
|
||
|
flag |= DB_GET_BOTH_RANGE;
|
||
|
break;
|
||
|
case DBCGET_KEY_BUF_SIZE:
|
||
|
result =
|
||
|
Tcl_GetIntFromObj(interp, objv[i], &key_buf_size);
|
||
|
if (result != TCL_OK)
|
||
|
goto out;
|
||
|
i++;
|
||
|
break;
|
||
|
case DBCGET_MULTI:
|
||
|
flag |= DB_MULTIPLE;
|
||
|
result =
|
||
|
Tcl_GetIntFromObj(interp, objv[i], &data_buf_size);
|
||
|
if (result != TCL_OK)
|
||
|
goto out;
|
||
|
i++;
|
||
|
break;
|
||
|
case DBCGET_MULTI_KEY:
|
||
|
flag |= DB_MULTIPLE_KEY;
|
||
|
result =
|
||
|
Tcl_GetIntFromObj(interp, objv[i], &data_buf_size);
|
||
|
if (result != TCL_OK)
|
||
|
goto out;
|
||
|
i++;
|
||
|
break;
|
||
|
case DBCGET_NOLEASE:
|
||
|
flag |= DB_IGNORE_LEASE;
|
||
|
break;
|
||
|
case DBCGET_READ_COMMITTED:
|
||
|
flag |= DB_READ_COMMITTED;
|
||
|
break;
|
||
|
case DBCGET_READ_UNCOMMITTED:
|
||
|
flag |= DB_READ_UNCOMMITTED;
|
||
|
break;
|
||
|
#endif
|
||
|
case DBCGET_RMW:
|
||
|
flag |= DB_RMW;
|
||
|
break;
|
||
|
case DBCGET_CURRENT:
|
||
|
FLAG_CHECK2(flag, FLAG_CHECK2_STDARG);
|
||
|
flag |= DB_CURRENT;
|
||
|
break;
|
||
|
case DBCGET_FIRST:
|
||
|
FLAG_CHECK2(flag, FLAG_CHECK2_STDARG);
|
||
|
flag |= DB_FIRST;
|
||
|
break;
|
||
|
case DBCGET_LAST:
|
||
|
FLAG_CHECK2(flag, FLAG_CHECK2_STDARG);
|
||
|
flag |= DB_LAST;
|
||
|
break;
|
||
|
case DBCGET_NEXT:
|
||
|
FLAG_CHECK2(flag, FLAG_CHECK2_STDARG);
|
||
|
flag |= DB_NEXT;
|
||
|
break;
|
||
|
case DBCGET_PREV:
|
||
|
FLAG_CHECK2(flag, FLAG_CHECK2_STDARG);
|
||
|
flag |= DB_PREV;
|
||
|
break;
|
||
|
case DBCGET_PREVDUP:
|
||
|
FLAG_CHECK2(flag, FLAG_CHECK2_STDARG);
|
||
|
flag |= DB_PREV_DUP;
|
||
|
break;
|
||
|
case DBCGET_PREVNODUP:
|
||
|
FLAG_CHECK2(flag, FLAG_CHECK2_STDARG);
|
||
|
flag |= DB_PREV_NODUP;
|
||
|
break;
|
||
|
case DBCGET_NEXTNODUP:
|
||
|
FLAG_CHECK2(flag, FLAG_CHECK2_STDARG);
|
||
|
flag |= DB_NEXT_NODUP;
|
||
|
break;
|
||
|
case DBCGET_NEXTDUP:
|
||
|
FLAG_CHECK2(flag, FLAG_CHECK2_STDARG);
|
||
|
flag |= DB_NEXT_DUP;
|
||
|
break;
|
||
|
case DBCGET_BOTH:
|
||
|
FLAG_CHECK2(flag, FLAG_CHECK2_STDARG);
|
||
|
flag |= DB_GET_BOTH;
|
||
|
break;
|
||
|
case DBCGET_RECNO:
|
||
|
FLAG_CHECK2(flag, FLAG_CHECK2_STDARG);
|
||
|
flag |= DB_GET_RECNO;
|
||
|
break;
|
||
|
case DBCGET_JOIN:
|
||
|
FLAG_CHECK2(flag, FLAG_CHECK2_STDARG);
|
||
|
flag |= DB_JOIN_ITEM;
|
||
|
break;
|
||
|
case DBCGET_SET:
|
||
|
FLAG_CHECK2(flag, FLAG_CHECK2_STDARG);
|
||
|
flag |= DB_SET;
|
||
|
break;
|
||
|
case DBCGET_SETRANGE:
|
||
|
FLAG_CHECK2(flag, FLAG_CHECK2_STDARG);
|
||
|
flag |= DB_SET_RANGE;
|
||
|
break;
|
||
|
case DBCGET_SETRECNO:
|
||
|
FLAG_CHECK2(flag, FLAG_CHECK2_STDARG);
|
||
|
flag |= DB_SET_RECNO;
|
||
|
break;
|
||
|
case DBCGET_PART:
|
||
|
if (i == objc) {
|
||
|
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_OK)
|
||
|
goto out;
|
||
|
heapflag = flag & ~DB_OPFLAGS_MASK;
|
||
|
heapflag &= ~DB_MULTIPLE_KEY;
|
||
|
if (F_ISSET(dbc, DBC_READ_COMMITTED))
|
||
|
heapflag |= DB_READ_COMMITTED;
|
||
|
if (F_ISSET(dbc, DBC_READ_UNCOMMITTED))
|
||
|
heapflag |= DB_READ_UNCOMMITTED;
|
||
|
|
||
|
/*
|
||
|
* We need to determine if we are a recno database
|
||
|
* or not. If we are, then key.data is a recno, not
|
||
|
* a string.
|
||
|
*/
|
||
|
dbcip = _PtrToInfo(dbc);
|
||
|
if (dbcip != NULL) {
|
||
|
dbip = dbcip->i_parent;
|
||
|
if (dbip == NULL) {
|
||
|
Tcl_SetResult(interp, "Cursor without parent database",
|
||
|
TCL_STATIC);
|
||
|
result = TCL_ERROR;
|
||
|
goto out;
|
||
|
}
|
||
|
thisdbp = dbip->i_dbp;
|
||
|
(void)thisdbp->get_type(thisdbp, &type);
|
||
|
if (ispget && thisdbp->s_primary != NULL)
|
||
|
(void)thisdbp->
|
||
|
s_primary->get_type(thisdbp->s_primary, &ptype);
|
||
|
else
|
||
|
ptype = DB_UNKNOWN;
|
||
|
if (type == DB_HEAP) {
|
||
|
hrdbp = dbip->hrdbp;
|
||
|
hsdbp = dbip->hsdbp;
|
||
|
}
|
||
|
}
|
||
|
/*
|
||
|
* When we get here, we better have:
|
||
|
* 2 args, key and data if GET_BOTH/GET_BOTH_RANGE was specified.
|
||
|
* 1 arg if -set, -set_range or -set_recno
|
||
|
* 0 in all other cases.
|
||
|
*/
|
||
|
op = flag & DB_OPFLAGS_MASK;
|
||
|
switch (op) {
|
||
|
case DB_GET_BOTH:
|
||
|
#ifdef CONFIG_TEST
|
||
|
case DB_GET_BOTH_RANGE:
|
||
|
#endif
|
||
|
if (i != (objc - 2)) {
|
||
|
Tcl_WrongNumArgs(interp, 2, objv,
|
||
|
"?-args? -get_both key data");
|
||
|
result = TCL_ERROR;
|
||
|
goto out;
|
||
|
} else {
|
||
|
if (type == DB_RECNO ||
|
||
|
type == DB_QUEUE || type == DB_HEAP) {
|
||
|
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],
|
||
|
&ktmp, &key.size, &freekey);
|
||
|
if (ret != 0) {
|
||
|
result = _ReturnSetup(interp, ret,
|
||
|
DB_RETOK_DBCGET(ret), "dbc get");
|
||
|
return (result);
|
||
|
}
|
||
|
key.data = ktmp;
|
||
|
}
|
||
|
if (ptype == DB_RECNO ||
|
||
|
ptype == DB_QUEUE || ptype == DB_HEAP) {
|
||
|
result = _GetUInt32(
|
||
|
interp, objv[objc-1], &precno);
|
||
|
if (result == TCL_OK) {
|
||
|
data.data = &precno;
|
||
|
data.size = sizeof(db_recno_t);
|
||
|
} else
|
||
|
goto out;
|
||
|
} else {
|
||
|
ret = _CopyObjBytes(interp, objv[objc-1],
|
||
|
&dtmp, &data.size, &freedata);
|
||
|
if (ret != 0) {
|
||
|
result = _ReturnSetup(interp, ret,
|
||
|
DB_RETOK_DBCGET(ret), "dbc get");
|
||
|
goto out;
|
||
|
}
|
||
|
data.data = dtmp;
|
||
|
}
|
||
|
}
|
||
|
break;
|
||
|
case DB_SET:
|
||
|
case DB_SET_RANGE:
|
||
|
case DB_SET_RECNO:
|
||
|
if (i != (objc - 1)) {
|
||
|
Tcl_WrongNumArgs(interp, 2, objv, "?-args? key");
|
||
|
result = TCL_ERROR;
|
||
|
goto out;
|
||
|
}
|
||
|
#ifdef CONFIG_TEST
|
||
|
if (data_buf_size != 0) {
|
||
|
(void)__os_malloc(
|
||
|
NULL, (size_t)data_buf_size, &data.data);
|
||
|
data.ulen = (u_int32_t)data_buf_size;
|
||
|
data.flags |= DB_DBT_USERMEM;
|
||
|
} else
|
||
|
#endif
|
||
|
data.flags |= DB_DBT_MALLOC;
|
||
|
if (op == DB_SET_RECNO ||
|
||
|
type == DB_HEAP || type == DB_RECNO || type == DB_QUEUE) {
|
||
|
result = _GetUInt32(interp, objv[objc - 1], &recno);
|
||
|
key.data = &recno;
|
||
|
key.size = sizeof(db_recno_t);
|
||
|
} 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],
|
||
|
&ktmp, &key.size, &freekey);
|
||
|
if (ret != 0) {
|
||
|
result = _ReturnSetup(interp, ret,
|
||
|
DB_RETOK_DBCGET(ret), "dbc get");
|
||
|
return (result);
|
||
|
}
|
||
|
key.data = ktmp;
|
||
|
}
|
||
|
break;
|
||
|
default:
|
||
|
if (i != objc) {
|
||
|
Tcl_WrongNumArgs(interp, 2, objv, "?-args?");
|
||
|
result = TCL_ERROR;
|
||
|
goto out;
|
||
|
}
|
||
|
#ifdef CONFIG_TEST
|
||
|
if (key_buf_size != 0) {
|
||
|
(void)__os_malloc(
|
||
|
NULL, (size_t)key_buf_size, &key.data);
|
||
|
key.ulen = (u_int32_t)key_buf_size;
|
||
|
key.flags |= DB_DBT_USERMEM;
|
||
|
} else
|
||
|
#endif
|
||
|
key.flags |= DB_DBT_MALLOC;
|
||
|
#ifdef CONFIG_TEST
|
||
|
if (data_buf_size != 0) {
|
||
|
(void)__os_malloc(
|
||
|
NULL, (size_t)data_buf_size, &data.data);
|
||
|
data.ulen = (u_int32_t)data_buf_size;
|
||
|
data.flags |= DB_DBT_USERMEM;
|
||
|
} else
|
||
|
#endif
|
||
|
data.flags |= DB_DBT_MALLOC;
|
||
|
}
|
||
|
|
||
|
_debug_check();
|
||
|
|
||
|
/*
|
||
|
* Heap cannot be a secondary, so with type == DB_HEAP we know that
|
||
|
* ispget is false.
|
||
|
*/
|
||
|
if (type == DB_HEAP && (op == DB_GET_BOTH ||
|
||
|
op == DB_GET_BOTH_RANGE || op == DB_SET || op == DB_SET_RANGE)) {
|
||
|
rkey.data = &recno;
|
||
|
rkey.ulen = rkey.size = sizeof(db_recno_t);
|
||
|
rkey.flags |= DB_DBT_USERMEM;
|
||
|
if (key.data != NULL && F_ISSET(&key, DB_DBT_USERMEM))
|
||
|
__os_free(NULL, key.data);
|
||
|
if (key.data != NULL && F_ISSET(&key, DB_DBT_MALLOC))
|
||
|
__os_ufree(NULL, key.data);
|
||
|
memset(&key, 0, sizeof(DBT));
|
||
|
key.data = &rid;
|
||
|
key.ulen = key.size = sizeof(DB_HEAP_RID);
|
||
|
key.flags |= DB_DBT_USERMEM;
|
||
|
|
||
|
/*
|
||
|
* This is a noncursor get on recno db, use heapflag because
|
||
|
* the cursor op flags have been removed.
|
||
|
*/
|
||
|
ret = hrdbp->get(hrdbp, dbc->txn, &rkey, &key, heapflag);
|
||
|
if (ret != 0) {
|
||
|
result = _ReturnSetup(
|
||
|
interp, ret, DB_RETOK_DBGET(ret), "db get");
|
||
|
retlist = Tcl_NewListObj(0, NULL);
|
||
|
goto out1;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
/*
|
||
|
* If we're doing a pget and DB_GET_BOTH is set, the primary key (stored
|
||
|
* in data) needs to match, too. For a HEAP primary, we're called with
|
||
|
* a recno primary key and we need to translate that to an RID. (ptype
|
||
|
* is only set if we're doing a pget.)
|
||
|
*/
|
||
|
if (ptype == DB_HEAP &&
|
||
|
(op == DB_GET_BOTH || op == DB_GET_BOTH_RANGE)) {
|
||
|
rkey.data = &precno;
|
||
|
rkey.size = rkey.ulen = sizeof(db_recno_t);
|
||
|
rkey.flags = DB_DBT_USERMEM;
|
||
|
if (data.data != NULL && F_ISSET(&data, DB_DBT_USERMEM))
|
||
|
__os_free(NULL, data.data);
|
||
|
if (data.data != NULL && F_ISSET(&data, DB_DBT_MALLOC))
|
||
|
__os_ufree(NULL, data.data);
|
||
|
memset(&data, 0, sizeof(DBT));
|
||
|
data.data = &rid;
|
||
|
data.size = data.ulen = sizeof(DB_HEAP_RID);
|
||
|
data.flags = DB_DBT_USERMEM;
|
||
|
ret = hrdbp->get(hrdbp, dbc->txn, &rkey, &data, heapflag);
|
||
|
if (ret != 0) {
|
||
|
result = _ReturnSetup(
|
||
|
interp, ret, DB_RETOK_DBGET(ret), "db get");
|
||
|
retlist = Tcl_NewListObj(0, NULL);
|
||
|
goto out1;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if (ispget) {
|
||
|
F_SET(&pdata, DB_DBT_MALLOC);
|
||
|
ret = dbc->pget(dbc, &key, &data, &pdata, flag);
|
||
|
if (ret == 0 && ptype == DB_HEAP) {
|
||
|
rid.pgno = ((DB_HEAP_RID *)data.data)->pgno;
|
||
|
rid.indx = ((DB_HEAP_RID *)data.data)->indx;
|
||
|
hkey.data = &rid;
|
||
|
hkey.ulen = hkey.size = data.size;
|
||
|
hkey.flags = DB_DBT_USERMEM;
|
||
|
hsdbp = ((DBTCL_INFO *)
|
||
|
dbc->dbp->s_primary->api_internal)->hsdbp;
|
||
|
ret = hsdbp->pget(hsdbp,
|
||
|
dbc->txn, &hkey, &data, &tmpdata, 0);
|
||
|
}
|
||
|
|
||
|
} else
|
||
|
ret = dbc->get(dbc, &key, &data, flag);
|
||
|
result = _ReturnSetup(interp, ret, DB_RETOK_DBCGET(ret), "dbc get");
|
||
|
if (result == TCL_ERROR)
|
||
|
goto out;
|
||
|
|
||
|
retlist = Tcl_NewListObj(0, NULL);
|
||
|
if (ret != 0)
|
||
|
goto out1;
|
||
|
if (op == DB_GET_RECNO) {
|
||
|
recno = *((db_recno_t *)data.data);
|
||
|
myobj = Tcl_NewWideIntObj((Tcl_WideInt)recno);
|
||
|
result = Tcl_ListObjAppendElement(interp, retlist, myobj);
|
||
|
} else {
|
||
|
if (flag & (DB_MULTIPLE|DB_MULTIPLE_KEY))
|
||
|
result = _SetMultiList(interp,
|
||
|
retlist, &key, &data, type, flag);
|
||
|
else if ((type == DB_RECNO || type == DB_QUEUE) &&
|
||
|
key.data != NULL) {
|
||
|
if (ispget)
|
||
|
result = _Set3DBTList(interp, retlist, &key, 1,
|
||
|
&data,
|
||
|
(ptype == DB_RECNO || ptype == DB_QUEUE),
|
||
|
&pdata);
|
||
|
else
|
||
|
result = _SetListRecnoElem(interp, retlist,
|
||
|
*(db_recno_t *)key.data,
|
||
|
data.data, data.size);
|
||
|
} else if (type == DB_HEAP) {
|
||
|
/*
|
||
|
* If given a record number, we're done. If we don't
|
||
|
* yet have a record number, we need to look it up.
|
||
|
*/
|
||
|
if (op != DB_GET_BOTH && op != DB_SET &&
|
||
|
op != DB_GET_BOTH_RANGE && op != DB_SET_RANGE) {
|
||
|
rdata.flags = DB_DBT_PARTIAL | DB_DBT_USERMEM;
|
||
|
rdata.dlen = 0;
|
||
|
rkey.data = &recno;
|
||
|
rkey.size = rkey.ulen = sizeof(db_recno_t);
|
||
|
rkey.flags = DB_DBT_USERMEM;
|
||
|
|
||
|
ret = hsdbp->pget(hsdbp, dbc->txn, &key,
|
||
|
&rkey, &rdata, heapflag);
|
||
|
result = _ReturnSetup(
|
||
|
interp, ret, DB_RETOK_DBGET(ret), "db get");
|
||
|
if (result == TCL_ERROR)
|
||
|
goto out;
|
||
|
retlist = Tcl_NewListObj(0, NULL);
|
||
|
if (ret != 0)
|
||
|
goto out1;
|
||
|
}
|
||
|
result = _SetListRecnoElem(interp, retlist,
|
||
|
*(db_recno_t *)rkey.data, data.data, data.size);
|
||
|
} else {
|
||
|
if (ispget)
|
||
|
result = _Set3DBTList(interp, retlist, &key, 0,
|
||
|
&data,
|
||
|
(ptype == DB_HEAP ||
|
||
|
ptype == DB_RECNO || ptype == DB_QUEUE),
|
||
|
&pdata);
|
||
|
else
|
||
|
result = _SetListElem(interp, retlist,
|
||
|
key.data, key.size, data.data, data.size);
|
||
|
}
|
||
|
}
|
||
|
out1:
|
||
|
if (result == TCL_OK)
|
||
|
Tcl_SetObjResult(interp, retlist);
|
||
|
/*
|
||
|
* If DB_DBT_MALLOC is set we need to free if DB allocated anything.
|
||
|
* If DB_DBT_USERMEM is set we need to free it because
|
||
|
* we allocated it (for data_buf_size/key_buf_size). That
|
||
|
* allocation does not apply to the pdata DBT. For heap, we do not
|
||
|
* malloc anything but move pointers around so nothing to free.
|
||
|
*/
|
||
|
out:
|
||
|
if (key.data != NULL && F_ISSET(&key, DB_DBT_MALLOC))
|
||
|
__os_ufree(dbc->env, key.data);
|
||
|
if (type != DB_HEAP &&
|
||
|
key.data != NULL && F_ISSET(&key, DB_DBT_USERMEM))
|
||
|
__os_free(dbc->env, key.data);
|
||
|
if (data.data != NULL && F_ISSET(&data, DB_DBT_MALLOC))
|
||
|
__os_ufree(dbc->env, data.data);
|
||
|
if (data.data != NULL && F_ISSET(&data, DB_DBT_USERMEM))
|
||
|
__os_free(dbc->env, data.data);
|
||
|
if (pdata.data != NULL && F_ISSET(&pdata, DB_DBT_MALLOC))
|
||
|
__os_ufree(dbc->env, pdata.data);
|
||
|
if (freedata)
|
||
|
__os_free(NULL, dtmp);
|
||
|
if (freekey)
|
||
|
__os_free(NULL, ktmp);
|
||
|
return (result);
|
||
|
|
||
|
}
|
||
|
|
||
|
/*
|
||
|
* tcl_DbcCompare --
|
||
|
*/
|
||
|
static int
|
||
|
tcl_DbcCompare(interp, objc, objv, dbc)
|
||
|
Tcl_Interp *interp; /* Interpreter */
|
||
|
int objc; /* How many arguments? */
|
||
|
Tcl_Obj *CONST objv[]; /* The argument objects */
|
||
|
DBC *dbc; /* Cursor pointer */
|
||
|
{
|
||
|
DBC *odbc;
|
||
|
DBTCL_INFO *dbcip, *dbip;
|
||
|
Tcl_Obj *res;
|
||
|
int cmp_res, result, ret;
|
||
|
char *arg, msg[MSG_SIZE];
|
||
|
|
||
|
result = TCL_OK;
|
||
|
res = NULL;
|
||
|
|
||
|
if (objc != 3) {
|
||
|
Tcl_WrongNumArgs(interp, 3, objv, "?-args?");
|
||
|
return (TCL_ERROR);
|
||
|
}
|
||
|
|
||
|
dbcip = _PtrToInfo(dbc);
|
||
|
if (dbcip == NULL) {
|
||
|
Tcl_SetResult(interp, "Cursor without info structure",
|
||
|
TCL_STATIC);
|
||
|
result = TCL_ERROR;
|
||
|
goto out;
|
||
|
} else {
|
||
|
dbip = dbcip->i_parent;
|
||
|
if (dbip == NULL) {
|
||
|
Tcl_SetResult(interp, "Cursor without parent database",
|
||
|
TCL_STATIC);
|
||
|
result = TCL_ERROR;
|
||
|
goto out;
|
||
|
}
|
||
|
}
|
||
|
/*
|
||
|
* When we get here, we better have:
|
||
|
* 2 args one DBC and an int address for the result
|
||
|
*/
|
||
|
arg = Tcl_GetStringFromObj(objv[2], NULL);
|
||
|
odbc = NAME_TO_DBC(arg);
|
||
|
if (odbc == NULL) {
|
||
|
snprintf(msg, MSG_SIZE,
|
||
|
"Cmp: Invalid cursor: %s\n", arg);
|
||
|
Tcl_SetResult(interp, msg, TCL_VOLATILE);
|
||
|
result = TCL_ERROR;
|
||
|
goto out;
|
||
|
}
|
||
|
|
||
|
ret = dbc->cmp(dbc, odbc, &cmp_res, 0);
|
||
|
if (ret != 0) {
|
||
|
result = _ReturnSetup(interp, ret,
|
||
|
DB_RETOK_STD(ret), "dbc cmp");
|
||
|
return (result);
|
||
|
}
|
||
|
res = Tcl_NewIntObj(cmp_res);
|
||
|
Tcl_SetObjResult(interp, res);
|
||
|
out:
|
||
|
return (result);
|
||
|
|
||
|
}
|
||
|
|
||
|
/*
|
||
|
* tcl_DbcDup --
|
||
|
*/
|
||
|
static int
|
||
|
tcl_DbcDup(interp, objc, objv, dbc)
|
||
|
Tcl_Interp *interp; /* Interpreter */
|
||
|
int objc; /* How many arguments? */
|
||
|
Tcl_Obj *CONST objv[]; /* The argument objects */
|
||
|
DBC *dbc; /* Cursor pointer */
|
||
|
{
|
||
|
static const char *dbcdupopts[] = {
|
||
|
"-position",
|
||
|
NULL
|
||
|
};
|
||
|
enum dbcdupopts {
|
||
|
DBCDUP_POS
|
||
|
};
|
||
|
DBC *newdbc;
|
||
|
DBTCL_INFO *dbcip, *newdbcip, *dbip;
|
||
|
Tcl_Obj *res;
|
||
|
u_int32_t flag;
|
||
|
int i, optindex, result, ret;
|
||
|
char newname[MSG_SIZE];
|
||
|
|
||
|
result = TCL_OK;
|
||
|
flag = 0;
|
||
|
res = NULL;
|
||
|
|
||
|
if (objc < 2) {
|
||
|
Tcl_WrongNumArgs(interp, 2, objv, "?-args?");
|
||
|
return (TCL_ERROR);
|
||
|
}
|
||
|
|
||
|
/*
|
||
|
* Get the command name index from the object based on the options
|
||
|
* defined above.
|
||
|
*/
|
||
|
i = 2;
|
||
|
while (i < objc) {
|
||
|
if (Tcl_GetIndexFromObj(interp, objv[i], dbcdupopts,
|
||
|
"option", TCL_EXACT, &optindex) != TCL_OK) {
|
||
|
/*
|
||
|
* Reset the result so we don't get
|
||
|
* an errant error message if there is another error.
|
||
|
*/
|
||
|
if (IS_HELP(objv[i]) == TCL_OK) {
|
||
|
result = TCL_OK;
|
||
|
goto out;
|
||
|
}
|
||
|
Tcl_ResetResult(interp);
|
||
|
break;
|
||
|
}
|
||
|
i++;
|
||
|
switch ((enum dbcdupopts)optindex) {
|
||
|
case DBCDUP_POS:
|
||
|
flag = DB_POSITION;
|
||
|
break;
|
||
|
}
|
||
|
if (result != TCL_OK)
|
||
|
break;
|
||
|
}
|
||
|
if (result != TCL_OK)
|
||
|
goto out;
|
||
|
|
||
|
/*
|
||
|
* We need to determine if we are a recno database
|
||
|
* or not. If we are, then key.data is a recno, not
|
||
|
* a string.
|
||
|
*/
|
||
|
dbcip = _PtrToInfo(dbc);
|
||
|
if (dbcip == NULL) {
|
||
|
Tcl_SetResult(interp, "Cursor without info structure",
|
||
|
TCL_STATIC);
|
||
|
result = TCL_ERROR;
|
||
|
goto out;
|
||
|
} else {
|
||
|
dbip = dbcip->i_parent;
|
||
|
if (dbip == NULL) {
|
||
|
Tcl_SetResult(interp, "Cursor without parent database",
|
||
|
TCL_STATIC);
|
||
|
result = TCL_ERROR;
|
||
|
goto out;
|
||
|
}
|
||
|
}
|
||
|
/*
|
||
|
* Now duplicate the cursor. If successful, we need to create
|
||
|
* a new cursor command.
|
||
|
*/
|
||
|
snprintf(newname, sizeof(newname),
|
||
|
"%s.c%d", dbip->i_name, dbip->i_dbdbcid);
|
||
|
newdbcip = _NewInfo(interp, NULL, newname, I_DBC);
|
||
|
if (newdbcip != NULL) {
|
||
|
ret = dbc->dup(dbc, &newdbc, flag);
|
||
|
if (ret == 0) {
|
||
|
dbip->i_dbdbcid++;
|
||
|
newdbcip->i_parent = dbip;
|
||
|
(void)Tcl_CreateObjCommand(interp, newname,
|
||
|
(Tcl_ObjCmdProc *)dbc_Cmd,
|
||
|
(ClientData)newdbc, NULL);
|
||
|
res = NewStringObj(newname, strlen(newname));
|
||
|
_SetInfoData(newdbcip, newdbc);
|
||
|
Tcl_SetObjResult(interp, res);
|
||
|
} else {
|
||
|
result = _ReturnSetup(interp, ret, DB_RETOK_STD(ret),
|
||
|
"db dup");
|
||
|
_DeleteInfo(newdbcip);
|
||
|
}
|
||
|
} else {
|
||
|
Tcl_SetResult(interp, "Could not set up info", TCL_STATIC);
|
||
|
result = TCL_ERROR;
|
||
|
}
|
||
|
out:
|
||
|
return (result);
|
||
|
|
||
|
}
|
||
|
|
||
|
/*
|
||
|
* tcl_DbcDel --
|
||
|
*/
|
||
|
static int
|
||
|
tcl_DbcDel(interp, objc, objv, dbc)
|
||
|
Tcl_Interp *interp; /* Interpreter */
|
||
|
int objc; /* How many arguments? */
|
||
|
Tcl_Obj *CONST objv[]; /* The argument objects */
|
||
|
DBC *dbc; /* Cursor pointer */
|
||
|
{
|
||
|
static const char *dbcdelopts[] = {
|
||
|
"-consume",
|
||
|
NULL
|
||
|
};
|
||
|
enum dbcdelopts {
|
||
|
DBCDEL_CONSUME
|
||
|
};
|
||
|
u_int32_t flag;
|
||
|
int i, optindex, result, ret;
|
||
|
|
||
|
result = TCL_OK;
|
||
|
flag = 0;
|
||
|
if (objc < 2) {
|
||
|
Tcl_WrongNumArgs(interp, 2, objv, "?-args?");
|
||
|
return (TCL_ERROR);
|
||
|
}
|
||
|
|
||
|
/*
|
||
|
* Get the command name index from the object based on the options
|
||
|
* defined above.
|
||
|
*/
|
||
|
i = 2;
|
||
|
while (i < objc) {
|
||
|
if (Tcl_GetIndexFromObj(interp, objv[i], dbcdelopts,
|
||
|
"option", TCL_EXACT, &optindex) != TCL_OK) {
|
||
|
/*
|
||
|
* Reset the result so we don't get
|
||
|
* an errant error message if there is another error.
|
||
|
*/
|
||
|
if (IS_HELP(objv[i]) == TCL_OK) {
|
||
|
result = TCL_OK;
|
||
|
goto out;
|
||
|
}
|
||
|
Tcl_ResetResult(interp);
|
||
|
break;
|
||
|
}
|
||
|
i++;
|
||
|
switch ((enum dbcdelopts)optindex) {
|
||
|
case DBCDEL_CONSUME:
|
||
|
flag = DB_CONSUME;
|
||
|
break;
|
||
|
}
|
||
|
}
|
||
|
if (dbc->dbp->type == DB_HEAP)
|
||
|
result = tcl_DbcHeapDel(interp, dbc);
|
||
|
else {
|
||
|
_debug_check();
|
||
|
ret = dbc->del(dbc, flag);
|
||
|
result = _ReturnSetup(
|
||
|
interp, ret, DB_RETOK_DBCDEL(ret), "dbc delete");
|
||
|
}
|
||
|
out:
|
||
|
return (result);
|
||
|
}
|