diff --git a/libs/sslstripplus/ClientRequest.py b/libs/sslstripplus/ClientRequest.py index f3628b5..d28d7db 100644 --- a/libs/sslstripplus/ClientRequest.py +++ b/libs/sslstripplus/ClientRequest.py @@ -32,7 +32,7 @@ from ServerConnection import ServerConnection from SSLServerConnection import SSLServerConnection from URLMonitor import URLMonitor from CookieCleaner import CookieCleaner -from DnsCache import DnsCache +from libs.sslstrip.DnsCache import DnsCache from libs.sergioproxy.ProxyPlugins import ProxyPlugins class ClientRequest(Request): diff --git a/libs/sslstripplus/DnsCache.py b/libs/sslstripplus/DnsCache.py deleted file mode 100644 index 45e19d5..0000000 --- a/libs/sslstripplus/DnsCache.py +++ /dev/null @@ -1,41 +0,0 @@ -import logging - -class DnsCache: - - ''' - The DnsCache maintains a cache of DNS lookups, mirroring the browser experience. - ''' - - _instance = None - - def __init__(self): - self.customAddress = None - self.cache = {} - - def cacheResolution(self, host, address): - self.cache[host] = address - - def getCachedAddress(self, host): - if host in self.cache: - return self.cache[host] - - return None - - def getInstance(): - if DnsCache._instance == None: - DnsCache._instance = DnsCache() - - return DnsCache._instance - - def setCustomRes(self, host, ip_address=None): - if ip_address is not None: - self.cache[host] = ip_address - logging.debug("DNS entry set: %s -> %s" %(host, ip_address)) - else: - if self.customAddress is not None: - self.cache[host] = self.customAddress - - def setCustomAddress(self, ip_address): - self.customAddress = ip_address - - getInstance = staticmethod(getInstance) \ No newline at end of file