Files
zeekay efee4c9fb1 Add the WAF middleware
OWASP Coraza with the Core Rule Set v4 embedded, declared as `waf` on
every provider: file, labels and the Middleware CR. Rules load Core Rule
Set, then files, then inline directives.

The engine state is written after every rule. The Core Rule Set's own
recommended configuration opens with SecRuleEngine DetectionOnly, so a
state written first is overridden by the rules it governs and the
firewall loads all of CRS and refuses nothing; detectionOnly on the
middleware is the one place the mode is decided.

geoBlock rode along: it was declared in dynamic config only, absent from
MiddlewareSpec, both deep copies and the CRD schema, so a Middleware CR
carrying it produced a middleware without it. The +groupName marker now
matches the registered group, which is what makes controller-gen's output
land where code-gen.sh reads it; the stale traefik.io_*.yaml dumps go.

CI asks for the arm64 runner git.hanzo.ai actually has and builds both
platforms.
2026-09-03 16:14:35 -07:00

104 lines
3.4 KiB
Go

package waf
import (
"context"
"net/http"
"net/http/httptest"
"testing"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
"github.com/hanzoai/ingress/pkg/config/dynamic"
)
func serve(t *testing.T, config dynamic.WAF, target string) (int, bool) {
t.Helper()
reached := false
next := http.HandlerFunc(func(rw http.ResponseWriter, _ *http.Request) {
reached = true
rw.WriteHeader(http.StatusOK)
})
h, err := New(context.Background(), next, config, "waf")
require.NoError(t, err)
rw := httptest.NewRecorder()
h.ServeHTTP(rw, httptest.NewRequest(http.MethodGet, target, nil))
return rw.Code, reached
}
// A ruleset that loads nothing inspects nothing while answering every request
// successfully, so it is refused where it is built rather than at the edge.
func TestNoRulesIsRefused(t *testing.T) {
_, err := New(context.Background(), http.NotFoundHandler(), dynamic.WAF{}, "waf")
assert.Error(t, err)
// The control: the same call with one rule builds.
_, err = New(context.Background(), http.NotFoundHandler(),
dynamic.WAF{Directives: `SecRule REQUEST_URI "@contains /nope" "id:1,phase:1,deny,status:403"`}, "waf")
assert.NoError(t, err)
}
func TestInlineRuleRefusesAndTheNextHandlerNeverRuns(t *testing.T) {
config := dynamic.WAF{
Directives: `SecRule REQUEST_URI "@contains /admin" "id:1,phase:1,deny,status:403"`,
}
code, reached := serve(t, config, "http://x/admin")
assert.Equal(t, http.StatusForbidden, code)
assert.False(t, reached, "a refused request must not reach the service")
code, reached = serve(t, config, "http://x/ok")
assert.Equal(t, http.StatusOK, code)
assert.True(t, reached)
}
// DetectionOnly is how a ruleset is measured against live traffic: the same
// rule matches and the request still proceeds.
func TestDetectionOnlyRefusesNothing(t *testing.T) {
config := dynamic.WAF{
Directives: `SecRule REQUEST_URI "@contains /admin" "id:1,phase:1,deny,status:403"`,
DetectionOnly: true,
}
code, reached := serve(t, config, "http://x/admin")
assert.Equal(t, http.StatusOK, code)
assert.True(t, reached)
}
// The embedded Core Rule Set is loaded with the setup file it reads, so the
// anomaly thresholds the rules score against are actually set.
func TestCoreRuleSetRefusesInjection(t *testing.T) {
config := dynamic.WAF{CoreRuleSet: true}
code, reached := serve(t, config, "http://x/?id=1%27%20OR%20%271%27%3D%271")
assert.Equal(t, http.StatusForbidden, code, "SQL injection must be refused")
assert.False(t, reached)
code, reached = serve(t, config, "http://x/?id=7")
assert.Equal(t, http.StatusOK, code, "an ordinary request must pass")
assert.True(t, reached)
}
func TestCoreRuleSetRefusesTraversal(t *testing.T) {
code, reached := serve(t, dynamic.WAF{CoreRuleSet: true}, "http://x/?f=../../../../etc/passwd")
assert.Equal(t, http.StatusForbidden, code)
assert.False(t, reached)
}
// The engine state is this middleware's to decide. A ruleset that names its
// own — as the Core Rule Set's recommended configuration does on line 7 —
// must not be able to disarm the firewall that loaded it.
func TestARulesetCannotDisarmTheFirewall(t *testing.T) {
config := dynamic.WAF{
Directives: `SecRuleEngine DetectionOnly
SecRule REQUEST_URI "@contains /admin" "id:1,phase:1,deny,status:403"`,
}
code, reached := serve(t, config, "http://x/admin")
assert.Equal(t, http.StatusForbidden, code)
assert.False(t, reached)
}