s3-librpc Call GSSAPI via the auth_generic layer and gensec
[metze/samba/wip.git] / source3 / librpc / rpc / dcerpc_helpers.c
index fc9915cf0843c2bc980d2620212658a75c8d6bf0..4cfe7933fe0c95002008aa6e9e6178ca52b306d3 100644 (file)
@@ -24,8 +24,9 @@
 #include "librpc/gen_ndr/ndr_schannel.h"
 #include "../libcli/auth/schannel.h"
 #include "../libcli/auth/spnego.h"
-#include "../libcli/auth/ntlmssp.h"
-#include "ntlmssp_wrap.h"
+#include "librpc/crypto/gse.h"
+#include "librpc/crypto/spnego.h"
+#include "auth/gensec/gensec.h"
 
 #undef DBGC_CLASS
 #define DBGC_CLASS DBGC_RPC_PARSE
@@ -91,6 +92,7 @@ NTSTATUS dcerpc_push_ncacn_packet(TALLOC_CTX *mem_ctx,
 *                      elements
 * @param blob          The blob of data to decode
 * @param r             An empty ncacn_packet, must not be NULL
+* @param bigendian     Whether the packet is bignedian encoded
 *
 * @return a NTSTATUS error code
 */
@@ -239,11 +241,141 @@ NTSTATUS dcerpc_pull_dcerpc_auth(TALLOC_CTX *mem_ctx,
        return NT_STATUS_OK;
 }
 
+/**
+* @brief Calculate how much data we can in a packet, including calculating
+*       auth token and pad lengths.
+*
+* @param auth          The pipe_auth_data structure for this pipe.
+* @param header_len    The length of the packet header
+* @param data_left     The data left in the send buffer
+* @param max_xmit_frag The max fragment size.
+* @param pad_alignment The NDR padding size.
+* @param data_to_send  [out] The max data we will send in the pdu
+* @param frag_len      [out] The total length of the fragment
+* @param auth_len      [out] The length of the auth trailer
+* @param pad_len       [out] The padding to be applied
+*
+* @return A NT Error status code.
+*/
+NTSTATUS dcerpc_guess_sizes(struct pipe_auth_data *auth,
+                           size_t header_len, size_t data_left,
+                           size_t max_xmit_frag, size_t pad_alignment,
+                           size_t *data_to_send, size_t *frag_len,
+                           size_t *auth_len, size_t *pad_len)
+{
+       size_t max_len;
+       size_t mod_len;
+       struct gensec_security *gensec_security;
+       struct schannel_state *schannel_auth;
+       struct spnego_context *spnego_ctx;
+       struct gse_context *gse_ctx;
+       enum spnego_mech auth_type;
+       void *auth_ctx;
+       bool seal = false;
+       NTSTATUS status;
+
+       /* no auth token cases first */
+       switch (auth->auth_level) {
+       case DCERPC_AUTH_LEVEL_NONE:
+       case DCERPC_AUTH_LEVEL_CONNECT:
+       case DCERPC_AUTH_LEVEL_PACKET:
+               max_len = max_xmit_frag - header_len;
+               *data_to_send = MIN(max_len, data_left);
+               *pad_len = 0;
+               *auth_len = 0;
+               *frag_len = header_len + *data_to_send;
+               return NT_STATUS_OK;
+
+       case DCERPC_AUTH_LEVEL_PRIVACY:
+               seal = true;
+               break;
+
+       case DCERPC_AUTH_LEVEL_INTEGRITY:
+               break;
+
+       default:
+               return NT_STATUS_INVALID_PARAMETER;
+       }
+
+
+       /* Sign/seal case, calculate auth and pad lengths */
+
+       max_len = max_xmit_frag - header_len - DCERPC_AUTH_TRAILER_LENGTH;
+
+       /* Treat the same for all authenticated rpc requests. */
+       switch (auth->auth_type) {
+       case DCERPC_AUTH_TYPE_SPNEGO:
+               spnego_ctx = talloc_get_type_abort(auth->auth_ctx,
+                                                  struct spnego_context);
+               status = spnego_get_negotiated_mech(spnego_ctx,
+                                                   &auth_type, &auth_ctx);
+               if (!NT_STATUS_IS_OK(status)) {
+                       return status;
+               }
+               switch (auth_type) {
+               case SPNEGO_NTLMSSP:
+                       gensec_security = talloc_get_type_abort(auth_ctx,
+                                                               struct gensec_security);
+                       *auth_len = gensec_sig_size(gensec_security, max_len);
+                       break;
+
+               case SPNEGO_KRB5:
+                       gse_ctx = talloc_get_type_abort(auth_ctx,
+                                                       struct gse_context);
+                       if (!gse_ctx) {
+                               return NT_STATUS_INVALID_PARAMETER;
+                       }
+                       *auth_len = gse_get_signature_length(gse_ctx,
+                                                            seal, max_len);
+                       break;
+
+               default:
+                       return NT_STATUS_INVALID_PARAMETER;
+               }
+               break;
+
+       case DCERPC_AUTH_TYPE_NTLMSSP:
+       case DCERPC_AUTH_TYPE_KRB5:
+               gensec_security = talloc_get_type_abort(auth->auth_ctx,
+                                                       struct gensec_security);
+               *auth_len = gensec_sig_size(gensec_security, max_len);
+               break;
+
+       case DCERPC_AUTH_TYPE_SCHANNEL:
+               schannel_auth = talloc_get_type_abort(auth->auth_ctx,
+                                                     struct schannel_state);
+               *auth_len = netsec_outgoing_sig_size(schannel_auth);
+               break;
+       default:
+               return NT_STATUS_INVALID_PARAMETER;
+       }
+
+       max_len -= *auth_len;
+
+       *data_to_send = MIN(max_len, data_left);
+
+       mod_len = (header_len + *data_to_send) % pad_alignment;
+       if (mod_len) {
+               *pad_len = pad_alignment - mod_len;
+       } else {
+               *pad_len = 0;
+       }
+
+       if (*data_to_send + *pad_len > max_len) {
+               *data_to_send -= pad_alignment;
+       }
+
+       *frag_len = header_len + *data_to_send + *pad_len
+                       + DCERPC_AUTH_TRAILER_LENGTH + *auth_len;
+
+       return NT_STATUS_OK;
+}
+
 /*******************************************************************
  Create and add the NTLMSSP sign/seal auth data.
  ********************************************************************/
 
-static NTSTATUS add_ntlmssp_auth_footer(struct auth_ntlmssp_state *auth_state,
+static NTSTATUS add_generic_auth_footer(struct gensec_security *gensec_security,
                                        enum dcerpc_AuthLevel auth_level,
                                        DATA_BLOB *rpc_out)
 {
@@ -253,21 +385,21 @@ static NTSTATUS add_ntlmssp_auth_footer(struct auth_ntlmssp_state *auth_state,
        DATA_BLOB auth_blob;
        NTSTATUS status;
 
-       if (!auth_state) {
+       if (!gensec_security) {
                return NT_STATUS_INVALID_PARAMETER;
        }
 
        switch (auth_level) {
        case DCERPC_AUTH_LEVEL_PRIVACY:
                /* Data portion is encrypted. */
-               status = auth_ntlmssp_seal_packet(auth_state,
-                                            rpc_out->data,
-                                            rpc_out->data
-                                               + DCERPC_RESPONSE_LENGTH,
-                                            data_and_pad_len,
-                                            rpc_out->data,
-                                            rpc_out->length,
-                                            &auth_blob);
+               status = gensec_seal_packet(gensec_security,
+                                           rpc_out->data,
+                                           rpc_out->data
+                                           + DCERPC_RESPONSE_LENGTH,
+                                           data_and_pad_len,
+                                           rpc_out->data,
+                                           rpc_out->length,
+                                           &auth_blob);
                if (!NT_STATUS_IS_OK(status)) {
                        return status;
                }
@@ -275,14 +407,14 @@ static NTSTATUS add_ntlmssp_auth_footer(struct auth_ntlmssp_state *auth_state,
 
        case DCERPC_AUTH_LEVEL_INTEGRITY:
                /* Data is signed. */
-               status = auth_ntlmssp_sign_packet(auth_state,
-                                            rpc_out->data,
-                                            rpc_out->data
-                                               + DCERPC_RESPONSE_LENGTH,
-                                            data_and_pad_len,
-                                            rpc_out->data,
-                                            rpc_out->length,
-                                            &auth_blob);
+               status = gensec_sign_packet(gensec_security,
+                                           rpc_out->data,
+                                           rpc_out->data
+                                           + DCERPC_RESPONSE_LENGTH,
+                                           data_and_pad_len,
+                                           rpc_out->data,
+                                           rpc_out->length,
+                                           &auth_blob);
                if (!NT_STATUS_IS_OK(status)) {
                        return status;
                }
@@ -307,6 +439,39 @@ static NTSTATUS add_ntlmssp_auth_footer(struct auth_ntlmssp_state *auth_state,
        return NT_STATUS_OK;
 }
 
+/*******************************************************************
+ Check/unseal the NTLMSSP auth data. (Unseal in place).
+ ********************************************************************/
+
+static NTSTATUS get_generic_auth_footer(struct gensec_security *gensec_security,
+                                       enum dcerpc_AuthLevel auth_level,
+                                       DATA_BLOB *data, DATA_BLOB *full_pkt,
+                                       DATA_BLOB *auth_token)
+{
+       switch (auth_level) {
+       case DCERPC_AUTH_LEVEL_PRIVACY:
+               /* Data portion is encrypted. */
+               return gensec_unseal_packet(gensec_security,
+                                           data->data,
+                                           data->length,
+                                           full_pkt->data,
+                                           full_pkt->length,
+                                           auth_token);
+
+       case DCERPC_AUTH_LEVEL_INTEGRITY:
+               /* Data is signed. */
+               return gensec_check_packet(gensec_security,
+                                          data->data,
+                                          data->length,
+                                          full_pkt->data,
+                                          full_pkt->length,
+                                          auth_token);
+
+       default:
+               return NT_STATUS_INVALID_PARAMETER;
+       }
+}
+
 /*******************************************************************
  Create and add the schannel sign/seal auth data.
  ********************************************************************/
@@ -371,6 +536,124 @@ static NTSTATUS add_schannel_auth_footer(struct schannel_state *sas,
        return NT_STATUS_OK;
 }
 
+/*******************************************************************
+ Check/unseal the Schannel auth data. (Unseal in place).
+ ********************************************************************/
+
+static NTSTATUS get_schannel_auth_footer(TALLOC_CTX *mem_ctx,
+                                        struct schannel_state *auth_state,
+                                        enum dcerpc_AuthLevel auth_level,
+                                        DATA_BLOB *data, DATA_BLOB *full_pkt,
+                                        DATA_BLOB *auth_token)
+{
+       switch (auth_level) {
+       case DCERPC_AUTH_LEVEL_PRIVACY:
+               /* Data portion is encrypted. */
+               return netsec_incoming_packet(auth_state,
+                                               true,
+                                               data->data,
+                                               data->length,
+                                               auth_token);
+
+       case DCERPC_AUTH_LEVEL_INTEGRITY:
+               /* Data is signed. */
+               return netsec_incoming_packet(auth_state,
+                                               false,
+                                               data->data,
+                                               data->length,
+                                               auth_token);
+
+       default:
+               return NT_STATUS_INVALID_PARAMETER;
+       }
+}
+
+/*******************************************************************
+ Create and add the spnego-negotiated sign/seal auth data.
+ ********************************************************************/
+
+static NTSTATUS add_spnego_auth_footer(struct spnego_context *spnego_ctx,
+                                       enum dcerpc_AuthLevel auth_level,
+                                       DATA_BLOB *rpc_out)
+{
+       DATA_BLOB auth_blob;
+       DATA_BLOB rpc_data;
+       NTSTATUS status;
+
+       if (!spnego_ctx) {
+               return NT_STATUS_INVALID_PARAMETER;
+       }
+
+       rpc_data = data_blob_const(rpc_out->data
+                                       + DCERPC_RESPONSE_LENGTH,
+                                  rpc_out->length
+                                       - DCERPC_RESPONSE_LENGTH
+                                       - DCERPC_AUTH_TRAILER_LENGTH);
+
+       switch (auth_level) {
+       case DCERPC_AUTH_LEVEL_PRIVACY:
+               /* Data portion is encrypted. */
+               status = spnego_seal(rpc_out->data, spnego_ctx,
+                                    &rpc_data, rpc_out, &auth_blob);
+               break;
+
+               if (!NT_STATUS_IS_OK(status)) {
+                       return status;
+               }
+               break;
+
+       case DCERPC_AUTH_LEVEL_INTEGRITY:
+               /* Data is signed. */
+               status = spnego_sign(rpc_out->data, spnego_ctx,
+                                    &rpc_data, rpc_out, &auth_blob);
+               break;
+
+               if (!NT_STATUS_IS_OK(status)) {
+                       return status;
+               }
+               break;
+
+       default:
+               /* Can't happen. */
+               smb_panic("bad auth level");
+               /* Notreached. */
+               return NT_STATUS_INVALID_PARAMETER;
+       }
+
+       /* Finally attach the blob. */
+       if (!data_blob_append(NULL, rpc_out,
+                               auth_blob.data, auth_blob.length)) {
+               DEBUG(0, ("Failed to add %u bytes auth blob.\n",
+                         (unsigned int)auth_blob.length));
+               return NT_STATUS_NO_MEMORY;
+       }
+       data_blob_free(&auth_blob);
+
+       return NT_STATUS_OK;
+}
+
+static NTSTATUS get_spnego_auth_footer(TALLOC_CTX *mem_ctx,
+                                       struct spnego_context *sp_ctx,
+                                       enum dcerpc_AuthLevel auth_level,
+                                       DATA_BLOB *data, DATA_BLOB *full_pkt,
+                                       DATA_BLOB *auth_token)
+{
+       switch (auth_level) {
+       case DCERPC_AUTH_LEVEL_PRIVACY:
+               /* Data portion is encrypted. */
+               return spnego_unseal(mem_ctx, sp_ctx,
+                                    data, full_pkt, auth_token);
+
+       case DCERPC_AUTH_LEVEL_INTEGRITY:
+               /* Data is signed. */
+               return spnego_sigcheck(mem_ctx, sp_ctx,
+                                      data, full_pkt, auth_token);
+
+       default:
+               return NT_STATUS_INVALID_PARAMETER;
+       }
+}
+
 /**
 * @brief   Append an auth footer according to what is the current mechanism
 *
@@ -383,13 +666,16 @@ static NTSTATUS add_schannel_auth_footer(struct schannel_state *sas,
 NTSTATUS dcerpc_add_auth_footer(struct pipe_auth_data *auth,
                                size_t pad_len, DATA_BLOB *rpc_out)
 {
-       enum dcerpc_AuthType auth_type;
+       struct schannel_state *schannel_auth;
+       struct gensec_security *gensec_security;
+       struct spnego_context *spnego_ctx;
        char pad[CLIENT_NDR_PADDING_SIZE] = { 0, };
        DATA_BLOB auth_info;
        DATA_BLOB auth_blob;
        NTSTATUS status;
 
-       if (auth->auth_type == PIPE_AUTH_TYPE_NONE) {
+       if (auth->auth_type == DCERPC_AUTH_TYPE_NONE ||
+           auth->auth_type == DCERPC_AUTH_TYPE_NCALRPC_AS_SYSTEM) {
                return NT_STATUS_OK;
        }
 
@@ -400,14 +686,12 @@ NTSTATUS dcerpc_add_auth_footer(struct pipe_auth_data *auth,
                }
        }
 
-       auth_type = map_pipe_auth_type_to_rpc_auth_type(auth->auth_type);
-
        /* marshall the dcerpc_auth with an actually empty auth_blob.
         * This is needed because the ntmlssp signature includes the
         * auth header. We will append the actual blob later. */
        auth_blob = data_blob_null;
        status = dcerpc_push_dcerpc_auth(rpc_out->data,
-                                        auth_type,
+                                        auth->auth_type,
                                         auth->auth_level,
                                         pad_len,
                                         1 /* context id. */,
@@ -428,17 +712,28 @@ NTSTATUS dcerpc_add_auth_footer(struct pipe_auth_data *auth,
 
        /* Generate any auth sign/seal and add the auth footer. */
        switch (auth->auth_type) {
-       case PIPE_AUTH_TYPE_NONE:
+       case DCERPC_AUTH_TYPE_NONE:
+       case DCERPC_AUTH_TYPE_NCALRPC_AS_SYSTEM:
                status = NT_STATUS_OK;
                break;
-       case PIPE_AUTH_TYPE_NTLMSSP:
-       case PIPE_AUTH_TYPE_SPNEGO_NTLMSSP:
-               status = add_ntlmssp_auth_footer(auth->a_u.auth_ntlmssp_state,
+       case DCERPC_AUTH_TYPE_SPNEGO:
+               spnego_ctx = talloc_get_type_abort(auth->auth_ctx,
+                                                  struct spnego_context);
+               status = add_spnego_auth_footer(spnego_ctx,
+                                               auth->auth_level, rpc_out);
+               break;
+       case DCERPC_AUTH_TYPE_KRB5:
+       case DCERPC_AUTH_TYPE_NTLMSSP:
+               gensec_security = talloc_get_type_abort(auth->auth_ctx,
+                                               struct gensec_security);
+               status = add_generic_auth_footer(gensec_security,
                                                 auth->auth_level,
                                                 rpc_out);
                break;
-       case PIPE_AUTH_TYPE_SCHANNEL:
-               status = add_schannel_auth_footer(auth->a_u.schannel_auth,
+       case DCERPC_AUTH_TYPE_SCHANNEL:
+               schannel_auth = talloc_get_type_abort(auth->auth_ctx,
+                                                     struct schannel_state);
+               status = add_schannel_auth_footer(schannel_auth,
                                                  auth->auth_level,
                                                  rpc_out);
                break;
@@ -469,6 +764,9 @@ NTSTATUS dcerpc_check_auth(struct pipe_auth_data *auth,
                           DATA_BLOB *raw_pkt,
                           size_t *pad_len)
 {
+       struct schannel_state *schannel_auth;
+       struct gensec_security *gensec_security;
+       struct spnego_context *spnego_ctx;
        NTSTATUS status;
        struct dcerpc_auth auth_info;
        uint32_t auth_length;
@@ -510,9 +808,9 @@ NTSTATUS dcerpc_check_auth(struct pipe_auth_data *auth,
        if (pkt->auth_length > pkt->frag_length) {
                return NT_STATUS_INFO_LENGTH_MISMATCH;
        }
-       if ((pkt->auth_length
-            + DCERPC_AUTH_TRAILER_LENGTH < pkt->auth_length) ||
-           (pkt->auth_length
+       if (((unsigned int)pkt->auth_length
+            + DCERPC_AUTH_TRAILER_LENGTH < (unsigned int)pkt->auth_length) ||
+           ((unsigned int)pkt->auth_length
             + DCERPC_AUTH_TRAILER_LENGTH < DCERPC_AUTH_TRAILER_LENGTH)) {
                /* Integer wrap attempt. */
                return NT_STATUS_INFO_LENGTH_MISMATCH;
@@ -530,83 +828,50 @@ NTSTATUS dcerpc_check_auth(struct pipe_auth_data *auth,
                                raw_pkt->length - auth_info.credentials.length);
 
        switch (auth->auth_type) {
-       case PIPE_AUTH_TYPE_NONE:
+       case DCERPC_AUTH_TYPE_NONE:
+       case DCERPC_AUTH_TYPE_NCALRPC_AS_SYSTEM:
                return NT_STATUS_OK;
 
-       case PIPE_AUTH_TYPE_SPNEGO_NTLMSSP:
-       case PIPE_AUTH_TYPE_NTLMSSP:
-
-               DEBUG(10, ("NTLMSSP auth\n"));
-
-               if (!auth->a_u.auth_ntlmssp_state) {
-                       DEBUG(0, ("Invalid auth level, "
-                                 "failed to process packet auth.\n"));
-                       return NT_STATUS_INVALID_PARAMETER;
+       case DCERPC_AUTH_TYPE_SPNEGO:
+               spnego_ctx = talloc_get_type_abort(auth->auth_ctx,
+                                                  struct spnego_context);
+               status = get_spnego_auth_footer(pkt, spnego_ctx,
+                                               auth->auth_level,
+                                               &data, &full_pkt,
+                                               &auth_info.credentials);
+               if (!NT_STATUS_IS_OK(status)) {
+                       return status;
                }
+               break;
 
-               switch (auth->auth_level) {
-               case DCERPC_AUTH_LEVEL_PRIVACY:
-                       status = auth_ntlmssp_unseal_packet(
-                                       auth->a_u.auth_ntlmssp_state,
-                                       data.data, data.length,
-                                       full_pkt.data, full_pkt.length,
-                                       &auth_info.credentials);
-                       if (!NT_STATUS_IS_OK(status)) {
-                               return status;
-                       }
-                       memcpy(pkt_trailer->data, data.data, data.length);
-                       break;
+       case DCERPC_AUTH_TYPE_KRB5:
+       case DCERPC_AUTH_TYPE_NTLMSSP:
 
-               case DCERPC_AUTH_LEVEL_INTEGRITY:
-                       status = auth_ntlmssp_check_packet(
-                                       auth->a_u.auth_ntlmssp_state,
-                                       data.data, data.length,
-                                       full_pkt.data, full_pkt.length,
-                                       &auth_info.credentials);
-                       if (!NT_STATUS_IS_OK(status)) {
-                               return status;
-                       }
-                       break;
+               DEBUG(10, ("GENSEC auth\n"));
 
-               default:
-                       DEBUG(0, ("Invalid auth level, "
-                                 "failed to process packet auth.\n"));
-                       return NT_STATUS_INVALID_PARAMETER;
+               gensec_security = talloc_get_type_abort(auth->auth_ctx,
+                                               struct gensec_security);
+               status = get_generic_auth_footer(gensec_security,
+                                                auth->auth_level,
+                                                &data, &full_pkt,
+                                                &auth_info.credentials);
+               if (!NT_STATUS_IS_OK(status)) {
+                       return status;
                }
                break;
 
-       case PIPE_AUTH_TYPE_SCHANNEL:
+       case DCERPC_AUTH_TYPE_SCHANNEL:
 
                DEBUG(10, ("SCHANNEL auth\n"));
 
-               switch (auth->auth_level) {
-               case DCERPC_AUTH_LEVEL_PRIVACY:
-                       status = netsec_incoming_packet(
-                                       auth->a_u.schannel_auth,
-                                       pkt, true,
-                                       data.data, data.length,
-                                       &auth_info.credentials);
-                       if (!NT_STATUS_IS_OK(status)) {
-                               return status;
-                       }
-                       memcpy(pkt_trailer->data, data.data, data.length);
-                       break;
-
-               case DCERPC_AUTH_LEVEL_INTEGRITY:
-                       status = netsec_incoming_packet(
-                                       auth->a_u.schannel_auth,
-                                       pkt, false,
-                                       data.data, data.length,
-                                       &auth_info.credentials);
-                       if (!NT_STATUS_IS_OK(status)) {
-                               return status;
-                       }
-                       break;
-
-               default:
-                       DEBUG(0, ("Invalid auth level, "
-                                 "failed to process packet auth.\n"));
-                       return NT_STATUS_INVALID_PARAMETER;
+               schannel_auth = talloc_get_type_abort(auth->auth_ctx,
+                                                     struct schannel_state);
+               status = get_schannel_auth_footer(pkt, schannel_auth,
+                                                 auth->auth_level,
+                                                 &data, &full_pkt,
+                                                 &auth_info.credentials);
+               if (!NT_STATUS_IS_OK(status)) {
+                       return status;
                }
                break;
 
@@ -617,6 +882,14 @@ NTSTATUS dcerpc_check_auth(struct pipe_auth_data *auth,
                return NT_STATUS_INVALID_PARAMETER;
        }
 
+       /* TODO: remove later
+        * this is still needed because in the server code the
+        * pkt_trailer actually has a copy of the raw data, and they
+        * are still both used in later calls */
+       if (auth->auth_level == DCERPC_AUTH_LEVEL_PRIVACY) {
+               memcpy(pkt_trailer->data, data.data, data.length);
+       }
+
        *pad_len = auth_info.auth_pad_length;
        data_blob_free(&auth_info.credentials);
        return NT_STATUS_OK;