ydb.c 235 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"
Rich Prohaska's avatar
Rich Prohaska committed
36
#include "log_header.h"
Yoni Fogel's avatar
Yoni Fogel committed
37

38

39 40 41 42 43 44 45 46 47 48
#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

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

51 52 53 54 55 56 57 58 59 60 61 62 63 64
// Set when env is panicked, never cleared.
static int env_is_panicked = 0;

static void
env_panic(DB_ENV * env, int cause, char * msg) {
    if (cause == 0)
	cause = -1;  // if unknown cause, at least guarantee panic
    if (msg == NULL)
	msg = "Unknown cause in env_panic\n";
    env_is_panicked = cause;
    env->i->is_panicked = cause;
    env->i->panic_string = toku_strdup(msg);
}

65 66
// Accountability: operation counters available for debugging and for "show engine status"
static u_int64_t num_inserts;
67
static u_int64_t num_inserts_fail;
68
static u_int64_t num_deletes;
69
static u_int64_t num_deletes_fail;
70 71
static u_int64_t num_updates;
static u_int64_t num_updates_fail;
72 73
static u_int64_t num_updates_broadcast;
static u_int64_t num_updates_broadcast_fail;
74 75 76 77 78 79
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;
80 81
static u_int64_t num_point_queries;
static u_int64_t num_sequential_queries;
82 83 84 85 86 87

static u_int64_t directory_read_locks;        /* total directory read locks taken */ 
static u_int64_t directory_read_locks_fail;   /* total directory read locks unable to be taken */ 
static u_int64_t directory_write_locks;       /* total directory write locks taken */ 
static u_int64_t directory_write_locks_fail;  /* total directory write locks unable to be taken */ 

88 89 90
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
91
static DB_ENV * volatile most_recent_env;   // most recently opened env, used for engine status on crash.  Note there are likely to be races on this if you have multiple threads creating and closing environments in parallel.  We'll declare it volatile since at least that helps make sure the compiler doesn't optimize away certain code (e.g., if while debugging, you write a code that spins on most_recent_env, you'd like to compiler not to optimize your code away.)
92

93 94
static uint32_t  engine_status_enable = 1;   // if zero, suppress engine status output on failed assert, for test programs only

95 96 97 98 99 100
static void
init_status_info(void) {
    num_inserts = 0;
    num_inserts_fail = 0;
    num_deletes = 0;
    num_deletes_fail = 0;
101 102
    num_updates = 0;
    num_updates_fail = 0;
103 104
    num_updates_broadcast = 0;
    num_updates_broadcast_fail = 0;
105 106 107 108 109 110
    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;
111 112
    num_point_queries = 0;
    num_sequential_queries = 0;
113 114 115 116
    directory_read_locks = 0;
    directory_read_locks_fail = 0;
    directory_write_locks = 0;
    directory_write_locks_fail = 0;
117 118 119 120
    logsuppress = 0;
    logsuppressfail = 0;
    startuptime = time(NULL);
}
121

122 123 124
const char * environmentdictionary = "tokudb.environment";
const char * fileopsdirectory = "tokudb.directory";

125
static int env_get_iname(DB_ENV* env, DBT* dname_dbt, DBT* iname_dbt);
126 127 128
static int toku_maybe_get_engine_status_text (char* buff, int buffsize);  // for use by toku_assert
static void toku_maybe_set_env_panic(int code, char * msg);               // for use by toku_assert

129

Yoni Fogel's avatar
Yoni Fogel committed
130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160
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;
}
161

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

166 167 168 169 170 171 172
static inline DBT*
init_dbt_realloc(DBT *dbt) {
    memset(dbt, 0, sizeof(*dbt));
    dbt->flags = DB_DBT_REALLOC;
    return dbt;
}

173 174 175 176 177 178
//Callback used for redirecting dictionaries.
static void
ydb_set_brt(DB *db, BRT brt) {
    db->i->brt = brt;
}

179 180
int 
toku_ydb_init(void) {
Yoni Fogel's avatar
Yoni Fogel committed
181 182 183
    int r = 0;
    //Lower level must be initialized first.
    if (r==0) 
184
        r = toku_brt_init(toku_ydb_lock, toku_ydb_unlock, ydb_set_brt);
Yoni Fogel's avatar
Yoni Fogel committed
185 186 187
    if (r==0) 
        r = toku_ydb_lock_init();
    return r;
188 189
}

190
// Do not clean up resources if env is panicked, just exit ugly
191 192
int 
toku_ydb_destroy(void) {
Yoni Fogel's avatar
Yoni Fogel committed
193
    int r = 0;
194
    if (env_is_panicked == 0) {
Yoni Fogel's avatar
Yoni Fogel committed
195
        r = toku_ydb_lock_destroy();
196 197 198 199
	//Lower level must be cleaned up last.
	if (r==0)
	    r = toku_brt_destroy();
    }
Yoni Fogel's avatar
Yoni Fogel committed
200
    return r;
201 202
}

203 204 205 206 207
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
208 209
/* env methods */
static int toku_env_close(DB_ENV *env, u_int32_t flags);
210 211 212
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
213

214 215
static inline int 
env_opened(DB_ENV *env) {
216
    return env->i->cachetable != 0;
Rich Prohaska's avatar
Rich Prohaska committed
217 218
}

219 220
static void 
env_init_open_txn(DB_ENV *env) {
221
    toku_list_init(&env->i->open_txns);
Rich Prohaska's avatar
Rich Prohaska committed
222 223
}

224
// add a txn to the list of open txn's
225 226
static void 
env_add_open_txn(DB_ENV *env, DB_TXN *txn) {
227
    toku_list_push(&env->i->open_txns, (struct toku_list *) (void *) &txn->open_txns);
228 229 230
}

// remove a txn from the list of open txn's
231 232
static void 
env_remove_open_txn(DB_ENV *UU(env), DB_TXN *txn) {
233
    toku_list_remove((struct toku_list *) (void *) &txn->open_txns);
Rich Prohaska's avatar
Rich Prohaska committed
234 235
}

236
static int toku_txn_abort(DB_TXN * txn, TXN_PROGRESS_POLL_FUNCTION, void*);
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
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
271
    int in_red;    // set true to prevent certain operations (returning ENOSPC)
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

    // 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:
Barry Perlman's avatar
Barry Perlman committed
306
        if (!in_red) {
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 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373
	    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; 
374
    if (r) env->i->enospc_redzone_ctr++;
375 376 377
    return r;
}

378 379 380 381 382 383
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
384 385
/* db methods */
static inline int db_opened(DB *db) {
386
    return db->i->opened != 0;
Rich Prohaska's avatar
Rich Prohaska committed
387 388
}

Zardosht Kasheff's avatar
Zardosht Kasheff committed
389

Rich Prohaska's avatar
Rich Prohaska committed
390
static int toku_db_put(DB * db, DB_TXN * txn, DBT * key, DBT * data, u_int32_t flags);
391 392
static int toku_db_update(DB *db, DB_TXN *txn, const DBT *key, const DBT *update_function_extra, u_int32_t flags);
static int toku_db_update_broadcast(DB *db, DB_TXN *txn, const DBT *update_function_extra, u_int32_t flags);
Rich Prohaska's avatar
Rich Prohaska committed
393
static int toku_db_get (DB * db, DB_TXN * txn, DBT * key, DBT * data, u_int32_t flags);
394
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
395 396 397

/* txn methods */

398
/* lightweight cursor methods. */
399 400 401
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
402

403
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
404

405
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
406

407 408
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
409

410 411
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);
412
static int toku_c_getf_set_range_reverse(DBC *c, u_int32_t flag, DBT *key, YDB_CALLBACK_FUNCTION f, void *extra);
413

Yoni Fogel's avatar
Yoni Fogel committed
414 415
// Effect: Lightweight cursor get

Rich Prohaska's avatar
Rich Prohaska committed
416 417 418 419 420
/* 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
421

422
static void
423 424 425
env_setup_real_dir(DB_ENV *env, char **real_dir, const char *nominal_dir) {
    toku_free(*real_dir);
    *real_dir = NULL;
426 427

    assert(env->i->dir);
428 429
    if (nominal_dir) 
	*real_dir = toku_construct_full_name(2, env->i->dir, nominal_dir);
430
    else
431 432 433 434 435 436
        *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);
437 438 439 440
}

static void
env_setup_real_log_dir(DB_ENV *env) {
441
    env_setup_real_dir(env, &env->i->real_log_dir, env->i->lg_dir);
442 443
}

444 445 446 447 448
static void
env_setup_real_tmp_dir(DB_ENV *env) {
    env_setup_real_dir(env, &env->i->real_tmp_dir, env->i->tmp_dir);
}

449 450
static int 
ydb_do_recovery (DB_ENV *env) {
451
    assert(env->i->real_log_dir);
452
    toku_ydb_unlock();
453
    int r = tokudb_recover(env->i->dir, env->i->real_log_dir, env->i->bt_compare,
454
                           env->i->update_function,
455
                           env->i->generate_row_for_put, env->i->generate_row_for_del,
456
                           env->i->cachetable_size);
457
    toku_ydb_lock();
458
    return r;
459 460
}

461 462
static int 
needs_recovery (DB_ENV *env) {
463 464
    assert(env->i->real_log_dir);
    int recovery_needed = tokudb_needs_recovery(env->i->real_log_dir, TRUE);
465
    return recovery_needed ? DB_RUNRECOVERY : 0;
466 467
}

468 469 470 471
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);
472
static int toku_txn_begin(DB_ENV *env, DB_TXN * stxn, DB_TXN ** txn, u_int32_t flags, int internal);
473
static int toku_txn_commit(DB_TXN * txn, u_int32_t flags, TXN_PROGRESS_POLL_FUNCTION, void*);
474 475
static int db_open_iname(DB * db, DB_TXN * txn, const char *iname, u_int32_t flags, int mode);

476
static void finalize_file_removal(DICTIONARY_ID dict_id, void * extra);
477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500

// 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;
}

501 502
// Keys used in persistent environment dictionary:
// Following keys added in version 12
503
static const char * orig_env_ver_key = "original_version";
504
static const char * curr_env_ver_key = "current_version";  
505
// Following keys added in version 14, add more keys for future versions
506
static const char * creation_time_key         = "creation_time";
507 508 509
static const char * last_lsn_of_v13_key       = "last_lsn_of_v13";
static const char * upgrade_v14_time_key      = "upgrade_v14_time";      
static const char * upgrade_v14_footprint_key = "upgrade_v14_footprint";
510 511 512 513 514 515

// 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;
516 517 518
static uint64_t persistent_last_lsn_of_v13;
static time_t   persistent_upgrade_v14_time;
static uint64_t persistent_upgrade_v14_footprint;
519 520 521 522

// 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.
523
static int
524
maybe_upgrade_persistent_environment_dictionary(DB_ENV * env, DB_TXN * txn, LSN last_lsn_of_clean_shutdown_read_from_log) {
525 526
    int r;
    DBT key, val;
527
    DB *persistent_environment = env->i->persistent_environment;
528

529
    toku_fill_dbt(&key, curr_env_ver_key, strlen(curr_env_ver_key));
530
    toku_init_dbt(&val);
531
    r = toku_db_get(persistent_environment, txn, &key, &val, 0);
532
    assert(r == 0);
533 534
    uint32_t stored_env_version = toku_dtoh32(*(uint32_t*)val.data);
    persistent_stored_env_version_at_startup = stored_env_version;
535
    if (stored_env_version > BRT_LAYOUT_VERSION)
536
	r = TOKUDB_DICTIONARY_TOO_NEW;
537 538 539
    else if (stored_env_version < BRT_LAYOUT_MIN_SUPPORTED_VERSION)
	r = TOKUDB_DICTIONARY_TOO_OLD;
    else if (stored_env_version < BRT_LAYOUT_VERSION) {
540
        const uint32_t curr_env_ver_d = toku_htod32(BRT_LAYOUT_VERSION);
541
        toku_fill_dbt(&key, curr_env_ver_key, strlen(curr_env_ver_key));
542
        toku_fill_dbt(&val, &curr_env_ver_d, sizeof(curr_env_ver_d));
543
        r = toku_db_put(persistent_environment, txn, &key, &val, 0);
544 545
        assert(r==0);
	
546 547 548
	uint64_t last_lsn_of_v13_d = toku_htod64(last_lsn_of_clean_shutdown_read_from_log.lsn);
	toku_fill_dbt(&key, last_lsn_of_v13_key, strlen(last_lsn_of_v13_key));
	toku_fill_dbt(&val, &last_lsn_of_v13_d, sizeof(last_lsn_of_v13_d));
549
	r = toku_db_put(persistent_environment, txn, &key, &val, 0);
550 551
        assert(r==0);
	
552 553 554
	time_t upgrade_v14_time_d = toku_htod64(time(NULL));
	toku_fill_dbt(&key, upgrade_v14_time_key, strlen(upgrade_v14_time_key));
	toku_fill_dbt(&val, &upgrade_v14_time_d, sizeof(upgrade_v14_time_d));
555
	r = toku_db_put(persistent_environment, txn, &key, &val, DB_NOOVERWRITE);
556
        assert(r==0);
557

558 559 560
	uint64_t upgrade_v14_footprint_d = toku_htod64(toku_log_upgrade_get_footprint());
	toku_fill_dbt(&key, upgrade_v14_footprint_key, strlen(upgrade_v14_footprint_key));
	toku_fill_dbt(&val, &upgrade_v14_footprint_d, sizeof(upgrade_v14_footprint_d));
561 562
	r = toku_db_put(persistent_environment, txn, &key, &val, DB_NOOVERWRITE);
        assert(r==0);
563
    }
564
    return r;
565 566
}

567

Barry Perlman's avatar
Barry Perlman committed
568
// Capture contents of persistent_environment dictionary so that it can be read by engine status
569
static void
Barry Perlman's avatar
Barry Perlman committed
570
capture_persistent_env_contents (DB_ENV * env, DB_TXN * txn) {
571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589
    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
590
    if (persistent_original_env_version >= BRT_LAYOUT_VERSION_14) {
591 592 593 594 595 596 597 598 599 600
	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
	
601
	toku_fill_dbt(&key, last_lsn_of_v13_key, strlen(last_lsn_of_v13_key));
602 603 604
	toku_init_dbt(&val);
	r = toku_db_get(persistent_environment, txn, &key, &val, 0);
	assert(r == 0);
605
	persistent_last_lsn_of_v13 = toku_dtoh64(*(uint32_t*)val.data);
606

607
	toku_fill_dbt(&key, upgrade_v14_time_key, strlen(upgrade_v14_time_key));
608 609 610
	toku_init_dbt(&val);
	r = toku_db_get(persistent_environment, txn, &key, &val, 0);
	assert(r == 0);
611
	persistent_upgrade_v14_time = toku_dtoh64((*(time_t*)val.data));
612

613
	toku_fill_dbt(&key, upgrade_v14_footprint_key, strlen(upgrade_v14_footprint_key));
614 615 616
	toku_init_dbt(&val);
	r = toku_db_get(persistent_environment, txn, &key, &val, 0);
	assert(r == 0);
617
	persistent_upgrade_v14_footprint = toku_dtoh64((*(uint64_t*)val.data));
618 619 620 621 622 623 624
    }

}




625 626 627
// return 0 if log exists or ENOENT if log does not exist
static int
ydb_recover_log_exists(DB_ENV *env) {
628
    int r = tokudb_recover_log_exists(env->i->real_log_dir);
629 630 631 632 633
    return r;
}


// Validate that all required files are present, no side effects.
634 635
// 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.
636 637 638
// 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 
639
validate_env(DB_ENV * env, BOOL * valid_newenv, BOOL need_rollback_cachefile) {
640
    int r;
641
    BOOL expect_newenv = FALSE;        // set true if we expect to create a new env
642 643
    toku_struct_stat buf;
    char* path = NULL;
644

645
    // Test for persistent environment
Yoni Fogel's avatar
Yoni Fogel committed
646
    path = toku_construct_full_name(2, env->i->dir, environmentdictionary);
647 648
    assert(path);
    r = toku_stat(path, &buf);
649
    int stat_errno = errno;
650
    toku_free(path);
651 652 653
    if (r == 0) {
	expect_newenv = FALSE;  // persistent info exists
    }
654
    else if (stat_errno == ENOENT) {
655 656
	expect_newenv = TRUE;
	r = 0;
657 658
    }
    else {
659 660
	r = toku_ydb_do_error(env, errno, "Unable to access persistent environment\n");
	assert(r);
661
    }
662

663
    // Test for existence of rollback cachefile if it is expected to exist
664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680
    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 {
681
	    r = toku_ydb_do_error(env, stat_errno, "Unable to access rollback cachefile\n");
682 683 684 685
	    assert(r);
	}
    }

686 687
    // Test for fileops directory
    if (r == 0) {
Yoni Fogel's avatar
Yoni Fogel committed
688
	path = toku_construct_full_name(2, env->i->dir, fileopsdirectory);
689 690
	assert(path);
	r = toku_stat(path, &buf);
691
	stat_errno = errno;
692 693 694 695 696
	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");
	}
697
	else if (stat_errno == ENOENT) {
698 699 700 701 702 703
	    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 {
704
	    r = toku_ydb_do_error(env, stat_errno, "Unable to access fileops directory\n");
705 706 707 708 709
	    assert(r);
	}
    }

    // Test for recovery log
710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727
    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;
}

Barry Perlman's avatar
Barry Perlman committed
728 729 730 731 732

// The version of the environment (on disk) is the version of the recovery log.  
// If the recovery log is of the current version, then there is no upgrade to be done.  
// If the recovery log is of an old version, then replacing it with a new recovery log
// of the current version is how the upgrade is done.  
Barry Perlman's avatar
Barry Perlman committed
733
// Note, the upgrade procedure takes a checkpoint, so we must release the ydb lock.
734
static int
735
ydb_maybe_upgrade_env (DB_ENV *env, LSN * last_lsn_of_clean_shutdown_read_from_log, BOOL * upgrade_in_progress) {
736 737 738
    int r = 0;
    if (env->i->open_flags & DB_INIT_TXN && env->i->open_flags & DB_INIT_LOG) {
        toku_ydb_unlock();
739
        r = toku_maybe_upgrade_log(env->i->dir, env->i->real_log_dir, last_lsn_of_clean_shutdown_read_from_log, upgrade_in_progress);
740 741 742 743 744 745
        toku_ydb_lock();
    }
    return r;
}


Yoni Fogel's avatar
Yoni Fogel committed
746 747 748 749 750 751 752 753 754 755 756 757
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);
}
758 759 760 761 762 763 764

// 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) {
765
    HANDLE_PANICKED_ENV(env);
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
766
    int r;
767
    BOOL newenv;  // true iff creating a new environment
768
    u_int32_t unused_flags=flags;
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
769

Rich Prohaska's avatar
Rich Prohaska committed
770
    if (env_opened(env)) {
Yoni Fogel's avatar
Yoni Fogel committed
771 772
	r = toku_ydb_do_error(env, EINVAL, "The environment is already open\n");
        goto cleanup;
773
    }
Yoni Fogel's avatar
Yoni Fogel committed
774

775 776
    most_recent_env = NULL;

777 778
    assert(sizeof(time_t) == sizeof(uint64_t));

779 780 781 782
    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);


783 784
    // 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
785
    if ((flags & DB_PRIVATE) && !(flags & DB_CREATE)) {
Yoni Fogel's avatar
Yoni Fogel committed
786 787
	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
788 789
    }

790
    if (!(flags & DB_PRIVATE)) {
Yoni Fogel's avatar
Yoni Fogel committed
791 792
	r = toku_ydb_do_error(env, ENOENT, "TokuDB requires DB_PRIVATE\n");
        goto cleanup;
Yoni Fogel's avatar
Yoni Fogel committed
793
    }
794

Yoni Fogel's avatar
Yoni Fogel committed
795 796 797 798
    if ((flags & DB_INIT_LOG) && !(flags & DB_INIT_TXN)) {
	r = toku_ydb_do_error(env, EINVAL, "TokuDB requires transactions for logging\n");
        goto cleanup;
    }
799

Yoni Fogel's avatar
Yoni Fogel committed
800
    if (!home) home = ".";
Yoni Fogel's avatar
Yoni Fogel committed
801

802 803 804
    // Verify that the home exists.
    {
	BOOL made_new_home = FALSE;
Zardosht Kasheff's avatar
Zardosht Kasheff committed
805
        char* new_home = NULL;
806
    	toku_struct_stat buf;
807
        if (strlen(home) > 1 && home[strlen(home)-1] == '\\') {
Zardosht Kasheff's avatar
Zardosht Kasheff committed
808 809 810 811 812
            new_home = toku_malloc(strlen(home));
            memcpy(new_home, home, strlen(home));
            new_home[strlen(home) - 1] = 0;
            made_new_home = TRUE;
        }
813
    	r = toku_stat(made_new_home? new_home : home, &buf);
Zardosht Kasheff's avatar
Zardosht Kasheff committed
814 815 816 817
        if (made_new_home) {
            toku_free(new_home);
        }
    	if (r!=0) {
Yoni Fogel's avatar
Yoni Fogel committed
818 819
    	    r = toku_ydb_do_error(env, errno, "Error from toku_stat(\"%s\",...)\n", home);
            goto cleanup;
Zardosht Kasheff's avatar
Zardosht Kasheff committed
820
    	}
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
821
    }
822
    unused_flags &= ~DB_PRIVATE;
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
823 824 825

    if (env->i->dir)
        toku_free(env->i->dir);
Yoni Fogel's avatar
Yoni Fogel committed
826
    env->i->dir = toku_strdup(home);
827
    if (env->i->dir == 0) {
Yoni Fogel's avatar
Yoni Fogel committed
828 829
	r = toku_ydb_do_error(env, ENOMEM, "Out of memory\n");
        goto cleanup;
830
    }
Yoni Fogel's avatar
Yoni Fogel committed
831 832 833 834
    if (0) {
        died1:
        toku_free(env->i->dir);
        env->i->dir = NULL;
Yoni Fogel's avatar
Yoni Fogel committed
835
        goto cleanup;
Yoni Fogel's avatar
Yoni Fogel committed
836
    }
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
837 838
    env->i->open_flags = flags;
    env->i->open_mode = mode;
Bradley C. Kuszmaul's avatar
Bradley C. Kuszmaul committed
839

840 841
    env_setup_real_data_dir(env);
    env_setup_real_log_dir(env);
842
    env_setup_real_tmp_dir(env);
843

Yoni Fogel's avatar
Yoni Fogel committed
844 845
    r = single_process_lock(env->i->dir, "environment", &env->i->envdir_lockfd);
    if (r!=0) goto cleanup;
846
    r = single_process_lock(env->i->real_data_dir, "data", &env->i->datadir_lockfd);
Yoni Fogel's avatar
Yoni Fogel committed
847
    if (r!=0) goto cleanup;
848
    r = single_process_lock(env->i->real_log_dir, "logs", &env->i->logdir_lockfd);
Yoni Fogel's avatar
Yoni Fogel committed
849
    if (r!=0) goto cleanup;
850
    r = single_process_lock(env->i->real_tmp_dir, "temp", &env->i->tmpdir_lockfd);
Yoni Fogel's avatar
Yoni Fogel committed
851 852 853
    if (r!=0) goto cleanup;


854 855 856 857 858
    BOOL need_rollback_cachefile = FALSE;
    if (flags & (DB_INIT_TXN | DB_INIT_LOG)) {
        need_rollback_cachefile = TRUE;
    }

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

861 862 863
    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
864
    if (r!=0) goto cleanup;
865

866 867 868 869 870 871 872 873 874 875 876
    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
    }
    
877
    r = validate_env(env, &newenv, need_rollback_cachefile);  // make sure that environment is either new or complete
Yoni Fogel's avatar
Yoni Fogel committed
878
    if (r != 0) goto cleanup;
879

880
    unused_flags &= ~DB_INIT_TXN & ~DB_INIT_LOG;
881

882 883 884 885 886 887 888
    // 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
889
                if (r != 0) goto cleanup;
890 891 892
            } 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
893
                if (r != 0) goto cleanup;
894
            }
895 896
        }
    }
897 898
    
    toku_loader_cleanup_temp_files(env);
899

Bradley C. Kuszmaul's avatar
Bradley C. Kuszmaul committed
900
    if (flags & (DB_INIT_TXN | DB_INIT_LOG)) {
901
	assert(env->i->logger);
902
        toku_logger_write_log_files(env->i->logger, (BOOL)((flags & DB_INIT_LOG) != 0));
903
        r = toku_logger_open(env->i->real_log_dir, env->i->logger);
904
	if (r!=0) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
905
	    toku_ydb_do_error(env, r, "Could not open logger\n");
Bradley C. Kuszmaul's avatar
Bradley C. Kuszmaul committed
906
	died2:
907
	    toku_logger_close(&env->i->logger);
Bradley C. Kuszmaul's avatar
Bradley C. Kuszmaul committed
908 909
	    goto died1;
	}
910 911 912
    } 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
913 914
    }

915 916 917 918 919 920 921
    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
922 923
//	r = toku_ydb_do_error(env, EINVAL, "TokuDB requires DB_THREAD");
//	goto cleanup;
924 925 926 927
//    }
    unused_flags &= ~DB_THREAD;

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

932
    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
933
    if (r!=0) goto died2;
934
    toku_cachetable_set_lock_unlock_for_io(env->i->cachetable, toku_ydb_lock, toku_ydb_unlock);
935

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

938 939
    int using_txns = env->i->open_flags & DB_INIT_TXN;
    if (env->i->logger) {
940 941
	// 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;
942 943 944
	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);
945
        r = toku_logger_open_rollback(env->i->logger, env->i->cachetable, create_new_rollback_file);
946
        assert(r==0);
947
    }
948

949 950
    DB_TXN *txn=NULL;
    if (using_txns) {
951
        r = toku_txn_begin(env, 0, &txn, 0, 1);
952
        assert(r==0);
953
    }
954

955 956
    {
        r = toku_db_create(&env->i->persistent_environment, env, 0);
957
        assert(r==0);
958 959
        r = db_use_builtin_key_cmp(env->i->persistent_environment);
        assert(r==0);
960
	r = db_open_iname(env->i->persistent_environment, txn, environmentdictionary, DB_CREATE, mode);
961
	assert(r==0);
962
	if (newenv) {
963 964
	    // create new persistent_environment
	    DBT key, val;
965 966 967
	    persistent_original_env_version = BRT_LAYOUT_VERSION;
	    const uint32_t environment_version = toku_htod32(persistent_original_env_version);

968
	    toku_fill_dbt(&key, orig_env_ver_key, strlen(orig_env_ver_key));
969 970 971
	    toku_fill_dbt(&val, &environment_version, sizeof(environment_version));
	    r = toku_db_put(env->i->persistent_environment, txn, &key, &val, 0);
	    assert(r==0);
972

973
	    toku_fill_dbt(&key, curr_env_ver_key, strlen(curr_env_ver_key));
974 975 976
	    toku_fill_dbt(&val, &environment_version, sizeof(environment_version));
	    r = toku_db_put(env->i->persistent_environment, txn, &key, &val, 0);
	    assert(r==0);
977 978 979 980 981 982

	    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);
983 984
	}
	else {
985
	    r = maybe_upgrade_persistent_environment_dictionary(env, txn, last_lsn_of_clean_shutdown_read_from_log);
986 987
	    assert(r==0);
	}
Barry Perlman's avatar
Barry Perlman committed
988
	capture_persistent_env_contents(env, txn);
989 990 991
    }
    {
        r = toku_db_create(&env->i->directory, env, 0);
992
        assert(r==0);
993 994
        r = db_use_builtin_key_cmp(env->i->directory);
        assert(r==0);
995 996
        r = db_open_iname(env->i->directory, txn, fileopsdirectory, DB_CREATE, mode);
        assert(r==0);
997 998
    }
    if (using_txns) {
999
        r = toku_txn_commit(txn, 0, NULL, NULL);
1000
        assert(r==0);
1001 1002
    }
    toku_ydb_unlock();
1003
    r = toku_checkpoint(env->i->cachetable, env->i->logger, NULL, NULL, NULL, NULL);
1004
    assert(r==0);
1005
    toku_ydb_lock();
1006 1007
    env_fs_poller(env);          // get the file system state at startup
    env_fs_init_minicron(env); 
Yoni Fogel's avatar
Yoni Fogel committed
1008 1009 1010 1011 1012 1013
cleanup:
    if (r!=0) {
        if (env && env->i) {
            unlock_single_process(env);
        }
    }
1014
    if (r == 0) {
1015
	errno = 0; // tabula rasa.   If there's a crash after env was successfully opened, no misleading errno will have been left around by this code.
1016
	most_recent_env = env;
1017
	toku_assert_set_fpointers(toku_maybe_get_engine_status_text, toku_maybe_set_env_panic);
1018
    }
Yoni Fogel's avatar
Yoni Fogel committed
1019
    return r;
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
1020
}
Bradley C. Kuszmaul's avatar
Bradley C. Kuszmaul committed
1021

1022

1023 1024
static int 
toku_env_close(DB_ENV * env, u_int32_t flags) {
1025
    int r = 0;
1026
    char * err_msg = NULL;
1027

1028 1029
    most_recent_env = NULL; // Set most_recent_env to NULL so that we don't have a dangling pointer (and if there's an error, the toku assert code would try to look at the env.)

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

1032
    if (toku_env_is_panicked(env)) goto panic_and_quit_early;
1033
    if (!toku_list_empty(&env->i->open_txns)) {
1034 1035
	err_msg = "Cannot close environment due to open transactions\n";
        r = toku_ydb_do_error(env, EINVAL, "%s", err_msg);
1036 1037
        goto panic_and_quit_early;
    }
1038 1039 1040 1041
    { //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) {
1042 1043
	    err_msg = "Cannot close environment due to open DBs\n";
            r = toku_ydb_do_error(env, EINVAL, "%s", err_msg);
1044 1045
            goto panic_and_quit_early;
        }
1046 1047 1048 1049 1050
    }
    {
        if (env->i->persistent_environment) {
            r = toku_db_close(env->i->persistent_environment, 0);
            if (r) {
1051 1052
		err_msg = "Cannot close persistent environment dictionary (DB->close error)\n";
                toku_ydb_do_error(env, r, "%s", err_msg);
1053 1054 1055 1056 1057 1058
                goto panic_and_quit_early;
            }
        }
        if (env->i->directory) {
            r = toku_db_close(env->i->directory, 0);
            if (r) {
1059 1060
		err_msg = "Cannot close Directory dictionary (DB->close error)\n";
                toku_ydb_do_error(env, r, "%s", err_msg);
1061 1062 1063 1064
                goto panic_and_quit_early;
            }
        }
    }
1065
    if (env->i->cachetable) {
1066 1067
	toku_ydb_unlock();  // ydb lock must not be held when shutting down minicron
	toku_cachetable_minicron_shutdown(env->i->cachetable);
1068
        if (env->i->logger) {
1069
            r = toku_checkpoint(env->i->cachetable, env->i->logger, NULL, NULL, NULL, NULL);
1070
            if (r) {
1071 1072
		err_msg = "Cannot close environment (error during checkpoint)\n";
                toku_ydb_do_error(env, r, "%s", err_msg);
1073 1074
                goto panic_and_quit_early;
            }
1075 1076 1077 1078
            { //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) {
1079 1080
		    err_msg = "Cannot close environment due to zombie DBs\n";
                    r = toku_ydb_do_error(env, EINVAL, "%s", err_msg);
1081 1082 1083
                    goto panic_and_quit_early;
                }
            }
1084 1085
            r = toku_logger_close_rollback(env->i->logger, FALSE);
            if (r) {
1086 1087
		err_msg = "Cannot close environment (error during closing rollback cachefile)\n";
                toku_ydb_do_error(env, r, "%s", err_msg);
1088 1089 1090 1091 1092
                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) {
1093 1094
		err_msg = "Cannot close environment (error during checkpoint)\n";
                toku_ydb_do_error(env, r, "%s", err_msg);
1095 1096
                goto panic_and_quit_early;
            }
1097 1098
            r = toku_logger_shutdown(env->i->logger); 
            if (r) {
1099 1100
		err_msg = "Cannot close environment (error during logger shutdown)\n";
                toku_ydb_do_error(env, r, "%s", err_msg);
1101 1102
                goto panic_and_quit_early;
            }
1103
        }
1104
	toku_ydb_lock();
1105 1106
        r=toku_cachetable_close(&env->i->cachetable);
	if (r) {
1107 1108
	    err_msg = "Cannot close environment (cachetable close error)\n";
	    toku_ydb_do_error(env, r, "%s", err_msg);
1109
            goto panic_and_quit_early;
1110 1111 1112
	}
    }
    if (env->i->logger) {
1113 1114
        r=toku_logger_close(&env->i->logger);
	if (r) {
1115
	    err_msg = "Cannot close environment (logger close error)\n";
1116
            env->i->logger = NULL;
1117
	    toku_ydb_do_error(env, r, "%s", err_msg);
1118
            goto panic_and_quit_early;
1119 1120 1121 1122
	}
    }
    // 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)
1123 1124 1125 1126
    if (toku_env_is_panicked(env))
        goto panic_and_quit_early;
    else
	assert(env->i->panic_string==0);
1127

1128
    env_fs_destroy(env);
Yoni Fogel's avatar
Yoni Fogel committed
1129
    toku_ltm_close(env->i->ltm);
1130 1131
    if (env->i->data_dir)
        toku_free(env->i->data_dir);
1132 1133
    if (env->i->lg_dir)
        toku_free(env->i->lg_dir);
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
1134 1135
    if (env->i->tmp_dir)
        toku_free(env->i->tmp_dir);
1136 1137 1138 1139
    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);
1140 1141
    if (env->i->real_tmp_dir)
	toku_free(env->i->real_tmp_dir);
1142 1143
    if (env->i->open_dbs)
        toku_omt_destroy(&env->i->open_dbs);
1144 1145
    if (env->i->dir)
	toku_free(env->i->dir);
1146 1147
    //Immediately before freeing internal environment unlock the directories
    unlock_single_process(env);
Bradley C. Kuszmaul's avatar
Bradley C. Kuszmaul committed
1148
    toku_free(env->i);
1149
    env->i = NULL;
Bradley C. Kuszmaul's avatar
Bradley C. Kuszmaul committed
1150
    toku_free(env);
1151
    env = NULL;
1152
    if (flags!=0)
1153 1154 1155 1156
        r = EINVAL;
    return r;

panic_and_quit_early:
Yoni Fogel's avatar
Yoni Fogel committed
1157 1158
    //release lock files.
    unlock_single_process(env);
1159 1160 1161 1162 1163
    //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);
    }
1164 1165 1166
    else {
	env_panic(env, r, err_msg);
    }
1167
    return r;
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
1168
}
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
1169

1170 1171
static int 
toku_env_log_archive(DB_ENV * env, char **list[], u_int32_t flags) {
1172
    return toku_logger_log_archive(env->i->logger, list, flags);
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
1173
}
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
1174

1175 1176
static int 
toku_env_log_flush(DB_ENV * env, const DB_LSN * lsn __attribute__((__unused__))) {
1177
    HANDLE_PANICKED_ENV(env);
1178 1179
    // 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
1180
}
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
1181

1182 1183
static int 
toku_env_set_cachesize(DB_ENV * env, u_int32_t gbytes, u_int32_t bytes, int ncache) {
1184
    HANDLE_PANICKED_ENV(env);
Rich Prohaska's avatar
Rich Prohaska committed
1185 1186
    if (ncache != 1)
        return EINVAL;
Rich Prohaska's avatar
Rich Prohaska committed
1187 1188 1189 1190 1191
    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
1192 1193 1194
    return 0;
}

1195 1196 1197 1198 1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 1210 1211 1212 1213 1214 1215 1216 1217 1218 1219
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
1220 1221
#if DB_VERSION_MAJOR == 4 && DB_VERSION_MINOR >= 3

1222 1223
static int 
toku_env_get_cachesize(DB_ENV * env, u_int32_t *gbytes, u_int32_t *bytes, int *ncache) {
1224
    HANDLE_PANICKED_ENV(env);
Rich Prohaska's avatar
Rich Prohaska committed
1225 1226 1227 1228 1229 1230
    *gbytes = env->i->cachetable_size >> 30;
    *bytes = env->i->cachetable_size & ((1<<30)-1);
    *ncache = 1;
    return 0;
}

1231 1232
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
1233
    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
1234
}
Rich Prohaska's avatar
Rich Prohaska committed
1235 1236
#endif

1237 1238
static int 
toku_env_set_data_dir(DB_ENV * env, const char *dir) {
1239
    HANDLE_PANICKED_ENV(env);
Yoni Fogel's avatar
Yoni Fogel committed
1240 1241
    int r;
    
Rich Prohaska's avatar
Rich Prohaska committed
1242
    if (env_opened(env) || !dir) {
1243
	r = toku_ydb_do_error(env, EINVAL, "You cannot set the data dir after opening the env\n");
1244
    }
1245 1246
    else if (env->i->data_dir)
	r = toku_ydb_do_error(env, EINVAL, "You cannot set the data dir more than once.\n");
1247 1248 1249 1250 1251
    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
1252
        }
1253
        else r = 0;
Yoni Fogel's avatar
Yoni Fogel committed
1254
    }
1255
    return r;
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
1256
}
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
1257

1258 1259
static void 
toku_env_set_errcall(DB_ENV * env, toku_env_errcall_t errcall) {
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
1260
    env->i->errcall = errcall;
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
1261
}
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
1262

1263 1264
static void 
toku_env_set_errfile(DB_ENV*env, FILE*errfile) {
1265 1266 1267
    env->i->errfile = errfile;
}

1268 1269
static void 
toku_env_set_errpfx(DB_ENV * env, const char *errpfx) {
1270
    env->i->errpfx = errpfx;
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
1271
}
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
1272

1273 1274
static int 
toku_env_set_flags(DB_ENV * env, u_int32_t flags, int onoff) {
1275
    HANDLE_PANICKED_ENV(env);
Yoni Fogel's avatar
Yoni Fogel committed
1276 1277 1278 1279 1280 1281

    u_int32_t change = 0;
    if (flags & DB_AUTO_COMMIT) {
        change |=  DB_AUTO_COMMIT;
        flags  &= ~DB_AUTO_COMMIT;
    }
1282
    if (flags != 0 && onoff) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1283
	return toku_ydb_do_error(env, EINVAL, "TokuDB does not (yet) support any nonzero ENV flags other than DB_AUTO_COMMIT\n");
1284
    }
Yoni Fogel's avatar
Yoni Fogel committed
1285 1286
    if   (onoff) env->i->open_flags |=  change;
    else         env->i->open_flags &= ~change;
Rich Prohaska's avatar
Rich Prohaska committed
1287
    return 0;
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
1288
}
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
1289

1290 1291
static int 
toku_env_set_lg_bsize(DB_ENV * env, u_int32_t bsize) {
1292
    HANDLE_PANICKED_ENV(env);
1293
    return toku_logger_set_lg_bsize(env->i->logger, bsize);
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
1294
}
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
1295

1296 1297
static int 
toku_env_set_lg_dir(DB_ENV * env, const char *dir) {
1298
    HANDLE_PANICKED_ENV(env);
Rich Prohaska's avatar
Rich Prohaska committed
1299
    if (env_opened(env)) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1300
	return toku_ydb_do_error(env, EINVAL, "Cannot set log dir after opening the env\n");
1301
    }
1302 1303

    if (env->i->lg_dir) toku_free(env->i->lg_dir);
1304 1305
    if (dir) {
        env->i->lg_dir = toku_strdup(dir);
1306
        if (!env->i->lg_dir) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1307
	    return toku_ydb_do_error(env, ENOMEM, "Out of memory\n");
1308
	}
1309
    }
1310 1311
    else env->i->lg_dir = NULL;
    return 0;
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
1312
}
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
1313

1314 1315
static int 
toku_env_set_lg_max(DB_ENV * env, u_int32_t lg_max) {
1316
    HANDLE_PANICKED_ENV(env);
1317 1318 1319
    return toku_logger_set_lg_max(env->i->logger, lg_max);
}

1320 1321
static int 
toku_env_get_lg_max(DB_ENV * env, u_int32_t *lg_maxp) {
1322 1323
    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
1324
}
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
1325

1326 1327
static int 
toku_env_set_lk_detect(DB_ENV * env, u_int32_t detect) {
1328
    HANDLE_PANICKED_ENV(env);
1329
    detect=detect;
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1330
    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
1331
}
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
1332

1333 1334
static int 
toku_env_set_lk_max_locks(DB_ENV *dbenv, u_int32_t max) {
Yoni Fogel's avatar
Yoni Fogel committed
1335
    int r = ENOSYS;
Yoni Fogel's avatar
Yoni Fogel committed
1336
    HANDLE_PANICKED_ENV(dbenv);
Yoni Fogel's avatar
Yoni Fogel committed
1337
    if (env_opened(dbenv))         { return EINVAL; }
1338
    r = toku_ltm_set_max_locks(dbenv->i->ltm, max);
Yoni Fogel's avatar
Yoni Fogel committed
1339
    return r;
Yoni Fogel's avatar
Yoni Fogel committed
1340 1341
}

1342
#if DB_VERSION_MAJOR == 4 && DB_VERSION_MINOR <= 4
1343 1344
static int 
toku_env_set_lk_max(DB_ENV * env, u_int32_t lk_max) {
Yoni Fogel's avatar
Yoni Fogel committed
1345
    return toku_env_set_lk_max_locks(env, lk_max);
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
1346
}
Rich Prohaska's avatar
Rich Prohaska committed
1347

1348 1349
static int 
locked_env_set_lk_max(DB_ENV * env, u_int32_t lk_max) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1350
    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
1351
}
1352
#endif
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
1353

1354 1355
static int 
toku_env_get_lk_max_locks(DB_ENV *dbenv, u_int32_t *lk_maxp) {
1356
    HANDLE_PANICKED_ENV(dbenv);
1357
    return toku_ltm_get_max_locks(dbenv->i->ltm, lk_maxp);
1358 1359
}

1360 1361
static int 
locked_env_set_lk_max_locks(DB_ENV *dbenv, u_int32_t max) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1362
    toku_ydb_lock(); int r = toku_env_set_lk_max_locks(dbenv, max); toku_ydb_unlock(); return r;
1363 1364
}

1365 1366
static int 
locked_env_get_lk_max_locks(DB_ENV *dbenv, u_int32_t *lk_maxp) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1367
    toku_ydb_lock(); int r = toku_env_get_lk_max_locks(dbenv, lk_maxp); toku_ydb_unlock(); return r;
1368 1369
}

1370 1371
static int 
toku_env_set_lk_max_memory(DB_ENV *dbenv, uint64_t max) {
1372 1373 1374 1375 1376 1377 1378
    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;
}

1379 1380
static int 
toku_env_get_lk_max_memory(DB_ENV *dbenv, uint64_t *lk_maxp) {
1381 1382 1383 1384
    HANDLE_PANICKED_ENV(dbenv);
    return toku_ltm_get_max_lock_memory(dbenv->i->ltm, lk_maxp);
}

1385 1386 1387 1388 1389 1390
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;
1391 1392 1393 1394 1395 1396
}

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
1397
//void toku__env_set_noticecall (DB_ENV *env, void (*noticecall)(DB_ENV *, db_notices)) {
Bradley C. Kuszmaul's avatar
Fixup  
Bradley C. Kuszmaul committed
1398 1399
//    env->i->noticecall = noticecall;
//}
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
1400

1401 1402
static int 
toku_env_set_tmp_dir(DB_ENV * env, const char *tmp_dir) {
1403
    HANDLE_PANICKED_ENV(env);
Rich Prohaska's avatar
Rich Prohaska committed
1404
    if (env_opened(env)) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1405
	return toku_ydb_do_error(env, EINVAL, "Cannot set the tmp dir after opening an env\n");
1406 1407
    }
    if (!tmp_dir) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1408
	return toku_ydb_do_error(env, EINVAL, "Tmp dir bust be non-null\n");
1409
    }
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
1410 1411
    if (env->i->tmp_dir)
        toku_free(env->i->tmp_dir);
Yoni Fogel's avatar
Yoni Fogel committed
1412
    env->i->tmp_dir = toku_strdup(tmp_dir);
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
1413
    return env->i->tmp_dir ? 0 : ENOMEM;
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
1414
}
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
1415

1416 1417
static int 
toku_env_set_verbose(DB_ENV * env, u_int32_t which, int onoff) {
1418 1419
    HANDLE_PANICKED_ENV(env);
    which=which; onoff=onoff;
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
1420
    return 1;
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
1421
}
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
1422

1423
// For test purposes only.
1424 1425
// These callbacks are never used in production code, only as a way to test the system
// (for example, by causing crashes at predictable times).
1426 1427
static void (*checkpoint_callback_f)(void*) = NULL;
static void * checkpoint_callback_extra     = NULL;
1428 1429
static void (*checkpoint_callback2_f)(void*) = NULL;
static void * checkpoint_callback2_extra     = NULL;
1430

1431 1432
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__))) {
1433
    int r = toku_checkpoint(env->i->cachetable, env->i->logger,
1434 1435
			    checkpoint_callback_f,  checkpoint_callback_extra,
			    checkpoint_callback2_f, checkpoint_callback2_extra);
1436
    if (r) {
1437 1438
	// Panicking the whole environment may be overkill, but I'm not sure what else to do.
	env_panic(env, r, "checkpoint error\n");
1439
        toku_ydb_do_error(env, r, "Checkpoint\n");
1440 1441
    }
    return r;
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
1442 1443
}

1444 1445
static int 
toku_env_txn_stat(DB_ENV * env, DB_TXN_STAT ** statp, u_int32_t flags) {
1446 1447
    HANDLE_PANICKED_ENV(env);
    statp=statp;flags=flags;
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
1448
    return 1;
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
1449 1450
}

1451 1452
static int 
locked_env_open(DB_ENV * env, const char *home, u_int32_t flags, int mode) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1453
    toku_ydb_lock(); int r = toku_env_open(env, home, flags, mode); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
1454 1455
}

1456 1457
static int 
locked_env_close(DB_ENV * env, u_int32_t flags) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1458
    toku_ydb_lock(); int r = toku_env_close(env, flags); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
1459 1460
}

1461 1462
static int 
locked_env_log_archive(DB_ENV * env, char **list[], u_int32_t flags) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1463
    toku_ydb_lock(); int r = toku_env_log_archive(env, list, flags); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
1464 1465
}

1466 1467
static int 
locked_env_log_flush(DB_ENV * env, const DB_LSN * lsn) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1468
    toku_ydb_lock(); int r = toku_env_log_flush(env, lsn); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
1469 1470
}

1471 1472
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
1473
    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
1474 1475
}

1476 1477
static int 
locked_env_set_data_dir(DB_ENV * env, const char *dir) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1478
    toku_ydb_lock(); int r = toku_env_set_data_dir(env, dir); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
1479 1480
}

1481 1482
static int 
locked_env_set_flags(DB_ENV * env, u_int32_t flags, int onoff) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1483
    toku_ydb_lock(); int r = toku_env_set_flags(env, flags, onoff); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
1484 1485
}

1486 1487
static int 
locked_env_set_lg_bsize(DB_ENV * env, u_int32_t bsize) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1488
    toku_ydb_lock(); int r = toku_env_set_lg_bsize(env, bsize); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
1489 1490
}

1491 1492
static int 
locked_env_set_lg_dir(DB_ENV * env, const char *dir) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1493
    toku_ydb_lock(); int r = toku_env_set_lg_dir(env, dir); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
1494 1495
}

1496 1497
static int 
locked_env_set_lg_max(DB_ENV * env, u_int32_t lg_max) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1498
    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
1499 1500
}

1501 1502
static int 
locked_env_get_lg_max(DB_ENV * env, u_int32_t *lg_maxp) {
1503 1504 1505
    toku_ydb_lock(); int r = toku_env_get_lg_max(env, lg_maxp); toku_ydb_unlock(); return r;
}

1506 1507
static int 
locked_env_set_lk_detect(DB_ENV * env, u_int32_t detect) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1508
    toku_ydb_lock(); int r = toku_env_set_lk_detect(env, detect); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
1509 1510
}

1511 1512
static int 
locked_env_set_tmp_dir(DB_ENV * env, const char *tmp_dir) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1513
    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
1514 1515
}

1516 1517
static int 
locked_env_set_verbose(DB_ENV * env, u_int32_t which, int onoff) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1518
    toku_ydb_lock(); int r = toku_env_set_verbose(env, which, onoff); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
1519 1520
}

1521 1522
static int 
locked_env_txn_stat(DB_ENV * env, DB_TXN_STAT ** statp, u_int32_t flags) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
1523
    toku_ydb_lock(); int r = toku_env_txn_stat(env, statp, flags); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
1524 1525
}

1526 1527 1528 1529 1530 1531 1532 1533 1534 1535 1536 1537 1538 1539 1540
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;
}

Dave Wells's avatar
Dave Wells committed
1541 1542 1543 1544 1545 1546 1547 1548 1549 1550 1551 1552 1553 1554 1555
static int
locked_env_create_indexer(DB_ENV *env,
                          DB_TXN *txn,
                          DB_INDEXER **indexerp,
                          DB *src_db,
                          int N,
                          DB *dest_dbs[N],
                          uint32_t db_flags[N],
                          uint32_t indexer_flags) {
    toku_ydb_lock();
    int r = toku_indexer_create_indexer(env, txn, indexerp, src_db, N, dest_dbs, db_flags, indexer_flags);
    toku_ydb_unlock();
    return r;
}

1556 1557 1558 1559 1560 1561 1562 1563 1564 1565 1566 1567 1568 1569 1570 1571 1572
static int
locked_env_create_loader(DB_ENV *env,
                         DB_TXN *txn, 
                         DB_LOADER **blp, 
                         DB *src_db, 
                         int N, 
                         DB *dbs[], 
                         uint32_t db_flags[N], 
                         uint32_t dbt_flags[N], 
                         uint32_t loader_flags) {
    toku_ydb_lock();
    int r = toku_loader_create_loader(env, txn, blp, src_db, N, dbs, db_flags, dbt_flags, loader_flags);
    toku_ydb_unlock();
    return r;
}


Dave Wells's avatar
Dave Wells committed
1573

1574 1575 1576 1577 1578 1579 1580 1581 1582 1583 1584 1585 1586 1587 1588
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;
}

1589 1590 1591 1592 1593 1594 1595 1596 1597 1598 1599 1600 1601 1602 1603 1604 1605 1606 1607 1608 1609 1610 1611 1612 1613 1614 1615 1616 1617 1618 1619 1620 1621 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_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;
}

1644 1645 1646 1647 1648 1649 1650 1651 1652 1653 1654 1655 1656 1657
static void
env_set_update (DB_ENV *env, int (*update_function)(DB *, const DBT *key, const DBT *old_val, const DBT *extra, void (*set_val)(const DBT *new_val, void *set_extra), void *set_extra)) {
    env->i->update_function = update_function;
}

static void
locked_env_set_update (DB_ENV *env, int (*update_function)(DB *, const DBT *key, const DBT *old_val, const DBT *extra, void (*set_val)(const DBT *new_val, void *set_extra), void *set_extra)) {
    toku_ydb_lock();
    env_set_update (env, update_function);
    toku_ydb_unlock();
}



1658
static int
1659
env_set_generate_row_callback_for_put(DB_ENV *env, generate_row_for_put_func generate_row_for_put) {
1660 1661 1662 1663
    HANDLE_PANICKED_ENV(env);
    int r = 0;
    if (env_opened(env)) r = EINVAL;
    else {
1664
        env->i->generate_row_for_put = generate_row_for_put;
1665 1666 1667 1668 1669
    }
    return r;
}

static int
1670 1671 1672 1673 1674 1675 1676 1677 1678 1679 1680 1681
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) {
1682
    toku_ydb_lock();
1683
    int r = env_set_generate_row_callback_for_put(env, generate_row_for_put);
1684 1685 1686 1687
    toku_ydb_unlock();
    return r;
}

1688 1689 1690 1691 1692 1693 1694 1695
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;
}

1696
static int env_put_multiple(DB_ENV *env, DB *src_db, DB_TXN *txn, 
1697
                            const DBT *src_key, const DBT *src_val, 
1698
                            uint32_t num_dbs, DB **db_array, DBT *keys, DBT *vals, uint32_t *flags_array);
1699 1700

static int env_del_multiple(DB_ENV *env, DB *src_db, DB_TXN *txn, 
1701
                            const DBT *src_key, const DBT *src_val, 
1702
                            uint32_t num_dbs, DB **db_array, DBT *keys, uint32_t *flags_array);
1703 1704

static int env_update_multiple(DB_ENV *env, DB *src_db, DB_TXN *txn, 
1705 1706 1707
                               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, 
1708
                               uint32_t num_keys, DBT *keys, 
1709
                               uint32_t num_vals, DBT *vals);
1710 1711

static int
1712
locked_env_put_multiple(DB_ENV *env, DB *src_db, DB_TXN *txn, const DBT *src_key, const DBT *src_val, uint32_t num_dbs, DB **db_array, DBT *keys, DBT *vals, uint32_t *flags_array) {
1713 1714 1715
    int r = env_check_avail_fs_space(env);
    if (r == 0) {
	toku_ydb_lock();
1716
	r = env_put_multiple(env, src_db, txn, src_key, src_val, num_dbs, db_array, keys, vals, flags_array);
1717 1718
	toku_ydb_unlock();
    }
1719 1720 1721 1722
    return r;
}

static int
1723
locked_env_del_multiple(DB_ENV *env, DB *src_db, DB_TXN *txn, const DBT *src_key, const DBT *src_val, uint32_t num_dbs, DB **db_array, DBT *keys, uint32_t *flags_array) {
1724
    toku_ydb_lock();
1725
    int r = env_del_multiple(env, src_db, txn, src_key, src_val, num_dbs, db_array, keys, flags_array);
1726 1727 1728 1729
    toku_ydb_unlock();
    return r;
}

1730 1731
static int
locked_env_update_multiple(DB_ENV *env, DB *src_db, DB_TXN *txn,                                
1732 1733 1734
                           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, 
1735
                           uint32_t num_keys, DBT *keys, 
1736
                           uint32_t num_vals, DBT *vals) {
1737
    toku_ydb_lock();
1738
    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);
1739 1740 1741
    toku_ydb_unlock();
    return r;
}
1742

1743 1744 1745 1746 1747 1748 1749 1750 1751 1752 1753 1754 1755 1756 1757 1758 1759 1760 1761 1762 1763
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;
}

1764
static int
1765 1766
env_get_lock_timeout(DB_ENV *env, uint64_t *lock_timeout_msec) {
    toku_ltm_get_lock_wait_time(env->i->ltm, lock_timeout_msec);
1767 1768 1769 1770
    return 0;
}

static int
1771
locked_env_get_lock_timeout(DB_ENV *env, uint64_t *lock_timeout_msec) {
1772
    toku_ydb_lock();
1773
    int r = env_get_lock_timeout(env, lock_timeout_msec);
1774 1775 1776 1777 1778
    toku_ydb_unlock();
    return r;
}

static int
1779 1780
env_set_lock_timeout(DB_ENV *env, uint64_t lock_timeout_msec) {
    toku_ltm_set_lock_wait_time(env->i->ltm, lock_timeout_msec);
1781 1782 1783 1784
    return 0;
}

static int
1785
locked_env_set_lock_timeout(DB_ENV *env, uint64_t lock_timeout_msec) {
1786
    toku_ydb_lock();
1787
    int r = env_set_lock_timeout(env, lock_timeout_msec);
1788 1789 1790 1791
    toku_ydb_unlock();
    return r;
}

1792 1793 1794 1795 1796 1797 1798 1799 1800 1801 1802 1803 1804 1805 1806 1807
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];
    }
}
1808

1809 1810
// 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
1811 1812 1813
// can help diagnose the problem.
// This function only collects information, and it does not matter if something gets garbled
// because of a race condition.  
1814
// Note, engine status is still collected even if the environment or logger is panicked
1815
static int
1816 1817 1818
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) {
1819
	if (env && env->i && env->i->is_panicked && env->i->panic_string) {
1820
	    strncpy(env_panic_string_buf, env->i->panic_string, env_panic_string_length);
1821 1822
	    env_panic_string_buf[env_panic_string_length - 1] = '\0';  // just in case
	}
1823 1824 1825 1826
	else 
	    *env_panic_string_buf = '\0';
    }

1827 1828 1829 1830
    if ( !(env)     || 
	 !(env->i)  || 
	 !(env_opened(env)) )
	r = EINVAL;
1831
    else {
1832 1833
	r = 0;
	engstat->env_panic = env->i->is_panicked;
1834
	format_time(&persistent_creation_time, engstat->creationtime);
1835 1836
	time_t now = time(NULL);
        format_time(&now, engstat->now);
1837
        format_time(&startuptime, engstat->startuptime);
1838 1839 1840
	{
	    SCHEDULE_STATUS_S schedstat;
	    toku_ydb_lock_get_status(&schedstat);
1841 1842 1843 1844 1845 1846 1847 1848
	    engstat->ydb_lock_ctr             = schedstat.ydb_lock_ctr;             /* How many times has ydb lock been taken/released?                                                                      */ 
	    engstat->num_waiters_now          = schedstat.num_waiters_now;          /* How many are waiting on on the ydb lock right now (including the current lock holder, if any)?                        */
	    engstat->max_waiters              = schedstat.max_waiters;              /* The maxium of num_waiters_now (since the system booted).                                                              */ 
	    engstat->total_sleep_time         = schedstat.total_sleep_time;         /* The total time spent (since the system booted) sleeping (by the indexer) to give foreground threads a chance to work .*/ 
	    engstat->max_time_ydb_lock_held   = schedstat.max_time_ydb_lock_held;   /* Maximum time that the ydb lock was held.                                                                              */ 
	    engstat->total_time_ydb_lock_held = schedstat.total_time_ydb_lock_held; /* Total time client threads held the ydb lock                                                                           */ 
	    engstat->total_time_since_start   = schedstat.total_time_since_start;   /* Total time since the lock was created.  Use this as total_time_ydb_lock_held/total_time_since_start to get a ratio.   */

1849
	}
1850 1851 1852 1853 1854 1855 1856 1857
        {
	    LE_STATUS_S lestat;                    // Rice's vampire
	    toku_le_get_status(&lestat);
            engstat->le_max_committed_xr    = lestat.max_committed_xr;
            engstat->le_max_provisional_xr  = lestat.max_provisional_xr;
            engstat->le_expanded            = lestat.expanded;
            engstat->le_max_memsize         = lestat.max_memsize;
        }
1858
	engstat->checkpoint_period = toku_get_checkpoint_period_unlocked(env->i->cachetable);  // do not take any locks (not even minicron lock)
1859 1860 1861 1862 1863 1864 1865
	{
            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);
1866 1867 1868 1869 1870 1871 1872 1873 1874 1875 1876 1877 1878
	    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;
1879
                time_t   oldest_starttime = 0;
1880 1881 1882
		uint64_t next_lsn   = 0;
		TOKULOGGER logger = env->i->logger;
		if (logger) {
1883
		    oldest_xid = toku_logger_get_oldest_living_xid(env->i->logger, &oldest_starttime);
1884 1885 1886 1887
		    next_lsn   = (toku_logger_get_next_lsn(env->i->logger)).lsn;
		}
		engstat->txn_oldest_live = oldest_xid;
		engstat->next_lsn = next_lsn;
1888
                format_time(&oldest_starttime, engstat->txn_oldest_live_starttime);
1889
	    }
1890
	}
1891 1892 1893
	{
	    CACHETABLE_STATUS_S ctstat;
	    toku_cachetable_get_status(env->i->cachetable, &ctstat);
1894 1895 1896 1897 1898 1899 1900 1901
	    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;
1902
	    engstat->cachetable_wait_checkpoint = ctstat.wait_checkpoint;
1903 1904 1905 1906 1907 1908
	    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;
1909
	    engstat->cachetable_size_max      = ctstat.size_max;
1910 1911
	    engstat->cachetable_size_writing  = ctstat.size_writing;
	    engstat->get_and_pin_footprint    = ctstat.get_and_pin_footprint;
1912 1913 1914
	    engstat->local_checkpoint         = ctstat.local_checkpoint;
	    engstat->local_checkpoint_files   = ctstat.local_checkpoint_files;
	    engstat->local_checkpoint_during_checkpoint = ctstat.local_checkpoint_during_checkpoint;
1915
            engstat->cachetable_evictions     = ctstat.evictions;
1916
	}
1917 1918
	{
	    toku_ltm* ltm = env->i->ltm;
1919
	    LTM_STATUS_S ltmstat;
1920 1921 1922 1923 1924
	    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);
1925 1926
	    engstat->range_locks_max                 = max_locks;
	    engstat->range_locks_curr                = curr_locks;
1927 1928
	    engstat->range_locks_max_memory          = max_lock_memory;
	    engstat->range_locks_curr_memory         = curr_lock_memory;
1929 1930
	    engstat->range_lock_escalation_successes = ltmstat.lock_escalation_successes;
	    engstat->range_lock_escalation_failures  = ltmstat.lock_escalation_failures;
1931 1932 1933 1934 1935 1936
	    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;
1937
	}
1938
	{
1939
     	    engstat->inserts            = num_inserts;
1940
	    engstat->inserts_fail       = num_inserts_fail;
1941
	    engstat->deletes            = num_deletes;
1942
	    engstat->deletes_fail       = num_deletes_fail;
1943 1944
	    engstat->updates            = num_updates;
	    engstat->updates_fail       = num_updates_fail;
1945 1946
	    engstat->updates_broadcast  = num_updates_broadcast;
	    engstat->updates_broadcast_fail  = num_updates_broadcast_fail;
1947 1948 1949 1950 1951 1952
     	    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;
1953 1954
	    engstat->point_queries      = num_point_queries;
	    engstat->sequential_queries = num_sequential_queries;
1955 1956 1957 1958
            engstat->directory_read_locks = directory_read_locks;
            engstat->directory_read_locks_fail = directory_read_locks_fail;
            engstat->directory_write_locks = directory_write_locks;
            engstat->directory_write_locks_fail = directory_write_locks_fail;
1959
	}
1960
	{
1961 1962 1963 1964 1965
	    BRT_STATUS_S brt_stat;
	    toku_brt_get_status(&brt_stat);
	    engstat->le_updates = brt_stat.updates;
	    engstat->le_updates_broadcast = brt_stat.updates_broadcast;
	    engstat->descriptor_set = brt_stat.descriptor_set;
1966 1967 1968 1969 1970 1971
	    engstat->partial_fetch_hit = brt_stat.partial_fetch_hit;
	    engstat->partial_fetch_miss = brt_stat.partial_fetch_miss;
	    engstat->partial_fetch_compressed = brt_stat.partial_fetch_compressed;
	    engstat->msn_discards = brt_stat.msn_discards;
	    engstat->max_workdone = brt_stat.max_workdone;
	    engstat->dsn_gap = brt_stat.dsn_gap;	    
1972 1973
	    engstat->cachetable_size_leaf = brt_stat.bytes_leaf;
	    engstat->cachetable_size_nonleaf = brt_stat.bytes_nonleaf;
1974
	}
1975 1976 1977 1978 1979 1980
	{
	    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;
	}
1981 1982 1983 1984 1985 1986 1987
	{
	    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;
1988 1989
	    engstat->logger_panic     = log_stat.panicked;
	    engstat->logger_panic_errno = log_stat.panic_errno;
1990
	}
1991 1992
	{
	    time_t    enospc_most_recent_timestamp;
1993 1994
	    u_int64_t enospc_threads_blocked, enospc_ctr;
	    toku_fs_get_write_info(&enospc_most_recent_timestamp, &enospc_threads_blocked, &enospc_ctr);
1995 1996
	    format_time(&enospc_most_recent_timestamp, engstat->enospc_most_recent);	    
	    engstat->enospc_threads_blocked = enospc_threads_blocked;
1997
	    engstat->enospc_ctr = enospc_ctr;
1998
	}
1999
	{
2000 2001 2002 2003 2004 2005 2006 2007 2008
	    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;
2009
	    engstat->loader_put_fail       = loader_stat.put_fail;
2010 2011 2012 2013 2014 2015 2016 2017
	    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;
2018
	}
2019 2020 2021 2022 2023 2024 2025 2026 2027 2028 2029 2030 2031
	{
	    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;
	}
2032
	{
2033 2034
	    BRT_UPGRADE_STATUS_S brt_upgrade_stat;
	    toku_brt_get_upgrade_status(&brt_upgrade_stat);
2035 2036 2037 2038 2039
	    uint64_t upgrade_footprint  = toku_log_upgrade_get_footprint();
	    // Footprint of upgrade maybe performed for this time environment is opened
	    // is provided in six least significant decimal digits, footprint of 
	    // upgrade performed when environment was actually upgraded is provided
	    // in most significant decimal digits.
2040
	    // If ver_at_startup == 13, then the footprint will have the same value in 
2041
	    // upper and lower digits.
2042 2043 2044 2045 2046
	    engstat->upgrade_env_status = (persistent_upgrade_v14_footprint * 1000000) + upgrade_footprint;
	    engstat->upgrade_header     = brt_upgrade_stat.header_13;
	    engstat->upgrade_nonleaf    = brt_upgrade_stat.nonleaf_13;
	    engstat->upgrade_leaf       = brt_upgrade_stat.leaf_13;
	    engstat->optimized_for_upgrade = brt_upgrade_stat.optimized_for_upgrade;
2047 2048
	    engstat->original_ver       = persistent_original_env_version;
	    engstat->ver_at_startup     = persistent_stored_env_version_at_startup;
2049 2050
	    engstat->last_lsn_v13       = persistent_last_lsn_of_v13;
	    format_time(&persistent_upgrade_v14_time, engstat->upgrade_v14_time);
2051
	}
2052 2053 2054
	{
	    MEMORY_STATUS_S memory_status;
	    toku_memory_get_status(&memory_status);
2055 2056 2057 2058 2059 2060 2061 2062 2063
	    engstat->malloc_count   = memory_status.malloc_count;
	    engstat->free_count     = memory_status.free_count;
	    engstat->realloc_count  = memory_status.realloc_count;
	    engstat->malloc_fail    = memory_status.malloc_fail;
	    engstat->realloc_fail   = memory_status.realloc_fail;
	    engstat->mem_requested  = memory_status.requested;
	    engstat->mem_used       = memory_status.used;
	    engstat->mem_freed      = memory_status.freed;
	    engstat->max_mem_in_use = memory_status.max_in_use;
2064
	}
2065 2066 2067 2068
    }
    return r;
}

2069

2070
// Fill buff with text description of engine status up to bufsiz bytes.
2071 2072
// 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.
2073 2074 2075
static int
env_get_engine_status_text(DB_ENV * env, char * buff, int bufsiz) {
    ENGINE_STATUS engstat;
2076
    uint32_t stringsize = 1024;
2077
    char panicstring[stringsize];
2078 2079
    int n = 0;  // number of characters printed so far

2080 2081 2082

    n = snprintf(buff, bufsiz - n, "BUILD_ID = %d\n", BUILD_ID);

2083 2084 2085 2086 2087 2088 2089
    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);
    }

2090 2091 2092 2093 2094 2095 2096 2097 2098 2099 2100 2101 2102
    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 {
2103
	n += snprintf(buff + n, bufsiz - n, "env panic                        %"PRIu64"\n", engstat.env_panic);
2104 2105 2106 2107
	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);
2108 2109
	n += snprintf(buff + n, bufsiz - n, "num_waiters_now                  %"PRIu32"\n", engstat.num_waiters_now);
	n += snprintf(buff + n, bufsiz - n, "max_waiters                      %"PRIu32"\n", engstat.max_waiters);
2110
	n += snprintf(buff + n, bufsiz - n, "total_sleep_time                 %"PRIu64"\n", engstat.total_sleep_time);
2111 2112
	n += snprintf(buff + n, bufsiz - n, "max_time_ydb_lock_held           %.6f\n",      tokutime_to_seconds(engstat.max_time_ydb_lock_held));
	n += snprintf(buff + n, bufsiz - n, "total_time_ydb_lock_held         %.6f\n",      tokutime_to_seconds(engstat.total_time_ydb_lock_held));
2113
	n += snprintf(buff + n, bufsiz - n, "total_time_since_start           %"PRIu64"\n", engstat.total_time_since_start);
2114 2115 2116 2117
	n += snprintf(buff + n, bufsiz - n, "le_max_committed_xr              %"PRIu64"\n", engstat.le_max_committed_xr);
	n += snprintf(buff + n, bufsiz - n, "le_max_provisional_xr            %"PRIu64"\n", engstat.le_max_provisional_xr);
	n += snprintf(buff + n, bufsiz - n, "le_expanded                      %"PRIu64"\n", engstat.le_expanded);
	n += snprintf(buff + n, bufsiz - n, "le_max_memsize                   %"PRIu64"\n", engstat.le_max_memsize);
2118 2119 2120 2121 2122 2123 2124 2125 2126 2127 2128 2129 2130 2131 2132 2133 2134 2135 2136 2137 2138 2139
	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);
2140
	n += snprintf(buff + n, bufsiz - n, "cachetable_evictions             %"PRIu64"\n", engstat.cachetable_evictions);
2141 2142 2143 2144 2145 2146
	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);
2147
	n += snprintf(buff + n, bufsiz - n, "cachetable_size_max              %"PRId64"\n", engstat.cachetable_size_max);
2148 2149
	n += snprintf(buff + n, bufsiz - n, "cachetable_size_leaf             %"PRIu64"\n", engstat.cachetable_size_leaf);
	n += snprintf(buff + n, bufsiz - n, "cachetable_size_nonleaf          %"PRIu64"\n", engstat.cachetable_size_nonleaf);
2150 2151 2152 2153 2154 2155 2156 2157 2158 2159 2160 2161 2162 2163 2164 2165 2166 2167 2168 2169 2170
	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);
2171 2172
	n += snprintf(buff + n, bufsiz - n, "updates                          %"PRIu64"\n", engstat.updates);
	n += snprintf(buff + n, bufsiz - n, "updates_fail                     %"PRIu64"\n", engstat.updates_fail);
2173 2174 2175 2176
	n += snprintf(buff + n, bufsiz - n, "updates_broadcast                %"PRIu64"\n", engstat.updates_broadcast);
	n += snprintf(buff + n, bufsiz - n, "updates_broadcast_fail           %"PRIu64"\n", engstat.updates_broadcast_fail);
	n += snprintf(buff + n, bufsiz - n, "le_updates                       %"PRIu64"\n", engstat.le_updates);
	n += snprintf(buff + n, bufsiz - n, "le_updates_broadcast             %"PRIu64"\n", engstat.le_updates_broadcast);
2177
	n += snprintf(buff + n, bufsiz - n, "descriptor_set                   %"PRIu64"\n", engstat.descriptor_set);
2178 2179 2180 2181 2182 2183
	n += snprintf(buff + n, bufsiz - n, "partial_fetch_hit                %"PRIu64"\n", engstat.partial_fetch_hit);
	n += snprintf(buff + n, bufsiz - n, "partial_fetch_miss               %"PRIu64"\n", engstat.partial_fetch_miss);
	n += snprintf(buff + n, bufsiz - n, "partial_fetch_compressed         %"PRIu64"\n", engstat.partial_fetch_compressed);
	n += snprintf(buff + n, bufsiz - n, "msn_discards                     %"PRIu64"\n", engstat.msn_discards);
	n += snprintf(buff + n, bufsiz - n, "max_workdone                     %"PRIu64"\n", engstat.max_workdone);
	n += snprintf(buff + n, bufsiz - n, "dsn_gap                          %"PRIu64"\n", engstat.dsn_gap);
2184 2185 2186 2187 2188 2189
	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);
2190 2191
	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);
2192 2193 2194 2195
	n += snprintf(buff + n, bufsiz - n, "directory_read_locks             %"PRIu64"\n", engstat.directory_read_locks);
	n += snprintf(buff + n, bufsiz - n, "directory_read_locks_fail        %"PRIu64"\n", engstat.directory_read_locks_fail);
	n += snprintf(buff + n, bufsiz - n, "directory_write_locks            %"PRIu64"\n", engstat.directory_write_locks);
	n += snprintf(buff + n, bufsiz - n, "directory_write_locks_fail       %"PRIu64"\n", engstat.directory_write_locks_fail);
2196 2197 2198 2199 2200
	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);
2201 2202
	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);
2203 2204 2205 2206 2207 2208
	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);
2209
	n += snprintf(buff + n, bufsiz - n, "loader_create_fail               %"PRIu64"\n", engstat.loader_create_fail);
2210
	n += snprintf(buff + n, bufsiz - n, "loader_put                       %"PRIu64"\n", engstat.loader_put);
2211
	n += snprintf(buff + n, bufsiz - n, "loader_put_fail                  %"PRIu64"\n", engstat.loader_put_fail);
2212 2213 2214 2215 2216 2217 2218
	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);
2219 2220 2221 2222 2223 2224 2225 2226 2227
	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);
2228 2229 2230 2231
	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);
2232
	n += snprintf(buff + n, bufsiz - n, "optimized_for_upgrade            %"PRIu64"\n", engstat.optimized_for_upgrade);
2233 2234
	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);
2235 2236
	n += snprintf(buff + n, bufsiz - n, "last_lsn_v13                     %"PRIu64"\n", engstat.last_lsn_v13);
	n += snprintf(buff + n, bufsiz - n, "upgrade_v14_time                 %s \n", engstat.upgrade_v14_time);
2237 2238 2239
	n += snprintf(buff + n, bufsiz - n, "malloc_count                     %"PRIu64"\n", engstat.malloc_count);
	n += snprintf(buff + n, bufsiz - n, "free_count                       %"PRIu64"\n", engstat.free_count);
	n += snprintf(buff + n, bufsiz - n, "realloc_count                    %"PRIu64"\n", engstat.realloc_count);
2240 2241
	n += snprintf(buff + n, bufsiz - n, "malloc_fail                      %"PRIu64"\n", engstat.malloc_fail);
	n += snprintf(buff + n, bufsiz - n, "realloc_fail                     %"PRIu64"\n", engstat.realloc_fail);
2242 2243 2244
	n += snprintf(buff + n, bufsiz - n, "mem_requested                    %"PRIu64"\n", engstat.mem_requested);
	n += snprintf(buff + n, bufsiz - n, "mem_used                         %"PRIu64"\n", engstat.mem_used);
	n += snprintf(buff + n, bufsiz - n, "mem_freed                        %"PRIu64"\n", engstat.mem_freed);
2245
	n += snprintf(buff + n, bufsiz - n, "max_mem_in_use                   %"PRIu64"\n", engstat.max_mem_in_use);
2246
    }
2247 2248 2249
    if (n > bufsiz) {
	char * errmsg = "BUFFER TOO SMALL\n";
	int len = strlen(errmsg) + 1;
2250
	(void) snprintf(buff + (bufsiz - 1) - len, len, "%s", errmsg);
2251 2252 2253 2254 2255
    }

    return r;
}

2256 2257 2258 2259
// 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;
2260 2261 2262 2263 2264 2265 2266 2267
    int r;
    if (engine_status_enable) {
	r = env_get_engine_status_text(env, buff, buffsize);
    }
    else {
	r = ENODATA;
	snprintf(buff, buffsize, "Engine status not available: disabled by user.  This should only happen in test programs.\n");
    }
2268 2269 2270
    return r;
}

2271 2272 2273 2274
// Set panic code and panic string if not already panicked,
// intended for use by toku_assert when about to abort().
static void 
toku_maybe_set_env_panic(int code, char * msg) {
2275 2276 2277 2278 2279
    if (code == 0) 
	code = -1;
    if (msg == NULL)
	msg = "Unknown cause from abort (failed assert)\n";
    env_is_panicked = code;  // disable library destructor no matter what
2280
    DB_ENV * env = most_recent_env;
2281 2282 2283 2284
    if (env && 
	env->i &&
	(env->i->is_panicked == 0)) {
	env_panic(env, code, msg);
2285 2286
    }
}
2287 2288


Rich Prohaska's avatar
Rich Prohaska committed
2289 2290
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
2291 2292 2293 2294
static int toku_db_lt_panic(DB* db, int r);

static toku_dbt_cmp toku_db_get_compare_fun(DB* db);

2295 2296
static int 
toku_env_create(DB_ENV ** envp, u_int32_t flags) {
Yoni Fogel's avatar
Yoni Fogel committed
2297 2298 2299
    int r = ENOSYS;
    DB_ENV* result = NULL;

2300 2301
    engine_status_enable = 1;

Yoni Fogel's avatar
Yoni Fogel committed
2302 2303 2304
    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
2305
    memset(result, 0, sizeof *result);
2306
    result->err = (void (*)(const DB_ENV * env, int error, const char *fmt, ...)) toku_locked_env_err;
2307 2308 2309 2310
#define SENV(name) result->name = locked_env_ ## name
    SENV(dbremove);
    SENV(dbrename);
    SENV(set_default_bt_compare);
2311
    SENV(set_update);
2312 2313
    SENV(set_generate_row_callback_for_put);
    SENV(set_generate_row_callback_for_del);
2314 2315
    SENV(put_multiple);
    SENV(del_multiple);
2316
    SENV(update_multiple);
2317 2318
    SENV(checkpointing_set_period);
    SENV(checkpointing_get_period);
2319 2320 2321 2322
    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;
2323
    result->get_engine_status = env_get_engine_status;
2324
    result->get_engine_status_text = env_get_engine_status_text;
2325 2326 2327
    result->get_iname = env_get_iname;
    SENV(open);
    SENV(close);
2328
    result->txn_checkpoint = toku_env_txn_checkpoint;
2329
    SENV(log_flush);
Rich Prohaska's avatar
Rich Prohaska committed
2330 2331 2332
    result->set_errcall = toku_env_set_errcall;
    result->set_errfile = toku_env_set_errfile;
    result->set_errpfx = toku_env_set_errpfx;
2333 2334 2335 2336 2337 2338 2339 2340 2341 2342 2343
    //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);
2344 2345
    SENV(set_lk_max_memory);
    SENV(get_lk_max_memory);
2346
    SENV(set_cachesize);
Rich Prohaska's avatar
Rich Prohaska committed
2347
#if DB_VERSION_MAJOR == 4 && DB_VERSION_MINOR >= 3
2348
    SENV(get_cachesize);
Rich Prohaska's avatar
Rich Prohaska committed
2349
#endif
2350
    SENV(set_lk_detect);
2351
#if DB_VERSION_MAJOR == 4 && DB_VERSION_MINOR <= 4
2352
    SENV(set_lk_max);
2353
#endif
2354 2355
    SENV(log_archive);
    SENV(txn_stat);
Rich Prohaska's avatar
Rich Prohaska committed
2356
    result->txn_begin = locked_txn_begin;
2357
    SENV(set_redzone);
Dave Wells's avatar
Dave Wells committed
2358
    SENV(create_indexer);
2359
    SENV(create_loader);
2360 2361
    SENV(get_lock_timeout);
    SENV(set_lock_timeout);
Dave Wells's avatar
Dave Wells committed
2362
#undef SENV
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
2363

2364
    MALLOC(result->i);
Yoni Fogel's avatar
Yoni Fogel committed
2365
    if (result->i == 0) { r = ENOMEM; goto cleanup; }
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
2366
    memset(result->i, 0, sizeof *result->i);
Yoni Fogel's avatar
Yoni Fogel committed
2367 2368 2369 2370
    result->i->envdir_lockfd  = -1;
    result->i->datadir_lockfd = -1;
    result->i->logdir_lockfd  = -1;
    result->i->tmpdir_lockfd  = -1;
2371
    env_init_open_txn(result);
2372
    env_fs_init(result);
Yoni Fogel's avatar
Yoni Fogel committed
2373

2374 2375
    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
2376
                         toku_db_lt_panic, 
2377
                         toku_db_get_compare_fun,
Yoni Fogel's avatar
Yoni Fogel committed
2378
                         toku_malloc, toku_free, toku_realloc);
Yoni Fogel's avatar
Yoni Fogel committed
2379
    if (r!=0) { goto cleanup; }
2380
    toku_ltm_set_mutex(result->i->ltm, toku_ydb_mutex());
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
2381

2382
    {
Yoni Fogel's avatar
Yoni Fogel committed
2383 2384
	r = toku_logger_create(&result->i->logger);
	if (r!=0) { goto cleanup; }
2385 2386
	assert(result->i->logger);
    }
2387 2388 2389 2390 2391
    {
        r = toku_omt_create(&result->i->open_dbs);
        if (r!=0) goto cleanup;
        assert(result->i->open_dbs);
    }
2392

Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
2393
    *envp = result;
Yoni Fogel's avatar
Yoni Fogel committed
2394 2395 2396 2397 2398 2399 2400 2401
    r = 0;
cleanup:
    if (r!=0) {
        if (result) {
            if (result->i) {
                if (result->i->ltm) {
                    toku_ltm_close(result->i->ltm);
                }
2402 2403
                if (result->i->open_dbs)
                    toku_omt_destroy(&result->i->open_dbs);
Yoni Fogel's avatar
Yoni Fogel committed
2404 2405 2406 2407 2408 2409
                toku_free(result->i);
            }
            toku_free(result);
        }
    }
    return r;
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
2410 2411
}

2412 2413
int 
DB_ENV_CREATE_FUN (DB_ENV ** envp, u_int32_t flags) {
2414 2415 2416 2417
    toku_ydb_lock(); 
    int r = toku_env_create(envp, flags); 
    toku_ydb_unlock(); 
    return r;
Rich Prohaska's avatar
Rich Prohaska committed
2418 2419
}

2420 2421
static int 
toku_txn_release_locks(DB_TXN* txn) {
Yoni Fogel's avatar
Yoni Fogel committed
2422
    assert(txn);
2423
    toku_lth* lth = db_txn_struct_i(txn)->lth;
Yoni Fogel's avatar
Yoni Fogel committed
2424

Yoni Fogel's avatar
Yoni Fogel committed
2425 2426
    int r = ENOSYS;
    int first_error = 0;
Yoni Fogel's avatar
Yoni Fogel committed
2427 2428 2429 2430
    if (lth) {
        toku_lth_start_scan(lth);
        toku_lock_tree* next = toku_lth_next(lth);
        while (next) {
2431
            r = toku_lt_unlock(next, toku_txn_get_txnid(db_txn_struct_i(txn)->tokutxn));
Yoni Fogel's avatar
Yoni Fogel committed
2432 2433 2434 2435 2436
            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
2437 2438 2439
            next = toku_lth_next(lth);
        }
        toku_lth_close(lth);
2440
        db_txn_struct_i(txn)->lth = NULL;
Yoni Fogel's avatar
Yoni Fogel committed
2441
    }
Yoni Fogel's avatar
Yoni Fogel committed
2442 2443
    r = first_error;

Yoni Fogel's avatar
Yoni Fogel committed
2444
    return r;
Yoni Fogel's avatar
Yoni Fogel committed
2445 2446
}

2447 2448
// 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.
2449 2450
static void 
ydb_yield (voidfp f, void *fv, void *UU(v)) {
2451
    toku_ydb_unlock(); 
2452 2453
    if (f) 
        f(fv);
2454 2455 2456
    toku_ydb_lock();
}

2457 2458 2459
static int 
toku_txn_commit(DB_TXN * txn, u_int32_t flags,
                TXN_PROGRESS_POLL_FUNCTION poll, void* poll_extra) {
2460
    if (!txn) return EINVAL;
2461
    HANDLE_PANICKED_ENV(txn->mgrp);
Yoni Fogel's avatar
Yoni Fogel committed
2462
    //Recursively kill off children
2463 2464
    if (db_txn_struct_i(txn)->child) {
        //commit of child sets the child pointer to NULL
2465
        int r_child = toku_txn_commit(db_txn_struct_i(txn)->child, flags, NULL, NULL);
2466
        if (r_child !=0 && !toku_env_is_panicked(txn->mgrp)) {
2467
	    env_panic(txn->mgrp, r_child, "Recursive child commit failed during parent commit.\n");
2468
        }
Yoni Fogel's avatar
Yoni Fogel committed
2469 2470
        //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
2471
    }
2472
    assert(!db_txn_struct_i(txn)->child);
Yoni Fogel's avatar
Yoni Fogel committed
2473 2474
    //Remove from parent
    if (txn->parent) {
2475 2476
        assert(db_txn_struct_i(txn->parent)->child == txn);
        db_txn_struct_i(txn->parent)->child=NULL;
Yoni Fogel's avatar
Yoni Fogel committed
2477
    }
2478
    env_remove_open_txn(txn->mgrp, txn);
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
2479
    //toku_ydb_notef("flags=%d\n", flags);
2480 2481 2482 2483
    if (flags & DB_TXN_SYNC) {
        toku_txn_force_fsync_on_commit(db_txn_struct_i(txn)->tokutxn);
        flags &= ~DB_TXN_SYNC;
    }
2484
    int nosync = (flags & DB_TXN_NOSYNC)!=0 || (db_txn_struct_i(txn)->flags&DB_TXN_NOSYNC);
2485
    flags &= ~DB_TXN_NOSYNC;
Yoni Fogel's avatar
Yoni Fogel committed
2486 2487

    int r;
2488
    if (flags!=0)
2489 2490
	// frees the tokutxn
	// Calls ydb_yield(NULL) occasionally
2491
        //r = toku_logger_abort(db_txn_struct_i(txn)->tokutxn, ydb_yield, NULL);
2492
        r = toku_txn_abort_txn(db_txn_struct_i(txn)->tokutxn, ydb_yield, NULL, poll, poll_extra);
Yoni Fogel's avatar
Yoni Fogel committed
2493
    else
2494 2495
	// frees the tokutxn
	// Calls ydb_yield(NULL) occasionally
2496
        //r = toku_logger_commit(db_txn_struct_i(txn)->tokutxn, nosync, ydb_yield, NULL);
2497
        r = toku_txn_commit_txn(db_txn_struct_i(txn)->tokutxn, nosync, ydb_yield, NULL,
2498
				poll, poll_extra);
2499

2500
    if (r!=0 && !toku_env_is_panicked(txn->mgrp)) {
2501
	env_panic(txn->mgrp, r, "Error during commit.\n");
2502 2503 2504 2505 2506
    }
    //If panicked, we're done.
    HANDLE_PANICKED_ENV(txn->mgrp);
    assert(r==0);

2507
    // Close the logger after releasing the locks
2508
    r = toku_txn_release_locks(txn);
2509
    //toku_logger_txn_close(db_txn_struct_i(txn)->tokutxn);
2510
    toku_txn_maybe_fsync_log(db_txn_struct_i(txn)->tokutxn, ydb_yield, NULL);
2511
    toku_txn_close_txn(db_txn_struct_i(txn)->tokutxn);
2512
    // the toxutxn is freed, and we must free the rest. */
Yoni Fogel's avatar
Yoni Fogel committed
2513

2514 2515 2516
    //Promote list to parent (dbs that must close before abort)
    if (txn->parent) {
        //Combine lists.
2517 2518 2519
        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);
2520 2521 2522 2523
        }
    }
    else {
        //Empty the list
2524 2525
        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);
2526 2527 2528
        }
    }

2529
    // The txn is no good after the commit even if the commit fails, so free it up.
2530 2531 2532
#if !TOKUDB_NATIVE_H
    toku_free(db_txn_struct_i(txn));
#endif
2533
    toku_free(txn);    txn = NULL;
Yoni Fogel's avatar
Yoni Fogel committed
2534
    if (flags!=0) return EINVAL;
2535
    return r;
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
2536 2537
}

2538 2539
static u_int32_t 
toku_txn_id(DB_TXN * txn) {
2540
    HANDLE_PANICKED_ENV(txn->mgrp);
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
2541
    toku_ydb_barf();
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
2542
    abort();
Rich Prohaska's avatar
Rich Prohaska committed
2543
    return -1;
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
2544 2545
}

2546 2547 2548
static int 
toku_txn_abort(DB_TXN * txn,
               TXN_PROGRESS_POLL_FUNCTION poll, void* poll_extra) {
2549
    HANDLE_PANICKED_ENV(txn->mgrp);
2550
    //Recursively kill off children (abort or commit are both correct, commit is cheaper)
2551 2552
    if (db_txn_struct_i(txn)->child) {
        //commit of child sets the child pointer to NULL
2553
        int r_child = toku_txn_commit(db_txn_struct_i(txn)->child, DB_TXN_NOSYNC, NULL, NULL);
2554
        if (r_child !=0 && !toku_env_is_panicked(txn->mgrp)) {
2555
	    env_panic(txn->mgrp, r_child, "Recursive child commit failed during parent abort.\n");
2556
        }
Yoni Fogel's avatar
Yoni Fogel committed
2557 2558
        //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
2559
    }
2560
    assert(!db_txn_struct_i(txn)->child);
Yoni Fogel's avatar
Yoni Fogel committed
2561 2562
    //Remove from parent
    if (txn->parent) {
2563 2564
        assert(db_txn_struct_i(txn->parent)->child == txn);
        db_txn_struct_i(txn->parent)->child=NULL;
Yoni Fogel's avatar
Yoni Fogel committed
2565
    }
2566
    env_remove_open_txn(txn->mgrp, txn);
2567 2568

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

2571
    //int r = toku_logger_abort(db_txn_struct_i(txn)->tokutxn, ydb_yield, NULL);
2572
    int r = toku_txn_abort_txn(db_txn_struct_i(txn)->tokutxn, ydb_yield, NULL, poll, poll_extra);
2573
    if (r!=0 && !toku_env_is_panicked(txn->mgrp)) {
2574
	env_panic(txn->mgrp, r, "Error during abort.\n");
2575 2576 2577 2578
    }
    HANDLE_PANICKED_ENV(txn->mgrp);
    assert(r==0);
    r = toku_txn_release_locks(txn);
2579
    //toku_logger_txn_close(db_txn_struct_i(txn)->tokutxn);
2580
    toku_txn_maybe_fsync_log(db_txn_struct_i(txn)->tokutxn, ydb_yield, NULL);
2581
    toku_txn_close_txn(db_txn_struct_i(txn)->tokutxn);
Yoni Fogel's avatar
Yoni Fogel committed
2582

2583 2584 2585
#if !TOKUDB_NATIVE_H
    toku_free(db_txn_struct_i(txn));
#endif
2586
    toku_free(txn);
2587
    return r;
2588 2589
}

2590 2591
static int 
locked_txn_begin(DB_ENV *env, DB_TXN * stxn, DB_TXN ** txn, u_int32_t flags) {
2592
    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
2593 2594
}

2595 2596
static u_int32_t 
locked_txn_id(DB_TXN *txn) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
2597
    toku_ydb_lock(); u_int32_t r = toku_txn_id(txn); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
2598 2599
}

2600 2601
static int 
toku_txn_stat (DB_TXN *txn, struct txn_stat **txn_stat) {
2602
    XMALLOC(*txn_stat);
2603
    return toku_logger_txn_rollback_raw_count(db_txn_struct_i(txn)->tokutxn, &(*txn_stat)->rollback_raw_count);
2604 2605
}

2606 2607
static int 
locked_txn_stat (DB_TXN *txn, struct txn_stat **txn_stat) {
2608 2609 2610
    toku_ydb_lock(); u_int32_t r = toku_txn_stat(txn, txn_stat); toku_ydb_unlock(); return r;
}

Zardosht Kasheff's avatar
Zardosht Kasheff committed
2611
static int
2612 2613
locked_txn_commit_with_progress(DB_TXN *txn, u_int32_t flags,
                                TXN_PROGRESS_POLL_FUNCTION poll, void* poll_extra) {
Zardosht Kasheff's avatar
Zardosht Kasheff committed
2614 2615 2616 2617 2618 2619
    TOKUTXN ttxn = db_txn_struct_i(txn)->tokutxn;
    //
    // We must unpin rollback log, otherwise, another thread that tries to checkpoint during commit
    // will grab the multi operation lock, and then not be able to complete the checkpoint because
    // this thread has its rollback log pinned and is trying to grab the multi operation lock.
    //
2620
    toku_ydb_lock();
Zardosht Kasheff's avatar
Zardosht Kasheff committed
2621
    int r = toku_unpin_inprogress_rollback_log(ttxn);
2622
    toku_ydb_unlock();
Zardosht Kasheff's avatar
Zardosht Kasheff committed
2623 2624 2625 2626
    assert(r==0);
    if (toku_txn_requires_checkpoint(ttxn)) {
        toku_checkpoint(txn->mgrp->i->cachetable, txn->mgrp->i->logger, NULL, NULL, NULL, NULL);
    }
2627
    toku_multi_operation_client_lock(); //Cannot checkpoint during a commit.
Zardosht Kasheff's avatar
Zardosht Kasheff committed
2628
    toku_ydb_lock(); r = toku_txn_commit(txn, flags, poll, poll_extra); toku_ydb_unlock();
2629
    toku_multi_operation_client_unlock(); //Cannot checkpoint during a commit.
2630
    return r;
Rich Prohaska's avatar
Rich Prohaska committed
2631 2632
}

2633 2634 2635
static int 
locked_txn_abort_with_progress(DB_TXN *txn,
                               TXN_PROGRESS_POLL_FUNCTION poll, void* poll_extra) {
2636
    toku_multi_operation_client_lock(); //Cannot checkpoint during an abort.
2637
    toku_ydb_lock(); int r = toku_txn_abort(txn, poll, poll_extra); toku_ydb_unlock();
2638
    toku_multi_operation_client_unlock(); //Cannot checkpoint during an abort.
2639
    return r;
Rich Prohaska's avatar
Rich Prohaska committed
2640 2641
}

2642 2643
static int 
locked_txn_commit(DB_TXN *txn, u_int32_t flags) {
2644 2645 2646 2647 2648
    int r;
    r = locked_txn_commit_with_progress(txn, flags, NULL, NULL);
    return r;
}

2649 2650
static int 
locked_txn_abort(DB_TXN *txn) {
2651 2652 2653 2654 2655
    int r;
    r = locked_txn_abort_with_progress(txn, NULL, NULL);
    return r;
}

2656 2657
static int 
toku_txn_begin(DB_ENV *env, DB_TXN * stxn, DB_TXN ** txn, u_int32_t flags, int internal) {
2658
    HANDLE_PANICKED_ENV(env);
2659
    HANDLE_ILLEGAL_WORKING_PARENT_TXN(env, stxn); //Cannot create child while child already exists.
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
2660 2661
    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");
2662 2663
    u_int32_t txn_flags = 0;
    txn_flags |= DB_TXN_NOWAIT; //We do not support blocking locks.
2664
    TOKU_ISOLATION child_isolation = TOKU_ISO_SERIALIZABLE;
2665 2666 2667 2668 2669 2670 2671 2672 2673
    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)
       ) 
    {
2674 2675 2676
        return toku_ydb_do_error(
            env, 
            EINVAL, 
2677
            "Invalid isolation flags set\n"
2678 2679
            );
    }
2680 2681 2682 2683 2684 2685 2686 2687 2688 2689 2690 2691 2692 2693 2694 2695 2696 2697 2698 2699 2700 2701 2702 2703 2704 2705 2706 2707 2708 2709 2710 2711 2712 2713 2714
    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;
2715
        }
2716
    }
2717 2718 2719 2720 2721 2722 2723
    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"
            );   
2724
    }
2725

2726 2727 2728 2729 2730 2731 2732 2733 2734 2735
    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");

2736 2737
    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
2738 2739
    if (result == 0)
        return ENOMEM;
2740
    memset(result, 0, result_size);
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
2741
    //toku_ydb_notef("parent=%p flags=0x%x\n", stxn, flags);
2742
    result->mgrp = env;
2743 2744 2745 2746 2747 2748 2749
#define STXN(name) result->name = locked_txn_ ## name
    STXN(abort);
    STXN(commit);
    STXN(abort_with_progress);
    STXN(commit_with_progress);
    STXN(id);
#undef STXN
2750
    result->txn_stat = locked_txn_stat;
2751 2752 2753


    result->parent = stxn;
2754 2755 2756
#if !TOKUDB_NATIVE_H
    MALLOC(db_txn_struct_i(result));
    if (!db_txn_struct_i(result)) {
Yoni Fogel's avatar
Yoni Fogel committed
2757 2758 2759
        toku_free(result);
        return ENOMEM;
    }
2760 2761 2762
#endif
    memset(db_txn_struct_i(result), 0, sizeof *db_txn_struct_i(result));
    db_txn_struct_i(result)->flags = txn_flags;
2763
    db_txn_struct_i(result)->iso = child_isolation;
2764
    toku_list_init(&db_txn_struct_i(result)->dbs_that_must_close_before_abort);
Yoni Fogel's avatar
Yoni Fogel committed
2765 2766

    int r;
Yoni Fogel's avatar
Yoni Fogel committed
2767
    if (env->i->open_flags & DB_INIT_LOCK && !stxn) {
2768
        r = toku_lth_create(&db_txn_struct_i(result)->lth,
Yoni Fogel's avatar
Yoni Fogel committed
2769 2770
                            toku_malloc, toku_free, toku_realloc);
        if (r!=0) {
2771 2772 2773
#if !TOKUDB_NATIVE_H
            toku_free(db_txn_struct_i(result));
#endif
Yoni Fogel's avatar
Yoni Fogel committed
2774 2775 2776
            toku_free(result);
            return r;
        }
Yoni Fogel's avatar
Yoni Fogel committed
2777 2778
    }
    
2779
    //r = toku_logger_txn_begin(stxn ? db_txn_struct_i(stxn)->tokutxn : 0, &db_txn_struct_i(result)->tokutxn, env->i->logger);
2780 2781 2782 2783 2784 2785 2786 2787 2788 2789 2790 2791 2792 2793 2794 2795 2796 2797
    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;
        }
    }
2798 2799 2800 2801 2802 2803
    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
2804 2805
    if (r != 0)
        return r;
2806

Yoni Fogel's avatar
Yoni Fogel committed
2807 2808
    //Add to the list of children for the parent.
    if (result->parent) {
2809 2810
        assert(!db_txn_struct_i(result->parent)->child);
        db_txn_struct_i(result->parent)->child = result;
Yoni Fogel's avatar
Yoni Fogel committed
2811
    }
2812
    env_add_open_txn(env, result);
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
2813 2814 2815 2816
    *txn = result;
    return 0;
}

2817 2818
int 
log_compare(const DB_LSN * a, const DB_LSN * b) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
2819
    toku_ydb_lock();
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
2820
    fprintf(stderr, "%s:%d log_compare(%p,%p)\n", __FILE__, __LINE__, a, b);
2821
    assert(0);
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
2822
    toku_ydb_unlock();
2823
    return 0;
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
2824 2825
}

2826 2827
static void env_note_zombie_db_closed(DB_ENV *env, DB *db);

2828 2829
static int
db_close_before_brt(DB *db, u_int32_t UU(flags)) {
2830 2831 2832
    int r;
    char *error_string = NULL;
    
2833 2834 2835 2836
    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)
    }
2837 2838 2839 2840 2841 2842 2843
    r = toku_close_brt(db->i->brt, &error_string);
    if (r) {
	if (!error_string)
	    error_string = "Closing file\n";
	// Panicking the whole environment may be overkill, but I'm not sure what else to do.
	env_panic(db->dbenv, r, error_string);
	toku_ydb_do_error(db->dbenv, r, "%s", error_string);
2844
    }
2845 2846 2847 2848 2849 2850 2851 2852 2853 2854
    else {
	if (db->i->lt) {
	    toku_lt_remove_db_ref(db->i->lt, db);
	}
	// printf("%s:%d %d=__toku_db_close(%p)\n", __FILE__, __LINE__, r, db);
	toku_sdbt_cleanup(&db->i->skey);
	toku_sdbt_cleanup(&db->i->sval);
	if (db->i->dname) toku_free(db->i->dname);
	toku_free(db->i);
	toku_free(db);
Yoni Fogel's avatar
Yoni Fogel committed
2855
    }
2856
    return r;
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
2857 2858
}

2859 2860 2861 2862 2863 2864 2865 2866
// 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;
2867 2868 2869 2870 2871 2872 2873
    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;
2874 2875
    if (cmp != 0) return cmp;
    if (db < dbfind) return -1;
2876
    if (db > dbfind) return  1;
2877 2878 2879 2880 2881 2882 2883
    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
2884
    assert(!db->i->is_zombie);
2885 2886 2887
    int r;
    OMTVALUE dbv;
    uint32_t idx;
2888
    env->i->num_open_dbs++;
2889
    r = toku_omt_find_zero(env->i->open_dbs, find_db_by_db, db, &dbv, &idx);
2890 2891 2892 2893 2894 2895 2896 2897
    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);
2898
    assert(!db->i->is_zombie);
2899
    assert(env->i->num_open_dbs);
2900 2901 2902
    int r;
    OMTVALUE dbv;
    uint32_t idx;
2903
    env->i->num_open_dbs--;
2904
    r = toku_omt_find_zero(env->i->open_dbs, find_db_by_db, db, &dbv, &idx);
2905 2906 2907 2908 2909 2910 2911 2912 2913 2914 2915 2916 2917 2918
    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;
2919
    env->i->num_zombie_dbs++;
2920
    r = toku_omt_find_zero(env->i->open_dbs, find_db_by_db, db, &dbv, &idx);
2921 2922 2923 2924 2925 2926 2927 2928 2929
    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);
2930
    assert(env->i->num_zombie_dbs);
2931 2932 2933
    int r;
    OMTVALUE dbv;
    uint32_t idx;
2934
    env->i->num_zombie_dbs--;
2935
    r = toku_omt_find_zero(env->i->open_dbs, find_db_by_db, db, &dbv, &idx);
2936 2937 2938 2939 2940 2941 2942
    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
2943 2944 2945 2946 2947 2948 2949 2950 2951 2952 2953 2954 2955 2956 2957 2958 2959
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;
2960 2961
    const char *dname     = db->i->dname;
    const char *dnamefind = dnamev;
2962 2963 2964 2965 2966 2967
    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;
2968 2969 2970 2971 2972 2973 2974 2975 2976
}

// 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;
2977
    r = toku_omt_find_zero(env->i->open_dbs, find_open_db_by_dname, (void*)dname, &dbv, &idx);
2978 2979 2980
    if (r==0) {
        DB *db = dbv;
        assert(strcmp(dname, db->i->dname) == 0);
2981
        assert(!db->i->is_zombie);
2982 2983 2984 2985 2986 2987 2988 2989 2990
        rval = TRUE;
    }
    else {
        assert(r==DB_NOTFOUND);
        rval = FALSE;
    }
    return rval;
}

2991 2992 2993 2994 2995 2996 2997
// 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;
2998
    r = toku_omt_find_zero(env->i->open_dbs, find_zombie_db_by_dname, (void*)dname, &dbv, &idx);
2999 3000 3001 3002 3003 3004 3005 3006 3007 3008 3009 3010 3011 3012
    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;
}

3013 3014 3015 3016 3017 3018 3019 3020 3021 3022
void
toku_db_add_ref(DB *db) {
    db->i->refs++;
}

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

3023
//DB->close()
3024 3025
static int 
toku_db_close(DB * db, u_int32_t flags) {
3026 3027 3028 3029 3030 3031 3032 3033 3034 3035 3036 3037 3038 3039
    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);
3040

3041 3042
        r = toku_brt_db_delay_closed(db->i->brt, db, db_close_before_brt, flags);
    }
3043 3044 3045 3046
    return r;
}


3047
//Get the main portion of a cursor flag (excluding the bitwise or'd components).
3048 3049
static int 
get_main_cursor_flag(u_int32_t flags) {
3050
    return flags & DB_OPFLAGS_MASK;
Yoni Fogel's avatar
Yoni Fogel committed
3051 3052
}

3053 3054
static int 
get_nonmain_cursor_flags(u_int32_t flags) {
3055
    return flags & ~(DB_OPFLAGS_MASK);
3056 3057
}

3058 3059
static inline BOOL 
toku_c_uninitialized(DBC* c) {
3060
    return toku_brt_cursor_uninitialized(dbc_struct_i(c)->c);
Yoni Fogel's avatar
Yoni Fogel committed
3061
}            
Yoni Fogel's avatar
Yoni Fogel committed
3062

3063 3064 3065 3066 3067 3068 3069 3070 3071 3072 3073
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;
3074 3075
    context->skey = dbc_struct_i(c)->skey;
    context->sval = dbc_struct_i(c)->sval;
3076 3077 3078 3079 3080 3081 3082 3083 3084 3085 3086
}

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;
}

3087 3088
static int 
toku_c_get_current_unconditional(DBC* c, u_int32_t flags, DBT* key, DBT* val) {
3089 3090 3091 3092
    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
3093 3094 3095
    return r;
}

3096 3097
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
3098 3099 3100 3101 3102
    *flag    -= *get_flag;
    *get_flag =  new_flag;
    *flag    += *get_flag;
}

Yoni Fogel's avatar
Yoni Fogel committed
3103 3104 3105 3106 3107 3108 3109
/*
    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.
*/
3110 3111
static inline DB_TXN* 
toku_txn_ancestor(DB_TXN* txn) {
Yoni Fogel's avatar
Yoni Fogel committed
3112
    while (txn && txn->parent) txn = txn->parent;
Yoni Fogel's avatar
Yoni Fogel committed
3113

Yoni Fogel's avatar
Yoni Fogel committed
3114 3115 3116
    return txn;
}

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

Yoni Fogel's avatar
Yoni Fogel committed
3119 3120 3121 3122 3123 3124 3125 3126 3127 3128 3129
/* 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
3130
    u_int32_t   lock_flags;         // The prelock flags.
Yoni Fogel's avatar
Yoni Fogel committed
3131 3132 3133 3134 3135 3136 3137 3138 3139 3140
    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;

3141

3142 3143
static inline u_int32_t 
get_prelocked_flags(u_int32_t flags) {
3144
    u_int32_t lock_flags = flags & (DB_PRELOCKED | DB_PRELOCKED_WRITE | DB_PRELOCKED_FILE_READ);
3145 3146
    return lock_flags;
}
3147

3148 3149
static inline u_int32_t 
get_cursor_prelocked_flags(u_int32_t flags, DBC* dbc) {
3150
    u_int32_t lock_flags = flags & (DB_PRELOCKED | DB_PRELOCKED_WRITE | DB_PRELOCKED_FILE_READ);
3151 3152

    //DB_READ_UNCOMMITTED and DB_READ_COMMITTED transactions 'own' all read locks for user-data dictionaries.
3153
    if (dbc_struct_i(dbc)->iso != TOKU_ISO_SERIALIZABLE) {
3154
        lock_flags |= DB_PRELOCKED;
3155
    }
3156
    return lock_flags;
Yoni Fogel's avatar
Yoni Fogel committed
3157 3158
}

3159 3160
static int
toku_c_get(DBC* c, DBT* key, DBT* val, u_int32_t flag) {
3161 3162 3163
    //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
3164

3165 3166 3167 3168 3169 3170 3171 3172 3173 3174 3175 3176
    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
3177
        case (DB_FIRST):
3178 3179
            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
3180
            break;
Yoni Fogel's avatar
Yoni Fogel committed
3181
        case (DB_LAST):
3182 3183
            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
3184
            break;
3185
        case (DB_NEXT):
Yoni Fogel's avatar
Yoni Fogel committed
3186
        case (DB_NEXT_NODUP):
3187
            query_context_wrapped_init(&context, c, key,  val);
3188
            r = toku_c_getf_next(c, remaining_flags, c_get_wrapper_callback, &context);
Yoni Fogel's avatar
Yoni Fogel committed
3189 3190
            break;
        case (DB_PREV):
3191
        case (DB_PREV_NODUP):
3192 3193
            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
3194 3195
            break;
#ifdef DB_PREV_DUP
Yoni Fogel's avatar
Yoni Fogel committed
3196
        case (DB_PREV_DUP):
3197 3198
            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
3199 3200
            break;
#endif
3201 3202 3203 3204 3205 3206 3207 3208 3209 3210 3211 3212 3213 3214 3215 3216 3217
        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;
3218 3219 3220 3221
        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
3222
        default:
Yoni Fogel's avatar
Yoni Fogel committed
3223
            r = EINVAL;
3224
            break;
Yoni Fogel's avatar
Yoni Fogel committed
3225
    }
Yoni Fogel's avatar
Yoni Fogel committed
3226
    return r;
Yoni Fogel's avatar
Yoni Fogel committed
3227 3228
}

3229 3230
static int 
locked_c_getf_first(DBC *c, u_int32_t flag, YDB_CALLBACK_FUNCTION f, void *extra) {
3231 3232 3233 3234
    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
3235 3236
}

3237 3238
static int 
locked_c_getf_last(DBC *c, u_int32_t flag, YDB_CALLBACK_FUNCTION f, void *extra) {
3239 3240
    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
3241

3242 3243
static int 
locked_c_getf_next(DBC *c, u_int32_t flag, YDB_CALLBACK_FUNCTION f, void *extra) {
3244
    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
3245 3246
}

3247 3248
static int 
locked_c_getf_prev(DBC *c, u_int32_t flag, YDB_CALLBACK_FUNCTION f, void *extra) {
3249
    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
3250 3251
}

3252 3253
static int 
locked_c_getf_current(DBC *c, u_int32_t flag, YDB_CALLBACK_FUNCTION f, void *extra) {
3254
    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
3255 3256
}

3257 3258
static int 
locked_c_getf_current_binding(DBC *c, u_int32_t flag, YDB_CALLBACK_FUNCTION f, void *extra) {
3259 3260
    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
3261

3262 3263
static int 
locked_c_getf_set(DBC *c, u_int32_t flag, DBT * key, YDB_CALLBACK_FUNCTION f, void *extra) {
3264 3265
    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
3266

3267 3268
static int 
locked_c_getf_set_range(DBC *c, u_int32_t flag, DBT * key, YDB_CALLBACK_FUNCTION f, void *extra) {
3269 3270
    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
3271

3272 3273
static int 
locked_c_getf_set_range_reverse(DBC *c, u_int32_t flag, DBT * key, YDB_CALLBACK_FUNCTION f, void *extra) {
3274 3275 3276
    toku_ydb_lock();  int r = toku_c_getf_set_range_reverse(c, flag, key, f, extra); toku_ydb_unlock(); return r;
}

3277 3278 3279
// Get a range lock.
// Return when the range lock is acquired or the default lock tree timeout has expired.  
// The ydb mutex must be held when called and may be released when waiting in the lock tree.
3280 3281 3282 3283 3284 3285 3286 3287 3288 3289 3290 3291 3292
static int
get_range_lock(DB *db, DB_TXN *txn, const DBT *left_key, const DBT *right_key, toku_lock_type lock_type) {
    int r;
    DB_TXN *txn_anc = toku_txn_ancestor(txn);
    r = toku_txn_add_lt(txn_anc, db->i->lt);
    if (r == 0) {
        TXNID txn_anc_id = toku_txn_get_txnid(db_txn_struct_i(txn_anc)->tokutxn);
        toku_lock_request lock_request;
        toku_lock_request_init(&lock_request, db, txn_anc_id, left_key, right_key, lock_type);
        r = toku_lt_acquire_lock_request_with_default_timeout_locked(db->i->lt, &lock_request);
        toku_lock_request_destroy(&lock_request);
    }
    return r;
3293 3294
}

3295
// Setup and start an asynchronous lock request.
3296
static int
3297
start_range_lock(DB *db, DB_TXN *txn, const DBT *left_key, const DBT *right_key, toku_lock_type lock_type, toku_lock_request *lock_request) {
3298
    int r;
3299 3300 3301
    DB_TXN *txn_anc = toku_txn_ancestor(txn);
    r = toku_txn_add_lt(txn_anc, db->i->lt);
    if (r == 0) {
3302
        TXNID txn_anc_id = toku_txn_get_txnid(db_txn_struct_i(txn_anc)->tokutxn);
3303 3304
        toku_lock_request_set(lock_request, db, txn_anc_id, left_key, right_key, lock_type);
        r = toku_lock_request_start_locked(lock_request, db->i->lt, true);
3305 3306
    }
    return r;
Yoni Fogel's avatar
Yoni Fogel committed
3307 3308
}

3309 3310 3311 3312 3313 3314 3315
static int 
get_point_write_lock(DB *db, DB_TXN *txn, const DBT *key) {
    int r = get_range_lock(db, txn, key, key, LOCK_REQUEST_WRITE);
    return r;
}

// assume ydb is locked
3316
int
Zardosht Kasheff's avatar
Zardosht Kasheff committed
3317 3318
toku_grab_read_lock_on_directory (DB* db, DB_TXN * txn) {
    // bad hack because some environment dictionaries do not have a dname
3319
    char *dname = db->i->dname;
3320
    if (!dname || (db->dbenv->i->directory->i->lt == NULL))
Zardosht Kasheff's avatar
Zardosht Kasheff committed
3321
        return 0;
3322

Zardosht Kasheff's avatar
Zardosht Kasheff committed
3323
    //Left end of range == right end of range (point lock)
3324 3325
    DBT key_in_directory = { .data = dname, .size = strlen(dname)+1 };
    int r = get_range_lock(db->dbenv->i->directory, txn, &key_in_directory, &key_in_directory, LOCK_REQUEST_READ);
3326 3327 3328 3329
    if (r == 0)
	directory_read_locks++;
    else
	directory_read_locks_fail++;
Zardosht Kasheff's avatar
Zardosht Kasheff committed
3330 3331 3332
    return r;
}

3333 3334
//This is the user level callback function given to ydb layer functions like
//toku_c_getf_first
Yoni Fogel's avatar
Yoni Fogel committed
3335

3336 3337 3338 3339
typedef struct query_context_base_t {
    BRT_CURSOR  c;
    DB_TXN     *txn;
    DB         *db;
3340
    YDB_CALLBACK_FUNCTION f;
3341 3342
    void       *f_extra;
    int         r_user_callback;
3343 3344
    BOOL        do_locking;
    BOOL        is_write_op;
3345
    toku_lock_request lock_request;
3346
} *QUERY_CONTEXT_BASE, QUERY_CONTEXT_BASE_S;
3347

3348 3349 3350
typedef struct query_context_t {
    QUERY_CONTEXT_BASE_S  base;
} *QUERY_CONTEXT, QUERY_CONTEXT_S;
Yoni Fogel's avatar
Yoni Fogel committed
3351

3352 3353 3354 3355 3356
typedef struct query_context_with_input_t {
    QUERY_CONTEXT_BASE_S  base;
    DBT                  *input_key;
    DBT                  *input_val;
} *QUERY_CONTEXT_WITH_INPUT, QUERY_CONTEXT_WITH_INPUT_S;
Yoni Fogel's avatar
Yoni Fogel committed
3357

3358
static void
3359
query_context_base_init(QUERY_CONTEXT_BASE context, DBC *c, u_int32_t flag, BOOL is_write_op, YDB_CALLBACK_FUNCTION f, void *extra) {
3360 3361
    context->c       = dbc_struct_i(c)->c;
    context->txn     = dbc_struct_i(c)->txn;
3362
    context->db      = c->dbp;
3363
    context->f       = f;
3364
    context->f_extra = extra;
3365
    context->is_write_op = is_write_op;
3366
    u_int32_t lock_flags = get_cursor_prelocked_flags(flag, c);
3367 3368
    if (context->is_write_op) 
        lock_flags &= DB_PRELOCKED_WRITE; // Only care about whether already locked for write
Zardosht Kasheff's avatar
Zardosht Kasheff committed
3369
    context->do_locking = (BOOL)(context->db->i->lt!=NULL && !(lock_flags & (DB_PRELOCKED|DB_PRELOCKED_WRITE)));
3370
    context->r_user_callback = 0;
3371 3372 3373 3374 3375 3376
    toku_lock_request_default_init(&context->lock_request);
}

static void
query_context_base_destroy(QUERY_CONTEXT_BASE context) {
    toku_lock_request_destroy(&context->lock_request);
Yoni Fogel's avatar
Yoni Fogel committed
3377 3378
}

3379
static void
3380 3381
query_context_init_read(QUERY_CONTEXT context, DBC *c, u_int32_t flag, YDB_CALLBACK_FUNCTION f, void *extra) {
    BOOL is_write = FALSE;
3382
    query_context_base_init(&context->base, c, flag, is_write, f, extra);
3383 3384 3385
}

static void
3386 3387
query_context_init_write(QUERY_CONTEXT context, DBC *c, u_int32_t flag, YDB_CALLBACK_FUNCTION f, void *extra) {
    BOOL is_write = TRUE;
3388
    query_context_base_init(&context->base, c, flag, is_write, f, extra);
3389
}
Yoni Fogel's avatar
Yoni Fogel committed
3390

3391 3392
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) {
3393 3394
    // grab write locks if the DB_RMW flag is set or the cursor was created with the DB_RMW flag
    BOOL is_write = ((flag & DB_RMW) != 0) || dbc_struct_i(c)->rmw;
3395
    query_context_base_init(&context->base, c, flag, is_write, f, extra);
3396 3397
    context->input_key = key;
    context->input_val = val;
3398 3399
}

3400 3401 3402 3403 3404
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) {
3405
    HANDLE_PANICKED_DB(c->dbp);
3406
    HANDLE_CURSOR_ILLEGAL_WORKING_PARENT_TXN(c);
3407

3408 3409 3410 3411
    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;
3412
    u_int32_t lock_flags = get_cursor_prelocked_flags(flags, c);
3413
    unchecked_flags &= ~lock_flags;
3414
    BOOL do_locking = (BOOL)(c->dbp->i->lt && !(lock_flags&DB_PRELOCKED_WRITE));
3415

3416
    int r = 0;
3417 3418
    if (unchecked_flags!=0) 
        r = EINVAL;
3419 3420 3421
    else {
        if (do_locking) {
            QUERY_CONTEXT_S context;
3422
            query_context_init_write(&context, c, lock_flags, NULL, NULL);
3423 3424 3425 3426 3427 3428 3429 3430 3431
            while (r == 0) {
                //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 == DB_LOCK_NOTGRANTED)
                    r = toku_lock_request_wait_with_default_timeout(&context.base.lock_request, c->dbp->i->lt);
                else
                    break;
            }
            query_context_base_destroy(&context.base);
3432 3433 3434
        }
        if (r==0) {
            //Do the actual delete.
3435 3436
            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);
3437 3438
        }
    }
3439 3440
    return r;
}
3441

3442 3443 3444 3445 3446
//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
3447

3448
    int r;
Yoni Fogel's avatar
Yoni Fogel committed
3449

3450
    assert(context->do_locking);
3451
    assert(context->is_write_op);
3452 3453
    assert(key!=NULL);
    assert(val!=NULL);
3454

3455 3456
    //Lock:
    //  left(key,val)==right(key,val) == (key, val);
3457
    r = start_range_lock(context->db, context->txn, key, key, LOCK_REQUEST_WRITE, &context->lock_request);
3458 3459

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

3463 3464
static int c_getf_first_callback(ITEMLEN keylen, bytevec key, ITEMLEN vallen, bytevec val, void *extra);

3465 3466
static void 
c_query_context_init(QUERY_CONTEXT context, DBC *c, u_int32_t flag, YDB_CALLBACK_FUNCTION f, void *extra) {
3467 3468 3469 3470 3471 3472 3473 3474 3475 3476
    BOOL is_write_op = FALSE;
    // grab write locks if the DB_RMW flag is set or the cursor was created with the DB_RMW flag
    if ((flag & DB_RMW) || dbc_struct_i(c)->rmw)
        is_write_op = TRUE;
    if (is_write_op)
        query_context_init_write(context, c, flag, f, extra);
    else
        query_context_init_read(context, c, flag, f, extra);
}

3477 3478 3479 3480 3481
static void 
c_query_context_destroy(QUERY_CONTEXT context) {
    query_context_base_destroy(&context->base);
}

3482 3483 3484
static int
toku_c_getf_first(DBC *c, u_int32_t flag, YDB_CALLBACK_FUNCTION f, void *extra) {
    HANDLE_PANICKED_DB(c->dbp);
3485
    HANDLE_CURSOR_ILLEGAL_WORKING_PARENT_TXN(c);
3486
    num_point_queries++;   // accountability
3487
    int r = 0;
3488
    QUERY_CONTEXT_S context; //Describes the context of this query.
3489
    c_query_context_init(&context, c, flag, f, extra);
3490 3491 3492 3493 3494 3495 3496 3497 3498 3499 3500 3501
    while (r == 0) {
        //toku_brt_cursor_first will call c_getf_first_callback(..., context) (if query is successful)
        r = toku_brt_cursor_first(dbc_struct_i(c)->c, c_getf_first_callback, &context);
        if (r == DB_LOCK_NOTGRANTED)
            r = toku_lock_request_wait_with_default_timeout(&context.base.lock_request, c->dbp->i->lt);
        else {
            if (r == TOKUDB_USER_CALLBACK_ERROR)
                r = context.base.r_user_callback;
            break;
        }
    }
    c_query_context_destroy(&context);
3502 3503 3504 3505 3506 3507 3508 3509 3510 3511
    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;
3512 3513
    DBT found_key = { .data = (void *) key, .size = keylen };
    DBT found_val = { .data = (void *) val, .size = vallen };
3514 3515

    if (context->do_locking) {
3516 3517
        const DBT *left_key = toku_lt_neg_infinity;
        const DBT *right_key = key != NULL ? &found_key : toku_lt_infinity;
3518 3519
        r = start_range_lock(context->db, context->txn, left_key, right_key, 
                             context->is_write_op ? LOCK_REQUEST_WRITE : LOCK_REQUEST_READ, &context->lock_request);
3520 3521
    } else 
        r = 0;
3522 3523 3524

    //Call application-layer callback if found and locks were successfully obtained.
    if (r==0 && key!=NULL) {
3525
        context->r_user_callback = context->f(&found_key, &found_val, context->f_extra);
3526
        r = context->r_user_callback;
3527 3528 3529 3530 3531 3532 3533 3534 3535 3536 3537
    }

    //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);
3538
    HANDLE_CURSOR_ILLEGAL_WORKING_PARENT_TXN(c);
3539
    num_point_queries++;   // accountability
3540
    int r = 0;
3541
    QUERY_CONTEXT_S context; //Describes the context of this query.
3542
    c_query_context_init(&context, c, flag, f, extra); 
3543 3544 3545 3546 3547 3548 3549 3550 3551 3552 3553 3554
    while (r == 0) {
        //toku_brt_cursor_last will call c_getf_last_callback(..., context) (if query is successful)
        r = toku_brt_cursor_last(dbc_struct_i(c)->c, c_getf_last_callback, &context);
        if (r == DB_LOCK_NOTGRANTED)
            r = toku_lock_request_wait_with_default_timeout(&context.base.lock_request, c->dbp->i->lt);
        else {
            if (r == TOKUDB_USER_CALLBACK_ERROR)
                r = context.base.r_user_callback;
            break;
        }
    }
    c_query_context_destroy(&context);
3555 3556 3557 3558 3559 3560 3561 3562 3563 3564
    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;
3565 3566
    DBT found_key = { .data = (void *) key, .size = keylen };
    DBT found_val = { .data = (void *) val, .size = vallen };
3567 3568

    if (context->do_locking) {
3569 3570
        const DBT *left_key = key != NULL ? &found_key : toku_lt_neg_infinity;
        const DBT *right_key = toku_lt_infinity;
3571 3572
        r = start_range_lock(context->db, context->txn, left_key, right_key, 
                             context->is_write_op ? LOCK_REQUEST_WRITE : LOCK_REQUEST_READ, &context->lock_request);
3573 3574
    } else 
        r = 0;
3575 3576 3577

    //Call application-layer callback if found and locks were successfully obtained.
    if (r==0 && key!=NULL) {
3578
        context->r_user_callback = context->f(&found_key, &found_val, context->f_extra);
3579
        r = context->r_user_callback;
3580 3581 3582 3583 3584 3585 3586 3587 3588 3589 3590 3591
    }

    //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);
3592
    HANDLE_CURSOR_ILLEGAL_WORKING_PARENT_TXN(c);
3593 3594
    if (toku_c_uninitialized(c)) 
        r = toku_c_getf_first(c, flag, f, extra);
3595
    else {
3596
        r = 0;
3597
        QUERY_CONTEXT_S context; //Describes the context of this query.
3598
        c_query_context_init(&context, c, flag, f, extra); 
3599 3600 3601 3602 3603 3604 3605 3606 3607 3608 3609 3610
        while (r == 0) {
            //toku_brt_cursor_next will call c_getf_next_callback(..., context) (if query is successful)
            r = toku_brt_cursor_next(dbc_struct_i(c)->c, c_getf_next_callback, &context);
            if (r == DB_LOCK_NOTGRANTED)
                r = toku_lock_request_wait_with_default_timeout(&context.base.lock_request, c->dbp->i->lt);
            else {
                if (r == TOKUDB_USER_CALLBACK_ERROR)
                    r = context.base.r_user_callback;
                break;
            }
        }
        c_query_context_destroy(&context);
3611 3612 3613 3614 3615 3616 3617 3618 3619 3620 3621 3622
    }
    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;

3623 3624 3625
    DBT found_key = { .data = (void *) key, .size = keylen };
    DBT found_val = { .data = (void *) val, .size = vallen };
    num_sequential_queries++;   // accountability
3626 3627

    if (context->do_locking) {
3628
        const DBT *prevkey, *prevval;
3629
        toku_brt_cursor_peek(context->c, &prevkey, &prevval);
3630 3631
        const DBT *left_key = prevkey;
        const DBT *right_key = key != NULL ? &found_key : toku_lt_infinity;
3632 3633
        r = start_range_lock(context->db, context->txn, left_key, right_key, 
                             context->is_write_op ? LOCK_REQUEST_WRITE : LOCK_REQUEST_READ, &context->lock_request);
3634 3635
    } else 
        r = 0;
3636 3637 3638

    //Call application-layer callback if found and locks were successfully obtained.
    if (r==0 && key!=NULL) {
3639
        context->r_user_callback = context->f(&found_key, &found_val, context->f_extra);
3640
        r = context->r_user_callback;
3641 3642 3643 3644 3645 3646 3647 3648 3649 3650 3651 3652
    }

    //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);
3653
    HANDLE_CURSOR_ILLEGAL_WORKING_PARENT_TXN(c);
3654 3655
    if (toku_c_uninitialized(c)) 
        r = toku_c_getf_last(c, flag, f, extra);
3656
    else {
3657
        r = 0;
3658
        QUERY_CONTEXT_S context; //Describes the context of this query.
3659 3660 3661 3662 3663 3664 3665 3666 3667 3668 3669 3670 3671
        c_query_context_init(&context, c, flag, f, extra);
        while (r == 0) {
            //toku_brt_cursor_prev will call c_getf_prev_callback(..., context) (if query is successful)
            r = toku_brt_cursor_prev(dbc_struct_i(c)->c, c_getf_prev_callback, &context);
            if (r == DB_LOCK_NOTGRANTED)
                r = toku_lock_request_wait_with_default_timeout(&context.base.lock_request, c->dbp->i->lt);
            else {
                if (r == TOKUDB_USER_CALLBACK_ERROR)
                    r = context.base.r_user_callback;
                break;
            }
        }
        c_query_context_destroy(&context);
Yoni Fogel's avatar
Yoni Fogel committed
3672 3673 3674 3675
    }
    return r;
}

3676 3677 3678 3679 3680 3681 3682
//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;
3683 3684
    DBT found_key = { .data = (void *) key, .size = keylen };
    DBT found_val = { .data = (void *) val, .size = vallen };
3685
    num_sequential_queries++;   // accountability
3686 3687

    if (context->do_locking) {
3688
        const DBT *prevkey, *prevval;
3689
        toku_brt_cursor_peek(context->c, &prevkey, &prevval);
3690 3691
        const DBT *left_key = key != NULL ? &found_key : toku_lt_neg_infinity;
        const DBT *right_key = prevkey;
3692 3693
        r = start_range_lock(context->db, context->txn, left_key, right_key, 
                             context->is_write_op ? LOCK_REQUEST_WRITE : LOCK_REQUEST_READ, &context->lock_request);
3694 3695
    } else 
        r = 0;
3696 3697 3698

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

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

3707 3708 3709 3710 3711
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);
3712
    HANDLE_CURSOR_ILLEGAL_WORKING_PARENT_TXN(c);
3713 3714

    QUERY_CONTEXT_S context; //Describes the context of this query.
3715
    num_sequential_queries++;   // accountability
3716
    c_query_context_init(&context, c, flag, f, extra); 
3717
    //toku_brt_cursor_current will call c_getf_current_callback(..., context) (if query is successful)
3718
    int r = toku_brt_cursor_current(dbc_struct_i(c)->c, DB_CURRENT, c_getf_current_callback, &context);
3719
    if (r == TOKUDB_USER_CALLBACK_ERROR) r = context.base.r_user_callback;
3720
    c_query_context_destroy(&context);
3721 3722 3723 3724 3725 3726 3727 3728 3729
    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;

3730 3731 3732
    int r;
    DBT found_key = { .data = (void *) key, .size = keylen };
    DBT found_val = { .data = (void *) val, .size = vallen };
3733 3734 3735

    //Call application-layer callback if found.
    if (key!=NULL) {
3736
        context->r_user_callback = context->f(&found_key, &found_val, context->f_extra);
3737
        r = context->r_user_callback;
3738 3739
    } else
        r = 0;
3740 3741 3742 3743 3744 3745 3746 3747

    //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);
3748
    HANDLE_CURSOR_ILLEGAL_WORKING_PARENT_TXN(c);
3749 3750

    QUERY_CONTEXT_S context; //Describes the context of this query.
3751
    num_sequential_queries++;   // accountability
3752
    c_query_context_init(&context, c, flag, f, extra); 
3753
    //toku_brt_cursor_current will call c_getf_current_callback(..., context) (if query is successful)
3754
    int r = toku_brt_cursor_current(dbc_struct_i(c)->c, DB_CURRENT_BINDING, c_getf_current_callback, &context);
3755
    if (r == TOKUDB_USER_CALLBACK_ERROR) r = context.base.r_user_callback;
3756
    c_query_context_destroy(&context);
3757 3758 3759 3760 3761 3762 3763 3764
    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);
3765
    HANDLE_CURSOR_ILLEGAL_WORKING_PARENT_TXN(c);
3766

3767
    int r = 0;
3768
    QUERY_CONTEXT_WITH_INPUT_S context; //Describes the context of this query.
3769
    num_point_queries++;   // accountability
3770
    query_context_with_input_init(&context, c, flag, key, NULL, f, extra); 
3771 3772 3773 3774 3775 3776 3777 3778 3779 3780 3781 3782
    while (r == 0) {
        //toku_brt_cursor_set will call c_getf_set_callback(..., context) (if query is successful)
        r = toku_brt_cursor_set(dbc_struct_i(c)->c, key, c_getf_set_callback, &context);
        if (r == DB_LOCK_NOTGRANTED)
            r = toku_lock_request_wait_with_default_timeout(&context.base.lock_request, c->dbp->i->lt);
        else {
            if (r == TOKUDB_USER_CALLBACK_ERROR)
                r = context.base.r_user_callback;
            break;
        }
    }
    query_context_base_destroy(&context.base);
3783 3784 3785 3786 3787 3788 3789 3790 3791 3792
    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;
3793 3794
    DBT found_key = { .data = (void *) key, .size = keylen };
    DBT found_val = { .data = (void *) val, .size = vallen };
3795 3796 3797 3798 3799

    //Lock:
    //  left(key,val)  = (input_key, -infinity)
    //  right(key,val) = (input_key, found ? found_val : infinity)
    if (context->do_locking) {
3800 3801
        r = start_range_lock(context->db, context->txn, super_context->input_key, super_context->input_key, 
                             context->is_write_op ? LOCK_REQUEST_WRITE : LOCK_REQUEST_READ, &context->lock_request);
3802 3803
    } else 
        r = 0;
3804 3805 3806

    //Call application-layer callback if found and locks were successfully obtained.
    if (r==0 && key!=NULL) {
3807
        context->r_user_callback = context->f(&found_key, &found_val, context->f_extra);
3808
        r = context->r_user_callback;
3809 3810 3811 3812 3813 3814 3815 3816 3817 3818 3819
    }

    //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);
3820
    HANDLE_CURSOR_ILLEGAL_WORKING_PARENT_TXN(c);
3821

3822
    int r = 0;
3823
    QUERY_CONTEXT_WITH_INPUT_S context; //Describes the context of this query.
3824
    num_point_queries++;   // accountability
3825
    query_context_with_input_init(&context, c, flag, key, NULL, f, extra); 
3826 3827 3828 3829 3830 3831 3832 3833 3834 3835 3836 3837
    while (r == 0) {
        //toku_brt_cursor_set_range will call c_getf_set_range_callback(..., context) (if query is successful)
        r = toku_brt_cursor_set_range(dbc_struct_i(c)->c, key, c_getf_set_range_callback, &context);
        if (r == DB_LOCK_NOTGRANTED)
            r = toku_lock_request_wait_with_default_timeout(&context.base.lock_request, c->dbp->i->lt);
        else {
            if (r == TOKUDB_USER_CALLBACK_ERROR)
                r = context.base.r_user_callback;
            break;
        }
    }
    query_context_base_destroy(&context.base);
3838 3839 3840 3841 3842 3843 3844 3845 3846 3847
    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;
3848 3849
    DBT found_key = { .data = (void *) key, .size = keylen };
    DBT found_val = { .data = (void *) val, .size = vallen };
3850 3851 3852 3853 3854 3855

    //Lock:
    //  left(key,val)  = (input_key, -infinity)
    //  right(key) = found ? found_key : infinity
    //  right(val) = found ? found_val : infinity
    if (context->do_locking) {
3856 3857
        const DBT *left_key = super_context->input_key;
        const DBT *right_key = key != NULL ? &found_key : toku_lt_infinity;
3858 3859
        r = start_range_lock(context->db, context->txn, left_key, right_key, 
                             context->is_write_op ? LOCK_REQUEST_WRITE : LOCK_REQUEST_READ, &context->lock_request);
3860 3861
    } else 
        r = 0;
3862 3863 3864

    //Call application-layer callback if found and locks were successfully obtained.
    if (r==0 && key!=NULL) {
3865
        context->r_user_callback = context->f(&found_key, &found_val, context->f_extra);
3866
        r = context->r_user_callback;
3867 3868 3869 3870 3871 3872
    }

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

3873 3874 3875 3876 3877 3878 3879
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);

3880
    int r = 0;
3881
    QUERY_CONTEXT_WITH_INPUT_S context; //Describes the context of this query.
3882
    num_point_queries++;   // accountability
3883
    query_context_with_input_init(&context, c, flag, key, NULL, f, extra); 
3884 3885 3886 3887 3888 3889 3890 3891 3892 3893 3894 3895
    while (r == 0) {
        //toku_brt_cursor_set_range_reverse will call c_getf_set_range_reverse_callback(..., context) (if query is successful)
        r = toku_brt_cursor_set_range_reverse(dbc_struct_i(c)->c, key, c_getf_set_range_reverse_callback, &context);
        if (r == DB_LOCK_NOTGRANTED)
            r = toku_lock_request_wait_with_default_timeout(&context.base.lock_request, c->dbp->i->lt);
        else {
            if (r == TOKUDB_USER_CALLBACK_ERROR)
                r = context.base.r_user_callback;
            break;
        }
    }
    query_context_base_destroy(&context.base);
3896 3897 3898 3899 3900 3901 3902 3903 3904 3905
    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;
3906 3907
    DBT found_key = { .data = (void *) key, .size = keylen };
    DBT found_val = { .data = (void *) val, .size = vallen };
3908 3909 3910 3911 3912 3913

    //Lock:
    //  left(key) = found ? found_key : -infinity
    //  left(val) = found ? found_val : -infinity
    //  right(key,val)  = (input_key, infinity)
    if (context->do_locking) {
3914 3915
        const DBT *left_key = key != NULL ? &found_key : toku_lt_neg_infinity;
        const DBT *right_key = super_context->input_key;
3916 3917
        r = start_range_lock(context->db, context->txn, left_key, right_key, 
                             context->is_write_op ? LOCK_REQUEST_WRITE : LOCK_REQUEST_READ, &context->lock_request);
3918 3919
    } else 
        r = 0;
3920 3921 3922

    //Call application-layer callback if found and locks were successfully obtained.
    if (r==0 && key!=NULL) {
3923
        context->r_user_callback = context->f(&found_key, &found_val, context->f_extra);
3924
        r = context->r_user_callback;
3925 3926 3927 3928 3929 3930
    }

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

3931
static int toku_c_close(DBC * c) {
3932 3933
    HANDLE_PANICKED_DB(c->dbp);
    HANDLE_CURSOR_ILLEGAL_WORKING_PARENT_TXN(c);
3934 3935 3936 3937 3938 3939
    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
3940
    toku_free(c);
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
3941 3942 3943
    return r;
}

3944 3945 3946 3947
// 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) {
3948 3949
    HANDLE_PANICKED_DB(cursor->dbp);
    HANDLE_CURSOR_ILLEGAL_WORKING_PARENT_TXN(cursor);
Rich Prohaska's avatar
Rich Prohaska committed
3950 3951
    int r;
    DBC *count_cursor = 0;
3952
    DBT currentkey;
Rich Prohaska's avatar
Rich Prohaska committed
3953

3954
    init_dbt_realloc(&currentkey);
3955
    u_int32_t lock_flags = get_cursor_prelocked_flags(flags, cursor);
3956
    flags &= ~lock_flags;
Rich Prohaska's avatar
Rich Prohaska committed
3957 3958 3959 3960
    if (flags != 0) {
        r = EINVAL; goto finish;
    }

3961
    r = toku_c_get_current_unconditional(cursor, lock_flags, &currentkey, NULL);
Rich Prohaska's avatar
Rich Prohaska committed
3962
    if (r != 0) goto finish;
3963 3964 3965 3966 3967 3968

    //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
3969
    
3970
    r = toku_db_cursor(cursor->dbp, dbc_struct_i(cursor)->txn, &count_cursor, 0, 0);
Rich Prohaska's avatar
Rich Prohaska committed
3971 3972
    if (r != 0) goto finish;

3973
    r = toku_c_getf_set(count_cursor, lock_flags, &currentkey, ydb_getf_do_nothing, NULL);
3974 3975 3976 3977
    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
3978
    }
3979
    r = 0;
Rich Prohaska's avatar
Rich Prohaska committed
3980 3981 3982 3983 3984 3985 3986 3987
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
3988

3989 3990
///////////
//db_getf_XXX is equivalent to c_getf_XXX, without a persistent cursor
Yoni Fogel's avatar
 
Yoni Fogel committed
3991

3992 3993
static int
db_getf_set(DB *db, DB_TXN *txn, u_int32_t flags, DBT *key, YDB_CALLBACK_FUNCTION f, void *extra) {
3994 3995
    HANDLE_PANICKED_DB(db);
    HANDLE_DB_ILLEGAL_WORKING_PARENT_TXN(db, txn);
3996
    DBC *c;
3997
    uint32_t create_flags = flags & (DB_ISOLATION_FLAGS | DB_RMW);
3998
    flags &= ~DB_ISOLATION_FLAGS;
3999
    int r = toku_db_cursor(db, txn, &c, create_flags, 1);
4000 4001 4002 4003
    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
4004
    }
4005
    return r;
Yoni Fogel's avatar
 
Yoni Fogel committed
4006 4007
}

4008 4009
static int
toku_db_del(DB *db, DB_TXN *txn, DBT *key, u_int32_t flags) {
4010
    HANDLE_PANICKED_DB(db);
4011
    HANDLE_DB_ILLEGAL_WORKING_PARENT_TXN(db, txn);
4012

4013 4014
    u_int32_t unchecked_flags = flags;
    //DB_DELETE_ANY means delete regardless of whether it exists in the db.
4015
    BOOL error_if_missing = (BOOL)(!(flags&DB_DELETE_ANY));
4016
    unchecked_flags &= ~DB_DELETE_ANY;
4017
    u_int32_t lock_flags = get_prelocked_flags(flags);
4018
    unchecked_flags &= ~lock_flags;
4019
    BOOL do_locking = (BOOL)(db->i->lt && !(lock_flags&DB_PRELOCKED_WRITE));
4020
    BOOL do_dir_locking = !(lock_flags&DB_PRELOCKED_FILE_READ);
4021

4022
    int r = 0;
4023 4024 4025
    if (unchecked_flags!=0) 
        r = EINVAL;

4026
    if (r == 0 && do_dir_locking) {
4027
        r = toku_grab_read_lock_on_directory(db, txn);
Zardosht Kasheff's avatar
Zardosht Kasheff committed
4028
    }
4029
    if (r == 0 && error_if_missing) {
4030
        //Check if the key exists in the db.
4031
        r = db_getf_set(db, txn, lock_flags|DB_SERIALIZABLE|DB_RMW, key, ydb_getf_do_nothing, NULL);
4032
    }
4033
    if (r == 0 && do_locking) {
4034
        //Do locking if necessary.
4035
        r = get_point_write_lock(db, txn, key);
4036
    }
4037
    if (r == 0) {
4038
        //Do the actual deleting.
4039
        r = toku_brt_delete(db->i->brt, key, txn ? db_txn_struct_i(txn)->tokutxn : 0);
4040
    }
4041

4042 4043 4044
    if (r == 0) 
        num_deletes++;       // accountability 
    else
Zardosht Kasheff's avatar
Zardosht Kasheff committed
4045
        num_deletes_fail++;
4046 4047 4048 4049 4050 4051 4052 4053 4054 4055 4056

    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;
}

4057 4058 4059 4060 4061 4062 4063 4064
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;
}

4065
static int
4066
log_del_multiple(DB_TXN *txn, DB *src_db, const DBT *key, const DBT *val, uint32_t num_dbs, BRT brts[], DBT keys[]) {
4067 4068 4069 4070
    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;
Rich Prohaska's avatar
Rich Prohaska committed
4071 4072
        uint32_t del_multiple_size = key->size + val->size + num_dbs*sizeof (uint32_t) + toku_log_enq_delete_multiple_overhead;
        uint32_t del_single_sizes = sum_size(num_dbs, keys, toku_log_enq_delete_any_overhead);
4073 4074 4075 4076 4077 4078
        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);
        }
4079 4080 4081 4082
    }
    return r;
}

4083 4084 4085 4086 4087 4088 4089 4090 4091
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;
}

4092
static int
4093 4094
do_del_multiple(DB_TXN *txn, uint32_t num_dbs, DB *db_array[], DBT keys[], DB *src_db, const DBT *src_key) {
    src_db = src_db; src_key = src_key;
4095 4096 4097 4098
    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];
4099 4100 4101

        // if db is being indexed by an indexer, then insert a delete message into the db if the src key is to the left or equal to the 
        // indexers cursor.  we have to get the src_db from the indexer and find it in the db_array.
4102 4103
	int do_delete = TRUE;
	DB_INDEXER *indexer = toku_db_get_indexer(db);
4104
	if (indexer) { // if this db is the index under construction
4105 4106 4107 4108 4109 4110 4111 4112 4113 4114 4115
            DB *indexer_src_db = toku_indexer_get_src_db(indexer);
            invariant(indexer_src_db != NULL);
            const DBT *indexer_src_key;
            if (src_db == indexer_src_db)
                indexer_src_key = src_key;
            else {
                uint32_t which_src_db = lookup_src_db(num_dbs, db_array, indexer_src_db);
                invariant(which_src_db < num_dbs);
                indexer_src_key = &keys[which_src_db];
            }
            do_delete = !toku_indexer_is_key_right_of_le_cursor(indexer, indexer_src_db, indexer_src_key);
4116 4117 4118 4119
        }
	if (r == 0 && do_delete) {
            r = toku_brt_maybe_delete(db->i->brt, &keys[which_db], ttxn, FALSE, ZERO_LSN, FALSE);
        }
Zardosht Kasheff's avatar
Zardosht Kasheff committed
4120
    }
4121
    return r;
Rich Prohaska's avatar
Rich Prohaska committed
4122 4123
}

4124
static int
4125 4126 4127 4128
env_del_multiple(
    DB_ENV *env, 
    DB *src_db, 
    DB_TXN *txn, 
4129 4130
    const DBT *src_key, 
    const DBT *src_val, 
4131 4132 4133
    uint32_t num_dbs, 
    DB **db_array, 
    DBT *keys, 
4134
    uint32_t *flags_array) 
4135
{
4136
    int r;
4137
    DBT del_keys[num_dbs];
4138 4139 4140 4141

    HANDLE_PANICKED_ENV(env);

    if (!txn) {
4142 4143 4144
        r = EINVAL;
        goto cleanup;
    }
4145
    if (!env->i->generate_row_for_del) {
4146 4147 4148 4149
        r = EINVAL;
        goto cleanup;
    }

4150 4151 4152 4153 4154 4155 4156 4157
    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++) {
4158
        DB *db = db_array[which_db];
4159 4160
        lock_flags[which_db] = get_prelocked_flags(flags_array[which_db]);
        remaining_flags[which_db] = flags_array[which_db] & ~lock_flags[which_db];
4161

4162 4163 4164 4165 4166
        //Do locking if necessary.
        if (!(lock_flags[which_db] & DB_PRELOCKED_FILE_READ)) {
            r = toku_grab_read_lock_on_directory(db, txn);
            if (r != 0) goto cleanup;
        }
4167
        if (db == src_db) {
4168
            del_keys[which_db] = *src_key;
4169 4170
        }
        else {
4171
        //Generate the key
4172
            r = env->i->generate_row_for_del(db, src_db, &keys[which_db], src_key, src_val);
4173 4174 4175
            if (r != 0) goto cleanup;
            del_keys[which_db] = keys[which_db];
        }
4176 4177 4178 4179 4180 4181 4182 4183

        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.
4184
            r = db_getf_set(db, txn, lock_flags[which_db]|DB_SERIALIZABLE|DB_RMW, &del_keys[which_db], ydb_getf_do_nothing, NULL);
4185
            if (r != 0) goto cleanup;
4186 4187 4188 4189 4190
        }

        //Do locking if necessary.
        if (db->i->lt && !(lock_flags[which_db] & DB_PRELOCKED_WRITE)) {
            //Needs locking
4191
            r = get_point_write_lock(db, txn, &del_keys[which_db]);
4192
            if (r != 0) goto cleanup;
4193 4194 4195
        }
        brts[which_db] = db->i->brt;
    }
4196 4197

    if (num_dbs == 1)
4198
        r = log_del_single(txn, brts[0], &del_keys[0]);
4199
    else
4200
        r = log_del_multiple(txn, src_db, src_key, src_val, num_dbs, brts, del_keys);
4201 4202

    if (r == 0) 
4203
        r = do_del_multiple(txn, num_dbs, db_array, del_keys, src_db, src_key);
4204 4205 4206
    }

cleanup:
4207 4208 4209 4210
    if (r == 0)
        num_multi_deletes += num_dbs;
    else
        num_multi_deletes_fail += num_dbs;
4211 4212 4213 4214
    return r;
}


4215 4216
static int 
locked_c_get(DBC * c, DBT * key, DBT * data, u_int32_t flag) {
4217 4218
    toku_ydb_lock(); int r = toku_c_get(c, key, data, flag); toku_ydb_unlock();
    return r;
Rich Prohaska's avatar
Rich Prohaska committed
4219 4220
}

4221 4222
static int 
locked_c_close(DBC * c) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
4223
    toku_ydb_lock(); int r = toku_c_close(c); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
4224 4225
}

4226 4227
static int 
locked_c_count(DBC *cursor, db_recno_t *count, u_int32_t flags) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
4228
    toku_ydb_lock(); int r = toku_c_count(cursor, count, flags); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
4229 4230
}

4231 4232
static int 
locked_c_del(DBC * c, u_int32_t flags) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
4233
    toku_ydb_lock(); int r = toku_c_del(c, flags); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
4234 4235
}

4236
static int locked_c_pre_acquire_range_lock(DBC *dbc, const DBT *key_left, const DBT *key_right);
4237

4238 4239
static int 
toku_db_cursor(DB * db, DB_TXN * txn, DBC ** c, u_int32_t flags, int is_temporary_cursor) {
4240
    HANDLE_PANICKED_DB(db);
4241
    HANDLE_DB_ILLEGAL_WORKING_PARENT_TXN(db, txn);
4242
    DB_ENV* env = db->dbenv;
Zardosht Kasheff's avatar
Zardosht Kasheff committed
4243
    int r;
4244
    size_t result_size = sizeof(DBC)+sizeof(struct __toku_dbc_internal); // internal stuff stuck on the end
4245
    if (flags & ~(DB_SERIALIZABLE | DB_INHERIT_ISOLATION | DB_RMW)) {
4246 4247 4248
        return toku_ydb_do_error(
            env, 
            EINVAL, 
4249
            "Invalid flags set for toku_db_cursor\n"
4250 4251
            );
    }
Zardosht Kasheff's avatar
Zardosht Kasheff committed
4252
    r = toku_grab_read_lock_on_directory(db, txn);
4253
    if (r != 0) 
Zardosht Kasheff's avatar
Zardosht Kasheff committed
4254 4255
        return r;
    
4256
    DBC *result = toku_malloc(result_size);
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
4257 4258
    if (result == 0)
        return ENOMEM;
4259
    memset(result, 0, result_size);
Yoni Fogel's avatar
Yoni Fogel committed
4260 4261 4262 4263 4264
#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
4265 4266
    SCRS(c_getf_first);
    SCRS(c_getf_last);
Yoni Fogel's avatar
Yoni Fogel committed
4267
    SCRS(c_getf_next);
4268 4269 4270 4271 4272
    SCRS(c_getf_prev);
    SCRS(c_getf_current);
    SCRS(c_getf_current_binding);
    SCRS(c_getf_set);
    SCRS(c_getf_set_range);
4273
    SCRS(c_getf_set_range_reverse);
4274
    SCRS(c_pre_acquire_range_lock);
Yoni Fogel's avatar
Yoni Fogel committed
4275
#undef SCRS
4276 4277 4278

#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
4279
    assert(result->i);
4280
#endif
4281
    result->dbp = db;
4282

4283 4284 4285
    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};
4286
    if (is_temporary_cursor) {
4287 4288
	dbc_struct_i(result)->skey = &db->i->skey;
	dbc_struct_i(result)->sval = &db->i->sval;
4289
    } else {
4290 4291
	dbc_struct_i(result)->skey = &dbc_struct_i(result)->skey_s;
	dbc_struct_i(result)->sval = &dbc_struct_i(result)->sval_s;
4292
    }
4293 4294 4295 4296
    if (flags & DB_SERIALIZABLE) {
        dbc_struct_i(result)->iso = TOKU_ISO_SERIALIZABLE;
    } else {
        dbc_struct_i(result)->iso = txn ? db_txn_struct_i(txn)->iso : TOKU_ISO_SERIALIZABLE;
4297
    }
4298
    dbc_struct_i(result)->rmw = (flags & DB_RMW) != 0;
4299
    BOOL is_snapshot_read = FALSE;
4300
    if (txn) {
4301 4302
        is_snapshot_read = (dbc_struct_i(result)->iso == TOKU_ISO_READ_COMMITTED || 
                            dbc_struct_i(result)->iso == TOKU_ISO_SNAPSHOT);
4303
    }
Zardosht Kasheff's avatar
Zardosht Kasheff committed
4304
    r = toku_brt_cursor(
4305
        db->i->brt, 
4306 4307 4308
        &dbc_struct_i(result)->c,
        txn ? db_txn_struct_i(txn)->tokutxn : NULL,
        is_snapshot_read
4309
        );
4310 4311 4312 4313 4314 4315 4316 4317 4318 4319 4320
    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
4321 4322
}

4323 4324
static inline int 
db_thread_need_flags(DBT *dbt) {
Rich Prohaska's avatar
Rich Prohaska committed
4325 4326 4327
    return (dbt->flags & (DB_DBT_MALLOC+DB_DBT_REALLOC+DB_DBT_USERMEM)) == 0;
}

4328 4329
static int 
toku_db_get (DB * db, DB_TXN * txn, DBT * key, DBT * data, u_int32_t flags) {
4330
    HANDLE_PANICKED_DB(db);
4331
    HANDLE_DB_ILLEGAL_WORKING_PARENT_TXN(db, txn);
Yoni Fogel's avatar
Yoni Fogel committed
4332
    int r;
4333
    u_int32_t iso_flags = flags & DB_ISOLATION_FLAGS;
4334

Rich Prohaska's avatar
Rich Prohaska committed
4335
    if ((db->i->open_flags & DB_THREAD) && db_thread_need_flags(data))
4336 4337
        return EINVAL;

4338
    u_int32_t lock_flags = flags & (DB_PRELOCKED | DB_PRELOCKED_WRITE | DB_PRELOCKED_FILE_READ);
Yoni Fogel's avatar
Yoni Fogel committed
4339
    flags &= ~lock_flags;
4340
    flags &= ~DB_ISOLATION_FLAGS;
4341 4342 4343
    // And DB_GET_BOTH is no longer supported. #2862.
    if (flags != 0) return EINVAL;

Yoni Fogel's avatar
Yoni Fogel committed
4344 4345

    DBC *dbc;
4346
    r = toku_db_cursor(db, txn, &dbc, iso_flags, 1);
Yoni Fogel's avatar
Yoni Fogel committed
4347
    if (r!=0) return r;
4348
    u_int32_t c_get_flags = DB_SET;
Yoni Fogel's avatar
Yoni Fogel committed
4349
    r = toku_c_get(dbc, key, data, c_get_flags | lock_flags);
Yoni Fogel's avatar
Yoni Fogel committed
4350 4351
    int r2 = toku_c_close(dbc);
    return r ? r : r2;
4352 4353
}

Rich Prohaska's avatar
Rich Prohaska committed
4354
#if 0
4355 4356
static int 
toku_db_key_range(DB * db, DB_TXN * txn, DBT * dbt, DB_KEY_RANGE * kr, u_int32_t flags) {
4357
    HANDLE_PANICKED_DB(db);
4358
    HANDLE_DB_ILLEGAL_WORKING_PARENT_TXN(db, txn);
4359
    txn=txn; dbt=dbt; kr=kr; flags=flags;
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
4360
    toku_ydb_barf();
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
4361
    abort();
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
4362
}
Rich Prohaska's avatar
Rich Prohaska committed
4363
#endif
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
4364

4365 4366
static int 
toku_db_lt_panic(DB* db, int r) {
4367
    assert(r!=0);
Yoni Fogel's avatar
Yoni Fogel committed
4368 4369
    assert(db && db->i && db->dbenv && db->dbenv->i);
    DB_ENV* env = db->dbenv;
4370 4371 4372 4373
    char * panic_string;

    if (r < 0) panic_string = toku_lt_strerror((TOKU_LT_ERROR)r);
    else       panic_string = "Error in locktree.\n";
4374

4375
    env_panic(env, r, panic_string);
4376

4377
    return toku_ydb_do_error(env, r, "%s", panic_string);
Yoni Fogel's avatar
Yoni Fogel committed
4378 4379
}

4380 4381
static int 
toku_txn_add_lt(DB_TXN* txn, toku_lock_tree* lt) {
Yoni Fogel's avatar
Yoni Fogel committed
4382
    int r = ENOSYS;
Yoni Fogel's avatar
Yoni Fogel committed
4383
    assert(txn && lt);
4384
    toku_lth* lth = db_txn_struct_i(txn)->lth;
Yoni Fogel's avatar
Yoni Fogel committed
4385 4386 4387 4388 4389
    assert(lth);

    toku_lock_tree* find = toku_lth_find(lth, lt);
    if (find) {
        assert(find == lt);
Yoni Fogel's avatar
Yoni Fogel committed
4390 4391
        r = 0;
        goto cleanup;
Yoni Fogel's avatar
Yoni Fogel committed
4392
    }
Yoni Fogel's avatar
Yoni Fogel committed
4393 4394 4395 4396 4397 4398
    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
4399 4400 4401
    return r;
}

4402 4403
static toku_dbt_cmp 
toku_db_get_compare_fun(DB* db) {
Yoni Fogel's avatar
Yoni Fogel committed
4404 4405
    return db->i->brt->compare_fun;
}
Yoni Fogel's avatar
Yoni Fogel committed
4406

4407

4408 4409 4410 4411 4412
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 {
4413 4414 4415 4416 4417 4418
        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);
    }
4419 4420
    return r;
}
4421

4422 4423 4424 4425 4426 4427 4428 4429
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)) {
4430 4431
            char c = *dname++;
            *hint++ = c;
4432 4433 4434 4435 4436 4437 4438
            underscored = FALSE;
        }
        else {
            if (!underscored)
                *hint++ = '_';
            dname++;
            underscored = TRUE;
4439 4440
        }
    }
4441 4442 4443
    *hint = '\0';
}

4444

4445
// n < 0  means to ignore mark and ignore n
4446
// n >= 0 means to include mark ("_B_" or "_P_") with hex value of n in iname
4447
// (intended for use by loader, which will create many inames using one txnid).
4448
static char *
4449
create_iname(DB_ENV *env, u_int64_t id, char *hint, char *mark, int n) {
4450
    int bytes;
4451
    char inamebase[strlen(hint) +
4452 4453 4454
		   8 +  // hex file format version
		   16 + // hex id (normally the txnid)
		   8  + // hex value of n if non-neg
4455
		   sizeof("_B___.tokudb")]; // extra pieces
4456
    if (n < 0)
4457 4458 4459
	bytes = snprintf(inamebase, sizeof(inamebase),
                         "%s_%"PRIx64"_%"PRIx32            ".tokudb",
                         hint, id, BRT_LAYOUT_VERSION);
4460 4461
    else {
	invariant(strlen(mark) == 1);
4462
	bytes = snprintf(inamebase, sizeof(inamebase),
4463 4464 4465
                         "%s_%"PRIx64"_%"PRIx32"_%s_%"PRIx32".tokudb",
                         hint, id, BRT_LAYOUT_VERSION, mark, n);
    }
4466 4467 4468 4469
    assert(bytes>0);
    assert(bytes<=(int)sizeof(inamebase)-1);
    char *rval;
    if (env->i->data_dir)
Yoni Fogel's avatar
Yoni Fogel committed
4470
        rval = toku_construct_full_name(2, env->i->data_dir, inamebase);
4471
    else
Yoni Fogel's avatar
Yoni Fogel committed
4472
        rval = toku_construct_full_name(1, inamebase);
4473 4474
    assert(rval);
    return rval;
4475 4476
}

4477 4478 4479 4480 4481 4482 4483 4484 4485 4486 4487 4488 4489

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) {
4490
    HANDLE_PANICKED_DB(db);
4491
    HANDLE_DB_ILLEGAL_WORKING_PARENT_TXN(db, txn);
4492
    if (dbname!=NULL) 
4493
        return db_open_subdb(db, txn, fname, dbname, dbtype, flags, mode);
4494

4495
    // at this point fname is the dname
4496 4497
    //This code ONLY supports single-db files.
    assert(dbname==NULL);
4498
    const char * dname = fname;  // db_open_subdb() converts (fname, dbname) to dname
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
4499

4500 4501 4502
    ////////////////////////////// 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
4503
    int r;
Yoni Fogel's avatar
Yoni Fogel committed
4504
    if (dbtype!=DB_BTREE && dbtype!=DB_UNKNOWN) return EINVAL;
4505 4506
    int is_db_excl    = flags & DB_EXCL;    unused_flags&=~DB_EXCL;
    int is_db_create  = flags & DB_CREATE;  unused_flags&=~DB_CREATE;
4507
    int is_db_hot_index  = flags & DB_IS_HOT_INDEX;  unused_flags&=~DB_IS_HOT_INDEX;
4508

4509
    //We support READ_UNCOMMITTED and READ_COMMITTED whether or not the flag is provided.
4510
                                            unused_flags&=~DB_READ_UNCOMMITTED;
4511
                                            unused_flags&=~DB_READ_COMMITTED;
4512
                                            unused_flags&=~DB_SERIALIZABLE;
4513 4514 4515 4516 4517 4518 4519 4520 4521 4522 4523 4524 4525 4526 4527 4528 4529 4530
    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) {
4531 4532
        r = toku_txn_begin(db->dbenv, txn, &child, DB_TXN_NOSYNC, 1);
        assert(r==0);
4533 4534 4535 4536 4537 4538
    }

    // 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
4539
    DBT iname_dbt;  // holds iname_in_env
4540 4541
    toku_fill_dbt(&dname_dbt, dname, strlen(dname)+1);
    init_dbt_realloc(&iname_dbt);  // sets iname_dbt.data = NULL
4542
    r = toku_db_get(db->dbenv->i->directory, child, &dname_dbt, &iname_dbt, DB_SERIALIZABLE);  // allocates memory for iname
4543 4544 4545 4546 4547 4548 4549
    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) {
4550
        char hint[strlen(dname) + 1];
4551

4552 4553 4554 4555 4556 4557 4558
        // 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);
4559
        iname = create_iname(db->dbenv, id, hint, NULL, -1);  // allocated memory for iname
4560
        toku_fill_dbt(&iname_dbt, iname, strlen(iname) + 1);
4561
        //
4562
        // 0 for performance only, avoid unnecessary query
4563 4564 4565
        // if we are creating a hot index, per #3166, we do not want the write lock  in directory grabbed.
        // directory read lock is grabbed in toku_db_get above
        //
4566
        u_int32_t put_flags = 0 | ((is_db_hot_index) ? DB_PRELOCKED_WRITE : 0); 
4567
        r = toku_db_put(db->dbenv->i->directory, child, &dname_dbt, &iname_dbt, put_flags);  
4568 4569 4570 4571
    }

    // we now have an iname
    if (r == 0) {
4572
        r = db_open_iname(db, child, iname, flags, mode);
4573 4574 4575 4576 4577 4578 4579 4580 4581 4582
        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) {
4583 4584 4585
        // close txn
        if (r == 0) {  // commit
            r = toku_txn_commit(child, DB_TXN_NOSYNC, NULL, NULL);
4586
            invariant(r==0);  // TODO panic
4587 4588 4589
        }
        else {         // abort
            int r2 = toku_txn_abort(child, NULL, NULL);
4590
            invariant(r2==0);  // TODO panic
4591
        }
4592 4593 4594 4595 4596 4597
    }

    return r;
}

static int 
4598
db_open_iname(DB * db, DB_TXN * txn, const char *iname_in_env, u_int32_t flags, int mode) {
4599 4600 4601 4602 4603 4604 4605 4606
    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;
    }
4607 4608 4609 4610
    if (db->dbenv->i->update_function) {
        r = toku_brt_set_update(db->i->brt,db->dbenv->i->update_function);
        assert(r==0);
    }
4611 4612 4613
    BOOL need_locktree = (BOOL)((db->dbenv->i->open_flags & DB_INIT_LOCK) &&
                                (db->dbenv->i->open_flags & DB_INIT_TXN));

4614 4615
    int is_db_excl    = flags & DB_EXCL;    flags&=~DB_EXCL;
    int is_db_create  = flags & DB_CREATE;  flags&=~DB_CREATE;
4616
    //We support READ_UNCOMMITTED and READ_COMMITTED whether or not the flag is provided.
4617
                                            flags&=~DB_READ_UNCOMMITTED;
4618
                                            flags&=~DB_READ_COMMITTED;
4619
                                            flags&=~DB_SERIALIZABLE;
4620
                                            flags&=~DB_IS_HOT_INDEX;
4621
    if (flags & ~DB_THREAD) return EINVAL; // unknown flags
4622 4623

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

4625 4626 4627 4628 4629 4630
    /* 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;

4631
    if (db_opened(db))
4632
        return EINVAL;              /* It was already open. */
Yoni Fogel's avatar
Yoni Fogel committed
4633
    
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
4634 4635
    db->i->open_flags = flags;
    db->i->open_mode = mode;
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
4636

Yoni Fogel's avatar
Yoni Fogel committed
4637
    r = toku_brt_open(db->i->brt, iname_in_env,
4638
		      is_db_create, is_db_excl,
4639
		      db->dbenv->i->cachetable,
4640
		      txn ? db_txn_struct_i(txn)->tokutxn : NULL_TXN,
4641
		      db);
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
4642 4643 4644
    if (r != 0)
        goto error_cleanup;

4645
    db->i->opened = 1;
Yoni Fogel's avatar
Yoni Fogel committed
4646
    if (need_locktree) {
4647
	db->i->dict_id = toku_brt_get_dictionary_id(db->i->brt);
4648
        r = toku_ltm_get_lt(db->dbenv->i->ltm, &db->i->lt, db->i->dict_id, db);
Yoni Fogel's avatar
Yoni Fogel committed
4649
        if (r!=0) { goto error_cleanup; }
Yoni Fogel's avatar
Yoni Fogel committed
4650
    }
4651 4652 4653
    //Add to transaction's list of 'must close' if necessary.
    if (txn) {
        //Do last so we don't have to undo.
4654
        toku_list_push(&db_txn_struct_i(txn)->dbs_that_must_close_before_abort,
4655 4656
                  &db->i->dbs_that_must_close_before_abort);
    }
Yoni Fogel's avatar
Yoni Fogel committed
4657

Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
4658
    return 0;
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
4659 4660
 
error_cleanup:
4661
    db->i->dict_id = DICTIONARY_ID_NONE;
4662
    db->i->opened = 0;
Yoni Fogel's avatar
Yoni Fogel committed
4663
    if (db->i->lt) {
4664
        toku_lt_remove_db_ref(db->i->lt, db);
Yoni Fogel's avatar
Yoni Fogel committed
4665 4666
        db->i->lt = NULL;
    }
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
4667
    return r;
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
4668
}
Bradley C. Kuszmaul's avatar
Bradley C. Kuszmaul committed
4669

4670 4671 4672 4673
//Return 0 if proposed pair do not violate size constraints of DB
//(insertion is legal)
//Return non zero otherwise.
static int
4674
db_put_check_size_constraints(DB *db, const DBT *key, const DBT *val) {
4675
    int r;
4676

4677
    //Check limits on size of key and val.
4678
    unsigned int nodesize;
4679
    r = toku_brt_get_nodesize(db->i->brt, &nodesize); assert(r == 0);
4680

4681 4682 4683 4684 4685 4686
    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);
    
4687 4688 4689
    return r;
}

4690 4691 4692 4693 4694 4695 4696 4697 4698 4699 4700 4701 4702 4703 4704 4705 4706 4707 4708 4709 4710
//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;
}

4711 4712
//Return 0 if insert is legal
static int
4713
db_put_check_overwrite_constraint(DB *db, DB_TXN *txn, DBT *key,
4714 4715 4716
                                  u_int32_t lock_flags, u_int32_t overwrite_flag) {
    int r;

4717
    if (overwrite_flag == 0) { // 0 (yesoverwrite) does not impose constraints.
4718 4719
        r = 0;
    } else if (overwrite_flag == DB_NOOVERWRITE) {
4720 4721 4722 4723
        // Check if (key,anything) exists in dictionary.
        // If exists, fail.  Otherwise, do insert.
        // The DB_RMW flag causes the cursor to grab a write lock instead of a read lock on the key if it exists.
        r = db_getf_set(db, txn, lock_flags|DB_SERIALIZABLE|DB_RMW, key, ydb_getf_do_nothing, NULL);
4724 4725 4726 4727
        if (r == DB_NOTFOUND) 
            r = 0;
        else if (r == 0)      
            r = DB_KEYEXIST;
4728
        //Any other error is passed through.
4729
    } else if (overwrite_flag == DB_NOOVERWRITE_NO_ERROR) {
4730
        r = 0;
4731
    } else {
4732 4733
        //Other flags are not (yet) supported.
        r = EINVAL;
4734 4735 4736 4737
    }
    return r;
}

4738 4739
static int
toku_db_put(DB *db, DB_TXN *txn, DBT *key, DBT *val, u_int32_t flags) {
4740
    HANDLE_PANICKED_DB(db);
4741
    HANDLE_DB_ILLEGAL_WORKING_PARENT_TXN(db, txn);
4742
    int r = 0;
4743

4744
    u_int32_t lock_flags = get_prelocked_flags(flags);
4745
    flags &= ~lock_flags;
4746

4747 4748 4749 4750
    if (!(lock_flags & DB_PRELOCKED_FILE_READ)) {
        r = toku_grab_read_lock_on_directory(db, txn);
    }
    
4751 4752 4753
    if (r == 0)
        r = db_put_check_size_constraints(db, key, val);
    if (r == 0) {
4754
        //Do any checking required by the flags.
4755
        r = db_put_check_overwrite_constraint(db, txn, key, lock_flags, flags);
4756
    }
4757 4758
    BOOL do_locking = (BOOL)(db->i->lt && !(lock_flags&DB_PRELOCKED_WRITE));
    if (r == 0 && do_locking) {
4759
        //Do locking if necessary.
4760
        r = get_point_write_lock(db, txn, key);
4761
    }
4762
    if (r == 0) {
4763
        //Insert into the brt.
4764 4765 4766 4767 4768
        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);
4769
    }
4770

4771 4772 4773
    if (r == 0)
	num_inserts++;
    else
Zardosht Kasheff's avatar
Zardosht Kasheff committed
4774
        num_inserts_fail++;
4775 4776 4777 4778

    return r;
}

4779 4780
static int toku_db_pre_acquire_fileops_lock(DB *db, DB_TXN *txn) {
    // bad hack because some environment dictionaries do not have a dname
4781 4782
    char *dname = db->i->dname;
    if (!dname)
4783
        return 0;
4784 4785

    DBT key_in_directory = { .data = dname, .size = strlen(dname)+1 };
4786
    //Left end of range == right end of range (point lock)
4787
    int r = get_range_lock(db->dbenv->i->directory, txn, &key_in_directory, &key_in_directory, LOCK_REQUEST_WRITE);
4788 4789 4790 4791 4792 4793 4794 4795 4796 4797 4798 4799 4800 4801 4802 4803 4804 4805 4806 4807 4808 4809 4810 4811 4812 4813 4814 4815 4816
    if (r == 0)
	directory_write_locks++;
    else
	directory_write_locks_fail++;
    return r;
}

static int
toku_db_update(DB *db, DB_TXN *txn,
               const DBT *key,
               const DBT *update_function_extra,
               u_int32_t flags) {
    HANDLE_PANICKED_DB(db);
    HANDLE_DB_ILLEGAL_WORKING_PARENT_TXN(db, txn);
    int r = 0;

    u_int32_t lock_flags = get_prelocked_flags(flags);
    flags &= ~lock_flags;

    if (!(lock_flags & DB_PRELOCKED_FILE_READ)) {
        r = toku_grab_read_lock_on_directory(db, txn);
        if (r != 0) { goto cleanup; }
    }

    r = db_put_check_size_constraints(db, key, update_function_extra);
    if (r != 0) { goto cleanup; }

    BOOL do_locking = (db->i->lt && !(lock_flags & DB_PRELOCKED_WRITE));
    if (do_locking) {
4817
        r = get_point_write_lock(db, txn, key);
4818 4819 4820 4821 4822 4823 4824 4825 4826 4827 4828 4829 4830 4831 4832 4833 4834 4835 4836 4837 4838 4839 4840 4841 4842 4843 4844 4845 4846 4847 4848 4849 4850 4851 4852 4853 4854 4855 4856 4857 4858 4859 4860 4861 4862 4863 4864 4865 4866 4867 4868 4869 4870 4871 4872 4873 4874 4875 4876 4877 4878 4879 4880 4881 4882 4883 4884 4885 4886 4887 4888 4889 4890
        if (r != 0) { goto cleanup; }
    }

    TOKUTXN ttxn = txn ? db_txn_struct_i(txn)->tokutxn : NULL;
    r = toku_brt_maybe_update(db->i->brt, key, update_function_extra, ttxn,
                              FALSE, ZERO_LSN, TRUE);

cleanup:
    if (r == 0) 
	num_updates++;
    else
	num_updates_fail++;
    return r;
}


// DB_IS_RESETTING_OP is true if the dictionary should be considered as if created by this transaction.
// For example, it will be true if toku_db_update_broadcast() is used to implement a schema change (such
// as adding a column), and will be false if used simply to update all the rows of a table (such as 
// incrementing a field).
static int
toku_db_update_broadcast(DB *db, DB_TXN *txn,
                         const DBT *update_function_extra,
                         u_int32_t flags) {
    HANDLE_PANICKED_DB(db);
    HANDLE_DB_ILLEGAL_WORKING_PARENT_TXN(db, txn);
    int r = 0;

    u_int32_t lock_flags = get_prelocked_flags(flags);
    flags &= ~lock_flags;
    u_int32_t is_resetting_op_flag = flags & DB_IS_RESETTING_OP;
    flags &= is_resetting_op_flag;
    BOOL is_resetting_op = (is_resetting_op_flag != 0);
    

    if (is_resetting_op) {
        if (txn->parent != NULL) {
            r = EINVAL; // cannot have a parent if you are a resetting op
            goto cleanup;
        }
        r = toku_db_pre_acquire_fileops_lock(db, txn);
        if (r != 0) { goto cleanup; }
    }
    else if (!(lock_flags & DB_PRELOCKED_FILE_READ)) {
        r = toku_grab_read_lock_on_directory(db, txn);
        if (r != 0) { goto cleanup; }
    }

    {
        DBT null_key;
        toku_init_dbt(&null_key);
        r = db_put_check_size_constraints(db, &null_key, update_function_extra);
        if (r != 0) { goto cleanup; }
    }

    BOOL do_locking = (db->i->lt && !(lock_flags & DB_PRELOCKED_WRITE));
    if (do_locking) {
        r = toku_db_pre_acquire_table_lock(db, txn, TRUE);
        if (r != 0) { goto cleanup; }
    }

    TOKUTXN ttxn = txn ? db_txn_struct_i(txn)->tokutxn : NULL;
    r = toku_brt_maybe_update_broadcast(db->i->brt, update_function_extra, ttxn,
                                        FALSE, ZERO_LSN, TRUE, is_resetting_op);

cleanup:
    if (r == 0) 
	num_updates_broadcast++;
    else
	num_updates_broadcast_fail++;
    return r;
}

4891 4892 4893 4894 4895 4896 4897 4898 4899 4900 4901 4902 4903 4904 4905 4906 4907 4908 4909
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
4910
do_put_multiple(DB_TXN *txn, uint32_t num_dbs, DB *db_array[], DBT keys[], DBT vals[], DB *src_db, const DBT *src_key) {
4911 4912 4913 4914
    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];
4915 4916 4917

        // if db is being indexed by an indexer, then put into that db if the src key is to the left or equal to the 
        // indexers cursor.  we have to get the src_db from the indexer and find it in the db_array.
4918 4919
	int do_put = TRUE;
	DB_INDEXER *indexer = toku_db_get_indexer(db);
4920
	if (indexer) { // if this db is the index under construction
4921 4922 4923 4924 4925 4926 4927 4928 4929 4930
            DB *indexer_src_db = toku_indexer_get_src_db(indexer);
            invariant(indexer_src_db != NULL);
            const DBT *indexer_src_key;
            if (src_db == indexer_src_db)
                indexer_src_key = src_key;
            else {
                uint32_t which_src_db = lookup_src_db(num_dbs, db_array, indexer_src_db);
                invariant(which_src_db < num_dbs);
                indexer_src_key = &keys[which_src_db];
            }
4931
            do_put = !toku_indexer_is_key_right_of_le_cursor(indexer, indexer_src_db, indexer_src_key);
4932 4933 4934 4935
        }
        if (r == 0 && do_put) {
            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
4936
    }
4937
    return r;
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
4938
}
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
4939

4940
static int
4941 4942 4943 4944
env_put_multiple(
    DB_ENV *env, 
    DB *src_db, 
    DB_TXN *txn, 
4945 4946
    const DBT *src_key, 
    const DBT *src_val, 
4947 4948 4949 4950
    uint32_t num_dbs, 
    DB **db_array, 
    DBT *keys, 
    DBT *vals, 
4951
    uint32_t *flags_array) 
4952
{
4953
    int r;
4954 4955
    DBT put_keys[num_dbs];
    DBT put_vals[num_dbs];
4956 4957 4958 4959

    HANDLE_PANICKED_ENV(env);

    {
4960 4961 4962
    uint32_t lock_flags[num_dbs];
    uint32_t remaining_flags[num_dbs];
    BRT brts[num_dbs];
4963

4964 4965 4966 4967
    if (!txn || !num_dbs) {
        r = EINVAL;
        goto cleanup;
    }
4968
    if (!env->i->generate_row_for_put) {
4969 4970 4971 4972
        r = EINVAL;
        goto cleanup;
    }

4973 4974 4975
    HANDLE_ILLEGAL_WORKING_PARENT_TXN(env, txn);

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

4978 4979 4980 4981 4982 4983 4984 4985 4986
        lock_flags[which_db] = get_prelocked_flags(flags_array[which_db]);
        remaining_flags[which_db] = flags_array[which_db] & ~lock_flags[which_db];

        //Do locking if necessary.
        if (!(lock_flags[which_db] & DB_PRELOCKED_FILE_READ)) {
            r = toku_grab_read_lock_on_directory(db, txn);
            if (r != 0) goto cleanup;
        }

4987
        //Generate the row
4988
        if (db == src_db) {
4989 4990
            put_keys[which_db] = *src_key;
            put_vals[which_db] = *src_val;
4991 4992
        }
        else {
4993
            r = env->i->generate_row_for_put(db, src_db, &keys[which_db], &vals[which_db], src_key, src_val);
4994 4995 4996 4997
            if (r != 0) goto cleanup;
            put_keys[which_db] = keys[which_db];
            put_vals[which_db] = vals[which_db];            
        }
4998 4999

        // check size constraints
5000
        r = db_put_check_size_constraints(db, &put_keys[which_db], &put_vals[which_db]);
5001 5002
        if (r != 0) goto cleanup;

5003 5004
        //Check overwrite constraints
        r = db_put_check_overwrite_constraint(db, txn,
5005
                                              &put_keys[which_db],
5006
                                              lock_flags[which_db], remaining_flags[which_db]);
5007
        if (r != 0) goto cleanup;
5008 5009 5010 5011 5012
        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;
        }
5013

5014 5015 5016
        //Do locking if necessary.
        if (db->i->lt && !(lock_flags[which_db] & DB_PRELOCKED_WRITE)) {
            //Needs locking
5017
            r = get_point_write_lock(db, txn, &put_keys[which_db]);
5018
            if (r != 0) goto cleanup;
5019 5020 5021
        }
        brts[which_db] = db->i->brt;
    }
5022 5023

    if (num_dbs == 1)
5024
        r = log_put_single(txn, brts[0], &put_keys[0], &put_vals[0]);
5025
    else
5026
        r = log_put_multiple(txn, src_db, src_key, src_val, num_dbs, brts);
5027 5028
    
    if (r == 0)
5029
        r = do_put_multiple(txn, num_dbs, db_array, put_keys, put_vals, src_db, src_key);
5030

5031 5032
    }

5033
cleanup:
5034
    if (r == 0)
5035
        num_multi_inserts += num_dbs;
5036
    else
5037
        num_multi_inserts_fail += num_dbs;
5038 5039 5040 5041 5042
    return r;
}

static int
env_update_multiple(DB_ENV *env, DB *src_db, DB_TXN *txn,                                
5043 5044 5045
                    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, 
5046
                    uint32_t num_keys, DBT keys[], 
5047
                    uint32_t num_vals, DBT vals[]) {
5048 5049 5050 5051 5052 5053 5054 5055 5056 5057 5058 5059 5060 5061 5062 5063 5064 5065 5066 5067 5068 5069 5070 5071 5072 5073 5074
    int r = 0;

    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];

5075 5076 5077
        uint32_t lock_flags[num_dbs];
        uint32_t remaining_flags[num_dbs];

5078 5079
        for (uint32_t which_db = 0; which_db < num_dbs; which_db++) {
            DB *db = db_array[which_db];
5080 5081 5082 5083
            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];
5084

5085 5086 5087 5088
            if (!(lock_flags[which_db] & DB_PRELOCKED_FILE_READ)) {
                r = toku_grab_read_lock_on_directory(db, txn);
                if (r != 0) goto cleanup;
            }
5089 5090 5091
            // 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
5092

5093 5094
            // Generate the old key and val
            if (which_db + num_dbs >= num_keys) {
5095 5096
                r = ENOMEM; goto cleanup;
            }
5097 5098 5099 5100
            if (db == src_db) {
                curr_old_key = *old_src_key;
            }
            else {
5101
                r = env->i->generate_row_for_put(db, src_db, &keys[which_db + num_dbs], NULL, old_src_key, old_src_data);
5102 5103 5104
                if (r != 0) goto cleanup;
                curr_old_key = keys[which_db + num_dbs];
            }
5105
            // Generate the new key and val
5106 5107 5108
            if (which_db >= num_keys || which_db >= num_vals) {
                r = ENOMEM; goto cleanup;
            }
5109 5110 5111 5112 5113
            if (db == src_db) {
                curr_new_key = *new_src_key;
                curr_new_val = *new_src_data;
            }
            else {
5114
                r = env->i->generate_row_for_put(db, src_db, &keys[which_db], &vals[which_db], new_src_key, new_src_data);
5115 5116 5117 5118
                if (r != 0) goto cleanup;
                curr_new_key = keys[which_db];
                curr_new_val = vals[which_db];
            }
5119
            toku_dbt_cmp cmpfun = toku_db_get_compare_fun(db);
5120
            BOOL key_eq = cmpfun(db, &curr_old_key, &curr_new_key) == 0;
5121
            if (!key_eq) {
5122 5123
                //Check overwrite constraints only in the case where 
                // the keys are not equal.
5124
                // If the keys are equal, then we do not care of the flag is DB_NOOVERWRITE or 0
5125 5126 5127 5128 5129 5130 5131 5132 5133 5134
                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;
                }

5135
                // lock old key
5136
                if (db->i->lt && !(lock_flags[which_db] & DB_PRELOCKED_WRITE)) {
5137
                    r = get_point_write_lock(db, txn, &curr_old_key);
5138 5139 5140 5141
                    if (r != 0) goto cleanup;
                }
                del_dbs[n_del_dbs] = db;
                del_brts[n_del_dbs] = db->i->brt;
5142
                del_keys[n_del_dbs] = curr_old_key;
5143
                n_del_dbs++;
5144
                
5145 5146
            }

5147 5148 5149
            // 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)) {
5150 5151
            if (!key_eq || curr_new_val.size > 0) {
                r = db_put_check_size_constraints(db, &curr_new_key, &curr_new_val);
5152 5153 5154 5155
                if (r != 0) goto cleanup;

                // lock new key
                if (db->i->lt) {
5156
                    r = get_point_write_lock(db, txn, &curr_new_key);
5157 5158 5159 5160
                    if (r != 0) goto cleanup;
                }
                put_dbs[n_put_dbs] = db;
                put_brts[n_put_dbs] = db->i->brt;
5161 5162
                put_keys[n_put_dbs] = curr_new_key;
                put_vals[n_put_dbs] = curr_new_val;
5163 5164 5165 5166 5167 5168 5169 5170
                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
5171
                r = log_del_multiple(txn, src_db, old_src_key, old_src_data, n_del_dbs, del_brts, del_keys);
5172 5173
            if (r == 0)
                r = do_del_multiple(txn, n_del_dbs, del_dbs, del_keys, src_db, old_src_key);
5174 5175 5176 5177 5178 5179 5180 5181
        }

        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)
5182
                r = do_put_multiple(txn, n_put_dbs, put_dbs, put_keys, put_vals, src_db, new_src_key);
5183 5184 5185 5186
        }
    }

cleanup:
5187 5188 5189 5190
    if (r == 0)
        num_multi_updates += num_dbs;
    else
        num_multi_updates_fail += num_dbs;
5191 5192 5193 5194
    return r;
}

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

//We do not (yet?) support deleting subdbs by deleting the enclosing 'fname'
5197
static int
5198 5199 5200 5201
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 {
5202 5203 5204 5205
        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;
5206
        r = toku_env_dbremove(env, txn, subdb_full_name, null_subdbname, flags);
5207
    }
5208 5209
    return r;
}
5210

5211 5212 5213 5214

//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
5215
finalize_file_removal(DICTIONARY_ID dict_id, void * extra) {
5216 5217 5218
    toku_ltm *ltm = (toku_ltm*) extra;
    if (ltm) {
        //Poison the lock tree to prevent a future file from re-using it.
5219
        toku_ltm_invalidate_lt(ltm, dict_id);
5220 5221 5222
    }
}

5223
//static int toku_db_pre_acquire_table_lock(DB *db, DB_TXN *txn);
5224

5225 5226 5227 5228 5229 5230 5231 5232 5233 5234 5235 5236 5237 5238 5239 5240 5241 5242 5243 5244 5245 5246 5247 5248 5249 5250
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) {
5251
	r = toku_txn_begin(env, txn, &child, DB_TXN_NOSYNC, 1);
5252 5253 5254 5255
	assert(r==0);
    }

    // get iname
5256
    r = toku_db_get(env->i->directory, child, &dname_dbt, &iname_dbt, DB_SERIALIZABLE);  // allocates memory for iname
5257 5258 5259 5260 5261 5262 5263 5264
    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
5265
                r = toku_brt_remove_on_commit(db_txn_struct_i(child)->tokutxn, &iname_dbt);
5266
		assert(r==0);
5267 5268 5269
                //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");
5270 5271 5272
                if (r==0) {
                    DB* zombie = env_get_zombie_db_with_dname(env, dname);
                    if (zombie)
5273
                        r = toku_db_pre_acquire_table_lock(zombie, child, TRUE);
5274
                    if (r!=0 && r!=DB_LOCK_NOTGRANTED)
5275 5276
                        toku_ydb_do_error(env, r, "Cannot remove dictionary.\n");
                }
5277 5278
            }
            else {
Yoni Fogel's avatar
Yoni Fogel committed
5279
                r = toku_brt_remove_now(env->i->cachetable, &iname_dbt);
5280 5281 5282 5283 5284 5285 5286 5287
		assert(r==0);
            }
	}
    }

    if (using_txns) {
	// close txn
	if (r == 0) {  // commit
5288
	    r = toku_txn_commit(child, DB_TXN_NOSYNC, NULL, NULL);
5289
	    invariant(r==0);  // TODO panic
5290 5291
	}
	else {         // abort
5292
	    int r2 = toku_txn_abort(child, NULL, NULL);
5293
	    invariant(r2==0);  // TODO panic
5294
	}
5295
    }
5296 5297

    if (iname) toku_free(iname);
5298
    return r;
5299

5300 5301
}

5302 5303 5304

static int
toku_db_remove(DB * db, const char *fname, const char *dbname, u_int32_t flags) {
5305
    HANDLE_PANICKED_DB(db);
5306 5307 5308 5309 5310 5311 5312 5313 5314 5315 5316 5317 5318 5319 5320 5321 5322 5323 5324 5325 5326
    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
5327
        }
5328 5329
        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
5330
    }
5331 5332 5333 5334 5335 5336 5337 5338 5339 5340 5341 5342 5343 5344 5345 5346 5347 5348 5349 5350 5351 5352
    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
5353
    
5354 5355 5356 5357 5358 5359 5360 5361 5362 5363 5364
    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) {
5365
	r = toku_txn_begin(env, txn, &child, DB_TXN_NOSYNC, 1);
5366 5367
	assert(r==0);
    }
5368

5369
    r = toku_db_get(env->i->directory, child, &old_dname_dbt, &iname_dbt, DB_SERIALIZABLE);  // allocates memory for iname
5370 5371 5372 5373 5374
    char *iname = iname_dbt.data;
    if (r==DB_NOTFOUND)
        r = ENOENT;
    else if (r==0) {
	// verify that newname does not already exist
5375
	r = db_getf_set(env->i->directory, child, DB_SERIALIZABLE, &new_dname_dbt, ydb_getf_do_nothing, NULL);
5376 5377 5378 5379 5380 5381
	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)
5382
		r = toku_db_put(env->i->directory, child, &new_dname_dbt, &iname_dbt, 0);
5383 5384 5385
            //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");
5386
            DB* zombie = NULL;
5387
            if (r==0) {
5388
                zombie = env_get_zombie_db_with_dname(env, dname);
5389
                if (zombie)
5390
                    r = toku_db_pre_acquire_table_lock(zombie, child, TRUE);
5391
                if (r!=0 && r!=DB_LOCK_NOTGRANTED)
5392 5393
                    toku_ydb_do_error(env, r, "Cannot rename dictionary.\n");
            }
5394 5395
            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");
5396 5397 5398 5399 5400 5401
            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
5402
            }
5403 5404 5405 5406 5407 5408
	}
    }

    if (using_txns) {
	// close txn
	if (r == 0) {  // commit
5409
	    r = toku_txn_commit(child, DB_TXN_NOSYNC, NULL, NULL);
5410
	    invariant(r==0);  // TODO panic
5411 5412
	}
	else {         // abort
5413
	    int r2 = toku_txn_abort(child, NULL, NULL);
5414
	    invariant(r2==0);  // TODO panic
5415 5416 5417 5418 5419
	}
    }

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

Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
5421
}
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
5422

5423 5424
static int
toku_db_rename(DB * db, const char *fname, const char *dbname, const char *newname, u_int32_t flags) {
5425
    HANDLE_PANICKED_DB(db);
5426 5427 5428 5429 5430
    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
5431
}
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
5432

5433 5434 5435
//
// This function is the only way to set a descriptor of a DB.
//
5436
static int 
5437
toku_db_change_descriptor(DB *db, DB_TXN* txn, const DBT* descriptor, u_int32_t flags) {
5438
    HANDLE_PANICKED_DB(db);
5439
    HANDLE_DB_ILLEGAL_WORKING_PARENT_TXN(db, txn);
5440
    int r;
5441 5442 5443 5444 5445 5446 5447 5448 5449 5450 5451 5452 5453 5454 5455 5456 5457 5458 5459 5460 5461 5462 5463 5464 5465 5466 5467 5468 5469 5470
    TOKUTXN ttxn = txn ? db_txn_struct_i(txn)->tokutxn : NULL;
    DBT old_descriptor;
    BOOL is_db_hot_index  = ((flags & DB_IS_HOT_INDEX) != 0);

    toku_init_dbt(&old_descriptor);
    if (!db_opened(db) || !txn || !descriptor || (descriptor->size>0 && !descriptor->data)){
        r = EINVAL;
        goto cleanup;
    }
    if (txn->parent != NULL) {
        r = EINVAL; // cannot have a parent if you are a resetting op
        goto cleanup;
    }
    //
    // If the DB is created for the purpose of being a hot index, 
    // then do not grab a write lock on the directory when setting the
    // descriptor, because the hot index DB must not have a write
    // lock grabbed in order to work
    //
    if (is_db_hot_index) {
        r = toku_grab_read_lock_on_directory(db, txn);
        if (r != 0) { goto cleanup; }    
    }
    else {
        r = toku_db_pre_acquire_fileops_lock(db, txn);
        if (r != 0) { goto cleanup; }    
    }
    
    old_descriptor.size = db->descriptor->dbt.size;
    old_descriptor.data = toku_memdup(db->descriptor->dbt.data, db->descriptor->dbt.size);
5471
    r = toku_brt_change_descriptor(db->i->brt, &old_descriptor, descriptor, TRUE, ttxn);
5472 5473
cleanup:
    if (old_descriptor.data) toku_free(old_descriptor.data);
5474 5475 5476
    return r;
}

5477 5478
static int 
toku_db_set_flags(DB *db, u_int32_t flags) {
5479
    HANDLE_PANICKED_DB(db);
5480

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

5484
    return 0;
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
5485 5486
}

5487 5488
static int 
toku_db_get_flags(DB *db, u_int32_t *pflags) {
5489
    HANDLE_PANICKED_DB(db);
5490
    if (!pflags) return EINVAL;
5491
    *pflags = 0;
5492 5493 5494
    return 0;
}

5495 5496
static int 
toku_db_set_pagesize(DB *db, u_int32_t pagesize) {
5497
    HANDLE_PANICKED_DB(db);
5498
    int r = toku_brt_set_nodesize(db->i->brt, pagesize);
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
5499
    return r;
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
5500
}
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
5501

5502 5503 5504 5505 5506 5507 5508
static int 
toku_db_get_pagesize(DB *db, u_int32_t *pagesize_ptr) {
    HANDLE_PANICKED_DB(db);
    int r = toku_brt_get_nodesize(db->i->brt, pagesize_ptr);
    return r;
}

5509 5510 5511 5512 5513 5514 5515 5516 5517 5518 5519 5520 5521 5522
static int 
toku_db_set_readpagesize(DB *db, u_int32_t readpagesize) {
    HANDLE_PANICKED_DB(db);
    int r = toku_brt_set_basementnodesize(db->i->brt, readpagesize);
    return r;
}

static int 
toku_db_get_readpagesize(DB *db, u_int32_t *readpagesize_ptr) {
    HANDLE_PANICKED_DB(db);
    int r = toku_brt_get_basementnodesize(db->i->brt, readpagesize_ptr);
    return r;
}

5523 5524
static int 
toku_db_stat64(DB * db, DB_TXN *txn, DB_BTREE_STAT64 *s) {
5525
    HANDLE_PANICKED_DB(db);
5526
    HANDLE_DB_ILLEGAL_WORKING_PARENT_TXN(db, txn);
5527
    struct brtstat64_s brtstat;
5528 5529 5530 5531 5532
    TOKUTXN tokutxn = NULL;
    if (txn != NULL) {
        tokutxn = db_txn_struct_i(txn)->tokutxn;
    }
    int r = toku_brt_stat64(db->i->brt, tokutxn, &brtstat);
5533
    if (r==0) {
John Esmet's avatar
John Esmet committed
5534 5535 5536 5537 5538 5539 5540
        s->bt_nkeys = brtstat.nkeys;
        s->bt_ndata = brtstat.ndata;
        s->bt_dsize = brtstat.dsize;
        s->bt_fsize = brtstat.fsize;
        // 4018
        s->bt_create_time_sec = brtstat.create_time_sec;
        s->bt_modify_time_sec = brtstat.modify_time_sec;
5541 5542
    }
    return r;
5543
}
5544 5545 5546

static int 
locked_db_stat64 (DB *db, DB_TXN *txn, DB_BTREE_STAT64 *s) {
5547 5548 5549 5550
    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
5551 5552
}

5553 5554
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
5555
    HANDLE_PANICKED_DB(db);
5556
    HANDLE_DB_ILLEGAL_WORKING_PARENT_TXN(db, txn);
Zardosht Kasheff's avatar
Zardosht Kasheff committed
5557 5558 5559 5560 5561 5562 5563 5564 5565 5566 5567 5568 5569

    // 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;
}

5570
static int
5571 5572 5573
toku_c_pre_acquire_range_lock(DBC *dbc, const DBT *key_left, const DBT *key_right) {
    DB *db = dbc->dbp;
    DB_TXN *txn = dbc_struct_i(dbc)->txn;
Yoni Fogel's avatar
Yoni Fogel committed
5574
    HANDLE_PANICKED_DB(db);
5575 5576 5577 5578 5579
    toku_brt_cursor_set_range_lock(dbc_struct_i(dbc)->c, key_left, key_right,
                                   (key_left == toku_lt_neg_infinity),
                                   (key_right == toku_lt_infinity));
    if (!db->i->lt || !txn)
        return 0;
5580
    //READ_UNCOMMITTED and READ_COMMITTED transactions do not need read locks.
5581 5582
    if (!dbc_struct_i(dbc)->rmw && dbc_struct_i(dbc)->iso != TOKU_ISO_SERIALIZABLE)
        return 0;
Yoni Fogel's avatar
Yoni Fogel committed
5583

5584 5585
    toku_lock_type lock_type = dbc_struct_i(dbc)->rmw ? LOCK_REQUEST_WRITE : LOCK_REQUEST_READ;
    int r = get_range_lock(db, txn, key_left, key_right, lock_type);
Yoni Fogel's avatar
Yoni Fogel committed
5586 5587 5588
    return r;
}

5589 5590
//static int toku_db_pre_acquire_table_lock(DB *db, DB_TXN *txn) {
// needed by loader.c
5591 5592
int 
toku_db_pre_acquire_table_lock(DB *db, DB_TXN *txn, BOOL just_lock) {
Yoni Fogel's avatar
Yoni Fogel committed
5593
    HANDLE_PANICKED_DB(db);
5594
    if (!db->i->lt || !txn) return 0;
Yoni Fogel's avatar
Yoni Fogel committed
5595 5596 5597

    int r;

5598
    r = get_range_lock(db, txn, toku_lt_neg_infinity, toku_lt_infinity, LOCK_REQUEST_WRITE);
5599

5600 5601
    if (r==0 && !just_lock &&
        !toku_brt_is_recovery_logging_suppressed(db->i->brt) &&
5602
        toku_brt_is_empty_fast(db->i->brt)
5603 5604 5605 5606 5607 5608
    ) {
        //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};
5609
        uint32_t loader_flags = DB_PRELOCKED_WRITE; //Don't recursively prelock
5610
        DB_ENV *env = db->dbenv;
5611 5612 5613 5614 5615 5616 5617
	DB_TXN *child = NULL;
	
	{
	    // begin child
	    int rt = toku_txn_begin(env, txn, &child, DB_TXN_NOSYNC, 1);
	    assert(rt==0);
	}
5618 5619

        toku_ydb_unlock(); //Cannot hold ydb lock when creating loader
5620 5621
	
        int r_loader = env->create_loader(env, child, &loader, NULL, 1, dbs, db_flags, dbt_flags, loader_flags);
5622
        if (r_loader==0) {
5623 5624 5625 5626
            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);
5627
            // close the loader
5628 5629 5630 5631
            r_loader = loader->close(loader);
	    if (r_loader==0) {
		toku_brt_suppress_recovery_logs(db->i->brt, db_txn_struct_i(child)->tokutxn);
	    }
5632 5633 5634 5635 5636 5637 5638
        }
        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;
        }
5639 5640 5641
	if (r_loader == 0) { // commit
	    r = locked_txn_commit(child, 0);
	    assert(r==0);
5642
	    logsuppress++;
5643 5644 5645 5646
	}
	else {  // abort
	    r = locked_txn_abort(child);
	    assert(r==0);
5647
	    logsuppressfail++;
5648
	}
5649
        toku_ydb_lock(); //Reaquire ydb lock.
5650 5651
    }

Yoni Fogel's avatar
Yoni Fogel committed
5652 5653
    return r;
}
Zardosht Kasheff's avatar
Zardosht Kasheff committed
5654

Yoni Fogel's avatar
Yoni Fogel committed
5655 5656 5657
//TODO: DB_AUTO_COMMIT.
//TODO: Nowait only conditionally?
//TODO: NOSYNC change to SYNC if DB_ENV has something in set_flags
5658 5659
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
5660 5661 5662 5663 5664 5665
    assert(db && txn && changed);
    DB_ENV* env = db->dbenv;
    if (*txn || !(env->i->open_flags & DB_INIT_TXN)) {
        *changed = FALSE;
        return 0;
    }
5666
    BOOL nosync = (BOOL)(!force_auto_commit && !(env->i->open_flags & DB_AUTO_COMMIT));
Yoni Fogel's avatar
Yoni Fogel committed
5667
    u_int32_t txn_flags = DB_TXN_NOWAIT | (nosync ? DB_TXN_NOSYNC : 0);
5668
    int r = toku_txn_begin(env, NULL, txn, txn_flags, 1);
Yoni Fogel's avatar
Yoni Fogel committed
5669 5670 5671 5672 5673
    if (r!=0) return r;
    *changed = TRUE;
    return 0;
}

5674 5675
static inline int 
toku_db_destruct_autotxn(DB_TXN *txn, int r, BOOL changed) {
Yoni Fogel's avatar
Yoni Fogel committed
5676
    if (!changed) return r;
5677 5678
    if (r==0) return toku_txn_commit(txn, 0, NULL, NULL);
    toku_txn_abort(txn, NULL, NULL);
Yoni Fogel's avatar
Yoni Fogel committed
5679 5680 5681
    return r; 
}

5682 5683
static int 
locked_db_close(DB * db, u_int32_t flags) {
5684 5685 5686 5687
    toku_ydb_lock(); 
    int r = toku_db_close(db, flags); 
    toku_ydb_unlock(); 
    return r;
Rich Prohaska's avatar
Rich Prohaska committed
5688 5689
}

5690 5691
static inline int 
autotxn_db_cursor(DB *db, DB_TXN *txn, DBC **c, u_int32_t flags) {
Yoni Fogel's avatar
Yoni Fogel committed
5692
    if (!txn && (db->dbenv->i->open_flags & DB_INIT_TXN)) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
5693
        return toku_ydb_do_error(db->dbenv, EINVAL,
Yoni Fogel's avatar
Yoni Fogel committed
5694 5695
              "Cursors in a transaction environment must have transactions.\n");
    }
5696
    return toku_db_cursor(db, txn, c, flags, 0);
Yoni Fogel's avatar
Yoni Fogel committed
5697 5698
}

5699 5700
static int 
locked_db_cursor(DB *db, DB_TXN *txn, DBC **c, u_int32_t flags) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
5701
    toku_ydb_lock(); int r = autotxn_db_cursor(db, txn, c, flags); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
5702 5703
}

5704 5705
static inline int 
autotxn_db_del(DB* db, DB_TXN* txn, DBT* key, u_int32_t flags) {
Yoni Fogel's avatar
Yoni Fogel committed
5706 5707 5708 5709 5710 5711 5712
    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);
}

5713 5714
static int 
locked_db_del(DB * db, DB_TXN * txn, DBT * key, u_int32_t flags) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
5715
    toku_ydb_lock(); int r = autotxn_db_del(db, txn, key, flags); toku_ydb_unlock(); return r;
Yoni Fogel's avatar
Yoni Fogel committed
5716 5717
}

5718 5719
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
5720 5721 5722 5723 5724
    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
5725 5726
}

5727 5728
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
5729
    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
5730 5731
}

5732 5733
static inline int 
autotxn_db_getf_set (DB *db, DB_TXN *txn, u_int32_t flags, DBT *key, YDB_CALLBACK_FUNCTION f, void *extra) {
5734 5735 5736 5737 5738 5739 5740
    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);
}

5741 5742
static int 
locked_db_getf_set (DB *db, DB_TXN *txn, u_int32_t flags, DBT *key, YDB_CALLBACK_FUNCTION f, void *extra) {
5743 5744 5745
    toku_ydb_lock(); int r = autotxn_db_getf_set(db, txn, flags, key, f, extra); toku_ydb_unlock(); return r;
}

5746
static int 
5747
locked_c_pre_acquire_range_lock(DBC *dbc, const DBT *key_left, const DBT *key_right) {
Yoni Fogel's avatar
Yoni Fogel committed
5748
    toku_ydb_lock();
5749
    int r = toku_c_pre_acquire_range_lock(dbc, key_left, key_right);
Yoni Fogel's avatar
Yoni Fogel committed
5750 5751 5752 5753
    toku_ydb_unlock();
    return r;
}

5754 5755
static int 
locked_db_pre_acquire_table_lock(DB *db, DB_TXN *txn) {
Yoni Fogel's avatar
Yoni Fogel committed
5756
    toku_ydb_lock();
5757
    int r = toku_db_pre_acquire_table_lock(db, txn, FALSE);
Yoni Fogel's avatar
Yoni Fogel committed
5758 5759 5760 5761
    toku_ydb_unlock();
    return r;
}

Zardosht Kasheff's avatar
Zardosht Kasheff committed
5762 5763 5764 5765 5766 5767 5768
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;
}

5769 5770 5771 5772 5773 5774 5775
static int locked_db_pre_acquire_fileops_shared_lock(DB *db, DB_TXN *txn) {
    toku_ydb_lock();
    int r = toku_grab_read_lock_on_directory(db, txn);
    toku_ydb_unlock();
    return r;
}

5776 5777
// truncate a database
// effect: remove all of the rows from a database
5778 5779
static int 
toku_db_truncate(DB *db, DB_TXN *txn, u_int32_t *row_count, u_int32_t flags) {
5780
    HANDLE_PANICKED_DB(db);
5781
    HANDLE_DB_ILLEGAL_WORKING_PARENT_TXN(db, txn);
5782 5783
    int r;

5784 5785 5786 5787 5788 5789 5790
    u_int32_t unhandled_flags = flags;
    int ignore_cursors = 0;
    if (flags & DB_TRUNCATE_WITHCURSORS) {
        ignore_cursors = 1;
        unhandled_flags &= ~DB_TRUNCATE_WITHCURSORS;
    }

5791
    // dont support flags (yet)
5792
    if (unhandled_flags)
5793
        return EINVAL;
5794 5795
    // dont support cursors unless explicitly told to
    if (!ignore_cursors && toku_brt_get_cursor_count(db->i->brt) > 0)
5796 5797 5798 5799
        return EINVAL;

    // acquire a table lock
    if (txn) {
Zardosht Kasheff's avatar
Zardosht Kasheff committed
5800 5801 5802 5803
        r = toku_db_pre_acquire_fileops_lock(db, txn);
        if (r != 0) {
            return r;
        }
5804
        r = toku_db_pre_acquire_table_lock(db, txn, TRUE);
Zardosht Kasheff's avatar
Zardosht Kasheff committed
5805
        if (r != 0) {
5806
            return r;
Zardosht Kasheff's avatar
Zardosht Kasheff committed
5807
        }
5808 5809 5810 5811
    }

    *row_count = 0;

5812
    r = toku_brt_truncate(db->i->brt);
5813 5814 5815 5816

    return r;
}

5817 5818
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
5819
    BOOL changed; int r;
5820
    r = toku_db_construct_autotxn(db, &txn, &changed, (BOOL)((flags & DB_AUTO_COMMIT) != 0));
Yoni Fogel's avatar
Yoni Fogel committed
5821 5822 5823
    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
5824 5825
}

5826 5827
static int 
locked_db_open(DB *db, DB_TXN *txn, const char *fname, const char *dbname, DBTYPE dbtype, u_int32_t flags, int mode) {
5828 5829 5830 5831
    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
5832 5833
}

5834 5835
static inline int 
autotxn_db_put(DB* db, DB_TXN* txn, DBT* key, DBT* data, u_int32_t flags) {
5836
    //{ 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
5837 5838 5839 5840 5841 5842 5843
    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);
}

5844 5845
static int 
locked_db_put(DB * db, DB_TXN * txn, DBT * key, DBT * data, u_int32_t flags) {
5846 5847 5848 5849 5850 5851 5852
    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
5853 5854
}

5855 5856 5857 5858 5859 5860 5861 5862 5863 5864 5865 5866 5867 5868 5869 5870 5871 5872 5873 5874 5875 5876 5877 5878 5879 5880 5881 5882 5883 5884 5885 5886 5887 5888 5889 5890 5891 5892 5893 5894 5895 5896 5897 5898 5899 5900 5901 5902 5903 5904
static inline int
autotxn_db_update(DB *db, DB_TXN *txn,
                  const DBT *key,
                  const DBT *update_function_extra,
                  u_int32_t flags) {
    BOOL changed; int r;
    r = toku_db_construct_autotxn(db, &txn, &changed, FALSE);
    if (r != 0) { return r; }
    r = toku_db_update(db, txn, key, update_function_extra, flags);
    return toku_db_destruct_autotxn(txn, r, changed);
}

static int
locked_db_update(DB *db, DB_TXN *txn,
                 const DBT *key,
                 const DBT *update_function_extra,
                 u_int32_t flags) {
    int r = env_check_avail_fs_space(db->dbenv);
    if (r != 0) { goto cleanup; }
    toku_ydb_lock();
    r = autotxn_db_update(db, txn, key, update_function_extra, flags);
    toku_ydb_unlock();
cleanup:
    return r;
}

static inline int
autotxn_db_update_broadcast(DB *db, DB_TXN *txn,
                            const DBT *update_function_extra,
                            u_int32_t flags) {
    BOOL changed; int r;
    r = toku_db_construct_autotxn(db, &txn, &changed, FALSE);
    if (r != 0) { return r; }
    r = toku_db_update_broadcast(db, txn, update_function_extra, flags);
    return toku_db_destruct_autotxn(txn, r, changed);
}

static int
locked_db_update_broadcast(DB *db, DB_TXN *txn,
                           const DBT *update_function_extra,
                           u_int32_t flags) {
    int r = env_check_avail_fs_space(db->dbenv);
    if (r != 0) { goto cleanup; }
    toku_ydb_lock();
    r = autotxn_db_update_broadcast(db, txn, update_function_extra, flags);
    toku_ydb_unlock();
cleanup:
    return r;
}

5905 5906
static int 
locked_db_remove(DB * db, const char *fname, const char *dbname, u_int32_t flags) {
5907
    toku_multi_operation_client_lock(); //Cannot begin checkpoint
5908 5909 5910
    toku_ydb_lock();
    int r = toku_db_remove(db, fname, dbname, flags);
    toku_ydb_unlock();
5911
    toku_multi_operation_client_unlock(); //Can now begin checkpoint
5912
    return r;
Rich Prohaska's avatar
Rich Prohaska committed
5913 5914
}

5915 5916
static int 
locked_db_rename(DB * db, const char *namea, const char *nameb, const char *namec, u_int32_t flags) {
5917
    toku_multi_operation_client_lock(); //Cannot begin checkpoint
5918 5919 5920
    toku_ydb_lock();
    int r = toku_db_rename(db, namea, nameb, namec, flags);
    toku_ydb_unlock();
5921
    toku_multi_operation_client_unlock(); //Can now begin checkpoint
5922
    return r;
Rich Prohaska's avatar
Rich Prohaska committed
5923 5924
}

5925
static int 
5926
locked_db_change_descriptor(DB *db, DB_TXN* txn, const DBT* descriptor, u_int32_t flags) {
5927
    toku_ydb_lock();
5928
    int r = toku_db_change_descriptor(db, txn, descriptor, flags);
5929 5930
    toku_ydb_unlock();
    return r;
5931 5932
}

5933 5934
static void 
locked_db_set_errfile (DB *db, FILE *errfile) {
Rich Prohaska's avatar
Rich Prohaska committed
5935 5936 5937
    db->dbenv->set_errfile(db->dbenv, errfile);
}

5938 5939
static int 
locked_db_set_flags(DB *db, u_int32_t flags) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
5940
    toku_ydb_lock(); int r = toku_db_set_flags(db, flags); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
5941 5942
}

5943 5944
static int 
locked_db_get_flags(DB *db, u_int32_t *flags) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
5945
    toku_ydb_lock(); int r = toku_db_get_flags(db, flags); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
5946 5947
}

5948 5949
static int 
locked_db_set_pagesize(DB *db, u_int32_t pagesize) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
5950
    toku_ydb_lock(); int r = toku_db_set_pagesize(db, pagesize); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
5951 5952
}

5953 5954 5955 5956 5957
static int 
locked_db_get_pagesize(DB *db, u_int32_t *pagesize_ptr) {
    toku_ydb_lock(); int r = toku_db_get_pagesize(db, pagesize_ptr); toku_ydb_unlock(); return r;
}

5958 5959 5960 5961 5962 5963 5964 5965 5966 5967
static int 
locked_db_set_readpagesize(DB *db, u_int32_t readpagesize) {
    toku_ydb_lock(); int r = toku_db_set_readpagesize(db, readpagesize); toku_ydb_unlock(); return r;
}

static int 
locked_db_get_readpagesize(DB *db, u_int32_t *readpagesize_ptr) {
    toku_ydb_lock(); int r = toku_db_get_readpagesize(db, readpagesize_ptr); toku_ydb_unlock(); return r;
}

5968
// TODO 2216 delete this
5969 5970
static int 
locked_db_fd(DB * UU(db), int * UU(fdp)) {
5971 5972 5973 5974 5975
    //    toku_ydb_lock(); 
    // int r = toku_db_fd(db, fdp); 
    //    toku_ydb_unlock(); 
    //    return r;
    return 0;
Rich Prohaska's avatar
Rich Prohaska committed
5976 5977
}

Zardosht Kasheff's avatar
Zardosht Kasheff committed
5978

Yoni Fogel's avatar
Yoni Fogel committed
5979 5980 5981 5982 5983
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));
5984 5985
static const DBT*
toku_db_dbt_pos_infty(void) {
Yoni Fogel's avatar
Yoni Fogel committed
5986 5987 5988 5989
    return toku_lt_infinity;
}

static const DBT* toku_db_dbt_neg_infty(void) __attribute__((pure));
5990 5991
static const DBT* 
toku_db_dbt_neg_infty(void) {
Yoni Fogel's avatar
Yoni Fogel committed
5992
    return toku_lt_neg_infinity;
Zardosht Kasheff's avatar
Zardosht Kasheff committed
5993 5994
}

5995 5996
static int 
locked_db_truncate(DB *db, DB_TXN *txn, u_int32_t *row_count, u_int32_t flags) {
5997 5998 5999 6000 6001 6002
    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;
6003 6004
}

6005 6006 6007 6008 6009 6010 6011
static int
toku_db_optimize(DB *db) {
    HANDLE_PANICKED_DB(db);
    int r = toku_brt_optimize(db->i->brt);
    return r;
}

6012 6013 6014
static int
toku_db_flatten(DB *db, DB_TXN *txn) {
    HANDLE_PANICKED_DB(db);
6015 6016
    TOKUTXN ttxn = txn ? db_txn_struct_i(txn)->tokutxn : NULL;
    int r = toku_brt_flatten(db->i->brt, ttxn);
6017 6018 6019
    return r;
}

6020 6021
static inline int 
autotxn_db_flatten(DB* db, DB_TXN* txn) {
6022 6023 6024 6025 6026 6027 6028 6029
    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);
}


6030 6031
static int 
locked_db_flatten(DB *db, DB_TXN *txn) {
6032 6033 6034
    toku_ydb_lock(); int r = autotxn_db_flatten(db, txn); toku_ydb_unlock(); return r;
}

6035 6036
static int 
locked_db_optimize(DB *db) {
6037 6038 6039 6040 6041 6042
    toku_ydb_lock();
    int r = toku_db_optimize(db);
    toku_ydb_unlock();
    return r;
}

6043 6044 6045 6046 6047 6048 6049 6050 6051 6052 6053 6054 6055 6056 6057 6058 6059 6060 6061
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;
}

6062 6063
int 
toku_db_set_indexer(DB *db, DB_INDEXER * indexer) {
Dave Wells's avatar
Dave Wells committed
6064 6065 6066 6067 6068 6069 6070 6071 6072
    int r = 0;
    if ( db->i->indexer != NULL && indexer != NULL ) {
        // you are trying to overwrite a valid indexer
        r = EINVAL;
    }
    else {
        db->i->indexer = indexer;
    }
    return r;
6073 6074 6075 6076 6077 6078 6079 6080 6081 6082 6083 6084 6085 6086 6087 6088 6089
}

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();
}

6090 6091 6092 6093 6094 6095 6096 6097 6098 6099 6100 6101 6102 6103 6104 6105 6106 6107 6108 6109 6110 6111 6112 6113 6114
struct ydb_verify_context {
    int (*progress_callback)(void *extra, float progress);
    void *progress_extra;
};

static int
ydb_verify_progress_callback(void *extra, float progress) {
    struct ydb_verify_context *context = (struct ydb_verify_context *) extra;
    toku_ydb_unlock_and_yield(1000);
    int r = 0;
    if (context->progress_callback)
        r = context->progress_callback(context->progress_extra, progress);
    toku_ydb_lock();
    return r;
}

static int
locked_db_verify_with_progress(DB *db, int (*progress_callback)(void *extra, float progress), void *progress_extra, int verbose, int keep_going) {
    struct ydb_verify_context context = { progress_callback, progress_extra };
    toku_ydb_lock();
    int r = toku_verify_brt_with_progress(db->i->brt, ydb_verify_progress_callback, &context, verbose, keep_going);
    toku_ydb_unlock();
    return r;
}

6115 6116
static int 
toku_db_create(DB ** db, DB_ENV * env, u_int32_t flags) {
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
6117 6118
    int r;

6119 6120 6121 6122 6123
    if (flags || env == NULL) 
        return EINVAL;

    if (!env_opened(env))
        return EINVAL;
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
6124
    
6125
    DB *MALLOC(result);
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
6126 6127
    if (result == 0) {
        return ENOMEM;
Bradley C. Kuszmaul's avatar
Bradley C. Kuszmaul committed
6128
    }
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
6129 6130
    memset(result, 0, sizeof *result);
    result->dbenv = env;
Yoni Fogel's avatar
Yoni Fogel committed
6131 6132 6133 6134 6135 6136 6137 6138 6139
#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);
6140 6141
    SDB(update);
    SDB(update_broadcast);
Yoni Fogel's avatar
Yoni Fogel committed
6142 6143
    SDB(remove);
    SDB(rename);
6144
    SDB(change_descriptor);
Yoni Fogel's avatar
Yoni Fogel committed
6145 6146
    SDB(set_errfile);
    SDB(set_pagesize);
6147
    SDB(get_pagesize);
6148 6149
    SDB(set_readpagesize);
    SDB(get_readpagesize);
Yoni Fogel's avatar
Yoni Fogel committed
6150 6151
    SDB(set_flags);
    SDB(get_flags);
6152
    SDB(stat64);
Yoni Fogel's avatar
Yoni Fogel committed
6153
    SDB(fd);
Yoni Fogel's avatar
Yoni Fogel committed
6154
    SDB(pre_acquire_table_lock);
Zardosht Kasheff's avatar
Zardosht Kasheff committed
6155
    SDB(pre_acquire_fileops_lock);
6156
    SDB(pre_acquire_fileops_shared_lock);
6157
    SDB(truncate);
6158
    SDB(row_size_supported);
6159
    SDB(getf_set);
6160
    SDB(flatten);
6161
    SDB(optimize);
6162
    SDB(get_fragmentation);
6163 6164
    SDB(set_indexer);
    SDB(get_indexer);
6165
    SDB(verify_with_progress);
Yoni Fogel's avatar
Yoni Fogel committed
6166 6167 6168
#undef SDB
    result->dbt_pos_infty = toku_db_dbt_pos_infty;
    result->dbt_neg_infty = toku_db_dbt_neg_infty;
6169
    MALLOC(result->i);
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
6170 6171 6172 6173 6174
    if (result->i == 0) {
        toku_free(result);
        return ENOMEM;
    }
    memset(result->i, 0, sizeof *result->i);
6175
    result->i->dict_id = DICTIONARY_ID_NONE;
6176
    result->i->opened = 0;
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
6177 6178 6179
    result->i->open_flags = 0;
    result->i->open_mode = 0;
    result->i->brt = 0;
6180 6181
    result->i->indexer = NULL;
    result->i->refs = 1;
6182
    toku_list_init(&result->i->dbs_that_must_close_before_abort);
6183
    r = toku_brt_create(&result->i->brt);
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
6184 6185 6186
    if (r != 0) {
        toku_free(result->i);
        toku_free(result);
Yoni Fogel's avatar
Yoni Fogel committed
6187
        return r;
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
6188 6189 6190
    }
    *db = result;
    return 0;
Bradley C. Kuszmaul's avatar
Bradley C. Kuszmaul committed
6191
}
Bradley C. Kuszmaul's avatar
Bradley C. Kuszmaul committed
6192

6193 6194
int 
DB_CREATE_FUN (DB ** db, DB_ENV * env, u_int32_t flags) {
6195 6196 6197 6198
    toku_ydb_lock(); 
    int r = toku_db_create(db, env, flags); 
    toku_ydb_unlock(); 
    return r;
Rich Prohaska's avatar
Rich Prohaska committed
6199 6200 6201 6202
}

/* need db_strerror_r for multiple threads */

6203 6204
char *
db_strerror(int error) {
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
6205 6206 6207 6208 6209 6210 6211
    char *errorstr;
    if (error >= 0) {
        errorstr = strerror(error);
        if (errorstr)
            return errorstr;
    }
    
6212 6213 6214 6215 6216 6217 6218 6219 6220 6221 6222 6223 6224 6225 6226
    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)";
6227
    }
6228

Rich Prohaska's avatar
Rich Prohaska committed
6229
    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
6230 6231 6232 6233 6234
    errorstr = unknown_result;
    snprintf(errorstr, sizeof unknown_result, "Unknown error code: %d", error);
    return errorstr;
}

6235 6236
const char *
db_version(int *major, int *minor, int *patch) {
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
6237 6238 6239 6240 6241 6242
    if (major)
        *major = DB_VERSION_MAJOR;
    if (minor)
        *minor = DB_VERSION_MINOR;
    if (patch)
        *patch = DB_VERSION_PATCH;
6243 6244 6245 6246 6247
#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
6248
#error
6249
#endif
Bradley C. Kuszmaul's avatar
Bradley C. Kuszmaul committed
6250
}
6251
 
6252 6253
int 
db_env_set_func_fsync (int (*fsync_function)(int)) {
6254 6255
    return toku_set_func_fsync(fsync_function);
}
Yoni Fogel's avatar
Yoni Fogel committed
6256

6257 6258
int 
db_env_set_func_pwrite (ssize_t (*pwrite_function)(int, const void *, size_t, toku_off_t)) {
6259 6260
    return toku_set_func_pwrite(pwrite_function);
}
6261 6262 6263

int 
db_env_set_func_full_pwrite (ssize_t (*pwrite_function)(int, const void *, size_t, toku_off_t)) {
6264 6265
    return toku_set_func_full_pwrite(pwrite_function);
}
6266 6267 6268

int 
db_env_set_func_write (ssize_t (*write_function)(int, const void *, size_t)) {
6269 6270
    return toku_set_func_write(write_function);
}
6271 6272 6273

int 
db_env_set_func_full_write (ssize_t (*write_function)(int, const void *, size_t)) {
6274 6275
    return toku_set_func_full_write(write_function);
}
6276

6277 6278
int 
db_env_set_func_fdopen (FILE * (*fdopen_function)(int, const char *)) {
6279 6280
    return toku_set_func_fdopen(fdopen_function);
}
6281 6282 6283

int 
db_env_set_func_fopen (FILE * (*fopen_function)(const char *, const char *)) {
6284 6285
    return toku_set_func_fopen(fopen_function);
}
6286 6287 6288

int 
db_env_set_func_open (int (*open_function)(const char *, int, int)) {
6289 6290
    return toku_set_func_open(open_function);
}
6291 6292 6293

int 
db_env_set_func_fclose (int (*fclose_function)(FILE*)) {
6294 6295 6296
    return toku_set_func_fclose(fclose_function);
}

6297 6298 6299 6300 6301
int
db_env_set_func_pread (ssize_t (*fun)(int, void *, size_t, off_t)) {
    return toku_set_func_pread(fun);
}

6302 6303 6304 6305 6306
void 
db_env_set_func_loader_fwrite (size_t (*fwrite_fun)(const void*,size_t,size_t,FILE*)) {
    brtloader_set_os_fwrite(fwrite_fun);
}

6307 6308
int 
db_env_set_func_malloc (void *(*f)(size_t)) {
6309 6310
    toku_set_func_malloc(f);
    return 0;
6311
}
6312 6313 6314

int 
db_env_set_func_realloc (void *(*f)(void*, size_t)) {
6315 6316
    toku_set_func_realloc(f);
    return 0;
6317
}
6318 6319 6320

int 
db_env_set_func_free (void (*f)(void*)) {
6321 6322
    toku_set_func_free(f);
    return 0;
6323
}
6324

6325

6326
// Got to call dlmalloc, or else it won't get included.
6327 6328
void 
setup_dlmalloc (void) {
6329 6330 6331 6332
    db_env_set_func_malloc(dlmalloc);
    db_env_set_func_realloc(dlrealloc);
    db_env_set_func_free(dlfree);
}
6333 6334

// For test purposes only.
6335
// With this interface, all checkpoint users get the same callbacks and the same extras.
6336 6337
void 
db_env_set_checkpoint_callback (void (*callback_f)(void*), void* extra) {
6338
    toku_checkpoint_safe_client_lock();
6339 6340
    checkpoint_callback_f = callback_f;
    checkpoint_callback_extra = extra;
6341
    toku_checkpoint_safe_client_unlock();
6342
    //printf("set callback = %p, extra = %p\n", callback_f, extra);
6343
}
6344 6345 6346

void 
db_env_set_checkpoint_callback2 (void (*callback_f)(void*), void* extra) {
6347 6348 6349 6350 6351 6352
    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
6353

6354 6355
void 
db_env_set_recover_callback (void (*callback_f)(void*), void* extra) {
6356 6357 6358
    toku_recover_set_callback(callback_f, extra);
}

6359 6360
void 
db_env_set_recover_callback2 (void (*callback_f)(void*), void* extra) {
6361 6362 6363
    toku_recover_set_callback2(callback_f, extra);
}

Zardosht Kasheff's avatar
Zardosht Kasheff committed
6364 6365 6366 6367 6368
void 
db_env_set_flusher_thread_callback(void (*callback_f)(int, void*), void* extra) {
    toku_flusher_thread_set_callback(callback_f, extra);
}

6369 6370
void 
db_env_set_loader_size_factor (uint32_t factor) {
6371 6372 6373
    toku_brtloader_set_size_factor(factor);
}

6374 6375
void 
db_env_set_mvcc_garbage_collection_verification(u_int32_t verification_mode) {
Zardosht Kasheff's avatar
Zardosht Kasheff committed
6376 6377
    garbage_collection_debug = (verification_mode != 0);
}
6378

6379 6380 6381 6382 6383
// Purpose: allow test programs that expect to fail to suppress engine status output on failed assert.
void
db_env_enable_engine_status(uint32_t enable) {
    engine_status_enable = enable;
}
6384

Yoni Fogel's avatar
Yoni Fogel committed
6385 6386 6387 6388 6389 6390 6391 6392 6393 6394
// 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();
}

6395 6396 6397 6398

// For test purposes only, translate dname to iname
static int 
env_get_iname(DB_ENV* env, DBT* dname_dbt, DBT* iname_dbt) {
6399
    toku_ydb_lock();
6400
    DB *directory = env->i->directory;
6401
    int r = autotxn_db_get(directory, NULL, dname_dbt, iname_dbt, DB_SERIALIZABLE|DB_PRELOCKED); // allocates memory for iname
6402
    toku_ydb_unlock();
6403 6404 6405
    return r;
}

6406 6407 6408 6409 6410 6411 6412 6413 6414 6415 6416 6417 6418
/* 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.
6419 6420
// If "mark_as_loader" is true, then include a mark in the iname
// to indicate that the file is created by the brt loader.
6421 6422
// Return 0 on success (could fail if write lock not available).
int
6423
ydb_load_inames(DB_ENV * env, DB_TXN * txn, int N, DB * dbs[N], char * new_inames_in_env[N], LSN *load_lsn, BOOL mark_as_loader) {
6424 6425 6426 6427 6428
    int rval;
    int i;
    
    int using_txns = env->i->open_flags & DB_INIT_TXN;
    DB_TXN * child = NULL;
6429
    TXNID xid = 0;
6430 6431 6432
    DBT dname_dbt;  // holds dname
    DBT iname_dbt;  // holds new iname
    
6433 6434 6435
    char * mark;

    if (mark_as_loader)
6436
	mark = "B";
6437 6438 6439
    else
	mark = "P";

6440 6441 6442 6443
    for (i=0; i<N; i++) {
	new_inames_in_env[i] = NULL;
    }

6444 6445 6446 6447 6448 6449 6450 6451 6452 6453 6454 6455
    // 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);
6456
	char * new_iname = create_iname(env, xid, hint, mark, i);               // allocates memory for iname_in_env
6457
	new_inames_in_env[i] = new_iname;
6458
        toku_fill_dbt(&iname_dbt, new_iname, strlen(new_iname) + 1);      // iname_in_env goes in directory
6459
        rval = toku_db_put(env->i->directory, child, &dname_dbt, &iname_dbt, 0);
6460
	if (rval) break;
Yoni Fogel's avatar
Yoni Fogel committed
6461 6462 6463 6464 6465 6466
    }

    // Generate load log entries.
    if (!rval && using_txns) {
        TOKUTXN ttxn = db_txn_struct_i(txn)->tokutxn;
        int do_fsync = 0;
6467
        LSN *get_lsn = NULL;
Yoni Fogel's avatar
Yoni Fogel committed
6468 6469 6470
        for (i = 0; i < N; i++) {
            BRT brt  = dbs[i]->i->brt;
            //Fsync is necessary for the last one only.
6471 6472 6473 6474 6475
            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
6476 6477
            if (rval) break;
        }
6478 6479 6480 6481 6482
    }
	
    if (using_txns) {
	// close txn
	if (rval == 0) {  // all well so far, commit child
6483
	    rval = toku_txn_commit(child, DB_TXN_NOSYNC, NULL, NULL);
6484 6485 6486
	    assert(rval==0);
	}
	else {         // abort child
6487
	    int r2 = toku_txn_abort(child, NULL, NULL);
6488 6489
	    assert(r2==0);
	    for (i=0; i<N; i++) {
6490 6491 6492 6493
		if (new_inames_in_env[i]) {
		    toku_free(new_inames_in_env[i]);
		    new_inames_in_env[i] = NULL;
		}
6494 6495 6496 6497 6498 6499 6500
	    }
	}
    }

    return rval;
}

6501
int
6502
locked_ydb_load_inames(DB_ENV * env, DB_TXN * txn, int N, DB * dbs[N], char * new_inames_in_env[N], LSN *load_lsn, BOOL mark_as_loader) {
6503
    toku_ydb_lock();
6504
    int r = ydb_load_inames(env, txn, N, dbs, new_inames_in_env, load_lsn, mark_as_loader);
6505 6506 6507 6508
    toku_ydb_unlock();
    return r;
}

6509 6510 6511 6512
// 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
6513
toku_test_db_redirect_dictionary(DB * db, char * dname_of_new_file, DB_TXN *dbtxn) {
6514 6515 6516 6517 6518 6519 6520 6521 6522 6523
    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
6524
    r = toku_db_get(db->dbenv->i->directory, dbtxn, &dname_dbt, &iname_dbt, DB_SERIALIZABLE);  // allocates memory for iname
6525 6526 6527
    assert(r==0);
    new_iname_in_env = iname_dbt.data;

Yoni Fogel's avatar
Yoni Fogel committed
6528
    r = toku_dictionary_redirect(new_iname_in_env, brt, tokutxn);
6529 6530 6531 6532

    toku_free(new_iname_in_env);
    return r;
}
6533 6534 6535 6536 6537 6538 6539 6540 6541 6542 6543

//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;
}

6544 6545
int 
toku_test_get_checkpointing_user_data_status (void) {
6546 6547
    return toku_cachetable_get_checkpointing_user_data_status();
}
Zardosht Kasheff's avatar
Zardosht Kasheff committed
6548

6549 6550
// acquire a point write lock on the key for a given txn.
// this does not block the calling thread.
6551
int
6552 6553 6554 6555 6556 6557 6558 6559
toku_grab_write_lock (DB *db, DBT *key, TOKUTXN tokutxn) {
    DB_TXN *txn = toku_txn_get_container_db_txn(tokutxn);
    DB_TXN *txn_anc = toku_txn_ancestor(txn);
    int r = toku_txn_add_lt(txn_anc, db->i->lt);
    if (r == 0) {
        TXNID txn_anc_id = toku_txn_get_txnid(db_txn_struct_i(txn_anc)->tokutxn);
        r = toku_lt_acquire_write_lock(db->i->lt, db, txn_anc_id, key);
    }
6560 6561
    return r;
}