datatracker/ietf/utils/pipe.py
2019-06-30 21:01:44 +00:00

32 lines
899 B
Python

# Copyright The IETF Trust 2010-2019, All Rights Reserved
# Simplified interface to os.popen3()
def pipe(cmd:bytes, str:bytes=None) -> (int, bytes, bytes):
from subprocess import Popen, PIPE
bufsize = 4096
MAX = 65536*16
if str and len(str) > 4096: # XXX: Hardcoded Linux 2.4, 2.6 pipe buffer size
bufsize = len(str)
pipe = Popen(cmd, stdin=PIPE, stdout=PIPE, stderr=PIPE, bufsize=bufsize, shell=True)
if not str is None:
pipe.stdin.write(str)
pipe.stdin.close()
out = b""
err = b""
while True:
str = pipe.stdout.read()
if str:
out += str
code = pipe.poll()
if code > -1:
err = pipe.stderr.read()
break
if len(out) >= MAX:
err = "Output exceeds %s bytes and has been truncated" % MAX
break
return (code, out, err)