source3/libsmb/ntlmssp_wrap.c

   1 /*
   2    NLTMSSP wrappers
   3
   4    Copyright (C) Andrew Tridgell      2001
   5    Copyright (C) Andrew Bartlett 2001-2003,2011
   6
   7    This program is free software; you can redistribute it and/or modify
   8    it under the terms of the GNU General Public License as published by
   9    the Free Software Foundation; either version 3 of the License, or
  10    (at your option) any later version.
  11
  12    This program is distributed in the hope that it will be useful,
  13    but WITHOUT ANY WARRANTY; without even the implied warranty of
  14    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15    GNU General Public License for more details.
  16
  17    You should have received a copy of the GNU General Public License
  18    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19 */
  20
  21 #include "includes.h"
  22 #include "auth/ntlmssp/ntlmssp.h"
  23 #include "ntlmssp_wrap.h"
  24 #include "auth/gensec/gensec.h"
  25 #include "auth/credentials/credentials.h"
  26 #include "librpc/rpc/dcerpc.h"
  27 #include "lib/param/param.h"
  28
  29 NTSTATUS auth_ntlmssp_sign_packet(struct auth_ntlmssp_state *ans,
  30                                   TALLOC_CTX *sig_mem_ctx,
  31                                   const uint8_t *data,
  32                                   size_t length,
  33                                   const uint8_t *whole_pdu,
  34                                   size_t pdu_length,
  35                                   DATA_BLOB *sig)
  36 {
  37         return gensec_sign_packet(ans->gensec_security,
  38                                   sig_mem_ctx, data, length, whole_pdu, pdu_length, sig);
  39 }
  40
  41 NTSTATUS auth_ntlmssp_check_packet(struct auth_ntlmssp_state *ans,
  42                                    const uint8_t *data,
  43                                    size_t length,
  44                                    const uint8_t *whole_pdu,
  45                                    size_t pdu_length,
  46                                    const DATA_BLOB *sig)
  47 {
  48         return gensec_check_packet(ans->gensec_security,
  49                                    data, length, whole_pdu, pdu_length, sig);
  50 }
  51
  52 NTSTATUS auth_ntlmssp_seal_packet(struct auth_ntlmssp_state *ans,
  53                                   TALLOC_CTX *sig_mem_ctx,
  54                                   uint8_t *data,
  55                                   size_t length,
  56                                   const uint8_t *whole_pdu,
  57                                   size_t pdu_length,
  58                                   DATA_BLOB *sig)
  59 {
  60         return gensec_seal_packet(ans->gensec_security,
  61                                   sig_mem_ctx, data, length, whole_pdu, pdu_length, sig);
  62 }
  63
  64 NTSTATUS auth_ntlmssp_unseal_packet(struct auth_ntlmssp_state *ans,
  65                                     uint8_t *data,
  66                                     size_t length,
  67                                     const uint8_t *whole_pdu,
  68                                     size_t pdu_length,
  69                                     const DATA_BLOB *sig)
  70 {
  71         return gensec_unseal_packet(ans->gensec_security,
  72                                     data, length, whole_pdu, pdu_length, sig);
  73 }
  74
  75 bool auth_ntlmssp_negotiated_sign(struct auth_ntlmssp_state *ans)
  76 {
  77         return gensec_have_feature(ans->gensec_security, GENSEC_FEATURE_SIGN);
  78 }
  79
  80 bool auth_ntlmssp_negotiated_seal(struct auth_ntlmssp_state *ans)
  81 {
  82         return gensec_have_feature(ans->gensec_security, GENSEC_FEATURE_SEAL);
  83 }
  84
  85 NTSTATUS auth_ntlmssp_set_username(struct auth_ntlmssp_state *ans,
  86                                    const char *user)
  87 {
  88         cli_credentials_set_username(ans->credentials, user, CRED_SPECIFIED);
  89         return NT_STATUS_OK;
  90 }
  91
  92 NTSTATUS auth_ntlmssp_set_domain(struct auth_ntlmssp_state *ans,
  93                                  const char *domain)
  94 {
  95         cli_credentials_set_domain(ans->credentials, domain, CRED_SPECIFIED);
  96         return NT_STATUS_OK;
  97 }
  98
  99 NTSTATUS auth_ntlmssp_set_password(struct auth_ntlmssp_state *ans,
 100                                    const char *password)
 101 {
 102         cli_credentials_set_password(ans->credentials, password, CRED_SPECIFIED);
 103         return NT_STATUS_OK;
 104 }
 105
 106 void auth_ntlmssp_want_feature(struct auth_ntlmssp_state *ans, uint32_t feature)
 107 {
 108         if (feature & NTLMSSP_FEATURE_SESSION_KEY) {
 109                 gensec_want_feature(ans->gensec_security, GENSEC_FEATURE_SESSION_KEY);
 110         }
 111         if (feature & NTLMSSP_FEATURE_SIGN) {
 112                 gensec_want_feature(ans->gensec_security, GENSEC_FEATURE_SIGN);
 113         }
 114         if (feature & NTLMSSP_FEATURE_SEAL) {
 115                 gensec_want_feature(ans->gensec_security, GENSEC_FEATURE_SEAL);
 116         }
 117 }
 118
 119 DATA_BLOB auth_ntlmssp_get_session_key(struct auth_ntlmssp_state *ans, TALLOC_CTX *mem_ctx)
 120 {
 121         DATA_BLOB session_key;
 122         NTSTATUS status = gensec_session_key(ans->gensec_security, mem_ctx, &session_key);
 123         if (NT_STATUS_IS_OK(status)) {
 124                 return session_key;
 125         } else {
 126                 return data_blob_null;
 127         }
 128 }
 129
 130 NTSTATUS auth_ntlmssp_update(struct auth_ntlmssp_state *ans,
 131                              TALLOC_CTX *mem_ctx,
 132                              const DATA_BLOB request, DATA_BLOB *reply)
 133 {
 134         return gensec_update(ans->gensec_security, mem_ctx, NULL, request, reply);
 135 }
 136
 137 static NTSTATUS gensec_ntlmssp3_client_update(struct gensec_security *gensec_security,
 138                                               TALLOC_CTX *out_mem_ctx,
 139                                               struct tevent_context *ev,
 140                                               const DATA_BLOB request,
 141                                               DATA_BLOB *reply)
 142 {
 143         NTSTATUS status;
 144         struct gensec_ntlmssp_context *gensec_ntlmssp =
 145                 talloc_get_type_abort(gensec_security->private_data,
 146                                       struct gensec_ntlmssp_context);
 147
 148         status = ntlmssp_update(gensec_ntlmssp->ntlmssp_state, request, reply);
 149         if (NT_STATUS_IS_OK(status) ||
 150             NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
 151                 talloc_steal(out_mem_ctx, reply->data);
 152         }
 153
 154         return status;
 155 }
 156
 157 static NTSTATUS gensec_ntlmssp3_client_start(struct gensec_security *gensec_security)
 158 {
 159         NTSTATUS nt_status;
 160         struct gensec_ntlmssp_context *gensec_ntlmssp;
 161         const char *user, *domain;
 162         const char *password;
 163
 164         nt_status = gensec_ntlmssp_start(gensec_security);
 165         NT_STATUS_NOT_OK_RETURN(nt_status);
 166
 167         gensec_ntlmssp =
 168                 talloc_get_type_abort(gensec_security->private_data,
 169                                       struct gensec_ntlmssp_context);
 170
 171         nt_status = ntlmssp_client_start(gensec_ntlmssp,
 172                                          lp_netbios_name(), lp_workgroup(),
 173                                          lp_client_ntlmv2_auth(), &gensec_ntlmssp->ntlmssp_state);
 174         if (!NT_STATUS_IS_OK(nt_status)) {
 175                 return nt_status;
 176         }
 177
 178         cli_credentials_get_ntlm_username_domain(gensec_security->credentials, gensec_ntlmssp, &user, &domain);
 179         if (!user || !domain) {
 180                 return NT_STATUS_NO_MEMORY;
 181         }
 182
 183         nt_status = ntlmssp_set_username(gensec_ntlmssp->ntlmssp_state, user);
 184         if (!NT_STATUS_IS_OK(nt_status)) {
 185                 return nt_status;
 186         }
 187
 188         nt_status = ntlmssp_set_domain(gensec_ntlmssp->ntlmssp_state, domain);
 189         if (!NT_STATUS_IS_OK(nt_status)) {
 190                 return nt_status;
 191         }
 192
 193         password = cli_credentials_get_password(gensec_security->credentials);
 194         if (!password) {
 195                 return NT_STATUS_NO_MEMORY;
 196         }
 197
 198         nt_status = ntlmssp_set_password(gensec_ntlmssp->ntlmssp_state, password);
 199         if (!NT_STATUS_IS_OK(nt_status)) {
 200                 return nt_status;
 201         }
 202
 203         if (gensec_ntlmssp->gensec_security->want_features & GENSEC_FEATURE_SESSION_KEY) {
 204                 gensec_ntlmssp->ntlmssp_state->neg_flags |= NTLMSSP_NEGOTIATE_SIGN;
 205         }
 206         if (gensec_ntlmssp->gensec_security->want_features & GENSEC_FEATURE_SIGN) {
 207                 gensec_ntlmssp->ntlmssp_state->neg_flags |= NTLMSSP_NEGOTIATE_SIGN;
 208         }
 209         if (gensec_ntlmssp->gensec_security->want_features & GENSEC_FEATURE_SEAL) {
 210                 gensec_ntlmssp->ntlmssp_state->neg_flags |= NTLMSSP_NEGOTIATE_SIGN;
 211                 gensec_ntlmssp->ntlmssp_state->neg_flags |= NTLMSSP_NEGOTIATE_SEAL;
 212         }
 213
 214         return NT_STATUS_OK;
 215 }
 216
 217 static const char *gensec_ntlmssp3_client_oids[] = {
 218         GENSEC_OID_NTLMSSP,
 219         NULL
 220 };
 221
 222 static const struct gensec_security_ops gensec_ntlmssp3_client_ops = {
 223         .name           = "ntlmssp3_client",
 224         .sasl_name      = GENSEC_SASL_NAME_NTLMSSP, /* "NTLM" */
 225         .auth_type      = DCERPC_AUTH_TYPE_NTLMSSP,
 226         .oid            = gensec_ntlmssp3_client_oids,
 227         .client_start   = gensec_ntlmssp3_client_start,
 228         .magic          = gensec_ntlmssp_magic,
 229         .update         = gensec_ntlmssp3_client_update,
 230         .sig_size       = gensec_ntlmssp_sig_size,
 231         .sign_packet    = gensec_ntlmssp_sign_packet,
 232         .check_packet   = gensec_ntlmssp_check_packet,
 233         .seal_packet    = gensec_ntlmssp_seal_packet,
 234         .unseal_packet  = gensec_ntlmssp_unseal_packet,
 235         .wrap           = gensec_ntlmssp_wrap,
 236         .unwrap         = gensec_ntlmssp_unwrap,
 237         .session_key    = gensec_ntlmssp_session_key,
 238         .have_feature   = gensec_ntlmssp_have_feature,
 239         .enabled        = true,
 240         .priority       = GENSEC_NTLMSSP
 241 };
 242
 243 NTSTATUS auth_ntlmssp_client_prepare(TALLOC_CTX *mem_ctx, struct auth_ntlmssp_state **auth_ntlmssp_state)
 244 {
 245         struct auth_ntlmssp_state *ans;
 246         NTSTATUS nt_status;
 247
 248         struct gensec_settings *gensec_settings;
 249         struct loadparm_context *lp_ctx;
 250
 251         ans = talloc_zero(mem_ctx, struct auth_ntlmssp_state);
 252         if (!ans) {
 253                 DEBUG(0,("auth_ntlmssp_start: talloc failed!\n"));
 254                 return NT_STATUS_NO_MEMORY;
 255         }
 256
 257         lp_ctx = loadparm_init_s3(ans, loadparm_s3_context());
 258         if (lp_ctx == NULL) {
 259                 DEBUG(10, ("loadparm_init_s3 failed\n"));
 260                 TALLOC_FREE(ans);
 261                 return NT_STATUS_INVALID_SERVER_STATE;
 262         }
 263
 264         gensec_settings = lpcfg_gensec_settings(ans, lp_ctx);
 265         if (lp_ctx == NULL) {
 266                 DEBUG(10, ("lpcfg_gensec_settings failed\n"));
 267                 TALLOC_FREE(ans);
 268                 return NT_STATUS_NO_MEMORY;
 269         }
 270
 271         nt_status = gensec_client_start(ans, &ans->gensec_security, gensec_settings);
 272
 273         if (!NT_STATUS_IS_OK(nt_status)) {
 274                 TALLOC_FREE(ans);
 275                 return nt_status;
 276         }
 277
 278         ans->credentials = cli_credentials_init(ans);
 279         if (!ans->credentials) {
 280                 TALLOC_FREE(ans);
 281                 return NT_STATUS_NO_MEMORY;
 282         }
 283
 284         cli_credentials_guess(ans->credentials, lp_ctx);
 285
 286         talloc_unlink(ans, lp_ctx);
 287         talloc_unlink(ans, gensec_settings);
 288
 289         *auth_ntlmssp_state = ans;
 290         return NT_STATUS_OK;
 291 }
 292
 293 NTSTATUS auth_ntlmssp_client_start(struct auth_ntlmssp_state *ans)
 294 {
 295         NTSTATUS status;
 296
 297         /* Transfer the credentials to gensec */
 298         status = gensec_set_credentials(ans->gensec_security, ans->credentials);
 299         if (!NT_STATUS_IS_OK(status)) {
 300                 DEBUG(1, ("Failed to set GENSEC credentials: %s\n",
 301                           nt_errstr(status)));
 302                 return status;
 303         }
 304         talloc_unlink(ans, ans->credentials);
 305         ans->credentials = NULL;
 306
 307         status = gensec_start_mech_by_ops(ans->gensec_security,
 308                                           &gensec_ntlmssp3_client_ops);
 309         if (!NT_STATUS_IS_OK(status)) {
 310                 return status;
 311         }
 312
 313         return NT_STATUS_OK;
 314 }