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
|
# Copyright (c) 2025, Oracle and/or its affiliates.
# SPDX-License-Identifier: LGPL-2.1-or-later
import os
import sys
import tempfile
import traceback
import types
import drgn.cli
from tests import TestCase
class TestCli(TestCase):
def run_cli(self, args, *, input=None):
stdout_r, stdout_w = os.pipe()
stderr_r, stderr_w = os.pipe()
if input is not None:
stdin_r, stdin_w = os.pipe()
pid = os.fork()
if pid == 0:
try:
os.close(stdout_r)
sys.stdout = open(stdout_w, "w")
os.close(stderr_r)
sys.stderr = open(stderr_w, "w")
if input is not None:
os.close(stdin_w)
sys.stdin = open(stdin_r, "r")
sys.argv = ["drgn"] + args
drgn.cli._main()
finally:
exception = sys.exc_info()[1] is not None
if exception:
traceback.print_exc()
sys.stdout.flush()
sys.stderr.flush()
os._exit(1 if exception else 0)
os.close(stdout_w)
os.close(stderr_w)
if input is not None:
os.close(stdin_r)
with open(stdin_w, "w") as f:
f.write(input)
with open(stdout_r, "r") as f:
stdout = f.read()
with open(stderr_r, "r") as f:
stderr = f.read()
_, wstatus = os.waitpid(pid, 0)
if not os.WIFEXITED(wstatus) or os.WEXITSTATUS(wstatus) != 0:
if os.WIFEXITED(wstatus):
msg = f"Exited with status {os.WEXITSTATUS(wstatus)}"
elif os.WIFSIGNALED(wstatus):
msg = f"Terminated by signal {os.WTERMSIG(wstatus)}"
else:
msg = "Exited abnormally"
self.fail(
f"""\
{msg}
STDOUT:
{stdout}
STDERR:
{stderr}
"""
)
return types.SimpleNamespace(stdout=stdout, stderr=stderr)
def test_e(self):
script = r"""
import sys
assert drgn.get_default_prog() is prog
assert __name__ == "__main__"
assert "__file__" not in globals()
assert sys.path[0] == ""
print(sys.argv)
"""
proc = self.run_cli(
["--quiet", "--pid", "0", "--no-default-symbols", "-e", script, "pass"]
)
self.assertEqual(proc.stdout, "['-e', 'pass']\n")
def test_e_empty(self):
self.run_cli(
["--quiet", "--pid", "0", "--no-default-symbols", "-e", ""],
# This shouldn't be executed.
input="raise Exception('-e was ignored')",
)
def test_script(self):
with tempfile.NamedTemporaryFile() as f:
f.write(
rb"""
assert "drgn" not in globals()
import drgn
import os.path
import sys
assert drgn.get_default_prog() is prog
assert __name__ == "__main__"
assert __file__ == sys.argv[0]
assert sys.path[0] == os.path.dirname(__file__)
print(sys.argv)
"""
)
f.flush()
proc = self.run_cli(
["--quiet", "--pid", "0", "--no-default-symbols", f.name, "pass"]
)
self.assertEqual(proc.stdout, f"[{f.name!r}, 'pass']\n")
def test_pipe(self):
script = r"""
import sys
assert drgn.get_default_prog() is prog
assert __name__ == "__main__"
assert __file__ == "<stdin>"
assert sys.path[0] == ""
# Dummy if statement to test handling of multi-line blocks.
if True:
print(sys.argv)
"""
proc = self.run_cli(
["--quiet", "--pid", "0", "--no-default-symbols"], input=script
)
self.assertEqual(proc.stdout, "['']\n")
|
