samba-mirror/source4/lib/registry/local.c

/* 
   Unix SMB/CIFS implementation.
   Transparent registry backend handling
   Copyright (C) Jelmer Vernooij			2003-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 2 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, write to the Free Software
   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
*/

#include "includes.h"
#include "lib/util/dlinklist.h"
#include "lib/registry/registry.h"
#include "system/filesys.h"
#include "build.h"

struct reg_key_path {
	uint32_t predefined_key;
	const char **elements;
};

struct registry_local {
	struct registry_context registry;

	struct mountpoint {
		struct reg_key_path path;
		struct hive_key *key;
		struct mountpoint *prev, *next;
	} *mountpoints;

	struct auth_session_info *session_info; 
	struct cli_credentials *credentials;
};

struct local_key {
	struct registry_key global;
	struct reg_key_path path;
	struct hive_key *hive_key;
};


struct registry_key *reg_import_hive_key(struct registry_context *ctx,
									     struct hive_key *hive, 
										 uint32_t predefined_key,
										 const char **elements)
{
	struct local_key *local_key;
	struct reg_key_path parent_path;

	parent_path.predefined_key = predefined_key;
	parent_path.elements = elements;

	local_key = talloc(ctx, struct local_key);
	local_key->hive_key = talloc_steal(local_key, hive);
	local_key->global.context = talloc_reference(local_key, ctx);
	local_key->path = parent_path;

	return (struct registry_key *)local_key;
}


static WERROR local_open_key(TALLOC_CTX *mem_ctx,
						   struct registry_key *parent, 
						   const char *path,
						   struct registry_key **result)
{
	char *orig = talloc_strdup(mem_ctx, path),
		 *curbegin = orig, 
		 *curend = strchr(orig, '\\');
	struct local_key *local_parent = talloc_get_type(parent, struct local_key);
	struct hive_key *curkey = local_parent->hive_key;
	WERROR error;
	const char **elements = NULL;
	int el;

	if (local_parent->path.elements != NULL) {
		elements = talloc_array(mem_ctx, const char *, 
							str_list_length(local_parent->path.elements) + 1);
		for (el = 0; local_parent->path.elements[el] != NULL; el++) {
			elements[el] = talloc_reference(elements, 
											local_parent->path.elements[el]);
		}
		elements[el] = NULL;
	} else {
		elements = NULL;
		el = 0;
	}

	while (curbegin != NULL && *curbegin) {
		if (curend != NULL)
			*curend = '\0';
		elements = talloc_realloc(mem_ctx, elements, const char *, el+2);
		elements[el] = talloc_strdup(elements, curbegin);
		el++;
		elements[el] = NULL;
		error = hive_get_key_by_name(mem_ctx, curkey, curbegin, &curkey);
		if (!W_ERROR_IS_OK(error)) {
			DEBUG(2, ("Opening key %s failed: %s\n", curbegin, win_errstr(error)));
			talloc_free(orig);
			return error;
		}
		if (curend == NULL) 
			break;
		curbegin = curend + 1;
		curend = strchr(curbegin, '\\');
	}
	talloc_free(orig);

	*result = reg_import_hive_key(local_parent->global.context, curkey, 
								  local_parent->path.predefined_key,
								  talloc_steal(curkey, elements));
	
	return WERR_OK;
}

WERROR local_get_predefined_key (const struct registry_context *ctx, 
	  uint32_t key_id, struct registry_key **key) 
{	
	struct registry_local *rctx = talloc_get_type(ctx, struct registry_local);
	struct mountpoint *mp;

	for (mp = rctx->mountpoints; mp != NULL; mp = mp->next) {
		if (mp->path.predefined_key == key_id && 
			mp->path.elements == NULL)
			break;
	}

	if (mp == NULL)
		return WERR_NOT_FOUND;
	
	*key = reg_import_hive_key(ctx, mp->key, 
							   mp->path.predefined_key, 
							   mp->path.elements
							   );

	return WERR_OK;
}

WERROR local_enum_key(TALLOC_CTX *mem_ctx,
					  const struct registry_key *key, uint32_t idx,
					  const char **name,
					  const char **keyclass,
					  NTTIME *last_changed_time)
{
	const struct local_key *local = (const struct local_key *)key;

	return hive_enum_key(mem_ctx, local->hive_key, idx, name, keyclass, 
						  last_changed_time);
}

static WERROR local_create_key (TALLOC_CTX *mem_ctx, 
								struct registry_key *parent_key, 
								const char *name,
								const char *key_class,
								struct security_descriptor *security,
								struct registry_key **key)
{
	const struct local_key *local_parent;
	struct hive_key *hivekey;
	const char **elements;
	int i;
	char *last_part;

	last_part = strrchr(name, '\\');
	if (last_part == NULL) {
		last_part = name;
		local_parent = (const struct local_key *)parent_key;
	} else {
		W_ERROR_NOT_OK_RETURN(reg_open_key(mem_ctx, parent_key, 
							  talloc_strndup(mem_ctx, name, last_part-name),
							  &local_parent));
		last_part++;
	}

	W_ERROR_NOT_OK_RETURN(hive_key_add_name(mem_ctx, local_parent->hive_key, 
								last_part, key_class, security, &hivekey));

	if (local_parent->path.elements != NULL) {
		elements = talloc_array(hivekey, const char *, 
								str_list_length(local_parent->path.elements)+2);
		for (i = 0; local_parent->path.elements[i] != NULL; i++) {
			elements[i] = talloc_reference(elements, 
										   local_parent->path.elements[i]);
		}
	} else {
		elements = talloc_array(hivekey, const char *, 2);
		i = 0;
	}

	elements[i] = talloc_strdup(elements, name);
	elements[i+1] = NULL;

	*key = reg_import_hive_key(local_parent->global.context, hivekey, 
							   local_parent->path.predefined_key,
							   elements);

	return WERR_OK;
}

static WERROR local_set_value (struct registry_key *key, const char *name,
						uint32_t type, const DATA_BLOB data)
{
	struct local_key *local = (struct local_key *)key;

	return hive_set_value(local->hive_key, name, type, data);
}

static WERROR local_get_value (TALLOC_CTX *mem_ctx,
						 const struct registry_key *key,
						 const char *name, uint32_t *type, DATA_BLOB *data)
{
	const struct local_key *local = (const struct local_key *)key;

	return hive_get_value(mem_ctx, local->hive_key, name, type, data);
}

static WERROR local_enum_value (TALLOC_CTX *mem_ctx,
						  const struct registry_key *key, uint32_t idx,
						  const char **name,
						  uint32_t *type,
						  DATA_BLOB *data)
{
	const struct local_key *local = (const struct local_key *)key;

	return hive_get_value_by_index(mem_ctx, local->hive_key, idx, 
								   name, type, data);
}

static WERROR local_delete_key (struct registry_key *key, const char *name)
{
	const struct local_key *local = (const struct local_key *)key;

	return hive_key_del(local->hive_key, name);
}

static WERROR local_delete_value (struct registry_key *key, const char *name)
{
	const struct local_key *local = (const struct local_key *)key;

	return hive_del_value(local->hive_key, name);
}

static WERROR local_flush_key (struct registry_key *key)
{
	const struct local_key *local = (const struct local_key *)key;

	return hive_key_flush(local->hive_key);
}

static WERROR local_get_key_info (TALLOC_CTX *mem_ctx,
						const struct registry_key *key,
						const char **classname,
						uint32_t *num_subkeys,
						uint32_t *num_values,
						NTTIME *last_change_time)
{
	const struct local_key *local = (const struct local_key *)key;

	return hive_key_get_info(mem_ctx, local->hive_key, 
							 classname, num_subkeys, num_values, 
							 last_change_time);
}

const static struct registry_operations local_ops = {
	.name = "local",
	.open_key = local_open_key,
	.get_predefined_key = local_get_predefined_key,
	.enum_key = local_enum_key,
	.create_key = local_create_key,
	.set_value = local_set_value,
	.get_value = local_get_value,
	.enum_value = local_enum_value,
	.delete_key = local_delete_key,
	.delete_value = local_delete_value,
	.flush_key = local_flush_key,
	.get_key_info = local_get_key_info,
};

WERROR reg_open_local(TALLOC_CTX *mem_ctx, struct registry_context **ctx, 
				struct auth_session_info *session_info, 
				struct cli_credentials *credentials)
{
	struct registry_local *ret = talloc_zero(mem_ctx, struct registry_local);

	W_ERROR_HAVE_NO_MEMORY(ret);

	ret->registry.ops = &local_ops;
	ret->session_info = session_info;
	ret->credentials = credentials;

	*ctx = (struct registry_context *)ret;
	
	return WERR_OK;
}

WERROR reg_mount_hive(struct registry_context *rctx, 
					  struct hive_key *hive_key,
					  uint32_t key_id,
					  const char **elements) 
{
	struct registry_local *reg_local = talloc_get_type(rctx, struct registry_local);
	struct mountpoint *mp = talloc(rctx, struct mountpoint);
	int i = 0;

	mp->path.predefined_key = key_id;
	mp->prev = mp->next = NULL;
	mp->key = hive_key;
	if (elements != NULL) {
		mp->path.elements = talloc_array(mp, const char *, 
										 str_list_length(elements));
		for (i = 0; elements[i] != NULL; i++) {
			mp->path.elements[i] = talloc_reference(mp->path.elements, 
													elements[i]);
		}
		mp->path.elements[i] = NULL;
	} else {
		mp->path.elements = NULL;
	}

	DLIST_ADD(reg_local->mountpoints, mp);

	return WERR_OK;
}