From 39dec5ee6645c8f3be3d4dac78efea47df61b606 Mon Sep 17 00:00:00 2001 From: Volker Lendecke Date: Mon, 19 Nov 2007 15:01:03 +0100 Subject: r26029: In RAW-STREAMS test listing of streams (This used to be commit 8dbe10be30798ea78cc2a5e6c70e6bc5202fa05c) --- source4/torture/raw/streams.c | 143 ++++++++++++++++++++++++++++++++++-------- 1 file changed, 116 insertions(+), 27 deletions(-) (limited to 'source4/torture/raw') diff --git a/source4/torture/raw/streams.c b/source4/torture/raw/streams.c index 14d674e482..ac9d0d36aa 100644 --- a/source4/torture/raw/streams.c +++ b/source4/torture/raw/streams.c @@ -46,7 +46,8 @@ /* check that a stream has the right contents */ -static bool check_stream(struct smbcli_state *cli, TALLOC_CTX *mem_ctx, +static bool check_stream(struct smbcli_state *cli, const char *location, + TALLOC_CTX *mem_ctx, const char *fname, const char *sname, const char *value) { @@ -61,15 +62,16 @@ static bool check_stream(struct smbcli_state *cli, TALLOC_CTX *mem_ctx, if (value == NULL) { if (fnum != -1) { - printf("should have failed stream open of %s\n", full_name); + printf("(%s) should have failed stream open of %s\n", + location, full_name); return false; } return true; } if (fnum == -1) { - printf("Failed to open stream '%s' - %s\n", - full_name, smbcli_errstr(cli->tree)); + printf("(%s) Failed to open stream '%s' - %s\n", + location, full_name, smbcli_errstr(cli->tree)); return false; } @@ -77,13 +79,13 @@ static bool check_stream(struct smbcli_state *cli, TALLOC_CTX *mem_ctx, ret = smbcli_read(cli->tree, fnum, buf, 0, strlen(value)+11); if (ret != strlen(value)) { - printf("Failed to read %lu bytes from stream '%s' - got %d\n", - (long)strlen(value), full_name, (int)ret); + printf("(%s) Failed to read %lu bytes from stream '%s' - got %d\n", + location, (long)strlen(value), full_name, (int)ret); return false; } if (memcmp(buf, value, strlen(value)) != 0) { - printf("Bad data in stream\n"); + printf("(%s) Bad data in stream\n", location); return false; } @@ -91,6 +93,82 @@ static bool check_stream(struct smbcli_state *cli, TALLOC_CTX *mem_ctx, return true; } +static int qsort_string(const void *v1, const void *v2) +{ + char * const *s1 = v1; + char * const *s2 = 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; + return strcmp(s1->stream_name.s, s2->stream_name.s); +} + +static bool check_stream_list(struct smbcli_state *cli, const char *fname, + int num_exp, const char **exp) +{ + union smb_fileinfo finfo; + NTSTATUS status; + int i; + TALLOC_CTX *tmp_ctx = talloc_new(cli); + char **exp_sort; + struct stream_struct *stream_sort; + bool ret = false; + + 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; + } + + 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; + } + + exp_sort = talloc_memdup(tmp_ctx, exp, num_exp * sizeof(*exp)); + + if (exp_sort == NULL) { + goto fail; + } + + 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)); + + if (stream_sort == NULL) { + goto fail; + } + + qsort(stream_sort, finfo.stream_info.out.num_streams, + sizeof(*stream_sort), qsort_stream); + + for (i=0; itree, mem_ctx, &io); CHECK_STATUS(status, NT_STATUS_NOT_A_DIRECTORY); - printf("creating a stream on a non-existant file\n"); + printf("(%s) creating a stream on a non-existant file\n", __location__); io.ntcreatex.in.create_options = 0; io.ntcreatex.in.fname = sname1; status = smb_raw_open(cli->tree, mem_ctx, &io); CHECK_STATUS(status, NT_STATUS_OK); fnum = io.ntcreatex.out.file.fnum; - ret &= check_stream(cli, mem_ctx, fname, "Stream One", NULL); + ret &= check_stream(cli, __location__, mem_ctx, fname, "Stream One", NULL); - printf("check that open of base file is allowed\n"); + printf("(%s) check that open of base file is allowed\n", __location__); io.ntcreatex.in.open_disposition = NTCREATEX_DISP_OPEN; io.ntcreatex.in.fname = fname; status = smb_raw_open(cli->tree, mem_ctx, &io); CHECK_STATUS(status, NT_STATUS_OK); smbcli_close(cli->tree, io.ntcreatex.out.file.fnum); - printf("writing to stream\n"); + printf("(%s) writing to stream\n", __location__); retsize = smbcli_write(cli->tree, fnum, 0, "test data", 0, 9); CHECK_VALUE(retsize, 9); smbcli_close(cli->tree, fnum); - ret &= check_stream(cli, mem_ctx, fname, "Stream One", "test data"); + ret &= check_stream(cli, __location__, mem_ctx, fname, "Stream One", "test data"); io.ntcreatex.in.open_disposition = NTCREATEX_DISP_OPEN; io.ntcreatex.in.fname = sname1; @@ -153,45 +236,50 @@ static bool test_stream_io(struct smbcli_state *cli, TALLOC_CTX *mem_ctx) CHECK_STATUS(status, NT_STATUS_OK); fnum = io.ntcreatex.out.file.fnum; - printf("modifying stream\n"); + printf("(%s) modifying stream\n", __location__); retsize = smbcli_write(cli->tree, fnum, 0, "MORE DATA ", 5, 10); CHECK_VALUE(retsize, 10); smbcli_close(cli->tree, fnum); - ret &= check_stream(cli, mem_ctx, fname, "Stream One:$FOO", NULL); + ret &= check_stream(cli, __location__, mem_ctx, fname, "Stream One:$FOO", NULL); - printf("creating a stream2 on a existing file\n"); + printf("(%s) creating a stream2 on a existing file\n", __location__); io.ntcreatex.in.fname = sname2; io.ntcreatex.in.open_disposition = NTCREATEX_DISP_OPEN_IF; status = smb_raw_open(cli->tree, mem_ctx, &io); CHECK_STATUS(status, NT_STATUS_OK); fnum = io.ntcreatex.out.file.fnum; - printf("modifying stream\n"); + printf("(%s) modifying stream\n", __location__); retsize = smbcli_write(cli->tree, fnum, 0, "SECOND STREAM", 0, 13); CHECK_VALUE(retsize, 13); smbcli_close(cli->tree, fnum); - ret &= check_stream(cli, mem_ctx, fname, "Stream One", "test MORE DATA "); - ret &= check_stream(cli, mem_ctx, fname, "Stream One:$DATA", "test MORE DATA "); - ret &= check_stream(cli, mem_ctx, fname, "Stream One:", NULL); - ret &= check_stream(cli, mem_ctx, fname, "Second Stream", "SECOND STREAM"); - ret &= check_stream(cli, mem_ctx, fname, "Second Stream:$DATA", "SECOND STREAM"); - ret &= check_stream(cli, mem_ctx, fname, "Second Stream:", NULL); - ret &= check_stream(cli, mem_ctx, fname, "Second Stream:$FOO", NULL); + ret &= check_stream(cli, __location__, mem_ctx, fname, "Stream One", "test MORE DATA "); + ret &= check_stream(cli, __location__, mem_ctx, fname, "Stream One:$DATA", "test MORE DATA "); + ret &= check_stream(cli, __location__, mem_ctx, fname, "Stream One:", NULL); + ret &= check_stream(cli, __location__, mem_ctx, fname, "Second Stream", "SECOND STREAM"); + ret &= check_stream(cli, __location__, mem_ctx, fname, "Second Stream:$DATA", "SECOND STREAM"); + ret &= check_stream(cli, __location__, mem_ctx, fname, "Second Stream:", NULL); + ret &= check_stream(cli, __location__, mem_ctx, fname, "Second Stream:$FOO", NULL); - printf("deleting stream\n"); + check_stream_list(cli, fname, 3, three); + + printf("(%s) deleting stream\n", __location__); status = smbcli_unlink(cli->tree, sname1); CHECK_STATUS(status, NT_STATUS_OK); - printf("delete a stream via delete-on-close\n"); + check_stream_list(cli, fname, 2, two); + + printf("(%s) delete a stream via delete-on-close\n", __location__); io.ntcreatex.in.fname = sname2; io.ntcreatex.in.create_options = NTCREATEX_OPTIONS_DELETE_ON_CLOSE; io.ntcreatex.in.share_access = NTCREATEX_SHARE_ACCESS_DELETE; io.ntcreatex.in.access_mask = SEC_RIGHTS_FILE_ALL; io.ntcreatex.in.open_disposition = NTCREATEX_DISP_OPEN; + status = smb_raw_open(cli->tree, mem_ctx, &io); CHECK_STATUS(status, NT_STATUS_OK); fnum = io.ntcreatex.out.file.fnum; @@ -200,8 +288,9 @@ static bool test_stream_io(struct smbcli_state *cli, TALLOC_CTX *mem_ctx) status = smbcli_unlink(cli->tree, sname2); CHECK_STATUS(status, NT_STATUS_OBJECT_NAME_NOT_FOUND); + check_stream_list(cli, fname, 1, one); - printf("deleting file\n"); + printf("(%s) deleting file\n", __location__); status = smbcli_unlink(cli->tree, fname); CHECK_STATUS(status, NT_STATUS_OK); -- cgit