1
0
mirror of https://github.com/fumiama/WireGold.git synced 2026-06-04 23:40:26 +08:00
This commit is contained in:
fumiama
2021-12-31 20:06:18 +08:00
parent 2761865178
commit f7cf029d00
2 changed files with 9 additions and 4 deletions

View File

@@ -82,13 +82,15 @@ func (l *Link) Write(p *head.Packet, istransfer bool) (n int, err error) {
}
return l.write(p, uint32(len(p.Data)), 0, istransfer, false)
}
p.FillHash()
p.Data = l.Encode(p.Data)
if !istransfer {
p.FillHash()
p.Data = l.Encode(p.Data)
}
data := p.Data
totl := uint32(len(data))
i := 0
for ; int(totl)-i > int(l.me.mtu); i += int(l.me.mtu) {
logrus.Infoln("[link] split frag", i, ":", i+int(l.me.mtu), ", remain:", int(totl)-i)
logrus.Infoln("[link] split frag", i, ":", i+int(l.me.mtu), ", remain:", int(totl)-i-int(l.me.mtu))
packet := *p
packet.Data = data[:int(l.me.mtu)]
cnt, err := l.write(&packet, totl, uint16(i), istransfer, true)
@@ -110,7 +112,7 @@ func (l *Link) Write(p *head.Packet, istransfer bool) (n int, err error) {
func (l *Link) String() (n string) {
n = "default"
if l.pubk != nil {
b, err := base14.UTF16be2utf8(base14.Encode(l.pubk[:10]))
b, err := base14.UTF16be2utf8(base14.Encode(l.pubk[:7]))
if err == nil {
n = helper.BytesToString(b)
} else {

View File

@@ -2,6 +2,7 @@ package link
import (
"encoding/binary"
"encoding/hex"
"time"
"unsafe"
@@ -54,6 +55,7 @@ func (m *Me) wait(data []byte) *head.Packet {
hsh := *(*[32]byte)(*(*unsafe.Pointer)(unsafe.Pointer(&hashd)))
h, ok := m.recving[hsh]
if ok {
logrus.Infoln("[recv] get another frag part of", hex.EncodeToString(hashd))
ok, err := h.Unmarshal(data)
if err == nil {
if ok {
@@ -65,6 +67,7 @@ func (m *Me) wait(data []byte) *head.Packet {
}
return nil
}
logrus.Infoln("[recv] get new frag part of", hex.EncodeToString(hashd))
h = &head.Packet{}
_, err := h.Unmarshal(data)
if err != nil {