diff options
author | Volker Lendecke <vl@samba.org> | 2010-10-06 18:46:43 +0200 |
---|---|---|
committer | Volker Lendecke <vl@samba.org> | 2010-10-08 21:11:45 +0200 |
commit | 220aa311d100814be6d396dff96f3825fc01985f (patch) | |
tree | ad156604f02a52b4cab09f7472e86c1a0e1fc2ac | |
parent | 5e482aeebf43f95aa003c68d25f7773a944b227e (diff) | |
download | samba-220aa311d100814be6d396dff96f3825fc01985f.tar.gz samba-220aa311d100814be6d396dff96f3825fc01985f.tar.bz2 samba-220aa311d100814be6d396dff96f3825fc01985f.zip |
s3: Add a little test for the echo responder
-rw-r--r-- | source3/Makefile.in | 3 | ||||
-rw-r--r-- | source3/torture/proto.h | 1 | ||||
-rw-r--r-- | source3/torture/test_async_echo.c | 142 | ||||
-rw-r--r-- | source3/torture/torture.c | 1 |
4 files changed, 146 insertions, 1 deletions
diff --git a/source3/Makefile.in b/source3/Makefile.in index c80a8742d7..4d2587fdaa 100644 --- a/source3/Makefile.in +++ b/source3/Makefile.in @@ -1192,6 +1192,7 @@ NMBLOOKUP_OBJ = utils/nmblookup.o $(PARAM_OBJ) $(LIBNMB_OBJ) \ SMBTORTURE_OBJ1 = torture/torture.o torture/nbio.o torture/scanner.o torture/utable.o \ torture/denytest.o torture/mangle_test.o \ torture/nbench.o \ + torture/test_async_echo.o \ torture/test_posix_append.o SMBTORTURE_OBJ = $(SMBTORTURE_OBJ1) $(PARAM_OBJ) $(TLDAP_OBJ) \ @@ -1199,7 +1200,7 @@ SMBTORTURE_OBJ = $(SMBTORTURE_OBJ1) $(PARAM_OBJ) $(TLDAP_OBJ) \ @LIBWBCLIENT_STATIC@ \ ../nsswitch/libwbclient/wbc_async.o \ ../nsswitch/libwbclient/wb_reqtrans.o \ - $(LIBNDR_GEN_OBJ0) + $(LIBMSRPC_OBJ) $(LIBMSRPC_GEN_OBJ) $(LIBCLI_ECHO_OBJ) MASKTEST_OBJ = torture/masktest.o $(PARAM_OBJ) $(LIBSMB_OBJ) $(KRBCLIENT_OBJ) \ $(LIB_NONSMBD_OBJ) \ diff --git a/source3/torture/proto.h b/source3/torture/proto.h index 4f9a1807aa..63004345ca 100644 --- a/source3/torture/proto.h +++ b/source3/torture/proto.h @@ -84,5 +84,6 @@ bool torture_casetable(int dummy); bool run_posix_append(int dummy); bool run_nbench2(int dummy); +bool run_async_echo(int dummy); #endif /* __TORTURE_H__ */ diff --git a/source3/torture/test_async_echo.c b/source3/torture/test_async_echo.c new file mode 100644 index 0000000000..13d4679bc9 --- /dev/null +++ b/source3/torture/test_async_echo.c @@ -0,0 +1,142 @@ +/* + Unix SMB/CIFS implementation. + Run the async echo responder + Copyright (C) Volker Lendecke 2010 + + 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 "librpc/gen_ndr/cli_echo.h" + +static void rpccli_sleep_done(struct tevent_req *req) +{ + int *done = (int *)tevent_req_callback_data_void(req); + NTSTATUS status; + uint32_t result; + + status = rpccli_echo_TestSleep_recv(req, talloc_tos(), &result); + TALLOC_FREE(req); + printf("sleep returned %s, %d\n", nt_errstr(status), (int)result); + *done -= 1; +} + +static void cli_echo_done(struct tevent_req *req) +{ + int *done = (int *)tevent_req_callback_data_void(req); + NTSTATUS status; + + status = cli_echo_recv(req); + TALLOC_FREE(req); + printf("echo returned %s\n", nt_errstr(status)); + *done -= 1; +} + +static void cli_close_done(struct tevent_req *req) +{ + int *done = (int *)tevent_req_callback_data_void(req); + NTSTATUS status; + size_t written; + + status = cli_write_andx_recv(req, &written); + TALLOC_FREE(req); + printf("close returned %s\n", nt_errstr(status)); + *done -= 1; +} + +bool run_async_echo(int dummy) +{ + struct cli_state *cli = NULL; + struct rpc_pipe_client *p; + struct tevent_context *ev; + struct tevent_req *req; + NTSTATUS status; + bool ret = false; + int i, num_reqs; + uint8_t buf[32768]; + + printf("Starting ASYNC_ECHO\n"); + + ev = tevent_context_init(talloc_tos()); + if (ev == NULL) { + printf("tevent_context_init failed\n"); + goto fail; + } + + if (!torture_open_connection(&cli, 0)) { + printf("torture_open_connection failed\n"); + goto fail; + } + status = cli_rpc_pipe_open_noauth(cli, &ndr_table_rpcecho.syntax_id, + &p); + if (!NT_STATUS_IS_OK(status)) { + printf("Could not open echo pipe: %s\n", nt_errstr(status)); + goto fail; + } + + num_reqs = 0; + + req = rpccli_echo_TestSleep_send(ev, ev, p, 15); + if (req == NULL) { + printf("rpccli_echo_TestSleep_send failed\n"); + goto fail; + } + tevent_req_set_callback(req, rpccli_sleep_done, &num_reqs); + num_reqs += 1; + + req = cli_echo_send(ev, ev, cli, 1, data_blob_const("hello", 5)); + if (req == NULL) { + printf("cli_echo_send failed\n"); + goto fail; + } + tevent_req_set_callback(req, cli_echo_done, &num_reqs); + num_reqs += 1; + + memset(buf, 0, sizeof(buf)); + + for (i=0; i<10; i++) { + req = cli_write_andx_send(ev, ev, cli, 4711, 0, buf, 0, sizeof(buf)); + if (req == NULL) { + printf("cli_close_send failed\n"); + goto fail; + } + tevent_req_set_callback(req, cli_close_done, &num_reqs); + num_reqs += 1; + + req = cli_echo_send(ev, ev, cli, 1, data_blob_const("hello", 5)); + if (req == NULL) { + printf("cli_echo_send failed\n"); + goto fail; + } + tevent_req_set_callback(req, cli_echo_done, &num_reqs); + num_reqs += 1; + } + + while (num_reqs > 0) { + if (tevent_loop_once(ev) != 0) { + printf("tevent_loop_once failed\n"); + goto fail; + } + } + + TALLOC_FREE(p); + + ret = true; +fail: + if (cli != NULL) { + torture_close_connection(cli); + } + return ret; +} diff --git a/source3/torture/torture.c b/source3/torture/torture.c index 122c69434d..abec6111bd 100644 --- a/source3/torture/torture.c +++ b/source3/torture/torture.c @@ -7777,6 +7777,7 @@ static struct { {"OPEN", run_opentest, 0}, {"POSIX", run_simple_posix_open_test, 0}, {"POSIX-APPEND", run_posix_append, 0}, + {"ASYNC-ECHO", run_async_echo, 0}, { "UID-REGRESSION-TEST", run_uid_regression_test, 0}, { "SHORTNAME-TEST", run_shortname_test, 0}, #if 1 |