mirror of
https://github.com/syntrex-lab/gomcp.git
synced 2026-05-10 03:52:36 +02:00
feat: SOC ghost sinkhole, rate limiter, RBAC, demo seed
This commit is contained in:
parent
cc7956d835
commit
b8097d3f1b
19 changed files with 1169 additions and 63 deletions
|
|
@ -151,3 +151,104 @@ func TestRBAC_RoleHierarchy(t *testing.T) {
|
|||
}
|
||||
}
|
||||
}
|
||||
|
||||
// ── Security Regression Tests (T4 bug bounty patches) ──────────────
|
||||
|
||||
// TestRBAC_QueryParamKey_Rejected verifies that API keys in query params
|
||||
// are no longer accepted (P1 fix: credential leakage via URL).
|
||||
func TestRBAC_QueryParamKey_Rejected(t *testing.T) {
|
||||
rbac := NewRBACMiddleware(RBACConfig{Enabled: true})
|
||||
rbac.RegisterKey("api-test", "sk-query-key-001", RoleAdmin)
|
||||
|
||||
handler := rbac.Require(RoleViewer, func(w http.ResponseWriter, r *http.Request) {
|
||||
w.WriteHeader(http.StatusOK)
|
||||
})
|
||||
|
||||
// Query param should NOT authenticate
|
||||
req := httptest.NewRequest("GET", "/test?api_key=sk-query-key-001", nil)
|
||||
rec := httptest.NewRecorder()
|
||||
handler(rec, req)
|
||||
|
||||
if rec.Code != http.StatusUnauthorized {
|
||||
t.Fatalf("query param auth should be rejected (P1 fix), got %d", rec.Code)
|
||||
}
|
||||
}
|
||||
|
||||
// TestRBAC_UndefinedRole_Denied verifies that undefined/fabricated roles
|
||||
// are rejected by the permission check (P3 fix: default-deny).
|
||||
func TestRBAC_UndefinedRole_Denied(t *testing.T) {
|
||||
tests := []struct {
|
||||
name string
|
||||
user Role
|
||||
required Role
|
||||
}{
|
||||
{"fabricated user role", Role("superadmin"), RoleViewer},
|
||||
{"empty user role", Role(""), RoleViewer},
|
||||
{"fabricated required role", RoleAdmin, Role("superviewer")},
|
||||
{"both undefined", Role("ghost"), Role("phantom")},
|
||||
}
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
if hasPermission(tt.user, tt.required) {
|
||||
t.Errorf("hasPermission(%q, %q) should be false (undefined role)", tt.user, tt.required)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
// TestRBAC_HMAC_MatchesRegisteredKey verifies that the HMAC-based
|
||||
// constant-time lookup correctly authenticates valid keys.
|
||||
func TestRBAC_HMAC_MatchesRegisteredKey(t *testing.T) {
|
||||
rbac := NewRBACMiddleware(RBACConfig{Enabled: true})
|
||||
// Register multiple keys to ensure iteration works
|
||||
rbac.RegisterKey("key-a", "sk-aaaa-1111-2222-3333", RoleAdmin)
|
||||
rbac.RegisterKey("key-b", "sk-bbbb-4444-5555-6666", RoleAnalyst)
|
||||
rbac.RegisterKey("key-c", "sk-cccc-7777-8888-9999", RoleViewer)
|
||||
|
||||
handler := rbac.Require(RoleViewer, func(w http.ResponseWriter, r *http.Request) {
|
||||
w.WriteHeader(http.StatusOK)
|
||||
})
|
||||
|
||||
// Each key should authenticate successfully
|
||||
for _, key := range []string{"sk-aaaa-1111-2222-3333", "sk-bbbb-4444-5555-6666", "sk-cccc-7777-8888-9999"} {
|
||||
req := httptest.NewRequest("GET", "/test", nil)
|
||||
req.Header.Set("Authorization", "Bearer "+key)
|
||||
rec := httptest.NewRecorder()
|
||||
handler(rec, req)
|
||||
|
||||
if rec.Code != http.StatusOK {
|
||||
t.Errorf("key %s should authenticate, got %d", key[:10]+"...", rec.Code)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// TestRBAC_HMAC_RejectsWrongKey verifies that similar-looking keys
|
||||
// are rejected even if they share a common prefix.
|
||||
func TestRBAC_HMAC_RejectsWrongKey(t *testing.T) {
|
||||
rbac := NewRBACMiddleware(RBACConfig{Enabled: true})
|
||||
rbac.RegisterKey("real-key", "sk-admin-secret-key-12345", RoleAdmin)
|
||||
|
||||
handler := rbac.Require(RoleViewer, func(w http.ResponseWriter, r *http.Request) {
|
||||
w.WriteHeader(http.StatusOK)
|
||||
})
|
||||
|
||||
// Wrong keys — including prefix collision and off-by-one
|
||||
wrongKeys := []string{
|
||||
"sk-admin-secret-key-12346", // off by last char
|
||||
"sk-admin-secret-key-1234", // one char short
|
||||
"sk-admin-secret-key-123456", // one char extra
|
||||
"sk-admin-secret-key-12345 ", // trailing space
|
||||
"SK-ADMIN-SECRET-KEY-12345", // wrong case
|
||||
}
|
||||
|
||||
for _, key := range wrongKeys {
|
||||
req := httptest.NewRequest("GET", "/test", nil)
|
||||
req.Header.Set("Authorization", "Bearer "+key)
|
||||
rec := httptest.NewRecorder()
|
||||
handler(rec, req)
|
||||
|
||||
if rec.Code != http.StatusUnauthorized {
|
||||
t.Errorf("wrong key %q should be rejected, got %d", key, rec.Code)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue