-
-
Notifications
You must be signed in to change notification settings - Fork 88
/
dht.go
39 lines (34 loc) · 1.06 KB
/
dht.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
package dht
import (
"github.com/bitmagnet-io/bitmagnet/internal/protocol"
"net/netip"
)
type ID = protocol.ID
const (
QPing = "ping"
QFindNode = "find_node"
QGetPeers = "get_peers"
QAnnouncePeer = "announce_peer"
QSampleInfohashes = "sample_infohashes"
)
type RecvMsg struct {
Msg Msg
From netip.AddrPort
}
// AnnouncePort returns the torrent port for the message.
// There is an optional argument called implied_port which value is either 0 or 1.
// If it is present and non-zero, the port argument should be ignored and the source port of the UDP packet should be used as the peer's port instead.
// This is useful for peers behind a NAT that may not know their external port, and supporting uTP,
// they accept incoming connections on the same port as the DHT port.
// https://www.bittorrent.org/beps/bep_0005.html
func (m RecvMsg) AnnouncePort() uint16 {
port := m.From.Port()
args := m.Msg.A
if args != nil && !args.ImpliedPort {
argsPort := args.Port
if argsPort != nil {
port = uint16(*argsPort)
}
}
return port
}