ydb.c 232 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 1765 1766 1767 1768 1769 1770 1771 1772 1773 1774 1775 1776 1777 1778 1779
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];
    }
}
1780

1781 1782
// 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
1783 1784 1785
// can help diagnose the problem.
// This function only collects information, and it does not matter if something gets garbled
// because of a race condition.  
1786
// Note, engine status is still collected even if the environment or logger is panicked
1787
static int
1788 1789 1790
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) {
1791
	if (env && env->i && env->i->is_panicked && env->i->panic_string) {
1792
	    strncpy(env_panic_string_buf, env->i->panic_string, env_panic_string_length);
1793 1794
	    env_panic_string_buf[env_panic_string_length - 1] = '\0';  // just in case
	}
1795 1796 1797 1798
	else 
	    *env_panic_string_buf = '\0';
    }

1799 1800 1801 1802
    if ( !(env)     || 
	 !(env->i)  || 
	 !(env_opened(env)) )
	r = EINVAL;
1803
    else {
1804 1805
	r = 0;
	engstat->env_panic = env->i->is_panicked;
1806
	format_time(&persistent_creation_time, engstat->creationtime);
1807 1808
	time_t now = time(NULL);
        format_time(&now, engstat->now);
1809
        format_time(&startuptime, engstat->startuptime);
1810 1811 1812
	{
	    SCHEDULE_STATUS_S schedstat;
	    toku_ydb_lock_get_status(&schedstat);
1813 1814 1815 1816 1817 1818 1819 1820
	    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.   */

1821
	}
1822 1823 1824 1825 1826 1827 1828 1829
        {
	    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;
        }
1830
	engstat->checkpoint_period = toku_get_checkpoint_period_unlocked(env->i->cachetable);  // do not take any locks (not even minicron lock)
1831 1832 1833 1834 1835 1836 1837
	{
            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);
1838 1839 1840 1841 1842 1843 1844 1845 1846 1847 1848 1849 1850
	    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;
1851
                time_t   oldest_starttime = 0;
1852 1853 1854
		uint64_t next_lsn   = 0;
		TOKULOGGER logger = env->i->logger;
		if (logger) {
1855
		    oldest_xid = toku_logger_get_oldest_living_xid(env->i->logger, &oldest_starttime);
1856 1857 1858 1859
		    next_lsn   = (toku_logger_get_next_lsn(env->i->logger)).lsn;
		}
		engstat->txn_oldest_live = oldest_xid;
		engstat->next_lsn = next_lsn;
1860
                format_time(&oldest_starttime, engstat->txn_oldest_live_starttime);
1861
	    }
1862
	}
1863 1864 1865
	{
	    CACHETABLE_STATUS_S ctstat;
	    toku_cachetable_get_status(env->i->cachetable, &ctstat);
1866 1867 1868 1869 1870 1871 1872 1873
	    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;
1874
	    engstat->cachetable_wait_checkpoint = ctstat.wait_checkpoint;
1875 1876 1877 1878 1879 1880
	    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;
1881
	    engstat->cachetable_size_max      = ctstat.size_max;
1882 1883
	    engstat->cachetable_size_writing  = ctstat.size_writing;
	    engstat->get_and_pin_footprint    = ctstat.get_and_pin_footprint;
1884 1885 1886
	    engstat->local_checkpoint         = ctstat.local_checkpoint;
	    engstat->local_checkpoint_files   = ctstat.local_checkpoint_files;
	    engstat->local_checkpoint_during_checkpoint = ctstat.local_checkpoint_during_checkpoint;
1887
            engstat->cachetable_evictions     = ctstat.evictions;
1888
	}
1889 1890
	{
	    toku_ltm* ltm = env->i->ltm;
1891
	    LTM_STATUS_S ltmstat;
1892 1893 1894 1895 1896
	    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);
1897 1898
	    engstat->range_locks_max                 = max_locks;
	    engstat->range_locks_curr                = curr_locks;
1899 1900
	    engstat->range_locks_max_memory          = max_lock_memory;
	    engstat->range_locks_curr_memory         = curr_lock_memory;
1901 1902
	    engstat->range_lock_escalation_successes = ltmstat.lock_escalation_successes;
	    engstat->range_lock_escalation_failures  = ltmstat.lock_escalation_failures;
1903 1904 1905 1906 1907 1908
	    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;
1909
	}
1910
	{
1911
     	    engstat->inserts            = num_inserts;
1912
	    engstat->inserts_fail       = num_inserts_fail;
1913
	    engstat->deletes            = num_deletes;
1914
	    engstat->deletes_fail       = num_deletes_fail;
1915 1916
	    engstat->updates            = num_updates;
	    engstat->updates_fail       = num_updates_fail;
1917 1918
	    engstat->updates_broadcast  = num_updates_broadcast;
	    engstat->updates_broadcast_fail  = num_updates_broadcast_fail;
1919 1920 1921 1922 1923 1924
     	    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;
1925 1926
	    engstat->point_queries      = num_point_queries;
	    engstat->sequential_queries = num_sequential_queries;
1927 1928 1929 1930
            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;
1931
	}
1932
	{
1933 1934 1935 1936 1937
	    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;
1938 1939 1940 1941 1942 1943
	    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;	    
1944 1945
	    engstat->cachetable_size_leaf = brt_stat.bytes_leaf;
	    engstat->cachetable_size_nonleaf = brt_stat.bytes_nonleaf;
1946
	}
1947 1948 1949 1950 1951 1952
	{
	    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;
	}
1953 1954 1955 1956 1957 1958 1959
	{
	    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;
1960 1961
	    engstat->logger_panic     = log_stat.panicked;
	    engstat->logger_panic_errno = log_stat.panic_errno;
1962
	}
1963 1964
	{
	    time_t    enospc_most_recent_timestamp;
1965 1966
	    u_int64_t enospc_threads_blocked, enospc_ctr;
	    toku_fs_get_write_info(&enospc_most_recent_timestamp, &enospc_threads_blocked, &enospc_ctr);
1967 1968
	    format_time(&enospc_most_recent_timestamp, engstat->enospc_most_recent);	    
	    engstat->enospc_threads_blocked = enospc_threads_blocked;
1969
	    engstat->enospc_ctr = enospc_ctr;
1970
	}
1971
	{
1972 1973 1974 1975 1976 1977 1978 1979 1980
	    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;
1981
	    engstat->loader_put_fail       = loader_stat.put_fail;
1982 1983 1984 1985 1986 1987 1988 1989
	    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;
1990
	}
1991 1992 1993 1994 1995 1996 1997 1998 1999 2000 2001 2002 2003
	{
	    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;
	}
2004
	{
2005 2006
	    BRT_UPGRADE_STATUS_S brt_upgrade_stat;
	    toku_brt_get_upgrade_status(&brt_upgrade_stat);
2007 2008 2009 2010 2011
	    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.
2012
	    // If ver_at_startup == 13, then the footprint will have the same value in 
2013
	    // upper and lower digits.
2014 2015 2016 2017 2018
	    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;
2019 2020
	    engstat->original_ver       = persistent_original_env_version;
	    engstat->ver_at_startup     = persistent_stored_env_version_at_startup;
2021 2022
	    engstat->last_lsn_v13       = persistent_last_lsn_of_v13;
	    format_time(&persistent_upgrade_v14_time, engstat->upgrade_v14_time);
2023
	}
2024 2025 2026
	{
	    MEMORY_STATUS_S memory_status;
	    toku_memory_get_status(&memory_status);
2027 2028 2029 2030 2031 2032 2033 2034 2035
	    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;
2036
	}
2037 2038 2039 2040
    }
    return r;
}

2041

2042
// Fill buff with text description of engine status up to bufsiz bytes.
2043 2044
// 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.
2045 2046 2047
static int
env_get_engine_status_text(DB_ENV * env, char * buff, int bufsiz) {
    ENGINE_STATUS engstat;
2048
    uint32_t stringsize = 1024;
2049
    char panicstring[stringsize];
2050 2051
    int n = 0;  // number of characters printed so far

2052 2053 2054

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

2055 2056 2057 2058 2059 2060 2061
    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);
    }

2062 2063 2064 2065 2066 2067 2068 2069 2070 2071 2072 2073 2074
    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 {
2075
	n += snprintf(buff + n, bufsiz - n, "env panic                        %"PRIu64"\n", engstat.env_panic);
2076 2077 2078 2079
	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);
2080 2081
	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);
2082
	n += snprintf(buff + n, bufsiz - n, "total_sleep_time                 %"PRIu64"\n", engstat.total_sleep_time);
2083 2084
	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));
2085
	n += snprintf(buff + n, bufsiz - n, "total_time_since_start           %"PRIu64"\n", engstat.total_time_since_start);
2086 2087 2088 2089
	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);
2090 2091 2092 2093 2094 2095 2096 2097 2098 2099 2100 2101 2102 2103 2104 2105 2106 2107 2108 2109 2110 2111
	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);
2112
	n += snprintf(buff + n, bufsiz - n, "cachetable_evictions             %"PRIu64"\n", engstat.cachetable_evictions);
2113 2114 2115 2116 2117 2118
	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);
2119 2120 2121
	n += snprintf(buff + n, bufsiz - n, "cachetable_size_max              %"PRId64"\n", engstat.cachetable_size_max);
	n += snprintf(buff + n, bufsiz - n, "cachetable_size_leaf             %"PRId64"\n", engstat.cachetable_size_leaf);
	n += snprintf(buff + n, bufsiz - n, "cachetable_size_nonleaf          %"PRId64"\n", engstat.cachetable_size_nonleaf);
2122 2123 2124 2125 2126 2127 2128 2129 2130 2131 2132 2133 2134 2135 2136 2137 2138 2139 2140 2141 2142
	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);
2143 2144
	n += snprintf(buff + n, bufsiz - n, "updates                          %"PRIu64"\n", engstat.updates);
	n += snprintf(buff + n, bufsiz - n, "updates_fail                     %"PRIu64"\n", engstat.updates_fail);
2145 2146 2147 2148
	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);
2149
	n += snprintf(buff + n, bufsiz - n, "descriptor_set                   %"PRIu64"\n", engstat.descriptor_set);
2150 2151 2152 2153 2154 2155
	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);
2156 2157 2158 2159 2160 2161
	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);
2162 2163
	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);
2164 2165 2166 2167
	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);
2168 2169 2170 2171 2172
	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);
2173 2174
	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);
2175 2176 2177 2178 2179 2180
	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);
2181
	n += snprintf(buff + n, bufsiz - n, "loader_create_fail               %"PRIu64"\n", engstat.loader_create_fail);
2182
	n += snprintf(buff + n, bufsiz - n, "loader_put                       %"PRIu64"\n", engstat.loader_put);
2183
	n += snprintf(buff + n, bufsiz - n, "loader_put_fail                  %"PRIu64"\n", engstat.loader_put_fail);
2184 2185 2186 2187 2188 2189 2190
	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);
2191 2192 2193 2194 2195 2196 2197 2198 2199
	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);
2200 2201 2202 2203
	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);
2204
	n += snprintf(buff + n, bufsiz - n, "optimized_for_upgrade_14         %"PRIu64"\n", engstat.optimized_for_upgrade);
2205 2206
	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);
2207 2208
	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);
2209 2210 2211
	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);
2212 2213
	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);
2214 2215 2216
	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);
2217
	n += snprintf(buff + n, bufsiz - n, "max_mem_in_use                   %"PRIu64"\n", engstat.max_mem_in_use);
2218
    }
2219 2220 2221
    if (n > bufsiz) {
	char * errmsg = "BUFFER TOO SMALL\n";
	int len = strlen(errmsg) + 1;
2222
	(void) snprintf(buff + (bufsiz - 1) - len, len, "%s", errmsg);
2223 2224 2225 2226 2227
    }

    return r;
}

2228 2229 2230 2231
// 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;
2232 2233 2234 2235 2236 2237 2238 2239
    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");
    }
2240 2241 2242
    return r;
}

2243 2244 2245 2246
// 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) {
2247 2248 2249 2250 2251
    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
2252
    DB_ENV * env = most_recent_env;
2253 2254 2255 2256
    if (env && 
	env->i &&
	(env->i->is_panicked == 0)) {
	env_panic(env, code, msg);
2257 2258
    }
}
2259 2260


Rich Prohaska's avatar
Rich Prohaska committed
2261 2262
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
2263 2264 2265 2266
static int toku_db_lt_panic(DB* db, int r);

static toku_dbt_cmp toku_db_get_compare_fun(DB* db);

2267 2268
static int 
toku_env_create(DB_ENV ** envp, u_int32_t flags) {
Yoni Fogel's avatar
Yoni Fogel committed
2269 2270 2271
    int r = ENOSYS;
    DB_ENV* result = NULL;

2272 2273
    engine_status_enable = 1;

Yoni Fogel's avatar
Yoni Fogel committed
2274 2275 2276
    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
2277
    memset(result, 0, sizeof *result);
2278
    result->err = (void (*)(const DB_ENV * env, int error, const char *fmt, ...)) toku_locked_env_err;
2279 2280 2281 2282
#define SENV(name) result->name = locked_env_ ## name
    SENV(dbremove);
    SENV(dbrename);
    SENV(set_default_bt_compare);
2283
    SENV(set_update);
2284 2285
    SENV(set_generate_row_callback_for_put);
    SENV(set_generate_row_callback_for_del);
2286 2287
    SENV(put_multiple);
    SENV(del_multiple);
2288
    SENV(update_multiple);
2289 2290
    SENV(checkpointing_set_period);
    SENV(checkpointing_get_period);
2291 2292 2293 2294
    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;
2295
    result->get_engine_status = env_get_engine_status;
2296
    result->get_engine_status_text = env_get_engine_status_text;
2297 2298 2299
    result->get_iname = env_get_iname;
    SENV(open);
    SENV(close);
2300
    result->txn_checkpoint = toku_env_txn_checkpoint;
2301
    SENV(log_flush);
Rich Prohaska's avatar
Rich Prohaska committed
2302 2303 2304
    result->set_errcall = toku_env_set_errcall;
    result->set_errfile = toku_env_set_errfile;
    result->set_errpfx = toku_env_set_errpfx;
2305 2306 2307 2308 2309 2310 2311 2312 2313 2314 2315
    //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);
2316 2317
    SENV(set_lk_max_memory);
    SENV(get_lk_max_memory);
2318
    SENV(set_cachesize);
Rich Prohaska's avatar
Rich Prohaska committed
2319
#if DB_VERSION_MAJOR == 4 && DB_VERSION_MINOR >= 3
2320
    SENV(get_cachesize);
Rich Prohaska's avatar
Rich Prohaska committed
2321
#endif
2322
    SENV(set_lk_detect);
2323
#if DB_VERSION_MAJOR == 4 && DB_VERSION_MINOR <= 4
2324
    SENV(set_lk_max);
2325
#endif
2326 2327
    SENV(log_archive);
    SENV(txn_stat);
Rich Prohaska's avatar
Rich Prohaska committed
2328
    result->txn_begin = locked_txn_begin;
2329
    SENV(set_redzone);
Dave Wells's avatar
Dave Wells committed
2330
    SENV(create_indexer);
2331
    SENV(create_loader);
Dave Wells's avatar
Dave Wells committed
2332
#undef SENV
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
2333

2334
    MALLOC(result->i);
Yoni Fogel's avatar
Yoni Fogel committed
2335
    if (result->i == 0) { r = ENOMEM; goto cleanup; }
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
2336
    memset(result->i, 0, sizeof *result->i);
Yoni Fogel's avatar
Yoni Fogel committed
2337 2338 2339 2340
    result->i->envdir_lockfd  = -1;
    result->i->datadir_lockfd = -1;
    result->i->logdir_lockfd  = -1;
    result->i->tmpdir_lockfd  = -1;
2341
    env_init_open_txn(result);
2342
    env_fs_init(result);
Yoni Fogel's avatar
Yoni Fogel committed
2343

2344 2345
    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
2346
                         toku_db_lt_panic, 
2347
                         toku_db_get_compare_fun,
Yoni Fogel's avatar
Yoni Fogel committed
2348
                         toku_malloc, toku_free, toku_realloc);
Yoni Fogel's avatar
Yoni Fogel committed
2349
    if (r!=0) { goto cleanup; }
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
2350

2351
    {
Yoni Fogel's avatar
Yoni Fogel committed
2352 2353
	r = toku_logger_create(&result->i->logger);
	if (r!=0) { goto cleanup; }
2354 2355
	assert(result->i->logger);
    }
2356 2357 2358 2359 2360
    {
        r = toku_omt_create(&result->i->open_dbs);
        if (r!=0) goto cleanup;
        assert(result->i->open_dbs);
    }
2361

Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
2362
    *envp = result;
Yoni Fogel's avatar
Yoni Fogel committed
2363 2364 2365 2366 2367 2368 2369 2370
    r = 0;
cleanup:
    if (r!=0) {
        if (result) {
            if (result->i) {
                if (result->i->ltm) {
                    toku_ltm_close(result->i->ltm);
                }
2371 2372
                if (result->i->open_dbs)
                    toku_omt_destroy(&result->i->open_dbs);
Yoni Fogel's avatar
Yoni Fogel committed
2373 2374 2375 2376 2377 2378
                toku_free(result->i);
            }
            toku_free(result);
        }
    }
    return r;
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
2379 2380
}

2381 2382
int 
DB_ENV_CREATE_FUN (DB_ENV ** envp, u_int32_t flags) {
2383 2384 2385 2386
    toku_ydb_lock(); 
    int r = toku_env_create(envp, flags); 
    toku_ydb_unlock(); 
    return r;
Rich Prohaska's avatar
Rich Prohaska committed
2387 2388
}

2389 2390
static int 
toku_txn_release_locks(DB_TXN* txn) {
Yoni Fogel's avatar
Yoni Fogel committed
2391
    assert(txn);
2392
    toku_lth* lth = db_txn_struct_i(txn)->lth;
Yoni Fogel's avatar
Yoni Fogel committed
2393

Yoni Fogel's avatar
Yoni Fogel committed
2394 2395
    int r = ENOSYS;
    int first_error = 0;
Yoni Fogel's avatar
Yoni Fogel committed
2396 2397 2398 2399
    if (lth) {
        toku_lth_start_scan(lth);
        toku_lock_tree* next = toku_lth_next(lth);
        while (next) {
2400
            r = toku_lt_unlock(next, toku_txn_get_txnid(db_txn_struct_i(txn)->tokutxn));
Yoni Fogel's avatar
Yoni Fogel committed
2401 2402 2403 2404 2405
            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
2406 2407 2408
            next = toku_lth_next(lth);
        }
        toku_lth_close(lth);
2409
        db_txn_struct_i(txn)->lth = NULL;
Yoni Fogel's avatar
Yoni Fogel committed
2410
    }
Yoni Fogel's avatar
Yoni Fogel committed
2411 2412
    r = first_error;

Yoni Fogel's avatar
Yoni Fogel committed
2413
    return r;
Yoni Fogel's avatar
Yoni Fogel committed
2414 2415
}

2416 2417
// 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.
2418 2419
static void 
ydb_yield (voidfp f, void *fv, void *UU(v)) {
2420
    toku_ydb_unlock(); 
2421 2422
    if (f) 
        f(fv);
2423 2424 2425
    toku_ydb_lock();
}

2426 2427 2428
static int 
toku_txn_commit(DB_TXN * txn, u_int32_t flags,
                TXN_PROGRESS_POLL_FUNCTION poll, void* poll_extra) {
2429
    if (!txn) return EINVAL;
2430
    HANDLE_PANICKED_ENV(txn->mgrp);
Yoni Fogel's avatar
Yoni Fogel committed
2431
    //Recursively kill off children
2432 2433
    if (db_txn_struct_i(txn)->child) {
        //commit of child sets the child pointer to NULL
2434
        int r_child = toku_txn_commit(db_txn_struct_i(txn)->child, flags, NULL, NULL);
2435
        if (r_child !=0 && !toku_env_is_panicked(txn->mgrp)) {
2436
	    env_panic(txn->mgrp, r_child, "Recursive child commit failed during parent commit.\n");
2437
        }
Yoni Fogel's avatar
Yoni Fogel committed
2438 2439
        //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
2440
    }
2441
    assert(!db_txn_struct_i(txn)->child);
Yoni Fogel's avatar
Yoni Fogel committed
2442 2443
    //Remove from parent
    if (txn->parent) {
2444 2445
        assert(db_txn_struct_i(txn->parent)->child == txn);
        db_txn_struct_i(txn->parent)->child=NULL;
Yoni Fogel's avatar
Yoni Fogel committed
2446
    }
2447
    env_remove_open_txn(txn->mgrp, txn);
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
2448
    //toku_ydb_notef("flags=%d\n", flags);
2449 2450 2451 2452
    if (flags & DB_TXN_SYNC) {
        toku_txn_force_fsync_on_commit(db_txn_struct_i(txn)->tokutxn);
        flags &= ~DB_TXN_SYNC;
    }
2453
    int nosync = (flags & DB_TXN_NOSYNC)!=0 || (db_txn_struct_i(txn)->flags&DB_TXN_NOSYNC);
2454
    flags &= ~DB_TXN_NOSYNC;
Yoni Fogel's avatar
Yoni Fogel committed
2455 2456

    int r;
2457
    if (flags!=0)
2458 2459
	// frees the tokutxn
	// Calls ydb_yield(NULL) occasionally
2460
        //r = toku_logger_abort(db_txn_struct_i(txn)->tokutxn, ydb_yield, NULL);
2461
        r = toku_txn_abort_txn(db_txn_struct_i(txn)->tokutxn, ydb_yield, NULL, poll, poll_extra);
Yoni Fogel's avatar
Yoni Fogel committed
2462
    else
2463 2464
	// frees the tokutxn
	// Calls ydb_yield(NULL) occasionally
2465
        //r = toku_logger_commit(db_txn_struct_i(txn)->tokutxn, nosync, ydb_yield, NULL);
2466
        r = toku_txn_commit_txn(db_txn_struct_i(txn)->tokutxn, nosync, ydb_yield, NULL,
2467
				poll, poll_extra);
2468

2469
    if (r!=0 && !toku_env_is_panicked(txn->mgrp)) {
2470
	env_panic(txn->mgrp, r, "Error during commit.\n");
2471 2472 2473 2474 2475
    }
    //If panicked, we're done.
    HANDLE_PANICKED_ENV(txn->mgrp);
    assert(r==0);

2476
    // Close the logger after releasing the locks
2477
    r = toku_txn_release_locks(txn);
2478
    //toku_logger_txn_close(db_txn_struct_i(txn)->tokutxn);
2479
    toku_txn_maybe_fsync_log(db_txn_struct_i(txn)->tokutxn, ydb_yield, NULL);
2480
    toku_txn_close_txn(db_txn_struct_i(txn)->tokutxn);
2481
    // the toxutxn is freed, and we must free the rest. */
Yoni Fogel's avatar
Yoni Fogel committed
2482

2483 2484 2485
    //Promote list to parent (dbs that must close before abort)
    if (txn->parent) {
        //Combine lists.
2486 2487 2488
        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);
2489 2490 2491 2492
        }
    }
    else {
        //Empty the list
2493 2494
        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);
2495 2496 2497
        }
    }

2498
    // The txn is no good after the commit even if the commit fails, so free it up.
2499 2500 2501
#if !TOKUDB_NATIVE_H
    toku_free(db_txn_struct_i(txn));
#endif
2502
    toku_free(txn);    txn = NULL;
Yoni Fogel's avatar
Yoni Fogel committed
2503
    if (flags!=0) return EINVAL;
2504
    return r;
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
2505 2506
}

2507 2508
static u_int32_t 
toku_txn_id(DB_TXN * txn) {
2509
    HANDLE_PANICKED_ENV(txn->mgrp);
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
2510
    toku_ydb_barf();
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
2511
    abort();
Rich Prohaska's avatar
Rich Prohaska committed
2512
    return -1;
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
2513 2514
}

2515 2516 2517
static int 
toku_txn_abort(DB_TXN * txn,
               TXN_PROGRESS_POLL_FUNCTION poll, void* poll_extra) {
2518
    HANDLE_PANICKED_ENV(txn->mgrp);
2519
    //Recursively kill off children (abort or commit are both correct, commit is cheaper)
2520 2521
    if (db_txn_struct_i(txn)->child) {
        //commit of child sets the child pointer to NULL
2522
        int r_child = toku_txn_commit(db_txn_struct_i(txn)->child, DB_TXN_NOSYNC, NULL, NULL);
2523
        if (r_child !=0 && !toku_env_is_panicked(txn->mgrp)) {
2524
	    env_panic(txn->mgrp, r_child, "Recursive child commit failed during parent abort.\n");
2525
        }
Yoni Fogel's avatar
Yoni Fogel committed
2526 2527
        //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
2528
    }
2529
    assert(!db_txn_struct_i(txn)->child);
Yoni Fogel's avatar
Yoni Fogel committed
2530 2531
    //Remove from parent
    if (txn->parent) {
2532 2533
        assert(db_txn_struct_i(txn->parent)->child == txn);
        db_txn_struct_i(txn->parent)->child=NULL;
Yoni Fogel's avatar
Yoni Fogel committed
2534
    }
2535
    env_remove_open_txn(txn->mgrp, txn);
2536 2537

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

2540
    //int r = toku_logger_abort(db_txn_struct_i(txn)->tokutxn, ydb_yield, NULL);
2541
    int r = toku_txn_abort_txn(db_txn_struct_i(txn)->tokutxn, ydb_yield, NULL, poll, poll_extra);
2542
    if (r!=0 && !toku_env_is_panicked(txn->mgrp)) {
2543
	env_panic(txn->mgrp, r, "Error during abort.\n");
2544 2545 2546 2547
    }
    HANDLE_PANICKED_ENV(txn->mgrp);
    assert(r==0);
    r = toku_txn_release_locks(txn);
2548
    //toku_logger_txn_close(db_txn_struct_i(txn)->tokutxn);
2549
    toku_txn_maybe_fsync_log(db_txn_struct_i(txn)->tokutxn, ydb_yield, NULL);
2550
    toku_txn_close_txn(db_txn_struct_i(txn)->tokutxn);
Yoni Fogel's avatar
Yoni Fogel committed
2551

2552 2553 2554
#if !TOKUDB_NATIVE_H
    toku_free(db_txn_struct_i(txn));
#endif
2555
    toku_free(txn);
2556
    return r;
2557 2558
}

2559 2560
static int 
locked_txn_begin(DB_ENV *env, DB_TXN * stxn, DB_TXN ** txn, u_int32_t flags) {
2561
    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
2562 2563
}

2564 2565
static u_int32_t 
locked_txn_id(DB_TXN *txn) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
2566
    toku_ydb_lock(); u_int32_t r = toku_txn_id(txn); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
2567 2568
}

2569 2570
static int 
toku_txn_stat (DB_TXN *txn, struct txn_stat **txn_stat) {
2571
    XMALLOC(*txn_stat);
2572
    return toku_logger_txn_rollback_raw_count(db_txn_struct_i(txn)->tokutxn, &(*txn_stat)->rollback_raw_count);
2573 2574
}

2575 2576
static int 
locked_txn_stat (DB_TXN *txn, struct txn_stat **txn_stat) {
2577 2578 2579
    toku_ydb_lock(); u_int32_t r = toku_txn_stat(txn, txn_stat); toku_ydb_unlock(); return r;
}

2580 2581 2582
static int 
locked_txn_commit_with_progress(DB_TXN *txn, u_int32_t flags,
                                TXN_PROGRESS_POLL_FUNCTION poll, void* poll_extra) {
2583
    toku_multi_operation_client_lock(); //Cannot checkpoint during a commit.
2584
    toku_ydb_lock(); int r = toku_txn_commit(txn, flags, poll, poll_extra); toku_ydb_unlock();
2585
    toku_multi_operation_client_unlock(); //Cannot checkpoint during a commit.
2586
    return r;
Rich Prohaska's avatar
Rich Prohaska committed
2587 2588
}

2589 2590 2591
static int 
locked_txn_abort_with_progress(DB_TXN *txn,
                               TXN_PROGRESS_POLL_FUNCTION poll, void* poll_extra) {
2592
    toku_multi_operation_client_lock(); //Cannot checkpoint during an abort.
2593
    toku_ydb_lock(); int r = toku_txn_abort(txn, poll, poll_extra); toku_ydb_unlock();
2594
    toku_multi_operation_client_unlock(); //Cannot checkpoint during an abort.
2595
    return r;
Rich Prohaska's avatar
Rich Prohaska committed
2596 2597
}

2598 2599
static int 
locked_txn_commit(DB_TXN *txn, u_int32_t flags) {
2600 2601 2602 2603 2604
    int r;
    r = locked_txn_commit_with_progress(txn, flags, NULL, NULL);
    return r;
}

2605 2606
static int 
locked_txn_abort(DB_TXN *txn) {
2607 2608 2609 2610 2611
    int r;
    r = locked_txn_abort_with_progress(txn, NULL, NULL);
    return r;
}

2612 2613
static int 
toku_txn_begin(DB_ENV *env, DB_TXN * stxn, DB_TXN ** txn, u_int32_t flags, int internal) {
2614
    HANDLE_PANICKED_ENV(env);
2615
    HANDLE_ILLEGAL_WORKING_PARENT_TXN(env, stxn); //Cannot create child while child already exists.
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
2616 2617
    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");
2618 2619
    u_int32_t txn_flags = 0;
    txn_flags |= DB_TXN_NOWAIT; //We do not support blocking locks.
2620
    TOKU_ISOLATION child_isolation = TOKU_ISO_SERIALIZABLE;
2621 2622 2623 2624 2625 2626 2627 2628 2629
    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)
       ) 
    {
2630 2631 2632
        return toku_ydb_do_error(
            env, 
            EINVAL, 
2633
            "Invalid isolation flags set\n"
2634 2635
            );
    }
2636 2637 2638 2639 2640 2641 2642 2643 2644 2645 2646 2647 2648 2649 2650 2651 2652 2653 2654 2655 2656 2657 2658 2659 2660 2661 2662 2663 2664 2665 2666 2667 2668 2669 2670
    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;
2671
        }
2672
    }
2673 2674 2675 2676 2677 2678 2679
    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"
            );   
2680
    }
2681

2682 2683 2684 2685 2686 2687 2688 2689 2690 2691
    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");

2692 2693
    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
2694 2695
    if (result == 0)
        return ENOMEM;
2696
    memset(result, 0, result_size);
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
2697
    //toku_ydb_notef("parent=%p flags=0x%x\n", stxn, flags);
2698
    result->mgrp = env;
2699 2700 2701 2702 2703 2704 2705
#define STXN(name) result->name = locked_txn_ ## name
    STXN(abort);
    STXN(commit);
    STXN(abort_with_progress);
    STXN(commit_with_progress);
    STXN(id);
#undef STXN
2706
    result->txn_stat = locked_txn_stat;
2707 2708 2709


    result->parent = stxn;
2710 2711 2712
#if !TOKUDB_NATIVE_H
    MALLOC(db_txn_struct_i(result));
    if (!db_txn_struct_i(result)) {
Yoni Fogel's avatar
Yoni Fogel committed
2713 2714 2715
        toku_free(result);
        return ENOMEM;
    }
2716 2717 2718
#endif
    memset(db_txn_struct_i(result), 0, sizeof *db_txn_struct_i(result));
    db_txn_struct_i(result)->flags = txn_flags;
2719
    db_txn_struct_i(result)->iso = child_isolation;
2720
    toku_list_init(&db_txn_struct_i(result)->dbs_that_must_close_before_abort);
Yoni Fogel's avatar
Yoni Fogel committed
2721 2722

    int r;
Yoni Fogel's avatar
Yoni Fogel committed
2723
    if (env->i->open_flags & DB_INIT_LOCK && !stxn) {
2724
        r = toku_lth_create(&db_txn_struct_i(result)->lth,
Yoni Fogel's avatar
Yoni Fogel committed
2725 2726
                            toku_malloc, toku_free, toku_realloc);
        if (r!=0) {
2727 2728 2729
#if !TOKUDB_NATIVE_H
            toku_free(db_txn_struct_i(result));
#endif
Yoni Fogel's avatar
Yoni Fogel committed
2730 2731 2732
            toku_free(result);
            return r;
        }
Yoni Fogel's avatar
Yoni Fogel committed
2733 2734
    }
    
2735
    //r = toku_logger_txn_begin(stxn ? db_txn_struct_i(stxn)->tokutxn : 0, &db_txn_struct_i(result)->tokutxn, env->i->logger);
2736 2737 2738 2739 2740 2741 2742 2743 2744 2745 2746 2747 2748 2749 2750 2751 2752 2753
    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;
        }
    }
2754 2755 2756 2757 2758 2759
    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
2760 2761
    if (r != 0)
        return r;
2762

Yoni Fogel's avatar
Yoni Fogel committed
2763 2764
    //Add to the list of children for the parent.
    if (result->parent) {
2765 2766
        assert(!db_txn_struct_i(result->parent)->child);
        db_txn_struct_i(result->parent)->child = result;
Yoni Fogel's avatar
Yoni Fogel committed
2767
    }
2768
    env_add_open_txn(env, result);
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
2769 2770 2771 2772
    *txn = result;
    return 0;
}

2773 2774
int 
log_compare(const DB_LSN * a, const DB_LSN * b) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
2775
    toku_ydb_lock();
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
2776
    fprintf(stderr, "%s:%d log_compare(%p,%p)\n", __FILE__, __LINE__, a, b);
2777
    assert(0);
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
2778
    toku_ydb_unlock();
2779
    return 0;
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
2780 2781
}

2782 2783
static void env_note_zombie_db_closed(DB_ENV *env, DB *db);

2784 2785
static int
db_close_before_brt(DB *db, u_int32_t UU(flags)) {
2786 2787 2788
    int r;
    char *error_string = NULL;
    
2789 2790 2791 2792
    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)
    }
2793 2794 2795 2796 2797 2798 2799
    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);
2800
    }
2801 2802 2803 2804 2805 2806 2807 2808 2809 2810
    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
2811
    }
2812
    return r;
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
2813 2814
}

2815 2816 2817 2818 2819 2820 2821 2822
// 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;
2823 2824 2825 2826 2827 2828 2829
    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;
2830 2831
    if (cmp != 0) return cmp;
    if (db < dbfind) return -1;
2832
    if (db > dbfind) return  1;
2833 2834 2835 2836 2837 2838 2839
    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
2840
    assert(!db->i->is_zombie);
2841 2842 2843
    int r;
    OMTVALUE dbv;
    uint32_t idx;
2844
    env->i->num_open_dbs++;
2845
    r = toku_omt_find_zero(env->i->open_dbs, find_db_by_db, db, &dbv, &idx);
2846 2847 2848 2849 2850 2851 2852 2853
    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);
2854
    assert(!db->i->is_zombie);
2855
    assert(env->i->num_open_dbs);
2856 2857 2858
    int r;
    OMTVALUE dbv;
    uint32_t idx;
2859
    env->i->num_open_dbs--;
2860
    r = toku_omt_find_zero(env->i->open_dbs, find_db_by_db, db, &dbv, &idx);
2861 2862 2863 2864 2865 2866 2867 2868 2869 2870 2871 2872 2873 2874
    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;
2875
    env->i->num_zombie_dbs++;
2876
    r = toku_omt_find_zero(env->i->open_dbs, find_db_by_db, db, &dbv, &idx);
2877 2878 2879 2880 2881 2882 2883 2884 2885
    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);
2886
    assert(env->i->num_zombie_dbs);
2887 2888 2889
    int r;
    OMTVALUE dbv;
    uint32_t idx;
2890
    env->i->num_zombie_dbs--;
2891
    r = toku_omt_find_zero(env->i->open_dbs, find_db_by_db, db, &dbv, &idx);
2892 2893 2894 2895 2896 2897 2898
    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
2899 2900 2901 2902 2903 2904 2905 2906 2907 2908 2909 2910 2911 2912 2913 2914 2915
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;
2916 2917
    const char *dname     = db->i->dname;
    const char *dnamefind = dnamev;
2918 2919 2920 2921 2922 2923
    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;
2924 2925 2926 2927 2928 2929 2930 2931 2932
}

// 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;
2933
    r = toku_omt_find_zero(env->i->open_dbs, find_open_db_by_dname, (void*)dname, &dbv, &idx);
2934 2935 2936
    if (r==0) {
        DB *db = dbv;
        assert(strcmp(dname, db->i->dname) == 0);
2937
        assert(!db->i->is_zombie);
2938 2939 2940 2941 2942 2943 2944 2945 2946
        rval = TRUE;
    }
    else {
        assert(r==DB_NOTFOUND);
        rval = FALSE;
    }
    return rval;
}

2947 2948 2949 2950 2951 2952 2953
// 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;
2954
    r = toku_omt_find_zero(env->i->open_dbs, find_zombie_db_by_dname, (void*)dname, &dbv, &idx);
2955 2956 2957 2958 2959 2960 2961 2962 2963 2964 2965 2966 2967 2968
    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;
}

2969 2970 2971 2972 2973 2974 2975 2976 2977 2978
void
toku_db_add_ref(DB *db) {
    db->i->refs++;
}

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

2979
//DB->close()
2980 2981
static int 
toku_db_close(DB * db, u_int32_t flags) {
2982 2983 2984 2985 2986 2987 2988 2989 2990 2991 2992 2993 2994 2995
    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);
2996

2997 2998
        r = toku_brt_db_delay_closed(db->i->brt, db, db_close_before_brt, flags);
    }
2999 3000 3001 3002
    return r;
}


3003
//Get the main portion of a cursor flag (excluding the bitwise or'd components).
3004 3005
static int 
get_main_cursor_flag(u_int32_t flags) {
3006
    return flags & DB_OPFLAGS_MASK;
Yoni Fogel's avatar
Yoni Fogel committed
3007 3008
}

3009 3010
static int 
get_nonmain_cursor_flags(u_int32_t flags) {
3011
    return flags & ~(DB_OPFLAGS_MASK);
3012 3013
}

3014 3015
static inline BOOL 
toku_c_uninitialized(DBC* c) {
3016
    return toku_brt_cursor_uninitialized(dbc_struct_i(c)->c);
Yoni Fogel's avatar
Yoni Fogel committed
3017
}            
Yoni Fogel's avatar
Yoni Fogel committed
3018

3019 3020 3021 3022 3023 3024 3025 3026 3027 3028 3029
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;
3030 3031
    context->skey = dbc_struct_i(c)->skey;
    context->sval = dbc_struct_i(c)->sval;
3032 3033 3034 3035 3036 3037 3038 3039 3040 3041 3042
}

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

3043 3044
static int 
toku_c_get_current_unconditional(DBC* c, u_int32_t flags, DBT* key, DBT* val) {
3045 3046 3047 3048
    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
3049 3050 3051
    return r;
}

3052 3053
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
3054 3055 3056 3057 3058
    *flag    -= *get_flag;
    *get_flag =  new_flag;
    *flag    += *get_flag;
}

Yoni Fogel's avatar
Yoni Fogel committed
3059 3060 3061 3062 3063 3064 3065
/*
    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.
*/
3066 3067
static inline DB_TXN* 
toku_txn_ancestor(DB_TXN* txn) {
Yoni Fogel's avatar
Yoni Fogel committed
3068
    while (txn && txn->parent) txn = txn->parent;
Yoni Fogel's avatar
Yoni Fogel committed
3069

Yoni Fogel's avatar
Yoni Fogel committed
3070 3071 3072
    return txn;
}

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

Yoni Fogel's avatar
Yoni Fogel committed
3075 3076 3077 3078 3079 3080 3081 3082 3083 3084 3085
/* 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
3086
    u_int32_t   lock_flags;         // The prelock flags.
Yoni Fogel's avatar
Yoni Fogel committed
3087 3088 3089 3090 3091 3092 3093 3094 3095 3096
    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;

3097

3098 3099
static inline u_int32_t 
get_prelocked_flags(u_int32_t flags) {
3100
    u_int32_t lock_flags = flags & (DB_PRELOCKED | DB_PRELOCKED_WRITE | DB_PRELOCKED_FILE_READ);
3101 3102
    return lock_flags;
}
3103

3104 3105
static inline u_int32_t 
get_cursor_prelocked_flags(u_int32_t flags, DBC* dbc) {
3106
    u_int32_t lock_flags = flags & (DB_PRELOCKED | DB_PRELOCKED_WRITE | DB_PRELOCKED_FILE_READ);
3107 3108

    //DB_READ_UNCOMMITTED and DB_READ_COMMITTED transactions 'own' all read locks for user-data dictionaries.
3109
    if (dbc_struct_i(dbc)->iso != TOKU_ISO_SERIALIZABLE) {
3110
        lock_flags |= DB_PRELOCKED;
3111
    }
3112
    return lock_flags;
Yoni Fogel's avatar
Yoni Fogel committed
3113 3114
}

3115 3116
static int
toku_c_get(DBC* c, DBT* key, DBT* val, u_int32_t flag) {
3117 3118 3119
    //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
3120

3121 3122 3123 3124 3125 3126 3127 3128 3129 3130 3131 3132
    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
3133
        case (DB_FIRST):
3134 3135
            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
3136
            break;
Yoni Fogel's avatar
Yoni Fogel committed
3137
        case (DB_LAST):
3138 3139
            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
3140
            break;
3141
        case (DB_NEXT):
Yoni Fogel's avatar
Yoni Fogel committed
3142
        case (DB_NEXT_NODUP):
3143
            query_context_wrapped_init(&context, c, key,  val);
3144
            r = toku_c_getf_next(c, remaining_flags, c_get_wrapper_callback, &context);
Yoni Fogel's avatar
Yoni Fogel committed
3145 3146
            break;
        case (DB_PREV):
3147
        case (DB_PREV_NODUP):
3148 3149
            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
3150 3151
            break;
#ifdef DB_PREV_DUP
Yoni Fogel's avatar
Yoni Fogel committed
3152
        case (DB_PREV_DUP):
3153 3154
            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
3155 3156
            break;
#endif
3157 3158 3159 3160 3161 3162 3163 3164 3165 3166 3167 3168 3169 3170 3171 3172 3173
        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;
3174 3175 3176 3177
        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
3178
        default:
Yoni Fogel's avatar
Yoni Fogel committed
3179
            r = EINVAL;
3180
            break;
Yoni Fogel's avatar
Yoni Fogel committed
3181
    }
Yoni Fogel's avatar
Yoni Fogel committed
3182
    return r;
Yoni Fogel's avatar
Yoni Fogel committed
3183 3184
}

3185 3186
static int 
locked_c_getf_first(DBC *c, u_int32_t flag, YDB_CALLBACK_FUNCTION f, void *extra) {
3187
    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
3188 3189
}

3190 3191
static int 
locked_c_getf_last(DBC *c, u_int32_t flag, YDB_CALLBACK_FUNCTION f, void *extra) {
3192 3193
    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
3194

3195 3196
static int 
locked_c_getf_next(DBC *c, u_int32_t flag, YDB_CALLBACK_FUNCTION f, void *extra) {
3197
    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
3198 3199
}

3200 3201
static int 
locked_c_getf_prev(DBC *c, u_int32_t flag, YDB_CALLBACK_FUNCTION f, void *extra) {
3202
    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
3203 3204
}

3205 3206
static int 
locked_c_getf_current(DBC *c, u_int32_t flag, YDB_CALLBACK_FUNCTION f, void *extra) {
3207
    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
3208 3209
}

3210 3211
static int 
locked_c_getf_current_binding(DBC *c, u_int32_t flag, YDB_CALLBACK_FUNCTION f, void *extra) {
3212 3213
    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
3214

3215 3216
static int 
locked_c_getf_set(DBC *c, u_int32_t flag, DBT * key, YDB_CALLBACK_FUNCTION f, void *extra) {
3217 3218
    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
3219

3220 3221
static int 
locked_c_getf_set_range(DBC *c, u_int32_t flag, DBT * key, YDB_CALLBACK_FUNCTION f, void *extra) {
3222 3223
    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
3224

3225 3226
static int 
locked_c_getf_set_range_reverse(DBC *c, u_int32_t flag, DBT * key, YDB_CALLBACK_FUNCTION f, void *extra) {
3227 3228 3229
    toku_ydb_lock();  int r = toku_c_getf_set_range_reverse(c, flag, key, f, extra); toku_ydb_unlock(); return r;
}

3230
typedef struct {
3231
    BOOL            is_write_lock;
3232 3233 3234 3235 3236 3237
    DB_TXN         *txn;
    DB             *db;
    toku_lock_tree *lt;
    DBT const      *left_key;
    DBT const      *right_key;
} *RANGE_LOCK_REQUEST, RANGE_LOCK_REQUEST_S;
Yoni Fogel's avatar
Yoni Fogel committed
3238

3239 3240
static void
range_lock_request_init(RANGE_LOCK_REQUEST request,
3241
                        BOOL       is_write_lock,
3242 3243 3244
                        DB_TXN    *txn,
                        DB        *db,
                        DBT const *left_key,
3245
                        DBT const *right_key) {
3246
    request->is_write_lock = is_write_lock;
3247 3248 3249 3250 3251
    request->txn = txn;
    request->db = db;
    request->lt = db->i->lt;
    request->left_key = left_key;
    request->right_key = right_key;
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
3252 3253
}

3254 3255 3256 3257 3258
static void
read_lock_request_init(RANGE_LOCK_REQUEST request,
                       DB_TXN    *txn,
                       DB        *db,
                       DBT const *left_key,
3259
                       DBT const *right_key) {
3260
    range_lock_request_init(request, FALSE, txn, db, left_key, right_key);
Yoni Fogel's avatar
Yoni Fogel committed
3261 3262
}

3263 3264 3265 3266 3267
static void
write_lock_request_init(RANGE_LOCK_REQUEST request,
                        DB_TXN    *txn,
                        DB        *db,
                        DBT const *left_key,
3268
                        DBT const *right_key) {
3269
    range_lock_request_init(request, TRUE, txn, db, left_key, right_key);
3270 3271 3272 3273 3274 3275 3276 3277 3278
}

static int
grab_range_lock(RANGE_LOCK_REQUEST request) {
    int r;
    //TODO: (Multithreading) Grab lock protecting lock tree
    DB_TXN *txn_anc = toku_txn_ancestor(request->txn);
    r = toku_txn_add_lt(txn_anc, request->lt);
    if (r==0) {
3279
        TXNID txn_anc_id = toku_txn_get_txnid(db_txn_struct_i(txn_anc)->tokutxn);
3280
        if (request->is_write_lock)
3281
            r = toku_lt_acquire_range_write_lock(request->lt, request->db, txn_anc_id,
3282
                                                 request->left_key, request->right_key);
3283 3284 3285
        else 
            r = toku_lt_acquire_range_read_lock(request->lt, request->db, txn_anc_id,
                                                request->left_key, request->right_key);
3286 3287 3288
    }
    //TODO: (Multithreading) Release lock protecting lock tree
    return r;
Yoni Fogel's avatar
Yoni Fogel committed
3289 3290
}

3291
int
Zardosht Kasheff's avatar
Zardosht Kasheff committed
3292 3293 3294
toku_grab_read_lock_on_directory (DB* db, DB_TXN * txn) {
    char *   dname = db->i->dname;
    DBT key_in_directory;
3295

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

Zardosht Kasheff's avatar
Zardosht Kasheff committed
3300 3301
    toku_fill_dbt(&key_in_directory, dname, strlen(dname)+1);
    //Left end of range == right end of range (point lock)
3302 3303 3304
    RANGE_LOCK_REQUEST_S request;
    read_lock_request_init(&request, txn, db->dbenv->i->directory,
                           &key_in_directory, &key_in_directory);
Zardosht Kasheff's avatar
Zardosht Kasheff committed
3305
    int r = grab_range_lock(&request);
3306 3307 3308 3309
    if (r == 0)
	directory_read_locks++;
    else
	directory_read_locks_fail++;
Zardosht Kasheff's avatar
Zardosht Kasheff committed
3310 3311 3312
    return r;
}

3313 3314
//This is the user level callback function given to ydb layer functions like
//toku_c_getf_first
Yoni Fogel's avatar
Yoni Fogel committed
3315

3316 3317 3318 3319 3320 3321
typedef struct query_context_base_t {
    BRT_CURSOR  c;
    DB_TXN     *txn;
    DB         *db;
    void       *f_extra;
    int         r_user_callback;
3322 3323
    BOOL        do_locking;
    BOOL        is_write_op;
3324
} *QUERY_CONTEXT_BASE, QUERY_CONTEXT_BASE_S;
3325

3326 3327 3328 3329
typedef struct query_context_t {
    QUERY_CONTEXT_BASE_S  base;
    YDB_CALLBACK_FUNCTION f;
} *QUERY_CONTEXT, QUERY_CONTEXT_S;
Yoni Fogel's avatar
Yoni Fogel committed
3330

3331 3332 3333 3334 3335 3336
typedef struct query_context_with_input_t {
    QUERY_CONTEXT_BASE_S  base;
    YDB_CALLBACK_FUNCTION f;
    DBT                  *input_key;
    DBT                  *input_val;
} *QUERY_CONTEXT_WITH_INPUT, QUERY_CONTEXT_WITH_INPUT_S;
Yoni Fogel's avatar
Yoni Fogel committed
3337

3338
static void
3339
query_context_base_init(QUERY_CONTEXT_BASE context, DBC *c, u_int32_t flag, BOOL is_write_op, void *extra) {
3340 3341
    context->c       = dbc_struct_i(c)->c;
    context->txn     = dbc_struct_i(c)->txn;
3342 3343
    context->db      = c->dbp;
    context->f_extra = extra;
3344
    context->is_write_op = is_write_op;
3345
    u_int32_t lock_flags = get_cursor_prelocked_flags(flag, c);
3346 3347
    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
3348
    context->do_locking = (BOOL)(context->db->i->lt!=NULL && !(lock_flags & (DB_PRELOCKED|DB_PRELOCKED_WRITE)));
3349
    context->r_user_callback = 0;
Yoni Fogel's avatar
Yoni Fogel committed
3350 3351
}

3352
static void
3353 3354
query_context_init_read(QUERY_CONTEXT context, DBC *c, u_int32_t flag, YDB_CALLBACK_FUNCTION f, void *extra) {
    BOOL is_write = FALSE;
3355 3356 3357 3358 3359
    query_context_base_init(&context->base, c, flag, is_write, extra);
    context->f = f;
}

static void
3360 3361
query_context_init_write(QUERY_CONTEXT context, DBC *c, u_int32_t flag, YDB_CALLBACK_FUNCTION f, void *extra) {
    BOOL is_write = TRUE;
3362
    query_context_base_init(&context->base, c, flag, is_write, extra);
3363 3364
    context->f = f;
}
Yoni Fogel's avatar
Yoni Fogel committed
3365

3366 3367
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) {
3368 3369
    // 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;
3370
    query_context_base_init(&context->base, c, flag, is_write, extra);
3371 3372 3373
    context->f         = f;
    context->input_key = key;
    context->input_val = val;
3374 3375
}

3376 3377 3378 3379 3380
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) {
3381
    HANDLE_PANICKED_DB(c->dbp);
3382
    HANDLE_CURSOR_ILLEGAL_WORKING_PARENT_TXN(c);
3383

3384 3385 3386 3387
    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;
3388
    u_int32_t lock_flags = get_cursor_prelocked_flags(flags, c);
3389
    unchecked_flags &= ~lock_flags;
3390
    BOOL do_locking = (BOOL)(c->dbp->i->lt && !(lock_flags&DB_PRELOCKED_WRITE));
3391

3392 3393 3394 3395 3396
    int r = 0;
    if (unchecked_flags!=0) r = EINVAL;
    else {
        if (do_locking) {
            QUERY_CONTEXT_S context;
3397
            query_context_init_write(&context, c, lock_flags, NULL, NULL);
3398 3399 3400 3401 3402
            //We do not need a read lock, we must already have it.
            r = toku_c_getf_current_binding(c, DB_PRELOCKED, c_del_callback, &context);
        }
        if (r==0) {
            //Do the actual delete.
3403 3404
            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);
3405 3406
        }
    }
3407 3408
    return r;
}
3409

3410 3411 3412 3413 3414
//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
3415

3416
    int r;
Yoni Fogel's avatar
Yoni Fogel committed
3417

3418
    assert(context->do_locking);
3419
    assert(context->is_write_op);
3420 3421 3422 3423 3424
    assert(key!=NULL);
    assert(val!=NULL);
    //Lock:
    //  left(key,val)==right(key,val) == (key, val);
    RANGE_LOCK_REQUEST_S request;
3425
    write_lock_request_init(&request, context->txn, context->db, key, key);
3426 3427 3428
    r = grab_range_lock(&request);

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

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

3434 3435 3436 3437 3438 3439 3440 3441 3442 3443 3444
static void c_query_context_init(QUERY_CONTEXT context, DBC *c, u_int32_t flag, YDB_CALLBACK_FUNCTION f, void *extra) {
    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);
}

3445 3446 3447
static int
toku_c_getf_first(DBC *c, u_int32_t flag, YDB_CALLBACK_FUNCTION f, void *extra) {
    HANDLE_PANICKED_DB(c->dbp);
3448
    HANDLE_CURSOR_ILLEGAL_WORKING_PARENT_TXN(c);
3449
    num_point_queries++;   // accountability
3450
    QUERY_CONTEXT_S context; //Describes the context of this query.
3451
    c_query_context_init(&context, c, flag, f, extra);
3452
    //toku_brt_cursor_first will call c_getf_first_callback(..., context) (if query is successful)
3453
    int r = toku_brt_cursor_first(dbc_struct_i(c)->c, c_getf_first_callback, &context);
3454 3455 3456 3457 3458 3459 3460 3461 3462 3463 3464 3465 3466 3467 3468 3469 3470 3471 3472 3473
    if (r == TOKUDB_USER_CALLBACK_ERROR) r = context.base.r_user_callback;
    return r;
}

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

    int r;

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

    if (context->do_locking) {
        RANGE_LOCK_REQUEST_S request;
        if (key!=NULL) {
3474 3475 3476 3477 3478
            range_lock_request_init(&request, context->is_write_op, context->txn, context->db,
                                    toku_lt_neg_infinity, &found_key);
        } else {
            range_lock_request_init(&request, context->is_write_op, context->txn, context->db,
                                    toku_lt_neg_infinity, toku_lt_infinity);
3479 3480 3481 3482 3483 3484 3485 3486
        }
        r = grab_range_lock(&request);
    }
    else r = 0;

    //Call application-layer callback if found and locks were successfully obtained.
    if (r==0 && key!=NULL) {
        context->r_user_callback = super_context->f(&found_key, &found_val, context->f_extra);
3487
        r = context->r_user_callback;
3488 3489 3490 3491 3492 3493 3494 3495 3496 3497 3498
    }

    //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);
3499
    HANDLE_CURSOR_ILLEGAL_WORKING_PARENT_TXN(c);
3500
    num_point_queries++;   // accountability
3501
    QUERY_CONTEXT_S context; //Describes the context of this query.
3502
    c_query_context_init(&context, c, flag, f, extra); 
3503
    //toku_brt_cursor_last will call c_getf_last_callback(..., context) (if query is successful)
3504
    int r = toku_brt_cursor_last(dbc_struct_i(c)->c, c_getf_last_callback, &context);
3505 3506 3507 3508 3509 3510 3511 3512 3513 3514 3515 3516 3517 3518 3519 3520 3521 3522 3523 3524
    if (r == TOKUDB_USER_CALLBACK_ERROR) r = context.base.r_user_callback;
    return r;
}

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

    int r;

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

    if (context->do_locking) {
        RANGE_LOCK_REQUEST_S request;
        if (key!=NULL) {
3525 3526 3527 3528 3529
            range_lock_request_init(&request, context->is_write_op, context->txn, context->db,
                                    &found_key,           toku_lt_infinity);
        } else {
            range_lock_request_init(&request, context->is_write_op, context->txn, context->db,
                                    toku_lt_neg_infinity, toku_lt_infinity);
3530 3531 3532 3533 3534 3535 3536 3537
        }
        r = grab_range_lock(&request);
    }
    else r = 0;

    //Call application-layer callback if found and locks were successfully obtained.
    if (r==0 && key!=NULL) {
        context->r_user_callback = super_context->f(&found_key, &found_val, context->f_extra);
3538
        r = context->r_user_callback;
3539 3540 3541 3542 3543 3544 3545 3546 3547 3548 3549 3550
    }

    //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);
3551
    HANDLE_CURSOR_ILLEGAL_WORKING_PARENT_TXN(c);
3552
    if (toku_c_uninitialized(c)) r = toku_c_getf_first(c, flag, f, extra);
3553 3554
    else {
        QUERY_CONTEXT_S context; //Describes the context of this query.
3555
        c_query_context_init(&context, c, flag, f, extra); 
3556
        //toku_brt_cursor_next will call c_getf_next_callback(..., context) (if query is successful)
3557
        r = toku_brt_cursor_next(dbc_struct_i(c)->c, c_getf_next_callback, &context);
3558 3559 3560 3561 3562 3563 3564 3565 3566 3567 3568 3569
        if (r == TOKUDB_USER_CALLBACK_ERROR) r = context.base.r_user_callback;
    }
    return r;
}

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

    int r;
3570
    num_sequential_queries++;   // accountability
3571 3572 3573 3574 3575 3576 3577 3578 3579 3580 3581 3582 3583

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

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

        toku_brt_cursor_peek(context->c, &prevkey, &prevval);
3584 3585
        range_lock_request_init(&request, context->is_write_op, context->txn, context->db,
                                prevkey, right_key);
3586 3587 3588 3589 3590 3591 3592
        r = grab_range_lock(&request);
    }
    else r = 0;

    //Call application-layer callback if found and locks were successfully obtained.
    if (r==0 && key!=NULL) {
        context->r_user_callback = super_context->f(&found_key, &found_val, context->f_extra);
3593
        r = context->r_user_callback;
3594 3595 3596 3597 3598 3599 3600 3601 3602 3603 3604 3605
    }

    //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);
3606
    HANDLE_CURSOR_ILLEGAL_WORKING_PARENT_TXN(c);
3607
    if (toku_c_uninitialized(c)) r = toku_c_getf_last(c, flag, f, extra);
3608 3609
    else {
        QUERY_CONTEXT_S context; //Describes the context of this query.
3610
        c_query_context_init(&context, c, flag, f, extra); 
3611
        //toku_brt_cursor_prev will call c_getf_prev_callback(..., context) (if query is successful)
3612
        r = toku_brt_cursor_prev(dbc_struct_i(c)->c, c_getf_prev_callback, &context);
3613
        if (r == TOKUDB_USER_CALLBACK_ERROR) r = context.base.r_user_callback;
Yoni Fogel's avatar
Yoni Fogel committed
3614 3615 3616 3617
    }
    return r;
}

3618 3619 3620 3621 3622 3623 3624 3625
//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;

3626
    num_sequential_queries++;   // accountability
3627 3628 3629 3630 3631 3632 3633 3634 3635 3636 3637 3638
    DBT found_key;
    DBT found_val;
    toku_fill_dbt(&found_key, key, keylen);
    toku_fill_dbt(&found_val, val, vallen);

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

        toku_brt_cursor_peek(context->c, &prevkey, &prevval);
3639 3640
        range_lock_request_init(&request, context->is_write_op, context->txn, context->db,
                                left_key, prevkey);
3641 3642 3643 3644 3645 3646 3647
        r = grab_range_lock(&request);
    }
    else r = 0;

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

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

3655 3656 3657 3658 3659
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);
3660
    HANDLE_CURSOR_ILLEGAL_WORKING_PARENT_TXN(c);
3661 3662

    QUERY_CONTEXT_S context; //Describes the context of this query.
3663
    num_sequential_queries++;   // accountability
3664
    c_query_context_init(&context, c, flag, f, extra); 
3665
    //toku_brt_cursor_current will call c_getf_current_callback(..., context) (if query is successful)
3666
    int r = toku_brt_cursor_current(dbc_struct_i(c)->c, DB_CURRENT, c_getf_current_callback, &context);
3667 3668 3669 3670 3671 3672 3673 3674 3675 3676 3677 3678 3679 3680 3681 3682 3683 3684 3685
    if (r == TOKUDB_USER_CALLBACK_ERROR) r = context.base.r_user_callback;
    return r;
}

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

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

    int r=0;
    //Call application-layer callback if found.
    if (key!=NULL) {
        context->r_user_callback = super_context->f(&found_key, &found_val, context->f_extra);
3686
        r = context->r_user_callback;
3687 3688 3689 3690 3691 3692 3693 3694 3695
    }

    //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);
3696
    HANDLE_CURSOR_ILLEGAL_WORKING_PARENT_TXN(c);
3697 3698

    QUERY_CONTEXT_S context; //Describes the context of this query.
3699
    num_sequential_queries++;   // accountability
3700
    c_query_context_init(&context, c, flag, f, extra); 
3701
    //toku_brt_cursor_current will call c_getf_current_callback(..., context) (if query is successful)
3702
    int r = toku_brt_cursor_current(dbc_struct_i(c)->c, DB_CURRENT_BINDING, c_getf_current_callback, &context);
3703 3704 3705 3706 3707 3708 3709 3710 3711
    if (r == TOKUDB_USER_CALLBACK_ERROR) r = context.base.r_user_callback;
    return r;
}

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

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

    QUERY_CONTEXT_WITH_INPUT_S context; //Describes the context of this query.
3715
    num_point_queries++;   // accountability
3716 3717
    query_context_with_input_init(&context, c, flag, key, NULL, f, extra); 
    //toku_brt_cursor_set will call c_getf_set_callback(..., context) (if query is successful)
3718
    int r = toku_brt_cursor_set(dbc_struct_i(c)->c, key, c_getf_set_callback, &context);
3719 3720 3721 3722 3723 3724 3725 3726 3727 3728 3729 3730 3731 3732 3733 3734 3735 3736 3737 3738 3739 3740
    if (r == TOKUDB_USER_CALLBACK_ERROR) r = context.base.r_user_callback;
    return r;
}

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

    int r;

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

    //Lock:
    //  left(key,val)  = (input_key, -infinity)
    //  right(key,val) = (input_key, found ? found_val : infinity)
    if (context->do_locking) {
        RANGE_LOCK_REQUEST_S request;
3741 3742
        range_lock_request_init(&request, context->is_write_op, context->txn, context->db, 
                                super_context->input_key, super_context->input_key);
3743
        r = grab_range_lock(&request);
3744 3745
    } else 
        r = 0;
3746 3747 3748 3749

    //Call application-layer callback if found and locks were successfully obtained.
    if (r==0 && key!=NULL) {
        context->r_user_callback = super_context->f(&found_key, &found_val, context->f_extra);
3750
        r = context->r_user_callback;
3751 3752 3753 3754 3755 3756 3757 3758 3759 3760 3761
    }

    //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);
3762
    HANDLE_CURSOR_ILLEGAL_WORKING_PARENT_TXN(c);
3763 3764

    QUERY_CONTEXT_WITH_INPUT_S context; //Describes the context of this query.
3765
    num_point_queries++;   // accountability
3766 3767
    query_context_with_input_init(&context, c, flag, key, NULL, f, extra); 
    //toku_brt_cursor_set_range will call c_getf_set_range_callback(..., context) (if query is successful)
3768
    int r = toku_brt_cursor_set_range(dbc_struct_i(c)->c, key, c_getf_set_range_callback, &context);
3769 3770 3771 3772 3773 3774 3775 3776 3777 3778 3779 3780 3781 3782 3783 3784 3785 3786 3787 3788 3789 3790 3791
    if (r == TOKUDB_USER_CALLBACK_ERROR) r = context.base.r_user_callback;
    return r;
}

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

    int r;

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

    //Lock:
    //  left(key,val)  = (input_key, -infinity)
    //  right(key) = found ? found_key : infinity
    //  right(val) = found ? found_val : infinity
    if (context->do_locking) {
        RANGE_LOCK_REQUEST_S request;
3792 3793 3794 3795 3796 3797
        if (key!=NULL)
            range_lock_request_init(&request, context->is_write_op, context->txn, context->db, 
                                    super_context->input_key, &found_key);
        else
            range_lock_request_init(&request, context->is_write_op, context->txn, context->db, 
                                    super_context->input_key, toku_lt_infinity);
3798 3799 3800 3801 3802 3803 3804
        r = grab_range_lock(&request);
    }
    else r = 0;

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

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

3812 3813 3814 3815 3816 3817 3818 3819
static int c_getf_set_range_reverse_callback(ITEMLEN keylen, bytevec key, ITEMLEN vallen, bytevec val, void *extra);

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

    QUERY_CONTEXT_WITH_INPUT_S context; //Describes the context of this query.
3820
    num_point_queries++;   // accountability
3821 3822
    query_context_with_input_init(&context, c, flag, key, NULL, f, extra); 
    //toku_brt_cursor_set_range_reverse will call c_getf_set_range_reverse_callback(..., context) (if query is successful)
3823
    int r = toku_brt_cursor_set_range_reverse(dbc_struct_i(c)->c, key, c_getf_set_range_reverse_callback, &context);
3824 3825 3826 3827 3828 3829 3830 3831 3832 3833 3834 3835 3836 3837 3838 3839 3840 3841 3842 3843 3844 3845 3846 3847
    if (r == TOKUDB_USER_CALLBACK_ERROR) r = context.base.r_user_callback;
    return r;
}

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

    int r;

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

    //Lock:
    //  left(key) = found ? found_key : -infinity
    //  left(val) = found ? found_val : -infinity
    //  right(key,val)  = (input_key, infinity)
    if (context->do_locking) {
        RANGE_LOCK_REQUEST_S request;
        if (key!=NULL) {
3848 3849 3850 3851 3852
            range_lock_request_init(&request, context->is_write_op, context->txn, context->db,
                                    &found_key, super_context->input_key);
        } else {
            range_lock_request_init(&request, context->is_write_op, context->txn, context->db,
                                    toku_lt_neg_infinity, super_context->input_key);
3853 3854 3855 3856 3857 3858 3859 3860
        }
        r = grab_range_lock(&request);
    }
    else r = 0;

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

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

3868
static int toku_c_close(DBC * c) {
3869 3870
    HANDLE_PANICKED_DB(c->dbp);
    HANDLE_CURSOR_ILLEGAL_WORKING_PARENT_TXN(c);
3871 3872 3873 3874 3875 3876
    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
3877
    toku_free(c);
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
3878 3879 3880
    return r;
}

3881 3882
static inline int 
keyeq(DBC *c, DBT *a, DBT *b) {
3883 3884 3885 3886
    DB *db = c->dbp;
    return db->i->brt->compare_fun(db, a, b) == 0;
}

3887 3888 3889 3890
// 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) {
3891 3892
    HANDLE_PANICKED_DB(cursor->dbp);
    HANDLE_CURSOR_ILLEGAL_WORKING_PARENT_TXN(cursor);
Rich Prohaska's avatar
Rich Prohaska committed
3893 3894
    int r;
    DBC *count_cursor = 0;
3895
    DBT currentkey;
Rich Prohaska's avatar
Rich Prohaska committed
3896

3897
    init_dbt_realloc(&currentkey);
3898
    u_int32_t lock_flags = get_cursor_prelocked_flags(flags, cursor);
3899
    flags &= ~lock_flags;
Rich Prohaska's avatar
Rich Prohaska committed
3900 3901 3902 3903
    if (flags != 0) {
        r = EINVAL; goto finish;
    }

3904
    r = toku_c_get_current_unconditional(cursor, lock_flags, &currentkey, NULL);
Rich Prohaska's avatar
Rich Prohaska committed
3905
    if (r != 0) goto finish;
3906 3907 3908 3909 3910 3911

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

3916
    r = toku_c_getf_set(count_cursor, lock_flags, &currentkey, ydb_getf_do_nothing, NULL);
3917 3918 3919 3920
    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
3921
    }
3922
    r = 0;
Rich Prohaska's avatar
Rich Prohaska committed
3923 3924 3925 3926 3927 3928 3929 3930
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
3931

3932 3933
///////////
//db_getf_XXX is equivalent to c_getf_XXX, without a persistent cursor
Yoni Fogel's avatar
 
Yoni Fogel committed
3934

3935 3936
static int
db_getf_set(DB *db, DB_TXN *txn, u_int32_t flags, DBT *key, YDB_CALLBACK_FUNCTION f, void *extra) {
3937 3938
    HANDLE_PANICKED_DB(db);
    HANDLE_DB_ILLEGAL_WORKING_PARENT_TXN(db, txn);
3939
    DBC *c;
3940
    uint32_t create_flags = flags & (DB_ISOLATION_FLAGS | DB_RMW);
3941
    flags &= ~DB_ISOLATION_FLAGS;
3942
    int r = toku_db_cursor(db, txn, &c, create_flags, 1);
3943 3944 3945 3946
    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
3947
    }
3948
    return r;
Yoni Fogel's avatar
 
Yoni Fogel committed
3949 3950
}

3951
////////////
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
3952

3953 3954 3955 3956 3957 3958 3959 3960 3961
static int 
get_point_lock(DB *db, DB_TXN *txn, const DBT *key) {
    RANGE_LOCK_REQUEST_S request;
    //Left end of range == right end of range (point lock)
    write_lock_request_init(&request, txn, db, key, key);
    int r = grab_range_lock(&request);
    return r;
}

3962 3963
static int
toku_db_del(DB *db, DB_TXN *txn, DBT *key, u_int32_t flags) {
3964
    HANDLE_PANICKED_DB(db);
3965
    HANDLE_DB_ILLEGAL_WORKING_PARENT_TXN(db, txn);
3966

3967 3968
    u_int32_t unchecked_flags = flags;
    //DB_DELETE_ANY means delete regardless of whether it exists in the db.
3969
    BOOL error_if_missing = (BOOL)(!(flags&DB_DELETE_ANY));
3970
    unchecked_flags &= ~DB_DELETE_ANY;
3971
    u_int32_t lock_flags = get_prelocked_flags(flags);
3972
    unchecked_flags &= ~lock_flags;
3973
    BOOL do_locking = (BOOL)(db->i->lt && !(lock_flags&DB_PRELOCKED_WRITE));
3974
    BOOL do_dir_locking = !(lock_flags&DB_PRELOCKED_FILE_READ);
3975

3976
    int r = 0;
3977 3978 3979
    if (unchecked_flags!=0) 
        r = EINVAL;

3980
    if (r == 0 && do_dir_locking) {
3981
        r = toku_grab_read_lock_on_directory(db, txn);
Zardosht Kasheff's avatar
Zardosht Kasheff committed
3982
    }
3983
    if (r == 0 && error_if_missing) {
3984
        //Check if the key exists in the db.
3985
        r = db_getf_set(db, txn, lock_flags|DB_SERIALIZABLE|DB_RMW, key, ydb_getf_do_nothing, NULL);
3986
    }
3987
    if (r == 0 && do_locking) {
3988
        //Do locking if necessary.
3989
        r = get_point_lock(db, txn, key);
3990
    }
3991
    if (r == 0) {
3992
        //Do the actual deleting.
3993
        r = toku_brt_delete(db->i->brt, key, txn ? db_txn_struct_i(txn)->tokutxn : 0);
3994
    }
3995

3996 3997 3998
    if (r == 0) 
        num_deletes++;       // accountability 
    else
Zardosht Kasheff's avatar
Zardosht Kasheff committed
3999
        num_deletes_fail++;
4000 4001 4002 4003 4004 4005 4006 4007 4008 4009 4010

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

4011 4012 4013 4014 4015 4016 4017 4018
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;
}

4019
static int
4020
log_del_multiple(DB_TXN *txn, DB *src_db, const DBT *key, const DBT *val, uint32_t num_dbs, BRT brts[], DBT keys[]) {
4021 4022 4023 4024
    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
4025 4026
        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);
4027 4028 4029 4030 4031 4032
        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);
        }
4033 4034 4035 4036
    }
    return r;
}

4037 4038 4039 4040 4041 4042 4043 4044 4045
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;
}

4046
static int
4047 4048
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;
4049 4050 4051 4052
    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];
4053 4054 4055

        // 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.
4056 4057
	int do_delete = TRUE;
	DB_INDEXER *indexer = toku_db_get_indexer(db);
4058
	if (indexer) { // if this db is the index under construction
4059 4060 4061 4062 4063 4064 4065 4066 4067 4068 4069
            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);
4070 4071 4072 4073
        }
	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
4074
    }
4075
    return r;
Rich Prohaska's avatar
Rich Prohaska committed
4076 4077
}

4078
static int
4079 4080 4081 4082
env_del_multiple(
    DB_ENV *env, 
    DB *src_db, 
    DB_TXN *txn, 
4083 4084
    const DBT *src_key, 
    const DBT *src_val, 
4085 4086 4087
    uint32_t num_dbs, 
    DB **db_array, 
    DBT *keys, 
4088
    uint32_t *flags_array) 
4089
{
4090
    int r;
4091
    DBT del_keys[num_dbs];
4092 4093 4094 4095

    HANDLE_PANICKED_ENV(env);

    if (!txn) {
4096 4097 4098
        r = EINVAL;
        goto cleanup;
    }
4099
    if (!env->i->generate_row_for_del) {
4100 4101 4102 4103
        r = EINVAL;
        goto cleanup;
    }

4104 4105 4106 4107 4108 4109 4110 4111
    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++) {
4112
        DB *db = db_array[which_db];
4113 4114
        lock_flags[which_db] = get_prelocked_flags(flags_array[which_db]);
        remaining_flags[which_db] = flags_array[which_db] & ~lock_flags[which_db];
4115

4116 4117 4118 4119 4120
        //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;
        }
4121
        if (db == src_db) {
4122
            del_keys[which_db] = *src_key;
4123 4124
        }
        else {
4125
        //Generate the key
4126
            r = env->i->generate_row_for_del(db, src_db, &keys[which_db], src_key, src_val);
4127 4128 4129
            if (r != 0) goto cleanup;
            del_keys[which_db] = keys[which_db];
        }
4130 4131 4132 4133 4134 4135 4136 4137

        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.
4138
            r = db_getf_set(db, txn, lock_flags[which_db]|DB_SERIALIZABLE|DB_RMW, &del_keys[which_db], ydb_getf_do_nothing, NULL);
4139
            if (r != 0) goto cleanup;
4140 4141 4142 4143 4144
        }

        //Do locking if necessary.
        if (db->i->lt && !(lock_flags[which_db] & DB_PRELOCKED_WRITE)) {
            //Needs locking
4145
            r = get_point_lock(db, txn, &del_keys[which_db]);
4146
            if (r != 0) goto cleanup;
4147 4148 4149
        }
        brts[which_db] = db->i->brt;
    }
4150 4151

    if (num_dbs == 1)
4152
        r = log_del_single(txn, brts[0], &del_keys[0]);
4153
    else
4154
        r = log_del_multiple(txn, src_db, src_key, src_val, num_dbs, brts, del_keys);
4155 4156

    if (r == 0) 
4157
        r = do_del_multiple(txn, num_dbs, db_array, del_keys, src_db, src_key);
4158 4159 4160
    }

cleanup:
4161 4162 4163 4164
    if (r == 0)
        num_multi_deletes += num_dbs;
    else
        num_multi_deletes_fail += num_dbs;
4165 4166 4167 4168
    return r;
}


4169 4170
static int 
locked_c_get(DBC * c, DBT * key, DBT * data, u_int32_t flag) {
4171 4172
    toku_ydb_lock(); int r = toku_c_get(c, key, data, flag); toku_ydb_unlock();
    return r;
Rich Prohaska's avatar
Rich Prohaska committed
4173 4174
}

4175 4176
static int 
locked_c_close(DBC * c) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
4177
    toku_ydb_lock(); int r = toku_c_close(c); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
4178 4179
}

4180 4181
static int 
locked_c_count(DBC *cursor, db_recno_t *count, u_int32_t flags) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
4182
    toku_ydb_lock(); int r = toku_c_count(cursor, count, flags); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
4183 4184
}

4185 4186
static int 
locked_c_del(DBC * c, u_int32_t flags) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
4187
    toku_ydb_lock(); int r = toku_c_del(c, flags); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
4188 4189
}

4190
static int locked_c_pre_acquire_range_lock(DBC *dbc, const DBT *key_left, const DBT *key_right);
4191

4192 4193
static int 
toku_db_cursor(DB * db, DB_TXN * txn, DBC ** c, u_int32_t flags, int is_temporary_cursor) {
4194
    HANDLE_PANICKED_DB(db);
4195
    HANDLE_DB_ILLEGAL_WORKING_PARENT_TXN(db, txn);
4196
    DB_ENV* env = db->dbenv;
Zardosht Kasheff's avatar
Zardosht Kasheff committed
4197
    int r;
4198
    size_t result_size = sizeof(DBC)+sizeof(struct __toku_dbc_internal); // internal stuff stuck on the end
4199
    if (flags & ~(DB_SERIALIZABLE | DB_INHERIT_ISOLATION | DB_RMW)) {
4200 4201 4202
        return toku_ydb_do_error(
            env, 
            EINVAL, 
4203
            "Invalid flags set for toku_db_cursor\n"
4204 4205
            );
    }
Zardosht Kasheff's avatar
Zardosht Kasheff committed
4206
    r = toku_grab_read_lock_on_directory(db, txn);
4207
    if (r != 0) 
Zardosht Kasheff's avatar
Zardosht Kasheff committed
4208 4209
        return r;
    
4210
    DBC *result = toku_malloc(result_size);
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
4211 4212
    if (result == 0)
        return ENOMEM;
4213
    memset(result, 0, result_size);
Yoni Fogel's avatar
Yoni Fogel committed
4214 4215 4216 4217 4218
#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
4219 4220
    SCRS(c_getf_first);
    SCRS(c_getf_last);
Yoni Fogel's avatar
Yoni Fogel committed
4221
    SCRS(c_getf_next);
4222 4223 4224 4225 4226
    SCRS(c_getf_prev);
    SCRS(c_getf_current);
    SCRS(c_getf_current_binding);
    SCRS(c_getf_set);
    SCRS(c_getf_set_range);
4227
    SCRS(c_getf_set_range_reverse);
4228
    SCRS(c_pre_acquire_range_lock);
Yoni Fogel's avatar
Yoni Fogel committed
4229
#undef SCRS
4230 4231 4232

#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
4233
    assert(result->i);
4234
#endif
4235
    result->dbp = db;
4236

4237 4238 4239
    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};
4240
    if (is_temporary_cursor) {
4241 4242
	dbc_struct_i(result)->skey = &db->i->skey;
	dbc_struct_i(result)->sval = &db->i->sval;
4243
    } else {
4244 4245
	dbc_struct_i(result)->skey = &dbc_struct_i(result)->skey_s;
	dbc_struct_i(result)->sval = &dbc_struct_i(result)->sval_s;
4246
    }
4247 4248 4249 4250
    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;
4251
    }
4252
    dbc_struct_i(result)->rmw = (flags & DB_RMW) != 0;
4253
    BOOL is_snapshot_read = FALSE;
4254
    if (txn) {
4255 4256
        is_snapshot_read = (dbc_struct_i(result)->iso == TOKU_ISO_READ_COMMITTED || 
                            dbc_struct_i(result)->iso == TOKU_ISO_SNAPSHOT);
4257
    }
Zardosht Kasheff's avatar
Zardosht Kasheff committed
4258
    r = toku_brt_cursor(
4259
        db->i->brt, 
4260 4261 4262
        &dbc_struct_i(result)->c,
        txn ? db_txn_struct_i(txn)->tokutxn : NULL,
        is_snapshot_read
4263
        );
4264 4265 4266 4267 4268 4269 4270 4271 4272 4273 4274
    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
4275 4276
}

4277 4278
static inline int 
db_thread_need_flags(DBT *dbt) {
Rich Prohaska's avatar
Rich Prohaska committed
4279 4280 4281
    return (dbt->flags & (DB_DBT_MALLOC+DB_DBT_REALLOC+DB_DBT_USERMEM)) == 0;
}

4282 4283
static int 
toku_db_get (DB * db, DB_TXN * txn, DBT * key, DBT * data, u_int32_t flags) {
4284
    HANDLE_PANICKED_DB(db);
4285
    HANDLE_DB_ILLEGAL_WORKING_PARENT_TXN(db, txn);
Yoni Fogel's avatar
Yoni Fogel committed
4286
    int r;
4287
    u_int32_t iso_flags = flags & DB_ISOLATION_FLAGS;
4288

Rich Prohaska's avatar
Rich Prohaska committed
4289
    if ((db->i->open_flags & DB_THREAD) && db_thread_need_flags(data))
4290 4291
        return EINVAL;

4292
    u_int32_t lock_flags = flags & (DB_PRELOCKED | DB_PRELOCKED_WRITE | DB_PRELOCKED_FILE_READ);
Yoni Fogel's avatar
Yoni Fogel committed
4293
    flags &= ~lock_flags;
4294
    flags &= ~DB_ISOLATION_FLAGS;
4295 4296 4297
    // And DB_GET_BOTH is no longer supported. #2862.
    if (flags != 0) return EINVAL;

Yoni Fogel's avatar
Yoni Fogel committed
4298 4299

    DBC *dbc;
4300
    r = toku_db_cursor(db, txn, &dbc, iso_flags, 1);
Yoni Fogel's avatar
Yoni Fogel committed
4301
    if (r!=0) return r;
4302
    u_int32_t c_get_flags = DB_SET;
Yoni Fogel's avatar
Yoni Fogel committed
4303
    r = toku_c_get(dbc, key, data, c_get_flags | lock_flags);
Yoni Fogel's avatar
Yoni Fogel committed
4304 4305
    int r2 = toku_c_close(dbc);
    return r ? r : r2;
4306 4307
}

Rich Prohaska's avatar
Rich Prohaska committed
4308
#if 0
4309 4310
static int 
toku_db_key_range(DB * db, DB_TXN * txn, DBT * dbt, DB_KEY_RANGE * kr, u_int32_t flags) {
4311
    HANDLE_PANICKED_DB(db);
4312
    HANDLE_DB_ILLEGAL_WORKING_PARENT_TXN(db, txn);
4313
    txn=txn; dbt=dbt; kr=kr; flags=flags;
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
4314
    toku_ydb_barf();
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
4315
    abort();
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
4316
}
Rich Prohaska's avatar
Rich Prohaska committed
4317
#endif
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
4318

4319 4320
static int 
toku_db_lt_panic(DB* db, int r) {
4321
    assert(r!=0);
Yoni Fogel's avatar
Yoni Fogel committed
4322 4323
    assert(db && db->i && db->dbenv && db->dbenv->i);
    DB_ENV* env = db->dbenv;
4324 4325 4326 4327
    char * panic_string;

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

4329
    env_panic(env, r, panic_string);
4330

4331
    return toku_ydb_do_error(env, r, "%s", panic_string);
Yoni Fogel's avatar
Yoni Fogel committed
4332 4333
}

4334 4335
static int 
toku_txn_add_lt(DB_TXN* txn, toku_lock_tree* lt) {
Yoni Fogel's avatar
Yoni Fogel committed
4336
    int r = ENOSYS;
Yoni Fogel's avatar
Yoni Fogel committed
4337
    assert(txn && lt);
4338
    toku_lth* lth = db_txn_struct_i(txn)->lth;
Yoni Fogel's avatar
Yoni Fogel committed
4339 4340 4341 4342 4343
    assert(lth);

    toku_lock_tree* find = toku_lth_find(lth, lt);
    if (find) {
        assert(find == lt);
Yoni Fogel's avatar
Yoni Fogel committed
4344 4345
        r = 0;
        goto cleanup;
Yoni Fogel's avatar
Yoni Fogel committed
4346
    }
Yoni Fogel's avatar
Yoni Fogel committed
4347 4348 4349 4350 4351 4352
    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
4353 4354 4355
    return r;
}

4356 4357
static toku_dbt_cmp 
toku_db_get_compare_fun(DB* db) {
Yoni Fogel's avatar
Yoni Fogel committed
4358 4359
    return db->i->brt->compare_fun;
}
Yoni Fogel's avatar
Yoni Fogel committed
4360

4361

4362 4363 4364 4365 4366
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 {
4367 4368 4369 4370 4371 4372
        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);
    }
4373 4374
    return r;
}
4375

4376 4377 4378 4379 4380 4381 4382 4383
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)) {
4384 4385
            char c = *dname++;
            *hint++ = c;
4386 4387 4388 4389 4390 4391 4392
            underscored = FALSE;
        }
        else {
            if (!underscored)
                *hint++ = '_';
            dname++;
            underscored = TRUE;
4393 4394
        }
    }
4395 4396 4397
    *hint = '\0';
}

4398

4399
// n < 0  means to ignore mark and ignore n
4400
// n >= 0 means to include mark ("_B_" or "_P_") with hex value of n in iname
4401
// (intended for use by loader, which will create many inames using one txnid).
4402
static char *
4403
create_iname(DB_ENV *env, u_int64_t id, char *hint, char *mark, int n) {
4404
    int bytes;
4405
    char inamebase[strlen(hint) +
4406 4407 4408
		   8 +  // hex file format version
		   16 + // hex id (normally the txnid)
		   8  + // hex value of n if non-neg
4409
		   sizeof("_B___.tokudb")]; // extra pieces
4410
    if (n < 0)
4411 4412 4413
	bytes = snprintf(inamebase, sizeof(inamebase),
                         "%s_%"PRIx64"_%"PRIx32            ".tokudb",
                         hint, id, BRT_LAYOUT_VERSION);
4414 4415
    else {
	invariant(strlen(mark) == 1);
4416
	bytes = snprintf(inamebase, sizeof(inamebase),
4417 4418 4419
                         "%s_%"PRIx64"_%"PRIx32"_%s_%"PRIx32".tokudb",
                         hint, id, BRT_LAYOUT_VERSION, mark, n);
    }
4420 4421 4422 4423
    assert(bytes>0);
    assert(bytes<=(int)sizeof(inamebase)-1);
    char *rval;
    if (env->i->data_dir)
Yoni Fogel's avatar
Yoni Fogel committed
4424
        rval = toku_construct_full_name(2, env->i->data_dir, inamebase);
4425
    else
Yoni Fogel's avatar
Yoni Fogel committed
4426
        rval = toku_construct_full_name(1, inamebase);
4427 4428
    assert(rval);
    return rval;
4429 4430
}

4431 4432 4433 4434 4435 4436 4437 4438 4439 4440 4441 4442 4443

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) {
4444
    HANDLE_PANICKED_DB(db);
4445
    HANDLE_DB_ILLEGAL_WORKING_PARENT_TXN(db, txn);
4446
    if (dbname!=NULL) 
4447
        return db_open_subdb(db, txn, fname, dbname, dbtype, flags, mode);
4448

4449
    // at this point fname is the dname
4450 4451
    //This code ONLY supports single-db files.
    assert(dbname==NULL);
4452
    const char * dname = fname;  // db_open_subdb() converts (fname, dbname) to dname
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
4453

4454 4455 4456
    ////////////////////////////// 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
4457
    int r;
Yoni Fogel's avatar
Yoni Fogel committed
4458
    if (dbtype!=DB_BTREE && dbtype!=DB_UNKNOWN) return EINVAL;
4459 4460
    int is_db_excl    = flags & DB_EXCL;    unused_flags&=~DB_EXCL;
    int is_db_create  = flags & DB_CREATE;  unused_flags&=~DB_CREATE;
4461
    int is_db_hot_index  = flags & DB_IS_HOT_INDEX;  unused_flags&=~DB_IS_HOT_INDEX;
4462

4463
    //We support READ_UNCOMMITTED and READ_COMMITTED whether or not the flag is provided.
4464
                                            unused_flags&=~DB_READ_UNCOMMITTED;
4465
                                            unused_flags&=~DB_READ_COMMITTED;
4466
                                            unused_flags&=~DB_SERIALIZABLE;
4467 4468 4469 4470 4471 4472 4473 4474 4475 4476 4477 4478 4479 4480 4481 4482 4483 4484
    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) {
4485 4486
        r = toku_txn_begin(db->dbenv, txn, &child, DB_TXN_NOSYNC, 1);
        assert(r==0);
4487 4488 4489 4490 4491 4492
    }

    // 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
4493
    DBT iname_dbt;  // holds iname_in_env
4494 4495
    toku_fill_dbt(&dname_dbt, dname, strlen(dname)+1);
    init_dbt_realloc(&iname_dbt);  // sets iname_dbt.data = NULL
4496
    r = toku_db_get(db->dbenv->i->directory, child, &dname_dbt, &iname_dbt, DB_SERIALIZABLE);  // allocates memory for iname
4497 4498 4499 4500 4501 4502 4503
    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) {
4504
        char hint[strlen(dname) + 1];
4505

4506 4507 4508 4509 4510 4511 4512
        // 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);
4513
        iname = create_iname(db->dbenv, id, hint, NULL, -1);  // allocated memory for iname
4514
        toku_fill_dbt(&iname_dbt, iname, strlen(iname) + 1);
4515
        //
4516
        // 0 for performance only, avoid unnecessary query
4517 4518 4519
        // 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
        //
4520
        u_int32_t put_flags = 0 | ((is_db_hot_index) ? DB_PRELOCKED_WRITE : 0); 
4521
        r = toku_db_put(db->dbenv->i->directory, child, &dname_dbt, &iname_dbt, put_flags);  
4522 4523 4524 4525
    }

    // we now have an iname
    if (r == 0) {
4526
        r = db_open_iname(db, child, iname, flags, mode);
4527 4528 4529 4530 4531 4532 4533 4534 4535 4536
        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) {
4537 4538 4539
        // close txn
        if (r == 0) {  // commit
            r = toku_txn_commit(child, DB_TXN_NOSYNC, NULL, NULL);
4540
            invariant(r==0);  // TODO panic
4541 4542 4543
        }
        else {         // abort
            int r2 = toku_txn_abort(child, NULL, NULL);
4544
            invariant(r2==0);  // TODO panic
4545
        }
4546 4547 4548 4549 4550 4551
    }

    return r;
}

static int 
4552
db_open_iname(DB * db, DB_TXN * txn, const char *iname_in_env, u_int32_t flags, int mode) {
4553 4554 4555 4556 4557 4558 4559 4560
    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;
    }
4561 4562 4563 4564
    if (db->dbenv->i->update_function) {
        r = toku_brt_set_update(db->i->brt,db->dbenv->i->update_function);
        assert(r==0);
    }
4565 4566 4567
    BOOL need_locktree = (BOOL)((db->dbenv->i->open_flags & DB_INIT_LOCK) &&
                                (db->dbenv->i->open_flags & DB_INIT_TXN));

4568 4569
    int is_db_excl    = flags & DB_EXCL;    flags&=~DB_EXCL;
    int is_db_create  = flags & DB_CREATE;  flags&=~DB_CREATE;
4570
    //We support READ_UNCOMMITTED and READ_COMMITTED whether or not the flag is provided.
4571
                                            flags&=~DB_READ_UNCOMMITTED;
4572
                                            flags&=~DB_READ_COMMITTED;
4573
                                            flags&=~DB_SERIALIZABLE;
4574
                                            flags&=~DB_IS_HOT_INDEX;
4575
    if (flags & ~DB_THREAD) return EINVAL; // unknown flags
4576 4577

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

4579 4580 4581 4582 4583 4584
    /* 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;

4585
    if (db_opened(db))
4586
        return EINVAL;              /* It was already open. */
Yoni Fogel's avatar
Yoni Fogel committed
4587
    
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
4588 4589
    db->i->open_flags = flags;
    db->i->open_mode = mode;
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
4590

Yoni Fogel's avatar
Yoni Fogel committed
4591
    r = toku_brt_open(db->i->brt, iname_in_env,
4592
		      is_db_create, is_db_excl,
4593
		      db->dbenv->i->cachetable,
4594
		      txn ? db_txn_struct_i(txn)->tokutxn : NULL_TXN,
4595
		      db);
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
4596 4597 4598
    if (r != 0)
        goto error_cleanup;

4599
    db->i->opened = 1;
Yoni Fogel's avatar
Yoni Fogel committed
4600
    if (need_locktree) {
4601
	db->i->dict_id = toku_brt_get_dictionary_id(db->i->brt);
4602
        r = toku_ltm_get_lt(db->dbenv->i->ltm, &db->i->lt, db->i->dict_id, db);
Yoni Fogel's avatar
Yoni Fogel committed
4603
        if (r!=0) { goto error_cleanup; }
Yoni Fogel's avatar
Yoni Fogel committed
4604
    }
4605 4606 4607
    //Add to transaction's list of 'must close' if necessary.
    if (txn) {
        //Do last so we don't have to undo.
4608
        toku_list_push(&db_txn_struct_i(txn)->dbs_that_must_close_before_abort,
4609 4610
                  &db->i->dbs_that_must_close_before_abort);
    }
Yoni Fogel's avatar
Yoni Fogel committed
4611

Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
4612
    return 0;
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
4613 4614
 
error_cleanup:
4615
    db->i->dict_id = DICTIONARY_ID_NONE;
4616
    db->i->opened = 0;
Yoni Fogel's avatar
Yoni Fogel committed
4617
    if (db->i->lt) {
4618
        toku_lt_remove_db_ref(db->i->lt, db);
Yoni Fogel's avatar
Yoni Fogel committed
4619 4620
        db->i->lt = NULL;
    }
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
4621
    return r;
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
4622
}
Bradley C. Kuszmaul's avatar
Bradley C. Kuszmaul committed
4623

4624 4625 4626 4627
//Return 0 if proposed pair do not violate size constraints of DB
//(insertion is legal)
//Return non zero otherwise.
static int
4628
db_put_check_size_constraints(DB *db, const DBT *key, const DBT *val) {
4629
    int r;
4630

4631
    //Check limits on size of key and val.
4632
    unsigned int nodesize;
4633
    r = toku_brt_get_nodesize(db->i->brt, &nodesize); assert(r == 0);
4634

4635 4636 4637 4638 4639 4640
    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);
    
4641 4642 4643
    return r;
}

4644 4645 4646 4647 4648 4649 4650 4651 4652 4653 4654 4655 4656 4657 4658 4659 4660 4661 4662 4663 4664
//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;
}

4665 4666
//Return 0 if insert is legal
static int
4667
db_put_check_overwrite_constraint(DB *db, DB_TXN *txn, DBT *key,
4668 4669 4670
                                  u_int32_t lock_flags, u_int32_t overwrite_flag) {
    int r;

4671
    if (overwrite_flag == 0) { // 0 (yesoverwrite) does not impose constraints.
4672 4673
        r = 0;
    } else if (overwrite_flag == DB_NOOVERWRITE) {
4674 4675 4676 4677
        // 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);
4678 4679 4680 4681
        if (r == DB_NOTFOUND) 
            r = 0;
        else if (r == 0)      
            r = DB_KEYEXIST;
4682
        //Any other error is passed through.
4683
    } else if (overwrite_flag == DB_NOOVERWRITE_NO_ERROR) {
4684
        r = 0;
4685
    } else {
4686 4687
        //Other flags are not (yet) supported.
        r = EINVAL;
4688 4689 4690 4691
    }
    return r;
}

4692 4693
static int
toku_db_put(DB *db, DB_TXN *txn, DBT *key, DBT *val, u_int32_t flags) {
4694
    HANDLE_PANICKED_DB(db);
4695
    HANDLE_DB_ILLEGAL_WORKING_PARENT_TXN(db, txn);
4696
    int r = 0;
4697

4698
    u_int32_t lock_flags = get_prelocked_flags(flags);
4699
    flags &= ~lock_flags;
4700

4701 4702 4703 4704
    if (!(lock_flags & DB_PRELOCKED_FILE_READ)) {
        r = toku_grab_read_lock_on_directory(db, txn);
    }
    
4705 4706 4707
    if (r == 0)
        r = db_put_check_size_constraints(db, key, val);
    if (r == 0) {
4708
        //Do any checking required by the flags.
4709
        r = db_put_check_overwrite_constraint(db, txn, key, lock_flags, flags);
4710
    }
4711 4712
    BOOL do_locking = (BOOL)(db->i->lt && !(lock_flags&DB_PRELOCKED_WRITE));
    if (r == 0 && do_locking) {
4713
        //Do locking if necessary.
4714
        r = get_point_lock(db, txn, key);
4715
    }
4716
    if (r == 0) {
4717
        //Insert into the brt.
4718 4719 4720 4721 4722
        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);
4723
    }
4724

4725 4726 4727
    if (r == 0)
	num_inserts++;
    else
Zardosht Kasheff's avatar
Zardosht Kasheff committed
4728
        num_inserts_fail++;
4729 4730 4731 4732

    return r;
}

4733 4734 4735 4736 4737 4738 4739 4740 4741 4742 4743
static int toku_db_pre_acquire_fileops_lock(DB *db, DB_TXN *txn) {
    char *   dname = db->i->dname;
    DBT key_in_directory;
    //
    // bad hack because some environment dictionaries do not have a dname
    //
    if (!dname) {
        return 0;
    }
    toku_fill_dbt(&key_in_directory, dname, strlen(dname)+1);
    //Left end of range == right end of range (point lock)
4744 4745 4746
    RANGE_LOCK_REQUEST_S request;
    write_lock_request_init(&request, txn, db->dbenv->i->directory,
                            &key_in_directory, &key_in_directory);
4747 4748 4749 4750 4751 4752 4753 4754 4755 4756 4757 4758 4759 4760 4761 4762 4763 4764 4765 4766 4767 4768 4769 4770 4771 4772 4773 4774 4775 4776 4777 4778 4779 4780 4781 4782 4783 4784 4785 4786 4787 4788 4789 4790 4791 4792 4793 4794 4795 4796 4797 4798 4799 4800 4801 4802 4803 4804 4805 4806 4807 4808 4809 4810 4811 4812 4813 4814 4815 4816 4817 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
    int r = grab_range_lock(&request);
    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) {
        r = get_point_lock(db, txn, key);
        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;
}

4853 4854 4855 4856 4857 4858 4859 4860 4861 4862 4863 4864 4865 4866 4867 4868 4869 4870 4871
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
4872
do_put_multiple(DB_TXN *txn, uint32_t num_dbs, DB *db_array[], DBT keys[], DBT vals[], DB *src_db, const DBT *src_key) {
4873 4874 4875 4876
    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];
4877 4878 4879

        // 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.
4880 4881
	int do_put = TRUE;
	DB_INDEXER *indexer = toku_db_get_indexer(db);
4882
	if (indexer) { // if this db is the index under construction
4883 4884 4885 4886 4887 4888 4889 4890 4891 4892
            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];
            }
4893
            do_put = !toku_indexer_is_key_right_of_le_cursor(indexer, indexer_src_db, indexer_src_key);
4894 4895 4896 4897
        }
        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
4898
    }
4899
    return r;
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
4900
}
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
4901

4902
static int
4903 4904 4905 4906
env_put_multiple(
    DB_ENV *env, 
    DB *src_db, 
    DB_TXN *txn, 
4907 4908
    const DBT *src_key, 
    const DBT *src_val, 
4909 4910 4911 4912
    uint32_t num_dbs, 
    DB **db_array, 
    DBT *keys, 
    DBT *vals, 
4913
    uint32_t *flags_array) 
4914
{
4915
    int r;
4916 4917
    DBT put_keys[num_dbs];
    DBT put_vals[num_dbs];
4918 4919 4920 4921

    HANDLE_PANICKED_ENV(env);

    {
4922 4923 4924
    uint32_t lock_flags[num_dbs];
    uint32_t remaining_flags[num_dbs];
    BRT brts[num_dbs];
4925

4926 4927 4928 4929
    if (!txn || !num_dbs) {
        r = EINVAL;
        goto cleanup;
    }
4930
    if (!env->i->generate_row_for_put) {
4931 4932 4933 4934
        r = EINVAL;
        goto cleanup;
    }

4935 4936 4937
    HANDLE_ILLEGAL_WORKING_PARENT_TXN(env, txn);

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

4940 4941 4942 4943 4944 4945 4946 4947 4948
        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;
        }

4949
        //Generate the row
4950
        if (db == src_db) {
4951 4952
            put_keys[which_db] = *src_key;
            put_vals[which_db] = *src_val;
4953 4954
        }
        else {
4955
            r = env->i->generate_row_for_put(db, src_db, &keys[which_db], &vals[which_db], src_key, src_val);
4956 4957 4958 4959
            if (r != 0) goto cleanup;
            put_keys[which_db] = keys[which_db];
            put_vals[which_db] = vals[which_db];            
        }
4960 4961

        // check size constraints
4962
        r = db_put_check_size_constraints(db, &put_keys[which_db], &put_vals[which_db]);
4963 4964
        if (r != 0) goto cleanup;

4965 4966
        //Check overwrite constraints
        r = db_put_check_overwrite_constraint(db, txn,
4967
                                              &put_keys[which_db],
4968
                                              lock_flags[which_db], remaining_flags[which_db]);
4969
        if (r != 0) goto cleanup;
4970 4971 4972 4973 4974
        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;
        }
4975

4976 4977 4978
        //Do locking if necessary.
        if (db->i->lt && !(lock_flags[which_db] & DB_PRELOCKED_WRITE)) {
            //Needs locking
4979
            r = get_point_lock(db, txn, &put_keys[which_db]);
4980
            if (r != 0) goto cleanup;
4981 4982 4983
        }
        brts[which_db] = db->i->brt;
    }
4984 4985

    if (num_dbs == 1)
4986
        r = log_put_single(txn, brts[0], &put_keys[0], &put_vals[0]);
4987
    else
4988
        r = log_put_multiple(txn, src_db, src_key, src_val, num_dbs, brts);
4989 4990
    
    if (r == 0)
4991
        r = do_put_multiple(txn, num_dbs, db_array, put_keys, put_vals, src_db, src_key);
4992

4993 4994
    }

4995
cleanup:
4996
    if (r == 0)
4997
        num_multi_inserts += num_dbs;
4998
    else
4999
        num_multi_inserts_fail += num_dbs;
5000 5001 5002 5003 5004
    return r;
}

static int
env_update_multiple(DB_ENV *env, DB *src_db, DB_TXN *txn,                                
5005 5006 5007
                    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, 
5008
                    uint32_t num_keys, DBT keys[], 
5009
                    uint32_t num_vals, DBT vals[]) {
5010 5011 5012 5013 5014 5015 5016 5017 5018 5019 5020 5021 5022 5023 5024 5025 5026 5027 5028 5029 5030 5031 5032 5033 5034 5035 5036
    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];

5037 5038 5039
        uint32_t lock_flags[num_dbs];
        uint32_t remaining_flags[num_dbs];

5040 5041
        for (uint32_t which_db = 0; which_db < num_dbs; which_db++) {
            DB *db = db_array[which_db];
5042 5043 5044 5045
            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];
5046

5047 5048 5049 5050
            if (!(lock_flags[which_db] & DB_PRELOCKED_FILE_READ)) {
                r = toku_grab_read_lock_on_directory(db, txn);
                if (r != 0) goto cleanup;
            }
5051 5052 5053
            // 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
5054

5055 5056
            // Generate the old key and val
            if (which_db + num_dbs >= num_keys) {
5057 5058
                r = ENOMEM; goto cleanup;
            }
5059 5060 5061 5062
            if (db == src_db) {
                curr_old_key = *old_src_key;
            }
            else {
5063
                r = env->i->generate_row_for_put(db, src_db, &keys[which_db + num_dbs], NULL, old_src_key, old_src_data);
5064 5065 5066
                if (r != 0) goto cleanup;
                curr_old_key = keys[which_db + num_dbs];
            }
5067
            // Generate the new key and val
5068 5069 5070
            if (which_db >= num_keys || which_db >= num_vals) {
                r = ENOMEM; goto cleanup;
            }
5071 5072 5073 5074 5075
            if (db == src_db) {
                curr_new_key = *new_src_key;
                curr_new_val = *new_src_data;
            }
            else {
5076
                r = env->i->generate_row_for_put(db, src_db, &keys[which_db], &vals[which_db], new_src_key, new_src_data);
5077 5078 5079 5080
                if (r != 0) goto cleanup;
                curr_new_key = keys[which_db];
                curr_new_val = vals[which_db];
            }
5081
            toku_dbt_cmp cmpfun = toku_db_get_compare_fun(db);
5082
            BOOL key_eq = cmpfun(db, &curr_old_key, &curr_new_key) == 0;
5083
            if (!key_eq) {
5084 5085
                //Check overwrite constraints only in the case where 
                // the keys are not equal.
5086
                // If the keys are equal, then we do not care of the flag is DB_NOOVERWRITE or 0
5087 5088 5089 5090 5091 5092 5093 5094 5095 5096
                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;
                }

5097
                // lock old key
5098 5099
                if (db->i->lt && !(lock_flags[which_db] & DB_PRELOCKED_WRITE)) {
                    r = get_point_lock(db, txn, &curr_old_key);
5100 5101 5102 5103
                    if (r != 0) goto cleanup;
                }
                del_dbs[n_del_dbs] = db;
                del_brts[n_del_dbs] = db->i->brt;
5104
                del_keys[n_del_dbs] = curr_old_key;
5105
                n_del_dbs++;
5106
                
5107 5108
            }

5109 5110 5111
            // 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)) {
5112 5113
            if (!key_eq || curr_new_val.size > 0) {
                r = db_put_check_size_constraints(db, &curr_new_key, &curr_new_val);
5114 5115 5116 5117
                if (r != 0) goto cleanup;

                // lock new key
                if (db->i->lt) {
5118
                    r = get_point_lock(db, txn, &curr_new_key);
5119 5120 5121 5122
                    if (r != 0) goto cleanup;
                }
                put_dbs[n_put_dbs] = db;
                put_brts[n_put_dbs] = db->i->brt;
5123 5124
                put_keys[n_put_dbs] = curr_new_key;
                put_vals[n_put_dbs] = curr_new_val;
5125 5126 5127 5128 5129 5130 5131 5132
                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
5133
                r = log_del_multiple(txn, src_db, old_src_key, old_src_data, n_del_dbs, del_brts, del_keys);
5134 5135
            if (r == 0)
                r = do_del_multiple(txn, n_del_dbs, del_dbs, del_keys, src_db, old_src_key);
5136 5137 5138 5139 5140 5141 5142 5143
        }

        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)
5144
                r = do_put_multiple(txn, n_put_dbs, put_dbs, put_keys, put_vals, src_db, new_src_key);
5145 5146 5147 5148
        }
    }

cleanup:
5149 5150 5151 5152
    if (r == 0)
        num_multi_updates += num_dbs;
    else
        num_multi_updates_fail += num_dbs;
5153 5154 5155 5156
    return r;
}

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

//We do not (yet?) support deleting subdbs by deleting the enclosing 'fname'
5159
static int
5160 5161 5162 5163
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 {
5164 5165 5166 5167
        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;
5168
        r = toku_env_dbremove(env, txn, subdb_full_name, null_subdbname, flags);
5169
    }
5170 5171
    return r;
}
5172

5173 5174 5175 5176

//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
5177
finalize_file_removal(DICTIONARY_ID dict_id, void * extra) {
5178 5179 5180
    toku_ltm *ltm = (toku_ltm*) extra;
    if (ltm) {
        //Poison the lock tree to prevent a future file from re-using it.
5181
        toku_ltm_invalidate_lt(ltm, dict_id);
5182 5183 5184
    }
}

5185
//static int toku_db_pre_acquire_table_lock(DB *db, DB_TXN *txn);
5186

5187 5188 5189 5190 5191 5192 5193 5194 5195 5196 5197 5198 5199 5200 5201 5202 5203 5204 5205 5206 5207 5208 5209 5210 5211 5212
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) {
5213
	r = toku_txn_begin(env, txn, &child, DB_TXN_NOSYNC, 1);
5214 5215 5216 5217
	assert(r==0);
    }

    // get iname
5218
    r = toku_db_get(env->i->directory, child, &dname_dbt, &iname_dbt, DB_SERIALIZABLE);  // allocates memory for iname
5219 5220 5221 5222 5223 5224 5225 5226
    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
5227
                r = toku_brt_remove_on_commit(db_txn_struct_i(child)->tokutxn, &iname_dbt);
5228
		assert(r==0);
5229 5230 5231
                //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");
5232 5233 5234
                if (r==0) {
                    DB* zombie = env_get_zombie_db_with_dname(env, dname);
                    if (zombie)
5235
                        r = toku_db_pre_acquire_table_lock(zombie, child, TRUE);
5236
                    if (r!=0 && r!=DB_LOCK_NOTGRANTED)
5237 5238
                        toku_ydb_do_error(env, r, "Cannot remove dictionary.\n");
                }
5239 5240
            }
            else {
Yoni Fogel's avatar
Yoni Fogel committed
5241
                r = toku_brt_remove_now(env->i->cachetable, &iname_dbt);
5242 5243 5244 5245 5246 5247 5248 5249
		assert(r==0);
            }
	}
    }

    if (using_txns) {
	// close txn
	if (r == 0) {  // commit
5250
	    r = toku_txn_commit(child, DB_TXN_NOSYNC, NULL, NULL);
5251
	    invariant(r==0);  // TODO panic
5252 5253
	}
	else {         // abort
5254
	    int r2 = toku_txn_abort(child, NULL, NULL);
5255
	    invariant(r2==0);  // TODO panic
5256
	}
5257
    }
5258 5259

    if (iname) toku_free(iname);
5260
    return r;
5261

5262 5263
}

5264 5265 5266

static int
toku_db_remove(DB * db, const char *fname, const char *dbname, u_int32_t flags) {
5267
    HANDLE_PANICKED_DB(db);
5268 5269 5270 5271 5272 5273 5274 5275 5276 5277 5278 5279 5280 5281 5282 5283 5284 5285 5286 5287 5288
    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
5289
        }
5290 5291
        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
5292
    }
5293 5294 5295 5296 5297 5298 5299 5300 5301 5302 5303 5304 5305 5306 5307 5308 5309 5310 5311 5312 5313 5314
    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
5315
    
5316 5317 5318 5319 5320 5321 5322 5323 5324 5325 5326
    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) {
5327
	r = toku_txn_begin(env, txn, &child, DB_TXN_NOSYNC, 1);
5328 5329
	assert(r==0);
    }
5330

5331
    r = toku_db_get(env->i->directory, child, &old_dname_dbt, &iname_dbt, DB_SERIALIZABLE);  // allocates memory for iname
5332 5333 5334 5335 5336
    char *iname = iname_dbt.data;
    if (r==DB_NOTFOUND)
        r = ENOENT;
    else if (r==0) {
	// verify that newname does not already exist
5337
	r = db_getf_set(env->i->directory, child, DB_SERIALIZABLE, &new_dname_dbt, ydb_getf_do_nothing, NULL);
5338 5339 5340 5341 5342 5343
	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)
5344
		r = toku_db_put(env->i->directory, child, &new_dname_dbt, &iname_dbt, 0);
5345 5346 5347
            //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");
5348
            DB* zombie = NULL;
5349
            if (r==0) {
5350
                zombie = env_get_zombie_db_with_dname(env, dname);
5351
                if (zombie)
5352
                    r = toku_db_pre_acquire_table_lock(zombie, child, TRUE);
5353
                if (r!=0 && r!=DB_LOCK_NOTGRANTED)
5354 5355
                    toku_ydb_do_error(env, r, "Cannot rename dictionary.\n");
            }
5356 5357
            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");
5358 5359 5360 5361 5362 5363
            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
5364
            }
5365 5366 5367 5368 5369 5370
	}
    }

    if (using_txns) {
	// close txn
	if (r == 0) {  // commit
5371
	    r = toku_txn_commit(child, DB_TXN_NOSYNC, NULL, NULL);
5372
	    invariant(r==0);  // TODO panic
5373 5374
	}
	else {         // abort
5375
	    int r2 = toku_txn_abort(child, NULL, NULL);
5376
	    invariant(r2==0);  // TODO panic
5377 5378 5379 5380 5381
	}
    }

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

Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
5383
}
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
5384

5385 5386
static int
toku_db_rename(DB * db, const char *fname, const char *dbname, const char *newname, u_int32_t flags) {
5387
    HANDLE_PANICKED_DB(db);
5388 5389 5390 5391 5392
    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
5393
}
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
5394

5395 5396 5397
//
// This function is the only way to set a descriptor of a DB.
//
5398
static int 
5399
toku_db_change_descriptor(DB *db, DB_TXN* txn, const DBT* descriptor, u_int32_t flags) {
5400
    HANDLE_PANICKED_DB(db);
5401
    HANDLE_DB_ILLEGAL_WORKING_PARENT_TXN(db, txn);
5402
    int r;
5403 5404 5405 5406 5407 5408 5409 5410 5411 5412 5413 5414 5415 5416 5417 5418 5419 5420 5421 5422 5423 5424 5425 5426 5427 5428 5429 5430 5431 5432
    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);
5433
    r = toku_brt_change_descriptor(db->i->brt, &old_descriptor, descriptor, TRUE, ttxn);
5434 5435
cleanup:
    if (old_descriptor.data) toku_free(old_descriptor.data);
5436 5437 5438
    return r;
}

5439 5440
static int 
toku_db_set_flags(DB *db, u_int32_t flags) {
5441
    HANDLE_PANICKED_DB(db);
5442

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

5446
    return 0;
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
5447 5448
}

5449 5450
static int 
toku_db_get_flags(DB *db, u_int32_t *pflags) {
5451
    HANDLE_PANICKED_DB(db);
5452
    if (!pflags) return EINVAL;
5453
    *pflags = 0;
5454 5455 5456
    return 0;
}

5457 5458
static int 
toku_db_set_pagesize(DB *db, u_int32_t pagesize) {
5459
    HANDLE_PANICKED_DB(db);
5460
    int r = toku_brt_set_nodesize(db->i->brt, pagesize);
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
5461
    return r;
Bradley C. Kuszmaul's avatar
Rename  
Bradley C. Kuszmaul committed
5462
}
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
5463

5464 5465 5466 5467 5468 5469 5470
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;
}

5471 5472 5473 5474 5475 5476 5477 5478 5479 5480 5481 5482 5483 5484
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;
}

5485 5486
static int 
toku_db_stat64(DB * db, DB_TXN *txn, DB_BTREE_STAT64 *s) {
5487
    HANDLE_PANICKED_DB(db);
5488
    HANDLE_DB_ILLEGAL_WORKING_PARENT_TXN(db, txn);
5489
    struct brtstat64_s brtstat;
5490 5491 5492 5493 5494
    TOKUTXN tokutxn = NULL;
    if (txn != NULL) {
        tokutxn = db_txn_struct_i(txn)->tokutxn;
    }
    int r = toku_brt_stat64(db->i->brt, tokutxn, &brtstat);
5495 5496 5497 5498 5499 5500 5501
    if (r==0) {
	s->bt_nkeys = brtstat.nkeys;
	s->bt_ndata = brtstat.ndata;
	s->bt_dsize = brtstat.dsize;
	s->bt_fsize = brtstat.fsize;
    }
    return r;
5502
}
5503 5504 5505

static int 
locked_db_stat64 (DB *db, DB_TXN *txn, DB_BTREE_STAT64 *s) {
5506 5507 5508 5509
    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
5510 5511
}

5512 5513
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
5514
    HANDLE_PANICKED_DB(db);
5515
    HANDLE_DB_ILLEGAL_WORKING_PARENT_TXN(db, txn);
Zardosht Kasheff's avatar
Zardosht Kasheff committed
5516 5517 5518 5519 5520 5521 5522 5523 5524 5525 5526 5527 5528

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

5529
static int
5530 5531 5532
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
5533
    HANDLE_PANICKED_DB(db);
5534 5535 5536 5537 5538
    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;
5539
    //READ_UNCOMMITTED and READ_COMMITTED transactions do not need read locks.
5540 5541
    if (!dbc_struct_i(dbc)->rmw && dbc_struct_i(dbc)->iso != TOKU_ISO_SERIALIZABLE)
        return 0;
Yoni Fogel's avatar
Yoni Fogel committed
5542

5543 5544 5545
    RANGE_LOCK_REQUEST_S request;
    range_lock_request_init(&request, dbc_struct_i(dbc)->rmw, txn, db, key_left, key_right);
    int r = grab_range_lock(&request);
Yoni Fogel's avatar
Yoni Fogel committed
5546 5547 5548
    return r;
}

5549 5550
//static int toku_db_pre_acquire_table_lock(DB *db, DB_TXN *txn) {
// needed by loader.c
5551 5552
int 
toku_db_pre_acquire_table_lock(DB *db, DB_TXN *txn, BOOL just_lock) {
Yoni Fogel's avatar
Yoni Fogel committed
5553
    HANDLE_PANICKED_DB(db);
5554
    if (!db->i->lt || !txn) return 0;
Yoni Fogel's avatar
Yoni Fogel committed
5555 5556 5557

    int r;

5558
    {
Zardosht Kasheff's avatar
Zardosht Kasheff committed
5559
        RANGE_LOCK_REQUEST_S request;
5560 5561
        write_lock_request_init(&request, txn, db,
                                toku_lt_neg_infinity, toku_lt_infinity);
5562 5563 5564
        r = grab_range_lock(&request);
    }

5565 5566
    if (r==0 && !just_lock &&
        !toku_brt_is_recovery_logging_suppressed(db->i->brt) &&
5567
        toku_brt_is_empty_fast(db->i->brt)
5568 5569 5570 5571 5572 5573
    ) {
        //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};
5574
        uint32_t loader_flags = DB_PRELOCKED_WRITE; //Don't recursively prelock
5575
        DB_ENV *env = db->dbenv;
5576 5577 5578 5579 5580 5581 5582
	DB_TXN *child = NULL;
	
	{
	    // begin child
	    int rt = toku_txn_begin(env, txn, &child, DB_TXN_NOSYNC, 1);
	    assert(rt==0);
	}
5583 5584

        toku_ydb_unlock(); //Cannot hold ydb lock when creating loader
5585 5586
	
        int r_loader = env->create_loader(env, child, &loader, NULL, 1, dbs, db_flags, dbt_flags, loader_flags);
5587
        if (r_loader==0) {
5588 5589 5590 5591
            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);
5592
            // close the loader
5593 5594 5595 5596
            r_loader = loader->close(loader);
	    if (r_loader==0) {
		toku_brt_suppress_recovery_logs(db->i->brt, db_txn_struct_i(child)->tokutxn);
	    }
5597 5598 5599 5600 5601 5602 5603
        }
        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;
        }
5604 5605 5606
	if (r_loader == 0) { // commit
	    r = locked_txn_commit(child, 0);
	    assert(r==0);
5607
	    logsuppress++;
5608 5609 5610 5611
	}
	else {  // abort
	    r = locked_txn_abort(child);
	    assert(r==0);
5612
	    logsuppressfail++;
5613
	}
5614
        toku_ydb_lock(); //Reaquire ydb lock.
5615 5616
    }

Yoni Fogel's avatar
Yoni Fogel committed
5617 5618
    return r;
}
Zardosht Kasheff's avatar
Zardosht Kasheff committed
5619

Yoni Fogel's avatar
Yoni Fogel committed
5620 5621 5622
//TODO: DB_AUTO_COMMIT.
//TODO: Nowait only conditionally?
//TODO: NOSYNC change to SYNC if DB_ENV has something in set_flags
5623 5624
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
5625 5626 5627 5628 5629 5630
    assert(db && txn && changed);
    DB_ENV* env = db->dbenv;
    if (*txn || !(env->i->open_flags & DB_INIT_TXN)) {
        *changed = FALSE;
        return 0;
    }
5631
    BOOL nosync = (BOOL)(!force_auto_commit && !(env->i->open_flags & DB_AUTO_COMMIT));
Yoni Fogel's avatar
Yoni Fogel committed
5632
    u_int32_t txn_flags = DB_TXN_NOWAIT | (nosync ? DB_TXN_NOSYNC : 0);
5633
    int r = toku_txn_begin(env, NULL, txn, txn_flags, 1);
Yoni Fogel's avatar
Yoni Fogel committed
5634 5635 5636 5637 5638
    if (r!=0) return r;
    *changed = TRUE;
    return 0;
}

5639 5640
static inline int 
toku_db_destruct_autotxn(DB_TXN *txn, int r, BOOL changed) {
Yoni Fogel's avatar
Yoni Fogel committed
5641
    if (!changed) return r;
5642 5643
    if (r==0) return toku_txn_commit(txn, 0, NULL, NULL);
    toku_txn_abort(txn, NULL, NULL);
Yoni Fogel's avatar
Yoni Fogel committed
5644 5645 5646
    return r; 
}

5647 5648
static int 
locked_db_close(DB * db, u_int32_t flags) {
5649 5650 5651 5652
    toku_ydb_lock(); 
    int r = toku_db_close(db, flags); 
    toku_ydb_unlock(); 
    return r;
Rich Prohaska's avatar
Rich Prohaska committed
5653 5654
}

5655 5656
static inline int 
autotxn_db_cursor(DB *db, DB_TXN *txn, DBC **c, u_int32_t flags) {
Yoni Fogel's avatar
Yoni Fogel committed
5657
    if (!txn && (db->dbenv->i->open_flags & DB_INIT_TXN)) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
5658
        return toku_ydb_do_error(db->dbenv, EINVAL,
Yoni Fogel's avatar
Yoni Fogel committed
5659 5660
              "Cursors in a transaction environment must have transactions.\n");
    }
5661
    return toku_db_cursor(db, txn, c, flags, 0);
Yoni Fogel's avatar
Yoni Fogel committed
5662 5663
}

5664 5665
static int 
locked_db_cursor(DB *db, DB_TXN *txn, DBC **c, u_int32_t flags) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
5666
    toku_ydb_lock(); int r = autotxn_db_cursor(db, txn, c, flags); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
5667 5668
}

5669 5670
static inline int 
autotxn_db_del(DB* db, DB_TXN* txn, DBT* key, u_int32_t flags) {
Yoni Fogel's avatar
Yoni Fogel committed
5671 5672 5673 5674 5675 5676 5677
    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);
}

5678 5679
static int 
locked_db_del(DB * db, DB_TXN * txn, DBT * key, u_int32_t flags) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
5680
    toku_ydb_lock(); int r = autotxn_db_del(db, txn, key, flags); toku_ydb_unlock(); return r;
Yoni Fogel's avatar
Yoni Fogel committed
5681 5682
}

5683 5684
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
5685 5686 5687 5688 5689
    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
5690 5691
}

5692 5693
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
5694
    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
5695 5696
}

5697 5698
static inline int 
autotxn_db_getf_set (DB *db, DB_TXN *txn, u_int32_t flags, DBT *key, YDB_CALLBACK_FUNCTION f, void *extra) {
5699 5700 5701 5702 5703 5704 5705
    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);
}

5706 5707
static int 
locked_db_getf_set (DB *db, DB_TXN *txn, u_int32_t flags, DBT *key, YDB_CALLBACK_FUNCTION f, void *extra) {
5708 5709 5710
    toku_ydb_lock(); int r = autotxn_db_getf_set(db, txn, flags, key, f, extra); toku_ydb_unlock(); return r;
}

5711
static int 
5712
locked_c_pre_acquire_range_lock(DBC *dbc, const DBT *key_left, const DBT *key_right) {
Yoni Fogel's avatar
Yoni Fogel committed
5713
    toku_ydb_lock();
5714
    int r = toku_c_pre_acquire_range_lock(dbc, key_left, key_right);
Yoni Fogel's avatar
Yoni Fogel committed
5715 5716 5717 5718
    toku_ydb_unlock();
    return r;
}

5719 5720
static int 
locked_db_pre_acquire_table_lock(DB *db, DB_TXN *txn) {
Yoni Fogel's avatar
Yoni Fogel committed
5721
    toku_ydb_lock();
5722
    int r = toku_db_pre_acquire_table_lock(db, txn, FALSE);
Yoni Fogel's avatar
Yoni Fogel committed
5723 5724 5725 5726
    toku_ydb_unlock();
    return r;
}

Zardosht Kasheff's avatar
Zardosht Kasheff committed
5727 5728 5729 5730 5731 5732 5733
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;
}

5734 5735 5736 5737 5738 5739 5740
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;
}

5741 5742
// truncate a database
// effect: remove all of the rows from a database
5743 5744
static int 
toku_db_truncate(DB *db, DB_TXN *txn, u_int32_t *row_count, u_int32_t flags) {
5745
    HANDLE_PANICKED_DB(db);
5746
    HANDLE_DB_ILLEGAL_WORKING_PARENT_TXN(db, txn);
5747 5748
    int r;

5749 5750 5751 5752 5753 5754 5755
    u_int32_t unhandled_flags = flags;
    int ignore_cursors = 0;
    if (flags & DB_TRUNCATE_WITHCURSORS) {
        ignore_cursors = 1;
        unhandled_flags &= ~DB_TRUNCATE_WITHCURSORS;
    }

5756
    // dont support flags (yet)
5757
    if (unhandled_flags)
5758
        return EINVAL;
5759 5760
    // dont support cursors unless explicitly told to
    if (!ignore_cursors && toku_brt_get_cursor_count(db->i->brt) > 0)
5761 5762 5763 5764
        return EINVAL;

    // acquire a table lock
    if (txn) {
Zardosht Kasheff's avatar
Zardosht Kasheff committed
5765 5766 5767 5768
        r = toku_db_pre_acquire_fileops_lock(db, txn);
        if (r != 0) {
            return r;
        }
5769
        r = toku_db_pre_acquire_table_lock(db, txn, TRUE);
Zardosht Kasheff's avatar
Zardosht Kasheff committed
5770
        if (r != 0) {
5771
            return r;
Zardosht Kasheff's avatar
Zardosht Kasheff committed
5772
        }
5773 5774 5775 5776
    }

    *row_count = 0;

5777
    r = toku_brt_truncate(db->i->brt);
5778 5779 5780 5781

    return r;
}

5782 5783
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
5784
    BOOL changed; int r;
5785
    r = toku_db_construct_autotxn(db, &txn, &changed, (BOOL)((flags & DB_AUTO_COMMIT) != 0));
Yoni Fogel's avatar
Yoni Fogel committed
5786 5787 5788
    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
5789 5790
}

5791 5792
static int 
locked_db_open(DB *db, DB_TXN *txn, const char *fname, const char *dbname, DBTYPE dbtype, u_int32_t flags, int mode) {
5793 5794 5795 5796
    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
5797 5798
}

5799 5800
static inline int 
autotxn_db_put(DB* db, DB_TXN* txn, DBT* key, DBT* data, u_int32_t flags) {
5801
    //{ 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
5802 5803 5804 5805 5806 5807 5808
    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);
}

5809 5810
static int 
locked_db_put(DB * db, DB_TXN * txn, DBT * key, DBT * data, u_int32_t flags) {
5811 5812 5813 5814 5815 5816 5817
    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
5818 5819
}

5820 5821 5822 5823 5824 5825 5826 5827 5828 5829 5830 5831 5832 5833 5834 5835 5836 5837 5838 5839 5840 5841 5842 5843 5844 5845 5846 5847 5848 5849 5850 5851 5852 5853 5854 5855 5856 5857 5858 5859 5860 5861 5862 5863 5864 5865 5866 5867 5868 5869
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;
}

5870 5871
static int 
locked_db_remove(DB * db, const char *fname, const char *dbname, u_int32_t flags) {
5872
    toku_multi_operation_client_lock(); //Cannot begin checkpoint
5873 5874 5875
    toku_ydb_lock();
    int r = toku_db_remove(db, fname, dbname, flags);
    toku_ydb_unlock();
5876
    toku_multi_operation_client_unlock(); //Can now begin checkpoint
5877
    return r;
Rich Prohaska's avatar
Rich Prohaska committed
5878 5879
}

5880 5881
static int 
locked_db_rename(DB * db, const char *namea, const char *nameb, const char *namec, u_int32_t flags) {
5882
    toku_multi_operation_client_lock(); //Cannot begin checkpoint
5883 5884 5885
    toku_ydb_lock();
    int r = toku_db_rename(db, namea, nameb, namec, flags);
    toku_ydb_unlock();
5886
    toku_multi_operation_client_unlock(); //Can now begin checkpoint
5887
    return r;
Rich Prohaska's avatar
Rich Prohaska committed
5888 5889
}

5890
static int 
5891
locked_db_change_descriptor(DB *db, DB_TXN* txn, const DBT* descriptor, u_int32_t flags) {
5892
    toku_ydb_lock();
5893
    int r = toku_db_change_descriptor(db, txn, descriptor, flags);
5894 5895
    toku_ydb_unlock();
    return r;
5896 5897
}

5898 5899
static void 
locked_db_set_errfile (DB *db, FILE *errfile) {
Rich Prohaska's avatar
Rich Prohaska committed
5900 5901 5902
    db->dbenv->set_errfile(db->dbenv, errfile);
}

5903 5904
static int 
locked_db_set_flags(DB *db, u_int32_t flags) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
5905
    toku_ydb_lock(); int r = toku_db_set_flags(db, flags); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
5906 5907
}

5908 5909
static int 
locked_db_get_flags(DB *db, u_int32_t *flags) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
5910
    toku_ydb_lock(); int r = toku_db_get_flags(db, flags); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
5911 5912
}

5913 5914
static int 
locked_db_set_pagesize(DB *db, u_int32_t pagesize) {
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
5915
    toku_ydb_lock(); int r = toku_db_set_pagesize(db, pagesize); toku_ydb_unlock(); return r;
Rich Prohaska's avatar
Rich Prohaska committed
5916 5917
}

5918 5919 5920 5921 5922
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;
}

5923 5924 5925 5926 5927 5928 5929 5930 5931 5932
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;
}

5933
// TODO 2216 delete this
5934 5935
static int 
locked_db_fd(DB * UU(db), int * UU(fdp)) {
5936 5937 5938 5939 5940
    //    toku_ydb_lock(); 
    // int r = toku_db_fd(db, fdp); 
    //    toku_ydb_unlock(); 
    //    return r;
    return 0;
Rich Prohaska's avatar
Rich Prohaska committed
5941 5942
}

Zardosht Kasheff's avatar
Zardosht Kasheff committed
5943

Yoni Fogel's avatar
Yoni Fogel committed
5944 5945 5946 5947 5948
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));
5949 5950
static const DBT*
toku_db_dbt_pos_infty(void) {
Yoni Fogel's avatar
Yoni Fogel committed
5951 5952 5953 5954
    return toku_lt_infinity;
}

static const DBT* toku_db_dbt_neg_infty(void) __attribute__((pure));
5955 5956
static const DBT* 
toku_db_dbt_neg_infty(void) {
Yoni Fogel's avatar
Yoni Fogel committed
5957
    return toku_lt_neg_infinity;
Zardosht Kasheff's avatar
Zardosht Kasheff committed
5958 5959
}

5960 5961
static int 
locked_db_truncate(DB *db, DB_TXN *txn, u_int32_t *row_count, u_int32_t flags) {
5962 5963 5964 5965 5966 5967
    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;
5968 5969
}

5970 5971 5972 5973 5974 5975 5976
static int
toku_db_optimize(DB *db) {
    HANDLE_PANICKED_DB(db);
    int r = toku_brt_optimize(db->i->brt);
    return r;
}

5977 5978 5979
static int
toku_db_flatten(DB *db, DB_TXN *txn) {
    HANDLE_PANICKED_DB(db);
5980 5981
    TOKUTXN ttxn = txn ? db_txn_struct_i(txn)->tokutxn : NULL;
    int r = toku_brt_flatten(db->i->brt, ttxn);
5982 5983 5984
    return r;
}

5985 5986
static inline int 
autotxn_db_flatten(DB* db, DB_TXN* txn) {
5987 5988 5989 5990 5991 5992 5993 5994
    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);
}


5995 5996
static int 
locked_db_flatten(DB *db, DB_TXN *txn) {
5997 5998 5999
    toku_ydb_lock(); int r = autotxn_db_flatten(db, txn); toku_ydb_unlock(); return r;
}

6000 6001
static int 
locked_db_optimize(DB *db) {
6002 6003 6004 6005 6006 6007
    toku_ydb_lock();
    int r = toku_db_optimize(db);
    toku_ydb_unlock();
    return r;
}

6008 6009 6010 6011 6012 6013 6014 6015 6016 6017 6018 6019 6020 6021 6022 6023 6024 6025 6026
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;
}

6027 6028
int 
toku_db_set_indexer(DB *db, DB_INDEXER * indexer) {
Dave Wells's avatar
Dave Wells committed
6029 6030 6031 6032 6033 6034 6035 6036 6037
    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;
6038 6039 6040 6041 6042 6043 6044 6045 6046 6047 6048 6049 6050 6051 6052 6053 6054
}

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

6055 6056 6057 6058 6059 6060 6061 6062 6063 6064 6065 6066 6067 6068 6069 6070 6071 6072 6073 6074 6075 6076 6077 6078 6079
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;
}

6080 6081
static int 
toku_db_create(DB ** db, DB_ENV * env, u_int32_t flags) {
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
6082 6083
    int r;

6084 6085 6086 6087 6088
    if (flags || env == NULL) 
        return EINVAL;

    if (!env_opened(env))
        return EINVAL;
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
6089
    
6090
    DB *MALLOC(result);
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
6091 6092
    if (result == 0) {
        return ENOMEM;
Bradley C. Kuszmaul's avatar
Bradley C. Kuszmaul committed
6093
    }
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
6094 6095
    memset(result, 0, sizeof *result);
    result->dbenv = env;
Yoni Fogel's avatar
Yoni Fogel committed
6096 6097 6098 6099 6100 6101 6102 6103 6104
#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);
6105 6106
    SDB(update);
    SDB(update_broadcast);
Yoni Fogel's avatar
Yoni Fogel committed
6107 6108
    SDB(remove);
    SDB(rename);
6109
    SDB(change_descriptor);
Yoni Fogel's avatar
Yoni Fogel committed
6110 6111
    SDB(set_errfile);
    SDB(set_pagesize);
6112
    SDB(get_pagesize);
6113 6114
    SDB(set_readpagesize);
    SDB(get_readpagesize);
Yoni Fogel's avatar
Yoni Fogel committed
6115 6116
    SDB(set_flags);
    SDB(get_flags);
6117
    SDB(stat64);
Yoni Fogel's avatar
Yoni Fogel committed
6118
    SDB(fd);
Yoni Fogel's avatar
Yoni Fogel committed
6119
    SDB(pre_acquire_table_lock);
Zardosht Kasheff's avatar
Zardosht Kasheff committed
6120
    SDB(pre_acquire_fileops_lock);
6121
    SDB(pre_acquire_fileops_shared_lock);
6122
    SDB(truncate);
6123
    SDB(row_size_supported);
6124
    SDB(getf_set);
6125
    SDB(flatten);
6126
    SDB(optimize);
6127
    SDB(get_fragmentation);
6128 6129
    SDB(set_indexer);
    SDB(get_indexer);
6130
    SDB(verify_with_progress);
Yoni Fogel's avatar
Yoni Fogel committed
6131 6132 6133
#undef SDB
    result->dbt_pos_infty = toku_db_dbt_pos_infty;
    result->dbt_neg_infty = toku_db_dbt_neg_infty;
6134
    MALLOC(result->i);
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
6135 6136 6137 6138 6139
    if (result->i == 0) {
        toku_free(result);
        return ENOMEM;
    }
    memset(result->i, 0, sizeof *result->i);
6140
    result->i->dict_id = DICTIONARY_ID_NONE;
6141
    result->i->opened = 0;
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
6142 6143 6144
    result->i->open_flags = 0;
    result->i->open_mode = 0;
    result->i->brt = 0;
6145 6146
    result->i->indexer = NULL;
    result->i->refs = 1;
6147
    toku_list_init(&result->i->dbs_that_must_close_before_abort);
6148
    r = toku_brt_create(&result->i->brt);
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
6149 6150 6151
    if (r != 0) {
        toku_free(result->i);
        toku_free(result);
Yoni Fogel's avatar
Yoni Fogel committed
6152
        return r;
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
6153 6154 6155
    }
    *db = result;
    return 0;
Bradley C. Kuszmaul's avatar
Bradley C. Kuszmaul committed
6156
}
Bradley C. Kuszmaul's avatar
Bradley C. Kuszmaul committed
6157

6158 6159
int 
DB_CREATE_FUN (DB ** db, DB_ENV * env, u_int32_t flags) {
6160 6161 6162 6163
    toku_ydb_lock(); 
    int r = toku_db_create(db, env, flags); 
    toku_ydb_unlock(); 
    return r;
Rich Prohaska's avatar
Rich Prohaska committed
6164 6165 6166 6167
}

/* need db_strerror_r for multiple threads */

6168 6169
char *
db_strerror(int error) {
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
6170 6171 6172 6173 6174 6175 6176
    char *errorstr;
    if (error >= 0) {
        errorstr = strerror(error);
        if (errorstr)
            return errorstr;
    }
    
6177 6178 6179 6180 6181 6182 6183 6184 6185 6186 6187 6188 6189 6190 6191
    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)";
6192
    }
6193

Rich Prohaska's avatar
Rich Prohaska committed
6194
    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
6195 6196 6197 6198 6199
    errorstr = unknown_result;
    snprintf(errorstr, sizeof unknown_result, "Unknown error code: %d", error);
    return errorstr;
}

6200 6201
const char *
db_version(int *major, int *minor, int *patch) {
Bradley C. Kuszmaul's avatar
up  
Bradley C. Kuszmaul committed
6202 6203 6204 6205 6206 6207
    if (major)
        *major = DB_VERSION_MAJOR;
    if (minor)
        *minor = DB_VERSION_MINOR;
    if (patch)
        *patch = DB_VERSION_PATCH;
6208 6209 6210 6211 6212
#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
6213
#error
6214
#endif
Bradley C. Kuszmaul's avatar
Bradley C. Kuszmaul committed
6215
}
6216
 
6217 6218
int 
db_env_set_func_fsync (int (*fsync_function)(int)) {
6219 6220
    return toku_set_func_fsync(fsync_function);
}
Yoni Fogel's avatar
Yoni Fogel committed
6221

6222 6223
int 
db_env_set_func_pwrite (ssize_t (*pwrite_function)(int, const void *, size_t, toku_off_t)) {
6224 6225
    return toku_set_func_pwrite(pwrite_function);
}
6226 6227 6228

int 
db_env_set_func_full_pwrite (ssize_t (*pwrite_function)(int, const void *, size_t, toku_off_t)) {
6229 6230
    return toku_set_func_full_pwrite(pwrite_function);
}
6231 6232 6233

int 
db_env_set_func_write (ssize_t (*write_function)(int, const void *, size_t)) {
6234 6235
    return toku_set_func_write(write_function);
}
6236 6237 6238

int 
db_env_set_func_full_write (ssize_t (*write_function)(int, const void *, size_t)) {
6239 6240
    return toku_set_func_full_write(write_function);
}
6241

6242 6243
int 
db_env_set_func_fdopen (FILE * (*fdopen_function)(int, const char *)) {
6244 6245
    return toku_set_func_fdopen(fdopen_function);
}
6246 6247 6248

int 
db_env_set_func_fopen (FILE * (*fopen_function)(const char *, const char *)) {
6249 6250
    return toku_set_func_fopen(fopen_function);
}
6251 6252 6253

int 
db_env_set_func_open (int (*open_function)(const char *, int, int)) {
6254 6255
    return toku_set_func_open(open_function);
}
6256 6257 6258

int 
db_env_set_func_fclose (int (*fclose_function)(FILE*)) {
6259 6260 6261
    return toku_set_func_fclose(fclose_function);
}

6262 6263 6264 6265 6266
int
db_env_set_func_pread (ssize_t (*fun)(int, void *, size_t, off_t)) {
    return toku_set_func_pread(fun);
}

6267 6268 6269 6270 6271
void 
db_env_set_func_loader_fwrite (size_t (*fwrite_fun)(const void*,size_t,size_t,FILE*)) {
    brtloader_set_os_fwrite(fwrite_fun);
}

6272 6273
int 
db_env_set_func_malloc (void *(*f)(size_t)) {
6274 6275
    toku_set_func_malloc(f);
    return 0;
6276
}
6277 6278 6279

int 
db_env_set_func_realloc (void *(*f)(void*, size_t)) {
6280 6281
    toku_set_func_realloc(f);
    return 0;
6282
}
6283 6284 6285

int 
db_env_set_func_free (void (*f)(void*)) {
6286 6287
    toku_set_func_free(f);
    return 0;
6288
}
6289

6290

6291
// Got to call dlmalloc, or else it won't get included.
6292 6293
void 
setup_dlmalloc (void) {
6294 6295 6296 6297
    db_env_set_func_malloc(dlmalloc);
    db_env_set_func_realloc(dlrealloc);
    db_env_set_func_free(dlfree);
}
6298 6299

// For test purposes only.
6300
// With this interface, all checkpoint users get the same callbacks and the same extras.
6301 6302
void 
db_env_set_checkpoint_callback (void (*callback_f)(void*), void* extra) {
6303
    toku_checkpoint_safe_client_lock();
6304 6305
    checkpoint_callback_f = callback_f;
    checkpoint_callback_extra = extra;
6306
    toku_checkpoint_safe_client_unlock();
6307
    //printf("set callback = %p, extra = %p\n", callback_f, extra);
6308
}
6309 6310 6311

void 
db_env_set_checkpoint_callback2 (void (*callback_f)(void*), void* extra) {
6312 6313 6314 6315 6316 6317
    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
6318

6319 6320
void 
db_env_set_recover_callback (void (*callback_f)(void*), void* extra) {
6321 6322 6323
    toku_recover_set_callback(callback_f, extra);
}

6324 6325
void 
db_env_set_recover_callback2 (void (*callback_f)(void*), void* extra) {
6326 6327 6328
    toku_recover_set_callback2(callback_f, extra);
}

6329 6330
void 
db_env_set_loader_size_factor (uint32_t factor) {
6331 6332 6333
    toku_brtloader_set_size_factor(factor);
}

6334 6335
void 
db_env_set_mvcc_garbage_collection_verification(u_int32_t verification_mode) {
Zardosht Kasheff's avatar
Zardosht Kasheff committed
6336 6337
    garbage_collection_debug = (verification_mode != 0);
}
6338

6339 6340 6341 6342 6343
// 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;
}
6344

Yoni Fogel's avatar
Yoni Fogel committed
6345 6346 6347 6348 6349 6350 6351 6352 6353 6354
// 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();
}

6355 6356 6357 6358

// For test purposes only, translate dname to iname
static int 
env_get_iname(DB_ENV* env, DBT* dname_dbt, DBT* iname_dbt) {
6359
    toku_ydb_lock();
6360
    DB *directory = env->i->directory;
6361
    int r = autotxn_db_get(directory, NULL, dname_dbt, iname_dbt, DB_SERIALIZABLE|DB_PRELOCKED); // allocates memory for iname
6362
    toku_ydb_unlock();
6363 6364 6365
    return r;
}

6366 6367 6368 6369 6370 6371 6372 6373 6374 6375 6376 6377 6378
/* 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.
6379 6380
// If "mark_as_loader" is true, then include a mark in the iname
// to indicate that the file is created by the brt loader.
6381 6382
// Return 0 on success (could fail if write lock not available).
int
6383
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) {
6384 6385 6386 6387 6388
    int rval;
    int i;
    
    int using_txns = env->i->open_flags & DB_INIT_TXN;
    DB_TXN * child = NULL;
6389
    TXNID xid = 0;
6390 6391 6392
    DBT dname_dbt;  // holds dname
    DBT iname_dbt;  // holds new iname
    
6393 6394 6395
    char * mark;

    if (mark_as_loader)
6396
	mark = "B";
6397 6398 6399
    else
	mark = "P";

6400 6401 6402 6403
    for (i=0; i<N; i++) {
	new_inames_in_env[i] = NULL;
    }

6404 6405 6406 6407 6408 6409 6410 6411 6412 6413 6414 6415
    // 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);
6416
	char * new_iname = create_iname(env, xid, hint, mark, i);               // allocates memory for iname_in_env
6417
	new_inames_in_env[i] = new_iname;
6418
        toku_fill_dbt(&iname_dbt, new_iname, strlen(new_iname) + 1);      // iname_in_env goes in directory
6419
        rval = toku_db_put(env->i->directory, child, &dname_dbt, &iname_dbt, 0);
6420
	if (rval) break;
Yoni Fogel's avatar
Yoni Fogel committed
6421 6422 6423 6424 6425 6426
    }

    // Generate load log entries.
    if (!rval && using_txns) {
        TOKUTXN ttxn = db_txn_struct_i(txn)->tokutxn;
        int do_fsync = 0;
6427
        LSN *get_lsn = NULL;
Yoni Fogel's avatar
Yoni Fogel committed
6428 6429 6430
        for (i = 0; i < N; i++) {
            BRT brt  = dbs[i]->i->brt;
            //Fsync is necessary for the last one only.
6431 6432 6433 6434 6435
            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
6436 6437
            if (rval) break;
        }
6438 6439 6440 6441 6442
    }
	
    if (using_txns) {
	// close txn
	if (rval == 0) {  // all well so far, commit child
6443
	    rval = toku_txn_commit(child, DB_TXN_NOSYNC, NULL, NULL);
6444 6445 6446
	    assert(rval==0);
	}
	else {         // abort child
6447
	    int r2 = toku_txn_abort(child, NULL, NULL);
6448 6449
	    assert(r2==0);
	    for (i=0; i<N; i++) {
6450 6451 6452 6453
		if (new_inames_in_env[i]) {
		    toku_free(new_inames_in_env[i]);
		    new_inames_in_env[i] = NULL;
		}
6454 6455 6456 6457 6458 6459 6460
	    }
	}
    }

    return rval;
}

6461
int
6462
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) {
6463
    toku_ydb_lock();
6464
    int r = ydb_load_inames(env, txn, N, dbs, new_inames_in_env, load_lsn, mark_as_loader);
6465 6466 6467 6468
    toku_ydb_unlock();
    return r;
}

6469 6470 6471 6472
// 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
6473
toku_test_db_redirect_dictionary(DB * db, char * dname_of_new_file, DB_TXN *dbtxn) {
6474 6475 6476 6477 6478 6479 6480 6481 6482 6483
    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
6484
    r = toku_db_get(db->dbenv->i->directory, dbtxn, &dname_dbt, &iname_dbt, DB_SERIALIZABLE);  // allocates memory for iname
6485 6486 6487
    assert(r==0);
    new_iname_in_env = iname_dbt.data;

Yoni Fogel's avatar
Yoni Fogel committed
6488
    r = toku_dictionary_redirect(new_iname_in_env, brt, tokutxn);
6489 6490 6491 6492

    toku_free(new_iname_in_env);
    return r;
}
6493 6494 6495 6496 6497 6498 6499 6500 6501 6502 6503

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

6504 6505
int 
toku_test_get_checkpointing_user_data_status (void) {
6506 6507
    return toku_cachetable_get_checkpointing_user_data_status();
}
Zardosht Kasheff's avatar
Zardosht Kasheff committed
6508

6509 6510 6511 6512
int
toku_grab_write_lock (DB* db, DBT* key, TOKUTXN tokutxn) {
    DB_TXN * txn = toku_txn_get_container_db_txn(tokutxn);
    //Left end of range == right end of range (point lock)
6513 6514
    RANGE_LOCK_REQUEST_S request;
    write_lock_request_init(&request, txn, db, key, key);
6515 6516 6517 6518
    int r = grab_range_lock(&request);
    return r;
}