43 lines
1.5 KiB
Python
43 lines
1.5 KiB
Python
from network import HostFinger
|
|
from model.host import VictimHost
|
|
|
|
|
|
class HTTPFinger(HostFinger):
|
|
"""
|
|
Goal is to recognise HTTP servers, where what we currently care about is apache.
|
|
"""
|
|
|
|
def __init__(self):
|
|
self._config = __import__('config').WormConfiguration
|
|
self.HTTP = [(port, str(port)) for port in self._config.HTTP_PORTS]
|
|
|
|
@staticmethod
|
|
def _banner_match(service, host, banner):
|
|
pass
|
|
|
|
def get_host_fingerprint(self, host):
|
|
assert isinstance(host, VictimHost)
|
|
from requests import head
|
|
from requests.exceptions import Timeout, ConnectionError
|
|
from contextlib import closing
|
|
|
|
for port in self.HTTP:
|
|
# check both http and https
|
|
http = "http://" + host.ip_addr + ":" + port[1]
|
|
https = "https://" + host.ip_addr + ":" + port[1]
|
|
|
|
# try http, we don't optimise for 443
|
|
for url in (https, http): # start with https and downgrade
|
|
try:
|
|
with closing(head(url, verify=False, timeout=1)) as req:
|
|
server = req.headers.get('Server')
|
|
ssl = True if 'https://' in url else False
|
|
host.services['tcp-' + port[1]] = (server,ssl)
|
|
break # https will be the same on the same port
|
|
except Timeout:
|
|
pass
|
|
except ConnectionError: # Someone doesn't like us
|
|
pass
|
|
|
|
return True
|