o "i d@sdZddlmZddlmZddlmZddlmZddlZddlZddlm Z ddlm Z dd lm Z dd lm Z dd l mZdd lmZd ZGdddeZGdddeZGdddeZGdddeZeddZ GdddeZGdddeZGdddeZdS)z1 This module provides ASN.1 encoder and decoder. )absolute_import)division)print_function)unicode_literalsNbytesint)range)str)contextmanager)IntEnumz2.8.0c@seZdZddZdS)HexEnumcCsdjt|j|j|jdS)Nz<{cls}.{name}: 0x{value:02x}>)clsnamevalue)formattype__name__rrselfrD/var/www/html/premium_crap/venv/lib/python3.10/site-packages/asn1.py__repr__ s zHexEnum.__repr__N)r __module__ __qualname__rrrrrrs rc@sHeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdS)Numbers N)rrrBooleanInteger BitString OctetStringNullObjectIdentifier Enumerated UTF8StringSequenceSetPrintableString IA5StringUTCTimeGeneralizedTime UnicodeStringrrrrr(s rc@seZdZdZdZdS)Types rN)rrr Constructed Primitiverrrrr;:sr;c@seZdZdZdZdZdZdS)Classesr@N)rrr Universal ApplicationContextPrivaterrrrr??s r?Tagz nr typ clsc@seZdZdZdS)Errorz, ASN.11 encoding or decoding error. N)rrr__doc__rrrrrHMsrHc@seZdZdZddZddZd.ddZd d Zed.d d Z d/d dZ ddZ ddZ ddZ ddZddZddZddZddZdd Zed!d"Zed#d$Zed%d&Zed'd(Zed)d*Zed+Zd,d-ZdS)0Encoderz+ ASN.1 encoder. Uses DER encoding. cCs d|_dS)z Constructor. Nm_stackrrrr__init__Xs zEncoder.__init__cCs gg|_dS)z This method instructs the encoder to start encoding a new ASN.1 output. This method may be called at any time to reset the encoder, and resets the current output (if any). NrKrrrrstart^s z Encoder.startNcCs@|jdur td|durtj}||tj||jgdS)a This method starts the construction of a constructed type. Args: nr (int): The desired ASN.1 type. Use ``Numbers`` enumeration. cls (int): This optional parameter specifies the class of the constructed type. The default class to use is the universal class. Use ``Classes`` enumeration. Returns: None Raises: `Error` N,Encoder not initialized. Call start() first.)rLrHr?rC _emit_tagr;r=appendrnrrrrrenterfs z Encoder.entercCs\|jdur tdt|jdkrtdd|jd}|jd=|t|||dS)z This method completes the construction of a constructed type and writes the encoded representation to the output buffer. NrOrTag stack is empty.)rLrHlenjoin _emit_length_emit)rrrrrleave~s z Encoder.leaveccs |||dV|dS)a This method - context manager calls enter and leave methods, for better code mapping. Usage: ``` with encoder.construct(asn1.Numbers.Sequence): encoder.write(1) with encoder.construct(asn1.Numbers.Sequence): encoder.write('foo') encoder.write('bar') encoder.write(2) ``` encoder.output() will result following structure: SEQUENCE: INTEGER: 1 SEQUENCE: STRING: foo STRING: bar INTEGER: 2 Args: nr (int): The desired ASN.1 type. Use ``Numbers`` enumeration. cls (int): This optional parameter specifies the class of the constructed type. The default class to use is the universal class. Use ``Classes`` enumeration. Returns: None Raises: `Error` N)rTr\rRrrr constructs % zEncoder.constructcCs|jdur td|durtj}|durtj}|tjkr$|dur$td|durSt|tr1tj }n"t|t r:tj }nt|t rCtj }nt|trLtj}n|durStj}||||}|||||t|||dS)a) This method encodes one ASN.1 tag and writes it to the output buffer. Note: Normally, ``value`` will be the only parameter to this method. In this case Python-ASN1 will autodetect the correct ASN.1 type from the type of ``value``, and will output the encoded value based on this type. Args: value (any): The value of the ASN.1 tag to write. Python-ASN1 will try to autodetect the correct ASN.1 type from the type of ``value``. nr (int): If the desired ASN.1 type cannot be autodetected or is autodetected wrongly, the ``nr`` parameter can be provided to specify the ASN.1 type to be used. Use ``Numbers`` enumeration. typ (int): This optional parameter can be used to write constructed types to the output by setting it to indicate the constructed encoding type. In this case, ``value`` must already be valid ASN.1 encoded data as plain Python bytes. This is not normally how constructed types should be encoded though, see `Encoder.enter()` and `Encoder.leave()` for the recommended way of doing this. Use ``Types`` enumeration. cls (int): This parameter can be used to override the class of the ``value``. The default class is the universal class. Use ``Classes`` enumeration. Returns: None Raises: `Error` NrOzSPlease specify a tag number (nr) when using classes Application, Context or Private)rLrHr;r>r?rC isinstanceboolrr,r r-r r6rr/r0 _encode_valuerPrZrXr[)rrrStyprrrrwrites. %    z Encoder.writecCs<|jdur tdt|jdkrtdd|jd}|S)a7 This method returns the encoded ASN.1 data as plain Python ``bytes``. This method can be called multiple times, also during encoding. In the latter case the data that has been encoded so far is returned. Note: It is an error to call this method if the encoder is still constructing a constructed type, i.e. if `Encoder.enter()` has been called more times that `Encoder.leave()`. Returns: bytes: The DER encoded ASN.1 data. Raises: `Error` NrOrzStack is not empty.rVr)rLrHrXrY)routputrrrrcs zEncoder.outputcCs,|dkr ||||dS||||dS)z Emit a tag.N)_emit_tag_short_emit_tag_longrrSrarrrrrPszEncoder._emit_tagcCs(|dksJ|t||B|BgdS)zEmit a short (< 31 bytes) tag.rdNr[rrgrrrres zEncoder._emit_tag_shortcCsvt||BdBg}|||d@g}|dL}|r(||d@dB|dL}|s||D] }|t|gq.dS)zEmit a long (>= 31 bytes) tag.rdrAN)rr[rQreverse)rrSrarheadvaluesvalrrrrfs  zEncoder._emit_tag_longcCs$|dkr ||dS||dS)zEmit length octects.rAN)_emit_length_short_emit_length_longrlengthrrrrZ(szEncoder._emit_lengthcCs |dksJ|t|gdS)z*Emit the short length form (< 128 octets).rANrhrqrrrro/s zEncoder._emit_length_shortcCstg}|r||d@|dL}|s|t|dksJtdt|Bg}|||D] }|t|gq-dS)z*Emit the long length form (>= 128 octets).rirAN)rQrkrXrr[)rrrrmrlrnrrrrp4s zEncoder._emit_length_longcCs"t|tsJ|jd|dS)zEmit raw bytes.rWN)r^rrLrQ)rsrrrr[Bsz Encoder._emitcCs|tjkr|S|tjtjfvr||S|tjtjtjtj tj tj tj fvr+| |S|tjkr5||S|tjkr?||S|tjkrH|S|tjkrR||S|S)zEncode a value.)r?rCrr-r2_encode_integerr/r6r3r7r:r8r9_encode_octet_stringr._encode_bit_stringr,_encode_booleanr0 _encode_nullr1_encode_object_identifier)rrrSrrrrr`Gs&           zEncoder._encode_valuecCs|rtdp tdS)zEncode a boolean.rrrrrry\szEncoder._encode_booleancCs|dkr | }d}d}nd}d}g}||kr%||d@|dL}||ks||d@|rftt|D] }d||||<q4tt|D] }||d7<||dkrWn|t|dksaJd||<qE|rw|t|ddkrw|d|t|S) zEncode an integer.rTrAFrirsrtr)rQr rXrkr)rnegativelimitrmirrrrvas2   zEncoder._encode_integercCs0t|ts t|ts Jt|tr|dS|S)zEncode an octetstring.utf-8)r^r rencoder~rrrrws  zEncoder._encode_octet_stringcCst|tsJd|S)z,Encode a bitstring. Assumes no unused bytes.r})r^rr~rrrrxszEncoder._encode_bit_stringcCstdS)zEncode a Null value.rVrrrrrrzszEncoder._encode_nullz^[0-9]+(\.[0-9]+)+$cCs|j|s tdttt|d}|ddkr |ddks&|ddkr*tdd|d|dg|dd}|g}|D]}||d @|d kra|d L}|d |d @B|d ksPqC|t |S) zEncode an object identifier.zIllegal object identifier.rr'r(NrirjrA) _re_oidmatchrHlistmapr splitrkrQr)roidcmpsresultcmp_datarrrr{s" $"z!Encoder._encode_object_identifierN)NNN)rrrrIrMrNrTr\r r]rbrcrPrerfrZrorpr[r` staticmethodryrvrwrxrzrecompilerr{rrrrrJSs<   (A        rJc@seZdZdZddZddZddZd+d d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZeddZedd Zed!d"Zed#d$Zed%d&Zed'd(Zed)d*ZdS),DecoderzE ASN.1 decoder. Understands BER (and DER which is a subset). cCsd|_d|_dS)z Constructor.N)rLm_tagrrrrrMs zDecoder.__init__cCs,t|ts tddt|gg|_d|_dS)a This method instructs the decoder to start decoding the ASN.1 input ``data``, which must be a passed in as plain Python bytes. This method may be called at any time to start a new decoding job. If this method is called while currently decoding another input, that decoding context is discarded. Note: It is not necessary to specify the encoding because the decoder assumes the input is in BER or DER format. Args: data (bytes): ASN.1 input, in BER or DER format, to be decoded. Returns: None Raises: `Error` zExpecting bytes instance.rN)r^rrHrLr)rdatarrrrNs  z Decoder.startcCs8|jdur td|rdS|jdur||_|jS)a This method returns the current ASN.1 tag (i.e. the tag that a subsequent `Decoder.read()` call would return) without updating the decoding offset. In case no more data is available from the input, this method returns ``None`` to signal end-of-file. This method is useful if you don't know whether the next tag will be a primitive or a constructed tag. Depending on the return value of `peek`, you would decide to either issue a `Decoder.read()` in case of a primitive type, or an `Decoder.enter()` in case of a constructed type. Note: Because this method does not advance the current offset in the input, calling it multiple times in a row will return the same value for all calls. Returns: `Tag`: The current ASN.1 tag. Raises: `Error` N&No input selected. Call start() first.)rLrH _end_of_inputr _read_tagrrrrpeeks   z Decoder.peekNcCsZ|jdur td|rdS|}|}|dur|j}||j||}d|_||fS)aB This method decodes one ASN.1 tag from the input and returns it as a ``(tag, value)`` tuple. ``tag`` is a 3-tuple ``(nr, typ, cls)``, while ``value`` is a Python object representing the ASN.1 value. The offset in the input is increased so that the next `Decoder.read()` call will return the next tag. In case no more data is available from the input, this method returns ``None`` to signal end-of-file. Returns: `Tag`, value: The current ASN.1 tag and its value. Raises: `Error` Nr) rLrHrr _read_lengthrS _read_valuerr)rtagnrtagrrrrrrreads z Decoder.readcCs|S)z Return True if we are at the end of input. Returns: bool: True if all input has been decoded, and False otherwise. )rrrrreofsz Decoder.eofcCsZ|jdur td|}|jtjkrtd|}||}|jd|gd|_ dS)a This method enters the constructed type that is at the current decoding offset. Note: It is an error to call `Decoder.enter()` if the to be decoded ASN.1 tag is not of a constructed type. Returns: None Nrz#Cannot enter a non-constructed tag.r) rLrHrrar;r=r _read_bytesrQr)rrrr bytes_datarrrrTs    z Decoder.entercCs:|jdur tdt|jdkrtd|jd=d|_dS)a This method leaves the last constructed type that was `Decoder.enter()`-ed. Note: It is an error to call `Decoder.leave()` if the current ASN.1 tag is not of a constructed type. Returns: None NrrrUrW)rLrHrXrrrrrr\'s  z Decoder.leavecCs|}|d@}|d@}|d@}|dkr)d} |}|d>|d@B}|d@s(nqzt|}Wn ty8Ynwzt|}Wn tyHYnw|tjkr^zt|}Wn ty]Ynwt|||d S) z, Read a tag from the input. rBr<rdrTrjrirA)rSrar) _read_byter; ValueErrorr?rCrrG)rbyterrarSrrrr:s:       zDecoder._read_tagcCs||}|d@r:|d@}|dkrtd||}d}|D] }|d>t|B}qzt|}W|Sty9Y|Sw|}|S)z/ Read a length from the input. rAriASN1 syntax errorrrt)rrHrr OverflowError)rrcountrrrrrrrXs$   zDecoder._read_lengthcCs||}|tjkr|}|S|tjkr||}|S|tjtjfvr)||}|S|tj kr5| |}|S|tj krA| |}|S|tj krM||}|S|tjtjtjtjtjfvrb||}|S|tjkrn||}|S|}|S)z. Read a value from the input. )rr?rCrr,_decode_booleanr-r2_decode_integerr/_decode_octet_stringr0 _decode_nullr1_decode_object_identifierr6r7r3r8r9_decode_printable_stringr._decode_bitstring)rrrSrrrrrrrrms<          zDecoder._read_valuecCsL|jd\}}z||}Wn tytdw|jddd7<|S)zP Return the next input byte, or raise an error on end-of-input. rWPremature end of input.rr)rL IndexErrorrH)rindex input_datarrrrrs  zDecoder._read_bytecCsL|jd\}}||||}t||krtd|jdd|7<|S)z` Return the next ``count`` bytes of input. Raise error on end-of-input. rWrr)rLrXrH)rrrrrrrrrs  zDecoder._read_bytescCs*|jd\}}|t|krJ|t|kS)z< Return True if we are at the end of input. rW)rLrX)rrrrrrrs zDecoder._end_of_inputcCs(t|dkr td|ddkrdSdS)z) Decode a boolean value. rrrFTrXrHrrrrrs  zDecoder._decode_booleancCsdd|D}t|dkr)|ddkr|dd@s%|ddkr)|dd@s)td|dd@}|ritt|D] }d||||<q7tt|dddD]}||d7<||dkr^n |dksdJd||<qLd}|D]}|d >|B}qm|r{| }zt|}W|StyY|Sw) z* Decode an integer value. cSsg|]}t|qSrr).0brrr sz+Decoder._decode_integer..rrrsrArrWrt)rXrHr r r)rrmrrrrnrrrrs2<      zDecoder._decode_integercCs|S)z) Decode an octet string. rrrrrrszDecoder._decode_octet_stringcCst|dkr tddS)z& Decode a Null value. rrNrrrrrrs zDecoder._decode_nullcCsg}d}tt|D]'}t||}|dkr|dkrtd|d>|d@B}|d@s1||d}q t|dkrd}d}tt|D]}t||}||?||>B||<||@}q@t|S)z% Decode a bitstring. rrrjrN)rXrH bytearrayr r r)rnum_unused_bits remainingbitmask removed_bitsrrrrrrs"     zDecoder._decode_bitstringr)rrrrIrMrNrrrrTr\rrrrrrrrrrrrrrrrrrrs:          r)rI __future__rrrr collectionsrbuiltinsrr r r contextlibr enumr __version__rrr;r? namedtuplerG ExceptionrHobjectrJrrrrrs2            Y