1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
|
#!/usr/bin/env python3
# focused host memory configuration and growth checks.
import os
import subprocess
source = '''fn main() {
int initial = syscall(9, 0, 0, 0);
put_n(syscall(9, 5, initial, 0) == initial); put_c(32);
put_n(syscall(9, 4, initial + 4096, 0) == -1); put_c(32);
put_n(syscall(9, 5, initial + 8192, 0) == initial + 8192); put_c(32);
put_n(syscall(9, 4, initial + 4096, 0) == initial + 4096); put_c(32);
put_n(syscall(9, 5, initial - 1, 0) == -1); put_c(10);
return 0;
}
'''
commands = [['bin/zy'], ['bin/zym'], ['node', 'src/js/tinyvm.js']]
environment = dict(os.environ, TINYVM_MAX_MEMORY_WORDS='16777216')
for command in commands:
result = subprocess.run(command, input=source, text=True, capture_output=True, env=environment, timeout=15)
assert result.returncode == 0 and result.stdout == '1 1 1 1 1\n', (command, result.stdout, result.stderr)
invalid = subprocess.run(command, input=source, text=True, capture_output=True, env=dict(environment, TINYVM_MEMORY_WORDS='1'), timeout=15)
assert invalid.returncode != 0 and invalid.stderr, (command, invalid.stdout)
print('ok runtime memory limits, denied growth, successful growth, and invalid startup: ' + command[0])
|
