s4-torture: never return false in the raw.composite torture test
[samba.git] / source4 / torture / raw / composite.c
index c6969574a86906509c119a3465f31bbb66772493..1c02e40013d473ab23068e54d2087289c2477f1f 100644 (file)
@@ -31,6 +31,8 @@
 #include "lib/cmdline/popt_common.h"
 #include "torture/util.h"
 #include "param/param.h"
+#include "libcli/resolve/resolve.h"
+#include "torture/raw/proto.h"
 
 #define BASEDIR "\\composite"
 
@@ -43,20 +45,20 @@ static void loadfile_complete(struct composite_context *c)
 /*
   test a simple savefile/loadfile combination
 */
-static bool test_loadfile(struct smbcli_state *cli, TALLOC_CTX *mem_ctx)
+static bool test_loadfile(struct torture_context *tctx, struct smbcli_state *cli)
 {
        const char *fname = BASEDIR "\\test.txt";
        NTSTATUS status;
        struct smb_composite_savefile io1;
-       struct smb_composite_loadfile io2;
+       struct smb_composite_loadfile *io2;
        struct composite_context **c;
        uint8_t *data;
        size_t len = random() % 100000;
        const int num_ops = 50;
        int i;
-       int *count = talloc_zero(mem_ctx, int);
+       int *count = talloc_zero(tctx, int);
 
-       data = talloc_array(mem_ctx, uint8_t, len);
+       data = talloc_array(tctx, uint8_t, len);
 
        generate_random_buffer(data, len);
 
@@ -64,53 +66,39 @@ static bool test_loadfile(struct smbcli_state *cli, TALLOC_CTX *mem_ctx)
        io1.in.data  = data;
        io1.in.size  = len;
 
-       printf("testing savefile\n");
+       torture_comment(tctx, "Testing savefile\n");
 
        status = smb_composite_savefile(cli->tree, &io1);
-       if (!NT_STATUS_IS_OK(status)) {
-               printf("(%s) savefile failed: %s\n", __location__,nt_errstr(status));
-               return false;
-       }
-
-       io2.in.fname = fname;
+       torture_assert_ntstatus_equal(tctx, status, NT_STATUS_OK, "savefile failed");
 
-       printf("testing parallel loadfile with %d ops\n", num_ops);
+       torture_comment(tctx, "Testing parallel loadfile with %d ops\n", num_ops);
 
-       c = talloc_array(mem_ctx, struct composite_context *, num_ops);
+       c = talloc_array(tctx, struct composite_context *, num_ops);
+       io2 = talloc_zero_array(tctx, struct smb_composite_loadfile, num_ops);
 
        for (i=0;i<num_ops;i++) {
-               c[i] = smb_composite_loadfile_send(cli->tree, &io2);
+               io2[i].in.fname = fname;
+               c[i] = smb_composite_loadfile_send(cli->tree, &io2[i]);
                c[i]->async.fn = loadfile_complete;
                c[i]->async.private_data = count;
        }
 
-       printf("waiting for completion\n");
+       torture_comment(tctx, "waiting for completion\n");
        while (*count != num_ops) {
-               event_loop_once(cli->transport->socket->event.ctx);
-               if (lp_parm_bool(global_loadparm, NULL, "torture", "progress", true)) {
-                       printf("(%s) count=%d\r", __location__, *count);
+               tevent_loop_once(tctx->ev);
+               if (torture_setting_bool(tctx, "progress", true)) {
+                       torture_comment(tctx, "(%s) count=%d\r", __location__, *count);
                        fflush(stdout);
                }
        }
-       printf("count=%d\n", *count);
+       torture_comment(tctx, "count=%d\n", *count);
        
        for (i=0;i<num_ops;i++) {
-               status = smb_composite_loadfile_recv(c[i], mem_ctx);
-               if (!NT_STATUS_IS_OK(status)) {
-                       printf("(%s) loadfile[%d] failed - %s\n", __location__, i, nt_errstr(status));
-                       return false;
-               }
+               status = smb_composite_loadfile_recv(c[i], tctx);
+               torture_assert_ntstatus_equal(tctx, status, NT_STATUS_OK, "loadfile failed");
 
-               if (io2.out.size != len) {
-                       printf("(%s) wrong length in returned data - %d should be %d\n",__location__,
-                              io2.out.size, (int)len);
-                       return false;
-               }
-               
-               if (memcmp(io2.out.data, data, len) != 0) {
-                       printf("(%s) wrong data in loadfile!\n",__location__);
-                       return false;
-               }
+               torture_assert_int_equal(tctx, io2[i].out.size, len, "wrong length in returned data");
+               torture_assert_mem_equal(tctx, io2[i].out.data, data, len, "wrong data in loadfile");
        }
 
        talloc_free(data);
@@ -118,10 +106,22 @@ static bool test_loadfile(struct smbcli_state *cli, TALLOC_CTX *mem_ctx)
        return true;
 }
 
+static bool test_loadfile_t(struct torture_context *tctx, struct smbcli_state *cli)
+{
+       int ret;
+       torture_assert(tctx, torture_setup_dir(cli, BASEDIR), "failed to setup " BASEDIR);
+
+       ret = test_loadfile(tctx, cli);
+       smb_raw_exit(cli->session);
+       smbcli_deltree(cli->tree, BASEDIR);
+
+       return ret;
+}
+
 /*
   test a simple savefile/loadfile combination
 */
-static bool test_fetchfile(struct smbcli_state *cli, TALLOC_CTX *mem_ctx)
+static bool test_fetchfile(struct torture_context *tctx, struct smbcli_state *cli)
 {
        const char *fname = BASEDIR "\\test.txt";
        NTSTATUS status;
@@ -132,40 +132,45 @@ static bool test_fetchfile(struct smbcli_state *cli, TALLOC_CTX *mem_ctx)
        int i;
        size_t len = random() % 10000;
        extern int torture_numops;
-       struct event_context *event_ctx;
-       int *count = talloc_zero(mem_ctx, int);
+       struct tevent_context *event_ctx;
+       int *count = talloc_zero(tctx, int);
        bool ret = true;
 
-       data = talloc_array(mem_ctx, uint8_t, len);
+       data = talloc_array(tctx, uint8_t, len);
 
        generate_random_buffer(data, len);
 
+       ZERO_STRUCT(io1);
        io1.in.fname = fname;
        io1.in.data  = data;
        io1.in.size  = len;
 
-       printf("testing savefile\n");
+       torture_comment(tctx, "Testing savefile\n");
 
        status = smb_composite_savefile(cli->tree, &io1);
-       if (!NT_STATUS_IS_OK(status)) {
-               printf("(%s) savefile failed: %s\n",__location__, nt_errstr(status));
-               return false;
-       }
+       torture_assert_ntstatus_equal(tctx, status, NT_STATUS_OK, "savefile failed");
+
+       ZERO_STRUCT(io2);
 
-       io2.in.dest_host = lp_parm_string(global_loadparm, NULL, "torture", "host");
-       io2.in.port = 0;
-       io2.in.called_name = lp_parm_string(global_loadparm, NULL, "torture", "host");
-       io2.in.service = lp_parm_string(global_loadparm, NULL, "torture", "share");
+       io2.in.dest_host = torture_setting_string(tctx, "host", NULL);
+       io2.in.ports = lpcfg_smb_ports(tctx->lp_ctx);
+       io2.in.called_name = torture_setting_string(tctx, "host", NULL);
+       io2.in.service = torture_setting_string(tctx, "share", NULL);
        io2.in.service_type = "A:";
+       io2.in.socket_options = lpcfg_socket_options(tctx->lp_ctx);
 
        io2.in.credentials = cmdline_credentials;
-       io2.in.workgroup  = lp_workgroup(global_loadparm);
+       io2.in.workgroup  = lpcfg_workgroup(tctx->lp_ctx);
        io2.in.filename = fname;
+       lpcfg_smbcli_options(tctx->lp_ctx, &io2.in.options);
+       lpcfg_smbcli_session_options(tctx->lp_ctx, &io2.in.session_options);
+       io2.in.resolve_ctx = lpcfg_resolve_context(tctx->lp_ctx);
+       io2.in.gensec_settings = lpcfg_gensec_settings(tctx, tctx->lp_ctx);
 
-       printf("testing parallel fetchfile with %d ops\n", torture_numops);
+       torture_comment(tctx, "Testing parallel fetchfile with %d ops\n", torture_numops);
 
-       event_ctx = cli->transport->socket->event.ctx;
-       c = talloc_array(mem_ctx, struct composite_context *, torture_numops);
+       event_ctx = tctx->ev;
+       c = talloc_array(tctx, struct composite_context *, torture_numops);
 
        for (i=0; i<torture_numops; i++) {
                c[i] = smb_composite_fetchfile_send(&io2, event_ctx);
@@ -173,66 +178,61 @@ static bool test_fetchfile(struct smbcli_state *cli, TALLOC_CTX *mem_ctx)
                c[i]->async.private_data = count;
        }
 
-       printf("waiting for completion\n");
+       torture_comment(tctx, "waiting for completion\n");
 
        while (*count != torture_numops) {
-               event_loop_once(event_ctx);
-               if (lp_parm_bool(global_loadparm, NULL, "torture", "progress", true)) {
-                       printf("(%s) count=%d\r", __location__, *count);
+               tevent_loop_once(event_ctx);
+               if (torture_setting_bool(tctx, "progress", true)) {
+                       torture_comment(tctx, "(%s) count=%d\r", __location__, *count);
                        fflush(stdout);
                }
        }
-       printf("count=%d\n", *count);
+       torture_comment(tctx, "count=%d\n", *count);
 
        for (i=0;i<torture_numops;i++) {
-               status = smb_composite_fetchfile_recv(c[i], mem_ctx);
-               if (!NT_STATUS_IS_OK(status)) {
-                       printf("(%s) loadfile[%d] failed - %s\n", __location__, i,
-                              nt_errstr(status));
-                       ret = false;
-                       continue;
-               }
+               status = smb_composite_fetchfile_recv(c[i], tctx);
+               torture_assert_ntstatus_equal(tctx, status, NT_STATUS_OK, "loadfile failed");
 
-               if (io2.out.size != len) {
-                       printf("(%s) wrong length in returned data - %d "
-                              "should be %d\n", __location__,
-                              io2.out.size, (int)len);
-                       ret = false;
-                       continue;
-               }
-               
-               if (memcmp(io2.out.data, data, len) != 0) {
-                       printf("(%s) wrong data in loadfile!\n", __location__);
-                       ret = false;
-                       continue;
-               }
+               torture_assert_int_equal(tctx, io2.out.size, len, "wrong length in returned data");
+               torture_assert_mem_equal(tctx, io2.out.data, data, len, "wrong data in loadfile");
        }
 
        return ret;
 }
 
+static bool test_fetchfile_t(struct torture_context *tctx, struct smbcli_state *cli)
+{
+       int ret;
+       torture_assert(tctx, torture_setup_dir(cli, BASEDIR), "failed to setup " BASEDIR);
+       ret = test_fetchfile(tctx, cli);
+       smb_raw_exit(cli->session);
+       smbcli_deltree(cli->tree, BASEDIR);
+
+       return ret;
+}
+
 /*
   test setfileacl
 */
-static bool test_appendacl(struct smbcli_state *cli, TALLOC_CTX *mem_ctx)
+static bool test_appendacl(struct torture_context *tctx, struct smbcli_state *cli)
 {
        struct smb_composite_appendacl **io;
        struct smb_composite_appendacl **io_orig;
        struct composite_context **c;
-       struct event_context *event_ctx;
+       struct tevent_context *event_ctx;
 
        struct security_descriptor *test_sd;
        struct security_ace *ace;
        struct dom_sid *test_sid;
 
        const int num_ops = 50;
-       int *count = talloc_zero(mem_ctx, int);
+       int *count = talloc_zero(tctx, int);
        struct smb_composite_savefile io1;
 
        NTSTATUS status;
        int i;
 
-       io_orig = talloc_array(mem_ctx, struct smb_composite_appendacl *, num_ops);
+       io_orig = talloc_array(tctx, struct smb_composite_appendacl *, num_ops);
 
        printf ("creating %d empty files and getting their acls with appendacl\n", num_ops);
 
@@ -242,28 +242,22 @@ static bool test_appendacl(struct smbcli_state *cli, TALLOC_CTX *mem_ctx)
                io1.in.size  = 0;
          
                status = smb_composite_savefile(cli->tree, &io1);
-               if (!NT_STATUS_IS_OK(status)) {
-                       printf("(%s) savefile failed: %s\n", __location__, nt_errstr(status));
-                       return false;
-               }
+               torture_assert_ntstatus_equal(tctx, status, NT_STATUS_OK, "savefile failed");
 
                io_orig[i] = talloc (io_orig, struct smb_composite_appendacl);
                io_orig[i]->in.fname = talloc_steal(io_orig[i], io1.in.fname);
                io_orig[i]->in.sd = security_descriptor_initialise(io_orig[i]);
                status = smb_composite_appendacl(cli->tree, io_orig[i], io_orig[i]);
-               if (!NT_STATUS_IS_OK(status)) {
-                       printf("(%s) appendacl failed: %s\n", __location__, nt_errstr(status));
-                       return false;
-               }
+               torture_assert_ntstatus_equal(tctx, status, NT_STATUS_OK, "appendacl failed");
        }
        
 
        /* fill Security Descriptor with aces to be added */
 
-       test_sd = security_descriptor_initialise(mem_ctx);
-       test_sid = dom_sid_parse_talloc (mem_ctx, "S-1-5-32-1234-5432");
+       test_sd = security_descriptor_initialise(tctx);
+       test_sid = dom_sid_parse_talloc (tctx, "S-1-5-32-1234-5432");
 
-       ace = talloc_zero(mem_ctx, struct security_ace);
+       ace = talloc_zero(tctx, struct security_ace);
 
        ace->type = SEC_ACE_TYPE_ACCESS_ALLOWED;
        ace->flags = 0;
@@ -271,17 +265,14 @@ static bool test_appendacl(struct smbcli_state *cli, TALLOC_CTX *mem_ctx)
        ace->trustee = *test_sid;
 
        status = security_descriptor_dacl_add(test_sd, ace);
-       if (!NT_STATUS_IS_OK(status)) {
-               printf("(%s) appendacl failed: %s\n", __location__, nt_errstr(status));
-               return false;
-       }
+       torture_assert_ntstatus_equal(tctx, status, NT_STATUS_OK, "appendacl failed");
 
        /* set parameters for appendacl async call */
 
-       printf("testing parallel appendacl with %d ops\n", num_ops);
+       torture_comment(tctx, "Testing parallel appendacl with %d ops\n", num_ops);
 
-       c = talloc_array(mem_ctx, struct composite_context *, num_ops);
-       io = talloc_array(mem_ctx, struct  smb_composite_appendacl *, num_ops);
+       c = talloc_array(tctx, struct composite_context *, num_ops);
+       io = talloc_array(tctx, struct  smb_composite_appendacl *, num_ops);
 
        for (i=0; i < num_ops; i++) {
                io[i] = talloc (io, struct smb_composite_appendacl);
@@ -293,29 +284,29 @@ static bool test_appendacl(struct smbcli_state *cli, TALLOC_CTX *mem_ctx)
                c[i]->async.private_data = count;
        }
 
-       event_ctx = talloc_reference(mem_ctx, cli->tree->session->transport->socket->event.ctx);
-       printf("waiting for completion\n");
+       event_ctx = tctx->ev;
+       torture_comment(tctx, "waiting for completion\n");
        while (*count != num_ops) {
-               event_loop_once(event_ctx);
-               if (lp_parm_bool(global_loadparm, NULL, "torture", "progress", true)) {
-                       printf("(%s) count=%d\r", __location__, *count);
+               tevent_loop_once(event_ctx);
+               if (torture_setting_bool(tctx, "progress", true)) {
+                       torture_comment(tctx, "(%s) count=%d\r", __location__, *count);
                        fflush(stdout);
                }
        }
-       printf("count=%d\n", *count);
+       torture_comment(tctx, "count=%d\n", *count);
 
        for (i=0; i < num_ops; i++) {
                status = smb_composite_appendacl_recv(c[i], io[i]);
                if (!NT_STATUS_IS_OK(status)) {
-                       printf("(%s) appendacl[%d] failed - %s\n", __location__, i, nt_errstr(status));
+                       torture_comment(tctx, "(%s) appendacl[%d] failed - %s\n", __location__, i, nt_errstr(status));
                        return false;
                }
                
                security_descriptor_dacl_add(io_orig[i]->out.sd, ace);
-               if (!security_acl_equal(io_orig[i]->out.sd->dacl, io[i]->out.sd->dacl)) {
-                       printf("(%s) appendacl[%d] failed - needed acl isn't set\n", __location__, i);
-                       return false;
-               }
+               torture_assert(tctx,
+                              security_acl_equal(io_orig[i]->out.sd->dacl,
+                                                 io[i]->out.sd->dacl),
+                              "appendacl failed - needed acl isn't set");
        }
        
 
@@ -326,8 +317,19 @@ static bool test_appendacl(struct smbcli_state *cli, TALLOC_CTX *mem_ctx)
        return true;
 }
 
+static bool test_appendacl_t(struct torture_context *tctx, struct smbcli_state *cli)
+{
+       int ret;
+       torture_assert(tctx, torture_setup_dir(cli, BASEDIR), "failed to setup " BASEDIR);
+       ret = test_appendacl(tctx, cli);
+       smb_raw_exit(cli->session);
+       smbcli_deltree(cli->tree, BASEDIR);
+
+       return ret;
+}
+
 /* test a query FS info by asking for share's GUID */
-static bool test_fsinfo(struct smbcli_state *cli, TALLOC_CTX *mem_ctx)
+static bool test_fsinfo(struct torture_context *tctx, struct smbcli_state *cli)
 {
        char *guid = NULL;
        NTSTATUS status;
@@ -336,86 +338,80 @@ static bool test_fsinfo(struct smbcli_state *cli, TALLOC_CTX *mem_ctx)
 
        int i;
        extern int torture_numops;
-       struct event_context *event_ctx;
-       int *count = talloc_zero(mem_ctx, int);
+       struct tevent_context *event_ctx;
+       int *count = talloc_zero(tctx, int);
        bool ret = true;
 
-       io1.in.dest_host = lp_parm_string(global_loadparm, NULL, "torture", "host");
-       io1.in.port = 0;
-       io1.in.called_name = lp_parm_string(global_loadparm, NULL, "torture", "host");
-       io1.in.service = lp_parm_string(global_loadparm, NULL, "torture", "share");
+       io1.in.dest_host = torture_setting_string(tctx, "host", NULL);
+       io1.in.dest_ports = lpcfg_smb_ports(tctx->lp_ctx);
+       io1.in.socket_options = lpcfg_socket_options(tctx->lp_ctx);
+       io1.in.called_name = torture_setting_string(tctx, "host", NULL);
+       io1.in.service = torture_setting_string(tctx, "share", NULL);
        io1.in.service_type = "A:";
        io1.in.credentials = cmdline_credentials;
-       io1.in.workgroup = lp_workgroup(global_loadparm);
+       io1.in.workgroup = lpcfg_workgroup(tctx->lp_ctx);
        io1.in.level = RAW_QFS_OBJECTID_INFORMATION;
+       io1.in.gensec_settings = lpcfg_gensec_settings(tctx, tctx->lp_ctx);
 
-       printf("testing parallel queryfsinfo [Object ID] with %d ops\n", torture_numops);
+       torture_comment(tctx, "Testing parallel queryfsinfo [Object ID] with %d ops\n",
+                       torture_numops);
 
-       event_ctx = talloc_reference(mem_ctx, cli->tree->session->transport->socket->event.ctx);
-       c = talloc_array(mem_ctx, struct composite_context *, torture_numops);
+       event_ctx = tctx->ev;
+       c = talloc_array(tctx, struct composite_context *, torture_numops);
 
        for (i=0; i<torture_numops; i++) {
-               c[i] = smb_composite_fsinfo_send(cli->tree,&io1);
+               c[i] = smb_composite_fsinfo_send(cli->tree, &io1, lpcfg_resolve_context(tctx->lp_ctx), event_ctx);
+               torture_assert(tctx, c[i], "smb_composite_fsinfo_send failed!");
                c[i]->async.fn = loadfile_complete;
                c[i]->async.private_data = count;
        }
 
-       printf("waiting for completion\n");
+       torture_comment(tctx, "waiting for completion\n");
 
        while (*count < torture_numops) {
-               event_loop_once(event_ctx);
-               if (lp_parm_bool(global_loadparm, NULL, "torture", "progress", true)) {
-                       printf("(%s) count=%d\r", __location__, *count);
+               tevent_loop_once(event_ctx);
+               if (torture_setting_bool(tctx, "progress", true)) {
+                       torture_comment(tctx, "(%s) count=%d\r", __location__, *count);
                        fflush(stdout);
                }
        }
-       printf("count=%d\n", *count);
+       torture_comment(tctx, "count=%d\n", *count);
 
        for (i=0;i<torture_numops;i++) {
-               status = smb_composite_fsinfo_recv(c[i], mem_ctx);
-               if (!NT_STATUS_IS_OK(status)) {
-                       printf("(%s) fsinfo[%d] failed - %s\n", __location__, i, nt_errstr(status));
-                       ret = false;
-                       continue;
-               }
-
-               if (io1.out.fsinfo->generic.level != RAW_QFS_OBJECTID_INFORMATION) {
-                       printf("(%s) wrong level in returned info - %d "
-                              "should be %d\n", __location__,
-                              io1.out.fsinfo->generic.level, RAW_QFS_OBJECTID_INFORMATION);
-                       ret = false;
-                       continue;
-               }
+               status = smb_composite_fsinfo_recv(c[i], tctx);
+               torture_assert_ntstatus_equal(tctx, status, NT_STATUS_OK, "smb_composite_fsinfo_recv failed");
 
-               guid=GUID_string(mem_ctx, &io1.out.fsinfo->objectid_information.out.guid);
-               printf("[%d] GUID: %s\n", i, guid);
+               torture_assert_int_equal(tctx, io1.out.fsinfo->generic.level, RAW_QFS_OBJECTID_INFORMATION, "wrong level in returned info");
 
-               
+               guid=GUID_string(tctx, &io1.out.fsinfo->objectid_information.out.guid);
+               torture_comment(tctx, "[%d] GUID: %s\n", i, guid);
        }
 
        return ret;
 }
 
-
-/* 
-   basic testing of libcli composite calls
-*/
-bool torture_raw_composite(struct torture_context *tctx, 
-                          struct smbcli_state *cli)
+static bool test_fsinfo_t(struct torture_context *tctx, struct smbcli_state *cli)
 {
-       bool ret = true;
-
-       if (!torture_setup_dir(cli, BASEDIR)) {
-               return false;
-       }
-
-       ret &= test_fetchfile(cli, tctx);
-       ret &= test_loadfile(cli, tctx);
-       ret &= test_appendacl(cli, tctx);
-       ret &= test_fsinfo(cli, tctx);
-
+       int ret;
+       torture_assert(tctx, torture_setup_dir(cli, BASEDIR), "failed to setup " BASEDIR);
+       ret = test_fsinfo(tctx, cli);
        smb_raw_exit(cli->session);
        smbcli_deltree(cli->tree, BASEDIR);
 
        return ret;
 }
+
+/*
+   basic testing of all RAW_SEARCH_* calls using a single file
+*/
+struct torture_suite *torture_raw_composite(TALLOC_CTX *mem_ctx)
+{
+       struct torture_suite *suite = torture_suite_create(mem_ctx, "composite");
+
+       torture_suite_add_1smb_test(suite, "fetchfile", test_fetchfile_t);
+       torture_suite_add_1smb_test(suite, "loadfile", test_loadfile_t);
+       torture_suite_add_1smb_test(suite, "appendacl", test_appendacl_t);
+       torture_suite_add_1smb_test(suite, "fsinfo", test_fsinfo_t);
+
+       return suite;
+}