U c@sddlZddlmZddlmZmZddlmZmZmZddlm Z ddlm Z m Z ddl mZdd lmZmZmZdd lmZd d d dddddgZGdd d ZGdddeZGdddeZGdddeZGdddeZGdddeZGdd d ejZGdd d ejZdS)N)datetime)IterableOptional)algoscmscore)pdf)tspx509)hashes)as_signing_certificate_v2get_pyca_cryptography_hashsimple_cms_attribute) TimeStamperSignedAttributeProviderSpecUnsignedAttributeProviderSpecCMSAttributeProviderSigningTimeProviderSigningCertificateV2ProviderAdobeRevinfoProviderCMSAlgorithmProtectionProvider TSTProviderc@s:eZdZUdZeed<d ddZd eej dddZ d S) rzB Base class to provide asynchronous CMS attribute values. attribute_typeFcstdS)a Build the attribute value asynchronously. :param dry_run: ``True`` if the signer is operating in dry-run (size estimation) mode. :return: An attribute value appropriate for the attribute type. NNotImplementedErrorselfdry_runr;/tmp/pip-unpacked-wheel-0kb_yl26/pyhanko/sign/attributes.pybuild_attr_value(s z%CMSAttributeProvider.build_attr_valuereturncs.|j|dIdH}|dk r&t|j|SdSdS)N)r)r!rr)rrvaluerrr get_attribute4s z"CMSAttributeProvider.get_attributeN)F)F) __name__ __module__ __qualname____doc__str__annotations__r!rrZ CMSAttributer%rrrr rs  c@s@eZdZUdZdZeed<ejdddZ d e j dd d Z d S) rz Provide a value for the signing-certificate-v2 attribute. :param signing_cert: Certificate containing the signer's public key. Zsigning_certificate_v2r signing_certcCs ||_dSNr,)rr-rrr __init__Gsz%SigningCertificateV2Provider.__init__Fr"cs t|jSr.)r r-rrrr r!Jsz-SigningCertificateV2Provider.build_attr_valueN)F) r&r'r(r)rr*r+r Z Certificater/r ZSigningCertificateV2r!rrrr r=s  c@s>eZdZUdZdZeed<edddZd e j dd d Z d S) rz Provide a value for the signing-time attribute (i.e. an otherwise unauthenticated timestamp). :param timestamp: Datetime object to include. Z signing_timer timestampcCs ||_dSr.r0)rr1rrr r/YszSigningTimeProvider.__init__Fr"cstdt|jiS)NZutc_time)rTimerZUTCTimer1rrrr r!\sz$SigningTimeProvider.build_attr_valueN)F) r&r'r(r)rr*r+rr/rr2r!rrrr rNs  c@sDeZdZUdZdZeed<ejdddZ d e ejdd d Z d S) rz Yield Adobe-style revocation information for inclusion into a CMS object. :param value: A (pre-formatted) RevocationInfoArchival object. Zadobe_revocation_info_archivalrr$cCs ||_dSr.r3)rr$rrr r/qszAdobeRevinfoProvider.__init__Fr"cs|jSr.r3rrrr r!tsz%AdobeRevinfoProvider.build_attr_valueN)F) r&r'r(r)rr*r+asn1_pdfZRevocationInfoArchivalr/rr!rrrr rfs  c@s>eZdZUdZeed<eejdddZd e j ddd Z d S) rZcms_algorithm_protectionr digest_algosignature_algocCs||_||_dSr.r5)rr6r7rrr r/|sz'CMSAlgorithmProtectionProvider.__init__Fr"csttd|ji|jdS)N algorithm)digest_algorithmZsignature_algorithm)rCMSAlgorithmProtectionrZDigestAlgorithmr6r7rrrr r!s z/CMSAlgorithmProtectionProvider.build_attr_valueN)F) r&r'r(rr*r+rZSignedDigestAlgorithmr/rr:r!rrrr rys   c@s4eZdZd eeeedddZd ejdddZ d S) rsignature_time_stamp_tokenF)r9 data_to_ts timestamper attr_typecCs"||_||_||_||_||_dSr.)rr9r=data prehashed)rr9r<r=r>r@rrr r/s zTSTProvider.__init__r"csd|j}|jr|j}n&t|}t|}||j|}|rL|j |}n|j ||}|IdHSr.) r9r@r?rr ZHashupdatefinalizer=Zasync_dummy_responseZasync_timestamp)rrr9digestZmd_specZmdZts_cororrr r!s  zTSTProvider.build_attr_valueN)r;F)F) r&r'r(r*bytesrr/rZ ContentInfor!rrrr rs c@s&eZdZdZeeeedddZdS)rz .. versionadded:: 0.14.0 Interface for setting up signed attributes, independently of the :class:`~pyhanko.sign.signers.pdf_cms.Signer` hierarchy. ) data_digestr9r#cCstdS)z Lazily set up signed attribute providers. :param data_digest: The digest of the data to be signed. :param digest_algorithm: The digest algorithm used. Nr)rrEr9rrr signed_attr_providerss z1SignedAttributeProviderSpec.signed_attr_providersN) r&r'r(r)rDr*rrrFrrrr rsc@s*eZdZdZeejeee dddZ dS)rz .. versionadded:: 0.14.0 Interface for setting up unsigned attributes, independently of the :class:`~pyhanko.sign.signers.pdf_cms.Signer` hierarchy. ) signature signed_attrsr9r#cCstdS)a> Lazily set up unsigned attribute providers. :param signature: The signature computed over the signed attributes. :param signed_attrs: Signed attributes over which the signature was taken. :param digest_algorithm: The digest algorithm used. Nr)rrGrHr9rrr unsigned_attr_providerss z5UnsignedAttributeProviderSpec.unsigned_attr_providersN) r&r'r(r)rDrZ CMSAttributesr*rrrIrrrr rs )abcrtypingrrZ asn1cryptorrrrr4r r Zcryptography.hazmat.primitivesr Zgeneralr rrZ timestampsr__all__rrrrrrABCrrrrrr s2