Fix multiple commands in judging

wxh06 · Jun 10, 2020 · 3dc2539 · 3dc2539
1 parent d0fe740
commit 3dc2539
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/dockerjudge/__init__.py b/dockerjudge/__init__.py
@@ -43,9 +43,13 @@ def _judge(dir, container, commands, ioput, timeout, iofile) -> (str, float):
                             '{}/{}'.format(shlex.quote(ioput[0]),
                                            dir, iofile[0])])
         result = container.exec_run(['bash', '-c', 'cd {}&&time timeout -s '
-                                     'KILL {} {}<{}'.format(dir, timeout,
-                                                            commands[1] % '..',
-                                                            iofile[0])],
+                                     'KILL {} sh -c {}<{}'.format(dir, timeout,
+                                                                  shlex.quote(
+                                                                      commands
+                                                                      [1]
+                                                                      % '..'
+                                                                  ),
+                                                                  iofile[0])],
                                     demux=True)
     else:
         result = container.exec_run(['bash', '-c', 'cd {}&&time echo {}|'