r1118: Read-only enumeration of keys in the LDB backend works now :-)

(This used to be commit 456f97eed05047b20acceb50708b47209c37ca20)

1 files changed, 77 insertions, 29 deletions

diff --git a/source4/lib/registry/reg_backend_ldb/reg_backend_ldb.c b/source4/lib/registry/reg_backend_ldb/reg_backend_ldb.c
index 0633c94616..408ce52e44 100644
--- a/source4/lib/registry/reg_backend_ldb/reg_backend_ldb.c
+++ b/source4/lib/registry/reg_backend_ldb/reg_backend_ldb.c
@@ -21,24 +21,37 @@
 #include "includes.h"
 #include "lib/registry/common/registry.h"
 
-static char *reg_path_to_ldb(TALLOC_CTX *mem_ctx, const char *path)
+static char *reg_path_to_ldb(TALLOC_CTX *mem_ctx, const char *path, const char *add)
 {
-	char *ret = talloc_strdup(mem_ctx, "(dn=");
-	char *begin = (char *)path;
-	char *end = NULL;
-
-	while(begin) {
-		end = strchr(begin, '\\');
-		if(end)end = '\0';
-		if(end - begin != 0) ret = talloc_asprintf_append(mem_ctx, ret, "key=%s,", begin);
+	char *ret = talloc_strdup(mem_ctx, "");
+	char *mypath = strdup(path);
+	char *end = mypath, *begin;
+
+	if(add) 
+		ret = talloc_asprintf_append(mem_ctx, 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(mem_ctx, ret, "key=%s,", keyname);
 			
-		if(end) {
-			*end = '\\';
-			begin = end+1;
-		} else begin = NULL;
+		if(begin) {
+			begin[0] = '\0';
+			end = begin-1;
+		} else {
+			end = NULL;
+		}
 	}
 
-	ret[strlen(ret)-1] = ')';
+	SAFE_FREE(mypath);
+
+	ret[strlen(ret)-1] = '\0';
+	
 	printf("Doing search for : %s\n", ret);
 	return ret;
 }
@@ -50,8 +63,12 @@ static char *reg_path_to_ldb(TALLOC_CTX *mem_ctx, const char *path)
 static WERROR ldb_open_registry(REG_HANDLE *handle, const char *location, const char *credentials)
 {
 	struct ldb_context *c;
+
+	if (!location) return WERR_INVALID_PARAM;
 	c = ldb_connect(location, 0, NULL);
 
+	ldb_set_debug_stderr(c);
+
 	if(!c) return WERR_FOOBAR;
 
 	handle->backend_data = c;
@@ -69,18 +86,36 @@ static WERROR ldb_fetch_subkeys(REG_KEY *k, int *count, REG_KEY ***subkeys)
 {
 	struct ldb_context *c = k->handle->backend_data;
 	char *path;
+	int ret, i, j;
 	struct ldb_message **msg;
-	char *ldap_path;
 	TALLOC_CTX *mem_ctx = talloc_init("ldb_path");
 	REG_KEY *key = NULL;
-	ldap_path = reg_path_to_ldb(mem_ctx, reg_key_get_path(k));
-	
-	if(ldb_search(c, NULL, LDB_SCOPE_ONELEVEL, ldap_path, NULL,&msg) > 0) {
-		key = reg_key_new_abs(reg_key_get_path(k), k->handle, ldap_path);
-		talloc_steal(mem_ctx, key->mem_ctx, ldap_path);
-		/* FIXME */
+
+	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", k->backend_data, ldb_errstring(c)));
+		return WERR_FOOBAR;
 	}
 
+	*subkeys = talloc_array_p(k->mem_ctx, REG_KEY *, ret);
+	j = 0;
+	for(i = 0; i < ret; i++) {
+		struct ldb_message_element *el;
+		char *name;
+		el = ldb_msg_find_element(msg[i], "key");
+
+		name = el->values[0].data;
+
+		/* Dirty hack to circumvent ldb_tdb bug */
+		if(k->backend_data && !strcmp(msg[i]->dn, (char *)k->backend_data)) continue;
+			
+		(*subkeys)[j] = reg_key_new_rel(name, k, NULL);
+		(*subkeys)[j]->backend_data = talloc_strdup((*subkeys)[j]->mem_ctx, msg[i]->dn);
+		j++;
+	}
+	*count = j;
+
 	ldb_search_free(c, msg);
 	talloc_destroy(mem_ctx);
 	return WERR_OK;
@@ -88,8 +123,10 @@ static WERROR ldb_fetch_subkeys(REG_KEY *k, int *count, REG_KEY ***subkeys)
 
 static WERROR ldb_get_hive(REG_HANDLE *h, int num, REG_KEY **key)
 {
-	/* FIXME */
-	return WERR_FOOBAR;	
+	if(num != 0) return WERR_NO_MORE_ITEMS;
+	*key = reg_key_new_abs("", h, NULL);
+//	(*key)->backend_data = talloc_strdup((*key)->mem_ctx, "");
+	return WERR_OK;
 }
 
 static WERROR ldb_open_key(REG_HANDLE *h, int num, const char *name, REG_KEY **key)
@@ -97,16 +134,26 @@ static WERROR ldb_open_key(REG_HANDLE *h, int num, const char *name, REG_KEY **k
 	struct ldb_context *c = h->backend_data;
 	char *path;
 	struct ldb_message **msg;
-	char *ldap_path;
+	char *ldap_path, *new_ldap_path;
+	int ret;
 	TALLOC_CTX *mem_ctx = talloc_init("ldb_path");
-	ldap_path = reg_path_to_ldb(mem_ctx, name);
+	if(num != 0) return WERR_NO_MORE_ITEMS;
+	ldap_path = reg_path_to_ldb(mem_ctx, name, NULL);
 	
-	if(ldb_search(c, NULL, LDB_SCOPE_BASE, ldap_path, NULL,&msg) > 0) {
-		*key = reg_key_new_abs(name, h, ldap_path);
-		talloc_steal(mem_ctx, (*key)->mem_ctx, ldap_path);
-		/* FIXME */
+	ret = ldb_search(c, ldap_path, LDB_SCOPE_BASE, "*", 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 = reg_key_new_abs(name, h, ldap_path);
+	talloc_steal(mem_ctx, (*key)->mem_ctx, ldap_path);
+	printf("Got something!\n");
+	/* FIXME */
+
 	ldb_search_free(c, msg);
 	talloc_destroy(mem_ctx);
 
@@ -116,6 +163,7 @@ static WERROR ldb_open_key(REG_HANDLE *h, int num, const char *name, REG_KEY **k
 static struct registry_ops reg_backend_ldb = {
 	.name = "ldb",
 	.open_registry = ldb_open_registry,
+	.get_hive = ldb_get_hive,
 	.close_registry = ldb_close_registry,
 	.open_key = ldb_open_key,
 	.fetch_subkeys = ldb_fetch_subkeys,