ipn: add c2n endpoint for sockstats logs

Signed-off-by: Will Norris <will@tailscale.com>
This commit is contained in:
Will Norris
2023-03-10 14:33:26 -08:00
committed by Will Norris
parent a1d9f65354
commit 09e0ccf4c2
3 changed files with 33 additions and 3 deletions

View File

@@ -6,12 +6,14 @@ package sockstatlog
import (
"encoding/json"
"io"
"os"
"path/filepath"
"time"
"tailscale.com/logtail/filch"
"tailscale.com/net/sockstats"
"tailscale.com/types/logger"
"tailscale.com/util/mak"
)
@@ -21,6 +23,7 @@ const pollPeriod = time.Second / 10
// Logger logs statistics about network sockets.
type Logger struct {
ticker time.Ticker
logf logger.Logf
logbuffer *filch.Filch
}
@@ -46,7 +49,7 @@ type event struct {
// NewLogger returns a new Logger that will store stats in logdir.
// On platforms that do not support sockstat logging, a nil Logger will be returned.
// The returned Logger is not yet running.
func NewLogger(logdir string) (*Logger, error) {
func NewLogger(logdir string, logf logger.Logf) (*Logger, error) {
if !sockstats.IsAvailable {
return nil, nil
}
@@ -62,6 +65,7 @@ func NewLogger(logdir string) (*Logger, error) {
return &Logger{
ticker: *time.NewTicker(pollPeriod),
logf: logf,
logbuffer: filch,
}, nil
}
@@ -92,7 +96,9 @@ func (l *Logger) poll() {
if stats.CurrentInterfaceCellular {
e.IsCellularInterface = 1
}
enc.Encode(e)
if err := enc.Encode(e); err != nil {
l.logf("sockstatlog: error encoding log: %v", err)
}
}
}
lastTime = t
@@ -105,6 +111,27 @@ func (l *Logger) Shutdown() {
l.logbuffer.Close()
}
// WriteLogs reads local logs, combining logs into events, and writes them to w.
// Logs within eventWindow are combined into the same event.
func (l *Logger) WriteLogs(w io.Writer) {
if l == nil || l.logbuffer == nil {
return
}
for {
b, err := l.logbuffer.TryReadLine()
if err != nil {
l.logf("sockstatlog: error reading log: %v", err)
return
}
if b == nil {
// no more log messages
return
}
w.Write(b)
}
}
// delta calculates the delta stats between two SockStats snapshots.
// b is assumed to have occurred after a.
// Zero values are omitted from the returned map, and an empty map is returned if no bytes were transferred.