TheArchive/tailscale
1
0
Fork 0
mirror of https://github.com/tailscale/tailscale.git synced 2025-08-21 02:17:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

ipn/ipnstate, wgengine/wgint: add handshake attempts accessors

Browse Source 
Not yet used. This is being made available so magicsock/wgengine can
use it to ignore certain sends (UDP + DERP) later on at least mobile,
letting wireguard-go think it's doing its full attempt schedule, but
we can cut it short conditionally based on what we know from the
control plane.

Updates #7617

Signed-off-by: Brad Fitzpatrick <bradfitz@tailscale.com>
Change-Id: Ia367cf6bd87b2aeedd3c6f4989528acdb6773ca7
This commit is contained in:
Brad Fitzpatrick
2024-02-25 06:40:35 -08:00
committed by Brad Fitzpatrick
parent eb28818403
commit 1cf85822d0
4 changed files with 33 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
wgengine/wgint/wgint.go
View File

@@ -17,6 +17,8 @@ var (
offHandshake = getPeerStatsOffset("lastHandshakeNano")
offRxBytes = getPeerStatsOffset("rxBytes")
offTxBytes = getPeerStatsOffset("txBytes")
offHandshakeAttempts = getPeerHandshakeAttemptsOffset()
)
func getPeerStatsOffset(name string) uintptr {
@@ -31,6 +33,22 @@ func getPeerStatsOffset(name string) uintptr {
return field.Offset
}
func getPeerHandshakeAttemptsOffset() uintptr {
peerType := reflect.TypeFor[device.Peer]()
field, ok := peerType.FieldByName("timers")
if !ok {
panic("no timers field in device.Peer")
}
field2, ok := field.Type.FieldByName("handshakeAttempts")
if !ok {
panic("no handshakeAttempts field in device.Peer.timers")
}
if g, w := field2.Type.String(), "atomic.Uint32"; g != w {
panic("unexpected type " + g + " of field handshakeAttempts in device.Peer.timers; want " + w)
}
return field.Offset + field2.Offset
}
// PeerLastHandshakeNano returns the last handshake time in nanoseconds since the
// unix epoch.
func PeerLastHandshakeNano(peer *device.Peer) int64 {
@@ -46,3 +64,9 @@ func PeerRxBytes(peer *device.Peer) uint64 {
func PeerTxBytes(peer *device.Peer) uint64 {
return (*atomic.Uint64)(unsafe.Add(unsafe.Pointer(peer), offTxBytes)).Load()
}
// PeerHandshakeAttempts returns the number of WireGuard handshake attempts
// made for the current handshake. It resets to zero before every new handshake.
func PeerHandshakeAttempts(peer *device.Peer) uint32 {
return (*atomic.Uint32)(unsafe.Add(unsafe.Pointer(peer), offHandshakeAttempts)).Load()
}