s4:dsdb Don't allow creation of systemOnly objectclasses
[sfrench/samba-autobuild/.git] / source4 / dsdb / samdb / ldb_modules / objectclass.c
index b048a8d8e1e2aa06a7fa05101f5db4fdea9cc84c..b3d54612dde9a15ddc7fad3872dded1b6d4242d8 100644 (file)
@@ -1,7 +1,7 @@
 /* 
    ldb database library
 
-   Copyright (C) Simo Sorce  2006
+   Copyright (C) Simo Sorce  2006-2008
    Copyright (C) Andrew Bartlett <abartlet@samba.org> 2005-2007
 
    This program is free software; you can redistribute it and/or modify
 
 
 #include "includes.h"
-#include "ldb/include/ldb.h"
-#include "ldb/include/ldb_errors.h"
-#include "ldb/include/ldb_private.h"
+#include "ldb_module.h"
+#include "dlinklist.h"
 #include "dsdb/samdb/samdb.h"
-#include "lib/util/dlinklist.h"
 #include "librpc/ndr/libndr.h"
 #include "librpc/gen_ndr/ndr_security.h"
 #include "libcli/security/security.h"
 
 struct oc_context {
 
-       enum oc_step {OC_DO_REQ, OC_SEARCH_SELF, OC_DO_MOD, 
-                     OC_SEARCH_ADD_PARENT, OC_DO_ADD, 
-                     OC_SEARCH_RENAME_PARENT, OC_DO_RENAME} step;
-
        struct ldb_module *module;
-       struct ldb_request *orig_req;
-
-       struct ldb_request *down_req;
+       struct ldb_request *req;
 
-       struct ldb_request *search_req;
        struct ldb_reply *search_res;
 
-       struct ldb_request *add_req;
-       struct ldb_request *mod_req;
-       struct ldb_request *rename_req;
+       int (*step_fn)(struct oc_context *);
 };
 
 struct class_list {
@@ -69,54 +58,46 @@ struct class_list {
        const struct dsdb_class *objectclass;
 };
 
-static int objectclass_do_add(struct ldb_handle *h);
-
-static struct ldb_handle *oc_init_handle(struct ldb_request *req, struct ldb_module *module)
+static struct oc_context *oc_init_context(struct ldb_module *module,
+                                         struct ldb_request *req)
 {
+       struct ldb_context *ldb;
        struct oc_context *ac;
-       struct ldb_handle *h;
-
-       h = talloc_zero(req, struct ldb_handle);
-       if (h == NULL) {
-               ldb_set_errstring(module->ldb, "Out of Memory");
-               return NULL;
-       }
 
-       h->module = module;
+       ldb = ldb_module_get_ctx(module);
 
-       ac = talloc_zero(h, struct oc_context);
+       ac = talloc_zero(req, struct oc_context);
        if (ac == NULL) {
-               ldb_set_errstring(module->ldb, "Out of Memory");
-               talloc_free(h);
+               ldb_set_errstring(ldb, "Out of Memory");
                return NULL;
        }
 
-       h->private_data = (void *)ac;
-
-       h->state = LDB_ASYNC_INIT;
-       h->status = LDB_SUCCESS;
-
        ac->module = module;
-       ac->orig_req = req;
+       ac->req = req;
 
-       return h;
+       return ac;
 }
 
+static int objectclass_do_add(struct oc_context *ac);
+
 /* Sort objectClasses into correct order, and validate that all
  * objectClasses specified actually exist in the schema
  */
 
 static int objectclass_sort(struct ldb_module *module,
                            const struct dsdb_schema *schema,
-                           struct ldb_message *msg, /* so that when we create new elements, we put it on the right parent */
                            TALLOC_CTX *mem_ctx,
                            struct ldb_message_element *objectclass_element,
                            struct class_list **sorted_out) 
 {
+       struct ldb_context *ldb;
        int i;
        int layer;
        struct class_list *sorted = NULL, *parent_class = NULL,
                *subclass = NULL, *unsorted = NULL, *current, *poss_subclass, *poss_parent, *new_parent;
+
+       ldb = ldb_module_get_ctx(module);
+
        /* DESIGN:
         *
         * We work on 4 different 'bins' (implemented here as linked lists):
@@ -150,14 +131,20 @@ static int objectclass_sort(struct ldb_module *module,
        for (i=0; i < objectclass_element->num_values; i++) {
                current = talloc(mem_ctx, struct class_list);
                if (!current) {
-                       ldb_set_errstring(module->ldb, "objectclass: out of memory allocating objectclass list");
-                       talloc_free(mem_ctx);
+                       ldb_oom(ldb);
                        return LDB_ERR_OPERATIONS_ERROR;
                }
-               current->objectclass = dsdb_class_by_lDAPDisplayName(schema, (const char *)objectclass_element->values[i].data);
+               current->objectclass = dsdb_class_by_lDAPDisplayName_ldb_val(schema, &objectclass_element->values[i]);
                if (!current->objectclass) {
-                       ldb_asprintf_errstring(module->ldb, "objectclass %s is not a valid objectClass in schema", (const char *)objectclass_element->values[i].data);
-                       return LDB_ERR_OBJECT_CLASS_VIOLATION;
+                       ldb_asprintf_errstring(ldb, "objectclass %.*s is not a valid objectClass in schema", 
+                                              (int)objectclass_element->values[i].length, (const char *)objectclass_element->values[i].data);
+                       /* This looks weird, but windows apparently returns this for invalid objectClass values */
+                       return LDB_ERR_NO_SUCH_ATTRIBUTE;
+               } else if (current->objectclass->isDefunct) {
+                       ldb_asprintf_errstring(ldb, "objectclass %.*s marked as isDefunct objectClass in schema - not valid for new objects", 
+                                              (int)objectclass_element->values[i].length, (const char *)objectclass_element->values[i].data);
+                       /* This looks weird, but windows apparently returns this for invalid objectClass values */
+                       return LDB_ERR_NO_SUCH_ATTRIBUTE;
                }
 
                /* this is the root of the tree.  We will start
@@ -246,72 +233,83 @@ static int objectclass_sort(struct ldb_module *module,
         * was no 'top', a conflict in the objectClasses or some other
         * schema error?
         */
-       ldb_asprintf_errstring(module->ldb, "objectclass %s is not a valid objectClass in objectClass chain", unsorted->objectclass->lDAPDisplayName);
+       ldb_asprintf_errstring(ldb, "objectclass %s is not a valid objectClass in objectClass chain", unsorted->objectclass->lDAPDisplayName);
        return LDB_ERR_OBJECT_CLASS_VIOLATION;
 }
 
-static DATA_BLOB *get_sd(struct ldb_module *module, TALLOC_CTX *mem_ctx, 
-                        const struct dsdb_class *objectclass) 
+static int get_search_callback(struct ldb_request *req, struct ldb_reply *ares)
 {
-       enum ndr_err_code ndr_err;
-       DATA_BLOB *linear_sd;
-       struct auth_session_info *session_info
-               = ldb_get_opaque(module->ldb, "sessionInfo");
-       struct security_descriptor *sd;
-       struct dom_sid *domain_sid = samdb_domain_sid(module->ldb);
-
-       if (!objectclass->defaultSecurityDescriptor || !domain_sid) {
-               return NULL;
-       }
-       
-       sd = sddl_decode(mem_ctx, 
-                        objectclass->defaultSecurityDescriptor,
-                        domain_sid);
+       struct ldb_context *ldb;
+       struct oc_context *ac;
+       int ret;
 
-       if (!sd || !session_info || !session_info->security_token) {
-               return NULL;
+       ac = talloc_get_type(req->context, struct oc_context);
+       ldb = ldb_module_get_ctx(ac->module);
+
+       if (!ares) {
+               return ldb_module_done(ac->req, NULL, NULL,
+                                       LDB_ERR_OPERATIONS_ERROR);
        }
-       
-       sd->owner_sid = session_info->security_token->user_sid;
-       sd->group_sid = session_info->security_token->group_sid;
-       
-       linear_sd = talloc(mem_ctx, DATA_BLOB);
-       if (!linear_sd) {
-               return NULL;
+       if (ares->error != LDB_SUCCESS &&
+           ares->error != LDB_ERR_NO_SUCH_OBJECT) {
+               return ldb_module_done(ac->req, ares->controls,
+                                       ares->response, ares->error);
        }
 
-       ndr_err = ndr_push_struct_blob(linear_sd, mem_ctx, 
-                                       lp_iconv_convenience(ldb_get_opaque(module->ldb, "loadparm")),
-                                      sd,
-                                      (ndr_push_flags_fn_t)ndr_push_security_descriptor);
-       if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
-               return NULL;
+       ldb_reset_err_string(ldb);
+
+       switch (ares->type) {
+       case LDB_REPLY_ENTRY:
+               if (ac->search_res != NULL) {
+                       ldb_set_errstring(ldb, "Too many results");
+                       talloc_free(ares);
+                       return ldb_module_done(ac->req, NULL, NULL,
+                                               LDB_ERR_OPERATIONS_ERROR);
+               }
+
+               ac->search_res = talloc_steal(ac, ares);
+               break;
+
+       case LDB_REPLY_REFERRAL:
+               /* ignore */
+               talloc_free(ares);
+               break;
+
+       case LDB_REPLY_DONE:
+               talloc_free(ares);
+               ret = ac->step_fn(ac);
+               if (ret != LDB_SUCCESS) {
+                       return ldb_module_done(ac->req, NULL, NULL, ret);
+               }
+               break;
        }
-       
-       return linear_sd;
 
+       return LDB_SUCCESS;
 }
 
-static int get_search_callback(struct ldb_context *ldb, void *context, struct ldb_reply *ares)
+static int oc_op_callback(struct ldb_request *req, struct ldb_reply *ares)
 {
        struct oc_context *ac;
 
-       ac = talloc_get_type(context, struct oc_context);
+       ac = talloc_get_type(req->context, struct oc_context);
 
-       /* we are interested only in the single reply (base search) we receive here */
-       if (ares->type == LDB_REPLY_ENTRY) {
-               if (ac->search_res != NULL) {
-                       ldb_set_errstring(ldb, "Too many results");
-                       talloc_free(ares);
-                       return LDB_ERR_OPERATIONS_ERROR;
-               }
+       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);
+       }
 
-               ac->search_res = talloc_move(ac, &ares);
-       } else {
+       if (ares->type != LDB_REPLY_DONE) {
                talloc_free(ares);
+               return ldb_module_done(ac->req, NULL, NULL,
+                                       LDB_ERR_OPERATIONS_ERROR);
        }
 
-       return LDB_SUCCESS;
+       return ldb_module_done(ac->req, ares->controls,
+                               ares->response, ares->error);
 }
 
 /* Fix up the DN to be in the standard form, taking particular care to match the parent DN
@@ -357,141 +355,155 @@ static int fix_attributes(struct ldb_context *ldb, const struct dsdb_schema *sch
        int i;
        for (i=0; i < msg->num_elements; i++) {
                const struct dsdb_attribute *attribute = dsdb_attribute_by_lDAPDisplayName(schema, msg->elements[i].name);
+               /* Add in a very special case for 'clearTextPassword',
+                * which is used for internal processing only, and is
+                * not presented in the schema */
                if (!attribute) {
-                       ldb_asprintf_errstring(ldb, "attribute %s is not a valid attribute in schema", msg->elements[i].name);
-                       return LDB_ERR_UNDEFINED_ATTRIBUTE_TYPE;
+                       if (strcasecmp(msg->elements[i].name, "clearTextPassword") != 0) {
+                               ldb_asprintf_errstring(ldb, "attribute %s is not a valid attribute in schema", msg->elements[i].name);
+                               /* Apparently Windows sends exactly this behaviour */
+                               return LDB_ERR_NO_SUCH_ATTRIBUTE;
+                       }
+               } else {
+                       msg->elements[i].name = attribute->lDAPDisplayName;
                }
-               msg->elements[i].name = attribute->lDAPDisplayName;
        }
 
        return LDB_SUCCESS;
 }
 
+static int objectclass_do_add(struct oc_context *ac);
+
 static int objectclass_add(struct ldb_module *module, struct ldb_request *req)
 {
-
-       static const char * const attrs[] = { NULL };
-
-       struct ldb_handle *h;
+       struct ldb_context *ldb;
+       struct ldb_request *search_req;
        struct oc_context *ac;
        struct ldb_dn *parent_dn;
        int ret;
-       
-       ldb_debug(module->ldb, LDB_DEBUG_TRACE, "objectclass_add\n");
+       static const char * const parent_attrs[] = { "objectGUID", NULL };
+
+       ldb = ldb_module_get_ctx(module);
+
+       ldb_debug(ldb, LDB_DEBUG_TRACE, "objectclass_add\n");
 
        /* do not manipulate our control entries */
        if (ldb_dn_is_special(req->op.add.message->dn)) {
                return ldb_next_request(module, req);
        }
 
-       /* Need to object to this, but cn=rootdse doesn't hae an objectClass... */
+       /* the objectClass must be specified on add */
        if (ldb_msg_find_element(req->op.add.message, 
                                 "objectClass") == NULL) {
-               return ldb_next_request(module, req);
+               return LDB_ERR_OBJECT_CLASS_VIOLATION;
        }
 
-       h = oc_init_handle(req, module);
-       if (!h) {
+       ac = oc_init_context(module, req);
+       if (ac == NULL) {
                return LDB_ERR_OPERATIONS_ERROR;
        }
-       ac = talloc_get_type(h->private_data, struct oc_context);
-       
-       /* return or own handle to deal with this call */
-       req->handle = h;
 
        /* If there isn't a parent, just go on to the add processing */
-       if (ldb_dn_get_comp_num(ac->orig_req->op.add.message->dn) == 1) {
-               return objectclass_do_add(h);
+       if (ldb_dn_get_comp_num(ac->req->op.add.message->dn) == 1) {
+               return objectclass_do_add(ac);
        }
 
-       parent_dn = ldb_dn_get_parent(ac, ac->orig_req->op.add.message->dn);
+       /* get copy of parent DN */
+       parent_dn = ldb_dn_get_parent(ac, ac->req->op.add.message->dn);
        if (parent_dn == NULL) {
-               ldb_oom(module->ldb);
+               ldb_oom(ldb);
                return LDB_ERR_OPERATIONS_ERROR;
        }
 
-       ret = ldb_build_search_req(&ac->search_req, module->ldb,
+       ret = ldb_build_search_req(&search_req, ldb,
                                   ac, parent_dn, LDB_SCOPE_BASE,
-                                  "(objectClass=*)",
-                                  attrs, NULL, 
-                                  ac, get_search_callback);
+                                  "(objectClass=*)", parent_attrs,
+                                  NULL,
+                                  ac, get_search_callback,
+                                  req);
        if (ret != LDB_SUCCESS) {
                return ret;
        }
+       talloc_steal(search_req, parent_dn);
 
-       talloc_steal(ac->search_req, parent_dn);
-
-       ldb_set_timeout_from_prev_req(ac->module->ldb, ac->orig_req, ac->search_req);
+       ac->step_fn = objectclass_do_add;
 
-       ac->step = OC_SEARCH_ADD_PARENT;
-
-       return ldb_next_request(ac->module, ac->search_req);
+       return ldb_next_request(ac->module, search_req);
 }
 
-static int objectclass_do_add(struct ldb_handle *h) 
+static int objectclass_do_add(struct oc_context *ac)
 {
+       struct ldb_context *ldb;
        const struct dsdb_schema *schema;
-       struct oc_context *ac;
+       struct ldb_request *add_req;
+       char *value;
        struct ldb_message_element *objectclass_element;
        struct ldb_message *msg;
        TALLOC_CTX *mem_ctx;
        struct class_list *sorted, *current;
        int ret;
-      
-       ac = talloc_get_type(h->private_data, struct oc_context);
-       schema = dsdb_get_schema(ac->module->ldb);
+
+       ldb = ldb_module_get_ctx(ac->module);
+       schema = dsdb_get_schema(ldb);
 
        mem_ctx = talloc_new(ac);
        if (mem_ctx == NULL) {
                return LDB_ERR_OPERATIONS_ERROR;
        }
 
-       ac->add_req = talloc(ac, struct ldb_request);
-       if (ac->add_req == NULL) {
-               talloc_free(mem_ctx);
-               return LDB_ERR_OPERATIONS_ERROR;
-       }
-
-       *ac->add_req = *ac->orig_req;
+       msg = ldb_msg_copy_shallow(ac, ac->req->op.add.message);
 
-       ac->add_req->op.add.message = msg = ldb_msg_copy_shallow(ac->add_req, ac->orig_req->op.add.message);
-
-       ldb_set_timeout_from_prev_req(ac->module->ldb, ac->orig_req, ac->add_req);
-       
        /* Check we have a valid parent */
        if (ac->search_res == NULL) {
-               if (ldb_dn_compare(ldb_get_root_basedn(ac->module->ldb), ac->orig_req->op.add.message->dn) == 0) {
+               if (ldb_dn_compare(ldb_get_root_basedn(ldb),
+                                                               msg->dn) == 0) {
                        /* Allow the tree to be started */
                        
                        /* but don't keep any error string, it's meaningless */
-                       ldb_set_errstring(ac->module->ldb, NULL);
+                       ldb_set_errstring(ldb, NULL);
                } else {
-                       ldb_asprintf_errstring(ac->module->ldb, "objectclass: Cannot add %s, parent does not exist!", 
-                                              ldb_dn_get_linearized(ac->orig_req->op.add.message->dn));
+                       ldb_asprintf_errstring(ldb, "objectclass: Cannot add %s, parent does not exist!", 
+                                              ldb_dn_get_linearized(msg->dn));
+                       talloc_free(mem_ctx);
                        return LDB_ERR_UNWILLING_TO_PERFORM;
                }
        } else {
-               
+               const struct ldb_val *parent_guid;
+
                /* Fix up the DN to be in the standard form, taking particular care to match the parent DN */
                ret = fix_dn(msg, 
-                            ac->orig_req->op.add.message->dn,
+                            ac->req->op.add.message->dn,
                             ac->search_res->message->dn,
                             &msg->dn);
 
                if (ret != LDB_SUCCESS) {
-                       ldb_asprintf_errstring(ac->module->ldb, "Could not munge DN %s into normal form", 
-                                              ldb_dn_get_linearized(ac->orig_req->op.add.message->dn));
+                       ldb_asprintf_errstring(ldb, "Could not munge DN %s into normal form", 
+                                              ldb_dn_get_linearized(ac->req->op.add.message->dn));
+                       talloc_free(mem_ctx);
                        return ret;
                }
 
+               parent_guid = ldb_msg_find_ldb_val(ac->search_res->message, "objectGUID");
+               if (parent_guid == NULL) {
+                       ldb_asprintf_errstring(ldb, "objectclass: Cannot add %s, parent does not have an objectGUID!", 
+                                              ldb_dn_get_linearized(msg->dn));
+                       talloc_free(mem_ctx);
+                       return LDB_ERR_UNWILLING_TO_PERFORM;                    
+               }
+
                /* TODO: Check this is a valid child to this parent,
                 * by reading the allowedChildClasses and
                 * allowedChildClasssesEffective attributes */
-
+               ret = ldb_msg_add_steal_value(msg, "parentGUID", discard_const(parent_guid));
+               if (ret != LDB_SUCCESS) {
+                       ldb_asprintf_errstring(ldb, "objectclass: Cannot add %s, failed to add parentGUID", 
+                                              ldb_dn_get_linearized(msg->dn));
+                       talloc_free(mem_ctx);
+                       return LDB_ERR_UNWILLING_TO_PERFORM;                                            
+               }
        }
-
        if (schema) {
-               ret = fix_attributes(ac->module->ldb, schema, msg);
+               ret = fix_attributes(ldb, schema, msg);
                if (ret != LDB_SUCCESS) {
                        talloc_free(mem_ctx);
                        return ret;
@@ -499,13 +511,13 @@ static int objectclass_do_add(struct ldb_handle *h)
 
                /* This is now the objectClass list from the database */
                objectclass_element = ldb_msg_find_element(msg, "objectClass");
-               
+
                if (!objectclass_element) {
                        /* Where did it go?  bail now... */
                        talloc_free(mem_ctx);
                        return LDB_ERR_OPERATIONS_ERROR;
                }
-               ret = objectclass_sort(ac->module, schema, msg, mem_ctx, objectclass_element, &sorted);
+               ret = objectclass_sort(ac->module, schema, mem_ctx, objectclass_element, &sorted);
                if (ret != LDB_SUCCESS) {
                        talloc_free(mem_ctx);
                        return ret;
@@ -518,15 +530,21 @@ static int objectclass_do_add(struct ldb_handle *h)
                        talloc_free(mem_ctx);
                        return ret;
                }
-               
+
                /* We must completely replace the existing objectClass entry,
                 * because we need it sorted */
-               
+
                /* Move from the linked list back into an ldb msg */
                for (current = sorted; current; current = current->next) {
-                       ret = ldb_msg_add_string(msg, "objectClass", current->objectclass->lDAPDisplayName);
+                       value = talloc_strdup(msg, current->objectclass->lDAPDisplayName);
+                       if (value == NULL) {
+                               ldb_oom(ldb);
+                               talloc_free(mem_ctx);
+                               return LDB_ERR_OPERATIONS_ERROR;
+                       }
+                       ret = ldb_msg_add_string(msg, "objectClass", value);
                        if (ret != LDB_SUCCESS) {
-                               ldb_set_errstring(ac->module->ldb, 
+                               ldb_set_errstring(ldb,
                                                  "objectclass: could not re-add sorted "
                                                  "objectclass to modify msg");
                                talloc_free(mem_ctx);
@@ -536,20 +554,32 @@ static int objectclass_do_add(struct ldb_handle *h)
                        if (!current->next) {
                                struct ldb_message_element *el;
                                int32_t systemFlags = 0;
+                               const char *rdn_name = ldb_dn_get_rdn_name(msg->dn);
+                               if (ldb_attr_cmp(rdn_name, current->objectclass->rDNAttID) != 0) {
+                                       ldb_asprintf_errstring(ldb, "RDN %s is not correct for most specific structural objectclass %s, should be %s", 
+                                                              rdn_name, current->objectclass->lDAPDisplayName, current->objectclass->rDNAttID);
+                                       return LDB_ERR_NAMING_VIOLATION;
+                               }
+
+                               if (current->objectclass->systemOnly && !ldb_request_get_control(ac->req, LDB_CONTROL_RELAX_OID)) {
+                                       ldb_asprintf_errstring(ldb, "objectClass %s is systemOnly, rejecting creation of %s",
+                                                              current->objectclass->lDAPDisplayName, ldb_dn_get_linearized(msg->dn));
+                                       return LDB_ERR_UNWILLING_TO_PERFORM;
+                               }
+
                                if (!ldb_msg_find_element(msg, "objectCategory")) {
-                                       ldb_msg_add_string(msg, "objectCategory", 
-                                                          current->objectclass->defaultObjectCategory);
+                                       value = talloc_strdup(msg, current->objectclass->defaultObjectCategory);
+                                       if (value == NULL) {
+                                               ldb_oom(ldb);
+                                               talloc_free(mem_ctx);
+                                               return LDB_ERR_OPERATIONS_ERROR;
+                                       }
+                                       ldb_msg_add_string(msg, "objectCategory", value);
                                }
                                if (!ldb_msg_find_element(msg, "showInAdvancedViewOnly") && (current->objectclass->defaultHidingValue == true)) {
-                                       ldb_msg_add_string(msg, "showInAdvancedViewOnly", 
+                                       ldb_msg_add_string(msg, "showInAdvancedViewOnly",
                                                           "TRUE");
                                }
-                               if (!ldb_msg_find_element(msg, "nTSecurityDescriptor")) {
-                                       DATA_BLOB *sd = get_sd(ac->module, mem_ctx, current->objectclass);
-                                       if (sd) {
-                                               ldb_msg_add_steal_value(msg, "nTSecurityDescriptor", sd);
-                                       }
-                               }
 
                                /* There are very special rules for systemFlags, see MS-ADTS 3.1.1.5.2.4 */
                                el = ldb_msg_find_element(msg, "systemFlags");
@@ -561,7 +591,7 @@ static int objectclass_do_add(struct ldb_handle *h)
                                        /* systemFlags &= ( SYSTEM_FLAG_CONFIG_ALLOW_RENAME | SYSTEM_FLAG_CONFIG_ALLOW_MOVE | SYSTEM_FLAG_CONFIG_LIMITED_MOVE); */
                                        ldb_msg_remove_element(msg, el);
                                }
-                               
+
                                /* This flag is only allowed on attributeSchema objects */
                                if (ldb_attr_cmp(current->objectclass->lDAPDisplayName, "attributeSchema") == 0) {
                                        systemFlags &= ~SYSTEM_FLAG_ATTR_IS_RDN;
@@ -574,7 +604,7 @@ static int objectclass_do_add(struct ldb_handle *h)
                                           || ldb_attr_cmp(current->objectclass->lDAPDisplayName, "ntDSDSA") == 0) {
                                        systemFlags |= (int32_t)(SYSTEM_FLAG_DISALLOW_MOVE_ON_DELETE);
 
-                               } else if (ldb_attr_cmp(current->objectclass->lDAPDisplayName, "siteLink") == 0 
+                               } else if (ldb_attr_cmp(current->objectclass->lDAPDisplayName, "siteLink") == 0
                                           || ldb_attr_cmp(current->objectclass->lDAPDisplayName, "siteLinkBridge") == 0
                                           || ldb_attr_cmp(current->objectclass->lDAPDisplayName, "nTDSConnection") == 0) {
                                        systemFlags |= (int32_t)(SYSTEM_FLAG_CONFIG_ALLOW_RENAME);
@@ -583,37 +613,51 @@ static int objectclass_do_add(struct ldb_handle *h)
                                /* TODO: If parent object is site or subnet, also add (SYSTEM_FLAG_CONFIG_ALLOW_RENAME) */
 
                                if (el || systemFlags != 0) {
-                                       samdb_msg_add_int(ac->module->ldb, msg, msg, "systemFlags", systemFlags);
+                                       samdb_msg_add_int(ldb, msg, msg, "systemFlags", systemFlags);
                                }
                        }
                }
        }
 
        talloc_free(mem_ctx);
-       ret = ldb_msg_sanity_check(ac->module->ldb, msg);
+       ret = ldb_msg_sanity_check(ldb, msg);
 
 
        if (ret != LDB_SUCCESS) {
                return ret;
        }
 
-       h->state = LDB_ASYNC_INIT;
-       h->status = LDB_SUCCESS;
-
-       ac->step = OC_DO_ADD;
+       ret = ldb_build_add_req(&add_req, ldb, ac,
+                               msg,
+                               ac->req->controls,
+                               ac, oc_op_callback,
+                               ac->req);
+       if (ret != LDB_SUCCESS) {
+               return ret;
+       }
 
        /* perform the add */
-       return ldb_next_request(ac->module, ac->add_req);
+       return ldb_next_request(ac->module, add_req);
 }
 
+static int oc_modify_callback(struct ldb_request *req,
+                               struct ldb_reply *ares);
+static int objectclass_do_mod(struct oc_context *ac);
+
 static int objectclass_modify(struct ldb_module *module, struct ldb_request *req)
 {
+       struct ldb_context *ldb = ldb_module_get_ctx(module);
        struct ldb_message_element *objectclass_element;
        struct ldb_message *msg;
-       const struct dsdb_schema *schema = dsdb_get_schema(module->ldb);
+       const struct dsdb_schema *schema = dsdb_get_schema(ldb);
+       struct class_list *sorted, *current;
+       struct ldb_request *down_req;
+       struct oc_context *ac;
+       TALLOC_CTX *mem_ctx;
+       char *value;
        int ret;
 
-       ldb_debug(module->ldb, LDB_DEBUG_TRACE, "objectclass_modify\n");
+       ldb_debug(ldb, LDB_DEBUG_TRACE, "objectclass_modify\n");
 
        /* do not manipulate our control entries */
        if (ldb_dn_is_special(req->op.mod.message->dn)) {
@@ -626,40 +670,38 @@ static int objectclass_modify(struct ldb_module *module, struct ldb_request *req
        }
        objectclass_element = ldb_msg_find_element(req->op.mod.message, "objectClass");
 
+       ac = oc_init_context(module, req);
+       if (ac == NULL) {
+               return LDB_ERR_OPERATIONS_ERROR;
+       }
+
        /* If no part of this touches the objectClass, then we don't
         * need to make any changes.  */
 
        /* If the only operation is the deletion of the objectClass
         * then go on with just fixing the attribute case */
        if (!objectclass_element) {
-               struct ldb_request *down_req = talloc(req, struct ldb_request);
-               if (down_req == NULL) {
-                       ldb_set_errstring(module->ldb, "Out of memory!");
+               msg = ldb_msg_copy_shallow(ac, req->op.mod.message);
+               if (msg == NULL) {
                        return LDB_ERR_OPERATIONS_ERROR;
                }
                
-               *down_req = *req; /* copy the request */
-               
-               down_req->op.mod.message = msg = ldb_msg_copy_shallow(down_req, req->op.mod.message);
-               
-               if (down_req->op.mod.message == NULL) {
-                       return LDB_ERR_OPERATIONS_ERROR;
+               ret = fix_attributes(ldb, schema, msg);
+               if (ret != LDB_SUCCESS) {
+                       return ret;
                }
-               
-               ret = fix_attributes(module->ldb, schema, msg);
+
+               ret = ldb_build_mod_req(&down_req, ldb, ac,
+                                       msg,
+                                       req->controls,
+                                       ac, oc_op_callback,
+                                       req);
                if (ret != LDB_SUCCESS) {
                        return ret;
                }
 
                /* 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;
-               }
-               return ret;
+               return ldb_next_request(module, down_req);
        }
 
        switch (objectclass_element->flags & LDB_FLAG_MOD_MASK) {
@@ -668,41 +710,28 @@ static int objectclass_modify(struct ldb_module *module, struct ldb_request *req
                        return LDB_ERR_OBJECT_CLASS_MODS_PROHIBITED;
                }
                break;
+
        case LDB_FLAG_MOD_REPLACE:
-       {
-               struct ldb_request *down_req;
-               struct class_list *sorted, *current;
-               TALLOC_CTX *mem_ctx;
-               mem_ctx = talloc_new(req);
+               mem_ctx = talloc_new(ac);
                if (mem_ctx == NULL) {
                        return LDB_ERR_OPERATIONS_ERROR;
                }
 
-               /* prepare the first operation */
-               down_req = talloc(req, struct ldb_request);
-               if (down_req == NULL) {
-                       ldb_set_errstring(module->ldb, "Out of memory!");
+               msg = ldb_msg_copy_shallow(ac, req->op.mod.message);
+               if (msg == NULL) {
                        talloc_free(mem_ctx);
                        return LDB_ERR_OPERATIONS_ERROR;
                }
-               
-               *down_req = *req; /* copy the request */
-               
-               down_req->op.mod.message = msg = ldb_msg_copy_shallow(down_req, req->op.mod.message);
-               
-               if (down_req->op.mod.message == NULL) {
-                       talloc_free(mem_ctx);
-                       return LDB_ERR_OPERATIONS_ERROR;
-               }
-               
-               ret = fix_attributes(module->ldb, schema, msg);
+
+               ret = fix_attributes(ldb, schema, msg);
                if (ret != LDB_SUCCESS) {
                        talloc_free(mem_ctx);
                        return ret;
                }
 
-               ret = objectclass_sort(module, schema, msg, mem_ctx, objectclass_element, &sorted);
+               ret = objectclass_sort(module, schema, mem_ctx, objectclass_element, &sorted);
                if (ret != LDB_SUCCESS) {
+                       talloc_free(mem_ctx);
                        return ret;
                }
 
@@ -719,9 +748,21 @@ static int objectclass_modify(struct ldb_module *module, struct ldb_request *req
 
                /* Move from the linked list back into an ldb msg */
                for (current = sorted; current; current = current->next) {
-                       ret = ldb_msg_add_string(msg, "objectClass", current->objectclass->lDAPDisplayName);
+                       /* copy the value as this string is on the schema
+                        * context and we can't rely on it not changing
+                        * before the operation is over */
+                       value = talloc_strdup(msg,
+                                       current->objectclass->lDAPDisplayName);
+                       if (value == NULL) {
+                               ldb_oom(ldb);
+                               talloc_free(mem_ctx);
+                               return LDB_ERR_OPERATIONS_ERROR;
+                       }
+                       ret = ldb_msg_add_string(msg, "objectClass", value);
                        if (ret != LDB_SUCCESS) {
-                               ldb_set_errstring(module->ldb, "objectclass: could not re-add sorted objectclass to modify msg");
+                               ldb_set_errstring(ldb,
+                                       "objectclass: could not re-add sorted "
+                                       "objectclass to modify msg");
                                talloc_free(mem_ctx);
                                return ret;
                        }
@@ -729,132 +770,127 @@ static int objectclass_modify(struct ldb_module *module, struct ldb_request *req
                
                talloc_free(mem_ctx);
 
-               ret = ldb_msg_sanity_check(module->ldb, msg);
+               ret = ldb_msg_sanity_check(ldb, msg);
                if (ret != LDB_SUCCESS) {
-                       talloc_free(mem_ctx);
                        return ret;
                }
-               
-               /* 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_mod_req(&down_req, ldb, ac,
+                                       msg,
+                                       req->controls,
+                                       ac, oc_op_callback,
+                                       req);
+               if (ret != LDB_SUCCESS) {
+                       return ret;
                }
-               return ret;
-       }
+
+               /* go on with the call chain */
+               return ldb_next_request(module, down_req);
        }
 
        /* This isn't the default branch of the switch, but a 'in any
         * other case'.  When a delete isn't for all objectClasses for
         * example
         */
-       {
-               struct ldb_handle *h;
-               struct oc_context *ac;
-               
-               h = oc_init_handle(req, module);
-               if (!h) {
-                       return LDB_ERR_OPERATIONS_ERROR;
-               }
-               ac = talloc_get_type(h->private_data, struct oc_context);
-               
-               /* return or own handle to deal with this call */
-               req->handle = h;
-               
-               /* prepare the first operation */
-               ac->down_req = talloc(ac, struct ldb_request);
-               if (ac->down_req == NULL) {
-                       ldb_oom(ac->module->ldb);
-                       return LDB_ERR_OPERATIONS_ERROR;
-               }
-               
-               *(ac->down_req) = *req; /* copy the request */
-               
-               ac->down_req->op.mod.message = msg = ldb_msg_copy_shallow(ac->down_req, req->op.mod.message);
-               
-               if (ac->down_req->op.mod.message == NULL) {
-                       ldb_oom(ac->module->ldb);
-                       return LDB_ERR_OPERATIONS_ERROR;
-               }
-               
-               ret = fix_attributes(ac->module->ldb, schema, msg);
-               if (ret != LDB_SUCCESS) {
-                       ldb_oom(ac->module->ldb);
-                       return ret;
-               }
 
-               ac->down_req->context = NULL;
-               ac->down_req->callback = NULL;
-               ldb_set_timeout_from_prev_req(module->ldb, req, ac->down_req);
-               
-               ac->step = OC_DO_REQ;
+       msg = ldb_msg_copy_shallow(ac, req->op.mod.message);
+       if (msg == NULL) {
+               ldb_oom(ldb);
+               return LDB_ERR_OPERATIONS_ERROR;
+       }
 
-               return ldb_next_request(module, ac->down_req);
+       ret = fix_attributes(ldb, schema, msg);
+       if (ret != LDB_SUCCESS) {
+               ldb_oom(ldb);
+               return ret;
+       }
+
+       ret = ldb_build_mod_req(&down_req, ldb, ac,
+                               msg,
+                               req->controls,
+                               ac, oc_modify_callback,
+                               req);
+       if (ret != LDB_SUCCESS) {
+               return ret;
        }
+
+       return ldb_next_request(module, down_req);
 }
 
-static int objectclass_search_self(struct ldb_handle *h) 
+static int oc_modify_callback(struct ldb_request *req, struct ldb_reply *ares)
 {
-       int ret;
-       struct oc_context *ac;
+       struct ldb_context *ldb;
        static const char * const attrs[] = { "objectClass", NULL };
+       struct ldb_request *search_req;
+       struct oc_context *ac;
+       int ret;
+
+       ac = talloc_get_type(req->context, struct oc_context);
+       ldb = ldb_module_get_ctx(ac->module);
 
-       ac = talloc_get_type(h->private_data, struct oc_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);
+       }
 
-       ret = ldb_build_search_req(&ac->search_req, ac->module->ldb,
-                                  ac, ac->orig_req->op.mod.message->dn, LDB_SCOPE_BASE,
+       if (ares->type != LDB_REPLY_DONE) {
+               talloc_free(ares);
+               return ldb_module_done(ac->req, NULL, NULL,
+                                       LDB_ERR_OPERATIONS_ERROR);
+       }
+
+       ret = ldb_build_search_req(&search_req, ldb, ac,
+                                  ac->req->op.mod.message->dn, LDB_SCOPE_BASE,
                                   "(objectClass=*)",
                                   attrs, NULL, 
-                                  ac, get_search_callback);
-
+                                  ac, get_search_callback,
+                                  ac->req);
        if (ret != LDB_SUCCESS) {
-               return ret;
+               return ldb_module_done(ac->req, NULL, NULL, ret);
        }
 
-       ldb_set_timeout_from_prev_req(ac->module->ldb, ac->orig_req, ac->search_req);
+       ac->step_fn = objectclass_do_mod;
 
-       ac->step = OC_SEARCH_SELF;
-
-       return ldb_next_request(ac->module, ac->search_req);
+       ret = ldb_next_request(ac->module, search_req);
+       if (ret != LDB_SUCCESS) {
+               return ldb_module_done(ac->req, NULL, NULL, ret);
+       }
+       return LDB_SUCCESS;
 }
 
-static int objectclass_do_mod(struct ldb_handle *h) {
-
+static int objectclass_do_mod(struct oc_context *ac)
+{
+       struct ldb_context *ldb;
        const struct dsdb_schema *schema;
-       struct oc_context *ac;
+       struct ldb_request *mod_req;
+       char *value;
        struct ldb_message_element *objectclass_element;
        struct ldb_message *msg;
        TALLOC_CTX *mem_ctx;
        struct class_list *sorted, *current;
        int ret;
-      
-       ac = talloc_get_type(h->private_data, struct oc_context);
-       schema = dsdb_get_schema(ac->module->ldb);
 
-       mem_ctx = talloc_new(ac);
-       if (mem_ctx == NULL) {
+       ldb = ldb_module_get_ctx(ac->module);
+
+       if (ac->search_res == NULL) {
                return LDB_ERR_OPERATIONS_ERROR;
        }
+       schema = dsdb_get_schema(ldb);
 
-       ac->mod_req = talloc(ac, struct ldb_request);
-       if (ac->mod_req == NULL) {
-               talloc_free(mem_ctx);
+       mem_ctx = talloc_new(ac);
+       if (mem_ctx == NULL) {
                return LDB_ERR_OPERATIONS_ERROR;
        }
 
-       ac->mod_req->operation = LDB_MODIFY;
-       ac->mod_req->controls = NULL;
-       ac->mod_req->context = ac;
-       ac->mod_req->callback = NULL;
-       ldb_set_timeout_from_prev_req(ac->module->ldb, ac->orig_req, ac->mod_req);
-       
        /* use a new message structure */
-       ac->mod_req->op.mod.message = msg = ldb_msg_new(ac->mod_req);
+       msg = ldb_msg_new(ac);
        if (msg == NULL) {
-               ldb_set_errstring(ac->module->ldb, "objectclass: could not create new modify msg");
+               ldb_set_errstring(ldb,
+                       "objectclass: could not create new modify msg");
                talloc_free(mem_ctx);
                return LDB_ERR_OPERATIONS_ERROR;
        }
@@ -869,9 +905,9 @@ static int objectclass_do_mod(struct ldb_handle *h) {
        }
        
        /* modify dn */
-       msg->dn = ac->orig_req->op.mod.message->dn;
+       msg->dn = ac->req->op.mod.message->dn;
 
-       ret = objectclass_sort(ac->module, schema, msg, mem_ctx, objectclass_element, &sorted);
+       ret = objectclass_sort(ac->module, schema, mem_ctx, objectclass_element, &sorted);
        if (ret != LDB_SUCCESS) {
                return ret;
        }
@@ -882,320 +918,212 @@ static int objectclass_do_mod(struct ldb_handle *h) {
 
        ret = ldb_msg_add_empty(msg, "objectClass", LDB_FLAG_MOD_REPLACE, NULL);
        if (ret != LDB_SUCCESS) {
-               ldb_set_errstring(ac->module->ldb, "objectclass: could not clear objectclass in modify msg");
+               ldb_set_errstring(ldb, "objectclass: could not clear objectclass in modify msg");
                talloc_free(mem_ctx);
                return ret;
        }
        
        /* Move from the linked list back into an ldb msg */
        for (current = sorted; current; current = current->next) {
-               ret = ldb_msg_add_string(msg, "objectClass", current->objectclass->lDAPDisplayName);
+               value = talloc_strdup(msg, current->objectclass->lDAPDisplayName);
+               if (value == NULL) {
+                       ldb_oom(ldb);
+                       return LDB_ERR_OPERATIONS_ERROR;
+               }
+               ret = ldb_msg_add_string(msg, "objectClass", value);
                if (ret != LDB_SUCCESS) {
-                       ldb_set_errstring(ac->module->ldb, "objectclass: could not re-add sorted objectclass to modify msg");
+                       ldb_set_errstring(ldb, "objectclass: could not re-add sorted objectclass to modify msg");
                        talloc_free(mem_ctx);
                        return ret;
                }
        }
 
-       ret = ldb_msg_sanity_check(ac->module->ldb, msg);
+       ret = ldb_msg_sanity_check(ldb, msg);
        if (ret != LDB_SUCCESS) {
                talloc_free(mem_ctx);
                return ret;
        }
 
-
-       h->state = LDB_ASYNC_INIT;
-       h->status = LDB_SUCCESS;
-
-       ac->step = OC_DO_MOD;
+       ret = ldb_build_mod_req(&mod_req, ldb, ac,
+                               msg,
+                               ac->req->controls,
+                               ac, oc_op_callback,
+                               ac->req);
+       if (ret != LDB_SUCCESS) {
+               talloc_free(mem_ctx);
+               return ret;
+       }
 
        talloc_free(mem_ctx);
-       /* perform the search */
-       return ldb_next_request(ac->module, ac->mod_req);
+       /* perform the modify */
+       return ldb_next_request(ac->module, mod_req);
 }
 
+static int objectclass_do_rename(struct oc_context *ac);
+
 static int objectclass_rename(struct ldb_module *module, struct ldb_request *req)
 {
-
-       static const char * const attrs[] = { NULL };
-
-       struct ldb_handle *h;
+       static const char * const attrs[] = { "objectGUID", NULL };
+       struct ldb_context *ldb;
+       struct ldb_request *search_req;
        struct oc_context *ac;
        struct ldb_dn *parent_dn;
        int ret;
-       
-       ldb_debug(module->ldb, LDB_DEBUG_TRACE, "objectclass_rename\n");
+
+       ldb = ldb_module_get_ctx(module);
+
+       ldb_debug(ldb, LDB_DEBUG_TRACE, "objectclass_rename\n");
 
        if (ldb_dn_is_special(req->op.rename.newdn)) { /* do not manipulate our control entries */
                return ldb_next_request(module, req);
        }
-       
+
        /* Firstly ensure we are not trying to rename it to be a child of itself */
        if ((ldb_dn_compare_base(req->op.rename.olddn, req->op.rename.newdn) == 0) 
            && (ldb_dn_compare(req->op.rename.olddn, req->op.rename.newdn) != 0)) {
-               ldb_asprintf_errstring(module->ldb, "Cannot rename %s to be a child of itself",
+               ldb_asprintf_errstring(ldb, "Cannot rename %s to be a child of itself",
                                       ldb_dn_get_linearized(req->op.rename.olddn));
                return LDB_ERR_UNWILLING_TO_PERFORM;
        }
 
-       h = oc_init_handle(req, module);
-       if (!h) {
+       ac = oc_init_context(module, req);
+       if (ac == NULL) {
                return LDB_ERR_OPERATIONS_ERROR;
        }
-       ac = talloc_get_type(h->private_data, struct oc_context);
-       
-       /* return or own handle to deal with this call */
-       req->handle = h;
 
-       parent_dn = ldb_dn_get_parent(ac, ac->orig_req->op.rename.newdn);
+       parent_dn = ldb_dn_get_parent(ac, req->op.rename.newdn);
        if (parent_dn == NULL) {
-               ldb_oom(module->ldb);
+               ldb_oom(ldb);
                return LDB_ERR_OPERATIONS_ERROR;
        }
-       ret = ldb_build_search_req(&ac->search_req, module->ldb,
+
+       /* note that the results of this search are kept and used to
+          update the parentGUID in objectclass_rename_callback() */
+       ret = ldb_build_search_req(&search_req, ldb,
                                   ac, parent_dn, LDB_SCOPE_BASE,
                                   "(objectClass=*)",
                                   attrs, NULL, 
-                                  ac, get_search_callback);
+                                  ac, get_search_callback,
+                                  req);
        if (ret != LDB_SUCCESS) {
                return ret;
        }
-       talloc_steal(ac->search_req, parent_dn);
-       ldb_set_timeout_from_prev_req(ac->module->ldb, ac->orig_req, ac->search_req);
-
-       ac->step = OC_SEARCH_RENAME_PARENT;
-
-       return ldb_next_request(ac->module, ac->search_req);
-}
-
-static int objectclass_do_rename(struct ldb_handle *h) 
-{
-       struct oc_context *ac;
-       int ret;
-      
-       ac = talloc_get_type(h->private_data, struct oc_context);
-
-       ac->rename_req = talloc(ac, struct ldb_request);
-       if (ac->rename_req == NULL) {
-               return LDB_ERR_OPERATIONS_ERROR;
-       }
 
-       *ac->rename_req = *ac->orig_req;
-
-       ldb_set_timeout_from_prev_req(ac->module->ldb, ac->orig_req, ac->rename_req);
-       
-       /* Check we have a valid parent */
-       if (ac->search_res == NULL) {
-               ldb_asprintf_errstring(ac->module->ldb, "objectclass: Cannot rename %s, parent does not exist!", 
-                                      ldb_dn_get_linearized(ac->orig_req->op.rename.newdn));
-               return LDB_ERR_UNWILLING_TO_PERFORM;
-       }
-       
-       /* Fix up the DN to be in the standard form, taking particular care to match the parent DN */
-       ret = fix_dn(ac->rename_req, 
-                    ac->orig_req->op.rename.newdn, 
-                    ac->search_res->message->dn, 
-                    &ac->rename_req->op.rename.newdn);
+       /* we have to add the show deleted control, as otherwise DRS
+          deletes will be refused as we will think the target parent
+          does not exist */
+       ret = ldb_request_add_control(search_req, LDB_CONTROL_SHOW_DELETED_OID, false, NULL);
 
        if (ret != LDB_SUCCESS) {
                return ret;
        }
 
-       /* TODO: Check this is a valid child to this parent,
-        * by reading the allowedChildClasses and
-        * allowedChildClasssesEffective attributes */
+       ac->step_fn = objectclass_do_rename;
 
-       h->state = LDB_ASYNC_INIT;
-       h->status = LDB_SUCCESS;
-
-       ac->step = OC_DO_RENAME;
-
-       /* perform the rename */
-       return ldb_next_request(ac->module, ac->rename_req);
+       return ldb_next_request(ac->module, search_req);
 }
 
-static int oc_wait(struct ldb_handle *handle) {
+/* 
+   called after the rename happens. 
+   We now need to fix the parentGUID of the object to be the objectGUID of
+   the new parent 
+*/
+static int objectclass_rename_callback(struct ldb_request *req, struct ldb_reply *ares)
+{
+       struct ldb_context *ldb;
        struct oc_context *ac;
+       const struct ldb_val *parent_guid;
+       struct ldb_request *mod_req = NULL;
        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 oc_context);
-
-       switch (ac->step) {
-       case OC_DO_REQ:
-               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;
-               }
-
-               /* mods done, go on */
-               return objectclass_search_self(handle);
-
-       case OC_SEARCH_SELF:
-               ret = ldb_wait(ac->search_req->handle, LDB_WAIT_NONE);
-
-               if (ret != LDB_SUCCESS) {
-                       handle->status = ret;
-                       goto done;
-               }
-               if (ac->search_req->handle->status != LDB_SUCCESS) {
-                       handle->status = ac->search_req->handle->status;
-                       goto done;
-               }
-
-               if (ac->search_req->handle->state != LDB_ASYNC_DONE) {
-                       return LDB_SUCCESS;
-               }
-
-               /* self search done, go on */
-               return objectclass_do_mod(handle);
-
-       case OC_DO_MOD:
-               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;
-               
-       case OC_SEARCH_ADD_PARENT:
-               ret = ldb_wait(ac->search_req->handle, LDB_WAIT_NONE);
-
-               if (ret != LDB_SUCCESS && ret != LDB_ERR_NO_SUCH_OBJECT) {
-                       handle->status = ret;
-                       goto done;
-               }
-               if (ac->search_req->handle->status != LDB_SUCCESS
-                   && ac->search_req->handle->status != LDB_ERR_NO_SUCH_OBJECT) {
-                       handle->status = ac->search_req->handle->status;
-                       goto done;
-               }
-
-               if (ac->search_req->handle->state != LDB_ASYNC_DONE) {
-                       return LDB_SUCCESS;
-               }
-
-               /* parent search done, go on */
-               return objectclass_do_add(handle);
-
-       case OC_DO_ADD:
-               ret = ldb_wait(ac->add_req->handle, LDB_WAIT_NONE);
-
-               if (ret != LDB_SUCCESS) {
-                       handle->status = ret;
-                       goto done;
-               }
-               if (ac->add_req->handle->status != LDB_SUCCESS) {
-                       handle->status = ac->add_req->handle->status;
-                       goto done;
-               }
-
-               if (ac->add_req->handle->state != LDB_ASYNC_DONE) {
-                       return LDB_SUCCESS;
-               }
+       struct ldb_message *msg;
+       struct ldb_message_element *el = NULL;
 
-               break;
-               
-       case OC_SEARCH_RENAME_PARENT:
-               ret = ldb_wait(ac->search_req->handle, LDB_WAIT_NONE);
+       ac = talloc_get_type(req->context, struct oc_context);
+       ldb = ldb_module_get_ctx(ac->module);
 
-               if (ret != LDB_SUCCESS && ret != LDB_ERR_NO_SUCH_OBJECT) {
-                       handle->status = ret;
-                       goto done;
-               }
-               if (ac->search_req->handle->status != LDB_SUCCESS && ac->search_req->handle->status != LDB_ERR_NO_SUCH_OBJECT) {
-                       handle->status = ac->search_req->handle->status;
-                       goto done;
-               }
+       /* make sure the rename succeeded */
+       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 (ac->search_req->handle->state != LDB_ASYNC_DONE) {
-                       return LDB_SUCCESS;
-               }
 
-               /* parent search done, go on */
-               return objectclass_do_rename(handle);
+       /* the ac->search_res should contain the new parents objectGUID */
+       parent_guid = ldb_msg_find_ldb_val(ac->search_res->message, "objectGUID");
+       if (parent_guid == NULL) {
+               ldb_asprintf_errstring(ldb, "objectclass: Cannot rename %s, new parent does not have an objectGUID!", 
+                                      ldb_dn_get_linearized(ac->req->op.rename.newdn));
+               return LDB_ERR_UNWILLING_TO_PERFORM;
 
-       case OC_DO_RENAME:
-               ret = ldb_wait(ac->rename_req->handle, LDB_WAIT_NONE);
+       }
 
-               if (ret != LDB_SUCCESS) {
-                       handle->status = ret;
-                       goto done;
-               }
-               if (ac->rename_req->handle->status != LDB_SUCCESS) {
-                       handle->status = ac->rename_req->handle->status;
-                       goto done;
-               }
+       /* construct the modify message */
+       msg = ldb_msg_new(ac);
+       if (msg == NULL) {
+               ldb_oom(ldb);
+               return LDB_ERR_OPERATIONS_ERROR;
+       }
 
-               if (ac->rename_req->handle->state != LDB_ASYNC_DONE) {
-                       return LDB_SUCCESS;
-               }
+       msg->dn = ac->req->op.rename.newdn;
 
-               break;
-               
-       default:
-               ret = LDB_ERR_OPERATIONS_ERROR;
-               goto done;
+       ret = ldb_msg_add_value(msg, "parentGUID", parent_guid, &el);
+       if (ret != LDB_SUCCESS) {
+               return ret;
        }
 
-       ret = LDB_SUCCESS;
+       el->flags = LDB_FLAG_MOD_REPLACE;
 
-done:
-       handle->state = LDB_ASYNC_DONE;
-       return ret;
-}
+       ret = ldb_build_mod_req(&mod_req, ldb, ac, msg,
+                               NULL, ac, oc_op_callback, req);
 
-static int oc_wait_all(struct ldb_handle *handle) {
+       return ldb_next_request(ac->module, mod_req);
+}
 
+static int objectclass_do_rename(struct oc_context *ac)
+{
+       struct ldb_context *ldb;
+       struct ldb_request *rename_req;
+       struct ldb_dn *fixed_dn;
        int ret;
 
-       while (handle->state != LDB_ASYNC_DONE) {
-               ret = oc_wait(handle);
-               if (ret != LDB_SUCCESS) {
-                       return ret;
-               }
+       ldb = ldb_module_get_ctx(ac->module);
+
+       /* Check we have a valid parent */
+       if (ac->search_res == NULL) {
+               ldb_asprintf_errstring(ldb, "objectclass: Cannot rename %s, parent does not exist!", 
+                                      ldb_dn_get_linearized(ac->req->op.rename.newdn));
+               return LDB_ERR_UNWILLING_TO_PERFORM;
+       }
+       
+       /* Fix up the DN to be in the standard form,
+        * taking particular care to match the parent DN */
+       ret = fix_dn(ac,
+                    ac->req->op.rename.newdn,
+                    ac->search_res->message->dn,
+                    &fixed_dn);
+       if (ret != LDB_SUCCESS) {
+               return ret;
        }
 
-       return handle->status;
-}
+       /* TODO: Check this is a valid child to this parent,
+        * by reading the allowedChildClasses and
+        * allowedChildClasssesEffective attributes */
 
-static int objectclass_wait(struct ldb_handle *handle, enum ldb_wait_type type)
-{
-       if (type == LDB_WAIT_ALL) {
-               return oc_wait_all(handle);
-       } else {
-               return oc_wait(handle);
+       ret = ldb_build_rename_req(&rename_req, ldb, ac,
+                                  ac->req->op.rename.olddn, fixed_dn,
+                                  ac->req->controls,
+                                  ac, objectclass_rename_callback,
+                                  ac->req);
+       if (ret != LDB_SUCCESS) {
+               return ret;
        }
+
+       /* perform the rename */
+       return ldb_next_request(ac->module, rename_req);
 }
 
 _PUBLIC_ const struct ldb_module_ops ldb_objectclass_module_ops = {
@@ -1203,5 +1131,4 @@ _PUBLIC_ const struct ldb_module_ops ldb_objectclass_module_ops = {
        .add           = objectclass_add,
        .modify        = objectclass_modify,
        .rename        = objectclass_rename,
-       .wait          = objectclass_wait
 };