今天添加端口探测功能,主要实现方式是通过nmap模块调用,扫描1-65535端口。上一篇中已经将UP的PC机全部获取到,这里直接从已知在线的PC中进行端口扫描就可以了,会节省很多时间。
代码如下,还是非常简单的:
import nmap
import time
import csv
import os
class Port_Scan(object):
def __init__(self):
self.rootpath = 'C:/cunhuo/'
self.dirpath = 'C:/portScan/'
self.savepath = 'C:/portScan/result.csv'
self.csv_write = []
def get_ip(self, file):
ipList = []
with open(file, 'r') as f:
for ip in f.readlines()[1:]:
ipList.append(ip.strip()[:-3])
return ipList
def scan(self, ip):
if not os.path.exists(self.dirpath):
os.mkdir(self.dirpath)
nm = nmap.PortScanner()
try:
nm.scan(ip, arguments='1-65535')
print '[*]Host:%s' % ip
fo