summaryrefslogtreecommitdiff
path: root/source4
diff options
context:
space:
mode:
Diffstat (limited to 'source4')
-rw-r--r--source4/torture/raw/lockbench.c6
-rw-r--r--source4/torture/raw/openbench.c3
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--;