1
0
mirror of https://github.com/samba-team/samba.git synced 2024-12-24 21:34:56 +03:00
samba-mirror/source3/lib/sessionid_tdb.c
2011-05-06 10:48:10 +02:00

143 lines
3.7 KiB
C

/*
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"
#include "system/filesys.h"
#include "dbwrap.h"
#include "session.h"
#include "util_tdb.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|TDB_INCOMPATIBLE_HASH,
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);
}