Fix for yet another race condition in LRU

andresriancho · Mar 26, 2019 · 7a375e6 · 7a375e6
1 parent bbdfc3f
commit 7a375e6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/w3af/core/data/misc/response_cache_key.py b/w3af/core/data/misc/response_cache_key.py
@@ -141,7 +141,7 @@ def get_response_cache_key(self,
 
         try:
             result = self._cache.get(cache_key, None)
-        except AssertionError:
+        except (AssertionError, KeyError) as _:
             # This is a rare race conditions which happens when another
             # thread modifies the cache and changes the __first item in
             # the cache.
@@ -165,7 +165,7 @@ def get_response_cache_key(self,
 
         try:
             self._cache[cache_key] = result
-        except AssertionError:
+        except (AssertionError, KeyError) as _:
             # This is a rare race conditions which happens when another
             # thread modifies the cache and changes the __first item in
             # the cache.