python 实现IP开放端口探测与是否部署WEB服务探测

由于工作中需要对多个IP段进行统计，并判断是否部署WEB服务，所以有了写这个脚本的想法。在中间测试脚本的时候，遇到了个坑，我在自己的电脑上运行脚本的时候，没有任何问题。但是当我放到我的服务器上去运行的时候，python在请求了几个站点之后就开始报错了。

ConnectionResetError: [Errno 104] Connection reset by peer

经过度娘，谷歌的一阵搜索，发现了问题，是由于我大量频繁的请求，导致服务器关闭了连接。由于技术有限，在这里我的解决办法就是在对URL进行请求的时候，添加个异常捕获(try except),成功解决了该问题！

以下是我昨天临时写的脚本源码，大致意思就是，获取文件中的IP列表，然后使用nmap对80,8080,443进行探测，并对开放这些端口的IP进行拼接并根据返回code判断是否有WEB服务。最后将结果写入到文件中。

import sys
import os
import nmap
import requests
import logging
import threading
from multiprocessing import Pool as ThreadPool
import time
from openpyxl import load_workbook,Workbook

th = 20 
code = [200,403,404]
os.mknod('AAA.txt')


def GetIp(site):
	url_list = []
	print ('[* ]**************************GET-IP-SITE**************************[ *]\n')
	ip_file = open(site,'r')
	lines = ip_file.readlines()
	print (lines)
	for ip in lines:
		ip = ip.rstrip('\r\n')
		url_list.append(ip)
	return url_list

def scan(url_list):
	print ('[* ]' + url_list + ' **************************Join-Thread**************************[ *]\n')
	ip_port = '80,8080,443'
	active_list = []
	nm = nmap.PortScanner()
	ret = nm.scan(url_list,ip_port)
	#print (str(ret['scan']))
	if ret['scan']:
		print ('[* ]' + url_list + ' **************************Runing**************************[ *]\n')
		if ret['scan'][url_list]['tcp'][80]['state'] == 'open':
			url = 'http://'+url_list
			#print ('http://'+url_list)
			active_list.append(url)
		elif ret['scan'][url_list]['tcp'][8080]['state'] == 'open':
			url1 = 'http://'+url_list+':8080'
			#print ('http://'+url_list+':8080')
			active_list.append(url1)
		elif ret['scan'][url_list]['tcp'][443]['state'] == 'open':
			url2 = 'https://'+url_list
			#print ('https://'+url_list)
			active_list.append(url2)
		else:
			print('[* ]'+'*******jump********')
	return active_list
def active_ip(active_list):
	if active_list:
		print ('[* ]' + '************' +'up'+ str(active_list) +'WEB-up-or-die'+ '************[ *]\n')
		logging.captureWarnings(True)
		try:
			response = requests.get(active_list[0],headers={'Host':'test.com'},verify=False,timeout=10)
			if response.status_code in code:
				print ('[* ]=========' + active_list[0] + ' =======up'+' code ' + str(response.status_code))
				info = str(active_list[0]) + ' ' + str(response.status_code) + '\n'
				return info
		except requests.RequestException as e:
			pass

def savefile(url_info):
	print ('=================================>>>>>>>>'+str(url_info))
	f = open('AAA.txt','r+')
	for i in url_info:
		if i:
			f.write(str(i))
	f.close()
	
if __name__ == '__main__':
	site = sys.argv[1]
	print ('[* ]******************************************GoGoGoGoing******************************************[ *]\n')
	start = time.time()
	pool = ThreadPool(th)
	url_list = GetIp(site)
	active_list1 = pool.map(scan,url_list)
	print(str(active_list1))
	url_info = pool.map(active_ip,active_list1)
	savefile(url_info)
	pool.close()
	pool.join()
	finish = time.time()
	times = finish - start
	print ('time:'+ str(times))
	

其实我本人基本上很少写python，也没有去系统的学习过，写的代码也是完全靠想象来写的，hh，所以我这个代码可能会有显得有点点糟糕。= =！。