2 * Copyright (C) Joerg Lenneis 2003
3 * Copyright (C) Frank Lahm 2009
4 * All Rights Reserved. See COPYING.
9 #endif /* HAVE_CONFIG_H */
15 #include <sys/types.h>
18 #include <arpa/inet.h>
22 #include <atalk/logger.h>
23 #include <atalk/util.h>
24 #include <atalk/errchk.h>
30 #define DB_ERRLOGFILE "db_errlog"
33 * Get the db stamp which is the st_ctime of the file "cnid2.db" and store it in buffer
35 static int dbif_stamp(DBD *dbd, void *buffer, int size)
45 if ((cwd = open(".", O_RDONLY)) < 0) {
46 LOG(log_error, logtype_cnid, "error opening cwd: %s", strerror(errno));
50 /* chdir to db_envhome */
51 if ((chdir(dbd->db_envhome)) != 0) {
52 LOG(log_error, logtype_cnid, "error chdiring to db_env '%s': %s", dbd->db_envhome, strerror(errno));
56 if (stat(dbd->db_table[DBIF_CNID].name, &st) < 0) {
57 LOG(log_error, logtype_cnid, "error stating database %s: %s", dbd->db_table[DBIF_CNID].name, db_strerror(errno));
61 LOG(log_maxdebug, logtype_cnid,"stamp: %s", asctime(localtime(&st.st_ctime)));
63 memset(buffer, 0, size);
64 memcpy(buffer, &st.st_ctime, sizeof(st.st_ctime));
68 if (fchdir(cwd) != 0) {
69 LOG(log_error, logtype_cnid, "error chdiring back: %s", strerror(errno));
78 * Inititialize rootinfo key (which has CNID 0 as key)
80 * This also "stamps" the database, which means storing st.st_ctime of the
81 * "cnid2.db" file in the rootinfo data at the DEV offset
83 * @param dbd (rw) database handle
84 * @param version (r) database version number
86 * @returns -1 on error, 0 on success
88 static int dbif_init_rootinfo(DBD *dbd, int version)
92 char buf[ROOTINFO_DATALEN];
94 LOG(log_debug, logtype_cnid, "Setting CNID database version to %u", version);
99 memset(&key, 0, sizeof(key));
100 memset(&data, 0, sizeof(data));
101 key.data = ROOTINFO_KEY;
102 key.size = ROOTINFO_KEYLEN;
104 data.size = ROOTINFO_DATALEN;
106 memcpy(buf, ROOTINFO_DATA, ROOTINFO_DATALEN);
107 memcpy(buf + CNID_DID_OFS, &v, sizeof(v));
108 if (dbif_stamp(dbd, buf + CNID_DEV_OFS, CNID_DEV_LEN) < 0)
111 if (dbif_put(dbd, DBIF_CNID, &key, &data, 0) < 0)
113 if (dbif_txn_commit(dbd) != 1) {
114 LOG(log_error, logtype_cnid, "dbif_init_rootinfo: cant commit txn");
122 * Return CNID database version number
124 * Returns version in *version
126 * @returns -1 on error, 0 if theres no rootinfo key yet, 1 if *version is returned
128 static int dbif_getversion(DBD *dbd, uint32_t *version)
133 LOG(log_maxdebug, logtype_cnid, "dbif_getversion: reading version info");
136 memset(&key, 0, sizeof(key));
137 memset(&data, 0, sizeof(data));
138 key.data = ROOTINFO_KEY;
139 key.size = ROOTINFO_KEYLEN;
141 switch (dbif_get(dbd, DBIF_CNID, &key, &data, 0)) {
143 memcpy(version, (char *)data.data + CNID_DID_OFS, sizeof(uint32_t));
144 *version = ntohl(*version);
145 LOG(log_debug, logtype_cnid, "CNID database version %u", *version);
148 case 0: /* not found */
149 LOG(log_debug, logtype_cnid, "dbif_getversion: no version info found");
153 LOG(log_error, logtype_cnid, "dbif_getversion: database error");
162 * Set CNID database version number
164 * Initializes rootinfo key as neccessary
165 * @returns -1 on error, 0 on success
167 static int dbif_setversion(DBD *dbd, uint32_t version)
173 LOG(log_debug, logtype_cnid, "Setting CNID database version to %u", version);
178 memset(&key, 0, sizeof(key));
179 memset(&data, 0, sizeof(data));
180 key.data = ROOTINFO_KEY;
181 key.size = ROOTINFO_KEYLEN;
183 if ((ret = dbif_get(dbd, DBIF_CNID, &key, &data, 0)) == -1)
186 /* No rootinfo key yet, init it */
187 if (dbif_init_rootinfo(dbd, CNID_VERSION) != 0)
190 if (dbif_get(dbd, DBIF_CNID, &key, &data, 0) == -1)
193 memcpy((char *)data.data + CNID_DID_OFS, &v, sizeof(v));
194 data.size = ROOTINFO_DATALEN;
195 if (dbif_put(dbd, DBIF_CNID, &key, &data, 0) < 0)
202 * Upgrade CNID database versions, initialize rootinfo key as as necessary in dbif_setversion()
204 * For now this does nothing, as upgrading from ver. 0 to 1 is done in dbif_open
206 #define UNINTIALIZED_DB UINT32_MAX
207 static int dbif_upgrade(DBD *dbd)
209 uint32_t version = CNID_VERSION_UNINTIALIZED_DB;
211 if (dbif_getversion(dbd, &version) == -1)
213 if (version == CNID_VERSION_UNINTIALIZED_DB) {
214 version = CNID_VERSION;
215 if (dbif_setversion(dbd, CNID_VERSION) != 0)
220 * Do upgrade stuff ...
223 /* Write current version to database */
224 if (version != CNID_VERSION) {
225 if (dbif_setversion(dbd, CNID_VERSION) != 0)
229 LOG(log_debug, logtype_cnid, "Finished CNID database version upgrade check");
234 /* --------------- */
235 static int dbif_openlog(DBD *dbd)
240 if ( ! dbd->db_filename)
245 if ((cwd = open(".", O_RDONLY)) < 0) {
246 LOG(log_error, logtype_cnid, "error opening cwd: %s", strerror(errno));
250 /* chdir to db_envhome */
251 if ((chdir(dbd->db_envhome)) != 0) {
252 LOG(log_error, logtype_cnid, "error chdiring to db_env '%s': %s", dbd->db_envhome, strerror(errno));
257 if ((dbd->db_errlog = fopen(DB_ERRLOGFILE, "a")) == NULL)
258 LOG(log_warning, logtype_cnid, "error creating/opening DB errlogfile: %s", strerror(errno));
260 if (dbd->db_errlog != NULL) {
261 dbd->db_env->set_errfile(dbd->db_env, dbd->db_errlog);
262 dbd->db_env->set_msgfile(dbd->db_env, dbd->db_errlog);
267 if ((fchdir(cwd)) != 0) {
268 LOG(log_error, logtype_cnid, "error chdiring back: %s", strerror(errno));
276 /* --------------- */
277 static int dbif_logautorem(DBD *dbd)
281 char **logfiles = NULL;
284 if ( ! dbd->db_filename)
289 if ((cwd = open(".", O_RDONLY)) < 0) {
290 LOG(log_error, logtype_cnid, "error opening cwd: %s", strerror(errno));
294 /* chdir to db_envhome */
295 if ((chdir(dbd->db_envhome)) != 0) {
296 LOG(log_error, logtype_cnid, "error chdiring to db_env '%s': %s", dbd->db_envhome, strerror(errno));
301 if ((ret = dbd->db_env->log_archive(dbd->db_env, &logfiles, 0)) != 0) {
302 LOG(log_error, logtype_cnid, "error getting list of stale logfiles: %s",
304 dbd->db_env->close(dbd->db_env, 0);
310 if (logfiles != NULL) {
311 for (file = logfiles; *file != NULL; file++) {
312 if (unlink(*file) < 0)
313 LOG(log_warning, logtype_cnid, "Error removing stale logfile %s: %s", *file, strerror(errno));
320 if ((fchdir(cwd)) != 0) {
321 LOG(log_error, logtype_cnid, "error chdiring back: %s", strerror(errno));
330 * Get lock on db lock file
332 * @args cmd (r) lock command:
333 * LOCK_FREE: close lockfd
334 * LOCK_UNLOCK: unlock lockm keep lockfd open
335 * LOCK_EXCL: F_WRLCK on lockfd
336 * LOCK_SHRD: F_RDLCK on lockfd
337 * @args dbpath (r) path to lockfile, only used on first call,
338 * later the stored fd is used
339 * @returns LOCK_FREE/LOCK_UNLOCK return 0 on success, -1 on error
340 * LOCK_EXCL/LOCK_SHRD return LOCK_EXCL or LOCK_SHRD respectively on
341 * success, 0 if the lock couldn't be acquired, -1 on other errors
343 int get_lock(int cmd, const char *dbpath)
345 static int lockfd = -1;
347 char lockpath[PATH_MAX];
350 LOG(log_debug, logtype_cnid, "get_lock(%s, \"%s\")",
351 cmd == LOCK_EXCL ? "LOCK_EXCL" :
352 cmd == LOCK_SHRD ? "LOCK_SHRD" :
353 cmd == LOCK_FREE ? "LOCK_FREE" :
354 cmd == LOCK_UNLOCK ? "LOCK_UNLOCK" : "UNKNOWN" , dbpath ? dbpath : "");
367 return unlock(lockfd, 0, SEEK_SET, 0);
372 if ( (strlen(dbpath) + strlen(LOCKFILENAME+1)) > (PATH_MAX - 1) ) {
373 LOG(log_error, logtype_cnid, ".AppleDB pathname too long");
376 strncpy(lockpath, dbpath, PATH_MAX - 1);
377 strcat(lockpath, "/");
378 strcat(lockpath, LOCKFILENAME);
380 if ((lockfd = open(lockpath, O_RDWR | O_CREAT, 0644)) < 0) {
381 LOG(log_error, logtype_cnid, "Error opening lockfile: %s", strerror(errno));
385 if ((stat(dbpath, &st)) != 0) {
386 LOG(log_error, logtype_cnid, "Error statting lockfile: %s", strerror(errno));
390 if ((chown(lockpath, st.st_uid, st.st_gid)) != 0) {
391 LOG(log_error, logtype_cnid, "Error inheriting lockfile permissions: %s",
397 if (cmd == LOCK_EXCL)
398 ret = write_lock(lockfd, 0, SEEK_SET, 0);
400 ret = read_lock(lockfd, 0, SEEK_SET, 0);
403 if (cmd == LOCK_SHRD)
404 LOG(log_error, logtype_cnid, "Volume CNID db is locked, try again...");
408 LOG(log_debug, logtype_cnid, "get_lock: got %s lock",
409 cmd == LOCK_EXCL ? "LOCK_EXCL" : "LOCK_SHRD");
416 /* deadc0de, never get here */
420 /* --------------- */
421 DBD *dbif_init(const char *envhome, const char *filename)
425 if ( NULL == (dbd = calloc(sizeof(DBD), 1)) )
428 /* filename == NULL means in memory db */
433 dbd->db_envhome = strdup(envhome);
434 if (NULL == dbd->db_envhome) {
439 dbd->db_filename = strdup(filename);
440 if (NULL == dbd->db_filename) {
441 free(dbd->db_envhome);
447 dbd->db_table[DBIF_CNID].name = "cnid2.db";
448 dbd->db_table[DBIF_IDX_DEVINO].name = "devino.db";
449 dbd->db_table[DBIF_IDX_DIDNAME].name = "didname.db";
450 dbd->db_table[DBIF_IDX_NAME].name = "name.db";
452 dbd->db_table[DBIF_CNID].type = DB_BTREE;
453 dbd->db_table[DBIF_IDX_DEVINO].type = DB_BTREE;
454 dbd->db_table[DBIF_IDX_DIDNAME].type = DB_BTREE;
455 dbd->db_table[DBIF_IDX_NAME].type = DB_BTREE;
457 dbd->db_table[DBIF_CNID].openflags = DB_CREATE;
458 dbd->db_table[DBIF_IDX_DEVINO].openflags = DB_CREATE;
459 dbd->db_table[DBIF_IDX_DIDNAME].openflags = DB_CREATE;
460 dbd->db_table[DBIF_IDX_NAME].openflags = DB_CREATE;
462 dbd->db_table[DBIF_IDX_NAME].flags = DB_DUPSORT;
468 We must open the db_env with an absolute pathname, as `dbd` keeps chdir'ing, which
469 breaks e.g. bdb logfile-rotation with relative pathnames.
470 But still we use relative paths with DB_ERRLOGFILE
471 in order to avoid creating absolute paths by copying. Both have no problem with
474 int dbif_env_open(DBD *dbd, struct db_param *dbp, uint32_t dbenv_oflags)
478 if ((ret = db_env_create(&dbd->db_env, 0))) {
479 LOG(log_error, logtype_cnid, "error creating DB environment: %s",
485 dbd->db_param = *dbp;
487 if ((dbif_openlog(dbd)) != 0)
490 if (dbenv_oflags & DB_RECOVER) {
492 LOG(log_debug, logtype_cnid, "Running recovery");
494 dbd->db_env->set_verbose(dbd->db_env, DB_VERB_RECOVERY, 1);
495 /* Open the database for recovery using DB_PRIVATE option which is faster */
496 if ((ret = dbd->db_env->open(dbd->db_env, dbd->db_envhome, dbenv_oflags | DB_PRIVATE, 0))) {
497 LOG(log_error, logtype_cnid, "error opening DB environment: %s",
499 dbd->db_env->close(dbd->db_env, 0);
503 dbenv_oflags = (dbenv_oflags & ~DB_RECOVER);
505 if (dbd->db_errlog != NULL)
506 fflush(dbd->db_errlog);
508 if ((ret = dbd->db_env->close(dbd->db_env, 0))) {
509 LOG(log_error, logtype_cnid, "error closing DB environment after recovery: %s",
514 dbd->db_errlog = NULL;
516 if ((ret = db_env_create(&dbd->db_env, 0))) {
517 LOG(log_error, logtype_cnid, "error creating DB environment after recovery: %s",
523 if ((dbif_openlog(dbd)) != 0)
526 LOG(log_debug, logtype_cnid, "Finished recovery.");
529 if ((ret = dbd->db_env->set_cachesize(dbd->db_env, 0, 1024 * dbp->cachesize, 0))) {
530 LOG(log_error, logtype_cnid, "error setting DB environment cachesize to %i: %s",
531 dbp->cachesize, db_strerror(ret));
532 dbd->db_env->close(dbd->db_env, 0);
537 if ((ret = dbd->db_env->set_lk_max_locks(dbd->db_env, dbp->maxlocks))) {
538 LOG(log_error, logtype_cnid, "error setting DB environment maxlocks to %i: %s",
539 10000, db_strerror(ret));
540 dbd->db_env->close(dbd->db_env, 0);
545 if ((ret = dbd->db_env->set_lk_max_objects(dbd->db_env, dbp->maxlockobjs))) {
546 LOG(log_error, logtype_cnid, "error setting DB environment max lockobjects to %i: %s",
547 10000, db_strerror(ret));
548 dbd->db_env->close(dbd->db_env, 0);
553 if ((ret = dbd->db_env->open(dbd->db_env, dbd->db_envhome, dbenv_oflags, 0))) {
554 LOG(log_error, logtype_cnid, "error opening DB environment after recovery: %s",
556 dbd->db_env->close(dbd->db_env, 0);
561 if ((ret = dbd->db_env->set_flags(dbd->db_env, DB_AUTO_COMMIT, 1))) {
562 LOG(log_error, logtype_cnid, "error setting DB_AUTO_COMMIT flag: %s",
564 dbd->db_env->close(dbd->db_env, 0);
569 if (dbp->logfile_autoremove) {
570 if ((dbif_logautorem(dbd)) != 0)
573 #if DB_VERSION_MAJOR > 4 || (DB_VERSION_MAJOR == 4 && DB_VERSION_MINOR >= 7)
574 if ((ret = dbd->db_env->log_set_config(dbd->db_env, DB_LOG_AUTO_REMOVE, 1))) {
575 LOG(log_error, logtype_cnid, "error setting DB_LOG_AUTO_REMOVE flag: %s",
577 dbd->db_env->close(dbd->db_env, 0);
582 if ((ret = dbd->db_env->set_flags(dbd->db_env, DB_LOG_AUTOREMOVE, 1))) {
583 LOG(log_error, logtype_cnid, "error setting DB_LOG_AUTOREMOVE flag: %s",
585 dbd->db_env->close(dbd->db_env, 0);
595 /* --------------- */
596 int dbif_open(DBD *dbd, struct db_param *dbp, int reindex)
603 /* Try to upgrade if it's a normal on-disk database */
604 if (dbd->db_envhome) {
606 if ((cwd = open(".", O_RDONLY)) < 0) {
607 LOG(log_error, logtype_cnid, "error opening cwd: %s", strerror(errno));
611 /* chdir to db_envhome. makes it easier checking for old db files and creating db_errlog file */
612 if ((chdir(dbd->db_envhome)) != 0) {
613 LOG(log_error, logtype_cnid, "error chdiring to db_env '%s': %s", dbd->db_envhome, strerror(errno));
617 if ((stat(dbd->db_filename, &st)) == 0) {
618 LOG(log_debug, logtype_cnid, "See if we can upgrade the CNID database...");
619 if ((ret = db_create(&upgrade_db, dbd->db_env, 0))) {
620 LOG(log_error, logtype_cnid, "error creating handle for database: %s", db_strerror(ret));
623 if ((ret = upgrade_db->upgrade(upgrade_db, dbd->db_filename, 0))) {
624 LOG(log_error, logtype_cnid, "error upgarding database: %s", db_strerror(ret));
627 if ((ret = upgrade_db->close(upgrade_db, 0))) {
628 LOG(log_error, logtype_cnid, "error closing database: %s", db_strerror(ret));
631 if ((ret = dbd->db_env->txn_checkpoint(dbd->db_env, 0, 0, DB_FORCE))) {
632 LOG(log_error, logtype_cnid, "error forcing checkpoint: %s", db_strerror(ret));
635 LOG(log_debug, logtype_cnid, "Finished BerkeleyBD upgrade check");
638 if ((fchdir(cwd)) != 0) {
639 LOG(log_error, logtype_cnid, "error chdiring back: %s", strerror(errno));
644 /* Now open databases ... */
645 for (i = 0; i != DBIF_DB_CNT; i++) {
646 if ((ret = db_create(&dbd->db_table[i].db, dbd->db_env, 0))) {
647 LOG(log_error, logtype_cnid, "error creating handle for database %s: %s",
648 dbd->db_table[i].name, db_strerror(ret));
652 if (dbd->db_table[i].flags) {
653 if ((ret = dbd->db_table[i].db->set_flags(dbd->db_table[i].db,
654 dbd->db_table[i].flags))) {
655 LOG(log_error, logtype_cnid, "error setting flags for database %s: %s",
656 dbd->db_table[i].name, db_strerror(ret));
661 if ( ! dbd->db_env) { /* In memory db */
662 if ((ret = dbd->db_table[i].db->set_cachesize(dbd->db_table[i].db,
665 4)) /* split in 4 memory chunks */
667 LOG(log_error, logtype_cnid, "error setting cachesize %u KB for database %s: %s",
668 dbp->cachesize / 1024, dbd->db_table[i].name, db_strerror(ret));
673 if (dbd->db_table[i].db->open(dbd->db_table[i].db,
676 dbd->db_table[i].name,
677 dbd->db_table[i].type,
678 dbd->db_table[i].openflags,
680 LOG(log_error, logtype_cnid, "Cant open database");
684 if (reindex && i > 0) {
685 LOG(log_info, logtype_cnid, "Truncating CNID index.");
686 if ((ret = dbd->db_table[i].db->truncate(dbd->db_table[i].db, NULL, &count, 0))) {
687 LOG(log_error, logtype_cnid, "error truncating database %s: %s",
688 dbd->db_table[i].name, db_strerror(ret));
694 /* TODO: Implement CNID DB versioning info on new databases. */
696 /* Associate the secondary with the primary. */
698 LOG(log_info, logtype_cnid, "Reindexing did/name index...");
699 if ((ret = dbd->db_table[0].db->associate(dbd->db_table[DBIF_CNID].db,
701 dbd->db_table[DBIF_IDX_DIDNAME].db,
703 (reindex) ? DB_CREATE : 0))
705 LOG(log_error, logtype_cnid, "Failed to associate didname database: %s",db_strerror(ret));
709 LOG(log_info, logtype_cnid, "... done.");
712 LOG(log_info, logtype_cnid, "Reindexing dev/ino index...");
713 if ((ret = dbd->db_table[0].db->associate(dbd->db_table[0].db,
715 dbd->db_table[DBIF_IDX_DEVINO].db,
717 (reindex) ? DB_CREATE : 0))
719 LOG(log_error, logtype_cnid, "Failed to associate devino database: %s",db_strerror(ret));
723 LOG(log_info, logtype_cnid, "... done.");
726 LOG(log_info, logtype_cnid, "Reindexing name index...");
729 * Upgrading from version 0 to 1 requires adding the name index below which
730 * must be done by specifying the DB_CREATE flag
732 uint32_t version = CNID_VERSION;
733 if (dbd->db_envhome && !reindex) {
734 if (dbif_getversion(dbd, &version) == -1)
738 if ((ret = dbd->db_table[0].db->associate(dbd->db_table[0].db,
740 dbd->db_table[DBIF_IDX_NAME].db,
744 ((CNID_VERSION == CNID_VERSION_1) && (version == CNID_VERSION_0)))
745 ? DB_CREATE : 0)) != 0) {
746 LOG(log_error, logtype_cnid, "Failed to associate name index: %s", db_strerror(ret));
750 LOG(log_info, logtype_cnid, "... done.");
752 if ((dbd->db_envhome) && ((ret = dbif_upgrade(dbd)) != 0)) {
753 LOG(log_error, logtype_cnid, "Error upgrading CNID database to version %d", CNID_VERSION);
760 /* ------------------------ */
761 static int dbif_closedb(DBD *dbd)
767 for (i = DBIF_DB_CNT -1; i >= 0; i--) {
768 if (dbd->db_table[i].db != NULL && (ret = dbd->db_table[i].db->close(dbd->db_table[i].db, 0))) {
769 LOG(log_error, logtype_cnid, "error closing database %s: %s", dbd->db_table[i].name, db_strerror(ret));
778 /* ------------------------ */
779 int dbif_close(DBD *dbd)
784 if (dbif_closedb(dbd))
787 if (dbd->db_env != NULL && (ret = dbd->db_env->close(dbd->db_env, 0))) {
788 LOG(log_error, logtype_cnid, "error closing DB environment: %s", db_strerror(ret));
791 if (dbd->db_errlog != NULL && fclose(dbd->db_errlog) == EOF) {
792 LOG(log_error, logtype_cnid, "error closing DB logfile: %s", strerror(errno));
796 free(dbd->db_filename);
806 In order to support silent database upgrades:
807 destroy env at cnid_dbd shutdown.
809 int dbif_env_remove(const char *path)
814 LOG(log_debug, logtype_cnid, "Trying to remove BerkeleyDB environment");
816 if (get_lock(LOCK_EXCL, path) != LOCK_EXCL) {
817 LOG(log_debug, logtype_cnid, "CNID db \"%s\" in use, not removing BerkeleyDB environment", path);
821 if (NULL == (dbd = dbif_init(path, "cnid2.db")))
824 /* Get db_env handle */
825 if ((ret = db_env_create(&dbd->db_env, 0))) {
826 LOG(log_error, logtype_cnid, "error creating DB environment: %s", db_strerror(ret));
831 if ((dbif_openlog(dbd)) != 0)
834 /* Open environment with recovery */
835 if ((ret = dbd->db_env->open(dbd->db_env,
837 DB_CREATE | DB_INIT_LOG | DB_INIT_MPOOL | DB_INIT_LOCK | DB_INIT_TXN | DB_RECOVER | DB_PRIVATE,
839 LOG(log_error, logtype_cnid, "error opening DB environment: %s",
841 dbd->db_env->close(dbd->db_env, 0);
846 if (dbd->db_errlog != NULL)
847 fflush(dbd->db_errlog);
849 /* Remove logfiles */
850 if ((ret = dbd->db_env->log_archive(dbd->db_env, NULL, DB_ARCH_REMOVE))) {
851 LOG(log_error, logtype_cnid, "error removing transaction logfiles: %s", db_strerror(ret));
855 if ((ret = dbd->db_env->close(dbd->db_env, 0))) {
856 LOG(log_error, logtype_cnid, "error closing DB environment after recovery: %s", db_strerror(ret));
861 LOG(log_debug, logtype_cnid, "BerkeleyDB environment recovery done.");
863 /* Get a new db_env handle and then remove environment */
864 if ((ret = db_env_create(&dbd->db_env, 0))) {
865 LOG(log_error, logtype_cnid, "error acquiring db_end handle: %s", db_strerror(ret));
869 if ((ret = dbd->db_env->remove(dbd->db_env, dbd->db_envhome, 0))) {
870 LOG(log_error, logtype_cnid, "error removing BerkeleyDB environment: %s", db_strerror(ret));
874 LOG(log_debug, logtype_cnid, "Removed BerkeleyDB environment.");
880 * The following three functions are wrappers for DB->get(), DB->put() and DB->del().
881 * All three return -1 on error. dbif_get()/dbif_del return 1 if the key was found and 0
882 * otherwise. dbif_put() returns 0 if key/val was successfully updated and 1 if
883 * the DB_NOOVERWRITE flag was specified and the key already exists.
885 * All return codes other than DB_NOTFOUND and DB_KEYEXIST from the DB->()
886 * functions are not expected and therefore error conditions.
889 int dbif_get(DBD *dbd, const int dbi, DBT *key, DBT *val, u_int32_t flags)
893 ret = dbd->db_table[dbi].db->get(dbd->db_table[dbi].db,
899 if (ret == DB_NOTFOUND)
902 LOG(log_error, logtype_cnid, "error retrieving value from %s: %s",
903 dbd->db_table[dbi].name, db_strerror(ret));
909 /* search by secondary return primary */
910 int dbif_pget(DBD *dbd, const int dbi, DBT *key, DBT *pkey, DBT *val, u_int32_t flags)
914 ret = dbd->db_table[dbi].db->pget(dbd->db_table[dbi].db,
921 if (ret == DB_NOTFOUND || ret == DB_SECONDARY_BAD) {
925 LOG(log_error, logtype_cnid, "error retrieving value from %s: %s",
926 dbd->db_table[dbi].name, db_strerror(ret));
932 /* -------------------------- */
933 int dbif_put(DBD *dbd, const int dbi, DBT *key, DBT *val, u_int32_t flags)
937 if (dbif_txn_begin(dbd) < 0) {
938 LOG(log_error, logtype_cnid, "error setting key/value in %s", dbd->db_table[dbi].name);
942 ret = dbd->db_table[dbi].db->put(dbd->db_table[dbi].db,
950 if ((flags & DB_NOOVERWRITE) && ret == DB_KEYEXIST) {
953 LOG(log_error, logtype_cnid, "error setting key/value in %s: %s",
954 dbd->db_table[dbi].name, db_strerror(ret));
961 int dbif_del(DBD *dbd, const int dbi, DBT *key, u_int32_t flags)
965 /* For cooperation with the dbd utility and its usage of a cursor */
967 dbd->db_cur->close(dbd->db_cur);
971 if (dbif_txn_begin(dbd) < 0) {
972 LOG(log_error, logtype_cnid, "error deleting key/value from %s", dbd->db_table[dbi].name);
976 ret = dbd->db_table[dbi].db->del(dbd->db_table[dbi].db,
981 if (ret == DB_NOTFOUND) {
982 LOG(log_debug, logtype_cnid, "key not found");
986 LOG(log_error, logtype_cnid, "error deleting key/value from %s: %s",
987 dbd->db_table[dbi].name, db_strerror(ret));
994 * Search the database by name
996 * @param resbuf (w) buffer for search results CNIDs, maxsize is assumed to be
997 * DBD_MAX_SRCH_RSLTS * sizefof(cnid_t)
999 * @returns -1 on error, 0 when nothing found, else the number of matches
1001 int dbif_search(DBD *dbd, DBT *key, char *resbuf)
1005 DBC *cursorp = NULL;
1007 char *cnids = resbuf;
1009 char *namebkp = key->data;
1010 int namelenbkp = key->size;
1012 memset(&pkey, 0, sizeof(DBT));
1013 memset(&data, 0, sizeof(DBT));
1016 ret = dbd->db_table[DBIF_IDX_NAME].db->cursor(dbd->db_table[DBIF_IDX_NAME].db,
1021 LOG(log_error, logtype_cnid, "Couldn't create cursor: %s", db_strerror(ret));
1026 ret = cursorp->pget(cursorp, key, &pkey, &data, DB_SET_RANGE);
1027 while (count < DBD_MAX_SRCH_RSLTS && ret != DB_NOTFOUND) {
1028 if (!((namelenbkp <= key->size) && (strncmp(namebkp, key->data, namelenbkp) == 0)))
1031 memcpy(cnids, pkey.data, sizeof(cnid_t));
1032 memcpy(&cnid, pkey.data, sizeof(cnid_t));
1033 cnids += sizeof(cnid_t);
1034 LOG(log_debug, logtype_cnid, "match: CNID %" PRIu32, ntohl(cnid));
1036 ret = cursorp->pget(cursorp, key, &pkey, &data, DB_NEXT);
1042 if (cursorp != NULL)
1043 cursorp->close(cursorp);
1047 int dbif_txn_begin(DBD *dbd)
1051 /* If we already have an active txn, just return */
1055 /* If our DBD has no env, just return (-> in memory db) */
1056 if (dbd->db_env == NULL)
1059 ret = dbd->db_env->txn_begin(dbd->db_env, NULL, &dbd->db_txn, 0);
1062 LOG(log_error, logtype_cnid, "error starting transaction: %s", db_strerror(ret));
1068 int dbif_txn_commit(DBD *dbd)
1075 /* If our DBD has no env, just return (-> in memory db) */
1076 if (dbd->db_env == NULL)
1079 ret = dbd->db_txn->commit(dbd->db_txn, 0);
1083 LOG(log_error, logtype_cnid, "error committing transaction: %s", db_strerror(ret));
1089 int dbif_txn_abort(DBD *dbd)
1096 /* If our DBD has no env, just return (-> in memory db) */
1097 if (dbd->db_env == NULL)
1100 ret = dbd->db_txn->abort(dbd->db_txn);
1104 LOG(log_error, logtype_cnid, "error aborting transaction: %s", db_strerror(ret));
1111 ret = 1 -> commit txn if db_param.txn_frequency
1112 ret = 0 -> abort txn db_param.txn_frequency -> exit!
1113 anything else -> exit!
1115 @returns 0 on success (abort or commit), -1 on error
1117 int dbif_txn_close(DBD *dbd, int ret)
1120 if (dbif_txn_abort(dbd) < 0) {
1121 LOG( log_error, logtype_cnid, "Fatal error aborting transaction. Exiting!");
1124 } else if (ret == 1) {
1125 ret = dbif_txn_commit(dbd);
1127 LOG( log_error, logtype_cnid, "Fatal error committing transaction. Exiting!");
1137 int dbif_txn_checkpoint(DBD *dbd, u_int32_t kbyte, u_int32_t min, u_int32_t flags)
1140 ret = dbd->db_env->txn_checkpoint(dbd->db_env, kbyte, min, flags);
1142 LOG(log_error, logtype_cnid, "error checkpointing transaction susystem: %s", db_strerror(ret));
1148 int dbif_count(DBD *dbd, const int dbi, u_int32_t *count)
1152 DB *db = dbd->db_table[dbi].db;
1154 ret = db->stat(db, NULL, &sp, 0);
1157 LOG(log_error, logtype_cnid, "error getting stat infotmation on database: %s", db_strerror(ret));
1161 *count = sp->bt_ndata;
1167 int dbif_copy_rootinfokey(DBD *srcdbd, DBD *destdbd)
1172 memset(&key, 0, sizeof(key));
1173 memset(&data, 0, sizeof(data));
1175 key.data = ROOTINFO_KEY;
1176 key.size = ROOTINFO_KEYLEN;
1178 if ((rc = dbif_get(srcdbd, DBIF_CNID, &key, &data, 0)) <= 0) {
1179 LOG(log_error, logtype_cnid, "dbif_copy_rootinfokey: Error getting rootinfo record");
1183 memset(&key, 0, sizeof(key));
1184 key.data = ROOTINFO_KEY;
1185 key.size = ROOTINFO_KEYLEN;
1187 if ((rc = dbif_put(destdbd, DBIF_CNID, &key, &data, 0))) {
1188 LOG(log_error, logtype_cnid, "dbif_copy_rootinfokey: Error writing rootinfo key");
1195 int dbif_dump(DBD *dbd, int dumpindexes)
1198 uint32_t max = 0, count = 0, cnid, type, did, lastid, version;
1202 DBT key = { 0 }, data = { 0 };
1203 DB *db = dbd->db_table[DBIF_CNID].db;
1204 char *typestring[2] = {"f", "d"};
1207 printf("CNID database dump:\n");
1209 rc = db->cursor(db, NULL, &cur, 0);
1211 LOG(log_error, logtype_cnid, "Couldn't create cursor: %s", db_strerror(rc));
1215 cur->c_get(cur, &key, &data, DB_FIRST);
1217 /* Parse and print data */
1218 memcpy(&cnid, key.data, 4);
1223 /* Rootinfo node ? */
1225 memcpy(&stamp, (char *)data.data + CNID_DEV_OFS, sizeof(time_t));
1226 memcpy(&lastid, (char *)data.data + CNID_TYPE_OFS, CNID_TYPE_LEN);
1227 lastid = ntohl(lastid);
1228 memcpy(&version, (char *)data.data + CNID_DID_OFS, CNID_DID_LEN);
1229 version = ntohl(version);
1231 strftime(timebuf, sizeof(timebuf), "%b %d %Y %H:%M:%S", localtime(&stamp));
1232 printf("CNID db version: %u, dbd stamp: 0x%08x (%s), next free CNID: %u\n",
1233 version, (unsigned int)stamp, timebuf, lastid + 1);
1236 memcpy(&dev, (char *)data.data + CNID_DEV_OFS, 8);
1239 memcpy(&ino, (char *)data.data + CNID_INO_OFS, 8);
1242 memcpy(&type, (char *)data.data + CNID_TYPE_OFS, 4);
1245 memcpy(&did, (char *)data.data + CNID_DID_OFS, 4);
1249 printf("id: %10u, did: %10u, type: %s, dev: 0x%llx, ino: 0x%016llx, name: %s\n",
1250 cnid, did, typestring[type],
1251 (long long unsigned int)dev, (long long unsigned int)ino,
1252 (char *)data.data + CNID_NAME_OFS);
1256 rc = cur->c_get(cur, &key, &data, DB_NEXT);
1259 if (rc != DB_NOTFOUND) {
1260 LOG(log_error, logtype_cnid, "Error iterating over btree: %s", db_strerror(rc));
1264 rc = cur->c_close(cur);
1266 LOG(log_error, logtype_cnid, "Couldn't close cursor: %s", db_strerror(rc));
1269 printf("%u CNIDs in database. Max CNID: %u.\n", count, max);
1271 /* Dump indexes too ? */
1273 /* DBIF_IDX_DEVINO */
1274 printf("\ndev/inode index:\n");
1276 db = dbd->db_table[DBIF_IDX_DEVINO].db;
1277 rc = db->cursor(db, NULL, &cur, 0);
1279 LOG(log_error, logtype_cnid, "Couldn't create cursor: %s", db_strerror(rc));
1283 cur->c_get(cur, &key, &data, DB_FIRST);
1285 /* Parse and print data */
1288 memcpy(&cnid, data.data, CNID_LEN);
1294 memcpy(&dev, key.data, CNID_DEV_LEN);
1297 memcpy(&ino, (char *)key.data + CNID_DEV_LEN, CNID_INO_LEN);
1300 printf("id: %10u <== dev: 0x%llx, ino: 0x%016llx\n",
1301 cnid, (unsigned long long int)dev, (unsigned long long int)ino);
1304 rc = cur->c_get(cur, &key, &data, DB_NEXT);
1306 if (rc != DB_NOTFOUND) {
1307 LOG(log_error, logtype_cnid, "Error iterating over btree: %s", db_strerror(rc));
1311 rc = cur->c_close(cur);
1313 LOG(log_error, logtype_cnid, "Couldn't close cursor: %s", db_strerror(rc));
1316 printf("%u items\n", count);
1318 /* Now dump DBIF_IDX_DIDNAME */
1319 printf("\ndid/name index:\n");
1321 db = dbd->db_table[DBIF_IDX_DIDNAME].db;
1322 rc = db->cursor(db, NULL, &cur, 0);
1324 LOG(log_error, logtype_cnid, "Couldn't create cursor: %s", db_strerror(rc));
1328 cur->c_get(cur, &key, &data, DB_FIRST);
1330 /* Parse and print data */
1333 memcpy(&cnid, data.data, CNID_LEN);
1339 memcpy(&did, key.data, CNID_LEN);
1342 printf("id: %10u <== did: %10u, name: %s\n", cnid, did, (char *)key.data + CNID_LEN);
1345 rc = cur->c_get(cur, &key, &data, DB_NEXT);
1347 if (rc != DB_NOTFOUND) {
1348 LOG(log_error, logtype_cnid, "Error iterating over btree: %s", db_strerror(rc));
1352 rc = cur->c_close(cur);
1354 LOG(log_error, logtype_cnid, "Couldn't close cursor: %s", db_strerror(rc));
1357 printf("%u items\n", count);
1364 Iterates over dbd, returning cnids.
1365 Uses in-value of cnid to seek to that cnid, then gets next and return that in cnid.
1366 If close=1, close cursor.
1367 Return -1 on error, 0 on EOD (end-of-database), 1 if returning cnid.
1369 int dbif_idwalk(DBD *dbd, cnid_t *cnid, int close)
1375 static DBT key = { 0 }, data = { 0 };
1376 DB *db = dbd->db_table[DBIF_CNID].db;
1380 dbd->db_cur->close(dbd->db_cur);
1386 /* An dbif_del will have closed our cursor too */
1387 if ( ! dbd->db_cur ) {
1388 if ((rc = db->cursor(db, NULL, &dbd->db_cur, 0)) != 0) {
1389 LOG(log_error, logtype_cnid, "Couldn't create cursor: %s", db_strerror(rc));
1392 flag = DB_SET_RANGE; /* This will seek to next cnid after the one just deleted */
1395 key.size = sizeof(cnid_t);
1399 if ((rc = dbd->db_cur->get(dbd->db_cur, &key, &data, flag)) == 0) {
1400 memcpy(cnid, key.data, sizeof(cnid_t));
1401 *cnid = ntohl(*cnid);
1405 if (rc != DB_NOTFOUND) {
1406 LOG(log_error, logtype_cnid, "Error iterating over btree: %s", db_strerror(rc));
1407 dbd->db_cur->close(dbd->db_cur);
1413 dbd->db_cur->close(dbd->db_cur);