librpc/rpc: pass struct dcerpc_pipe to dcerpc_auth3()
[tprouty/samba.git] / source4 / librpc / rpc / dcerpc_auth.c
index 9587fb9390ca2d0583a5213205384c3e1cdada22..661cd13c5a5054dea1f8fa49317379f2a924df91 100644 (file)
@@ -4,12 +4,12 @@
    Generic Authentication Interface
 
    Copyright (C) Andrew Tridgell 2003
-   Copyright (C) Andrew Bartlett <abartlet@samba.org> 2004
-
+   Copyright (C) Andrew Bartlett <abartlet@samba.org> 2004-2005
+   Copyright (C) Stefan Metzmacher 2004
    
    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 "libcli/composite/composite.h"
+#include "auth/gensec/gensec.h"
+#include "librpc/rpc/dcerpc.h"
+#include "librpc/rpc/dcerpc_proto.h"
+#include "param/param.h"
+
+/*
+  return the rpc syntax and transfer syntax given the pipe uuid and version
+*/
+static NTSTATUS dcerpc_init_syntaxes(const struct ndr_interface_table *table,
+                             struct ndr_syntax_id *syntax,
+                             struct ndr_syntax_id *transfer_syntax)
+{
+       syntax->uuid = table->syntax_id.uuid;
+       syntax->if_version = table->syntax_id.if_version;
+
+       *transfer_syntax = ndr_transfer_syntax;
+
+       return NT_STATUS_OK;
+}
+
 
 /*
-  do a non-athenticated dcerpc bind
+  Send request to do a non-authenticated dcerpc bind
 */
-NTSTATUS dcerpc_bind_auth_none(struct dcerpc_pipe *p,
-                              const char *uuid, uint_t version)
+struct composite_context *dcerpc_bind_auth_none_send(TALLOC_CTX *mem_ctx,
+                                                    struct dcerpc_pipe *p,
+                                                    const struct ndr_interface_table *table)
 {
-       TALLOC_CTX *mem_ctx;
-       NTSTATUS status;
+       struct ndr_syntax_id syntax;
+       struct ndr_syntax_id transfer_syntax;
 
-       mem_ctx = talloc_init("dcerpc_bind_auth_ntlm");
-       if (!mem_ctx) {
-               return NT_STATUS_NO_MEMORY;
+       struct composite_context *c;
+
+       c = composite_create(mem_ctx, p->conn->event_ctx);
+       if (c == NULL) return NULL;
+
+       c->status = dcerpc_init_syntaxes(table,
+                                        &syntax, &transfer_syntax);
+       if (!NT_STATUS_IS_OK(c->status)) {
+               DEBUG(2,("Invalid uuid string in "
+                        "dcerpc_bind_auth_none_send\n"));
+               composite_error(c, c->status);
+               return c;
        }
 
-       status = dcerpc_bind_byuuid(p, mem_ctx, uuid, version);
-       talloc_destroy(mem_ctx);
+       /* c was only allocated as a container for a possible error */
+       talloc_free(c);
+
+       return dcerpc_bind_send(p, mem_ctx, &syntax, &transfer_syntax);
+}
+
 
-       return status;
+/*
+  Receive result of a non-authenticated dcerpc bind
+*/
+NTSTATUS dcerpc_bind_auth_none_recv(struct composite_context *ctx)
+{
+       return dcerpc_bind_recv(ctx);
 }
 
-NTSTATUS dcerpc_bind_auth3(struct dcerpc_pipe *p, uint8_t auth_type, uint8_t auth_level,
-                         const char *uuid, uint_t version)
+
+/*
+  Perform sync non-authenticated dcerpc bind
+*/
+_PUBLIC_ NTSTATUS dcerpc_bind_auth_none(struct dcerpc_pipe *p,
+                              const struct ndr_interface_table *table)
 {
-       NTSTATUS status;
-       TALLOC_CTX *mem_ctx;
+       struct composite_context *ctx;
+
+       ctx = dcerpc_bind_auth_none_send(p, p, table);
+       return dcerpc_bind_auth_none_recv(ctx);
+}
+
+
+struct bind_auth_state {
+       struct dcerpc_pipe *pipe;
        DATA_BLOB credentials;
-       DATA_BLOB null_data_blob = data_blob(NULL, 0);
+       bool more_processing;   /* Is there anything more to do after the
+                                * first bind itself received? */
+};
 
-       mem_ctx = talloc_init("dcerpc_bind_auth");
-       if (!mem_ctx) {
-               return NT_STATUS_NO_MEMORY;
+static void bind_auth_recv_alter(struct composite_context *creq);
+
+static void bind_auth_next_step(struct composite_context *c)
+{
+       struct bind_auth_state *state;
+       struct dcerpc_security *sec;
+       struct composite_context *creq;
+       bool more_processing = false;
+
+       state = talloc_get_type(c->private_data, struct bind_auth_state);
+       sec = &state->pipe->conn->security_state;
+
+       /* The status value here, from GENSEC is vital to the security
+        * of the system.  Even if the other end accepts, if GENSEC
+        * claims 'MORE_PROCESSING_REQUIRED' then you must keep
+        * feeding it blobs, or else the remote host/attacker might
+        * avoid mutal authentication requirements.
+        *
+        * Likewise, you must not feed GENSEC too much (after the OK),
+        * it doesn't like that either
+        */
+
+       c->status = gensec_update(sec->generic_state, state,
+                                 sec->auth_info->credentials,
+                                 &state->credentials);
+
+       if (NT_STATUS_EQUAL(c->status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
+               more_processing = true;
+               c->status = NT_STATUS_OK;
        }
-       
-       if (!p->security_state.generic_state) {
-               status = gensec_client_start(&p->security_state.generic_state);
-               if (!NT_STATUS_IS_OK(status)) {
-                       return status;
-               }
 
-               status = gensec_start_mech_by_authtype(p->security_state.generic_state, auth_type, auth_level);
+       if (!composite_is_ok(c)) return;
 
-               if (!NT_STATUS_IS_OK(status)) {
-                       return status;
-               }
+       if (state->credentials.length == 0) {
+               composite_done(c);
+               return;
        }
 
-       p->security_state.auth_info = talloc(p, sizeof(*p->security_state.auth_info));
-       if (!p->security_state.auth_info) {
-               status = NT_STATUS_NO_MEMORY;
-               goto done;
-       }
+       sec->auth_info->credentials = state->credentials;
 
-       p->security_state.auth_info->auth_type = auth_type;
-       p->security_state.auth_info->auth_level = auth_level;
-       p->security_state.auth_info->auth_pad_length = 0;
-       p->security_state.auth_info->auth_reserved = 0;
-       p->security_state.auth_info->auth_context_id = random();
-       p->security_state.auth_info->credentials = null_data_blob;
+       if (!more_processing) {
+               /* NO reply expected, so just send it */
+               c->status = dcerpc_auth3(state->pipe, state);
+               if (!composite_is_ok(c)) return;
 
-       status = gensec_update(p->security_state.generic_state, mem_ctx,
-                              null_data_blob,
-                              &credentials);
-       
-       if (!NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
-               goto done;
+               composite_done(c);
+               return;
        }
 
-       p->security_state.auth_info->credentials = credentials;
+       /* We are demanding a reply, so use a request that will get us one */
 
-       status = dcerpc_bind_byuuid(p, mem_ctx, uuid, version);
-       if (!NT_STATUS_IS_OK(status)) {
-               goto done;
-       }
+       creq = dcerpc_alter_context_send(state->pipe, state,
+                                        &state->pipe->syntax,
+                                        &state->pipe->transfer_syntax);
+       if (composite_nomem(creq, c)) return;
 
-       status = gensec_update(p->security_state.generic_state, mem_ctx,
-                              p->security_state.auth_info->credentials,
-                              &credentials);
+       composite_continue(c, creq, bind_auth_recv_alter, c);
+}
 
-       if (!NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
-               goto done;
-       }
 
-       p->security_state.auth_info->credentials = credentials;
+static void bind_auth_recv_alter(struct composite_context *creq)
+{
+       struct composite_context *c = talloc_get_type(creq->async.private_data,
+                                                     struct composite_context);
+
+       c->status = dcerpc_alter_context_recv(creq);
+       if (!composite_is_ok(c)) return;
 
-       status = dcerpc_auth3(p, mem_ctx);
-done:
-       talloc_destroy(mem_ctx);
+       bind_auth_next_step(c);
+}
 
-       if (!NT_STATUS_IS_OK(status)) {
-               ZERO_STRUCT(p->security_state);
+
+static void bind_auth_recv_bindreply(struct composite_context *creq)
+{
+       struct composite_context *c = talloc_get_type(creq->async.private_data,
+                                                     struct composite_context);
+       struct bind_auth_state *state = talloc_get_type(c->private_data,
+                                                       struct bind_auth_state);
+
+       c->status = dcerpc_bind_recv(creq);
+       if (!composite_is_ok(c)) return;
+
+       if (!state->more_processing) {
+               /* The first gensec_update has not requested a second run, so
+                * we're done here. */
+               composite_done(c);
+               return;
        }
 
-       return status;
+       bind_auth_next_step(c);
 }
 
-NTSTATUS dcerpc_bind_alter(struct dcerpc_pipe *p, uint8_t auth_type, uint8_t auth_level,
-                         const char *uuid, uint_t version)
+
+/**
+   Bind to a DCE/RPC pipe, send async request
+   @param mem_ctx TALLOC_CTX for the allocation of the composite_context
+   @param p The dcerpc_pipe to bind (must already be connected)
+   @param table The interface table to use (the DCE/RPC bind both selects and interface and authenticates)
+   @param credentials The credentials of the account to connect with 
+   @param auth_type Select the authentication scheme to use
+   @param auth_level Chooses between unprotected (connect), signed or sealed
+   @param service The service (used by Kerberos to select the service principal to contact)
+   @retval A composite context describing the partial state of the bind
+*/
+
+struct composite_context *dcerpc_bind_auth_send(TALLOC_CTX *mem_ctx,
+                                               struct dcerpc_pipe *p,
+                                               const struct ndr_interface_table *table,
+                                               struct cli_credentials *credentials,
+                                               struct loadparm_context *lp_ctx,
+                                               uint8_t auth_type, uint8_t auth_level,
+                                               const char *service)
 {
-       NTSTATUS status;
-       TALLOC_CTX *mem_ctx;
-       DATA_BLOB credentials;
-       DATA_BLOB null_data_blob = data_blob(NULL, 0);
+       struct composite_context *c, *creq;
+       struct bind_auth_state *state;
+       struct dcerpc_security *sec;
 
-       mem_ctx = talloc_init("dcerpc_bind_auth");
-       if (!mem_ctx) {
-               return NT_STATUS_NO_MEMORY;
+       struct ndr_syntax_id syntax, transfer_syntax;
+
+       /* composite context allocation and setup */
+       c = composite_create(mem_ctx, p->conn->event_ctx);
+       if (c == NULL) return NULL;
+
+       state = talloc(c, struct bind_auth_state);
+       if (composite_nomem(state, c)) return c;
+       c->private_data = state;
+
+       state->pipe = p;
+
+       c->status = dcerpc_init_syntaxes(table,
+                                        &syntax,
+                                        &transfer_syntax);
+       if (!composite_is_ok(c)) return c;
+
+       sec = &p->conn->security_state;
+
+       c->status = gensec_client_start(p, &sec->generic_state,
+                                       p->conn->event_ctx,
+                                       lp_ctx);
+       if (!NT_STATUS_IS_OK(c->status)) {
+               DEBUG(1, ("Failed to start GENSEC client mode: %s\n",
+                         nt_errstr(c->status)));
+               composite_error(c, c->status);
+               return c;
        }
-       
-       if (!p->security_state.generic_state) {
-               status = gensec_client_start(&p->security_state.generic_state);
-               if (!NT_STATUS_IS_OK(status)) {
-                       return status;
-               }
 
-               status = gensec_start_mech_by_authtype(p->security_state.generic_state, 
-                                                      auth_type, auth_level);
+       c->status = gensec_set_credentials(sec->generic_state, credentials);
+       if (!NT_STATUS_IS_OK(c->status)) {
+               DEBUG(1, ("Failed to set GENSEC client credentials: %s\n",
+                         nt_errstr(c->status)));
+               composite_error(c, c->status);
+               return c;
+       }
 
-               if (!NT_STATUS_IS_OK(status)) {
-                       return status;
-               }
+       c->status = gensec_set_target_hostname(sec->generic_state,
+                                              p->conn->transport.target_hostname(p->conn));
+       if (!NT_STATUS_IS_OK(c->status)) {
+               DEBUG(1, ("Failed to set GENSEC target hostname: %s\n", 
+                         nt_errstr(c->status)));
+               composite_error(c, c->status);
+               return c;
        }
 
-       p->security_state.auth_info = talloc(p, sizeof(*p->security_state.auth_info));
-       if (!p->security_state.auth_info) {
-               status = NT_STATUS_NO_MEMORY;
-               goto done;
+       if (service != NULL) {
+               c->status = gensec_set_target_service(sec->generic_state,
+                                                     service);
+               if (!NT_STATUS_IS_OK(c->status)) {
+                       DEBUG(1, ("Failed to set GENSEC target service: %s\n",
+                                 nt_errstr(c->status)));
+                       composite_error(c, c->status);
+                       return c;
+               }
        }
 
-       p->security_state.auth_info->auth_type = auth_type;
-       p->security_state.auth_info->auth_level = auth_level;
-       p->security_state.auth_info->auth_pad_length = 0;
-       p->security_state.auth_info->auth_reserved = 0;
-       p->security_state.auth_info->auth_context_id = random();
-       p->security_state.auth_info->credentials = null_data_blob;
+       c->status = gensec_start_mech_by_authtype(sec->generic_state,
+                                                 auth_type, auth_level);
+       if (!NT_STATUS_IS_OK(c->status)) {
+               DEBUG(1, ("Failed to start GENSEC client mechanism %s: %s\n",
+                         gensec_get_name_by_authtype(auth_type),
+                         nt_errstr(c->status)));
+               composite_error(c, c->status);
+               return c;
+       }
 
-       status = gensec_update(p->security_state.generic_state, mem_ctx,
-                              null_data_blob,
-                              &credentials);
-       
-       if (!NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
-               goto done;
+       sec->auth_info = talloc(p, struct dcerpc_auth);
+       if (composite_nomem(sec->auth_info, c)) return c;
+
+       sec->auth_info->auth_type = auth_type;
+       sec->auth_info->auth_level = auth_level,
+       sec->auth_info->auth_pad_length = 0;
+       sec->auth_info->auth_reserved = 0;
+       sec->auth_info->auth_context_id = random();
+       sec->auth_info->credentials = data_blob(NULL, 0);
+
+       /* The status value here, from GENSEC is vital to the security
+        * of the system.  Even if the other end accepts, if GENSEC
+        * claims 'MORE_PROCESSING_REQUIRED' then you must keep
+        * feeding it blobs, or else the remote host/attacker might
+        * avoid mutal authentication requirements.
+        *
+        * Likewise, you must not feed GENSEC too much (after the OK),
+        * it doesn't like that either
+        */
+
+       c->status = gensec_update(sec->generic_state, state,
+                                 sec->auth_info->credentials,
+                                 &state->credentials);
+       if (!NT_STATUS_IS_OK(c->status) &&
+           !NT_STATUS_EQUAL(c->status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
+               composite_error(c, c->status);
+               return c;
        }
 
-       p->security_state.auth_info->credentials = credentials;
+       state->more_processing = NT_STATUS_EQUAL(c->status,
+                                                NT_STATUS_MORE_PROCESSING_REQUIRED);
 
-       status = dcerpc_bind_byuuid(p, mem_ctx, uuid, version);
-       if (!NT_STATUS_IS_OK(status)) {
-               goto done;
+       if (state->credentials.length == 0) {
+               composite_done(c);
+               return c;
        }
 
-       while(1) {
-               status = gensec_update(p->security_state.generic_state, mem_ctx,
-                              p->security_state.auth_info->credentials,
-                              &credentials);
+       sec->auth_info->credentials = state->credentials;
 
-               if (!NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
-                       goto done;
-               }
+       /* The first request always is a dcerpc_bind. The subsequent ones
+        * depend on gensec results */
+       creq = dcerpc_bind_send(p, state, &syntax, &transfer_syntax);
+       if (composite_nomem(creq, c)) return c;
 
-               p->security_state.auth_info->credentials = credentials;
+       composite_continue(c, creq, bind_auth_recv_bindreply, c);
+       return c;
+}
 
-               status = dcerpc_alter(p, mem_ctx);
-               if (!NT_STATUS_IS_OK(status)) {
-                       goto done;
-               }
-       }
 
-done:
-       talloc_destroy(mem_ctx);
+/**
+   Bind to a DCE/RPC pipe, receive result
+   @param creq A composite context describing state of async call
+   @retval NTSTATUS code
+*/
 
-       if (!NT_STATUS_IS_OK(status)) {
-               ZERO_STRUCT(p->security_state);
+NTSTATUS dcerpc_bind_auth_recv(struct composite_context *creq)
+{
+       NTSTATUS result = composite_wait(creq);
+       struct bind_auth_state *state = talloc_get_type(creq->private_data,
+                                                       struct bind_auth_state);
+
+       if (NT_STATUS_IS_OK(result)) {
+               /*
+                 after a successful authenticated bind the session
+                 key reverts to the generic session key
+               */
+               state->pipe->conn->security_state.session_key = dcerpc_generic_session_key;
        }
+       
+       talloc_free(creq);
+       return result;
+}
 
-       return status;
+
+/**
+   Perform a GENSEC authenticated bind to a DCE/RPC pipe, sync
+   @param p The dcerpc_pipe to bind (must already be connected)
+   @param table The interface table to use (the DCE/RPC bind both selects and interface and authenticates)
+   @param credentials The credentials of the account to connect with 
+   @param auth_type Select the authentication scheme to use
+   @param auth_level Chooses between unprotected (connect), signed or sealed
+   @param service The service (used by Kerberos to select the service principal to contact)
+   @retval NTSTATUS status code
+*/
+
+_PUBLIC_ NTSTATUS dcerpc_bind_auth(struct dcerpc_pipe *p,
+                         const struct ndr_interface_table *table,
+                         struct cli_credentials *credentials,
+                         struct loadparm_context *lp_ctx,
+                         uint8_t auth_type, uint8_t auth_level,
+                         const char *service)
+{
+       struct composite_context *creq;
+       creq = dcerpc_bind_auth_send(p, p, table, credentials, lp_ctx,
+                                    auth_type, auth_level, service);
+       return dcerpc_bind_auth_recv(creq);
 }