1
0
mirror of https://github.com/samba-team/samba.git synced 2024-12-22 13:34:15 +03:00
samba-mirror/source4/torture/smb2/credits.c
Jeremy Allison b871d5fdae s4: torture: Add a TALLOC_CTX * to torture_smb2_crediting_init().
Signed-off-by: Jeremy Allison <jra@samba.org>
Reviewed-by: Ralph Böhme <slow@samba.org>
2017-05-05 15:52:12 +02:00

269 lines
7.8 KiB
C

/*
Unix SMB/CIFS implementation.
test suite for SMB2 credits
Copyright (C) Ralph Boehme 2017
This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#include "includes.h"
#include "libcli/smb2/smb2.h"
#include "libcli/smb2/smb2_calls.h"
#include "torture/torture.h"
#include "torture/smb2/proto.h"
#include "../libcli/smb/smbXcli_base.h"
#include "lib/param/param.h"
/**
* Request 64k credits in negprot/sessionsetup and require at least 8k
*
* This passes against Windows 2016
**/
static bool test_session_setup_credits_granted(struct torture_context *tctx,
struct smb2_tree *_tree)
{
struct smbcli_options options;
struct smb2_transport *transport = NULL;
struct smb2_tree *tree = NULL;
uint16_t cur_credits;
NTSTATUS status;
bool ret = true;
transport = _tree->session->transport;
options = transport->options;
status = smb2_logoff(_tree->session);
torture_assert_ntstatus_ok_goto(tctx, status, ret, done,
"smb2_logoff failed\n");
TALLOC_FREE(_tree);
options.max_credits = 65535;
ret = torture_smb2_connection_ext(tctx, 0, &options, &tree);
torture_assert_goto(tctx, ret == true, ret, done,
"torture_smb2_connection_ext failed\n");
transport = tree->session->transport;
cur_credits = smb2cli_conn_get_cur_credits(transport->conn);
if (cur_credits < 8192) {
torture_result(tctx, TORTURE_FAIL,
"Server only granted %" PRIu16" credits\n",
cur_credits);
ret = false;
goto done;
}
done:
TALLOC_FREE(tree);
return ret;
}
/**
* Request 64K credits in a single SMB2 request and requite at least 8192
*
* This passes against Windows 2016
**/
static bool test_single_req_credits_granted(struct torture_context *tctx,
struct smb2_tree *_tree)
{
struct smbcli_options options;
struct smb2_transport *transport = NULL;
struct smb2_tree *tree = NULL;
struct smb2_handle h = {{0}};
struct smb2_create create;
const char *fname = "single_req_credits_granted.dat";
uint16_t cur_credits;
NTSTATUS status;
bool ret = true;
smb2_util_unlink(_tree, fname);
transport = _tree->session->transport;
options = transport->options;
status = smb2_logoff(_tree->session);
torture_assert_ntstatus_ok_goto(tctx, status, ret, done,
"smb2_logoff failed\n");
TALLOC_FREE(_tree);
options.max_credits = 1;
ret = torture_smb2_connection_ext(tctx, 0, &options, &tree);
torture_assert_goto(tctx, ret == true, ret, done,
"torture_smb2_connection_ext failed\n");
transport = tree->session->transport;
cur_credits = smb2cli_conn_get_cur_credits(transport->conn);
if (cur_credits != 1) {
torture_result(tctx, TORTURE_FAIL,
"Only wanted 1 credit but server granted %" PRIu16"\n",
cur_credits);
ret = false;
goto done;
}
smb2cli_conn_set_max_credits(transport->conn, 65535);
ZERO_STRUCT(create);
create.in.impersonation_level = NTCREATEX_IMPERSONATION_IMPERSONATION;
create.in.desired_access = SEC_RIGHTS_FILE_ALL;
create.in.file_attributes = FILE_ATTRIBUTE_NORMAL;
create.in.create_disposition = NTCREATEX_DISP_OPEN_IF;
create.in.fname = fname;
status = smb2_create(tree, tctx, &create);
torture_assert_ntstatus_ok_goto(tctx, status, ret, done,
"smb2_create failed\n");
h = create.out.file.handle;
cur_credits = smb2cli_conn_get_cur_credits(transport->conn);
if (cur_credits < 8192) {
torture_result(tctx, TORTURE_FAIL,
"Server only granted %" PRIu16" credits\n",
cur_credits);
ret = false;
goto done;
}
done:
if (!smb2_util_handle_empty(h)) {
smb2_util_close(tree, h);
}
smb2_util_unlink(tree, fname);
TALLOC_FREE(tree);
return ret;
}
static bool test_crediting_skipped_mid(struct torture_context *tctx,
struct smb2_tree *_tree)
{
struct smbcli_options options;
struct smb2_transport *transport = NULL;
struct smb2_tree *tree = NULL;
struct smb2_handle h = {{0}};
struct smb2_create create;
const char *fname = "skipped_mid.dat";
uint64_t mid;
uint16_t cur_credits;
NTSTATUS status;
bool ret = true;
int i;
smb2_util_unlink(_tree, fname);
transport = _tree->session->transport;
options = transport->options;
status = smb2_logoff(_tree->session);
torture_assert_ntstatus_ok_goto(tctx, status, ret, done, "smb2_logoff failed\n");
TALLOC_FREE(_tree);
options.max_credits = 8192;
ret = torture_smb2_connection_ext(tctx, 0, &options, &tree);
torture_assert_goto(tctx, ret == true, ret, done, "torture_smb2_connection_ext failed\n");
transport = tree->session->transport;
cur_credits = smb2cli_conn_get_cur_credits(transport->conn);
if (cur_credits != 8192) {
torture_result(tctx, TORTURE_FAIL, "Server only granted %" PRIu16" credits\n", cur_credits);
ret = false;
goto done;
}
ZERO_STRUCT(create);
create.in.impersonation_level = NTCREATEX_IMPERSONATION_IMPERSONATION;
create.in.desired_access = SEC_RIGHTS_FILE_ALL;
create.in.file_attributes = FILE_ATTRIBUTE_NORMAL;
create.in.create_disposition = NTCREATEX_DISP_OPEN_IF;
create.in.fname = fname;
status = smb2_create(tree, tctx, &create);
torture_assert_ntstatus_ok_goto(tctx, status, ret, done, "smb2_create failed\n");
h = create.out.file.handle;
/*
* See what happens if we skip a mid. As we want to avoid triggering our
* client side mid window check we keep conn->smb2.cur_credits
* unchanged so the server keeps granting credits until it's max mid
* windows size is reached at which point it will disconnect us:
*
* o Windows 2016 currently has a maximum mid window size of 8192 by
* default
*
* o Samba's limit is 512
*
* o Windows 2008r2 uses some special algorithm (MS-SMB2 3.3.1.1
* footnote <167>) that kicks in once a mid is skipped, resulting in a
* maximum window size of 100-300 depending on the number of granted
* credits at the moment of skipping a mid.
*/
mid = smb2cli_conn_get_mid(tree->session->transport->conn);
smb2cli_conn_set_mid(tree->session->transport->conn, mid + 1);
for (i = 0; i < 8191; i++) {
status = smb2_util_write(tree, h, "\0", 0, 1);
if (!NT_STATUS_IS_OK(status)) {
torture_result(tctx, TORTURE_FAIL, "Server only allowed %d writes\n", i);
ret = false;
goto done;
}
}
/*
* Now use the skipped mid (the smb2_util_close...), we should
* immediately get a full mid window of size 8192.
*/
smb2cli_conn_set_mid(tree->session->transport->conn, mid);
status = smb2_util_close(tree, h);
torture_assert_ntstatus_ok_goto(tctx, status, ret, done, "smb2_close failed\n");
ZERO_STRUCT(h);
cur_credits = smb2cli_conn_get_cur_credits(transport->conn);
if (cur_credits != 8192) {
torture_result(tctx, TORTURE_FAIL, "Server only granted %" PRIu16" credits\n", cur_credits);
ret = false;
goto done;
}
smb2cli_conn_set_mid(tree->session->transport->conn, mid + 8192);
done:
if (!smb2_util_handle_empty(h)) {
smb2_util_close(tree, h);
}
smb2_util_unlink(tree, fname);
TALLOC_FREE(tree);
return ret;
}
struct torture_suite *torture_smb2_crediting_init(TALLOC_CTX *ctx)
{
struct torture_suite *suite = torture_suite_create(ctx, "credits");
torture_suite_add_1smb2_test(suite, "session_setup_credits_granted", test_session_setup_credits_granted);
torture_suite_add_1smb2_test(suite, "single_req_credits_granted", test_single_req_credits_granted);
torture_suite_add_1smb2_test(suite, "skipped_mid", test_crediting_skipped_mid);
suite->description = talloc_strdup(suite, "SMB2-CREDITS tests");
return suite;
}