Allow Java access to all parent directories of agent

Riolku · Sep 5, 2021 · e20c1d6 · e20c1d6
1 parent 31a4862
commit e20c1d6
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/dmoj/executors/java_executor.py b/dmoj/executors/java_executor.py
@@ -4,6 +4,7 @@
 import subprocess
 import sys
 from collections import deque
+from pathlib import PurePath
 from typing import Optional
 
 from dmoj.error import CompileError, InternalError
@@ -81,7 +82,8 @@ def get_executable(self):
         return self.get_vm()
 
     def get_fs(self):
-        return super().get_fs() + [self._agent_file]
+        return super().get_fs() + [f'{re.escape(self._agent_file)}$'] + \
+            [f'{re.escape(str(parent))}$' for parent in PurePath(self._agent_file).parents]
 
     def get_write_fs(self):
         return super().get_write_fs() + [os.path.join(self._dir, 'submission_jvm_crash.log')]