Compare commits
3 Commits
04f370a050
...
31228608d1
| Author | SHA1 | Date | |
|---|---|---|---|
| 31228608d1 | |||
| a13e5b7ded | |||
| bb30b3ad64 |
37
README.md
37
README.md
@@ -107,14 +107,26 @@ TTY detection, and security headers are always applied.
|
|||||||
| `DATA_DIR` | Directory for all SQLite databases | `/var/lib/webhooker` |
|
| `DATA_DIR` | Directory for all SQLite databases | `/var/lib/webhooker` |
|
||||||
| `DEBUG` | Enable debug logging | `false` |
|
| `DEBUG` | Enable debug logging | `false` |
|
||||||
| `MAINTENANCE_MODE` | Report `maintenanceMode: true` in the healthcheck JSON. It does not change how any request is served — no maintenance page exists | `false` |
|
| `MAINTENANCE_MODE` | Report `maintenanceMode: true` in the healthcheck JSON. It does not change how any request is served — no maintenance page exists | `false` |
|
||||||
| `METRICS_USERNAME` | Basic auth username for `/metrics` | `""` |
|
| `METRICS_USERNAME` | Basic auth username for `/metrics`. Must be set together with `METRICS_PASSWORD`; one without the other fails startup | `""` |
|
||||||
| `METRICS_PASSWORD` | Basic auth password for `/metrics` | `""` |
|
| `METRICS_PASSWORD` | Basic auth password for `/metrics`. Must be set together with `METRICS_USERNAME`; one without the other fails startup | `""` |
|
||||||
| `SENTRY_DSN` | Sentry error reporting DSN | `""` |
|
| `SENTRY_DSN` | Sentry error reporting DSN | `""` |
|
||||||
| `RETENTION_SWEEP_INTERVAL` | How often the retention reaper and archive sweeper run (Go duration, must be positive) | `1h` |
|
| `RETENTION_SWEEP_INTERVAL` | How often the retention reaper and archive sweeper run (Go duration, must be positive) | `1h` |
|
||||||
| `SESSION_IDLE_TIMEOUT` | Idle session timeout (Go duration) | `24h` |
|
| `SESSION_IDLE_TIMEOUT` | Idle session timeout (Go duration) | `24h` |
|
||||||
| `RECEIVER_RATE_LIMIT` | Receiver requests/minute per IP per entrypoint (10x that per IP across the route) | `120` |
|
| `RECEIVER_RATE_LIMIT` | Receiver requests/minute per IP per entrypoint (10x that per IP across the route) | `120` |
|
||||||
| `TRUSTED_PROXIES` | CIDRs whose forwarded headers are trusted (unset: all clients behind a proxy share one rate-limit bucket; a correct login password is never throttled either way) | `""` (none) |
|
| `TRUSTED_PROXIES` | CIDRs whose forwarded headers are trusted (unset: all clients behind a proxy share one rate-limit bucket; a correct login password is never throttled either way) | `""` (none) |
|
||||||
|
|
||||||
|
#### Metrics credentials
|
||||||
|
|
||||||
|
`METRICS_USERNAME` and `METRICS_PASSWORD` are set together or not at
|
||||||
|
all. With both set, `/metrics` is served behind basic auth. With
|
||||||
|
neither set, the route is not registered and returns 404. With one set
|
||||||
|
and the other empty or unset, the process refuses to start and exits
|
||||||
|
non-zero with an error naming both variables — mounting the endpoint
|
||||||
|
on the username alone would publish it behind a password that is the
|
||||||
|
empty string, and quietly withholding it would deny an endpoint that
|
||||||
|
was asked for. The `hasMetricsAuth` field in the startup log and the
|
||||||
|
existence of the route are the same value, so they cannot disagree.
|
||||||
|
|
||||||
#### Trusted proxies
|
#### Trusted proxies
|
||||||
|
|
||||||
`TRUSTED_PROXIES` is a comma-separated list of CIDR blocks (a bare
|
`TRUSTED_PROXIES` is a comma-separated list of CIDR blocks (a bare
|
||||||
@@ -269,6 +281,10 @@ What it does **not** put in the log:
|
|||||||
`internal/gormlog/firstboot_test.go` boots the real graph with
|
`internal/gormlog/firstboot_test.go` boots the real graph with
|
||||||
`DEBUG=true` against an empty `DATA_DIR` and asserts that neither
|
`DEBUG=true` against an empty `DATA_DIR` and asserts that neither
|
||||||
secret appears in what that boot wrote to stdout.
|
secret appears in what that boot wrote to stdout.
|
||||||
|
The one exception is `(*gorm.DB).Scan`, which GORM logs through its
|
||||||
|
own trace recorder rather than through this filter. No production
|
||||||
|
code path calls it, and `internal/gormlog/scan_guard_test.go` fails
|
||||||
|
if a non-test file adds one.
|
||||||
- **Session cookies, API keys or target credentials.** None of these is
|
- **Session cookies, API keys or target credentials.** None of these is
|
||||||
logged at any level.
|
logged at any level.
|
||||||
|
|
||||||
@@ -1458,8 +1474,11 @@ on all three arms of `Trace`, including the routine one an operator
|
|||||||
reaches at `DEBUG`, which is the only level at which a successful
|
reaches at `DEBUG`, which is the only level at which a successful
|
||||||
`INSERT` is written at all. One GORM path does not consult the filter —
|
`INSERT` is written at all. One GORM path does not consult the filter —
|
||||||
`(*gorm.DB).Scan`, which records the statement through GORM's own trace
|
`(*gorm.DB).Scan`, which records the statement through GORM's own trace
|
||||||
recorder — and nothing in this service calls it; `Pluck`, `Row` and
|
recorder. No production code path calls it; its one caller is
|
||||||
`Raw` all run through the normal callback processor and are filtered.
|
`internal/database/database_test.go:91`, whose `SELECT 1` binds
|
||||||
|
nothing, and `internal/gormlog/scan_guard_test.go` fails if a non-test
|
||||||
|
file calls it. `Pluck`, `Row` and `Raw` all run through the normal
|
||||||
|
callback processor and are filtered.
|
||||||
See `#### What DEBUG=true exposes` under Configuration.
|
See `#### What DEBUG=true exposes` under Configuration.
|
||||||
|
|
||||||
What that ceiling does **not** cover, stated here so the figure is not
|
What that ceiling does **not** cover, stated here so the figure is not
|
||||||
@@ -1738,7 +1757,7 @@ abuse limit later; they are tracked as future work.
|
|||||||
|
|
||||||
| Method | Path | Description |
|
| Method | Path | Description |
|
||||||
| ------ | ---------- | ----------- |
|
| ------ | ---------- | ----------- |
|
||||||
| `GET` | `/metrics` | Prometheus metrics, behind basic auth. The route is registered only when `METRICS_USERNAME` is set; otherwise it does not exist and returns 404 |
|
| `GET` | `/metrics` | Prometheus metrics, behind basic auth. The route is registered only when `METRICS_USERNAME` and `METRICS_PASSWORD` are both set; with neither set it does not exist and returns 404, and with only one set the process refuses to start |
|
||||||
|
|
||||||
#### API (Planned)
|
#### API (Planned)
|
||||||
|
|
||||||
@@ -1893,7 +1912,8 @@ Applied to all routes in this order:
|
|||||||
Permissions-Policy)
|
Permissions-Policy)
|
||||||
3. **Logging** — Structured request logging (method, URL, status,
|
3. **Logging** — Structured request logging (method, URL, status,
|
||||||
latency, remote IP, user agent, request ID)
|
latency, remote IP, user agent, request ID)
|
||||||
4. **Metrics** — Prometheus HTTP metrics (if `METRICS_USERNAME` is set)
|
4. **Metrics** — Prometheus HTTP metrics (if `METRICS_USERNAME` and
|
||||||
|
`METRICS_PASSWORD` are both set)
|
||||||
5. **CORS** — Cross-origin resource sharing headers
|
5. **CORS** — Cross-origin resource sharing headers
|
||||||
6. **Timeout** — 60-second request timeout
|
6. **Timeout** — 60-second request timeout
|
||||||
7. **Recoverer** — Panic recovery: one `ERROR` record through
|
7. **Recoverer** — Panic recovery: one `ERROR` record through
|
||||||
@@ -1925,8 +1945,9 @@ being read and without reaching CSRF, the route group's remaining
|
|||||||
middleware, or the handler. It is not rejected before *any* other
|
middleware, or the handler. It is not rejected before *any* other
|
||||||
middleware, though: the global entries listed above all run first, so
|
middleware, though: the global entries listed above all run first, so
|
||||||
such a request is still logged and given the security headers — and
|
such a request is still logged and given the security headers — and
|
||||||
counted in the metrics, on a deployment where `METRICS_USERNAME` is
|
counted in the metrics, on a deployment where the `/metrics`
|
||||||
set and the Metrics middleware is therefore registered at all. The
|
credentials are set and the Metrics middleware is therefore registered
|
||||||
|
at all. The
|
||||||
rejection itself is logged at `WARN` with the method, path and
|
rejection itself is logged at `WARN` with the method, path and
|
||||||
declared length. A chunked request, or
|
declared length. A chunked request, or
|
||||||
one that lies about its length, is hard-capped by
|
one that lies about its length, is hard-capped by
|
||||||
|
|||||||
@@ -71,6 +71,16 @@ var ErrInvalidPort = errors.New("invalid port")
|
|||||||
// nor a bare IP address.
|
// nor a bare IP address.
|
||||||
var ErrInvalidCIDR = errors.New("invalid CIDR")
|
var ErrInvalidCIDR = errors.New("invalid CIDR")
|
||||||
|
|
||||||
|
// ErrIncompleteMetricsAuth is returned when exactly one of
|
||||||
|
// METRICS_USERNAME and METRICS_PASSWORD carries a value. Neither
|
||||||
|
// fallback is acceptable: serving /metrics on the username alone
|
||||||
|
// publishes an endpoint whose password is the empty string, and
|
||||||
|
// silently leaving it unmounted withholds an endpoint the operator
|
||||||
|
// asked for. Half-set is a configuration error, so startup fails.
|
||||||
|
var ErrIncompleteMetricsAuth = errors.New(
|
||||||
|
"incomplete metrics credentials",
|
||||||
|
)
|
||||||
|
|
||||||
//nolint:revive // ConfigParams is a standard fx naming convention.
|
//nolint:revive // ConfigParams is a standard fx naming convention.
|
||||||
type ConfigParams struct {
|
type ConfigParams struct {
|
||||||
fx.In
|
fx.In
|
||||||
@@ -128,6 +138,21 @@ func (c *Config) IsProd() bool {
|
|||||||
return c.Environment == EnvironmentProd
|
return c.Environment == EnvironmentProd
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// MetricsAuthEnabled reports whether /metrics is served behind basic
|
||||||
|
// auth. It is the only answer to that question in the codebase: the
|
||||||
|
// route mount, the Prometheus recording middleware and the startup
|
||||||
|
// log's hasMetricsAuth field all read this one method, so the log
|
||||||
|
// cannot report auth as off while the route is mounted.
|
||||||
|
//
|
||||||
|
// It requires both credentials rather than the username alone.
|
||||||
|
// loadFromEnv already rejects a half-set pair, but a Config built in
|
||||||
|
// code bypasses that, and the failure mode this guards is an endpoint
|
||||||
|
// mounted with a credential map whose only password is the empty
|
||||||
|
// string.
|
||||||
|
func (c *Config) MetricsAuthEnabled() bool {
|
||||||
|
return c.MetricsUsername != "" && c.MetricsPassword != ""
|
||||||
|
}
|
||||||
|
|
||||||
// envString returns the value of the named environment variable,
|
// envString returns the value of the named environment variable,
|
||||||
// or an empty string if not set.
|
// or an empty string if not set.
|
||||||
func envString(key string) string {
|
func envString(key string) string {
|
||||||
@@ -329,6 +354,30 @@ func envPrefixList(key string) ([]netip.Prefix, error) {
|
|||||||
return prefixes, nil
|
return prefixes, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// resolveMetricsAuth reads the /metrics basic-auth credentials and
|
||||||
|
// rejects a half-set pair, naming both variables either way. The
|
||||||
|
// error carries neither value: the password is a secret.
|
||||||
|
func resolveMetricsAuth() (string, string, error) {
|
||||||
|
username := envString("METRICS_USERNAME")
|
||||||
|
password := envString("METRICS_PASSWORD")
|
||||||
|
|
||||||
|
if (username == "") == (password == "") {
|
||||||
|
return username, password, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
set, empty := "METRICS_USERNAME", "METRICS_PASSWORD"
|
||||||
|
if username == "" {
|
||||||
|
set, empty = empty, set
|
||||||
|
}
|
||||||
|
|
||||||
|
return "", "", fmt.Errorf(
|
||||||
|
"%w: %s is set but %s is empty; METRICS_USERNAME and "+
|
||||||
|
"METRICS_PASSWORD must both be set to serve /metrics, "+
|
||||||
|
"or both be empty to leave it unmounted",
|
||||||
|
ErrIncompleteMetricsAuth, set, empty,
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
// resolveEnvironment reads WEBHOOKER_ENVIRONMENT, defaulting to
|
// resolveEnvironment reads WEBHOOKER_ENVIRONMENT, defaulting to
|
||||||
// dev, and rejects unrecognised values.
|
// dev, and rejects unrecognised values.
|
||||||
func resolveEnvironment() (string, error) {
|
func resolveEnvironment() (string, error) {
|
||||||
@@ -406,13 +455,18 @@ func loadFromEnv() (*Config, error) {
|
|||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
metricsUsername, metricsPassword, err := resolveMetricsAuth()
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
return &Config{
|
return &Config{
|
||||||
DataDir: envString("DATA_DIR"),
|
DataDir: envString("DATA_DIR"),
|
||||||
Debug: debug,
|
Debug: debug,
|
||||||
MaintenanceMode: maintenanceMode,
|
MaintenanceMode: maintenanceMode,
|
||||||
Environment: environment,
|
Environment: environment,
|
||||||
MetricsUsername: envString("METRICS_USERNAME"),
|
MetricsUsername: metricsUsername,
|
||||||
MetricsPassword: envString("METRICS_PASSWORD"),
|
MetricsPassword: metricsPassword,
|
||||||
Port: port,
|
Port: port,
|
||||||
SentryDSN: envString("SENTRY_DSN"),
|
SentryDSN: envString("SENTRY_DSN"),
|
||||||
RetentionSweepInterval: retentionSweepInterval,
|
RetentionSweepInterval: retentionSweepInterval,
|
||||||
@@ -512,8 +566,7 @@ func New(lc fx.Lifecycle, params ConfigParams) (*Config, error) {
|
|||||||
"receiverRateLimit", s.ReceiverRateLimit,
|
"receiverRateLimit", s.ReceiverRateLimit,
|
||||||
"trustedProxies", len(s.TrustedProxies),
|
"trustedProxies", len(s.TrustedProxies),
|
||||||
"hasSentryDSN", s.SentryDSN != "",
|
"hasSentryDSN", s.SentryDSN != "",
|
||||||
"hasMetricsAuth",
|
"hasMetricsAuth", s.MetricsAuthEnabled(),
|
||||||
s.MetricsUsername != "" && s.MetricsPassword != "",
|
|
||||||
)
|
)
|
||||||
|
|
||||||
s.warnSharedRateLimitBucket(log)
|
s.warnSharedRateLimitBucket(log)
|
||||||
|
|||||||
@@ -26,6 +26,12 @@ const (
|
|||||||
// cidrPrivateV4 is the sample trusted-proxy block the
|
// cidrPrivateV4 is the sample trusted-proxy block the
|
||||||
// TRUSTED_PROXIES cases are built from.
|
// TRUSTED_PROXIES cases are built from.
|
||||||
cidrPrivateV4 = "10.0.0.0/8"
|
cidrPrivateV4 = "10.0.0.0/8"
|
||||||
|
|
||||||
|
// metricsAuthValue is the sample METRICS_PASSWORD the metrics
|
||||||
|
// credential cases are built from. It is asserted absent from
|
||||||
|
// the startup error, so it must not be a substring of either
|
||||||
|
// variable name that error prints.
|
||||||
|
metricsAuthValue = "s3cret"
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestEnvironmentConfig(t *testing.T) {
|
func TestEnvironmentConfig(t *testing.T) {
|
||||||
@@ -726,3 +732,168 @@ func TestSharedRateLimitBucketWarning(t *testing.T) {
|
|||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// metricsEnv describes what one subtest below puts in the
|
||||||
|
// environment for a single METRICS_ variable. A variable that is
|
||||||
|
// set to the empty string and one that is not set at all are
|
||||||
|
// distinct inputs here, because the reported bug arrived through
|
||||||
|
// the first of them.
|
||||||
|
type metricsEnv struct {
|
||||||
|
set bool
|
||||||
|
value string
|
||||||
|
}
|
||||||
|
|
||||||
|
// unset leaves the variable out of the environment entirely.
|
||||||
|
func unset() metricsEnv {
|
||||||
|
return metricsEnv{set: false, value: ""}
|
||||||
|
}
|
||||||
|
|
||||||
|
// setTo sets the variable, including to the empty string.
|
||||||
|
func setTo(value string) metricsEnv {
|
||||||
|
return metricsEnv{set: true, value: value}
|
||||||
|
}
|
||||||
|
|
||||||
|
// metricsAuthCase is one row of the table in TestMetricsAuthConfig,
|
||||||
|
// named so the table can live in its own function and keep the test
|
||||||
|
// itself short.
|
||||||
|
type metricsAuthCase struct {
|
||||||
|
name string
|
||||||
|
username metricsEnv
|
||||||
|
password metricsEnv
|
||||||
|
expectError bool
|
||||||
|
expectAuth bool
|
||||||
|
}
|
||||||
|
|
||||||
|
// metricsAuthCases enumerates every combination of the two
|
||||||
|
// credentials, counting "set to the empty string" and "not set at
|
||||||
|
// all" as separate inputs on each side.
|
||||||
|
func metricsAuthCases() []metricsAuthCase {
|
||||||
|
return []metricsAuthCase{
|
||||||
|
{
|
||||||
|
name: "both unset leaves metrics unmounted",
|
||||||
|
username: unset(),
|
||||||
|
password: unset(),
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "both empty leaves metrics unmounted",
|
||||||
|
username: setTo(""),
|
||||||
|
password: setTo(""),
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "both set enables metrics auth",
|
||||||
|
username: setTo("metrics"),
|
||||||
|
password: setTo(metricsAuthValue),
|
||||||
|
expectAuth: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "username with unset password fails",
|
||||||
|
username: setTo("metrics"),
|
||||||
|
password: unset(),
|
||||||
|
expectError: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "username with empty password fails",
|
||||||
|
username: setTo("metrics"),
|
||||||
|
password: setTo(""),
|
||||||
|
expectError: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "password with unset username fails",
|
||||||
|
username: unset(),
|
||||||
|
password: setTo(metricsAuthValue),
|
||||||
|
expectError: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "password with empty username fails",
|
||||||
|
username: setTo(""),
|
||||||
|
password: setTo(metricsAuthValue),
|
||||||
|
expectError: true,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestMetricsAuthConfig covers every combination of METRICS_USERNAME
|
||||||
|
// and METRICS_PASSWORD. Either both carry a value, in which case
|
||||||
|
// /metrics is served behind basic auth, or neither does, in which
|
||||||
|
// case the route is never mounted. One without the other is a
|
||||||
|
// startup error rather than a fallback: mounting on the username
|
||||||
|
// alone published /metrics behind a credential map that accepted an
|
||||||
|
// empty password, which is the defect this test exists to pin. See
|
||||||
|
// https://git.eeqj.de/sneak/webhooker/issues/205.
|
||||||
|
func TestMetricsAuthConfig(t *testing.T) {
|
||||||
|
for _, tt := range metricsAuthCases() {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
// Cannot use t.Parallel() here because t.Setenv
|
||||||
|
// is incompatible with parallel subtests.
|
||||||
|
if tt.username.set {
|
||||||
|
t.Setenv("METRICS_USERNAME", tt.username.value)
|
||||||
|
} else {
|
||||||
|
require.NoError(
|
||||||
|
t, os.Unsetenv("METRICS_USERNAME"),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
if tt.password.set {
|
||||||
|
t.Setenv("METRICS_PASSWORD", tt.password.value)
|
||||||
|
} else {
|
||||||
|
require.NoError(
|
||||||
|
t, os.Unsetenv("METRICS_PASSWORD"),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
if tt.expectError {
|
||||||
|
assertMetricsAuthRejected(t)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
assertMetricsAuthAccepted(t, tt.expectAuth)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// assertMetricsAuthRejected requires that fx refused to build the
|
||||||
|
// graph, that the failure is ErrIncompleteMetricsAuth, and that the
|
||||||
|
// operator is told both variable names — the point of failing here
|
||||||
|
// rather than degrading is that the message says what to fix.
|
||||||
|
func assertMetricsAuthRejected(t *testing.T) {
|
||||||
|
t.Helper()
|
||||||
|
|
||||||
|
var cfg *config.Config
|
||||||
|
|
||||||
|
app := fx.New(
|
||||||
|
fx.NopLogger,
|
||||||
|
fx.Provide(globals.New, logger.New, config.New),
|
||||||
|
fx.Populate(&cfg),
|
||||||
|
)
|
||||||
|
|
||||||
|
err := app.Err()
|
||||||
|
require.Error(t, err)
|
||||||
|
require.ErrorIs(t, err, config.ErrIncompleteMetricsAuth)
|
||||||
|
assert.Contains(t, err.Error(), "METRICS_USERNAME")
|
||||||
|
assert.Contains(t, err.Error(), "METRICS_PASSWORD")
|
||||||
|
// The password is a secret and must not reach a startup error.
|
||||||
|
assert.NotContains(t, err.Error(), metricsAuthValue)
|
||||||
|
}
|
||||||
|
|
||||||
|
// assertMetricsAuthAccepted requires that startup succeeded and that
|
||||||
|
// MetricsAuthEnabled — the single value the /metrics mount and the
|
||||||
|
// startup log both read — reports what the environment asked for.
|
||||||
|
func assertMetricsAuthAccepted(t *testing.T, expectAuth bool) {
|
||||||
|
t.Helper()
|
||||||
|
|
||||||
|
var cfg *config.Config
|
||||||
|
|
||||||
|
app := fxtest.New(
|
||||||
|
t,
|
||||||
|
fx.Provide(globals.New, logger.New, config.New),
|
||||||
|
fx.Populate(&cfg),
|
||||||
|
)
|
||||||
|
require.NoError(t, app.Err())
|
||||||
|
|
||||||
|
app.RequireStart()
|
||||||
|
|
||||||
|
defer app.RequireStop()
|
||||||
|
|
||||||
|
assert.Equal(t, expectAuth, cfg.MetricsAuthEnabled())
|
||||||
|
}
|
||||||
|
|||||||
@@ -111,9 +111,11 @@ func (l *Logger) LogMode(gormlogger.LogLevel) gormlogger.Interface {
|
|||||||
//
|
//
|
||||||
// One GORM path does not consult this: (*gorm.DB).Scan records the
|
// One GORM path does not consult this: (*gorm.DB).Scan records the
|
||||||
// statement through gorm's own traceRecorder, which does not implement
|
// statement through gorm's own traceRecorder, which does not implement
|
||||||
// this interface. Nothing in this service calls it. (*gorm.DB).Pluck,
|
// this interface. No production code path calls it; its one caller is
|
||||||
// Row and Raw all run through the normal callback processor and are
|
// internal/database/database_test.go:91, whose SELECT 1 binds nothing.
|
||||||
// filtered.
|
// scan_guard_test.go fails if a non-test file calls it.
|
||||||
|
// (*gorm.DB).Pluck, Row and Raw all run through the normal callback
|
||||||
|
// processor and are filtered.
|
||||||
func (l *Logger) ParamsFilter(
|
func (l *Logger) ParamsFilter(
|
||||||
_ context.Context, sql string, _ ...any,
|
_ context.Context, sql string, _ ...any,
|
||||||
) (string, []any) {
|
) (string, []any) {
|
||||||
|
|||||||
260
internal/gormlog/scan_guard_test.go
Normal file
260
internal/gormlog/scan_guard_test.go
Normal file
@@ -0,0 +1,260 @@
|
|||||||
|
package gormlog_test
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"go/ast"
|
||||||
|
"go/parser"
|
||||||
|
"go/token"
|
||||||
|
"io/fs"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"strings"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/stretchr/testify/require"
|
||||||
|
)
|
||||||
|
|
||||||
|
// minNonTestFiles guards the walk below against passing because it
|
||||||
|
// found nothing to look at. The tree held 60 non-test .go files when
|
||||||
|
// this was written.
|
||||||
|
const minNonTestFiles = 40
|
||||||
|
|
||||||
|
// isRowProducer reports whether name is a method that returns a
|
||||||
|
// database/sql row handle. GORM's Row and Rows return *sql.Row and
|
||||||
|
// *sql.Rows, so Scan on the result of one of them is database/sql's
|
||||||
|
// Scan and never (*gorm.DB).Scan.
|
||||||
|
func isRowProducer(name string) bool {
|
||||||
|
switch name {
|
||||||
|
case "Row", "Rows", "QueryRow", "QueryRowContext":
|
||||||
|
return true
|
||||||
|
default:
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// receiverIsRowHandle reports whether x is syntactically a call to a
|
||||||
|
// row producer, which is the only receiver form this check accepts for
|
||||||
|
// a Scan.
|
||||||
|
func receiverIsRowHandle(x ast.Expr) bool {
|
||||||
|
call, ok := x.(*ast.CallExpr)
|
||||||
|
if !ok {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
sel, ok := call.Fun.(*ast.SelectorExpr)
|
||||||
|
if !ok {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
return isRowProducer(sel.Sel.Name)
|
||||||
|
}
|
||||||
|
|
||||||
|
// unguardedScans returns the position of every Scan call in file whose
|
||||||
|
// receiver is not a row handle. It fails closed: a receiver it cannot
|
||||||
|
// resolve syntactically — a local variable, a struct field — is
|
||||||
|
// reported rather than assumed safe.
|
||||||
|
func unguardedScans(
|
||||||
|
fset *token.FileSet, file *ast.File,
|
||||||
|
) []token.Position {
|
||||||
|
var found []token.Position
|
||||||
|
|
||||||
|
ast.Inspect(file, func(n ast.Node) bool {
|
||||||
|
call, ok := n.(*ast.CallExpr)
|
||||||
|
if !ok {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
sel, ok := call.Fun.(*ast.SelectorExpr)
|
||||||
|
if !ok || sel.Sel.Name != "Scan" {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
if !receiverIsRowHandle(sel.X) {
|
||||||
|
found = append(found, fset.Position(sel.Sel.Pos()))
|
||||||
|
}
|
||||||
|
|
||||||
|
return true
|
||||||
|
})
|
||||||
|
|
||||||
|
return found
|
||||||
|
}
|
||||||
|
|
||||||
|
// moduleRoot walks up from the working directory to the directory
|
||||||
|
// holding go.mod.
|
||||||
|
func moduleRoot(t *testing.T) string {
|
||||||
|
t.Helper()
|
||||||
|
|
||||||
|
dir, err := os.Getwd()
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
for {
|
||||||
|
_, statErr := os.Stat(filepath.Join(dir, "go.mod"))
|
||||||
|
if statErr == nil {
|
||||||
|
return dir
|
||||||
|
}
|
||||||
|
|
||||||
|
parent := filepath.Dir(dir)
|
||||||
|
require.NotEqual(t, parent, dir, "no go.mod above %s", dir)
|
||||||
|
|
||||||
|
dir = parent
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// skipDir reports whether a directory holds no source this check
|
||||||
|
// governs.
|
||||||
|
func skipDir(name string) bool {
|
||||||
|
switch name {
|
||||||
|
case ".git", "bin", "node_modules", "testdata":
|
||||||
|
return true
|
||||||
|
default:
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// walkNonTestGo parses every non-test .go file under root and returns
|
||||||
|
// how many it parsed along with every unguarded Scan it found.
|
||||||
|
func walkNonTestGo(t *testing.T, root string) (int, []string) {
|
||||||
|
t.Helper()
|
||||||
|
|
||||||
|
var (
|
||||||
|
parsed int
|
||||||
|
hits []string
|
||||||
|
)
|
||||||
|
|
||||||
|
fset := token.NewFileSet()
|
||||||
|
|
||||||
|
require.NoError(t, filepath.WalkDir(
|
||||||
|
root,
|
||||||
|
func(path string, d fs.DirEntry, err error) error {
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if d.IsDir() {
|
||||||
|
if skipDir(d.Name()) {
|
||||||
|
return fs.SkipDir
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
if !isNonTestGo(d.Name()) {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
file, err := parser.ParseFile(fset, path, nil, 0)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
parsed++
|
||||||
|
|
||||||
|
for _, pos := range unguardedScans(fset, file) {
|
||||||
|
hits = append(hits, relPosition(root, pos))
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
},
|
||||||
|
))
|
||||||
|
|
||||||
|
return parsed, hits
|
||||||
|
}
|
||||||
|
|
||||||
|
// isNonTestGo reports whether a file name is Go source this check
|
||||||
|
// governs.
|
||||||
|
func isNonTestGo(name string) bool {
|
||||||
|
return strings.HasSuffix(name, ".go") &&
|
||||||
|
!strings.HasSuffix(name, "_test.go")
|
||||||
|
}
|
||||||
|
|
||||||
|
// relPosition renders pos with its path relative to root, so a failure
|
||||||
|
// names the file the way the repository does.
|
||||||
|
func relPosition(root string, pos token.Position) string {
|
||||||
|
name := pos.Filename
|
||||||
|
|
||||||
|
rel, err := filepath.Rel(root, name)
|
||||||
|
if err == nil {
|
||||||
|
name = rel
|
||||||
|
}
|
||||||
|
|
||||||
|
return fmt.Sprintf("%s:%d:%d", name, pos.Line, pos.Column)
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestGormScanIsNeverCalledOutsideTests keeps (*gorm.DB).Scan out of
|
||||||
|
// non-test code.
|
||||||
|
//
|
||||||
|
// It is the one statement path (*Logger).ParamsFilter does not reach:
|
||||||
|
// Scan swaps GORM's own trace recorder in for the adapter, and that
|
||||||
|
// recorder does not implement gorm.ParamsFilter, so the statement is
|
||||||
|
// logged with its values interpolated. The package comment states the
|
||||||
|
// limit; this fails when someone adds a call site anyway.
|
||||||
|
//
|
||||||
|
// The current tree has one caller, internal/database/database_test.go,
|
||||||
|
// which this check does not govern: it is test-only and its SELECT 1
|
||||||
|
// binds nothing.
|
||||||
|
func TestGormScanIsNeverCalledOutsideTests(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
parsed, offenders := walkNonTestGo(t, moduleRoot(t))
|
||||||
|
|
||||||
|
require.GreaterOrEqual(
|
||||||
|
t, parsed, minNonTestFiles,
|
||||||
|
"parsed %d non-test .go files, so this check found "+
|
||||||
|
"nothing to look at", parsed,
|
||||||
|
)
|
||||||
|
require.Empty(
|
||||||
|
t, offenders,
|
||||||
|
"Scan called on a receiver this check cannot show is a "+
|
||||||
|
"database/sql row handle. (*gorm.DB).Scan logs the "+
|
||||||
|
"statement with its bound values interpolated — use "+
|
||||||
|
"Find, Pluck, or Raw(...).Row().Scan instead. A "+
|
||||||
|
"database/sql Scan reached through a variable is "+
|
||||||
|
"reported too; write it as <producer>().Scan rather "+
|
||||||
|
"than widening this check.",
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
// scanGuardCase is one planted snippet and whether the check above
|
||||||
|
// should report it.
|
||||||
|
type scanGuardCase struct {
|
||||||
|
name string
|
||||||
|
body string
|
||||||
|
want int
|
||||||
|
}
|
||||||
|
|
||||||
|
func scanGuardCases() []scanGuardCase {
|
||||||
|
return []scanGuardCase{
|
||||||
|
{"gorm chain", `db.DB().Raw("SELECT 1").Scan(&v)`, 1},
|
||||||
|
{"gorm receiver", `gdb.Scan(&v)`, 1},
|
||||||
|
{"gorm via variable", "q := gdb.Raw(\"x\")\nq.Scan(&v)", 1},
|
||||||
|
{"gorm model chain", `gdb.Model(&x).Scan(&v)`, 1},
|
||||||
|
{"sql row", `gdb.Raw("SELECT 1").Row().Scan(&v)`, 0},
|
||||||
|
{"sql rows", `gdb.Raw("SELECT 1").Rows().Scan(&v)`, 0},
|
||||||
|
{"unrelated call", `gdb.Find(&v)`, 0},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestScanGuard_ReportsPlantedCalls proves the check fires. Without it
|
||||||
|
// a detector that matched nothing would satisfy the walk above no
|
||||||
|
// matter what the tree contained.
|
||||||
|
func TestScanGuard_ReportsPlantedCalls(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
for _, tc := range scanGuardCases() {
|
||||||
|
t.Run(tc.name, func(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
fset := token.NewFileSet()
|
||||||
|
src := fmt.Sprintf(
|
||||||
|
"package p\n\nfunc f() {\n\t%s\n}\n", tc.body,
|
||||||
|
)
|
||||||
|
|
||||||
|
file, err := parser.ParseFile(
|
||||||
|
fset, tc.name+".go", src, 0,
|
||||||
|
)
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
require.Len(t, unguardedScans(fset, file), tc.want)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -43,10 +43,7 @@ func (s *Server) serveUntilShutdown() {
|
|||||||
err := s.httpServer.ListenAndServe()
|
err := s.httpServer.ListenAndServe()
|
||||||
if err != nil && !errors.Is(err, http.ErrServerClosed) {
|
if err != nil && !errors.Is(err, http.ErrServerClosed) {
|
||||||
s.log.Error("listen error", "error", err)
|
s.log.Error("listen error", "error", err)
|
||||||
|
s.shutdownOnListenFailure()
|
||||||
if s.cancelFunc != nil {
|
|
||||||
s.cancelFunc()
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
95
internal/server/listen_failure_test.go
Normal file
95
internal/server/listen_failure_test.go
Normal file
@@ -0,0 +1,95 @@
|
|||||||
|
package server_test
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"net"
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/stretchr/testify/require"
|
||||||
|
"go.uber.org/fx"
|
||||||
|
"sneak.berlin/go/webhooker/internal/globals"
|
||||||
|
"sneak.berlin/go/webhooker/internal/server"
|
||||||
|
)
|
||||||
|
|
||||||
|
// listenFailureDeadline is how long the app gets to give up after a
|
||||||
|
// listen it cannot satisfy. The defect this pins left the process
|
||||||
|
// reporting RUNNING for 183 seconds with nothing bound; a bind error
|
||||||
|
// is known instantly, so anything past a moment here is that defect
|
||||||
|
// back.
|
||||||
|
const listenFailureDeadline = 2 * time.Second
|
||||||
|
|
||||||
|
// lifecycleTimeout bounds the app's start and stop sequences so a
|
||||||
|
// wedged hook fails the test instead of hanging it.
|
||||||
|
const lifecycleTimeout = 15 * time.Second
|
||||||
|
|
||||||
|
// TestListenFailure_ShutsDownTheApp pins that a listener the server
|
||||||
|
// cannot bind terminates the application with a non-zero status.
|
||||||
|
//
|
||||||
|
// The fx OnStart hook returns as soon as the serving goroutine is
|
||||||
|
// spawned, so a bind failure is discovered after fx has already
|
||||||
|
// reported RUNNING. Nothing else in the graph observes it, and the
|
||||||
|
// process used to stay alive with no listener: down, but indis-
|
||||||
|
// tinguishable from healthy to systemd's Restart=on-failure and to
|
||||||
|
// Docker's restart policies, which is the state this test exists to
|
||||||
|
// keep from returning.
|
||||||
|
//
|
||||||
|
// The port is occupied by a listener this test holds open, on a
|
||||||
|
// kernel-chosen port, so the failure is the real EADDRINUSE the
|
||||||
|
// operator hits when a second instance starts. Loopback is enough to
|
||||||
|
// collide with the server's wildcard bind: a listening socket on a
|
||||||
|
// specific address blocks the wildcard from claiming the same port.
|
||||||
|
func TestListenFailure_ShutsDownTheApp(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
var listenCfg net.ListenConfig
|
||||||
|
|
||||||
|
occupied, err := listenCfg.Listen(
|
||||||
|
t.Context(), "tcp", "127.0.0.1:0",
|
||||||
|
)
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
t.Cleanup(func() { _ = occupied.Close() })
|
||||||
|
|
||||||
|
addr, ok := occupied.Addr().(*net.TCPAddr)
|
||||||
|
require.True(t, ok, "listener is not TCP")
|
||||||
|
|
||||||
|
// The collaborators come from the wired graph rather than stubs,
|
||||||
|
// so the Server under test is the one that ships. Only the port
|
||||||
|
// is test-specific.
|
||||||
|
env := newTestEnv(t)
|
||||||
|
env.cfg.Port = addr.Port
|
||||||
|
|
||||||
|
app := fx.New(
|
||||||
|
fx.NopLogger,
|
||||||
|
fx.Supply(env.log, env.cfg, env.mw, env.hnd),
|
||||||
|
fx.Provide(globals.New, server.New),
|
||||||
|
fx.Invoke(func(*server.Server) {}),
|
||||||
|
)
|
||||||
|
|
||||||
|
startCtx, cancelStart := context.WithTimeout(
|
||||||
|
context.Background(), lifecycleTimeout,
|
||||||
|
)
|
||||||
|
defer cancelStart()
|
||||||
|
|
||||||
|
require.NoError(t, app.Start(startCtx))
|
||||||
|
|
||||||
|
select {
|
||||||
|
case sig := <-app.Wait():
|
||||||
|
require.Equal(
|
||||||
|
t, server.ListenFailureExitCode, sig.ExitCode,
|
||||||
|
"listen failure must exit non-zero",
|
||||||
|
)
|
||||||
|
case <-time.After(listenFailureDeadline):
|
||||||
|
t.Fatal("listen failure left the app running")
|
||||||
|
}
|
||||||
|
|
||||||
|
// The stop sequence still has to complete: the fix must reach
|
||||||
|
// shutdown through fx rather than around it.
|
||||||
|
stopCtx, cancelStop := context.WithTimeout(
|
||||||
|
context.Background(), lifecycleTimeout,
|
||||||
|
)
|
||||||
|
defer cancelStop()
|
||||||
|
|
||||||
|
require.NoError(t, app.Stop(stopCtx))
|
||||||
|
}
|
||||||
@@ -55,8 +55,11 @@ func (s *Server) setupGlobalMiddleware() {
|
|||||||
s.router.Use(s.mw.SecurityHeaders())
|
s.router.Use(s.mw.SecurityHeaders())
|
||||||
s.router.Use(s.mw.Logging())
|
s.router.Use(s.mw.Logging())
|
||||||
|
|
||||||
// Metrics middleware (only if credentials are configured)
|
// Metrics recording middleware, registered only when the
|
||||||
if s.params.Config.MetricsUsername != "" {
|
// endpoint that exposes what it records is served. The
|
||||||
|
// condition is the same MetricsAuthEnabled the /metrics mount
|
||||||
|
// in setupRoutes reads.
|
||||||
|
if s.params.Config.MetricsAuthEnabled() {
|
||||||
s.router.Use(s.mw.Metrics())
|
s.router.Use(s.mw.Metrics())
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -103,8 +106,14 @@ func (s *Server) setupRoutes() {
|
|||||||
s.h.HandleHealthCheck(),
|
s.h.HandleHealthCheck(),
|
||||||
)
|
)
|
||||||
|
|
||||||
// set up authenticated /metrics route:
|
// Authenticated /metrics route. The condition is
|
||||||
if s.params.Config.MetricsUsername != "" {
|
// Config.MetricsAuthEnabled and never the username alone: a
|
||||||
|
// username with an empty password would otherwise mount the
|
||||||
|
// endpoint behind a credential map that accepts an empty
|
||||||
|
// password. Config rejects that combination at startup, and
|
||||||
|
// this reads the same value the startup log reports, so the
|
||||||
|
// two cannot disagree about whether the route exists.
|
||||||
|
if s.params.Config.MetricsAuthEnabled() {
|
||||||
s.router.Group(func(r chi.Router) {
|
s.router.Group(func(r chi.Router) {
|
||||||
r.Use(s.mw.MetricsAuth())
|
r.Use(s.mw.MetricsAuth())
|
||||||
r.Get(
|
r.Get(
|
||||||
|
|||||||
@@ -34,6 +34,13 @@ import (
|
|||||||
// the CSRF middleware executed.
|
// the CSRF middleware executed.
|
||||||
const csrfCookieName = "_gorilla_csrf"
|
const csrfCookieName = "_gorilla_csrf"
|
||||||
|
|
||||||
|
const (
|
||||||
|
// metricsUser and metricsAuthValue are the /metrics basic-auth
|
||||||
|
// credentials the metrics routing tests below configure.
|
||||||
|
metricsUser = "metrics"
|
||||||
|
metricsAuthValue = "s3cret"
|
||||||
|
)
|
||||||
|
|
||||||
type noopNotifier struct{}
|
type noopNotifier struct{}
|
||||||
|
|
||||||
func (n *noopNotifier) Notify([]delivery.Task) {}
|
func (n *noopNotifier) Notify([]delivery.Task) {}
|
||||||
@@ -69,9 +76,23 @@ type testEnv struct {
|
|||||||
func newTestEnv(t *testing.T) *testEnv {
|
func newTestEnv(t *testing.T) *testEnv {
|
||||||
t.Helper()
|
t.Helper()
|
||||||
|
|
||||||
|
return newTestEnvWithConfig(t, &config.Config{
|
||||||
|
DataDir: t.TempDir(),
|
||||||
|
Environment: config.EnvironmentDev,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// newTestEnvWithConfig is newTestEnv over a caller-supplied Config,
|
||||||
|
// for the routes whose existence the configuration decides. The same
|
||||||
|
// pointer reaches the router and every middleware, so a test cannot
|
||||||
|
// accidentally configure one and not the other.
|
||||||
|
func newTestEnvWithConfig(
|
||||||
|
t *testing.T, cfg *config.Config,
|
||||||
|
) *testEnv {
|
||||||
|
t.Helper()
|
||||||
|
|
||||||
var (
|
var (
|
||||||
log *logger.Logger
|
log *logger.Logger
|
||||||
cfg *config.Config
|
|
||||||
mw *middleware.Middleware
|
mw *middleware.Middleware
|
||||||
hnd *handlers.Handlers
|
hnd *handlers.Handlers
|
||||||
sess *session.Session
|
sess *session.Session
|
||||||
@@ -84,12 +105,7 @@ func newTestEnv(t *testing.T) *testEnv {
|
|||||||
fx.Provide(
|
fx.Provide(
|
||||||
globals.New,
|
globals.New,
|
||||||
logger.New,
|
logger.New,
|
||||||
func() *config.Config {
|
func() *config.Config { return cfg },
|
||||||
return &config.Config{
|
|
||||||
DataDir: t.TempDir(),
|
|
||||||
Environment: config.EnvironmentDev,
|
|
||||||
}
|
|
||||||
},
|
|
||||||
database.New,
|
database.New,
|
||||||
database.NewWebhookDBManager,
|
database.NewWebhookDBManager,
|
||||||
healthcheck.New,
|
healthcheck.New,
|
||||||
@@ -99,7 +115,7 @@ func newTestEnv(t *testing.T) *testEnv {
|
|||||||
middleware.New,
|
middleware.New,
|
||||||
handlers.New,
|
handlers.New,
|
||||||
),
|
),
|
||||||
fx.Populate(&log, &cfg, &mw, &hnd, &sess, &db, &dbMgr),
|
fx.Populate(&log, &mw, &hnd, &sess, &db, &dbMgr),
|
||||||
)
|
)
|
||||||
app.RequireStart()
|
app.RequireStart()
|
||||||
t.Cleanup(app.RequireStop)
|
t.Cleanup(app.RequireStop)
|
||||||
@@ -657,3 +673,119 @@ func TestSourceLogsBody_OtherUser404s(t *testing.T) {
|
|||||||
assert.Equal(t, http.StatusSeeOther, anon.Code)
|
assert.Equal(t, http.StatusSeeOther, anon.Code)
|
||||||
assert.Equal(t, "/pages/login", anon.Header().Get("Location"))
|
assert.Equal(t, "/pages/login", anon.Header().Get("Location"))
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// metricsConfig is a Config differing from the routing default only
|
||||||
|
// in the two /metrics credentials.
|
||||||
|
func metricsConfig(
|
||||||
|
t *testing.T, username, password string,
|
||||||
|
) *config.Config {
|
||||||
|
t.Helper()
|
||||||
|
|
||||||
|
return &config.Config{
|
||||||
|
DataDir: t.TempDir(),
|
||||||
|
Environment: config.EnvironmentDev,
|
||||||
|
MetricsUsername: username,
|
||||||
|
MetricsPassword: password,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// metricsRequest asks the real router for /metrics with the given
|
||||||
|
// basic-auth credentials, or with no Authorization header when
|
||||||
|
// username is empty.
|
||||||
|
func (e *testEnv) metricsRequest(
|
||||||
|
username, password string,
|
||||||
|
) *httptest.ResponseRecorder {
|
||||||
|
req := httptest.NewRequestWithContext(
|
||||||
|
context.Background(), http.MethodGet, "/metrics", nil,
|
||||||
|
)
|
||||||
|
|
||||||
|
if username != "" {
|
||||||
|
req.SetBasicAuth(username, password)
|
||||||
|
}
|
||||||
|
|
||||||
|
w := httptest.NewRecorder()
|
||||||
|
e.router.ServeHTTP(w, req)
|
||||||
|
|
||||||
|
return w
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestMetricsRouteUnmountedWithoutCredentials pins that with neither
|
||||||
|
// credential configured the route does not exist, which is the
|
||||||
|
// documented behaviour and the only valid way for /metrics to be
|
||||||
|
// absent.
|
||||||
|
func TestMetricsRouteUnmountedWithoutCredentials(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
env := newTestEnvWithConfig(t, metricsConfig(t, "", ""))
|
||||||
|
|
||||||
|
assert.Equal(
|
||||||
|
t, http.StatusNotFound,
|
||||||
|
env.metricsRequest("", "").Code,
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestMetricsRouteRequiresCredentials pins that with both credentials
|
||||||
|
// configured the route exists and every request that does not carry
|
||||||
|
// the configured pair is refused — including the empty password that
|
||||||
|
// a half-set configuration used to make sufficient.
|
||||||
|
func TestMetricsRouteRequiresCredentials(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
env := newTestEnvWithConfig(
|
||||||
|
t, metricsConfig(t, metricsUser, metricsAuthValue),
|
||||||
|
)
|
||||||
|
|
||||||
|
assert.Equal(
|
||||||
|
t, http.StatusUnauthorized,
|
||||||
|
env.metricsRequest("", "").Code,
|
||||||
|
"no credentials must not reach the metrics handler",
|
||||||
|
)
|
||||||
|
assert.Equal(
|
||||||
|
t, http.StatusUnauthorized,
|
||||||
|
env.metricsRequest(metricsUser, "").Code,
|
||||||
|
"an empty password must not reach the metrics handler",
|
||||||
|
)
|
||||||
|
assert.Equal(
|
||||||
|
t, http.StatusUnauthorized,
|
||||||
|
env.metricsRequest(metricsUser, "wrong").Code,
|
||||||
|
)
|
||||||
|
|
||||||
|
ok := env.metricsRequest(metricsUser, metricsAuthValue)
|
||||||
|
assert.Equal(t, http.StatusOK, ok.Code)
|
||||||
|
assert.Contains(t, ok.Body.String(), "go_goroutines")
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestMetricsRouteUnmountedOnHalfSetConfig pins the defect from
|
||||||
|
// https://git.eeqj.de/sneak/webhooker/issues/205 at the routing
|
||||||
|
// layer. Config rejects a half-set pair at startup, so this Config
|
||||||
|
// cannot be reached from the environment; the assertion is that the
|
||||||
|
// route tree does not publish an endpoint accepting an empty
|
||||||
|
// password even when handed one anyway, because the mount and the
|
||||||
|
// startup log's hasMetricsAuth read the same value.
|
||||||
|
func TestMetricsRouteUnmountedOnHalfSetConfig(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
for _, tc := range []struct {
|
||||||
|
name string
|
||||||
|
username string
|
||||||
|
password string
|
||||||
|
}{
|
||||||
|
{name: "username only", username: metricsUser},
|
||||||
|
{name: "password only", password: metricsAuthValue},
|
||||||
|
} {
|
||||||
|
t.Run(tc.name, func(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
cfg := metricsConfig(t, tc.username, tc.password)
|
||||||
|
env := newTestEnvWithConfig(t, cfg)
|
||||||
|
|
||||||
|
assert.False(t, cfg.MetricsAuthEnabled())
|
||||||
|
assert.Equal(
|
||||||
|
t, http.StatusNotFound,
|
||||||
|
env.metricsRequest(
|
||||||
|
tc.username, tc.password,
|
||||||
|
).Code,
|
||||||
|
)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
@@ -50,6 +50,13 @@ const (
|
|||||||
minSentryFlush = 250 * time.Millisecond
|
minSentryFlush = 250 * time.Millisecond
|
||||||
)
|
)
|
||||||
|
|
||||||
|
// ListenFailureExitCode is the status the process exits with when the
|
||||||
|
// HTTP listener cannot be established, or dies for a reason other
|
||||||
|
// than a requested shutdown. It must stay non-zero: systemd
|
||||||
|
// `Restart=on-failure` and Docker's restart policies key off it, and a
|
||||||
|
// zero exit would read as a deliberate stop.
|
||||||
|
const ListenFailureExitCode = 1
|
||||||
|
|
||||||
// SentryFlushBudget reports how long the Sentry flush may run when
|
// SentryFlushBudget reports how long the Sentry flush may run when
|
||||||
// remaining is the time left on the fx stop context after the HTTP
|
// remaining is the time left on the fx stop context after the HTTP
|
||||||
// drain. sentry.Flush takes a bare duration and honours no context,
|
// drain. sentry.Flush takes a bare duration and honours no context,
|
||||||
@@ -75,13 +82,13 @@ type ServerParams struct {
|
|||||||
Config *config.Config
|
Config *config.Config
|
||||||
Middleware *middleware.Middleware
|
Middleware *middleware.Middleware
|
||||||
Handlers *handlers.Handlers
|
Handlers *handlers.Handlers
|
||||||
|
Shutdowner fx.Shutdowner
|
||||||
}
|
}
|
||||||
|
|
||||||
// Server is the main HTTP server that wires up routes and manages
|
// Server is the main HTTP server that wires up routes and manages
|
||||||
// graceful shutdown.
|
// graceful shutdown.
|
||||||
type Server struct {
|
type Server struct {
|
||||||
startupTime time.Time
|
startupTime time.Time
|
||||||
exitCode int
|
|
||||||
sentryEnabled bool
|
sentryEnabled bool
|
||||||
log *slog.Logger
|
log *slog.Logger
|
||||||
cancelFunc context.CancelFunc
|
cancelFunc context.CancelFunc
|
||||||
@@ -159,7 +166,12 @@ func (s *Server) enableSentry() {
|
|||||||
s.sentryEnabled = true
|
s.sentryEnabled = true
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *Server) serve() int {
|
// serve installs the signal watcher, starts the listener and blocks
|
||||||
|
// until the server's context is cancelled. The process exit status is
|
||||||
|
// fx's to decide — from a signal, or from the code
|
||||||
|
// shutdownOnListenFailure hands the Shutdowner — so this reports
|
||||||
|
// nothing back to its caller.
|
||||||
|
func (s *Server) serve() {
|
||||||
ctx, cancelFunc := context.WithCancel(context.Background())
|
ctx, cancelFunc := context.WithCancel(context.Background())
|
||||||
s.cancelFunc = cancelFunc
|
s.cancelFunc = cancelFunc
|
||||||
|
|
||||||
@@ -185,7 +197,30 @@ func (s *Server) serve() int {
|
|||||||
<-ctx.Done()
|
<-ctx.Done()
|
||||||
// Shutdown is handled by the fx OnStop hook (cleanShutdown).
|
// Shutdown is handled by the fx OnStop hook (cleanShutdown).
|
||||||
// Do not call cleanShutdown() here to avoid double invocation.
|
// Do not call cleanShutdown() here to avoid double invocation.
|
||||||
return s.exitCode
|
}
|
||||||
|
|
||||||
|
// shutdownOnListenFailure ends the application after the HTTP
|
||||||
|
// listener failed. The fx OnStart hook returns as soon as the serving
|
||||||
|
// goroutine is spawned, so nothing downstream of it ever learns that
|
||||||
|
// the listen failed: fx reports RUNNING and the process sits alive
|
||||||
|
// with nothing bound, which is invisible to systemd and Docker
|
||||||
|
// restart policies. Asking the Shutdowner to stop the app with a
|
||||||
|
// non-zero code is what turns that into a visible failure.
|
||||||
|
//
|
||||||
|
// The context cancel that follows only unwinds serve()'s own wait.
|
||||||
|
// The shutdown itself runs through fx's normal stop sequence, so the
|
||||||
|
// clean-shutdown drain in cleanShutdown is reached unchanged.
|
||||||
|
func (s *Server) shutdownOnListenFailure() {
|
||||||
|
err := s.params.Shutdowner.Shutdown(
|
||||||
|
fx.ExitCode(ListenFailureExitCode),
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
s.log.Error("shutdown request failed", "error", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if s.cancelFunc != nil {
|
||||||
|
s.cancelFunc()
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *Server) cleanupForExit() {
|
func (s *Server) cleanupForExit() {
|
||||||
@@ -193,9 +228,6 @@ func (s *Server) cleanupForExit() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (s *Server) cleanShutdown(ctx context.Context) {
|
func (s *Server) cleanShutdown(ctx context.Context) {
|
||||||
// initiate clean shutdown
|
|
||||||
s.exitCode = 0
|
|
||||||
|
|
||||||
ctxShutdown, shutdownCancel := context.WithTimeout(
|
ctxShutdown, shutdownCancel := context.WithTimeout(
|
||||||
ctx, ShutdownTimeout,
|
ctx, ShutdownTimeout,
|
||||||
)
|
)
|
||||||
|
|||||||
Reference in New Issue
Block a user