r17318: make better usage of the composite api

metze
(This used to be commit 683fc25f6524a3821ba70529251aabe97bad9370)

1 files changed, 9 insertions, 33 deletions

diff --git a/source4/libcli/smb2/session.c b/source4/libcli/smb2/session.c
index 7518a3d6d3..8ebdc93bd4 100644
--- a/source4/libcli/smb2/session.c
+++ b/source4/libcli/smb2/session.c
@@ -196,18 +196,12 @@ struct composite_context *smb2_session_setup_spnego_send(struct smb2_session *se
 	struct composite_context *c;
 	struct smb2_session_state *state;
 
-	c = talloc_zero(session, struct composite_context);
+	c = composite_create(session, session->transport->socket->event.ctx);
 	if (c == NULL) return NULL;
 
 	state = talloc(c, struct smb2_session_state);
-	if (state == NULL) {
-		c->status = NT_STATUS_NO_MEMORY;
-		goto failed;
-	}
-
-	c->state = COMPOSITE_STATE_IN_PROGRESS;
+	if (composite_nomem(state, c)) return c;
 	c->private_data = state;
-	c->event_ctx = session->transport->socket->event.ctx;
 
 	ZERO_STRUCT(state->io);
 	state->io.in._pad = 0x0000;
@@ -216,47 +210,29 @@ struct composite_context *smb2_session_setup_spnego_send(struct smb2_session *se
 	state->io.in.unknown4 = 0; /* uint64_t */
 
 	c->status = gensec_set_credentials(session->gensec, credentials);
-	if (!NT_STATUS_IS_OK(c->status)) {
-		goto failed;
-	}
+	if (!composite_is_ok(c)) return c;
 
 	c->status = gensec_set_target_hostname(session->gensec, 
 					       session->transport->socket->hostname);
-	if (!NT_STATUS_IS_OK(c->status)) {
-		goto failed;
-	}
+	if (!composite_is_ok(c)) return c;
 
 	c->status = gensec_set_target_service(session->gensec, "cifs");
-	if (!NT_STATUS_IS_OK(c->status)) {
-		goto failed;
-	}
+	if (!composite_is_ok(c)) return c;
 
 	c->status = gensec_start_mech_by_oid(session->gensec, GENSEC_OID_SPNEGO);
-	if (!NT_STATUS_IS_OK(c->status)) {
-		goto failed;
-	}
+	if (!composite_is_ok(c)) return c;
 
 	c->status = gensec_update(session->gensec, c, 
 				  session->transport->negotiate.secblob,
 				  &state->io.in.secblob);
 	if (!NT_STATUS_EQUAL(c->status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
-		goto failed;
+		composite_error(c, c->status);
+		return c;
 	}
 	state->gensec_status = c->status;
 		
 	state->req = smb2_session_setup_send(session, &state->io);
-	if (state->req == NULL) {
-		c->status = NT_STATUS_NO_MEMORY;
-		goto failed;
-	}
-
-	state->req->async.fn = session_request_handler;
-	state->req->async.private = c;
-
-	return c;
-
-failed:
-	composite_error(c, c->status);
+	composite_continue_smb2(c, state->req, session_request_handler, c);
 	return c;
 }