[PATCH 4 of 5] procutil: unroll uin/uout loop in protectstdio()
Yuya Nishihara
yuya at tcha.org
Sun Mar 25 08:54:15 EDT 2018
# HG changeset patch
# User Yuya Nishihara <yuya at tcha.org>
# Date 1521947238 -32400
# Sun Mar 25 12:07:18 2018 +0900
# Node ID 768b03c733d00da88fa99a698c0274ec5aa2cf5a
# Parent a315bc2695f7faff23e71acea37d2c4a13961b14
procutil: unroll uin/uout loop in protectstdio()
I'll change uout to be redirected to stderr.
diff --git a/mercurial/utils/procutil.py b/mercurial/utils/procutil.py
--- a/mercurial/utils/procutil.py
+++ b/mercurial/utils/procutil.py
@@ -221,17 +221,18 @@ def protectstdio(uin, uout):
"owned" in that print(), exec(), etc. never reach to them.
"""
uout.flush()
- newfiles = []
nullfd = os.open(os.devnull, os.O_RDWR)
- for f, sysf, mode in [(uin, stdin, r'rb'),
- (uout, stdout, r'wb')]:
- if f is sysf:
- newfd = os.dup(f.fileno())
- os.dup2(nullfd, f.fileno())
- f = os.fdopen(newfd, mode)
- newfiles.append(f)
+ fin, fout = uin, uout
+ if uin is stdin:
+ newfd = os.dup(uin.fileno())
+ os.dup2(nullfd, uin.fileno())
+ fin = os.fdopen(newfd, r'rb')
+ if uout is stdout:
+ newfd = os.dup(uout.fileno())
+ os.dup2(nullfd, uout.fileno())
+ fout = os.fdopen(newfd, r'wb')
os.close(nullfd)
- return tuple(newfiles)
+ return fin, fout
def restorestdio(uin, uout, fin, fout):
"""Restore (uin, uout) streams from possibly duplicated (fin, fout)"""
More information about the Mercurial-devel
mailing list