-
Notifications
You must be signed in to change notification settings - Fork 0
/
port_scanner.py
40 lines (32 loc) · 988 Bytes
/
port_scanner.py
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
import socket
from utils import timefunc
class Scanner:
def __init__(self, ip):
self.ip = ip
self.open_ports = [];
def __repr__(self):
return 'Scanner: {}'.format(self.ip)
def add_port(self, port):
self.open_ports.append(port)
def scan(self, lowerport, upperport):
for port in range(lowerport, upperport + 1):
if self.is_open(port):
self.add_port(port)
def is_open(self, port):
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
result = s.connect_ex((self.ip, port))
s.close()
return result == 0
def write(self, filepath):
openports = map(str, self.open_ports)
with open(filepath, 'w') as f:
f.write('\n'.join(openports))
@timefunc
def main():
ip = '192.168.42.42'
scanner = Scanner(ip)
scanner.scan(1, 65535)
scanner.write('./open_ports')
print(scanner.open_ports)
if __name__ == '__main__':
main()