mirror of
https://github.com/fumiama/WireGold.git
synced 2026-06-09 02:02:41 +08:00
87 lines
2.4 KiB
Go
87 lines
2.4 KiB
Go
package link
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"math/rand"
|
|
"sync/atomic"
|
|
|
|
"github.com/fumiama/WireGold/gold/head"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
// WriteAndPut 向 peer 发包并将包放回缓存池
|
|
func (l *Link) WriteAndPut(p *head.Packet, istransfer bool) (n int, err error) {
|
|
teatype := uint8(rand.Intn(16))
|
|
sndcnt := atomic.AddUintptr(&l.sendcount, 1)
|
|
logrus.Debugln("[send] count:", sndcnt, "additional data:", uint16(sndcnt))
|
|
if len(p.Data) <= int(l.mtu) {
|
|
if !istransfer {
|
|
p.FillHash()
|
|
if l.aead != nil {
|
|
p.Data = l.EncodePreshared(uint16(sndcnt), p.Data)
|
|
}
|
|
p.Data = l.Encode(teatype, p.Data)
|
|
}
|
|
defer p.Put()
|
|
return l.write(p, teatype, uint16(sndcnt), uint32(len(p.Data)), 0, istransfer, false)
|
|
}
|
|
if !istransfer {
|
|
p.FillHash()
|
|
if l.aead != nil {
|
|
p.Data = l.EncodePreshared(uint16(sndcnt), p.Data)
|
|
}
|
|
p.Data = l.Encode(teatype, p.Data)
|
|
}
|
|
data := p.Data
|
|
ttl := p.TTL
|
|
totl := uint32(len(data))
|
|
i := 0
|
|
packet := head.SelectPacket()
|
|
*packet = *p
|
|
for ; int(totl)-i > int(l.mtu); i += int(l.mtu) {
|
|
logrus.Debugln("[send] split frag", i, ":", i+int(l.mtu), ", remain:", int(totl)-i-int(l.mtu))
|
|
packet.Data = data[:int(l.mtu)]
|
|
cnt, err := l.write(packet, teatype, uint16(sndcnt), totl, uint16(i>>3), istransfer, true)
|
|
n += cnt
|
|
if err != nil {
|
|
return n, err
|
|
}
|
|
data = data[int(l.mtu):]
|
|
packet.TTL = ttl
|
|
}
|
|
packet.Put()
|
|
p.Data = data
|
|
cnt, err := l.write(p, teatype, uint16(sndcnt), totl, uint16(i>>3), istransfer, false)
|
|
p.Put()
|
|
n += cnt
|
|
return n, err
|
|
}
|
|
|
|
// write 向 peer 发一个包
|
|
func (l *Link) write(p *head.Packet, teatype uint8, additional uint16, datasz uint32, offset uint16, istransfer, hasmore bool) (n int, err error) {
|
|
var d []byte
|
|
var cl func()
|
|
if istransfer {
|
|
if p.Flags&0x4000 == 0x4000 && len(p.Data) > int(l.mtu) {
|
|
return len(p.Data), errors.New("drop dont fragmnet big trans packet")
|
|
}
|
|
d, cl = p.Marshal(nil, teatype, additional, 0, 0, false, false)
|
|
} else {
|
|
d, cl = p.Marshal(l.me.me, teatype, additional, datasz, offset, false, hasmore)
|
|
}
|
|
if d == nil {
|
|
return 0, errors.New("[send] ttl exceeded")
|
|
}
|
|
if err == nil {
|
|
peerep := l.endpoint
|
|
if peerep == nil {
|
|
return 0, errors.New("[send] nil endpoint of " + p.Dst.String())
|
|
}
|
|
logrus.Debugln("[send] write", len(d), "bytes data from ep", l.me.myep.LocalAddr(), "to", peerep, "offset:", fmt.Sprintf("%04x", offset))
|
|
n, err = l.me.myep.WriteToUDP(d, peerep)
|
|
cl()
|
|
}
|
|
return
|
|
}
|