diff options
| -rw-r--r-- | source4/torture/raw/lockbench.c | 6 | ||||
| -rw-r--r-- | source4/torture/raw/openbench.c | 3 |
2 files changed, 6 insertions, 3 deletions
diff --git a/source4/torture/raw/lockbench.c b/source4/torture/raw/lockbench.c index 361f509c40..e929218ead 100644 --- a/source4/torture/raw/lockbench.c +++ b/source4/torture/raw/lockbench.c @@ -212,7 +212,8 @@ static void lock_completion(struct smbcli_request *req) NTSTATUS status = smbcli_request_simple_recv(req); state->req = NULL; if (!NT_STATUS_IS_OK(status)) { - if (NT_STATUS_EQUAL(status, NT_STATUS_END_OF_FILE)) { + if (NT_STATUS_EQUAL(status, NT_STATUS_END_OF_FILE) || + NT_STATUS_EQUAL(status, NT_STATUS_LOCAL_DISCONNECT)) { talloc_free(state->tree); state->tree = NULL; num_connected--; @@ -249,7 +250,8 @@ static void echo_completion(struct smbcli_request *req) struct benchlock_state *state = talloc_get_type(req->async.private, struct benchlock_state); NTSTATUS status = smbcli_request_simple_recv(req); - if (NT_STATUS_EQUAL(status, NT_STATUS_END_OF_FILE)) { + if (NT_STATUS_EQUAL(status, NT_STATUS_END_OF_FILE) || + NT_STATUS_EQUAL(status, NT_STATUS_LOCAL_DISCONNECT)) { talloc_free(state->tree); state->tree = NULL; num_connected--; diff --git a/source4/torture/raw/openbench.c b/source4/torture/raw/openbench.c index 86b97fb9c8..fe01f1f5e2 100644 --- a/source4/torture/raw/openbench.c +++ b/source4/torture/raw/openbench.c @@ -283,7 +283,8 @@ static void echo_completion(struct smbcli_request *req) struct benchopen_state *state = talloc_get_type(req->async.private, struct benchopen_state); NTSTATUS status = smbcli_request_simple_recv(req); - if (NT_STATUS_EQUAL(status, NT_STATUS_END_OF_FILE)) { + if (NT_STATUS_EQUAL(status, NT_STATUS_END_OF_FILE) || + NT_STATUS_EQUAL(status, NT_STATUS_LOCAL_DISCONNECT)) { talloc_free(state->tree); state->tree = NULL; num_connected--; |