python端口掃描

前段時間寫了個端口掃描的python腳本,發現如今忘記的差很少了,趕忙記錄下來吧,雖然是併發的,可是在實際的使用場景中,若是掃描的機器過多,依舊有性能上的問題,好在我須要掃描的機子就那麼幾臺.腳本寫的生爛...

下面附源碼:

#!/usr/bin/env python
#-*- coding:utf-8 -*-
import nmap
import threading
import smtplib
import string
from email.mime.text import MIMEText
from email.header import Header
import sys
#防止linux下中文出問題
reload(sys)
sys.setdefaultencoding('utf-8')

#設置白名單接口
PortList=[22,80,xxx,xxx]

#設置收件人列表:多個須要用逗號隔開,如:['shinianyijian@qq.com','4153@qq.com','http@163.com']
MailList=['shinianyijian@qq.com','4153@qq.com','http@163.com']

#定義個全局變量以接受scan1函數中的變量
result = ''

#總共執行的主機數,從ip.txt裏面讀取主機,固然有的需求多是從數據庫中,這裏能夠自行修改
num=file('ip.txt','r')
HostNum=len(num.readlines())
num.close()

#定義發郵件函數
def SendMail(sender,receiver,subject,content,smtpserver,smtpuser,smtppass):
  msg = MIMEText(content,'html','utf-8')
  msg['Subject'] = Header(subject, 'utf-8')
  msg['From'] = '<%s>' % sender 
  msg['To'] = ";".join(receiver)
  try: 
    smtp = smtplib.SMTP()
    smtp.connect(smtpserver)
    smtp.login(smtpuser, smtppass)
    smtp.sendmail(sender, receiver, msg.as_string())
    smtp.quit()
  except Exception,e:
      print e
#定義掃面端口函數,默認端口是1-65535
def scan1(ip,port='1-65535'):
  nm = nmap.PortScanner()
  nm.scan(ip,port)
  global result
  result = result + "<h2>ip地址: %s</h2><hr>" %(ip)
  for proto in nm[ip].all_protocols():
    lport = nm[ip][proto].keys()
   # print lport
    lport.sort()
    for port in lport:
      if port in PortList:
        info = '<strong><font color=green>Info:正常開放端口:</font></strong>&nbsp;&nbsp;'
        portinfo='%s<strong> port </strong>: %s&nbsp;&nbsp;<strong>state</strong>: %s &nbsp;&nbsp;<strong>product</strong>: %s<br>' %(info,port,nm[ip][proto][port]['state'],nm[ip]['tcp'][port]['name'])
        result = result + portinfo
      else:
        info = '<strong><font color=red>Info:非預期端口</font></strong>&nbsp;&nbsp;'
        portinfo='%s<strong> port </strong>: %s&nbsp;&nbsp;<strong>state</strong>: %s &nbsp;&nbsp;<strong>product</strong>: %s<br>' %(info,port,nm[ip][proto][port]['state'],nm[ip]['tcp'][port]['name'])
        result = result + portinfo 
  return result
#定義多線程掃描
def main():
  threads=[]
  #使用的ip列表文件
  f=file('./ip.txt','rU')
  nloops = len(f.readlines())
  f.seek(0)
  global ErrorNum
  global CurNum
  global ErrHostList
  ErrorNum=0
  CurNum=0
  ErrHostList = ''
  for i in f.readlines():
    t=threading.Thread(target=scan1,args=(i.strip(),))
    threads.append(t)
  for i in range(nloops):
    try:
      threads[i].start()
    except Exception,e:
      ErrorNum=ErrorNum+1
      ErrHostList = ErrHostList + '%s' % num[ip]
    else:
      CurNum=CurNum+1
  for i in range(nloops):
    threads[i].join()
  f.close()
if __name__ == "__main__":
  main()
  sender = 'lidefu@lingyuwangluo.com'
  receiver = MailList
  #郵件主題,如:xxx端口掃描
  subject = 'xxx端口掃描'
  smtpserver = 'smtp.xxxx.com'
  smtpuser = 'xxxx@xxxx.com'
  smtppass = 'xxxx'
  #mailcontent = '<h1>共執行%s 臺主機<h1><br>' % HostNum + '<h2>成功%s臺<h2><br>' % CurNum + '<h2>失敗%s臺<h2><br>' % ErrorNum + '<h2>失敗的ip爲:%s<h2><br>' % ErrHostList + result 
  mailcontent = '<h1>共執行%s 臺主機</h1>' % HostNum + '<h2>成功%s臺</h2>' % CurNum + '<h2>失敗%s臺</h2>' % ErrorNum + '<h2>失敗的ip爲:%s</h2><hr>' % ErrHostList + result 
  #print mailcontent
  SendMail(sender,receiver,subject,mailcontent,smtpserver,smtpuser,smtppass)
  #print result