Issue #3708: os.urandom no longer goes into an infinite loop when passed a

non-integer floating point number.
This commit is contained in:
Gregory P. Smith 2008-09-02 05:36:11 +00:00
parent 2bb25cc1e2
commit d712203d17
3 changed files with 14 additions and 5 deletions

View file

@ -753,8 +753,10 @@ if not _exists("urandom"):
_urandomfd = open("/dev/urandom", O_RDONLY)
except (OSError, IOError):
raise NotImplementedError("/dev/urandom (or equivalent) not found")
bytes = ""
while len(bytes) < n:
bytes += read(_urandomfd, n - len(bytes))
close(_urandomfd)
return bytes
try:
bs = b""
while n - len(bs) >= 1:
bs += read(_urandomfd, n - len(bs))
finally:
close(_urandomfd)
return bs