diff options
Diffstat (limited to 'src/third_party/wiredtiger/test/format/bdb.c')
| -rw-r--r-- | src/third_party/wiredtiger/test/format/bdb.c | 211 |
1 files changed, 211 insertions, 0 deletions
diff --git a/src/third_party/wiredtiger/test/format/bdb.c b/src/third_party/wiredtiger/test/format/bdb.c new file mode 100644 index 00000000000..fec23112549 --- /dev/null +++ b/src/third_party/wiredtiger/test/format/bdb.c @@ -0,0 +1,211 @@ +/*- + * Public Domain 2014-2015 MongoDB, Inc. + * Public Domain 2008-2014 WiredTiger, Inc. + * + * This is free and unencumbered software released into the public domain. + * + * Anyone is free to copy, modify, publish, use, compile, sell, or + * distribute this software, either in source code form or as a compiled + * binary, for any purpose, commercial or non-commercial, and by any + * means. + * + * In jurisdictions that recognize copyright laws, the author or authors + * of this software dedicate any and all copyright interest in the + * software to the public domain. We make this dedication for the benefit + * of the public at large and to the detriment of our heirs and + * successors. We intend this dedication to be an overt act of + * relinquishment in perpetuity of all present and future rights to this + * software under copyright law. + * + * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, + * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF + * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. + * IN NO EVENT SHALL THE AUTHORS BE LIABLE FOR ANY CLAIM, DAMAGES OR + * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, + * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR + * OTHER DEALINGS IN THE SOFTWARE. + */ + +#define BDB 1 /* Berkeley DB header files */ +#include "format.h" + +static DBT key, value; +static uint8_t *keybuf; + +static int +bdb_compare_reverse(DB *dbp, const DBT *k1, const DBT *k2 +#if DB_VERSION_MAJOR >= 6 + , size_t *locp +#endif +) +{ + size_t len; + int cmp; + + (void)(dbp); +#if DB_VERSION_MAJOR >= 6 + (void)(locp); +#endif + + len = (k1->size < k2->size) ? k1->size : k2->size; + if ((cmp = memcmp(k2->data, k1->data, len)) == 0) + cmp = ((int)k1->size - (int)k2->size); + return (cmp); +} + +void +bdb_open(void) +{ + DB *db; + DBC *dbc; + DB_ENV *dbenv; + + assert(db_env_create(&dbenv, 0) == 0); + dbenv->set_errpfx(dbenv, "bdb"); + dbenv->set_errfile(dbenv, stderr); + assert(dbenv->mutex_set_max(dbenv, 10000) == 0); + assert(dbenv->set_cachesize(dbenv, 0, 50 * 1024 * 1024, 1) == 0); + assert(dbenv->open(dbenv, NULL, + DB_CREATE | DB_INIT_LOCK | DB_INIT_MPOOL | DB_PRIVATE, 0) == 0); + assert(db_create(&db, dbenv, 0) == 0); + + if (g.type == ROW && g.c_reverse) + assert(db->set_bt_compare(db, bdb_compare_reverse) == 0); + + assert(db->open( + db, NULL, g.home_bdb, NULL, DB_BTREE, DB_CREATE, 0) == 0); + g.bdb = db; + assert(db->cursor(db, NULL, &dbc, 0) == 0); + g.dbc = dbc; + + key_gen_setup(&keybuf); +} + +void +bdb_close(void) +{ + DB *db; + DBC *dbc; + DB_ENV *dbenv; + + dbc = g.dbc; + db = g.bdb; + dbenv = db->dbenv; + assert(dbc->close(dbc) == 0); + assert(db->close(db, 0) == 0); + assert(dbenv->close(dbenv, 0) == 0); + + free(keybuf); + keybuf = NULL; +} + +void +bdb_insert( + const void *key_data, size_t key_size, + const void *value_data, size_t value_size) +{ + DBC *dbc; + + key.data = (void *)key_data; + key.size = (uint32_t)key_size; + value.data = (void *)value_data; + value.size = (uint32_t)value_size; + + dbc = g.dbc; + + assert(dbc->put(dbc, &key, &value, DB_KEYFIRST) == 0); +} + +void +bdb_np(int next, + void *keyp, size_t *keysizep, + void *valuep, size_t *valuesizep, int *notfoundp) +{ + DBC *dbc = g.dbc; + int ret; + + *notfoundp = 0; + if ((ret = + dbc->get(dbc, &key, &value, next ? DB_NEXT : DB_PREV)) != 0) { + if (ret != DB_NOTFOUND) + die(ret, "dbc.get: %s: {%.*s}", + next ? "DB_NEXT" : "DB_PREV", + (int)key.size, (char *)key.data); + *notfoundp = 1; + } else { + *(void **)keyp = key.data; + *keysizep = key.size; + *(void **)valuep = value.data; + *valuesizep = value.size; + } +} + +void +bdb_read(uint64_t keyno, void *valuep, size_t *valuesizep, int *notfoundp) +{ + DBC *dbc = g.dbc; + size_t size; + int ret; + + key_gen(keybuf, &size, keyno); + key.data = keybuf; + key.size = (uint32_t)size; + + *notfoundp = 0; + if ((ret = dbc->get(dbc, &key, &value, DB_SET)) != 0) { + if (ret != DB_NOTFOUND) + die(ret, "dbc.get: DB_SET: {%.*s}", + (int)key.size, (char *)key.data); + *notfoundp = 1; + } else { + *(void **)valuep = value.data; + *valuesizep = value.size; + } +} + +void +bdb_update(const void *arg_key, size_t arg_key_size, + const void *arg_value, size_t arg_value_size, int *notfoundp) +{ + DBC *dbc = g.dbc; + int ret; + + key.data = (void *)arg_key; + key.size = (uint32_t)arg_key_size; + value.data = (void *)arg_value; + value.size = (uint32_t)arg_value_size; + + *notfoundp = 0; + if ((ret = dbc->put(dbc, &key, &value, DB_KEYFIRST)) != 0) { + if (ret != DB_NOTFOUND) { + die(ret, "dbc.put: DB_KEYFIRST: {%.*s}{%.*s}", + (int)key.size, (char *)key.data, + (int)value.size, (char *)value.data); + } + *notfoundp = 1; + } +} + +void +bdb_remove(uint64_t keyno, int *notfoundp) +{ + DBC *dbc = g.dbc; + size_t size; + int ret; + + key_gen(keybuf, &size, keyno); + key.data = keybuf; + key.size = (uint32_t)size; + + bdb_read(keyno, &value.data, &size, notfoundp); + value.size = (uint32_t)size; + if (*notfoundp) + return; + + if ((ret = dbc->del(dbc, 0)) != 0) { + if (ret != DB_NOTFOUND) + die(ret, "dbc.del: {%.*s}", + (int)key.size, (char *)key.data); + *notfoundp = 1; + } +} |