Merge pull request #104 from jbenet/feat/fix-ping

Fixed Ping (minor)

crypto/spipe/handshake.go

@@ -67,8 +67,12 @@ func (s *SecurePipe) handshake() error {
 		return err
 	}
 
-	// u.POut("sending encoded handshake\n")
-	s.insecure.Out <- encoded
+	// Send our Propose packet
+	select {
+	case s.insecure.Out <- encoded:
+	case <-s.ctx.Done():
+		return ErrClosed
+	}
 
 	// Parse their Propose packet and generate an Exchange packet.
 	// Exchange = (EphemeralPubKey, Signature)
@@ -137,7 +141,12 @@ func (s *SecurePipe) handshake() error {
 
 	exEncoded, err := proto.Marshal(exPacket)
 
-	s.insecure.Out <- exEncoded
+	// send out Exchange packet
+	select {
+	case s.insecure.Out <- exEncoded:
+	case <-s.ctx.Done():
+		return ErrClosed
+	}
 
 	// Parse their Exchange packet and generate a Finish packet.
 	// Finish = E('Finish')
@@ -182,7 +191,14 @@ func (s *SecurePipe) handshake() error {
 
 	finished := []byte("Finished")
 
-	s.Out <- finished
+	// send finished msg
+	select {
+	case <-s.ctx.Done():
+		return ErrClosed
+	case s.Out <- finished:
+	}
+
+	// recv finished msg
 	var resp2 []byte
 	select {
 	case <-s.ctx.Done():
@@ -194,7 +210,7 @@ func (s *SecurePipe) handshake() error {
 		return errors.New("Negotiation failed.")
 	}
 
-	u.DOut("[%s] identify: Got node id: %s\n", s.local.ID.Pretty(), s.remote.ID.Pretty())
+	u.DOut("[%s] handshake: Got node id: %s\n", s.local.ID.Pretty(), s.remote.ID.Pretty())
 	return nil
 }
 

routing/dht/handlers.go

@@ -104,8 +104,7 @@ func (dht *IpfsDHT) handlePutValue(p *peer.Peer, pmes *Message) (*Message, error
 
 func (dht *IpfsDHT) handlePing(p *peer.Peer, pmes *Message) (*Message, error) {
 	u.DOut("[%s] Responding to ping from [%s]!\n", dht.self.ID.Pretty(), p.ID.Pretty())
-
-	return newMessage(pmes.GetType(), "", int(pmes.GetClusterLevel())), nil
+	return pmes, nil
 }
 
 func (dht *IpfsDHT) handleFindPeer(p *peer.Peer, pmes *Message) (*Message, error) {