mirror of
https://github.com/status-im/status-go.git
synced 2025-02-22 19:58:29 +00:00
eeca435064
Update vendor Integrate rendezvous into status node Add a test with failover using rendezvous Use multiple servers in client Use discovery V5 by default and test that node can be started with rendezvous discovet Fix linter Update rendezvous client to one with instrumented stream Address feedback Fix test with updated topic limits Apply several suggestions Change log to debug for request errors because we continue execution Remove web3js after rebase Update rendezvous package
99 lines
2.5 KiB
Go
99 lines
2.5 KiB
Go
package relay
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"errors"
|
|
"io"
|
|
|
|
pb "github.com/libp2p/go-libp2p-circuit/pb"
|
|
|
|
ggio "github.com/gogo/protobuf/io"
|
|
proto "github.com/gogo/protobuf/proto"
|
|
peer "github.com/libp2p/go-libp2p-peer"
|
|
pstore "github.com/libp2p/go-libp2p-peerstore"
|
|
ma "github.com/multiformats/go-multiaddr"
|
|
mh "github.com/multiformats/go-multihash"
|
|
)
|
|
|
|
func peerToPeerInfo(p *pb.CircuitRelay_Peer) (pstore.PeerInfo, error) {
|
|
if p == nil {
|
|
return pstore.PeerInfo{}, errors.New("nil peer")
|
|
}
|
|
|
|
h, err := mh.Cast(p.Id)
|
|
if err != nil {
|
|
return pstore.PeerInfo{}, err
|
|
}
|
|
|
|
addrs := make([]ma.Multiaddr, 0, len(p.Addrs))
|
|
for _, addrBytes := range p.Addrs {
|
|
a, err := ma.NewMultiaddrBytes(addrBytes)
|
|
if err == nil {
|
|
addrs = append(addrs, a)
|
|
}
|
|
}
|
|
|
|
return pstore.PeerInfo{ID: peer.ID(h), Addrs: addrs}, nil
|
|
}
|
|
|
|
func peerInfoToPeer(pi pstore.PeerInfo) *pb.CircuitRelay_Peer {
|
|
addrs := make([][]byte, len(pi.Addrs))
|
|
for i, addr := range pi.Addrs {
|
|
addrs[i] = addr.Bytes()
|
|
}
|
|
|
|
p := new(pb.CircuitRelay_Peer)
|
|
p.Id = []byte(pi.ID)
|
|
p.Addrs = addrs
|
|
|
|
return p
|
|
}
|
|
|
|
type delimitedReader struct {
|
|
r io.Reader
|
|
buf []byte
|
|
}
|
|
|
|
// The gogo protobuf NewDelimitedReader is buffered, which may eat up stream data.
|
|
// So we need to implement a compatible delimited reader that reads unbuffered.
|
|
// There is a slowdown from unbuffered reading: when reading the message
|
|
// it can take multiple single byte Reads to read the length and another Read
|
|
// to read the message payload.
|
|
// However, this is not critical performance degradation as
|
|
// - the reader is utilized to read one (dialer, stop) or two messages (hop) during
|
|
// the handshake, so it's a drop in the water for the connection lifetime.
|
|
// - messages are small (max 4k) and the length fits in a couple of bytes,
|
|
// so overall we have at most three reads per message.
|
|
func newDelimitedReader(r io.Reader, maxSize int) *delimitedReader {
|
|
return &delimitedReader{r: r, buf: make([]byte, maxSize)}
|
|
}
|
|
|
|
func (d *delimitedReader) ReadByte() (byte, error) {
|
|
buf := d.buf[:1]
|
|
_, err := d.r.Read(buf)
|
|
return buf[0], err
|
|
}
|
|
|
|
func (d *delimitedReader) ReadMsg(msg proto.Message) error {
|
|
mlen, err := binary.ReadUvarint(d)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if uint64(len(d.buf)) < mlen {
|
|
return errors.New("Message too large")
|
|
}
|
|
|
|
buf := d.buf[:mlen]
|
|
_, err = io.ReadFull(d.r, buf)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return proto.Unmarshal(buf, msg)
|
|
}
|
|
|
|
func newDelimitedWriter(w io.Writer) ggio.WriteCloser {
|
|
return ggio.NewDelimitedWriter(w)
|
|
}
|