-
Notifications
You must be signed in to change notification settings - Fork 8
/
scanner_04.py
69 lines (56 loc) · 2.06 KB
/
scanner_04.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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
import argparse
import nmap
import csv
import os
import sys
def scan_host(ip, ports):
nm = nmap.PortScanner()
nm.scan(ip, ports)
host_infos = []
for proto in nm[ip].all_protocols():
lport = nm[ip][proto].keys()
for port in lport:
host_info = {
'ip': ip,
'os': nm[ip].get('osclass', {}).get('osfamily', 'Unknown'),
'port': port,
'name': nm[ip][proto][port]['name'],
'product': nm[ip][proto][port]['product'],
'version': nm[ip][proto][port]['version'],
}
host_infos.append(host_info)
return host_infos
def output_to_csv(output_file, host_info):
fieldnames = ["ip", "os", "port", "name", "product", "version"]
file_exists = os.path.isfile(output_file)
with open(output_file, "a") as f:
writer = csv.DictWriter(f, fieldnames=fieldnames)
if not file_exists:
writer.writeheader()
writer.writerow(host_info)
def main():
parser = argparse.ArgumentParser(description="Scan a single host for open ports and services")
parser.add_argument("host", help="The target host IP address")
parser.add_argument("-p", "--ports", help="Ports to scan", type=str, required=True)
parser.add_argument("-o", "--output", help="The output file", default="scan_results.csv")
args = parser.parse_args()
ip = args.host
ports = args.ports
output_file = args.output
print(f"Scanning IP: {ip}")
print(f"Scanning ports: {ports}")
sys.stdout.write("Scanning ")
sys.stdout.flush()
host_infos = scan_host(ip, ports)
for host_info in host_infos:
output_to_csv(output_file, host_info)
print("\n\nScan results:")
for host_info in host_infos:
print(f"IP: {host_info['ip']}")
print(f"OS: {host_info['os']}")
print(f"Port: {host_info['port']}")
print(f"Name: {host_info['name']}")
print(f"Product: {host_info['product']}")
print(f"Version: {host_info['version']}\n")
if __name__ == "__main__":
main()