mirror of
https://github.com/samba-team/samba.git
synced 2025-03-11 16:58:40 +03:00
s4:torture: Add IPC test for 'smb2 max async credits'
BUG: https://bugzilla.samba.org/show_bug.cgi?id=14430 Pair-Programmed-With: Stefan Metzmacher <metze@samba.org> Signed-off-by: Stefan Metzmacher <metze@samba.org> Signed-off-by: Andreas Schneider <asn@samba.org>
This commit is contained in:
Andreas Schneider
Andreas Schneider
parent
6cb46e5cb6
commit
3884a7c085
1
selftest/knownfail.d/smb2_credits
Normal file
1
selftest/knownfail.d/smb2_credits
Normal file
@ -0,0 +1 @@
|
||||
samba3.smb2.credits.*1conn_ipc_max_async_credits
|
||||
@ -20,12 +20,17 @@
|
||||
*/
|
||||
|
||||
#include "includes.h"
|
||||
#include "torture/torture.h"
|
||||
|
||||
#include "lib/cmdline/cmdline.h"
|
||||
#include "lib/param/param.h"
|
||||
#include "libcli/resolve/resolve.h"
|
||||
#include "libcli/smb/smbXcli_base.h"
|
||||
#include "libcli/smb2/smb2.h"
|
||||
#include "libcli/smb2/smb2_calls.h"
|
||||
#include "torture/torture.h"
|
||||
#include "librpc/gen_ndr/ndr_lsa.h"
|
||||
#include "torture/smb2/proto.h"
|
||||
#include "../libcli/smb/smbXcli_base.h"
|
||||
#include "lib/param/param.h"
|
||||
#include "util/tevent_ntstatus.h"
|
||||
|
||||
/**
|
||||
* Request 64k credits in negprot/sessionsetup and require at least 8k
|
||||
@ -254,6 +259,482 @@ done:
|
||||
return ret;
|
||||
}
|
||||
|
||||
#define SMBXCLI_NP_DESIRED_ACCESS \
|
||||
(SEC_STD_READ_CONTROL | SEC_FILE_READ_DATA | SEC_FILE_WRITE_DATA | \
|
||||
SEC_FILE_APPEND_DATA | SEC_FILE_READ_EA | SEC_FILE_WRITE_EA | \
|
||||
SEC_FILE_READ_ATTRIBUTE | SEC_FILE_WRITE_ATTRIBUTE | 0)
|
||||
|
||||
struct test_ipc_async_credits_loop;
|
||||
|
||||
struct test_ipc_async_credits_state {
|
||||
struct torture_context *tctx;
|
||||
struct smbXcli_conn *conn;
|
||||
struct smbXcli_session *session;
|
||||
struct smbXcli_tcon *tcon;
|
||||
uint32_t timeout_msec;
|
||||
uint16_t pid;
|
||||
const char *pipe_name;
|
||||
|
||||
size_t num_loops;
|
||||
struct test_ipc_async_credits_loop *loops;
|
||||
size_t num_status_received;
|
||||
|
||||
size_t num_status_pending;
|
||||
size_t num_status_insufficient;
|
||||
|
||||
bool stop;
|
||||
};
|
||||
|
||||
struct test_ipc_async_credits_loop {
|
||||
size_t idx;
|
||||
struct test_ipc_async_credits_state *state;
|
||||
struct tevent_req *req;
|
||||
size_t num_started;
|
||||
|
||||
uint32_t max_data;
|
||||
uint64_t fid_persistent;
|
||||
uint64_t fid_volatile;
|
||||
|
||||
struct tevent_immediate *im;
|
||||
|
||||
NTSTATUS status;
|
||||
};
|
||||
|
||||
static void test_ipc_async_credits_read_loop_do(
|
||||
struct test_ipc_async_credits_loop *loop);
|
||||
|
||||
static void test_ipc_async_credits_read_loop_start(struct tevent_context *ctx,
|
||||
struct tevent_immediate *im,
|
||||
void *private_data)
|
||||
{
|
||||
struct test_ipc_async_credits_loop *loop = private_data;
|
||||
|
||||
test_ipc_async_credits_read_loop_do(loop);
|
||||
}
|
||||
|
||||
static void test_ipc_async_credits_read_loop_done(struct tevent_req *req);
|
||||
|
||||
static void test_ipc_async_credits_read_loop_do(
|
||||
struct test_ipc_async_credits_loop *loop)
|
||||
{
|
||||
struct test_ipc_async_credits_state *state = loop->state;
|
||||
bool ok;
|
||||
|
||||
loop->num_started += 1;
|
||||
loop->req = smb2cli_read_send(state->loops,
|
||||
state->tctx->ev,
|
||||
state->conn,
|
||||
state->timeout_msec,
|
||||
state->session,
|
||||
state->tcon,
|
||||
loop->max_data, /* length */
|
||||
0, /* offset */
|
||||
loop->fid_persistent,
|
||||
loop->fid_volatile,
|
||||
0, /* minimum_count */
|
||||
0); /* remaining_bytes */
|
||||
torture_assert_not_null_goto(
|
||||
state->tctx, loop->req, ok, asserted, "smb2cli_read_send");
|
||||
smb2cli_read_set_notify_async(loop->req);
|
||||
tevent_req_set_callback(loop->req,
|
||||
test_ipc_async_credits_read_loop_done,
|
||||
loop);
|
||||
|
||||
return;
|
||||
asserted:
|
||||
(void)ok;
|
||||
state->stop = true;
|
||||
}
|
||||
|
||||
static void test_ipc_async_credits_read_loop_done(struct tevent_req *req)
|
||||
{
|
||||
struct test_ipc_async_credits_loop *loop = _tevent_req_callback_data(
|
||||
req);
|
||||
struct test_ipc_async_credits_state *state = loop->state;
|
||||
TALLOC_CTX *frame = talloc_stackframe();
|
||||
uint8_t *data = NULL;
|
||||
uint32_t data_length = 0;
|
||||
bool ok;
|
||||
bool was_pending = false;
|
||||
bool in_progress = false;
|
||||
|
||||
torture_assert_goto(state->tctx,
|
||||
loop->req == req,
|
||||
ok,
|
||||
asserted,
|
||||
__location__);
|
||||
|
||||
if (NT_STATUS_EQUAL(loop->status, NT_STATUS_PENDING)) {
|
||||
was_pending = true;
|
||||
}
|
||||
|
||||
loop->status = smb2cli_read_recv(req, frame, &data, &data_length);
|
||||
#if 0
|
||||
torture_comment(state->tctx,
|
||||
"loop->status: %s\n",
|
||||
nt_errstr(loop->status));
|
||||
#endif
|
||||
in_progress = tevent_req_is_in_progress(req);
|
||||
state->num_status_received += 1;
|
||||
|
||||
if (NT_STATUS_EQUAL(loop->status, NT_STATUS_PENDING)) {
|
||||
state->num_status_pending += 1;
|
||||
torture_assert_goto(state->tctx,
|
||||
in_progress,
|
||||
ok,
|
||||
asserted,
|
||||
__location__);
|
||||
goto done;
|
||||
}
|
||||
loop->req = NULL;
|
||||
/* We will cancel the request, don't free it. */
|
||||
torture_assert_goto(state->tctx,
|
||||
!in_progress,
|
||||
ok,
|
||||
asserted,
|
||||
__location__);
|
||||
if (was_pending &&
|
||||
NT_STATUS_EQUAL(loop->status, NT_STATUS_CANCELLED))
|
||||
{
|
||||
goto done;
|
||||
}
|
||||
torture_assert_ntstatus_equal_goto(state->tctx,
|
||||
loop->status,
|
||||
NT_STATUS_INSUFFICIENT_RESOURCES,
|
||||
ok,
|
||||
asserted,
|
||||
__location__);
|
||||
state->num_status_insufficient += 1;
|
||||
|
||||
done:
|
||||
TALLOC_FREE(frame);
|
||||
return;
|
||||
asserted:
|
||||
(void)ok;
|
||||
state->stop = true;
|
||||
TALLOC_FREE(frame);
|
||||
}
|
||||
|
||||
static const uint8_t dcerpc_bind_lsa_bytes[] = {
|
||||
0x05, 0x00, 0x0b, 0x03, 0x10, 0x00, 0x00, 0x00,
|
||||
0xa0, 0x00, 0x00, 0x00, 0x02, 0x00, 0x00, 0x00,
|
||||
0xb8, 0x10, 0xb8, 0x10, 0x00, 0x00, 0x00, 0x00,
|
||||
0x03, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01, 0x00,
|
||||
0x78, 0x57, 0x34, 0x12, 0x34, 0x12, 0xcd, 0xab,
|
||||
0xef, 0x00, 0x01, 0x23, 0x45, 0x67, 0x89, 0xab,
|
||||
0x00, 0x00, 0x00, 0x00, 0x04, 0x5d, 0x88, 0x8a,
|
||||
0xeb, 0x1c, 0xc9, 0x11, 0x9f, 0xe8, 0x08, 0x00,
|
||||
0x2b, 0x10, 0x48, 0x60, 0x02, 0x00, 0x00, 0x00,
|
||||
0x01, 0x00, 0x01, 0x00, 0x78, 0x57, 0x34, 0x12,
|
||||
0x34, 0x12, 0xcd, 0xab, 0xef, 0x00, 0x01, 0x23,
|
||||
0x45, 0x67, 0x89, 0xab, 0x00, 0x00, 0x00, 0x00,
|
||||
0x33, 0x05, 0x71, 0x71, 0xba, 0xbe, 0x37, 0x49,
|
||||
0x83, 0x19, 0xb5, 0xdb, 0xef, 0x9c, 0xcc, 0x36,
|
||||
0x01, 0x00, 0x00, 0x00, 0x02, 0x00, 0x01, 0x00,
|
||||
0x78, 0x57, 0x34, 0x12, 0x34, 0x12, 0xcd, 0xab,
|
||||
0xef, 0x00, 0x01, 0x23, 0x45, 0x67, 0x89, 0xab,
|
||||
0x00, 0x00, 0x00, 0x00, 0x2c, 0x1c, 0xb7, 0x6c,
|
||||
0x12, 0x98, 0x40, 0x45, 0x03, 0x00, 0x00, 0x00,
|
||||
0x00, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00
|
||||
};
|
||||
|
||||
static bool test_ipc_max_async_credits(struct torture_context *tctx,
|
||||
struct smb2_tree *trees[],
|
||||
size_t num_trees,
|
||||
size_t num_loops)
|
||||
{
|
||||
struct test_ipc_async_credits_state *states[num_trees];
|
||||
bool stop_loop = false;
|
||||
NTSTATUS status;
|
||||
size_t i, t;
|
||||
bool ok;
|
||||
|
||||
for (t = 0; t < num_trees; t++) {
|
||||
uint16_t cur_credits;
|
||||
|
||||
states[t] = NULL;
|
||||
|
||||
cur_credits = smb2cli_conn_get_cur_credits(
|
||||
trees[t]->session->transport->conn);
|
||||
torture_assert_int_equal_goto(
|
||||
tctx,
|
||||
cur_credits,
|
||||
num_loops,
|
||||
ok,
|
||||
out,
|
||||
"Invalid number of granted credits");
|
||||
}
|
||||
|
||||
torture_assert_int_not_equal_goto(
|
||||
tctx, num_trees, 0, ok, out, "Invalid number of trees");
|
||||
|
||||
for (i = 0; i < num_trees; i++) {
|
||||
struct smb2_tree *tree = trees[i];
|
||||
struct test_ipc_async_credits_state *state = NULL;
|
||||
|
||||
state = talloc_zero(tctx,
|
||||
struct test_ipc_async_credits_state);
|
||||
torture_assert_not_null_goto(
|
||||
tctx, state, ok, out, "tevent_zero failed");
|
||||
|
||||
state->tctx = tctx;
|
||||
state->conn = tree->session->transport->conn;
|
||||
state->session = tree->session->smbXcli;
|
||||
state->tcon = tree->smbXcli;
|
||||
state->timeout_msec = 0; /* 10 * 1000; */
|
||||
state->pid = 0;
|
||||
state->pipe_name = NDR_LSARPC_NAME;
|
||||
|
||||
state->num_loops = num_loops;
|
||||
state->loops = talloc_zero_array(
|
||||
state,
|
||||
struct test_ipc_async_credits_loop,
|
||||
state->num_loops);
|
||||
torture_assert_not_null_goto(tctx,
|
||||
state->loops,
|
||||
ok,
|
||||
out,
|
||||
"tevent_zero_array failed");
|
||||
|
||||
states[i] = state;
|
||||
}
|
||||
|
||||
for (i = 0; i < num_loops; i++) {
|
||||
for (t = 0; t < num_trees; t++) {
|
||||
struct test_ipc_async_credits_state
|
||||
*state = states[t];
|
||||
struct test_ipc_async_credits_loop
|
||||
*loop = &state->loops[i];
|
||||
DATA_BLOB in_input_buffer = data_blob_const(
|
||||
dcerpc_bind_lsa_bytes,
|
||||
ARRAY_SIZE(dcerpc_bind_lsa_bytes));
|
||||
DATA_BLOB in_output_buffer = data_blob_null;
|
||||
DATA_BLOB out_input_buffer = data_blob_null;
|
||||
DATA_BLOB out_output_buffer = data_blob_null;
|
||||
|
||||
loop->idx = i;
|
||||
loop->state = state;
|
||||
loop->status = NT_STATUS_UNSUCCESSFUL;
|
||||
loop->max_data = 1024;
|
||||
|
||||
loop->im = tevent_create_immediate(state->loops);
|
||||
torture_assert_not_null_goto(
|
||||
tctx,
|
||||
loop->im,
|
||||
ok,
|
||||
out,
|
||||
"tevent_create_immediate failed");
|
||||
|
||||
status = smb2cli_create(
|
||||
state->conn,
|
||||
state->timeout_msec,
|
||||
state->session,
|
||||
state->tcon,
|
||||
state->pipe_name,
|
||||
SMB2_OPLOCK_LEVEL_NONE,
|
||||
SMB2_IMPERSONATION_IMPERSONATION,
|
||||
SMBXCLI_NP_DESIRED_ACCESS,
|
||||
0, /* file_attributes */
|
||||
FILE_SHARE_READ | FILE_SHARE_WRITE,
|
||||
FILE_OPEN,
|
||||
0, /* create_options */
|
||||
NULL, /* blobs */
|
||||
&loop->fid_persistent,
|
||||
&loop->fid_volatile,
|
||||
NULL, /* cr */
|
||||
state, /* mem_ctx */
|
||||
NULL, /* ret_blobs */
|
||||
NULL); /* psymlink */
|
||||
torture_assert_ntstatus_ok_goto(
|
||||
tctx, status, ok, out, "smb2cli_create failed");
|
||||
|
||||
status = smb2cli_ioctl(
|
||||
state->conn,
|
||||
state->timeout_msec,
|
||||
state->session,
|
||||
state->tcon,
|
||||
loop->fid_persistent,
|
||||
loop->fid_volatile,
|
||||
FSCTL_NAMED_PIPE_READ_WRITE,
|
||||
0, /* in_max_input_length */
|
||||
&in_input_buffer,
|
||||
1024, /* in_max_output_length */
|
||||
&in_output_buffer,
|
||||
SMB2_IOCTL_FLAG_IS_FSCTL,
|
||||
state, /* mem_ctx */
|
||||
&out_input_buffer,
|
||||
&out_output_buffer);
|
||||
torture_assert_ntstatus_ok_goto(
|
||||
tctx,
|
||||
status,
|
||||
ok,
|
||||
out,
|
||||
"FSCTL_NAMED_PIPE_READ_WRITE failed");
|
||||
|
||||
tevent_schedule_immediate(
|
||||
loop->im,
|
||||
tctx->ev,
|
||||
test_ipc_async_credits_read_loop_start,
|
||||
loop);
|
||||
}
|
||||
}
|
||||
|
||||
/* Loop to send and receive packets */
|
||||
while (!stop_loop) {
|
||||
size_t loops_ready = 0;
|
||||
int rc;
|
||||
|
||||
rc = tevent_loop_once(tctx->ev);
|
||||
torture_assert_int_equal_goto(
|
||||
tctx, rc, 0, ok, out, "tevent_loop_once");
|
||||
|
||||
for (i = 0; i < num_trees; i++) {
|
||||
struct test_ipc_async_credits_state
|
||||
*state = states[i];
|
||||
if (state->stop) {
|
||||
stop_loop = true;
|
||||
}
|
||||
|
||||
if (state->num_status_received >= state->num_loops) {
|
||||
loops_ready += 1;
|
||||
}
|
||||
}
|
||||
|
||||
if (loops_ready >= num_trees) {
|
||||
stop_loop = true;
|
||||
}
|
||||
}
|
||||
|
||||
for (t = 0; t < num_trees; t++) {
|
||||
struct test_ipc_async_credits_state *state = states[t];
|
||||
size_t max_async_credits = 512;
|
||||
size_t max_credits = max_async_credits + 2;
|
||||
|
||||
torture_assert_goto(state->tctx, !state->stop, ok, out, "");
|
||||
torture_assert_int_equal_goto(state->tctx,
|
||||
state->num_status_received,
|
||||
state->num_loops,
|
||||
ok,
|
||||
out,
|
||||
"");
|
||||
torture_assert_int_equal_goto(state->tctx,
|
||||
state->num_status_pending,
|
||||
max_async_credits - 1,
|
||||
ok,
|
||||
out,
|
||||
"");
|
||||
torture_assert_int_equal_goto(state->tctx,
|
||||
state->num_status_insufficient,
|
||||
max_credits - max_async_credits + 1,
|
||||
ok,
|
||||
out,
|
||||
"");
|
||||
}
|
||||
|
||||
for (t = 0; t < num_trees; t++) {
|
||||
struct test_ipc_async_credits_state *state = states[t];
|
||||
|
||||
for (i = 0; i < num_loops; i++) {
|
||||
struct test_ipc_async_credits_loop
|
||||
*loop = &state->loops[i];
|
||||
if (loop->req != NULL) {
|
||||
ok = tevent_req_cancel(loop->req);
|
||||
torture_assert_goto(tctx,
|
||||
ok,
|
||||
ok,
|
||||
out,
|
||||
"tevent_req_cancel failed");
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
for (t = 0; t < num_trees; t++) {
|
||||
struct test_ipc_async_credits_state *state = states[t];
|
||||
|
||||
for (i = 0; i < num_loops; i++) {
|
||||
struct test_ipc_async_credits_loop
|
||||
*loop = &state->loops[i];
|
||||
|
||||
if (NT_STATUS_EQUAL(loop->status,
|
||||
NT_STATUS_INSUFFICIENT_RESOURCES))
|
||||
{
|
||||
continue;
|
||||
}
|
||||
|
||||
if (loop->req != NULL) {
|
||||
ok = tevent_req_poll(loop->req, tctx->ev);
|
||||
torture_assert_goto(state->tctx,
|
||||
ok,
|
||||
ok,
|
||||
out,
|
||||
"tevent_req_poll failed");
|
||||
}
|
||||
torture_assert_ntstatus_equal_goto(
|
||||
tctx,
|
||||
loop->status,
|
||||
NT_STATUS_CANCELLED,
|
||||
ok,
|
||||
out,
|
||||
__location__);
|
||||
}
|
||||
}
|
||||
|
||||
ok = true;
|
||||
out:
|
||||
for (t = 0; t < num_trees; t++) {
|
||||
TALLOC_FREE(states[t]);
|
||||
}
|
||||
|
||||
return ok;
|
||||
}
|
||||
|
||||
static bool test_1conn_ipc_max_async_credits(struct torture_context *tctx,
|
||||
struct smb2_tree *tree0)
|
||||
{
|
||||
struct smb2_transport *transport0 = tree0->session->transport;
|
||||
struct smbcli_options options = transport0->options;
|
||||
struct smb2_tree *tree1 = NULL;
|
||||
struct smb2_tree *trees[1] = {};
|
||||
const char *host = torture_setting_string(tctx, "host", NULL);
|
||||
const char *share = "IPC$";
|
||||
/* We want to check if we IPC connections add to "max async credits" */
|
||||
uint16_t max_async_credits = torture_setting_int(
|
||||
tctx,
|
||||
"maxasynccredits",
|
||||
512 /* lpcfg_smb2_max_async_credits(tctx->lp_ctx) */);
|
||||
bool ok = false;
|
||||
NTSTATUS status;
|
||||
uint16_t max_credits = max_async_credits + 2;
|
||||
|
||||
options.client_guid = GUID_random();
|
||||
options.max_credits = max_credits;
|
||||
|
||||
/* Create connection to IPC$ */
|
||||
status = smb2_connect(tctx,
|
||||
host,
|
||||
lpcfg_smb_ports(tctx->lp_ctx),
|
||||
share,
|
||||
lpcfg_resolve_context(tctx->lp_ctx),
|
||||
samba_cmdline_get_creds(),
|
||||
&tree1,
|
||||
tctx->ev,
|
||||
&options,
|
||||
lpcfg_socket_options(tctx->lp_ctx),
|
||||
lpcfg_gensec_settings(tctx, tctx->lp_ctx));
|
||||
torture_assert_ntstatus_ok_goto(
|
||||
tctx, status, ok, out, "smb2_connect failed");
|
||||
|
||||
trees[0] = tree1;
|
||||
ok = test_ipc_max_async_credits(tctx,
|
||||
trees,
|
||||
ARRAY_SIZE(trees),
|
||||
max_credits);
|
||||
out:
|
||||
TALLOC_FREE(tree1);
|
||||
|
||||
return ok;
|
||||
}
|
||||
|
||||
struct torture_suite *torture_smb2_crediting_init(TALLOC_CTX *ctx)
|
||||
{
|
||||
struct torture_suite *suite = torture_suite_create(ctx, "credits");
|
||||
@ -262,6 +743,10 @@ struct torture_suite *torture_smb2_crediting_init(TALLOC_CTX *ctx)
|
||||
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);
|
||||
|
||||
torture_suite_add_1smb2_test(suite,
|
||||
"1conn_ipc_max_async_credits",
|
||||
test_1conn_ipc_max_async_credits);
|
||||
|
||||
suite->description = talloc_strdup(suite, "SMB2-CREDITS tests");
|
||||
|
||||
return suite;
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user