Grab SSH Banner

grab_banner.py

# !/usr/bin/python
import socket
import json
import re
import time
import random
from multiprocessing import Pool, TimeoutError
socket.setdefaulttimeout(4)

def grab_banner(ip_address, port):
    ip_address = ip_address.strip()
    #print(ip_address)
    time.sleep(random.randint(0, 1000) / 1000)
    for i in range(2):
        try:
            s = socket.socket()
            s.connect((ip_address, port))
            banner = s.recv(1024)
            s.close()
            return banner
        except socket.gaierror:
            if i == 1:
                return b'dnsfail'
        except socket.timeout:
            if i == 1:
                return b'timeout'
        i += 1
        time.sleep(random.randint(100, 2000)/1000)



if __name__ == '__main__':
    // Load IPS
    ips = []
    results = []
    with Pool(processes=50) as pool:
            for ip in ips:
                  results.append( ( pool.apply_async(grab_banner, (ip, 22)), ip) )
                  if len(results) == 50:
                      for r in results:
                          print("%s;%s"%(r[1], r[0].get()))
                      results = []
        pass