Prevent third-party plugins with faulty hooks to crash the server (#158)

Author: ccordoba12

pylsp/config/config.py

@@ -4,9 +4,11 @@
 
 import logging
 from functools import lru_cache
-import pkg_resources
+from typing import List, Mapping, Sequence, Union
 
+import pkg_resources
 import pluggy
+from pluggy._hooks import HookImpl
 
 from pylsp import _utils, hookspecs, uris, PYLSP
 
@@ -16,6 +18,24 @@
 DEFAULT_CONFIG_SOURCES = ['pycodestyle']
 
 
+class PluginManager(pluggy.PluginManager):
+
+    def _hookexec(
+        self,
+        hook_name: str,
+        methods: Sequence[HookImpl],
+        kwargs: Mapping[str, object],
+        firstresult: bool,
+    ) -> Union[object, List[object]]:
+        # called from all hookcaller instances.
+        # enable_tracing will set its own wrapping function at self._inner_hookexec
+        try:
+            return self._inner_hookexec(hook_name, methods, kwargs, firstresult)
+        except Exception as e:  # pylint: disable=broad-except
+            log.warning(f"Failed to load hook {hook_name}: {e}")
+            return []
+
+
 class Config:
     def __init__(self, root_uri, init_opts, process_id, capabilities):
         self._root_path = uris.to_fs_path(root_uri)
@@ -39,7 +59,7 @@ def __init__(self, root_uri, init_opts, process_id, capabilities):
         except ImportError:
             pass
 
-        self._pm = pluggy.PluginManager(PYLSP)
+        self._pm = PluginManager(PYLSP)
         self._pm.trace.root.setwriter(log.debug)
         self._pm.enable_tracing()
         self._pm.add_hookspecs(hookspecs)

setup.cfg

@@ -14,7 +14,7 @@ python_requires = >=3.7
 install_requires =
     jedi>=0.17.2,<0.19.0
     python-lsp-jsonrpc>=1.0.0
-    pluggy
+    pluggy>=1.0.0
     ujson>=3.0.0
     setuptools>=39.0.0
 setup_requires = setuptools>=44; wheel; setuptools_scm[toml]>=3.4.3