s3:auth_generic: add auth_generic_client_start_by_name()
[samba.git] / source3 / libsmb / ntlmssp_wrap.c
index ba22300cab216ae54fae391e7a7d13b55c9313bc..46f68aede79271e189a883e06f4e411d6fd64cc4 100644 (file)
@@ -2,7 +2,7 @@
    NLTMSSP wrappers
 
    Copyright (C) Andrew Tridgell      2001
-   Copyright (C) Andrew Bartlett 2001-2003
+   Copyright (C) Andrew Bartlett 2001-2003,2011
 
    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
 */
 
 #include "includes.h"
-#include "libcli/auth/ntlmssp.h"
-#include "ntlmssp_wrap.h"
+#include "auth/ntlmssp/ntlmssp.h"
+#include "auth/ntlmssp/ntlmssp_private.h"
+#include "auth_generic.h"
 #include "auth/gensec/gensec.h"
-
-NTSTATUS auth_ntlmssp_sign_packet(struct auth_ntlmssp_state *ans,
-                                 TALLOC_CTX *sig_mem_ctx,
-                                 const uint8_t *data,
-                                 size_t length,
-                                 const uint8_t *whole_pdu,
-                                 size_t pdu_length,
-                                 DATA_BLOB *sig)
+#include "auth/gensec/gensec_internal.h"
+#include "auth/credentials/credentials.h"
+#include "librpc/rpc/dcerpc.h"
+#include "lib/param/param.h"
+
+static NTSTATUS gensec_ntlmssp3_client_update(struct gensec_security *gensec_security,
+                                             TALLOC_CTX *out_mem_ctx,
+                                             struct tevent_context *ev,
+                                             const DATA_BLOB request,
+                                             DATA_BLOB *reply)
 {
-       if (ans->gensec_security) {
-               return gensec_sign_packet(ans->gensec_security,
-                                         sig_mem_ctx, data, length, whole_pdu, pdu_length, sig);
+       NTSTATUS status;
+       struct gensec_ntlmssp_context *gensec_ntlmssp =
+               talloc_get_type_abort(gensec_security->private_data,
+                                     struct gensec_ntlmssp_context);
+
+       status = ntlmssp_update(gensec_ntlmssp->ntlmssp_state, request, reply);
+       if (NT_STATUS_IS_OK(status) ||
+           NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
+               talloc_steal(out_mem_ctx, reply->data);
        }
-       return ntlmssp_sign_packet(ans->ntlmssp_state,
-                                  sig_mem_ctx,
-                                  data, length,
-                                  whole_pdu, pdu_length,
-                                  sig);
-}
 
-NTSTATUS auth_ntlmssp_check_packet(struct auth_ntlmssp_state *ans,
-                                  const uint8_t *data,
-                                  size_t length,
-                                  const uint8_t *whole_pdu,
-                                  size_t pdu_length,
-                                  const DATA_BLOB *sig)
-{
-       if (ans->gensec_security) {
-               return gensec_check_packet(ans->gensec_security,
-                                          data, length, whole_pdu, pdu_length, sig);
-       }
-       return ntlmssp_check_packet(ans->ntlmssp_state,
-                                   data, length,
-                                   whole_pdu, pdu_length,
-                                   sig);
+       return status;
 }
 
-NTSTATUS auth_ntlmssp_seal_packet(struct auth_ntlmssp_state *ans,
-                                 TALLOC_CTX *sig_mem_ctx,
-                                 uint8_t *data,
-                                 size_t length,
-                                 const uint8_t *whole_pdu,
-                                 size_t pdu_length,
-                                 DATA_BLOB *sig)
+static NTSTATUS gensec_ntlmssp3_client_start(struct gensec_security *gensec_security)
 {
-       if (ans->gensec_security) {
-               return gensec_seal_packet(ans->gensec_security,
-                                         sig_mem_ctx, data, length, whole_pdu, pdu_length, sig);
+       NTSTATUS nt_status;
+       struct gensec_ntlmssp_context *gensec_ntlmssp;
+       const char *user, *domain;
+       const char *password;
+
+       nt_status = gensec_ntlmssp_start(gensec_security);
+       NT_STATUS_NOT_OK_RETURN(nt_status);
+
+       gensec_ntlmssp =
+               talloc_get_type_abort(gensec_security->private_data,
+                                     struct gensec_ntlmssp_context);
+
+       nt_status = ntlmssp_client_start(gensec_ntlmssp,
+                                        lp_netbios_name(), lp_workgroup(),
+                                        lp_client_ntlmv2_auth(), &gensec_ntlmssp->ntlmssp_state);
+       if (!NT_STATUS_IS_OK(nt_status)) {
+               return nt_status;
        }
-       return ntlmssp_seal_packet(ans->ntlmssp_state,
-                                  sig_mem_ctx,
-                                  data, length,
-                                  whole_pdu, pdu_length,
-                                  sig);
-}
 
-NTSTATUS auth_ntlmssp_unseal_packet(struct auth_ntlmssp_state *ans,
-                                   uint8_t *data,
-                                   size_t length,
-                                   const uint8_t *whole_pdu,
-                                   size_t pdu_length,
-                                   const DATA_BLOB *sig)
-{
-       if (ans->gensec_security) {
-               return gensec_unseal_packet(ans->gensec_security,
-                                           data, length, whole_pdu, pdu_length, sig);
+       cli_credentials_get_ntlm_username_domain(gensec_security->credentials, gensec_ntlmssp, &user, &domain);
+       if (!user || !domain) {
+               return NT_STATUS_NO_MEMORY;
        }
-       return ntlmssp_unseal_packet(ans->ntlmssp_state,
-                                    data, length,
-                                    whole_pdu, pdu_length,
-                                    sig);
-}
 
-bool auth_ntlmssp_negotiated_sign(struct auth_ntlmssp_state *ans)
-{
-       if (ans->gensec_security) {
-               return gensec_have_feature(ans->gensec_security, GENSEC_FEATURE_SIGN);
+       nt_status = ntlmssp_set_username(gensec_ntlmssp->ntlmssp_state, user);
+       if (!NT_STATUS_IS_OK(nt_status)) {
+               return nt_status;
        }
-       return ans->ntlmssp_state->neg_flags & NTLMSSP_NEGOTIATE_SIGN;
-}
 
-bool auth_ntlmssp_negotiated_seal(struct auth_ntlmssp_state *ans)
-{
-       if (ans->gensec_security) {
-               return gensec_have_feature(ans->gensec_security, GENSEC_FEATURE_SEAL);
+       nt_status = ntlmssp_set_domain(gensec_ntlmssp->ntlmssp_state, domain);
+       if (!NT_STATUS_IS_OK(nt_status)) {
+               return nt_status;
        }
-       return ans->ntlmssp_state->neg_flags & NTLMSSP_NEGOTIATE_SEAL;
-}
-
-NTSTATUS auth_ntlmssp_set_username(struct auth_ntlmssp_state *ans,
-                                  const char *user)
-{
-       return ntlmssp_set_username(ans->ntlmssp_state, user);
-}
-
-NTSTATUS auth_ntlmssp_set_domain(struct auth_ntlmssp_state *ans,
-                                const char *domain)
-{
-       return ntlmssp_set_domain(ans->ntlmssp_state, domain);
-}
 
-NTSTATUS auth_ntlmssp_set_password(struct auth_ntlmssp_state *ans,
-                                  const char *password)
-{
-       return ntlmssp_set_password(ans->ntlmssp_state, password);
-}
-
-void auth_ntlmssp_want_feature(struct auth_ntlmssp_state *ans, uint32_t feature)
-{
-       if (ans->gensec_security) {
-               /* You need to negotiate signing to get a windows server to calculate a session key */
-               if (feature & NTLMSSP_FEATURE_SESSION_KEY) {
-                       return gensec_want_feature(ans->gensec_security, GENSEC_FEATURE_SESSION_KEY);
-               }
-               if (feature & NTLMSSP_FEATURE_SIGN) {
-                       return gensec_want_feature(ans->gensec_security, GENSEC_FEATURE_SIGN);
-               }
-               if (feature & NTLMSSP_FEATURE_SEAL) {
-                       return gensec_want_feature(ans->gensec_security, GENSEC_FEATURE_SEAL);
-               }
-       } else {
-               ntlmssp_want_feature(ans->ntlmssp_state, feature);
+       password = cli_credentials_get_password(gensec_security->credentials);
+       if (!password) {
+               return NT_STATUS_NO_MEMORY;
        }
-}
 
-DATA_BLOB auth_ntlmssp_get_session_key(struct auth_ntlmssp_state *ans, TALLOC_CTX *mem_ctx)
-{
-       if (ans->gensec_security) {
-               DATA_BLOB session_key;
-               NTSTATUS status = gensec_session_key(ans->gensec_security, mem_ctx, &session_key);
-               if (NT_STATUS_IS_OK(status)) {
-                       return session_key;
-               } else {
-                       return data_blob_null;
-               }
+       nt_status = ntlmssp_set_password(gensec_ntlmssp->ntlmssp_state, password);
+       if (!NT_STATUS_IS_OK(nt_status)) {
+               return nt_status;
        }
-       return data_blob_talloc(mem_ctx, ans->ntlmssp_state->session_key.data, ans->ntlmssp_state->session_key.length);
-}
 
-NTSTATUS auth_ntlmssp_update(struct auth_ntlmssp_state *ans,
-                            TALLOC_CTX *mem_ctx,
-                            const DATA_BLOB request, DATA_BLOB *reply)
-{
-       NTSTATUS status;
-       if (ans->gensec_security) {
-               return gensec_update(ans->gensec_security, mem_ctx, request, reply);
+       if (gensec_security->want_features & GENSEC_FEATURE_SESSION_KEY) {
+               gensec_ntlmssp->ntlmssp_state->neg_flags |= NTLMSSP_NEGOTIATE_SIGN;
        }
-       status = ntlmssp_update(ans->ntlmssp_state, request, reply);
-       if (!NT_STATUS_IS_OK(status) && !NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
-               return status;
+       if (gensec_security->want_features & GENSEC_FEATURE_SIGN) {
+               gensec_ntlmssp->ntlmssp_state->neg_flags |= NTLMSSP_NEGOTIATE_SIGN;
        }
-       talloc_steal(mem_ctx, reply->data);
-       return status;
-}
-
-NTSTATUS auth_ntlmssp_client_start(TALLOC_CTX *mem_ctx,
-                                  const char *netbios_name,
-                                  const char *netbios_domain,
-                                  bool use_ntlmv2,
-                                  struct auth_ntlmssp_state **_ans)
-{
-       struct auth_ntlmssp_state *ans;
-       NTSTATUS status;
-
-       ans = talloc_zero(mem_ctx, struct auth_ntlmssp_state);
-
-       status = ntlmssp_client_start(ans,
-                                       netbios_name, netbios_domain,
-                                       use_ntlmv2, &ans->ntlmssp_state);
-       if (!NT_STATUS_IS_OK(status)) {
-               return status;
+       if (gensec_security->want_features & GENSEC_FEATURE_SEAL) {
+               gensec_ntlmssp->ntlmssp_state->neg_flags |= NTLMSSP_NEGOTIATE_SIGN;
+               gensec_ntlmssp->ntlmssp_state->neg_flags |= NTLMSSP_NEGOTIATE_SEAL;
        }
 
-       *_ans = ans;
        return NT_STATUS_OK;
 }
+
+static const char *gensec_ntlmssp3_client_oids[] = {
+       GENSEC_OID_NTLMSSP,
+       NULL
+};
+
+const struct gensec_security_ops gensec_ntlmssp3_client_ops = {
+       .name           = "ntlmssp3_client",
+       .sasl_name      = GENSEC_SASL_NAME_NTLMSSP, /* "NTLM" */
+       .auth_type      = DCERPC_AUTH_TYPE_NTLMSSP,
+       .oid            = gensec_ntlmssp3_client_oids,
+       .client_start   = gensec_ntlmssp3_client_start,
+       .magic          = gensec_ntlmssp_magic,
+       .update         = gensec_ntlmssp3_client_update,
+       .sig_size       = gensec_ntlmssp_sig_size,
+       .sign_packet    = gensec_ntlmssp_sign_packet,
+       .check_packet   = gensec_ntlmssp_check_packet,
+       .seal_packet    = gensec_ntlmssp_seal_packet,
+       .unseal_packet  = gensec_ntlmssp_unseal_packet,
+       .wrap           = gensec_ntlmssp_wrap,
+       .unwrap         = gensec_ntlmssp_unwrap,
+       .session_key    = gensec_ntlmssp_session_key,
+       .have_feature   = gensec_ntlmssp_have_feature,
+       .enabled        = true,
+       .priority       = GENSEC_NTLMSSP
+};