s4/net_drs: 'net drs kcc' command implementation

Signed-off-by: Andrew Tridgell <tridge@samba.org>

3 files changed, 177 insertions, 2 deletions

diff --git a/source4/utils/net/config.mk b/source4/utils/net/config.mk
index 70c6bafd0a..e3dfb35ce0 100644
--- a/source4/utils/net/config.mk
+++ b/source4/utils/net/config.mk
@@ -14,7 +14,8 @@ PRIVATE_DEPENDENCIES = \
 
 net_drs_OBJ_FILES = $(addprefix $(utilssrcdir)/net/drs/,  \
 		net_drs.o \
-		net_drs_bind.o)
+		net_drs_bind.o \
+		net_drs_kcc.o)
 
 $(eval $(call proto_header_template,$(utilssrcdir)/net/drs/net_drs_proto.h,$(net_drs_OBJ_FILES:.o=.c)))
 
diff --git a/source4/utils/net/drs/net_drs.c b/source4/utils/net/drs/net_drs.c
index 547d642bfb..7772d83ace 100644
--- a/source4/utils/net/drs/net_drs.c
+++ b/source4/utils/net/drs/net_drs.c
@@ -34,6 +34,8 @@
  */
 static const struct net_functable net_drs_functable[] = {
 	{ "bind", "Display replication features for a domain controller\n", net_drs_bind_cmd, net_drs_bind_usage },
+	{ "kcc", "Forces the KCC to recalculate replication topology for a specified domain controller\n",
+		  net_drs_kcc_cmd, net_drs_kcc_usage },
 	{ NULL, NULL }
 };
 
@@ -53,7 +55,8 @@ int net_drs_usage(struct net_context *ctx, int argc, const char **argv)
 	d_printf("net drs <command> [options]\n");
 	d_printf("\n");
 	d_printf("Currently implemented commands:\n");
-	d_printf("  bind - Display DC replication features");
+	d_printf("  bind - Display DC replication features\n");
+	d_printf("  kcc - Forces the KCC to recalculate replication topology for a specified domain controller\n");
 	return 0;
 }
 
diff --git a/source4/utils/net/drs/net_drs_kcc.c b/source4/utils/net/drs/net_drs_kcc.c
new file mode 100644
index 0000000000..fb2c661d9b
--- /dev/null
+++ b/source4/utils/net/drs/net_drs_kcc.c
@@ -0,0 +1,171 @@
+/*
+   Unix SMB/CIFS implementation.
+
+   Implements functions offered by repadmin.exe tool under Windows
+
+   Copyright (C) Kamen Mazdrashki <kamen.mazdrashki@postpath.com> 2010
+
+   This program is free software; you can redistribute it and/or modify
+   it under the terms of the GNU General Public License as published by
+   the Free Software Foundation; either version 3 of the License, or
+   (at your option) any later version.
+
+   This program is distributed in the hope that it will be useful,
+   but WITHOUT ANY WARRANTY; without even the implied warranty of
+   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+   GNU General Public License for more details.
+
+   You should have received a copy of the GNU General Public License
+   along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#include "includes.h"
+#include "utils/net/net.h"
+#include "net_drs.h"
+#include "lib/ldb/include/ldb.h"
+
+
+static bool net_drs_kcc_site_info(struct net_drs_context *drs_ctx,
+				  const char **site_name,
+				  uint32_t *site_options)
+{
+	struct ldb_dn *dn;
+	const struct ldb_val *ldb_val;
+	TALLOC_CTX *mem_ctx;
+	int ret;
+	struct ldb_result *ldb_res;
+	static const char *attrs[] = {
+		"options",
+		"whenChanged",
+		NULL
+	};
+
+	mem_ctx = talloc_new(drs_ctx);
+
+	/* get dsServiceName, which is NTDS Settings
+	 * object for the server
+	 * e.g.: CN=NTDS Settings,CN=<DC_NAME>,CN=Servers,CN=<SITE_NAME>,CN=Sites,<CONFIG> */
+	dn = ldb_msg_find_attr_as_dn(drs_ctx->ldap.ldb, mem_ctx, drs_ctx->ldap.rootdse, "dsServiceName");
+	if (!dn) {
+		d_printf("No dsServiceName value in RootDSE!\n");
+		goto failed;
+	}
+
+	/* work out site name */
+	if (!ldb_dn_remove_child_components(dn, 3)) {
+		d_printf("Failed to find Site DN.\n");
+		goto failed;
+	}
+
+	ldb_val = ldb_dn_get_rdn_val(dn);
+	if (!ldb_val) {
+		d_printf("Failed to get site name.\n");
+		goto failed;
+	}
+	*site_name = talloc_strndup(drs_ctx, (const char*)ldb_val->data, ldb_val->length);
+
+	/* get 'NTDS Site Settings' */
+	if (!ldb_dn_add_child_fmt(dn, "CN=NTDS Site Settings")) {
+		d_printf("Failed to create NTDS Site Settings DN.\n");
+		goto failed;
+	}
+
+	ret = ldb_search(drs_ctx->ldap.ldb, mem_ctx, &ldb_res,
+	                     dn, LDB_SCOPE_BASE,  attrs, "(objectClass=*)");
+	if (ret != LDB_SUCCESS) {
+		d_printf("Failed to get Site object\n");
+		goto failed;
+	}
+	if (ldb_res->count != 1) {
+		d_printf("Error: returned %d messages for Site object.\n", ldb_res->count);
+		goto failed;
+	}
+	*site_options = ldb_msg_find_attr_as_uint(ldb_res->msgs[0], "options", 0);
+
+	talloc_free(mem_ctx);
+	return true;
+
+failed:
+	talloc_free(mem_ctx);
+	return false;
+}
+
+/**
+ * 'net drs kcc' command entry point
+ */
+int net_drs_kcc_cmd(struct net_context *ctx, int argc, const char **argv)
+{
+	NTSTATUS status;
+	struct net_drs_context *drs_ctx;
+	struct drsuapi_DsBindInfo48 *info48;
+	struct drsuapi_DsExecuteKCC req;
+	union drsuapi_DsExecuteKCCRequest kcc_req;
+	const char *site_name;
+	uint32_t site_options;
+
+	/* only one arg expected */
+	if (argc != 1) {
+		return net_drs_kcc_usage(ctx, argc, argv);
+	}
+
+	if (!net_drs_create_context(ctx, argv[0], &drs_ctx)) {
+		return -1;
+	}
+	info48 = &drs_ctx->info48;
+
+	/* check if target DC supports ExecuteKCC */
+	if (!(info48->supported_extensions & DRSUAPI_SUPPORTED_EXTENSION_KCC_EXECUTE)) {
+		d_printf("%s does not support EXECUTE_KCC extension.\n", drs_ctx->dc_name);
+		goto failed;
+	}
+
+	/* gather some Site info */
+	if (!net_drs_kcc_site_info(drs_ctx, &site_name, &site_options)) {
+		goto failed;
+	}
+
+	d_printf("%s\n", site_name);
+	if (site_options) {
+		// TODO: print meaningfull site options here
+		d_printf("Current Site Options: 0x%X\n", site_options);
+	} else {
+		d_printf("Current Site Options: (none)\n");
+	}
+
+	/* execute KCC */
+	ZERO_STRUCT(req);
+	ZERO_STRUCT(kcc_req);
+	req.in.bind_handle = &drs_ctx->bind_handle;
+	req.in.level = 1;
+	req.in.req = &kcc_req;
+	status = dcerpc_drsuapi_DsExecuteKCC(drs_ctx->drs_pipe, drs_ctx, &req);
+	if (!NT_STATUS_IS_OK(status)) {
+		const char *errstr = nt_errstr(status);
+		if (NT_STATUS_EQUAL(status, NT_STATUS_NET_WRITE_FAULT)) {
+			errstr = dcerpc_errstr(drs_ctx, drs_ctx->drs_pipe->last_fault_code);
+		}
+		d_printf("dcerpc_drsuapi_DsExecuteKCC failed - %s.\n", errstr);
+		goto failed;
+	} else if (!W_ERROR_IS_OK(req.out.result)) {
+		d_printf("DsExecuteKCC failed - %s.\n", win_errstr(req.out.result));
+		goto failed;
+	}
+
+	d_printf("Consistency check on %s successful.\n", drs_ctx->dc_name);
+
+	talloc_free(drs_ctx);
+	return 0;
+
+failed:
+	talloc_free(drs_ctx);
+	return -1;
+}
+
+/**
+ * 'net drs kcc' usage
+ */
+int net_drs_kcc_usage(struct net_context *ctx, int argc, const char **argv)
+{
+	d_printf("net drs kcc <DC_NAME>\n");
+	return 0;
+}