Author: jelmer
Date: 2006-06-12 00:33:15 +0000 (Mon, 12 Jun 2006)
New Revision: 16145

WebSVN: 
http://websvn.samba.org/cgi-bin/viewcvs.cgi?view=rev&root=samba&rev=16145

Log:
Add test program for passdb backends. Patch by Wilco Baan Hofman with 
some fixes by me.

Added:
   trunk/source/torture/pdbtest.c
Modified:
   trunk/source/Makefile.in


Changeset:
Modified: trunk/source/Makefile.in
===================================================================
--- trunk/source/Makefile.in    2006-06-12 00:27:21 UTC (rev 16144)
+++ trunk/source/Makefile.in    2006-06-12 00:33:15 UTC (rev 16145)
@@ -149,7 +149,8 @@
 
 TORTURE_PROGS = bin/[EMAIL PROTECTED]@ bin/[EMAIL PROTECTED]@ \
        bin/[EMAIL PROTECTED]@ bin/[EMAIL PROTECTED]@ \
-       bin/[EMAIL PROTECTED]@ bin/[EMAIL PROTECTED]@ bin/[EMAIL PROTECTED]@
+       bin/[EMAIL PROTECTED]@ bin/[EMAIL PROTECTED]@ bin/[EMAIL PROTECTED]@ \
+       bin/[EMAIL PROTECTED]@
 
 BIN_PROGS = $(BIN_PROGS1) $(BIN_PROGS2) $(BIN_PROGS3) @EXTRA_BIN_PROGS@
 
@@ -638,6 +639,11 @@
 NSSTEST_OBJ = torture/nsstest.o $(PARAM_OBJ) $(LIBSMB_OBJ) $(KRBCLIENT_OBJ) \
                  $(LIB_NONSMBD_OBJ) $(SECRETS_OBJ)
 
+PDBTEST_OBJ = torture/pdbtest.o $(PARAM_OBJ) $(LIBSMB_OBJ) $(KRBCLIENT_OBJ) \
+               $(LIB_NONSMBD_OBJ) $(SECRETS_OBJ) $(PASSDB_OBJ) $(GROUPDB_OBJ) \
+               $(SMBLDAP_OBJ) $(POPT_LIB_OBJ) 
+
+
 VFSTEST_OBJ = torture/cmd_vfs.o torture/vfstest.o $(SMBD_OBJ_BASE) 
$(READLINE_OBJ)
 
 SMBICONV_OBJ = $(PARAM_OBJ) torture/smbiconv.o $(LIB_NONSMBD_OBJ) 
$(POPT_LIB_OBJ) $(SECRETS_OBJ) $(LIBSAMBA_OBJ)
@@ -1075,6 +1081,10 @@
        @echo Linking $@
        @$(CC) $(FLAGS) @PIE_LDFLAGS@ -o $@ $(NSSTEST_OBJ) $(LDFLAGS) $(DYNEXP) 
$(LIBS) $(KRB5LIBS)  $(LDAP_LIBS)
 
+bin/[EMAIL PROTECTED]@: $(PDBTEST_OBJ) bin/.dummy
+       @echo Linking $@
+       @$(CC) $(FLAGS) @PIE_LDFLAGS@ -o $@ $(PDBTEST_OBJ) $(LDFLAGS) $(DYNEXP) 
$(LIBS) $(KRB5LIBS) $(LDAP_LIBS) $(PASSDB_LIBS) $(LIBS) @POPTLIBS@
+
 bin/[EMAIL PROTECTED]@: $(VFSTEST_OBJ) @BUILD_POPT@ bin/.dummy
        @echo Linking $@
        @$(CC) $(FLAGS) @PIE_LDFLAGS@ -o $@ $(VFSTEST_OBJ) $(LDFLAGS) 
$(TERMLDFLAGS) $(TERMLIBS) $(DYNEXP) $(PRINT_LIBS) $(AUTH_LIBS) $(ACL_LIBS) 
$(LIBS) @POPTLIBS@ $(KRB5LIBS) $(LDAP_LIBS) @SMBD_LIBS@

Added: trunk/source/torture/pdbtest.c
===================================================================
--- trunk/source/torture/pdbtest.c      2006-06-12 00:27:21 UTC (rev 16144)
+++ trunk/source/torture/pdbtest.c      2006-06-12 00:33:15 UTC (rev 16145)
@@ -0,0 +1,391 @@
+/* 
+   Unix SMB/CIFS implementation.
+   passdb testing utility
+   
+   Copyright (C) Wilco Baan Hofman 2006
+   Copyright (C) Jelmer Vernooij 2006
+
+   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 2 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, write to the Free Software
+   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+*/
+
+
+#include "includes.h"
+
+static BOOL samu_correct(struct samu *s1, struct samu *s2)
+{
+       BOOL ret = True;
+       uint32 s1_len, s2_len;
+       const char *s1_buf, *s2_buf;
+       const uint8 *d1_buf, *d2_buf;
+               
+       /* Check Unix username */
+       s1_buf = pdb_get_username(s1);
+       s2_buf = pdb_get_username(s2);
+       if (s2_buf == NULL && s1_buf != NULL) {
+               DEBUG(0, ("Username is not set\n"));
+               ret = False;
+       } else if (s1_buf == NULL) {
+               /* Do nothing */
+       } else if (strcmp(s1_buf,s2_buf)) {
+               DEBUG(0, ("Username not written correctly, want %s, got 
\"%s\"\n",
+                                       pdb_get_username(s1),
+                                       pdb_get_username(s2)));
+               ret = False;
+       }
+
+       /* Check NT username */
+       s1_buf = pdb_get_nt_username(s1);
+       s2_buf = pdb_get_nt_username(s2);
+       if (s2_buf == NULL && s1_buf != NULL) {
+               DEBUG(0, ("NT Username is not set\n"));
+               ret = False;
+       } else if (s1_buf == NULL) {
+               /* Do nothing */
+       } else if (strcmp(s1_buf, s2_buf)) {
+               DEBUG(0, ("NT Username not written correctly, want \"%s\", got 
\"%s\"\n",
+                                       pdb_get_nt_username(s1),
+                                       pdb_get_nt_username(s2)));
+               ret = False;
+       }
+
+       /* Check acct ctrl */
+       if (pdb_get_acct_ctrl(s1) != pdb_get_acct_ctrl(s2)) {
+               DEBUG(0, ("Acct ctrl field not written correctly, want %d 
(0x%X), got %d (0x%X)\n",
+                                       pdb_get_acct_ctrl(s1),
+                                       pdb_get_acct_ctrl(s1),
+                                       pdb_get_acct_ctrl(s2),
+                                       pdb_get_acct_ctrl(s2)));
+               ret = False;
+       }
+
+       /* Check NT password */
+       d1_buf = pdb_get_nt_passwd(s1);
+       d2_buf = pdb_get_nt_passwd(s2);
+       if (d2_buf == NULL && d1_buf != NULL) {
+               DEBUG(0, ("NT password is not set\n"));
+               ret = False;
+       } else if (d1_buf == NULL) {
+               /* Do nothing */
+       } else if (memcmp(d1_buf, d2_buf, NT_HASH_LEN)) {
+               DEBUG(0, ("NT password not written correctly\n"));
+               ret = False;
+       }
+
+       /* Check lanman password */
+       d1_buf = pdb_get_lanman_passwd(s1);
+       d2_buf = pdb_get_lanman_passwd(s2);
+       if (d2_buf == NULL && d1_buf != NULL) {
+               DEBUG(0, ("Lanman password is not set\n"));
+       } else if (d1_buf == NULL) {
+               /* Do nothing */
+       } else if (memcmp(d1_buf, d2_buf, NT_HASH_LEN)) {
+               DEBUG(0, ("Lanman password not written correctly\n"));
+               ret = False;
+       }
+
+       /* Check password history */
+       d1_buf = pdb_get_pw_history(s1, &s1_len);
+       d2_buf = pdb_get_pw_history(s2, &s2_len);
+       if (d2_buf == NULL && d1_buf != NULL) {
+               DEBUG(0, ("Password history is not set\n"));
+       } else if (d1_buf == NULL) {
+               /* Do nothing */
+       } else if (s1_len != s1_len) {
+               DEBUG(0, ("Password history not written correctly, lengths 
differ, want %d, got %d\n",
+                                       s1_len, s2_len));
+               ret = False;
+       } else if (strncmp(s1_buf, s2_buf, s1_len)) {
+               DEBUG(0, ("Password history not written correctly\n"));
+               ret = False;
+       }
+
+       /* Check logon time */
+       if (pdb_get_logon_time(s1) != pdb_get_logon_time(s2)) {
+               DEBUG(0, ("Logon time is not written correctly\n"));
+               ret = False;
+       }
+
+       /* Check logoff time */
+       if (pdb_get_logoff_time(s1) != pdb_get_logoff_time(s2)) {
+               DEBUG(0, ("Logoff time is not written correctly\n"));
+               ret = False;
+       }
+       
+       /* Check kickoff time */
+       if (pdb_get_kickoff_time(s1) != pdb_get_logoff_time(s2)) {
+               DEBUG(0, ("Kickoff time is not written correctly\n"));
+               ret = False;
+       }
+       
+       /* Check bad password time */
+       if (pdb_get_bad_password_time(s1) != pdb_get_bad_password_time(s2)) {
+               DEBUG(0, ("Bad password time is not written correctly\n"));
+               ret = False;
+       }
+       
+       /* Check password last set time */
+       if (pdb_get_pass_last_set_time(s1) != pdb_get_pass_last_set_time(s2)) {
+               DEBUG(0, ("Password last set time is not written correctly\n"));
+               ret = False;
+       }
+       
+       /* Check password can change time */
+       if (pdb_get_pass_can_change_time(s1) != 
pdb_get_pass_can_change_time(s2)) {
+               DEBUG(0, ("Password can change time is not written 
correctly\n"));
+               ret = False;
+       }
+       
+       /* Check password must change time */
+       if (pdb_get_pass_must_change_time(s1) != 
pdb_get_pass_must_change_time(s2)) {
+               DEBUG(0, ("Password must change time is not written 
correctly\n"));
+               ret = False;
+       }
+       
+       /* Check logon divs */
+       if (pdb_get_logon_divs(s1) != pdb_get_logon_divs(s2)) {
+               DEBUG(0, ("Logon divs not written correctly\n"));
+               ret = False;
+       }
+       
+       /* Check logon hours */
+       if (pdb_get_hours_len(s1) != pdb_get_hours_len(s2)) {
+               DEBUG(0, ("Logon hours length not written correctly\n"));
+               ret = False;
+       } else if (pdb_get_hours_len(s1) != 0) {
+               d1_buf = pdb_get_hours(s1);
+               d2_buf = pdb_get_hours(s2);
+               if (d2_buf == NULL && d2_buf != NULL) {
+                       DEBUG(0, ("Logon hours is not set\n"));
+                       ret = False;
+               } else if (d1_buf == NULL) {
+                       /* Do nothing */
+               } else if (memcmp(d1_buf, d2_buf, MAX_HOURS_LEN)) {
+                       DEBUG(0, ("Logon hours is not written correctly\n"));
+                       ret = False;
+               }
+       }
+       
+       /* Check profile path */
+       s1_buf = pdb_get_profile_path(s1);
+       s2_buf = pdb_get_profile_path(s2);
+       if (s2_buf == NULL && s1_buf != NULL) {
+               DEBUG(0, ("Profile path is not set\n"));
+               ret = False;
+       } else if (s1_buf == NULL) {
+               /* Do nothing */
+       } else if (strcmp(s1_buf, s2_buf)) {
+               DEBUG(0, ("Profile path is not written correctly\n"));
+               ret = False;
+       }
+
+       /* Check home dir */
+       s1_buf = pdb_get_homedir(s1);
+       s2_buf = pdb_get_homedir(s2);
+       if (s2_buf == NULL && s1_buf != NULL) {
+               DEBUG(0, ("Home dir is not set\n"));
+               ret = False;
+       } else if (s1_buf == NULL) {
+               /* Do nothing */
+       } else if (strcmp(s1_buf, s2_buf)) {
+               DEBUG(0, ("Home dir is not written correctly\n"));
+               ret = False;
+       }
+       
+       /* Check logon script */
+       s1_buf = pdb_get_logon_script(s1);
+       s2_buf = pdb_get_logon_script(s2);
+       if (s2_buf == NULL && s1_buf != NULL) {
+               DEBUG(0, ("Logon script not set\n"));
+               ret = False;
+       } else if (s1_buf == NULL) {
+               /* Do nothing */
+       } else if (strcmp(s1_buf, s2_buf)) {
+               DEBUG(0, ("Logon script is not written correctly\n"));
+               ret = False;
+       }
+       
+       /* TODO Check user and group sids */
+               
+       return ret;     
+}
+
+
+int main(int argc, char **argv)
+{
+       TALLOC_CTX *ctx;
+       struct samu *out = NULL;
+       struct samu *in = NULL;
+       NTSTATUS rv;
+       int i;
+       struct timeval tv;
+       BOOL error = False;
+       struct passwd *pwd;
+       uint8 *buf;
+       uint32 expire, min_age, history;
+       struct pdb_methods *pdb;
+       poptContext pc;
+       const char *backend = NULL, *unix_user = "nobody";
+       struct poptOption long_options[] = {
+               {"username", 'u', POPT_ARG_STRING, &unix_user, 0, "Unix user to 
use for testing", "USERNAME" },
+               {"backend", 'b', POPT_ARG_STRING, &backend, 0, "Backend to use 
if not default", "BACKEND[:SETTINGS]" },
+               POPT_AUTOHELP
+               POPT_COMMON_SAMBA
+               POPT_TABLEEND
+       };
+
+       load_case_tables();
+
+       pc = poptGetContext("vfstest", argc, (const char **) argv,
+                           long_options, 0);
+
+       poptSetOtherOptionHelp(pc, "backend[:settings] username");
+       
+       while(poptGetNextOpt(pc) != -1);
+
+       poptFreeContext(pc);
+
+       /* Load configuration */
+       lp_load(dyn_CONFIGFILE, False, False, True, True);
+       setup_logging("pdbtest", True);
+
+       if (backend == NULL) {
+               backend = lp_passdb_backend();
+       }
+
+       rv = make_pdb_method_name(&pdb, backend);
+       if (NT_STATUS_IS_ERR(rv)) {
+               fprintf(stderr, "Error initializing '%s': %s\n", backend, 
get_friendly_nt_error_msg(rv));
+               exit(1);
+       }
+       
+       ctx = talloc_init("PDBTEST");
+       
+       if (!(out = samu_new(ctx))) {
+               fprintf(stderr, "Can't create samu structure.\n");
+               exit(1);
+       }
+       
+       if ((pwd = getpwnam_alloc(ctx, unix_user)) == NULL) {
+               fprintf(stderr, "Error getting user information for %s\n", 
unix_user);
+               exit(1);
+       }
+       
+       samu_set_unix(out, pwd);
+
+       pdb_set_profile_path(out, "\\\\torture\\profile", PDB_SET);
+       pdb_set_homedir(out, "\\\\torture\\home", PDB_SET);
+       pdb_set_logon_script(out, "torture_script.cmd", PDB_SET);
+
+       pdb_get_account_policy(AP_PASSWORD_HISTORY, &history);
+       if (history * PW_HISTORY_ENTRY_LEN < NT_HASH_LEN) {
+               buf = TALLOC(ctx, NT_HASH_LEN);
+       } else {
+               buf = TALLOC(ctx, history * PW_HISTORY_ENTRY_LEN);
+       }
+
+       /* Generate some random hashes */
+       GetTimeOfDay(&tv);
+       srand(tv.tv_usec);
+       for (i = 0; i < NT_HASH_LEN; i++) {
+               buf[i] = (uint8) rand();
+       }
+       pdb_set_nt_passwd(out, buf, PDB_SET);
+       for (i = 0; i < LM_HASH_LEN; i++) {
+               buf[i] = (uint8) rand();
+       }
+       pdb_set_lanman_passwd(out, buf, PDB_SET);
+       for (i = 0; i < history * PW_HISTORY_ENTRY_LEN; i++) {
+               buf[i] = (uint8) rand();
+       }
+       pdb_set_pw_history(out, buf, history, PDB_SET);
+
+       pdb_get_account_policy(AP_MAX_PASSWORD_AGE, &expire);
+       pdb_get_account_policy(AP_MIN_PASSWORD_AGE, &min_age);
+       pdb_set_pass_last_set_time(out, time(NULL), PDB_SET);
+       
+       if (expire == 0 || expire == (uint32)-1) {
+               pdb_set_pass_must_change_time(out, get_time_t_max(), PDB_SET);
+       } else {
+               pdb_set_pass_must_change_time(out, time(NULL)+expire, PDB_SET);
+       }
+
+       if (min_age == (uint32)-1) {
+               pdb_set_pass_can_change_time(out, 0, PDB_SET);
+       } else {
+               pdb_set_pass_can_change_time(out, time(NULL)+min_age, PDB_SET);
+       }
+       
+       /* Create account */
+       if (!NT_STATUS_IS_OK(rv = pdb->add_sam_account(pdb, out))) {
+               fprintf(stderr, "Error in add_sam_account: %s\n", 
+                               get_friendly_nt_error_msg(rv));
+               exit(1);
+       }
+
+       if (!(in = samu_new(ctx))) {
+               fprintf(stderr, "Can't create samu structure.\n");
+               exit(1);
+       }
+       
+       /* Get account information through getsampwnam() */
+       if (NT_STATUS_IS_ERR(pdb->getsampwnam(pdb, in, out->username))) {
+               fprintf(stderr, "Error getting sampw of added user %s.\n",
+                               out->username);
+               if (!NT_STATUS_IS_OK(rv = pdb->delete_sam_account(pdb, out))) {
+                       fprintf(stderr, "Error in delete_sam_account %s\n", 
+                                       get_friendly_nt_error_msg(rv));
+               }
+               TALLOC_FREE(ctx);
+       }
+       
+       /* Verify integrity */
+       if (samu_correct(out, in)) {
+               printf("User info written correctly\n");
+       } else {
+               printf("User info NOT written correctly\n");
+               error = True;
+       }
+       
+       /* Delete account */
+       if (!NT_STATUS_IS_OK(rv = pdb->delete_sam_account(pdb, out))) {
+               fprintf(stderr, "Error in delete_sam_account %s\n", 
+                                       get_friendly_nt_error_msg(rv));
+       }
+
+       pdb->setsampwent(pdb, False, 0);
+       while (NT_STATUS_IS_OK(pdb->getsampwent(pdb, out))) {
+               if (pdb_get_username(out) == NULL) {
+                       fprintf(stderr, "Got bad username through 
getsampwent()\n");
+                       error = True;
+                       break;
+               }
+               if (NT_STATUS_IS_ERR(pdb->getsampwnam(pdb, in, 
pdb_get_username(out)))) {
+                       fprintf(stderr, "Error getting samu through 
getsampwnam() of an account we got through getsampwent!\n");
+                       error = True;
+                       continue;
+               }
+               if (!samu_correct(out, in)) {
+                       printf("Record gotten through getsampwnam() differs 
from same record through getsampwent()\n");
+               }
+       }
+       pdb->endsampwent(pdb);
+       
+       TALLOC_FREE(ctx);
+
+       if (error) {
+               return 1;
+       }
+       return 0;
+}

Reply via email to