s4:librpc: fix netlogon connections against servers without AES support
[metze/samba/wip.git] / source4 / librpc / rpc / dcerpc_schannel.c
index 7cd2d1654f19d75b4778d349ce7478a6e019ebc4..130ebebd9092aa2a5161a9b0b4f6426969160723 100644 (file)
@@ -4,10 +4,12 @@
    dcerpc schannel operations
 
    Copyright (C) Andrew Tridgell 2004
-   
+   Copyright (C) Andrew Bartlett <abartlet@samba.org> 2004-2005
+   Copyright (C) Rafal Szczesniak 2006
+
    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 2 of the License, or
+   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,
    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, write to the Free Software
-   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+   along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
 #include "includes.h"
-#include "librpc/gen_ndr/ndr_schannel.h"
+#include <tevent.h>
 #include "auth/auth.h"
-
-enum schannel_position {
-       DCERPC_SCHANNEL_STATE_START = 0,
-       DCERPC_SCHANNEL_STATE_UPDATE_1
+#include "libcli/composite/composite.h"
+#include "libcli/auth/libcli_auth.h"
+#include "librpc/gen_ndr/ndr_netlogon.h"
+#include "librpc/gen_ndr/ndr_netlogon_c.h"
+#include "auth/credentials/credentials.h"
+#include "librpc/rpc/dcerpc_proto.h"
+#include "param/param.h"
+
+struct schannel_key_state {
+       struct dcerpc_pipe *pipe;
+       struct dcerpc_pipe *pipe2;
+       struct dcerpc_binding *binding;
+       bool dcerpc_schannel_auto;
+       struct cli_credentials *credentials;
+       struct netlogon_creds_CredentialState *creds;
+       uint32_t local_negotiate_flags;
+       uint32_t remote_negotiate_flags;
+       struct netr_Credential credentials1;
+       struct netr_Credential credentials2;
+       struct netr_Credential credentials3;
+       struct netr_ServerReqChallenge r;
+       struct netr_ServerAuthenticate2 a;
+       const struct samr_Password *mach_pwd;
 };
 
-struct dcerpc_schannel_state {
-       enum schannel_position state;
-       struct schannel_state *schannel_state;
-       struct creds_CredentialState *creds;
-       char *account_name;
-};
 
-static NTSTATUS dcerpc_schannel_key(struct dcerpc_pipe *p,
-                                   const char *domain,
-                                   const char *username,
-                                   const char *password,
-                                   int chan_type,
-                                   struct creds_CredentialState *creds);
+static void continue_secondary_connection(struct composite_context *ctx);
+static void continue_bind_auth_none(struct composite_context *ctx);
+static void continue_srv_challenge(struct tevent_req *subreq);
+static void continue_srv_auth2(struct tevent_req *subreq);
+static void continue_get_capabilities(struct tevent_req *subreq);
 
-/*
-  wrappers for the schannel_*() functions
 
-  These will become static again, when we get dynamic registration, and
-  decrpc_schannel_security_ops come back here.
+/*
+  Stage 2 of schannel_key: Receive endpoint mapping and request secondary
+  rpc connection
 */
-static NTSTATUS dcerpc_schannel_unseal_packet(struct gensec_security *gensec_security, 
-                                             TALLOC_CTX *mem_ctx, 
-                                             uint8_t *data, size_t length, 
-                                             const uint8_t *whole_pdu, size_t pdu_length, 
-                                             DATA_BLOB *sig)
+static void continue_epm_map_binding(struct composite_context *ctx)
 {
-       struct dcerpc_schannel_state *dce_schan_state = gensec_security->private_data;
-       
-       return schannel_unseal_packet(dce_schan_state->schannel_state, mem_ctx, data, length, sig);
-}
+       struct composite_context *c;
+       struct schannel_key_state *s;
+       struct composite_context *sec_conn_req;
+
+       c = talloc_get_type(ctx->async.private_data, struct composite_context);
+       s = talloc_get_type(c->private_data, struct schannel_key_state);
+
+       /* receive endpoint mapping */
+       c->status = dcerpc_epm_map_binding_recv(ctx);
+       if (!NT_STATUS_IS_OK(c->status)) {
+               DEBUG(0,("Failed to map DCERPC/TCP NCACN_NP pipe for '%s' - %s\n",
+                        NDR_NETLOGON_UUID, nt_errstr(c->status)));
+               composite_error(c, c->status);
+               return;
+       }
 
-static NTSTATUS dcerpc_schannel_check_packet(struct gensec_security *gensec_security, 
-                                            TALLOC_CTX *mem_ctx, 
-                                            const uint8_t *data, size_t length, 
-                                            const uint8_t *whole_pdu, size_t pdu_length, 
-                                            const DATA_BLOB *sig)
-{
-       struct dcerpc_schannel_state *dce_schan_state = gensec_security->private_data;
+       /* send a request for secondary rpc connection */
+       sec_conn_req = dcerpc_secondary_connection_send(s->pipe,
+                                                       s->binding);
+       if (composite_nomem(sec_conn_req, c)) return;
 
-       return schannel_check_packet(dce_schan_state->schannel_state, data, length, sig);
+       composite_continue(c, sec_conn_req, continue_secondary_connection, c);
 }
 
-static NTSTATUS dcerpc_schannel_seal_packet(struct gensec_security *gensec_security, 
-                                           TALLOC_CTX *mem_ctx, 
-                                           uint8_t *data, size_t length, 
-                                           const uint8_t *whole_pdu, size_t pdu_length, 
-                                           DATA_BLOB *sig)
+
+/*
+  Stage 3 of schannel_key: Receive secondary rpc connection and perform
+  non-authenticated bind request
+*/
+static void continue_secondary_connection(struct composite_context *ctx)
 {
-       struct dcerpc_schannel_state *dce_schan_state = gensec_security->private_data;
+       struct composite_context *c;
+       struct schannel_key_state *s;
+       struct composite_context *auth_none_req;
 
-       return schannel_seal_packet(dce_schan_state->schannel_state, mem_ctx, data, length, sig);
-}
+       c = talloc_get_type(ctx->async.private_data, struct composite_context);
+       s = talloc_get_type(c->private_data, struct schannel_key_state);
 
-static NTSTATUS dcerpc_schannel_sign_packet(struct gensec_security *gensec_security, 
-                                           TALLOC_CTX *mem_ctx, 
-                                           const uint8_t *data, size_t length, 
-                                           const uint8_t *whole_pdu, size_t pdu_length, 
-                                           DATA_BLOB *sig)
-{
-       struct dcerpc_schannel_state *dce_schan_state = gensec_security->private_data;
+       /* receive secondary rpc connection */
+       c->status = dcerpc_secondary_connection_recv(ctx, &s->pipe2);
+       if (!composite_is_ok(c)) return;
+
+       talloc_steal(s, s->pipe2);
 
-       return schannel_sign_packet(dce_schan_state->schannel_state, mem_ctx, data, length, sig);
+       /* initiate a non-authenticated bind */
+       auth_none_req = dcerpc_bind_auth_none_send(c, s->pipe2, &ndr_table_netlogon);
+       if (composite_nomem(auth_none_req, c)) return;
+
+       composite_continue(c, auth_none_req, continue_bind_auth_none, c);
 }
 
-static size_t dcerpc_schannel_sig_size(struct gensec_security *gensec_security)
+
+/*
+  Stage 4 of schannel_key: Receive non-authenticated bind and get
+  a netlogon challenge
+*/
+static void continue_bind_auth_none(struct composite_context *ctx)
 {
-       return 32;
+       struct composite_context *c;
+       struct schannel_key_state *s;
+       struct tevent_req *subreq;
+
+       c = talloc_get_type(ctx->async.private_data, struct composite_context);
+       s = talloc_get_type(c->private_data, struct schannel_key_state);
+
+       /* receive result of non-authenticated bind request */
+       c->status = dcerpc_bind_auth_none_recv(ctx);
+       if (!composite_is_ok(c)) return;
+       
+       /* prepare a challenge request */
+       s->r.in.server_name   = talloc_asprintf(c, "\\\\%s", dcerpc_server_name(s->pipe));
+       if (composite_nomem(s->r.in.server_name, c)) return;
+       s->r.in.computer_name = cli_credentials_get_workstation(s->credentials);
+       s->r.in.credentials   = &s->credentials1;
+       s->r.out.return_credentials  = &s->credentials2;
+       
+       generate_random_buffer(s->credentials1.data, sizeof(s->credentials1.data));
+
+       /*
+         request a netlogon challenge - a rpc request over opened secondary pipe
+       */
+       subreq = dcerpc_netr_ServerReqChallenge_r_send(s, c->event_ctx,
+                                                      s->pipe2->binding_handle,
+                                                      &s->r);
+       if (composite_nomem(subreq, c)) return;
+
+       tevent_req_set_callback(subreq, continue_srv_challenge, c);
 }
 
-static NTSTATUS dcerpc_schannel_session_key(struct gensec_security *gensec_security, 
-                                           DATA_BLOB *session_key)
+
+/*
+  Stage 5 of schannel_key: Receive a challenge and perform authentication
+  on the netlogon pipe
+*/
+static void continue_srv_challenge(struct tevent_req *subreq)
 {
-       return NT_STATUS_NOT_IMPLEMENTED;
+       struct composite_context *c;
+       struct schannel_key_state *s;
+
+       c = tevent_req_callback_data(subreq, struct composite_context);
+       s = talloc_get_type(c->private_data, struct schannel_key_state);
+
+       /* receive rpc request result - netlogon challenge */
+       c->status = dcerpc_netr_ServerReqChallenge_r_recv(subreq, s);
+       TALLOC_FREE(subreq);
+       if (!composite_is_ok(c)) return;
+
+       /* prepare credentials for auth2 request */
+       s->mach_pwd = cli_credentials_get_nt_hash(s->credentials, c);
+
+       /* auth2 request arguments */
+       s->a.in.server_name      = s->r.in.server_name;
+       s->a.in.account_name     = cli_credentials_get_username(s->credentials);
+       s->a.in.secure_channel_type =
+               cli_credentials_get_secure_channel_type(s->credentials);
+       s->a.in.computer_name    = cli_credentials_get_workstation(s->credentials);
+       s->a.in.negotiate_flags  = &s->local_negotiate_flags;
+       s->a.in.credentials      = &s->credentials3;
+       s->a.out.negotiate_flags = &s->remote_negotiate_flags;
+       s->a.out.return_credentials     = &s->credentials3;
+
+       s->creds = netlogon_creds_client_init(s, 
+                                             s->a.in.account_name, 
+                                             s->a.in.computer_name,
+                                             s->a.in.secure_channel_type,
+                                             &s->credentials1, &s->credentials2,
+                                             s->mach_pwd, &s->credentials3,
+                                             s->local_negotiate_flags);
+       if (composite_nomem(s->creds, c)) {
+               return;
+       }
+       /*
+         authenticate on the netlogon pipe - a rpc request over secondary pipe
+       */
+       subreq = dcerpc_netr_ServerAuthenticate2_r_send(s, c->event_ctx,
+                                                       s->pipe2->binding_handle,
+                                                       &s->a);
+       if (composite_nomem(subreq, c)) return;
+
+       tevent_req_set_callback(subreq, continue_srv_auth2, c);
 }
 
-static NTSTATUS dcerpc_schannel_update(struct gensec_security *gensec_security, TALLOC_CTX *out_mem_ctx, 
-                                      const DATA_BLOB in, DATA_BLOB *out) 
+
+/*
+  Stage 6 of schannel_key: Receive authentication request result and verify
+  received credentials
+*/
+static void continue_srv_auth2(struct tevent_req *subreq)
 {
-       struct dcerpc_schannel_state *dce_schan_state = gensec_security->private_data;
-       NTSTATUS status;
-       struct schannel_bind bind_schannel;
-       struct schannel_bind_ack bind_schannel_ack;
-       const char *account_name;
-       *out = data_blob(NULL, 0);
-
-       switch (gensec_security->gensec_role) {
-       case GENSEC_CLIENT:
-               if (dce_schan_state->state != DCERPC_SCHANNEL_STATE_START) {
-                       /* we could parse the bind ack, but we don't know what it is yet */
-                       return NT_STATUS_OK;
-               }
-               
-               status = schannel_start(&dce_schan_state->schannel_state, 
-                                       dce_schan_state->creds->session_key,
-                                       True);
-               if (!NT_STATUS_IS_OK(status)) {
-                       DEBUG(1, ("Failed to start schannel client\n"));
-                       return status;
-               }
-               talloc_steal(dce_schan_state, dce_schan_state->schannel_state);
-       
-               bind_schannel.unknown1 = 0;
-#if 0
-               /* to support this we'd need to have access to the full domain name */
-               bind_schannel.bind_type = 23;
-               bind_schannel.u.info23.domain = gensec_security->user.domain;
-               bind_schannel.u.info23.account_name = gensec_security->user.name;
-               bind_schannel.u.info23.dnsdomain = str_format_nbt_domain(out_mem_ctx, fulldomainname);
-               bind_schannel.u.info23.workstation = str_format_nbt_domain(out_mem_ctx, gensec_security->user.name);
-#else
-               bind_schannel.bind_type = 3;
-               bind_schannel.u.info3.domain = gensec_security->user.domain;
-               bind_schannel.u.info3.account_name = gensec_security->user.name;
-#endif
-               
-               status = ndr_push_struct_blob(out, out_mem_ctx, &bind_schannel,
-                                             (ndr_push_flags_fn_t)ndr_push_schannel_bind);
-               if (!NT_STATUS_IS_OK(status)) {
-                       DEBUG(3, ("Could not create schannel bind: %s\n",
-                                 nt_errstr(status)));
-                       return status;
-               }
-               
-               dce_schan_state->state = DCERPC_SCHANNEL_STATE_UPDATE_1;
-
-               return NT_STATUS_MORE_PROCESSING_REQUIRED;
-       case GENSEC_SERVER:
-               
-               if (dce_schan_state->state != DCERPC_SCHANNEL_STATE_START) {
-                       /* no third leg on this protocol */
-                       return NT_STATUS_INVALID_PARAMETER;
-               }
-               
-               /* parse the schannel startup blob */
-               status = ndr_pull_struct_blob(&in, out_mem_ctx, &bind_schannel, 
-                                             (ndr_pull_flags_fn_t)ndr_pull_schannel_bind);
-               if (!NT_STATUS_IS_OK(status)) {
-                       return status;
+       struct composite_context *c;
+       struct schannel_key_state *s;
+
+       c = tevent_req_callback_data(subreq, struct composite_context);
+       s = talloc_get_type(c->private_data, struct schannel_key_state);
+
+       /* receive rpc request result - auth2 credentials */ 
+       c->status = dcerpc_netr_ServerAuthenticate2_r_recv(subreq, s);
+       TALLOC_FREE(subreq);
+       if (!composite_is_ok(c)) return;
+
+       if (!NT_STATUS_EQUAL(s->a.out.result, NT_STATUS_ACCESS_DENIED) &&
+           !NT_STATUS_IS_OK(s->a.out.result)) {
+               composite_error(c, s->a.out.result);
+               return;
+       }
+
+       /*
+        * Strong keys could be unsupported (NT4) or disables. So retry with the
+        * flags returned by the server. - asn
+        */
+       if (NT_STATUS_EQUAL(s->a.out.result, NT_STATUS_ACCESS_DENIED)) {
+               uint32_t lf = s->local_negotiate_flags;
+               const char *ln = NULL;
+               uint32_t rf = s->remote_negotiate_flags;
+               const char *rn = NULL;
+
+               if (!s->dcerpc_schannel_auto) {
+                       composite_error(c, s->a.out.result);
+                       return;
                }
-               
-               if (bind_schannel.bind_type == 23) {
-                       account_name = bind_schannel.u.info23.account_name;
+               s->dcerpc_schannel_auto = false;
+
+               if (lf & NETLOGON_NEG_SUPPORTS_AES)  {
+                       ln = "aes";
+                       if (rf & NETLOGON_NEG_SUPPORTS_AES) {
+                               composite_error(c, s->a.out.result);
+                               return;
+                       }
+               } else if (lf & NETLOGON_NEG_STRONG_KEYS) {
+                       ln = "strong";
+                       if (rf & NETLOGON_NEG_STRONG_KEYS) {
+                               composite_error(c, s->a.out.result);
+                               return;
+                       }
                } else {
-                       account_name = bind_schannel.u.info3.account_name;
-               }
-               
-               /* pull the session key for this client */
-               status = schannel_fetch_session_key(out_mem_ctx, account_name, &dce_schan_state->creds);
-               if (!NT_STATUS_IS_OK(status)) {
-                       DEBUG(3, ("Could not find session key for attempted schannel connection on %s: %s\n",
-                                 account_name, nt_errstr(status)));
-                       return status;
+                       ln = "des";
                }
 
-               dce_schan_state->account_name = talloc_strdup(dce_schan_state, account_name);
-               
-               /* start up the schannel server code */
-               status = schannel_start(&dce_schan_state->schannel_state, 
-                                       dce_schan_state->creds->session_key, False);
-               if (!NT_STATUS_IS_OK(status)) {
-                       DEBUG(3, ("Could not initialise schannel state for account %s: %s\n",
-                                 account_name, nt_errstr(status)));
-                       return status;
-               }
-               talloc_steal(dce_schan_state, dce_schan_state->schannel_state);
-               
-               bind_schannel_ack.unknown1 = 1;
-               bind_schannel_ack.unknown2 = 0;
-               bind_schannel_ack.unknown3 = 0x6c0000;
-               
-               status = ndr_push_struct_blob(out, out_mem_ctx, &bind_schannel_ack, 
-                                             (ndr_push_flags_fn_t)ndr_push_schannel_bind_ack);
-               if (!NT_STATUS_IS_OK(status)) {
-                       DEBUG(3, ("Could not return schannel bind ack for account %s: %s\n",
-                                 account_name, nt_errstr(status)));
-                       return status;
+               if (rf & NETLOGON_NEG_SUPPORTS_AES)  {
+                       rn = "aes";
+               } else if (rf & NETLOGON_NEG_STRONG_KEYS) {
+                       rn = "strong";
+               } else {
+                       rn = "des";
                }
 
-               dce_schan_state->state = DCERPC_SCHANNEL_STATE_UPDATE_1;
+               DEBUG(3, ("Server doesn't support %s keys, downgrade to %s"
+                         "and retry! local[0x%08X] remote[0x%08X]\n",
+                         ln, rn, lf, rf));
 
-               return NT_STATUS_OK;
-       }
-       return NT_STATUS_INVALID_PARAMETER;
-}
+               s->local_negotiate_flags = s->remote_negotiate_flags;
 
-/** 
- * Return the credentials of a logged on user, including session keys
- * etc.
- *
- * Only valid after a successful authentication
- *
- * May only be called once per authentication.
- *
- */
+               generate_random_buffer(s->credentials1.data,
+                                      sizeof(s->credentials1.data));
 
-NTSTATUS dcerpc_schannel_session_info(struct gensec_security *gensec_security,
-                                     struct auth_session_info **session_info)
-{
-       (*session_info) = talloc(gensec_security, struct auth_session_info);
-       NT_STATUS_HAVE_NO_MEMORY(*session_info);
+               subreq = dcerpc_netr_ServerReqChallenge_r_send(s,
+                                                              c->event_ctx,
+                                                              s->pipe2->binding_handle,
+                                                              &s->r);
+               if (composite_nomem(subreq, c)) return;
 
-       ZERO_STRUCTP(*session_info);
+               tevent_req_set_callback(subreq, continue_srv_challenge, c);
+               return;
+       }
 
-       return NT_STATUS_OK;
-}
+       s->creds->negotiate_flags = s->remote_negotiate_flags;
 
-/**
- * Return the struct creds_CredentialState.
- *
- * Make sure not to call this unless gensec is using schannel...
- */
+       /* verify credentials */
+       if (!netlogon_creds_client_check(s->creds, s->a.out.return_credentials)) {
+               composite_error(c, NT_STATUS_UNSUCCESSFUL);
+               return;
+       }
 
-NTSTATUS dcerpc_schannel_creds(struct gensec_security *gensec_security,
-                              TALLOC_CTX *mem_ctx,
-                              struct creds_CredentialState **creds)
-{ 
-       struct dcerpc_schannel_state *dce_schan_state = gensec_security->private_data;
+       /* setup current netlogon credentials */
+       cli_credentials_set_netlogon_creds(s->credentials, s->creds);
 
-       *creds = talloc_reference(mem_ctx, dce_schan_state->creds);
-       if (!*creds) {
-               return NT_STATUS_NO_MEMORY;
-       }
-       return NT_STATUS_OK;
+       composite_done(c);
 }
-               
 
 /*
-  end crypto state
+  Initiate establishing a schannel key using netlogon challenge
+  on a secondary pipe
 */
-static int dcerpc_schannel_destroy(void *ptr)
+struct composite_context *dcerpc_schannel_key_send(TALLOC_CTX *mem_ctx,
+                                                  struct dcerpc_pipe *p,
+                                                  struct cli_credentials *credentials,
+                                                  struct loadparm_context *lp_ctx)
 {
-       struct dcerpc_schannel_state *dce_schan_state = ptr;
+       struct composite_context *c;
+       struct schannel_key_state *s;
+       struct composite_context *epm_map_req;
+       enum netr_SchannelType schannel_type = cli_credentials_get_secure_channel_type(credentials);
+       
+       /* composite context allocation and setup */
+       c = composite_create(mem_ctx, p->conn->event_ctx);
+       if (c == NULL) return NULL;
+
+       s = talloc_zero(c, struct schannel_key_state);
+       if (composite_nomem(s, c)) return c;
+       c->private_data = s;
+
+       /* store parameters in the state structure */
+       s->pipe        = p;
+       s->credentials = credentials;
+       s->local_negotiate_flags = NETLOGON_NEG_AUTH2_FLAGS;
+
+       /* allocate credentials */
+       if (s->pipe->conn->flags & DCERPC_SCHANNEL_128) {
+               s->local_negotiate_flags = NETLOGON_NEG_AUTH2_ADS_FLAGS;
+       }
+       if (s->pipe->conn->flags & DCERPC_SCHANNEL_AES) {
+               s->local_negotiate_flags = NETLOGON_NEG_AUTH2_ADS_FLAGS;
+               s->local_negotiate_flags |= NETLOGON_NEG_SUPPORTS_AES;
+       }
+       if (s->pipe->conn->flags & DCERPC_SCHANNEL_AUTO) {
+               s->local_negotiate_flags = NETLOGON_NEG_AUTH2_ADS_FLAGS;
+               s->local_negotiate_flags |= NETLOGON_NEG_SUPPORTS_AES;
+               s->dcerpc_schannel_auto = true;
+       }
 
-       schannel_end(&dce_schan_state->schannel_state);
+       /* type of authentication depends on schannel type */
+       if (schannel_type == SEC_CHAN_RODC) {
+               s->local_negotiate_flags |= NETLOGON_NEG_RODC_PASSTHROUGH;
+       }
 
-       return 0;
-}
+       /* allocate binding structure */
+       s->binding = talloc_zero(c, struct dcerpc_binding);
+       if (composite_nomem(s->binding, c)) return c;
 
-static NTSTATUS dcerpc_schannel_start(struct gensec_security *gensec_security)
-{
-       struct dcerpc_schannel_state *dce_schan_state;
+       *s->binding = *s->pipe->binding;
 
-       dce_schan_state = talloc(gensec_security, struct dcerpc_schannel_state);
-       if (!dce_schan_state) {
-               return NT_STATUS_NO_MEMORY;
-       }
+       /* request the netlogon endpoint mapping */
+       epm_map_req = dcerpc_epm_map_binding_send(c, s->binding,
+                                                 &ndr_table_netlogon,
+                                                 s->pipe->conn->event_ctx,
+                                                 lp_ctx);
+       if (composite_nomem(epm_map_req, c)) return c;
+
+       composite_continue(c, epm_map_req, continue_epm_map_binding, c);
+       return c;
+}
 
-       dce_schan_state->state = DCERPC_SCHANNEL_STATE_START;
-       gensec_security->private_data = dce_schan_state;
 
-       talloc_set_destructor(dce_schan_state, dcerpc_schannel_destroy);
+/*
+  Receive result of schannel key request
+ */
+NTSTATUS dcerpc_schannel_key_recv(struct composite_context *c)
+{
+       NTSTATUS status = composite_wait(c);
        
-       return NT_STATUS_OK;
+       talloc_free(c);
+       return status;
 }
 
-static NTSTATUS dcerpc_schannel_server_start(struct gensec_security *gensec_security) 
-{
-       NTSTATUS status;
 
-       status = dcerpc_schannel_start(gensec_security);
-       if (!NT_STATUS_IS_OK(status)) {
-               return status;
-       }
+struct auth_schannel_state {
+       struct dcerpc_pipe *pipe;
+       struct cli_credentials *credentials;
+       const struct ndr_interface_table *table;
+       struct loadparm_context *lp_ctx;
+       uint8_t auth_level;
+       struct netlogon_creds_CredentialState *creds_state;
+       struct netlogon_creds_CredentialState save_creds_state;
+       struct netr_Authenticator auth;
+       struct netr_Authenticator return_auth;
+       union netr_Capabilities capabilities;
+       struct netr_LogonGetCapabilities c;
+};
 
-       return NT_STATUS_OK;
-}
 
-static NTSTATUS dcerpc_schannel_client_start(struct gensec_security *gensec_security) 
+static void continue_bind_auth(struct composite_context *ctx);
+
+
+/*
+  Stage 2 of auth_schannel: Receive schannel key and intitiate an
+  authenticated bind using received credentials
+ */
+static void continue_schannel_key(struct composite_context *ctx)
 {
+       struct composite_context *auth_req;
+       struct composite_context *c = talloc_get_type(ctx->async.private_data,
+                                                     struct composite_context);
+       struct auth_schannel_state *s = talloc_get_type(c->private_data,
+                                                       struct auth_schannel_state);
        NTSTATUS status;
 
-       status = dcerpc_schannel_start(gensec_security);
-       if (!NT_STATUS_IS_OK(status)) {
-               return status;
+       /* receive schannel key */
+       status = c->status = dcerpc_schannel_key_recv(ctx);
+       if (!composite_is_ok(c)) {
+               DEBUG(1, ("Failed to setup credentials: %s\n", nt_errstr(status)));
+               return;
        }
 
-       return NT_STATUS_OK;
+       /* send bind auth request with received creds */
+       auth_req = dcerpc_bind_auth_send(c, s->pipe, s->table, s->credentials, 
+                                        lpcfg_gensec_settings(c, s->lp_ctx),
+                                        DCERPC_AUTH_TYPE_SCHANNEL, s->auth_level,
+                                        NULL);
+       if (composite_nomem(auth_req, c)) return;
+       
+       composite_continue(c, auth_req, continue_bind_auth, c);
 }
 
 
 /*
-  get a schannel key using a netlogon challenge on a secondary pipe
+  Stage 3 of auth_schannel: Receivce result of authenticated bind
+  and say if we're done ok.
 */
-static NTSTATUS dcerpc_schannel_key(struct dcerpc_pipe *p,
-                                   const char *domain,
-                                   const char *username,
-                                   const char *password,
-                                   int chan_type,
-                                   struct creds_CredentialState *creds)
+static void continue_bind_auth(struct composite_context *ctx)
 {
-       NTSTATUS status;
-       struct dcerpc_pipe *p2;
-       struct netr_ServerReqChallenge r;
-       struct netr_ServerAuthenticate2 a;
-       struct netr_Credential credentials1, credentials2, credentials3;
-       struct samr_Password mach_pwd;
-       const char *workgroup, *workstation;
-       uint32_t negotiate_flags;
-
-       if (p->conn->flags & DCERPC_SCHANNEL_128) {
-               negotiate_flags = NETLOGON_NEG_AUTH2_ADS_FLAGS;
-       } else {
-               negotiate_flags = NETLOGON_NEG_AUTH2_FLAGS;
+       struct composite_context *c = talloc_get_type(ctx->async.private_data,
+                                                     struct composite_context);
+       struct auth_schannel_state *s = talloc_get_type(c->private_data,
+                                                       struct auth_schannel_state);
+       struct tevent_req *subreq;
+
+       c->status = dcerpc_bind_auth_recv(ctx);
+       if (!composite_is_ok(c)) return;
+
+       /* if we have a AES encrypted connection, verify the capabilities */
+       if (ndr_syntax_id_equal(&s->table->syntax_id,
+                               &ndr_table_netlogon.syntax_id)) {
+               ZERO_STRUCT(s->return_auth);
+
+               s->creds_state = cli_credentials_get_netlogon_creds(s->credentials);
+               if (composite_nomem(s->creds_state, c)) return;
+
+               s->save_creds_state = *s->creds_state;
+               netlogon_creds_client_authenticator(&s->save_creds_state, &s->auth);
+
+               s->c.in.server_name = talloc_asprintf(c,
+                                                     "\\\\%s",
+                                                     dcerpc_server_name(s->pipe));
+               if (composite_nomem(s->c.in.server_name, c)) return;
+               s->c.in.computer_name         = cli_credentials_get_workstation(s->credentials);
+               s->c.in.credential            = &s->auth;
+               s->c.in.return_authenticator  = &s->return_auth;
+               s->c.in.query_level           = 1;
+
+               s->c.out.capabilities         = &s->capabilities;
+               s->c.out.return_authenticator = &s->return_auth;
+
+               DEBUG(5, ("We established a AES connection, verifying logon "
+                         "capabilities\n"));
+
+               subreq = dcerpc_netr_LogonGetCapabilities_r_send(s,
+                                                                c->event_ctx,
+                                                                s->pipe->binding_handle,
+                                                                &s->c);
+               if (composite_nomem(subreq, c)) return;
+
+               tevent_req_set_callback(subreq, continue_get_capabilities, c);
+               return;
        }
 
-       workstation = username;
-       workgroup = domain;
+       composite_done(c);
+}
 
-       /*
-         step 1 - establish a netlogon connection, with no authentication
-       */
-       status = dcerpc_secondary_connection(p, &p2, 
-                                            DCERPC_NETLOGON_NAME, 
-                                            DCERPC_NETLOGON_UUID, 
-                                            DCERPC_NETLOGON_VERSION);
-       if (!NT_STATUS_IS_OK(status)) {
-               return status;
+/*
+  Stage 4 of auth_schannel: Get the Logon Capablities and verify them.
+*/
+static void continue_get_capabilities(struct tevent_req *subreq)
+{
+       struct composite_context *c;
+       struct auth_schannel_state *s;
+
+       c = tevent_req_callback_data(subreq, struct composite_context);
+       s = talloc_get_type(c->private_data, struct auth_schannel_state);
+
+       /* receive rpc request result */
+       c->status = dcerpc_netr_LogonGetCapabilities_r_recv(subreq, s);
+       TALLOC_FREE(subreq);
+       if (NT_STATUS_EQUAL(c->status, NT_STATUS_RPC_PROCNUM_OUT_OF_RANGE)) {
+               if (s->creds_state->negotiate_flags & NETLOGON_NEG_SUPPORTS_AES) {
+                       composite_error(c, NT_STATUS_INVALID_NETWORK_RESPONSE);
+                       return;
+               } else {
+                       /* This is probably NT */
+                       composite_done(c);
+                       return;
+               }
+       } else if (!composite_is_ok(c)) {
+               return;
        }
 
+       if (NT_STATUS_EQUAL(s->c.out.result, NT_STATUS_NOT_IMPLEMENTED)) {
+               if (s->creds_state->negotiate_flags & NETLOGON_NEG_SUPPORTS_AES) {
+                       /* This means AES isn't supported. */
+                       composite_error(c, NT_STATUS_INVALID_NETWORK_RESPONSE);
+                       return;
+               }
 
-       /*
-         step 2 - request a netlogon challenge
-       */
-       r.in.server_name = talloc_asprintf(p, "\\\\%s", dcerpc_server_name(p));
-       r.in.computer_name = workstation;
-       r.in.credentials = &credentials1;
-       r.out.credentials = &credentials2;
-
-       generate_random_buffer(credentials1.data, sizeof(credentials1.data));
-
-       status = dcerpc_netr_ServerReqChallenge(p2, p, &r);
-       if (!NT_STATUS_IS_OK(status)) {
-               return status;
+               /* This is probably an old Samba version */
+               composite_done(c);
+               return;
        }
 
-       /*
-         step 3 - authenticate on the netlogon pipe
-       */
-       E_md4hash(password, mach_pwd.hash);
-       creds_client_init(creds, &credentials1, &credentials2, &mach_pwd, &credentials3,
-                         negotiate_flags);
-
-       a.in.server_name = r.in.server_name;
-       a.in.account_name = talloc_asprintf(p, "%s$", workstation);
-       a.in.secure_channel_type = chan_type;
-       a.in.computer_name = workstation;
-       a.in.negotiate_flags = &negotiate_flags;
-       a.out.negotiate_flags = &negotiate_flags;
-       a.in.credentials = &credentials3;
-       a.out.credentials = &credentials3;
-
-       status = dcerpc_netr_ServerAuthenticate2(p2, p, &a);
-       if (!NT_STATUS_IS_OK(status)) {
-               return status;
+       /* verify credentials */
+       if (!netlogon_creds_client_check(&s->save_creds_state,
+                                        &s->c.out.return_authenticator->cred)) {
+               composite_error(c, NT_STATUS_UNSUCCESSFUL);
+               return;
        }
 
-       if (!creds_client_check(creds, a.out.credentials)) {
-               return NT_STATUS_UNSUCCESSFUL;
+       *s->creds_state = s->save_creds_state;
+
+       if (!NT_STATUS_IS_OK(s->c.out.result)) {
+               composite_error(c, s->c.out.result);
+               return;
        }
 
-       /*
-         the schannel session key is now in creds.session_key
+       /* compare capabilities */
+       if (s->creds_state->negotiate_flags != s->capabilities.server_capabilities) {
+               DEBUG(2, ("The client capabilities don't match the server "
+                         "capabilities: local[0x%08X] remote[0x%08X]\n",
+                         s->creds_state->negotiate_flags,
+                         s->capabilities.server_capabilities));
+               composite_error(c, NT_STATUS_INVALID_NETWORK_RESPONSE);
+               return;
+       }
 
-         we no longer need the netlogon pipe open
-       */
-       dcerpc_pipe_close(p2);
+       /* TODO: Add downgrade dectection. */
 
-       return NT_STATUS_OK;
+       composite_done(c);
 }
 
+
 /*
-  do a schannel style bind on a dcerpc pipe. The username is usually
-  of the form HOSTNAME$ and the password is the domain trust password
+  Initiate schannel authentication request
 */
-NTSTATUS dcerpc_bind_auth_schannel_withkey(struct dcerpc_pipe *p,
-                                          const char *uuid, uint_t version,
-                                          const char *domain,
-                                          const char *username,
-                                          const char *password,
-                                          struct creds_CredentialState *creds)
+struct composite_context *dcerpc_bind_auth_schannel_send(TALLOC_CTX *tmp_ctx, 
+                                                        struct dcerpc_pipe *p,
+                                                        const struct ndr_interface_table *table,
+                                                        struct cli_credentials *credentials,
+                                                        struct loadparm_context *lp_ctx,
+                                                        uint8_t auth_level)
 {
-       NTSTATUS status;
-       struct dcerpc_schannel_state *dce_schan_state;
+       struct composite_context *c;
+       struct auth_schannel_state *s;
+       struct composite_context *schan_key_req;
 
-       status = gensec_client_start(p, &p->conn->security_state.generic_state);
-       if (!NT_STATUS_IS_OK(status)) {
-               return status;
-       }
-
-       status = gensec_set_username(p->conn->security_state.generic_state, username);
-       if (!NT_STATUS_IS_OK(status)) {
-               DEBUG(1, ("Failed to set schannel username to %s: %s\n", username, nt_errstr(status)));
-               talloc_free(p->conn->security_state.generic_state);
-               p->conn->security_state.generic_state = NULL;
-               return status;
-       }
-       
-       status = gensec_set_domain(p->conn->security_state.generic_state, domain);
-       if (!NT_STATUS_IS_OK(status)) {
-               DEBUG(1, ("Failed to set schannel domain to %s: %s\n", domain, nt_errstr(status)));
-               talloc_free(p->conn->security_state.generic_state);
-               p->conn->security_state.generic_state = NULL;
-               return status;
-       }
+       /* composite context allocation and setup */
+       c = composite_create(tmp_ctx, p->conn->event_ctx);
+       if (c == NULL) return NULL;
        
-       status = gensec_start_mech_by_authtype(p->conn->security_state.generic_state, 
-                                              DCERPC_AUTH_TYPE_SCHANNEL, 
-                                              dcerpc_auth_level(p->conn));
-
-       if (!NT_STATUS_IS_OK(status)) {
-               DEBUG(1, ("Failed to start SCHANNEL GENSEC backend: %s\n", nt_errstr(status)));
-               talloc_free(p->conn->security_state.generic_state);
-               p->conn->security_state.generic_state = NULL;
-               return status;
-       }
-
-       dce_schan_state = p->conn->security_state.generic_state->private_data;
-       dce_schan_state->creds = talloc_reference(dce_schan_state, creds);
-
-       status = dcerpc_bind_auth(p, DCERPC_AUTH_TYPE_SCHANNEL, dcerpc_auth_level(p->conn),
-                                 uuid, version);
-
-       if (!NT_STATUS_IS_OK(status)) {
-               DEBUG(1, ("Failed to bind to pipe with SCHANNEL: %s\n", nt_errstr(status)));
-               talloc_free(p->conn->security_state.generic_state);
-               p->conn->security_state.generic_state = NULL;
-               return status;
-       }
-
-       return NT_STATUS_OK;
+       s = talloc_zero(c, struct auth_schannel_state);
+       if (composite_nomem(s, c)) return c;
+       c->private_data = s;
+
+       /* store parameters in the state structure */
+       s->pipe        = p;
+       s->credentials = credentials;
+       s->table       = table;
+       s->auth_level  = auth_level;
+       s->lp_ctx      = lp_ctx;
+
+       /* start getting schannel key first */
+       schan_key_req = dcerpc_schannel_key_send(c, p, credentials, lp_ctx);
+       if (composite_nomem(schan_key_req, c)) return c;
+
+       composite_continue(c, schan_key_req, continue_schannel_key, c);
+       return c;
 }
 
-NTSTATUS dcerpc_bind_auth_schannel(struct dcerpc_pipe *p,
-                                  const char *uuid, uint_t version,
-                                  const char *domain,
-                                  const char *username,
-                                  const char *password)
-{
-       NTSTATUS status;
-       int chan_type = 0;
-       struct creds_CredentialState *creds;
-       creds = talloc(p, struct creds_CredentialState);
-       if (!creds) {
-               return NT_STATUS_NO_MEMORY;
-       }
-
-       if (p->conn->flags & DCERPC_SCHANNEL_BDC) {
-               chan_type = SEC_CHAN_BDC;
-       } else if (p->conn->flags & DCERPC_SCHANNEL_WORKSTATION) {
-               chan_type = SEC_CHAN_WKSTA;
-       } else if (p->conn->flags & DCERPC_SCHANNEL_DOMAIN) {
-               chan_type = SEC_CHAN_DOMAIN;
-       }
 
-       status = dcerpc_schannel_key(p, domain, 
-                                    username,
-                                    password, 
-                                    chan_type,
-                                    creds);
-
-       if (!NT_STATUS_IS_OK(status)) {
-               DEBUG(1, ("Failed to fetch schannel session key: %s\n",
-                         nt_errstr(status)));
-               return status;
-       }
-
-       return dcerpc_bind_auth_schannel_withkey(p, uuid, version, domain,
-                                                username, password,
-                                                creds);
-}
-
-static BOOL dcerpc_schannel_have_feature(struct gensec_security *gensec_security,
-                                         uint32 feature)
+/*
+  Receive result of schannel authentication request
+*/
+NTSTATUS dcerpc_bind_auth_schannel_recv(struct composite_context *c)
 {
-       if (feature & (GENSEC_FEATURE_SESSION_KEY | 
-                      GENSEC_FEATURE_SIGN | 
-                      GENSEC_FEATURE_SEAL)) {
-               return True;
-       }
-       return False;
+       NTSTATUS status = composite_wait(c);
+       
+       talloc_free(c);
+       return status;
 }
 
 
-static const struct gensec_security_ops gensec_dcerpc_schannel_security_ops = {
-       .name           = "dcerpc_schannel",
-       .auth_type      = DCERPC_AUTH_TYPE_SCHANNEL,
-       .client_start   = dcerpc_schannel_client_start,
-       .server_start   = dcerpc_schannel_server_start,
-       .update         = dcerpc_schannel_update,
-       .seal_packet    = dcerpc_schannel_seal_packet,
-       .sign_packet    = dcerpc_schannel_sign_packet,
-       .check_packet   = dcerpc_schannel_check_packet,
-       .unseal_packet  = dcerpc_schannel_unseal_packet,
-       .session_key    = dcerpc_schannel_session_key,
-       .session_info   = dcerpc_schannel_session_info,
-       .sig_size       = dcerpc_schannel_sig_size,
-       .have_feature   = dcerpc_schannel_have_feature,
-       .enabled        = True
-};
-
-NTSTATUS gensec_dcerpc_schannel_init(void)
+/*
+  Perform schannel authenticated bind - sync version
+ */
+_PUBLIC_ NTSTATUS dcerpc_bind_auth_schannel(TALLOC_CTX *tmp_ctx, 
+                                  struct dcerpc_pipe *p,
+                                  const struct ndr_interface_table *table,
+                                  struct cli_credentials *credentials,
+                                  struct loadparm_context *lp_ctx,
+                                  uint8_t auth_level)
 {
-       NTSTATUS ret;
-       ret = gensec_register(&gensec_dcerpc_schannel_security_ops);
-       if (!NT_STATUS_IS_OK(ret)) {
-               DEBUG(0,("Failed to register '%s' gensec backend!\n",
-                       gensec_dcerpc_schannel_security_ops.name));
-               return ret;
-       }
+       struct composite_context *c;
 
-       return ret;
+       c = dcerpc_bind_auth_schannel_send(tmp_ctx, p, table, credentials, lp_ctx,
+                                          auth_level);
+       return dcerpc_bind_auth_schannel_recv(c);
 }