p2.5: log host=- for permit_restricted callers

Scrape, fetch, refused, fetch-failed, and internal lines redact the
host for callers in context.permit_restricted. final_url stays in the
response. Test captures slog records, not the journal.
This commit is contained in:
2026-09-23 11:40:45 -07:00
parent e36da19f57
commit 85113f9c0b
4 changed files with 165 additions and 3 deletions
+136
View File
@@ -2,6 +2,7 @@ package api
import (
"bytes"
"context"
"crypto/tls"
"encoding/json"
"io"
@@ -12,6 +13,7 @@ import (
"os"
"path/filepath"
"strings"
"sync"
"testing"
"time"
@@ -262,3 +264,137 @@ func TestHTTPServerWriteTimeout(t *testing.T) {
t.Fatalf("WriteTimeout %s", hs.WriteTimeout)
}
}
type recHandler struct {
mu sync.Mutex
recs []slog.Record
}
func (h *recHandler) Enabled(context.Context, slog.Level) bool { return true }
func (h *recHandler) Handle(_ context.Context, r slog.Record) error {
h.mu.Lock()
h.recs = append(h.recs, r.Clone())
h.mu.Unlock()
return nil
}
func (h *recHandler) WithAttrs([]slog.Attr) slog.Handler { return h }
func (h *recHandler) WithGroup(string) slog.Handler { return h }
func recHost(r slog.Record) string {
var host string
r.Attrs(func(a slog.Attr) bool {
if a.Key == "host" {
host = fmtString(a.Value)
}
return true
})
return host
}
func recMsg(r slog.Record) string { return r.Message }
func fmtString(v slog.Value) string {
if v.Kind() == slog.KindString {
return v.String()
}
return v.Any().(string)
}
func TestRestrictedCallerLogsBlankHost(t *testing.T) {
dir := t.TempDir()
keys := filepath.Join(dir, "keys.toml")
fleetOut := filepath.Join(dir, "fleet")
gsOut := filepath.Join(dir, "gs")
if err := auth.Add(keys, "test", "hermes-fleet", fleetOut); err != nil {
t.Fatal(err)
}
if err := auth.Add(keys, "straylight-gs", "guardian-star", gsOut); err != nil {
t.Fatal(err)
}
fleetKey, err := os.ReadFile(fleetOut)
if err != nil {
t.Fatal(err)
}
gsKey, err := os.ReadFile(gsOut)
if err != nil {
t.Fatal(err)
}
store, err := auth.Load(keys)
if err != nil {
t.Fatal(err)
}
g := &guard.Guard{AllowPrefixes: []netip.Prefix{netip.MustParsePrefix("127.0.0.0/8")}}
h := &recHandler{}
s := New("test", g, &fetch.Fetcher{Guard: g, UA: "sift-test"}, store, slog.New(h))
s.PermitRestricted = []string{"straylight-gs"}
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Content-Type", "text/html; charset=utf-8")
_, _ = w.Write([]byte(`<!doctype html><html><head><title>Hi</title></head><body><article><p>ok</p></article></body></html>`))
}))
defer ts.Close()
rec := postJSON(t, s, "/v1/fetch", strings.TrimSpace(string(gsKey)), map[string]any{"url": ts.URL, "mode": "clean"})
if rec.Code != 200 {
t.Fatalf("gs fetch: %d %s", rec.Code, rec.Body.String())
}
var env map[string]any
if err := json.Unmarshal(rec.Body.Bytes(), &env); err != nil {
t.Fatal(err)
}
src, _ := env["source"].(map[string]any)
final, _ := src["final_url"].(string)
if !strings.Contains(final, hostOf(ts.URL)) {
t.Fatalf("final_url still in body, got %q", final)
}
rec = postJSON(t, s, "/v1/fetch", strings.TrimSpace(string(gsKey)), map[string]any{"url": "http://10.0.0.5/", "mode": "clean"})
if rec.Code != 403 {
t.Fatalf("gs refused: %d %s", rec.Code, rec.Body.String())
}
rec = postJSON(t, s, "/v1/fetch", strings.TrimSpace(string(fleetKey)), map[string]any{"url": ts.URL, "mode": "clean"})
if rec.Code != 200 {
t.Fatalf("fleet fetch: %d %s", rec.Code, rec.Body.String())
}
var gsFetch, gsRefused, fleetFetch string
h.mu.Lock()
for _, r := range h.recs {
switch recMsg(r) {
case "fetch":
if attrCaller(r) == "straylight-gs" {
gsFetch = recHost(r)
}
if attrCaller(r) == "test" {
fleetFetch = recHost(r)
}
case "refused":
if attrCaller(r) == "straylight-gs" {
gsRefused = recHost(r)
}
}
}
h.mu.Unlock()
if gsFetch != "-" {
t.Fatalf("gs fetch host=%q want -", gsFetch)
}
if gsRefused != "-" {
t.Fatalf("gs refused host=%q want -", gsRefused)
}
wantHost := hostOf(ts.URL)
if fleetFetch != wantHost {
t.Fatalf("fleet fetch host=%q want %q", fleetFetch, wantHost)
}
}
func attrCaller(r slog.Record) string {
var name string
r.Attrs(func(a slog.Attr) bool {
if a.Key == "caller" {
name = a.Value.String()
}
return true
})
return name
}