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 1070 1071
            if ( flags && DB_CLOSE_DONT_TRIM_LOG ) {
                toku_logger_trim_log_files(env->i->logger, FALSE);
            }
1072
            r = toku_checkpoint(env->i->cachetable, env->i->logger, NULL, NULL, NULL, NULL);
1073
            if (r) {
1074 1075
		err_msg = "Cannot close environment (error during checkpoint)\n";
                toku_ydb_do_error(env, r, "%s", err_msg);
1076 1077
                goto panic_and_quit_early;
            }
1078 1079 1080 1081
            { //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) {
1082 1083
		    err_msg = "Cannot close environment due to zombie DBs\n";
                    r = toku_ydb_do_error(env, EINVAL, "%s", err_msg);
1084 1085 1086
                    goto panic_and_quit_early;
                }
            }
1087 1088
            r = toku_logger_close_rollback(env->i->logger, FALSE);
            if (r) {
1089 1090
		err_msg = "Cannot close environment (error during closing rollback cachefile)\n";
                toku_ydb_do_error(env, r, "%s", err_msg);
1091 1092 1093 1094 1095
                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) {
1096 1097
		err_msg = "Cannot close environment (error during checkpoint)\n";
                toku_ydb_do_error(env, r, "%s", err_msg);
1098 1099
                goto panic_and_quit_early;
            }
1100 1101
            r = toku_logger_shutdown(env->i->logger); 
            if (r) {
1102 1103
		err_msg = "Cannot close environment (error during logger shutdown)\n";
                toku_ydb_do_error(env, r, "%s", err_msg);
1104 1105
                goto panic_and_quit_early;
            }
1106
        }
1107
	toku_ydb_lock();
1108 1109
        r=toku_cachetable_close(&env->i->cachetable);
	if (r) {
1110 1111
	    err_msg = "Cannot close environment (cachetable close error)\n";
	    toku_ydb_do_error(env, r, "%s", err_msg);
1112
            goto panic_and_quit_early;
1113 1114 1115
	}
    }
    if (env->i->logger) {
1116 1117
        r=toku_logger_close(&env->i->logger);
	if (r) {
1118
	    err_msg = "Cannot close environment (logger close error)\n";
1119
            env->i->logger = NULL;
1120
	    toku_ydb_do_error(env, r, "%s", err_msg);
1121
            goto panic_and_quit_early;
1122 1123 1124 1125
	}
    }
    // 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)
1126 1127 1128 1129
    if (toku_env_is_panicked(env))
        goto panic_and_quit_early;
    else
	assert(env->i->panic_string==0);
1130

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

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

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

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

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

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

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

1234 1235
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
1236
    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
1237
}
Rich Prohaska's avatar
Rich Prohaska committed
1238 1239
#endif

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

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

1266 1267
static void 
toku_env_set_errfile(DB_ENV*env, FILE*errfile) {
1268 1269 1270
    env->i->errfile = errfile;
}

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

1388 1389 1390 1391 1392 1393
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;
1394 1395 1396 1397 1398 1399
}

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

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

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

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

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

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

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

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

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

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

1474 1475
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
1476
    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
1477 1478
}

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

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

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

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

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

1504 1505
static int 
locked_env_get_lg_max(DB_ENV * env, u_int32_t *lg_maxp) {
1506 1507 1508
    toku_ydb_lock(); int r = toku_env_get_lg_max(env, lg_maxp); toku_ydb_unlock(); return r;
}

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

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

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

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

1529 1530 1531 1532 1533 1534 1535 1536 1537 1538 1539 1540 1541 1542 1543
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
1544 1545 1546 1547 1548 1549 1550 1551 1552 1553 1554 1555 1556 1557 1558
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;
}

1559 1560 1561 1562 1563 1564 1565 1566 1567 1568 1569 1570 1571 1572 1573 1574 1575
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
1576

1577 1578 1579 1580 1581 1582 1583 1584 1585 1586 1587 1588 1589 1590 1591
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;
}

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 1644 1645 1646
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;
}

1647 1648 1649 1650 1651 1652 1653 1654 1655 1656 1657 1658 1659 1660
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();
}



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

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

1691 1692 1693 1694 1695 1696 1697 1698
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;
}

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

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

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

static int
1715
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) {
1716 1717 1718
    int r = env_check_avail_fs_space(env);
    if (r == 0) {
	toku_ydb_lock();
1719
	r = env_put_multiple(env, src_db, txn, src_key, src_val, num_dbs, db_array, keys, vals, flags_array);
1720 1721
	toku_ydb_unlock();
    }
1722 1723 1724 1725
    return r;
}

static int
1726
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) {
1727
    toku_ydb_lock();
1728
    int r = env_del_multiple(env, src_db, txn, src_key, src_val, num_dbs, db_array, keys, flags_array);
1729 1730 1731 1732
    toku_ydb_unlock();
    return r;
}

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

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

1767 1768 1769 1770 1771 1772 1773 1774 1775 1776 1777 1778 1779 1780 1781 1782
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];
    }
}
1783

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

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

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

    return r;
}

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

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


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

static toku_dbt_cmp toku_db_get_compare_fun(DB* db);

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

2271 2272
    engine_status_enable = 1;

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

2611 2612
static int 
toku_txn_begin(DB_ENV *env, DB_TXN * stxn, DB_TXN ** txn, u_int32_t flags, int internal) {
2613
    HANDLE_PANICKED_ENV(env);
2614
    HANDLE_ILLEGAL_WORKING_PARENT_TXN(env, stxn); //Cannot create child while child already exists.
Vincenzo Liberatore's avatar
Vincenzo Liberatore committed
2615 2616
    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");
2617 2618
    u_int32_t txn_flags = 0;
    txn_flags |= DB_TXN_NOWAIT; //We do not support blocking locks.
2619
    TOKU_ISOLATION child_isolation = TOKU_ISO_SERIALIZABLE;
2620 2621 2622 2623 2624 2625 2626 2627 2628
    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)
       ) 
    {
2629 2630 2631
        return toku_ydb_do_error(
            env, 
            EINVAL, 
2632
            "Invalid isolation flags set\n"
2633 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
    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;
2670
        }
2671
    }
2672 2673 2674 2675 2676 2677 2678
    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"
            );   
2679
    }
2680

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

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


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

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

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

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

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

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

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

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

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

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

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

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

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


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

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

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

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

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

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

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

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

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

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

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

3096

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

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

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

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

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

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

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

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

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

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

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

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

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

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

3229
typedef struct {
3230
    BOOL            is_write_lock;
3231 3232 3233 3234 3235 3236
    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
3237

3238 3239
static void
range_lock_request_init(RANGE_LOCK_REQUEST request,
3240
                        BOOL       is_write_lock,
3241 3242 3243
                        DB_TXN    *txn,
                        DB        *db,
                        DBT const *left_key,
3244
                        DBT const *right_key) {
3245
    request->is_write_lock = is_write_lock;
3246 3247 3248 3249 3250
    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
3251 3252
}

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

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

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) {
3278
        TXNID txn_anc_id = toku_txn_get_txnid(db_txn_struct_i(txn_anc)->tokutxn);
3279
        if (request->is_write_lock)
3280
            r = toku_lt_acquire_range_write_lock(request->lt, request->db, txn_anc_id,
3281
                                                 request->left_key, request->right_key);
3282 3283 3284
        else 
            r = toku_lt_acquire_range_read_lock(request->lt, request->db, txn_anc_id,
                                                request->left_key, request->right_key);
3285 3286 3287
    }
    //TODO: (Multithreading) Release lock protecting lock tree
    return r;
Yoni Fogel's avatar
Yoni Fogel committed
3288 3289
}

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

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

Zardosht Kasheff's avatar
Zardosht Kasheff committed
3299 3300
    toku_fill_dbt(&key_in_directory, dname, strlen(dname)+1);
    //Left end of range == right end of range (point lock)
3301 3302 3303
    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
3304
    int r = grab_range_lock(&request);
3305 3306 3307 3308
    if (r == 0)
	directory_read_locks++;
    else
	directory_read_locks_fail++;
Zardosht Kasheff's avatar
Zardosht Kasheff committed
3309 3310 3311
    return r;
}

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

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

3325 3326 3327 3328
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
3329

3330 3331 3332 3333 3334 3335
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
3336

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

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

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

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

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

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

3391 3392 3393 3394 3395
    int r = 0;
    if (unchecked_flags!=0) r = EINVAL;
    else {
        if (do_locking) {
            QUERY_CONTEXT_S context;
3396
            query_context_init_write(&context, c, lock_flags, NULL, NULL);
3397 3398 3399 3400 3401
            //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.
3402 3403
            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);
3404 3405
        }
    }
3406 3407
    return r;
}
3408

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

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

3417
    assert(context->do_locking);
3418
    assert(context->is_write_op);
3419 3420 3421 3422 3423
    assert(key!=NULL);
    assert(val!=NULL);
    //Lock:
    //  left(key,val)==right(key,val) == (key, val);
    RANGE_LOCK_REQUEST_S request;
3424
    write_lock_request_init(&request, context->txn, context->db, key, key);
3425 3426 3427
    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
3428 3429 3430
    return r;
}

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

3433 3434 3435 3436 3437 3438 3439 3440 3441 3442 3443
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);
}

3444 3445 3446
static int
toku_c_getf_first(DBC *c, u_int32_t flag, YDB_CALLBACK_FUNCTION f, void *extra) {
    HANDLE_PANICKED_DB(c->dbp);
3447
    HANDLE_CURSOR_ILLEGAL_WORKING_PARENT_TXN(c);
3448
    num_point_queries++;   // accountability
3449
    QUERY_CONTEXT_S context; //Describes the context of this query.
3450
    c_query_context_init(&context, c, flag, f, extra);
3451
    //toku_brt_cursor_first will call c_getf_first_callback(..., context) (if query is successful)
3452
    int r = toku_brt_cursor_first(dbc_struct_i(c)->c, c_getf_first_callback, &context);
3453 3454 3455 3456 3457 3458 3459 3460 3461 3462 3463 3464 3465 3466 3467 3468 3469 3470 3471 3472
    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) {
3473 3474 3475 3476 3477
            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);
3478 3479 3480 3481 3482 3483 3484 3485
        }
        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);
3486
        r = context->r_user_callback;
3487 3488 3489 3490 3491 3492 3493 3494 3495 3496 3497
    }

    //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);
3498
    HANDLE_CURSOR_ILLEGAL_WORKING_PARENT_TXN(c);
3499
    num_point_queries++;   // accountability
3500
    QUERY_CONTEXT_S context; //Describes the context of this query.
3501
    c_query_context_init(&context, c, flag, f, extra); 
3502
    //toku_brt_cursor_last will call c_getf_last_callback(..., context) (if query is successful)
3503
    int r = toku_brt_cursor_last(dbc_struct_i(c)->c, c_getf_last_callback, &context);
3504 3505 3506 3507 3508 3509 3510 3511 3512 3513 3514 3515 3516 3517 3518 3519 3520 3521 3522 3523
    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) {
3524 3525 3526 3527 3528
            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);
3529 3530 3531 3532 3533 3534 3535 3536
        }
        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);
3537
        r = context->r_user_callback;
3538 3539 3540 3541 3542 3543 3544 3545 3546 3547 3548 3549
    }

    //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);
3550
    HANDLE_CURSOR_ILLEGAL_WORKING_PARENT_TXN(c);
3551
    if (toku_c_uninitialized(c)) r = toku_c_getf_first(c, flag, f, extra);
3552 3553
    else {
        QUERY_CONTEXT_S context; //Describes the context of this query.
3554
        num_sequential_queries++;   // accountability
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 3570 3571 3572 3573 3574 3575 3576 3577 3578 3579 3580 3581 3582
        if (r == TOKUDB_USER_CALLBACK_ERROR) r = context.base.r_user_callback;
    }
    return r;
}

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

    int r;

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

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

        toku_brt_cursor_peek(context->c, &prevkey, &prevval);
3583 3584
        range_lock_request_init(&request, context->is_write_op, context->txn, context->db,
                                prevkey, right_key);
3585 3586 3587 3588 3589 3590 3591
        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);
3592
        r = context->r_user_callback;
3593 3594 3595 3596 3597 3598 3599 3600 3601 3602 3603 3604
    }

    //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);
3605
    HANDLE_CURSOR_ILLEGAL_WORKING_PARENT_TXN(c);
3606
    if (toku_c_uninitialized(c)) r = toku_c_getf_last(c, flag, f, extra);
3607 3608
    else {
        QUERY_CONTEXT_S context; //Describes the context of this query.
3609
        num_sequential_queries++;   // accountability
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 3626 3627 3628 3629 3630 3631 3632 3633 3634 3635 3636 3637
//result is the result of the query (i.e. 0 means found, DB_NOTFOUND, etc..)
static int
c_getf_prev_callback(ITEMLEN keylen, bytevec key, ITEMLEN vallen, bytevec val, void *extra) {
    QUERY_CONTEXT      super_context = extra;
    QUERY_CONTEXT_BASE context       = &super_context->base;

    int r;

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

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

        toku_brt_cursor_peek(context->c, &prevkey, &prevval);
3638 3639
        range_lock_request_init(&request, context->is_write_op, context->txn, context->db,
                                left_key, prevkey);
3640 3641 3642 3643 3644 3645 3646
        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);
3647
        r = context->r_user_callback;
3648 3649 3650 3651
    }

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

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

    QUERY_CONTEXT_S context; //Describes the context of this query.
3662
    num_sequential_queries++;   // accountability
3663
    c_query_context_init(&context, c, flag, f, extra); 
3664
    //toku_brt_cursor_current will call c_getf_current_callback(..., context) (if query is successful)
3665
    int r = toku_brt_cursor_current(dbc_struct_i(c)->c, DB_CURRENT, c_getf_current_callback, &context);
3666 3667 3668 3669 3670 3671 3672 3673 3674 3675 3676 3677 3678 3679 3680 3681 3682 3683 3684
    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);
3685
        r = context->r_user_callback;
3686 3687 3688 3689 3690 3691 3692 3693 3694
    }

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

    QUERY_CONTEXT_S context; //Describes the context of this query.
3698
    num_sequential_queries++;   // accountability
3699
    c_query_context_init(&context, c, flag, f, extra); 
3700
    //toku_brt_cursor_current will call c_getf_current_callback(..., context) (if query is successful)
3701
    int r = toku_brt_cursor_current(dbc_struct_i(c)->c, DB_CURRENT_BINDING, c_getf_current_callback, &context);
3702 3703 3704 3705 3706 3707 3708 3709 3710
    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);
3711
    HANDLE_CURSOR_ILLEGAL_WORKING_PARENT_TXN(c);
3712 3713

    QUERY_CONTEXT_WITH_INPUT_S context; //Describes the context of this query.
3714
    num_point_queries++;   // accountability
3715 3716
    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)
3717
    int r = toku_brt_cursor_set(dbc_struct_i(c)->c, key, c_getf_set_callback, &context);
3718 3719 3720 3721 3722 3723 3724 3725 3726 3727 3728 3729 3730 3731 3732 3733 3734 3735 3736 3737 3738 3739
    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;
3740 3741
        range_lock_request_init(&request, context->is_write_op, context->txn, context->db, 
                                super_context->input_key, super_context->input_key);
3742
        r = grab_range_lock(&request);
3743 3744
    } else 
        r = 0;
3745 3746 3747 3748

    //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);
3749
        r = context->r_user_callback;
3750 3751 3752 3753 3754 3755 3756 3757 3758 3759 3760
    }

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

    QUERY_CONTEXT_WITH_INPUT_S context; //Describes the context of this query.
3764
    num_point_queries++;   // accountability
3765 3766
    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)
3767
    int r = toku_brt_cursor_set_range(dbc_struct_i(c)->c, key, c_getf_set_range_callback, &context);
3768 3769 3770 3771 3772 3773 3774 3775 3776 3777 3778 3779 3780 3781 3782 3783 3784 3785 3786 3787 3788 3789 3790
    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;
3791 3792 3793 3794 3795 3796
        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);
3797 3798 3799 3800 3801 3802 3803
        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);
3804
        r = context->r_user_callback;
3805 3806 3807 3808 3809 3810
    }

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

3811 3812 3813 3814 3815 3816 3817 3818
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.
3819
    num_point_queries++;   // accountability
3820 3821
    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)
3822
    int r = toku_brt_cursor_set_range_reverse(dbc_struct_i(c)->c, key, c_getf_set_range_reverse_callback, &context);
3823 3824 3825 3826 3827 3828 3829 3830 3831 3832 3833 3834 3835 3836 3837 3838 3839 3840 3841 3842 3843 3844 3845 3846
    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) {
3847 3848 3849 3850 3851
            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);
3852 3853 3854 3855 3856 3857 3858 3859
        }
        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);
3860
        r = context->r_user_callback;
3861 3862 3863 3864 3865 3866
    }

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

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

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

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

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

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

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

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

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

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

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

3952 3953 3954 3955 3956 3957 3958 3959 3960
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;
}

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

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

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

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

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

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

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

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

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

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

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

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

    HANDLE_PANICKED_ENV(env);

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

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

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

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

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

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

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

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


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

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

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

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

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

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

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

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

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

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

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

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

Yoni Fogel's avatar
Yoni Fogel committed
4297 4298

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

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

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

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

4328
    env_panic(env, r, panic_string);
4329

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

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

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

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

4360

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

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

4397

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

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

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

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

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

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

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

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

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

    return r;
}

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

    return r;
}

4732 4733 4734 4735 4736 4737 4738 4739 4740 4741 4742
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)
4743 4744 4745
    RANGE_LOCK_REQUEST_S request;
    write_lock_request_init(&request, txn, db->dbenv->i->directory,
                            &key_in_directory, &key_in_directory);
4746 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
    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;
}

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

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

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

    HANDLE_PANICKED_ENV(env);

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

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

4934 4935 4936
    HANDLE_ILLEGAL_WORKING_PARENT_TXN(env, txn);

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

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

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

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

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

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

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

4992 4993
    }

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

static int
env_update_multiple(DB_ENV *env, DB *src_db, DB_TXN *txn,                                
5004 5005 5006
                    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, 
5007
                    uint32_t num_keys, DBT keys[], 
5008
                    uint32_t num_vals, DBT vals[]) {
5009 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
    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];

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

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

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

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

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

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

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

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

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

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

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

5172 5173 5174 5175

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

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

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
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) {
5212
	r = toku_txn_begin(env, txn, &child, DB_TXN_NOSYNC, 1);
5213 5214 5215 5216
	assert(r==0);
    }

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

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

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

5261 5262
}

5263 5264 5265

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

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

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

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

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

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

5394 5395 5396
//
// This function is the only way to set a descriptor of a DB.
//
5397
static int 
5398
toku_db_change_descriptor(DB *db, DB_TXN* txn, const DBT* descriptor, u_int32_t flags) {
5399
    HANDLE_PANICKED_DB(db);
5400
    HANDLE_DB_ILLEGAL_WORKING_PARENT_TXN(db, txn);
5401
    int r;
5402 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
    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);
5432
    r = toku_brt_change_descriptor(db->i->brt, &old_descriptor, descriptor, TRUE, ttxn);
5433 5434
cleanup:
    if (old_descriptor.data) toku_free(old_descriptor.data);
5435 5436 5437
    return r;
}

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

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

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

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

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

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

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

5484 5485
static int 
toku_db_stat64(DB * db, DB_TXN *txn, DB_BTREE_STAT64 *s) {
5486
    HANDLE_PANICKED_DB(db);
5487
    HANDLE_DB_ILLEGAL_WORKING_PARENT_TXN(db, txn);
5488
    struct brtstat64_s brtstat;
5489 5490 5491 5492 5493
    TOKUTXN tokutxn = NULL;
    if (txn != NULL) {
        tokutxn = db_txn_struct_i(txn)->tokutxn;
    }
    int r = toku_brt_stat64(db->i->brt, tokutxn, &brtstat);
5494 5495 5496 5497 5498 5499 5500
    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;
5501
}
5502 5503 5504

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

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

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

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

5542 5543 5544
    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
5545 5546 5547
    return r;
}

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

    int r;

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

    *row_count = 0;

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

    return r;
}

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

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

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

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

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

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

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

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

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

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

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

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

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

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

Zardosht Kasheff's avatar
Zardosht Kasheff committed
5942

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

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

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

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

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

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


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

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

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

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

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

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

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

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

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

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

/* need db_strerror_r for multiple threads */

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

6289

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

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

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

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

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

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

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

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

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

6354 6355 6356 6357

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

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

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

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

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

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

    return rval;
}

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

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

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

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

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

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

6508 6509 6510 6511
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)
6512 6513
    RANGE_LOCK_REQUEST_S request;
    write_lock_request_init(&request, txn, db, key, key);
6514 6515 6516 6517
    int r = grab_range_lock(&request);
    return r;
}