diff options
-rw-r--r-- | source4/torture/rpc/echo.c | 62 |
1 files changed, 62 insertions, 0 deletions
diff --git a/source4/torture/rpc/echo.c b/source4/torture/rpc/echo.c index ad09e00294..94400247a6 100644 --- a/source4/torture/rpc/echo.c +++ b/source4/torture/rpc/echo.c @@ -383,6 +383,67 @@ static BOOL test_doublepointer(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx) return ret; } + +/* + test request timeouts +*/ +static BOOL test_timeout(struct dcerpc_pipe *p, TALLOC_CTX *mem_ctx) +{ + NTSTATUS status; + struct rpc_request *req; + struct echo_TestSleep r; + int timeout_saved = p->request_timeout; + + if (!lp_parm_bool(-1, "torture", "echo_TestSleep", True)) { + printf("timeout testing disabled - use \"torture:echo_TestSleep=yes\" to enable\n"); + return True; + } + + printf("testing request timeouts\n"); + r.in.seconds = 2; + p->request_timeout = 1; + + req = dcerpc_echo_TestSleep_send(p, mem_ctx, &r); + if (!req) { + printf("Failed to send async sleep request\n"); + goto failed; + } + + status = dcerpc_ndr_request_recv(req); + if (!NT_STATUS_EQUAL(status, NT_STATUS_IO_TIMEOUT)) { + printf("request should have timed out - %s\n", nt_errstr(status)); + goto failed; + } + + printf("testing request destruction\n"); + req = dcerpc_echo_TestSleep_send(p, mem_ctx, &r); + if (!req) { + printf("Failed to send async sleep request\n"); + goto failed; + } + talloc_free(req); + + req = dcerpc_echo_TestSleep_send(p, mem_ctx, &r); + if (!req) { + printf("Failed to send async sleep request\n"); + goto failed; + } + status = dcerpc_ndr_request_recv(req); + if (!NT_STATUS_EQUAL(status, NT_STATUS_IO_TIMEOUT)) { + printf("request should have timed out - %s\n", nt_errstr(status)); + goto failed; + } + + + p->request_timeout = timeout_saved; + return True; + +failed: + p->request_timeout = timeout_saved; + return False; +} + + BOOL torture_rpc_echo(void) { NTSTATUS status; @@ -411,6 +472,7 @@ BOOL torture_rpc_echo(void) ret &= test_surrounding(p, mem_ctx); ret &= test_doublepointer(p, mem_ctx); ret &= test_sleep(p, mem_ctx); + ret &= test_timeout(p, mem_ctx); printf("\n"); |