fumiama/WireGold
1
0
Fork 0
mirror of https://github.com/fumiama/WireGold.git synced 2026-07-01 00:00:23 +08:00
Code Activity

fix mtu error

Browse Source
This commit is contained in:
fumiama
2021-12-30 23:48:17 +08:00
parent 4013c482b4
commit c207a5daf6
4 changed files with 109 additions and 77 deletions
Download Patch File Download Diff File Expand all files Collapse all files

46
gold/link/link.go
View File

@@ -75,6 +75,39 @@ func (l *Link) Read() *head.Packet {
// Write 向 peer 发包
func (l *Link) Write(p *head.Packet, istransfer bool) (n int, err error) {
if len(p.Data) <= int(l.me.mtu) {
return l.write(p, istransfer)
}
data := p.Data
offset := 0
for len(data) > int(l.me.mtu) {
packet := *p
packet.Data = data[offset*int(l.me.mtu) : (offset+1)*int(l.me.mtu)]
i, err := l.write(&packet, istransfer)
n += i
if err != nil {
return n, err
}
data = data[(offset+1)*int(l.me.mtu):]
}
return n, nil
}
func (l *Link) String() (n string) {
n = "default"
if l.pubk != nil {
b, err := base14.UTF16be2utf8(base14.Encode(l.pubk[:10]))
if err == nil {
n = helper.BytesToString(b)
} else {
n = err.Error()
}
}
return
}
// write 向 peer 发一个包
func (l *Link) write(p *head.Packet, istransfer bool) (n int, err error) {
var d []byte
if istransfer {
d = p.Marshal(nil)
@@ -102,16 +135,3 @@ func (l *Link) Write(p *head.Packet, istransfer bool) (n int, err error) {
}
return
}
func (l *Link) String() (n string) {
n = "default"
if l.pubk != nil {
b, err := base14.UTF16be2utf8(base14.Encode(l.pubk[:10]))
if err == nil {
n = helper.BytesToString(b)
} else {
n = err.Error()
}
}
return
}