Add a customizable HTTP/HTTPS block page for sinkholed queries
Serves a page explaining why a domain was blocked instead of leaving a sinkholed client with a dead connection. Binds its own HTTP/HTTPS listeners with self-signed, per-hostname TLS certs generated on the fly, re-evaluates the requesting client against the policy engine per request, and renders an HTML template editable from Settings with a live preview. Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01TTKpGMQzpfDsvedu1hvSUf
This commit is contained in:
@@ -0,0 +1,119 @@
|
||||
package app_test
|
||||
|
||||
import (
|
||||
"context"
|
||||
"crypto/tls"
|
||||
"io"
|
||||
"net/http"
|
||||
"strings"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/owen/vibedns/internal/app"
|
||||
)
|
||||
|
||||
// TestBlockPageServesMatchedPolicy exercises the whole path an operator sets
|
||||
// up: a blacklist, a policy that sinkholes to this host, and a network that
|
||||
// covers the test client. Hitting the block page listener for the blocked
|
||||
// hostname should re-evaluate the same policy the DNS engine would have used,
|
||||
// and the rendered page should carry the matched list and policy names.
|
||||
func TestBlockPageServesMatchedPolicy(t *testing.T) {
|
||||
a := newTestApp(t)
|
||||
ctx := context.Background()
|
||||
|
||||
list, err := a.CreateDomainList(ctx, testActor(), app.ListInput{
|
||||
Kind: "blacklist", Name: "Test Ads",
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatalf("create list: %v", err)
|
||||
}
|
||||
if _, err := a.AddDomain(ctx, testActor(), list.ID, "ads.example.com", true, ""); err != nil {
|
||||
t.Fatalf("add domain: %v", err)
|
||||
}
|
||||
|
||||
policy, err := a.CreatePolicy(ctx, testActor(), app.PolicyInput{
|
||||
Name: "Block Test",
|
||||
BlockAction: "sinkhole",
|
||||
SinkholeIPv4: "127.0.0.1",
|
||||
SinkholeIPv6: "::1",
|
||||
ListIDs: []int64{list.ID},
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatalf("create policy: %v", err)
|
||||
}
|
||||
|
||||
if _, err := a.CreateNetwork(ctx, testActor(), app.NetworkInput{
|
||||
Name: "Test Client", CIDR: "127.0.0.1/32", PolicyIDs: []int64{policy.ID},
|
||||
}); err != nil {
|
||||
t.Fatalf("create network: %v", err)
|
||||
}
|
||||
if err := a.Reload(ctx); err != nil {
|
||||
t.Fatalf("reload: %v", err)
|
||||
}
|
||||
|
||||
const httpAddr = "127.0.0.1:18280"
|
||||
const httpsAddr = "127.0.0.1:18243"
|
||||
if err := a.BlockPage.Start(httpAddr, httpsAddr); err != nil {
|
||||
t.Fatalf("start block page server: %v", err)
|
||||
}
|
||||
t.Cleanup(func() {
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||||
defer cancel()
|
||||
_ = a.BlockPage.Shutdown(ctx)
|
||||
})
|
||||
|
||||
client := &http.Client{
|
||||
Transport: &http.Transport{TLSClientConfig: &tls.Config{InsecureSkipVerify: true}},
|
||||
Timeout: 5 * time.Second,
|
||||
}
|
||||
|
||||
// Wait for the listener to actually accept, rather than sleeping blindly.
|
||||
deadline := time.Now().Add(2 * time.Second)
|
||||
for {
|
||||
if a.BlockPage.Running() {
|
||||
break
|
||||
}
|
||||
if time.Now().After(deadline) {
|
||||
t.Fatal("block page server never reported running")
|
||||
}
|
||||
time.Sleep(10 * time.Millisecond)
|
||||
}
|
||||
|
||||
req, err := http.NewRequest(http.MethodGet, "http://"+httpAddr+"/tracker.js", nil)
|
||||
if err != nil {
|
||||
t.Fatalf("build request: %v", err)
|
||||
}
|
||||
req.Host = "ads.example.com"
|
||||
|
||||
resp, err := client.Do(req)
|
||||
if err != nil {
|
||||
t.Fatalf("GET block page: %v", err)
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
body, err := io.ReadAll(resp.Body)
|
||||
if err != nil {
|
||||
t.Fatalf("read body: %v", err)
|
||||
}
|
||||
if resp.StatusCode != http.StatusOK {
|
||||
t.Fatalf("status = %d, want 200", resp.StatusCode)
|
||||
}
|
||||
page := string(body)
|
||||
if !strings.Contains(page, "Test Ads") {
|
||||
t.Errorf("page does not mention the matched list name: %s", page)
|
||||
}
|
||||
if !strings.Contains(page, "Block Test") {
|
||||
t.Errorf("page does not mention the matched policy name: %s", page)
|
||||
}
|
||||
if !strings.Contains(page, "ads.example.com") {
|
||||
t.Errorf("page does not mention the requested domain: %s", page)
|
||||
}
|
||||
|
||||
httpsResp, err := client.Get("https://" + httpsAddr + "/")
|
||||
if err != nil {
|
||||
t.Fatalf("GET https block page: %v", err)
|
||||
}
|
||||
defer httpsResp.Body.Close()
|
||||
if httpsResp.StatusCode != http.StatusOK {
|
||||
t.Fatalf("https status = %d, want 200", httpsResp.StatusCode)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user