From bbe41bc900f2216639fe2aa8d3ce5a57abf5794a Mon Sep 17 00:00:00 2001 From: Vangelis Banos Date: Thu, 9 May 2019 15:57:01 +0000 Subject: [PATCH] Add bad_hostnames_ports in PlaybackProxy These vars are required also there in addition to `SingleThreadedWarcProxy`. --- warcprox/playback.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/warcprox/playback.py b/warcprox/playback.py index 91f86aa..8bfa42f 100644 --- a/warcprox/playback.py +++ b/warcprox/playback.py @@ -42,6 +42,7 @@ from warcprox.mitmproxy import MitmProxyHandler import warcprox import sqlite3 import threading +from cachetools import TTLCache class PlaybackProxyHandler(MitmProxyHandler): logger = logging.getLogger("warcprox.playback.PlaybackProxyHandler") @@ -219,6 +220,8 @@ class PlaybackProxy(socketserver.ThreadingMixIn, http_server.HTTPServer): self.playback_index_db = playback_index_db self.warcs_dir = options.directory self.options = options + self.bad_hostnames_ports = TTLCache(maxsize=1024, ttl=60) + self.bad_hostnames_ports_lock = threading.RLock() def server_activate(self): http_server.HTTPServer.server_activate(self)