1. 程式人生 > >Python實現端口檢測

Python實現端口檢測

ado 掃描 獲取 map [1] com 指定 RoCE 站長

一、背景:

在平時工作中有遇到端口檢測,查看服務端特定端口是否對外開放,常用nmap,tcping,telnet等,同時也可以利用站長工具等web掃描端口等。
但是在使用站長工具發現:

  • 每次只能輸入一個檢測的地址;
  • 雖然可以輸入多個端口,但是不能指定一個端口範圍來進行批量檢測;
  • 沒有批量任務記錄日誌等;
    技術分享圖片

因避免由於局域網檢測發起端網絡限制而導致的端口檢測異常,未使用python-nmap
想通過調用站長工具,實現

  • 單次可多個地址或域名檢測
  • 單詞可指定端口範圍,批量檢測
  • 記錄日誌

二、代碼:

2.1 結構

技術分享圖片

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

#檢查的端口,如多個端口使用,隔開,端口範圍使用‘-‘
#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):
        """
        進行端口檢測
        :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 查看檢測結果

技術分享圖片
此處可以查看51cto的8080端口也是開放的。

3.2 查看日誌

技術分享圖片

四、改進:

  • 後期可以添加異步多進程等來提升效率
  • 可以對比多個站點檢測結果,使結果更準確
  • 整合nmap內網也可檢測

Python實現端口檢測