LDB:rdn name module - change counters to "unsigned" where appropriate
[ira/wip.git] / source4 / lib / ldb / modules / rdn_name.c
index 0da8b8f5738e42044e0abad716a4f03cd1d46e25..a473f251af61aadaadf7d1e444ead25c36fd8fe3 100644 (file)
@@ -1,8 +1,8 @@
 /* 
    ldb database library
 
-   Copyright (C) Andrew Bartlet 2005
-   Copyright (C) Simo Sorce     2006
+   Copyright (C) Andrew Bartlett 2005-2009
+   Copyright (C) Simo Sorce 2006-2008
 
      ** NOTE! The following LGPL license applies to the ldb
      ** library. This does NOT imply that all of Samba is released
@@ -11,7 +11,7 @@
    This library is free software; you can redistribute it and/or
    modify it under the terms of the GNU Lesser General Public
    License as published by the Free Software Foundation; either
-   version 2 of the License, or (at your option) any later version.
+   version 3 of the License, or (at your option) any later version.
 
    This library is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    Lesser General Public License for more details.
 
    You should have received a copy of the GNU Lesser General Public
-   License along with this library; if not, write to the Free Software
-   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+   License along with this library; if not, see <http://www.gnu.org/licenses/>.
 */
 
 /*
- *  Name: rdb_name
+ *  Name: rdn_name
  *
  *  Component: ldb rdn name module
  *
  *  Description: keep a consistent name attribute on objects manpulations
  *
- *  Author: Andrew Bartlet
+ *  Author: Andrew Bartlett
  *
  *  Modifications:
  *    - made the module async
  *      Simo Sorce Mar 2006
  */
 
-#include "includes.h"
-#include "ldb/include/includes.h"
+#include "ldb_includes.h"
+#include "ldb_module.h"
+
+struct rename_context {
+       struct ldb_module *module;
+       struct ldb_request *req;
+
+       struct ldb_reply *ares;
+};
 
 static struct ldb_message_element *rdn_name_find_attribute(const struct ldb_message *msg, const char *name)
 {
-       int i;
+       unsigned int i;
 
        for (i = 0; i < msg->num_elements; i++) {
                if (ldb_attr_cmp(name, msg->elements[i].name) == 0) {
@@ -53,37 +59,66 @@ static struct ldb_message_element *rdn_name_find_attribute(const struct ldb_mess
        return NULL;
 }
 
+static int rdn_name_add_callback(struct ldb_request *req,
+                                struct ldb_reply *ares)
+{
+       struct rename_context *ac;
+
+       ac = talloc_get_type(req->context, struct rename_context);
+
+       if (!ares) {
+               return ldb_module_done(ac->req, NULL, NULL,
+                                       LDB_ERR_OPERATIONS_ERROR);
+       }
+       if (ares->error != LDB_SUCCESS) {
+               return ldb_module_done(ac->req, ares->controls,
+                                       ares->response, ares->error);
+       }
+
+       if (ares->type != LDB_REPLY_DONE) {
+               return ldb_module_done(ac->req, NULL, NULL,
+                                       LDB_ERR_OPERATIONS_ERROR);
+       }
+
+       return ldb_module_done(ac->req, ares->controls,
+                                       ares->response, LDB_SUCCESS);
+}
+
 static int rdn_name_add(struct ldb_module *module, struct ldb_request *req)
 {
+       struct ldb_context *ldb;
        struct ldb_request *down_req;
+       struct rename_context *ac;
        struct ldb_message *msg;
        struct ldb_message_element *attribute;
+       const struct ldb_schema_attribute *a;
        const char *rdn_name;
        struct ldb_val rdn_val;
-       int i, ret;
+       unsigned int i;
+       int ret;
 
-       ldb_debug(module->ldb, LDB_DEBUG_TRACE, "rdn_name_add_record\n");
+       ldb = ldb_module_get_ctx(module);
 
        /* do not manipulate our control entries */
        if (ldb_dn_is_special(req->op.add.message->dn)) {
                return ldb_next_request(module, req);
        }
 
-       down_req = talloc(req, struct ldb_request);
-       if (down_req == NULL) {
+       ac = talloc_zero(req, struct rename_context);
+       if (ac == NULL) {
                return LDB_ERR_OPERATIONS_ERROR;
        }
 
-       *down_req = *req;
+       ac->module = module;
+       ac->req = req;
 
-       down_req->op.add.message = msg = ldb_msg_copy_shallow(down_req, req->op.add.message);
+       msg = ldb_msg_copy_shallow(req, req->op.add.message);
        if (msg == NULL) {
                return LDB_ERR_OPERATIONS_ERROR;
        }
 
        rdn_name = ldb_dn_get_rdn_name(msg->dn);
        if (rdn_name == NULL) {
-               talloc_free(down_req);
                return LDB_ERR_OPERATIONS_ERROR;
        }
        
@@ -95,7 +130,6 @@ static int rdn_name_add(struct ldb_module *module, struct ldb_request *req)
        }
 
        if (ldb_msg_add_value(msg, "name", &rdn_val, NULL) != 0) {
-               talloc_free(down_req);
                return LDB_ERR_OPERATIONS_ERROR;
        }
 
@@ -103,241 +137,222 @@ static int rdn_name_add(struct ldb_module *module, struct ldb_request *req)
 
        if (!attribute) {
                if (ldb_msg_add_value(msg, rdn_name, &rdn_val, NULL) != 0) {
-                       talloc_free(down_req);
                        return LDB_ERR_OPERATIONS_ERROR;
                }
        } else {
-               const struct ldb_attrib_handler *handler = ldb_attrib_handler(module->ldb, rdn_name);
+               a = ldb_schema_attribute_by_name(ldb, rdn_name);
 
                for (i = 0; i < attribute->num_values; i++) {
-                       if (handler->comparison_fn(module->ldb, msg, &rdn_val, &attribute->values[i]) == 0) {
+                       ret = a->syntax->comparison_fn(ldb, msg,
+                                       &rdn_val, &attribute->values[i]);
+                       if (ret == 0) {
                                /* overwrite so it matches in case */
                                attribute->values[i] = rdn_val;
                                break;
                        }
                }
                if (i == attribute->num_values) {
-                       ldb_debug_set(module->ldb, LDB_DEBUG_FATAL, 
-                                     "RDN mismatch on %s: %s (%s)", 
-                                     ldb_dn_get_linearized(msg->dn), rdn_name, rdn_val.data);
-                       talloc_free(down_req);
-                       return LDB_ERR_OPERATIONS_ERROR;
+                       char *rdn_errstring = talloc_asprintf(ac,
+                               "RDN mismatch on %s: %s (%.*s) should match one of:", 
+                               ldb_dn_get_linearized(msg->dn), rdn_name, 
+                               (int)rdn_val.length, (const char *)rdn_val.data);
+                       for (i = 0; i < attribute->num_values; i++) {
+                               rdn_errstring = talloc_asprintf_append(
+                                       rdn_errstring, " (%.*s)",
+                                       (int)attribute->values[i].length, 
+                                       (const char *)attribute->values[i].data);
+                       }
+                       ldb_set_errstring(ldb, rdn_errstring);
+                       /* Match AD's error here */
+                       return LDB_ERR_INVALID_DN_SYNTAX;
                }
        }
 
-       /* go on with the call chain */
-       ret = ldb_next_request(module, down_req);
-
-       /* do not free down_req as the call results may be linked to it,
-        * it will be freed when the upper level request get freed */
-       if (ret == LDB_SUCCESS) {
-               req->handle = down_req->handle;
+       ret = ldb_build_add_req(&down_req, ldb, req,
+                               msg,
+                               req->controls,
+                               ac, rdn_name_add_callback,
+                               req);
+       if (ret != LDB_SUCCESS) {
+               return ret;
        }
 
-       return ret;
-}
-
-struct rename_context {
+       talloc_steal(down_req, msg);
 
-       enum {RENAME_RENAME, RENAME_MODIFY} step;
-       struct ldb_request *orig_req;
-       struct ldb_request *down_req;
-       struct ldb_request *mod_req;
-};
+       /* go on with the call chain */
+       return ldb_next_request(module, down_req);
+}
 
-static int rdn_name_rename(struct ldb_module *module, struct ldb_request *req)
+static int rdn_modify_callback(struct ldb_request *req, struct ldb_reply *ares)
 {
-       struct ldb_handle *h;
        struct rename_context *ac;
 
-       ldb_debug(module->ldb, LDB_DEBUG_TRACE, "rdn_name_rename\n");
+       ac = talloc_get_type(req->context, struct rename_context);
 
-       /* do not manipulate our control entries */
-       if (ldb_dn_is_special(req->op.rename.newdn)) {
-               return ldb_next_request(module, req);
+       if (!ares) {
+               return ldb_module_done(ac->req, NULL, NULL,
+                                       LDB_ERR_OPERATIONS_ERROR);
        }
-
-       h = talloc_zero(req, struct ldb_handle);
-       if (h == NULL) {
-               return LDB_ERR_OPERATIONS_ERROR;
+       if (ares->error != LDB_SUCCESS) {
+               return ldb_module_done(ac->req, ares->controls,
+                                       ares->response, ares->error);
        }
 
-       h->module = module;
-
-       ac = talloc_zero(h, struct rename_context);
-       if (ac == NULL) {
-               return LDB_ERR_OPERATIONS_ERROR;
+       /* the only supported reply right now is a LDB_REPLY_DONE */
+       if (ares->type != LDB_REPLY_DONE) {
+               return ldb_module_done(ac->req, NULL, NULL,
+                                       LDB_ERR_OPERATIONS_ERROR);
        }
 
-       h->private_data = (void *)ac;
-
-       h->state = LDB_ASYNC_INIT;
-       h->status = LDB_SUCCESS;
-
-       ac->orig_req = req;
-       ac->down_req = talloc(req, struct ldb_request);
-       if (ac->down_req == NULL) {
-               return LDB_ERR_OPERATIONS_ERROR;
-       }
-
-       *(ac->down_req) = *req;
-
-       ac->step = RENAME_RENAME;
-
-       req->handle = h;
-
-       /* rename first, modify "name" if rename is ok */
-       return ldb_next_request(module, ac->down_req);
+       /* send saved controls eventually */
+       return ldb_module_done(ac->req, ac->ares->controls,
+                               ac->ares->response, LDB_SUCCESS);
 }
 
-static int rdn_name_rename_do_mod(struct ldb_handle *h) {
-
+static int rdn_rename_callback(struct ldb_request *req, struct ldb_reply *ares)
+{
+       struct ldb_context *ldb;
        struct rename_context *ac;
+       struct ldb_request *mod_req;
        const char *rdn_name;
        struct ldb_val rdn_val;
        struct ldb_message *msg;
+       int ret;
+
+       ac = talloc_get_type(req->context, struct rename_context);
+       ldb = ldb_module_get_ctx(ac->module);
+
+       if (!ares) {
+               goto error;
+       }
+       if (ares->error != LDB_SUCCESS) {
+               return ldb_module_done(ac->req, ares->controls,
+                                       ares->response, ares->error);
+       }
 
-       ac = talloc_get_type(h->private_data, struct rename_context);
+       /* the only supported reply right now is a LDB_REPLY_DONE */
+       if (ares->type != LDB_REPLY_DONE) {
+               goto error;
+       }
 
-       ac->mod_req = talloc_zero(ac, struct ldb_request);
+       /* save reply for caller */
+       ac->ares = talloc_steal(ac, ares);
 
-       ac->mod_req->operation = LDB_MODIFY;
-       ac->mod_req->op.mod.message = msg = ldb_msg_new(ac->mod_req);
+       msg = ldb_msg_new(ac);
        if (msg == NULL) {
-               return LDB_ERR_OPERATIONS_ERROR;
+               goto error;
        }
-
-       msg->dn = ldb_dn_copy(msg, ac->orig_req->op.rename.newdn);
+       msg->dn = ldb_dn_copy(msg, ac->req->op.rename.newdn);
        if (msg->dn == NULL) {
-               return LDB_ERR_OPERATIONS_ERROR;
+               goto error;
        }
-
-       rdn_name = ldb_dn_get_rdn_name(ac->orig_req->op.rename.newdn);
+       rdn_name = ldb_dn_get_rdn_name(ac->req->op.rename.newdn);
        if (rdn_name == NULL) {
-               return LDB_ERR_OPERATIONS_ERROR;
+               goto error;
        }
        
-       rdn_val = ldb_val_dup(msg, ldb_dn_get_rdn_val(ac->orig_req->op.rename.newdn));
+       rdn_val = ldb_val_dup(msg, ldb_dn_get_rdn_val(ac->req->op.rename.newdn));
        
        if (ldb_msg_add_empty(msg, rdn_name, LDB_FLAG_MOD_REPLACE, NULL) != 0) {
-               return LDB_ERR_OPERATIONS_ERROR;
+               goto error;
        }
        if (ldb_msg_add_value(msg, rdn_name, &rdn_val, NULL) != 0) {
-               return LDB_ERR_OPERATIONS_ERROR;
+               goto error;
        }
        if (ldb_msg_add_empty(msg, "name", LDB_FLAG_MOD_REPLACE, NULL) != 0) {
-               return LDB_ERR_OPERATIONS_ERROR;
+               goto error;
        }
        if (ldb_msg_add_value(msg, "name", &rdn_val, NULL) != 0) {
-               return LDB_ERR_OPERATIONS_ERROR;
+               goto error;
        }
 
-       ldb_set_timeout_from_prev_req(h->module->ldb, ac->orig_req, ac->mod_req);
+       ret = ldb_build_mod_req(&mod_req, ldb,
+                               ac, msg, NULL,
+                               ac, rdn_modify_callback,
+                               req);
+       if (ret != LDB_SUCCESS) {
+               return ldb_module_done(ac->req, NULL, NULL, ret);
+       }
+       talloc_steal(mod_req, msg);
 
-       ac->step = RENAME_MODIFY;
+       /* go on with the call chain */
+       return ldb_next_request(ac->module, mod_req);
 
-       /* do the mod call */
-       return ldb_request(h->module->ldb, ac->mod_req);
+error:
+       return ldb_module_done(ac->req, NULL, NULL,
+                                                LDB_ERR_OPERATIONS_ERROR);
 }
 
-static int rename_wait(struct ldb_handle *handle)
+static int rdn_name_rename(struct ldb_module *module, struct ldb_request *req)
 {
+       struct ldb_context *ldb;
        struct rename_context *ac;
+       struct ldb_request *down_req;
        int ret;
-    
-       if (!handle || !handle->private_data) {
-               return LDB_ERR_OPERATIONS_ERROR;
-       }
-
-       if (handle->state == LDB_ASYNC_DONE) {
-               return handle->status;
-       }
-
-       handle->state = LDB_ASYNC_PENDING;
-       handle->status = LDB_SUCCESS;
-
-       ac = talloc_get_type(handle->private_data, struct rename_context);
-
-       switch(ac->step) {
-       case RENAME_RENAME:
-               ret = ldb_wait(ac->down_req->handle, LDB_WAIT_NONE);
-               if (ret != LDB_SUCCESS) {
-                       handle->status = ret;
-                       goto done;
-               }
-               if (ac->down_req->handle->status != LDB_SUCCESS) {
-                       handle->status = ac->down_req->handle->status;
-                       goto done;
-               }
-
-               if (ac->down_req->handle->state != LDB_ASYNC_DONE) {
-                       return LDB_SUCCESS;
-               }
 
-               /* rename operation done */
-               return rdn_name_rename_do_mod(handle);
+       ldb = ldb_module_get_ctx(module);
 
-       case RENAME_MODIFY:
-               ret = ldb_wait(ac->mod_req->handle, LDB_WAIT_NONE);
-               if (ret != LDB_SUCCESS) {
-                       handle->status = ret;
-                       goto done;
-               }
-               if (ac->mod_req->handle->status != LDB_SUCCESS) {
-                       handle->status = ac->mod_req->handle->status;
-                       goto done;
-               }
-
-               if (ac->mod_req->handle->state != LDB_ASYNC_DONE) {
-                       return LDB_SUCCESS;
-               }
-
-               break;
+       /* do not manipulate our control entries */
+       if (ldb_dn_is_special(req->op.rename.newdn)) {
+               return ldb_next_request(module, req);
+       }
 
-       default:
-               ret = LDB_ERR_OPERATIONS_ERROR;
-               goto done;
+       ac = talloc_zero(req, struct rename_context);
+       if (ac == NULL) {
+               return LDB_ERR_OPERATIONS_ERROR;
        }
 
-       ret = LDB_SUCCESS;
+       ac->module = module;
+       ac->req = req;
+
+       ret = ldb_build_rename_req(&down_req,
+                                  ldb,
+                                  ac,
+                                  req->op.rename.olddn,
+                                  req->op.rename.newdn,
+                                  req->controls,
+                                  ac,
+                                  rdn_rename_callback,
+                                  req);
+
+       if (ret != LDB_SUCCESS) {
+               return ret;
+       }
 
-done:
-       handle->state = LDB_ASYNC_DONE;
-       return ret;
+       /* rename first, modify "name" if rename is ok */
+       return ldb_next_request(module, down_req);
 }
 
-static int rename_wait_all(struct ldb_handle *handle) {
+static int rdn_name_modify(struct ldb_module *module, struct ldb_request *req)
+{
+       struct ldb_context *ldb;
 
-       int ret;
+       ldb = ldb_module_get_ctx(module);
 
-       while (handle->state != LDB_ASYNC_DONE) {
-               ret = rename_wait(handle);
-               if (ret != LDB_SUCCESS) {
-                       return ret;
-               }
+       /* do not manipulate our control entries */
+       if (ldb_dn_is_special(req->op.mod.message->dn)) {
+               return ldb_next_request(module, req);
        }
 
-       return handle->status;
-}
+       if (ldb_msg_find_element(req->op.mod.message, "name")) {
+               ldb_asprintf_errstring(ldb, "Modify of 'name' on %s not permitted, must use 'rename' operation instead",
+                                      ldb_dn_get_linearized(req->op.mod.message->dn));
+               return LDB_ERR_NOT_ALLOWED_ON_RDN;
+       }
 
-static int rdn_name_wait(struct ldb_handle *handle, enum ldb_wait_type type)
-{
-       if (type == LDB_WAIT_ALL) {
-               return rename_wait_all(handle);
-       } else {
-               return rename_wait(handle);
+       if (ldb_msg_find_element(req->op.mod.message, ldb_dn_get_rdn_name(req->op.mod.message->dn))) {
+               ldb_asprintf_errstring(ldb, "Modify of RDN '%s' on %s not permitted, must use 'rename' operation instead",
+                                      ldb_dn_get_rdn_name(req->op.mod.message->dn), ldb_dn_get_linearized(req->op.mod.message->dn));
+               return LDB_ERR_NOT_ALLOWED_ON_RDN;
        }
+
+       /* All OK, they kept their fingers out of the special attributes */
+       return ldb_next_request(module, req);
 }
 
-static const struct ldb_module_ops rdn_name_ops = {
+const struct ldb_module_ops ldb_rdn_name_module_ops = {
        .name              = "rdn_name",
        .add               = rdn_name_add,
-       .rename            = rdn_name_rename,
-       .wait              = rdn_name_wait
+       .modify            = rdn_name_modify,
+       .rename            = rdn_name_rename
 };
-
-
-int ldb_rdn_name_init(void)
-{
-       return ldb_register_module(&rdn_name_ops);
-}