ntlm_auth: Use libwbclient in get_require_membership_sid()
[nivanova/samba-autobuild/.git] / source3 / utils / ntlm_auth.c
index 2216816bb48483274219435ae38295f901e6c3bc..c0e1c2a6525adf9d1d5203a63fa27d3845f255e9 100644 (file)
@@ -8,6 +8,7 @@
    Copyright (C) Francesco Chemolli <kinkie@kame.usr.dsi.unimi.it> 2000
    Copyright (C) Robert O'Callahan 2006 (added cached credential code).
    Copyright (C) Kai Blin <kai@samba.org> 2008
+   Copyright (C) Simo Sorce 2010
 
    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 "lib/param/param.h"
 #include "popt_common.h"
+#include "libcli/security/security.h"
 #include "utils/ntlm_auth.h"
 #include "../libcli/auth/libcli_auth.h"
-#include "../libcli/auth/spnego.h"
 #include "auth/ntlmssp/ntlmssp.h"
 #include "auth/gensec/gensec.h"
+#include "auth/gensec/gensec_internal.h"
 #include "auth/credentials/credentials.h"
 #include "librpc/crypto/gse.h"
 #include "smb_krb5.h"
-#include <iniparser.h>
+#include "lib/util/tiniparser.h"
 #include "../lib/crypto/arcfour.h"
-#include "libads/kerberos_proto.h"
 #include "nsswitch/winbind_client.h"
 #include "librpc/gen_ndr/krb5pac.h"
 #include "../lib/util/asn1.h"
 #include "auth/common_auth.h"
+#include "source3/include/auth.h"
+#include "source3/auth/proto.h"
+#include "nsswitch/libwbclient/wbclient.h"
+#include "lib/param/loadparm.h"
+#include "lib/util/base64.h"
+
+#if HAVE_KRB5
+#include "auth/kerberos/pac_utils.h"
+#endif
 
 #ifndef PAM_WINBIND_CONFIG_FILE
 #define PAM_WINBIND_CONFIG_FILE "/etc/security/pam_winbind.conf"
@@ -59,7 +69,7 @@ enum stdio_helper_mode {
        SQUID_2_5_BASIC,
        SQUID_2_5_NTLMSSP,
        NTLMSSP_CLIENT_1,
-       GSS_SPNEGO,
+       GSS_SPNEGO_SERVER,
        GSS_SPNEGO_CLIENT,
        NTLM_SERVER_1,
        NTLM_CHANGE_PASSWORD_1,
@@ -73,52 +83,66 @@ enum ntlm_auth_cli_state {
        CLIENT_ERROR
 };
 
-enum ntlm_auth_svr_state {
-       SERVER_INITIAL = 0,
-       SERVER_CHALLENGE,
-       SERVER_FINISHED,
-       SERVER_ERROR
-};
-
 struct ntlm_auth_state {
        TALLOC_CTX *mem_ctx;
        enum stdio_helper_mode helper_mode;
        enum ntlm_auth_cli_state cli_state;
-       enum ntlm_auth_svr_state svr_state;
        struct ntlmssp_state *ntlmssp_state;
-       struct gensec_security *gensec_security;
        uint32_t neg_flags;
        char *want_feature_list;
-       char *spnego_mech;
-       char *spnego_mech_oid;
        bool have_session_key;
        DATA_BLOB session_key;
        DATA_BLOB initial_message;
+       void *gensec_private_1;
 };
-
-typedef void (*stdio_helper_function)(struct ntlm_auth_state *state, char *buf,
-                                       int length);
-
-static void manage_squid_basic_request (struct ntlm_auth_state *state,
-                                       char *buf, int length);
-
-static void manage_squid_ntlmssp_request (struct ntlm_auth_state *state,
-                                       char *buf, int length);
-
-static void manage_client_ntlmssp_request (struct ntlm_auth_state *state,
-                                       char *buf, int length);
-
-static void manage_gss_spnego_request (struct ntlm_auth_state *state,
-                                       char *buf, int length);
-
-static void manage_gss_spnego_client_request (struct ntlm_auth_state *state,
-                                       char *buf, int length);
-
-static void manage_ntlm_server_1_request (struct ntlm_auth_state *state,
-                                       char *buf, int length);
-
-static void manage_ntlm_change_password_1_request(struct ntlm_auth_state *state,
-                                       char *buf, int length);
+typedef void (*stdio_helper_function)(enum stdio_helper_mode stdio_helper_mode,
+                                     struct loadparm_context *lp_ctx,
+                                     struct ntlm_auth_state *state, char *buf,
+                                       int length, void **private2);
+
+static void manage_gensec_request(enum stdio_helper_mode stdio_helper_mode,
+                                 struct loadparm_context *lp_ctx,
+                                 char *buf, int length, void **private1);
+
+static void manage_squid_request(enum stdio_helper_mode stdio_helper_mode,
+                                struct loadparm_context *lp_ctx,
+                                struct ntlm_auth_state *state,
+                                stdio_helper_function fn, void **private2);
+
+static void manage_squid_basic_request (enum stdio_helper_mode stdio_helper_mode,
+                                     struct loadparm_context *lp_ctx,
+                                     struct ntlm_auth_state *state,
+                                       char *buf, int length, void **private2);
+
+static void manage_squid_ntlmssp_request (enum stdio_helper_mode stdio_helper_mode,
+                                     struct loadparm_context *lp_ctx,
+                                     struct ntlm_auth_state *state,
+                                       char *buf, int length, void **private2);
+
+static void manage_client_ntlmssp_request (enum stdio_helper_mode stdio_helper_mode,
+                                     struct loadparm_context *lp_ctx,
+                                     struct ntlm_auth_state *state,
+                                       char *buf, int length, void **private2);
+
+static void manage_gss_spnego_request (enum stdio_helper_mode stdio_helper_mode,
+                                     struct loadparm_context *lp_ctx,
+                                     struct ntlm_auth_state *state,
+                                       char *buf, int length, void **private2);
+
+static void manage_gss_spnego_client_request (enum stdio_helper_mode stdio_helper_mode,
+                                     struct loadparm_context *lp_ctx,
+                                     struct ntlm_auth_state *state,
+                                       char *buf, int length, void **private2);
+
+static void manage_ntlm_server_1_request (enum stdio_helper_mode stdio_helper_mode,
+                                     struct loadparm_context *lp_ctx,
+                                     struct ntlm_auth_state *state,
+                                       char *buf, int length, void **private2);
+
+static void manage_ntlm_change_password_1_request(enum stdio_helper_mode stdio_helper_mode,
+                                     struct loadparm_context *lp_ctx,
+                                     struct ntlm_auth_state *state,
+                                       char *buf, int length, void **private2);
 
 static const struct {
        enum stdio_helper_mode mode;
@@ -129,7 +153,7 @@ static const struct {
        { SQUID_2_5_BASIC, "squid-2.5-basic", manage_squid_basic_request},
        { SQUID_2_5_NTLMSSP, "squid-2.5-ntlmssp", manage_squid_ntlmssp_request},
        { NTLMSSP_CLIENT_1, "ntlmssp-client-1", manage_client_ntlmssp_request},
-       { GSS_SPNEGO, "gss-spnego", manage_gss_spnego_request},
+       { GSS_SPNEGO_SERVER, "gss-spnego", manage_gss_spnego_request},
        { GSS_SPNEGO_CLIENT, "gss-spnego-client", manage_gss_spnego_client_request},
        { NTLM_SERVER_1, "ntlm-server-1", manage_ntlm_server_1_request},
        { NTLM_CHANGE_PASSWORD_1, "ntlm-change-password-1", manage_ntlm_change_password_1_request},
@@ -146,6 +170,8 @@ static DATA_BLOB opt_nt_response;
 static int request_lm_key;
 static int request_user_session_key;
 static int use_cached_creds;
+static int offline_logon;
+static int opt_allow_mschapv2;
 
 static const char *require_membership_of;
 static const char *require_membership_of_sid;
@@ -154,6 +180,79 @@ static const char *opt_pam_winbind_conf;
 const char *opt_target_service;
 const char *opt_target_hostname;
 
+
+/* This is a bit hairy, but the basic idea is to do a password callback
+   to the calling application.  The callback comes from within gensec */
+
+static void manage_gensec_get_pw_request(enum stdio_helper_mode stdio_helper_mode,
+                                        struct loadparm_context *lp_ctx,
+                                        struct ntlm_auth_state *state, char *buf, int length,
+                                        void **password)
+{
+       DATA_BLOB in;
+       if (strlen(buf) < 2) {
+               DEBUG(1, ("query [%s] invalid", buf));
+               printf("BH Query invalid\n");
+               return;
+       }
+
+       if (strlen(buf) > 3) {
+               in = base64_decode_data_blob(buf + 3);
+       } else {
+               in = data_blob(NULL, 0);
+       }
+
+       if (strncmp(buf, "PW ", 3) == 0) {
+
+               *password = talloc_strndup(NULL,
+                                          (const char *)in.data, in.length);
+
+               if (*password == NULL) {
+                       DEBUG(1, ("Out of memory\n"));
+                       printf("BH Out of memory\n");
+                       data_blob_free(&in);
+                       return;
+               }
+
+               printf("OK\n");
+               data_blob_free(&in);
+               return;
+       }
+       DEBUG(1, ("Asked for (and expected) a password\n"));
+       printf("BH Expected a password\n");
+       data_blob_free(&in);
+}
+
+/**
+ * Callback for password credentials.  This is not async, and when
+ * GENSEC and the credentials code is made async, it will look rather
+ * different.
+ */
+
+static const char *get_password(struct cli_credentials *credentials)
+{
+       TALLOC_CTX *frame = talloc_stackframe();
+       char *password = NULL;
+       struct ntlm_auth_state *state;
+
+       state = talloc_zero(frame, struct ntlm_auth_state);
+       if (state == NULL) {
+               DEBUG(0, ("squid_stream: Failed to talloc ntlm_auth_state\n"));
+               fprintf(stderr, "ERR\n");
+               exit(1);
+       }
+
+       state->mem_ctx = state;
+
+       /* Ask for a password */
+       printf("PW\n");
+
+       manage_squid_request(NUM_HELPER_MODES /* bogus */, NULL, state, manage_gensec_get_pw_request, (void **)&password);
+       talloc_steal(credentials, password);
+       TALLOC_FREE(frame);
+       return password;
+}
+
 /**
  * A limited set of features are defined with text strings as needed
  * by ntlm_auth
@@ -173,32 +272,36 @@ static void gensec_want_feature_list(struct gensec_security *state, char* featur
                DEBUG(10, ("want GENSEC_FEATURE_SEAL\n"));
                gensec_want_feature(state, GENSEC_FEATURE_SEAL);
        }
+       if (in_list("NTLMSSP_FEATURE_CCACHE", feature_list, true)) {
+               DEBUG(10, ("want GENSEC_FEATURE_NTLM_CCACHE\n"));
+               gensec_want_feature(state, GENSEC_FEATURE_NTLM_CCACHE);
+       }
 }
 
 static char winbind_separator(void)
 {
-       struct winbindd_response response;
+       struct wbcInterfaceDetails *details;
+       wbcErr ret;
        static bool got_sep;
        static char sep;
 
        if (got_sep)
                return sep;
 
-       ZERO_STRUCT(response);
-
-       /* Send off request */
-
-       if (winbindd_request_response(WINBINDD_INFO, NULL, &response) !=
-           NSS_STATUS_SUCCESS) {
-               d_printf("could not obtain winbind separator!\n");
+       ret = wbcInterfaceDetails(&details);
+       if (!WBC_ERROR_IS_OK(ret)) {
+               d_fprintf(stderr, "could not obtain winbind separator!\n");
                return *lp_winbind_separator();
        }
 
-       sep = response.data.info.winbind_separator;
+       sep = details->winbind_separator;
+
+       wbcFreeMemory(details);
+
        got_sep = True;
 
        if (!sep) {
-               d_printf("winbind separator was NULL!\n");
+               d_fprintf(stderr, "winbind separator was NULL!\n");
                return *lp_winbind_separator();
        }
 
@@ -207,24 +310,25 @@ static char winbind_separator(void)
 
 const char *get_winbind_domain(void)
 {
-       struct winbindd_response response;
+       struct wbcInterfaceDetails *details;
+       wbcErr ret;
 
        static fstring winbind_domain;
        if (*winbind_domain) {
                return winbind_domain;
        }
 
-       ZERO_STRUCT(response);
-
        /* Send off request */
 
-       if (winbindd_request_response(WINBINDD_DOMAIN_NAME, NULL, &response) !=
-           NSS_STATUS_SUCCESS) {
-               DEBUG(0, ("could not obtain winbind domain name!\n"));
+       ret = wbcInterfaceDetails(&details);
+       if (!WBC_ERROR_IS_OK(ret)) {
+               DEBUG(1, ("could not obtain winbind domain name!\n"));
                return lp_workgroup();
        }
 
-       fstrcpy(winbind_domain, response.data.domain_name);
+       fstrcpy(winbind_domain, details->netbios_domain);
+
+       wbcFreeMemory(details);
 
        return winbind_domain;
 
@@ -244,9 +348,9 @@ const char *get_winbind_netbios_name(void)
 
        /* Send off request */
 
-       if (winbindd_request_response(WINBINDD_NETBIOS_NAME, NULL, &response) !=
+       if (winbindd_request_response(NULL, WINBINDD_NETBIOS_NAME, NULL, &response) !=
            NSS_STATUS_SUCCESS) {
-               DEBUG(0, ("could not obtain winbind netbios name!\n"));
+               DEBUG(1, ("could not obtain winbind netbios name!\n"));
                return lp_netbios_name();
        }
 
@@ -284,14 +388,14 @@ static bool parse_ntlm_auth_domain_user(const char *domuser, fstring domain,
        fstrcpy(user, p+1);
        fstrcpy(domain, domuser);
        domain[PTR_DIFF(p, domuser)] = 0;
-       strupper_m(domain);
-
-       return True;
+       return strupper_m(domain);
 }
 
 static bool get_require_membership_sid(void) {
-       struct winbindd_request request;
-       struct winbindd_response response;
+       fstring domain, name, sidbuf;
+       struct wbcDomainSid sid;
+       enum wbcSidType type;
+       wbcErr ret;
 
        if (!require_membership_of) {
                return True;
@@ -303,25 +407,23 @@ static bool get_require_membership_sid(void) {
 
        /* Otherwise, ask winbindd for the name->sid request */
 
-       ZERO_STRUCT(request);
-       ZERO_STRUCT(response);
-
-       if (!parse_ntlm_auth_domain_user(require_membership_of, 
-                                        request.data.name.dom_name, 
-                                        request.data.name.name)) {
-               DEBUG(0, ("Could not parse %s into seperate domain/name parts!\n", 
+       if (!parse_ntlm_auth_domain_user(require_membership_of,
+                                        domain, name)) {
+               DEBUG(0, ("Could not parse %s into separate domain/name parts!\n",
                          require_membership_of));
                return False;
        }
 
-       if (winbindd_request_response(WINBINDD_LOOKUPNAME, &request, &response) !=
-           NSS_STATUS_SUCCESS) {
+       ret = wbcLookupName(domain, name, &sid, &type);
+       if (!WBC_ERROR_IS_OK(ret)) {
                DEBUG(0, ("Winbindd lookupname failed to resolve %s into a SID!\n", 
                          require_membership_of));
                return False;
        }
 
-       require_membership_of_sid = SMB_STRDUP(response.data.sid.sid);
+       wbcSidToStringBuf(&sid, sidbuf, sizeof(sidbuf));
+
+       require_membership_of_sid = SMB_STRDUP(sidbuf);
 
        if (require_membership_of_sid)
                return True;
@@ -337,23 +439,23 @@ static bool get_require_membership_sid(void) {
 int get_pam_winbind_config()
 {
        int ctrl = 0;
-       dictionary *d = NULL;
+       struct tiniparser_dictionary *d = NULL;
 
        if (!opt_pam_winbind_conf || !*opt_pam_winbind_conf) {
                opt_pam_winbind_conf = PAM_WINBIND_CONFIG_FILE;
        }
 
-       d = iniparser_load(discard_const_p(char, opt_pam_winbind_conf));
+       d = tiniparser_load(opt_pam_winbind_conf);
 
        if (!d) {
                return 0;
        }
 
-       if (iniparser_getboolean(d, discard_const_p(char, "global:krb5_auth"), false)) {
+       if (tiniparser_getboolean(d, "global:krb5_auth", false)) {
                ctrl |= WINBIND_KRB5_AUTH;
        }
 
-       iniparser_freedict(d);
+       tiniparser_freedict(d);
 
        return ctrl;
 }
@@ -384,13 +486,17 @@ static bool check_plaintext_auth(const char *user, const char *pass,
                        sizeof(request.data.auth.require_membership_of_sid));
        }
 
-       result = winbindd_request_response(WINBINDD_PAM_AUTH, &request, &response);
+       if (offline_logon) {
+               request.flags |= WBFLAG_PAM_CACHED_LOGIN;
+       }
+
+       result = winbindd_request_response(NULL, WINBINDD_PAM_AUTH, &request, &response);
 
        /* Display response */
 
        if (stdout_diagnostics) {
                if ((result != NSS_STATUS_SUCCESS) && (response.data.auth.nt_status == 0)) {
-                       d_printf("Reading winbind reply failed! (0x01)\n");
+                       d_fprintf(stderr, "Reading winbind reply failed! (0x01)\n");
                }
 
                d_printf("%s: %s (0x%x)\n",
@@ -419,10 +525,11 @@ NTSTATUS contact_winbind_auth_crap(const char *username,
                                   const DATA_BLOB *challenge,
                                   const DATA_BLOB *lm_response,
                                   const DATA_BLOB *nt_response,
-                                  uint32 flags,
-                                  uint32 extra_logon_parameters,
-                                  uint8 lm_key[8],
-                                  uint8 user_session_key[16],
+                                  uint32_t flags,
+                                  uint32_t extra_logon_parameters,
+                                  uint8_t lm_key[8],
+                                  uint8_t user_session_key[16],
+                                  uint8_t *pauthoritative,
                                   char **error_string,
                                   char **unix_name)
 {
@@ -431,6 +538,8 @@ NTSTATUS contact_winbind_auth_crap(const char *username,
        struct winbindd_request request;
        struct winbindd_response response;
 
+       *pauthoritative = 1;
+
        if (!get_require_membership_sid()) {
                return NT_STATUS_INVALID_PARAMETER;
        }
@@ -443,6 +552,10 @@ NTSTATUS contact_winbind_auth_crap(const char *username,
        request.data.auth_crap.logon_parameters = extra_logon_parameters
                | MSV1_0_ALLOW_WORKSTATION_TRUST_ACCOUNT | MSV1_0_ALLOW_SERVER_TRUST_ACCOUNT;
 
+       if (opt_allow_mschapv2) {
+                       request.data.auth_crap.logon_parameters |= MSV1_0_ALLOW_MSVCHAPV2;
+       }
+
        if (require_membership_of_sid)
                fstrcpy(request.data.auth_crap.require_membership_of_sid, require_membership_of_sid);
 
@@ -479,7 +592,7 @@ NTSTATUS contact_winbind_auth_crap(const char *username,
                 request.data.auth_crap.nt_resp_len = nt_response->length;
        }
 
-       result = winbindd_request_response(WINBINDD_PAM_AUTH_CRAP, &request, &response);
+       result = winbindd_request_response(NULL, WINBINDD_PAM_AUTH_CRAP, &request, &response);
        SAFE_FREE(request.extra_data.data);
 
        /* Display response */
@@ -496,6 +609,7 @@ NTSTATUS contact_winbind_auth_crap(const char *username,
        if (!NT_STATUS_IS_OK(nt_status)) {
                if (error_string) 
                        *error_string = smb_xstrdup(response.data.auth.error_string);
+               *pauthoritative = response.data.auth.authoritative;
                winbindd_free_response(&response);
                return nt_status;
        }
@@ -574,7 +688,7 @@ static NTSTATUS contact_winbind_change_pswd_auth_crap(const char *username,
                request.data.chng_pswd_auth_crap.old_lm_hash_enc_len = old_lm_hash_enc.length;
        }
 
-       result = winbindd_request_response(WINBINDD_PAM_CHNG_PSWD_AUTH_CRAP, &request, &response);
+       result = winbindd_request_response(NULL, WINBINDD_PAM_CHNG_PSWD_AUTH_CRAP, &request, &response);
 
        /* Display response */
 
@@ -608,24 +722,179 @@ static NTSTATUS ntlm_auth_generate_session_info(struct auth4_context *auth_conte
                                                uint32_t session_info_flags,
                                                struct auth_session_info **session_info_out)
 {
-       char *unix_username = (char *)server_returned_info;
-       struct auth_session_info *session_info = talloc_zero(mem_ctx, struct auth_session_info);
-       if (!session_info) {
+       const char *unix_username = (const char *)server_returned_info;
+       bool ok;
+       struct dom_sid *sids = NULL;
+       struct auth_session_info *session_info = NULL;
+
+       session_info = talloc_zero(mem_ctx, struct auth_session_info);
+       if (session_info == NULL) {
                return NT_STATUS_NO_MEMORY;
        }
 
        session_info->unix_info = talloc_zero(session_info, struct auth_user_info_unix);
-       if (!session_info->unix_info) {
+       if (session_info->unix_info == NULL) {
                TALLOC_FREE(session_info);
                return NT_STATUS_NO_MEMORY;
        }
-       session_info->unix_info->unix_name = talloc_steal(session_info->unix_info, unix_username);
+       session_info->unix_info->unix_name = talloc_strdup(session_info->unix_info,
+                                                          unix_username);
+       if (session_info->unix_info->unix_name == NULL) {
+               TALLOC_FREE(session_info);
+               return NT_STATUS_NO_MEMORY;
+       }
+
+       session_info->security_token = talloc_zero(session_info, struct security_token);
+       if (session_info->security_token == NULL) {
+               TALLOC_FREE(session_info);
+               return NT_STATUS_NO_MEMORY;
+       }
+
+       sids = talloc_zero_array(session_info->security_token,
+                                struct dom_sid, 3);
+       if (sids == NULL) {
+               TALLOC_FREE(session_info);
+               return NT_STATUS_NO_MEMORY;
+       }
+       ok = dom_sid_parse(SID_WORLD, &sids[0]);
+       if (!ok) {
+               TALLOC_FREE(session_info);
+               return NT_STATUS_INTERNAL_ERROR;
+       }
+       ok = dom_sid_parse(SID_NT_NETWORK, &sids[1]);
+       if (!ok) {
+               TALLOC_FREE(session_info);
+               return NT_STATUS_INTERNAL_ERROR;
+       }
+       ok = dom_sid_parse(SID_NT_AUTHENTICATED_USERS, &sids[2]);
+       if (!ok) {
+               TALLOC_FREE(session_info);
+               return NT_STATUS_INTERNAL_ERROR;
+       }
+
+       session_info->security_token->num_sids = talloc_array_length(sids);
+       session_info->security_token->sids = sids;
 
        *session_info_out = session_info;
 
        return NT_STATUS_OK;
 }
 
+static NTSTATUS ntlm_auth_generate_session_info_pac(struct auth4_context *auth_ctx,
+                                                   TALLOC_CTX *mem_ctx,
+                                                   struct smb_krb5_context *smb_krb5_context,
+                                                   DATA_BLOB *pac_blob,
+                                                   const char *princ_name,
+                                                   const struct tsocket_address *remote_address,
+                                                   uint32_t session_info_flags,
+                                                   struct auth_session_info **session_info)
+{
+       TALLOC_CTX *tmp_ctx;
+       struct PAC_LOGON_INFO *logon_info = NULL;
+       char *unixuser;
+       NTSTATUS status;
+       char *domain = NULL;
+       char *realm = NULL;
+       char *user = NULL;
+       char *p;
+
+       tmp_ctx = talloc_new(mem_ctx);
+       if (!tmp_ctx) {
+               return NT_STATUS_NO_MEMORY;
+       }
+
+       if (pac_blob) {
+#ifdef HAVE_KRB5
+               status = kerberos_pac_logon_info(tmp_ctx, *pac_blob, NULL, NULL,
+                                                NULL, NULL, 0, &logon_info);
+#else
+               status = NT_STATUS_ACCESS_DENIED;
+#endif
+               if (!NT_STATUS_IS_OK(status)) {
+                       goto done;
+               }
+       }
+
+       DEBUG(3, ("Kerberos ticket principal name is [%s]\n", princ_name));
+
+       p = strchr_m(princ_name, '@');
+       if (!p) {
+               DEBUG(3, ("[%s] Doesn't look like a valid principal\n",
+                         princ_name));
+               return NT_STATUS_LOGON_FAILURE;
+       }
+
+       user = talloc_strndup(mem_ctx, princ_name, p - princ_name);
+       if (!user) {
+               return NT_STATUS_NO_MEMORY;
+       }
+
+       realm = talloc_strdup(talloc_tos(), p + 1);
+       if (!realm) {
+               return NT_STATUS_NO_MEMORY;
+       }
+
+       if (!strequal(realm, lp_realm())) {
+               DEBUG(3, ("Ticket for foreign realm %s@%s\n", user, realm));
+               if (!lp_allow_trusted_domains()) {
+                       return NT_STATUS_LOGON_FAILURE;
+               }
+       }
+
+       if (logon_info && logon_info->info3.base.logon_domain.string) {
+               domain = talloc_strdup(mem_ctx,
+                                       logon_info->info3.base.logon_domain.string);
+               if (!domain) {
+                       return NT_STATUS_NO_MEMORY;
+               }
+               DEBUG(10, ("Domain is [%s] (using PAC)\n", domain));
+       } else {
+
+               /* If we have winbind running, we can (and must) shorten the
+                  username by using the short netbios name. Otherwise we will
+                  have inconsistent user names. With Kerberos, we get the
+                  fully qualified realm, with ntlmssp we get the short
+                  name. And even w2k3 does use ntlmssp if you for example
+                  connect to an ip address. */
+
+               wbcErr wbc_status;
+               struct wbcDomainInfo *info = NULL;
+
+               DEBUG(10, ("Mapping [%s] to short name using winbindd\n",
+                          realm));
+
+               wbc_status = wbcDomainInfo(realm, &info);
+
+               if (WBC_ERROR_IS_OK(wbc_status)) {
+                       domain = talloc_strdup(mem_ctx,
+                                               info->short_name);
+                       wbcFreeMemory(info);
+               } else {
+                       DEBUG(3, ("Could not find short name: %s\n",
+                                 wbcErrorString(wbc_status)));
+                       domain = talloc_strdup(mem_ctx, realm);
+               }
+               if (!domain) {
+                       return NT_STATUS_NO_MEMORY;
+               }
+               DEBUG(10, ("Domain is [%s] (using Winbind)\n", domain));
+       }
+
+       unixuser = talloc_asprintf(tmp_ctx, "%s%c%s", domain, winbind_separator(), user);
+       if (!unixuser) {
+               status = NT_STATUS_NO_MEMORY;
+               goto done;
+       }
+
+       status = ntlm_auth_generate_session_info(auth_ctx, mem_ctx, unixuser, NULL, session_info_flags, session_info);
+
+done:
+       TALLOC_FREE(tmp_ctx);
+       return status;
+}
+
+
+
 /**
  * Return the challenge as determined by the authentication subsystem 
  * @return an 8 byte random challenge
@@ -647,8 +916,6 @@ static NTSTATUS ntlm_auth_get_challenge(struct auth4_context *auth_ctx,
                auth_ctx->challenge.data                = data_blob_talloc(auth_ctx, chal, 8);
                NT_STATUS_HAVE_NO_MEMORY(auth_ctx->challenge.data.data);
                auth_ctx->challenge.set_by              = "random";
-
-               auth_ctx->challenge.may_be_modified     = true;
        }
 
        DEBUG(10,("auth_get_challenge: challenge set by %s\n",
@@ -657,16 +924,6 @@ static NTSTATUS ntlm_auth_get_challenge(struct auth4_context *auth_ctx,
        return NT_STATUS_OK;
 }
 
-/**
- * Some authentication methods 'fix' the challenge, so we may not be able to set it
- *
- * @return If the effective challenge used by the auth subsystem may be modified
- */
-static bool ntlm_auth_may_set_challenge(struct auth4_context *auth_ctx)
-{
-       return auth_ctx->challenge.may_be_modified;
-}
-
 /**
  * NTLM2 authentication modifies the effective challenge, 
  * @param challenge The new challenge value
@@ -691,14 +948,14 @@ static NTSTATUS ntlm_auth_set_challenge(struct auth4_context *auth_ctx, const ui
 static NTSTATUS winbind_pw_check(struct auth4_context *auth4_context, 
                                 TALLOC_CTX *mem_ctx,
                                 const struct auth_usersupplied_info *user_info, 
+                                uint8_t *pauthoritative,
                                 void **server_returned_info,
                                 DATA_BLOB *session_key, DATA_BLOB *lm_session_key)
 {
-       static const char zeros[16] = { 0, };
        NTSTATUS nt_status;
        char *error_string = NULL;
-       uint8 lm_key[8]; 
-       uint8 user_sess_key[16]; 
+       uint8_t lm_key[8]; 
+       uint8_t user_sess_key[16]; 
        char *unix_name = NULL;
 
        nt_status = contact_winbind_auth_crap(user_info->client.account_name, user_info->client.domain_name, 
@@ -709,16 +966,17 @@ static NTSTATUS winbind_pw_check(struct auth4_context *auth4_context,
                                              WBFLAG_PAM_LMKEY | WBFLAG_PAM_USER_SESSION_KEY | WBFLAG_PAM_UNIX_NAME,
                                              0,
                                              lm_key, user_sess_key, 
+                                             pauthoritative,
                                              &error_string, &unix_name);
 
        if (NT_STATUS_IS_OK(nt_status)) {
-               if (memcmp(lm_key, zeros, 8) != 0) {
+               if (!all_zero(lm_key, 8)) {
                        *lm_session_key = data_blob_talloc(mem_ctx, NULL, 16);
                        memcpy(lm_session_key->data, lm_key, 8);
                        memset(lm_session_key->data+8, '\0', 8);
                }
 
-               if (memcmp(user_sess_key, zeros, 16) != 0) {
+               if (!all_zero(user_sess_key, 16)) {
                        *session_key = data_blob_talloc(mem_ctx, user_sess_key, 16);
                }
                *server_returned_info = talloc_strdup(mem_ctx,
@@ -738,7 +996,8 @@ static NTSTATUS winbind_pw_check(struct auth4_context *auth4_context,
 
 static NTSTATUS local_pw_check(struct auth4_context *auth4_context, 
                                TALLOC_CTX *mem_ctx,
-                               const struct auth_usersupplied_info *user_info, 
+                               const struct auth_usersupplied_info *user_info,
+                               uint8_t *pauthoritative,
                                void **server_returned_info,
                                DATA_BLOB *session_key, DATA_BLOB *lm_session_key)
 {
@@ -747,6 +1006,8 @@ static NTSTATUS local_pw_check(struct auth4_context *auth4_context,
 
        nt_lm_owf_gen (opt_password, nt_pw.hash, lm_pw.hash);
 
+       *pauthoritative = 1;
+
        nt_status = ntlm_password_check(mem_ctx,
                                        true, true, 0,
                                        &auth4_context->challenge.data,
@@ -771,57 +1032,75 @@ static NTSTATUS local_pw_check(struct auth4_context *auth4_context,
        return nt_status;
 }
 
-static NTSTATUS ntlm_auth_start_ntlmssp_client(struct ntlmssp_state **client_ntlmssp_state)
+static NTSTATUS ntlm_auth_prepare_gensec_client(TALLOC_CTX *mem_ctx,
+                                               struct loadparm_context *lp_ctx,
+                                               struct gensec_security **gensec_security_out)
 {
-       NTSTATUS status;
-       if ( (opt_username == NULL) || (opt_domain == NULL) ) {
-               status = NT_STATUS_UNSUCCESSFUL;
-               DEBUG(1, ("Need username and domain for NTLMSSP\n"));
-               return NT_STATUS_INVALID_PARAMETER;
-       }
+       struct gensec_security *gensec_security = NULL;
+       NTSTATUS nt_status;
+       TALLOC_CTX *tmp_ctx;
+       const struct gensec_security_ops **backends = NULL;
+       struct gensec_settings *gensec_settings = NULL;
+       size_t idx = 0;
+
+       tmp_ctx = talloc_new(mem_ctx);
+       NT_STATUS_HAVE_NO_MEMORY(tmp_ctx);
 
-       status = ntlmssp_client_start(NULL,
-                                     lp_netbios_name(),
-                                     lp_workgroup(),
-                                     lp_client_ntlmv2_auth(),
-                                     client_ntlmssp_state);
+       gensec_settings = lpcfg_gensec_settings(tmp_ctx, lp_ctx);
+       if (gensec_settings == NULL) {
+               DEBUG(10, ("lpcfg_gensec_settings failed\n"));
+               TALLOC_FREE(tmp_ctx);
+               return NT_STATUS_NO_MEMORY;
+       }
 
-       if (!NT_STATUS_IS_OK(status)) {
-               DEBUG(1, ("Could not start NTLMSSP client: %s\n",
-                         nt_errstr(status)));
-               TALLOC_FREE(*client_ntlmssp_state);
-               return status;
+       backends = talloc_zero_array(gensec_settings,
+                                    const struct gensec_security_ops *, 4);
+       if (backends == NULL) {
+               TALLOC_FREE(tmp_ctx);
+               return NT_STATUS_NO_MEMORY;
        }
+       gensec_settings->backends = backends;
 
-       status = ntlmssp_set_username(*client_ntlmssp_state, opt_username);
+       gensec_init();
 
-       if (!NT_STATUS_IS_OK(status)) {
-               DEBUG(1, ("Could not set username: %s\n",
-                         nt_errstr(status)));
-               TALLOC_FREE(*client_ntlmssp_state);
-               return status;
-       }
+       /* These need to be in priority order, krb5 before NTLMSSP */
+#if defined(HAVE_KRB5)
+       backends[idx++] = &gensec_gse_krb5_security_ops;
+#endif
+
+       backends[idx++] = gensec_security_by_oid(NULL, GENSEC_OID_NTLMSSP);
 
-       status = ntlmssp_set_domain(*client_ntlmssp_state, opt_domain);
+       backends[idx++] = gensec_security_by_oid(NULL, GENSEC_OID_SPNEGO);
 
-       if (!NT_STATUS_IS_OK(status)) {
-               DEBUG(1, ("Could not set domain: %s\n",
-                         nt_errstr(status)));
-               TALLOC_FREE(*client_ntlmssp_state);
-               return status;
+       nt_status = gensec_client_start(NULL, &gensec_security,
+                                       gensec_settings);
+       if (!NT_STATUS_IS_OK(nt_status)) {
+               TALLOC_FREE(tmp_ctx);
+               return nt_status;
        }
 
-       if (opt_password) {
-               status = ntlmssp_set_password(*client_ntlmssp_state, opt_password);
+       talloc_unlink(tmp_ctx, gensec_settings);
 
-               if (!NT_STATUS_IS_OK(status)) {
-                       DEBUG(1, ("Could not set password: %s\n",
-                                 nt_errstr(status)));
-                       TALLOC_FREE(*client_ntlmssp_state);
-                       return status;
+       if (opt_target_service != NULL) {
+               nt_status = gensec_set_target_service(gensec_security,
+                                                     opt_target_service);
+               if (!NT_STATUS_IS_OK(nt_status)) {
+                       TALLOC_FREE(tmp_ctx);
+                       return nt_status;
+               }
+       }
+
+       if (opt_target_hostname != NULL) {
+               nt_status = gensec_set_target_hostname(gensec_security,
+                                                      opt_target_hostname);
+               if (!NT_STATUS_IS_OK(nt_status)) {
+                       TALLOC_FREE(tmp_ctx);
+                       return nt_status;
                }
        }
 
+       *gensec_security_out = talloc_steal(mem_ctx, gensec_security);
+       TALLOC_FREE(tmp_ctx);
        return NT_STATUS_OK;
 }
 
@@ -833,9 +1112,9 @@ static struct auth4_context *make_auth4_context_ntlm_auth(TALLOC_CTX *mem_ctx, b
                return NULL;
        }
        auth4_context->generate_session_info = ntlm_auth_generate_session_info;
+       auth4_context->generate_session_info_pac = ntlm_auth_generate_session_info_pac;
        auth4_context->get_ntlm_challenge = ntlm_auth_get_challenge;
        auth4_context->set_ntlm_challenge = ntlm_auth_set_challenge;
-       auth4_context->challenge_may_be_modified = ntlm_auth_may_set_challenge;
        if (local_pw) {
                auth4_context->check_ntlm_password = local_pw_check;
        } else {
@@ -845,33 +1124,30 @@ static struct auth4_context *make_auth4_context_ntlm_auth(TALLOC_CTX *mem_ctx, b
        return auth4_context;
 }
 
-static NTSTATUS ntlm_auth_start_ntlmssp_server(TALLOC_CTX *mem_ctx,
-                                              struct gensec_security **gensec_security_out)
+static NTSTATUS ntlm_auth_prepare_gensec_server(TALLOC_CTX *mem_ctx,
+                                               struct loadparm_context *lp_ctx,
+                                               struct gensec_security **gensec_security_out)
 {
        struct gensec_security *gensec_security;
        NTSTATUS nt_status;
 
-       TALLOC_CTX *tmp_ctx = talloc_new(mem_ctx);
-       NT_STATUS_HAVE_NO_MEMORY(tmp_ctx);
-
+       TALLOC_CTX *tmp_ctx;
+       const struct gensec_security_ops **backends;
        struct gensec_settings *gensec_settings;
-       struct loadparm_context *lp_ctx;
        size_t idx = 0;
        struct cli_credentials *server_credentials;
        
-       struct auth4_context *auth4_context = make_auth4_context_ntlm_auth(tmp_ctx, opt_password);
+       struct auth4_context *auth4_context;
+
+       tmp_ctx = talloc_new(mem_ctx);
+       NT_STATUS_HAVE_NO_MEMORY(tmp_ctx);
+
+       auth4_context = make_auth4_context_ntlm_auth(tmp_ctx, opt_password);
        if (auth4_context == NULL) {
                TALLOC_FREE(tmp_ctx);
                return NT_STATUS_NO_MEMORY;
        }
        
-       lp_ctx = loadparm_init_s3(tmp_ctx, loadparm_s3_context());
-       if (lp_ctx == NULL) {
-               DEBUG(10, ("loadparm_init_s3 failed\n"));
-               TALLOC_FREE(tmp_ctx);
-               return NT_STATUS_INVALID_SERVER_STATE;
-       }
-       
        gensec_settings = lpcfg_gensec_settings(tmp_ctx, lp_ctx);
        if (lp_ctx == NULL) {
                DEBUG(10, ("lpcfg_gensec_settings failed\n"));
@@ -900,25 +1176,26 @@ static NTSTATUS ntlm_auth_start_ntlmssp_server(TALLOC_CTX *mem_ctx,
        gensec_settings->server_dns_name = strlower_talloc(gensec_settings,
                                                           get_mydnsfullname());
        
-       gensec_settings->backends = talloc_zero_array(gensec_settings,
-                                                     struct gensec_security_ops *, 4);
+       backends = talloc_zero_array(gensec_settings,
+                                    const struct gensec_security_ops *, 4);
        
-       if (gensec_settings->backends == NULL) {
+       if (backends == NULL) {
                TALLOC_FREE(tmp_ctx);
                return NT_STATUS_NO_MEMORY;
        }
-       
+       gensec_settings->backends = backends;
+
        gensec_init();
        
-       gensec_settings->backends[idx++] = gensec_security_by_oid(NULL, GENSEC_OID_NTLMSSP);
-       
-#if defined(HAVE_KRB5) && defined(HAVE_GSS_WRAP_IOV)
-       gensec_settings->backends[idx++] = &gensec_gse_krb5_security_ops;
+       /* These need to be in priority order, krb5 before NTLMSSP */
+#if defined(HAVE_KRB5)
+       backends[idx++] = &gensec_gse_krb5_security_ops;
 #endif
-       
-       gensec_settings->backends[idx++] = gensec_security_by_oid(NULL,
-                                                                 GENSEC_OID_SPNEGO);
-       
+
+       backends[idx++] = gensec_security_by_oid(NULL, GENSEC_OID_NTLMSSP);
+
+       backends[idx++] = gensec_security_by_oid(NULL, GENSEC_OID_SPNEGO);
+
        /*
         * This is anonymous for now, because we just use it
         * to set the kerberos state at the moment
@@ -931,7 +1208,7 @@ static NTSTATUS ntlm_auth_start_ntlmssp_server(TALLOC_CTX *mem_ctx,
        
        cli_credentials_set_conf(server_credentials, lp_ctx);
        
-       if (lp_security() == SEC_ADS || USE_KERBEROS_KEYTAB) {
+       if (lp_server_role() == ROLE_ACTIVE_DIRECTORY_DC || lp_security() == SEC_ADS || USE_KERBEROS_KEYTAB) {
                cli_credentials_set_kerberos_state(server_credentials, CRED_AUTO_USE_KERBEROS);
        } else {
                cli_credentials_set_kerberos_state(server_credentials, CRED_DONT_USE_KERBEROS);
@@ -946,1274 +1223,438 @@ static NTSTATUS ntlm_auth_start_ntlmssp_server(TALLOC_CTX *mem_ctx,
        }
        
        gensec_set_credentials(gensec_security, server_credentials);
-       
-       gensec_want_feature(gensec_security, GENSEC_FEATURE_SIGN);
-       gensec_want_feature(gensec_security, GENSEC_FEATURE_SEAL);
+
+       /*
+        * TODO: Allow the caller to pass their own description here
+        * via a command-line option
+        */
+       nt_status = gensec_set_target_service_description(gensec_security,
+                                                         "ntlm_auth");
+       if (!NT_STATUS_IS_OK(nt_status)) {
+               TALLOC_FREE(tmp_ctx);
+               return nt_status;
+       }
 
        talloc_unlink(tmp_ctx, lp_ctx);
        talloc_unlink(tmp_ctx, server_credentials);
        talloc_unlink(tmp_ctx, gensec_settings);
        talloc_unlink(tmp_ctx, auth4_context);
 
-       nt_status = gensec_start_mech_by_oid(gensec_security, GENSEC_OID_NTLMSSP);
-       if (!NT_STATUS_IS_OK(nt_status)) {
-               TALLOC_FREE(tmp_ctx);
-               return nt_status;
-       }
-       
        *gensec_security_out = talloc_steal(mem_ctx, gensec_security);
        TALLOC_FREE(tmp_ctx);
        return NT_STATUS_OK;
 }
 
-/*******************************************************************
- Used by firefox to drive NTLM auth to IIS servers.
-*******************************************************************/
+static void manage_client_ntlmssp_request(enum stdio_helper_mode stdio_helper_mode,
+                                  struct loadparm_context *lp_ctx,
+                                  struct ntlm_auth_state *state,
+                                               char *buf, int length, void **private2)
+{
+       manage_gensec_request(stdio_helper_mode, lp_ctx, buf, length, &state->gensec_private_1);
+       return;
+}
 
-static NTSTATUS do_ccache_ntlm_auth(DATA_BLOB initial_msg, DATA_BLOB challenge_msg,
-                               DATA_BLOB *reply)
+static void manage_squid_basic_request(enum stdio_helper_mode stdio_helper_mode,
+                                  struct loadparm_context *lp_ctx,
+                                  struct ntlm_auth_state *state,
+                                       char *buf, int length, void **private2)
 {
-       struct winbindd_request wb_request;
-       struct winbindd_response wb_response;
-       int ctrl = 0;
-       NSS_STATUS result;
+       char *user, *pass;      
+       user=buf;
 
-       /* get winbindd to do the ntlmssp step on our behalf */
-       ZERO_STRUCT(wb_request);
-       ZERO_STRUCT(wb_response);
+       pass=(char *)memchr(buf,' ',length);
+       if (!pass) {
+               DEBUG(2, ("Password not found. Denying access\n"));
+               printf("ERR\n");
+               return;
+       }
+       *pass='\0';
+       pass++;
 
-       /*
-        * This is tricky here. If we set krb5_auth in pam_winbind.conf
-        * creds for users in trusted domain will be stored the winbindd
-        * child of the trusted domain. If we ask the primary domain for
-        * ntlm_ccache_auth, it will fail. So, we have to ask the trusted
-        * domain's child for ccache_ntlm_auth. that is to say, we have to 
-        * set WBFLAG_PAM_CONTACT_TRUSTDOM in request.flags.
-        */
-       ctrl = get_pam_winbind_config();
+       if (state->helper_mode == SQUID_2_5_BASIC) {
+               rfc1738_unescape(user);
+               rfc1738_unescape(pass);
+       }
 
-       if (ctrl & WINBIND_KRB5_AUTH) {
-               wb_request.flags |= WBFLAG_PAM_CONTACT_TRUSTDOM;
+       if (check_plaintext_auth(user, pass, False)) {
+               printf("OK\n");
+       } else {
+               printf("ERR\n");
        }
+}
 
-       fstr_sprintf(wb_request.data.ccache_ntlm_auth.user,
-               "%s%c%s", opt_domain, winbind_separator(), opt_username);
-       wb_request.data.ccache_ntlm_auth.uid = geteuid();
-       wb_request.data.ccache_ntlm_auth.initial_blob_len = initial_msg.length;
-       wb_request.data.ccache_ntlm_auth.challenge_blob_len = challenge_msg.length;
-       wb_request.extra_len = initial_msg.length + challenge_msg.length;
+static void manage_gensec_request(enum stdio_helper_mode stdio_helper_mode,
+                                 struct loadparm_context *lp_ctx,
+                                 char *buf, int length, void **private1)
+{
+       DATA_BLOB in;
+       DATA_BLOB out = data_blob(NULL, 0);
+       char *out_base64 = NULL;
+       const char *reply_arg = NULL;
+       struct gensec_ntlm_state {
+               struct gensec_security *gensec_state;
+               const char *set_password;
+       };
+       struct gensec_ntlm_state *state;
 
-       if (wb_request.extra_len > 0) {
-               wb_request.extra_data.data = SMB_MALLOC_ARRAY(char, wb_request.extra_len);
-               if (wb_request.extra_data.data == NULL) {
-                       return NT_STATUS_NO_MEMORY;
-               }
+       NTSTATUS nt_status;
+       bool first = false;
+       const char *reply_code;
+       struct cli_credentials *creds;
 
-               memcpy(wb_request.extra_data.data, initial_msg.data, initial_msg.length);
-               memcpy(wb_request.extra_data.data + initial_msg.length,
-                       challenge_msg.data, challenge_msg.length);
-       }
+       static char *want_feature_list = NULL;
+       static DATA_BLOB session_key;
 
-       result = winbindd_request_response(WINBINDD_CCACHE_NTLMAUTH, &wb_request, &wb_response);
-       SAFE_FREE(wb_request.extra_data.data);
+       TALLOC_CTX *mem_ctx;
 
-       if (result != NSS_STATUS_SUCCESS) {
-               winbindd_free_response(&wb_response);
-               return NT_STATUS_UNSUCCESSFUL;
-       }
+       mem_ctx = talloc_named(NULL, 0, "manage_gensec_request internal mem_ctx");
 
-       if (reply) {
-               *reply = data_blob(wb_response.extra_data.data,
-                               wb_response.data.ccache_ntlm_auth.auth_blob_len);
-               if (wb_response.data.ccache_ntlm_auth.auth_blob_len > 0 &&
-                               reply->data == NULL) {
-                       winbindd_free_response(&wb_response);
-                       return NT_STATUS_NO_MEMORY;
+       if (*private1) {
+               state = (struct gensec_ntlm_state *)*private1;
+       } else {
+               state = talloc_zero(NULL, struct gensec_ntlm_state);
+               if (!state) {
+                       printf("BH No Memory\n");
+                       exit(1);
+               }
+               *private1 = state;
+               if (opt_password) {
+                       state->set_password = opt_password;
                }
        }
 
-       winbindd_free_response(&wb_response);
-       return NT_STATUS_MORE_PROCESSING_REQUIRED;
-}
-
-static void manage_squid_ntlmssp_request_int(struct ntlm_auth_state *state,
-                                            char *buf, int length,
-                                            TALLOC_CTX *mem_ctx,
-                                            char **response)
-{
-       DATA_BLOB request, reply;
-       NTSTATUS nt_status;
-
-       if (strlen(buf) < 2) {
-               DEBUG(1, ("NTLMSSP query [%s] invalid\n", buf));
-               *response = talloc_strdup(mem_ctx, "BH NTLMSSP query invalid");
-               return;
-       }
-
-       if (strlen(buf) > 3) {
-               if(strncmp(buf, "SF ", 3) == 0){
-                       DEBUG(10, ("Setting flags to negotioate\n"));
-                       TALLOC_FREE(state->want_feature_list);
-                       state->want_feature_list = talloc_strdup(state->mem_ctx,
-                                       buf+3);
-                       *response = talloc_strdup(mem_ctx, "OK");
-                       return;
-               }
-               request = base64_decode_data_blob(buf + 3);
-       } else {
-               request = data_blob_null;
-       }
-
-       if ((strncmp(buf, "PW ", 3) == 0)) {
-               /* The calling application wants us to use a local password
-                * (rather than winbindd) */
-
-               opt_password = SMB_STRNDUP((const char *)request.data,
-                               request.length);
-
-               if (opt_password == NULL) {
-                       DEBUG(1, ("Out of memory\n"));
-                       *response = talloc_strdup(mem_ctx, "BH Out of memory");
-                       data_blob_free(&request);
-                       return;
-               }
-
-               *response = talloc_strdup(mem_ctx, "OK");
-               data_blob_free(&request);
-               return;
-       }
-
-       if (strncmp(buf, "YR", 2) == 0) {
-               TALLOC_FREE(state->gensec_security);
-               state->svr_state = SERVER_INITIAL;
-       } else if (strncmp(buf, "KK", 2) == 0) {
-               /* No special preprocessing required */
-       } else if (strncmp(buf, "GF", 2) == 0) {
-               DEBUG(10, ("Requested negotiated NTLMSSP flags\n"));
-
-               if (state->svr_state == SERVER_FINISHED) {
-                       *response = talloc_asprintf(mem_ctx, "GF 0x%08x",
-                                                   gensec_ntlmssp_neg_flags(state->gensec_security));
-               }
-               else {
-                       *response = talloc_strdup(mem_ctx, "BH\n");
-               }
-               data_blob_free(&request);
-               return;
-       } else if (strncmp(buf, "GK", 2) == 0) {
-               DEBUG(10, ("Requested NTLMSSP session key\n"));
-               if(state->have_session_key) {
-                       DATA_BLOB session_key;
-                       char *key64;
-                       nt_status = gensec_session_key(state->gensec_security, talloc_tos(), &session_key);
-                       if (!NT_STATUS_IS_OK(nt_status)) {
-                               *response = talloc_asprintf(
-                                       mem_ctx, "BH %s", nt_errstr(nt_status));
-                               return;
-                       }
-                       key64 = base64_encode_data_blob(state->mem_ctx, session_key);
-                       data_blob_free(&session_key);
-                       *response = talloc_asprintf(mem_ctx, "GK %s",
-                                                key64 ? key64 : "<NULL>");
-                       TALLOC_FREE(key64);
-               } else {
-                       *response = talloc_strdup(mem_ctx, "BH");
-               }
-
-               data_blob_free(&request);
-               return;
-       } else {
-               DEBUG(1, ("NTLMSSP query [%s] invalid\n", buf));
-               *response = talloc_strdup(mem_ctx, "BH NTLMSSP query invalid");
-               return;
-       }
-
-       if (!state->gensec_security) {
-               nt_status = ntlm_auth_start_ntlmssp_server(state, 
-                               &state->gensec_security);
-               if (!NT_STATUS_IS_OK(nt_status)) {
-                       *response = talloc_asprintf(
-                               mem_ctx, "BH %s", nt_errstr(nt_status));
-                       return;
-               }
-               gensec_want_feature_list(state->gensec_security,
-                                        state->want_feature_list);
-       }
-
-       DEBUG(10, ("got NTLMSSP packet:\n"));
-       dump_data(10, request.data, request.length);
-
-       nt_status = gensec_update(state->gensec_security, talloc_tos(), NULL, request, &reply);
-
-       if (NT_STATUS_EQUAL(nt_status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
-               char *reply_base64 = base64_encode_data_blob(state->mem_ctx,
-                               reply);
-               *response = talloc_asprintf(mem_ctx, "TT %s", reply_base64);
-               TALLOC_FREE(reply_base64);
-               data_blob_free(&reply);
-               state->svr_state = SERVER_CHALLENGE;
-               DEBUG(10, ("NTLMSSP challenge\n"));
-       } else if (NT_STATUS_EQUAL(nt_status, NT_STATUS_ACCESS_DENIED)) {
-               *response = talloc_asprintf(mem_ctx, "BH %s",
-                                        nt_errstr(nt_status));
-               DEBUG(0, ("NTLMSSP BH: %s\n", nt_errstr(nt_status)));
-
-               TALLOC_FREE(state->gensec_security);
-       } else if (!NT_STATUS_IS_OK(nt_status)) {
-               *response = talloc_asprintf(mem_ctx, "NA %s",
-                                        nt_errstr(nt_status));
-               DEBUG(10, ("NTLMSSP %s\n", nt_errstr(nt_status)));
-       } else {
-               struct auth_session_info *session_info;
-               nt_status = gensec_session_info(state->gensec_security, state->gensec_security, &session_info);
-               if (!NT_STATUS_IS_OK(nt_status)) {
-                       DEBUG(0, ("NTLMSSP BH: %s\n", nt_errstr(nt_status)));
-                       TALLOC_FREE(state->gensec_security);
-                       return;
-               }
-               *response = talloc_asprintf(
-                       mem_ctx, "AF %s",
-                       session_info->unix_info->unix_name);
-               DEBUG(10, ("NTLMSSP OK!\n"));
-
-               state->have_session_key = true;
-               state->svr_state = SERVER_FINISHED;
-       }
-
-       data_blob_free(&request);
-}
-
-static void manage_squid_ntlmssp_request(struct ntlm_auth_state *state,
-                                        char *buf, int length)
-{
-       char *response;
-
-       manage_squid_ntlmssp_request_int(state, buf, length,
-                                        talloc_tos(), &response);
-
-       if (response == NULL) {
-               x_fprintf(x_stdout, "BH Out of memory\n");
-               return;
-       }
-       x_fprintf(x_stdout, "%s\n", response);
-       TALLOC_FREE(response);
-}
-
-static void manage_client_ntlmssp_request(struct ntlm_auth_state *state,
-                                               char *buf, int length)
-{
-       DATA_BLOB request, reply;
-       NTSTATUS nt_status;
-
-       if (!opt_username || !*opt_username) {
-               x_fprintf(x_stderr, "username must be specified!\n\n");
-               exit(1);
-       }
-
-       if (strlen(buf) < 2) {
-               DEBUG(1, ("NTLMSSP query [%s] invalid\n", buf));
-               x_fprintf(x_stdout, "BH NTLMSSP query invalid\n");
-               return;
-       }
+       if (strlen(buf) < 2) {
+               DEBUG(1, ("query [%s] invalid", buf));
+               printf("BH Query invalid\n");
+               talloc_free(mem_ctx);
+               return;
+       }
 
        if (strlen(buf) > 3) {
                if(strncmp(buf, "SF ", 3) == 0) {
-                       DEBUG(10, ("Looking for flags to negotiate\n"));
-                       talloc_free(state->want_feature_list);
-                       state->want_feature_list = talloc_strdup(state->mem_ctx,
-                                       buf+3);
-                       x_fprintf(x_stdout, "OK\n");
+                       DEBUG(10, ("Setting flags to negotiate\n"));
+                       talloc_free(want_feature_list);
+                       want_feature_list = talloc_strndup(state, buf+3, strlen(buf)-3);
+                       printf("OK\n");
+                       talloc_free(mem_ctx);
                        return;
                }
-               request = base64_decode_data_blob(buf + 3);
+               in = base64_decode_data_blob_talloc(mem_ctx, buf + 3);
        } else {
-               request = data_blob_null;
-       }
-
-       if (strncmp(buf, "PW ", 3) == 0) {
-               /* We asked for a password and obviously got it :-) */
-
-               opt_password = SMB_STRNDUP((const char *)request.data,
-                               request.length);
-
-               if (opt_password == NULL) {
-                       DEBUG(1, ("Out of memory\n"));
-                       x_fprintf(x_stdout, "BH Out of memory\n");
-                       data_blob_free(&request);
-                       return;
-               }
-
-               x_fprintf(x_stdout, "OK\n");
-               data_blob_free(&request);
-               return;
-       }
-
-       if (!state->ntlmssp_state && use_cached_creds) {
-               /* check whether cached credentials are usable. */
-               DATA_BLOB empty_blob = data_blob_null;
-
-               nt_status = do_ccache_ntlm_auth(empty_blob, empty_blob, NULL);
-               if (!NT_STATUS_EQUAL(nt_status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
-                       /* failed to use cached creds */
-                       use_cached_creds = False;
-               }
-       }
-
-       if (opt_password == NULL && !use_cached_creds) {
-               /* Request a password from the calling process.  After
-                  sending it, the calling process should retry asking for the
-                  negotiate. */
-
-               DEBUG(10, ("Requesting password\n"));
-               x_fprintf(x_stdout, "PW\n");
-               return;
+               in = data_blob(NULL, 0);
        }
 
        if (strncmp(buf, "YR", 2) == 0) {
-               TALLOC_FREE(state->ntlmssp_state);
-               state->cli_state = CLIENT_INITIAL;
-       } else if (strncmp(buf, "TT", 2) == 0) {
-               /* No special preprocessing required */
-       } else if (strncmp(buf, "GF", 2) == 0) {
-               DEBUG(10, ("Requested negotiated NTLMSSP flags\n"));
-
-               if(state->cli_state == CLIENT_FINISHED) {
-                       x_fprintf(x_stdout, "GF 0x%08x\n", state->neg_flags);
-               }
-               else {
-                       x_fprintf(x_stdout, "BH\n");
-               }
-
-               data_blob_free(&request);
-               return;
-       } else if (strncmp(buf, "GK", 2) == 0 ) {
-               DEBUG(10, ("Requested session key\n"));
-
-               if(state->cli_state == CLIENT_FINISHED) {
-                       char *key64 = base64_encode_data_blob(state->mem_ctx,
-                                       state->session_key);
-                       x_fprintf(x_stdout, "GK %s\n", key64?key64:"<NULL>");
-                       TALLOC_FREE(key64);
-               }
-               else {
-                       x_fprintf(x_stdout, "BH\n");
-               }
-
-               data_blob_free(&request);
+               if (state->gensec_state) {
+                       talloc_free(state->gensec_state);
+                       state->gensec_state = NULL;
+               }
+       } else if ( (strncmp(buf, "OK", 2) == 0)) {
+               /* Just return BH, like ntlm_auth from Samba 3 does. */
+               printf("BH Command expected\n");
+               talloc_free(mem_ctx);
                return;
-       } else {
-               DEBUG(1, ("NTLMSSP query [%s] invalid\n", buf));
-               x_fprintf(x_stdout, "BH NTLMSSP query invalid\n");
+       } else if ( (strncmp(buf, "TT ", 3) != 0) &&
+                   (strncmp(buf, "KK ", 3) != 0) &&
+                   (strncmp(buf, "AF ", 3) != 0) &&
+                   (strncmp(buf, "NA ", 3) != 0) &&
+                   (strncmp(buf, "UG", 2) != 0) &&
+                   (strncmp(buf, "PW ", 3) != 0) &&
+                   (strncmp(buf, "GK", 2) != 0) &&
+                   (strncmp(buf, "GF", 2) != 0)) {
+               DEBUG(1, ("SPNEGO request [%s] invalid prefix\n", buf));
+               printf("BH SPNEGO request invalid prefix\n");
+               talloc_free(mem_ctx);
                return;
        }
 
-       if (!state->ntlmssp_state) {
-               nt_status = ntlm_auth_start_ntlmssp_client(
-                               &state->ntlmssp_state);
-               if (!NT_STATUS_IS_OK(nt_status)) {
-                       x_fprintf(x_stdout, "BH %s\n", nt_errstr(nt_status));
-                       return;
-               }
-               ntlmssp_want_feature_list(state->ntlmssp_state,
-                               state->want_feature_list);
-               state->initial_message = data_blob_null;
-       }
-
-       DEBUG(10, ("got NTLMSSP packet:\n"));
-       dump_data(10, request.data, request.length);
-
-       if (use_cached_creds && !opt_password &&
-                       (state->cli_state == CLIENT_RESPONSE)) {
-               nt_status = do_ccache_ntlm_auth(state->initial_message, request,
-                               &reply);
-       } else {
-               nt_status = ntlmssp_update(state->ntlmssp_state, request,
-                               &reply);
-       }
-
-       if (NT_STATUS_EQUAL(nt_status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
-               char *reply_base64 = base64_encode_data_blob(state->mem_ctx,
-                               reply);
-               if (state->cli_state == CLIENT_INITIAL) {
-                       x_fprintf(x_stdout, "YR %s\n", reply_base64);
-                       state->initial_message = reply;
-                       state->cli_state = CLIENT_RESPONSE;
-               } else {
-                       x_fprintf(x_stdout, "KK %s\n", reply_base64);
-                       data_blob_free(&reply);
-               }
-               TALLOC_FREE(reply_base64);
-               DEBUG(10, ("NTLMSSP challenge\n"));
-       } else if (NT_STATUS_IS_OK(nt_status)) {
-               char *reply_base64 = base64_encode_data_blob(talloc_tos(),
-                               reply);
-               x_fprintf(x_stdout, "AF %s\n", reply_base64);
-               TALLOC_FREE(reply_base64);
-
-               if(state->have_session_key)
-                       data_blob_free(&state->session_key);
-
-               state->session_key = data_blob(
-                               state->ntlmssp_state->session_key.data,
-                               state->ntlmssp_state->session_key.length);
-               state->neg_flags = state->ntlmssp_state->neg_flags;
-               state->have_session_key = true;
-
-               DEBUG(10, ("NTLMSSP OK!\n"));
-               state->cli_state = CLIENT_FINISHED;
-               TALLOC_FREE(state->ntlmssp_state);
-       } else {
-               x_fprintf(x_stdout, "BH %s\n", nt_errstr(nt_status));
-               DEBUG(0, ("NTLMSSP BH: %s\n", nt_errstr(nt_status)));
-               state->cli_state = CLIENT_ERROR;
-               TALLOC_FREE(state->ntlmssp_state);
-       }
-
-       data_blob_free(&request);
-}
-
-static void manage_squid_basic_request(struct ntlm_auth_state *state,
-                                       char *buf, int length)
-{
-       char *user, *pass;      
-       user=buf;
-
-       pass=(char *)memchr(buf,' ',length);
-       if (!pass) {
-               DEBUG(2, ("Password not found. Denying access\n"));
-               x_fprintf(x_stdout, "ERR\n");
-               return;
-       }
-       *pass='\0';
-       pass++;
-
-       if (state->helper_mode == SQUID_2_5_BASIC) {
-               rfc1738_unescape(user);
-               rfc1738_unescape(pass);
-       }
-
-       if (check_plaintext_auth(user, pass, False)) {
-               x_fprintf(x_stdout, "OK\n");
-       } else {
-               x_fprintf(x_stdout, "ERR\n");
-       }
-}
-
-static void offer_gss_spnego_mechs(void) {
-
-       DATA_BLOB token;
-       struct spnego_data spnego;
-       ssize_t len;
-       char *reply_base64;
-       TALLOC_CTX *ctx = talloc_tos();
-       char *principal;
-       char *myname_lower;
-
-       ZERO_STRUCT(spnego);
-
-       myname_lower = talloc_strdup(ctx, lp_netbios_name());
-       if (!myname_lower) {
-               return;
-       }
-       strlower_m(myname_lower);
-
-       principal = talloc_asprintf(ctx, "%s$@%s", myname_lower, lp_realm());
-       if (!principal) {
-               return;
-       }
-
-       /* Server negTokenInit (mech offerings) */
-       spnego.type = SPNEGO_NEG_TOKEN_INIT;
-       spnego.negTokenInit.mechTypes = talloc_array(ctx, const char *, 4);
-#ifdef HAVE_KRB5
-       spnego.negTokenInit.mechTypes[0] = talloc_strdup(ctx, OID_KERBEROS5_OLD);
-       spnego.negTokenInit.mechTypes[1] = talloc_strdup(ctx, OID_KERBEROS5);
-       spnego.negTokenInit.mechTypes[2] = talloc_strdup(ctx, OID_NTLMSSP);
-       spnego.negTokenInit.mechTypes[3] = NULL;
-#else
-       spnego.negTokenInit.mechTypes[0] = talloc_strdup(ctx, OID_NTLMSSP);
-       spnego.negTokenInit.mechTypes[1] = NULL;
-#endif
-
-
-       spnego.negTokenInit.mechListMIC = data_blob_talloc(ctx, principal,
-                                                   strlen(principal));
-
-       len = spnego_write_data(ctx, &token, &spnego);
-       spnego_free_data(&spnego);
-
-       if (len == -1) {
-               DEBUG(1, ("Could not write SPNEGO data blob\n"));
-               x_fprintf(x_stdout, "BH Could not write SPNEGO data blob\n");
-               return;
-       }
-
-       reply_base64 = base64_encode_data_blob(talloc_tos(), token);
-       x_fprintf(x_stdout, "TT %s *\n", reply_base64);
-
-       TALLOC_FREE(reply_base64);
-       data_blob_free(&token);
-       DEBUG(10, ("sent SPNEGO negTokenInit\n"));
-       return;
-}
-
-static void manage_gss_spnego_request(struct ntlm_auth_state *state,
-                                       char *buf, int length)
-{
-       struct spnego_data request, response;
-       DATA_BLOB token;
-       DATA_BLOB raw_in_token = data_blob_null;
-       DATA_BLOB raw_out_token = data_blob_null;
-       NTSTATUS status;
-       ssize_t len;
-       TALLOC_CTX *ctx = talloc_tos();
-
-       const char *unix_username = NULL;
-
-       const char *reply_code;
-       char       *reply_base64;
-       char *reply_argument = NULL;
-       char *supportedMech = NULL;
-
-       if (strlen(buf) < 2) {
-               DEBUG(1, ("SPENGO query [%s] invalid\n", buf));
-               x_fprintf(x_stdout, "BH SPENGO query invalid\n");
-               return;
-       }
-
-       if (strncmp(buf, "YR", 2) == 0) {
-               TALLOC_FREE(state->gensec_security);
-               TALLOC_FREE(state->spnego_mech);
-               TALLOC_FREE(state->spnego_mech_oid);
-       } else if (strncmp(buf, "KK", 2) == 0) {
-               ;
-       } else {
-               DEBUG(1, ("SPENGO query [%s] invalid\n", buf));
-               x_fprintf(x_stdout, "BH SPENGO query invalid\n");
-               return;
-       }
-
-       if ( (strlen(buf) == 2)) {
-
-               /* no client data, get the negTokenInit offering
-                   mechanisms */
-
-               offer_gss_spnego_mechs();
-               return;
-       }
-
-       /* All subsequent requests have a blob. This might be negTokenInit or negTokenTarg */
-
-       if (strlen(buf) <= 3) {
-               DEBUG(1, ("GSS-SPNEGO query [%s] invalid\n", buf));
-               x_fprintf(x_stdout, "BH GSS-SPNEGO query invalid\n");
-               return;
-       }
-
-       token = base64_decode_data_blob(buf + 3);
-
-       if ((token.length >= 7)
-           && (strncmp((char *)token.data, "NTLMSSP", 7) == 0)) {
-               char *reply;
-
-               data_blob_free(&token);
-
-               DEBUG(10, ("Could not parse GSS-SPNEGO, trying raw "
-                          "ntlmssp\n"));
-
-               manage_squid_ntlmssp_request_int(state, buf, length,
-                                                talloc_tos(), &reply);
-               if (reply == NULL) {
-                       x_fprintf(x_stdout, "BH Out of memory\n");
-                       return;
-               }
-
-               if (strncmp(reply, "AF ", 3) == 0) {
-                       x_fprintf(x_stdout, "AF * %s\n", reply+3);
-               } else {
-                       x_fprintf(x_stdout, "%s *\n", reply);
-               }
-
-               TALLOC_FREE(reply);
-               return;
-       }
-
-       ZERO_STRUCT(request);
-       len = spnego_read_data(ctx, token, &request);
-       data_blob_free(&token);
-
-       if (len == -1) {
-               DEBUG(1, ("GSS-SPNEGO query [%s] invalid\n", buf));
-               x_fprintf(x_stdout, "BH GSS-SPNEGO query invalid\n");
-               return;
-       }
-
-       if (request.type == SPNEGO_NEG_TOKEN_INIT) {
-#ifdef HAVE_KRB5
-               int krb5_idx = -1;
-#endif
-               int ntlm_idx = -1;
-               int used_idx = -1;
-               int i;
-
-               if (state->spnego_mech) {
-                       DEBUG(1, ("Client restarted SPNEGO with NegTokenInit "
-                                 "while mech[%s] was already negotiated\n",
-                                 state->spnego_mech));
-                       x_fprintf(x_stdout, "BH Client send NegTokenInit twice\n");
-                       return;
-               }
-
-               /* Second request from Client. This is where the
-                  client offers its mechanism to use. */
-
-               if ( (request.negTokenInit.mechTypes == NULL) ||
-                    (request.negTokenInit.mechTypes[0] == NULL) ) {
-                       DEBUG(1, ("Client did not offer any mechanism\n"));
-                       x_fprintf(x_stdout, "BH Client did not offer any "
-                                           "mechanism\n");
-                       return;
-               }
-
-               status = NT_STATUS_UNSUCCESSFUL;
-               for (i = 0; request.negTokenInit.mechTypes[i] != NULL; i++) {
-                       DEBUG(10,("got mech[%d][%s]\n",
-                               i, request.negTokenInit.mechTypes[i]));
-#ifdef HAVE_KRB5
-                       if (strcmp(request.negTokenInit.mechTypes[i], OID_KERBEROS5_OLD) == 0) {
-                               krb5_idx = i;
-                               break;
+       /* setup gensec */
+       if (!(state->gensec_state)) {
+               switch (stdio_helper_mode) {
+               case GSS_SPNEGO_CLIENT:
+                       /*
+                        * cached credentials are only supported by
+                        * NTLMSSP_CLIENT_1 for now.
+                        */
+                       use_cached_creds = false;
+                       /* fall through */
+               case NTLMSSP_CLIENT_1:
+                       /* setup the client side */
+
+                       if (state->set_password != NULL) {
+                               use_cached_creds = false;
                        }
-                       if (strcmp(request.negTokenInit.mechTypes[i], OID_KERBEROS5) == 0) {
-                               krb5_idx = i;
-                               break;
-                       }
-#endif
-                       if (strcmp(request.negTokenInit.mechTypes[i], OID_NTLMSSP) == 0) {
-                               ntlm_idx = i;
-                               break;
-                       }
-               }
 
-               used_idx = ntlm_idx;
-#ifdef HAVE_KRB5
-               if (krb5_idx != -1) {
-                       ntlm_idx = -1;
-                       used_idx = krb5_idx;
-               }
-#endif
-               if (ntlm_idx > -1) {
-                       state->spnego_mech = talloc_strdup(state, "ntlmssp");
-                       if (state->spnego_mech == NULL) {
-                               x_fprintf(x_stdout, "BH Out of memory\n");
-                               return;
+                       if (use_cached_creds) {
+                               struct wbcCredentialCacheParams params;
+                               struct wbcCredentialCacheInfo *info = NULL;
+                               struct wbcAuthErrorInfo *error = NULL;
+                               wbcErr wbc_status;
+
+                               params.account_name = opt_username;
+                               params.domain_name = opt_domain;
+                               params.level = WBC_CREDENTIAL_CACHE_LEVEL_NTLMSSP;
+                               params.num_blobs = 0;
+                               params.blobs = NULL;
+
+                               wbc_status = wbcCredentialCache(&params, &info,
+                                                               &error);
+                               wbcFreeMemory(error);
+                               if (!WBC_ERROR_IS_OK(wbc_status)) {
+                                       use_cached_creds = false;
+                               }
+                               wbcFreeMemory(info);
                        }
 
-                       if (state->gensec_security) {
-                               DEBUG(1, ("Client wants a new NTLMSSP challenge, but "
-                                         "already got one\n"));
-                               x_fprintf(x_stdout, "BH Client wants a new "
-                                                   "NTLMSSP challenge, but "
-                                                   "already got one\n");
-                               TALLOC_FREE(state->gensec_security);
+                       nt_status = ntlm_auth_prepare_gensec_client(state, lp_ctx,
+                                                                   &state->gensec_state);
+                       if (!NT_STATUS_IS_OK(nt_status)) {
+                               printf("BH GENSEC mech failed to start: %s\n",
+                                      nt_errstr(nt_status));
+                               talloc_free(mem_ctx);
                                return;
                        }
 
-                       status = ntlm_auth_start_ntlmssp_server(state, &state->gensec_security);
-                       if (!NT_STATUS_IS_OK(status)) {
-                               x_fprintf(x_stdout, "BH %s\n", nt_errstr(status));
-                               return;
+                       creds = cli_credentials_init(state->gensec_state);
+                       cli_credentials_set_conf(creds, lp_ctx);
+                       if (opt_username) {
+                               cli_credentials_set_username(creds, opt_username, CRED_SPECIFIED);
                        }
-               }
-
-#ifdef HAVE_KRB5
-               if (krb5_idx > -1) {
-                       state->spnego_mech = talloc_strdup(state, "krb5");
-                       if (state->spnego_mech == NULL) {
-                               x_fprintf(x_stdout, "BH Out of memory\n");
-                               return;
+                       if (opt_domain) {
+                               cli_credentials_set_domain(creds, opt_domain, CRED_SPECIFIED);
                        }
-               }
-#endif
-               if (used_idx > -1) {
-                       state->spnego_mech_oid = talloc_strdup(state,
-                               request.negTokenInit.mechTypes[used_idx]);
-                       if (state->spnego_mech_oid == NULL) {
-                               x_fprintf(x_stdout, "BH Out of memory\n");
-                               return;
+                       if (use_cached_creds) {
+                               gensec_want_feature(state->gensec_state,
+                                                   GENSEC_FEATURE_NTLM_CCACHE);
+                       } else if (state->set_password) {
+                               cli_credentials_set_password(creds, state->set_password, CRED_SPECIFIED);
+                       } else {
+                               cli_credentials_set_password_callback(creds, get_password);
                        }
-                       supportedMech = talloc_strdup(ctx, state->spnego_mech_oid);
-                       if (supportedMech == NULL) {
-                               x_fprintf(x_stdout, "BH Out of memory\n");
-                               return;
+                       if (opt_workstation) {
+                               cli_credentials_set_workstation(creds, opt_workstation, CRED_SPECIFIED);
                        }
 
-                       status = NT_STATUS_MORE_PROCESSING_REQUIRED;
-               } else {
-                       status = NT_STATUS_NOT_SUPPORTED;
-               }
-               if (used_idx == 0) {
-                       status = NT_STATUS_OK;
-                       raw_in_token = request.negTokenInit.mechToken;
-               }
-       } else {
-               if (state->spnego_mech == NULL) {
-                       DEBUG(1,("Got netTokenTarg without negTokenInit\n"));
-                       x_fprintf(x_stdout, "BH Got a negTokenTarg without "
-                                           "negTokenInit\n");
-                       return;
-               }
-
-               if ((request.negTokenTarg.supportedMech != NULL) &&
-                    (strcmp(request.negTokenTarg.supportedMech, state->spnego_mech_oid) != 0 ) ) {
-                       DEBUG(1, ("Got a negTokenTarg with mech[%s] while [%s] was already negotiated\n",
-                                 request.negTokenTarg.supportedMech,
-                                 state->spnego_mech_oid));
-                       x_fprintf(x_stdout, "BH Got a negTokenTarg with speficied mech\n");
-                       return;
-               }
-
-               status = NT_STATUS_OK;
-               raw_in_token = request.negTokenTarg.responseToken;
-       }
+                       gensec_set_credentials(state->gensec_state, creds);
 
-       if (!NT_STATUS_IS_OK(status)) {
-               /* error or more processing */
-       } else if (strcmp(state->spnego_mech, "ntlmssp") == 0) {
-
-               DEBUG(10, ("got NTLMSSP packet:\n"));
-               dump_data(10, raw_in_token.data, raw_in_token.length);
-
-               status = gensec_update(state->gensec_security,
-                                      talloc_tos(), NULL,
-                                      raw_in_token,
-                                      &raw_out_token);
-               if (NT_STATUS_IS_OK(status)) {
-                       struct auth_session_info *session_info;
-                       status = gensec_session_info(state->gensec_security, state->gensec_security, &session_info);
-                       if (!NT_STATUS_IS_OK(status)) {
-                               DEBUG(0, ("NTLMSSP BH: %s\n", nt_errstr(status)));
-                               TALLOC_FREE(state->gensec_security);
+                       break;
+               case GSS_SPNEGO_SERVER:
+               case SQUID_2_5_NTLMSSP:
+               {
+                       nt_status = ntlm_auth_prepare_gensec_server(state, lp_ctx,
+                                                                   &state->gensec_state);
+                       if (!NT_STATUS_IS_OK(nt_status)) {
+                               printf("BH GENSEC mech failed to start: %s\n",
+                                      nt_errstr(nt_status));
+                               talloc_free(mem_ctx);
                                return;
                        }
-                       unix_username = session_info->unix_info->unix_name;
-                       DEBUG(10, ("NTLMSSP OK!\n"));
+                       break;
                }
-               if (!NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
-                       TALLOC_FREE(state->ntlmssp_state);
+               default:
+                       talloc_free(mem_ctx);
+                       abort();
                }
-#ifdef HAVE_KRB5
-       } else if (strcmp(state->spnego_mech, "krb5") == 0) {
-               char *principal;
-               DATA_BLOB ap_rep;
-               DATA_BLOB session_key;
-               struct PAC_LOGON_INFO *logon_info = NULL;
-               DATA_BLOB ticket;
-               uint8_t tok_id[2];
-
-               if (!spnego_parse_krb5_wrap(ctx, raw_in_token,
-                                           &ticket, tok_id)) {
-                       DEBUG(1, ("spnego_parse_krb5_wrap failed\n"));
-                       x_fprintf(x_stdout, "BH spnego_parse_krb5_wrap failed\n");
-                       return;
-               }
-
-               status = ads_verify_ticket(ctx, lp_realm(), 0,
-                                          &ticket,
-                                          &principal, &logon_info, &ap_rep,
-                                          &session_key, True);
 
-               /* Now in "principal" we have the name we are authenticated as. */
+               gensec_want_feature_list(state->gensec_state, want_feature_list);
 
-               if (NT_STATUS_IS_OK(status)) {
-                       char *domain;
-                       char *user;
-                       domain = strchr_m(principal, '@');
+               /* Session info is not complete, do not pass to auth log */
+               gensec_want_feature(state->gensec_state, GENSEC_FEATURE_NO_AUTHZ_LOG);
 
-                       if (domain == NULL) {
-                               DEBUG(1, ("Did not get a valid principal "
-                                         "from ads_verify_ticket\n"));
-                               x_fprintf(x_stdout, "BH Did not get a "
-                                         "valid principal from "
-                                         "ads_verify_ticket\n");
-                               return;
+               switch (stdio_helper_mode) {
+               case GSS_SPNEGO_CLIENT:
+               case GSS_SPNEGO_SERVER:
+                       nt_status = gensec_start_mech_by_oid(state->gensec_state, GENSEC_OID_SPNEGO);
+                       if (!in.length) {
+                               first = true;
                        }
-
-                       *domain++ = '\0';
-                       domain = talloc_strdup(ctx, domain);
-                       user = talloc_strdup(ctx, principal);
-
-                       if (logon_info) {
-                               netsamlogon_cache_store(
-                                       user, &logon_info->info3);
+                       break;
+               case NTLMSSP_CLIENT_1:
+                       if (!in.length) {
+                               first = true;
                        }
-
-                       data_blob_free(&ap_rep);
-                       data_blob_free(&session_key);
-                       unix_username = talloc_asprintf(ctx, "%s\\%s", domain, user);
+                       /* fall through */
+               case SQUID_2_5_NTLMSSP:
+                       nt_status = gensec_start_mech_by_oid(state->gensec_state, GENSEC_OID_NTLMSSP);
+                       break;
+               default:
+                       talloc_free(mem_ctx);
+                       abort();
                }
-               data_blob_free(&ticket);
-#endif
-       }
-
-       spnego_free_data(&request);
-       ZERO_STRUCT(response);
-       response.type = SPNEGO_NEG_TOKEN_TARG;
 
-       if (NT_STATUS_IS_OK(status)) {
-               TALLOC_FREE(state->spnego_mech);
-               TALLOC_FREE(state->spnego_mech_oid);
-               response.negTokenTarg.negResult = SPNEGO_ACCEPT_COMPLETED;
-               response.negTokenTarg.responseToken = raw_out_token;
-               reply_code = "AF";
-               reply_argument = unix_username;
-       } else if (NT_STATUS_EQUAL(status,
-                                  NT_STATUS_MORE_PROCESSING_REQUIRED)) {
-               response.negTokenTarg.supportedMech = supportedMech;
-               response.negTokenTarg.responseToken = raw_out_token;
-               response.negTokenTarg.negResult = SPNEGO_ACCEPT_INCOMPLETE;
-               reply_code = "TT";
-               reply_argument = talloc_strdup(ctx, "*");
-       } else {
-               TALLOC_FREE(state->spnego_mech);
-               TALLOC_FREE(state->spnego_mech_oid);
-               data_blob_free(&raw_out_token);
-               response.negTokenTarg.negResult = SPNEGO_REJECT;
-               reply_code = "NA";
-               reply_argument = talloc_strdup(ctx, nt_errstr(status));
-       }
+               if (!NT_STATUS_IS_OK(nt_status)) {
+                       DEBUG(1, ("GENSEC mech failed to start: %s\n", nt_errstr(nt_status)));
+                       printf("BH GENSEC mech failed to start\n");
+                       talloc_free(mem_ctx);
+                       return;
+               }
 
-       if (!reply_argument) {
-               DEBUG(1, ("Could not write SPNEGO data blob\n"));
-               x_fprintf(x_stdout, "BH Could not write SPNEGO data blob\n");
-               spnego_free_data(&response);
-               return;
        }
 
-       len = spnego_write_data(ctx, &token, &response);
-       spnego_free_data(&response);
+       /* update */
 
-       if (len == -1) {
-               DEBUG(1, ("Could not write SPNEGO data blob\n"));
-               x_fprintf(x_stdout, "BH Could not write SPNEGO data blob\n");
+       if (strncmp(buf, "PW ", 3) == 0) {
+               state->set_password = talloc_strndup(state,
+                                                    (const char *)in.data,
+                                                    in.length);
+
+               cli_credentials_set_password(gensec_get_credentials(state->gensec_state),
+                                            state->set_password,
+                                            CRED_SPECIFIED);
+               printf("OK\n");
+               talloc_free(mem_ctx);
                return;
        }
 
-       reply_base64 = base64_encode_data_blob(talloc_tos(), token);
-
-       x_fprintf(x_stdout, "%s %s %s\n",
-                 reply_code, reply_base64, reply_argument);
-
-       TALLOC_FREE(reply_base64);
-       data_blob_free(&token);
-
-       return;
-}
-
-static struct ntlmssp_state *client_ntlmssp_state = NULL;
-
-static bool manage_client_ntlmssp_init(struct spnego_data spnego)
-{
-       NTSTATUS status;
-       DATA_BLOB null_blob = data_blob_null;
-       DATA_BLOB to_server;
-       char *to_server_base64;
-       const char *my_mechs[] = {OID_NTLMSSP, NULL};
-       TALLOC_CTX *ctx = talloc_tos();
-
-       DEBUG(10, ("Got spnego negTokenInit with NTLMSSP\n"));
-
-       if (client_ntlmssp_state != NULL) {
-               DEBUG(1, ("Request for initial SPNEGO request where "
-                         "we already have a state\n"));
-               return False;
-       }
-
-       if (!client_ntlmssp_state) {
-               if (!NT_STATUS_IS_OK(status = ntlm_auth_start_ntlmssp_client(&client_ntlmssp_state))) {
-                       x_fprintf(x_stdout, "BH %s\n", nt_errstr(status));
-                       return False;
+       if (strncmp(buf, "GK", 2) == 0) {
+               char *base64_key;
+               DEBUG(10, ("Requested session key\n"));
+               nt_status = gensec_session_key(state->gensec_state, mem_ctx, &session_key);
+               if(!NT_STATUS_IS_OK(nt_status)) {
+                       DEBUG(1, ("gensec_session_key failed: %s\n", nt_errstr(nt_status)));
+                       printf("BH No session key\n");
+                       talloc_free(mem_ctx);
+                       return;
+               } else {
+                       base64_key = base64_encode_data_blob(state, session_key);
+                       SMB_ASSERT(base64_key != NULL);
+                       printf("GK %s\n", base64_key);
+                       talloc_free(base64_key);
                }
-       }
-
-
-       if (opt_password == NULL) {
-
-               /* Request a password from the calling process.  After
-                  sending it, the calling process should retry with
-                  the negTokenInit. */
-
-               DEBUG(10, ("Requesting password\n"));
-               x_fprintf(x_stdout, "PW\n");
-               return True;
-       }
-
-       spnego.type = SPNEGO_NEG_TOKEN_INIT;
-       spnego.negTokenInit.mechTypes = my_mechs;
-       spnego.negTokenInit.reqFlags = data_blob_null;
-       spnego.negTokenInit.reqFlagsPadding = 0;
-       spnego.negTokenInit.mechListMIC = null_blob;
-
-       status = ntlmssp_update(client_ntlmssp_state, null_blob,
-                                      &spnego.negTokenInit.mechToken);
-
-       if ( !(NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED) ||
-                       NT_STATUS_IS_OK(status)) ) {
-               DEBUG(1, ("Expected OK or MORE_PROCESSING_REQUIRED, got: %s\n",
-                         nt_errstr(status)));
-               TALLOC_FREE(client_ntlmssp_state);
-               return False;
-       }
-
-       spnego_write_data(ctx, &to_server, &spnego);
-       data_blob_free(&spnego.negTokenInit.mechToken);
-
-       to_server_base64 = base64_encode_data_blob(talloc_tos(), to_server);
-       data_blob_free(&to_server);
-       x_fprintf(x_stdout, "KK %s\n", to_server_base64);
-       TALLOC_FREE(to_server_base64);
-       return True;
-}
-
-static void manage_client_ntlmssp_targ(struct spnego_data spnego)
-{
-       NTSTATUS status;
-       DATA_BLOB null_blob = data_blob_null;
-       DATA_BLOB request;
-       DATA_BLOB to_server;
-       char *to_server_base64;
-       TALLOC_CTX *ctx = talloc_tos();
-
-       DEBUG(10, ("Got spnego negTokenTarg with NTLMSSP\n"));
-
-       if (client_ntlmssp_state == NULL) {
-               DEBUG(1, ("Got NTLMSSP tArg without a client state\n"));
-               x_fprintf(x_stdout, "BH Got NTLMSSP tArg without a client state\n");
+               talloc_free(mem_ctx);
                return;
        }
 
-       if (spnego.negTokenTarg.negResult == SPNEGO_REJECT) {
-               x_fprintf(x_stdout, "NA\n");
-               TALLOC_FREE(client_ntlmssp_state);
-               return;
-       }
+       if (strncmp(buf, "GF", 2) == 0) {
+               uint32_t neg_flags;
 
-       if (spnego.negTokenTarg.negResult == SPNEGO_ACCEPT_COMPLETED) {
-               x_fprintf(x_stdout, "AF\n");
-               TALLOC_FREE(client_ntlmssp_state);
-               return;
-       }
+               DEBUG(10, ("Requested negotiated NTLMSSP feature flags\n"));
 
-       status = ntlmssp_update(client_ntlmssp_state,
-                                      spnego.negTokenTarg.responseToken,
-                                      &request);
+               neg_flags = gensec_ntlmssp_neg_flags(state->gensec_state);
+               if (neg_flags == 0) {
+                       printf("BH\n");
+                       talloc_free(mem_ctx);
+                       return;
+               }
 
-       if (!NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED) && !NT_STATUS_IS_OK(status)) {
-               DEBUG(1, ("Expected MORE_PROCESSING_REQUIRED or OK from "
-                         "ntlmssp_client_update, got: %s\n",
-                         nt_errstr(status)));
-               x_fprintf(x_stdout, "BH Expected MORE_PROCESSING_REQUIRED from "
-                                   "ntlmssp_client_update\n");
-               data_blob_free(&request);
-               TALLOC_FREE(client_ntlmssp_state);
+               printf("GF 0x%08x\n", neg_flags);
+               talloc_free(mem_ctx);
                return;
        }
 
-       spnego.type = SPNEGO_NEG_TOKEN_TARG;
-       spnego.negTokenTarg.negResult = SPNEGO_ACCEPT_INCOMPLETE;
-       spnego.negTokenTarg.supportedMech = (const char *)OID_NTLMSSP;
-       spnego.negTokenTarg.responseToken = request;
-       spnego.negTokenTarg.mechListMIC = null_blob;
-
-       spnego_write_data(ctx, &to_server, &spnego);
-       data_blob_free(&request);
+       nt_status = gensec_update(state->gensec_state, mem_ctx, in, &out);
 
-       to_server_base64 = base64_encode_data_blob(talloc_tos(), to_server);
-       data_blob_free(&to_server);
-       x_fprintf(x_stdout, "KK %s\n", to_server_base64);
-       TALLOC_FREE(to_server_base64);
-       return;
-}
-
-#ifdef HAVE_KRB5
-
-static bool manage_client_krb5_init(struct spnego_data spnego)
-{
-       char *principal;
-       DATA_BLOB tkt, tkt_wrapped, to_server;
-       DATA_BLOB session_key_krb5 = data_blob_null;
-       struct spnego_data reply;
-       char *reply_base64;
-       int retval;
-
-       const char *my_mechs[] = {OID_KERBEROS5_OLD, NULL};
-       ssize_t len;
-       TALLOC_CTX *ctx = talloc_tos();
+       /* don't leak 'bad password'/'no such user' info to the network client */
+       nt_status = nt_status_squash(nt_status);
 
-       principal = spnego.negTokenInit.targetPrincipal;
-
-       /* We may not be allowed to use the server-supplied SPNEGO principal, or it may not have been supplied to us
-        */
-       if (!lp_client_use_spnego_principal() || strequal(principal, ADS_IGNORE_PRINCIPAL)) {
-               principal = NULL;
+       if (out.length) {
+               out_base64 = base64_encode_data_blob(mem_ctx, out);
+               SMB_ASSERT(out_base64 != NULL);
+       } else {
+               out_base64 = NULL;
        }
-       
-       if (principal == NULL &&
-           opt_target_service && opt_target_hostname && !is_ipaddress(opt_target_hostname)) {
-               DEBUG(3,("manage_client_krb5_init: using target "
-                        "hostname not SPNEGO principal\n"));
 
-               principal = kerberos_get_principal_from_service_hostname(talloc_tos(),
-                                                                        opt_target_service,
-                                                                        opt_target_hostname);
-
-               if (!principal) {
-                       return false;
+       if (NT_STATUS_EQUAL(nt_status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
+               reply_arg = "*";
+               if (first && state->gensec_state->gensec_role == GENSEC_CLIENT) {
+                       reply_code = "YR";
+               } else if (state->gensec_state->gensec_role == GENSEC_CLIENT) {
+                       reply_code = "KK";
+               } else if (state->gensec_state->gensec_role == GENSEC_SERVER) {
+                       reply_code = "TT";
+               } else {
+                       abort();
                }
-               
-               DEBUG(3,("manage_client_krb5_init: guessed "
-                        "server principal=%s\n",
-                        principal ? principal : "<null>"));
-       }
-       
-       if (principal == NULL) {
-               DEBUG(3,("manage_client_krb5_init: could not guess server principal\n"));
-               return false;
-       }
-
-       retval = cli_krb5_get_ticket(ctx, principal, 0,
-                                         &tkt, &session_key_krb5,
-                                         0, NULL, NULL, NULL);
-       if (retval) {
-               char *user = NULL;
 
-               /* Let's try to first get the TGT, for that we need a
-                   password. */
 
-               if (opt_password == NULL) {
-                       DEBUG(10, ("Requesting password\n"));
-                       x_fprintf(x_stdout, "PW\n");
-                       return True;
-               }
-
-               user = talloc_asprintf(talloc_tos(), "%s@%s", opt_username, opt_domain);
-               if (!user) {
-                       return false;
-               }
+       } else if (NT_STATUS_EQUAL(nt_status, NT_STATUS_ACCESS_DENIED)) {
+               reply_code = "BH NT_STATUS_ACCESS_DENIED";
+               reply_arg = nt_errstr(nt_status);
+               DEBUG(1, ("GENSEC login failed: %s\n", nt_errstr(nt_status)));
+       } else if (NT_STATUS_EQUAL(nt_status, NT_STATUS_UNSUCCESSFUL)) {
+               reply_code = "BH NT_STATUS_UNSUCCESSFUL";
+               reply_arg = nt_errstr(nt_status);
+               DEBUG(1, ("GENSEC login failed: %s\n", nt_errstr(nt_status)));
+       } else if (!NT_STATUS_IS_OK(nt_status)) {
+               reply_code = "NA";
+               reply_arg = nt_errstr(nt_status);
+               DEBUG(1, ("GENSEC login failed: %s\n", nt_errstr(nt_status)));
+       } else if /* OK */ (state->gensec_state->gensec_role == GENSEC_SERVER) {
+               struct auth_session_info *session_info;
 
-               if ((retval = kerberos_kinit_password(user, opt_password, 0, NULL))) {
-                       DEBUG(10, ("Requesting TGT failed: %s\n", error_message(retval)));
-                       return False;
-               }
+               nt_status = gensec_session_info(state->gensec_state, mem_ctx, &session_info);
+               if (!NT_STATUS_IS_OK(nt_status)) {
+                       reply_code = "BH Failed to retrive session info";
+                       reply_arg = nt_errstr(nt_status);
+                       DEBUG(1, ("GENSEC failed to retrieve the session info: %s\n", nt_errstr(nt_status)));
+               } else {
 
-               retval = cli_krb5_get_ticket(ctx, principal, 0,
-                                                 &tkt, &session_key_krb5,
-                                                 0, NULL, NULL, NULL);
-               if (retval) {
-                       DEBUG(10, ("Kinit suceeded, but getting a ticket failed: %s\n", error_message(retval)));
-                       return False;
+                       reply_code = "AF";
+                       reply_arg = talloc_strdup(state->gensec_state, session_info->unix_info->unix_name);
+                       if (reply_arg == NULL) {
+                               reply_code = "BH out of memory";
+                               reply_arg = nt_errstr(NT_STATUS_NO_MEMORY);
+                       }
+                       talloc_free(session_info);
                }
-
+       } else if (state->gensec_state->gensec_role == GENSEC_CLIENT) {
+               reply_code = "AF";
+               reply_arg = out_base64;
+       } else {
+               abort();
        }
 
-       /* wrap that up in a nice GSS-API wrapping */
-       tkt_wrapped = spnego_gen_krb5_wrap(ctx, tkt, TOK_ID_KRB_AP_REQ);
-
-       data_blob_free(&session_key_krb5);
-
-       ZERO_STRUCT(reply);
-
-       reply.type = SPNEGO_NEG_TOKEN_INIT;
-       reply.negTokenInit.mechTypes = my_mechs;
-       reply.negTokenInit.reqFlags = data_blob_null;
-       reply.negTokenInit.reqFlagsPadding = 0;
-       reply.negTokenInit.mechToken = tkt_wrapped;
-       reply.negTokenInit.mechListMIC = data_blob_null;
-
-       len = spnego_write_data(ctx, &to_server, &reply);
-       data_blob_free(&tkt);
-
-       if (len == -1) {
-               DEBUG(1, ("Could not write SPNEGO data blob\n"));
-               return False;
+       switch (stdio_helper_mode) {
+       case GSS_SPNEGO_SERVER:
+               printf("%s %s %s\n", reply_code,
+                      out_base64 ? out_base64 : "*",
+                      reply_arg ? reply_arg : "*");
+               break;
+       default:
+               if (out_base64) {
+                       printf("%s %s\n", reply_code, out_base64);
+               } else if (reply_arg) {
+                       printf("%s %s\n", reply_code, reply_arg);
+               } else {
+                       printf("%s\n", reply_code);
+               }
        }
 
-       reply_base64 = base64_encode_data_blob(talloc_tos(), to_server);
-       x_fprintf(x_stdout, "KK %s *\n", reply_base64);
-
-       TALLOC_FREE(reply_base64);
-       data_blob_free(&to_server);
-       DEBUG(10, ("sent GSS-SPNEGO KERBEROS5 negTokenInit\n"));
-       return True;
+       talloc_free(mem_ctx);
+       return;
 }
 
-static void manage_client_krb5_targ(struct spnego_data spnego)
+static void manage_gss_spnego_request(enum stdio_helper_mode stdio_helper_mode,
+                                  struct loadparm_context *lp_ctx,
+                                  struct ntlm_auth_state *state,
+                                     char *buf, int length, void **private2)
 {
-       switch (spnego.negTokenTarg.negResult) {
-       case SPNEGO_ACCEPT_INCOMPLETE:
-               DEBUG(1, ("Got a Kerberos negTokenTarg with ACCEPT_INCOMPLETE\n"));
-               x_fprintf(x_stdout, "BH Got a Kerberos negTokenTarg with "
-                                   "ACCEPT_INCOMPLETE\n");
-               break;
-       case SPNEGO_ACCEPT_COMPLETED:
-               DEBUG(10, ("Accept completed\n"));
-               x_fprintf(x_stdout, "AF\n");
-               break;
-       case SPNEGO_REJECT:
-               DEBUG(10, ("Rejected\n"));
-               x_fprintf(x_stdout, "NA\n");
-               break;
-       default:
-               DEBUG(1, ("Got an invalid negTokenTarg\n"));
-               x_fprintf(x_stdout, "AF\n");
-       }
+       manage_gensec_request(stdio_helper_mode, lp_ctx, buf, length, &state->gensec_private_1);
+       return;
 }
 
-#endif
-
-static void manage_gss_spnego_client_request(struct ntlm_auth_state *state,
-                                               char *buf, int length)
+static void manage_squid_ntlmssp_request(enum stdio_helper_mode stdio_helper_mode,
+                                  struct loadparm_context *lp_ctx,
+                                  struct ntlm_auth_state *state,
+                                        char *buf, int length, void **private2)
 {
-       DATA_BLOB request;
-       struct spnego_data spnego;
-       ssize_t len;
-       TALLOC_CTX *ctx = talloc_tos();
-
-       if (!opt_username || !*opt_username) {
-               x_fprintf(x_stderr, "username must be specified!\n\n");
-               exit(1);
-       }
-
-       if (strlen(buf) <= 3) {
-               DEBUG(1, ("SPNEGO query [%s] too short\n", buf));
-               x_fprintf(x_stdout, "BH SPNEGO query too short\n");
-               return;
-       }
-
-       request = base64_decode_data_blob(buf+3);
-
-       if (strncmp(buf, "PW ", 3) == 0) {
-
-               /* We asked for a password and obviously got it :-) */
-
-               opt_password = SMB_STRNDUP((const char *)request.data, request.length);
-
-               if (opt_password == NULL) {
-                       DEBUG(1, ("Out of memory\n"));
-                       x_fprintf(x_stdout, "BH Out of memory\n");
-                       data_blob_free(&request);
-                       return;
-               }
-
-               x_fprintf(x_stdout, "OK\n");
-               data_blob_free(&request);
-               return;
-       }
-
-       if ( (strncmp(buf, "TT ", 3) != 0) &&
-            (strncmp(buf, "AF ", 3) != 0) &&
-            (strncmp(buf, "NA ", 3) != 0) ) {
-               DEBUG(1, ("SPNEGO request [%s] invalid\n", buf));
-               x_fprintf(x_stdout, "BH SPNEGO request invalid\n");
-               data_blob_free(&request);
-               return;
-       }
-
-       /* So we got a server challenge to generate a SPNEGO
-           client-to-server request... */
-
-       len = spnego_read_data(ctx, request, &spnego);
-       data_blob_free(&request);
-
-       if (len == -1) {
-               DEBUG(1, ("Could not read SPNEGO data for [%s]\n", buf));
-               x_fprintf(x_stdout, "BH Could not read SPNEGO data\n");
-               return;
-       }
-
-       if (spnego.type == SPNEGO_NEG_TOKEN_INIT) {
-
-               /* The server offers a list of mechanisms */
-
-               const char **mechType = (const char **)spnego.negTokenInit.mechTypes;
-
-               while (*mechType != NULL) {
-
-#ifdef HAVE_KRB5
-                       if ( (strcmp(*mechType, OID_KERBEROS5_OLD) == 0) ||
-                            (strcmp(*mechType, OID_KERBEROS5) == 0) ) {
-                               if (manage_client_krb5_init(spnego))
-                                       goto out;
-                       }
-#endif
-
-                       if (strcmp(*mechType, OID_NTLMSSP) == 0) {
-                               if (manage_client_ntlmssp_init(spnego))
-                                       goto out;
-                       }
-
-                       mechType++;
-               }
-
-               DEBUG(1, ("Server offered no compatible mechanism\n"));
-               x_fprintf(x_stdout, "BH Server offered no compatible mechanism\n");
-               return;
-       }
-
-       if (spnego.type == SPNEGO_NEG_TOKEN_TARG) {
-
-               if (spnego.negTokenTarg.supportedMech == NULL) {
-                       /* On accept/reject Windows does not send the
-                           mechanism anymore. Handle that here and
-                           shut down the mechanisms. */
-
-                       switch (spnego.negTokenTarg.negResult) {
-                       case SPNEGO_ACCEPT_COMPLETED:
-                               x_fprintf(x_stdout, "AF\n");
-                               break;
-                       case SPNEGO_REJECT:
-                               x_fprintf(x_stdout, "NA\n");
-                               break;
-                       default:
-                               DEBUG(1, ("Got a negTokenTarg with no mech and an "
-                                         "unknown negResult: %d\n",
-                                         spnego.negTokenTarg.negResult));
-                               x_fprintf(x_stdout, "BH Got a negTokenTarg with"
-                                                   " no mech and an unknown "
-                                                   "negResult\n");
-                       }
-
-                       TALLOC_FREE(client_ntlmssp_state);
-                       goto out;
-               }
-
-               if (strcmp(spnego.negTokenTarg.supportedMech,
-                          OID_NTLMSSP) == 0) {
-                       manage_client_ntlmssp_targ(spnego);
-                       goto out;
-               }
-
-#if HAVE_KRB5
-               if (strcmp(spnego.negTokenTarg.supportedMech,
-                          OID_KERBEROS5_OLD) == 0) {
-                       manage_client_krb5_targ(spnego);
-                       goto out;
-               }
-#endif
-
-       }
-
-       DEBUG(1, ("Got an SPNEGO token I could not handle [%s]!\n", buf));
-       x_fprintf(x_stdout, "BH Got an SPNEGO token I could not handle\n");
+       manage_gensec_request(stdio_helper_mode, lp_ctx, buf, length, &state->gensec_private_1);
        return;
+}
 
- out:
-       spnego_free_data(&spnego);
+static void manage_gss_spnego_client_request(enum stdio_helper_mode stdio_helper_mode,
+                                  struct loadparm_context *lp_ctx,
+                                  struct ntlm_auth_state *state,
+                                            char *buf, int length, void **private2)
+{
+       manage_gensec_request(stdio_helper_mode, lp_ctx, buf, length, &state->gensec_private_1);
        return;
 }
 
-static void manage_ntlm_server_1_request(struct ntlm_auth_state *state,
-                                               char *buf, int length)
+static void manage_ntlm_server_1_request(enum stdio_helper_mode stdio_helper_mode,
+                                  struct loadparm_context *lp_ctx,
+                                  struct ntlm_auth_state *state,
+                                               char *buf, int length, void **private2)
 {
        char *request, *parameter;      
        static DATA_BLOB challenge;
@@ -2228,37 +1669,39 @@ static void manage_ntlm_server_1_request(struct ntlm_auth_state *state,
 
        if (strequal(buf, ".")) {
                if (!full_username && !username) {      
-                       x_fprintf(x_stdout, "Error: No username supplied!\n");
+                       printf("Error: No username supplied!\n");
                } else if (plaintext_password) {
                        /* handle this request as plaintext */
                        if (!full_username) {
                                if (asprintf(&full_username, "%s%c%s", domain, winbind_separator(), username) == -1) {
-                                       x_fprintf(x_stdout, "Error: Out of memory in asprintf!\n.\n");
+                                       printf("Error: Out of memory in "
+                                              "asprintf!\n.\n");
                                        return;
                                }
                        }
                        if (check_plaintext_auth(full_username, plaintext_password, False)) {
-                               x_fprintf(x_stdout, "Authenticated: Yes\n");
+                               printf("Authenticated: Yes\n");
                        } else {
-                               x_fprintf(x_stdout, "Authenticated: No\n");
+                               printf("Authenticated: No\n");
                        }
                } else if (!lm_response.data && !nt_response.data) {
-                       x_fprintf(x_stdout, "Error: No password supplied!\n");
+                       printf("Error: No password supplied!\n");
                } else if (!challenge.data) {   
-                       x_fprintf(x_stdout, "Error: No lanman-challenge supplied!\n");
+                       printf("Error: No lanman-challenge supplied!\n");
                } else {
                        char *error_string = NULL;
                        uchar lm_key[8];
                        uchar user_session_key[16];
-                       uint32 flags = 0;
-
+                       uint32_t flags = 0;
+                       NTSTATUS nt_status;
                        if (full_username && !username) {
                                fstring fstr_user;
                                fstring fstr_domain;
 
                                if (!parse_ntlm_auth_domain_user(full_username, fstr_user, fstr_domain)) {
                                        /* username might be 'tainted', don't print into our new-line deleimianted stream */
-                                       x_fprintf(x_stdout, "Error: Could not parse into domain and username\n");
+                                       printf("Error: Could not parse into "
+                                              "domain and username\n");
                                }
                                SAFE_FREE(username);
                                SAFE_FREE(domain);
@@ -2266,55 +1709,98 @@ static void manage_ntlm_server_1_request(struct ntlm_auth_state *state,
                                domain = smb_xstrdup(fstr_domain);
                        }
 
-                       if (!domain) {
-                               domain = smb_xstrdup(get_winbind_domain());
+                       if (opt_password) {
+                               DATA_BLOB nt_session_key, lm_session_key;
+                               struct samr_Password lm_pw, nt_pw;
+                               TALLOC_CTX *mem_ctx = talloc_new(NULL);
+                               ZERO_STRUCT(user_session_key);
+                               ZERO_STRUCT(lm_key);
+
+                               nt_lm_owf_gen (opt_password, nt_pw.hash, lm_pw.hash);
+                               nt_status = ntlm_password_check(mem_ctx,
+                                                               true, true, 0,
+                                                               &challenge,
+                                                               &lm_response,
+                                                               &nt_response,
+                                                               username,
+                                                               username,
+                                                               domain,
+                                                               &lm_pw, &nt_pw,
+                                                               &nt_session_key,
+                                                               &lm_session_key);
+                               error_string = smb_xstrdup(get_friendly_nt_error_msg(nt_status));
+                               if (ntlm_server_1_user_session_key) {
+                                       if (nt_session_key.length == sizeof(user_session_key)) {
+                                               memcpy(user_session_key,
+                                                      nt_session_key.data,
+                                                      sizeof(user_session_key));
+                                       }
+                               }
+                               if (ntlm_server_1_lm_session_key) {
+                                       if (lm_session_key.length == sizeof(lm_key)) {
+                                               memcpy(lm_key,
+                                                      lm_session_key.data,
+                                                      sizeof(lm_key));
+                                       }
+                               }
+                               TALLOC_FREE(mem_ctx);
+
+                       } else {
+                               uint8_t authoritative = 0;
+
+                               if (!domain) {
+                                       domain = smb_xstrdup(get_winbind_domain());
+                               }
+
+                               if (ntlm_server_1_lm_session_key)
+                                       flags |= WBFLAG_PAM_LMKEY;
+
+                               if (ntlm_server_1_user_session_key)
+                                       flags |= WBFLAG_PAM_USER_SESSION_KEY;
+
+                               nt_status = contact_winbind_auth_crap(username,
+                                                                     domain,
+                                                                     lp_netbios_name(),
+                                                                     &challenge,
+                                                                     &lm_response,
+                                                                     &nt_response,
+                                                                     flags, 0,
+                                                                     lm_key,
+                                                                     user_session_key,
+                                                                     &authoritative,
+                                                                     &error_string,
+                                                                     NULL);
                        }
 
-                       if (ntlm_server_1_lm_session_key) 
-                               flags |= WBFLAG_PAM_LMKEY;
-
-                       if (ntlm_server_1_user_session_key) 
-                               flags |= WBFLAG_PAM_USER_SESSION_KEY;
-
-                       if (!NT_STATUS_IS_OK(
-                                   contact_winbind_auth_crap(username, 
-                                                             domain, 
-                                                             lp_netbios_name(),
-                                                             &challenge, 
-                                                             &lm_response, 
-                                                             &nt_response, 
-                                                             flags, 0,
-                                                             lm_key, 
-                                                             user_session_key,
-                                                             &error_string,
-                                                             NULL))) {
-
-                               x_fprintf(x_stdout, "Authenticated: No\n");
-                               x_fprintf(x_stdout, "Authentication-Error: %s\n.\n", error_string);
+                       if (!NT_STATUS_IS_OK(nt_status)) {
+                               printf("Authenticated: No\n");
+                               printf("Authentication-Error: %s\n.\n",
+                                      error_string);
                        } else {
-                               static char zeros[16];
                                char *hex_lm_key;
                                char *hex_user_session_key;
 
-                               x_fprintf(x_stdout, "Authenticated: Yes\n");
+                               printf("Authenticated: Yes\n");
 
                                if (ntlm_server_1_lm_session_key 
-                                   && (memcmp(zeros, lm_key, 
-                                              sizeof(lm_key)) != 0)) {
+                                   && (!all_zero(lm_key,
+                                                 sizeof(lm_key)))) {
                                        hex_lm_key = hex_encode_talloc(NULL,
                                                                (const unsigned char *)lm_key,
                                                                sizeof(lm_key));
-                                       x_fprintf(x_stdout, "LANMAN-Session-Key: %s\n", hex_lm_key);
+                                       printf("LANMAN-Session-Key: %s\n",
+                                              hex_lm_key);
                                        TALLOC_FREE(hex_lm_key);
                                }
 
                                if (ntlm_server_1_user_session_key 
-                                   && (memcmp(zeros, user_session_key, 
-                                              sizeof(user_session_key)) != 0)) {
+                                   && (!all_zero(user_session_key,
+                                                 sizeof(user_session_key)))) {
                                        hex_user_session_key = hex_encode_talloc(NULL,
                                                                          (const unsigned char *)user_session_key, 
                                                                          sizeof(user_session_key));
-                                       x_fprintf(x_stdout, "User-Session-Key: %s\n", hex_user_session_key);
+                                       printf("User-Session-Key: %s\n",
+                                              hex_user_session_key);
                                        TALLOC_FREE(hex_user_session_key);
                                }
                        }
@@ -2330,7 +1816,7 @@ static void manage_ntlm_server_1_request(struct ntlm_auth_state *state,
                SAFE_FREE(plaintext_password);
                ntlm_server_1_user_session_key = False;
                ntlm_server_1_lm_session_key = False;
-               x_fprintf(x_stdout, ".\n");
+               printf(".\n");
 
                return;
        }
@@ -2344,7 +1830,7 @@ static void manage_ntlm_server_1_request(struct ntlm_auth_state *state,
 
                if (!parameter) {
                        DEBUG(0, ("Parameter not found!\n"));
-                       x_fprintf(x_stdout, "Error: Parameter not found!\n.\n");
+                       printf("Error: Parameter not found!\n.\n");
                        return;
                }
 
@@ -2367,25 +1853,28 @@ static void manage_ntlm_server_1_request(struct ntlm_auth_state *state,
        if (strequal(request, "LANMAN-Challenge")) {
                challenge = strhex_to_data_blob(NULL, parameter);
                if (challenge.length != 8) {
-                       x_fprintf(x_stdout, "Error: hex decode of %s failed! (got %d bytes, expected 8)\n.\n", 
-                                 parameter,
-                                 (int)challenge.length);
+                       printf("Error: hex decode of %s failed! "
+                              "(got %d bytes, expected 8)\n.\n",
+                              parameter,
+                              (int)challenge.length);
                        challenge = data_blob_null;
                }
        } else if (strequal(request, "NT-Response")) {
                nt_response = strhex_to_data_blob(NULL, parameter);
                if (nt_response.length < 24) {
-                       x_fprintf(x_stdout, "Error: hex decode of %s failed! (only got %d bytes, needed at least 24)\n.\n", 
-                                 parameter,
-                                 (int)nt_response.length);
+                       printf("Error: hex decode of %s failed! "
+                              "(only got %d bytes, needed at least 24)\n.\n",
+                              parameter,
+                              (int)nt_response.length);
                        nt_response = data_blob_null;
                }
        } else if (strequal(request, "LANMAN-Response")) {
                lm_response = strhex_to_data_blob(NULL, parameter);
                if (lm_response.length != 24) {
-                       x_fprintf(x_stdout, "Error: hex decode of %s failed! (got %d bytes, expected 24)\n.\n", 
-                                 parameter,
-                                 (int)lm_response.length);
+                       printf("Error: hex decode of %s failed! "
+                              "(got %d bytes, expected 24)\n.\n",
+                              parameter,
+                              (int)lm_response.length);
                        lm_response = data_blob_null;
                }
        } else if (strequal(request, "Password")) {
@@ -2401,12 +1890,14 @@ static void manage_ntlm_server_1_request(struct ntlm_auth_state *state,
        } else if (strequal(request, "Request-LanMan-Session-Key")) {
                ntlm_server_1_lm_session_key = strequal(parameter, "Yes");
        } else {
-               x_fprintf(x_stdout, "Error: Unknown request %s\n.\n", request);
+               printf("Error: Unknown request %s\n.\n", request);
        }
 }
 
-static void manage_ntlm_change_password_1_request(struct ntlm_auth_state *state,
-                                                       char *buf, int length)
+static void manage_ntlm_change_password_1_request(enum stdio_helper_mode stdio_helper_mode,
+                                                 struct loadparm_context *lp_ctx,
+                                                 struct ntlm_auth_state *state,
+                                                 char *buf, int length, void **private2)
 {
        char *request, *parameter;      
        static DATA_BLOB new_nt_pswd;
@@ -2472,11 +1963,11 @@ static void manage_ntlm_change_password_1_request(struct ntlm_auth_state *state,
                }
 
                if (!full_username && !username) {      
-                       x_fprintf(x_stdout, "Error: No username supplied!\n");
+                       printf("Error: No username supplied!\n");
                } else if ((!new_nt_pswd.data || !old_nt_hash_enc.data) &&
                           (!new_lm_pswd.data || old_lm_hash_enc.data) ) {
-                       x_fprintf(x_stdout, "Error: No NT or LM password "
-                                 "blobs supplied!\n");
+                       printf("Error: No NT or LM password "
+                              "blobs supplied!\n");
                } else {
                        char *error_string = NULL;
 
@@ -2490,9 +1981,9 @@ static void manage_ntlm_change_password_1_request(struct ntlm_auth_state *state,
                                        /* username might be 'tainted', don't
                                         * print into our new-line
                                         * deleimianted stream */
-                                       x_fprintf(x_stdout, "Error: Could not "
-                                                 "parse into domain and "
-                                                 "username\n");
+                                       printf("Error: Could not "
+                                              "parse into domain and "
+                                              "username\n");
                                        SAFE_FREE(username);
                                        username = smb_xstrdup(full_username);
                                } else {
@@ -2511,11 +2002,11 @@ static void manage_ntlm_change_password_1_request(struct ntlm_auth_state *state,
                                                    new_lm_pswd,
                                                    old_lm_hash_enc,
                                                    &error_string))) {
-                               x_fprintf(x_stdout, "Password-Change: No\n");
-                               x_fprintf(x_stdout, "Password-Change-Error: "
-                                         "%s\n.\n", error_string);
+                               printf("Password-Change: No\n");
+                               printf("Password-Change-Error: %s\n.\n",
+                                      error_string);
                        } else {
-                               x_fprintf(x_stdout, "Password-Change: Yes\n");
+                               printf("Password-Change: Yes\n");
                        }
 
                        SAFE_FREE(error_string);
@@ -2530,7 +2021,7 @@ static void manage_ntlm_change_password_1_request(struct ntlm_auth_state *state,
                SAFE_FREE(domain);
                SAFE_FREE(newpswd);
                SAFE_FREE(oldpswd);
-               x_fprintf(x_stdout, ".\n");
+               printf(".\n");
 
                return;
        }
@@ -2544,7 +2035,7 @@ static void manage_ntlm_change_password_1_request(struct ntlm_auth_state *state,
 
                if (!parameter) {
                        DEBUG(0, ("Parameter not found!\n"));
-                       x_fprintf(x_stdout, "Error: Parameter not found!\n.\n");
+                       printf("Error: Parameter not found!\n.\n");
                        return;
                }
 
@@ -2566,28 +2057,28 @@ static void manage_ntlm_change_password_1_request(struct ntlm_auth_state *state,
        if (strequal(request, "new-nt-password-blob")) {
                new_nt_pswd = strhex_to_data_blob(NULL, parameter);
                if (new_nt_pswd.length != 516) {
-                       x_fprintf(x_stdout, "Error: hex decode of %s failed! "
-                                 "(got %d bytes, expected 516)\n.\n", 
-                                 parameter,
-                                 (int)new_nt_pswd.length);
+                       printf("Error: hex decode of %s failed! "
+                              "(got %d bytes, expected 516)\n.\n",
+                              parameter,
+                              (int)new_nt_pswd.length);
                        new_nt_pswd = data_blob_null;
                }
        } else if (strequal(request, "old-nt-hash-blob")) {
                old_nt_hash_enc = strhex_to_data_blob(NULL, parameter);
                if (old_nt_hash_enc.length != 16) {
-                       x_fprintf(x_stdout, "Error: hex decode of %s failed! "
-                                 "(got %d bytes, expected 16)\n.\n", 
-                                 parameter,
-                                 (int)old_nt_hash_enc.length);
+                       printf("Error: hex decode of %s failed! "
+                              "(got %d bytes, expected 16)\n.\n",
+                              parameter,
+                              (int)old_nt_hash_enc.length);
                        old_nt_hash_enc = data_blob_null;
                }
        } else if (strequal(request, "new-lm-password-blob")) {
                new_lm_pswd = strhex_to_data_blob(NULL, parameter);
                if (new_lm_pswd.length != 516) {
-                       x_fprintf(x_stdout, "Error: hex decode of %s failed! "
-                                 "(got %d bytes, expected 516)\n.\n", 
-                                 parameter,
-                                 (int)new_lm_pswd.length);
+                       printf("Error: hex decode of %s failed! "
+                              "(got %d bytes, expected 516)\n.\n",
+                              parameter,
+                              (int)new_lm_pswd.length);
                        new_lm_pswd = data_blob_null;
                }
        }
@@ -2595,10 +2086,10 @@ static void manage_ntlm_change_password_1_request(struct ntlm_auth_state *state,
                old_lm_hash_enc = strhex_to_data_blob(NULL, parameter);
                if (old_lm_hash_enc.length != 16)
                {
-                       x_fprintf(x_stdout, "Error: hex decode of %s failed! "
-                                 "(got %d bytes, expected 16)\n.\n", 
-                                 parameter,
-                                 (int)old_lm_hash_enc.length);
+                       printf("Error: hex decode of %s failed! "
+                              "(got %d bytes, expected 16)\n.\n",
+                              parameter,
+                              (int)old_lm_hash_enc.length);
                        old_lm_hash_enc = data_blob_null;
                }
        } else if (strequal(request, "nt-domain")) {
@@ -2612,12 +2103,14 @@ static void manage_ntlm_change_password_1_request(struct ntlm_auth_state *state,
        } else if (strequal(request, "old-password")) {
                oldpswd = smb_xstrdup(parameter);
        } else {
-               x_fprintf(x_stdout, "Error: Unknown request %s\n.\n", request);
+               printf("Error: Unknown request %s\n.\n", request);
        }
 }
 
-static void manage_squid_request(struct ntlm_auth_state *state,
-               stdio_helper_function fn)
+static void manage_squid_request(enum stdio_helper_mode stdio_helper_mode,
+                                  struct loadparm_context *lp_ctx,
+                                  struct ntlm_auth_state *state,
+               stdio_helper_function fn, void **private2)
 {
        char *buf;
        char tmp[INITIAL_BUFFER_SIZE+1];
@@ -2627,7 +2120,7 @@ static void manage_squid_request(struct ntlm_auth_state *state,
        buf = talloc_strdup(state->mem_ctx, "");
        if (!buf) {
                DEBUG(0, ("Failed to allocate input buffer.\n"));
-               x_fprintf(x_stderr, "ERR\n");
+               fprintf(stderr, "ERR\n");
                exit(1);
        }
 
@@ -2651,7 +2144,7 @@ static void manage_squid_request(struct ntlm_auth_state *state,
 
                if (buf_size > MAX_BUFFER_SIZE) {
                        DEBUG(2, ("Oversized message\n"));
-                       x_fprintf(x_stderr, "ERR\n");
+                       fprintf(stderr, "ERR\n");
                        talloc_free(buf);
                        return;
                }
@@ -2666,35 +2159,37 @@ static void manage_squid_request(struct ntlm_auth_state *state,
 
        if (buf[0] == '\0') {
                DEBUG(2, ("Invalid Request\n"));
-               x_fprintf(x_stderr, "ERR\n");
+               fprintf(stderr, "ERR\n");
                talloc_free(buf);
                return;
        }
 
-       fn(state, buf, length);
+       fn(stdio_helper_mode, lp_ctx, state, buf, length, private2);
        talloc_free(buf);
 }
 
 
-static void squid_stream(enum stdio_helper_mode stdio_mode, stdio_helper_function fn) {
+static void squid_stream(enum stdio_helper_mode stdio_mode,
+                        struct loadparm_context *lp_ctx,
+                        stdio_helper_function fn) {
        TALLOC_CTX *mem_ctx;
        struct ntlm_auth_state *state;
 
        /* initialize FDescs */
-       x_setbuf(x_stdout, NULL);
-       x_setbuf(x_stderr, NULL);
+       setbuf(stdout, NULL);
+       setbuf(stderr, NULL);
 
        mem_ctx = talloc_init("ntlm_auth");
        if (!mem_ctx) {
                DEBUG(0, ("squid_stream: Failed to create talloc context\n"));
-               x_fprintf(x_stderr, "ERR\n");
+               fprintf(stderr, "ERR\n");
                exit(1);
        }
 
        state = talloc_zero(mem_ctx, struct ntlm_auth_state);
        if (!state) {
                DEBUG(0, ("squid_stream: Failed to talloc ntlm_auth_state\n"));
-               x_fprintf(x_stderr, "ERR\n");
+               fprintf(stderr, "ERR\n");
                exit(1);
        }
 
@@ -2703,7 +2198,7 @@ static void squid_stream(enum stdio_helper_mode stdio_mode, stdio_helper_functio
 
        while(1) {
                TALLOC_CTX *frame = talloc_stackframe();
-               manage_squid_request(state, fn);
+               manage_squid_request(stdio_mode, lp_ctx, state, fn, NULL);
                TALLOC_FREE(frame);
        }
 }
@@ -2714,15 +2209,15 @@ static void squid_stream(enum stdio_helper_mode stdio_mode, stdio_helper_functio
 static bool check_auth_crap(void)
 {
        NTSTATUS nt_status;
-       uint32 flags = 0;
+       uint32_t flags = 0;
        char lm_key[8];
        char user_session_key[16];
        char *hex_lm_key;
        char *hex_user_session_key;
        char *error_string;
-       static uint8 zeros[16];
+       uint8_t authoritative = 0;
 
-       x_setbuf(x_stdout, NULL);
+       setbuf(stdout, NULL);
 
        if (request_lm_key) 
                flags |= WBFLAG_PAM_LMKEY;
@@ -2740,30 +2235,29 @@ static bool check_auth_crap(void)
                                              flags, 0,
                                              (unsigned char *)lm_key, 
                                              (unsigned char *)user_session_key, 
+                                             &authoritative,
                                              &error_string, NULL);
 
        if (!NT_STATUS_IS_OK(nt_status)) {
-               x_fprintf(x_stdout, "%s (0x%x)\n", 
-                         error_string,
-                         NT_STATUS_V(nt_status));
+               printf("%s (0x%x)\n", error_string,
+                      NT_STATUS_V(nt_status));
                SAFE_FREE(error_string);
                return False;
        }
 
        if (request_lm_key 
-           && (memcmp(zeros, lm_key, 
-                      sizeof(lm_key)) != 0)) {
+           && (!all_zero((uint8_t *)lm_key, sizeof(lm_key)))) {
                hex_lm_key = hex_encode_talloc(talloc_tos(), (const unsigned char *)lm_key,
                                        sizeof(lm_key));
-               x_fprintf(x_stdout, "LM_KEY: %s\n", hex_lm_key);
+               printf("LM_KEY: %s\n", hex_lm_key);
                TALLOC_FREE(hex_lm_key);
        }
        if (request_user_session_key 
-           && (memcmp(zeros, user_session_key, 
-                      sizeof(user_session_key)) != 0)) {
+           && (!all_zero((uint8_t *)user_session_key,
+                         sizeof(user_session_key)))) {
                hex_user_session_key = hex_encode_talloc(talloc_tos(), (const unsigned char *)user_session_key, 
                                                  sizeof(user_session_key));
-               x_fprintf(x_stdout, "NT_KEY: %s\n", hex_user_session_key);
+               printf("NT_KEY: %s\n", hex_user_session_key);
                TALLOC_FREE(hex_user_session_key);
        }
 
@@ -2786,22 +2280,24 @@ enum {
        OPT_DIAGNOSTICS,
        OPT_REQUIRE_MEMBERSHIP,
        OPT_USE_CACHED_CREDS,
+       OPT_ALLOW_MSCHAPV2,
        OPT_PAM_WINBIND_CONF,
        OPT_TARGET_SERVICE,
-       OPT_TARGET_HOSTNAME
+       OPT_TARGET_HOSTNAME,
+       OPT_OFFLINE_LOGON
 };
 
  int main(int argc, const char **argv)
 {
        TALLOC_CTX *frame = talloc_stackframe();
        int opt;
-       static const char *helper_protocol;
-       static int diagnostics;
-
-       static const char *hex_challenge;
-       static const char *hex_lm_response;
-       static const char *hex_nt_response;
+       const char *helper_protocol = NULL;
+       int diagnostics = 0;
 
+       const char *hex_challenge = NULL;
+       const char *hex_lm_response = NULL;
+       const char *hex_nt_response = NULL;
+       struct loadparm_context *lp_ctx;
        poptContext pc;
 
        /* NOTE: DO NOT change this interface without considering the implications!
@@ -2825,6 +2321,10 @@ enum {
                { "request-lm-key", 0, POPT_ARG_NONE, &request_lm_key, OPT_LM_KEY, "Retrieve LM session key"},
                { "request-nt-key", 0, POPT_ARG_NONE, &request_user_session_key, OPT_USER_SESSION_KEY, "Retrieve User (NT) session key"},
                { "use-cached-creds", 0, POPT_ARG_NONE, &use_cached_creds, OPT_USE_CACHED_CREDS, "Use cached credentials if no password is given"},
+               { "allow-mschapv2", 0, POPT_ARG_NONE, &opt_allow_mschapv2, OPT_ALLOW_MSCHAPV2, "Explicitly allow MSCHAPv2" },
+               { "offline-logon", 0, POPT_ARG_NONE, &offline_logon,
+                 OPT_OFFLINE_LOGON,
+                 "Use cached passwords when DC is offline"},
                { "diagnostics", 0, POPT_ARG_NONE, &diagnostics,
                  OPT_DIAGNOSTICS,
                  "Perform diagnostics on the authentication chain"},
@@ -2834,13 +2334,15 @@ enum {
                { "target-hostname", 0, POPT_ARG_STRING, &opt_target_hostname, OPT_TARGET_HOSTNAME, "Target hostname" },
                POPT_COMMON_CONFIGFILE
                POPT_COMMON_VERSION
+               POPT_COMMON_OPTION
                POPT_TABLEEND
        };
 
        /* Samba client initialisation */
-       load_case_tables();
+       smb_init_locale();
 
        setup_logging("ntlm_auth", DEBUG_STDERR);
+       fault_setup();
 
        /* Parse options */
 
@@ -2873,18 +2375,20 @@ enum {
                case OPT_CHALLENGE:
                        opt_challenge = strhex_to_data_blob(NULL, hex_challenge);
                        if (opt_challenge.length != 8) {
-                               x_fprintf(x_stderr, "hex decode of %s failed! (only got %d bytes)\n", 
-                                         hex_challenge,
-                                         (int)opt_challenge.length);
+                               fprintf(stderr, "hex decode of %s failed! "
+                                       "(only got %d bytes)\n",
+                                       hex_challenge,
+                                       (int)opt_challenge.length);
                                exit(1);
                        }
                        break;
                case OPT_LM: 
                        opt_lm_response = strhex_to_data_blob(NULL, hex_lm_response);
                        if (opt_lm_response.length != 24) {
-                               x_fprintf(x_stderr, "hex decode of %s failed! (only got %d bytes)\n", 
-                                         hex_lm_response,
-                                         (int)opt_lm_response.length);
+                               fprintf(stderr, "hex decode of %s failed! "
+                                       "(only got %d bytes)\n",
+                                       hex_lm_response,
+                                       (int)opt_lm_response.length);
                                exit(1);
                        }
                        break;
@@ -2892,9 +2396,10 @@ enum {
                case OPT_NT: 
                        opt_nt_response = strhex_to_data_blob(NULL, hex_nt_response);
                        if (opt_nt_response.length < 24) {
-                               x_fprintf(x_stderr, "hex decode of %s failed! (only got %d bytes)\n", 
-                                         hex_nt_response,
-                                         (int)opt_nt_response.length);
+                               fprintf(stderr, "hex decode of %s failed! "
+                                       "(only got %d bytes)\n",
+                                       hex_nt_response,
+                                       (int)opt_nt_response.length);
                                exit(1);
                        }
                        break;
@@ -2914,7 +2419,7 @@ enum {
                        opt_username = p+1;
                        *p = '\0';
                        if (opt_domain && !strequal(opt_domain, domain)) {
-                               x_fprintf(x_stderr, "Domain specified in username (%s) "
+                               fprintf(stderr, "Domain specified in username (%s) "
                                        "doesn't match specified domain (%s)!\n\n",
                                        domain, opt_domain);
                                poptPrintHelp(pc, stderr, 0);
@@ -2935,25 +2440,33 @@ enum {
                opt_workstation = "";
        }
 
+       lp_ctx = loadparm_init_s3(NULL, loadparm_s3_helpers());
+       if (lp_ctx == NULL) {
+               fprintf(stderr, "loadparm_init_s3() failed!\n");
+               exit(1);
+       }
+
        if (helper_protocol) {
                int i;
                for (i=0; i<NUM_HELPER_MODES; i++) {
                        if (strcmp(helper_protocol, stdio_helper_protocols[i].name) == 0) {
-                               squid_stream(stdio_helper_protocols[i].mode, stdio_helper_protocols[i].fn);
+                               squid_stream(stdio_helper_protocols[i].mode, lp_ctx, stdio_helper_protocols[i].fn);
                                exit(0);
                        }
                }
-               x_fprintf(x_stderr, "unknown helper protocol [%s]\n\nValid helper protools:\n\n", helper_protocol);
+               fprintf(stderr, "unknown helper protocol [%s]\n\n"
+                       "Valid helper protools:\n\n", helper_protocol);
 
                for (i=0; i<NUM_HELPER_MODES; i++) {
-                       x_fprintf(x_stderr, "%s\n", stdio_helper_protocols[i].name);
+                       fprintf(stderr, "%s\n",
+                               stdio_helper_protocols[i].name);
                }
 
                exit(1);
        }
 
        if (!opt_username || !*opt_username) {
-               x_fprintf(x_stderr, "username must be specified!\n\n");
+               fprintf(stderr, "username must be specified!\n\n");
                poptPrintHelp(pc, stderr, 0);
                exit(1);
        }
@@ -2966,7 +2479,13 @@ enum {
        } 
 
        if (!opt_password) {
-               opt_password = getpass("password: ");
+               char pwd[256] = {0};
+               int rc;
+
+               rc = samba_getpass("Password: ", pwd, sizeof(pwd), false, false);
+               if (rc == 0) {
+                       opt_password = SMB_STRDUP(pwd);
+               }
        }
 
        if (diagnostics) {