source4/torture/rpc/echo.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    test suite for echo rpc operations
   4
   5    Copyright (C) Andrew Tridgell 2003
   6    Copyright (C) Stefan (metze) Metzmacher 2005
   7    Copyright (C) Jelmer Vernooij 2005
   8
   9    This program is free software; you can redistribute it and/or modify
  10    it under the terms of the GNU General Public License as published by
  11    the Free Software Foundation; either version 3 of the License, or
  12    (at your option) any later version.
  13
  14    This program is distributed in the hope that it will be useful,
  15    but WITHOUT ANY WARRANTY; without even the implied warranty of
  16    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17    GNU General Public License for more details.
  18
  19    You should have received a copy of the GNU General Public License
  20    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  21 */
  22
  23 #include "includes.h"
  24 #include "torture/rpc/torture_rpc.h"
  25 #include "lib/events/events.h"
  26 #include "librpc/gen_ndr/ndr_echo_c.h"
  27
  28
  29 /*
  30   test the AddOne interface
  31 */
  32 #define TEST_ADDONE(tctx, value) do { \
  33         n = i = value; \
  34         r.in.in_data = n; \
  35         r.out.out_data = &n; \
  36         torture_assert_ntstatus_ok(tctx, dcerpc_echo_AddOne_r(b, tctx, &r), \
  37                 talloc_asprintf(tctx, "AddOne(%d) failed", i)); \
  38         torture_assert (tctx, n == i+1, talloc_asprintf(tctx, "%d + 1 != %u (should be %u)\n", i, n, i+1)); \
  39         torture_comment (tctx, "%d + 1 = %u\n", i, n); \
  40 } while(0)
  41
  42 static bool test_addone(struct torture_context *tctx,
  43                         struct dcerpc_pipe *p)
  44 {
  45         uint32_t i;
  46         uint32_t n;
  47         struct echo_AddOne r;
  48         struct dcerpc_binding_handle *b = p->binding_handle;
  49
  50         for (i=0;i<10;i++) {
  51                 TEST_ADDONE(tctx, i);
  52         }
  53
  54         TEST_ADDONE(tctx, 0x7FFFFFFE);
  55         TEST_ADDONE(tctx, 0xFFFFFFFE);
  56         TEST_ADDONE(tctx, 0xFFFFFFFF);
  57         TEST_ADDONE(tctx, random() & 0xFFFFFFFF);
  58         return true;
  59 }
  60
  61 /*
  62   test the EchoData interface
  63 */
  64 static bool test_echodata(struct torture_context *tctx,
  65                           struct dcerpc_pipe *p)
  66 {
  67         int i;
  68         uint8_t *data_in, *data_out;
  69         int len;
  70         struct echo_EchoData r;
  71         struct dcerpc_binding_handle *b = p->binding_handle;
  72         const struct dcerpc_binding *binding = dcerpc_binding_handle_get_binding(b);
  73         uint16_t flags = dcerpc_binding_get_flags(binding);
  74
  75         if (torture_setting_bool(tctx, "quick", false) &&
  76             (flags & DCERPC_DEBUG_VALIDATE_BOTH)) {
  77                 len = 1 + (random() % 500);
  78         } else {
  79                 len = 1 + (random() % 5000);
  80         }
  81
  82         data_in = talloc_array(tctx, uint8_t, len);
  83         data_out = talloc_array(tctx, uint8_t, len);
  84         for (i=0;i<len;i++) {
  85                 data_in[i] = i;
  86         }
  87
  88         r.in.len = len;
  89         r.in.in_data = data_in;
  90
  91         torture_assert_ntstatus_ok(tctx, dcerpc_echo_EchoData_r(b, tctx, &r),
  92                 talloc_asprintf(tctx, "EchoData(%d) failed\n", len));
  93
  94         data_out = r.out.out_data;
  95
  96         for (i=0;i<len;i++) {
  97                 if (data_in[i] != data_out[i]) {
  98                         torture_comment(tctx, "Bad data returned for len %d at offset %d\n",
  99                                len, i);
 100                         torture_comment(tctx, "in:\n");
 101                         dump_data(0, data_in+i, MIN(len-i, 16));
 102                         torture_comment(tctx, "out:\n");
 103                         dump_data(0, data_out+i, MIN(len-1, 16));
 104                         return false;
 105                 }
 106         }
 107         return true;
 108 }
 109
 110 /*
 111   test the SourceData interface
 112 */
 113 static bool test_sourcedata(struct torture_context *tctx,
 114                             struct dcerpc_pipe *p)
 115 {
 116         int i;
 117         int len;
 118         struct echo_SourceData r;
 119         struct dcerpc_binding_handle *b = p->binding_handle;
 120         const struct dcerpc_binding *binding = dcerpc_binding_handle_get_binding(b);
 121         uint16_t flags = dcerpc_binding_get_flags(binding);
 122
 123         if (torture_setting_bool(tctx, "quick", false) &&
 124             (flags & DCERPC_DEBUG_VALIDATE_BOTH)) {
 125                 len = 100 + (random() % 500);
 126         } else {
 127                 len = 200000 + (random() % 5000);
 128         }
 129
 130         r.in.len = len;
 131
 132         torture_assert_ntstatus_ok(tctx, dcerpc_echo_SourceData_r(b, tctx, &r),
 133                 talloc_asprintf(tctx, "SourceData(%d) failed", len));
 134
 135         for (i=0;i<len;i++) {
 136                 uint8_t *v = (uint8_t *)r.out.data;
 137                 torture_assert(tctx, v[i] == (i & 0xFF),
 138                         talloc_asprintf(tctx,
 139                                                 "bad data 0x%x at %d\n", (uint8_t)r.out.data[i], i));
 140         }
 141         return true;
 142 }
 143
 144 /*
 145   test the SinkData interface
 146 */
 147 static bool test_sinkdata(struct torture_context *tctx,
 148                           struct dcerpc_pipe *p)
 149 {
 150         int i;
 151         uint8_t *data_in;
 152         int len;
 153         struct echo_SinkData r;
 154         struct dcerpc_binding_handle *b = p->binding_handle;
 155         const struct dcerpc_binding *binding = dcerpc_binding_handle_get_binding(b);
 156         uint16_t flags = dcerpc_binding_get_flags(binding);
 157
 158         if (torture_setting_bool(tctx, "quick", false) &&
 159             (flags & DCERPC_DEBUG_VALIDATE_BOTH)) {
 160                 len = 100 + (random() % 5000);
 161         } else {
 162                 len = 200000 + (random() % 5000);
 163         }
 164
 165         data_in = talloc_array(tctx, uint8_t, len);
 166         for (i=0;i<len;i++) {
 167                 data_in[i] = i+1;
 168         }
 169
 170         r.in.len = len;
 171         r.in.data = data_in;
 172
 173         torture_assert_ntstatus_ok(tctx, dcerpc_echo_SinkData_r(b, tctx, &r),
 174                 talloc_asprintf(tctx, "SinkData(%d) failed", len));
 175
 176         torture_comment(tctx, "sunk %d bytes\n", len);
 177         return true;
 178 }
 179
 180
 181 /*
 182   test the testcall interface
 183 */
 184 static bool test_testcall(struct torture_context *tctx,
 185                           struct dcerpc_pipe *p)
 186 {
 187         struct echo_TestCall r;
 188         const char *s = NULL;
 189         struct dcerpc_binding_handle *b = p->binding_handle;
 190
 191         r.in.s1 = "input string";
 192         r.out.s2 = &s;
 193
 194         torture_assert_ntstatus_ok(tctx, dcerpc_echo_TestCall_r(b, tctx, &r),
 195                 "TestCall failed");
 196
 197         torture_assert_str_equal(tctx, s, "input string", "Didn't receive back same string");
 198
 199         return true;
 200 }
 201
 202 /*
 203   test the testcall interface
 204 */
 205 static bool test_testcall2(struct torture_context *tctx,
 206                            struct dcerpc_pipe *p)
 207 {
 208         struct echo_TestCall2 r;
 209         int i;
 210         struct dcerpc_binding_handle *b = p->binding_handle;
 211
 212         for (i=1;i<=7;i++) {
 213                 r.in.level = i;
 214                 r.out.info = talloc(tctx, union echo_Info);
 215
 216                 torture_comment(tctx, "Testing TestCall2 level %d\n", i);
 217                 torture_assert_ntstatus_ok(tctx, dcerpc_echo_TestCall2_r(b, tctx, &r),
 218                         "TestCall2 failed");
 219                 torture_assert_ntstatus_ok(tctx, r.out.result, "TestCall2 failed");
 220         }
 221         return true;
 222 }
 223
 224 static void test_sleep_done(struct tevent_req *subreq)
 225 {
 226         bool *done1 = (bool *)tevent_req_callback_data_void(subreq);
 227         *done1 = true;
 228 }
 229
 230 /*
 231   test the TestSleep interface
 232 */
 233 static bool test_sleep(struct torture_context *tctx,
 234                        struct dcerpc_pipe *p)
 235 {
 236         int i;
 237 #define ASYNC_COUNT 3
 238         struct tevent_req *req[ASYNC_COUNT];
 239         struct echo_TestSleep r[ASYNC_COUNT];
 240         bool done1[ASYNC_COUNT];
 241         bool done2[ASYNC_COUNT];
 242         struct timeval snd[ASYNC_COUNT];
 243         struct timeval rcv[ASYNC_COUNT];
 244         struct timeval diff[ASYNC_COUNT];
 245         int total_done = 0;
 246         struct dcerpc_binding_handle *b = p->binding_handle;
 247         const struct dcerpc_binding *bd = dcerpc_binding_handle_get_binding(b);
 248         enum dcerpc_transport_t transport;
 249         uint32_t assoc_group_id;
 250         struct dcerpc_pipe *p2 = NULL;
 251         NTSTATUS status;
 252
 253         if (torture_setting_bool(tctx, "quick", false)) {
 254                 torture_skip(tctx, "TestSleep disabled - use \"torture:quick=no\" to enable\n");
 255         }
 256         torture_comment(tctx, "Testing TestSleep - use \"torture:quick=yes\" to disable\n");
 257
 258         transport       = dcerpc_binding_get_transport(bd);
 259         assoc_group_id  = dcerpc_binding_get_assoc_group_id(bd);
 260
 261         torture_comment(tctx, "connect echo connection 2 with "
 262                         "DCERPC_CONCURRENT_MULTIPLEX\n");
 263         status = torture_rpc_connection_transport(tctx, &p2,
 264                                                   &ndr_table_rpcecho,
 265                                                   transport,
 266                                                   assoc_group_id,
 267                                                   DCERPC_CONCURRENT_MULTIPLEX);
 268         torture_assert_ntstatus_ok(tctx, status, "opening echo connection 2");
 269         b = p2->binding_handle;
 270
 271         for (i=0;i<ASYNC_COUNT;i++) {
 272                 done1[i]        = false;
 273                 done2[i]        = false;
 274                 snd[i]          = timeval_current();
 275                 rcv[i]          = timeval_zero();
 276                 r[i].in.seconds = ASYNC_COUNT-i;
 277                 req[i] = dcerpc_echo_TestSleep_r_send(tctx, tctx->ev, b, &r[i]);
 278                 torture_assert(tctx, req[i], "Failed to send async sleep request\n");
 279                 tevent_req_set_callback(req[i], test_sleep_done, &done1[i]);
 280         }
 281
 282         while (total_done < ASYNC_COUNT) {
 283                 torture_assert(tctx, tevent_loop_once(tctx->ev) == 0,
 284                                            "Event context loop failed");
 285                 for (i=0;i<ASYNC_COUNT;i++) {
 286                         if (done2[i] == false && done1[i] == true) {
 287                                 int rounded_tdiff;
 288                                 total_done++;
 289                                 done2[i] = true;
 290                                 rcv[i]  = timeval_current();
 291                                 diff[i] = tevent_timeval_until(
 292                                         &snd[i], &rcv[i]);
 293                                 rounded_tdiff = (int)(0.5 + diff[i].tv_sec + (1.0e-6*diff[i].tv_usec));
 294                                 torture_comment(tctx, "rounded_tdiff=%d\n", rounded_tdiff);
 295                                 torture_assert_ntstatus_ok(tctx,
 296                                         dcerpc_echo_TestSleep_r_recv(req[i], tctx),
 297                                         talloc_asprintf(tctx, "TestSleep(%d) failed", i));
 298                                 torture_assert(tctx, r[i].out.result == r[i].in.seconds,
 299                                         talloc_asprintf(tctx, "Failed - Asked to sleep for %u seconds (server replied with %u seconds and the reply takes only %u seconds)",
 300                                                 r[i].out.result, r[i].in.seconds, (unsigned int)diff[i].tv_sec));
 301                                 torture_assert(tctx, r[i].out.result <= rounded_tdiff,
 302                                         talloc_asprintf(tctx, "Failed - Slept for %u seconds (but reply takes only %u.%06u seconds)",
 303                                                 r[i].out.result, (unsigned int)diff[i].tv_sec, (unsigned int)diff[i].tv_usec));
 304                                 if (r[i].out.result+1 == rounded_tdiff) {
 305                                         torture_comment(tctx, "Slept for %u seconds (but reply takes %u.%06u seconds - busy server?)\n",
 306                                                         r[i].out.result, (unsigned int)diff[i].tv_sec, (unsigned int)diff[i].tv_usec);
 307                                 } else if (r[i].out.result == rounded_tdiff) {
 308                                         torture_comment(tctx, "Slept for %u seconds (reply takes %u.%06u seconds - ok)\n",
 309                                                         r[i].out.result, (unsigned int)diff[i].tv_sec, (unsigned int)diff[i].tv_usec);
 310                                 } else {
 311                                         torture_fail(tctx, talloc_asprintf(tctx,
 312                                                      "(Failed) - Not async - Slept for %u seconds (but reply takes %u.%06u seconds)\n",
 313                                                      r[i].out.result, (unsigned int)diff[i].tv_sec, (unsigned int)diff[i].tv_usec));
 314                                 }
 315                         }
 316                 }
 317         }
 318         torture_comment(tctx, "\n");
 319         return true;
 320 }
 321
 322 /*
 323   test enum handling
 324 */
 325 static bool test_enum(struct torture_context *tctx,
 326                                                   struct dcerpc_pipe *p)
 327 {
 328         struct echo_TestEnum r;
 329         enum echo_Enum1 v = ECHO_ENUM1;
 330         struct echo_Enum2 e2;
 331         union echo_Enum3 e3;
 332         struct dcerpc_binding_handle *b = p->binding_handle;
 333
 334         r.in.foo1 = &v;
 335         r.in.foo2 = &e2;
 336         r.in.foo3 = &e3;
 337         r.out.foo1 = &v;
 338         r.out.foo2 = &e2;
 339         r.out.foo3 = &e3;
 340
 341         e2.e1 = 76;
 342         e2.e2 = ECHO_ENUM1_32;
 343         e3.e1 = ECHO_ENUM2;
 344
 345         torture_assert_ntstatus_ok(tctx, dcerpc_echo_TestEnum_r(b, tctx, &r),
 346                 "TestEnum failed");
 347         return true;
 348 }
 349
 350 /*
 351   test surrounding conformant array handling
 352 */
 353 static bool test_surrounding(struct torture_context *tctx,
 354                                                   struct dcerpc_pipe *p)
 355 {
 356         struct echo_TestSurrounding r;
 357         struct dcerpc_binding_handle *b = p->binding_handle;
 358
 359         ZERO_STRUCT(r);
 360         r.in.data = talloc(tctx, struct echo_Surrounding);
 361
 362         r.in.data->x = 20;
 363         r.in.data->surrounding = talloc_zero_array(tctx, uint16_t, r.in.data->x);
 364
 365         r.out.data = talloc(tctx, struct echo_Surrounding);
 366
 367         torture_assert_ntstatus_ok(tctx, dcerpc_echo_TestSurrounding_r(b, tctx, &r),
 368                 "TestSurrounding failed");
 369
 370         torture_assert(tctx, r.out.data->x == 2 * r.in.data->x,
 371                 "TestSurrounding did not make the array twice as large");
 372
 373         return true;
 374 }
 375
 376 /*
 377   test multiple levels of pointers
 378 */
 379 static bool test_doublepointer(struct torture_context *tctx,
 380                                struct dcerpc_pipe *p)
 381 {
 382         struct echo_TestDoublePointer r;
 383         uint16_t value = 12;
 384         uint16_t *pvalue = &value;
 385         uint16_t **ppvalue = &pvalue;
 386         struct dcerpc_binding_handle *b = p->binding_handle;
 387
 388         ZERO_STRUCT(r);
 389         r.in.data = &ppvalue;
 390
 391         torture_assert_ntstatus_ok(tctx, dcerpc_echo_TestDoublePointer_r(b, tctx, &r),
 392                 "TestDoublePointer failed");
 393
 394         torture_assert_int_equal(tctx, value, r.out.result,
 395                                         "TestDoublePointer did not return original value");
 396         return true;
 397 }
 398
 399
 400 /*
 401   test request timeouts
 402 */
 403 #if 0 /* this test needs fixing to work over ncacn_np */
 404 static bool test_timeout(struct torture_context *tctx,
 405                                                  struct dcerpc_pipe *p)
 406 {
 407         NTSTATUS status;
 408         struct rpc_request *req;
 409         struct echo_TestSleep r;
 410         int timeout_saved = p->request_timeout;
 411
 412         if (torture_setting_bool(tctx, "quick", false)) {
 413                 torture_skip(tctx, "timeout testing disabled - use \"torture:quick=no\" to enable\n");
 414         }
 415
 416         torture_comment(tctx, "testing request timeouts\n");
 417         r.in.seconds = 2;
 418         p->request_timeout = 1;
 419
 420         req = dcerpc_echo_TestSleep_send(p, tctx, &r);
 421         if (!req) {
 422                 torture_comment(tctx, "Failed to send async sleep request\n");
 423                 goto failed;
 424         }
 425         req->ignore_timeout = true;
 426
 427         status  = dcerpc_echo_TestSleep_recv(req);
 428         torture_assert_ntstatus_equal(tctx, status, NT_STATUS_IO_TIMEOUT,
 429                                                                   "request should have timed out");
 430
 431         torture_comment(tctx, "testing request destruction\n");
 432         req = dcerpc_echo_TestSleep_send(p, tctx, &r);
 433         if (!req) {
 434                 torture_comment(tctx, "Failed to send async sleep request\n");
 435                 goto failed;
 436         }
 437         talloc_free(req);
 438
 439         req = dcerpc_echo_TestSleep_send(p, tctx, &r);
 440         if (!req) {
 441                 torture_comment(tctx, "Failed to send async sleep request\n");
 442                 goto failed;
 443         }
 444         req->ignore_timeout = true;
 445         status  = dcerpc_echo_TestSleep_recv(req);
 446         torture_assert_ntstatus_equal(tctx, status, NT_STATUS_IO_TIMEOUT,
 447                 "request should have timed out");
 448
 449         p->request_timeout = timeout_saved;
 450
 451         return test_addone(tctx, p);
 452
 453 failed:
 454         p->request_timeout = timeout_saved;
 455         return false;
 456 }
 457 #endif
 458
 459 struct torture_suite *torture_rpc_echo(TALLOC_CTX *mem_ctx)
 460 {
 461         struct torture_suite *suite = torture_suite_create(mem_ctx, "echo");
 462         struct torture_rpc_tcase *tcase;
 463
 464         tcase = torture_suite_add_rpc_iface_tcase(suite, "echo",
 465                                                   &ndr_table_rpcecho);
 466
 467         torture_rpc_tcase_add_test(tcase, "addone", test_addone);
 468         torture_rpc_tcase_add_test(tcase, "sinkdata", test_sinkdata);
 469         torture_rpc_tcase_add_test(tcase, "echodata", test_echodata);
 470         torture_rpc_tcase_add_test(tcase, "sourcedata", test_sourcedata);
 471         torture_rpc_tcase_add_test(tcase, "testcall", test_testcall);
 472         torture_rpc_tcase_add_test(tcase, "testcall2", test_testcall2);
 473         torture_rpc_tcase_add_test(tcase, "enum", test_enum);
 474         torture_rpc_tcase_add_test(tcase, "surrounding", test_surrounding);
 475         torture_rpc_tcase_add_test(tcase, "doublepointer", test_doublepointer);
 476         torture_rpc_tcase_add_test(tcase, "sleep", test_sleep);
 477 #if 0 /* this test needs fixing to work over ncacn_np */
 478         torture_rpc_tcase_add_test(tcase, "timeout", test_timeout);
 479 #endif
 480
 481         return suite;
 482 }