WireGold/gold/link/peer.go

package link

import (
	"net"
	"time"

	"github.com/fumiama/WireGold/gold/head"
	curve "github.com/fumiama/go-x25519"
	tea "github.com/fumiama/gofastTEA"
	"github.com/sirupsen/logrus"
)

// AddPeer 添加一个 peer
func (m *Me) AddPeer(peerip string, pubicKey *[32]byte, endPoint string, allowedIPs, querys []string, keepAliveDur, queryTick int64, allowTrans, nopipe bool) (l *Link) {
	peerip = net.ParseIP(peerip).String()
	var ok bool
	l, ok = m.IsInPeer(peerip)
	if ok {
		return
	}
	l = &Link{
		pubk:       pubicKey,
		peerip:     net.ParseIP(peerip),
		allowtrans: allowTrans,
		me:         m,
	}

	if !nopipe {
		l.pipe = make(chan *head.Packet, 32)
	}
	if pubicKey != nil {
		c := curve.Get(m.privKey[:])
		k, err := c.Shared(pubicKey)
		if err == nil {
			l.key = make([]tea.TEA, 16)
			for i := range l.key {
				l.key[i] = tea.NewTeaCipherLittleEndian(k[i : 16+i])
			}
		}
	}
	if endPoint != "" {
		e, err := net.ResolveUDPAddr("udp", endPoint)
		if err != nil {
			panic(err)
		}
		l.endpoint = e
	}
	if allowedIPs != nil {
		l.allowedips = make([]*net.IPNet, 0, len(allowedIPs))
		for _, ipnet := range allowedIPs {
			_, cidr, err := net.ParseCIDR(ipnet)
			if err == nil {
				l.allowedips = append(l.allowedips, cidr)
				l.me.router.SetItem(cidr, l)
				l.me.connmapmu.Lock()
				l.me.connections[peerip] = l
				l.me.connmapmu.Unlock()
			} else {
				panic(err)
			}
		}
	}
	logrus.Infoln("[peer] add peer:", peerip, "allow:", allowedIPs)
	go l.keepAlive(keepAliveDur)
	go l.sendquery(time.Second*time.Duration(queryTick), querys...)
	return
}

// IsInPeer 查找 peer 是否已经在册
func (m *Me) IsInPeer(peer string) (p *Link, ok bool) {
	m.connmapmu.RLock()
	p, ok = m.connections[peer]
	m.connmapmu.RUnlock()
	return
}