使用python实现端口连通性探测

背景

公司需要添加新的网段,所以需要对业务所有的机器进行端口探测,看是否需要放开iptables以及安全组,以下是脚本,大家修改后就可以使用

python 复制代码
from datetime import datetime
import socket
import logging
def log_format():
    # 输出到console
    ch_formatter = logging.Formatter('%(asctime)s - %(filename)s[line:%(lineno)d] - %(levelname)s: %(message)s')
    ch = logging.StreamHandler()
    ch.setLevel(logging.INFO)  # 指定被处理的信息级别为最低级INFO,低于level级别的信息将被忽略
    ch.setFormatter(ch_formatter)
    # 输出到file
    formatter = logging.Formatter('%(asctime)s - %(filename)s[line:%(lineno)d] - %(levelname)s: %(message)s')
    fh = logging.FileHandler(filename="./output_%s.log" % (datetime.now().strftime('%Y%m%d')), mode='w',
                             encoding='utf-8')  # 不拆分日志文件,a指追加模式,w为覆盖模式
    fh.setLevel(logging.INFO)
    fh.setFormatter(formatter)

    logger = logging.getLogger()
    logger.setLevel(logging.INFO)
    logger.addHandler(ch)
    logger.addHandler(fh)
    return logger



def probe(ip, port=22, timeout=5):
    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    sock.settimeout(timeout)
    try:
        sock.connect((ip, port))
        return True
    except (socket.timeout, ConnectionRefusedError):
        return False
    finally:
        sock.close()

now = datetime.now().strftime('%Y-%m-%d_%H-%M-%S')
result_file_name = f'/root/jarvisyqliu/result_{now}.txt'
log_format()
with open('/root/jarvisyqliu/ip.txt', 'r') as ip_file:
    for line in ip_file:
        ip = line.strip()
        logging.info(f'scaning...{ip}')
        with open(result_file_name, 'a') as result_file:
            if probe(ip):
                result_file.write(f'{ip} ok\n')
            else:
                result_file.write(f'{ip} error\n')
相关推荐
张槊哲21 分钟前
如何打包python程序为可执行文件
开发语言·python
张3蜂25 分钟前
Jupyter Notebook 全面介绍:从原理到实战部署
ide·python·jupyter
AI_RSER1 小时前
基于 Google Earth Engine (GEE) 的土地利用变化监测
python·随机森林·分类·土地利用·gee·遥感影像·landsat
uncle_ll1 小时前
FastAPI 零基础入门指南:10 分钟搭建高性能 API
后端·python·微服务·api·fastapi
程序猿与金融与科技1 小时前
PyQt6基础_pyqtgraph_折线图with缩放调节
python·pyqt6·pyqtgraph
guest_8811 小时前
用 Python 打造打篮球字符动画!控制台彩色炫酷输出,抖音搞怪视频灵感还原
开发语言·python·numpy·pillow
没有不重的名么1 小时前
在Pytorch中使用Tensorboard可视化训练过程
人工智能·pytorch·python
程序媛徐师姐1 小时前
Python基于Django的全国二手房可视化分析系统【附源码】
python·django·全国二手房可视化分析系统·二手房可视化分析系统·二手房房价可视化·二手房可视化·二手房房价可视化分析系统
东木月2 小时前
Python解析地址中省市区街道
开发语言·python
意.远2 小时前
PyTorch卷积层填充(Padding)与步幅(Stride)详解及代码示例
人工智能·pytorch·python·深度学习