summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--source4/torture/raw/streams.c131
1 files changed, 54 insertions, 77 deletions
diff --git a/source4/torture/raw/streams.c b/source4/torture/raw/streams.c
index 9640d78029..c0cb64b106 100644
--- a/source4/torture/raw/streams.c
+++ b/source4/torture/raw/streams.c
@@ -29,28 +29,14 @@
#define BASEDIR "\\teststreams"
-#define CHECK_STATUS(status, correct) do { \
- if (!NT_STATUS_EQUAL(status, correct)) { \
- printf("(%s) Incorrect status %s - should be %s\n", \
- __location__, nt_errstr(status), nt_errstr(correct)); \
- ret = false; \
- goto done; \
- }} while (0)
-
-#define CHECK_VALUE(v, correct) do { \
- if ((v) != (correct)) { \
- printf("(%s) Incorrect value %s=%d - should be %d\n", \
- __location__, #v, (int)v, (int)correct); \
- ret = false; \
- }} while (0)
-
-#define CHECK_NTTIME(v, correct) do { \
- if ((v) != (correct)) { \
- printf("(%s) Incorrect value %s=%llu - should be %llu\n", \
- __location__, #v, (unsigned long long)v, \
- (unsigned long long)correct); \
- ret = false; \
- }} while (0)
+#define CHECK_STATUS(status, correct) \
+ torture_assert_ntstatus_equal_goto(tctx,status,correct,ret,done,"CHECK_STATUS")
+
+#define CHECK_VALUE(v, correct) \
+ torture_assert_int_equal(tctx,v,correct,"CHECK_VALUE")
+
+#define CHECK_NTTIME(v, correct) \
+ torture_assert_u64_equal(tctx,v,correct,"CHECK_NTTIME")
#define CHECK_STR(v, correct) do { \
bool ok; \
@@ -65,12 +51,10 @@
} else { \
ok = false; \
} \
- if (!ok) { \
- printf("(%s) Incorrect value %s='%s' - should be '%s'\n", \
- __location__, #v, (v)?(v):"NULL", \
- (correct)?(correct):"NULL"); \
- ret = false; \
- }} while (0)
+ torture_assert(tctx,ok,\
+ talloc_asprintf(tctx, "got '%s', expected '%s'",\
+ (v)?(v):"NULL", (correct)?(correct):"NULL")); \
+} while (0)
/*
check that a stream has the right contents
@@ -124,19 +108,20 @@ static bool check_stream(struct smbcli_state *cli, const char *location,
static int qsort_string(const void *v1, const void *v2)
{
- char * const *s1 = v1;
- char * const *s2 = v2;
+ char * const *s1 = (char * const *)v1;
+ char * const *s2 = (char * const *)v2;
return strcmp(*s1, *s2);
}
static int qsort_stream(const void *v1, const void *v2)
{
- const struct stream_struct * s1 = v1;
- const struct stream_struct * s2 = v2;
+ const struct stream_struct * s1 = (const struct stream_struct *)v1;
+ const struct stream_struct * s2 = (const struct stream_struct *)v2;
return strcmp(s1->stream_name.s, s2->stream_name.s);
}
-static bool check_stream_list(struct smbcli_state *cli, const char *fname,
+static bool check_stream_list(struct torture_context *tctx,
+ struct smbcli_state *cli, const char *fname,
int num_exp, const char **exp)
{
union smb_fileinfo finfo;
@@ -146,43 +131,36 @@ static bool check_stream_list(struct smbcli_state *cli, const char *fname,
char **exp_sort;
struct stream_struct *stream_sort;
bool ret = false;
+ int fail = -1;
finfo.generic.level = RAW_FILEINFO_STREAM_INFO;
finfo.generic.in.file.path = fname;
status = smb_raw_pathinfo(cli->tree, tmp_ctx, &finfo);
- if (!NT_STATUS_IS_OK(status)) {
- d_fprintf(stderr, "(%s) smb_raw_pathinfo failed: %s\n",
- __location__, nt_errstr(status));
- goto fail;
- }
+ CHECK_STATUS(status, NT_STATUS_OK);
- if (finfo.stream_info.out.num_streams != num_exp) {
- d_fprintf(stderr, "(%s) expected %d streams, got %d\n",
- __location__, num_exp,
- finfo.stream_info.out.num_streams);
- goto fail;
- }
+ CHECK_VALUE(finfo.stream_info.out.num_streams, num_exp);
if (num_exp == 0) {
ret = true;
- goto fail;
+ goto done;
}
- exp_sort = talloc_memdup(tmp_ctx, exp, num_exp * sizeof(*exp));
+ exp_sort = (char **)talloc_memdup(tmp_ctx, exp, num_exp * sizeof(*exp));
if (exp_sort == NULL) {
- goto fail;
+ goto done;
}
qsort(exp_sort, num_exp, sizeof(*exp_sort), qsort_string);
- stream_sort = talloc_memdup(tmp_ctx, finfo.stream_info.out.streams,
- finfo.stream_info.out.num_streams *
- sizeof(*stream_sort));
+ stream_sort = (struct stream_struct *)talloc_memdup(tmp_ctx,
+ finfo.stream_info.out.streams,
+ finfo.stream_info.out.num_streams *
+ sizeof(*stream_sort));
if (stream_sort == NULL) {
- goto fail;
+ goto done;
}
qsort(stream_sort, finfo.stream_info.out.num_streams,
@@ -190,23 +168,22 @@ static bool check_stream_list(struct smbcli_state *cli, const char *fname,
for (i=0; i<num_exp; i++) {
if (strcmp(exp_sort[i], stream_sort[i].stream_name.s) != 0) {
- d_fprintf(stderr, "(%s) expected stream name %s, got "
- "%s\n", __location__, exp_sort[i],
- stream_sort[i].stream_name.s);
+ fail = i;
goto show_streams;
}
}
ret = true;
- fail:
+done:
talloc_free(tmp_ctx);
return ret;
show_streams:
for (i=0; i<num_exp; i++) {
- d_fprintf(stderr, "stream names '%s' '%s'\n",
- exp_sort[i], stream_sort[i].stream_name.s);
+ torture_comment(tctx, "stream names '%s' '%s'\n",
+ exp_sort[i], stream_sort[i].stream_name.s);
}
+ CHECK_STR(stream_sort[fail].stream_name.s, exp_sort[fail]);
talloc_free(tmp_ctx);
return ret;
}
@@ -280,7 +257,7 @@ static bool test_stream_dir(struct torture_context *tctx,
CHECK_STATUS(status, NT_STATUS_FILE_IS_A_DIRECTORY);
printf("(%s) list the streams on the basedir\n", __location__);
- ret &= check_stream_list(cli, BASEDIR, 0, NULL);
+ ret &= check_stream_list(tctx, cli, BASEDIR, 0, NULL);
done:
smbcli_deltree(cli->tree, BASEDIR);
return ret;
@@ -383,13 +360,13 @@ static bool test_stream_io(struct torture_context *tctx,
ret &= check_stream(cli, __location__, tctx, fname, "Second Stream:", NULL);
ret &= check_stream(cli, __location__, tctx, fname, "Second Stream:$FOO", NULL);
- check_stream_list(cli, fname, 3, three);
+ check_stream_list(tctx, cli, fname, 3, three);
printf("(%s) deleting stream\n", __location__);
status = smbcli_unlink(cli->tree, sname1);
CHECK_STATUS(status, NT_STATUS_OK);
- check_stream_list(cli, fname, 2, two);
+ check_stream_list(tctx, cli, fname, 2, two);
printf("(%s) delete a stream via delete-on-close\n", __location__);
io.ntcreatex.in.fname = sname2;
@@ -406,7 +383,7 @@ static bool test_stream_io(struct torture_context *tctx,
status = smbcli_unlink(cli->tree, sname2);
CHECK_STATUS(status, NT_STATUS_OBJECT_NAME_NOT_FOUND);
- check_stream_list(cli, fname, 1, one);
+ check_stream_list(tctx, cli, fname, 1, one);
io.ntcreatex.in.open_disposition = NTCREATEX_DISP_CREATE;
io.ntcreatex.in.fname = sname1;
@@ -794,7 +771,7 @@ static bool test_stream_names(struct torture_context *tctx,
status = smb_raw_open(cli->tree, tctx, &io);
CHECK_STATUS(status, NT_STATUS_OBJECT_NAME_INVALID);
- ret &= check_stream_list(cli, fname, 4, four);
+ ret &= check_stream_list(tctx, cli, fname, 4, four);
smbcli_close(cli->tree, fnum1);
smbcli_close(cli->tree, fnum2);
@@ -805,7 +782,7 @@ static bool test_stream_names(struct torture_context *tctx,
status = smb_raw_pathinfo(cli->tree, tctx, &finfo);
CHECK_STATUS(status, NT_STATUS_OK);
- ret &= check_stream_list(cli, fname, 4, four);
+ ret &= check_stream_list(tctx, cli, fname, 4, four);
for (i=0; i < 4; i++) {
NTTIME write_time;
@@ -911,7 +888,7 @@ static bool test_stream_names(struct torture_context *tctx,
CHECK_VALUE(stinfo.all_info.out.ea_size,
finfo.all_info.out.ea_size);
- ret &= check_stream_list(cli, fname, 4, four);
+ ret &= check_stream_list(tctx, cli, fname, 4, four);
smbcli_close(cli->tree, fnum1);
talloc_free(path);
@@ -927,7 +904,7 @@ static bool test_stream_names(struct torture_context *tctx,
CHECK_STATUS(status, NT_STATUS_OK);
fnum1 = io.ntcreatex.out.file.fnum;
- ret &= check_stream_list(cli, fname, 5, five1);
+ ret &= check_stream_list(tctx, cli, fname, 5, five1);
ZERO_STRUCT(sinfo);
sinfo.rename_information.level = RAW_SFILEINFO_RENAME_INFORMATION;
@@ -938,7 +915,7 @@ static bool test_stream_names(struct torture_context *tctx,
status = smb_raw_setfileinfo(cli->tree, &sinfo);
CHECK_STATUS(status, NT_STATUS_OK);
- ret &= check_stream_list(cli, fname, 5, five2);
+ ret &= check_stream_list(tctx, cli, fname, 5, five2);
ZERO_STRUCT(sinfo);
sinfo.rename_information.level = RAW_SFILEINFO_RENAME_INFORMATION;
@@ -949,7 +926,7 @@ static bool test_stream_names(struct torture_context *tctx,
status = smb_raw_setfileinfo(cli->tree, &sinfo);
CHECK_STATUS(status, NT_STATUS_OBJECT_NAME_COLLISION);
- ret &= check_stream_list(cli, fname, 5, five2);
+ ret &= check_stream_list(tctx, cli, fname, 5, five2);
ZERO_STRUCT(sinfo);
sinfo.rename_information.level = RAW_SFILEINFO_RENAME_INFORMATION;
@@ -962,10 +939,10 @@ static bool test_stream_names(struct torture_context *tctx,
torture_setting_bool(tctx, "samba3", false)) {
/* why should this rename be considered invalid?? */
CHECK_STATUS(status, NT_STATUS_OK);
- ret &= check_stream_list(cli, fname, 4, four);
+ ret &= check_stream_list(tctx, cli, fname, 4, four);
} else {
CHECK_STATUS(status, NT_STATUS_INVALID_PARAMETER);
- ret &= check_stream_list(cli, fname, 5, five2);
+ ret &= check_stream_list(tctx, cli, fname, 5, five2);
}
@@ -1430,7 +1407,7 @@ static bool test_stream_create_disposition(struct torture_context *tctx,
status = smb_raw_open(cli->tree, tctx, &io);
CHECK_STATUS(status, NT_STATUS_OK);
smbcli_close(cli->tree, io.ntcreatex.out.file.fnum);
- if (!check_stream_list(cli, fname, 2, stream_list)) {
+ if (!check_stream_list(tctx, cli, fname, 2, stream_list)) {
goto done;
}
@@ -1442,7 +1419,7 @@ static bool test_stream_create_disposition(struct torture_context *tctx,
status = smb_raw_open(cli->tree, tctx, &io);
CHECK_STATUS(status, NT_STATUS_OK);
smbcli_close(cli->tree, io.ntcreatex.out.file.fnum);
- if (!check_stream_list(cli, fname, 1, &default_stream_name)) {
+ if (!check_stream_list(tctx, cli, fname, 1, &default_stream_name)) {
goto done;
}
@@ -1459,7 +1436,7 @@ static bool test_stream_create_disposition(struct torture_context *tctx,
status = smb_raw_open(cli->tree, tctx, &io);
CHECK_STATUS(status, NT_STATUS_OK);
smbcli_close(cli->tree, io.ntcreatex.out.file.fnum);
- if (!check_stream_list(cli, fname, 1, &default_stream_name)) {
+ if (!check_stream_list(tctx, cli, fname, 1, &default_stream_name)) {
goto done;
}
@@ -1476,7 +1453,7 @@ static bool test_stream_create_disposition(struct torture_context *tctx,
status = smb_raw_open(cli->tree, tctx, &io);
CHECK_STATUS(status, NT_STATUS_OK);
smbcli_close(cli->tree, io.ntcreatex.out.file.fnum);
- if (!check_stream_list(cli, fname, 1, &default_stream_name)) {
+ if (!check_stream_list(tctx, cli, fname, 1, &default_stream_name)) {
goto done;
}
@@ -1495,7 +1472,7 @@ static bool test_stream_create_disposition(struct torture_context *tctx,
status = smb_raw_open(cli->tree, tctx, &io);
CHECK_STATUS(status, NT_STATUS_OK);
smbcli_close(cli->tree, io.ntcreatex.out.file.fnum);
- if (!check_stream_list(cli, fname, 2, stream_list)) {
+ if (!check_stream_list(tctx, cli, fname, 2, stream_list)) {
goto done;
}
@@ -1522,7 +1499,7 @@ static bool test_stream_create_disposition(struct torture_context *tctx,
status = smb_raw_open(cli->tree, tctx, &io);
CHECK_STATUS(status, NT_STATUS_OK);
smbcli_close(cli->tree, io.openx.out.file.fnum);
- if (!check_stream_list(cli, fname, 1, &default_stream_name)) {
+ if (!check_stream_list(tctx, cli, fname, 1, &default_stream_name)) {
goto done;
}
@@ -1559,7 +1536,7 @@ static bool test_stream_large_streaminfo(struct torture_context *tctx,
}
lstream_name[LONG_STREAM_SIZE - 1] = '\0';
- printf("(%s) Creating a file with a lot of streams\n", __location__);
+ torture_comment(tctx, "(%s) Creating a file with a lot of streams\n", __location__);
for (i = 0; i < 10000; i++) {
fname_stream = talloc_asprintf(tctx, "%s:%s%d", fname,
lstream_name, i);
@@ -1600,7 +1577,7 @@ static bool test_stream_attributes(struct torture_context *tctx,
return false;
}
- printf ("(%s) testing attribute setting on stream\n", __location__);
+ torture_comment(tctx, "(%s) testing attribute setting on stream\n", __location__);
fname_stream = talloc_asprintf(tctx, "%s:%s", fname, stream);