From 5c0636eb3d24f6f26a27387f9863f9c9d8e5eded Mon Sep 17 00:00:00 2001
From: Arceliar <Arceliar@users.noreply.github.com>
Date: Fri, 18 May 2018 20:41:02 -0500
Subject: [PATCH] add uptime and bytes sent/recvd to peer struct and getPeers

---
 src/yggdrasil/admin.go |  4 ++++
 src/yggdrasil/peer.go  | 25 +++++++++++++++++--------
 2 files changed, 21 insertions(+), 8 deletions(-)

diff --git a/src/yggdrasil/admin.go b/src/yggdrasil/admin.go
index 75561bd..8e6ee91 100644
--- a/src/yggdrasil/admin.go
+++ b/src/yggdrasil/admin.go
@@ -10,6 +10,7 @@ import "net/url"
 import "sort"
 import "strings"
 import "strconv"
+import "sync/atomic"
 import "time"
 
 // TODO? Make all of this JSON
@@ -322,6 +323,9 @@ func (a *admin) getData_getPeers() []admin_nodeInfo {
 		info := admin_nodeInfo{
 			{"IP", net.IP(addr[:]).String()},
 			{"port", fmt.Sprint(port)},
+			{"uptime", fmt.Sprint(time.Since(p.firstSeen))},
+			{"bytesSent", fmt.Sprint(atomic.LoadUint64(&p.bytesSent))},
+			{"bytesRecvd", fmt.Sprint(atomic.LoadUint64(&p.bytesRecvd))},
 		}
 		peerInfos = append(peerInfos, info)
 	}
diff --git a/src/yggdrasil/peer.go b/src/yggdrasil/peer.go
index bec9135..9325c33 100644
--- a/src/yggdrasil/peer.go
+++ b/src/yggdrasil/peer.go
@@ -86,11 +86,14 @@ func (ps *peers) putPorts(ports map[switchPort]*peer) {
 type peer struct {
 	// Rolling approximation of bandwidth, in bps, used by switch, updated by packet sends
 	// use get/update methods only! (atomic accessors as float64)
-	bandwidth uint64
+	bandwidth  uint64
+	bytesSent  uint64 // To track bandwidth usage for getPeers
+	bytesRecvd uint64 // To track bandwidth usage for getPeers
 	// BUG: sync/atomic, 32 bit platforms need the above to be the first element
-	box    boxPubKey
-	sig    sigPubKey
-	shared boxSharedKey
+	firstSeen time.Time // To track uptime for getPeers
+	box       boxPubKey
+	sig       sigPubKey
+	shared    boxSharedKey
 	//in <-chan []byte
 	//out chan<- []byte
 	//in func([]byte)
@@ -133,11 +136,13 @@ func (p *peer) updateBandwidth(bytes int, duration time.Duration) {
 
 func (ps *peers) newPeer(box *boxPubKey,
 	sig *sigPubKey) *peer {
+	now := time.Now()
 	p := peer{box: *box,
-		sig:     *sig,
-		shared:  *getSharedKey(&ps.core.boxPriv, box),
-		lastAnc: time.Now(),
-		core:    ps.core}
+		sig:       *sig,
+		shared:    *getSharedKey(&ps.core.boxPriv, box),
+		lastAnc:   now,
+		firstSeen: now,
+		core:      ps.core}
 	ps.mutex.Lock()
 	defer ps.mutex.Unlock()
 	oldPorts := ps.getPorts()
@@ -221,6 +226,8 @@ func (p *peer) linkLoop(in <-chan []byte) {
 }
 
 func (p *peer) handlePacket(packet []byte, linkIn chan<- []byte) {
+	// TODO See comment in sendPacket about atomics technically being done wrong
+	atomic.AddUint64(&p.bytesRecvd, uint64(len(packet)))
 	pType, pTypeLen := wire_decode_uint64(packet)
 	if pTypeLen == 0 {
 		return
@@ -277,6 +284,8 @@ func (p *peer) sendPacket(packet []byte) {
 	// Is there ever a case where something more complicated is needed?
 	// What if p.out blocks?
 	p.out(packet)
+	// TODO this should really happen at the interface, to account for LIFO packet drops and additional per-packet/per-message overhead, but this should be pretty close... better to move it to the tcp/udp stuff *after* rewriting both to give a common interface
+	atomic.AddUint64(&p.bytesSent, uint64(len(packet)))
 }
 
 func (p *peer) sendLinkPacket(packet []byte) {