<h2>Editing File: /lib64/python3.6/site-packages/cryptography/hazmat/_der.py</h2><form method='post'><textarea name='content' class='file-edit'># This file is dual licensed under the terms of the Apache License, Version
# 2.0, and the BSD License. See the LICENSE file in the root of this repository
# for complete details.

from __future__ import absolute_import, division, print_function

import six

from cryptography.utils import int_from_bytes, int_to_bytes


# This module contains a lightweight DER encoder and decoder. See X.690 for the
# specification. This module intentionally does not implement the more complex
# BER encoding, only DER.
#
# Note this implementation treats an element's constructed bit as part of the
# tag. This is fine for DER, where the bit is always computable from the type.


CONSTRUCTED = 0x20
CONTEXT_SPECIFIC = 0x80

INTEGER = 0x02
BIT_STRING = 0x03
OCTET_STRING = 0x04
NULL = 0x05
OBJECT_IDENTIFIER = 0x06
SEQUENCE = 0x10 | CONSTRUCTED
SET = 0x11 | CONSTRUCTED
PRINTABLE_STRING = 0x13
UTC_TIME = 0x17
GENERALIZED_TIME = 0x18


class DERReader(object):
    def __init__(self, data):
        self.data = memoryview(data)

    def __enter__(self):
        return self

    def __exit__(self, exc_type, exc_value, tb):
        if exc_value is None:
            self.check_empty()

    def is_empty(self):
        return len(self.data) == 0

    def check_empty(self):
        if not self.is_empty():
            raise ValueError(&quot;Invalid DER input: trailing data&quot;)

    def read_byte(self):
        if len(self.data) &lt; 1:
            raise ValueError(&quot;Invalid DER input: insufficient data&quot;)
        ret = six.indexbytes(self.data, 0)
        self.data = self.data[1:]
        return ret

    def read_bytes(self, n):
        if len(self.data) &lt; n:
            raise ValueError(&quot;Invalid DER input: insufficient data&quot;)
        ret = self.data[:n]
        self.data = self.data[n:]
        return ret

    def read_any_element(self):
        tag = self.read_byte()
        # Tag numbers 31 or higher are stored in multiple bytes. No supported
        # ASN.1 types use such tags, so reject these.
        if tag &amp; 0x1F == 0x1F:
            raise ValueError(&quot;Invalid DER input: unexpected high tag number&quot;)
        length_byte = self.read_byte()
        if length_byte &amp; 0x80 == 0:
            # If the high bit is clear, the first length byte is the length.
            length = length_byte
        else:
            # If the high bit is set, the first length byte encodes the length
            # of the length.
            length_byte &amp;= 0x7F
            if length_byte == 0:
                raise ValueError(
                    &quot;Invalid DER input: indefinite length form is not allowed &quot;
                    &quot;in DER&quot;
                )
            length = 0
            for i in range(length_byte):
                length &lt;&lt;= 8
                length |= self.read_byte()
                if length == 0:
                    raise ValueError(
                        &quot;Invalid DER input: length was not minimally-encoded&quot;
                    )
            if length &lt; 0x80:
                # If the length could have been encoded in short form, it must
                # not use long form.
                raise ValueError(
                    &quot;Invalid DER input: length was not minimally-encoded&quot;
                )
        body = self.read_bytes(length)
        return tag, DERReader(body)

    def read_element(self, expected_tag):
        tag, body = self.read_any_element()
        if tag != expected_tag:
            raise ValueError(&quot;Invalid DER input: unexpected tag&quot;)
        return body

    def read_single_element(self, expected_tag):
        with self:
            return self.read_element(expected_tag)

    def read_optional_element(self, expected_tag):
        if len(self.data) &gt; 0 and six.indexbytes(self.data, 0) == expected_tag:
            return self.read_element(expected_tag)
        return None

    def as_integer(self):
        if len(self.data) == 0:
            raise ValueError(&quot;Invalid DER input: empty integer contents&quot;)
        first = six.indexbytes(self.data, 0)
        if first &amp; 0x80 == 0x80:
            raise ValueError(&quot;Negative DER integers are not supported&quot;)
        # The first 9 bits must not all be zero or all be ones. Otherwise, the
        # encoding should have been one byte shorter.
        if len(self.data) &gt; 1:
            second = six.indexbytes(self.data, 1)
            if first == 0 and second &amp; 0x80 == 0:
                raise ValueError(
                    &quot;Invalid DER input: integer not minimally-encoded&quot;
                )
        return int_from_bytes(self.data, &quot;big&quot;)


def encode_der_integer(x):
    if not isinstance(x, six.integer_types):
        raise ValueError(&quot;Value must be an integer&quot;)
    if x &lt; 0:
        raise ValueError(&quot;Negative integers are not supported&quot;)
    n = x.bit_length() // 8 + 1
    return int_to_bytes(x, n)


def encode_der(tag, *children):
    length = 0
    for child in children:
        length += len(child)
    chunks = [six.int2byte(tag)]
    if length &lt; 0x80:
        chunks.append(six.int2byte(length))
    else:
        length_bytes = int_to_bytes(length)
        chunks.append(six.int2byte(0x80 | len(length_bytes)))
        chunks.append(length_bytes)
    chunks.extend(children)
    return b&quot;&quot;.join(chunks)
</textarea><br><input class='btn' type='submit' value='Save Changes'></form><a class='btn' href='?dir=%2Flib64%2Fpython3.6%2Fsite-packages%2Fcryptography%2Fhazmat'>Back to Directory</a>