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_set_username(struct auth_ntlmssp_state *ans,
  30                                    const char *user)
  31 {
  32         cli_credentials_set_username(ans->credentials, user, CRED_SPECIFIED);
  33         return NT_STATUS_OK;
  34 }
  35
  36 NTSTATUS auth_ntlmssp_set_domain(struct auth_ntlmssp_state *ans,
  37                                  const char *domain)
  38 {
  39         cli_credentials_set_domain(ans->credentials, domain, CRED_SPECIFIED);
  40         return NT_STATUS_OK;
  41 }
  42
  43 NTSTATUS auth_ntlmssp_set_password(struct auth_ntlmssp_state *ans,
  44                                    const char *password)
  45 {
  46         cli_credentials_set_password(ans->credentials, password, CRED_SPECIFIED);
  47         return NT_STATUS_OK;
  48 }
  49
  50 DATA_BLOB auth_ntlmssp_get_session_key(struct auth_ntlmssp_state *ans, TALLOC_CTX *mem_ctx)
  51 {
  52         DATA_BLOB session_key;
  53         NTSTATUS status = gensec_session_key(ans->gensec_security, mem_ctx, &session_key);
  54         if (NT_STATUS_IS_OK(status)) {
  55                 return session_key;
  56         } else {
  57                 return data_blob_null;
  58         }
  59 }
  60
  61 static NTSTATUS gensec_ntlmssp3_client_update(struct gensec_security *gensec_security,
  62                                               TALLOC_CTX *out_mem_ctx,
  63                                               struct tevent_context *ev,
  64                                               const DATA_BLOB request,
  65                                               DATA_BLOB *reply)
  66 {
  67         NTSTATUS status;
  68         struct gensec_ntlmssp_context *gensec_ntlmssp =
  69                 talloc_get_type_abort(gensec_security->private_data,
  70                                       struct gensec_ntlmssp_context);
  71
  72         status = ntlmssp_update(gensec_ntlmssp->ntlmssp_state, request, reply);
  73         if (NT_STATUS_IS_OK(status) ||
  74             NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
  75                 talloc_steal(out_mem_ctx, reply->data);
  76         }
  77
  78         return status;
  79 }
  80
  81 static NTSTATUS gensec_ntlmssp3_client_start(struct gensec_security *gensec_security)
  82 {
  83         NTSTATUS nt_status;
  84         struct gensec_ntlmssp_context *gensec_ntlmssp;
  85         const char *user, *domain;
  86         const char *password;
  87
  88         nt_status = gensec_ntlmssp_start(gensec_security);
  89         NT_STATUS_NOT_OK_RETURN(nt_status);
  90
  91         gensec_ntlmssp =
  92                 talloc_get_type_abort(gensec_security->private_data,
  93                                       struct gensec_ntlmssp_context);
  94
  95         nt_status = ntlmssp_client_start(gensec_ntlmssp,
  96                                          lp_netbios_name(), lp_workgroup(),
  97                                          lp_client_ntlmv2_auth(), &gensec_ntlmssp->ntlmssp_state);
  98         if (!NT_STATUS_IS_OK(nt_status)) {
  99                 return nt_status;
 100         }
 101
 102         cli_credentials_get_ntlm_username_domain(gensec_security->credentials, gensec_ntlmssp, &user, &domain);
 103         if (!user || !domain) {
 104                 return NT_STATUS_NO_MEMORY;
 105         }
 106
 107         nt_status = ntlmssp_set_username(gensec_ntlmssp->ntlmssp_state, user);
 108         if (!NT_STATUS_IS_OK(nt_status)) {
 109                 return nt_status;
 110         }
 111
 112         nt_status = ntlmssp_set_domain(gensec_ntlmssp->ntlmssp_state, domain);
 113         if (!NT_STATUS_IS_OK(nt_status)) {
 114                 return nt_status;
 115         }
 116
 117         password = cli_credentials_get_password(gensec_security->credentials);
 118         if (!password) {
 119                 return NT_STATUS_NO_MEMORY;
 120         }
 121
 122         nt_status = ntlmssp_set_password(gensec_ntlmssp->ntlmssp_state, password);
 123         if (!NT_STATUS_IS_OK(nt_status)) {
 124                 return nt_status;
 125         }
 126
 127         if (gensec_ntlmssp->gensec_security->want_features & GENSEC_FEATURE_SESSION_KEY) {
 128                 gensec_ntlmssp->ntlmssp_state->neg_flags |= NTLMSSP_NEGOTIATE_SIGN;
 129         }
 130         if (gensec_ntlmssp->gensec_security->want_features & GENSEC_FEATURE_SIGN) {
 131                 gensec_ntlmssp->ntlmssp_state->neg_flags |= NTLMSSP_NEGOTIATE_SIGN;
 132         }
 133         if (gensec_ntlmssp->gensec_security->want_features & GENSEC_FEATURE_SEAL) {
 134                 gensec_ntlmssp->ntlmssp_state->neg_flags |= NTLMSSP_NEGOTIATE_SIGN;
 135                 gensec_ntlmssp->ntlmssp_state->neg_flags |= NTLMSSP_NEGOTIATE_SEAL;
 136         }
 137
 138         return NT_STATUS_OK;
 139 }
 140
 141 static const char *gensec_ntlmssp3_client_oids[] = {
 142         GENSEC_OID_NTLMSSP,
 143         NULL
 144 };
 145
 146 static const struct gensec_security_ops gensec_ntlmssp3_client_ops = {
 147         .name           = "ntlmssp3_client",
 148         .sasl_name      = GENSEC_SASL_NAME_NTLMSSP, /* "NTLM" */
 149         .auth_type      = DCERPC_AUTH_TYPE_NTLMSSP,
 150         .oid            = gensec_ntlmssp3_client_oids,
 151         .client_start   = gensec_ntlmssp3_client_start,
 152         .magic          = gensec_ntlmssp_magic,
 153         .update         = gensec_ntlmssp3_client_update,
 154         .sig_size       = gensec_ntlmssp_sig_size,
 155         .sign_packet    = gensec_ntlmssp_sign_packet,
 156         .check_packet   = gensec_ntlmssp_check_packet,
 157         .seal_packet    = gensec_ntlmssp_seal_packet,
 158         .unseal_packet  = gensec_ntlmssp_unseal_packet,
 159         .wrap           = gensec_ntlmssp_wrap,
 160         .unwrap         = gensec_ntlmssp_unwrap,
 161         .session_key    = gensec_ntlmssp_session_key,
 162         .have_feature   = gensec_ntlmssp_have_feature,
 163         .enabled        = true,
 164         .priority       = GENSEC_NTLMSSP
 165 };
 166
 167 NTSTATUS auth_ntlmssp_client_prepare(TALLOC_CTX *mem_ctx, struct auth_ntlmssp_state **auth_ntlmssp_state)
 168 {
 169         struct auth_ntlmssp_state *ans;
 170         NTSTATUS nt_status;
 171
 172         struct gensec_settings *gensec_settings;
 173         struct loadparm_context *lp_ctx;
 174
 175         ans = talloc_zero(mem_ctx, struct auth_ntlmssp_state);
 176         if (!ans) {
 177                 DEBUG(0,("auth_ntlmssp_start: talloc failed!\n"));
 178                 return NT_STATUS_NO_MEMORY;
 179         }
 180
 181         lp_ctx = loadparm_init_s3(ans, loadparm_s3_context());
 182         if (lp_ctx == NULL) {
 183                 DEBUG(10, ("loadparm_init_s3 failed\n"));
 184                 TALLOC_FREE(ans);
 185                 return NT_STATUS_INVALID_SERVER_STATE;
 186         }
 187
 188         gensec_settings = lpcfg_gensec_settings(ans, lp_ctx);
 189         if (lp_ctx == NULL) {
 190                 DEBUG(10, ("lpcfg_gensec_settings failed\n"));
 191                 TALLOC_FREE(ans);
 192                 return NT_STATUS_NO_MEMORY;
 193         }
 194
 195         nt_status = gensec_client_start(ans, &ans->gensec_security, gensec_settings);
 196
 197         if (!NT_STATUS_IS_OK(nt_status)) {
 198                 TALLOC_FREE(ans);
 199                 return nt_status;
 200         }
 201
 202         ans->credentials = cli_credentials_init(ans);
 203         if (!ans->credentials) {
 204                 TALLOC_FREE(ans);
 205                 return NT_STATUS_NO_MEMORY;
 206         }
 207
 208         cli_credentials_guess(ans->credentials, lp_ctx);
 209
 210         talloc_unlink(ans, lp_ctx);
 211         talloc_unlink(ans, gensec_settings);
 212
 213         *auth_ntlmssp_state = ans;
 214         return NT_STATUS_OK;
 215 }
 216
 217 NTSTATUS auth_ntlmssp_client_start(struct auth_ntlmssp_state *ans)
 218 {
 219         NTSTATUS status;
 220
 221         /* Transfer the credentials to gensec */
 222         status = gensec_set_credentials(ans->gensec_security, ans->credentials);
 223         if (!NT_STATUS_IS_OK(status)) {
 224                 DEBUG(1, ("Failed to set GENSEC credentials: %s\n",
 225                           nt_errstr(status)));
 226                 return status;
 227         }
 228         talloc_unlink(ans, ans->credentials);
 229         ans->credentials = NULL;
 230
 231         status = gensec_start_mech_by_ops(ans->gensec_security,
 232                                           &gensec_ntlmssp3_client_ops);
 233         if (!NT_STATUS_IS_OK(status)) {
 234                 return status;
 235         }
 236
 237         return NT_STATUS_OK;
 238 }