The branch, master has been updated via 6c2c189 s4-torture: smb2 copychunk data integrity checks via c9d1303 s4-torture: add smb2 copychunk ioctl tests via 2dad001 s4-torture: split common copychunk ioctl test code into helpers via 8abaafa s4-torture: Add tests for the smb2 copychunk ioctl via 2e1168d idl: add copychunk and request_resume_key ioctl definitions from 63c7107 s3:smbd: also the parent smbd needs FLAG_MSG_PRINT_GENERAL (bug #8553)
http://gitweb.samba.org/?p=samba.git;a=shortlog;h=master - Log ----------------------------------------------------------------- commit 6c2c189caafe90659a8f43d338523f0cbb0a1505 Author: David Disseldorp <dd...@suse.de> Date: Tue Oct 11 11:44:03 2011 +0200 s4-torture: smb2 copychunk data integrity checks Perform simple integrity checks on destination file data following a successful copychunk request. Signed-off-by: Stefan Metzmacher <me...@samba.org> Autobuild-User: David Disseldorp <dd...@samba.org> Autobuild-Date: Mon Oct 31 19:28:20 CET 2011 on sn-devel-104 commit c9d1303734b003bb37e1b7a5ae437c3ce20b58c9 Author: David Disseldorp <dd...@suse.de> Date: Fri Sep 30 14:33:36 2011 +0200 s4-torture: add smb2 copychunk ioctl tests test_ioctl_copy_chunk_multi: Issue a copychunk request with multiple chunk descriptors. test_ioctl_copy_chunk_over: Issue a copychunk request with overlapping chunks. test_ioctl_copy_chunk_append: append data as part of a copy chunk test_ioctl_copy_chunk_tiny: issue request with sub FS block size chunk descriptors. Signed-off-by: Stefan Metzmacher <me...@samba.org> commit 2dad001e05ba46c1e6cde28a04d5c9e6e39ac304 Author: David Disseldorp <dd...@suse.de> Date: Fri Sep 30 12:50:36 2011 +0200 s4-torture: split common copychunk ioctl test code into helpers Signed-off-by: Stefan Metzmacher <me...@samba.org> commit 8abaafadde887a22e803f93b3b801b732495ad98 Author: David Disseldorp <dd...@suse.de> Date: Tue Sep 27 16:40:20 2011 +0200 s4-torture: Add tests for the smb2 copychunk ioctl Add idls for parsing of copychunk ioctl args. Signed-off-by: Stefan Metzmacher <me...@samba.org> commit 2e1168d0a84d49b7858725d580491e970d6b4f64 Author: David Disseldorp <dd...@suse.de> Date: Wed Sep 28 00:37:54 2011 +0200 idl: add copychunk and request_resume_key ioctl definitions ----------------------------------------------------------------------- Summary of changes: librpc/idl/ioctl.idl | 33 ++ librpc/idl/wscript_build | 2 +- librpc/wscript_build | 5 + source4/torture/smb2/ioctl.c | 584 +++++++++++++++++++++++++++++++++++- source4/torture/smb2/wscript_build | 2 +- 5 files changed, 622 insertions(+), 4 deletions(-) create mode 100644 librpc/idl/ioctl.idl Changeset truncated at 500 lines: diff --git a/librpc/idl/ioctl.idl b/librpc/idl/ioctl.idl new file mode 100644 index 0000000..1c319ea --- /dev/null +++ b/librpc/idl/ioctl.idl @@ -0,0 +1,33 @@ +#include "idl_types.h" +[ + pointer_default(unique) +] +interface copychunk +{ + typedef [public] struct { + uint8 resume_key[24]; + uint32 context_len; + /* <56> Windows sends 4 bytes of zero for the context field. */ + uint8 context[4]; + } req_resume_key_rsp; + + typedef struct { + hyper source_off; + hyper target_off; + uint32 length; + uint32 reserved; + } srv_copychunk; + + typedef [public] struct { + uint8 source_key[24]; + uint32 chunk_count; + uint32 reserved; + srv_copychunk chunks[chunk_count]; + } srv_copychunk_copy; + + typedef [public] struct { + uint32 chunks_written; + uint32 chunk_bytes_written; + uint32 total_bytes_written; + } srv_copychunk_rsp; +} diff --git a/librpc/idl/wscript_build b/librpc/idl/wscript_build index ffb7a9c..d2112b4 100644 --- a/librpc/idl/wscript_build +++ b/librpc/idl/wscript_build @@ -27,7 +27,7 @@ bld.SAMBA_PIDL_LIST('PIDL', output_dir='../gen_ndr') bld.SAMBA_PIDL_LIST('PIDL', - 'rap.idl ntprinting.idl preg.idl', + 'rap.idl ntprinting.idl preg.idl ioctl.idl', options='--header --ndr-parser', output_dir='../gen_ndr') diff --git a/librpc/wscript_build b/librpc/wscript_build index 14289a1..dc35de5 100644 --- a/librpc/wscript_build +++ b/librpc/wscript_build @@ -563,6 +563,11 @@ bld.SAMBA_SUBSYSTEM('RPC_NDR_DNSSERVER', public_deps='dcerpc-binding ndr-standard' ) +bld.SAMBA_SUBSYSTEM('NDR_IOCTL', + source='gen_ndr/ndr_ioctl.c', + public_deps='ndr' + ) + # a grouping library for NDR subsystems that may be used by more than one target bld.SAMBA_LIBRARY('ndr-samba', source=[], diff --git a/source4/torture/smb2/ioctl.c b/source4/torture/smb2/ioctl.c index 169a001..a54702d 100644 --- a/source4/torture/smb2/ioctl.c +++ b/source4/torture/smb2/ioctl.c @@ -25,8 +25,10 @@ #include "libcli/smb2/smb2_calls.h" #include "torture/torture.h" #include "torture/smb2/proto.h" +#include "librpc/gen_ndr/ndr_ioctl.h" -#define FNAME "testfsctl.dat" +#define FNAME "testfsctl.dat" +#define FNAME2 "testfsctl2.dat" /* basic testing of SMB2 shadow copy calls @@ -72,13 +74,591 @@ static bool test_ioctl_get_shadow_copy(struct torture_context *torture, } /* + basic testing of the SMB2 server side copy ioctls +*/ +static bool test_ioctl_req_resume_key(struct torture_context *torture, + struct smb2_tree *tree) +{ + struct smb2_handle h; + uint8_t buf[100]; + NTSTATUS status; + union smb_ioctl ioctl; + TALLOC_CTX *tmp_ctx = talloc_new(tree); + struct req_resume_key_rsp res_key; + enum ndr_err_code ndr_ret; + + smb2_util_unlink(tree, FNAME); + + status = torture_smb2_testfile(tree, FNAME, &h); + if (!NT_STATUS_IS_OK(status)) { + printf("create write\n"); + return false; + } + + ZERO_ARRAY(buf); + status = smb2_util_write(tree, h, buf, 0, ARRAY_SIZE(buf)); + if (!NT_STATUS_IS_OK(status)) { + printf("failed write\n"); + return false; + } + + ZERO_STRUCT(ioctl); + ioctl.smb2.level = RAW_IOCTL_SMB2; + ioctl.smb2.in.file.handle = h; + ioctl.smb2.in.function = FSCTL_SRV_REQUEST_RESUME_KEY; + ioctl.smb2.in.max_response_size = 32; + ioctl.smb2.in.flags = SMB2_IOCTL_FLAG_IS_FSCTL; + + status = smb2_ioctl(tree, tmp_ctx, &ioctl.smb2); + if (!NT_STATUS_IS_OK(status)) { + printf("FSCTL_SRV_REQUEST_RESUME_KEY failed\n"); + return false; + } + + ndr_ret = ndr_pull_struct_blob(&ioctl.smb2.out.out, tmp_ctx, &res_key, + (ndr_pull_flags_fn_t)ndr_pull_req_resume_key_rsp); + if (ndr_ret != NDR_ERR_SUCCESS) { + return false; + } + + ndr_print_debug((ndr_print_fn_t)ndr_print_req_resume_key_rsp, "yo", &res_key); + + talloc_free(tmp_ctx); + return true; +} + +static uint64_t patt_hash(uint64_t off) +{ + return off; +} + +static bool check_pattern(struct smb2_tree *tree, TALLOC_CTX *mem_ctx, + struct smb2_handle h, uint64_t off, uint64_t len, + uint64_t patt_off) +{ + uint64_t i; + struct smb2_read r; + NTSTATUS status; + + ZERO_STRUCT(r); + r.in.file.handle = h; + r.in.length = len; + r.in.offset = off; + status = smb2_read(tree, mem_ctx, &r); + if (!NT_STATUS_IS_OK(status)) { + printf("read failed - %s\n", nt_errstr(status)); + return false; + } else if (len != r.out.data.length) { + printf("read data len mismatch got %zd, expected %lu\n", + r.out.data.length, len); + return false; + } + + for (i = 0; i <= len - 8; i += 8, patt_off += 8) { + if (BVAL(r.out.data.data, i) != patt_hash(patt_off)) { + printf("pattern bad at %lu, got %lx, expected %lx\n", + i, BVAL(r.out.data.data, i), + patt_hash(patt_off)); + return false; + } + } + + talloc_free(r.out.data.data); + return true; +} + +static bool test_setup_copy_chunk(struct smb2_tree *tree, TALLOC_CTX *mem_ctx, + uint32_t nchunks, + struct smb2_handle *src_h, + uint64_t src_size, + struct smb2_handle *dest_h, + uint64_t dest_size, + struct srv_copychunk_copy *cc_copy, + union smb_ioctl *ioctl) +{ + struct req_resume_key_rsp res_key; + NTSTATUS status; + enum ndr_err_code ndr_ret; + uint64_t i; + uint8_t *buf = talloc_zero_size(mem_ctx, MAX(src_size, dest_size)); + if (buf == NULL) { + printf("no mem for file data buffer\n"); + return false; + } + + smb2_util_unlink(tree, FNAME); + smb2_util_unlink(tree, FNAME2); + + status = torture_smb2_testfile(tree, FNAME, src_h); + if (!NT_STATUS_IS_OK(status)) { + printf("create write\n"); + return false; + } + + if (src_size > 0) { + for (i = 0; i <= src_size - 8; i += 8) { + SBVAL(buf, i, patt_hash(i)); + } + status = smb2_util_write(tree, *src_h, buf, 0, src_size); + if (!NT_STATUS_IS_OK(status)) { + printf("failed src write\n"); + return false; + } + } + + status = torture_smb2_testfile(tree, FNAME2, dest_h); + if (!NT_STATUS_IS_OK(status)) { + printf("create write\n"); + return false; + } + + if (dest_size > 0) { + for (i = 0; i <= src_size - 8; i += 8) { + SBVAL(buf, i, patt_hash(i)); + } + status = smb2_util_write(tree, *dest_h, buf, 0, dest_size); + if (!NT_STATUS_IS_OK(status)) { + printf("failed dest write\n"); + return false; + } + } + + ZERO_STRUCTPN(ioctl); + ioctl->smb2.level = RAW_IOCTL_SMB2; + ioctl->smb2.in.file.handle = *src_h; + ioctl->smb2.in.function = FSCTL_SRV_REQUEST_RESUME_KEY; + /* Allow for Key + ContextLength + Context */ + ioctl->smb2.in.max_response_size = 32; + ioctl->smb2.in.flags = SMB2_IOCTL_FLAG_IS_FSCTL; + + status = smb2_ioctl(tree, mem_ctx, &ioctl->smb2); + if (!NT_STATUS_IS_OK(status)) { + printf("FSCTL_SRV_REQUEST_RESUME_KEY failed\n"); + return false; + } + + ndr_ret = ndr_pull_struct_blob(&ioctl->smb2.out.out, mem_ctx, &res_key, + (ndr_pull_flags_fn_t)ndr_pull_req_resume_key_rsp); + if (ndr_ret != NDR_ERR_SUCCESS) { + return false; + } + + ZERO_STRUCTPN(ioctl); + ioctl->smb2.level = RAW_IOCTL_SMB2; + ioctl->smb2.in.file.handle = *dest_h; + ioctl->smb2.in.function = FSCTL_SRV_COPYCHUNK; + ioctl->smb2.in.max_response_size = sizeof(struct srv_copychunk_rsp); + ioctl->smb2.in.flags = SMB2_IOCTL_FLAG_IS_FSCTL; + + ZERO_STRUCTPN(cc_copy); + memcpy(cc_copy->source_key, res_key.resume_key, ARRAY_SIZE(cc_copy->source_key)); + cc_copy->chunk_count = nchunks; + cc_copy->chunks = talloc_zero_array(mem_ctx, struct srv_copychunk, nchunks); + if (cc_copy->chunks == NULL) { + printf("not enough memory to allocate %u chunks\n", nchunks); + return false; + } + + return true; +} + + +static bool check_copy_chunk_rsp(struct srv_copychunk_rsp *cc_rsp, + uint32_t ex_chunks_written, + uint32_t ex_chunk_bytes_written, + uint32_t ex_total_bytes_written) +{ + if (cc_rsp->chunks_written != ex_chunks_written) { + printf("expected %u chunks, got %u\n", + ex_chunks_written, cc_rsp->chunks_written); + return false; + } + if (cc_rsp->chunk_bytes_written != ex_chunk_bytes_written) { + printf("expected %u chunk bytes remaining, got %u\n", + ex_chunk_bytes_written, cc_rsp->chunk_bytes_written); + return false; + } + if (cc_rsp->total_bytes_written != ex_total_bytes_written) { + printf("expected %u total bytes, got %u\n", + ex_total_bytes_written, cc_rsp->total_bytes_written); + return false; + } + return true; +} + +static bool test_ioctl_copy_chunk_simple(struct torture_context *torture, + struct smb2_tree *tree) +{ + struct smb2_handle src_h; + struct smb2_handle dest_h; + NTSTATUS status; + union smb_ioctl ioctl; + TALLOC_CTX *tmp_ctx = talloc_new(tree); + struct srv_copychunk_copy cc_copy; + struct srv_copychunk_rsp cc_rsp; + enum ndr_err_code ndr_ret; + bool ok; + + ok = test_setup_copy_chunk(tree, tmp_ctx, + 1, /* 1 chunk */ + &src_h, 4096, /* fill 4096 byte src file */ + &dest_h, 0, /* 0 byte dest file */ + &cc_copy, + &ioctl); + if (!ok) { + return false; + } + + /* copy all src file data (via a single chunk desc) */ + cc_copy.chunks[0].source_off = 0; + cc_copy.chunks[0].target_off = 0; + cc_copy.chunks[0].length = 4096; + + ndr_ret = ndr_push_struct_blob(&ioctl.smb2.in.out, tmp_ctx, + &cc_copy, + (ndr_push_flags_fn_t)ndr_push_srv_copychunk_copy); + if (ndr_ret != NDR_ERR_SUCCESS) { + return false; + } + + status = smb2_ioctl(tree, tmp_ctx, &ioctl.smb2); + if (!NT_STATUS_IS_OK(status)) { + printf("FSCTL_SRV_COPYCHUNK failed\n"); + return false; + } + + ndr_ret = ndr_pull_struct_blob(&ioctl.smb2.out.out, tmp_ctx, + &cc_rsp, + (ndr_pull_flags_fn_t)ndr_pull_srv_copychunk_rsp); + if (ndr_ret != NDR_ERR_SUCCESS) { + return false; + } + + ok = check_copy_chunk_rsp(&cc_rsp, + 1, /* chunks written */ + 0, /* chunk bytes unsuccessfully written */ + 4096); /* total bytes written */ + if (!ok) { + return false; + } + + ok = check_pattern(tree, tmp_ctx, dest_h, 0, 4096, 0); + if (!ok) { + return false; + } + + smb2_util_close(tree, src_h); + smb2_util_close(tree, dest_h); + talloc_free(tmp_ctx); + return true; +} + +static bool test_ioctl_copy_chunk_multi(struct torture_context *torture, + struct smb2_tree *tree) +{ + struct smb2_handle src_h; + struct smb2_handle dest_h; + NTSTATUS status; + union smb_ioctl ioctl; + TALLOC_CTX *tmp_ctx = talloc_new(tree); + struct srv_copychunk_copy cc_copy; + struct srv_copychunk_rsp cc_rsp; + enum ndr_err_code ndr_ret; + bool ok; + + ok = test_setup_copy_chunk(tree, tmp_ctx, + 2, /* chunks */ + &src_h, 8192, /* src file */ + &dest_h, 0, /* dest file */ + &cc_copy, + &ioctl); + if (!ok) { + return false; + } + + /* copy all src file data via two chunks */ + cc_copy.chunks[0].source_off = 0; + cc_copy.chunks[0].target_off = 0; + cc_copy.chunks[0].length = 4096; + + cc_copy.chunks[1].source_off = 4096; + cc_copy.chunks[1].target_off = 4096; + cc_copy.chunks[1].length = 4096; + + ndr_ret = ndr_push_struct_blob(&ioctl.smb2.in.out, tmp_ctx, + &cc_copy, + (ndr_push_flags_fn_t)ndr_push_srv_copychunk_copy); + if (ndr_ret != NDR_ERR_SUCCESS) { + return false; + } + + status = smb2_ioctl(tree, tmp_ctx, &ioctl.smb2); + if (!NT_STATUS_IS_OK(status)) { + printf("FSCTL_SRV_COPYCHUNK failed\n"); + return false; + } + + ndr_ret = ndr_pull_struct_blob(&ioctl.smb2.out.out, tmp_ctx, + &cc_rsp, + (ndr_pull_flags_fn_t)ndr_pull_srv_copychunk_rsp); + if (ndr_ret != NDR_ERR_SUCCESS) { + return false; + } + + ok = check_copy_chunk_rsp(&cc_rsp, + 2, /* chunks written */ + 0, /* chunk bytes unsuccessfully written */ + 8192); /* total bytes written */ + if (!ok) { + return false; + } + + smb2_util_close(tree, src_h); + smb2_util_close(tree, dest_h); + talloc_free(tmp_ctx); + return true; +} + +static bool test_ioctl_copy_chunk_tiny(struct torture_context *torture, + struct smb2_tree *tree) +{ + struct smb2_handle src_h; + struct smb2_handle dest_h; + NTSTATUS status; + union smb_ioctl ioctl; + TALLOC_CTX *tmp_ctx = talloc_new(tree); + struct srv_copychunk_copy cc_copy; + struct srv_copychunk_rsp cc_rsp; + enum ndr_err_code ndr_ret; + bool ok; + + ok = test_setup_copy_chunk(tree, tmp_ctx, + 2, /* chunks */ + &src_h, 100, /* src file */ + &dest_h, 0, /* dest file */ + &cc_copy, + &ioctl); + if (!ok) { + return false; + } + + /* copy all src file data via two chunks, sub block size chunks */ + cc_copy.chunks[0].source_off = 0; + cc_copy.chunks[0].target_off = 0; + cc_copy.chunks[0].length = 50; + + cc_copy.chunks[1].source_off = 50; + cc_copy.chunks[1].target_off = 50; + cc_copy.chunks[1].length = 50; + + ndr_ret = ndr_push_struct_blob(&ioctl.smb2.in.out, tmp_ctx, + &cc_copy, + (ndr_push_flags_fn_t)ndr_push_srv_copychunk_copy); + if (ndr_ret != NDR_ERR_SUCCESS) { + return false; + } + + status = smb2_ioctl(tree, tmp_ctx, &ioctl.smb2); + if (!NT_STATUS_IS_OK(status)) { + printf("FSCTL_SRV_COPYCHUNK failed\n"); + return false; + } + + ndr_ret = ndr_pull_struct_blob(&ioctl.smb2.out.out, tmp_ctx, + &cc_rsp, + (ndr_pull_flags_fn_t)ndr_pull_srv_copychunk_rsp); + if (ndr_ret != NDR_ERR_SUCCESS) { + return false; + } + + ok = check_copy_chunk_rsp(&cc_rsp, + 2, /* chunks written */ + 0, /* chunk bytes unsuccessfully written */ + 100); /* total bytes written */ + if (!ok) { + return false; + } + + ok = check_pattern(tree, tmp_ctx, dest_h, 0, 100, 0); + if (!ok) { + return false; + } + + smb2_util_close(tree, src_h); + smb2_util_close(tree, dest_h); -- Samba Shared Repository