mcp-proxy-adapter 6.2.24__py3-none-any.whl → 6.2.26__py3-none-any.whl

mcp_proxy_adapter/examples/run_security_tests.py

@@ -81,11 +81,15 @@ class SecurityTestRunner:
     def _pids_on_port(self, port: int) -> List[int]:
         pids: List[int] = []
         try:
-            for proc in psutil.process_iter(attrs=["pid", "connections"]):
-                for c in proc.connections(kind="inet"):
-                    if c.laddr and c.laddr.port == port:
-                        pids.append(proc.pid)
-                        break
+            for proc in psutil.process_iter(attrs=["pid"]):
+                try:
+                    connections = proc.connections(kind="inet")
+                    for c in connections:
+                        if c.laddr and c.laddr.port == port:
+                            pids.append(proc.pid)
+                            break
+                except (psutil.NoSuchProcess, psutil.AccessDenied):
+                    pass
         except Exception:
             pass
         return list(set(pids))
@@ -187,7 +191,12 @@ class SecurityTestRunner:
             "certs/localhost_server.crt",
             "keys/localhost_server.key"
         ]
+        
         missing_certs = []
+        # Check if roles.json exists
+        roles_file = "configs/roles.json"
+        if not os.path.exists(roles_file):
+            missing_certs.append(f"Missing roles file: {roles_file}")
         for cert_file in cert_files:
             if not Path(cert_file).exists():
                 missing_certs.append(cert_file)