diff options
author | Volker Lendecke <vlendec@samba.org> | 2006-08-08 13:54:43 +0000 |
---|---|---|
committer | Gerald (Jerry) Carter <jerry@samba.org> | 2007-10-10 11:38:35 -0500 |
commit | 21da07ba1fdbef9d8fadfb9bf9fa23afc0a665d1 (patch) | |
tree | a865f0c7630a42d0c565a73da8b483a18210a0d2 | |
parent | 72c605b4fc518b498688ec032004fded0d2250d7 (diff) | |
download | samba-21da07ba1fdbef9d8fadfb9bf9fa23afc0a665d1.tar.gz samba-21da07ba1fdbef9d8fadfb9bf9fa23afc0a665d1.tar.bz2 samba-21da07ba1fdbef9d8fadfb9bf9fa23afc0a665d1.zip |
r17457: Add a test to do some operations on group mapping.
Volker
(This used to be commit 68db058fdf508d5b4d38731ece21f5d92feca00c)
-rw-r--r-- | source3/Makefile.in | 8 | ||||
-rw-r--r-- | source3/include/smb.h | 5 | ||||
-rw-r--r-- | source3/lib/tdb_multikey.c | 59 | ||||
-rwxr-xr-x | source3/script/tests/test_smbtorture_s3.sh | 4 | ||||
-rw-r--r-- | source3/torture/local-groupmap.c | 340 | ||||
-rw-r--r-- | source3/torture/local-multikey.c | 211 | ||||
-rw-r--r-- | source3/torture/torture.c | 169 |
7 files changed, 625 insertions, 171 deletions
diff --git a/source3/Makefile.in b/source3/Makefile.in index fad8763247..1f5a39d9da 100644 --- a/source3/Makefile.in +++ b/source3/Makefile.in @@ -608,10 +608,12 @@ NMBLOOKUP_OBJ = utils/nmblookup.o $(PARAM_OBJ) $(LIBNMB_OBJ) $(RPC_PARSE_OBJ1) $ $(LIB_NONSMBD_OBJ) $(POPT_LIB_OBJ) $(SECRETS_OBJ) $(LIBSAMBA_OBJ) SMBTORTURE_OBJ1 = torture/torture.o torture/nbio.o torture/scanner.o torture/utable.o \ - torture/denytest.o torture/mangle_test.o + torture/denytest.o torture/mangle_test.o \ + torture/local-multikey.o torture/local-groupmap.o SMBTORTURE_OBJ = $(SMBTORTURE_OBJ1) $(PARAM_OBJ) \ - $(LIBSMB_OBJ) $(KRBCLIENT_OBJ) $(LIB_NONSMBD_OBJ) $(SECRETS_OBJ) + $(LIBSMB_OBJ) $(KRBCLIENT_OBJ) $(LIB_NONSMBD_OBJ) $(SECRETS_OBJ) \ + $(PASSDB_OBJ) $(GROUPDB_OBJ) $(SMBLDAP_OBJ) MASKTEST_OBJ = torture/masktest.o $(PARAM_OBJ) $(LIBSMB_OBJ) $(KRBCLIENT_OBJ) \ $(LIB_NONSMBD_OBJ) $(SECRETS_OBJ) @@ -1013,7 +1015,7 @@ bin/nmblookup@EXEEXT@: $(NMBLOOKUP_OBJ) @BUILD_POPT@ bin/.dummy bin/smbtorture@EXEEXT@: $(SMBTORTURE_OBJ) bin/.dummy @echo Linking $@ - @$(CC) $(FLAGS) @PIE_LDFLAGS@ -o $@ $(SMBTORTURE_OBJ) $(LDFLAGS) $(DYNEXP) $(LIBS) $(KRB5LIBS) $(LDAP_LIBS) + @$(CC) $(FLAGS) @PIE_LDFLAGS@ -o $@ $(SMBTORTURE_OBJ) $(LDFLAGS) $(DYNEXP) $(LIBS) $(KRB5LIBS) $(LDAP_LIBS) $(PASSDB_LIBS) bin/talloctort@EXEEXT@: $(TALLOCTORT_OBJ) bin/.dummy @echo Linking $@ diff --git a/source3/include/smb.h b/source3/include/smb.h index 1a55bef428..c80c029f3e 100644 --- a/source3/include/smb.h +++ b/source3/include/smb.h @@ -1853,4 +1853,9 @@ enum usershare_err { /* Different reasons for closing a file. */ enum file_close_type {NORMAL_CLOSE=0,SHUTDOWN_CLOSE,ERROR_CLOSE}; +struct tdb_keyed_iterator { + struct tdb_context *tdb; + TDB_DATA key; +}; + #endif /* _SMB_H */ diff --git a/source3/lib/tdb_multikey.c b/source3/lib/tdb_multikey.c index 77e63c5aaa..e8febd6f7b 100644 --- a/source3/lib/tdb_multikey.c +++ b/source3/lib/tdb_multikey.c @@ -34,7 +34,7 @@ static struct { enum TDB_ERROR t; NTSTATUS n; } tdb_to_ntstatus_map[] = { { 0, NT_STATUS_OK }, }; -static NTSTATUS map_ntstatus_from_tdb(struct tdb_context *t) +NTSTATUS map_ntstatus_from_tdb(struct tdb_context *t) { enum TDB_ERROR err = tdb_error(t); int i = 0; @@ -474,7 +474,7 @@ NTSTATUS tdb_update_keyed(struct tdb_context *tdb, const char *primary_key, if ((primary_key == NULL) || (strlen(primary_key) != PRIMARY_KEY_LENGTH) || - (strncmp(primary_key, "KEYPRIM/", 7) != 0)) { + (strncmp(primary_key, "KEYPRIM/", 8) != 0)) { return NT_STATUS_INVALID_PARAMETER; } @@ -528,3 +528,58 @@ NTSTATUS tdb_update_keyed(struct tdb_context *tdb, const char *primary_key, return status; } + +static int iterator_destructor(void *p) +{ + struct tdb_keyed_iterator *i = (struct tdb_keyed_iterator *)p; + SAFE_FREE(i->key.dptr); + return 0; +} + +struct tdb_keyed_iterator *tdb_enum_keyed(TALLOC_CTX *mem_ctx, + struct tdb_context *tdb) +{ + struct tdb_keyed_iterator *result = TALLOC_P( + mem_ctx, struct tdb_keyed_iterator); + + if (result == NULL) { + DEBUG(0, ("talloc failed\n")); + return result; + } + + result->tdb = tdb; + result->key = tdb_firstkey(tdb); + talloc_set_destructor(result, iterator_destructor); + return result; +} + +BOOL tdb_next_keyed(struct tdb_keyed_iterator *it, TDB_DATA *data) +{ + if (it->key.dptr == NULL) { + return False; + } + + while (True) { + TDB_DATA tmp; + + if ((it->key.dsize == PRIMARY_KEY_LENGTH+1) && + (strncmp(it->key.dptr, "KEYPRIM/", 8) == 0)) { + + *data = tdb_fetch(it->tdb, it->key); + + tmp = tdb_nextkey(it->tdb, it->key); + SAFE_FREE(it->key.dptr); + it->key = tmp; + + return (data->dptr != NULL); + } + + tmp = tdb_nextkey(it->tdb, it->key); + SAFE_FREE(it->key.dptr); + it->key = tmp; + + if (it->key.dptr == NULL) { + return False; + } + } +} diff --git a/source3/script/tests/test_smbtorture_s3.sh b/source3/script/tests/test_smbtorture_s3.sh index 3675f39257..2ee2584b22 100755 --- a/source3/script/tests/test_smbtorture_s3.sh +++ b/source3/script/tests/test_smbtorture_s3.sh @@ -25,7 +25,7 @@ tests="$tests OPLOCK1 OPLOCK2 OPLOCK3" tests="$tests DIR DIR1 TCON TCONDEV RW1 RW2 RW3" tests="$tests OPEN XCOPY RENAME DELETE PROPERTIES W2K" tests="$tests PIPE_NUMBER TCON2 IOCTL CHKPATH FDSESS LOCAL-SUBSTITUTE" -tests="$tests LOCAL-MULTIKEY" +tests="$tests LOCAL-MULTIKEY LOCAL-GROUPMAP" skipped1="RANDOMIPC NEGNOWAIT NBENCH ERRMAPEXTRACT TRANS2SCAN NTTRANSSCAN" skipped2="DENY1 DENY2 OPENATTR CASETABLE EATEST" @@ -42,7 +42,7 @@ for t in $tests; do fi start="" name="$t" - testit "$name" $VALGRIND $SRCDIR/bin/smbtorture $ADDARGS $unc -U"$username"%"$password" $t || failed=`expr $failed + 1` + testit "$name" $VALGRIND $SRCDIR/bin/smbtorture $ADDARGS $unc -U"$username"%"$password" -S "$CONFFILE" $t || failed=`expr $failed + 1` done testok $0 $failed diff --git a/source3/torture/local-groupmap.c b/source3/torture/local-groupmap.c new file mode 100644 index 0000000000..bd6f060e71 --- /dev/null +++ b/source3/torture/local-groupmap.c @@ -0,0 +1,340 @@ +/* + Unix SMB/CIFS implementation. + Run some local tests for group mapping + Copyright (C) Volker Lendecke 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" + +#define CHECK_STATUS(_status, _expected) do { \ + if (!NT_STATUS_EQUAL(_status, _expected)) { \ + printf("(%d) Incorrect status %s - should be %s\n", \ + __LINE__, nt_errstr(status), nt_errstr(_expected)); \ + goto fail; \ + }} while (0) + +static NTSTATUS create_v2_mapping(struct tdb_context *tdb, + const char *sid, gid_t gid, + enum SID_NAME_USE type, + const char *nt_name, + const char *comment) +{ + TDB_DATA key, data; + NTSTATUS status; + + ZERO_STRUCT(data); + + if (asprintf(&key.dptr, "UNIXGROUP/%s", sid) < 0) { + d_fprintf(stderr, "(%s) asprintf failed\n", + __location__); + return NT_STATUS_NO_MEMORY; + } + key.dsize = strlen(key.dptr)+1; + + if (!tdb_pack_append(NULL, &data.dptr, &data.dsize, "ddff", + (uint32)gid, (uint32)type, nt_name, comment)) { + d_fprintf(stderr, "(%s) tdb_pack_append failed\n", + __location__); + SAFE_FREE(key.dptr); + return NT_STATUS_NO_MEMORY; + } + + if (tdb_store(tdb, key, data, TDB_INSERT) < 0) { + status = map_ntstatus_from_tdb(tdb); + d_fprintf(stderr, "(%s) tdb_store failed: %s\n", __location__, + nt_errstr(status)); + SAFE_FREE(key.dptr); + TALLOC_FREE(data.dptr); + return status; + } + + SAFE_FREE(key.dptr); + TALLOC_FREE(data.dptr); + return NT_STATUS_OK; +} + +#define NUM_ENTRIES (50) + +static NTSTATUS create_v2_db(BOOL invalid) +{ + struct tdb_context *tdb; + NTSTATUS status; + int i; + + tdb = tdb_open_log(lock_path("group_mapping.tdb"), 0, TDB_DEFAULT, + O_RDWR|O_CREAT, 0600); + if (tdb == NULL) { + d_fprintf(stderr, "(%s) tdb_open_log failed: %s\n", + __location__, strerror(errno)); + status = map_nt_error_from_unix(errno); + goto fail; + } + + /* Empty the database */ + tdb_traverse(tdb, tdb_traverse_delete_fn, NULL); + + if (tdb_store_int32(tdb, "INFO/version", 2) < 0) { + status = map_ntstatus_from_tdb(tdb); + d_fprintf(stderr, "(%s) tdb_store_uint32 failed: %s\n", + __location__, nt_errstr(status)); + goto fail; + } + + for (i=1000; i<1000+NUM_ENTRIES; i++) { + char *sid, *name; + if (asprintf(&sid, "S-1-5-21-744032650-3806004166-77016029-%d", + i) < 0) { + d_fprintf(stderr, "(%s) asprintf failed\n", + __location__); + goto fail; + } + if (asprintf(&name, "Unix group %d", i) < 0) { + d_fprintf(stderr, "(%s) asprintf failed\n", + __location__); + SAFE_FREE(sid); + goto fail; + } + status = create_v2_mapping(tdb, sid, (gid_t)i, + SID_NAME_DOM_GRP, name, name); + SAFE_FREE(sid); + SAFE_FREE(name); + CHECK_STATUS(status, NT_STATUS_OK); + } + status = create_v2_mapping(tdb, "S-1-5-32-544", 10000, + SID_NAME_ALIAS, "Administrators", + "Machine Admins"); + CHECK_STATUS(status, NT_STATUS_OK); + status = create_v2_mapping(tdb, "S-1-5-32-545", 10001, + SID_NAME_ALIAS, "Users", "Machine Users"); + CHECK_STATUS(status, NT_STATUS_OK); + + if (invalid) { + /* Map 10001 to two different SIDs */ + status = create_v2_mapping(tdb, "S-1-5-32-999", 10001, + SID_NAME_ALIAS, "Overlapping", + "Invalid mapping"); + CHECK_STATUS(status, NT_STATUS_OK); + } + + status = create_v2_mapping(tdb, "S-1-5-32-546", -1, + SID_NAME_ALIAS, "notthere", "To remove"); + CHECK_STATUS(status, NT_STATUS_OK); + + status = NT_STATUS_OK; + fail: + if (tdb != NULL) { + tdb_close(tdb); + } + return status; +} + +static BOOL groupmap_diff(const GROUP_MAP *m1, const GROUP_MAP *m2) +{ + return ((sid_compare(&m1->sid, &m2->sid) != 0) || + (m1->gid != m2->gid) || + (m1->sid_name_use != m2->sid_name_use) || + (strcmp(m1->nt_name, m2->nt_name) != 0) || + (strcmp(m1->comment, m2->comment) != 0)); +} + +#undef GROUPDB_V3 + +BOOL run_local_groupmap(int dummy) +{ + TALLOC_CTX *mem_ctx; + BOOL ret = False; + NTSTATUS status; + GROUP_MAP *maps = NULL; + size_t num_maps = 0; + + mem_ctx = talloc_init("run_local_groupmap"); + if (mem_ctx == NULL) { + d_fprintf(stderr, "(%s) talloc_init failed\n", + __location__); + return False; + } + +#ifdef GROUPDB_V3 + status = create_v2_db(True); + if (!NT_STATUS_IS_OK(status)) { + goto fail; + } + + { + GROUP_MAP map; + if (pdb_getgrgid(&map, 10001)) { + d_fprintf(stderr, "(%s) upgrading an invalid group db " + "worked\n", __location__); + goto fail; + } + } +#endif + + status = create_v2_db(False); + if (!NT_STATUS_IS_OK(status)) { + goto fail; + } + + /* This tests upgrading the database, as well as listing */ + + if (!pdb_enum_group_mapping(NULL, SID_NAME_UNKNOWN, &maps, &num_maps, + False)) { + d_fprintf(stderr, "(%s) pdb_enum_group_mapping failed\n", + __location__); + goto fail; + } + + if (num_maps != NUM_ENTRIES+2) { + d_fprintf(stderr, "(%s) expected %d entries, got %d\n", + __location__, NUM_ENTRIES+2, num_maps); + goto fail; + } + + /* See if getgrsid, getgrgid and getgrnam find the same entry */ + + { + DOM_SID sid; + GROUP_MAP map, map1; + string_to_sid(&sid, "S-1-5-32-545"); + + ZERO_STRUCT(map); + if (!pdb_getgrsid(&map, &sid)) { + d_fprintf(stderr, "(%s) pdb_getgrsid failed\n", + __location__); + goto fail; + } + + ZERO_STRUCT(map1); + if (!pdb_getgrgid(&map1, map.gid)) { + d_fprintf(stderr, "(%s) pdb_getgrgid failed\n", + __location__); + goto fail; + } + + if (groupmap_diff(&map1, &map)) { + d_fprintf(stderr, "(%s) getgrsid/getgrgid disagree\n", + __location__); + goto fail; + } + + ZERO_STRUCT(map1); + if (!pdb_getgrnam(&map1, map.nt_name)) { + d_fprintf(stderr, "(%s) pdb_getgrnam failed\n", + __location__); + goto fail; + } + + if (groupmap_diff(&map1, &map)) { + d_fprintf(stderr, "(%s) getgrsid/getgrnam disagree\n", + __location__); + goto fail; + } + } + + /* See if pdb_delete_group_mapping_entry works */ + + { + DOM_SID sid; + GROUP_MAP map, map1; + string_to_sid(&sid, "S-1-5-32-545"); + + if (!pdb_getgrsid(&map, &sid)) { + d_fprintf(stderr, "(%s) did not find S-1-5-32-545\n", + __location__); + goto fail; + } + + status = pdb_delete_group_mapping_entry(sid); + CHECK_STATUS(status, NT_STATUS_OK); + status = pdb_delete_group_mapping_entry(sid); +#ifdef GROUPDB_V3 + CHECK_STATUS(status, NT_STATUS_NOT_FOUND); +#else + CHECK_STATUS(status, NT_STATUS_UNSUCCESSFUL); +#endif + + if (pdb_getgrsid(&map1, &sid)) { + d_fprintf(stderr, "(%s) getgrsid found deleted " + "entry\n", __location__); + goto fail; + } + + if (pdb_getgrgid(&map1, map.gid)) { + d_fprintf(stderr, "(%s) getgrgid found deleted " + "entry\n", __location__); + goto fail; + } + + if (pdb_getgrnam(&map1, map.nt_name)) { + d_fprintf(stderr, "(%s) getgrnam found deleted " + "entry\n", __location__); + goto fail; + } + + } + + /* See if pdb_update_group_mapping_entry works */ + + { + DOM_SID sid; + gid_t oldgid; + GROUP_MAP map, map1; + string_to_sid(&sid, "S-1-5-32-544"); + + if (!pdb_getgrsid(&map, &sid)) { + d_fprintf(stderr, "(%s) did not find S-1-5-32-544\n", + __location__); + goto fail; + } + + oldgid = map.gid; + map.gid = 4711; + + status = pdb_update_group_mapping_entry(&map); + CHECK_STATUS(status, NT_STATUS_OK); + + if (pdb_getgrgid(&map1, oldgid)) { + d_fprintf(stderr, "(%s) getgrgid found outdated " + "entry\n", __location__); + goto fail; + } + + /* Change to an existing entry, see "create_db_v2" */ + + map.gid = 1000; + status = pdb_update_group_mapping_entry(&map); +#ifdef GROUPDB_V3 + CHECK_STATUS(status, NT_STATUS_OBJECTID_EXISTS); + if (!pdb_getgrgid(&map1, 4711)) { + d_fprintf(stderr, "(%s) update_group changed entry " + "upon failure\n", __location__); + goto fail; + } +#else + CHECK_STATUS(status, NT_STATUS_OK); +#endif + } + + ret = True; + fail: + if (maps != NULL) { + SAFE_FREE(maps); + } + TALLOC_FREE(mem_ctx); + return ret; +} + diff --git a/source3/torture/local-multikey.c b/source3/torture/local-multikey.c new file mode 100644 index 0000000000..83de2f0b40 --- /dev/null +++ b/source3/torture/local-multikey.c @@ -0,0 +1,211 @@ +/* + Unix SMB/CIFS implementation. + Run some local tests on the local tdb multikey wrapper + Copyright (C) Volker Lendecke 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 char **key_fn(TALLOC_CTX *mem_ctx, TDB_DATA data, + void *private_data) +{ + fstring key, value; + char **result; + + result = TALLOC_ARRAY(mem_ctx, char *, 3); + if (result == NULL) { + return NULL; + } + + if (tdb_unpack(data.dptr, data.dsize, "ff", key, value) < 0) { + d_fprintf(stderr, "tdb_unpack failed\n"); + TALLOC_FREE(result); + return NULL; + } + result[0] = talloc_strdup(result, key); + result[1] = talloc_strdup(result, value); + result[2] = NULL; + + if ((result[0] == NULL) || (result[1] == NULL)) { + d_fprintf(stderr, "talloc_strdup failed\n"); + TALLOC_FREE(result); + return NULL; + } + + return result; +} + +static NTSTATUS multikey_add(struct tdb_context *tdb, const char *key, + const char *value) +{ + NTSTATUS status; + TDB_DATA data; + + data.dptr = NULL; + data.dsize = 0; + + if (!tdb_pack_append(NULL, &data.dptr, &data.dsize, + "ff", key, value)) { + return NT_STATUS_NO_MEMORY; + } + + status = tdb_add_keyed(tdb, key_fn, data, NULL); + TALLOC_FREE(data.dptr); + return status; +} + +#define CHECK_STATUS(_status, _expected) do { \ + if (!NT_STATUS_EQUAL(_status, _expected)) { \ + printf("(%d) Incorrect status %s - should be %s\n", \ + __LINE__, nt_errstr(status), nt_errstr(_expected)); \ + ret = False; \ + goto fail; \ + }} while (0) + +#define NUM_ELEMENTS (50) + +BOOL run_local_multikey(int dummy) +{ + TALLOC_CTX *mem_ctx; + char *prim; + const char *tdbname = "multi_key_test.tdb"; + struct tdb_context *tdb = NULL; + NTSTATUS status; + BOOL ret = False; + TDB_DATA data; + int i; + fstring key,value; + + unlink(tdbname); + + mem_ctx = talloc_init("run_local_multikey"); + if (mem_ctx == NULL) { + d_fprintf(stderr, "talloc_init failed\n"); + return False; + } + + tdb = tdb_open(tdbname, 0, 0, O_CREAT|O_RDWR, 0644); + if (tdb == NULL) { + d_fprintf(stderr, "tdb_open failed: %s\n", strerror(errno)); + goto fail; + } + + for (i=0; i<NUM_ELEMENTS; i++) { + fstr_sprintf(key, "KEY%d", i); + fstr_sprintf(value, "VAL%d", i); + + status = multikey_add(tdb, key, value); + if (!NT_STATUS_IS_OK(status)) { + d_fprintf(stderr, "tdb_add_keyed failed: %s\n", + nt_errstr(status)); + goto fail; + } + } + + { + struct tdb_keyed_iterator *it = tdb_enum_keyed(mem_ctx, tdb); + if (it == NULL) { + d_printf("tdb_enum_keyed failed\n"); + goto fail; + } + + i = 0; + + while (tdb_next_keyed(it, &data)) { + i += 1; + if (i > 1000) { + d_printf("tdb_next_keyed overrun\n"); + goto fail; + } + } + + if (i != NUM_ELEMENTS) { + d_printf("counted %d, elements, expected %d\n", + i, NUM_ELEMENTS); + goto fail; + } + } + + status = multikey_add(tdb, "KEY35", "FOOO"); + CHECK_STATUS(status, NT_STATUS_OBJECTID_EXISTS); + status = multikey_add(tdb, "KEY42", "VAL45"); + CHECK_STATUS(status, NT_STATUS_OBJECTID_EXISTS); + status = multikey_add(tdb, "FOO", "VAL45"); + CHECK_STATUS(status, NT_STATUS_OBJECTID_EXISTS); + + for (i=0; i<NUM_ELEMENTS; i++) { + fstr_sprintf(key, "KEY%d", i); + fstr_sprintf(value, "VAL%d", i); + + status = tdb_find_keyed(mem_ctx, tdb, 0, key, &data, &prim); + CHECK_STATUS(status, NT_STATUS_OK); + status = tdb_find_keyed(mem_ctx, tdb, 1, value, &data, &prim); + CHECK_STATUS(status, NT_STATUS_OK); + status = tdb_find_keyed(mem_ctx, tdb, 1, key, &data, &prim); + CHECK_STATUS(status, NT_STATUS_NOT_FOUND); + status = tdb_find_keyed(mem_ctx, tdb, 0, value, &data, &prim); + CHECK_STATUS(status, NT_STATUS_NOT_FOUND); + } + + status = tdb_find_keyed(mem_ctx, tdb, 0, "FOO", &data, &prim); + CHECK_STATUS(status, NT_STATUS_NOT_FOUND); + status = tdb_find_keyed(mem_ctx, tdb, 1, "BAR", &data, &prim); + CHECK_STATUS(status, NT_STATUS_NOT_FOUND); + + status = tdb_find_keyed(mem_ctx, tdb, 0, "KEY0", &data, &prim); + CHECK_STATUS(status, NT_STATUS_OK); + + ZERO_STRUCT(data); + if (tdb_pack_append(mem_ctx, &data.dptr, &data.dsize, "ff", + "NEWKEY", "NEWVAL") < 0) { + d_printf("tdb_pack_alloc failed\n"); + goto fail; + } + + status = tdb_update_keyed(tdb, prim, key_fn, data, NULL); + CHECK_STATUS(status, NT_STATUS_OK); + + status = tdb_find_keyed(mem_ctx, tdb, 0, "KEY0", &data, &prim); + CHECK_STATUS(status, NT_STATUS_NOT_FOUND); + status = tdb_find_keyed(mem_ctx, tdb, 1, "VAL0", &data, &prim); + CHECK_STATUS(status, NT_STATUS_NOT_FOUND); + status = tdb_find_keyed(mem_ctx, tdb, 0, "NEWKEY", &data, &prim); + CHECK_STATUS(status, NT_STATUS_OK); + status = tdb_find_keyed(mem_ctx, tdb, 1, "NEWVAL", &data, &prim); + CHECK_STATUS(status, NT_STATUS_OK); + + status = tdb_del_keyed(tdb, key_fn, prim, NULL); + CHECK_STATUS(status, NT_STATUS_OK); + + for (i=1; i<NUM_ELEMENTS; i++) { + fstr_sprintf(key, "KEY%d", i); + status = tdb_find_keyed(mem_ctx, tdb, 0, key, &data, &prim); + CHECK_STATUS(status, NT_STATUS_OK); + status = tdb_del_keyed(tdb, key_fn, prim, NULL); + CHECK_STATUS(status, NT_STATUS_OK); + } + + ret = True; + fail: + if (tdb != NULL) { + tdb_close(tdb); + } + unlink(tdbname); + TALLOC_FREE(mem_ctx); + return ret; +} + diff --git a/source3/torture/torture.c b/source3/torture/torture.c index 009eb84787..67d13068bb 100644 --- a/source3/torture/torture.c +++ b/source3/torture/torture.c @@ -4797,169 +4797,6 @@ static BOOL run_local_substitute(int dummy) return (diff == 0); } -static char **key_fn(TALLOC_CTX *mem_ctx, TDB_DATA data, - void *private_data) -{ - fstring key, value; - char **result; - - result = TALLOC_ARRAY(mem_ctx, char *, 3); - if (result == NULL) { - return NULL; - } - - if (tdb_unpack(data.dptr, data.dsize, "ff", key, value) < 0) { - d_fprintf(stderr, "tdb_unpack failed\n"); - TALLOC_FREE(result); - return NULL; - } - result[0] = talloc_strdup(result, key); - result[1] = talloc_strdup(result, value); - result[2] = NULL; - - if ((result[0] == NULL) || (result[1] == NULL)) { - d_fprintf(stderr, "talloc_strdup failed\n"); - TALLOC_FREE(result); - return NULL; - } - - return result; -} - -static NTSTATUS multikey_add(struct tdb_context *tdb, const char *key, - const char *value) -{ - NTSTATUS status; - TDB_DATA data; - - data.dptr = NULL; - data.dsize = 0; - - if (!tdb_pack_append(NULL, &data.dptr, &data.dsize, - "ff", key, value)) { - return NT_STATUS_NO_MEMORY; - } - - status = tdb_add_keyed(tdb, key_fn, data, NULL); - TALLOC_FREE(data.dptr); - return status; -} - -#define CHECK_STATUS(_status, _expected) do { \ - if (!NT_STATUS_EQUAL(_status, _expected)) { \ - printf("(%d) Incorrect status %s - should be %s\n", \ - __LINE__, nt_errstr(status), nt_errstr(_expected)); \ - ret = False; \ - goto fail; \ - }} while (0) - -static BOOL run_local_multikey(int dummy) -{ - TALLOC_CTX *mem_ctx; - char *prim; - const char *tdbname = "multi_key_test.tdb"; - struct tdb_context *tdb = NULL; - NTSTATUS status; - BOOL ret = False; - TDB_DATA data; - int i; - fstring key,value; - - unlink(tdbname); - - mem_ctx = talloc_init("run_local_multikey"); - if (mem_ctx == NULL) { - d_fprintf(stderr, "talloc_init failed\n"); - return False; - } - - tdb = tdb_open(tdbname, 0, 0, O_CREAT|O_RDWR, 0644); - if (tdb == NULL) { - d_fprintf(stderr, "tdb_open failed: %s\n", strerror(errno)); - goto fail; - } - - for (i=0; i<200; i++) { - fstr_sprintf(key, "KEY%d", i); - fstr_sprintf(value, "VAL%d", i); - - status = multikey_add(tdb, key, value); - if (!NT_STATUS_IS_OK(status)) { - d_fprintf(stderr, "tdb_add_keyed failed: %s\n", - nt_errstr(status)); - goto fail; - } - } - - status = multikey_add(tdb, "KEY35", "FOOO"); - CHECK_STATUS(status, NT_STATUS_OBJECTID_EXISTS); - status = multikey_add(tdb, "KEY42", "VAL45"); - CHECK_STATUS(status, NT_STATUS_OBJECTID_EXISTS); - status = multikey_add(tdb, "FOO", "VAL45"); - CHECK_STATUS(status, NT_STATUS_OBJECTID_EXISTS); - - for (i=0; i<200; i++) { - fstr_sprintf(key, "KEY%d", i); - fstr_sprintf(value, "VAL%d", i); - - status = tdb_find_keyed(mem_ctx, tdb, 0, key, &data, &prim); - CHECK_STATUS(status, NT_STATUS_OK); - status = tdb_find_keyed(mem_ctx, tdb, 1, value, &data, &prim); - CHECK_STATUS(status, NT_STATUS_OK); - status = tdb_find_keyed(mem_ctx, tdb, 1, key, &data, &prim); - CHECK_STATUS(status, NT_STATUS_NOT_FOUND); - status = tdb_find_keyed(mem_ctx, tdb, 0, value, &data, &prim); - CHECK_STATUS(status, NT_STATUS_NOT_FOUND); - } - - status = tdb_find_keyed(mem_ctx, tdb, 0, "FOO", &data, &prim); - CHECK_STATUS(status, NT_STATUS_NOT_FOUND); - status = tdb_find_keyed(mem_ctx, tdb, 1, "BAR", &data, &prim); - CHECK_STATUS(status, NT_STATUS_NOT_FOUND); - - status = tdb_find_keyed(mem_ctx, tdb, 0, "KEY0", &data, &prim); - CHECK_STATUS(status, NT_STATUS_OK); - - ZERO_STRUCT(data); - if (tdb_pack_append(mem_ctx, &data.dptr, &data.dsize, "ff", - "NEWKEY", "NEWVAL") < 0) { - d_printf("tdb_pack_alloc failed\n"); - goto fail; - } - - status = tdb_update_keyed(tdb, prim, key_fn, data, NULL); - CHECK_STATUS(status, NT_STATUS_OK); - - status = tdb_find_keyed(mem_ctx, tdb, 0, "KEY0", &data, &prim); - CHECK_STATUS(status, NT_STATUS_NOT_FOUND); - status = tdb_find_keyed(mem_ctx, tdb, 1, "VAL0", &data, &prim); - CHECK_STATUS(status, NT_STATUS_NOT_FOUND); - status = tdb_find_keyed(mem_ctx, tdb, 0, "NEWKEY", &data, &prim); - CHECK_STATUS(status, NT_STATUS_OK); - status = tdb_find_keyed(mem_ctx, tdb, 1, "NEWVAL", &data, &prim); - CHECK_STATUS(status, NT_STATUS_OK); - - status = tdb_del_keyed(tdb, key_fn, prim, NULL); - CHECK_STATUS(status, NT_STATUS_OK); - - for (i=1; i<200; i++) { - fstr_sprintf(key, "KEY%d", i); - status = tdb_find_keyed(mem_ctx, tdb, 0, key, &data, &prim); - CHECK_STATUS(status, NT_STATUS_OK); - status = tdb_del_keyed(tdb, key_fn, prim, NULL); - CHECK_STATUS(status, NT_STATUS_OK); - } - - ret = True; - fail: - if (tdb != NULL) { - tdb_close(tdb); - } - unlink(tdbname); - TALLOC_FREE(mem_ctx); - return ret; -} - static double create_procs(BOOL (*fn)(int), BOOL *result) { int i, status; @@ -5112,6 +4949,7 @@ static struct { { "EATEST", run_eatest, 0}, { "LOCAL-SUBSTITUTE", run_local_substitute, 0}, { "LOCAL-MULTIKEY", run_local_multikey, 0}, + { "LOCAL-GROUPMAP", run_local_groupmap, 0}, {NULL, NULL, 0}}; @@ -5260,7 +5098,7 @@ static void usage(void) fstrcpy(workgroup, lp_workgroup()); - while ((opt = getopt(argc, argv, "p:hW:U:n:N:O:o:m:Ld:Ac:ks:b:")) != EOF) { + while ((opt = getopt(argc, argv, "p:hW:U:n:N:O:o:m:Ld:Ac:ks:b:S:")) != EOF) { switch (opt) { case 'p': port_to_use = atoi(optarg); @@ -5268,6 +5106,9 @@ static void usage(void) case 's': srandom(atoi(optarg)); break; + case 'S': + lp_load(optarg,True,False,False,True); + break; case 'W': fstrcpy(workgroup,optarg); break; |