Python實(shí)現(xiàn)端口檢測的方法

更新時(shí)間：2018年07月24日 10:30:20 作者：KaliArch

這篇文章主要介紹了Python實(shí)現(xiàn)端口檢測的方法,非常不錯(cuò)，具有一定的參考借鑒價(jià)值，需要的朋友可以參考下

一、背景：

在平時(shí)工作中有遇到端口檢測，查看服務(wù)端特定端口是否對外開放，常用nmap，tcping，telnet等，同時(shí)也可以利用站長工具等web掃描端口等。

但是在使用站長工具發(fā)現(xiàn)：

每次只能輸入一個(gè)檢測的地址；
雖然可以輸入多個(gè)端口，但是不能指定一個(gè)端口范圍來進(jìn)行批量檢測；
沒有批量任務(wù)記錄日志等；

因避免由于局域網(wǎng)檢測發(fā)起端網(wǎng)絡(luò)限制而導(dǎo)致的端口檢測異常，未使用python-nmap

想通過調(diào)用站長工具，實(shí)現(xiàn)

單次可多個(gè)地址或域名檢測
單詞可指定端口范圍，批量檢測
記錄日志

二、代碼：

2.1 結(jié)構(gòu)

2.2 代碼

github地址

部分代碼

#配置文件
#端口檢測配置
[port_check_info]
#檢測ip地址或域名
#address = baidu.com
#address = 127.0.0.1
address = www.anchnet.com,www.51cto.com,www.baidu.com
#檢查的端口，如多個(gè)端口使用,隔開,端口范圍使用'-'
#ports = 80,8080....
ports = 20-25,80,443,1433,1521,3306,3389,6379,8080,27017
#日志配置
[loginfo]
#日志目錄
logdir_name = logdir
#日志文件名稱
logfile_name = check_port.log
  def _get_body(self):
    """
    獲取address和port
    :return: list
    """
    address_list = self.address_list.split(',')
    port_list = self.port_list.split(',')
    # 處理端口范圍，返回range
    range_flag = False
    port_range = None
    content_List_range = []
    for port in port_list:
      if '-' in port:
        range_flag = True
        port_range = range(int(port.split('-')[0]),int(port.split('-')[1])+1)
        port_list.remove(port)
    # 處理總體list
    for add in address_list:
      if range_flag:
        for port in port_range:
          content_List_range.append(add + ':' + str(port))
    # 合并range和普通list
    content_List = [ add+':'+port for add in address_list for port in port_list ]
    content_List_range.extend(content_List)
    return content_List_range
  def run(self):
    """
    進(jìn)行端口檢測
    :return:
    """
    for content in self._get_body():
      content_list = content.split(':')
      body = {
        'host': content_list[0],
        'port': content_list[1],
        'encode': 'tlCHS1u3IgF4sC57m6KOP3Oaj1Y1kfLq'
      }
      try:
        response = requests.post(url=self.url,data=body,headers=self.headers)
        port_status = re.findall("msg:'(.*?)'", response.text)
        if len(port_status) > 0:
          # print('%s,port status is:%s' % (content, port_status))
          self.logoper.info('%s,port status is:%s' % (content, port_status))
        else:
          self.logoper.info('%s,port status is:%s' % (content, port_status))
          # print('Occer error！請輸入正確的地址和端口')
      except Exception as e:
        print(e)

三、測試：

3.1 查看檢測結(jié)果