From cda9f6e73a3bf3af09481bf31d57cf35c8b342de Mon Sep 17 00:00:00 2001 From: Josh Bleecher Snyder Date: Tue, 26 Jan 2021 10:54:59 -0800 Subject: device: add PeerStats Per-peer statistics are the sort of thing clients refresh very frequently, for UIs. They are available via UAPI, but the serialization and deserialization is quite heavy for frequent use. This commit adds a very cheap, direct means of retrieving peer stats. Signed-off-by: Josh Bleecher Snyder --- device/stats.go | 39 +++++++++++++++++++++++++++++++++++++++ 1 file changed, 39 insertions(+) create mode 100644 device/stats.go diff --git a/device/stats.go b/device/stats.go new file mode 100644 index 0000000..6ccc501 --- /dev/null +++ b/device/stats.go @@ -0,0 +1,39 @@ +/* SPDX-License-Identifier: MIT + * + * Copyright (C) 2017-2020 WireGuard LLC. All Rights Reserved. + */ + +package device + +import ( + "sync/atomic" + "time" +) + +// PeerStats are connection statistics for a given Peer. +type PeerStats struct { + RxBytes uint64 + TxBytes uint64 + LastHandshakeInitiated time.Time +} + +// PeerStats returns statistics for the peer with public key pk, +// and reports whether the peer lookup succeeded. +func (device *Device) PeerStats(pk NoisePublicKey) (stats PeerStats, ok bool) { + device.peers.RLock() + peer := device.peers.keyMap[pk] + device.peers.RUnlock() + + if peer == nil { + return PeerStats{}, false + } + + peer.RLock() + defer peer.RUnlock() + stats = PeerStats{ + RxBytes: atomic.LoadUint64(&peer.stats.rxBytes), + TxBytes: atomic.LoadUint64(&peer.stats.txBytes), + LastHandshakeInitiated: time.Unix(0, atomic.LoadInt64(&peer.stats.lastHandshakeNano)), + } + return stats, true +} -- cgit v1.2.3-59-g8ed1b