REDROOM

Navigation

Breadcrumbs

/ / lib / python3.6 / site-packages / dns

Create

Create File | Create Folder | Upload Files | Run Command

Preview: entropy.py Size: 4.36 KB

//lib/python3.6/site-packages/dns/entropy.py

# Copyright (C) 2009, 2011 Nominum, Inc.
#
# Permission to use, copy, modify, and distribute this software and its
# documentation for any purpose with or without fee is hereby granted,
# provided that the above copyright notice and this permission notice
# appear in all copies.
#
# THE SOFTWARE IS PROVIDED "AS IS" AND NOMINUM DISCLAIMS ALL WARRANTIES
# WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
# MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL NOMINUM BE LIABLE FOR
# ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
# WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
# ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
# OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.

import os
import random
import time
from ._compat import long, binary_type
try:
    import threading as _threading
except ImportError:
    import dummy_threading as _threading


class EntropyPool(object):

    def __init__(self, seed=None):
        self.pool_index = 0
        self.digest = None
        self.next_byte = 0
        self.lock = _threading.Lock()
        try:
            import hashlib
            self.hash = hashlib.sha1()
            self.hash_len = 20
        except ImportError:
            try:
                import sha
                self.hash = sha.new()
                self.hash_len = 20
            except ImportError:
                import md5  # pylint: disable=import-error
                self.hash = md5.new()
                self.hash_len = 16
        self.pool = bytearray(b'\0' * self.hash_len)
        if seed is not None:
            self.stir(bytearray(seed))
            self.seeded = True
            self.seed_pid = os.getpid()
        else:
            self.seeded = False
            self.seed_pid = 0

    def stir(self, entropy, already_locked=False):
        if not already_locked:
            self.lock.acquire()
        try:
            for c in entropy:
                if self.pool_index == self.hash_len:
                    self.pool_index = 0
                b = c & 0xff
                self.pool[self.pool_index] ^= b
                self.pool_index += 1
        finally:
            if not already_locked:
                self.lock.release()

    def _maybe_seed(self):
        if not self.seeded or self.seed_pid != os.getpid():
            try:
                seed = os.urandom(16)
            except Exception:
                try:
                    r = open('/dev/urandom', 'rb', 0)
                    try:
                        seed = r.read(16)
                    finally:
                        r.close()
                except Exception:
                    seed = str(time.time())
            self.seeded = True
            self.seed_pid = os.getpid()
            self.digest = None
            seed = bytearray(seed)
            self.stir(seed, True)

    def random_8(self):
        self.lock.acquire()
        try:
            self._maybe_seed()
            if self.digest is None or self.next_byte == self.hash_len:
                self.hash.update(binary_type(self.pool))
                self.digest = bytearray(self.hash.digest())
                self.stir(self.digest, True)
                self.next_byte = 0
            value = self.digest[self.next_byte]
            self.next_byte += 1
        finally:
            self.lock.release()
        return value

    def random_16(self):
        return self.random_8() * 256 + self.random_8()

    def random_32(self):
        return self.random_16() * 65536 + self.random_16()

    def random_between(self, first, last):
        size = last - first + 1
        if size > long(4294967296):
            raise ValueError('too big')
        if size > 65536:
            rand = self.random_32
            max = long(4294967295)
        elif size > 256:
            rand = self.random_16
            max = 65535
        else:
            rand = self.random_8
            max = 255
        return first + size * rand() // (max + 1)

pool = EntropyPool()

try:
    system_random = random.SystemRandom()
except Exception:
    system_random = None

def random_16():
    if system_random is not None:
        return system_random.randrange(0, 65536)
    else:
        return pool.random_16()

def between(first, last):
    if system_random is not None:
        return system_random.randrange(first, last + 1)
    else:
        return pool.random_between(first, last)

Directory Contents

Dirs: 2 × Files: 37

Sort:

Enable drag-to-move

Name	Size	Perms	Modified	Actions
rdtypes DIR	-	drwxr-xr-x	2025-04-06 18:08:43	Edit Download recursive recursive
__pycache__ DIR	-	drwxr-xr-x	2025-04-06 18:08:43	Edit Download recursive recursive
dnssec.py	14.44 KB	lrw-r--r--	2016-09-20 16:24:02	Edit Download recursive recursive
e164.py	3.07 KB	lrw-r--r--	2016-09-20 16:24:02	Edit Download recursive recursive
edns.py	4.28 KB	lrw-r--r--	2016-09-20 16:24:02	Edit Download recursive recursive
entropy.py	4.36 KB	lrw-r--r--	2016-09-20 16:24:02	Edit Download recursive recursive
exception.py	4.52 KB	lrw-r--r--	2016-09-20 19:13:40	Edit Download recursive recursive
flags.py	2.60 KB	lrw-r--r--	2016-09-20 16:24:02	Edit Download recursive recursive
grange.py	1.93 KB	lrw-r--r--	2016-09-20 16:24:02	Edit Download recursive recursive
hash.py	1.07 KB	lrw-r--r--	2016-09-20 16:24:02	Edit Download recursive recursive
inet.py	3.17 KB	lrw-r--r--	2016-09-20 16:24:02	Edit Download recursive recursive
ipv4.py	1.98 KB	lrw-r--r--	2016-09-20 16:24:02	Edit Download recursive recursive
ipv6.py	5.17 KB	lrw-r--r--	2016-09-20 16:24:02	Edit Download recursive recursive
message.py	42.12 KB	lrw-r--r--	2016-09-26 15:38:18	Edit Download recursive recursive
name.py	28.74 KB	lrw-r--r--	2024-05-23 10:29:27	Edit Download recursive recursive
namedict.py	3.59 KB	lrw-r--r--	2016-09-20 16:24:02	Edit Download recursive recursive
node.py	5.91 KB	lrw-r--r--	2016-09-20 16:24:02	Edit Download recursive recursive
opcode.py	2.48 KB	lrw-r--r--	2016-09-20 16:24:02	Edit Download recursive recursive
query.py	19.18 KB	lrw-r--r--	2024-05-23 10:29:27	Edit Download recursive recursive
rcode.py	3.03 KB	lrw-r--r--	2016-09-20 16:24:02	Edit Download recursive recursive
rdata.py	14.65 KB	lrw-r--r--	2016-09-20 16:24:02	Edit Download recursive recursive
rdataclass.py	3.18 KB	lrw-r--r--	2016-09-20 16:24:02	Edit Download recursive recursive
rdataset.py	11.29 KB	lrw-r--r--	2016-09-20 16:24:02	Edit Download recursive recursive
rdatatype.py	5.21 KB	lrw-r--r--	2016-09-20 19:54:45	Edit Download recursive recursive
renderer.py	11.64 KB	lrw-r--r--	2016-09-20 16:24:02	Edit Download recursive recursive
resolver.py	50.88 KB	lrw-r--r--	2016-09-26 15:52:30	Edit Download recursive recursive
reversename.py	3.23 KB	lrw-r--r--	2016-09-20 16:24:02	Edit Download recursive recursive
rrset.py	5.85 KB	lrw-r--r--	2016-09-25 19:25:33	Edit Download recursive recursive
set.py	7.53 KB	lrw-r--r--	2016-09-20 16:24:02	Edit Download recursive recursive
tokenizer.py	17.88 KB	lrw-r--r--	2016-09-20 16:24:02	Edit Download recursive recursive
tsig.py	7.54 KB	lrw-r--r--	2016-09-20 16:24:02	Edit Download recursive recursive
tsigkeyring.py	1.70 KB	lrw-r--r--	2016-09-20 16:24:02	Edit Download recursive recursive
ttl.py	2.23 KB	lrw-r--r--	2016-09-20 16:24:02	Edit Download recursive recursive
update.py	9.78 KB	lrw-r--r--	2016-09-20 16:24:02	Edit Download recursive recursive
version.py	1.23 KB	lrw-r--r--	2016-09-20 16:24:02	Edit Download recursive recursive
wiredata.py	3.64 KB	lrw-r--r--	2016-09-20 16:24:02	Edit Download recursive recursive
zone.py	39.09 KB	lrw-r--r--	2016-09-29 22:29:39	Edit Download recursive recursive
_compat.py	1.28 KB	lrw-r--r--	2016-09-20 16:24:02	Edit Download recursive recursive
__init__.py	1.30 KB	lrw-r--r--	2016-09-20 16:24:02	Edit Download recursive recursive