python脚本批量检测端口连通性

傷城~ 2023-05-29 13:06 92阅读 0赞

### python 脚本批量检测端口连通性 ###

*  前言
 *  脚本
 *  命令行执行
 *  返回结果

# 前言 #

测试连通性的工具有很多，比如telnet，nc，netcat等等。但telnet检测一些超时的连接时会变成交互式，影响批量脚本书写，而nc和netcat等工具又由于一些限制，并不是每一台服务器上都有这些工具，因此基于python的socket模块写了一个检测的脚本

# 脚本 #

*  在list\_str按照例子中的写法添加ip和port即可(python2.7)
 *  socket.setdefaulttimeout(2) 为设置超时时间

import socket
    
    list_str = ''' 10.78.199.62 9101 10.78.199.62 9102 '''
    
    OK_list = []
    Timeout_list = []
    DOWN_list = []
    ip_list = list(list_str.split())
    
    
    # socket try connect
    def IsOpen(ip, port):
        socket.setdefaulttimeout(2)
        s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        try:
            s.connect((ip, port))
            s.shutdown(2)
            OK_list.append([ip, port])
            return True
        except socket.timeout:
            Timeout_list.append([ip, port])
            return False
        except:
            # print '%s %d is DOWN' % (ip, port)
            DOWN_list.append([ip, port])
            return False
    
    
    if __name__ == '__main__':
        for _ in range(0, len(ip_list), 2):
          IsOpen(ip_list[_], int(ip_list[_ + 1]))
        lists={ 'ok': OK_list, 'timeout': Timeout_list, 'donw': DOWN_list}
        for status, _ in lists.items():
          print status.center(40, '-')
          # print '-' * 30
          # print 'status: %s is :' % status
          if not _:
            print 'null'
            continue
          for ip, port in _:
            print ip, port

# 命令行执行 #

只需脚本内容放入其中，然后复制到命令行中回车即可

cat << eof | python
    # 脚本内容
    eof

# 返回结果 #

返回3个数据，ok为连接没有问题，down为拒绝连接，timeout为超时  
![在这里插入图片描述][20191106124017565.png]

[20191106124017565.png]: https://img-blog.csdnimg.cn/20191106124017565.png