source3/torture/test_async_echo.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    Run the async echo responder
   4    Copyright (C) Volker Lendecke 2010
   5
   6    This program is free software; you can redistribute it and/or modify
   7    it under the terms of the GNU General Public License as published by
   8    the Free Software Foundation; either version 3 of the License, or
   9    (at your option) any later version.
  10
  11    This program is distributed in the hope that it will be useful,
  12    but WITHOUT ANY WARRANTY; without even the implied warranty of
  13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14    GNU General Public License for more details.
  15
  16    You should have received a copy of the GNU General Public License
  17    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18 */
  19
  20 #include "includes.h"
  21 #include "torture/proto.h"
  22 #include "libsmb/libsmb.h"
  23 #include "rpc_client/cli_pipe.h"
  24 #include "librpc/gen_ndr/ndr_echo_c.h"
  25
  26 static void rpccli_sleep_done(struct tevent_req *req)
  27 {
  28         int *done = (int *)tevent_req_callback_data_void(req);
  29         NTSTATUS status;
  30         uint32_t result = UINT32_MAX;
  31
  32         status = dcerpc_echo_TestSleep_recv(req, talloc_tos(), &result);
  33         TALLOC_FREE(req);
  34         printf("sleep returned %s, %d\n", nt_errstr(status), (int)result);
  35         *done -= 1;
  36 }
  37
  38 static void cli_echo_done(struct tevent_req *req)
  39 {
  40         int *done = (int *)tevent_req_callback_data_void(req);
  41         NTSTATUS status;
  42
  43         status = cli_echo_recv(req);
  44         TALLOC_FREE(req);
  45         printf("echo returned %s\n", nt_errstr(status));
  46         *done -= 1;
  47 }
  48
  49 static void write_andx_done(struct tevent_req *req)
  50 {
  51         int *done = (int *)tevent_req_callback_data_void(req);
  52         NTSTATUS status;
  53         size_t written;
  54
  55         status = cli_write_andx_recv(req, &written);
  56         TALLOC_FREE(req);
  57         printf("cli_write_andx returned %s\n", nt_errstr(status));
  58         *done -= 1;
  59 }
  60
  61 bool run_async_echo(int dummy)
  62 {
  63         struct cli_state *cli = NULL;
  64         struct rpc_pipe_client *p;
  65         struct dcerpc_binding_handle *b;
  66         struct tevent_context *ev;
  67         struct tevent_req *req;
  68         NTSTATUS status;
  69         bool ret = false;
  70         int i, num_reqs;
  71         uint8_t buf[65536];
  72
  73         printf("Starting ASYNC_ECHO\n");
  74
  75         ev = samba_tevent_context_init(talloc_tos());
  76         if (ev == NULL) {
  77                 printf("tevent_context_init failed\n");
  78                 goto fail;
  79         }
  80
  81         if (!torture_open_connection(&cli, 0)) {
  82                 printf("torture_open_connection failed\n");
  83                 goto fail;
  84         }
  85         status = cli_rpc_pipe_open_noauth(cli, &ndr_table_rpcecho,
  86                                           &p);
  87         if (!NT_STATUS_IS_OK(status)) {
  88                 printf("Could not open echo pipe: %s\n", nt_errstr(status));
  89                 goto fail;
  90         }
  91         b = p->binding_handle;
  92
  93         num_reqs = 0;
  94
  95         req = dcerpc_echo_TestSleep_send(ev, ev, b, 15);
  96         if (req == NULL) {
  97                 printf("rpccli_echo_TestSleep_send failed\n");
  98                 goto fail;
  99         }
 100         tevent_req_set_callback(req, rpccli_sleep_done, &num_reqs);
 101         num_reqs += 1;
 102
 103         req = cli_echo_send(ev, ev, cli, 1, data_blob_const("hello", 5));
 104         if (req == NULL) {
 105                 printf("cli_echo_send failed\n");
 106                 goto fail;
 107         }
 108         tevent_req_set_callback(req, cli_echo_done, &num_reqs);
 109         num_reqs += 1;
 110
 111         memset(buf, 0, sizeof(buf));
 112
 113         for (i=0; i<10; i++) {
 114                 req = cli_write_andx_send(ev, ev, cli, 4711, 0, buf, 0,
 115                                           sizeof(buf));
 116                 if (req == NULL) {
 117                         printf("cli_write_andx_send failed\n");
 118                         goto fail;
 119                 }
 120                 tevent_req_set_callback(req, write_andx_done, &num_reqs);
 121                 num_reqs += 1;
 122
 123                 req = cli_echo_send(ev, ev, cli, 1,
 124                                     data_blob_const("hello", 5));
 125                 if (req == NULL) {
 126                         printf("cli_echo_send failed\n");
 127                         goto fail;
 128                 }
 129                 tevent_req_set_callback(req, cli_echo_done, &num_reqs);
 130                 num_reqs += 1;
 131         }
 132
 133         while (num_reqs > 0) {
 134                 if (tevent_loop_once(ev) != 0) {
 135                         printf("tevent_loop_once failed\n");
 136                         goto fail;
 137                 }
 138         }
 139
 140         TALLOC_FREE(p);
 141
 142         ret = true;
 143 fail:
 144         if (cli != NULL) {
 145                 torture_close_connection(cli);
 146         }
 147         return ret;
 148 }