s3:libsmb: add tstream_cli_np_set_timeout()

metze

1 files changed, 15 insertions, 0 deletions

diff --git a/source3/libsmb/cli_np_tstream.c b/source3/libsmb/cli_np_tstream.c
index a46b511905..1cab1ed460 100644
--- a/source3/libsmb/cli_np_tstream.c
+++ b/source3/libsmb/cli_np_tstream.c
@@ -35,6 +35,7 @@ struct tstream_cli_np {
 	struct cli_state *cli;
 	const char *npipe;
 	uint16_t fnum;
+	unsigned int default_timeout;
 
 	struct {
 		bool active;
@@ -176,6 +177,7 @@ NTSTATUS _tstream_cli_np_open_recv(struct tevent_req *req,
 	cli_nps->cli = state->cli;
 	cli_nps->npipe = talloc_move(cli_nps, &state->npipe);
 	cli_nps->fnum = state->fnum;
+	cli_nps->default_timeout = state->cli->timeout;
 
 	talloc_set_destructor(cli_nps, tstream_cli_np_destructor);
 
@@ -233,6 +235,19 @@ NTSTATUS tstream_cli_np_use_trans(struct tstream_context *stream)
 	return NT_STATUS_OK;
 }
 
+unsigned int tstream_cli_np_set_timeout(struct tstream_context *stream,
+					unsigned int timeout)
+{
+	struct tstream_cli_np *cli_nps = tstream_context_data(stream,
+					 struct tstream_cli_np);
+
+	if (!cli_state_is_connected(cli_nps->cli)) {
+		return cli_nps->default_timeout;
+	}
+
+	return cli_set_timeout(cli_nps->cli, timeout);
+}
+
 struct tstream_cli_np_writev_state {
 	struct tstream_context *stream;
 	struct tevent_context *ev;