ctypescrypto/rand.py

   1 """
   2         Interface to the OpenSSL pseudo-random generator
   3 """
   4
   5 from ctypes import create_string_buffer, c_char_p, c_int, c_double
   6 from ctypescrypto import libcrypto
   7 from ctypescrypto.exception import LibCryptoError
   8
   9 __all__ = ['RandError','bytes','pseudo_bytes','seed','status']
  10
  11 class RandError(LibCryptoError):
  12         pass
  13
  14 def bytes( num, check_result=False):
  15         """
  16                 Returns num bytes of cryptographically strong pseudo-random
  17                 bytes. If checkc_result is True, raises error if PRNG is not
  18                 seeded enough
  19         """
  20
  21         if num <= 0 :
  22                 raise ValueError("'num' should be > 0")
  23         buffer = create_string_buffer(num)
  24         result = libcrypto.RAND_bytes(buffer, num)
  25         if check_result and result == 0:
  26                 raise RandError("Random Number Generator not seeded sufficiently")
  27         return buffer.raw[:num]
  28
  29 def pseudo_bytes(num):
  30         """
  31                 Returns num bytes of pseudo random data.  Pseudo- random byte
  32                 sequences generated by pseudo_bytes() will be unique if
  33                 they are of sufficient length, but are not necessarily
  34                 unpredictable. They can be used for non-cryptographic purposes
  35                 and for certain purposes in cryptographic protocols, but usually
  36                 not for key generation etc.
  37         """
  38         if num <= 0 :
  39                 raise ValueError("'num' should be > 0")
  40         buffer = create_string_buffer(num)
  41         libcrypto.RAND_pseudo_bytes(buffer, num)
  42         return buffer.raw[:num]
  43
  44 def seed(data, entropy=None):
  45         """
  46                 Seeds random generator with data.
  47                 If entropy is not None, it should be floating point(double)
  48                 value estimating amount of entropy  in the data (in bytes).
  49         """
  50         if type(data) != type(""):
  51                 raise TypeError("A string is expected")
  52         ptr = c_char_p(data)
  53         size = len(data)
  54         if entropy is None:
  55                 libcrypto.RAND_seed(ptr, size)
  56         else :
  57                 libcrypto.RAND_add(ptr, size, entropy)
  58
  59 def status():
  60         """
  61                 Returns 1 if random generator is sufficiently seeded and 0
  62                 otherwise
  63         """
  64
  65         return libcrypto.RAND_status()
  66
  67 libcrypto.RAND_add.argtypes=(c_char_p,c_int,c_double)
  68 libcrypto.RAND_seed.argtypes=(c_char_p,c_int)
  69 libcrypto.RAND_pseudo_bytes.argtypes=(c_char_p,c_int)
  70 libcrypto.RAND_bytes.argtypes=(c_char_p,c_int)