s3: Abstract access to sessionid.tdb, similar to conn_tdb.c

1 files changed, 138 insertions, 0 deletions

diff --git a/source3/lib/sessionid_tdb.c b/source3/lib/sessionid_tdb.c
new file mode 100644
index 0000000000..6efbafd46d
--- /dev/null
+++ b/source3/lib/sessionid_tdb.c
@@ -0,0 +1,138 @@
+/*
+   Unix SMB/CIFS implementation.
+   Low-level sessionid.tdb access functions
+   Copyright (C) Volker Lendecke 2010
+
+   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"
+
+static struct db_context *session_db_ctx(void)
+{
+	static struct db_context *session_db_ctx_ptr;
+
+	if (session_db_ctx_ptr != NULL) {
+		return session_db_ctx_ptr;
+	}
+
+	session_db_ctx_ptr = db_open(NULL, lock_path("sessionid.tdb"), 0,
+				     TDB_CLEAR_IF_FIRST|TDB_DEFAULT,
+				     O_RDWR | O_CREAT, 0644);
+	return session_db_ctx_ptr;
+}
+
+bool sessionid_init(void)
+{
+	if (session_db_ctx() == NULL) {
+		DEBUG(1,("session_init: failed to open sessionid tdb\n"));
+		return False;
+	}
+
+	return True;
+}
+
+struct db_record *sessionid_fetch_record(TALLOC_CTX *mem_ctx, const char *key)
+{
+	struct db_context *db;
+
+	db = session_db_ctx();
+	if (db == NULL) {
+		return NULL;
+	}
+	return db->fetch_locked(db, mem_ctx, string_term_tdb_data(key));
+}
+
+struct sessionid_traverse_state {
+	int (*fn)(struct db_record *rec, const char *key,
+		  struct sessionid *session, void *private_data);
+	void *private_data;
+};
+
+static int sessionid_traverse_fn(struct db_record *rec, void *private_data)
+{
+	struct sessionid_traverse_state *state =
+		(struct sessionid_traverse_state *)private_data;
+	struct sessionid session;
+
+	if ((rec->key.dptr[rec->key.dsize-1] != '\0')
+	    || (rec->value.dsize != sizeof(struct sessionid))) {
+		DEBUG(1, ("Found invalid record in sessionid.tdb\n"));
+		return 0;
+	}
+
+	memcpy(&session, rec->value.dptr, sizeof(session));
+
+	return state->fn(rec, (char *)rec->key.dptr, &session,
+			 state->private_data);
+}
+
+int sessionid_traverse(int (*fn)(struct db_record *rec, const char *key,
+				 struct sessionid *session,
+				 void *private_data),
+		       void *private_data)
+{
+	struct db_context *db;
+	struct sessionid_traverse_state state;
+
+	db = session_db_ctx();
+	if (db == NULL) {
+		return -1;
+	}
+	state.fn = fn;
+	state.private_data = private_data;
+	return db->traverse(db, sessionid_traverse_fn, &state);
+}
+
+struct sessionid_traverse_read_state {
+	int (*fn)(const char *key, struct sessionid *session,
+		  void *private_data);
+	void *private_data;
+};
+
+static int sessionid_traverse_read_fn(struct db_record *rec,
+				      void *private_data)
+{
+	struct sessionid_traverse_read_state *state =
+		(struct sessionid_traverse_read_state *)private_data;
+	struct sessionid session;
+
+	if ((rec->key.dptr[rec->key.dsize-1] != '\0')
+	    || (rec->value.dsize != sizeof(struct sessionid))) {
+		DEBUG(1, ("Found invalid record in sessionid.tdb\n"));
+		return 0;
+	}
+
+	memcpy(&session, rec->value.dptr, sizeof(session));
+
+	return state->fn((char *)rec->key.dptr, &session,
+			 state->private_data);
+}
+
+int sessionid_traverse_read(int (*fn)(const char *key,
+				      struct sessionid *session,
+				      void *private_data),
+			    void *private_data)
+{
+	struct db_context *db;
+	struct sessionid_traverse_read_state state;
+
+	db = session_db_ctx();
+	if (db == NULL) {
+		return -1;
+	}
+	state.fn = fn;
+	state.private_data = private_data;
+	return db->traverse(db, sessionid_traverse_read_fn, &state);
+}