s4/drs_util: Add public function for binding to a DC

diff --git a/source4/utils/net/drs/net_drs.c b/source4/utils/net/drs/net_drs.c
index 6e103aa417f373c0fe2e4a5a46a53a353e8a7d2c..87ebe0890af310d9da10090009d7232e272934b6 100644 (file)
--- a/source4/utils/net/drs/net_drs.c
+++ b/source4/utils/net/drs/net_drs.c
@@ -137,15 +137,63 @@ static bool net_drs_DsUnbind(struct net_drs_connection *conn)
        struct drsuapi_DsUnbind r;
        struct policy_handle bind_handle;
 
+       SMB_ASSERT(conn->drs_pipe);
+
        ZERO_STRUCT(r);
        r.out.bind_handle = &bind_handle;
 
        r.in.bind_handle = &conn->bind_handle;
        dcerpc_drsuapi_DsUnbind(conn->drs_pipe, conn, &r);
 
+       /* free dcerpc pipe in case we get called more than once */
+       talloc_free(conn->drs_pipe);
+       conn->drs_pipe = NULL;
+
        return true;
 }
 
+/**
+ * Destroy drsuapi connection
+ */
+static int net_drs_connection_destructor(struct net_drs_connection *conn)
+{
+       if (conn->drs_pipe) {
+               net_drs_DsUnbind(conn);
+       }
+       return 0;
+}
+
+/**
+ * Create DRSUAPI connection to target DC
+ * @return ptr to net_drs_connection or NULL on failure
+ */
+struct net_drs_connection * net_drs_connect_dc(struct net_drs_context *drs_ctx, const char *dc_name)
+{
+       struct net_drs_connection *conn = NULL;
+
+       conn = talloc_zero(drs_ctx, struct net_drs_connection);
+       NET_DRS_NOMEM_GOTO(conn, failed);
+
+       /* init binding */
+       conn->binding = talloc_zero(conn, struct dcerpc_binding);
+       conn->binding->transport = NCACN_IP_TCP;
+       conn->binding->flags = drs_ctx->drs_conn->binding->flags;
+       conn->binding->host = talloc_strdup(conn, dc_name);
+       conn->binding->target_hostname = conn->binding->host;
+
+       if (!net_drs_DsBind(drs_ctx, conn)) {
+               goto failed;
+       }
+
+       talloc_set_destructor(conn, net_drs_connection_destructor);
+
+       return conn;
+
+failed:
+       talloc_free(conn);
+       return NULL;
+}
+
 /**
  * Open secured LDAP connection to remote DC
  */