fumiama/WireGold
1
0
Fork 0
mirror of https://github.com/fumiama/WireGold.git synced 2026-06-19 09:40:23 +08:00
Code Activity

fix 分片

Browse Source
This commit is contained in:
fumiama
2021-12-31 13:48:27 +08:00
parent fb88615092
commit 5b2d16674b
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
gold/link/link.go
View File

@@ -81,16 +81,16 @@ func (l *Link) Write(p *head.Packet, istransfer bool) (n int, err error) {
data := p.Data data := p.Data
totl := len(data) totl := len(data)
i := 0 i := 0
for ; totl > int(l.me.mtu); i += int(l.me.mtu) { for ; totl-i > int(l.me.mtu); i += int(l.me.mtu) {
logrus.Infoln("[link] split frag", i, ":", i+int(l.me.mtu), ", remain:", totl-i)
packet := *p packet := *p
packet.Data = data[i : i+int(l.me.mtu)] packet.Data = data[:int(l.me.mtu)]
cnt, err := l.write(&packet, uint16(i), istransfer, true) cnt, err := l.write(&packet, uint16(i), istransfer, true)
n += cnt n += cnt
if err != nil { if err != nil {
return n, err return n, err
} }
data = data[i+int(l.me.mtu):] data = data[int(l.me.mtu):]
totl -= int(l.me.mtu)
} }
p.Data = data p.Data = data
cnt, err := l.write(p, uint16(i), istransfer, false) cnt, err := l.write(p, uint16(i), istransfer, false)

2
gold/link/me.go
View File

@@ -71,7 +71,7 @@ func NewMe(privateKey *[32]byte, myipwithmask string, myEndpoint string, nopipei
m.loop = m.AddPeer(m.me.String(), nil, "127.0.0.1:56789", []string{myipwithmask}, 0, false, nopipeinlink) m.loop = m.AddPeer(m.me.String(), nil, "127.0.0.1:56789", []string{myipwithmask}, 0, false, nopipeinlink)
m.srcport = srcport m.srcport = srcport
m.dstport = dstport m.dstport = dstport
m.mtu = mtu m.mtu = mtu & 0xfff8
go m.initrecvpool() go m.initrecvpool()
return return
} }