feat(s3): add STS GetFederationToken support (#8891)
* feat(s3): add STS GetFederationToken support Implement the AWS STS GetFederationToken API, which allows long-term IAM users to obtain temporary credentials scoped down by an optional inline session policy. This is useful for server-side applications that mint per-user temporary credentials. Key behaviors: - Requires SigV4 authentication from a long-term IAM user - Rejects calls from temporary credentials (session tokens) - Name parameter (2-64 chars) identifies the federated user - DurationSeconds supports 900-129600 (15 min to 36 hours, default 12h) - Optional inline session policy for permission scoping - Caller's attached policies are embedded in the JWT token - Returns federated user ARN: arn:aws:sts::<account>:federated-user/<Name> No performance impact on the S3 hot path — credential vending is a separate control-plane operation, and all policy data is embedded in the stateless JWT token. * fix(s3): address GetFederationToken PR review feedback - Fix Name validation: max 32 chars (not 64) per AWS spec, add regex validation for [\w+=,.@-]+ character whitelist - Refactor parseDurationSeconds into parseDurationSecondsWithBounds to eliminate duplicated duration parsing logic - Add sts:GetFederationToken permission check via VerifyActionPermission mirroring the AssumeRole authorization pattern - Change GetPoliciesForUser to return ([]string, error) so callers fail closed on policy-resolution failures instead of silently returning nil - Move temporary-credentials rejection before SigV4 verification for early rejection and proper test coverage - Update tests: verify specific error message for temp cred rejection, add regex validation test cases (spaces, slashes rejected) * refactor(s3): use sts.Action* constants instead of hard-coded strings Replace hard-coded "sts:AssumeRole" and "sts:GetFederationToken" strings in VerifyActionPermission calls with sts.ActionAssumeRole and sts.ActionGetFederationToken package constants. * fix(s3): pass through sts: prefix in action resolver and merge policies Two fixes: 1. mapBaseActionToS3Format now passes through "sts:" prefix alongside "s3:" and "iam:", preventing sts:GetFederationToken from being rewritten to s3:sts:GetFederationToken in VerifyActionPermission. This also fixes the existing sts:AssumeRole permission checks. 2. GetFederationToken policy embedding now merges identity.PolicyNames (from SigV4 identity) with policies from the IAM manager (which may include group-attached policies), deduplicated via a map. Previously the IAM manager lookup was skipped when identity.PolicyNames was non-empty, causing group policies to be omitted from the token. * test(s3): add integration tests for sts: action passthrough and policy merge Action resolver tests: - TestMapBaseActionToS3Format_ServicePrefixPassthrough: verifies s3:, iam:, and sts: prefixed actions pass through unchanged while coarse actions (Read, Write) are mapped to S3 format - TestResolveS3Action_STSActionsPassthrough: verifies sts:AssumeRole, sts:GetFederationToken, sts:GetCallerIdentity pass through ResolveS3Action unchanged with both nil and real HTTP requests Policy merge tests: - TestGetFederationToken_GetPoliciesForUser: tests IAMManager.GetPoliciesForUser with no user store (error), missing user, user with policies, user without - TestGetFederationToken_PolicyMergeAndDedup: tests that identity.PolicyNames and IAM-manager-resolved policies are merged and deduplicated (SharedPolicy appears in both sources, result has 3 unique policies) - TestGetFederationToken_PolicyMergeNoManager: tests that when IAM manager is unavailable, identity.PolicyNames alone are embedded * test(s3): add end-to-end integration tests for GetFederationToken Add integration tests that call GetFederationToken using real AWS SigV4 signed HTTP requests against a running SeaweedFS instance, following the existing pattern in test/s3/iam/s3_sts_assume_role_test.go. Tests: - TestSTSGetFederationTokenValidation: missing name, name too short/long, invalid characters, duration too short/long, malformed policy, anonymous rejection (7 subtests) - TestSTSGetFederationTokenRejectTemporaryCredentials: obtains temp creds via AssumeRole then verifies GetFederationToken rejects them - TestSTSGetFederationTokenSuccess: basic success, custom 1h duration, 36h max duration with expiration time verification - TestSTSGetFederationTokenWithSessionPolicy: creates a bucket, obtains federated creds with GetObject-only session policy, verifies GetObject succeeds and PutObject is denied using the AWS SDK S3 client
This commit is contained in:
511
test/s3/iam/s3_sts_get_federation_token_test.go
Normal file
511
test/s3/iam/s3_sts_get_federation_token_test.go
Normal file
@@ -0,0 +1,511 @@
|
|||||||
|
package iam
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/xml"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"net/http"
|
||||||
|
"net/url"
|
||||||
|
"os"
|
||||||
|
"strings"
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/aws/aws-sdk-go/aws"
|
||||||
|
"github.com/aws/aws-sdk-go/aws/credentials"
|
||||||
|
"github.com/aws/aws-sdk-go/aws/session"
|
||||||
|
"github.com/aws/aws-sdk-go/service/s3"
|
||||||
|
v4 "github.com/aws/aws-sdk-go/aws/signer/v4"
|
||||||
|
"github.com/stretchr/testify/assert"
|
||||||
|
"github.com/stretchr/testify/require"
|
||||||
|
)
|
||||||
|
|
||||||
|
// GetFederationTokenTestResponse represents the STS GetFederationToken response
|
||||||
|
type GetFederationTokenTestResponse struct {
|
||||||
|
XMLName xml.Name `xml:"GetFederationTokenResponse"`
|
||||||
|
Result struct {
|
||||||
|
Credentials struct {
|
||||||
|
AccessKeyId string `xml:"AccessKeyId"`
|
||||||
|
SecretAccessKey string `xml:"SecretAccessKey"`
|
||||||
|
SessionToken string `xml:"SessionToken"`
|
||||||
|
Expiration string `xml:"Expiration"`
|
||||||
|
} `xml:"Credentials"`
|
||||||
|
FederatedUser struct {
|
||||||
|
FederatedUserId string `xml:"FederatedUserId"`
|
||||||
|
Arn string `xml:"Arn"`
|
||||||
|
} `xml:"FederatedUser"`
|
||||||
|
} `xml:"GetFederationTokenResult"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func getTestCredentials() (string, string) {
|
||||||
|
accessKey := os.Getenv("STS_TEST_ACCESS_KEY")
|
||||||
|
if accessKey == "" {
|
||||||
|
accessKey = "admin"
|
||||||
|
}
|
||||||
|
secretKey := os.Getenv("STS_TEST_SECRET_KEY")
|
||||||
|
if secretKey == "" {
|
||||||
|
secretKey = "admin"
|
||||||
|
}
|
||||||
|
return accessKey, secretKey
|
||||||
|
}
|
||||||
|
|
||||||
|
// isGetFederationTokenImplemented checks if the running server supports GetFederationToken
|
||||||
|
func isGetFederationTokenImplemented(t *testing.T) bool {
|
||||||
|
accessKey, secretKey := getTestCredentials()
|
||||||
|
resp, err := callSTSAPIWithSigV4(t, url.Values{
|
||||||
|
"Action": {"GetFederationToken"},
|
||||||
|
"Version": {"2011-06-15"},
|
||||||
|
"Name": {"probe"},
|
||||||
|
}, accessKey, secretKey)
|
||||||
|
if err != nil {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
defer resp.Body.Close()
|
||||||
|
|
||||||
|
body, _ := io.ReadAll(resp.Body)
|
||||||
|
var errResp STSErrorTestResponse
|
||||||
|
if xml.Unmarshal(body, &errResp) == nil {
|
||||||
|
if errResp.Error.Code == "InvalidAction" || errResp.Error.Code == "NotImplemented" {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestSTSGetFederationTokenValidation tests input validation for the GetFederationToken endpoint
|
||||||
|
func TestSTSGetFederationTokenValidation(t *testing.T) {
|
||||||
|
if testing.Short() {
|
||||||
|
t.Skip("Skipping integration test in short mode")
|
||||||
|
}
|
||||||
|
|
||||||
|
if !isSTSEndpointRunning(t) {
|
||||||
|
t.Fatal("SeaweedFS STS endpoint is not running at", TestSTSEndpoint, "- please run 'make setup-all-tests' first")
|
||||||
|
}
|
||||||
|
|
||||||
|
if !isGetFederationTokenImplemented(t) {
|
||||||
|
t.Fatal("GetFederationToken action is not implemented in the running server")
|
||||||
|
}
|
||||||
|
|
||||||
|
accessKey, secretKey := getTestCredentials()
|
||||||
|
|
||||||
|
t.Run("missing_name", func(t *testing.T) {
|
||||||
|
resp, err := callSTSAPIWithSigV4(t, url.Values{
|
||||||
|
"Action": {"GetFederationToken"},
|
||||||
|
"Version": {"2011-06-15"},
|
||||||
|
// Name is missing
|
||||||
|
}, accessKey, secretKey)
|
||||||
|
require.NoError(t, err)
|
||||||
|
defer resp.Body.Close()
|
||||||
|
|
||||||
|
body, _ := io.ReadAll(resp.Body)
|
||||||
|
var errResp STSErrorTestResponse
|
||||||
|
require.NoError(t, xml.Unmarshal(body, &errResp), "Failed to parse: %s", string(body))
|
||||||
|
assert.Equal(t, "MissingParameter", errResp.Error.Code)
|
||||||
|
})
|
||||||
|
|
||||||
|
t.Run("name_too_short", func(t *testing.T) {
|
||||||
|
resp, err := callSTSAPIWithSigV4(t, url.Values{
|
||||||
|
"Action": {"GetFederationToken"},
|
||||||
|
"Version": {"2011-06-15"},
|
||||||
|
"Name": {"A"},
|
||||||
|
}, accessKey, secretKey)
|
||||||
|
require.NoError(t, err)
|
||||||
|
defer resp.Body.Close()
|
||||||
|
|
||||||
|
body, _ := io.ReadAll(resp.Body)
|
||||||
|
var errResp STSErrorTestResponse
|
||||||
|
require.NoError(t, xml.Unmarshal(body, &errResp), "Failed to parse: %s", string(body))
|
||||||
|
assert.Equal(t, "InvalidParameterValue", errResp.Error.Code)
|
||||||
|
})
|
||||||
|
|
||||||
|
t.Run("name_too_long", func(t *testing.T) {
|
||||||
|
resp, err := callSTSAPIWithSigV4(t, url.Values{
|
||||||
|
"Action": {"GetFederationToken"},
|
||||||
|
"Version": {"2011-06-15"},
|
||||||
|
"Name": {strings.Repeat("A", 33)},
|
||||||
|
}, accessKey, secretKey)
|
||||||
|
require.NoError(t, err)
|
||||||
|
defer resp.Body.Close()
|
||||||
|
|
||||||
|
body, _ := io.ReadAll(resp.Body)
|
||||||
|
var errResp STSErrorTestResponse
|
||||||
|
require.NoError(t, xml.Unmarshal(body, &errResp), "Failed to parse: %s", string(body))
|
||||||
|
assert.Equal(t, "InvalidParameterValue", errResp.Error.Code)
|
||||||
|
})
|
||||||
|
|
||||||
|
t.Run("name_invalid_characters", func(t *testing.T) {
|
||||||
|
resp, err := callSTSAPIWithSigV4(t, url.Values{
|
||||||
|
"Action": {"GetFederationToken"},
|
||||||
|
"Version": {"2011-06-15"},
|
||||||
|
"Name": {"bad name"},
|
||||||
|
}, accessKey, secretKey)
|
||||||
|
require.NoError(t, err)
|
||||||
|
defer resp.Body.Close()
|
||||||
|
|
||||||
|
body, _ := io.ReadAll(resp.Body)
|
||||||
|
var errResp STSErrorTestResponse
|
||||||
|
require.NoError(t, xml.Unmarshal(body, &errResp), "Failed to parse: %s", string(body))
|
||||||
|
assert.Equal(t, "InvalidParameterValue", errResp.Error.Code)
|
||||||
|
})
|
||||||
|
|
||||||
|
t.Run("duration_too_short", func(t *testing.T) {
|
||||||
|
resp, err := callSTSAPIWithSigV4(t, url.Values{
|
||||||
|
"Action": {"GetFederationToken"},
|
||||||
|
"Version": {"2011-06-15"},
|
||||||
|
"Name": {"TestApp"},
|
||||||
|
"DurationSeconds": {"100"},
|
||||||
|
}, accessKey, secretKey)
|
||||||
|
require.NoError(t, err)
|
||||||
|
defer resp.Body.Close()
|
||||||
|
|
||||||
|
body, _ := io.ReadAll(resp.Body)
|
||||||
|
var errResp STSErrorTestResponse
|
||||||
|
require.NoError(t, xml.Unmarshal(body, &errResp), "Failed to parse: %s", string(body))
|
||||||
|
assert.Equal(t, "InvalidParameterValue", errResp.Error.Code)
|
||||||
|
})
|
||||||
|
|
||||||
|
t.Run("duration_too_long", func(t *testing.T) {
|
||||||
|
resp, err := callSTSAPIWithSigV4(t, url.Values{
|
||||||
|
"Action": {"GetFederationToken"},
|
||||||
|
"Version": {"2011-06-15"},
|
||||||
|
"Name": {"TestApp"},
|
||||||
|
"DurationSeconds": {"200000"},
|
||||||
|
}, accessKey, secretKey)
|
||||||
|
require.NoError(t, err)
|
||||||
|
defer resp.Body.Close()
|
||||||
|
|
||||||
|
body, _ := io.ReadAll(resp.Body)
|
||||||
|
var errResp STSErrorTestResponse
|
||||||
|
require.NoError(t, xml.Unmarshal(body, &errResp), "Failed to parse: %s", string(body))
|
||||||
|
assert.Equal(t, "InvalidParameterValue", errResp.Error.Code)
|
||||||
|
})
|
||||||
|
|
||||||
|
t.Run("malformed_policy", func(t *testing.T) {
|
||||||
|
resp, err := callSTSAPIWithSigV4(t, url.Values{
|
||||||
|
"Action": {"GetFederationToken"},
|
||||||
|
"Version": {"2011-06-15"},
|
||||||
|
"Name": {"TestApp"},
|
||||||
|
"Policy": {"not-valid-json"},
|
||||||
|
}, accessKey, secretKey)
|
||||||
|
require.NoError(t, err)
|
||||||
|
defer resp.Body.Close()
|
||||||
|
|
||||||
|
body, _ := io.ReadAll(resp.Body)
|
||||||
|
var errResp STSErrorTestResponse
|
||||||
|
require.NoError(t, xml.Unmarshal(body, &errResp), "Failed to parse: %s", string(body))
|
||||||
|
assert.Equal(t, "MalformedPolicyDocument", errResp.Error.Code)
|
||||||
|
})
|
||||||
|
|
||||||
|
t.Run("anonymous_rejected", func(t *testing.T) {
|
||||||
|
// GetFederationToken requires SigV4, anonymous should fail
|
||||||
|
resp, err := callSTSAPI(t, url.Values{
|
||||||
|
"Action": {"GetFederationToken"},
|
||||||
|
"Version": {"2011-06-15"},
|
||||||
|
"Name": {"TestApp"},
|
||||||
|
})
|
||||||
|
require.NoError(t, err)
|
||||||
|
defer resp.Body.Close()
|
||||||
|
|
||||||
|
assert.NotEqual(t, http.StatusOK, resp.StatusCode)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestSTSGetFederationTokenRejectTemporaryCredentials tests that temporary
|
||||||
|
// credentials (session tokens) are rejected by GetFederationToken
|
||||||
|
func TestSTSGetFederationTokenRejectTemporaryCredentials(t *testing.T) {
|
||||||
|
if testing.Short() {
|
||||||
|
t.Skip("Skipping integration test in short mode")
|
||||||
|
}
|
||||||
|
|
||||||
|
if !isSTSEndpointRunning(t) {
|
||||||
|
t.Skip("SeaweedFS STS endpoint is not running at", TestSTSEndpoint)
|
||||||
|
}
|
||||||
|
|
||||||
|
if !isGetFederationTokenImplemented(t) {
|
||||||
|
t.Skip("GetFederationToken not implemented")
|
||||||
|
}
|
||||||
|
|
||||||
|
accessKey, secretKey := getTestCredentials()
|
||||||
|
|
||||||
|
// First, obtain temporary credentials via AssumeRole
|
||||||
|
resp, err := callSTSAPIWithSigV4(t, url.Values{
|
||||||
|
"Action": {"AssumeRole"},
|
||||||
|
"Version": {"2011-06-15"},
|
||||||
|
"RoleArn": {"arn:aws:iam::role/admin"},
|
||||||
|
"RoleSessionName": {"temp-session"},
|
||||||
|
}, accessKey, secretKey)
|
||||||
|
require.NoError(t, err)
|
||||||
|
defer resp.Body.Close()
|
||||||
|
|
||||||
|
body, err := io.ReadAll(resp.Body)
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
if resp.StatusCode != http.StatusOK {
|
||||||
|
t.Skipf("AssumeRole failed (may not be configured): status=%d body=%s", resp.StatusCode, string(body))
|
||||||
|
}
|
||||||
|
|
||||||
|
var assumeResp AssumeRoleTestResponse
|
||||||
|
require.NoError(t, xml.Unmarshal(body, &assumeResp), "Parse AssumeRole response: %s", string(body))
|
||||||
|
|
||||||
|
tempAccessKey := assumeResp.Result.Credentials.AccessKeyId
|
||||||
|
tempSecretKey := assumeResp.Result.Credentials.SecretAccessKey
|
||||||
|
tempSessionToken := assumeResp.Result.Credentials.SessionToken
|
||||||
|
require.NotEmpty(t, tempAccessKey)
|
||||||
|
require.NotEmpty(t, tempSessionToken)
|
||||||
|
|
||||||
|
// Now try GetFederationToken with the temporary credentials
|
||||||
|
// Include X-Amz-Security-Token header which marks this as a temp credential call
|
||||||
|
params := url.Values{
|
||||||
|
"Action": {"GetFederationToken"},
|
||||||
|
"Version": {"2011-06-15"},
|
||||||
|
"Name": {"ShouldFail"},
|
||||||
|
}
|
||||||
|
|
||||||
|
reqBody := params.Encode()
|
||||||
|
req, err := http.NewRequest(http.MethodPost, TestSTSEndpoint+"/", strings.NewReader(reqBody))
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
|
||||||
|
req.Header.Set("X-Amz-Security-Token", tempSessionToken)
|
||||||
|
|
||||||
|
creds := credentials.NewStaticCredentials(tempAccessKey, tempSecretKey, tempSessionToken)
|
||||||
|
signer := v4.NewSigner(creds)
|
||||||
|
_, err = signer.Sign(req, strings.NewReader(reqBody), "sts", "us-east-1", time.Now())
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
client := &http.Client{Timeout: 30 * time.Second}
|
||||||
|
resp2, err := client.Do(req)
|
||||||
|
require.NoError(t, err)
|
||||||
|
defer resp2.Body.Close()
|
||||||
|
|
||||||
|
body2, _ := io.ReadAll(resp2.Body)
|
||||||
|
assert.Equal(t, http.StatusForbidden, resp2.StatusCode,
|
||||||
|
"GetFederationToken should reject temporary credentials: %s", string(body2))
|
||||||
|
assert.Contains(t, string(body2), "temporary credentials",
|
||||||
|
"Error should mention temporary credentials")
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestSTSGetFederationTokenSuccess tests a successful GetFederationToken call
|
||||||
|
// and verifies the returned credentials can be used to access S3
|
||||||
|
func TestSTSGetFederationTokenSuccess(t *testing.T) {
|
||||||
|
if testing.Short() {
|
||||||
|
t.Skip("Skipping integration test in short mode")
|
||||||
|
}
|
||||||
|
|
||||||
|
if !isSTSEndpointRunning(t) {
|
||||||
|
t.Skip("SeaweedFS STS endpoint is not running at", TestSTSEndpoint)
|
||||||
|
}
|
||||||
|
|
||||||
|
if !isGetFederationTokenImplemented(t) {
|
||||||
|
t.Skip("GetFederationToken not implemented")
|
||||||
|
}
|
||||||
|
|
||||||
|
accessKey, secretKey := getTestCredentials()
|
||||||
|
|
||||||
|
t.Run("basic_success", func(t *testing.T) {
|
||||||
|
resp, err := callSTSAPIWithSigV4(t, url.Values{
|
||||||
|
"Action": {"GetFederationToken"},
|
||||||
|
"Version": {"2011-06-15"},
|
||||||
|
"Name": {"AppClient"},
|
||||||
|
}, accessKey, secretKey)
|
||||||
|
require.NoError(t, err)
|
||||||
|
defer resp.Body.Close()
|
||||||
|
|
||||||
|
body, err := io.ReadAll(resp.Body)
|
||||||
|
require.NoError(t, err)
|
||||||
|
t.Logf("Response status: %d, body: %s", resp.StatusCode, string(body))
|
||||||
|
|
||||||
|
if resp.StatusCode != http.StatusOK {
|
||||||
|
var errResp STSErrorTestResponse
|
||||||
|
_ = xml.Unmarshal(body, &errResp)
|
||||||
|
t.Fatalf("GetFederationToken failed: code=%s message=%s", errResp.Error.Code, errResp.Error.Message)
|
||||||
|
}
|
||||||
|
|
||||||
|
var stsResp GetFederationTokenTestResponse
|
||||||
|
require.NoError(t, xml.Unmarshal(body, &stsResp), "Parse response: %s", string(body))
|
||||||
|
|
||||||
|
creds := stsResp.Result.Credentials
|
||||||
|
assert.NotEmpty(t, creds.AccessKeyId)
|
||||||
|
assert.NotEmpty(t, creds.SecretAccessKey)
|
||||||
|
assert.NotEmpty(t, creds.SessionToken)
|
||||||
|
assert.NotEmpty(t, creds.Expiration)
|
||||||
|
|
||||||
|
fedUser := stsResp.Result.FederatedUser
|
||||||
|
assert.Contains(t, fedUser.Arn, "federated-user/AppClient")
|
||||||
|
assert.Contains(t, fedUser.FederatedUserId, "AppClient")
|
||||||
|
})
|
||||||
|
|
||||||
|
t.Run("with_custom_duration", func(t *testing.T) {
|
||||||
|
resp, err := callSTSAPIWithSigV4(t, url.Values{
|
||||||
|
"Action": {"GetFederationToken"},
|
||||||
|
"Version": {"2011-06-15"},
|
||||||
|
"Name": {"DurationTest"},
|
||||||
|
"DurationSeconds": {"3600"},
|
||||||
|
}, accessKey, secretKey)
|
||||||
|
require.NoError(t, err)
|
||||||
|
defer resp.Body.Close()
|
||||||
|
|
||||||
|
body, _ := io.ReadAll(resp.Body)
|
||||||
|
t.Logf("Response status: %d, body: %s", resp.StatusCode, string(body))
|
||||||
|
|
||||||
|
if resp.StatusCode == http.StatusOK {
|
||||||
|
var stsResp GetFederationTokenTestResponse
|
||||||
|
require.NoError(t, xml.Unmarshal(body, &stsResp))
|
||||||
|
assert.NotEmpty(t, stsResp.Result.Credentials.AccessKeyId)
|
||||||
|
|
||||||
|
// Verify expiration is roughly 1 hour from now
|
||||||
|
expTime, err := time.Parse(time.RFC3339, stsResp.Result.Credentials.Expiration)
|
||||||
|
require.NoError(t, err)
|
||||||
|
diff := time.Until(expTime)
|
||||||
|
assert.InDelta(t, 3600, diff.Seconds(), 60,
|
||||||
|
"Expiration should be ~1 hour from now")
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
t.Run("with_36_hour_duration", func(t *testing.T) {
|
||||||
|
// GetFederationToken allows up to 36 hours (unlike AssumeRole's 12h max)
|
||||||
|
resp, err := callSTSAPIWithSigV4(t, url.Values{
|
||||||
|
"Action": {"GetFederationToken"},
|
||||||
|
"Version": {"2011-06-15"},
|
||||||
|
"Name": {"LongDuration"},
|
||||||
|
"DurationSeconds": {"129600"}, // 36 hours
|
||||||
|
}, accessKey, secretKey)
|
||||||
|
require.NoError(t, err)
|
||||||
|
defer resp.Body.Close()
|
||||||
|
|
||||||
|
body, _ := io.ReadAll(resp.Body)
|
||||||
|
if resp.StatusCode == http.StatusOK {
|
||||||
|
var stsResp GetFederationTokenTestResponse
|
||||||
|
require.NoError(t, xml.Unmarshal(body, &stsResp))
|
||||||
|
|
||||||
|
expTime, err := time.Parse(time.RFC3339, stsResp.Result.Credentials.Expiration)
|
||||||
|
require.NoError(t, err)
|
||||||
|
diff := time.Until(expTime)
|
||||||
|
assert.InDelta(t, 129600, diff.Seconds(), 60,
|
||||||
|
"Expiration should be ~36 hours from now")
|
||||||
|
} else {
|
||||||
|
// Duration should not cause a rejection
|
||||||
|
var errResp STSErrorTestResponse
|
||||||
|
_ = xml.Unmarshal(body, &errResp)
|
||||||
|
assert.NotContains(t, errResp.Error.Message, "DurationSeconds",
|
||||||
|
"36-hour duration should be accepted by GetFederationToken")
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestSTSGetFederationTokenWithSessionPolicy tests that vended credentials
|
||||||
|
// are scoped down by an inline session policy
|
||||||
|
func TestSTSGetFederationTokenWithSessionPolicy(t *testing.T) {
|
||||||
|
if testing.Short() {
|
||||||
|
t.Skip("Skipping integration test in short mode")
|
||||||
|
}
|
||||||
|
|
||||||
|
if !isSTSEndpointRunning(t) {
|
||||||
|
t.Skip("SeaweedFS STS endpoint is not running at", TestSTSEndpoint)
|
||||||
|
}
|
||||||
|
|
||||||
|
if !isGetFederationTokenImplemented(t) {
|
||||||
|
t.Skip("GetFederationToken not implemented")
|
||||||
|
}
|
||||||
|
|
||||||
|
accessKey, secretKey := getTestCredentials()
|
||||||
|
|
||||||
|
// Create a test bucket using admin credentials
|
||||||
|
adminSess, err := session.NewSession(&aws.Config{
|
||||||
|
Region: aws.String("us-east-1"),
|
||||||
|
Endpoint: aws.String(TestSTSEndpoint),
|
||||||
|
DisableSSL: aws.Bool(true),
|
||||||
|
S3ForcePathStyle: aws.Bool(true),
|
||||||
|
Credentials: credentials.NewStaticCredentials(accessKey, secretKey, ""),
|
||||||
|
})
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
adminS3 := s3.New(adminSess)
|
||||||
|
bucket := fmt.Sprintf("fed-token-test-%d", time.Now().UnixNano())
|
||||||
|
|
||||||
|
_, err = adminS3.CreateBucket(&s3.CreateBucketInput{Bucket: aws.String(bucket)})
|
||||||
|
require.NoError(t, err)
|
||||||
|
defer adminS3.DeleteBucket(&s3.DeleteBucketInput{Bucket: aws.String(bucket)})
|
||||||
|
|
||||||
|
_, err = adminS3.PutObject(&s3.PutObjectInput{
|
||||||
|
Bucket: aws.String(bucket),
|
||||||
|
Key: aws.String("test.txt"),
|
||||||
|
Body: strings.NewReader("hello"),
|
||||||
|
})
|
||||||
|
require.NoError(t, err)
|
||||||
|
defer adminS3.DeleteObject(&s3.DeleteObjectInput{Bucket: aws.String(bucket), Key: aws.String("test.txt")})
|
||||||
|
|
||||||
|
// Get federated credentials with a session policy that only allows GetObject
|
||||||
|
sessionPolicy := fmt.Sprintf(`{
|
||||||
|
"Version": "2012-10-17",
|
||||||
|
"Statement": [{
|
||||||
|
"Effect": "Allow",
|
||||||
|
"Action": ["s3:GetObject"],
|
||||||
|
"Resource": ["arn:aws:s3:::%s/*"]
|
||||||
|
}]
|
||||||
|
}`, bucket)
|
||||||
|
|
||||||
|
resp, err := callSTSAPIWithSigV4(t, url.Values{
|
||||||
|
"Action": {"GetFederationToken"},
|
||||||
|
"Version": {"2011-06-15"},
|
||||||
|
"Name": {"ScopedClient"},
|
||||||
|
"Policy": {sessionPolicy},
|
||||||
|
}, accessKey, secretKey)
|
||||||
|
require.NoError(t, err)
|
||||||
|
defer resp.Body.Close()
|
||||||
|
|
||||||
|
body, _ := io.ReadAll(resp.Body)
|
||||||
|
t.Logf("GetFederationToken response: status=%d body=%s", resp.StatusCode, string(body))
|
||||||
|
|
||||||
|
if resp.StatusCode != http.StatusOK {
|
||||||
|
t.Skipf("GetFederationToken failed (may need IAM policy config): %s", string(body))
|
||||||
|
}
|
||||||
|
|
||||||
|
var stsResp GetFederationTokenTestResponse
|
||||||
|
require.NoError(t, xml.Unmarshal(body, &stsResp))
|
||||||
|
|
||||||
|
fedCreds := stsResp.Result.Credentials
|
||||||
|
require.NotEmpty(t, fedCreds.AccessKeyId)
|
||||||
|
require.NotEmpty(t, fedCreds.SessionToken)
|
||||||
|
|
||||||
|
// Create S3 client with the federated credentials
|
||||||
|
fedSess, err := session.NewSession(&aws.Config{
|
||||||
|
Region: aws.String("us-east-1"),
|
||||||
|
Endpoint: aws.String(TestSTSEndpoint),
|
||||||
|
DisableSSL: aws.Bool(true),
|
||||||
|
S3ForcePathStyle: aws.Bool(true),
|
||||||
|
Credentials: credentials.NewStaticCredentials(
|
||||||
|
fedCreds.AccessKeyId, fedCreds.SecretAccessKey, fedCreds.SessionToken),
|
||||||
|
})
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
fedS3 := s3.New(fedSess)
|
||||||
|
|
||||||
|
// GetObject should succeed (allowed by session policy)
|
||||||
|
getResp, err := fedS3.GetObject(&s3.GetObjectInput{
|
||||||
|
Bucket: aws.String(bucket),
|
||||||
|
Key: aws.String("test.txt"),
|
||||||
|
})
|
||||||
|
if err == nil {
|
||||||
|
defer getResp.Body.Close()
|
||||||
|
t.Log("GetObject with federated credentials succeeded (as expected)")
|
||||||
|
} else {
|
||||||
|
t.Logf("GetObject with federated credentials: %v (session policy enforcement may vary)", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// PutObject should be denied (not allowed by session policy)
|
||||||
|
_, err = fedS3.PutObject(&s3.PutObjectInput{
|
||||||
|
Bucket: aws.String(bucket),
|
||||||
|
Key: aws.String("denied.txt"),
|
||||||
|
Body: strings.NewReader("should fail"),
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
t.Log("PutObject correctly denied with federated credentials")
|
||||||
|
assert.Contains(t, err.Error(), "AccessDenied",
|
||||||
|
"PutObject should be denied by session policy")
|
||||||
|
} else {
|
||||||
|
// Clean up if unexpectedly succeeded
|
||||||
|
adminS3.DeleteObject(&s3.DeleteObjectInput{Bucket: aws.String(bucket), Key: aws.String("denied.txt")})
|
||||||
|
t.Log("PutObject unexpectedly succeeded — session policy enforcement may not be active")
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -725,6 +725,23 @@ func (m *IAMManager) ExpireSessionForTesting(ctx context.Context, sessionToken s
|
|||||||
return m.stsService.ExpireSessionForTesting(ctx, sessionToken)
|
return m.stsService.ExpireSessionForTesting(ctx, sessionToken)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// GetPoliciesForUser returns the policy names attached to an IAM user.
|
||||||
|
// Returns an error if the user store is not configured or the lookup fails,
|
||||||
|
// so callers can fail closed on policy-resolution failures.
|
||||||
|
func (m *IAMManager) GetPoliciesForUser(ctx context.Context, username string) ([]string, error) {
|
||||||
|
if m.userStore == nil {
|
||||||
|
return nil, fmt.Errorf("user store not configured")
|
||||||
|
}
|
||||||
|
user, err := m.userStore.GetUser(ctx, username)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("failed to look up user %q: %w", username, err)
|
||||||
|
}
|
||||||
|
if user == nil {
|
||||||
|
return nil, nil
|
||||||
|
}
|
||||||
|
return user.PolicyNames, nil
|
||||||
|
}
|
||||||
|
|
||||||
// GetSTSService returns the STS service instance
|
// GetSTSService returns the STS service instance
|
||||||
func (m *IAMManager) GetSTSService() *sts.STSService {
|
func (m *IAMManager) GetSTSService() *sts.STSService {
|
||||||
return m.stsService
|
return m.stsService
|
||||||
|
|||||||
@@ -124,6 +124,7 @@ const (
|
|||||||
ActionAssumeRole = "sts:AssumeRole"
|
ActionAssumeRole = "sts:AssumeRole"
|
||||||
ActionAssumeRoleWithWebIdentity = "sts:AssumeRoleWithWebIdentity"
|
ActionAssumeRoleWithWebIdentity = "sts:AssumeRoleWithWebIdentity"
|
||||||
ActionAssumeRoleWithCredentials = "sts:AssumeRoleWithCredentials"
|
ActionAssumeRoleWithCredentials = "sts:AssumeRoleWithCredentials"
|
||||||
|
ActionGetFederationToken = "sts:GetFederationToken"
|
||||||
ActionValidateSession = "sts:ValidateSession"
|
ActionValidateSession = "sts:ValidateSession"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|||||||
@@ -296,8 +296,8 @@ func resolveBucketLevelAction(method string, baseAction string) string {
|
|||||||
// mapBaseActionToS3Format converts coarse-grained base actions to S3 format
|
// mapBaseActionToS3Format converts coarse-grained base actions to S3 format
|
||||||
// This is the fallback when no specific resolution is found
|
// This is the fallback when no specific resolution is found
|
||||||
func mapBaseActionToS3Format(baseAction string) string {
|
func mapBaseActionToS3Format(baseAction string) string {
|
||||||
// Handle actions that already have s3: or iam: prefix
|
// Handle actions that already have a known service prefix
|
||||||
if strings.HasPrefix(baseAction, "s3:") || strings.HasPrefix(baseAction, "iam:") {
|
if strings.HasPrefix(baseAction, "s3:") || strings.HasPrefix(baseAction, "iam:") || strings.HasPrefix(baseAction, "sts:") {
|
||||||
return baseAction
|
return baseAction
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -7,6 +7,59 @@ import (
|
|||||||
"github.com/seaweedfs/seaweedfs/weed/s3api/s3_constants"
|
"github.com/seaweedfs/seaweedfs/weed/s3api/s3_constants"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
// TestMapBaseActionToS3Format_ServicePrefixPassthrough verifies that actions
|
||||||
|
// with known service prefixes (s3:, iam:, sts:) are returned unchanged.
|
||||||
|
func TestMapBaseActionToS3Format_ServicePrefixPassthrough(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
input string
|
||||||
|
expect string
|
||||||
|
}{
|
||||||
|
{"s3 prefix", "s3:GetObject", "s3:GetObject"},
|
||||||
|
{"iam prefix", "iam:CreateUser", "iam:CreateUser"},
|
||||||
|
{"sts:AssumeRole", "sts:AssumeRole", "sts:AssumeRole"},
|
||||||
|
{"sts:GetFederationToken", "sts:GetFederationToken", "sts:GetFederationToken"},
|
||||||
|
{"sts:GetCallerIdentity", "sts:GetCallerIdentity", "sts:GetCallerIdentity"},
|
||||||
|
{"coarse Read maps to s3:GetObject", "Read", s3_constants.S3_ACTION_GET_OBJECT},
|
||||||
|
{"coarse Write maps to s3:PutObject", "Write", s3_constants.S3_ACTION_PUT_OBJECT},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
got := mapBaseActionToS3Format(tt.input)
|
||||||
|
if got != tt.expect {
|
||||||
|
t.Errorf("mapBaseActionToS3Format(%q) = %q, want %q", tt.input, got, tt.expect)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestResolveS3Action_STSActionsPassthrough verifies that STS actions flow
|
||||||
|
// through ResolveS3Action unchanged, both with and without an HTTP request.
|
||||||
|
func TestResolveS3Action_STSActionsPassthrough(t *testing.T) {
|
||||||
|
stsActions := []string{
|
||||||
|
"sts:AssumeRole",
|
||||||
|
"sts:GetFederationToken",
|
||||||
|
"sts:GetCallerIdentity",
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, action := range stsActions {
|
||||||
|
t.Run("nil_request_"+action, func(t *testing.T) {
|
||||||
|
got := ResolveS3Action(nil, action, "", "")
|
||||||
|
if got != action {
|
||||||
|
t.Errorf("ResolveS3Action(nil, %q) = %q, want %q", action, got, action)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
t.Run("with_request_"+action, func(t *testing.T) {
|
||||||
|
r, _ := http.NewRequest(http.MethodPost, "http://localhost/", nil)
|
||||||
|
got := ResolveS3Action(r, action, "", "")
|
||||||
|
if got != action {
|
||||||
|
t.Errorf("ResolveS3Action(r, %q) = %q, want %q", action, got, action)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func TestResolveS3Action_AttributesBeforeVersionId(t *testing.T) {
|
func TestResolveS3Action_AttributesBeforeVersionId(t *testing.T) {
|
||||||
tests := []struct {
|
tests := []struct {
|
||||||
name string
|
name string
|
||||||
|
|||||||
@@ -10,6 +10,7 @@ import (
|
|||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"net/http"
|
"net/http"
|
||||||
|
"regexp"
|
||||||
"strconv"
|
"strconv"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
@@ -37,6 +38,10 @@ const (
|
|||||||
actionAssumeRoleWithWebIdentity = "AssumeRoleWithWebIdentity"
|
actionAssumeRoleWithWebIdentity = "AssumeRoleWithWebIdentity"
|
||||||
actionAssumeRoleWithLDAPIdentity = "AssumeRoleWithLDAPIdentity"
|
actionAssumeRoleWithLDAPIdentity = "AssumeRoleWithLDAPIdentity"
|
||||||
actionGetCallerIdentity = "GetCallerIdentity"
|
actionGetCallerIdentity = "GetCallerIdentity"
|
||||||
|
actionGetFederationToken = "GetFederationToken"
|
||||||
|
|
||||||
|
// GetFederationToken-specific parameters
|
||||||
|
stsFederationName = "Name"
|
||||||
|
|
||||||
// LDAP parameter names
|
// LDAP parameter names
|
||||||
stsLDAPUsername = "LDAPUsername"
|
stsLDAPUsername = "LDAPUsername"
|
||||||
@@ -44,15 +49,20 @@ const (
|
|||||||
stsLDAPProviderName = "LDAPProviderName"
|
stsLDAPProviderName = "LDAPProviderName"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
// federationNameRegex validates the Name parameter for GetFederationToken per AWS spec
|
||||||
|
var federationNameRegex = regexp.MustCompile(`^[\w+=,.@-]+$`)
|
||||||
|
|
||||||
// STS duration constants (AWS specification)
|
// STS duration constants (AWS specification)
|
||||||
const (
|
const (
|
||||||
minDurationSeconds = int64(900) // 15 minutes
|
minDurationSeconds = int64(900) // 15 minutes
|
||||||
maxDurationSeconds = int64(43200) // 12 hours
|
maxDurationSeconds = int64(43200) // 12 hours (AssumeRole)
|
||||||
|
defaultFederationDurationSeconds = int64(43200) // 12 hours (GetFederationToken default)
|
||||||
|
maxFederationDurationSeconds = int64(129600) // 36 hours (GetFederationToken max)
|
||||||
)
|
)
|
||||||
|
|
||||||
// parseDurationSeconds parses and validates the DurationSeconds parameter
|
// parseDurationSecondsWithBounds parses and validates the DurationSeconds parameter
|
||||||
// Returns nil if the parameter is not provided, or a pointer to the parsed value
|
// against the given min and max bounds. Returns nil if the parameter is not provided.
|
||||||
func parseDurationSeconds(r *http.Request) (*int64, STSErrorCode, error) {
|
func parseDurationSecondsWithBounds(r *http.Request, minSec, maxSec int64) (*int64, STSErrorCode, error) {
|
||||||
dsStr := r.FormValue("DurationSeconds")
|
dsStr := r.FormValue("DurationSeconds")
|
||||||
if dsStr == "" {
|
if dsStr == "" {
|
||||||
return nil, "", nil
|
return nil, "", nil
|
||||||
@@ -63,14 +73,19 @@ func parseDurationSeconds(r *http.Request) (*int64, STSErrorCode, error) {
|
|||||||
return nil, STSErrInvalidParameterValue, fmt.Errorf("invalid DurationSeconds: %w", err)
|
return nil, STSErrInvalidParameterValue, fmt.Errorf("invalid DurationSeconds: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
if ds < minDurationSeconds || ds > maxDurationSeconds {
|
if ds < minSec || ds > maxSec {
|
||||||
return nil, STSErrInvalidParameterValue,
|
return nil, STSErrInvalidParameterValue,
|
||||||
fmt.Errorf("DurationSeconds must be between %d and %d seconds", minDurationSeconds, maxDurationSeconds)
|
fmt.Errorf("DurationSeconds must be between %d and %d seconds", minSec, maxSec)
|
||||||
}
|
}
|
||||||
|
|
||||||
return &ds, "", nil
|
return &ds, "", nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// parseDurationSeconds parses DurationSeconds for AssumeRole (15 min to 12 hours)
|
||||||
|
func parseDurationSeconds(r *http.Request) (*int64, STSErrorCode, error) {
|
||||||
|
return parseDurationSecondsWithBounds(r, minDurationSeconds, maxDurationSeconds)
|
||||||
|
}
|
||||||
|
|
||||||
// Removed generateSecureCredentials - now using STS service's JWT token generation
|
// Removed generateSecureCredentials - now using STS service's JWT token generation
|
||||||
// The STS service generates proper JWT tokens with embedded claims that can be validated
|
// The STS service generates proper JWT tokens with embedded claims that can be validated
|
||||||
// across distributed instances without shared state.
|
// across distributed instances without shared state.
|
||||||
@@ -124,6 +139,8 @@ func (h *STSHandlers) HandleSTSRequest(w http.ResponseWriter, r *http.Request) {
|
|||||||
h.handleAssumeRoleWithLDAPIdentity(w, r)
|
h.handleAssumeRoleWithLDAPIdentity(w, r)
|
||||||
case actionGetCallerIdentity:
|
case actionGetCallerIdentity:
|
||||||
h.handleGetCallerIdentity(w, r)
|
h.handleGetCallerIdentity(w, r)
|
||||||
|
case actionGetFederationToken:
|
||||||
|
h.handleGetFederationToken(w, r)
|
||||||
default:
|
default:
|
||||||
h.writeSTSErrorResponse(w, r, STSErrInvalidAction,
|
h.writeSTSErrorResponse(w, r, STSErrInvalidAction,
|
||||||
fmt.Errorf("unsupported action: %s", action))
|
fmt.Errorf("unsupported action: %s", action))
|
||||||
@@ -296,7 +313,7 @@ func (h *STSHandlers) handleAssumeRole(w http.ResponseWriter, r *http.Request) {
|
|||||||
// Check authorizations
|
// Check authorizations
|
||||||
if roleArn != "" {
|
if roleArn != "" {
|
||||||
// Check if the caller is authorized to assume the role (sts:AssumeRole permission)
|
// Check if the caller is authorized to assume the role (sts:AssumeRole permission)
|
||||||
if authErr := h.iam.VerifyActionPermission(r, identity, Action("sts:AssumeRole"), "", roleArn); authErr != s3err.ErrNone {
|
if authErr := h.iam.VerifyActionPermission(r, identity, Action(sts.ActionAssumeRole), "", roleArn); authErr != s3err.ErrNone {
|
||||||
glog.V(2).Infof("AssumeRole: caller %s is not authorized to assume role %s", identity.Name, roleArn)
|
glog.V(2).Infof("AssumeRole: caller %s is not authorized to assume role %s", identity.Name, roleArn)
|
||||||
h.writeSTSErrorResponse(w, r, STSErrAccessDenied,
|
h.writeSTSErrorResponse(w, r, STSErrAccessDenied,
|
||||||
fmt.Errorf("user %s is not authorized to assume role %s", identity.Name, roleArn))
|
fmt.Errorf("user %s is not authorized to assume role %s", identity.Name, roleArn))
|
||||||
@@ -320,7 +337,7 @@ func (h *STSHandlers) handleAssumeRole(w http.ResponseWriter, r *http.Request) {
|
|||||||
// For safety/consistency with previous logic, we keep the check but strictly it might not be required by AWS for GetSessionToken.
|
// For safety/consistency with previous logic, we keep the check but strictly it might not be required by AWS for GetSessionToken.
|
||||||
// But since this IS AssumeRole, let's keep it.
|
// But since this IS AssumeRole, let's keep it.
|
||||||
// Admin/Global check when no specific role is requested
|
// Admin/Global check when no specific role is requested
|
||||||
if authErr := h.iam.VerifyActionPermission(r, identity, Action("sts:AssumeRole"), "", ""); authErr != s3err.ErrNone {
|
if authErr := h.iam.VerifyActionPermission(r, identity, Action(sts.ActionAssumeRole), "", ""); authErr != s3err.ErrNone {
|
||||||
glog.Warningf("AssumeRole: caller %s attempted to assume role without RoleArn and lacks global sts:AssumeRole permission", identity.Name)
|
glog.Warningf("AssumeRole: caller %s attempted to assume role without RoleArn and lacks global sts:AssumeRole permission", identity.Name)
|
||||||
h.writeSTSErrorResponse(w, r, STSErrAccessDenied, fmt.Errorf("access denied"))
|
h.writeSTSErrorResponse(w, r, STSErrAccessDenied, fmt.Errorf("access denied"))
|
||||||
return
|
return
|
||||||
@@ -505,6 +522,202 @@ func (h *STSHandlers) handleAssumeRoleWithLDAPIdentity(w http.ResponseWriter, r
|
|||||||
s3err.WriteXMLResponse(w, r, http.StatusOK, xmlResponse)
|
s3err.WriteXMLResponse(w, r, http.StatusOK, xmlResponse)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// handleGetFederationToken handles the GetFederationToken API action.
|
||||||
|
// This allows long-term IAM users to obtain temporary credentials scoped down
|
||||||
|
// by an optional inline session policy. Temporary credentials cannot call this action.
|
||||||
|
func (h *STSHandlers) handleGetFederationToken(w http.ResponseWriter, r *http.Request) {
|
||||||
|
// Extract parameters
|
||||||
|
name := r.FormValue(stsFederationName)
|
||||||
|
|
||||||
|
// Validate required parameters
|
||||||
|
if name == "" {
|
||||||
|
h.writeSTSErrorResponse(w, r, STSErrMissingParameter,
|
||||||
|
fmt.Errorf("Name is required"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// AWS requires Name to be 2-32 characters matching [\w+=,.@-]+
|
||||||
|
if len(name) < 2 || len(name) > 32 {
|
||||||
|
h.writeSTSErrorResponse(w, r, STSErrInvalidParameterValue,
|
||||||
|
fmt.Errorf("Name must be between 2 and 32 characters"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if !federationNameRegex.MatchString(name) {
|
||||||
|
h.writeSTSErrorResponse(w, r, STSErrInvalidParameterValue,
|
||||||
|
fmt.Errorf("Name contains invalid characters, must match [\\w+=,.@-]+"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Parse and validate DurationSeconds (GetFederationToken allows up to 36 hours)
|
||||||
|
durationSeconds, errCode, err := parseDurationSecondsWithBounds(r, minDurationSeconds, maxFederationDurationSeconds)
|
||||||
|
if err != nil {
|
||||||
|
h.writeSTSErrorResponse(w, r, errCode, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Reject calls from temporary credentials (session tokens) early,
|
||||||
|
// before SigV4 verification — no need to authenticate first.
|
||||||
|
// GetFederationToken can only be called by long-term IAM users.
|
||||||
|
securityToken := r.Header.Get("X-Amz-Security-Token")
|
||||||
|
if securityToken == "" {
|
||||||
|
securityToken = r.URL.Query().Get("X-Amz-Security-Token")
|
||||||
|
}
|
||||||
|
if securityToken != "" {
|
||||||
|
h.writeSTSErrorResponse(w, r, STSErrAccessDenied,
|
||||||
|
fmt.Errorf("GetFederationToken cannot be called with temporary credentials"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check if STS service is initialized
|
||||||
|
if h.stsService == nil || !h.stsService.IsInitialized() {
|
||||||
|
h.writeSTSErrorResponse(w, r, STSErrSTSNotReady,
|
||||||
|
fmt.Errorf("STS service not initialized"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check if IAM is available for SigV4 verification
|
||||||
|
if h.iam == nil {
|
||||||
|
h.writeSTSErrorResponse(w, r, STSErrSTSNotReady,
|
||||||
|
fmt.Errorf("IAM not configured for STS"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Validate AWS SigV4 authentication
|
||||||
|
identity, _, _, _, sigErrCode := h.iam.verifyV4Signature(r, false)
|
||||||
|
if sigErrCode != s3err.ErrNone {
|
||||||
|
glog.V(2).Infof("GetFederationToken SigV4 verification failed: %v", sigErrCode)
|
||||||
|
h.writeSTSErrorResponse(w, r, STSErrAccessDenied,
|
||||||
|
fmt.Errorf("invalid AWS signature: %v", sigErrCode))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if identity == nil {
|
||||||
|
h.writeSTSErrorResponse(w, r, STSErrAccessDenied,
|
||||||
|
fmt.Errorf("unable to identify caller"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
glog.V(2).Infof("GetFederationToken: caller identity=%s, name=%s", identity.Name, name)
|
||||||
|
|
||||||
|
// Check if the caller is authorized to call GetFederationToken
|
||||||
|
if authErr := h.iam.VerifyActionPermission(r, identity, Action(sts.ActionGetFederationToken), "", ""); authErr != s3err.ErrNone {
|
||||||
|
glog.V(2).Infof("GetFederationToken: caller %s is not authorized to call GetFederationToken", identity.Name)
|
||||||
|
h.writeSTSErrorResponse(w, r, STSErrAccessDenied,
|
||||||
|
fmt.Errorf("user %s is not authorized to call GetFederationToken", identity.Name))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Validate session policy if provided
|
||||||
|
sessionPolicyJSON, err := sts.NormalizeSessionPolicy(r.FormValue("Policy"))
|
||||||
|
if err != nil {
|
||||||
|
h.writeSTSErrorResponse(w, r, STSErrMalformedPolicyDocument,
|
||||||
|
fmt.Errorf("invalid Policy document: %w", err))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Calculate duration (default 12 hours for GetFederationToken)
|
||||||
|
duration := time.Duration(defaultFederationDurationSeconds) * time.Second
|
||||||
|
if durationSeconds != nil {
|
||||||
|
duration = time.Duration(*durationSeconds) * time.Second
|
||||||
|
}
|
||||||
|
|
||||||
|
// Generate session ID
|
||||||
|
sessionId, err := sts.GenerateSessionId()
|
||||||
|
if err != nil {
|
||||||
|
h.writeSTSErrorResponse(w, r, STSErrInternalError,
|
||||||
|
fmt.Errorf("failed to generate session ID: %w", err))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
expiration := time.Now().Add(duration)
|
||||||
|
accountID := h.getAccountID()
|
||||||
|
|
||||||
|
// Build federated user ARN: arn:aws:sts::<account>:federated-user/<Name>
|
||||||
|
federatedUserArn := fmt.Sprintf("arn:aws:sts::%s:federated-user/%s", accountID, name)
|
||||||
|
federatedUserId := fmt.Sprintf("%s:%s", accountID, name)
|
||||||
|
|
||||||
|
// Create session claims — use the caller's principal ARN as the RoleArn
|
||||||
|
// so that policy evaluation resolves the caller's attached policies
|
||||||
|
claims := sts.NewSTSSessionClaims(sessionId, h.stsService.Config.Issuer, expiration).
|
||||||
|
WithSessionName(name).
|
||||||
|
WithRoleInfo(identity.PrincipalArn, federatedUserId, federatedUserArn)
|
||||||
|
|
||||||
|
// Embed the caller's effective policies into the token.
|
||||||
|
// Merge identity.PolicyNames (from SigV4 identity) with policies resolved
|
||||||
|
// from the IAM manager (which may include group-attached policies).
|
||||||
|
policySet := make(map[string]struct{})
|
||||||
|
for _, p := range identity.PolicyNames {
|
||||||
|
policySet[p] = struct{}{}
|
||||||
|
}
|
||||||
|
|
||||||
|
var policyManager *integration.IAMManager
|
||||||
|
if h.iam.iamIntegration != nil {
|
||||||
|
if provider, ok := h.iam.iamIntegration.(IAMManagerProvider); ok {
|
||||||
|
policyManager = provider.GetIAMManager()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if policyManager != nil {
|
||||||
|
userPolicies, err := policyManager.GetPoliciesForUser(r.Context(), identity.Name)
|
||||||
|
if err != nil {
|
||||||
|
glog.V(2).Infof("GetFederationToken: failed to resolve policies for %s: %v", identity.Name, err)
|
||||||
|
h.writeSTSErrorResponse(w, r, STSErrInternalError,
|
||||||
|
fmt.Errorf("failed to resolve caller policies"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
for _, p := range userPolicies {
|
||||||
|
policySet[p] = struct{}{}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(policySet) > 0 {
|
||||||
|
merged := make([]string, 0, len(policySet))
|
||||||
|
for p := range policySet {
|
||||||
|
merged = append(merged, p)
|
||||||
|
}
|
||||||
|
claims.WithPolicies(merged)
|
||||||
|
}
|
||||||
|
|
||||||
|
if sessionPolicyJSON != "" {
|
||||||
|
claims.WithSessionPolicy(sessionPolicyJSON)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Generate JWT session token
|
||||||
|
sessionToken, err := h.stsService.GetTokenGenerator().GenerateJWTWithClaims(claims)
|
||||||
|
if err != nil {
|
||||||
|
h.writeSTSErrorResponse(w, r, STSErrInternalError,
|
||||||
|
fmt.Errorf("failed to generate session token: %w", err))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Generate temporary credentials
|
||||||
|
stsCredGen := sts.NewCredentialGenerator()
|
||||||
|
stsCredsDet, err := stsCredGen.GenerateTemporaryCredentials(sessionId, expiration)
|
||||||
|
if err != nil {
|
||||||
|
h.writeSTSErrorResponse(w, r, STSErrInternalError,
|
||||||
|
fmt.Errorf("failed to generate temporary credentials: %w", err))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Build and return response
|
||||||
|
xmlResponse := &GetFederationTokenResponse{
|
||||||
|
Result: GetFederationTokenResult{
|
||||||
|
Credentials: STSCredentials{
|
||||||
|
AccessKeyId: stsCredsDet.AccessKeyId,
|
||||||
|
SecretAccessKey: stsCredsDet.SecretAccessKey,
|
||||||
|
SessionToken: sessionToken,
|
||||||
|
Expiration: expiration.Format(time.RFC3339),
|
||||||
|
},
|
||||||
|
FederatedUser: FederatedUser{
|
||||||
|
FederatedUserId: federatedUserId,
|
||||||
|
Arn: federatedUserArn,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
xmlResponse.ResponseMetadata.RequestId = request_id.GetFromRequest(r)
|
||||||
|
|
||||||
|
s3err.WriteXMLResponse(w, r, http.StatusOK, xmlResponse)
|
||||||
|
}
|
||||||
|
|
||||||
// prepareSTSCredentials extracts common shared logic for credential generation
|
// prepareSTSCredentials extracts common shared logic for credential generation
|
||||||
func (h *STSHandlers) prepareSTSCredentials(ctx context.Context, roleArn, roleSessionName string,
|
func (h *STSHandlers) prepareSTSCredentials(ctx context.Context, roleArn, roleSessionName string,
|
||||||
durationSeconds *int64, sessionPolicy string, modifyClaims func(*sts.STSSessionClaims)) (STSCredentials, *AssumedRoleUser, error) {
|
durationSeconds *int64, sessionPolicy string, modifyClaims func(*sts.STSSessionClaims)) (STSCredentials, *AssumedRoleUser, error) {
|
||||||
@@ -743,6 +956,27 @@ type GetCallerIdentityResult struct {
|
|||||||
Account string `xml:"Account"`
|
Account string `xml:"Account"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// GetFederationTokenResponse is the response for GetFederationToken
|
||||||
|
type GetFederationTokenResponse struct {
|
||||||
|
XMLName xml.Name `xml:"https://sts.amazonaws.com/doc/2011-06-15/ GetFederationTokenResponse"`
|
||||||
|
Result GetFederationTokenResult `xml:"GetFederationTokenResult"`
|
||||||
|
ResponseMetadata struct {
|
||||||
|
RequestId string `xml:"RequestId,omitempty"`
|
||||||
|
} `xml:"ResponseMetadata,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetFederationTokenResult contains the result of GetFederationToken
|
||||||
|
type GetFederationTokenResult struct {
|
||||||
|
Credentials STSCredentials `xml:"Credentials"`
|
||||||
|
FederatedUser FederatedUser `xml:"FederatedUser"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// FederatedUser contains information about the federated user
|
||||||
|
type FederatedUser struct {
|
||||||
|
FederatedUserId string `xml:"FederatedUserId"`
|
||||||
|
Arn string `xml:"Arn"`
|
||||||
|
}
|
||||||
|
|
||||||
// STS Error types
|
// STS Error types
|
||||||
|
|
||||||
// STSErrorCode represents STS error codes
|
// STSErrorCode represents STS error codes
|
||||||
|
|||||||
746
weed/s3api/s3api_sts_get_federation_token_test.go
Normal file
746
weed/s3api/s3api_sts_get_federation_token_test.go
Normal file
@@ -0,0 +1,746 @@
|
|||||||
|
package s3api
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"encoding/xml"
|
||||||
|
"fmt"
|
||||||
|
"net/http"
|
||||||
|
"net/http/httptest"
|
||||||
|
"net/url"
|
||||||
|
"sort"
|
||||||
|
"strings"
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/seaweedfs/seaweedfs/weed/iam/integration"
|
||||||
|
"github.com/seaweedfs/seaweedfs/weed/iam/policy"
|
||||||
|
"github.com/seaweedfs/seaweedfs/weed/iam/sts"
|
||||||
|
"github.com/seaweedfs/seaweedfs/weed/pb/iam_pb"
|
||||||
|
"github.com/stretchr/testify/assert"
|
||||||
|
"github.com/stretchr/testify/require"
|
||||||
|
)
|
||||||
|
|
||||||
|
// mockUserStore implements integration.UserStore for testing GetPoliciesForUser
|
||||||
|
type mockUserStore struct {
|
||||||
|
users map[string]*iam_pb.Identity
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *mockUserStore) GetUser(_ context.Context, username string) (*iam_pb.Identity, error) {
|
||||||
|
u, ok := m.users[username]
|
||||||
|
if !ok {
|
||||||
|
return nil, nil
|
||||||
|
}
|
||||||
|
return u, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestGetFederationToken_BasicFlow tests basic credential generation for GetFederationToken
|
||||||
|
func TestGetFederationToken_BasicFlow(t *testing.T) {
|
||||||
|
stsService, _ := setupTestSTSService(t)
|
||||||
|
|
||||||
|
iam := &IdentityAccessManagement{
|
||||||
|
iamIntegration: &MockIAMIntegration{},
|
||||||
|
}
|
||||||
|
stsHandlers := NewSTSHandlers(stsService, iam)
|
||||||
|
|
||||||
|
// Simulate the core logic of handleGetFederationToken
|
||||||
|
name := "BobApp"
|
||||||
|
callerIdentity := &Identity{
|
||||||
|
Name: "alice",
|
||||||
|
PrincipalArn: fmt.Sprintf("arn:aws:iam::%s:user/alice", defaultAccountID),
|
||||||
|
PolicyNames: []string{"S3ReadPolicy"},
|
||||||
|
}
|
||||||
|
|
||||||
|
accountID := stsHandlers.getAccountID()
|
||||||
|
|
||||||
|
// Generate session ID and credentials
|
||||||
|
sessionId, err := sts.GenerateSessionId()
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
expiration := time.Now().Add(12 * time.Hour)
|
||||||
|
federatedUserArn := fmt.Sprintf("arn:aws:sts::%s:federated-user/%s", accountID, name)
|
||||||
|
federatedUserId := fmt.Sprintf("%s:%s", accountID, name)
|
||||||
|
|
||||||
|
claims := sts.NewSTSSessionClaims(sessionId, stsService.Config.Issuer, expiration).
|
||||||
|
WithSessionName(name).
|
||||||
|
WithRoleInfo(callerIdentity.PrincipalArn, federatedUserId, federatedUserArn).
|
||||||
|
WithPolicies(callerIdentity.PolicyNames)
|
||||||
|
|
||||||
|
sessionToken, err := stsService.GetTokenGenerator().GenerateJWTWithClaims(claims)
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
// Validate the session token
|
||||||
|
sessionInfo, err := stsService.ValidateSessionToken(context.Background(), sessionToken)
|
||||||
|
require.NoError(t, err)
|
||||||
|
require.NotNil(t, sessionInfo)
|
||||||
|
|
||||||
|
// Verify the session info contains caller's policies
|
||||||
|
assert.Equal(t, []string{"S3ReadPolicy"}, sessionInfo.Policies)
|
||||||
|
|
||||||
|
// Verify principal is the federated user ARN
|
||||||
|
assert.Equal(t, federatedUserArn, sessionInfo.Principal)
|
||||||
|
|
||||||
|
// Verify the RoleArn points to the caller's identity (for policy resolution)
|
||||||
|
assert.Equal(t, callerIdentity.PrincipalArn, sessionInfo.RoleArn)
|
||||||
|
|
||||||
|
// Verify session name
|
||||||
|
assert.Equal(t, name, sessionInfo.SessionName)
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestGetFederationToken_WithSessionPolicy tests session policy scoping
|
||||||
|
func TestGetFederationToken_WithSessionPolicy(t *testing.T) {
|
||||||
|
stsService, _ := setupTestSTSService(t)
|
||||||
|
|
||||||
|
stsHandlers := NewSTSHandlers(stsService, &IdentityAccessManagement{
|
||||||
|
iamIntegration: &MockIAMIntegration{},
|
||||||
|
})
|
||||||
|
|
||||||
|
accountID := stsHandlers.getAccountID()
|
||||||
|
name := "ScopedApp"
|
||||||
|
|
||||||
|
sessionPolicyJSON := `{"Version":"2012-10-17","Statement":[{"Effect":"Allow","Action":["s3:GetObject"],"Resource":["arn:aws:s3:::my-bucket/*"]}]}`
|
||||||
|
normalizedPolicy, err := sts.NormalizeSessionPolicy(sessionPolicyJSON)
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
sessionId, err := sts.GenerateSessionId()
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
expiration := time.Now().Add(12 * time.Hour)
|
||||||
|
federatedUserArn := fmt.Sprintf("arn:aws:sts::%s:federated-user/%s", accountID, name)
|
||||||
|
federatedUserId := fmt.Sprintf("%s:%s", accountID, name)
|
||||||
|
|
||||||
|
claims := sts.NewSTSSessionClaims(sessionId, stsService.Config.Issuer, expiration).
|
||||||
|
WithSessionName(name).
|
||||||
|
WithRoleInfo("arn:aws:iam::000000000000:user/caller", federatedUserId, federatedUserArn).
|
||||||
|
WithPolicies([]string{"S3FullAccess"}).
|
||||||
|
WithSessionPolicy(normalizedPolicy)
|
||||||
|
|
||||||
|
sessionToken, err := stsService.GetTokenGenerator().GenerateJWTWithClaims(claims)
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
sessionInfo, err := stsService.ValidateSessionToken(context.Background(), sessionToken)
|
||||||
|
require.NoError(t, err)
|
||||||
|
require.NotNil(t, sessionInfo)
|
||||||
|
|
||||||
|
// Verify session policy is embedded
|
||||||
|
assert.NotEmpty(t, sessionInfo.SessionPolicy)
|
||||||
|
assert.Contains(t, sessionInfo.SessionPolicy, "s3:GetObject")
|
||||||
|
|
||||||
|
// Verify caller's policies are still present
|
||||||
|
assert.Equal(t, []string{"S3FullAccess"}, sessionInfo.Policies)
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestGetFederationToken_RejectTemporaryCredentials tests that requests with
|
||||||
|
// session tokens are rejected.
|
||||||
|
func TestGetFederationToken_RejectTemporaryCredentials(t *testing.T) {
|
||||||
|
stsService, _ := setupTestSTSService(t)
|
||||||
|
stsHandlers := NewSTSHandlers(stsService, &IdentityAccessManagement{
|
||||||
|
iamIntegration: &MockIAMIntegration{},
|
||||||
|
})
|
||||||
|
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
setToken func(r *http.Request)
|
||||||
|
description string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "SessionTokenInHeader",
|
||||||
|
setToken: func(r *http.Request) {
|
||||||
|
r.Header.Set("X-Amz-Security-Token", "some-session-token")
|
||||||
|
},
|
||||||
|
description: "Session token in X-Amz-Security-Token header should be rejected",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "SessionTokenInQuery",
|
||||||
|
setToken: func(r *http.Request) {
|
||||||
|
q := r.URL.Query()
|
||||||
|
q.Set("X-Amz-Security-Token", "some-session-token")
|
||||||
|
r.URL.RawQuery = q.Encode()
|
||||||
|
},
|
||||||
|
description: "Session token in query string should be rejected",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
form := url.Values{}
|
||||||
|
form.Set("Action", "GetFederationToken")
|
||||||
|
form.Set("Name", "TestUser")
|
||||||
|
form.Set("Version", "2011-06-15")
|
||||||
|
|
||||||
|
req := httptest.NewRequest("POST", "/", strings.NewReader(form.Encode()))
|
||||||
|
req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
|
||||||
|
tt.setToken(req)
|
||||||
|
|
||||||
|
// Parse form so the handler can read it
|
||||||
|
require.NoError(t, req.ParseForm())
|
||||||
|
// Re-set values after parse
|
||||||
|
req.Form.Set("Action", "GetFederationToken")
|
||||||
|
req.Form.Set("Name", "TestUser")
|
||||||
|
req.Form.Set("Version", "2011-06-15")
|
||||||
|
|
||||||
|
rr := httptest.NewRecorder()
|
||||||
|
stsHandlers.HandleSTSRequest(rr, req)
|
||||||
|
|
||||||
|
// The handler rejects temporary credentials before SigV4 verification
|
||||||
|
assert.Equal(t, http.StatusForbidden, rr.Code, tt.description)
|
||||||
|
assert.Contains(t, rr.Body.String(), "AccessDenied")
|
||||||
|
assert.Contains(t, rr.Body.String(), "cannot be called with temporary credentials")
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestGetFederationToken_MissingName tests that a missing Name parameter returns an error
|
||||||
|
func TestGetFederationToken_MissingName(t *testing.T) {
|
||||||
|
stsService, _ := setupTestSTSService(t)
|
||||||
|
stsHandlers := NewSTSHandlers(stsService, &IdentityAccessManagement{
|
||||||
|
iamIntegration: &MockIAMIntegration{},
|
||||||
|
})
|
||||||
|
|
||||||
|
req := httptest.NewRequest("POST", "/", nil)
|
||||||
|
req.Form = url.Values{}
|
||||||
|
req.Form.Set("Action", "GetFederationToken")
|
||||||
|
req.Form.Set("Version", "2011-06-15")
|
||||||
|
// Name is intentionally omitted
|
||||||
|
|
||||||
|
rr := httptest.NewRecorder()
|
||||||
|
stsHandlers.HandleSTSRequest(rr, req)
|
||||||
|
|
||||||
|
assert.Equal(t, http.StatusBadRequest, rr.Code)
|
||||||
|
assert.Contains(t, rr.Body.String(), "Name is required")
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestGetFederationToken_NameValidation tests Name parameter validation
|
||||||
|
func TestGetFederationToken_NameValidation(t *testing.T) {
|
||||||
|
stsService, _ := setupTestSTSService(t)
|
||||||
|
stsHandlers := NewSTSHandlers(stsService, &IdentityAccessManagement{
|
||||||
|
iamIntegration: &MockIAMIntegration{},
|
||||||
|
})
|
||||||
|
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
federName string
|
||||||
|
expectError bool
|
||||||
|
errContains string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "TooShort",
|
||||||
|
federName: "A",
|
||||||
|
expectError: true,
|
||||||
|
errContains: "between 2 and 32",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "TooLong",
|
||||||
|
federName: strings.Repeat("A", 33),
|
||||||
|
expectError: true,
|
||||||
|
errContains: "between 2 and 32",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "MinLength",
|
||||||
|
federName: "AB",
|
||||||
|
expectError: false,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "MaxLength",
|
||||||
|
federName: strings.Repeat("A", 32),
|
||||||
|
expectError: false,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "ValidSpecialChars",
|
||||||
|
federName: "user+=,.@-test",
|
||||||
|
expectError: false,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "InvalidChars_Space",
|
||||||
|
federName: "bad name",
|
||||||
|
expectError: true,
|
||||||
|
errContains: "invalid characters",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "InvalidChars_Slash",
|
||||||
|
federName: "bad/name",
|
||||||
|
expectError: true,
|
||||||
|
errContains: "invalid characters",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
req := httptest.NewRequest("POST", "/", nil)
|
||||||
|
req.Form = url.Values{}
|
||||||
|
req.Form.Set("Action", "GetFederationToken")
|
||||||
|
req.Form.Set("Name", tt.federName)
|
||||||
|
req.Form.Set("Version", "2011-06-15")
|
||||||
|
|
||||||
|
rr := httptest.NewRecorder()
|
||||||
|
stsHandlers.HandleSTSRequest(rr, req)
|
||||||
|
|
||||||
|
if tt.expectError {
|
||||||
|
assert.Equal(t, http.StatusBadRequest, rr.Code)
|
||||||
|
assert.Contains(t, rr.Body.String(), tt.errContains)
|
||||||
|
} else {
|
||||||
|
// Valid name should proceed past validation — will fail at SigV4
|
||||||
|
// (returns 403 because we have no real signature)
|
||||||
|
assert.NotEqual(t, http.StatusBadRequest, rr.Code,
|
||||||
|
"Valid name should not produce a 400 for name validation")
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestGetFederationToken_DurationValidation tests DurationSeconds validation
|
||||||
|
func TestGetFederationToken_DurationValidation(t *testing.T) {
|
||||||
|
stsService, _ := setupTestSTSService(t)
|
||||||
|
stsHandlers := NewSTSHandlers(stsService, &IdentityAccessManagement{
|
||||||
|
iamIntegration: &MockIAMIntegration{},
|
||||||
|
})
|
||||||
|
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
duration string
|
||||||
|
expectError bool
|
||||||
|
errContains string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "BelowMinimum",
|
||||||
|
duration: "899",
|
||||||
|
expectError: true,
|
||||||
|
errContains: "between",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "AboveMaximum",
|
||||||
|
duration: "129601",
|
||||||
|
expectError: true,
|
||||||
|
errContains: "between",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "InvalidFormat",
|
||||||
|
duration: "not-a-number",
|
||||||
|
expectError: true,
|
||||||
|
errContains: "invalid DurationSeconds",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "MinimumValid",
|
||||||
|
duration: "900",
|
||||||
|
expectError: false,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "MaximumValid_36Hours",
|
||||||
|
duration: "129600",
|
||||||
|
expectError: false,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Default12Hours",
|
||||||
|
duration: "43200",
|
||||||
|
expectError: false,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
req := httptest.NewRequest("POST", "/", nil)
|
||||||
|
req.Form = url.Values{}
|
||||||
|
req.Form.Set("Action", "GetFederationToken")
|
||||||
|
req.Form.Set("Name", "TestUser")
|
||||||
|
req.Form.Set("DurationSeconds", tt.duration)
|
||||||
|
req.Form.Set("Version", "2011-06-15")
|
||||||
|
|
||||||
|
rr := httptest.NewRecorder()
|
||||||
|
stsHandlers.HandleSTSRequest(rr, req)
|
||||||
|
|
||||||
|
if tt.expectError {
|
||||||
|
assert.Equal(t, http.StatusBadRequest, rr.Code)
|
||||||
|
assert.Contains(t, rr.Body.String(), tt.errContains)
|
||||||
|
} else {
|
||||||
|
// Valid duration should proceed past validation — will fail at SigV4
|
||||||
|
assert.NotEqual(t, http.StatusBadRequest, rr.Code,
|
||||||
|
"Valid duration should not produce a 400 for duration validation")
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestGetFederationToken_ResponseFormat tests the XML response structure
|
||||||
|
func TestGetFederationToken_ResponseFormat(t *testing.T) {
|
||||||
|
// Verify the response XML structure matches AWS format
|
||||||
|
response := GetFederationTokenResponse{
|
||||||
|
Result: GetFederationTokenResult{
|
||||||
|
Credentials: STSCredentials{
|
||||||
|
AccessKeyId: "ASIA1234567890",
|
||||||
|
SecretAccessKey: "secret123",
|
||||||
|
SessionToken: "token123",
|
||||||
|
Expiration: "2026-04-02T12:00:00Z",
|
||||||
|
},
|
||||||
|
FederatedUser: FederatedUser{
|
||||||
|
FederatedUserId: "000000000000:BobApp",
|
||||||
|
Arn: "arn:aws:sts::000000000000:federated-user/BobApp",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
response.ResponseMetadata.RequestId = "test-request-id"
|
||||||
|
|
||||||
|
data, err := xml.MarshalIndent(response, "", " ")
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
xmlStr := string(data)
|
||||||
|
assert.Contains(t, xmlStr, "GetFederationTokenResponse")
|
||||||
|
assert.Contains(t, xmlStr, "GetFederationTokenResult")
|
||||||
|
assert.Contains(t, xmlStr, "FederatedUser")
|
||||||
|
assert.Contains(t, xmlStr, "FederatedUserId")
|
||||||
|
assert.Contains(t, xmlStr, "federated-user/BobApp")
|
||||||
|
assert.Contains(t, xmlStr, "ASIA1234567890")
|
||||||
|
assert.Contains(t, xmlStr, "test-request-id")
|
||||||
|
|
||||||
|
// Verify it can be unmarshaled back
|
||||||
|
var parsed GetFederationTokenResponse
|
||||||
|
err = xml.Unmarshal(data, &parsed)
|
||||||
|
require.NoError(t, err)
|
||||||
|
assert.Equal(t, "ASIA1234567890", parsed.Result.Credentials.AccessKeyId)
|
||||||
|
assert.Equal(t, "arn:aws:sts::000000000000:federated-user/BobApp", parsed.Result.FederatedUser.Arn)
|
||||||
|
assert.Equal(t, "000000000000:BobApp", parsed.Result.FederatedUser.FederatedUserId)
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestGetFederationToken_PolicyEmbedding tests that the caller's policies are embedded
|
||||||
|
// into the session token using the IAM integration manager
|
||||||
|
func TestGetFederationToken_PolicyEmbedding(t *testing.T) {
|
||||||
|
ctx := context.Background()
|
||||||
|
manager := newTestSTSIntegrationManager(t)
|
||||||
|
|
||||||
|
// Create a policy that the user has attached
|
||||||
|
userPolicy := &policy.PolicyDocument{
|
||||||
|
Version: "2012-10-17",
|
||||||
|
Statement: []policy.Statement{
|
||||||
|
{
|
||||||
|
Effect: "Allow",
|
||||||
|
Action: []string{"s3:GetObject", "s3:PutObject"},
|
||||||
|
Resource: []string{"arn:aws:s3:::user-bucket/*"},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
require.NoError(t, manager.CreatePolicy(ctx, "", "UserS3Policy", userPolicy))
|
||||||
|
|
||||||
|
stsService := manager.GetSTSService()
|
||||||
|
|
||||||
|
// Simulate what handleGetFederationToken does for policy embedding
|
||||||
|
name := "AppClient"
|
||||||
|
callerPolicies := []string{"UserS3Policy"}
|
||||||
|
|
||||||
|
sessionId, err := sts.GenerateSessionId()
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
expiration := time.Now().Add(12 * time.Hour)
|
||||||
|
accountID := defaultAccountID
|
||||||
|
federatedUserArn := fmt.Sprintf("arn:aws:sts::%s:federated-user/%s", accountID, name)
|
||||||
|
federatedUserId := fmt.Sprintf("%s:%s", accountID, name)
|
||||||
|
|
||||||
|
claims := sts.NewSTSSessionClaims(sessionId, stsService.Config.Issuer, expiration).
|
||||||
|
WithSessionName(name).
|
||||||
|
WithRoleInfo("arn:aws:iam::000000000000:user/caller", federatedUserId, federatedUserArn).
|
||||||
|
WithPolicies(callerPolicies)
|
||||||
|
|
||||||
|
sessionToken, err := stsService.GetTokenGenerator().GenerateJWTWithClaims(claims)
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
sessionInfo, err := stsService.ValidateSessionToken(ctx, sessionToken)
|
||||||
|
require.NoError(t, err)
|
||||||
|
require.NotNil(t, sessionInfo)
|
||||||
|
|
||||||
|
// Verify the caller's policy names are embedded
|
||||||
|
assert.Equal(t, []string{"UserS3Policy"}, sessionInfo.Policies)
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestGetFederationToken_PolicyIntersection tests that both the caller's base policies
|
||||||
|
// and the restrictive session policy are embedded in the token, enabling the
|
||||||
|
// authorization layer to compute their intersection at request time.
|
||||||
|
func TestGetFederationToken_PolicyIntersection(t *testing.T) {
|
||||||
|
ctx := context.Background()
|
||||||
|
manager := newTestSTSIntegrationManager(t)
|
||||||
|
|
||||||
|
// Create a broad policy for the caller
|
||||||
|
broadPolicy := &policy.PolicyDocument{
|
||||||
|
Version: "2012-10-17",
|
||||||
|
Statement: []policy.Statement{
|
||||||
|
{
|
||||||
|
Effect: "Allow",
|
||||||
|
Action: []string{"s3:*"},
|
||||||
|
Resource: []string{"arn:aws:s3:::*", "arn:aws:s3:::*/*"},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
require.NoError(t, manager.CreatePolicy(ctx, "", "S3FullAccess", broadPolicy))
|
||||||
|
|
||||||
|
stsService := manager.GetSTSService()
|
||||||
|
|
||||||
|
// Session policy restricts to one bucket and one action
|
||||||
|
sessionPolicyJSON := `{"Version":"2012-10-17","Statement":[{"Effect":"Allow","Action":["s3:GetObject"],"Resource":["arn:aws:s3:::restricted-bucket/*"]}]}`
|
||||||
|
normalizedPolicy, err := sts.NormalizeSessionPolicy(sessionPolicyJSON)
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
sessionId, err := sts.GenerateSessionId()
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
expiration := time.Now().Add(12 * time.Hour)
|
||||||
|
name := "RestrictedApp"
|
||||||
|
accountID := defaultAccountID
|
||||||
|
federatedUserArn := fmt.Sprintf("arn:aws:sts::%s:federated-user/%s", accountID, name)
|
||||||
|
federatedUserId := fmt.Sprintf("%s:%s", accountID, name)
|
||||||
|
|
||||||
|
claims := sts.NewSTSSessionClaims(sessionId, stsService.Config.Issuer, expiration).
|
||||||
|
WithSessionName(name).
|
||||||
|
WithRoleInfo("arn:aws:iam::000000000000:user/caller", federatedUserId, federatedUserArn).
|
||||||
|
WithPolicies([]string{"S3FullAccess"}).
|
||||||
|
WithSessionPolicy(normalizedPolicy)
|
||||||
|
|
||||||
|
sessionToken, err := stsService.GetTokenGenerator().GenerateJWTWithClaims(claims)
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
sessionInfo, err := stsService.ValidateSessionToken(ctx, sessionToken)
|
||||||
|
require.NoError(t, err)
|
||||||
|
require.NotNil(t, sessionInfo)
|
||||||
|
|
||||||
|
// Verify both the broad base policies and the restrictive session policy are embedded
|
||||||
|
// The authorization layer computes intersection at request time
|
||||||
|
assert.Equal(t, []string{"S3FullAccess"}, sessionInfo.Policies,
|
||||||
|
"Caller's base policies should be embedded in token")
|
||||||
|
assert.Contains(t, sessionInfo.SessionPolicy, "restricted-bucket",
|
||||||
|
"Session policy should restrict to specific bucket")
|
||||||
|
assert.Contains(t, sessionInfo.SessionPolicy, "s3:GetObject",
|
||||||
|
"Session policy should restrict to specific action")
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestGetFederationToken_MalformedPolicy tests that invalid policy JSON is rejected
|
||||||
|
// by the session policy normalization used in the handler
|
||||||
|
func TestGetFederationToken_MalformedPolicy(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
policyStr string
|
||||||
|
expectErr bool
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "InvalidJSON",
|
||||||
|
policyStr: "not-valid-json",
|
||||||
|
expectErr: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "EmptyObject",
|
||||||
|
policyStr: "{}",
|
||||||
|
expectErr: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "TooLarge",
|
||||||
|
policyStr: `{"Version":"2012-10-17","Statement":[{"Effect":"Allow","Action":["s3:GetObject"],"Resource":["` + strings.Repeat("a", 2048) + `"]}]}`,
|
||||||
|
expectErr: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "ValidPolicy",
|
||||||
|
policyStr: `{"Version":"2012-10-17","Statement":[{"Effect":"Allow","Action":["s3:GetObject"],"Resource":["arn:aws:s3:::bucket/*"]}]}`,
|
||||||
|
expectErr: false,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "EmptyString",
|
||||||
|
policyStr: "",
|
||||||
|
expectErr: false,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
_, err := sts.NormalizeSessionPolicy(tt.policyStr)
|
||||||
|
if tt.expectErr {
|
||||||
|
assert.Error(t, err)
|
||||||
|
} else {
|
||||||
|
assert.NoError(t, err)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestGetFederationToken_STSNotReady tests that the handler returns 503 when STS is not initialized
|
||||||
|
func TestGetFederationToken_STSNotReady(t *testing.T) {
|
||||||
|
// Create handlers with nil STS service
|
||||||
|
stsHandlers := NewSTSHandlers(nil, &IdentityAccessManagement{
|
||||||
|
iamIntegration: &MockIAMIntegration{},
|
||||||
|
})
|
||||||
|
|
||||||
|
req := httptest.NewRequest("POST", "/", nil)
|
||||||
|
req.Form = url.Values{}
|
||||||
|
req.Form.Set("Action", "GetFederationToken")
|
||||||
|
req.Form.Set("Name", "TestUser")
|
||||||
|
req.Form.Set("Version", "2011-06-15")
|
||||||
|
|
||||||
|
rr := httptest.NewRecorder()
|
||||||
|
stsHandlers.HandleSTSRequest(rr, req)
|
||||||
|
|
||||||
|
assert.Equal(t, http.StatusServiceUnavailable, rr.Code)
|
||||||
|
assert.Contains(t, rr.Body.String(), "ServiceUnavailable")
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestGetFederationToken_DefaultDuration tests that the default duration is 12 hours
|
||||||
|
func TestGetFederationToken_DefaultDuration(t *testing.T) {
|
||||||
|
assert.Equal(t, int64(43200), defaultFederationDurationSeconds, "Default duration should be 12 hours (43200 seconds)")
|
||||||
|
assert.Equal(t, int64(129600), maxFederationDurationSeconds, "Max duration should be 36 hours (129600 seconds)")
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestGetFederationToken_GetPoliciesForUser tests that GetPoliciesForUser
|
||||||
|
// correctly resolves user policies from the UserStore and returns errors
|
||||||
|
// when the store is unavailable.
|
||||||
|
func TestGetFederationToken_GetPoliciesForUser(t *testing.T) {
|
||||||
|
ctx := context.Background()
|
||||||
|
manager := newTestSTSIntegrationManager(t)
|
||||||
|
|
||||||
|
t.Run("NoUserStore", func(t *testing.T) {
|
||||||
|
// UserStore not set — should return error
|
||||||
|
policies, err := manager.GetPoliciesForUser(ctx, "alice")
|
||||||
|
assert.Error(t, err)
|
||||||
|
assert.Nil(t, policies)
|
||||||
|
assert.Contains(t, err.Error(), "user store not configured")
|
||||||
|
})
|
||||||
|
|
||||||
|
t.Run("UserNotFound", func(t *testing.T) {
|
||||||
|
manager.SetUserStore(&mockUserStore{users: map[string]*iam_pb.Identity{}})
|
||||||
|
policies, err := manager.GetPoliciesForUser(ctx, "nonexistent")
|
||||||
|
assert.NoError(t, err)
|
||||||
|
assert.Nil(t, policies)
|
||||||
|
})
|
||||||
|
|
||||||
|
t.Run("UserWithPolicies", func(t *testing.T) {
|
||||||
|
manager.SetUserStore(&mockUserStore{
|
||||||
|
users: map[string]*iam_pb.Identity{
|
||||||
|
"alice": {
|
||||||
|
Name: "alice",
|
||||||
|
PolicyNames: []string{"GroupReadPolicy", "GroupWritePolicy"},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
})
|
||||||
|
policies, err := manager.GetPoliciesForUser(ctx, "alice")
|
||||||
|
assert.NoError(t, err)
|
||||||
|
assert.Equal(t, []string{"GroupReadPolicy", "GroupWritePolicy"}, policies)
|
||||||
|
})
|
||||||
|
|
||||||
|
t.Run("UserWithNoPolicies", func(t *testing.T) {
|
||||||
|
manager.SetUserStore(&mockUserStore{
|
||||||
|
users: map[string]*iam_pb.Identity{
|
||||||
|
"bob": {Name: "bob"},
|
||||||
|
},
|
||||||
|
})
|
||||||
|
policies, err := manager.GetPoliciesForUser(ctx, "bob")
|
||||||
|
assert.NoError(t, err)
|
||||||
|
assert.Empty(t, policies)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestGetFederationToken_PolicyMergeAndDedup tests that the handler's policy
|
||||||
|
// merge logic correctly combines identity.PolicyNames with IAM-manager-resolved
|
||||||
|
// policies and deduplicates the result.
|
||||||
|
func TestGetFederationToken_PolicyMergeAndDedup(t *testing.T) {
|
||||||
|
ctx := context.Background()
|
||||||
|
manager := newTestSTSIntegrationManager(t)
|
||||||
|
|
||||||
|
// Create policies so they exist in the engine
|
||||||
|
for _, name := range []string{"DirectPolicy", "GroupPolicy", "SharedPolicy"} {
|
||||||
|
require.NoError(t, manager.CreatePolicy(ctx, "", name, &policy.PolicyDocument{
|
||||||
|
Version: "2012-10-17",
|
||||||
|
Statement: []policy.Statement{
|
||||||
|
{Effect: "Allow", Action: []string{"s3:GetObject"}, Resource: []string{"arn:aws:s3:::*/*"}},
|
||||||
|
},
|
||||||
|
}))
|
||||||
|
}
|
||||||
|
|
||||||
|
// Set up a user store that returns group-attached policies
|
||||||
|
manager.SetUserStore(&mockUserStore{
|
||||||
|
users: map[string]*iam_pb.Identity{
|
||||||
|
"alice": {
|
||||||
|
Name: "alice",
|
||||||
|
PolicyNames: []string{"GroupPolicy", "SharedPolicy"},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
})
|
||||||
|
|
||||||
|
stsService := manager.GetSTSService()
|
||||||
|
|
||||||
|
// Simulate what the handler does: merge identity.PolicyNames with GetPoliciesForUser
|
||||||
|
identityPolicies := []string{"DirectPolicy", "SharedPolicy"} // SharedPolicy overlaps
|
||||||
|
|
||||||
|
policySet := make(map[string]struct{})
|
||||||
|
for _, p := range identityPolicies {
|
||||||
|
policySet[p] = struct{}{}
|
||||||
|
}
|
||||||
|
|
||||||
|
userPolicies, err := manager.GetPoliciesForUser(ctx, "alice")
|
||||||
|
require.NoError(t, err)
|
||||||
|
for _, p := range userPolicies {
|
||||||
|
policySet[p] = struct{}{}
|
||||||
|
}
|
||||||
|
|
||||||
|
merged := make([]string, 0, len(policySet))
|
||||||
|
for p := range policySet {
|
||||||
|
merged = append(merged, p)
|
||||||
|
}
|
||||||
|
sort.Strings(merged) // deterministic for assertion
|
||||||
|
|
||||||
|
// Should contain all three unique policies, no duplicates
|
||||||
|
assert.Equal(t, []string{"DirectPolicy", "GroupPolicy", "SharedPolicy"}, merged)
|
||||||
|
|
||||||
|
// Verify the merged policies can be embedded in a token and recovered
|
||||||
|
sessionId, err := sts.GenerateSessionId()
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
expiration := time.Now().Add(time.Hour)
|
||||||
|
claims := sts.NewSTSSessionClaims(sessionId, stsService.Config.Issuer, expiration).
|
||||||
|
WithSessionName("test").
|
||||||
|
WithRoleInfo("arn:aws:iam::000000000000:user/alice", "000000000000:test", "arn:aws:sts::000000000000:federated-user/test").
|
||||||
|
WithPolicies(merged)
|
||||||
|
|
||||||
|
token, err := stsService.GetTokenGenerator().GenerateJWTWithClaims(claims)
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
sessionInfo, err := stsService.ValidateSessionToken(ctx, token)
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
sort.Strings(sessionInfo.Policies)
|
||||||
|
assert.Equal(t, []string{"DirectPolicy", "GroupPolicy", "SharedPolicy"}, sessionInfo.Policies,
|
||||||
|
"Token should contain the deduplicated merge of identity and group policies")
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestGetFederationToken_PolicyMergeNoManager tests that when the IAM manager
|
||||||
|
// is unavailable, identity.PolicyNames alone are still embedded.
|
||||||
|
func TestGetFederationToken_PolicyMergeNoManager(t *testing.T) {
|
||||||
|
ctx := context.Background()
|
||||||
|
stsService, _ := setupTestSTSService(t)
|
||||||
|
|
||||||
|
// No IAM manager — only identity.PolicyNames should be used
|
||||||
|
identityPolicies := []string{"UserDirectPolicy"}
|
||||||
|
|
||||||
|
policySet := make(map[string]struct{})
|
||||||
|
for _, p := range identityPolicies {
|
||||||
|
policySet[p] = struct{}{}
|
||||||
|
}
|
||||||
|
|
||||||
|
// IAM manager is nil — skip GetPoliciesForUser (mirrors handler logic)
|
||||||
|
var policyManager *integration.IAMManager // nil
|
||||||
|
if policyManager != nil {
|
||||||
|
t.Fatal("policyManager should be nil in this test")
|
||||||
|
}
|
||||||
|
|
||||||
|
merged := make([]string, 0, len(policySet))
|
||||||
|
for p := range policySet {
|
||||||
|
merged = append(merged, p)
|
||||||
|
}
|
||||||
|
|
||||||
|
sessionId, err := sts.GenerateSessionId()
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
expiration := time.Now().Add(time.Hour)
|
||||||
|
claims := sts.NewSTSSessionClaims(sessionId, stsService.Config.Issuer, expiration).
|
||||||
|
WithSessionName("test").
|
||||||
|
WithRoleInfo("arn:aws:iam::000000000000:user/alice", "000000000000:test", "arn:aws:sts::000000000000:federated-user/test").
|
||||||
|
WithPolicies(merged)
|
||||||
|
|
||||||
|
token, err := stsService.GetTokenGenerator().GenerateJWTWithClaims(claims)
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
sessionInfo, err := stsService.ValidateSessionToken(ctx, token)
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
assert.Equal(t, []string{"UserDirectPolicy"}, sessionInfo.Policies,
|
||||||
|
"Without IAM manager, only identity policies should be embedded")
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user