Fix for schemaUpdateNow command
[samba.git] / source4 / dsdb / schema / schema_set.c
index 0ca26c0fc71404be4f1825ca0b5da9c6a33d69f2..5ded04e9da1ef53243c7fe63addcb12601d723f9 100644 (file)
 */
 
 #include "includes.h"
-#include "dsdb/samdb/samdb.h"
-#include "lib/ldb/include/ldb_errors.h"
-#include "lib/ldb/include/ldb_private.h"
 #include "lib/util/dlinklist.h"
+#include "dsdb/samdb/samdb.h"
+#include "lib/ldb/include/ldb_module.h"
 #include "param/param.h"
 
+/*
+  override the name to attribute handler function
+ */
+const struct ldb_schema_attribute *dsdb_attribute_handler_override(struct ldb_context *ldb, 
+                                                                  void *private_data,
+                                                                  const char *name)
+{
+       struct dsdb_schema *schema = talloc_get_type_abort(private_data, struct dsdb_schema);
+       const struct dsdb_attribute *a = dsdb_attribute_by_lDAPDisplayName(schema, name);
+       if (a == NULL) {
+               /* this will fall back to ldb internal handling */
+               return NULL;
+       }
+       return a->ldb_schema_attribute;
+}
 
 static int dsdb_schema_set_attributes(struct ldb_context *ldb, struct dsdb_schema *schema, bool write_attributes)
 {
@@ -35,11 +49,18 @@ static int dsdb_schema_set_attributes(struct ldb_context *ldb, struct dsdb_schem
        struct ldb_result *res_idx;
        struct dsdb_attribute *attr;
        struct ldb_message *mod_msg;
-       TALLOC_CTX *mem_ctx = talloc_new(ldb);
-       
+       TALLOC_CTX *mem_ctx;
        struct ldb_message *msg;
        struct ldb_message *msg_idx;
 
+       /* setup our own attribute name to schema handler */
+       ldb_schema_attribute_set_override_handler(ldb, dsdb_attribute_handler_override, schema);
+
+       if (!write_attributes) {
+               return ret;
+       }
+
+       mem_ctx = talloc_new(ldb);
        if (!mem_ctx) {
                return LDB_ERR_OPERATIONS_ERROR;
        }
@@ -47,27 +68,32 @@ static int dsdb_schema_set_attributes(struct ldb_context *ldb, struct dsdb_schem
        msg = ldb_msg_new(mem_ctx);
        if (!msg) {
                ldb_oom(ldb);
-               return LDB_ERR_OPERATIONS_ERROR;
+               goto op_error;
        }
        msg_idx = ldb_msg_new(mem_ctx);
        if (!msg_idx) {
                ldb_oom(ldb);
-               return LDB_ERR_OPERATIONS_ERROR;
+               goto op_error;
        }
        msg->dn = ldb_dn_new(msg, ldb, "@ATTRIBUTES");
        if (!msg->dn) {
                ldb_oom(ldb);
-               return LDB_ERR_OPERATIONS_ERROR;
+               goto op_error;
        }
        msg_idx->dn = ldb_dn_new(msg, ldb, "@INDEXLIST");
        if (!msg_idx->dn) {
                ldb_oom(ldb);
-               return LDB_ERR_OPERATIONS_ERROR;
+               goto op_error;
+       }
+
+       ret = ldb_msg_add_string(msg_idx, "@IDXONE", "1");
+       if (ret != LDB_SUCCESS) {
+               goto op_error;
        }
 
        for (attr = schema->attributes; attr; attr = attr->next) {
-               const struct ldb_schema_syntax *s;
                const char *syntax = attr->syntax->ldb_syntax;
+               
                if (!syntax) {
                        syntax = attr->syntax->ldap_oid;
                }
@@ -79,120 +105,232 @@ static int dsdb_schema_set_attributes(struct ldb_context *ldb, struct dsdb_schem
                        ret = ldb_msg_add_string(msg, attr->lDAPDisplayName, "CASE_INSENSITIVE");
                } 
                if (ret != LDB_SUCCESS) {
-                       return ret;
+                       break;
                }
 
                if (attr->searchFlags & SEARCH_FLAG_ATTINDEX) {
                        ret = ldb_msg_add_string(msg_idx, "@IDXATTR", attr->lDAPDisplayName);
                        if (ret != LDB_SUCCESS) {
-                               return ret;
-                       }
-               }
-
-               if (!attr->syntax) {
-                       continue;
-               }
-
-               ret = ldb_schema_attribute_add(ldb, attr->lDAPDisplayName, LDB_ATTR_FLAG_FIXED,
-                                              syntax);
-               if (ret != LDB_SUCCESS) {
-                       s = ldb_samba_syntax_by_name(ldb, attr->syntax->ldap_oid);
-                       if (s) {
-                               ret = ldb_schema_attribute_add_with_syntax(ldb, attr->lDAPDisplayName, LDB_ATTR_FLAG_FIXED, s);
-                       } else {
-                               ret = LDB_SUCCESS; /* Nothing to do here */
+                               break;
                        }
                }
-               
-               if (ret != LDB_SUCCESS) {
-                       return ret;
-               }
-       }
-
-       if (!write_attributes) {
-               talloc_free(mem_ctx);
-               return ret;
        }
 
-
-       ret = ldb_transaction_start(ldb);
        if (ret != LDB_SUCCESS) {
+               talloc_free(mem_ctx);
                return ret;
        }
 
        /* Try to avoid churning the attributes too much - we only want to do this if they have changed */
-       ret = ldb_search_exp_fmt(ldb, mem_ctx, &res, msg->dn, LDB_SCOPE_BASE, NULL, "dn=%s", ldb_dn_get_linearized(msg->dn));
+       ret = ldb_search(ldb, mem_ctx, &res, msg->dn, LDB_SCOPE_BASE, NULL, "dn=%s", ldb_dn_get_linearized(msg->dn));
        if (ret == LDB_ERR_NO_SUCH_OBJECT) {
                ret = ldb_add(ldb, msg);
        } else if (ret != LDB_SUCCESS) {
-               talloc_free(mem_ctx);
-               ldb_transaction_cancel(ldb);
-               return ret;
+       } else if (res->count != 1) {
+               ret = ldb_add(ldb, msg);
        } else {
-
-               if (res->count != 1) {
-                       talloc_free(mem_ctx);
-                       ldb_transaction_cancel(ldb);
-                       return LDB_ERR_NO_SUCH_OBJECT;
-               }
-               
                ret = LDB_SUCCESS;
                /* Annoyingly added to our search results */
                ldb_msg_remove_attr(res->msgs[0], "distinguishedName");
-
+               
                mod_msg = ldb_msg_diff(ldb, res->msgs[0], msg);
                if (mod_msg->num_elements > 0) {
-                       ret = ldb_modify(ldb, mod_msg);
+                       ret = samdb_replace(ldb, mem_ctx, mod_msg);
                }
        }
 
        if (ret == LDB_ERR_INSUFFICIENT_ACCESS_RIGHTS) {
                /* We might be on a read-only DB */
+               ret = LDB_SUCCESS;
+       }
+       if (ret != LDB_SUCCESS) {
                talloc_free(mem_ctx);
-               ret = ldb_transaction_cancel(ldb);
-               return ret;
-       } else if (ret != LDB_SUCCESS) {
-               ldb_transaction_cancel(ldb);
                return ret;
        }
 
        /* Now write out the indexs, as found in the schema (if they have changed) */
 
-       ret = ldb_search_exp_fmt(ldb, mem_ctx, &res_idx, msg_idx->dn, LDB_SCOPE_BASE, NULL, "dn=%s", ldb_dn_get_linearized(msg_idx->dn));
+       ret = ldb_search(ldb, mem_ctx, &res_idx, msg_idx->dn, LDB_SCOPE_BASE, NULL, "dn=%s", ldb_dn_get_linearized(msg_idx->dn));
        if (ret == LDB_ERR_NO_SUCH_OBJECT) {
                ret = ldb_add(ldb, msg_idx);
        } else if (ret != LDB_SUCCESS) {
-               talloc_free(mem_ctx);
-               ldb_transaction_cancel(ldb);
-               return ret;
+       } else if (res_idx->count != 1) {
+               ret = ldb_add(ldb, msg_idx);
        } else {
-               if (res_idx->count != 1) {
-                       talloc_free(mem_ctx);
-                       ldb_transaction_cancel(ldb);
-                       return LDB_ERR_NO_SUCH_OBJECT;
-               }
-               
+               ret = LDB_SUCCESS;
                /* Annoyingly added to our search results */
                ldb_msg_remove_attr(res_idx->msgs[0], "distinguishedName");
 
                mod_msg = ldb_msg_diff(ldb, res_idx->msgs[0], msg_idx);
                if (mod_msg->num_elements > 0) {
-                       ret = ldb_modify(ldb, mod_msg);
+                       ret = samdb_replace(ldb, mem_ctx, mod_msg);
                }
        }
        if (ret == LDB_ERR_INSUFFICIENT_ACCESS_RIGHTS) {
                /* We might be on a read-only DB */
-               talloc_free(mem_ctx);
-               return ldb_transaction_cancel(ldb);
-       } else if (ret == LDB_SUCCESS) {
-               ret = ldb_transaction_commit(ldb);
-       } else {
-               ldb_transaction_cancel(ldb);
+               ret = LDB_SUCCESS;
        }
        talloc_free(mem_ctx);
        return ret;
+
+op_error:
+       talloc_free(mem_ctx);
+       return LDB_ERR_OPERATIONS_ERROR;
+}
+
+static int dsdb_compare_class_by_lDAPDisplayName(struct dsdb_class **c1, struct dsdb_class **c2)
+{
+       return strcasecmp((*c1)->lDAPDisplayName, (*c2)->lDAPDisplayName);
+}
+static int dsdb_compare_class_by_governsID_id(struct dsdb_class **c1, struct dsdb_class **c2)
+{
+       return (*c1)->governsID_id - (*c2)->governsID_id;
+}
+static int dsdb_compare_class_by_governsID_oid(struct dsdb_class **c1, struct dsdb_class **c2)
+{
+       return strcasecmp((*c1)->governsID_oid, (*c2)->governsID_oid);
+}
+static int dsdb_compare_class_by_cn(struct dsdb_class **c1, struct dsdb_class **c2)
+{
+       return strcasecmp((*c1)->cn, (*c2)->cn);
+}
+
+static int dsdb_compare_attribute_by_lDAPDisplayName(struct dsdb_attribute **a1, struct dsdb_attribute **a2)
+{
+       return strcasecmp((*a1)->lDAPDisplayName, (*a2)->lDAPDisplayName);
+}
+static int dsdb_compare_attribute_by_attributeID_id(struct dsdb_attribute **a1, struct dsdb_attribute **a2)
+{
+       return (*a1)->attributeID_id - (*a2)->attributeID_id;
+}
+static int dsdb_compare_attribute_by_attributeID_oid(struct dsdb_attribute **a1, struct dsdb_attribute **a2)
+{
+       return strcasecmp((*a1)->attributeID_oid, (*a2)->attributeID_oid);
+}
+static int dsdb_compare_attribute_by_linkID(struct dsdb_attribute **a1, struct dsdb_attribute **a2)
+{
+       return (*a1)->linkID - (*a2)->linkID;
+}
+
+/*
+  create the sorted accessor arrays for the schema
+ */
+static int dsdb_setup_sorted_accessors(struct ldb_context *ldb,
+                                      struct dsdb_schema *schema)
+{
+       struct dsdb_class *cur;
+       struct dsdb_attribute *a;
+       uint32_t i;
+
+       talloc_free(schema->classes_by_lDAPDisplayName);
+       talloc_free(schema->classes_by_governsID_id);
+       talloc_free(schema->classes_by_governsID_oid);
+       talloc_free(schema->classes_by_cn);
+
+       /* count the classes */
+       for (i=0, cur=schema->classes; cur; i++, cur=cur->next) /* noop */ ;
+       schema->num_classes = i;
+
+       /* setup classes_by_* */
+       schema->classes_by_lDAPDisplayName = talloc_array(schema, struct dsdb_class *, i);
+       schema->classes_by_governsID_id    = talloc_array(schema, struct dsdb_class *, i);
+       schema->classes_by_governsID_oid   = talloc_array(schema, struct dsdb_class *, i);
+       schema->classes_by_cn              = talloc_array(schema, struct dsdb_class *, i);
+       if (schema->classes_by_lDAPDisplayName == NULL ||
+           schema->classes_by_governsID_id == NULL ||
+           schema->classes_by_governsID_oid == NULL ||
+           schema->classes_by_cn == NULL) {
+               goto failed;
+       }
+
+       for (i=0, cur=schema->classes; cur; i++, cur=cur->next) {
+               schema->classes_by_lDAPDisplayName[i] = cur;
+               schema->classes_by_governsID_id[i]    = cur;
+               schema->classes_by_governsID_oid[i]   = cur;
+               schema->classes_by_cn[i]              = cur;
+       }
+
+       /* sort the arrays */
+       qsort(schema->classes_by_lDAPDisplayName, schema->num_classes, 
+             sizeof(struct dsdb_class *), QSORT_CAST dsdb_compare_class_by_lDAPDisplayName);
+       qsort(schema->classes_by_governsID_id, schema->num_classes, 
+             sizeof(struct dsdb_class *), QSORT_CAST dsdb_compare_class_by_governsID_id);
+       qsort(schema->classes_by_governsID_oid, schema->num_classes, 
+             sizeof(struct dsdb_class *), QSORT_CAST dsdb_compare_class_by_governsID_oid);
+       qsort(schema->classes_by_cn, schema->num_classes, 
+             sizeof(struct dsdb_class *), QSORT_CAST dsdb_compare_class_by_cn);
+
+       /* now build the attribute accessor arrays */
+       talloc_free(schema->attributes_by_lDAPDisplayName);
+       talloc_free(schema->attributes_by_attributeID_id);
+       talloc_free(schema->attributes_by_attributeID_oid);
+       talloc_free(schema->attributes_by_linkID);
+
+       /* count the attributes */
+       for (i=0, a=schema->attributes; a; i++, a=a->next) /* noop */ ;
+       schema->num_attributes = i;
+
+       /* setup attributes_by_* */
+       schema->attributes_by_lDAPDisplayName = talloc_array(schema, struct dsdb_attribute *, i);
+       schema->attributes_by_attributeID_id    = talloc_array(schema, struct dsdb_attribute *, i);
+       schema->attributes_by_attributeID_oid   = talloc_array(schema, struct dsdb_attribute *, i);
+       schema->attributes_by_linkID              = talloc_array(schema, struct dsdb_attribute *, i);
+       if (schema->attributes_by_lDAPDisplayName == NULL ||
+           schema->attributes_by_attributeID_id == NULL ||
+           schema->attributes_by_attributeID_oid == NULL ||
+           schema->attributes_by_linkID == NULL) {
+               goto failed;
+       }
+
+       for (i=0, a=schema->attributes; a; i++, a=a->next) {
+               schema->attributes_by_lDAPDisplayName[i] = a;
+               schema->attributes_by_attributeID_id[i]    = a;
+               schema->attributes_by_attributeID_oid[i]   = a;
+               schema->attributes_by_linkID[i]          = a;
+       }
+
+       /* sort the arrays */
+       qsort(schema->attributes_by_lDAPDisplayName, schema->num_attributes, 
+             sizeof(struct dsdb_attribute *), QSORT_CAST dsdb_compare_attribute_by_lDAPDisplayName);
+       qsort(schema->attributes_by_attributeID_id, schema->num_attributes, 
+             sizeof(struct dsdb_attribute *), QSORT_CAST dsdb_compare_attribute_by_attributeID_id);
+       qsort(schema->attributes_by_attributeID_oid, schema->num_attributes, 
+             sizeof(struct dsdb_attribute *), QSORT_CAST dsdb_compare_attribute_by_attributeID_oid);
+       qsort(schema->attributes_by_linkID, schema->num_attributes, 
+             sizeof(struct dsdb_attribute *), QSORT_CAST dsdb_compare_attribute_by_linkID);
+
+       return LDB_SUCCESS;
+
+failed:
+       schema->classes_by_lDAPDisplayName = NULL;
+       schema->classes_by_governsID_id = NULL;
+       schema->classes_by_governsID_oid = NULL;
+       schema->classes_by_cn = NULL;
+       schema->attributes_by_lDAPDisplayName = NULL;
+       schema->attributes_by_attributeID_id = NULL;
+       schema->attributes_by_attributeID_oid = NULL;
+       schema->attributes_by_linkID = NULL;
+       ldb_oom(ldb);
+       return LDB_ERR_OPERATIONS_ERROR;
 }
 
+int dsdb_setup_schema_inversion(struct ldb_context *ldb, struct dsdb_schema *schema)
+{
+       /* Walk the list of schema classes */
+
+       /*  For each subClassOf, add us to subclasses of the parent */
+
+       /* collect these subclasses into a recursive list of total subclasses, preserving order */
+
+       /* For each subclass under 'top', write the index from it's
+        * order as an integer in the dsdb_class (for sorting
+        * objectClass lists efficiently) */
+
+       /* Walk the list of scheam classes */
+       
+       /*  Create a 'total possible superiors' on each class */
+       return LDB_SUCCESS;
+}
 
 /**
  * Attach the schema to an opaque pointer on the ldb, so ldb modules
@@ -203,6 +341,13 @@ int dsdb_set_schema(struct ldb_context *ldb, struct dsdb_schema *schema)
 {
        int ret;
 
+       ret = dsdb_setup_sorted_accessors(ldb, schema);
+       if (ret != LDB_SUCCESS) {
+               return ret;
+       }
+
+       schema_fill_constructed(schema);
+
        ret = ldb_set_opaque(ldb, "dsdb_schema", schema);
        if (ret != LDB_SUCCESS) {
                return ret;
@@ -233,6 +378,7 @@ int dsdb_set_global_schema(struct ldb_context *ldb)
        if (!global_schema) {
                return LDB_SUCCESS;
        }
+
        ret = ldb_set_opaque(ldb, "dsdb_schema", global_schema);
        if (ret != LDB_SUCCESS) {
                return ret;
@@ -287,10 +433,11 @@ void dsdb_make_schema_global(struct ldb_context *ldb)
        }
 
        if (global_schema) {
-               talloc_unlink(talloc_autofree_context(), schema);
+               talloc_unlink(talloc_autofree_context(), global_schema);
        }
 
-       talloc_steal(talloc_autofree_context(), schema);
+       /* we want the schema to be around permanently */
+       talloc_reference(talloc_autofree_context(), schema);
        global_schema = schema;
 
        dsdb_set_global_schema(ldb);
@@ -303,7 +450,7 @@ void dsdb_make_schema_global(struct ldb_context *ldb)
  * schema itself to the directory.
  */
 
-WERROR dsdb_attach_schema_from_ldif_file(struct ldb_context *ldb, const char *pf, const char *df)
+WERROR dsdb_attach_schema_from_ldif(struct ldb_context *ldb, const char *pf, const char *df)
 {
        struct ldb_ldif *ldif;
        struct ldb_message *msg;
@@ -353,11 +500,10 @@ WERROR dsdb_attach_schema_from_ldif_file(struct ldb_context *ldb, const char *pf
 
        info_val = ldb_msg_find_ldb_val(msg, "schemaInfo");
        if (!info_val) {
-               info_val_default = strhex_to_data_blob("FF0000000000000000000000000000000000000000");
+               info_val_default = strhex_to_data_blob(mem_ctx, "FF0000000000000000000000000000000000000000");
                if (!info_val_default.data) {
                        goto nomem;
                }
-               talloc_steal(mem_ctx, info_val_default.data);
                info_val = &info_val_default;
        }
 
@@ -394,7 +540,7 @@ WERROR dsdb_attach_schema_from_ldif_file(struct ldb_context *ldb, const char *pf
                                goto nomem;
                        }
 
-                       status = dsdb_attribute_from_ldb(schema, msg, sa, sa);
+                       status = dsdb_attribute_from_ldb(ldb, schema, msg, sa, sa);
                        if (!W_ERROR_IS_OK(status)) {
                                goto failed;
                        }