Richard Oudkerk | 84ed9a6 | 2013-08-14 15:35:41 +0100 | [diff] [blame] | 1 | import io |
| 2 | import os |
| 3 | |
Davin Potts | 5458647 | 2016-09-09 18:03:10 -0500 | [diff] [blame] | 4 | from .context import reduction, set_spawning_popen |
Richard Oudkerk | 84ed9a6 | 2013-08-14 15:35:41 +0100 | [diff] [blame] | 5 | from . import popen_fork |
Richard Oudkerk | 84ed9a6 | 2013-08-14 15:35:41 +0100 | [diff] [blame] | 6 | from . import spawn |
| 7 | from . import util |
| 8 | |
Richard Oudkerk | 84ed9a6 | 2013-08-14 15:35:41 +0100 | [diff] [blame] | 9 | __all__ = ['Popen'] |
| 10 | |
| 11 | |
| 12 | # |
| 13 | # Wrapper for an fd used while launching a process |
| 14 | # |
| 15 | |
| 16 | class _DupFd(object): |
| 17 | def __init__(self, fd): |
| 18 | self.fd = fd |
| 19 | def detach(self): |
| 20 | return self.fd |
| 21 | |
| 22 | # |
| 23 | # Start child process using a fresh interpreter |
| 24 | # |
| 25 | |
| 26 | class Popen(popen_fork.Popen): |
| 27 | method = 'spawn' |
| 28 | DupFd = _DupFd |
| 29 | |
| 30 | def __init__(self, process_obj): |
| 31 | self._fds = [] |
| 32 | super().__init__(process_obj) |
| 33 | |
| 34 | def duplicate_for_child(self, fd): |
| 35 | self._fds.append(fd) |
| 36 | return fd |
| 37 | |
| 38 | def _launch(self, process_obj): |
Richard Oudkerk | 7d2d43c | 2013-08-22 11:38:57 +0100 | [diff] [blame] | 39 | from . import semaphore_tracker |
Richard Oudkerk | b1694cf | 2013-10-16 16:41:56 +0100 | [diff] [blame] | 40 | tracker_fd = semaphore_tracker.getfd() |
Richard Oudkerk | 84ed9a6 | 2013-08-14 15:35:41 +0100 | [diff] [blame] | 41 | self._fds.append(tracker_fd) |
| 42 | prep_data = spawn.get_preparation_data(process_obj._name) |
| 43 | fp = io.BytesIO() |
Davin Potts | 5458647 | 2016-09-09 18:03:10 -0500 | [diff] [blame] | 44 | set_spawning_popen(self) |
Richard Oudkerk | 84ed9a6 | 2013-08-14 15:35:41 +0100 | [diff] [blame] | 45 | try: |
| 46 | reduction.dump(prep_data, fp) |
| 47 | reduction.dump(process_obj, fp) |
| 48 | finally: |
Davin Potts | 5458647 | 2016-09-09 18:03:10 -0500 | [diff] [blame] | 49 | set_spawning_popen(None) |
Richard Oudkerk | 84ed9a6 | 2013-08-14 15:35:41 +0100 | [diff] [blame] | 50 | |
| 51 | parent_r = child_w = child_r = parent_w = None |
| 52 | try: |
Victor Stinner | daf4555 | 2013-08-28 00:53:59 +0200 | [diff] [blame] | 53 | parent_r, child_w = os.pipe() |
| 54 | child_r, parent_w = os.pipe() |
Richard Oudkerk | 7d2d43c | 2013-08-22 11:38:57 +0100 | [diff] [blame] | 55 | cmd = spawn.get_command_line(tracker_fd=tracker_fd, |
| 56 | pipe_handle=child_r) |
Richard Oudkerk | 84ed9a6 | 2013-08-14 15:35:41 +0100 | [diff] [blame] | 57 | self._fds.extend([child_r, child_w]) |
| 58 | self.pid = util.spawnv_passfds(spawn.get_executable(), |
| 59 | cmd, self._fds) |
| 60 | self.sentinel = parent_r |
| 61 | with open(parent_w, 'wb', closefd=False) as f: |
| 62 | f.write(fp.getbuffer()) |
| 63 | finally: |
| 64 | if parent_r is not None: |
| 65 | util.Finalize(self, os.close, (parent_r,)) |
| 66 | for fd in (child_r, child_w, parent_w): |
| 67 | if fd is not None: |
| 68 | os.close(fd) |