s3:libsmb: add SMB2/3 support to cli_dfs_get_referral()
[mat/samba.git] / source3 / libsmb / clidfs.c
index 2a39ee9b58e6d045e12b2f1b7e51a7dc70ee1681..ff48719c76b6b94717b6d9b5047e23ab4f834bde 100644 (file)
 */
 
 #include "includes.h"
+#include "libsmb/libsmb.h"
+#include "libsmb/clirap.h"
+#include "msdfs.h"
+#include "trans2.h"
+#include "libsmb/nmblib.h"
+#include "../libcli/smb/smbXcli_base.h"
 
 /********************************************************************
  Important point.
@@ -42,7 +48,23 @@ NTSTATUS cli_cm_force_encryption(struct cli_state *c,
                        const char *domain,
                        const char *sharename)
 {
-       NTSTATUS status = cli_force_encryption(c,
+       NTSTATUS status;
+
+       if (smbXcli_conn_protocol(c->conn) >= PROTOCOL_SMB2_02) {
+               status = smb2cli_session_encryption_on(c->smb2.session);
+               if (NT_STATUS_EQUAL(status,NT_STATUS_NOT_SUPPORTED)) {
+                       d_printf("Encryption required and "
+                               "server doesn't support "
+                               "SMB3 encryption - failing connect\n");
+               } else if (!NT_STATUS_IS_OK(status)) {
+                       d_printf("Encryption required and "
+                               "setup failed with error %s.\n",
+                               nt_errstr(status));
+               }
+               return status;
+       }
+
+       status = cli_force_encryption(c,
                                        username,
                                        password,
                                        domain);
@@ -72,7 +94,7 @@ NTSTATUS cli_cm_force_encryption(struct cli_state *c,
  Return a connection to a server.
 ********************************************************************/
 
-static struct cli_state *do_connect(TALLOC_CTX *ctx,
+static NTSTATUS do_connect(TALLOC_CTX *ctx,
                                        const char *server,
                                        const char *share,
                                        const struct user_auth_info *auth_info,
@@ -80,131 +102,109 @@ static struct cli_state *do_connect(TALLOC_CTX *ctx,
                                        bool force_encrypt,
                                        int max_protocol,
                                        int port,
-                                       int name_type)
+                                       int name_type,
+                                       struct cli_state **pcli)
 {
        struct cli_state *c = NULL;
-       struct nmb_name called, calling;
-       const char *called_str;
-       const char *server_n;
-       struct sockaddr_storage ss;
        char *servicename;
        char *sharename;
        char *newserver, *newshare;
        const char *username;
        const char *password;
        NTSTATUS status;
+       int flags = 0;
 
        /* make a copy so we don't modify the global string 'service' */
        servicename = talloc_strdup(ctx,share);
        if (!servicename) {
-               return NULL;
+               return NT_STATUS_NO_MEMORY;
        }
        sharename = servicename;
        if (*sharename == '\\') {
                sharename += 2;
-               called_str = sharename;
                if (server == NULL) {
                        server = sharename;
                }
                sharename = strchr_m(sharename,'\\');
                if (!sharename) {
-                       return NULL;
+                       return NT_STATUS_NO_MEMORY;
                }
                *sharename = 0;
                sharename++;
-       } else {
-               called_str = server;
+       }
+       if (server == NULL) {
+               return NT_STATUS_INVALID_PARAMETER;
        }
 
-       server_n = server;
-
-       zero_sockaddr(&ss);
-
-       make_nmb_name(&calling, global_myname(), 0x0);
-       make_nmb_name(&called , called_str, name_type);
-
- again:
-       zero_sockaddr(&ss);
-
-       /* have to open a new connection */
-       if (!(c=cli_initialise_ex(get_cmdline_auth_info_signing_state(auth_info)))) {
-               d_printf("Connection to %s failed\n", server_n);
-               if (c) {
-                       cli_shutdown(c);
-               }
-               return NULL;
+       if (get_cmdline_auth_info_use_kerberos(auth_info)) {
+               flags |= CLI_FULL_CONNECTION_USE_KERBEROS;
        }
-       if (port) {
-               cli_set_port(c, port);
+       if (get_cmdline_auth_info_fallback_after_kerberos(auth_info)) {
+               flags |= CLI_FULL_CONNECTION_FALLBACK_AFTER_KERBEROS;
+       }
+       if (get_cmdline_auth_info_use_ccache(auth_info)) {
+               flags |= CLI_FULL_CONNECTION_USE_CCACHE;
+       }
+       if (get_cmdline_auth_info_use_pw_nt_hash(auth_info)) {
+               flags |= CLI_FULL_CONNECTION_USE_NT_HASH;
        }
 
-       status = cli_connect(c, server_n, &ss);
+       status = cli_connect_nb(
+               server, NULL, port, name_type, NULL,
+               get_cmdline_auth_info_signing_state(auth_info),
+               flags, &c);
+
        if (!NT_STATUS_IS_OK(status)) {
                d_printf("Connection to %s failed (Error %s)\n",
-                               server_n,
+                               server,
                                nt_errstr(status));
-               cli_shutdown(c);
-               return NULL;
+               return status;
        }
 
        if (max_protocol == 0) {
                max_protocol = PROTOCOL_NT1;
        }
-       c->protocol = max_protocol;
-       c->use_kerberos = get_cmdline_auth_info_use_kerberos(auth_info);
-       c->fallback_after_kerberos =
-               get_cmdline_auth_info_fallback_after_kerberos(auth_info);
-       c->use_ccache = get_cmdline_auth_info_use_ccache(auth_info);
-
-       if (!cli_session_request(c, &calling, &called)) {
-               char *p;
-               d_printf("session request to %s failed (%s)\n",
-                        called.name, cli_errstr(c));
-               cli_shutdown(c);
-               c = NULL;
-               if ((p=strchr_m(called.name, '.'))) {
-                       *p = 0;
-                       goto again;
-               }
-               if (strcmp(called.name, "*SMBSERVER")) {
-                       make_nmb_name(&called , "*SMBSERVER", 0x20);
-                       goto again;
-               }
-               return NULL;
-       }
-
        DEBUG(4,(" session request ok\n"));
 
-       status = cli_negprot(c);
+       status = smbXcli_negprot(c->conn, c->timeout,
+                                lp_cli_minprotocol(),
+                                max_protocol);
 
        if (!NT_STATUS_IS_OK(status)) {
                d_printf("protocol negotiation failed: %s\n",
                         nt_errstr(status));
                cli_shutdown(c);
-               return NULL;
+               return status;
+       }
+
+       if (smbXcli_conn_protocol(c->conn) >= PROTOCOL_SMB2_02) {
+               /* Ensure we ask for some initial credits. */
+               smb2cli_conn_set_max_credits(c->conn, DEFAULT_SMB2_MAX_CREDITS);
        }
 
        username = get_cmdline_auth_info_username(auth_info);
        password = get_cmdline_auth_info_password(auth_info);
 
-       if (!NT_STATUS_IS_OK(cli_session_setup(c, username,
-                                              password, strlen(password),
-                                              password, strlen(password),
-                                              lp_workgroup()))) {
+       status = cli_session_setup(c, username,
+                                  password, strlen(password),
+                                  password, strlen(password),
+                                  lp_workgroup());
+       if (!NT_STATUS_IS_OK(status)) {
                /* If a password was not supplied then
                 * try again with a null username. */
                if (password[0] || !username[0] ||
                        get_cmdline_auth_info_use_kerberos(auth_info) ||
-                       !NT_STATUS_IS_OK(cli_session_setup(c, "",
+                       !NT_STATUS_IS_OK(status = cli_session_setup(c, "",
                                                "", 0,
                                                "", 0,
                                               lp_workgroup()))) {
-                       d_printf("session setup failed: %s\n", cli_errstr(c));
-                       if (NT_STATUS_V(cli_nt_error(c)) ==
-                           NT_STATUS_V(NT_STATUS_MORE_PROCESSING_REQUIRED))
+                       d_printf("session setup failed: %s\n",
+                                nt_errstr(status));
+                       if (NT_STATUS_EQUAL(status,
+                                           NT_STATUS_MORE_PROCESSING_REQUIRED))
                                d_printf("did you forget to run kinit?\n");
                        cli_shutdown(c);
-                       return NULL;
+                       return status;
                }
                d_printf("Anonymous login successful\n");
                status = cli_init_creds(c, "", lp_workgroup(), "");
@@ -215,7 +215,7 @@ static struct cli_state *do_connect(TALLOC_CTX *ctx,
        if (!NT_STATUS_IS_OK(status)) {
                DEBUG(10,("cli_init_creds() failed: %s\n", nt_errstr(status)));
                cli_shutdown(c);
-               return NULL;
+               return status;
        }
 
        if ( show_sessetup ) {
@@ -234,7 +234,7 @@ static struct cli_state *do_connect(TALLOC_CTX *ctx,
           here before trying to connect to the original share.
           cli_check_msdfs_proxy() will fail if it is a normal share. */
 
-       if ((c->capabilities & CAP_DFS) &&
+       if (smbXcli_conn_dfs_supported(c->conn) &&
                        cli_check_msdfs_proxy(ctx, c, sharename,
                                &newserver, &newshare,
                                force_encrypt,
@@ -245,17 +245,17 @@ static struct cli_state *do_connect(TALLOC_CTX *ctx,
                return do_connect(ctx, newserver,
                                newshare, auth_info, false,
                                force_encrypt, max_protocol,
-                               port, name_type);
+                               port, name_type, pcli);
        }
 
        /* must be a normal share */
 
-       status = cli_tcon_andx(c, sharename, "?????",
-                              password, strlen(password)+1);
+       status = cli_tree_connect(c, sharename, "?????",
+                                 password, strlen(password)+1);
        if (!NT_STATUS_IS_OK(status)) {
                d_printf("tree connect failed: %s\n", nt_errstr(status));
                cli_shutdown(c);
-               return NULL;
+               return status;
        }
 
        if (force_encrypt) {
@@ -266,12 +266,13 @@ static struct cli_state *do_connect(TALLOC_CTX *ctx,
                                        sharename);
                if (!NT_STATUS_IS_OK(status)) {
                        cli_shutdown(c);
-                       return NULL;
+                       return status;
                }
        }
 
        DEBUG(4,(" tconx ok\n"));
-       return c;
+       *pcli = c;
+       return NT_STATUS_OK;
 }
 
 /****************************************************************************
@@ -293,26 +294,28 @@ static void cli_set_mntpoint(struct cli_state *cli, const char *mnt)
  referring_cli == NULL means a new initial connection.
 ********************************************************************/
 
-static struct cli_state *cli_cm_connect(TALLOC_CTX *ctx,
-                                       struct cli_state *referring_cli,
-                                       const char *server,
-                                       const char *share,
-                                       const struct user_auth_info *auth_info,
-                                       bool show_hdr,
-                                       bool force_encrypt,
-                                       int max_protocol,
-                                       int port,
-                                       int name_type)
+static NTSTATUS cli_cm_connect(TALLOC_CTX *ctx,
+                              struct cli_state *referring_cli,
+                              const char *server,
+                              const char *share,
+                              const struct user_auth_info *auth_info,
+                              bool show_hdr,
+                              bool force_encrypt,
+                              int max_protocol,
+                              int port,
+                              int name_type,
+                              struct cli_state **pcli)
 {
        struct cli_state *cli;
+       NTSTATUS status;
 
-       cli = do_connect(ctx, server, share,
+       status = do_connect(ctx, server, share,
                                auth_info,
                                show_hdr, force_encrypt, max_protocol,
-                               port, name_type);
+                               port, name_type, &cli);
 
-       if (!cli ) {
-               return NULL;
+       if (!NT_STATUS_IS_OK(status)) {
+               return status;
        }
 
        /* Enter into the list. */
@@ -320,10 +323,9 @@ static struct cli_state *cli_cm_connect(TALLOC_CTX *ctx,
                DLIST_ADD_END(referring_cli, cli, struct cli_state *);
        }
 
-       if (referring_cli && referring_cli->posix_capabilities) {
+       if (referring_cli && referring_cli->requested_posix_capabilities) {
                uint16 major, minor;
                uint32 caplow, caphigh;
-               NTSTATUS status;
                status = cli_unix_extensions_version(cli, &major, &minor,
                                                     &caplow, &caphigh);
                if (NT_STATUS_IS_OK(status)) {
@@ -333,7 +335,8 @@ static struct cli_state *cli_cm_connect(TALLOC_CTX *ctx,
                }
        }
 
-       return cli;
+       *pcli = cli;
+       return NT_STATUS_OK;
 }
 
 /********************************************************************
@@ -351,8 +354,11 @@ static struct cli_state *cli_cm_find(struct cli_state *cli,
        }
 
        /* Search to the start of the list. */
-       for (p = cli; p; p = p->prev) {
-               if (strequal(server, p->desthost) &&
+       for (p = cli; p; p = DLIST_PREV(p)) {
+               const char *remote_name =
+                       smbXcli_conn_remote_name(p->conn);
+
+               if (strequal(server, remote_name) &&
                                strequal(share,p->share)) {
                        return p;
                }
@@ -360,7 +366,10 @@ static struct cli_state *cli_cm_find(struct cli_state *cli,
 
        /* Search to the end of the list. */
        for (p = cli->next; p; p = p->next) {
-               if (strequal(server, p->desthost) &&
+               const char *remote_name =
+                       smbXcli_conn_remote_name(p->conn);
+
+               if (strequal(server, remote_name) &&
                                strequal(share,p->share)) {
                        return p;
                }
@@ -373,7 +382,7 @@ static struct cli_state *cli_cm_find(struct cli_state *cli,
  Open a client connection to a \\server\share.
 ****************************************************************************/
 
-struct cli_state *cli_cm_open(TALLOC_CTX *ctx,
+NTSTATUS cli_cm_open(TALLOC_CTX *ctx,
                                struct cli_state *referring_cli,
                                const char *server,
                                const char *share,
@@ -382,13 +391,16 @@ struct cli_state *cli_cm_open(TALLOC_CTX *ctx,
                                bool force_encrypt,
                                int max_protocol,
                                int port,
-                               int name_type)
+                               int name_type,
+                               struct cli_state **pcli)
 {
        /* Try to reuse an existing connection in this list. */
        struct cli_state *c = cli_cm_find(referring_cli, server, share);
+       NTSTATUS status;
 
        if (c) {
-               return c;
+               *pcli = c;
+               return NT_STATUS_OK;
        }
 
        if (auth_info == NULL) {
@@ -397,10 +409,10 @@ struct cli_state *cli_cm_open(TALLOC_CTX *ctx,
                d_printf("cli_cm_open() Unable to open connection [\\%s\\%s] "
                        "without auth info\n",
                        server, share );
-               return NULL;
+               return NT_STATUS_INVALID_PARAMETER;
        }
 
-       return cli_cm_connect(ctx,
+       status = cli_cm_connect(ctx,
                                referring_cli,
                                server,
                                share,
@@ -409,19 +421,25 @@ struct cli_state *cli_cm_open(TALLOC_CTX *ctx,
                                force_encrypt,
                                max_protocol,
                                port,
-                               name_type);
+                               name_type,
+                               &c);
+       if (!NT_STATUS_IS_OK(status)) {
+               return status;
+       }
+       *pcli = c;
+       return NT_STATUS_OK;
 }
 
 /****************************************************************************
 ****************************************************************************/
 
-void cli_cm_display(const struct cli_state *cli)
+void cli_cm_display(struct cli_state *cli)
 {
        int i;
 
        for (i=0; cli; cli = cli->next,i++ ) {
                d_printf("%d:\tserver=%s, share=%s\n",
-                       i, cli->desthost, cli->share );
+                       i, smbXcli_conn_remote_name(cli->conn), cli->share);
        }
 }
 
@@ -452,7 +470,7 @@ void cli_cm_set_credentials(struct user_auth_info *auth_info)
  split a dfs path into the server, share name, and extrapath components
 **********************************************************************/
 
-static void split_dfs_path(TALLOC_CTX *ctx,
+static bool split_dfs_path(TALLOC_CTX *ctx,
                                const char *nodepath,
                                char **pp_server,
                                char **pp_share,
@@ -467,16 +485,16 @@ static void split_dfs_path(TALLOC_CTX *ctx,
 
        path = talloc_strdup(ctx, nodepath);
        if (!path) {
-               return;
+               goto fail;
        }
 
        if ( path[0] != '\\' ) {
-               return;
+               goto fail;
        }
 
        p = strchr_m( path + 1, '\\' );
        if ( !p ) {
-               return;
+               goto fail;
        }
 
        *p = '\0';
@@ -491,9 +509,28 @@ static void split_dfs_path(TALLOC_CTX *ctx,
        } else {
                *pp_extrapath = talloc_strdup(ctx, "");
        }
+       if (*pp_extrapath == NULL) {
+               goto fail;
+       }
 
        *pp_share = talloc_strdup(ctx, p);
+       if (*pp_share == NULL) {
+               goto fail;
+       }
+
        *pp_server = talloc_strdup(ctx, &path[1]);
+       if (*pp_server == NULL) {
+               goto fail;
+       }
+
+       TALLOC_FREE(path);
+       return true;
+
+fail:
+       TALLOC_FREE(*pp_share);
+       TALLOC_FREE(*pp_extrapath);
+       TALLOC_FREE(path);
+       return false;
 }
 
 /****************************************************************************
@@ -564,12 +601,12 @@ static char *cli_dfs_make_full_path(TALLOC_CTX *ctx,
                dir++;
        }
 
-       if (cli->posix_capabilities & CIFS_UNIX_POSIX_PATHNAMES_CAP) {
+       if (cli->requested_posix_capabilities & CIFS_UNIX_POSIX_PATHNAMES_CAP) {
                path_sep = '/';
        }
        return talloc_asprintf(ctx, "%c%s%c%s%c%s",
                        path_sep,
-                       cli->desthost,
+                       smbXcli_conn_remote_name(cli->conn),
                        path_sep,
                        cli->share,
                        path_sep,
@@ -580,19 +617,20 @@ static char *cli_dfs_make_full_path(TALLOC_CTX *ctx,
  check for dfs referral
 ********************************************************************/
 
-static bool cli_dfs_check_error( struct cli_state *cli, NTSTATUS status )
+static bool cli_dfs_check_error(struct cli_state *cli, NTSTATUS expected,
+                               NTSTATUS status)
 {
-       uint32 flgs2 = SVAL(cli->inbuf,smb_flg2);
-
        /* only deal with DS when we negotiated NT_STATUS codes and UNICODE */
 
-       if (!((flgs2&FLAGS2_32_BIT_ERROR_CODES) &&
-                               (flgs2&FLAGS2_UNICODE_STRINGS)))
+       if (!(smbXcli_conn_use_unicode(cli->conn))) {
                return false;
-
-       if (NT_STATUS_EQUAL(status, NT_STATUS(IVAL(cli->inbuf,smb_rcls))))
+       }
+       if (!(smb1cli_conn_capabilities(cli->conn) & CAP_STATUS32)) {
+               return false;
+       }
+       if (NT_STATUS_EQUAL(status, expected)) {
                return true;
-
+       }
        return false;
 }
 
@@ -600,63 +638,104 @@ static bool cli_dfs_check_error( struct cli_state *cli, NTSTATUS status )
  Get the dfs referral link.
 ********************************************************************/
 
-bool cli_dfs_get_referral(TALLOC_CTX *ctx,
+NTSTATUS cli_dfs_get_referral(TALLOC_CTX *ctx,
                        struct cli_state *cli,
                        const char *path,
                        struct client_dfs_referral **refs,
                        size_t *num_refs,
                        size_t *consumed)
 {
-       unsigned int data_len = 0;
        unsigned int param_len = 0;
-       uint16 setup = TRANSACT2_GET_DFS_REFERRAL;
-       char *param = NULL;
-       char *rparam=NULL, *rdata=NULL;
+       uint16_t recv_flags2;
+       uint8_t *param = NULL;
+       uint8_t *rdata = NULL;
        char *p;
        char *endp;
-       size_t pathlen = 2*(strlen(path)+1);
        smb_ucs2_t *path_ucs;
        char *consumed_path = NULL;
        uint16_t consumed_ucs;
        uint16 num_referrals;
        struct client_dfs_referral *referrals = NULL;
-       bool ret = false;
+       NTSTATUS status;
+       TALLOC_CTX *frame = talloc_stackframe();
 
        *num_refs = 0;
        *refs = NULL;
 
-       param = SMB_MALLOC_ARRAY(char, 2+pathlen+2);
+       param = talloc_array(talloc_tos(), uint8_t, 2);
        if (!param) {
+               status = NT_STATUS_NO_MEMORY;
                goto out;
        }
        SSVAL(param, 0, 0x03);  /* max referral level */
-       p = &param[2];
-
-       path_ucs = (smb_ucs2_t *)p;
-       p += clistr_push(cli, p, path, pathlen, STR_TERMINATE);
-       param_len = PTR_DIFF(p, param);
-
-       if (!cli_send_trans(cli, SMBtrans2,
-                       NULL,                        /* name */
-                       -1, 0,                          /* fid, flags */
-                       &setup, 1, 0,                   /* setup, length, max */
-                       param, param_len, 2,            /* param, length, max */
-                       NULL, 0, cli->max_xmit /* data, length, max */
-                       )) {
-               goto out;
-       }
 
-       if (!cli_receive_trans(cli, SMBtrans2,
-               &rparam, &param_len,
-               &rdata, &data_len)) {
+       param = trans2_bytes_push_str(param, smbXcli_conn_use_unicode(cli->conn),
+                                     path, strlen(path)+1,
+                                     NULL);
+       if (!param) {
+               status = NT_STATUS_NO_MEMORY;
                goto out;
        }
+       param_len = talloc_get_size(param);
+       path_ucs = (smb_ucs2_t *)&param[2];
+
+       if (smbXcli_conn_protocol(cli->conn) >= PROTOCOL_SMB2_02) {
+               DATA_BLOB in_input_buffer;
+               DATA_BLOB in_output_buffer = data_blob_null;
+               DATA_BLOB out_input_buffer = data_blob_null;
+               DATA_BLOB out_output_buffer = data_blob_null;
+
+               in_input_buffer.data = param;
+               in_input_buffer.length = param_len;
+
+               status = smb2cli_ioctl(cli->conn,
+                                      cli->timeout,
+                                      cli->smb2.session,
+                                      cli->smb2.tcon,
+                                      UINT64_MAX, /* in_fid_persistent */
+                                      UINT64_MAX, /* in_fid_volatile */
+                                      FSCTL_DFS_GET_REFERRALS,
+                                      0, /* in_max_input_length */
+                                      &in_input_buffer,
+                                      CLI_BUFFER_SIZE, /* in_max_output_length */
+                                      &in_output_buffer,
+                                      SMB2_IOCTL_FLAG_IS_FSCTL,
+                                      talloc_tos(),
+                                      &out_input_buffer,
+                                      &out_output_buffer);
+               if (!NT_STATUS_IS_OK(status)) {
+                       goto out;
+               }
 
-       if (data_len < 4) {
-               goto out;
-       }
+               if (out_output_buffer.length < 4) {
+                       status = NT_STATUS_INVALID_NETWORK_RESPONSE;
+                       goto out;
+               }
+
+               recv_flags2 = FLAGS2_UNICODE_STRINGS;
+               rdata = out_output_buffer.data;
+               endp = (char *)rdata + out_output_buffer.length;
+       } else {
+               unsigned int data_len = 0;
+               uint16_t setup[1];
+
+               SSVAL(setup, 0, TRANSACT2_GET_DFS_REFERRAL);
+
+               status = cli_trans(talloc_tos(), cli, SMBtrans2,
+                                  NULL, 0xffff, 0, 0,
+                                  setup, 1, 0,
+                                  param, param_len, 2,
+                                  NULL, 0, CLI_BUFFER_SIZE,
+                                  &recv_flags2,
+                                  NULL, 0, NULL, /* rsetup */
+                                  NULL, 0, NULL,
+                                  &rdata, 4, &data_len);
+               if (!NT_STATUS_IS_OK(status)) {
+                       goto out;
+               }
 
-       endp = rdata + data_len;
+               endp = (char *)rdata + data_len;
+       }
 
        consumed_ucs  = SVAL(rdata, 0);
        num_referrals = SVAL(rdata, 2);
@@ -668,6 +747,7 @@ bool cli_dfs_get_referral(TALLOC_CTX *ctx,
         * to get the number of bytes consumed from
         * the incoming path. */
 
+       errno = 0;
        if (pull_string_talloc(talloc_tos(),
                        NULL,
                        0,
@@ -675,9 +755,15 @@ bool cli_dfs_get_referral(TALLOC_CTX *ctx,
                        path_ucs,
                        consumed_ucs,
                        STR_UNICODE) == 0) {
+               if (errno != 0) {
+                       status = map_nt_error_from_unix(errno);
+               } else {
+                       status = NT_STATUS_INVALID_NETWORK_RESPONSE;
+               }
                goto out;
        }
        if (consumed_path == NULL) {
+               status = map_nt_error_from_unix(errno);
                goto out;
        }
        *consumed = strlen(consumed_path);
@@ -692,11 +778,12 @@ bool cli_dfs_get_referral(TALLOC_CTX *ctx,
                                         num_referrals);
 
                if (!referrals) {
+                       status = NT_STATUS_NO_MEMORY;
                        goto out;
                }
                /* start at the referrals array */
 
-               p = rdata+8;
+               p = (char *)rdata+8;
                for (i=0; i<num_referrals && p < endp; i++) {
                        if (p + 18 > endp) {
                                goto out;
@@ -714,47 +801,48 @@ bool cli_dfs_get_referral(TALLOC_CTX *ctx,
                        referrals[i].ttl       = SVAL(p, 10);
 
                        if (p + node_offset > endp) {
+                               status = NT_STATUS_INVALID_NETWORK_RESPONSE;
                                goto out;
                        }
-                       clistr_pull_talloc(ctx, cli->inbuf,
+                       clistr_pull_talloc(referrals,
+                                          (const char *)rdata,
+                                          recv_flags2,
                                           &referrals[i].dfspath,
-                                          p+node_offset, -1,
+                                          p+node_offset,
+                                          PTR_DIFF(endp, p+node_offset),
                                           STR_TERMINATE|STR_UNICODE);
 
                        if (!referrals[i].dfspath) {
+                               status = map_nt_error_from_unix(errno);
                                goto out;
                        }
                        p += ref_size;
                }
                if (i < num_referrals) {
+                       status = NT_STATUS_INVALID_NETWORK_RESPONSE;
                        goto out;
                }
        }
 
-       ret = true;
-
        *num_refs = num_referrals;
        *refs = referrals;
 
   out:
 
-       TALLOC_FREE(consumed_path);
-       SAFE_FREE(param);
-       SAFE_FREE(rdata);
-       SAFE_FREE(rparam);
-       return ret;
+       TALLOC_FREE(frame);
+       return status;
 }
 
 /********************************************************************
 ********************************************************************/
 
-bool cli_resolve_path(TALLOC_CTX *ctx,
-                       const char *mountpt,
-                       const struct user_auth_info *dfs_auth_info,
-                       struct cli_state *rootcli,
-                       const char *path,
-                       struct cli_state **targetcli,
-                       char **pp_targetpath)
+NTSTATUS cli_resolve_path(TALLOC_CTX *ctx,
+                         const char *mountpt,
+                         const struct user_auth_info *dfs_auth_info,
+                         struct cli_state *rootcli,
+                         const char *path,
+                         struct cli_state **targetcli,
+                         char **pp_targetpath)
 {
        struct client_dfs_referral *refs = NULL;
        size_t num_refs = 0;
@@ -772,20 +860,29 @@ bool cli_resolve_path(TALLOC_CTX *ctx,
        char *ppath = NULL;
        SMB_STRUCT_STAT sbuf;
        uint32 attributes;
+       NTSTATUS status;
+       struct smbXcli_tcon *root_tcon = NULL;
+       struct smbXcli_tcon *target_tcon = NULL;
 
        if ( !rootcli || !path || !targetcli ) {
-               return false;
+               return NT_STATUS_INVALID_PARAMETER;
        }
 
        /* Don't do anything if this is not a DFS root. */
 
-       if ( !rootcli->dfsroot) {
+       if (smbXcli_conn_protocol(rootcli->conn) >= PROTOCOL_SMB2_02) {
+               root_tcon = rootcli->smb2.tcon;
+       } else {
+               root_tcon = rootcli->smb1.tcon;
+       }
+
+       if (!smbXcli_tcon_is_dfs_share(root_tcon)) {
                *targetcli = rootcli;
                *pp_targetpath = talloc_strdup(ctx, path);
                if (!*pp_targetpath) {
-                       return false;
+                       return NT_STATUS_NO_MEMORY;
                }
-               return true;
+               return NT_STATUS_OK;
        }
 
        *targetcli = NULL;
@@ -794,83 +891,88 @@ bool cli_resolve_path(TALLOC_CTX *ctx,
 
        cleanpath = clean_path(ctx, path);
        if (!cleanpath) {
-               return false;
+               return NT_STATUS_NO_MEMORY;
        }
 
        dfs_path = cli_dfs_make_full_path(ctx, rootcli, cleanpath);
        if (!dfs_path) {
-               return false;
+               return NT_STATUS_NO_MEMORY;
        }
 
-       if (cli_qpathinfo_basic( rootcli, dfs_path, &sbuf, &attributes)) {
+       status = cli_qpathinfo_basic( rootcli, dfs_path, &sbuf, &attributes);
+       if (NT_STATUS_IS_OK(status)) {
                /* This is an ordinary path, just return it. */
                *targetcli = rootcli;
                *pp_targetpath = talloc_strdup(ctx, path);
                if (!*pp_targetpath) {
-                       return false;
+                       return NT_STATUS_NO_MEMORY;
                }
                goto done;
        }
 
        /* Special case where client asked for a path that does not exist */
 
-       if (cli_dfs_check_error(rootcli, NT_STATUS_OBJECT_NAME_NOT_FOUND)) {
+       if (cli_dfs_check_error(rootcli, NT_STATUS_OBJECT_NAME_NOT_FOUND,
+                               status)) {
                *targetcli = rootcli;
                *pp_targetpath = talloc_strdup(ctx, path);
                if (!*pp_targetpath) {
-                       return false;
+                       return NT_STATUS_NO_MEMORY;
                }
                goto done;
        }
 
        /* We got an error, check for DFS referral. */
 
-       if (!cli_dfs_check_error(rootcli, NT_STATUS_PATH_NOT_COVERED)) {
-               return false;
+       if (!cli_dfs_check_error(rootcli, NT_STATUS_PATH_NOT_COVERED,
+                                status)) {
+               return status;
        }
 
        /* Check for the referral. */
 
-       if (!(cli_ipc = cli_cm_open(ctx,
-                               rootcli,
-                               rootcli->desthost,
-                               "IPC$",
-                               dfs_auth_info,
-                               false,
-                               (rootcli->trans_enc_state != NULL),
-                               rootcli->protocol,
-                               0,
-                               0x20))) {
-               return false;
+       status = cli_cm_open(ctx,
+                            rootcli,
+                            smbXcli_conn_remote_name(rootcli->conn),
+                            "IPC$",
+                            dfs_auth_info,
+                            false,
+                            smb1cli_conn_encryption_on(rootcli->conn),
+                            smbXcli_conn_protocol(rootcli->conn),
+                            0,
+                            0x20,
+                            &cli_ipc);
+       if (!NT_STATUS_IS_OK(status)) {
+               return status;
        }
 
-       if (!cli_dfs_get_referral(ctx, cli_ipc, dfs_path, &refs,
-                       &num_refs, &consumed) || !num_refs) {
-               return false;
+       status = cli_dfs_get_referral(ctx, cli_ipc, dfs_path, &refs,
+                                     &num_refs, &consumed);
+       if (!NT_STATUS_IS_OK(status) || !num_refs) {
+               return status;
        }
 
        /* Just store the first referral for now. */
 
        if (!refs[0].dfspath) {
-               return false;
+               return NT_STATUS_NOT_FOUND;
        }
-       split_dfs_path(ctx, refs[0].dfspath, &server, &share, &extrapath );
-
-       if (!server || !share) {
-               return false;
+       if (!split_dfs_path(ctx, refs[0].dfspath, &server, &share,
+                           &extrapath)) {
+               return NT_STATUS_NOT_FOUND;
        }
 
        /* Make sure to recreate the original string including any wildcards. */
 
        dfs_path = cli_dfs_make_full_path(ctx, rootcli, path);
        if (!dfs_path) {
-               return false;
+               return NT_STATUS_NO_MEMORY;
        }
        pathlen = strlen(dfs_path);
        consumed = MIN(pathlen, consumed);
        *pp_targetpath = talloc_strdup(ctx, &dfs_path[consumed]);
        if (!*pp_targetpath) {
-               return false;
+               return NT_STATUS_NO_MEMORY;
        }
        dfs_path[consumed] = '\0';
 
@@ -881,27 +983,38 @@ bool cli_resolve_path(TALLOC_CTX *ctx,
         */
 
        /* Open the connection to the target server & share */
-       if ((*targetcli = cli_cm_open(ctx, rootcli,
-                                       server,
-                                       share,
-                                       dfs_auth_info,
-                                       false,
-                                       (rootcli->trans_enc_state != NULL),
-                                       rootcli->protocol,
-                                       0,
-                                       0x20)) == NULL) {
+       status = cli_cm_open(ctx, rootcli,
+                            server,
+                            share,
+                            dfs_auth_info,
+                            false,
+                            smb1cli_conn_encryption_on(rootcli->conn),
+                            smbXcli_conn_protocol(rootcli->conn),
+                            0,
+                            0x20,
+                            targetcli);
+       if (!NT_STATUS_IS_OK(status)) {
                d_printf("Unable to follow dfs referral [\\%s\\%s]\n",
                        server, share );
-               return false;
+               return status;
        }
 
        if (extrapath && strlen(extrapath) > 0) {
-               *pp_targetpath = talloc_asprintf(ctx,
-                                               "%s%s",
-                                               extrapath,
-                                               *pp_targetpath);
+               /* EMC Celerra NAS version 5.6.50 (at least) doesn't appear to */
+               /* put the trailing \ on the path, so to be save we put one in if needed */
+               if (extrapath[strlen(extrapath)-1] != '\\' && **pp_targetpath != '\\') {
+                       *pp_targetpath = talloc_asprintf(ctx,
+                                                 "%s\\%s",
+                                                 extrapath,
+                                                 *pp_targetpath);
+               } else {
+                       *pp_targetpath = talloc_asprintf(ctx,
+                                                 "%s%s",
+                                                 extrapath,
+                                                 *pp_targetpath);
+               }
                if (!*pp_targetpath) {
-                       return false;
+                       return NT_STATUS_NO_MEMORY;
                }
        }
 
@@ -914,26 +1027,26 @@ bool cli_resolve_path(TALLOC_CTX *ctx,
                d_printf("cli_resolve_path: "
                        "dfs_path (%s) not in correct format.\n",
                        dfs_path );
-               return false;
+               return NT_STATUS_NOT_FOUND;
        }
 
        ppath++; /* Now pointing at start of server name. */
 
        if ((ppath = strchr_m( dfs_path, '\\' )) == NULL) {
-               return false;
+               return NT_STATUS_NOT_FOUND;
        }
 
        ppath++; /* Now pointing at start of share name. */
 
        if ((ppath = strchr_m( ppath+1, '\\' )) == NULL) {
-               return false;
+               return NT_STATUS_NOT_FOUND;
        }
 
        ppath++; /* Now pointing at path component. */
 
        newmount = talloc_asprintf(ctx, "%s\\%s", mountpt, ppath );
        if (!newmount) {
-               return false;
+               return NT_STATUS_NOT_FOUND;
        }
 
        cli_set_mntpoint(*targetcli, newmount);
@@ -942,13 +1055,14 @@ bool cli_resolve_path(TALLOC_CTX *ctx,
           checking for loops here. */
 
        if (!strequal(*pp_targetpath, "\\") && !strequal(*pp_targetpath, "/")) {
-               if (cli_resolve_path(ctx,
-                                       newmount,
-                                       dfs_auth_info,
-                                       *targetcli,
-                                       *pp_targetpath,
-                                       &newcli,
-                                       &newpath)) {
+               status = cli_resolve_path(ctx,
+                                         newmount,
+                                         dfs_auth_info,
+                                         *targetcli,
+                                         *pp_targetpath,
+                                         &newcli,
+                                         &newpath);
+               if (NT_STATUS_IS_OK(status)) {
                        /*
                         * When cli_resolve_path returns true here it's always
                         * returning the complete path in newpath, so we're done
@@ -956,22 +1070,31 @@ bool cli_resolve_path(TALLOC_CTX *ctx,
                         */
                        *targetcli = newcli;
                        *pp_targetpath = newpath;
-                       return true;
+                       return status;
                }
        }
 
   done:
 
+       if (smbXcli_conn_protocol((*targetcli)->conn) >= PROTOCOL_SMB2_02) {
+               target_tcon = (*targetcli)->smb2.tcon;
+       } else {
+               target_tcon = (*targetcli)->smb1.tcon;
+       }
+
        /* If returning true ensure we return a dfs root full path. */
-       if ((*targetcli)->dfsroot) {
+       if (smbXcli_tcon_is_dfs_share(target_tcon)) {
                dfs_path = talloc_strdup(ctx, *pp_targetpath);
                if (!dfs_path) {
-                       return false;
+                       return NT_STATUS_NO_MEMORY;
                }
                *pp_targetpath = cli_dfs_make_full_path(ctx, *targetcli, dfs_path);
+               if (*pp_targetpath == NULL) {
+                       return NT_STATUS_NO_MEMORY;
+               }
        }
 
-       return true;
+       return NT_STATUS_OK;
 }
 
 /********************************************************************
@@ -995,12 +1118,14 @@ bool cli_check_msdfs_proxy(TALLOC_CTX *ctx,
        uint16 cnum;
        char *newextrapath = NULL;
        NTSTATUS status;
+       const char *remote_name;
 
        if (!cli || !sharename) {
                return false;
        }
 
-       cnum = cli->cnum;
+       remote_name = smbXcli_conn_remote_name(cli->conn);
+       cnum = cli_state_get_tid(cli);
 
        /* special case.  never check for a referral on the IPC$ share */
 
@@ -1010,14 +1135,14 @@ bool cli_check_msdfs_proxy(TALLOC_CTX *ctx,
 
        /* send a trans2_query_path_info to check for a referral */
 
-       fullpath = talloc_asprintf(ctx, "\\%s\\%s", cli->desthost, sharename );
+       fullpath = talloc_asprintf(ctx, "\\%s\\%s", remote_name, sharename);
        if (!fullpath) {
                return false;
        }
 
        /* check for the referral */
 
-       if (!NT_STATUS_IS_OK(cli_tcon_andx(cli, "IPC$", "IPC", NULL, 0))) {
+       if (!NT_STATUS_IS_OK(cli_tree_connect(cli, "IPC$", "IPC", NULL, 0))) {
                return false;
        }
 
@@ -1032,14 +1157,16 @@ bool cli_check_msdfs_proxy(TALLOC_CTX *ctx,
                }
        }
 
-       res = cli_dfs_get_referral(ctx, cli, fullpath, &refs, &num_refs, &consumed);
+       status = cli_dfs_get_referral(ctx, cli, fullpath, &refs,
+                                     &num_refs, &consumed);
+       res = NT_STATUS_IS_OK(status);
 
        status = cli_tdis(cli);
        if (!NT_STATUS_IS_OK(status)) {
                return false;
        }
 
-       cli->cnum = cnum;
+       cli_state_set_tid(cli, cnum);
 
        if (!res || !num_refs) {
                return false;
@@ -1049,16 +1176,14 @@ bool cli_check_msdfs_proxy(TALLOC_CTX *ctx,
                return false;
        }
 
-       split_dfs_path(ctx, refs[0].dfspath, pp_newserver,
-                       pp_newshare, &newextrapath );
-
-       if ((*pp_newserver == NULL) || (*pp_newshare == NULL)) {
+       if (!split_dfs_path(ctx, refs[0].dfspath, pp_newserver,
+                           pp_newshare, &newextrapath)) {
                return false;
        }
 
        /* check that this is not a self-referral */
 
-       if (strequal(cli->desthost, *pp_newserver) &&
+       if (strequal(remote_name, *pp_newserver) &&
                        strequal(sharename, *pp_newshare)) {
                return false;
        }