Update app.py

app.py

@@ -19,6 +19,7 @@ import json
 import resource
 import platform
 from loguru import logger
+import subprocess
 
 app = FastAPI(title="Code Execution and Input Analysis API", docs_url="/")
 
@@ -62,6 +63,68 @@ class CodeExecutor:
             "java": "javac",
             "python": "python3",
         }
+        # Check and log ulimits on startup
+        self._check_ulimits()
+    
+    def _check_ulimits(self):
+        """Check current ulimits and log them"""
+        try:
+            # Check various ulimits
+            limits = {
+                "RLIMIT_NOFILE": resource.RLIMIT_NOFILE,  # file descriptors
+                "RLIMIT_NPROC": resource.RLIMIT_NPROC,    # processes
+                "RLIMIT_STACK": resource.RLIMIT_STACK,    # stack size
+            }
+            
+            logger.info("Current ulimits:")
+            for name, limit_type in limits.items():
+                try:
+                    soft, hard = resource.getrlimit(limit_type)
+                    logger.info(f"{name}: soft={soft}, hard={hard}")
+                except:
+                    pass
+                    
+            # Try to increase limits if they're too low
+            try:
+                # Increase file descriptors
+                resource.setrlimit(resource.RLIMIT_NOFILE, (65536, 65536))
+            except:
+                try:
+                    # Try a lower value if the high one fails
+                    resource.setrlimit(resource.RLIMIT_NOFILE, (4096, 4096))
+                except:
+                    pass
+                    
+        except Exception as e:
+            logger.warning(f"Could not check/set ulimits: {e}")
+
+    async def _create_java_wrapper(self, workspace: str, jvm_opts: List[str]) -> str:
+        """Create a wrapper script for Java execution with ulimit settings"""
+        
+        wrapper_content = f"""#!/bin/bash
+# Set ulimits to prevent memory allocation issues
+ulimit -n 65536 2>/dev/null || ulimit -n 4096
+ulimit -u 32768 2>/dev/null || ulimit -u 2048
+ulimit -s 8192 2>/dev/null || ulimit -s 4096
+ulimit -v unlimited 2>/dev/null || true
+ulimit -m unlimited 2>/dev/null || true
+
+# Log current ulimits for debugging
+echo "Current ulimits:" >&2
+ulimit -a >&2
+
+# Execute Java with the provided options
+exec java {' '.join(jvm_opts)} "$@"
+"""
+        
+        wrapper_path = os.path.join(workspace, "java_wrapper.sh")
+        with open(wrapper_path, 'w') as f:
+            f.write(wrapper_content)
+        
+        # Make executable
+        os.chmod(wrapper_path, 0o755)
+        
+        return wrapper_path
 
     def set_resource_limits(self):
         """Set resource limits for subprocess (Unix only)"""
@@ -72,6 +135,32 @@ class CodeExecutor:
             resource.setrlimit(resource.RLIMIT_AS, (MAX_MEMORY, MAX_MEMORY))
 
     def check_available_memory(self) -> int:
+        """Create a wrapper script for Java execution with ulimit settings"""
+        
+        wrapper_content = f"""#!/bin/bash
+# Set ulimits to prevent memory allocation issues
+ulimit -n 65536 2>/dev/null || ulimit -n 4096
+ulimit -u 32768 2>/dev/null || ulimit -u 2048
+ulimit -s 8192 2>/dev/null || ulimit -s 4096
+ulimit -v unlimited 2>/dev/null || true
+ulimit -m unlimited 2>/dev/null || true
+
+# Log current ulimits for debugging
+echo "Current ulimits:" >&2
+ulimit -a >&2
+
+# Execute Java with the provided options
+exec java {' '.join(jvm_opts)} "$@"
+"""
+        
+        wrapper_path = os.path.join(workspace, "java_wrapper.sh")
+        with open(wrapper_path, 'w') as f:
+            f.write(wrapper_content)
+        
+        # Make executable
+        os.chmod(wrapper_path, 0o755)
+        
+        return wrapper_path
         """Check available memory in MB"""
         try:
             with open('/proc/meminfo', 'r') as f:
@@ -564,41 +653,38 @@ class CodeExecutor:
             
             # Try different JVM configurations
             jvm_configs = [
-                # Config 1: Minimal memory with explicit small code cache
+                # Config 1: Ultra minimal with no code cache reservation
                 [
-                    "-XX:+UseSerialGC",
+                    "-Xint",  # Interpreter only mode (no JIT)
                     "-Xmx32m",
-                    "-Xms4m",
-                    "-XX:ReservedCodeCacheSize=16m",
-                    "-XX:InitialCodeCacheSize=4m",
-                    "-XX:MaxMetaspaceSize=32m",
-                    "-XX:CompressedClassSpaceSize=8m",
-                    "-XX:-TieredCompilation",
-                    "-XX:CICompilerCount=1",
-                    "-Xss256k",
-                    "-XX:+UseStringDeduplication",
-                    "-XX:MaxDirectMemorySize=16m",
+                    "-Xms2m",
+                    "-XX:MaxMetaspaceSize=16m",
                     "-Djava.awt.headless=true",
-                    "-Dfile.encoding=UTF-8",
                 ],
-                # Config 2: Even more minimal
+                # Config 2: Minimal with small code cache
                 [
-                    "-Xmx16m",
-                    "-Xms2m",
+                    "-XX:+UseSerialGC",
+                    "-Xmx32m",
+                    "-Xms4m",
                     "-XX:ReservedCodeCacheSize=8m",
-                    "-XX:MaxMetaspaceSize=16m",
+                    "-XX:InitialCodeCacheSize=2m",
+                    "-XX:MaxMetaspaceSize=24m",
                     "-XX:-TieredCompilation",
                     "-Djava.awt.headless=true",
                 ],
-                # Config 3: Absolute minimum
+                # Config 3: Slightly more memory
                 [
-                    "-Xmx8m",
-                    "-XX:ReservedCodeCacheSize=4m",
+                    "-Xmx48m",
+                    "-Xms8m",
+                    "-XX:ReservedCodeCacheSize=16m",
+                    "-XX:MaxMetaspaceSize=32m",
+                    "-XX:-TieredCompilation",
                     "-Djava.awt.headless=true",
                 ],
-                # Config 4: Let JVM decide with hints
+                # Config 4: Use container support
                 [
-                    "-XX:+UseSerialGC",
+                    "-XX:+UseContainerSupport",
+                    "-XX:MaxRAMPercentage=25.0",
                     "-XX:-TieredCompilation",
                     "-Djava.awt.headless=true",
                 ],
@@ -610,7 +696,12 @@ class CodeExecutor:
                     logger.info(f"Trying JVM config {config_idx + 1} for {class_name}")
                     start_time = asyncio.get_event_loop().time()
                     
-                    java_cmd = ["java"] + jvm_opts + [class_name]
+                    # For first two configs, try with wrapper script
+                    if config_idx < 2:
+                        wrapper_path = await self._create_java_wrapper(workspace, jvm_opts)
+                        java_cmd = [wrapper_path, class_name]
+                    else:
+                        java_cmd = ["java"] + jvm_opts + [class_name]
                     
                     stdout, stderr, returncode = await self._execute_with_input(
                         java_cmd, workspace, input_data