+++ /dev/null
-/*
- * $Id: cnid_db3_add.c,v 1.5 2009-10-29 13:38:16 didg Exp $
- *
- * Copyright (c) 1999. Adrian Sun (asun@zoology.washington.edu)
- * All Rights Reserved. See COPYRIGHT.
- *
- * cnid_add (db, dev, ino, did, name, hint):
- * add a name to the CNID database. we use both dev/ino and did/name
- * to keep track of things.
- */
-
-#ifdef HAVE_CONFIG_H
-#include "config.h"
-#endif /* HAVE_CONFIG_H */
-
-#ifdef CNID_BACKEND_DB3
-
-#include <stdio.h>
-#include <sys/param.h>
-#include <sys/stat.h>
-#include <string.h>
-#ifdef HAVE_UNISTD_H
-#include <unistd.h>
-#endif /* HAVE_UNISTD_H */
-#ifdef HAVE_FCNTL_H
-#include <fcntl.h>
-#endif /* HAVE_FCNTL_H */
-#include <errno.h>
-#include <atalk/logger.h>
-#ifdef HAVE_SYS_TIME_H
-#include <sys/time.h>
-#endif /* HAVE_SYS_TIME_H */
-
-#ifdef HAVE_DB4_DB_H
-#include <db4/db.h>
-#else
-#include <db.h>
-#endif
-#include <netatalk/endian.h>
-
-#include <atalk/adouble.h>
-#include "cnid_db3.h"
-#include <atalk/util.h>
-
-#include "cnid_db3_private.h"
-
-/* add an entry to the CNID databases. we do this as a transaction
- * to prevent messiness. */
-static int add_cnid(CNID_private *db, DBT *key, DBT *data) {
- DBT altkey, altdata;
- DB_TXN *tid;
- int rc, ret;
-
- memset(&altkey, 0, sizeof(altkey));
- memset(&altdata, 0, sizeof(altdata));
-
-retry:
- if ((rc = db3_txn_begin(db->dbenv, NULL, &tid, 0)) != 0) {
- return rc;
- }
-
- /* main database */
- if ((rc = db->db_cnid->put(db->db_cnid, tid, key, data, DB_NOOVERWRITE))) {
- if (rc == DB_LOCK_DEADLOCK) {
- if ((ret = db3_txn_abort(tid)) != 0) {
- return ret;
- }
- goto retry;
- }
- goto abort;
- }
-
- /* dev/ino database */
- altkey.data = data->data;
- altkey.size = CNID_DEVINO_LEN;
- altdata.data = key->data;
- altdata.size = key->size;
- if ((rc = db->db_devino->put(db->db_devino, tid, &altkey, &altdata, 0))) {
- if (rc == DB_LOCK_DEADLOCK) {
- if ((ret = db3_txn_abort(tid)) != 0) {
- return ret;
- }
- goto retry;
- }
- goto abort;
- }
-
- /* did/name database */
- altkey.data = (char *) data->data + CNID_DEVINO_LEN;
- altkey.size = data->size - CNID_DEVINO_LEN;
- if ((rc = db->db_didname->put(db->db_didname, tid, &altkey, &altdata, 0))) {
- if (rc == DB_LOCK_DEADLOCK) {
- if ((ret = db3_txn_abort(tid)) != 0) {
- return ret;
- }
- goto retry;
- }
- goto abort;
- }
-
- if ((rc = db3_txn_commit(tid, 0)) != 0) {
- LOG(log_error, logtype_default, "add_cnid: Failed to commit transaction: %s", db_strerror(rc));
- return rc;
- }
-
- return 0;
-
-abort:
- if ((ret = db3_txn_abort(tid)) != 0) {
- return ret;
- }
- return rc;
-}
-
-/* ---------------------- */
-static cnid_t get_cnid(CNID_private *db)
-{
- DBT rootinfo_key, rootinfo_data;
- DB_TXN *tid;
- int rc;
- cnid_t hint,id;
-
- memset(&rootinfo_key, 0, sizeof(rootinfo_key));
- memset(&rootinfo_data, 0, sizeof(rootinfo_data));
- rootinfo_key.data = ROOTINFO_KEY;
- rootinfo_key.size = ROOTINFO_KEYLEN;
-
-retry:
- if ((rc = db3_txn_begin(db->dbenv, NULL, &tid, 0)) != 0) {
- LOG(log_error, logtype_default, "cnid_add: Failed to begin transaction: %s", db_strerror(rc));
- errno = CNID_ERR_DB;
- return CNID_INVALID;
- }
- switch (rc = db->db_didname->get(db->db_didname, tid, &rootinfo_key,
- &rootinfo_data, DB_RMW)) {
- case DB_LOCK_DEADLOCK:
- if ((rc = db3_txn_abort(tid)) != 0) {
- LOG(log_error, logtype_default, "cnid_add: txn_abort: %s", db_strerror(rc));
- errno = CNID_ERR_DB;
- return CNID_INVALID;
- }
- goto retry;
- case 0:
- memcpy(&hint, rootinfo_data.data, sizeof(hint));
- id = ntohl(hint);
- /* If we've hit the max CNID allowed, we return a fatal error. CNID
- * needs to be recycled before proceding. */
- if (++id == CNID_INVALID) {
- db3_txn_abort(tid);
- LOG(log_error, logtype_default, "cnid_add: FATAL: Cannot add CNID. CNID database has reached its limit.");
- errno = CNID_ERR_MAX;
- return CNID_INVALID;
- }
- hint = htonl(id);
-/* #ifdef DEBUG */
-#if 0
- LOG(log_debug9, logtype_default, "cnid_add: Found rootinfo for did %u, name %s as %u", ntohl(did), name, ntohl(hint));
-#endif
- break;
- case DB_NOTFOUND:
- hint = htonl(CNID_START);
-/* #ifdef DEBUG */
-#if 0
- LOG(log_debug9, logtype_default, "cnid_add: Using CNID_START for did %u, name %s", ntohl(did), name);
-#endif
- break;
- default:
- LOG(log_error, logtype_default, "cnid_add: Unable to lookup rootinfo: %s", db_strerror(rc));
- goto cleanup_abort;
- }
-
- rootinfo_data.data = &hint;
- rootinfo_data.size = sizeof(hint);
-
- switch (rc = db->db_didname->put(db->db_didname, tid, &rootinfo_key, &rootinfo_data, 0)) {
- case DB_LOCK_DEADLOCK:
- if ((rc = db3_txn_abort(tid)) != 0) {
- LOG(log_error, logtype_default, "cnid_add: txn_abort: %s", db_strerror(rc));
- errno = CNID_ERR_DB;
- return CNID_INVALID;
- }
- goto retry;
- case 0:
- /* The transaction finished, commit it. */
- if ((rc = db3_txn_commit(tid, 0)) != 0) {
- LOG(log_error, logtype_default, "cnid_add: Unable to commit transaction: %s", db_strerror(rc));
- errno = CNID_ERR_DB;
- return CNID_INVALID;
- }
- break;
- default:
- LOG(log_error, logtype_default, "cnid_add: Unable to update rootinfo: %s", db_strerror(rc));
- goto cleanup_abort;
- }
- return hint;
-
-cleanup_abort:
- db3_txn_abort(tid);
- errno = CNID_ERR_DB;
- return CNID_INVALID;
-}
-
-/* ------------------------ */
-cnid_t cnid_db3_add(struct _cnid_db *cdb, const struct stat *st,
- const cnid_t did, char *name, const size_t len,
- cnid_t hint)
-{
- CNID_private *db;
- DBT key, data;
- cnid_t id;
- int rc;
-
- if (!cdb || !(db = cdb->_private) || !st || !name) {
- errno = CNID_ERR_PARAM;
- return CNID_INVALID;
- }
-
- /* Do a lookup. */
- id = cnid_db3_lookup(cdb, st, did, name, len);
- /* ... Return id if it is valid, or if Rootinfo is read-only. */
- if (id || (db->flags & CNIDFLAG_DB_RO)) {
-#ifdef DEBUG
- LOG(log_debug9, logtype_default, "cnid_add: Looked up did %u, name %s as %u", ntohl(did), name, ntohl(id));
-#endif
- return id;
- }
-
- /* Initialize our DBT data structures. */
- memset(&key, 0, sizeof(key));
- memset(&data, 0, sizeof(data));
-
- /* Just tickle hint, and the key will change (gotta love pointers). */
- key.data = &hint;
- key.size = sizeof(hint);
-
- if ((data.data = make_cnid_data(st, did, name, len)) == NULL) {
- LOG(log_error, logtype_default, "cnid_add: Path name is too long");
- errno = CNID_ERR_PATH;
- return CNID_INVALID;
- }
-
- data.size = CNID_HEADER_LEN + len + 1;
-
- /* Start off with the hint. It should be in network byte order.
- * We need to make sure that somebody doesn't add in restricted
- * cnid's to the database. */
- if (ntohl(hint) >= CNID_START) {
- /* If the key doesn't exist, add it in. Don't fiddle with nextID. */
- rc = add_cnid(db, &key, &data);
- switch (rc) {
- case DB_KEYEXIST: /* Need to use RootInfo after all. */
- break;
- default:
- LOG(log_error, logtype_default, "cnid_add: Unable to add CNID %u: %s", ntohl(hint), db_strerror(rc));
- errno = CNID_ERR_DB;
- return CNID_INVALID;
- case 0:
-#ifdef DEBUG
- LOG(log_debug9, logtype_default, "cnid_add: Used hint for did %u, name %s as %u", ntohl(did), name, ntohl(hint));
-#endif
- return hint;
- }
- }
- hint = get_cnid(db);
- if (hint == 0) {
- errno = CNID_ERR_DB;
- return CNID_INVALID;
- }
-
- /* Now we need to add the CNID data to the databases. */
- rc = add_cnid(db, &key, &data);
- if (rc) {
- LOG(log_error, logtype_default, "cnid_add: Failed to add CNID for %s to database using hint %u: %s", name, ntohl(hint), db_strerror(rc));
- errno = CNID_ERR_DB;
- return CNID_INVALID;
- }
-
-#ifdef DEBUG
- LOG(log_debug9, logtype_default, "cnid_add: Returned CNID for did %u, name %s as %u", ntohl(did), name, ntohl(hint));
-#endif
-
- return hint;
-}
-
-#endif /* CNID_BACKEND_DB3 */
projects / netatalk.git / blobdiff