Fix the mess with ldb includes.
[metze/samba/wip.git] / source4 / lib / ldb-samba / ldif_handlers.c
index 22a57da10bfd6b029c0fc98b65b57b14268ac8e9..d92f21194651387be5f3869fdd8df2e8edd3b16f 100644 (file)
@@ -22,7 +22,7 @@
 */
 
 #include "includes.h"
-#include "lib/ldb/include/ldb_includes.h"
+#include "ldb_private.h"
 #include "dsdb/samdb/samdb.h"
 #include "librpc/gen_ndr/ndr_security.h"
 #include "librpc/gen_ndr/ndr_misc.h"
@@ -38,7 +38,7 @@ static int ldif_read_objectSid(struct ldb_context *ldb, void *mem_ctx,
 {
        enum ndr_err_code ndr_err;
        struct dom_sid *sid;
-       sid = dom_sid_parse_talloc(mem_ctx, (const char *)in->data);
+       sid = dom_sid_parse_length(mem_ctx, in);
        if (sid == NULL) {
                return -1;
        }
@@ -64,18 +64,17 @@ static int ldif_write_objectSid(struct ldb_context *ldb, void *mem_ctx,
        if (sid == NULL) {
                return -1;
        }
-       ndr_err = ndr_pull_struct_blob(in, sid, NULL, sid,
-                                      (ndr_pull_flags_fn_t)ndr_pull_dom_sid);
+       ndr_err = ndr_pull_struct_blob_all(in, sid, NULL, sid,
+                                          (ndr_pull_flags_fn_t)ndr_pull_dom_sid);
        if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
                talloc_free(sid);
                return -1;
        }
-       out->data = (uint8_t *)dom_sid_string(mem_ctx, sid);
+       *out = data_blob_string_const(dom_sid_string(mem_ctx, sid));
        talloc_free(sid);
        if (out->data == NULL) {
                return -1;
        }
-       out->length = strlen((const char *)out->data);
        return 0;
 }
 
@@ -97,13 +96,14 @@ static int ldb_comparison_objectSid(struct ldb_context *ldb, void *mem_ctx,
                                    const struct ldb_val *v1, const struct ldb_val *v2)
 {
        if (ldb_comparision_objectSid_isString(v1) && ldb_comparision_objectSid_isString(v2)) {
-               return strcmp((const char *)v1->data, (const char *)v2->data);
+               return ldb_comparison_binary(ldb, mem_ctx, v1, v2);
        } else if (ldb_comparision_objectSid_isString(v1)
                   && !ldb_comparision_objectSid_isString(v2)) {
                struct ldb_val v;
                int ret;
                if (ldif_read_objectSid(ldb, mem_ctx, v1, &v) != 0) {
-                       return -1;
+                       /* Perhaps not a string after all */
+                       return ldb_comparison_binary(ldb, mem_ctx, v1, v2);
                }
                ret = ldb_comparison_binary(ldb, mem_ctx, &v, v2);
                talloc_free(v.data);
@@ -113,7 +113,8 @@ static int ldb_comparison_objectSid(struct ldb_context *ldb, void *mem_ctx,
                struct ldb_val v;
                int ret;
                if (ldif_read_objectSid(ldb, mem_ctx, v2, &v) != 0) {
-                       return -1;
+                       /* Perhaps not a string after all */
+                       return ldb_comparison_binary(ldb, mem_ctx, v1, v2);
                }
                ret = ldb_comparison_binary(ldb, mem_ctx, v1, &v);
                talloc_free(v.data);
@@ -129,11 +130,45 @@ static int ldb_canonicalise_objectSid(struct ldb_context *ldb, void *mem_ctx,
                                      const struct ldb_val *in, struct ldb_val *out)
 {
        if (ldb_comparision_objectSid_isString(in)) {
-               return ldif_read_objectSid(ldb, mem_ctx, in, out);
+               if (ldif_read_objectSid(ldb, mem_ctx, in, out) != 0) {
+                       /* Perhaps not a string after all */
+                       return ldb_handler_copy(ldb, mem_ctx, in, out);
+               }
+               return 0;
        }
        return ldb_handler_copy(ldb, mem_ctx, in, out);
 }
 
+static int extended_dn_read_SID(struct ldb_context *ldb, void *mem_ctx,
+                             const struct ldb_val *in, struct ldb_val *out)
+{
+       struct dom_sid sid;
+       enum ndr_err_code ndr_err;
+       if (ldb_comparision_objectSid_isString(in)) {
+               if (ldif_read_objectSid(ldb, mem_ctx, in, out) == 0) {
+                       return 0;
+               }
+       }
+       
+       /* Perhaps not a string after all */
+       *out = data_blob_talloc(mem_ctx, NULL, in->length/2+1);
+
+       if (!out->data) {
+               return -1;
+       }
+
+       (*out).length = strhex_to_str((char *)out->data, out->length,
+                                    (const char *)in->data, in->length);
+
+       /* Check it looks like a SID */
+       ndr_err = ndr_pull_struct_blob_all(out, mem_ctx, NULL, &sid,
+                                          (ndr_pull_flags_fn_t)ndr_pull_dom_sid);
+       if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
+               return -1;
+       }
+       return 0;
+}
+
 /*
   convert a ldif formatted objectGUID to a NDR formatted blob
 */
@@ -144,7 +179,7 @@ static int ldif_read_objectGUID(struct ldb_context *ldb, void *mem_ctx,
        NTSTATUS status;
        enum ndr_err_code ndr_err;
 
-       status = GUID_from_string((const char *)in->data, &guid);
+       status = GUID_from_data_blob(in, &guid);
        if (!NT_STATUS_IS_OK(status)) {
                return -1;
        }
@@ -165,8 +200,8 @@ static int ldif_write_objectGUID(struct ldb_context *ldb, void *mem_ctx,
 {
        struct GUID guid;
        enum ndr_err_code ndr_err;
-       ndr_err = ndr_pull_struct_blob(in, mem_ctx, NULL, &guid,
-                                      (ndr_pull_flags_fn_t)ndr_pull_GUID);
+       ndr_err = ndr_pull_struct_blob_all(in, mem_ctx, NULL, &guid,
+                                          (ndr_pull_flags_fn_t)ndr_pull_GUID);
        if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
                return -1;
        }
@@ -180,20 +215,42 @@ static int ldif_write_objectGUID(struct ldb_context *ldb, void *mem_ctx,
 
 static bool ldb_comparision_objectGUID_isString(const struct ldb_val *v)
 {
-       struct GUID guid;
-       NTSTATUS status;
+       if (v->length != 36 && v->length != 38) return false;
 
-       if (v->length < 33) return false;
-
-       /* see if the input if null-terninated (safety check for the below) */
-       if (v->data[v->length] != '\0') return false;
+       /* Might be a GUID string, can't be a binary GUID (fixed 16 bytes) */
+       return true;
+}
 
-       status = GUID_from_string((const char *)v->data, &guid);
-       if (!NT_STATUS_IS_OK(status)) {
-               return false;
+static int extended_dn_read_GUID(struct ldb_context *ldb, void *mem_ctx,
+                             const struct ldb_val *in, struct ldb_val *out)
+{
+       struct GUID guid;
+       enum ndr_err_code ndr_err;
+       if (in->length == 36 && ldif_read_objectGUID(ldb, mem_ctx, in, out) == 0) {
+               return 0;
        }
 
-       return true;
+       /* Try as 'hex' form */
+       if (in->length != 32) {
+               return -1;
+       }
+               
+       *out = data_blob_talloc(mem_ctx, NULL, in->length/2+1);
+       
+       if (!out->data) {
+               return -1;
+       }
+       
+       (*out).length = strhex_to_str((char *)out->data, out->length,
+                                     (const char *)in->data, in->length);
+       
+       /* Check it looks like a GUID */
+       ndr_err = ndr_pull_struct_blob_all(out, mem_ctx, NULL, &guid,
+                                          (ndr_pull_flags_fn_t)ndr_pull_GUID);
+       if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
+               return -1;
+       }
+       return 0;
 }
 
 /*
@@ -203,13 +260,14 @@ static int ldb_comparison_objectGUID(struct ldb_context *ldb, void *mem_ctx,
                                     const struct ldb_val *v1, const struct ldb_val *v2)
 {
        if (ldb_comparision_objectGUID_isString(v1) && ldb_comparision_objectGUID_isString(v2)) {
-               return strcmp((const char *)v1->data, (const char *)v2->data);
+               return ldb_comparison_binary(ldb, mem_ctx, v1, v2);
        } else if (ldb_comparision_objectGUID_isString(v1)
                   && !ldb_comparision_objectGUID_isString(v2)) {
                struct ldb_val v;
                int ret;
                if (ldif_read_objectGUID(ldb, mem_ctx, v1, &v) != 0) {
-                       return -1;
+                       /* Perhaps it wasn't a valid string after all */
+                       return ldb_comparison_binary(ldb, mem_ctx, v1, v2);
                }
                ret = ldb_comparison_binary(ldb, mem_ctx, &v, v2);
                talloc_free(v.data);
@@ -219,7 +277,8 @@ static int ldb_comparison_objectGUID(struct ldb_context *ldb, void *mem_ctx,
                struct ldb_val v;
                int ret;
                if (ldif_read_objectGUID(ldb, mem_ctx, v2, &v) != 0) {
-                       return -1;
+                       /* Perhaps it wasn't a valid string after all */
+                       return ldb_comparison_binary(ldb, mem_ctx, v1, v2);
                }
                ret = ldb_comparison_binary(ldb, mem_ctx, v1, &v);
                talloc_free(v.data);
@@ -235,7 +294,11 @@ static int ldb_canonicalise_objectGUID(struct ldb_context *ldb, void *mem_ctx,
                                       const struct ldb_val *in, struct ldb_val *out)
 {
        if (ldb_comparision_objectGUID_isString(in)) {
-               return ldif_read_objectGUID(ldb, mem_ctx, in, out);
+               if (ldif_read_objectGUID(ldb, mem_ctx, in, out) != 0) {
+                       /* Perhaps it wasn't a valid string after all */
+                       return ldb_handler_copy(ldb, mem_ctx, in, out);
+               }
+               return 0;
        }
        return ldb_handler_copy(ldb, mem_ctx, in, out);
 }
@@ -276,8 +339,9 @@ static int ldif_write_ntSecurityDescriptor(struct ldb_context *ldb, void *mem_ct
        if (sd == NULL) {
                return -1;
        }
+       /* We can't use ndr_pull_struct_blob_all because this contains relative pointers */
        ndr_err = ndr_pull_struct_blob(in, sd, NULL, sd,
-                                      (ndr_pull_flags_fn_t)ndr_pull_security_descriptor);
+                                          (ndr_pull_flags_fn_t)ndr_pull_security_descriptor);
        if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
                talloc_free(sd);
                return -1;
@@ -314,7 +378,7 @@ static int ldif_canonicalise_objectCategory(struct ldb_context *ldb, void *mem_c
                }
                return LDB_SUCCESS;
        }
-       dn1 = ldb_dn_new(tmp_ctx, ldb, (char *)in->data);
+       dn1 = ldb_dn_from_ldb_val(tmp_ctx, ldb, in);
        if ( ! ldb_dn_validate(dn1)) {
                const char *lDAPDisplayName = talloc_strndup(tmp_ctx, (char *)in->data, in->length);
                class = dsdb_class_by_lDAPDisplayName(schema, lDAPDisplayName);
@@ -477,10 +541,10 @@ static int ldif_write_prefixMap(struct ldb_context *ldb, void *mem_ctx,
        if (blob == NULL) {
                return -1;
        }
-       ndr_err = ndr_pull_struct_blob(in, blob, 
-                                      lp_iconv_convenience(ldb_get_opaque(ldb, "loadparm")), 
-                                      blob,
-                                      (ndr_pull_flags_fn_t)ndr_pull_prefixMapBlob);
+       ndr_err = ndr_pull_struct_blob_all(in, blob, 
+                                          lp_iconv_convenience(ldb_get_opaque(ldb, "loadparm")), 
+                                          blob,
+                                          (ndr_pull_flags_fn_t)ndr_pull_prefixMapBlob);
        if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
                talloc_free(blob);
                return -1;
@@ -561,41 +625,71 @@ static int ldif_comparison_prefixMap(struct ldb_context *ldb, void *mem_ctx,
        return ret;
 }
 
+static int extended_dn_write_hex(struct ldb_context *ldb, void *mem_ctx,
+                                const struct ldb_val *in, struct ldb_val *out)
+{
+       *out = data_blob_string_const(data_blob_hex_string(mem_ctx, in));
+       if (!out->data) {
+               return -1;
+       }
+       return 0;
+}
+
+
 #define LDB_SYNTAX_SAMBA_GUID                  "LDB_SYNTAX_SAMBA_GUID"
 #define LDB_SYNTAX_SAMBA_OBJECT_CATEGORY       "LDB_SYNTAX_SAMBA_OBJECT_CATEGORY"
 #define LDB_SYNTAX_SAMBA_PREFIX_MAP    "LDB_SYNTAX_SAMBA_PREFIX_MAP"
 
 static const struct ldb_schema_syntax samba_syntaxes[] = {
        {
-               .name           = LDB_SYNTAX_SAMBA_SID,
-               .ldif_read_fn   = ldif_read_objectSid,
-               .ldif_write_fn  = ldif_write_objectSid,
-               .canonicalise_fn= ldb_canonicalise_objectSid,
-               .comparison_fn  = ldb_comparison_objectSid
+               .name             = LDB_SYNTAX_SAMBA_SID,
+               .ldif_read_fn     = ldif_read_objectSid,
+               .ldif_write_fn    = ldif_write_objectSid,
+               .canonicalise_fn  = ldb_canonicalise_objectSid,
+               .comparison_fn    = ldb_comparison_objectSid
+       },{
+               .name             = LDB_SYNTAX_SAMBA_SECURITY_DESCRIPTOR,
+               .ldif_read_fn     = ldif_read_ntSecurityDescriptor,
+               .ldif_write_fn    = ldif_write_ntSecurityDescriptor,
+               .canonicalise_fn  = ldb_handler_copy,
+               .comparison_fn    = ldb_comparison_binary
        },{
-               .name           = LDB_SYNTAX_SAMBA_SECURITY_DESCRIPTOR,
-               .ldif_read_fn   = ldif_read_ntSecurityDescriptor,
-               .ldif_write_fn  = ldif_write_ntSecurityDescriptor,
-               .canonicalise_fn= ldb_handler_copy,
-               .comparison_fn  = ldb_comparison_binary
+               .name             = LDB_SYNTAX_SAMBA_GUID,
+               .ldif_read_fn     = ldif_read_objectGUID,
+               .ldif_write_fn    = ldif_write_objectGUID,
+               .canonicalise_fn  = ldb_canonicalise_objectGUID,
+               .comparison_fn    = ldb_comparison_objectGUID
        },{
-               .name           = LDB_SYNTAX_SAMBA_GUID,
-               .ldif_read_fn   = ldif_read_objectGUID,
-               .ldif_write_fn  = ldif_write_objectGUID,
-               .canonicalise_fn= ldb_canonicalise_objectGUID,
-               .comparison_fn  = ldb_comparison_objectGUID
+               .name             = LDB_SYNTAX_SAMBA_OBJECT_CATEGORY,
+               .ldif_read_fn     = ldb_handler_copy,
+               .ldif_write_fn    = ldb_handler_copy,
+               .canonicalise_fn  = ldif_canonicalise_objectCategory,
+               .comparison_fn    = ldif_comparison_objectCategory
+       },{
+               .name             = LDB_SYNTAX_SAMBA_PREFIX_MAP,
+               .ldif_read_fn     = ldif_read_prefixMap,
+               .ldif_write_fn    = ldif_write_prefixMap,
+               .canonicalise_fn  = ldif_canonicalise_prefixMap,
+               .comparison_fn    = ldif_comparison_prefixMap
+       }
+};
+
+static const struct ldb_dn_extended_syntax samba_dn_syntax[] = {
+       {
+               .name             = "SID",
+               .read_fn          = extended_dn_read_SID,
+               .write_clear_fn   = ldif_write_objectSid,
+               .write_hex_fn     = extended_dn_write_hex
        },{
-               .name           = LDB_SYNTAX_SAMBA_OBJECT_CATEGORY,
-               .ldif_read_fn   = ldb_handler_copy,
-               .ldif_write_fn  = ldb_handler_copy,
-               .canonicalise_fn= ldif_canonicalise_objectCategory,
-               .comparison_fn  = ldif_comparison_objectCategory
+               .name             = "GUID",
+               .read_fn          = extended_dn_read_GUID,
+               .write_clear_fn   = ldif_write_objectGUID,
+               .write_hex_fn     = extended_dn_write_hex
        },{
-               .name           = LDB_SYNTAX_SAMBA_PREFIX_MAP,
-               .ldif_read_fn   = ldif_read_prefixMap,
-               .ldif_write_fn  = ldif_write_prefixMap,
-               .canonicalise_fn= ldif_canonicalise_prefixMap,
-               .comparison_fn  = ldif_comparison_prefixMap
+               .name             = "WKGUID",
+               .read_fn          = ldb_handler_copy,
+               .write_clear_fn   = ldb_handler_copy,
+               .write_hex_fn     = ldb_handler_copy
        }
 };
 
@@ -617,19 +711,6 @@ static const struct {
        { "fRSReplicaSetGUID",          LDB_SYNTAX_SAMBA_GUID },
        { "netbootGUID",                LDB_SYNTAX_SAMBA_GUID },
        { "objectCategory",             LDB_SYNTAX_SAMBA_OBJECT_CATEGORY },
-       { "member",                     LDB_SYNTAX_DN },
-       { "memberOf",                   LDB_SYNTAX_DN },
-       { "nCName",                     LDB_SYNTAX_DN },
-       { "schemaNamingContext",        LDB_SYNTAX_DN },
-       { "configurationNamingContext", LDB_SYNTAX_DN },
-       { "rootDomainNamingContext",    LDB_SYNTAX_DN },
-       { "defaultNamingContext",       LDB_SYNTAX_DN },
-       { "subRefs",                    LDB_SYNTAX_DN },
-       { "dMDLocation",                LDB_SYNTAX_DN },
-       { "serverReference",            LDB_SYNTAX_DN },
-       { "masteredBy",                 LDB_SYNTAX_DN },
-       { "msDs-masteredBy",            LDB_SYNTAX_DN },
-       { "fSMORoleOwner",              LDB_SYNTAX_DN },
        { "prefixMap",                  LDB_SYNTAX_SAMBA_PREFIX_MAP }
 };
 
@@ -669,11 +750,21 @@ int ldb_register_samba_handlers(struct ldb_context *ldb)
                        return -1;
                }
 
-               ret = ldb_schema_attribute_add_with_syntax(ldb, samba_attributes[i].name, 0, s);
+               ret = ldb_schema_attribute_add_with_syntax(ldb, samba_attributes[i].name, LDB_ATTR_FLAG_FIXED, s);
                if (ret != LDB_SUCCESS) {
                        return ret;
                }
        }
 
+       for (i=0; i < ARRAY_SIZE(samba_dn_syntax); i++) {
+               int ret;
+               ret = ldb_dn_extended_add_syntax(ldb, LDB_ATTR_FLAG_FIXED, &samba_dn_syntax[i]);
+               if (ret != LDB_SUCCESS) {
+                       return ret;
+               }
+
+               
+       }
+
        return LDB_SUCCESS;
 }