Rob Wu
e9f22af949
Before this patch, an attempt to read from a program that closed its stdout would result in a never-resolving read promise on Windows. Before this patch, an attempt to write to a program that closed its stdin would result in a never-resolving write promise on Windows. This patch fixes both issues by fixing the typo that prevented the implementation from detecting errors, and adds test coverage. Differential Revision: https://phabricator.services.mozilla.com/D261332
79 lines
2.1 KiB
Python
79 lines
2.1 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import os
|
|
import signal
|
|
import struct
|
|
import sys
|
|
|
|
|
|
def output(line, stream=sys.stdout, print_only=False):
|
|
if isinstance(line, str):
|
|
line = line.encode("utf-8", "surrogateescape")
|
|
if not print_only:
|
|
stream.buffer.write(struct.pack("@I", len(line)))
|
|
stream.buffer.write(line)
|
|
stream.flush()
|
|
|
|
|
|
def echo_loop():
|
|
while True:
|
|
line = sys.stdin.buffer.readline()
|
|
if not line:
|
|
break
|
|
|
|
output(line)
|
|
|
|
|
|
if sys.platform == "win32":
|
|
import msvcrt
|
|
|
|
msvcrt.setmode(sys.stderr.fileno(), os.O_BINARY)
|
|
|
|
|
|
cmd = sys.argv[1]
|
|
if cmd == "echo":
|
|
echo_loop()
|
|
elif cmd == "exit":
|
|
sys.exit(int(sys.argv[2]))
|
|
elif cmd == "env":
|
|
for var in sys.argv[2:]:
|
|
output(os.environ.get(var, "!"))
|
|
elif cmd == "pwd":
|
|
output(os.path.abspath(os.curdir))
|
|
elif cmd == "print_args":
|
|
for arg in sys.argv[2:]:
|
|
output(arg)
|
|
elif cmd == "print_python_executable_path":
|
|
import psutil # not stdlib, put already part of our virtualenv.
|
|
|
|
# This returns the final executable that launched this script.
|
|
# We cannot use sys.executable because on Windows that would point to the
|
|
# python.exe parent that spawns this process.
|
|
# See getRealPythonExecutable in test_subprocess.js.
|
|
output(psutil.Process().exe())
|
|
elif cmd == "ignore_sigterm":
|
|
signal.signal(signal.SIGTERM, signal.SIG_IGN)
|
|
|
|
output("Ready")
|
|
while True:
|
|
try:
|
|
signal.pause()
|
|
except AttributeError:
|
|
import time
|
|
|
|
time.sleep(3600)
|
|
elif cmd == "close_stdin_and_wait_forever":
|
|
# Do NOT use non-stdlib modules here; this runs outside our virtualenv,
|
|
# via the program identified by print_python_executable_path.
|
|
os.close(sys.stdin.fileno())
|
|
output("stdin_closed")
|
|
while True:
|
|
pass # Test should kill the program
|
|
elif cmd == "close_pipes_and_wait_for_stdin":
|
|
os.close(sys.stdout.fileno())
|
|
os.close(sys.stderr.fileno())
|
|
sys.stdin.buffer.read(1)
|
|
elif cmd == "print":
|
|
output(sys.argv[2], stream=sys.stdout, print_only=True)
|
|
output(sys.argv[3], stream=sys.stderr, print_only=True)
|