r24645: add in the ctdb dbwrap backend

(This used to be commit e2d173fde17ab903bddf2139d8100785280856bd)

diff --git a/source4/lib/dbwrap/config.mk b/source4/lib/dbwrap/config.mk
index e2df04ac966d4574cef6362710f27882b2bfc83c..01ce4ddd8e6a65da8c753dae3f32a4f7cc619e8d 100644 (file)
--- a/source4/lib/dbwrap/config.mk
+++ b/source4/lib/dbwrap/config.mk
@@ -1,5 +1,6 @@
 [SUBSYSTEM::LIBDBWRAP]
 OBJ_FILES = dbwrap.o \
-               dbwrap_tdb.o
+               dbwrap_tdb.o \
+               dbwrap_ctdb.o
 PUBLIC_DEPENDENCIES = \
-               LIBTDB
+               LIBTDB ctdb

diff --git a/source4/lib/dbwrap/dbwrap.c b/source4/lib/dbwrap/dbwrap.c
index e853967aa294b8bba8ab86fa301bee87e9f52bd7..c0bac672aa9b24765f1f86cbdfe030b004a0fa9c 100644 (file)
--- a/source4/lib/dbwrap/dbwrap.c
+++ b/source4/lib/dbwrap/dbwrap.c
@@ -29,5 +29,10 @@
  */
 struct db_context *db_tmp_open(TALLOC_CTX *mem_ctx, const char *name, int tdb_flags)
 {
+       if (lp_parm_bool(-1, "ctdb", "enable", False) &&
+           lp_parm_bool(-1, "ctdb", name, True)) {
+                   return db_tmp_open_ctdb(mem_ctx, name, tdb_flags);
+       }
+
        return db_tmp_open_tdb(mem_ctx, name, tdb_flags);
 }

diff --git a/source4/lib/dbwrap/dbwrap.h b/source4/lib/dbwrap/dbwrap.h
index 4b9654b8cf580f164a9d05909a7d88cfaf10cd4b..b4267d802dff8574e9d1f0a4d85712ae1920543b 100644 (file)
--- a/source4/lib/dbwrap/dbwrap.h
+++ b/source4/lib/dbwrap/dbwrap.h
@@ -50,4 +50,4 @@ struct db_context *db_tmp_open(TALLOC_CTX *mem_ctx, const char *name, int tdb_fl
 
 /* backends */
 struct db_context *db_tmp_open_tdb(TALLOC_CTX *mem_ctx, const char *name, int tdb_flags);
-
+struct db_context *db_tmp_open_ctdb(TALLOC_CTX *mem_ctx, const char *name, int tdb_flags);

diff --git a/source4/lib/dbwrap/dbwrap_ctdb.c b/source4/lib/dbwrap/dbwrap_ctdb.c
new file mode 100644 (file)
index 0000000..f635af0
--- /dev/null
+++ b/source4/lib/dbwrap/dbwrap_ctdb.c
@@ -0,0 +1,191 @@
+/* 
+   Unix SMB/CIFS implementation.
+
+   Database interface wrapper around ctdbd
+
+   Copyright (C) Andrew Tridgell 2007
+   
+   This program is free software; you can redistribute it and/or modify
+   it under the terms of the GNU General Public License as published by
+   the Free Software Foundation; either version 3 of the License, or
+   (at your option) any later version.
+   
+   This program is distributed in the hope that it will be useful,
+   but WITHOUT ANY WARRANTY; without even the implied warranty of
+   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+   GNU General Public License for more details.
+   
+   You should have received a copy of the GNU General Public License
+   along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#include "includes.h"
+#include "lib/tdb/include/tdb.h"
+#include "lib/dbwrap/dbwrap.h"
+#include "cluster/cluster.h"
+#include "cluster/ctdb/include/ctdb.h"
+
+static NTSTATUS db_ctdb_store(struct db_record *rec, TDB_DATA data, int flag)
+{
+       struct ctdb_record_handle *h = talloc_get_type(rec->private_data, struct ctdb_record_handle);
+       int ret;
+
+       ret = ctdb_record_store(h, data);
+       if (ret != 0) {
+               return NT_STATUS_INTERNAL_DB_CORRUPTION;
+       }
+       return NT_STATUS_OK;
+}
+
+static NTSTATUS db_ctdb_delete(struct db_record *rec)
+{
+       return rec->store(rec, tdb_null, TDB_REPLACE);
+}
+
+
+static struct db_record *db_ctdb_fetch_locked(struct db_context *db,
+                                             TALLOC_CTX *mem_ctx,
+                                             TDB_DATA key)
+{
+       struct db_record *rec;
+       struct ctdb_record_handle *h;
+       struct ctdb_db_context *cdb = talloc_get_type(db->private_data, struct ctdb_db_context);
+
+       rec = talloc(mem_ctx, struct db_record);
+       if (!rec) return NULL;
+
+       h = ctdb_fetch_lock(cdb, rec, key, &rec->value);
+       if (h == NULL) {
+               talloc_free(rec);
+               return NULL;
+       }
+
+       rec->private_data = h;
+       rec->store        = db_ctdb_store;
+       rec->delete_rec   = db_ctdb_delete;
+
+       return rec;
+}
+
+/*
+  fetch (unlocked, no migration) operation on ctdb
+ */
+static int db_ctdb_fetch(struct db_context *db, TALLOC_CTX *mem_ctx,
+                        TDB_DATA key, TDB_DATA *data)
+{
+       struct ctdb_db_context *cdb = talloc_get_type(db->private_data, struct ctdb_db_context);
+
+       return ctdb_fetch(cdb, mem_ctx, key, data);
+}
+
+struct traverse_state {
+       struct db_context *db;
+       int (*fn)(struct db_record *rec, void *private_data);
+       void *private_data;
+};
+
+static int traverse_callback(struct ctdb_context *ctdb, TDB_DATA key, TDB_DATA data, void *private_data)
+{
+       struct traverse_state *state = (struct traverse_state *)private_data;
+       struct db_record *rec;
+       TALLOC_CTX *tmp_ctx = talloc_new(state->db);
+       /* we have to give them a locked record to prevent races */
+       rec = db_ctdb_fetch_locked(state->db, tmp_ctx, key);
+       if (rec && rec->value.dsize > 0) {
+               state->fn(rec, state->private_data);
+       }
+       talloc_free(tmp_ctx);
+       return 0;
+}
+
+static int db_ctdb_traverse(struct db_context *db,
+                           int (*fn)(struct db_record *rec, void *private_data),
+                           void *private_data)
+{
+       struct ctdb_db_context *cdb = talloc_get_type(db->private_data, struct ctdb_db_context);
+       struct traverse_state state;
+
+       state.db = db;
+       state.fn = fn;
+       state.private_data = private_data;
+
+       ctdb_traverse(cdb, traverse_callback, &state);
+       return 0;
+}
+
+static NTSTATUS db_ctdb_store_deny(struct db_record *rec, TDB_DATA data, int flag)
+{
+       return NT_STATUS_MEDIA_WRITE_PROTECTED;
+}
+
+static NTSTATUS db_ctdb_delete_deny(struct db_record *rec)
+{
+       return NT_STATUS_MEDIA_WRITE_PROTECTED;
+}
+
+static int traverse_read_callback(struct ctdb_context *ctdb, 
+                                 TDB_DATA key, TDB_DATA data, void *private_data)
+{
+       struct traverse_state *state = (struct traverse_state *)private_data;
+       struct db_record rec;
+       rec.key = key;
+       rec.value = data;
+       rec.store = db_ctdb_store_deny;
+       rec.delete_rec = db_ctdb_delete_deny;
+       rec.private_data = state->db;
+       state->fn(&rec, state->private_data);
+       return 0;
+}
+
+static int db_ctdb_traverse_read(struct db_context *db,
+                                int (*fn)(struct db_record *rec,
+                                          void *private_data),
+                                void *private_data)
+{
+       struct traverse_state state;
+       struct ctdb_db_context *cdb = talloc_get_type(db->private_data, struct ctdb_db_context);
+
+       state.db = db;
+       state.fn = fn;
+       state.private_data = private_data;
+
+       ctdb_traverse(cdb, traverse_read_callback, &state);
+       return 0;
+}
+
+static int db_ctdb_get_seqnum(struct db_context *db)
+{
+       DEBUG(0,("ctdb_get_seqnum not implemented\n"));
+       return -1;
+}
+
+struct db_context *db_tmp_open_ctdb(TALLOC_CTX *mem_ctx, const char *name, int tdb_flags)
+{
+       struct db_context *db;
+       struct ctdb_context *ctdb = talloc_get_type(cluster_backend_handle(), 
+                                                   struct ctdb_context);
+       struct ctdb_db_context *cdb;
+
+       db = talloc_zero(mem_ctx, struct db_context);
+       if (db == NULL) {
+               return NULL;
+       }
+
+       cdb = ctdb_attach(ctdb, name);
+       if (!cdb) {
+               DEBUG(0,("Failed to attach to ctdb database '%s'\n", name));
+               talloc_free(db);
+               return NULL;
+       }
+
+       db->private_data  = cdb;
+       db->fetch_locked  = db_ctdb_fetch_locked;
+       db->fetch         = db_ctdb_fetch;
+       db->traverse      = db_ctdb_traverse;
+       db->traverse_read = db_ctdb_traverse_read;
+       db->get_seqnum    = db_ctdb_get_seqnum;
+
+       DEBUG(3,("db_tmp_open_ctdb: opened database '%s'\n", name));
+
+       return db;
+}