r3783: - don't use make proto for ldb anymore
[kamenim/samba.git] / source4 / lib / registry / reg_backend_ldb / reg_backend_ldb.c
index afa60c5c4a1769234b223d151608c0d288fa2c06..380a8010f22b19d5ddba20a35dc0538f4d435f71 100644 (file)
@@ -1,7 +1,7 @@
 /* 
    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", &reg_backend_ldb);
+       return registry_register(&reg_backend_ldb);
 }