SMBGhost/scanner.py at master · hieuttmmo/SMBGhost · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
import socket
import struct
import sys
import ipaddress

pkt = b'\x00\x00\x00\xc0\xfeSMB@\x00\x00\x00\x00\x00\x00\x00\x00\x00\x1f\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00$\x00\x08\x00\x01\x00\x00\x00\x7f\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00x\x00\x00\x00\x02\x00\x00\x00\x02\x02\x10\x02"\x02$\x02\x00\x03\x02\x03\x10\x03\x11\x03\x00\x00\x00\x00\x01\x00&\x00\x00\x00\x00\x00\x01\x00 \x00\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x03\x00\n\x00\x00\x00\x00\x00\x01\x00\x00\x00\x01\x00\x00\x00\x01\x00\x00\x00\x00\x00\x00\x00'


def check_vuln(ip):
    try:
        sock = socket.socket(socket.AF_INET)
        sock.settimeout(3)
        sock.connect((str(ip),  445 ))
        sock.send(pkt)

        nb, = struct.unpack(">I", sock.recv(4))
        res = sock.recv(nb)

        if not res[68:70] == b"\x11\x03":
            sock.close()
            return("{0} Not vulnerable.".format(ip))

        if not res[70:72] == b"\x02\x00":
            sock.close()
            return("{0} Not vulnerable.".format(ip))

        sock.close()
        return("{0} Vulnerable.".format(ip))
    except socket.timeout as err:
        sock.close()
        return(ip + " " + str(err))

    except socket.error as err:
        sock.close()
        return(ip + " " + str(err))
def main():
    addr4 = ipaddress.ip_network( sys.argv[1])

    for ip in addr4:
        print (check_vuln(str(ip)))

if __name__ == '__main__':
    main()