auth/ntlmssp/gensec_ntlmssp.c

   1 /*
   2  *  Unix SMB/CIFS implementation.
   3  *  Version 3.0
   4  *  NTLMSSP Signing routines
   5  *  Copyright (C) Luke Kenneth Casson Leighton 1996-2001
   6  *  Copyright (C) Andrew Bartlett <abartlet@samba.org> 2003-2005
   7  *
   8  *  This program is free software; you can redistribute it and/or modify
   9  *  it under the terms of the GNU General Public License as published by
  10  *  the Free Software Foundation; either version 3 of the License, or
  11  *  (at your option) any later version.
  12  *
  13  *  This program is distributed in the hope that it will be useful,
  14  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16  *  GNU General Public License for more details.
  17  *
  18  *  You should have received a copy of the GNU General Public License
  19  *  along with this program; if not, see <http://www.gnu.org/licenses/>.
  20  */
  21
  22 #include "includes.h"
  23 #include "auth/ntlmssp/ntlmssp.h"
  24 #include "auth/gensec/gensec.h"
  25 #include "auth/gensec/gensec_internal.h"
  26 #include "auth/ntlmssp/ntlmssp_private.h"
  27
  28 NTSTATUS gensec_ntlmssp_magic(struct gensec_security *gensec_security,
  29                               const DATA_BLOB *first_packet)
  30 {
  31         if (ntlmssp_blob_matches_magic(first_packet)) {
  32                 return NT_STATUS_OK;
  33         } else {
  34                 return NT_STATUS_INVALID_PARAMETER;
  35         }
  36 }
  37
  38 /**
  39  * Return the NTLMSSP master session key
  40  *
  41  * @param ntlmssp_state NTLMSSP State
  42  */
  43
  44 NTSTATUS gensec_ntlmssp_session_key(struct gensec_security *gensec_security,
  45                                     TALLOC_CTX *mem_ctx,
  46                                     DATA_BLOB *session_key)
  47 {
  48         struct gensec_ntlmssp_context *gensec_ntlmssp =
  49                 talloc_get_type_abort(gensec_security->private_data,
  50                                       struct gensec_ntlmssp_context);
  51         struct ntlmssp_state *ntlmssp_state = gensec_ntlmssp->ntlmssp_state;
  52
  53         if (ntlmssp_state->expected_state != NTLMSSP_DONE) {
  54                 return NT_STATUS_NO_USER_SESSION_KEY;
  55         }
  56
  57         if (!ntlmssp_state->session_key.data) {
  58                 return NT_STATUS_NO_USER_SESSION_KEY;
  59         }
  60         *session_key = data_blob_talloc(mem_ctx, ntlmssp_state->session_key.data, ntlmssp_state->session_key.length);
  61         if (!session_key->data) {
  62                 return NT_STATUS_NO_MEMORY;
  63         }
  64
  65         return NT_STATUS_OK;
  66 }
  67
  68 bool gensec_ntlmssp_have_feature(struct gensec_security *gensec_security,
  69                                  uint32_t feature)
  70 {
  71         struct gensec_ntlmssp_context *gensec_ntlmssp =
  72                 talloc_get_type_abort(gensec_security->private_data,
  73                                       struct gensec_ntlmssp_context);
  74         struct ntlmssp_state *ntlmssp_state = gensec_ntlmssp->ntlmssp_state;
  75
  76         if (feature & GENSEC_FEATURE_SIGN) {
  77                 if (!ntlmssp_state->session_key.length) {
  78                         return false;
  79                 }
  80                 if (ntlmssp_state->neg_flags & NTLMSSP_NEGOTIATE_SIGN) {
  81                         return true;
  82                 }
  83         }
  84         if (feature & GENSEC_FEATURE_SEAL) {
  85                 if (!ntlmssp_state->session_key.length) {
  86                         return false;
  87                 }
  88                 if (ntlmssp_state->neg_flags & NTLMSSP_NEGOTIATE_SEAL) {
  89                         return true;
  90                 }
  91         }
  92         if (feature & GENSEC_FEATURE_SESSION_KEY) {
  93                 if (ntlmssp_state->session_key.length) {
  94                         return true;
  95                 }
  96         }
  97         if (feature & GENSEC_FEATURE_DCE_STYLE) {
  98                 return true;
  99         }
 100         if (feature & GENSEC_FEATURE_ASYNC_REPLIES) {
 101                 if (ntlmssp_state->neg_flags & NTLMSSP_NEGOTIATE_NTLM2) {
 102                         return true;
 103                 }
 104         }
 105         return false;
 106 }
 107
 108 NTSTATUS gensec_ntlmssp_start(struct gensec_security *gensec_security)
 109 {
 110         struct gensec_ntlmssp_context *gensec_ntlmssp;
 111
 112         gensec_ntlmssp = talloc_zero(gensec_security,
 113                                      struct gensec_ntlmssp_context);
 114         if (!gensec_ntlmssp) {
 115                 return NT_STATUS_NO_MEMORY;
 116         }
 117
 118         gensec_security->private_data = gensec_ntlmssp;
 119         return NT_STATUS_OK;
 120 }
 121
 122 NTSTATUS gensec_ntlmssp_sign_packet(struct gensec_security *gensec_security,
 123                                     TALLOC_CTX *sig_mem_ctx,
 124                                     const uint8_t *data, size_t length,
 125                                     const uint8_t *whole_pdu, size_t pdu_length,
 126                                     DATA_BLOB *sig)
 127 {
 128         struct gensec_ntlmssp_context *gensec_ntlmssp =
 129                 talloc_get_type_abort(gensec_security->private_data,
 130                                       struct gensec_ntlmssp_context);
 131         NTSTATUS nt_status;
 132
 133         nt_status = ntlmssp_sign_packet(gensec_ntlmssp->ntlmssp_state,
 134                                         sig_mem_ctx,
 135                                         data, length,
 136                                         whole_pdu, pdu_length,
 137                                         sig);
 138
 139         return nt_status;
 140 }
 141
 142 NTSTATUS gensec_ntlmssp_check_packet(struct gensec_security *gensec_security,
 143                                      const uint8_t *data, size_t length,
 144                                      const uint8_t *whole_pdu, size_t pdu_length,
 145                                      const DATA_BLOB *sig)
 146 {
 147         struct gensec_ntlmssp_context *gensec_ntlmssp =
 148                 talloc_get_type_abort(gensec_security->private_data,
 149                                       struct gensec_ntlmssp_context);
 150         NTSTATUS nt_status;
 151
 152         nt_status = ntlmssp_check_packet(gensec_ntlmssp->ntlmssp_state,
 153                                          data, length,
 154                                          whole_pdu, pdu_length,
 155                                          sig);
 156
 157         return nt_status;
 158 }
 159
 160 NTSTATUS gensec_ntlmssp_seal_packet(struct gensec_security *gensec_security,
 161                                     TALLOC_CTX *sig_mem_ctx,
 162                                     uint8_t *data, size_t length,
 163                                     const uint8_t *whole_pdu, size_t pdu_length,
 164                                     DATA_BLOB *sig)
 165 {
 166         struct gensec_ntlmssp_context *gensec_ntlmssp =
 167                 talloc_get_type_abort(gensec_security->private_data,
 168                                       struct gensec_ntlmssp_context);
 169         NTSTATUS nt_status;
 170
 171         nt_status = ntlmssp_seal_packet(gensec_ntlmssp->ntlmssp_state,
 172                                         sig_mem_ctx,
 173                                         data, length,
 174                                         whole_pdu, pdu_length,
 175                                         sig);
 176
 177         return nt_status;
 178 }
 179
 180 /*
 181   wrappers for the ntlmssp_*() functions
 182 */
 183 NTSTATUS gensec_ntlmssp_unseal_packet(struct gensec_security *gensec_security,
 184                                       uint8_t *data, size_t length,
 185                                       const uint8_t *whole_pdu, size_t pdu_length,
 186                                       const DATA_BLOB *sig)
 187 {
 188         struct gensec_ntlmssp_context *gensec_ntlmssp =
 189                 talloc_get_type_abort(gensec_security->private_data,
 190                                       struct gensec_ntlmssp_context);
 191         NTSTATUS nt_status;
 192
 193         nt_status = ntlmssp_unseal_packet(gensec_ntlmssp->ntlmssp_state,
 194                                           data, length,
 195                                           whole_pdu, pdu_length,
 196                                           sig);
 197
 198         return nt_status;
 199 }
 200
 201 size_t gensec_ntlmssp_sig_size(struct gensec_security *gensec_security, size_t data_size)
 202 {
 203         return NTLMSSP_SIG_SIZE;
 204 }
 205
 206 NTSTATUS gensec_ntlmssp_wrap(struct gensec_security *gensec_security,
 207                              TALLOC_CTX *out_mem_ctx,
 208                              const DATA_BLOB *in,
 209                              DATA_BLOB *out)
 210 {
 211         struct gensec_ntlmssp_context *gensec_ntlmssp =
 212                 talloc_get_type_abort(gensec_security->private_data,
 213                                       struct gensec_ntlmssp_context);
 214
 215         return ntlmssp_wrap(gensec_ntlmssp->ntlmssp_state,
 216                             out_mem_ctx,
 217                             in, out);
 218 }
 219
 220
 221 NTSTATUS gensec_ntlmssp_unwrap(struct gensec_security *gensec_security,
 222                                TALLOC_CTX *out_mem_ctx,
 223                                const DATA_BLOB *in,
 224                                DATA_BLOB *out)
 225 {
 226         struct gensec_ntlmssp_context *gensec_ntlmssp =
 227                 talloc_get_type_abort(gensec_security->private_data,
 228                                       struct gensec_ntlmssp_context);
 229
 230         return ntlmssp_unwrap(gensec_ntlmssp->ntlmssp_state,
 231                               out_mem_ctx,
 232                               in, out);
 233 }