mirror of
https://github.com/MariaDB/server.git
synced 2025-01-16 12:02:42 +01:00
1771 lines
42 KiB
C
1771 lines
42 KiB
C
/*-
|
|
* See the file LICENSE for redistribution information.
|
|
*
|
|
* Copyright (c) 1999, 2000
|
|
* Sleepycat Software. All rights reserved.
|
|
*/
|
|
|
|
#include "db_config.h"
|
|
|
|
#ifndef lint
|
|
static const char revid[] = "$Id: tcl_db.c,v 11.55 2000/11/28 20:12:31 bostic Exp $";
|
|
#endif /* not lint */
|
|
|
|
#ifndef NO_SYSTEM_INCLUDES
|
|
#include <sys/types.h>
|
|
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <tcl.h>
|
|
#endif
|
|
|
|
#include "db_int.h"
|
|
#include "tcl_db.h"
|
|
|
|
/*
|
|
* Prototypes for procedures defined later in this file:
|
|
*/
|
|
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 *));
|
|
static int tcl_DbKeyRange __P((Tcl_Interp *, int, Tcl_Obj * CONST*, DB *));
|
|
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_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_DbGetjoin __P((Tcl_Interp *, int, Tcl_Obj * CONST*, DB *));
|
|
static int tcl_DbCount __P((Tcl_Interp *, int, Tcl_Obj * CONST*, DB *));
|
|
|
|
/*
|
|
*
|
|
* 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 char *dbcmds[] = {
|
|
"close",
|
|
"count",
|
|
"cursor",
|
|
"del",
|
|
"get",
|
|
"get_join",
|
|
"get_type",
|
|
"is_byteswapped",
|
|
"join",
|
|
"keyrange",
|
|
"put",
|
|
"stat",
|
|
"sync",
|
|
#if CONFIG_TEST
|
|
"test",
|
|
#endif
|
|
NULL
|
|
};
|
|
enum dbcmds {
|
|
DBCLOSE,
|
|
DBCOUNT,
|
|
DBCURSOR,
|
|
DBDELETE,
|
|
DBGET,
|
|
DBGETJOIN,
|
|
DBGETTYPE,
|
|
DBSWAPPED,
|
|
DBJOIN,
|
|
DBKEYRANGE,
|
|
DBPUT,
|
|
DBSTAT,
|
|
DBSYNC
|
|
#if CONFIG_TEST
|
|
, DBTEST
|
|
#endif
|
|
};
|
|
DB *dbp;
|
|
DBC *dbc;
|
|
DBTCL_INFO *dbip;
|
|
DBTCL_INFO *ip;
|
|
Tcl_Obj *res;
|
|
int cmdindex, result, ret;
|
|
char newname[MSG_SIZE];
|
|
|
|
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) {
|
|
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);
|
|
break;
|
|
case DBKEYRANGE:
|
|
result = tcl_DbKeyRange(interp, objc, objv, dbp);
|
|
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);
|
|
res = Tcl_NewIntObj(ret);
|
|
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);
|
|
if (ret == DB_BTREE)
|
|
res = Tcl_NewStringObj("btree", strlen("btree"));
|
|
else if (ret == DB_HASH)
|
|
res = Tcl_NewStringObj("hash", strlen("hash"));
|
|
else if (ret == DB_RECNO)
|
|
res = Tcl_NewStringObj("recno", strlen("recno"));
|
|
else if (ret == DB_QUEUE)
|
|
res = Tcl_NewStringObj("queue", strlen("queue"));
|
|
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 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;
|
|
}
|
|
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;
|
|
Tcl_CreateObjCommand(interp, newname,
|
|
(Tcl_ObjCmdProc *)dbc_Cmd,
|
|
(ClientData)dbc, NULL);
|
|
res =
|
|
Tcl_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;
|
|
Tcl_CreateObjCommand(interp, newname,
|
|
(Tcl_ObjCmdProc *)dbc_Cmd,
|
|
(ClientData)dbc, NULL);
|
|
res =
|
|
Tcl_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 DBGETJOIN:
|
|
result = tcl_DbGetjoin(interp, objc, objv, dbp);
|
|
break;
|
|
#if CONFIG_TEST
|
|
case DBTEST:
|
|
result = tcl_EnvTest(interp, objc, objv, dbp->dbenv);
|
|
break;
|
|
#endif
|
|
}
|
|
/*
|
|
* 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 */
|
|
{
|
|
DB_BTREE_STAT *bsp;
|
|
DB_HASH_STAT *hsp;
|
|
DB_QUEUE_STAT *qsp;
|
|
void *sp;
|
|
Tcl_Obj *res;
|
|
DBTYPE type;
|
|
u_int32_t flag;
|
|
int result, ret;
|
|
char *arg;
|
|
|
|
result = TCL_OK;
|
|
flag = 0;
|
|
|
|
if (objc > 3) {
|
|
Tcl_WrongNumArgs(interp, 2, objv, "?-recordcount?");
|
|
return (TCL_ERROR);
|
|
}
|
|
|
|
if (objc == 3) {
|
|
arg = Tcl_GetStringFromObj(objv[2], NULL);
|
|
if (strcmp(arg, "-recordcount") == 0)
|
|
flag = DB_RECORDCOUNT;
|
|
else if (strcmp(arg, "-cachedcounts") == 0)
|
|
flag = DB_CACHED_COUNTS;
|
|
else {
|
|
Tcl_SetResult(interp,
|
|
"db stat: unknown arg", TCL_STATIC);
|
|
return (TCL_ERROR);
|
|
}
|
|
}
|
|
|
|
_debug_check();
|
|
ret = dbp->stat(dbp, &sp, NULL, flag);
|
|
result = _ReturnSetup(interp, ret, "db stat");
|
|
if (result == TCL_ERROR)
|
|
return (result);
|
|
|
|
type = dbp->get_type(dbp);
|
|
/*
|
|
* 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("Number of keys", hsp->hash_nkeys);
|
|
MAKE_STAT_LIST("Number of records", hsp->hash_ndata);
|
|
MAKE_STAT_LIST("Estim. number of elements", hsp->hash_nelem);
|
|
MAKE_STAT_LIST("Fill factor", hsp->hash_ffactor);
|
|
MAKE_STAT_LIST("Buckets", hsp->hash_buckets);
|
|
MAKE_STAT_LIST("Free pages", hsp->hash_free);
|
|
MAKE_STAT_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_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("Number of records", qsp->qs_ndata);
|
|
MAKE_STAT_LIST("Number of pages", qsp->qs_pages);
|
|
MAKE_STAT_LIST("Bytes free", qsp->qs_pgfree);
|
|
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);
|
|
} else { /* BTREE and RECNO are same stats */
|
|
bsp = (DB_BTREE_STAT *)sp;
|
|
MAKE_STAT_LIST("Number of keys", bsp->bt_nkeys);
|
|
MAKE_STAT_LIST("Number of records", bsp->bt_ndata);
|
|
if (flag != DB_RECORDCOUNT) {
|
|
MAKE_STAT_LIST("Magic", bsp->bt_magic);
|
|
MAKE_STAT_LIST("Version", bsp->bt_version);
|
|
MAKE_STAT_LIST("Flags", bsp->bt_metaflags);
|
|
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("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("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);
|
|
}
|
|
}
|
|
Tcl_SetObjResult(interp, res);
|
|
error:
|
|
__os_free(sp, 0);
|
|
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 */
|
|
{
|
|
DBTCL_INFO *p, *nextp;
|
|
u_int32_t flag;
|
|
int result, ret;
|
|
char *arg;
|
|
|
|
result = TCL_OK;
|
|
flag = 0;
|
|
if (objc > 3) {
|
|
Tcl_WrongNumArgs(interp, 2, objv, "?-nosync?");
|
|
return (TCL_ERROR);
|
|
}
|
|
|
|
if (objc == 3) {
|
|
arg = Tcl_GetStringFromObj(objv[2], NULL);
|
|
if (strcmp(arg, "-nosync") == 0)
|
|
flag = DB_NOSYNC;
|
|
else {
|
|
Tcl_SetResult(interp,
|
|
"dbclose: unknown arg", TCL_STATIC);
|
|
return (TCL_ERROR);
|
|
}
|
|
}
|
|
|
|
/*
|
|
* 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);
|
|
_debug_check();
|
|
ret = (dbp)->close(dbp, flag);
|
|
result = _ReturnSetup(interp, 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 char *dbputopts[] = {
|
|
"-append",
|
|
"-nodupdata",
|
|
"-nooverwrite",
|
|
"-partial",
|
|
"-txn",
|
|
NULL
|
|
};
|
|
enum dbputopts {
|
|
DBPUT_APPEND,
|
|
DBGET_NODUPDATA,
|
|
DBPUT_NOOVER,
|
|
DBPUT_PART,
|
|
DBPUT_TXN
|
|
};
|
|
static char *dbputapp[] = {
|
|
"-append", NULL
|
|
};
|
|
enum dbputapp { DBPUT_APPEND0 };
|
|
DBT key, data;
|
|
DBTYPE type;
|
|
DB_TXN *txn;
|
|
Tcl_Obj **elemv, *res;
|
|
db_recno_t recno;
|
|
u_int32_t flag;
|
|
int elemc, end, i, itmp, optindex, result, ret;
|
|
char *arg, msg[MSG_SIZE];
|
|
|
|
txn = NULL;
|
|
result = TCL_OK;
|
|
flag = 0;
|
|
if (objc <= 3) {
|
|
Tcl_WrongNumArgs(interp, 2, objv, "?-args? key data");
|
|
return (TCL_ERROR);
|
|
}
|
|
|
|
memset(&key, 0, sizeof(key));
|
|
memset(&data, 0, sizeof(data));
|
|
|
|
/*
|
|
* 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".
|
|
*/
|
|
type = dbp->get_type(dbp);
|
|
|
|
/*
|
|
* We need to determine where the end of required args are. If we
|
|
* are using a QUEUE/RECNO db and -append, 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;
|
|
if (type == DB_QUEUE || type == DB_RECNO) {
|
|
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:
|
|
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) {
|
|
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 DBGET_NODUPDATA:
|
|
FLAG_CHECK(flag);
|
|
flag = DB_NODUPDATA;
|
|
break;
|
|
case DBPUT_NOOVER:
|
|
FLAG_CHECK(flag);
|
|
flag = DB_NOOVERWRITE;
|
|
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 = Tcl_GetIntFromObj(interp, elemv[0], &itmp);
|
|
data.doff = itmp;
|
|
if (result != TCL_OK)
|
|
break;
|
|
result = Tcl_GetIntFromObj(interp, elemv[1], &itmp);
|
|
data.dlen = itmp;
|
|
/*
|
|
* 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 we are a recno db and we are NOT using append, then the 2nd
|
|
* last arg is the key.
|
|
*/
|
|
if (type == DB_QUEUE || type == DB_RECNO) {
|
|
key.data = &recno;
|
|
key.ulen = key.size = sizeof(db_recno_t);
|
|
key.flags = DB_DBT_USERMEM;
|
|
if (flag == DB_APPEND)
|
|
recno = 0;
|
|
else {
|
|
result = Tcl_GetIntFromObj(interp, objv[objc-2], &itmp);
|
|
recno = itmp;
|
|
if (result != TCL_OK)
|
|
return (result);
|
|
}
|
|
} else {
|
|
key.data = Tcl_GetByteArrayFromObj(objv[objc-2], &itmp);
|
|
key.size = itmp;
|
|
}
|
|
/*
|
|
* XXX
|
|
* Tcl 8.1 Tcl_GetByteArrayFromObj/Tcl_GetIntFromObj bug.
|
|
*
|
|
* This line (and the line for key.data above) were moved from
|
|
* the beginning of the function to here.
|
|
*
|
|
* There is a bug in Tcl 8.1 and byte arrays in that if it happens
|
|
* to use an object as both a byte array and something else like
|
|
* an int, and you've done a Tcl_GetByteArrayFromObj, then you
|
|
* do a Tcl_GetIntFromObj, your memory is deleted.
|
|
*
|
|
* Workaround is to make sure all Tcl_GetByteArrayFromObj calls
|
|
* are done last.
|
|
*/
|
|
data.data = Tcl_GetByteArrayFromObj(objv[objc-1], &itmp);
|
|
data.size = itmp;
|
|
_debug_check();
|
|
ret = dbp->put(dbp, txn, &key, &data, flag);
|
|
result = _ReturnSetup(interp, ret, "db put");
|
|
if (ret == 0 &&
|
|
(type == DB_RECNO || type == DB_QUEUE) && flag == DB_APPEND) {
|
|
res = Tcl_NewIntObj(recno);
|
|
Tcl_SetObjResult(interp, res);
|
|
}
|
|
return (result);
|
|
}
|
|
|
|
/*
|
|
* tcl_db_get --
|
|
*/
|
|
static int
|
|
tcl_DbGet(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 char *dbgetopts[] = {
|
|
"-consume",
|
|
"-consume_wait",
|
|
"-get_both",
|
|
"-glob",
|
|
"-partial",
|
|
"-recno",
|
|
"-rmw",
|
|
"-txn",
|
|
NULL
|
|
};
|
|
enum dbgetopts {
|
|
DBGET_CONSUME,
|
|
DBGET_CONSUME_WAIT,
|
|
DBGET_BOTH,
|
|
DBGET_GLOB,
|
|
DBGET_PART,
|
|
DBGET_RECNO,
|
|
DBGET_RMW,
|
|
DBGET_TXN
|
|
};
|
|
DBC *dbc;
|
|
DBT key, data, save;
|
|
DBTYPE type;
|
|
DB_TXN *txn;
|
|
Tcl_Obj **elemv, *retlist;
|
|
db_recno_t recno;
|
|
u_int32_t flag, cflag, isdup, rmw;
|
|
int elemc, end, i, itmp, optindex, result, ret, useglob, userecno;
|
|
char *arg, *pattern, *prefix, msg[MSG_SIZE];
|
|
|
|
result = TCL_OK;
|
|
cflag = flag = rmw = 0;
|
|
useglob = userecno = 0;
|
|
txn = NULL;
|
|
pattern = prefix = NULL;
|
|
|
|
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));
|
|
|
|
/*
|
|
* Get the command name index from the object based on the options
|
|
* defined above.
|
|
*/
|
|
i = 2;
|
|
type = dbp->get_type(dbp);
|
|
end = objc;
|
|
while (i < end) {
|
|
if (Tcl_GetIndexFromObj(interp, objv[i], dbgetopts, "option",
|
|
TCL_EXACT, &optindex) != TCL_OK) {
|
|
if (IS_HELP(objv[i]) == TCL_OK)
|
|
return (TCL_OK);
|
|
Tcl_ResetResult(interp);
|
|
break;
|
|
}
|
|
i++;
|
|
switch ((enum dbgetopts)optindex) {
|
|
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 - 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,
|
|
"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) {
|
|
FLAG_CHECK(flag);
|
|
flag = DB_SET_RECNO;
|
|
}
|
|
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 = Tcl_GetIntFromObj(interp, elemv[0], &itmp);
|
|
save.doff = itmp;
|
|
if (result != TCL_OK)
|
|
break;
|
|
result = Tcl_GetIntFromObj(interp, elemv[1], &itmp);
|
|
save.dlen = itmp;
|
|
/*
|
|
* 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;
|
|
|
|
if (type == DB_RECNO || type == DB_QUEUE)
|
|
userecno = 1;
|
|
/*
|
|
* Check for illegal combos of options.
|
|
*/
|
|
if (useglob && (userecno || flag == DB_SET_RECNO ||
|
|
type == DB_RECNO || type == DB_QUEUE)) {
|
|
Tcl_SetResult(interp,
|
|
"Cannot use -glob and record numbers.\n",
|
|
TCL_STATIC);
|
|
result = TCL_ERROR;
|
|
goto out;
|
|
}
|
|
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;
|
|
|
|
/*
|
|
* isdup is used to know if we support duplicates. If not, we
|
|
* can just do a db->get call and avoid using cursors.
|
|
* XXX
|
|
* When there is a db->get_flags method, it should be used.
|
|
* isdup = dbp->get_flags(dbp) & DB_DUP;
|
|
* For now we illegally peek.
|
|
* XXX
|
|
*/
|
|
isdup = dbp->flags & DB_AM_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.
|
|
*/
|
|
if (pattern == NULL && (isdup == 0 ||
|
|
flag == DB_SET_RECNO || flag == DB_GET_BOTH ||
|
|
flag == DB_CONSUME || flag == DB_CONSUME_WAIT)) {
|
|
if (flag == DB_GET_BOTH) {
|
|
if (userecno) {
|
|
result = Tcl_GetIntFromObj(interp,
|
|
objv[(objc - 2)], &itmp);
|
|
recno = itmp;
|
|
if (result == TCL_OK) {
|
|
key.data = &recno;
|
|
key.size = sizeof(db_recno_t);
|
|
} else
|
|
return (result);
|
|
} else {
|
|
key.data =
|
|
Tcl_GetByteArrayFromObj(objv[objc-2],
|
|
&itmp);
|
|
key.size = itmp;
|
|
}
|
|
/*
|
|
* Already checked args above. Fill in key and save.
|
|
* Save is used in the dbp->get call below to fill in
|
|
* data.
|
|
*/
|
|
save.data =
|
|
Tcl_GetByteArrayFromObj(objv[objc-1], &itmp);
|
|
save.size = itmp;
|
|
} else if (flag != DB_CONSUME && flag != DB_CONSUME_WAIT) {
|
|
if (userecno) {
|
|
result = Tcl_GetIntFromObj(
|
|
interp, objv[(objc - 1)], &itmp);
|
|
recno = itmp;
|
|
if (result == TCL_OK) {
|
|
key.data = &recno;
|
|
key.size = sizeof(db_recno_t);
|
|
} else
|
|
return (result);
|
|
} else {
|
|
key.data = Tcl_GetByteArrayFromObj(objv[objc-1], &itmp);
|
|
key.size = itmp;
|
|
}
|
|
}
|
|
|
|
memset(&data, 0, sizeof(data));
|
|
data = save;
|
|
|
|
_debug_check();
|
|
|
|
ret = dbp->get(dbp, txn, &key, &data, flag | rmw);
|
|
result = _ReturnSetup(interp, 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 (type == DB_RECNO || type == DB_QUEUE)
|
|
result = _SetListRecnoElem(interp, retlist,
|
|
*(db_recno_t *)key.data, data.data,
|
|
data.size);
|
|
else
|
|
result = _SetListElem(interp, retlist,
|
|
key.data, key.size, data.data, data.size);
|
|
/*
|
|
* Free space from DB_DBT_MALLOC
|
|
*/
|
|
__os_free(data.data, data.size);
|
|
}
|
|
if (result == TCL_OK)
|
|
Tcl_SetObjResult(interp, retlist);
|
|
goto out;
|
|
}
|
|
|
|
if (userecno) {
|
|
result = Tcl_GetIntFromObj(interp, objv[(objc - 1)], &itmp);
|
|
recno = itmp;
|
|
if (result == TCL_OK) {
|
|
key.data = &recno;
|
|
key.size = sizeof(db_recno_t);
|
|
} else
|
|
return (result);
|
|
} else {
|
|
key.data = Tcl_GetByteArrayFromObj(objv[objc-1], &itmp);
|
|
key.size = itmp;
|
|
}
|
|
ret = dbp->cursor(dbp, txn, &dbc, 0);
|
|
result = _ReturnSetup(interp, 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 = 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;
|
|
_debug_check();
|
|
ret = dbc->c_get(dbc, &key, &data, cflag | rmw);
|
|
result = _ReturnSetup(interp, ret, "db get (cursor)");
|
|
if (result == TCL_ERROR)
|
|
goto out1;
|
|
if (pattern)
|
|
cflag = DB_NEXT;
|
|
else
|
|
cflag = DB_NEXT_DUP;
|
|
|
|
while (ret == 0 && result == TCL_OK) {
|
|
/*
|
|
* Build up our {name value} sublist
|
|
*/
|
|
result = _SetListElem(interp, retlist,
|
|
key.data, key.size,
|
|
data.data, data.size);
|
|
/*
|
|
* Free space from DB_DBT_MALLOC
|
|
*/
|
|
__os_free(data.data, data.size);
|
|
if (result != TCL_OK)
|
|
break;
|
|
/*
|
|
* Append {name value} to return list
|
|
*/
|
|
memset(&key, 0, sizeof(key));
|
|
memset(&data, 0, sizeof(data));
|
|
/*
|
|
* Restore any "partial" info we have saved.
|
|
*/
|
|
data = save;
|
|
ret = dbc->c_get(dbc, &key, &data, cflag | rmw);
|
|
if (ret == 0 && pattern &&
|
|
memcmp(key.data, prefix, strlen(prefix)) != 0) {
|
|
/*
|
|
* Free space from DB_DBT_MALLOC
|
|
*/
|
|
__os_free(data.data, data.size);
|
|
break;
|
|
}
|
|
}
|
|
dbc->c_close(dbc);
|
|
out1:
|
|
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(prefix,0);
|
|
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 char *dbdelopts[] = {
|
|
"-glob",
|
|
"-txn",
|
|
NULL
|
|
};
|
|
enum dbdelopts {
|
|
DBDEL_GLOB,
|
|
DBDEL_TXN
|
|
};
|
|
DBC *dbc;
|
|
DBT key, data;
|
|
DBTYPE type;
|
|
DB_TXN *txn;
|
|
db_recno_t recno;
|
|
int i, itmp, optindex, result, ret;
|
|
u_int32_t flag;
|
|
char *arg, *pattern, *prefix, msg[MSG_SIZE];
|
|
|
|
result = TCL_OK;
|
|
flag = 0;
|
|
pattern = prefix = NULL;
|
|
txn = NULL;
|
|
if (objc < 3) {
|
|
Tcl_WrongNumArgs(interp, 2, objv, "?-args? key");
|
|
return (TCL_ERROR);
|
|
}
|
|
|
|
memset(&key, 0, sizeof(key));
|
|
/*
|
|
* The first arg must be -txn, -glob 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;
|
|
}
|
|
if (result != TCL_OK)
|
|
break;
|
|
}
|
|
|
|
if (result != TCL_OK)
|
|
goto out;
|
|
|
|
/*
|
|
* If we have a pattern AND more keys to process, then there
|
|
* is an error. Either we have some number of exact keys,
|
|
* or we have a pattern.
|
|
*/
|
|
if (pattern != NULL && i != objc) {
|
|
Tcl_WrongNumArgs(interp, 2, objv, "?args? -glob pattern | key");
|
|
result = TCL_ERROR;
|
|
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 (pattern == NULL && i != (objc - 1)) {
|
|
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".
|
|
*/
|
|
type = dbp->get_type(dbp);
|
|
ret = 0;
|
|
while (i < objc && ret == 0) {
|
|
memset(&key, 0, sizeof(key));
|
|
if (type == DB_RECNO || type == DB_QUEUE) {
|
|
result = Tcl_GetIntFromObj(interp, objv[i++], &itmp);
|
|
recno = itmp;
|
|
if (result == TCL_OK) {
|
|
key.data = &recno;
|
|
key.size = sizeof(db_recno_t);
|
|
} else
|
|
return (result);
|
|
} else {
|
|
key.data = Tcl_GetByteArrayFromObj(objv[i++], &itmp);
|
|
key.size = itmp;
|
|
}
|
|
_debug_check();
|
|
ret = dbp->del(dbp, txn, &key, 0);
|
|
/*
|
|
* If we have any error, set up return result and stop
|
|
* processing keys.
|
|
*/
|
|
if (ret != 0)
|
|
break;
|
|
}
|
|
result = _ReturnSetup(interp, 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 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 = strlen(prefix);
|
|
if (strlen(prefix) == 0)
|
|
flag = DB_FIRST;
|
|
else
|
|
flag = DB_SET_RANGE;
|
|
ret = dbc->c_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->c_del(dbc, 0);
|
|
if (ret != 0) {
|
|
result = _ReturnSetup(interp, 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->c_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(prefix,0);
|
|
dbc->c_close(dbc);
|
|
result = _ReturnSetup(interp, ret, "db del");
|
|
}
|
|
out:
|
|
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 char *dbcuropts[] = {
|
|
"-txn", "-update",
|
|
NULL
|
|
};
|
|
enum dbcuropts {
|
|
DBCUR_TXN, DBCUR_UPDATE
|
|
};
|
|
DB_TXN *txn;
|
|
u_int32_t flag;
|
|
int i, optindex, result, ret;
|
|
char *arg, msg[MSG_SIZE];
|
|
|
|
result = TCL_OK;
|
|
flag = 0;
|
|
txn = NULL;
|
|
/*
|
|
* If the user asks for -glob or -recno, it MUST be the second
|
|
* last arg given. If it isn't given, then we must check if
|
|
* they gave us a correct key.
|
|
*/
|
|
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 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;
|
|
case DBCUR_UPDATE:
|
|
flag = DB_WRITECURSOR;
|
|
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_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 char *dbjopts[] = {
|
|
"-nosort",
|
|
NULL
|
|
};
|
|
enum dbjopts {
|
|
DBJ_NOSORT
|
|
};
|
|
DBC **listp;
|
|
u_int32_t flag;
|
|
int adj, i, j, optindex, size, 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);
|
|
}
|
|
|
|
i = 2;
|
|
adj = i;
|
|
while (i < objc) {
|
|
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;
|
|
}
|
|
i++;
|
|
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 *) * ((objc - adj) + 1);
|
|
ret = __os_malloc(dbp->dbenv, size, NULL, &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 join");
|
|
|
|
out:
|
|
__os_free(listp, size);
|
|
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 char *dbgetjopts[] = {
|
|
"-nosort",
|
|
"-txn",
|
|
NULL
|
|
};
|
|
enum dbgetjopts {
|
|
DBGETJ_NOSORT,
|
|
DBGETJ_TXN
|
|
};
|
|
DB_TXN *txn;
|
|
DB *elemdbp;
|
|
DBC **listp;
|
|
DBC *dbc;
|
|
DBT key, data;
|
|
Tcl_Obj **elemv, *retlist;
|
|
u_int32_t flag;
|
|
int adj, elemc, i, itmp, j, optindex, result, ret, size;
|
|
char *arg, msg[MSG_SIZE];
|
|
|
|
result = TCL_OK;
|
|
flag = 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) {
|
|
case DBGETJ_NOSORT:
|
|
flag |= DB_JOIN_NOSORT;
|
|
adj++;
|
|
break;
|
|
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 *) * ((objc - adj) + 1);
|
|
ret = __os_malloc(NULL, size, NULL, &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 cursor")) ==
|
|
TCL_ERROR)
|
|
goto out;
|
|
memset(&key, 0, sizeof(key));
|
|
memset(&data, 0, sizeof(data));
|
|
key.data = Tcl_GetByteArrayFromObj(elemv[elemc-1], &itmp);
|
|
key.size = itmp;
|
|
ret = (listp[j])->c_get(listp[j], &key, &data, DB_SET);
|
|
if ((result = _ReturnSetup(interp, ret, "db cget")) ==
|
|
TCL_ERROR)
|
|
goto out;
|
|
}
|
|
listp[j] = NULL;
|
|
_debug_check();
|
|
ret = dbp->join(dbp, listp, &dbc, flag);
|
|
result = _ReturnSetup(interp, 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->c_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_free(key.data, key.size);
|
|
__os_free(data.data, data.size);
|
|
}
|
|
}
|
|
dbc->c_close(dbc);
|
|
if (result == TCL_OK)
|
|
Tcl_SetObjResult(interp, retlist);
|
|
out:
|
|
while (j) {
|
|
if (listp[j])
|
|
(listp[j])->c_close(listp[j]);
|
|
j--;
|
|
}
|
|
__os_free(listp, size);
|
|
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 */
|
|
{
|
|
Tcl_Obj *res;
|
|
DBC *dbc;
|
|
DBT key, data;
|
|
db_recno_t count, recno;
|
|
int itmp, len, result, ret;
|
|
|
|
result = TCL_OK;
|
|
count = 0;
|
|
res = NULL;
|
|
if (objc != 3) {
|
|
Tcl_WrongNumArgs(interp, 2, objv, "key");
|
|
return (TCL_ERROR);
|
|
}
|
|
|
|
memset(&key, 0, sizeof(key));
|
|
|
|
/*
|
|
* 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.
|
|
*/
|
|
ret = 0;
|
|
memset(&key, 0, sizeof(key));
|
|
memset(&data, 0, sizeof(data));
|
|
/*
|
|
* If it's a 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_RECNO || dbp->type == DB_QUEUE) {
|
|
result = Tcl_GetIntFromObj(interp, objv[2], &itmp);
|
|
recno = itmp;
|
|
if (result == TCL_OK) {
|
|
key.data = &recno;
|
|
key.size = sizeof(db_recno_t);
|
|
} else
|
|
return (result);
|
|
} else {
|
|
key.data = Tcl_GetByteArrayFromObj(objv[2], &len);
|
|
key.size = len;
|
|
}
|
|
_debug_check();
|
|
ret = dbp->cursor(dbp, NULL, &dbc, 0);
|
|
if (ret != 0) {
|
|
result = _ReturnSetup(interp, ret, "db cursor");
|
|
goto out;
|
|
}
|
|
/*
|
|
* Move our cursor to the key.
|
|
*/
|
|
ret = dbc->c_get(dbc, &key, &data, DB_SET);
|
|
if (ret == DB_NOTFOUND)
|
|
count = 0;
|
|
else {
|
|
ret = dbc->c_count(dbc, &count, 0);
|
|
if (ret != 0) {
|
|
result = _ReturnSetup(interp, ret, "db cursor");
|
|
goto out;
|
|
}
|
|
}
|
|
res = Tcl_NewIntObj(count);
|
|
Tcl_SetObjResult(interp, res);
|
|
out:
|
|
return (result);
|
|
}
|
|
|
|
/*
|
|
* 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 char *dbkeyropts[] = {
|
|
"-txn",
|
|
NULL
|
|
};
|
|
enum dbkeyropts {
|
|
DBKEYR_TXN
|
|
};
|
|
DB_TXN *txn;
|
|
DB_KEY_RANGE range;
|
|
DBT key;
|
|
DBTYPE type;
|
|
Tcl_Obj *myobjv[3], *retlist;
|
|
db_recno_t recno;
|
|
u_int32_t flag;
|
|
int i, itmp, myobjc, optindex, result, ret;
|
|
char *arg, msg[MSG_SIZE];
|
|
|
|
result = TCL_OK;
|
|
flag = 0;
|
|
if (objc < 3) {
|
|
Tcl_WrongNumArgs(interp, 2, objv, "?-txn id? key");
|
|
return (TCL_ERROR);
|
|
}
|
|
|
|
txn = NULL;
|
|
i = 2;
|
|
while (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);
|
|
type = dbp->get_type(dbp);
|
|
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 = Tcl_GetIntFromObj(interp, objv[i], &itmp);
|
|
recno = itmp;
|
|
if (result == TCL_OK) {
|
|
key.data = &recno;
|
|
key.size = sizeof(db_recno_t);
|
|
} else
|
|
return (result);
|
|
} else {
|
|
key.data = Tcl_GetByteArrayFromObj(objv[i++], &itmp);
|
|
key.size = itmp;
|
|
}
|
|
_debug_check();
|
|
ret = dbp->key_range(dbp, txn, &key, &range, flag);
|
|
result = _ReturnSetup(interp, ret, "db join");
|
|
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:
|
|
return (result);
|
|
}
|