s4-torture: handle NT_STATUS_CONNECTION_RESET

[sfrench/samba-autobuild/.git] / source4 / torture / raw / openbench.c

diff --git a/source4/torture/raw/openbench.c b/source4/torture/raw/openbench.c
index 2716f6a528507d0585d6ac913bd563e038e4663a..837b7c345a620f103e36ffe43ce274244edfd044 100644 (file)
--- 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--;