elie/ssh-audit
1
0
Fork 0
mirror of https://github.com/jtesta/ssh-audit.git synced 2024-06-30 19:24:46 +02:00
Code Issues Packages Projects Releases Wiki Activity

Do not capture unnecessary regex groups.

Browse Source
This commit is contained in:
Andris Raugulis 2016-09-07 19:22:47 +03:00
parent c68211b8e7
commit d07d5078cb
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
ssh-audit.py
View File

@ -210,8 +210,9 @@ class SSH(object):
MSG_KEXDH_REPLY = 32
class Banner(object):
_RXP, _RXR = r'SSH-(\d)\.\s*?(\d+)', r'(|-([^\s]*)(\s+(.*))?)'
RX_BANNER = re.compile(r'^({0}((-{0})*)){1}$'.format(_RXP, _RXR))
_RXP, _RXR = r'SSH-\d\.\s*?\d+', r'(-([^\s]*)(?:\s+(.*))?)?'
RX_PROTOCOL = re.compile(_RXP.replace('\d', '(\d)'))
RX_BANNER = re.compile(r'^({0}(?:(?:-{0})*)){1}$'.format(_RXP, _RXR))
def __init__(self, protocol, software, comments):
self.__protocol = protocol
@ -252,12 +253,12 @@ class SSH(object):
mx = cls.RX_BANNER.match(banner)
if mx is None:
return None
protocol = min(re.findall(cls._RXP, mx.group(1)))
protocol = min(re.findall(cls.RX_PROTOCOL, mx.group(1)))
protocol = (int(protocol[0]), int(protocol[1]))
software = (mx.group(9) or '').strip() or None
if software is None and mx.group(8).startswith('-'):
software = (mx.group(3) or '').strip() or None
if software is None and (mx.group(2) or '').startswith('-'):
software = ''
comments = (mx.group(11) or '').strip() or None
comments = (mx.group(4) or '').strip() or None
return cls(protocol, software, comments)
class Socket(ReadBuf, WriteBuf):