2 * Copyright (C) Joerg Lenneis 2003
3 * Copyright (C) Frank Lahm 2009
4 * All Rights Reserved. See COPYING.
9 #endif /* HAVE_CONFIG_H */
14 #ifdef HAVE_SYS_TYPES_H
15 #include <sys/types.h>
16 #endif /* HAVE_SYS_TYPES_H */
18 #include <sys/types.h>
20 #include <sys/cdefs.h>
22 #include <atalk/logger.h>
28 #define DB_ERRLOGFILE "db_errlog"
30 static char *old_dbfiles[] = {"cnid.db", NULL};
33 static int upgrade_required(const DBD *dbd)
41 if ( ! dbd->db_filename)
46 if ((cwd = open(".", O_RDONLY)) < 0) {
47 LOG(log_error, logtype_cnid, "error opening cwd: %s", strerror(errno));
51 /* chdir to db_envhome */
52 if ((chdir(dbd->db_envhome)) != 0) {
53 LOG(log_error, logtype_cnid, "error chdiring to db_env '%s': %s", dbd->db_envhome, strerror(errno));
58 for (i = 0; old_dbfiles[i] != NULL; i++) {
59 if ( !(stat(old_dbfiles[i], &st) < 0) ) {
63 if (errno != ENOENT) {
64 LOG(log_error, logtype_cnid, "cnid_open: Checking %s gave %s", old_dbfiles[i], strerror(errno));
71 if ((fchdir(cwd)) != 0) {
72 LOG(log_error, logtype_cnid, "error chdiring back: %s", strerror(errno));
77 return (ret < 0 ? ret : found);
81 * Upgrade CNID database versions
83 * For now this does nothing, as upgrading from ver. 0 to 1 is done in dbif_open
85 static int dbif_upgrade(DBD *dbd)
89 if ((version = dbif_getversion(dbd)) == -1)
93 * Do upgrade stuff ...
96 /* Write current version to database */
97 if (version != CNID_VERSION) {
98 if (dbif_setversion(dbd, CNID_VERSION) != 0)
102 LOG(log_debug, logtype_cnid, "Finished CNID database version upgrade check");
107 /* --------------- */
108 static int dbif_openlog(DBD *dbd)
113 if ( ! dbd->db_filename)
118 if ((cwd = open(".", O_RDONLY)) < 0) {
119 LOG(log_error, logtype_cnid, "error opening cwd: %s", strerror(errno));
123 /* chdir to db_envhome */
124 if ((chdir(dbd->db_envhome)) != 0) {
125 LOG(log_error, logtype_cnid, "error chdiring to db_env '%s': %s", dbd->db_envhome, strerror(errno));
130 if ((dbd->db_errlog = fopen(DB_ERRLOGFILE, "a")) == NULL)
131 LOG(log_warning, logtype_cnid, "error creating/opening DB errlogfile: %s", strerror(errno));
133 if (dbd->db_errlog != NULL) {
134 dbd->db_env->set_errfile(dbd->db_env, dbd->db_errlog);
135 dbd->db_env->set_msgfile(dbd->db_env, dbd->db_errlog);
140 if ((fchdir(cwd)) != 0) {
141 LOG(log_error, logtype_cnid, "error chdiring back: %s", strerror(errno));
149 /* --------------- */
150 static int dbif_logautorem(DBD *dbd)
154 char **logfiles = NULL;
157 if ( ! dbd->db_filename)
162 if ((cwd = open(".", O_RDONLY)) < 0) {
163 LOG(log_error, logtype_cnid, "error opening cwd: %s", strerror(errno));
167 /* chdir to db_envhome */
168 if ((chdir(dbd->db_envhome)) != 0) {
169 LOG(log_error, logtype_cnid, "error chdiring to db_env '%s': %s", dbd->db_envhome, strerror(errno));
174 if ((ret = dbd->db_env->log_archive(dbd->db_env, &logfiles, 0)) != 0) {
175 LOG(log_error, logtype_cnid, "error getting list of stale logfiles: %s",
177 dbd->db_env->close(dbd->db_env, 0);
183 if (logfiles != NULL) {
184 for (file = logfiles; *file != NULL; file++) {
185 if (unlink(*file) < 0)
186 LOG(log_warning, logtype_cnid, "Error removing stale logfile %s: %s", *file, strerror(errno));
193 if ((fchdir(cwd)) != 0) {
194 LOG(log_error, logtype_cnid, "error chdiring back: %s", strerror(errno));
202 #include <atalk/util.h>
204 /* --------------- */
205 int dbif_stamp(DBD *dbd, void *buffer, int size)
214 if ((cwd = open(".", O_RDONLY)) < 0) {
215 LOG(log_error, logtype_cnid, "error opening cwd: %s", strerror(errno));
219 /* chdir to db_envhome */
220 if ((chdir(dbd->db_envhome)) != 0) {
221 LOG(log_error, logtype_cnid, "error chdiring to db_env '%s': %s", dbd->db_envhome, strerror(errno));
225 if ((rc = stat(dbd->db_table[DBIF_CNID].name, &st)) < 0) {
226 LOG(log_error, logtype_cnid, "error stating database %s: %s", dbd->db_table[DBIF_CNID].name, db_strerror(errno));
230 LOG(log_maxdebug, logtype_cnid,"stamp: %s", asctime(localtime(&st.st_ctime)));
232 memset(buffer, 0, size);
233 memcpy(buffer, &st.st_ctime, sizeof(st.st_ctime));
235 if ((fchdir(cwd)) != 0) {
236 LOG(log_error, logtype_cnid, "error chdiring back: %s", strerror(errno));
243 /* --------------- */
244 DBD *dbif_init(const char *envhome, const char *filename)
248 if ( NULL == (dbd = calloc(sizeof(DBD), 1)) )
251 /* filename == NULL means in memory db */
256 dbd->db_envhome = strdup(envhome);
257 if (NULL == dbd->db_envhome) {
262 dbd->db_filename = strdup(filename);
263 if (NULL == dbd->db_filename) {
264 free(dbd->db_envhome);
270 dbd->db_table[DBIF_CNID].name = "cnid2.db";
271 dbd->db_table[DBIF_IDX_DEVINO].name = "devino.db";
272 dbd->db_table[DBIF_IDX_DIDNAME].name = "didname.db";
273 dbd->db_table[DBIF_IDX_NAME].name = "name.db";
275 dbd->db_table[DBIF_CNID].type = DB_BTREE;
276 dbd->db_table[DBIF_IDX_DEVINO].type = DB_BTREE;
277 dbd->db_table[DBIF_IDX_DIDNAME].type = DB_BTREE;
278 dbd->db_table[DBIF_IDX_NAME].type = DB_BTREE;
280 dbd->db_table[DBIF_CNID].openflags = DB_CREATE;
281 dbd->db_table[DBIF_IDX_DEVINO].openflags = DB_CREATE;
282 dbd->db_table[DBIF_IDX_DIDNAME].openflags = DB_CREATE;
283 dbd->db_table[DBIF_IDX_NAME].openflags = DB_CREATE;
285 dbd->db_table[DBIF_IDX_NAME].flags = DB_DUPSORT;
291 We must open the db_env with an absolute pathname, as `dbd` keeps chdir'ing, which
292 breaks e.g. bdb logfile-rotation with relative pathnames.
293 But still we use relative paths with upgrade_required() and the DB_ERRLOGFILE
294 in order to avoid creating absolute paths by copying. Both have no problem with
297 int dbif_env_open(DBD *dbd, struct db_param *dbp, uint32_t dbenv_oflags)
301 /* Refuse to do anything if this is an old version of the CNID database */
302 if (upgrade_required(dbd)) {
303 LOG(log_error, logtype_cnid, "Found version 1 of the CNID database. Please upgrade to version 2");
307 if ((ret = db_env_create(&dbd->db_env, 0))) {
308 LOG(log_error, logtype_cnid, "error creating DB environment: %s",
314 if ((dbif_openlog(dbd)) != 0)
317 if (dbenv_oflags & DB_RECOVER) {
319 LOG(log_debug, logtype_cnid, "Running recovery");
321 dbd->db_env->set_verbose(dbd->db_env, DB_VERB_RECOVERY, 1);
322 /* Open the database for recovery using DB_PRIVATE option which is faster */
323 if ((ret = dbd->db_env->open(dbd->db_env, dbd->db_envhome, dbenv_oflags | DB_PRIVATE, 0))) {
324 LOG(log_error, logtype_cnid, "error opening DB environment: %s",
326 dbd->db_env->close(dbd->db_env, 0);
330 dbenv_oflags = (dbenv_oflags & ~DB_RECOVER);
332 if (dbd->db_errlog != NULL)
333 fflush(dbd->db_errlog);
335 if ((ret = dbd->db_env->close(dbd->db_env, 0))) {
336 LOG(log_error, logtype_cnid, "error closing DB environment after recovery: %s",
341 dbd->db_errlog = NULL;
343 if ((ret = db_env_create(&dbd->db_env, 0))) {
344 LOG(log_error, logtype_cnid, "error creating DB environment after recovery: %s",
350 if ((dbif_openlog(dbd)) != 0)
353 LOG(log_debug, logtype_cnid, "Finished recovery.");
356 if ((ret = dbd->db_env->set_cachesize(dbd->db_env, 0, 1024 * dbp->cachesize, 0))) {
357 LOG(log_error, logtype_cnid, "error setting DB environment cachesize to %i: %s",
358 dbp->cachesize, db_strerror(ret));
359 dbd->db_env->close(dbd->db_env, 0);
364 if ((ret = dbd->db_env->set_lk_max_locks(dbd->db_env, dbp->maxlocks))) {
365 LOG(log_error, logtype_cnid, "error setting DB environment maxlocks to %i: %s",
366 10000, db_strerror(ret));
367 dbd->db_env->close(dbd->db_env, 0);
372 if ((ret = dbd->db_env->set_lk_max_objects(dbd->db_env, dbp->maxlockobjs))) {
373 LOG(log_error, logtype_cnid, "error setting DB environment max lockobjects to %i: %s",
374 10000, db_strerror(ret));
375 dbd->db_env->close(dbd->db_env, 0);
380 if ((ret = dbd->db_env->open(dbd->db_env, dbd->db_envhome, dbenv_oflags, 0))) {
381 LOG(log_error, logtype_cnid, "error opening DB environment after recovery: %s",
383 dbd->db_env->close(dbd->db_env, 0);
388 if ((ret = dbd->db_env->set_flags(dbd->db_env, DB_AUTO_COMMIT, 1))) {
389 LOG(log_error, logtype_cnid, "error setting DB_AUTO_COMMIT flag: %s",
391 dbd->db_env->close(dbd->db_env, 0);
396 if (dbp->logfile_autoremove) {
397 if ((dbif_logautorem(dbd)) != 0)
400 #if DB_VERSION_MAJOR > 4 || (DB_VERSION_MAJOR == 4 && DB_VERSION_MINOR >= 7)
401 if ((ret = dbd->db_env->log_set_config(dbd->db_env, DB_LOG_AUTO_REMOVE, 1))) {
402 LOG(log_error, logtype_cnid, "error setting DB_LOG_AUTO_REMOVE flag: %s",
404 dbd->db_env->close(dbd->db_env, 0);
409 if ((ret = dbd->db_env->set_flags(dbd->db_env, DB_LOG_AUTOREMOVE, 1))) {
410 LOG(log_error, logtype_cnid, "error setting DB_LOG_AUTOREMOVE flag: %s",
412 dbd->db_env->close(dbd->db_env, 0);
422 /* --------------- */
423 int dbif_open(DBD *dbd, struct db_param *dbp, int reindex)
430 /* Try to upgrade if it's a normal on-disk database */
431 if (dbd->db_envhome) {
433 if ((cwd = open(".", O_RDONLY)) < 0) {
434 LOG(log_error, logtype_cnid, "error opening cwd: %s", strerror(errno));
438 /* chdir to db_envhome. makes it easier checking for old db files and creating db_errlog file */
439 if ((chdir(dbd->db_envhome)) != 0) {
440 LOG(log_error, logtype_cnid, "error chdiring to db_env '%s': %s", dbd->db_envhome, strerror(errno));
444 if ((stat(dbd->db_filename, &st)) == 0) {
445 LOG(log_debug, logtype_cnid, "See if we can upgrade the CNID database...");
446 if ((ret = db_create(&upgrade_db, dbd->db_env, 0))) {
447 LOG(log_error, logtype_cnid, "error creating handle for database: %s", db_strerror(ret));
450 if ((ret = upgrade_db->upgrade(upgrade_db, dbd->db_filename, 0))) {
451 LOG(log_error, logtype_cnid, "error upgarding database: %s", db_strerror(ret));
454 if ((ret = upgrade_db->close(upgrade_db, 0))) {
455 LOG(log_error, logtype_cnid, "error closing database: %s", db_strerror(ret));
458 if ((ret = dbd->db_env->txn_checkpoint(dbd->db_env, 0, 0, DB_FORCE))) {
459 LOG(log_error, logtype_cnid, "error forcing checkpoint: %s", db_strerror(ret));
462 LOG(log_debug, logtype_cnid, "Finished BerkeleyBD upgrade check");
465 if ((fchdir(cwd)) != 0) {
466 LOG(log_error, logtype_cnid, "error chdiring back: %s", strerror(errno));
471 /* Now open databases ... */
472 for (i = 0; i != DBIF_DB_CNT; i++) {
473 if ((ret = db_create(&dbd->db_table[i].db, dbd->db_env, 0))) {
474 LOG(log_error, logtype_cnid, "error creating handle for database %s: %s",
475 dbd->db_table[i].name, db_strerror(ret));
479 if (dbd->db_table[i].flags) {
480 if ((ret = dbd->db_table[i].db->set_flags(dbd->db_table[i].db,
481 dbd->db_table[i].flags))) {
482 LOG(log_error, logtype_cnid, "error setting flags for database %s: %s",
483 dbd->db_table[i].name, db_strerror(ret));
488 if ( ! dbd->db_env) { /* In memory db */
489 if ((ret = dbd->db_table[i].db->set_cachesize(dbd->db_table[i].db,
492 4)) /* split in 4 memory chunks */
494 LOG(log_error, logtype_cnid, "error setting cachesize %u KB for database %s: %s",
495 dbp->cachesize / 1024, dbd->db_table[i].name, db_strerror(ret));
500 if (dbd->db_table[i].db->open(dbd->db_table[i].db,
503 dbd->db_table[i].name,
504 dbd->db_table[i].type,
505 dbd->db_table[i].openflags,
507 LOG(log_error, logtype_cnid, "Cant open database");
511 if (reindex && i > 0) {
512 LOG(log_info, logtype_cnid, "Truncating CNID index.");
513 if ((ret = dbd->db_table[i].db->truncate(dbd->db_table[i].db, NULL, &count, 0))) {
514 LOG(log_error, logtype_cnid, "error truncating database %s: %s",
515 dbd->db_table[i].name, db_strerror(ret));
521 /* TODO: Implement CNID DB versioning info on new databases. */
523 /* Associate the secondary with the primary. */
525 LOG(log_info, logtype_cnid, "Reindexing did/name index...");
526 if ((ret = dbd->db_table[0].db->associate(dbd->db_table[DBIF_CNID].db,
528 dbd->db_table[DBIF_IDX_DIDNAME].db,
530 (reindex) ? DB_CREATE : 0))
532 LOG(log_error, logtype_cnid, "Failed to associate didname database: %s",db_strerror(ret));
536 LOG(log_info, logtype_cnid, "... done.");
539 LOG(log_info, logtype_cnid, "Reindexing dev/ino index...");
540 if ((ret = dbd->db_table[0].db->associate(dbd->db_table[0].db,
542 dbd->db_table[DBIF_IDX_DEVINO].db,
544 (reindex) ? DB_CREATE : 0))
546 LOG(log_error, logtype_cnid, "Failed to associate devino database: %s",db_strerror(ret));
550 LOG(log_info, logtype_cnid, "... done.");
553 LOG(log_info, logtype_cnid, "Reindexing name index...");
555 int version = CNID_VERSION;
556 if (dbd->db_envhome && !reindex) {
557 if ((version = dbif_getversion(dbd)) == -1)
561 if ((ret = dbd->db_table[0].db->associate(dbd->db_table[0].db,
563 dbd->db_table[DBIF_IDX_NAME].db,
567 ((CNID_VERSION == CNID_VERSION_1) && (version == CNID_VERSION_0)))
568 ? DB_CREATE : 0)) != 0) {
569 LOG(log_error, logtype_cnid, "Failed to associate name index: %s", db_strerror(ret));
573 LOG(log_info, logtype_cnid, "... done.");
575 if ((dbd->db_envhome) && ((ret = dbif_upgrade(dbd)) != 0)) {
576 LOG(log_error, logtype_cnid, "Error upgrading CNID database to version %d", CNID_VERSION);
583 /* ------------------------ */
584 static int dbif_closedb(DBD *dbd)
590 for (i = DBIF_DB_CNT -1; i >= 0; i--) {
591 if (dbd->db_table[i].db != NULL && (ret = dbd->db_table[i].db->close(dbd->db_table[i].db, 0))) {
592 LOG(log_error, logtype_cnid, "error closing database %s: %s", dbd->db_table[i].name, db_strerror(ret));
601 /* ------------------------ */
602 int dbif_close(DBD *dbd)
607 if (dbif_closedb(dbd))
610 if (dbd->db_env != NULL && (ret = dbd->db_env->close(dbd->db_env, 0))) {
611 LOG(log_error, logtype_cnid, "error closing DB environment: %s", db_strerror(ret));
614 if (dbd->db_errlog != NULL && fclose(dbd->db_errlog) == EOF) {
615 LOG(log_error, logtype_cnid, "error closing DB logfile: %s", strerror(errno));
619 free(dbd->db_filename);
629 In order to support silent database upgrades:
630 destroy env at cnid_dbd shutdown.
632 int dbif_env_remove(const char *path)
637 LOG(log_debug, logtype_cnid, "Reopening BerkeleyDB environment");
639 if (NULL == (dbd = dbif_init(path, "cnid2.db")))
642 /* Get db_env handle */
643 if ((ret = db_env_create(&dbd->db_env, 0))) {
644 LOG(log_error, logtype_cnid, "error creating DB environment: %s", db_strerror(ret));
649 if ((dbif_openlog(dbd)) != 0)
652 /* Open environment with recovery */
653 if ((ret = dbd->db_env->open(dbd->db_env,
655 DB_CREATE | DB_INIT_LOG | DB_INIT_MPOOL | DB_INIT_LOCK | DB_INIT_TXN | DB_RECOVER | DB_PRIVATE,
657 LOG(log_error, logtype_cnid, "error opening DB environment: %s",
659 dbd->db_env->close(dbd->db_env, 0);
664 if (dbd->db_errlog != NULL)
665 fflush(dbd->db_errlog);
667 /* Remove logfiles */
668 if ((ret = dbd->db_env->log_archive(dbd->db_env, NULL, DB_ARCH_REMOVE))) {
669 LOG(log_error, logtype_cnid, "error removing transaction logfiles: %s", db_strerror(ret));
673 if ((ret = dbd->db_env->close(dbd->db_env, 0))) {
674 LOG(log_error, logtype_cnid, "error closing DB environment after recovery: %s", db_strerror(ret));
679 LOG(log_debug, logtype_cnid, "BerkeleyDB environment recovery done.");
681 /* Get a new db_env handle and then remove environment */
682 if ((ret = db_env_create(&dbd->db_env, 0))) {
683 LOG(log_error, logtype_cnid, "error acquiring db_end handle: %s", db_strerror(ret));
687 if ((ret = dbd->db_env->remove(dbd->db_env, dbd->db_envhome, 0))) {
688 LOG(log_error, logtype_cnid, "error removing BerkeleyDB environment: %s", db_strerror(ret));
692 LOG(log_debug, logtype_cnid, "Removed BerkeleyDB environment.");
698 * The following three functions are wrappers for DB->get(), DB->put() and DB->del().
699 * All three return -1 on error. dbif_get()/dbif_del return 1 if the key was found and 0
700 * otherwise. dbif_put() returns 0 if key/val was successfully updated and 1 if
701 * the DB_NOOVERWRITE flag was specified and the key already exists.
703 * All return codes other than DB_NOTFOUND and DB_KEYEXIST from the DB->()
704 * functions are not expected and therefore error conditions.
707 int dbif_get(DBD *dbd, const int dbi, DBT *key, DBT *val, u_int32_t flags)
711 ret = dbd->db_table[dbi].db->get(dbd->db_table[dbi].db,
717 if (ret == DB_NOTFOUND)
720 LOG(log_error, logtype_cnid, "error retrieving value from %s: %s",
721 dbd->db_table[dbi].name, db_strerror(ret));
727 /* search by secondary return primary */
728 int dbif_pget(DBD *dbd, const int dbi, DBT *key, DBT *pkey, DBT *val, u_int32_t flags)
732 ret = dbd->db_table[dbi].db->pget(dbd->db_table[dbi].db,
739 if (ret == DB_NOTFOUND || ret == DB_SECONDARY_BAD) {
743 LOG(log_error, logtype_cnid, "error retrieving value from %s: %s",
744 dbd->db_table[dbi].name, db_strerror(ret));
750 /* -------------------------- */
751 int dbif_put(DBD *dbd, const int dbi, DBT *key, DBT *val, u_int32_t flags)
755 if (dbif_txn_begin(dbd) < 0) {
756 LOG(log_error, logtype_cnid, "error setting key/value in %s", dbd->db_table[dbi].name);
760 ret = dbd->db_table[dbi].db->put(dbd->db_table[dbi].db,
768 if ((flags & DB_NOOVERWRITE) && ret == DB_KEYEXIST) {
771 LOG(log_error, logtype_cnid, "error setting key/value in %s: %s",
772 dbd->db_table[dbi].name, db_strerror(ret));
779 int dbif_del(DBD *dbd, const int dbi, DBT *key, u_int32_t flags)
783 /* For cooperation with the dbd utility and its usage of a cursor */
785 dbd->db_cur->close(dbd->db_cur);
789 if (dbif_txn_begin(dbd) < 0) {
790 LOG(log_error, logtype_cnid, "error deleting key/value from %s", dbd->db_table[dbi].name);
794 ret = dbd->db_table[dbi].db->del(dbd->db_table[dbi].db,
799 if (ret == DB_NOTFOUND) {
800 LOG(log_info, logtype_cnid, "key not found");
804 LOG(log_error, logtype_cnid, "error deleting key/value from %s: %s",
805 dbd->db_table[dbi].name, db_strerror(ret));
812 * Inititialize rootinfo key (which has CNID 0 as key)
814 * This also "stamps" the database, which means storing st.st_ctime of the
815 * "cnid2.db" file in the rootinfo data at the DEV offset
817 * @param dbd (rw) database handle
818 * @param version (r) database version number
820 * @returns -1 on error, 0 on success
822 static int dbif_init_rootinfo(DBD *dbd, int version)
826 char buf[ROOTINFO_DATALEN];
828 LOG(log_debug, logtype_cnid, "Setting CNID database version to %u", version);
833 memset(&key, 0, sizeof(key));
834 memset(&data, 0, sizeof(data));
835 key.data = ROOTINFO_KEY;
836 key.size = ROOTINFO_KEYLEN;
838 data.size = ROOTINFO_DATALEN;
840 memcpy(buf, ROOTINFO_DATA, ROOTINFO_DATALEN);
841 memcpy(buf + CNID_DID_OFS, &v, sizeof(v));
842 if (dbif_stamp(dbd, buf + CNID_DEV_OFS, CNID_DEV_LEN) < 0)
845 if (dbif_put(dbd, DBIF_CNID, &key, &data, 0) < 0)
852 * Initialize or return CNID database version number
854 * Calls dbif_init_rootinfo if the rootinfo key does not exist yet
856 * @returns -1 on error, version number otherwise
858 int dbif_getversion(DBD *dbd)
864 memset(&key, 0, sizeof(key));
865 memset(&data, 0, sizeof(data));
866 key.data = ROOTINFO_KEY;
867 key.size = ROOTINFO_KEYLEN;
869 switch (dbif_get(dbd, DBIF_CNID, &key, &data, 0)) {
871 memcpy(&version, (char *)data.data + CNID_DID_OFS, sizeof(version));
872 version = ntohl(version);
873 LOG(log_debug, logtype_cnid, "CNID database version %u", version);
876 case 0: /* not found */
877 if (dbif_init_rootinfo(dbd, CNID_VERSION) != 0)
888 * Set CNID database version number
890 * Initializes rootinfo key as neccessary, as does dbif_getversion
891 * @returns -1 on error, version number otherwise
893 int dbif_setversion(DBD *dbd, int version)
899 LOG(log_debug, logtype_cnid, "Setting CNID database version to %u", version);
904 memset(&key, 0, sizeof(key));
905 memset(&data, 0, sizeof(data));
906 key.data = ROOTINFO_KEY;
907 key.size = ROOTINFO_KEYLEN;
909 if ((ret = dbif_get(dbd, DBIF_CNID, &key, &data, 0)) == -1)
912 /* No rootinfo key yet, init it */
913 if (dbif_init_rootinfo(dbd, CNID_VERSION) != 0)
916 if (dbif_get(dbd, DBIF_CNID, &key, &data, 0) == -1)
919 memcpy((char *)data.data + CNID_DID_OFS, &v, sizeof(v));
920 data.size = ROOTINFO_DATALEN;
921 if (dbif_put(dbd, DBIF_CNID, &key, &data, 0) < 0)
928 * Search the database by name
930 * @param resbuf (w) buffer for search results CNIDs, maxsize is assumed to be
931 * DBD_MAX_SRCH_RSLTS * sizefof(cnid_t)
933 * @returns -1 on error, 0 when nothing found, else the number of matches
935 int dbif_search(DBD *dbd, DBT *key, char *resbuf)
941 char *cnids = resbuf;
943 char *namebkp = key->data;
944 int namelenbkp = key->size;
946 memset(&pkey, 0, sizeof(DBT));
947 memset(&data, 0, sizeof(DBT));
950 ret = dbd->db_table[DBIF_IDX_NAME].db->cursor(dbd->db_table[DBIF_IDX_NAME].db,
955 LOG(log_error, logtype_cnid, "Couldn't create cursor: %s", db_strerror(ret));
960 ret = cursorp->pget(cursorp, key, &pkey, &data, DB_SET_RANGE);
961 while (count < DBD_MAX_SRCH_RSLTS && ret != DB_NOTFOUND) {
962 if (!((namelenbkp <= key->size) && (strncmp(namebkp, key->data, namelenbkp) == 0)))
965 memcpy(cnids, pkey.data, sizeof(cnid_t));
966 memcpy(&cnid, pkey.data, sizeof(cnid_t));
967 cnids += sizeof(cnid_t);
968 LOG(log_debug, logtype_cnid, "match: CNID %" PRIu32, ntohl(cnid));
970 ret = cursorp->pget(cursorp, key, &pkey, &data, DB_NEXT);
977 cursorp->close(cursorp);
981 int dbif_txn_begin(DBD *dbd)
985 /* If we already have an active txn, just return */
989 /* If our DBD has no env, just return (-> in memory db) */
990 if (dbd->db_env == NULL)
993 ret = dbd->db_env->txn_begin(dbd->db_env, NULL, &dbd->db_txn, 0);
996 LOG(log_error, logtype_cnid, "error starting transaction: %s", db_strerror(ret));
1002 int dbif_txn_commit(DBD *dbd)
1009 /* If our DBD has no env, just return (-> in memory db) */
1010 if (dbd->db_env == NULL)
1013 ret = dbd->db_txn->commit(dbd->db_txn, 0);
1017 LOG(log_error, logtype_cnid, "error committing transaction: %s", db_strerror(ret));
1023 int dbif_txn_abort(DBD *dbd)
1030 /* If our DBD has no env, just return (-> in memory db) */
1031 if (dbd->db_env == NULL)
1034 ret = dbd->db_txn->abort(dbd->db_txn);
1038 LOG(log_error, logtype_cnid, "error aborting transaction: %s", db_strerror(ret));
1045 ret = 1 -> commit txn
1046 ret = 0 -> abort txn -> exit!
1047 anything else -> exit!
1049 void dbif_txn_close(DBD *dbd, int ret)
1052 if (dbif_txn_abort(dbd) < 0) {
1053 LOG( log_error, logtype_cnid, "Fatal error aborting transaction. Exiting!");
1056 } else if (ret == 1) {
1057 ret = dbif_txn_commit(dbd);
1059 LOG( log_error, logtype_cnid, "Fatal error committing transaction. Exiting!");
1066 int dbif_txn_checkpoint(DBD *dbd, u_int32_t kbyte, u_int32_t min, u_int32_t flags)
1069 ret = dbd->db_env->txn_checkpoint(dbd->db_env, kbyte, min, flags);
1071 LOG(log_error, logtype_cnid, "error checkpointing transaction susystem: %s", db_strerror(ret));
1077 int dbif_count(DBD *dbd, const int dbi, u_int32_t *count)
1081 DB *db = dbd->db_table[dbi].db;
1083 ret = db->stat(db, NULL, &sp, 0);
1086 LOG(log_error, logtype_cnid, "error getting stat infotmation on database: %s", db_strerror(ret));
1090 *count = sp->bt_ndata;
1096 int dbif_copy_rootinfokey(DBD *srcdbd, DBD *destdbd)
1101 memset(&key, 0, sizeof(key));
1102 memset(&data, 0, sizeof(data));
1104 key.data = ROOTINFO_KEY;
1105 key.size = ROOTINFO_KEYLEN;
1107 if ((rc = dbif_get(srcdbd, DBIF_CNID, &key, &data, 0)) <= 0) {
1108 LOG(log_error, logtype_cnid, "dbif_copy_rootinfokey: Error getting rootinfo record");
1112 memset(&key, 0, sizeof(key));
1113 key.data = ROOTINFO_KEY;
1114 key.size = ROOTINFO_KEYLEN;
1116 if ((rc = dbif_put(destdbd, DBIF_CNID, &key, &data, 0))) {
1117 LOG(log_error, logtype_cnid, "dbif_copy_rootinfokey: Error writing rootinfo key");
1124 int dbif_dump(DBD *dbd, int dumpindexes)
1127 uint32_t max = 0, count = 0, cnid, type, did, lastid, version;
1131 DBT key = { 0 }, data = { 0 };
1132 DB *db = dbd->db_table[DBIF_CNID].db;
1133 char *typestring[2] = {"f", "d"};
1136 printf("CNID database dump:\n");
1138 rc = db->cursor(db, NULL, &cur, 0);
1140 LOG(log_error, logtype_cnid, "Couldn't create cursor: %s", db_strerror(rc));
1144 cur->c_get(cur, &key, &data, DB_FIRST);
1146 /* Parse and print data */
1147 memcpy(&cnid, key.data, 4);
1152 /* Rootinfo node ? */
1154 memcpy(&stamp, (char *)data.data + CNID_DEV_OFS, sizeof(time_t));
1155 memcpy(&lastid, (char *)data.data + CNID_TYPE_OFS, CNID_TYPE_LEN);
1156 lastid = ntohl(lastid);
1157 memcpy(&version, (char *)data.data + CNID_DID_OFS, CNID_DID_LEN);
1158 version = ntohl(version);
1160 strftime(timebuf, sizeof(timebuf), "%b %d %Y %H:%M:%S", localtime(&stamp));
1161 printf("CNID db version: %u, dbd stamp: 0x%08x (%s), next free CNID: %u\n",
1162 version, (unsigned int)stamp, timebuf, lastid + 1);
1165 memcpy(&dev, (char *)data.data + CNID_DEV_OFS, 8);
1168 memcpy(&ino, (char *)data.data + CNID_INO_OFS, 8);
1171 memcpy(&type, (char *)data.data + CNID_TYPE_OFS, 4);
1174 memcpy(&did, (char *)data.data + CNID_DID_OFS, 4);
1178 printf("id: %10u, did: %10u, type: %s, dev: 0x%llx, ino: 0x%016llx, name: %s\n",
1179 cnid, did, typestring[type],
1180 (long long unsigned int)dev, (long long unsigned int)ino,
1181 (char *)data.data + CNID_NAME_OFS);
1185 rc = cur->c_get(cur, &key, &data, DB_NEXT);
1188 if (rc != DB_NOTFOUND) {
1189 LOG(log_error, logtype_cnid, "Error iterating over btree: %s", db_strerror(rc));
1193 rc = cur->c_close(cur);
1195 LOG(log_error, logtype_cnid, "Couldn't close cursor: %s", db_strerror(rc));
1198 printf("%u CNIDs in database. Max CNID: %u.\n", count, max);
1200 /* Dump indexes too ? */
1202 /* DBIF_IDX_DEVINO */
1203 printf("\ndev/inode index:\n");
1205 db = dbd->db_table[DBIF_IDX_DEVINO].db;
1206 rc = db->cursor(db, NULL, &cur, 0);
1208 LOG(log_error, logtype_cnid, "Couldn't create cursor: %s", db_strerror(rc));
1212 cur->c_get(cur, &key, &data, DB_FIRST);
1214 /* Parse and print data */
1217 memcpy(&cnid, data.data, CNID_LEN);
1223 memcpy(&dev, key.data, CNID_DEV_LEN);
1226 memcpy(&ino, (char *)key.data + CNID_DEV_LEN, CNID_INO_LEN);
1229 printf("id: %10u <== dev: 0x%llx, ino: 0x%016llx\n",
1230 cnid, (unsigned long long int)dev, (unsigned long long int)ino);
1233 rc = cur->c_get(cur, &key, &data, DB_NEXT);
1235 if (rc != DB_NOTFOUND) {
1236 LOG(log_error, logtype_cnid, "Error iterating over btree: %s", db_strerror(rc));
1240 rc = cur->c_close(cur);
1242 LOG(log_error, logtype_cnid, "Couldn't close cursor: %s", db_strerror(rc));
1245 printf("%u items\n", count);
1247 /* Now dump DBIF_IDX_DIDNAME */
1248 printf("\ndid/name index:\n");
1250 db = dbd->db_table[DBIF_IDX_DIDNAME].db;
1251 rc = db->cursor(db, NULL, &cur, 0);
1253 LOG(log_error, logtype_cnid, "Couldn't create cursor: %s", db_strerror(rc));
1257 cur->c_get(cur, &key, &data, DB_FIRST);
1259 /* Parse and print data */
1262 memcpy(&cnid, data.data, CNID_LEN);
1268 memcpy(&did, key.data, CNID_LEN);
1271 printf("id: %10u <== did: %10u, name: %s\n", cnid, did, (char *)key.data + CNID_LEN);
1274 rc = cur->c_get(cur, &key, &data, DB_NEXT);
1276 if (rc != DB_NOTFOUND) {
1277 LOG(log_error, logtype_cnid, "Error iterating over btree: %s", db_strerror(rc));
1281 rc = cur->c_close(cur);
1283 LOG(log_error, logtype_cnid, "Couldn't close cursor: %s", db_strerror(rc));
1286 printf("%u items\n", count);
1293 Iterates over dbd, returning cnids.
1294 Uses in-value of cnid to seek to that cnid, then gets next and return that in cnid.
1295 If close=1, close cursor.
1296 Return -1 on error, 0 on EOD (end-of-database), 1 if returning cnid.
1298 int dbif_idwalk(DBD *dbd, cnid_t *cnid, int close)
1304 static DBT key = { 0 }, data = { 0 };
1305 DB *db = dbd->db_table[DBIF_CNID].db;
1309 dbd->db_cur->close(dbd->db_cur);
1315 /* An dbif_del will have closed our cursor too */
1316 if ( ! dbd->db_cur ) {
1317 if ((rc = db->cursor(db, NULL, &dbd->db_cur, 0)) != 0) {
1318 LOG(log_error, logtype_cnid, "Couldn't create cursor: %s", db_strerror(rc));
1321 flag = DB_SET_RANGE; /* This will seek to next cnid after the one just deleted */
1324 key.size = sizeof(cnid_t);
1328 if ((rc = dbd->db_cur->get(dbd->db_cur, &key, &data, flag)) == 0) {
1329 memcpy(cnid, key.data, sizeof(cnid_t));
1330 *cnid = ntohl(*cnid);
1334 if (rc != DB_NOTFOUND) {
1335 LOG(log_error, logtype_cnid, "Error iterating over btree: %s", db_strerror(rc));
1336 dbd->db_cur->close(dbd->db_cur);
1342 dbd->db_cur->close(dbd->db_cur);