fumiama/WireGold
1
0
Fork 0
mirror of https://github.com/fumiama/WireGold.git synced 2026-06-11 04:06:14 +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

4
gold/head/packet.go
View File

@@ -86,7 +86,9 @@ func (p *Packet) Marshal(src net.IP) []byte {
}
p.DataSZ = uint32(len(p.Data))
p.Src = src
if src != nil {
p.Src = src
}
packet := make([]byte, 52+len(p.Data))
binary.LittleEndian.PutUint32(packet[:4], p.DataSZ)

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")
}

4
gold/link/listen.go
View File

@@ -46,7 +46,7 @@ func (m *Me) listen() (conn *net.UDPConn, err error) {
case head.ProtoHello:
switch p.status {
case LINK_STATUS_DOWN:
_, _ = p.Write(head.NewPacket(head.ProtoHello, 0, p.peerip, 0, nil))
_, _ = p.Write(head.NewPacket(head.ProtoHello, 0, p.peerip, 0, nil), false)
logrus.Infoln("[link] send hello ack packet")
p.status = LINK_STATUS_HALFUP
case LINK_STATUS_HALFUP:
@@ -77,7 +77,7 @@ func (m *Me) listen() (conn *net.UDPConn, err error) {
} else if p.Accept(packet.Dst) {
if p.allowtrans {
// 转发
p.Write(&packet)
p.Write(&packet, true)
logrus.Infoln("[link] trans packet to", packet.Dst.String()+":"+strconv.Itoa(int(packet.DstPort)))
} else {
logrus.Warnln("[link] refused to trans packet to", packet.Dst.String()+":"+strconv.Itoa(int(packet.DstPort)))

2
gold/link/nat.go
View File

@@ -15,7 +15,7 @@ func (l *Link) keepAlive() {
go func() {
t := time.NewTicker(time.Second * time.Duration(l.keepalive))
for range t.C {
_, _ = l.Write(head.NewPacket(head.ProtoHello, 0, l.peerip, 0, nil))
_, _ = l.Write(head.NewPacket(head.ProtoHello, 0, l.peerip, 0, nil), false)
logrus.Infoln("[link.nat] send keep alive packet")
}
}()

2
gold/link/query.go
View File

@@ -27,6 +27,6 @@ func (l *Link) SendQuery(peers ...string) error {
if err != nil {
return err
}
_, err = l.Write(head.NewPacket(head.ProtoQuery, 0, l.peerip, 0, data))
_, err = l.Write(head.NewPacket(head.ProtoQuery, 0, l.peerip, 0, data), false)
return err
}