From: Stefan Metzmacher <metze@samba.org>
Date: Tue, 5 Jun 2018 01:36:39 +0000 (+0200)
Subject: libcli/lsarpc: add trust_forest_info_tln[_ex]_match()
X-Git-Tag: tevent-0.17.0~712
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=6f6aa8d661333add059760c52e530a47bf656b6f;p=thirdparty%2Fsamba.git

libcli/lsarpc: add trust_forest_info_tln[_ex]_match()

These are copies of dsdb_trust_find_tln[_ex]_match()
in source4/dsdb/common/util_trusts.c, which gets replaced
in the next commits.

Signed-off-by: Stefan Metzmacher <metze@samba.org>
Reviewed-by: Ralph Boehme <slow@samba.org>
---

diff --git a/libcli/lsarpc/util_lsarpc.c b/libcli/lsarpc/util_lsarpc.c
index 018bbcc0eec..05126304758 100644
--- a/libcli/lsarpc/util_lsarpc.c
+++ b/libcli/lsarpc/util_lsarpc.c
@@ -18,6 +18,7 @@
 */
 
 #include "includes.h"
+#include "lib/util/dns_cmp.h"
 #include "../librpc/gen_ndr/ndr_drsblobs.h"
 #include "../librpc/gen_ndr/ndr_lsa.h"
 #include "libcli/lsarpc/util_lsarpc.h"
@@ -587,3 +588,88 @@ NTSTATUS trust_forest_info_to_lsa(TALLOC_CTX *mem_ctx,
 	*_lfti = lfti;
 	return NT_STATUS_OK;
 }
+
+static int trust_forest_info_tln_match_internal(
+		const struct lsa_ForestTrustInformation *info,
+		enum lsa_ForestTrustRecordType type,
+		uint32_t disable_mask,
+		const char *tln)
+{
+	uint32_t i;
+
+	for (i = 0; i < info->count; i++) {
+		struct lsa_ForestTrustRecord *e = info->entries[i];
+		struct lsa_StringLarge *t = NULL;
+		int cmp;
+
+		if (e == NULL) {
+			continue;
+		}
+
+		if (e->type != type) {
+			continue;
+		}
+
+		if (e->flags & disable_mask) {
+			continue;
+		}
+
+		switch (type) {
+		case LSA_FOREST_TRUST_TOP_LEVEL_NAME:
+			t = &e->forest_trust_data.top_level_name;
+			break;
+		case LSA_FOREST_TRUST_TOP_LEVEL_NAME_EX:
+			t = &e->forest_trust_data.top_level_name_ex;
+			break;
+		default:
+			break;
+		}
+
+		if (t == NULL) {
+			continue;
+		}
+
+		cmp = dns_cmp(tln, t->string);
+		switch (cmp) {
+		case DNS_CMP_MATCH:
+		case DNS_CMP_FIRST_IS_CHILD:
+			return i;
+		}
+	}
+
+	return -1;
+}
+
+bool trust_forest_info_tln_match(
+		const struct lsa_ForestTrustInformation *info,
+		const char *tln)
+{
+	int m;
+
+	m = trust_forest_info_tln_match_internal(info,
+					LSA_FOREST_TRUST_TOP_LEVEL_NAME,
+					LSA_TLN_DISABLED_MASK,
+					tln);
+	if (m != -1) {
+		return true;
+	}
+
+	return false;
+}
+
+bool trust_forest_info_tln_ex_match(
+		const struct lsa_ForestTrustInformation *info,
+		const char *tln)
+{
+	int m;
+
+	m = trust_forest_info_tln_match_internal(info,
+					LSA_FOREST_TRUST_TOP_LEVEL_NAME_EX,
+					0,
+					tln);
+	if (m != -1) {
+		return true;
+	}
+
+	return false;
+}
diff --git a/libcli/lsarpc/util_lsarpc.h b/libcli/lsarpc/util_lsarpc.h
index 6795c148357..b12ab4f4292 100644
--- a/libcli/lsarpc/util_lsarpc.h
+++ b/libcli/lsarpc/util_lsarpc.h
@@ -46,4 +46,10 @@ NTSTATUS trust_forest_info_to_lsa(TALLOC_CTX *mem_ctx,
 				  const struct ForestTrustInfo *fti,
 				  struct lsa_ForestTrustInformation **_lfti);
 
+bool trust_forest_info_tln_match(
+		const struct lsa_ForestTrustInformation *info,
+		const char *tln);
+bool trust_forest_info_tln_ex_match(
+		const struct lsa_ForestTrustInformation *info,
+		const char *tln);
 #endif /* _LIBCLI_AUTH_UTIL_LSARPC_H_ */