s4/test: Replace hand-made field count to test
[kamenim/samba.git] / source4 / torture / libnet / userman.c
index 1d630e5ecdb3c08ea86a17a9e7a8076ac33e2f22..c67f133356cf41b7462ed4e231d46ca31ff7a1fd 100644 (file)
@@ -1,25 +1,25 @@
-/* 
+/*
    Unix SMB/CIFS implementation.
    Test suite for libnet calls.
 
    Copyright (C) Rafal Szczesniak 2005
-   
+
    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 "torture/rpc/rpc.h"
+#include "torture/rpc/torture_rpc.h"
 #include "torture/libnet/usertest.h"
 #include "libnet/libnet.h"
 #include "librpc/gen_ndr/ndr_samr_c.h"
 #include "torture/libnet/utils.h"
 
 
-static bool test_useradd(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
+static bool test_useradd(struct torture_context *tctx,
+                        struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
                         struct policy_handle *domain_handle,
                         const char *name)
 {
        NTSTATUS status;
        bool ret = true;
        struct libnet_rpc_useradd user;
-       
+
        user.in.domain_handle = *domain_handle;
        user.in.username      = name;
 
-       printf("Testing libnet_rpc_useradd\n");
+       torture_comment(tctx, "Testing libnet_rpc_useradd\n");
 
        status = libnet_rpc_useradd(p, mem_ctx, &user);
        if (!NT_STATUS_IS_OK(status)) {
-               printf("Failed to call libnet_rpc_useradd - %s\n", nt_errstr(status));
+               torture_comment(tctx, "Failed to call libnet_rpc_useradd - %s\n", nt_errstr(status));
                return false;
        }
-       
+
        return ret;
 }
 
 
-static bool test_useradd_async(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
+static bool test_useradd_async(struct torture_context *tctx,
+                              struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
                               struct policy_handle *handle, const char* username)
 {
        NTSTATUS status;
@@ -60,18 +62,18 @@ static bool test_useradd_async(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
 
        user.in.domain_handle = *handle;
        user.in.username      = username;
-       
-       printf("Testing async libnet_rpc_useradd\n");
-       
+
+       torture_comment(tctx, "Testing async libnet_rpc_useradd\n");
+
        c = libnet_rpc_useradd_send(p, &user, msg_handler);
        if (!c) {
-               printf("Failed to call async libnet_rpc_useradd\n");
+               torture_comment(tctx, "Failed to call async libnet_rpc_useradd\n");
                return false;
        }
 
        status = libnet_rpc_useradd_recv(c, mem_ctx, &user);
        if (!NT_STATUS_IS_OK(status)) {
-               printf("Calling async libnet_rpc_useradd failed - %s\n", nt_errstr(status));
+               torture_comment(tctx, "Calling async libnet_rpc_useradd failed - %s\n", nt_errstr(status));
                return false;
        }
 
@@ -79,7 +81,8 @@ static bool test_useradd_async(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
 
 }
 
-static bool test_usermod(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
+static bool test_usermod(struct torture_context *tctx, struct dcerpc_pipe *p,
+                        TALLOC_CTX *mem_ctx,
                         struct policy_handle *handle, int num_changes,
                         struct libnet_rpc_usermod *mod, char **username)
 {
@@ -87,9 +90,9 @@ static bool test_usermod(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
        const char* home_dirs[] = { "\\\\srv\\home", "\\\\homesrv\\home\\user", "\\\\pdcsrv\\domain" };
        const char* home_drives[] = { "H:", "z:", "I:", "J:", "n:" };
        const char *homedir, *homedrive, *logonscript;
-       const uint32_t flags[] = { (ACB_DISABLED | ACB_NORMAL),
+       const uint32_t flags[] = { (ACB_DISABLED | ACB_NORMAL | ACB_PW_EXPIRED),
                                   (ACB_NORMAL | ACB_PWNOEXP),
-                                  (ACB_NORMAL) };
+                                  (ACB_NORMAL | ACB_PW_EXPIRED) };
 
        NTSTATUS status;
        struct timeval now;
@@ -102,14 +105,15 @@ static bool test_usermod(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
        mod->in.username = talloc_strdup(mem_ctx, *username);
        mod->in.domain_handle = *handle;
 
-       printf("modifying user (%d simultaneous change(s))\n", num_changes);
+       torture_comment(tctx, "modifying user (%d simultaneous change(s))\n",
+                       num_changes);
 
-       printf("fields to change: [");
+       torture_comment(tctx, "fields to change: [");
 
-       for (i = 0; i < num_changes && i < FIELDS_NUM - 1; i++) {
+       for (i = 0; i < num_changes && i <= USER_FIELD_LAST; i++) {
                const char *fldname;
 
-               testfld = (random() % (FIELDS_NUM - 1)) + 1;
+               testfld = (random() % USER_FIELD_LAST) + 1;
 
                gettimeofday(&now, NULL);
 
@@ -138,7 +142,7 @@ static bool test_usermod(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
                        mod->in.change.fields |= USERMOD_FIELD_DESCRIPTION;
                        fldname = "description";
                        break;
-                       
+
                case home_directory:
                        continue_if_field_set(mod->in.change.home_directory);
                        homedir = home_dirs[random() % (sizeof(home_dirs)/sizeof(char*))];
@@ -199,32 +203,30 @@ static bool test_usermod(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
                        break;
                }
 
-               printf(((i < num_changes - 1) ? "%s," : "%s"), fldname);
+               torture_comment(tctx, ((i < num_changes - 1) ? "%s," : "%s"), fldname);
        }
-       printf("]\n");
+       torture_comment(tctx, "]\n");
 
        status = libnet_rpc_usermod(p, mem_ctx, mod);
-       if (!NT_STATUS_IS_OK(status)) {
-               printf("Failed to call sync libnet_rpc_usermod - %s\n", nt_errstr(status));
-               return false;
-       }
+       torture_assert_ntstatus_ok(tctx, status, "Failed to call sync libnet_rpc_usermod");
 
        return true;
 }
 
 
-static bool test_userdel(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
+static bool test_userdel(struct torture_context *tctx,
+                        struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
                         struct policy_handle *handle, const char *username)
 {
        NTSTATUS status;
        struct libnet_rpc_userdel user;
-       
+
        user.in.domain_handle = *handle;
        user.in.username = username;
-       
+
        status = libnet_rpc_userdel(p, mem_ctx, &user);
        if (!NT_STATUS_IS_OK(status)) {
-               printf("Failed to call sync libnet_rpc_userdel - %s\n", nt_errstr(status));
+               torture_comment(tctx, "Failed to call sync libnet_rpc_userdel - %s\n", nt_errstr(status));
                return false;
        }
 
@@ -235,9 +237,9 @@ static bool test_userdel(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
 #define CMP_LSA_STRING_FLD(fld, flags) \
        if ((mod->in.change.fields & flags) && \
            !strequal(i->fld.string, mod->in.change.fld)) { \
-               printf("'%s' field does not match\n", #fld); \
-               printf("received: '%s'\n", i->fld.string); \
-               printf("expected: '%s'\n", mod->in.change.fld); \
+               torture_comment(tctx, "'%s' field does not match\n", #fld); \
+               torture_comment(tctx, "received: '%s'\n", i->fld.string); \
+               torture_comment(tctx, "expected: '%s'\n", mod->in.change.fld); \
                return false; \
        }
 
@@ -246,10 +248,10 @@ static bool test_userdel(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
        if (mod->in.change.fields & flags) { \
                nttime_to_timeval(&t, i->fld); \
                if (timeval_compare(&t, mod->in.change.fld)) { \
-                       printf("'%s' field does not match\n", #fld); \
-                       printf("received: '%s (+%ld us)'\n", \
+                       torture_comment(tctx, "'%s' field does not match\n", #fld); \
+                       torture_comment(tctx, "received: '%s (+%ld us)'\n", \
                               timestring(mem_ctx, t.tv_sec), t.tv_usec); \
-                       printf("expected: '%s (+%ld us)'\n", \
+                       torture_comment(tctx, "expected: '%s (+%ld us)'\n", \
                               timestring(mem_ctx, mod->in.change.fld->tv_sec), \
                               mod->in.change.fld->tv_usec); \
                        return false; \
@@ -259,14 +261,15 @@ static bool test_userdel(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
 #define CMP_NUM_FLD(fld, flags) \
        if ((mod->in.change.fields & flags) && \
            (i->fld != mod->in.change.fld)) { \
-               printf("'%s' field does not match\n", #fld); \
-               printf("received: '%04x'\n", i->fld); \
-               printf("expected: '%04x'\n", mod->in.change.fld); \
+               torture_comment(tctx, "'%s' field does not match\n", #fld); \
+               torture_comment(tctx, "received: '%04x'\n", i->fld); \
+               torture_comment(tctx, "expected: '%04x'\n", mod->in.change.fld); \
                return false; \
        }
 
 
-static bool test_compare(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
+static bool test_compare(struct torture_context *tctx,
+                        struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
                         struct policy_handle *handle, struct libnet_rpc_usermod *mod,
                         const char *username)
 {
@@ -282,10 +285,7 @@ static bool test_compare(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx,
        info.in.level = 21;             /* the most rich infolevel available */
 
        status = libnet_rpc_userinfo(p, mem_ctx, &info);
-       if (!NT_STATUS_IS_OK(status)) {
-               printf("Failed to call sync libnet_rpc_userinfo - %s\n", nt_errstr(status));
-               return false;
-       }
+       torture_assert_ntstatus_ok(tctx, status, "Failed to call sync libnet_rpc_userinfo");
 
        i = &info.out.info.info21;
 
@@ -314,44 +314,44 @@ bool torture_useradd(struct torture_context *torture)
        const char *name = TEST_USERNAME;
        TALLOC_CTX *mem_ctx;
        bool ret = true;
+       struct dcerpc_binding_handle *b;
 
        mem_ctx = talloc_init("test_useradd");
 
-       status = torture_rpc_connection(torture, 
+       status = torture_rpc_connection(torture,
                                        &p,
                                        &ndr_table_samr);
-       
-       if (!NT_STATUS_IS_OK(status)) {
-               return false;
-       }
+
+       torture_assert_ntstatus_ok(torture, status, "RPC connect failed");
+       b = p->binding_handle;
 
        domain_name.string = lp_workgroup(torture->lp_ctx);
-       if (!test_opendomain(p, mem_ctx, &h, &domain_name, &sid)) {
+       if (!test_opendomain(torture, b, mem_ctx, &h, &domain_name, &sid)) {
                ret = false;
                goto done;
        }
 
-       if (!test_useradd(p, mem_ctx, &h, name)) {
+       if (!test_useradd(torture, p, mem_ctx, &h, name)) {
                ret = false;
                goto done;
        }
 
-       if (!test_user_cleanup(p, mem_ctx, &h, name)) {
+       if (!test_user_cleanup(torture, b, mem_ctx, &h, name)) {
                ret = false;
                goto done;
        }
 
-       if (!test_opendomain(p, mem_ctx, &h, &domain_name, &sid)) {
+       if (!test_opendomain(torture, b, mem_ctx, &h, &domain_name, &sid)) {
                ret = false;
                goto done;
        }
 
-       if (!test_useradd_async(p, mem_ctx, &h, name)) {
+       if (!test_useradd_async(torture, p, mem_ctx, &h, name)) {
                ret = false;
                goto done;
        }
 
-       if (!test_user_cleanup(p, mem_ctx, &h, name)) {
+       if (!test_user_cleanup(torture, b, mem_ctx, &h, name)) {
                ret = false;
                goto done;
        }
@@ -373,33 +373,35 @@ bool torture_userdel(struct torture_context *torture)
        const char *name = TEST_USERNAME;
        TALLOC_CTX *mem_ctx;
        bool ret = true;
+       struct dcerpc_binding_handle *b;
 
        mem_ctx = talloc_init("test_userdel");
 
-       status = torture_rpc_connection(torture, 
+       status = torture_rpc_connection(torture,
                                        &p,
                                        &ndr_table_samr);
-       
+
        if (!NT_STATUS_IS_OK(status)) {
                return false;
        }
+       b = p->binding_handle;
 
        domain_name.string = lp_workgroup(torture->lp_ctx);
-       if (!test_opendomain(p, mem_ctx, &h, &domain_name, &sid)) {
+       if (!test_opendomain(torture, b, mem_ctx, &h, &domain_name, &sid)) {
                ret = false;
                goto done;
        }
 
-       if (!test_user_create(p, mem_ctx, &h, name, &rid)) {
+       if (!test_user_create(torture, b, mem_ctx, &h, name, &rid)) {
                ret = false;
                goto done;
        }
-       
-               if (!test_userdel(p, mem_ctx, &h, name)) {
+
+       if (!test_userdel(torture, p, mem_ctx, &h, name)) {
                ret = false;
                goto done;
        }
-       
+
 done:
        talloc_free(mem_ctx);
        return ret;
@@ -418,47 +420,46 @@ bool torture_usermod(struct torture_context *torture)
        char *name;
        TALLOC_CTX *mem_ctx;
        bool ret = true;
+       struct dcerpc_binding_handle *b;
 
        mem_ctx = talloc_init("test_userdel");
 
-       status = torture_rpc_connection(torture, 
+       status = torture_rpc_connection(torture,
                                        &p,
                                        &ndr_table_samr);
-       
-       if (!NT_STATUS_IS_OK(status)) {
-               ret = false;
-               goto done;
-       }
+
+       torture_assert_ntstatus_ok(torture, status, "RPC connect");
+       b = p->binding_handle;
 
        domain_name.string = lp_workgroup(torture->lp_ctx);
        name = talloc_strdup(mem_ctx, TEST_USERNAME);
 
-       if (!test_opendomain(p, mem_ctx, &h, &domain_name, &sid)) {
+       if (!test_opendomain(torture, b, mem_ctx, &h, &domain_name, &sid)) {
                ret = false;
                goto done;
        }
 
-       if (!test_user_create(p, mem_ctx, &h, name, &rid)) {
+       if (!test_user_create(torture, b, mem_ctx, &h, name, &rid)) {
                ret = false;
                goto done;
        }
-       
-       for (i = 1; i < FIELDS_NUM; i++) {
+
+       for (i = USER_FIELD_FIRST; i <= USER_FIELD_LAST; i++) {
                struct libnet_rpc_usermod m;
 
-               if (!test_usermod(p, mem_ctx, &h, i, &m, &name)) {
+               if (!test_usermod(torture, p, mem_ctx, &h, i, &m, &name)) {
                        ret = false;
                        goto cleanup;
                }
 
-               if (!test_compare(p, mem_ctx, &h, &m, name)) {
+               if (!test_compare(torture, p, mem_ctx, &h, &m, name)) {
                        ret = false;
                        goto cleanup;
                }
        }
-       
-cleanup:       
-       if (!test_user_cleanup(p, mem_ctx, &h, name)) {
+
+cleanup:
+       if (!test_user_cleanup(torture, b, mem_ctx, &h, name)) {
                ret = false;
                goto done;
        }