diff options
Diffstat (limited to 'source4/torture/raw/openbench.c')
-rw-r--r-- | source4/torture/raw/openbench.c | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/source4/torture/raw/openbench.c b/source4/torture/raw/openbench.c index 2716f6a528..837b7c345a 100644 --- a/source4/torture/raw/openbench.c +++ b/source4/torture/raw/openbench.c @@ -222,7 +222,8 @@ static void open_completed(struct smbcli_request *req) state->req_open = NULL; if (NT_STATUS_EQUAL(status, NT_STATUS_END_OF_FILE) || - NT_STATUS_EQUAL(status, NT_STATUS_LOCAL_DISCONNECT)) { + NT_STATUS_EQUAL(status, NT_STATUS_LOCAL_DISCONNECT) || + NT_STATUS_EQUAL(status, NT_STATUS_CONNECTION_RESET)) { talloc_free(state->tree); talloc_free(state->cli); state->tree = NULL; @@ -281,7 +282,8 @@ static void close_completed(struct smbcli_request *req) state->req_close = NULL; if (NT_STATUS_EQUAL(status, NT_STATUS_END_OF_FILE) || - NT_STATUS_EQUAL(status, NT_STATUS_LOCAL_DISCONNECT)) { + NT_STATUS_EQUAL(status, NT_STATUS_LOCAL_DISCONNECT) || + NT_STATUS_EQUAL(status, NT_STATUS_CONNECTION_RESET)) { talloc_free(state->tree); talloc_free(state->cli); state->tree = NULL; @@ -315,7 +317,8 @@ static void echo_completion(struct smbcli_request *req) struct benchopen_state *state = (struct benchopen_state *)req->async.private_data; NTSTATUS status = smbcli_request_simple_recv(req); if (NT_STATUS_EQUAL(status, NT_STATUS_END_OF_FILE) || - NT_STATUS_EQUAL(status, NT_STATUS_LOCAL_DISCONNECT)) { + NT_STATUS_EQUAL(status, NT_STATUS_LOCAL_DISCONNECT) || + NT_STATUS_EQUAL(status, NT_STATUS_CONNECTION_RESET)) { talloc_free(state->tree); state->tree = NULL; num_connected--; |