import sys
import nmap

scan_row = []
input_data = input('Please input hosts and port: ')
#scan_row以空格分隔
scan_row = input_data.split(' ')

if len(scan_row) != 2:
 print("Input errors, example \"192.168.209.0/24 80,443,22 \"")
 sys.exit(0)

#接收用戶輸入的主機(jī)
hosts = scan_row[0]
#接收用戶收入的端口
port = scan_row[1]

try:
 #創(chuàng)建端口掃描對象
 nm = nmap.PortScanner()
except nmap.PortScannerError:
 print('Nmap not found', sys.exc_info()[0])
 sys.exit(0)
except Exception as e:
 print("Unexpected error:", sys.exc_info()[0])
 print(str(e))
 sys.exit(0)

try:
 #調(diào)用掃描方法，參數(shù)指定掃描主機(jī)hosts，nmap掃描命令行參數(shù)arguments
 nm.scan(hosts=hosts, arguments=' -v -sS -p ' + port)
except Exception as e:
 print("Scan error:" + str(e))

for host in nm.all_hosts():
 print('---------------------------------------------------------------------')
 #輸出主機(jī)及主機(jī)名
 print('Host : %s (%s)' % (host, nm[host].hostname()))
 #輸出主機(jī)狀態(tài)，如up、down
 print('State : %s' % nm[host].state())
 #遍歷掃描協(xié)議，tcp、udp
 for proto in nm[host].all_protocols():
  print('--------------')
  #輸出協(xié)議名
  print('Protocol : %s' % proto)

  #獲取協(xié)議的所有掃描端口
  lport = list(nm[host][proto].keys())
  #端口列表排序
  lport.sort()
  #遍歷端口輸出端口與狀態(tài)
  for port in lport:
   print('port %s\tstate : %s' % (port, nm[host][proto][port]['state']))