source4/lib/registry/reg_backend_ldb/reg_backend_ldb.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    Registry interface
   4    Copyright (C) Jelmer Vernooij  2004.
   5
   6    This program is free software; you can redistribute it and/or modify
   7    it under the terms of the GNU General Public License as published by
   8    the Free Software Foundation; either version 2 of the License, or
   9    (at your option) any later version.
  10
  11    This program is distributed in the hope that it will be useful,
  12    but WITHOUT ANY WARRANTY; without even the implied warranty of
  13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14    GNU General Public License for more details.
  15
  16    You should have received a copy of the GNU General Public License
  17    along with this program; if not, write to the Free Software
  18    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  19 */
  20
  21 #include "includes.h"
  22 #include "lib/registry/common/registry.h"
  23
  24 static char *reg_path_to_ldb(TALLOC_CTX *mem_ctx, const char *path, const char *add)
  25 {
  26         char *ret = talloc_strdup(mem_ctx, "");
  27         char *mypath = strdup(path);
  28         char *end = mypath, *begin;
  29
  30         if(add)
  31                 ret = talloc_asprintf_append(ret, "%s", add);
  32
  33         while(end) {
  34                 char *keyname;
  35                 begin = strrchr(end, '\\');
  36
  37                 if(begin) keyname = begin + 1;
  38                 else keyname = mypath;
  39
  40                 if(strlen(keyname))
  41                         ret = talloc_asprintf_append(ret, "key=%s,", keyname);
  42
  43                 if(begin) {
  44                         begin[0] = '\0';
  45                         end = begin-1;
  46                 } else {
  47                         end = NULL;
  48                 }
  49         }
  50
  51         SAFE_FREE(mypath);
  52
  53         ret[strlen(ret)-1] = '\0';
  54
  55         return ret;
  56 }
  57
  58 /*
  59  * Saves the dn as private_data for every key/val
  60  */
  61
  62 static WERROR ldb_open_registry(REG_HANDLE *handle, const char *location, const char *credentials)
  63 {
  64         struct ldb_context *c;
  65
  66         if (!location) return WERR_INVALID_PARAM;
  67         c = ldb_connect(location, 0, NULL);
  68
  69         ldb_set_debug_stderr(c);
  70
  71         if(!c) return WERR_FOOBAR;
  72
  73         handle->backend_data = c;
  74
  75         return WERR_OK;
  76 }
  77
  78 static WERROR ldb_close_registry(REG_HANDLE *h)
  79 {
  80         ldb_close((struct ldb_context *)h->backend_data);
  81         return WERR_OK;
  82 }
  83
  84 static WERROR ldb_add_key(REG_KEY *p, const char *name, uint32_t access_mask, SEC_DESC *sec, REG_KEY **new)
  85 {
  86         return WERR_NOT_SUPPORTED;
  87 }
  88
  89 static WERROR ldb_fetch_subkeys(REG_KEY *k, int *count, REG_KEY ***subkeys)
  90 {
  91         struct ldb_context *c = k->handle->backend_data;
  92         int ret, i, j;
  93         struct ldb_message **msg;
  94
  95         ret = ldb_search(c, (char *)k->backend_data, LDB_SCOPE_ONELEVEL, "(key=*)", NULL,&msg);
  96
  97         if(ret < 0) {
  98                 DEBUG(0, ("Error getting subkeys for '%s': %s\n", (char *)k->backend_data, ldb_errstring(c)));
  99                 return WERR_FOOBAR;
 100         }
 101
 102         *subkeys = talloc_array_p(k->mem_ctx, REG_KEY *, ret);
 103         j = 0;
 104         for(i = 0; i < ret; i++) {
 105                 struct ldb_message_element *el;
 106                 char *name;
 107                 el = ldb_msg_find_element(msg[i], "key");
 108
 109                 name = el->values[0].data;
 110
 111                 /* Dirty hack to circumvent ldb_tdb bug */
 112                 if(k->backend_data && !strcmp(msg[i]->dn, (char *)k->backend_data)) continue;
 113
 114                 (*subkeys)[j] = reg_key_new_rel(name, k, NULL);
 115                 (*subkeys)[j]->backend_data = talloc_strdup((*subkeys)[j]->mem_ctx, msg[i]->dn);
 116                 j++;
 117         }
 118         *count = j;
 119
 120         ldb_search_free(c, msg);
 121         return WERR_OK;
 122 }
 123
 124 static WERROR ldb_fetch_values(REG_KEY *k, int *count, REG_VAL ***values)
 125 {
 126         struct ldb_context *c = k->handle->backend_data;
 127         int ret, i, j;
 128         struct ldb_message **msg;
 129
 130         ret = ldb_search(c, (char *)k->backend_data, LDB_SCOPE_ONELEVEL, "(value=*)", NULL,&msg);
 131
 132         if(ret < 0) {
 133                 DEBUG(0, ("Error getting values for '%s': %s\n", (char *)k->backend_data, ldb_errstring(c)));
 134                 return WERR_FOOBAR;
 135         }
 136
 137         *values = talloc_array_p(k->mem_ctx, REG_VAL *, ret);
 138         j = 0;
 139         for(i = 0; i < ret; i++) {
 140                 struct ldb_message_element *el;
 141                 char *name;
 142                 el = ldb_msg_find_element(msg[i], "key");
 143
 144                 name = el->values[0].data;
 145
 146                 /* Dirty hack to circumvent ldb_tdb bug */
 147                 if(k->backend_data && !strcmp(msg[i]->dn, (char *)k->backend_data)) continue;
 148
 149                 (*values)[j] = reg_val_new(k, NULL);
 150                 (*values)[j]->backend_data = talloc_strdup((*values)[j]->mem_ctx, msg[i]->dn);
 151                 j++;
 152         }
 153         *count = j;
 154
 155         ldb_search_free(c, msg);
 156         return WERR_OK;
 157 }
 158
 159 static WERROR ldb_get_hive(REG_HANDLE *h, int num, REG_KEY **key)
 160 {
 161         if(num != 0) return WERR_NO_MORE_ITEMS;
 162         *key = reg_key_new_abs("", h, NULL);
 163         return WERR_OK;
 164 }
 165
 166 static WERROR ldb_open_key(REG_HANDLE *h, int num, const char *name, REG_KEY **key)
 167 {
 168         struct ldb_context *c = h->backend_data;
 169         struct ldb_message **msg;
 170         char *ldap_path;
 171         int ret;
 172         TALLOC_CTX *mem_ctx = talloc_init("ldb_path");
 173         if(num != 0) return WERR_NO_MORE_ITEMS;
 174         ldap_path = reg_path_to_ldb(mem_ctx, name, NULL);
 175
 176         ret = ldb_search(c, ldap_path, LDB_SCOPE_BASE, "*", NULL,&msg);
 177
 178         if(ret == 0) {
 179                 return WERR_NO_MORE_ITEMS;
 180         } else if(ret < 0) {
 181                 DEBUG(0, ("Error opening key '%s': %s\n", ldap_path, ldb_errstring(c)));
 182                 return WERR_FOOBAR;
 183         }
 184
 185         *key = reg_key_new_abs(name, h, ldap_path);
 186         talloc_steal((*key)->mem_ctx, ldap_path);
 187         printf("Got something!\n");
 188         /* FIXME */
 189
 190         ldb_search_free(c, msg);
 191         talloc_destroy(mem_ctx);
 192
 193         return WERR_OK;
 194 }
 195
 196 static struct registry_ops reg_backend_ldb = {
 197         .name = "ldb",
 198         .open_registry = ldb_open_registry,
 199         .get_hive = ldb_get_hive,
 200         .close_registry = ldb_close_registry,
 201         .open_key = ldb_open_key,
 202         .fetch_subkeys = ldb_fetch_subkeys,
 203         .fetch_values = ldb_fetch_values,
 204         .add_key = ldb_add_key,
 205 };
 206
 207 NTSTATUS registry_ldb_init(void)
 208 {
 209         return register_backend("registry", &reg_backend_ldb);
 210 }