From cda9f6e73a3bf3af09481bf31d57cf35c8b342de Mon Sep 17 00:00:00 2001
From: Josh Bleecher Snyder <josh@tailscale.com>
Date: Tue, 26 Jan 2021 10:54:59 -0800
Subject: [PATCH] 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 <josh@tailscale.com>
---
 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
+}