o j@sdZddlmZmZmZmZddlZddlZddlZddl m Z ddl m Z mZmZejdkr6ddlmZndd lmZd d Zdd d ZddZdddZdS)zq Encoding DER to PEM and decoding PEM to DER. Exports the following items: - armor() - detect() - unarmor() )unicode_literalsdivisionabsolute_importprint_functionN)unwrap) type_namestr_clsbyte_cls))StringIO)BytesIOcCs8t|tsttdt||ddkp|ddkS)z Detect if a byte string seems to contain a PEM-encoded block :param byte_string: A byte string to look through :return: A boolean, indicating if a PEM-encoded block is contained in the byte string zC byte_string must be a byte string, not %s s -----BEGINs ---- BEGIN) isinstancer TypeErrorr _type_namefind) byte_stringr/root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/asn1crypto/pem.pydetects rcCs0t|tsttdt|t|tsttdt||d}t}| d| || d|r^|D]}| |d| d| ||d| dq:| dt |}t |}d}||kr| |||d | d|d 7}||ksm| d | || d| S) a  Armors a DER-encoded byte string in PEM :param type_name: A unicode string that will be capitalized and placed in the header and footer of the block. E.g. "CERTIFICATE", "PRIVATE KEY", etc. This will appear as "-----BEGIN CERTIFICATE-----" and "-----END CERTIFICATE-----". :param der_bytes: A byte string to be armored :param headers: An OrderedDict of the header lines to write after the BEGIN line :return: A byte string of the PEM block zA der_bytes must be a byte string, not %s zD type_name must be a unicode string, not %s asciis -----BEGIN s----- s:  r@s -----END )rr rrrr upperencoder writebase64 b64encodelengetvalue)r der_bytesheadersoutputkey b64_bytesb64_lenirrrarmor2sF             r(c cs,t|tsttdt|d}i}d}d}d}d}|dD]i}|dkr'q |dkrAtd|}|s4q |d d}d }d }q |d krc| d d krOd }n| d} | dd\} } | || <q |d kr|dddvrt |} ||| fVd}i}d}d}d }q ||7}q |r|sttddS)ax Convert a PEM-encoded byte string into one or more DER-encoded byte strings :param pem_bytes: A byte string of the PEM-encoded data :raises: ValueError - when the pem_bytes do not appear to be PEM-encoded bytes :return: A generator of 3-element tuples in the format: (object_type, headers, der_bytes). The object_type is a unicode string of what is between "-----BEGIN " and "-----". Examples include: "CERTIFICATE", "PUBLIC KEY", "PRIVATE KEY". The headers is a dict containing any lines in the form "Name: Value" that are right after the begin line. zA pem_bytes must be a byte string, not %s trashNFs1^(?:---- |-----)BEGIN ([A-Z0-9 ]+)(?: ----|-----)rrTr":rbody:r)s-----s---- z| pem_bytes does not appear to contain PEM-encoded data - no BEGIN/END combination found )rr rrr splitlinesrematchgroupdecodersplitstripr b64decode ValueError) pem_bytesstater" base64_data object_type found_start found_endlinetype_name_match decoded_linenamevaluer!rrr_unarmorps^      rCFcCst|}|s t|S|S)a Convert a PEM-encoded byte string into a DER-encoded byte string :param pem_bytes: A byte string of the PEM-encoded data :param multiple: If True, function will return a generator :raises: ValueError - when the pem_bytes do not appear to be PEM-encoded bytes :return: A 3-element tuple (object_name, headers, der_bytes). The object_name is a unicode string of what is between "-----BEGIN " and "-----". Examples include: "CERTIFICATE", "PUBLIC KEY", "PRIVATE KEY". The headers is a dict containing any lines in the form "Name: Value" that are right after the begin line. )rCnext)r8multiple generatorrrrunarmorsrG)N)F)__doc__ __future__rrrrrr0sys_errorsr_typesrrr r version_info cStringIOr r iorr(rCrGrrrrs     >T