add accessor for tstream_context

Signed-off-by: Noel Power <noel.power@suse.com>
Reviewed-by: Andrew Bartlett <abartlet@samba.org>

diff --git a/source3/rpc_client/rpc_transport.h b/source3/rpc_client/rpc_transport.h
index dccfa36cdc52982f4050b05f3a9a6ccfb47cabc2..f352f60cf2a4f323e7b27666f57310e4290e0008 100644 (file)
--- a/source3/rpc_client/rpc_transport.h
+++ b/source3/rpc_client/rpc_transport.h
@@ -99,5 +99,6 @@ NTSTATUS rpc_transport_sock_init(TALLOC_CTX *mem_ctx, int fd,
 NTSTATUS rpc_transport_tstream_init(TALLOC_CTX *mem_ctx,
                                struct tstream_context **stream,
                                struct rpc_cli_transport **presult);
-
+struct tstream_context *rpc_transport_get_tstream(
+               struct rpc_cli_transport *transport);
 #endif /* _RPC_CLIENT_RPC_TRANSPORT_H_ */

diff --git a/source3/rpc_client/rpc_transport_tstream.c b/source3/rpc_client/rpc_transport_tstream.c
index 3c64d03bb126166319088265c078e600d6ce139d..02fc320570379e5b050ca5cf1ed091c13f460d97 100644 (file)
--- a/source3/rpc_client/rpc_transport_tstream.c
+++ b/source3/rpc_client/rpc_transport_tstream.c
@@ -554,3 +554,11 @@ NTSTATUS rpc_transport_tstream_init(TALLOC_CTX *mem_ctx,
        *presult = result;
        return NT_STATUS_OK;
 }
+
+struct tstream_context *rpc_transport_get_tstream(
+               struct rpc_cli_transport *transport)
+{
+       struct rpc_tstream_state *state = talloc_get_type_abort(
+                       transport->priv, struct rpc_tstream_state);
+       return state->stream;
+}