Return per-entry controls in ldb_module_send_entry()
[abartlet/samba.git/.git] / source4 / lib / ldb / common / ldb_modules.c
index ddbe0f23a60cee35c904c92f20eff172b728836b..4ac404b9d6e341ff30e7b3edeb2263257591afd8 100644 (file)
@@ -1,7 +1,7 @@
 /* 
    ldb database library
 
-   Copyright (C) Simo Sorce  2004
+   Copyright (C) Simo Sorce  2004-2008
 
      ** NOTE! The following LGPL license applies to the ldb
      ** library. This does NOT imply that all of Samba is released
@@ -120,36 +120,149 @@ const char **ldb_modules_list_from_string(struct ldb_context *ldb, TALLOC_CTX *m
        return m;
 }
 
+static struct backends_list_entry {
+       struct ldb_backend_ops *ops;
+       struct backends_list_entry *prev, *next;
+} *ldb_backends = NULL;
+
 static struct ops_list_entry {
        const struct ldb_module_ops *ops;
        struct ops_list_entry *next;    
 } *registered_modules = NULL;
 
-#define LDB_MODULE(name) (&ldb_ ## name ## _module_ops)
+static const struct ldb_builtins {
+       const struct ldb_backend_ops *backend_ops;
+       const struct ldb_module_ops *module_ops;
+} builtins[];
 
-#ifndef STATIC_LIBLDB_MODULES
+static ldb_connect_fn ldb_find_backend(const char *url)
+{
+       struct backends_list_entry *backend;
+       int i;
 
-#define STATIC_LIBLDB_MODULES \
-       LDB_MODULE(operational),        \
-       LDB_MODULE(rdn_name),   \
-       LDB_MODULE(paged_results),      \
-       LDB_MODULE(server_sort),                \
-       LDB_MODULE(asq), \
-       NULL
-#endif
+       for (i = 0; builtins[i].backend_ops || builtins[i].module_ops; i++) {
+               if (builtins[i].backend_ops == NULL) continue;
 
-const static struct ldb_module_ops *builtin_modules[] = {
-       STATIC_LIBLDB_MODULES
-};
+               if (strncmp(builtins[i].backend_ops->name, url,
+                           strlen(builtins[i].backend_ops->name)) == 0) {
+                       return builtins[i].backend_ops->connect_fn;
+               }
+       }
+
+       for (backend = ldb_backends; backend; backend = backend->next) {
+               if (strncmp(backend->ops->name, url,
+                           strlen(backend->ops->name)) == 0) {
+                       return backend->ops->connect_fn;
+               }
+       }
+
+       return NULL;
+}
+
+/*
+ register a new ldb backend
+*/
+int ldb_register_backend(const char *url_prefix, ldb_connect_fn connectfn)
+{
+       struct ldb_backend_ops *backend;
+       struct backends_list_entry *entry;
+
+       backend = talloc(talloc_autofree_context(), struct ldb_backend_ops);
+       if (!backend) return LDB_ERR_OPERATIONS_ERROR;
+
+       entry = talloc(talloc_autofree_context(), struct backends_list_entry);
+       if (!entry) {
+               talloc_free(backend);
+               return LDB_ERR_OPERATIONS_ERROR;
+       }
+
+       if (ldb_find_backend(url_prefix)) {
+               return LDB_SUCCESS;
+       }
+
+       /* Maybe check for duplicity here later on? */
+
+       backend->name = talloc_strdup(backend, url_prefix);
+       backend->connect_fn = connectfn;
+       entry->ops = backend;
+       DLIST_ADD(ldb_backends, entry);
+
+       return LDB_SUCCESS;
+}
+
+/*
+   Return the ldb module form of a database.
+   The URL can either be one of the following forms
+   ldb://path
+   ldapi://path
+
+   flags is made up of LDB_FLG_*
+
+   the options are passed uninterpreted to the backend, and are
+   backend specific.
+
+   This allows modules to get at only the backend module, for example where a
+   module may wish to direct certain requests at a particular backend.
+*/
+int ldb_connect_backend(struct ldb_context *ldb,
+                       const char *url,
+                       const char *options[],
+                       struct ldb_module **backend_module)
+{
+       int ret;
+       char *backend;
+       ldb_connect_fn fn;
+
+       if (strchr(url, ':') != NULL) {
+               backend = talloc_strndup(ldb, url, strchr(url, ':')-url);
+       } else {
+               /* Default to tdb */
+               backend = talloc_strdup(ldb, "tdb");
+       }
+
+       fn = ldb_find_backend(backend);
+
+       if (fn == NULL) {
+               struct ldb_backend_ops *ops;
+               char *symbol_name = talloc_asprintf(ldb, "ldb_%s_backend_ops", backend);
+               if (symbol_name == NULL) {
+                       return LDB_ERR_OPERATIONS_ERROR;
+               }
+               ops = ldb_dso_load_symbol(ldb, backend, symbol_name);
+               if (ops != NULL) {
+                       fn = ops->connect_fn;
+               }
+               talloc_free(symbol_name);
+       }
+
+       talloc_free(backend);
+
+       if (fn == NULL) {
+               ldb_debug(ldb, LDB_DEBUG_FATAL,
+                         "Unable to find backend for '%s'\n", url);
+               return LDB_ERR_OTHER;
+       }
+
+       ret = fn(ldb, url, ldb->flags, options, backend_module);
+
+       if (ret != LDB_SUCCESS) {
+               ldb_debug(ldb, LDB_DEBUG_ERROR,
+                         "Failed to connect to '%s'\n", url);
+               return ret;
+       }
+       return ret;
+}
 
 static const struct ldb_module_ops *ldb_find_module_ops(const char *name)
 {
        struct ops_list_entry *e;
        int i;
 
-       for (i = 0; builtin_modules[i]; i++) {
-               if (strcmp(builtin_modules[i]->name, name) == 0)
-                       return builtin_modules[i];
+       for (i = 0; builtins[i].backend_ops || builtins[i].module_ops; i++) {
+               if (builtins[i].module_ops == NULL) continue;
+
+               if (strcmp(builtins[i].module_ops->name, name) == 0)
+                       return builtins[i].module_ops;
        }
  
        for (e = registered_modules; e; e = e->next) {
@@ -223,16 +336,6 @@ int ldb_load_modules_list(struct ldb_context *ldb, const char **module_list, str
                const struct ldb_module_ops *ops;
                
                ops = ldb_find_module_ops(module_list[i]);
-               if (ops == NULL) {
-                       int (*init_fn) (void);
-
-                       init_fn = ldb_dso_load_symbol(ldb, module_list[i], 
-                                                     "init_module");
-                       if (init_fn != NULL && init_fn() == 0) {
-                               ops = ldb_find_module_ops(module_list[i]);
-                       }
-               }
-
                if (ops == NULL) {
                        char *symbol_name = talloc_asprintf(ldb, "ldb_%s_module_ops", 
                                                                                                module_list[i]);
@@ -316,7 +419,7 @@ int ldb_load_modules(struct ldb_context *ldb, const char *options[])
                        return -1;
                }
 
-               ret = ldb_search_exp_fmt(ldb, mods_dn, &res, mods_dn, LDB_SCOPE_BASE, attrs, "@LIST=*");
+               ret = ldb_search(ldb, mods_dn, &res, mods_dn, LDB_SCOPE_BASE, attrs, "@LIST=*");
                
                if (ret == LDB_ERR_NO_SUCH_OBJECT) {
                        ldb_debug(ldb, LDB_DEBUG_TRACE, "no modules required by the db");
@@ -383,6 +486,12 @@ int ldb_load_modules(struct ldb_context *ldb, const char *options[])
 int ldb_next_request(struct ldb_module *module, struct ldb_request *request)
 {
        int ret;
+
+       if (request->callback == NULL) {
+               ldb_set_errstring(module->ldb, "Requests MUST define callbacks");
+               return LDB_ERR_UNWILLING_TO_PERFORM;
+       }
+
        switch (request->operation) {
        case LDB_SEARCH:
                FIND_OP(module, search);
@@ -408,10 +517,6 @@ int ldb_next_request(struct ldb_module *module, struct ldb_request *request)
                FIND_OP(module, extended);
                ret = module->ops->extended(module, request);
                break;
-       case LDB_SEQUENCE_NUMBER:
-               FIND_OP(module, sequence_number);
-               ret = module->ops->sequence_number(module, request);
-               break;
        default:
                FIND_OP(module, request);
                ret = module->ops->request(module, request);
@@ -451,3 +556,209 @@ int ldb_next_del_trans(struct ldb_module *module)
        FIND_OP(module, del_transaction);
        return module->ops->del_transaction(module);
 }
+
+struct ldb_handle *ldb_handle_new(TALLOC_CTX *mem_ctx, struct ldb_context *ldb)
+{
+       struct ldb_handle *h;
+
+       h = talloc_zero(mem_ctx, struct ldb_handle);
+       if (h == NULL) {
+               ldb_set_errstring(ldb, "Out of Memory");
+               return NULL;
+       }
+
+       h->status = LDB_SUCCESS;
+       h->state = LDB_ASYNC_INIT;
+       h->ldb = ldb;
+
+       return h;
+}
+
+/* calls the request callback to send an entry
+ *
+ * params:
+ *      req: the original request passed to your module
+ *      msg: reply message (must be a talloc pointer, and it will be stolen
+ *           on the ldb_reply that is sent to the callback)
+ *     ctrls: controls to send in the reply  (must be a talloc pointer, and it will be stolen
+ *           on the ldb_reply that is sent to the callback)
+ */
+
+int ldb_module_send_entry(struct ldb_request *req,
+                         struct ldb_message *msg,
+                         struct ldb_control **ctrls)
+{
+       struct ldb_reply *ares;
+
+       ares = talloc_zero(req, struct ldb_reply);
+       if (!ares) {
+               ldb_oom(req->handle->ldb);
+               req->callback(req, NULL);
+               return LDB_ERR_OPERATIONS_ERROR;
+       }
+       ares->type = LDB_REPLY_ENTRY;
+       ares->message = talloc_steal(ares, msg);
+       ares->controls = talloc_steal(ares, ctrls);
+       ares->error = LDB_SUCCESS;
+
+       return req->callback(req, ares);
+}
+
+/* calls the request callback to send an referrals
+ *
+ * params:
+ *      req: the original request passed to your module
+ *      ref: referral string (must be a talloc pointeri, steal)
+ */
+
+int ldb_module_send_referral(struct ldb_request *req,
+                                          char *ref)
+{
+       struct ldb_reply *ares;
+
+       ares = talloc_zero(req, struct ldb_reply);
+       if (!ares) {
+               ldb_oom(req->handle->ldb);
+               req->callback(req, NULL);
+               return LDB_ERR_OPERATIONS_ERROR;
+       }
+       ares->type = LDB_REPLY_REFERRAL;
+       ares->referral = talloc_steal(ares, ref);
+       ares->error = LDB_SUCCESS;
+
+       return req->callback(req, ares);
+}
+
+/* calls the original request callback
+ *
+ * params:
+ *     req:   the original request passed to your module
+ *     ctrls: controls to send in the reply (must be a talloc pointer, steal)
+ *     response: results for extended request (steal)
+ *     error: LDB_SUCCESS for a succesful return
+ *            any other ldb error otherwise
+ */
+int ldb_module_done(struct ldb_request *req,
+                   struct ldb_control **ctrls,
+                   struct ldb_extended *response,
+                   int error)
+{
+       struct ldb_reply *ares;
+
+       ares = talloc_zero(req, struct ldb_reply);
+       if (!ares) {
+               ldb_oom(req->handle->ldb);
+               req->callback(req, NULL);
+               return LDB_ERR_OPERATIONS_ERROR;
+       }
+       ares->type = LDB_REPLY_DONE;
+       ares->controls = talloc_steal(ares, ctrls);
+       ares->response = talloc_steal(ares, response);
+       ares->error = error;
+
+       req->callback(req, ares);
+       return error;
+}
+
+/* to be used *only* in modules init functions.
+ * this function i synchronous and will register
+ * the requested OID in the rootdse module if present
+ * otherwise it will return an error */
+int ldb_mod_register_control(struct ldb_module *module, const char *oid)
+{
+       struct ldb_request *req;
+       int ret;
+
+       req = talloc_zero(module, struct ldb_request);
+       if (req == NULL) {
+               return LDB_ERR_OPERATIONS_ERROR;
+       }
+
+       req->operation = LDB_REQ_REGISTER_CONTROL;
+       req->op.reg_control.oid = oid;
+       req->callback = ldb_op_default_callback;
+
+       ldb_set_timeout(module->ldb, req, 0);
+
+       req->handle = ldb_handle_new(req, module->ldb);
+       if (req->handle == NULL) {
+               return LDB_ERR_OPERATIONS_ERROR;
+       }
+
+       ret = ldb_request(module->ldb, req);
+       if (ret == LDB_SUCCESS) {
+               ret = ldb_wait(req->handle, LDB_WAIT_ALL);
+       }
+       talloc_free(req);
+
+       return ret;
+}
+
+#ifndef STATIC_LIBLDB_MODULES
+
+#ifdef HAVE_LDB_LDAP
+#define LDAP_BACKEND LDB_BACKEND(ldap), LDB_BACKEND(ldapi), LDB_BACKEND(ldaps),
+#else
+#define LDAP_BACKEND
+#endif
+
+#ifdef HAVE_LDB_SQLITE3
+#define SQLITE3_BACKEND LDB_BACKEND(sqlite3),
+#else
+#define SQLITE3_BACKEND
+#endif
+
+#define STATIC_LIBLDB_MODULES \
+       LDB_BACKEND(tdb),       \
+       LDAP_BACKEND    \
+       SQLITE3_BACKEND \
+       LDB_MODULE(operational),        \
+       LDB_MODULE(rdn_name),   \
+       LDB_MODULE(paged_results),      \
+       LDB_MODULE(server_sort),                \
+       LDB_MODULE(asq), \
+       NULL
+#endif
+
+/*
+ * this is a bit hacked, as STATIC_LIBLDB_MODULES contains ','
+ * between the elements and we want to autogenerate the
+ * extern struct declarations, so we do some hacks and let the
+ * ',' appear in an unused function prototype.
+ */
+#undef NULL
+#define NULL LDB_MODULE(NULL),
+
+#define LDB_BACKEND(name) \
+       int); \
+       extern const struct ldb_backend_ops ldb_ ## name ## _backend_ops;\
+       extern void ldb_noop ## name (int
+#define LDB_MODULE(name) \
+       int); \
+       extern const struct ldb_module_ops ldb_ ## name ## _module_ops;\
+       extern void ldb_noop ## name (int
+
+extern void ldb_start_noop(int,
+STATIC_LIBLDB_MODULES
+int);
+
+#undef NULL
+#define NULL { \
+       .backend_ops = (void *)0, \
+       .module_ops = (void *)0 \
+}
+
+#undef LDB_BACKEND
+#define LDB_BACKEND(name) { \
+       .backend_ops = &ldb_ ## name ## _backend_ops, \
+       .module_ops = (void *)0 \
+}
+#undef LDB_MODULE
+#define LDB_MODULE(name) { \
+       .backend_ops = (void *)0, \
+       .module_ops = &ldb_ ## name ## _module_ops \
+}
+
+static const struct ldb_builtins builtins[] = {
+       STATIC_LIBLDB_MODULES
+};