Spaces:
Running
Running
File size: 2,876 Bytes
41e79e2 |
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 |
from . import run
import time
from pathlib import Path
ROOT = Path(__file__).resolve().parent / "evil_programs"
def assert_no_running_evil():
result = run(
["pgrep", "-f", ROOT], timeout_seconds=1, max_output_size=1024
)
assert (
result.exit_code == 1
), f"There are still evil processes running: {result.stdout}"
assert len(result.stderr) == 0
assert len(result.stdout) == 0
def test_fork_once():
# The program exits cleanly and immediately. But, it forks a child that runs
# forever.
result = run(
["python3", ROOT / "fork_once.py"],
timeout_seconds=2,
max_output_size=1024,
)
assert result.exit_code == 0
assert result.timeout == False
assert len(result.stderr) == 0
assert len(result.stdout) == 0
assert_no_running_evil()
def test_close_outputs():
# The program prints to stdout, closes its output, and then runs forever.
result = run(
["python3", ROOT / "close_outputs.py"],
timeout_seconds=2,
max_output_size=1024,
)
assert result.exit_code == -1
assert result.timeout == True
assert len(result.stderr) == 0
assert result.stdout == "This is the end\n"
assert_no_running_evil()
def test_unbounded_output():
result = run(
["python3", ROOT / "unbounded_output.py"],
timeout_seconds=3,
max_output_size=1024,
)
assert result.exit_code == -1
assert result.timeout == True
assert len(result.stderr) == 0
assert len(result.stdout) == 1024
assert_no_running_evil()
def test_sleep_forever():
result = run(
["python3", ROOT / "sleep_forever.py"],
timeout_seconds=2,
max_output_size=1024,
)
assert result.exit_code == -1
assert result.timeout == True
assert len(result.stderr) == 0
assert len(result.stdout) == 0
assert_no_running_evil()
def test_fork_bomb():
result = run(
["python3", ROOT / "fork_bomb.py"],
timeout_seconds=2,
max_output_size=1024,
)
assert result.exit_code == -1
assert result.timeout == True
assert len(result.stderr) == 0
assert len(result.stdout) == 0
# Unfortunately, this sleep seems to be necessary. My theories:
# 1. os.killpg doesn't block until the whole process group is dead.
# 2. pgrep can produce stale output
time.sleep(2)
assert_no_running_evil()
def test_block_on_inputs():
# We run the subprocess with /dev/null as input. So, any program that tries
# to read input will error.
result = run(
["python3", ROOT / "block_on_inputs.py"],
timeout_seconds=2,
max_output_size=1024,
)
assert result.exit_code == 1
assert result.timeout == False
assert len(result.stdout) == 0
assert "EOF when reading a line" in result.stderr
assert_no_running_evil()
|