mirror of
https://github.com/fumiama/WireGold.git
synced 2026-06-05 07:50:24 +08:00
103 lines
2.4 KiB
Go
103 lines
2.4 KiB
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"
|
|
"golang.org/x/crypto/chacha20poly1305"
|
|
)
|
|
|
|
type PeerConfig struct {
|
|
PeerIP string
|
|
EndPoint string
|
|
AllowedIPs, Querys []string
|
|
PubicKey *[32]byte
|
|
PresharedKey *[32]byte
|
|
KeepAliveDur, QueryTick int64
|
|
MTU uint16
|
|
MTURandomRange uint16
|
|
AllowTrans, NoPipe bool
|
|
UseZstd bool
|
|
}
|
|
|
|
// AddPeer 添加一个 peer
|
|
func (m *Me) AddPeer(cfg *PeerConfig) (l *Link) {
|
|
cfg.PeerIP = net.ParseIP(cfg.PeerIP).String()
|
|
var ok bool
|
|
l, ok = m.IsInPeer(cfg.PeerIP)
|
|
if ok {
|
|
return
|
|
}
|
|
if cfg.MTU == 0 || (m.mtu != 0 && cfg.MTU > m.mtu) {
|
|
panic("invalid mtu for peer " + cfg.PeerIP)
|
|
}
|
|
l = &Link{
|
|
pubk: cfg.PubicKey,
|
|
peerip: net.ParseIP(cfg.PeerIP),
|
|
allowtrans: cfg.AllowTrans,
|
|
usezstd: cfg.UseZstd,
|
|
me: m,
|
|
mtu: cfg.MTU,
|
|
mturandomrange: cfg.MTURandomRange,
|
|
}
|
|
|
|
if !cfg.NoPipe {
|
|
l.pipe = make(chan *head.Packet, 32)
|
|
}
|
|
if cfg.PubicKey != nil {
|
|
c := curve.Get(m.privKey[:])
|
|
k, err := c.Shared(cfg.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 cfg.PresharedKey != nil {
|
|
var err error
|
|
l.aead, err = chacha20poly1305.NewX(cfg.PresharedKey[:])
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
if cfg.EndPoint != "" {
|
|
e, err := net.ResolveUDPAddr("udp", cfg.EndPoint)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
l.endpoint = e
|
|
}
|
|
if cfg.AllowedIPs != nil {
|
|
l.allowedips = make([]*net.IPNet, 0, len(cfg.AllowedIPs))
|
|
for _, ipnet := range cfg.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[cfg.PeerIP] = l
|
|
l.me.connmapmu.Unlock()
|
|
} else {
|
|
panic(err)
|
|
}
|
|
}
|
|
}
|
|
logrus.Infoln("[peer] add peer:", cfg.PeerIP, "allow:", cfg.AllowedIPs)
|
|
go l.keepAlive(cfg.KeepAliveDur)
|
|
go l.sendquery(time.Second*time.Duration(cfg.QueryTick), cfg.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
|
|
}
|