diff options
Diffstat (limited to 'source4/torture/raw/lockbench.c')
| -rw-r--r-- | source4/torture/raw/lockbench.c | 6 |
1 files changed, 4 insertions, 2 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--; |