s3-rpc_server: Add tevent based endpoint setup functions.

1 files changed, 121 insertions, 0 deletions

diff --git a/source3/rpc_server/rpc_ep_setup.c b/source3/rpc_server/rpc_ep_setup.c
index 32afa1d7bd..dd4c4c8053 100644
--- a/source3/rpc_server/rpc_ep_setup.c
+++ b/source3/rpc_server/rpc_ep_setup.c
@@ -176,6 +176,127 @@ static NTSTATUS _rpc_ep_unregister(const struct ndr_interface_table *iface)
 	return status;
 }
 
+static void rpc_ep_setup_register_loop(struct tevent_req *subreq);
+static NTSTATUS rpc_ep_setup_try_register(struct tevent_context *ev_ctx,
+					  struct messaging_context *msg_ctx,
+					  const struct ndr_interface_table *iface,
+					  const char *name,
+					  uint16_t port);
+
+struct rpc_ep_regsiter_state {
+	uint32_t wait_time;
+
+	struct tevent_context *ev_ctx;
+	struct messaging_context *msg_ctx;
+
+	const struct ndr_interface_table *iface;
+
+	const char *ncalrpc;
+	uint16_t port;
+};
+
+static NTSTATUS rpc_ep_setup_register(struct tevent_context *ev_ctx,
+				      struct messaging_context *msg_ctx,
+				      const struct ndr_interface_table *iface,
+				      const char *ncalrpc,
+				      uint16_t port)
+{
+	struct rpc_ep_regsiter_state *state;
+	struct tevent_req *req;
+
+	state = talloc(ev_ctx, struct rpc_ep_regsiter_state);
+	if (state == NULL) {
+		return NT_STATUS_NO_MEMORY;
+	}
+
+	state->wait_time = 1;
+	state->ev_ctx = ev_ctx;
+	state->msg_ctx = msg_ctx;
+	state->iface = iface;
+	state->ncalrpc = talloc_strdup(state, ncalrpc);
+	state->port = port;
+
+	req = tevent_wakeup_send(state, ev_ctx, timeval_current_ofs(1, 0));
+	if (tevent_req_nomem(state, req)) {
+		talloc_free(state);
+		return NT_STATUS_NO_MEMORY;
+	}
+
+	tevent_req_set_callback(req, rpc_ep_setup_register_loop, state);
+
+	return NT_STATUS_OK;
+}
+
+static void rpc_ep_setup_register_loop(struct tevent_req *req)
+{
+	struct rpc_ep_regsiter_state *state =
+		tevent_req_callback_data(req, struct rpc_ep_regsiter_state);
+	NTSTATUS status;
+	bool ok;
+
+	ok = tevent_wakeup_recv(req);
+	TALLOC_FREE(req);
+	if (!ok) {
+		return;
+	}
+
+	status = rpc_ep_setup_try_register(state->ev_ctx,
+					   state->msg_ctx,
+					   state->iface,
+					   state->ncalrpc,
+					   state->port);
+	if (NT_STATUS_IS_OK(status)) {
+		talloc_free(state);
+		return;
+	}
+
+	state->wait_time = state->wait_time * 2;
+	if (state->wait_time > 16) {
+		talloc_free(state);
+		return;
+	}
+
+	req = tevent_wakeup_send(state,
+				 state->ev_ctx,
+				 timeval_current_ofs(state->wait_time, 0));
+	if (tevent_req_nomem(state, req)) {
+		talloc_free(state);
+		return;
+	}
+
+	tevent_req_set_callback(req, rpc_ep_setup_register_loop, state);
+	return;
+}
+
+static NTSTATUS rpc_ep_setup_try_register(struct tevent_context *ev_ctx,
+					  struct messaging_context *msg_ctx,
+					  const struct ndr_interface_table *iface,
+					  const char *name,
+					  uint16_t port)
+{
+	struct dcerpc_binding_vector *v = NULL;
+	NTSTATUS status;
+
+	status = dcerpc_binding_vector_create(talloc_tos(),
+					      iface,
+					      port,
+					      name,
+					      &v);
+	if (!NT_STATUS_IS_OK(status)) {
+		return status;
+	}
+
+	status = dcerpc_ep_register(iface,
+				    v,
+				    &iface->syntax_id.uuid,
+				    name);
+	if (!NT_STATUS_IS_OK(status)) {
+		return status;
+	}
+
+	return status;
+}
+
 static bool epmapper_init_cb(void *ptr)
 {
 	struct dcesrv_ep_context *ep_ctx =