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

Reply via email to