2 * $Id: cnid_index.c,v 1.1.2.1 2004-12-21 13:36:11 didg Exp $
4 * All Rights Reserved. See COPYING.
9 #endif /* HAVE_CONFIG_H */
13 #endif /* HAVE_UNISTD_H */
16 #endif /* HAVE_FCNTL_H */
22 #ifdef HAVE_SYS_TYPES_H
23 #include <sys/types.h>
24 #endif /* HAVE_SYS_TYPES_H */
25 #include <sys/param.h>
26 #ifdef HAVE_SYS_STAT_H
28 #endif /* HAVE_SYS_STAT_H */
32 #include <netatalk/endian.h>
33 #include <atalk/cnid_dbd_private.h>
34 #include <atalk/logger.h>
36 #define LOCKFILENAME "lock"
37 static int exit_sig = 0;
39 /* Copy and past from etc/cnid_dbd/dbif.c */
42 #define DB_ERRLOGFILE "db_errlog"
44 static DB_ENV *db_env = NULL;
45 static DB_TXN *db_txn = NULL;
46 static FILE *db_errlog = NULL;
48 #ifdef CNID_BACKEND_DBD_TXN
49 #define DBOPTIONS (DB_CREATE | DB_INIT_LOCK | DB_INIT_LOG | DB_INIT_MPOOL | DB_INIT_TXN)
51 #define DBOPTIONS (DB_CREATE | DB_INIT_CDB | DB_INIT_MPOOL | DB_PRIVATE)
56 #define DBIF_IDX_CNID 0
57 #define DBIF_IDX_DEVINO 1
58 #define DBIF_IDX_DIDNAME 2
60 static struct db_table {
63 u_int32_t general_flags;
67 { "cnid2.db", NULL, 0, DB_BTREE},
68 { "devino.db", NULL, 0, DB_BTREE},
69 { "didname.db", NULL, 0, DB_BTREE},
72 static char *old_dbfiles[] = {"cnid.db", NULL};
75 int didname(dbp, pkey, pdata, skey)
77 const DBT *pkey, *pdata;
82 memset(skey, 0, sizeof(DBT));
83 skey->data = (char *)pdata->data + CNID_DID_OFS;
84 /* FIXME: At least DB 4.0.14 and 4.1.25 pass in the correct length of
85 pdata.size. strlen is therefore not needed. Also, skey should be zeroed
87 len = strlen((char *)skey->data + CNID_DID_LEN);
88 skey->size = CNID_DID_LEN + len + 1;
93 int devino(dbp, pkey, pdata, skey)
95 const DBT *pkey, *pdata;
98 memset(skey, 0, sizeof(DBT));
99 skey->data = (char *)pdata->data + CNID_DEVINO_OFS;
100 skey->size = CNID_DEVINO_LEN;
104 /* --------------- */
105 static int db_compat_associate (DB *p, DB *s,
106 int (*callback)(DB *, const DBT *,const DBT *, DBT *),
109 #if DB_VERSION_MAJOR > 4 || (DB_VERSION_MAJOR == 4 && DB_VERSION_MINOR >= 1)
110 return p->associate(p, db_txn, s, callback, flags);
112 #if (DB_VERSION_MAJOR == 4 && DB_VERSION_MINOR == 0)
113 return p->associate(p, s, callback, flags);
120 /* --------------- */
121 static int db_compat_open(DB *db, char *file, char *name, DBTYPE type, int mode)
125 #if DB_VERSION_MAJOR > 4 || (DB_VERSION_MAJOR == 4 && DB_VERSION_MINOR >= 1)
126 ret = db->open(db, db_txn, file, name, type, DB_CREATE, mode);
128 ret = db->open(db, file, name, type, DB_CREATE, mode);
132 LOG(log_error, logtype_cnid, "error opening database %s: %s", name, db_strerror(ret));
139 /* --------------- */
140 static int dbif_open(int do_truncate)
146 for (i = 0; i != DBIF_DB_CNT; i++) {
147 if ((ret = db_create(&(db_table[i].db), db_env, 0))) {
148 LOG(log_error, logtype_cnid, "error creating handle for database %s: %s",
149 db_table[i].name, db_strerror(ret));
153 if (db_table[i].general_flags) {
154 if ((ret = db_table[i].db->set_flags(db_table[i].db, db_table[i].general_flags))) {
155 LOG(log_error, logtype_cnid, "error setting flags for database %s: %s",
156 db_table[i].name, db_strerror(ret));
161 if (db_compat_open(db_table[i].db, db_table[0].name, db_table[i].name, db_table[i].type, 0664) < 0)
163 if (db_errlog != NULL)
164 db_table[i].db->set_errfile(db_table[i].db, db_errlog);
166 if (do_truncate && i > 0) {
167 if ((ret = db_table[i].db->truncate(db_table[i].db, db_txn, &count, 0))) {
168 LOG(log_error, logtype_cnid, "error truncating database %s: %s",
169 db_table[i].name, db_strerror(ret));
175 /* TODO: Implement CNID DB versioning info on new databases. */
176 /* TODO: Make transaction support a runtime option. */
177 /* Associate the secondary with the primary. */
178 if ((ret = db_compat_associate(db_table[0].db, db_table[DBIF_IDX_DIDNAME].db, didname, (do_truncate)?DB_CREATE:0)) != 0) {
179 LOG(log_error, logtype_cnid, "Failed to associate didname database: %s",db_strerror(ret));
183 if ((ret = db_compat_associate(db_table[0].db, db_table[DBIF_IDX_DEVINO].db, devino, (do_truncate)?DB_CREATE:0)) != 0) {
184 LOG(log_error, logtype_cnid, "Failed to associate devino database: %s",db_strerror(ret));
191 /* ------------------------ */
192 static int dbif_closedb(void)
198 for (i = DBIF_DB_CNT -1; i >= 0; i--) {
199 if (db_table[i].db != NULL && (ret = db_table[i].db->close(db_table[i].db, 0))) {
200 LOG(log_error, logtype_cnid, "error closing database %s: %s", db_table[i].name, db_strerror(ret));
209 /* ------------------------ */
210 static int dbif_close(void)
218 if (db_env != NULL && (ret = db_env->close(db_env, 0))) {
219 LOG(log_error, logtype_cnid, "error closing DB environment: %s", db_strerror(ret));
222 if (db_errlog != NULL && fclose(db_errlog) == EOF) {
223 LOG(log_error, logtype_cnid, "error closing DB logfile: %s", strerror(errno));
230 /* --------------- */
231 static int upgrade_required(void)
237 for (i = 0; old_dbfiles[i] != NULL; i++) {
238 if ( !(stat(old_dbfiles[i], &st) < 0) ) {
242 if (errno != ENOENT) {
243 LOG(log_error, logtype_cnid, "cnid_open: Checking %s gave %s", old_dbfiles[i], strerror(errno));
250 /* -------------------------- */
251 static int dbif_sync(void)
257 for (i = 0; i != /* DBIF_DB_CNT*/ 1; i++) {
258 if ((ret = db_table[i].db->sync(db_table[i].db, 0))) {
259 LOG(log_error, logtype_cnid, "error syncing database %s: %s", db_table[i].name, db_strerror(ret));
270 /* -------------------------- */
271 static int dbif_count(const int dbi, u_int32_t *count)
275 DB *db = db_table[dbi].db;
277 ret = db->stat(db, &sp, 0);
280 LOG(log_error, logtype_cnid, "error getting stat infotmation on database: %s", db_strerror(errno));
284 *count = sp->bt_ndata;
290 /* -------------------------- */
291 static int dbd_check(char *dbdir)
293 u_int32_t c_didname = 0, c_devino = 0, c_cnid = 0;
295 LOG(log_debug, logtype_cnid, "CNID database at `%s' is being checked (quick)", dbdir);
297 if (dbif_count(DBIF_IDX_CNID, &c_cnid))
300 if (dbif_count(DBIF_IDX_DIDNAME, &c_didname))
303 /* bailout after the first error */
304 if (dbif_count(DBIF_IDX_DEVINO, &c_devino))
307 if ( c_cnid != c_devino) {
308 LOG(log_error, logtype_cnid, "CNID database at `%s' corrupted (%u/%u)", dbdir, c_cnid, c_devino);
312 if ( c_cnid != c_didname) {
313 LOG(log_error, logtype_cnid, "CNID database at `%s' corrupted (%u/%u)", dbdir, c_cnid, c_didname);
317 LOG(log_debug, logtype_cnid, "CNID database at `%s' seems ok, %u entries.", dbdir, c_cnid);
321 /* --------------- */
323 * We assume our current directory is already the BDB homedir. Otherwise
324 * opening the databases will not work as expected. If we use transactions,
325 * dbif_env_init(), dbif_close() and dbif_stamp() are the only interface
326 * functions that can be called without a valid transaction handle in db_txn.
328 static int dbif_env_init(void)
331 #ifdef CNID_BACKEND_DBD_TXN
332 char **logfiles = NULL;
336 /* Refuse to do anything if this is an old version of the CNID database */
337 if (upgrade_required()) {
338 LOG(log_error, logtype_cnid, "Found version 1 of the CNID database. Please upgrade to version 2");
342 if ((db_errlog = fopen(DB_ERRLOGFILE, "a")) == NULL)
343 LOG(log_warning, logtype_cnid, "error creating/opening DB errlogfile: %s", strerror(errno));
345 #ifdef CNID_BACKEND_DBD_TXN
346 if ((ret = db_env_create(&db_env, 0))) {
347 LOG(log_error, logtype_cnid, "error creating DB environment: %s",
352 if (db_errlog != NULL)
353 db_env->set_errfile(db_env, db_errlog);
354 db_env->set_verbose(db_env, DB_VERB_RECOVERY, 1);
355 db_env->set_verbose(db_env, DB_VERB_CHKPOINT, 1);
356 if ((ret = db_env->open(db_env, ".", DBOPTIONS | DB_PRIVATE | DB_RECOVER, 0))) {
357 LOG(log_error, logtype_cnid, "error opening DB environment: %s",
359 db_env->close(db_env, 0);
364 if (db_errlog != NULL)
367 if ((ret = db_env->close(db_env, 0))) {
368 LOG(log_error, logtype_cnid, "error closing DB environment after recovery: %s",
374 if ((ret = db_env_create(&db_env, 0))) {
375 LOG(log_error, logtype_cnid, "error creating DB environment after recovery: %s",
380 if (db_errlog != NULL)
381 db_env->set_errfile(db_env, db_errlog);
382 if ((ret = db_env->open(db_env, ".", DBOPTIONS , 0))) {
383 LOG(log_error, logtype_cnid, "error opening DB environment after recovery: %s",
385 db_env->close(db_env, 0);
390 #ifdef CNID_BACKEND_DBD_TXN
391 if (db_env->log_archive(db_env, &logfiles, 0)) {
392 LOG(log_error, logtype_cnid, "error getting list of stale logfiles: %s",
394 db_env->close(db_env, 0);
398 if (logfiles != NULL) {
399 for (file = logfiles; *file != NULL; file++) {
400 if (unlink(*file) < 0)
401 LOG(log_warning, logtype_cnid, "Error removing stale logfile %s: %s", *file, strerror(errno));
410 static void sig_exit(int signo)
416 static void block_sigs_onoff(int block)
421 sigaddset(&set, SIGINT);
422 sigaddset(&set, SIGTERM);
424 sigprocmask(SIG_BLOCK, &set, NULL);
426 sigprocmask(SIG_UNBLOCK, &set, NULL);
430 /* ------------------------ */
436 if ((lockfd = open(LOCKFILENAME, O_RDWR | O_CREAT, 0644)) < 0) {
437 LOG(log_error, logtype_cnid, "main: error opening lockfile: %s", strerror(errno));
442 lock.l_whence = SEEK_SET;
444 lock.l_type = F_WRLCK;
446 if (fcntl(lockfd, F_SETLK, &lock) < 0) {
447 if (errno == EACCES || errno == EAGAIN) {
450 LOG(log_error, logtype_cnid, "main: fcntl F_WRLCK lockfile: %s", strerror(errno));
458 /* ----------------------- */
459 void set_signal(void)
463 sv.sa_handler = sig_exit;
465 sigemptyset(&sv.sa_mask);
466 sigaddset(&sv.sa_mask, SIGINT);
467 sigaddset(&sv.sa_mask, SIGTERM);
468 if (sigaction(SIGINT, &sv, NULL) < 0 || sigaction(SIGTERM, &sv, NULL) < 0) {
469 LOG(log_error, logtype_cnid, "main: sigaction: %s", strerror(errno));
472 sv.sa_handler = SIG_IGN;
473 sigemptyset(&sv.sa_mask);
474 if (sigaction(SIGPIPE, &sv, NULL) < 0) {
475 LOG(log_error, logtype_cnid, "main: sigaction: %s", strerror(errno));
480 /* ----------------------- */
481 void free_lock(int lockfd)
486 lock.l_whence = SEEK_SET;
488 lock.l_type = F_UNLCK;
489 fcntl(lockfd, F_SETLK, &lock);
493 /* ------------------------ */
494 int main(int argc, char *argv[])
501 set_processname("cnid_index");
502 syslog_setup(log_debug, logtype_default, logoption_ndelay | logoption_pid, logfacility_daemon);
505 LOG(log_error, logtype_cnid, "main: not enough arguments");
511 if (chdir(dir) < 0) {
512 LOG(log_error, logtype_cnid, "chdir to %s failed: %s", dir, strerror(errno));
516 /* Before we do anything else, check if there is an instance of cnid_dbd
517 running already and silently exit if yes. */
520 LOG(log_info, logtype_cnid, "Startup, DB dir %s", dir);
524 /* SIGINT and SIGTERM are always off, unless we get a return code of 0 from
525 comm_rcv (no requests for one second, see above in loop()). That means we
526 only shut down after one second of inactivity. */
529 if (dbif_env_init() < 0)
530 exit(2); /* FIXME: same exit code as failure for dbif_open() */
532 #ifdef CNID_BACKEND_DBD_TXN
533 if (dbif_txn_begin() < 0)
537 if (dbif_open(0) < 0) {
538 #ifdef CNID_BACKEND_DBD_TXN
545 #ifndef CNID_BACKEND_DBD_TXN
546 if ((ret = dbd_check(dir))) {
552 LOG(log_info, logtype_cnid, "main: re-opening, secondaries will be rebuilt. This may take some time");
553 if (dbif_open(1) < 0) {
554 LOG(log_info, logtype_cnid, "main: re-opening databases failed");
558 LOG(log_info, logtype_cnid, "main: rebuilt done");
562 #ifdef CNID_BACKEND_DBD_TXN
563 if (dbif_txn_commit() < 0)
567 #ifndef CNID_BACKEND_DBD_TXN
568 /* FIXME: Do we really need to sync before closing the DB? Just closing it
574 if (dbif_close() < 0)