fumiama/WireGold
1
0
Fork 0
mirror of https://github.com/fumiama/WireGold.git synced 2026-06-05 07:50:24 +08:00
Code Activity

add istransfer option in link.Write

Browse Source
This commit is contained in:
fumiama
2021-12-30 16:03:08 +08:00
parent 6627881c63
commit c3d05635f0
7 changed files with 17 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
gold/link/link.go
View File

@@ -72,10 +72,15 @@ func (l *Link) Read() *head.Packet {
}
// Write 向 peer 发包
func (l *Link) Write(p *head.Packet) (n int, err error) {
func (l *Link) Write(p *head.Packet, istransfer bool) (n int, err error) {
p.FillHash()
p.Data = l.Encode(p.Data)
d := p.Marshal(l.me.me)
var d []byte
if istransfer {
d = p.Marshal(nil)
} else {
d = p.Marshal(l.me.me)
}
if d == nil {
return 0, errors.New("[link] ttl exceeded")
}