blob: b90655b5eb98222946fe5aca022f609b2086d0ed (
plain)
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
|
#!/usr/bin/env python
"""This server runs an XMLRPC server that can be submitted code and tests
and returns the output. It *should* be run as root and will run as the user
'nobody' so as to minimize any damange by errant code.
"""
import traceback
from SimpleXMLRPCServer import SimpleXMLRPCServer
import pwd
import os
from os.path import isdir
# Set the effective uid
nobody = pwd.getpwnam('nobody')
os.setegid(nobody.pw_gid)
os.seteuid(nobody.pw_uid)
def run_code(answer, test_code, in_dir=None):
"""Tests given Python function (`answer`) with the `test_code` supplied.
If the optional `in_dir` keyword argument is supplied it changes the
directory to that directory (it does not change it back to the original when
done).
Returns
-------
A tuple: (success, error message).
"""
if in_dir is not None and isdir(in_dir):
os.chdir(in_dir)
try:
submitted = compile(answer, '<string>', mode='exec')
g = {}
exec submitted in g
_tests = compile(test_code, '<string>', mode='exec')
exec _tests in g
except:
success = False
err = traceback.format_exc(limit=1)
else:
success = True
err = 'Correct answer'
return success, err
def main():
server = SimpleXMLRPCServer(("localhost", 8001))
server.register_function(run_code)
server.serve_forever()
if __name__ == '__main__':
main()
|
