35
34
r.in.in_data = n; \
36
35
r.out.out_data = &n; \
37
status = dcerpc_echo_AddOne(p, tctx, &r); \
38
torture_assert_ntstatus_ok(tctx, status, talloc_asprintf(tctx, "AddOne(%d) failed", i)); \
36
torture_assert_ntstatus_ok(tctx, dcerpc_echo_AddOne_r(b, tctx, &r), \
37
talloc_asprintf(tctx, "AddOne(%d) failed", i)); \
39
38
torture_assert (tctx, n == i+1, talloc_asprintf(tctx, "%d + 1 != %u (should be %u)\n", i, n, i+1)); \
40
39
torture_comment (tctx, "%d + 1 = %u\n", i, n); \
43
42
static bool test_addone(struct torture_context *tctx,
44
struct dcerpc_pipe *p)
43
struct dcerpc_pipe *p)
49
47
struct echo_AddOne r;
48
struct dcerpc_binding_handle *b = p->binding_handle;
51
50
for (i=0;i<10;i++) {
52
51
TEST_ADDONE(tctx, i);
88
87
r.in.in_data = data_in;
90
status = dcerpc_echo_EchoData(p, tctx, &r);
91
torture_assert_ntstatus_ok(tctx, status, talloc_asprintf(tctx,
92
"EchoData(%d) failed\n", len));
89
torture_assert_ntstatus_ok(tctx, dcerpc_echo_EchoData_r(b, tctx, &r),
90
talloc_asprintf(tctx, "EchoData(%d) failed\n", len));
94
92
data_out = r.out.out_data;
130
status = dcerpc_echo_SourceData(p, tctx, &r);
131
torture_assert_ntstatus_ok(tctx, status, talloc_asprintf(tctx,
132
"SourceData(%d) failed", len));
128
torture_assert_ntstatus_ok(tctx, dcerpc_echo_SourceData_r(b, tctx, &r),
129
talloc_asprintf(tctx, "SourceData(%d) failed", len));
134
131
for (i=0;i<len;i++) {
135
132
uint8_t *v = (uint8_t *)r.out.data;
168
165
r.in.data = data_in;
170
status = dcerpc_echo_SinkData(p, tctx, &r);
171
torture_assert_ntstatus_ok(tctx, status, talloc_asprintf(tctx,
172
"SinkData(%d) failed",
167
torture_assert_ntstatus_ok(tctx, dcerpc_echo_SinkData_r(b, tctx, &r),
168
talloc_asprintf(tctx, "SinkData(%d) failed", len));
175
170
torture_comment(tctx, "sunk %d bytes\n", len);
181
176
test the testcall interface
183
178
static bool test_testcall(struct torture_context *tctx,
184
struct dcerpc_pipe *p)
179
struct dcerpc_pipe *p)
187
181
struct echo_TestCall r;
188
182
const char *s = NULL;
183
struct dcerpc_binding_handle *b = p->binding_handle;
190
185
r.in.s1 = "input string";
193
status = dcerpc_echo_TestCall(p, tctx, &r);
194
torture_assert_ntstatus_ok(tctx, status, "TestCall failed");
188
torture_assert_ntstatus_ok(tctx, dcerpc_echo_TestCall_r(b, tctx, &r),
196
191
torture_assert_str_equal(tctx, s, "input string", "Didn't receive back same string");
202
197
test the testcall interface
204
199
static bool test_testcall2(struct torture_context *tctx,
205
struct dcerpc_pipe *p)
200
struct dcerpc_pipe *p)
208
202
struct echo_TestCall2 r;
204
struct dcerpc_binding_handle *b = p->binding_handle;
211
206
for (i=1;i<=7;i++) {
213
208
r.out.info = talloc(tctx, union echo_Info);
215
210
torture_comment(tctx, "Testing TestCall2 level %d\n", i);
216
status = dcerpc_echo_TestCall2(p, tctx, &r);
217
torture_assert_ntstatus_ok(tctx, status, "TestCall2 failed");
211
torture_assert_ntstatus_ok(tctx, dcerpc_echo_TestCall2_r(b, tctx, &r),
213
torture_assert_ntstatus_ok(tctx, r.out.result, "TestCall2 failed");
218
static void test_sleep_done(struct tevent_req *subreq)
220
bool *done1 = (bool *)tevent_req_callback_data_void(subreq);
223
225
test the TestSleep interface
225
227
static bool test_sleep(struct torture_context *tctx,
226
struct dcerpc_pipe *p)
228
struct dcerpc_pipe *p)
230
231
#define ASYNC_COUNT 3
231
struct rpc_request *req[ASYNC_COUNT];
232
struct tevent_req *req[ASYNC_COUNT];
232
233
struct echo_TestSleep r[ASYNC_COUNT];
233
bool done[ASYNC_COUNT];
234
bool done1[ASYNC_COUNT];
235
bool done2[ASYNC_COUNT];
234
236
struct timeval snd[ASYNC_COUNT];
235
237
struct timeval rcv[ASYNC_COUNT];
236
238
struct timeval diff[ASYNC_COUNT];
237
239
struct tevent_context *ctx;
238
240
int total_done = 0;
241
struct dcerpc_binding_handle *b = p->binding_handle;
240
243
if (torture_setting_bool(tctx, "quick", false)) {
241
244
torture_skip(tctx, "TestSleep disabled - use \"torture:quick=no\" to enable\n");
243
246
torture_comment(tctx, "Testing TestSleep - use \"torture:quick=yes\" to disable\n");
245
248
for (i=0;i<ASYNC_COUNT;i++) {
247
251
snd[i] = timeval_current();
248
252
rcv[i] = timeval_zero();
249
253
r[i].in.seconds = ASYNC_COUNT-i;
250
req[i] = dcerpc_echo_TestSleep_send(p, tctx, &r[i]);
254
req[i] = dcerpc_echo_TestSleep_r_send(tctx, tctx->ev, b, &r[i]);
251
255
torture_assert(tctx, req[i], "Failed to send async sleep request\n");
256
tevent_req_set_callback(req[i], test_sleep_done, &done1[i]);
254
259
ctx = dcerpc_event_context(p);
256
261
torture_assert(tctx, event_loop_once(ctx) == 0,
257
262
"Event context loop failed");
258
263
for (i=0;i<ASYNC_COUNT;i++) {
259
if (done[i] == false && req[i]->state == RPC_REQUEST_DONE) {
264
if (done2[i] == false && done1[i] == true) {
260
265
int rounded_tdiff;
263
268
rcv[i] = timeval_current();
264
269
diff[i] = timeval_until(&snd[i], &rcv[i]);
265
270
rounded_tdiff = (int)(0.5 + diff[i].tv_sec + (1.0e-6*diff[i].tv_usec));
266
status = dcerpc_ndr_request_recv(req[i]);
267
271
torture_comment(tctx, "rounded_tdiff=%d\n", rounded_tdiff);
268
torture_assert_ntstatus_ok(tctx, status,
269
talloc_asprintf(tctx, "TestSleep(%d) failed", i));
272
torture_assert_ntstatus_ok(tctx,
273
dcerpc_echo_TestSleep_r_recv(req[i], tctx),
274
talloc_asprintf(tctx, "TestSleep(%d) failed", i));
270
275
torture_assert(tctx, r[i].out.result == r[i].in.seconds,
271
276
talloc_asprintf(tctx, "Failed - Asked to sleep for %u seconds (server replied with %u seconds and the reply takes only %u seconds)",
272
r[i].out.result, r[i].in.seconds, (uint_t)diff[i].tv_sec));
277
r[i].out.result, r[i].in.seconds, (unsigned int)diff[i].tv_sec));
273
278
torture_assert(tctx, r[i].out.result <= rounded_tdiff,
274
279
talloc_asprintf(tctx, "Failed - Slept for %u seconds (but reply takes only %u.%06u seconds)",
275
r[i].out.result, (uint_t)diff[i].tv_sec, (uint_t)diff[i].tv_usec));
280
r[i].out.result, (unsigned int)diff[i].tv_sec, (unsigned int)diff[i].tv_usec));
276
281
if (r[i].out.result+1 == rounded_tdiff) {
277
282
torture_comment(tctx, "Slept for %u seconds (but reply takes %u.%06u seconds - busy server?)\n",
278
r[i].out.result, (uint_t)diff[i].tv_sec, (uint_t)diff[i].tv_usec);
283
r[i].out.result, (unsigned int)diff[i].tv_sec, (unsigned int)diff[i].tv_usec);
279
284
} else if (r[i].out.result == rounded_tdiff) {
280
285
torture_comment(tctx, "Slept for %u seconds (reply takes %u.%06u seconds - ok)\n",
281
r[i].out.result, (uint_t)diff[i].tv_sec, (uint_t)diff[i].tv_usec);
286
r[i].out.result, (unsigned int)diff[i].tv_sec, (unsigned int)diff[i].tv_usec);
283
torture_comment(tctx, "(Failed) - Not async - Slept for %u seconds (but reply takes %u.%06u seconds)",
284
r[i].out.result, (uint_t)diff[i].tv_sec, (uint_t)diff[i].tv_usec);
285
/* TODO: let the test fail here, when we support async rpc on ncacn_np */
288
torture_fail(tctx, talloc_asprintf(tctx,
289
"(Failed) - Not async - Slept for %u seconds (but reply takes %u.%06u seconds)\n",
290
r[i].out.result, (unsigned int)diff[i].tv_sec, (unsigned int)diff[i].tv_usec));
337
342
r.out.data = talloc(tctx, struct echo_Surrounding);
339
status = dcerpc_echo_TestSurrounding(p, tctx, &r);
340
torture_assert_ntstatus_ok(tctx, status, "TestSurrounding failed");
344
torture_assert_ntstatus_ok(tctx, dcerpc_echo_TestSurrounding_r(b, tctx, &r),
345
"TestSurrounding failed");
342
347
torture_assert(tctx, r.out.data->x == 2 * r.in.data->x,
343
348
"TestSurrounding did not make the array twice as large");
349
354
test multiple levels of pointers
351
356
static bool test_doublepointer(struct torture_context *tctx,
352
struct dcerpc_pipe *p)
357
struct dcerpc_pipe *p)
355
359
struct echo_TestDoublePointer r;
356
360
uint16_t value = 12;
357
361
uint16_t *pvalue = &value;
358
362
uint16_t **ppvalue = &pvalue;
363
struct dcerpc_binding_handle *b = p->binding_handle;
361
366
r.in.data = &ppvalue;
363
status = dcerpc_echo_TestDoublePointer(p, tctx, &r);
364
torture_assert_ntstatus_ok(tctx, status, "TestDoublePointer failed");
368
torture_assert_ntstatus_ok(tctx, dcerpc_echo_TestDoublePointer_r(b, tctx, &r),
369
"TestDoublePointer failed");
366
371
torture_assert_int_equal(tctx, value, r.out.result,
367
372
"TestDoublePointer did not return original value");