ydb.c 213 KB
Newer Older
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
1
/* -*- mode: C; c-basic-offset: 4 -*- */
2
#ident "Copyright (c) 2007-2009 Tokutek Inc.  All rights reserved."
Zardosht Kasheff's avatar
Zardosht Kasheff committed
3
 
4 5 6
#ident "The technology is licensed by the Massachusetts Institute of Technology, Rutgers State University of New Jersey, and the Research Foundation of State University of New York at Stony Brook under United States of America Serial No. 11/760379 and to the patents and/or patent applications resulting from it."

const char *toku_patent_string = "The technology is licensed by the Massachusetts Institute of Technology, Rutgers State University of New Jersey, and the Research Foundation of State University of New York at Stony Brook under United States of America Serial No. 11/760379 and to the patents and/or patent applications resulting from it.";
7
const char *toku_copyright_string = "Copyright (c) 2007-2009 Tokutek Inc.  All rights reserved.";
8

9
#include <toku_portability.h>
10
#include <toku_pthread.h>
11
#include <ctype.h>
Bradley C. Kuszmaul's avatar
Bradley C. Kuszmaul committed
12 13
#include <errno.h>
#include <limits.h>
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
14 15 16
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
17
#include <fcntl.h>
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
18
#include <sys/stat.h>
Bradley C. Kuszmaul's avatar
Bradley C. Kuszmaul committed
19 20
#include <sys/types.h>
#include <unistd.h>
21
#include <db.h>
22
#include "toku_assert.h"
23
#include "ydb.h"
24
#include "ydb-internal.h"
25
#include "brt-internal.h"
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
26
#include "cachetable.h"
Bradley C. Kuszmaul's avatar
Bradley C. Kuszmaul committed
27 28
#include "log.h"
#include "memory.h"
29
#include "dlmalloc.h"
30
#include "checkpoint.h"
31
#include "key.h"
32
#include "loader.h"
33
#include "indexer.h"
34
#include "ydb_load.h"
35
#include "brtloader.h"
Yoni Fogel's avatar
Yoni Fogel committed
36

37 38 39 40 41 42 43 44 45 46
#ifdef TOKUTRACE
 #define DB_ENV_CREATE_FUN db_env_create_toku10
 #define DB_CREATE_FUN db_create_toku10
#else
 #define DB_ENV_CREATE_FUN db_env_create
 #define DB_CREATE_FUN db_create
 int toku_set_trace_file (char *fname __attribute__((__unused__))) { return 0; }
 int toku_close_trace_file (void) { return 0; } 
#endif

47
#define DB_ISOLATION_FLAGS (DB_READ_COMMITTED | DB_READ_UNCOMMITTED | DB_TXN_SNAPSHOT | DB_SERIALIZABLE | DB_INHERIT_ISOLATION)
48 49 50

// Accountability: operation counters available for debugging and for "show engine status"
static u_int64_t num_inserts;
51
static u_int64_t num_inserts_fail;
52
static u_int64_t num_deletes;
53
static u_int64_t num_deletes_fail;
54 55 56 57 58 59 60 61
static u_int64_t num_updates;
static u_int64_t num_updates_fail;
static u_int64_t num_multi_inserts;
static u_int64_t num_multi_inserts_fail;
static u_int64_t num_multi_deletes;
static u_int64_t num_multi_deletes_fail;
static u_int64_t num_multi_updates;
static u_int64_t num_multi_updates_fail;
62 63
static u_int64_t num_point_queries;
static u_int64_t num_sequential_queries;
64 65 66
static u_int64_t logsuppress;                // number of times logs are suppressed for empty table (2440)
static u_int64_t logsuppressfail;            // number of times unable to suppress logs for empty table (2440)
static time_t    startuptime;                // timestamp of system startup
67 68 69 70 71 72 73 74
static DB_ENV *  most_recent_env;            // most recently opened env, used for engine status on crash

static void
init_status_info(void) {
    num_inserts = 0;
    num_inserts_fail = 0;
    num_deletes = 0;
    num_deletes_fail = 0;
75 76 77 78 79 80 81 82
    num_updates = 0;
    num_updates_fail = 0;
    num_multi_inserts = 0;
    num_multi_inserts_fail = 0;
    num_multi_deletes = 0;
    num_multi_deletes_fail = 0;
    num_multi_updates = 0;
    num_multi_updates_fail = 0;
83 84 85 86 87 88
    num_point_queries = 0;
    num_sequential_queries = 0;
    logsuppress = 0;
    logsuppressfail = 0;
    startuptime = time(NULL);
}
89

90 91 92
const char * environmentdictionary = "tokudb.environment";
const char * fileopsdirectory = "tokudb.directory";

93
static int env_get_iname(DB_ENV* env, DBT* dname_dbt, DBT* iname_dbt);
94

Yoni Fogel's avatar
Yoni Fogel committed
95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125
static const char single_process_lock_file[] = "/__tokudb_lock_dont_delete_me_";

static int
single_process_lock(const char *lock_dir, const char *which, int *lockfd) {
    if (!lock_dir)
        return ENOENT;
    int namelen=strlen(lock_dir)+strlen(which);
    char lockfname[namelen+sizeof(single_process_lock_file)];

    int l = snprintf(lockfname, sizeof(lockfname), "%s%s%s", lock_dir, single_process_lock_file, which);
    assert(l+1 == (signed)(sizeof(lockfname)));
    *lockfd = toku_os_lock_file(lockfname);
    if (*lockfd < 0) {
        int e = errno;
        fprintf(stderr, "Couldn't start tokudb because some other tokudb process is using the same directory [%s] for [%s]\n", lock_dir, which);
        return e;
    }
    return 0;
}

static int
single_process_unlock(int *lockfd) {
    int fd = *lockfd;
    *lockfd = -1;
    if (fd>=0) {
        int r = toku_os_unlock_file(fd);
        if (r != 0)
            return errno;
    }
    return 0;
}
126

Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
127
/** The default maximum number of persistent locks in a lock tree  */
128
const u_int32_t __toku_env_default_max_locks = 0x7FFFFFFF;
129
const uint64_t __toku_env_default_max_lock_memory = 1000*1024;
Rich Prohaska's avatar
Rich Prohaska committed
130

131 132 133 134 135 136 137
static inline DBT*
init_dbt_realloc(DBT *dbt) {
    memset(dbt, 0, sizeof(*dbt));
    dbt->flags = DB_DBT_REALLOC;
    return dbt;
}

138 139 140 141 142 143
//Callback used for redirecting dictionaries.
static void
ydb_set_brt(DB *db, BRT brt) {
    db->i->brt = brt;
}

144 145
int 
toku_ydb_init(void) {
Yoni Fogel's avatar
Yoni Fogel committed
146 147 148
    int r = 0;
    //Lower level must be initialized first.
    if (r==0) 
149
        r = toku_brt_init(toku_ydb_lock, toku_ydb_unlock, ydb_set_brt);
Yoni Fogel's avatar
Yoni Fogel committed
150 151 152
    if (r==0) 
        r = toku_ydb_lock_init();
    return r;
153 154
}

155 156
int 
toku_ydb_destroy(void) {
Yoni Fogel's avatar
Yoni Fogel committed
157 158 159 160 161 162 163
    int r = 0;
    if (r==0)
        r = toku_ydb_lock_destroy();
    //Lower level must be cleaned up last.
    if (r==0)
        r = toku_brt_destroy();
    return r;
164 165
}

166 167 168 169 170
static int
ydb_getf_do_nothing(DBT const* UU(key), DBT const* UU(val), void* UU(extra)) {
    return 0;
}

Rich Prohaska's avatar
Rich Prohaska committed
171 172
/* env methods */
static int toku_env_close(DB_ENV *env, u_int32_t flags);
173 174 175
static int toku_env_set_data_dir(DB_ENV * env, const char *dir);
static int toku_env_set_lg_dir(DB_ENV * env, const char *dir);
static int toku_env_set_tmp_dir(DB_ENV * env, const char *tmp_dir);
Rich Prohaska's avatar
Rich Prohaska committed
176

177 178
static inline int 
env_opened(DB_ENV *env) {
179
    return env->i->cachetable != 0;
Rich Prohaska's avatar
Rich Prohaska committed
180 181
}

182 183
static void 
env_init_open_txn(DB_ENV *env) {
184
    toku_list_init(&env->i->open_txns);
Rich Prohaska's avatar
Rich Prohaska committed
185 186
}

187
// add a txn to the list of open txn's
188 189
static void 
env_add_open_txn(DB_ENV *env, DB_TXN *txn) {
190
    toku_list_push(&env->i->open_txns, (struct toku_list *) (void *) &txn->open_txns);
191 192 193
}

// remove a txn from the list of open txn's
194 195
static void 
env_remove_open_txn(DB_ENV *UU(env), DB_TXN *txn) {
196
    toku_list_remove((struct toku_list *) (void *) &txn->open_txns);
Rich Prohaska's avatar
Rich Prohaska committed
197 198
}

199
static int toku_txn_abort(DB_TXN * txn, TXN_PROGRESS_POLL_FUNCTION, void*);
200

201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233
static void
env_fs_report_in_yellow(DB_ENV *UU(env)) {
    char tbuf[26];
    time_t tnow = time(NULL);
    fprintf(stderr, "%.24s Tokudb file system space is low\n", ctime_r(&tnow, tbuf)); fflush(stderr);
}

static void
env_fs_report_in_red(DB_ENV *UU(env)) {
    char tbuf[26];
    time_t tnow = time(NULL);
    fprintf(stderr, "%.24s Tokudb file system space is really low and access is restricted\n", ctime_r(&tnow, tbuf)); fflush(stderr);
}

static inline uint64_t
env_fs_redzone(DB_ENV *env, uint64_t total) {
    return total * env->i->redzone / 100;
}

#define ZONEREPORTLIMIT 12
// Check the available space in the file systems used by tokudb and erect barriers when available space gets low.
static int
env_fs_poller(void *arg) {
    if (0) printf("%s:%d %p\n", __FUNCTION__, __LINE__, arg);
 
    DB_ENV *env = (DB_ENV *) arg;
    int r;
#if 0
    // get the cachetable size limit (not yet needed)
    uint64_t cs = toku_cachetable_get_size_limit(env->i->cachetable);
#endif

    int in_yellow; // set true to issue warning to user
234
    int in_red;    // set true to prevent certain operations (returning ENOSPC)
235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336

    // get the fs sizes for the home dir
    uint64_t avail_size, total_size;
    r = toku_get_filesystem_sizes(env->i->dir, &avail_size, NULL, &total_size);
    assert(r == 0);
    if (0) fprintf(stderr, "%s %"PRIu64" %"PRIu64"\n", env->i->dir, avail_size, total_size);
    in_yellow = (avail_size < 2 * env_fs_redzone(env, total_size));
    in_red = (avail_size < env_fs_redzone(env, total_size));
    
    // get the fs sizes for the data dir if different than the home dir
    if (strcmp(env->i->dir, env->i->real_data_dir) != 0) {
        r = toku_get_filesystem_sizes(env->i->real_data_dir, &avail_size, NULL, &total_size);
        assert(r == 0);
        if (0) fprintf(stderr, "%s %"PRIu64" %"PRIu64"\n", env->i->real_data_dir, avail_size, total_size);
        in_yellow += (avail_size < 2 * env_fs_redzone(env, total_size));
        in_red += (avail_size < env_fs_redzone(env, total_size));
    }

    // get the fs sizes for the log dir if different than the home dir and data dir
    if (strcmp(env->i->dir, env->i->real_log_dir) != 0 && strcmp(env->i->real_data_dir, env->i->real_log_dir) != 0) {
        r = toku_get_filesystem_sizes(env->i->real_log_dir, &avail_size, NULL, &total_size);
        assert(r == 0);
        if (0) fprintf(stderr, "%s %"PRIu64" %"PRIu64"\n", env->i->real_log_dir, avail_size, total_size);
        in_yellow += (avail_size < 2 * env_fs_redzone(env, total_size));
        in_red += (avail_size < env_fs_redzone(env, total_size));
    }

    
    env->i->fs_seq++;                    // how many times through this polling loop?
    uint64_t now = env->i->fs_seq;

    // Don't issue report if we have not been out of this fs_state for a while, unless we're at system startup
    switch (env->i->fs_state) {
    case FS_RED:
	if (!in_red) {
	    if (in_yellow) {
		env->i->fs_state = FS_YELLOW;
	    } else {
		env->i->fs_state = FS_GREEN;
	    }
	}
        break;
    case FS_YELLOW:
        if (in_red) {
	    if ((now - env->i->last_seq_entered_red > ZONEREPORTLIMIT) || (now < ZONEREPORTLIMIT))
		env_fs_report_in_red(env);
            env->i->fs_state = FS_RED;
	    env->i->last_seq_entered_red = now;
        } else if (!in_yellow) {
            env->i->fs_state = FS_GREEN;
        }
        break;
    case FS_GREEN:
        if (in_red) {
	    if ((now - env->i->last_seq_entered_red > ZONEREPORTLIMIT) || (now < ZONEREPORTLIMIT))
		env_fs_report_in_red(env);
            env->i->fs_state = FS_RED;
	    env->i->last_seq_entered_red = now;
        } else if (in_yellow) {
	    if ((now - env->i->last_seq_entered_yellow > ZONEREPORTLIMIT) || (now < ZONEREPORTLIMIT))
		env_fs_report_in_yellow(env);
            env->i->fs_state = FS_YELLOW;
	    env->i->last_seq_entered_yellow = now;
        }
        break;
    }
    return 0;
}
#undef ZONEREPORTLIMIT

static void
env_fs_init(DB_ENV *env) {
    env->i->fs_state = FS_GREEN;
    env->i->fs_poll_time = 5;  // seconds
    env->i->redzone = 5;       // percent of total space
    env->i->fs_poller_is_init = FALSE;
}

// Initialize the minicron that polls file system space
static int
env_fs_init_minicron(DB_ENV *env) {
    int r = toku_minicron_setup(&env->i->fs_poller, env->i->fs_poll_time, env_fs_poller, env); 
    assert(r == 0);
    env->i->fs_poller_is_init = TRUE;
    return r;
}

// Destroy the file system space minicron
static void
env_fs_destroy(DB_ENV *env) {
    if (env->i->fs_poller_is_init) {
        int r = toku_minicron_shutdown(&env->i->fs_poller);
        assert(r == 0);
        env->i->fs_poller_is_init = FALSE;
    }
}

// Check if the available file system space is less than the reserve
// Returns ENOSPC if not enough space, othersize 0
static inline int 
env_check_avail_fs_space(DB_ENV *env) {
    int r = env->i->fs_state == FS_RED ? ENOSPC : 0; 
337
    if (r) env->i->enospc_redzone_ctr++;
338 339 340
    return r;
}

341 342 343 344 345 346
int
toku_ydb_check_avail_fs_space(DB_ENV *env) {
    int rval = env_check_avail_fs_space(env);
    return rval;
}

Rich Prohaska's avatar
Rich Prohaska committed
347 348
/* db methods */
static inline int db_opened(DB *db) {
349
    return db->i->opened != 0;
Rich Prohaska's avatar
Rich Prohaska committed
350 351
}

Zardosht Kasheff's avatar
Zardosht Kasheff committed
352

Rich Prohaska's avatar
Rich Prohaska committed
353 354
static int toku_db_put(DB * db, DB_TXN * txn, DBT * key, DBT * data, u_int32_t flags);
static int toku_db_get (DB * db, DB_TXN * txn, DBT * key, DBT * data, u_int32_t flags);
355
static int toku_db_cursor(DB *db, DB_TXN * txn, DBC **c, u_int32_t flags, int is_temporary_cursor);
Rich Prohaska's avatar
Rich Prohaska committed
356 357 358

/* txn methods */

359
/* lightweight cursor methods. */
360 361 362
static int toku_c_getf_first(DBC *c, u_int32_t flag, YDB_CALLBACK_FUNCTION f, void *extra);

static int toku_c_getf_last(DBC *c, u_int32_t flag, YDB_CALLBACK_FUNCTION f, void *extra);
Yoni Fogel's avatar
Yoni Fogel committed
363

364
static int toku_c_getf_next(DBC *c, u_int32_t flag, YDB_CALLBACK_FUNCTION f, void *extra);
Yoni Fogel's avatar
Yoni Fogel committed
365

366
static int toku_c_getf_prev(DBC *c, u_int32_t flag, YDB_CALLBACK_FUNCTION f, void *extra);
Yoni Fogel's avatar
Yoni Fogel committed
367

368 369
static int toku_c_getf_current(DBC *c, u_int32_t flag, YDB_CALLBACK_FUNCTION f, void *extra);
static int toku_c_getf_current_binding(DBC *c, u_int32_t flag, YDB_CALLBACK_FUNCTION f, void *extra);
Yoni Fogel's avatar
Yoni Fogel committed
370

371 372
static int toku_c_getf_set(DBC *c, u_int32_t flag, DBT *key, YDB_CALLBACK_FUNCTION f, void *extra);
static int toku_c_getf_set_range(DBC *c, u_int32_t flag, DBT *key, YDB_CALLBACK_FUNCTION f, void *extra);
373
static int toku_c_getf_set_range_reverse(DBC *c, u_int32_t flag, DBT *key, YDB_CALLBACK_FUNCTION f, void *extra);
374

Yoni Fogel's avatar
Yoni Fogel committed
375 376
// Effect: Lightweight cursor get

Rich Prohaska's avatar
Rich Prohaska committed
377 378 379 380 381
/* cursor methods */
static int toku_c_get(DBC * c, DBT * key, DBT * data, u_int32_t flag);
static int toku_c_del(DBC *c, u_int32_t flags);
static int toku_c_count(DBC *cursor, db_recno_t *count, u_int32_t flags);
static int toku_c_close(DBC * c);
Bradley C. Kuszmaul's avatar
Bradley C. Kuszmaul committed
382

383
static void
384 385 386
env_setup_real_dir(DB_ENV *env, char **real_dir, const char *nominal_dir) {
    toku_free(*real_dir);
    *real_dir = NULL;
387 388

    assert(env->i->dir);
389 390
    if (nominal_dir) 
	*real_dir = toku_construct_full_name(2, env->i->dir, nominal_dir);
391
    else
392 393 394 395 396 397
        *real_dir = toku_strdup(env->i->dir);
}

static void
env_setup_real_data_dir(DB_ENV *env) {
    env_setup_real_dir(env, &env->i->real_data_dir, env->i->data_dir);
398 399 400 401
}

static void
env_setup_real_log_dir(DB_ENV *env) {
402
    env_setup_real_dir(env, &env->i->real_log_dir, env->i->lg_dir);
403 404
}

405 406 407 408 409
static void
env_setup_real_tmp_dir(DB_ENV *env) {
    env_setup_real_dir(env, &env->i->real_tmp_dir, env->i->tmp_dir);
}

410 411
static int 
ydb_do_recovery (DB_ENV *env) {
412
    assert(env->i->real_log_dir);
413
    toku_ydb_unlock();
414
    int r = tokudb_recover(env->i->dir, env->i->real_log_dir, env->i->bt_compare,
415
                           env->i->generate_row_for_put, env->i->generate_row_for_del,
416
                           env->i->cachetable_size);
417
    toku_ydb_lock();
418
    return r;
419 420
}

421 422
static int 
needs_recovery (DB_ENV *env) {
423 424
    assert(env->i->real_log_dir);
    int recovery_needed = tokudb_needs_recovery(env->i->real_log_dir, TRUE);
425
    return recovery_needed ? DB_RUNRECOVERY : 0;
426 427
}

428 429 430 431
static int toku_db_create(DB ** db, DB_ENV * env, u_int32_t flags);
static int toku_db_open(DB * db, DB_TXN * txn, const char *fname, const char *dbname, DBTYPE dbtype, u_int32_t flags, int mode);
static int toku_env_txn_checkpoint(DB_ENV * env, u_int32_t kbyte, u_int32_t min, u_int32_t flags);
static int toku_db_close(DB * db, u_int32_t flags);
432
static int toku_txn_begin(DB_ENV *env, DB_TXN * stxn, DB_TXN ** txn, u_int32_t flags, int internal);
433
static int toku_txn_commit(DB_TXN * txn, u_int32_t flags, TXN_PROGRESS_POLL_FUNCTION, void*);
434 435
static int db_open_iname(DB * db, DB_TXN * txn, const char *iname, u_int32_t flags, int mode);

436
static void finalize_file_removal(DICTIONARY_ID dict_id, void * extra);
437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460

// Instruct db to use the default (built-in) key comparison function
// by setting the flag bits in the db and brt structs
static int
db_use_builtin_key_cmp(DB *db) {
    HANDLE_PANICKED_DB(db);
    int r;
    if (db_opened(db))
        r = toku_ydb_do_error(db->dbenv, EINVAL, "Comparison functions cannot be set after DB open.\n");
    else if (db->i->key_compare_was_set)
        r = toku_ydb_do_error(db->dbenv, EINVAL, "Key comparison function already set.\n");
    else {
        u_int32_t tflags;
        r = toku_brt_get_flags(db->i->brt, &tflags);
        if (r!=0) return r;

        tflags |= TOKU_DB_KEYCMP_BUILTIN;
        r = toku_brt_set_flags(db->i->brt, tflags);
        if (!r)
            db->i->key_compare_was_set = TRUE;
    }
    return r;
}

461 462
// Keys used in persistent environment dictionary:
// Following keys added in version 12
463
static const char * orig_env_ver_key = "original_version";
464 465 466 467
static const char * curr_env_ver_key = "current_version";  
// Following keys added in version 13
static const char * creation_time_key = "creation_time";
static const char * last_lsn_of_v12_key = "last_lsn_of_v12";
468
static const char * upgrade_v13_time_key = "upgrade_v13_time";  // Add more keys for future upgrades
469 470 471 472 473 474 475

// Values read from (or written into) persistent environment,
// kept here for read-only access from engine status.
static uint32_t persistent_original_env_version;
static uint32_t persistent_stored_env_version_at_startup;    // read from curr_env_ver_key, prev version as of this startup
static time_t   persistent_creation_time;
static uint64_t persistent_last_lsn_of_v12;
476
static time_t   persistent_upgrade_v13_time;
477 478 479 480

// Requires: persistent environment dictionary is already open.
// Input arg is lsn of clean shutdown of previous version,
// or ZERO_LSN if no upgrade or if crash between log upgrade and here.
481
static int
482
maybe_upgrade_persistent_environment_dictionary(DB_ENV * env, DB_TXN * txn, LSN last_lsn_of_clean_shutdown_read_from_log) {
483 484
    int r;
    DBT key, val;
485
    DB *persistent_environment = env->i->persistent_environment;
486

487
    toku_fill_dbt(&key, curr_env_ver_key, strlen(curr_env_ver_key));
488
    toku_init_dbt(&val);
489
    r = toku_db_get(persistent_environment, txn, &key, &val, 0);
490
    assert(r == 0);
491 492
    uint32_t stored_env_version = toku_dtoh32(*(uint32_t*)val.data);
    persistent_stored_env_version_at_startup = stored_env_version;
493
    if (stored_env_version > BRT_LAYOUT_VERSION)
494
	r = TOKUDB_DICTIONARY_TOO_NEW;
495 496 497
    else if (stored_env_version < BRT_LAYOUT_MIN_SUPPORTED_VERSION)
	r = TOKUDB_DICTIONARY_TOO_OLD;
    else if (stored_env_version < BRT_LAYOUT_VERSION) {
498
        const uint32_t curr_env_ver_d = toku_htod32(BRT_LAYOUT_VERSION);
499
        toku_fill_dbt(&key, curr_env_ver_key, strlen(curr_env_ver_key));
500 501 502 503 504 505 506 507 508 509
        toku_fill_dbt(&val, &curr_env_ver_d, sizeof(curr_env_ver_d));
        r = toku_db_put(persistent_environment, txn, &key, &val, DB_YESOVERWRITE);
        assert(r==0);
	
	uint64_t last_lsn_of_v12_d = toku_htod64(last_lsn_of_clean_shutdown_read_from_log.lsn);
	toku_fill_dbt(&key, last_lsn_of_v12_key, strlen(last_lsn_of_v12_key));
	toku_fill_dbt(&val, &last_lsn_of_v12_d, sizeof(last_lsn_of_v12_d));
	r = toku_db_put(persistent_environment, txn, &key, &val, DB_YESOVERWRITE);
        assert(r==0);
	
510 511 512
	time_t upgrade_v13_time_d = toku_htod64(time(NULL));
	toku_fill_dbt(&key, upgrade_v13_time_key, strlen(upgrade_v13_time_key));
	toku_fill_dbt(&val, &upgrade_v13_time_d, sizeof(upgrade_v13_time_d));
513
	r = toku_db_put(persistent_environment, txn, &key, &val, DB_NOOVERWRITE);
514 515
        assert(r==0);
    }
516
    return r;
517 518
}

519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558

// Capture persistent env contents to be read by engine status
static void
capture_persistent_env (DB_ENV * env, DB_TXN * txn) {
    int r;
    DBT key, val;
    DB *persistent_environment = env->i->persistent_environment;

    toku_fill_dbt(&key, curr_env_ver_key, strlen(curr_env_ver_key));
    toku_init_dbt(&val);
    r = toku_db_get(persistent_environment, txn, &key, &val, 0);
    assert(r == 0);
    uint32_t curr_env_version = toku_dtoh32(*(uint32_t*)val.data);
    assert(curr_env_version == BRT_LAYOUT_VERSION);

    toku_fill_dbt(&key, orig_env_ver_key, strlen(orig_env_ver_key));
    toku_init_dbt(&val);
    r = toku_db_get(persistent_environment, txn, &key, &val, 0);
    assert(r == 0);
    persistent_original_env_version = toku_dtoh32(*(uint32_t*)val.data);
    assert(persistent_original_env_version <= curr_env_version);

    // make no assertions about timestamps, clock may have been reset
    if (persistent_original_env_version >= BRT_LAYOUT_VERSION_13) {
	toku_fill_dbt(&key, creation_time_key, strlen(creation_time_key));
	toku_init_dbt(&val);
	r = toku_db_get(persistent_environment, txn, &key, &val, 0);
	assert(r == 0);
	persistent_creation_time = toku_dtoh64((*(time_t*)val.data));
    }

    if (persistent_original_env_version != curr_env_version) {
	// an upgrade was performed at some time, capture info about the upgrade
	
	toku_fill_dbt(&key, last_lsn_of_v12_key, strlen(last_lsn_of_v12_key));
	toku_init_dbt(&val);
	r = toku_db_get(persistent_environment, txn, &key, &val, 0);
	assert(r == 0);
	persistent_last_lsn_of_v12 = toku_dtoh64(*(uint32_t*)val.data);

559
	toku_fill_dbt(&key, upgrade_v13_time_key, strlen(upgrade_v13_time_key));
560 561 562
	toku_init_dbt(&val);
	r = toku_db_get(persistent_environment, txn, &key, &val, 0);
	assert(r == 0);
563
	persistent_upgrade_v13_time = toku_dtoh64((*(time_t*)val.data));
564 565 566 567 568 569 570
    }

}




571 572 573
// return 0 if log exists or ENOENT if log does not exist
static int
ydb_recover_log_exists(DB_ENV *env) {
574
    int r = tokudb_recover_log_exists(env->i->real_log_dir);
575 576 577 578 579
    return r;
}


// Validate that all required files are present, no side effects.
580 581
// Return 0 if all is well, ENOENT if some files are present but at least one is missing, 
// other non-zero value if some other error occurs.
582 583 584
// Set *valid_newenv if creating a new environment (all files missing).
// (Note, if special dictionaries exist, then they were created transactionally and log should exist.)
static int 
585
validate_env(DB_ENV * env, BOOL * valid_newenv, BOOL need_rollback_cachefile) {
586
    int r;
587
    BOOL expect_newenv = FALSE;        // set true if we expect to create a new env
588 589
    toku_struct_stat buf;
    char* path = NULL;
590

591
    // Test for persistent environment
Yoni Fogel's avatar
Yoni Fogel committed
592
    path = toku_construct_full_name(2, env->i->dir, environmentdictionary);
593 594
    assert(path);
    r = toku_stat(path, &buf);
595
    int stat_errno = errno;
596
    toku_free(path);
597 598 599
    if (r == 0) {
	expect_newenv = FALSE;  // persistent info exists
    }
600
    else if (stat_errno == ENOENT) {
601 602
	expect_newenv = TRUE;
	r = 0;
603 604
    }
    else {
605 606
	r = toku_ydb_do_error(env, errno, "Unable to access persistent environment\n");
	assert(r);
607
    }
608

609
    // Test for existence of rollback cachefile if it is expected to exist
610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626
    if (r == 0 && need_rollback_cachefile) {
	path = toku_construct_full_name(2, env->i->dir, ROLLBACK_CACHEFILE_NAME);
	assert(path);
	r = toku_stat(path, &buf);
	stat_errno = errno;
	toku_free(path);
	if (r == 0) {  
	    if (expect_newenv)  // rollback cachefile exists, but persistent env is missing
		r = toku_ydb_do_error(env, ENOENT, "Persistent environment is missing\n");
	}
	else if (stat_errno == ENOENT) {
	    if (!expect_newenv)  // rollback cachefile is missing but persistent env exists
		r = toku_ydb_do_error(env, ENOENT, "rollback cachefile directory is missing\n");
	    else 
		r = 0;           // both rollback cachefile and persistent env are missing
	}
	else {
627
	    r = toku_ydb_do_error(env, stat_errno, "Unable to access rollback cachefile\n");
628 629 630 631
	    assert(r);
	}
    }

632 633
    // Test for fileops directory
    if (r == 0) {
Yoni Fogel's avatar
Yoni Fogel committed
634
	path = toku_construct_full_name(2, env->i->dir, fileopsdirectory);
635 636
	assert(path);
	r = toku_stat(path, &buf);
637
	stat_errno = errno;
638 639 640 641 642
	toku_free(path);
	if (r == 0) {  
	    if (expect_newenv)  // fileops directory exists, but persistent env is missing
		r = toku_ydb_do_error(env, ENOENT, "Persistent environment is missing\n");
	}
643
	else if (stat_errno == ENOENT) {
644 645 646 647 648 649
	    if (!expect_newenv)  // fileops directory is missing but persistent env exists
		r = toku_ydb_do_error(env, ENOENT, "Fileops directory is missing\n");
	    else 
		r = 0;           // both fileops directory and persistent env are missing
	}
	else {
650
	    r = toku_ydb_do_error(env, stat_errno, "Unable to access fileops directory\n");
651 652 653 654 655
	    assert(r);
	}
    }

    // Test for recovery log
656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673
    if ((r == 0) && (env->i->open_flags & DB_INIT_LOG)) {
	// if using transactions, test for existence of log
	r = ydb_recover_log_exists(env);  // return 0 or ENOENT
	if (expect_newenv && (r != ENOENT))
	    r = toku_ydb_do_error(env, ENOENT, "Persistent environment information is missing (but log exists)\n");
	else if (!expect_newenv && r == ENOENT)
	    r = toku_ydb_do_error(env, ENOENT, "Recovery log is missing (persistent environment information is present)\n");
	else
	    r = 0;
    }

    if (r == 0)
	*valid_newenv = expect_newenv;
    else 
	*valid_newenv = FALSE;
    return r;
}

674
static int
675
ydb_maybe_upgrade_env (DB_ENV *env, LSN * last_lsn_of_clean_shutdown_read_from_log, BOOL * upgrade_in_progress) {
676 677 678
    int r = 0;
    if (env->i->open_flags & DB_INIT_TXN && env->i->open_flags & DB_INIT_LOG) {
        toku_ydb_unlock();
679
        r = toku_maybe_upgrade_log(env->i->dir, env->i->real_log_dir, last_lsn_of_clean_shutdown_read_from_log, upgrade_in_progress);
680 681 682 683 684 685
        toku_ydb_lock();
    }
    return r;
}


Yoni Fogel's avatar
Yoni Fogel committed
686 687 688 689 690 691 692 693 694 695 696 697
static void
unlock_single_process(DB_ENV *env) {
    int r;
    r = single_process_unlock(&env->i->envdir_lockfd);
    lazy_assert(r==0);
    r = single_process_unlock(&env->i->datadir_lockfd);
    lazy_assert(r==0);
    r = single_process_unlock(&env->i->logdir_lockfd);
    lazy_assert(r==0);
    r = single_process_unlock(&env->i->tmpdir_lockfd);
    lazy_assert(r==0);
}
698 699 700 701 702 703 704

// Open the environment.
// If this is a new environment, then create the necessary files.
// Return 0 on success, ENOENT if any of the expected necessary files are missing.
// (The set of necessary files is defined in the function validate_env() above.)
static int 
toku_env_open(DB_ENV * env, const char *home, u_int32_t flags, int mode) {
705
    HANDLE_PANICKED_ENV(env);
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
706
    int r;
707
    BOOL newenv;  // true iff creating a new environment
708
    u_int32_t unused_flags=flags;
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
709

Rich Prohaska's avatar
Rich Prohaska committed
710
    if (env_opened(env)) {
Yoni Fogel's avatar
Yoni Fogel committed
711 712
	r = toku_ydb_do_error(env, EINVAL, "The environment is already open\n");
        goto cleanup;
713
    }
Yoni Fogel's avatar
Yoni Fogel committed
714

715 716
    assert(sizeof(time_t) == sizeof(uint64_t));

717 718 719 720
    HANDLE_EXTRA_FLAGS(env, flags, 
                       DB_CREATE|DB_PRIVATE|DB_INIT_LOG|DB_INIT_TXN|DB_RECOVER|DB_INIT_MPOOL|DB_INIT_LOCK|DB_THREAD);


721 722
    // DB_CREATE means create if env does not exist, and Tokudb requires it because
    // Tokudb requries DB_PRIVATE.
Bradley C. Kuszmaul's avatar
Bradley C. Kuszmaul committed
723
    if ((flags & DB_PRIVATE) && !(flags & DB_CREATE)) {
Yoni Fogel's avatar
Yoni Fogel committed
724 725
	r = toku_ydb_do_error(env, ENOENT, "DB_PRIVATE requires DB_CREATE (seems gratuitous to us, but that's BDB's behavior\n");
        goto cleanup;
Bradley C. Kuszmaul's avatar
Bradley C. Kuszmaul committed
726 727
    }

728
    if (!(flags & DB_PRIVATE)) {
Yoni Fogel's avatar
Yoni Fogel committed
729 730
	r = toku_ydb_do_error(env, ENOENT, "TokuDB requires DB_PRIVATE\n");
        goto cleanup;
Yoni Fogel's avatar
Yoni Fogel committed
731
    }
732

Yoni Fogel's avatar
Yoni Fogel committed
733 734 735 736
    if ((flags & DB_INIT_LOG) && !(flags & DB_INIT_TXN)) {
	r = toku_ydb_do_error(env, EINVAL, "TokuDB requires transactions for logging\n");
        goto cleanup;
    }
737

Yoni Fogel's avatar
Yoni Fogel committed
738
    if (!home) home = ".";
Yoni Fogel's avatar
Yoni Fogel committed
739

740 741 742
    // Verify that the home exists.
    {
	BOOL made_new_home = FALSE;
Zardosht Kasheff's avatar
Zardosht Kasheff committed
743
        char* new_home = NULL;
744
    	toku_struct_stat buf;
745
        if (strlen(home) > 1 && home[strlen(home)-1] == '\\') {
Zardosht Kasheff's avatar
Zardosht Kasheff committed
746 747 748 749 750
            new_home = toku_malloc(strlen(home));
            memcpy(new_home, home, strlen(home));
            new_home[strlen(home) - 1] = 0;
            made_new_home = TRUE;
        }
751
    	r = toku_stat(made_new_home? new_home : home, &buf);
Zardosht Kasheff's avatar
Zardosht Kasheff committed
752 753 754 755
        if (made_new_home) {
            toku_free(new_home);
        }
    	if (r!=0) {
Yoni Fogel's avatar
Yoni Fogel committed
756 757
    	    r = toku_ydb_do_error(env, errno, "Error from toku_stat(\"%s\",...)\n", home);
            goto cleanup;
Zardosht Kasheff's avatar
Zardosht Kasheff committed
758
    	}
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
759
    }
760
    unused_flags &= ~DB_PRIVATE;
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
761 762 763

    if (env->i->dir)
        toku_free(env->i->dir);
Yoni Fogel's avatar
Yoni Fogel committed
764
    env->i->dir = toku_strdup(home);
765
    if (env->i->dir == 0) {
Yoni Fogel's avatar
Yoni Fogel committed
766 767
	r = toku_ydb_do_error(env, ENOMEM, "Out of memory\n");
        goto cleanup;
768
    }
Yoni Fogel's avatar
Yoni Fogel committed
769 770 771 772
    if (0) {
        died1:
        toku_free(env->i->dir);
        env->i->dir = NULL;
Yoni Fogel's avatar
Yoni Fogel committed
773
        goto cleanup;
Yoni Fogel's avatar
Yoni Fogel committed
774
    }
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
775 776
    env->i->open_flags = flags;
    env->i->open_mode = mode;
Bradley C. Kuszmaul's avatar
Bradley C. Kuszmaul committed
777

778 779
    env_setup_real_data_dir(env);
    env_setup_real_log_dir(env);
780
    env_setup_real_tmp_dir(env);
781

Yoni Fogel's avatar
Yoni Fogel committed
782 783
    r = single_process_lock(env->i->dir, "environment", &env->i->envdir_lockfd);
    if (r!=0) goto cleanup;
784
    r = single_process_lock(env->i->real_data_dir, "data", &env->i->datadir_lockfd);
Yoni Fogel's avatar
Yoni Fogel committed
785
    if (r!=0) goto cleanup;
786
    r = single_process_lock(env->i->real_log_dir, "logs", &env->i->logdir_lockfd);
Yoni Fogel's avatar
Yoni Fogel committed
787
    if (r!=0) goto cleanup;
788
    r = single_process_lock(env->i->real_tmp_dir, "temp", &env->i->tmpdir_lockfd);
Yoni Fogel's avatar
Yoni Fogel committed
789 790 791
    if (r!=0) goto cleanup;


792 793 794 795 796
    BOOL need_rollback_cachefile = FALSE;
    if (flags & (DB_INIT_TXN | DB_INIT_LOG)) {
        need_rollback_cachefile = TRUE;
    }

797 798
    init_status_info();  // do this before possibly upgrading, so upgrade work is counted in status counters

799 800 801
    LSN last_lsn_of_clean_shutdown_read_from_log = ZERO_LSN;
    BOOL upgrade_in_progress = FALSE;
    r = ydb_maybe_upgrade_env(env, &last_lsn_of_clean_shutdown_read_from_log, &upgrade_in_progress);
Yoni Fogel's avatar
Yoni Fogel committed
802
    if (r!=0) goto cleanup;
803

804 805 806 807 808 809 810 811 812 813 814
    if (upgrade_in_progress) {
	// Delete old rollback file.  There was a clean shutdown, so it has nothing useful,
	// and there is no value in upgrading it.  It is simpler to just create a new one.
	char* rollback_filename = toku_construct_full_name(2, env->i->dir, ROLLBACK_CACHEFILE_NAME);
	assert(rollback_filename);
	r = unlink(rollback_filename);
	toku_free(rollback_filename);
	assert(r==0 || errno==ENOENT);	
	need_rollback_cachefile = FALSE;  // we're not expecting it to exist now
    }
    
815
    r = validate_env(env, &newenv, need_rollback_cachefile);  // make sure that environment is either new or complete
Yoni Fogel's avatar
Yoni Fogel committed
816
    if (r != 0) goto cleanup;
817

818
    unused_flags &= ~DB_INIT_TXN & ~DB_INIT_LOG;
819

820 821 822 823 824 825 826
    // do recovery only if there exists a log and recovery is requested
    // otherwise, a log is created when the logger is opened later
    if (!newenv) {
        if (flags & DB_INIT_LOG) {
            // the log does exist
            if (flags & DB_RECOVER) {
                r = ydb_do_recovery(env);
Yoni Fogel's avatar
Yoni Fogel committed
827
                if (r != 0) goto cleanup;
828 829 830
            } else {
                // the log is required to have clean shutdown if recovery is not requested
                r = needs_recovery(env);
Yoni Fogel's avatar
Yoni Fogel committed
831
                if (r != 0) goto cleanup;
832
            }
833 834
        }
    }
835 836
    
    toku_loader_cleanup_temp_files(env);
837

Bradley C. Kuszmaul's avatar
Bradley C. Kuszmaul committed
838
    if (flags & (DB_INIT_TXN | DB_INIT_LOG)) {
839
	assert(env->i->logger);
840
        toku_logger_write_log_files(env->i->logger, (BOOL)((flags & DB_INIT_LOG) != 0));
841
        r = toku_logger_open(env->i->real_log_dir, env->i->logger);
842
	if (r!=0) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
843
	    toku_ydb_do_error(env, r, "Could not open logger\n");
Bradley C. Kuszmaul's avatar
Bradley C. Kuszmaul committed
844
	died2:
845
	    toku_logger_close(&env->i->logger);
Bradley C. Kuszmaul's avatar
Bradley C. Kuszmaul committed
846 847
	    goto died1;
	}
848 849 850
    } else {
	r = toku_logger_close(&env->i->logger); // if no logging system, then kill the logger
	assert(r==0);
Bradley C. Kuszmaul's avatar
Bradley C. Kuszmaul committed
851 852
    }

853 854 855 856 857 858 859
    unused_flags &= ~DB_INIT_MPOOL; // we always init an mpool.
    unused_flags &= ~DB_CREATE;     // we always do DB_CREATE
    unused_flags &= ~DB_INIT_LOCK;  // we check this later (e.g. in db->open)
    unused_flags &= ~DB_RECOVER;

// This is probably correct, but it will be pain...
//    if ((flags & DB_THREAD)==0) {
Yoni Fogel's avatar
Yoni Fogel committed
860 861
//	r = toku_ydb_do_error(env, EINVAL, "TokuDB requires DB_THREAD");
//	goto cleanup;
862 863 864 865
//    }
    unused_flags &= ~DB_THREAD;

    if (unused_flags!=0) {
Yoni Fogel's avatar
Yoni Fogel committed
866 867
	r = toku_ydb_do_error(env, EINVAL, "Extra flags not understood by tokudb: %u\n", unused_flags);
        goto cleanup;
868 869
    }

870
    r = toku_brt_create_cachetable(&env->i->cachetable, env->i->cachetable_size, ZERO_LSN, env->i->logger);
Bradley C. Kuszmaul's avatar
Bradley C. Kuszmaul committed
871
    if (r!=0) goto died2;
872

Yoni Fogel's avatar
Yoni Fogel committed
873 874
    toku_cachetable_set_env_dir(env->i->cachetable, env->i->dir);

875 876
    int using_txns = env->i->open_flags & DB_INIT_TXN;
    if (env->i->logger) {
877 878
	// if this is a newborn env or if this is an upgrade, then create a brand new rollback file
	BOOL create_new_rollback_file = newenv | upgrade_in_progress;
879 880 881
	assert (using_txns);
	toku_logger_set_cachetable(env->i->logger, env->i->cachetable);
	toku_logger_set_remove_finalize_callback(env->i->logger, finalize_file_removal, env->i->ltm);
882
        r = toku_logger_open_rollback(env->i->logger, env->i->cachetable, create_new_rollback_file);
883
        assert(r==0);
884
    }
885

886 887
    DB_TXN *txn=NULL;
    if (using_txns) {
888
        r = toku_txn_begin(env, 0, &txn, 0, 1);
889
        assert(r==0);
890
    }
891

892 893
    {
        r = toku_db_create(&env->i->persistent_environment, env, 0);
894
        assert(r==0);
895 896
        r = db_use_builtin_key_cmp(env->i->persistent_environment);
        assert(r==0);
897
	r = db_open_iname(env->i->persistent_environment, txn, environmentdictionary, DB_CREATE, mode);
898
	assert(r==0);
899
	if (newenv) {
900 901
	    // create new persistent_environment
	    DBT key, val;
902 903 904
	    persistent_original_env_version = BRT_LAYOUT_VERSION;
	    const uint32_t environment_version = toku_htod32(persistent_original_env_version);

905
	    toku_fill_dbt(&key, orig_env_ver_key, strlen(orig_env_ver_key));
906 907 908
	    toku_fill_dbt(&val, &environment_version, sizeof(environment_version));
	    r = toku_db_put(env->i->persistent_environment, txn, &key, &val, 0);
	    assert(r==0);
909

910
	    toku_fill_dbt(&key, curr_env_ver_key, strlen(curr_env_ver_key));
911 912 913
	    toku_fill_dbt(&val, &environment_version, sizeof(environment_version));
	    r = toku_db_put(env->i->persistent_environment, txn, &key, &val, 0);
	    assert(r==0);
914 915 916 917 918 919

	    time_t creation_time_d = toku_htod64(time(NULL));
	    toku_fill_dbt(&key, creation_time_key, strlen(creation_time_key));
	    toku_fill_dbt(&val, &creation_time_d, sizeof(creation_time_d));
	    r = toku_db_put(env->i->persistent_environment, txn, &key, &val, 0);
	    assert(r==0);
920 921
	}
	else {
922
	    r = maybe_upgrade_persistent_environment_dictionary(env, txn, last_lsn_of_clean_shutdown_read_from_log);
923 924
	    assert(r==0);
	}
925
	capture_persistent_env(env, txn);
926 927 928
    }
    {
        r = toku_db_create(&env->i->directory, env, 0);
929
        assert(r==0);
930 931
        r = db_use_builtin_key_cmp(env->i->directory);
        assert(r==0);
932 933
        r = db_open_iname(env->i->directory, txn, fileopsdirectory, DB_CREATE, mode);
        assert(r==0);
934 935
    }
    if (using_txns) {
936
        r = toku_txn_commit(txn, 0, NULL, NULL);
937
        assert(r==0);
938 939
    }
    toku_ydb_unlock();
940
    r = toku_checkpoint(env->i->cachetable, env->i->logger, NULL, NULL, NULL, NULL);
941
    assert(r==0);
942
    toku_ydb_lock();
943 944
    env_fs_poller(env);          // get the file system state at startup
    env_fs_init_minicron(env); 
Yoni Fogel's avatar
Yoni Fogel committed
945 946 947 948 949 950
cleanup:
    if (r!=0) {
        if (env && env->i) {
            unlock_single_process(env);
        }
    }
951
    if (r == 0) {
952
	errno = 0; // tabula rasa
953 954
	most_recent_env = env;
    }
Yoni Fogel's avatar
Yoni Fogel committed
955
    return r;
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
956
}
Bradley C. Kuszmaul's avatar
Bradley C. Kuszmaul committed
957

958 959
static int 
toku_env_close(DB_ENV * env, u_int32_t flags) {
960 961 962
    int r = 0;

    // if panicked, or if any open transactions, or any open dbs, then do nothing.
963

964
    if (toku_env_is_panicked(env)) goto panic_and_quit_early;
965
    if (!toku_list_empty(&env->i->open_txns)) {
966 967 968
        r = toku_ydb_do_error(env, EINVAL, "Cannot close environment due to open transactions\n");
        goto panic_and_quit_early;
    }
969 970 971 972 973 974 975
    { //Verify open dbs. Zombies are ok at this stage, fully open is not.
        uint32_t size = toku_omt_size(env->i->open_dbs);
        assert(size == env->i->num_open_dbs + env->i->num_zombie_dbs);
        if (env->i->num_open_dbs > 0) {
            r = toku_ydb_do_error(env, EINVAL, "Cannot close environment due to open DBs\n");
            goto panic_and_quit_early;
        }
976 977 978 979 980 981 982 983 984 985 986 987 988 989 990 991 992
    }
    {
        if (env->i->persistent_environment) {
            r = toku_db_close(env->i->persistent_environment, 0);
            if (r) {
                toku_ydb_do_error(env, r, "Cannot close persistent environment dictionary (DB->close error)\n");
                goto panic_and_quit_early;
            }
        }
        if (env->i->directory) {
            r = toku_db_close(env->i->directory, 0);
            if (r) {
                toku_ydb_do_error(env, r, "Cannot close Directory dictionary (DB->close error)\n");
                goto panic_and_quit_early;
            }
        }
    }
993
    if (env->i->cachetable) {
994 995
	toku_ydb_unlock();  // ydb lock must not be held when shutting down minicron
	toku_cachetable_minicron_shutdown(env->i->cachetable);
996
        if (env->i->logger) {
997 998 999
            if ( flags && DB_CLOSE_DONT_TRIM_LOG ) {
                toku_logger_trim_log_files(env->i->logger, FALSE);
            }
1000
            r = toku_checkpoint(env->i->cachetable, env->i->logger, NULL, NULL, NULL, NULL);
1001 1002 1003 1004
            if (r) {
                toku_ydb_do_error(env, r, "Cannot close environment (error during checkpoint)\n");
                goto panic_and_quit_early;
            }
1005 1006 1007 1008 1009 1010 1011 1012
            { //Verify open dbs. Neither Zombies nor fully open are ok at this stage.
                uint32_t size = toku_omt_size(env->i->open_dbs);
                assert(size == env->i->num_open_dbs + env->i->num_zombie_dbs);
                if (size > 0) {
                    r = toku_ydb_do_error(env, EINVAL, "Cannot close environment due to zombie DBs\n");
                    goto panic_and_quit_early;
                }
            }
1013 1014 1015 1016 1017 1018 1019 1020 1021 1022 1023
            r = toku_logger_close_rollback(env->i->logger, FALSE);
            if (r) {
                toku_ydb_do_error(env, r, "Cannot close environment (error during closing rollback cachefile)\n");
                goto panic_and_quit_early;
            }
            //Do a second checkpoint now that the rollback cachefile is closed.
            r = toku_checkpoint(env->i->cachetable, env->i->logger, NULL, NULL, NULL, NULL);
            if (r) {
                toku_ydb_do_error(env, r, "Cannot close environment (error during checkpoint)\n");
                goto panic_and_quit_early;
            }
1024 1025 1026 1027 1028
            r = toku_logger_shutdown(env->i->logger); 
            if (r) {
                toku_ydb_do_error(env, r, "Cannot close environment (error during logger shutdown)\n");
                goto panic_and_quit_early;
            }
1029
        }
1030
	toku_ydb_lock();
1031 1032 1033 1034
        r=toku_cachetable_close(&env->i->cachetable);
	if (r) {
	    toku_ydb_do_error(env, r, "Cannot close environment (cachetable close error)\n");
            goto panic_and_quit_early;
1035 1036 1037
	}
    }
    if (env->i->logger) {
1038 1039 1040 1041 1042
        r=toku_logger_close(&env->i->logger);
	if (r) {
            env->i->logger = NULL;
	    toku_ydb_do_error(env, r, "Cannot close environment (logger close error)\n");
            goto panic_and_quit_early;
1043 1044 1045 1046
	}
    }
    // Even if nothing else went wrong, but we were panicked, then raise an error.
    // But if something else went wrong then raise that error (above)
1047 1048 1049 1050
    if (toku_env_is_panicked(env))
        goto panic_and_quit_early;
    else
	assert(env->i->panic_string==0);
1051

1052
    env_fs_destroy(env);
Yoni Fogel's avatar
Yoni Fogel committed
1053
    toku_ltm_close(env->i->ltm);
1054 1055
    if (env->i->data_dir)
        toku_free(env->i->data_dir);
1056 1057
    if (env->i->lg_dir)
        toku_free(env->i->lg_dir);
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
1058 1059
    if (env->i->tmp_dir)
        toku_free(env->i->tmp_dir);
1060 1061 1062 1063
    if (env->i->real_data_dir)
	toku_free(env->i->real_data_dir);
    if (env->i->real_log_dir)
	toku_free(env->i->real_log_dir);
1064 1065
    if (env->i->real_tmp_dir)
	toku_free(env->i->real_tmp_dir);
1066 1067
    if (env->i->open_dbs)
        toku_omt_destroy(&env->i->open_dbs);
1068 1069
    if (env->i->dir)
	toku_free(env->i->dir);
1070 1071
    //Immediately before freeing internal environment unlock the directories
    unlock_single_process(env);
Bradley C. Kuszmaul's avatar
Bradley C. Kuszmaul committed
1072
    toku_free(env->i);
1073
    env->i = NULL;
Bradley C. Kuszmaul's avatar
Bradley C. Kuszmaul committed
1074
    toku_free(env);
1075 1076 1077 1078 1079 1080
    env = NULL;
    if ((flags!=0) && !(flags==DB_CLOSE_DONT_TRIM_LOG))
        r = EINVAL;
    return r;

panic_and_quit_early:
Yoni Fogel's avatar
Yoni Fogel committed
1081 1082
    //release lock files.
    unlock_single_process(env);
1083 1084 1085 1086 1087 1088 1089 1090
    //r is the panic error
    if (toku_env_is_panicked(env)) {
        char *panic_string = env->i->panic_string;
        r = toku_ydb_do_error(env, toku_env_is_panicked(env), "Cannot close environment due to previous error: %s\n", panic_string);
    }
    else
        env->i->is_panicked = r;
    return r;
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
1091
}
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
1092

1093 1094
static int 
toku_env_log_archive(DB_ENV * env, char **list[], u_int32_t flags) {
1095
    return toku_logger_log_archive(env->i->logger, list, flags);
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
1096
}
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
1097

1098 1099
static int 
toku_env_log_flush(DB_ENV * env, const DB_LSN * lsn __attribute__((__unused__))) {
1100
    HANDLE_PANICKED_ENV(env);
1101 1102
    // We just flush everything.  MySQL uses lsn==0 which means flush everything.  For anyone else using the log, it is correct to flush too much, so we are OK.
    return toku_logger_fsync(env->i->logger);
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
1103
}
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
1104

1105 1106
static int 
toku_env_set_cachesize(DB_ENV * env, u_int32_t gbytes, u_int32_t bytes, int ncache) {
1107
    HANDLE_PANICKED_ENV(env);
Rich Prohaska's avatar
Rich Prohaska committed
1108 1109
    if (ncache != 1)
        return EINVAL;
Rich Prohaska's avatar
Rich Prohaska committed
1110 1111 1112 1113 1114
    u_int64_t cs64 = ((u_int64_t) gbytes << 30) + bytes;
    unsigned long cs = cs64;
    if (cs64 > cs)
        return EINVAL;
    env->i->cachetable_size = cs;
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
1115 1116 1117
    return 0;
}

1118 1119 1120 1121 1122 1123 1124 1125 1126 1127 1128 1129 1130 1131 1132 1133 1134 1135 1136 1137 1138 1139 1140 1141 1142
static int toku_env_dbremove(DB_ENV * env, DB_TXN *txn, const char *fname, const char *dbname, u_int32_t flags);

static int
locked_env_dbremove(DB_ENV * env, DB_TXN *txn, const char *fname, const char *dbname, u_int32_t flags) {
    toku_multi_operation_client_lock(); //Cannot begin checkpoint
    toku_ydb_lock();
    int r = toku_env_dbremove(env, txn, fname, dbname, flags);
    toku_ydb_unlock();
    toku_multi_operation_client_unlock(); //Can now begin checkpoint
    return r;
}

static int toku_env_dbrename(DB_ENV *env, DB_TXN *txn, const char *fname, const char *dbname, const char *newname, u_int32_t flags);

static int
locked_env_dbrename(DB_ENV *env, DB_TXN *txn, const char *fname, const char *dbname, const char *newname, u_int32_t flags) {
    toku_multi_operation_client_lock(); //Cannot begin checkpoint
    toku_ydb_lock();
    int r = toku_env_dbrename(env, txn, fname, dbname, newname, flags);
    toku_ydb_unlock();
    toku_multi_operation_client_unlock(); //Can now begin checkpoint
    return r;
}


Rich Prohaska's avatar
Rich Prohaska committed
1143 1144
#if DB_VERSION_MAJOR == 4 && DB_VERSION_MINOR >= 3

1145 1146
static int 
toku_env_get_cachesize(DB_ENV * env, u_int32_t *gbytes, u_int32_t *bytes, int *ncache) {
1147
    HANDLE_PANICKED_ENV(env);
Rich Prohaska's avatar
Rich Prohaska committed
1148 1149 1150 1151 1152 1153
    *gbytes = env->i->cachetable_size >> 30;
    *bytes = env->i->cachetable_size & ((1<<30)-1);
    *ncache = 1;
    return 0;
}

1154 1155
static int 
locked_env_get_cachesize(DB_ENV *env, u_int32_t *gbytes, u_int32_t *bytes, int *ncache) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1156
    toku_ydb_lock(); int r = toku_env_get_cachesize(env, gbytes, bytes, ncache); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
1157
}
Rich Prohaska's avatar
Rich Prohaska committed
1158 1159
#endif

1160 1161
static int 
toku_env_set_data_dir(DB_ENV * env, const char *dir) {
1162
    HANDLE_PANICKED_ENV(env);
Yoni Fogel's avatar
Yoni Fogel committed
1163 1164
    int r;
    
Rich Prohaska's avatar
Rich Prohaska committed
1165
    if (env_opened(env) || !dir) {
1166
	r = toku_ydb_do_error(env, EINVAL, "You cannot set the data dir after opening the env\n");
1167
    }
1168 1169
    else if (env->i->data_dir)
	r = toku_ydb_do_error(env, EINVAL, "You cannot set the data dir more than once.\n");
1170 1171 1172 1173 1174
    else {
        env->i->data_dir = toku_strdup(dir);
        if (env->i->data_dir==NULL) {
            assert(errno == ENOMEM);
            r = toku_ydb_do_error(env, ENOMEM, "Out of memory\n");
Yoni Fogel's avatar
Yoni Fogel committed
1175
        }
1176
        else r = 0;
Yoni Fogel's avatar
Yoni Fogel committed
1177
    }
1178
    return r;
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
1179
}
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
1180

1181 1182
static void 
toku_env_set_errcall(DB_ENV * env, toku_env_errcall_t errcall) {
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
1183
    env->i->errcall = errcall;
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
1184
}
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
1185

1186 1187
static void 
toku_env_set_errfile(DB_ENV*env, FILE*errfile) {
1188 1189 1190
    env->i->errfile = errfile;
}

1191 1192
static void 
toku_env_set_errpfx(DB_ENV * env, const char *errpfx) {
1193
    env->i->errpfx = errpfx;
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
1194
}
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
1195

1196 1197
static int 
toku_env_set_flags(DB_ENV * env, u_int32_t flags, int onoff) {
1198
    HANDLE_PANICKED_ENV(env);
Yoni Fogel's avatar
Yoni Fogel committed
1199 1200 1201 1202 1203 1204

    u_int32_t change = 0;
    if (flags & DB_AUTO_COMMIT) {
        change |=  DB_AUTO_COMMIT;
        flags  &= ~DB_AUTO_COMMIT;
    }
1205
    if (flags != 0 && onoff) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1206
	return toku_ydb_do_error(env, EINVAL, "TokuDB does not (yet) support any nonzero ENV flags other than DB_AUTO_COMMIT\n");
1207
    }
Yoni Fogel's avatar
Yoni Fogel committed
1208 1209
    if   (onoff) env->i->open_flags |=  change;
    else         env->i->open_flags &= ~change;
Rich Prohaska's avatar
Rich Prohaska committed
1210
    return 0;
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
1211
}
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
1212

1213 1214
static int 
toku_env_set_lg_bsize(DB_ENV * env, u_int32_t bsize) {
1215
    HANDLE_PANICKED_ENV(env);
1216
    return toku_logger_set_lg_bsize(env->i->logger, bsize);
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
1217
}
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
1218

1219 1220
static int 
toku_env_set_lg_dir(DB_ENV * env, const char *dir) {
1221
    HANDLE_PANICKED_ENV(env);
Rich Prohaska's avatar
Rich Prohaska committed
1222
    if (env_opened(env)) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1223
	return toku_ydb_do_error(env, EINVAL, "Cannot set log dir after opening the env\n");
1224
    }
1225 1226

    if (env->i->lg_dir) toku_free(env->i->lg_dir);
1227 1228
    if (dir) {
        env->i->lg_dir = toku_strdup(dir);
1229
        if (!env->i->lg_dir) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1230
	    return toku_ydb_do_error(env, ENOMEM, "Out of memory\n");
1231
	}
1232
    }
1233 1234
    else env->i->lg_dir = NULL;
    return 0;
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
1235
}
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
1236

1237 1238
static int 
toku_env_set_lg_max(DB_ENV * env, u_int32_t lg_max) {
1239
    HANDLE_PANICKED_ENV(env);
1240 1241 1242
    return toku_logger_set_lg_max(env->i->logger, lg_max);
}

1243 1244
static int 
toku_env_get_lg_max(DB_ENV * env, u_int32_t *lg_maxp) {
1245 1246
    HANDLE_PANICKED_ENV(env);
    return toku_logger_get_lg_max(env->i->logger, lg_maxp);
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
1247
}
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
1248

1249 1250
static int 
toku_env_set_lk_detect(DB_ENV * env, u_int32_t detect) {
1251
    HANDLE_PANICKED_ENV(env);
1252
    detect=detect;
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1253
    return toku_ydb_do_error(env, EINVAL, "TokuDB does not (yet) support set_lk_detect\n");
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
1254
}
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
1255

1256 1257
static int 
toku_env_set_lk_max_locks(DB_ENV *dbenv, u_int32_t max) {
Yoni Fogel's avatar
Yoni Fogel committed
1258
    int r = ENOSYS;
Yoni Fogel's avatar
Yoni Fogel committed
1259
    HANDLE_PANICKED_ENV(dbenv);
Yoni Fogel's avatar
Yoni Fogel committed
1260
    if (env_opened(dbenv))         { return EINVAL; }
1261
    r = toku_ltm_set_max_locks(dbenv->i->ltm, max);
Yoni Fogel's avatar
Yoni Fogel committed
1262
    return r;
Yoni Fogel's avatar
Yoni Fogel committed
1263 1264
}

1265
#if DB_VERSION_MAJOR == 4 && DB_VERSION_MINOR <= 4
1266 1267
static int 
toku_env_set_lk_max(DB_ENV * env, u_int32_t lk_max) {
Yoni Fogel's avatar
Yoni Fogel committed
1268
    return toku_env_set_lk_max_locks(env, lk_max);
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
1269
}
Rich Prohaska's avatar
Rich Prohaska committed
1270

1271 1272
static int 
locked_env_set_lk_max(DB_ENV * env, u_int32_t lk_max) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1273
    toku_ydb_lock(); int r = toku_env_set_lk_max(env, lk_max); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
1274
}
1275
#endif
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
1276

1277 1278
static int 
toku_env_get_lk_max_locks(DB_ENV *dbenv, u_int32_t *lk_maxp) {
1279
    HANDLE_PANICKED_ENV(dbenv);
1280
    return toku_ltm_get_max_locks(dbenv->i->ltm, lk_maxp);
1281 1282
}

1283 1284
static int 
locked_env_set_lk_max_locks(DB_ENV *dbenv, u_int32_t max) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1285
    toku_ydb_lock(); int r = toku_env_set_lk_max_locks(dbenv, max); toku_ydb_unlock(); return r;
1286 1287
}

1288 1289
static int 
locked_env_get_lk_max_locks(DB_ENV *dbenv, u_int32_t *lk_maxp) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1290
    toku_ydb_lock(); int r = toku_env_get_lk_max_locks(dbenv, lk_maxp); toku_ydb_unlock(); return r;
1291 1292
}

1293 1294
static int 
toku_env_set_lk_max_memory(DB_ENV *dbenv, uint64_t max) {
1295 1296 1297 1298 1299 1300 1301
    int r = ENOSYS;
    HANDLE_PANICKED_ENV(dbenv);
    if (env_opened(dbenv))         { return EINVAL; }
    r = toku_ltm_set_max_lock_memory(dbenv->i->ltm, max);
    return r;
}

1302 1303
static int 
toku_env_get_lk_max_memory(DB_ENV *dbenv, uint64_t *lk_maxp) {
1304 1305 1306 1307
    HANDLE_PANICKED_ENV(dbenv);
    return toku_ltm_get_max_lock_memory(dbenv->i->ltm, lk_maxp);
}

1308 1309 1310 1311 1312 1313
static int 
locked_env_set_lk_max_memory(DB_ENV *dbenv, uint64_t max) {
    toku_ydb_lock(); 
    int r = toku_env_set_lk_max_memory(dbenv, max); 
    toku_ydb_unlock(); 
    return r;
1314 1315 1316 1317 1318 1319
}

static int locked_env_get_lk_max_memory(DB_ENV *dbenv, uint64_t *lk_maxp) {
    toku_ydb_lock(); int r = toku_env_get_lk_max_memory(dbenv, lk_maxp); toku_ydb_unlock(); return r;
}

Yoni Fogel's avatar
Yoni Fogel committed
1320
//void toku__env_set_noticecall (DB_ENV *env, void (*noticecall)(DB_ENV *, db_notices)) {
Bradley C. Kuszmaul's avatar
Fixup  
Bradley C. Kuszmaul committed
1321 1322
//    env->i->noticecall = noticecall;
//}
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
1323

1324 1325
static int 
toku_env_set_tmp_dir(DB_ENV * env, const char *tmp_dir) {
1326
    HANDLE_PANICKED_ENV(env);
Rich Prohaska's avatar
Rich Prohaska committed
1327
    if (env_opened(env)) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1328
	return toku_ydb_do_error(env, EINVAL, "Cannot set the tmp dir after opening an env\n");
1329 1330
    }
    if (!tmp_dir) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1331
	return toku_ydb_do_error(env, EINVAL, "Tmp dir bust be non-null\n");
1332
    }
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
1333 1334
    if (env->i->tmp_dir)
        toku_free(env->i->tmp_dir);
Yoni Fogel's avatar
Yoni Fogel committed
1335
    env->i->tmp_dir = toku_strdup(tmp_dir);
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
1336
    return env->i->tmp_dir ? 0 : ENOMEM;
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
1337
}
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
1338

1339 1340
static int 
toku_env_set_verbose(DB_ENV * env, u_int32_t which, int onoff) {
1341 1342
    HANDLE_PANICKED_ENV(env);
    which=which; onoff=onoff;
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
1343
    return 1;
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
1344
}
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
1345

1346
// For test purposes only.
1347 1348
// These callbacks are never used in production code, only as a way to test the system
// (for example, by causing crashes at predictable times).
1349 1350
static void (*checkpoint_callback_f)(void*) = NULL;
static void * checkpoint_callback_extra     = NULL;
1351 1352
static void (*checkpoint_callback2_f)(void*) = NULL;
static void * checkpoint_callback2_extra     = NULL;
1353

1354 1355
static int 
toku_env_txn_checkpoint(DB_ENV * env, u_int32_t kbyte __attribute__((__unused__)), u_int32_t min __attribute__((__unused__)), u_int32_t flags __attribute__((__unused__))) {
1356
    int r = toku_checkpoint(env->i->cachetable, env->i->logger,
1357 1358
			    checkpoint_callback_f,  checkpoint_callback_extra,
			    checkpoint_callback2_f, checkpoint_callback2_extra);
1359 1360
    if (r) {
	env->i->is_panicked = r; // Panicking the whole environment may be overkill, but I'm not sure what else to do.
1361 1362
	env->i->panic_string = toku_strdup("checkpoint error");
        toku_ydb_do_error(env, r, "Checkpoint\n");
1363 1364
    }
    return r;
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
1365 1366
}

1367 1368
static int 
toku_env_txn_stat(DB_ENV * env, DB_TXN_STAT ** statp, u_int32_t flags) {
1369 1370
    HANDLE_PANICKED_ENV(env);
    statp=statp;flags=flags;
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
1371
    return 1;
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
1372 1373
}

1374 1375
static int 
locked_env_open(DB_ENV * env, const char *home, u_int32_t flags, int mode) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1376
    toku_ydb_lock(); int r = toku_env_open(env, home, flags, mode); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
1377 1378
}

1379 1380
static int 
locked_env_close(DB_ENV * env, u_int32_t flags) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1381
    toku_ydb_lock(); int r = toku_env_close(env, flags); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
1382 1383
}

1384 1385
static int 
locked_env_log_archive(DB_ENV * env, char **list[], u_int32_t flags) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1386
    toku_ydb_lock(); int r = toku_env_log_archive(env, list, flags); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
1387 1388
}

1389 1390
static int 
locked_env_log_flush(DB_ENV * env, const DB_LSN * lsn) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1391
    toku_ydb_lock(); int r = toku_env_log_flush(env, lsn); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
1392 1393
}

1394 1395
static int 
locked_env_set_cachesize(DB_ENV *env, u_int32_t gbytes, u_int32_t bytes, int ncache) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1396
    toku_ydb_lock(); int r = toku_env_set_cachesize(env, gbytes, bytes, ncache); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
1397 1398
}

1399 1400
static int 
locked_env_set_data_dir(DB_ENV * env, const char *dir) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1401
    toku_ydb_lock(); int r = toku_env_set_data_dir(env, dir); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
1402 1403
}

1404 1405
static int 
locked_env_set_flags(DB_ENV * env, u_int32_t flags, int onoff) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1406
    toku_ydb_lock(); int r = toku_env_set_flags(env, flags, onoff); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
1407 1408
}

1409 1410
static int 
locked_env_set_lg_bsize(DB_ENV * env, u_int32_t bsize) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1411
    toku_ydb_lock(); int r = toku_env_set_lg_bsize(env, bsize); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
1412 1413
}

1414 1415
static int 
locked_env_set_lg_dir(DB_ENV * env, const char *dir) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1416
    toku_ydb_lock(); int r = toku_env_set_lg_dir(env, dir); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
1417 1418
}

1419 1420
static int 
locked_env_set_lg_max(DB_ENV * env, u_int32_t lg_max) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1421
    toku_ydb_lock(); int r = toku_env_set_lg_max(env, lg_max); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
1422 1423
}

1424 1425
static int 
locked_env_get_lg_max(DB_ENV * env, u_int32_t *lg_maxp) {
1426 1427 1428
    toku_ydb_lock(); int r = toku_env_get_lg_max(env, lg_maxp); toku_ydb_unlock(); return r;
}

1429 1430
static int 
locked_env_set_lk_detect(DB_ENV * env, u_int32_t detect) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1431
    toku_ydb_lock(); int r = toku_env_set_lk_detect(env, detect); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
1432 1433
}

1434 1435
static int 
locked_env_set_tmp_dir(DB_ENV * env, const char *tmp_dir) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1436
    toku_ydb_lock(); int r = toku_env_set_tmp_dir(env, tmp_dir); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
1437 1438
}

1439 1440
static int 
locked_env_set_verbose(DB_ENV * env, u_int32_t which, int onoff) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1441
    toku_ydb_lock(); int r = toku_env_set_verbose(env, which, onoff); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
1442 1443
}

1444 1445
static int 
locked_env_txn_stat(DB_ENV * env, DB_TXN_STAT ** statp, u_int32_t flags) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1446
    toku_ydb_lock(); int r = toku_env_txn_stat(env, statp, flags); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
1447 1448
}

1449 1450 1451 1452 1453 1454 1455 1456 1457 1458 1459 1460 1461 1462 1463 1464 1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476 1477 1478
static int
env_checkpointing_set_period(DB_ENV * env, u_int32_t seconds) {
    HANDLE_PANICKED_ENV(env);
    int r;
    if (!env_opened(env)) r = EINVAL;
    else
        r = toku_set_checkpoint_period(env->i->cachetable, seconds);
    return r;
}

static int
locked_env_checkpointing_set_period(DB_ENV * env, u_int32_t seconds) {
    toku_ydb_lock(); int r = env_checkpointing_set_period(env, seconds); toku_ydb_unlock(); return r;
}

static int
env_checkpointing_get_period(DB_ENV * env, u_int32_t *seconds) {
    HANDLE_PANICKED_ENV(env);
    int r = 0;
    if (!env_opened(env)) r = EINVAL;
    else 
        *seconds = toku_get_checkpoint_period(env->i->cachetable);
    return r;
}

static int
locked_env_checkpointing_get_period(DB_ENV * env, u_int32_t *seconds) {
    toku_ydb_lock(); int r = env_checkpointing_get_period(env, seconds); toku_ydb_unlock(); return r;
}

1479 1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 1496 1497 1498 1499 1500 1501 1502 1503 1504 1505 1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521 1522 1523 1524 1525 1526 1527 1528 1529 1530 1531 1532 1533
static int
env_checkpointing_postpone(DB_ENV * env) {
    HANDLE_PANICKED_ENV(env);
    int r = 0;
    if (!env_opened(env)) r = EINVAL;
    else toku_checkpoint_safe_client_lock();
    return r;
}

static int
env_checkpointing_resume(DB_ENV * env) {
    HANDLE_PANICKED_ENV(env);
    int r = 0;
    if (!env_opened(env)) r = EINVAL;
    else toku_checkpoint_safe_client_unlock();
    return r;
}

static int
env_checkpointing_begin_atomic_operation(DB_ENV * env) {
    HANDLE_PANICKED_ENV(env);
    int r = 0;
    if (!env_opened(env)) r = EINVAL;
    else toku_multi_operation_client_lock();
    return r;
}

static int
env_checkpointing_end_atomic_operation(DB_ENV * env) {
    HANDLE_PANICKED_ENV(env);
    int r = 0;
    if (!env_opened(env)) r = EINVAL;
    else toku_multi_operation_client_unlock();
    return r;
}

static int
env_set_default_bt_compare(DB_ENV * env, int (*bt_compare) (DB *, const DBT *, const DBT *)) {
    HANDLE_PANICKED_ENV(env);
    int r = 0;
    if (env_opened(env)) r = EINVAL;
    else {
        env->i->bt_compare = bt_compare;
    }
    return r;
}

static int
locked_env_set_default_bt_compare(DB_ENV * env, int (*bt_compare) (DB *, const DBT *, const DBT *)) {
    toku_ydb_lock();
    int r = env_set_default_bt_compare(env, bt_compare);
    toku_ydb_unlock();
    return r;
}

1534
static int
1535
env_set_generate_row_callback_for_put(DB_ENV *env, generate_row_for_put_func generate_row_for_put) {
1536 1537 1538 1539
    HANDLE_PANICKED_ENV(env);
    int r = 0;
    if (env_opened(env)) r = EINVAL;
    else {
1540
        env->i->generate_row_for_put = generate_row_for_put;
1541 1542 1543 1544 1545
    }
    return r;
}

static int
1546 1547 1548 1549 1550 1551 1552 1553 1554 1555 1556 1557
env_set_generate_row_callback_for_del(DB_ENV *env, generate_row_for_del_func generate_row_for_del) {
    HANDLE_PANICKED_ENV(env);
    int r = 0;
    if (env_opened(env)) r = EINVAL;
    else {
        env->i->generate_row_for_del = generate_row_for_del;
    }
    return r;
}

static int
locked_env_set_generate_row_callback_for_put(DB_ENV *env, generate_row_for_put_func generate_row_for_put) {
1558
    toku_ydb_lock();
1559
    int r = env_set_generate_row_callback_for_put(env, generate_row_for_put);
1560 1561 1562 1563
    toku_ydb_unlock();
    return r;
}

1564 1565 1566 1567 1568 1569 1570 1571
static int
locked_env_set_generate_row_callback_for_del(DB_ENV *env, generate_row_for_del_func generate_row_for_del) {
    toku_ydb_lock();
    int r = env_set_generate_row_callback_for_del(env, generate_row_for_del);
    toku_ydb_unlock();
    return r;
}

1572 1573 1574 1575 1576 1577 1578 1579 1580 1581 1582
static int env_put_multiple(DB_ENV *env, DB *src_db, DB_TXN *txn, 
                            const DBT *key, const DBT *val, 
                            uint32_t num_dbs, DB **db_array, DBT *keys, DBT *vals, uint32_t *flags_array, 
                            void *extra);

static int env_del_multiple(DB_ENV *env, DB *src_db, DB_TXN *txn, 
                            const DBT *key, const DBT *val, 
                            uint32_t num_dbs, DB **db_array, DBT *keys, uint32_t *flags_array, 
                            void *extra);

static int env_update_multiple(DB_ENV *env, DB *src_db, DB_TXN *txn, 
1583 1584 1585
                               DBT *old_src_key, DBT *old_src_data,
                               DBT *new_src_key, DBT *new_src_data,
                               uint32_t num_dbs, DB **db_array, uint32_t* flags_array, 
1586 1587
                               uint32_t num_keys, DBT *keys, 
                               uint32_t num_vals, DBT *vals,
1588
                               void *extra);
1589 1590

static int
1591
locked_env_put_multiple(DB_ENV *env, DB *src_db, DB_TXN *txn, const DBT *key, const DBT *val, uint32_t num_dbs, DB **db_array, DBT *keys, DBT *vals, uint32_t *flags_array, void *extra) {
1592 1593 1594 1595 1596 1597
    int r = env_check_avail_fs_space(env);
    if (r == 0) {
	toku_ydb_lock();
	r = env_put_multiple(env, src_db, txn, key, val, num_dbs, db_array, keys, vals, flags_array, extra);
	toku_ydb_unlock();
    }
1598 1599 1600 1601
    return r;
}

static int
1602
locked_env_del_multiple(DB_ENV *env, DB *src_db, DB_TXN *txn, const DBT *key, const DBT *val, uint32_t num_dbs, DB **db_array, DBT *keys, uint32_t *flags_array, void *extra) {
1603
    toku_ydb_lock();
1604
    int r = env_del_multiple(env, src_db, txn, key, val, num_dbs, db_array, keys, flags_array, extra);
1605 1606 1607 1608
    toku_ydb_unlock();
    return r;
}

1609 1610
static int
locked_env_update_multiple(DB_ENV *env, DB *src_db, DB_TXN *txn,                                
1611 1612 1613
                           DBT *old_src_key, DBT *old_src_data,
                           DBT *new_src_key, DBT *new_src_data,
                           uint32_t num_dbs, DB **db_array, uint32_t* flags_array, 
1614 1615
                           uint32_t num_keys, DBT *keys, 
                           uint32_t num_vals, DBT *vals,
1616 1617
                           void *extra) {
    toku_ydb_lock();
1618
    int r = env_update_multiple(env, src_db, txn, old_src_key, old_src_data, new_src_key, new_src_data, num_dbs, db_array, flags_array, num_keys, keys, num_vals, vals, extra);
1619 1620 1621
    toku_ydb_unlock();
    return r;
}
1622

1623 1624 1625 1626 1627 1628 1629 1630 1631 1632 1633 1634 1635 1636 1637 1638 1639 1640 1641 1642 1643
static int
env_set_redzone(DB_ENV *env, int redzone) {
    HANDLE_PANICKED_ENV(env);
    int r;
    if (env_opened(env))
        r = EINVAL;
    else {
        env->i->redzone = redzone;
        r = 0;
    }
    return r;
}

static int 
locked_env_set_redzone(DB_ENV *env, int redzone) {
    toku_ydb_lock();
    int r= env_set_redzone(env, redzone);
    toku_ydb_unlock();
    return r;
}

1644 1645 1646 1647 1648 1649 1650 1651 1652 1653 1654 1655 1656 1657 1658 1659
static void
format_time(const time_t *timer, char *buf) {
    ctime_r(timer, buf);
    size_t len = strlen(buf);
    assert(len < 26);
    char end;

    assert(len>=1);
    end = buf[len-1];
    while (end == '\n' || end == '\r') {
        buf[len-1] = '\0';
        len--;
        assert(len>=1);
        end = buf[len-1];
    }
}
1660

1661 1662
// Do not take ydb lock or any other lock around or in this function.  
// If the engine is blocked because some thread is holding a lock, this function
1663 1664 1665
// can help diagnose the problem.
// This function only collects information, and it does not matter if something gets garbled
// because of a race condition.  
1666
// Note, engine status is still collected even if the environment or logger is panicked
1667
static int
1668 1669 1670 1671 1672 1673 1674 1675 1676
env_get_engine_status(DB_ENV * env, ENGINE_STATUS * engstat, char * env_panic_string_buf, int env_panic_string_length) {
    int r;
    if (env_panic_string_buf) {
	if (env && env->i && env->i->is_panicked && env->i->panic_string)
	    strncpy(env_panic_string_buf, env->i->panic_string, env_panic_string_length);
	else 
	    *env_panic_string_buf = '\0';
    }

1677 1678 1679 1680
    if ( !(env)     || 
	 !(env->i)  || 
	 !(env_opened(env)) )
	r = EINVAL;
1681
    else {
1682 1683
	r = 0;
	engstat->env_panic = env->i->is_panicked;
1684
	format_time(&persistent_creation_time, engstat->creationtime);
1685 1686
	time_t now = time(NULL);
        format_time(&now, engstat->now);
1687
        format_time(&startuptime, engstat->startuptime);
1688 1689 1690 1691 1692 1693 1694 1695 1696 1697 1698 1699 1700 1701 1702 1703 1704 1705
	{
	    SCHEDULE_STATUS_S schedstat;
	    toku_ydb_lock_get_status(&schedstat);
	    engstat->ydb_lock_ctr = schedstat.ydb_lock_ctr;                        /* how many times has ydb lock been taken/released */ 
	    engstat->max_possible_sleep = schedstat.max_possible_sleep;            /* max possible sleep time for ydb lock scheduling (constant) */ 
	    engstat->processor_freq_mhz = schedstat.processor_freq_mhz;            /* clock frequency in MHz */
	    engstat->max_requested_sleep = schedstat.max_requested_sleep;          /* max sleep time requested, can be larger than max possible */ 
	    engstat->times_max_sleep_used = schedstat.times_max_sleep_used;        /* number of times the max_possible_sleep was used to sleep */ 
	    engstat->total_sleepers = schedstat.total_sleepers;                    /* total number of times a client slept for ydb lock scheduling */ 
	    engstat->total_sleep_time = schedstat.total_sleep_time;                /* total time spent sleeping for ydb lock scheduling */ 
	    engstat->max_waiters = schedstat.max_waiters;                          /* max number of simultaneous client threads kept waiting for ydb lock  */ 
	    engstat->total_waiters = schedstat.total_waiters;                      /* total number of times a client thread waited for ydb lock  */ 
	    engstat->total_clients = schedstat.total_clients;                      /* total number of separate client threads that use ydb lock  */ 
	    engstat->time_ydb_lock_held_unavailable = schedstat.time_ydb_lock_held_unavailable;  /* number of times a thread migrated and theld is unavailable */ 
	    engstat->total_time_ydb_lock_held = schedstat.total_time_ydb_lock_held;/* total time client threads held the ydb lock  */ 
	    engstat->max_time_ydb_lock_held = schedstat.max_time_ydb_lock_held;    /* max time client threads held the ydb lock  */ 
	}

1706
	engstat->checkpoint_period = toku_get_checkpoint_period_unlocked(env->i->cachetable);  // do not take any locks (not even minicron lock)
1707 1708 1709 1710 1711 1712 1713
	{
            CHECKPOINT_STATUS_S cpstat;
            toku_checkpoint_get_status(&cpstat);
            engstat->checkpoint_footprint = cpstat.footprint;
	    format_time(&cpstat.time_last_checkpoint_begin_complete, engstat->checkpoint_time_begin_complete);
	    format_time(&cpstat.time_last_checkpoint_begin,          engstat->checkpoint_time_begin);
	    format_time(&cpstat.time_last_checkpoint_end,            engstat->checkpoint_time_end);
1714 1715 1716 1717 1718 1719 1720 1721 1722 1723 1724 1725 1726 1727 1728 1729 1730 1731 1732 1733 1734 1735
	    engstat->checkpoint_last_lsn   = cpstat.last_lsn;
	    engstat->checkpoint_count      = cpstat.checkpoint_count;
	    engstat->checkpoint_count_fail = cpstat.checkpoint_count_fail;
	}
	{
	    TXN_STATUS_S txnstat;
	    toku_txn_get_status(&txnstat);
	    engstat->txn_begin   = txnstat.begin;
	    engstat->txn_commit  = txnstat.commit;
	    engstat->txn_abort   = txnstat.abort;
	    engstat->txn_close   = txnstat.close;
	    {
		uint64_t oldest_xid = 0;
		uint64_t next_lsn   = 0;
		TOKULOGGER logger = env->i->logger;
		if (logger) {
		    oldest_xid = toku_logger_get_oldest_living_xid(env->i->logger);
		    next_lsn   = (toku_logger_get_next_lsn(env->i->logger)).lsn;
		}
		engstat->txn_oldest_live = oldest_xid;
		engstat->next_lsn = next_lsn;
	    }
1736
	}
1737 1738 1739
	{
	    CACHETABLE_STATUS_S ctstat;
	    toku_cachetable_get_status(env->i->cachetable, &ctstat);
1740 1741 1742 1743 1744 1745 1746 1747
	    engstat->cachetable_lock_taken    = ctstat.lock_taken;
	    engstat->cachetable_lock_released = ctstat.lock_released;
	    engstat->cachetable_hit           = ctstat.hit;
	    engstat->cachetable_miss          = ctstat.miss;
	    engstat->cachetable_misstime      = ctstat.misstime;
	    engstat->cachetable_waittime      = ctstat.waittime;
	    engstat->cachetable_wait_reading  = ctstat.wait_reading;
	    engstat->cachetable_wait_writing  = ctstat.wait_writing;
1748
	    engstat->cachetable_wait_checkpoint = ctstat.wait_checkpoint;
1749 1750 1751 1752 1753 1754 1755 1756
	    engstat->puts                     = ctstat.puts;
	    engstat->prefetches               = ctstat.prefetches;
	    engstat->maybe_get_and_pins       = ctstat.maybe_get_and_pins;
	    engstat->maybe_get_and_pin_hits   = ctstat.maybe_get_and_pin_hits;
	    engstat->cachetable_size_current  = ctstat.size_current;
	    engstat->cachetable_size_limit    = ctstat.size_limit;
	    engstat->cachetable_size_writing  = ctstat.size_writing;
	    engstat->get_and_pin_footprint    = ctstat.get_and_pin_footprint;
1757 1758 1759
	    engstat->local_checkpoint         = ctstat.local_checkpoint;
	    engstat->local_checkpoint_files   = ctstat.local_checkpoint_files;
	    engstat->local_checkpoint_during_checkpoint = ctstat.local_checkpoint_during_checkpoint;
1760
	}
1761 1762
	{
	    toku_ltm* ltm = env->i->ltm;
1763
	    LTM_STATUS_S ltmstat;
1764 1765 1766 1767 1768
	    uint32_t max_locks, curr_locks;
	    uint64_t max_lock_memory, curr_lock_memory;
	    toku_ltm_get_status(ltm, &max_locks, &curr_locks, 
				&max_lock_memory, &curr_lock_memory,
				&ltmstat);
1769 1770
	    engstat->range_locks_max                 = max_locks;
	    engstat->range_locks_curr                = curr_locks;
1771 1772
	    engstat->range_locks_max_memory          = max_lock_memory;
	    engstat->range_locks_curr_memory         = curr_lock_memory;
1773 1774
	    engstat->range_lock_escalation_successes = ltmstat.lock_escalation_successes;
	    engstat->range_lock_escalation_failures  = ltmstat.lock_escalation_failures;
1775 1776 1777 1778 1779 1780
	    engstat->range_read_locks                = ltmstat.read_lock;
	    engstat->range_read_locks_fail           = ltmstat.read_lock_fail;
	    engstat->range_out_of_read_locks         = ltmstat.out_of_read_locks;
	    engstat->range_write_locks               = ltmstat.write_lock;
	    engstat->range_write_locks_fail          = ltmstat.write_lock_fail;
	    engstat->range_out_of_write_locks        = ltmstat.out_of_write_locks;
1781
	}
1782
	{
1783
     	    engstat->inserts            = num_inserts;
1784
	    engstat->inserts_fail       = num_inserts_fail;
1785
	    engstat->deletes            = num_deletes;
1786
	    engstat->deletes_fail       = num_deletes_fail;
1787 1788 1789 1790 1791 1792 1793 1794
	    engstat->updates            = num_updates;
	    engstat->updates_fail       = num_updates_fail;
     	    engstat->multi_inserts      = num_multi_inserts;
	    engstat->multi_inserts_fail = num_multi_inserts_fail;
	    engstat->multi_deletes      = num_multi_deletes;
	    engstat->multi_deletes_fail = num_multi_deletes_fail;
	    engstat->multi_updates      = num_multi_updates;
	    engstat->multi_updates_fail = num_multi_updates_fail;
1795 1796 1797
	    engstat->point_queries      = num_point_queries;
	    engstat->sequential_queries = num_sequential_queries;
	}
1798 1799 1800 1801 1802 1803
	{
	    u_int64_t fsync_count, fsync_time;
	    toku_get_fsync_times(&fsync_count, &fsync_time);
	    engstat->fsync_count = fsync_count;
	    engstat->fsync_time  = fsync_time;
	}
1804 1805 1806 1807 1808 1809 1810
	{
	    LOGGER_STATUS_S log_stat;
	    TOKULOGGER logger = env->i->logger;
	    toku_logger_get_status(logger, &log_stat);
	    engstat->logger_ilock_ctr = log_stat.ilock_ctr;
	    engstat->logger_olock_ctr = log_stat.olock_ctr;
	    engstat->logger_swap_ctr  = log_stat.swap_ctr;
1811 1812
	    engstat->logger_panic     = log_stat.panicked;
	    engstat->logger_panic_errno = log_stat.panic_errno;
1813
	}
1814 1815
	{
	    time_t    enospc_most_recent_timestamp;
1816 1817
	    u_int64_t enospc_threads_blocked, enospc_ctr;
	    toku_fs_get_write_info(&enospc_most_recent_timestamp, &enospc_threads_blocked, &enospc_ctr);
1818 1819
	    format_time(&enospc_most_recent_timestamp, engstat->enospc_most_recent);	    
	    engstat->enospc_threads_blocked = enospc_threads_blocked;
1820
	    engstat->enospc_ctr = enospc_ctr;
1821
	}
1822
	{
1823 1824 1825 1826 1827 1828 1829 1830 1831 1832 1833 1834 1835 1836 1837 1838 1839
	    engstat->enospc_redzone_ctr   = env->i->enospc_redzone_ctr;   // number of operations rejected by enospc prevention (red zone)
	    engstat->enospc_state         = env->i->fs_state;
	}
	{
	    LOADER_STATUS_S loader_stat;
	    toku_loader_get_status(&loader_stat);
	    engstat->loader_create         = loader_stat.create;
	    engstat->loader_create_fail    = loader_stat.create_fail;
	    engstat->loader_put            = loader_stat.put;
	    engstat->loader_close          = loader_stat.close;
	    engstat->loader_close_fail     = loader_stat.close_fail;
	    engstat->loader_abort          = loader_stat.abort;
	    engstat->loader_current        = loader_stat.current;
	    engstat->loader_max            = loader_stat.max;
	    
	    engstat->logsuppress     = logsuppress;
	    engstat->logsuppressfail = logsuppressfail;
1840
	}
1841 1842 1843 1844 1845 1846 1847 1848 1849 1850 1851 1852 1853
	{
	    INDEXER_STATUS_S indexer_stat;
	    toku_indexer_get_status(&indexer_stat);
	    engstat->indexer_create         = indexer_stat.create;
	    engstat->indexer_create_fail    = indexer_stat.create_fail;
	    engstat->indexer_build          = indexer_stat.build;
	    engstat->indexer_build_fail     = indexer_stat.build_fail;
	    engstat->indexer_close          = indexer_stat.close;
	    engstat->indexer_close_fail     = indexer_stat.close_fail;
	    engstat->indexer_abort          = indexer_stat.abort;
	    engstat->indexer_current        = indexer_stat.current;
	    engstat->indexer_max            = indexer_stat.max;
	}
1854
	{
1855 1856 1857 1858
	    BRT_UPGRADE_STATUS_S brt_upgrade_stat;
	    toku_brt_get_upgrade_status(&brt_upgrade_stat);

	    engstat->upgrade_env_status = toku_log_upgrade_get_footprint();
1859 1860 1861
	    engstat->upgrade_header     = brt_upgrade_stat.header_12;
	    engstat->upgrade_nonleaf    = brt_upgrade_stat.nonleaf_12;
	    engstat->upgrade_leaf       = brt_upgrade_stat.leaf_12;
1862 1863 1864 1865
	    engstat->original_ver       = persistent_original_env_version;
	    engstat->ver_at_startup     = persistent_stored_env_version_at_startup;
	    engstat->last_lsn_v12       = persistent_last_lsn_of_v12;
	    format_time(&persistent_upgrade_v13_time, engstat->upgrade_v13_time);
1866
	}
1867 1868 1869 1870
    }
    return r;
}

1871

1872
// Fill buff with text description of engine status up to bufsiz bytes.
1873 1874
// Intended for use by test programs that do not have the handlerton available,
// and for use by toku_assert logic to print diagnostic info on crash.
1875 1876 1877
static int
env_get_engine_status_text(DB_ENV * env, char * buff, int bufsiz) {
    ENGINE_STATUS engstat;
Zardosht Kasheff's avatar
Zardosht Kasheff committed
1878
    uint32_t stringsize = 80;
1879
    char panicstring[stringsize];
1880 1881
    int n = 0;  // number of characters printed so far

1882 1883 1884 1885 1886 1887 1888
    int r = env_get_engine_status(env, &engstat, panicstring, stringsize);    

    if (strlen(panicstring)) {
	invariant(strlen(panicstring) <= stringsize);
	n += snprintf(buff + n, bufsiz - n, "Env panic: %s\n", panicstring);
    }

1889 1890 1891 1892 1893 1894 1895 1896 1897 1898 1899 1900 1901
    if (r) {
	n += snprintf(buff + n, bufsiz - n, "Engine status not available: ");
	if (!env) {
	    n += snprintf(buff + n, bufsiz - n, "no environment\n");
	}
	else if (!(env->i)) {
	    n += snprintf(buff + n, bufsiz - n, "environment internal struct is null\n");
	}
	else if (!env_opened(env)) {
	    n += snprintf(buff + n, bufsiz - n, "environment is not open\n");
	}
    }
    else {
1902
	n += snprintf(buff + n, bufsiz - n, "env panic                        %"PRIu64"\n", engstat.env_panic);
1903 1904 1905 1906 1907 1908 1909 1910 1911 1912 1913 1914 1915 1916 1917 1918 1919 1920 1921 1922 1923 1924 1925 1926 1927 1928 1929 1930 1931 1932 1933 1934 1935 1936 1937 1938 1939 1940 1941 1942 1943 1944 1945 1946 1947 1948 1949 1950 1951 1952 1953 1954 1955 1956 1957 1958 1959 1960 1961 1962 1963 1964 1965 1966 1967
	n += snprintf(buff + n, bufsiz - n, "creationtime                     %s \n", engstat.creationtime);
	n += snprintf(buff + n, bufsiz - n, "startuptime                      %s \n", engstat.startuptime);
	n += snprintf(buff + n, bufsiz - n, "now                              %s \n", engstat.now);
	n += snprintf(buff + n, bufsiz - n, "ydb_lock_ctr                     %"PRIu64"\n", engstat.ydb_lock_ctr);
	n += snprintf(buff + n, bufsiz - n, "max_possible_sleep               %"PRIu64"\n", engstat.max_possible_sleep);
	n += snprintf(buff + n, bufsiz - n, "processor_freq_mhz               %"PRIu64"\n", engstat.processor_freq_mhz);
	n += snprintf(buff + n, bufsiz - n, "max_requested_sleep              %"PRIu64"\n", engstat.max_requested_sleep);
	n += snprintf(buff + n, bufsiz - n, "times_max_sleep_used             %"PRIu64"\n", engstat.times_max_sleep_used);
	n += snprintf(buff + n, bufsiz - n, "total_sleepers                   %"PRIu64"\n", engstat.total_sleepers);
	n += snprintf(buff + n, bufsiz - n, "total_sleep_time                 %"PRIu64"\n", engstat.total_sleep_time);
	n += snprintf(buff + n, bufsiz - n, "max_waiters                      %"PRIu64"\n", engstat.max_waiters);
	n += snprintf(buff + n, bufsiz - n, "total_waiters                    %"PRIu64"\n", engstat.total_waiters);
	n += snprintf(buff + n, bufsiz - n, "total_clients                    %"PRIu64"\n", engstat.total_clients);
	n += snprintf(buff + n, bufsiz - n, "time_ydb_lock_held_unavailable   %"PRIu64"\n", engstat.time_ydb_lock_held_unavailable);
	n += snprintf(buff + n, bufsiz - n, "max_time_ydb_lock_held           %"PRIu64"\n", engstat.max_time_ydb_lock_held);
	n += snprintf(buff + n, bufsiz - n, "total_time_ydb_lock_held         %"PRIu64"\n", engstat.total_time_ydb_lock_held);
	n += snprintf(buff + n, bufsiz - n, "checkpoint_period                %d \n", engstat.checkpoint_period);
	n += snprintf(buff + n, bufsiz - n, "checkpoint_footprint             %d \n", engstat.checkpoint_footprint);
	n += snprintf(buff + n, bufsiz - n, "checkpoint_time_begin            %s \n", engstat.checkpoint_time_begin);
	n += snprintf(buff + n, bufsiz - n, "checkpoint_time_begin_complete   %s \n", engstat.checkpoint_time_begin_complete);
	n += snprintf(buff + n, bufsiz - n, "checkpoint_time_end              %s \n", engstat.checkpoint_time_end);
	n += snprintf(buff + n, bufsiz - n, "checkpoint_last_lsn              %"PRIu64"\n", engstat.checkpoint_last_lsn);
	n += snprintf(buff + n, bufsiz - n, "checkpoint_count                 %"PRIu32"\n", engstat.checkpoint_count);
	n += snprintf(buff + n, bufsiz - n, "checkpoint_count_fail            %"PRIu32"\n", engstat.checkpoint_count_fail);
	n += snprintf(buff + n, bufsiz - n, "txn_begin                        %"PRIu64"\n", engstat.txn_begin);
	n += snprintf(buff + n, bufsiz - n, "txn_commit                       %"PRIu64"\n", engstat.txn_commit);
	n += snprintf(buff + n, bufsiz - n, "txn_abort                        %"PRIu64"\n", engstat.txn_abort);
	n += snprintf(buff + n, bufsiz - n, "txn_close                        %"PRIu64"\n", engstat.txn_close);
	n += snprintf(buff + n, bufsiz - n, "txn_oldest_live                  %"PRIu64"\n", engstat.txn_oldest_live);
	n += snprintf(buff + n, bufsiz - n, "next_lsn                         %"PRIu64"\n", engstat.next_lsn);
	n += snprintf(buff + n, bufsiz - n, "cachetable_lock_taken            %"PRIu64"\n", engstat.cachetable_lock_taken);
	n += snprintf(buff + n, bufsiz - n, "cachetable_lock_released         %"PRIu64"\n", engstat.cachetable_lock_released);
	n += snprintf(buff + n, bufsiz - n, "cachetable_hit                   %"PRIu64"\n", engstat.cachetable_hit);
	n += snprintf(buff + n, bufsiz - n, "cachetable_miss                  %"PRIu64"\n", engstat.cachetable_miss);
	n += snprintf(buff + n, bufsiz - n, "cachetable_misstime              %"PRIu64"\n", engstat.cachetable_misstime);
	n += snprintf(buff + n, bufsiz - n, "cachetable_waittime              %"PRIu64"\n", engstat.cachetable_waittime);
	n += snprintf(buff + n, bufsiz - n, "cachetable_wait_reading          %"PRIu64"\n", engstat.cachetable_wait_reading);
	n += snprintf(buff + n, bufsiz - n, "cachetable_wait_writing          %"PRIu64"\n", engstat.cachetable_wait_writing);
	n += snprintf(buff + n, bufsiz - n, "puts                             %"PRIu64"\n", engstat.puts);
	n += snprintf(buff + n, bufsiz - n, "prefetches                       %"PRIu64"\n", engstat.prefetches);
	n += snprintf(buff + n, bufsiz - n, "maybe_get_and_pins               %"PRIu64"\n", engstat.maybe_get_and_pins);
	n += snprintf(buff + n, bufsiz - n, "maybe_get_and_pin_hits           %"PRIu64"\n", engstat.maybe_get_and_pin_hits);
	n += snprintf(buff + n, bufsiz - n, "cachetable_size_current          %"PRId64"\n", engstat.cachetable_size_current);
	n += snprintf(buff + n, bufsiz - n, "cachetable_size_limit            %"PRId64"\n", engstat.cachetable_size_limit);
	n += snprintf(buff + n, bufsiz - n, "cachetable_size_writing          %"PRId64"\n", engstat.cachetable_size_writing);
	n += snprintf(buff + n, bufsiz - n, "get_and_pin_footprint            %"PRId64"\n", engstat.get_and_pin_footprint);
	n += snprintf(buff + n, bufsiz - n, "local_checkpoint                 %"PRId64"\n", engstat.local_checkpoint);
	n += snprintf(buff + n, bufsiz - n, "local_checkpoint_files           %"PRId64"\n", engstat.local_checkpoint_files);
	n += snprintf(buff + n, bufsiz - n, "local_checkpoint_during_checkpoint  %"PRId64"\n", engstat.local_checkpoint_during_checkpoint);
	n += snprintf(buff + n, bufsiz - n, "range_locks_max                  %"PRIu32"\n", engstat.range_locks_max);
	n += snprintf(buff + n, bufsiz - n, "range_locks_curr                 %"PRIu32"\n", engstat.range_locks_curr);
	n += snprintf(buff + n, bufsiz - n, "range_locks_max_memory           %"PRIu64"\n", engstat.range_locks_max_memory);
	n += snprintf(buff + n, bufsiz - n, "range_locks_curr_memory          %"PRIu64"\n", engstat.range_locks_curr_memory);
	n += snprintf(buff + n, bufsiz - n, "range_locks_escalation_successes %"PRIu32"\n", engstat.range_lock_escalation_successes);
	n += snprintf(buff + n, bufsiz - n, "range_locks_escalation_failures  %"PRIu32"\n", engstat.range_lock_escalation_failures);
	n += snprintf(buff + n, bufsiz - n, "range_read_locks                 %"PRIu64"\n", engstat.range_read_locks);
	n += snprintf(buff + n, bufsiz - n, "range_read_locks_fail            %"PRIu64"\n", engstat.range_read_locks_fail);
	n += snprintf(buff + n, bufsiz - n, "range_out_of_read_locks          %"PRIu64"\n", engstat.range_out_of_read_locks);
	n += snprintf(buff + n, bufsiz - n, "range_write_locks                %"PRIu64"\n", engstat.range_write_locks);
	n += snprintf(buff + n, bufsiz - n, "range_write_locks_fail           %"PRIu64"\n", engstat.range_write_locks_fail);
	n += snprintf(buff + n, bufsiz - n, "range_out_of_write_locks         %"PRIu64"\n", engstat.range_out_of_write_locks);
	n += snprintf(buff + n, bufsiz - n, "inserts                          %"PRIu64"\n", engstat.inserts);
	n += snprintf(buff + n, bufsiz - n, "inserts_fail                     %"PRIu64"\n", engstat.inserts_fail);
	n += snprintf(buff + n, bufsiz - n, "deletes                          %"PRIu64"\n", engstat.deletes);
	n += snprintf(buff + n, bufsiz - n, "deletes_fail                     %"PRIu64"\n", engstat.deletes_fail);
1968 1969 1970 1971 1972 1973 1974 1975
	n += snprintf(buff + n, bufsiz - n, "updates                          %"PRIu64"\n", engstat.updates);
	n += snprintf(buff + n, bufsiz - n, "updates_fail                     %"PRIu64"\n", engstat.updates_fail);
	n += snprintf(buff + n, bufsiz - n, "multi_inserts                    %"PRIu64"\n", engstat.multi_inserts);
	n += snprintf(buff + n, bufsiz - n, "multi_inserts_fail               %"PRIu64"\n", engstat.multi_inserts_fail);
	n += snprintf(buff + n, bufsiz - n, "multi_deletes                    %"PRIu64"\n", engstat.multi_deletes);
	n += snprintf(buff + n, bufsiz - n, "multi_deletes_fail               %"PRIu64"\n", engstat.multi_deletes_fail);
	n += snprintf(buff + n, bufsiz - n, "multi_updates                    %"PRIu64"\n", engstat.multi_updates);
	n += snprintf(buff + n, bufsiz - n, "multi_updates_fail               %"PRIu64"\n", engstat.multi_updates_fail);
1976 1977 1978 1979 1980 1981 1982
	n += snprintf(buff + n, bufsiz - n, "point_queries                    %"PRIu64"\n", engstat.point_queries);
	n += snprintf(buff + n, bufsiz - n, "sequential_queries               %"PRIu64"\n", engstat.sequential_queries);
	n += snprintf(buff + n, bufsiz - n, "fsync_count                      %"PRIu64"\n", engstat.fsync_count);
	n += snprintf(buff + n, bufsiz - n, "fsync_time                       %"PRIu64"\n", engstat.fsync_time);
	n += snprintf(buff + n, bufsiz - n, "logger ilock count               %"PRIu64"\n", engstat.logger_ilock_ctr);
	n += snprintf(buff + n, bufsiz - n, "logger olock count               %"PRIu64"\n", engstat.logger_olock_ctr);
	n += snprintf(buff + n, bufsiz - n, "logger swap count                %"PRIu64"\n", engstat.logger_swap_ctr);
1983 1984
	n += snprintf(buff + n, bufsiz - n, "logger panic                     %"PRIu64"\n", engstat.logger_panic);
	n += snprintf(buff + n, bufsiz - n, "logger panic_errno               %"PRIu64"\n", engstat.logger_panic_errno);
1985 1986 1987 1988 1989 1990
	n += snprintf(buff + n, bufsiz - n, "enospc_most_recent               %s \n", engstat.enospc_most_recent);
	n += snprintf(buff + n, bufsiz - n, "enospc threads blocked           %"PRIu64"\n", engstat.enospc_threads_blocked);
	n += snprintf(buff + n, bufsiz - n, "enospc count                     %"PRIu64"\n", engstat.enospc_ctr);
	n += snprintf(buff + n, bufsiz - n, "enospc redzone ctr               %"PRIu64"\n", engstat.enospc_redzone_ctr);
	n += snprintf(buff + n, bufsiz - n, "enospc state                     %"PRIu64"\n", engstat.enospc_state);
	n += snprintf(buff + n, bufsiz - n, "loader_create                    %"PRIu64"\n", engstat.loader_create);
1991
	n += snprintf(buff + n, bufsiz - n, "loader_create_fail               %"PRIu64"\n", engstat.loader_create_fail);
1992 1993 1994 1995 1996 1997 1998 1999
	n += snprintf(buff + n, bufsiz - n, "loader_put                       %"PRIu64"\n", engstat.loader_put);
	n += snprintf(buff + n, bufsiz - n, "loader_close                     %"PRIu64"\n", engstat.loader_close);
	n += snprintf(buff + n, bufsiz - n, "loader_close_fail                %"PRIu64"\n", engstat.loader_close_fail);
	n += snprintf(buff + n, bufsiz - n, "loader_abort                     %"PRIu64"\n", engstat.loader_abort);
	n += snprintf(buff + n, bufsiz - n, "loader_current                   %"PRIu32"\n", engstat.loader_current);
	n += snprintf(buff + n, bufsiz - n, "loader_max                       %"PRIu32"\n", engstat.loader_max);
	n += snprintf(buff + n, bufsiz - n, "logsuppress                      %"PRIu64"\n", engstat.logsuppress);
	n += snprintf(buff + n, bufsiz - n, "logsuppressfail                  %"PRIu64"\n", engstat.logsuppressfail);
2000 2001 2002 2003 2004 2005 2006 2007 2008
	n += snprintf(buff + n, bufsiz - n, "indexer_create                   %"PRIu64"\n", engstat.indexer_create);
	n += snprintf(buff + n, bufsiz - n, "indexer_create_fail              %"PRIu64"\n", engstat.indexer_create_fail);
	n += snprintf(buff + n, bufsiz - n, "indexer_build                    %"PRIu64"\n", engstat.indexer_build);
	n += snprintf(buff + n, bufsiz - n, "indexer_build_fail               %"PRIu64"\n", engstat.indexer_build_fail);
	n += snprintf(buff + n, bufsiz - n, "indexer_close                    %"PRIu64"\n", engstat.indexer_close);
	n += snprintf(buff + n, bufsiz - n, "indexer_close_fail               %"PRIu64"\n", engstat.indexer_close_fail);
	n += snprintf(buff + n, bufsiz - n, "indexer_abort                    %"PRIu64"\n", engstat.indexer_abort);
	n += snprintf(buff + n, bufsiz - n, "indexer_current                  %"PRIu32"\n", engstat.indexer_current);
	n += snprintf(buff + n, bufsiz - n, "indexer_max                      %"PRIu32"\n", engstat.indexer_max);
2009 2010 2011 2012 2013 2014 2015 2016 2017
	n += snprintf(buff + n, bufsiz - n, "upgrade_env_status               %"PRIu64"\n", engstat.upgrade_env_status);
	n += snprintf(buff + n, bufsiz - n, "upgrade_header                   %"PRIu64"\n", engstat.upgrade_header);
	n += snprintf(buff + n, bufsiz - n, "upgrade_nonleaf                  %"PRIu64"\n", engstat.upgrade_nonleaf);
	n += snprintf(buff + n, bufsiz - n, "upgrade_leaf                     %"PRIu64"\n", engstat.upgrade_leaf);
	n += snprintf(buff + n, bufsiz - n, "original_ver                     %"PRIu64"\n", engstat.original_ver);
	n += snprintf(buff + n, bufsiz - n, "ver_at_startup                   %"PRIu64"\n", engstat.ver_at_startup);
	n += snprintf(buff + n, bufsiz - n, "last_lsn_v12                     %"PRIu64"\n", engstat.last_lsn_v12);
	n += snprintf(buff + n, bufsiz - n, "upgrade_v13_time                 %s \n", engstat.upgrade_v13_time);
    }
2018 2019 2020
    if (n > bufsiz) {
	char * errmsg = "BUFFER TOO SMALL\n";
	int len = strlen(errmsg) + 1;
2021
	(void) snprintf(buff + (bufsiz - 1) - len, len, "%s", errmsg);
2022 2023 2024 2025 2026
    }

    return r;
}

2027 2028 2029 2030 2031 2032 2033 2034 2035 2036 2037 2038 2039 2040 2041
static int toku_maybe_get_engine_status_text (char* buff, int buffsize);

// assign value to global pointer so that other files can access via tentative definition if linked to this library (.so)
int (*toku_maybe_get_engine_status_text_p)(char* buff, int buffsize) = toku_maybe_get_engine_status_text;

// intended for use by toku_assert logic, when env is not known
static int 
toku_maybe_get_engine_status_text (char * buff, int buffsize) {
    DB_ENV * env = most_recent_env;
    int r = env_get_engine_status_text(env, buff, buffsize);
    return r;
}



Rich Prohaska's avatar
Rich Prohaska committed
2042 2043
static int locked_txn_begin(DB_ENV * env, DB_TXN * stxn, DB_TXN ** txn, u_int32_t flags);

Yoni Fogel's avatar
Yoni Fogel committed
2044 2045 2046 2047
static int toku_db_lt_panic(DB* db, int r);

static toku_dbt_cmp toku_db_get_compare_fun(DB* db);

2048 2049
static int 
toku_env_create(DB_ENV ** envp, u_int32_t flags) {
Yoni Fogel's avatar
Yoni Fogel committed
2050 2051 2052 2053 2054 2055
    int r = ENOSYS;
    DB_ENV* result = NULL;

    if (flags!=0)    { r = EINVAL; goto cleanup; }
    MALLOC(result);
    if (result == 0) { r = ENOMEM; goto cleanup; }
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
2056
    memset(result, 0, sizeof *result);
2057
    result->err = (void (*)(const DB_ENV * env, int error, const char *fmt, ...)) toku_locked_env_err;
2058 2059 2060 2061
#define SENV(name) result->name = locked_env_ ## name
    SENV(dbremove);
    SENV(dbrename);
    SENV(set_default_bt_compare);
2062 2063
    SENV(set_generate_row_callback_for_put);
    SENV(set_generate_row_callback_for_del);
2064 2065
    SENV(put_multiple);
    SENV(del_multiple);
2066
    SENV(update_multiple);
2067 2068
    SENV(checkpointing_set_period);
    SENV(checkpointing_get_period);
2069 2070 2071 2072
    result->checkpointing_postpone = env_checkpointing_postpone;
    result->checkpointing_resume = env_checkpointing_resume;
    result->checkpointing_begin_atomic_operation = env_checkpointing_begin_atomic_operation;
    result->checkpointing_end_atomic_operation = env_checkpointing_end_atomic_operation;
2073
    result->get_engine_status = env_get_engine_status;
2074
    result->get_engine_status_text = env_get_engine_status_text;
2075 2076 2077
    result->get_iname = env_get_iname;
    SENV(open);
    SENV(close);
2078
    result->txn_checkpoint = toku_env_txn_checkpoint;
2079
    SENV(log_flush);
Rich Prohaska's avatar
Rich Prohaska committed
2080 2081 2082
    result->set_errcall = toku_env_set_errcall;
    result->set_errfile = toku_env_set_errfile;
    result->set_errpfx = toku_env_set_errpfx;
2083 2084 2085 2086 2087 2088 2089 2090 2091 2092 2093
    //SENV(set_noticecall);
    SENV(set_flags);
    SENV(set_data_dir);
    SENV(set_tmp_dir);
    SENV(set_verbose);
    SENV(set_lg_bsize);
    SENV(set_lg_dir);
    SENV(set_lg_max);
    SENV(get_lg_max);
    SENV(set_lk_max_locks);
    SENV(get_lk_max_locks);
2094 2095
    SENV(set_lk_max_memory);
    SENV(get_lk_max_memory);
2096
    SENV(set_cachesize);
Rich Prohaska's avatar
Rich Prohaska committed
2097
#if DB_VERSION_MAJOR == 4 && DB_VERSION_MINOR >= 3
2098
    SENV(get_cachesize);
Rich Prohaska's avatar
Rich Prohaska committed
2099
#endif
2100
    SENV(set_lk_detect);
2101
#if DB_VERSION_MAJOR == 4 && DB_VERSION_MINOR <= 4
2102
    SENV(set_lk_max);
2103
#endif
2104 2105
    SENV(log_archive);
    SENV(txn_stat);
Rich Prohaska's avatar
Rich Prohaska committed
2106
    result->txn_begin = locked_txn_begin;
2107
    SENV(set_redzone);
2108
#undef SENV
2109
    result->create_loader = toku_loader_create_loader;
2110
    result->create_indexer = toku_indexer_create_indexer;
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
2111

2112
    MALLOC(result->i);
Yoni Fogel's avatar
Yoni Fogel committed
2113
    if (result->i == 0) { r = ENOMEM; goto cleanup; }
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
2114
    memset(result->i, 0, sizeof *result->i);
Yoni Fogel's avatar
Yoni Fogel committed
2115 2116 2117 2118
    result->i->envdir_lockfd  = -1;
    result->i->datadir_lockfd = -1;
    result->i->logdir_lockfd  = -1;
    result->i->tmpdir_lockfd  = -1;
2119
    env_init_open_txn(result);
2120
    env_fs_init(result);
Yoni Fogel's avatar
Yoni Fogel committed
2121

2122 2123
    r = toku_ltm_create(&result->i->ltm,
                        __toku_env_default_max_locks, __toku_env_default_max_lock_memory,
Yoni Fogel's avatar
Yoni Fogel committed
2124
                         toku_db_lt_panic, 
2125
                         toku_db_get_compare_fun,
Yoni Fogel's avatar
Yoni Fogel committed
2126
                         toku_malloc, toku_free, toku_realloc);
Yoni Fogel's avatar
Yoni Fogel committed
2127
    if (r!=0) { goto cleanup; }
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
2128

2129
    {
Yoni Fogel's avatar
Yoni Fogel committed
2130 2131
	r = toku_logger_create(&result->i->logger);
	if (r!=0) { goto cleanup; }
2132 2133
	assert(result->i->logger);
    }
2134 2135 2136 2137 2138
    {
        r = toku_omt_create(&result->i->open_dbs);
        if (r!=0) goto cleanup;
        assert(result->i->open_dbs);
    }
2139

Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
2140
    *envp = result;
Yoni Fogel's avatar
Yoni Fogel committed
2141 2142 2143 2144 2145 2146 2147 2148
    r = 0;
cleanup:
    if (r!=0) {
        if (result) {
            if (result->i) {
                if (result->i->ltm) {
                    toku_ltm_close(result->i->ltm);
                }
2149 2150
                if (result->i->open_dbs)
                    toku_omt_destroy(&result->i->open_dbs);
Yoni Fogel's avatar
Yoni Fogel committed
2151 2152 2153 2154 2155 2156
                toku_free(result->i);
            }
            toku_free(result);
        }
    }
    return r;
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
2157 2158
}

2159 2160
int 
DB_ENV_CREATE_FUN (DB_ENV ** envp, u_int32_t flags) {
2161 2162 2163 2164
    toku_ydb_lock(); 
    int r = toku_env_create(envp, flags); 
    toku_ydb_unlock(); 
    return r;
Rich Prohaska's avatar
Rich Prohaska committed
2165 2166
}

2167 2168
static int 
toku_txn_release_locks(DB_TXN* txn) {
Yoni Fogel's avatar
Yoni Fogel committed
2169
    assert(txn);
2170
    toku_lth* lth = db_txn_struct_i(txn)->lth;
Yoni Fogel's avatar
Yoni Fogel committed
2171

Yoni Fogel's avatar
Yoni Fogel committed
2172 2173
    int r = ENOSYS;
    int first_error = 0;
Yoni Fogel's avatar
Yoni Fogel committed
2174 2175 2176 2177
    if (lth) {
        toku_lth_start_scan(lth);
        toku_lock_tree* next = toku_lth_next(lth);
        while (next) {
2178
            r = toku_lt_unlock(next, toku_txn_get_txnid(db_txn_struct_i(txn)->tokutxn));
Yoni Fogel's avatar
Yoni Fogel committed
2179 2180 2181 2182 2183
            if (!first_error && r!=0) { first_error = r; }
            if (r == 0) {
                r = toku_lt_remove_ref(next);
                if (!first_error && r!=0) { first_error = r; }
            }
Yoni Fogel's avatar
Yoni Fogel committed
2184 2185 2186
            next = toku_lth_next(lth);
        }
        toku_lth_close(lth);
2187
        db_txn_struct_i(txn)->lth = NULL;
Yoni Fogel's avatar
Yoni Fogel committed
2188
    }
Yoni Fogel's avatar
Yoni Fogel committed
2189 2190
    r = first_error;

Yoni Fogel's avatar
Yoni Fogel committed
2191
    return r;
Yoni Fogel's avatar
Yoni Fogel committed
2192 2193
}

2194 2195
// Yield the lock so someone else can work, and then reacquire the lock.
// Useful while processing commit or rollback logs, to allow others to access the system.
2196 2197
static void 
ydb_yield (voidfp f, void *fv, void *UU(v)) {
2198
    toku_ydb_unlock(); 
2199
    if (f) f(fv);
2200 2201 2202
    toku_ydb_lock();
}

2203 2204 2205
static int 
toku_txn_commit(DB_TXN * txn, u_int32_t flags,
                TXN_PROGRESS_POLL_FUNCTION poll, void* poll_extra) {
2206
    if (!txn) return EINVAL;
2207
    HANDLE_PANICKED_ENV(txn->mgrp);
Yoni Fogel's avatar
Yoni Fogel committed
2208
    //Recursively kill off children
2209 2210
    if (db_txn_struct_i(txn)->child) {
        //commit of child sets the child pointer to NULL
2211
        int r_child = toku_txn_commit(db_txn_struct_i(txn)->child, flags, NULL, NULL);
2212 2213 2214 2215
        if (r_child !=0 && !toku_env_is_panicked(txn->mgrp)) {
            txn->mgrp->i->is_panicked = r_child;
            txn->mgrp->i->panic_string = toku_strdup("Recursive child commit failed during parent commit.\n");
        }
Yoni Fogel's avatar
Yoni Fogel committed
2216 2217
        //In a panicked env, the child may not be removed from the list.
        HANDLE_PANICKED_ENV(txn->mgrp);
Yoni Fogel's avatar
Yoni Fogel committed
2218
    }
2219
    assert(!db_txn_struct_i(txn)->child);
Yoni Fogel's avatar
Yoni Fogel committed
2220 2221
    //Remove from parent
    if (txn->parent) {
2222 2223
        assert(db_txn_struct_i(txn->parent)->child == txn);
        db_txn_struct_i(txn->parent)->child=NULL;
Yoni Fogel's avatar
Yoni Fogel committed
2224
    }
2225
    env_remove_open_txn(txn->mgrp, txn);
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
2226
    //toku_ydb_notef("flags=%d\n", flags);
2227 2228 2229 2230
    if (flags & DB_TXN_SYNC) {
        toku_txn_force_fsync_on_commit(db_txn_struct_i(txn)->tokutxn);
        flags &= ~DB_TXN_SYNC;
    }
2231
    int nosync = (flags & DB_TXN_NOSYNC)!=0 || (db_txn_struct_i(txn)->flags&DB_TXN_NOSYNC);
2232
    flags &= ~DB_TXN_NOSYNC;
Yoni Fogel's avatar
Yoni Fogel committed
2233 2234

    int r;
2235
    if (flags!=0)
2236 2237
	// frees the tokutxn
	// Calls ydb_yield(NULL) occasionally
2238
        //r = toku_logger_abort(db_txn_struct_i(txn)->tokutxn, ydb_yield, NULL);
2239
        r = toku_txn_abort_txn(db_txn_struct_i(txn)->tokutxn, ydb_yield, NULL, poll, poll_extra);
Yoni Fogel's avatar
Yoni Fogel committed
2240
    else
2241 2242
	// frees the tokutxn
	// Calls ydb_yield(NULL) occasionally
2243
        //r = toku_logger_commit(db_txn_struct_i(txn)->tokutxn, nosync, ydb_yield, NULL);
2244
        r = toku_txn_commit_txn(db_txn_struct_i(txn)->tokutxn, nosync, ydb_yield, NULL,
2245
				poll, poll_extra);
2246

2247 2248 2249 2250 2251 2252 2253 2254
    if (r!=0 && !toku_env_is_panicked(txn->mgrp)) {
        txn->mgrp->i->is_panicked = r;
        txn->mgrp->i->panic_string = toku_strdup("Error during commit.\n");
    }
    //If panicked, we're done.
    HANDLE_PANICKED_ENV(txn->mgrp);
    assert(r==0);

2255
    // Close the logger after releasing the locks
2256
    r = toku_txn_release_locks(txn);
2257 2258
    //toku_logger_txn_close(db_txn_struct_i(txn)->tokutxn);
    toku_txn_close_txn(db_txn_struct_i(txn)->tokutxn);
2259
    // the toxutxn is freed, and we must free the rest. */
Yoni Fogel's avatar
Yoni Fogel committed
2260

2261 2262 2263
    //Promote list to parent (dbs that must close before abort)
    if (txn->parent) {
        //Combine lists.
2264 2265 2266
        while (!toku_list_empty(&db_txn_struct_i(txn)->dbs_that_must_close_before_abort)) {
            struct toku_list *list = toku_list_pop(&db_txn_struct_i(txn)->dbs_that_must_close_before_abort);
            toku_list_push(&db_txn_struct_i(txn->parent)->dbs_that_must_close_before_abort, list);
2267 2268 2269 2270
        }
    }
    else {
        //Empty the list
2271 2272
        while (!toku_list_empty(&db_txn_struct_i(txn)->dbs_that_must_close_before_abort)) {
            toku_list_pop(&db_txn_struct_i(txn)->dbs_that_must_close_before_abort);
2273 2274 2275
        }
    }

2276
    // The txn is no good after the commit even if the commit fails, so free it up.
2277 2278 2279
#if !TOKUDB_NATIVE_H
    toku_free(db_txn_struct_i(txn));
#endif
2280
    toku_free(txn);    txn = NULL;
Yoni Fogel's avatar
Yoni Fogel committed
2281
    if (flags!=0) return EINVAL;
2282
    return r;
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
2283 2284
}

2285 2286
static u_int32_t 
toku_txn_id(DB_TXN * txn) {
2287
    HANDLE_PANICKED_ENV(txn->mgrp);
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
2288
    toku_ydb_barf();
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
2289
    abort();
Rich Prohaska's avatar
Rich Prohaska committed
2290
    return -1;
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
2291 2292
}

2293 2294 2295
static int 
toku_txn_abort(DB_TXN * txn,
               TXN_PROGRESS_POLL_FUNCTION poll, void* poll_extra) {
2296
    HANDLE_PANICKED_ENV(txn->mgrp);
2297
    //Recursively kill off children (abort or commit are both correct, commit is cheaper)
2298 2299
    if (db_txn_struct_i(txn)->child) {
        //commit of child sets the child pointer to NULL
2300
        int r_child = toku_txn_commit(db_txn_struct_i(txn)->child, DB_TXN_NOSYNC, NULL, NULL);
2301 2302 2303 2304
        if (r_child !=0 && !toku_env_is_panicked(txn->mgrp)) {
            txn->mgrp->i->is_panicked = r_child;
            txn->mgrp->i->panic_string = toku_strdup("Recursive child commit failed during parent abort.\n");
        }
Yoni Fogel's avatar
Yoni Fogel committed
2305 2306
        //In a panicked env, the child may not be removed from the list.
        HANDLE_PANICKED_ENV(txn->mgrp);
Yoni Fogel's avatar
Yoni Fogel committed
2307
    }
2308
    assert(!db_txn_struct_i(txn)->child);
Yoni Fogel's avatar
Yoni Fogel committed
2309 2310
    //Remove from parent
    if (txn->parent) {
2311 2312
        assert(db_txn_struct_i(txn->parent)->child == txn);
        db_txn_struct_i(txn->parent)->child=NULL;
Yoni Fogel's avatar
Yoni Fogel committed
2313
    }
2314
    env_remove_open_txn(txn->mgrp, txn);
2315 2316

    //All dbs that must close before abort, must now be closed
2317
    assert(toku_list_empty(&db_txn_struct_i(txn)->dbs_that_must_close_before_abort));
2318

2319
    //int r = toku_logger_abort(db_txn_struct_i(txn)->tokutxn, ydb_yield, NULL);
2320
    int r = toku_txn_abort_txn(db_txn_struct_i(txn)->tokutxn, ydb_yield, NULL, poll, poll_extra);
2321 2322 2323 2324 2325 2326 2327
    if (r!=0 && !toku_env_is_panicked(txn->mgrp)) {
        txn->mgrp->i->is_panicked = r;
        txn->mgrp->i->panic_string = toku_strdup("Error during abort.\n");
    }
    HANDLE_PANICKED_ENV(txn->mgrp);
    assert(r==0);
    r = toku_txn_release_locks(txn);
2328 2329
    //toku_logger_txn_close(db_txn_struct_i(txn)->tokutxn);
    toku_txn_close_txn(db_txn_struct_i(txn)->tokutxn);
Yoni Fogel's avatar
Yoni Fogel committed
2330

2331 2332 2333
#if !TOKUDB_NATIVE_H
    toku_free(db_txn_struct_i(txn));
#endif
2334
    toku_free(txn);
2335
    return r;
2336 2337
}

2338 2339
static int 
locked_txn_begin(DB_ENV *env, DB_TXN * stxn, DB_TXN ** txn, u_int32_t flags) {
2340
    toku_ydb_lock(); int r = toku_txn_begin(env, stxn, txn, flags, 0); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
2341 2342
}

2343 2344
static u_int32_t 
locked_txn_id(DB_TXN *txn) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
2345
    toku_ydb_lock(); u_int32_t r = toku_txn_id(txn); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
2346 2347
}

2348 2349
static int 
toku_txn_stat (DB_TXN *txn, struct txn_stat **txn_stat) {
2350
    XMALLOC(*txn_stat);
2351
    return toku_logger_txn_rollback_raw_count(db_txn_struct_i(txn)->tokutxn, &(*txn_stat)->rollback_raw_count);
2352 2353
}

2354 2355
static int 
locked_txn_stat (DB_TXN *txn, struct txn_stat **txn_stat) {
2356 2357 2358
    toku_ydb_lock(); u_int32_t r = toku_txn_stat(txn, txn_stat); toku_ydb_unlock(); return r;
}

2359 2360 2361
static int 
locked_txn_commit_with_progress(DB_TXN *txn, u_int32_t flags,
                                TXN_PROGRESS_POLL_FUNCTION poll, void* poll_extra) {
2362
    toku_multi_operation_client_lock(); //Cannot checkpoint during a commit.
2363
    toku_ydb_lock(); int r = toku_txn_commit(txn, flags, poll, poll_extra); toku_ydb_unlock();
2364
    toku_multi_operation_client_unlock(); //Cannot checkpoint during a commit.
2365
    return r;
Rich Prohaska's avatar
Rich Prohaska committed
2366 2367
}

2368 2369 2370
static int 
locked_txn_abort_with_progress(DB_TXN *txn,
                               TXN_PROGRESS_POLL_FUNCTION poll, void* poll_extra) {
2371
    toku_multi_operation_client_lock(); //Cannot checkpoint during an abort.
2372
    toku_ydb_lock(); int r = toku_txn_abort(txn, poll, poll_extra); toku_ydb_unlock();
2373
    toku_multi_operation_client_unlock(); //Cannot checkpoint during an abort.
2374
    return r;
Rich Prohaska's avatar
Rich Prohaska committed
2375 2376
}

2377 2378
static int 
locked_txn_commit(DB_TXN *txn, u_int32_t flags) {
2379 2380 2381 2382 2383
    int r;
    r = locked_txn_commit_with_progress(txn, flags, NULL, NULL);
    return r;
}

2384 2385
static int 
locked_txn_abort(DB_TXN *txn) {
2386 2387 2388 2389 2390
    int r;
    r = locked_txn_abort_with_progress(txn, NULL, NULL);
    return r;
}

2391 2392
static int 
toku_txn_begin(DB_ENV *env, DB_TXN * stxn, DB_TXN ** txn, u_int32_t flags, int internal) {
2393
    HANDLE_PANICKED_ENV(env);
2394
    HANDLE_ILLEGAL_WORKING_PARENT_TXN(env, stxn); //Cannot create child while child already exists.
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
2395 2396
    if (!toku_logger_is_open(env->i->logger)) return toku_ydb_do_error(env, EINVAL, "Environment does not have logging enabled\n");
    if (!(env->i->open_flags & DB_INIT_TXN))  return toku_ydb_do_error(env, EINVAL, "Environment does not have transactions enabled\n");
2397 2398
    u_int32_t txn_flags = 0;
    txn_flags |= DB_TXN_NOWAIT; //We do not support blocking locks.
2399 2400 2401 2402 2403 2404 2405 2406 2407 2408
    TOKU_ISOLATION child_isolation = 0;
    u_int32_t iso_flags = flags & DB_ISOLATION_FLAGS;
    if (!(iso_flags == 0 || 
          iso_flags == DB_TXN_SNAPSHOT || 
          iso_flags == DB_READ_COMMITTED || 
          iso_flags == DB_READ_UNCOMMITTED || 
          iso_flags == DB_SERIALIZABLE || 
          iso_flags == DB_INHERIT_ISOLATION)
       ) 
    {
2409 2410 2411
        return toku_ydb_do_error(
            env, 
            EINVAL, 
2412
            "Invalid isolation flags set\n"
2413 2414
            );
    }
2415 2416 2417 2418 2419 2420 2421 2422 2423 2424 2425 2426 2427 2428 2429 2430 2431 2432 2433 2434 2435 2436 2437 2438 2439 2440 2441 2442 2443 2444 2445 2446 2447 2448 2449
    flags &= ~iso_flags;

    if (internal && stxn) {
        child_isolation = db_txn_struct_i(stxn)->iso;
    }
    else {
        switch (iso_flags) {
            case (DB_INHERIT_ISOLATION):
                if (stxn) {
                    child_isolation = db_txn_struct_i(stxn)->iso;
                }
                else {
                    return toku_ydb_do_error(
                        env, 
                        EINVAL, 
                        "Cannot set DB_INHERIT_ISOLATION when no parent exists\n"
                        );                    
                }
                break;
            case (DB_READ_COMMITTED):
                child_isolation = TOKU_ISO_READ_COMMITTED;
                break;
            case (DB_READ_UNCOMMITTED):
                child_isolation = TOKU_ISO_READ_UNCOMMITTED;
                break;
            case (DB_TXN_SNAPSHOT):
                child_isolation = TOKU_ISO_SNAPSHOT;
                break;
            case (DB_SERIALIZABLE):
            case (0):
                child_isolation = TOKU_ISO_SERIALIZABLE;
                break;
            default:
                assert(FALSE); // error path is above, so this should not happen
                break;
2450
        }
2451
    }
2452 2453 2454 2455 2456 2457 2458
    if (stxn && child_isolation != db_txn_struct_i(stxn)->iso) {
        return toku_ydb_do_error(
            env, 
            EINVAL, 
            "Cannot set isolation level of transaction to something different \
                isolation level\n"
            );   
2459
    }
2460

2461 2462 2463 2464 2465 2466 2467 2468 2469 2470
    if (flags&DB_TXN_NOWAIT) {
        txn_flags |=  DB_TXN_NOWAIT;
        flags     &= ~DB_TXN_NOWAIT;
    }
    if (flags&DB_TXN_NOSYNC) {
        txn_flags |=  DB_TXN_NOSYNC;
        flags     &= ~DB_TXN_NOSYNC;
    }
    if (flags!=0) return toku_ydb_do_error(env, EINVAL, "Invalid flags passed to DB_ENV->txn_begin\n");

2471 2472
    size_t result_size = sizeof(DB_TXN)+sizeof(struct __toku_db_txn_internal); // the internal stuff is stuck on the end.
    DB_TXN *result = toku_malloc(result_size);
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
2473 2474
    if (result == 0)
        return ENOMEM;
2475
    memset(result, 0, result_size);
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
2476
    //toku_ydb_notef("parent=%p flags=0x%x\n", stxn, flags);
2477
    result->mgrp = env;
2478 2479 2480 2481 2482 2483 2484
#define STXN(name) result->name = locked_txn_ ## name
    STXN(abort);
    STXN(commit);
    STXN(abort_with_progress);
    STXN(commit_with_progress);
    STXN(id);
#undef STXN
2485
    result->txn_stat = locked_txn_stat;
2486 2487 2488


    result->parent = stxn;
2489 2490 2491
#if !TOKUDB_NATIVE_H
    MALLOC(db_txn_struct_i(result));
    if (!db_txn_struct_i(result)) {
Yoni Fogel's avatar
Yoni Fogel committed
2492 2493 2494
        toku_free(result);
        return ENOMEM;
    }
2495 2496 2497
#endif
    memset(db_txn_struct_i(result), 0, sizeof *db_txn_struct_i(result));
    db_txn_struct_i(result)->flags = txn_flags;
2498
    db_txn_struct_i(result)->iso = child_isolation;
2499
    toku_list_init(&db_txn_struct_i(result)->dbs_that_must_close_before_abort);
Yoni Fogel's avatar
Yoni Fogel committed
2500 2501

    int r;
Yoni Fogel's avatar
Yoni Fogel committed
2502
    if (env->i->open_flags & DB_INIT_LOCK && !stxn) {
2503
        r = toku_lth_create(&db_txn_struct_i(result)->lth,
Yoni Fogel's avatar
Yoni Fogel committed
2504 2505
                            toku_malloc, toku_free, toku_realloc);
        if (r!=0) {
2506 2507 2508
#if !TOKUDB_NATIVE_H
            toku_free(db_txn_struct_i(result));
#endif
Yoni Fogel's avatar
Yoni Fogel committed
2509 2510 2511
            toku_free(result);
            return r;
        }
Yoni Fogel's avatar
Yoni Fogel committed
2512 2513
    }
    
2514
    //r = toku_logger_txn_begin(stxn ? db_txn_struct_i(stxn)->tokutxn : 0, &db_txn_struct_i(result)->tokutxn, env->i->logger);
2515 2516 2517 2518 2519 2520 2521 2522 2523 2524 2525 2526 2527 2528 2529 2530 2531 2532
    TXN_SNAPSHOT_TYPE snapshot_type;
    switch(db_txn_struct_i(result)->iso){
        case(TOKU_ISO_SNAPSHOT):
        {
            snapshot_type = TXN_SNAPSHOT_ROOT;
            break;
        }
        case(TOKU_ISO_READ_COMMITTED):
        {
            snapshot_type = TXN_SNAPSHOT_CHILD;
            break;
        }
        default:
        {
            snapshot_type = TXN_SNAPSHOT_NONE;
            break;
        }
    }
2533 2534 2535 2536 2537 2538
    r = toku_txn_begin_txn(result,		   
			   stxn ? db_txn_struct_i(stxn)->tokutxn : 0, 
			   &db_txn_struct_i(result)->tokutxn, 
			   env->i->logger,
			   snapshot_type
			   );
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
2539 2540
    if (r != 0)
        return r;
2541

Yoni Fogel's avatar
Yoni Fogel committed
2542 2543
    //Add to the list of children for the parent.
    if (result->parent) {
2544 2545
        assert(!db_txn_struct_i(result->parent)->child);
        db_txn_struct_i(result->parent)->child = result;
Yoni Fogel's avatar
Yoni Fogel committed
2546
    }
2547
    env_add_open_txn(env, result);
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
2548 2549 2550 2551
    *txn = result;
    return 0;
}

2552 2553
int 
log_compare(const DB_LSN * a, const DB_LSN * b) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
2554
    toku_ydb_lock();
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
2555
    fprintf(stderr, "%s:%d log_compare(%p,%p)\n", __FILE__, __LINE__, a, b);
2556
    assert(0);
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
2557
    toku_ydb_unlock();
2558
    return 0;
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
2559 2560
}

2561 2562
static void env_note_zombie_db_closed(DB_ENV *env, DB *db);

2563 2564
static int
db_close_before_brt(DB *db, u_int32_t UU(flags)) {
2565 2566 2567 2568
    if (db_opened(db) && db->i->dname) {
        // internal (non-user) dictionary has no dname
        env_note_zombie_db_closed(db->dbenv, db);  // tell env that this db is no longer a zombie (it is completely closed)
    }
2569
    char *error_string = 0;
2570
    int r1 = toku_close_brt(db->i->brt, &error_string);
2571 2572 2573 2574 2575 2576 2577 2578 2579 2580 2581 2582
    if (r1) {
	db->dbenv->i->is_panicked = r1; // Panicking the whole environment may be overkill, but I'm not sure what else to do.
	db->dbenv->i->panic_string = error_string;
	if (error_string) {
	    toku_ydb_do_error(db->dbenv, r1, "%s\n", error_string);
	} else {
	    toku_ydb_do_error(db->dbenv, r1, "Closing file\n");
	}
	error_string=0;
    }
    assert(error_string==0);
    int r2 = 0;
Yoni Fogel's avatar
Yoni Fogel committed
2583
    if (db->i->lt) {
2584
        toku_lt_remove_db_ref(db->i->lt, db);
Yoni Fogel's avatar
Yoni Fogel committed
2585
    }
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
2586
    // printf("%s:%d %d=__toku_db_close(%p)\n", __FILE__, __LINE__, r, db);
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
2587 2588
    // Even if panicked, let's close as much as we can.
    int is_panicked = toku_env_is_panicked(db->dbenv); 
2589 2590
    toku_sdbt_cleanup(&db->i->skey);
    toku_sdbt_cleanup(&db->i->sval);
2591
    if (db->i->dname) toku_free(db->i->dname);
Bradley C. Kuszmaul's avatar
Bradley C. Kuszmaul committed
2592 2593
    toku_free(db->i);
    toku_free(db);
2594 2595 2596 2597
    if (r1) return r1;
    if (r2) return r2;
    if (is_panicked) return EINVAL;
    return 0;
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
2598 2599
}

2600 2601 2602 2603 2604 2605 2606 2607
// return 0 if v and dbv refer to same db (including same dname)
// return <0 if v is earlier in omt than dbv
// return >0 if v is later in omt than dbv
static int
find_db_by_db (OMTVALUE v, void *dbv) {
    DB *db = v;            // DB* that is stored in the omt
    DB *dbfind = dbv;      // extra, to be compared to v
    int cmp;
2608 2609 2610 2611 2612 2613 2614
    const char *dname     = db->i->dname;
    const char *dnamefind = dbfind->i->dname;
    cmp = strcmp(dname, dnamefind);
    if (cmp != 0) return cmp;
    int is_zombie     = db->i->is_zombie != 0;
    int is_zombiefind = dbfind->i->is_zombie != 0;
    cmp = is_zombie - is_zombiefind;
2615 2616
    if (cmp != 0) return cmp;
    if (db < dbfind) return -1;
2617
    if (db > dbfind) return  1;
2618 2619 2620 2621 2622 2623 2624
    return 0;
}

// Tell env that there is a new db handle (with non-unique dname in db->i-dname)
static void
env_note_db_opened(DB_ENV *env, DB *db) {
    assert(db->i->dname);  // internal (non-user) dictionary has no dname
2625
    assert(!db->i->is_zombie);
2626 2627 2628
    int r;
    OMTVALUE dbv;
    uint32_t idx;
2629
    env->i->num_open_dbs++;
2630 2631 2632 2633 2634 2635 2636 2637 2638
    r = toku_omt_find_zero(env->i->open_dbs, find_db_by_db, db, &dbv, &idx, NULL);
    assert(r==DB_NOTFOUND); //Must not already be there.
    r = toku_omt_insert_at(env->i->open_dbs, db, idx);
    assert(r==0);
}

static void
env_note_db_closed(DB_ENV *env, DB *db) {
    assert(db->i->dname);
2639
    assert(!db->i->is_zombie);
2640
    assert(env->i->num_open_dbs);
2641 2642 2643
    int r;
    OMTVALUE dbv;
    uint32_t idx;
2644
    env->i->num_open_dbs--;
2645 2646 2647 2648 2649 2650 2651 2652 2653 2654 2655 2656 2657 2658 2659
    r = toku_omt_find_zero(env->i->open_dbs, find_db_by_db, db, &dbv, &idx, NULL);
    assert(r==0); //Must already be there.
    assert((DB*)dbv == db);
    r = toku_omt_delete_at(env->i->open_dbs, idx);
    assert(r==0);
}

// Tell env that there is a new db handle (with non-unique dname in db->i-dname)
static void
env_note_zombie_db(DB_ENV *env, DB *db) {
    assert(db->i->dname);  // internal (non-user) dictionary has no dname
    assert(db->i->is_zombie);
    int r;
    OMTVALUE dbv;
    uint32_t idx;
2660
    env->i->num_zombie_dbs++;
2661 2662 2663 2664 2665 2666 2667 2668 2669 2670
    r = toku_omt_find_zero(env->i->open_dbs, find_db_by_db, db, &dbv, &idx, NULL);
    assert(r==DB_NOTFOUND); //Must not already be there.
    r = toku_omt_insert_at(env->i->open_dbs, db, idx);
    assert(r==0);
}

static void
env_note_zombie_db_closed(DB_ENV *env, DB *db) {
    assert(db->i->dname);
    assert(db->i->is_zombie);
2671
    assert(env->i->num_zombie_dbs);
2672 2673 2674
    int r;
    OMTVALUE dbv;
    uint32_t idx;
2675
    env->i->num_zombie_dbs--;
2676 2677 2678 2679 2680 2681 2682 2683
    r = toku_omt_find_zero(env->i->open_dbs, find_db_by_db, db, &dbv, &idx, NULL);
    assert(r==0); //Must already be there.
    assert((DB*)dbv == db);
    r = toku_omt_delete_at(env->i->open_dbs, idx);
    assert(r==0);
}

static int
2684 2685 2686 2687 2688 2689 2690 2691 2692 2693 2694 2695 2696 2697 2698 2699 2700
find_zombie_db_by_dname (OMTVALUE v, void *dnamev) {
    DB *db = v;            // DB* that is stored in the omt
    int cmp;
    const char *dname     = db->i->dname;
    const char *dnamefind = dnamev;
    cmp = strcmp(dname, dnamefind);
    if (cmp != 0) return cmp;
    int is_zombie     = db->i->is_zombie != 0;
    int is_zombiefind = 1;
    cmp = is_zombie - is_zombiefind;
    return cmp;
}

static int
find_open_db_by_dname (OMTVALUE v, void *dnamev) {
    DB *db = v;            // DB* that is stored in the omt
    int cmp;
2701 2702
    const char *dname     = db->i->dname;
    const char *dnamefind = dnamev;
2703 2704 2705 2706 2707 2708
    cmp = strcmp(dname, dnamefind);
    if (cmp != 0) return cmp;
    int is_zombie     = db->i->is_zombie != 0;
    int is_zombiefind = 0;
    cmp = is_zombie - is_zombiefind;
    return cmp;
2709 2710 2711 2712 2713 2714 2715 2716 2717
}

// return true if there is any db open with the given dname
static BOOL
env_is_db_with_dname_open(DB_ENV *env, const char *dname) {
    int r;
    BOOL rval;
    OMTVALUE dbv;
    uint32_t idx;
2718
    r = toku_omt_find_zero(env->i->open_dbs, find_open_db_by_dname, (void*)dname, &dbv, &idx, NULL);
2719 2720 2721
    if (r==0) {
        DB *db = dbv;
        assert(strcmp(dname, db->i->dname) == 0);
2722
        assert(!db->i->is_zombie);
2723 2724 2725 2726 2727 2728 2729 2730 2731
        rval = TRUE;
    }
    else {
        assert(r==DB_NOTFOUND);
        rval = FALSE;
    }
    return rval;
}

2732 2733 2734 2735 2736 2737 2738 2739 2740 2741 2742 2743 2744 2745 2746 2747 2748 2749 2750 2751 2752 2753
// return true if there is any db open with the given dname
static DB*
env_get_zombie_db_with_dname(DB_ENV *env, const char *dname) {
    int r;
    DB* rval;
    OMTVALUE dbv;
    uint32_t idx;
    r = toku_omt_find_zero(env->i->open_dbs, find_zombie_db_by_dname, (void*)dname, &dbv, &idx, NULL);
    if (r==0) {
        DB *db = dbv;
        assert(db);
        assert(strcmp(dname, db->i->dname) == 0);
        assert(db->i->is_zombie);
        rval = db;
    }
    else {
        assert(r==DB_NOTFOUND);
        rval = NULL;
    }
    return rval;
}

2754 2755 2756 2757 2758 2759 2760 2761 2762 2763
void
toku_db_add_ref(DB *db) {
    db->i->refs++;
}

void
toku_db_release_ref(DB *db){
    db->i->refs--;
}

2764
//DB->close()
2765 2766
static int 
toku_db_close(DB * db, u_int32_t flags) {
2767 2768 2769 2770 2771 2772 2773 2774 2775 2776 2777 2778 2779 2780
    int r = 0;
    if (db->i->refs != 1) {
        r = EBUSY;
    } else {
        db->i->refs = 0;
        if (db_opened(db) && db->i->dname) {
            // internal (non-user) dictionary has no dname
            env_note_db_closed(db->dbenv, db);  // tell env that this db is no longer in use by the user of this api (user-closed, may still be in use by fractal tree internals)
            db->i->is_zombie = TRUE;
            env_note_zombie_db(db->dbenv, db);  // tell env that this db is a zombie
        }
        //Remove from transaction's list of 'must close' if necessary.
        if (!toku_list_empty(&db->i->dbs_that_must_close_before_abort))
            toku_list_remove(&db->i->dbs_that_must_close_before_abort);
2781

2782 2783
        r = toku_brt_db_delay_closed(db->i->brt, db, db_close_before_brt, flags);
    }
2784 2785 2786 2787
    return r;
}


2788
//Get the main portion of a cursor flag (excluding the bitwise or'd components).
2789 2790
static int 
get_main_cursor_flag(u_int32_t flags) {
2791
    return flags & DB_OPFLAGS_MASK;
Yoni Fogel's avatar
Yoni Fogel committed
2792 2793
}

2794 2795
static int 
get_nonmain_cursor_flags(u_int32_t flags) {
2796
    return flags & ~(DB_OPFLAGS_MASK);
2797 2798
}

2799 2800
static inline BOOL 
toku_c_uninitialized(DBC* c) {
2801
    return toku_brt_cursor_uninitialized(dbc_struct_i(c)->c);
Yoni Fogel's avatar
Yoni Fogel committed
2802
}            
Yoni Fogel's avatar
Yoni Fogel committed
2803

2804 2805 2806 2807 2808 2809 2810 2811 2812 2813 2814
typedef struct query_context_wrapped_t {
    DBT               *key;
    DBT               *val;
    struct simple_dbt *skey;
    struct simple_dbt *sval;
} *QUERY_CONTEXT_WRAPPED, QUERY_CONTEXT_WRAPPED_S;

static inline void
query_context_wrapped_init(QUERY_CONTEXT_WRAPPED context, DBC *c, DBT *key, DBT *val) {
    context->key  = key;
    context->val  = val;
2815 2816
    context->skey = dbc_struct_i(c)->skey;
    context->sval = dbc_struct_i(c)->sval;
2817 2818 2819 2820 2821 2822 2823 2824 2825 2826 2827
}

static int
c_get_wrapper_callback(DBT const *key, DBT const *val, void *extra) {
    QUERY_CONTEXT_WRAPPED context = extra;
    int r;
              r = toku_dbt_set(key->size, key->data, context->key, context->skey);
    if (r==0) r = toku_dbt_set(val->size, val->data, context->val, context->sval);
    return r;
}

2828 2829
static int 
toku_c_get_current_unconditional(DBC* c, u_int32_t flags, DBT* key, DBT* val) {
2830 2831 2832 2833
    int r;
    QUERY_CONTEXT_WRAPPED_S context; 
    query_context_wrapped_init(&context, c, key, val);
    r = toku_c_getf_current_binding(c, flags, c_get_wrapper_callback, &context);
Yoni Fogel's avatar
Yoni Fogel committed
2834 2835 2836
    return r;
}

2837 2838
static inline void 
toku_swap_flag(u_int32_t* flag, u_int32_t* get_flag, u_int32_t new_flag) {
Yoni Fogel's avatar
Yoni Fogel committed
2839 2840 2841 2842 2843
    *flag    -= *get_flag;
    *get_flag =  new_flag;
    *flag    += *get_flag;
}

Yoni Fogel's avatar
Yoni Fogel committed
2844 2845 2846 2847 2848 2849 2850
/*
    Used for partial implementation of nested transactions.
    Work is done by children as normal, but all locking is done by the
    root of the nested txn tree.
    This may hold extra locks, and will not work as expected when
    a node has two non-completed txns at any time.
*/
2851 2852
static inline DB_TXN* 
toku_txn_ancestor(DB_TXN* txn) {
Yoni Fogel's avatar
Yoni Fogel committed
2853
    while (txn && txn->parent) txn = txn->parent;
Yoni Fogel's avatar
Yoni Fogel committed
2854

Yoni Fogel's avatar
Yoni Fogel committed
2855 2856 2857
    return txn;
}

Yoni Fogel's avatar
Yoni Fogel committed
2858 2859
static int toku_txn_add_lt(DB_TXN* txn, toku_lock_tree* lt);

Yoni Fogel's avatar
Yoni Fogel committed
2860 2861 2862 2863 2864 2865 2866 2867 2868 2869 2870
/* c_get has many subfunctions with lots of parameters
 * this structure exists to simplify it. */
typedef struct {
    DBC*        c;                  // The cursor
    DB*         db;                 // db the cursor is iterating over
    DB_TXN*     txn_anc;            // The (root) ancestor of the transaction
    TXNID       id_anc;
    DBT         tmp_key;            // Temporary key to protect out param
    DBT         tmp_val;            // Temporary val to protect out param
    u_int32_t   flag;               // The c_get flag
    u_int32_t   op;                 // The operation portion of the c_get flag
Yoni Fogel's avatar
Yoni Fogel committed
2871
    u_int32_t   lock_flags;         // The prelock flags.
Yoni Fogel's avatar
Yoni Fogel committed
2872 2873 2874 2875 2876 2877 2878 2879 2880 2881
    BOOL        cursor_is_write;    // Whether op can change position of cursor
    BOOL        key_is_read;        
    BOOL        key_is_write;
    BOOL        val_is_read;
    BOOL        val_is_write;
    BOOL        duplicates;
    BOOL        tmp_key_malloced;
    BOOL        tmp_val_malloced;
} C_GET_VARS;

2882

2883 2884
static inline u_int32_t 
get_prelocked_flags(u_int32_t flags) {
2885
    u_int32_t lock_flags = flags & (DB_PRELOCKED | DB_PRELOCKED_WRITE);
2886 2887
    return lock_flags;
}
2888

2889 2890
static inline u_int32_t 
get_cursor_prelocked_flags(u_int32_t flags, DBC* dbc) {
2891 2892 2893 2894 2895 2896
    u_int32_t lock_flags = flags & (DB_PRELOCKED | DB_PRELOCKED_WRITE);

    //DB_READ_UNCOMMITTED and DB_READ_COMMITTED transactions 'own' all read locks for user-data dictionaries.
    if (dbc_struct_i(dbc)->iso != TOKU_ISO_SERIALIZABLE)
    {
        lock_flags |= DB_PRELOCKED;
2897
    }
2898
    return lock_flags;
Yoni Fogel's avatar
Yoni Fogel committed
2899 2900
}

2901 2902
static int
toku_c_get(DBC* c, DBT* key, DBT* val, u_int32_t flag) {
2903 2904 2905
    //This function exists for legacy (test compatibility) purposes/parity with bdb.
    HANDLE_PANICKED_DB(c->dbp);
    HANDLE_CURSOR_ILLEGAL_WORKING_PARENT_TXN(c);
Yoni Fogel's avatar
Yoni Fogel committed
2906

2907 2908 2909 2910 2911 2912 2913 2914 2915 2916 2917 2918
    u_int32_t main_flag       = get_main_cursor_flag(flag);
    u_int32_t remaining_flags = get_nonmain_cursor_flags(flag);
    int r;
    QUERY_CONTEXT_WRAPPED_S context;
    //Passing in NULL for a key or val means that it is NOT an output.
    //    Both key and val are output:
    //        query_context_wrapped_init(&context, c, key,  val);
    //    Val is output, key is not:
    //            query_context_wrapped_init(&context, c, NULL, val);
    //    Neither key nor val are output:
    //	    query_context_wrapped_init(&context, c, NULL, NULL); // Used for DB_GET_BOTH
    switch (main_flag) {
Yoni Fogel's avatar
Yoni Fogel committed
2919
        case (DB_FIRST):
2920 2921
            query_context_wrapped_init(&context, c, key,  val);
            r = toku_c_getf_first(c, remaining_flags, c_get_wrapper_callback, &context);
Yoni Fogel's avatar
Yoni Fogel committed
2922
            break;
Yoni Fogel's avatar
Yoni Fogel committed
2923
        case (DB_LAST):
2924 2925
            query_context_wrapped_init(&context, c, key,  val);
            r = toku_c_getf_last(c, remaining_flags, c_get_wrapper_callback, &context);
Yoni Fogel's avatar
Yoni Fogel committed
2926
            break;
2927
        case (DB_NEXT):
Yoni Fogel's avatar
Yoni Fogel committed
2928
        case (DB_NEXT_NODUP):
2929
            query_context_wrapped_init(&context, c, key,  val);
2930
            r = toku_c_getf_next(c, remaining_flags, c_get_wrapper_callback, &context);
Yoni Fogel's avatar
Yoni Fogel committed
2931 2932
            break;
        case (DB_PREV):
2933
        case (DB_PREV_NODUP):
2934 2935
            query_context_wrapped_init(&context, c, key,  val);
            r = toku_c_getf_prev(c, remaining_flags, c_get_wrapper_callback, &context);
Yoni Fogel's avatar
Yoni Fogel committed
2936 2937
            break;
#ifdef DB_PREV_DUP
Yoni Fogel's avatar
Yoni Fogel committed
2938
        case (DB_PREV_DUP):
2939 2940
            query_context_wrapped_init(&context, c, key,  val);
            r = toku_c_getf_prev_dup(c, remaining_flags, c_get_wrapper_callback, &context);
Yoni Fogel's avatar
Yoni Fogel committed
2941 2942
            break;
#endif
2943 2944 2945 2946 2947 2948 2949 2950 2951 2952 2953 2954 2955 2956 2957 2958 2959
        case (DB_CURRENT):
            query_context_wrapped_init(&context, c, key,  val);
            r = toku_c_getf_current(c, remaining_flags, c_get_wrapper_callback, &context);
            break;
        case (DB_CURRENT_BINDING):
            query_context_wrapped_init(&context, c, key,  val);
            r = toku_c_getf_current_binding(c, remaining_flags, c_get_wrapper_callback, &context);
            break;

        case (DB_SET):
            query_context_wrapped_init(&context, c, NULL, val);
            r = toku_c_getf_set(c, remaining_flags, key, c_get_wrapper_callback, &context);
            break;
        case (DB_SET_RANGE):
            query_context_wrapped_init(&context, c, key,  val);
            r = toku_c_getf_set_range(c, remaining_flags, key, c_get_wrapper_callback, &context);
            break;
2960 2961 2962 2963
        case (DB_SET_RANGE_REVERSE):
            query_context_wrapped_init(&context, c, key,  val);
            r = toku_c_getf_set_range_reverse(c, remaining_flags, key, c_get_wrapper_callback, &context);
            break;
Yoni Fogel's avatar
Yoni Fogel committed
2964
        default:
Yoni Fogel's avatar
Yoni Fogel committed
2965
            r = EINVAL;
2966
            break;
Yoni Fogel's avatar
Yoni Fogel committed
2967
    }
Yoni Fogel's avatar
Yoni Fogel committed
2968
    return r;
Yoni Fogel's avatar
Yoni Fogel committed
2969 2970
}

2971 2972
static int 
locked_c_getf_first(DBC *c, u_int32_t flag, YDB_CALLBACK_FUNCTION f, void *extra) {
2973
    toku_ydb_lock();  int r = toku_c_getf_first(c, flag, f, extra); toku_ydb_unlock(); return r;
Yoni Fogel's avatar
Yoni Fogel committed
2974 2975
}

2976 2977
static int 
locked_c_getf_last(DBC *c, u_int32_t flag, YDB_CALLBACK_FUNCTION f, void *extra) {
2978 2979
    toku_ydb_lock();  int r = toku_c_getf_last(c, flag, f, extra); toku_ydb_unlock(); return r;
}
Yoni Fogel's avatar
Yoni Fogel committed
2980

2981 2982
static int 
locked_c_getf_next(DBC *c, u_int32_t flag, YDB_CALLBACK_FUNCTION f, void *extra) {
2983
    toku_ydb_lock();  int r = toku_c_getf_next(c, flag, f, extra); toku_ydb_unlock(); return r;
Yoni Fogel's avatar
Yoni Fogel committed
2984 2985
}

2986 2987
static int 
locked_c_getf_prev(DBC *c, u_int32_t flag, YDB_CALLBACK_FUNCTION f, void *extra) {
2988
    toku_ydb_lock();  int r = toku_c_getf_prev(c, flag, f, extra); toku_ydb_unlock(); return r;
Yoni Fogel's avatar
Yoni Fogel committed
2989 2990
}

2991 2992
static int 
locked_c_getf_current(DBC *c, u_int32_t flag, YDB_CALLBACK_FUNCTION f, void *extra) {
2993
    toku_ydb_lock();  int r = toku_c_getf_current(c, flag, f, extra); toku_ydb_unlock(); return r;
Yoni Fogel's avatar
Yoni Fogel committed
2994 2995
}

2996 2997
static int 
locked_c_getf_current_binding(DBC *c, u_int32_t flag, YDB_CALLBACK_FUNCTION f, void *extra) {
2998 2999
    toku_ydb_lock();  int r = toku_c_getf_current_binding(c, flag, f, extra); toku_ydb_unlock(); return r;
}
Yoni Fogel's avatar
Yoni Fogel committed
3000

3001 3002
static int 
locked_c_getf_set(DBC *c, u_int32_t flag, DBT * key, YDB_CALLBACK_FUNCTION f, void *extra) {
3003 3004
    toku_ydb_lock();  int r = toku_c_getf_set(c, flag, key, f, extra); toku_ydb_unlock(); return r;
}
Yoni Fogel's avatar
Yoni Fogel committed
3005

3006 3007
static int 
locked_c_getf_set_range(DBC *c, u_int32_t flag, DBT * key, YDB_CALLBACK_FUNCTION f, void *extra) {
3008 3009
    toku_ydb_lock();  int r = toku_c_getf_set_range(c, flag, key, f, extra); toku_ydb_unlock(); return r;
}
Yoni Fogel's avatar
Yoni Fogel committed
3010

3011 3012
static int 
locked_c_getf_set_range_reverse(DBC *c, u_int32_t flag, DBT * key, YDB_CALLBACK_FUNCTION f, void *extra) {
3013 3014 3015
    toku_ydb_lock();  int r = toku_c_getf_set_range_reverse(c, flag, key, f, extra); toku_ydb_unlock(); return r;
}

3016 3017 3018 3019 3020 3021 3022 3023
typedef struct {
    BOOL            is_read_lock;
    DB_TXN         *txn;
    DB             *db;
    toku_lock_tree *lt;
    DBT const      *left_key;
    DBT const      *right_key;
} *RANGE_LOCK_REQUEST, RANGE_LOCK_REQUEST_S;
Yoni Fogel's avatar
Yoni Fogel committed
3024

3025 3026 3027 3028 3029 3030
static void
range_lock_request_init(RANGE_LOCK_REQUEST request,
                        BOOL       is_read_lock,
                        DB_TXN    *txn,
                        DB        *db,
                        DBT const *left_key,
3031
                        DBT const *right_key) {
3032 3033 3034 3035 3036 3037
    request->is_read_lock = is_read_lock;
    request->txn = txn;
    request->db = db;
    request->lt = db->i->lt;
    request->left_key = left_key;
    request->right_key = right_key;
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
3038 3039
}

3040 3041 3042 3043 3044
static void
read_lock_request_init(RANGE_LOCK_REQUEST request,
                       DB_TXN    *txn,
                       DB        *db,
                       DBT const *left_key,
3045
                       DBT const *right_key) {
3046
    range_lock_request_init(request, TRUE, txn, db,
3047
                            left_key, right_key);
Yoni Fogel's avatar
Yoni Fogel committed
3048 3049
}

3050 3051 3052 3053 3054
static void
write_lock_request_init(RANGE_LOCK_REQUEST request,
                        DB_TXN    *txn,
                        DB        *db,
                        DBT const *left_key,
3055
                        DBT const *right_key) {
3056
    range_lock_request_init(request, FALSE, txn, db,
3057
                            left_key, right_key);
3058 3059 3060 3061 3062 3063 3064 3065 3066
}

static int
grab_range_lock(RANGE_LOCK_REQUEST request) {
    int r;
    //TODO: (Multithreading) Grab lock protecting lock tree
    DB_TXN *txn_anc = toku_txn_ancestor(request->txn);
    r = toku_txn_add_lt(txn_anc, request->lt);
    if (r==0) {
3067
        TXNID txn_anc_id = toku_txn_get_txnid(db_txn_struct_i(txn_anc)->tokutxn);
3068 3069
        if (request->is_read_lock)
            r = toku_lt_acquire_range_read_lock(request->lt, request->db, txn_anc_id,
3070
                                                request->left_key, request->right_key);
3071 3072
        else 
            r = toku_lt_acquire_range_write_lock(request->lt, request->db, txn_anc_id,
3073
                                                 request->left_key, request->right_key);
3074 3075 3076
    }
    //TODO: (Multithreading) Release lock protecting lock tree
    return r;
Yoni Fogel's avatar
Yoni Fogel committed
3077 3078
}

3079
int
Zardosht Kasheff's avatar
Zardosht Kasheff committed
3080 3081 3082 3083
toku_grab_read_lock_on_directory (DB* db, DB_TXN * txn) {
    RANGE_LOCK_REQUEST_S request;
    char *   dname = db->i->dname;
    DBT key_in_directory;
3084

Zardosht Kasheff's avatar
Zardosht Kasheff committed
3085
    // bad hack because some environment dictionaries do not have a dname
3086
    if (!dname || (db->dbenv->i->directory->i->lt == NULL))
Zardosht Kasheff's avatar
Zardosht Kasheff committed
3087
        return 0;
3088

Zardosht Kasheff's avatar
Zardosht Kasheff committed
3089 3090 3091 3092 3093 3094 3095 3096 3097 3098 3099 3100 3101
    toku_fill_dbt(&key_in_directory, dname, strlen(dname)+1);
    //Left end of range == right end of range (point lock)
    read_lock_request_init(
        &request, 
        txn, 
        db->dbenv->i->directory,
        &key_in_directory,
        &key_in_directory
        );
    int r = grab_range_lock(&request);
    return r;
}

3102 3103
//This is the user level callback function given to ydb layer functions like
//toku_c_getf_first
Yoni Fogel's avatar
Yoni Fogel committed
3104

3105 3106 3107 3108
typedef struct __toku_is_write_op {
    BOOL is_write_op;
} WRITE_OP;

3109 3110 3111 3112 3113 3114
typedef struct query_context_base_t {
    BRT_CURSOR  c;
    DB_TXN     *txn;
    DB         *db;
    void       *f_extra;
    int         r_user_callback;
3115 3116
    BOOL        do_locking;
    BOOL        is_write_op;
3117
} *QUERY_CONTEXT_BASE, QUERY_CONTEXT_BASE_S;
3118

3119 3120 3121 3122
typedef struct query_context_t {
    QUERY_CONTEXT_BASE_S  base;
    YDB_CALLBACK_FUNCTION f;
} *QUERY_CONTEXT, QUERY_CONTEXT_S;
Yoni Fogel's avatar
Yoni Fogel committed
3123

3124 3125 3126 3127 3128 3129
typedef struct query_context_with_input_t {
    QUERY_CONTEXT_BASE_S  base;
    YDB_CALLBACK_FUNCTION f;
    DBT                  *input_key;
    DBT                  *input_val;
} *QUERY_CONTEXT_WITH_INPUT, QUERY_CONTEXT_WITH_INPUT_S;
Yoni Fogel's avatar
Yoni Fogel committed
3130

3131 3132

static void
3133
query_context_base_init(QUERY_CONTEXT_BASE context, DBC *c, u_int32_t flag, WRITE_OP is_write_op, void *extra) {
3134 3135
    context->c       = dbc_struct_i(c)->c;
    context->txn     = dbc_struct_i(c)->txn;
3136 3137
    context->db      = c->dbp;
    context->f_extra = extra;
3138
    context->is_write_op = is_write_op.is_write_op;
3139
    u_int32_t lock_flags = get_cursor_prelocked_flags(flag, c);
Yoni Fogel's avatar
Yoni Fogel committed
3140
    flag &= ~lock_flags;
3141
    if (context->is_write_op) lock_flags &= DB_PRELOCKED_WRITE; // Only care about whether already locked for write
Yoni Fogel's avatar
Yoni Fogel committed
3142
    assert(flag==0);
3143
    context->do_locking = (BOOL)(context->db->i->lt!=NULL && !lock_flags);
3144
    context->r_user_callback = 0;
Yoni Fogel's avatar
Yoni Fogel committed
3145 3146
}

3147 3148
static void
query_context_init(QUERY_CONTEXT context, DBC *c, u_int32_t flag, YDB_CALLBACK_FUNCTION f, void *extra) {
3149 3150 3151 3152 3153 3154 3155 3156 3157
    WRITE_OP is_write = {FALSE};
    query_context_base_init(&context->base, c, flag, is_write, extra);
    context->f = f;
}

static void
query_context_init_write_op(QUERY_CONTEXT context, DBC *c, u_int32_t flag, YDB_CALLBACK_FUNCTION f, void *extra) {
    WRITE_OP is_write = {TRUE};
    query_context_base_init(&context->base, c, flag, is_write, extra);
3158 3159
    context->f = f;
}
Yoni Fogel's avatar
Yoni Fogel committed
3160

3161 3162
static void
query_context_with_input_init(QUERY_CONTEXT_WITH_INPUT context, DBC *c, u_int32_t flag, DBT *key, DBT *val, YDB_CALLBACK_FUNCTION f, void *extra) {
3163 3164
    WRITE_OP is_write = {FALSE};
    query_context_base_init(&context->base, c, flag, is_write, extra);
3165 3166 3167
    context->f         = f;
    context->input_key = key;
    context->input_val = val;
3168 3169
}

3170 3171 3172 3173 3174
static int c_del_callback(DBT const *key, DBT const *val, void *extra);

//Delete whatever the cursor is pointing at.
static int
toku_c_del(DBC * c, u_int32_t flags) {
3175
    HANDLE_PANICKED_DB(c->dbp);
3176
    HANDLE_CURSOR_ILLEGAL_WORKING_PARENT_TXN(c);
3177

3178 3179 3180 3181
    u_int32_t unchecked_flags = flags;
    //DB_DELETE_ANY means delete regardless of whether it exists in the db.
    u_int32_t flag_for_brt = flags&DB_DELETE_ANY;
    unchecked_flags &= ~flag_for_brt;
3182
    u_int32_t lock_flags = get_cursor_prelocked_flags(flags, c);
3183
    unchecked_flags &= ~lock_flags;
3184
    BOOL do_locking = (BOOL)(c->dbp->i->lt && !(lock_flags&DB_PRELOCKED_WRITE));
3185

3186 3187 3188 3189 3190
    int r = 0;
    if (unchecked_flags!=0) r = EINVAL;
    else {
        if (do_locking) {
            QUERY_CONTEXT_S context;
3191
            query_context_init_write_op(&context, c, lock_flags, NULL, NULL);
3192 3193 3194 3195 3196
            //We do not need a read lock, we must already have it.
            r = toku_c_getf_current_binding(c, DB_PRELOCKED, c_del_callback, &context);
        }
        if (r==0) {
            //Do the actual delete.
3197 3198
            TOKUTXN txn = dbc_struct_i(c)->txn ? db_txn_struct_i(dbc_struct_i(c)->txn)->tokutxn : 0;
            r = toku_brt_cursor_delete(dbc_struct_i(c)->c, flag_for_brt, txn);
3199 3200
        }
    }
3201 3202
    return r;
}
3203

3204 3205 3206 3207 3208
//result is the result of the query (i.e. 0 means found, DB_NOTFOUND, etc..)
static int
c_del_callback(DBT const *key, DBT const *val, void *extra) {
    QUERY_CONTEXT_WITH_INPUT super_context = extra;
    QUERY_CONTEXT_BASE       context       = &super_context->base;
Yoni Fogel's avatar
Yoni Fogel committed
3209

3210
    int r;
Yoni Fogel's avatar
Yoni Fogel committed
3211

3212
    assert(context->do_locking);
3213
    assert(context->is_write_op);
3214 3215 3216 3217 3218 3219
    assert(key!=NULL);
    assert(val!=NULL);
    //Lock:
    //  left(key,val)==right(key,val) == (key, val);
    RANGE_LOCK_REQUEST_S request;
    write_lock_request_init(&request, context->txn, context->db,
3220
                            key, key);
3221 3222 3223
    r = grab_range_lock(&request);

    //Give brt-layer an error (if any) to return from toku_c_getf_current_binding
Yoni Fogel's avatar
Yoni Fogel committed
3224 3225 3226
    return r;
}

3227 3228 3229 3230 3231
static int c_getf_first_callback(ITEMLEN keylen, bytevec key, ITEMLEN vallen, bytevec val, void *extra);

static int
toku_c_getf_first(DBC *c, u_int32_t flag, YDB_CALLBACK_FUNCTION f, void *extra) {
    HANDLE_PANICKED_DB(c->dbp);
3232
    HANDLE_CURSOR_ILLEGAL_WORKING_PARENT_TXN(c);
3233
    num_point_queries++;   // accountability
3234 3235 3236
    QUERY_CONTEXT_S context; //Describes the context of this query.
    query_context_init(&context, c, flag, f, extra); 
    //toku_brt_cursor_first will call c_getf_first_callback(..., context) (if query is successful)
3237
    int r = toku_brt_cursor_first(dbc_struct_i(c)->c, c_getf_first_callback, &context);
3238 3239 3240 3241 3242 3243 3244 3245 3246 3247 3248 3249 3250 3251 3252 3253 3254 3255 3256 3257 3258
    if (r == TOKUDB_USER_CALLBACK_ERROR) r = context.base.r_user_callback;
    return r;
}

//result is the result of the query (i.e. 0 means found, DB_NOTFOUND, etc..)
static int
c_getf_first_callback(ITEMLEN keylen, bytevec key, ITEMLEN vallen, bytevec val, void *extra) {
    QUERY_CONTEXT      super_context = extra;
    QUERY_CONTEXT_BASE context       = &super_context->base;

    int r;

    DBT found_key;
    DBT found_val;
    toku_fill_dbt(&found_key, key, keylen);
    toku_fill_dbt(&found_val, val, vallen);

    if (context->do_locking) {
        RANGE_LOCK_REQUEST_S request;
        if (key!=NULL) {
            read_lock_request_init(&request, context->txn, context->db,
3259
                                   toku_lt_neg_infinity, &found_key);
3260 3261 3262
        }
        else {
            read_lock_request_init(&request, context->txn, context->db,
3263
                                   toku_lt_neg_infinity, toku_lt_infinity);
3264 3265 3266 3267 3268 3269 3270 3271 3272 3273 3274 3275 3276 3277 3278 3279 3280 3281 3282 3283
        }
        r = grab_range_lock(&request);
    }
    else r = 0;

    //Call application-layer callback if found and locks were successfully obtained.
    if (r==0 && key!=NULL) {
        context->r_user_callback = super_context->f(&found_key, &found_val, context->f_extra);
        if (context->r_user_callback) r = TOKUDB_USER_CALLBACK_ERROR;
    }

    //Give brt-layer an error (if any) to return from toku_brt_cursor_first
    return r;
}

static int c_getf_last_callback(ITEMLEN keylen, bytevec key, ITEMLEN vallen, bytevec val, void *extra);

static int
toku_c_getf_last(DBC *c, u_int32_t flag, YDB_CALLBACK_FUNCTION f, void *extra) {
    HANDLE_PANICKED_DB(c->dbp);
3284
    HANDLE_CURSOR_ILLEGAL_WORKING_PARENT_TXN(c);
3285
    num_point_queries++;   // accountability
3286 3287 3288
    QUERY_CONTEXT_S context; //Describes the context of this query.
    query_context_init(&context, c, flag, f, extra); 
    //toku_brt_cursor_last will call c_getf_last_callback(..., context) (if query is successful)
3289
    int r = toku_brt_cursor_last(dbc_struct_i(c)->c, c_getf_last_callback, &context);
3290 3291 3292 3293 3294 3295 3296 3297 3298 3299 3300 3301 3302 3303 3304 3305 3306 3307 3308 3309 3310
    if (r == TOKUDB_USER_CALLBACK_ERROR) r = context.base.r_user_callback;
    return r;
}

//result is the result of the query (i.e. 0 means found, DB_NOTFOUND, etc..)
static int
c_getf_last_callback(ITEMLEN keylen, bytevec key, ITEMLEN vallen, bytevec val, void *extra) {
    QUERY_CONTEXT      super_context = extra;
    QUERY_CONTEXT_BASE context       = &super_context->base;

    int r;

    DBT found_key;
    DBT found_val;
    toku_fill_dbt(&found_key, key, keylen);
    toku_fill_dbt(&found_val, val, vallen);

    if (context->do_locking) {
        RANGE_LOCK_REQUEST_S request;
        if (key!=NULL) {
            read_lock_request_init(&request, context->txn, context->db,
3311
                                   &found_key,           toku_lt_infinity);
3312 3313 3314
        }
        else {
            read_lock_request_init(&request, context->txn, context->db,
3315
                                   toku_lt_neg_infinity, toku_lt_infinity);
3316 3317 3318 3319 3320 3321 3322 3323 3324 3325 3326 3327 3328 3329 3330 3331 3332 3333 3334 3335 3336
        }
        r = grab_range_lock(&request);
    }
    else r = 0;

    //Call application-layer callback if found and locks were successfully obtained.
    if (r==0 && key!=NULL) {
        context->r_user_callback = super_context->f(&found_key, &found_val, context->f_extra);
        if (context->r_user_callback) r = TOKUDB_USER_CALLBACK_ERROR;
    }

    //Give brt-layer an error (if any) to return from toku_brt_cursor_last
    return r;
}

static int c_getf_next_callback(ITEMLEN keylen, bytevec key, ITEMLEN vallen, bytevec val, void *extra);

static int
toku_c_getf_next(DBC *c, u_int32_t flag, YDB_CALLBACK_FUNCTION f, void *extra) {
    int r;
    HANDLE_PANICKED_DB(c->dbp);
3337
    HANDLE_CURSOR_ILLEGAL_WORKING_PARENT_TXN(c);
3338
    if (toku_c_uninitialized(c)) r = toku_c_getf_first(c, flag, f, extra);
3339 3340
    else {
        QUERY_CONTEXT_S context; //Describes the context of this query.
3341
        num_sequential_queries++;   // accountability
3342 3343
        query_context_init(&context, c, flag, f, extra); 
        //toku_brt_cursor_next will call c_getf_next_callback(..., context) (if query is successful)
3344
        r = toku_brt_cursor_next(dbc_struct_i(c)->c, c_getf_next_callback, &context);
3345 3346 3347 3348 3349 3350 3351 3352 3353 3354 3355 3356 3357 3358 3359 3360 3361 3362 3363 3364 3365 3366 3367 3368 3369 3370
        if (r == TOKUDB_USER_CALLBACK_ERROR) r = context.base.r_user_callback;
    }
    return r;
}

//result is the result of the query (i.e. 0 means found, DB_NOTFOUND, etc..)
static int
c_getf_next_callback(ITEMLEN keylen, bytevec key, ITEMLEN vallen, bytevec val, void *extra) {
    QUERY_CONTEXT      super_context = extra;
    QUERY_CONTEXT_BASE context       = &super_context->base;

    int r;

    DBT found_key;
    DBT found_val;
    toku_fill_dbt(&found_key, key, keylen);
    toku_fill_dbt(&found_val, val, vallen);

    if (context->do_locking) {
        RANGE_LOCK_REQUEST_S request;
        const DBT *prevkey;
        const DBT *prevval;
        const DBT *right_key = key==NULL ? toku_lt_infinity : &found_key;

        toku_brt_cursor_peek(context->c, &prevkey, &prevval);
        read_lock_request_init(&request, context->txn, context->db,
3371 3372
                               prevkey,
                               right_key);
3373 3374 3375 3376 3377 3378 3379 3380 3381 3382 3383 3384 3385 3386 3387 3388 3389 3390 3391 3392
        r = grab_range_lock(&request);
    }
    else r = 0;

    //Call application-layer callback if found and locks were successfully obtained.
    if (r==0 && key!=NULL) {
        context->r_user_callback = super_context->f(&found_key, &found_val, context->f_extra);
        if (context->r_user_callback) r = TOKUDB_USER_CALLBACK_ERROR;
    }

    //Give brt-layer an error (if any) to return from toku_brt_cursor_next
    return r;
}

static int c_getf_prev_callback(ITEMLEN keylen, bytevec key, ITEMLEN vallen, bytevec val, void *extra);

static int
toku_c_getf_prev(DBC *c, u_int32_t flag, YDB_CALLBACK_FUNCTION f, void *extra) {
    int r;
    HANDLE_PANICKED_DB(c->dbp);
3393
    HANDLE_CURSOR_ILLEGAL_WORKING_PARENT_TXN(c);
3394
    if (toku_c_uninitialized(c)) r = toku_c_getf_last(c, flag, f, extra);
3395 3396
    else {
        QUERY_CONTEXT_S context; //Describes the context of this query.
3397
        num_sequential_queries++;   // accountability
3398 3399
        query_context_init(&context, c, flag, f, extra); 
        //toku_brt_cursor_prev will call c_getf_prev_callback(..., context) (if query is successful)
3400
        r = toku_brt_cursor_prev(dbc_struct_i(c)->c, c_getf_prev_callback, &context);
3401
        if (r == TOKUDB_USER_CALLBACK_ERROR) r = context.base.r_user_callback;
Yoni Fogel's avatar
Yoni Fogel committed
3402 3403 3404 3405
    }
    return r;
}

3406 3407 3408 3409 3410 3411 3412 3413 3414 3415 3416 3417 3418 3419 3420 3421 3422 3423 3424 3425 3426
//result is the result of the query (i.e. 0 means found, DB_NOTFOUND, etc..)
static int
c_getf_prev_callback(ITEMLEN keylen, bytevec key, ITEMLEN vallen, bytevec val, void *extra) {
    QUERY_CONTEXT      super_context = extra;
    QUERY_CONTEXT_BASE context       = &super_context->base;

    int r;

    DBT found_key;
    DBT found_val;
    toku_fill_dbt(&found_key, key, keylen);
    toku_fill_dbt(&found_val, val, vallen);

    if (context->do_locking) {
        RANGE_LOCK_REQUEST_S request;
        const DBT *prevkey;
        const DBT *prevval;
        const DBT *left_key = key==NULL ? toku_lt_neg_infinity : &found_key;

        toku_brt_cursor_peek(context->c, &prevkey, &prevval);
        read_lock_request_init(&request, context->txn, context->db,
3427 3428
                               left_key,
                               prevkey);
3429 3430 3431 3432 3433 3434 3435 3436 3437 3438 3439 3440
        r = grab_range_lock(&request);
    }
    else r = 0;

    //Call application-layer callback if found and locks were successfully obtained.
    if (r==0 && key!=NULL) {
        context->r_user_callback = super_context->f(&found_key, &found_val, context->f_extra);
        if (context->r_user_callback) r = TOKUDB_USER_CALLBACK_ERROR;
    }

    //Give brt-layer an error (if any) to return from toku_brt_cursor_prev
    return r;
Yoni Fogel's avatar
Yoni Fogel committed
3441 3442
}

3443 3444 3445 3446 3447
static int c_getf_current_callback(ITEMLEN keylen, bytevec key, ITEMLEN vallen, bytevec val, void *extra);

static int
toku_c_getf_current(DBC *c, u_int32_t flag, YDB_CALLBACK_FUNCTION f, void *extra) {
    HANDLE_PANICKED_DB(c->dbp);
3448
    HANDLE_CURSOR_ILLEGAL_WORKING_PARENT_TXN(c);
3449 3450

    QUERY_CONTEXT_S context; //Describes the context of this query.
3451
    num_sequential_queries++;   // accountability
3452 3453
    query_context_init(&context, c, flag, f, extra); 
    //toku_brt_cursor_current will call c_getf_current_callback(..., context) (if query is successful)
3454
    int r = toku_brt_cursor_current(dbc_struct_i(c)->c, DB_CURRENT, c_getf_current_callback, &context);
3455 3456 3457 3458 3459 3460 3461 3462 3463 3464 3465 3466 3467 3468 3469 3470 3471 3472 3473 3474 3475 3476 3477 3478 3479 3480 3481 3482 3483
    if (r == TOKUDB_USER_CALLBACK_ERROR) r = context.base.r_user_callback;
    return r;
}

//result is the result of the query (i.e. 0 means found, DB_NOTFOUND, etc..)
static int
c_getf_current_callback(ITEMLEN keylen, bytevec key, ITEMLEN vallen, bytevec val, void *extra) {
    QUERY_CONTEXT      super_context = extra;
    QUERY_CONTEXT_BASE context       = &super_context->base;

    DBT found_key;
    DBT found_val;
    toku_fill_dbt(&found_key, key, keylen);
    toku_fill_dbt(&found_val, val, vallen);

    int r=0;
    //Call application-layer callback if found.
    if (key!=NULL) {
        context->r_user_callback = super_context->f(&found_key, &found_val, context->f_extra);
        if (context->r_user_callback) r = TOKUDB_USER_CALLBACK_ERROR;
    }

    //Give brt-layer an error (if any) to return from toku_brt_cursor_current
    return r;
}

static int
toku_c_getf_current_binding(DBC *c, u_int32_t flag, YDB_CALLBACK_FUNCTION f, void *extra) {
    HANDLE_PANICKED_DB(c->dbp);
3484
    HANDLE_CURSOR_ILLEGAL_WORKING_PARENT_TXN(c);
3485 3486

    QUERY_CONTEXT_S context; //Describes the context of this query.
3487
    num_sequential_queries++;   // accountability
3488 3489
    query_context_init(&context, c, flag, f, extra); 
    //toku_brt_cursor_current will call c_getf_current_callback(..., context) (if query is successful)
3490
    int r = toku_brt_cursor_current(dbc_struct_i(c)->c, DB_CURRENT_BINDING, c_getf_current_callback, &context);
3491 3492 3493 3494 3495 3496 3497 3498 3499
    if (r == TOKUDB_USER_CALLBACK_ERROR) r = context.base.r_user_callback;
    return r;
}

static int c_getf_set_callback(ITEMLEN keylen, bytevec key, ITEMLEN vallen, bytevec val, void *extra);

static int
toku_c_getf_set(DBC *c, u_int32_t flag, DBT *key, YDB_CALLBACK_FUNCTION f, void *extra) {
    HANDLE_PANICKED_DB(c->dbp);
3500
    HANDLE_CURSOR_ILLEGAL_WORKING_PARENT_TXN(c);
3501 3502

    QUERY_CONTEXT_WITH_INPUT_S context; //Describes the context of this query.
3503
    num_point_queries++;   // accountability
3504 3505
    query_context_with_input_init(&context, c, flag, key, NULL, f, extra); 
    //toku_brt_cursor_set will call c_getf_set_callback(..., context) (if query is successful)
3506
    int r = toku_brt_cursor_set(dbc_struct_i(c)->c, key, c_getf_set_callback, &context);
3507 3508 3509 3510 3511 3512 3513 3514 3515 3516 3517 3518 3519 3520 3521 3522 3523 3524 3525 3526 3527 3528 3529 3530
    if (r == TOKUDB_USER_CALLBACK_ERROR) r = context.base.r_user_callback;
    return r;
}

//result is the result of the query (i.e. 0 means found, DB_NOTFOUND, etc..)
static int
c_getf_set_callback(ITEMLEN keylen, bytevec key, ITEMLEN vallen, bytevec val, void *extra) {
    QUERY_CONTEXT_WITH_INPUT super_context = extra;
    QUERY_CONTEXT_BASE       context       = &super_context->base;

    int r;

    DBT found_key;
    DBT found_val;
    toku_fill_dbt(&found_key, key, keylen);
    toku_fill_dbt(&found_val, val, vallen);

    //Lock:
    //  left(key,val)  = (input_key, -infinity)
    //  right(key,val) = (input_key, found ? found_val : infinity)
    if (context->do_locking) {
        RANGE_LOCK_REQUEST_S request;
        if (key!=NULL) {
            read_lock_request_init(&request, context->txn, context->db,
3531 3532
                                   super_context->input_key,
                                   super_context->input_key);
3533 3534 3535
        }
        else {
            read_lock_request_init(&request, context->txn, context->db,
3536 3537
                                   super_context->input_key,
                                   super_context->input_key);
3538 3539 3540 3541 3542 3543 3544 3545 3546 3547 3548 3549 3550 3551 3552 3553 3554 3555 3556 3557
        }
        r = grab_range_lock(&request);
    }
    else r = 0;

    //Call application-layer callback if found and locks were successfully obtained.
    if (r==0 && key!=NULL) {
        context->r_user_callback = super_context->f(&found_key, &found_val, context->f_extra);
        if (context->r_user_callback) r = TOKUDB_USER_CALLBACK_ERROR;
    }

    //Give brt-layer an error (if any) to return from toku_brt_cursor_set
    return r;
}

static int c_getf_set_range_callback(ITEMLEN keylen, bytevec key, ITEMLEN vallen, bytevec val, void *extra);

static int
toku_c_getf_set_range(DBC *c, u_int32_t flag, DBT *key, YDB_CALLBACK_FUNCTION f, void *extra) {
    HANDLE_PANICKED_DB(c->dbp);
3558
    HANDLE_CURSOR_ILLEGAL_WORKING_PARENT_TXN(c);
3559 3560

    QUERY_CONTEXT_WITH_INPUT_S context; //Describes the context of this query.
3561
    num_point_queries++;   // accountability
3562 3563
    query_context_with_input_init(&context, c, flag, key, NULL, f, extra); 
    //toku_brt_cursor_set_range will call c_getf_set_range_callback(..., context) (if query is successful)
3564
    int r = toku_brt_cursor_set_range(dbc_struct_i(c)->c, key, c_getf_set_range_callback, &context);
3565 3566 3567 3568 3569 3570 3571 3572 3573 3574 3575 3576 3577 3578 3579 3580 3581 3582 3583 3584 3585 3586 3587 3588 3589
    if (r == TOKUDB_USER_CALLBACK_ERROR) r = context.base.r_user_callback;
    return r;
}

//result is the result of the query (i.e. 0 means found, DB_NOTFOUND, etc..)
static int
c_getf_set_range_callback(ITEMLEN keylen, bytevec key, ITEMLEN vallen, bytevec val, void *extra) {
    QUERY_CONTEXT_WITH_INPUT super_context = extra;
    QUERY_CONTEXT_BASE       context       = &super_context->base;

    int r;

    DBT found_key;
    DBT found_val;
    toku_fill_dbt(&found_key, key, keylen);
    toku_fill_dbt(&found_val, val, vallen);

    //Lock:
    //  left(key,val)  = (input_key, -infinity)
    //  right(key) = found ? found_key : infinity
    //  right(val) = found ? found_val : infinity
    if (context->do_locking) {
        RANGE_LOCK_REQUEST_S request;
        if (key!=NULL) {
            read_lock_request_init(&request, context->txn, context->db,
3590 3591
                                   super_context->input_key,
                                   &found_key);
3592 3593 3594
        }
        else {
            read_lock_request_init(&request, context->txn, context->db,
3595 3596
                                   super_context->input_key,
                                   toku_lt_infinity);
3597 3598 3599 3600 3601 3602 3603 3604 3605 3606 3607 3608 3609 3610 3611
        }
        r = grab_range_lock(&request);
    }
    else r = 0;

    //Call application-layer callback if found and locks were successfully obtained.
    if (r==0 && key!=NULL) {
        context->r_user_callback = super_context->f(&found_key, &found_val, context->f_extra);
        if (context->r_user_callback) r = TOKUDB_USER_CALLBACK_ERROR;
    }

    //Give brt-layer an error (if any) to return from toku_brt_cursor_set_range
    return r;
}

3612 3613 3614 3615 3616 3617 3618 3619
static int c_getf_set_range_reverse_callback(ITEMLEN keylen, bytevec key, ITEMLEN vallen, bytevec val, void *extra);

static int
toku_c_getf_set_range_reverse(DBC *c, u_int32_t flag, DBT *key, YDB_CALLBACK_FUNCTION f, void *extra) {
    HANDLE_PANICKED_DB(c->dbp);
    HANDLE_CURSOR_ILLEGAL_WORKING_PARENT_TXN(c);

    QUERY_CONTEXT_WITH_INPUT_S context; //Describes the context of this query.
3620
    num_point_queries++;   // accountability
3621 3622
    query_context_with_input_init(&context, c, flag, key, NULL, f, extra); 
    //toku_brt_cursor_set_range_reverse will call c_getf_set_range_reverse_callback(..., context) (if query is successful)
3623
    int r = toku_brt_cursor_set_range_reverse(dbc_struct_i(c)->c, key, c_getf_set_range_reverse_callback, &context);
3624 3625 3626 3627 3628 3629 3630 3631 3632 3633 3634 3635 3636 3637 3638 3639 3640 3641 3642 3643 3644 3645 3646 3647 3648
    if (r == TOKUDB_USER_CALLBACK_ERROR) r = context.base.r_user_callback;
    return r;
}

//result is the result of the query (i.e. 0 means found, DB_NOTFOUND, etc..)
static int
c_getf_set_range_reverse_callback(ITEMLEN keylen, bytevec key, ITEMLEN vallen, bytevec val, void *extra) {
    QUERY_CONTEXT_WITH_INPUT super_context = extra;
    QUERY_CONTEXT_BASE       context       = &super_context->base;

    int r;

    DBT found_key;
    DBT found_val;
    toku_fill_dbt(&found_key, key, keylen);
    toku_fill_dbt(&found_val, val, vallen);

    //Lock:
    //  left(key) = found ? found_key : -infinity
    //  left(val) = found ? found_val : -infinity
    //  right(key,val)  = (input_key, infinity)
    if (context->do_locking) {
        RANGE_LOCK_REQUEST_S request;
        if (key!=NULL) {
            read_lock_request_init(&request, context->txn, context->db,
3649 3650
                                   &found_key,
                                   super_context->input_key);
3651 3652 3653
        }
        else {
            read_lock_request_init(&request, context->txn, context->db,
3654 3655
                                   toku_lt_neg_infinity,
                                   super_context->input_key);
3656 3657 3658 3659 3660 3661 3662 3663 3664 3665 3666 3667 3668 3669 3670
        }
        r = grab_range_lock(&request);
    }
    else r = 0;

    //Call application-layer callback if found and locks were successfully obtained.
    if (r==0 && key!=NULL) {
        context->r_user_callback = super_context->f(&found_key, &found_val, context->f_extra);
        if (context->r_user_callback) r = TOKUDB_USER_CALLBACK_ERROR;
    }

    //Give brt-layer an error (if any) to return from toku_brt_cursor_set_range_reverse
    return r;
}

3671
static int toku_c_close(DBC * c) {
3672 3673
    HANDLE_PANICKED_DB(c->dbp);
    HANDLE_CURSOR_ILLEGAL_WORKING_PARENT_TXN(c);
3674 3675 3676 3677 3678 3679
    int r = toku_brt_cursor_close(dbc_struct_i(c)->c);
    toku_sdbt_cleanup(&dbc_struct_i(c)->skey_s);
    toku_sdbt_cleanup(&dbc_struct_i(c)->sval_s);
#if !TOKUDB_NATIVE_H
    toku_free(dbc_struct_i(c));
#endif
Bradley C. Kuszmaul's avatar
Bradley C. Kuszmaul committed
3680
    toku_free(c);
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
3681 3682 3683
    return r;
}

3684 3685
static inline int 
keyeq(DBC *c, DBT *a, DBT *b) {
3686 3687 3688 3689
    DB *db = c->dbp;
    return db->i->brt->compare_fun(db, a, b) == 0;
}

3690 3691 3692 3693
// Return the number of entries whose key matches the key currently 
// pointed to by the brt cursor.  
static int 
toku_c_count(DBC *cursor, db_recno_t *count, u_int32_t flags) {
3694 3695
    HANDLE_PANICKED_DB(cursor->dbp);
    HANDLE_CURSOR_ILLEGAL_WORKING_PARENT_TXN(cursor);
Rich Prohaska's avatar
Rich Prohaska committed
3696 3697
    int r;
    DBC *count_cursor = 0;
3698
    DBT currentkey;
Rich Prohaska's avatar
Rich Prohaska committed
3699

3700
    init_dbt_realloc(&currentkey);
3701
    u_int32_t lock_flags = get_cursor_prelocked_flags(flags, cursor);
3702
    flags &= ~lock_flags;
Rich Prohaska's avatar
Rich Prohaska committed
3703 3704 3705 3706
    if (flags != 0) {
        r = EINVAL; goto finish;
    }

3707
    r = toku_c_get_current_unconditional(cursor, lock_flags, &currentkey, NULL);
Rich Prohaska's avatar
Rich Prohaska committed
3708
    if (r != 0) goto finish;
3709 3710 3711 3712 3713 3714

    //TODO: Optimization
    //if (do_locking) {
    //   do a lock from currentkey,-infinity to currentkey,infinity
    //   lock_flags |= DB_PRELOCKED
    //}
Rich Prohaska's avatar
Rich Prohaska committed
3715
    
3716
    r = toku_db_cursor(cursor->dbp, dbc_struct_i(cursor)->txn, &count_cursor, 0, 0);
Rich Prohaska's avatar
Rich Prohaska committed
3717 3718
    if (r != 0) goto finish;

3719
    r = toku_c_getf_set(count_cursor, lock_flags, &currentkey, ydb_getf_do_nothing, NULL);
3720 3721 3722 3723
    if (r==0) {
	*count = 1; // there is a key, so the count is one (since we don't have DUP dbs anymore, the only answers are 0 or 1.
    } else {
	*count = 0;
Rich Prohaska's avatar
Rich Prohaska committed
3724
    }
3725
    r = 0;
Rich Prohaska's avatar
Rich Prohaska committed
3726 3727 3728 3729 3730 3731 3732 3733
finish:
    if (currentkey.data) toku_free(currentkey.data);
    if (count_cursor) {
        int rr = toku_c_close(count_cursor); assert(rr == 0);
    }
    return r;
}

Yoni Fogel's avatar
 
Yoni Fogel committed
3734

3735 3736
///////////
//db_getf_XXX is equivalent to c_getf_XXX, without a persistent cursor
Yoni Fogel's avatar
 
Yoni Fogel committed
3737

3738 3739
static int
db_getf_set(DB *db, DB_TXN *txn, u_int32_t flags, DBT *key, YDB_CALLBACK_FUNCTION f, void *extra) {
3740 3741
    HANDLE_PANICKED_DB(db);
    HANDLE_DB_ILLEGAL_WORKING_PARENT_TXN(db, txn);
3742
    DBC *c;
3743 3744 3745
    uint32_t iso_flags = flags & DB_ISOLATION_FLAGS;
    flags &= ~DB_ISOLATION_FLAGS;
    int r = toku_db_cursor(db, txn, &c, iso_flags, 1);
3746 3747 3748 3749
    if (r==0) {
        r = toku_c_getf_set(c, flags, key, f, extra);
        int r2 = toku_c_close(c);
        if (r==0) r = r2;
Yoni Fogel's avatar
 
Yoni Fogel committed
3750
    }
3751
    return r;
Yoni Fogel's avatar
 
Yoni Fogel committed
3752 3753
}

3754
////////////
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
3755

3756 3757 3758 3759 3760 3761 3762 3763 3764
static int 
get_point_lock(DB *db, DB_TXN *txn, const DBT *key) {
    RANGE_LOCK_REQUEST_S request;
    //Left end of range == right end of range (point lock)
    write_lock_request_init(&request, txn, db, key, key);
    int r = grab_range_lock(&request);
    return r;
}

3765 3766
static int
toku_db_del(DB *db, DB_TXN *txn, DBT *key, u_int32_t flags) {
3767
    HANDLE_PANICKED_DB(db);
3768
    HANDLE_DB_ILLEGAL_WORKING_PARENT_TXN(db, txn);
3769

3770 3771
    u_int32_t unchecked_flags = flags;
    //DB_DELETE_ANY means delete regardless of whether it exists in the db.
3772
    BOOL error_if_missing = (BOOL)(!(flags&DB_DELETE_ANY));
3773
    unchecked_flags &= ~DB_DELETE_ANY;
3774
    u_int32_t lock_flags = get_prelocked_flags(flags);
3775
    unchecked_flags &= ~lock_flags;
3776
    BOOL do_locking = (BOOL)(db->i->lt && !(lock_flags&DB_PRELOCKED_WRITE));
3777

3778
    int r = 0;
3779 3780 3781 3782 3783
    if (unchecked_flags!=0) 
        r = EINVAL;

    if (r == 0) {
        r = toku_grab_read_lock_on_directory(db, txn);
Zardosht Kasheff's avatar
Zardosht Kasheff committed
3784
    }
3785
    if (r == 0 && error_if_missing) {
3786
        //Check if the key exists in the db.
3787
        r = db_getf_set(db, txn, lock_flags|DB_SERIALIZABLE, key, ydb_getf_do_nothing, NULL);
3788
    }
3789
    if (r == 0 && do_locking) {
3790
        //Do locking if necessary.
3791
        r = get_point_lock(db, txn, key);
3792
    }
3793
    if (r == 0) {
3794
        //Do the actual deleting.
3795
        num_deletes++;       // accountability 
3796
        r = toku_brt_delete(db->i->brt, key, txn ? db_txn_struct_i(txn)->tokutxn : 0);
3797
    }
3798 3799

    if (r) 
Zardosht Kasheff's avatar
Zardosht Kasheff committed
3800
        num_deletes_fail++;
3801 3802 3803 3804 3805 3806 3807 3808 3809 3810 3811

    return r;
}

static int
log_del_single(DB_TXN *txn, BRT brt, const DBT *key) {
    TOKUTXN ttxn = db_txn_struct_i(txn)->tokutxn;
    int r = toku_brt_log_del(ttxn, brt, key);
    return r;
}

3812 3813 3814 3815 3816 3817 3818 3819
static uint32_t
sum_size(uint32_t num_keys, DBT keys[], uint32_t overhead) {
    uint32_t sum = 0;
    for (uint32_t i = 0; i < num_keys; i++) 
        sum += keys[i].size + overhead;
    return sum;
}

3820
static int
3821
log_del_multiple(DB_TXN *txn, DB *src_db, const DBT *key, const DBT *val, uint32_t num_dbs, BRT brts[], DBT keys[]) {
3822 3823 3824 3825
    int r = 0;
    if (num_dbs > 0) {
        TOKUTXN ttxn = db_txn_struct_i(txn)->tokutxn;
        BRT src_brt  = src_db ? src_db->i->brt : NULL;
3826 3827 3828 3829 3830 3831 3832 3833 3834
        const uint32_t log_entry_overhead = 24; // rough approximation of the log entry overhead for deletes
        uint32_t del_multiple_size = key->size + val->size + log_entry_overhead;
        uint32_t del_single_sizes = sum_size(num_dbs, keys, log_entry_overhead);
        if (del_single_sizes < del_multiple_size) {
            for (uint32_t i = 0; r == 0 && i < num_dbs; i++)
                r = log_del_single(txn, brts[i], &keys[i]);
        } else {
            r = toku_brt_log_del_multiple(ttxn, src_brt, brts, num_dbs, key, val);
        }
3835 3836 3837 3838
    }
    return r;
}

3839 3840 3841 3842 3843 3844 3845 3846 3847
static uint32_t 
lookup_src_db(uint32_t num_dbs, DB *db_array[], DB *src_db) {
    uint32_t which_db;
    for (which_db = 0; which_db < num_dbs; which_db++) 
        if (db_array[which_db] == src_db)
            break;
    return which_db;
}

3848 3849 3850 3851 3852 3853
static int
do_del_multiple(DB_TXN *txn, uint32_t num_dbs, DB *db_array[], DBT keys[]) {
    int r = 0;
    TOKUTXN ttxn = db_txn_struct_i(txn)->tokutxn;
    for (uint32_t which_db = 0; r == 0 && which_db < num_dbs; which_db++) {
        DB *db = db_array[which_db];
3854 3855 3856 3857 3858 3859 3860 3861 3862 3863 3864 3865 3866 3867 3868
	int do_delete = TRUE;
	DB_INDEXER *indexer = toku_db_get_indexer(db);
	if (indexer) {
            DB *src_db = toku_indexer_get_src_db(indexer);
            invariant(src_db != NULL);
            uint32_t which_src_db = lookup_src_db(num_dbs, db_array, src_db);
            if (which_src_db >= num_dbs)
                r = EINVAL;
            else
                do_delete = !toku_indexer_is_key_right_of_le_cursor(indexer, src_db, &keys[which_src_db]);
        }
	if (r == 0 && do_delete) {
            num_deletes++;
            r = toku_brt_maybe_delete(db->i->brt, &keys[which_db], ttxn, FALSE, ZERO_LSN, FALSE);
        }
Zardosht Kasheff's avatar
Zardosht Kasheff committed
3869
    }
3870
    return r;
Rich Prohaska's avatar
Rich Prohaska committed
3871 3872
}

3873
static int
3874 3875 3876 3877 3878 3879 3880 3881 3882 3883 3884 3885 3886
env_del_multiple(
    DB_ENV *env, 
    DB *src_db, 
    DB_TXN *txn, 
    const DBT *key, 
    const DBT *val, 
    uint32_t num_dbs, 
    DB **db_array, 
    DBT *keys, 
    uint32_t *flags_array, 
    void *extra
    ) 
{
3887
    int r;
3888
    DBT del_keys[num_dbs];
3889
    BOOL multi_accounting = TRUE;  // use num_multi_delete accountability counters 
3890 3891 3892

    // special case single DB
    if (num_dbs == 1 && src_db == db_array[0]) {
3893
	multi_accounting = FALSE;
3894 3895 3896 3897 3898 3899 3900
        r = toku_db_del(db_array[0], txn, (DBT *) key, flags_array[0]);
        goto cleanup;
    }

    HANDLE_PANICKED_ENV(env);

    if (!txn) {
3901 3902 3903
        r = EINVAL;
        goto cleanup;
    }
3904
    if (!env->i->generate_row_for_del) {
3905 3906 3907 3908
        r = EINVAL;
        goto cleanup;
    }

3909 3910 3911 3912 3913 3914 3915 3916
    HANDLE_ILLEGAL_WORKING_PARENT_TXN(env, txn);

    {
    uint32_t lock_flags[num_dbs];
    uint32_t remaining_flags[num_dbs];
    BRT brts[num_dbs];

    for (uint32_t which_db = 0; which_db < num_dbs; which_db++) {
3917
        DB *db = db_array[which_db];
3918

3919 3920 3921 3922
        if (db == src_db) {
            del_keys[which_db] = *key;
        }
        else {
3923
        //Generate the key
3924 3925 3926 3927
            r = env->i->generate_row_for_del(db, src_db, &keys[which_db], key, val, extra);
            if (r != 0) goto cleanup;
            del_keys[which_db] = keys[which_db];
        }
3928
        lock_flags[which_db] = get_prelocked_flags(flags_array[which_db]);
3929
        remaining_flags[which_db] = flags_array[which_db] & ~lock_flags[which_db];
3930 3931 3932 3933 3934 3935 3936 3937

        if (remaining_flags[which_db] & ~DB_DELETE_ANY) {
            r = EINVAL;
            goto cleanup;
        }
        BOOL error_if_missing = (BOOL)(!(remaining_flags[which_db]&DB_DELETE_ANY));
        if (error_if_missing) {
            //Check if the key exists in the db.
3938
            r = db_getf_set(db, txn, lock_flags[which_db]|DB_SERIALIZABLE, &del_keys[which_db], ydb_getf_do_nothing, NULL);
3939
            if (r != 0) goto cleanup;
3940 3941 3942
        }

        //Do locking if necessary.
Zardosht Kasheff's avatar
Zardosht Kasheff committed
3943
        r = toku_grab_read_lock_on_directory(db, txn);
3944 3945
        if (r != 0) goto cleanup;

3946 3947
        if (db->i->lt && !(lock_flags[which_db] & DB_PRELOCKED_WRITE)) {
            //Needs locking
3948
            r = get_point_lock(db, txn, &del_keys[which_db]);
3949
            if (r != 0) goto cleanup;
3950 3951 3952
        }
        brts[which_db] = db->i->brt;
    }
3953 3954

    if (num_dbs == 1)
3955
        r = log_del_single(txn, brts[0], &del_keys[0]);
3956
    else
3957
        r = log_del_multiple(txn, src_db, key, val, num_dbs, brts, del_keys);
3958 3959

    if (r == 0) 
3960
        r = do_del_multiple(txn, num_dbs, db_array, del_keys);
3961 3962 3963
    }

cleanup:
3964
    if (multi_accounting) {
3965
	if (r == 0)
3966 3967 3968 3969
	    num_multi_deletes += num_dbs;
	else
	    num_multi_deletes_fail += num_dbs;
    }
3970 3971 3972 3973
    return r;
}


3974 3975
static int 
locked_c_get(DBC * c, DBT * key, DBT * data, u_int32_t flag) {
3976 3977 3978 3979
    //{ unsigned int i; printf("cget flags=%d keylen=%d key={", flag, key->size); for(i=0; i<key->size; i++) printf("%d,", ((char*)key->data)[i]); printf("} datalen=%d data={", data->size); for(i=0; i<data->size; i++) printf("%d,", ((char*)data->data)[i]); printf("}\n"); }
    toku_ydb_lock(); int r = toku_c_get(c, key, data, flag); toku_ydb_unlock();
    //{ unsigned int i; printf("cgot r=%d keylen=%d key={", r, key->size); for(i=0; i<key->size; i++) printf("%d,", ((char*)key->data)[i]); printf("} datalen=%d data={", data->size); for(i=0; i<data->size; i++) printf("%d,", ((char*)data->data)[i]); printf("}\n"); }
    return r;
Rich Prohaska's avatar
Rich Prohaska committed
3980 3981
}

3982 3983
static int 
locked_c_close(DBC * c) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
3984
    toku_ydb_lock(); int r = toku_c_close(c); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
3985 3986
}

3987 3988
static int 
locked_c_count(DBC *cursor, db_recno_t *count, u_int32_t flags) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
3989
    toku_ydb_lock(); int r = toku_c_count(cursor, count, flags); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
3990 3991
}

3992 3993
static int 
locked_c_del(DBC * c, u_int32_t flags) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
3994
    toku_ydb_lock(); int r = toku_c_del(c, flags); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
3995 3996
}

3997 3998
static int locked_c_pre_acquire_read_lock(DBC *dbc, const DBT *key_left, const DBT *key_right);

3999 4000
static int 
toku_db_cursor(DB * db, DB_TXN * txn, DBC ** c, u_int32_t flags, int is_temporary_cursor) {
4001
    HANDLE_PANICKED_DB(db);
4002
    HANDLE_DB_ILLEGAL_WORKING_PARENT_TXN(db, txn);
4003
    DB_ENV* env = db->dbenv;
Zardosht Kasheff's avatar
Zardosht Kasheff committed
4004
    int r;
4005
    size_t result_size = sizeof(DBC)+sizeof(struct __toku_dbc_internal); // internal stuff stuck on the end
4006 4007 4008 4009 4010 4011 4012 4013 4014 4015 4016
    if (!(flags == 0 || 
          flags == DB_SERIALIZABLE || 
          flags == DB_INHERIT_ISOLATION)
       ) 
    {
        return toku_ydb_do_error(
            env, 
            EINVAL, 
            "Invalid isolation flags set for toku_db_cursor\n"
            );
    }
Zardosht Kasheff's avatar
Zardosht Kasheff committed
4017 4018 4019 4020 4021
    r = toku_grab_read_lock_on_directory(db, txn);
    if (r != 0) {
        return r;
    }
    
4022
    DBC *result = toku_malloc(result_size);
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
4023 4024
    if (result == 0)
        return ENOMEM;
4025
    memset(result, 0, result_size);
Yoni Fogel's avatar
Yoni Fogel committed
4026 4027 4028 4029 4030
#define SCRS(name) result->name = locked_ ## name
    SCRS(c_get);
    SCRS(c_close);
    SCRS(c_del);
    SCRS(c_count);
Yoni Fogel's avatar
Yoni Fogel committed
4031 4032
    SCRS(c_getf_first);
    SCRS(c_getf_last);
Yoni Fogel's avatar
Yoni Fogel committed
4033
    SCRS(c_getf_next);
4034 4035 4036 4037 4038
    SCRS(c_getf_prev);
    SCRS(c_getf_current);
    SCRS(c_getf_current_binding);
    SCRS(c_getf_set);
    SCRS(c_getf_set_range);
4039
    SCRS(c_getf_set_range_reverse);
4040
    SCRS(c_pre_acquire_read_lock);
Yoni Fogel's avatar
Yoni Fogel committed
4041
#undef SCRS
4042 4043 4044

#if !TOKUDB_NATIVE_H
    MALLOC(result->i); // otherwise it is allocated as part of result->ii
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
4045
    assert(result->i);
4046
#endif
4047
    result->dbp = db;
4048 4049 4050
    dbc_struct_i(result)->txn = txn;
    dbc_struct_i(result)->skey_s = (struct simple_dbt){0,0};
    dbc_struct_i(result)->sval_s = (struct simple_dbt){0,0};
4051
    if (is_temporary_cursor) {
4052 4053
	dbc_struct_i(result)->skey = &db->i->skey;
	dbc_struct_i(result)->sval = &db->i->sval;
4054
    } else {
4055 4056
	dbc_struct_i(result)->skey = &dbc_struct_i(result)->skey_s;
	dbc_struct_i(result)->sval = &dbc_struct_i(result)->sval_s;
4057
    }
4058 4059 4060 4061 4062 4063 4064 4065 4066
    switch(flags) {
        case (DB_SERIALIZABLE):
            dbc_struct_i(result)->iso = TOKU_ISO_SERIALIZABLE;
            break;
        default:
            dbc_struct_i(result)->iso = txn ? db_txn_struct_i(txn)->iso : TOKU_ISO_SERIALIZABLE;
            break;
    }
    BOOL is_snapshot_read = FALSE;
4067
    if (txn) {
4068 4069
        is_snapshot_read = (dbc_struct_i(result)->iso == TOKU_ISO_READ_COMMITTED || 
                            dbc_struct_i(result)->iso == TOKU_ISO_SNAPSHOT);
4070
    }
Zardosht Kasheff's avatar
Zardosht Kasheff committed
4071
    r = toku_brt_cursor(
4072
        db->i->brt, 
4073 4074 4075
        &dbc_struct_i(result)->c,
        txn ? db_txn_struct_i(txn)->tokutxn : NULL,
        is_snapshot_read
4076
        );
4077 4078 4079 4080 4081 4082 4083 4084 4085 4086 4087
    assert(r == 0 || r == TOKUDB_MVCC_DICTIONARY_TOO_NEW);
    if (r == 0) {
        *c = result;
    }
    else {
#if !TOKUDB_NATIVE_H
        toku_free(result->i); // otherwise it is allocated as part of result->ii
#endif
        toku_free(result);
    }
    return r;
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
4088 4089
}

4090 4091
static inline int 
db_thread_need_flags(DBT *dbt) {
Rich Prohaska's avatar
Rich Prohaska committed
4092 4093 4094
    return (dbt->flags & (DB_DBT_MALLOC+DB_DBT_REALLOC+DB_DBT_USERMEM)) == 0;
}

4095 4096
static int 
toku_db_get (DB * db, DB_TXN * txn, DBT * key, DBT * data, u_int32_t flags) {
4097
    HANDLE_PANICKED_DB(db);
4098
    HANDLE_DB_ILLEGAL_WORKING_PARENT_TXN(db, txn);
Yoni Fogel's avatar
Yoni Fogel committed
4099
    int r;
4100
    u_int32_t iso_flags = flags & DB_ISOLATION_FLAGS;
4101

Rich Prohaska's avatar
Rich Prohaska committed
4102
    if ((db->i->open_flags & DB_THREAD) && db_thread_need_flags(data))
4103 4104
        return EINVAL;

4105
    u_int32_t lock_flags = flags & (DB_PRELOCKED | DB_PRELOCKED_WRITE);
Yoni Fogel's avatar
Yoni Fogel committed
4106
    flags &= ~lock_flags;
4107
    flags &= ~DB_ISOLATION_FLAGS;
4108 4109 4110
    // And DB_GET_BOTH is no longer supported. #2862.
    if (flags != 0) return EINVAL;

Yoni Fogel's avatar
Yoni Fogel committed
4111 4112

    DBC *dbc;
4113
    r = toku_db_cursor(db, txn, &dbc, iso_flags, 1);
Yoni Fogel's avatar
Yoni Fogel committed
4114
    if (r!=0) return r;
4115
    u_int32_t c_get_flags = DB_SET;
Yoni Fogel's avatar
Yoni Fogel committed
4116
    r = toku_c_get(dbc, key, data, c_get_flags | lock_flags);
Yoni Fogel's avatar
Yoni Fogel committed
4117 4118
    int r2 = toku_c_close(dbc);
    return r ? r : r2;
4119 4120
}

Rich Prohaska's avatar
Rich Prohaska committed
4121
#if 0
4122 4123
static int 
toku_db_key_range(DB * db, DB_TXN * txn, DBT * dbt, DB_KEY_RANGE * kr, u_int32_t flags) {
4124
    HANDLE_PANICKED_DB(db);
4125
    HANDLE_DB_ILLEGAL_WORKING_PARENT_TXN(db, txn);
4126
    txn=txn; dbt=dbt; kr=kr; flags=flags;
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
4127
    toku_ydb_barf();
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
4128
    abort();
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
4129
}
Rich Prohaska's avatar
Rich Prohaska committed
4130
#endif
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
4131

4132 4133
static int 
toku_db_lt_panic(DB* db, int r) {
4134
    assert(r!=0);
Yoni Fogel's avatar
Yoni Fogel committed
4135 4136
    assert(db && db->i && db->dbenv && db->dbenv->i);
    DB_ENV* env = db->dbenv;
4137 4138 4139 4140 4141
    env->i->is_panicked = r;

    if (r < 0) env->i->panic_string = toku_strdup(toku_lt_strerror((TOKU_LT_ERROR)r));
    else       env->i->panic_string = toku_strdup("Error in locktree.\n");

4142
    return toku_ydb_do_error(env, r, "%s", env->i->panic_string);
Yoni Fogel's avatar
Yoni Fogel committed
4143 4144
}

4145 4146
static int 
toku_txn_add_lt(DB_TXN* txn, toku_lock_tree* lt) {
Yoni Fogel's avatar
Yoni Fogel committed
4147
    int r = ENOSYS;
Yoni Fogel's avatar
Yoni Fogel committed
4148
    assert(txn && lt);
4149
    toku_lth* lth = db_txn_struct_i(txn)->lth;
Yoni Fogel's avatar
Yoni Fogel committed
4150 4151 4152 4153 4154
    assert(lth);

    toku_lock_tree* find = toku_lth_find(lth, lt);
    if (find) {
        assert(find == lt);
Yoni Fogel's avatar
Yoni Fogel committed
4155 4156
        r = 0;
        goto cleanup;
Yoni Fogel's avatar
Yoni Fogel committed
4157
    }
Yoni Fogel's avatar
Yoni Fogel committed
4158 4159 4160 4161 4162 4163
    r = toku_lth_insert(lth, lt);
    if (r != 0) { goto cleanup; }
    
    toku_lt_add_ref(lt);
    r = 0;
cleanup:
Yoni Fogel's avatar
Yoni Fogel committed
4164 4165 4166
    return r;
}

4167 4168
static toku_dbt_cmp 
toku_db_get_compare_fun(DB* db) {
Yoni Fogel's avatar
Yoni Fogel committed
4169 4170
    return db->i->brt->compare_fun;
}
Yoni Fogel's avatar
Yoni Fogel committed
4171

4172
/***** TODO 2216 delete this 
4173 4174
static int 
toku_db_fd(DB *db, int *fdp) {
4175 4176 4177 4178
    HANDLE_PANICKED_DB(db);
    if (!db_opened(db)) return EINVAL;
    return toku_brt_get_fd(db->i->brt, fdp);
}
4179
*******/
4180

4181 4182 4183 4184 4185
static int
db_open_subdb(DB * db, DB_TXN * txn, const char *fname, const char *dbname, DBTYPE dbtype, u_int32_t flags, int mode) {
    int r;
    if (!fname || !dbname) r = EINVAL;
    else {
4186 4187 4188 4189 4190 4191
        char subdb_full_name[strlen(fname) + sizeof("/") + strlen(dbname)];
        int bytes = snprintf(subdb_full_name, sizeof(subdb_full_name), "%s/%s", fname, dbname);
        assert(bytes==(int)sizeof(subdb_full_name)-1);
        const char *null_subdbname = NULL;
        r = toku_db_open(db, txn, subdb_full_name, null_subdbname, dbtype, flags, mode);
    }
4192 4193
    return r;
}
4194

4195 4196 4197 4198 4199 4200 4201 4202
static void
create_iname_hint(const char *dname, char *hint) {
    //Requires: size of hint array must be > strlen(dname)
    //Copy alphanumeric characters only.
    //Replace strings of non-alphanumeric characters with a single underscore.
    BOOL underscored = FALSE;
    while (*dname) {
        if (isalnum(*dname)) {
4203 4204
            char c = *dname++;
            *hint++ = c;
4205 4206 4207 4208 4209 4210 4211
            underscored = FALSE;
        }
        else {
            if (!underscored)
                *hint++ = '_';
            dname++;
            underscored = TRUE;
4212 4213
        }
    }
4214 4215 4216
    *hint = '\0';
}

4217 4218 4219

// n >= 0 means to include "_L_" with hex value of n in iname
// (intended for use by loader, which will create many inames using one txnid).
4220
static char *
4221 4222
create_iname(DB_ENV *env, u_int64_t id, char *hint, int n) {
    int bytes;
4223
    char inamebase[strlen(hint) +
4224 4225 4226 4227 4228
		   8 +  // hex file format version
		   16 + // hex id (normally the txnid)
		   8  + // hex value of n if non-neg
		   sizeof("_L___.tokudb")]; // extra pieces
    if (n < 0)
4229 4230 4231
	bytes = snprintf(inamebase, sizeof(inamebase),
                         "%s_%"PRIx64"_%"PRIx32            ".tokudb",
                         hint, id, BRT_LAYOUT_VERSION);
4232
    else
4233 4234 4235
	bytes = snprintf(inamebase, sizeof(inamebase),
                         "%s_%"PRIx64"_%"PRIx32"_L_%"PRIx32".tokudb",
                         hint, id, BRT_LAYOUT_VERSION, n);
4236 4237 4238 4239
    assert(bytes>0);
    assert(bytes<=(int)sizeof(inamebase)-1);
    char *rval;
    if (env->i->data_dir)
Yoni Fogel's avatar
Yoni Fogel committed
4240
        rval = toku_construct_full_name(2, env->i->data_dir, inamebase);
4241
    else
Yoni Fogel's avatar
Yoni Fogel committed
4242
        rval = toku_construct_full_name(1, inamebase);
4243 4244
    assert(rval);
    return rval;
4245 4246
}

4247 4248 4249 4250 4251 4252 4253 4254 4255 4256 4257 4258 4259

static int db_open_iname(DB * db, DB_TXN * txn, const char *iname, u_int32_t flags, int mode);


// inames are created here.
// algorithm:
//  begin txn
//  convert dname to iname (possibly creating new iname)
//  open file (toku_brt_open() will handle logging)
//  close txn
//  if created a new iname, take full range lock
static int 
toku_db_open(DB * db, DB_TXN * txn, const char *fname, const char *dbname, DBTYPE dbtype, u_int32_t flags, int mode) {
4260
    HANDLE_PANICKED_DB(db);
4261
    HANDLE_DB_ILLEGAL_WORKING_PARENT_TXN(db, txn);
4262
    if (dbname!=NULL) 
4263
        return db_open_subdb(db, txn, fname, dbname, dbtype, flags, mode);
4264

4265
    // at this point fname is the dname
4266 4267
    //This code ONLY supports single-db files.
    assert(dbname==NULL);
4268
    const char * dname = fname;  // db_open_subdb() converts (fname, dbname) to dname
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
4269

4270 4271 4272
    ////////////////////////////// do some level of parameter checking.
    u_int32_t unused_flags = flags;
    int using_txns = db->dbenv->i->open_flags & DB_INIT_TXN;
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
4273
    int r;
Yoni Fogel's avatar
Yoni Fogel committed
4274
    if (dbtype!=DB_BTREE && dbtype!=DB_UNKNOWN) return EINVAL;
4275 4276 4277
    int is_db_excl    = flags & DB_EXCL;    unused_flags&=~DB_EXCL;
    int is_db_create  = flags & DB_CREATE;  unused_flags&=~DB_CREATE;

4278
    //We support READ_UNCOMMITTED and READ_COMMITTED whether or not the flag is provided.
4279
                                            unused_flags&=~DB_READ_UNCOMMITTED;
4280
                                            unused_flags&=~DB_READ_COMMITTED;
4281
                                            unused_flags&=~DB_SERIALIZABLE;
4282 4283 4284 4285 4286 4287 4288 4289 4290 4291 4292 4293 4294 4295 4296 4297 4298 4299
    if (unused_flags & ~DB_THREAD) return EINVAL; // unknown flags

    if (is_db_excl && !is_db_create) return EINVAL;
    if (dbtype==DB_UNKNOWN && is_db_excl) return EINVAL;

    /* tokudb supports no duplicates and sorted duplicates only */
    unsigned int tflags;
    r = toku_brt_get_flags(db->i->brt, &tflags);
    if (r != 0) 
        return r;

    if (db_opened(db))
        return EINVAL;              /* It was already open. */
    //////////////////////////////

    DB_TXN *child = NULL;
    // begin child (unless transactionless)
    if (using_txns) {
4300 4301
        r = toku_txn_begin(db->dbenv, txn, &child, DB_TXN_NOSYNC, 1);
        assert(r==0);
4302 4303 4304 4305 4306 4307
    }

    // convert dname to iname
    //  - look up dname, get iname
    //  - if dname does not exist, create iname and make entry in directory
    DBT dname_dbt;  // holds dname
4308
    DBT iname_dbt;  // holds iname_in_env
4309 4310
    toku_fill_dbt(&dname_dbt, dname, strlen(dname)+1);
    init_dbt_realloc(&iname_dbt);  // sets iname_dbt.data = NULL
4311
    r = toku_db_get(db->dbenv->i->directory, child, &dname_dbt, &iname_dbt, DB_SERIALIZABLE);  // allocates memory for iname
4312 4313 4314 4315 4316 4317 4318
    char *iname = iname_dbt.data;
    if (r==DB_NOTFOUND && !is_db_create)
        r = ENOENT;
    else if (r==0 && is_db_excl) {
        r = EEXIST;
    }
    else if (r==DB_NOTFOUND) {
4319
        char hint[strlen(dname) + 1];
4320

4321 4322 4323 4324 4325 4326 4327
        // create iname and make entry in directory
        u_int64_t id = 0;

        if (using_txns) {
            id = toku_txn_get_txnid(db_txn_struct_i(child)->tokutxn);
        }
        create_iname_hint(dname, hint);
4328
        iname = create_iname(db->dbenv, id, hint, -1);  // allocated memory for iname
4329 4330 4331 4332 4333 4334
        toku_fill_dbt(&iname_dbt, iname, strlen(iname) + 1);
        r = toku_db_put(db->dbenv->i->directory, child, &dname_dbt, &iname_dbt, DB_YESOVERWRITE);  // DB_YESOVERWRITE for performance only, avoid unnecessary query
    }

    // we now have an iname
    if (r == 0) {
4335
        r = db_open_iname(db, child, iname, flags, mode);
4336 4337 4338 4339 4340 4341 4342 4343 4344 4345
        if (r==0) {
            db->i->dname = toku_xstrdup(dname);
            env_note_db_opened(db->dbenv, db);  // tell env that a new db handle is open (using dname)
        }
    }

    // free string holding iname
    if (iname) toku_free(iname);

    if (using_txns) {
4346 4347 4348 4349 4350 4351 4352 4353 4354
        // close txn
        if (r == 0) {  // commit
            r = toku_txn_commit(child, DB_TXN_NOSYNC, NULL, NULL);
            assert(r==0);  // TODO panic
        }
        else {         // abort
            int r2 = toku_txn_abort(child, NULL, NULL);
            assert(r2==0);  // TODO panic
        }
4355 4356 4357 4358 4359 4360
    }

    return r;
}

static int 
4361
db_open_iname(DB * db, DB_TXN * txn, const char *iname_in_env, u_int32_t flags, int mode) {
4362 4363 4364 4365 4366 4367 4368 4369 4370 4371 4372
    int r;

    //Set comparison functions if not yet set.
    if (!db->i->key_compare_was_set && db->dbenv->i->bt_compare) {
        r = toku_brt_set_bt_compare(db->i->brt, db->dbenv->i->bt_compare);
        assert(r==0);
        db->i->key_compare_was_set = TRUE;
    }
    BOOL need_locktree = (BOOL)((db->dbenv->i->open_flags & DB_INIT_LOCK) &&
                                (db->dbenv->i->open_flags & DB_INIT_TXN));

4373 4374
    int is_db_excl    = flags & DB_EXCL;    flags&=~DB_EXCL;
    int is_db_create  = flags & DB_CREATE;  flags&=~DB_CREATE;
4375
    //We support READ_UNCOMMITTED and READ_COMMITTED whether or not the flag is provided.
4376
                                            flags&=~DB_READ_UNCOMMITTED;
4377
                                            flags&=~DB_READ_COMMITTED;
4378
                                            flags&=~DB_SERIALIZABLE;
4379
    if (flags & ~DB_THREAD) return EINVAL; // unknown flags
4380 4381

    if (is_db_excl && !is_db_create) return EINVAL;
4382

4383 4384 4385 4386 4387 4388
    /* tokudb supports no duplicates and sorted duplicates only */
    unsigned int tflags;
    r = toku_brt_get_flags(db->i->brt, &tflags);
    if (r != 0) 
        return r;

4389
    if (db_opened(db))
4390
        return EINVAL;              /* It was already open. */
Yoni Fogel's avatar
Yoni Fogel committed
4391
    
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
4392 4393
    db->i->open_flags = flags;
    db->i->open_mode = mode;
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
4394

Yoni Fogel's avatar
Yoni Fogel committed
4395
    r = toku_brt_open(db->i->brt, iname_in_env,
4396
		      is_db_create, is_db_excl,
4397
		      db->dbenv->i->cachetable,
4398
		      txn ? db_txn_struct_i(txn)->tokutxn : NULL_TXN,
4399
		      db);
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
4400 4401 4402
    if (r != 0)
        goto error_cleanup;

4403
    db->i->opened = 1;
Yoni Fogel's avatar
Yoni Fogel committed
4404
    if (need_locktree) {
4405
	db->i->dict_id = toku_brt_get_dictionary_id(db->i->brt);
4406
        r = toku_ltm_get_lt(db->dbenv->i->ltm, &db->i->lt, db->i->dict_id, db);
Yoni Fogel's avatar
Yoni Fogel committed
4407
        if (r!=0) { goto error_cleanup; }
Yoni Fogel's avatar
Yoni Fogel committed
4408
    }
4409 4410 4411
    //Add to transaction's list of 'must close' if necessary.
    if (txn) {
        //Do last so we don't have to undo.
4412
        toku_list_push(&db_txn_struct_i(txn)->dbs_that_must_close_before_abort,
4413 4414
                  &db->i->dbs_that_must_close_before_abort);
    }
Yoni Fogel's avatar
Yoni Fogel committed
4415

Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
4416
    return 0;
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
4417 4418
 
error_cleanup:
4419
    db->i->dict_id = DICTIONARY_ID_NONE;
4420
    db->i->opened = 0;
Yoni Fogel's avatar
Yoni Fogel committed
4421
    if (db->i->lt) {
4422
        toku_lt_remove_db_ref(db->i->lt, db);
Yoni Fogel's avatar
Yoni Fogel committed
4423 4424
        db->i->lt = NULL;
    }
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
4425
    return r;
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
4426
}
Bradley C. Kuszmaul's avatar
Bradley C. Kuszmaul committed
4427

4428 4429 4430 4431 4432
//Return 0 if proposed pair do not violate size constraints of DB
//(insertion is legal)
//Return non zero otherwise.
static int
db_put_check_size_constraints(DB *db, DBT *key, DBT *val) {
4433
    int r;
4434

4435
    //Check limits on size of key and val.
4436
    unsigned int nodesize;
4437
    r = toku_brt_get_nodesize(db->i->brt, &nodesize); assert(r == 0);
4438

4439 4440 4441 4442 4443 4444
    u_int32_t limit = nodesize / BRT_FANOUT;
    if (key->size > limit)
	r = toku_ydb_do_error(db->dbenv, EINVAL, "The largest key allowed is %u bytes", limit);
    else if (val->size > nodesize)
	r = toku_ydb_do_error(db->dbenv, EINVAL, "The largest value allowed is %u bytes", nodesize);
    
4445 4446 4447
    return r;
}

4448 4449 4450 4451 4452 4453 4454 4455 4456 4457 4458 4459 4460 4461 4462 4463 4464 4465 4466 4467 4468
//Return 0 if supported.
//Return ERANGE if out of range.
static int
db_row_size_supported(DB *db, u_int32_t size) {
    DBT key, val;

    toku_fill_dbt(&key, NULL, size);
    toku_fill_dbt(&val, NULL, 0);
    int r = db_put_check_size_constraints(db, &key, &val);
    if (r!=0) r = ERANGE;
    return r;
}

static int
locked_db_row_size_supported(DB *db, u_int32_t size) {
    toku_ydb_lock();
    int r = db_row_size_supported(db, size);
    toku_ydb_unlock();
    return r;
}

4469 4470
//Return 0 if insert is legal
static int
4471
db_put_check_overwrite_constraint(DB *db, DB_TXN *txn, DBT *key,
4472 4473 4474 4475 4476 4477 4478 4479
                                  u_int32_t lock_flags, u_int32_t overwrite_flag) {
    int r;

    //DB_YESOVERWRITE does not impose constraints.
    if (overwrite_flag==DB_YESOVERWRITE) r = 0;
    else if (overwrite_flag==DB_NOOVERWRITE) {
        //Check if (key,anything) exists in dictionary.
        //If exists, fail.  Otherwise, do insert.
4480
        r = db_getf_set(db, txn, lock_flags|DB_SERIALIZABLE, key, ydb_getf_do_nothing, NULL);
4481 4482 4483 4484 4485 4486 4487
        if (r==DB_NOTFOUND) r = 0;
        else if (r==0)      r = DB_KEYEXIST;
        //Any other error is passed through.
    }
    else if (overwrite_flag==0) {
        //in a nodup db:   overwrite_flag==0 is an alias for DB_YESOVERWRITE
        //in a dupsort db: overwrite_flag==0 is an error
4488
	r = 0;
4489
    }
4490 4491 4492
    else if (overwrite_flag==DB_NOOVERWRITE_NO_ERROR) {
        r = 0;
    }
4493 4494 4495
    else {
        //Other flags are not (yet) supported.
        r = EINVAL;
4496 4497 4498 4499
    }
    return r;
}

4500 4501
static int
toku_db_put(DB *db, DB_TXN *txn, DBT *key, DBT *val, u_int32_t flags) {
4502
    HANDLE_PANICKED_DB(db);
4503
    HANDLE_DB_ILLEGAL_WORKING_PARENT_TXN(db, txn);
4504 4505
    int r;

4506
    num_inserts++;
4507
    u_int32_t lock_flags = get_prelocked_flags(flags);
4508
    flags &= ~lock_flags;
4509

Zardosht Kasheff's avatar
Zardosht Kasheff committed
4510
    r = toku_grab_read_lock_on_directory(db, txn);
4511

4512 4513 4514
    if (r == 0)
        r = db_put_check_size_constraints(db, key, val);
    if (r == 0) {
4515
        //Do any checking required by the flags.
4516
        r = db_put_check_overwrite_constraint(db, txn, key, lock_flags, flags);
4517
    }
4518 4519
    BOOL do_locking = (BOOL)(db->i->lt && !(lock_flags&DB_PRELOCKED_WRITE));
    if (r == 0 && do_locking) {
4520
        //Do locking if necessary.
4521
        r = get_point_lock(db, txn, key);
4522
    }
4523
    if (r == 0) {
4524
        //Insert into the brt.
4525 4526 4527 4528 4529
        TOKUTXN ttxn = txn ? db_txn_struct_i(txn)->tokutxn : NULL;
        enum brt_msg_type type = BRT_INSERT;
        if (flags==DB_NOOVERWRITE_NO_ERROR)
            type = BRT_INSERT_NO_OVERWRITE;
        r = toku_brt_maybe_insert(db->i->brt, key, val, ttxn, FALSE, ZERO_LSN, TRUE, type);
4530
    }
4531 4532

    if (r)
Zardosht Kasheff's avatar
Zardosht Kasheff committed
4533
        num_inserts_fail++;
4534 4535 4536 4537 4538 4539 4540 4541 4542 4543 4544 4545 4546 4547 4548 4549 4550 4551 4552 4553 4554 4555 4556 4557 4558 4559 4560 4561

    return r;
}

static int
log_put_single(DB_TXN *txn, BRT brt, const DBT *key, const DBT *val) {
    TOKUTXN ttxn = db_txn_struct_i(txn)->tokutxn;
    int r = toku_brt_log_put(ttxn, brt, key, val);
    return r;
}

static int
log_put_multiple(DB_TXN *txn, DB *src_db, const DBT *src_key, const DBT *src_val, uint32_t num_dbs, BRT brts[]) {
    int r = 0;
    if (num_dbs > 0) {
        TOKUTXN ttxn = db_txn_struct_i(txn)->tokutxn;
        BRT src_brt  = src_db ? src_db->i->brt : NULL;
        r = toku_brt_log_put_multiple(ttxn, src_brt, brts, num_dbs, src_key, src_val);
    }
    return r;
}

static int
do_put_multiple(DB_TXN *txn, uint32_t num_dbs, DB *db_array[], DBT keys[], DBT vals[]) {
    int r = 0;
    TOKUTXN ttxn = db_txn_struct_i(txn)->tokutxn;
    for (uint32_t which_db = 0; r == 0 && which_db < num_dbs; which_db++) {
        DB *db = db_array[which_db];
4562 4563 4564 4565 4566 4567 4568 4569 4570 4571 4572 4573 4574 4575 4576
	int do_put = TRUE;
	DB_INDEXER *indexer = toku_db_get_indexer(db);
	if (indexer) {
            DB *src_db = toku_indexer_get_src_db(indexer);
            invariant(src_db != NULL);
            uint32_t which_src_db = lookup_src_db(num_dbs, db_array, src_db);
            if (which_src_db >= num_dbs)
                r = EINVAL;
            else
                do_put = !toku_indexer_is_key_right_of_le_cursor(indexer, src_db, &keys[which_src_db]);
        }
        if (r == 0 && do_put) {
            num_inserts++;
            r = toku_brt_maybe_insert(db->i->brt, &keys[which_db], &vals[which_db], ttxn, FALSE, ZERO_LSN, FALSE, BRT_INSERT);
        }
Zardosht Kasheff's avatar
Zardosht Kasheff committed
4577
    }
4578
    return r;
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
4579
}
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
4580

4581
static int
4582 4583 4584 4585 4586 4587 4588 4589 4590 4591 4592 4593 4594 4595
env_put_multiple(
    DB_ENV *env, 
    DB *src_db, 
    DB_TXN *txn, 
    const DBT *key, 
    const DBT *val, 
    uint32_t num_dbs, 
    DB **db_array, 
    DBT *keys, 
    DBT *vals, 
    uint32_t *flags_array, 
    void *extra
    ) 
{
4596
    int r;
4597 4598
    DBT put_keys[num_dbs];
    DBT put_vals[num_dbs];
4599
    BOOL multi_accounting = TRUE;  // use num_multi_insert accountability counters 
4600 4601

    // special case for a single DB
4602
    if (num_dbs == 1 && src_db == db_array[0]) {
4603
	multi_accounting = FALSE;
4604 4605 4606 4607 4608 4609 4610
        r = toku_db_put(src_db, txn, (DBT *) key, (DBT *) val, flags_array[0]);
        goto cleanup;
    }

    HANDLE_PANICKED_ENV(env);

    {
4611 4612 4613
    uint32_t lock_flags[num_dbs];
    uint32_t remaining_flags[num_dbs];
    BRT brts[num_dbs];
4614

4615 4616 4617 4618
    if (!txn || !num_dbs) {
        r = EINVAL;
        goto cleanup;
    }
4619
    if (!env->i->generate_row_for_put) {
4620 4621 4622 4623
        r = EINVAL;
        goto cleanup;
    }

4624 4625 4626
    HANDLE_ILLEGAL_WORKING_PARENT_TXN(env, txn);

    for (uint32_t which_db = 0; which_db < num_dbs; which_db++) {
4627
        DB *db = db_array[which_db];
4628

4629
        //Generate the row
4630 4631 4632 4633 4634 4635 4636 4637 4638 4639
        if (db == src_db) {
            put_keys[which_db] = *key;
            put_vals[which_db] = *val;
        }
        else {
            r = env->i->generate_row_for_put(db, src_db, &keys[which_db], &vals[which_db], key, val, extra);
            if (r != 0) goto cleanup;
            put_keys[which_db] = keys[which_db];
            put_vals[which_db] = vals[which_db];            
        }
4640

4641
        lock_flags[which_db] = get_prelocked_flags(flags_array[which_db]);
4642
        remaining_flags[which_db] = flags_array[which_db] & ~lock_flags[which_db];
4643 4644

        // check size constraints
4645
        r = db_put_check_size_constraints(db, &put_keys[which_db], &put_vals[which_db]);
4646 4647
        if (r != 0) goto cleanup;

4648 4649
        //Check overwrite constraints
        r = db_put_check_overwrite_constraint(db, txn,
4650
                                              &put_keys[which_db],
4651
                                              lock_flags[which_db], remaining_flags[which_db]);
4652
        if (r != 0) goto cleanup;
4653 4654 4655 4656 4657
        if (remaining_flags[which_db] == DB_NOOVERWRITE_NO_ERROR) {
            //put_multiple does not support delaying the no error, since we would
            //have to log the flag in the put_multiple.
            r = EINVAL; goto cleanup;
        }
4658

4659
        //Do locking if necessary.
Zardosht Kasheff's avatar
Zardosht Kasheff committed
4660
        r = toku_grab_read_lock_on_directory(db, txn);
4661 4662
        if (r != 0) goto cleanup;

4663 4664
        if (db->i->lt && !(lock_flags[which_db] & DB_PRELOCKED_WRITE)) {
            //Needs locking
4665
            r = get_point_lock(db, txn, &put_keys[which_db]);
4666
            if (r != 0) goto cleanup;
4667 4668 4669
        }
        brts[which_db] = db->i->brt;
    }
4670 4671

    if (num_dbs == 1)
4672
        r = log_put_single(txn, brts[0], &put_keys[0], &put_vals[0]);
4673 4674 4675 4676
    else
        r = log_put_multiple(txn, src_db, key, val, num_dbs, brts);
    
    if (r == 0)
4677
        r = do_put_multiple(txn, num_dbs, db_array, put_keys, put_vals);
4678

4679 4680 4681
    }

cleanup:
4682
    if (multi_accounting) {
4683
	if (r == 0)
4684 4685 4686 4687
	    num_multi_inserts += num_dbs;
	else
	    num_multi_inserts_fail += num_dbs;
    }
4688 4689 4690
    return r;
}

4691 4692 4693 4694 4695 4696 4697 4698
static int
dbt_cmp(const DBT *a, const DBT *b) {
    if (a->size < b->size)
        return -1;
    if (a->size > b->size)
        return +1;
    return memcmp(a->data, b->data, a->size);
}
4699

4700
static int
4701
update_single(
4702
    DB_ENV *env,
4703 4704 4705 4706 4707 4708
    DB *db, 
    uint32_t flags, 
    DB_TXN *txn, 
    DBT *old_key, 
    DBT *old_data, 
    DBT *new_key, 
4709
    DBT *new_data) 
4710
{
4711
    int r = 0;
4712 4713 4714 4715
    uint32_t lock_flags;
    uint32_t remaining_flags;
    lock_flags = get_prelocked_flags(flags);
    remaining_flags = flags & ~lock_flags;
4716

4717 4718 4719
    r = toku_grab_read_lock_on_directory(db, txn);
    if (r != 0) goto cleanup;

4720 4721 4722 4723 4724
    int (*cmpfun)(DB *db, const DBT *a, const DBT *b) = toku_builtin_compare_fun;
    if (env->i->bt_compare)
        cmpfun = env->i->bt_compare;

    BOOL key_eq = cmpfun(db, old_key, new_key) == 0;
4725 4726 4727 4728 4729 4730 4731 4732
    if (!key_eq) {
        //Check overwrite constraints only in the case where 
        // the keys are not equal.
        // If the keys are equal, then we do not care of the flag is DB_NOOVERWRITE or DB_YESOVERWRITE
        r = db_put_check_overwrite_constraint(db, txn,
                                              new_key,
                                              lock_flags, remaining_flags);
        if (r != 0) goto cleanup;
4733

4734 4735
        r = toku_db_del(db, txn, (DBT *) old_key, DB_DELETE_ANY);
    }
4736
    
4737 4738 4739 4740
    if (r == 0 && (!key_eq || !(dbt_cmp(old_data, new_data) == 0))) {
        r = toku_db_put(db, txn, (DBT *) new_key, (DBT *) new_data, DB_YESOVERWRITE);
    }
cleanup:
4741 4742 4743 4744
    if (r == 0)
	num_updates++;
    else
	num_updates_fail++;
4745 4746 4747 4748 4749
    return r;
}

static int
env_update_multiple(DB_ENV *env, DB *src_db, DB_TXN *txn,                                
4750 4751 4752
                    DBT *old_src_key, DBT *old_src_data,
                    DBT *new_src_key, DBT *new_src_data,
                    uint32_t num_dbs, DB **db_array, uint32_t* flags_array, 
4753 4754
                    uint32_t num_keys, DBT keys[], 
                    uint32_t num_vals, DBT vals[],
4755 4756
                    void *extra) {
    int r = 0;
4757
    BOOL multi_accounting = TRUE;  // use num_multi_update accountability counters 
4758 4759

    // special case for a single DB 
Zardosht Kasheff's avatar
Zardosht Kasheff committed
4760
    if (num_dbs == 1 && src_db == db_array[0]) {
4761
	multi_accounting = FALSE;	
4762
        r = update_single(env,
4763 4764 4765 4766 4767 4768 4769 4770
            db_array[0], 
            flags_array[0], 
            txn, 
            old_src_key, 
            old_src_data, 
            new_src_key, 
            new_src_data
            );
4771 4772 4773 4774 4775 4776 4777 4778 4779 4780 4781 4782 4783 4784 4785 4786 4787 4788 4789 4790 4791 4792 4793 4794 4795 4796 4797 4798
        goto cleanup;
    }

    HANDLE_PANICKED_ENV(env);

    if (!txn) {
        r = EINVAL;
        goto cleanup;
    }
    if (!env->i->generate_row_for_put) {
        r = EINVAL;
        goto cleanup;
    }

    HANDLE_ILLEGAL_WORKING_PARENT_TXN(env, txn);

    {
        uint32_t n_del_dbs = 0;
        DB *del_dbs[num_dbs];
        BRT del_brts[num_dbs];
        DBT del_keys[num_dbs];
        
        uint32_t n_put_dbs = 0;
        DB *put_dbs[num_dbs];
        BRT put_brts[num_dbs];
        DBT put_keys[num_dbs];
        DBT put_vals[num_dbs];

4799 4800 4801
        uint32_t lock_flags[num_dbs];
        uint32_t remaining_flags[num_dbs];

4802 4803 4804 4805
        int (*cmpfun)(DB *db, const DBT *a, const DBT *b) = toku_builtin_compare_fun;
        if (env->i->bt_compare)
            cmpfun = env->i->bt_compare;

4806 4807
        for (uint32_t which_db = 0; which_db < num_dbs; which_db++) {
            DB *db = db_array[which_db];
4808 4809 4810 4811 4812
            DBT curr_old_key, curr_new_key, curr_new_val;
            

            lock_flags[which_db] = get_prelocked_flags(flags_array[which_db]);
            remaining_flags[which_db] = flags_array[which_db] & ~lock_flags[which_db];
4813

4814 4815 4816
            // keys[0..num_dbs-1] are the new keys
            // keys[num_dbs..2*num_dbs-1] are the old keys
            // vals[0..num_dbs-1] are the new vals
4817

4818 4819
            // Generate the old key and val
            if (which_db + num_dbs >= num_keys) {
4820 4821
                r = ENOMEM; goto cleanup;
            }
4822 4823 4824 4825 4826 4827 4828 4829
            if (db == src_db) {
                curr_old_key = *old_src_key;
            }
            else {
                r = env->i->generate_row_for_put(db, src_db, &keys[which_db + num_dbs], NULL, old_src_key, old_src_data, extra);
                if (r != 0) goto cleanup;
                curr_old_key = keys[which_db + num_dbs];
            }
4830
            // Generate the new key and val
4831 4832 4833
            if (which_db >= num_keys || which_db >= num_vals) {
                r = ENOMEM; goto cleanup;
            }
4834 4835 4836 4837 4838 4839 4840 4841 4842 4843 4844
            if (db == src_db) {
                curr_new_key = *new_src_key;
                curr_new_val = *new_src_data;
            }
            else {
                r = env->i->generate_row_for_put(db, src_db, &keys[which_db], &vals[which_db], new_src_key, new_src_data, extra);
                if (r != 0) goto cleanup;
                curr_new_key = keys[which_db];
                curr_new_val = vals[which_db];
            }
            BOOL key_eq = cmpfun(db, &curr_old_key, &curr_new_key) == 0;
4845 4846 4847 4848
            if (!key_eq) {
                r = toku_grab_read_lock_on_directory(db, txn);
                if (r != 0) goto cleanup;

4849 4850 4851 4852 4853 4854 4855 4856 4857 4858 4859 4860 4861
                //Check overwrite constraints only in the case where 
                // the keys are not equal.
                // If the keys are equal, then we do not care of the flag is DB_NOOVERWRITE or DB_YESOVERWRITE
                r = db_put_check_overwrite_constraint(db, txn,
                                                      &curr_new_key,
                                                      lock_flags[which_db], remaining_flags[which_db]);
                if (r != 0) goto cleanup;
                if (remaining_flags[which_db] == DB_NOOVERWRITE_NO_ERROR) {
                    //update_multiple does not support delaying the no error, since we would
                    //have to log the flag in the put_multiple.
                    r = EINVAL; goto cleanup;
                }

4862
                // lock old key
4863 4864
                if (db->i->lt && !(lock_flags[which_db] & DB_PRELOCKED_WRITE)) {
                    r = get_point_lock(db, txn, &curr_old_key);
4865 4866 4867 4868
                    if (r != 0) goto cleanup;
                }
                del_dbs[n_del_dbs] = db;
                del_brts[n_del_dbs] = db->i->brt;
4869
                del_keys[n_del_dbs] = curr_old_key;
4870
                n_del_dbs++;
4871
                
4872 4873
            }

4874 4875 4876
            // we take a shortcut and avoid generating the old val
            // we assume that any new vals with size > 0 are different than the old val
            // if (!key_eq || !(dbt_cmp(&vals[which_db], &vals[which_db + num_dbs]) == 0)) {
4877 4878
            if (!key_eq || curr_new_val.size > 0) {
                r = db_put_check_size_constraints(db, &curr_new_key, &curr_new_val);
4879 4880 4881 4882 4883 4884 4885
                if (r != 0) goto cleanup;

                r = toku_grab_read_lock_on_directory(db, txn);
                if (r != 0) goto cleanup;

                // lock new key
                if (db->i->lt) {
4886
                    r = get_point_lock(db, txn, &curr_new_key);
4887 4888 4889 4890
                    if (r != 0) goto cleanup;
                }
                put_dbs[n_put_dbs] = db;
                put_brts[n_put_dbs] = db->i->brt;
4891 4892
                put_keys[n_put_dbs] = curr_new_key;
                put_vals[n_put_dbs] = curr_new_val;
4893 4894 4895 4896 4897 4898 4899 4900
                n_put_dbs++;
            }
        }

        if (r == 0 && n_del_dbs > 0) {
            if (n_del_dbs == 1)
                r = log_del_single(txn, del_brts[0], &del_keys[0]);
            else
4901
                r = log_del_multiple(txn, src_db, old_src_key, old_src_data, n_del_dbs, del_brts, del_keys);
4902 4903 4904 4905 4906 4907 4908 4909 4910 4911 4912 4913 4914 4915 4916
            if (r == 0) 
                r = do_del_multiple(txn, n_del_dbs, del_dbs, del_keys);
        }

        if (r == 0 && n_put_dbs > 0) {
            if (n_put_dbs == 1)
                r = log_put_single(txn, put_brts[0], &put_keys[0], &put_vals[0]);
            else
                r = log_put_multiple(txn, src_db, new_src_key, new_src_data, n_put_dbs, put_brts);
            if (r == 0)
                r = do_put_multiple(txn, n_put_dbs, put_dbs, put_keys, put_vals);
        }
    }

cleanup:
4917
    if (multi_accounting) {
4918
	if (r == 0)
4919 4920 4921 4922
	    num_multi_updates += num_dbs;
	else
	    num_multi_updates_fail += num_dbs;
    }
4923 4924 4925 4926
    return r;
}

static int toku_db_remove(DB * db, const char *fname, const char *dbname, u_int32_t flags);
4927 4928

//We do not (yet?) support deleting subdbs by deleting the enclosing 'fname'
4929
static int
4930 4931 4932 4933
env_dbremove_subdb(DB_ENV * env, DB_TXN * txn, const char *fname, const char *dbname, int32_t flags) {
    int r;
    if (!fname || !dbname) r = EINVAL;
    else {
4934 4935 4936 4937
        char subdb_full_name[strlen(fname) + sizeof("/") + strlen(dbname)];
        int bytes = snprintf(subdb_full_name, sizeof(subdb_full_name), "%s/%s", fname, dbname);
        assert(bytes==(int)sizeof(subdb_full_name)-1);
        const char *null_subdbname = NULL;
4938
        r = toku_env_dbremove(env, txn, subdb_full_name, null_subdbname, flags);
4939
    }
4940 4941
    return r;
}
4942

4943 4944 4945 4946

//Called during committing an fdelete ONLY IF you still have an fd AND it is not connected to /dev/null
//Called during aborting an fcreate (harmless to do, and definitely correct)
static void
4947
finalize_file_removal(DICTIONARY_ID dict_id, void * extra) {
4948 4949 4950
    toku_ltm *ltm = (toku_ltm*) extra;
    if (ltm) {
        //Poison the lock tree to prevent a future file from re-using it.
4951
        toku_ltm_invalidate_lt(ltm, dict_id);
4952 4953 4954
    }
}

4955
//static int toku_db_pre_acquire_table_lock(DB *db, DB_TXN *txn);
4956

4957 4958 4959 4960 4961 4962 4963 4964 4965 4966 4967 4968 4969 4970 4971 4972 4973 4974 4975 4976 4977 4978 4979 4980 4981 4982
static int
toku_env_dbremove(DB_ENV * env, DB_TXN *txn, const char *fname, const char *dbname, u_int32_t flags) {
    int r;
    HANDLE_PANICKED_ENV(env);
    HANDLE_ILLEGAL_WORKING_PARENT_TXN(env, txn);
    if (!env_opened(env)) return EINVAL;
    if (dbname!=NULL) 
        return env_dbremove_subdb(env, txn, fname, dbname, flags);
    // env_dbremove_subdb() converts (fname, dbname) to dname

    const char * dname = fname;
    assert(dbname == NULL);

    if (flags!=0) return EINVAL;
    if (env_is_db_with_dname_open(env, dname))
        return toku_ydb_do_error(env, EINVAL, "Cannot remove dictionary with an open handle.\n");
    
    DBT dname_dbt;  
    DBT iname_dbt;  
    toku_fill_dbt(&dname_dbt, dname, strlen(dname)+1);
    init_dbt_realloc(&iname_dbt);  // sets iname_dbt.data = NULL

    int using_txns = env->i->open_flags & DB_INIT_TXN;
    DB_TXN *child = NULL;
    // begin child (unless transactionless)
    if (using_txns) {
4983
	r = toku_txn_begin(env, txn, &child, DB_TXN_NOSYNC, 1);
4984 4985 4986 4987
	assert(r==0);
    }

    // get iname
4988
    r = toku_db_get(env->i->directory, child, &dname_dbt, &iname_dbt, DB_SERIALIZABLE);  // allocates memory for iname
4989 4990 4991 4992 4993 4994 4995 4996
    char *iname = iname_dbt.data;
    if (r==DB_NOTFOUND)
        r = ENOENT;
    else if (r==0) {
	// remove (dname,iname) from directory
	r = toku_db_del(env->i->directory, child, &dname_dbt, DB_DELETE_ANY);
	if (r == 0) {
            if (using_txns) {
Yoni Fogel's avatar
Yoni Fogel committed
4997
                r = toku_brt_remove_on_commit(db_txn_struct_i(child)->tokutxn, &iname_dbt);
4998
		assert(r==0);
4999 5000 5001
                //Now that we have a writelock on dname, verify that there are still no handles open. (to prevent race conditions)
                if (r==0 && env_is_db_with_dname_open(env, dname))
                    r = toku_ydb_do_error(env, EINVAL, "Cannot remove dictionary with an open handle.\n");
5002 5003 5004
                if (r==0) {
                    DB* zombie = env_get_zombie_db_with_dname(env, dname);
                    if (zombie)
5005
                        r = toku_db_pre_acquire_table_lock(zombie, child, TRUE);
5006
                    if (r!=0 && r!=DB_LOCK_NOTGRANTED)
5007 5008
                        toku_ydb_do_error(env, r, "Cannot remove dictionary.\n");
                }
5009 5010
            }
            else {
Yoni Fogel's avatar
Yoni Fogel committed
5011
                r = toku_brt_remove_now(env->i->cachetable, &iname_dbt);
5012 5013 5014 5015 5016 5017 5018 5019
		assert(r==0);
            }
	}
    }

    if (using_txns) {
	// close txn
	if (r == 0) {  // commit
5020
	    r = toku_txn_commit(child, DB_TXN_NOSYNC, NULL, NULL);
5021 5022 5023
	    assert(r==0);  // TODO panic
	}
	else {         // abort
5024
	    int r2 = toku_txn_abort(child, NULL, NULL);
5025 5026
	    assert(r2==0);  // TODO panic
	}
5027
    }
5028 5029

    if (iname) toku_free(iname);
5030
    return r;
5031

5032 5033
}

5034 5035 5036

static int
toku_db_remove(DB * db, const char *fname, const char *dbname, u_int32_t flags) {
5037
    HANDLE_PANICKED_DB(db);
5038 5039 5040 5041 5042 5043 5044 5045 5046 5047 5048 5049 5050 5051 5052 5053 5054 5055 5056 5057 5058
    DB_TXN *null_txn = NULL;
    int r  = toku_env_dbremove(db->dbenv, null_txn, fname, dbname, flags);
    int r2 = toku_db_close(db, 0);
    if (r==0) r = r2;
    return r;
}

static int
env_dbrename_subdb(DB_ENV *env, DB_TXN *txn, const char *fname, const char *dbname, const char *newname, u_int32_t flags) {
    int r;
    if (!fname || !dbname || !newname) r = EINVAL;
    else {
        char subdb_full_name[strlen(fname) + sizeof("/") + strlen(dbname)];
        {
            int bytes = snprintf(subdb_full_name, sizeof(subdb_full_name), "%s/%s", fname, dbname);
            assert(bytes==(int)sizeof(subdb_full_name)-1);
        }
        char new_full_name[strlen(fname) + sizeof("/") + strlen(dbname)];
        {
            int bytes = snprintf(new_full_name, sizeof(new_full_name), "%s/%s", fname, dbname);
            assert(bytes==(int)sizeof(new_full_name)-1);
Yoni Fogel's avatar
Yoni Fogel committed
5059
        }
5060 5061
        const char *null_subdbname = NULL;
        r = toku_env_dbrename(env, txn, subdb_full_name, null_subdbname, new_full_name, flags);
Yoni Fogel's avatar
Yoni Fogel committed
5062
    }
5063 5064 5065 5066 5067 5068 5069 5070 5071 5072 5073 5074 5075 5076 5077 5078 5079 5080 5081 5082 5083 5084
    return r;
}


static int
toku_env_dbrename(DB_ENV *env, DB_TXN *txn, const char *fname, const char *dbname, const char *newname, u_int32_t flags) {
    int r;
    HANDLE_PANICKED_ENV(env);
    HANDLE_ILLEGAL_WORKING_PARENT_TXN(env, txn);
    if (!env_opened(env)) return EINVAL;
    if (dbname!=NULL) 
        return env_dbrename_subdb(env, txn, fname, dbname, newname, flags);
    // env_dbrename_subdb() converts (fname, dbname) to dname and (fname, newname) to newdname

    const char * dname = fname;
    assert(dbname == NULL);

    if (flags!=0) return EINVAL;
    if (env_is_db_with_dname_open(env, dname))
        return toku_ydb_do_error(env, EINVAL, "Cannot rename dictionary with an open handle.\n");
    if (env_is_db_with_dname_open(env, newname))
        return toku_ydb_do_error(env, EINVAL, "Cannot rename dictionary; Dictionary with target name has an open handle.\n");
Yoni Fogel's avatar
Yoni Fogel committed
5085
    
5086 5087 5088 5089 5090 5091 5092 5093 5094 5095 5096
    DBT old_dname_dbt;  
    DBT new_dname_dbt;  
    DBT iname_dbt;  
    toku_fill_dbt(&old_dname_dbt, dname, strlen(dname)+1);
    toku_fill_dbt(&new_dname_dbt, newname, strlen(newname)+1);
    init_dbt_realloc(&iname_dbt);  // sets iname_dbt.data = NULL

    int using_txns = env->i->open_flags & DB_INIT_TXN;
    DB_TXN *child = NULL;
    // begin child (unless transactionless)
    if (using_txns) {
5097
	r = toku_txn_begin(env, txn, &child, DB_TXN_NOSYNC, 1);
5098 5099
	assert(r==0);
    }
5100

5101
    r = toku_db_get(env->i->directory, child, &old_dname_dbt, &iname_dbt, DB_SERIALIZABLE);  // allocates memory for iname
5102 5103 5104 5105 5106
    char *iname = iname_dbt.data;
    if (r==DB_NOTFOUND)
        r = ENOENT;
    else if (r==0) {
	// verify that newname does not already exist
5107
	r = db_getf_set(env->i->directory, child, DB_SERIALIZABLE, &new_dname_dbt, ydb_getf_do_nothing, NULL);
5108 5109 5110 5111 5112 5113 5114
	if (r == 0) 
	    r = EEXIST;
	else if (r == DB_NOTFOUND) {
	    // remove old (dname,iname) and insert (newname,iname) in directory
	    r = toku_db_del(env->i->directory, child, &old_dname_dbt, DB_DELETE_ANY);
	    if (r == 0)
		r = toku_db_put(env->i->directory, child, &new_dname_dbt, &iname_dbt, DB_YESOVERWRITE);
5115 5116 5117
            //Now that we have writelocks on both dnames, verify that there are still no handles open. (to prevent race conditions)
            if (r==0 && env_is_db_with_dname_open(env, dname))
                r = toku_ydb_do_error(env, EINVAL, "Cannot rename dictionary with an open handle.\n");
5118
            DB* zombie = NULL;
5119
            if (r==0) {
5120
                zombie = env_get_zombie_db_with_dname(env, dname);
5121
                if (zombie)
5122
                    r = toku_db_pre_acquire_table_lock(zombie, child, TRUE);
5123
                if (r!=0 && r!=DB_LOCK_NOTGRANTED)
5124 5125
                    toku_ydb_do_error(env, r, "Cannot rename dictionary.\n");
            }
5126 5127
            if (r==0 && env_is_db_with_dname_open(env, newname))
                r = toku_ydb_do_error(env, EINVAL, "Cannot rename dictionary; Dictionary with target name has an open handle.\n");
5128 5129 5130 5131 5132 5133
            if (r==0 && zombie) {
                //Update zombie in list if exists.
                env_note_zombie_db_closed(env, zombie);  // tell env that this db is no longer a zombie (it is completely closed)
                toku_free(zombie->i->dname);
                zombie->i->dname = toku_xstrdup(newname);
                env_note_zombie_db(env, zombie);  // tell env that this db is a zombie
5134
            }
5135 5136 5137 5138 5139 5140
	}
    }

    if (using_txns) {
	// close txn
	if (r == 0) {  // commit
5141
	    r = toku_txn_commit(child, DB_TXN_NOSYNC, NULL, NULL);
5142 5143 5144
	    assert(r==0);  // TODO panic
	}
	else {         // abort
5145
	    int r2 = toku_txn_abort(child, NULL, NULL);
5146 5147 5148 5149 5150 5151
	    assert(r2==0);  // TODO panic
	}
    }

    if (iname) toku_free(iname);
    return r;
Yoni Fogel's avatar
Yoni Fogel committed
5152

Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
5153
}
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
5154

5155 5156
static int
toku_db_rename(DB * db, const char *fname, const char *dbname, const char *newname, u_int32_t flags) {
5157
    HANDLE_PANICKED_DB(db);
5158 5159 5160 5161 5162
    DB_TXN *null_txn = NULL;
    int r  = toku_env_dbrename(db->dbenv, null_txn, fname, dbname, newname, flags);
    int r2 = toku_db_close(db, 0);
    if (r==0) r = r2;
    return r;
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
5163
}
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
5164

5165 5166
static int 
toku_db_set_descriptor(DB *db, u_int32_t version, const DBT* descriptor) {
5167 5168 5169 5170 5171
    HANDLE_PANICKED_DB(db);
    int r;
    if (db_opened(db)) return EINVAL;
    else if (!descriptor) r = EINVAL;
    else if (descriptor->size>0 && !descriptor->data) r = EINVAL;
5172
    else r = toku_brt_set_descriptor(db->i->brt, version, descriptor);
5173 5174 5175
    return r;
}

5176 5177
static int 
toku_db_set_flags(DB *db, u_int32_t flags) {
5178
    HANDLE_PANICKED_DB(db);
5179

Rich Prohaska's avatar
Rich Prohaska committed
5180
    /* the following matches BDB */
5181 5182
    if (db_opened(db) && flags != 0) return EINVAL;

5183
    return 0;
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
5184 5185
}

5186 5187
static int 
toku_db_get_flags(DB *db, u_int32_t *pflags) {
5188
    HANDLE_PANICKED_DB(db);
5189
    if (!pflags) return EINVAL;
5190
    *pflags = 0;
5191 5192 5193
    return 0;
}

5194 5195
static int 
toku_db_set_pagesize(DB *db, u_int32_t pagesize) {
5196
    HANDLE_PANICKED_DB(db);
5197
    int r = toku_brt_set_nodesize(db->i->brt, pagesize);
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
5198
    return r;
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
5199
}
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
5200

5201 5202
static int 
toku_db_stat64(DB * db, DB_TXN *txn, DB_BTREE_STAT64 *s) {
5203
    HANDLE_PANICKED_DB(db);
5204
    HANDLE_DB_ILLEGAL_WORKING_PARENT_TXN(db, txn);
5205 5206 5207 5208 5209 5210 5211 5212 5213
    struct brtstat64_s brtstat;
    int r = toku_brt_stat64(db->i->brt, db_txn_struct_i(txn)->tokutxn, &brtstat);
    if (r==0) {
	s->bt_nkeys = brtstat.nkeys;
	s->bt_ndata = brtstat.ndata;
	s->bt_dsize = brtstat.dsize;
	s->bt_fsize = brtstat.fsize;
    }
    return r;
5214
}
5215 5216 5217

static int 
locked_db_stat64 (DB *db, DB_TXN *txn, DB_BTREE_STAT64 *s) {
5218 5219 5220 5221
    toku_ydb_lock();
    int r = toku_db_stat64(db, txn, s);
    toku_ydb_unlock();
    return r;
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
5222 5223
}

5224 5225
static int 
toku_db_key_range64(DB* db, DB_TXN* txn __attribute__((__unused__)), DBT* key, u_int64_t* less, u_int64_t* equal, u_int64_t* greater, int* is_exact) {
Zardosht Kasheff's avatar
Zardosht Kasheff committed
5226
    HANDLE_PANICKED_DB(db);
5227
    HANDLE_DB_ILLEGAL_WORKING_PARENT_TXN(db, txn);
Zardosht Kasheff's avatar
Zardosht Kasheff committed
5228 5229 5230 5231 5232 5233 5234 5235 5236 5237 5238 5239 5240

    // note that toku_brt_keyrange does not have a txn param
    // this will be fixed later
    // temporarily, because the caller, locked_db_keyrange, 
    // has the ydb lock, we are ok
    int r = toku_brt_keyrange(db->i->brt, key, less, equal, greater);
    if (r != 0) { goto cleanup; }
    // temporarily set is_exact to 0 because brt_keyrange does not have this parameter
    *is_exact = 0;
cleanup:
    return r;
}

5241 5242
static int 
toku_c_pre_acquire_read_lock(DBC *dbc, const DBT *key_left, const DBT *key_right) {
5243 5244
    DB* db = dbc->dbp;
    DB_TXN* txn = dbc_struct_i(dbc)->txn;
Yoni Fogel's avatar
Yoni Fogel committed
5245 5246
    HANDLE_PANICKED_DB(db);
    if (!db->i->lt || !txn) return EINVAL;
5247
    //READ_UNCOMMITTED and READ_COMMITTED transactions do not need read locks.
5248 5249 5250
    if (dbc_struct_i(dbc)->iso != TOKU_ISO_SERIALIZABLE) {
         return 0;
    }
Yoni Fogel's avatar
Yoni Fogel committed
5251 5252

    int r;
5253 5254 5255
    {
	RANGE_LOCK_REQUEST_S request;
	read_lock_request_init(&request, txn, db,
5256 5257
			       key_left,
			       key_right);
5258 5259
        r = grab_range_lock(&request);
    }
Yoni Fogel's avatar
Yoni Fogel committed
5260 5261 5262
    return r;
}

Zardosht Kasheff's avatar
Zardosht Kasheff committed
5263 5264 5265 5266 5267 5268 5269 5270 5271 5272 5273 5274 5275 5276 5277 5278 5279 5280 5281 5282 5283 5284 5285
static int toku_db_pre_acquire_fileops_lock(DB *db, DB_TXN *txn) {
    RANGE_LOCK_REQUEST_S request;
    char *   dname = db->i->dname;
    DBT key_in_directory;
    //
    // bad hack because some environment dictionaries do not have a dname
    //
    if (!dname) {
        return 0;
    }
    toku_fill_dbt(&key_in_directory, dname, strlen(dname)+1);
    //Left end of range == right end of range (point lock)
    write_lock_request_init(
        &request, 
        txn, 
        db->dbenv->i->directory,
        &key_in_directory,
        &key_in_directory
        );
    int r = grab_range_lock(&request);
    return r;
}

5286 5287
//static int toku_db_pre_acquire_table_lock(DB *db, DB_TXN *txn) {
// needed by loader.c
5288 5289
int 
toku_db_pre_acquire_table_lock(DB *db, DB_TXN *txn, BOOL just_lock) {
Yoni Fogel's avatar
Yoni Fogel committed
5290 5291 5292 5293 5294
    HANDLE_PANICKED_DB(db);
    if (!db->i->lt || !txn) return EINVAL;

    int r;

5295
    {
Zardosht Kasheff's avatar
Zardosht Kasheff committed
5296 5297 5298 5299 5300 5301 5302 5303
        RANGE_LOCK_REQUEST_S request;
        write_lock_request_init(
            &request, 
            txn, 
            db,
            toku_lt_neg_infinity,
            toku_lt_infinity
            );
5304 5305 5306
        r = grab_range_lock(&request);
    }

5307 5308
    if (r==0 && !just_lock &&
        !toku_brt_is_recovery_logging_suppressed(db->i->brt) &&
5309
        toku_brt_is_empty_fast(db->i->brt)
5310 5311 5312 5313 5314 5315
    ) {
        //Try to suppress both rollback and recovery logs
        DB_LOADER *loader;
        DB *dbs[1] = {db};
        uint32_t db_flags[1]  = {DB_NOOVERWRITE};
        uint32_t dbt_flags[1] = {0};
5316
        uint32_t loader_flags = DB_PRELOCKED_WRITE; //Don't recursively prelock
5317
        DB_ENV *env = db->dbenv;
5318 5319 5320 5321 5322 5323 5324
	DB_TXN *child = NULL;
	
	{
	    // begin child
	    int rt = toku_txn_begin(env, txn, &child, DB_TXN_NOSYNC, 1);
	    assert(rt==0);
	}
5325 5326

        toku_ydb_unlock(); //Cannot hold ydb lock when creating loader
5327 5328
	
        int r_loader = env->create_loader(env, child, &loader, NULL, 1, dbs, db_flags, dbt_flags, loader_flags);
5329
        if (r_loader==0) {
5330 5331 5332 5333
            r_loader = loader->set_error_callback(loader, NULL, NULL);
            assert(r_loader==0);
            r_loader = loader->set_poll_function(loader, NULL, NULL);
            assert(r_loader==0);
5334
            // close the loader
5335 5336 5337 5338
            r_loader = loader->close(loader);
	    if (r_loader==0) {
		toku_brt_suppress_recovery_logs(db->i->brt, db_txn_struct_i(child)->tokutxn);
	    }
5339 5340 5341 5342 5343 5344 5345
        }
        else if (r_loader != DB_LOCK_NOTGRANTED) {
            //Lock not granted is not an error.
            //It just means we cannot use the loader optimization.
            assert(r==0);
            r = r_loader;
        }
5346 5347 5348
	if (r_loader == 0) { // commit
	    r = locked_txn_commit(child, 0);
	    assert(r==0);
5349
	    logsuppress++;
5350 5351 5352 5353
	}
	else {  // abort
	    r = locked_txn_abort(child);
	    assert(r==0);
5354
	    logsuppressfail++;
5355
	}
5356
        toku_ydb_lock(); //Reaquire ydb lock.
5357 5358
    }

Yoni Fogel's avatar
Yoni Fogel committed
5359 5360
    return r;
}
Zardosht Kasheff's avatar
Zardosht Kasheff committed
5361

Yoni Fogel's avatar
Yoni Fogel committed
5362 5363 5364
//TODO: DB_AUTO_COMMIT.
//TODO: Nowait only conditionally?
//TODO: NOSYNC change to SYNC if DB_ENV has something in set_flags
5365 5366
static inline int 
toku_db_construct_autotxn(DB* db, DB_TXN **txn, BOOL* changed, BOOL force_auto_commit) {
Yoni Fogel's avatar
Yoni Fogel committed
5367 5368 5369 5370 5371 5372
    assert(db && txn && changed);
    DB_ENV* env = db->dbenv;
    if (*txn || !(env->i->open_flags & DB_INIT_TXN)) {
        *changed = FALSE;
        return 0;
    }
5373
    BOOL nosync = (BOOL)(!force_auto_commit && !(env->i->open_flags & DB_AUTO_COMMIT));
Yoni Fogel's avatar
Yoni Fogel committed
5374
    u_int32_t txn_flags = DB_TXN_NOWAIT | (nosync ? DB_TXN_NOSYNC : 0);
5375
    int r = toku_txn_begin(env, NULL, txn, txn_flags, 1);
Yoni Fogel's avatar
Yoni Fogel committed
5376 5377 5378 5379 5380
    if (r!=0) return r;
    *changed = TRUE;
    return 0;
}

5381 5382
static inline int 
toku_db_destruct_autotxn(DB_TXN *txn, int r, BOOL changed) {
Yoni Fogel's avatar
Yoni Fogel committed
5383
    if (!changed) return r;
5384 5385
    if (r==0) return toku_txn_commit(txn, 0, NULL, NULL);
    toku_txn_abort(txn, NULL, NULL);
Yoni Fogel's avatar
Yoni Fogel committed
5386 5387 5388
    return r; 
}

5389 5390
static int 
locked_db_close(DB * db, u_int32_t flags) {
5391 5392 5393 5394
    toku_ydb_lock(); 
    int r = toku_db_close(db, flags); 
    toku_ydb_unlock(); 
    return r;
Rich Prohaska's avatar
Rich Prohaska committed
5395 5396
}

5397 5398
static inline int 
autotxn_db_cursor(DB *db, DB_TXN *txn, DBC **c, u_int32_t flags) {
Yoni Fogel's avatar
Yoni Fogel committed
5399
    if (!txn && (db->dbenv->i->open_flags & DB_INIT_TXN)) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
5400
        return toku_ydb_do_error(db->dbenv, EINVAL,
Yoni Fogel's avatar
Yoni Fogel committed
5401 5402
              "Cursors in a transaction environment must have transactions.\n");
    }
5403
    return toku_db_cursor(db, txn, c, flags, 0);
Yoni Fogel's avatar
Yoni Fogel committed
5404 5405
}

5406 5407
static int 
locked_db_cursor(DB *db, DB_TXN *txn, DBC **c, u_int32_t flags) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
5408
    toku_ydb_lock(); int r = autotxn_db_cursor(db, txn, c, flags); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
5409 5410
}

5411 5412
static inline int 
autotxn_db_del(DB* db, DB_TXN* txn, DBT* key, u_int32_t flags) {
Yoni Fogel's avatar
Yoni Fogel committed
5413 5414 5415 5416 5417 5418 5419
    BOOL changed; int r;
    r = toku_db_construct_autotxn(db, &txn, &changed, FALSE);
    if (r!=0) return r;
    r = toku_db_del(db, txn, key, flags);
    return toku_db_destruct_autotxn(txn, r, changed);
}

5420 5421
static int 
locked_db_del(DB * db, DB_TXN * txn, DBT * key, u_int32_t flags) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
5422
    toku_ydb_lock(); int r = autotxn_db_del(db, txn, key, flags); toku_ydb_unlock(); return r;
Yoni Fogel's avatar
Yoni Fogel committed
5423 5424
}

5425 5426
static inline int 
autotxn_db_get(DB* db, DB_TXN* txn, DBT* key, DBT* data, u_int32_t flags) {
Yoni Fogel's avatar
Yoni Fogel committed
5427 5428 5429 5430 5431
    BOOL changed; int r;
    r = toku_db_construct_autotxn(db, &txn, &changed, FALSE);
    if (r!=0) return r;
    r = toku_db_get(db, txn, key, data, flags);
    return toku_db_destruct_autotxn(txn, r, changed);
Rich Prohaska's avatar
Rich Prohaska committed
5432 5433
}

5434 5435
static int 
locked_db_get (DB * db, DB_TXN * txn, DBT * key, DBT * data, u_int32_t flags) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
5436
    toku_ydb_lock(); int r = autotxn_db_get(db, txn, key, data, flags); toku_ydb_unlock(); return r;
Yoni Fogel's avatar
Yoni Fogel committed
5437 5438
}

5439 5440
static inline int 
autotxn_db_getf_set (DB *db, DB_TXN *txn, u_int32_t flags, DBT *key, YDB_CALLBACK_FUNCTION f, void *extra) {
5441 5442 5443 5444 5445 5446 5447
    BOOL changed; int r;
    r = toku_db_construct_autotxn(db, &txn, &changed, FALSE);
    if (r!=0) return r;
    r = db_getf_set(db, txn, flags, key, f, extra);
    return toku_db_destruct_autotxn(txn, r, changed);
}

5448 5449
static int 
locked_db_getf_set (DB *db, DB_TXN *txn, u_int32_t flags, DBT *key, YDB_CALLBACK_FUNCTION f, void *extra) {
5450 5451 5452
    toku_ydb_lock(); int r = autotxn_db_getf_set(db, txn, flags, key, f, extra); toku_ydb_unlock(); return r;
}

5453 5454
static int 
locked_c_pre_acquire_read_lock(DBC *dbc, const DBT *key_left, const DBT *key_right) {
Yoni Fogel's avatar
Yoni Fogel committed
5455
    toku_ydb_lock();
5456
    int r = toku_c_pre_acquire_read_lock(dbc, key_left, key_right);
Yoni Fogel's avatar
Yoni Fogel committed
5457 5458 5459 5460
    toku_ydb_unlock();
    return r;
}

5461 5462
static int 
locked_db_pre_acquire_table_lock(DB *db, DB_TXN *txn) {
Yoni Fogel's avatar
Yoni Fogel committed
5463
    toku_ydb_lock();
5464
    int r = toku_db_pre_acquire_table_lock(db, txn, FALSE);
Yoni Fogel's avatar
Yoni Fogel committed
5465 5466 5467 5468
    toku_ydb_unlock();
    return r;
}

Zardosht Kasheff's avatar
Zardosht Kasheff committed
5469 5470 5471 5472 5473 5474 5475
static int locked_db_pre_acquire_fileops_lock(DB *db, DB_TXN *txn) {
    toku_ydb_lock();
    int r = toku_db_pre_acquire_fileops_lock(db, txn);
    toku_ydb_unlock();
    return r;
}

5476 5477
// truncate a database
// effect: remove all of the rows from a database
5478 5479
static int 
toku_db_truncate(DB *db, DB_TXN *txn, u_int32_t *row_count, u_int32_t flags) {
5480
    HANDLE_PANICKED_DB(db);
5481
    HANDLE_DB_ILLEGAL_WORKING_PARENT_TXN(db, txn);
5482 5483
    int r;

5484 5485 5486 5487 5488 5489 5490
    u_int32_t unhandled_flags = flags;
    int ignore_cursors = 0;
    if (flags & DB_TRUNCATE_WITHCURSORS) {
        ignore_cursors = 1;
        unhandled_flags &= ~DB_TRUNCATE_WITHCURSORS;
    }

5491
    // dont support flags (yet)
5492
    if (unhandled_flags)
5493
        return EINVAL;
5494 5495
    // dont support cursors unless explicitly told to
    if (!ignore_cursors && toku_brt_get_cursor_count(db->i->brt) > 0)
5496 5497 5498 5499
        return EINVAL;

    // acquire a table lock
    if (txn) {
Zardosht Kasheff's avatar
Zardosht Kasheff committed
5500 5501 5502 5503
        r = toku_db_pre_acquire_fileops_lock(db, txn);
        if (r != 0) {
            return r;
        }
5504
        r = toku_db_pre_acquire_table_lock(db, txn, TRUE);
Zardosht Kasheff's avatar
Zardosht Kasheff committed
5505
        if (r != 0) {
5506
            return r;
Zardosht Kasheff's avatar
Zardosht Kasheff committed
5507
        }
5508 5509 5510 5511
    }

    *row_count = 0;

5512
    r = toku_brt_truncate(db->i->brt);
5513 5514 5515 5516

    return r;
}

5517 5518
static inline int 
autotxn_db_open(DB* db, DB_TXN* txn, const char *fname, const char *dbname, DBTYPE dbtype, u_int32_t flags, int mode) {
Yoni Fogel's avatar
Yoni Fogel committed
5519
    BOOL changed; int r;
5520
    r = toku_db_construct_autotxn(db, &txn, &changed, (BOOL)((flags & DB_AUTO_COMMIT) != 0));
Yoni Fogel's avatar
Yoni Fogel committed
5521 5522 5523
    if (r!=0) return r;
    r = toku_db_open(db, txn, fname, dbname, dbtype, flags & ~DB_AUTO_COMMIT, mode);
    return toku_db_destruct_autotxn(txn, r, changed);
Rich Prohaska's avatar
Rich Prohaska committed
5524 5525
}

5526 5527
static int 
locked_db_open(DB *db, DB_TXN *txn, const char *fname, const char *dbname, DBTYPE dbtype, u_int32_t flags, int mode) {
5528 5529 5530 5531
    toku_multi_operation_client_lock(); //Cannot begin checkpoint
    toku_ydb_lock(); int r = autotxn_db_open(db, txn, fname, dbname, dbtype, flags, mode); toku_ydb_unlock();
    toku_multi_operation_client_unlock(); //Can now begin checkpoint
    return r;
Rich Prohaska's avatar
Rich Prohaska committed
5532 5533
}

5534 5535
static inline int 
autotxn_db_put(DB* db, DB_TXN* txn, DBT* key, DBT* data, u_int32_t flags) {
5536
    //{ unsigned i; printf("put %p keylen=%d key={", db, key->size); for(i=0; i<key->size; i++) printf("%d,", ((char*)key->data)[i]); printf("} datalen=%d data={", data->size); for(i=0; i<data->size; i++) printf("%d,", ((char*)data->data)[i]); printf("}\n"); }
Yoni Fogel's avatar
Yoni Fogel committed
5537 5538 5539 5540 5541 5542 5543
    BOOL changed; int r;
    r = toku_db_construct_autotxn(db, &txn, &changed, FALSE);
    if (r!=0) return r;
    r = toku_db_put(db, txn, key, data, flags);
    return toku_db_destruct_autotxn(txn, r, changed);
}

5544 5545
static int 
locked_db_put(DB * db, DB_TXN * txn, DBT * key, DBT * data, u_int32_t flags) {
5546 5547 5548 5549 5550 5551 5552
    int r = env_check_avail_fs_space(db->dbenv);
    if (r == 0) {
	toku_ydb_lock(); 
	r = autotxn_db_put(db, txn, key, data, flags); 
	toku_ydb_unlock(); 
    }
    return r;
Rich Prohaska's avatar
Rich Prohaska committed
5553 5554
}

5555 5556
static int 
locked_db_remove(DB * db, const char *fname, const char *dbname, u_int32_t flags) {
5557
    toku_multi_operation_client_lock(); //Cannot begin checkpoint
5558 5559 5560
    toku_ydb_lock();
    int r = toku_db_remove(db, fname, dbname, flags);
    toku_ydb_unlock();
5561
    toku_multi_operation_client_unlock(); //Can now begin checkpoint
5562
    return r;
Rich Prohaska's avatar
Rich Prohaska committed
5563 5564
}

5565 5566
static int 
locked_db_rename(DB * db, const char *namea, const char *nameb, const char *namec, u_int32_t flags) {
5567
    toku_multi_operation_client_lock(); //Cannot begin checkpoint
5568 5569 5570
    toku_ydb_lock();
    int r = toku_db_rename(db, namea, nameb, namec, flags);
    toku_ydb_unlock();
5571
    toku_multi_operation_client_unlock(); //Can now begin checkpoint
5572
    return r;
Rich Prohaska's avatar
Rich Prohaska committed
5573 5574
}

5575 5576
static int 
locked_db_set_descriptor(DB *db, u_int32_t version, const DBT* descriptor) {
5577
    toku_ydb_lock();
5578
    int r = toku_db_set_descriptor(db, version, descriptor);
5579 5580
    toku_ydb_unlock();
    return r;
5581 5582
}

5583 5584
static void 
locked_db_set_errfile (DB *db, FILE *errfile) {
Rich Prohaska's avatar
Rich Prohaska committed
5585 5586 5587
    db->dbenv->set_errfile(db->dbenv, errfile);
}

5588 5589
static int 
locked_db_set_flags(DB *db, u_int32_t flags) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
5590
    toku_ydb_lock(); int r = toku_db_set_flags(db, flags); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
5591 5592
}

5593 5594
static int 
locked_db_get_flags(DB *db, u_int32_t *flags) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
5595
    toku_ydb_lock(); int r = toku_db_get_flags(db, flags); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
5596 5597
}

5598 5599
static int 
locked_db_set_pagesize(DB *db, u_int32_t pagesize) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
5600
    toku_ydb_lock(); int r = toku_db_set_pagesize(db, pagesize); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
5601 5602
}

5603
// TODO 2216 delete this
5604 5605
static int 
locked_db_fd(DB * UU(db), int * UU(fdp)) {
5606 5607 5608 5609 5610
    //    toku_ydb_lock(); 
    // int r = toku_db_fd(db, fdp); 
    //    toku_ydb_unlock(); 
    //    return r;
    return 0;
Rich Prohaska's avatar
Rich Prohaska committed
5611 5612
}

Zardosht Kasheff's avatar
Zardosht Kasheff committed
5613

Yoni Fogel's avatar
Yoni Fogel committed
5614 5615 5616 5617 5618
static int locked_db_key_range64(DB* db, DB_TXN* txn, DBT* dbt, u_int64_t* less, u_int64_t* equal, u_int64_t* greater, int* is_exact) {
    toku_ydb_lock(); int r = toku_db_key_range64(db, txn, dbt, less, equal, greater, is_exact); toku_ydb_unlock(); return r;
}

static const DBT* toku_db_dbt_pos_infty(void) __attribute__((pure));
5619 5620
static const DBT*
toku_db_dbt_pos_infty(void) {
Yoni Fogel's avatar
Yoni Fogel committed
5621 5622 5623 5624
    return toku_lt_infinity;
}

static const DBT* toku_db_dbt_neg_infty(void) __attribute__((pure));
5625 5626
static const DBT* 
toku_db_dbt_neg_infty(void) {
Yoni Fogel's avatar
Yoni Fogel committed
5627
    return toku_lt_neg_infinity;
Zardosht Kasheff's avatar
Zardosht Kasheff committed
5628 5629
}

5630 5631
static int 
locked_db_truncate(DB *db, DB_TXN *txn, u_int32_t *row_count, u_int32_t flags) {
5632 5633 5634 5635 5636 5637
    toku_checkpoint_safe_client_lock();
    toku_ydb_lock();
    int r = toku_db_truncate(db, txn, row_count, flags);
    toku_ydb_unlock();
    toku_checkpoint_safe_client_unlock();
    return r;
5638 5639
}

5640 5641 5642 5643 5644 5645 5646
static int
toku_db_optimize(DB *db) {
    HANDLE_PANICKED_DB(db);
    int r = toku_brt_optimize(db->i->brt);
    return r;
}

5647 5648 5649
static int
toku_db_flatten(DB *db, DB_TXN *txn) {
    HANDLE_PANICKED_DB(db);
5650 5651
    TOKUTXN ttxn = txn ? db_txn_struct_i(txn)->tokutxn : NULL;
    int r = toku_brt_flatten(db->i->brt, ttxn);
5652 5653 5654
    return r;
}

5655 5656
static inline int 
autotxn_db_flatten(DB* db, DB_TXN* txn) {
5657 5658 5659 5660 5661 5662 5663 5664
    BOOL changed; int r;
    r = toku_db_construct_autotxn(db, &txn, &changed, FALSE);
    if (r!=0) return r;
    r = toku_db_flatten(db, txn);
    return toku_db_destruct_autotxn(txn, r, changed);
}


5665 5666
static int 
locked_db_flatten(DB *db, DB_TXN *txn) {
5667 5668 5669
    toku_ydb_lock(); int r = autotxn_db_flatten(db, txn); toku_ydb_unlock(); return r;
}

5670 5671
static int 
locked_db_optimize(DB *db) {
5672 5673 5674 5675 5676 5677
    toku_ydb_lock();
    int r = toku_db_optimize(db);
    toku_ydb_unlock();
    return r;
}

5678 5679 5680 5681 5682 5683 5684 5685 5686 5687 5688 5689 5690 5691 5692 5693 5694 5695 5696
static int
db_get_fragmentation(DB * db, TOKU_DB_FRAGMENTATION report) {
    HANDLE_PANICKED_DB(db);
    int r;
    if (!db_opened(db))
        r = toku_ydb_do_error(db->dbenv, EINVAL, "Fragmentation report available only on open DBs.\n");
    else
        r = toku_brt_get_fragmentation(db->i->brt, report);
    return r;
}

static int
locked_db_get_fragmentation(DB * db, TOKU_DB_FRAGMENTATION report) {
    toku_ydb_lock();
    int r = db_get_fragmentation(db, report);
    toku_ydb_unlock();
    return r;
}

5697 5698 5699 5700 5701 5702 5703 5704 5705 5706 5707 5708 5709 5710 5711 5712 5713 5714 5715 5716 5717
int 
toku_db_set_indexer(DB *db, DB_INDEXER * indexer) {
    db->i->indexer = indexer;
    return 0;
}

static int 
locked_db_set_indexer(DB *db, DB_INDEXER *indexer) {
    toku_ydb_lock(); int r = toku_db_set_indexer(db, indexer); toku_ydb_unlock(); return r;
}

DB_INDEXER *
toku_db_get_indexer(DB *db) {
    return db->i->indexer;
}

static void 
locked_db_get_indexer(DB *db, DB_INDEXER **indexer_ptr) {
    toku_ydb_lock(); *indexer_ptr = toku_db_get_indexer(db); toku_ydb_unlock();
}

5718 5719
static int 
toku_db_create(DB ** db, DB_ENV * env, u_int32_t flags) {
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
5720 5721
    int r;

5722 5723 5724 5725 5726
    if (flags || env == NULL) 
        return EINVAL;

    if (!env_opened(env))
        return EINVAL;
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
5727
    
5728
    DB *MALLOC(result);
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
5729 5730
    if (result == 0) {
        return ENOMEM;
Bradley C. Kuszmaul's avatar
Bradley C. Kuszmaul committed
5731
    }
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
5732 5733
    memset(result, 0, sizeof *result);
    result->dbenv = env;
Yoni Fogel's avatar
Yoni Fogel committed
5734 5735 5736 5737 5738 5739 5740 5741 5742 5743 5744
#define SDB(name) result->name = locked_db_ ## name
    SDB(key_range64);
    SDB(close);
    SDB(cursor);
    SDB(del);
    SDB(get);
    //    SDB(key_range);
    SDB(open);
    SDB(put);
    SDB(remove);
    SDB(rename);
5745
    SDB(set_descriptor);
Yoni Fogel's avatar
Yoni Fogel committed
5746 5747 5748 5749
    SDB(set_errfile);
    SDB(set_pagesize);
    SDB(set_flags);
    SDB(get_flags);
5750
    SDB(stat64);
Yoni Fogel's avatar
Yoni Fogel committed
5751
    SDB(fd);
Yoni Fogel's avatar
Yoni Fogel committed
5752
    SDB(pre_acquire_table_lock);
Zardosht Kasheff's avatar
Zardosht Kasheff committed
5753
    SDB(pre_acquire_fileops_lock);
5754
    SDB(truncate);
5755
    SDB(row_size_supported);
5756
    SDB(getf_set);
5757
    SDB(flatten);
5758
    SDB(optimize);
5759
    SDB(get_fragmentation);
5760 5761
    SDB(set_indexer);
    SDB(get_indexer);
Yoni Fogel's avatar
Yoni Fogel committed
5762 5763 5764
#undef SDB
    result->dbt_pos_infty = toku_db_dbt_pos_infty;
    result->dbt_neg_infty = toku_db_dbt_neg_infty;
5765
    MALLOC(result->i);
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
5766 5767 5768 5769 5770
    if (result->i == 0) {
        toku_free(result);
        return ENOMEM;
    }
    memset(result->i, 0, sizeof *result->i);
5771
    result->i->dict_id = DICTIONARY_ID_NONE;
5772
    result->i->opened = 0;
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
5773 5774 5775
    result->i->open_flags = 0;
    result->i->open_mode = 0;
    result->i->brt = 0;
5776 5777
    result->i->indexer = NULL;
    result->i->refs = 1;
5778
    toku_list_init(&result->i->dbs_that_must_close_before_abort);
5779
    r = toku_brt_create(&result->i->brt);
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
5780 5781 5782
    if (r != 0) {
        toku_free(result->i);
        toku_free(result);
Yoni Fogel's avatar
Yoni Fogel committed
5783
        return r;
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
5784 5785 5786
    }
    *db = result;
    return 0;
Bradley C. Kuszmaul's avatar
Bradley C. Kuszmaul committed
5787
}
Bradley C. Kuszmaul's avatar
Bradley C. Kuszmaul committed
5788

5789 5790
int 
DB_CREATE_FUN (DB ** db, DB_ENV * env, u_int32_t flags) {
5791 5792 5793 5794
    toku_ydb_lock(); 
    int r = toku_db_create(db, env, flags); 
    toku_ydb_unlock(); 
    return r;
Rich Prohaska's avatar
Rich Prohaska committed
5795 5796 5797 5798
}

/* need db_strerror_r for multiple threads */

5799 5800
char *
db_strerror(int error) {
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
5801 5802 5803 5804 5805 5806 5807
    char *errorstr;
    if (error >= 0) {
        errorstr = strerror(error);
        if (errorstr)
            return errorstr;
    }
    
5808 5809 5810 5811 5812 5813 5814 5815 5816 5817 5818 5819 5820 5821 5822
    switch (error) {
        case DB_BADFORMAT:
            return "Database Bad Format (probably a corrupted database)";
        case DB_NOTFOUND:
            return "Not found";
        case TOKUDB_OUT_OF_LOCKS:
            return "Out of locks";
        case TOKUDB_DICTIONARY_TOO_OLD:
            return "Dictionary too old for this version of TokuDB";
        case TOKUDB_DICTIONARY_TOO_NEW:
            return "Dictionary too new for this version of TokuDB";
        case TOKUDB_CANCELED:
            return "User cancelled operation";
        case TOKUDB_NO_DATA:
            return "Ran out of data (not EOF)";
5823
    }
5824

Rich Prohaska's avatar
Rich Prohaska committed
5825
    static char unknown_result[100];    // Race condition if two threads call this at the same time. However even in a bad case, it should be some sort of null-terminated string.
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
5826 5827 5828 5829 5830
    errorstr = unknown_result;
    snprintf(errorstr, sizeof unknown_result, "Unknown error code: %d", error);
    return errorstr;
}

5831 5832
const char *
db_version(int *major, int *minor, int *patch) {
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
5833 5834 5835 5836 5837 5838
    if (major)
        *major = DB_VERSION_MAJOR;
    if (minor)
        *minor = DB_VERSION_MINOR;
    if (patch)
        *patch = DB_VERSION_PATCH;
5839 5840 5841 5842 5843
#if defined(TOKUDB_REVISION)
#define xstr(X) str(X)
#define str(X) #X
    return "tokudb " xstr(DB_VERSION_MAJOR) "." xstr(DB_VERSION_MINOR) "." xstr(DB_VERSION_PATCH) " build " xstr(TOKUDB_REVISION);
#else
Bradley C. Kuszmaul's avatar
Bradley C. Kuszmaul committed
5844
    return DB_VERSION_STRING;
5845
#endif
Bradley C. Kuszmaul's avatar
Bradley C. Kuszmaul committed
5846
}
5847
 
5848 5849
int 
db_env_set_func_fsync (int (*fsync_function)(int)) {
5850 5851
    return toku_set_func_fsync(fsync_function);
}
Yoni Fogel's avatar
Yoni Fogel committed
5852

5853 5854
int 
db_env_set_func_pwrite (ssize_t (*pwrite_function)(int, const void *, size_t, toku_off_t)) {
5855 5856
    return toku_set_func_pwrite(pwrite_function);
}
5857 5858 5859

int 
db_env_set_func_full_pwrite (ssize_t (*pwrite_function)(int, const void *, size_t, toku_off_t)) {
5860 5861
    return toku_set_func_full_pwrite(pwrite_function);
}
5862 5863 5864

int 
db_env_set_func_write (ssize_t (*write_function)(int, const void *, size_t)) {
5865 5866
    return toku_set_func_write(write_function);
}
5867 5868 5869

int 
db_env_set_func_full_write (ssize_t (*write_function)(int, const void *, size_t)) {
5870 5871
    return toku_set_func_full_write(write_function);
}
5872

5873 5874
int 
db_env_set_func_fdopen (FILE * (*fdopen_function)(int, const char *)) {
5875 5876
    return toku_set_func_fdopen(fdopen_function);
}
5877 5878 5879

int 
db_env_set_func_fopen (FILE * (*fopen_function)(const char *, const char *)) {
5880 5881
    return toku_set_func_fopen(fopen_function);
}
5882 5883 5884

int 
db_env_set_func_open (int (*open_function)(const char *, int, int)) {
5885 5886
    return toku_set_func_open(open_function);
}
5887 5888 5889

int 
db_env_set_func_fclose (int (*fclose_function)(FILE*)) {
5890 5891 5892
    return toku_set_func_fclose(fclose_function);
}

5893 5894 5895 5896 5897
void 
db_env_set_func_loader_fwrite (size_t (*fwrite_fun)(const void*,size_t,size_t,FILE*)) {
    brtloader_set_os_fwrite(fwrite_fun);
}

5898 5899
int 
db_env_set_func_malloc (void *(*f)(size_t)) {
5900 5901
    return toku_set_func_malloc(f);
}
5902 5903 5904

int 
db_env_set_func_realloc (void *(*f)(void*, size_t)) {
5905 5906
    return toku_set_func_realloc(f);
}
5907 5908 5909

int 
db_env_set_func_free (void (*f)(void*)) {
5910 5911
    return toku_set_func_free(f);
}
5912

5913
// Got to call dlmalloc, or else it won't get included.
5914 5915
void 
setup_dlmalloc (void) {
5916 5917 5918 5919
    db_env_set_func_malloc(dlmalloc);
    db_env_set_func_realloc(dlrealloc);
    db_env_set_func_free(dlfree);
}
5920 5921

// For test purposes only.
5922
// With this interface, all checkpoint users get the same callbacks and the same extras.
5923 5924
void 
db_env_set_checkpoint_callback (void (*callback_f)(void*), void* extra) {
5925
    toku_checkpoint_safe_client_lock();
5926 5927
    checkpoint_callback_f = callback_f;
    checkpoint_callback_extra = extra;
5928
    toku_checkpoint_safe_client_unlock();
5929
    //printf("set callback = %p, extra = %p\n", callback_f, extra);
5930
}
5931 5932 5933

void 
db_env_set_checkpoint_callback2 (void (*callback_f)(void*), void* extra) {
5934 5935 5936 5937 5938 5939
    toku_checkpoint_safe_client_lock();
    checkpoint_callback2_f = callback_f;
    checkpoint_callback2_extra = extra;
    toku_checkpoint_safe_client_unlock();
    //printf("set callback2 = %p, extra2 = %p\n", callback2_f, extra2);
}
Yoni Fogel's avatar
Yoni Fogel committed
5940

5941 5942
void 
db_env_set_recover_callback (void (*callback_f)(void*), void* extra) {
5943 5944 5945
    toku_recover_set_callback(callback_f, extra);
}

5946 5947
void 
db_env_set_recover_callback2 (void (*callback_f)(void*), void* extra) {
5948 5949 5950
    toku_recover_set_callback2(callback_f, extra);
}

5951 5952
void 
db_env_set_loader_size_factor (uint32_t factor) {
5953 5954 5955
    toku_brtloader_set_size_factor(factor);
}

5956 5957
void 
db_env_set_mvcc_garbage_collection_verification(u_int32_t verification_mode) {
Zardosht Kasheff's avatar
Zardosht Kasheff committed
5958 5959
    garbage_collection_debug = (verification_mode != 0);
}
5960 5961 5962



Yoni Fogel's avatar
Yoni Fogel committed
5963 5964 5965 5966 5967 5968 5969 5970 5971 5972
// HACK: To ensure toku_pthread_yield gets included in the .so
// non-static would require a prototype in a header
// static (since unused) would give a warning
// static + unused would not actually help toku_pthread_yield get in the .so
// static + used avoids all the warnings and makes sure toku_pthread_yield is in the .so
static void __attribute__((__used__))
include_toku_pthread_yield (void) {
    toku_pthread_yield();
}

5973 5974 5975 5976

// For test purposes only, translate dname to iname
static int 
env_get_iname(DB_ENV* env, DBT* dname_dbt, DBT* iname_dbt) {
5977
    toku_ydb_lock();
5978
    DB *directory = env->i->directory;
5979
    int r = autotxn_db_get(directory, NULL, dname_dbt, iname_dbt, DB_SERIALIZABLE|DB_PRELOCKED); // allocates memory for iname
5980
    toku_ydb_unlock();
5981 5982 5983
    return r;
}

5984 5985 5986 5987 5988 5989 5990 5991 5992 5993 5994 5995 5996 5997 5998
/* Following functions (ydb_load_xxx()) are used by loader:
 */


// When the loader is created, it makes this call.
// For each dictionary to be loaded, replace old iname in directory
// with a newly generated iname.  This will also take a write lock
// on the directory entries.  The write lock will be released when
// the transaction of the loader is completed.
// If the transaction commits, the new inames are in place.
// If the transaction aborts, the old inames will be restored.
// The new inames are returned to the caller.  
// It is the caller's responsibility to free them.
// Return 0 on success (could fail if write lock not available).
int
5999
ydb_load_inames(DB_ENV * env, DB_TXN * txn, int N, DB * dbs[N], char * new_inames_in_env[N], LSN *load_lsn) {
6000 6001 6002 6003 6004
    int rval;
    int i;
    
    int using_txns = env->i->open_flags & DB_INIT_TXN;
    DB_TXN * child = NULL;
6005
    TXNID xid = 0;
6006 6007 6008
    DBT dname_dbt;  // holds dname
    DBT iname_dbt;  // holds new iname
    
6009 6010 6011 6012
    for (i=0; i<N; i++) {
	new_inames_in_env[i] = NULL;
    }

6013 6014 6015 6016 6017 6018 6019 6020 6021 6022 6023 6024
    // begin child (unless transactionless)
    if (using_txns) {
	rval = toku_txn_begin(env, txn, &child, DB_TXN_NOSYNC, 1);
	assert(rval == 0);
	xid = toku_txn_get_txnid(db_txn_struct_i(child)->tokutxn);
    }
    for (i = 0; i < N; i++) {
	char * dname = dbs[i]->i->dname;
	toku_fill_dbt(&dname_dbt, dname, strlen(dname)+1);
	// now create new iname
	char hint[strlen(dname) + 1];
	create_iname_hint(dname, hint);
6025
	char * new_iname = create_iname(env, xid, hint, i);               // allocates memory for iname_in_env
6026
	new_inames_in_env[i] = new_iname;
6027
        toku_fill_dbt(&iname_dbt, new_iname, strlen(new_iname) + 1);      // iname_in_env goes in directory
6028 6029
        rval = toku_db_put(env->i->directory, child, &dname_dbt, &iname_dbt, DB_YESOVERWRITE);  // DB_YESOVERWRITE necessary
	if (rval) break;
Yoni Fogel's avatar
Yoni Fogel committed
6030 6031 6032 6033 6034 6035
    }

    // Generate load log entries.
    if (!rval && using_txns) {
        TOKUTXN ttxn = db_txn_struct_i(txn)->tokutxn;
        int do_fsync = 0;
6036
        LSN *get_lsn = NULL;
Yoni Fogel's avatar
Yoni Fogel committed
6037 6038 6039
        for (i = 0; i < N; i++) {
            BRT brt  = dbs[i]->i->brt;
            //Fsync is necessary for the last one only.
6040 6041 6042 6043 6044
            if (i==N-1) {
                do_fsync = 1; //We only need a single fsync of logs.
                get_lsn  = load_lsn; //Set pointer to capture the last lsn.
            }
            rval = toku_brt_load(brt, ttxn, new_inames_in_env[i], do_fsync, get_lsn);
Yoni Fogel's avatar
Yoni Fogel committed
6045 6046
            if (rval) break;
        }
6047 6048 6049 6050 6051
    }
	
    if (using_txns) {
	// close txn
	if (rval == 0) {  // all well so far, commit child
6052
	    rval = toku_txn_commit(child, DB_TXN_NOSYNC, NULL, NULL);
6053 6054 6055
	    assert(rval==0);
	}
	else {         // abort child
6056
	    int r2 = toku_txn_abort(child, NULL, NULL);
6057 6058
	    assert(r2==0);
	    for (i=0; i<N; i++) {
6059 6060 6061 6062
		if (new_inames_in_env[i]) {
		    toku_free(new_inames_in_env[i]);
		    new_inames_in_env[i] = NULL;
		}
6063 6064 6065 6066 6067 6068 6069
	    }
	}
    }

    return rval;
}

6070
int
6071
locked_ydb_load_inames(DB_ENV * env, DB_TXN * txn, int N, DB * dbs[N], char * new_inames_in_env[N], LSN *load_lsn) {
6072
    toku_ydb_lock();
6073
    int r = ydb_load_inames(env, txn, N, dbs, new_inames_in_env, load_lsn);
6074 6075 6076 6077
    toku_ydb_unlock();
    return r;
}

6078 6079 6080 6081
// TODO 2216:  Patch out this (dangerous) function when loader is working and 
//             we don't need to test the low-level redirect anymore.
// for use by test programs only, just a wrapper around brt call:
int
6082
toku_test_db_redirect_dictionary(DB * db, char * dname_of_new_file, DB_TXN *dbtxn) {
6083 6084 6085 6086 6087 6088 6089 6090 6091 6092
    int r;
    DBT dname_dbt;
    DBT iname_dbt;
    char * new_iname_in_env;

    BRT brt = db->i->brt;
    TOKUTXN tokutxn = db_txn_struct_i(dbtxn)->tokutxn;

    toku_fill_dbt(&dname_dbt, dname_of_new_file, strlen(dname_of_new_file)+1);
    init_dbt_realloc(&iname_dbt);  // sets iname_dbt.data = NULL
6093
    r = toku_db_get(db->dbenv->i->directory, dbtxn, &dname_dbt, &iname_dbt, DB_SERIALIZABLE);  // allocates memory for iname
6094 6095 6096
    assert(r==0);
    new_iname_in_env = iname_dbt.data;

Yoni Fogel's avatar
Yoni Fogel committed
6097
    r = toku_dictionary_redirect(new_iname_in_env, brt, tokutxn);
6098 6099 6100 6101

    toku_free(new_iname_in_env);
    return r;
}
6102 6103 6104 6105 6106 6107 6108 6109 6110 6111 6112

//Tets only function
uint64_t
toku_test_get_latest_lsn(DB_ENV *env) {
    LSN rval = ZERO_LSN;
    if (env && env->i->logger) {
        rval = toku_logger_last_lsn(env->i->logger);
    }
    return rval.lsn;
}

6113 6114
int 
toku_test_get_checkpointing_user_data_status (void) {
6115 6116
    return toku_cachetable_get_checkpointing_user_data_status();
}
Zardosht Kasheff's avatar
Zardosht Kasheff committed
6117

6118 6119 6120 6121 6122 6123 6124 6125 6126 6127 6128 6129
int
toku_grab_write_lock (DB* db, DBT* key, TOKUTXN tokutxn) {
    RANGE_LOCK_REQUEST_S request;
    DB_TXN * txn = toku_txn_get_container_db_txn(tokutxn);
    //Left end of range == right end of range (point lock)
    write_lock_request_init(&request, txn, db,
			    key,
			    key);
    int r = grab_range_lock(&request);
    return r;
}