hvdZddlZddlmZmZmZddlmZddlm Z ddl m Z m Z m Z mZdgZGd dZdS) z Module to handle the timestamping functionality in pyHanko. Many PDF signature profiles require trusted timestamp tokens. The tools in this module allow pyHanko to obtain such tokens from :rfc:`3161`-compliant time stamping authorities. N)algoscmstsp)CertificateValidator)SimpleCertificateStore) dummy_digestextract_ts_certs get_noncehandle_tsp_response TimeStamperceZdZdZddZdZdZdZdZde j fd Z d e j de jfd Zde j fd Zd S)r z .. versionchanged:: 0.9.0 Made API more asyncio-friendly _(breaking change)_ Class to make :rfc:`3161` timestamp requests. TcVi|_i|_t|_||_dSN)_dummy_response_cache_certsr cert_registry include_nonce)selfrs W/var/www/html/Sam_Eipo/venv/lib/python3.11/site-packages/pyhanko/sign/timestamps/api.py__init__zTimeStamper.__init__!s-%'" 355*cdtjtjd|i|ddd}|jr&t }t j||d<nd}|tj|fS)a& Format the body of an :rfc:`3161` request as a CMS object. Subclasses with more specific needs may want to override this. :param message_digest: Message digest to which the timestamp will apply. :param md_algorithm: Message digest algorithm to use. .. note:: As per :rfc:`8933`, ``md_algorithm`` should also be the algorithm used to compute ``message_digest``. :return: An :class:`.asn1crypto.tsp.TimeStampReq` object. r algorithm)hash_algorithmhashed_messageT)versionmessage_imprintcert_reqnonceN) rMessageImprintrDigestAlgorithmrr rInteger TimeStampReq)rmessage_digest md_algorithmreqr s r request_cmszTimeStamper.request_cms's""1&+&;$l3'''5         KKE;u--CLLEc&s++++rcKd{Vfd}t|j}t j|D] }|d{VWVdS)a+ Produce validation paths for the certificates gathered by this :class:`.TimeStamper`. This is internal API. :param validation_context: The validation context to apply. :return: An asynchronous generator of validation paths. Ncxt|j}|tdhS)N)intermediate_certsvalidation_context time_stamping)rrasync_validate_usageset)cert validatorrr,s r_validation_jobz5TimeStamper.validation_paths.._validation_jobYsB,#'#5#5I 11#%%/9JKK Kr) _ensure_dummymaprvaluesasyncio as_completed)rr,r2jobsjobs`` rvalidation_pathszTimeStamper.validation_pathsKs  """"""""" L L L L L L?DK$6$6$8$899'--  C))))))OOOOO  rch||j|<t||jD]}||j|j<dSr)rr rr issuer_serial)rr&dummyr0s r_register_dummyzTimeStamper._register_dummyfsG38"<0$UD,>?? 3 3D.2DK* + + 3 3rc^K|js#ddlm}||d{VdSdS)Nr) DEFAULT_MD)r pyhanko.signr@async_dummy_response)rr@s rr3zTimeStamper._ensure_dummyksY) 8 / / / / / /++J77 7 7 7 7 7 7 7 7 7 8 8rreturncK |j|S#t$r,|t||d{V}YnwxYw||||S)a9 Return a dummy response for use in CMS object size estimation. For every new ``md_algorithm`` passed in, this method will call the :meth:`timestamp` method exactly once, with a dummy digest. The resulting object will be cached and reused for future invocations of :meth:`dummy_response` with the same ``md_algorithm`` value. :param md_algorithm: Message digest algorithm to use. :return: A timestamp token, encoded as an :class:`.asn1crypto.cms.ContentInfo` object. N)rKeyErrorasync_timestampr r>)rr&r=s rrBz TimeStamper.async_dummy_responsers$ -l; ;   ..\**LEEE  \5111 s 3AAr'cKt)aF Submit the specified timestamp request to the server. :param req: Request body to submit. :return: A timestamp response from the server. :raises IOError: Raised in case of an I/O issue in the communication with the timestamping server. )NotImplementedError)rr's rasync_request_tsa_responsez&TimeStamper.async_request_tsa_responses"!rcK|||\}}||d{V}t||S)a+ Request a timestamp for the given message digest. :param message_digest: Message digest to which the timestamp will apply. :param md_algorithm: Message digest algorithm to use. .. note:: As per :rfc:`8933`, ``md_algorithm`` should also be the algorithm used to compute ``message_digest``. :return: A timestamp token, encoded as an :class:`.asn1crypto.cms.ContentInfo` object. :raises IOError: Raised in case of an I/O issue in the communication with the timestamping server. :raises TimestampRequestError: Raised if the timestamp server did not return a success response, or if the server's response is invalid. N)r(rIr )rr%r&r r'ress rrFzTimeStamper.async_timestampsV2%%nlCC s33C88888888"3...rN)T)__name__ __module__ __qualname____doc__rr(r:r>r3r ContentInforBrr$ TimeStampResprIrFrrr r s++++ ",",",H6333 888#/6"#" """" / //////r)rOr6 asn1cryptorrrpyhanko_certvalidatorrpyhanko_certvalidator.registryr common_utilsr r r r __all__r rRrrrXs&&&&&&&&&&666666AAAAAA /_/_/_/_/_/_/_/_/_/_/r