/*
Unix SMB/CIFS implementation.
Registry interface
- Copyright (C) Jelmer Vernooij 2004.
+ Copyright (C) Jelmer Vernooij 2004.
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
*/
#include "includes.h"
-#include "lib/registry/common/registry.h"
+#include "registry.h"
+#include "lib/ldb/include/ldb.h"
-/*
- * Saves the dn as private_data for every key/val
- */
-
-static BOOL ldb_open_registry(REG_HANDLE *handle, const char *location, BOOL try_full_load)
+static char *reg_path_to_ldb(TALLOC_CTX *mem_ctx, const char *path, const char *add)
{
- struct ldb_context *c;
- c = ldb_connect(location, 0, NULL);
+ char *ret = talloc_strdup(mem_ctx, "");
+ char *mypath = strdup(path);
+ char *end = mypath, *begin;
+
+ if(add)
+ ret = talloc_asprintf_append(ret, "%s", add);
+
+ while(end) {
+ char *keyname;
+ begin = strrchr(end, '\\');
+
+ if(begin) keyname = begin + 1;
+ else keyname = mypath;
+
+ if(strlen(keyname))
+ ret = talloc_asprintf_append(ret, "key=%s,", keyname);
+
+ if(begin) {
+ *begin = '\0';
+ end = begin-1;
+ } else {
+ end = NULL;
+ }
+ }
- if(!c) return False;
+ SAFE_FREE(mypath);
- handle->backend_data = c;
+ ret[strlen(ret)-1] = '\0';
+
+ if(strlen(ret) == 0) return NULL;
- return True;
+ return ret;
}
-static BOOL ldb_close_registry(REG_HANDLE *h)
+
+static WERROR ldb_get_subkey_by_id(TALLOC_CTX *mem_ctx, struct registry_key *k, int idx, struct registry_key **subkey)
{
- ldb_close(h);
- return True;
+ struct ldb_context *c = k->hive->backend_data;
+ int ret;
+ struct ldb_message **msg;
+ struct ldb_message_element *el;
+
+ ret = ldb_search(c, (char *)k->backend_data, LDB_SCOPE_ONELEVEL, "(key=*)", NULL,&msg);
+
+ if(ret < 0) {
+ DEBUG(0, ("Error getting subkeys for '%s': %s\n", (char *)k->backend_data, ldb_errstring(c)));
+ return WERR_FOOBAR;
+ }
+
+ if(idx >= ret) return WERR_NO_MORE_ITEMS;
+
+ el = ldb_msg_find_element(msg[idx], "key");
+
+ *subkey = talloc_p(mem_ctx, struct registry_key);
+ (*subkey)->name = talloc_strdup(mem_ctx, el->values[0].data);
+ (*subkey)->backend_data = talloc_strdup(mem_ctx, msg[idx]->dn);
+
+ ldb_search_free(c, msg);
+ return WERR_OK;
}
-static BOOL ldb_fetch_subkeys(REG_KEY *k, int *count, REG_KEY ***subkeys)
+static WERROR ldb_get_value_by_id(TALLOC_CTX *mem_ctx, struct registry_key *k, int idx, struct registry_value **value)
{
- ldb_search();
+ struct ldb_context *c = k->hive->backend_data;
+ int ret;
+ struct ldb_message **msg;
+ struct ldb_message_element *el;
+
+ ret = ldb_search(c, (char *)k->backend_data, LDB_SCOPE_ONELEVEL, "(value=*)", NULL,&msg);
+
+ if(ret < 0) {
+ DEBUG(0, ("Error getting values for '%s': %s\n", (char *)k->backend_data, ldb_errstring(c)));
+ return WERR_FOOBAR;
+ }
+
+ if(idx >= ret) return WERR_NO_MORE_ITEMS;
+
+ el = ldb_msg_find_element(msg[idx], "value");
+
+ *value = talloc_p(mem_ctx, struct registry_value);
+ (*value)->name = talloc_strdup(mem_ctx, el->values[0].data);
+ (*value)->backend_data = talloc_strdup(mem_ctx, msg[idx]->dn);
+
+ ldb_search_free(c, msg);
+ return WERR_OK;
}
-static REG_KEY *ldb_open_key(REG_HANDLE *h, const char *name)
+static WERROR ldb_open_key(TALLOC_CTX *mem_ctx, struct registry_hive *h, const char *name, struct registry_key **key)
{
struct ldb_context *c = h->backend_data;
- char *path;
struct ldb_message **msg;
- REG_KEY *key = NULL;
- (dn=key=Systems,
- if(ldb_search(c, NULL, LDP_SCOPE_BASE, "", NULL,&msg) > 0) {
- key = reg_key_new_abs(name, h, base);
+ char *ldap_path;
+ int ret;
+ ldap_path = reg_path_to_ldb(mem_ctx, name, NULL);
+
+ ret = ldb_search(c, ldap_path, LDB_SCOPE_BASE, "(key=*)", NULL,&msg);
+
+ if(ret == 0) {
+ return WERR_NO_MORE_ITEMS;
+ } else if(ret < 0) {
+ DEBUG(0, ("Error opening key '%s': %s\n", ldap_path, ldb_errstring(c)));
+ return WERR_FOOBAR;
+ }
+
+ *key = talloc_p(mem_ctx, struct registry_key);
+ (*key)->name = talloc_strdup(mem_ctx, strrchr(name, '\\'));
+ (*key)->backend_data = talloc_strdup(mem_ctx, msg[0]->dn);
+
+ ldb_search_free(c, msg);
+
+ return WERR_OK;
+}
+
+static WERROR ldb_open_hive(TALLOC_CTX *mem_ctx, struct registry_hive *hive, struct registry_key **k)
+{
+ struct ldb_context *c;
+
+ if (!hive->location) return WERR_INVALID_PARAM;
+ c = ldb_connect(hive->location, 0, NULL);
+
+ if(!c) {
+ DEBUG(1, ("ldb_open_hive: %s\n", ldb_errstring(hive->backend_data)));
+ return WERR_FOOBAR;
}
+ ldb_set_debug_stderr(c);
+ hive->backend_data = c;
+
+ hive->root = talloc_zero_p(mem_ctx, struct registry_key);
+ hive->root->name = talloc_strdup(mem_ctx, "");
+
+ return WERR_OK;
+}
+
+static WERROR ldb_add_key (TALLOC_CTX *mem_ctx, struct registry_key *parent, const char *name, uint32_t access_mask, SEC_DESC *sd, struct registry_key **newkey)
+{
+ struct ldb_context *ctx = parent->hive->backend_data;
+ struct ldb_message msg;
+ int ret;
+
+ ZERO_STRUCT(msg);
- ldap_search_free(c, msg);
+ msg.dn = reg_path_to_ldb(mem_ctx, parent->path, talloc_asprintf(mem_ctx, "key=%s,", name));
- return key;
+ ldb_msg_add_string(ctx, &msg, "key", talloc_strdup(mem_ctx, name));
+
+ ret = ldb_add(ctx, &msg);
+ if (ret < 0) {
+ DEBUG(1, ("ldb_msg_add: %s\n", ldb_errstring(parent->hive->backend_data)));
+ return WERR_FOOBAR;
+ }
+
+ *newkey = talloc_zero_p(mem_ctx, struct registry_key);
+ (*newkey)->backend_data = msg.dn;
+ (*newkey)->name = talloc_strdup(mem_ctx, name);
+
+ return WERR_OK;
+}
+
+static WERROR ldb_del_key (struct registry_key *key)
+{
+ int ret;
+
+ ret = ldb_delete(key->hive->backend_data, key->backend_data);
+
+ if (ret < 0) {
+ DEBUG(1, ("ldb_del_key: %s\n", ldb_errstring(key->hive->backend_data)));
+ return WERR_FOOBAR;
+ }
+
+ return WERR_OK;
+}
+
+static WERROR ldb_close_hive (struct registry_hive *hive)
+{
+ ldb_close (hive->backend_data);
+ return WERR_OK;
}
-static REG_OPS reg_backend_ldb = {
+static struct registry_operations reg_backend_ldb = {
.name = "ldb",
- .open_registry = ldb_open_registry,
- .close_registry = ldb_close_registry,
+ .add_key = ldb_add_key,
+ .del_key = ldb_del_key,
+ .open_hive = ldb_open_hive,
+ .close_hive = ldb_close_hive,
.open_key = ldb_open_key,
- .fetch_subkeys = ldb_fetch_subkeys,
+ .get_value_by_index = ldb_get_value_by_id,
+ .get_subkey_by_index = ldb_get_subkey_by_id,
};
-NTSTATUS reg_ldb_init(void)
+NTSTATUS registry_ldb_init(void)
{
- return register_backend("registry", ®_backend_ldb);
+ return registry_register(®_backend_ldb);
}