fix(net) use NetMessage interface

net/interface.go

@@ -26,7 +26,7 @@ type Network interface {
 	GetProtocols() *mux.ProtocolMap
 
 	// SendMessage sends given Message out
-	SendMessage(*msg.Message) error
+	SendMessage(msg.NetMessage) error
 
 	// Close terminates all network operation
 	Close() error

net/net.go

@@ -86,7 +86,7 @@ func (n *IpfsNetwork) GetProtocols() *mux.ProtocolMap {
 }
 
 // SendMessage sends given Message out
-func (n *IpfsNetwork) SendMessage(m *msg.Message) error {
+func (n *IpfsNetwork) SendMessage(m msg.NetMessage) error {
 	n.swarm.Outgoing <- m
 	return nil
 }

net/swarm/conn.go

@@ -153,7 +153,7 @@ func (s *Swarm) fanOut() {
 			}
 
 			s.connsLock.RLock()
-			conn, found := s.conns[msg.Peer.Key()]
+			conn, found := s.conns[msg.Peer().Key()]
 			s.connsLock.RUnlock()
 
 			if !found {
@@ -164,7 +164,7 @@ func (s *Swarm) fanOut() {
 			}
 
 			// queue it in the connection's buffer
-			conn.Outgoing.MsgChan <- msg.Data
+			conn.Outgoing.MsgChan <- msg.Data()
 		}
 	}
 }
@@ -189,7 +189,7 @@ func (s *Swarm) fanIn(c *conn.Conn) {
 				goto out
 			}
 
-			msg := &msg.Message{Peer: c.Peer, Data: data}
+			msg := msg.New(c.Peer, data)
 			s.Incoming <- msg
 		}
 	}