source4/lib/ldb/common/ldb_modules.c

   1
   2 /*
   3    ldb database library
   4
   5    Copyright (C) Simo Sorce  2004
   6
   7      ** NOTE! The following LGPL license applies to the ldb
   8      ** library. This does NOT imply that all of Samba is released
   9      ** under the LGPL
  10
  11    This library is free software; you can redistribute it and/or
  12    modify it under the terms of the GNU Lesser General Public
  13    License as published by the Free Software Foundation; either
  14    version 2 of the License, or (at your option) any later version.
  15
  16    This library is distributed in the hope that it will be useful,
  17    but WITHOUT ANY WARRANTY; without even the implied warranty of
  18    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  19    Lesser General Public License for more details.
  20
  21    You should have received a copy of the GNU Lesser General Public
  22    License along with this library; if not, write to the Free Software
  23    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  24 */
  25
  26 /*
  27  *  Name: ldb
  28  *
  29  *  Component: ldb modules core
  30  *
  31  *  Description: core modules routines
  32  *
  33  *  Author: Simo Sorce
  34  */
  35
  36 #include "includes.h"
  37 #include "ldb/include/ldb.h"
  38 #include "ldb/include/ldb_errors.h"
  39 #include "ldb/include/ldb_private.h"
  40 #include "dlinklist.h"
  41 #include <sys/types.h>
  42 #include <sys/stat.h>
  43 #include <unistd.h>
  44
  45 #ifdef HAVE_DLOPEN_DISABLED
  46 #include <dlfcn.h>
  47 #endif
  48
  49 #define LDB_MODULE_PREFIX       "modules:"
  50 #define LDB_MODULE_PREFIX_LEN   8
  51
  52 static char *talloc_strdup_no_spaces(struct ldb_context *ldb, const char *string)
  53 {
  54         int i, len;
  55         char *trimmed;
  56
  57         trimmed = talloc_strdup(ldb, string);
  58         if (!trimmed) {
  59                 ldb_debug(ldb, LDB_DEBUG_FATAL, "Out of Memory in talloc_strdup_trim_spaces()\n");
  60                 return NULL;
  61         }
  62
  63         len = strlen(trimmed);
  64         for (i = 0; trimmed[i] != '\0'; i++) {
  65                 switch (trimmed[i]) {
  66                 case ' ':
  67                 case '\t':
  68                 case '\n':
  69                         memmove(&trimmed[i], &trimmed[i + 1], len -i -1);
  70                         break;
  71                 }
  72         }
  73
  74         return trimmed;
  75 }
  76
  77
  78 /* modules are called in inverse order on the stack.
  79    Lets place them as an admin would think the right order is.
  80    Modules order is imprtant */
  81 static char **ldb_modules_list_from_string(struct ldb_context *ldb, const char *string)
  82 {
  83         char **modules = NULL;
  84         char *modstr, *p;
  85         int i;
  86
  87         /* spaces not admitted */
  88         modstr = talloc_strdup_no_spaces(ldb, string);
  89         if ( ! modstr) {
  90                 return NULL;
  91         }
  92
  93         modules = talloc_realloc(ldb, modules, char *, 2);
  94         if ( ! modules ) {
  95                 ldb_debug(ldb, LDB_DEBUG_FATAL, "Out of Memory in ldb_modules_list_from_string()\n");
  96                 talloc_free(modstr);
  97                 return NULL;
  98         }
  99         talloc_steal(modules, modstr);
 100
 101         i = 0;
 102         while ((p = strrchr(modstr, ',')) != NULL) {
 103                 *p = '\0';
 104                 p++;
 105                 modules[i] = p;
 106
 107                 i++;
 108                 modules = talloc_realloc(ldb, modules, char *, i + 2);
 109                 if ( ! modules ) {
 110                         ldb_debug(ldb, LDB_DEBUG_FATAL, "Out of Memory in ldb_modules_list_from_string()\n");
 111                         return NULL;
 112                 }
 113
 114         }
 115         modules[i] = modstr;
 116
 117         modules[i + 1] = NULL;
 118
 119         return modules;
 120 }
 121
 122 int ldb_load_modules(struct ldb_context *ldb, const char *options[])
 123 {
 124         char **modules = NULL;
 125         int i;
 126         struct {
 127                 const char *name;
 128                 ldb_module_init_t init;
 129         } well_known_modules[] = {
 130                 { "schema", schema_module_init },
 131                 { "operational", operational_module_init },
 132                 { "rdn_name", rdn_name_module_init },
 133 #ifdef _SAMBA_BUILD_
 134                 { "objectguid", objectguid_module_init },
 135                 { "samldb", samldb_module_init },
 136                 { "samba3sam", ldb_samba3sam_module_init },
 137                 { "proxy", proxy_module_init },
 138                 { "rootdse", rootdse_module_init },
 139 #endif
 140                 { NULL, NULL }
 141         };
 142
 143         /* find out which modules we are requested to activate */
 144
 145         /* check if we have a custom module list passd as ldb option */
 146         if (options) {
 147                 for (i = 0; options[i] != NULL; i++) {
 148                         if (strncmp(options[i], LDB_MODULE_PREFIX, LDB_MODULE_PREFIX_LEN) == 0) {
 149                                 modules = ldb_modules_list_from_string(ldb, &options[i][LDB_MODULE_PREFIX_LEN]);
 150                         }
 151                 }
 152         }
 153
 154         /* if not overloaded by options and the backend is not ldap try to load the modules list form ldb */
 155         if ((modules == NULL) && (strcmp("ldap", ldb->modules->ops->name) != 0)) {
 156                 int ret;
 157                 const char * const attrs[] = { "@LIST" , NULL};
 158                 struct ldb_result *res = NULL;
 159                 struct ldb_dn *mods;
 160
 161                 mods = ldb_dn_explode(ldb, "@MODULES");
 162                 if (mods == NULL) {
 163                         return -1;
 164                 }
 165
 166                 ret = ldb_search(ldb, mods, LDB_SCOPE_BASE, "", attrs, &res);
 167                 talloc_free(mods);
 168                 if (ret == LDB_SUCCESS && (res->count == 0 || res->msgs[0]->num_elements == 0)) {
 169                         ldb_debug(ldb, LDB_DEBUG_TRACE, "no modules required by the db\n");
 170                 } else {
 171                         if (ret != LDB_SUCCESS) {
 172                                 ldb_debug(ldb, LDB_DEBUG_FATAL, "ldb error (%s) occurred searching for modules, bailing out\n", ldb_errstring(ldb));
 173                                 return -1;
 174                         }
 175                         if (res->count > 1) {
 176                                 ldb_debug(ldb, LDB_DEBUG_FATAL, "Too many records found (%d), bailing out\n", res->count);
 177                                 talloc_free(res);
 178                                 return -1;
 179                         }
 180
 181                         modules = ldb_modules_list_from_string(ldb,
 182                                                                (const char *)res->msgs[0]->elements[0].values[0].data);
 183
 184                 }
 185
 186                 talloc_free(res);
 187         }
 188
 189         if (modules == NULL) {
 190                 ldb_debug(ldb, LDB_DEBUG_TRACE, "No modules specified for this database\n");
 191                 return 0;
 192         }
 193
 194         for (i = 0; modules[i] != NULL; i++) {
 195                 struct ldb_module *current;
 196                 int m;
 197                 for (m=0;well_known_modules[m].name;m++) {
 198                         if (strcmp(modules[i], well_known_modules[m].name) == 0) {
 199                                 current = well_known_modules[m].init(ldb, options);
 200                                 if (current == NULL) {
 201                                         ldb_debug(ldb, LDB_DEBUG_FATAL, "function 'init_module' in %s fails\n", modules[i]);
 202                                         return -1;
 203                                 }
 204                                 DLIST_ADD(ldb->modules, current);
 205                                 break;
 206                         }
 207                 }
 208                 if (well_known_modules[m].name == NULL) {
 209                         ldb_debug(ldb, LDB_DEBUG_WARNING, "WARNING: Module [%s] not found\n",
 210                                   modules[i]);
 211                 }
 212         }
 213
 214         talloc_free(modules);
 215         return 0;
 216 }
 217
 218 /*
 219   by using this we allow ldb modules to only implement the functions they care about,
 220   which makes writing a module simpler, and makes it more likely to keep working
 221   when ldb is extended
 222 */
 223 #define FIND_OP(module, op) do { \
 224         module = module->next; \
 225         while (module && module->ops->op == NULL) module = module->next; \
 226         if (module == NULL) return -1; \
 227 } while (0)
 228
 229
 230 /*
 231    helper functions to call the next module in chain
 232 */
 233 int ldb_next_request(struct ldb_module *module, struct ldb_request *request)
 234 {
 235         FIND_OP(module, request);
 236         return module->ops->request(module, request);
 237 }
 238
 239 int ldb_next_start_trans(struct ldb_module *module)
 240 {
 241         FIND_OP(module, start_transaction);
 242         return module->ops->start_transaction(module);
 243 }
 244
 245 int ldb_next_end_trans(struct ldb_module *module)
 246 {
 247         FIND_OP(module, end_transaction);
 248         return module->ops->end_transaction(module);
 249 }
 250
 251 int ldb_next_del_trans(struct ldb_module *module)
 252 {
 253         FIND_OP(module, del_transaction);
 254         return module->ops->del_transaction(module);
 255 }
 256
 257 void ldb_set_errstring(struct ldb_module *module, char *err_string)
 258 {
 259         if (module->ldb->err_string) {
 260                 talloc_free(module->ldb->err_string);
 261         }
 262
 263         module->ldb->err_string = err_string;
 264 }
 265