From fb4c897ec609484022d69ac93f28b9ca87d3364e Mon Sep 17 00:00:00 2001 From: Diego Date: Thu, 20 Apr 2017 22:53:32 +0200 Subject: [PATCH] improve __random_bytes function performance --- magneticod/magneticod/bittorrent.py | 3 ++- magneticod/magneticod/dht.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/magneticod/magneticod/bittorrent.py b/magneticod/magneticod/bittorrent.py index 472dae4..f499d0b 100644 --- a/magneticod/magneticod/bittorrent.py +++ b/magneticod/magneticod/bittorrent.py @@ -19,6 +19,7 @@ import math import socket import random import typing +import os from . import bencode @@ -287,4 +288,4 @@ class DisposablePeer: @staticmethod def __random_bytes(n: int) -> bytes: - return random.getrandbits(n * 8).to_bytes(n, "big") + return os.urandom(n) diff --git a/magneticod/magneticod/dht.py b/magneticod/magneticod/dht.py index a4437c4..8362d34 100644 --- a/magneticod/magneticod/dht.py +++ b/magneticod/magneticod/dht.py @@ -19,6 +19,7 @@ import logging import random import socket import typing +import os from . import bencode @@ -244,7 +245,7 @@ class SybilNode: @staticmethod def __random_bytes(n: int) -> bytes: - return random.getrandbits(n * 8).to_bytes(n, "big") + return os.urandom(n) def __build_FIND_NODE_query(self, id_: bytes) -> bytes: """ BENCODE IMPLEMENTATION