source3/libsmb/cliconnect.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    client connect/disconnect routines
   4    Copyright (C) Andrew Tridgell 1994-1998
   5    Copyright (C) Andrew Bartlett 2001-2003
   6
   7    This program is free software; you can redistribute it and/or modify
   8    it under the terms of the GNU General Public License as published by
   9    the Free Software Foundation; either version 3 of the License, or
  10    (at your option) any later version.
  11
  12    This program is distributed in the hope that it will be useful,
  13    but WITHOUT ANY WARRANTY; without even the implied warranty of
  14    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15    GNU General Public License for more details.
  16
  17    You should have received a copy of the GNU General Public License
  18    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19 */
  20
  21 #include "includes.h"
  22 #include "../libcli/auth/libcli_auth.h"
  23 #include "../libcli/auth/spnego.h"
  24 #include "smb_krb5.h"
  25 #include "ntlmssp.h"
  26
  27 static const struct {
  28         int prot;
  29         const char name[24];
  30 } prots[10] = {
  31         {PROTOCOL_CORE,         "PC NETWORK PROGRAM 1.0"},
  32         {PROTOCOL_COREPLUS,     "MICROSOFT NETWORKS 1.03"},
  33         {PROTOCOL_LANMAN1,      "MICROSOFT NETWORKS 3.0"},
  34         {PROTOCOL_LANMAN1,      "LANMAN1.0"},
  35         {PROTOCOL_LANMAN2,      "LM1.2X002"},
  36         {PROTOCOL_LANMAN2,      "DOS LANMAN2.1"},
  37         {PROTOCOL_LANMAN2,      "LANMAN2.1"},
  38         {PROTOCOL_LANMAN2,      "Samba"},
  39         {PROTOCOL_NT1,          "NT LANMAN 1.0"},
  40         {PROTOCOL_NT1,          "NT LM 0.12"},
  41 };
  42
  43 #define STAR_SMBSERVER "*SMBSERVER"
  44
  45 /**
  46  * Set the user session key for a connection
  47  * @param cli The cli structure to add it too
  48  * @param session_key The session key used.  (A copy of this is taken for the cli struct)
  49  *
  50  */
  51
  52 static void cli_set_session_key (struct cli_state *cli, const DATA_BLOB session_key)
  53 {
  54         cli->user_session_key = data_blob(session_key.data, session_key.length);
  55 }
  56
  57 /****************************************************************************
  58  Do an old lanman2 style session setup.
  59 ****************************************************************************/
  60
  61 static NTSTATUS cli_session_setup_lanman2(struct cli_state *cli,
  62                                           const char *user,
  63                                           const char *pass, size_t passlen,
  64                                           const char *workgroup)
  65 {
  66         DATA_BLOB session_key = data_blob_null;
  67         DATA_BLOB lm_response = data_blob_null;
  68         NTSTATUS status;
  69         fstring pword;
  70         char *p;
  71
  72         if (passlen > sizeof(pword)-1) {
  73                 return NT_STATUS_INVALID_PARAMETER;
  74         }
  75
  76         /* LANMAN servers predate NT status codes and Unicode and ignore those
  77            smb flags so we must disable the corresponding default capabilities
  78            that would otherwise cause the Unicode and NT Status flags to be
  79            set (and even returned by the server) */
  80
  81         cli->capabilities &= ~(CAP_UNICODE | CAP_STATUS32);
  82
  83         /* if in share level security then don't send a password now */
  84         if (!(cli->sec_mode & NEGOTIATE_SECURITY_USER_LEVEL))
  85                 passlen = 0;
  86
  87         if (passlen > 0 && (cli->sec_mode & NEGOTIATE_SECURITY_CHALLENGE_RESPONSE) && passlen != 24) {
  88                 /* Encrypted mode needed, and non encrypted password supplied. */
  89                 lm_response = data_blob(NULL, 24);
  90                 if (!SMBencrypt(pass, cli->secblob.data,(uchar *)lm_response.data)) {
  91                         DEBUG(1, ("Password is > 14 chars in length, and is therefore incompatible with Lanman authentication\n"));
  92                         return NT_STATUS_ACCESS_DENIED;
  93                 }
  94         } else if ((cli->sec_mode & NEGOTIATE_SECURITY_CHALLENGE_RESPONSE) && passlen == 24) {
  95                 /* Encrypted mode needed, and encrypted password supplied. */
  96                 lm_response = data_blob(pass, passlen);
  97         } else if (passlen > 0) {
  98                 /* Plaintext mode needed, assume plaintext supplied. */
  99                 passlen = clistr_push(cli, pword, pass, sizeof(pword), STR_TERMINATE);
 100                 lm_response = data_blob(pass, passlen);
 101         }
 102
 103         /* send a session setup command */
 104         memset(cli->outbuf,'\0',smb_size);
 105         cli_set_message(cli->outbuf,10, 0, True);
 106         SCVAL(cli->outbuf,smb_com,SMBsesssetupX);
 107         cli_setup_packet(cli);
 108
 109         SCVAL(cli->outbuf,smb_vwv0,0xFF);
 110         SSVAL(cli->outbuf,smb_vwv2,cli->max_xmit);
 111         SSVAL(cli->outbuf,smb_vwv3,2);
 112         SSVAL(cli->outbuf,smb_vwv4,1);
 113         SIVAL(cli->outbuf,smb_vwv5,cli->sesskey);
 114         SSVAL(cli->outbuf,smb_vwv7,lm_response.length);
 115
 116         p = smb_buf(cli->outbuf);
 117         memcpy(p,lm_response.data,lm_response.length);
 118         p += lm_response.length;
 119         p += clistr_push(cli, p, user, -1, STR_TERMINATE|STR_UPPER);
 120         p += clistr_push(cli, p, workgroup, -1, STR_TERMINATE|STR_UPPER);
 121         p += clistr_push(cli, p, "Unix", -1, STR_TERMINATE);
 122         p += clistr_push(cli, p, "Samba", -1, STR_TERMINATE);
 123         cli_setup_bcc(cli, p);
 124
 125         if (!cli_send_smb(cli) || !cli_receive_smb(cli)) {
 126                 return cli_nt_error(cli);
 127         }
 128
 129         show_msg(cli->inbuf);
 130
 131         if (cli_is_error(cli)) {
 132                 return cli_nt_error(cli);
 133         }
 134
 135         /* use the returned vuid from now on */
 136         cli->vuid = SVAL(cli->inbuf,smb_uid);
 137         status = cli_set_username(cli, user);
 138         if (!NT_STATUS_IS_OK(status)) {
 139                 return status;
 140         }
 141
 142         if (session_key.data) {
 143                 /* Have plaintext orginal */
 144                 cli_set_session_key(cli, session_key);
 145         }
 146
 147         return NT_STATUS_OK;
 148 }
 149
 150 /****************************************************************************
 151  Work out suitable capabilities to offer the server.
 152 ****************************************************************************/
 153
 154 static uint32 cli_session_setup_capabilities(struct cli_state *cli)
 155 {
 156         uint32 capabilities = CAP_NT_SMBS;
 157
 158         if (!cli->force_dos_errors)
 159                 capabilities |= CAP_STATUS32;
 160
 161         if (cli->use_level_II_oplocks)
 162                 capabilities |= CAP_LEVEL_II_OPLOCKS;
 163
 164         capabilities |= (cli->capabilities & (CAP_UNICODE|CAP_LARGE_FILES|CAP_LARGE_READX|CAP_LARGE_WRITEX|CAP_DFS));
 165         return capabilities;
 166 }
 167
 168 /****************************************************************************
 169  Do a NT1 guest session setup.
 170 ****************************************************************************/
 171
 172 struct cli_session_setup_guest_state {
 173         struct cli_state *cli;
 174         uint16_t vwv[16];
 175         struct iovec bytes;
 176 };
 177
 178 static void cli_session_setup_guest_done(struct tevent_req *subreq);
 179
 180 struct tevent_req *cli_session_setup_guest_create(TALLOC_CTX *mem_ctx,
 181                                                   struct event_context *ev,
 182                                                   struct cli_state *cli,
 183                                                   struct tevent_req **psmbreq)
 184 {
 185         struct tevent_req *req, *subreq;
 186         struct cli_session_setup_guest_state *state;
 187         uint16_t *vwv;
 188         uint8_t *bytes;
 189
 190         req = tevent_req_create(mem_ctx, &state,
 191                                 struct cli_session_setup_guest_state);
 192         if (req == NULL) {
 193                 return NULL;
 194         }
 195         state->cli = cli;
 196         vwv = state->vwv;
 197
 198         SCVAL(vwv+0, 0, 0xFF);
 199         SCVAL(vwv+0, 1, 0);
 200         SSVAL(vwv+1, 0, 0);
 201         SSVAL(vwv+2, 0, CLI_BUFFER_SIZE);
 202         SSVAL(vwv+3, 0, 2);
 203         SSVAL(vwv+4, 0, cli->pid);
 204         SIVAL(vwv+5, 0, cli->sesskey);
 205         SSVAL(vwv+7, 0, 0);
 206         SSVAL(vwv+8, 0, 0);
 207         SSVAL(vwv+9, 0, 0);
 208         SSVAL(vwv+10, 0, 0);
 209         SIVAL(vwv+11, 0, cli_session_setup_capabilities(cli));
 210
 211         bytes = talloc_array(state, uint8_t, 0);
 212
 213         bytes = smb_bytes_push_str(bytes, cli_ucs2(cli), "",  1, /* username */
 214                                    NULL);
 215         bytes = smb_bytes_push_str(bytes, cli_ucs2(cli), "", 1, /* workgroup */
 216                                    NULL);
 217         bytes = smb_bytes_push_str(bytes, cli_ucs2(cli), "Unix", 5, NULL);
 218         bytes = smb_bytes_push_str(bytes, cli_ucs2(cli), "Samba", 6, NULL);
 219
 220         if (bytes == NULL) {
 221                 TALLOC_FREE(req);
 222                 return NULL;
 223         }
 224
 225         state->bytes.iov_base = (void *)bytes;
 226         state->bytes.iov_len = talloc_get_size(bytes);
 227
 228         subreq = cli_smb_req_create(state, ev, cli, SMBsesssetupX, 0, 13, vwv,
 229                                     1, &state->bytes);
 230         if (subreq == NULL) {
 231                 TALLOC_FREE(req);
 232                 return NULL;
 233         }
 234         tevent_req_set_callback(subreq, cli_session_setup_guest_done, req);
 235         *psmbreq = subreq;
 236         return req;
 237 }
 238
 239 struct tevent_req *cli_session_setup_guest_send(TALLOC_CTX *mem_ctx,
 240                                                 struct event_context *ev,
 241                                                 struct cli_state *cli)
 242 {
 243         struct tevent_req *req, *subreq;
 244         NTSTATUS status;
 245
 246         req = cli_session_setup_guest_create(mem_ctx, ev, cli, &subreq);
 247         if (req == NULL) {
 248                 return NULL;
 249         }
 250
 251         status = cli_smb_req_send(subreq);
 252         if (NT_STATUS_IS_OK(status)) {
 253                 tevent_req_nterror(req, status);
 254                 return tevent_req_post(req, ev);
 255         }
 256         return req;
 257 }
 258
 259 static void cli_session_setup_guest_done(struct tevent_req *subreq)
 260 {
 261         struct tevent_req *req = tevent_req_callback_data(
 262                 subreq, struct tevent_req);
 263         struct cli_session_setup_guest_state *state = tevent_req_data(
 264                 req, struct cli_session_setup_guest_state);
 265         struct cli_state *cli = state->cli;
 266         uint32_t num_bytes;
 267         char *inbuf;
 268         uint8_t *bytes;
 269         uint8_t *p;
 270         NTSTATUS status;
 271
 272         status = cli_smb_recv(subreq, 0, NULL, NULL, &num_bytes, &bytes);
 273         if (!NT_STATUS_IS_OK(status)) {
 274                 TALLOC_FREE(subreq);
 275                 tevent_req_nterror(req, status);
 276                 return;
 277         }
 278
 279         inbuf = (char *)cli_smb_inbuf(subreq);
 280         p = bytes;
 281
 282         cli->vuid = SVAL(inbuf, smb_uid);
 283
 284         p += clistr_pull(inbuf, cli->server_os, (char *)p, sizeof(fstring),
 285                          bytes+num_bytes-p, STR_TERMINATE);
 286         p += clistr_pull(inbuf, cli->server_type, (char *)p, sizeof(fstring),
 287                          bytes+num_bytes-p, STR_TERMINATE);
 288         p += clistr_pull(inbuf, cli->server_domain, (char *)p, sizeof(fstring),
 289                          bytes+num_bytes-p, STR_TERMINATE);
 290
 291         if (strstr(cli->server_type, "Samba")) {
 292                 cli->is_samba = True;
 293         }
 294
 295         TALLOC_FREE(subreq);
 296
 297         status = cli_set_username(cli, "");
 298         if (!NT_STATUS_IS_OK(status)) {
 299                 tevent_req_nterror(req, status);
 300                 return;
 301         }
 302         tevent_req_done(req);
 303 }
 304
 305 NTSTATUS cli_session_setup_guest_recv(struct tevent_req *req)
 306 {
 307         return tevent_req_simple_recv_ntstatus(req);
 308 }
 309
 310 static NTSTATUS cli_session_setup_guest(struct cli_state *cli)
 311 {
 312         TALLOC_CTX *frame = talloc_stackframe();
 313         struct event_context *ev;
 314         struct tevent_req *req;
 315         NTSTATUS status = NT_STATUS_OK;
 316
 317         if (cli_has_async_calls(cli)) {
 318                 /*
 319                  * Can't use sync call while an async call is in flight
 320                  */
 321                 status = NT_STATUS_INVALID_PARAMETER;
 322                 goto fail;
 323         }
 324
 325         ev = event_context_init(frame);
 326         if (ev == NULL) {
 327                 status = NT_STATUS_NO_MEMORY;
 328                 goto fail;
 329         }
 330
 331         req = cli_session_setup_guest_send(frame, ev, cli);
 332         if (req == NULL) {
 333                 status = NT_STATUS_NO_MEMORY;
 334                 goto fail;
 335         }
 336
 337         if (!tevent_req_poll(req, ev)) {
 338                 status = map_nt_error_from_unix(errno);
 339                 goto fail;
 340         }
 341
 342         status = cli_session_setup_guest_recv(req);
 343  fail:
 344         TALLOC_FREE(frame);
 345         if (!NT_STATUS_IS_OK(status)) {
 346                 cli_set_error(cli, status);
 347         }
 348         return status;
 349 }
 350
 351 /****************************************************************************
 352  Do a NT1 plaintext session setup.
 353 ****************************************************************************/
 354
 355 static NTSTATUS cli_session_setup_plaintext(struct cli_state *cli,
 356                                             const char *user, const char *pass,
 357                                             const char *workgroup)
 358 {
 359         uint32 capabilities = cli_session_setup_capabilities(cli);
 360         char *p;
 361         NTSTATUS status;
 362         fstring lanman;
 363
 364         fstr_sprintf( lanman, "Samba %s", samba_version_string());
 365
 366         memset(cli->outbuf, '\0', smb_size);
 367         cli_set_message(cli->outbuf,13,0,True);
 368         SCVAL(cli->outbuf,smb_com,SMBsesssetupX);
 369         cli_setup_packet(cli);
 370
 371         SCVAL(cli->outbuf,smb_vwv0,0xFF);
 372         SSVAL(cli->outbuf,smb_vwv2,CLI_BUFFER_SIZE);
 373         SSVAL(cli->outbuf,smb_vwv3,2);
 374         SSVAL(cli->outbuf,smb_vwv4,cli->pid);
 375         SIVAL(cli->outbuf,smb_vwv5,cli->sesskey);
 376         SSVAL(cli->outbuf,smb_vwv8,0);
 377         SIVAL(cli->outbuf,smb_vwv11,capabilities);
 378         p = smb_buf(cli->outbuf);
 379
 380         /* check wether to send the ASCII or UNICODE version of the password */
 381
 382         if ( (capabilities & CAP_UNICODE) == 0 ) {
 383                 p += clistr_push(cli, p, pass, -1, STR_TERMINATE); /* password */
 384                 SSVAL(cli->outbuf,smb_vwv7,PTR_DIFF(p, smb_buf(cli->outbuf)));
 385         }
 386         else {
 387                 /* For ucs2 passwords clistr_push calls ucs2_align, which causes
 388                  * the space taken by the unicode password to be one byte too
 389                  * long (as we're on an odd byte boundary here). Reduce the
 390                  * count by 1 to cope with this. Fixes smbclient against NetApp
 391                  * servers which can't cope. Fix from
 392                  * bryan.kolodziej@allenlund.com in bug #3840.
 393                  */
 394                 p += clistr_push(cli, p, pass, -1, STR_UNICODE|STR_TERMINATE); /* unicode password */
 395                 SSVAL(cli->outbuf,smb_vwv8,PTR_DIFF(p, smb_buf(cli->outbuf))-1);
 396         }
 397
 398         p += clistr_push(cli, p, user, -1, STR_TERMINATE); /* username */
 399         p += clistr_push(cli, p, workgroup, -1, STR_TERMINATE); /* workgroup */
 400         p += clistr_push(cli, p, "Unix", -1, STR_TERMINATE);
 401         p += clistr_push(cli, p, lanman, -1, STR_TERMINATE);
 402         cli_setup_bcc(cli, p);
 403
 404         if (!cli_send_smb(cli) || !cli_receive_smb(cli)) {
 405                 return cli_nt_error(cli);
 406         }
 407
 408         show_msg(cli->inbuf);
 409
 410         if (cli_is_error(cli)) {
 411                 return cli_nt_error(cli);
 412         }
 413
 414         cli->vuid = SVAL(cli->inbuf,smb_uid);
 415         p = smb_buf(cli->inbuf);
 416         p += clistr_pull(cli->inbuf, cli->server_os, p, sizeof(fstring),
 417                          -1, STR_TERMINATE);
 418         p += clistr_pull(cli->inbuf, cli->server_type, p, sizeof(fstring),
 419                          -1, STR_TERMINATE);
 420         p += clistr_pull(cli->inbuf, cli->server_domain, p, sizeof(fstring),
 421                          -1, STR_TERMINATE);
 422         status = cli_set_username(cli, user);
 423         if (!NT_STATUS_IS_OK(status)) {
 424                 return status;
 425         }
 426         if (strstr(cli->server_type, "Samba")) {
 427                 cli->is_samba = True;
 428         }
 429
 430         return NT_STATUS_OK;
 431 }
 432
 433 /****************************************************************************
 434    do a NT1 NTLM/LM encrypted session setup - for when extended security
 435    is not negotiated.
 436    @param cli client state to create do session setup on
 437    @param user username
 438    @param pass *either* cleartext password (passlen !=24) or LM response.
 439    @param ntpass NT response, implies ntpasslen >=24, implies pass is not clear
 440    @param workgroup The user's domain.
 441 ****************************************************************************/
 442
 443 static NTSTATUS cli_session_setup_nt1(struct cli_state *cli, const char *user,
 444                                       const char *pass, size_t passlen,
 445                                       const char *ntpass, size_t ntpasslen,
 446                                       const char *workgroup)
 447 {
 448         uint32 capabilities = cli_session_setup_capabilities(cli);
 449         DATA_BLOB lm_response = data_blob_null;
 450         DATA_BLOB nt_response = data_blob_null;
 451         DATA_BLOB session_key = data_blob_null;
 452         NTSTATUS result;
 453         char *p;
 454         bool ok;
 455
 456         if (passlen == 0) {
 457                 /* do nothing - guest login */
 458         } else if (passlen != 24) {
 459                 if (lp_client_ntlmv2_auth()) {
 460                         DATA_BLOB server_chal;
 461                         DATA_BLOB names_blob;
 462                         server_chal = data_blob(cli->secblob.data, MIN(cli->secblob.length, 8));
 463
 464                         /* note that the 'workgroup' here is a best guess - we don't know
 465                            the server's domain at this point.  The 'server name' is also
 466                            dodgy...
 467                         */
 468                         names_blob = NTLMv2_generate_names_blob(NULL, cli->called.name, workgroup);
 469
 470                         if (!SMBNTLMv2encrypt(NULL, user, workgroup, pass, &server_chal,
 471                                               &names_blob,
 472                                               &lm_response, &nt_response, NULL, &session_key)) {
 473                                 data_blob_free(&names_blob);
 474                                 data_blob_free(&server_chal);
 475                                 return NT_STATUS_ACCESS_DENIED;
 476                         }
 477                         data_blob_free(&names_blob);
 478                         data_blob_free(&server_chal);
 479
 480                 } else {
 481                         uchar nt_hash[16];
 482                         E_md4hash(pass, nt_hash);
 483
 484 #ifdef LANMAN_ONLY
 485                         nt_response = data_blob_null;
 486 #else
 487                         nt_response = data_blob(NULL, 24);
 488                         SMBNTencrypt(pass,cli->secblob.data,nt_response.data);
 489 #endif
 490                         /* non encrypted password supplied. Ignore ntpass. */
 491                         if (lp_client_lanman_auth()) {
 492                                 lm_response = data_blob(NULL, 24);
 493                                 if (!SMBencrypt(pass,cli->secblob.data, lm_response.data)) {
 494                                         /* Oops, the LM response is invalid, just put
 495                                            the NT response there instead */
 496                                         data_blob_free(&lm_response);
 497                                         lm_response = data_blob(nt_response.data, nt_response.length);
 498                                 }
 499                         } else {
 500                                 /* LM disabled, place NT# in LM field instead */
 501                                 lm_response = data_blob(nt_response.data, nt_response.length);
 502                         }
 503
 504                         session_key = data_blob(NULL, 16);
 505 #ifdef LANMAN_ONLY
 506                         E_deshash(pass, session_key.data);
 507                         memset(&session_key.data[8], '\0', 8);
 508 #else
 509                         SMBsesskeygen_ntv1(nt_hash, session_key.data);
 510 #endif
 511                 }
 512                 cli_temp_set_signing(cli);
 513         } else {
 514                 /* pre-encrypted password supplied.  Only used for
 515                    security=server, can't do
 516                    signing because we don't have original key */
 517
 518                 lm_response = data_blob(pass, passlen);
 519                 nt_response = data_blob(ntpass, ntpasslen);
 520         }
 521
 522         /* send a session setup command */
 523         memset(cli->outbuf,'\0',smb_size);
 524
 525         cli_set_message(cli->outbuf,13,0,True);
 526         SCVAL(cli->outbuf,smb_com,SMBsesssetupX);
 527         cli_setup_packet(cli);
 528
 529         SCVAL(cli->outbuf,smb_vwv0,0xFF);
 530         SSVAL(cli->outbuf,smb_vwv2,CLI_BUFFER_SIZE);
 531         SSVAL(cli->outbuf,smb_vwv3,2);
 532         SSVAL(cli->outbuf,smb_vwv4,cli->pid);
 533         SIVAL(cli->outbuf,smb_vwv5,cli->sesskey);
 534         SSVAL(cli->outbuf,smb_vwv7,lm_response.length);
 535         SSVAL(cli->outbuf,smb_vwv8,nt_response.length);
 536         SIVAL(cli->outbuf,smb_vwv11,capabilities);
 537         p = smb_buf(cli->outbuf);
 538         if (lm_response.length) {
 539                 memcpy(p,lm_response.data, lm_response.length); p += lm_response.length;
 540         }
 541         if (nt_response.length) {
 542                 memcpy(p,nt_response.data, nt_response.length); p += nt_response.length;
 543         }
 544         p += clistr_push(cli, p, user, -1, STR_TERMINATE);
 545
 546         /* Upper case here might help some NTLMv2 implementations */
 547         p += clistr_push(cli, p, workgroup, -1, STR_TERMINATE|STR_UPPER);
 548         p += clistr_push(cli, p, "Unix", -1, STR_TERMINATE);
 549         p += clistr_push(cli, p, "Samba", -1, STR_TERMINATE);
 550         cli_setup_bcc(cli, p);
 551
 552         if (!cli_send_smb(cli) || !cli_receive_smb(cli)) {
 553                 result = cli_nt_error(cli);
 554                 goto end;
 555         }
 556
 557         /* show_msg(cli->inbuf); */
 558
 559         if (cli_is_error(cli)) {
 560                 result = cli_nt_error(cli);
 561                 goto end;
 562         }
 563
 564 #ifdef LANMAN_ONLY
 565         ok = cli_simple_set_signing(cli, session_key, lm_response);
 566 #else
 567         ok = cli_simple_set_signing(cli, session_key, nt_response);
 568 #endif
 569         if (ok) {
 570                 if (!cli_check_sign_mac(cli, cli->inbuf, 1)) {
 571                         result = NT_STATUS_ACCESS_DENIED;
 572                         goto end;
 573                 }
 574         }
 575
 576         /* use the returned vuid from now on */
 577         cli->vuid = SVAL(cli->inbuf,smb_uid);
 578
 579         p = smb_buf(cli->inbuf);
 580         p += clistr_pull(cli->inbuf, cli->server_os, p, sizeof(fstring),
 581                          -1, STR_TERMINATE);
 582         p += clistr_pull(cli->inbuf, cli->server_type, p, sizeof(fstring),
 583                          -1, STR_TERMINATE);
 584         p += clistr_pull(cli->inbuf, cli->server_domain, p, sizeof(fstring),
 585                          -1, STR_TERMINATE);
 586
 587         if (strstr(cli->server_type, "Samba")) {
 588                 cli->is_samba = True;
 589         }
 590
 591         result = cli_set_username(cli, user);
 592         if (!NT_STATUS_IS_OK(result)) {
 593                 goto end;
 594         }
 595
 596         if (session_key.data) {
 597                 /* Have plaintext orginal */
 598                 cli_set_session_key(cli, session_key);
 599         }
 600
 601         result = NT_STATUS_OK;
 602 end:
 603         data_blob_free(&lm_response);
 604         data_blob_free(&nt_response);
 605         data_blob_free(&session_key);
 606         return result;
 607 }
 608
 609 /* The following is calculated from :
 610  * (smb_size-4) = 35
 611  * (smb_wcnt * 2) = 24 (smb_wcnt == 12 in cli_session_setup_blob_send() )
 612  * (strlen("Unix") + 1 + strlen("Samba") + 1) * 2 = 22 (unicode strings at
 613  * end of packet.
 614  */
 615
 616 #define BASE_SESSSETUP_BLOB_PACKET_SIZE (35 + 24 + 22)
 617
 618 struct cli_sesssetup_blob_state {
 619         struct tevent_context *ev;
 620         struct cli_state *cli;
 621         DATA_BLOB blob;
 622         uint16_t max_blob_size;
 623         uint16_t vwv[12];
 624         uint8_t *buf;
 625
 626         NTSTATUS status;
 627         char *inbuf;
 628         DATA_BLOB ret_blob;
 629 };
 630
 631 static bool cli_sesssetup_blob_next(struct cli_sesssetup_blob_state *state,
 632                                     struct tevent_req **psubreq);
 633 static void cli_sesssetup_blob_done(struct tevent_req *subreq);
 634
 635 static struct tevent_req *cli_sesssetup_blob_send(TALLOC_CTX *mem_ctx,
 636                                                   struct tevent_context *ev,
 637                                                   struct cli_state *cli,
 638                                                   DATA_BLOB blob)
 639 {
 640         struct tevent_req *req, *subreq;
 641         struct cli_sesssetup_blob_state *state;
 642
 643         req = tevent_req_create(mem_ctx, &state,
 644                                 struct cli_sesssetup_blob_state);
 645         if (req == NULL) {
 646                 return NULL;
 647         }
 648         state->ev = ev;
 649         state->blob = blob;
 650         state->cli = cli;
 651
 652         if (cli->max_xmit < BASE_SESSSETUP_BLOB_PACKET_SIZE + 1) {
 653                 DEBUG(1, ("cli_session_setup_blob: cli->max_xmit too small "
 654                           "(was %u, need minimum %u)\n",
 655                           (unsigned int)cli->max_xmit,
 656                           BASE_SESSSETUP_BLOB_PACKET_SIZE));
 657                 tevent_req_nterror(req, NT_STATUS_INVALID_PARAMETER);
 658                 return tevent_req_post(req, ev);
 659         }
 660         state->max_blob_size =
 661                 MIN(cli->max_xmit - BASE_SESSSETUP_BLOB_PACKET_SIZE, 0xFFFF);
 662
 663         if (!cli_sesssetup_blob_next(state, &subreq)) {
 664                 tevent_req_nomem(NULL, req);
 665                 return tevent_req_post(req, ev);
 666         }
 667         tevent_req_set_callback(subreq, cli_sesssetup_blob_done, req);
 668         return req;
 669 }
 670
 671 static bool cli_sesssetup_blob_next(struct cli_sesssetup_blob_state *state,
 672                                     struct tevent_req **psubreq)
 673 {
 674         struct tevent_req *subreq;
 675         uint16_t thistime;
 676
 677         SCVAL(state->vwv+0, 0, 0xFF);
 678         SCVAL(state->vwv+0, 1, 0);
 679         SSVAL(state->vwv+1, 0, 0);
 680         SSVAL(state->vwv+2, 0, CLI_BUFFER_SIZE);
 681         SSVAL(state->vwv+3, 0, 2);
 682         SSVAL(state->vwv+4, 0, 1);
 683         SIVAL(state->vwv+5, 0, 0);
 684
 685         thistime = MIN(state->blob.length, state->max_blob_size);
 686         SSVAL(state->vwv+7, 0, thistime);
 687
 688         SSVAL(state->vwv+8, 0, 0);
 689         SSVAL(state->vwv+9, 0, 0);
 690         SIVAL(state->vwv+10, 0,
 691               cli_session_setup_capabilities(state->cli)
 692               | CAP_EXTENDED_SECURITY);
 693
 694         state->buf = (uint8_t *)talloc_memdup(state, state->blob.data,
 695                                               thistime);
 696         if (state->buf == NULL) {
 697                 return false;
 698         }
 699         state->blob.data += thistime;
 700         state->blob.length -= thistime;
 701
 702         state->buf = smb_bytes_push_str(state->buf, cli_ucs2(state->cli),
 703                                         "Unix", 5, NULL);
 704         state->buf = smb_bytes_push_str(state->buf, cli_ucs2(state->cli),
 705                                         "Samba", 6, NULL);
 706         if (state->buf == NULL) {
 707                 return false;
 708         }
 709         subreq = cli_smb_send(state, state->ev, state->cli, SMBsesssetupX, 0,
 710                               12, state->vwv,
 711                               talloc_get_size(state->buf), state->buf);
 712         if (subreq == NULL) {
 713                 return false;
 714         }
 715         *psubreq = subreq;
 716         return true;
 717 }
 718
 719 static void cli_sesssetup_blob_done(struct tevent_req *subreq)
 720 {
 721         struct tevent_req *req = tevent_req_callback_data(
 722                 subreq, struct tevent_req);
 723         struct cli_sesssetup_blob_state *state = tevent_req_data(
 724                 req, struct cli_sesssetup_blob_state);
 725         struct cli_state *cli = state->cli;
 726         uint8_t wct;
 727         uint16_t *vwv;
 728         uint32_t num_bytes;
 729         uint8_t *bytes;
 730         NTSTATUS status;
 731         uint8_t *p;
 732         uint16_t blob_length;
 733
 734         status = cli_smb_recv(subreq, 1, &wct, &vwv, &num_bytes, &bytes);
 735         if (!NT_STATUS_IS_OK(status)
 736             && !NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
 737                 TALLOC_FREE(subreq);
 738                 tevent_req_nterror(req, status);
 739                 return;
 740         }
 741
 742         state->status = status;
 743         TALLOC_FREE(state->buf);
 744
 745         state->inbuf = (char *)cli_smb_inbuf(subreq);
 746         cli->vuid = SVAL(state->inbuf, smb_uid);
 747
 748         blob_length = SVAL(vwv+3, 0);
 749         if (blob_length > num_bytes) {
 750                 TALLOC_FREE(subreq);
 751                 tevent_req_nterror(req, NT_STATUS_INVALID_NETWORK_RESPONSE);
 752                 return;
 753         }
 754         state->ret_blob = data_blob_const(bytes, blob_length);
 755
 756         p = bytes + blob_length;
 757
 758         p += clistr_pull(state->inbuf, cli->server_os,
 759                          (char *)p, sizeof(fstring),
 760                          bytes+num_bytes-p, STR_TERMINATE);
 761         p += clistr_pull(state->inbuf, cli->server_type,
 762                          (char *)p, sizeof(fstring),
 763                          bytes+num_bytes-p, STR_TERMINATE);
 764         p += clistr_pull(state->inbuf, cli->server_domain,
 765                          (char *)p, sizeof(fstring),
 766                          bytes+num_bytes-p, STR_TERMINATE);
 767
 768         if (strstr(cli->server_type, "Samba")) {
 769                 cli->is_samba = True;
 770         }
 771
 772         if (state->blob.length != 0) {
 773                 TALLOC_FREE(subreq);
 774                 /*
 775                  * More to send
 776                  */
 777                 if (!cli_sesssetup_blob_next(state, &subreq)) {
 778                         tevent_req_nomem(NULL, req);
 779                         return;
 780                 }
 781                 tevent_req_set_callback(subreq, cli_sesssetup_blob_done, req);
 782                 return;
 783         }
 784         tevent_req_done(req);
 785 }
 786
 787 static NTSTATUS cli_sesssetup_blob_recv(struct tevent_req *req,
 788                                         TALLOC_CTX *mem_ctx,
 789                                         DATA_BLOB *pblob,
 790                                         char **pinbuf)
 791 {
 792         struct cli_sesssetup_blob_state *state = tevent_req_data(
 793                 req, struct cli_sesssetup_blob_state);
 794         NTSTATUS status;
 795         char *inbuf;
 796
 797         if (tevent_req_is_nterror(req, &status)) {
 798                 state->cli->vuid = 0;
 799                 return status;
 800         }
 801
 802         inbuf = talloc_move(mem_ctx, &state->inbuf);
 803         if (pblob != NULL) {
 804                 *pblob = state->ret_blob;
 805         }
 806         if (pinbuf != NULL) {
 807                 *pinbuf = inbuf;
 808         }
 809         /* could be NT_STATUS_MORE_PROCESSING_REQUIRED */
 810         return state->status;
 811 }
 812
 813 #ifdef HAVE_KRB5
 814
 815 /****************************************************************************
 816  Use in-memory credentials cache
 817 ****************************************************************************/
 818
 819 static void use_in_memory_ccache(void) {
 820         setenv(KRB5_ENV_CCNAME, "MEMORY:cliconnect", 1);
 821 }
 822
 823 /****************************************************************************
 824  Do a spnego/kerberos encrypted session setup.
 825 ****************************************************************************/
 826
 827 struct cli_session_setup_kerberos_state {
 828         struct cli_state *cli;
 829         DATA_BLOB negTokenTarg;
 830         DATA_BLOB session_key_krb5;
 831         ADS_STATUS ads_status;
 832 };
 833
 834 static void cli_session_setup_kerberos_done(struct tevent_req *subreq);
 835
 836 static struct tevent_req *cli_session_setup_kerberos_send(
 837         TALLOC_CTX *mem_ctx, struct tevent_context *ev, struct cli_state *cli,
 838         const char *principal, const char *workgroup)
 839 {
 840         struct tevent_req *req, *subreq;
 841         struct cli_session_setup_kerberos_state *state;
 842         int rc;
 843
 844         DEBUG(2,("Doing kerberos session setup\n"));
 845
 846         req = tevent_req_create(mem_ctx, &state,
 847                                 struct cli_session_setup_kerberos_state);
 848         if (req == NULL) {
 849                 return NULL;
 850         }
 851         state->cli = cli;
 852         state->ads_status = ADS_SUCCESS;
 853
 854         cli_temp_set_signing(cli);
 855
 856         /*
 857          * Ok, this is cheated: spnego_gen_negTokenTarg can block if
 858          * we have to acquire a ticket. To be fixed later :-)
 859          */
 860         rc = spnego_gen_negTokenTarg(principal, 0, &state->negTokenTarg,
 861                                      &state->session_key_krb5, 0, NULL);
 862         if (rc) {
 863                 DEBUG(1, ("cli_session_setup_kerberos: "
 864                           "spnego_gen_negTokenTarg failed: %s\n",
 865                           error_message(rc)));
 866                 state->ads_status = ADS_ERROR_KRB5(rc);
 867                 tevent_req_nterror(req, NT_STATUS_UNSUCCESSFUL);
 868                 return tevent_req_post(req, ev);
 869         }
 870
 871 #if 0
 872         file_save("negTokenTarg.dat", state->negTokenTarg.data,
 873                   state->negTokenTarg.length);
 874 #endif
 875
 876         subreq = cli_sesssetup_blob_send(state, ev, cli, state->negTokenTarg);
 877         if (tevent_req_nomem(subreq, req)) {
 878                 return tevent_req_post(req, ev);
 879         }
 880         tevent_req_set_callback(subreq, cli_session_setup_kerberos_done, req);
 881         return req;
 882 }
 883
 884 static void cli_session_setup_kerberos_done(struct tevent_req *subreq)
 885 {
 886         struct tevent_req *req = tevent_req_callback_data(
 887                 subreq, struct tevent_req);
 888         struct cli_session_setup_kerberos_state *state = tevent_req_data(
 889                 req, struct cli_session_setup_kerberos_state);
 890         char *inbuf = NULL;
 891         NTSTATUS status;
 892
 893         status = cli_sesssetup_blob_recv(subreq, talloc_tos(), NULL, &inbuf);
 894         if (!NT_STATUS_IS_OK(status)) {
 895                 TALLOC_FREE(subreq);
 896                 tevent_req_nterror(req, status);
 897                 return;
 898         }
 899
 900         cli_set_session_key(state->cli, state->session_key_krb5);
 901
 902         if (cli_simple_set_signing(state->cli, state->session_key_krb5,
 903                                    data_blob_null)
 904             && !cli_check_sign_mac(state->cli, inbuf, 1)) {
 905                 TALLOC_FREE(subreq);
 906                 tevent_req_nterror(req, NT_STATUS_ACCESS_DENIED);
 907                 return;
 908         }
 909         TALLOC_FREE(subreq);
 910         tevent_req_done(req);
 911 }
 912
 913 static ADS_STATUS cli_session_setup_kerberos_recv(struct tevent_req *req)
 914 {
 915         struct cli_session_setup_kerberos_state *state = tevent_req_data(
 916                 req, struct cli_session_setup_kerberos_state);
 917         NTSTATUS status;
 918
 919         if (tevent_req_is_nterror(req, &status)) {
 920                 return ADS_ERROR_NT(status);
 921         }
 922         return state->ads_status;
 923 }
 924
 925 static ADS_STATUS cli_session_setup_kerberos(struct cli_state *cli,
 926                                              const char *principal,
 927                                              const char *workgroup)
 928 {
 929         struct tevent_context *ev;
 930         struct tevent_req *req;
 931         ADS_STATUS status = ADS_ERROR_NT(NT_STATUS_NO_MEMORY);
 932
 933         if (cli_has_async_calls(cli)) {
 934                 return ADS_ERROR_NT(NT_STATUS_INVALID_PARAMETER);
 935         }
 936         ev = tevent_context_init(talloc_tos());
 937         if (ev == NULL) {
 938                 goto fail;
 939         }
 940         req = cli_session_setup_kerberos_send(ev, ev, cli, principal,
 941                                               workgroup);
 942         if (req == NULL) {
 943                 goto fail;
 944         }
 945         if (!tevent_req_poll(req, ev)) {
 946                 status = ADS_ERROR_SYSTEM(errno);
 947                 goto fail;
 948         }
 949         status = cli_session_setup_kerberos_recv(req);
 950 fail:
 951         TALLOC_FREE(ev);
 952         return status;
 953 }
 954 #endif  /* HAVE_KRB5 */
 955
 956 /****************************************************************************
 957  Do a spnego/NTLMSSP encrypted session setup.
 958 ****************************************************************************/
 959
 960 struct cli_session_setup_ntlmssp_state {
 961         struct tevent_context *ev;
 962         struct cli_state *cli;
 963         struct ntlmssp_state *ntlmssp_state;
 964         int turn;
 965         DATA_BLOB blob_out;
 966 };
 967
 968 static int cli_session_setup_ntlmssp_state_destructor(
 969         struct cli_session_setup_ntlmssp_state *state)
 970 {
 971         if (state->ntlmssp_state != NULL) {
 972                 ntlmssp_end(&state->ntlmssp_state);
 973         }
 974         return 0;
 975 }
 976
 977 static void cli_session_setup_ntlmssp_done(struct tevent_req *req);
 978
 979 static struct tevent_req *cli_session_setup_ntlmssp_send(
 980         TALLOC_CTX *mem_ctx, struct tevent_context *ev, struct cli_state *cli,
 981         const char *user, const char *pass, const char *domain)
 982 {
 983         struct tevent_req *req, *subreq;
 984         struct cli_session_setup_ntlmssp_state *state;
 985         NTSTATUS status;
 986         DATA_BLOB blob_out;
 987
 988         req = tevent_req_create(mem_ctx, &state,
 989                                 struct cli_session_setup_ntlmssp_state);
 990         if (req == NULL) {
 991                 return NULL;
 992         }
 993         state->ev = ev;
 994         state->cli = cli;
 995         state->turn = 1;
 996
 997         state->ntlmssp_state = NULL;
 998         talloc_set_destructor(
 999                 state, cli_session_setup_ntlmssp_state_destructor);
1000
1001         cli_temp_set_signing(cli);
1002
1003         status = ntlmssp_client_start(&state->ntlmssp_state);
1004         if (!NT_STATUS_IS_OK(status)) {
1005                 goto fail;
1006         }
1007         ntlmssp_want_feature(state->ntlmssp_state,
1008                              NTLMSSP_FEATURE_SESSION_KEY);
1009         status = ntlmssp_set_username(state->ntlmssp_state, user);
1010         if (!NT_STATUS_IS_OK(status)) {
1011                 goto fail;
1012         }
1013         status = ntlmssp_set_domain(state->ntlmssp_state, domain);
1014         if (!NT_STATUS_IS_OK(status)) {
1015                 goto fail;
1016         }
1017         status = ntlmssp_set_password(state->ntlmssp_state, pass);
1018         if (!NT_STATUS_IS_OK(status)) {
1019                 goto fail;
1020         }
1021         status = ntlmssp_update(state->ntlmssp_state, data_blob_null,
1022                                 &blob_out);
1023         if (!NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
1024                 goto fail;
1025         }
1026
1027         state->blob_out = gen_negTokenInit(OID_NTLMSSP, blob_out);
1028         data_blob_free(&blob_out);
1029
1030         subreq = cli_sesssetup_blob_send(state, ev, cli, state->blob_out);
1031         if (tevent_req_nomem(subreq, req)) {
1032                 return tevent_req_post(req, ev);
1033         }
1034         tevent_req_set_callback(subreq, cli_session_setup_ntlmssp_done, req);
1035         return req;
1036 fail:
1037         tevent_req_nterror(req, status);
1038         return tevent_req_post(req, ev);
1039 }
1040
1041 static void cli_session_setup_ntlmssp_done(struct tevent_req *subreq)
1042 {
1043         struct tevent_req *req = tevent_req_callback_data(
1044                 subreq, struct tevent_req);
1045         struct cli_session_setup_ntlmssp_state *state = tevent_req_data(
1046                 req, struct cli_session_setup_ntlmssp_state);
1047         DATA_BLOB blob_in, msg_in, blob_out;
1048         char *inbuf = NULL;
1049         bool parse_ret;
1050         NTSTATUS status;
1051
1052         status = cli_sesssetup_blob_recv(subreq, talloc_tos(), &blob_in,
1053                                          &inbuf);
1054         TALLOC_FREE(subreq);
1055         data_blob_free(&state->blob_out);
1056
1057         if (NT_STATUS_IS_OK(status)) {
1058                 if (state->cli->server_domain[0] == '\0') {
1059                         fstrcpy(state->cli->server_domain,
1060                                 state->ntlmssp_state->server_domain);
1061                 }
1062                 cli_set_session_key(
1063                         state->cli, state->ntlmssp_state->session_key);
1064
1065                 if (cli_simple_set_signing(
1066                             state->cli, state->ntlmssp_state->session_key,
1067                             data_blob_null)
1068                     && !cli_check_sign_mac(state->cli, inbuf, 1)) {
1069                         TALLOC_FREE(subreq);
1070                         tevent_req_nterror(req, NT_STATUS_ACCESS_DENIED);
1071                         return;
1072                 }
1073                 TALLOC_FREE(subreq);
1074                 ntlmssp_end(&state->ntlmssp_state);
1075                 tevent_req_done(req);
1076                 return;
1077         }
1078         if (!NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
1079                 tevent_req_nterror(req, status);
1080                 return;
1081         }
1082
1083         if (blob_in.length == 0) {
1084                 tevent_req_nterror(req, NT_STATUS_UNSUCCESSFUL);
1085                 return;
1086         }
1087
1088         if ((state->turn == 1)
1089             && NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
1090                 DATA_BLOB tmp_blob = data_blob_null;
1091                 /* the server might give us back two challenges */
1092                 parse_ret = spnego_parse_challenge(blob_in, &msg_in,
1093                                                    &tmp_blob);
1094                 data_blob_free(&tmp_blob);
1095         } else {
1096                 parse_ret = spnego_parse_auth_response(blob_in, status,
1097                                                        OID_NTLMSSP, &msg_in);
1098         }
1099         state->turn += 1;
1100
1101         if (!parse_ret) {
1102                 DEBUG(3,("Failed to parse auth response\n"));
1103                 if (NT_STATUS_IS_OK(status)
1104                     || NT_STATUS_EQUAL(status,
1105                                        NT_STATUS_MORE_PROCESSING_REQUIRED)) {
1106                         tevent_req_nterror(
1107                                 req, NT_STATUS_INVALID_NETWORK_RESPONSE);
1108                         return;
1109                 }
1110         }
1111
1112         status = ntlmssp_update(state->ntlmssp_state, msg_in, &blob_out);
1113
1114         if (!NT_STATUS_IS_OK(status)
1115             && !NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED)) {
1116                 TALLOC_FREE(subreq);
1117                 ntlmssp_end(&state->ntlmssp_state);
1118                 tevent_req_nterror(req, status);
1119                 return;
1120         }
1121
1122         state->blob_out = spnego_gen_auth(blob_out);
1123         TALLOC_FREE(subreq);
1124         if (tevent_req_nomem(state->blob_out.data, req)) {
1125                 return;
1126         }
1127
1128         subreq = cli_sesssetup_blob_send(state, state->ev, state->cli,
1129                                          state->blob_out);
1130         if (tevent_req_nomem(subreq, req)) {
1131                 return;
1132         }
1133         tevent_req_set_callback(subreq, cli_session_setup_ntlmssp_done, req);
1134 }
1135
1136 static NTSTATUS cli_session_setup_ntlmssp_recv(struct tevent_req *req)
1137 {
1138         struct cli_session_setup_ntlmssp_state *state = tevent_req_data(
1139                 req, struct cli_session_setup_ntlmssp_state);
1140         NTSTATUS status;
1141
1142         if (tevent_req_is_nterror(req, &status)) {
1143                 state->cli->vuid = 0;
1144                 return status;
1145         }
1146         return NT_STATUS_OK;
1147 }
1148
1149 static NTSTATUS cli_session_setup_ntlmssp(struct cli_state *cli,
1150                                           const char *user,
1151                                           const char *pass,
1152                                           const char *domain)
1153 {
1154         struct tevent_context *ev;
1155         struct tevent_req *req;
1156         NTSTATUS status = NT_STATUS_NO_MEMORY;
1157
1158         if (cli_has_async_calls(cli)) {
1159                 return NT_STATUS_INVALID_PARAMETER;
1160         }
1161         ev = tevent_context_init(talloc_tos());
1162         if (ev == NULL) {
1163                 goto fail;
1164         }
1165         req = cli_session_setup_ntlmssp_send(ev, ev, cli, user, pass, domain);
1166         if (req == NULL) {
1167                 goto fail;
1168         }
1169         if (!tevent_req_poll_ntstatus(req, ev, &status)) {
1170                 goto fail;
1171         }
1172         status = cli_session_setup_ntlmssp_recv(req);
1173 fail:
1174         TALLOC_FREE(ev);
1175         if (!NT_STATUS_IS_OK(status)) {
1176                 cli_set_error(cli, status);
1177         }
1178         return status;
1179 }
1180
1181 /****************************************************************************
1182  Do a spnego encrypted session setup.
1183
1184  user_domain: The shortname of the domain the user/machine is a member of.
1185  dest_realm: The realm we're connecting to, if NULL we use our default realm.
1186 ****************************************************************************/
1187
1188 ADS_STATUS cli_session_setup_spnego(struct cli_state *cli, const char *user,
1189                               const char *pass, const char *user_domain,
1190                               const char * dest_realm)
1191 {
1192         char *principal = NULL;
1193         char *OIDs[ASN1_MAX_OIDS];
1194         int i;
1195         DATA_BLOB blob;
1196         const char *p = NULL;
1197         char *account = NULL;
1198         NTSTATUS status;
1199
1200         DEBUG(3,("Doing spnego session setup (blob length=%lu)\n", (unsigned long)cli->secblob.length));
1201
1202         /* the server might not even do spnego */
1203         if (cli->secblob.length <= 16) {
1204                 DEBUG(3,("server didn't supply a full spnego negprot\n"));
1205                 goto ntlmssp;
1206         }
1207
1208 #if 0
1209         file_save("negprot.dat", cli->secblob.data, cli->secblob.length);
1210 #endif
1211
1212         /* there is 16 bytes of GUID before the real spnego packet starts */
1213         blob = data_blob(cli->secblob.data+16, cli->secblob.length-16);
1214
1215         /* The server sent us the first part of the SPNEGO exchange in the
1216          * negprot reply. It is WRONG to depend on the principal sent in the
1217          * negprot reply, but right now we do it. If we don't receive one,
1218          * we try to best guess, then fall back to NTLM.  */
1219         if (!spnego_parse_negTokenInit(blob, OIDs, &principal)) {
1220                 data_blob_free(&blob);
1221                 return ADS_ERROR_NT(NT_STATUS_INVALID_PARAMETER);
1222         }
1223         data_blob_free(&blob);
1224
1225         /* make sure the server understands kerberos */
1226         for (i=0;OIDs[i];i++) {
1227                 if (i == 0)
1228                         DEBUG(3,("got OID=%s\n", OIDs[i]));
1229                 else
1230                         DEBUGADD(3,("got OID=%s\n", OIDs[i]));
1231                 if (strcmp(OIDs[i], OID_KERBEROS5_OLD) == 0 ||
1232                     strcmp(OIDs[i], OID_KERBEROS5) == 0) {
1233                         cli->got_kerberos_mechanism = True;
1234                 }
1235                 talloc_free(OIDs[i]);
1236         }
1237
1238         DEBUG(3,("got principal=%s\n", principal ? principal : "<null>"));
1239
1240         status = cli_set_username(cli, user);
1241         if (!NT_STATUS_IS_OK(status)) {
1242                 return ADS_ERROR_NT(status);
1243         }
1244
1245 #ifdef HAVE_KRB5
1246         /* If password is set we reauthenticate to kerberos server
1247          * and do not store results */
1248
1249         if (cli->got_kerberos_mechanism && cli->use_kerberos) {
1250                 ADS_STATUS rc;
1251
1252                 if (pass && *pass) {
1253                         int ret;
1254
1255                         use_in_memory_ccache();
1256                         ret = kerberos_kinit_password(user, pass, 0 /* no time correction for now */, NULL);
1257
1258                         if (ret){
1259                                 TALLOC_FREE(principal);
1260                                 DEBUG(0, ("Kinit failed: %s\n", error_message(ret)));
1261                                 if (cli->fallback_after_kerberos)
1262                                         goto ntlmssp;
1263                                 return ADS_ERROR_KRB5(ret);
1264                         }
1265                 }
1266
1267                 /* If we get a bad principal, try to guess it if
1268                    we have a valid host NetBIOS name.
1269                  */
1270                 if (strequal(principal, ADS_IGNORE_PRINCIPAL)) {
1271                         TALLOC_FREE(principal);
1272                 }
1273
1274                 if (principal == NULL &&
1275                         !is_ipaddress(cli->desthost) &&
1276                         !strequal(STAR_SMBSERVER,
1277                                 cli->desthost)) {
1278                         char *realm = NULL;
1279                         char *machine = NULL;
1280                         char *host = NULL;
1281                         DEBUG(3,("cli_session_setup_spnego: got a "
1282                                 "bad server principal, trying to guess ...\n"));
1283
1284                         host = strchr_m(cli->desthost, '.');
1285                         if (host) {
1286                                 machine = SMB_STRNDUP(cli->desthost,
1287                                         host - cli->desthost);
1288                         } else {
1289                                 machine = SMB_STRDUP(cli->desthost);
1290                         }
1291                         if (machine == NULL) {
1292                                 return ADS_ERROR_NT(NT_STATUS_NO_MEMORY);
1293                         }
1294
1295                         if (dest_realm) {
1296                                 realm = SMB_STRDUP(dest_realm);
1297                                 strupper_m(realm);
1298                         } else {
1299                                 realm = kerberos_get_default_realm_from_ccache();
1300                         }
1301                         if (realm && *realm) {
1302                                 principal = talloc_asprintf(NULL, "%s$@%s",
1303                                                         machine, realm);
1304                                 if (!principal) {
1305                                         SAFE_FREE(machine);
1306                                         SAFE_FREE(realm);
1307                                         return ADS_ERROR_NT(NT_STATUS_NO_MEMORY);
1308                                 }
1309                                 DEBUG(3,("cli_session_setup_spnego: guessed "
1310                                         "server principal=%s\n",
1311                                         principal ? principal : "<null>"));
1312                         }
1313                         SAFE_FREE(machine);
1314                         SAFE_FREE(realm);
1315                 }
1316
1317                 if (principal) {
1318                         rc = cli_session_setup_kerberos(cli, principal,
1319                                 dest_realm);
1320                         if (ADS_ERR_OK(rc) || !cli->fallback_after_kerberos) {
1321                                 TALLOC_FREE(principal);
1322                                 return rc;
1323                         }
1324                 }
1325         }
1326 #endif
1327
1328         TALLOC_FREE(principal);
1329
1330 ntlmssp:
1331
1332         account = talloc_strdup(talloc_tos(), user);
1333         if (!account) {
1334                 return ADS_ERROR_NT(NT_STATUS_NO_MEMORY);
1335         }
1336
1337         /* when falling back to ntlmssp while authenticating with a machine
1338          * account strip off the realm - gd */
1339
1340         if ((p = strchr_m(user, '@')) != NULL) {
1341                 account[PTR_DIFF(p,user)] = '\0';
1342         }
1343
1344         return ADS_ERROR_NT(cli_session_setup_ntlmssp(cli, account, pass, user_domain));
1345 }
1346
1347 /****************************************************************************
1348  Send a session setup. The username and workgroup is in UNIX character
1349  format and must be converted to DOS codepage format before sending. If the
1350  password is in plaintext, the same should be done.
1351 ****************************************************************************/
1352
1353 NTSTATUS cli_session_setup(struct cli_state *cli,
1354                            const char *user,
1355                            const char *pass, int passlen,
1356                            const char *ntpass, int ntpasslen,
1357                            const char *workgroup)
1358 {
1359         char *p;
1360         fstring user2;
1361
1362         if (user) {
1363                 fstrcpy(user2, user);
1364         } else {
1365                 user2[0] ='\0';
1366         }
1367
1368         if (!workgroup) {
1369                 workgroup = "";
1370         }
1371
1372         /* allow for workgroups as part of the username */
1373         if ((p=strchr_m(user2,'\\')) || (p=strchr_m(user2,'/')) ||
1374             (p=strchr_m(user2,*lp_winbind_separator()))) {
1375                 *p = 0;
1376                 user = p+1;
1377                 workgroup = user2;
1378         }
1379
1380         if (cli->protocol < PROTOCOL_LANMAN1) {
1381                 return NT_STATUS_OK;
1382         }
1383
1384         /* now work out what sort of session setup we are going to
1385            do. I have split this into separate functions to make the
1386            flow a bit easier to understand (tridge) */
1387
1388         /* if its an older server then we have to use the older request format */
1389
1390         if (cli->protocol < PROTOCOL_NT1) {
1391                 if (!lp_client_lanman_auth() && passlen != 24 && (*pass)) {
1392                         DEBUG(1, ("Server requested LM password but 'client lanman auth'"
1393                                   " is disabled\n"));
1394                         return NT_STATUS_ACCESS_DENIED;
1395                 }
1396
1397                 if ((cli->sec_mode & NEGOTIATE_SECURITY_CHALLENGE_RESPONSE) == 0 &&
1398                     !lp_client_plaintext_auth() && (*pass)) {
1399                         DEBUG(1, ("Server requested plaintext password but "
1400                                   "'client plaintext auth' is disabled\n"));
1401                         return NT_STATUS_ACCESS_DENIED;
1402                 }
1403
1404                 return cli_session_setup_lanman2(cli, user, pass, passlen,
1405                                                  workgroup);
1406         }
1407
1408         /* if no user is supplied then we have to do an anonymous connection.
1409            passwords are ignored */
1410
1411         if (!user || !*user)
1412                 return cli_session_setup_guest(cli);
1413
1414         /* if the server is share level then send a plaintext null
1415            password at this point. The password is sent in the tree
1416            connect */
1417
1418         if ((cli->sec_mode & NEGOTIATE_SECURITY_USER_LEVEL) == 0)
1419                 return cli_session_setup_plaintext(cli, user, "", workgroup);
1420
1421         /* if the server doesn't support encryption then we have to use
1422            plaintext. The second password is ignored */
1423
1424         if ((cli->sec_mode & NEGOTIATE_SECURITY_CHALLENGE_RESPONSE) == 0) {
1425                 if (!lp_client_plaintext_auth() && (*pass)) {
1426                         DEBUG(1, ("Server requested plaintext password but "
1427                                   "'client plaintext auth' is disabled\n"));
1428                         return NT_STATUS_ACCESS_DENIED;
1429                 }
1430                 return cli_session_setup_plaintext(cli, user, pass, workgroup);
1431         }
1432
1433         /* if the server supports extended security then use SPNEGO */
1434
1435         if (cli->capabilities & CAP_EXTENDED_SECURITY) {
1436                 ADS_STATUS status = cli_session_setup_spnego(cli, user, pass,
1437                                                              workgroup, NULL);
1438                 if (!ADS_ERR_OK(status)) {
1439                         DEBUG(3, ("SPNEGO login failed: %s\n", ads_errstr(status)));
1440                         return ads_ntstatus(status);
1441                 }
1442         } else {
1443                 NTSTATUS status;
1444
1445                 /* otherwise do a NT1 style session setup */
1446                 status = cli_session_setup_nt1(cli, user, pass, passlen,
1447                                                ntpass, ntpasslen, workgroup);
1448                 if (!NT_STATUS_IS_OK(status)) {
1449                         DEBUG(3,("cli_session_setup: NT1 session setup "
1450                                  "failed: %s\n", nt_errstr(status)));
1451                         return status;
1452                 }
1453         }
1454
1455         if (strstr(cli->server_type, "Samba")) {
1456                 cli->is_samba = True;
1457         }
1458
1459         return NT_STATUS_OK;
1460 }
1461
1462 /****************************************************************************
1463  Send a uloggoff.
1464 *****************************************************************************/
1465
1466 struct cli_ulogoff_state {
1467         struct cli_state *cli;
1468         uint16_t vwv[2];
1469 };
1470
1471 static void cli_ulogoff_done(struct tevent_req *subreq);
1472
1473 struct tevent_req *cli_ulogoff_send(TALLOC_CTX *mem_ctx,
1474                                     struct tevent_context *ev,
1475                                     struct cli_state *cli)
1476 {
1477         struct tevent_req *req, *subreq;
1478         struct cli_ulogoff_state *state;
1479
1480         req = tevent_req_create(mem_ctx, &state, struct cli_ulogoff_state);
1481         if (req == NULL) {
1482                 return NULL;
1483         }
1484         state->cli = cli;
1485
1486         SCVAL(state->vwv+0, 0, 0xFF);
1487         SCVAL(state->vwv+1, 0, 0);
1488         SSVAL(state->vwv+2, 0, 0);
1489
1490         subreq = cli_smb_send(state, ev, cli, SMBulogoffX, 0, 2, state->vwv,
1491                               0, NULL);
1492         if (tevent_req_nomem(subreq, req)) {
1493                 return tevent_req_post(req, ev);
1494         }
1495         tevent_req_set_callback(subreq, cli_ulogoff_done, req);
1496         return req;
1497 }
1498
1499 static void cli_ulogoff_done(struct tevent_req *subreq)
1500 {
1501         struct tevent_req *req = tevent_req_callback_data(
1502                 subreq, struct tevent_req);
1503         struct cli_ulogoff_state *state = tevent_req_data(
1504                 req, struct cli_ulogoff_state);
1505         NTSTATUS status;
1506
1507         status = cli_smb_recv(subreq, 0, NULL, NULL, NULL, NULL);
1508         if (!NT_STATUS_IS_OK(status)) {
1509                 tevent_req_nterror(req, status);
1510                 return;
1511         }
1512         state->cli->vuid = -1;
1513         tevent_req_done(req);
1514 }
1515
1516 NTSTATUS cli_ulogoff_recv(struct tevent_req *req)
1517 {
1518         return tevent_req_simple_recv_ntstatus(req);
1519 }
1520
1521 NTSTATUS cli_ulogoff(struct cli_state *cli)
1522 {
1523         struct tevent_context *ev;
1524         struct tevent_req *req;
1525         NTSTATUS status = NT_STATUS_NO_MEMORY;
1526
1527         if (cli_has_async_calls(cli)) {
1528                 return NT_STATUS_INVALID_PARAMETER;
1529         }
1530         ev = tevent_context_init(talloc_tos());
1531         if (ev == NULL) {
1532                 goto fail;
1533         }
1534         req = cli_ulogoff_send(ev, ev, cli);
1535         if (req == NULL) {
1536                 goto fail;
1537         }
1538         if (!tevent_req_poll_ntstatus(req, ev, &status)) {
1539                 goto fail;
1540         }
1541         status = cli_ulogoff_recv(req);
1542 fail:
1543         TALLOC_FREE(ev);
1544         if (!NT_STATUS_IS_OK(status)) {
1545                 cli_set_error(cli, status);
1546         }
1547         return status;
1548 }
1549
1550 /****************************************************************************
1551  Send a tconX.
1552 ****************************************************************************/
1553
1554 struct cli_tcon_andx_state {
1555         struct cli_state *cli;
1556         uint16_t vwv[4];
1557         struct iovec bytes;
1558 };
1559
1560 static void cli_tcon_andx_done(struct tevent_req *subreq);
1561
1562 struct tevent_req *cli_tcon_andx_create(TALLOC_CTX *mem_ctx,
1563                                         struct event_context *ev,
1564                                         struct cli_state *cli,
1565                                         const char *share, const char *dev,
1566                                         const char *pass, int passlen,
1567                                         struct tevent_req **psmbreq)
1568 {
1569         struct tevent_req *req, *subreq;
1570         struct cli_tcon_andx_state *state;
1571         fstring pword;
1572         uint16_t *vwv;
1573         char *tmp = NULL;
1574         uint8_t *bytes;
1575
1576         req = tevent_req_create(mem_ctx, &state, struct cli_tcon_andx_state);
1577         if (req == NULL) {
1578                 return NULL;
1579         }
1580         state->cli = cli;
1581         vwv = state->vwv;
1582
1583         fstrcpy(cli->share, share);
1584
1585         /* in user level security don't send a password now */
1586         if (cli->sec_mode & NEGOTIATE_SECURITY_USER_LEVEL) {
1587                 passlen = 1;
1588                 pass = "";
1589         } else if (pass == NULL) {
1590                 DEBUG(1, ("Server not using user level security and no "
1591                           "password supplied.\n"));
1592                 goto access_denied;
1593         }
1594
1595         if ((cli->sec_mode & NEGOTIATE_SECURITY_CHALLENGE_RESPONSE) &&
1596             *pass && passlen != 24) {
1597                 if (!lp_client_lanman_auth()) {
1598                         DEBUG(1, ("Server requested LANMAN password "
1599                                   "(share-level security) but "
1600                                   "'client lanman auth' is disabled\n"));
1601                         goto access_denied;
1602                 }
1603
1604                 /*
1605                  * Non-encrypted passwords - convert to DOS codepage before
1606                  * encryption.
1607                  */
1608                 passlen = 24;
1609                 SMBencrypt(pass, cli->secblob.data, (uchar *)pword);
1610         } else {
1611                 if((cli->sec_mode & (NEGOTIATE_SECURITY_USER_LEVEL
1612                                      |NEGOTIATE_SECURITY_CHALLENGE_RESPONSE))
1613                    == 0) {
1614                         if (!lp_client_plaintext_auth() && (*pass)) {
1615                                 DEBUG(1, ("Server requested plaintext "
1616                                           "password but 'client plaintext "
1617                                           "auth' is disabled\n"));
1618                                 goto access_denied;
1619                         }
1620
1621                         /*
1622                          * Non-encrypted passwords - convert to DOS codepage
1623                          * before using.
1624                          */
1625                         passlen = clistr_push(cli, pword, pass, sizeof(pword),
1626                                               STR_TERMINATE);
1627                         if (passlen == -1) {
1628                                 DEBUG(1, ("clistr_push(pword) failed\n"));
1629                                 goto access_denied;
1630                         }
1631                 } else {
1632                         if (passlen) {
1633                                 memcpy(pword, pass, passlen);
1634                         }
1635                 }
1636         }
1637
1638         SCVAL(vwv+0, 0, 0xFF);
1639         SCVAL(vwv+0, 1, 0);
1640         SSVAL(vwv+1, 0, 0);
1641         SSVAL(vwv+2, 0, TCONX_FLAG_EXTENDED_RESPONSE);
1642         SSVAL(vwv+3, 0, passlen);
1643
1644         if (passlen) {
1645                 bytes = (uint8_t *)talloc_memdup(state, pword, passlen);
1646         } else {
1647                 bytes = talloc_array(state, uint8_t, 0);
1648         }
1649
1650         /*
1651          * Add the sharename
1652          */
1653         tmp = talloc_asprintf_strupper_m(talloc_tos(), "\\\\%s\\%s",
1654                                          cli->desthost, share);
1655         if (tmp == NULL) {
1656                 TALLOC_FREE(req);
1657                 return NULL;
1658         }
1659         bytes = smb_bytes_push_str(bytes, cli_ucs2(cli), tmp, strlen(tmp)+1,
1660                                    NULL);
1661         TALLOC_FREE(tmp);
1662
1663         /*
1664          * Add the devicetype
1665          */
1666         tmp = talloc_strdup_upper(talloc_tos(), dev);
1667         if (tmp == NULL) {
1668                 TALLOC_FREE(req);
1669                 return NULL;
1670         }
1671         bytes = smb_bytes_push_str(bytes, false, tmp, strlen(tmp)+1, NULL);
1672         TALLOC_FREE(tmp);
1673
1674         if (bytes == NULL) {
1675                 TALLOC_FREE(req);
1676                 return NULL;
1677         }
1678
1679         state->bytes.iov_base = (void *)bytes;
1680         state->bytes.iov_len = talloc_get_size(bytes);
1681
1682         subreq = cli_smb_req_create(state, ev, cli, SMBtconX, 0, 4, vwv,
1683                                     1, &state->bytes);
1684         if (subreq == NULL) {
1685                 TALLOC_FREE(req);
1686                 return NULL;
1687         }
1688         tevent_req_set_callback(subreq, cli_tcon_andx_done, req);
1689         *psmbreq = subreq;
1690         return req;
1691
1692  access_denied:
1693         tevent_req_nterror(req, NT_STATUS_ACCESS_DENIED);
1694         return tevent_req_post(req, ev);
1695 }
1696
1697 struct tevent_req *cli_tcon_andx_send(TALLOC_CTX *mem_ctx,
1698                                       struct event_context *ev,
1699                                       struct cli_state *cli,
1700                                       const char *share, const char *dev,
1701                                       const char *pass, int passlen)
1702 {
1703         struct tevent_req *req, *subreq;
1704         NTSTATUS status;
1705
1706         req = cli_tcon_andx_create(mem_ctx, ev, cli, share, dev, pass, passlen,
1707                                    &subreq);
1708         if (req == NULL) {
1709                 return NULL;
1710         }
1711         status = cli_smb_req_send(subreq);
1712         if (!NT_STATUS_IS_OK(status)) {
1713                 tevent_req_nterror(req, status);
1714                 return tevent_req_post(req, ev);
1715         }
1716         return req;
1717 }
1718
1719 static void cli_tcon_andx_done(struct tevent_req *subreq)
1720 {
1721         struct tevent_req *req = tevent_req_callback_data(
1722                 subreq, struct tevent_req);
1723         struct cli_tcon_andx_state *state = tevent_req_data(
1724                 req, struct cli_tcon_andx_state);
1725         struct cli_state *cli = state->cli;
1726         char *inbuf = (char *)cli_smb_inbuf(subreq);
1727         uint8_t wct;
1728         uint16_t *vwv;
1729         uint32_t num_bytes;
1730         uint8_t *bytes;
1731         NTSTATUS status;
1732
1733         status = cli_smb_recv(subreq, 0, &wct, &vwv, &num_bytes, &bytes);
1734         if (!NT_STATUS_IS_OK(status)) {
1735                 TALLOC_FREE(subreq);
1736                 tevent_req_nterror(req, status);
1737                 return;
1738         }
1739
1740         clistr_pull(inbuf, cli->dev, bytes, sizeof(fstring), num_bytes,
1741                     STR_TERMINATE|STR_ASCII);
1742
1743         if ((cli->protocol >= PROTOCOL_NT1) && (num_bytes == 3)) {
1744                 /* almost certainly win95 - enable bug fixes */
1745                 cli->win95 = True;
1746         }
1747
1748         /*
1749          * Make sure that we have the optional support 16-bit field. WCT > 2.
1750          * Avoids issues when connecting to Win9x boxes sharing files
1751          */
1752
1753         cli->dfsroot = false;
1754
1755         if ((wct > 2) && (cli->protocol >= PROTOCOL_LANMAN2)) {
1756                 cli->dfsroot = ((SVAL(vwv+2, 0) & SMB_SHARE_IN_DFS) != 0);
1757         }
1758
1759         cli->cnum = SVAL(inbuf,smb_tid);
1760         tevent_req_done(req);
1761 }
1762
1763 NTSTATUS cli_tcon_andx_recv(struct tevent_req *req)
1764 {
1765         return tevent_req_simple_recv_ntstatus(req);
1766 }
1767
1768 NTSTATUS cli_tcon_andx(struct cli_state *cli, const char *share,
1769                        const char *dev, const char *pass, int passlen)
1770 {
1771         TALLOC_CTX *frame = talloc_stackframe();
1772         struct event_context *ev;
1773         struct tevent_req *req;
1774         NTSTATUS status = NT_STATUS_OK;
1775
1776         if (cli_has_async_calls(cli)) {
1777                 /*
1778                  * Can't use sync call while an async call is in flight
1779                  */
1780                 status = NT_STATUS_INVALID_PARAMETER;
1781                 goto fail;
1782         }
1783
1784         ev = event_context_init(frame);
1785         if (ev == NULL) {
1786                 status = NT_STATUS_NO_MEMORY;
1787                 goto fail;
1788         }
1789
1790         req = cli_tcon_andx_send(frame, ev, cli, share, dev, pass, passlen);
1791         if (req == NULL) {
1792                 status = NT_STATUS_NO_MEMORY;
1793                 goto fail;
1794         }
1795
1796         if (!tevent_req_poll(req, ev)) {
1797                 status = map_nt_error_from_unix(errno);
1798                 goto fail;
1799         }
1800
1801         status = cli_tcon_andx_recv(req);
1802  fail:
1803         TALLOC_FREE(frame);
1804         if (!NT_STATUS_IS_OK(status)) {
1805                 cli_set_error(cli, status);
1806         }
1807         return status;
1808 }
1809
1810 /****************************************************************************
1811  Send a tree disconnect.
1812 ****************************************************************************/
1813
1814 struct cli_tdis_state {
1815         struct cli_state *cli;
1816 };
1817
1818 static void cli_tdis_done(struct tevent_req *subreq);
1819
1820 struct tevent_req *cli_tdis_send(TALLOC_CTX *mem_ctx,
1821                                  struct tevent_context *ev,
1822                                  struct cli_state *cli)
1823 {
1824         struct tevent_req *req, *subreq;
1825         struct cli_tdis_state *state;
1826
1827         req = tevent_req_create(mem_ctx, &state, struct cli_tdis_state);
1828         if (req == NULL) {
1829                 return NULL;
1830         }
1831         state->cli = cli;
1832
1833         subreq = cli_smb_send(state, ev, cli, SMBtdis, 0, 0, NULL, 0, NULL);
1834         if (tevent_req_nomem(subreq, req)) {
1835                 return tevent_req_post(req, ev);
1836         }
1837         tevent_req_set_callback(subreq, cli_tdis_done, req);
1838         return req;
1839 }
1840
1841 static void cli_tdis_done(struct tevent_req *subreq)
1842 {
1843         struct tevent_req *req = tevent_req_callback_data(
1844                 subreq, struct tevent_req);
1845         struct cli_tdis_state *state = tevent_req_data(
1846                 req, struct cli_tdis_state);
1847         NTSTATUS status;
1848
1849         status = cli_smb_recv(subreq, 0, NULL, NULL, NULL, NULL);
1850         TALLOC_FREE(subreq);
1851         if (!NT_STATUS_IS_OK(status)) {
1852                 tevent_req_nterror(req, status);
1853                 return;
1854         }
1855         state->cli->cnum = -1;
1856         tevent_req_done(req);
1857 }
1858
1859 NTSTATUS cli_tdis_recv(struct tevent_req *req)
1860 {
1861         return tevent_req_simple_recv_ntstatus(req);
1862 }
1863
1864 NTSTATUS cli_tdis(struct cli_state *cli)
1865 {
1866         struct tevent_context *ev;
1867         struct tevent_req *req;
1868         NTSTATUS status = NT_STATUS_NO_MEMORY;
1869
1870         if (cli_has_async_calls(cli)) {
1871                 return NT_STATUS_INVALID_PARAMETER;
1872         }
1873         ev = tevent_context_init(talloc_tos());
1874         if (ev == NULL) {
1875                 goto fail;
1876         }
1877         req = cli_tdis_send(ev, ev, cli);
1878         if (req == NULL) {
1879                 goto fail;
1880         }
1881         if (!tevent_req_poll_ntstatus(req, ev, &status)) {
1882                 goto fail;
1883         }
1884         status = cli_tdis_recv(req);
1885 fail:
1886         TALLOC_FREE(ev);
1887         if (!NT_STATUS_IS_OK(status)) {
1888                 cli_set_error(cli, status);
1889         }
1890         return status;
1891 }
1892
1893 /****************************************************************************
1894  Send a negprot command.
1895 ****************************************************************************/
1896
1897 void cli_negprot_sendsync(struct cli_state *cli)
1898 {
1899         char *p;
1900         int numprots;
1901
1902         if (cli->protocol < PROTOCOL_NT1)
1903                 cli->use_spnego = False;
1904
1905         memset(cli->outbuf,'\0',smb_size);
1906
1907         /* setup the protocol strings */
1908         cli_set_message(cli->outbuf,0,0,True);
1909
1910         p = smb_buf(cli->outbuf);
1911         for (numprots=0; numprots < ARRAY_SIZE(prots); numprots++) {
1912                 if (prots[numprots].prot > cli->protocol) {
1913                         break;
1914                 }
1915                 *p++ = 2;
1916                 p += clistr_push(cli, p, prots[numprots].name, -1, STR_TERMINATE);
1917         }
1918
1919         SCVAL(cli->outbuf,smb_com,SMBnegprot);
1920         cli_setup_bcc(cli, p);
1921         cli_setup_packet(cli);
1922
1923         SCVAL(smb_buf(cli->outbuf),0,2);
1924
1925         cli_send_smb(cli);
1926 }
1927
1928 /****************************************************************************
1929  Send a negprot command.
1930 ****************************************************************************/
1931
1932 struct cli_negprot_state {
1933         struct cli_state *cli;
1934 };
1935
1936 static void cli_negprot_done(struct tevent_req *subreq);
1937
1938 struct tevent_req *cli_negprot_send(TALLOC_CTX *mem_ctx,
1939                                     struct event_context *ev,
1940                                     struct cli_state *cli)
1941 {
1942         struct tevent_req *req, *subreq;
1943         struct cli_negprot_state *state;
1944         uint8_t *bytes = NULL;
1945         int numprots;
1946         uint16_t cnum;
1947
1948         req = tevent_req_create(mem_ctx, &state, struct cli_negprot_state);
1949         if (req == NULL) {
1950                 return NULL;
1951         }
1952         state->cli = cli;
1953
1954         if (cli->protocol < PROTOCOL_NT1)
1955                 cli->use_spnego = False;
1956
1957         /* setup the protocol strings */
1958         for (numprots=0; numprots < ARRAY_SIZE(prots); numprots++) {
1959                 uint8_t c = 2;
1960                 if (prots[numprots].prot > cli->protocol) {
1961                         break;
1962                 }
1963                 bytes = (uint8_t *)talloc_append_blob(
1964                         state, bytes, data_blob_const(&c, sizeof(c)));
1965                 if (tevent_req_nomem(bytes, req)) {
1966                         return tevent_req_post(req, ev);
1967                 }
1968                 bytes = smb_bytes_push_str(bytes, false,
1969                                            prots[numprots].name,
1970                                            strlen(prots[numprots].name)+1,
1971                                            NULL);
1972                 if (tevent_req_nomem(bytes, req)) {
1973                         return tevent_req_post(req, ev);
1974                 }
1975         }
1976
1977         cnum = cli->cnum;
1978
1979         cli->cnum = 0;
1980         subreq = cli_smb_send(state, ev, cli, SMBnegprot, 0, 0, NULL,
1981                               talloc_get_size(bytes), bytes);
1982         cli->cnum = cnum;
1983
1984         if (tevent_req_nomem(subreq, req)) {
1985                 return tevent_req_post(req, ev);
1986         }
1987         tevent_req_set_callback(subreq, cli_negprot_done, req);
1988         return req;
1989 }
1990
1991 static void cli_negprot_done(struct tevent_req *subreq)
1992 {
1993         struct tevent_req *req = tevent_req_callback_data(
1994                 subreq, struct tevent_req);
1995         struct cli_negprot_state *state = tevent_req_data(
1996                 req, struct cli_negprot_state);
1997         struct cli_state *cli = state->cli;
1998         uint8_t wct;
1999         uint16_t *vwv;
2000         uint32_t num_bytes;
2001         uint8_t *bytes;
2002         NTSTATUS status;
2003         uint16_t protnum;
2004
2005         status = cli_smb_recv(subreq, 1, &wct, &vwv, &num_bytes, &bytes);
2006         if (!NT_STATUS_IS_OK(status)) {
2007                 TALLOC_FREE(subreq);
2008                 tevent_req_nterror(req, status);
2009                 return;
2010         }
2011
2012         protnum = SVAL(vwv, 0);
2013
2014         if ((protnum >= ARRAY_SIZE(prots))
2015             || (prots[protnum].prot > cli->protocol)) {
2016                 tevent_req_nterror(req, NT_STATUS_INVALID_NETWORK_RESPONSE);
2017                 return;
2018         }
2019
2020         cli->protocol = prots[protnum].prot;
2021
2022         if ((cli->protocol < PROTOCOL_NT1) &&
2023             client_is_signing_mandatory(cli)) {
2024                 DEBUG(0,("cli_negprot: SMB signing is mandatory and the selected protocol level doesn't support it.\n"));
2025                 tevent_req_nterror(req, NT_STATUS_ACCESS_DENIED);
2026                 return;
2027         }
2028
2029         if (cli->protocol >= PROTOCOL_NT1) {
2030                 struct timespec ts;
2031                 bool negotiated_smb_signing = false;
2032
2033                 /* NT protocol */
2034                 cli->sec_mode = CVAL(vwv + 1, 0);
2035                 cli->max_mux = SVAL(vwv + 1, 1);
2036                 cli->max_xmit = IVAL(vwv + 3, 1);
2037                 cli->sesskey = IVAL(vwv + 7, 1);
2038                 cli->serverzone = SVALS(vwv + 15, 1);
2039                 cli->serverzone *= 60;
2040                 /* this time arrives in real GMT */
2041                 ts = interpret_long_date(((char *)(vwv+11))+1);
2042                 cli->servertime = ts.tv_sec;
2043                 cli->secblob = data_blob(bytes, num_bytes);
2044                 cli->capabilities = IVAL(vwv + 9, 1);
2045                 if (cli->capabilities & CAP_RAW_MODE) {
2046                         cli->readbraw_supported = True;
2047                         cli->writebraw_supported = True;
2048                 }
2049                 /* work out if they sent us a workgroup */
2050                 if (!(cli->capabilities & CAP_EXTENDED_SECURITY) &&
2051                     smb_buflen(cli->inbuf) > 8) {
2052                         clistr_pull(cli->inbuf, cli->server_domain,
2053                                     bytes+8, sizeof(cli->server_domain),
2054                                     num_bytes-8,
2055                                     STR_UNICODE|STR_NOALIGN);
2056                 }
2057
2058                 /*
2059                  * As signing is slow we only turn it on if either the client or
2060                  * the server require it. JRA.
2061                  */
2062
2063                 if (cli->sec_mode & NEGOTIATE_SECURITY_SIGNATURES_REQUIRED) {
2064                         /* Fail if server says signing is mandatory and we don't want to support it. */
2065                         if (!client_is_signing_allowed(cli)) {
2066                                 DEBUG(0,("cli_negprot: SMB signing is mandatory and we have disabled it.\n"));
2067                                 tevent_req_nterror(req,
2068                                                    NT_STATUS_ACCESS_DENIED);
2069                                 return;
2070                         }
2071                         negotiated_smb_signing = true;
2072                 } else if (client_is_signing_mandatory(cli) && client_is_signing_allowed(cli)) {
2073                         /* Fail if client says signing is mandatory and the server doesn't support it. */
2074                         if (!(cli->sec_mode & NEGOTIATE_SECURITY_SIGNATURES_ENABLED)) {
2075                                 DEBUG(1,("cli_negprot: SMB signing is mandatory and the server doesn't support it.\n"));
2076                                 tevent_req_nterror(req,
2077                                                    NT_STATUS_ACCESS_DENIED);
2078                                 return;
2079                         }
2080                         negotiated_smb_signing = true;
2081                 } else if (cli->sec_mode & NEGOTIATE_SECURITY_SIGNATURES_ENABLED) {
2082                         negotiated_smb_signing = true;
2083                 }
2084
2085                 if (negotiated_smb_signing) {
2086                         cli_set_signing_negotiated(cli);
2087                 }
2088
2089                 if (cli->capabilities & (CAP_LARGE_READX|CAP_LARGE_WRITEX)) {
2090                         SAFE_FREE(cli->outbuf);
2091                         SAFE_FREE(cli->inbuf);
2092                         cli->outbuf = (char *)SMB_MALLOC(CLI_SAMBA_MAX_LARGE_READX_SIZE+LARGE_WRITEX_HDR_SIZE+SAFETY_MARGIN);
2093                         cli->inbuf = (char *)SMB_MALLOC(CLI_SAMBA_MAX_LARGE_READX_SIZE+LARGE_WRITEX_HDR_SIZE+SAFETY_MARGIN);
2094                         cli->bufsize = CLI_SAMBA_MAX_LARGE_READX_SIZE + LARGE_WRITEX_HDR_SIZE;
2095                 }
2096
2097         } else if (cli->protocol >= PROTOCOL_LANMAN1) {
2098                 cli->use_spnego = False;
2099                 cli->sec_mode = SVAL(vwv + 1, 0);
2100                 cli->max_xmit = SVAL(vwv + 2, 0);
2101                 cli->max_mux = SVAL(vwv + 3, 0);
2102                 cli->sesskey = IVAL(vwv + 6, 0);
2103                 cli->serverzone = SVALS(vwv + 10, 0);
2104                 cli->serverzone *= 60;
2105                 /* this time is converted to GMT by make_unix_date */
2106                 cli->servertime = cli_make_unix_date(
2107                         cli, (char *)(vwv + 8));
2108                 cli->readbraw_supported = ((SVAL(vwv + 5, 0) & 0x1) != 0);
2109                 cli->writebraw_supported = ((SVAL(vwv + 5, 0) & 0x2) != 0);
2110                 cli->secblob = data_blob(bytes, num_bytes);
2111         } else {
2112                 /* the old core protocol */
2113                 cli->use_spnego = False;
2114                 cli->sec_mode = 0;
2115                 cli->serverzone = get_time_zone(time(NULL));
2116         }
2117
2118         cli->max_xmit = MIN(cli->max_xmit, CLI_BUFFER_SIZE);
2119
2120         /* a way to force ascii SMB */
2121         if (getenv("CLI_FORCE_ASCII"))
2122                 cli->capabilities &= ~CAP_UNICODE;
2123
2124         tevent_req_done(req);
2125 }
2126
2127 NTSTATUS cli_negprot_recv(struct tevent_req *req)
2128 {
2129         return tevent_req_simple_recv_ntstatus(req);
2130 }
2131
2132 NTSTATUS cli_negprot(struct cli_state *cli)
2133 {
2134         TALLOC_CTX *frame = talloc_stackframe();
2135         struct event_context *ev;
2136         struct tevent_req *req;
2137         NTSTATUS status = NT_STATUS_OK;
2138
2139         if (cli_has_async_calls(cli)) {
2140                 /*
2141                  * Can't use sync call while an async call is in flight
2142                  */
2143                 status = NT_STATUS_INVALID_PARAMETER;
2144                 goto fail;
2145         }
2146
2147         ev = event_context_init(frame);
2148         if (ev == NULL) {
2149                 status = NT_STATUS_NO_MEMORY;
2150                 goto fail;
2151         }
2152
2153         req = cli_negprot_send(frame, ev, cli);
2154         if (req == NULL) {
2155                 status = NT_STATUS_NO_MEMORY;
2156                 goto fail;
2157         }
2158
2159         if (!tevent_req_poll(req, ev)) {
2160                 status = map_nt_error_from_unix(errno);
2161                 goto fail;
2162         }
2163
2164         status = cli_negprot_recv(req);
2165  fail:
2166         TALLOC_FREE(frame);
2167         if (!NT_STATUS_IS_OK(status)) {
2168                 cli_set_error(cli, status);
2169         }
2170         return status;
2171 }
2172
2173 /****************************************************************************
2174  Send a session request. See rfc1002.txt 4.3 and 4.3.2.
2175 ****************************************************************************/
2176
2177 bool cli_session_request(struct cli_state *cli,
2178                          struct nmb_name *calling, struct nmb_name *called)
2179 {
2180         char *p;
2181         int len = 4;
2182         char *tmp;
2183
2184         /* 445 doesn't have session request */
2185         if (cli->port == 445)
2186                 return True;
2187
2188         memcpy(&(cli->calling), calling, sizeof(*calling));
2189         memcpy(&(cli->called ), called , sizeof(*called ));
2190
2191         /* put in the destination name */
2192
2193         tmp = name_mangle(talloc_tos(), cli->called.name,
2194                           cli->called.name_type);
2195         if (tmp == NULL) {
2196                 return false;
2197         }
2198
2199         p = cli->outbuf+len;
2200         memcpy(p, tmp, name_len(tmp));
2201         len += name_len(tmp);
2202         TALLOC_FREE(tmp);
2203
2204         /* and my name */
2205
2206         tmp = name_mangle(talloc_tos(), cli->calling.name,
2207                           cli->calling.name_type);
2208         if (tmp == NULL) {
2209                 return false;
2210         }
2211
2212         p = cli->outbuf+len;
2213         memcpy(p, tmp, name_len(tmp));
2214         len += name_len(tmp);
2215         TALLOC_FREE(tmp);
2216
2217         /* send a session request (RFC 1002) */
2218         /* setup the packet length
2219          * Remove four bytes from the length count, since the length
2220          * field in the NBT Session Service header counts the number
2221          * of bytes which follow.  The cli_send_smb() function knows
2222          * about this and accounts for those four bytes.
2223          * CRH.
2224          */
2225         len -= 4;
2226         _smb_setlen(cli->outbuf,len);
2227         SCVAL(cli->outbuf,0,0x81);
2228
2229         cli_send_smb(cli);
2230         DEBUG(5,("Sent session request\n"));
2231
2232         if (!cli_receive_smb(cli))
2233                 return False;
2234
2235         if (CVAL(cli->inbuf,0) == 0x84) {
2236                 /* C. Hoch  9/14/95 Start */
2237                 /* For information, here is the response structure.
2238                  * We do the byte-twiddling to for portability.
2239                 struct RetargetResponse{
2240                 unsigned char type;
2241                 unsigned char flags;
2242                 int16 length;
2243                 int32 ip_addr;
2244                 int16 port;
2245                 };
2246                 */
2247                 uint16_t port = (CVAL(cli->inbuf,8)<<8)+CVAL(cli->inbuf,9);
2248                 struct in_addr dest_ip;
2249                 NTSTATUS status;
2250
2251                 /* SESSION RETARGET */
2252                 putip((char *)&dest_ip,cli->inbuf+4);
2253                 in_addr_to_sockaddr_storage(&cli->dest_ss, dest_ip);
2254
2255                 status = open_socket_out(&cli->dest_ss, port,
2256                                          LONG_CONNECT_TIMEOUT, &cli->fd);
2257                 if (!NT_STATUS_IS_OK(status)) {
2258                         return False;
2259                 }
2260
2261                 DEBUG(3,("Retargeted\n"));
2262
2263                 set_socket_options(cli->fd, lp_socket_options());
2264
2265                 /* Try again */
2266                 {
2267                         static int depth;
2268                         bool ret;
2269                         if (depth > 4) {
2270                                 DEBUG(0,("Retarget recursion - failing\n"));
2271                                 return False;
2272                         }
2273                         depth++;
2274                         ret = cli_session_request(cli, calling, called);
2275                         depth--;
2276                         return ret;
2277                 }
2278         } /* C. Hoch 9/14/95 End */
2279
2280         if (CVAL(cli->inbuf,0) != 0x82) {
2281                 /* This is the wrong place to put the error... JRA. */
2282                 cli->rap_error = CVAL(cli->inbuf,4);
2283                 return False;
2284         }
2285         return(True);
2286 }
2287
2288 struct fd_struct {
2289         int fd;
2290 };
2291
2292 static void smb_sock_connected(struct tevent_req *req)
2293 {
2294         struct fd_struct *pfd = tevent_req_callback_data(
2295                 req, struct fd_struct);
2296         int fd;
2297         NTSTATUS status;
2298
2299         status = open_socket_out_defer_recv(req, &fd);
2300         if (NT_STATUS_IS_OK(status)) {
2301                 pfd->fd = fd;
2302         }
2303 }
2304
2305 static NTSTATUS open_smb_socket(const struct sockaddr_storage *pss,
2306                                 uint16_t *port, int timeout, int *pfd)
2307 {
2308         struct event_context *ev;
2309         struct tevent_req *r139, *r445;
2310         struct fd_struct *fd139, *fd445;
2311         NTSTATUS status = NT_STATUS_NO_MEMORY;
2312
2313         if (*port != 0) {
2314                 return open_socket_out(pss, *port, timeout, pfd);
2315         }
2316
2317         ev = event_context_init(talloc_tos());
2318         if (ev == NULL) {
2319                 return NT_STATUS_NO_MEMORY;
2320         }
2321
2322         fd139 = talloc(ev, struct fd_struct);
2323         if (fd139 == NULL) {
2324                 goto done;
2325         }
2326         fd139->fd = -1;
2327
2328         fd445 = talloc(ev, struct fd_struct);
2329         if (fd445 == NULL) {
2330                 goto done;
2331         }
2332         fd445->fd = -1;
2333
2334         r445 = open_socket_out_defer_send(ev, ev, timeval_set(0, 0),
2335                                           pss, 445, timeout);
2336         r139 = open_socket_out_defer_send(ev, ev, timeval_set(0, 3000),
2337                                           pss, 139, timeout);
2338         if ((r445 == NULL) || (r139 == NULL)) {
2339                 goto done;
2340         }
2341         tevent_req_set_callback(r445, smb_sock_connected, fd445);
2342         tevent_req_set_callback(r139, smb_sock_connected, fd139);
2343
2344         while ((fd445->fd == -1) && (fd139->fd == -1)
2345                && (tevent_req_is_in_progress(r139)
2346                    || tevent_req_is_in_progress(r445))) {
2347                 event_loop_once(ev);
2348         }
2349
2350         if ((fd139->fd != -1) && (fd445->fd != -1)) {
2351                 close(fd139->fd);
2352                 fd139->fd = -1;
2353         }
2354
2355         if (fd445->fd != -1) {
2356                 *port = 445;
2357                 *pfd = fd445->fd;
2358                 status = NT_STATUS_OK;
2359                 goto done;
2360         }
2361         if (fd139->fd != -1) {
2362                 *port = 139;
2363                 *pfd = fd139->fd;
2364                 status = NT_STATUS_OK;
2365                 goto done;
2366         }
2367
2368         status = open_socket_out_defer_recv(r445, &fd445->fd);
2369  done:
2370         TALLOC_FREE(ev);
2371         return status;
2372 }
2373
2374 /****************************************************************************
2375  Open the client sockets.
2376 ****************************************************************************/
2377
2378 NTSTATUS cli_connect(struct cli_state *cli,
2379                 const char *host,
2380                 struct sockaddr_storage *dest_ss)
2381
2382 {
2383         int name_type = 0x20;
2384         TALLOC_CTX *frame = talloc_stackframe();
2385         unsigned int num_addrs = 0;
2386         unsigned int i = 0;
2387         struct sockaddr_storage *ss_arr = NULL;
2388         char *p = NULL;
2389
2390         /* reasonable default hostname */
2391         if (!host) {
2392                 host = STAR_SMBSERVER;
2393         }
2394
2395         fstrcpy(cli->desthost, host);
2396
2397         /* allow hostnames of the form NAME#xx and do a netbios lookup */
2398         if ((p = strchr(cli->desthost, '#'))) {
2399                 name_type = strtol(p+1, NULL, 16);
2400                 *p = 0;
2401         }
2402
2403         if (!dest_ss || is_zero_addr((struct sockaddr *)dest_ss)) {
2404                 NTSTATUS status =resolve_name_list(frame,
2405                                         cli->desthost,
2406                                         name_type,
2407                                         &ss_arr,
2408                                         &num_addrs);
2409                 if (!NT_STATUS_IS_OK(status)) {
2410                         TALLOC_FREE(frame);
2411                         return NT_STATUS_BAD_NETWORK_NAME;
2412                 }
2413         } else {
2414                 num_addrs = 1;
2415                 ss_arr = TALLOC_P(frame, struct sockaddr_storage);
2416                 if (!ss_arr) {
2417                         TALLOC_FREE(frame);
2418                         return NT_STATUS_NO_MEMORY;
2419                 }
2420                 *ss_arr = *dest_ss;
2421         }
2422
2423         for (i = 0; i < num_addrs; i++) {
2424                 cli->dest_ss = ss_arr[i];
2425                 if (getenv("LIBSMB_PROG")) {
2426                         cli->fd = sock_exec(getenv("LIBSMB_PROG"));
2427                 } else {
2428                         uint16_t port = cli->port;
2429                         NTSTATUS status;
2430                         status = open_smb_socket(&cli->dest_ss, &port,
2431                                                  cli->timeout, &cli->fd);
2432                         if (NT_STATUS_IS_OK(status)) {
2433                                 cli->port = port;
2434                         }
2435                 }
2436                 if (cli->fd == -1) {
2437                         char addr[INET6_ADDRSTRLEN];
2438                         print_sockaddr(addr, sizeof(addr), &ss_arr[i]);
2439                         DEBUG(2,("Error connecting to %s (%s)\n",
2440                                  dest_ss?addr:host,strerror(errno)));
2441                 } else {
2442                         /* Exit from loop on first connection. */
2443                         break;
2444                 }
2445         }
2446
2447         if (cli->fd == -1) {
2448                 TALLOC_FREE(frame);
2449                 return map_nt_error_from_unix(errno);
2450         }
2451
2452         if (dest_ss) {
2453                 *dest_ss = cli->dest_ss;
2454         }
2455
2456         set_socket_options(cli->fd, lp_socket_options());
2457
2458         TALLOC_FREE(frame);
2459         return NT_STATUS_OK;
2460 }
2461
2462 /**
2463    establishes a connection to after the negprot.
2464    @param output_cli A fully initialised cli structure, non-null only on success
2465    @param dest_host The netbios name of the remote host
2466    @param dest_ss (optional) The the destination IP, NULL for name based lookup
2467    @param port (optional) The destination port (0 for default)
2468    @param retry bool. Did this connection fail with a retryable error ?
2469
2470 */
2471 NTSTATUS cli_start_connection(struct cli_state **output_cli,
2472                               const char *my_name,
2473                               const char *dest_host,
2474                               struct sockaddr_storage *dest_ss, int port,
2475                               int signing_state, int flags,
2476                               bool *retry)
2477 {
2478         NTSTATUS nt_status;
2479         struct nmb_name calling;
2480         struct nmb_name called;
2481         struct cli_state *cli;
2482         struct sockaddr_storage ss;
2483
2484         if (retry)
2485                 *retry = False;
2486
2487         if (!my_name)
2488                 my_name = global_myname();
2489
2490         if (!(cli = cli_initialise_ex(signing_state))) {
2491                 return NT_STATUS_NO_MEMORY;
2492         }
2493
2494         make_nmb_name(&calling, my_name, 0x0);
2495         make_nmb_name(&called , dest_host, 0x20);
2496
2497         cli_set_port(cli, port);
2498         cli_set_timeout(cli, 10000); /* 10 seconds. */
2499
2500         if (dest_ss) {
2501                 ss = *dest_ss;
2502         } else {
2503                 zero_sockaddr(&ss);
2504         }
2505
2506 again:
2507
2508         DEBUG(3,("Connecting to host=%s\n", dest_host));
2509
2510         nt_status = cli_connect(cli, dest_host, &ss);
2511         if (!NT_STATUS_IS_OK(nt_status)) {
2512                 char addr[INET6_ADDRSTRLEN];
2513                 print_sockaddr(addr, sizeof(addr), &ss);
2514                 DEBUG(1,("cli_start_connection: failed to connect to %s (%s). Error %s\n",
2515                          nmb_namestr(&called), addr, nt_errstr(nt_status) ));
2516                 cli_shutdown(cli);
2517                 return nt_status;
2518         }
2519
2520         if (retry)
2521                 *retry = True;
2522
2523         if (!cli_session_request(cli, &calling, &called)) {
2524                 char *p;
2525                 DEBUG(1,("session request to %s failed (%s)\n",
2526                          called.name, cli_errstr(cli)));
2527                 if ((p=strchr(called.name, '.')) && !is_ipaddress(called.name)) {
2528                         *p = 0;
2529                         goto again;
2530                 }
2531                 if (strcmp(called.name, STAR_SMBSERVER)) {
2532                         make_nmb_name(&called , STAR_SMBSERVER, 0x20);
2533                         goto again;
2534                 }
2535                 return NT_STATUS_BAD_NETWORK_NAME;
2536         }
2537
2538         if (flags & CLI_FULL_CONNECTION_DONT_SPNEGO)
2539                 cli->use_spnego = False;
2540         else if (flags & CLI_FULL_CONNECTION_USE_KERBEROS)
2541                 cli->use_kerberos = True;
2542
2543         if ((flags & CLI_FULL_CONNECTION_FALLBACK_AFTER_KERBEROS) &&
2544              cli->use_kerberos) {
2545                 cli->fallback_after_kerberos = true;
2546         }
2547
2548         nt_status = cli_negprot(cli);
2549         if (!NT_STATUS_IS_OK(nt_status)) {
2550                 DEBUG(1, ("failed negprot: %s\n", nt_errstr(nt_status)));
2551                 cli_shutdown(cli);
2552                 return nt_status;
2553         }
2554
2555         *output_cli = cli;
2556         return NT_STATUS_OK;
2557 }
2558
2559
2560 /**
2561    establishes a connection right up to doing tconX, password specified.
2562    @param output_cli A fully initialised cli structure, non-null only on success
2563    @param dest_host The netbios name of the remote host
2564    @param dest_ip (optional) The the destination IP, NULL for name based lookup
2565    @param port (optional) The destination port (0 for default)
2566    @param service (optional) The share to make the connection to.  Should be 'unqualified' in any way.
2567    @param service_type The 'type' of serivice.
2568    @param user Username, unix string
2569    @param domain User's domain
2570    @param password User's password, unencrypted unix string.
2571    @param retry bool. Did this connection fail with a retryable error ?
2572 */
2573
2574 NTSTATUS cli_full_connection(struct cli_state **output_cli,
2575                              const char *my_name,
2576                              const char *dest_host,
2577                              struct sockaddr_storage *dest_ss, int port,
2578                              const char *service, const char *service_type,
2579                              const char *user, const char *domain,
2580                              const char *password, int flags,
2581                              int signing_state,
2582                              bool *retry)
2583 {
2584         NTSTATUS nt_status;
2585         struct cli_state *cli = NULL;
2586         int pw_len = password ? strlen(password)+1 : 0;
2587
2588         *output_cli = NULL;
2589
2590         if (password == NULL) {
2591                 password = "";
2592         }
2593
2594         nt_status = cli_start_connection(&cli, my_name, dest_host,
2595                                          dest_ss, port, signing_state,
2596                                          flags, retry);
2597
2598         if (!NT_STATUS_IS_OK(nt_status)) {
2599                 return nt_status;
2600         }
2601
2602         cli->use_oplocks = ((flags & CLI_FULL_CONNECTION_OPLOCKS) != 0);
2603         cli->use_level_II_oplocks =
2604                 ((flags & CLI_FULL_CONNECTION_LEVEL_II_OPLOCKS) != 0);
2605
2606         nt_status = cli_session_setup(cli, user, password, pw_len, password,
2607                                       pw_len, domain);
2608         if (!NT_STATUS_IS_OK(nt_status)) {
2609
2610                 if (!(flags & CLI_FULL_CONNECTION_ANONYMOUS_FALLBACK)) {
2611                         DEBUG(1,("failed session setup with %s\n",
2612                                  nt_errstr(nt_status)));
2613                         cli_shutdown(cli);
2614                         return nt_status;
2615                 }
2616
2617                 nt_status = cli_session_setup(cli, "", "", 0, "", 0, domain);
2618                 if (!NT_STATUS_IS_OK(nt_status)) {
2619                         DEBUG(1,("anonymous failed session setup with %s\n",
2620                                  nt_errstr(nt_status)));
2621                         cli_shutdown(cli);
2622                         return nt_status;
2623                 }
2624         }
2625
2626         if (service) {
2627                 nt_status = cli_tcon_andx(cli, service, service_type, password,
2628                                           pw_len);
2629                 if (!NT_STATUS_IS_OK(nt_status)) {
2630                         DEBUG(1,("failed tcon_X with %s\n", nt_errstr(nt_status)));
2631                         cli_shutdown(cli);
2632                         if (NT_STATUS_IS_OK(nt_status)) {
2633                                 nt_status = NT_STATUS_UNSUCCESSFUL;
2634                         }
2635                         return nt_status;
2636                 }
2637         }
2638
2639         nt_status = cli_init_creds(cli, user, domain, password);
2640         if (!NT_STATUS_IS_OK(nt_status)) {
2641                 cli_shutdown(cli);
2642                 return nt_status;
2643         }
2644
2645         *output_cli = cli;
2646         return NT_STATUS_OK;
2647 }
2648
2649 /****************************************************************************
2650  Attempt a NetBIOS session request, falling back to *SMBSERVER if needed.
2651 ****************************************************************************/
2652
2653 bool attempt_netbios_session_request(struct cli_state **ppcli, const char *srchost, const char *desthost,
2654                                      struct sockaddr_storage *pdest_ss)
2655 {
2656         struct nmb_name calling, called;
2657
2658         make_nmb_name(&calling, srchost, 0x0);
2659
2660         /*
2661          * If the called name is an IP address
2662          * then use *SMBSERVER immediately.
2663          */
2664
2665         if(is_ipaddress(desthost)) {
2666                 make_nmb_name(&called, STAR_SMBSERVER, 0x20);
2667         } else {
2668                 make_nmb_name(&called, desthost, 0x20);
2669         }
2670
2671         if (!cli_session_request(*ppcli, &calling, &called)) {
2672                 NTSTATUS status;
2673                 struct nmb_name smbservername;
2674
2675                 make_nmb_name(&smbservername, STAR_SMBSERVER, 0x20);
2676
2677                 /*
2678                  * If the name wasn't *SMBSERVER then
2679                  * try with *SMBSERVER if the first name fails.
2680                  */
2681
2682                 if (nmb_name_equal(&called, &smbservername)) {
2683
2684                         /*
2685                          * The name used was *SMBSERVER, don't bother with another name.
2686                          */
2687
2688                         DEBUG(0,("attempt_netbios_session_request: %s rejected the session for name *SMBSERVER \
2689 with error %s.\n", desthost, cli_errstr(*ppcli) ));
2690                         return False;
2691                 }
2692
2693                 /* Try again... */
2694                 cli_shutdown(*ppcli);
2695
2696                 *ppcli = cli_initialise();
2697                 if (!*ppcli) {
2698                         /* Out of memory... */
2699                         return False;
2700                 }
2701
2702                 status = cli_connect(*ppcli, desthost, pdest_ss);
2703                 if (!NT_STATUS_IS_OK(status) ||
2704                                 !cli_session_request(*ppcli, &calling, &smbservername)) {
2705                         DEBUG(0,("attempt_netbios_session_request: %s rejected the session for \
2706 name *SMBSERVER with error %s\n", desthost, cli_errstr(*ppcli) ));
2707                         return False;
2708                 }
2709         }
2710
2711         return True;
2712 }
2713
2714 /****************************************************************************
2715  Send an old style tcon.
2716 ****************************************************************************/
2717 NTSTATUS cli_raw_tcon(struct cli_state *cli,
2718                       const char *service, const char *pass, const char *dev,
2719                       uint16 *max_xmit, uint16 *tid)
2720 {
2721         char *p;
2722
2723         if (!lp_client_plaintext_auth() && (*pass)) {
2724                 DEBUG(1, ("Server requested plaintext password but 'client "
2725                           "plaintext auth' is disabled\n"));
2726                 return NT_STATUS_ACCESS_DENIED;
2727         }
2728
2729         memset(cli->outbuf,'\0',smb_size);
2730         memset(cli->inbuf,'\0',smb_size);
2731
2732         cli_set_message(cli->outbuf, 0, 0, True);
2733         SCVAL(cli->outbuf,smb_com,SMBtcon);
2734         cli_setup_packet(cli);
2735
2736         p = smb_buf(cli->outbuf);
2737         *p++ = 4; p += clistr_push(cli, p, service, -1, STR_TERMINATE | STR_NOALIGN);
2738         *p++ = 4; p += clistr_push(cli, p, pass, -1, STR_TERMINATE | STR_NOALIGN);
2739         *p++ = 4; p += clistr_push(cli, p, dev, -1, STR_TERMINATE | STR_NOALIGN);
2740
2741         cli_setup_bcc(cli, p);
2742
2743         cli_send_smb(cli);
2744         if (!cli_receive_smb(cli)) {
2745                 return NT_STATUS_UNEXPECTED_NETWORK_ERROR;
2746         }
2747
2748         if (cli_is_error(cli)) {
2749                 return cli_nt_error(cli);
2750         }
2751
2752         *max_xmit = SVAL(cli->inbuf, smb_vwv0);
2753         *tid = SVAL(cli->inbuf, smb_vwv1);
2754
2755         return NT_STATUS_OK;
2756 }
2757
2758 /* Return a cli_state pointing at the IPC$ share for the given server */
2759
2760 struct cli_state *get_ipc_connect(char *server,
2761                                 struct sockaddr_storage *server_ss,
2762                                 const struct user_auth_info *user_info)
2763 {
2764         struct cli_state *cli;
2765         NTSTATUS nt_status;
2766         uint32_t flags = CLI_FULL_CONNECTION_ANONYMOUS_FALLBACK;
2767
2768         if (user_info->use_kerberos) {
2769                 flags |= CLI_FULL_CONNECTION_USE_KERBEROS;
2770         }
2771
2772         nt_status = cli_full_connection(&cli, NULL, server, server_ss, 0, "IPC$", "IPC",
2773                                         user_info->username ? user_info->username : "",
2774                                         lp_workgroup(),
2775                                         user_info->password ? user_info->password : "",
2776                                         flags,
2777                                         Undefined, NULL);
2778
2779         if (NT_STATUS_IS_OK(nt_status)) {
2780                 return cli;
2781         } else if (is_ipaddress(server)) {
2782             /* windows 9* needs a correct NMB name for connections */
2783             fstring remote_name;
2784
2785             if (name_status_find("*", 0, 0, server_ss, remote_name)) {
2786                 cli = get_ipc_connect(remote_name, server_ss, user_info);
2787                 if (cli)
2788                     return cli;
2789             }
2790         }
2791         return NULL;
2792 }
2793
2794 /*
2795  * Given the IP address of a master browser on the network, return its
2796  * workgroup and connect to it.
2797  *
2798  * This function is provided to allow additional processing beyond what
2799  * get_ipc_connect_master_ip_bcast() does, e.g. to retrieve the list of master
2800  * browsers and obtain each master browsers' list of domains (in case the
2801  * first master browser is recently on the network and has not yet
2802  * synchronized with other master browsers and therefore does not yet have the
2803  * entire network browse list)
2804  */
2805
2806 struct cli_state *get_ipc_connect_master_ip(TALLOC_CTX *ctx,
2807                                 struct ip_service *mb_ip,
2808                                 const struct user_auth_info *user_info,
2809                                 char **pp_workgroup_out)
2810 {
2811         char addr[INET6_ADDRSTRLEN];
2812         fstring name;
2813         struct cli_state *cli;
2814         struct sockaddr_storage server_ss;
2815
2816         *pp_workgroup_out = NULL;
2817
2818         print_sockaddr(addr, sizeof(addr), &mb_ip->ss);
2819         DEBUG(99, ("Looking up name of master browser %s\n",
2820                    addr));
2821
2822         /*
2823          * Do a name status query to find out the name of the master browser.
2824          * We use <01><02>__MSBROWSE__<02>#01 if *#00 fails because a domain
2825          * master browser will not respond to a wildcard query (or, at least,
2826          * an NT4 server acting as the domain master browser will not).
2827          *
2828          * We might be able to use ONLY the query on MSBROWSE, but that's not
2829          * yet been tested with all Windows versions, so until it is, leave
2830          * the original wildcard query as the first choice and fall back to
2831          * MSBROWSE if the wildcard query fails.
2832          */
2833         if (!name_status_find("*", 0, 0x1d, &mb_ip->ss, name) &&
2834             !name_status_find(MSBROWSE, 1, 0x1d, &mb_ip->ss, name)) {
2835
2836                 DEBUG(99, ("Could not retrieve name status for %s\n",
2837                            addr));
2838                 return NULL;
2839         }
2840
2841         if (!find_master_ip(name, &server_ss)) {
2842                 DEBUG(99, ("Could not find master ip for %s\n", name));
2843                 return NULL;
2844         }
2845
2846         *pp_workgroup_out = talloc_strdup(ctx, name);
2847
2848         DEBUG(4, ("found master browser %s, %s\n", name, addr));
2849
2850         print_sockaddr(addr, sizeof(addr), &server_ss);
2851         cli = get_ipc_connect(addr, &server_ss, user_info);
2852
2853         return cli;
2854 }
2855
2856 /*
2857  * Return the IP address and workgroup of a master browser on the network, and
2858  * connect to it.
2859  */
2860
2861 struct cli_state *get_ipc_connect_master_ip_bcast(TALLOC_CTX *ctx,
2862                                         const struct user_auth_info *user_info,
2863                                         char **pp_workgroup_out)
2864 {
2865         struct ip_service *ip_list;
2866         struct cli_state *cli;
2867         int i, count;
2868
2869         *pp_workgroup_out = NULL;
2870
2871         DEBUG(99, ("Do broadcast lookup for workgroups on local network\n"));
2872
2873         /* Go looking for workgroups by broadcasting on the local network */
2874
2875         if (!NT_STATUS_IS_OK(name_resolve_bcast(MSBROWSE, 1, &ip_list,
2876                                                 &count))) {
2877                 DEBUG(99, ("No master browsers responded\n"));
2878                 return False;
2879         }
2880
2881         for (i = 0; i < count; i++) {
2882                 char addr[INET6_ADDRSTRLEN];
2883                 print_sockaddr(addr, sizeof(addr), &ip_list[i].ss);
2884                 DEBUG(99, ("Found master browser %s\n", addr));
2885
2886                 cli = get_ipc_connect_master_ip(ctx, &ip_list[i],
2887                                 user_info, pp_workgroup_out);
2888                 if (cli)
2889                         return(cli);
2890         }
2891
2892         return NULL;
2893 }