Gecko [ bayviewconstructionltd.com ]

Name	Size	Permission	Date
__pycache__	[ DIR ]	drwxr-xr-x	2023-06-04 17:23
__init__.py	872 B	-rw-r--r--	2019-08-01 02:26
challenges.py	19.37 KB	-rw-r--r--	2019-08-01 02:26
challenges_test.py	20.03 KB	-rw-r--r--	2019-08-01 02:26
client.py	45.61 KB	-rw-r--r--	2019-08-01 02:26
client_test.py	55.13 KB	-rw-r--r--	2019-08-01 02:26
crypto_util.py	10.99 KB	-rw-r--r--	2018-11-07 21:14
crypto_util_test.py	9.99 KB	-rw-r--r--	2018-11-07 21:14
errors.py	3.57 KB	-rw-r--r--	2018-11-07 21:14
errors_test.py	1.48 KB	-rw-r--r--	2018-11-07 21:14
fields.py	1.7 KB	-rw-r--r--	2018-11-07 21:14
fields_test.py	2.03 KB	-rw-r--r--	2018-11-07 21:14
jose_test.py	1.92 KB	-rw-r--r--	2019-08-01 02:26
jws.py	2.09 KB	-rw-r--r--	2018-11-07 21:14
jws_test.py	2.03 KB	-rw-r--r--	2018-11-07 21:14
magic_typing.py	534 B	-rw-r--r--	2018-11-07 21:14
magic_typing_test.py	1.42 KB	-rw-r--r--	2018-11-07 21:14
messages.py	17.97 KB	-rw-r--r--	2018-11-07 21:14
messages_test.py	14.84 KB	-rw-r--r--	2018-11-07 21:14
standalone.py	11.09 KB	-rw-r--r--	2018-11-07 21:14
standalone_test.py	10.54 KB	-rw-r--r--	2018-11-07 21:14
test_util.py	3.12 KB	-rw-r--r--	2018-11-07 21:14
util.py	166 B	-rw-r--r--	2018-11-07 21:14
util_test.py	456 B	-rw-r--r--	2018-11-07 21:14

Rename

"""ACME-specific JWS.

The JWS implementation in josepy only implements the base JOSE standard. In
order to support the new header fields defined in ACME, this module defines some
ACME-specific classes that layer on top of josepy.
"""
import josepy as jose

class Header(jose.Header):
    """ACME-specific JOSE Header. Implements nonce, kid, and url.
    """
    nonce = jose.Field('nonce', omitempty=True, encoder=jose.encode_b64jose)
    kid = jose.Field('kid', omitempty=True)
    url = jose.Field('url', omitempty=True)

@nonce.decoder
    def nonce(value):  # pylint: disable=missing-docstring,no-self-argument
        try:
            return jose.decode_b64jose(value)
        except jose.DeserializationError as error:
            # TODO: custom error
            raise jose.DeserializationError("Invalid nonce: {0}".format(error))

class Signature(jose.Signature):
    """ACME-specific Signature. Uses ACME-specific Header for customer fields."""
    __slots__ = jose.Signature._orig_slots  # pylint: disable=no-member

# TODO: decoder/encoder should accept cls? Otherwise, subclassing
    # JSONObjectWithFields is tricky...
    header_cls = Header
    header = jose.Field(
        'header', omitempty=True, default=header_cls(),
        decoder=header_cls.from_json)

# TODO: decoder should check that nonce is in the protected header

class JWS(jose.JWS):
    """ACME-specific JWS. Includes none, url, and kid in protected header."""
    signature_cls = Signature
    __slots__ = jose.JWS._orig_slots  # pylint: disable=no-member

@classmethod
    # pylint: disable=arguments-differ,too-many-arguments
    def sign(cls, payload, key, alg, nonce, url=None, kid=None):
        # Per ACME spec, jwk and kid are mutually exclusive, so only include a
        # jwk field if kid is not provided.
        include_jwk = kid is None
        return super(JWS, cls).sign(payload, key=key, alg=alg,
                                    protect=frozenset(['nonce', 'url', 'kid', 'jwk', 'alg']),
                                    nonce=nonce, url=url, kid=kid,
                                    include_jwk=include_jwk)