Allow packet_stream_reconnect() to cut offHEADmaster

 a connecting stream. Cleanup timer.

Signed-off-by: Andrew Opalach <andrew@akon.city>

1 files changed, 2 insertions, 1 deletions

diff --git a/src/rpc.c b/src/rpc.c
index 1b00e63..7938341 100644
--- a/src/rpc.c
+++ b/src/rpc.c
@@ -125,10 +125,11 @@ void nn_rpc_connect(struct nn_rpc *rpc, u8 id, u8 type, str *addr, u16 port)
     nn_packet_stream_connect(rpc->conn->stream, rpc->loop, id, type, addr, port);
 }
 
-void nn_rpc_reconnect(struct nn_rpc *rpc, str *addr, u16 port)
+struct nn_rpc_connection *nn_rpc_reconnect(struct nn_rpc *rpc, str *addr, u16 port)
 {
     al_assert(rpc->conn);
     nn_packet_stream_reconnect(rpc->conn->stream, addr, port);
+    return rpc->conn;
 }
 
 // @TODO: Pack opcode into a u32, reduce ID by 8 bits (make struct with bitmask)