Setting timeout to 2 and refactoring a little

Setting default timeout to 2 seconds.  Also made the run_as_nobody a
function so the rest of the server can be tested without the need to
sudo it.

1 files changed, 7 insertions, 5 deletions

diff --git a/python_server.py b/python_server.py
index 6ade0ff..dee5631 100755
--- a/python_server.py
+++ b/python_server.py
@@ -12,12 +12,13 @@ from os.path import isdir
 import signal
 
 # Timeout for the code to run in seconds.
-TIMEOUT = 3
+TIMEOUT = 2
 
-# Set the effective uid
-nobody = pwd.getpwnam('nobody')
-os.setegid(nobody.pw_gid)
-os.seteuid(nobody.pw_uid)
+def run_as_nobody():
+    # Set the effective uid
+    nobody = pwd.getpwnam('nobody')
+    os.setegid(nobody.pw_gid)
+    os.seteuid(nobody.pw_uid)
 
 # Raised when the code times-out.
 # c.f. http://pguides.net/python/timeout-a-function
@@ -82,6 +83,7 @@ def run_code(answer, test_code, in_dir=None):
 
 
 def main():
+    run_as_nobody()
     server = SimpleXMLRPCServer(("localhost", 8001))
     server.register_function(run_code)
     server.serve_forever()