diff mercurial/commandserver.py @ 37124:6715e8035b4f

procutil: introduce context-manager interface for protect/restorestdio

The code looks slightly cleaner since it was pretty easy to pass arguments
in wrong order.

--- a/mercurial/commandserver.py	Sun Mar 25 11:40:30 2018 +0900
+++ b/mercurial/commandserver.py	Sun Mar 25 11:58:05 2018 +0900
@@ -318,13 +318,12 @@
         ui = self.ui
         # redirect stdio to null device so that broken extensions or in-process
         # hooks will never cause corruption of channel protocol.
-        fin, fout = procutil.protectstdio(ui.fin, ui.fout)
-        try:
-            sv = server(ui, self.repo, fin, fout)
-            return sv.serve()
-        finally:
-            sv.cleanup()
-            procutil.restorestdio(ui.fin, ui.fout, fin, fout)
+        with procutil.protectedstdio(ui.fin, ui.fout) as (fin, fout):
+            try:
+                sv = server(ui, self.repo, fin, fout)
+                return sv.serve()
+            finally:
+                sv.cleanup()
 
 def _initworkerprocess():
     # use a different process group from the master process, in order to: