s3: Add a test that makes a chained open break an oplock

Autobuild-User: Volker Lendecke <vl@samba.org>
Autobuild-Date: Wed Feb 29 01:13:03 CET 2012 on sn-devel-104

diff --git a/source3/Makefile.in b/source3/Makefile.in
index 0a189b54ddf74fbec8dd5f84158675c3cf57b764..71a18d6263dbab33d19fd61e43bc784d9ea0bdaf 100644 (file)
--- a/source3/Makefile.in
+++ b/source3/Makefile.in
@@ -1266,6 +1266,7 @@ SMBTORTURE_OBJ1 = torture/torture.o torture/nbio.o torture/scanner.o torture/uta
                torture/test_case_insensitive.o \
                torture/test_posix_append.o \
                torture/test_smb2.o \
+               torture/test_chain3.o \
                torture/test_authinfo_structs.o \
                torture/test_cleanup.o \
                torture/t_strappend.o

diff --git a/source3/selftest/tests.py b/source3/selftest/tests.py
index 163cfb45b4dc12ba19e7a1fdbcb0e1bd0f4e84db..eab1356c006eb914d62aeecf5d6a569d838ce80b 100755 (executable)
--- a/source3/selftest/tests.py
+++ b/source3/selftest/tests.py
@@ -69,6 +69,7 @@ tests=[ "FDPASS", "LOCK1", "LOCK2", "LOCK3", "LOCK4", "LOCK5", "LOCK6", "LOCK7",
         "DIR", "DIR1", "DIR-CREATETIME", "TCON", "TCONDEV", "RW1", "RW2", "RW3", "RW-SIGNING",
         "OPEN", "XCOPY", "RENAME", "DELETE", "DELETE-LN", "PROPERTIES", "W2K",
         "TCON2", "IOCTL", "CHKPATH", "FDSESS", "CHAIN1", "CHAIN2",
+        "CHAIN3",
         "GETADDRINFO", "POSIX", "UID-REGRESSION-TEST", "SHORTNAME-TEST",
         "POSIX-APPEND",
         "CASE-INSENSITIVE-CREATE", "SMB2-BASIC", "NTTRANS-FSCTL", "SMB2-NEGPROT",

diff --git a/source3/torture/proto.h b/source3/torture/proto.h
index 8d661aada91094ceb8040944c687e6713b08c7de..e65b2722d0a1d320cb1a360534efcea65b395ff2 100644 (file)
--- a/source3/torture/proto.h
+++ b/source3/torture/proto.h
@@ -99,6 +99,7 @@ bool run_smb2_session_reconnect(int dummy);
 bool run_smb2_tcon_dependence(int dummy);
 bool run_smb2_multi_channel(int dummy);
 bool run_smb2_session_reauth(int dummy);
+bool run_chain3(int dummy);
 bool run_local_conv_auth_info(int dummy);
 bool run_local_sprintf_append(int dummy);
 bool run_cleanup1(int dummy);

diff --git a/source3/torture/test_chain3.c b/source3/torture/test_chain3.c
new file mode 100644 (file)
index 0000000..7b9eeb0
--- /dev/null
+++ b/source3/torture/test_chain3.c
@@ -0,0 +1,294 @@
+/*
+   Unix SMB/CIFS implementation.
+   Test smbd chain routines
+
+   Copyright (C) Volker Lendecke 2012
+
+   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/proto.h"
+#include "libsmb/libsmb.h"
+#include "system/filesys.h"
+#include "async_smb.h"
+#include "lib/util/tevent_ntstatus.h"
+#include "libcli/security/security.h"
+
+struct chain3_andx_state {
+       uint16_t fnum;
+       size_t written;
+       char str[6];
+};
+
+static void chain3_andx_open_done(struct tevent_req *subreq);
+static void chain3_andx_write_done(struct tevent_req *subreq);
+static void chain3_andx_close_done(struct tevent_req *subreq);
+
+static struct tevent_req *chain3_andx_send(TALLOC_CTX *mem_ctx,
+                                          struct tevent_context *ev,
+                                          struct cli_state *cli,
+                                          const char *fname)
+{
+       struct tevent_req *req, *subreq;
+       struct tevent_req *smbreqs[3];
+       struct chain3_andx_state *state;
+       NTSTATUS status;
+
+       req = tevent_req_create(mem_ctx, &state, struct chain3_andx_state);
+       if (req == NULL) {
+               return NULL;
+       }
+
+       strlcpy(state->str, "hello", sizeof(state->str));
+
+       subreq = cli_openx_create(state, ev, cli, fname,
+                                 O_CREAT|O_RDWR, 0, &smbreqs[0]);
+       if (tevent_req_nomem(subreq, req)) {
+               return tevent_req_post(req, ev);
+       }
+       tevent_req_set_callback(subreq, chain3_andx_open_done, req);
+
+       subreq = cli_write_andx_create(state, ev, cli, 0, 0,
+                                      (const uint8_t *)state->str, 0,
+                                      strlen(state->str)+1,
+                                      smbreqs, 1, &smbreqs[1]);
+       if (tevent_req_nomem(subreq, req)) {
+               return tevent_req_post(req, ev);
+       }
+       tevent_req_set_callback(subreq, chain3_andx_write_done, req);
+
+       subreq = cli_close_create(state, ev, cli, 0, &smbreqs[2]);
+       if (tevent_req_nomem(subreq, req)) {
+               return tevent_req_post(req, ev);
+       }
+       tevent_req_set_callback(subreq, chain3_andx_close_done, req);
+
+       status = cli_smb_chain_send(smbreqs, ARRAY_SIZE(smbreqs));
+       if (tevent_req_nterror(req, status)) {
+               return tevent_req_post(req, ev);
+       }
+       return req;
+}
+
+static void chain3_andx_open_done(struct tevent_req *subreq)
+{
+       struct tevent_req *req = tevent_req_callback_data(
+               subreq, struct tevent_req);
+       struct chain3_andx_state *state = tevent_req_data(
+               req, struct chain3_andx_state);
+       NTSTATUS status;
+
+       status = cli_openx_recv(subreq, &state->fnum);
+       printf("cli_openx returned %s, fnum=%u\n", nt_errstr(status),
+              (unsigned)state->fnum);
+       TALLOC_FREE(subreq);
+       if (tevent_req_nterror(req, status)) {
+               return;
+       }
+}
+
+static void chain3_andx_write_done(struct tevent_req *subreq)
+{
+       struct tevent_req *req = tevent_req_callback_data(
+               subreq, struct tevent_req);
+       struct chain3_andx_state *state = tevent_req_data(
+               req, struct chain3_andx_state);
+       NTSTATUS status;
+
+       status = cli_write_andx_recv(subreq, &state->written);
+       printf("cli_write_andx returned %s, written=%u\n", nt_errstr(status),
+              (unsigned)state->written);
+       TALLOC_FREE(subreq);
+       if (tevent_req_nterror(req, status)) {
+               return;
+       }
+}
+
+static void chain3_andx_close_done(struct tevent_req *subreq)
+{
+       struct tevent_req *req = tevent_req_callback_data(
+               subreq, struct tevent_req);
+       NTSTATUS status;
+
+       status = cli_close_recv(subreq);
+       printf("cli_close returned %s\n", nt_errstr(status));
+       TALLOC_FREE(subreq);
+       if (tevent_req_nterror(req, status)) {
+               return;
+       }
+       tevent_req_done(req);
+}
+
+static NTSTATUS chain3_andx_recv(struct tevent_req *req)
+{
+       return tevent_req_simple_recv_ntstatus(req);
+}
+
+struct chain3_state {
+       struct tevent_context *ev;
+       struct cli_state *cli;
+       const char *fname;
+       uint16_t fnum;
+};
+
+static void chain3_got_break(struct tevent_req *subreq);
+static void chain3_ntcreate_done(struct tevent_req *subreq);
+static void chain3_break_close_done(struct tevent_req *subreq);
+static void chain3_andx_done(struct tevent_req *subreq);
+
+static struct tevent_req *chain3_send(TALLOC_CTX *mem_ctx,
+                                     struct tevent_context *ev)
+{
+       struct tevent_req *req, *subreq;
+       struct chain3_state *state;
+
+       req = tevent_req_create(mem_ctx, &state, struct chain3_state);
+       if (req == NULL) {
+               return NULL;
+       }
+       state->ev = ev;
+       state->fname = "chain3.txt";
+
+       if (!torture_open_connection(&state->cli, 0)) {
+               tevent_req_nterror(req, NT_STATUS_UNSUCCESSFUL);
+               return tevent_req_post(req, ev);
+       }
+
+       subreq = cli_smb_oplock_break_waiter_send(
+               state, state->ev, state->cli);
+       if (tevent_req_nomem(subreq, req)) {
+               return tevent_req_post(req, ev);
+       }
+       tevent_req_set_callback(subreq, chain3_got_break, req);
+
+       subreq = cli_ntcreate_send(
+               state, state->ev, state->cli, state->fname,
+               REQUEST_OPLOCK|REQUEST_BATCH_OPLOCK,
+               GENERIC_READ_ACCESS|GENERIC_WRITE_ACCESS,
+               FILE_ATTRIBUTE_NORMAL,
+               FILE_SHARE_READ|FILE_SHARE_WRITE|FILE_SHARE_DELETE,
+               FILE_OVERWRITE_IF, 0, 0);
+       if (tevent_req_nomem(subreq, req)) {
+               return tevent_req_post(req, ev);
+       }
+       tevent_req_set_callback(subreq, chain3_ntcreate_done, req);
+       return req;
+}
+
+static void chain3_got_break(struct tevent_req *subreq)
+{
+       struct tevent_req *req = tevent_req_callback_data(
+               subreq, struct tevent_req);
+       struct chain3_state *state = tevent_req_data(
+               req, struct chain3_state);
+       uint16_t fnum;
+       uint8_t level;
+       NTSTATUS status;
+
+       status = cli_smb_oplock_break_waiter_recv(subreq, &fnum, &level);
+       TALLOC_FREE(subreq);
+       printf("cli_smb_oplock_break_waiter_recv returned %s\n",
+              nt_errstr(status));
+       if (tevent_req_nterror(req, status)) {
+               return;
+       }
+       subreq = cli_close_send(state, state->ev, state->cli, fnum);
+       if (tevent_req_nomem(subreq, req)) {
+               return;
+       }
+       tevent_req_set_callback(subreq, chain3_break_close_done, req);
+}
+
+static void chain3_break_close_done(struct tevent_req *subreq)
+{
+       struct tevent_req *req = tevent_req_callback_data(
+               subreq, struct tevent_req);
+       NTSTATUS status;
+
+       status = cli_close_recv(subreq);
+       TALLOC_FREE(subreq);
+       printf("cli_close_recv returned %s\n", nt_errstr(status));
+       if (tevent_req_nterror(req, status)) {
+               return;
+       }
+}
+
+static void chain3_ntcreate_done(struct tevent_req *subreq)
+{
+       struct tevent_req *req = tevent_req_callback_data(
+               subreq, struct tevent_req);
+       struct chain3_state *state = tevent_req_data(
+               req, struct chain3_state);
+       NTSTATUS status;
+
+       status = cli_ntcreate_recv(subreq, &state->fnum);
+       TALLOC_FREE(subreq);
+       printf("cli_ntcreate returned %s, fnum=%u\n", nt_errstr(status),
+              (unsigned)state->fnum);
+       if (tevent_req_nterror(req, status)) {
+               return;
+       }
+
+       subreq = chain3_andx_send(state, state->ev, state->cli, state->fname);
+       if (tevent_req_nomem(subreq, req)) {
+               return;
+       }
+       tevent_req_set_callback(subreq, chain3_andx_done, req);
+}
+
+static void chain3_andx_done(struct tevent_req *subreq)
+{
+       struct tevent_req *req = tevent_req_callback_data(
+               subreq, struct tevent_req);
+       NTSTATUS status;
+
+       status = chain3_andx_recv(subreq);
+       TALLOC_FREE(subreq);
+       printf("chain3_andx_recv returned %s\n", nt_errstr(status));
+       if (tevent_req_nterror(req, status)) {
+               return;
+       }
+       tevent_req_done(req);
+}
+
+static NTSTATUS chain3_recv(struct tevent_req *req)
+{
+       return tevent_req_simple_recv_ntstatus(req);
+}
+
+bool run_chain3(int dummy)
+{
+       TALLOC_CTX *frame = talloc_stackframe();
+       struct tevent_context *ev;
+       struct tevent_req *req;
+       NTSTATUS status = NT_STATUS_NO_MEMORY;
+
+       ev = tevent_context_init(frame);
+       if (ev == NULL) {
+               goto fail;
+       }
+       req = chain3_send(frame, ev);
+       if (req == NULL) {
+               goto fail;
+       }
+       if (!tevent_req_poll_ntstatus(req, ev, &status)) {
+               goto fail;
+       }
+       status = chain3_recv(req);
+fail:
+       TALLOC_FREE(frame);
+       printf("run_chain3 returns %s\n", nt_errstr(status));
+       return NT_STATUS_IS_OK(status);
+}

diff --git a/source3/torture/torture.c b/source3/torture/torture.c
index 0b2727cfcab337deb560cc8edd2f4fb212938442..99896ecc812eeb93057e5cd56f142fc271746c75 100644 (file)
--- a/source3/torture/torture.c
+++ b/source3/torture/torture.c
@@ -8881,6 +8881,7 @@ static struct {
        { "SESSSETUP_BENCH", run_sesssetup_bench, 0},
        { "CHAIN1", run_chain1, 0},
        { "CHAIN2", run_chain2, 0},
+       { "CHAIN3", run_chain3, 0},
        { "WINDOWS-WRITE", run_windows_write, 0},
        { "NTTRANS-CREATE", run_nttrans_create, 0},
        { "NTTRANS-FSCTL", run_nttrans_fsctl, 0},

diff --git a/source3/wscript_build b/source3/wscript_build
index b24090e07db576601f713c1d613cefc7e589f212..8e17c374fe14baea892271954ef6dc7470ebbf25 100755 (executable)
--- a/source3/wscript_build
+++ b/source3/wscript_build
@@ -566,6 +566,7 @@ SMBTORTURE_SRC1 = '''torture/torture.c torture/nbio.c torture/scanner.c torture/
                torture/test_nttrans_fsctl.c
                torture/test_case_insensitive.c
                torture/test_notify_online.c
+               torture/test_chain3.c
                torture/test_smb2.c
                torture/test_authinfo_structs.c
                 torture/test_smbsock_any_connect.c