1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
|
# Copyright 2010-2024 Gentoo Authors
# Distributed under the terms of the GNU General Public License v2
import tempfile
import time
from portage import os
from portage import shutil
from portage import _python_interpreter
from portage.tests import TestCase
from portage.const import PORTAGE_BIN_PATH
from portage.const import PORTAGE_PYM_PATH
from portage.const import BASH_BINARY
from portage.locks import hardlock_cleanup
from portage.package.ebuild._ipc.ExitCommand import ExitCommand
from portage.util import ensure_dirs
from portage.util._async.ForkProcess import ForkProcess
from portage.util._async.TaskScheduler import TaskScheduler
from portage.util._eventloop.global_event_loop import global_event_loop
from _emerge.SpawnProcess import SpawnProcess
from _emerge.EbuildBuildDir import EbuildBuildDir
from _emerge.EbuildIpcDaemon import EbuildIpcDaemon
class IpcDaemonTestCase(TestCase):
_SCHEDULE_TIMEOUT = 40 # seconds
def testIpcDaemon(self):
event_loop = global_event_loop()
tmpdir = tempfile.mkdtemp()
build_dir = None
try:
env = {}
# Pass along PORTAGE_USERNAME and PORTAGE_GRPNAME since they
# need to be inherited by ebuild subprocesses.
if "PORTAGE_USERNAME" in os.environ:
env["PORTAGE_USERNAME"] = os.environ["PORTAGE_USERNAME"]
if "PORTAGE_GRPNAME" in os.environ:
env["PORTAGE_GRPNAME"] = os.environ["PORTAGE_GRPNAME"]
env["PORTAGE_PYTHON"] = _python_interpreter
env["PORTAGE_BIN_PATH"] = PORTAGE_BIN_PATH
env["PORTAGE_PYM_PATH"] = PORTAGE_PYM_PATH
env["PORTAGE_BUILDDIR"] = os.path.join(tmpdir, "cat", "pkg-1")
env["PYTHONDONTWRITEBYTECODE"] = os.environ.get(
"PYTHONDONTWRITEBYTECODE", ""
)
if "__PORTAGE_TEST_HARDLINK_LOCKS" in os.environ:
env["__PORTAGE_TEST_HARDLINK_LOCKS"] = os.environ[
"__PORTAGE_TEST_HARDLINK_LOCKS"
]
build_dir = EbuildBuildDir(scheduler=event_loop, settings=env)
event_loop.run_until_complete(build_dir.async_lock())
ensure_dirs(os.path.join(env["PORTAGE_BUILDDIR"], ".ipc"))
input_fifo = os.path.join(env["PORTAGE_BUILDDIR"], ".ipc", "in")
output_fifo = os.path.join(env["PORTAGE_BUILDDIR"], ".ipc", "out")
os.mkfifo(input_fifo)
os.mkfifo(output_fifo)
for exitcode in (0, 1, 2):
exit_command = ExitCommand()
commands = {"exit": exit_command}
daemon = EbuildIpcDaemon(
commands=commands, input_fifo=input_fifo, output_fifo=output_fifo
)
proc = SpawnProcess(
args=[
BASH_BINARY,
"-c",
'"$PORTAGE_BIN_PATH"/ebuild-ipc exit %d' % exitcode,
],
env=env,
)
task_scheduler = TaskScheduler(
iter([daemon, proc]), max_jobs=2, event_loop=event_loop
)
self.assertTrue(task_scheduler._loop is event_loop._loop)
self.received_command = False
def exit_command_callback():
self.received_command = True
task_scheduler.cancel()
exit_command.reply_hook = exit_command_callback
start_time = time.time()
self._run(event_loop, task_scheduler, self._SCHEDULE_TIMEOUT)
hardlock_cleanup(env["PORTAGE_BUILDDIR"], remove_all_locks=True)
self.assertEqual(
self.received_command,
True,
"command not received after %d seconds"
% (time.time() - start_time,),
)
self.assertEqual(proc.isAlive(), False)
self.assertEqual(daemon.isAlive(), False)
self.assertEqual(exit_command.exitcode, exitcode)
# Intentionally short timeout test for EventLoop/AsyncScheduler.
# Use a ridiculously long sleep_time_s in case the user's
# system is heavily loaded (see bug #436334).
sleep_time_s = 600 # seconds
short_timeout_s = 0.010 # seconds
for i in range(3):
exit_command = ExitCommand()
commands = {"exit": exit_command}
daemon = EbuildIpcDaemon(
commands=commands, input_fifo=input_fifo, output_fifo=output_fifo
)
# Emulate the sleep command, in order to ensure a consistent
# return code when it is killed by SIGTERM (see bug #437180).
proc = ForkProcess(target=time.sleep, args=(sleep_time_s,))
task_scheduler = TaskScheduler(
iter([daemon, proc]), max_jobs=2, event_loop=event_loop
)
self.received_command = False
def exit_command_callback():
self.received_command = True
task_scheduler.cancel()
exit_command.reply_hook = exit_command_callback
start_time = time.time()
self._run(event_loop, task_scheduler, short_timeout_s)
hardlock_cleanup(env["PORTAGE_BUILDDIR"], remove_all_locks=True)
self.assertEqual(
self.received_command,
False,
"command received after %d seconds" % (time.time() - start_time,),
)
self.assertEqual(proc.isAlive(), False)
self.assertEqual(daemon.isAlive(), False)
self.assertEqual(proc.returncode == os.EX_OK, False)
finally:
if build_dir is not None:
event_loop.run_until_complete(build_dir.async_unlock())
shutil.rmtree(tmpdir)
def _timeout_callback(self, task_scheduler):
task_scheduler.cancel()
self._exit_callback(task_scheduler)
def _exit_callback(self, task_scheduler):
if not self._run_done.done():
self._run_done.set_result(True)
def _run(self, event_loop, task_scheduler, timeout):
self._run_done = event_loop.create_future()
timeout_handle = event_loop.call_later(
timeout, self._timeout_callback, task_scheduler
)
task_scheduler.addExitListener(self._exit_callback)
async def start_task_scheduler():
# This fails unless the event loop is running, since it needs
# the loop to setup a ChildWatcher.
task_scheduler.start()
try:
event_loop.run_until_complete(start_task_scheduler())
event_loop.run_until_complete(self._run_done)
event_loop.run_until_complete(task_scheduler.async_wait())
finally:
timeout_handle.cancel()
|