propublica-mcp

from authlib.common.urls import is_valid_url from authlib.jose import BaseClaims from authlib.jose.errors import InvalidClaimError class ClientMetadataClaims(BaseClaims): REGISTERED_CLAIMS = [ "token_endpoint_auth_signing_alg", "application_type", "sector_identifier_uri", "subject_type", "id_token_signed_response_alg", "id_token_encrypted_response_alg", "id_token_encrypted_response_enc", "userinfo_signed_response_alg", "userinfo_encrypted_response_alg", "userinfo_encrypted_response_enc", "default_max_age", "require_auth_time", "default_acr_values", "initiate_login_uri", "request_object_signing_alg", "request_object_encryption_alg", "request_object_encryption_enc", "request_uris", ] def validate(self): self._validate_essential_claims() self.validate_token_endpoint_auth_signing_alg() self.validate_application_type() self.validate_sector_identifier_uri() self.validate_subject_type() self.validate_id_token_signed_response_alg() self.validate_id_token_encrypted_response_alg() self.validate_id_token_encrypted_response_enc() self.validate_userinfo_signed_response_alg() self.validate_userinfo_encrypted_response_alg() self.validate_userinfo_encrypted_response_enc() self.validate_default_max_age() self.validate_require_auth_time() self.validate_default_acr_values() self.validate_initiate_login_uri() self.validate_request_object_signing_alg() self.validate_request_object_encryption_alg() self.validate_request_object_encryption_enc() self.validate_request_uris() def _validate_uri(self, key): uri = self.get(key) uris = uri if isinstance(uri, list) else [uri] for uri in uris: if uri and not is_valid_url(uri): raise InvalidClaimError(key) @classmethod def get_claims_options(self, metadata): """Generate claims options validation from Authorization Server metadata.""" options = {} if acr_values_supported := metadata.get("acr_values_supported"): def _validate_default_acr_values(claims, value): return not value or set(value).issubset(set(acr_values_supported)) options["default_acr_values"] = {"validate": _validate_default_acr_values} values_mapping = { "token_endpoint_auth_signing_alg_values_supported": "token_endpoint_auth_signing_alg", "subject_types_supported": "subject_type", "id_token_signing_alg_values_supported": "id_token_signed_response_alg", "id_token_encryption_alg_values_supported": "id_token_encrypted_response_alg", "id_token_encryption_enc_values_supported": "id_token_encrypted_response_enc", "userinfo_signing_alg_values_supported": "userinfo_signed_response_alg", "userinfo_encryption_alg_values_supported": "userinfo_encrypted_response_alg", "userinfo_encryption_enc_values_supported": "userinfo_encrypted_response_enc", "request_object_signing_alg_values_supported": "request_object_signing_alg", "request_object_encryption_alg_values_supported": "request_object_encryption_alg", "request_object_encryption_enc_values_supported": "request_object_encryption_enc", } def make_validator(metadata_claim_values): def _validate(claims, value): return not value or value in metadata_claim_values return _validate for metadata_claim_name, request_claim_name in values_mapping.items(): if metadata_claim_values := metadata.get(metadata_claim_name): options[request_claim_name] = { "validate": make_validator(metadata_claim_values) } return options def validate_token_endpoint_auth_signing_alg(self): """JWS [JWS] alg algorithm [JWA] that MUST be used for signing the JWT [JWT] used to authenticate the Client at the Token Endpoint for the private_key_jwt and client_secret_jwt authentication methods. All Token Requests using these authentication methods from this Client MUST be rejected, if the JWT is not signed with this algorithm. Servers SHOULD support RS256. The value none MUST NOT be used. The default, if omitted, is that any algorithm supported by the OP and the RP MAY be used. """ if self.get("token_endpoint_auth_signing_alg") == "none": raise InvalidClaimError("token_endpoint_auth_signing_alg") self._validate_claim_value("token_endpoint_auth_signing_alg") def validate_application_type(self): """Kind of the application. The default, if omitted, is web. The defined values are native or web. Web Clients using the OAuth Implicit Grant Type MUST only register URLs using the https scheme as redirect_uris; they MUST NOT use localhost as the hostname. Native Clients MUST only register redirect_uris using custom URI schemes or loopback URLs using the http scheme; loopback URLs use localhost or the IP loopback literals 127.0.0.1 or [::1] as the hostname. Authorization Servers MAY place additional constraints on Native Clients. Authorization Servers MAY reject Redirection URI values using the http scheme, other than the loopback case for Native Clients. The Authorization Server MUST verify that all the registered redirect_uris conform to these constraints. This prevents sharing a Client ID across different types of Clients. """ self.setdefault("application_type", "web") if self.get("application_type") not in ("web", "native"): raise InvalidClaimError("application_type") self._validate_claim_value("application_type") def validate_sector_identifier_uri(self): """URL using the https scheme to be used in calculating Pseudonymous Identifiers by the OP. The URL references a file with a single JSON array of redirect_uri values. Please see Section 5. Providers that use pairwise sub (subject) values SHOULD utilize the sector_identifier_uri value provided in the Subject Identifier calculation for pairwise identifiers. """ self._validate_uri("sector_identifier_uri") def validate_subject_type(self): """subject_type requested for responses to this Client. The subject_types_supported discovery parameter contains a list of the supported subject_type values for the OP. Valid types include pairwise and public. """ self._validate_claim_value("subject_type") def validate_id_token_signed_response_alg(self): """JWS alg algorithm [JWA] REQUIRED for signing the ID Token issued to this Client. The value none MUST NOT be used as the ID Token alg value unless the Client uses only Response Types that return no ID Token from the Authorization Endpoint (such as when only using the Authorization Code Flow). The default, if omitted, is RS256. The public key for validating the signature is provided by retrieving the JWK Set referenced by the jwks_uri element from OpenID Connect Discovery 1.0 [OpenID.Discovery]. """ if self.get( "id_token_signed_response_alg" ) == "none" and "id_token" in self.get("response_type", ""): raise InvalidClaimError("id_token_signed_response_alg") self.setdefault("id_token_signed_response_alg", "RS256") self._validate_claim_value("id_token_signed_response_alg") def validate_id_token_encrypted_response_alg(self): """JWE alg algorithm [JWA] REQUIRED for encrypting the ID Token issued to this Client. If this is requested, the response will be signed then encrypted, with the result being a Nested JWT, as defined in [JWT]. The default, if omitted, is that no encryption is performed. """ self._validate_claim_value("id_token_encrypted_response_alg") def validate_id_token_encrypted_response_enc(self): """JWE enc algorithm [JWA] REQUIRED for encrypting the ID Token issued to this Client. If id_token_encrypted_response_alg is specified, the default id_token_encrypted_response_enc value is A128CBC-HS256. When id_token_encrypted_response_enc is included, id_token_encrypted_response_alg MUST also be provided. """ if self.get("id_token_encrypted_response_enc") and not self.get( "id_token_encrypted_response_alg" ): raise InvalidClaimError("id_token_encrypted_response_enc") if self.get("id_token_encrypted_response_alg"): self.setdefault("id_token_encrypted_response_enc", "A128CBC-HS256") self._validate_claim_value("id_token_encrypted_response_enc") def validate_userinfo_signed_response_alg(self): """JWS alg algorithm [JWA] REQUIRED for signing UserInfo Responses. If this is specified, the response will be JWT [JWT] serialized, and signed using JWS. The default, if omitted, is for the UserInfo Response to return the Claims as a UTF-8 [RFC3629] encoded JSON object using the application/json content-type. """ self._validate_claim_value("userinfo_signed_response_alg") def validate_userinfo_encrypted_response_alg(self): """JWE [JWE] alg algorithm [JWA] REQUIRED for encrypting UserInfo Responses. If both signing and encryption are requested, the response will be signed then encrypted, with the result being a Nested JWT, as defined in [JWT]. The default, if omitted, is that no encryption is performed. """ self._validate_claim_value("userinfo_encrypted_response_alg") def validate_userinfo_encrypted_response_enc(self): """JWE enc algorithm [JWA] REQUIRED for encrypting UserInfo Responses. If userinfo_encrypted_response_alg is specified, the default userinfo_encrypted_response_enc value is A128CBC-HS256. When userinfo_encrypted_response_enc is included, userinfo_encrypted_response_alg MUST also be provided. """ if self.get("userinfo_encrypted_response_enc") and not self.get( "userinfo_encrypted_response_alg" ): raise InvalidClaimError("userinfo_encrypted_response_enc") if self.get("userinfo_encrypted_response_alg"): self.setdefault("userinfo_encrypted_response_enc", "A128CBC-HS256") self._validate_claim_value("userinfo_encrypted_response_enc") def validate_default_max_age(self): """Default Maximum Authentication Age. Specifies that the End-User MUST be actively authenticated if the End-User was authenticated longer ago than the specified number of seconds. The max_age request parameter overrides this default value. If omitted, no default Maximum Authentication Age is specified. """ if self.get("default_max_age") is not None and not isinstance( self["default_max_age"], (int, float) ): raise InvalidClaimError("default_max_age") self._validate_claim_value("default_max_age") def validate_require_auth_time(self): """Boolean value specifying whether the auth_time Claim in the ID Token is REQUIRED. It is REQUIRED when the value is true. (If this is false, the auth_time Claim can still be dynamically requested as an individual Claim for the ID Token using the claims request parameter described in Section 5.5.1 of OpenID Connect Core 1.0 [OpenID.Core].) If omitted, the default value is false. """ self.setdefault("require_auth_time", False) if self.get("require_auth_time") is not None and not isinstance( self["require_auth_time"], bool ): raise InvalidClaimError("require_auth_time") self._validate_claim_value("require_auth_time") def validate_default_acr_values(self): """Default requested Authentication Context Class Reference values. Array of strings that specifies the default acr values that the OP is being requested to use for processing requests from this Client, with the values appearing in order of preference. The Authentication Context Class satisfied by the authentication performed is returned as the acr Claim Value in the issued ID Token. The acr Claim is requested as a Voluntary Claim by this parameter. The acr_values_supported discovery element contains a list of the supported acr values supported by the OP. Values specified in the acr_values request parameter or an individual acr Claim request override these default values. """ self._validate_claim_value("default_acr_values") def validate_initiate_login_uri(self): """RI using the https scheme that a third party can use to initiate a login by the RP, as specified in Section 4 of OpenID Connect Core 1.0 [OpenID.Core]. The URI MUST accept requests via both GET and POST. The Client MUST understand the login_hint and iss parameters and SHOULD support the target_link_uri parameter. """ self._validate_uri("initiate_login_uri") def validate_request_object_signing_alg(self): """JWS [JWS] alg algorithm [JWA] that MUST be used for signing Request Objects sent to the OP. All Request Objects from this Client MUST be rejected, if not signed with this algorithm. Request Objects are described in Section 6.1 of OpenID Connect Core 1.0 [OpenID.Core]. This algorithm MUST be used both when the Request Object is passed by value (using the request parameter) and when it is passed by reference (using the request_uri parameter). Servers SHOULD support RS256. The value none MAY be used. The default, if omitted, is that any algorithm supported by the OP and the RP MAY be used. """ self._validate_claim_value("request_object_signing_alg") def validate_request_object_encryption_alg(self): """JWE [JWE] alg algorithm [JWA] the RP is declaring that it may use for encrypting Request Objects sent to the OP. This parameter SHOULD be included when symmetric encryption will be used, since this signals to the OP that a client_secret value needs to be returned from which the symmetric key will be derived, that might not otherwise be returned. The RP MAY still use other supported encryption algorithms or send unencrypted Request Objects, even when this parameter is present. If both signing and encryption are requested, the Request Object will be signed then encrypted, with the result being a Nested JWT, as defined in [JWT]. The default, if omitted, is that the RP is not declaring whether it might encrypt any Request Objects. """ self._validate_claim_value("request_object_encryption_alg") def validate_request_object_encryption_enc(self): """JWE enc algorithm [JWA] the RP is declaring that it may use for encrypting Request Objects sent to the OP. If request_object_encryption_alg is specified, the default request_object_encryption_enc value is A128CBC-HS256. When request_object_encryption_enc is included, request_object_encryption_alg MUST also be provided. """ if self.get("request_object_encryption_enc") and not self.get( "request_object_encryption_alg" ): raise InvalidClaimError("request_object_encryption_enc") if self.get("request_object_encryption_alg"): self.setdefault("request_object_encryption_enc", "A128CBC-HS256") self._validate_claim_value("request_object_encryption_enc") def validate_request_uris(self): """Array of request_uri values that are pre-registered by the RP for use at the OP. These URLs MUST use the https scheme unless the target Request Object is signed in a way that is verifiable by the OP. Servers MAY cache the contents of the files referenced by these URIs and not retrieve them at the time they are used in a request. OPs can require that request_uri values used be pre-registered with the require_request_uri_registration discovery parameter. If the contents of the request file could ever change, these URI values SHOULD include the base64url-encoded SHA-256 hash value of the file contents referenced by the URI as the value of the URI fragment. If the fragment value used for a URI changes, that signals the server that its cached value for that URI with the old fragment value is no longer valid. """ self._validate_uri("request_uris")