fumiama/WireGold
1
0
Fork 0
mirror of https://github.com/fumiama/WireGold.git synced 2026-06-27 06:10:26 +08:00
Code Activity

add peerwise mtu

Browse Source
This commit is contained in:
源文雨
2022-05-13 13:50:41 +08:00
parent 4ccd99b63f
commit 4e833c877d
10 changed files with 19 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
gold/link/send.go
View File

@@ -12,7 +12,7 @@ import (
// WriteAndPut 向 peer 发包并将包放回缓存池
func (l *Link) WriteAndPut(p *head.Packet, istransfer bool) (n int, err error) {
teatype := uint8(rand.Intn(16))
if len(p.Data) <= int(l.me.mtu) {
if len(p.Data) <= int(l.mtu) {
if !istransfer {
p.FillHash()
p.Data = l.Encode(teatype, p.Data)
@@ -30,15 +30,15 @@ func (l *Link) WriteAndPut(p *head.Packet, istransfer bool) (n int, err error) {
i := 0
packet := head.SelectPacket()
*packet = *p
for ; int(totl)-i > int(l.me.mtu); i += int(l.me.mtu) {
logrus.Debugln("[link] split frag", i, ":", i+int(l.me.mtu), ", remain:", int(totl)-i-int(l.me.mtu))
packet.Data = data[:int(l.me.mtu)]
for ; int(totl)-i > int(l.mtu); i += int(l.mtu) {
logrus.Debugln("[link] 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, totl, uint16(uint(i)>>3), istransfer, true)
n += cnt
if err != nil {
return n, err
}
data = data[int(l.me.mtu):]
data = data[int(l.mtu):]
packet.TTL = ttl
}
packet.Put()
@@ -54,7 +54,7 @@ func (l *Link) write(p *head.Packet, teatype uint8, datasz uint32, offset uint16
var d []byte
var cl func()
if istransfer {
if p.Flags&0x4000 == 0x4000 && len(p.Data) > int(l.me.mtu) {
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, 0, 0, false, false)