r26676: libcli: Fill in lp_workgroup() again, should fix my breakage of cifsdd tests...
[ira/wip.git] / source / libcli / smb_composite / sesssetup.c
index 1b7756a3f4927058d769795e92d466651ebc76c2..ce7bcc143ed44d745d96a18c4955757405e3b88c 100644 (file)
@@ -5,7 +5,7 @@
    
    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
-   the Free Software Foundation; either version 2 of the License, or
+   the Free Software Foundation; either version 3 of the License, or
    (at your option) any later version.
    
    This program is distributed in the hope that it will be useful,
@@ -14,8 +14,7 @@
    GNU General Public License for more details.
    
    You should have received a copy of the GNU General Public License
-   along with this program; if not, write to the Free Software
-   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+   along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 /*
   a composite API for making handling a generic async session setup
 #include "libcli/smb_composite/smb_composite.h"
 #include "libcli/auth/libcli_auth.h"
 #include "auth/auth.h"
+#include "auth/gensec/gensec.h"
+#include "auth/credentials/credentials.h"
 #include "version.h"
+#include "param/param.h"
 
 struct sesssetup_state {
        union smb_sesssetup setup;
@@ -36,6 +38,16 @@ struct sesssetup_state {
        struct smbcli_request *req;
 };
 
+static int sesssetup_state_destructor(struct sesssetup_state *state)
+{
+       if (state->req) {
+               talloc_free(state->req);
+               state->req = NULL;
+       }
+
+       return 0;
+}
+
 static NTSTATUS session_setup_old(struct composite_context *c,
                                  struct smbcli_session *session, 
                                  struct smb_composite_sesssetup *io,
@@ -65,7 +77,7 @@ static void set_user_session_key(struct smbcli_session *session,
 */
 static void request_handler(struct smbcli_request *req)
 {
-       struct composite_context *c = req->async.private;
+       struct composite_context *c = (struct composite_context *)req->async.private;
        struct sesssetup_state *state = talloc_get_type(c->private_data, struct sesssetup_state);
        struct smbcli_session *session = req->session;
        DATA_BLOB session_key = data_blob(NULL, 0);
@@ -73,6 +85,7 @@ static void request_handler(struct smbcli_request *req)
        NTSTATUS session_key_err, nt_status;
 
        c->status = smb_raw_sesssetup_recv(req, state, &state->setup);
+       state->req = NULL;
 
        switch (state->setup.old.level) {
        case RAW_SESSSETUP_OLD:
@@ -80,14 +93,15 @@ static void request_handler(struct smbcli_request *req)
                /* This doesn't work, as this only happens on old
                 * protocols, where this comparison won't match. */
                if (NT_STATUS_EQUAL(c->status, NT_STATUS_LOGON_FAILURE)) {
+                       /* we neet to reset the vuid for a new try */
+                       session->vuid = 0;
                        if (cli_credentials_wrong_password(state->io->in.credentials)) {
                                nt_status = session_setup_old(c, session, 
                                                              state->io, 
                                                              &state->req);
                                if (NT_STATUS_IS_OK(nt_status)) {
                                        c->status = nt_status;
-                                       state->req->async.fn = request_handler;
-                                       state->req->async.private = c;
+                                       composite_continue_smb(c, state->req, request_handler, c);
                                        return;
                                }
                        }
@@ -97,14 +111,15 @@ static void request_handler(struct smbcli_request *req)
        case RAW_SESSSETUP_NT1:
                state->io->out.vuid = state->setup.nt1.out.vuid;
                if (NT_STATUS_EQUAL(c->status, NT_STATUS_LOGON_FAILURE)) {
+                       /* we neet to reset the vuid for a new try */
+                       session->vuid = 0;
                        if (cli_credentials_wrong_password(state->io->in.credentials)) {
                                nt_status = session_setup_nt1(c, session, 
                                                              state->io, 
                                                              &state->req);
                                if (NT_STATUS_IS_OK(nt_status)) {
                                        c->status = nt_status;
-                                       state->req->async.fn = request_handler;
-                                       state->req->async.private = c;
+                                       composite_continue_smb(c, state->req, request_handler, c);
                                        return;
                                }
                        }
@@ -112,16 +127,17 @@ static void request_handler(struct smbcli_request *req)
                break;
 
        case RAW_SESSSETUP_SPNEGO:
-               session->vuid = state->io->out.vuid = state->setup.spnego.out.vuid;
+               state->io->out.vuid = state->setup.spnego.out.vuid;
                if (NT_STATUS_EQUAL(c->status, NT_STATUS_LOGON_FAILURE)) {
+                       /* we need to reset the vuid for a new try */
+                       session->vuid = 0;
                        if (cli_credentials_wrong_password(state->io->in.credentials)) {
                                nt_status = session_setup_spnego(c, session, 
                                                                      state->io, 
                                                                      &state->req);
                                if (NT_STATUS_IS_OK(nt_status)) {
                                        c->status = nt_status;
-                                       state->req->async.fn = request_handler;
-                                       state->req->async.private = c;
+                                       composite_continue_smb(c, state->req, request_handler, c);
                                        return;
                                }
                        }
@@ -150,7 +166,7 @@ static void request_handler(struct smbcli_request *req)
                } else {
                        state->setup.spnego.in.secblob = data_blob(NULL, 0);
                }
-                       
+
                /* we need to do another round of session setup. We keep going until both sides
                   are happy */
                session_key_err = gensec_session_key(session->gensec, &session_key);
@@ -160,11 +176,22 @@ static void request_handler(struct smbcli_request *req)
                }
 
                if (state->setup.spnego.in.secblob.length) {
+                       /* 
+                        * set the session->vuid value only for calling
+                        * smb_raw_sesssetup_send()
+                        */
+                       uint16_t vuid = session->vuid;
+                       session->vuid = state->io->out.vuid;
                        state->req = smb_raw_sesssetup_send(session, &state->setup);
-                       state->req->async.fn = request_handler;
-                       state->req->async.private = c;
+                       session->vuid = vuid;
+                       composite_continue_smb(c, state->req, request_handler, c);
                        return;
                }
+               break;
+
+       case RAW_SESSSETUP_SMB2:
+               c->status = NT_STATUS_INTERNAL_ERROR;
+               break;
        }
 
        /* enforce the local signing required flag */
@@ -176,14 +203,12 @@ static void request_handler(struct smbcli_request *req)
                }
        }
 
-       if (NT_STATUS_IS_OK(c->status)) {
-               c->state = COMPOSITE_STATE_DONE;
-       } else {
-               c->state = COMPOSITE_STATE_ERROR;
-       }
-       if (c->async.fn) {
-               c->async.fn(c);
+       if (!NT_STATUS_IS_OK(c->status)) {
+               composite_error(c, c->status);
+               return;
        }
+
+       composite_done(c);
 }
 
 
@@ -198,14 +223,14 @@ static NTSTATUS session_setup_nt1(struct composite_context *c,
        NTSTATUS nt_status;
        struct sesssetup_state *state = talloc_get_type(c->private_data, struct sesssetup_state);
        const char *password = cli_credentials_get_password(io->in.credentials);
-       DATA_BLOB names_blob = NTLMv2_generate_names_blob(state, session->transport->socket->hostname, lp_workgroup());
+       DATA_BLOB names_blob = NTLMv2_generate_names_blob(state, session->transport->socket->hostname, lp_workgroup(global_loadparm));
        DATA_BLOB session_key;
        int flags = CLI_CRED_NTLM_AUTH;
-       if (lp_client_lanman_auth()) {
+       if (lp_client_lanman_auth(global_loadparm)) {
                flags |= CLI_CRED_LANMAN_AUTH;
        }
 
-       if (lp_client_ntlmv2_auth()) {
+       if (lp_client_ntlmv2_auth(global_loadparm)) {
                flags |= CLI_CRED_NTLMv2_AUTH;
        }
 
@@ -238,7 +263,7 @@ static NTSTATUS session_setup_nt1(struct composite_context *c,
                set_user_session_key(session, &session_key);
                
                data_blob_free(&session_key);
-       } else if (lp_client_plaintext_auth()) {
+       } else if (lp_client_plaintext_auth(global_loadparm)) {
                state->setup.nt1.in.password1 = data_blob_talloc(state, password, strlen(password));
                state->setup.nt1.in.password2 = data_blob(NULL, 0);
        } else {
@@ -265,14 +290,14 @@ static NTSTATUS session_setup_old(struct composite_context *c,
        NTSTATUS nt_status;
        struct sesssetup_state *state = talloc_get_type(c->private_data, struct sesssetup_state);
        const char *password = cli_credentials_get_password(io->in.credentials);
-       DATA_BLOB names_blob = NTLMv2_generate_names_blob(state, session->transport->socket->hostname, lp_workgroup());
+       DATA_BLOB names_blob = NTLMv2_generate_names_blob(state, session->transport->socket->hostname, lp_workgroup(global_loadparm));
        DATA_BLOB session_key;
        int flags = 0;
-       if (lp_client_lanman_auth()) {
+       if (lp_client_lanman_auth(global_loadparm)) {
                flags |= CLI_CRED_LANMAN_AUTH;
        }
 
-       if (lp_client_ntlmv2_auth()) {
+       if (lp_client_ntlmv2_auth(global_loadparm)) {
                flags |= CLI_CRED_NTLMv2_AUTH;
        }
 
@@ -299,7 +324,7 @@ static NTSTATUS session_setup_old(struct composite_context *c,
                set_user_session_key(session, &session_key);
                
                data_blob_free(&session_key);
-       } else if (lp_client_plaintext_auth()) {
+       } else if (lp_client_plaintext_auth(global_loadparm)) {
                state->setup.old.in.password = data_blob_talloc(state, password, strlen(password));
        } else {
                /* could match windows client and return 'cannot logon from this workstation', but it just confuses everybody */
@@ -338,11 +363,10 @@ static NTSTATUS session_setup_spnego(struct composite_context *c,
        state->setup.spnego.in.lanman       = talloc_asprintf(state, "Samba %s", SAMBA_VERSION_STRING);
        state->setup.spnego.in.workgroup    = io->in.workgroup;
 
-       state->setup.spnego.out.vuid        = session->vuid;
-
        smbcli_temp_set_signing(session->transport);
 
-       status = gensec_client_start(session, &session->gensec, c->event_ctx);
+       status = gensec_client_start(session, &session->gensec, c->event_ctx,
+                                    global_loadparm);
        if (!NT_STATUS_IS_OK(status)) {
                DEBUG(1, ("Failed to start GENSEC client mode: %s\n", nt_errstr(status)));
                return status;
@@ -352,7 +376,7 @@ static NTSTATUS session_setup_spnego(struct composite_context *c,
 
        status = gensec_set_credentials(session->gensec, io->in.credentials);
        if (!NT_STATUS_IS_OK(status)) {
-               DEBUG(1, ("Failed to start set GENSEC client credentails: %s\n", 
+               DEBUG(1, ("Failed to start set GENSEC client credentials: %s\n", 
                          nt_errstr(status)));
                return status;
        }
@@ -395,7 +419,7 @@ static NTSTATUS session_setup_spnego(struct composite_context *c,
                }
        }
 
-       if (chosen_oid == GENSEC_OID_SPNEGO) {
+       if ((const void *)chosen_oid == (const void *)GENSEC_OID_SPNEGO) {
                status = gensec_update(session->gensec, state,
                                       session->transport->negotiate.secblob,
                                       &state->setup.spnego.in.secblob);
@@ -439,25 +463,21 @@ struct composite_context *smb_composite_sesssetup_send(struct smbcli_session *se
        struct sesssetup_state *state;
        NTSTATUS status;
 
-       c = talloc_zero(session, struct composite_context);
+       c = composite_create(session, session->transport->socket->event.ctx);
        if (c == NULL) return NULL;
 
-       state = talloc(c, struct sesssetup_state);
-       if (state == NULL) {
-               talloc_free(c);
-               return NULL;
-       }
+       state = talloc_zero(c, struct sesssetup_state);
+       if (composite_nomem(state, c)) return c;
+       c->private_data = state;
 
        state->io = io;
 
-       c->state = COMPOSITE_STATE_IN_PROGRESS;
-       c->private_data = state;
-       c->event_ctx = session->transport->socket->event.ctx;
+       talloc_set_destructor(state, sesssetup_state_destructor);
 
        /* no session setup at all in earliest protocol varients */
        if (session->transport->negotiate.protocol < PROTOCOL_LANMAN1) {
                ZERO_STRUCT(io->out);
-               c->state = COMPOSITE_STATE_DONE;
+               composite_done(c);
                return c;
        }
 
@@ -473,13 +493,11 @@ struct composite_context *smb_composite_sesssetup_send(struct smbcli_session *se
 
        if (NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED) || 
            NT_STATUS_IS_OK(status)) {
-               state->req->async.fn = request_handler;
-               state->req->async.private = c;
+               composite_continue_smb(c, state->req, request_handler, c);      
                return c;
        }
 
-       c->state = COMPOSITE_STATE_ERROR;
-       c->status = status;
+       composite_error(c, status);
        return c;
 }