mirror of
https://github.com/samba-team/samba.git
synced 2024-12-31 17:18:04 +03:00
beaa01e403
This doessn't work against Windows yet, and I've submitted a WSPP
request for clarification of the docs to try and find out
why. Meanwhile this is no worse than what we had, as it only gets used
when the server demands signing, and we didn't work then anyway.
(This used to be commit b788096add)
246 lines
6.2 KiB
C
246 lines
6.2 KiB
C
/*
|
|
Unix SMB/CIFS implementation.
|
|
|
|
SMB2 read test suite
|
|
|
|
Copyright (C) Andrew Tridgell 2008
|
|
|
|
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 "librpc/gen_ndr/ndr_security.h"
|
|
|
|
#define CHECK_STATUS(status, correct) do { \
|
|
if (!NT_STATUS_EQUAL(status, correct)) { \
|
|
printf("(%s) Incorrect status %s - should be %s\n", \
|
|
__location__, nt_errstr(status), nt_errstr(correct)); \
|
|
ret = false; \
|
|
goto done; \
|
|
}} while (0)
|
|
|
|
#define CHECK_VALUE(v, correct) do { \
|
|
if ((v) != (correct)) { \
|
|
printf("(%s) Incorrect value %s=%u - should be %u\n", \
|
|
__location__, #v, (unsigned)v, (unsigned)correct); \
|
|
ret = false; \
|
|
goto done; \
|
|
}} while (0)
|
|
|
|
#define FNAME "smb2_readtest.dat"
|
|
#define DNAME "smb2_readtest.dir"
|
|
|
|
static bool test_read_eof(struct torture_context *torture, struct smb2_tree *tree)
|
|
{
|
|
bool ret = true;
|
|
NTSTATUS status;
|
|
struct smb2_handle h;
|
|
uint8_t buf[70000];
|
|
struct smb2_read rd;
|
|
TALLOC_CTX *tmp_ctx = talloc_new(tree);
|
|
|
|
ZERO_STRUCT(buf);
|
|
|
|
status = torture_smb2_testfile(tree, FNAME, &h);
|
|
CHECK_STATUS(status, NT_STATUS_OK);
|
|
|
|
status = smb2_util_write(tree, h, buf, 0, ARRAY_SIZE(buf));
|
|
CHECK_STATUS(status, NT_STATUS_OK);
|
|
|
|
ZERO_STRUCT(rd);
|
|
rd.in.file.handle = h;
|
|
rd.in.length = 10;
|
|
rd.in.offset = 0;
|
|
rd.in.min_count = 1;
|
|
|
|
status = smb2_read(tree, tmp_ctx, &rd);
|
|
CHECK_STATUS(status, NT_STATUS_OK);
|
|
CHECK_VALUE(rd.out.data.length, 10);
|
|
|
|
rd.in.min_count = 0;
|
|
rd.in.length = 10;
|
|
rd.in.offset = sizeof(buf);
|
|
status = smb2_read(tree, tmp_ctx, &rd);
|
|
CHECK_STATUS(status, NT_STATUS_END_OF_FILE);
|
|
|
|
rd.in.min_count = 0;
|
|
rd.in.length = 0;
|
|
rd.in.offset = sizeof(buf);
|
|
status = smb2_read(tree, tmp_ctx, &rd);
|
|
CHECK_STATUS(status, NT_STATUS_OK);
|
|
CHECK_VALUE(rd.out.data.length, 0);
|
|
|
|
rd.in.min_count = 1;
|
|
rd.in.length = 0;
|
|
rd.in.offset = sizeof(buf);
|
|
status = smb2_read(tree, tmp_ctx, &rd);
|
|
CHECK_STATUS(status, NT_STATUS_END_OF_FILE);
|
|
|
|
rd.in.min_count = 0;
|
|
rd.in.length = 2;
|
|
rd.in.offset = sizeof(buf) - 1;
|
|
status = smb2_read(tree, tmp_ctx, &rd);
|
|
CHECK_STATUS(status, NT_STATUS_OK);
|
|
CHECK_VALUE(rd.out.data.length, 1);
|
|
|
|
rd.in.min_count = 2;
|
|
rd.in.length = 1;
|
|
rd.in.offset = sizeof(buf) - 1;
|
|
status = smb2_read(tree, tmp_ctx, &rd);
|
|
CHECK_STATUS(status, NT_STATUS_END_OF_FILE);
|
|
|
|
rd.in.min_count = 0x10000;
|
|
rd.in.length = 1;
|
|
rd.in.offset = 0;
|
|
status = smb2_read(tree, tmp_ctx, &rd);
|
|
CHECK_STATUS(status, NT_STATUS_END_OF_FILE);
|
|
|
|
rd.in.min_count = 0x10000 - 2;
|
|
rd.in.length = 1;
|
|
rd.in.offset = 0;
|
|
status = smb2_read(tree, tmp_ctx, &rd);
|
|
CHECK_STATUS(status, NT_STATUS_END_OF_FILE);
|
|
|
|
rd.in.min_count = 10;
|
|
rd.in.length = 5;
|
|
rd.in.offset = 0;
|
|
status = smb2_read(tree, tmp_ctx, &rd);
|
|
CHECK_STATUS(status, NT_STATUS_END_OF_FILE);
|
|
|
|
done:
|
|
talloc_free(tmp_ctx);
|
|
return ret;
|
|
}
|
|
|
|
|
|
static bool test_read_position(struct torture_context *torture, struct smb2_tree *tree)
|
|
{
|
|
bool ret = true;
|
|
NTSTATUS status;
|
|
struct smb2_handle h;
|
|
uint8_t buf[70000];
|
|
struct smb2_read rd;
|
|
TALLOC_CTX *tmp_ctx = talloc_new(tree);
|
|
union smb_fileinfo info;
|
|
|
|
ZERO_STRUCT(buf);
|
|
|
|
status = torture_smb2_testfile(tree, FNAME, &h);
|
|
CHECK_STATUS(status, NT_STATUS_OK);
|
|
|
|
status = smb2_util_write(tree, h, buf, 0, ARRAY_SIZE(buf));
|
|
CHECK_STATUS(status, NT_STATUS_OK);
|
|
|
|
ZERO_STRUCT(rd);
|
|
rd.in.file.handle = h;
|
|
rd.in.length = 10;
|
|
rd.in.offset = 0;
|
|
rd.in.min_count = 1;
|
|
|
|
status = smb2_read(tree, tmp_ctx, &rd);
|
|
CHECK_STATUS(status, NT_STATUS_OK);
|
|
CHECK_VALUE(rd.out.data.length, 10);
|
|
|
|
info.generic.level = RAW_FILEINFO_SMB2_ALL_INFORMATION;
|
|
info.generic.in.file.handle = h;
|
|
|
|
status = smb2_getinfo_file(tree, tmp_ctx, &info);
|
|
CHECK_STATUS(status, NT_STATUS_OK);
|
|
if (torture_setting_bool(torture, "windows", false)) {
|
|
CHECK_VALUE(info.all_info2.out.position, 0);
|
|
} else {
|
|
CHECK_VALUE(info.all_info2.out.position, 10);
|
|
}
|
|
|
|
|
|
done:
|
|
talloc_free(tmp_ctx);
|
|
return ret;
|
|
}
|
|
|
|
static bool test_read_dir(struct torture_context *torture, struct smb2_tree *tree)
|
|
{
|
|
bool ret = true;
|
|
NTSTATUS status;
|
|
struct smb2_handle h;
|
|
struct smb2_read rd;
|
|
TALLOC_CTX *tmp_ctx = talloc_new(tree);
|
|
|
|
status = torture_smb2_testdir(tree, DNAME, &h);
|
|
if (!NT_STATUS_IS_OK(status)) {
|
|
printf(__location__ " Unable to create test directory '%s' - %s\n", DNAME, nt_errstr(status));
|
|
return false;
|
|
}
|
|
|
|
ZERO_STRUCT(rd);
|
|
rd.in.file.handle = h;
|
|
rd.in.length = 10;
|
|
rd.in.offset = 0;
|
|
rd.in.min_count = 1;
|
|
|
|
status = smb2_read(tree, tmp_ctx, &rd);
|
|
CHECK_STATUS(status, NT_STATUS_INVALID_DEVICE_REQUEST);
|
|
|
|
rd.in.min_count = 11;
|
|
status = smb2_read(tree, tmp_ctx, &rd);
|
|
CHECK_STATUS(status, NT_STATUS_INVALID_DEVICE_REQUEST);
|
|
|
|
rd.in.length = 0;
|
|
rd.in.min_count = 2592;
|
|
status = smb2_read(tree, tmp_ctx, &rd);
|
|
if (torture_setting_bool(torture, "windows", false)) {
|
|
CHECK_STATUS(status, NT_STATUS_END_OF_FILE);
|
|
} else {
|
|
CHECK_STATUS(status, NT_STATUS_INVALID_DEVICE_REQUEST);
|
|
}
|
|
|
|
rd.in.length = 0;
|
|
rd.in.min_count = 0;
|
|
rd.in.channel = 0;
|
|
status = smb2_read(tree, tmp_ctx, &rd);
|
|
if (torture_setting_bool(torture, "windows", false)) {
|
|
CHECK_STATUS(status, NT_STATUS_OK);
|
|
} else {
|
|
CHECK_STATUS(status, NT_STATUS_INVALID_DEVICE_REQUEST);
|
|
}
|
|
|
|
done:
|
|
talloc_free(tmp_ctx);
|
|
return ret;
|
|
}
|
|
|
|
|
|
/*
|
|
basic testing of SMB2 read
|
|
*/
|
|
struct torture_suite *torture_smb2_read_init(void)
|
|
{
|
|
struct torture_suite *suite = torture_suite_create(talloc_autofree_context(), "READ");
|
|
|
|
torture_suite_add_1smb2_test(suite, "EOF", test_read_eof);
|
|
torture_suite_add_1smb2_test(suite, "POSITION", test_read_position);
|
|
torture_suite_add_1smb2_test(suite, "DIR", test_read_dir);
|
|
|
|
suite->description = talloc_strdup(suite, "SMB2-READ tests");
|
|
|
|
return suite;
|
|
}
|
|
|