/* test the SinkData interface */ static bool test_sinkdata(struct torture_context *tctx, struct dcerpc_pipe *p) { int i; NTSTATUS status; uint8_t *data_in; int len; struct echo_SinkData r; if (torture_setting_bool(tctx, "quick", false) && (p->conn->flags & DCERPC_DEBUG_VALIDATE_BOTH)) { len = 100 + (random() % 5000); } else { len = 200000 + (random() % 5000); } data_in = talloc_array(tctx, uint8_t, len); for (i=0;i<len;i++) { data_in[i] = i+1; } r.in.len = len; r.in.data = data_in; status = dcerpc_echo_SinkData(p, tctx, &r); torture_assert_ntstatus_ok(tctx, status, talloc_asprintf(tctx, "SinkData(%d) failed", len)); torture_comment(tctx, "sunk %d bytes\n", len); return true; }
static NTSTATUS cmd_echo_sink_data(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx, int argc, const char **argv) { struct dcerpc_binding_handle *b = cli->binding_handle; uint32 size, i; NTSTATUS status; uint8_t *in_data = NULL; if (argc != 2) { printf("Usage: %s num\n", argv[0]); return NT_STATUS_OK; } size = atoi(argv[1]); if ( (in_data = (uint8_t*)SMB_MALLOC(size)) == NULL ) { printf("Failure to allocate buff of %d bytes\n", size); status = NT_STATUS_NO_MEMORY; goto done; } for (i = 0; i < size; i++) { in_data[i] = i & 0xff; } status = dcerpc_echo_SinkData(b, mem_ctx, size, in_data); if (!NT_STATUS_IS_OK(status)) { goto done; } done: SAFE_FREE(in_data); return status; }