swarm: codebase split from go-ethereum (#1405)

network/enr.go

@@ -0,0 +1,93 @@
+package network
+
+import (
+	"fmt"
+	"io"
+
+	"github.com/ethereum/go-ethereum/p2p"
+	"github.com/ethereum/go-ethereum/p2p/enode"
+	"github.com/ethereum/go-ethereum/p2p/protocols"
+	"github.com/ethereum/go-ethereum/rlp"
+	"github.com/ethersphere/swarm/log"
+)
+
+// ENRAddrEntry is the entry type to store the bzz key in the enode
+type ENRAddrEntry struct {
+	data []byte
+}
+
+func NewENRAddrEntry(addr []byte) *ENRAddrEntry {
+	return &ENRAddrEntry{
+		data: addr,
+	}
+}
+
+func (b ENRAddrEntry) Address() []byte {
+	return b.data
+}
+
+// ENRKey implements enr.Entry
+func (b ENRAddrEntry) ENRKey() string {
+	return "bzzkey"
+}
+
+// EncodeRLP implements rlp.Encoder
+func (b ENRAddrEntry) EncodeRLP(w io.Writer) error {
+	log.Debug("in encoderlp", "b", b, "p", fmt.Sprintf("%p", &b))
+	return rlp.Encode(w, &b.data)
+}
+
+// DecodeRLP implements rlp.Decoder
+func (b *ENRAddrEntry) DecodeRLP(s *rlp.Stream) error {
+	byt, err := s.Bytes()
+	if err != nil {
+		return err
+	}
+	b.data = byt
+	log.Debug("in decoderlp", "b", b, "p", fmt.Sprintf("%p", &b))
+	return nil
+}
+
+type ENRLightNodeEntry bool
+
+func (b ENRLightNodeEntry) ENRKey() string {
+	return "bzzlightnode"
+}
+
+type ENRBootNodeEntry bool
+
+func (b ENRBootNodeEntry) ENRKey() string {
+	return "bzzbootnode"
+}
+
+func getENRBzzPeer(p *p2p.Peer, rw p2p.MsgReadWriter, spec *protocols.Spec) *BzzPeer {
+	var lightnode ENRLightNodeEntry
+	var bootnode ENRBootNodeEntry
+
+	// retrieve the ENR Record data
+	record := p.Node().Record()
+	record.Load(&lightnode)
+	record.Load(&bootnode)
+
+	// get the address; separate function as long as we need swarm/network:NewAddr() to call it
+	addr := getENRBzzAddr(p.Node())
+
+	// build the peer using the retrieved data
+	return &BzzPeer{
+		Peer:      protocols.NewPeer(p, rw, spec),
+		LightNode: bool(lightnode),
+		BzzAddr:   addr,
+	}
+}
+
+func getENRBzzAddr(nod *enode.Node) *BzzAddr {
+	var addr ENRAddrEntry
+
+	record := nod.Record()
+	record.Load(&addr)
+
+	return &BzzAddr{
+		OAddr: addr.data,
+		UAddr: []byte(nod.String()),
+	}
+}