From 9c95eca0f4116d52c8522a2066783e4722b21337 Mon Sep 17 00:00:00 2001
From: Ralph Boehme <slow@samba.org>
Date: Wed, 11 Jan 2017 15:36:38 +0100
Subject: [PATCH] s3/smbd: add "smbd:find async delay usec" to SMB2 FIND

This is just a hack for selftest that will be used in subsequent commits
for torturing compound find requests.

Signed-off-by: Ralph Boehme <slow@samba.org>
Reviewed-by: Stefan Metzmacher <metze@samba.org>
---
 source3/smbd/smb2_query_directory.c | 66 +++++++++++++++++++++++++++++++++++++
 1 file changed, 66 insertions(+)

diff --git a/source3/smbd/smb2_query_directory.c b/source3/smbd/smb2_query_directory.c
index fc9ce522397..915067315de 100644
--- a/source3/smbd/smb2_query_directory.c
+++ b/source3/smbd/smb2_query_directory.c
@@ -204,12 +204,15 @@ static NTSTATUS fetch_write_time_recv(struct tevent_req *req);
 
 
 struct smbd_smb2_query_directory_state {
+	struct tevent_context *ev;
 	struct smbd_smb2_request *smb2req;
 	uint64_t async_count;
+	uint32_t find_async_delay_usec;
 	DATA_BLOB out_output_buffer;
 };
 
 static void smb2_query_directory_fetch_write_time_done(struct tevent_req *subreq);
+static void smb2_query_directory_waited(struct tevent_req *subreq);
 
 static struct tevent_req *smbd_smb2_query_directory_send(TALLOC_CTX *mem_ctx,
 					      struct tevent_context *ev,
@@ -249,6 +252,7 @@ static struct tevent_req *smbd_smb2_query_directory_send(TALLOC_CTX *mem_ctx,
 	if (req == NULL) {
 		return NULL;
 	}
+	state->ev = ev;
 	state->smb2req = smb2req;
 	state->out_output_buffer = data_blob_null;
 
@@ -488,6 +492,13 @@ static struct tevent_req *smbd_smb2_query_directory_send(TALLOC_CTX *mem_ctx,
 		smb2_request_set_async_internal(smb2req, true);
 	}
 
+	/*
+	 * This gets set in autobuild for some tests
+	 */
+	state->find_async_delay_usec = lp_parm_ulong(SNUM(conn), "smbd",
+						     "find async delay usec",
+						     0);
+
 	while (true) {
 		bool got_exact_match = false;
 		int space_remaining = in_output_buffer_length - off;
@@ -577,6 +588,30 @@ last_entry_done:
 				  state->async_count);
 			return req;
 		}
+
+		if (state->find_async_delay_usec > 0) {
+			struct timeval tv;
+			struct tevent_req *subreq = NULL;
+
+			/*
+			 * Should we only set async_internal
+			 * if we're not the last request in
+			 * a compound chain?
+			 */
+			smb2_request_set_async_internal(smb2req, true);
+
+			tv = timeval_current_ofs(0, state->find_async_delay_usec);
+
+			subreq = tevent_wakeup_send(state, ev, tv);
+			if (tevent_req_nomem(subreq, req)) {
+				return tevent_req_post(req, ev);
+			}
+			tevent_req_set_callback(subreq,
+						smb2_query_directory_waited,
+						req);
+			return req;
+		}
+
 		tevent_req_done(req);
 		return tevent_req_post(req, ev);
 	}
@@ -605,10 +640,41 @@ static void smb2_query_directory_fetch_write_time_done(struct tevent_req *subreq
 		return;
 	}
 
+	if (state->find_async_delay_usec > 0) {
+		struct timeval tv;
+
+		tv = timeval_current_ofs(0, state->find_async_delay_usec);
+
+		subreq = tevent_wakeup_send(state, state->ev, tv);
+		if (tevent_req_nomem(subreq, req)) {
+			tevent_req_post(req, state->ev);
+			return;
+		}
+		tevent_req_set_callback(subreq,
+					smb2_query_directory_waited,
+					req);
+		return;
+	}
+
 	tevent_req_done(req);
 	return;
 }
 
+static void smb2_query_directory_waited(struct tevent_req *subreq)
+{
+	struct tevent_req *req = tevent_req_callback_data(
+		subreq, struct tevent_req);
+	bool ok;
+
+	ok = tevent_wakeup_recv(subreq);
+	TALLOC_FREE(subreq);
+	if (!ok) {
+		tevent_req_oom(req);
+		return;
+	}
+	tevent_req_done(req);
+}
+
 static NTSTATUS smbd_smb2_query_directory_recv(struct tevent_req *req,
 				    TALLOC_CTX *mem_ctx,
 				    DATA_BLOB *out_output_buffer)
-- 
2.11.4.GIT