plugeth/swarm/network/enr.go

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/ethereum/go-ethereum/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()),
	}
}