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>
21 #include <atalk/logger.h>
22 #include <atalk/util.h>
28 #define DB_ERRLOGFILE "db_errlog"
31 * Get the db stamp which is the st_ctime of the file "cnid2.db" and store it in buffer
33 static int dbif_stamp(DBD *dbd, void *buffer, int size)
42 if ((cwd = open(".", O_RDONLY)) < 0) {
43 LOG(log_error, logtype_cnid, "error opening cwd: %s", strerror(errno));
47 /* chdir to db_envhome */
48 if ((chdir(dbd->db_envhome)) != 0) {
49 LOG(log_error, logtype_cnid, "error chdiring to db_env '%s': %s", dbd->db_envhome, strerror(errno));
53 if ((rc = stat(dbd->db_table[DBIF_CNID].name, &st)) < 0) {
54 LOG(log_error, logtype_cnid, "error stating database %s: %s", dbd->db_table[DBIF_CNID].name, db_strerror(errno));
58 LOG(log_maxdebug, logtype_cnid,"stamp: %s", asctime(localtime(&st.st_ctime)));
60 memset(buffer, 0, size);
61 memcpy(buffer, &st.st_ctime, sizeof(st.st_ctime));
63 if ((fchdir(cwd)) != 0) {
64 LOG(log_error, logtype_cnid, "error chdiring back: %s", strerror(errno));
72 * Inititialize rootinfo key (which has CNID 0 as key)
74 * This also "stamps" the database, which means storing st.st_ctime of the
75 * "cnid2.db" file in the rootinfo data at the DEV offset
77 * @param dbd (rw) database handle
78 * @param version (r) database version number
80 * @returns -1 on error, 0 on success
82 static int dbif_init_rootinfo(DBD *dbd, int version)
86 char buf[ROOTINFO_DATALEN];
88 LOG(log_debug, logtype_cnid, "Setting CNID database version to %u", version);
93 memset(&key, 0, sizeof(key));
94 memset(&data, 0, sizeof(data));
95 key.data = ROOTINFO_KEY;
96 key.size = ROOTINFO_KEYLEN;
98 data.size = ROOTINFO_DATALEN;
100 memcpy(buf, ROOTINFO_DATA, ROOTINFO_DATALEN);
101 memcpy(buf + CNID_DID_OFS, &v, sizeof(v));
102 if (dbif_stamp(dbd, buf + CNID_DEV_OFS, CNID_DEV_LEN) < 0)
105 if (dbif_put(dbd, DBIF_CNID, &key, &data, 0) < 0)
112 * Return CNID database version number
114 * Returns version in *version
116 * @returns -1 on error, 0 if theres no rootinfo key yet, 1 if *version is returned
118 static int dbif_getversion(DBD *dbd, uint32_t *version)
123 memset(&key, 0, sizeof(key));
124 memset(&data, 0, sizeof(data));
125 key.data = ROOTINFO_KEY;
126 key.size = ROOTINFO_KEYLEN;
128 switch (dbif_get(dbd, DBIF_CNID, &key, &data, 0)) {
130 memcpy(version, (char *)data.data + CNID_DID_OFS, sizeof(uint32_t));
131 *version = ntohl(*version);
132 LOG(log_debug, logtype_cnid, "CNID database version %u", *version);
135 case 0: /* not found */
147 * Set CNID database version number
149 * Initializes rootinfo key as neccessary, as does dbif_getversion
150 * @returns -1 on error, 0 on success
152 static int dbif_setversion(DBD *dbd, uint32_t version)
158 LOG(log_debug, logtype_cnid, "Setting CNID database version to %u", version);
163 memset(&key, 0, sizeof(key));
164 memset(&data, 0, sizeof(data));
165 key.data = ROOTINFO_KEY;
166 key.size = ROOTINFO_KEYLEN;
168 if ((ret = dbif_get(dbd, DBIF_CNID, &key, &data, 0)) == -1)
171 /* No rootinfo key yet, init it */
172 if (dbif_init_rootinfo(dbd, CNID_VERSION) != 0)
175 if (dbif_get(dbd, DBIF_CNID, &key, &data, 0) == -1)
178 memcpy((char *)data.data + CNID_DID_OFS, &v, sizeof(v));
179 data.size = ROOTINFO_DATALEN;
180 if (dbif_put(dbd, DBIF_CNID, &key, &data, 0) < 0)
187 * Upgrade CNID database versions
189 * For now this does nothing, as upgrading from ver. 0 to 1 is done in dbif_open
191 static int dbif_upgrade(DBD *dbd)
195 if (dbif_getversion(dbd, &version) == -1)
199 * Do upgrade stuff ...
202 /* Write current version to database */
203 if (version != CNID_VERSION) {
204 if (dbif_setversion(dbd, CNID_VERSION) != 0)
208 LOG(log_debug, logtype_cnid, "Finished CNID database version upgrade check");
213 /* --------------- */
214 static int dbif_openlog(DBD *dbd)
219 if ( ! dbd->db_filename)
224 if ((cwd = open(".", O_RDONLY)) < 0) {
225 LOG(log_error, logtype_cnid, "error opening cwd: %s", strerror(errno));
229 /* chdir to db_envhome */
230 if ((chdir(dbd->db_envhome)) != 0) {
231 LOG(log_error, logtype_cnid, "error chdiring to db_env '%s': %s", dbd->db_envhome, strerror(errno));
236 if ((dbd->db_errlog = fopen(DB_ERRLOGFILE, "a")) == NULL)
237 LOG(log_warning, logtype_cnid, "error creating/opening DB errlogfile: %s", strerror(errno));
239 if (dbd->db_errlog != NULL) {
240 dbd->db_env->set_errfile(dbd->db_env, dbd->db_errlog);
241 dbd->db_env->set_msgfile(dbd->db_env, dbd->db_errlog);
246 if ((fchdir(cwd)) != 0) {
247 LOG(log_error, logtype_cnid, "error chdiring back: %s", strerror(errno));
255 /* --------------- */
256 static int dbif_logautorem(DBD *dbd)
260 char **logfiles = NULL;
263 if ( ! dbd->db_filename)
268 if ((cwd = open(".", O_RDONLY)) < 0) {
269 LOG(log_error, logtype_cnid, "error opening cwd: %s", strerror(errno));
273 /* chdir to db_envhome */
274 if ((chdir(dbd->db_envhome)) != 0) {
275 LOG(log_error, logtype_cnid, "error chdiring to db_env '%s': %s", dbd->db_envhome, strerror(errno));
280 if ((ret = dbd->db_env->log_archive(dbd->db_env, &logfiles, 0)) != 0) {
281 LOG(log_error, logtype_cnid, "error getting list of stale logfiles: %s",
283 dbd->db_env->close(dbd->db_env, 0);
289 if (logfiles != NULL) {
290 for (file = logfiles; *file != NULL; file++) {
291 if (unlink(*file) < 0)
292 LOG(log_warning, logtype_cnid, "Error removing stale logfile %s: %s", *file, strerror(errno));
299 if ((fchdir(cwd)) != 0) {
300 LOG(log_error, logtype_cnid, "error chdiring back: %s", strerror(errno));
308 /* --------------- */
309 DBD *dbif_init(const char *envhome, const char *filename)
313 if ( NULL == (dbd = calloc(sizeof(DBD), 1)) )
316 /* filename == NULL means in memory db */
321 dbd->db_envhome = strdup(envhome);
322 if (NULL == dbd->db_envhome) {
327 dbd->db_filename = strdup(filename);
328 if (NULL == dbd->db_filename) {
329 free(dbd->db_envhome);
335 dbd->db_table[DBIF_CNID].name = "cnid2.db";
336 dbd->db_table[DBIF_IDX_DEVINO].name = "devino.db";
337 dbd->db_table[DBIF_IDX_DIDNAME].name = "didname.db";
338 dbd->db_table[DBIF_IDX_NAME].name = "name.db";
340 dbd->db_table[DBIF_CNID].type = DB_BTREE;
341 dbd->db_table[DBIF_IDX_DEVINO].type = DB_BTREE;
342 dbd->db_table[DBIF_IDX_DIDNAME].type = DB_BTREE;
343 dbd->db_table[DBIF_IDX_NAME].type = DB_BTREE;
345 dbd->db_table[DBIF_CNID].openflags = DB_CREATE;
346 dbd->db_table[DBIF_IDX_DEVINO].openflags = DB_CREATE;
347 dbd->db_table[DBIF_IDX_DIDNAME].openflags = DB_CREATE;
348 dbd->db_table[DBIF_IDX_NAME].openflags = DB_CREATE;
350 dbd->db_table[DBIF_IDX_NAME].flags = DB_DUPSORT;
356 We must open the db_env with an absolute pathname, as `dbd` keeps chdir'ing, which
357 breaks e.g. bdb logfile-rotation with relative pathnames.
358 But still we use relative paths with DB_ERRLOGFILE
359 in order to avoid creating absolute paths by copying. Both have no problem with
362 int dbif_env_open(DBD *dbd, struct db_param *dbp, uint32_t dbenv_oflags)
366 if ((ret = db_env_create(&dbd->db_env, 0))) {
367 LOG(log_error, logtype_cnid, "error creating DB environment: %s",
373 if ((dbif_openlog(dbd)) != 0)
376 if (dbenv_oflags & DB_RECOVER) {
378 LOG(log_debug, logtype_cnid, "Running recovery");
380 dbd->db_env->set_verbose(dbd->db_env, DB_VERB_RECOVERY, 1);
381 /* Open the database for recovery using DB_PRIVATE option which is faster */
382 if ((ret = dbd->db_env->open(dbd->db_env, dbd->db_envhome, dbenv_oflags | DB_PRIVATE, 0))) {
383 LOG(log_error, logtype_cnid, "error opening DB environment: %s",
385 dbd->db_env->close(dbd->db_env, 0);
389 dbenv_oflags = (dbenv_oflags & ~DB_RECOVER);
391 if (dbd->db_errlog != NULL)
392 fflush(dbd->db_errlog);
394 if ((ret = dbd->db_env->close(dbd->db_env, 0))) {
395 LOG(log_error, logtype_cnid, "error closing DB environment after recovery: %s",
400 dbd->db_errlog = NULL;
402 if ((ret = db_env_create(&dbd->db_env, 0))) {
403 LOG(log_error, logtype_cnid, "error creating DB environment after recovery: %s",
409 if ((dbif_openlog(dbd)) != 0)
412 LOG(log_debug, logtype_cnid, "Finished recovery.");
415 if ((ret = dbd->db_env->set_cachesize(dbd->db_env, 0, 1024 * dbp->cachesize, 0))) {
416 LOG(log_error, logtype_cnid, "error setting DB environment cachesize to %i: %s",
417 dbp->cachesize, db_strerror(ret));
418 dbd->db_env->close(dbd->db_env, 0);
423 if ((ret = dbd->db_env->set_lk_max_locks(dbd->db_env, dbp->maxlocks))) {
424 LOG(log_error, logtype_cnid, "error setting DB environment maxlocks to %i: %s",
425 10000, db_strerror(ret));
426 dbd->db_env->close(dbd->db_env, 0);
431 if ((ret = dbd->db_env->set_lk_max_objects(dbd->db_env, dbp->maxlockobjs))) {
432 LOG(log_error, logtype_cnid, "error setting DB environment max lockobjects to %i: %s",
433 10000, db_strerror(ret));
434 dbd->db_env->close(dbd->db_env, 0);
439 if ((ret = dbd->db_env->open(dbd->db_env, dbd->db_envhome, dbenv_oflags, 0))) {
440 LOG(log_error, logtype_cnid, "error opening DB environment after recovery: %s",
442 dbd->db_env->close(dbd->db_env, 0);
447 if ((ret = dbd->db_env->set_flags(dbd->db_env, DB_AUTO_COMMIT, 1))) {
448 LOG(log_error, logtype_cnid, "error setting DB_AUTO_COMMIT flag: %s",
450 dbd->db_env->close(dbd->db_env, 0);
455 if (dbp->logfile_autoremove) {
456 if ((dbif_logautorem(dbd)) != 0)
459 #if DB_VERSION_MAJOR > 4 || (DB_VERSION_MAJOR == 4 && DB_VERSION_MINOR >= 7)
460 if ((ret = dbd->db_env->log_set_config(dbd->db_env, DB_LOG_AUTO_REMOVE, 1))) {
461 LOG(log_error, logtype_cnid, "error setting DB_LOG_AUTO_REMOVE flag: %s",
463 dbd->db_env->close(dbd->db_env, 0);
468 if ((ret = dbd->db_env->set_flags(dbd->db_env, DB_LOG_AUTOREMOVE, 1))) {
469 LOG(log_error, logtype_cnid, "error setting DB_LOG_AUTOREMOVE flag: %s",
471 dbd->db_env->close(dbd->db_env, 0);
481 /* --------------- */
482 int dbif_open(DBD *dbd, struct db_param *dbp, int reindex)
489 /* Try to upgrade if it's a normal on-disk database */
490 if (dbd->db_envhome) {
492 if ((cwd = open(".", O_RDONLY)) < 0) {
493 LOG(log_error, logtype_cnid, "error opening cwd: %s", strerror(errno));
497 /* chdir to db_envhome. makes it easier checking for old db files and creating db_errlog file */
498 if ((chdir(dbd->db_envhome)) != 0) {
499 LOG(log_error, logtype_cnid, "error chdiring to db_env '%s': %s", dbd->db_envhome, strerror(errno));
503 if ((stat(dbd->db_filename, &st)) == 0) {
504 LOG(log_debug, logtype_cnid, "See if we can upgrade the CNID database...");
505 if ((ret = db_create(&upgrade_db, dbd->db_env, 0))) {
506 LOG(log_error, logtype_cnid, "error creating handle for database: %s", db_strerror(ret));
509 if ((ret = upgrade_db->upgrade(upgrade_db, dbd->db_filename, 0))) {
510 LOG(log_error, logtype_cnid, "error upgarding database: %s", db_strerror(ret));
513 if ((ret = upgrade_db->close(upgrade_db, 0))) {
514 LOG(log_error, logtype_cnid, "error closing database: %s", db_strerror(ret));
517 if ((ret = dbd->db_env->txn_checkpoint(dbd->db_env, 0, 0, DB_FORCE))) {
518 LOG(log_error, logtype_cnid, "error forcing checkpoint: %s", db_strerror(ret));
521 LOG(log_debug, logtype_cnid, "Finished BerkeleyBD upgrade check");
524 if ((fchdir(cwd)) != 0) {
525 LOG(log_error, logtype_cnid, "error chdiring back: %s", strerror(errno));
530 /* Now open databases ... */
531 for (i = 0; i != DBIF_DB_CNT; i++) {
532 if ((ret = db_create(&dbd->db_table[i].db, dbd->db_env, 0))) {
533 LOG(log_error, logtype_cnid, "error creating handle for database %s: %s",
534 dbd->db_table[i].name, db_strerror(ret));
538 if (dbd->db_table[i].flags) {
539 if ((ret = dbd->db_table[i].db->set_flags(dbd->db_table[i].db,
540 dbd->db_table[i].flags))) {
541 LOG(log_error, logtype_cnid, "error setting flags for database %s: %s",
542 dbd->db_table[i].name, db_strerror(ret));
547 if ( ! dbd->db_env) { /* In memory db */
548 if ((ret = dbd->db_table[i].db->set_cachesize(dbd->db_table[i].db,
551 4)) /* split in 4 memory chunks */
553 LOG(log_error, logtype_cnid, "error setting cachesize %u KB for database %s: %s",
554 dbp->cachesize / 1024, dbd->db_table[i].name, db_strerror(ret));
559 if (dbd->db_table[i].db->open(dbd->db_table[i].db,
562 dbd->db_table[i].name,
563 dbd->db_table[i].type,
564 dbd->db_table[i].openflags,
566 LOG(log_error, logtype_cnid, "Cant open database");
570 if (reindex && i > 0) {
571 LOG(log_info, logtype_cnid, "Truncating CNID index.");
572 if ((ret = dbd->db_table[i].db->truncate(dbd->db_table[i].db, NULL, &count, 0))) {
573 LOG(log_error, logtype_cnid, "error truncating database %s: %s",
574 dbd->db_table[i].name, db_strerror(ret));
580 /* TODO: Implement CNID DB versioning info on new databases. */
582 /* Associate the secondary with the primary. */
584 LOG(log_info, logtype_cnid, "Reindexing did/name index...");
585 if ((ret = dbd->db_table[0].db->associate(dbd->db_table[DBIF_CNID].db,
587 dbd->db_table[DBIF_IDX_DIDNAME].db,
589 (reindex) ? DB_CREATE : 0))
591 LOG(log_error, logtype_cnid, "Failed to associate didname database: %s",db_strerror(ret));
595 LOG(log_info, logtype_cnid, "... done.");
598 LOG(log_info, logtype_cnid, "Reindexing dev/ino index...");
599 if ((ret = dbd->db_table[0].db->associate(dbd->db_table[0].db,
601 dbd->db_table[DBIF_IDX_DEVINO].db,
603 (reindex) ? DB_CREATE : 0))
605 LOG(log_error, logtype_cnid, "Failed to associate devino database: %s",db_strerror(ret));
609 LOG(log_info, logtype_cnid, "... done.");
612 LOG(log_info, logtype_cnid, "Reindexing name index...");
614 uint32_t version = CNID_VERSION;
615 if (dbd->db_envhome && !reindex) {
616 if (dbif_getversion(dbd, &version) == -1)
620 if ((ret = dbd->db_table[0].db->associate(dbd->db_table[0].db,
622 dbd->db_table[DBIF_IDX_NAME].db,
626 ((CNID_VERSION == CNID_VERSION_1) && (version == CNID_VERSION_0)))
627 ? DB_CREATE : 0)) != 0) {
628 LOG(log_error, logtype_cnid, "Failed to associate name index: %s", db_strerror(ret));
632 LOG(log_info, logtype_cnid, "... done.");
634 if ((dbd->db_envhome) && ((ret = dbif_upgrade(dbd)) != 0)) {
635 LOG(log_error, logtype_cnid, "Error upgrading CNID database to version %d", CNID_VERSION);
642 /* ------------------------ */
643 static int dbif_closedb(DBD *dbd)
649 for (i = DBIF_DB_CNT -1; i >= 0; i--) {
650 if (dbd->db_table[i].db != NULL && (ret = dbd->db_table[i].db->close(dbd->db_table[i].db, 0))) {
651 LOG(log_error, logtype_cnid, "error closing database %s: %s", dbd->db_table[i].name, db_strerror(ret));
660 /* ------------------------ */
661 int dbif_close(DBD *dbd)
666 if (dbif_closedb(dbd))
669 if (dbd->db_env != NULL && (ret = dbd->db_env->close(dbd->db_env, 0))) {
670 LOG(log_error, logtype_cnid, "error closing DB environment: %s", db_strerror(ret));
673 if (dbd->db_errlog != NULL && fclose(dbd->db_errlog) == EOF) {
674 LOG(log_error, logtype_cnid, "error closing DB logfile: %s", strerror(errno));
678 free(dbd->db_filename);
688 In order to support silent database upgrades:
689 destroy env at cnid_dbd shutdown.
691 int dbif_env_remove(const char *path)
696 LOG(log_debug, logtype_cnid, "Reopening BerkeleyDB environment");
698 if (NULL == (dbd = dbif_init(path, "cnid2.db")))
701 /* Get db_env handle */
702 if ((ret = db_env_create(&dbd->db_env, 0))) {
703 LOG(log_error, logtype_cnid, "error creating DB environment: %s", db_strerror(ret));
708 if ((dbif_openlog(dbd)) != 0)
711 /* Open environment with recovery */
712 if ((ret = dbd->db_env->open(dbd->db_env,
714 DB_CREATE | DB_INIT_LOG | DB_INIT_MPOOL | DB_INIT_LOCK | DB_INIT_TXN | DB_RECOVER | DB_PRIVATE,
716 LOG(log_error, logtype_cnid, "error opening DB environment: %s",
718 dbd->db_env->close(dbd->db_env, 0);
723 if (dbd->db_errlog != NULL)
724 fflush(dbd->db_errlog);
726 /* Remove logfiles */
727 if ((ret = dbd->db_env->log_archive(dbd->db_env, NULL, DB_ARCH_REMOVE))) {
728 LOG(log_error, logtype_cnid, "error removing transaction logfiles: %s", db_strerror(ret));
732 if ((ret = dbd->db_env->close(dbd->db_env, 0))) {
733 LOG(log_error, logtype_cnid, "error closing DB environment after recovery: %s", db_strerror(ret));
738 LOG(log_debug, logtype_cnid, "BerkeleyDB environment recovery done.");
740 /* Get a new db_env handle and then remove environment */
741 if ((ret = db_env_create(&dbd->db_env, 0))) {
742 LOG(log_error, logtype_cnid, "error acquiring db_end handle: %s", db_strerror(ret));
746 if ((ret = dbd->db_env->remove(dbd->db_env, dbd->db_envhome, 0))) {
747 LOG(log_error, logtype_cnid, "error removing BerkeleyDB environment: %s", db_strerror(ret));
751 LOG(log_debug, logtype_cnid, "Removed BerkeleyDB environment.");
757 * The following three functions are wrappers for DB->get(), DB->put() and DB->del().
758 * All three return -1 on error. dbif_get()/dbif_del return 1 if the key was found and 0
759 * otherwise. dbif_put() returns 0 if key/val was successfully updated and 1 if
760 * the DB_NOOVERWRITE flag was specified and the key already exists.
762 * All return codes other than DB_NOTFOUND and DB_KEYEXIST from the DB->()
763 * functions are not expected and therefore error conditions.
766 int dbif_get(DBD *dbd, const int dbi, DBT *key, DBT *val, u_int32_t flags)
770 ret = dbd->db_table[dbi].db->get(dbd->db_table[dbi].db,
776 if (ret == DB_NOTFOUND)
779 LOG(log_error, logtype_cnid, "error retrieving value from %s: %s",
780 dbd->db_table[dbi].name, db_strerror(ret));
786 /* search by secondary return primary */
787 int dbif_pget(DBD *dbd, const int dbi, DBT *key, DBT *pkey, DBT *val, u_int32_t flags)
791 ret = dbd->db_table[dbi].db->pget(dbd->db_table[dbi].db,
798 if (ret == DB_NOTFOUND || ret == DB_SECONDARY_BAD) {
802 LOG(log_error, logtype_cnid, "error retrieving value from %s: %s",
803 dbd->db_table[dbi].name, db_strerror(ret));
809 /* -------------------------- */
810 int dbif_put(DBD *dbd, const int dbi, DBT *key, DBT *val, u_int32_t flags)
814 if (dbif_txn_begin(dbd) < 0) {
815 LOG(log_error, logtype_cnid, "error setting key/value in %s", dbd->db_table[dbi].name);
819 ret = dbd->db_table[dbi].db->put(dbd->db_table[dbi].db,
827 if ((flags & DB_NOOVERWRITE) && ret == DB_KEYEXIST) {
830 LOG(log_error, logtype_cnid, "error setting key/value in %s: %s",
831 dbd->db_table[dbi].name, db_strerror(ret));
838 int dbif_del(DBD *dbd, const int dbi, DBT *key, u_int32_t flags)
842 /* For cooperation with the dbd utility and its usage of a cursor */
844 dbd->db_cur->close(dbd->db_cur);
848 if (dbif_txn_begin(dbd) < 0) {
849 LOG(log_error, logtype_cnid, "error deleting key/value from %s", dbd->db_table[dbi].name);
853 ret = dbd->db_table[dbi].db->del(dbd->db_table[dbi].db,
858 if (ret == DB_NOTFOUND) {
859 LOG(log_info, logtype_cnid, "key not found");
863 LOG(log_error, logtype_cnid, "error deleting key/value from %s: %s",
864 dbd->db_table[dbi].name, db_strerror(ret));
871 * Search the database by name
873 * @param resbuf (w) buffer for search results CNIDs, maxsize is assumed to be
874 * DBD_MAX_SRCH_RSLTS * sizefof(cnid_t)
876 * @returns -1 on error, 0 when nothing found, else the number of matches
878 int dbif_search(DBD *dbd, DBT *key, char *resbuf)
884 char *cnids = resbuf;
886 char *namebkp = key->data;
887 int namelenbkp = key->size;
889 memset(&pkey, 0, sizeof(DBT));
890 memset(&data, 0, sizeof(DBT));
893 ret = dbd->db_table[DBIF_IDX_NAME].db->cursor(dbd->db_table[DBIF_IDX_NAME].db,
898 LOG(log_error, logtype_cnid, "Couldn't create cursor: %s", db_strerror(ret));
903 ret = cursorp->pget(cursorp, key, &pkey, &data, DB_SET_RANGE);
904 while (count < DBD_MAX_SRCH_RSLTS && ret != DB_NOTFOUND) {
905 if (!((namelenbkp <= key->size) && (strncmp(namebkp, key->data, namelenbkp) == 0)))
908 memcpy(cnids, pkey.data, sizeof(cnid_t));
909 memcpy(&cnid, pkey.data, sizeof(cnid_t));
910 cnids += sizeof(cnid_t);
911 LOG(log_debug, logtype_cnid, "match: CNID %" PRIu32, ntohl(cnid));
913 ret = cursorp->pget(cursorp, key, &pkey, &data, DB_NEXT);
920 cursorp->close(cursorp);
924 int dbif_txn_begin(DBD *dbd)
928 /* If we already have an active txn, just return */
932 /* If our DBD has no env, just return (-> in memory db) */
933 if (dbd->db_env == NULL)
936 ret = dbd->db_env->txn_begin(dbd->db_env, NULL, &dbd->db_txn, 0);
939 LOG(log_error, logtype_cnid, "error starting transaction: %s", db_strerror(ret));
945 int dbif_txn_commit(DBD *dbd)
952 /* If our DBD has no env, just return (-> in memory db) */
953 if (dbd->db_env == NULL)
956 ret = dbd->db_txn->commit(dbd->db_txn, 0);
960 LOG(log_error, logtype_cnid, "error committing transaction: %s", db_strerror(ret));
966 int dbif_txn_abort(DBD *dbd)
973 /* If our DBD has no env, just return (-> in memory db) */
974 if (dbd->db_env == NULL)
977 ret = dbd->db_txn->abort(dbd->db_txn);
981 LOG(log_error, logtype_cnid, "error aborting transaction: %s", db_strerror(ret));
988 ret = 1 -> commit txn
989 ret = 0 -> abort txn -> exit!
990 anything else -> exit!
992 void dbif_txn_close(DBD *dbd, int ret)
995 if (dbif_txn_abort(dbd) < 0) {
996 LOG( log_error, logtype_cnid, "Fatal error aborting transaction. Exiting!");
999 } else if (ret == 1) {
1000 ret = dbif_txn_commit(dbd);
1002 LOG( log_error, logtype_cnid, "Fatal error committing transaction. Exiting!");
1009 int dbif_txn_checkpoint(DBD *dbd, u_int32_t kbyte, u_int32_t min, u_int32_t flags)
1012 ret = dbd->db_env->txn_checkpoint(dbd->db_env, kbyte, min, flags);
1014 LOG(log_error, logtype_cnid, "error checkpointing transaction susystem: %s", db_strerror(ret));
1020 int dbif_count(DBD *dbd, const int dbi, u_int32_t *count)
1024 DB *db = dbd->db_table[dbi].db;
1026 ret = db->stat(db, NULL, &sp, 0);
1029 LOG(log_error, logtype_cnid, "error getting stat infotmation on database: %s", db_strerror(ret));
1033 *count = sp->bt_ndata;
1039 int dbif_copy_rootinfokey(DBD *srcdbd, DBD *destdbd)
1044 memset(&key, 0, sizeof(key));
1045 memset(&data, 0, sizeof(data));
1047 key.data = ROOTINFO_KEY;
1048 key.size = ROOTINFO_KEYLEN;
1050 if ((rc = dbif_get(srcdbd, DBIF_CNID, &key, &data, 0)) <= 0) {
1051 LOG(log_error, logtype_cnid, "dbif_copy_rootinfokey: Error getting rootinfo record");
1055 memset(&key, 0, sizeof(key));
1056 key.data = ROOTINFO_KEY;
1057 key.size = ROOTINFO_KEYLEN;
1059 if ((rc = dbif_put(destdbd, DBIF_CNID, &key, &data, 0))) {
1060 LOG(log_error, logtype_cnid, "dbif_copy_rootinfokey: Error writing rootinfo key");
1067 int dbif_dump(DBD *dbd, int dumpindexes)
1070 uint32_t max = 0, count = 0, cnid, type, did, lastid, version;
1074 DBT key = { 0 }, data = { 0 };
1075 DB *db = dbd->db_table[DBIF_CNID].db;
1076 char *typestring[2] = {"f", "d"};
1079 printf("CNID database dump:\n");
1081 rc = db->cursor(db, NULL, &cur, 0);
1083 LOG(log_error, logtype_cnid, "Couldn't create cursor: %s", db_strerror(rc));
1087 cur->c_get(cur, &key, &data, DB_FIRST);
1089 /* Parse and print data */
1090 memcpy(&cnid, key.data, 4);
1095 /* Rootinfo node ? */
1097 memcpy(&stamp, (char *)data.data + CNID_DEV_OFS, sizeof(time_t));
1098 memcpy(&lastid, (char *)data.data + CNID_TYPE_OFS, CNID_TYPE_LEN);
1099 lastid = ntohl(lastid);
1100 memcpy(&version, (char *)data.data + CNID_DID_OFS, CNID_DID_LEN);
1101 version = ntohl(version);
1103 strftime(timebuf, sizeof(timebuf), "%b %d %Y %H:%M:%S", localtime(&stamp));
1104 printf("CNID db version: %u, dbd stamp: 0x%08x (%s), next free CNID: %u\n",
1105 version, (unsigned int)stamp, timebuf, lastid + 1);
1108 memcpy(&dev, (char *)data.data + CNID_DEV_OFS, 8);
1111 memcpy(&ino, (char *)data.data + CNID_INO_OFS, 8);
1114 memcpy(&type, (char *)data.data + CNID_TYPE_OFS, 4);
1117 memcpy(&did, (char *)data.data + CNID_DID_OFS, 4);
1121 printf("id: %10u, did: %10u, type: %s, dev: 0x%llx, ino: 0x%016llx, name: %s\n",
1122 cnid, did, typestring[type],
1123 (long long unsigned int)dev, (long long unsigned int)ino,
1124 (char *)data.data + CNID_NAME_OFS);
1128 rc = cur->c_get(cur, &key, &data, DB_NEXT);
1131 if (rc != DB_NOTFOUND) {
1132 LOG(log_error, logtype_cnid, "Error iterating over btree: %s", db_strerror(rc));
1136 rc = cur->c_close(cur);
1138 LOG(log_error, logtype_cnid, "Couldn't close cursor: %s", db_strerror(rc));
1141 printf("%u CNIDs in database. Max CNID: %u.\n", count, max);
1143 /* Dump indexes too ? */
1145 /* DBIF_IDX_DEVINO */
1146 printf("\ndev/inode index:\n");
1148 db = dbd->db_table[DBIF_IDX_DEVINO].db;
1149 rc = db->cursor(db, NULL, &cur, 0);
1151 LOG(log_error, logtype_cnid, "Couldn't create cursor: %s", db_strerror(rc));
1155 cur->c_get(cur, &key, &data, DB_FIRST);
1157 /* Parse and print data */
1160 memcpy(&cnid, data.data, CNID_LEN);
1166 memcpy(&dev, key.data, CNID_DEV_LEN);
1169 memcpy(&ino, (char *)key.data + CNID_DEV_LEN, CNID_INO_LEN);
1172 printf("id: %10u <== dev: 0x%llx, ino: 0x%016llx\n",
1173 cnid, (unsigned long long int)dev, (unsigned long long int)ino);
1176 rc = cur->c_get(cur, &key, &data, DB_NEXT);
1178 if (rc != DB_NOTFOUND) {
1179 LOG(log_error, logtype_cnid, "Error iterating over btree: %s", db_strerror(rc));
1183 rc = cur->c_close(cur);
1185 LOG(log_error, logtype_cnid, "Couldn't close cursor: %s", db_strerror(rc));
1188 printf("%u items\n", count);
1190 /* Now dump DBIF_IDX_DIDNAME */
1191 printf("\ndid/name index:\n");
1193 db = dbd->db_table[DBIF_IDX_DIDNAME].db;
1194 rc = db->cursor(db, NULL, &cur, 0);
1196 LOG(log_error, logtype_cnid, "Couldn't create cursor: %s", db_strerror(rc));
1200 cur->c_get(cur, &key, &data, DB_FIRST);
1202 /* Parse and print data */
1205 memcpy(&cnid, data.data, CNID_LEN);
1211 memcpy(&did, key.data, CNID_LEN);
1214 printf("id: %10u <== did: %10u, name: %s\n", cnid, did, (char *)key.data + CNID_LEN);
1217 rc = cur->c_get(cur, &key, &data, DB_NEXT);
1219 if (rc != DB_NOTFOUND) {
1220 LOG(log_error, logtype_cnid, "Error iterating over btree: %s", db_strerror(rc));
1224 rc = cur->c_close(cur);
1226 LOG(log_error, logtype_cnid, "Couldn't close cursor: %s", db_strerror(rc));
1229 printf("%u items\n", count);
1236 Iterates over dbd, returning cnids.
1237 Uses in-value of cnid to seek to that cnid, then gets next and return that in cnid.
1238 If close=1, close cursor.
1239 Return -1 on error, 0 on EOD (end-of-database), 1 if returning cnid.
1241 int dbif_idwalk(DBD *dbd, cnid_t *cnid, int close)
1247 static DBT key = { 0 }, data = { 0 };
1248 DB *db = dbd->db_table[DBIF_CNID].db;
1252 dbd->db_cur->close(dbd->db_cur);
1258 /* An dbif_del will have closed our cursor too */
1259 if ( ! dbd->db_cur ) {
1260 if ((rc = db->cursor(db, NULL, &dbd->db_cur, 0)) != 0) {
1261 LOG(log_error, logtype_cnid, "Couldn't create cursor: %s", db_strerror(rc));
1264 flag = DB_SET_RANGE; /* This will seek to next cnid after the one just deleted */
1267 key.size = sizeof(cnid_t);
1271 if ((rc = dbd->db_cur->get(dbd->db_cur, &key, &data, flag)) == 0) {
1272 memcpy(cnid, key.data, sizeof(cnid_t));
1273 *cnid = ntohl(*cnid);
1277 if (rc != DB_NOTFOUND) {
1278 LOG(log_error, logtype_cnid, "Error iterating over btree: %s", db_strerror(rc));
1279 dbd->db_cur->close(dbd->db_cur);
1285 dbd->db_cur->close(dbd->db_cur);