s3: smbtorture: Show correct use of cli_state_save_tcon() / cli_state_restore_tcon().

BUG: https://bugzilla.samba.org/show_bug.cgi?id=12831

Signed-off-by: Jeremy Allison <jra@samba.org>
Reviewed-by: Richard Sharpe <realrichardsharpe@gmail.com>

diff --git a/source3/torture/test_smb2.c b/source3/torture/test_smb2.c
index c0d11e610876f02f3082804144c3148f795e3b41..9368ab3264dc1e723410212a3edcaf80a116955c 100644 (file)
--- a/source3/torture/test_smb2.c
+++ b/source3/torture/test_smb2.c
@@ -170,7 +170,10 @@ bool run_smb2_basic(int dummy)
        }
 
        saved_tid = smb2cli_tcon_current_id(cli->smb2.tcon);
-       saved_tcon = cli->smb2.tcon;
+       saved_tcon = cli_state_save_tcon(cli);
+       if (saved_tcon == NULL) {
+               return false;
+       }
        cli->smb2.tcon = smbXcli_tcon_create(cli);
        smb2cli_tcon_set_values(cli->smb2.tcon,
                                NULL, /* session */
@@ -187,8 +190,7 @@ bool run_smb2_basic(int dummy)
                printf("smb2cli_tdis returned %s\n", nt_errstr(status));
                return false;
        }
-       talloc_free(cli->smb2.tcon);
-       cli->smb2.tcon = saved_tcon;
+       cli_state_restore_tcon(cli, saved_tcon);
 
        status = smb2cli_tdis(cli->conn,
                              cli->timeout,