s4:libcli/wrepl: convert wrepl_connect_send to tevent_req

metze

1 files changed, 95 insertions, 65 deletions

diff --git a/source4/libcli/wrepl/winsrepl.c b/source4/libcli/wrepl/winsrepl.c
index b3b1381601..907a7b9709 100644
--- a/source4/libcli/wrepl/winsrepl.c
+++ b/source4/libcli/wrepl/winsrepl.c
@@ -267,34 +267,89 @@ static NTSTATUS wrepl_request_wait(struct wrepl_request *req)
 	return req->status;
 }
 
+const char *wrepl_best_ip(struct loadparm_context *lp_ctx, const char *peer_ip)
+{
+	struct interface *ifaces;
+	load_interfaces(lp_ctx, lp_interfaces(lp_ctx), &ifaces);
+	return iface_best_ip(ifaces, peer_ip);
+}
+
 struct wrepl_connect_state {
-	struct composite_context *result;
 	struct wrepl_socket *wrepl_socket;
-	struct composite_context *creq;
 };
 
-/*
-  handler for winrepl connection completion
-*/
-static void wrepl_connect_handler(struct composite_context *creq)
+static void wrepl_connect_handler(struct composite_context *creq);
+
+struct tevent_req *wrepl_connect_send(TALLOC_CTX *mem_ctx,
+				      struct tevent_context *ev,
+				      struct wrepl_socket *wrepl_socket,
+				      const char *our_ip, const char *peer_ip)
+{
+	struct tevent_req *req;
+	struct wrepl_connect_state *state;
+	struct composite_context *subreq;
+	struct socket_address *peer, *us;
+
+	req = tevent_req_create(mem_ctx, &state,
+				struct wrepl_connect_state);
+	if (req == NULL) {
+		return NULL;
+	}
+
+	state->wrepl_socket	= wrepl_socket;
+
+	us = socket_address_from_strings(state, wrepl_socket->sock->backend_name,
+					 our_ip, 0);
+	if (tevent_req_nomem(us, req)) {
+		return tevent_req_post(req, ev);
+	}
+
+	peer = socket_address_from_strings(state, wrepl_socket->sock->backend_name,
+					   peer_ip, WINS_REPLICATION_PORT);
+	if (tevent_req_nomem(peer, req)) {
+		return tevent_req_post(req, ev);
+	}
+
+	subreq = socket_connect_send(wrepl_socket->sock, us, peer,
+				     0, wrepl_socket->event.ctx);
+	if (tevent_req_nomem(subreq, req)) {
+		return tevent_req_post(req, ev);
+	}
+
+	subreq->async.fn = wrepl_connect_handler;
+	subreq->async.private_data = req;
+
+	return req;
+}
+
+static void wrepl_connect_handler(struct composite_context *subreq)
 {
-	struct wrepl_connect_state *state = talloc_get_type(creq->async.private_data, 
+	struct tevent_req *req = talloc_get_type_abort(subreq->async.private_data,
+				 struct tevent_req);
+	struct wrepl_connect_state *state = tevent_req_data(req,
 					    struct wrepl_connect_state);
 	struct wrepl_socket *wrepl_socket = state->wrepl_socket;
-	struct composite_context *result = state->result;
+	NTSTATUS status;
 
-	result->status = socket_connect_recv(state->creq);
-	if (!composite_is_ok(result)) return;
+	status = socket_connect_recv(subreq);
+	if (!NT_STATUS_IS_OK(status)) {
+		tevent_req_nterror(req, status);
+		return;
+	}
 
 	wrepl_socket->event.fde = event_add_fd(wrepl_socket->event.ctx, wrepl_socket, 
 					       socket_get_fd(wrepl_socket->sock), 
 					       EVENT_FD_READ,
 					       wrepl_handler, wrepl_socket);
-	if (composite_nomem(wrepl_socket->event.fde, result)) return;
+	if (tevent_req_nomem(wrepl_socket->event.fde, req)) {
+		return;
+	}
 
 	/* setup the stream -> packet parser */
 	wrepl_socket->packet = packet_init(wrepl_socket);
-	if (composite_nomem(wrepl_socket->packet, result)) return;
+	if (tevent_req_nomem(wrepl_socket->packet, req)) {
+		return;
+	}
 	packet_set_private(wrepl_socket->packet, wrepl_socket);
 	packet_set_socket(wrepl_socket->packet, wrepl_socket->sock);
 	packet_set_callback(wrepl_socket->packet, wrepl_finish_recv);
@@ -304,69 +359,27 @@ static void wrepl_connect_handler(struct composite_context *creq)
 	packet_set_fde(wrepl_socket->packet, wrepl_socket->event.fde);
 	packet_set_serialise(wrepl_socket->packet);
 
-	composite_done(result);
-}
-
-const char *wrepl_best_ip(struct loadparm_context *lp_ctx, const char *peer_ip)
-{
-	struct interface *ifaces;
-	load_interfaces(lp_ctx, lp_interfaces(lp_ctx), &ifaces);
-	return iface_best_ip(ifaces, peer_ip);
-}
-
-
-/*
-  connect a wrepl_socket to a WINS server
-*/
-struct composite_context *wrepl_connect_send(struct wrepl_socket *wrepl_socket,
-					     const char *our_ip, const char *peer_ip)
-{
-	struct composite_context *result;
-	struct wrepl_connect_state *state;
-	struct socket_address *peer, *us;
-
-	result = talloc_zero(wrepl_socket, struct composite_context);
-	if (!result) return NULL;
-
-	result->state		= COMPOSITE_STATE_IN_PROGRESS;
-	result->event_ctx	= wrepl_socket->event.ctx;
-
-	state = talloc_zero(result, struct wrepl_connect_state);
-	if (composite_nomem(state, result)) return result;
-	result->private_data	= state;
-	state->result		= result;
-	state->wrepl_socket	= wrepl_socket;
-
-	us = socket_address_from_strings(state, wrepl_socket->sock->backend_name, 
-					 our_ip, 0);
-	if (composite_nomem(us, result)) return result;
-
-	peer = socket_address_from_strings(state, wrepl_socket->sock->backend_name, 
-					   peer_ip, WINS_REPLICATION_PORT);
-	if (composite_nomem(peer, result)) return result;
-
-	state->creq = socket_connect_send(wrepl_socket->sock, us, peer,
-					  0, wrepl_socket->event.ctx);
-	composite_continue(result, state->creq, wrepl_connect_handler, state);
-	return result;
+	tevent_req_done(req);
 }
 
 /*
   connect a wrepl_socket to a WINS server - recv side
 */
-NTSTATUS wrepl_connect_recv(struct composite_context *result)
+NTSTATUS wrepl_connect_recv(struct tevent_req *req)
 {
-	struct wrepl_connect_state *state = talloc_get_type(result->private_data,
+	struct wrepl_connect_state *state = tevent_req_data(req,
 					    struct wrepl_connect_state);
 	struct wrepl_socket *wrepl_socket = state->wrepl_socket;
-	NTSTATUS status = composite_wait(result);
+	NTSTATUS status;
 
-	if (!NT_STATUS_IS_OK(status)) {
+	if (tevent_req_is_nterror(req, &status)) {
 		wrepl_socket_dead(wrepl_socket, status);
+		tevent_req_received(req);
+		return status;
 	}
 
-	talloc_free(result);
-	return status;
+	tevent_req_received(req);
+	return NT_STATUS_OK;
 }
 
 /*
@@ -375,8 +388,25 @@ NTSTATUS wrepl_connect_recv(struct composite_context *result)
 NTSTATUS wrepl_connect(struct wrepl_socket *wrepl_socket,
 		       const char *our_ip, const char *peer_ip)
 {
-	struct composite_context *c_req = wrepl_connect_send(wrepl_socket, our_ip, peer_ip);
-	return wrepl_connect_recv(c_req);
+	struct tevent_req *subreq;
+	bool ok;
+	NTSTATUS status;
+
+	subreq = wrepl_connect_send(wrepl_socket, wrepl_socket->event.ctx,
+				    wrepl_socket, our_ip, peer_ip);
+	NT_STATUS_HAVE_NO_MEMORY(subreq);
+
+	ok = tevent_req_poll(subreq, wrepl_socket->event.ctx);
+	if (!ok) {
+		TALLOC_FREE(subreq);
+		return NT_STATUS_INTERNAL_ERROR;
+	}
+
+	status = wrepl_connect_recv(subreq);
+	TALLOC_FREE(subreq);
+	NT_STATUS_NOT_OK_RETURN(status);
+
+	return NT_STATUS_OK;
 }
 
 /*