Update pc_agent.py

pc_agent.py

@@ -1,26 +1,52 @@
 import subprocess, os, tempfile, shutil
 from pathlib import Path
 
-WORKSPACE = Path(tempfile.mkdtemp(prefix="zai_workspace_"))
+# مجلد عمل مرن يُفحص ويُعاد إنشاؤه تلقائياً عند الحاجة
+WORKSPACE_ROOT = Path(tempfile.gettempdir()) / "zai_workspace"
+
+def ensure_workspace():
+    WORKSPACE_ROOT.mkdir(parents=True, exist_ok=True)
+    (WORKSPACE_ROOT / "uploads").mkdir(exist_ok=True)
+    return WORKSPACE_ROOT
 
 def run_command(cmd: str, timeout: int = 30) -> dict:
-    dangerous = ["rm -rf /", "mkfs", "dd if=", "shutdown", "reboot", "curl", "wget"]
-    if any(d in cmd.lower() for d in dangerous):
-        return {"output": "⛔ أمر غير مسموح به لأسباب أمنية.", "error": True}
+    ws = ensure_workspace()
+    
+    # قائمة سوداء شاملة + منع تجاوز الصلاحيات
+    dangerous_patterns = [
+        "rm -rf", "mkfs", "dd if=", "shutdown", "reboot", "curl", "wget",
+        "pip install", "pip3", "python -c", "python3 -c", "bash -c", "sh -c",
+        "eval(", "exec(", "sudo", "chmod 777", "chown", "nc ", "nmap", "base64 -d"
+    ]
+    cmd_lower = cmd.lower()
+    if any(p in cmd_lower for p in dangerous_patterns):
+        return {"output": "⛔ أمر محظور لأسباب أمنية.", "error": True, "code": 403}
+    
+    # عزل البيئة لمنع الوصول لمتغيرات النظام الحساسة
+    safe_env = {"PATH": "/usr/local/bin:/usr/bin:/bin", "HOME": str(ws), "TERM": "xterm"}
+    
     try:
         result = subprocess.run(
-            cmd, shell=True, cwd=str(WORKSPACE),
-            capture_output=True, text=True, timeout=timeout
+            cmd, shell=True, cwd=str(ws),
+            capture_output=True, text=True, timeout=timeout, env=safe_env
         )
         out = result.stdout.strip() or result.stderr.strip() or "(لا يوجد مخرج)"
-        return {"output": out, "error": result.returncode != 0}
+        return {"output": out, "error": result.returncode != 0, "code": 200 if result.returncode == 0 else 400}
     except subprocess.TimeoutExpired:
-        return {"output": "⏳ انتهى الوقت المحدد للأمر (30 ثانية).", "error": True}
+        return {"output": "⏳ انتهى الوقت المحدد للأمر (30 ثانية).", "error": True, "code": 408}
     except Exception as e:
-        return {"output": f"❌ خطأ: {str(e)}", "error": True}
+        return {"output": f"❌ خطأ تنفيذ: {str(e)}", "error": True, "code": 500}
 
 def list_files():
-    return [str(p.relative_to(WORKSPACE)) for p in WORKSPACE.rglob("*") if p.is_file()]
+    ws = ensure_workspace()
+    return [str(p.relative_to(ws)) for p in ws.rglob("*") if p.is_file()]
 
 def cleanup():
-    shutil.rmtree(WORKSPACE, ignore_errors=True)
+    ws = ensure_workspace()
+    # حذف المحتويات فقط مع الحفاظ على مجلد العمل نفسه
+    for item in ws.iterdir():
+        if item.is_dir():
+            shutil.rmtree(item)
+        else:
+            item.unlink()
+    return True