87 lines
2.3 KiB
Go
87 lines
2.3 KiB
Go
package discovery
|
|
|
|
import (
|
|
"fmt"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/ethereum/go-ethereum/crypto"
|
|
"github.com/ethereum/go-ethereum/log"
|
|
"github.com/ethereum/go-ethereum/p2p/discv5"
|
|
"github.com/ethereum/go-ethereum/p2p/enr"
|
|
ma "github.com/multiformats/go-multiaddr"
|
|
)
|
|
|
|
// ProxyToRendezvous proxies records discovered using original to rendezvous servers for specified topic.
|
|
func ProxyToRendezvous(original Discovery, servers []ma.Multiaddr, topic string, stop chan struct{}) error {
|
|
var (
|
|
identities = map[discv5.NodeID]*Rendezvous{}
|
|
period = make(chan time.Duration, 1)
|
|
found = make(chan *discv5.Node, 10)
|
|
lookup = make(chan bool)
|
|
wg sync.WaitGroup
|
|
)
|
|
period <- 1 * time.Second
|
|
wg.Add(1)
|
|
go func() {
|
|
if err := original.Discover(topic, period, found, lookup); err != nil {
|
|
log.Error("discover request failed", "topic", topic, "error", err)
|
|
}
|
|
wg.Done()
|
|
}()
|
|
for {
|
|
select {
|
|
case <-stop:
|
|
close(period)
|
|
wg.Wait()
|
|
return nil
|
|
case <-lookup:
|
|
case n := <-found:
|
|
if _, exist := identities[n.ID]; exist {
|
|
continue
|
|
}
|
|
log.Info("proxying new record", "topic", topic, "identity", n.String())
|
|
record, err := makeProxiedENR(n)
|
|
if err != nil {
|
|
log.Error("error converting discovered node to ENR", "node", n.String(), "error", err)
|
|
}
|
|
r := NewRendezvousWithENR(servers, record)
|
|
identities[n.ID] = r
|
|
if err := r.Start(); err != nil {
|
|
log.Error("unable to start rendezvous proxying", "servers", servers, "error", err)
|
|
}
|
|
wg.Add(1)
|
|
go func() {
|
|
if err := r.Register(topic, stop); err != nil {
|
|
log.Error("register error", "topic", topic, "error", err)
|
|
}
|
|
wg.Done()
|
|
}()
|
|
}
|
|
}
|
|
}
|
|
|
|
func makeProxiedENR(n *discv5.Node) (enr.Record, error) {
|
|
record := enr.Record{}
|
|
record.Set(enr.IP(n.IP))
|
|
record.Set(enr.TCP(n.TCP))
|
|
record.Set(enr.UDP(n.UDP))
|
|
record.Set(Proxied(n.ID))
|
|
key, err := crypto.GenerateKey() // we need separate key for each identity, records are stored based on it
|
|
if err != nil {
|
|
return record, fmt.Errorf("unable to generate private key. error : %v", err)
|
|
}
|
|
if err := enr.SignV4(&record, key); err != nil {
|
|
return record, fmt.Errorf("unable to sign enr record. error: %v", err)
|
|
}
|
|
return record, nil
|
|
}
|
|
|
|
// Proxied is an Entry for ENR
|
|
type Proxied discv5.NodeID
|
|
|
|
// ENRKey returns unique key that is used by ENR.
|
|
func (p Proxied) ENRKey() string {
|
|
return "proxied"
|
|
}
|