From: Gregory P. Smith <greg@mad-scientist.com>
Date: Tue, 2 Sep 2008 05:36:11 +0000 (+0000)
Subject: Issue #3708: os.urandom no longer goes into an infinite loop when passed a
X-Git-Tag: v2.6rc1~81
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=d712203d17d406b3ebdb1276e5f1694a2a7db120;p=python

Issue #3708: os.urandom no longer goes into an infinite loop when passed a
non-integer floating point number.
---

diff --git a/Lib/os.py b/Lib/os.py
index 715ff202f4..abbadc41e4 100644
--- a/Lib/os.py
+++ b/Lib/os.py
@@ -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
diff --git a/Lib/test/test_os.py b/Lib/test/test_os.py
index cec023d308..a618af42b4 100644
--- a/Lib/test/test_os.py
+++ b/Lib/test/test_os.py
@@ -497,6 +497,10 @@ class URandomTests (unittest.TestCase):
             self.assertEqual(len(os.urandom(10)), 10)
             self.assertEqual(len(os.urandom(100)), 100)
             self.assertEqual(len(os.urandom(1000)), 1000)
+            # see http://bugs.python.org/issue3708
+            self.assertEqual(len(os.urandom(0.9)), 0)
+            self.assertEqual(len(os.urandom(1.1)), 1)
+            self.assertEqual(len(os.urandom(2.0)), 2)
         except NotImplementedError:
             pass
 
diff --git a/Misc/NEWS b/Misc/NEWS
index a1074736ae..00dfd33d82 100644
--- a/Misc/NEWS
+++ b/Misc/NEWS
@@ -72,6 +72,9 @@ Library
 - Issue #3703: _fileio.FileIO gave unhelpful error message when trying to open a
   directory.
 
+- Issue #3708: os.urandom no longer goes into an infinite loop when passed a
+  non-integer floating point number.
+
 Extension Modules
 -----------------