From 03fdb3cce08d8cebf6a3ca49f4b2179bdd659ac2 Mon Sep 17 00:00:00 2001 From: lanjelot Date: Sun, 13 Oct 2019 17:12:46 +1000 Subject: [PATCH] Add ssl option to tcp_fuzz --- patator.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/patator.py b/patator.py index a3cfc21..5f8adc4 100755 --- a/patator.py +++ b/patator.py @@ -899,6 +899,7 @@ import signal import ctypes import glob from xml.sax.saxutils import escape as xmlescape, quoteattr as xmlquoteattr +from ssl import wrap_socket try: # python3+ from queue import Empty, Full @@ -3112,8 +3113,6 @@ class Rlogin_login(TCP_Cache): # }}} # VMauthd {{{ -from ssl import wrap_socket - class LineReceiver_Error(Exception): pass @@ -4793,10 +4792,9 @@ class TCP_fuzz: Response = Response_Base - def execute(self, host, port, data='', timeout='2',ssl='0'): + def execute(self, host, port, data='', timeout='2', ssl='0'): fp = socket.create_connection((host, port), int(timeout)) - if ssl!='0': - from ssl import wrap_socket + if ssl != '0': fp = wrap_socket(fp) fp.send(data.decode('hex')) with Timing() as timing: