Add some client callbacks

[btrtrc.git] / client.go

diff --git a/client.go b/client.go
index fe7786ac4ec96d27d7bda9d4a99e461c0e208ec0..69ef1ef5d3271b855b6153ec517b65f6b2962460 100644 (file)
--- a/client.go
+++ b/client.go
@@ -860,6 +860,9 @@ func (cl *Client) connBtHandshake(c *PeerConn, ih *metainfo.Hash) (ret metainfo.
        c.PeerExtensionBytes = res.PeerExtensionBits
        c.PeerID = res.PeerID
        c.completedHandshake = time.Now()
+       if cb := cl.config.Callbacks.CompletedHandshake; cb != nil {
+               cb(c, res.Hash)
+       }
        return
 }