/* Copyright (C) 2005 Red Hat, Inc. */

#include <semanage/handle.h>

#include "semanage_store.h"

#include "semanage_conf.h"

#include "database.h"

#include "debug.h"

static int assert_init(semanage_handle_t * handle, dbase_config_t * dconfig)

{

	if (dconfig->dtable == NULL) {

		ERR(handle,

		    "A direct or server connection is needed "

		    "to use this function - please call "

		    "the corresponding connect() method");

		return STATUS_ERR;

	}

	return STATUS_SUCCESS;

}

static int enter_ro(semanage_handle_t * handle, dbase_config_t * dconfig)

{

	if (assert_init(handle, dconfig) < 0)

		goto err;

	if (!handle->is_in_transaction &&

	    handle->conf->store_type == SEMANAGE_CON_DIRECT) {

		if (semanage_get_active_lock(handle) < 0) {

			ERR(handle, "could not get the active lock");

			goto err;

		}

	}

	if (dconfig->dtable->cache(handle, dconfig->dbase) < 0)

		goto err;

	return STATUS_SUCCESS;

      err:

	ERR(handle, "could not enter read-only section");

	return STATUS_ERR;

}

static inline int exit_ro(semanage_handle_t * handle)

{

	int commit_num = handle->funcs->get_serial(handle);

	if (!handle->is_in_transaction &&

	    handle->conf->store_type == SEMANAGE_CON_DIRECT)

		semanage_release_active_lock(handle);

	return commit_num;

}

static int enter_rw(semanage_handle_t * handle, dbase_config_t * dconfig)

{

	if (assert_init(handle, dconfig) < 0)

		goto err;

	if (!handle->is_in_transaction) {

		ERR(handle, "this operation requires a transaction");

		goto err;

	}

	if (dconfig->dtable->cache(handle, dconfig->dbase) < 0)

		goto err;

	return STATUS_SUCCESS;

      err:

	ERR(handle, "could not enter read-write section");

	return STATUS_ERR;

}

int dbase_modify(semanage_handle_t * handle,

		 dbase_config_t * dconfig,

		 const record_key_t * key, const record_t * data)

{

	if (enter_rw(handle, dconfig) < 0)

		return STATUS_ERR;

	if (dconfig->dtable->modify(handle, dconfig->dbase, key, data) < 0)

		return STATUS_ERR;

	return STATUS_SUCCESS;

}

int dbase_set(semanage_handle_t * handle,

	      dbase_config_t * dconfig,

	      const record_key_t * key, const record_t * data)

{

	if (enter_rw(handle, dconfig) < 0)

		return STATUS_ERR;

	if (dconfig->dtable->set(handle, dconfig->dbase, key, data) < 0)

		return STATUS_ERR;

	return STATUS_SUCCESS;

}

int dbase_del(semanage_handle_t * handle,

	      dbase_config_t * dconfig, const record_key_t * key)

{

	if (enter_rw(handle, dconfig) < 0)

		return STATUS_ERR;

	if (dconfig->dtable->del(handle, dconfig->dbase, key) < 0)

		return STATUS_ERR;

	return STATUS_SUCCESS;

}

int dbase_query(semanage_handle_t * handle,

		dbase_config_t * dconfig,

		const record_key_t * key, record_t ** response)

{

	if (enter_ro(handle, dconfig) < 0)

		return STATUS_ERR;

	if (dconfig->dtable->query(handle, dconfig->dbase, key, response) < 0) {

		exit_ro(handle);

		return STATUS_ERR;

	}

	return exit_ro(handle);

}

int dbase_exists(semanage_handle_t * handle,

		 dbase_config_t * dconfig,

		 const record_key_t * key, int *response)

{

	if (enter_ro(handle, dconfig) < 0)

		return STATUS_ERR;

	if (dconfig->dtable->exists(handle, dconfig->dbase, key, response) < 0) {

		exit_ro(handle);

		return STATUS_ERR;

	}

	return exit_ro(handle);

}

int dbase_count(semanage_handle_t * handle,

		dbase_config_t * dconfig, unsigned int *response)

{

	if (enter_ro(handle, dconfig) < 0)

		return STATUS_ERR;

	if (dconfig->dtable->count(handle, dconfig->dbase, response) < 0) {

		exit_ro(handle);

		return STATUS_ERR;

	}

	return exit_ro(handle);

}

int dbase_iterate(semanage_handle_t * handle,

		  dbase_config_t * dconfig,

		  int (*fn) (const record_t * record,

			     void *fn_arg), void *fn_arg)

{

	if (enter_ro(handle, dconfig) < 0)

		return STATUS_ERR;

	if (dconfig->dtable->iterate(handle, dconfig->dbase, fn, fn_arg) < 0) {

		exit_ro(handle);

		return STATUS_ERR;

	}

	return exit_ro(handle);

}

int dbase_list(semanage_handle_t * handle,

	       dbase_config_t * dconfig,

	       record_t *** records, unsigned int *count)

{

	if (enter_ro(handle, dconfig) < 0)

		return STATUS_ERR;

	if (dconfig->dtable->list(handle, dconfig->dbase, records, count) < 0) {

		exit_ro(handle);

		return STATUS_ERR;

	}

	return exit_ro(handle);

}