gpdZddlZddlZddlZddlZddlmZmZmZddl m Z ddl Z ddl mZGddZy)z:SSL Certificate class for handling certificate operations.N)DictAnyOptional)urlparse)Pathc BeZdZdZdeeeffdZeddede de dfdZ ededefd Z dd e ede efd Z dd e ede efd Zdd e ede efdZedeeeffdZedeeeffdZedefdZedefdZedefdZy )SSLCertificatean A class representing an SSL certificate with methods to export in various formats. Attributes: cert_info (Dict[str, Any]): The certificate information. Methods: from_url(url: str, timeout: int = 10) -> Optional['SSLCertificate']: Create SSLCertificate instance from a URL. from_file(file_path: str) -> Optional['SSLCertificate']: Create SSLCertificate instance from a file. from_binary(binary_data: bytes) -> Optional['SSLCertificate']: Create SSLCertificate instance from binary data. export_as_pem() -> str: Export the certificate as PEM format. export_as_der() -> bytes: Export the certificate as DER format. export_as_json() -> Dict[str, Any]: Export the certificate as JSON format. export_as_text() -> str: Export the certificate as text format. cert_infoc0|j||_yN)_decode_cert_data _cert_info)selfr s M/var/www/openai/venv/lib/python3.12/site-packages/crawl4ai/ssl_certificate.py__init__zSSLCertificate.__init__s00;urltimeoutreturnc  t|j}d|vr|jdd}tj}t j |df|5}|j||5}|jd}tjjtjj|}t|jjt|j!j|j#t%|j'|j)|j+|j-dj%|j/t1j2|d }g} t5|j7D]>} |j9| } | j;| j=t?| d @| |d <tA|cd d d cd d d S#1swYnxYw d d d y #1swYy xYw#tB$r } Yd } ~ y d } ~ wwxYw) a: Create SSLCertificate instance from a URL. Args: url (str): URL of the website. timeout (int): Timeout for the connection (default: 10). Returns: Optional[SSLCertificate]: SSLCertificate instance if successful, None otherwise. :ri)r)server_hostnameT) binary_formsha256) subjectissuerversion serial_number not_before not_after fingerprintsignature_algorithmraw_cert)namevalue extensionsN)"rnetlocsplitsslcreate_default_contextsocketcreate_connection wrap_socket getpeercertOpenSSLcryptoload_certificate FILETYPE_ASN1dict get_subjectget_components get_issuer get_versionhexget_serial_number get_notBefore get_notAfterdigestget_signature_algorithmbase64 b64encoderangeget_extension_count get_extensionappendget_short_namestrr Exception) rrhostnamecontextsockssock cert_binaryx509r r&iextes rfrom_urlzSSLCertificate.from_url s$ }++Hh#>>#.q1002G))8S/7Kt((x(HE"'"3"3"3"EK">>::7>>;W;WYdeD$((8(8(:(I(I(K#L"&t'8'G'G'I"J#'#3#3#5),T-C-C-E)F&*&8&8&:%)%6%6%8'+{{8'<'@'@'B/3/K/K/M$*$4$4[$A !I"$J"4#;#;#=>"003"))$'$6$6$8%(X+? /9Il+))45IHLKHHHLKK:  sOAH0H$0F H; H$ H0H H$H0$H-)H0-H00 I>Idatact|tr|jdSt|trX|j Dcic]<\}}t|tr|jdn|t j |>c}}St|tr#|Dcgc]}t j |c}S|Scc}}wcc}w)z2Helper method to decode bytes in certificate data.utf-8) isinstancebytesdecoder3itemsr r list)rQkvitems rr z SSLCertificate._decode_cert_dataRs dE ";;w' ' d #!JJL(DAq'1E&:'"NDdDdefDgg( d #GKLttN44T:tL L  Ms AB?CNfilepathctj|jdd}|rt|j |dy|S)z Export certificate as JSON. Args: filepath (Optional[str]): Path to save the JSON file (default: None). Returns: Optional[str]: JSON string if successful, None otherwise. F)indent ensure_asciirSencodingN)jsondumpsrr write_text)rr\json_strs rto_jsonzSSLCertificate.to_json`s<::dooaeL  N % %h % Arc tjjtjjt j |j d}tjjtjj|jd}|rt|j|dy|S#t$r }Yd}~yd}~wwxYw)z Export certificate as PEM. Args: filepath (Optional[str]): Path to save the PEM file (default: None). Returns: Optional[str]: PEM string if successful, None otherwise. r#rSraN) r/r0r1r2r> b64decoderdump_certificate FILETYPE_PEMrVrrerF)rr\rLpem_datarOs rto_pemzSSLCertificate.to_pemps >>22,,  !<=D~~66++fWo  X))(W)EO  sB=CC CCc tj|jd}|rt|j |y|S#t $rYywxYw)z Export certificate as DER. Args: filepath (Optional[str]): Path to save the DER file (default: None). Returns: Optional[bytes]: DER bytes if successful, None otherwise. r#N)r>rirr write_bytesrF)rr\der_datas rto_derzSSLCertificate.to_dersO '' (CDHX**84O  s>AA AAc:|jjdiS)z#Get certificate issuer information.rrgetrs rrzSSLCertificate.issuers""8R00rc:|jjdiS)z$Get certificate subject information.rrsrus rrzSSLCertificate.subjects""9b11rc:|jjddS)z$Get certificate validity start date.rrsrus r valid_fromzSSLCertificate.valid_froms""<44rc:|jjddS)z"Get certificate validity end date.r rxrsrus r valid_untilzSSLCertificate.valid_untils"";33rc:|jjddS)zGet certificate fingerprint.r!rxrsrus rr!zSSLCertificate.fingerprints""="55r) r )__name__ __module__ __qualname____doc__rrErr staticmethodintrrPr rgrmrUrqpropertyrrryr{r!rrr r sU<$sCx.</c/C/:J1K//b     # x} 6x}&1S#X112c3h225C554S446S66rr )rr)r+r>rctypingrrr urllib.parserOpenSSL.cryptor/pathlibrr rrrrs/@ &&!h6h6r