crypto.py 553 B

123456789101112131415
  1. import os as overlay_fake_os
  2. from .crypto_orig import *
  3. if overlay_fake_os.getenv('MMGEN_TEST_SUITE_DETERMINISTIC'):
  4. overlay_fake_get_random_orig = Crypto.get_random
  5. overlay_fake_add_user_random_orig = Crypto.add_user_random
  6. from .test import fake_urandom as overlay_fake_urandom
  7. Crypto.get_random = lambda self, length: overlay_fake_urandom(
  8. len(overlay_fake_get_random_orig(self, length)))
  9. Crypto.add_user_random = lambda self, rand_bytes, desc: overlay_fake_urandom(
  10. len(overlay_fake_add_user_random_orig(self, rand_bytes, desc=desc)))