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/ntlmssp/ntlmssp_private.h"
  26
  27 NTSTATUS gensec_ntlmssp_magic(struct gensec_security *gensec_security,
  28                               const DATA_BLOB *first_packet)
  29 {
  30         if (ntlmssp_blob_matches_magic(first_packet)) {
  31                 return NT_STATUS_OK;
  32         } else {
  33                 return NT_STATUS_INVALID_PARAMETER;
  34         }
  35 }
  36
  37 /**
  38  * Return the NTLMSSP master session key
  39  *
  40  * @param ntlmssp_state NTLMSSP State
  41  */
  42
  43 NTSTATUS gensec_ntlmssp_session_key(struct gensec_security *gensec_security,
  44                                     TALLOC_CTX *mem_ctx,
  45                                     DATA_BLOB *session_key)
  46 {
  47         struct gensec_ntlmssp_context *gensec_ntlmssp =
  48                 talloc_get_type_abort(gensec_security->private_data,
  49                                       struct gensec_ntlmssp_context);
  50         struct ntlmssp_state *ntlmssp_state = gensec_ntlmssp->ntlmssp_state;
  51
  52         if (ntlmssp_state->expected_state != NTLMSSP_DONE) {
  53                 return NT_STATUS_NO_USER_SESSION_KEY;
  54         }
  55
  56         if (!ntlmssp_state->session_key.data) {
  57                 return NT_STATUS_NO_USER_SESSION_KEY;
  58         }
  59         *session_key = data_blob_talloc(mem_ctx, ntlmssp_state->session_key.data, ntlmssp_state->session_key.length);
  60         if (!session_key->data) {
  61                 return NT_STATUS_NO_MEMORY;
  62         }
  63
  64         return NT_STATUS_OK;
  65 }
  66
  67 bool gensec_ntlmssp_have_feature(struct gensec_security *gensec_security,
  68                                  uint32_t feature)
  69 {
  70         struct gensec_ntlmssp_context *gensec_ntlmssp =
  71                 talloc_get_type_abort(gensec_security->private_data,
  72                                       struct gensec_ntlmssp_context);
  73         struct ntlmssp_state *ntlmssp_state = gensec_ntlmssp->ntlmssp_state;
  74
  75         if (feature & GENSEC_FEATURE_SIGN) {
  76                 if (!ntlmssp_state->session_key.length) {
  77                         return false;
  78                 }
  79                 if (ntlmssp_state->neg_flags & NTLMSSP_NEGOTIATE_SIGN) {
  80                         return true;
  81                 }
  82         }
  83         if (feature & GENSEC_FEATURE_SEAL) {
  84                 if (!ntlmssp_state->session_key.length) {
  85                         return false;
  86                 }
  87                 if (ntlmssp_state->neg_flags & NTLMSSP_NEGOTIATE_SEAL) {
  88                         return true;
  89                 }
  90         }
  91         if (feature & GENSEC_FEATURE_SESSION_KEY) {
  92                 if (ntlmssp_state->session_key.length) {
  93                         return true;
  94                 }
  95         }
  96         if (feature & GENSEC_FEATURE_DCE_STYLE) {
  97                 return true;
  98         }
  99         if (feature & GENSEC_FEATURE_ASYNC_REPLIES) {
 100                 if (ntlmssp_state->neg_flags & NTLMSSP_NEGOTIATE_NTLM2) {
 101                         return true;
 102                 }
 103         }
 104         return false;
 105 }
 106
 107 NTSTATUS gensec_ntlmssp_start(struct gensec_security *gensec_security)
 108 {
 109         struct gensec_ntlmssp_context *gensec_ntlmssp;
 110         struct ntlmssp_state *ntlmssp_state;
 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_ntlmssp->gensec_security = gensec_security;
 119
 120         ntlmssp_state = talloc_zero(gensec_ntlmssp,
 121                                     struct ntlmssp_state);
 122         if (!ntlmssp_state) {
 123                 return NT_STATUS_NO_MEMORY;
 124         }
 125
 126         ntlmssp_state->callback_private = gensec_ntlmssp;
 127
 128         gensec_ntlmssp->ntlmssp_state = ntlmssp_state;
 129
 130         gensec_security->private_data = gensec_ntlmssp;
 131         return NT_STATUS_OK;
 132 }
 133
 134 NTSTATUS gensec_ntlmssp_sign_packet(struct gensec_security *gensec_security,
 135                                     TALLOC_CTX *sig_mem_ctx,
 136                                     const uint8_t *data, size_t length,
 137                                     const uint8_t *whole_pdu, size_t pdu_length,
 138                                     DATA_BLOB *sig)
 139 {
 140         struct gensec_ntlmssp_context *gensec_ntlmssp =
 141                 talloc_get_type_abort(gensec_security->private_data,
 142                                       struct gensec_ntlmssp_context);
 143         NTSTATUS nt_status;
 144
 145         nt_status = ntlmssp_sign_packet(gensec_ntlmssp->ntlmssp_state,
 146                                         sig_mem_ctx,
 147                                         data, length,
 148                                         whole_pdu, pdu_length,
 149                                         sig);
 150
 151         return nt_status;
 152 }
 153
 154 NTSTATUS gensec_ntlmssp_check_packet(struct gensec_security *gensec_security,
 155                                      const uint8_t *data, size_t length,
 156                                      const uint8_t *whole_pdu, size_t pdu_length,
 157                                      const DATA_BLOB *sig)
 158 {
 159         struct gensec_ntlmssp_context *gensec_ntlmssp =
 160                 talloc_get_type_abort(gensec_security->private_data,
 161                                       struct gensec_ntlmssp_context);
 162         NTSTATUS nt_status;
 163
 164         nt_status = ntlmssp_check_packet(gensec_ntlmssp->ntlmssp_state,
 165                                          data, length,
 166                                          whole_pdu, pdu_length,
 167                                          sig);
 168
 169         return nt_status;
 170 }
 171
 172 NTSTATUS gensec_ntlmssp_seal_packet(struct gensec_security *gensec_security,
 173                                     TALLOC_CTX *sig_mem_ctx,
 174                                     uint8_t *data, size_t length,
 175                                     const uint8_t *whole_pdu, size_t pdu_length,
 176                                     DATA_BLOB *sig)
 177 {
 178         struct gensec_ntlmssp_context *gensec_ntlmssp =
 179                 talloc_get_type_abort(gensec_security->private_data,
 180                                       struct gensec_ntlmssp_context);
 181         NTSTATUS nt_status;
 182
 183         nt_status = ntlmssp_seal_packet(gensec_ntlmssp->ntlmssp_state,
 184                                         sig_mem_ctx,
 185                                         data, length,
 186                                         whole_pdu, pdu_length,
 187                                         sig);
 188
 189         return nt_status;
 190 }
 191
 192 /*
 193   wrappers for the ntlmssp_*() functions
 194 */
 195 NTSTATUS gensec_ntlmssp_unseal_packet(struct gensec_security *gensec_security,
 196                                       uint8_t *data, size_t length,
 197                                       const uint8_t *whole_pdu, size_t pdu_length,
 198                                       const DATA_BLOB *sig)
 199 {
 200         struct gensec_ntlmssp_context *gensec_ntlmssp =
 201                 talloc_get_type_abort(gensec_security->private_data,
 202                                       struct gensec_ntlmssp_context);
 203         NTSTATUS nt_status;
 204
 205         nt_status = ntlmssp_unseal_packet(gensec_ntlmssp->ntlmssp_state,
 206                                           data, length,
 207                                           whole_pdu, pdu_length,
 208                                           sig);
 209
 210         return nt_status;
 211 }
 212
 213 size_t gensec_ntlmssp_sig_size(struct gensec_security *gensec_security, size_t data_size)
 214 {
 215         return NTLMSSP_SIG_SIZE;
 216 }
 217
 218 NTSTATUS gensec_ntlmssp_wrap(struct gensec_security *gensec_security,
 219                              TALLOC_CTX *out_mem_ctx,
 220                              const DATA_BLOB *in,
 221                              DATA_BLOB *out)
 222 {
 223         struct gensec_ntlmssp_context *gensec_ntlmssp =
 224                 talloc_get_type_abort(gensec_security->private_data,
 225                                       struct gensec_ntlmssp_context);
 226
 227         return ntlmssp_wrap(gensec_ntlmssp->ntlmssp_state,
 228                             out_mem_ctx,
 229                             in, out);
 230 }
 231
 232
 233 NTSTATUS gensec_ntlmssp_unwrap(struct gensec_security *gensec_security,
 234                                TALLOC_CTX *out_mem_ctx,
 235                                const DATA_BLOB *in,
 236                                DATA_BLOB *out)
 237 {
 238         struct gensec_ntlmssp_context *gensec_ntlmssp =
 239                 talloc_get_type_abort(gensec_security->private_data,
 240                                       struct gensec_ntlmssp_context);
 241
 242         return ntlmssp_unwrap(gensec_ntlmssp->ntlmssp_state,
 243                               out_mem_ctx,
 244                               in, out);
 245 }