identify: implement identify push protocol

p2p/protocol/identify/id.go

@@ -3,6 +3,7 @@ package identify
 import (
 	"context"
 	"sync"
+	"time"
 
 	pb "github.com/libp2p/go-libp2p/p2p/protocol/identify/pb"
 
@@ -23,6 +24,9 @@ var log = logging.Logger("net/identify")
 // ID is the protocol.ID of the Identify Service.
 const ID = "/ipfs/id/1.0.0"
 
+// IDPush is the protocol.ID of the Identify push protocol
+const IDPush = "/ipfs/id/push/1.0.0"
+
 // LibP2PVersion holds the current protocol version for a client running this code
 // TODO(jbenet): fix the versioning mess.
 const LibP2PVersion = "ipfs/0.1.0"
@@ -60,6 +64,7 @@ func NewIDService(h host.Host) *IDService {
 		currid: make(map[inet.Conn]chan struct{}),
 	}
 	h.SetStreamHandler(ID, s.requestHandler)
+	h.SetStreamHandler(IDPush, s.pushHandler)
 	h.Network().Notify((*netNotifiee)(s))
 	return s
 }
@@ -138,6 +143,24 @@ func (ids *IDService) responseHandler(s inet.Stream) {
 	go inet.FullClose(s)
 }
 
+func (ids *IDService) pushHandler(s inet.Stream) {
+	ids.responseHandler(s)
+}
+
+func (ids *IDService) Push() {
+	for _, p := range ids.Host.Network().Peers() {
+		ctx, cancel := context.WithTimeout(context.Background(), 60*time.Second)
+		s, err := ids.Host.NewStream(ctx, p, IDPush)
+		cancel()
+		if err != nil {
+			log.Debugf("error opening push stream: %s", err.Error())
+			continue
+		}
+
+		ids.requestHandler(s)
+	}
+}
+
 func (ids *IDService) populateMessage(mes *pb.Identify, c inet.Conn) {
 
 	// set protocols this node is currently handling