Fix #27: Close Pipes at parallel version of genprime.

Now closing pipes after reading result, and added a unittest.

1 files changed, 9 insertions, 5 deletions

diff --git a/rsa/parallel.py b/rsa/parallel.py
index e5034ac..17a10ed 100644
--- a/rsa/parallel.py
+++ b/rsa/parallel.py
@@ -65,11 +65,15 @@ def getprime(nbits, poolsize):
     (pipe_recv, pipe_send) = mp.Pipe(duplex=False)
 
     # Create processes
-    procs = [mp.Process(target=_find_prime, args=(nbits, pipe_send))
-             for _ in range(poolsize)]
-    [p.start() for p in procs]
-
-    result = pipe_recv.recv()
+    try:
+        procs = [mp.Process(target=_find_prime, args=(nbits, pipe_send))
+                 for _ in range(poolsize)]
+        [p.start() for p in procs]
+
+        result = pipe_recv.recv()
+    finally:
+        pipe_recv.close()
+        pipe_send.close()
 
     [p.terminate() for p in procs]