Avoid including libds/common/roles.h in public loadparm.h header.
[obnox/samba/samba-obnox.git] / source4 / rpc_server / lsa / dcesrv_lsa.c
index b2c4edaffe1338896f129805c9effbec3da3069c..c55f679bfd9deb861fbf0ba895b72f1d60216122 100644 (file)
@@ -33,6 +33,7 @@
 #include "libcli/security/session.h"
 #include "libcli/lsarpc/util_lsarpc.h"
 #include "lib/messaging/irpc.h"
+#include "libds/common/roles.h"
 
 /*
   this type allows us to distinguish handle types
@@ -4214,375 +4215,80 @@ static NTSTATUS dcesrv_lsa_LSARUNREGISTERAUDITEVENT(struct dcesrv_call_state *dc
 static NTSTATUS dcesrv_lsa_lsaRQueryForestTrustInformation(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx,
                       struct lsa_lsaRQueryForestTrustInformation *r)
 {
-       DCESRV_FAULT(DCERPC_FAULT_OP_RNG_ERROR);
-}
-
-#define DNS_CMP_MATCH 0
-#define DNS_CMP_FIRST_IS_CHILD 1
-#define DNS_CMP_SECOND_IS_CHILD 2
-#define DNS_CMP_NO_MATCH 3
-
-/* this function assumes names are well formed DNS names.
- * it doesn't validate them */
-static int dns_cmp(const char *s1, size_t l1,
-                  const char *s2, size_t l2)
-{
-       const char *p1, *p2;
-       size_t t1, t2;
-       int cret;
-
-       if (l1 == l2) {
-               if (strcasecmp_m(s1, s2) == 0) {
-                       return DNS_CMP_MATCH;
-               }
-               return DNS_CMP_NO_MATCH;
-       }
-
-       if (l1 > l2) {
-               p1 = s1;
-               p2 = s2;
-               t1 = l1;
-               t2 = l2;
-               cret = DNS_CMP_FIRST_IS_CHILD;
-       } else {
-               p1 = s2;
-               p2 = s1;
-               t1 = l2;
-               t2 = l1;
-               cret = DNS_CMP_SECOND_IS_CHILD;
-       }
-
-       if (p1[t1 - t2 - 1] != '.') {
-               return DNS_CMP_NO_MATCH;
-       }
-
-       if (strcasecmp_m(&p1[t1 - t2], p2) == 0) {
-               return cret;
-       }
+       struct dcesrv_handle *h = NULL;
+       struct lsa_policy_state *p_state = NULL;
+       int forest_level = DS_DOMAIN_FUNCTION_2000;
+       const char * const trust_attrs[] = {
+               "securityIdentifier",
+               "flatName",
+               "trustPartner",
+               "trustAttributes",
+               "trustDirection",
+               "trustType",
+               "msDS-TrustForestTrustInfo",
+               NULL
+       };
+       struct ldb_message *trust_tdo_msg = NULL;
+       struct lsa_TrustDomainInfoInfoEx *trust_tdo = NULL;
+       struct ForestTrustInfo *trust_fti = NULL;
+       struct lsa_ForestTrustInformation *trust_lfti = NULL;
+       NTSTATUS status;
 
-       return DNS_CMP_NO_MATCH;
-}
+       DCESRV_PULL_HANDLE(h, r->in.handle, LSA_HANDLE_POLICY);
 
-/* decode all TDOs forest trust info blobs */
-static NTSTATUS get_ft_info(TALLOC_CTX *mem_ctx,
-                           struct ldb_message *msg,
-                           struct ForestTrustInfo *info)
-{
-       const struct ldb_val *ft_blob;
-       enum ndr_err_code ndr_err;
+       p_state = h->data;
 
-       ft_blob = ldb_msg_find_ldb_val(msg, "msDS-TrustForestTrustInfo");
-       if (!ft_blob || !ft_blob->data) {
-               return NT_STATUS_OBJECT_NAME_NOT_FOUND;
-       }
-       /* ldb_val is equivalent to DATA_BLOB */
-       ndr_err = ndr_pull_struct_blob_all(ft_blob, mem_ctx, info,
-                                          (ndr_pull_flags_fn_t)ndr_pull_ForestTrustInfo);
-       if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
+       if (strcmp(p_state->domain_dns, p_state->forest_dns)) {
                return NT_STATUS_INVALID_DOMAIN_STATE;
        }
 
-       return NT_STATUS_OK;
-}
-
-static NTSTATUS own_ft_info(struct lsa_policy_state *ps,
-                           struct ForestTrustInfo *fti)
-{
-       struct ForestTrustDataDomainInfo *info;
-       struct ForestTrustInfoRecord *rec;
-
-       fti->version = 1;
-       fti->count = 2;
-       fti->records = talloc_array(fti,
-                                   struct ForestTrustInfoRecordArmor, 2);
-       if (!fti->records) {
-               return NT_STATUS_NO_MEMORY;
+       forest_level = dsdb_forest_functional_level(p_state->sam_ldb);
+       if (forest_level < DS_DOMAIN_FUNCTION_2003) {
+               return NT_STATUS_INVALID_DOMAIN_STATE;
        }
 
-        /* TLN info */
-       rec = &fti->records[0].record;
-
-       rec->flags = 0;
-       rec->timestamp = 0;
-       rec->type = LSA_FOREST_TRUST_TOP_LEVEL_NAME;
-
-       rec->data.name.string = talloc_strdup(fti, ps->forest_dns);
-       if (!rec->data.name.string) {
-               return NT_STATUS_NO_MEMORY;
+       if (r->in.trusted_domain_name->string == NULL) {
+               return NT_STATUS_NO_SUCH_DOMAIN;
        }
-       rec->data.name.size = strlen(rec->data.name.string);
-
-        /* DOMAIN info */
-       rec = &fti->records[1].record;
 
-       rec->flags = 0;
-       rec->timestamp = 0;
-       rec->type = LSA_FOREST_TRUST_DOMAIN_INFO;
-
-        info = &rec->data.info;
-
-       info->sid = *ps->domain_sid;
-       info->dns_name.string = talloc_strdup(fti, ps->domain_dns);
-       if (!info->dns_name.string) {
-               return NT_STATUS_NO_MEMORY;
+       status = dsdb_trust_search_tdo(p_state->sam_ldb,
+                                      r->in.trusted_domain_name->string,
+                                      r->in.trusted_domain_name->string,
+                                      trust_attrs, mem_ctx, &trust_tdo_msg);
+       if (NT_STATUS_EQUAL(status, NT_STATUS_OBJECT_NAME_NOT_FOUND)) {
+               return NT_STATUS_NO_SUCH_DOMAIN;
        }
-       info->dns_name.size = strlen(info->dns_name.string);
-       info->netbios_name.string = talloc_strdup(fti, ps->domain_name);
-       if (!info->netbios_name.string) {
-               return NT_STATUS_NO_MEMORY;
+       if (!NT_STATUS_IS_OK(status)) {
+               return status;
        }
-       info->netbios_name.size = strlen(info->netbios_name.string);
 
-       return NT_STATUS_OK;
-}
-
-static NTSTATUS make_ft_info(TALLOC_CTX *mem_ctx,
-                            struct lsa_ForestTrustInformation *lfti,
-                            struct ForestTrustInfo *fti)
-{
-       struct lsa_ForestTrustRecord *lrec;
-       struct ForestTrustInfoRecord *rec;
-       struct lsa_StringLarge *tln;
-       struct lsa_ForestTrustDomainInfo *info;
-       uint32_t i;
-
-       fti->version = 1;
-       fti->count = lfti->count;
-       fti->records = talloc_array(mem_ctx,
-                                   struct ForestTrustInfoRecordArmor,
-                                   fti->count);
-       if (!fti->records) {
-               return NT_STATUS_NO_MEMORY;
-       }
-       for (i = 0; i < fti->count; i++) {
-               lrec = lfti->entries[i];
-               rec = &fti->records[i].record;
-
-               rec->flags = lrec->flags;
-               rec->timestamp = lrec->time;
-               rec->type = lrec->type;
-
-               switch (lrec->type) {
-               case LSA_FOREST_TRUST_TOP_LEVEL_NAME:
-               case LSA_FOREST_TRUST_TOP_LEVEL_NAME_EX:
-                       tln = &lrec->forest_trust_data.top_level_name;
-                       rec->data.name.string =
-                               talloc_strdup(mem_ctx, tln->string);
-                       if (!rec->data.name.string) {
-                               return NT_STATUS_NO_MEMORY;
-                       }
-                       rec->data.name.size = strlen(rec->data.name.string);
-                       break;
-               case LSA_FOREST_TRUST_DOMAIN_INFO:
-                       info = &lrec->forest_trust_data.domain_info;
-                       rec->data.info.sid = *info->domain_sid;
-                       rec->data.info.dns_name.string =
-                               talloc_strdup(mem_ctx,
-                                           info->dns_domain_name.string);
-                       if (!rec->data.info.dns_name.string) {
-                               return NT_STATUS_NO_MEMORY;
-                       }
-                       rec->data.info.dns_name.size =
-                               strlen(rec->data.info.dns_name.string);
-                       rec->data.info.netbios_name.string =
-                               talloc_strdup(mem_ctx,
-                                           info->netbios_domain_name.string);
-                       if (!rec->data.info.netbios_name.string) {
-                               return NT_STATUS_NO_MEMORY;
-                       }
-                       rec->data.info.netbios_name.size =
-                               strlen(rec->data.info.netbios_name.string);
-                       break;
-               default:
-                       return NT_STATUS_INVALID_DOMAIN_STATE;
-               }
+       status = dsdb_trust_parse_tdo_info(mem_ctx, trust_tdo_msg, &trust_tdo);
+       if (!NT_STATUS_IS_OK(status)) {
+               return status;
        }
 
-       return NT_STATUS_OK;
-}
-
-static NTSTATUS add_collision(struct lsa_ForestTrustCollisionInfo *c_info,
-                             uint32_t idx, uint32_t collision_type,
-                             uint32_t conflict_type, const char *tdo_name);
-
-static NTSTATUS check_ft_info(TALLOC_CTX *mem_ctx,
-                             const char *tdo_name,
-                             struct ForestTrustInfo *tdo_fti,
-                             struct ForestTrustInfo *new_fti,
-                             struct lsa_ForestTrustCollisionInfo *c_info)
-{
-       struct ForestTrustInfoRecord *nrec;
-       struct ForestTrustInfoRecord *trec;
-       const char *dns_name;
-       const char *nb_name;
-       struct dom_sid *sid = NULL;
-       const char *tname = NULL;
-       size_t dns_len;
-       size_t tlen = 0;
-       NTSTATUS nt_status = NT_STATUS_OK;
-       uint32_t new_fti_idx;
-       uint32_t i;
-       /* use always TDO type, until we understand when Xref can be used */
-       uint32_t collision_type = LSA_FOREST_TRUST_COLLISION_TDO;
-       bool tln_conflict;
-       bool sid_conflict;
-       bool nb_conflict;
-       bool exclusion;
-       bool ex_rule = false;
-       int ret;
-
-       for (new_fti_idx = 0; new_fti_idx < new_fti->count; new_fti_idx++) {
-
-               nrec = &new_fti->records[new_fti_idx].record;
-               dns_name = NULL;
-               nb_name = NULL;
-               tln_conflict = false;
-               sid_conflict = false;
-               nb_conflict = false;
-               exclusion = false;
-
-               switch (nrec->type) {
-               case LSA_FOREST_TRUST_TOP_LEVEL_NAME_EX:
-                       /* exclusions do not conflict by definition */
-                       break;
-
-               case FOREST_TRUST_TOP_LEVEL_NAME:
-                       dns_name = nrec->data.name.string;
-                       dns_len = nrec->data.name.size;
-                       break;
-
-               case LSA_FOREST_TRUST_DOMAIN_INFO:
-                       dns_name = nrec->data.info.dns_name.string;
-                       dns_len = nrec->data.info.dns_name.size;
-                       nb_name = nrec->data.info.netbios_name.string;
-                       sid = &nrec->data.info.sid;
-                       break;
-               }
-
-               if (!dns_name) continue;
-
-               /* check if this is already taken and not excluded */
-               for (i = 0; i < tdo_fti->count; i++) {
-                       trec = &tdo_fti->records[i].record;
-
-                       switch (trec->type) {
-                       case FOREST_TRUST_TOP_LEVEL_NAME:
-                               ex_rule = false;
-                               tname = trec->data.name.string;
-                               tlen = trec->data.name.size;
-                               break;
-                       case FOREST_TRUST_TOP_LEVEL_NAME_EX:
-                               ex_rule = true;
-                               tname = trec->data.name.string;
-                               tlen = trec->data.name.size;
-                               break;
-                       case FOREST_TRUST_DOMAIN_INFO:
-                               ex_rule = false;
-                               tname = trec->data.info.dns_name.string;
-                               tlen = trec->data.info.dns_name.size;
-                       }
-                       ret = dns_cmp(dns_name, dns_len, tname, tlen);
-                       switch (ret) {
-                       case DNS_CMP_MATCH:
-                               /* if it matches exclusion,
-                                * it doesn't conflict */
-                               if (ex_rule) {
-                                       exclusion = true;
-                                       break;
-                               }
-                               /* fall through */
-                       case DNS_CMP_FIRST_IS_CHILD:
-                       case DNS_CMP_SECOND_IS_CHILD:
-                               tln_conflict = true;
-                               /* fall through */
-                       default:
-                               break;
-                       }
-
-                       /* explicit exclusion, no dns name conflict here */
-                       if (exclusion) {
-                               tln_conflict = false;
-                       }
-
-                       if (trec->type != FOREST_TRUST_DOMAIN_INFO) {
-                               continue;
-                       }
-
-                       /* also test for domain info */
-                       if (!(trec->flags & LSA_SID_DISABLED_ADMIN) &&
-                           dom_sid_compare(&trec->data.info.sid, sid) == 0) {
-                               sid_conflict = true;
-                       }
-                       if (!(trec->flags & LSA_NB_DISABLED_ADMIN) &&
-                           (nb_name != NULL) &&
-                           strcasecmp_m(trec->data.info.netbios_name.string,
-                                        nb_name) == 0) {
-                               nb_conflict = true;
-                       }
-               }
-
-               if (tln_conflict) {
-                       nt_status = add_collision(c_info, new_fti_idx,
-                                                 collision_type,
-                                                 LSA_TLN_DISABLED_CONFLICT,
-                                                 tdo_name);
-                       if (!NT_STATUS_IS_OK(nt_status)) {
-                               goto done;
-                       }
-               }
-               if (sid_conflict) {
-                       nt_status = add_collision(c_info, new_fti_idx,
-                                                 collision_type,
-                                                 LSA_SID_DISABLED_CONFLICT,
-                                                 tdo_name);
-                       if (!NT_STATUS_IS_OK(nt_status)) {
-                               goto done;
-                       }
-               }
-               if (nb_conflict) {
-                       nt_status = add_collision(c_info, new_fti_idx,
-                                                 collision_type,
-                                                 LSA_NB_DISABLED_CONFLICT,
-                                                 tdo_name);
-                       if (!NT_STATUS_IS_OK(nt_status)) {
-                               goto done;
-                       }
-               }
+       if (!(trust_tdo->trust_attributes & LSA_TRUST_ATTRIBUTE_FOREST_TRANSITIVE)) {
+               return NT_STATUS_INVALID_PARAMETER;
        }
 
-done:
-       return nt_status;
-}
-
-static NTSTATUS add_collision(struct lsa_ForestTrustCollisionInfo *c_info,
-                             uint32_t idx, uint32_t collision_type,
-                             uint32_t conflict_type, const char *tdo_name)
-{
-       struct lsa_ForestTrustCollisionRecord **es;
-       uint32_t i = c_info->count;
-
-       es = talloc_realloc(c_info, c_info->entries,
-                           struct lsa_ForestTrustCollisionRecord *, i + 1);
-       if (!es) {
-               return NT_STATUS_NO_MEMORY;
+       if (r->in.highest_record_type >= LSA_FOREST_TRUST_RECORD_TYPE_LAST) {
+               return NT_STATUS_INVALID_PARAMETER;
        }
-       c_info->entries = es;
-       c_info->count = i + 1;
 
-       es[i] = talloc(es, struct lsa_ForestTrustCollisionRecord);
-       if (!es[i]) {
-               return NT_STATUS_NO_MEMORY;
+       status = dsdb_trust_parse_forest_info(mem_ctx,
+                                             trust_tdo_msg,
+                                             &trust_fti);
+       if (!NT_STATUS_IS_OK(status)) {
+               return status;
        }
 
-       es[i]->index = idx;
-       es[i]->type = collision_type;
-       es[i]->flags = conflict_type;
-       es[i]->name.string = talloc_strdup(es[i], tdo_name);
-       if (!es[i]->name.string) {
-               return NT_STATUS_NO_MEMORY;
+       status = dsdb_trust_forest_info_to_lsa(mem_ctx, trust_fti,
+                                              &trust_lfti);
+       if (!NT_STATUS_IS_OK(status)) {
+               return status;
        }
-       es[i]->name.size = strlen(es[i]->name.string);
 
+       *r->out.forest_trust_info = trust_lfti;
        return NT_STATUS_OK;
 }
 
@@ -4595,22 +4301,32 @@ static NTSTATUS dcesrv_lsa_lsaRSetForestTrustInformation(struct dcesrv_call_stat
 {
        struct dcesrv_handle *h;
        struct lsa_policy_state *p_state;
-       const char *trust_attrs[] = { "trustPartner", "trustAttributes",
-                                     "msDS-TrustForestTrustInfo", NULL };
-       struct ldb_message **dom_res = NULL;
-       struct ldb_dn *tdo_dn;
-       struct ldb_message *msg;
-       int num_res, i;
-       const char *td_name;
-       uint32_t trust_attributes;
-       struct lsa_ForestTrustCollisionInfo *c_info;
-       struct ForestTrustInfo *nfti;
-       struct ForestTrustInfo *fti;
-       DATA_BLOB ft_blob;
+       const char * const trust_attrs[] = {
+               "securityIdentifier",
+               "flatName",
+               "trustPartner",
+               "trustAttributes",
+               "trustDirection",
+               "trustType",
+               "msDS-TrustForestTrustInfo",
+               NULL
+       };
+       struct ldb_message *trust_tdo_msg = NULL;
+       struct lsa_TrustDomainInfoInfoEx *trust_tdo = NULL;
+       struct lsa_ForestTrustInformation *step1_lfti = NULL;
+       struct lsa_ForestTrustInformation *step2_lfti = NULL;
+       struct ForestTrustInfo *trust_fti = NULL;
+       struct ldb_result *trusts_res = NULL;
+       unsigned int i;
+       struct lsa_TrustDomainInfoInfoEx *xref_tdo = NULL;
+       struct lsa_ForestTrustInformation *xref_lfti = NULL;
+       struct lsa_ForestTrustCollisionInfo *c_info = NULL;
+       DATA_BLOB ft_blob = {};
+       struct ldb_message *msg = NULL;
+       NTSTATUS status;
        enum ndr_err_code ndr_err;
-       NTSTATUS nt_status;
-       bool am_rodc;
        int ret;
+       bool in_transaction = false;
 
        DCESRV_PULL_HANDLE(h, r->in.handle, LSA_HANDLE_POLICY);
 
@@ -4620,165 +4336,244 @@ static NTSTATUS dcesrv_lsa_lsaRSetForestTrustInformation(struct dcesrv_call_stat
                return NT_STATUS_INVALID_DOMAIN_STATE;
        }
 
-       /* abort if we are not a PDC */
-       if (!samdb_is_pdc(p_state->sam_ldb)) {
-               return NT_STATUS_INVALID_DOMAIN_ROLE;
+       if (r->in.check_only == 0) {
+               ret = ldb_transaction_start(p_state->sam_ldb);
+               if (ret != LDB_SUCCESS) {
+                       return NT_STATUS_INTERNAL_DB_CORRUPTION;
+               }
+               in_transaction = true;
        }
 
-       ret = samdb_rodc(p_state->sam_ldb, &am_rodc);
-       if (ret == LDB_SUCCESS && am_rodc) {
-               return NT_STATUS_NO_SUCH_DOMAIN;
+       /*
+        * abort if we are not a PDC
+        *
+        * In future we should use a function like IsEffectiveRoleOwner()
+        */
+       if (!samdb_is_pdc(p_state->sam_ldb)) {
+               status = NT_STATUS_INVALID_DOMAIN_ROLE;
+               goto done;
        }
 
-       /* check caller has TRUSTED_SET_AUTH */
-
-       /* fetch all trusted domain objects */
-       num_res = gendb_search(p_state->sam_ldb, mem_ctx,
-                              p_state->system_dn,
-                              &dom_res, trust_attrs,
-                              "(objectclass=trustedDomain)");
-       if (num_res == 0) {
-               return NT_STATUS_NO_SUCH_DOMAIN;
+       if (r->in.trusted_domain_name->string == NULL) {
+               status = NT_STATUS_NO_SUCH_DOMAIN;
+               goto done;
        }
 
-       for (i = 0; i < num_res; i++) {
-               td_name = ldb_msg_find_attr_as_string(dom_res[i],
-                                                     "trustPartner", NULL);
-               if (!td_name) {
-                       return NT_STATUS_INVALID_DOMAIN_STATE;
-               }
-               if (strcasecmp_m(td_name,
-                                r->in.trusted_domain_name->string) == 0) {
-                       break;
-               }
+       status = dsdb_trust_search_tdo(p_state->sam_ldb,
+                                      r->in.trusted_domain_name->string,
+                                      r->in.trusted_domain_name->string,
+                                      trust_attrs, mem_ctx, &trust_tdo_msg);
+       if (NT_STATUS_EQUAL(status, NT_STATUS_OBJECT_NAME_NOT_FOUND)) {
+               status = NT_STATUS_NO_SUCH_DOMAIN;
+               goto done;
        }
-       if (i >= num_res) {
-               return NT_STATUS_NO_SUCH_DOMAIN;
+       if (!NT_STATUS_IS_OK(status)) {
+               goto done;
        }
 
-       tdo_dn = dom_res[i]->dn;
-
-       trust_attributes = ldb_msg_find_attr_as_uint(dom_res[i],
-                                                    "trustAttributes", 0);
-       if (!(trust_attributes & LSA_TRUST_ATTRIBUTE_FOREST_TRANSITIVE)) {
-               return NT_STATUS_INVALID_PARAMETER;
+       status = dsdb_trust_parse_tdo_info(mem_ctx, trust_tdo_msg, &trust_tdo);
+       if (!NT_STATUS_IS_OK(status)) {
+               goto done;
        }
 
-       if (r->in.highest_record_type >= LSA_FOREST_TRUST_RECORD_TYPE_LAST) {
-               return NT_STATUS_INVALID_PARAMETER;
+       if (!(trust_tdo->trust_attributes & LSA_TRUST_ATTRIBUTE_FOREST_TRANSITIVE)) {
+               status = NT_STATUS_INVALID_PARAMETER;
+               goto done;
        }
 
-       nfti = talloc(mem_ctx, struct ForestTrustInfo);
-       if (!nfti) {
-               return NT_STATUS_NO_MEMORY;
+       if (r->in.highest_record_type >= LSA_FOREST_TRUST_RECORD_TYPE_LAST) {
+               status = NT_STATUS_INVALID_PARAMETER;
+               goto done;
        }
 
-       nt_status = make_ft_info(nfti, r->in.forest_trust_info, nfti);
-       if (!NT_STATUS_IS_OK(nt_status)) {
-               return nt_status;
+       /*
+        * verify and normalize the given forest trust info.
+        *
+        * Step1: doesn't reorder yet, so step1_lfti might contain
+        * NULL entries. This means dsdb_trust_verify_forest_info()
+        * can generate collision entries with the callers index.
+        */
+       status = dsdb_trust_normalize_forest_info_step1(mem_ctx,
+                                                       r->in.forest_trust_info,
+                                                       &step1_lfti);
+       if (!NT_STATUS_IS_OK(status)) {
+               goto done;
        }
 
        c_info = talloc_zero(r->out.collision_info,
                             struct lsa_ForestTrustCollisionInfo);
-       if (!c_info) {
-               return NT_STATUS_NO_MEMORY;
+       if (c_info == NULL) {
+               status = NT_STATUS_NO_MEMORY;
+               goto done;
        }
 
-        /* first check own info, then other domains */
-       fti = talloc(mem_ctx, struct ForestTrustInfo);
-       if (!fti) {
-               return NT_STATUS_NO_MEMORY;
+       /*
+        * First check our own forest, then other domains/forests
+        */
+
+       status = dsdb_trust_xref_tdo_info(mem_ctx, p_state->sam_ldb,
+                                         &xref_tdo);
+       if (!NT_STATUS_IS_OK(status)) {
+               goto done;
+       }
+       status = dsdb_trust_xref_forest_info(mem_ctx, p_state->sam_ldb,
+                                            &xref_lfti);
+       if (!NT_STATUS_IS_OK(status)) {
+               goto done;
        }
 
-        nt_status = own_ft_info(p_state, fti);
-       if (!NT_STATUS_IS_OK(nt_status)) {
-               return nt_status;
+       /*
+        * The documentation proposed to generate
+        * LSA_FOREST_TRUST_COLLISION_XREF collisions.
+        * But Windows always uses LSA_FOREST_TRUST_COLLISION_TDO.
+        */
+       status = dsdb_trust_verify_forest_info(xref_tdo, xref_lfti,
+                                              LSA_FOREST_TRUST_COLLISION_TDO,
+                                              c_info, step1_lfti);
+       if (!NT_STATUS_IS_OK(status)) {
+               goto done;
        }
 
-       nt_status = check_ft_info(c_info, p_state->domain_dns,
-                                  fti, nfti, c_info);
-       if (!NT_STATUS_IS_OK(nt_status)) {
-               return nt_status;
+       /* fetch all other trusted domain objects */
+       status = dsdb_trust_search_tdos(p_state->sam_ldb,
+                                       trust_tdo->domain_name.string,
+                                       trust_attrs,
+                                       mem_ctx, &trusts_res);
+       if (!NT_STATUS_IS_OK(status)) {
+               goto done;
        }
 
-       for (i = 0; i < num_res; i++) {
-               fti = talloc(mem_ctx, struct ForestTrustInfo);
-               if (!fti) {
-                       return NT_STATUS_NO_MEMORY;
+       /*
+        * now check against the other domains.
+        * and generate LSA_FOREST_TRUST_COLLISION_TDO collisions.
+        */
+       for (i = 0; i < trusts_res->count; i++) {
+               struct lsa_TrustDomainInfoInfoEx *tdo = NULL;
+               struct ForestTrustInfo *fti = NULL;
+               struct lsa_ForestTrustInformation *lfti = NULL;
+
+               status = dsdb_trust_parse_tdo_info(mem_ctx,
+                                                  trusts_res->msgs[i],
+                                                  &tdo);
+               if (!NT_STATUS_IS_OK(status)) {
+                       goto done;
                }
 
-               nt_status = get_ft_info(mem_ctx, dom_res[i], fti);
-               if (!NT_STATUS_IS_OK(nt_status)) {
-                       if (NT_STATUS_EQUAL(nt_status,
-                           NT_STATUS_OBJECT_NAME_NOT_FOUND)) {
-                               continue;
-                       }
-                       return nt_status;
+               status = dsdb_trust_parse_forest_info(tdo,
+                                                     trusts_res->msgs[i],
+                                                     &fti);
+               if (NT_STATUS_EQUAL(status, NT_STATUS_NOT_FOUND)) {
+                       continue;
+               }
+               if (!NT_STATUS_IS_OK(status)) {
+                       goto done;
                }
 
-               td_name = ldb_msg_find_attr_as_string(dom_res[i],
-                                                     "trustPartner", NULL);
-               if (!td_name) {
-                       return NT_STATUS_INVALID_DOMAIN_STATE;
+               status = dsdb_trust_forest_info_to_lsa(tdo, fti, &lfti);
+               if (!NT_STATUS_IS_OK(status)) {
+                       goto done;
                }
 
-               nt_status = check_ft_info(c_info, td_name, fti, nfti, c_info);
-               if (!NT_STATUS_IS_OK(nt_status)) {
-                       return nt_status;
+               status = dsdb_trust_verify_forest_info(tdo, lfti,
+                                               LSA_FOREST_TRUST_COLLISION_TDO,
+                                               c_info, step1_lfti);
+               if (!NT_STATUS_IS_OK(status)) {
+                       goto done;
                }
-       }
 
-       if (c_info->count != 0) {
-               *r->out.collision_info = c_info;
+               TALLOC_FREE(tdo);
        }
 
        if (r->in.check_only != 0) {
-               return NT_STATUS_OK;
+               status = NT_STATUS_OK;
+               goto done;
        }
 
-       /* not just a check, write info back */
+       /*
+        * not just a check, write info back
+        */
 
-       ndr_err = ndr_push_struct_blob(&ft_blob, mem_ctx, nfti,
+       /*
+        * normalize the given forest trust info.
+        *
+        * Step2: adds TOP_LEVEL_NAME[_EX] in reverse order,
+        * followed by DOMAIN_INFO in reverse order. It also removes
+        * possible NULL entries from Step1.
+        */
+       status = dsdb_trust_normalize_forest_info_step2(mem_ctx, step1_lfti,
+                                                       &step2_lfti);
+       if (!NT_STATUS_IS_OK(status)) {
+               goto done;
+       }
+
+       status = dsdb_trust_forest_info_from_lsa(mem_ctx, step2_lfti,
+                                                &trust_fti);
+       if (!NT_STATUS_IS_OK(status)) {
+               goto done;
+       }
+
+       ndr_err = ndr_push_struct_blob(&ft_blob, mem_ctx, trust_fti,
                                       (ndr_push_flags_fn_t)ndr_push_ForestTrustInfo);
        if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
-               return NT_STATUS_INVALID_PARAMETER;
+               status = NT_STATUS_INVALID_PARAMETER;
+               goto done;
        }
 
        msg = ldb_msg_new(mem_ctx);
        if (msg == NULL) {
-               return NT_STATUS_NO_MEMORY;
+               status = NT_STATUS_NO_MEMORY;
+               goto done;
        }
 
-       msg->dn = ldb_dn_copy(mem_ctx, tdo_dn);
+       msg->dn = ldb_dn_copy(mem_ctx, trust_tdo_msg->dn);
        if (!msg->dn) {
-               return NT_STATUS_NO_MEMORY;
+               status = NT_STATUS_NO_MEMORY;
+               goto done;
        }
 
        ret = ldb_msg_add_empty(msg, "msDS-TrustForestTrustInfo",
                                LDB_FLAG_MOD_REPLACE, NULL);
        if (ret != LDB_SUCCESS) {
-               return NT_STATUS_NO_MEMORY;
+               status = NT_STATUS_NO_MEMORY;
+               goto done;
        }
        ret = ldb_msg_add_value(msg, "msDS-TrustForestTrustInfo",
                                &ft_blob, NULL);
        if (ret != LDB_SUCCESS) {
-               return NT_STATUS_NO_MEMORY;
+               status = NT_STATUS_NO_MEMORY;
+               goto done;
        }
 
        ret = ldb_modify(p_state->sam_ldb, msg);
        if (ret != LDB_SUCCESS) {
+               status = dsdb_ldb_err_to_ntstatus(ret);
+
                DEBUG(0, ("Failed to store Forest Trust Info: %s\n",
                          ldb_errstring(p_state->sam_ldb)));
 
-               switch (ret) {
-               case LDB_ERR_INSUFFICIENT_ACCESS_RIGHTS:
-                       return NT_STATUS_ACCESS_DENIED;
-               default:
-                       return NT_STATUS_INTERNAL_DB_CORRUPTION;
-               }
+               goto done;
        }
 
-       return NT_STATUS_OK;
+       /* ok, all fine, commit transaction and return */
+       in_transaction = false;
+       ret = ldb_transaction_commit(p_state->sam_ldb);
+       if (ret != LDB_SUCCESS) {
+               status = NT_STATUS_INTERNAL_DB_CORRUPTION;
+               goto done;
+       }
+
+       status = NT_STATUS_OK;
+
+done:
+       if (NT_STATUS_IS_OK(status) && c_info->count != 0) {
+               *r->out.collision_info = c_info;
+       }
+
+       if (in_transaction) {
+               ldb_transaction_cancel(p_state->sam_ldb);
+       }
+
+       return status;
 }
 
 /*