mirror of
https://github.com/seaweedfs/seaweedfs.git
synced 2026-10-26 06:50:45 +01:00
Chris Lu
f9dfc0ea37
- OIDC discovery via .well-known/openid-configuration; falls back to /.well-known/jwks.json when discovery is absent. Reject discovery docs whose issuer claim does not match the configured issuer to defend against issuer-substitution. - ComputeParentUser derives a stable per-identity hash from (sub, iss). Surface as aws:userid in the request context and as a parent_user claim in the session JWT so per-user state survives token rotation. - Per-role MaxSessionDuration (3600..43200) clamps requested DurationSeconds before the STS service applies its own caps. - Tighten RoleSessionName to the AWS contract: 2..64 chars from [\w+=,.@-]. - Populate PackedPolicySize in AssumeRole / AssumeRoleWithWebIdentity / AssumeRoleWithLDAPIdentity responses as a percentage of the 2048-byte inline session policy budget.
66 lines
2.2 KiB
Go
66 lines
2.2 KiB
Go
package sts
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestComputeParentUserStability(t *testing.T) {
|
|
// Same (sub, iss) must produce the same hash, regardless of order or
|
|
// whitespace mutations callers should never apply.
|
|
a := ComputeParentUser("alice", "https://idp.example/")
|
|
b := ComputeParentUser("alice", "https://idp.example/")
|
|
if a == "" {
|
|
t.Fatal("parent user should not be empty for non-empty sub")
|
|
}
|
|
if a != b {
|
|
t.Fatalf("expected stable hash, got %q vs %q", a, b)
|
|
}
|
|
}
|
|
|
|
func TestComputeParentUserDistinguishesIssuer(t *testing.T) {
|
|
// The point of incorporating iss is that the same `sub` from two providers
|
|
// must not collide. If this assertion ever fails, the hash input is wrong.
|
|
a := ComputeParentUser("alice", "https://idp-a.example/")
|
|
b := ComputeParentUser("alice", "https://idp-b.example/")
|
|
if a == b {
|
|
t.Fatalf("hashes for different issuers must differ, both = %q", a)
|
|
}
|
|
}
|
|
|
|
func TestComputeParentUserDistinguishesSubject(t *testing.T) {
|
|
a := ComputeParentUser("alice", "https://idp.example/")
|
|
b := ComputeParentUser("bob", "https://idp.example/")
|
|
if a == b {
|
|
t.Fatalf("hashes for different subjects must differ, both = %q", a)
|
|
}
|
|
}
|
|
|
|
func TestComputeParentUserEmptySub(t *testing.T) {
|
|
if got := ComputeParentUser("", "https://idp.example/"); got != "" {
|
|
t.Fatalf("empty sub should produce empty parent user, got %q", got)
|
|
}
|
|
}
|
|
|
|
func TestComputeParentUserEncoding(t *testing.T) {
|
|
got := ComputeParentUser("alice", "https://idp.example/")
|
|
// Base64 RawURL has no padding and uses URL-safe alphabet — important
|
|
// because parent_user shows up in filer paths and audit log fields.
|
|
if strings.ContainsAny(got, "=+/") {
|
|
t.Fatalf("parent user should be base64 raw url, got %q", got)
|
|
}
|
|
}
|
|
|
|
func TestSessionClaimsRoundTripParentUser(t *testing.T) {
|
|
parent := ComputeParentUser("alice", "https://idp.example/")
|
|
claims := NewSTSSessionClaims("sid-1", "issuer", time.Now().Add(time.Hour)).
|
|
WithRoleInfo("arn:aws:iam::123:role/r", "arn:aws:sts::123:assumed-role/r/s", "arn:aws:sts::123:assumed-role/r/s").
|
|
WithParentUser(parent)
|
|
|
|
info := claims.ToSessionInfo()
|
|
if info.ParentUser != parent {
|
|
t.Fatalf("ParentUser lost on round-trip: got %q want %q", info.ParentUser, parent)
|
|
}
|
|
}
|