Avoid O(N**2) bottleneck in _communicate_(). Fixes #1598181.

This commit is contained in:
Peter Astrand 2007-01-07 14:34:16 +00:00
parent ec05a2d580
commit 1812f8cf3f
1 changed files with 4 additions and 3 deletions

View File

@ -1111,6 +1111,7 @@ class Popen(object):
read_set.append(self.stderr)
stderr = []
input_offset = 0
while read_set or write_set:
rlist, wlist, xlist = select.select(read_set, write_set, [])
@ -1118,9 +1119,9 @@ class Popen(object):
# When select has indicated that the file is writable,
# we can write up to PIPE_BUF bytes without risk
# blocking. POSIX defines PIPE_BUF >= 512
bytes_written = os.write(self.stdin.fileno(), input[:512])
input = input[bytes_written:]
if not input:
bytes_written = os.write(self.stdin.fileno(), buffer(input, input_offset, 512))
input_offset += bytes_written
if input_offset >= len(input):
self.stdin.close()
write_set.remove(self.stdin)