r5902: A rather large change...
[nivanova/samba-autobuild/.git] / source4 / librpc / rpc / dcerpc_schannel.c
index df15edfb6f1327b2f0d590d3dc0d68efee092cc9..170ddee1f300545135d53e802c73ff561c243288 100644 (file)
 */
 
 #include "includes.h"
+#include "librpc/gen_ndr/ndr_schannel.h"
+#include "auth/auth.h"
 
-#define DCERPC_SCHANNEL_STATE_START 0
-#define DCERPC_SCHANNEL_STATE_UPDATE_1 1
+enum schannel_position {
+       DCERPC_SCHANNEL_STATE_START = 0,
+       DCERPC_SCHANNEL_STATE_UPDATE_1
+};
 
 struct dcerpc_schannel_state {
-       TALLOC_CTX *mem_ctx;
-       uint8_t state;
-       struct schannel_bind bind_schannel;
+       enum schannel_position state;
        struct schannel_state *schannel_state;
+       struct creds_CredentialState *creds;
 };
 
 /*
   wrappers for the schannel_*() functions
+
+  These will become static again, when we get dynamic registration, and
+  decrpc_schannel_security_ops come back here.
 */
-static NTSTATUS dcerpc_schannel_unseal(struct dcerpc_security *dcerpc_security, 
-                                   TALLOC_CTX *mem_ctx, 
-                                   uint8_t *data, size_t length, DATA_BLOB *sig)
+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)
 {
-       struct dcerpc_schannel_state *dce_schan_state = dcerpc_security->private_data;
-
+       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);
 }
 
-static NTSTATUS dcerpc_schannel_check_sig(struct dcerpc_security *dcerpc_security, 
-                                  TALLOC_CTX *mem_ctx, 
-                                  const uint8_t *data, size_t length, 
-                                  const DATA_BLOB *sig)
+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 = dcerpc_security->private_data;
+       struct dcerpc_schannel_state *dce_schan_state = gensec_security->private_data;
 
        return schannel_check_packet(dce_schan_state->schannel_state, data, length, sig);
 }
 
-static NTSTATUS dcerpc_schannel_seal(struct dcerpc_security *dcerpc_security, 
-                                 TALLOC_CTX *mem_ctx, 
-                                 uint8_t *data, size_t length, 
-                                 DATA_BLOB *sig)
+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)
 {
-       struct dcerpc_schannel_state *dce_schan_state = dcerpc_security->private_data;
+       struct dcerpc_schannel_state *dce_schan_state = gensec_security->private_data;
 
        return schannel_seal_packet(dce_schan_state->schannel_state, mem_ctx, data, length, sig);
 }
 
-static NTSTATUS dcerpc_schannel_sign(struct dcerpc_security *dcerpc_security, 
-                                TALLOC_CTX *mem_ctx, 
-                                const uint8_t *data, size_t length, 
-                                DATA_BLOB *sig)
+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 = dcerpc_security->private_data;
+       struct dcerpc_schannel_state *dce_schan_state = gensec_security->private_data;
 
        return schannel_sign_packet(dce_schan_state->schannel_state, mem_ctx, data, length, sig);
 }
 
-static NTSTATUS dcerpc_schannel_session_key(struct dcerpc_security *dcerpc_security, 
-                                 DATA_BLOB *session_key)
+static size_t dcerpc_schannel_sig_size(struct gensec_security *gensec_security)
+{
+       return 32;
+}
+
+static NTSTATUS dcerpc_schannel_session_key(struct gensec_security *gensec_security, 
+                                           DATA_BLOB *session_key)
 {
        return NT_STATUS_NOT_IMPLEMENTED;
 }
 
-static NTSTATUS dcerpc_schannel_start(struct dcerpc_pipe *p, struct dcerpc_security *dcerpc_security)
+static NTSTATUS dcerpc_schannel_update(struct gensec_security *gensec_security, TALLOC_CTX *out_mem_ctx, 
+                                      const DATA_BLOB in, DATA_BLOB *out) 
 {
-       struct dcerpc_schannel_state *dce_schan_state;
-       TALLOC_CTX *mem_ctx;
+       struct dcerpc_schannel_state *dce_schan_state = gensec_security->private_data;
        NTSTATUS status;
-       uint8_t session_key[16];
-       int chan_type = 0;
+       struct schannel_bind bind_schannel;
+       struct schannel_bind_ack bind_schannel_ack;
+       const char *workstation;
+       const char *domain;
+       *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,
+                                       &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_get_workstation(gensec_security));
+#else
+               bind_schannel.bind_type = 3;
+               bind_schannel.u.info3.domain = gensec_security->user.domain;
+               bind_schannel.u.info3.workstation = gensec_get_workstation(gensec_security);
+#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;
+               }
+               
+               if (bind_schannel.bind_type == 23) {
+                       workstation = bind_schannel.u.info23.workstation;
+                       domain = bind_schannel.u.info23.domain;
+               } else {
+                       workstation = bind_schannel.u.info3.workstation;
+                       domain = bind_schannel.u.info3.domain;
+               }
+               
+               /* pull the session key for this client */
+               status = schannel_fetch_session_key(out_mem_ctx, workstation, 
+                                                   domain, &dce_schan_state->creds);
+               if (!NT_STATUS_IS_OK(status)) {
+                       DEBUG(3, ("Could not find session key for attempted schannel connection from %s: %s\n",
+                                 workstation, nt_errstr(status)));
+                       return status;
+               }
+
+               /* start up the schannel server code */
+               status = schannel_start(dce_schan_state,
+                                       &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 from client %s: %s\n",
+                                 workstation, 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 client %s: %s\n",
+                                 workstation, nt_errstr(status)));
+                       return status;
+               }
+
+               dce_schan_state->state = DCERPC_SCHANNEL_STATE_UPDATE_1;
 
-       if (p->flags & DCERPC_SCHANNEL_BDC) {
-               chan_type = SEC_CHAN_BDC;
-       } else if (p->flags & DCERPC_SCHANNEL_WORKSTATION) {
-               chan_type = SEC_CHAN_WKSTA;
-       } else if (p->flags & DCERPC_SCHANNEL_DOMAIN) {
-               chan_type = SEC_CHAN_DOMAIN;
+               return NT_STATUS_OK;
        }
+       return NT_STATUS_INVALID_PARAMETER;
+}
 
-       status = dcerpc_schannel_key(p, dcerpc_security->user.domain, 
-                                       dcerpc_security->user.name,
-                                       dcerpc_security->user.password, 
-                                       chan_type, session_key);
-       if (!NT_STATUS_IS_OK(status)) {
-               return status;
-       }
+/** 
+ * 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.
+ *
+ */
+
+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);
 
-       mem_ctx = talloc_init("dcerpc_schannel_start");
-       if (!mem_ctx) {
-               return NT_STATUS_NO_MEMORY;
-       }
+       ZERO_STRUCTP(*session_info);
 
-       dce_schan_state = talloc_p(mem_ctx, struct dcerpc_schannel_state);
-       if (!dce_schan_state) {
-               talloc_destroy(mem_ctx);
+       return NT_STATUS_OK;
+}
+
+/**
+ * Return the struct creds_CredentialState.
+ *
+ * Make sure not to call this unless gensec is using schannel...
+ */
+
+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;
+
+       *creds = talloc_reference(mem_ctx, dce_schan_state->creds);
+       if (!*creds) {
                return NT_STATUS_NO_MEMORY;
        }
+       return NT_STATUS_OK;
+}
+               
 
-       dce_schan_state->mem_ctx = mem_ctx;
+static NTSTATUS dcerpc_schannel_start(struct gensec_security *gensec_security)
+{
+       struct dcerpc_schannel_state *dce_schan_state;
 
-       status = schannel_start(&dce_schan_state->schannel_state, session_key, True);
-       if (!NT_STATUS_IS_OK(status)) {
-               return status;
+       dce_schan_state = talloc(gensec_security, struct dcerpc_schannel_state);
+       if (!dce_schan_state) {
+               return NT_STATUS_NO_MEMORY;
        }
 
        dce_schan_state->state = DCERPC_SCHANNEL_STATE_START;
+       gensec_security->private_data = dce_schan_state;
 
-       dcerpc_security->private_data = dce_schan_state;
-
-       dump_data_pw("session key:\n", dce_schan_state->schannel_state->session_key, 16);
-
-       return status;
+       return NT_STATUS_OK;
 }
 
-static NTSTATUS dcerpc_schannel_update(struct dcerpc_security *dcerpc_security, TALLOC_CTX *out_mem_ctx, 
-                                               const DATA_BLOB in, DATA_BLOB *out) 
+static NTSTATUS dcerpc_schannel_server_start(struct gensec_security *gensec_security) 
 {
-       struct dcerpc_schannel_state *dce_schan_state = dcerpc_security->private_data;
        NTSTATUS status;
-       struct schannel_bind bind_schannel;
-
-       if (dce_schan_state->state != DCERPC_SCHANNEL_STATE_START) {
-               return NT_STATUS_OK;
-       }
-
-       dce_schan_state->state = DCERPC_SCHANNEL_STATE_UPDATE_1;
 
-       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 = dcerpc_security->user.domain;
-       bind_schannel.u.info23.account_name = dcerpc_security->user.name;
-       bind_schannel.u.info23.dnsdomain = str_format_nbt_domain(dce_schan_state->mem_ctx, fulldomainname);
-       bind_schannel.u.info23.workstation = str_format_nbt_domain(dce_schan_state->mem_ctx, dcerpc_security->user.name);
-#else
-       bind_schannel.bind_type = 3;
-       bind_schannel.u.info3.domain = dcerpc_security->user.domain;
-       bind_schannel.u.info3.account_name = dcerpc_security->user.name;
-#endif
-
-       status = ndr_push_struct_blob(out, dce_schan_state->mem_ctx, &bind_schannel,
-                                     (ndr_push_flags_fn_t)ndr_push_schannel_bind);
+       status = dcerpc_schannel_start(gensec_security);
        if (!NT_STATUS_IS_OK(status)) {
                return status;
        }
 
-       return NT_STATUS_MORE_PROCESSING_REQUIRED;
+       return NT_STATUS_OK;
 }
 
-static void dcerpc_schannel_end(struct dcerpc_security *dcerpc_security)
+static NTSTATUS dcerpc_schannel_client_start(struct gensec_security *gensec_security) 
 {
-       struct dcerpc_schannel_state *dce_schan_state = dcerpc_security->private_data;
-
-       schannel_end(&dce_schan_state->schannel_state);
+       NTSTATUS status;
 
-       talloc_destroy(dce_schan_state->mem_ctx);
+       status = dcerpc_schannel_start(gensec_security);
+       if (!NT_STATUS_IS_OK(status)) {
+               return status;
+       }
 
-       dcerpc_security->private_data = NULL;
+       return NT_STATUS_OK;
 }
 
 
 /*
   get a schannel key using a netlogon challenge on a secondary pipe
 */
-NTSTATUS dcerpc_schannel_key(struct dcerpc_pipe *p,
-                            const char *domain,
-                            const char *username,
-                            const char *password,
-                            int chan_type,
-                            uint8_t new_session_key[16])
+static NTSTATUS dcerpc_schannel_key(struct dcerpc_pipe *p,
+                                   const char *workstation,
+                                   const char *domain,
+                                   const char *username,
+                                   const char *password,
+                                   int chan_type,
+                                   struct creds_CredentialState *creds)
 {
        NTSTATUS status;
+       struct dcerpc_binding *b;
        struct dcerpc_pipe *p2;
        struct netr_ServerReqChallenge r;
        struct netr_ServerAuthenticate2 a;
        struct netr_Credential credentials1, credentials2, credentials3;
        struct samr_Password mach_pwd;
-       struct creds_CredentialState creds;
-       const char *workgroup, *workstation;
+       const char *workgroup;
        uint32_t negotiate_flags;
+       TALLOC_CTX *tmp_ctx;
 
-       if (p->flags & DCERPC_SCHANNEL_128) {
+       if (p->conn->flags & DCERPC_SCHANNEL_128) {
                negotiate_flags = NETLOGON_NEG_AUTH2_ADS_FLAGS;
        } else {
                negotiate_flags = NETLOGON_NEG_AUTH2_FLAGS;
        }
 
-       workstation = username;
        workgroup = domain;
 
+       tmp_ctx = talloc_new(NULL);
+
        /*
          step 1 - establish a netlogon connection, with no authentication
        */
-       status = dcerpc_secondary_connection(p, &p2, 
-                                            DCERPC_NETLOGON_NAME, 
-                                            DCERPC_NETLOGON_UUID, 
-                                            DCERPC_NETLOGON_VERSION);
 
+       /* Find the original binding string */
+       status = dcerpc_parse_binding(tmp_ctx, p->conn->binding_string, &b);
+       if (!NT_STATUS_IS_OK(status)) {
+               DEBUG(0,("Failed to parse dcerpc binding '%s'\n", p->conn->binding_string));
+               talloc_free(tmp_ctx);
+               return status;
+       }
+
+       /* Make binding string for netlogon, not the other pipe */
+       status = dcerpc_epm_map_binding(tmp_ctx, b, DCERPC_NETLOGON_UUID, DCERPC_NETLOGON_VERSION);
+       if (!NT_STATUS_IS_OK(status)) {
+               DEBUG(0,("Failed to map DCERPC/TCP NCACN_NP pipe for '%s' - %s\n", 
+                        DCERPC_NETLOGON_UUID, nt_errstr(status)));
+               talloc_free(p);
+               return status;
+       }
+
+       status = dcerpc_secondary_connection(p, &p2, b);
+       if (!NT_STATUS_IS_OK(status)) {
+               talloc_free(tmp_ctx);
+               return status;
+       }
+
+       talloc_free(tmp_ctx);
+
+       status = dcerpc_bind_auth_none(p2, DCERPC_NETLOGON_UUID, 
+                                      DCERPC_NETLOGON_VERSION);
+       if (!NT_STATUS_IS_OK(status)) {
+               talloc_free(p2);
+                return status;
+        }
 
        /*
          step 2 - request a netlogon challenge
        */
-       r.in.server_name = talloc_asprintf(p->mem_ctx, "\\\\%s", dcerpc_server_name(p));
+       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), False);
+       generate_random_buffer(credentials1.data, sizeof(credentials1.data));
 
-       status = dcerpc_netr_ServerReqChallenge(p2, p->mem_ctx, &r);
+       status = dcerpc_netr_ServerReqChallenge(p2, p, &r);
        if (!NT_STATUS_IS_OK(status)) {
                return status;
        }
@@ -236,11 +378,13 @@ NTSTATUS dcerpc_schannel_key(struct dcerpc_pipe *p,
          step 3 - authenticate on the netlogon pipe
        */
        E_md4hash(password, mach_pwd.hash);
-       creds_client_init(&creds, &credentials1, &credentials2, &mach_pwd, &credentials3,
+       creds_client_init(creds, &credentials1, &credentials2, 
+                         workstation, domain, username, 
+                         &mach_pwd, &credentials3,
                          negotiate_flags);
 
        a.in.server_name = r.in.server_name;
-       a.in.account_name = talloc_asprintf(p->mem_ctx, "%s$", workstation);
+       a.in.account_name = username;
        a.in.secure_channel_type = chan_type;
        a.in.computer_name = workstation;
        a.in.negotiate_flags = &negotiate_flags;
@@ -248,12 +392,12 @@ NTSTATUS dcerpc_schannel_key(struct dcerpc_pipe *p,
        a.in.credentials = &credentials3;
        a.out.credentials = &credentials3;
 
-       status = dcerpc_netr_ServerAuthenticate2(p2, p->mem_ctx, &a);
+       status = dcerpc_netr_ServerAuthenticate2(p2, p, &a);
        if (!NT_STATUS_IS_OK(status)) {
                return status;
        }
 
-       if (!creds_client_check(&creds, a.out.credentials)) {
+       if (!creds_client_check(creds, a.out.credentials)) {
                return NT_STATUS_UNSUCCESSFUL;
        }
 
@@ -264,45 +408,156 @@ NTSTATUS dcerpc_schannel_key(struct dcerpc_pipe *p,
        */
        dcerpc_pipe_close(p2);
 
-       memcpy(new_session_key, creds.session_key, 16);
-
        return NT_STATUS_OK;
 }
 
-const struct dcesrv_security_ops dcerpc_schannel_security_ops = {
-       .name           = "schannel",
-       .auth_type      = DCERPC_AUTH_TYPE_SCHANNEL,
-       .start          = dcerpc_schannel_start,
-       .update         = dcerpc_schannel_update,
-       .seal           = dcerpc_schannel_seal,
-       .sign           = dcerpc_schannel_sign,
-       .check_sig      = dcerpc_schannel_check_sig,
-       .unseal         = dcerpc_schannel_unseal,
-       .session_key    = dcerpc_schannel_session_key,
-       .end            = dcerpc_schannel_end
-};
-
-const struct dcesrv_security_ops *dcerpc_schannel_security_get_ops(void)
-{
-       return &dcerpc_schannel_security_ops;
-}
-
 /*
   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
 */
+NTSTATUS dcerpc_bind_auth_schannel_withkey(struct dcerpc_pipe *p,
+                                          const char *uuid, uint_t version,
+                                          struct creds_CredentialState *creds)
+{
+       NTSTATUS status;
+       struct dcerpc_schannel_state *dce_schan_state;
+
+       status = gensec_client_start(p, &p->conn->security_state.generic_state);
+       if (!NT_STATUS_IS_OK(status)) {
+               return status;
+       }
+
+       status = gensec_set_workstation(p->conn->security_state.generic_state, creds->computer_name);
+       if (!NT_STATUS_IS_OK(status)) {
+               DEBUG(1, ("Failed to set schannel workstation to %s: %s\n", 
+                         creds->computer_name, nt_errstr(status)));
+               talloc_free(p->conn->security_state.generic_state);
+               p->conn->security_state.generic_state = NULL;
+               return status;
+       }
+       
+       status = gensec_set_username(p->conn->security_state.generic_state, creds->account_name);
+       if (!NT_STATUS_IS_OK(status)) {
+               DEBUG(1, ("Failed to set schannel username to %s: %s\n", 
+                         creds->account_name, 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, creds->domain);
+       if (!NT_STATUS_IS_OK(status)) {
+               DEBUG(1, ("Failed to set schannel domain to %s: %s\n", 
+                         creds->domain, nt_errstr(status)));
+               talloc_free(p->conn->security_state.generic_state);
+               p->conn->security_state.generic_state = NULL;
+               return status;
+       }
+       
+       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;
+}
+
 NTSTATUS dcerpc_bind_auth_schannel(struct dcerpc_pipe *p,
                                   const char *uuid, uint_t version,
+                                  const char *workstation,
                                   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, 
+                                    workstation,
+                                    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, creds);
+}
+
+static BOOL dcerpc_schannel_have_feature(struct gensec_security *gensec_security,
+                                         uint32_t feature)
+{
+       if (feature & (GENSEC_FEATURE_SESSION_KEY | 
+                      GENSEC_FEATURE_SIGN | 
+                      GENSEC_FEATURE_SEAL)) {
+               return True;
+       }
+       return False;
+}
 
-       status = dcerpc_bind_auth(p, DCERPC_AUTH_TYPE_SCHANNEL,
-                               uuid, version,
-                               domain, username, 
-                               password);
 
-       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)
+{
+       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;
+       }
+
+       return ret;
 }