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 bool cli_ulogoff(struct cli_state *cli)
1467 {
1468         memset(cli->outbuf,'\0',smb_size);
1469         cli_set_message(cli->outbuf,2,0,True);
1470         SCVAL(cli->outbuf,smb_com,SMBulogoffX);
1471         cli_setup_packet(cli);
1472         SSVAL(cli->outbuf,smb_vwv0,0xFF);
1473         SSVAL(cli->outbuf,smb_vwv2,0);  /* no additional info */
1474
1475         cli_send_smb(cli);
1476         if (!cli_receive_smb(cli))
1477                 return False;
1478
1479         if (cli_is_error(cli)) {
1480                 return False;
1481         }
1482
1483         cli->vuid = -1;
1484         return True;
1485 }
1486
1487 /****************************************************************************
1488  Send a tconX.
1489 ****************************************************************************/
1490
1491 struct cli_tcon_andx_state {
1492         struct cli_state *cli;
1493         uint16_t vwv[4];
1494         struct iovec bytes;
1495 };
1496
1497 static void cli_tcon_andx_done(struct tevent_req *subreq);
1498
1499 struct tevent_req *cli_tcon_andx_create(TALLOC_CTX *mem_ctx,
1500                                         struct event_context *ev,
1501                                         struct cli_state *cli,
1502                                         const char *share, const char *dev,
1503                                         const char *pass, int passlen,
1504                                         struct tevent_req **psmbreq)
1505 {
1506         struct tevent_req *req, *subreq;
1507         struct cli_tcon_andx_state *state;
1508         fstring pword;
1509         uint16_t *vwv;
1510         char *tmp = NULL;
1511         uint8_t *bytes;
1512
1513         req = tevent_req_create(mem_ctx, &state, struct cli_tcon_andx_state);
1514         if (req == NULL) {
1515                 return NULL;
1516         }
1517         state->cli = cli;
1518         vwv = state->vwv;
1519
1520         fstrcpy(cli->share, share);
1521
1522         /* in user level security don't send a password now */
1523         if (cli->sec_mode & NEGOTIATE_SECURITY_USER_LEVEL) {
1524                 passlen = 1;
1525                 pass = "";
1526         } else if (pass == NULL) {
1527                 DEBUG(1, ("Server not using user level security and no "
1528                           "password supplied.\n"));
1529                 goto access_denied;
1530         }
1531
1532         if ((cli->sec_mode & NEGOTIATE_SECURITY_CHALLENGE_RESPONSE) &&
1533             *pass && passlen != 24) {
1534                 if (!lp_client_lanman_auth()) {
1535                         DEBUG(1, ("Server requested LANMAN password "
1536                                   "(share-level security) but "
1537                                   "'client lanman auth' is disabled\n"));
1538                         goto access_denied;
1539                 }
1540
1541                 /*
1542                  * Non-encrypted passwords - convert to DOS codepage before
1543                  * encryption.
1544                  */
1545                 passlen = 24;
1546                 SMBencrypt(pass, cli->secblob.data, (uchar *)pword);
1547         } else {
1548                 if((cli->sec_mode & (NEGOTIATE_SECURITY_USER_LEVEL
1549                                      |NEGOTIATE_SECURITY_CHALLENGE_RESPONSE))
1550                    == 0) {
1551                         if (!lp_client_plaintext_auth() && (*pass)) {
1552                                 DEBUG(1, ("Server requested plaintext "
1553                                           "password but 'client plaintext "
1554                                           "auth' is disabled\n"));
1555                                 goto access_denied;
1556                         }
1557
1558                         /*
1559                          * Non-encrypted passwords - convert to DOS codepage
1560                          * before using.
1561                          */
1562                         passlen = clistr_push(cli, pword, pass, sizeof(pword),
1563                                               STR_TERMINATE);
1564                         if (passlen == -1) {
1565                                 DEBUG(1, ("clistr_push(pword) failed\n"));
1566                                 goto access_denied;
1567                         }
1568                 } else {
1569                         if (passlen) {
1570                                 memcpy(pword, pass, passlen);
1571                         }
1572                 }
1573         }
1574
1575         SCVAL(vwv+0, 0, 0xFF);
1576         SCVAL(vwv+0, 1, 0);
1577         SSVAL(vwv+1, 0, 0);
1578         SSVAL(vwv+2, 0, TCONX_FLAG_EXTENDED_RESPONSE);
1579         SSVAL(vwv+3, 0, passlen);
1580
1581         if (passlen) {
1582                 bytes = (uint8_t *)talloc_memdup(state, pword, passlen);
1583         } else {
1584                 bytes = talloc_array(state, uint8_t, 0);
1585         }
1586
1587         /*
1588          * Add the sharename
1589          */
1590         tmp = talloc_asprintf_strupper_m(talloc_tos(), "\\\\%s\\%s",
1591                                          cli->desthost, share);
1592         if (tmp == NULL) {
1593                 TALLOC_FREE(req);
1594                 return NULL;
1595         }
1596         bytes = smb_bytes_push_str(bytes, cli_ucs2(cli), tmp, strlen(tmp)+1,
1597                                    NULL);
1598         TALLOC_FREE(tmp);
1599
1600         /*
1601          * Add the devicetype
1602          */
1603         tmp = talloc_strdup_upper(talloc_tos(), dev);
1604         if (tmp == NULL) {
1605                 TALLOC_FREE(req);
1606                 return NULL;
1607         }
1608         bytes = smb_bytes_push_str(bytes, false, tmp, strlen(tmp)+1, NULL);
1609         TALLOC_FREE(tmp);
1610
1611         if (bytes == NULL) {
1612                 TALLOC_FREE(req);
1613                 return NULL;
1614         }
1615
1616         state->bytes.iov_base = (void *)bytes;
1617         state->bytes.iov_len = talloc_get_size(bytes);
1618
1619         subreq = cli_smb_req_create(state, ev, cli, SMBtconX, 0, 4, vwv,
1620                                     1, &state->bytes);
1621         if (subreq == NULL) {
1622                 TALLOC_FREE(req);
1623                 return NULL;
1624         }
1625         tevent_req_set_callback(subreq, cli_tcon_andx_done, req);
1626         *psmbreq = subreq;
1627         return req;
1628
1629  access_denied:
1630         tevent_req_nterror(req, NT_STATUS_ACCESS_DENIED);
1631         return tevent_req_post(req, ev);
1632 }
1633
1634 struct tevent_req *cli_tcon_andx_send(TALLOC_CTX *mem_ctx,
1635                                       struct event_context *ev,
1636                                       struct cli_state *cli,
1637                                       const char *share, const char *dev,
1638                                       const char *pass, int passlen)
1639 {
1640         struct tevent_req *req, *subreq;
1641         NTSTATUS status;
1642
1643         req = cli_tcon_andx_create(mem_ctx, ev, cli, share, dev, pass, passlen,
1644                                    &subreq);
1645         if (req == NULL) {
1646                 return NULL;
1647         }
1648         status = cli_smb_req_send(subreq);
1649         if (!NT_STATUS_IS_OK(status)) {
1650                 tevent_req_nterror(req, status);
1651                 return tevent_req_post(req, ev);
1652         }
1653         return req;
1654 }
1655
1656 static void cli_tcon_andx_done(struct tevent_req *subreq)
1657 {
1658         struct tevent_req *req = tevent_req_callback_data(
1659                 subreq, struct tevent_req);
1660         struct cli_tcon_andx_state *state = tevent_req_data(
1661                 req, struct cli_tcon_andx_state);
1662         struct cli_state *cli = state->cli;
1663         char *inbuf = (char *)cli_smb_inbuf(subreq);
1664         uint8_t wct;
1665         uint16_t *vwv;
1666         uint32_t num_bytes;
1667         uint8_t *bytes;
1668         NTSTATUS status;
1669
1670         status = cli_smb_recv(subreq, 0, &wct, &vwv, &num_bytes, &bytes);
1671         if (!NT_STATUS_IS_OK(status)) {
1672                 TALLOC_FREE(subreq);
1673                 tevent_req_nterror(req, status);
1674                 return;
1675         }
1676
1677         clistr_pull(inbuf, cli->dev, bytes, sizeof(fstring), num_bytes,
1678                     STR_TERMINATE|STR_ASCII);
1679
1680         if ((cli->protocol >= PROTOCOL_NT1) && (num_bytes == 3)) {
1681                 /* almost certainly win95 - enable bug fixes */
1682                 cli->win95 = True;
1683         }
1684
1685         /*
1686          * Make sure that we have the optional support 16-bit field. WCT > 2.
1687          * Avoids issues when connecting to Win9x boxes sharing files
1688          */
1689
1690         cli->dfsroot = false;
1691
1692         if ((wct > 2) && (cli->protocol >= PROTOCOL_LANMAN2)) {
1693                 cli->dfsroot = ((SVAL(vwv+2, 0) & SMB_SHARE_IN_DFS) != 0);
1694         }
1695
1696         cli->cnum = SVAL(inbuf,smb_tid);
1697         tevent_req_done(req);
1698 }
1699
1700 NTSTATUS cli_tcon_andx_recv(struct tevent_req *req)
1701 {
1702         return tevent_req_simple_recv_ntstatus(req);
1703 }
1704
1705 NTSTATUS cli_tcon_andx(struct cli_state *cli, const char *share,
1706                        const char *dev, const char *pass, int passlen)
1707 {
1708         TALLOC_CTX *frame = talloc_stackframe();
1709         struct event_context *ev;
1710         struct tevent_req *req;
1711         NTSTATUS status = NT_STATUS_OK;
1712
1713         if (cli_has_async_calls(cli)) {
1714                 /*
1715                  * Can't use sync call while an async call is in flight
1716                  */
1717                 status = NT_STATUS_INVALID_PARAMETER;
1718                 goto fail;
1719         }
1720
1721         ev = event_context_init(frame);
1722         if (ev == NULL) {
1723                 status = NT_STATUS_NO_MEMORY;
1724                 goto fail;
1725         }
1726
1727         req = cli_tcon_andx_send(frame, ev, cli, share, dev, pass, passlen);
1728         if (req == NULL) {
1729                 status = NT_STATUS_NO_MEMORY;
1730                 goto fail;
1731         }
1732
1733         if (!tevent_req_poll(req, ev)) {
1734                 status = map_nt_error_from_unix(errno);
1735                 goto fail;
1736         }
1737
1738         status = cli_tcon_andx_recv(req);
1739  fail:
1740         TALLOC_FREE(frame);
1741         if (!NT_STATUS_IS_OK(status)) {
1742                 cli_set_error(cli, status);
1743         }
1744         return status;
1745 }
1746
1747 /****************************************************************************
1748  Send a tree disconnect.
1749 ****************************************************************************/
1750
1751 struct cli_tdis_state {
1752         struct cli_state *cli;
1753 };
1754
1755 static void cli_tdis_done(struct tevent_req *subreq);
1756
1757 struct tevent_req *cli_tdis_send(TALLOC_CTX *mem_ctx,
1758                                  struct tevent_context *ev,
1759                                  struct cli_state *cli)
1760 {
1761         struct tevent_req *req, *subreq;
1762         struct cli_tdis_state *state;
1763
1764         req = tevent_req_create(mem_ctx, &state, struct cli_tdis_state);
1765         if (req == NULL) {
1766                 return NULL;
1767         }
1768         state->cli = cli;
1769
1770         subreq = cli_smb_send(state, ev, cli, SMBtdis, 0, 0, NULL, 0, NULL);
1771         if (tevent_req_nomem(subreq, req)) {
1772                 return tevent_req_post(req, ev);
1773         }
1774         tevent_req_set_callback(subreq, cli_tdis_done, req);
1775         return req;
1776 }
1777
1778 static void cli_tdis_done(struct tevent_req *subreq)
1779 {
1780         struct tevent_req *req = tevent_req_callback_data(
1781                 subreq, struct tevent_req);
1782         struct cli_tdis_state *state = tevent_req_data(
1783                 req, struct cli_tdis_state);
1784         NTSTATUS status;
1785
1786         status = cli_smb_recv(subreq, 0, NULL, NULL, NULL, NULL);
1787         TALLOC_FREE(subreq);
1788         if (!NT_STATUS_IS_OK(status)) {
1789                 tevent_req_nterror(req, status);
1790                 return;
1791         }
1792         state->cli->cnum = -1;
1793         tevent_req_done(req);
1794 }
1795
1796 NTSTATUS cli_tdis_recv(struct tevent_req *req)
1797 {
1798         return tevent_req_simple_recv_ntstatus(req);
1799 }
1800
1801 NTSTATUS cli_tdis(struct cli_state *cli)
1802 {
1803         struct tevent_context *ev;
1804         struct tevent_req *req;
1805         NTSTATUS status = NT_STATUS_NO_MEMORY;
1806
1807         if (cli_has_async_calls(cli)) {
1808                 return NT_STATUS_INVALID_PARAMETER;
1809         }
1810         ev = tevent_context_init(talloc_tos());
1811         if (ev == NULL) {
1812                 goto fail;
1813         }
1814         req = cli_tdis_send(ev, ev, cli);
1815         if (req == NULL) {
1816                 goto fail;
1817         }
1818         if (!tevent_req_poll_ntstatus(req, ev, &status)) {
1819                 goto fail;
1820         }
1821         status = cli_tdis_recv(req);
1822 fail:
1823         TALLOC_FREE(ev);
1824         if (!NT_STATUS_IS_OK(status)) {
1825                 cli_set_error(cli, status);
1826         }
1827         return status;
1828 }
1829
1830 /****************************************************************************
1831  Send a negprot command.
1832 ****************************************************************************/
1833
1834 void cli_negprot_sendsync(struct cli_state *cli)
1835 {
1836         char *p;
1837         int numprots;
1838
1839         if (cli->protocol < PROTOCOL_NT1)
1840                 cli->use_spnego = False;
1841
1842         memset(cli->outbuf,'\0',smb_size);
1843
1844         /* setup the protocol strings */
1845         cli_set_message(cli->outbuf,0,0,True);
1846
1847         p = smb_buf(cli->outbuf);
1848         for (numprots=0; numprots < ARRAY_SIZE(prots); numprots++) {
1849                 if (prots[numprots].prot > cli->protocol) {
1850                         break;
1851                 }
1852                 *p++ = 2;
1853                 p += clistr_push(cli, p, prots[numprots].name, -1, STR_TERMINATE);
1854         }
1855
1856         SCVAL(cli->outbuf,smb_com,SMBnegprot);
1857         cli_setup_bcc(cli, p);
1858         cli_setup_packet(cli);
1859
1860         SCVAL(smb_buf(cli->outbuf),0,2);
1861
1862         cli_send_smb(cli);
1863 }
1864
1865 /****************************************************************************
1866  Send a negprot command.
1867 ****************************************************************************/
1868
1869 struct cli_negprot_state {
1870         struct cli_state *cli;
1871 };
1872
1873 static void cli_negprot_done(struct tevent_req *subreq);
1874
1875 struct tevent_req *cli_negprot_send(TALLOC_CTX *mem_ctx,
1876                                     struct event_context *ev,
1877                                     struct cli_state *cli)
1878 {
1879         struct tevent_req *req, *subreq;
1880         struct cli_negprot_state *state;
1881         uint8_t *bytes = NULL;
1882         int numprots;
1883         uint16_t cnum;
1884
1885         req = tevent_req_create(mem_ctx, &state, struct cli_negprot_state);
1886         if (req == NULL) {
1887                 return NULL;
1888         }
1889         state->cli = cli;
1890
1891         if (cli->protocol < PROTOCOL_NT1)
1892                 cli->use_spnego = False;
1893
1894         /* setup the protocol strings */
1895         for (numprots=0; numprots < ARRAY_SIZE(prots); numprots++) {
1896                 uint8_t c = 2;
1897                 if (prots[numprots].prot > cli->protocol) {
1898                         break;
1899                 }
1900                 bytes = (uint8_t *)talloc_append_blob(
1901                         state, bytes, data_blob_const(&c, sizeof(c)));
1902                 if (tevent_req_nomem(bytes, req)) {
1903                         return tevent_req_post(req, ev);
1904                 }
1905                 bytes = smb_bytes_push_str(bytes, false,
1906                                            prots[numprots].name,
1907                                            strlen(prots[numprots].name)+1,
1908                                            NULL);
1909                 if (tevent_req_nomem(bytes, req)) {
1910                         return tevent_req_post(req, ev);
1911                 }
1912         }
1913
1914         cnum = cli->cnum;
1915
1916         cli->cnum = 0;
1917         subreq = cli_smb_send(state, ev, cli, SMBnegprot, 0, 0, NULL,
1918                               talloc_get_size(bytes), bytes);
1919         cli->cnum = cnum;
1920
1921         if (tevent_req_nomem(subreq, req)) {
1922                 return tevent_req_post(req, ev);
1923         }
1924         tevent_req_set_callback(subreq, cli_negprot_done, req);
1925         return req;
1926 }
1927
1928 static void cli_negprot_done(struct tevent_req *subreq)
1929 {
1930         struct tevent_req *req = tevent_req_callback_data(
1931                 subreq, struct tevent_req);
1932         struct cli_negprot_state *state = tevent_req_data(
1933                 req, struct cli_negprot_state);
1934         struct cli_state *cli = state->cli;
1935         uint8_t wct;
1936         uint16_t *vwv;
1937         uint32_t num_bytes;
1938         uint8_t *bytes;
1939         NTSTATUS status;
1940         uint16_t protnum;
1941
1942         status = cli_smb_recv(subreq, 1, &wct, &vwv, &num_bytes, &bytes);
1943         if (!NT_STATUS_IS_OK(status)) {
1944                 TALLOC_FREE(subreq);
1945                 tevent_req_nterror(req, status);
1946                 return;
1947         }
1948
1949         protnum = SVAL(vwv, 0);
1950
1951         if ((protnum >= ARRAY_SIZE(prots))
1952             || (prots[protnum].prot > cli->protocol)) {
1953                 tevent_req_nterror(req, NT_STATUS_INVALID_NETWORK_RESPONSE);
1954                 return;
1955         }
1956
1957         cli->protocol = prots[protnum].prot;
1958
1959         if ((cli->protocol < PROTOCOL_NT1) &&
1960             client_is_signing_mandatory(cli)) {
1961                 DEBUG(0,("cli_negprot: SMB signing is mandatory and the selected protocol level doesn't support it.\n"));
1962                 tevent_req_nterror(req, NT_STATUS_ACCESS_DENIED);
1963                 return;
1964         }
1965
1966         if (cli->protocol >= PROTOCOL_NT1) {
1967                 struct timespec ts;
1968                 bool negotiated_smb_signing = false;
1969
1970                 /* NT protocol */
1971                 cli->sec_mode = CVAL(vwv + 1, 0);
1972                 cli->max_mux = SVAL(vwv + 1, 1);
1973                 cli->max_xmit = IVAL(vwv + 3, 1);
1974                 cli->sesskey = IVAL(vwv + 7, 1);
1975                 cli->serverzone = SVALS(vwv + 15, 1);
1976                 cli->serverzone *= 60;
1977                 /* this time arrives in real GMT */
1978                 ts = interpret_long_date(((char *)(vwv+11))+1);
1979                 cli->servertime = ts.tv_sec;
1980                 cli->secblob = data_blob(bytes, num_bytes);
1981                 cli->capabilities = IVAL(vwv + 9, 1);
1982                 if (cli->capabilities & CAP_RAW_MODE) {
1983                         cli->readbraw_supported = True;
1984                         cli->writebraw_supported = True;
1985                 }
1986                 /* work out if they sent us a workgroup */
1987                 if (!(cli->capabilities & CAP_EXTENDED_SECURITY) &&
1988                     smb_buflen(cli->inbuf) > 8) {
1989                         clistr_pull(cli->inbuf, cli->server_domain,
1990                                     bytes+8, sizeof(cli->server_domain),
1991                                     num_bytes-8,
1992                                     STR_UNICODE|STR_NOALIGN);
1993                 }
1994
1995                 /*
1996                  * As signing is slow we only turn it on if either the client or
1997                  * the server require it. JRA.
1998                  */
1999
2000                 if (cli->sec_mode & NEGOTIATE_SECURITY_SIGNATURES_REQUIRED) {
2001                         /* Fail if server says signing is mandatory and we don't want to support it. */
2002                         if (!client_is_signing_allowed(cli)) {
2003                                 DEBUG(0,("cli_negprot: SMB signing is mandatory and we have disabled it.\n"));
2004                                 tevent_req_nterror(req,
2005                                                    NT_STATUS_ACCESS_DENIED);
2006                                 return;
2007                         }
2008                         negotiated_smb_signing = true;
2009                 } else if (client_is_signing_mandatory(cli) && client_is_signing_allowed(cli)) {
2010                         /* Fail if client says signing is mandatory and the server doesn't support it. */
2011                         if (!(cli->sec_mode & NEGOTIATE_SECURITY_SIGNATURES_ENABLED)) {
2012                                 DEBUG(1,("cli_negprot: SMB signing is mandatory and the server doesn't support it.\n"));
2013                                 tevent_req_nterror(req,
2014                                                    NT_STATUS_ACCESS_DENIED);
2015                                 return;
2016                         }
2017                         negotiated_smb_signing = true;
2018                 } else if (cli->sec_mode & NEGOTIATE_SECURITY_SIGNATURES_ENABLED) {
2019                         negotiated_smb_signing = true;
2020                 }
2021
2022                 if (negotiated_smb_signing) {
2023                         cli_set_signing_negotiated(cli);
2024                 }
2025
2026                 if (cli->capabilities & (CAP_LARGE_READX|CAP_LARGE_WRITEX)) {
2027                         SAFE_FREE(cli->outbuf);
2028                         SAFE_FREE(cli->inbuf);
2029                         cli->outbuf = (char *)SMB_MALLOC(CLI_SAMBA_MAX_LARGE_READX_SIZE+LARGE_WRITEX_HDR_SIZE+SAFETY_MARGIN);
2030                         cli->inbuf = (char *)SMB_MALLOC(CLI_SAMBA_MAX_LARGE_READX_SIZE+LARGE_WRITEX_HDR_SIZE+SAFETY_MARGIN);
2031                         cli->bufsize = CLI_SAMBA_MAX_LARGE_READX_SIZE + LARGE_WRITEX_HDR_SIZE;
2032                 }
2033
2034         } else if (cli->protocol >= PROTOCOL_LANMAN1) {
2035                 cli->use_spnego = False;
2036                 cli->sec_mode = SVAL(vwv + 1, 0);
2037                 cli->max_xmit = SVAL(vwv + 2, 0);
2038                 cli->max_mux = SVAL(vwv + 3, 0);
2039                 cli->sesskey = IVAL(vwv + 6, 0);
2040                 cli->serverzone = SVALS(vwv + 10, 0);
2041                 cli->serverzone *= 60;
2042                 /* this time is converted to GMT by make_unix_date */
2043                 cli->servertime = cli_make_unix_date(
2044                         cli, (char *)(vwv + 8));
2045                 cli->readbraw_supported = ((SVAL(vwv + 5, 0) & 0x1) != 0);
2046                 cli->writebraw_supported = ((SVAL(vwv + 5, 0) & 0x2) != 0);
2047                 cli->secblob = data_blob(bytes, num_bytes);
2048         } else {
2049                 /* the old core protocol */
2050                 cli->use_spnego = False;
2051                 cli->sec_mode = 0;
2052                 cli->serverzone = get_time_zone(time(NULL));
2053         }
2054
2055         cli->max_xmit = MIN(cli->max_xmit, CLI_BUFFER_SIZE);
2056
2057         /* a way to force ascii SMB */
2058         if (getenv("CLI_FORCE_ASCII"))
2059                 cli->capabilities &= ~CAP_UNICODE;
2060
2061         tevent_req_done(req);
2062 }
2063
2064 NTSTATUS cli_negprot_recv(struct tevent_req *req)
2065 {
2066         return tevent_req_simple_recv_ntstatus(req);
2067 }
2068
2069 NTSTATUS cli_negprot(struct cli_state *cli)
2070 {
2071         TALLOC_CTX *frame = talloc_stackframe();
2072         struct event_context *ev;
2073         struct tevent_req *req;
2074         NTSTATUS status = NT_STATUS_OK;
2075
2076         if (cli_has_async_calls(cli)) {
2077                 /*
2078                  * Can't use sync call while an async call is in flight
2079                  */
2080                 status = NT_STATUS_INVALID_PARAMETER;
2081                 goto fail;
2082         }
2083
2084         ev = event_context_init(frame);
2085         if (ev == NULL) {
2086                 status = NT_STATUS_NO_MEMORY;
2087                 goto fail;
2088         }
2089
2090         req = cli_negprot_send(frame, ev, cli);
2091         if (req == NULL) {
2092                 status = NT_STATUS_NO_MEMORY;
2093                 goto fail;
2094         }
2095
2096         if (!tevent_req_poll(req, ev)) {
2097                 status = map_nt_error_from_unix(errno);
2098                 goto fail;
2099         }
2100
2101         status = cli_negprot_recv(req);
2102  fail:
2103         TALLOC_FREE(frame);
2104         if (!NT_STATUS_IS_OK(status)) {
2105                 cli_set_error(cli, status);
2106         }
2107         return status;
2108 }
2109
2110 /****************************************************************************
2111  Send a session request. See rfc1002.txt 4.3 and 4.3.2.
2112 ****************************************************************************/
2113
2114 bool cli_session_request(struct cli_state *cli,
2115                          struct nmb_name *calling, struct nmb_name *called)
2116 {
2117         char *p;
2118         int len = 4;
2119         char *tmp;
2120
2121         /* 445 doesn't have session request */
2122         if (cli->port == 445)
2123                 return True;
2124
2125         memcpy(&(cli->calling), calling, sizeof(*calling));
2126         memcpy(&(cli->called ), called , sizeof(*called ));
2127
2128         /* put in the destination name */
2129
2130         tmp = name_mangle(talloc_tos(), cli->called.name,
2131                           cli->called.name_type);
2132         if (tmp == NULL) {
2133                 return false;
2134         }
2135
2136         p = cli->outbuf+len;
2137         memcpy(p, tmp, name_len(tmp));
2138         len += name_len(tmp);
2139         TALLOC_FREE(tmp);
2140
2141         /* and my name */
2142
2143         tmp = name_mangle(talloc_tos(), cli->calling.name,
2144                           cli->calling.name_type);
2145         if (tmp == NULL) {
2146                 return false;
2147         }
2148
2149         p = cli->outbuf+len;
2150         memcpy(p, tmp, name_len(tmp));
2151         len += name_len(tmp);
2152         TALLOC_FREE(tmp);
2153
2154         /* send a session request (RFC 1002) */
2155         /* setup the packet length
2156          * Remove four bytes from the length count, since the length
2157          * field in the NBT Session Service header counts the number
2158          * of bytes which follow.  The cli_send_smb() function knows
2159          * about this and accounts for those four bytes.
2160          * CRH.
2161          */
2162         len -= 4;
2163         _smb_setlen(cli->outbuf,len);
2164         SCVAL(cli->outbuf,0,0x81);
2165
2166         cli_send_smb(cli);
2167         DEBUG(5,("Sent session request\n"));
2168
2169         if (!cli_receive_smb(cli))
2170                 return False;
2171
2172         if (CVAL(cli->inbuf,0) == 0x84) {
2173                 /* C. Hoch  9/14/95 Start */
2174                 /* For information, here is the response structure.
2175                  * We do the byte-twiddling to for portability.
2176                 struct RetargetResponse{
2177                 unsigned char type;
2178                 unsigned char flags;
2179                 int16 length;
2180                 int32 ip_addr;
2181                 int16 port;
2182                 };
2183                 */
2184                 uint16_t port = (CVAL(cli->inbuf,8)<<8)+CVAL(cli->inbuf,9);
2185                 struct in_addr dest_ip;
2186                 NTSTATUS status;
2187
2188                 /* SESSION RETARGET */
2189                 putip((char *)&dest_ip,cli->inbuf+4);
2190                 in_addr_to_sockaddr_storage(&cli->dest_ss, dest_ip);
2191
2192                 status = open_socket_out(&cli->dest_ss, port,
2193                                          LONG_CONNECT_TIMEOUT, &cli->fd);
2194                 if (!NT_STATUS_IS_OK(status)) {
2195                         return False;
2196                 }
2197
2198                 DEBUG(3,("Retargeted\n"));
2199
2200                 set_socket_options(cli->fd, lp_socket_options());
2201
2202                 /* Try again */
2203                 {
2204                         static int depth;
2205                         bool ret;
2206                         if (depth > 4) {
2207                                 DEBUG(0,("Retarget recursion - failing\n"));
2208                                 return False;
2209                         }
2210                         depth++;
2211                         ret = cli_session_request(cli, calling, called);
2212                         depth--;
2213                         return ret;
2214                 }
2215         } /* C. Hoch 9/14/95 End */
2216
2217         if (CVAL(cli->inbuf,0) != 0x82) {
2218                 /* This is the wrong place to put the error... JRA. */
2219                 cli->rap_error = CVAL(cli->inbuf,4);
2220                 return False;
2221         }
2222         return(True);
2223 }
2224
2225 struct fd_struct {
2226         int fd;
2227 };
2228
2229 static void smb_sock_connected(struct tevent_req *req)
2230 {
2231         struct fd_struct *pfd = tevent_req_callback_data(
2232                 req, struct fd_struct);
2233         int fd;
2234         NTSTATUS status;
2235
2236         status = open_socket_out_defer_recv(req, &fd);
2237         if (NT_STATUS_IS_OK(status)) {
2238                 pfd->fd = fd;
2239         }
2240 }
2241
2242 static NTSTATUS open_smb_socket(const struct sockaddr_storage *pss,
2243                                 uint16_t *port, int timeout, int *pfd)
2244 {
2245         struct event_context *ev;
2246         struct tevent_req *r139, *r445;
2247         struct fd_struct *fd139, *fd445;
2248         NTSTATUS status = NT_STATUS_NO_MEMORY;
2249
2250         if (*port != 0) {
2251                 return open_socket_out(pss, *port, timeout, pfd);
2252         }
2253
2254         ev = event_context_init(talloc_tos());
2255         if (ev == NULL) {
2256                 return NT_STATUS_NO_MEMORY;
2257         }
2258
2259         fd139 = talloc(ev, struct fd_struct);
2260         if (fd139 == NULL) {
2261                 goto done;
2262         }
2263         fd139->fd = -1;
2264
2265         fd445 = talloc(ev, struct fd_struct);
2266         if (fd445 == NULL) {
2267                 goto done;
2268         }
2269         fd445->fd = -1;
2270
2271         r445 = open_socket_out_defer_send(ev, ev, timeval_set(0, 0),
2272                                           pss, 445, timeout);
2273         r139 = open_socket_out_defer_send(ev, ev, timeval_set(0, 3000),
2274                                           pss, 139, timeout);
2275         if ((r445 == NULL) || (r139 == NULL)) {
2276                 goto done;
2277         }
2278         tevent_req_set_callback(r445, smb_sock_connected, fd445);
2279         tevent_req_set_callback(r139, smb_sock_connected, fd139);
2280
2281         while ((fd445->fd == -1) && (fd139->fd == -1)
2282                && (tevent_req_is_in_progress(r139)
2283                    || tevent_req_is_in_progress(r445))) {
2284                 event_loop_once(ev);
2285         }
2286
2287         if ((fd139->fd != -1) && (fd445->fd != -1)) {
2288                 close(fd139->fd);
2289                 fd139->fd = -1;
2290         }
2291
2292         if (fd445->fd != -1) {
2293                 *port = 445;
2294                 *pfd = fd445->fd;
2295                 status = NT_STATUS_OK;
2296                 goto done;
2297         }
2298         if (fd139->fd != -1) {
2299                 *port = 139;
2300                 *pfd = fd139->fd;
2301                 status = NT_STATUS_OK;
2302                 goto done;
2303         }
2304
2305         status = open_socket_out_defer_recv(r445, &fd445->fd);
2306  done:
2307         TALLOC_FREE(ev);
2308         return status;
2309 }
2310
2311 /****************************************************************************
2312  Open the client sockets.
2313 ****************************************************************************/
2314
2315 NTSTATUS cli_connect(struct cli_state *cli,
2316                 const char *host,
2317                 struct sockaddr_storage *dest_ss)
2318
2319 {
2320         int name_type = 0x20;
2321         TALLOC_CTX *frame = talloc_stackframe();
2322         unsigned int num_addrs = 0;
2323         unsigned int i = 0;
2324         struct sockaddr_storage *ss_arr = NULL;
2325         char *p = NULL;
2326
2327         /* reasonable default hostname */
2328         if (!host) {
2329                 host = STAR_SMBSERVER;
2330         }
2331
2332         fstrcpy(cli->desthost, host);
2333
2334         /* allow hostnames of the form NAME#xx and do a netbios lookup */
2335         if ((p = strchr(cli->desthost, '#'))) {
2336                 name_type = strtol(p+1, NULL, 16);
2337                 *p = 0;
2338         }
2339
2340         if (!dest_ss || is_zero_addr((struct sockaddr *)dest_ss)) {
2341                 NTSTATUS status =resolve_name_list(frame,
2342                                         cli->desthost,
2343                                         name_type,
2344                                         &ss_arr,
2345                                         &num_addrs);
2346                 if (!NT_STATUS_IS_OK(status)) {
2347                         TALLOC_FREE(frame);
2348                         return NT_STATUS_BAD_NETWORK_NAME;
2349                 }
2350         } else {
2351                 num_addrs = 1;
2352                 ss_arr = TALLOC_P(frame, struct sockaddr_storage);
2353                 if (!ss_arr) {
2354                         TALLOC_FREE(frame);
2355                         return NT_STATUS_NO_MEMORY;
2356                 }
2357                 *ss_arr = *dest_ss;
2358         }
2359
2360         for (i = 0; i < num_addrs; i++) {
2361                 cli->dest_ss = ss_arr[i];
2362                 if (getenv("LIBSMB_PROG")) {
2363                         cli->fd = sock_exec(getenv("LIBSMB_PROG"));
2364                 } else {
2365                         uint16_t port = cli->port;
2366                         NTSTATUS status;
2367                         status = open_smb_socket(&cli->dest_ss, &port,
2368                                                  cli->timeout, &cli->fd);
2369                         if (NT_STATUS_IS_OK(status)) {
2370                                 cli->port = port;
2371                         }
2372                 }
2373                 if (cli->fd == -1) {
2374                         char addr[INET6_ADDRSTRLEN];
2375                         print_sockaddr(addr, sizeof(addr), &ss_arr[i]);
2376                         DEBUG(2,("Error connecting to %s (%s)\n",
2377                                  dest_ss?addr:host,strerror(errno)));
2378                 } else {
2379                         /* Exit from loop on first connection. */
2380                         break;
2381                 }
2382         }
2383
2384         if (cli->fd == -1) {
2385                 TALLOC_FREE(frame);
2386                 return map_nt_error_from_unix(errno);
2387         }
2388
2389         if (dest_ss) {
2390                 *dest_ss = cli->dest_ss;
2391         }
2392
2393         set_socket_options(cli->fd, lp_socket_options());
2394
2395         TALLOC_FREE(frame);
2396         return NT_STATUS_OK;
2397 }
2398
2399 /**
2400    establishes a connection to after the negprot.
2401    @param output_cli A fully initialised cli structure, non-null only on success
2402    @param dest_host The netbios name of the remote host
2403    @param dest_ss (optional) The the destination IP, NULL for name based lookup
2404    @param port (optional) The destination port (0 for default)
2405    @param retry bool. Did this connection fail with a retryable error ?
2406
2407 */
2408 NTSTATUS cli_start_connection(struct cli_state **output_cli,
2409                               const char *my_name,
2410                               const char *dest_host,
2411                               struct sockaddr_storage *dest_ss, int port,
2412                               int signing_state, int flags,
2413                               bool *retry)
2414 {
2415         NTSTATUS nt_status;
2416         struct nmb_name calling;
2417         struct nmb_name called;
2418         struct cli_state *cli;
2419         struct sockaddr_storage ss;
2420
2421         if (retry)
2422                 *retry = False;
2423
2424         if (!my_name)
2425                 my_name = global_myname();
2426
2427         if (!(cli = cli_initialise_ex(signing_state))) {
2428                 return NT_STATUS_NO_MEMORY;
2429         }
2430
2431         make_nmb_name(&calling, my_name, 0x0);
2432         make_nmb_name(&called , dest_host, 0x20);
2433
2434         cli_set_port(cli, port);
2435         cli_set_timeout(cli, 10000); /* 10 seconds. */
2436
2437         if (dest_ss) {
2438                 ss = *dest_ss;
2439         } else {
2440                 zero_sockaddr(&ss);
2441         }
2442
2443 again:
2444
2445         DEBUG(3,("Connecting to host=%s\n", dest_host));
2446
2447         nt_status = cli_connect(cli, dest_host, &ss);
2448         if (!NT_STATUS_IS_OK(nt_status)) {
2449                 char addr[INET6_ADDRSTRLEN];
2450                 print_sockaddr(addr, sizeof(addr), &ss);
2451                 DEBUG(1,("cli_start_connection: failed to connect to %s (%s). Error %s\n",
2452                          nmb_namestr(&called), addr, nt_errstr(nt_status) ));
2453                 cli_shutdown(cli);
2454                 return nt_status;
2455         }
2456
2457         if (retry)
2458                 *retry = True;
2459
2460         if (!cli_session_request(cli, &calling, &called)) {
2461                 char *p;
2462                 DEBUG(1,("session request to %s failed (%s)\n",
2463                          called.name, cli_errstr(cli)));
2464                 if ((p=strchr(called.name, '.')) && !is_ipaddress(called.name)) {
2465                         *p = 0;
2466                         goto again;
2467                 }
2468                 if (strcmp(called.name, STAR_SMBSERVER)) {
2469                         make_nmb_name(&called , STAR_SMBSERVER, 0x20);
2470                         goto again;
2471                 }
2472                 return NT_STATUS_BAD_NETWORK_NAME;
2473         }
2474
2475         if (flags & CLI_FULL_CONNECTION_DONT_SPNEGO)
2476                 cli->use_spnego = False;
2477         else if (flags & CLI_FULL_CONNECTION_USE_KERBEROS)
2478                 cli->use_kerberos = True;
2479
2480         if ((flags & CLI_FULL_CONNECTION_FALLBACK_AFTER_KERBEROS) &&
2481              cli->use_kerberos) {
2482                 cli->fallback_after_kerberos = true;
2483         }
2484
2485         nt_status = cli_negprot(cli);
2486         if (!NT_STATUS_IS_OK(nt_status)) {
2487                 DEBUG(1, ("failed negprot: %s\n", nt_errstr(nt_status)));
2488                 cli_shutdown(cli);
2489                 return nt_status;
2490         }
2491
2492         *output_cli = cli;
2493         return NT_STATUS_OK;
2494 }
2495
2496
2497 /**
2498    establishes a connection right up to doing tconX, password specified.
2499    @param output_cli A fully initialised cli structure, non-null only on success
2500    @param dest_host The netbios name of the remote host
2501    @param dest_ip (optional) The the destination IP, NULL for name based lookup
2502    @param port (optional) The destination port (0 for default)
2503    @param service (optional) The share to make the connection to.  Should be 'unqualified' in any way.
2504    @param service_type The 'type' of serivice.
2505    @param user Username, unix string
2506    @param domain User's domain
2507    @param password User's password, unencrypted unix string.
2508    @param retry bool. Did this connection fail with a retryable error ?
2509 */
2510
2511 NTSTATUS cli_full_connection(struct cli_state **output_cli,
2512                              const char *my_name,
2513                              const char *dest_host,
2514                              struct sockaddr_storage *dest_ss, int port,
2515                              const char *service, const char *service_type,
2516                              const char *user, const char *domain,
2517                              const char *password, int flags,
2518                              int signing_state,
2519                              bool *retry)
2520 {
2521         NTSTATUS nt_status;
2522         struct cli_state *cli = NULL;
2523         int pw_len = password ? strlen(password)+1 : 0;
2524
2525         *output_cli = NULL;
2526
2527         if (password == NULL) {
2528                 password = "";
2529         }
2530
2531         nt_status = cli_start_connection(&cli, my_name, dest_host,
2532                                          dest_ss, port, signing_state,
2533                                          flags, retry);
2534
2535         if (!NT_STATUS_IS_OK(nt_status)) {
2536                 return nt_status;
2537         }
2538
2539         cli->use_oplocks = ((flags & CLI_FULL_CONNECTION_OPLOCKS) != 0);
2540         cli->use_level_II_oplocks =
2541                 ((flags & CLI_FULL_CONNECTION_LEVEL_II_OPLOCKS) != 0);
2542
2543         nt_status = cli_session_setup(cli, user, password, pw_len, password,
2544                                       pw_len, domain);
2545         if (!NT_STATUS_IS_OK(nt_status)) {
2546
2547                 if (!(flags & CLI_FULL_CONNECTION_ANONYMOUS_FALLBACK)) {
2548                         DEBUG(1,("failed session setup with %s\n",
2549                                  nt_errstr(nt_status)));
2550                         cli_shutdown(cli);
2551                         return nt_status;
2552                 }
2553
2554                 nt_status = cli_session_setup(cli, "", "", 0, "", 0, domain);
2555                 if (!NT_STATUS_IS_OK(nt_status)) {
2556                         DEBUG(1,("anonymous failed session setup with %s\n",
2557                                  nt_errstr(nt_status)));
2558                         cli_shutdown(cli);
2559                         return nt_status;
2560                 }
2561         }
2562
2563         if (service) {
2564                 nt_status = cli_tcon_andx(cli, service, service_type, password,
2565                                           pw_len);
2566                 if (!NT_STATUS_IS_OK(nt_status)) {
2567                         DEBUG(1,("failed tcon_X with %s\n", nt_errstr(nt_status)));
2568                         cli_shutdown(cli);
2569                         if (NT_STATUS_IS_OK(nt_status)) {
2570                                 nt_status = NT_STATUS_UNSUCCESSFUL;
2571                         }
2572                         return nt_status;
2573                 }
2574         }
2575
2576         nt_status = cli_init_creds(cli, user, domain, password);
2577         if (!NT_STATUS_IS_OK(nt_status)) {
2578                 cli_shutdown(cli);
2579                 return nt_status;
2580         }
2581
2582         *output_cli = cli;
2583         return NT_STATUS_OK;
2584 }
2585
2586 /****************************************************************************
2587  Attempt a NetBIOS session request, falling back to *SMBSERVER if needed.
2588 ****************************************************************************/
2589
2590 bool attempt_netbios_session_request(struct cli_state **ppcli, const char *srchost, const char *desthost,
2591                                      struct sockaddr_storage *pdest_ss)
2592 {
2593         struct nmb_name calling, called;
2594
2595         make_nmb_name(&calling, srchost, 0x0);
2596
2597         /*
2598          * If the called name is an IP address
2599          * then use *SMBSERVER immediately.
2600          */
2601
2602         if(is_ipaddress(desthost)) {
2603                 make_nmb_name(&called, STAR_SMBSERVER, 0x20);
2604         } else {
2605                 make_nmb_name(&called, desthost, 0x20);
2606         }
2607
2608         if (!cli_session_request(*ppcli, &calling, &called)) {
2609                 NTSTATUS status;
2610                 struct nmb_name smbservername;
2611
2612                 make_nmb_name(&smbservername, STAR_SMBSERVER, 0x20);
2613
2614                 /*
2615                  * If the name wasn't *SMBSERVER then
2616                  * try with *SMBSERVER if the first name fails.
2617                  */
2618
2619                 if (nmb_name_equal(&called, &smbservername)) {
2620
2621                         /*
2622                          * The name used was *SMBSERVER, don't bother with another name.
2623                          */
2624
2625                         DEBUG(0,("attempt_netbios_session_request: %s rejected the session for name *SMBSERVER \
2626 with error %s.\n", desthost, cli_errstr(*ppcli) ));
2627                         return False;
2628                 }
2629
2630                 /* Try again... */
2631                 cli_shutdown(*ppcli);
2632
2633                 *ppcli = cli_initialise();
2634                 if (!*ppcli) {
2635                         /* Out of memory... */
2636                         return False;
2637                 }
2638
2639                 status = cli_connect(*ppcli, desthost, pdest_ss);
2640                 if (!NT_STATUS_IS_OK(status) ||
2641                                 !cli_session_request(*ppcli, &calling, &smbservername)) {
2642                         DEBUG(0,("attempt_netbios_session_request: %s rejected the session for \
2643 name *SMBSERVER with error %s\n", desthost, cli_errstr(*ppcli) ));
2644                         return False;
2645                 }
2646         }
2647
2648         return True;
2649 }
2650
2651 /****************************************************************************
2652  Send an old style tcon.
2653 ****************************************************************************/
2654 NTSTATUS cli_raw_tcon(struct cli_state *cli,
2655                       const char *service, const char *pass, const char *dev,
2656                       uint16 *max_xmit, uint16 *tid)
2657 {
2658         char *p;
2659
2660         if (!lp_client_plaintext_auth() && (*pass)) {
2661                 DEBUG(1, ("Server requested plaintext password but 'client "
2662                           "plaintext auth' is disabled\n"));
2663                 return NT_STATUS_ACCESS_DENIED;
2664         }
2665
2666         memset(cli->outbuf,'\0',smb_size);
2667         memset(cli->inbuf,'\0',smb_size);
2668
2669         cli_set_message(cli->outbuf, 0, 0, True);
2670         SCVAL(cli->outbuf,smb_com,SMBtcon);
2671         cli_setup_packet(cli);
2672
2673         p = smb_buf(cli->outbuf);
2674         *p++ = 4; p += clistr_push(cli, p, service, -1, STR_TERMINATE | STR_NOALIGN);
2675         *p++ = 4; p += clistr_push(cli, p, pass, -1, STR_TERMINATE | STR_NOALIGN);
2676         *p++ = 4; p += clistr_push(cli, p, dev, -1, STR_TERMINATE | STR_NOALIGN);
2677
2678         cli_setup_bcc(cli, p);
2679
2680         cli_send_smb(cli);
2681         if (!cli_receive_smb(cli)) {
2682                 return NT_STATUS_UNEXPECTED_NETWORK_ERROR;
2683         }
2684
2685         if (cli_is_error(cli)) {
2686                 return cli_nt_error(cli);
2687         }
2688
2689         *max_xmit = SVAL(cli->inbuf, smb_vwv0);
2690         *tid = SVAL(cli->inbuf, smb_vwv1);
2691
2692         return NT_STATUS_OK;
2693 }
2694
2695 /* Return a cli_state pointing at the IPC$ share for the given server */
2696
2697 struct cli_state *get_ipc_connect(char *server,
2698                                 struct sockaddr_storage *server_ss,
2699                                 const struct user_auth_info *user_info)
2700 {
2701         struct cli_state *cli;
2702         NTSTATUS nt_status;
2703         uint32_t flags = CLI_FULL_CONNECTION_ANONYMOUS_FALLBACK;
2704
2705         if (user_info->use_kerberos) {
2706                 flags |= CLI_FULL_CONNECTION_USE_KERBEROS;
2707         }
2708
2709         nt_status = cli_full_connection(&cli, NULL, server, server_ss, 0, "IPC$", "IPC",
2710                                         user_info->username ? user_info->username : "",
2711                                         lp_workgroup(),
2712                                         user_info->password ? user_info->password : "",
2713                                         flags,
2714                                         Undefined, NULL);
2715
2716         if (NT_STATUS_IS_OK(nt_status)) {
2717                 return cli;
2718         } else if (is_ipaddress(server)) {
2719             /* windows 9* needs a correct NMB name for connections */
2720             fstring remote_name;
2721
2722             if (name_status_find("*", 0, 0, server_ss, remote_name)) {
2723                 cli = get_ipc_connect(remote_name, server_ss, user_info);
2724                 if (cli)
2725                     return cli;
2726             }
2727         }
2728         return NULL;
2729 }
2730
2731 /*
2732  * Given the IP address of a master browser on the network, return its
2733  * workgroup and connect to it.
2734  *
2735  * This function is provided to allow additional processing beyond what
2736  * get_ipc_connect_master_ip_bcast() does, e.g. to retrieve the list of master
2737  * browsers and obtain each master browsers' list of domains (in case the
2738  * first master browser is recently on the network and has not yet
2739  * synchronized with other master browsers and therefore does not yet have the
2740  * entire network browse list)
2741  */
2742
2743 struct cli_state *get_ipc_connect_master_ip(TALLOC_CTX *ctx,
2744                                 struct ip_service *mb_ip,
2745                                 const struct user_auth_info *user_info,
2746                                 char **pp_workgroup_out)
2747 {
2748         char addr[INET6_ADDRSTRLEN];
2749         fstring name;
2750         struct cli_state *cli;
2751         struct sockaddr_storage server_ss;
2752
2753         *pp_workgroup_out = NULL;
2754
2755         print_sockaddr(addr, sizeof(addr), &mb_ip->ss);
2756         DEBUG(99, ("Looking up name of master browser %s\n",
2757                    addr));
2758
2759         /*
2760          * Do a name status query to find out the name of the master browser.
2761          * We use <01><02>__MSBROWSE__<02>#01 if *#00 fails because a domain
2762          * master browser will not respond to a wildcard query (or, at least,
2763          * an NT4 server acting as the domain master browser will not).
2764          *
2765          * We might be able to use ONLY the query on MSBROWSE, but that's not
2766          * yet been tested with all Windows versions, so until it is, leave
2767          * the original wildcard query as the first choice and fall back to
2768          * MSBROWSE if the wildcard query fails.
2769          */
2770         if (!name_status_find("*", 0, 0x1d, &mb_ip->ss, name) &&
2771             !name_status_find(MSBROWSE, 1, 0x1d, &mb_ip->ss, name)) {
2772
2773                 DEBUG(99, ("Could not retrieve name status for %s\n",
2774                            addr));
2775                 return NULL;
2776         }
2777
2778         if (!find_master_ip(name, &server_ss)) {
2779                 DEBUG(99, ("Could not find master ip for %s\n", name));
2780                 return NULL;
2781         }
2782
2783         *pp_workgroup_out = talloc_strdup(ctx, name);
2784
2785         DEBUG(4, ("found master browser %s, %s\n", name, addr));
2786
2787         print_sockaddr(addr, sizeof(addr), &server_ss);
2788         cli = get_ipc_connect(addr, &server_ss, user_info);
2789
2790         return cli;
2791 }
2792
2793 /*
2794  * Return the IP address and workgroup of a master browser on the network, and
2795  * connect to it.
2796  */
2797
2798 struct cli_state *get_ipc_connect_master_ip_bcast(TALLOC_CTX *ctx,
2799                                         const struct user_auth_info *user_info,
2800                                         char **pp_workgroup_out)
2801 {
2802         struct ip_service *ip_list;
2803         struct cli_state *cli;
2804         int i, count;
2805
2806         *pp_workgroup_out = NULL;
2807
2808         DEBUG(99, ("Do broadcast lookup for workgroups on local network\n"));
2809
2810         /* Go looking for workgroups by broadcasting on the local network */
2811
2812         if (!NT_STATUS_IS_OK(name_resolve_bcast(MSBROWSE, 1, &ip_list,
2813                                                 &count))) {
2814                 DEBUG(99, ("No master browsers responded\n"));
2815                 return False;
2816         }
2817
2818         for (i = 0; i < count; i++) {
2819                 char addr[INET6_ADDRSTRLEN];
2820                 print_sockaddr(addr, sizeof(addr), &ip_list[i].ss);
2821                 DEBUG(99, ("Found master browser %s\n", addr));
2822
2823                 cli = get_ipc_connect_master_ip(ctx, &ip_list[i],
2824                                 user_info, pp_workgroup_out);
2825                 if (cli)
2826                         return(cli);
2827         }
2828
2829         return NULL;
2830 }