From 051ea341b5573fe3edcd53042f347929b92c2b92 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jos=C3=A9=20Padilla?= Date: Thu, 12 Mar 2026 12:46:08 -0400 Subject: [PATCH] Merge commit from fork MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-authored-by: José Padilla --- CHANGELOG.rst | 1 jwt/api_jws.py | 27 ++++++++++++- tests/test_api_jws.py | 100 ++++++++++++++++++++++++++++++++++++++++++++++++++ tests/test_api_jwt.py | 19 +++++++++ 4 files changed, 145 insertions(+), 2 deletions(-) Index: pyjwt-2.9.0/CHANGELOG.rst =================================================================== --- pyjwt-2.9.0.orig/CHANGELOG.rst 2024-08-01 16:58:37.000000000 +0200 +++ pyjwt-2.9.0/CHANGELOG.rst 2026-03-19 19:29:53.632129845 +0100 @@ -12,6 +12,7 @@ Fixed ~~~~~ +- Validate the crit (Critical) Header Parameter defined in RFC 7515 §4.1.11. by @dmbs335 in `GHSA-752w-5fwx-jx9f `__ Added ~~~~~ Index: pyjwt-2.9.0/jwt/api_jws.py =================================================================== --- pyjwt-2.9.0.orig/jwt/api_jws.py 2024-08-01 16:58:37.000000000 +0200 +++ pyjwt-2.9.0/jwt/api_jws.py 2026-03-19 19:29:18.793675073 +0100 @@ -130,7 +130,7 @@ header: dict[str, Any] = {"typ": self.header_typ, "alg": algorithm_} if headers: - self._validate_headers(headers) + self._validate_headers(headers, encoding=True) header.update(headers) if not header["typ"]: @@ -198,6 +198,8 @@ payload, signing_input, header, signature = self._load(jwt) + self._validate_headers(header) + if header.get("b64", True) is False: if detached_payload is None: raise DecodeError( @@ -316,14 +318,35 @@ if not alg_obj.verify(signing_input, prepared_key, signature): raise InvalidSignatureError("Signature verification failed") - def _validate_headers(self, headers: dict[str, Any]) -> None: + # Extensions that PyJWT actually understands and supports + _supported_crit: set[str] = {"b64"} + + def _validate_headers( + self, headers: dict[str, Any], *, encoding: bool = False + ) -> None: if "kid" in headers: self._validate_kid(headers["kid"]) + if not encoding and "crit" in headers: + self._validate_crit(headers) def _validate_kid(self, kid: Any) -> None: if not isinstance(kid, str): raise InvalidTokenError("Key ID header parameter must be a string") + def _validate_crit(self, headers: dict[str, Any]) -> None: + crit = headers["crit"] + if not isinstance(crit, list) or len(crit) == 0: + raise InvalidTokenError("Invalid 'crit' header: must be a non-empty list") + for ext in crit: + if not isinstance(ext, str): + raise InvalidTokenError("Invalid 'crit' header: values must be strings") + if ext not in self._supported_crit: + raise InvalidTokenError(f"Unsupported critical extension: {ext}") + if ext not in headers: + raise InvalidTokenError( + f"Critical extension '{ext}' is missing from headers" + ) + _jws_global_obj = PyJWS() encode = _jws_global_obj.encode Index: pyjwt-2.9.0/tests/test_api_jws.py =================================================================== --- pyjwt-2.9.0.orig/tests/test_api_jws.py 2024-08-01 16:58:37.000000000 +0200 +++ pyjwt-2.9.0/tests/test_api_jws.py 2026-03-19 19:33:40.366377055 +0100 @@ -869,3 +869,103 @@ ) assert len(record) == 1 assert "foo" in str(record[0].message) + + def test_decode_rejects_unknown_crit_extension( + self, jws: PyJWS, payload: bytes + ) -> None: + secret = "secret" + token = jws.encode( + payload, + secret, + algorithm="HS256", + headers={"crit": ["x-custom-policy"], "x-custom-policy": "require-mfa"}, + ) + + with pytest.raises(InvalidTokenError, match="Unsupported critical extension"): + jws.decode(token, secret, algorithms=["HS256"]) + + def test_decode_rejects_empty_crit(self, jws: PyJWS, payload: bytes) -> None: + secret = "secret" + token = jws.encode( + payload, + secret, + algorithm="HS256", + headers={"crit": []}, + ) + + with pytest.raises(InvalidTokenError, match="must be a non-empty list"): + jws.decode(token, secret, algorithms=["HS256"]) + + def test_decode_rejects_non_list_crit(self, jws: PyJWS, payload: bytes) -> None: + secret = "secret" + token = jws.encode( + payload, + secret, + algorithm="HS256", + headers={"crit": "b64"}, + ) + + with pytest.raises(InvalidTokenError, match="must be a non-empty list"): + jws.decode(token, secret, algorithms=["HS256"]) + + def test_decode_rejects_crit_with_non_string_values( + self, jws: PyJWS, payload: bytes + ) -> None: + secret = "secret" + token = jws.encode( + payload, + secret, + algorithm="HS256", + headers={"crit": [123]}, + ) + + with pytest.raises(InvalidTokenError, match="values must be strings"): + jws.decode(token, secret, algorithms=["HS256"]) + + def test_decode_rejects_crit_extension_missing_from_header( + self, jws: PyJWS, payload: bytes + ) -> None: + secret = "secret" + token = jws.encode( + payload, + secret, + algorithm="HS256", + headers={"crit": ["b64"]}, + ) + + with pytest.raises(InvalidTokenError, match="missing from headers"): + jws.decode(token, secret, algorithms=["HS256"]) + + def test_decode_accepts_supported_crit_extension( + self, jws: PyJWS, payload: bytes + ) -> None: + secret = "secret" + token = jws.encode( + payload, + secret, + algorithm="HS256", + headers={"crit": ["b64"], "b64": False}, + is_payload_detached=True, + ) + + decoded = jws.decode( + token, + secret, + algorithms=["HS256"], + detached_payload=payload, + ) + assert decoded == payload + + def test_get_unverified_header_rejects_unknown_crit( + self, jws: PyJWS, payload: bytes + ) -> None: + secret = "secret" + token = jws.encode( + payload, + secret, + algorithm="HS256", + headers={"crit": ["x-unknown"], "x-unknown": "value"}, + ) + + with pytest.raises(InvalidTokenError, match="Unsupported critical extension"): + jws.get_unverified_header(token) Index: pyjwt-2.9.0/tests/test_api_jwt.py =================================================================== --- pyjwt-2.9.0.orig/tests/test_api_jwt.py 2024-08-01 16:58:37.000000000 +0200 +++ pyjwt-2.9.0/tests/test_api_jwt.py 2026-03-19 19:31:39.077273095 +0100 @@ -816,3 +816,22 @@ options={"strict_aud": True}, algorithms=["HS256"], ) + + + # -------------------- Crit Header Tests -------------------- + + def test_decode_rejects_token_with_unknown_crit_extension(self, jwt: PyJWT) -> None: + """RFC 7515 §4.1.11: tokens with unsupported critical extensions MUST be rejected.""" + from jwt.exceptions import InvalidTokenError + + secret = "secret" + payload = {"sub": "attacker", "role": "admin"} + token = jwt.encode( + payload, + secret, + algorithm="HS256", + headers={"crit": ["x-custom-policy"], "x-custom-policy": "require-mfa"}, + ) + + with pytest.raises(InvalidTokenError, match="Unsupported critical extension"): + jwt.decode(token, secret, algorithms=["HS256"])