X-Git-Url: https://arthur.barton.de/cgi-bin/gitweb.cgi?a=blobdiff_plain;f=etc%2Fcnid_dbd%2Fdbif.c;h=b96b2e6878c38f0f807a10387d12a69b487faa34;hb=f9a95c83ef9379a0cf6d4d46f35ca0a12b84dd9b;hp=38401fc577be5f46a39de489fdbbb4ea4adb0e54;hpb=d848539ce9cec21075139f543cd506e3c5a102d6;p=netatalk.git diff --git a/etc/cnid_dbd/dbif.c b/etc/cnid_dbd/dbif.c index 38401fc5..b96b2e68 100644 --- a/etc/cnid_dbd/dbif.c +++ b/etc/cnid_dbd/dbif.c @@ -1,6 +1,4 @@ /* - * $Id: dbif.c,v 1.12 2009-05-22 20:48:44 franklahm Exp $ - * * Copyright (C) Joerg Lenneis 2003 * Copyright (C) Frank Lahm 2009 * All Rights Reserved. See COPYING. @@ -13,52 +11,235 @@ #include #include #include -#ifdef HAVE_SYS_TYPES_H -#include -#endif /* HAVE_SYS_TYPES_H */ #include #include #include #include #include -#include + #include + +#include +#include +#include + #include "db_param.h" #include "dbif.h" #include "pack.h" #define DB_ERRLOGFILE "db_errlog" -static char *old_dbfiles[] = {"cnid.db", NULL}; - -/* --------------- */ -static int upgrade_required() +/*! + * Get the db stamp which is the st_ctime of the file "cnid2.db" and store it in buffer + */ +static int dbif_stamp(DBD *dbd, void *buffer, int size) { - int i; - int found = 0; + EC_INIT; struct stat st; + int cwd = -1; - for (i = 0; old_dbfiles[i] != NULL; i++) { - if ( !(stat(old_dbfiles[i], &st) < 0) ) { - found++; - continue; - } - if (errno != ENOENT) { - LOG(log_error, logtype_cnid, "cnid_open: Checking %s gave %s", old_dbfiles[i], strerror(errno)); - found++; + if (size < 8) + EC_FAIL; + + /* Remember cwd */ + if ((cwd = open(".", O_RDONLY)) < 0) { + LOG(log_error, logtype_cnid, "error opening cwd: %s", strerror(errno)); + EC_FAIL; + } + + /* chdir to db_envhome */ + if ((chdir(dbd->db_envhome)) != 0) { + LOG(log_error, logtype_cnid, "error chdiring to db_env '%s': %s", dbd->db_envhome, strerror(errno)); + EC_FAIL; + } + + if (stat(dbd->db_table[DBIF_CNID].name, &st) < 0) { + LOG(log_error, logtype_cnid, "error stating database %s: %s", dbd->db_table[DBIF_CNID].name, db_strerror(errno)); + EC_FAIL; + } + + LOG(log_maxdebug, logtype_cnid,"stamp: %s", asctime(localtime(&st.st_ctime))); + + memset(buffer, 0, size); + memcpy(buffer, &st.st_ctime, sizeof(st.st_ctime)); + +EC_CLEANUP: + if (cwd != -1) { + if (fchdir(cwd) != 0) { + LOG(log_error, logtype_cnid, "error chdiring back: %s", strerror(errno)); + EC_STATUS(-1); } + close(cwd); } - return found; + EC_EXIT; } -/* --------------- */ -int dbif_stamp(DBD *dbd, void *buffer, int size) +/*! + * Inititialize rootinfo key (which has CNID 0 as key) + * + * This also "stamps" the database, which means storing st.st_ctime of the + * "cnid2.db" file in the rootinfo data at the DEV offset + * + * @param dbd (rw) database handle + * @param version (r) database version number + * + * @returns -1 on error, 0 on success + */ +static int dbif_init_rootinfo(DBD *dbd, int version) { - struct stat st; - int rc,cwd; + DBT key, data; + uint32_t v; + char buf[ROOTINFO_DATALEN]; - if (size < 8) + LOG(log_debug, logtype_cnid, "Setting CNID database version to %u", version); + + v = version; + v = htonl(v); + + memset(&key, 0, sizeof(key)); + memset(&data, 0, sizeof(data)); + key.data = ROOTINFO_KEY; + key.size = ROOTINFO_KEYLEN; + data.data = buf; + data.size = ROOTINFO_DATALEN; + + memcpy(buf, ROOTINFO_DATA, ROOTINFO_DATALEN); + memcpy(buf + CNID_DID_OFS, &v, sizeof(v)); + if (dbif_stamp(dbd, buf + CNID_DEV_OFS, CNID_DEV_LEN) < 0) + return -1; + + if (dbif_put(dbd, DBIF_CNID, &key, &data, 0) < 0) return -1; + if (dbif_txn_commit(dbd) != 1) { + LOG(log_error, logtype_cnid, "dbif_init_rootinfo: cant commit txn"); + return -1; + } + + return 0; +} + +/*! + * Return CNID database version number + * + * Returns version in *version + * + * @returns -1 on error, 0 if theres no rootinfo key yet, 1 if *version is returned + */ +static int dbif_getversion(DBD *dbd, uint32_t *version) +{ + DBT key, data; + int ret; + + LOG(log_maxdebug, logtype_cnid, "dbif_getversion: reading version info"); + + *version = -1; + memset(&key, 0, sizeof(key)); + memset(&data, 0, sizeof(data)); + key.data = ROOTINFO_KEY; + key.size = ROOTINFO_KEYLEN; + + switch (dbif_get(dbd, DBIF_CNID, &key, &data, 0)) { + case 1: /* found */ + memcpy(version, (char *)data.data + CNID_DID_OFS, sizeof(uint32_t)); + *version = ntohl(*version); + LOG(log_debug, logtype_cnid, "CNID database version %u", *version); + ret = 1; + break; + case 0: /* not found */ + LOG(log_debug, logtype_cnid, "dbif_getversion: no version info found"); + ret = 0; + break; + default: + LOG(log_error, logtype_cnid, "dbif_getversion: database error"); + ret = -1; + break; + } + + return ret; +} + +/*! + * Set CNID database version number + * + * Initializes rootinfo key as neccessary + * @returns -1 on error, 0 on success + */ +static int dbif_setversion(DBD *dbd, uint32_t version) +{ + int ret; + DBT key, data; + uint32_t v; + + LOG(log_debug, logtype_cnid, "Setting CNID database version to %u", version); + + v = version; + v = htonl(v); + + memset(&key, 0, sizeof(key)); + memset(&data, 0, sizeof(data)); + key.data = ROOTINFO_KEY; + key.size = ROOTINFO_KEYLEN; + + if ((ret = dbif_get(dbd, DBIF_CNID, &key, &data, 0)) == -1) + return -1; + if (ret == 0) { + /* No rootinfo key yet, init it */ + if (dbif_init_rootinfo(dbd, CNID_VERSION) != 0) + return -1; + /* Now try again */ + if (dbif_get(dbd, DBIF_CNID, &key, &data, 0) == -1) + return -1; + } + memcpy((char *)data.data + CNID_DID_OFS, &v, sizeof(v)); + data.size = ROOTINFO_DATALEN; + if (dbif_put(dbd, DBIF_CNID, &key, &data, 0) < 0) + return -1; + + return 0; +} + +/*! + * Upgrade CNID database versions, initialize rootinfo key as as necessary in dbif_setversion() + * + * For now this does nothing, as upgrading from ver. 0 to 1 is done in dbif_open + */ +#define UNINTIALIZED_DB UINT32_MAX +static int dbif_upgrade(DBD *dbd) +{ + uint32_t version = CNID_VERSION_UNINTIALIZED_DB; + + if (dbif_getversion(dbd, &version) == -1) + return -1; + if (version == CNID_VERSION_UNINTIALIZED_DB) { + version = CNID_VERSION; + if (dbif_setversion(dbd, CNID_VERSION) != 0) + return -1; + } + + /* + * Do upgrade stuff ... + */ + + /* Write current version to database */ + if (version != CNID_VERSION) { + if (dbif_setversion(dbd, CNID_VERSION) != 0) + return -1; + } + + LOG(log_debug, logtype_cnid, "Finished CNID database version upgrade check"); + + return 0; +} + +/* --------------- */ +static int dbif_openlog(DBD *dbd) +{ + int ret = 0; + int cwd = -1; + + if ( ! dbd->db_filename) + /* in memory db */ + return 0; /* Remember cwd */ if ((cwd = open(".", O_RDONLY)) < 0) { @@ -69,22 +250,165 @@ int dbif_stamp(DBD *dbd, void *buffer, int size) /* chdir to db_envhome */ if ((chdir(dbd->db_envhome)) != 0) { LOG(log_error, logtype_cnid, "error chdiring to db_env '%s': %s", dbd->db_envhome, strerror(errno)); - return -1; + ret = -1; + goto exit; } - if ((rc = stat(dbd->db_table[DBIF_CNID].name, &st)) < 0) { - LOG(log_error, logtype_cnid, "error stating database %s: %s", dbd->db_table[DBIF_CNID].name, db_strerror(rc)); - return -1; + if ((dbd->db_errlog = fopen(DB_ERRLOGFILE, "a")) == NULL) + LOG(log_warning, logtype_cnid, "error creating/opening DB errlogfile: %s", strerror(errno)); + + if (dbd->db_errlog != NULL) { + dbd->db_env->set_errfile(dbd->db_env, dbd->db_errlog); + dbd->db_env->set_msgfile(dbd->db_env, dbd->db_errlog); } - memset(buffer, 0, size); - memcpy(buffer, &st.st_ctime, sizeof(st.st_ctime)); - if ((fchdir(cwd)) != 0) { - LOG(log_error, logtype_cnid, "error chdiring back: %s", strerror(errno)); +exit: + if (cwd != -1) { + if ((fchdir(cwd)) != 0) { + LOG(log_error, logtype_cnid, "error chdiring back: %s", strerror(errno)); + ret = -1; + } + close(cwd); + } + return ret; +} + +/* --------------- */ +static int dbif_logautorem(DBD *dbd) +{ + int ret = 0; + int cwd = -1; + char **logfiles = NULL; + char **file; + + if ( ! dbd->db_filename) + /* in memory db */ + return 0; + + /* Remember cwd */ + if ((cwd = open(".", O_RDONLY)) < 0) { + LOG(log_error, logtype_cnid, "error opening cwd: %s", strerror(errno)); return -1; } - return 0; + /* chdir to db_envhome */ + if ((chdir(dbd->db_envhome)) != 0) { + LOG(log_error, logtype_cnid, "error chdiring to db_env '%s': %s", dbd->db_envhome, strerror(errno)); + ret = -1; + goto exit; + } + + if ((ret = dbd->db_env->log_archive(dbd->db_env, &logfiles, 0)) != 0) { + LOG(log_error, logtype_cnid, "error getting list of stale logfiles: %s", + db_strerror(ret)); + dbd->db_env->close(dbd->db_env, 0); + dbd->db_env = NULL; + ret = -1; + goto exit; + } + + if (logfiles != NULL) { + for (file = logfiles; *file != NULL; file++) { + if (unlink(*file) < 0) + LOG(log_warning, logtype_cnid, "Error removing stale logfile %s: %s", *file, strerror(errno)); + } + free(logfiles); + } + +exit: + if (cwd != -1) { + if ((fchdir(cwd)) != 0) { + LOG(log_error, logtype_cnid, "error chdiring back: %s", strerror(errno)); + ret = -1; + } + close(cwd); + } + return ret; +} + +/*! + * Get lock on db lock file + * + * @args cmd (r) lock command: + * LOCK_FREE: close lockfd + * LOCK_UNLOCK: unlock lockm keep lockfd open + * LOCK_EXCL: F_WRLCK on lockfd + * LOCK_SHRD: F_RDLCK on lockfd + * @args dbpath (r) path to lockfile, only used on first call, + * later the stored fd is used + * @returns LOCK_FREE/LOCK_UNLOCK return 0 on success, -1 on error + * LOCK_EXCL/LOCK_SHRD return LOCK_EXCL or LOCK_SHRD respectively on + * success, 0 if the lock couldn't be acquired, -1 on other errors + */ +int get_lock(int cmd, const char *dbpath) +{ + static int lockfd = -1; + int ret; + char lockpath[PATH_MAX]; + struct stat st; + + switch (cmd) { + case LOCK_FREE: + if (lockfd == -1) + return -1; + close(lockfd); + lockfd = -1; + return 0; + + case LOCK_UNLOCK: + if (lockfd == -1) + return -1; + return unlock(lockfd, 0, SEEK_SET, 0); + + case LOCK_EXCL: + case LOCK_SHRD: + if (lockfd == -1) { + if ( (strlen(dbpath) + strlen(LOCKFILENAME+1)) > (PATH_MAX - 1) ) { + LOG(log_error, logtype_cnid, ".AppleDB pathname too long"); + return -1; + } + strncpy(lockpath, dbpath, PATH_MAX - 1); + strcat(lockpath, "/"); + strcat(lockpath, LOCKFILENAME); + + if ((lockfd = open(lockpath, O_RDWR | O_CREAT, 0644)) < 0) { + LOG(log_error, logtype_cnid, "Error opening lockfile: %s", strerror(errno)); + return -1; + } + + if ((stat(dbpath, &st)) != 0) { + LOG(log_error, logtype_cnid, "Error statting lockfile: %s", strerror(errno)); + return -1; + } + + if ((chown(lockpath, st.st_uid, st.st_gid)) != 0) { + LOG(log_error, logtype_cnid, "Error inheriting lockfile permissions: %s", + strerror(errno)); + return -1; + } + } + + if (cmd == LOCK_EXCL) + ret = write_lock(lockfd, 0, SEEK_SET, 0); + else + ret = read_lock(lockfd, 0, SEEK_SET, 0); + + if (ret != 0) { + if (cmd == LOCK_SHRD) + LOG(log_error, logtype_cnid, "Volume CNID db is locked, try again..."); + return 0; + } + + LOG(log_debug, logtype_cnid, "get_lock: got %s lock", + cmd == LOCK_EXCL ? "LOCK_EXCL" : "LOCK_SHRD"); + return cmd; + + default: + return -1; + } /* switch(cmd) */ + + /* deadc0de, never get here */ + return -1; } /* --------------- */ @@ -117,14 +441,19 @@ DBD *dbif_init(const char *envhome, const char *filename) dbd->db_table[DBIF_CNID].name = "cnid2.db"; dbd->db_table[DBIF_IDX_DEVINO].name = "devino.db"; dbd->db_table[DBIF_IDX_DIDNAME].name = "didname.db"; + dbd->db_table[DBIF_IDX_NAME].name = "name.db"; dbd->db_table[DBIF_CNID].type = DB_BTREE; dbd->db_table[DBIF_IDX_DEVINO].type = DB_BTREE; dbd->db_table[DBIF_IDX_DIDNAME].type = DB_BTREE; + dbd->db_table[DBIF_IDX_NAME].type = DB_BTREE; dbd->db_table[DBIF_CNID].openflags = DB_CREATE; dbd->db_table[DBIF_IDX_DEVINO].openflags = DB_CREATE; dbd->db_table[DBIF_IDX_DIDNAME].openflags = DB_CREATE; + dbd->db_table[DBIF_IDX_NAME].openflags = DB_CREATE; + + dbd->db_table[DBIF_IDX_NAME].flags = DB_DUPSORT; return dbd; } @@ -132,41 +461,13 @@ DBD *dbif_init(const char *envhome, const char *filename) /* We must open the db_env with an absolute pathname, as `dbd` keeps chdir'ing, which breaks e.g. bdb logfile-rotation with relative pathnames. - But still we use relative paths with upgrade_required() and the DB_ERRLOGFILE + But still we use relative paths with DB_ERRLOGFILE in order to avoid creating absolute paths by copying. Both have no problem with a relative path. */ int dbif_env_open(DBD *dbd, struct db_param *dbp, uint32_t dbenv_oflags) { - int ret, cwd; - char **logfiles = NULL; - char **file; - - /* Remember cwd */ - if ((cwd = open(".", O_RDONLY)) < 0) { - LOG(log_error, logtype_cnid, "error opening cwd: %s", strerror(errno)); - return -1; - } - - /* chdir to db_envhome. makes it easier checking for old db files and creating db_errlog file */ - if ((chdir(dbd->db_envhome)) != 0) { - LOG(log_error, logtype_cnid, "error chdiring to db_env '%s': %s", dbd->db_envhome, strerror(errno)); - return -1; - } - - /* Refuse to do anything if this is an old version of the CNID database */ - if (upgrade_required()) { - LOG(log_error, logtype_cnid, "Found version 1 of the CNID database. Please upgrade to version 2"); - return -1; - } - - if ((dbd->db_errlog = fopen(DB_ERRLOGFILE, "a")) == NULL) - LOG(log_warning, logtype_cnid, "error creating/opening DB errlogfile: %s", strerror(errno)); - - if ((fchdir(cwd)) != 0) { - LOG(log_error, logtype_cnid, "error chdiring back: %s", strerror(errno)); - return -1; - } + int ret; if ((ret = db_env_create(&dbd->db_env, 0))) { LOG(log_error, logtype_cnid, "error creating DB environment: %s", @@ -175,12 +476,15 @@ int dbif_env_open(DBD *dbd, struct db_param *dbp, uint32_t dbenv_oflags) return -1; } - if (dbd->db_errlog != NULL) { - dbd->db_env->set_errfile(dbd->db_env, dbd->db_errlog); - dbd->db_env->set_msgfile(dbd->db_env, dbd->db_errlog); - } + dbd->db_param = *dbp; + + if ((dbif_openlog(dbd)) != 0) + return -1; if (dbenv_oflags & DB_RECOVER) { + + LOG(log_debug, logtype_cnid, "Running recovery"); + dbd->db_env->set_verbose(dbd->db_env, DB_VERB_RECOVERY, 1); /* Open the database for recovery using DB_PRIVATE option which is faster */ if ((ret = dbd->db_env->open(dbd->db_env, dbd->db_envhome, dbenv_oflags | DB_PRIVATE, 0))) { @@ -201,6 +505,7 @@ int dbif_env_open(DBD *dbd, struct db_param *dbp, uint32_t dbenv_oflags) dbd->db_env = NULL; return -1; } + dbd->db_errlog = NULL; if ((ret = db_env_create(&dbd->db_env, 0))) { LOG(log_error, logtype_cnid, "error creating DB environment after recovery: %s", @@ -208,6 +513,11 @@ int dbif_env_open(DBD *dbd, struct db_param *dbp, uint32_t dbenv_oflags) dbd->db_env = NULL; return -1; } + + if ((dbif_openlog(dbd)) != 0) + return -1; + + LOG(log_debug, logtype_cnid, "Finished recovery."); } if ((ret = dbd->db_env->set_cachesize(dbd->db_env, 0, 1024 * dbp->cachesize, 0))) { @@ -218,10 +528,22 @@ int dbif_env_open(DBD *dbd, struct db_param *dbp, uint32_t dbenv_oflags) return -1; } - if (dbd->db_errlog != NULL) { - dbd->db_env->set_errfile(dbd->db_env, dbd->db_errlog); - dbd->db_env->set_msgfile(dbd->db_env, dbd->db_errlog); + if ((ret = dbd->db_env->set_lk_max_locks(dbd->db_env, dbp->maxlocks))) { + LOG(log_error, logtype_cnid, "error setting DB environment maxlocks to %i: %s", + 10000, db_strerror(ret)); + dbd->db_env->close(dbd->db_env, 0); + dbd->db_env = NULL; + return -1; + } + + if ((ret = dbd->db_env->set_lk_max_objects(dbd->db_env, dbp->maxlockobjs))) { + LOG(log_error, logtype_cnid, "error setting DB environment max lockobjects to %i: %s", + 10000, db_strerror(ret)); + dbd->db_env->close(dbd->db_env, 0); + dbd->db_env = NULL; + return -1; } + if ((ret = dbd->db_env->open(dbd->db_env, dbd->db_envhome, dbenv_oflags, 0))) { LOG(log_error, logtype_cnid, "error opening DB environment after recovery: %s", db_strerror(ret)); @@ -239,20 +561,8 @@ int dbif_env_open(DBD *dbd, struct db_param *dbp, uint32_t dbenv_oflags) } if (dbp->logfile_autoremove) { - if (dbd->db_env->log_archive(dbd->db_env, &logfiles, 0)) { - LOG(log_error, logtype_cnid, "error getting list of stale logfiles: %s", - db_strerror(ret)); - dbd->db_env->close(dbd->db_env, 0); - dbd->db_env = NULL; + if ((dbif_logautorem(dbd)) != 0) return -1; - } - if (logfiles != NULL) { - for (file = logfiles; *file != NULL; file++) { - if (unlink(*file) < 0) - LOG(log_warning, logtype_cnid, "Error removing stale logfile %s: %s", *file, strerror(errno)); - } - free(logfiles); - } #if DB_VERSION_MAJOR > 4 || (DB_VERSION_MAJOR == 4 && DB_VERSION_MINOR >= 7) if ((ret = dbd->db_env->log_set_config(dbd->db_env, DB_LOG_AUTO_REMOVE, 1))) { @@ -277,12 +587,55 @@ int dbif_env_open(DBD *dbd, struct db_param *dbp, uint32_t dbenv_oflags) } /* --------------- */ -int dbif_open(DBD *dbd, struct db_param *dbp _U_, int reindex) +int dbif_open(DBD *dbd, struct db_param *dbp, int reindex) { - int ret; - int i; + int ret, i, cwd; u_int32_t count; + struct stat st; + DB *upgrade_db; + + /* Try to upgrade if it's a normal on-disk database */ + if (dbd->db_envhome) { + /* Remember cwd */ + if ((cwd = open(".", O_RDONLY)) < 0) { + LOG(log_error, logtype_cnid, "error opening cwd: %s", strerror(errno)); + return -1; + } + + /* chdir to db_envhome. makes it easier checking for old db files and creating db_errlog file */ + if ((chdir(dbd->db_envhome)) != 0) { + LOG(log_error, logtype_cnid, "error chdiring to db_env '%s': %s", dbd->db_envhome, strerror(errno)); + return -1; + } + + if ((stat(dbd->db_filename, &st)) == 0) { + LOG(log_debug, logtype_cnid, "See if we can upgrade the CNID database..."); + if ((ret = db_create(&upgrade_db, dbd->db_env, 0))) { + LOG(log_error, logtype_cnid, "error creating handle for database: %s", db_strerror(ret)); + return -1; + } + if ((ret = upgrade_db->upgrade(upgrade_db, dbd->db_filename, 0))) { + LOG(log_error, logtype_cnid, "error upgarding database: %s", db_strerror(ret)); + return -1; + } + if ((ret = upgrade_db->close(upgrade_db, 0))) { + LOG(log_error, logtype_cnid, "error closing database: %s", db_strerror(ret)); + return -1; + } + if ((ret = dbd->db_env->txn_checkpoint(dbd->db_env, 0, 0, DB_FORCE))) { + LOG(log_error, logtype_cnid, "error forcing checkpoint: %s", db_strerror(ret)); + return -1; + } + LOG(log_debug, logtype_cnid, "Finished BerkeleyBD upgrade check"); + } + + if ((fchdir(cwd)) != 0) { + LOG(log_error, logtype_cnid, "error chdiring back: %s", strerror(errno)); + return -1; + } + } + /* Now open databases ... */ for (i = 0; i != DBIF_DB_CNT; i++) { if ((ret = db_create(&dbd->db_table[i].db, dbd->db_env, 0))) { LOG(log_error, logtype_cnid, "error creating handle for database %s: %s", @@ -299,6 +652,18 @@ int dbif_open(DBD *dbd, struct db_param *dbp _U_, int reindex) } } + if ( ! dbd->db_env) { /* In memory db */ + if ((ret = dbd->db_table[i].db->set_cachesize(dbd->db_table[i].db, + 0, + dbp->cachesize, + 4)) /* split in 4 memory chunks */ + < 0) { + LOG(log_error, logtype_cnid, "error setting cachesize %u KB for database %s: %s", + dbp->cachesize / 1024, dbd->db_table[i].name, db_strerror(ret)); + return -1; + } + } + if (dbd->db_table[i].db->open(dbd->db_table[i].db, dbd->db_txn, dbd->db_filename, @@ -309,8 +674,6 @@ int dbif_open(DBD *dbd, struct db_param *dbp _U_, int reindex) LOG(log_error, logtype_cnid, "Cant open database"); return -1; } - if (dbd->db_errlog != NULL) - dbd->db_table[i].db->set_errfile(dbd->db_table[i].db, dbd->db_errlog); if (reindex && i > 0) { LOG(log_info, logtype_cnid, "Truncating CNID index."); @@ -352,12 +715,44 @@ int dbif_open(DBD *dbd, struct db_param *dbp _U_, int reindex) } if (reindex) LOG(log_info, logtype_cnid, "... done."); + + if (reindex) + LOG(log_info, logtype_cnid, "Reindexing name index..."); + + /* + * Upgrading from version 0 to 1 requires adding the name index below which + * must be done by specifying the DB_CREATE flag + */ + uint32_t version = CNID_VERSION; + if (dbd->db_envhome && !reindex) { + if (dbif_getversion(dbd, &version) == -1) + return -1; + } + + if ((ret = dbd->db_table[0].db->associate(dbd->db_table[0].db, + dbd->db_txn, + dbd->db_table[DBIF_IDX_NAME].db, + idxname, + (reindex + || + ((CNID_VERSION == CNID_VERSION_1) && (version == CNID_VERSION_0))) + ? DB_CREATE : 0)) != 0) { + LOG(log_error, logtype_cnid, "Failed to associate name index: %s", db_strerror(ret)); + return -1; + } + if (reindex) + LOG(log_info, logtype_cnid, "... done."); + + if ((dbd->db_envhome) && ((ret = dbif_upgrade(dbd)) != 0)) { + LOG(log_error, logtype_cnid, "Error upgrading CNID database to version %d", CNID_VERSION); + return -1; + } return 0; } /* ------------------------ */ -int dbif_closedb(DBD *dbd) +static int dbif_closedb(DBD *dbd) { int i; int ret; @@ -401,6 +796,75 @@ int dbif_close(DBD *dbd) return 0; } +/* + In order to support silent database upgrades: + destroy env at cnid_dbd shutdown. + */ +int dbif_env_remove(const char *path) +{ + int ret; + DBD *dbd; + + LOG(log_debug, logtype_cnid, "Reopening BerkeleyDB environment"); + + if (NULL == (dbd = dbif_init(path, "cnid2.db"))) + return -1; + + /* Get db_env handle */ + if ((ret = db_env_create(&dbd->db_env, 0))) { + LOG(log_error, logtype_cnid, "error creating DB environment: %s", db_strerror(ret)); + dbd->db_env = NULL; + return -1; + } + + if ((dbif_openlog(dbd)) != 0) + return -1; + + /* Open environment with recovery */ + if ((ret = dbd->db_env->open(dbd->db_env, + dbd->db_envhome, + DB_CREATE | DB_INIT_LOG | DB_INIT_MPOOL | DB_INIT_LOCK | DB_INIT_TXN | DB_RECOVER | DB_PRIVATE, + 0))) { + LOG(log_error, logtype_cnid, "error opening DB environment: %s", + db_strerror(ret)); + dbd->db_env->close(dbd->db_env, 0); + dbd->db_env = NULL; + return -1; + } + + if (dbd->db_errlog != NULL) + fflush(dbd->db_errlog); + + /* Remove logfiles */ + if ((ret = dbd->db_env->log_archive(dbd->db_env, NULL, DB_ARCH_REMOVE))) { + LOG(log_error, logtype_cnid, "error removing transaction logfiles: %s", db_strerror(ret)); + return -1; + } + + if ((ret = dbd->db_env->close(dbd->db_env, 0))) { + LOG(log_error, logtype_cnid, "error closing DB environment after recovery: %s", db_strerror(ret)); + dbd->db_env = NULL; + return -1; + } + + LOG(log_debug, logtype_cnid, "BerkeleyDB environment recovery done."); + + /* Get a new db_env handle and then remove environment */ + if ((ret = db_env_create(&dbd->db_env, 0))) { + LOG(log_error, logtype_cnid, "error acquiring db_end handle: %s", db_strerror(ret)); + dbd->db_env = NULL; + return -1; + } + if ((ret = dbd->db_env->remove(dbd->db_env, dbd->db_envhome, 0))) { + LOG(log_error, logtype_cnid, "error removing BerkeleyDB environment: %s", db_strerror(ret)); + return -1; + } + + LOG(log_debug, logtype_cnid, "Removed BerkeleyDB environment."); + + return 0; +} + /* * The following three functions are wrappers for DB->get(), DB->put() and DB->del(). * All three return -1 on error. dbif_get()/dbif_del return 1 if the key was found and 0 @@ -425,7 +889,7 @@ int dbif_get(DBD *dbd, const int dbi, DBT *key, DBT *val, u_int32_t flags) return 0; if (ret) { LOG(log_error, logtype_cnid, "error retrieving value from %s: %s", - dbd->db_table[dbi].name, db_strerror(errno)); + dbd->db_table[dbi].name, db_strerror(ret)); return -1; } else return 1; @@ -448,7 +912,7 @@ int dbif_pget(DBD *dbd, const int dbi, DBT *key, DBT *pkey, DBT *val, u_int32_t } if (ret) { LOG(log_error, logtype_cnid, "error retrieving value from %s: %s", - dbd->db_table[dbi].name, db_strerror(errno)); + dbd->db_table[dbi].name, db_strerror(ret)); return -1; } else return 1; @@ -460,8 +924,7 @@ int dbif_put(DBD *dbd, const int dbi, DBT *key, DBT *val, u_int32_t flags) int ret; if (dbif_txn_begin(dbd) < 0) { - LOG(log_error, logtype_cnid, "error setting key/value in %s: %s", - dbd->db_table[dbi].name, db_strerror(errno)); + LOG(log_error, logtype_cnid, "error setting key/value in %s", dbd->db_table[dbi].name); return -1; } @@ -470,13 +933,14 @@ int dbif_put(DBD *dbd, const int dbi, DBT *key, DBT *val, u_int32_t flags) key, val, flags); + if (ret) { if ((flags & DB_NOOVERWRITE) && ret == DB_KEYEXIST) { return 1; } else { LOG(log_error, logtype_cnid, "error setting key/value in %s: %s", - dbd->db_table[dbi].name, db_strerror(errno)); + dbd->db_table[dbi].name, db_strerror(ret)); return -1; } } else @@ -494,8 +958,7 @@ int dbif_del(DBD *dbd, const int dbi, DBT *key, u_int32_t flags) } if (dbif_txn_begin(dbd) < 0) { - LOG(log_error, logtype_cnid, "error deleting key/value from %s: %s", - dbd->db_table[dbi].name, db_strerror(errno)); + LOG(log_error, logtype_cnid, "error deleting key/value from %s", dbd->db_table[dbi].name); return -1; } @@ -504,16 +967,72 @@ int dbif_del(DBD *dbd, const int dbi, DBT *key, u_int32_t flags) key, flags); - if (ret == DB_NOTFOUND || ret == DB_SECONDARY_BAD) + if (ret == DB_NOTFOUND) { + LOG(log_debug, logtype_cnid, "key not found"); return 0; + } if (ret) { LOG(log_error, logtype_cnid, "error deleting key/value from %s: %s", - dbd->db_table[dbi].name, db_strerror(errno)); + dbd->db_table[dbi].name, db_strerror(ret)); return -1; } else return 1; } +/*! + * Search the database by name + * + * @param resbuf (w) buffer for search results CNIDs, maxsize is assumed to be + * DBD_MAX_SRCH_RSLTS * sizefof(cnid_t) + * + * @returns -1 on error, 0 when nothing found, else the number of matches + */ +int dbif_search(DBD *dbd, DBT *key, char *resbuf) +{ + int ret = 0; + int count = 0; + DBC *cursorp = NULL; + DBT pkey, data; + char *cnids = resbuf; + cnid_t cnid; + char *namebkp = key->data; + int namelenbkp = key->size; + + memset(&pkey, 0, sizeof(DBT)); + memset(&data, 0, sizeof(DBT)); + + /* Get a cursor */ + ret = dbd->db_table[DBIF_IDX_NAME].db->cursor(dbd->db_table[DBIF_IDX_NAME].db, + NULL, + &cursorp, + 0); + if (ret != 0) { + LOG(log_error, logtype_cnid, "Couldn't create cursor: %s", db_strerror(ret)); + ret = -1; + goto exit; + } + + ret = cursorp->pget(cursorp, key, &pkey, &data, DB_SET_RANGE); + while (count < DBD_MAX_SRCH_RSLTS && ret != DB_NOTFOUND) { + if (!((namelenbkp <= key->size) && (strncmp(namebkp, key->data, namelenbkp) == 0))) + break; + count++; + memcpy(cnids, pkey.data, sizeof(cnid_t)); + memcpy(&cnid, pkey.data, sizeof(cnid_t)); + cnids += sizeof(cnid_t); + LOG(log_debug, logtype_cnid, "match: CNID %" PRIu32, ntohl(cnid)); + + ret = cursorp->pget(cursorp, key, &pkey, &data, DB_NEXT); + } + + ret = count; + +exit: + if (cursorp != NULL) + cursorp->close(cursorp); + return ret; +} + int dbif_txn_begin(DBD *dbd) { int ret; @@ -529,7 +1048,7 @@ int dbif_txn_begin(DBD *dbd) ret = dbd->db_env->txn_begin(dbd->db_env, NULL, &dbd->db_txn, 0); if (ret) { - LOG(log_error, logtype_cnid, "error starting transaction: %s", db_strerror(errno)); + LOG(log_error, logtype_cnid, "error starting transaction: %s", db_strerror(ret)); return -1; } else return 0; @@ -550,7 +1069,7 @@ int dbif_txn_commit(DBD *dbd) dbd->db_txn = NULL; if (ret) { - LOG(log_error, logtype_cnid, "error committing transaction: %s", db_strerror(errno)); + LOG(log_error, logtype_cnid, "error committing transaction: %s", db_strerror(ret)); return -1; } else return 1; @@ -571,32 +1090,37 @@ int dbif_txn_abort(DBD *dbd) dbd->db_txn = NULL; if (ret) { - LOG(log_error, logtype_cnid, "error aborting transaction: %s", db_strerror(errno)); + LOG(log_error, logtype_cnid, "error aborting transaction: %s", db_strerror(ret)); return -1; } else return 0; } /* - ret = 1 -> commit txn - ret = 0 -> abort txn -> exit! + ret = 1 -> commit txn if db_param.txn_frequency + ret = 0 -> abort txn db_param.txn_frequency -> exit! anything else -> exit! + + @returns 0 on success (abort or commit), -1 on error */ -void dbif_txn_close(DBD *dbd, int ret) +int dbif_txn_close(DBD *dbd, int ret) { if (ret == 0) { if (dbif_txn_abort(dbd) < 0) { LOG( log_error, logtype_cnid, "Fatal error aborting transaction. Exiting!"); - exit(EXIT_FAILURE); + return -1; } } else if (ret == 1) { ret = dbif_txn_commit(dbd); if ( ret < 0) { LOG( log_error, logtype_cnid, "Fatal error committing transaction. Exiting!"); - exit(EXIT_FAILURE); + return -1; } - } else - exit(EXIT_FAILURE); + } else { + return -1; + } + + return 0; } int dbif_txn_checkpoint(DBD *dbd, u_int32_t kbyte, u_int32_t min, u_int32_t flags) @@ -604,7 +1128,7 @@ int dbif_txn_checkpoint(DBD *dbd, u_int32_t kbyte, u_int32_t min, u_int32_t flag int ret; ret = dbd->db_env->txn_checkpoint(dbd->db_env, kbyte, min, flags); if (ret) { - LOG(log_error, logtype_cnid, "error checkpointing transaction susystem: %s", db_strerror(errno)); + LOG(log_error, logtype_cnid, "error checkpointing transaction susystem: %s", db_strerror(ret)); return -1; } else return 0; @@ -619,7 +1143,7 @@ int dbif_count(DBD *dbd, const int dbi, u_int32_t *count) ret = db->stat(db, NULL, &sp, 0); if (ret) { - LOG(log_error, logtype_cnid, "error getting stat infotmation on database: %s", db_strerror(errno)); + LOG(log_error, logtype_cnid, "error getting stat infotmation on database: %s", db_strerror(ret)); return -1; } @@ -660,7 +1184,7 @@ int dbif_copy_rootinfokey(DBD *srcdbd, DBD *destdbd) int dbif_dump(DBD *dbd, int dumpindexes) { int rc; - uint32_t max = 0, count = 0, cnid, type, did, lastid; + uint32_t max = 0, count = 0, cnid, type, did, lastid, version; uint64_t dev, ino; time_t stamp; DBC *cur; @@ -673,7 +1197,7 @@ int dbif_dump(DBD *dbd, int dumpindexes) rc = db->cursor(db, NULL, &cur, 0); if (rc) { - LOG(log_error, logtype_cnid, "Couldn't create cursor: %s", db_strerror(errno)); + LOG(log_error, logtype_cnid, "Couldn't create cursor: %s", db_strerror(rc)); return -1; } @@ -687,23 +1211,27 @@ int dbif_dump(DBD *dbd, int dumpindexes) /* Rootinfo node ? */ if (cnid == 0) { - memcpy(&stamp, data.data + 4, sizeof(time_t)); - memcpy(&lastid, data.data + 20, sizeof(cnid_t)); + memcpy(&stamp, (char *)data.data + CNID_DEV_OFS, sizeof(time_t)); + memcpy(&lastid, (char *)data.data + CNID_TYPE_OFS, CNID_TYPE_LEN); lastid = ntohl(lastid); + memcpy(&version, (char *)data.data + CNID_DID_OFS, CNID_DID_LEN); + version = ntohl(version); + strftime(timebuf, sizeof(timebuf), "%b %d %Y %H:%M:%S", localtime(&stamp)); - printf("dbd stamp: 0x%08x (%s), next free CNID: %u\n", (unsigned int)stamp, timebuf, lastid + 1); + printf("CNID db version: %u, dbd stamp: 0x%08x (%s), next free CNID: %u\n", + version, (unsigned int)stamp, timebuf, lastid + 1); } else { /* dev */ - memcpy(&dev, data.data + CNID_DEV_OFS, 8); + memcpy(&dev, (char *)data.data + CNID_DEV_OFS, 8); dev = ntoh64(dev); /* ino */ - memcpy(&ino, data.data + CNID_INO_OFS, 8); + memcpy(&ino, (char *)data.data + CNID_INO_OFS, 8); ino = ntoh64(ino); /* type */ - memcpy(&type, data.data + CNID_TYPE_OFS, 4); + memcpy(&type, (char *)data.data + CNID_TYPE_OFS, 4); type = ntohl(type); /* did */ - memcpy(&did, data.data + CNID_DID_OFS, 4); + memcpy(&did, (char *)data.data + CNID_DID_OFS, 4); did = ntohl(did); count++; @@ -718,13 +1246,13 @@ int dbif_dump(DBD *dbd, int dumpindexes) } if (rc != DB_NOTFOUND) { - LOG(log_error, logtype_cnid, "Error iterating over btree: %s", db_strerror(errno)); + LOG(log_error, logtype_cnid, "Error iterating over btree: %s", db_strerror(rc)); return -1; } rc = cur->c_close(cur); if (rc) { - LOG(log_error, logtype_cnid, "Couldn't close cursor: %s", db_strerror(errno)); + LOG(log_error, logtype_cnid, "Couldn't close cursor: %s", db_strerror(rc)); return -1; } printf("%u CNIDs in database. Max CNID: %u.\n", count, max); @@ -737,7 +1265,7 @@ int dbif_dump(DBD *dbd, int dumpindexes) db = dbd->db_table[DBIF_IDX_DEVINO].db; rc = db->cursor(db, NULL, &cur, 0); if (rc) { - LOG(log_error, logtype_cnid, "Couldn't create cursor: %s", db_strerror(errno)); + LOG(log_error, logtype_cnid, "Couldn't create cursor: %s", db_strerror(rc)); return -1; } @@ -755,7 +1283,7 @@ int dbif_dump(DBD *dbd, int dumpindexes) memcpy(&dev, key.data, CNID_DEV_LEN); dev = ntoh64(dev); /* ino */ - memcpy(&ino, key.data + CNID_DEV_LEN, CNID_INO_LEN); + memcpy(&ino, (char *)key.data + CNID_DEV_LEN, CNID_INO_LEN); ino = ntoh64(ino); printf("id: %10u <== dev: 0x%llx, ino: 0x%016llx\n", @@ -765,13 +1293,13 @@ int dbif_dump(DBD *dbd, int dumpindexes) rc = cur->c_get(cur, &key, &data, DB_NEXT); } if (rc != DB_NOTFOUND) { - LOG(log_error, logtype_cnid, "Error iterating over btree: %s", db_strerror(errno)); + LOG(log_error, logtype_cnid, "Error iterating over btree: %s", db_strerror(rc)); return -1; } rc = cur->c_close(cur); if (rc) { - LOG(log_error, logtype_cnid, "Couldn't close cursor: %s", db_strerror(errno)); + LOG(log_error, logtype_cnid, "Couldn't close cursor: %s", db_strerror(rc)); return -1; } printf("%u items\n", count); @@ -782,7 +1310,7 @@ int dbif_dump(DBD *dbd, int dumpindexes) db = dbd->db_table[DBIF_IDX_DIDNAME].db; rc = db->cursor(db, NULL, &cur, 0); if (rc) { - LOG(log_error, logtype_cnid, "Couldn't create cursor: %s", db_strerror(errno)); + LOG(log_error, logtype_cnid, "Couldn't create cursor: %s", db_strerror(rc)); return -1; } @@ -806,13 +1334,13 @@ int dbif_dump(DBD *dbd, int dumpindexes) rc = cur->c_get(cur, &key, &data, DB_NEXT); } if (rc != DB_NOTFOUND) { - LOG(log_error, logtype_cnid, "Error iterating over btree: %s", db_strerror(errno)); + LOG(log_error, logtype_cnid, "Error iterating over btree: %s", db_strerror(rc)); return -1; } rc = cur->c_close(cur); if (rc) { - LOG(log_error, logtype_cnid, "Couldn't close cursor: %s", db_strerror(errno)); + LOG(log_error, logtype_cnid, "Couldn't close cursor: %s", db_strerror(rc)); return -1; } printf("%u items\n", count); @@ -836,16 +1364,18 @@ int dbif_idwalk(DBD *dbd, cnid_t *cnid, int close) static DBT key = { 0 }, data = { 0 }; DB *db = dbd->db_table[DBIF_CNID].db; - if (close && dbd->db_cur) { - dbd->db_cur->close(dbd->db_cur); - dbd->db_cur = NULL; + if (close) { + if (dbd->db_cur) { + dbd->db_cur->close(dbd->db_cur); + dbd->db_cur = NULL; + } return 0; } /* An dbif_del will have closed our cursor too */ if ( ! dbd->db_cur ) { - if (db->cursor(db, NULL, &dbd->db_cur, 0) != 0) { - LOG(log_error, logtype_cnid, "Couldn't create cursor: %s", db_strerror(errno)); + if ((rc = db->cursor(db, NULL, &dbd->db_cur, 0)) != 0) { + LOG(log_error, logtype_cnid, "Couldn't create cursor: %s", db_strerror(rc)); return -1; } flag = DB_SET_RANGE; /* This will seek to next cnid after the one just deleted */ @@ -862,7 +1392,7 @@ int dbif_idwalk(DBD *dbd, cnid_t *cnid, int close) } if (rc != DB_NOTFOUND) { - LOG(log_error, logtype_cnid, "Error iterating over btree: %s", db_strerror(errno)); + LOG(log_error, logtype_cnid, "Error iterating over btree: %s", db_strerror(rc)); dbd->db_cur->close(dbd->db_cur); dbd->db_cur = NULL; return -1;